{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.19859395480001588,
  "global_step": 10000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.0,
      "learning_rate": 1.984126984126984e-06,
      "loss": 11.0771,
      "theoretical_loss": 20.841282884190854,
      "tokens_seen": 65536
    },
    {
      "epoch": 0.0,
      "learning_rate": 3.968253968253968e-06,
      "loss": 11.0714,
      "theoretical_loss": 17.594666186972347,
      "tokens_seen": 131072
    },
    {
      "epoch": 0.0,
      "learning_rate": 5.9523809523809525e-06,
      "loss": 10.9477,
      "theoretical_loss": 15.967942175480273,
      "tokens_seen": 196608
    },
    {
      "epoch": 0.0,
      "learning_rate": 7.936507936507936e-06,
      "loss": 10.6658,
      "theoretical_loss": 14.920781838632275,
      "tokens_seen": 262144
    },
    {
      "epoch": 0.0,
      "learning_rate": 9.92063492063492e-06,
      "loss": 10.3389,
      "theoretical_loss": 14.16468202040668,
      "tokens_seen": 327680
    },
    {
      "epoch": 0.0,
      "learning_rate": 1.1904761904761905e-05,
      "loss": 10.0572,
      "theoretical_loss": 13.581026555193928,
      "tokens_seen": 393216
    },
    {
      "epoch": 0.0,
      "learning_rate": 1.3888888888888888e-05,
      "loss": 9.8012,
      "theoretical_loss": 13.110274084044894,
      "tokens_seen": 458752
    },
    {
      "epoch": 0.0,
      "learning_rate": 1.5873015873015872e-05,
      "loss": 9.6269,
      "theoretical_loss": 12.718594708127029,
      "tokens_seen": 524288
    },
    {
      "epoch": 0.0,
      "learning_rate": 1.7857142857142855e-05,
      "loss": 9.5127,
      "theoretical_loss": 12.385057546517329,
      "tokens_seen": 589824
    },
    {
      "epoch": 0.0,
      "learning_rate": 1.984126984126984e-05,
      "loss": 9.3822,
      "theoretical_loss": 12.095877689678783,
      "tokens_seen": 655360
    },
    {
      "epoch": 0.0,
      "learning_rate": 2.1825396825396824e-05,
      "loss": 9.3134,
      "theoretical_loss": 11.841531313745381,
      "tokens_seen": 720896
    },
    {
      "epoch": 0.0,
      "learning_rate": 2.380952380952381e-05,
      "loss": 9.19,
      "theoretical_loss": 11.615184291350435,
      "tokens_seen": 786432
    },
    {
      "epoch": 0.0,
      "learning_rate": 2.5793650793650793e-05,
      "loss": 9.1376,
      "theoretical_loss": 11.411779222830493,
      "tokens_seen": 851968
    },
    {
      "epoch": 0.0,
      "learning_rate": 2.7777777777777776e-05,
      "loss": 9.0561,
      "theoretical_loss": 11.227476784755577,
      "tokens_seen": 917504
    },
    {
      "epoch": 0.0,
      "learning_rate": 2.9761904761904762e-05,
      "loss": 8.9674,
      "theoretical_loss": 11.059299000621618,
      "tokens_seen": 983040
    },
    {
      "epoch": 0.0,
      "learning_rate": 3.1746031746031745e-05,
      "loss": 8.9181,
      "theoretical_loss": 10.904893169100655,
      "tokens_seen": 1048576
    },
    {
      "epoch": 0.0,
      "learning_rate": 3.373015873015873e-05,
      "loss": 8.9953,
      "theoretical_loss": 10.76237082270643,
      "tokens_seen": 1114112
    },
    {
      "epoch": 0.0,
      "learning_rate": 3.571428571428571e-05,
      "loss": 8.7793,
      "theoretical_loss": 10.630194958873984,
      "tokens_seen": 1179648
    },
    {
      "epoch": 0.0,
      "learning_rate": 3.76984126984127e-05,
      "loss": 8.7566,
      "theoretical_loss": 10.507099255000346,
      "tokens_seen": 1245184
    },
    {
      "epoch": 0.0,
      "learning_rate": 3.968253968253968e-05,
      "loss": 8.7115,
      "theoretical_loss": 10.392029026407034,
      "tokens_seen": 1310720
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.1666666666666665e-05,
      "loss": 8.487,
      "theoretical_loss": 10.28409730354045,
      "tokens_seen": 1376256
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.365079365079365e-05,
      "loss": 8.5783,
      "theoretical_loss": 10.182551635913724,
      "tokens_seen": 1441792
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.563492063492063e-05,
      "loss": 8.5185,
      "theoretical_loss": 10.086748644376303,
      "tokens_seen": 1507328
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.761904761904762e-05,
      "loss": 8.4859,
      "theoretical_loss": 9.996134261483984,
      "tokens_seen": 1572864
    },
    {
      "epoch": 0.0,
      "learning_rate": 4.96031746031746e-05,
      "loss": 8.3094,
      "theoretical_loss": 9.910228209036813,
      "tokens_seen": 1638400
    },
    {
      "epoch": 0.0,
      "learning_rate": 5.1587301587301586e-05,
      "loss": 8.4173,
      "theoretical_loss": 9.828611674184266,
      "tokens_seen": 1703936
    },
    {
      "epoch": 0.0,
      "learning_rate": 5.357142857142857e-05,
      "loss": 8.3191,
      "theoretical_loss": 9.750917429551151,
      "tokens_seen": 1769472
    },
    {
      "epoch": 0.0,
      "learning_rate": 5.555555555555555e-05,
      "loss": 8.3262,
      "theoretical_loss": 9.67682184172525,
      "tokens_seen": 1835008
    },
    {
      "epoch": 0.0,
      "learning_rate": 5.753968253968254e-05,
      "loss": 8.1858,
      "theoretical_loss": 9.606038353789344,
      "tokens_seen": 1900544
    },
    {
      "epoch": 0.0,
      "learning_rate": 5.9523809523809524e-05,
      "loss": 8.1585,
      "theoretical_loss": 9.538312129408556,
      "tokens_seen": 1966080
    },
    {
      "epoch": 0.0,
      "learning_rate": 6.15079365079365e-05,
      "loss": 8.1146,
      "theoretical_loss": 9.47341562027564,
      "tokens_seen": 2031616
    },
    {
      "epoch": 0.0,
      "learning_rate": 6.349206349206349e-05,
      "loss": 8.0534,
      "theoretical_loss": 9.41114487355416,
      "tokens_seen": 2097152
    },
    {
      "epoch": 0.0,
      "learning_rate": 6.547619047619048e-05,
      "loss": 7.962,
      "theoretical_loss": 9.351316436877685,
      "tokens_seen": 2162688
    },
    {
      "epoch": 0.0,
      "learning_rate": 6.746031746031745e-05,
      "loss": 7.8543,
      "theoretical_loss": 9.29376474930398,
      "tokens_seen": 2228224
    },
    {
      "epoch": 0.0,
      "learning_rate": 6.944444444444444e-05,
      "loss": 8.0138,
      "theoretical_loss": 9.238339930088461,
      "tokens_seen": 2293760
    },
    {
      "epoch": 0.0,
      "learning_rate": 7.142857142857142e-05,
      "loss": 7.7162,
      "theoretical_loss": 9.184905895151996,
      "tokens_seen": 2359296
    },
    {
      "epoch": 0.0,
      "learning_rate": 7.341269841269842e-05,
      "loss": 7.7563,
      "theoretical_loss": 9.133338745057056,
      "tokens_seen": 2424832
    },
    {
      "epoch": 0.0,
      "learning_rate": 7.53968253968254e-05,
      "loss": 7.5792,
      "theoretical_loss": 9.083525379177438,
      "tokens_seen": 2490368
    },
    {
      "epoch": 0.0,
      "learning_rate": 7.738095238095239e-05,
      "loss": 7.591,
      "theoretical_loss": 9.035362299287534,
      "tokens_seen": 2555904
    },
    {
      "epoch": 0.0,
      "learning_rate": 7.936507936507937e-05,
      "loss": 7.6518,
      "theoretical_loss": 8.988754572553061,
      "tokens_seen": 2621440
    },
    {
      "epoch": 0.0,
      "learning_rate": 8.134920634920635e-05,
      "loss": 7.4322,
      "theoretical_loss": 8.943614929284315,
      "tokens_seen": 2686976
    },
    {
      "epoch": 0.0,
      "learning_rate": 8.333333333333333e-05,
      "loss": 7.4551,
      "theoretical_loss": 8.899862975121929,
      "tokens_seen": 2752512
    },
    {
      "epoch": 0.0,
      "learning_rate": 8.531746031746032e-05,
      "loss": 7.24,
      "theoretical_loss": 8.857424500797205,
      "tokens_seen": 2818048
    },
    {
      "epoch": 0.0,
      "learning_rate": 8.73015873015873e-05,
      "loss": 7.2072,
      "theoretical_loss": 8.816230875422118,
      "tokens_seen": 2883584
    },
    {
      "epoch": 0.0,
      "learning_rate": 8.928571428571429e-05,
      "loss": 7.3823,
      "theoretical_loss": 8.776218511555161,
      "tokens_seen": 2949120
    },
    {
      "epoch": 0.0,
      "learning_rate": 9.126984126984126e-05,
      "loss": 7.1742,
      "theoretical_loss": 8.737328392164539,
      "tokens_seen": 3014656
    },
    {
      "epoch": 0.0,
      "learning_rate": 9.325396825396825e-05,
      "loss": 7.1599,
      "theoretical_loss": 8.699505651152553,
      "tokens_seen": 3080192
    },
    {
      "epoch": 0.0,
      "learning_rate": 9.523809523809524e-05,
      "loss": 7.1427,
      "theoretical_loss": 8.66269920037918,
      "tokens_seen": 3145728
    },
    {
      "epoch": 0.0,
      "learning_rate": 9.722222222222223e-05,
      "loss": 7.0218,
      "theoretical_loss": 8.62686139718003,
      "tokens_seen": 3211264
    },
    {
      "epoch": 0.0,
      "objective/train/advantage_avg": 0.49531280994415283,
      "objective/train/docs_used": 8371,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 6.728449821472168,
      "objective/train/original_loss": 6.728449821472168,
      "objective/train/theoretical_loss": 8.591947747254773,
      "objective/train/tokens_used": 23736800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24548619985580444,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507793426513672,
      "objective/train/weighted_lm_loss": 7.069992542266846,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0216598510742188,
      "theoretical_loss": 8.591947747254773,
      "tokens_seen": 3276800
    },
    {
      "epoch": 0.0,
      "learning_rate": 9.92063492063492e-05,
      "loss": 6.9774,
      "theoretical_loss": 8.591947747254773,
      "tokens_seen": 3276800
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001011904761904762,
      "loss": 6.8521,
      "theoretical_loss": 8.55791663853882,
      "tokens_seen": 3342336
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00010317460317460317,
      "loss": 6.7841,
      "theoretical_loss": 8.524729102289708,
      "tokens_seen": 3407872
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00010515873015873016,
      "loss": 6.8152,
      "theoretical_loss": 8.492348598140893,
      "tokens_seen": 3473408
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00010714285714285714,
      "loss": 6.7547,
      "theoretical_loss": 8.460740820316484,
      "tokens_seen": 3538944
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00010912698412698413,
      "loss": 6.7581,
      "theoretical_loss": 8.429873522574422,
      "tokens_seen": 3604480
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001111111111111111,
      "loss": 6.7221,
      "theoretical_loss": 8.399716359763914,
      "tokens_seen": 3670016
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00011309523809523809,
      "loss": 6.6347,
      "theoretical_loss": 8.370240744154781,
      "tokens_seen": 3735552
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00011507936507936508,
      "loss": 6.6158,
      "theoretical_loss": 8.341419714929035,
      "tokens_seen": 3801088
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00011706349206349207,
      "loss": 6.4159,
      "theoretical_loss": 8.313227819424887,
      "tokens_seen": 3866624
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00011904761904761905,
      "loss": 6.4367,
      "theoretical_loss": 8.285641004895568,
      "tokens_seen": 3932160
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00012103174603174604,
      "loss": 6.5093,
      "theoretical_loss": 8.25863651969382,
      "tokens_seen": 3997696
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000123015873015873,
      "loss": 6.3352,
      "theoretical_loss": 8.232192822921675,
      "tokens_seen": 4063232
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000125,
      "loss": 6.3951,
      "theoretical_loss": 8.206289501696752,
      "tokens_seen": 4128768
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00012698412698412698,
      "loss": 6.4355,
      "theoretical_loss": 8.180907195283321,
      "tokens_seen": 4194304
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00012896825396825398,
      "loss": 6.2604,
      "theoretical_loss": 8.156027525421205,
      "tokens_seen": 4259840
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00013095238095238096,
      "loss": 6.2779,
      "theoretical_loss": 8.131633032259415,
      "tokens_seen": 4325376
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00013293650793650793,
      "loss": 6.3027,
      "theoretical_loss": 8.107707115366345,
      "tokens_seen": 4390912
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001349206349206349,
      "loss": 6.2102,
      "theoretical_loss": 8.084233979345122,
      "tokens_seen": 4456448
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001369047619047619,
      "loss": 6.1439,
      "theoretical_loss": 8.061198583632713,
      "tokens_seen": 4521984
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001388888888888889,
      "loss": 6.1627,
      "theoretical_loss": 8.03858659610554,
      "tokens_seen": 4587520
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00014087301587301586,
      "loss": 5.9802,
      "theoretical_loss": 8.016384350153084,
      "tokens_seen": 4653056
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00014285714285714284,
      "loss": 6.1413,
      "theoretical_loss": 7.9945788049155055,
      "tokens_seen": 4718592
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00014484126984126984,
      "loss": 6.2189,
      "theoretical_loss": 7.973157508411693,
      "tokens_seen": 4784128
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00014682539682539685,
      "loss": 6.0561,
      "theoretical_loss": 7.952108563311223,
      "tokens_seen": 4849664
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00014880952380952382,
      "loss": 6.1682,
      "theoretical_loss": 7.931420595127772,
      "tokens_seen": 4915200
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001507936507936508,
      "loss": 6.0328,
      "theoretical_loss": 7.911082722632908,
      "tokens_seen": 4980736
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001527777777777778,
      "loss": 5.9525,
      "theoretical_loss": 7.891084530308333,
      "tokens_seen": 5046272
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00015476190476190478,
      "loss": 6.0668,
      "theoretical_loss": 7.871416042671642,
      "tokens_seen": 5111808
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00015674603174603175,
      "loss": 6.054,
      "theoretical_loss": 7.8520677003260175,
      "tokens_seen": 5177344
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00015873015873015873,
      "loss": 5.8692,
      "theoretical_loss": 7.83303033759787,
      "tokens_seen": 5242880
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00016071428571428573,
      "loss": 5.9204,
      "theoretical_loss": 7.8142951616387775,
      "tokens_seen": 5308416
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001626984126984127,
      "loss": 5.8646,
      "theoretical_loss": 7.795853732879069,
      "tokens_seen": 5373952
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00016468253968253969,
      "loss": 6.0267,
      "theoretical_loss": 7.777697946730324,
      "tokens_seen": 5439488
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00016666666666666666,
      "loss": 6.0271,
      "theoretical_loss": 7.759820016443023,
      "tokens_seen": 5505024
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00016865079365079366,
      "loss": 5.9168,
      "theoretical_loss": 7.742212457033625,
      "tokens_seen": 5570560
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00017063492063492064,
      "loss": 5.8456,
      "theoretical_loss": 7.724868070202639,
      "tokens_seen": 5636096
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00017261904761904762,
      "loss": 5.7787,
      "theoretical_loss": 7.707779930171886,
      "tokens_seen": 5701632
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001746031746031746,
      "loss": 5.7974,
      "theoretical_loss": 7.690941370375033,
      "tokens_seen": 5767168
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001765873015873016,
      "loss": 5.8043,
      "theoretical_loss": 7.674345970940984,
      "tokens_seen": 5832704
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00017857142857142857,
      "loss": 5.7159,
      "theoretical_loss": 7.6579875469145655,
      "tokens_seen": 5898240
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00018055555555555555,
      "loss": 5.8569,
      "theoretical_loss": 7.641860137163401,
      "tokens_seen": 5963776
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00018253968253968252,
      "loss": 5.8218,
      "theoretical_loss": 7.6259579939239845,
      "tokens_seen": 6029312
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00018452380952380953,
      "loss": 5.8325,
      "theoretical_loss": 7.610275572943593,
      "tokens_seen": 6094848
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001865079365079365,
      "loss": 5.7266,
      "theoretical_loss": 7.5948075241780755,
      "tokens_seen": 6160384
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001884920634920635,
      "loss": 5.7308,
      "theoretical_loss": 7.579548683008654,
      "tokens_seen": 6225920
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00019047619047619048,
      "loss": 5.8053,
      "theoretical_loss": 7.564494061943624,
      "tokens_seen": 6291456
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00019246031746031748,
      "loss": 5.7645,
      "theoretical_loss": 7.549638842773467,
      "tokens_seen": 6356992
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00019444444444444446,
      "loss": 5.6472,
      "theoretical_loss": 7.534978369150202,
      "tokens_seen": 6422528
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00019642857142857144,
      "loss": 5.6673,
      "theoretical_loss": 7.520508139563954,
      "tokens_seen": 6488064
    },
    {
      "epoch": 0.0,
      "objective/train/advantage_avg": 0.4908171594142914,
      "objective/train/docs_used": 9704,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 5.399485111236572,
      "objective/train/original_loss": 5.3994855880737305,
      "objective/train/theoretical_loss": 7.5062238006917354,
      "objective/train/tokens_used": 27013600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2412356734275818,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503079891204834,
      "objective/train/weighted_lm_loss": 5.670967102050781,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 7.5062238006917354,
      "tokens_seen": 6553600
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0001984126984126984,
      "loss": 5.7328,
      "theoretical_loss": 7.5062238006917354,
      "tokens_seen": 6553600
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00020039682539682542,
      "loss": 5.7643,
      "theoretical_loss": 7.492121141095167,
      "tokens_seen": 6619136
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002023809523809524,
      "loss": 5.5097,
      "theoretical_loss": 7.478196085245621,
      "tokens_seen": 6684672
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00020436507936507937,
      "loss": 5.6864,
      "theoretical_loss": 7.464444687856724,
      "tokens_seen": 6750208
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00020634920634920634,
      "loss": 5.7318,
      "theoretical_loss": 7.45086312850561,
      "tokens_seen": 6815744
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00020833333333333335,
      "loss": 5.4749,
      "theoretical_loss": 7.43744770652558,
      "tokens_seen": 6881280
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00021031746031746032,
      "loss": 5.5339,
      "theoretical_loss": 7.42419483615405,
      "tokens_seen": 6946816
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002123015873015873,
      "loss": 5.6535,
      "theoretical_loss": 7.4111010419207375,
      "tokens_seen": 7012352
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00021428571428571427,
      "loss": 5.7321,
      "theoretical_loss": 7.398162954262078,
      "tokens_seen": 7077888
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00021626984126984128,
      "loss": 5.5421,
      "theoretical_loss": 7.385377305348793,
      "tokens_seen": 7143424
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00021825396825396825,
      "loss": 5.5598,
      "theoretical_loss": 7.3727409251143925,
      "tokens_seen": 7208960
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00022023809523809523,
      "loss": 5.5974,
      "theoretical_loss": 7.360250737473224,
      "tokens_seen": 7274496
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002222222222222222,
      "loss": 5.5628,
      "theoretical_loss": 7.347903756717382,
      "tokens_seen": 7340032
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002242063492063492,
      "loss": 5.5724,
      "theoretical_loss": 7.335697084082529,
      "tokens_seen": 7405568
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00022619047619047618,
      "loss": 5.6969,
      "theoretical_loss": 7.323627904473269,
      "tokens_seen": 7471104
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00022817460317460316,
      "loss": 5.3875,
      "theoretical_loss": 7.311693483339357,
      "tokens_seen": 7536640
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00023015873015873016,
      "loss": 5.4919,
      "theoretical_loss": 7.299891163694537,
      "tokens_seen": 7602176
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00023214285714285717,
      "loss": 5.5971,
      "theoretical_loss": 7.288218363270328,
      "tokens_seen": 7667712
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00023412698412698414,
      "loss": 5.5551,
      "theoretical_loss": 7.276672571797575,
      "tokens_seen": 7733248
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00023611111111111112,
      "loss": 5.5232,
      "theoretical_loss": 7.265251348408967,
      "tokens_seen": 7798784
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002380952380952381,
      "loss": 5.6741,
      "theoretical_loss": 7.253952319156202,
      "tokens_seen": 7864320
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002400793650793651,
      "loss": 5.5124,
      "theoretical_loss": 7.242773174635825,
      "tokens_seen": 7929856
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00024206349206349207,
      "loss": 5.6728,
      "theoretical_loss": 7.2317116677181055,
      "tokens_seen": 7995392
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00024404761904761905,
      "loss": 5.3354,
      "theoretical_loss": 7.22076561137372,
      "tokens_seen": 8060928
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000246031746031746,
      "loss": 5.5085,
      "theoretical_loss": 7.2099328765932205,
      "tokens_seen": 8126464
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000248015873015873,
      "loss": 5.5368,
      "theoretical_loss": 7.1992113903946695,
      "tokens_seen": 8192000
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00025,
      "loss": 5.4071,
      "theoretical_loss": 7.188599133914977,
      "tokens_seen": 8257536
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000251984126984127,
      "loss": 5.4527,
      "theoretical_loss": 7.178094140580843,
      "tokens_seen": 8323072
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00025396825396825396,
      "loss": 5.36,
      "theoretical_loss": 7.167694494355343,
      "tokens_seen": 8388608
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00025595238095238096,
      "loss": 5.3949,
      "theoretical_loss": 7.157398328056499,
      "tokens_seen": 8454144
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00025793650793650796,
      "loss": 5.4218,
      "theoretical_loss": 7.147203821744321,
      "tokens_seen": 8519680
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00025992063492063497,
      "loss": 5.4023,
      "theoretical_loss": 7.1371092011730495,
      "tokens_seen": 8585216
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002619047619047619,
      "loss": 5.3711,
      "theoretical_loss": 7.127112736305475,
      "tokens_seen": 8650752
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002638888888888889,
      "loss": 5.2684,
      "theoretical_loss": 7.117212739886398,
      "tokens_seen": 8716288
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00026587301587301587,
      "loss": 5.2483,
      "theoretical_loss": 7.107407566072454,
      "tokens_seen": 8781824
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00026785714285714287,
      "loss": 5.5102,
      "theoretical_loss": 7.097695609115671,
      "tokens_seen": 8847360
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002698412698412698,
      "loss": 5.3072,
      "theoretical_loss": 7.0880753020982725,
      "tokens_seen": 8912896
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002718253968253968,
      "loss": 5.1466,
      "theoretical_loss": 7.0785451157163735,
      "tokens_seen": 8978432
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002738095238095238,
      "loss": 5.439,
      "theoretical_loss": 7.06910355711033,
      "tokens_seen": 9043968
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00027579365079365083,
      "loss": 5.2897,
      "theoretical_loss": 7.059749168739638,
      "tokens_seen": 9109504
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002777777777777778,
      "loss": 5.3691,
      "theoretical_loss": 7.050480527300383,
      "tokens_seen": 9175040
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002797619047619048,
      "loss": 5.3555,
      "theoretical_loss": 7.041296242683314,
      "tokens_seen": 9240576
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00028174603174603173,
      "loss": 5.3703,
      "theoretical_loss": 7.032194956970776,
      "tokens_seen": 9306112
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00028373015873015873,
      "loss": 5.3154,
      "theoretical_loss": 7.023175343470762,
      "tokens_seen": 9371648
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002857142857142857,
      "loss": 5.3311,
      "theoretical_loss": 7.014236105786485,
      "tokens_seen": 9437184
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002876984126984127,
      "loss": 5.2603,
      "theoretical_loss": 7.005375976919894,
      "tokens_seen": 9502720
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002896825396825397,
      "loss": 5.3003,
      "theoretical_loss": 6.99659371840772,
      "tokens_seen": 9568256
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002916666666666667,
      "loss": 5.3527,
      "theoretical_loss": 6.987888119488599,
      "tokens_seen": 9633792
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002936507936507937,
      "loss": 5.3101,
      "theoretical_loss": 6.979257996300014,
      "tokens_seen": 9699328
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00029563492063492064,
      "loss": 5.27,
      "theoretical_loss": 6.970702191103734,
      "tokens_seen": 9764864
    },
    {
      "epoch": 0.0,
      "objective/train/advantage_avg": 0.49107539653778076,
      "objective/train/docs_used": 11015,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 5.094058036804199,
      "objective/train/original_loss": 5.094058036804199,
      "objective/train/theoretical_loss": 6.962219571538605,
      "objective/train/tokens_used": 30290400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24188564717769623,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503370761871338,
      "objective/train/weighted_lm_loss": 5.350419521331787,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 6.962219571538605,
      "tokens_seen": 9830400
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00029761904761904765,
      "loss": 5.3296,
      "theoretical_loss": 6.962219571538605,
      "tokens_seen": 9830400
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0002996031746031746,
      "loss": 5.3081,
      "theoretical_loss": 6.953809029899522,
      "tokens_seen": 9895936
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003015873015873016,
      "loss": 5.2714,
      "theoretical_loss": 6.945469482441503,
      "tokens_seen": 9961472
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00030357142857142855,
      "loss": 5.2838,
      "theoretical_loss": 6.937199868707852,
      "tokens_seen": 10027008
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003055555555555556,
      "loss": 5.3265,
      "theoretical_loss": 6.928999150881393,
      "tokens_seen": 10092544
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00030753968253968255,
      "loss": 5.2493,
      "theoretical_loss": 6.920866313157865,
      "tokens_seen": 10158080
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00030952380952380956,
      "loss": 5.3069,
      "theoretical_loss": 6.912800361140576,
      "tokens_seen": 10223616
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003115079365079365,
      "loss": 5.4451,
      "theoretical_loss": 6.904800321255455,
      "tokens_seen": 10289152
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003134920634920635,
      "loss": 5.1631,
      "theoretical_loss": 6.896865240185702,
      "tokens_seen": 10354688
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00031547619047619046,
      "loss": 5.2823,
      "theoretical_loss": 6.888994184325256,
      "tokens_seen": 10420224
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00031746031746031746,
      "loss": 5.2868,
      "theoretical_loss": 6.881186239250335,
      "tokens_seen": 10485760
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003194444444444444,
      "loss": 5.2119,
      "theoretical_loss": 6.873440509208347,
      "tokens_seen": 10551296
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00032142857142857147,
      "loss": 5.414,
      "theoretical_loss": 6.8657561166234995,
      "tokens_seen": 10616832
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003234126984126984,
      "loss": 5.0445,
      "theoretical_loss": 6.858132201618454,
      "tokens_seen": 10682368
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003253968253968254,
      "loss": 5.1709,
      "theoretical_loss": 6.8505679215514235,
      "tokens_seen": 10747904
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00032738095238095237,
      "loss": 5.1904,
      "theoretical_loss": 6.8430624505681,
      "tokens_seen": 10813440
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00032936507936507937,
      "loss": 5.2185,
      "theoretical_loss": 6.835614979167886,
      "tokens_seen": 10878976
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003313492063492063,
      "loss": 5.0221,
      "theoretical_loss": 6.82822471378385,
      "tokens_seen": 10944512
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003333333333333333,
      "loss": 5.1341,
      "theoretical_loss": 6.8208908763759295,
      "tokens_seen": 11010048
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003353174603174603,
      "loss": 4.9061,
      "theoretical_loss": 6.813612704036853,
      "tokens_seen": 11075584
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00033730158730158733,
      "loss": 5.1869,
      "theoretical_loss": 6.806389448610343,
      "tokens_seen": 11141120
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00033928571428571433,
      "loss": 5.1124,
      "theoretical_loss": 6.799220376321128,
      "tokens_seen": 11206656
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003412698412698413,
      "loss": 5.2221,
      "theoretical_loss": 6.79210476741633,
      "tokens_seen": 11272192
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003432539682539683,
      "loss": 5.035,
      "theoretical_loss": 6.785041915817828,
      "tokens_seen": 11337728
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00034523809523809523,
      "loss": 5.1263,
      "theoretical_loss": 6.778031128785191,
      "tokens_seen": 11403264
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00034722222222222224,
      "loss": 5.2328,
      "theoretical_loss": 6.771071726588795,
      "tokens_seen": 11468800
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003492063492063492,
      "loss": 5.1011,
      "theoretical_loss": 6.76416304219278,
      "tokens_seen": 11534336
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003511904761904762,
      "loss": 5.0111,
      "theoretical_loss": 6.757304420947466,
      "tokens_seen": 11599872
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003531746031746032,
      "loss": 5.1246,
      "theoretical_loss": 6.750495220290931,
      "tokens_seen": 11665408
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003551587301587302,
      "loss": 5.0826,
      "theoretical_loss": 6.743734809459394,
      "tokens_seen": 11730944
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00035714285714285714,
      "loss": 4.9246,
      "theoretical_loss": 6.737022569206117,
      "tokens_seen": 11796480
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00035912698412698415,
      "loss": 4.9429,
      "theoretical_loss": 6.730357891528529,
      "tokens_seen": 11862016
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003611111111111111,
      "loss": 5.1395,
      "theoretical_loss": 6.723740179403264,
      "tokens_seen": 11927552
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003630952380952381,
      "loss": 5.1806,
      "theoretical_loss": 6.717168846528885,
      "tokens_seen": 11993088
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00036507936507936505,
      "loss": 5.0837,
      "theoretical_loss": 6.710643317075979,
      "tokens_seen": 12058624
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00036706349206349205,
      "loss": 5.0545,
      "theoretical_loss": 6.70416302544442,
      "tokens_seen": 12124160
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00036904761904761905,
      "loss": 5.051,
      "theoretical_loss": 6.6977274160275195,
      "tokens_seen": 12189696
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00037103174603174606,
      "loss": 4.936,
      "theoretical_loss": 6.691335942982863,
      "tokens_seen": 12255232
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000373015873015873,
      "loss": 5.2205,
      "theoretical_loss": 6.684988070009584,
      "tokens_seen": 12320768
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000375,
      "loss": 5.1985,
      "theoretical_loss": 6.6786832701318755,
      "tokens_seen": 12386304
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000376984126984127,
      "loss": 5.0665,
      "theoretical_loss": 6.672421025488536,
      "tokens_seen": 12451840
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00037896825396825396,
      "loss": 5.2492,
      "theoretical_loss": 6.666200827128331,
      "tokens_seen": 12517376
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00038095238095238096,
      "loss": 5.0901,
      "theoretical_loss": 6.660022174811009,
      "tokens_seen": 12582912
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00038293650793650797,
      "loss": 5.0826,
      "theoretical_loss": 6.653884576813754,
      "tokens_seen": 12648448
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00038492063492063497,
      "loss": 4.9834,
      "theoretical_loss": 6.647787549742928,
      "tokens_seen": 12713984
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003869047619047619,
      "loss": 5.0116,
      "theoretical_loss": 6.641730618350916,
      "tokens_seen": 12779520
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003888888888888889,
      "loss": 5.0323,
      "theoretical_loss": 6.6357133153579175,
      "tokens_seen": 12845056
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00039087301587301587,
      "loss": 5.0115,
      "theoretical_loss": 6.629735181278515,
      "tokens_seen": 12910592
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003928571428571429,
      "loss": 4.9832,
      "theoretical_loss": 6.623795764252885,
      "tokens_seen": 12976128
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003948412698412698,
      "loss": 5.0157,
      "theoretical_loss": 6.61789461988249,
      "tokens_seen": 13041664
    },
    {
      "epoch": 0.0,
      "objective/train/advantage_avg": 0.4912474751472473,
      "objective/train/docs_used": 12170,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 4.883951187133789,
      "objective/train/original_loss": 4.883951187133789,
      "objective/train/theoretical_loss": 6.612031311070119,
      "objective/train/tokens_used": 33567200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2418125718832016,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.05035400390625,
      "objective/train/weighted_lm_loss": 5.129752159118652,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 6.612031311070119,
      "tokens_seen": 13107200
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0003968253968253968,
      "loss": 5.0655,
      "theoretical_loss": 6.612031311070119,
      "tokens_seen": 13107200
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00039880952380952383,
      "loss": 4.8914,
      "theoretical_loss": 6.6062054078641355,
      "tokens_seen": 13172736
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00040079365079365083,
      "loss": 5.1053,
      "theoretical_loss": 6.600416487306803,
      "tokens_seen": 13238272
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004027777777777778,
      "loss": 4.9361,
      "theoretical_loss": 6.594664133286568,
      "tokens_seen": 13303808
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004047619047619048,
      "loss": 5.1403,
      "theoretical_loss": 6.588947936394168,
      "tokens_seen": 13369344
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00040674603174603173,
      "loss": 4.9681,
      "theoretical_loss": 6.58326749378245,
      "tokens_seen": 13434880
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00040873015873015874,
      "loss": 5.061,
      "theoretical_loss": 6.577622409029796,
      "tokens_seen": 13500416
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004107142857142857,
      "loss": 4.8943,
      "theoretical_loss": 6.5720122920070345,
      "tokens_seen": 13565952
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004126984126984127,
      "loss": 4.9159,
      "theoretical_loss": 6.566436758747731,
      "tokens_seen": 13631488
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004146825396825397,
      "loss": 4.8941,
      "theoretical_loss": 6.560895431321773,
      "tokens_seen": 13697024
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004166666666666667,
      "loss": 4.8609,
      "theoretical_loss": 6.5553879377121245,
      "tokens_seen": 13762560
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004186507936507937,
      "loss": 4.9139,
      "theoretical_loss": 6.5499139116946825,
      "tokens_seen": 13828096
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00042063492063492065,
      "loss": 4.9303,
      "theoretical_loss": 6.544472992721121,
      "tokens_seen": 13893632
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00042261904761904765,
      "loss": 4.9147,
      "theoretical_loss": 6.539064825804655,
      "tokens_seen": 13959168
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004246031746031746,
      "loss": 5.0321,
      "theoretical_loss": 6.533689061408615,
      "tokens_seen": 14024704
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004265873015873016,
      "loss": 4.9443,
      "theoretical_loss": 6.528345355337773,
      "tokens_seen": 14090240
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00042857142857142855,
      "loss": 4.9354,
      "theoretical_loss": 6.523033368632323,
      "tokens_seen": 14155776
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004305555555555556,
      "loss": 4.9561,
      "theoretical_loss": 6.51775276746445,
      "tokens_seen": 14221312
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00043253968253968256,
      "loss": 4.9391,
      "theoretical_loss": 6.512503223037404,
      "tokens_seen": 14286848
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00043452380952380956,
      "loss": 4.85,
      "theoretical_loss": 6.507284411487015,
      "tokens_seen": 14352384
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004365079365079365,
      "loss": 4.9346,
      "theoretical_loss": 6.502096013785574,
      "tokens_seen": 14417920
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004384920634920635,
      "loss": 4.8673,
      "theoretical_loss": 6.496937715648013,
      "tokens_seen": 14483456
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00044047619047619046,
      "loss": 4.7758,
      "theoretical_loss": 6.491809207440321,
      "tokens_seen": 14548992
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00044246031746031746,
      "loss": 4.8789,
      "theoretical_loss": 6.486710184090135,
      "tokens_seen": 14614528
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004444444444444444,
      "loss": 4.8879,
      "theoretical_loss": 6.481640344999435,
      "tokens_seen": 14680064
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00044642857142857147,
      "loss": 4.9789,
      "theoretical_loss": 6.476599393959298,
      "tokens_seen": 14745600
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004484126984126984,
      "loss": 4.8676,
      "theoretical_loss": 6.471587039066644,
      "tokens_seen": 14811136
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004503968253968254,
      "loss": 4.8724,
      "theoretical_loss": 6.466602992642919,
      "tokens_seen": 14876672
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00045238095238095237,
      "loss": 4.9345,
      "theoretical_loss": 6.461646971154669,
      "tokens_seen": 14942208
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00045436507936507937,
      "loss": 5.0052,
      "theoretical_loss": 6.456718695135944,
      "tokens_seen": 15007744
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004563492063492063,
      "loss": 4.899,
      "theoretical_loss": 6.45181788911249,
      "tokens_seen": 15073280
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004583333333333333,
      "loss": 4.7496,
      "theoretical_loss": 6.446944281527674,
      "tokens_seen": 15138816
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00046031746031746033,
      "loss": 4.7412,
      "theoretical_loss": 6.442097604670096,
      "tokens_seen": 15204352
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00046230158730158733,
      "loss": 4.6114,
      "theoretical_loss": 6.437277594602845,
      "tokens_seen": 15269888
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00046428571428571433,
      "loss": 4.7578,
      "theoretical_loss": 6.43248399109436,
      "tokens_seen": 15335424
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004662698412698413,
      "loss": 4.763,
      "theoretical_loss": 6.427716537550838,
      "tokens_seen": 15400960
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004682539682539683,
      "loss": 4.7541,
      "theoretical_loss": 6.422974980950157,
      "tokens_seen": 15466496
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00047023809523809523,
      "loss": 4.8955,
      "theoretical_loss": 6.41825907177728,
      "tokens_seen": 15532032
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00047222222222222224,
      "loss": 4.9811,
      "theoretical_loss": 6.413568563961092,
      "tokens_seen": 15597568
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004742063492063492,
      "loss": 4.7225,
      "theoretical_loss": 6.408903214812629,
      "tokens_seen": 15663104
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004761904761904762,
      "loss": 4.8861,
      "theoretical_loss": 6.404262784964672,
      "tokens_seen": 15728640
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004781746031746032,
      "loss": 4.7362,
      "theoretical_loss": 6.399647038312671,
      "tokens_seen": 15794176
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004801587301587302,
      "loss": 4.7356,
      "theoretical_loss": 6.395055741956952,
      "tokens_seen": 15859712
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00048214285714285715,
      "loss": 4.887,
      "theoretical_loss": 6.390488666146185,
      "tokens_seen": 15925248
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.00048412698412698415,
      "loss": 4.8904,
      "theoretical_loss": 6.3859455842220765,
      "tokens_seen": 15990784
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004861111111111111,
      "loss": 4.952,
      "theoretical_loss": 6.381426272565258,
      "tokens_seen": 16056320
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004880952380952381,
      "loss": 4.917,
      "theoretical_loss": 6.376930510542334,
      "tokens_seen": 16121856
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004900793650793651,
      "loss": 4.8962,
      "theoretical_loss": 6.372458080454071,
      "tokens_seen": 16187392
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000492063492063492,
      "loss": 4.8315,
      "theoretical_loss": 6.368008767484675,
      "tokens_seen": 16252928
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.0004940476190476191,
      "loss": 4.8416,
      "theoretical_loss": 6.363582359652165,
      "tokens_seen": 16318464
    },
    {
      "epoch": 0.0,
      "objective/train/advantage_avg": 0.49277380108833313,
      "objective/train/docs_used": 13382,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 4.31853723526001,
      "objective/train/original_loss": 4.318537712097168,
      "objective/train/theoretical_loss": 6.359178647759789,
      "objective/train/tokens_used": 36844000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2433290332555771,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505143404006958,
      "objective/train/weighted_lm_loss": 4.537021636962891,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 6.359178647759789,
      "tokens_seen": 16384000
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000496031746031746,
      "loss": 4.7369,
      "theoretical_loss": 6.359178647759789,
      "tokens_seen": 16384000
    },
    {
      "epoch": 0.0,
      "learning_rate": 0.000498015873015873,
      "loss": 4.7574,
      "theoretical_loss": 6.354797425348455,
      "tokens_seen": 16449536
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005,
      "loss": 4.61,
      "theoretical_loss": 6.350438488650175,
      "tokens_seen": 16515072
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000501984126984127,
      "loss": 4.6561,
      "theoretical_loss": 6.3461016365424685,
      "tokens_seen": 16580608
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000503968253968254,
      "loss": 4.7203,
      "theoretical_loss": 6.341786670503726,
      "tokens_seen": 16646144
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005059523809523809,
      "loss": 4.7175,
      "theoretical_loss": 6.33749339456949,
      "tokens_seen": 16711680
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005079365079365079,
      "loss": 4.7454,
      "theoretical_loss": 6.333221615289645,
      "tokens_seen": 16777216
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005099206349206349,
      "loss": 4.9234,
      "theoretical_loss": 6.328971141686479,
      "tokens_seen": 16842752
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005119047619047619,
      "loss": 4.7527,
      "theoretical_loss": 6.324741785213623,
      "tokens_seen": 16908288
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005138888888888888,
      "loss": 4.7599,
      "theoretical_loss": 6.320533359715803,
      "tokens_seen": 16973824
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005158730158730159,
      "loss": 4.7576,
      "theoretical_loss": 6.316345681389436,
      "tokens_seen": 17039360
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005178571428571429,
      "loss": 4.7286,
      "theoretical_loss": 6.312178568744001,
      "tokens_seen": 17104896
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005198412698412699,
      "loss": 4.7687,
      "theoretical_loss": 6.308031842564212,
      "tokens_seen": 17170432
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005218253968253968,
      "loss": 4.4944,
      "theoretical_loss": 6.303905325872933,
      "tokens_seen": 17235968
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005238095238095238,
      "loss": 4.9019,
      "theoretical_loss": 6.2997988438948465,
      "tokens_seen": 17301504
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005257936507936508,
      "loss": 4.6807,
      "theoretical_loss": 6.295712224020842,
      "tokens_seen": 17367040
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005277777777777778,
      "loss": 4.7674,
      "theoretical_loss": 6.2916452957731135,
      "tokens_seen": 17432576
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005297619047619047,
      "loss": 4.6331,
      "theoretical_loss": 6.287597890770951,
      "tokens_seen": 17498112
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005317460317460317,
      "loss": 4.5051,
      "theoretical_loss": 6.283569842697203,
      "tokens_seen": 17563648
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005337301587301587,
      "loss": 4.4246,
      "theoretical_loss": 6.279560987265395,
      "tokens_seen": 17629184
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005357142857142857,
      "loss": 4.6671,
      "theoretical_loss": 6.275571162187505,
      "tokens_seen": 17694720
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005376984126984126,
      "loss": 4.7847,
      "theoretical_loss": 6.271600207142356,
      "tokens_seen": 17760256
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005396825396825396,
      "loss": 4.7324,
      "theoretical_loss": 6.26764796374462,
      "tokens_seen": 17825792
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005416666666666666,
      "loss": 4.6055,
      "theoretical_loss": 6.263714275514436,
      "tokens_seen": 17891328
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005436507936507936,
      "loss": 4.5236,
      "theoretical_loss": 6.259798987847603,
      "tokens_seen": 17956864
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005456349206349206,
      "loss": 4.599,
      "theoretical_loss": 6.255901947986348,
      "tokens_seen": 18022400
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005476190476190477,
      "loss": 4.664,
      "theoretical_loss": 6.25202300499066,
      "tokens_seen": 18087936
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005496031746031747,
      "loss": 4.5562,
      "theoretical_loss": 6.248162009710162,
      "tokens_seen": 18153472
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005515873015873017,
      "loss": 4.7609,
      "theoretical_loss": 6.244318814756527,
      "tokens_seen": 18219008
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005535714285714287,
      "loss": 4.6061,
      "theoretical_loss": 6.240493274476407,
      "tokens_seen": 18284544
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005555555555555556,
      "loss": 4.5964,
      "theoretical_loss": 6.236685244924882,
      "tokens_seen": 18350080
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005575396825396826,
      "loss": 4.5874,
      "theoretical_loss": 6.232894583839405,
      "tokens_seen": 18415616
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005595238095238096,
      "loss": 4.4101,
      "theoretical_loss": 6.229121150614235,
      "tokens_seen": 18481152
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005615079365079366,
      "loss": 4.7411,
      "theoretical_loss": 6.225364806275344,
      "tokens_seen": 18546688
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005634920634920635,
      "loss": 4.5962,
      "theoretical_loss": 6.2216254134558024,
      "tokens_seen": 18612224
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005654761904761905,
      "loss": 4.5972,
      "theoretical_loss": 6.217902836371607,
      "tokens_seen": 18677760
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005674603174603175,
      "loss": 4.6108,
      "theoretical_loss": 6.214196940797967,
      "tokens_seen": 18743296
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005694444444444445,
      "loss": 4.582,
      "theoretical_loss": 6.210507594046018,
      "tokens_seen": 18808832
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005714285714285714,
      "loss": 4.7575,
      "theoretical_loss": 6.206834664939976,
      "tokens_seen": 18874368
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005734126984126984,
      "loss": 4.7611,
      "theoretical_loss": 6.203178023794688,
      "tokens_seen": 18939904
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005753968253968254,
      "loss": 4.6871,
      "theoretical_loss": 6.199537542393618,
      "tokens_seen": 19005440
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005773809523809524,
      "loss": 4.6562,
      "theoretical_loss": 6.195913093967214,
      "tokens_seen": 19070976
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005793650793650794,
      "loss": 4.5739,
      "theoretical_loss": 6.192304553171669,
      "tokens_seen": 19136512
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005813492063492064,
      "loss": 4.5292,
      "theoretical_loss": 6.188711796068077,
      "tokens_seen": 19202048
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005833333333333334,
      "loss": 4.3859,
      "theoretical_loss": 6.185134700101948,
      "tokens_seen": 19267584
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005853174603174604,
      "loss": 4.6491,
      "theoretical_loss": 6.181573144083102,
      "tokens_seen": 19333120
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005873015873015874,
      "loss": 4.633,
      "theoretical_loss": 6.178027008165916,
      "tokens_seen": 19398656
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005892857142857143,
      "loss": 4.6405,
      "theoretical_loss": 6.174496173829926,
      "tokens_seen": 19464192
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005912698412698413,
      "loss": 4.602,
      "theoretical_loss": 6.170980523860769,
      "tokens_seen": 19529728
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005932539682539683,
      "loss": 4.4713,
      "theoretical_loss": 6.167479942331466,
      "tokens_seen": 19595264
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.49178165197372437,
      "objective/train/docs_used": 14656,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 4.28451681137085,
      "objective/train/original_loss": 4.284516334533691,
      "objective/train/theoretical_loss": 6.163994314584031,
      "objective/train/tokens_used": 40120800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24251613020896912,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504111051559448,
      "objective/train/weighted_lm_loss": 4.500372409820557,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 6.163994314584031,
      "tokens_seen": 19660800
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005952380952380953,
      "loss": 4.5733,
      "theoretical_loss": 6.163994314584031,
      "tokens_seen": 19660800
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005972222222222222,
      "loss": 4.5806,
      "theoretical_loss": 6.16052352721141,
      "tokens_seen": 19726336
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0005992063492063492,
      "loss": 4.5064,
      "theoretical_loss": 6.157067468039726,
      "tokens_seen": 19791872
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006011904761904762,
      "loss": 4.4136,
      "theoretical_loss": 6.153626026110843,
      "tokens_seen": 19857408
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006031746031746032,
      "loss": 4.5153,
      "theoretical_loss": 6.150199091665225,
      "tokens_seen": 19922944
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006051587301587301,
      "loss": 4.5327,
      "theoretical_loss": 6.146786556125103,
      "tokens_seen": 19988480
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006071428571428571,
      "loss": 4.4621,
      "theoretical_loss": 6.143388312077918,
      "tokens_seen": 20054016
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006091269841269841,
      "loss": 4.5998,
      "theoretical_loss": 6.140004253260065,
      "tokens_seen": 20119552
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006111111111111112,
      "loss": 4.6879,
      "theoretical_loss": 6.136634274540901,
      "tokens_seen": 20185088
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006130952380952381,
      "loss": 4.5875,
      "theoretical_loss": 6.13327827190704,
      "tokens_seen": 20250624
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006150793650793651,
      "loss": 4.5669,
      "theoretical_loss": 6.129936142446908,
      "tokens_seen": 20316160
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006170634920634921,
      "loss": 4.475,
      "theoretical_loss": 6.12660778433556,
      "tokens_seen": 20381696
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006190476190476191,
      "loss": 4.4535,
      "theoretical_loss": 6.123293096819758,
      "tokens_seen": 20447232
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000621031746031746,
      "loss": 4.6288,
      "theoretical_loss": 6.119991980203299,
      "tokens_seen": 20512768
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000623015873015873,
      "loss": 4.3293,
      "theoretical_loss": 6.11670433583259,
      "tokens_seen": 20578304
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000625,
      "loss": 4.5311,
      "theoretical_loss": 6.113430066082468,
      "tokens_seen": 20643840
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000626984126984127,
      "loss": 4.4825,
      "theoretical_loss": 6.1101690743422505,
      "tokens_seen": 20709376
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000628968253968254,
      "loss": 4.4968,
      "theoretical_loss": 6.106921265002029,
      "tokens_seen": 20774912
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006309523809523809,
      "loss": 4.4781,
      "theoretical_loss": 6.103686543439177,
      "tokens_seen": 20840448
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006329365079365079,
      "loss": 4.6214,
      "theoretical_loss": 6.100464816005092,
      "tokens_seen": 20905984
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006349206349206349,
      "loss": 4.5962,
      "theoretical_loss": 6.097255990012153,
      "tokens_seen": 20971520
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006369047619047619,
      "loss": 4.5147,
      "theoretical_loss": 6.094059973720896,
      "tokens_seen": 21037056
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006388888888888888,
      "loss": 4.4541,
      "theoretical_loss": 6.090876676327392,
      "tokens_seen": 21102592
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006408730158730159,
      "loss": 4.4346,
      "theoretical_loss": 6.087706007950839,
      "tokens_seen": 21168128
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006428571428571429,
      "loss": 4.5464,
      "theoretical_loss": 6.084547879621354,
      "tokens_seen": 21233664
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006448412698412699,
      "loss": 4.5378,
      "theoretical_loss": 6.081402203267963,
      "tokens_seen": 21299200
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006468253968253968,
      "loss": 4.5303,
      "theoretical_loss": 6.078268891706788,
      "tokens_seen": 21364736
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006488095238095238,
      "loss": 4.4849,
      "theoretical_loss": 6.075147858629419,
      "tokens_seen": 21430272
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006507936507936508,
      "loss": 4.3543,
      "theoretical_loss": 6.072039018591484,
      "tokens_seen": 21495808
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006527777777777778,
      "loss": 4.5641,
      "theoretical_loss": 6.068942287001391,
      "tokens_seen": 21561344
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006547619047619047,
      "loss": 4.4829,
      "theoretical_loss": 6.065857580109258,
      "tokens_seen": 21626880
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006567460317460317,
      "loss": 4.3724,
      "theoretical_loss": 6.0627848149960135,
      "tokens_seen": 21692416
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006587301587301587,
      "loss": 4.5818,
      "theoretical_loss": 6.059723909562683,
      "tokens_seen": 21757952
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006607142857142857,
      "loss": 4.3904,
      "theoretical_loss": 6.0566747825198295,
      "tokens_seen": 21823488
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006626984126984126,
      "loss": 4.4051,
      "theoretical_loss": 6.0536373533771695,
      "tokens_seen": 21889024
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006646825396825396,
      "loss": 4.5032,
      "theoretical_loss": 6.050611542433352,
      "tokens_seen": 21954560
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006666666666666666,
      "loss": 4.3651,
      "theoretical_loss": 6.047597270765904,
      "tokens_seen": 22020096
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006686507936507936,
      "loss": 4.5418,
      "theoretical_loss": 6.044594460221319,
      "tokens_seen": 22085632
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006706349206349207,
      "loss": 4.4352,
      "theoretical_loss": 6.041603033405314,
      "tokens_seen": 22151168
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006726190476190477,
      "loss": 4.3779,
      "theoretical_loss": 6.038622913673237,
      "tokens_seen": 22216704
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006746031746031747,
      "loss": 4.2616,
      "theoretical_loss": 6.035654025120612,
      "tokens_seen": 22282240
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006765873015873017,
      "loss": 4.319,
      "theoretical_loss": 6.032696292573847,
      "tokens_seen": 22347776
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006785714285714287,
      "loss": 4.2787,
      "theoretical_loss": 6.029749641581073,
      "tokens_seen": 22413312
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006805555555555556,
      "loss": 4.343,
      "theoretical_loss": 6.02681399840313,
      "tokens_seen": 22478848
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006825396825396826,
      "loss": 4.4465,
      "theoretical_loss": 6.023889290004692,
      "tokens_seen": 22544384
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006845238095238096,
      "loss": 4.4453,
      "theoretical_loss": 6.020975444045523,
      "tokens_seen": 22609920
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006865079365079366,
      "loss": 4.28,
      "theoretical_loss": 6.018072388871873,
      "tokens_seen": 22675456
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006884920634920635,
      "loss": 4.4475,
      "theoretical_loss": 6.015180053507997,
      "tokens_seen": 22740992
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006904761904761905,
      "loss": 4.2961,
      "theoretical_loss": 6.012298367647816,
      "tokens_seen": 22806528
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006924603174603175,
      "loss": 4.3297,
      "theoretical_loss": 6.009427261646688,
      "tokens_seen": 22872064
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.49345701932907104,
      "objective/train/docs_used": 15907,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 4.267792701721191,
      "objective/train/original_loss": 4.267792701721191,
      "objective/train/theoretical_loss": 6.006566666513313,
      "objective/train/tokens_used": 43397600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24379104375839233,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505850315093994,
      "objective/train/weighted_lm_loss": 4.483581066131592,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 6.006566666513313,
      "tokens_seen": 22937600
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006944444444444445,
      "loss": 4.4127,
      "theoretical_loss": 6.006566666513313,
      "tokens_seen": 22937600
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006964285714285714,
      "loss": 4.09,
      "theoretical_loss": 6.003716513901762,
      "tokens_seen": 23003136
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0006984126984126984,
      "loss": 4.2815,
      "theoretical_loss": 6.000876736103618,
      "tokens_seen": 23068672
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007003968253968254,
      "loss": 4.4355,
      "theoretical_loss": 5.998047266040238,
      "tokens_seen": 23134208
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007023809523809524,
      "loss": 4.4496,
      "theoretical_loss": 5.995228037255137,
      "tokens_seen": 23199744
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007043650793650794,
      "loss": 4.3207,
      "theoretical_loss": 5.992418983906469,
      "tokens_seen": 23265280
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007063492063492064,
      "loss": 4.1968,
      "theoretical_loss": 5.989620040759641,
      "tokens_seen": 23330816
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007083333333333334,
      "loss": 4.2346,
      "theoretical_loss": 5.986831143180019,
      "tokens_seen": 23396352
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007103174603174604,
      "loss": 4.2309,
      "theoretical_loss": 5.9840522271257495,
      "tokens_seen": 23461888
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007123015873015874,
      "loss": 4.2863,
      "theoretical_loss": 5.981283229140683,
      "tokens_seen": 23527424
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007142857142857143,
      "loss": 4.2112,
      "theoretical_loss": 5.978524086347409,
      "tokens_seen": 23592960
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007162698412698413,
      "loss": 4.2578,
      "theoretical_loss": 5.975774736440385,
      "tokens_seen": 23658496
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007182539682539683,
      "loss": 4.2903,
      "theoretical_loss": 5.973035117679164,
      "tokens_seen": 23724032
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007202380952380953,
      "loss": 4.3281,
      "theoretical_loss": 5.970305168881732,
      "tokens_seen": 23789568
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007222222222222222,
      "loss": 4.4398,
      "theoretical_loss": 5.967584829417934,
      "tokens_seen": 23855104
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007242063492063492,
      "loss": 4.2352,
      "theoretical_loss": 5.964874039202989,
      "tokens_seen": 23920640
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007261904761904762,
      "loss": 4.2132,
      "theoretical_loss": 5.96217273869112,
      "tokens_seen": 23986176
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007281746031746032,
      "loss": 4.1317,
      "theoretical_loss": 5.959480868869245,
      "tokens_seen": 24051712
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007301587301587301,
      "loss": 4.3792,
      "theoretical_loss": 5.956798371250791,
      "tokens_seen": 24117248
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007321428571428571,
      "loss": 4.1713,
      "theoretical_loss": 5.954125187869572,
      "tokens_seen": 24182784
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007341269841269841,
      "loss": 4.2422,
      "theoretical_loss": 5.951461261273764,
      "tokens_seen": 24248320
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007361111111111112,
      "loss": 4.137,
      "theoretical_loss": 5.948806534519972,
      "tokens_seen": 24313856
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007380952380952381,
      "loss": 4.1628,
      "theoretical_loss": 5.9461609511673625,
      "tokens_seen": 24379392
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007400793650793651,
      "loss": 4.2607,
      "theoretical_loss": 5.943524455271907,
      "tokens_seen": 24444928
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007420634920634921,
      "loss": 4.2082,
      "theoretical_loss": 5.940896991380676,
      "tokens_seen": 24510464
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007440476190476191,
      "loss": 4.3838,
      "theoretical_loss": 5.938278504526239,
      "tokens_seen": 24576000
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000746031746031746,
      "loss": 4.2242,
      "theoretical_loss": 5.935668940221127,
      "tokens_seen": 24641536
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000748015873015873,
      "loss": 4.1961,
      "theoretical_loss": 5.933068244452382,
      "tokens_seen": 24707072
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.00075,
      "loss": 4.1996,
      "theoretical_loss": 5.930476363676177,
      "tokens_seen": 24772608
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000751984126984127,
      "loss": 4.2341,
      "theoretical_loss": 5.927893244812513,
      "tokens_seen": 24838144
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000753968253968254,
      "loss": 4.2983,
      "theoretical_loss": 5.92531883523999,
      "tokens_seen": 24903680
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007559523809523809,
      "loss": 4.144,
      "theoretical_loss": 5.922753082790654,
      "tokens_seen": 24969216
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007579365079365079,
      "loss": 4.1519,
      "theoretical_loss": 5.920195935744903,
      "tokens_seen": 25034752
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007599206349206349,
      "loss": 4.3062,
      "theoretical_loss": 5.917647342826485,
      "tokens_seen": 25100288
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007619047619047619,
      "loss": 4.164,
      "theoretical_loss": 5.915107253197538,
      "tokens_seen": 25165824
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007638888888888888,
      "loss": 4.2066,
      "theoretical_loss": 5.91257561645372,
      "tokens_seen": 25231360
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007658730158730159,
      "loss": 4.236,
      "theoretical_loss": 5.910052382619399,
      "tokens_seen": 25296896
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007678571428571429,
      "loss": 4.2914,
      "theoretical_loss": 5.907537502142899,
      "tokens_seen": 25362432
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007698412698412699,
      "loss": 4.2342,
      "theoretical_loss": 5.905030925891829,
      "tokens_seen": 25427968
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007718253968253968,
      "loss": 4.1399,
      "theoretical_loss": 5.9025326051484575,
      "tokens_seen": 25493504
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007738095238095238,
      "loss": 4.3571,
      "theoretical_loss": 5.900042491605162,
      "tokens_seen": 25559040
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007757936507936508,
      "loss": 4.313,
      "theoretical_loss": 5.897560537359936,
      "tokens_seen": 25624576
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007777777777777778,
      "loss": 4.1413,
      "theoretical_loss": 5.895086694911951,
      "tokens_seen": 25690112
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007797619047619047,
      "loss": 4.2811,
      "theoretical_loss": 5.892620917157192,
      "tokens_seen": 25755648
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007817460317460317,
      "loss": 4.1719,
      "theoretical_loss": 5.8901631573841335,
      "tokens_seen": 25821184
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007837301587301587,
      "loss": 4.2168,
      "theoretical_loss": 5.887713369269493,
      "tokens_seen": 25886720
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007857142857142857,
      "loss": 4.2532,
      "theoretical_loss": 5.88527150687402,
      "tokens_seen": 25952256
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007876984126984126,
      "loss": 4.298,
      "theoretical_loss": 5.88283752463836,
      "tokens_seen": 26017792
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007896825396825396,
      "loss": 4.0948,
      "theoretical_loss": 5.880411377378962,
      "tokens_seen": 26083328
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007916666666666666,
      "loss": 4.0214,
      "theoretical_loss": 5.87799302028405,
      "tokens_seen": 26148864
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.4926624894142151,
      "objective/train/docs_used": 17033,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 4.048245906829834,
      "objective/train/original_loss": 4.048246383666992,
      "objective/train/theoretical_loss": 5.8755824089096285,
      "objective/train/tokens_used": 46674400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24356918036937714,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050504446029663,
      "objective/train/weighted_lm_loss": 4.252634048461914,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.8755824089096285,
      "tokens_seen": 26214400
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007936507936507937,
      "loss": 4.3117,
      "theoretical_loss": 5.8755824089096285,
      "tokens_seen": 26214400
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007956349206349207,
      "loss": 3.9992,
      "theoretical_loss": 5.873179499175565,
      "tokens_seen": 26279936
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007976190476190477,
      "loss": 4.0696,
      "theoretical_loss": 5.87078424736171,
      "tokens_seen": 26345472
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0007996031746031747,
      "loss": 4.1271,
      "theoretical_loss": 5.868396610104066,
      "tokens_seen": 26411008
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008015873015873017,
      "loss": 4.1637,
      "theoretical_loss": 5.866016544391016,
      "tokens_seen": 26476544
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008035714285714287,
      "loss": 4.0211,
      "theoretical_loss": 5.863644007559595,
      "tokens_seen": 26542080
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008055555555555556,
      "loss": 4.1509,
      "theoretical_loss": 5.861278957291807,
      "tokens_seen": 26607616
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008075396825396826,
      "loss": 4.0588,
      "theoretical_loss": 5.858921351611005,
      "tokens_seen": 26673152
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008095238095238096,
      "loss": 4.0002,
      "theoretical_loss": 5.856571148878293,
      "tokens_seen": 26738688
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008115079365079366,
      "loss": 4.1366,
      "theoretical_loss": 5.854228307788999,
      "tokens_seen": 26804224
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008134920634920635,
      "loss": 4.1982,
      "theoretical_loss": 5.851892787369178,
      "tokens_seen": 26869760
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008154761904761905,
      "loss": 4.044,
      "theoretical_loss": 5.849564546972166,
      "tokens_seen": 26935296
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008174603174603175,
      "loss": 3.8531,
      "theoretical_loss": 5.847243546275179,
      "tokens_seen": 27000832
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008194444444444445,
      "loss": 4.0057,
      "theoretical_loss": 5.844929745275948,
      "tokens_seen": 27066368
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008214285714285714,
      "loss": 4.1101,
      "theoretical_loss": 5.842623104289412,
      "tokens_seen": 27131904
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008234126984126984,
      "loss": 3.9774,
      "theoretical_loss": 5.840323583944434,
      "tokens_seen": 27197440
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008253968253968254,
      "loss": 3.8177,
      "theoretical_loss": 5.838031145180573,
      "tokens_seen": 27262976
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008273809523809524,
      "loss": 4.1632,
      "theoretical_loss": 5.835745749244896,
      "tokens_seen": 27328512
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008293650793650794,
      "loss": 4.1018,
      "theoretical_loss": 5.833467357688816,
      "tokens_seen": 27394048
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008313492063492064,
      "loss": 4.1213,
      "theoretical_loss": 5.8311959323649925,
      "tokens_seen": 27459584
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008333333333333334,
      "loss": 4.1399,
      "theoretical_loss": 5.82893143542425,
      "tokens_seen": 27525120
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008353174603174604,
      "loss": 4.0058,
      "theoretical_loss": 5.826673829312552,
      "tokens_seen": 27590656
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008373015873015874,
      "loss": 4.1258,
      "theoretical_loss": 5.824423076767996,
      "tokens_seen": 27656192
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008392857142857143,
      "loss": 3.9901,
      "theoretical_loss": 5.8221791408178625,
      "tokens_seen": 27721728
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008412698412698413,
      "loss": 4.0441,
      "theoretical_loss": 5.81994198477569,
      "tokens_seen": 27787264
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008432539682539683,
      "loss": 4.0542,
      "theoretical_loss": 5.817711572238393,
      "tokens_seen": 27852800
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008452380952380953,
      "loss": 4.0306,
      "theoretical_loss": 5.815487867083405,
      "tokens_seen": 27918336
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008472222222222222,
      "loss": 3.9466,
      "theoretical_loss": 5.813270833465875,
      "tokens_seen": 27983872
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008492063492063492,
      "loss": 3.9006,
      "theoretical_loss": 5.811060435815881,
      "tokens_seen": 28049408
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008511904761904762,
      "loss": 4.0148,
      "theoretical_loss": 5.808856638835682,
      "tokens_seen": 28114944
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008531746031746032,
      "loss": 3.9268,
      "theoretical_loss": 5.806659407497016,
      "tokens_seen": 28180480
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008551587301587301,
      "loss": 4.1383,
      "theoretical_loss": 5.804468707038416,
      "tokens_seen": 28246016
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008571428571428571,
      "loss": 4.0713,
      "theoretical_loss": 5.802284502962563,
      "tokens_seen": 28311552
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008591269841269841,
      "loss": 4.0327,
      "theoretical_loss": 5.8001067610336765,
      "tokens_seen": 28377088
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008611111111111112,
      "loss": 4.1369,
      "theoretical_loss": 5.797935447274931,
      "tokens_seen": 28442624
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008630952380952381,
      "loss": 3.9127,
      "theoretical_loss": 5.795770527965905,
      "tokens_seen": 28508160
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008650793650793651,
      "loss": 3.959,
      "theoretical_loss": 5.793611969640068,
      "tokens_seen": 28573696
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008670634920634921,
      "loss": 3.7726,
      "theoretical_loss": 5.791459739082283,
      "tokens_seen": 28639232
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008690476190476191,
      "loss": 4.043,
      "theoretical_loss": 5.789313803326356,
      "tokens_seen": 28704768
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000871031746031746,
      "loss": 3.9936,
      "theoretical_loss": 5.787174129652598,
      "tokens_seen": 28770304
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000873015873015873,
      "loss": 4.0503,
      "theoretical_loss": 5.785040685585437,
      "tokens_seen": 28835840
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000875,
      "loss": 3.9107,
      "theoretical_loss": 5.782913438891035,
      "tokens_seen": 28901376
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000876984126984127,
      "loss": 3.9906,
      "theoretical_loss": 5.780792357574953,
      "tokens_seen": 28966912
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000878968253968254,
      "loss": 4.0754,
      "theoretical_loss": 5.778677409879835,
      "tokens_seen": 29032448
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008809523809523809,
      "loss": 3.9564,
      "theoretical_loss": 5.7765685642831155,
      "tokens_seen": 29097984
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008829365079365079,
      "loss": 4.0196,
      "theoretical_loss": 5.7744657894947675,
      "tokens_seen": 29163520
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008849206349206349,
      "loss": 4.0888,
      "theoretical_loss": 5.772369054455065,
      "tokens_seen": 29229056
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008869047619047619,
      "loss": 3.9354,
      "theoretical_loss": 5.770278328332372,
      "tokens_seen": 29294592
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008888888888888888,
      "loss": 4.0387,
      "theoretical_loss": 5.768193580520972,
      "tokens_seen": 29360128
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008908730158730159,
      "loss": 3.8796,
      "theoretical_loss": 5.766114780638902,
      "tokens_seen": 29425664
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.49078458547592163,
      "objective/train/docs_used": 18174,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 3.943504571914673,
      "objective/train/original_loss": 3.9435043334960938,
      "objective/train/theoretical_loss": 5.7640418985258295,
      "objective/train/tokens_used": 49951200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2413143813610077,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503051280975342,
      "objective/train/weighted_lm_loss": 4.141692161560059,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.7640418985258295,
      "tokens_seen": 29491200
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008928571428571429,
      "loss": 4.0427,
      "theoretical_loss": 5.7640418985258295,
      "tokens_seen": 29491200
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008948412698412699,
      "loss": 3.9764,
      "theoretical_loss": 5.761974904240946,
      "tokens_seen": 29556736
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008968253968253968,
      "loss": 3.8796,
      "theoretical_loss": 5.759913768060882,
      "tokens_seen": 29622272
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0008988095238095238,
      "loss": 3.9728,
      "theoretical_loss": 5.757858460477662,
      "tokens_seen": 29687808
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009007936507936508,
      "loss": 4.0234,
      "theoretical_loss": 5.755808952196658,
      "tokens_seen": 29753344
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009027777777777778,
      "loss": 3.8761,
      "theoretical_loss": 5.753765214134593,
      "tokens_seen": 29818880
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009047619047619047,
      "loss": 4.0411,
      "theoretical_loss": 5.7517272174175496,
      "tokens_seen": 29884416
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009067460317460317,
      "loss": 4.0655,
      "theoretical_loss": 5.7496949333790095,
      "tokens_seen": 29949952
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009087301587301587,
      "loss": 3.8995,
      "theoretical_loss": 5.747668333557913,
      "tokens_seen": 30015488
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009107142857142857,
      "loss": 3.9744,
      "theoretical_loss": 5.745647389696742,
      "tokens_seen": 30081024
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009126984126984126,
      "loss": 4.072,
      "theoretical_loss": 5.743632073739626,
      "tokens_seen": 30146560
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009146825396825396,
      "loss": 3.9495,
      "theoretical_loss": 5.741622357830466,
      "tokens_seen": 30212096
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009166666666666666,
      "loss": 3.9492,
      "theoretical_loss": 5.7396182143110845,
      "tokens_seen": 30277632
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009186507936507937,
      "loss": 4.0284,
      "theoretical_loss": 5.737619615719392,
      "tokens_seen": 30343168
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009206349206349207,
      "loss": 3.892,
      "theoretical_loss": 5.735626534787584,
      "tokens_seen": 30408704
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009226190476190477,
      "loss": 3.8598,
      "theoretical_loss": 5.7336389444403375,
      "tokens_seen": 30474240
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009246031746031747,
      "loss": 3.8676,
      "theoretical_loss": 5.731656817793057,
      "tokens_seen": 30539776
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009265873015873017,
      "loss": 3.8372,
      "theoretical_loss": 5.729680128150115,
      "tokens_seen": 30605312
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009285714285714287,
      "loss": 3.7756,
      "theoretical_loss": 5.727708849003127,
      "tokens_seen": 30670848
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009305555555555556,
      "loss": 3.9913,
      "theoretical_loss": 5.72574295402924,
      "tokens_seen": 30736384
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009325396825396826,
      "loss": 3.9164,
      "theoretical_loss": 5.723782417089441,
      "tokens_seen": 30801920
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009345238095238096,
      "loss": 3.8756,
      "theoretical_loss": 5.721827212226889,
      "tokens_seen": 30867456
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009365079365079366,
      "loss": 3.7424,
      "theoretical_loss": 5.719877313665254,
      "tokens_seen": 30932992
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009384920634920635,
      "loss": 3.8924,
      "theoretical_loss": 5.717932695807093,
      "tokens_seen": 30998528
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009404761904761905,
      "loss": 3.9166,
      "theoretical_loss": 5.715993333232223,
      "tokens_seen": 31064064
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009424603174603175,
      "loss": 3.8553,
      "theoretical_loss": 5.7140592006961315,
      "tokens_seen": 31129600
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009444444444444445,
      "loss": 3.832,
      "theoretical_loss": 5.712130273128388,
      "tokens_seen": 31195136
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009464285714285714,
      "loss": 3.7882,
      "theoretical_loss": 5.710206525631086,
      "tokens_seen": 31260672
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009484126984126984,
      "loss": 4.0133,
      "theoretical_loss": 5.708287933477296,
      "tokens_seen": 31326208
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009503968253968254,
      "loss": 4.0449,
      "theoretical_loss": 5.706374472109536,
      "tokens_seen": 31391744
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009523809523809524,
      "loss": 3.8148,
      "theoretical_loss": 5.704466117138258,
      "tokens_seen": 31457280
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009543650793650794,
      "loss": 3.91,
      "theoretical_loss": 5.702562844340358,
      "tokens_seen": 31522816
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009563492063492064,
      "loss": 3.7582,
      "theoretical_loss": 5.700664629657688,
      "tokens_seen": 31588352
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009583333333333334,
      "loss": 4.045,
      "theoretical_loss": 5.698771449195602,
      "tokens_seen": 31653888
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009603174603174604,
      "loss": 3.8957,
      "theoretical_loss": 5.696883279221504,
      "tokens_seen": 31719424
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009623015873015874,
      "loss": 3.8502,
      "theoretical_loss": 5.695000096163421,
      "tokens_seen": 31784960
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009642857142857143,
      "loss": 3.7878,
      "theoretical_loss": 5.693121876608577,
      "tokens_seen": 31850496
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009662698412698413,
      "loss": 3.9058,
      "theoretical_loss": 5.691248597302009,
      "tokens_seen": 31916032
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009682539682539683,
      "loss": 3.8435,
      "theoretical_loss": 5.689380235145171,
      "tokens_seen": 31981568
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009702380952380953,
      "loss": 3.8857,
      "theoretical_loss": 5.687516767194563,
      "tokens_seen": 32047104
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009722222222222222,
      "loss": 3.8421,
      "theoretical_loss": 5.685658170660382,
      "tokens_seen": 32112640
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009742063492063492,
      "loss": 3.9891,
      "theoretical_loss": 5.683804422905179,
      "tokens_seen": 32178176
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009761904761904762,
      "loss": 3.7935,
      "theoretical_loss": 5.6819555014425305,
      "tokens_seen": 32243712
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009781746031746032,
      "loss": 3.7785,
      "theoretical_loss": 5.680111383935727,
      "tokens_seen": 32309248
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009801587301587302,
      "loss": 3.9492,
      "theoretical_loss": 5.678272048196479,
      "tokens_seen": 32374784
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009821428571428572,
      "loss": 3.8507,
      "theoretical_loss": 5.6764374721836255,
      "tokens_seen": 32440320
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000984126984126984,
      "loss": 3.7337,
      "theoretical_loss": 5.674607634001871,
      "tokens_seen": 32505856
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009861111111111112,
      "loss": 3.9994,
      "theoretical_loss": 5.672782511900524,
      "tokens_seen": 32571392
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009880952380952382,
      "loss": 3.9901,
      "theoretical_loss": 5.670962084272254,
      "tokens_seen": 32636928
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009900793650793652,
      "loss": 3.9945,
      "theoretical_loss": 5.669146329651861,
      "tokens_seen": 32702464
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.49417802691459656,
      "objective/train/docs_used": 19239,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 3.745575189590454,
      "objective/train/original_loss": 3.745574951171875,
      "objective/train/theoretical_loss": 5.667335226715059,
      "objective/train/tokens_used": 53228000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2445439100265503,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506609678268433,
      "objective/train/weighted_lm_loss": 3.935272216796875,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.667335226715059,
      "tokens_seen": 32768000
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000992063492063492,
      "loss": 3.9393,
      "theoretical_loss": 5.667335226715059,
      "tokens_seen": 32768000
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000994047619047619,
      "loss": 3.8418,
      "theoretical_loss": 5.66552875427727,
      "tokens_seen": 32833536
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996031746031746,
      "loss": 3.8581,
      "theoretical_loss": 5.663726891292432,
      "tokens_seen": 32899072
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998015873015873,
      "loss": 3.6826,
      "theoretical_loss": 5.6619296168518245,
      "tokens_seen": 32964608
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.001,
      "loss": 3.9296,
      "theoretical_loss": 5.6601369101828904,
      "tokens_seen": 33030144
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009999799398194584,
      "loss": 3.8296,
      "theoretical_loss": 5.658348750648093,
      "tokens_seen": 33095680
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009999598796389167,
      "loss": 3.7304,
      "theoretical_loss": 5.656565117743766,
      "tokens_seen": 33161216
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999939819458375,
      "loss": 3.7348,
      "theoretical_loss": 5.654785991098984,
      "tokens_seen": 33226752
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009999197592778335,
      "loss": 3.807,
      "theoretical_loss": 5.6530113504744435,
      "tokens_seen": 33292288
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999899699097292,
      "loss": 3.7786,
      "theoretical_loss": 5.651241175761357,
      "tokens_seen": 33357824
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009998796389167502,
      "loss": 3.79,
      "theoretical_loss": 5.649475446980356,
      "tokens_seen": 33423360
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009998595787362088,
      "loss": 3.9043,
      "theoretical_loss": 5.647714144280403,
      "tokens_seen": 33488896
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999839518555667,
      "loss": 3.8137,
      "theoretical_loss": 5.645957247937725,
      "tokens_seen": 33554432
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009998194583751255,
      "loss": 3.7915,
      "theoretical_loss": 5.644204738354742,
      "tokens_seen": 33619968
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009997993981945839,
      "loss": 3.837,
      "theoretical_loss": 5.642456596059025,
      "tokens_seen": 33685504
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009997793380140422,
      "loss": 3.7575,
      "theoretical_loss": 5.6407128017022465,
      "tokens_seen": 33751040
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009997592778335006,
      "loss": 3.8424,
      "theoretical_loss": 5.638973336059157,
      "tokens_seen": 33816576
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999739217652959,
      "loss": 3.6526,
      "theoretical_loss": 5.637238180026563,
      "tokens_seen": 33882112
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009997191574724173,
      "loss": 3.6857,
      "theoretical_loss": 5.6355073146223145,
      "tokens_seen": 33947648
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009996990972918757,
      "loss": 3.6679,
      "theoretical_loss": 5.633780720984311,
      "tokens_seen": 34013184
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999679037111334,
      "loss": 3.7352,
      "theoretical_loss": 5.632058380369512,
      "tokens_seen": 34078720
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009996589769307924,
      "loss": 3.6703,
      "theoretical_loss": 5.630340274152954,
      "tokens_seen": 34144256
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009996389167502508,
      "loss": 3.8389,
      "theoretical_loss": 5.62862638382679,
      "tokens_seen": 34209792
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009996188565697091,
      "loss": 3.7032,
      "theoretical_loss": 5.626916690999319,
      "tokens_seen": 34275328
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009995987963891675,
      "loss": 3.6495,
      "theoretical_loss": 5.625211177394046,
      "tokens_seen": 34340864
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009995787362086258,
      "loss": 3.6542,
      "theoretical_loss": 5.6235098248487425,
      "tokens_seen": 34406400
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009995586760280842,
      "loss": 3.5303,
      "theoretical_loss": 5.621812615314505,
      "tokens_seen": 34471936
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009995386158475428,
      "loss": 3.752,
      "theoretical_loss": 5.620119530854848,
      "tokens_seen": 34537472
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999518555667001,
      "loss": 3.7169,
      "theoretical_loss": 5.618430553644782,
      "tokens_seen": 34603008
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009994984954864595,
      "loss": 3.6834,
      "theoretical_loss": 5.616745665969917,
      "tokens_seen": 34668544
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009994784353059177,
      "loss": 3.6833,
      "theoretical_loss": 5.615064850225565,
      "tokens_seen": 34734080
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009994583751253762,
      "loss": 3.7607,
      "theoretical_loss": 5.613388088915855,
      "tokens_seen": 34799616
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009994383149448346,
      "loss": 3.707,
      "theoretical_loss": 5.611715364652864,
      "tokens_seen": 34865152
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999418254764293,
      "loss": 3.6116,
      "theoretical_loss": 5.610046660155737,
      "tokens_seen": 34930688
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009993981945837513,
      "loss": 3.6668,
      "theoretical_loss": 5.608381958249839,
      "tokens_seen": 34996224
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009993781344032097,
      "loss": 3.6732,
      "theoretical_loss": 5.6067212418659,
      "tokens_seen": 35061760
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999358074222668,
      "loss": 3.6028,
      "theoretical_loss": 5.605064494039176,
      "tokens_seen": 35127296
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009993380140421264,
      "loss": 3.6866,
      "theoretical_loss": 5.603411697908612,
      "tokens_seen": 35192832
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009993179538615848,
      "loss": 3.6574,
      "theoretical_loss": 5.601762836716016,
      "tokens_seen": 35258368
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009992978936810431,
      "loss": 3.4094,
      "theoretical_loss": 5.600117893805246,
      "tokens_seen": 35323904
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009992778335005015,
      "loss": 3.5417,
      "theoretical_loss": 5.598476852621397,
      "tokens_seen": 35389440
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009992577733199599,
      "loss": 3.5647,
      "theoretical_loss": 5.596839696710004,
      "tokens_seen": 35454976
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009992377131394182,
      "loss": 3.7131,
      "theoretical_loss": 5.59520640971624,
      "tokens_seen": 35520512
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009992176529588766,
      "loss": 3.5971,
      "theoretical_loss": 5.5935769753841384,
      "tokens_seen": 35586048
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999197592778335,
      "loss": 3.9092,
      "theoretical_loss": 5.591951377555809,
      "tokens_seen": 35651584
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009991775325977935,
      "loss": 3.7656,
      "theoretical_loss": 5.590329600170675,
      "tokens_seen": 35717120
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009991574724172517,
      "loss": 3.8054,
      "theoretical_loss": 5.588711627264699,
      "tokens_seen": 35782656
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009991374122367102,
      "loss": 3.6167,
      "theoretical_loss": 5.587097442969634,
      "tokens_seen": 35848192
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009991173520561684,
      "loss": 3.7877,
      "theoretical_loss": 5.585487031512276,
      "tokens_seen": 35913728
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999097291875627,
      "loss": 3.7781,
      "theoretical_loss": 5.583880377213717,
      "tokens_seen": 35979264
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.4903879463672638,
      "objective/train/docs_used": 20470,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 3.424363613128662,
      "objective/train/original_loss": 3.424363613128662,
      "objective/train/theoretical_loss": 5.582277464488612,
      "objective/train/tokens_used": 56504800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2408606857061386,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502631664276123,
      "objective/train/weighted_lm_loss": 3.596794605255127,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0253151655197144,
      "theoretical_loss": 5.582277464488612,
      "tokens_seen": 36044800
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009990772316950853,
      "loss": 3.5566,
      "theoretical_loss": 5.582277464488612,
      "tokens_seen": 36044800
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009990571715145437,
      "loss": 3.8534,
      "theoretical_loss": 5.58067827784446,
      "tokens_seen": 36110336
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000999037111334002,
      "loss": 3.6709,
      "theoretical_loss": 5.579082801880871,
      "tokens_seen": 36175872
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009990170511534604,
      "loss": 3.7757,
      "theoretical_loss": 5.577491021288861,
      "tokens_seen": 36241408
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009989969909729188,
      "loss": 3.5704,
      "theoretical_loss": 5.57590292085014,
      "tokens_seen": 36306944
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009989769307923771,
      "loss": 3.7544,
      "theoretical_loss": 5.574318485436422,
      "tokens_seen": 36372480
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009989568706118355,
      "loss": 3.512,
      "theoretical_loss": 5.572737700008718,
      "tokens_seen": 36438016
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009989368104312939,
      "loss": 3.5182,
      "theoretical_loss": 5.5711605496166605,
      "tokens_seen": 36503552
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009989167502507522,
      "loss": 3.785,
      "theoretical_loss": 5.569587019397816,
      "tokens_seen": 36569088
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009988966900702106,
      "loss": 3.6573,
      "theoretical_loss": 5.5680170945770175,
      "tokens_seen": 36634624
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998876629889669,
      "loss": 3.6389,
      "theoretical_loss": 5.56645076046569,
      "tokens_seen": 36700160
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009988565697091273,
      "loss": 3.4898,
      "theoretical_loss": 5.5648880024612,
      "tokens_seen": 36765696
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009988365095285857,
      "loss": 3.6444,
      "theoretical_loss": 5.5633288060461865,
      "tokens_seen": 36831232
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009988164493480443,
      "loss": 3.6373,
      "theoretical_loss": 5.561773156787925,
      "tokens_seen": 36896768
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009987963891675024,
      "loss": 3.667,
      "theoretical_loss": 5.5602210403376775,
      "tokens_seen": 36962304
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998776328986961,
      "loss": 3.5937,
      "theoretical_loss": 5.558672442430058,
      "tokens_seen": 37027840
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009987562688064191,
      "loss": 3.7541,
      "theoretical_loss": 5.557127348882403,
      "tokens_seen": 37093376
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009987362086258777,
      "loss": 3.6276,
      "theoretical_loss": 5.555585745594145,
      "tokens_seen": 37158912
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998716148445336,
      "loss": 3.6938,
      "theoretical_loss": 5.554047618546193,
      "tokens_seen": 37224448
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009986960882647944,
      "loss": 3.6552,
      "theoretical_loss": 5.552512953800322,
      "tokens_seen": 37289984
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009986760280842528,
      "loss": 3.5607,
      "theoretical_loss": 5.550981737498564,
      "tokens_seen": 37355520
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009986559679037112,
      "loss": 3.7794,
      "theoretical_loss": 5.549453955862602,
      "tokens_seen": 37421056
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009986359077231695,
      "loss": 3.7612,
      "theoretical_loss": 5.547929595193182,
      "tokens_seen": 37486592
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009986158475426279,
      "loss": 3.5894,
      "theoretical_loss": 5.5464086418695135,
      "tokens_seen": 37552128
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009985957873620862,
      "loss": 3.5894,
      "theoretical_loss": 5.544891082348692,
      "tokens_seen": 37617664
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009985757271815446,
      "loss": 3.6129,
      "theoretical_loss": 5.543376903165111,
      "tokens_seen": 37683200
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998555667001003,
      "loss": 3.5844,
      "theoretical_loss": 5.5418660909298945,
      "tokens_seen": 37748736
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009985356068204613,
      "loss": 3.6763,
      "theoretical_loss": 5.540358632330321,
      "tokens_seen": 37814272
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009985155466399197,
      "loss": 3.4916,
      "theoretical_loss": 5.538854514129265,
      "tokens_seen": 37879808
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998495486459378,
      "loss": 3.4597,
      "theoretical_loss": 5.5373537231646335,
      "tokens_seen": 37945344
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009984754262788366,
      "loss": 3.5356,
      "theoretical_loss": 5.535856246348814,
      "tokens_seen": 38010880
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998455366098295,
      "loss": 3.6021,
      "theoretical_loss": 5.534362070668127,
      "tokens_seen": 38076416
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009984353059177534,
      "loss": 3.5672,
      "theoretical_loss": 5.532871183182278,
      "tokens_seen": 38141952
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009984152457372117,
      "loss": 3.5533,
      "theoretical_loss": 5.53138357102382,
      "tokens_seen": 38207488
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.00099839518555667,
      "loss": 3.6106,
      "theoretical_loss": 5.529899221397624,
      "tokens_seen": 38273024
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009983751253761284,
      "loss": 3.4664,
      "theoretical_loss": 5.528418121580339,
      "tokens_seen": 38338560
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009983550651955868,
      "loss": 3.3992,
      "theoretical_loss": 5.526940258919881,
      "tokens_seen": 38404096
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009983350050150452,
      "loss": 3.6064,
      "theoretical_loss": 5.5254656208349004,
      "tokens_seen": 38469632
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009983149448345035,
      "loss": 3.5422,
      "theoretical_loss": 5.523994194814273,
      "tokens_seen": 38535168
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998294884653962,
      "loss": 3.3976,
      "theoretical_loss": 5.522525968416592,
      "tokens_seen": 38600704
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009982748244734203,
      "loss": 3.6676,
      "theoretical_loss": 5.521060929269655,
      "tokens_seen": 38666240
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009982547642928786,
      "loss": 3.5906,
      "theoretical_loss": 5.519599065069971,
      "tokens_seen": 38731776
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998234704112337,
      "loss": 3.5077,
      "theoretical_loss": 5.518140363582252,
      "tokens_seen": 38797312
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009982146439317953,
      "loss": 3.4583,
      "theoretical_loss": 5.516684812638932,
      "tokens_seen": 38862848
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998194583751254,
      "loss": 3.6336,
      "theoretical_loss": 5.515232400139671,
      "tokens_seen": 38928384
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998174523570712,
      "loss": 3.667,
      "theoretical_loss": 5.5137831140508755,
      "tokens_seen": 38993920
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009981544633901706,
      "loss": 3.4816,
      "theoretical_loss": 5.512336942405216,
      "tokens_seen": 39059456
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009981344032096288,
      "loss": 3.5222,
      "theoretical_loss": 5.510893873301154,
      "tokens_seen": 39124992
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009981143430290874,
      "loss": 3.5284,
      "theoretical_loss": 5.509453894902469,
      "tokens_seen": 39190528
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009980942828485457,
      "loss": 3.483,
      "theoretical_loss": 5.508016995437791,
      "tokens_seen": 39256064
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.4876997768878937,
      "objective/train/docs_used": 21760,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 3.0592448711395264,
      "objective/train/original_loss": 3.0592451095581055,
      "objective/train/theoretical_loss": 5.506583163200142,
      "objective/train/tokens_used": 59781600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.23880939185619354,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0499838590621948,
      "objective/train/weighted_lm_loss": 3.212702512741089,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.506583163200142,
      "tokens_seen": 39321600
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000998074222668004,
      "loss": 3.3564,
      "theoretical_loss": 5.506583163200142,
      "tokens_seen": 39321600
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009980541624874625,
      "loss": 3.5219,
      "theoretical_loss": 5.505152386546474,
      "tokens_seen": 39387136
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009980341023069208,
      "loss": 3.6034,
      "theoretical_loss": 5.503724653897207,
      "tokens_seen": 39452672
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009980140421263792,
      "loss": 3.5868,
      "theoretical_loss": 5.502299953735795,
      "tokens_seen": 39518208
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009979939819458375,
      "loss": 3.43,
      "theoretical_loss": 5.5008782746082625,
      "tokens_seen": 39583744
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997973921765296,
      "loss": 3.6451,
      "theoretical_loss": 5.499459605122775,
      "tokens_seen": 39649280
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009979538615847543,
      "loss": 3.3693,
      "theoretical_loss": 5.498043933949187,
      "tokens_seen": 39714816
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009979338014042126,
      "loss": 3.5735,
      "theoretical_loss": 5.496631249818616,
      "tokens_seen": 39780352
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997913741223671,
      "loss": 3.4721,
      "theoretical_loss": 5.495221541523011,
      "tokens_seen": 39845888
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009978936810431294,
      "loss": 3.4922,
      "theoretical_loss": 5.493814797914718,
      "tokens_seen": 39911424
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009978736208625877,
      "loss": 3.6077,
      "theoretical_loss": 5.49241100790606,
      "tokens_seen": 39976960
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997853560682046,
      "loss": 3.5721,
      "theoretical_loss": 5.491010160468919,
      "tokens_seen": 40042496
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009978335005015047,
      "loss": 3.399,
      "theoretical_loss": 5.489612244634316,
      "tokens_seen": 40108032
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009978134403209628,
      "loss": 3.4734,
      "theoretical_loss": 5.4882172494919965,
      "tokens_seen": 40173568
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009977933801404214,
      "loss": 3.542,
      "theoretical_loss": 5.486825164190026,
      "tokens_seen": 40239104
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009977733199598795,
      "loss": 3.4436,
      "theoretical_loss": 5.4854359779343795,
      "tokens_seen": 40304640
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009977532597793381,
      "loss": 3.5102,
      "theoretical_loss": 5.48404967998854,
      "tokens_seen": 40370176
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009977331995987965,
      "loss": 3.4231,
      "theoretical_loss": 5.482666259673101,
      "tokens_seen": 40435712
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009977131394182548,
      "loss": 3.6534,
      "theoretical_loss": 5.481285706365366,
      "tokens_seen": 40501248
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009976930792377132,
      "loss": 3.5207,
      "theoretical_loss": 5.479908009498965,
      "tokens_seen": 40566784
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009976730190571716,
      "loss": 3.4939,
      "theoretical_loss": 5.478533158563456,
      "tokens_seen": 40632320
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.00099765295887663,
      "loss": 3.7361,
      "theoretical_loss": 5.477161143103945,
      "tokens_seen": 40697856
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009976328986960883,
      "loss": 3.4448,
      "theoretical_loss": 5.4757919527207,
      "tokens_seen": 40763392
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009976128385155466,
      "loss": 3.5947,
      "theoretical_loss": 5.474425577068777,
      "tokens_seen": 40828928
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997592778335005,
      "loss": 3.4699,
      "theoretical_loss": 5.473062005857637,
      "tokens_seen": 40894464
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009975727181544634,
      "loss": 3.6217,
      "theoretical_loss": 5.471701228850778,
      "tokens_seen": 40960000
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009975526579739217,
      "loss": 3.6091,
      "theoretical_loss": 5.470343235865368,
      "tokens_seen": 41025536
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.00099753259779338,
      "loss": 3.4589,
      "theoretical_loss": 5.468988016771869,
      "tokens_seen": 41091072
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009975125376128385,
      "loss": 3.4358,
      "theoretical_loss": 5.467635561493681,
      "tokens_seen": 41156608
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009974924774322968,
      "loss": 3.4588,
      "theoretical_loss": 5.466285860006783,
      "tokens_seen": 41222144
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009974724172517554,
      "loss": 3.6569,
      "theoretical_loss": 5.46493890233937,
      "tokens_seen": 41287680
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009974523570712135,
      "loss": 3.453,
      "theoretical_loss": 5.4635946785715,
      "tokens_seen": 41353216
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009974322968906721,
      "loss": 3.3928,
      "theoretical_loss": 5.462253178834744,
      "tokens_seen": 41418752
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009974122367101303,
      "loss": 3.3296,
      "theoretical_loss": 5.460914393311839,
      "tokens_seen": 41484288
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009973921765295888,
      "loss": 3.4467,
      "theoretical_loss": 5.4595783122363395,
      "tokens_seen": 41549824
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009973721163490472,
      "loss": 3.4153,
      "theoretical_loss": 5.458244925892275,
      "tokens_seen": 41615360
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009973520561685056,
      "loss": 3.4421,
      "theoretical_loss": 5.456914224613812,
      "tokens_seen": 41680896
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997331995987964,
      "loss": 3.4096,
      "theoretical_loss": 5.455586198784916,
      "tokens_seen": 41746432
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009973119358074223,
      "loss": 3.4896,
      "theoretical_loss": 5.454260838839016,
      "tokens_seen": 41811968
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009972918756268807,
      "loss": 3.5514,
      "theoretical_loss": 5.452938135258673,
      "tokens_seen": 41877504
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997271815446339,
      "loss": 3.3437,
      "theoretical_loss": 5.451618078575256,
      "tokens_seen": 41943040
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009972517552657974,
      "loss": 3.2631,
      "theoretical_loss": 5.450300659368612,
      "tokens_seen": 42008576
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009972316950852557,
      "loss": 3.3576,
      "theoretical_loss": 5.448985868266736,
      "tokens_seen": 42074112
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997211634904714,
      "loss": 3.5492,
      "theoretical_loss": 5.447673695945468,
      "tokens_seen": 42139648
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009971915747241725,
      "loss": 3.406,
      "theoretical_loss": 5.446364133128155,
      "tokens_seen": 42205184
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009971715145436308,
      "loss": 3.4232,
      "theoretical_loss": 5.4450571705853505,
      "tokens_seen": 42270720
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009971514543630892,
      "loss": 3.1882,
      "theoretical_loss": 5.443752799134492,
      "tokens_seen": 42336256
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009971313941825476,
      "loss": 3.4595,
      "theoretical_loss": 5.442451009639596,
      "tokens_seen": 42401792
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009971113340020061,
      "loss": 3.3603,
      "theoretical_loss": 5.44115179301095,
      "tokens_seen": 42467328
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009970912738214643,
      "loss": 3.5864,
      "theoretical_loss": 5.439855140204802,
      "tokens_seen": 42532864
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.49280866980552673,
      "objective/train/docs_used": 22938,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 3.297335386276245,
      "objective/train/original_loss": 3.297335624694824,
      "objective/train/theoretical_loss": 5.438561042223066,
      "objective/train/tokens_used": 63058400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24311578273773193,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505168437957764,
      "objective/train/weighted_lm_loss": 3.463775634765625,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0253151655197144,
      "theoretical_loss": 5.438561042223066,
      "tokens_seen": 42598400
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009970712136409229,
      "loss": 3.5201,
      "theoretical_loss": 5.438561042223066,
      "tokens_seen": 42598400
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997051153460381,
      "loss": 3.3525,
      "theoretical_loss": 5.437269490113014,
      "tokens_seen": 42663936
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009970310932798396,
      "loss": 3.3086,
      "theoretical_loss": 5.435980474966981,
      "tokens_seen": 42729472
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000997011033099298,
      "loss": 3.3753,
      "theoretical_loss": 5.434693987922072,
      "tokens_seen": 42795008
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009969909729187563,
      "loss": 3.3824,
      "theoretical_loss": 5.433410020159865,
      "tokens_seen": 42860544
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009969709127382147,
      "loss": 3.2654,
      "theoretical_loss": 5.432128562906122,
      "tokens_seen": 42926080
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996950852557673,
      "loss": 3.384,
      "theoretical_loss": 5.430849607430501,
      "tokens_seen": 42991616
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009969307923771314,
      "loss": 3.2841,
      "theoretical_loss": 5.429573145046266,
      "tokens_seen": 43057152
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009969107321965898,
      "loss": 3.422,
      "theoretical_loss": 5.428299167110012,
      "tokens_seen": 43122688
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009968906720160481,
      "loss": 3.3227,
      "theoretical_loss": 5.427027665021371,
      "tokens_seen": 43188224
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009968706118355065,
      "loss": 3.285,
      "theoretical_loss": 5.425758630222747,
      "tokens_seen": 43253760
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996850551654965,
      "loss": 3.3776,
      "theoretical_loss": 5.424492054199025,
      "tokens_seen": 43319296
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009968304914744232,
      "loss": 3.267,
      "theoretical_loss": 5.423227928477306,
      "tokens_seen": 43384832
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009968104312938818,
      "loss": 3.3353,
      "theoretical_loss": 5.421966244626631,
      "tokens_seen": 43450368
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.00099679037111334,
      "loss": 3.5885,
      "theoretical_loss": 5.42070699425771,
      "tokens_seen": 43515904
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009967703109327985,
      "loss": 3.4124,
      "theoretical_loss": 5.419450169022653,
      "tokens_seen": 43581440
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009967502507522569,
      "loss": 3.4499,
      "theoretical_loss": 5.418195760614707,
      "tokens_seen": 43646976
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009967301905717152,
      "loss": 3.395,
      "theoretical_loss": 5.416943760767989,
      "tokens_seen": 43712512
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009967101303911736,
      "loss": 3.4137,
      "theoretical_loss": 5.415694161257225,
      "tokens_seen": 43778048
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996690070210632,
      "loss": 3.3275,
      "theoretical_loss": 5.414446953897489,
      "tokens_seen": 43843584
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009966700100300903,
      "loss": 3.1889,
      "theoretical_loss": 5.413202130543947,
      "tokens_seen": 43909120
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009966499498495487,
      "loss": 3.2611,
      "theoretical_loss": 5.411959683091602,
      "tokens_seen": 43974656
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996629889669007,
      "loss": 3.339,
      "theoretical_loss": 5.410719603475034,
      "tokens_seen": 44040192
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009966098294884654,
      "loss": 3.5504,
      "theoretical_loss": 5.409481883668155,
      "tokens_seen": 44105728
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009965897693079238,
      "loss": 3.3063,
      "theoretical_loss": 5.408246515683956,
      "tokens_seen": 44171264
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009965697091273821,
      "loss": 3.145,
      "theoretical_loss": 5.407013491574263,
      "tokens_seen": 44236800
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009965496489468405,
      "loss": 3.3912,
      "theoretical_loss": 5.405782803429483,
      "tokens_seen": 44302336
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009965295887662989,
      "loss": 3.2364,
      "theoretical_loss": 5.404554443378366,
      "tokens_seen": 44367872
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009965095285857572,
      "loss": 3.3157,
      "theoretical_loss": 5.403328403587762,
      "tokens_seen": 44433408
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009964894684052158,
      "loss": 3.2918,
      "theoretical_loss": 5.402104676262381,
      "tokens_seen": 44498944
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996469408224674,
      "loss": 3.2743,
      "theoretical_loss": 5.400883253644551,
      "tokens_seen": 44564480
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009964493480441325,
      "loss": 3.3715,
      "theoretical_loss": 5.399664128013988,
      "tokens_seen": 44630016
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009964292878635907,
      "loss": 3.2805,
      "theoretical_loss": 5.398447291687555,
      "tokens_seen": 44695552
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009964092276830492,
      "loss": 3.3267,
      "theoretical_loss": 5.397232737019031,
      "tokens_seen": 44761088
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009963891675025076,
      "loss": 3.3381,
      "theoretical_loss": 5.396020456398885,
      "tokens_seen": 44826624
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996369107321966,
      "loss": 3.1136,
      "theoretical_loss": 5.394810442254032,
      "tokens_seen": 44892160
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009963490471414243,
      "loss": 3.2009,
      "theoretical_loss": 5.393602687047627,
      "tokens_seen": 44957696
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009963289869608827,
      "loss": 3.3298,
      "theoretical_loss": 5.392397183278822,
      "tokens_seen": 45023232
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996308926780341,
      "loss": 3.1392,
      "theoretical_loss": 5.391193923482547,
      "tokens_seen": 45088768
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009962888665997994,
      "loss": 3.3421,
      "theoretical_loss": 5.38999290022929,
      "tokens_seen": 45154304
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009962688064192578,
      "loss": 3.2676,
      "theoretical_loss": 5.388794106124874,
      "tokens_seen": 45219840
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009962487462387161,
      "loss": 3.2214,
      "theoretical_loss": 5.387597533810238,
      "tokens_seen": 45285376
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009962286860581745,
      "loss": 3.245,
      "theoretical_loss": 5.386403175961223,
      "tokens_seen": 45350912
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009962086258776329,
      "loss": 3.177,
      "theoretical_loss": 5.385211025288349,
      "tokens_seen": 45416448
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009961885656970912,
      "loss": 3.1048,
      "theoretical_loss": 5.38402107453661,
      "tokens_seen": 45481984
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009961685055165496,
      "loss": 3.2362,
      "theoretical_loss": 5.382833316485255,
      "tokens_seen": 45547520
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000996148445336008,
      "loss": 3.3837,
      "theoretical_loss": 5.381647743947578,
      "tokens_seen": 45613056
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009961283851554665,
      "loss": 3.3365,
      "theoretical_loss": 5.3804643497707145,
      "tokens_seen": 45678592
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009961083249749247,
      "loss": 3.2244,
      "theoretical_loss": 5.379283126835426,
      "tokens_seen": 45744128
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009960882647943833,
      "loss": 3.3835,
      "theoretical_loss": 5.378104068055899,
      "tokens_seen": 45809664
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.4905271530151367,
      "objective/train/docs_used": 24173,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.979590654373169,
      "objective/train/original_loss": 2.97959041595459,
      "objective/train/theoretical_loss": 5.37692716637954,
      "objective/train/tokens_used": 66335200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24144382774829865,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502800941467285,
      "objective/train/weighted_lm_loss": 3.129429817199707,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.37692716637954,
      "tokens_seen": 45875200
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009960682046138414,
      "loss": 3.1644,
      "theoretical_loss": 5.37692716637954,
      "tokens_seen": 45875200
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009960481444333,
      "loss": 3.2088,
      "theoretical_loss": 5.375752414786769,
      "tokens_seen": 45940736
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009960280842527583,
      "loss": 3.3534,
      "theoretical_loss": 5.374579806290823,
      "tokens_seen": 46006272
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009960080240722167,
      "loss": 3.1675,
      "theoretical_loss": 5.373409333937557,
      "tokens_seen": 46071808
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995987963891675,
      "loss": 3.1467,
      "theoretical_loss": 5.372240990805237,
      "tokens_seen": 46137344
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009959679037111334,
      "loss": 3.2232,
      "theoretical_loss": 5.37107477000435,
      "tokens_seen": 46202880
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009959478435305918,
      "loss": 3.2601,
      "theoretical_loss": 5.369910664677409,
      "tokens_seen": 46268416
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009959277833500502,
      "loss": 2.9964,
      "theoretical_loss": 5.368748667998759,
      "tokens_seen": 46333952
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009959077231695085,
      "loss": 3.1695,
      "theoretical_loss": 5.367588773174377,
      "tokens_seen": 46399488
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009958876629889669,
      "loss": 3.2412,
      "theoretical_loss": 5.366430973441693,
      "tokens_seen": 46465024
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009958676028084252,
      "loss": 3.4332,
      "theoretical_loss": 5.365275262069391,
      "tokens_seen": 46530560
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009958475426278836,
      "loss": 3.0728,
      "theoretical_loss": 5.364121632357221,
      "tokens_seen": 46596096
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995827482447342,
      "loss": 3.3624,
      "theoretical_loss": 5.36297007763582,
      "tokens_seen": 46661632
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009958074222668003,
      "loss": 3.2145,
      "theoretical_loss": 5.36182059126652,
      "tokens_seen": 46727168
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009957873620862587,
      "loss": 3.2506,
      "theoretical_loss": 5.360673166641163,
      "tokens_seen": 46792704
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009957673019057173,
      "loss": 3.229,
      "theoretical_loss": 5.359527797181926,
      "tokens_seen": 46858240
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009957472417251754,
      "loss": 3.2049,
      "theoretical_loss": 5.358384476341126,
      "tokens_seen": 46923776
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995727181544634,
      "loss": 3.129,
      "theoretical_loss": 5.357243197601056,
      "tokens_seen": 46989312
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009957071213640921,
      "loss": 3.2963,
      "theoretical_loss": 5.356103954473793,
      "tokens_seen": 47054848
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009956870611835507,
      "loss": 3.3261,
      "theoretical_loss": 5.35496674050103,
      "tokens_seen": 47120384
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995667001003009,
      "loss": 3.083,
      "theoretical_loss": 5.353831549253895,
      "tokens_seen": 47185920
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009956469408224674,
      "loss": 3.1944,
      "theoretical_loss": 5.352698374332774,
      "tokens_seen": 47251456
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009956268806419258,
      "loss": 3.2186,
      "theoretical_loss": 5.351567209367143,
      "tokens_seen": 47316992
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009956068204613842,
      "loss": 3.387,
      "theoretical_loss": 5.350438048015396,
      "tokens_seen": 47382528
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009955867602808425,
      "loss": 3.0156,
      "theoretical_loss": 5.349310883964664,
      "tokens_seen": 47448064
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995566700100301,
      "loss": 3.0316,
      "theoretical_loss": 5.348185710930661,
      "tokens_seen": 47513600
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009955466399197593,
      "loss": 3.2195,
      "theoretical_loss": 5.3470625226575,
      "tokens_seen": 47579136
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009955265797392176,
      "loss": 3.1434,
      "theoretical_loss": 5.345941312917536,
      "tokens_seen": 47644672
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995506519558676,
      "loss": 3.2259,
      "theoretical_loss": 5.344822075511196,
      "tokens_seen": 47710208
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009954864593781343,
      "loss": 3.0835,
      "theoretical_loss": 5.343704804266816,
      "tokens_seen": 47775744
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995466399197593,
      "loss": 3.1539,
      "theoretical_loss": 5.342589493040472,
      "tokens_seen": 47841280
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995446339017051,
      "loss": 3.1759,
      "theoretical_loss": 5.3414761357158245,
      "tokens_seen": 47906816
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009954262788365096,
      "loss": 3.0574,
      "theoretical_loss": 5.340364726203955,
      "tokens_seen": 47972352
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995406218655968,
      "loss": 3.0117,
      "theoretical_loss": 5.339255258443202,
      "tokens_seen": 48037888
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009953861584754264,
      "loss": 2.9506,
      "theoretical_loss": 5.338147726399006,
      "tokens_seen": 48103424
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009953660982948847,
      "loss": 2.8871,
      "theoretical_loss": 5.33704212406375,
      "tokens_seen": 48168960
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995346038114343,
      "loss": 3.272,
      "theoretical_loss": 5.3359384454566055,
      "tokens_seen": 48234496
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009953259779338015,
      "loss": 3.1474,
      "theoretical_loss": 5.334836684623371,
      "tokens_seen": 48300032
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009953059177532598,
      "loss": 3.013,
      "theoretical_loss": 5.333736835636325,
      "tokens_seen": 48365568
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009952858575727182,
      "loss": 2.9694,
      "theoretical_loss": 5.332638892594063,
      "tokens_seen": 48431104
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009952657973921765,
      "loss": 3.0434,
      "theoretical_loss": 5.331542849621357,
      "tokens_seen": 48496640
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995245737211635,
      "loss": 3.0976,
      "theoretical_loss": 5.330448700868995,
      "tokens_seen": 48562176
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009952256770310933,
      "loss": 3.132,
      "theoretical_loss": 5.329356440513633,
      "tokens_seen": 48627712
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009952056168505516,
      "loss": 3.1594,
      "theoretical_loss": 5.32826606275765,
      "tokens_seen": 48693248
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.00099518555667001,
      "loss": 3.0358,
      "theoretical_loss": 5.327177561828993,
      "tokens_seen": 48758784
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009951654964894684,
      "loss": 3.04,
      "theoretical_loss": 5.326090931981036,
      "tokens_seen": 48824320
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995145436308927,
      "loss": 3.1244,
      "theoretical_loss": 5.3250061674924325,
      "tokens_seen": 48889856
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.000995125376128385,
      "loss": 3.1184,
      "theoretical_loss": 5.323923262666968,
      "tokens_seen": 48955392
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009951053159478437,
      "loss": 3.13,
      "theoretical_loss": 5.32284221183342,
      "tokens_seen": 49020928
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009950852557673018,
      "loss": 3.1912,
      "theoretical_loss": 5.321763009345411,
      "tokens_seen": 49086464
    },
    {
      "epoch": 0.01,
      "objective/train/advantage_avg": 0.4820156991481781,
      "objective/train/docs_used": 25394,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.984027862548828,
      "objective/train/original_loss": 2.98402738571167,
      "objective/train/theoretical_loss": 5.3206856495812715,
      "objective/train/tokens_used": 69612000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2343626171350479,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0493927001953125,
      "objective/train/weighted_lm_loss": 3.1309571266174316,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.3206856495812715,
      "tokens_seen": 49152000
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009950651955867604,
      "loss": 3.0822,
      "theoretical_loss": 5.3206856495812715,
      "tokens_seen": 49152000
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009950451354062187,
      "loss": 3.0233,
      "theoretical_loss": 5.3196101269438945,
      "tokens_seen": 49217536
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009950250752256771,
      "loss": 2.9487,
      "theoretical_loss": 5.318536435860599,
      "tokens_seen": 49283072
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009950050150451355,
      "loss": 3.0769,
      "theoretical_loss": 5.317464570782989,
      "tokens_seen": 49348608
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009949849548645938,
      "loss": 3.0474,
      "theoretical_loss": 5.316394526186816,
      "tokens_seen": 49414144
    },
    {
      "epoch": 0.01,
      "learning_rate": 0.0009949648946840522,
      "loss": 2.7961,
      "theoretical_loss": 5.3153262965718415,
      "tokens_seen": 49479680
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009949448345035106,
      "loss": 2.9753,
      "theoretical_loss": 5.314259876461705,
      "tokens_seen": 49545216
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000994924774322969,
      "loss": 3.0748,
      "theoretical_loss": 5.3131952604037815,
      "tokens_seen": 49610752
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009949047141424273,
      "loss": 2.9606,
      "theoretical_loss": 5.312132442969055,
      "tokens_seen": 49676288
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009948846539618856,
      "loss": 2.9401,
      "theoretical_loss": 5.3110714187519825,
      "tokens_seen": 49741824
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000994864593781344,
      "loss": 2.9654,
      "theoretical_loss": 5.310012182370359,
      "tokens_seen": 49807360
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009948445336008024,
      "loss": 2.8686,
      "theoretical_loss": 5.308954728465188,
      "tokens_seen": 49872896
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009948244734202607,
      "loss": 3.0768,
      "theoretical_loss": 5.307899051700552,
      "tokens_seen": 49938432
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000994804413239719,
      "loss": 2.9913,
      "theoretical_loss": 5.306845146763486,
      "tokens_seen": 50003968
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009947843530591777,
      "loss": 2.9679,
      "theoretical_loss": 5.305793008363841,
      "tokens_seen": 50069504
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009947642928786358,
      "loss": 3.0225,
      "theoretical_loss": 5.304742631234162,
      "tokens_seen": 50135040
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009947442326980944,
      "loss": 3.04,
      "theoretical_loss": 5.3036940101295595,
      "tokens_seen": 50200576
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009947241725175525,
      "loss": 2.8716,
      "theoretical_loss": 5.302647139827585,
      "tokens_seen": 50266112
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009947041123370111,
      "loss": 2.9582,
      "theoretical_loss": 5.301602015128104,
      "tokens_seen": 50331648
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009946840521564695,
      "loss": 3.0407,
      "theoretical_loss": 5.300558630853171,
      "tokens_seen": 50397184
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009946639919759278,
      "loss": 2.9239,
      "theoretical_loss": 5.299516981846912,
      "tokens_seen": 50462720
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009946439317953862,
      "loss": 2.909,
      "theoretical_loss": 5.29847706297539,
      "tokens_seen": 50528256
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009946238716148446,
      "loss": 2.952,
      "theoretical_loss": 5.297438869126498,
      "tokens_seen": 50593792
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000994603811434303,
      "loss": 2.9815,
      "theoretical_loss": 5.296402395209828,
      "tokens_seen": 50659328
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009945837512537613,
      "loss": 2.9763,
      "theoretical_loss": 5.295367636156552,
      "tokens_seen": 50724864
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009945636910732197,
      "loss": 3.1347,
      "theoretical_loss": 5.294334586919307,
      "tokens_seen": 50790400
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000994543630892678,
      "loss": 3.0942,
      "theoretical_loss": 5.293303242472074,
      "tokens_seen": 50855936
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009945235707121364,
      "loss": 2.9039,
      "theoretical_loss": 5.29227359781006,
      "tokens_seen": 50921472
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009945035105315947,
      "loss": 3.0075,
      "theoretical_loss": 5.29124564794958,
      "tokens_seen": 50987008
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000994483450351053,
      "loss": 2.8655,
      "theoretical_loss": 5.290219387927946,
      "tokens_seen": 51052544
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009944633901705115,
      "loss": 2.8902,
      "theoretical_loss": 5.289194812803347,
      "tokens_seen": 51118080
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009944433299899698,
      "loss": 3.0833,
      "theoretical_loss": 5.288171917654733,
      "tokens_seen": 51183616
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009944232698094284,
      "loss": 2.801,
      "theoretical_loss": 5.287150697581708,
      "tokens_seen": 51249152
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009944032096288866,
      "loss": 2.7836,
      "theoretical_loss": 5.286131147704413,
      "tokens_seen": 51314688
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009943831494483451,
      "loss": 2.9264,
      "theoretical_loss": 5.285113263163414,
      "tokens_seen": 51380224
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009943630892678035,
      "loss": 2.9479,
      "theoretical_loss": 5.284097039119587,
      "tokens_seen": 51445760
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009943430290872619,
      "loss": 3.0413,
      "theoretical_loss": 5.283082470754016,
      "tokens_seen": 51511296
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009943229689067202,
      "loss": 2.8968,
      "theoretical_loss": 5.282069553267874,
      "tokens_seen": 51576832
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009943029087261786,
      "loss": 2.7193,
      "theoretical_loss": 5.2810582818823235,
      "tokens_seen": 51642368
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000994282848545637,
      "loss": 2.9009,
      "theoretical_loss": 5.2800486518383964,
      "tokens_seen": 51707904
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009942627883650953,
      "loss": 2.8084,
      "theoretical_loss": 5.279040658396896,
      "tokens_seen": 51773440
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009942427281845537,
      "loss": 2.9595,
      "theoretical_loss": 5.278034296838284,
      "tokens_seen": 51838976
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000994222668004012,
      "loss": 2.8929,
      "theoretical_loss": 5.27702956246258,
      "tokens_seen": 51904512
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009942026078234704,
      "loss": 2.8877,
      "theoretical_loss": 5.276026450589249,
      "tokens_seen": 51970048
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009941825476429288,
      "loss": 2.9187,
      "theoretical_loss": 5.275024956557102,
      "tokens_seen": 52035584
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009941624874623871,
      "loss": 2.7947,
      "theoretical_loss": 5.274025075724189,
      "tokens_seen": 52101120
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009941424272818455,
      "loss": 2.7595,
      "theoretical_loss": 5.273026803467695,
      "tokens_seen": 52166656
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009941223671013038,
      "loss": 2.9168,
      "theoretical_loss": 5.272030135183842,
      "tokens_seen": 52232192
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009941023069207622,
      "loss": 2.872,
      "theoretical_loss": 5.271035066287779,
      "tokens_seen": 52297728
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009940822467402206,
      "loss": 2.9237,
      "theoretical_loss": 5.27004159221349,
      "tokens_seen": 52363264
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.49426913261413574,
      "objective/train/docs_used": 26541,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.603318452835083,
      "objective/train/original_loss": 2.603318691253662,
      "objective/train/theoretical_loss": 5.269049708413682,
      "objective/train/tokens_used": 72888800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24451129138469696,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506700277328491,
      "objective/train/weighted_lm_loss": 2.735368251800537,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0304546356201172,
      "theoretical_loss": 5.269049708413682,
      "tokens_seen": 52428800
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009940621865596791,
      "loss": 2.7216,
      "theoretical_loss": 5.269049708413682,
      "tokens_seen": 52428800
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009940421263791373,
      "loss": 2.857,
      "theoretical_loss": 5.268059410359697,
      "tokens_seen": 52494336
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009940220661985959,
      "loss": 2.9196,
      "theoretical_loss": 5.267070693541403,
      "tokens_seen": 52559872
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009940020060180542,
      "loss": 2.8979,
      "theoretical_loss": 5.2660835534671,
      "tokens_seen": 52625408
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009939819458375126,
      "loss": 2.862,
      "theoretical_loss": 5.265097985663418,
      "tokens_seen": 52690944
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993961885656971,
      "loss": 2.9804,
      "theoretical_loss": 5.264113985675227,
      "tokens_seen": 52756480
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009939418254764293,
      "loss": 2.8927,
      "theoretical_loss": 5.263131549065529,
      "tokens_seen": 52822016
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009939217652958877,
      "loss": 2.786,
      "theoretical_loss": 5.262150671415373,
      "tokens_seen": 52887552
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993901705115346,
      "loss": 2.7832,
      "theoretical_loss": 5.261171348323755,
      "tokens_seen": 52953088
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009938816449348044,
      "loss": 2.9547,
      "theoretical_loss": 5.2601935754075155,
      "tokens_seen": 53018624
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009938615847542628,
      "loss": 2.8122,
      "theoretical_loss": 5.259217348301259,
      "tokens_seen": 53084160
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009938415245737211,
      "loss": 2.7562,
      "theoretical_loss": 5.25824266265725,
      "tokens_seen": 53149696
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009938214643931795,
      "loss": 2.8343,
      "theoretical_loss": 5.257269514145324,
      "tokens_seen": 53215232
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993801404212638,
      "loss": 2.7475,
      "theoretical_loss": 5.256297898452793,
      "tokens_seen": 53280768
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009937813440320962,
      "loss": 2.7761,
      "theoretical_loss": 5.2553278112843556,
      "tokens_seen": 53346304
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009937612838515548,
      "loss": 2.7001,
      "theoretical_loss": 5.254359248362006,
      "tokens_seen": 53411840
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993741223671013,
      "loss": 2.8198,
      "theoretical_loss": 5.2533922054249365,
      "tokens_seen": 53477376
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009937211634904715,
      "loss": 2.7642,
      "theoretical_loss": 5.25242667822946,
      "tokens_seen": 53542912
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009937011033099299,
      "loss": 2.8012,
      "theoretical_loss": 5.251462662548908,
      "tokens_seen": 53608448
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009936810431293882,
      "loss": 2.7885,
      "theoretical_loss": 5.25050015417355,
      "tokens_seen": 53673984
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009936609829488466,
      "loss": 2.8942,
      "theoretical_loss": 5.2495391489104986,
      "tokens_seen": 53739520
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993640922768305,
      "loss": 2.9782,
      "theoretical_loss": 5.248579642583628,
      "tokens_seen": 53805056
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009936208625877633,
      "loss": 2.8523,
      "theoretical_loss": 5.24762163103348,
      "tokens_seen": 53870592
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009936008024072217,
      "loss": 2.7695,
      "theoretical_loss": 5.246665110117186,
      "tokens_seen": 53936128
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00099358074222668,
      "loss": 2.8059,
      "theoretical_loss": 5.24571007570837,
      "tokens_seen": 54001664
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009935606820461384,
      "loss": 2.8622,
      "theoretical_loss": 5.244756523697072,
      "tokens_seen": 54067200
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009935406218655968,
      "loss": 2.8403,
      "theoretical_loss": 5.243804449989659,
      "tokens_seen": 54132736
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009935205616850551,
      "loss": 2.5313,
      "theoretical_loss": 5.242853850508741,
      "tokens_seen": 54198272
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009935005015045135,
      "loss": 2.8811,
      "theoretical_loss": 5.2419047211930865,
      "tokens_seen": 54263808
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009934804413239719,
      "loss": 2.7909,
      "theoretical_loss": 5.24095705799754,
      "tokens_seen": 54329344
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009934603811434302,
      "loss": 2.6954,
      "theoretical_loss": 5.240010856892939,
      "tokens_seen": 54394880
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009934403209628888,
      "loss": 2.7536,
      "theoretical_loss": 5.239066113866029,
      "tokens_seen": 54460416
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993420260782347,
      "loss": 2.7655,
      "theoretical_loss": 5.238122824919387,
      "tokens_seen": 54525952
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009934002006018055,
      "loss": 2.8289,
      "theoretical_loss": 5.237180986071335,
      "tokens_seen": 54591488
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993380140421264,
      "loss": 2.7578,
      "theoretical_loss": 5.236240593355863,
      "tokens_seen": 54657024
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009933600802407223,
      "loss": 2.6962,
      "theoretical_loss": 5.235301642822543,
      "tokens_seen": 54722560
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009933400200601806,
      "loss": 2.7935,
      "theoretical_loss": 5.234364130536457,
      "tokens_seen": 54788096
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993319959879639,
      "loss": 2.7993,
      "theoretical_loss": 5.2334280525781125,
      "tokens_seen": 54853632
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009932998996990973,
      "loss": 2.8282,
      "theoretical_loss": 5.232493405043362,
      "tokens_seen": 54919168
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009932798395185557,
      "loss": 2.8451,
      "theoretical_loss": 5.231560184043333,
      "tokens_seen": 54984704
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993259779338014,
      "loss": 2.7866,
      "theoretical_loss": 5.230628385704337,
      "tokens_seen": 55050240
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009932397191574724,
      "loss": 2.9565,
      "theoretical_loss": 5.229698006167809,
      "tokens_seen": 55115776
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009932196589769308,
      "loss": 2.5017,
      "theoretical_loss": 5.228769041590215,
      "tokens_seen": 55181312
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009931995987963892,
      "loss": 2.7794,
      "theoretical_loss": 5.227841488142982,
      "tokens_seen": 55246848
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009931795386158475,
      "loss": 2.6695,
      "theoretical_loss": 5.2269153420124255,
      "tokens_seen": 55312384
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009931594784353059,
      "loss": 2.7819,
      "theoretical_loss": 5.22599059939967,
      "tokens_seen": 55377920
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009931394182547642,
      "loss": 2.75,
      "theoretical_loss": 5.225067256520572,
      "tokens_seen": 55443456
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009931193580742226,
      "loss": 2.5643,
      "theoretical_loss": 5.224145309605651,
      "tokens_seen": 55508992
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000993099297893681,
      "loss": 2.7934,
      "theoretical_loss": 5.223224754900014,
      "tokens_seen": 55574528
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009930792377131395,
      "loss": 2.7763,
      "theoretical_loss": 5.222305588663275,
      "tokens_seen": 55640064
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.4954495131969452,
      "objective/train/docs_used": 27758,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.493486166000366,
      "objective/train/original_loss": 2.493486166000366,
      "objective/train/theoretical_loss": 5.221387807169494,
      "objective/train/tokens_used": 76165600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24568335711956024,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507941246032715,
      "objective/train/weighted_lm_loss": 2.6202352046966553,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.221387807169494,
      "tokens_seen": 55705600
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009930591775325977,
      "loss": 2.6668,
      "theoretical_loss": 5.221387807169494,
      "tokens_seen": 55705600
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009930391173520563,
      "loss": 2.9847,
      "theoretical_loss": 5.2204714067070945,
      "tokens_seen": 55771136
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009930190571715146,
      "loss": 2.6578,
      "theoretical_loss": 5.219556383578795,
      "tokens_seen": 55836672
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992998996990973,
      "loss": 2.7279,
      "theoretical_loss": 5.218642734101538,
      "tokens_seen": 55902208
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009929789368104314,
      "loss": 2.6868,
      "theoretical_loss": 5.217730454606417,
      "tokens_seen": 55967744
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009929588766298897,
      "loss": 2.79,
      "theoretical_loss": 5.216819541438607,
      "tokens_seen": 56033280
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992938816449348,
      "loss": 2.7179,
      "theoretical_loss": 5.215909990957291,
      "tokens_seen": 56098816
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009929187562688064,
      "loss": 2.7788,
      "theoretical_loss": 5.215001799535598,
      "tokens_seen": 56164352
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009928986960882648,
      "loss": 2.739,
      "theoretical_loss": 5.214094963560519,
      "tokens_seen": 56229888
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009928786359077232,
      "loss": 2.5606,
      "theoretical_loss": 5.213189479432854,
      "tokens_seen": 56295424
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009928585757271815,
      "loss": 2.734,
      "theoretical_loss": 5.212285343567135,
      "tokens_seen": 56360960
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00099283851554664,
      "loss": 2.5668,
      "theoretical_loss": 5.211382552391555,
      "tokens_seen": 56426496
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009928184553660983,
      "loss": 2.7951,
      "theoretical_loss": 5.210481102347908,
      "tokens_seen": 56492032
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009927983951855566,
      "loss": 2.6594,
      "theoretical_loss": 5.2095809898915135,
      "tokens_seen": 56557568
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992778335005015,
      "loss": 2.7474,
      "theoretical_loss": 5.208682211491157,
      "tokens_seen": 56623104
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009927582748244733,
      "loss": 2.687,
      "theoretical_loss": 5.207784763629014,
      "tokens_seen": 56688640
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009927382146439317,
      "loss": 2.5976,
      "theoretical_loss": 5.206888642800598,
      "tokens_seen": 56754176
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009927181544633903,
      "loss": 2.6364,
      "theoretical_loss": 5.205993845514678,
      "tokens_seen": 56819712
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009926980942828484,
      "loss": 2.6975,
      "theoretical_loss": 5.205100368293225,
      "tokens_seen": 56885248
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992678034102307,
      "loss": 2.6616,
      "theoretical_loss": 5.204208207671343,
      "tokens_seen": 56950784
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009926579739217654,
      "loss": 2.6684,
      "theoretical_loss": 5.203317360197204,
      "tokens_seen": 57016320
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009926379137412237,
      "loss": 2.629,
      "theoretical_loss": 5.202427822431982,
      "tokens_seen": 57081856
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992617853560682,
      "loss": 2.6768,
      "theoretical_loss": 5.201539590949796,
      "tokens_seen": 57147392
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009925977933801405,
      "loss": 2.5754,
      "theoretical_loss": 5.200652662337637,
      "tokens_seen": 57212928
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009925777331995988,
      "loss": 2.5187,
      "theoretical_loss": 5.199767033195316,
      "tokens_seen": 57278464
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009925576730190572,
      "loss": 2.7063,
      "theoretical_loss": 5.198882700135387,
      "tokens_seen": 57344000
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009925376128385155,
      "loss": 2.7405,
      "theoretical_loss": 5.1979996597831,
      "tokens_seen": 57409536
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992517552657974,
      "loss": 2.6938,
      "theoretical_loss": 5.1971179087763275,
      "tokens_seen": 57475072
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009924974924774323,
      "loss": 2.6516,
      "theoretical_loss": 5.19623744376551,
      "tokens_seen": 57540608
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009924774322968906,
      "loss": 2.6739,
      "theoretical_loss": 5.195358261413589,
      "tokens_seen": 57606144
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009924573721163492,
      "loss": 2.5905,
      "theoretical_loss": 5.1944803583959525,
      "tokens_seen": 57671680
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009924373119358074,
      "loss": 2.6303,
      "theoretical_loss": 5.193603731400369,
      "tokens_seen": 57737216
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992417251755266,
      "loss": 2.6194,
      "theoretical_loss": 5.192728377126931,
      "tokens_seen": 57802752
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009923971915747243,
      "loss": 2.7367,
      "theoretical_loss": 5.191854292287992,
      "tokens_seen": 57868288
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009923771313941827,
      "loss": 2.7908,
      "theoretical_loss": 5.190981473608112,
      "tokens_seen": 57933824
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992357071213641,
      "loss": 2.5301,
      "theoretical_loss": 5.1901099178239924,
      "tokens_seen": 57999360
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009923370110330994,
      "loss": 2.5255,
      "theoretical_loss": 5.189239621684421,
      "tokens_seen": 58064896
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009923169508525577,
      "loss": 2.7432,
      "theoretical_loss": 5.188370581950215,
      "tokens_seen": 58130432
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009922968906720161,
      "loss": 2.7216,
      "theoretical_loss": 5.18750279539416,
      "tokens_seen": 58195968
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009922768304914745,
      "loss": 2.493,
      "theoretical_loss": 5.1866362588009505,
      "tokens_seen": 58261504
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009922567703109328,
      "loss": 2.6899,
      "theoretical_loss": 5.185770968967139,
      "tokens_seen": 58327040
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009922367101303912,
      "loss": 2.6763,
      "theoretical_loss": 5.184906922701076,
      "tokens_seen": 58392576
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009922166499498496,
      "loss": 2.5682,
      "theoretical_loss": 5.184044116822849,
      "tokens_seen": 58458112
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992196589769308,
      "loss": 2.5105,
      "theoretical_loss": 5.183182548164234,
      "tokens_seen": 58523648
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009921765295887663,
      "loss": 2.5142,
      "theoretical_loss": 5.182322213568634,
      "tokens_seen": 58589184
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009921564694082246,
      "loss": 2.7216,
      "theoretical_loss": 5.181463109891023,
      "tokens_seen": 58654720
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992136409227683,
      "loss": 2.8247,
      "theoretical_loss": 5.1806052339978965,
      "tokens_seen": 58720256
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009921163490471414,
      "loss": 2.5839,
      "theoretical_loss": 5.179748582767209,
      "tokens_seen": 58785792
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009920962888666,
      "loss": 2.6148,
      "theoretical_loss": 5.178893153088326,
      "tokens_seen": 58851328
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992076228686058,
      "loss": 2.5025,
      "theoretical_loss": 5.178038941861966,
      "tokens_seen": 58916864
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.4918819069862366,
      "objective/train/docs_used": 29071,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.2515406608581543,
      "objective/train/original_loss": 2.2515406608581543,
      "objective/train/theoretical_loss": 5.1771859460001455,
      "objective/train/tokens_used": 79442400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24229572713375092,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504199266433716,
      "objective/train/weighted_lm_loss": 2.365377187728882,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.1771859460001455,
      "tokens_seen": 58982400
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009920561685055167,
      "loss": 2.4743,
      "theoretical_loss": 5.1771859460001455,
      "tokens_seen": 58982400
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000992036108324975,
      "loss": 2.5575,
      "theoretical_loss": 5.176334162426128,
      "tokens_seen": 59047936
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009920160481444334,
      "loss": 2.6685,
      "theoretical_loss": 5.175483588074371,
      "tokens_seen": 59113472
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009919959879638918,
      "loss": 2.5362,
      "theoretical_loss": 5.17463421989047,
      "tokens_seen": 59179008
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009919759277833501,
      "loss": 2.6476,
      "theoretical_loss": 5.1737860548311065,
      "tokens_seen": 59244544
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009919558676028085,
      "loss": 2.5998,
      "theoretical_loss": 5.172939089863998,
      "tokens_seen": 59310080
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009919358074222668,
      "loss": 2.6016,
      "theoretical_loss": 5.172093321967843,
      "tokens_seen": 59375616
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009919157472417252,
      "loss": 2.432,
      "theoretical_loss": 5.171248748132273,
      "tokens_seen": 59441152
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009918956870611836,
      "loss": 2.7428,
      "theoretical_loss": 5.170405365357794,
      "tokens_seen": 59506688
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000991875626880642,
      "loss": 2.5627,
      "theoretical_loss": 5.169563170655742,
      "tokens_seen": 59572224
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009918555667001003,
      "loss": 2.5943,
      "theoretical_loss": 5.16872216104823,
      "tokens_seen": 59637760
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009918355065195587,
      "loss": 2.4533,
      "theoretical_loss": 5.167882333568095,
      "tokens_seen": 59703296
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000991815446339017,
      "loss": 2.6007,
      "theoretical_loss": 5.167043685258852,
      "tokens_seen": 59768832
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009917953861584754,
      "loss": 2.5313,
      "theoretical_loss": 5.16620621317464,
      "tokens_seen": 59834368
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009917753259779337,
      "loss": 2.5483,
      "theoretical_loss": 5.1653699143801735,
      "tokens_seen": 59899904
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000991755265797392,
      "loss": 2.6718,
      "theoretical_loss": 5.164534785950695,
      "tokens_seen": 59965440
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009917352056168507,
      "loss": 2.4902,
      "theoretical_loss": 5.163700824971922,
      "tokens_seen": 60030976
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009917151454363088,
      "loss": 2.7512,
      "theoretical_loss": 5.162868028540002,
      "tokens_seen": 60096512
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009916950852557674,
      "loss": 2.6651,
      "theoretical_loss": 5.1620363937614595,
      "tokens_seen": 60162048
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009916750250752258,
      "loss": 2.6301,
      "theoretical_loss": 5.161205917753154,
      "tokens_seen": 60227584
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009916549648946841,
      "loss": 2.5769,
      "theoretical_loss": 5.160376597642223,
      "tokens_seen": 60293120
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009916349047141425,
      "loss": 2.5911,
      "theoretical_loss": 5.159548430566044,
      "tokens_seen": 60358656
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009916148445336009,
      "loss": 2.6131,
      "theoretical_loss": 5.158721413672179,
      "tokens_seen": 60424192
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009915947843530592,
      "loss": 2.5863,
      "theoretical_loss": 5.1578955441183325,
      "tokens_seen": 60489728
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009915747241725176,
      "loss": 2.5351,
      "theoretical_loss": 5.157070819072301,
      "tokens_seen": 60555264
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000991554663991976,
      "loss": 2.7029,
      "theoretical_loss": 5.156247235711929,
      "tokens_seen": 60620800
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009915346038114343,
      "loss": 2.6004,
      "theoretical_loss": 5.155424791225061,
      "tokens_seen": 60686336
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009915145436308927,
      "loss": 2.6118,
      "theoretical_loss": 5.1546034828094935,
      "tokens_seen": 60751872
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000991494483450351,
      "loss": 2.5272,
      "theoretical_loss": 5.153783307672935,
      "tokens_seen": 60817408
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009914744232698094,
      "loss": 2.6638,
      "theoretical_loss": 5.152964263032954,
      "tokens_seen": 60882944
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009914543630892678,
      "loss": 2.5694,
      "theoretical_loss": 5.152146346116936,
      "tokens_seen": 60948480
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009914343029087261,
      "loss": 2.5069,
      "theoretical_loss": 5.151329554162039,
      "tokens_seen": 61014016
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009914142427281847,
      "loss": 2.6197,
      "theoretical_loss": 5.150513884415149,
      "tokens_seen": 61079552
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009913941825476428,
      "loss": 2.6271,
      "theoretical_loss": 5.1496993341328325,
      "tokens_seen": 61145088
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009913741223671014,
      "loss": 2.5083,
      "theoretical_loss": 5.148885900581297,
      "tokens_seen": 61210624
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009913540621865596,
      "loss": 2.6221,
      "theoretical_loss": 5.148073581036338,
      "tokens_seen": 61276160
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009913340020060181,
      "loss": 2.5627,
      "theoretical_loss": 5.14726237278331,
      "tokens_seen": 61341696
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009913139418254765,
      "loss": 2.5517,
      "theoretical_loss": 5.146452273117065,
      "tokens_seen": 61407232
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009912938816449349,
      "loss": 2.7035,
      "theoretical_loss": 5.1456432793419244,
      "tokens_seen": 61472768
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009912738214643932,
      "loss": 2.5791,
      "theoretical_loss": 5.144835388771626,
      "tokens_seen": 61538304
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009912537612838516,
      "loss": 2.6198,
      "theoretical_loss": 5.144028598729285,
      "tokens_seen": 61603840
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00099123370110331,
      "loss": 2.6126,
      "theoretical_loss": 5.1432229065473525,
      "tokens_seen": 61669376
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009912136409227683,
      "loss": 2.48,
      "theoretical_loss": 5.142418309567569,
      "tokens_seen": 61734912
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009911935807422267,
      "loss": 2.5514,
      "theoretical_loss": 5.141614805140927,
      "tokens_seen": 61800448
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000991173520561685,
      "loss": 2.6091,
      "theoretical_loss": 5.140812390627624,
      "tokens_seen": 61865984
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009911534603811434,
      "loss": 2.5793,
      "theoretical_loss": 5.1400110633970275,
      "tokens_seen": 61931520
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009911334002006018,
      "loss": 2.7038,
      "theoretical_loss": 5.139210820827625,
      "tokens_seen": 61997056
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009911133400200601,
      "loss": 2.5758,
      "theoretical_loss": 5.138411660306989,
      "tokens_seen": 62062592
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009910932798395185,
      "loss": 2.4817,
      "theoretical_loss": 5.137613579231737,
      "tokens_seen": 62128128
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009910732196589769,
      "loss": 2.6186,
      "theoretical_loss": 5.136816575007483,
      "tokens_seen": 62193664
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.4942206144332886,
      "objective/train/docs_used": 30276,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.677966356277466,
      "objective/train/original_loss": 2.677966594696045,
      "objective/train/theoretical_loss": 5.136020645048806,
      "objective/train/tokens_used": 82719200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24524354934692383,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506689548492432,
      "objective/train/weighted_lm_loss": 2.8136417865753174,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.136020645048806,
      "tokens_seen": 62259200
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009910531594784354,
      "loss": 2.6662,
      "theoretical_loss": 5.136020645048806,
      "tokens_seen": 62259200
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009910330992978936,
      "loss": 2.4777,
      "theoretical_loss": 5.135225786779204,
      "tokens_seen": 62324736
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009910130391173522,
      "loss": 2.6514,
      "theoretical_loss": 5.134431997631053,
      "tokens_seen": 62390272
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009909929789368103,
      "loss": 2.4079,
      "theoretical_loss": 5.1336392750455735,
      "tokens_seen": 62455808
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009909729187562689,
      "loss": 2.5303,
      "theoretical_loss": 5.132847616472784,
      "tokens_seen": 62521344
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009909528585757272,
      "loss": 2.5487,
      "theoretical_loss": 5.132057019371466,
      "tokens_seen": 62586880
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009909327983951856,
      "loss": 2.4935,
      "theoretical_loss": 5.1312674812091235,
      "tokens_seen": 62652416
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990912738214644,
      "loss": 2.4875,
      "theoretical_loss": 5.130478999461939,
      "tokens_seen": 62717952
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009908926780341023,
      "loss": 2.5383,
      "theoretical_loss": 5.129691571614747,
      "tokens_seen": 62783488
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009908726178535607,
      "loss": 2.4761,
      "theoretical_loss": 5.128905195160981,
      "tokens_seen": 62849024
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990852557673019,
      "loss": 2.589,
      "theoretical_loss": 5.128119867602646,
      "tokens_seen": 62914560
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009908324974924774,
      "loss": 2.601,
      "theoretical_loss": 5.127335586450272,
      "tokens_seen": 62980096
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009908124373119358,
      "loss": 2.634,
      "theoretical_loss": 5.126552349222884,
      "tokens_seen": 63045632
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009907923771313941,
      "loss": 2.5431,
      "theoretical_loss": 5.12577015344796,
      "tokens_seen": 63111168
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009907723169508525,
      "loss": 2.5072,
      "theoretical_loss": 5.124988996661393,
      "tokens_seen": 63176704
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990752256770311,
      "loss": 2.5105,
      "theoretical_loss": 5.124208876407453,
      "tokens_seen": 63242240
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009907321965897692,
      "loss": 2.4315,
      "theoretical_loss": 5.1234297902387524,
      "tokens_seen": 63307776
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009907121364092278,
      "loss": 2.4456,
      "theoretical_loss": 5.12265173571621,
      "tokens_seen": 63373312
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009906920762286862,
      "loss": 2.5354,
      "theoretical_loss": 5.121874710409012,
      "tokens_seen": 63438848
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009906720160481445,
      "loss": 2.526,
      "theoretical_loss": 5.121098711894572,
      "tokens_seen": 63504384
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990651955867603,
      "loss": 2.4092,
      "theoretical_loss": 5.120323737758502,
      "tokens_seen": 63569920
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009906318956870613,
      "loss": 2.443,
      "theoretical_loss": 5.119549785594572,
      "tokens_seen": 63635456
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009906118355065196,
      "loss": 2.5223,
      "theoretical_loss": 5.118776853004677,
      "tokens_seen": 63700992
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990591775325978,
      "loss": 2.5233,
      "theoretical_loss": 5.118004937598793,
      "tokens_seen": 63766528
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009905717151454363,
      "loss": 2.4831,
      "theoretical_loss": 5.1172340369949545,
      "tokens_seen": 63832064
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009905516549648947,
      "loss": 2.5658,
      "theoretical_loss": 5.116464148819206,
      "tokens_seen": 63897600
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990531594784353,
      "loss": 2.5614,
      "theoretical_loss": 5.115695270705579,
      "tokens_seen": 63963136
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009905115346038114,
      "loss": 2.3751,
      "theoretical_loss": 5.114927400296046,
      "tokens_seen": 64028672
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009904914744232698,
      "loss": 2.4838,
      "theoretical_loss": 5.114160535240492,
      "tokens_seen": 64094208
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009904714142427282,
      "loss": 2.5281,
      "theoretical_loss": 5.113394673196682,
      "tokens_seen": 64159744
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009904513540621865,
      "loss": 2.4431,
      "theoretical_loss": 5.112629811830217,
      "tokens_seen": 64225280
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990431293881645,
      "loss": 2.4407,
      "theoretical_loss": 5.111865948814511,
      "tokens_seen": 64290816
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009904112337011032,
      "loss": 2.4222,
      "theoretical_loss": 5.111103081830752,
      "tokens_seen": 64356352
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009903911735205618,
      "loss": 2.545,
      "theoretical_loss": 5.1103412085678634,
      "tokens_seen": 64421888
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00099037111334002,
      "loss": 2.5683,
      "theoretical_loss": 5.10958032672248,
      "tokens_seen": 64487424
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009903510531594785,
      "loss": 2.572,
      "theoretical_loss": 5.108820433998908,
      "tokens_seen": 64552960
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990330992978937,
      "loss": 2.381,
      "theoretical_loss": 5.108061528109092,
      "tokens_seen": 64618496
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009903109327983953,
      "loss": 2.4737,
      "theoretical_loss": 5.107303606772582,
      "tokens_seen": 64684032
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009902908726178536,
      "loss": 2.4322,
      "theoretical_loss": 5.106546667716508,
      "tokens_seen": 64749568
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990270812437312,
      "loss": 2.592,
      "theoretical_loss": 5.1057907086755305,
      "tokens_seen": 64815104
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009902507522567704,
      "loss": 2.5258,
      "theoretical_loss": 5.105035727391828,
      "tokens_seen": 64880640
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009902306920762287,
      "loss": 2.6152,
      "theoretical_loss": 5.104281721615046,
      "tokens_seen": 64946176
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990210631895687,
      "loss": 2.5266,
      "theoretical_loss": 5.103528689102281,
      "tokens_seen": 65011712
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009901905717151454,
      "loss": 2.5043,
      "theoretical_loss": 5.102776627618036,
      "tokens_seen": 65077248
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009901705115346038,
      "loss": 2.5268,
      "theoretical_loss": 5.102025534934196,
      "tokens_seen": 65142784
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009901504513540622,
      "loss": 2.5599,
      "theoretical_loss": 5.10127540882999,
      "tokens_seen": 65208320
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009901303911735205,
      "loss": 2.403,
      "theoretical_loss": 5.100526247091967,
      "tokens_seen": 65273856
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990110330992979,
      "loss": 2.4966,
      "theoretical_loss": 5.099778047513957,
      "tokens_seen": 65339392
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009900902708124373,
      "loss": 2.5187,
      "theoretical_loss": 5.099030807897048,
      "tokens_seen": 65404928
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009900702106318958,
      "loss": 2.5642,
      "theoretical_loss": 5.098284526049547,
      "tokens_seen": 65470464
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.49179795384407043,
      "objective/train/docs_used": 31526,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.5606729984283447,
      "objective/train/original_loss": 2.5606727600097656,
      "objective/train/theoretical_loss": 5.097539199786951,
      "objective/train/tokens_used": 85996000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24245329201221466,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504124164581299,
      "objective/train/weighted_lm_loss": 2.6897695064544678,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9753099083900452,
      "theoretical_loss": 5.097539199786951,
      "tokens_seen": 65536000
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000990050150451354,
      "loss": 2.5972,
      "theoretical_loss": 5.097539199786951,
      "tokens_seen": 65536000
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009900300902708126,
      "loss": 2.5759,
      "theoretical_loss": 5.096794826931921,
      "tokens_seen": 65601536
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009900100300902707,
      "loss": 2.5983,
      "theoretical_loss": 5.096051405314247,
      "tokens_seen": 65667072
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009899899699097293,
      "loss": 2.6292,
      "theoretical_loss": 5.095308932770816,
      "tokens_seen": 65732608
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009899699097291876,
      "loss": 2.4965,
      "theoretical_loss": 5.094567407145588,
      "tokens_seen": 65798144
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989949849548646,
      "loss": 2.4289,
      "theoretical_loss": 5.093826826289559,
      "tokens_seen": 65863680
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009899297893681044,
      "loss": 2.5533,
      "theoretical_loss": 5.093087188060739,
      "tokens_seen": 65929216
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009899097291875627,
      "loss": 2.4959,
      "theoretical_loss": 5.09234849032411,
      "tokens_seen": 65994752
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989889669007021,
      "loss": 2.4689,
      "theoretical_loss": 5.09161073095161,
      "tokens_seen": 66060288
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009898696088264795,
      "loss": 2.4506,
      "theoretical_loss": 5.090873907822093,
      "tokens_seen": 66125824
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009898495486459378,
      "loss": 2.408,
      "theoretical_loss": 5.090138018821309,
      "tokens_seen": 66191360
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009898294884653962,
      "loss": 2.4846,
      "theoretical_loss": 5.089403061841864,
      "tokens_seen": 66256896
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009898094282848545,
      "loss": 2.502,
      "theoretical_loss": 5.0886690347832015,
      "tokens_seen": 66322432
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989789368104313,
      "loss": 2.4468,
      "theoretical_loss": 5.087935935551566,
      "tokens_seen": 66387968
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009897693079237713,
      "loss": 2.4499,
      "theoretical_loss": 5.0872037620599775,
      "tokens_seen": 66453504
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009897492477432296,
      "loss": 2.4336,
      "theoretical_loss": 5.0864725122282035,
      "tokens_seen": 66519040
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989729187562688,
      "loss": 2.5683,
      "theoretical_loss": 5.0857421839827275,
      "tokens_seen": 66584576
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009897091273821466,
      "loss": 2.4402,
      "theoretical_loss": 5.085012775256727,
      "tokens_seen": 66650112
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009896890672016047,
      "loss": 2.441,
      "theoretical_loss": 5.084284283990035,
      "tokens_seen": 66715648
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009896690070210633,
      "loss": 2.4647,
      "theoretical_loss": 5.083556708129126,
      "tokens_seen": 66781184
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009896489468405214,
      "loss": 2.3144,
      "theoretical_loss": 5.082830045627072,
      "tokens_seen": 66846720
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00098962888665998,
      "loss": 2.4836,
      "theoretical_loss": 5.08210429444353,
      "tokens_seen": 66912256
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009896088264794384,
      "loss": 2.4112,
      "theoretical_loss": 5.081379452544702,
      "tokens_seen": 66977792
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009895887662988967,
      "loss": 2.5422,
      "theoretical_loss": 5.080655517903319,
      "tokens_seen": 67043328
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989568706118355,
      "loss": 2.4475,
      "theoretical_loss": 5.079932488498602,
      "tokens_seen": 67108864
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009895486459378135,
      "loss": 2.6382,
      "theoretical_loss": 5.079210362316245,
      "tokens_seen": 67174400
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009895285857572718,
      "loss": 2.4146,
      "theoretical_loss": 5.078489137348383,
      "tokens_seen": 67239936
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009895085255767302,
      "loss": 2.4907,
      "theoretical_loss": 5.077768811593563,
      "tokens_seen": 67305472
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009894884653961886,
      "loss": 2.3637,
      "theoretical_loss": 5.077049383056725,
      "tokens_seen": 67371008
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989468405215647,
      "loss": 2.3986,
      "theoretical_loss": 5.076330849749166,
      "tokens_seen": 67436544
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009894483450351055,
      "loss": 2.5128,
      "theoretical_loss": 5.0756132096885205,
      "tokens_seen": 67502080
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009894282848545636,
      "loss": 2.6255,
      "theoretical_loss": 5.0748964608987315,
      "tokens_seen": 67567616
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009894082246740222,
      "loss": 2.5026,
      "theoretical_loss": 5.074180601410026,
      "tokens_seen": 67633152
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009893881644934804,
      "loss": 2.3306,
      "theoretical_loss": 5.073465629258886,
      "tokens_seen": 67698688
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989368104312939,
      "loss": 2.4388,
      "theoretical_loss": 5.072751542488024,
      "tokens_seen": 67764224
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009893480441323973,
      "loss": 2.4484,
      "theoretical_loss": 5.072038339146357,
      "tokens_seen": 67829760
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009893279839518557,
      "loss": 2.3594,
      "theoretical_loss": 5.0713260172889845,
      "tokens_seen": 67895296
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989307923771314,
      "loss": 2.4074,
      "theoretical_loss": 5.070614574977157,
      "tokens_seen": 67960832
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009892878635907724,
      "loss": 2.4488,
      "theoretical_loss": 5.069904010278254,
      "tokens_seen": 68026368
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009892678034102308,
      "loss": 2.3926,
      "theoretical_loss": 5.069194321265758,
      "tokens_seen": 68091904
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009892477432296891,
      "loss": 2.2627,
      "theoretical_loss": 5.068485506019231,
      "tokens_seen": 68157440
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009892276830491475,
      "loss": 2.3413,
      "theoretical_loss": 5.0677775626242845,
      "tokens_seen": 68222976
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009892076228686058,
      "loss": 2.4958,
      "theoretical_loss": 5.067070489172562,
      "tokens_seen": 68288512
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009891875626880642,
      "loss": 2.3923,
      "theoretical_loss": 5.066364283761709,
      "tokens_seen": 68354048
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009891675025075226,
      "loss": 2.4773,
      "theoretical_loss": 5.06565894449535,
      "tokens_seen": 68419584
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989147442326981,
      "loss": 2.3944,
      "theoretical_loss": 5.064954469483064,
      "tokens_seen": 68485120
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009891273821464393,
      "loss": 2.4673,
      "theoretical_loss": 5.064250856840359,
      "tokens_seen": 68550656
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009891073219658977,
      "loss": 2.5153,
      "theoretical_loss": 5.06354810468865,
      "tokens_seen": 68616192
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009890872617853562,
      "loss": 2.367,
      "theoretical_loss": 5.06284621115523,
      "tokens_seen": 68681728
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009890672016048144,
      "loss": 2.4625,
      "theoretical_loss": 5.062145174373255,
      "tokens_seen": 68747264
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.49227413535118103,
      "objective/train/docs_used": 32629,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.2441399097442627,
      "objective/train/original_loss": 2.2441399097442627,
      "objective/train/theoretical_loss": 5.061444992481711,
      "objective/train/tokens_used": 89272800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24276849627494812,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504614114761353,
      "objective/train/weighted_lm_loss": 2.3572661876678467,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.061444992481711,
      "tokens_seen": 68812800
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989047141424273,
      "loss": 2.4288,
      "theoretical_loss": 5.061444992481711,
      "tokens_seen": 68812800
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000989027081243731,
      "loss": 2.5382,
      "theoretical_loss": 5.060745663625397,
      "tokens_seen": 68878336
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009890070210631897,
      "loss": 2.5068,
      "theoretical_loss": 5.060047185954893,
      "tokens_seen": 68943872
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000988986960882648,
      "loss": 2.3632,
      "theoretical_loss": 5.059349557626547,
      "tokens_seen": 69009408
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009889669007021064,
      "loss": 2.346,
      "theoretical_loss": 5.058652776802443,
      "tokens_seen": 69074944
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009889468405215648,
      "loss": 2.4677,
      "theoretical_loss": 5.057956841650385,
      "tokens_seen": 69140480
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009889267803410231,
      "loss": 2.471,
      "theoretical_loss": 5.057261750343864,
      "tokens_seen": 69206016
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009889067201604815,
      "loss": 2.3835,
      "theoretical_loss": 5.0565675010620454,
      "tokens_seen": 69271552
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009888866599799399,
      "loss": 2.319,
      "theoretical_loss": 5.05587409198974,
      "tokens_seen": 69337088
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009888665997993982,
      "loss": 2.5456,
      "theoretical_loss": 5.055181521317383,
      "tokens_seen": 69402624
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009888465396188566,
      "loss": 2.4407,
      "theoretical_loss": 5.0544897872410095,
      "tokens_seen": 69468160
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000988826479438315,
      "loss": 2.3936,
      "theoretical_loss": 5.053798887962236,
      "tokens_seen": 69533696
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009888064192577733,
      "loss": 2.3531,
      "theoretical_loss": 5.053108821688232,
      "tokens_seen": 69599232
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009887863590772317,
      "loss": 2.4097,
      "theoretical_loss": 5.052419586631703,
      "tokens_seen": 69664768
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00098876629889669,
      "loss": 2.4689,
      "theoretical_loss": 5.051731181010866,
      "tokens_seen": 69730304
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009887462387161484,
      "loss": 2.3541,
      "theoretical_loss": 5.051043603049426,
      "tokens_seen": 69795840
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000988726178535607,
      "loss": 2.3945,
      "theoretical_loss": 5.0503568509765575,
      "tokens_seen": 69861376
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009887061183550651,
      "loss": 2.3833,
      "theoretical_loss": 5.049670923026879,
      "tokens_seen": 69926912
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009886860581745237,
      "loss": 2.3977,
      "theoretical_loss": 5.048985817440432,
      "tokens_seen": 69992448
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009886659979939818,
      "loss": 2.4533,
      "theoretical_loss": 5.048301532462662,
      "tokens_seen": 70057984
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009886459378134404,
      "loss": 2.5417,
      "theoretical_loss": 5.047618066344392,
      "tokens_seen": 70123520
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009886258776328988,
      "loss": 2.3964,
      "theoretical_loss": 5.046935417341806,
      "tokens_seen": 70189056
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009886058174523571,
      "loss": 2.4526,
      "theoretical_loss": 5.046253583716425,
      "tokens_seen": 70254592
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009885857572718155,
      "loss": 2.425,
      "theoretical_loss": 5.045572563735084,
      "tokens_seen": 70320128
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009885656970912739,
      "loss": 2.3847,
      "theoretical_loss": 5.0448923556699174,
      "tokens_seen": 70385664
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009885456369107322,
      "loss": 2.458,
      "theoretical_loss": 5.044212957798328,
      "tokens_seen": 70451200
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009885255767301906,
      "loss": 2.3902,
      "theoretical_loss": 5.043534368402973,
      "tokens_seen": 70516736
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000988505516549649,
      "loss": 2.2822,
      "theoretical_loss": 5.042856585771746,
      "tokens_seen": 70582272
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009884854563691073,
      "loss": 2.3476,
      "theoretical_loss": 5.0421796081977455,
      "tokens_seen": 70647808
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009884653961885657,
      "loss": 2.4328,
      "theoretical_loss": 5.041503433979264,
      "tokens_seen": 70713344
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000988445336008024,
      "loss": 2.5019,
      "theoretical_loss": 5.040828061419762,
      "tokens_seen": 70778880
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009884252758274824,
      "loss": 2.4293,
      "theoretical_loss": 5.040153488827851,
      "tokens_seen": 70844416
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009884052156469408,
      "loss": 2.5365,
      "theoretical_loss": 5.039479714517272,
      "tokens_seen": 70909952
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009883851554663991,
      "loss": 2.4693,
      "theoretical_loss": 5.038806736806871,
      "tokens_seen": 70975488
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009883650952858577,
      "loss": 2.3275,
      "theoretical_loss": 5.038134554020587,
      "tokens_seen": 71041024
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009883450351053159,
      "loss": 2.3314,
      "theoretical_loss": 5.037463164487425,
      "tokens_seen": 71106560
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009883249749247744,
      "loss": 2.3667,
      "theoretical_loss": 5.036792566541441,
      "tokens_seen": 71172096
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009883049147442326,
      "loss": 2.3339,
      "theoretical_loss": 5.036122758521715,
      "tokens_seen": 71237632
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009882848545636912,
      "loss": 2.5154,
      "theoretical_loss": 5.03545373877234,
      "tokens_seen": 71303168
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009882647943831495,
      "loss": 2.4845,
      "theoretical_loss": 5.034785505642397,
      "tokens_seen": 71368704
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009882447342026079,
      "loss": 2.4743,
      "theoretical_loss": 5.034118057485935,
      "tokens_seen": 71434240
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009882246740220662,
      "loss": 2.5096,
      "theoretical_loss": 5.033451392661957,
      "tokens_seen": 71499776
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009882046138415246,
      "loss": 2.3763,
      "theoretical_loss": 5.032785509534391,
      "tokens_seen": 71565312
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000988184553660983,
      "loss": 2.4245,
      "theoretical_loss": 5.032120406472082,
      "tokens_seen": 71630848
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009881644934804413,
      "loss": 2.4523,
      "theoretical_loss": 5.031456081848761,
      "tokens_seen": 71696384
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009881444332998997,
      "loss": 2.2822,
      "theoretical_loss": 5.030792534043039,
      "tokens_seen": 71761920
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000988124373119358,
      "loss": 2.4078,
      "theoretical_loss": 5.030129761438376,
      "tokens_seen": 71827456
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009881043129388164,
      "loss": 2.4901,
      "theoretical_loss": 5.029467762423069,
      "tokens_seen": 71892992
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009880842527582748,
      "loss": 2.4351,
      "theoretical_loss": 5.028806535390228,
      "tokens_seen": 71958528
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009880641925777331,
      "loss": 2.4051,
      "theoretical_loss": 5.028146078737766,
      "tokens_seen": 72024064
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.49358925223350525,
      "objective/train/docs_used": 33987,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.276097536087036,
      "objective/train/original_loss": 2.2760977745056152,
      "objective/train/theoretical_loss": 5.0274863908683685,
      "objective/train/tokens_used": 92549600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24399147927761078,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050599455833435,
      "objective/train/weighted_lm_loss": 2.3911280632019043,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 5.0274863908683685,
      "tokens_seen": 72089600
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009880441323971915,
      "loss": 2.258,
      "theoretical_loss": 5.0274863908683685,
      "tokens_seen": 72089600
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009880240722166499,
      "loss": 2.4077,
      "theoretical_loss": 5.026827470189487,
      "tokens_seen": 72155136
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009880040120361084,
      "loss": 2.4191,
      "theoretical_loss": 5.026169315113311,
      "tokens_seen": 72220672
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009879839518555668,
      "loss": 2.3911,
      "theoretical_loss": 5.025511924056753,
      "tokens_seen": 72286208
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009879638916750252,
      "loss": 2.5566,
      "theoretical_loss": 5.024855295441432,
      "tokens_seen": 72351744
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009879438314944835,
      "loss": 2.3835,
      "theoretical_loss": 5.024199427693654,
      "tokens_seen": 72417280
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987923771313942,
      "loss": 2.4704,
      "theoretical_loss": 5.023544319244391,
      "tokens_seen": 72482816
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009879037111334003,
      "loss": 2.3699,
      "theoretical_loss": 5.0228899685292685,
      "tokens_seen": 72548352
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009878836509528586,
      "loss": 2.1989,
      "theoretical_loss": 5.022236373988544,
      "tokens_seen": 72613888
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987863590772317,
      "loss": 2.4072,
      "theoretical_loss": 5.021583534067085,
      "tokens_seen": 72679424
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009878435305917753,
      "loss": 2.3943,
      "theoretical_loss": 5.020931447214364,
      "tokens_seen": 72744960
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009878234704112337,
      "loss": 2.4555,
      "theoretical_loss": 5.020280111884426,
      "tokens_seen": 72810496
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987803410230692,
      "loss": 2.3937,
      "theoretical_loss": 5.01962952653588,
      "tokens_seen": 72876032
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009877833500501504,
      "loss": 2.2974,
      "theoretical_loss": 5.018979689631879,
      "tokens_seen": 72941568
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009877632898696088,
      "loss": 2.2553,
      "theoretical_loss": 5.018330599640101,
      "tokens_seen": 73007104
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009877432296890674,
      "loss": 2.3621,
      "theoretical_loss": 5.017682255032735,
      "tokens_seen": 73072640
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009877231695085255,
      "loss": 2.5088,
      "theoretical_loss": 5.017034654286462,
      "tokens_seen": 73138176
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987703109327984,
      "loss": 2.4084,
      "theoretical_loss": 5.0163877958824346,
      "tokens_seen": 73203712
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009876830491474422,
      "loss": 2.4194,
      "theoretical_loss": 5.0157416783062665,
      "tokens_seen": 73269248
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009876629889669008,
      "loss": 2.4157,
      "theoretical_loss": 5.015096300048009,
      "tokens_seen": 73334784
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009876429287863592,
      "loss": 2.4174,
      "theoretical_loss": 5.0144516596021385,
      "tokens_seen": 73400320
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009876228686058175,
      "loss": 2.205,
      "theoretical_loss": 5.013807755467537,
      "tokens_seen": 73465856
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987602808425276,
      "loss": 2.2951,
      "theoretical_loss": 5.013164586147477,
      "tokens_seen": 73531392
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009875827482447343,
      "loss": 2.3731,
      "theoretical_loss": 5.012522150149602,
      "tokens_seen": 73596928
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009875626880641926,
      "loss": 2.4183,
      "theoretical_loss": 5.011880445985916,
      "tokens_seen": 73662464
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987542627883651,
      "loss": 2.3,
      "theoretical_loss": 5.011239472172758,
      "tokens_seen": 73728000
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009875225677031094,
      "loss": 2.5542,
      "theoretical_loss": 5.010599227230792,
      "tokens_seen": 73793536
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009875025075225677,
      "loss": 2.5132,
      "theoretical_loss": 5.009959709684992,
      "tokens_seen": 73859072
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987482447342026,
      "loss": 2.409,
      "theoretical_loss": 5.009320918064615,
      "tokens_seen": 73924608
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009874623871614844,
      "loss": 2.4014,
      "theoretical_loss": 5.008682850903201,
      "tokens_seen": 73990144
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009874423269809428,
      "loss": 2.4533,
      "theoretical_loss": 5.00804550673854,
      "tokens_seen": 74055680
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009874222668004012,
      "loss": 2.309,
      "theoretical_loss": 5.007408884112671,
      "tokens_seen": 74121216
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009874022066198595,
      "loss": 2.5961,
      "theoretical_loss": 5.006772981571855,
      "tokens_seen": 74186752
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009873821464393181,
      "loss": 2.4414,
      "theoretical_loss": 5.006137797666559,
      "tokens_seen": 74252288
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009873620862587763,
      "loss": 2.2648,
      "theoretical_loss": 5.005503330951454,
      "tokens_seen": 74317824
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009873420260782348,
      "loss": 2.4416,
      "theoretical_loss": 5.004869579985381,
      "tokens_seen": 74383360
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987321965897693,
      "loss": 2.3453,
      "theoretical_loss": 5.004236543331345,
      "tokens_seen": 74448896
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009873019057171516,
      "loss": 2.3843,
      "theoretical_loss": 5.0036042195565,
      "tokens_seen": 74514432
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00098728184553661,
      "loss": 2.3351,
      "theoretical_loss": 5.002972607232129,
      "tokens_seen": 74579968
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009872617853560683,
      "loss": 2.3041,
      "theoretical_loss": 5.002341704933631,
      "tokens_seen": 74645504
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009872417251755266,
      "loss": 2.4829,
      "theoretical_loss": 5.001711511240506,
      "tokens_seen": 74711040
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987221664994985,
      "loss": 2.3724,
      "theoretical_loss": 5.0010820247363394,
      "tokens_seen": 74776576
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009872016048144434,
      "loss": 2.3896,
      "theoretical_loss": 5.000453244008782,
      "tokens_seen": 74842112
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009871815446339017,
      "loss": 2.4098,
      "theoretical_loss": 4.9998251676495435,
      "tokens_seen": 74907648
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00098716148445336,
      "loss": 2.425,
      "theoretical_loss": 4.999197794254371,
      "tokens_seen": 74973184
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009871414242728185,
      "loss": 2.5423,
      "theoretical_loss": 4.998571122423035,
      "tokens_seen": 75038720
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009871213640922768,
      "loss": 2.2918,
      "theoretical_loss": 4.997945150759316,
      "tokens_seen": 75104256
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009871013039117352,
      "loss": 2.3774,
      "theoretical_loss": 4.997319877870986,
      "tokens_seen": 75169792
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009870812437311935,
      "loss": 2.3024,
      "theoretical_loss": 4.9966953023697975,
      "tokens_seen": 75235328
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987061183550652,
      "loss": 2.4583,
      "theoretical_loss": 4.99607142287147,
      "tokens_seen": 75300864
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.49300244450569153,
      "objective/train/docs_used": 35047,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.2772889137268066,
      "objective/train/original_loss": 2.2772884368896484,
      "objective/train/theoretical_loss": 4.995448237995667,
      "objective/train/tokens_used": 95826400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24339433014392853,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505375862121582,
      "objective/train/weighted_lm_loss": 2.392406463623047,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.995448237995667,
      "tokens_seen": 75366400
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009870411233701103,
      "loss": 2.4079,
      "theoretical_loss": 4.995448237995667,
      "tokens_seen": 75366400
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009870210631895688,
      "loss": 2.3811,
      "theoretical_loss": 4.994825746365992,
      "tokens_seen": 75431936
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000987001003009027,
      "loss": 2.4155,
      "theoretical_loss": 4.994203946609964,
      "tokens_seen": 75497472
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009869809428284856,
      "loss": 2.2558,
      "theoretical_loss": 4.993582837359012,
      "tokens_seen": 75563008
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009869608826479437,
      "loss": 2.4505,
      "theoretical_loss": 4.992962417248453,
      "tokens_seen": 75628544
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009869408224674023,
      "loss": 2.4398,
      "theoretical_loss": 4.992342684917482,
      "tokens_seen": 75694080
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009869207622868607,
      "loss": 2.2416,
      "theoretical_loss": 4.991723639009154,
      "tokens_seen": 75759616
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000986900702106319,
      "loss": 2.2164,
      "theoretical_loss": 4.991105278170377,
      "tokens_seen": 75825152
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009868806419257774,
      "loss": 2.5057,
      "theoretical_loss": 4.990487601051887,
      "tokens_seen": 75890688
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009868605817452357,
      "loss": 2.4679,
      "theoretical_loss": 4.989870606308244,
      "tokens_seen": 75956224
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000986840521564694,
      "loss": 2.3433,
      "theoretical_loss": 4.989254292597813,
      "tokens_seen": 76021760
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009868204613841525,
      "loss": 2.5069,
      "theoretical_loss": 4.9886386585827465,
      "tokens_seen": 76087296
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009868004012036108,
      "loss": 2.4961,
      "theoretical_loss": 4.988023702928979,
      "tokens_seen": 76152832
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009867803410230692,
      "loss": 2.3971,
      "theoretical_loss": 4.9874094243062075,
      "tokens_seen": 76218368
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009867602808425276,
      "loss": 2.4396,
      "theoretical_loss": 4.986795821387878,
      "tokens_seen": 76283904
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000986740220661986,
      "loss": 2.4312,
      "theoretical_loss": 4.986182892851171,
      "tokens_seen": 76349440
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009867201604814443,
      "loss": 2.3369,
      "theoretical_loss": 4.985570637376993,
      "tokens_seen": 76414976
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009867001003009026,
      "loss": 2.3453,
      "theoretical_loss": 4.984959053649959,
      "tokens_seen": 76480512
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000986680040120361,
      "loss": 2.3224,
      "theoretical_loss": 4.984348140358374,
      "tokens_seen": 76546048
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009866599799398196,
      "loss": 2.3089,
      "theoretical_loss": 4.983737896194229,
      "tokens_seen": 76611584
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009866399197592777,
      "loss": 2.3881,
      "theoretical_loss": 4.983128319853182,
      "tokens_seen": 76677120
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009866198595787363,
      "loss": 2.3237,
      "theoretical_loss": 4.982519410034545,
      "tokens_seen": 76742656
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009865997993981945,
      "loss": 2.2656,
      "theoretical_loss": 4.981911165441273,
      "tokens_seen": 76808192
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000986579739217653,
      "loss": 2.358,
      "theoretical_loss": 4.981303584779946,
      "tokens_seen": 76873728
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009865596790371114,
      "loss": 2.4148,
      "theoretical_loss": 4.980696666760761,
      "tokens_seen": 76939264
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009865396188565698,
      "loss": 2.3153,
      "theoretical_loss": 4.980090410097517,
      "tokens_seen": 77004800
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009865195586760281,
      "loss": 2.3638,
      "theoretical_loss": 4.979484813507599,
      "tokens_seen": 77070336
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009864994984954865,
      "loss": 2.3893,
      "theoretical_loss": 4.97887987571197,
      "tokens_seen": 77135872
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009864794383149448,
      "loss": 2.2725,
      "theoretical_loss": 4.9782755954351545,
      "tokens_seen": 77201408
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009864593781344032,
      "loss": 2.3585,
      "theoretical_loss": 4.977671971405225,
      "tokens_seen": 77266944
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009864393179538616,
      "loss": 2.3069,
      "theoretical_loss": 4.977069002353792,
      "tokens_seen": 77332480
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00098641925777332,
      "loss": 2.3565,
      "theoretical_loss": 4.976466687015991,
      "tokens_seen": 77398016
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009863991975927785,
      "loss": 2.375,
      "theoretical_loss": 4.975865024130467,
      "tokens_seen": 77463552
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009863791374122367,
      "loss": 2.2839,
      "theoretical_loss": 4.975264012439363,
      "tokens_seen": 77529088
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009863590772316952,
      "loss": 2.3887,
      "theoretical_loss": 4.974663650688306,
      "tokens_seen": 77594624
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009863390170511534,
      "loss": 2.3619,
      "theoretical_loss": 4.974063937626402,
      "tokens_seen": 77660160
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000986318956870612,
      "loss": 2.3161,
      "theoretical_loss": 4.973464872006213,
      "tokens_seen": 77725696
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009862988966900703,
      "loss": 2.2241,
      "theoretical_loss": 4.972866452583748,
      "tokens_seen": 77791232
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009862788365095287,
      "loss": 2.5561,
      "theoretical_loss": 4.972268678118454,
      "tokens_seen": 77856768
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000986258776328987,
      "loss": 2.2846,
      "theoretical_loss": 4.971671547373202,
      "tokens_seen": 77922304
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009862387161484454,
      "loss": 2.3102,
      "theoretical_loss": 4.971075059114272,
      "tokens_seen": 77987840
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009862186559679038,
      "loss": 2.2737,
      "theoretical_loss": 4.970479212111342,
      "tokens_seen": 78053376
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009861985957873621,
      "loss": 2.2125,
      "theoretical_loss": 4.969884005137479,
      "tokens_seen": 78118912
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009861785356068205,
      "loss": 2.4044,
      "theoretical_loss": 4.969289436969124,
      "tokens_seen": 78184448
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009861584754262789,
      "loss": 2.3445,
      "theoretical_loss": 4.968695506386077,
      "tokens_seen": 78249984
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009861384152457372,
      "loss": 2.1897,
      "theoretical_loss": 4.968102212171493,
      "tokens_seen": 78315520
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009861183550651956,
      "loss": 2.3495,
      "theoretical_loss": 4.967509553111862,
      "tokens_seen": 78381056
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000986098294884654,
      "loss": 2.197,
      "theoretical_loss": 4.966917527996999,
      "tokens_seen": 78446592
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009860782347041123,
      "loss": 2.3427,
      "theoretical_loss": 4.9663261356200366,
      "tokens_seen": 78512128
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009860581745235707,
      "loss": 2.339,
      "theoretical_loss": 4.965735374777409,
      "tokens_seen": 78577664
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.4910324811935425,
      "objective/train/docs_used": 36308,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.3306944370269775,
      "objective/train/original_loss": 2.3306941986083984,
      "objective/train/theoretical_loss": 4.96514524426884,
      "objective/train/tokens_used": 99103200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24158093333244324,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503313541412354,
      "objective/train/weighted_lm_loss": 2.448256015777588,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.96514524426884,
      "tokens_seen": 78643200
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009860381143430292,
      "loss": 2.4381,
      "theoretical_loss": 4.96514524426884,
      "tokens_seen": 78643200
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009860180541624874,
      "loss": 2.333,
      "theoretical_loss": 4.964555742897334,
      "tokens_seen": 78708736
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985997993981946,
      "loss": 2.45,
      "theoretical_loss": 4.963966869469162,
      "tokens_seen": 78774272
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009859779338014041,
      "loss": 2.2028,
      "theoretical_loss": 4.963378622793849,
      "tokens_seen": 78839808
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009859578736208627,
      "loss": 2.2797,
      "theoretical_loss": 4.962791001684167,
      "tokens_seen": 78905344
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985937813440321,
      "loss": 2.2991,
      "theoretical_loss": 4.962204004956117,
      "tokens_seen": 78970880
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009859177532597794,
      "loss": 2.232,
      "theoretical_loss": 4.961617631428927,
      "tokens_seen": 79036416
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009858976930792378,
      "loss": 2.2345,
      "theoretical_loss": 4.961031879925029,
      "tokens_seen": 79101952
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009858776328986961,
      "loss": 2.3468,
      "theoretical_loss": 4.960446749270055,
      "tokens_seen": 79167488
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009858575727181545,
      "loss": 2.348,
      "theoretical_loss": 4.959862238292827,
      "tokens_seen": 79233024
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009858375125376129,
      "loss": 2.469,
      "theoretical_loss": 4.959278345825336,
      "tokens_seen": 79298560
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009858174523570712,
      "loss": 2.3083,
      "theoretical_loss": 4.958695070702744,
      "tokens_seen": 79364096
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009857973921765296,
      "loss": 2.3116,
      "theoretical_loss": 4.958112411763365,
      "tokens_seen": 79429632
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985777331995988,
      "loss": 2.328,
      "theoretical_loss": 4.957530367848651,
      "tokens_seen": 79495168
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009857572718154463,
      "loss": 2.4212,
      "theoretical_loss": 4.95694893780319,
      "tokens_seen": 79560704
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009857372116349047,
      "loss": 2.325,
      "theoretical_loss": 4.956368120474689,
      "tokens_seen": 79626240
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985717151454363,
      "loss": 2.3468,
      "theoretical_loss": 4.955787914713962,
      "tokens_seen": 79691776
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009856970912738214,
      "loss": 2.3578,
      "theoretical_loss": 4.955208319374924,
      "tokens_seen": 79757312
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.00098567703109328,
      "loss": 2.4362,
      "theoretical_loss": 4.954629333314573,
      "tokens_seen": 79822848
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009856569709127381,
      "loss": 2.2681,
      "theoretical_loss": 4.954050955392988,
      "tokens_seen": 79888384
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009856369107321967,
      "loss": 2.2174,
      "theoretical_loss": 4.953473184473312,
      "tokens_seen": 79953920
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009856168505516549,
      "loss": 2.2807,
      "theoretical_loss": 4.952896019421743,
      "tokens_seen": 80019456
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009855967903711134,
      "loss": 2.329,
      "theoretical_loss": 4.952319459107521,
      "tokens_seen": 80084992
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009855767301905718,
      "loss": 2.2275,
      "theoretical_loss": 4.951743502402923,
      "tokens_seen": 80150528
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009855566700100302,
      "loss": 2.2448,
      "theoretical_loss": 4.951168148183246,
      "tokens_seen": 80216064
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009855366098294885,
      "loss": 2.3235,
      "theoretical_loss": 4.9505933953268,
      "tokens_seen": 80281600
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009855165496489469,
      "loss": 2.3358,
      "theoretical_loss": 4.9500192427149,
      "tokens_seen": 80347136
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009854964894684052,
      "loss": 2.2623,
      "theoretical_loss": 4.949445689231847,
      "tokens_seen": 80412672
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009854764292878636,
      "loss": 2.2786,
      "theoretical_loss": 4.948872733764926,
      "tokens_seen": 80478208
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985456369107322,
      "loss": 2.1879,
      "theoretical_loss": 4.9483003752043935,
      "tokens_seen": 80543744
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009854363089267803,
      "loss": 2.1886,
      "theoretical_loss": 4.9477286124434645,
      "tokens_seen": 80609280
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009854162487462387,
      "loss": 2.3044,
      "theoretical_loss": 4.947157444378303,
      "tokens_seen": 80674816
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985396188565697,
      "loss": 2.2717,
      "theoretical_loss": 4.946586869908014,
      "tokens_seen": 80740352
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009853761283851554,
      "loss": 2.268,
      "theoretical_loss": 4.946016887934633,
      "tokens_seen": 80805888
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009853560682046138,
      "loss": 2.37,
      "theoretical_loss": 4.945447497363112,
      "tokens_seen": 80871424
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009853360080240721,
      "loss": 2.3511,
      "theoretical_loss": 4.944878697101316,
      "tokens_seen": 80936960
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009853159478435307,
      "loss": 2.3928,
      "theoretical_loss": 4.944310486060004,
      "tokens_seen": 81002496
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009852958876629889,
      "loss": 2.2323,
      "theoretical_loss": 4.943742863152828,
      "tokens_seen": 81068032
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009852758274824474,
      "loss": 2.203,
      "theoretical_loss": 4.943175827296319,
      "tokens_seen": 81133568
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009852557673019056,
      "loss": 2.3613,
      "theoretical_loss": 4.942609377409873,
      "tokens_seen": 81199104
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009852357071213642,
      "loss": 2.3922,
      "theoretical_loss": 4.942043512415751,
      "tokens_seen": 81264640
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009852156469408225,
      "loss": 2.3077,
      "theoretical_loss": 4.94147823123906,
      "tokens_seen": 81330176
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985195586760281,
      "loss": 2.4319,
      "theoretical_loss": 4.940913532807745,
      "tokens_seen": 81395712
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009851755265797393,
      "loss": 2.3457,
      "theoretical_loss": 4.940349416052586,
      "tokens_seen": 81461248
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009851554663991976,
      "loss": 2.3576,
      "theoretical_loss": 4.939785879907176,
      "tokens_seen": 81526784
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985135406218656,
      "loss": 2.1788,
      "theoretical_loss": 4.939222923307925,
      "tokens_seen": 81592320
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009851153460381143,
      "loss": 2.3332,
      "theoretical_loss": 4.938660545194038,
      "tokens_seen": 81657856
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009850952858575727,
      "loss": 2.4885,
      "theoretical_loss": 4.938098744507517,
      "tokens_seen": 81723392
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000985075225677031,
      "loss": 2.4229,
      "theoretical_loss": 4.937537520193139,
      "tokens_seen": 81788928
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009850551654964894,
      "loss": 2.2958,
      "theoretical_loss": 4.936976871198457,
      "tokens_seen": 81854464
    },
    {
      "epoch": 0.02,
      "objective/train/advantage_avg": 0.48865529894828796,
      "objective/train/docs_used": 37569,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.141408681869507,
      "objective/train/original_loss": 2.141408920288086,
      "objective/train/theoretical_loss": 4.936416796473786,
      "objective/train/tokens_used": 102380000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24008464813232422,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0500857830047607,
      "objective/train/weighted_lm_loss": 2.2487919330596924,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.936416796473786,
      "tokens_seen": 81920000
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009850351053159478,
      "loss": 2.3225,
      "theoretical_loss": 4.936416796473786,
      "tokens_seen": 81920000
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009850150451354062,
      "loss": 2.3456,
      "theoretical_loss": 4.935857294972193,
      "tokens_seen": 81985536
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009849949849548645,
      "loss": 2.1802,
      "theoretical_loss": 4.93529836564949,
      "tokens_seen": 82051072
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000984974924774323,
      "loss": 2.3381,
      "theoretical_loss": 4.934740007464219,
      "tokens_seen": 82116608
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009849548645937815,
      "loss": 2.3167,
      "theoretical_loss": 4.934182219377654,
      "tokens_seen": 82182144
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009849348044132398,
      "loss": 2.2076,
      "theoretical_loss": 4.9336250003537785,
      "tokens_seen": 82247680
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009849147442326982,
      "loss": 2.2008,
      "theoretical_loss": 4.933068349359283,
      "tokens_seen": 82313216
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.0009848946840521565,
      "loss": 2.255,
      "theoretical_loss": 4.93251226536356,
      "tokens_seen": 82378752
    },
    {
      "epoch": 0.02,
      "learning_rate": 0.000984874623871615,
      "loss": 2.3702,
      "theoretical_loss": 4.931956747338681,
      "tokens_seen": 82444288
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009848545636910733,
      "loss": 2.2657,
      "theoretical_loss": 4.931401794259408,
      "tokens_seen": 82509824
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009848345035105316,
      "loss": 2.3108,
      "theoretical_loss": 4.93084740510316,
      "tokens_seen": 82575360
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00098481444332999,
      "loss": 2.289,
      "theoretical_loss": 4.930293578850026,
      "tokens_seen": 82640896
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009847943831494484,
      "loss": 2.2171,
      "theoretical_loss": 4.9297403144827445,
      "tokens_seen": 82706432
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009847743229689067,
      "loss": 2.2979,
      "theoretical_loss": 4.929187610986693,
      "tokens_seen": 82771968
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984754262788365,
      "loss": 2.2295,
      "theoretical_loss": 4.928635467349885,
      "tokens_seen": 82837504
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009847342026078234,
      "loss": 2.3672,
      "theoretical_loss": 4.9280838825629605,
      "tokens_seen": 82903040
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009847141424272818,
      "loss": 2.3961,
      "theoretical_loss": 4.927532855619173,
      "tokens_seen": 82968576
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009846940822467404,
      "loss": 2.2037,
      "theoretical_loss": 4.926982385514384,
      "tokens_seen": 83034112
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009846740220661985,
      "loss": 2.3054,
      "theoretical_loss": 4.92643247124705,
      "tokens_seen": 83099648
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009846539618856571,
      "loss": 2.2816,
      "theoretical_loss": 4.925883111818225,
      "tokens_seen": 83165184
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009846339017051153,
      "loss": 2.2191,
      "theoretical_loss": 4.925334306231532,
      "tokens_seen": 83230720
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009846138415245738,
      "loss": 2.36,
      "theoretical_loss": 4.924786053493177,
      "tokens_seen": 83296256
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009845937813440322,
      "loss": 2.3395,
      "theoretical_loss": 4.924238352611924,
      "tokens_seen": 83361792
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009845737211634906,
      "loss": 2.3407,
      "theoretical_loss": 4.923691202599091,
      "tokens_seen": 83427328
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984553660982949,
      "loss": 2.0856,
      "theoretical_loss": 4.923144602468545,
      "tokens_seen": 83492864
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009845336008024073,
      "loss": 2.2128,
      "theoretical_loss": 4.922598551236689,
      "tokens_seen": 83558400
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009845135406218656,
      "loss": 2.4467,
      "theoretical_loss": 4.922053047922455,
      "tokens_seen": 83623936
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984493480441324,
      "loss": 2.266,
      "theoretical_loss": 4.921508091547297,
      "tokens_seen": 83689472
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009844734202607824,
      "loss": 2.2651,
      "theoretical_loss": 4.920963681135177,
      "tokens_seen": 83755008
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009844533600802407,
      "loss": 2.2465,
      "theoretical_loss": 4.9204198157125685,
      "tokens_seen": 83820544
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984433299899699,
      "loss": 2.2151,
      "theoretical_loss": 4.919876494308432,
      "tokens_seen": 83886080
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009844132397191575,
      "loss": 2.3275,
      "theoretical_loss": 4.91933371595422,
      "tokens_seen": 83951616
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009843931795386158,
      "loss": 2.3892,
      "theoretical_loss": 4.918791479683863,
      "tokens_seen": 84017152
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009843731193580742,
      "loss": 2.31,
      "theoretical_loss": 4.91824978453376,
      "tokens_seen": 84082688
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009843530591775325,
      "loss": 2.3452,
      "theoretical_loss": 4.917708629542775,
      "tokens_seen": 84148224
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009843329989969911,
      "loss": 2.2599,
      "theoretical_loss": 4.917168013752225,
      "tokens_seen": 84213760
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009843129388164493,
      "loss": 2.2059,
      "theoretical_loss": 4.916627936205872,
      "tokens_seen": 84279296
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009842928786359078,
      "loss": 2.2837,
      "theoretical_loss": 4.916088395949915,
      "tokens_seen": 84344832
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984272818455366,
      "loss": 2.4008,
      "theoretical_loss": 4.915549392032985,
      "tokens_seen": 84410368
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009842527582748246,
      "loss": 2.4535,
      "theoretical_loss": 4.915010923506134,
      "tokens_seen": 84475904
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984232698094283,
      "loss": 2.3609,
      "theoretical_loss": 4.914472989422826,
      "tokens_seen": 84541440
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009842126379137413,
      "loss": 2.1934,
      "theoretical_loss": 4.913935588838932,
      "tokens_seen": 84606976
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009841925777331997,
      "loss": 2.3023,
      "theoretical_loss": 4.913398720812719,
      "tokens_seen": 84672512
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984172517552658,
      "loss": 2.2574,
      "theoretical_loss": 4.9128623844048445,
      "tokens_seen": 84738048
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009841524573721164,
      "loss": 2.2642,
      "theoretical_loss": 4.91232657867835,
      "tokens_seen": 84803584
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009841323971915747,
      "loss": 2.2327,
      "theoretical_loss": 4.911791302698646,
      "tokens_seen": 84869120
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984112337011033,
      "loss": 2.3648,
      "theoretical_loss": 4.9112565555335115,
      "tokens_seen": 84934656
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009840922768304915,
      "loss": 2.3775,
      "theoretical_loss": 4.910722336253086,
      "tokens_seen": 85000192
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009840722166499498,
      "loss": 2.2554,
      "theoretical_loss": 4.910188643929853,
      "tokens_seen": 85065728
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009840521564694082,
      "loss": 2.386,
      "theoretical_loss": 4.909655477638648,
      "tokens_seen": 85131264
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.4955958127975464,
      "objective/train/docs_used": 38678,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.1719961166381836,
      "objective/train/original_loss": 2.1719963550567627,
      "objective/train/theoretical_loss": 4.909122836456632,
      "objective/train/tokens_used": 105656800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24580909311771393,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0508092641830444,
      "objective/train/weighted_lm_loss": 2.282179594039917,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0304546356201172,
      "theoretical_loss": 4.909122836456632,
      "tokens_seen": 85196800
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009840320962888666,
      "loss": 2.2857,
      "theoretical_loss": 4.909122836456632,
      "tokens_seen": 85196800
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000984012036108325,
      "loss": 2.243,
      "theoretical_loss": 4.908590719463298,
      "tokens_seen": 85262336
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009839919759277833,
      "loss": 2.3113,
      "theoretical_loss": 4.9080591257404595,
      "tokens_seen": 85327872
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009839719157472419,
      "loss": 2.2506,
      "theoretical_loss": 4.907528054372239,
      "tokens_seen": 85393408
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009839518555667,
      "loss": 2.1794,
      "theoretical_loss": 4.906997504445066,
      "tokens_seen": 85458944
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009839317953861586,
      "loss": 2.2323,
      "theoretical_loss": 4.906467475047664,
      "tokens_seen": 85524480
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009839117352056167,
      "loss": 2.4132,
      "theoretical_loss": 4.905937965271047,
      "tokens_seen": 85590016
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009838916750250753,
      "loss": 2.314,
      "theoretical_loss": 4.905408974208514,
      "tokens_seen": 85655552
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009838716148445337,
      "loss": 2.3108,
      "theoretical_loss": 4.904880500955633,
      "tokens_seen": 85721088
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000983851554663992,
      "loss": 2.1642,
      "theoretical_loss": 4.904352544610242,
      "tokens_seen": 85786624
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009838314944834504,
      "loss": 2.354,
      "theoretical_loss": 4.903825104272437,
      "tokens_seen": 85852160
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009838114343029088,
      "loss": 2.2949,
      "theoretical_loss": 4.903298179044569,
      "tokens_seen": 85917696
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009837913741223671,
      "loss": 2.2796,
      "theoretical_loss": 4.90277176803123,
      "tokens_seen": 85983232
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009837713139418255,
      "loss": 2.259,
      "theoretical_loss": 4.902245870339252,
      "tokens_seen": 86048768
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009837512537612838,
      "loss": 2.3348,
      "theoretical_loss": 4.9017204850776945,
      "tokens_seen": 86114304
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009837311935807422,
      "loss": 2.3361,
      "theoretical_loss": 4.901195611357844,
      "tokens_seen": 86179840
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009837111334002006,
      "loss": 2.1493,
      "theoretical_loss": 4.9006712482931984,
      "tokens_seen": 86245376
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000983691073219659,
      "loss": 2.3035,
      "theoretical_loss": 4.9001473949994665,
      "tokens_seen": 86310912
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009836710130391173,
      "loss": 2.1573,
      "theoretical_loss": 4.899624050594557,
      "tokens_seen": 86376448
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009836509528585757,
      "loss": 2.3964,
      "theoretical_loss": 4.899101214198575,
      "tokens_seen": 86441984
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000983630892678034,
      "loss": 2.1748,
      "theoretical_loss": 4.89857888493381,
      "tokens_seen": 86507520
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009836108324974926,
      "loss": 2.2292,
      "theoretical_loss": 4.898057061924732,
      "tokens_seen": 86573056
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009835907723169507,
      "loss": 2.3679,
      "theoretical_loss": 4.897535744297986,
      "tokens_seen": 86638592
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009835707121364093,
      "loss": 2.3989,
      "theoretical_loss": 4.89701493118238,
      "tokens_seen": 86704128
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009835506519558675,
      "loss": 2.1716,
      "theoretical_loss": 4.896494621708882,
      "tokens_seen": 86769664
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000983530591775326,
      "loss": 2.2875,
      "theoretical_loss": 4.895974815010612,
      "tokens_seen": 86835200
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009835105315947844,
      "loss": 2.2411,
      "theoretical_loss": 4.895455510222835,
      "tokens_seen": 86900736
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009834904714142428,
      "loss": 2.2779,
      "theoretical_loss": 4.8949367064829525,
      "tokens_seen": 86966272
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009834704112337011,
      "loss": 2.2055,
      "theoretical_loss": 4.8944184029305,
      "tokens_seen": 87031808
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009834503510531595,
      "loss": 2.3822,
      "theoretical_loss": 4.893900598707134,
      "tokens_seen": 87097344
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009834302908726179,
      "loss": 2.1673,
      "theoretical_loss": 4.893383292956631,
      "tokens_seen": 87162880
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009834102306920762,
      "loss": 2.3037,
      "theoretical_loss": 4.892866484824879,
      "tokens_seen": 87228416
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009833901705115346,
      "loss": 2.1902,
      "theoretical_loss": 4.892350173459863,
      "tokens_seen": 87293952
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000983370110330993,
      "loss": 2.2473,
      "theoretical_loss": 4.891834358011673,
      "tokens_seen": 87359488
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009833500501504515,
      "loss": 2.1875,
      "theoretical_loss": 4.891319037632486,
      "tokens_seen": 87425024
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009833299899699097,
      "loss": 2.2182,
      "theoretical_loss": 4.890804211476562,
      "tokens_seen": 87490560
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009833099297893682,
      "loss": 2.2744,
      "theoretical_loss": 4.890289878700239,
      "tokens_seen": 87556096
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009832898696088264,
      "loss": 2.202,
      "theoretical_loss": 4.889776038461924,
      "tokens_seen": 87621632
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000983269809428285,
      "loss": 2.203,
      "theoretical_loss": 4.88926268992209,
      "tokens_seen": 87687168
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009832497492477433,
      "loss": 2.216,
      "theoretical_loss": 4.888749832243265,
      "tokens_seen": 87752704
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009832296890672017,
      "loss": 2.2196,
      "theoretical_loss": 4.888237464590028,
      "tokens_seen": 87818240
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00098320962888666,
      "loss": 2.3488,
      "theoretical_loss": 4.887725586129003,
      "tokens_seen": 87883776
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009831895687061184,
      "loss": 2.2586,
      "theoretical_loss": 4.887214196028848,
      "tokens_seen": 87949312
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009831695085255768,
      "loss": 2.2706,
      "theoretical_loss": 4.886703293460259,
      "tokens_seen": 88014848
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009831494483450351,
      "loss": 2.2567,
      "theoretical_loss": 4.8861928775959464,
      "tokens_seen": 88080384
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009831293881644935,
      "loss": 2.2866,
      "theoretical_loss": 4.885682947610648,
      "tokens_seen": 88145920
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009831093279839519,
      "loss": 2.2333,
      "theoretical_loss": 4.885173502681107,
      "tokens_seen": 88211456
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009830892678034102,
      "loss": 2.2126,
      "theoretical_loss": 4.884664541986075,
      "tokens_seen": 88276992
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009830692076228686,
      "loss": 2.3505,
      "theoretical_loss": 4.884156064706302,
      "tokens_seen": 88342528
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000983049147442327,
      "loss": 2.1561,
      "theoretical_loss": 4.883648070024529,
      "tokens_seen": 88408064
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.49346664547920227,
      "objective/train/docs_used": 39491,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.0473103523254395,
      "objective/train/original_loss": 2.0473103523254395,
      "objective/train/theoretical_loss": 4.8831405571254844,
      "objective/train/tokens_used": 108933600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24416816234588623,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505878925323486,
      "objective/train/weighted_lm_loss": 2.150813102722168,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.8831405571254844,
      "tokens_seen": 88473600
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009830290872617853,
      "loss": 2.2569,
      "theoretical_loss": 4.8831405571254844,
      "tokens_seen": 88473600
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009830090270812437,
      "loss": 2.2434,
      "theoretical_loss": 4.8826335251958755,
      "tokens_seen": 88539136
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009829889669007023,
      "loss": 2.2955,
      "theoretical_loss": 4.882126973424384,
      "tokens_seen": 88604672
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009829689067201604,
      "loss": 2.3253,
      "theoretical_loss": 4.881620901001661,
      "tokens_seen": 88670208
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982948846539619,
      "loss": 2.2168,
      "theoretical_loss": 4.881115307120314,
      "tokens_seen": 88735744
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009829287863590771,
      "loss": 2.155,
      "theoretical_loss": 4.88061019097491,
      "tokens_seen": 88801280
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009829087261785357,
      "loss": 2.3044,
      "theoretical_loss": 4.880105551761961,
      "tokens_seen": 88866816
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982888665997994,
      "loss": 2.2709,
      "theoretical_loss": 4.879601388679923,
      "tokens_seen": 88932352
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009828686058174524,
      "loss": 2.2732,
      "theoretical_loss": 4.879097700929192,
      "tokens_seen": 88997888
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009828485456369108,
      "loss": 2.1625,
      "theoretical_loss": 4.878594487712088,
      "tokens_seen": 89063424
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009828284854563692,
      "loss": 2.4076,
      "theoretical_loss": 4.87809174823286,
      "tokens_seen": 89128960
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009828084252758275,
      "loss": 2.3207,
      "theoretical_loss": 4.877589481697674,
      "tokens_seen": 89194496
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009827883650952859,
      "loss": 2.1592,
      "theoretical_loss": 4.877087687314608,
      "tokens_seen": 89260032
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009827683049147442,
      "loss": 2.1353,
      "theoretical_loss": 4.876586364293646,
      "tokens_seen": 89325568
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009827482447342026,
      "loss": 2.2606,
      "theoretical_loss": 4.876085511846673,
      "tokens_seen": 89391104
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982728184553661,
      "loss": 2.2478,
      "theoretical_loss": 4.875585129187467,
      "tokens_seen": 89456640
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009827081243731193,
      "loss": 2.279,
      "theoretical_loss": 4.875085215531697,
      "tokens_seen": 89522176
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009826880641925777,
      "loss": 2.2824,
      "theoretical_loss": 4.87458577009691,
      "tokens_seen": 89587712
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982668004012036,
      "loss": 2.2527,
      "theoretical_loss": 4.874086792102535,
      "tokens_seen": 89653248
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009826479438314944,
      "loss": 2.2688,
      "theoretical_loss": 4.873588280769868,
      "tokens_seen": 89718784
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982627883650953,
      "loss": 2.0824,
      "theoretical_loss": 4.873090235322069,
      "tokens_seen": 89784320
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009826078234704111,
      "loss": 2.0821,
      "theoretical_loss": 4.87259265498416,
      "tokens_seen": 89849856
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009825877632898697,
      "loss": 2.2779,
      "theoretical_loss": 4.872095538983015,
      "tokens_seen": 89915392
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009825677031093279,
      "loss": 2.2936,
      "theoretical_loss": 4.871598886547356,
      "tokens_seen": 89980928
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009825476429287864,
      "loss": 2.2121,
      "theoretical_loss": 4.871102696907745,
      "tokens_seen": 90046464
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009825275827482448,
      "loss": 2.2357,
      "theoretical_loss": 4.870606969296581,
      "tokens_seen": 90112000
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009825075225677032,
      "loss": 2.5256,
      "theoretical_loss": 4.870111702948094,
      "tokens_seen": 90177536
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009824874623871615,
      "loss": 2.2016,
      "theoretical_loss": 4.869616897098336,
      "tokens_seen": 90243072
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00098246740220662,
      "loss": 2.1991,
      "theoretical_loss": 4.869122550985182,
      "tokens_seen": 90308608
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009824473420260783,
      "loss": 2.3257,
      "theoretical_loss": 4.868628663848316,
      "tokens_seen": 90374144
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009824272818455366,
      "loss": 2.3091,
      "theoretical_loss": 4.868135234929232,
      "tokens_seen": 90439680
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982407221664995,
      "loss": 2.3082,
      "theoretical_loss": 4.867642263471224,
      "tokens_seen": 90505216
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009823871614844533,
      "loss": 2.1111,
      "theoretical_loss": 4.867149748719386,
      "tokens_seen": 90570752
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009823671013039117,
      "loss": 2.3432,
      "theoretical_loss": 4.866657689920601,
      "tokens_seen": 90636288
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00098234704112337,
      "loss": 2.1497,
      "theoretical_loss": 4.866166086323535,
      "tokens_seen": 90701824
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009823269809428284,
      "loss": 2.1864,
      "theoretical_loss": 4.865674937178635,
      "tokens_seen": 90767360
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009823069207622868,
      "loss": 2.1773,
      "theoretical_loss": 4.865184241738126,
      "tokens_seen": 90832896
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009822868605817452,
      "loss": 2.2805,
      "theoretical_loss": 4.864693999255998,
      "tokens_seen": 90898432
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009822668004012037,
      "loss": 2.3565,
      "theoretical_loss": 4.864204208988003,
      "tokens_seen": 90963968
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009822467402206619,
      "loss": 2.1671,
      "theoretical_loss": 4.8637148701916555,
      "tokens_seen": 91029504
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009822266800401205,
      "loss": 2.2914,
      "theoretical_loss": 4.863225982126219,
      "tokens_seen": 91095040
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009822066198595786,
      "loss": 2.2608,
      "theoretical_loss": 4.862737544052706,
      "tokens_seen": 91160576
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009821865596790372,
      "loss": 2.1416,
      "theoretical_loss": 4.86224955523387,
      "tokens_seen": 91226112
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009821664994984955,
      "loss": 2.3752,
      "theoretical_loss": 4.8617620149342,
      "tokens_seen": 91291648
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982146439317954,
      "loss": 2.2634,
      "theoretical_loss": 4.861274922419918,
      "tokens_seen": 91357184
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009821263791374123,
      "loss": 2.322,
      "theoretical_loss": 4.86078827695897,
      "tokens_seen": 91422720
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009821063189568706,
      "loss": 2.2562,
      "theoretical_loss": 4.860302077821023,
      "tokens_seen": 91488256
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982086258776329,
      "loss": 2.1679,
      "theoretical_loss": 4.859816324277461,
      "tokens_seen": 91553792
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009820661985957874,
      "loss": 2.2282,
      "theoretical_loss": 4.859331015601375,
      "tokens_seen": 91619328
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009820461384152457,
      "loss": 2.1868,
      "theoretical_loss": 4.858846151067562,
      "tokens_seen": 91684864
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.49189305305480957,
      "objective/train/docs_used": 40748,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.9773281812667847,
      "objective/train/original_loss": 1.9773280620574951,
      "objective/train/theoretical_loss": 4.858361729952518,
      "objective/train/tokens_used": 112210400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24240271747112274,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504215955734253,
      "objective/train/weighted_lm_loss": 2.0769896507263184,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.858361729952518,
      "tokens_seen": 91750400
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000982026078234704,
      "loss": 2.2005,
      "theoretical_loss": 4.858361729952518,
      "tokens_seen": 91750400
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009820060180541624,
      "loss": 2.199,
      "theoretical_loss": 4.857877751534435,
      "tokens_seen": 91815936
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009819859578736208,
      "loss": 2.2238,
      "theoretical_loss": 4.857394215093191,
      "tokens_seen": 91881472
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009819658976930794,
      "loss": 2.2805,
      "theoretical_loss": 4.856911119910352,
      "tokens_seen": 91947008
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009819458375125375,
      "loss": 2.1968,
      "theoretical_loss": 4.856428465269159,
      "tokens_seen": 92012544
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009819257773319961,
      "loss": 2.1694,
      "theoretical_loss": 4.85594625045453,
      "tokens_seen": 92078080
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009819057171514545,
      "loss": 2.2446,
      "theoretical_loss": 4.855464474753049,
      "tokens_seen": 92143616
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009818856569709128,
      "loss": 2.1264,
      "theoretical_loss": 4.854983137452964,
      "tokens_seen": 92209152
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009818655967903712,
      "loss": 2.2233,
      "theoretical_loss": 4.8545022378441836,
      "tokens_seen": 92274688
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009818455366098296,
      "loss": 2.3523,
      "theoretical_loss": 4.854021775218266,
      "tokens_seen": 92340224
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981825476429288,
      "loss": 2.2655,
      "theoretical_loss": 4.853541748868423,
      "tokens_seen": 92405760
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009818054162487463,
      "loss": 2.3512,
      "theoretical_loss": 4.853062158089504,
      "tokens_seen": 92471296
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009817853560682046,
      "loss": 2.1191,
      "theoretical_loss": 4.852583002178001,
      "tokens_seen": 92536832
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981765295887663,
      "loss": 2.1088,
      "theoretical_loss": 4.852104280432039,
      "tokens_seen": 92602368
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009817452357071214,
      "loss": 2.2118,
      "theoretical_loss": 4.85162599215137,
      "tokens_seen": 92667904
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009817251755265797,
      "loss": 2.256,
      "theoretical_loss": 4.85114813663737,
      "tokens_seen": 92733440
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981705115346038,
      "loss": 2.2957,
      "theoretical_loss": 4.8506707131930344,
      "tokens_seen": 92798976
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009816850551654965,
      "loss": 2.0396,
      "theoretical_loss": 4.850193721122972,
      "tokens_seen": 92864512
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009816649949849548,
      "loss": 2.2755,
      "theoretical_loss": 4.849717159733401,
      "tokens_seen": 92930048
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009816449348044134,
      "loss": 2.2867,
      "theoretical_loss": 4.849241028332142,
      "tokens_seen": 92995584
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009816248746238715,
      "loss": 2.2158,
      "theoretical_loss": 4.848765326228618,
      "tokens_seen": 93061120
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009816048144433301,
      "loss": 2.1635,
      "theoretical_loss": 4.848290052733843,
      "tokens_seen": 93126656
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009815847542627883,
      "loss": 2.186,
      "theoretical_loss": 4.847815207160422,
      "tokens_seen": 93192192
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009815646940822468,
      "loss": 2.2273,
      "theoretical_loss": 4.847340788822546,
      "tokens_seen": 93257728
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009815446339017052,
      "loss": 2.1433,
      "theoretical_loss": 4.846866797035984,
      "tokens_seen": 93323264
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009815245737211636,
      "loss": 2.2482,
      "theoretical_loss": 4.846393231118083,
      "tokens_seen": 93388800
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981504513540622,
      "loss": 2.2104,
      "theoretical_loss": 4.845920090387757,
      "tokens_seen": 93454336
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009814844533600803,
      "loss": 2.1955,
      "theoretical_loss": 4.84544737416549,
      "tokens_seen": 93519872
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009814643931795387,
      "loss": 2.2442,
      "theoretical_loss": 4.844975081773322,
      "tokens_seen": 93585408
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981444332998997,
      "loss": 2.1941,
      "theoretical_loss": 4.844503212534856,
      "tokens_seen": 93650944
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009814242728184554,
      "loss": 2.1822,
      "theoretical_loss": 4.844031765775242,
      "tokens_seen": 93716480
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009814042126379137,
      "loss": 2.1861,
      "theoretical_loss": 4.843560740821177,
      "tokens_seen": 93782016
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981384152457372,
      "loss": 2.3743,
      "theoretical_loss": 4.843090137000904,
      "tokens_seen": 93847552
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009813640922768305,
      "loss": 2.0544,
      "theoretical_loss": 4.842619953644203,
      "tokens_seen": 93913088
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009813440320962888,
      "loss": 2.3207,
      "theoretical_loss": 4.842150190082384,
      "tokens_seen": 93978624
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009813239719157472,
      "loss": 2.0991,
      "theoretical_loss": 4.841680845648289,
      "tokens_seen": 94044160
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009813039117352056,
      "loss": 2.163,
      "theoretical_loss": 4.841211919676287,
      "tokens_seen": 94109696
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009812838515546641,
      "loss": 2.1717,
      "theoretical_loss": 4.84074341150226,
      "tokens_seen": 94175232
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009812637913741223,
      "loss": 2.1603,
      "theoretical_loss": 4.840275320463608,
      "tokens_seen": 94240768
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009812437311935809,
      "loss": 2.2077,
      "theoretical_loss": 4.839807645899245,
      "tokens_seen": 94306304
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981223671013039,
      "loss": 2.0532,
      "theoretical_loss": 4.839340387149586,
      "tokens_seen": 94371840
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009812036108324976,
      "loss": 2.1507,
      "theoretical_loss": 4.838873543556552,
      "tokens_seen": 94437376
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981183550651956,
      "loss": 2.1606,
      "theoretical_loss": 4.838407114463558,
      "tokens_seen": 94502912
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009811634904714143,
      "loss": 2.2777,
      "theoretical_loss": 4.837941099215515,
      "tokens_seen": 94568448
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009811434302908727,
      "loss": 2.1049,
      "theoretical_loss": 4.837475497158817,
      "tokens_seen": 94633984
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000981123370110331,
      "loss": 2.2131,
      "theoretical_loss": 4.83701030764135,
      "tokens_seen": 94699520
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009811033099297894,
      "loss": 2.0608,
      "theoretical_loss": 4.836545530012472,
      "tokens_seen": 94765056
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009810832497492478,
      "loss": 2.1397,
      "theoretical_loss": 4.83608116362302,
      "tokens_seen": 94830592
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009810631895687061,
      "loss": 2.1879,
      "theoretical_loss": 4.835617207825303,
      "tokens_seen": 94896128
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009810431293881645,
      "loss": 2.2057,
      "theoretical_loss": 4.835153661973092,
      "tokens_seen": 94961664
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.48443371057510376,
      "objective/train/docs_used": 42039,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8953099250793457,
      "objective/train/original_loss": 1.8953099250793457,
      "objective/train/theoretical_loss": 4.8346905254216255,
      "objective/train/tokens_used": 115487200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.23641405999660492,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0496450662612915,
      "objective/train/weighted_lm_loss": 1.9899815320968628,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.8346905254216255,
      "tokens_seen": 95027200
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009810230692076228,
      "loss": 2.0976,
      "theoretical_loss": 4.8346905254216255,
      "tokens_seen": 95027200
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009810030090270812,
      "loss": 2.1386,
      "theoretical_loss": 4.834227797527595,
      "tokens_seen": 95092736
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009809829488465396,
      "loss": 2.1371,
      "theoretical_loss": 4.83376547764915,
      "tokens_seen": 95158272
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980962888665998,
      "loss": 2.1342,
      "theoretical_loss": 4.833303565145885,
      "tokens_seen": 95223808
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009809428284854563,
      "loss": 2.2161,
      "theoretical_loss": 4.832842059378843,
      "tokens_seen": 95289344
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009809227683049149,
      "loss": 2.1805,
      "theoretical_loss": 4.832380959710505,
      "tokens_seen": 95354880
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980902708124373,
      "loss": 2.1964,
      "theoretical_loss": 4.831920265504792,
      "tokens_seen": 95420416
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009808826479438316,
      "loss": 2.1742,
      "theoretical_loss": 4.831459976127054,
      "tokens_seen": 95485952
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009808625877632897,
      "loss": 2.2361,
      "theoretical_loss": 4.831000090944068,
      "tokens_seen": 95551488
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009808425275827483,
      "loss": 2.2147,
      "theoretical_loss": 4.830540609324041,
      "tokens_seen": 95617024
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009808224674022067,
      "loss": 2.2732,
      "theoretical_loss": 4.830081530636594,
      "tokens_seen": 95682560
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980802407221665,
      "loss": 2.1651,
      "theoretical_loss": 4.829622854252765,
      "tokens_seen": 95748096
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009807823470411234,
      "loss": 2.2456,
      "theoretical_loss": 4.829164579545006,
      "tokens_seen": 95813632
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009807622868605818,
      "loss": 2.1658,
      "theoretical_loss": 4.828706705887171,
      "tokens_seen": 95879168
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009807422266800401,
      "loss": 2.2252,
      "theoretical_loss": 4.8282492326545245,
      "tokens_seen": 95944704
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009807221664994985,
      "loss": 2.2021,
      "theoretical_loss": 4.827792159223722,
      "tokens_seen": 96010240
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009807021063189569,
      "loss": 2.2582,
      "theoretical_loss": 4.82733548497282,
      "tokens_seen": 96075776
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009806820461384152,
      "loss": 2.0183,
      "theoretical_loss": 4.8268792092812625,
      "tokens_seen": 96141312
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009806619859578736,
      "loss": 2.0873,
      "theoretical_loss": 4.826423331529884,
      "tokens_seen": 96206848
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980641925777332,
      "loss": 2.1451,
      "theoretical_loss": 4.825967851100897,
      "tokens_seen": 96272384
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009806218655967903,
      "loss": 2.3053,
      "theoretical_loss": 4.825512767377897,
      "tokens_seen": 96337920
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009806018054162487,
      "loss": 2.149,
      "theoretical_loss": 4.825058079745853,
      "tokens_seen": 96403456
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980581745235707,
      "loss": 2.1526,
      "theoretical_loss": 4.824603787591102,
      "tokens_seen": 96468992
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009805616850551656,
      "loss": 2.1577,
      "theoretical_loss": 4.824149890301353,
      "tokens_seen": 96534528
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009805416248746238,
      "loss": 2.1158,
      "theoretical_loss": 4.823696387265674,
      "tokens_seen": 96600064
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009805215646940823,
      "loss": 2.2835,
      "theoretical_loss": 4.823243277874491,
      "tokens_seen": 96665600
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009805015045135407,
      "loss": 2.2474,
      "theoretical_loss": 4.822790561519591,
      "tokens_seen": 96731136
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980481444332999,
      "loss": 2.1669,
      "theoretical_loss": 4.822338237594105,
      "tokens_seen": 96796672
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009804613841524574,
      "loss": 2.1423,
      "theoretical_loss": 4.821886305492513,
      "tokens_seen": 96862208
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009804413239719158,
      "loss": 2.2884,
      "theoretical_loss": 4.8214347646106415,
      "tokens_seen": 96927744
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009804212637913741,
      "loss": 2.162,
      "theoretical_loss": 4.8209836143456535,
      "tokens_seen": 96993280
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009804012036108325,
      "loss": 2.0766,
      "theoretical_loss": 4.820532854096047,
      "tokens_seen": 97058816
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009803811434302909,
      "loss": 2.2347,
      "theoretical_loss": 4.820082483261652,
      "tokens_seen": 97124352
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009803610832497492,
      "loss": 2.1397,
      "theoretical_loss": 4.81963250124363,
      "tokens_seen": 97189888
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009803410230692076,
      "loss": 2.0618,
      "theoretical_loss": 4.81918290744446,
      "tokens_seen": 97255424
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980320962888666,
      "loss": 2.0536,
      "theoretical_loss": 4.818733701267945,
      "tokens_seen": 97320960
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009803009027081245,
      "loss": 2.2106,
      "theoretical_loss": 4.818284882119203,
      "tokens_seen": 97386496
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009802808425275827,
      "loss": 2.3802,
      "theoretical_loss": 4.817836449404666,
      "tokens_seen": 97452032
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009802607823470413,
      "loss": 2.2977,
      "theoretical_loss": 4.817388402532074,
      "tokens_seen": 97517568
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009802407221664994,
      "loss": 2.316,
      "theoretical_loss": 4.81694074091047,
      "tokens_seen": 97583104
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980220661985958,
      "loss": 2.4769,
      "theoretical_loss": 4.816493463950202,
      "tokens_seen": 97648640
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009802006018054163,
      "loss": 2.2114,
      "theoretical_loss": 4.816046571062912,
      "tokens_seen": 97714176
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009801805416248747,
      "loss": 2.2655,
      "theoretical_loss": 4.815600061661536,
      "tokens_seen": 97779712
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000980160481444333,
      "loss": 2.0975,
      "theoretical_loss": 4.815153935160305,
      "tokens_seen": 97845248
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009801404212637914,
      "loss": 2.1909,
      "theoretical_loss": 4.814708190974729,
      "tokens_seen": 97910784
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009801203610832498,
      "loss": 2.2142,
      "theoretical_loss": 4.8142628285216045,
      "tokens_seen": 97976320
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009801003009027082,
      "loss": 2.1922,
      "theoretical_loss": 4.813817847219008,
      "tokens_seen": 98041856
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009800802407221665,
      "loss": 2.1104,
      "theoretical_loss": 4.81337324648629,
      "tokens_seen": 98107392
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009800601805416249,
      "loss": 2.0705,
      "theoretical_loss": 4.812929025744072,
      "tokens_seen": 98172928
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009800401203610832,
      "loss": 2.0202,
      "theoretical_loss": 4.812485184414244,
      "tokens_seen": 98238464
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.4935273230075836,
      "objective/train/docs_used": 43181,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.0869107246398926,
      "objective/train/original_loss": 2.0869107246398926,
      "objective/train/theoretical_loss": 4.812041721919962,
      "objective/train/tokens_used": 118764000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24405071139335632,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050593376159668,
      "objective/train/weighted_lm_loss": 2.192361831665039,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.812041721919962,
      "tokens_seen": 98304000
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009800200601805416,
      "loss": 2.1919,
      "theoretical_loss": 4.812041721919962,
      "tokens_seen": 98304000
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00098,
      "loss": 2.1269,
      "theoretical_loss": 4.8115986376856394,
      "tokens_seen": 98369536
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009799799398194583,
      "loss": 2.1634,
      "theoretical_loss": 4.81115593113695,
      "tokens_seen": 98435072
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009799598796389167,
      "loss": 2.0142,
      "theoretical_loss": 4.8107136017008205,
      "tokens_seen": 98500608
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009799398194583753,
      "loss": 2.0447,
      "theoretical_loss": 4.810271648805427,
      "tokens_seen": 98566144
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009799197592778334,
      "loss": 2.1669,
      "theoretical_loss": 4.809830071880194,
      "tokens_seen": 98631680
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979899699097292,
      "loss": 2.2777,
      "theoretical_loss": 4.809388870355785,
      "tokens_seen": 98697216
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009798796389167501,
      "loss": 2.1788,
      "theoretical_loss": 4.808948043664107,
      "tokens_seen": 98762752
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009798595787362087,
      "loss": 2.1686,
      "theoretical_loss": 4.8085075912383015,
      "tokens_seen": 98828288
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979839518555667,
      "loss": 2.0359,
      "theoretical_loss": 4.808067512512741,
      "tokens_seen": 98893824
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009798194583751254,
      "loss": 2.0967,
      "theoretical_loss": 4.807627806923029,
      "tokens_seen": 98959360
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009797993981945838,
      "loss": 2.2084,
      "theoretical_loss": 4.807188473905994,
      "tokens_seen": 99024896
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009797793380140422,
      "loss": 2.267,
      "theoretical_loss": 4.806749512899687,
      "tokens_seen": 99090432
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009797592778335005,
      "loss": 2.1556,
      "theoretical_loss": 4.806310923343375,
      "tokens_seen": 99155968
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979739217652959,
      "loss": 2.1009,
      "theoretical_loss": 4.805872704677542,
      "tokens_seen": 99221504
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009797191574724173,
      "loss": 2.1948,
      "theoretical_loss": 4.805434856343886,
      "tokens_seen": 99287040
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009796990972918756,
      "loss": 2.2327,
      "theoretical_loss": 4.804997377785307,
      "tokens_seen": 99352576
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979679037111334,
      "loss": 2.2464,
      "theoretical_loss": 4.8045602684459165,
      "tokens_seen": 99418112
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009796589769307923,
      "loss": 2.168,
      "theoretical_loss": 4.804123527771022,
      "tokens_seen": 99483648
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009796389167502507,
      "loss": 2.1951,
      "theoretical_loss": 4.803687155207134,
      "tokens_seen": 99549184
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979618856569709,
      "loss": 2.1594,
      "theoretical_loss": 4.8032511502019535,
      "tokens_seen": 99614720
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009795987963891674,
      "loss": 2.1376,
      "theoretical_loss": 4.802815512204375,
      "tokens_seen": 99680256
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979578736208626,
      "loss": 2.1674,
      "theoretical_loss": 4.80238024066448,
      "tokens_seen": 99745792
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009795586760280842,
      "loss": 2.0653,
      "theoretical_loss": 4.801945335033535,
      "tokens_seen": 99811328
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009795386158475427,
      "loss": 2.3154,
      "theoretical_loss": 4.801510794763988,
      "tokens_seen": 99876864
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979518555667001,
      "loss": 2.0517,
      "theoretical_loss": 4.801076619309466,
      "tokens_seen": 99942400
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009794984954864595,
      "loss": 2.0963,
      "theoretical_loss": 4.800642808124767,
      "tokens_seen": 100007936
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009794784353059178,
      "loss": 2.3449,
      "theoretical_loss": 4.800209360665864,
      "tokens_seen": 100073472
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009794583751253762,
      "loss": 2.1891,
      "theoretical_loss": 4.799776276389897,
      "tokens_seen": 100139008
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009794383149448345,
      "loss": 2.183,
      "theoretical_loss": 4.79934355475517,
      "tokens_seen": 100204544
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979418254764293,
      "loss": 2.1676,
      "theoretical_loss": 4.79891119522115,
      "tokens_seen": 100270080
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009793981945837513,
      "loss": 2.1522,
      "theoretical_loss": 4.798479197248461,
      "tokens_seen": 100335616
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009793781344032096,
      "loss": 2.364,
      "theoretical_loss": 4.798047560298882,
      "tokens_seen": 100401152
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979358074222668,
      "loss": 2.2696,
      "theoretical_loss": 4.797616283835346,
      "tokens_seen": 100466688
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009793380140421264,
      "loss": 2.0181,
      "theoretical_loss": 4.797185367321932,
      "tokens_seen": 100532224
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009793179538615847,
      "loss": 2.1756,
      "theoretical_loss": 4.796754810223866,
      "tokens_seen": 100597760
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979297893681043,
      "loss": 2.0873,
      "theoretical_loss": 4.796324612007515,
      "tokens_seen": 100663296
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009792778335005014,
      "loss": 2.1305,
      "theoretical_loss": 4.795894772140386,
      "tokens_seen": 100728832
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009792577733199598,
      "loss": 2.2556,
      "theoretical_loss": 4.795465290091123,
      "tokens_seen": 100794368
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009792377131394182,
      "loss": 2.0559,
      "theoretical_loss": 4.795036165329499,
      "tokens_seen": 100859904
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009792176529588767,
      "loss": 2.1793,
      "theoretical_loss": 4.794607397326421,
      "tokens_seen": 100925440
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979197592778335,
      "loss": 2.1214,
      "theoretical_loss": 4.794178985553919,
      "tokens_seen": 100990976
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009791775325977935,
      "loss": 2.2544,
      "theoretical_loss": 4.793750929485149,
      "tokens_seen": 101056512
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009791574724172518,
      "loss": 2.3256,
      "theoretical_loss": 4.793323228594385,
      "tokens_seen": 101122048
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009791374122367102,
      "loss": 2.087,
      "theoretical_loss": 4.792895882357019,
      "tokens_seen": 101187584
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009791173520561686,
      "loss": 2.0553,
      "theoretical_loss": 4.792468890249557,
      "tokens_seen": 101253120
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979097291875627,
      "loss": 2.1112,
      "theoretical_loss": 4.792042251749617,
      "tokens_seen": 101318656
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009790772316950853,
      "loss": 2.1636,
      "theoretical_loss": 4.791615966335922,
      "tokens_seen": 101384192
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009790571715145436,
      "loss": 2.1967,
      "theoretical_loss": 4.791190033488302,
      "tokens_seen": 101449728
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000979037111334002,
      "loss": 2.1536,
      "theoretical_loss": 4.790764452687689,
      "tokens_seen": 101515264
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.49196192622184753,
      "objective/train/docs_used": 44310,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.0242908000946045,
      "objective/train/original_loss": 2.0242908000946045,
      "objective/train/theoretical_loss": 4.790339223416113,
      "objective/train/tokens_used": 122040800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2426905632019043,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504299402236938,
      "objective/train/weighted_lm_loss": 2.1261332035064697,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.790339223416113,
      "tokens_seen": 101580800
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009790170511534604,
      "loss": 2.0924,
      "theoretical_loss": 4.790339223416113,
      "tokens_seen": 101580800
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009789969909729187,
      "loss": 2.1643,
      "theoretical_loss": 4.7899143451566975,
      "tokens_seen": 101646336
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000978976930792377,
      "loss": 2.1168,
      "theoretical_loss": 4.7894898173936635,
      "tokens_seen": 101711872
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009789568706118357,
      "loss": 2.0482,
      "theoretical_loss": 4.789065639612316,
      "tokens_seen": 101777408
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009789368104312938,
      "loss": 2.17,
      "theoretical_loss": 4.78864181129905,
      "tokens_seen": 101842944
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009789167502507524,
      "loss": 2.0711,
      "theoretical_loss": 4.788218331941344,
      "tokens_seen": 101908480
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009788966900702105,
      "loss": 2.0941,
      "theoretical_loss": 4.787795201027757,
      "tokens_seen": 101974016
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009788766298896691,
      "loss": 2.149,
      "theoretical_loss": 4.787372418047926,
      "tokens_seen": 102039552
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009788565697091275,
      "loss": 2.2792,
      "theoretical_loss": 4.786949982492559,
      "tokens_seen": 102105088
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009788365095285858,
      "loss": 2.1955,
      "theoretical_loss": 4.786527893853441,
      "tokens_seen": 102170624
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009788164493480442,
      "loss": 2.046,
      "theoretical_loss": 4.786106151623423,
      "tokens_seen": 102236160
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009787963891675026,
      "loss": 2.0414,
      "theoretical_loss": 4.785684755296422,
      "tokens_seen": 102301696
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000978776328986961,
      "loss": 2.0363,
      "theoretical_loss": 4.785263704367418,
      "tokens_seen": 102367232
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009787562688064193,
      "loss": 2.0321,
      "theoretical_loss": 4.7848429983324525,
      "tokens_seen": 102432768
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009787362086258777,
      "loss": 2.1357,
      "theoretical_loss": 4.784422636688622,
      "tokens_seen": 102498304
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000978716148445336,
      "loss": 2.168,
      "theoretical_loss": 4.784002618934078,
      "tokens_seen": 102563840
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009786960882647944,
      "loss": 2.0982,
      "theoretical_loss": 4.783582944568026,
      "tokens_seen": 102629376
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009786760280842527,
      "loss": 2.145,
      "theoretical_loss": 4.783163613090714,
      "tokens_seen": 102694912
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000978655967903711,
      "loss": 2.0991,
      "theoretical_loss": 4.782744624003442,
      "tokens_seen": 102760448
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009786359077231695,
      "loss": 2.2595,
      "theoretical_loss": 4.782325976808549,
      "tokens_seen": 102825984
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009786158475426278,
      "loss": 2.2138,
      "theoretical_loss": 4.7819076710094155,
      "tokens_seen": 102891520
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009785957873620864,
      "loss": 2.1458,
      "theoretical_loss": 4.781489706110458,
      "tokens_seen": 102957056
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009785757271815446,
      "loss": 2.0066,
      "theoretical_loss": 4.781072081617127,
      "tokens_seen": 103022592
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009785556670010031,
      "loss": 2.1166,
      "theoretical_loss": 4.780654797035907,
      "tokens_seen": 103088128
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009785356068204615,
      "loss": 2.1703,
      "theoretical_loss": 4.780237851874308,
      "tokens_seen": 103153664
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009785155466399199,
      "loss": 2.1396,
      "theoretical_loss": 4.779821245640869,
      "tokens_seen": 103219200
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009784954864593782,
      "loss": 2.1718,
      "theoretical_loss": 4.779404977845148,
      "tokens_seen": 103284736
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009784754262788366,
      "loss": 2.1923,
      "theoretical_loss": 4.778989047997728,
      "tokens_seen": 103350272
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000978455366098295,
      "loss": 2.0623,
      "theoretical_loss": 4.778573455610207,
      "tokens_seen": 103415808
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009784353059177533,
      "loss": 2.1395,
      "theoretical_loss": 4.778158200195196,
      "tokens_seen": 103481344
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009784152457372117,
      "loss": 2.007,
      "theoretical_loss": 4.777743281266321,
      "tokens_seen": 103546880
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00097839518555667,
      "loss": 2.0959,
      "theoretical_loss": 4.777328698338218,
      "tokens_seen": 103612416
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009783751253761284,
      "loss": 2.146,
      "theoretical_loss": 4.776914450926526,
      "tokens_seen": 103677952
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009783550651955868,
      "loss": 2.1683,
      "theoretical_loss": 4.776500538547891,
      "tokens_seen": 103743488
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009783350050150451,
      "loss": 2.1065,
      "theoretical_loss": 4.776086960719956,
      "tokens_seen": 103809024
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009783149448345035,
      "loss": 2.0547,
      "theoretical_loss": 4.775673716961369,
      "tokens_seen": 103874560
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009782948846539618,
      "loss": 2.2282,
      "theoretical_loss": 4.775260806791767,
      "tokens_seen": 103940096
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009782748244734202,
      "loss": 2.2749,
      "theoretical_loss": 4.774848229731784,
      "tokens_seen": 104005632
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009782547642928786,
      "loss": 2.2745,
      "theoretical_loss": 4.774435985303043,
      "tokens_seen": 104071168
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009782347041123371,
      "loss": 2.1922,
      "theoretical_loss": 4.774024073028156,
      "tokens_seen": 104136704
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009782146439317953,
      "loss": 1.9764,
      "theoretical_loss": 4.7736124924307175,
      "tokens_seen": 104202240
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009781945837512539,
      "loss": 1.9248,
      "theoretical_loss": 4.773201243035307,
      "tokens_seen": 104267776
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009781745235707122,
      "loss": 2.1946,
      "theoretical_loss": 4.772790324367482,
      "tokens_seen": 104333312
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009781544633901706,
      "loss": 2.1539,
      "theoretical_loss": 4.772379735953779,
      "tokens_seen": 104398848
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000978134403209629,
      "loss": 2.1859,
      "theoretical_loss": 4.771969477321704,
      "tokens_seen": 104464384
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009781143430290873,
      "loss": 2.0899,
      "theoretical_loss": 4.771559547999743,
      "tokens_seen": 104529920
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009780942828485457,
      "loss": 2.1757,
      "theoretical_loss": 4.771149947517346,
      "tokens_seen": 104595456
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000978074222668004,
      "loss": 2.159,
      "theoretical_loss": 4.770740675404928,
      "tokens_seen": 104660992
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009780541624874624,
      "loss": 2.044,
      "theoretical_loss": 4.770331731193872,
      "tokens_seen": 104726528
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009780341023069208,
      "loss": 2.0043,
      "theoretical_loss": 4.769923114416522,
      "tokens_seen": 104792064
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.49358728528022766,
      "objective/train/docs_used": 45444,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.1862094402313232,
      "objective/train/original_loss": 2.1862096786499023,
      "objective/train/theoretical_loss": 4.769514824606178,
      "objective/train/tokens_used": 125317600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2443910837173462,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506013631820679,
      "objective/train/weighted_lm_loss": 2.2968454360961914,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.769514824606178,
      "tokens_seen": 104857600
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009780140421263791,
      "loss": 2.1955,
      "theoretical_loss": 4.769514824606178,
      "tokens_seen": 104857600
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009779939819458375,
      "loss": 2.0942,
      "theoretical_loss": 4.7691068612971,
      "tokens_seen": 104923136
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009779739217652959,
      "loss": 1.9648,
      "theoretical_loss": 4.768699224024498,
      "tokens_seen": 104988672
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009779538615847542,
      "loss": 2.12,
      "theoretical_loss": 4.768291912324538,
      "tokens_seen": 105054208
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009779338014042126,
      "loss": 2.025,
      "theoretical_loss": 4.7678849257343305,
      "tokens_seen": 105119744
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977913741223671,
      "loss": 2.1114,
      "theoretical_loss": 4.767478263791933,
      "tokens_seen": 105185280
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009778936810431293,
      "loss": 2.0773,
      "theoretical_loss": 4.7670719260363486,
      "tokens_seen": 105250816
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009778736208625879,
      "loss": 2.0793,
      "theoretical_loss": 4.76666591200752,
      "tokens_seen": 105316352
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977853560682046,
      "loss": 2.0647,
      "theoretical_loss": 4.766260221246329,
      "tokens_seen": 105381888
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009778335005015046,
      "loss": 2.0955,
      "theoretical_loss": 4.765854853294592,
      "tokens_seen": 105447424
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977813440320963,
      "loss": 2.0974,
      "theoretical_loss": 4.765449807695062,
      "tokens_seen": 105512960
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009777933801404213,
      "loss": 2.1978,
      "theoretical_loss": 4.765045083991422,
      "tokens_seen": 105578496
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009777733199598797,
      "loss": 2.0725,
      "theoretical_loss": 4.764640681728281,
      "tokens_seen": 105644032
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977753259779338,
      "loss": 2.3367,
      "theoretical_loss": 4.764236600451178,
      "tokens_seen": 105709568
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009777331995987964,
      "loss": 2.19,
      "theoretical_loss": 4.763832839706574,
      "tokens_seen": 105775104
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009777131394182548,
      "loss": 2.0348,
      "theoretical_loss": 4.763429399041852,
      "tokens_seen": 105840640
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009776930792377131,
      "loss": 2.025,
      "theoretical_loss": 4.763026278005314,
      "tokens_seen": 105906176
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009776730190571715,
      "loss": 2.2138,
      "theoretical_loss": 4.762623476146176,
      "tokens_seen": 105971712
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009776529588766299,
      "loss": 2.0778,
      "theoretical_loss": 4.762220993014572,
      "tokens_seen": 106037248
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009776328986960882,
      "loss": 2.0225,
      "theoretical_loss": 4.761818828161545,
      "tokens_seen": 106102784
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009776128385155466,
      "loss": 2.1458,
      "theoretical_loss": 4.761416981139046,
      "tokens_seen": 106168320
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977592778335005,
      "loss": 2.018,
      "theoretical_loss": 4.7610154514999365,
      "tokens_seen": 106233856
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009775727181544633,
      "loss": 2.1109,
      "theoretical_loss": 4.760614238797981,
      "tokens_seen": 106299392
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977552657973922,
      "loss": 2.1624,
      "theoretical_loss": 4.760213342587843,
      "tokens_seen": 106364928
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00097753259779338,
      "loss": 2.1445,
      "theoretical_loss": 4.75981276242509,
      "tokens_seen": 106430464
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009775125376128386,
      "loss": 2.0961,
      "theoretical_loss": 4.759412497866186,
      "tokens_seen": 106496000
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009774924774322968,
      "loss": 2.1643,
      "theoretical_loss": 4.759012548468487,
      "tokens_seen": 106561536
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009774724172517553,
      "loss": 2.0578,
      "theoretical_loss": 4.758612913790243,
      "tokens_seen": 106627072
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009774523570712137,
      "loss": 2.0945,
      "theoretical_loss": 4.758213593390595,
      "tokens_seen": 106692608
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977432296890672,
      "loss": 2.0254,
      "theoretical_loss": 4.757814586829572,
      "tokens_seen": 106758144
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009774122367101304,
      "loss": 2.1265,
      "theoretical_loss": 4.757415893668089,
      "tokens_seen": 106823680
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009773921765295888,
      "loss": 2.1485,
      "theoretical_loss": 4.75701751346794,
      "tokens_seen": 106889216
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009773721163490472,
      "loss": 2.0516,
      "theoretical_loss": 4.756619445791808,
      "tokens_seen": 106954752
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009773520561685055,
      "loss": 2.1517,
      "theoretical_loss": 4.756221690203244,
      "tokens_seen": 107020288
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009773319959879639,
      "loss": 2.2329,
      "theoretical_loss": 4.755824246266685,
      "tokens_seen": 107085824
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009773119358074222,
      "loss": 2.1316,
      "theoretical_loss": 4.755427113547437,
      "tokens_seen": 107151360
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009772918756268806,
      "loss": 2.0784,
      "theoretical_loss": 4.755030291611678,
      "tokens_seen": 107216896
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977271815446339,
      "loss": 2.1254,
      "theoretical_loss": 4.754633780026456,
      "tokens_seen": 107282432
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009772517552657975,
      "loss": 2.122,
      "theoretical_loss": 4.754237578359687,
      "tokens_seen": 107347968
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009772316950852557,
      "loss": 2.093,
      "theoretical_loss": 4.753841686180152,
      "tokens_seen": 107413504
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009772116349047143,
      "loss": 2.1048,
      "theoretical_loss": 4.753446103057492,
      "tokens_seen": 107479040
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009771915747241726,
      "loss": 2.1523,
      "theoretical_loss": 4.753050828562211,
      "tokens_seen": 107544576
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977171514543631,
      "loss": 2.1161,
      "theoretical_loss": 4.752655862265669,
      "tokens_seen": 107610112
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009771514543630894,
      "loss": 2.0962,
      "theoretical_loss": 4.752261203740085,
      "tokens_seen": 107675648
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009771313941825477,
      "loss": 2.1709,
      "theoretical_loss": 4.751866852558529,
      "tokens_seen": 107741184
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977111334002006,
      "loss": 2.0602,
      "theoretical_loss": 4.751472808294924,
      "tokens_seen": 107806720
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009770912738214644,
      "loss": 2.2369,
      "theoretical_loss": 4.751079070524041,
      "tokens_seen": 107872256
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009770712136409228,
      "loss": 2.185,
      "theoretical_loss": 4.750685638821498,
      "tokens_seen": 107937792
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009770511534603812,
      "loss": 2.0072,
      "theoretical_loss": 4.7502925127637585,
      "tokens_seen": 108003328
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009770310932798395,
      "loss": 2.132,
      "theoretical_loss": 4.749899691928128,
      "tokens_seen": 108068864
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.49381500482559204,
      "objective/train/docs_used": 46682,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.180783271789551,
      "objective/train/original_loss": 2.180783748626709,
      "objective/train/theoretical_loss": 4.749507175892756,
      "objective/train/tokens_used": 128594400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24419686198234558,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506229400634766,
      "objective/train/weighted_lm_loss": 2.291189432144165,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.749507175892756,
      "tokens_seen": 108134400
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000977011033099298,
      "loss": 2.1949,
      "theoretical_loss": 4.749507175892756,
      "tokens_seen": 108134400
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009769909729187563,
      "loss": 2.1011,
      "theoretical_loss": 4.7491149642366235,
      "tokens_seen": 108199936
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009769709127382146,
      "loss": 2.0941,
      "theoretical_loss": 4.7487230565395535,
      "tokens_seen": 108265472
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000976950852557673,
      "loss": 2.1314,
      "theoretical_loss": 4.7483314523822004,
      "tokens_seen": 108331008
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009769307923771313,
      "loss": 2.2006,
      "theoretical_loss": 4.74794015134605,
      "tokens_seen": 108396544
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009769107321965897,
      "loss": 2.0744,
      "theoretical_loss": 4.74754915301342,
      "tokens_seen": 108462080
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009768906720160483,
      "loss": 2.047,
      "theoretical_loss": 4.747158456967452,
      "tokens_seen": 108527616
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009768706118355064,
      "loss": 2.1796,
      "theoretical_loss": 4.746768062792116,
      "tokens_seen": 108593152
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000976850551654965,
      "loss": 2.0949,
      "theoretical_loss": 4.746377970072205,
      "tokens_seen": 108658688
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009768304914744234,
      "loss": 2.1516,
      "theoretical_loss": 4.745988178393331,
      "tokens_seen": 108724224
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009768104312938817,
      "loss": 2.1688,
      "theoretical_loss": 4.745598687341927,
      "tokens_seen": 108789760
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00097679037111334,
      "loss": 2.1327,
      "theoretical_loss": 4.745209496505241,
      "tokens_seen": 108855296
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009767703109327985,
      "loss": 2.1034,
      "theoretical_loss": 4.744820605471338,
      "tokens_seen": 108920832
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009767502507522568,
      "loss": 2.1114,
      "theoretical_loss": 4.744432013829094,
      "tokens_seen": 108986368
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009767301905717152,
      "loss": 2.1984,
      "theoretical_loss": 4.744043721168196,
      "tokens_seen": 109051904
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009767101303911735,
      "loss": 2.0895,
      "theoretical_loss": 4.7436557270791395,
      "tokens_seen": 109117440
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000976690070210632,
      "loss": 2.0292,
      "theoretical_loss": 4.743268031153226,
      "tokens_seen": 109182976
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009766700100300903,
      "loss": 2.0267,
      "theoretical_loss": 4.742880632982564,
      "tokens_seen": 109248512
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009766499498495486,
      "loss": 2.1127,
      "theoretical_loss": 4.74249353216006,
      "tokens_seen": 109314048
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000976629889669007,
      "loss": 1.9842,
      "theoretical_loss": 4.742106728279422,
      "tokens_seen": 109379584
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009766098294884654,
      "loss": 2.1229,
      "theoretical_loss": 4.741720220935159,
      "tokens_seen": 109445120
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009765897693079237,
      "loss": 2.0699,
      "theoretical_loss": 4.741334009722573,
      "tokens_seen": 109510656
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009765697091273823,
      "loss": 2.1166,
      "theoretical_loss": 4.740948094237761,
      "tokens_seen": 109576192
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009765496489468406,
      "loss": 2.1423,
      "theoretical_loss": 4.7405624740776116,
      "tokens_seen": 109641728
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009765295887662989,
      "loss": 2.1827,
      "theoretical_loss": 4.7401771488398055,
      "tokens_seen": 109707264
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009765095285857573,
      "loss": 2.1094,
      "theoretical_loss": 4.739792118122809,
      "tokens_seen": 109772800
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009764894684052156,
      "loss": 2.0291,
      "theoretical_loss": 4.739407381525874,
      "tokens_seen": 109838336
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000976469408224674,
      "loss": 2.1474,
      "theoretical_loss": 4.73902293864904,
      "tokens_seen": 109903872
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009764493480441325,
      "loss": 2.1552,
      "theoretical_loss": 4.738638789093124,
      "tokens_seen": 109969408
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009764292878635907,
      "loss": 2.0372,
      "theoretical_loss": 4.738254932459727,
      "tokens_seen": 110034944
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009764092276830492,
      "loss": 2.1613,
      "theoretical_loss": 4.7378713683512235,
      "tokens_seen": 110100480
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009763891675025074,
      "loss": 2.0998,
      "theoretical_loss": 4.737488096370768,
      "tokens_seen": 110166016
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009763691073219659,
      "loss": 2.0198,
      "theoretical_loss": 4.737105116122288,
      "tokens_seen": 110231552
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009763490471414243,
      "loss": 2.1154,
      "theoretical_loss": 4.736722427210483,
      "tokens_seen": 110297088
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009763289869608826,
      "loss": 2.2146,
      "theoretical_loss": 4.7363400292408215,
      "tokens_seen": 110362624
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000976308926780341,
      "loss": 2.1661,
      "theoretical_loss": 4.735957921819542,
      "tokens_seen": 110428160
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009762888665997995,
      "loss": 2.1288,
      "theoretical_loss": 4.735576104553646,
      "tokens_seen": 110493696
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009762688064192577,
      "loss": 2.0877,
      "theoretical_loss": 4.735194577050903,
      "tokens_seen": 110559232
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009762487462387162,
      "loss": 2.2459,
      "theoretical_loss": 4.734813338919842,
      "tokens_seen": 110624768
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009762286860581745,
      "loss": 2.0061,
      "theoretical_loss": 4.734432389769756,
      "tokens_seen": 110690304
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009762086258776329,
      "loss": 2.0924,
      "theoretical_loss": 4.734051729210693,
      "tokens_seen": 110755840
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009761885656970913,
      "loss": 2.0756,
      "theoretical_loss": 4.733671356853456,
      "tokens_seen": 110821376
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009761685055165496,
      "loss": 2.111,
      "theoretical_loss": 4.733291272309609,
      "tokens_seen": 110886912
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000976148445336008,
      "loss": 2.1974,
      "theoretical_loss": 4.732911475191464,
      "tokens_seen": 110952448
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009761283851554664,
      "loss": 1.9964,
      "theoretical_loss": 4.732531965112083,
      "tokens_seen": 111017984
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009761083249749247,
      "loss": 2.1016,
      "theoretical_loss": 4.732152741685281,
      "tokens_seen": 111083520
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009760882647943832,
      "loss": 2.0433,
      "theoretical_loss": 4.731773804525616,
      "tokens_seen": 111149056
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009760682046138415,
      "loss": 2.0489,
      "theoretical_loss": 4.731395153248394,
      "tokens_seen": 111214592
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009760481444332999,
      "loss": 1.9634,
      "theoretical_loss": 4.731016787469663,
      "tokens_seen": 111280128
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009760280842527583,
      "loss": 2.1297,
      "theoretical_loss": 4.730638706806213,
      "tokens_seen": 111345664
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.49222612380981445,
      "objective/train/docs_used": 47788,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.9619340896606445,
      "objective/train/original_loss": 1.961933970451355,
      "objective/train/theoretical_loss": 4.73026091087557,
      "objective/train/tokens_used": 131871200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24286691844463348,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050457239151001,
      "objective/train/weighted_lm_loss": 2.060546636581421,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.73026091087557,
      "tokens_seen": 111411200
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009760080240722167,
      "loss": 2.0969,
      "theoretical_loss": 4.73026091087557,
      "tokens_seen": 111411200
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000975987963891675,
      "loss": 2.1887,
      "theoretical_loss": 4.729883399296005,
      "tokens_seen": 111476736
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009759679037111334,
      "loss": 2.049,
      "theoretical_loss": 4.729506171686518,
      "tokens_seen": 111542272
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009759478435305919,
      "loss": 2.0852,
      "theoretical_loss": 4.729129227666847,
      "tokens_seen": 111607808
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009759277833500502,
      "loss": 2.1017,
      "theoretical_loss": 4.728752566857459,
      "tokens_seen": 111673344
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009759077231695086,
      "loss": 2.1405,
      "theoretical_loss": 4.728376188879554,
      "tokens_seen": 111738880
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009758876629889669,
      "loss": 2.1897,
      "theoretical_loss": 4.7280000933550586,
      "tokens_seen": 111804416
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009758676028084253,
      "loss": 2.1838,
      "theoretical_loss": 4.727624279906628,
      "tokens_seen": 111869952
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009758475426278837,
      "loss": 2.047,
      "theoretical_loss": 4.72724874815764,
      "tokens_seen": 111935488
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009758274824473421,
      "loss": 2.0746,
      "theoretical_loss": 4.726873497732196,
      "tokens_seen": 112001024
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009758074222668004,
      "loss": 2.2697,
      "theoretical_loss": 4.726498528255121,
      "tokens_seen": 112066560
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009757873620862589,
      "loss": 2.1296,
      "theoretical_loss": 4.7261238393519545,
      "tokens_seen": 112132096
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009757673019057171,
      "loss": 2.1274,
      "theoretical_loss": 4.725749430648958,
      "tokens_seen": 112197632
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009757472417251756,
      "loss": 2.0796,
      "theoretical_loss": 4.725375301773107,
      "tokens_seen": 112263168
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009757271815446339,
      "loss": 2.0802,
      "theoretical_loss": 4.725001452352091,
      "tokens_seen": 112328704
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009757071213640923,
      "loss": 2.1455,
      "theoretical_loss": 4.724627882014311,
      "tokens_seen": 112394240
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009756870611835507,
      "loss": 2.0957,
      "theoretical_loss": 4.724254590388881,
      "tokens_seen": 112459776
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009756670010030091,
      "loss": 1.9454,
      "theoretical_loss": 4.723881577105621,
      "tokens_seen": 112525312
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009756469408224674,
      "loss": 2.157,
      "theoretical_loss": 4.723508841795058,
      "tokens_seen": 112590848
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009756268806419259,
      "loss": 2.1352,
      "theoretical_loss": 4.723136384088427,
      "tokens_seen": 112656384
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009756068204613841,
      "loss": 2.078,
      "theoretical_loss": 4.722764203617663,
      "tokens_seen": 112721920
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009755867602808426,
      "loss": 2.1686,
      "theoretical_loss": 4.722392300015404,
      "tokens_seen": 112787456
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000975566700100301,
      "loss": 2.1921,
      "theoretical_loss": 4.7220206729149865,
      "tokens_seen": 112852992
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009755466399197593,
      "loss": 2.0499,
      "theoretical_loss": 4.72164932195045,
      "tokens_seen": 112918528
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009755265797392177,
      "loss": 2.1789,
      "theoretical_loss": 4.7212782467565235,
      "tokens_seen": 112984064
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000975506519558676,
      "loss": 2.0737,
      "theoretical_loss": 4.720907446968635,
      "tokens_seen": 113049600
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009754864593781344,
      "loss": 2.0673,
      "theoretical_loss": 4.720536922222907,
      "tokens_seen": 113115136
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009754663991975929,
      "loss": 2.1072,
      "theoretical_loss": 4.720166672156147,
      "tokens_seen": 113180672
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009754463390170511,
      "loss": 2.0866,
      "theoretical_loss": 4.719796696405858,
      "tokens_seen": 113246208
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009754262788365096,
      "loss": 2.1345,
      "theoretical_loss": 4.719426994610227,
      "tokens_seen": 113311744
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009754062186559678,
      "loss": 2.0008,
      "theoretical_loss": 4.719057566408132,
      "tokens_seen": 113377280
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009753861584754263,
      "loss": 2.0752,
      "theoretical_loss": 4.718688411439129,
      "tokens_seen": 113442816
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009753660982948847,
      "loss": 2.0776,
      "theoretical_loss": 4.718319529343462,
      "tokens_seen": 113508352
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.000975346038114343,
      "loss": 2.1316,
      "theoretical_loss": 4.7179509197620515,
      "tokens_seen": 113573888
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009753259779338014,
      "loss": 2.082,
      "theoretical_loss": 4.717582582336503,
      "tokens_seen": 113639424
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009753059177532599,
      "loss": 1.9908,
      "theoretical_loss": 4.7172145167090935,
      "tokens_seen": 113704960
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009752858575727181,
      "loss": 2.1448,
      "theoretical_loss": 4.716846722522781,
      "tokens_seen": 113770496
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009752657973921766,
      "loss": 2.0554,
      "theoretical_loss": 4.716479199421195,
      "tokens_seen": 113836032
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009752457372116349,
      "loss": 1.9584,
      "theoretical_loss": 4.716111947048638,
      "tokens_seen": 113901568
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009752256770310933,
      "loss": 2.0346,
      "theoretical_loss": 4.715744965050087,
      "tokens_seen": 113967104
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009752056168505517,
      "loss": 2.1214,
      "theoretical_loss": 4.715378253071181,
      "tokens_seen": 114032640
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.00097518555667001,
      "loss": 2.01,
      "theoretical_loss": 4.715011810758236,
      "tokens_seen": 114098176
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009751654964894684,
      "loss": 2.0735,
      "theoretical_loss": 4.714645637758226,
      "tokens_seen": 114163712
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009751454363089268,
      "loss": 2.0352,
      "theoretical_loss": 4.714279733718793,
      "tokens_seen": 114229248
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009751253761283851,
      "loss": 2.1201,
      "theoretical_loss": 4.713914098288242,
      "tokens_seen": 114294784
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009751053159478436,
      "loss": 2.0151,
      "theoretical_loss": 4.71354873111554,
      "tokens_seen": 114360320
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009750852557673019,
      "loss": 1.9781,
      "theoretical_loss": 4.7131836318503115,
      "tokens_seen": 114425856
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009750651955867603,
      "loss": 2.0794,
      "theoretical_loss": 4.7128188001428395,
      "tokens_seen": 114491392
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009750451354062187,
      "loss": 2.075,
      "theoretical_loss": 4.712454235644064,
      "tokens_seen": 114556928
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009750250752256771,
      "loss": 2.1529,
      "theoretical_loss": 4.71208993800558,
      "tokens_seen": 114622464
    },
    {
      "epoch": 0.03,
      "objective/train/advantage_avg": 0.49528729915618896,
      "objective/train/docs_used": 48994,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.978676676750183,
      "objective/train/original_loss": 1.9786767959594727,
      "objective/train/theoretical_loss": 4.711725906879634,
      "objective/train/tokens_used": 135148000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24554309248924255,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507770776748657,
      "objective/train/weighted_lm_loss": 2.0792107582092285,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.711725906879634,
      "tokens_seen": 114688000
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009750050150451354,
      "loss": 2.0281,
      "theoretical_loss": 4.711725906879634,
      "tokens_seen": 114688000
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009749849548645938,
      "loss": 2.0641,
      "theoretical_loss": 4.7113621419191265,
      "tokens_seen": 114753536
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009749648946840521,
      "loss": 2.1124,
      "theoretical_loss": 4.710998642777606,
      "tokens_seen": 114819072
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009749448345035106,
      "loss": 2.0097,
      "theoretical_loss": 4.710635409109269,
      "tokens_seen": 114884608
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009749247743229689,
      "loss": 2.2112,
      "theoretical_loss": 4.710272440568962,
      "tokens_seen": 114950144
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009749047141424273,
      "loss": 2.1056,
      "theoretical_loss": 4.709909736812172,
      "tokens_seen": 115015680
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009748846539618856,
      "loss": 2.2072,
      "theoretical_loss": 4.709547297495034,
      "tokens_seen": 115081216
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009748645937813441,
      "loss": 2.1081,
      "theoretical_loss": 4.709185122274322,
      "tokens_seen": 115146752
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009748445336008024,
      "loss": 2.0524,
      "theoretical_loss": 4.708823210807452,
      "tokens_seen": 115212288
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009748244734202608,
      "loss": 1.8597,
      "theoretical_loss": 4.708461562752479,
      "tokens_seen": 115277824
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009748044132397191,
      "loss": 2.0625,
      "theoretical_loss": 4.708100177768094,
      "tokens_seen": 115343360
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009747843530591775,
      "loss": 2.0811,
      "theoretical_loss": 4.707739055513627,
      "tokens_seen": 115408896
    },
    {
      "epoch": 0.03,
      "learning_rate": 0.0009747642928786359,
      "loss": 2.183,
      "theoretical_loss": 4.707378195649038,
      "tokens_seen": 115474432
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009747442326980943,
      "loss": 1.9737,
      "theoretical_loss": 4.7070175978349225,
      "tokens_seen": 115539968
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009747241725175526,
      "loss": 2.1049,
      "theoretical_loss": 4.7066572617325075,
      "tokens_seen": 115605504
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009747041123370111,
      "loss": 2.0388,
      "theoretical_loss": 4.706297187003647,
      "tokens_seen": 115671040
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009746840521564694,
      "loss": 2.0434,
      "theoretical_loss": 4.705937373310828,
      "tokens_seen": 115736576
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009746639919759278,
      "loss": 2.2422,
      "theoretical_loss": 4.705577820317158,
      "tokens_seen": 115802112
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009746439317953862,
      "loss": 2.1174,
      "theoretical_loss": 4.705218527686375,
      "tokens_seen": 115867648
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009746238716148445,
      "loss": 2.1772,
      "theoretical_loss": 4.704859495082837,
      "tokens_seen": 115933184
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009746038114343029,
      "loss": 2.1444,
      "theoretical_loss": 4.704500722171526,
      "tokens_seen": 115998720
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009745837512537614,
      "loss": 2.2104,
      "theoretical_loss": 4.704142208618045,
      "tokens_seen": 116064256
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009745636910732196,
      "loss": 2.0155,
      "theoretical_loss": 4.703783954088612,
      "tokens_seen": 116129792
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009745436308926781,
      "loss": 2.3143,
      "theoretical_loss": 4.703425958250071,
      "tokens_seen": 116195328
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009745235707121363,
      "loss": 2.1153,
      "theoretical_loss": 4.703068220769872,
      "tokens_seen": 116260864
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009745035105315948,
      "loss": 2.0486,
      "theoretical_loss": 4.702710741316087,
      "tokens_seen": 116326400
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009744834503510532,
      "loss": 2.1845,
      "theoretical_loss": 4.702353519557398,
      "tokens_seen": 116391936
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009744633901705115,
      "loss": 2.2034,
      "theoretical_loss": 4.701996555163101,
      "tokens_seen": 116457472
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00097444332998997,
      "loss": 2.0476,
      "theoretical_loss": 4.701639847803098,
      "tokens_seen": 116523008
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009744232698094282,
      "loss": 2.1367,
      "theoretical_loss": 4.701283397147904,
      "tokens_seen": 116588544
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009744032096288867,
      "loss": 2.0433,
      "theoretical_loss": 4.700927202868639,
      "tokens_seen": 116654080
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009743831494483451,
      "loss": 2.0795,
      "theoretical_loss": 4.700571264637029,
      "tokens_seen": 116719616
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009743630892678034,
      "loss": 1.985,
      "theoretical_loss": 4.700215582125406,
      "tokens_seen": 116785152
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009743430290872618,
      "loss": 2.0779,
      "theoretical_loss": 4.699860155006702,
      "tokens_seen": 116850688
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009743229689067203,
      "loss": 2.0759,
      "theoretical_loss": 4.699504982954452,
      "tokens_seen": 116916224
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009743029087261785,
      "loss": 2.1299,
      "theoretical_loss": 4.699150065642794,
      "tokens_seen": 116981760
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000974282848545637,
      "loss": 1.9488,
      "theoretical_loss": 4.698795402746457,
      "tokens_seen": 117047296
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009742627883650953,
      "loss": 2.0521,
      "theoretical_loss": 4.6984409939407765,
      "tokens_seen": 117112832
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009742427281845537,
      "loss": 2.3108,
      "theoretical_loss": 4.698086838901676,
      "tokens_seen": 117178368
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009742226680040121,
      "loss": 2.0137,
      "theoretical_loss": 4.697732937305678,
      "tokens_seen": 117243904
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009742026078234705,
      "loss": 2.1527,
      "theoretical_loss": 4.697379288829895,
      "tokens_seen": 117309440
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009741825476429288,
      "loss": 1.9996,
      "theoretical_loss": 4.697025893152032,
      "tokens_seen": 117374976
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009741624874623872,
      "loss": 2.1059,
      "theoretical_loss": 4.696672749950385,
      "tokens_seen": 117440512
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009741424272818455,
      "loss": 1.938,
      "theoretical_loss": 4.696319858903838,
      "tokens_seen": 117506048
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000974122367101304,
      "loss": 2.1274,
      "theoretical_loss": 4.69596721969186,
      "tokens_seen": 117571584
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009741023069207623,
      "loss": 2.1286,
      "theoretical_loss": 4.69561483199451,
      "tokens_seen": 117637120
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009740822467402207,
      "loss": 2.0024,
      "theoretical_loss": 4.695262695492428,
      "tokens_seen": 117702656
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009740621865596791,
      "loss": 1.99,
      "theoretical_loss": 4.694910809866839,
      "tokens_seen": 117768192
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009740421263791375,
      "loss": 2.1947,
      "theoretical_loss": 4.694559174799549,
      "tokens_seen": 117833728
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009740220661985958,
      "loss": 2.0581,
      "theoretical_loss": 4.694207789972943,
      "tokens_seen": 117899264
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.4920607805252075,
      "objective/train/docs_used": 50256,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.054048776626587,
      "objective/train/original_loss": 2.054048776626587,
      "objective/train/theoretical_loss": 4.693856655069986,
      "objective/train/tokens_used": 138424800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2430429458618164,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504416227340698,
      "objective/train/weighted_lm_loss": 2.157548666000366,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9048373699188232,
      "theoretical_loss": 4.693856655069986,
      "tokens_seen": 117964800
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009740020060180542,
      "loss": 2.1324,
      "theoretical_loss": 4.693856655069986,
      "tokens_seen": 117964800
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009739819458375125,
      "loss": 2.1721,
      "theoretical_loss": 4.693505769774223,
      "tokens_seen": 118030336
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000973961885656971,
      "loss": 2.1294,
      "theoretical_loss": 4.6931551337697694,
      "tokens_seen": 118095872
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009739418254764293,
      "loss": 2.0215,
      "theoretical_loss": 4.692804746741322,
      "tokens_seen": 118161408
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009739217652958877,
      "loss": 2.1654,
      "theoretical_loss": 4.692454608374145,
      "tokens_seen": 118226944
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000973901705115346,
      "loss": 2.1852,
      "theoretical_loss": 4.692104718354077,
      "tokens_seen": 118292480
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009738816449348045,
      "loss": 1.9563,
      "theoretical_loss": 4.69175507636753,
      "tokens_seen": 118358016
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009738615847542628,
      "loss": 2.0985,
      "theoretical_loss": 4.691405682101482,
      "tokens_seen": 118423552
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009738415245737212,
      "loss": 2.0069,
      "theoretical_loss": 4.6910565352434785,
      "tokens_seen": 118489088
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009738214643931796,
      "loss": 2.1439,
      "theoretical_loss": 4.690707635481634,
      "tokens_seen": 118554624
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009738014042126379,
      "loss": 2.0671,
      "theoretical_loss": 4.690358982504628,
      "tokens_seen": 118620160
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009737813440320963,
      "loss": 1.9935,
      "theoretical_loss": 4.690010576001702,
      "tokens_seen": 118685696
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009737612838515547,
      "loss": 2.0938,
      "theoretical_loss": 4.689662415662662,
      "tokens_seen": 118751232
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000973741223671013,
      "loss": 2.0523,
      "theoretical_loss": 4.689314501177874,
      "tokens_seen": 118816768
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009737211634904715,
      "loss": 2.0498,
      "theoretical_loss": 4.6889668322382665,
      "tokens_seen": 118882304
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009737011033099298,
      "loss": 2.025,
      "theoretical_loss": 4.688619408535323,
      "tokens_seen": 118947840
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009736810431293882,
      "loss": 2.1013,
      "theoretical_loss": 4.688272229761087,
      "tokens_seen": 119013376
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009736609829488466,
      "loss": 2.1606,
      "theoretical_loss": 4.687925295608157,
      "tokens_seen": 119078912
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009736409227683049,
      "loss": 2.1686,
      "theoretical_loss": 4.687578605769687,
      "tokens_seen": 119144448
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009736208625877633,
      "loss": 2.0253,
      "theoretical_loss": 4.6872321599393825,
      "tokens_seen": 119209984
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009736008024072218,
      "loss": 2.0954,
      "theoretical_loss": 4.686885957811503,
      "tokens_seen": 119275520
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00097358074222668,
      "loss": 1.9549,
      "theoretical_loss": 4.6865399990808605,
      "tokens_seen": 119341056
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009735606820461385,
      "loss": 1.9711,
      "theoretical_loss": 4.686194283442811,
      "tokens_seen": 119406592
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009735406218655967,
      "loss": 2.1033,
      "theoretical_loss": 4.685848810593263,
      "tokens_seen": 119472128
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009735205616850552,
      "loss": 1.9815,
      "theoretical_loss": 4.685503580228671,
      "tokens_seen": 119537664
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009735005015045136,
      "loss": 2.1605,
      "theoretical_loss": 4.685158592046037,
      "tokens_seen": 119603200
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009734804413239719,
      "loss": 1.963,
      "theoretical_loss": 4.684813845742902,
      "tokens_seen": 119668736
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009734603811434303,
      "loss": 2.0143,
      "theoretical_loss": 4.684469341017357,
      "tokens_seen": 119734272
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009734403209628886,
      "loss": 2.0491,
      "theoretical_loss": 4.684125077568028,
      "tokens_seen": 119799808
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000973420260782347,
      "loss": 2.2728,
      "theoretical_loss": 4.683781055094087,
      "tokens_seen": 119865344
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009734002006018055,
      "loss": 2.0802,
      "theoretical_loss": 4.683437273295245,
      "tokens_seen": 119930880
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009733801404212637,
      "loss": 2.1258,
      "theoretical_loss": 4.683093731871746,
      "tokens_seen": 119996416
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009733600802407222,
      "loss": 1.933,
      "theoretical_loss": 4.682750430524376,
      "tokens_seen": 120061952
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009733400200601806,
      "loss": 2.0386,
      "theoretical_loss": 4.682407368954454,
      "tokens_seen": 120127488
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009733199598796389,
      "loss": 2.1227,
      "theoretical_loss": 4.682064546863834,
      "tokens_seen": 120193024
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009732998996990973,
      "loss": 2.0572,
      "theoretical_loss": 4.681721963954903,
      "tokens_seen": 120258560
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009732798395185557,
      "loss": 2.2008,
      "theoretical_loss": 4.6813796199305795,
      "tokens_seen": 120324096
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000973259779338014,
      "loss": 2.0648,
      "theoretical_loss": 4.681037514494313,
      "tokens_seen": 120389632
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009732397191574725,
      "loss": 2.1425,
      "theoretical_loss": 4.680695647350082,
      "tokens_seen": 120455168
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009732196589769307,
      "loss": 2.1755,
      "theoretical_loss": 4.680354018202394,
      "tokens_seen": 120520704
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009731995987963892,
      "loss": 2.08,
      "theoretical_loss": 4.68001262675628,
      "tokens_seen": 120586240
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009731795386158475,
      "loss": 1.8476,
      "theoretical_loss": 4.6796714727173025,
      "tokens_seen": 120651776
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009731594784353059,
      "loss": 2.1225,
      "theoretical_loss": 4.679330555791543,
      "tokens_seen": 120717312
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009731394182547643,
      "loss": 2.1533,
      "theoretical_loss": 4.678989875685607,
      "tokens_seen": 120782848
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009731193580742227,
      "loss": 2.0241,
      "theoretical_loss": 4.678649432106627,
      "tokens_seen": 120848384
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000973099297893681,
      "loss": 1.9959,
      "theoretical_loss": 4.678309224762249,
      "tokens_seen": 120913920
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009730792377131395,
      "loss": 2.239,
      "theoretical_loss": 4.677969253360644,
      "tokens_seen": 120979456
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009730591775325977,
      "loss": 2.1644,
      "theoretical_loss": 4.677629517610498,
      "tokens_seen": 121044992
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009730391173520562,
      "loss": 1.9964,
      "theoretical_loss": 4.677290017221017,
      "tokens_seen": 121110528
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009730190571715145,
      "loss": 2.1193,
      "theoretical_loss": 4.67695075190192,
      "tokens_seen": 121176064
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.4931885600090027,
      "objective/train/docs_used": 51395,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.00555157661438,
      "objective/train/original_loss": 2.00555157661438,
      "objective/train/theoretical_loss": 4.676611721363443,
      "objective/train/tokens_used": 141701600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24360352754592896,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050557255744934,
      "objective/train/weighted_lm_loss": 2.1069064140319824,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.676611721363443,
      "tokens_seen": 121241600
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009729989969909729,
      "loss": 2.107,
      "theoretical_loss": 4.676611721363443,
      "tokens_seen": 121241600
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009729789368104313,
      "loss": 1.9974,
      "theoretical_loss": 4.6762729253163355,
      "tokens_seen": 121307136
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009729588766298897,
      "loss": 2.0427,
      "theoretical_loss": 4.675934363471857,
      "tokens_seen": 121372672
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009729388164493481,
      "loss": 1.9648,
      "theoretical_loss": 4.675596035541782,
      "tokens_seen": 121438208
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009729187562688064,
      "loss": 2.0273,
      "theoretical_loss": 4.675257941238391,
      "tokens_seen": 121503744
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009728986960882649,
      "loss": 2.041,
      "theoretical_loss": 4.674920080274475,
      "tokens_seen": 121569280
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009728786359077232,
      "loss": 2.0907,
      "theoretical_loss": 4.674582452363334,
      "tokens_seen": 121634816
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009728585757271816,
      "loss": 2.0461,
      "theoretical_loss": 4.674245057218775,
      "tokens_seen": 121700352
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00097283851554664,
      "loss": 2.1216,
      "theoretical_loss": 4.673907894555105,
      "tokens_seen": 121765888
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009728184553660983,
      "loss": 1.9807,
      "theoretical_loss": 4.673570964087141,
      "tokens_seen": 121831424
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009727983951855567,
      "loss": 2.1125,
      "theoretical_loss": 4.673234265530201,
      "tokens_seen": 121896960
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009727783350050151,
      "loss": 1.9752,
      "theoretical_loss": 4.672897798600104,
      "tokens_seen": 121962496
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009727582748244734,
      "loss": 2.0491,
      "theoretical_loss": 4.672561563013172,
      "tokens_seen": 122028032
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009727382146439319,
      "loss": 2.019,
      "theoretical_loss": 4.672225558486223,
      "tokens_seen": 122093568
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009727181544633902,
      "loss": 2.064,
      "theoretical_loss": 4.671889784736576,
      "tokens_seen": 122159104
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009726980942828486,
      "loss": 2.0167,
      "theoretical_loss": 4.671554241482047,
      "tokens_seen": 122224640
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000972678034102307,
      "loss": 2.2564,
      "theoretical_loss": 4.671218928440947,
      "tokens_seen": 122290176
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009726579739217653,
      "loss": 2.0625,
      "theoretical_loss": 4.670883845332084,
      "tokens_seen": 122355712
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009726379137412237,
      "loss": 1.9611,
      "theoretical_loss": 4.670548991874758,
      "tokens_seen": 122421248
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009726178535606822,
      "loss": 1.9603,
      "theoretical_loss": 4.670214367788761,
      "tokens_seen": 122486784
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009725977933801404,
      "loss": 2.095,
      "theoretical_loss": 4.66987997279438,
      "tokens_seen": 122552320
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009725777331995989,
      "loss": 1.9932,
      "theoretical_loss": 4.669545806612389,
      "tokens_seen": 122617856
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009725576730190571,
      "loss": 2.0098,
      "theoretical_loss": 4.669211868964052,
      "tokens_seen": 122683392
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009725376128385156,
      "loss": 2.0485,
      "theoretical_loss": 4.668878159571123,
      "tokens_seen": 122748928
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000972517552657974,
      "loss": 2.0402,
      "theoretical_loss": 4.6685446781558415,
      "tokens_seen": 122814464
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009724974924774323,
      "loss": 2.0854,
      "theoretical_loss": 4.668211424440934,
      "tokens_seen": 122880000
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009724774322968907,
      "loss": 2.1726,
      "theoretical_loss": 4.66787839814961,
      "tokens_seen": 122945536
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000972457372116349,
      "loss": 2.0551,
      "theoretical_loss": 4.667545599005566,
      "tokens_seen": 123011072
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009724373119358074,
      "loss": 2.0483,
      "theoretical_loss": 4.667213026732977,
      "tokens_seen": 123076608
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009724172517552659,
      "loss": 1.9799,
      "theoretical_loss": 4.6668806810565036,
      "tokens_seen": 123142144
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009723971915747241,
      "loss": 2.1432,
      "theoretical_loss": 4.666548561701285,
      "tokens_seen": 123207680
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009723771313941826,
      "loss": 2.091,
      "theoretical_loss": 4.666216668392939,
      "tokens_seen": 123273216
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000972357071213641,
      "loss": 2.0325,
      "theoretical_loss": 4.665885000857562,
      "tokens_seen": 123338752
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009723370110330993,
      "loss": 2.0689,
      "theoretical_loss": 4.665553558821729,
      "tokens_seen": 123404288
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009723169508525577,
      "loss": 2.125,
      "theoretical_loss": 4.665222342012491,
      "tokens_seen": 123469824
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009722968906720161,
      "loss": 1.9438,
      "theoretical_loss": 4.664891350157374,
      "tokens_seen": 123535360
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009722768304914744,
      "loss": 2.0849,
      "theoretical_loss": 4.664560582984374,
      "tokens_seen": 123600896
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009722567703109329,
      "loss": 1.9947,
      "theoretical_loss": 4.664230040221966,
      "tokens_seen": 123666432
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009722367101303911,
      "loss": 2.2468,
      "theoretical_loss": 4.663899721599093,
      "tokens_seen": 123731968
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009722166499498496,
      "loss": 1.9996,
      "theoretical_loss": 4.66356962684517,
      "tokens_seen": 123797504
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009721965897693079,
      "loss": 2.03,
      "theoretical_loss": 4.663239755690082,
      "tokens_seen": 123863040
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009721765295887663,
      "loss": 2.0711,
      "theoretical_loss": 4.66291010786418,
      "tokens_seen": 123928576
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009721564694082247,
      "loss": 1.9784,
      "theoretical_loss": 4.6625806830982865,
      "tokens_seen": 123994112
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009721364092276831,
      "loss": 1.9567,
      "theoretical_loss": 4.662251481123688,
      "tokens_seen": 124059648
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009721163490471414,
      "loss": 2.0798,
      "theoretical_loss": 4.661922501672136,
      "tokens_seen": 124125184
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009720962888665999,
      "loss": 2.0622,
      "theoretical_loss": 4.66159374447585,
      "tokens_seen": 124190720
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009720762286860581,
      "loss": 2.0201,
      "theoretical_loss": 4.661265209267507,
      "tokens_seen": 124256256
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009720561685055166,
      "loss": 1.9692,
      "theoretical_loss": 4.660936895780253,
      "tokens_seen": 124321792
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009720361083249749,
      "loss": 2.1423,
      "theoretical_loss": 4.660608803747688,
      "tokens_seen": 124387328
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009720160481444333,
      "loss": 2.0592,
      "theoretical_loss": 4.660280932903879,
      "tokens_seen": 124452864
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.4943527281284332,
      "objective/train/docs_used": 52572,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8934916257858276,
      "objective/train/original_loss": 1.8934917449951172,
      "objective/train/theoretical_loss": 4.659953282983348,
      "objective/train/tokens_used": 144978400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2446935772895813,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050679326057434,
      "objective/train/weighted_lm_loss": 1.9894511699676514,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.659953282983348,
      "tokens_seen": 124518400
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009719959879638917,
      "loss": 1.9832,
      "theoretical_loss": 4.659953282983348,
      "tokens_seen": 124518400
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009719759277833501,
      "loss": 2.0029,
      "theoretical_loss": 4.659625853721076,
      "tokens_seen": 124583936
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009719558676028084,
      "loss": 2.167,
      "theoretical_loss": 4.659298644852502,
      "tokens_seen": 124649472
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009719358074222668,
      "loss": 2.0752,
      "theoretical_loss": 4.65897165611352,
      "tokens_seen": 124715008
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009719157472417252,
      "loss": 2.1872,
      "theoretical_loss": 4.658644887240481,
      "tokens_seen": 124780544
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009718956870611836,
      "loss": 2.0446,
      "theoretical_loss": 4.6583183379701865,
      "tokens_seen": 124846080
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009718756268806419,
      "loss": 2.0463,
      "theoretical_loss": 4.657992008039894,
      "tokens_seen": 124911616
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009718555667001004,
      "loss": 2.1342,
      "theoretical_loss": 4.657665897187313,
      "tokens_seen": 124977152
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009718355065195586,
      "loss": 2.1419,
      "theoretical_loss": 4.657340005150602,
      "tokens_seen": 125042688
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009718154463390171,
      "loss": 2.1212,
      "theoretical_loss": 4.657014331668371,
      "tokens_seen": 125108224
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009717953861584754,
      "loss": 2.106,
      "theoretical_loss": 4.656688876479679,
      "tokens_seen": 125173760
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009717753259779338,
      "loss": 2.0815,
      "theoretical_loss": 4.656363639324032,
      "tokens_seen": 125239296
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009717552657973922,
      "loss": 2.0729,
      "theoretical_loss": 4.656038619941382,
      "tokens_seen": 125304832
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009717352056168506,
      "loss": 1.8985,
      "theoretical_loss": 4.655713818072132,
      "tokens_seen": 125370368
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009717151454363089,
      "loss": 2.0584,
      "theoretical_loss": 4.655389233457123,
      "tokens_seen": 125435904
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009716950852557674,
      "loss": 2.067,
      "theoretical_loss": 4.6550648658376454,
      "tokens_seen": 125501440
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009716750250752256,
      "loss": 2.0179,
      "theoretical_loss": 4.654740714955429,
      "tokens_seen": 125566976
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009716549648946841,
      "loss": 2.0203,
      "theoretical_loss": 4.654416780552648,
      "tokens_seen": 125632512
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009716349047141424,
      "loss": 2.1031,
      "theoretical_loss": 4.654093062371917,
      "tokens_seen": 125698048
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009716148445336008,
      "loss": 2.1167,
      "theoretical_loss": 4.65376956015629,
      "tokens_seen": 125763584
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009715947843530592,
      "loss": 2.0359,
      "theoretical_loss": 4.653446273649259,
      "tokens_seen": 125829120
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009715747241725175,
      "loss": 2.0423,
      "theoretical_loss": 4.653123202594757,
      "tokens_seen": 125894656
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009715546639919759,
      "loss": 2.1098,
      "theoretical_loss": 4.652800346737152,
      "tokens_seen": 125960192
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009715346038114344,
      "loss": 2.062,
      "theoretical_loss": 4.652477705821248,
      "tokens_seen": 126025728
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009715145436308926,
      "loss": 1.9808,
      "theoretical_loss": 4.652155279592286,
      "tokens_seen": 126091264
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009714944834503511,
      "loss": 2.0248,
      "theoretical_loss": 4.651833067795939,
      "tokens_seen": 126156800
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009714744232698093,
      "loss": 1.9461,
      "theoretical_loss": 4.651511070178314,
      "tokens_seen": 126222336
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009714543630892678,
      "loss": 2.1108,
      "theoretical_loss": 4.65118928648595,
      "tokens_seen": 126287872
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009714343029087263,
      "loss": 2.0036,
      "theoretical_loss": 4.650867716465819,
      "tokens_seen": 126353408
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009714142427281845,
      "loss": 1.8282,
      "theoretical_loss": 4.65054635986532,
      "tokens_seen": 126418944
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000971394182547643,
      "loss": 2.0785,
      "theoretical_loss": 4.650225216432285,
      "tokens_seen": 126484480
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009713741223671014,
      "loss": 2.1061,
      "theoretical_loss": 4.649904285914971,
      "tokens_seen": 126550016
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009713540621865597,
      "loss": 2.1424,
      "theoretical_loss": 4.6495835680620665,
      "tokens_seen": 126615552
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009713340020060181,
      "loss": 2.0539,
      "theoretical_loss": 4.649263062622682,
      "tokens_seen": 126681088
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009713139418254765,
      "loss": 1.9926,
      "theoretical_loss": 4.648942769346356,
      "tokens_seen": 126746624
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009712938816449348,
      "loss": 2.0468,
      "theoretical_loss": 4.648622687983053,
      "tokens_seen": 126812160
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009712738214643933,
      "loss": 1.9613,
      "theoretical_loss": 4.648302818283158,
      "tokens_seen": 126877696
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009712537612838515,
      "loss": 2.1172,
      "theoretical_loss": 4.64798315999748,
      "tokens_seen": 126943232
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00097123370110331,
      "loss": 2.1304,
      "theoretical_loss": 4.647663712877252,
      "tokens_seen": 127008768
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009712136409227683,
      "loss": 2.0174,
      "theoretical_loss": 4.6473444766741245,
      "tokens_seen": 127074304
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009711935807422267,
      "loss": 2.11,
      "theoretical_loss": 4.64702545114017,
      "tokens_seen": 127139840
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009711735205616851,
      "loss": 2.0202,
      "theoretical_loss": 4.646706636027881,
      "tokens_seen": 127205376
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009711534603811435,
      "loss": 2.0189,
      "theoretical_loss": 4.646388031090163,
      "tokens_seen": 127270912
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009711334002006018,
      "loss": 2.0739,
      "theoretical_loss": 4.646069636080346,
      "tokens_seen": 127336448
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009711133400200603,
      "loss": 2.0184,
      "theoretical_loss": 4.645751450752172,
      "tokens_seen": 127401984
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009710932798395185,
      "loss": 1.9785,
      "theoretical_loss": 4.645433474859798,
      "tokens_seen": 127467520
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000971073219658977,
      "loss": 1.9877,
      "theoretical_loss": 4.645115708157796,
      "tokens_seen": 127533056
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009710531594784353,
      "loss": 1.9167,
      "theoretical_loss": 4.6447981504011535,
      "tokens_seen": 127598592
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009710330992978937,
      "loss": 2.11,
      "theoretical_loss": 4.644480801345268,
      "tokens_seen": 127664128
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009710130391173521,
      "loss": 2.0886,
      "theoretical_loss": 4.644163660745951,
      "tokens_seen": 127729664
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.49203452467918396,
      "objective/train/docs_used": 53741,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.9939597845077515,
      "objective/train/original_loss": 1.993959665298462,
      "objective/train/theoretical_loss": 4.6438467283594225,
      "objective/train/tokens_used": 148255200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2424752116203308,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050436019897461,
      "objective/train/weighted_lm_loss": 2.0946204662323,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.6438467283594225,
      "tokens_seen": 127795200
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009709929789368105,
      "loss": 2.1437,
      "theoretical_loss": 4.6438467283594225,
      "tokens_seen": 127795200
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009709729187562688,
      "loss": 2.1764,
      "theoretical_loss": 4.643530003942313,
      "tokens_seen": 127860736
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009709528585757272,
      "loss": 1.9627,
      "theoretical_loss": 4.643213487251664,
      "tokens_seen": 127926272
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009709327983951856,
      "loss": 1.9454,
      "theoretical_loss": 4.642897178044924,
      "tokens_seen": 127991808
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000970912738214644,
      "loss": 1.9753,
      "theoretical_loss": 4.6425810760799475,
      "tokens_seen": 128057344
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009708926780341023,
      "loss": 2.1277,
      "theoretical_loss": 4.642265181114996,
      "tokens_seen": 128122880
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009708726178535608,
      "loss": 1.9976,
      "theoretical_loss": 4.641949492908737,
      "tokens_seen": 128188416
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000970852557673019,
      "loss": 2.0455,
      "theoretical_loss": 4.641634011220242,
      "tokens_seen": 128253952
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009708324974924775,
      "loss": 2.1228,
      "theoretical_loss": 4.641318735808988,
      "tokens_seen": 128319488
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009708124373119358,
      "loss": 1.9554,
      "theoretical_loss": 4.641003666434852,
      "tokens_seen": 128385024
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009707923771313942,
      "loss": 2.0163,
      "theoretical_loss": 4.640688802858113,
      "tokens_seen": 128450560
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009707723169508526,
      "loss": 2.0832,
      "theoretical_loss": 4.640374144839454,
      "tokens_seen": 128516096
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000970752256770311,
      "loss": 2.1972,
      "theoretical_loss": 4.640059692139955,
      "tokens_seen": 128581632
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009707321965897693,
      "loss": 1.9591,
      "theoretical_loss": 4.639745444521098,
      "tokens_seen": 128647168
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009707121364092278,
      "loss": 2.0579,
      "theoretical_loss": 4.63943140174476,
      "tokens_seen": 128712704
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000970692076228686,
      "loss": 2.126,
      "theoretical_loss": 4.63911756357322,
      "tokens_seen": 128778240
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009706720160481445,
      "loss": 2.0433,
      "theoretical_loss": 4.63880392976915,
      "tokens_seen": 128843776
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009706519558676028,
      "loss": 1.974,
      "theoretical_loss": 4.638490500095619,
      "tokens_seen": 128909312
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009706318956870612,
      "loss": 2.0815,
      "theoretical_loss": 4.6381772743160905,
      "tokens_seen": 128974848
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009706118355065196,
      "loss": 2.1851,
      "theoretical_loss": 4.637864252194423,
      "tokens_seen": 129040384
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009705917753259779,
      "loss": 2.0548,
      "theoretical_loss": 4.637551433494869,
      "tokens_seen": 129105920
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009705717151454363,
      "loss": 2.0605,
      "theoretical_loss": 4.637238817982071,
      "tokens_seen": 129171456
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009705516549648948,
      "loss": 1.9794,
      "theoretical_loss": 4.636926405421065,
      "tokens_seen": 129236992
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000970531594784353,
      "loss": 2.1056,
      "theoretical_loss": 4.636614195577276,
      "tokens_seen": 129302528
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009705115346038115,
      "loss": 1.8769,
      "theoretical_loss": 4.63630218821652,
      "tokens_seen": 129368064
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009704914744232697,
      "loss": 2.0787,
      "theoretical_loss": 4.635990383105003,
      "tokens_seen": 129433600
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009704714142427282,
      "loss": 1.9702,
      "theoretical_loss": 4.635678780009318,
      "tokens_seen": 129499136
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009704513540621866,
      "loss": 2.0665,
      "theoretical_loss": 4.6353673786964436,
      "tokens_seen": 129564672
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009704312938816449,
      "loss": 1.9906,
      "theoretical_loss": 4.635056178933747,
      "tokens_seen": 129630208
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009704112337011033,
      "loss": 2.0691,
      "theoretical_loss": 4.6347451804889825,
      "tokens_seen": 129695744
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009703911735205618,
      "loss": 1.9572,
      "theoretical_loss": 4.634434383130284,
      "tokens_seen": 129761280
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00097037111334002,
      "loss": 2.1197,
      "theoretical_loss": 4.634123786626175,
      "tokens_seen": 129826816
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009703510531594785,
      "loss": 2.0133,
      "theoretical_loss": 4.633813390745559,
      "tokens_seen": 129892352
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009703309929789367,
      "loss": 2.1973,
      "theoretical_loss": 4.633503195257724,
      "tokens_seen": 129957888
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009703109327983952,
      "loss": 2.1009,
      "theoretical_loss": 4.633193199932336,
      "tokens_seen": 130023424
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009702908726178536,
      "loss": 2.1452,
      "theoretical_loss": 4.632883404539445,
      "tokens_seen": 130088960
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009702708124373119,
      "loss": 2.0481,
      "theoretical_loss": 4.632573808849479,
      "tokens_seen": 130154496
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009702507522567703,
      "loss": 2.0503,
      "theoretical_loss": 4.632264412633246,
      "tokens_seen": 130220032
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009702306920762287,
      "loss": 2.0984,
      "theoretical_loss": 4.631955215661932,
      "tokens_seen": 130285568
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000970210631895687,
      "loss": 2.1714,
      "theoretical_loss": 4.6316462177071,
      "tokens_seen": 130351104
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009701905717151455,
      "loss": 2.1332,
      "theoretical_loss": 4.63133741854069,
      "tokens_seen": 130416640
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009701705115346038,
      "loss": 2.0043,
      "theoretical_loss": 4.631028817935018,
      "tokens_seen": 130482176
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009701504513540622,
      "loss": 1.9436,
      "theoretical_loss": 4.630720415662774,
      "tokens_seen": 130547712
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009701303911735205,
      "loss": 1.9561,
      "theoretical_loss": 4.630412211497022,
      "tokens_seen": 130613248
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000970110330992979,
      "loss": 1.8725,
      "theoretical_loss": 4.630104205211202,
      "tokens_seen": 130678784
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009700902708124373,
      "loss": 2.0146,
      "theoretical_loss": 4.629796396579124,
      "tokens_seen": 130744320
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009700702106318957,
      "loss": 1.9803,
      "theoretical_loss": 4.629488785374969,
      "tokens_seen": 130809856
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000970050150451354,
      "loss": 2.1308,
      "theoretical_loss": 4.629181371373292,
      "tokens_seen": 130875392
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009700300902708125,
      "loss": 2.0169,
      "theoretical_loss": 4.628874154349017,
      "tokens_seen": 130940928
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009700100300902708,
      "loss": 1.9983,
      "theoretical_loss": 4.628567134077436,
      "tokens_seen": 131006464
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.49247100949287415,
      "objective/train/docs_used": 54897,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.9558112621307373,
      "objective/train/original_loss": 1.9558112621307373,
      "objective/train/theoretical_loss": 4.628260310334209,
      "objective/train/tokens_used": 151532000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24331486225128174,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.05048406124115,
      "objective/train/weighted_lm_loss": 2.0544002056121826,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9048373699188232,
      "theoretical_loss": 4.628260310334209,
      "tokens_seen": 131072000
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009699899699097292,
      "loss": 1.9695,
      "theoretical_loss": 4.628260310334209,
      "tokens_seen": 131072000
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009699699097291876,
      "loss": 2.2511,
      "theoretical_loss": 4.6279536828953685,
      "tokens_seen": 131137536
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000969949849548646,
      "loss": 1.9499,
      "theoretical_loss": 4.627647251537309,
      "tokens_seen": 131203072
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009699297893681044,
      "loss": 1.9427,
      "theoretical_loss": 4.627341016036794,
      "tokens_seen": 131268608
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009699097291875627,
      "loss": 2.0919,
      "theoretical_loss": 4.6270349761709495,
      "tokens_seen": 131334144
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009698896690070212,
      "loss": 1.9728,
      "theoretical_loss": 4.626729131717271,
      "tokens_seen": 131399680
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009698696088264794,
      "loss": 1.9052,
      "theoretical_loss": 4.626423482453612,
      "tokens_seen": 131465216
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009698495486459379,
      "loss": 2.1468,
      "theoretical_loss": 4.626118028158196,
      "tokens_seen": 131530752
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009698294884653962,
      "loss": 1.9891,
      "theoretical_loss": 4.625812768609601,
      "tokens_seen": 131596288
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009698094282848546,
      "loss": 1.9941,
      "theoretical_loss": 4.6255077035867735,
      "tokens_seen": 131661824
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000969789368104313,
      "loss": 2.2123,
      "theoretical_loss": 4.625202832869016,
      "tokens_seen": 131727360
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009697693079237714,
      "loss": 2.0563,
      "theoretical_loss": 4.624898156235994,
      "tokens_seen": 131792896
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009697492477432297,
      "loss": 2.0261,
      "theoretical_loss": 4.62459367346773,
      "tokens_seen": 131858432
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009697291875626882,
      "loss": 2.0436,
      "theoretical_loss": 4.624289384344609,
      "tokens_seen": 131923968
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009697091273821464,
      "loss": 2.1076,
      "theoretical_loss": 4.623985288647368,
      "tokens_seen": 131989504
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009696890672016049,
      "loss": 1.9861,
      "theoretical_loss": 4.623681386157104,
      "tokens_seen": 132055040
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009696690070210632,
      "loss": 2.0016,
      "theoretical_loss": 4.623377676655271,
      "tokens_seen": 132120576
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009696489468405216,
      "loss": 2.0137,
      "theoretical_loss": 4.623074159923678,
      "tokens_seen": 132186112
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00096962888665998,
      "loss": 2.0859,
      "theoretical_loss": 4.622770835744487,
      "tokens_seen": 132251648
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009696088264794383,
      "loss": 2.0853,
      "theoretical_loss": 4.622467703900216,
      "tokens_seen": 132317184
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009695887662988967,
      "loss": 2.1195,
      "theoretical_loss": 4.622164764173735,
      "tokens_seen": 132382720
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009695687061183552,
      "loss": 2.0064,
      "theoretical_loss": 4.621862016348267,
      "tokens_seen": 132448256
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009695486459378134,
      "loss": 2.1665,
      "theoretical_loss": 4.621559460207386,
      "tokens_seen": 132513792
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009695285857572719,
      "loss": 2.141,
      "theoretical_loss": 4.621257095535019,
      "tokens_seen": 132579328
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009695085255767301,
      "loss": 2.0802,
      "theoretical_loss": 4.62095492211544,
      "tokens_seen": 132644864
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009694884653961886,
      "loss": 2.1012,
      "theoretical_loss": 4.620652939733276,
      "tokens_seen": 132710400
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000969468405215647,
      "loss": 2.2375,
      "theoretical_loss": 4.6203511481735,
      "tokens_seen": 132775936
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009694483450351053,
      "loss": 2.2068,
      "theoretical_loss": 4.6200495472214325,
      "tokens_seen": 132841472
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009694282848545637,
      "loss": 2.1315,
      "theoretical_loss": 4.6197481366627455,
      "tokens_seen": 132907008
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009694082246740222,
      "loss": 2.1076,
      "theoretical_loss": 4.6194469162834535,
      "tokens_seen": 132972544
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009693881644934804,
      "loss": 2.0728,
      "theoretical_loss": 4.6191458858699175,
      "tokens_seen": 133038080
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009693681043129389,
      "loss": 2.0247,
      "theoretical_loss": 4.618845045208845,
      "tokens_seen": 133103616
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009693480441323971,
      "loss": 2.0239,
      "theoretical_loss": 4.618544394087287,
      "tokens_seen": 133169152
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009693279839518556,
      "loss": 2.0453,
      "theoretical_loss": 4.618243932292639,
      "tokens_seen": 133234688
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000969307923771314,
      "loss": 2.0592,
      "theoretical_loss": 4.617943659612637,
      "tokens_seen": 133300224
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009692878635907723,
      "loss": 1.944,
      "theoretical_loss": 4.617643575835361,
      "tokens_seen": 133365760
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009692678034102307,
      "loss": 1.8841,
      "theoretical_loss": 4.617343680749233,
      "tokens_seen": 133431296
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009692477432296891,
      "loss": 2.0001,
      "theoretical_loss": 4.617043974143014,
      "tokens_seen": 133496832
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009692276830491474,
      "loss": 1.9382,
      "theoretical_loss": 4.616744455805806,
      "tokens_seen": 133562368
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009692076228686059,
      "loss": 2.1043,
      "theoretical_loss": 4.616445125527051,
      "tokens_seen": 133627904
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009691875626880642,
      "loss": 1.9803,
      "theoretical_loss": 4.61614598309653,
      "tokens_seen": 133693440
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009691675025075226,
      "loss": 2.0213,
      "theoretical_loss": 4.6158470283043584,
      "tokens_seen": 133758976
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009691474423269809,
      "loss": 1.9767,
      "theoretical_loss": 4.615548260940992,
      "tokens_seen": 133824512
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009691273821464394,
      "loss": 2.1192,
      "theoretical_loss": 4.615249680797223,
      "tokens_seen": 133890048
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009691073219658977,
      "loss": 2.1185,
      "theoretical_loss": 4.614951287664179,
      "tokens_seen": 133955584
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009690872617853561,
      "loss": 2.1068,
      "theoretical_loss": 4.614653081333321,
      "tokens_seen": 134021120
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009690672016048144,
      "loss": 2.0312,
      "theoretical_loss": 4.614355061596447,
      "tokens_seen": 134086656
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009690471414242729,
      "loss": 2.1178,
      "theoretical_loss": 4.614057228245686,
      "tokens_seen": 134152192
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009690270812437312,
      "loss": 2.1038,
      "theoretical_loss": 4.613759581073502,
      "tokens_seen": 134217728
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009690070210631896,
      "loss": 1.9887,
      "theoretical_loss": 4.61346211987269,
      "tokens_seen": 134283264
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.49365389347076416,
      "objective/train/docs_used": 56110,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.9194551706314087,
      "objective/train/original_loss": 1.9194552898406982,
      "objective/train/theoretical_loss": 4.613164844436379,
      "objective/train/tokens_used": 154808800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24395570158958435,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506056547164917,
      "objective/train/weighted_lm_loss": 2.0164098739624023,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0168063640594482,
      "theoretical_loss": 4.613164844436379,
      "tokens_seen": 134348800
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009689869608826479,
      "loss": 2.1666,
      "theoretical_loss": 4.613164844436379,
      "tokens_seen": 134348800
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009689669007021064,
      "loss": 2.1292,
      "theoretical_loss": 4.6128677545580254,
      "tokens_seen": 134414336
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009689468405215647,
      "loss": 2.0502,
      "theoretical_loss": 4.612570850031418,
      "tokens_seen": 134479872
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009689267803410231,
      "loss": 2.0157,
      "theoretical_loss": 4.612274130650675,
      "tokens_seen": 134545408
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009689067201604814,
      "loss": 1.9991,
      "theoretical_loss": 4.611977596210242,
      "tokens_seen": 134610944
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009688866599799398,
      "loss": 1.9506,
      "theoretical_loss": 4.6116812465048955,
      "tokens_seen": 134676480
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009688665997993982,
      "loss": 2.0571,
      "theoretical_loss": 4.611385081329736,
      "tokens_seen": 134742016
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009688465396188566,
      "loss": 2.0394,
      "theoretical_loss": 4.611089100480193,
      "tokens_seen": 134807552
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009688264794383149,
      "loss": 1.8767,
      "theoretical_loss": 4.610793303752023,
      "tokens_seen": 134873088
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009688064192577734,
      "loss": 2.0354,
      "theoretical_loss": 4.610497690941305,
      "tokens_seen": 134938624
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009687863590772317,
      "loss": 2.1914,
      "theoretical_loss": 4.610202261844444,
      "tokens_seen": 135004160
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009687662988966901,
      "loss": 2.0961,
      "theoretical_loss": 4.609907016258171,
      "tokens_seen": 135069696
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009687462387161484,
      "loss": 2.1632,
      "theoretical_loss": 4.609611953979537,
      "tokens_seen": 135135232
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009687261785356068,
      "loss": 1.9911,
      "theoretical_loss": 4.609317074805919,
      "tokens_seen": 135200768
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009687061183550652,
      "loss": 1.9436,
      "theoretical_loss": 4.6090223785350135,
      "tokens_seen": 135266304
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009686860581745236,
      "loss": 1.974,
      "theoretical_loss": 4.608727864964839,
      "tokens_seen": 135331840
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009686659979939819,
      "loss": 1.9855,
      "theoretical_loss": 4.608433533893738,
      "tokens_seen": 135397376
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009686459378134404,
      "loss": 2.0362,
      "theoretical_loss": 4.608139385120367,
      "tokens_seen": 135462912
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009686258776328986,
      "loss": 2.0603,
      "theoretical_loss": 4.607845418443706,
      "tokens_seen": 135528448
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009686058174523571,
      "loss": 1.95,
      "theoretical_loss": 4.6075516336630535,
      "tokens_seen": 135593984
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009685857572718155,
      "loss": 2.0118,
      "theoretical_loss": 4.607258030578024,
      "tokens_seen": 135659520
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009685656970912738,
      "loss": 2.0902,
      "theoretical_loss": 4.606964608988553,
      "tokens_seen": 135725056
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009685456369107322,
      "loss": 2.266,
      "theoretical_loss": 4.606671368694888,
      "tokens_seen": 135790592
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009685255767301905,
      "loss": 1.9943,
      "theoretical_loss": 4.6063783094975985,
      "tokens_seen": 135856128
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009685055165496489,
      "loss": 2.0378,
      "theoretical_loss": 4.606085431197563,
      "tokens_seen": 135921664
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009684854563691074,
      "loss": 2.0629,
      "theoretical_loss": 4.6057927335959805,
      "tokens_seen": 135987200
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009684653961885657,
      "loss": 2.0653,
      "theoretical_loss": 4.6055002164943595,
      "tokens_seen": 136052736
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009684453360080241,
      "loss": 2.1102,
      "theoretical_loss": 4.605207879694525,
      "tokens_seen": 136118272
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009684252758274826,
      "loss": 1.9009,
      "theoretical_loss": 4.604915722998614,
      "tokens_seen": 136183808
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009684052156469408,
      "loss": 1.9044,
      "theoretical_loss": 4.604623746209077,
      "tokens_seen": 136249344
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009683851554663993,
      "loss": 2.0011,
      "theoretical_loss": 4.604331949128672,
      "tokens_seen": 136314880
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009683650952858575,
      "loss": 2.173,
      "theoretical_loss": 4.604040331560473,
      "tokens_seen": 136380416
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000968345035105316,
      "loss": 2.0807,
      "theoretical_loss": 4.603748893307861,
      "tokens_seen": 136445952
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009683249749247744,
      "loss": 2.007,
      "theoretical_loss": 4.603457634174527,
      "tokens_seen": 136511488
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009683049147442327,
      "loss": 2.0617,
      "theoretical_loss": 4.603166553964474,
      "tokens_seen": 136577024
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009682848545636911,
      "loss": 2.2422,
      "theoretical_loss": 4.602875652482009,
      "tokens_seen": 136642560
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009682647943831495,
      "loss": 1.9923,
      "theoretical_loss": 4.60258492953175,
      "tokens_seen": 136708096
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009682447342026078,
      "loss": 2.0762,
      "theoretical_loss": 4.60229438491862,
      "tokens_seen": 136773632
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009682246740220663,
      "loss": 2.1774,
      "theoretical_loss": 4.60200401844785,
      "tokens_seen": 136839168
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009682046138415246,
      "loss": 1.9386,
      "theoretical_loss": 4.601713829924977,
      "tokens_seen": 136904704
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000968184553660983,
      "loss": 2.037,
      "theoretical_loss": 4.601423819155842,
      "tokens_seen": 136970240
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009681644934804413,
      "loss": 1.9515,
      "theoretical_loss": 4.601133985946591,
      "tokens_seen": 137035776
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009681444332998998,
      "loss": 2.0307,
      "theoretical_loss": 4.6008443301036746,
      "tokens_seen": 137101312
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009681243731193581,
      "loss": 1.9943,
      "theoretical_loss": 4.600554851433845,
      "tokens_seen": 137166848
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009681043129388165,
      "loss": 2.1094,
      "theoretical_loss": 4.60026554974416,
      "tokens_seen": 137232384
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009680842527582748,
      "loss": 2.0925,
      "theoretical_loss": 4.599976424841977,
      "tokens_seen": 137297920
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009680641925777333,
      "loss": 2.0283,
      "theoretical_loss": 4.5996874765349585,
      "tokens_seen": 137363456
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009680441323971916,
      "loss": 2.0397,
      "theoretical_loss": 4.599398704631062,
      "tokens_seen": 137428992
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00096802407221665,
      "loss": 1.9583,
      "theoretical_loss": 4.599110108938549,
      "tokens_seen": 137494528
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009680040120361083,
      "loss": 1.9251,
      "theoretical_loss": 4.598821689265982,
      "tokens_seen": 137560064
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.49249258637428284,
      "objective/train/docs_used": 57180,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.132284641265869,
      "objective/train/original_loss": 2.1322851181030273,
      "objective/train/theoretical_loss": 4.598533445422221,
      "objective/train/tokens_used": 158085600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24288922548294067,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.05048406124115,
      "objective/train/weighted_lm_loss": 2.2398180961608887,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0168063640594482,
      "theoretical_loss": 4.598533445422221,
      "tokens_seen": 137625600
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009679839518555668,
      "loss": 2.1706,
      "theoretical_loss": 4.598533445422221,
      "tokens_seen": 137625600
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009679638916750251,
      "loss": 1.9657,
      "theoretical_loss": 4.598245377216422,
      "tokens_seen": 137691136
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009679438314944835,
      "loss": 1.9385,
      "theoretical_loss": 4.5979574844580435,
      "tokens_seen": 137756672
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009679237713139418,
      "loss": 2.0471,
      "theoretical_loss": 4.597669766956837,
      "tokens_seen": 137822208
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009679037111334002,
      "loss": 2.1599,
      "theoretical_loss": 4.597382224522855,
      "tokens_seen": 137887744
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009678836509528586,
      "loss": 1.9596,
      "theoretical_loss": 4.597094856966441,
      "tokens_seen": 137953280
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967863590772317,
      "loss": 1.9701,
      "theoretical_loss": 4.596807664098238,
      "tokens_seen": 138018816
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009678435305917753,
      "loss": 2.0581,
      "theoretical_loss": 4.596520645729181,
      "tokens_seen": 138084352
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009678234704112338,
      "loss": 2.1893,
      "theoretical_loss": 4.596233801670502,
      "tokens_seen": 138149888
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009678034102306921,
      "loss": 1.8744,
      "theoretical_loss": 4.595947131733725,
      "tokens_seen": 138215424
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009677833500501505,
      "loss": 2.1235,
      "theoretical_loss": 4.595660635730665,
      "tokens_seen": 138280960
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009677632898696088,
      "loss": 2.1337,
      "theoretical_loss": 4.595374313473435,
      "tokens_seen": 138346496
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009677432296890672,
      "loss": 1.9614,
      "theoretical_loss": 4.595088164774435,
      "tokens_seen": 138412032
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009677231695085256,
      "loss": 1.9998,
      "theoretical_loss": 4.594802189446356,
      "tokens_seen": 138477568
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967703109327984,
      "loss": 2.0338,
      "theoretical_loss": 4.594516387302184,
      "tokens_seen": 138543104
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009676830491474423,
      "loss": 2.118,
      "theoretical_loss": 4.5942307581551916,
      "tokens_seen": 138608640
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009676629889669008,
      "loss": 2.167,
      "theoretical_loss": 4.593945301818941,
      "tokens_seen": 138674176
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967642928786359,
      "loss": 1.9812,
      "theoretical_loss": 4.593660018107286,
      "tokens_seen": 138739712
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009676228686058175,
      "loss": 1.9162,
      "theoretical_loss": 4.593374906834365,
      "tokens_seen": 138805248
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009676028084252759,
      "loss": 1.8815,
      "theoretical_loss": 4.593089967814606,
      "tokens_seen": 138870784
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009675827482447342,
      "loss": 2.0387,
      "theoretical_loss": 4.592805200862726,
      "tokens_seen": 138936320
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009675626880641926,
      "loss": 2.0424,
      "theoretical_loss": 4.592520605793725,
      "tokens_seen": 139001856
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009675426278836509,
      "loss": 2.0179,
      "theoretical_loss": 4.592236182422891,
      "tokens_seen": 139067392
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009675225677031093,
      "loss": 1.9477,
      "theoretical_loss": 4.591951930565798,
      "tokens_seen": 139132928
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009675025075225678,
      "loss": 1.9914,
      "theoretical_loss": 4.591667850038302,
      "tokens_seen": 139198464
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967482447342026,
      "loss": 2.1217,
      "theoretical_loss": 4.591383940656547,
      "tokens_seen": 139264000
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009674623871614845,
      "loss": 2.0834,
      "theoretical_loss": 4.59110020223696,
      "tokens_seen": 139329536
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009674423269809429,
      "loss": 2.0082,
      "theoretical_loss": 4.590816634596247,
      "tokens_seen": 139395072
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009674222668004012,
      "loss": 1.883,
      "theoretical_loss": 4.590533237551401,
      "tokens_seen": 139460608
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009674022066198596,
      "loss": 2.0717,
      "theoretical_loss": 4.590250010919696,
      "tokens_seen": 139526144
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967382146439318,
      "loss": 1.941,
      "theoretical_loss": 4.5899669545186885,
      "tokens_seen": 139591680
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009673620862587763,
      "loss": 2.1246,
      "theoretical_loss": 4.589684068166212,
      "tokens_seen": 139657216
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009673420260782348,
      "loss": 1.9705,
      "theoretical_loss": 4.589401351680385,
      "tokens_seen": 139722752
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967321965897693,
      "loss": 2.0961,
      "theoretical_loss": 4.5891188048796,
      "tokens_seen": 139788288
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009673019057171515,
      "loss": 2.0898,
      "theoretical_loss": 4.588836427582536,
      "tokens_seen": 139853824
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009672818455366098,
      "loss": 1.9653,
      "theoretical_loss": 4.588554219608145,
      "tokens_seen": 139919360
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009672617853560682,
      "loss": 2.0289,
      "theoretical_loss": 4.588272180775659,
      "tokens_seen": 139984896
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009672417251755266,
      "loss": 2.0156,
      "theoretical_loss": 4.587990310904587,
      "tokens_seen": 140050432
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967221664994985,
      "loss": 1.8817,
      "theoretical_loss": 4.587708609814715,
      "tokens_seen": 140115968
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009672016048144433,
      "loss": 2.0011,
      "theoretical_loss": 4.587427077326108,
      "tokens_seen": 140181504
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009671815446339017,
      "loss": 2.0467,
      "theoretical_loss": 4.587145713259102,
      "tokens_seen": 140247040
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00096716148445336,
      "loss": 1.9799,
      "theoretical_loss": 4.586864517434312,
      "tokens_seen": 140312576
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009671414242728185,
      "loss": 2.0688,
      "theoretical_loss": 4.586583489672625,
      "tokens_seen": 140378112
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009671213640922768,
      "loss": 1.9412,
      "theoretical_loss": 4.586302629795206,
      "tokens_seen": 140443648
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009671013039117352,
      "loss": 1.9925,
      "theoretical_loss": 4.5860219376234905,
      "tokens_seen": 140509184
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009670812437311936,
      "loss": 2.0053,
      "theoretical_loss": 4.585741412979187,
      "tokens_seen": 140574720
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967061183550652,
      "loss": 1.9645,
      "theoretical_loss": 4.585461055684279,
      "tokens_seen": 140640256
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009670411233701103,
      "loss": 2.204,
      "theoretical_loss": 4.58518086556102,
      "tokens_seen": 140705792
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009670210631895687,
      "loss": 1.9912,
      "theoretical_loss": 4.584900842431934,
      "tokens_seen": 140771328
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000967001003009027,
      "loss": 2.0303,
      "theoretical_loss": 4.58462098611982,
      "tokens_seen": 140836864
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.49588945508003235,
      "objective/train/docs_used": 58312,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7948875427246094,
      "objective/train/original_loss": 1.7948875427246094,
      "objective/train/theoretical_loss": 4.584341296447742,
      "objective/train/tokens_used": 161362400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24614021182060242,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0508403778076172,
      "objective/train/weighted_lm_loss": 1.8861085176467896,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.584341296447742,
      "tokens_seen": 140902400
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009669809428284855,
      "loss": 1.8735,
      "theoretical_loss": 4.584341296447742,
      "tokens_seen": 140902400
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009669608826479439,
      "loss": 2.0184,
      "theoretical_loss": 4.584061773239039,
      "tokens_seen": 140967936
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009669408224674022,
      "loss": 1.9653,
      "theoretical_loss": 4.583782416317316,
      "tokens_seen": 141033472
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009669207622868606,
      "loss": 1.8443,
      "theoretical_loss": 4.583503225506448,
      "tokens_seen": 141099008
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000966900702106319,
      "loss": 2.1068,
      "theoretical_loss": 4.583224200630577,
      "tokens_seen": 141164544
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009668806419257774,
      "loss": 2.0342,
      "theoretical_loss": 4.582945341514114,
      "tokens_seen": 141230080
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009668605817452357,
      "loss": 2.0212,
      "theoretical_loss": 4.582666647981739,
      "tokens_seen": 141295616
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009668405215646942,
      "loss": 2.0599,
      "theoretical_loss": 4.582388119858393,
      "tokens_seen": 141361152
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009668204613841525,
      "loss": 2.0357,
      "theoretical_loss": 4.582109756969289,
      "tokens_seen": 141426688
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009668004012036109,
      "loss": 1.7923,
      "theoretical_loss": 4.581831559139902,
      "tokens_seen": 141492224
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009667803410230693,
      "loss": 1.9931,
      "theoretical_loss": 4.581553526195974,
      "tokens_seen": 141557760
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009667602808425276,
      "loss": 2.0577,
      "theoretical_loss": 4.581275657963511,
      "tokens_seen": 141623296
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000966740220661986,
      "loss": 2.0293,
      "theoretical_loss": 4.580997954268781,
      "tokens_seen": 141688832
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009667201604814444,
      "loss": 1.9801,
      "theoretical_loss": 4.5807204149383205,
      "tokens_seen": 141754368
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009667001003009027,
      "loss": 2.028,
      "theoretical_loss": 4.580443039798922,
      "tokens_seen": 141819904
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009666800401203612,
      "loss": 2.027,
      "theoretical_loss": 4.580165828677647,
      "tokens_seen": 141885440
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009666599799398194,
      "loss": 2.0958,
      "theoretical_loss": 4.579888781401816,
      "tokens_seen": 141950976
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009666399197592779,
      "loss": 2.0398,
      "theoretical_loss": 4.579611897799008,
      "tokens_seen": 142016512
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009666198595787363,
      "loss": 1.9667,
      "theoretical_loss": 4.57933517769707,
      "tokens_seen": 142082048
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009665997993981946,
      "loss": 2.0251,
      "theoretical_loss": 4.579058620924103,
      "tokens_seen": 142147584
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000966579739217653,
      "loss": 1.9583,
      "theoretical_loss": 4.578782227308471,
      "tokens_seen": 142213120
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009665596790371113,
      "loss": 2.063,
      "theoretical_loss": 4.578505996678796,
      "tokens_seen": 142278656
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009665396188565697,
      "loss": 2.1415,
      "theoretical_loss": 4.578229928863959,
      "tokens_seen": 142344192
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009665195586760282,
      "loss": 2.0813,
      "theoretical_loss": 4.577954023693101,
      "tokens_seen": 142409728
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009664994984954864,
      "loss": 1.9436,
      "theoretical_loss": 4.577678280995619,
      "tokens_seen": 142475264
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009664794383149449,
      "loss": 1.9627,
      "theoretical_loss": 4.577402700601168,
      "tokens_seen": 142540800
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009664593781344033,
      "loss": 2.0339,
      "theoretical_loss": 4.57712728233966,
      "tokens_seen": 142606336
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009664393179538616,
      "loss": 1.9898,
      "theoretical_loss": 4.576852026041261,
      "tokens_seen": 142671872
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00096641925777332,
      "loss": 1.9128,
      "theoretical_loss": 4.5765769315363976,
      "tokens_seen": 142737408
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009663991975927783,
      "loss": 2.0191,
      "theoretical_loss": 4.576301998655747,
      "tokens_seen": 142802944
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009663791374122367,
      "loss": 1.9703,
      "theoretical_loss": 4.576027227230245,
      "tokens_seen": 142868480
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009663590772316952,
      "loss": 2.0606,
      "theoretical_loss": 4.5757526170910765,
      "tokens_seen": 142934016
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009663390170511534,
      "loss": 2.0374,
      "theoretical_loss": 4.575478168069687,
      "tokens_seen": 142999552
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009663189568706119,
      "loss": 2.121,
      "theoretical_loss": 4.575203879997771,
      "tokens_seen": 143065088
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009662988966900702,
      "loss": 2.0917,
      "theoretical_loss": 4.574929752707274,
      "tokens_seen": 143130624
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009662788365095286,
      "loss": 1.9206,
      "theoretical_loss": 4.5746557860304,
      "tokens_seen": 143196160
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000966258776328987,
      "loss": 2.0839,
      "theoretical_loss": 4.574381979799599,
      "tokens_seen": 143261696
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009662387161484454,
      "loss": 2.0296,
      "theoretical_loss": 4.574108333847575,
      "tokens_seen": 143327232
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009662186559679037,
      "loss": 1.9963,
      "theoretical_loss": 4.573834848007284,
      "tokens_seen": 143392768
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009661985957873621,
      "loss": 2.0374,
      "theoretical_loss": 4.573561522111927,
      "tokens_seen": 143458304
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009661785356068204,
      "loss": 2.0905,
      "theoretical_loss": 4.573288355994963,
      "tokens_seen": 143523840
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009661584754262789,
      "loss": 1.8426,
      "theoretical_loss": 4.573015349490094,
      "tokens_seen": 143589376
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009661384152457372,
      "loss": 2.0254,
      "theoretical_loss": 4.572742502431272,
      "tokens_seen": 143654912
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009661183550651956,
      "loss": 2.0699,
      "theoretical_loss": 4.572469814652698,
      "tokens_seen": 143720448
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000966098294884654,
      "loss": 2.0789,
      "theoretical_loss": 4.572197285988824,
      "tokens_seen": 143785984
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009660782347041124,
      "loss": 1.925,
      "theoretical_loss": 4.571924916274344,
      "tokens_seen": 143851520
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009660581745235707,
      "loss": 1.8722,
      "theoretical_loss": 4.571652705344202,
      "tokens_seen": 143917056
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009660381143430291,
      "loss": 2.0093,
      "theoretical_loss": 4.571380653033588,
      "tokens_seen": 143982592
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009660180541624874,
      "loss": 2.0186,
      "theoretical_loss": 4.571108759177939,
      "tokens_seen": 144048128
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009659979939819459,
      "loss": 2.0186,
      "theoretical_loss": 4.570837023612935,
      "tokens_seen": 144113664
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.4903619587421417,
      "objective/train/docs_used": 59583,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8303672075271606,
      "objective/train/original_loss": 1.830367088317871,
      "objective/train/theoretical_loss": 4.570565446174504,
      "objective/train/tokens_used": 164639200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2413649708032608,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502631664276123,
      "objective/train/weighted_lm_loss": 1.9222012758255005,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.570565446174504,
      "tokens_seen": 144179200
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009659779338014042,
      "loss": 2.0127,
      "theoretical_loss": 4.570565446174504,
      "tokens_seen": 144179200
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009659578736208626,
      "loss": 1.9161,
      "theoretical_loss": 4.570294026698815,
      "tokens_seen": 144244736
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009659378134403209,
      "loss": 2.0468,
      "theoretical_loss": 4.570022765022285,
      "tokens_seen": 144310272
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009659177532597794,
      "loss": 2.0074,
      "theoretical_loss": 4.569751660981572,
      "tokens_seen": 144375808
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009658976930792377,
      "loss": 1.9636,
      "theoretical_loss": 4.569480714413578,
      "tokens_seen": 144441344
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009658776328986961,
      "loss": 2.0375,
      "theoretical_loss": 4.569209925155445,
      "tokens_seen": 144506880
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009658575727181545,
      "loss": 1.9634,
      "theoretical_loss": 4.568939293044564,
      "tokens_seen": 144572416
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009658375125376129,
      "loss": 2.0068,
      "theoretical_loss": 4.56866881791856,
      "tokens_seen": 144637952
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009658174523570712,
      "loss": 2.078,
      "theoretical_loss": 4.568398499615305,
      "tokens_seen": 144703488
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009657973921765297,
      "loss": 2.0601,
      "theoretical_loss": 4.568128337972907,
      "tokens_seen": 144769024
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009657773319959879,
      "loss": 2.0646,
      "theoretical_loss": 4.567858332829719,
      "tokens_seen": 144834560
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009657572718154464,
      "loss": 1.9944,
      "theoretical_loss": 4.567588484024331,
      "tokens_seen": 144900096
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009657372116349047,
      "loss": 2.0121,
      "theoretical_loss": 4.56731879139557,
      "tokens_seen": 144965632
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009657171514543631,
      "loss": 2.0377,
      "theoretical_loss": 4.567049254782509,
      "tokens_seen": 145031168
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009656970912738215,
      "loss": 1.9483,
      "theoretical_loss": 4.566779874024453,
      "tokens_seen": 145096704
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009656770310932798,
      "loss": 1.8921,
      "theoretical_loss": 4.566510648960948,
      "tokens_seen": 145162240
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009656569709127382,
      "loss": 1.9893,
      "theoretical_loss": 4.566241579431776,
      "tokens_seen": 145227776
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009656369107321967,
      "loss": 1.9413,
      "theoretical_loss": 4.565972665276957,
      "tokens_seen": 145293312
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009656168505516549,
      "loss": 2.1933,
      "theoretical_loss": 4.56570390633675,
      "tokens_seen": 145358848
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009655967903711134,
      "loss": 1.882,
      "theoretical_loss": 4.565435302451645,
      "tokens_seen": 145424384
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009655767301905716,
      "loss": 2.0421,
      "theoretical_loss": 4.565166853462371,
      "tokens_seen": 145489920
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009655566700100301,
      "loss": 2.0148,
      "theoretical_loss": 4.56489855920989,
      "tokens_seen": 145555456
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009655366098294885,
      "loss": 2.1477,
      "theoretical_loss": 4.564630419535405,
      "tokens_seen": 145620992
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009655165496489468,
      "loss": 1.9786,
      "theoretical_loss": 4.564362434280344,
      "tokens_seen": 145686528
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009654964894684052,
      "loss": 2.0122,
      "theoretical_loss": 4.564094603286375,
      "tokens_seen": 145752064
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009654764292878637,
      "loss": 2.0539,
      "theoretical_loss": 4.5638269263954,
      "tokens_seen": 145817600
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000965456369107322,
      "loss": 2.0813,
      "theoretical_loss": 4.563559403449551,
      "tokens_seen": 145883136
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009654363089267804,
      "loss": 1.826,
      "theoretical_loss": 4.563292034291193,
      "tokens_seen": 145948672
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009654162487462387,
      "loss": 1.9389,
      "theoretical_loss": 4.5630248187629245,
      "tokens_seen": 146014208
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009653961885656971,
      "loss": 1.9886,
      "theoretical_loss": 4.562757756707574,
      "tokens_seen": 146079744
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009653761283851556,
      "loss": 2.0087,
      "theoretical_loss": 4.5624908479682045,
      "tokens_seen": 146145280
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009653560682046138,
      "loss": 1.9831,
      "theoretical_loss": 4.562224092388105,
      "tokens_seen": 146210816
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009653360080240723,
      "loss": 1.9913,
      "theoretical_loss": 4.561957489810798,
      "tokens_seen": 146276352
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009653159478435306,
      "loss": 1.9859,
      "theoretical_loss": 4.561691040080035,
      "tokens_seen": 146341888
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000965295887662989,
      "loss": 1.9011,
      "theoretical_loss": 4.561424743039797,
      "tokens_seen": 146407424
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009652758274824474,
      "loss": 2.0157,
      "theoretical_loss": 4.561158598534295,
      "tokens_seen": 146472960
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009652557673019058,
      "loss": 1.9363,
      "theoretical_loss": 4.5608926064079665,
      "tokens_seen": 146538496
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009652357071213641,
      "loss": 2.0811,
      "theoretical_loss": 4.560626766505479,
      "tokens_seen": 146604032
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009652156469408225,
      "loss": 1.9238,
      "theoretical_loss": 4.5603610786717255,
      "tokens_seen": 146669568
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009651955867602808,
      "loss": 2.1267,
      "theoretical_loss": 4.56009554275183,
      "tokens_seen": 146735104
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009651755265797393,
      "loss": 1.9993,
      "theoretical_loss": 4.559830158591139,
      "tokens_seen": 146800640
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009651554663991976,
      "loss": 1.9902,
      "theoretical_loss": 4.559564926035229,
      "tokens_seen": 146866176
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000965135406218656,
      "loss": 1.8406,
      "theoretical_loss": 4.5592998449299,
      "tokens_seen": 146931712
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009651153460381144,
      "loss": 1.9684,
      "theoretical_loss": 4.559034915121179,
      "tokens_seen": 146997248
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009650952858575728,
      "loss": 1.9249,
      "theoretical_loss": 4.558770136455316,
      "tokens_seen": 147062784
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009650752256770311,
      "loss": 2.013,
      "theoretical_loss": 4.55850550877879,
      "tokens_seen": 147128320
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009650551654964895,
      "loss": 2.001,
      "theoretical_loss": 4.558241031938298,
      "tokens_seen": 147193856
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009650351053159478,
      "loss": 1.8982,
      "theoretical_loss": 4.557976705780766,
      "tokens_seen": 147259392
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009650150451354063,
      "loss": 1.9986,
      "theoretical_loss": 4.557712530153342,
      "tokens_seen": 147324928
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009649949849548646,
      "loss": 1.932,
      "theoretical_loss": 4.557448504903396,
      "tokens_seen": 147390464
    },
    {
      "epoch": 0.04,
      "objective/train/advantage_avg": 0.493988960981369,
      "objective/train/docs_used": 60747,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7616697549819946,
      "objective/train/original_loss": 1.761669635772705,
      "objective/train/theoretical_loss": 4.557184629878522,
      "objective/train/tokens_used": 167916000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2443990260362625,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050641417503357,
      "objective/train/weighted_lm_loss": 1.8508374691009521,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 4.557184629878522,
      "tokens_seen": 147456000
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.000964974924774323,
      "loss": 1.8862,
      "theoretical_loss": 4.557184629878522,
      "tokens_seen": 147456000
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009649548645937813,
      "loss": 1.9551,
      "theoretical_loss": 4.556920904926534,
      "tokens_seen": 147521536
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009649348044132398,
      "loss": 1.9787,
      "theoretical_loss": 4.556657329895469,
      "tokens_seen": 147587072
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009649147442326981,
      "loss": 2.0888,
      "theoretical_loss": 4.556393904633584,
      "tokens_seen": 147652608
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009648946840521565,
      "loss": 1.9027,
      "theoretical_loss": 4.5561306289893615,
      "tokens_seen": 147718144
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009648746238716149,
      "loss": 1.9725,
      "theoretical_loss": 4.555867502811498,
      "tokens_seen": 147783680
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009648545636910733,
      "loss": 2.0147,
      "theoretical_loss": 4.5556045259489135,
      "tokens_seen": 147849216
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009648345035105316,
      "loss": 2.0284,
      "theoretical_loss": 4.555341698250748,
      "tokens_seen": 147914752
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.00096481444332999,
      "loss": 2.0678,
      "theoretical_loss": 4.555079019566359,
      "tokens_seen": 147980288
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009647943831494483,
      "loss": 1.9821,
      "theoretical_loss": 4.554816489745324,
      "tokens_seen": 148045824
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009647743229689068,
      "loss": 1.9895,
      "theoretical_loss": 4.554554108637437,
      "tokens_seen": 148111360
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009647542627883651,
      "loss": 2.0104,
      "theoretical_loss": 4.554291876092713,
      "tokens_seen": 148176896
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009647342026078235,
      "loss": 1.8984,
      "theoretical_loss": 4.554029791961382,
      "tokens_seen": 148242432
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009647141424272819,
      "loss": 1.936,
      "theoretical_loss": 4.5537678560938915,
      "tokens_seen": 148307968
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009646940822467402,
      "loss": 2.0408,
      "theoretical_loss": 4.553506068340907,
      "tokens_seen": 148373504
    },
    {
      "epoch": 0.04,
      "learning_rate": 0.0009646740220661986,
      "loss": 1.88,
      "theoretical_loss": 4.55324442855331,
      "tokens_seen": 148439040
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009646539618856571,
      "loss": 1.8749,
      "theoretical_loss": 4.552982936582194,
      "tokens_seen": 148504576
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009646339017051153,
      "loss": 2.0451,
      "theoretical_loss": 4.552721592278875,
      "tokens_seen": 148570112
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009646138415245738,
      "loss": 1.9803,
      "theoretical_loss": 4.552460395494878,
      "tokens_seen": 148635648
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000964593781344032,
      "loss": 1.9701,
      "theoretical_loss": 4.552199346081946,
      "tokens_seen": 148701184
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009645737211634905,
      "loss": 2.0254,
      "theoretical_loss": 4.551938443892035,
      "tokens_seen": 148766720
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009645536609829489,
      "loss": 1.8414,
      "theoretical_loss": 4.551677688777316,
      "tokens_seen": 148832256
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009645336008024072,
      "loss": 2.0927,
      "theoretical_loss": 4.55141708059017,
      "tokens_seen": 148897792
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009645135406218656,
      "loss": 1.9209,
      "theoretical_loss": 4.551156619183196,
      "tokens_seen": 148963328
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009644934804413241,
      "loss": 1.7724,
      "theoretical_loss": 4.5508963044092035,
      "tokens_seen": 149028864
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009644734202607823,
      "loss": 2.0814,
      "theoretical_loss": 4.550636136121211,
      "tokens_seen": 149094400
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009644533600802408,
      "loss": 1.8497,
      "theoretical_loss": 4.5503761141724555,
      "tokens_seen": 149159936
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000964433299899699,
      "loss": 1.8274,
      "theoretical_loss": 4.550116238416379,
      "tokens_seen": 149225472
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009644132397191575,
      "loss": 1.9875,
      "theoretical_loss": 4.54985650870664,
      "tokens_seen": 149291008
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009643931795386159,
      "loss": 1.929,
      "theoretical_loss": 4.549596924897102,
      "tokens_seen": 149356544
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009643731193580742,
      "loss": 1.9954,
      "theoretical_loss": 4.549337486841843,
      "tokens_seen": 149422080
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009643530591775326,
      "loss": 1.8849,
      "theoretical_loss": 4.54907819439515,
      "tokens_seen": 149487616
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000964332998996991,
      "loss": 1.9833,
      "theoretical_loss": 4.54881904741152,
      "tokens_seen": 149553152
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009643129388164493,
      "loss": 1.8151,
      "theoretical_loss": 4.548560045745656,
      "tokens_seen": 149618688
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009642928786359078,
      "loss": 2.1873,
      "theoretical_loss": 4.548301189252473,
      "tokens_seen": 149684224
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000964272818455366,
      "loss": 1.9426,
      "theoretical_loss": 4.548042477787093,
      "tokens_seen": 149749760
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009642527582748245,
      "loss": 2.0007,
      "theoretical_loss": 4.547783911204846,
      "tokens_seen": 149815296
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009642326980942828,
      "loss": 2.0073,
      "theoretical_loss": 4.54752548936127,
      "tokens_seen": 149880832
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009642126379137412,
      "loss": 1.9919,
      "theoretical_loss": 4.54726721211211,
      "tokens_seen": 149946368
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009641925777331996,
      "loss": 1.9581,
      "theoretical_loss": 4.547009079313316,
      "tokens_seen": 150011904
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000964172517552658,
      "loss": 1.9835,
      "theoretical_loss": 4.546751090821047,
      "tokens_seen": 150077440
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009641524573721163,
      "loss": 1.9467,
      "theoretical_loss": 4.546493246491666,
      "tokens_seen": 150142976
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009641323971915748,
      "loss": 1.88,
      "theoretical_loss": 4.546235546181743,
      "tokens_seen": 150208512
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009641123370110331,
      "loss": 2.0288,
      "theoretical_loss": 4.545977989748052,
      "tokens_seen": 150274048
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009640922768304915,
      "loss": 1.8078,
      "theoretical_loss": 4.5457205770475735,
      "tokens_seen": 150339584
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009640722166499498,
      "loss": 1.8783,
      "theoretical_loss": 4.5454633079374895,
      "tokens_seen": 150405120
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009640521564694082,
      "loss": 1.853,
      "theoretical_loss": 4.545206182275189,
      "tokens_seen": 150470656
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009640320962888666,
      "loss": 1.8686,
      "theoretical_loss": 4.544949199918264,
      "tokens_seen": 150536192
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000964012036108325,
      "loss": 1.9122,
      "theoretical_loss": 4.544692360724509,
      "tokens_seen": 150601728
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009639919759277833,
      "loss": 1.8843,
      "theoretical_loss": 4.544435664551921,
      "tokens_seen": 150667264
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.49405166506767273,
      "objective/train/docs_used": 61752,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8966041803359985,
      "objective/train/original_loss": 1.8966041803359985,
      "objective/train/theoretical_loss": 4.5441791112587016,
      "objective/train/tokens_used": 171192800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2443924993276596,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506476163864136,
      "objective/train/weighted_lm_loss": 1.9925545454025269,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 4.5441791112587016,
      "tokens_seen": 150732800
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009639719157472417,
      "loss": 1.9816,
      "theoretical_loss": 4.5441791112587016,
      "tokens_seen": 150732800
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009639518555667002,
      "loss": 2.1063,
      "theoretical_loss": 4.543922700703251,
      "tokens_seen": 150798336
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009639317953861585,
      "loss": 2.0471,
      "theoretical_loss": 4.5436664327441765,
      "tokens_seen": 150863872
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009639117352056169,
      "loss": 2.0619,
      "theoretical_loss": 4.543410307240281,
      "tokens_seen": 150929408
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009638916750250753,
      "loss": 1.9035,
      "theoretical_loss": 4.5431543240505725,
      "tokens_seen": 150994944
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009638716148445337,
      "loss": 1.9168,
      "theoretical_loss": 4.542898483034259,
      "tokens_seen": 151060480
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000963851554663992,
      "loss": 1.9746,
      "theoretical_loss": 4.542642784050747,
      "tokens_seen": 151126016
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009638314944834505,
      "loss": 2.0691,
      "theoretical_loss": 4.542387226959644,
      "tokens_seen": 151191552
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009638114343029087,
      "loss": 1.9151,
      "theoretical_loss": 4.5421318116207585,
      "tokens_seen": 151257088
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009637913741223672,
      "loss": 1.8876,
      "theoretical_loss": 4.541876537894095,
      "tokens_seen": 151322624
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009637713139418255,
      "loss": 1.8877,
      "theoretical_loss": 4.54162140563986,
      "tokens_seen": 151388160
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009637512537612839,
      "loss": 1.8514,
      "theoretical_loss": 4.541366414718456,
      "tokens_seen": 151453696
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009637311935807423,
      "loss": 1.9069,
      "theoretical_loss": 4.541111564990485,
      "tokens_seen": 151519232
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009637111334002006,
      "loss": 1.9673,
      "theoretical_loss": 4.540856856316747,
      "tokens_seen": 151584768
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000963691073219659,
      "loss": 1.8871,
      "theoretical_loss": 4.540602288558238,
      "tokens_seen": 151650304
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009636710130391175,
      "loss": 1.8173,
      "theoretical_loss": 4.5403478615761514,
      "tokens_seen": 151715840
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009636509528585757,
      "loss": 2.0094,
      "theoretical_loss": 4.540093575231879,
      "tokens_seen": 151781376
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009636308926780342,
      "loss": 1.8458,
      "theoretical_loss": 4.539839429387005,
      "tokens_seen": 151846912
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009636108324974924,
      "loss": 1.9317,
      "theoretical_loss": 4.539585423903313,
      "tokens_seen": 151912448
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009635907723169509,
      "loss": 1.9469,
      "theoretical_loss": 4.539331558642782,
      "tokens_seen": 151977984
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009635707121364093,
      "loss": 1.8159,
      "theoretical_loss": 4.539077833467582,
      "tokens_seen": 152043520
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009635506519558676,
      "loss": 2.0562,
      "theoretical_loss": 4.5388242482400845,
      "tokens_seen": 152109056
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000963530591775326,
      "loss": 1.9007,
      "theoretical_loss": 4.53857080282285,
      "tokens_seen": 152174592
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009635105315947845,
      "loss": 1.8905,
      "theoretical_loss": 4.538317497078634,
      "tokens_seen": 152240128
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009634904714142427,
      "loss": 1.8761,
      "theoretical_loss": 4.538064330870389,
      "tokens_seen": 152305664
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009634704112337012,
      "loss": 2.1206,
      "theoretical_loss": 4.537811304061257,
      "tokens_seen": 152371200
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009634503510531594,
      "loss": 1.8516,
      "theoretical_loss": 4.537558416514575,
      "tokens_seen": 152436736
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009634302908726179,
      "loss": 1.955,
      "theoretical_loss": 4.537305668093871,
      "tokens_seen": 152502272
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009634102306920763,
      "loss": 2.0363,
      "theoretical_loss": 4.537053058662869,
      "tokens_seen": 152567808
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009633901705115346,
      "loss": 2.0132,
      "theoretical_loss": 4.53680058808548,
      "tokens_seen": 152633344
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000963370110330993,
      "loss": 1.9282,
      "theoretical_loss": 4.53654825622581,
      "tokens_seen": 152698880
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009633500501504514,
      "loss": 1.9805,
      "theoretical_loss": 4.536296062948156,
      "tokens_seen": 152764416
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009633299899699097,
      "loss": 1.8875,
      "theoretical_loss": 4.536044008117005,
      "tokens_seen": 152829952
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009633099297893682,
      "loss": 2.0565,
      "theoretical_loss": 4.535792091597035,
      "tokens_seen": 152895488
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009632898696088264,
      "loss": 2.0856,
      "theoretical_loss": 4.535540313253112,
      "tokens_seen": 152961024
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009632698094282849,
      "loss": 1.9599,
      "theoretical_loss": 4.535288672950296,
      "tokens_seen": 153026560
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009632497492477432,
      "loss": 1.9393,
      "theoretical_loss": 4.535037170553833,
      "tokens_seen": 153092096
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009632296890672016,
      "loss": 1.9247,
      "theoretical_loss": 4.534785805929161,
      "tokens_seen": 153157632
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.00096320962888666,
      "loss": 1.8164,
      "theoretical_loss": 4.534534578941903,
      "tokens_seen": 153223168
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009631895687061184,
      "loss": 2.0316,
      "theoretical_loss": 4.534283489457875,
      "tokens_seen": 153288704
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009631695085255767,
      "loss": 1.9613,
      "theoretical_loss": 4.534032537343078,
      "tokens_seen": 153354240
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009631494483450352,
      "loss": 1.8766,
      "theoretical_loss": 4.5337817224637025,
      "tokens_seen": 153419776
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009631293881644935,
      "loss": 2.105,
      "theoretical_loss": 4.533531044686123,
      "tokens_seen": 153485312
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009631093279839519,
      "loss": 1.89,
      "theoretical_loss": 4.533280503876906,
      "tokens_seen": 153550848
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009630892678034102,
      "loss": 2.0764,
      "theoretical_loss": 4.533030099902803,
      "tokens_seen": 153616384
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009630692076228686,
      "loss": 1.9391,
      "theoretical_loss": 4.532779832630748,
      "tokens_seen": 153681920
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000963049147442327,
      "loss": 2.0657,
      "theoretical_loss": 4.532529701927868,
      "tokens_seen": 153747456
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009630290872617854,
      "loss": 1.9845,
      "theoretical_loss": 4.53227970766147,
      "tokens_seen": 153812992
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009630090270812437,
      "loss": 1.9766,
      "theoretical_loss": 4.53202984969905,
      "tokens_seen": 153878528
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009629889669007021,
      "loss": 1.949,
      "theoretical_loss": 4.531780127908286,
      "tokens_seen": 153944064
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.49506837129592896,
      "objective/train/docs_used": 63027,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.0724551677703857,
      "objective/train/original_loss": 2.0724549293518066,
      "objective/train/theoretical_loss": 4.531530542157043,
      "objective/train/tokens_used": 174469600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24586887657642365,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507569313049316,
      "objective/train/weighted_lm_loss": 2.1776838302612305,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.531530542157043,
      "tokens_seen": 154009600
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009629689067201605,
      "loss": 2.0624,
      "theoretical_loss": 4.531530542157043,
      "tokens_seen": 154009600
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009629488465396189,
      "loss": 1.997,
      "theoretical_loss": 4.53128109231337,
      "tokens_seen": 154075136
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009629287863590772,
      "loss": 2.1624,
      "theoretical_loss": 4.531031778245499,
      "tokens_seen": 154140672
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009629087261785357,
      "loss": 1.9835,
      "theoretical_loss": 4.530782599821848,
      "tokens_seen": 154206208
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009628886659979939,
      "loss": 1.8497,
      "theoretical_loss": 4.5305335569110134,
      "tokens_seen": 154271744
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009628686058174524,
      "loss": 1.9732,
      "theoretical_loss": 4.530284649381782,
      "tokens_seen": 154337280
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009628485456369107,
      "loss": 2.0882,
      "theoretical_loss": 4.530035877103115,
      "tokens_seen": 154402816
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009628284854563691,
      "loss": 2.0315,
      "theoretical_loss": 4.529787239944165,
      "tokens_seen": 154468352
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009628084252758275,
      "loss": 1.9269,
      "theoretical_loss": 4.529538737774259,
      "tokens_seen": 154533888
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009627883650952859,
      "loss": 2.0243,
      "theoretical_loss": 4.52929037046291,
      "tokens_seen": 154599424
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009627683049147442,
      "loss": 1.9704,
      "theoretical_loss": 4.529042137879809,
      "tokens_seen": 154664960
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009627482447342027,
      "loss": 1.9537,
      "theoretical_loss": 4.528794039894832,
      "tokens_seen": 154730496
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009627281845536609,
      "loss": 1.9413,
      "theoretical_loss": 4.528546076378033,
      "tokens_seen": 154796032
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009627081243731194,
      "loss": 1.8682,
      "theoretical_loss": 4.528298247199649,
      "tokens_seen": 154861568
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009626880641925777,
      "loss": 1.9414,
      "theoretical_loss": 4.528050552230092,
      "tokens_seen": 154927104
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009626680040120361,
      "loss": 1.8703,
      "theoretical_loss": 4.52780299133996,
      "tokens_seen": 154992640
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009626479438314945,
      "loss": 2.079,
      "theoretical_loss": 4.527555564400026,
      "tokens_seen": 155058176
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009626278836509528,
      "loss": 2.0089,
      "theoretical_loss": 4.527308271281243,
      "tokens_seen": 155123712
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009626078234704112,
      "loss": 1.9527,
      "theoretical_loss": 4.527061111854746,
      "tokens_seen": 155189248
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009625877632898697,
      "loss": 2.0116,
      "theoretical_loss": 4.526814085991843,
      "tokens_seen": 155254784
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009625677031093279,
      "loss": 2.0765,
      "theoretical_loss": 4.5265671935640235,
      "tokens_seen": 155320320
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009625476429287864,
      "loss": 1.9764,
      "theoretical_loss": 4.526320434442955,
      "tokens_seen": 155385856
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009625275827482446,
      "loss": 1.9078,
      "theoretical_loss": 4.526073808500481,
      "tokens_seen": 155451392
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009625075225677031,
      "loss": 1.8425,
      "theoretical_loss": 4.525827315608623,
      "tokens_seen": 155516928
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009624874623871615,
      "loss": 1.7777,
      "theoretical_loss": 4.5255809556395805,
      "tokens_seen": 155582464
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009624674022066198,
      "loss": 2.0199,
      "theoretical_loss": 4.525334728465727,
      "tokens_seen": 155648000
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009624473420260783,
      "loss": 2.0546,
      "theoretical_loss": 4.525088633959613,
      "tokens_seen": 155713536
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009624272818455367,
      "loss": 1.8745,
      "theoretical_loss": 4.524842671993966,
      "tokens_seen": 155779072
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000962407221664995,
      "loss": 1.9245,
      "theoretical_loss": 4.524596842441689,
      "tokens_seen": 155844608
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009623871614844534,
      "loss": 2.1329,
      "theoretical_loss": 4.524351145175858,
      "tokens_seen": 155910144
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009623671013039118,
      "loss": 1.8084,
      "theoretical_loss": 4.524105580069728,
      "tokens_seen": 155975680
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009623470411233701,
      "loss": 1.9775,
      "theoretical_loss": 4.523860146996725,
      "tokens_seen": 156041216
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009623269809428286,
      "loss": 1.8121,
      "theoretical_loss": 4.5236148458304495,
      "tokens_seen": 156106752
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009623069207622868,
      "loss": 1.9081,
      "theoretical_loss": 4.523369676444679,
      "tokens_seen": 156172288
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009622868605817453,
      "loss": 1.9997,
      "theoretical_loss": 4.523124638713361,
      "tokens_seen": 156237824
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009622668004012036,
      "loss": 2.0708,
      "theoretical_loss": 4.52287973251062,
      "tokens_seen": 156303360
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000962246740220662,
      "loss": 1.901,
      "theoretical_loss": 4.52263495771075,
      "tokens_seen": 156368896
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009622266800401204,
      "loss": 1.9521,
      "theoretical_loss": 4.522390314188221,
      "tokens_seen": 156434432
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009622066198595788,
      "loss": 1.9751,
      "theoretical_loss": 4.522145801817673,
      "tokens_seen": 156499968
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009621865596790371,
      "loss": 1.9347,
      "theoretical_loss": 4.5219014204739185,
      "tokens_seen": 156565504
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009621664994984956,
      "loss": 1.7942,
      "theoretical_loss": 4.521657170031944,
      "tokens_seen": 156631040
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009621464393179539,
      "loss": 1.9491,
      "theoretical_loss": 4.521413050366905,
      "tokens_seen": 156696576
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009621263791374123,
      "loss": 1.8449,
      "theoretical_loss": 4.521169061354129,
      "tokens_seen": 156762112
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009621063189568706,
      "loss": 1.8015,
      "theoretical_loss": 4.520925202869115,
      "tokens_seen": 156827648
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000962086258776329,
      "loss": 1.9523,
      "theoretical_loss": 4.520681474787532,
      "tokens_seen": 156893184
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009620661985957874,
      "loss": 1.9431,
      "theoretical_loss": 4.520437876985218,
      "tokens_seen": 156958720
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009620461384152458,
      "loss": 1.8395,
      "theoretical_loss": 4.520194409338185,
      "tokens_seen": 157024256
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009620260782347041,
      "loss": 2.0112,
      "theoretical_loss": 4.519951071722611,
      "tokens_seen": 157089792
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009620060180541625,
      "loss": 1.8646,
      "theoretical_loss": 4.519707864014844,
      "tokens_seen": 157155328
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009619859578736209,
      "loss": 1.8881,
      "theoretical_loss": 4.519464786091401,
      "tokens_seen": 157220864
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.4935777187347412,
      "objective/train/docs_used": 64097,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 2.037578582763672,
      "objective/train/original_loss": 2.037578582763672,
      "objective/train/theoretical_loss": 4.519221837828971,
      "objective/train/tokens_used": 177746400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24397633969783783,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505980253219604,
      "objective/train/weighted_lm_loss": 2.1407289505004883,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0168063640594482,
      "theoretical_loss": 4.519221837828971,
      "tokens_seen": 157286400
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009619658976930793,
      "loss": 2.1463,
      "theoretical_loss": 4.519221837828971,
      "tokens_seen": 157286400
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009619458375125376,
      "loss": 1.8261,
      "theoretical_loss": 4.518979019104407,
      "tokens_seen": 157351936
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009619257773319961,
      "loss": 2.0235,
      "theoretical_loss": 4.518736329794732,
      "tokens_seen": 157417472
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009619057171514543,
      "loss": 1.9399,
      "theoretical_loss": 4.5184937697771375,
      "tokens_seen": 157483008
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009618856569709128,
      "loss": 2.0169,
      "theoretical_loss": 4.51825133892898,
      "tokens_seen": 157548544
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009618655967903711,
      "loss": 1.9519,
      "theoretical_loss": 4.51800903712779,
      "tokens_seen": 157614080
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009618455366098295,
      "loss": 1.9318,
      "theoretical_loss": 4.517766864251255,
      "tokens_seen": 157679616
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009618254764292879,
      "loss": 1.878,
      "theoretical_loss": 4.517524820177239,
      "tokens_seen": 157745152
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009618054162487463,
      "loss": 1.8226,
      "theoretical_loss": 4.517282904783764,
      "tokens_seen": 157810688
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009617853560682046,
      "loss": 1.9889,
      "theoretical_loss": 4.5170411179490255,
      "tokens_seen": 157876224
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009617652958876631,
      "loss": 1.9731,
      "theoretical_loss": 4.516799459551378,
      "tokens_seen": 157941760
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009617452357071213,
      "loss": 1.8943,
      "theoretical_loss": 4.516557929469347,
      "tokens_seen": 158007296
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009617251755265798,
      "loss": 1.9482,
      "theoretical_loss": 4.516316527581621,
      "tokens_seen": 158072832
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009617051153460381,
      "loss": 1.9452,
      "theoretical_loss": 4.5160752537670525,
      "tokens_seen": 158138368
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009616850551654965,
      "loss": 1.9644,
      "theoretical_loss": 4.515834107904659,
      "tokens_seen": 158203904
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009616649949849549,
      "loss": 2.0442,
      "theoretical_loss": 4.515593089873624,
      "tokens_seen": 158269440
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009616449348044132,
      "loss": 1.9659,
      "theoretical_loss": 4.515352199553295,
      "tokens_seen": 158334976
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009616248746238716,
      "loss": 2.0096,
      "theoretical_loss": 4.51511143682318,
      "tokens_seen": 158400512
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009616048144433301,
      "loss": 1.9158,
      "theoretical_loss": 4.514870801562955,
      "tokens_seen": 158466048
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009615847542627883,
      "loss": 1.904,
      "theoretical_loss": 4.514630293652454,
      "tokens_seen": 158531584
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009615646940822468,
      "loss": 1.9249,
      "theoretical_loss": 4.514389912971679,
      "tokens_seen": 158597120
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000961544633901705,
      "loss": 1.8967,
      "theoretical_loss": 4.5141496594007915,
      "tokens_seen": 158662656
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009615245737211635,
      "loss": 1.9135,
      "theoretical_loss": 4.513909532820117,
      "tokens_seen": 158728192
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009615045135406219,
      "loss": 1.9641,
      "theoretical_loss": 4.513669533110141,
      "tokens_seen": 158793728
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009614844533600802,
      "loss": 1.9775,
      "theoretical_loss": 4.513429660151513,
      "tokens_seen": 158859264
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009614643931795386,
      "loss": 1.9063,
      "theoretical_loss": 4.513189913825043,
      "tokens_seen": 158924800
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009614443329989971,
      "loss": 1.9086,
      "theoretical_loss": 4.5129502940117,
      "tokens_seen": 158990336
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009614242728184553,
      "loss": 1.803,
      "theoretical_loss": 4.512710800592618,
      "tokens_seen": 159055872
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009614042126379138,
      "loss": 1.919,
      "theoretical_loss": 4.51247143344909,
      "tokens_seen": 159121408
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000961384152457372,
      "loss": 1.8886,
      "theoretical_loss": 4.512232192462566,
      "tokens_seen": 159186944
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009613640922768305,
      "loss": 1.9788,
      "theoretical_loss": 4.511993077514662,
      "tokens_seen": 159252480
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009613440320962889,
      "loss": 1.9061,
      "theoretical_loss": 4.51175408848715,
      "tokens_seen": 159318016
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009613239719157472,
      "loss": 1.9861,
      "theoretical_loss": 4.511515225261961,
      "tokens_seen": 159383552
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009613039117352056,
      "loss": 1.8786,
      "theoretical_loss": 4.511276487721187,
      "tokens_seen": 159449088
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000961283851554664,
      "loss": 1.9009,
      "theoretical_loss": 4.511037875747078,
      "tokens_seen": 159514624
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009612637913741223,
      "loss": 1.9976,
      "theoretical_loss": 4.5107993892220435,
      "tokens_seen": 159580160
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009612437311935808,
      "loss": 1.9536,
      "theoretical_loss": 4.5105610280286506,
      "tokens_seen": 159645696
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009612236710130391,
      "loss": 1.9695,
      "theoretical_loss": 4.5103227920496245,
      "tokens_seen": 159711232
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009612036108324975,
      "loss": 1.9963,
      "theoretical_loss": 4.510084681167848,
      "tokens_seen": 159776768
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009611835506519559,
      "loss": 1.8545,
      "theoretical_loss": 4.509846695266361,
      "tokens_seen": 159842304
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009611634904714143,
      "loss": 1.9444,
      "theoretical_loss": 4.509608834228365,
      "tokens_seen": 159907840
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009611434302908726,
      "loss": 1.897,
      "theoretical_loss": 4.50937109793721,
      "tokens_seen": 159973376
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000961123370110331,
      "loss": 1.7706,
      "theoretical_loss": 4.5091334862764105,
      "tokens_seen": 160038912
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009611033099297893,
      "loss": 1.9551,
      "theoretical_loss": 4.508895999129634,
      "tokens_seen": 160104448
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009610832497492478,
      "loss": 1.9135,
      "theoretical_loss": 4.508658636380705,
      "tokens_seen": 160169984
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009610631895687061,
      "loss": 1.8921,
      "theoretical_loss": 4.508421397913603,
      "tokens_seen": 160235520
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009610431293881645,
      "loss": 2.0087,
      "theoretical_loss": 4.508184283612462,
      "tokens_seen": 160301056
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009610230692076228,
      "loss": 1.9208,
      "theoretical_loss": 4.5079472933615765,
      "tokens_seen": 160366592
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009610030090270813,
      "loss": 2.1351,
      "theoretical_loss": 4.507710427045389,
      "tokens_seen": 160432128
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009609829488465396,
      "loss": 2.0943,
      "theoretical_loss": 4.507473684548502,
      "tokens_seen": 160497664
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.49687397480010986,
      "objective/train/docs_used": 65316,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8788886070251465,
      "objective/train/original_loss": 1.8788886070251465,
      "objective/train/theoretical_loss": 4.507237065755672,
      "objective/train/tokens_used": 181023200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24701941013336182,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0509432554244995,
      "objective/train/weighted_lm_loss": 1.97454035282135,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.507237065755672,
      "tokens_seen": 160563200
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000960962888665998,
      "loss": 1.985,
      "theoretical_loss": 4.507237065755672,
      "tokens_seen": 160563200
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009609428284854565,
      "loss": 1.8503,
      "theoretical_loss": 4.507000570551806,
      "tokens_seen": 160628736
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009609227683049147,
      "loss": 1.8316,
      "theoretical_loss": 4.50676419882197,
      "tokens_seen": 160694272
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009609027081243732,
      "loss": 2.0776,
      "theoretical_loss": 4.506527950451379,
      "tokens_seen": 160759808
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009608826479438315,
      "loss": 1.8979,
      "theoretical_loss": 4.506291825325405,
      "tokens_seen": 160825344
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009608625877632899,
      "loss": 1.9678,
      "theoretical_loss": 4.5060558233295716,
      "tokens_seen": 160890880
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009608425275827483,
      "loss": 1.8766,
      "theoretical_loss": 4.505819944349556,
      "tokens_seen": 160956416
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009608224674022067,
      "loss": 1.95,
      "theoretical_loss": 4.505584188271186,
      "tokens_seen": 161021952
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000960802407221665,
      "loss": 1.9329,
      "theoretical_loss": 4.505348554980444,
      "tokens_seen": 161087488
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009607823470411235,
      "loss": 1.8921,
      "theoretical_loss": 4.5051130443634655,
      "tokens_seen": 161153024
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009607622868605817,
      "loss": 1.7781,
      "theoretical_loss": 4.504877656306535,
      "tokens_seen": 161218560
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009607422266800402,
      "loss": 2.0076,
      "theoretical_loss": 4.504642390696089,
      "tokens_seen": 161284096
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009607221664994985,
      "loss": 1.9829,
      "theoretical_loss": 4.504407247418717,
      "tokens_seen": 161349632
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009607021063189569,
      "loss": 1.9316,
      "theoretical_loss": 4.50417222636116,
      "tokens_seen": 161415168
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009606820461384153,
      "loss": 1.973,
      "theoretical_loss": 4.503937327410306,
      "tokens_seen": 161480704
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009606619859578736,
      "loss": 1.9379,
      "theoretical_loss": 4.5037025504531965,
      "tokens_seen": 161546240
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000960641925777332,
      "loss": 1.9915,
      "theoretical_loss": 4.503467895377025,
      "tokens_seen": 161611776
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009606218655967905,
      "loss": 1.9256,
      "theoretical_loss": 4.50323336206913,
      "tokens_seen": 161677312
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009606018054162487,
      "loss": 1.984,
      "theoretical_loss": 4.502998950417004,
      "tokens_seen": 161742848
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009605817452357072,
      "loss": 1.8998,
      "theoretical_loss": 4.502764660308286,
      "tokens_seen": 161808384
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009605616850551654,
      "loss": 2.0802,
      "theoretical_loss": 4.502530491630768,
      "tokens_seen": 161873920
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009605416248746239,
      "loss": 2.0452,
      "theoretical_loss": 4.502296444272387,
      "tokens_seen": 161939456
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009605215646940823,
      "loss": 1.8631,
      "theoretical_loss": 4.502062518121232,
      "tokens_seen": 162004992
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009605015045135406,
      "loss": 1.9151,
      "theoretical_loss": 4.501828713065537,
      "tokens_seen": 162070528
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000960481444332999,
      "loss": 1.8775,
      "theoretical_loss": 4.501595028993686,
      "tokens_seen": 162136064
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009604613841524575,
      "loss": 1.9531,
      "theoretical_loss": 4.501361465794213,
      "tokens_seen": 162201600
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009604413239719157,
      "loss": 1.9826,
      "theoretical_loss": 4.501128023355796,
      "tokens_seen": 162267136
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009604212637913742,
      "loss": 1.9779,
      "theoretical_loss": 4.500894701567263,
      "tokens_seen": 162332672
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009604012036108325,
      "loss": 1.8989,
      "theoretical_loss": 4.500661500317587,
      "tokens_seen": 162398208
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009603811434302909,
      "loss": 1.8547,
      "theoretical_loss": 4.500428419495892,
      "tokens_seen": 162463744
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009603610832497493,
      "loss": 1.9796,
      "theoretical_loss": 4.500195458991443,
      "tokens_seen": 162529280
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009603410230692076,
      "loss": 2.0481,
      "theoretical_loss": 4.4999626186936545,
      "tokens_seen": 162594816
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000960320962888666,
      "loss": 2.0033,
      "theoretical_loss": 4.499729898492089,
      "tokens_seen": 162660352
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009603009027081244,
      "loss": 2.0799,
      "theoretical_loss": 4.4994972982764505,
      "tokens_seen": 162725888
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009602808425275827,
      "loss": 1.8236,
      "theoretical_loss": 4.499264817936593,
      "tokens_seen": 162791424
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009602607823470412,
      "loss": 2.1797,
      "theoretical_loss": 4.499032457362514,
      "tokens_seen": 162856960
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009602407221664995,
      "loss": 1.9804,
      "theoretical_loss": 4.498800216444353,
      "tokens_seen": 162922496
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009602206619859579,
      "loss": 1.9246,
      "theoretical_loss": 4.498568095072401,
      "tokens_seen": 162988032
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009602006018054163,
      "loss": 1.9251,
      "theoretical_loss": 4.498336093137089,
      "tokens_seen": 163053568
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009601805416248747,
      "loss": 1.9896,
      "theoretical_loss": 4.498104210528993,
      "tokens_seen": 163119104
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000960160481444333,
      "loss": 1.9048,
      "theoretical_loss": 4.497872447138834,
      "tokens_seen": 163184640
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009601404212637914,
      "loss": 1.9159,
      "theoretical_loss": 4.497640802857477,
      "tokens_seen": 163250176
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009601203610832497,
      "loss": 2.0265,
      "theoretical_loss": 4.49740927757593,
      "tokens_seen": 163315712
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009601003009027082,
      "loss": 1.7832,
      "theoretical_loss": 4.497177871185343,
      "tokens_seen": 163381248
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009600802407221665,
      "loss": 2.0497,
      "theoretical_loss": 4.4969465835770155,
      "tokens_seen": 163446784
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009600601805416249,
      "loss": 1.8603,
      "theoretical_loss": 4.496715414642381,
      "tokens_seen": 163512320
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009600401203610832,
      "loss": 1.9919,
      "theoretical_loss": 4.496484364273021,
      "tokens_seen": 163577856
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009600200601805417,
      "loss": 2.0578,
      "theoretical_loss": 4.49625343236066,
      "tokens_seen": 163643392
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.00096,
      "loss": 1.97,
      "theoretical_loss": 4.496022618797161,
      "tokens_seen": 163708928
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009599799398194584,
      "loss": 1.8953,
      "theoretical_loss": 4.495791923474533,
      "tokens_seen": 163774464
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.4943779706954956,
      "objective/train/docs_used": 66569,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7600229978561401,
      "objective/train/original_loss": 1.7600231170654297,
      "objective/train/theoretical_loss": 4.495561346284925,
      "objective/train/tokens_used": 184300000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2446385622024536,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050681471824646,
      "objective/train/weighted_lm_loss": 1.849135160446167,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.495561346284925,
      "tokens_seen": 163840000
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009599598796389167,
      "loss": 1.7395,
      "theoretical_loss": 4.495561346284925,
      "tokens_seen": 163840000
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009599398194583751,
      "loss": 1.8734,
      "theoretical_loss": 4.495330887120624,
      "tokens_seen": 163905536
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009599197592778335,
      "loss": 2.0076,
      "theoretical_loss": 4.495100545874065,
      "tokens_seen": 163971072
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009598996990972919,
      "loss": 1.955,
      "theoretical_loss": 4.4948703224378175,
      "tokens_seen": 164036608
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009598796389167502,
      "loss": 1.9929,
      "theoretical_loss": 4.494640216704598,
      "tokens_seen": 164102144
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009598595787362087,
      "loss": 2.0399,
      "theoretical_loss": 4.494410228567258,
      "tokens_seen": 164167680
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000959839518555667,
      "loss": 1.9737,
      "theoretical_loss": 4.494180357918792,
      "tokens_seen": 164233216
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009598194583751254,
      "loss": 1.9149,
      "theoretical_loss": 4.493950604652333,
      "tokens_seen": 164298752
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009597993981945838,
      "loss": 1.9785,
      "theoretical_loss": 4.493720968661158,
      "tokens_seen": 164364288
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009597793380140421,
      "loss": 1.8819,
      "theoretical_loss": 4.493491449838676,
      "tokens_seen": 164429824
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009597592778335005,
      "loss": 1.9185,
      "theoretical_loss": 4.493262048078442,
      "tokens_seen": 164495360
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000959739217652959,
      "loss": 1.928,
      "theoretical_loss": 4.493032763274148,
      "tokens_seen": 164560896
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009597191574724172,
      "loss": 1.9115,
      "theoretical_loss": 4.492803595319623,
      "tokens_seen": 164626432
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009596990972918757,
      "loss": 1.8792,
      "theoretical_loss": 4.492574544108838,
      "tokens_seen": 164691968
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009596790371113339,
      "loss": 1.7851,
      "theoretical_loss": 4.492345609535899,
      "tokens_seen": 164757504
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009596589769307924,
      "loss": 2.0436,
      "theoretical_loss": 4.492116791495052,
      "tokens_seen": 164823040
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009596389167502508,
      "loss": 1.8955,
      "theoretical_loss": 4.49188808988068,
      "tokens_seen": 164888576
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009596188565697091,
      "loss": 1.7672,
      "theoretical_loss": 4.491659504587304,
      "tokens_seen": 164954112
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009595987963891675,
      "loss": 1.8993,
      "theoretical_loss": 4.491431035509584,
      "tokens_seen": 165019648
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009595787362086258,
      "loss": 1.8166,
      "theoretical_loss": 4.4912026825423155,
      "tokens_seen": 165085184
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009595586760280842,
      "loss": 1.8962,
      "theoretical_loss": 4.490974445580429,
      "tokens_seen": 165150720
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009595386158475427,
      "loss": 1.8735,
      "theoretical_loss": 4.490746324518995,
      "tokens_seen": 165216256
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009595185556670009,
      "loss": 1.8615,
      "theoretical_loss": 4.490518319253221,
      "tokens_seen": 165281792
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009594984954864594,
      "loss": 1.8063,
      "theoretical_loss": 4.490290429678447,
      "tokens_seen": 165347328
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009594784353059178,
      "loss": 1.9261,
      "theoretical_loss": 4.490062655690153,
      "tokens_seen": 165412864
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009594583751253761,
      "loss": 1.8684,
      "theoretical_loss": 4.4898349971839515,
      "tokens_seen": 165478400
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009594383149448346,
      "loss": 1.7628,
      "theoretical_loss": 4.489607454055593,
      "tokens_seen": 165543936
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009594182547642929,
      "loss": 1.9646,
      "theoretical_loss": 4.489380026200961,
      "tokens_seen": 165609472
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009593981945837513,
      "loss": 1.9232,
      "theoretical_loss": 4.489152713516077,
      "tokens_seen": 165675008
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009593781344032097,
      "loss": 1.7531,
      "theoretical_loss": 4.488925515897095,
      "tokens_seen": 165740544
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000959358074222668,
      "loss": 1.6834,
      "theoretical_loss": 4.488698433240304,
      "tokens_seen": 165806080
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009593380140421264,
      "loss": 1.949,
      "theoretical_loss": 4.48847146544213,
      "tokens_seen": 165871616
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009593179538615848,
      "loss": 1.8811,
      "theoretical_loss": 4.488244612399129,
      "tokens_seen": 165937152
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009592978936810431,
      "loss": 1.8608,
      "theoretical_loss": 4.488017874007993,
      "tokens_seen": 166002688
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009592778335005016,
      "loss": 1.873,
      "theoretical_loss": 4.487791250165549,
      "tokens_seen": 166068224
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009592577733199599,
      "loss": 1.8062,
      "theoretical_loss": 4.487564740768756,
      "tokens_seen": 166133760
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009592377131394183,
      "loss": 1.8608,
      "theoretical_loss": 4.487338345714707,
      "tokens_seen": 166199296
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009592176529588767,
      "loss": 1.9687,
      "theoretical_loss": 4.487112064900627,
      "tokens_seen": 166264832
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009591975927783351,
      "loss": 1.9348,
      "theoretical_loss": 4.486885898223873,
      "tokens_seen": 166330368
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009591775325977934,
      "loss": 1.8068,
      "theoretical_loss": 4.48665984558194,
      "tokens_seen": 166395904
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009591574724172518,
      "loss": 1.8497,
      "theoretical_loss": 4.486433906872448,
      "tokens_seen": 166461440
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009591374122367101,
      "loss": 1.971,
      "theoretical_loss": 4.486208081993155,
      "tokens_seen": 166526976
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009591173520561686,
      "loss": 1.9106,
      "theoretical_loss": 4.485982370841946,
      "tokens_seen": 166592512
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009590972918756269,
      "loss": 1.878,
      "theoretical_loss": 4.485756773316844,
      "tokens_seen": 166658048
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009590772316950853,
      "loss": 1.9501,
      "theoretical_loss": 4.485531289315997,
      "tokens_seen": 166723584
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009590571715145436,
      "loss": 1.8681,
      "theoretical_loss": 4.485305918737687,
      "tokens_seen": 166789120
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009590371113340021,
      "loss": 1.8118,
      "theoretical_loss": 4.48508066148033,
      "tokens_seen": 166854656
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009590170511534604,
      "loss": 1.8352,
      "theoretical_loss": 4.484855517442467,
      "tokens_seen": 166920192
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009589969909729188,
      "loss": 1.7397,
      "theoretical_loss": 4.484630486522775,
      "tokens_seen": 166985728
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009589769307923771,
      "loss": 1.9343,
      "theoretical_loss": 4.4844055686200575,
      "tokens_seen": 167051264
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.4909823536872864,
      "objective/train/docs_used": 67770,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.918248176574707,
      "objective/train/original_loss": 1.9182484149932861,
      "objective/train/theoretical_loss": 4.48418076363325,
      "objective/train/tokens_used": 187576800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2417370080947876,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503270626068115,
      "objective/train/weighted_lm_loss": 2.014829158782959,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.48418076363325,
      "tokens_seen": 167116800
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009589568706118355,
      "loss": 1.9665,
      "theoretical_loss": 4.48418076363325,
      "tokens_seen": 167116800
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009589368104312939,
      "loss": 1.9255,
      "theoretical_loss": 4.483956071461419,
      "tokens_seen": 167182336
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009589167502507523,
      "loss": 2.0147,
      "theoretical_loss": 4.483731492003757,
      "tokens_seen": 167247872
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009588966900702106,
      "loss": 1.825,
      "theoretical_loss": 4.483507025159591,
      "tokens_seen": 167313408
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009588766298896691,
      "loss": 1.8508,
      "theoretical_loss": 4.483282670828374,
      "tokens_seen": 167378944
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009588565697091274,
      "loss": 1.9944,
      "theoretical_loss": 4.483058428909688,
      "tokens_seen": 167444480
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009588365095285858,
      "loss": 1.8854,
      "theoretical_loss": 4.482834299303246,
      "tokens_seen": 167510016
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009588164493480442,
      "loss": 1.9292,
      "theoretical_loss": 4.482610281908888,
      "tokens_seen": 167575552
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009587963891675025,
      "loss": 1.9696,
      "theoretical_loss": 4.482386376626582,
      "tokens_seen": 167641088
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009587763289869609,
      "loss": 1.8661,
      "theoretical_loss": 4.482162583356427,
      "tokens_seen": 167706624
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009587562688064194,
      "loss": 1.7834,
      "theoretical_loss": 4.481938901998647,
      "tokens_seen": 167772160
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009587362086258776,
      "loss": 2.0171,
      "theoretical_loss": 4.481715332453595,
      "tokens_seen": 167837696
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009587161484453361,
      "loss": 1.8928,
      "theoretical_loss": 4.481491874621751,
      "tokens_seen": 167903232
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009586960882647943,
      "loss": 1.6732,
      "theoretical_loss": 4.481268528403723,
      "tokens_seen": 167968768
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009586760280842528,
      "loss": 1.8741,
      "theoretical_loss": 4.481045293700248,
      "tokens_seen": 168034304
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009586559679037112,
      "loss": 1.8851,
      "theoretical_loss": 4.480822170412186,
      "tokens_seen": 168099840
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009586359077231695,
      "loss": 1.8634,
      "theoretical_loss": 4.480599158440526,
      "tokens_seen": 168165376
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009586158475426279,
      "loss": 1.7684,
      "theoretical_loss": 4.480376257686384,
      "tokens_seen": 168230912
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009585957873620862,
      "loss": 2.012,
      "theoretical_loss": 4.480153468051001,
      "tokens_seen": 168296448
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009585757271815446,
      "loss": 1.9126,
      "theoretical_loss": 4.479930789435746,
      "tokens_seen": 168361984
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009585556670010031,
      "loss": 1.9654,
      "theoretical_loss": 4.479708221742111,
      "tokens_seen": 168427520
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009585356068204613,
      "loss": 1.8602,
      "theoretical_loss": 4.479485764871717,
      "tokens_seen": 168493056
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009585155466399198,
      "loss": 1.9912,
      "theoretical_loss": 4.4792634187263065,
      "tokens_seen": 168558592
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009584954864593782,
      "loss": 2.0017,
      "theoretical_loss": 4.479041183207752,
      "tokens_seen": 168624128
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009584754262788365,
      "loss": 1.8021,
      "theoretical_loss": 4.478819058218047,
      "tokens_seen": 168689664
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009584553660982949,
      "loss": 1.9251,
      "theoretical_loss": 4.478597043659311,
      "tokens_seen": 168755200
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009584353059177533,
      "loss": 1.9435,
      "theoretical_loss": 4.4783751394337905,
      "tokens_seen": 168820736
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009584152457372116,
      "loss": 1.7251,
      "theoretical_loss": 4.478153345443854,
      "tokens_seen": 168886272
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009583951855566701,
      "loss": 1.964,
      "theoretical_loss": 4.477931661591995,
      "tokens_seen": 168951808
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009583751253761283,
      "loss": 1.7283,
      "theoretical_loss": 4.47771008778083,
      "tokens_seen": 169017344
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009583550651955868,
      "loss": 1.7976,
      "theoretical_loss": 4.4774886239131,
      "tokens_seen": 169082880
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009583350050150451,
      "loss": 2.0903,
      "theoretical_loss": 4.477267269891671,
      "tokens_seen": 169148416
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009583149448345035,
      "loss": 1.8024,
      "theoretical_loss": 4.477046025619531,
      "tokens_seen": 169213952
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009582948846539619,
      "loss": 1.9521,
      "theoretical_loss": 4.47682489099979,
      "tokens_seen": 169279488
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009582748244734203,
      "loss": 1.9436,
      "theoretical_loss": 4.476603865935683,
      "tokens_seen": 169345024
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009582547642928786,
      "loss": 1.9069,
      "theoretical_loss": 4.476382950330569,
      "tokens_seen": 169410560
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009582347041123371,
      "loss": 1.9511,
      "theoretical_loss": 4.476162144087925,
      "tokens_seen": 169476096
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009582146439317953,
      "loss": 1.9537,
      "theoretical_loss": 4.475941447111355,
      "tokens_seen": 169541632
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009581945837512538,
      "loss": 2.0403,
      "theoretical_loss": 4.475720859304583,
      "tokens_seen": 169607168
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009581745235707121,
      "loss": 1.9219,
      "theoretical_loss": 4.4755003805714555,
      "tokens_seen": 169672704
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009581544633901705,
      "loss": 1.8615,
      "theoretical_loss": 4.47528001081594,
      "tokens_seen": 169738240
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009581344032096289,
      "loss": 1.9764,
      "theoretical_loss": 4.475059749942126,
      "tokens_seen": 169803776
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009581143430290873,
      "loss": 1.6865,
      "theoretical_loss": 4.474839597854226,
      "tokens_seen": 169869312
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009580942828485456,
      "loss": 2.0214,
      "theoretical_loss": 4.474619554456571,
      "tokens_seen": 169934848
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000958074222668004,
      "loss": 1.8969,
      "theoretical_loss": 4.4743996196536155,
      "tokens_seen": 170000384
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009580541624874624,
      "loss": 1.9962,
      "theoretical_loss": 4.474179793349933,
      "tokens_seen": 170065920
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009580341023069208,
      "loss": 1.7917,
      "theoretical_loss": 4.473960075450218,
      "tokens_seen": 170131456
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009580140421263791,
      "loss": 1.9129,
      "theoretical_loss": 4.473740465859285,
      "tokens_seen": 170196992
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009579939819458375,
      "loss": 1.993,
      "theoretical_loss": 4.47352096448207,
      "tokens_seen": 170262528
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009579739217652958,
      "loss": 2.0726,
      "theoretical_loss": 4.473301571223629,
      "tokens_seen": 170328064
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.4946981370449066,
      "objective/train/docs_used": 69022,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.9940156936645508,
      "objective/train/original_loss": 1.9940155744552612,
      "objective/train/theoretical_loss": 4.473082285989134,
      "objective/train/tokens_used": 190853600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24500080943107605,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507153272628784,
      "objective/train/weighted_lm_loss": 2.094956636428833,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.473082285989134,
      "tokens_seen": 170393600
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009579538615847543,
      "loss": 1.9899,
      "theoretical_loss": 4.473082285989134,
      "tokens_seen": 170393600
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009579338014042127,
      "loss": 2.0756,
      "theoretical_loss": 4.472863108683884,
      "tokens_seen": 170459136
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000957913741223671,
      "loss": 1.9973,
      "theoretical_loss": 4.47264403921329,
      "tokens_seen": 170524672
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009578936810431295,
      "loss": 1.8127,
      "theoretical_loss": 4.472425077482886,
      "tokens_seen": 170590208
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009578736208625878,
      "loss": 1.8438,
      "theoretical_loss": 4.472206223398325,
      "tokens_seen": 170655744
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009578535606820462,
      "loss": 1.9837,
      "theoretical_loss": 4.471987476865376,
      "tokens_seen": 170721280
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009578335005015046,
      "loss": 1.8991,
      "theoretical_loss": 4.471768837789931,
      "tokens_seen": 170786816
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009578134403209629,
      "loss": 1.8756,
      "theoretical_loss": 4.4715503060779955,
      "tokens_seen": 170852352
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009577933801404213,
      "loss": 1.8956,
      "theoretical_loss": 4.471331881635698,
      "tokens_seen": 170917888
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009577733199598798,
      "loss": 1.9264,
      "theoretical_loss": 4.471113564369282,
      "tokens_seen": 170983424
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000957753259779338,
      "loss": 1.7323,
      "theoretical_loss": 4.47089535418511,
      "tokens_seen": 171048960
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009577331995987965,
      "loss": 2.1019,
      "theoretical_loss": 4.470677250989661,
      "tokens_seen": 171114496
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009577131394182547,
      "loss": 1.8691,
      "theoretical_loss": 4.470459254689533,
      "tokens_seen": 171180032
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009576930792377132,
      "loss": 1.8631,
      "theoretical_loss": 4.470241365191441,
      "tokens_seen": 171245568
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009576730190571716,
      "loss": 1.9141,
      "theoretical_loss": 4.4700235824022165,
      "tokens_seen": 171311104
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009576529588766299,
      "loss": 1.9757,
      "theoretical_loss": 4.469805906228807,
      "tokens_seen": 171376640
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009576328986960883,
      "loss": 1.8581,
      "theoretical_loss": 4.469588336578277,
      "tokens_seen": 171442176
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009576128385155466,
      "loss": 1.905,
      "theoretical_loss": 4.469370873357811,
      "tokens_seen": 171507712
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000957592778335005,
      "loss": 1.7967,
      "theoretical_loss": 4.469153516474703,
      "tokens_seen": 171573248
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009575727181544635,
      "loss": 1.893,
      "theoretical_loss": 4.468936265836371,
      "tokens_seen": 171638784
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009575526579739217,
      "loss": 1.8793,
      "theoretical_loss": 4.468719121350343,
      "tokens_seen": 171704320
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009575325977933802,
      "loss": 1.816,
      "theoretical_loss": 4.468502082924266,
      "tokens_seen": 171769856
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009575125376128386,
      "loss": 1.905,
      "theoretical_loss": 4.468285150465899,
      "tokens_seen": 171835392
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009574924774322969,
      "loss": 1.888,
      "theoretical_loss": 4.468068323883122,
      "tokens_seen": 171900928
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009574724172517553,
      "loss": 1.8005,
      "theoretical_loss": 4.467851603083923,
      "tokens_seen": 171966464
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009574523570712137,
      "loss": 1.8414,
      "theoretical_loss": 4.467634987976411,
      "tokens_seen": 172032000
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000957432296890672,
      "loss": 2.0145,
      "theoretical_loss": 4.467418478468808,
      "tokens_seen": 172097536
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009574122367101305,
      "loss": 1.9675,
      "theoretical_loss": 4.467202074469449,
      "tokens_seen": 172163072
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009573921765295887,
      "loss": 1.8405,
      "theoretical_loss": 4.466985775886784,
      "tokens_seen": 172228608
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009573721163490472,
      "loss": 1.9227,
      "theoretical_loss": 4.4667695826293805,
      "tokens_seen": 172294144
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009573520561685055,
      "loss": 1.8589,
      "theoretical_loss": 4.4665534946059156,
      "tokens_seen": 172359680
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009573319959879639,
      "loss": 1.9755,
      "theoretical_loss": 4.4663375117251825,
      "tokens_seen": 172425216
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009573119358074223,
      "loss": 1.8297,
      "theoretical_loss": 4.466121633896087,
      "tokens_seen": 172490752
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009572918756268807,
      "loss": 2.0146,
      "theoretical_loss": 4.4659058610276485,
      "tokens_seen": 172556288
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000957271815446339,
      "loss": 1.837,
      "theoretical_loss": 4.465690193029003,
      "tokens_seen": 172621824
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009572517552657975,
      "loss": 2.0488,
      "theoretical_loss": 4.465474629809394,
      "tokens_seen": 172687360
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009572316950852557,
      "loss": 1.8981,
      "theoretical_loss": 4.465259171278182,
      "tokens_seen": 172752896
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009572116349047142,
      "loss": 1.9211,
      "theoretical_loss": 4.4650438173448395,
      "tokens_seen": 172818432
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009571915747241725,
      "loss": 1.886,
      "theoretical_loss": 4.464828567918951,
      "tokens_seen": 172883968
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009571715145436309,
      "loss": 1.8676,
      "theoretical_loss": 4.464613422910213,
      "tokens_seen": 172949504
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009571514543630893,
      "loss": 1.8263,
      "theoretical_loss": 4.464398382228435,
      "tokens_seen": 173015040
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009571313941825477,
      "loss": 1.9294,
      "theoretical_loss": 4.464183445783537,
      "tokens_seen": 173080576
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000957111334002006,
      "loss": 1.9157,
      "theoretical_loss": 4.463968613485555,
      "tokens_seen": 173146112
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009570912738214644,
      "loss": 2.0705,
      "theoretical_loss": 4.463753885244631,
      "tokens_seen": 173211648
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009570712136409228,
      "loss": 1.8366,
      "theoretical_loss": 4.463539260971023,
      "tokens_seen": 173277184
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009570511534603812,
      "loss": 2.0209,
      "theoretical_loss": 4.463324740575098,
      "tokens_seen": 173342720
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009570310932798395,
      "loss": 1.8134,
      "theoretical_loss": 4.463110323967335,
      "tokens_seen": 173408256
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000957011033099298,
      "loss": 2.0315,
      "theoretical_loss": 4.462896011058323,
      "tokens_seen": 173473792
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009569909729187562,
      "loss": 1.936,
      "theoretical_loss": 4.462681801758762,
      "tokens_seen": 173539328
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009569709127382147,
      "loss": 1.8807,
      "theoretical_loss": 4.4624676959794645,
      "tokens_seen": 173604864
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.4944310486316681,
      "objective/train/docs_used": 70257,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.9424092769622803,
      "objective/train/original_loss": 1.9424092769622803,
      "objective/train/theoretical_loss": 4.462253693631352,
      "objective/train/tokens_used": 194130400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24486778676509857,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506879091262817,
      "objective/train/weighted_lm_loss": 2.0407896041870117,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9834713935852051,
      "theoretical_loss": 4.462253693631352,
      "tokens_seen": 173670400
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000956950852557673,
      "loss": 1.9977,
      "theoretical_loss": 4.462253693631352,
      "tokens_seen": 173670400
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009569307923771314,
      "loss": 1.9687,
      "theoretical_loss": 4.462039794625454,
      "tokens_seen": 173735936
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009569107321965898,
      "loss": 1.9434,
      "theoretical_loss": 4.461825998872914,
      "tokens_seen": 173801472
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009568906720160482,
      "loss": 1.8063,
      "theoretical_loss": 4.461612306284984,
      "tokens_seen": 173867008
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009568706118355065,
      "loss": 1.8382,
      "theoretical_loss": 4.461398716773022,
      "tokens_seen": 173932544
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000956850551654965,
      "loss": 1.9081,
      "theoretical_loss": 4.461185230248503,
      "tokens_seen": 173998080
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009568304914744232,
      "loss": 1.7065,
      "theoretical_loss": 4.460971846623005,
      "tokens_seen": 174063616
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009568104312938817,
      "loss": 1.854,
      "theoretical_loss": 4.4607585658082165,
      "tokens_seen": 174129152
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.00095679037111334,
      "loss": 1.8371,
      "theoretical_loss": 4.460545387715937,
      "tokens_seen": 174194688
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009567703109327984,
      "loss": 1.9075,
      "theoretical_loss": 4.460332312258075,
      "tokens_seen": 174260224
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009567502507522568,
      "loss": 1.9443,
      "theoretical_loss": 4.460119339346643,
      "tokens_seen": 174325760
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009567301905717151,
      "loss": 1.9899,
      "theoretical_loss": 4.4599064688937675,
      "tokens_seen": 174391296
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009567101303911735,
      "loss": 1.7698,
      "theoretical_loss": 4.45969370081168,
      "tokens_seen": 174456832
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000956690070210632,
      "loss": 1.7344,
      "theoretical_loss": 4.459481035012722,
      "tokens_seen": 174522368
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009566700100300902,
      "loss": 1.8361,
      "theoretical_loss": 4.45926847140934,
      "tokens_seen": 174587904
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009566499498495487,
      "loss": 1.8969,
      "theoretical_loss": 4.459056009914093,
      "tokens_seen": 174653440
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009566298896690069,
      "loss": 1.7649,
      "theoretical_loss": 4.458843650439643,
      "tokens_seen": 174718976
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009566098294884654,
      "loss": 1.8389,
      "theoretical_loss": 4.458631392898763,
      "tokens_seen": 174784512
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009565897693079238,
      "loss": 1.9442,
      "theoretical_loss": 4.45841923720433,
      "tokens_seen": 174850048
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009565697091273821,
      "loss": 2.0796,
      "theoretical_loss": 4.45820718326933,
      "tokens_seen": 174915584
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009565496489468405,
      "loss": 2.0647,
      "theoretical_loss": 4.457995231006857,
      "tokens_seen": 174981120
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000956529588766299,
      "loss": 1.8885,
      "theoretical_loss": 4.457783380330108,
      "tokens_seen": 175046656
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009565095285857572,
      "loss": 1.8958,
      "theoretical_loss": 4.4575716311523905,
      "tokens_seen": 175112192
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009564894684052157,
      "loss": 1.8034,
      "theoretical_loss": 4.457359983387116,
      "tokens_seen": 175177728
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000956469408224674,
      "loss": 1.9068,
      "theoretical_loss": 4.457148436947804,
      "tokens_seen": 175243264
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009564493480441324,
      "loss": 1.9427,
      "theoretical_loss": 4.456936991748078,
      "tokens_seen": 175308800
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009564292878635909,
      "loss": 1.9902,
      "theoretical_loss": 4.456725647701669,
      "tokens_seen": 175374336
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009564092276830491,
      "loss": 1.8762,
      "theoretical_loss": 4.456514404722413,
      "tokens_seen": 175439872
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009563891675025076,
      "loss": 1.9261,
      "theoretical_loss": 4.456303262724252,
      "tokens_seen": 175505408
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009563691073219659,
      "loss": 1.8886,
      "theoretical_loss": 4.456092221621233,
      "tokens_seen": 175570944
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009563490471414243,
      "loss": 1.8757,
      "theoretical_loss": 4.455881281327508,
      "tokens_seen": 175636480
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009563289869608827,
      "loss": 1.9739,
      "theoretical_loss": 4.455670441757334,
      "tokens_seen": 175702016
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009563089267803411,
      "loss": 1.8367,
      "theoretical_loss": 4.455459702825074,
      "tokens_seen": 175767552
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009562888665997994,
      "loss": 1.9812,
      "theoretical_loss": 4.455249064445195,
      "tokens_seen": 175833088
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009562688064192579,
      "loss": 1.9976,
      "theoretical_loss": 4.4550385265322685,
      "tokens_seen": 175898624
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009562487462387161,
      "loss": 1.9515,
      "theoretical_loss": 4.454828089000969,
      "tokens_seen": 175964160
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009562286860581746,
      "loss": 1.8709,
      "theoretical_loss": 4.454617751766078,
      "tokens_seen": 176029696
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009562086258776329,
      "loss": 1.8376,
      "theoretical_loss": 4.454407514742479,
      "tokens_seen": 176095232
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009561885656970913,
      "loss": 1.9017,
      "theoretical_loss": 4.45419737784516,
      "tokens_seen": 176160768
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009561685055165497,
      "loss": 1.9527,
      "theoretical_loss": 4.4539873409892135,
      "tokens_seen": 176226304
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009561484453360081,
      "loss": 2.0055,
      "theoretical_loss": 4.453777404089832,
      "tokens_seen": 176291840
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009561283851554664,
      "loss": 1.8616,
      "theoretical_loss": 4.453567567062318,
      "tokens_seen": 176357376
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009561083249749248,
      "loss": 1.9151,
      "theoretical_loss": 4.45335782982207,
      "tokens_seen": 176422912
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009560882647943832,
      "loss": 1.865,
      "theoretical_loss": 4.453148192284595,
      "tokens_seen": 176488448
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009560682046138416,
      "loss": 2.0749,
      "theoretical_loss": 4.4529386543655,
      "tokens_seen": 176553984
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009560481444332999,
      "loss": 2.0762,
      "theoretical_loss": 4.452729215980495,
      "tokens_seen": 176619520
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009560280842527584,
      "loss": 1.8006,
      "theoretical_loss": 4.452519877045393,
      "tokens_seen": 176685056
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009560080240722166,
      "loss": 2.0119,
      "theoretical_loss": 4.452310637476112,
      "tokens_seen": 176750592
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009559879638916751,
      "loss": 1.9248,
      "theoretical_loss": 4.452101497188666,
      "tokens_seen": 176816128
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009559679037111334,
      "loss": 2.002,
      "theoretical_loss": 4.451892456099176,
      "tokens_seen": 176881664
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.49354079365730286,
      "objective/train/docs_used": 71338,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6465367078781128,
      "objective/train/original_loss": 1.6465368270874023,
      "objective/train/theoretical_loss": 4.451683514123864,
      "objective/train/tokens_used": 197407200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2438439577817917,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505937337875366,
      "objective/train/weighted_lm_loss": 1.7298942804336548,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.451683514123864,
      "tokens_seen": 176947200
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009559478435305918,
      "loss": 1.8323,
      "theoretical_loss": 4.451683514123864,
      "tokens_seen": 176947200
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009559277833500502,
      "loss": 1.8433,
      "theoretical_loss": 4.451474671179055,
      "tokens_seen": 177012736
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009559077231695086,
      "loss": 1.9551,
      "theoretical_loss": 4.451265927181171,
      "tokens_seen": 177078272
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009558876629889669,
      "loss": 2.0386,
      "theoretical_loss": 4.451057282046741,
      "tokens_seen": 177143808
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009558676028084254,
      "loss": 1.8875,
      "theoretical_loss": 4.450848735692391,
      "tokens_seen": 177209344
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009558475426278836,
      "loss": 2.0172,
      "theoretical_loss": 4.45064028803485,
      "tokens_seen": 177274880
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009558274824473421,
      "loss": 1.8569,
      "theoretical_loss": 4.450431938990949,
      "tokens_seen": 177340416
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009558074222668004,
      "loss": 1.8493,
      "theoretical_loss": 4.450223688477617,
      "tokens_seen": 177405952
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009557873620862588,
      "loss": 1.9685,
      "theoretical_loss": 4.450015536411886,
      "tokens_seen": 177471488
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009557673019057172,
      "loss": 1.906,
      "theoretical_loss": 4.449807482710888,
      "tokens_seen": 177537024
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009557472417251755,
      "loss": 1.8314,
      "theoretical_loss": 4.449599527291854,
      "tokens_seen": 177602560
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009557271815446339,
      "loss": 1.8091,
      "theoretical_loss": 4.449391670072117,
      "tokens_seen": 177668096
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009557071213640924,
      "loss": 1.9665,
      "theoretical_loss": 4.449183910969108,
      "tokens_seen": 177733632
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009556870611835506,
      "loss": 1.9209,
      "theoretical_loss": 4.44897624990036,
      "tokens_seen": 177799168
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009556670010030091,
      "loss": 1.7531,
      "theoretical_loss": 4.448768686783504,
      "tokens_seen": 177864704
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009556469408224673,
      "loss": 1.8657,
      "theoretical_loss": 4.448561221536272,
      "tokens_seen": 177930240
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009556268806419258,
      "loss": 1.8918,
      "theoretical_loss": 4.448353854076494,
      "tokens_seen": 177995776
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009556068204613842,
      "loss": 1.9234,
      "theoretical_loss": 4.4481465843220995,
      "tokens_seen": 178061312
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009555867602808425,
      "loss": 1.9232,
      "theoretical_loss": 4.447939412191117,
      "tokens_seen": 178126848
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009555667001003009,
      "loss": 1.9091,
      "theoretical_loss": 4.447732337601675,
      "tokens_seen": 178192384
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009555466399197594,
      "loss": 1.8798,
      "theoretical_loss": 4.4475253604719995,
      "tokens_seen": 178257920
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009555265797392176,
      "loss": 1.7858,
      "theoretical_loss": 4.447318480720416,
      "tokens_seen": 178323456
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009555065195586761,
      "loss": 2.0034,
      "theoretical_loss": 4.447111698265346,
      "tokens_seen": 178388992
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009554864593781343,
      "loss": 1.7928,
      "theoretical_loss": 4.446905013025313,
      "tokens_seen": 178454528
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009554663991975928,
      "loss": 1.7196,
      "theoretical_loss": 4.446698424918937,
      "tokens_seen": 178520064
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009554463390170512,
      "loss": 1.9755,
      "theoretical_loss": 4.446491933864936,
      "tokens_seen": 178585600
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009554262788365095,
      "loss": 1.849,
      "theoretical_loss": 4.4462855397821235,
      "tokens_seen": 178651136
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009554062186559679,
      "loss": 1.9696,
      "theoretical_loss": 4.4460792425894144,
      "tokens_seen": 178716672
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009553861584754263,
      "loss": 1.8554,
      "theoretical_loss": 4.44587304220582,
      "tokens_seen": 178782208
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009553660982948846,
      "loss": 1.7727,
      "theoretical_loss": 4.445666938550447,
      "tokens_seen": 178847744
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009553460381143431,
      "loss": 1.8321,
      "theoretical_loss": 4.445460931542502,
      "tokens_seen": 178913280
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009553259779338014,
      "loss": 1.6822,
      "theoretical_loss": 4.4452550211012865,
      "tokens_seen": 178978816
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009553059177532598,
      "loss": 1.9822,
      "theoretical_loss": 4.4450492071462,
      "tokens_seen": 179044352
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009552858575727181,
      "loss": 1.8321,
      "theoretical_loss": 4.444843489596739,
      "tokens_seen": 179109888
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009552657973921765,
      "loss": 2.0049,
      "theoretical_loss": 4.444637868372495,
      "tokens_seen": 179175424
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009552457372116349,
      "loss": 1.9795,
      "theoretical_loss": 4.444432343393158,
      "tokens_seen": 179240960
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009552256770310933,
      "loss": 2.046,
      "theoretical_loss": 4.444226914578513,
      "tokens_seen": 179306496
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009552056168505516,
      "loss": 2.0291,
      "theoretical_loss": 4.444021581848442,
      "tokens_seen": 179372032
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009551855566700101,
      "loss": 2.0813,
      "theoretical_loss": 4.443816345122922,
      "tokens_seen": 179437568
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009551654964894684,
      "loss": 1.7821,
      "theoretical_loss": 4.443611204322026,
      "tokens_seen": 179503104
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009551454363089268,
      "loss": 1.916,
      "theoretical_loss": 4.4434061593659235,
      "tokens_seen": 179568640
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009551253761283851,
      "loss": 1.9274,
      "theoretical_loss": 4.4432012101748795,
      "tokens_seen": 179634176
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009551053159478436,
      "loss": 1.8894,
      "theoretical_loss": 4.442996356669253,
      "tokens_seen": 179699712
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009550852557673019,
      "loss": 1.9832,
      "theoretical_loss": 4.4427915987695,
      "tokens_seen": 179765248
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009550651955867603,
      "loss": 1.8851,
      "theoretical_loss": 4.442586936396171,
      "tokens_seen": 179830784
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009550451354062186,
      "loss": 1.8177,
      "theoretical_loss": 4.442382369469909,
      "tokens_seen": 179896320
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000955025075225677,
      "loss": 1.9614,
      "theoretical_loss": 4.442177897911458,
      "tokens_seen": 179961856
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009550050150451354,
      "loss": 1.9223,
      "theoretical_loss": 4.441973521641649,
      "tokens_seen": 180027392
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009549849548645938,
      "loss": 1.9658,
      "theoretical_loss": 4.441769240581412,
      "tokens_seen": 180092928
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009549648946840521,
      "loss": 2.0016,
      "theoretical_loss": 4.441565054651772,
      "tokens_seen": 180158464
    },
    {
      "epoch": 0.05,
      "objective/train/advantage_avg": 0.4942334294319153,
      "objective/train/docs_used": 72550,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7501457929611206,
      "objective/train/original_loss": 1.7501459121704102,
      "objective/train/theoretical_loss": 4.441360963773846,
      "objective/train/tokens_used": 200684000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2446201890707016,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506670475006104,
      "objective/train/weighted_lm_loss": 1.8388445377349854,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.441360963773846,
      "tokens_seen": 180224000
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009549448345035106,
      "loss": 1.9136,
      "theoretical_loss": 4.441360963773846,
      "tokens_seen": 180224000
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000954924774322969,
      "loss": 2.0606,
      "theoretical_loss": 4.441156967868846,
      "tokens_seen": 180289536
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009549047141424273,
      "loss": 2.0416,
      "theoretical_loss": 4.440953066858077,
      "tokens_seen": 180355072
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009548846539618858,
      "loss": 2.0186,
      "theoretical_loss": 4.440749260662939,
      "tokens_seen": 180420608
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000954864593781344,
      "loss": 1.8847,
      "theoretical_loss": 4.440545549204925,
      "tokens_seen": 180486144
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009548445336008025,
      "loss": 1.7716,
      "theoretical_loss": 4.440341932405623,
      "tokens_seen": 180551680
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009548244734202608,
      "loss": 1.8777,
      "theoretical_loss": 4.4401384101867105,
      "tokens_seen": 180617216
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009548044132397192,
      "loss": 1.8929,
      "theoretical_loss": 4.439934982469962,
      "tokens_seen": 180682752
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009547843530591776,
      "loss": 1.9978,
      "theoretical_loss": 4.439731649177244,
      "tokens_seen": 180748288
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009547642928786359,
      "loss": 1.9419,
      "theoretical_loss": 4.439528410230514,
      "tokens_seen": 180813824
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009547442326980943,
      "loss": 1.767,
      "theoretical_loss": 4.439325265551826,
      "tokens_seen": 180879360
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009547241725175528,
      "loss": 1.9141,
      "theoretical_loss": 4.439122215063323,
      "tokens_seen": 180944896
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.000954704112337011,
      "loss": 1.9685,
      "theoretical_loss": 4.438919258687242,
      "tokens_seen": 181010432
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009546840521564695,
      "loss": 1.826,
      "theoretical_loss": 4.438716396345913,
      "tokens_seen": 181075968
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009546639919759277,
      "loss": 1.9419,
      "theoretical_loss": 4.438513627961757,
      "tokens_seen": 181141504
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009546439317953862,
      "loss": 1.8724,
      "theoretical_loss": 4.438310953457286,
      "tokens_seen": 181207040
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009546238716148446,
      "loss": 1.9956,
      "theoretical_loss": 4.438108372755108,
      "tokens_seen": 181272576
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009546038114343029,
      "loss": 1.9181,
      "theoretical_loss": 4.43790588577792,
      "tokens_seen": 181338112
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009545837512537613,
      "loss": 1.9048,
      "theoretical_loss": 4.437703492448509,
      "tokens_seen": 181403648
    },
    {
      "epoch": 0.05,
      "learning_rate": 0.0009545636910732198,
      "loss": 2.0959,
      "theoretical_loss": 4.437501192689755,
      "tokens_seen": 181469184
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000954543630892678,
      "loss": 1.9178,
      "theoretical_loss": 4.437298986424631,
      "tokens_seen": 181534720
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009545235707121365,
      "loss": 1.9351,
      "theoretical_loss": 4.437096873576199,
      "tokens_seen": 181600256
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009545035105315947,
      "loss": 1.8552,
      "theoretical_loss": 4.436894854067614,
      "tokens_seen": 181665792
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009544834503510532,
      "loss": 1.8738,
      "theoretical_loss": 4.4366929278221185,
      "tokens_seen": 181731328
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009544633901705116,
      "loss": 2.0089,
      "theoretical_loss": 4.436491094763049,
      "tokens_seen": 181796864
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009544433299899699,
      "loss": 1.8173,
      "theoretical_loss": 4.436289354813832,
      "tokens_seen": 181862400
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009544232698094283,
      "loss": 1.9785,
      "theoretical_loss": 4.436087707897984,
      "tokens_seen": 181927936
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009544032096288867,
      "loss": 1.8956,
      "theoretical_loss": 4.435886153939111,
      "tokens_seen": 181993472
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000954383149448345,
      "loss": 1.901,
      "theoretical_loss": 4.435684692860912,
      "tokens_seen": 182059008
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009543630892678035,
      "loss": 1.9325,
      "theoretical_loss": 4.435483324587172,
      "tokens_seen": 182124544
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009543430290872618,
      "loss": 1.9463,
      "theoretical_loss": 4.435282049041769,
      "tokens_seen": 182190080
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009543229689067202,
      "loss": 1.8567,
      "theoretical_loss": 4.43508086614867,
      "tokens_seen": 182255616
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009543029087261785,
      "loss": 2.0057,
      "theoretical_loss": 4.4348797758319325,
      "tokens_seen": 182321152
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000954282848545637,
      "loss": 1.8366,
      "theoretical_loss": 4.434678778015701,
      "tokens_seen": 182386688
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009542627883650953,
      "loss": 1.8694,
      "theoretical_loss": 4.434477872624212,
      "tokens_seen": 182452224
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009542427281845537,
      "loss": 2.0037,
      "theoretical_loss": 4.434277059581789,
      "tokens_seen": 182517760
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000954222668004012,
      "loss": 1.925,
      "theoretical_loss": 4.434076338812848,
      "tokens_seen": 182583296
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009542026078234705,
      "loss": 1.9509,
      "theoretical_loss": 4.43387571024189,
      "tokens_seen": 182648832
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009541825476429288,
      "loss": 1.9147,
      "theoretical_loss": 4.433675173793507,
      "tokens_seen": 182714368
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009541624874623872,
      "loss": 2.0302,
      "theoretical_loss": 4.433474729392379,
      "tokens_seen": 182779904
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009541424272818455,
      "loss": 2.053,
      "theoretical_loss": 4.433274376963276,
      "tokens_seen": 182845440
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000954122367101304,
      "loss": 1.8329,
      "theoretical_loss": 4.4330741164310545,
      "tokens_seen": 182910976
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009541023069207623,
      "loss": 1.7745,
      "theoretical_loss": 4.43287394772066,
      "tokens_seen": 182976512
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009540822467402207,
      "loss": 1.9235,
      "theoretical_loss": 4.432673870757127,
      "tokens_seen": 183042048
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000954062186559679,
      "loss": 1.956,
      "theoretical_loss": 4.432473885465576,
      "tokens_seen": 183107584
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009540421263791374,
      "loss": 2.0946,
      "theoretical_loss": 4.432273991771218,
      "tokens_seen": 183173120
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009540220661985958,
      "loss": 1.9536,
      "theoretical_loss": 4.43207418959935,
      "tokens_seen": 183238656
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009540020060180542,
      "loss": 1.8331,
      "theoretical_loss": 4.431874478875356,
      "tokens_seen": 183304192
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009539819458375125,
      "loss": 1.8043,
      "theoretical_loss": 4.431674859524709,
      "tokens_seen": 183369728
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000953961885656971,
      "loss": 1.8608,
      "theoretical_loss": 4.43147533147297,
      "tokens_seen": 183435264
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.4887767434120178,
      "objective/train/docs_used": 73723,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8926588296890259,
      "objective/train/original_loss": 1.8926584720611572,
      "objective/train/theoretical_loss": 4.431275894645784,
      "objective/train/tokens_used": 203960800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24062563478946686,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0501008033752441,
      "objective/train/weighted_lm_loss": 1.9868710041046143,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.431275894645784,
      "tokens_seen": 183500800
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009539418254764293,
      "loss": 1.91,
      "theoretical_loss": 4.431275894645784,
      "tokens_seen": 183500800
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009539217652958877,
      "loss": 1.9714,
      "theoretical_loss": 4.431076548968887,
      "tokens_seen": 183566336
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000953901705115346,
      "loss": 1.9316,
      "theoretical_loss": 4.430877294368099,
      "tokens_seen": 183631872
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009538816449348044,
      "loss": 1.8594,
      "theoretical_loss": 4.430678130769329,
      "tokens_seen": 183697408
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009538615847542628,
      "loss": 1.8918,
      "theoretical_loss": 4.43047905809857,
      "tokens_seen": 183762944
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009538415245737212,
      "loss": 1.9531,
      "theoretical_loss": 4.430280076281904,
      "tokens_seen": 183828480
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009538214643931795,
      "loss": 1.7753,
      "theoretical_loss": 4.430081185245498,
      "tokens_seen": 183894016
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000953801404212638,
      "loss": 1.9356,
      "theoretical_loss": 4.429882384915607,
      "tokens_seen": 183959552
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009537813440320962,
      "loss": 1.9047,
      "theoretical_loss": 4.42968367521857,
      "tokens_seen": 184025088
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009537612838515547,
      "loss": 1.9452,
      "theoretical_loss": 4.4294850560808126,
      "tokens_seen": 184090624
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009537412236710131,
      "loss": 1.8756,
      "theoretical_loss": 4.429286527428848,
      "tokens_seen": 184156160
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009537211634904714,
      "loss": 1.9095,
      "theoretical_loss": 4.429088089189273,
      "tokens_seen": 184221696
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009537011033099298,
      "loss": 1.9531,
      "theoretical_loss": 4.428889741288771,
      "tokens_seen": 184287232
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009536810431293881,
      "loss": 2.0226,
      "theoretical_loss": 4.428691483654109,
      "tokens_seen": 184352768
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009536609829488465,
      "loss": 1.9104,
      "theoretical_loss": 4.428493316212144,
      "tokens_seen": 184418304
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000953640922768305,
      "loss": 1.9173,
      "theoretical_loss": 4.428295238889815,
      "tokens_seen": 184483840
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009536208625877632,
      "loss": 1.9288,
      "theoretical_loss": 4.428097251614145,
      "tokens_seen": 184549376
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009536008024072217,
      "loss": 1.8834,
      "theoretical_loss": 4.4278993543122445,
      "tokens_seen": 184614912
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009535807422266801,
      "loss": 2.0234,
      "theoretical_loss": 4.427701546911307,
      "tokens_seen": 184680448
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009535606820461384,
      "loss": 1.846,
      "theoretical_loss": 4.427503829338613,
      "tokens_seen": 184745984
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009535406218655968,
      "loss": 1.8591,
      "theoretical_loss": 4.427306201521524,
      "tokens_seen": 184811520
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009535205616850551,
      "loss": 1.8564,
      "theoretical_loss": 4.427108663387489,
      "tokens_seen": 184877056
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009535005015045135,
      "loss": 1.85,
      "theoretical_loss": 4.426911214864043,
      "tokens_seen": 184942592
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000953480441323972,
      "loss": 1.8551,
      "theoretical_loss": 4.426713855878798,
      "tokens_seen": 185008128
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009534603811434302,
      "loss": 2.0234,
      "theoretical_loss": 4.426516586359458,
      "tokens_seen": 185073664
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009534403209628887,
      "loss": 2.0515,
      "theoretical_loss": 4.426319406233807,
      "tokens_seen": 185139200
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009534202607823471,
      "loss": 1.9792,
      "theoretical_loss": 4.426122315429713,
      "tokens_seen": 185204736
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009534002006018054,
      "loss": 1.8541,
      "theoretical_loss": 4.425925313875129,
      "tokens_seen": 185270272
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009533801404212639,
      "loss": 1.8656,
      "theoretical_loss": 4.425728401498089,
      "tokens_seen": 185335808
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009533600802407222,
      "loss": 2.041,
      "theoretical_loss": 4.425531578226715,
      "tokens_seen": 185401344
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009533400200601806,
      "loss": 1.9088,
      "theoretical_loss": 4.425334843989207,
      "tokens_seen": 185466880
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009533199598796389,
      "loss": 2.0045,
      "theoretical_loss": 4.425138198713853,
      "tokens_seen": 185532416
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009532998996990973,
      "loss": 1.8641,
      "theoretical_loss": 4.424941642329019,
      "tokens_seen": 185597952
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009532798395185557,
      "loss": 2.0065,
      "theoretical_loss": 4.42474517476316,
      "tokens_seen": 185663488
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009532597793380141,
      "loss": 1.8994,
      "theoretical_loss": 4.424548795944808,
      "tokens_seen": 185729024
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009532397191574724,
      "loss": 1.8996,
      "theoretical_loss": 4.424352505802582,
      "tokens_seen": 185794560
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009532196589769309,
      "loss": 1.898,
      "theoretical_loss": 4.42415630426518,
      "tokens_seen": 185860096
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009531995987963892,
      "loss": 1.9314,
      "theoretical_loss": 4.423960191261387,
      "tokens_seen": 185925632
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009531795386158476,
      "loss": 1.8369,
      "theoretical_loss": 4.423764166720066,
      "tokens_seen": 185991168
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009531594784353059,
      "loss": 1.8829,
      "theoretical_loss": 4.4235682305701625,
      "tokens_seen": 186056704
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009531394182547644,
      "loss": 1.8661,
      "theoretical_loss": 4.423372382740707,
      "tokens_seen": 186122240
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009531193580742227,
      "loss": 1.8,
      "theoretical_loss": 4.423176623160811,
      "tokens_seen": 186187776
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009530992978936811,
      "loss": 1.8912,
      "theoretical_loss": 4.422980951759665,
      "tokens_seen": 186253312
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009530792377131394,
      "loss": 1.8467,
      "theoretical_loss": 4.4227853684665455,
      "tokens_seen": 186318848
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009530591775325978,
      "loss": 1.8339,
      "theoretical_loss": 4.422589873210806,
      "tokens_seen": 186384384
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009530391173520562,
      "loss": 1.8041,
      "theoretical_loss": 4.422394465921887,
      "tokens_seen": 186449920
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009530190571715146,
      "loss": 1.8145,
      "theoretical_loss": 4.422199146529305,
      "tokens_seen": 186515456
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009529989969909729,
      "loss": 1.8298,
      "theoretical_loss": 4.42200391496266,
      "tokens_seen": 186580992
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009529789368104314,
      "loss": 1.9237,
      "theoretical_loss": 4.4218087711516345,
      "tokens_seen": 186646528
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009529588766298897,
      "loss": 1.9574,
      "theoretical_loss": 4.421613715025988,
      "tokens_seen": 186712064
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.49304935336112976,
      "objective/train/docs_used": 74904,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7934205532073975,
      "objective/train/original_loss": 1.7934205532073975,
      "objective/train/theoretical_loss": 4.421418746515567,
      "objective/train/tokens_used": 207237600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24341203272342682,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505423545837402,
      "objective/train/weighted_lm_loss": 1.8839329481124878,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.421418746515567,
      "tokens_seen": 186777600
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009529388164493481,
      "loss": 1.8573,
      "theoretical_loss": 4.421418746515567,
      "tokens_seen": 186777600
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009529187562688064,
      "loss": 2.0024,
      "theoretical_loss": 4.421223865550291,
      "tokens_seen": 186843136
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009528986960882648,
      "loss": 1.9979,
      "theoretical_loss": 4.421029072060167,
      "tokens_seen": 186908672
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009528786359077232,
      "loss": 2.0378,
      "theoretical_loss": 4.420834365975278,
      "tokens_seen": 186974208
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009528585757271816,
      "loss": 1.8248,
      "theoretical_loss": 4.42063974722579,
      "tokens_seen": 187039744
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009528385155466399,
      "loss": 1.8471,
      "theoretical_loss": 4.420445215741949,
      "tokens_seen": 187105280
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009528184553660984,
      "loss": 1.8749,
      "theoretical_loss": 4.420250771454078,
      "tokens_seen": 187170816
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009527983951855566,
      "loss": 1.9967,
      "theoretical_loss": 4.420056414292584,
      "tokens_seen": 187236352
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009527783350050151,
      "loss": 1.9333,
      "theoretical_loss": 4.419862144187952,
      "tokens_seen": 187301888
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009527582748244735,
      "loss": 1.93,
      "theoretical_loss": 4.419667961070747,
      "tokens_seen": 187367424
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009527382146439318,
      "loss": 2.0093,
      "theoretical_loss": 4.419473864871613,
      "tokens_seen": 187432960
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009527181544633902,
      "loss": 1.9498,
      "theoretical_loss": 4.419279855521276,
      "tokens_seen": 187498496
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009526980942828485,
      "loss": 2.0281,
      "theoretical_loss": 4.419085932950537,
      "tokens_seen": 187564032
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009526780341023069,
      "loss": 1.9989,
      "theoretical_loss": 4.41889209709028,
      "tokens_seen": 187629568
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009526579739217654,
      "loss": 1.8987,
      "theoretical_loss": 4.4186983478714685,
      "tokens_seen": 187695104
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009526379137412236,
      "loss": 1.9608,
      "theoretical_loss": 4.418504685225142,
      "tokens_seen": 187760640
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009526178535606821,
      "loss": 1.9515,
      "theoretical_loss": 4.4183111090824205,
      "tokens_seen": 187826176
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009525977933801405,
      "loss": 2.0248,
      "theoretical_loss": 4.418117619374502,
      "tokens_seen": 187891712
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009525777331995988,
      "loss": 1.9202,
      "theoretical_loss": 4.417924216032667,
      "tokens_seen": 187957248
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009525576730190572,
      "loss": 1.9305,
      "theoretical_loss": 4.417730898988268,
      "tokens_seen": 188022784
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009525376128385155,
      "loss": 1.9592,
      "theoretical_loss": 4.417537668172743,
      "tokens_seen": 188088320
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009525175526579739,
      "loss": 1.8921,
      "theoretical_loss": 4.417344523517602,
      "tokens_seen": 188153856
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009524974924774324,
      "loss": 1.9449,
      "theoretical_loss": 4.417151464954437,
      "tokens_seen": 188219392
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009524774322968906,
      "loss": 1.9775,
      "theoretical_loss": 4.416958492414918,
      "tokens_seen": 188284928
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009524573721163491,
      "loss": 1.9599,
      "theoretical_loss": 4.416765605830792,
      "tokens_seen": 188350464
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009524373119358074,
      "loss": 1.8162,
      "theoretical_loss": 4.4165728051338835,
      "tokens_seen": 188416000
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009524172517552658,
      "loss": 1.9151,
      "theoretical_loss": 4.416380090256095,
      "tokens_seen": 188481536
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009523971915747242,
      "loss": 1.8405,
      "theoretical_loss": 4.416187461129409,
      "tokens_seen": 188547072
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009523771313941826,
      "loss": 2.0353,
      "theoretical_loss": 4.415994917685881,
      "tokens_seen": 188612608
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009523570712136409,
      "loss": 2.0348,
      "theoretical_loss": 4.415802459857648,
      "tokens_seen": 188678144
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009523370110330993,
      "loss": 2.0916,
      "theoretical_loss": 4.415610087576923,
      "tokens_seen": 188743680
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009523169508525576,
      "loss": 1.8956,
      "theoretical_loss": 4.4154178007759946,
      "tokens_seen": 188809216
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009522968906720161,
      "loss": 2.0073,
      "theoretical_loss": 4.415225599387231,
      "tokens_seen": 188874752
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009522768304914744,
      "loss": 1.916,
      "theoretical_loss": 4.415033483343075,
      "tokens_seen": 188940288
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009522567703109328,
      "loss": 1.9299,
      "theoretical_loss": 4.414841452576049,
      "tokens_seen": 189005824
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009522367101303912,
      "loss": 1.9453,
      "theoretical_loss": 4.4146495070187495,
      "tokens_seen": 189071360
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009522166499498496,
      "loss": 1.961,
      "theoretical_loss": 4.41445764660385,
      "tokens_seen": 189136896
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009521965897693079,
      "loss": 1.7822,
      "theoretical_loss": 4.414265871264102,
      "tokens_seen": 189202432
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009521765295887663,
      "loss": 1.8333,
      "theoretical_loss": 4.414074180932333,
      "tokens_seen": 189267968
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009521564694082246,
      "loss": 1.9526,
      "theoretical_loss": 4.413882575541446,
      "tokens_seen": 189333504
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009521364092276831,
      "loss": 1.8715,
      "theoretical_loss": 4.41369105502442,
      "tokens_seen": 189399040
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009521163490471414,
      "loss": 1.8555,
      "theoretical_loss": 4.41349961931431,
      "tokens_seen": 189464576
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009520962888665998,
      "loss": 1.8327,
      "theoretical_loss": 4.413308268344249,
      "tokens_seen": 189530112
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009520762286860581,
      "loss": 1.989,
      "theoretical_loss": 4.413117002047443,
      "tokens_seen": 189595648
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009520561685055166,
      "loss": 1.9247,
      "theoretical_loss": 4.4129258203571755,
      "tokens_seen": 189661184
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009520361083249749,
      "loss": 1.8879,
      "theoretical_loss": 4.412734723206805,
      "tokens_seen": 189726720
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009520160481444333,
      "loss": 1.9158,
      "theoretical_loss": 4.412543710529766,
      "tokens_seen": 189792256
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009519959879638917,
      "loss": 1.9396,
      "theoretical_loss": 4.412352782259568,
      "tokens_seen": 189857792
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009519759277833501,
      "loss": 1.7889,
      "theoretical_loss": 4.4121619383297945,
      "tokens_seen": 189923328
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009519558676028084,
      "loss": 2.0035,
      "theoretical_loss": 4.411971178674106,
      "tokens_seen": 189988864
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.4934732913970947,
      "objective/train/docs_used": 76077,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8127355575561523,
      "objective/train/original_loss": 1.8127355575561523,
      "objective/train/theoretical_loss": 4.411780503226238,
      "objective/train/tokens_used": 210514400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24385586380958557,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505869388580322,
      "objective/train/weighted_lm_loss": 1.9044004678726196,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.411780503226238,
      "tokens_seen": 190054400
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009519358074222668,
      "loss": 1.9662,
      "theoretical_loss": 4.411780503226238,
      "tokens_seen": 190054400
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009519157472417252,
      "loss": 1.9113,
      "theoretical_loss": 4.411589911919999,
      "tokens_seen": 190119936
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009518956870611836,
      "loss": 1.9649,
      "theoretical_loss": 4.411399404689275,
      "tokens_seen": 190185472
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951875626880642,
      "loss": 2.0698,
      "theoretical_loss": 4.411208981468025,
      "tokens_seen": 190251008
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009518555667001003,
      "loss": 1.8534,
      "theoretical_loss": 4.4110186421902835,
      "tokens_seen": 190316544
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009518355065195588,
      "loss": 1.7358,
      "theoretical_loss": 4.410828386790158,
      "tokens_seen": 190382080
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951815446339017,
      "loss": 1.7901,
      "theoretical_loss": 4.410638215201831,
      "tokens_seen": 190447616
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009517953861584755,
      "loss": 1.9016,
      "theoretical_loss": 4.410448127359561,
      "tokens_seen": 190513152
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009517753259779339,
      "loss": 2.0252,
      "theoretical_loss": 4.4102581231976785,
      "tokens_seen": 190578688
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009517552657973922,
      "loss": 1.9728,
      "theoretical_loss": 4.410068202650588,
      "tokens_seen": 190644224
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009517352056168506,
      "loss": 1.9752,
      "theoretical_loss": 4.40987836565277,
      "tokens_seen": 190709760
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009517151454363089,
      "loss": 1.8996,
      "theoretical_loss": 4.409688612138778,
      "tokens_seen": 190775296
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009516950852557673,
      "loss": 1.9769,
      "theoretical_loss": 4.409498942043237,
      "tokens_seen": 190840832
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009516750250752258,
      "loss": 2.0145,
      "theoretical_loss": 4.4093093553008496,
      "tokens_seen": 190906368
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951654964894684,
      "loss": 1.9347,
      "theoretical_loss": 4.409119851846388,
      "tokens_seen": 190971904
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009516349047141425,
      "loss": 1.8087,
      "theoretical_loss": 4.4089304316147,
      "tokens_seen": 191037440
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009516148445336009,
      "loss": 1.9947,
      "theoretical_loss": 4.408741094540707,
      "tokens_seen": 191102976
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009515947843530592,
      "loss": 1.8363,
      "theoretical_loss": 4.408551840559403,
      "tokens_seen": 191168512
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009515747241725176,
      "loss": 1.8745,
      "theoretical_loss": 4.408362669605854,
      "tokens_seen": 191234048
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951554663991976,
      "loss": 1.9859,
      "theoretical_loss": 4.408173581615199,
      "tokens_seen": 191299584
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009515346038114343,
      "loss": 1.8372,
      "theoretical_loss": 4.407984576522653,
      "tokens_seen": 191365120
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009515145436308928,
      "loss": 2.0925,
      "theoretical_loss": 4.4077956542635,
      "tokens_seen": 191430656
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951494483450351,
      "loss": 1.9144,
      "theoretical_loss": 4.407606814773099,
      "tokens_seen": 191496192
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009514744232698095,
      "loss": 1.7939,
      "theoretical_loss": 4.407418057986881,
      "tokens_seen": 191561728
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009514543630892678,
      "loss": 1.9196,
      "theoretical_loss": 4.407229383840347,
      "tokens_seen": 191627264
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009514343029087262,
      "loss": 1.9596,
      "theoretical_loss": 4.407040792269075,
      "tokens_seen": 191692800
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009514142427281846,
      "loss": 1.9563,
      "theoretical_loss": 4.406852283208711,
      "tokens_seen": 191758336
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951394182547643,
      "loss": 1.8223,
      "theoretical_loss": 4.406663856594977,
      "tokens_seen": 191823872
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009513741223671013,
      "loss": 1.8939,
      "theoretical_loss": 4.406475512363663,
      "tokens_seen": 191889408
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009513540621865597,
      "loss": 1.8934,
      "theoretical_loss": 4.4062872504506325,
      "tokens_seen": 191954944
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951334002006018,
      "loss": 1.7905,
      "theoretical_loss": 4.406099070791822,
      "tokens_seen": 192020480
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009513139418254765,
      "loss": 1.768,
      "theoretical_loss": 4.405910973323239,
      "tokens_seen": 192086016
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009512938816449348,
      "loss": 1.8877,
      "theoretical_loss": 4.405722957980962,
      "tokens_seen": 192151552
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009512738214643932,
      "loss": 1.9167,
      "theoretical_loss": 4.405535024701141,
      "tokens_seen": 192217088
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009512537612838516,
      "loss": 1.9675,
      "theoretical_loss": 4.405347173419998,
      "tokens_seen": 192282624
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.00095123370110331,
      "loss": 1.9097,
      "theoretical_loss": 4.405159404073826,
      "tokens_seen": 192348160
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009512136409227683,
      "loss": 1.8425,
      "theoretical_loss": 4.40497171659899,
      "tokens_seen": 192413696
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009511935807422267,
      "loss": 2.008,
      "theoretical_loss": 4.404784110931924,
      "tokens_seen": 192479232
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951173520561685,
      "loss": 1.7656,
      "theoretical_loss": 4.404596587009136,
      "tokens_seen": 192544768
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009511534603811435,
      "loss": 1.7931,
      "theoretical_loss": 4.404409144767201,
      "tokens_seen": 192610304
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009511334002006018,
      "loss": 1.8794,
      "theoretical_loss": 4.404221784142768,
      "tokens_seen": 192675840
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009511133400200602,
      "loss": 1.8634,
      "theoretical_loss": 4.404034505072556,
      "tokens_seen": 192741376
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009510932798395185,
      "loss": 1.8689,
      "theoretical_loss": 4.403847307493353,
      "tokens_seen": 192806912
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000951073219658977,
      "loss": 1.9959,
      "theoretical_loss": 4.40366019134202,
      "tokens_seen": 192872448
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009510531594784353,
      "loss": 1.9952,
      "theoretical_loss": 4.403473156555487,
      "tokens_seen": 192937984
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009510330992978937,
      "loss": 1.9028,
      "theoretical_loss": 4.403286203070751,
      "tokens_seen": 193003520
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009510130391173521,
      "loss": 1.7787,
      "theoretical_loss": 4.403099330824887,
      "tokens_seen": 193069056
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009509929789368105,
      "loss": 1.9463,
      "theoretical_loss": 4.4029125397550315,
      "tokens_seen": 193134592
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009509729187562688,
      "loss": 1.834,
      "theoretical_loss": 4.402725829798397,
      "tokens_seen": 193200128
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009509528585757272,
      "loss": 1.9024,
      "theoretical_loss": 4.402539200892264,
      "tokens_seen": 193265664
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.49331486225128174,
      "objective/train/docs_used": 77266,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8808577060699463,
      "objective/train/original_loss": 1.8808577060699463,
      "objective/train/theoretical_loss": 4.40235265297398,
      "objective/train/tokens_used": 213791200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24369393289089203,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505704879760742,
      "objective/train/weighted_lm_loss": 1.9757757186889648,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.40235265297398,
      "tokens_seen": 193331200
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009509327983951855,
      "loss": 1.8993,
      "theoretical_loss": 4.40235265297398,
      "tokens_seen": 193331200
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000950912738214644,
      "loss": 1.8289,
      "theoretical_loss": 4.402166185980967,
      "tokens_seen": 193396736
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009508926780341023,
      "loss": 1.8115,
      "theoretical_loss": 4.4019797998507135,
      "tokens_seen": 193462272
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009508726178535607,
      "loss": 1.8387,
      "theoretical_loss": 4.401793494520778,
      "tokens_seen": 193527808
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009508525576730191,
      "loss": 1.7944,
      "theoretical_loss": 4.401607269928788,
      "tokens_seen": 193593344
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009508324974924774,
      "loss": 1.7242,
      "theoretical_loss": 4.4014211260124405,
      "tokens_seen": 193658880
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009508124373119358,
      "loss": 1.8879,
      "theoretical_loss": 4.401235062709502,
      "tokens_seen": 193724416
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009507923771313943,
      "loss": 1.8485,
      "theoretical_loss": 4.401049079957809,
      "tokens_seen": 193789952
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009507723169508525,
      "loss": 1.8177,
      "theoretical_loss": 4.400863177695264,
      "tokens_seen": 193855488
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000950752256770311,
      "loss": 1.8554,
      "theoretical_loss": 4.400677355859841,
      "tokens_seen": 193921024
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009507321965897692,
      "loss": 2.0593,
      "theoretical_loss": 4.400491614389582,
      "tokens_seen": 193986560
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009507121364092277,
      "loss": 1.8376,
      "theoretical_loss": 4.400305953222596,
      "tokens_seen": 194052096
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009506920762286861,
      "loss": 1.9119,
      "theoretical_loss": 4.4001203722970645,
      "tokens_seen": 194117632
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009506720160481444,
      "loss": 1.8258,
      "theoretical_loss": 4.399934871551234,
      "tokens_seen": 194183168
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009506519558676028,
      "loss": 1.6996,
      "theoretical_loss": 4.3997494509234185,
      "tokens_seen": 194248704
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009506318956870613,
      "loss": 1.8192,
      "theoretical_loss": 4.399564110352005,
      "tokens_seen": 194314240
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009506118355065195,
      "loss": 1.9074,
      "theoretical_loss": 4.399378849775443,
      "tokens_seen": 194379776
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000950591775325978,
      "loss": 1.8622,
      "theoretical_loss": 4.399193669132255,
      "tokens_seen": 194445312
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009505717151454362,
      "loss": 1.9398,
      "theoretical_loss": 4.399008568361027,
      "tokens_seen": 194510848
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009505516549648947,
      "loss": 1.8967,
      "theoretical_loss": 4.398823547400417,
      "tokens_seen": 194576384
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009505315947843531,
      "loss": 1.9654,
      "theoretical_loss": 4.398638606189149,
      "tokens_seen": 194641920
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009505115346038114,
      "loss": 1.8578,
      "theoretical_loss": 4.398453744666012,
      "tokens_seen": 194707456
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009504914744232698,
      "loss": 1.9715,
      "theoretical_loss": 4.398268962769867,
      "tokens_seen": 194772992
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009504714142427282,
      "loss": 2.0093,
      "theoretical_loss": 4.39808426043964,
      "tokens_seen": 194838528
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009504513540621865,
      "loss": 1.8911,
      "theoretical_loss": 4.397899637614326,
      "tokens_seen": 194904064
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000950431293881645,
      "loss": 1.9758,
      "theoretical_loss": 4.397715094232984,
      "tokens_seen": 194969600
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009504112337011034,
      "loss": 1.9613,
      "theoretical_loss": 4.397530630234744,
      "tokens_seen": 195035136
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009503911735205617,
      "loss": 1.9392,
      "theoretical_loss": 4.3973462455588,
      "tokens_seen": 195100672
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009503711133400201,
      "loss": 1.8308,
      "theoretical_loss": 4.397161940144415,
      "tokens_seen": 195166208
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009503510531594784,
      "loss": 1.7817,
      "theoretical_loss": 4.396977713930919,
      "tokens_seen": 195231744
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009503309929789369,
      "loss": 1.9156,
      "theoretical_loss": 4.396793566857708,
      "tokens_seen": 195297280
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009503109327983952,
      "loss": 1.8466,
      "theoretical_loss": 4.396609498864243,
      "tokens_seen": 195362816
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009502908726178536,
      "loss": 1.8075,
      "theoretical_loss": 4.396425509890054,
      "tokens_seen": 195428352
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000950270812437312,
      "loss": 1.8059,
      "theoretical_loss": 4.396241599874738,
      "tokens_seen": 195493888
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009502507522567704,
      "loss": 1.8906,
      "theoretical_loss": 4.396057768757957,
      "tokens_seen": 195559424
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009502306920762287,
      "loss": 1.8618,
      "theoretical_loss": 4.395874016479438,
      "tokens_seen": 195624960
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009502106318956871,
      "loss": 1.8689,
      "theoretical_loss": 4.395690342978977,
      "tokens_seen": 195690496
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009501905717151454,
      "loss": 1.7489,
      "theoretical_loss": 4.395506748196435,
      "tokens_seen": 195756032
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009501705115346039,
      "loss": 2.0602,
      "theoretical_loss": 4.395323232071737,
      "tokens_seen": 195821568
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009501504513540622,
      "loss": 1.8469,
      "theoretical_loss": 4.395139794544878,
      "tokens_seen": 195887104
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009501303911735206,
      "loss": 1.9692,
      "theoretical_loss": 4.394956435555916,
      "tokens_seen": 195952640
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009501103309929789,
      "loss": 1.9052,
      "theoretical_loss": 4.394773155044976,
      "tokens_seen": 196018176
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009500902708124374,
      "loss": 1.8058,
      "theoretical_loss": 4.394589952952247,
      "tokens_seen": 196083712
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009500702106318957,
      "loss": 1.787,
      "theoretical_loss": 4.394406829217983,
      "tokens_seen": 196149248
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009500501504513541,
      "loss": 1.8961,
      "theoretical_loss": 4.394223783782509,
      "tokens_seen": 196214784
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009500300902708125,
      "loss": 1.7633,
      "theoretical_loss": 4.394040816586209,
      "tokens_seen": 196280320
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009500100300902709,
      "loss": 1.7421,
      "theoretical_loss": 4.393857927569534,
      "tokens_seen": 196345856
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009499899699097292,
      "loss": 1.9565,
      "theoretical_loss": 4.393675116673003,
      "tokens_seen": 196411392
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009499699097291876,
      "loss": 1.8156,
      "theoretical_loss": 4.393492383837196,
      "tokens_seen": 196476928
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009499498495486459,
      "loss": 1.8552,
      "theoretical_loss": 4.393309729002761,
      "tokens_seen": 196542464
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.4933859407901764,
      "objective/train/docs_used": 78313,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5727931261062622,
      "objective/train/original_loss": 1.5727932453155518,
      "objective/train/theoretical_loss": 4.393127152110409,
      "objective/train/tokens_used": 217068000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24412523210048676,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.05057954788208,
      "objective/train/weighted_lm_loss": 1.652202844619751,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.393127152110409,
      "tokens_seen": 196608000
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009499297893681044,
      "loss": 1.7561,
      "theoretical_loss": 4.393127152110409,
      "tokens_seen": 196608000
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009499097291875627,
      "loss": 1.855,
      "theoretical_loss": 4.392944653100918,
      "tokens_seen": 196673536
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009498896690070211,
      "loss": 1.9183,
      "theoretical_loss": 4.392762231915128,
      "tokens_seen": 196739072
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009498696088264795,
      "loss": 1.8333,
      "theoretical_loss": 4.392579888493946,
      "tokens_seen": 196804608
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009498495486459378,
      "loss": 1.8654,
      "theoretical_loss": 4.392397622778343,
      "tokens_seen": 196870144
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009498294884653962,
      "loss": 1.8546,
      "theoretical_loss": 4.392215434709351,
      "tokens_seen": 196935680
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009498094282848547,
      "loss": 1.9817,
      "theoretical_loss": 4.392033324228073,
      "tokens_seen": 197001216
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009497893681043129,
      "loss": 1.8226,
      "theoretical_loss": 4.391851291275671,
      "tokens_seen": 197066752
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009497693079237714,
      "loss": 1.8504,
      "theoretical_loss": 4.391669335793372,
      "tokens_seen": 197132288
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009497492477432296,
      "loss": 1.7561,
      "theoretical_loss": 4.391487457722469,
      "tokens_seen": 197197824
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009497291875626881,
      "loss": 1.671,
      "theoretical_loss": 4.391305657004317,
      "tokens_seen": 197263360
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009497091273821465,
      "loss": 1.9367,
      "theoretical_loss": 4.391123933580336,
      "tokens_seen": 197328896
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009496890672016048,
      "loss": 1.9203,
      "theoretical_loss": 4.39094228739201,
      "tokens_seen": 197394432
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009496690070210632,
      "loss": 1.8818,
      "theoretical_loss": 4.390760718380884,
      "tokens_seen": 197459968
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009496489468405217,
      "loss": 1.8564,
      "theoretical_loss": 4.390579226488571,
      "tokens_seen": 197525504
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009496288866599799,
      "loss": 1.9317,
      "theoretical_loss": 4.3903978116567455,
      "tokens_seen": 197591040
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009496088264794384,
      "loss": 1.9909,
      "theoretical_loss": 4.390216473827143,
      "tokens_seen": 197656576
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009495887662988966,
      "loss": 1.7772,
      "theoretical_loss": 4.390035212941568,
      "tokens_seen": 197722112
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009495687061183551,
      "loss": 1.7826,
      "theoretical_loss": 4.3898540289418815,
      "tokens_seen": 197787648
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009495486459378135,
      "loss": 1.8518,
      "theoretical_loss": 4.389672921770013,
      "tokens_seen": 197853184
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009495285857572718,
      "loss": 1.7901,
      "theoretical_loss": 4.389491891367953,
      "tokens_seen": 197918720
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009495085255767302,
      "loss": 1.8022,
      "theoretical_loss": 4.389310937677754,
      "tokens_seen": 197984256
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009494884653961886,
      "loss": 1.9155,
      "theoretical_loss": 4.389130060641534,
      "tokens_seen": 198049792
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009494684052156469,
      "loss": 1.8942,
      "theoretical_loss": 4.388949260201471,
      "tokens_seen": 198115328
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009494483450351054,
      "loss": 1.8778,
      "theoretical_loss": 4.388768536299808,
      "tokens_seen": 198180864
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009494282848545636,
      "loss": 1.9305,
      "theoretical_loss": 4.388587888878849,
      "tokens_seen": 198246400
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009494082246740221,
      "loss": 1.9257,
      "theoretical_loss": 4.388407317880962,
      "tokens_seen": 198311936
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009493881644934804,
      "loss": 1.9663,
      "theoretical_loss": 4.388226823248576,
      "tokens_seen": 198377472
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009493681043129388,
      "loss": 1.9069,
      "theoretical_loss": 4.388046404924184,
      "tokens_seen": 198443008
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009493480441323972,
      "loss": 1.8286,
      "theoretical_loss": 4.387866062850339,
      "tokens_seen": 198508544
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009493279839518556,
      "loss": 1.915,
      "theoretical_loss": 4.3876857969696585,
      "tokens_seen": 198574080
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009493079237713139,
      "loss": 1.8648,
      "theoretical_loss": 4.387505607224821,
      "tokens_seen": 198639616
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009492878635907724,
      "loss": 1.8246,
      "theoretical_loss": 4.387325493558566,
      "tokens_seen": 198705152
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009492678034102307,
      "loss": 1.8195,
      "theoretical_loss": 4.387145455913698,
      "tokens_seen": 198770688
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009492477432296891,
      "loss": 1.7988,
      "theoretical_loss": 4.38696549423308,
      "tokens_seen": 198836224
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009492276830491474,
      "loss": 1.9393,
      "theoretical_loss": 4.386785608459639,
      "tokens_seen": 198901760
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009492076228686058,
      "loss": 1.9559,
      "theoretical_loss": 4.386605798536362,
      "tokens_seen": 198967296
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009491875626880642,
      "loss": 1.945,
      "theoretical_loss": 4.3864260644063,
      "tokens_seen": 199032832
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009491675025075226,
      "loss": 1.8763,
      "theoretical_loss": 4.386246406012562,
      "tokens_seen": 199098368
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009491474423269809,
      "loss": 1.9405,
      "theoretical_loss": 4.3860668232983215,
      "tokens_seen": 199163904
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009491273821464393,
      "loss": 1.962,
      "theoretical_loss": 4.385887316206812,
      "tokens_seen": 199229440
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009491073219658977,
      "loss": 1.8279,
      "theoretical_loss": 4.385707884681327,
      "tokens_seen": 199294976
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009490872617853561,
      "loss": 1.7523,
      "theoretical_loss": 4.385528528665225,
      "tokens_seen": 199360512
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009490672016048144,
      "loss": 1.8032,
      "theoretical_loss": 4.385349248101922,
      "tokens_seen": 199426048
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009490471414242729,
      "loss": 1.8202,
      "theoretical_loss": 4.385170042934896,
      "tokens_seen": 199491584
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009490270812437311,
      "loss": 1.8708,
      "theoretical_loss": 4.384990913107686,
      "tokens_seen": 199557120
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009490070210631896,
      "loss": 1.8222,
      "theoretical_loss": 4.384811858563891,
      "tokens_seen": 199622656
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009489869608826479,
      "loss": 1.7764,
      "theoretical_loss": 4.384632879247173,
      "tokens_seen": 199688192
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009489669007021063,
      "loss": 1.865,
      "theoretical_loss": 4.384453975101251,
      "tokens_seen": 199753728
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009489468405215648,
      "loss": 1.8772,
      "theoretical_loss": 4.384275146069909,
      "tokens_seen": 199819264
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.4942682981491089,
      "objective/train/docs_used": 79488,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6611281633377075,
      "objective/train/original_loss": 1.661128282546997,
      "objective/train/theoretical_loss": 4.384096392096988,
      "objective/train/tokens_used": 220344800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24450337886810303,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.05066978931427,
      "objective/train/weighted_lm_loss": 1.7454839944839478,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.384096392096988,
      "tokens_seen": 199884800
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009489267803410231,
      "loss": 1.8157,
      "theoretical_loss": 4.384096392096988,
      "tokens_seen": 199884800
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009489067201604815,
      "loss": 1.9058,
      "theoretical_loss": 4.38391771312639,
      "tokens_seen": 199950336
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009488866599799399,
      "loss": 1.9625,
      "theoretical_loss": 4.38373910910208,
      "tokens_seen": 200015872
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009488665997993982,
      "loss": 1.8666,
      "theoretical_loss": 4.383560579968078,
      "tokens_seen": 200081408
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009488465396188566,
      "loss": 1.7205,
      "theoretical_loss": 4.3833821256684695,
      "tokens_seen": 200146944
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009488264794383151,
      "loss": 1.8827,
      "theoretical_loss": 4.383203746147397,
      "tokens_seen": 200212480
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009488064192577733,
      "loss": 1.9402,
      "theoretical_loss": 4.383025441349063,
      "tokens_seen": 200278016
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009487863590772318,
      "loss": 1.8922,
      "theoretical_loss": 4.382847211217731,
      "tokens_seen": 200343552
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.00094876629889669,
      "loss": 1.9755,
      "theoretical_loss": 4.3826690556977255,
      "tokens_seen": 200409088
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009487462387161485,
      "loss": 1.9237,
      "theoretical_loss": 4.382490974733425,
      "tokens_seen": 200474624
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009487261785356069,
      "loss": 2.0019,
      "theoretical_loss": 4.382312968269276,
      "tokens_seen": 200540160
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009487061183550652,
      "loss": 1.8139,
      "theoretical_loss": 4.382135036249777,
      "tokens_seen": 200605696
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009486860581745236,
      "loss": 1.8938,
      "theoretical_loss": 4.3819571786194915,
      "tokens_seen": 200671232
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009486659979939821,
      "loss": 1.8056,
      "theoretical_loss": 4.381779395323038,
      "tokens_seen": 200736768
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009486459378134403,
      "loss": 1.8419,
      "theoretical_loss": 4.381601686305098,
      "tokens_seen": 200802304
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009486258776328988,
      "loss": 1.7723,
      "theoretical_loss": 4.381424051510409,
      "tokens_seen": 200867840
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000948605817452357,
      "loss": 1.8334,
      "theoretical_loss": 4.38124649088377,
      "tokens_seen": 200933376
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009485857572718155,
      "loss": 1.8382,
      "theoretical_loss": 4.381069004370039,
      "tokens_seen": 200998912
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009485656970912739,
      "loss": 1.8665,
      "theoretical_loss": 4.38089159191413,
      "tokens_seen": 201064448
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009485456369107322,
      "loss": 1.7653,
      "theoretical_loss": 4.38071425346102,
      "tokens_seen": 201129984
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009485255767301906,
      "loss": 1.8926,
      "theoretical_loss": 4.380536988955742,
      "tokens_seen": 201195520
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000948505516549649,
      "loss": 1.7422,
      "theoretical_loss": 4.380359798343388,
      "tokens_seen": 201261056
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009484854563691073,
      "loss": 1.707,
      "theoretical_loss": 4.380182681569111,
      "tokens_seen": 201326592
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009484653961885658,
      "loss": 1.8416,
      "theoretical_loss": 4.380005638578119,
      "tokens_seen": 201392128
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000948445336008024,
      "loss": 1.8569,
      "theoretical_loss": 4.37982866931568,
      "tokens_seen": 201457664
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009484252758274825,
      "loss": 1.82,
      "theoretical_loss": 4.379651773727122,
      "tokens_seen": 201523200
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009484052156469408,
      "loss": 1.978,
      "theoretical_loss": 4.379474951757829,
      "tokens_seen": 201588736
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009483851554663992,
      "loss": 1.7683,
      "theoretical_loss": 4.379298203353244,
      "tokens_seen": 201654272
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009483650952858576,
      "loss": 1.9227,
      "theoretical_loss": 4.379121528458869,
      "tokens_seen": 201719808
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000948345035105316,
      "loss": 1.8948,
      "theoretical_loss": 4.378944927020262,
      "tokens_seen": 201785344
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009483249749247743,
      "loss": 1.939,
      "theoretical_loss": 4.378768398983042,
      "tokens_seen": 201850880
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009483049147442328,
      "loss": 1.8403,
      "theoretical_loss": 4.378591944292882,
      "tokens_seen": 201916416
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000948284854563691,
      "loss": 1.8222,
      "theoretical_loss": 4.378415562895518,
      "tokens_seen": 201981952
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009482647943831495,
      "loss": 1.7433,
      "theoretical_loss": 4.378239254736738,
      "tokens_seen": 202047488
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009482447342026078,
      "loss": 1.9094,
      "theoretical_loss": 4.378063019762392,
      "tokens_seen": 202113024
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009482246740220662,
      "loss": 1.7928,
      "theoretical_loss": 4.377886857918385,
      "tokens_seen": 202178560
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009482046138415246,
      "loss": 1.8734,
      "theoretical_loss": 4.377710769150681,
      "tokens_seen": 202244096
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000948184553660983,
      "loss": 1.9099,
      "theoretical_loss": 4.3775347534053015,
      "tokens_seen": 202309632
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009481644934804413,
      "loss": 1.7542,
      "theoretical_loss": 4.377358810628324,
      "tokens_seen": 202375168
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009481444332998997,
      "loss": 1.8399,
      "theoretical_loss": 4.377182940765883,
      "tokens_seen": 202440704
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009481243731193581,
      "loss": 1.7524,
      "theoretical_loss": 4.377007143764175,
      "tokens_seen": 202506240
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009481043129388165,
      "loss": 1.8736,
      "theoretical_loss": 4.376831419569445,
      "tokens_seen": 202571776
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009480842527582748,
      "loss": 1.9028,
      "theoretical_loss": 4.3766557681280025,
      "tokens_seen": 202637312
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009480641925777333,
      "loss": 1.9198,
      "theoretical_loss": 4.3764801893862115,
      "tokens_seen": 202702848
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009480441323971915,
      "loss": 1.8034,
      "theoretical_loss": 4.376304683290491,
      "tokens_seen": 202768384
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.00094802407221665,
      "loss": 1.8115,
      "theoretical_loss": 4.37612924978732,
      "tokens_seen": 202833920
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009480040120361083,
      "loss": 1.8673,
      "theoretical_loss": 4.375953888823233,
      "tokens_seen": 202899456
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009479839518555667,
      "loss": 1.7649,
      "theoretical_loss": 4.375778600344819,
      "tokens_seen": 202964992
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009479638916750251,
      "loss": 1.7969,
      "theoretical_loss": 4.375603384298726,
      "tokens_seen": 203030528
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009479438314944835,
      "loss": 1.8562,
      "theoretical_loss": 4.375428240631659,
      "tokens_seen": 203096064
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.49045994877815247,
      "objective/train/docs_used": 80605,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8460792303085327,
      "objective/train/original_loss": 1.8460791110992432,
      "objective/train/theoretical_loss": 4.375253169290376,
      "objective/train/tokens_used": 223621600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24119678139686584,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050271987915039,
      "objective/train/weighted_lm_loss": 1.9390350580215454,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.375253169290376,
      "tokens_seen": 203161600
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009479237713139418,
      "loss": 1.9466,
      "theoretical_loss": 4.375253169290376,
      "tokens_seen": 203161600
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009479037111334003,
      "loss": 1.9524,
      "theoretical_loss": 4.3750781702216965,
      "tokens_seen": 203227136
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009478836509528585,
      "loss": 1.7591,
      "theoretical_loss": 4.37490324337249,
      "tokens_seen": 203292672
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000947863590772317,
      "loss": 1.8594,
      "theoretical_loss": 4.374728388689688,
      "tokens_seen": 203358208
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009478435305917753,
      "loss": 1.8902,
      "theoretical_loss": 4.374553606120274,
      "tokens_seen": 203423744
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009478234704112337,
      "loss": 1.8355,
      "theoretical_loss": 4.374378895611289,
      "tokens_seen": 203489280
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009478034102306921,
      "loss": 1.9065,
      "theoretical_loss": 4.374204257109832,
      "tokens_seen": 203554816
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009477833500501504,
      "loss": 1.7489,
      "theoretical_loss": 4.374029690563053,
      "tokens_seen": 203620352
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009477632898696088,
      "loss": 1.8441,
      "theoretical_loss": 4.373855195918162,
      "tokens_seen": 203685888
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009477432296890673,
      "loss": 1.9567,
      "theoretical_loss": 4.373680773122424,
      "tokens_seen": 203751424
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009477231695085255,
      "loss": 1.8593,
      "theoretical_loss": 4.3735064221231585,
      "tokens_seen": 203816960
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000947703109327984,
      "loss": 1.9252,
      "theoretical_loss": 4.373332142867739,
      "tokens_seen": 203882496
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009476830491474422,
      "loss": 1.8711,
      "theoretical_loss": 4.3731579353036,
      "tokens_seen": 203948032
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009476629889669007,
      "loss": 1.8338,
      "theoretical_loss": 4.372983799378225,
      "tokens_seen": 204013568
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009476429287863591,
      "loss": 1.8558,
      "theoretical_loss": 4.372809735039157,
      "tokens_seen": 204079104
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009476228686058174,
      "loss": 1.8181,
      "theoretical_loss": 4.372635742233991,
      "tokens_seen": 204144640
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009476028084252758,
      "loss": 1.7501,
      "theoretical_loss": 4.372461820910382,
      "tokens_seen": 204210176
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009475827482447343,
      "loss": 1.8748,
      "theoretical_loss": 4.372287971016036,
      "tokens_seen": 204275712
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009475626880641925,
      "loss": 1.7815,
      "theoretical_loss": 4.372114192498714,
      "tokens_seen": 204341248
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000947542627883651,
      "loss": 1.9619,
      "theoretical_loss": 4.371940485306234,
      "tokens_seen": 204406784
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009475225677031093,
      "loss": 1.8881,
      "theoretical_loss": 4.371766849386468,
      "tokens_seen": 204472320
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009475025075225677,
      "loss": 1.9036,
      "theoretical_loss": 4.371593284687343,
      "tokens_seen": 204537856
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009474824473420261,
      "loss": 1.7205,
      "theoretical_loss": 4.371419791156841,
      "tokens_seen": 204603392
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009474623871614844,
      "loss": 1.8684,
      "theoretical_loss": 4.371246368742996,
      "tokens_seen": 204668928
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009474423269809429,
      "loss": 2.0226,
      "theoretical_loss": 4.3710730173939005,
      "tokens_seen": 204734464
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009474222668004012,
      "loss": 1.7672,
      "theoretical_loss": 4.3708997370577,
      "tokens_seen": 204800000
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009474022066198596,
      "loss": 1.9649,
      "theoretical_loss": 4.370726527682593,
      "tokens_seen": 204865536
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000947382146439318,
      "loss": 1.9361,
      "theoretical_loss": 4.370553389216834,
      "tokens_seen": 204931072
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009473620862587764,
      "loss": 1.8473,
      "theoretical_loss": 4.370380321608731,
      "tokens_seen": 204996608
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009473420260782347,
      "loss": 1.978,
      "theoretical_loss": 4.370207324806645,
      "tokens_seen": 205062144
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009473219658976932,
      "loss": 1.8565,
      "theoretical_loss": 4.370034398758996,
      "tokens_seen": 205127680
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009473019057171515,
      "loss": 1.7406,
      "theoretical_loss": 4.369861543414251,
      "tokens_seen": 205193216
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009472818455366099,
      "loss": 2.0285,
      "theoretical_loss": 4.369688758720937,
      "tokens_seen": 205258752
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009472617853560682,
      "loss": 1.851,
      "theoretical_loss": 4.36951604462763,
      "tokens_seen": 205324288
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009472417251755266,
      "loss": 1.8151,
      "theoretical_loss": 4.369343401082966,
      "tokens_seen": 205389824
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000947221664994985,
      "loss": 1.7439,
      "theoretical_loss": 4.369170828035628,
      "tokens_seen": 205455360
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009472016048144434,
      "loss": 1.8962,
      "theoretical_loss": 4.368998325434355,
      "tokens_seen": 205520896
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009471815446339017,
      "loss": 1.8172,
      "theoretical_loss": 4.368825893227943,
      "tokens_seen": 205586432
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009471614844533601,
      "loss": 1.7986,
      "theoretical_loss": 4.368653531365237,
      "tokens_seen": 205651968
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009471414242728185,
      "loss": 1.8557,
      "theoretical_loss": 4.3684812397951385,
      "tokens_seen": 205717504
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009471213640922769,
      "loss": 1.7482,
      "theoretical_loss": 4.3683090184666,
      "tokens_seen": 205783040
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009471013039117352,
      "loss": 1.7962,
      "theoretical_loss": 4.368136867328628,
      "tokens_seen": 205848576
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009470812437311937,
      "loss": 1.7665,
      "theoretical_loss": 4.367964786330285,
      "tokens_seen": 205914112
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009470611835506519,
      "loss": 1.8658,
      "theoretical_loss": 4.367792775420682,
      "tokens_seen": 205979648
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009470411233701104,
      "loss": 1.7218,
      "theoretical_loss": 4.367620834548987,
      "tokens_seen": 206045184
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009470210631895687,
      "loss": 1.8305,
      "theoretical_loss": 4.367448963664419,
      "tokens_seen": 206110720
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009470010030090271,
      "loss": 1.8604,
      "theoretical_loss": 4.367277162716249,
      "tokens_seen": 206176256
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009469809428284855,
      "loss": 1.8282,
      "theoretical_loss": 4.367105431653806,
      "tokens_seen": 206241792
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009469608826479439,
      "loss": 1.8322,
      "theoretical_loss": 4.3669337704264635,
      "tokens_seen": 206307328
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009469408224674022,
      "loss": 1.891,
      "theoretical_loss": 4.366762178983656,
      "tokens_seen": 206372864
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.4905458092689514,
      "objective/train/docs_used": 81724,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7321252822875977,
      "objective/train/original_loss": 1.7321252822875977,
      "objective/train/theoretical_loss": 4.366590657274865,
      "objective/train/tokens_used": 226898400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24100235104560852,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502796173095703,
      "objective/train/weighted_lm_loss": 1.8194676637649536,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0304546356201172,
      "theoretical_loss": 4.366590657274865,
      "tokens_seen": 206438400
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009469207622868607,
      "loss": 1.82,
      "theoretical_loss": 4.366590657274865,
      "tokens_seen": 206438400
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009469007021063189,
      "loss": 1.8988,
      "theoretical_loss": 4.366419205249628,
      "tokens_seen": 206503936
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009468806419257774,
      "loss": 1.9824,
      "theoretical_loss": 4.366247822857533,
      "tokens_seen": 206569472
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009468605817452357,
      "loss": 1.8839,
      "theoretical_loss": 4.36607651004822,
      "tokens_seen": 206635008
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009468405215646941,
      "loss": 1.8377,
      "theoretical_loss": 4.365905266771383,
      "tokens_seen": 206700544
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009468204613841525,
      "loss": 1.9625,
      "theoretical_loss": 4.36573409297677,
      "tokens_seen": 206766080
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009468004012036108,
      "loss": 1.7877,
      "theoretical_loss": 4.365562988614176,
      "tokens_seen": 206831616
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009467803410230692,
      "loss": 1.9814,
      "theoretical_loss": 4.365391953633453,
      "tokens_seen": 206897152
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009467602808425277,
      "loss": 1.8504,
      "theoretical_loss": 4.365220987984502,
      "tokens_seen": 206962688
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009467402206619859,
      "loss": 1.5948,
      "theoretical_loss": 4.365050091617277,
      "tokens_seen": 207028224
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009467201604814444,
      "loss": 1.8719,
      "theoretical_loss": 4.364879264481787,
      "tokens_seen": 207093760
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009467001003009026,
      "loss": 1.9276,
      "theoretical_loss": 4.364708506528087,
      "tokens_seen": 207159296
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009466800401203611,
      "loss": 1.7526,
      "theoretical_loss": 4.364537817706289,
      "tokens_seen": 207224832
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009466599799398195,
      "loss": 1.8202,
      "theoretical_loss": 4.364367197966552,
      "tokens_seen": 207290368
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009466399197592778,
      "loss": 1.7798,
      "theoretical_loss": 4.364196647259092,
      "tokens_seen": 207355904
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009466198595787362,
      "loss": 1.8522,
      "theoretical_loss": 4.364026165534174,
      "tokens_seen": 207421440
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009465997993981947,
      "loss": 1.7636,
      "theoretical_loss": 4.363855752742112,
      "tokens_seen": 207486976
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009465797392176529,
      "loss": 1.9009,
      "theoretical_loss": 4.363685408833276,
      "tokens_seen": 207552512
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009465596790371114,
      "loss": 1.9654,
      "theoretical_loss": 4.363515133758084,
      "tokens_seen": 207618048
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009465396188565697,
      "loss": 1.9998,
      "theoretical_loss": 4.363344927467009,
      "tokens_seen": 207683584
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009465195586760281,
      "loss": 1.9099,
      "theoretical_loss": 4.36317478991057,
      "tokens_seen": 207749120
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009464994984954865,
      "loss": 1.8827,
      "theoretical_loss": 4.363004721039341,
      "tokens_seen": 207814656
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009464794383149448,
      "loss": 1.8781,
      "theoretical_loss": 4.3628347208039475,
      "tokens_seen": 207880192
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009464593781344032,
      "loss": 1.7533,
      "theoretical_loss": 4.362664789155063,
      "tokens_seen": 207945728
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009464393179538616,
      "loss": 1.661,
      "theoretical_loss": 4.362494926043414,
      "tokens_seen": 208011264
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009464192577733199,
      "loss": 1.7946,
      "theoretical_loss": 4.362325131419779,
      "tokens_seen": 208076800
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009463991975927784,
      "loss": 1.9939,
      "theoretical_loss": 4.362155405234985,
      "tokens_seen": 208142336
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009463791374122367,
      "loss": 1.8894,
      "theoretical_loss": 4.361985747439911,
      "tokens_seen": 208207872
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009463590772316951,
      "loss": 1.9331,
      "theoretical_loss": 4.361816157985485,
      "tokens_seen": 208273408
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009463390170511535,
      "loss": 1.8647,
      "theoretical_loss": 4.361646636822689,
      "tokens_seen": 208338944
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009463189568706119,
      "loss": 2.0205,
      "theoretical_loss": 4.361477183902554,
      "tokens_seen": 208404480
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009462988966900702,
      "loss": 1.8844,
      "theoretical_loss": 4.361307799176159,
      "tokens_seen": 208470016
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009462788365095286,
      "loss": 1.8218,
      "theoretical_loss": 4.361138482594637,
      "tokens_seen": 208535552
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009462587763289869,
      "loss": 1.8755,
      "theoretical_loss": 4.360969234109169,
      "tokens_seen": 208601088
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009462387161484454,
      "loss": 1.8704,
      "theoretical_loss": 4.360800053670989,
      "tokens_seen": 208666624
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009462186559679037,
      "loss": 1.7776,
      "theoretical_loss": 4.360630941231378,
      "tokens_seen": 208732160
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009461985957873621,
      "loss": 1.7952,
      "theoretical_loss": 4.360461896741668,
      "tokens_seen": 208797696
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009461785356068204,
      "loss": 1.8615,
      "theoretical_loss": 4.360292920153244,
      "tokens_seen": 208863232
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009461584754262789,
      "loss": 1.8697,
      "theoretical_loss": 4.360124011417536,
      "tokens_seen": 208928768
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009461384152457372,
      "loss": 1.9524,
      "theoretical_loss": 4.359955170486029,
      "tokens_seen": 208994304
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009461183550651956,
      "loss": 1.7694,
      "theoretical_loss": 4.359786397310254,
      "tokens_seen": 209059840
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000946098294884654,
      "loss": 1.7871,
      "theoretical_loss": 4.359617691841795,
      "tokens_seen": 209125376
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009460782347041123,
      "loss": 1.9203,
      "theoretical_loss": 4.359449054032282,
      "tokens_seen": 209190912
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009460581745235707,
      "loss": 1.8939,
      "theoretical_loss": 4.3592804838334,
      "tokens_seen": 209256448
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009460381143430291,
      "loss": 1.7299,
      "theoretical_loss": 4.359111981196877,
      "tokens_seen": 209321984
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009460180541624874,
      "loss": 1.9227,
      "theoretical_loss": 4.358943546074497,
      "tokens_seen": 209387520
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009459979939819459,
      "loss": 1.8355,
      "theoretical_loss": 4.358775178418089,
      "tokens_seen": 209453056
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009459779338014042,
      "loss": 1.9658,
      "theoretical_loss": 4.358606878179533,
      "tokens_seen": 209518592
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009459578736208626,
      "loss": 1.8921,
      "theoretical_loss": 4.358438645310758,
      "tokens_seen": 209584128
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009459378134403211,
      "loss": 1.7757,
      "theoretical_loss": 4.3582704797637435,
      "tokens_seen": 209649664
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.49177494645118713,
      "objective/train/docs_used": 82917,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6986033916473389,
      "objective/train/original_loss": 1.6986033916473389,
      "objective/train/theoretical_loss": 4.358102381490517,
      "objective/train/tokens_used": 230175200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24229730665683746,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050409197807312,
      "objective/train/weighted_lm_loss": 1.7842323780059814,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.358102381490517,
      "tokens_seen": 209715200
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009459177532597793,
      "loss": 1.8477,
      "theoretical_loss": 4.358102381490517,
      "tokens_seen": 209715200
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009458976930792378,
      "loss": 1.8678,
      "theoretical_loss": 4.357934350443156,
      "tokens_seen": 209780736
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009458776328986961,
      "loss": 1.8636,
      "theoretical_loss": 4.357766386573785,
      "tokens_seen": 209846272
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009458575727181545,
      "loss": 1.8975,
      "theoretical_loss": 4.357598489834581,
      "tokens_seen": 209911808
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009458375125376129,
      "loss": 1.8138,
      "theoretical_loss": 4.3574306601777675,
      "tokens_seen": 209977344
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009458174523570712,
      "loss": 1.8093,
      "theoretical_loss": 4.357262897555617,
      "tokens_seen": 210042880
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009457973921765296,
      "loss": 1.8011,
      "theoretical_loss": 4.357095201920452,
      "tokens_seen": 210108416
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009457773319959881,
      "loss": 1.7639,
      "theoretical_loss": 4.356927573224643,
      "tokens_seen": 210173952
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009457572718154463,
      "loss": 1.7829,
      "theoretical_loss": 4.356760011420608,
      "tokens_seen": 210239488
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009457372116349048,
      "loss": 1.9724,
      "theoretical_loss": 4.356592516460817,
      "tokens_seen": 210305024
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000945717151454363,
      "loss": 1.992,
      "theoretical_loss": 4.356425088297786,
      "tokens_seen": 210370560
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009456970912738215,
      "loss": 1.8751,
      "theoretical_loss": 4.3562577268840785,
      "tokens_seen": 210436096
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009456770310932799,
      "loss": 1.9101,
      "theoretical_loss": 4.3560904321723095,
      "tokens_seen": 210501632
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009456569709127382,
      "loss": 1.9617,
      "theoretical_loss": 4.355923204115141,
      "tokens_seen": 210567168
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009456369107321966,
      "loss": 1.9317,
      "theoretical_loss": 4.3557560426652815,
      "tokens_seen": 210632704
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009456168505516551,
      "loss": 1.7983,
      "theoretical_loss": 4.355588947775491,
      "tokens_seen": 210698240
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009455967903711133,
      "loss": 1.8471,
      "theoretical_loss": 4.355421919398576,
      "tokens_seen": 210763776
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009455767301905718,
      "loss": 1.8127,
      "theoretical_loss": 4.35525495748739,
      "tokens_seen": 210829312
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.00094555667001003,
      "loss": 1.8958,
      "theoretical_loss": 4.355088061994837,
      "tokens_seen": 210894848
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009455366098294885,
      "loss": 1.9029,
      "theoretical_loss": 4.354921232873868,
      "tokens_seen": 210960384
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009455165496489469,
      "loss": 1.876,
      "theoretical_loss": 4.354754470077481,
      "tokens_seen": 211025920
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009454964894684052,
      "loss": 1.9396,
      "theoretical_loss": 4.354587773558722,
      "tokens_seen": 211091456
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009454764292878636,
      "loss": 1.8428,
      "theoretical_loss": 4.354421143270687,
      "tokens_seen": 211156992
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000945456369107322,
      "loss": 1.7118,
      "theoretical_loss": 4.354254579166517,
      "tokens_seen": 211222528
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009454363089267803,
      "loss": 1.8104,
      "theoretical_loss": 4.354088081199402,
      "tokens_seen": 211288064
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009454162487462388,
      "loss": 1.8452,
      "theoretical_loss": 4.353921649322579,
      "tokens_seen": 211353600
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009453961885656971,
      "loss": 1.8087,
      "theoretical_loss": 4.353755283489334,
      "tokens_seen": 211419136
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009453761283851555,
      "loss": 1.8845,
      "theoretical_loss": 4.353588983652999,
      "tokens_seen": 211484672
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009453560682046139,
      "loss": 1.8368,
      "theoretical_loss": 4.3534227497669535,
      "tokens_seen": 211550208
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009453360080240723,
      "loss": 1.9266,
      "theoretical_loss": 4.353256581784625,
      "tokens_seen": 211615744
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009453159478435306,
      "loss": 1.7949,
      "theoretical_loss": 4.353090479659488,
      "tokens_seen": 211681280
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000945295887662989,
      "loss": 1.8264,
      "theoretical_loss": 4.3529244433450645,
      "tokens_seen": 211746816
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009452758274824473,
      "loss": 1.7892,
      "theoretical_loss": 4.352758472794923,
      "tokens_seen": 211812352
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009452557673019058,
      "loss": 1.8234,
      "theoretical_loss": 4.35259256796268,
      "tokens_seen": 211877888
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009452357071213641,
      "loss": 1.8501,
      "theoretical_loss": 4.352426728801999,
      "tokens_seen": 211943424
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009452156469408225,
      "loss": 1.8889,
      "theoretical_loss": 4.352260955266589,
      "tokens_seen": 212008960
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009451955867602808,
      "loss": 1.9609,
      "theoretical_loss": 4.352095247310208,
      "tokens_seen": 212074496
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009451755265797393,
      "loss": 1.7887,
      "theoretical_loss": 4.351929604886658,
      "tokens_seen": 212140032
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009451554663991976,
      "loss": 1.8833,
      "theoretical_loss": 4.351764027949793,
      "tokens_seen": 212205568
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000945135406218656,
      "loss": 1.9028,
      "theoretical_loss": 4.3515985164535085,
      "tokens_seen": 212271104
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009451153460381143,
      "loss": 1.8628,
      "theoretical_loss": 4.351433070351748,
      "tokens_seen": 212336640
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009450952858575727,
      "loss": 1.9107,
      "theoretical_loss": 4.351267689598503,
      "tokens_seen": 212402176
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009450752256770311,
      "loss": 1.9297,
      "theoretical_loss": 4.3511023741478105,
      "tokens_seen": 212467712
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009450551654964895,
      "loss": 1.8527,
      "theoretical_loss": 4.350937123953755,
      "tokens_seen": 212533248
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009450351053159478,
      "loss": 1.8359,
      "theoretical_loss": 4.350771938970466,
      "tokens_seen": 212598784
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009450150451354063,
      "loss": 1.9545,
      "theoretical_loss": 4.35060681915212,
      "tokens_seen": 212664320
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009449949849548646,
      "loss": 1.6537,
      "theoretical_loss": 4.35044176445294,
      "tokens_seen": 212729856
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000944974924774323,
      "loss": 1.9184,
      "theoretical_loss": 4.350276774827195,
      "tokens_seen": 212795392
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009449548645937814,
      "loss": 1.7896,
      "theoretical_loss": 4.350111850229202,
      "tokens_seen": 212860928
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009449348044132397,
      "loss": 1.8893,
      "theoretical_loss": 4.3499469906133195,
      "tokens_seen": 212926464
    },
    {
      "epoch": 0.06,
      "objective/train/advantage_avg": 0.49271607398986816,
      "objective/train/docs_used": 84140,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6249154806137085,
      "objective/train/original_loss": 1.624915599822998,
      "objective/train/theoretical_loss": 4.349782195933957,
      "objective/train/tokens_used": 233452000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24310225248336792,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505075454711914,
      "objective/train/weighted_lm_loss": 1.7071648836135864,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.349782195933957,
      "tokens_seen": 212992000
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009449147442326981,
      "loss": 1.723,
      "theoretical_loss": 4.349782195933957,
      "tokens_seen": 212992000
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009448946840521565,
      "loss": 1.7778,
      "theoretical_loss": 4.349617466145568,
      "tokens_seen": 213057536
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009448746238716148,
      "loss": 1.7859,
      "theoretical_loss": 4.34945280120265,
      "tokens_seen": 213123072
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009448545636910733,
      "loss": 1.8328,
      "theoretical_loss": 4.349288201059751,
      "tokens_seen": 213188608
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009448345035105315,
      "loss": 1.7755,
      "theoretical_loss": 4.34912366567146,
      "tokens_seen": 213254144
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.00094481444332999,
      "loss": 1.9151,
      "theoretical_loss": 4.348959194992415,
      "tokens_seen": 213319680
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009447943831494484,
      "loss": 1.8385,
      "theoretical_loss": 4.348794788977298,
      "tokens_seen": 213385216
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009447743229689067,
      "loss": 1.8569,
      "theoretical_loss": 4.348630447580839,
      "tokens_seen": 213450752
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009447542627883651,
      "loss": 1.8746,
      "theoretical_loss": 4.34846617075781,
      "tokens_seen": 213516288
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009447342026078234,
      "loss": 1.7508,
      "theoretical_loss": 4.34830195846303,
      "tokens_seen": 213581824
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009447141424272818,
      "loss": 1.9108,
      "theoretical_loss": 4.348137810651366,
      "tokens_seen": 213647360
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009446940822467403,
      "loss": 1.8,
      "theoretical_loss": 4.347973727277726,
      "tokens_seen": 213712896
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009446740220661985,
      "loss": 1.8573,
      "theoretical_loss": 4.347809708297067,
      "tokens_seen": 213778432
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000944653961885657,
      "loss": 1.7998,
      "theoretical_loss": 4.347645753664389,
      "tokens_seen": 213843968
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009446339017051154,
      "loss": 1.7913,
      "theoretical_loss": 4.347481863334738,
      "tokens_seen": 213909504
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009446138415245737,
      "loss": 1.849,
      "theoretical_loss": 4.347318037263206,
      "tokens_seen": 213975040
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009445937813440321,
      "loss": 1.8244,
      "theoretical_loss": 4.34715427540493,
      "tokens_seen": 214040576
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009445737211634905,
      "loss": 1.8991,
      "theoretical_loss": 4.34699057771509,
      "tokens_seen": 214106112
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009445536609829488,
      "loss": 1.8225,
      "theoretical_loss": 4.346826944148912,
      "tokens_seen": 214171648
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009445336008024073,
      "loss": 1.8634,
      "theoretical_loss": 4.346663374661669,
      "tokens_seen": 214237184
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009445135406218655,
      "loss": 1.8842,
      "theoretical_loss": 4.346499869208676,
      "tokens_seen": 214302720
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.000944493480441324,
      "loss": 1.888,
      "theoretical_loss": 4.346336427745294,
      "tokens_seen": 214368256
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009444734202607823,
      "loss": 1.8777,
      "theoretical_loss": 4.3461730502269305,
      "tokens_seen": 214433792
    },
    {
      "epoch": 0.06,
      "learning_rate": 0.0009444533600802407,
      "loss": 1.9699,
      "theoretical_loss": 4.346009736609034,
      "tokens_seen": 214499328
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009444332998996992,
      "loss": 1.7312,
      "theoretical_loss": 4.345846486847099,
      "tokens_seen": 214564864
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009444132397191575,
      "loss": 1.89,
      "theoretical_loss": 4.345683300896668,
      "tokens_seen": 214630400
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009443931795386159,
      "loss": 1.7744,
      "theoretical_loss": 4.345520178713323,
      "tokens_seen": 214695936
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009443731193580743,
      "loss": 1.8952,
      "theoretical_loss": 4.345357120252692,
      "tokens_seen": 214761472
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009443530591775327,
      "loss": 1.988,
      "theoretical_loss": 4.34519412547045,
      "tokens_seen": 214827008
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000944332998996991,
      "loss": 1.908,
      "theoretical_loss": 4.345031194322313,
      "tokens_seen": 214892544
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009443129388164494,
      "loss": 1.9555,
      "theoretical_loss": 4.344868326764045,
      "tokens_seen": 214958080
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009442928786359077,
      "loss": 1.9617,
      "theoretical_loss": 4.344705522751449,
      "tokens_seen": 215023616
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009442728184553662,
      "loss": 1.8555,
      "theoretical_loss": 4.344542782240376,
      "tokens_seen": 215089152
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009442527582748245,
      "loss": 1.8447,
      "theoretical_loss": 4.344380105186721,
      "tokens_seen": 215154688
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009442326980942829,
      "loss": 1.8825,
      "theoretical_loss": 4.344217491546422,
      "tokens_seen": 215220224
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009442126379137412,
      "loss": 2.0065,
      "theoretical_loss": 4.344054941275459,
      "tokens_seen": 215285760
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009441925777331997,
      "loss": 1.8606,
      "theoretical_loss": 4.343892454329863,
      "tokens_seen": 215351296
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000944172517552658,
      "loss": 1.8549,
      "theoretical_loss": 4.343730030665699,
      "tokens_seen": 215416832
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009441524573721164,
      "loss": 1.691,
      "theoretical_loss": 4.343567670239084,
      "tokens_seen": 215482368
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009441323971915747,
      "loss": 1.7313,
      "theoretical_loss": 4.343405373006175,
      "tokens_seen": 215547904
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009441123370110331,
      "loss": 1.9053,
      "theoretical_loss": 4.343243138923173,
      "tokens_seen": 215613440
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009440922768304915,
      "loss": 2.0116,
      "theoretical_loss": 4.343080967946323,
      "tokens_seen": 215678976
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009440722166499499,
      "loss": 1.9259,
      "theoretical_loss": 4.342918860031914,
      "tokens_seen": 215744512
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009440521564694082,
      "loss": 1.7324,
      "theoretical_loss": 4.342756815136278,
      "tokens_seen": 215810048
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009440320962888667,
      "loss": 1.9033,
      "theoretical_loss": 4.3425948332157915,
      "tokens_seen": 215875584
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000944012036108325,
      "loss": 1.7331,
      "theoretical_loss": 4.3424329142268725,
      "tokens_seen": 215941120
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009439919759277834,
      "loss": 1.7929,
      "theoretical_loss": 4.342271058125983,
      "tokens_seen": 216006656
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009439719157472418,
      "loss": 1.9002,
      "theoretical_loss": 4.342109264869631,
      "tokens_seen": 216072192
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009439518555667001,
      "loss": 1.9297,
      "theoretical_loss": 4.341947534414364,
      "tokens_seen": 216137728
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009439317953861585,
      "loss": 1.83,
      "theoretical_loss": 4.341785866716774,
      "tokens_seen": 216203264
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.4911940395832062,
      "objective/train/docs_used": 85191,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.79802668094635,
      "objective/train/original_loss": 1.79802668094635,
      "objective/train/theoretical_loss": 4.341624261733497,
      "objective/train/tokens_used": 236728800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24179859459400177,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503485202789307,
      "objective/train/weighted_lm_loss": 1.8884522914886475,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.341624261733497,
      "tokens_seen": 216268800
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000943911735205617,
      "loss": 1.8136,
      "theoretical_loss": 4.341624261733497,
      "tokens_seen": 216268800
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009438916750250752,
      "loss": 1.8533,
      "theoretical_loss": 4.341462719421211,
      "tokens_seen": 216334336
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009438716148445337,
      "loss": 1.8464,
      "theoretical_loss": 4.341301239736639,
      "tokens_seen": 216399872
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009438515546639919,
      "loss": 1.9912,
      "theoretical_loss": 4.341139822636544,
      "tokens_seen": 216465408
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009438314944834504,
      "loss": 1.8442,
      "theoretical_loss": 4.340978468077735,
      "tokens_seen": 216530944
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009438114343029088,
      "loss": 1.9045,
      "theoretical_loss": 4.34081717601706,
      "tokens_seen": 216596480
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009437913741223671,
      "loss": 1.7926,
      "theoretical_loss": 4.340655946411413,
      "tokens_seen": 216662016
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009437713139418255,
      "loss": 1.896,
      "theoretical_loss": 4.340494779217732,
      "tokens_seen": 216727552
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009437512537612838,
      "loss": 1.8232,
      "theoretical_loss": 4.340333674392992,
      "tokens_seen": 216793088
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009437311935807422,
      "loss": 1.9165,
      "theoretical_loss": 4.340172631894218,
      "tokens_seen": 216858624
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009437111334002007,
      "loss": 1.9249,
      "theoretical_loss": 4.340011651678472,
      "tokens_seen": 216924160
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009436910732196589,
      "loss": 1.757,
      "theoretical_loss": 4.33985073370286,
      "tokens_seen": 216989696
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009436710130391174,
      "loss": 1.8661,
      "theoretical_loss": 4.339689877924531,
      "tokens_seen": 217055232
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009436509528585758,
      "loss": 1.8468,
      "theoretical_loss": 4.339529084300676,
      "tokens_seen": 217120768
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009436308926780341,
      "loss": 1.8,
      "theoretical_loss": 4.339368352788531,
      "tokens_seen": 217186304
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009436108324974925,
      "loss": 1.8137,
      "theoretical_loss": 4.339207683345369,
      "tokens_seen": 217251840
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009435907723169509,
      "loss": 1.8983,
      "theoretical_loss": 4.3390470759285105,
      "tokens_seen": 217317376
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009435707121364092,
      "loss": 1.8911,
      "theoretical_loss": 4.338886530495314,
      "tokens_seen": 217382912
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009435506519558677,
      "loss": 1.8343,
      "theoretical_loss": 4.338726047003183,
      "tokens_seen": 217448448
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009435305917753259,
      "loss": 1.797,
      "theoretical_loss": 4.338565625409563,
      "tokens_seen": 217513984
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009435105315947844,
      "loss": 1.8644,
      "theoretical_loss": 4.338405265671941,
      "tokens_seen": 217579520
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009434904714142427,
      "loss": 1.6793,
      "theoretical_loss": 4.338244967747844,
      "tokens_seen": 217645056
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009434704112337011,
      "loss": 1.834,
      "theoretical_loss": 4.338084731594844,
      "tokens_seen": 217710592
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009434503510531595,
      "loss": 1.7865,
      "theoretical_loss": 4.3379245571705525,
      "tokens_seen": 217776128
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009434302908726179,
      "loss": 1.7227,
      "theoretical_loss": 4.337764444432625,
      "tokens_seen": 217841664
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009434102306920762,
      "loss": 1.8882,
      "theoretical_loss": 4.337604393338758,
      "tokens_seen": 217907200
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009433901705115347,
      "loss": 1.8603,
      "theoretical_loss": 4.337444403846687,
      "tokens_seen": 217972736
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000943370110330993,
      "loss": 1.9331,
      "theoretical_loss": 4.3372844759141955,
      "tokens_seen": 218038272
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009433500501504514,
      "loss": 1.7618,
      "theoretical_loss": 4.337124609499101,
      "tokens_seen": 218103808
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009433299899699097,
      "loss": 1.8607,
      "theoretical_loss": 4.336964804559269,
      "tokens_seen": 218169344
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009433099297893681,
      "loss": 1.8718,
      "theoretical_loss": 4.336805061052601,
      "tokens_seen": 218234880
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009432898696088265,
      "loss": 1.6282,
      "theoretical_loss": 4.336645378937046,
      "tokens_seen": 218300416
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009432698094282849,
      "loss": 1.7722,
      "theoretical_loss": 4.336485758170589,
      "tokens_seen": 218365952
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009432497492477432,
      "loss": 1.8395,
      "theoretical_loss": 4.336326198711259,
      "tokens_seen": 218431488
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009432296890672016,
      "loss": 1.8712,
      "theoretical_loss": 4.336166700517125,
      "tokens_seen": 218497024
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.00094320962888666,
      "loss": 1.8655,
      "theoretical_loss": 4.3360072635463,
      "tokens_seen": 218562560
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009431895687061184,
      "loss": 1.989,
      "theoretical_loss": 4.335847887756934,
      "tokens_seen": 218628096
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009431695085255767,
      "loss": 1.8191,
      "theoretical_loss": 4.335688573107221,
      "tokens_seen": 218693632
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009431494483450351,
      "loss": 1.8258,
      "theoretical_loss": 4.3355293195553966,
      "tokens_seen": 218759168
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009431293881644934,
      "loss": 1.8348,
      "theoretical_loss": 4.335370127059735,
      "tokens_seen": 218824704
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009431093279839519,
      "loss": 1.7862,
      "theoretical_loss": 4.335210995578553,
      "tokens_seen": 218890240
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009430892678034102,
      "loss": 1.8832,
      "theoretical_loss": 4.335051925070209,
      "tokens_seen": 218955776
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009430692076228686,
      "loss": 1.8708,
      "theoretical_loss": 4.334892915493099,
      "tokens_seen": 219021312
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000943049147442327,
      "loss": 1.73,
      "theoretical_loss": 4.334733966805665,
      "tokens_seen": 219086848
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009430290872617854,
      "loss": 1.9105,
      "theoretical_loss": 4.334575078966383,
      "tokens_seen": 219152384
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009430090270812437,
      "loss": 1.8589,
      "theoretical_loss": 4.334416251933778,
      "tokens_seen": 219217920
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009429889669007022,
      "loss": 1.8095,
      "theoretical_loss": 4.334257485666408,
      "tokens_seen": 219283456
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009429689067201604,
      "loss": 1.9426,
      "theoretical_loss": 4.334098780122876,
      "tokens_seen": 219348992
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009429488465396189,
      "loss": 1.9143,
      "theoretical_loss": 4.333940135261823,
      "tokens_seen": 219414528
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009429287863590774,
      "loss": 1.8288,
      "theoretical_loss": 4.333781551041935,
      "tokens_seen": 219480064
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.4948246479034424,
      "objective/train/docs_used": 86440,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6815640926361084,
      "objective/train/original_loss": 1.6815643310546875,
      "objective/train/theoretical_loss": 4.3336230274219325,
      "objective/train/tokens_used": 240005600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24505706131458282,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507283210754395,
      "objective/train/weighted_lm_loss": 1.7667181491851807,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.3336230274219325,
      "tokens_seen": 219545600
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009429087261785356,
      "loss": 1.7774,
      "theoretical_loss": 4.3336230274219325,
      "tokens_seen": 219545600
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009428886659979941,
      "loss": 1.7581,
      "theoretical_loss": 4.333464564360581,
      "tokens_seen": 219611136
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009428686058174523,
      "loss": 1.9298,
      "theoretical_loss": 4.333306161816684,
      "tokens_seen": 219676672
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009428485456369108,
      "loss": 1.9582,
      "theoretical_loss": 4.333147819749085,
      "tokens_seen": 219742208
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009428284854563692,
      "loss": 1.9185,
      "theoretical_loss": 4.332989538116671,
      "tokens_seen": 219807744
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009428084252758275,
      "loss": 1.8663,
      "theoretical_loss": 4.332831316878364,
      "tokens_seen": 219873280
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009427883650952859,
      "loss": 1.8499,
      "theoretical_loss": 4.332673155993131,
      "tokens_seen": 219938816
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009427683049147442,
      "loss": 1.894,
      "theoretical_loss": 4.332515055419976,
      "tokens_seen": 220004352
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009427482447342026,
      "loss": 1.7753,
      "theoretical_loss": 4.332357015117946,
      "tokens_seen": 220069888
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009427281845536611,
      "loss": 1.8143,
      "theoretical_loss": 4.332199035046124,
      "tokens_seen": 220135424
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009427081243731193,
      "loss": 1.7558,
      "theoretical_loss": 4.332041115163636,
      "tokens_seen": 220200960
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009426880641925778,
      "loss": 1.8822,
      "theoretical_loss": 4.3318832554296485,
      "tokens_seen": 220266496
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009426680040120362,
      "loss": 1.8592,
      "theoretical_loss": 4.3317254558033635,
      "tokens_seen": 220332032
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009426479438314945,
      "loss": 1.9003,
      "theoretical_loss": 4.331567716244027,
      "tokens_seen": 220397568
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009426278836509529,
      "loss": 1.9003,
      "theoretical_loss": 4.331410036710925,
      "tokens_seen": 220463104
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009426078234704113,
      "loss": 1.792,
      "theoretical_loss": 4.331252417163379,
      "tokens_seen": 220528640
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009425877632898696,
      "loss": 1.8564,
      "theoretical_loss": 4.331094857560754,
      "tokens_seen": 220594176
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009425677031093281,
      "loss": 1.8892,
      "theoretical_loss": 4.330937357862453,
      "tokens_seen": 220659712
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009425476429287863,
      "loss": 1.8259,
      "theoretical_loss": 4.330779918027919,
      "tokens_seen": 220725248
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009425275827482448,
      "loss": 1.9108,
      "theoretical_loss": 4.330622538016635,
      "tokens_seen": 220790784
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009425075225677031,
      "loss": 1.8322,
      "theoretical_loss": 4.330465217788122,
      "tokens_seen": 220856320
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009424874623871615,
      "loss": 1.933,
      "theoretical_loss": 4.33030795730194,
      "tokens_seen": 220921856
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009424674022066199,
      "loss": 1.808,
      "theoretical_loss": 4.330150756517692,
      "tokens_seen": 220987392
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009424473420260783,
      "loss": 1.716,
      "theoretical_loss": 4.329993615395017,
      "tokens_seen": 221052928
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009424272818455366,
      "loss": 1.8106,
      "theoretical_loss": 4.329836533893593,
      "tokens_seen": 221118464
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009424072216649951,
      "loss": 1.9543,
      "theoretical_loss": 4.329679511973138,
      "tokens_seen": 221184000
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009423871614844533,
      "loss": 1.8496,
      "theoretical_loss": 4.3295225495934115,
      "tokens_seen": 221249536
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009423671013039118,
      "loss": 1.6599,
      "theoretical_loss": 4.3293656467142085,
      "tokens_seen": 221315072
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009423470411233701,
      "loss": 1.8111,
      "theoretical_loss": 4.329208803295365,
      "tokens_seen": 221380608
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009423269809428285,
      "loss": 1.734,
      "theoretical_loss": 4.329052019296754,
      "tokens_seen": 221446144
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009423069207622869,
      "loss": 1.795,
      "theoretical_loss": 4.328895294678292,
      "tokens_seen": 221511680
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009422868605817453,
      "loss": 1.9343,
      "theoretical_loss": 4.328738629399928,
      "tokens_seen": 221577216
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009422668004012036,
      "loss": 1.7639,
      "theoretical_loss": 4.328582023421655,
      "tokens_seen": 221642752
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000942246740220662,
      "loss": 1.8256,
      "theoretical_loss": 4.328425476703503,
      "tokens_seen": 221708288
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009422266800401204,
      "loss": 1.8873,
      "theoretical_loss": 4.32826898920554,
      "tokens_seen": 221773824
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009422066198595788,
      "loss": 1.8476,
      "theoretical_loss": 4.328112560887875,
      "tokens_seen": 221839360
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009421865596790371,
      "loss": 1.9737,
      "theoretical_loss": 4.327956191710651,
      "tokens_seen": 221904896
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009421664994984955,
      "loss": 1.884,
      "theoretical_loss": 4.327799881634056,
      "tokens_seen": 221970432
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009421464393179538,
      "loss": 1.8819,
      "theoretical_loss": 4.3276436306183115,
      "tokens_seen": 222035968
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009421263791374123,
      "loss": 1.7373,
      "theoretical_loss": 4.327487438623679,
      "tokens_seen": 222101504
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009421063189568706,
      "loss": 1.8465,
      "theoretical_loss": 4.32733130561046,
      "tokens_seen": 222167040
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000942086258776329,
      "loss": 1.7857,
      "theoretical_loss": 4.327175231538991,
      "tokens_seen": 222232576
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009420661985957874,
      "loss": 1.7946,
      "theoretical_loss": 4.327019216369651,
      "tokens_seen": 222298112
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009420461384152458,
      "loss": 1.8871,
      "theoretical_loss": 4.326863260062854,
      "tokens_seen": 222363648
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009420260782347041,
      "loss": 1.7786,
      "theoretical_loss": 4.326707362579054,
      "tokens_seen": 222429184
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009420060180541626,
      "loss": 1.7791,
      "theoretical_loss": 4.326551523878742,
      "tokens_seen": 222494720
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009419859578736208,
      "loss": 1.7992,
      "theoretical_loss": 4.32639574392245,
      "tokens_seen": 222560256
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009419658976930793,
      "loss": 1.735,
      "theoretical_loss": 4.326240022670743,
      "tokens_seen": 222625792
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009419458375125376,
      "loss": 1.8356,
      "theoretical_loss": 4.326084360084229,
      "tokens_seen": 222691328
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000941925777331996,
      "loss": 1.72,
      "theoretical_loss": 4.325928756123551,
      "tokens_seen": 222756864
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.4933840334415436,
      "objective/train/docs_used": 87601,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.578743815422058,
      "objective/train/original_loss": 1.5787436962127686,
      "objective/train/theoretical_loss": 4.325773210749392,
      "objective/train/tokens_used": 243282400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24385935068130493,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505781173706055,
      "objective/train/weighted_lm_loss": 1.658752679824829,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.325773210749392,
      "tokens_seen": 222822400
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009419057171514544,
      "loss": 1.7028,
      "theoretical_loss": 4.325773210749392,
      "tokens_seen": 222822400
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009418856569709127,
      "loss": 1.9186,
      "theoretical_loss": 4.325617723922472,
      "tokens_seen": 222887936
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009418655967903711,
      "loss": 1.7035,
      "theoretical_loss": 4.325462295603547,
      "tokens_seen": 222953472
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009418455366098296,
      "loss": 1.8367,
      "theoretical_loss": 4.325306925753415,
      "tokens_seen": 223019008
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009418254764292878,
      "loss": 1.8699,
      "theoretical_loss": 4.325151614332908,
      "tokens_seen": 223084544
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009418054162487463,
      "loss": 1.8548,
      "theoretical_loss": 4.324996361302899,
      "tokens_seen": 223150080
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009417853560682045,
      "loss": 1.8911,
      "theoretical_loss": 4.324841166624294,
      "tokens_seen": 223215616
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000941765295887663,
      "loss": 1.8347,
      "theoretical_loss": 4.324686030258042,
      "tokens_seen": 223281152
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009417452357071214,
      "loss": 1.8127,
      "theoretical_loss": 4.3245309521651265,
      "tokens_seen": 223346688
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009417251755265797,
      "loss": 1.8398,
      "theoretical_loss": 4.324375932306568,
      "tokens_seen": 223412224
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009417051153460381,
      "loss": 1.8697,
      "theoretical_loss": 4.324220970643426,
      "tokens_seen": 223477760
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009416850551654966,
      "loss": 1.8429,
      "theoretical_loss": 4.324066067136798,
      "tokens_seen": 223543296
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009416649949849548,
      "loss": 1.807,
      "theoretical_loss": 4.323911221747817,
      "tokens_seen": 223608832
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009416449348044133,
      "loss": 1.9073,
      "theoretical_loss": 4.323756434437655,
      "tokens_seen": 223674368
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009416248746238715,
      "loss": 1.7969,
      "theoretical_loss": 4.323601705167521,
      "tokens_seen": 223739904
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.00094160481444333,
      "loss": 1.8434,
      "theoretical_loss": 4.323447033898661,
      "tokens_seen": 223805440
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009415847542627884,
      "loss": 1.9763,
      "theoretical_loss": 4.323292420592356,
      "tokens_seen": 223870976
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009415646940822467,
      "loss": 1.7727,
      "theoretical_loss": 4.323137865209928,
      "tokens_seen": 223936512
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009415446339017051,
      "loss": 1.8935,
      "theoretical_loss": 4.322983367712735,
      "tokens_seen": 224002048
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009415245737211635,
      "loss": 1.7683,
      "theoretical_loss": 4.322828928062171,
      "tokens_seen": 224067584
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009415045135406218,
      "loss": 1.918,
      "theoretical_loss": 4.322674546219666,
      "tokens_seen": 224133120
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009414844533600803,
      "loss": 1.8897,
      "theoretical_loss": 4.322520222146689,
      "tokens_seen": 224198656
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009414643931795386,
      "loss": 1.811,
      "theoretical_loss": 4.322365955804749,
      "tokens_seen": 224264192
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000941444332998997,
      "loss": 1.8297,
      "theoretical_loss": 4.322211747155383,
      "tokens_seen": 224329728
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009414242728184555,
      "loss": 1.8643,
      "theoretical_loss": 4.322057596160174,
      "tokens_seen": 224395264
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009414042126379137,
      "loss": 1.6737,
      "theoretical_loss": 4.321903502780736,
      "tokens_seen": 224460800
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009413841524573722,
      "loss": 1.8232,
      "theoretical_loss": 4.321749466978722,
      "tokens_seen": 224526336
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009413640922768305,
      "loss": 1.7283,
      "theoretical_loss": 4.321595488715822,
      "tokens_seen": 224591872
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009413440320962889,
      "loss": 1.8719,
      "theoretical_loss": 4.321441567953762,
      "tokens_seen": 224657408
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009413239719157473,
      "loss": 1.8208,
      "theoretical_loss": 4.3212877046543055,
      "tokens_seen": 224722944
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009413039117352057,
      "loss": 1.7285,
      "theoretical_loss": 4.32113389877925,
      "tokens_seen": 224788480
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000941283851554664,
      "loss": 1.7993,
      "theoretical_loss": 4.320980150290432,
      "tokens_seen": 224854016
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009412637913741224,
      "loss": 1.9261,
      "theoretical_loss": 4.320826459149725,
      "tokens_seen": 224919552
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009412437311935808,
      "loss": 1.911,
      "theoretical_loss": 4.320672825319035,
      "tokens_seen": 224985088
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009412236710130392,
      "loss": 1.9533,
      "theoretical_loss": 4.320519248760309,
      "tokens_seen": 225050624
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009412036108324975,
      "loss": 1.9988,
      "theoretical_loss": 4.3203657294355295,
      "tokens_seen": 225116160
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000941183550651956,
      "loss": 1.8758,
      "theoretical_loss": 4.3202122673067125,
      "tokens_seen": 225181696
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009411634904714142,
      "loss": 1.8191,
      "theoretical_loss": 4.3200588623359115,
      "tokens_seen": 225247232
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009411434302908727,
      "loss": 1.8512,
      "theoretical_loss": 4.319905514485218,
      "tokens_seen": 225312768
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000941123370110331,
      "loss": 1.7275,
      "theoretical_loss": 4.319752223716758,
      "tokens_seen": 225378304
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009411033099297894,
      "loss": 1.7934,
      "theoretical_loss": 4.319598989992695,
      "tokens_seen": 225443840
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009410832497492478,
      "loss": 1.8376,
      "theoretical_loss": 4.319445813275225,
      "tokens_seen": 225509376
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009410631895687062,
      "loss": 2.0349,
      "theoretical_loss": 4.3192926935265845,
      "tokens_seen": 225574912
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009410431293881645,
      "loss": 1.9335,
      "theoretical_loss": 4.319139630709043,
      "tokens_seen": 225640448
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000941023069207623,
      "loss": 1.7679,
      "theoretical_loss": 4.318986624784908,
      "tokens_seen": 225705984
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009410030090270812,
      "loss": 1.8803,
      "theoretical_loss": 4.3188336757165215,
      "tokens_seen": 225771520
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009409829488465397,
      "loss": 1.9487,
      "theoretical_loss": 4.318680783466261,
      "tokens_seen": 225837056
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000940962888665998,
      "loss": 1.7355,
      "theoretical_loss": 4.318527947996542,
      "tokens_seen": 225902592
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009409428284854564,
      "loss": 1.8677,
      "theoretical_loss": 4.318375169269813,
      "tokens_seen": 225968128
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009409227683049148,
      "loss": 1.8113,
      "theoretical_loss": 4.3182224472485595,
      "tokens_seen": 226033664
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.49421149492263794,
      "objective/train/docs_used": 88830,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.722211480140686,
      "objective/train/original_loss": 1.7222118377685547,
      "objective/train/theoretical_loss": 4.3180697818953035,
      "objective/train/tokens_used": 246559200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24472033977508545,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050665259361267,
      "objective/train/weighted_lm_loss": 1.809336543083191,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.3180697818953035,
      "tokens_seen": 226099200
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009409027081243731,
      "loss": 1.7996,
      "theoretical_loss": 4.3180697818953035,
      "tokens_seen": 226099200
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009408826479438315,
      "loss": 1.9476,
      "theoretical_loss": 4.317917173172602,
      "tokens_seen": 226164736
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.00094086258776329,
      "loss": 1.7379,
      "theoretical_loss": 4.317764621043046,
      "tokens_seen": 226230272
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009408425275827482,
      "loss": 1.8394,
      "theoretical_loss": 4.317612125469264,
      "tokens_seen": 226295808
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009408224674022067,
      "loss": 1.7664,
      "theoretical_loss": 4.31745968641392,
      "tokens_seen": 226361344
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009408024072216649,
      "loss": 1.8748,
      "theoretical_loss": 4.317307303839712,
      "tokens_seen": 226426880
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009407823470411234,
      "loss": 1.7081,
      "theoretical_loss": 4.317154977709375,
      "tokens_seen": 226492416
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009407622868605818,
      "loss": 1.6976,
      "theoretical_loss": 4.3170027079856785,
      "tokens_seen": 226557952
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009407422266800401,
      "loss": 1.795,
      "theoretical_loss": 4.316850494631427,
      "tokens_seen": 226623488
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009407221664994985,
      "loss": 1.881,
      "theoretical_loss": 4.316698337609461,
      "tokens_seen": 226689024
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000940702106318957,
      "loss": 1.8121,
      "theoretical_loss": 4.3165462368826555,
      "tokens_seen": 226754560
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009406820461384152,
      "loss": 1.729,
      "theoretical_loss": 4.316394192413922,
      "tokens_seen": 226820096
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009406619859578737,
      "loss": 1.8137,
      "theoretical_loss": 4.316242204166205,
      "tokens_seen": 226885632
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000940641925777332,
      "loss": 1.8044,
      "theoretical_loss": 4.316090272102486,
      "tokens_seen": 226951168
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009406218655967904,
      "loss": 1.8453,
      "theoretical_loss": 4.315938396185782,
      "tokens_seen": 227016704
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009406018054162488,
      "loss": 1.7808,
      "theoretical_loss": 4.315786576379143,
      "tokens_seen": 227082240
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009405817452357071,
      "loss": 1.7636,
      "theoretical_loss": 4.315634812645654,
      "tokens_seen": 227147776
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009405616850551655,
      "loss": 1.7784,
      "theoretical_loss": 4.315483104948438,
      "tokens_seen": 227213312
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009405416248746239,
      "loss": 1.885,
      "theoretical_loss": 4.315331453250648,
      "tokens_seen": 227278848
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009405215646940822,
      "loss": 1.7154,
      "theoretical_loss": 4.315179857515477,
      "tokens_seen": 227344384
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009405015045135407,
      "loss": 1.7473,
      "theoretical_loss": 4.315028317706149,
      "tokens_seen": 227409920
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000940481444332999,
      "loss": 1.8366,
      "theoretical_loss": 4.314876833785926,
      "tokens_seen": 227475456
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009404613841524574,
      "loss": 1.8375,
      "theoretical_loss": 4.314725405718099,
      "tokens_seen": 227540992
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009404413239719157,
      "loss": 1.7129,
      "theoretical_loss": 4.314574033466001,
      "tokens_seen": 227606528
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009404212637913741,
      "loss": 1.789,
      "theoretical_loss": 4.314422716992994,
      "tokens_seen": 227672064
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009404012036108325,
      "loss": 1.8073,
      "theoretical_loss": 4.314271456262479,
      "tokens_seen": 227737600
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009403811434302909,
      "loss": 1.8551,
      "theoretical_loss": 4.314120251237887,
      "tokens_seen": 227803136
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009403610832497492,
      "loss": 1.8417,
      "theoretical_loss": 4.313969101882687,
      "tokens_seen": 227868672
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009403410230692077,
      "loss": 1.7539,
      "theoretical_loss": 4.31381800816038,
      "tokens_seen": 227934208
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000940320962888666,
      "loss": 1.8444,
      "theoretical_loss": 4.313666970034505,
      "tokens_seen": 227999744
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009403009027081244,
      "loss": 1.9733,
      "theoretical_loss": 4.31351598746863,
      "tokens_seen": 228065280
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009402808425275827,
      "loss": 1.7642,
      "theoretical_loss": 4.313365060426363,
      "tokens_seen": 228130816
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009402607823470412,
      "loss": 1.7757,
      "theoretical_loss": 4.313214188871341,
      "tokens_seen": 228196352
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009402407221664995,
      "loss": 1.8742,
      "theoretical_loss": 4.31306337276724,
      "tokens_seen": 228261888
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009402206619859579,
      "loss": 1.8931,
      "theoretical_loss": 4.312912612077767,
      "tokens_seen": 228327424
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009402006018054162,
      "loss": 1.848,
      "theoretical_loss": 4.312761906766664,
      "tokens_seen": 228392960
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009401805416248746,
      "loss": 1.8055,
      "theoretical_loss": 4.312611256797708,
      "tokens_seen": 228458496
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000940160481444333,
      "loss": 1.8979,
      "theoretical_loss": 4.3124606621347095,
      "tokens_seen": 228524032
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009401404212637914,
      "loss": 1.8403,
      "theoretical_loss": 4.312310122741512,
      "tokens_seen": 228589568
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009401203610832497,
      "loss": 1.8505,
      "theoretical_loss": 4.312159638581995,
      "tokens_seen": 228655104
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009401003009027082,
      "loss": 1.8639,
      "theoretical_loss": 4.31200920962007,
      "tokens_seen": 228720640
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009400802407221665,
      "loss": 1.8759,
      "theoretical_loss": 4.311858835819684,
      "tokens_seen": 228786176
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009400601805416249,
      "loss": 1.8338,
      "theoretical_loss": 4.311708517144817,
      "tokens_seen": 228851712
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009400401203610832,
      "loss": 1.838,
      "theoretical_loss": 4.3115582535594825,
      "tokens_seen": 228917248
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009400200601805416,
      "loss": 1.796,
      "theoretical_loss": 4.311408045027728,
      "tokens_seen": 228982784
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.00094,
      "loss": 1.9516,
      "theoretical_loss": 4.3112578915136375,
      "tokens_seen": 229048320
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009399799398194584,
      "loss": 1.8333,
      "theoretical_loss": 4.311107792981323,
      "tokens_seen": 229113856
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009399598796389167,
      "loss": 1.8905,
      "theoretical_loss": 4.310957749394935,
      "tokens_seen": 229179392
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009399398194583752,
      "loss": 1.8143,
      "theoretical_loss": 4.310807760718656,
      "tokens_seen": 229244928
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009399197592778335,
      "loss": 1.8554,
      "theoretical_loss": 4.310657826916701,
      "tokens_seen": 229310464
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.4928604066371918,
      "objective/train/docs_used": 90061,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8705188035964966,
      "objective/train/original_loss": 1.8705189228057861,
      "objective/train/theoretical_loss": 4.310507947953321,
      "objective/train/tokens_used": 249836000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24335232377052307,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050523281097412,
      "objective/train/weighted_lm_loss": 1.9649579524993896,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.310507947953321,
      "tokens_seen": 229376000
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009398996990972919,
      "loss": 1.8946,
      "theoretical_loss": 4.310507947953321,
      "tokens_seen": 229376000
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009398796389167504,
      "loss": 1.7751,
      "theoretical_loss": 4.310358123792798,
      "tokens_seen": 229441536
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009398595787362086,
      "loss": 1.9335,
      "theoretical_loss": 4.31020835439945,
      "tokens_seen": 229507072
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009398395185556671,
      "loss": 1.9429,
      "theoretical_loss": 4.310058639737626,
      "tokens_seen": 229572608
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009398194583751253,
      "loss": 1.8902,
      "theoretical_loss": 4.309908979771709,
      "tokens_seen": 229638144
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009397993981945838,
      "loss": 1.7913,
      "theoretical_loss": 4.3097593744661165,
      "tokens_seen": 229703680
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009397793380140422,
      "loss": 1.9064,
      "theoretical_loss": 4.309609823785298,
      "tokens_seen": 229769216
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009397592778335005,
      "loss": 1.8685,
      "theoretical_loss": 4.309460327693737,
      "tokens_seen": 229834752
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009397392176529589,
      "loss": 1.9508,
      "theoretical_loss": 4.3093108861559495,
      "tokens_seen": 229900288
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009397191574724174,
      "loss": 1.839,
      "theoretical_loss": 4.3091614991364855,
      "tokens_seen": 229965824
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009396990972918756,
      "loss": 1.8797,
      "theoretical_loss": 4.309012166599929,
      "tokens_seen": 230031360
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009396790371113341,
      "loss": 1.8448,
      "theoretical_loss": 4.308862888510893,
      "tokens_seen": 230096896
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009396589769307923,
      "loss": 1.7767,
      "theoretical_loss": 4.308713664834029,
      "tokens_seen": 230162432
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009396389167502508,
      "loss": 1.6558,
      "theoretical_loss": 4.308564495534017,
      "tokens_seen": 230227968
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009396188565697092,
      "loss": 1.8051,
      "theoretical_loss": 4.308415380575573,
      "tokens_seen": 230293504
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009395987963891675,
      "loss": 1.71,
      "theoretical_loss": 4.308266319923445,
      "tokens_seen": 230359040
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009395787362086259,
      "loss": 1.669,
      "theoretical_loss": 4.308117313542413,
      "tokens_seen": 230424576
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009395586760280843,
      "loss": 1.7442,
      "theoretical_loss": 4.3079683613972914,
      "tokens_seen": 230490112
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009395386158475426,
      "loss": 2.0136,
      "theoretical_loss": 4.307819463452925,
      "tokens_seen": 230555648
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009395185556670011,
      "loss": 1.7937,
      "theoretical_loss": 4.3076706196741945,
      "tokens_seen": 230621184
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009394984954864594,
      "loss": 1.8568,
      "theoretical_loss": 4.30752183002601,
      "tokens_seen": 230686720
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009394784353059178,
      "loss": 1.7888,
      "theoretical_loss": 4.307373094473318,
      "tokens_seen": 230752256
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009394583751253761,
      "loss": 1.758,
      "theoretical_loss": 4.307224412981095,
      "tokens_seen": 230817792
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009394383149448345,
      "loss": 1.7212,
      "theoretical_loss": 4.3070757855143516,
      "tokens_seen": 230883328
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009394182547642929,
      "loss": 1.8631,
      "theoretical_loss": 4.3069272120381275,
      "tokens_seen": 230948864
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009393981945837513,
      "loss": 1.7568,
      "theoretical_loss": 4.3067786925175,
      "tokens_seen": 231014400
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009393781344032096,
      "loss": 1.8251,
      "theoretical_loss": 4.3066302269175765,
      "tokens_seen": 231079936
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009393580742226681,
      "loss": 1.9332,
      "theoretical_loss": 4.306481815203496,
      "tokens_seen": 231145472
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009393380140421264,
      "loss": 1.8374,
      "theoretical_loss": 4.30633345734043,
      "tokens_seen": 231211008
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009393179538615848,
      "loss": 1.8336,
      "theoretical_loss": 4.306185153293584,
      "tokens_seen": 231276544
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009392978936810431,
      "loss": 1.6981,
      "theoretical_loss": 4.306036903028196,
      "tokens_seen": 231342080
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009392778335005016,
      "loss": 1.7224,
      "theoretical_loss": 4.305888706509534,
      "tokens_seen": 231407616
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009392577733199599,
      "loss": 1.9788,
      "theoretical_loss": 4.3057405637029,
      "tokens_seen": 231473152
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009392377131394183,
      "loss": 1.7976,
      "theoretical_loss": 4.305592474573627,
      "tokens_seen": 231538688
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009392176529588766,
      "loss": 1.9443,
      "theoretical_loss": 4.305444439087082,
      "tokens_seen": 231604224
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000939197592778335,
      "loss": 1.7595,
      "theoretical_loss": 4.305296457208662,
      "tokens_seen": 231669760
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009391775325977934,
      "loss": 1.8406,
      "theoretical_loss": 4.305148528903798,
      "tokens_seen": 231735296
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009391574724172518,
      "loss": 1.9642,
      "theoretical_loss": 4.3050006541379515,
      "tokens_seen": 231800832
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009391374122367101,
      "loss": 1.7075,
      "theoretical_loss": 4.304852832876618,
      "tokens_seen": 231866368
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009391173520561686,
      "loss": 1.7245,
      "theoretical_loss": 4.304705065085322,
      "tokens_seen": 231931904
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009390972918756269,
      "loss": 1.972,
      "theoretical_loss": 4.304557350729623,
      "tokens_seen": 231997440
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009390772316950853,
      "loss": 1.9232,
      "theoretical_loss": 4.304409689775111,
      "tokens_seen": 232062976
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009390571715145436,
      "loss": 1.8148,
      "theoretical_loss": 4.304262082187407,
      "tokens_seen": 232128512
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000939037111334002,
      "loss": 1.6574,
      "theoretical_loss": 4.304114527932166,
      "tokens_seen": 232194048
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009390170511534604,
      "loss": 1.8246,
      "theoretical_loss": 4.303967026975072,
      "tokens_seen": 232259584
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009389969909729188,
      "loss": 1.7932,
      "theoretical_loss": 4.303819579281844,
      "tokens_seen": 232325120
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009389769307923771,
      "loss": 1.8462,
      "theoretical_loss": 4.30367218481823,
      "tokens_seen": 232390656
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009389568706118356,
      "loss": 1.8204,
      "theoretical_loss": 4.30352484355001,
      "tokens_seen": 232456192
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009389368104312938,
      "loss": 1.9215,
      "theoretical_loss": 4.303377555442998,
      "tokens_seen": 232521728
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009389167502507523,
      "loss": 1.7366,
      "theoretical_loss": 4.3032303204630376,
      "tokens_seen": 232587264
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.49400627613067627,
      "objective/train/docs_used": 91258,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6414120197296143,
      "objective/train/original_loss": 1.6414121389389038,
      "objective/train/theoretical_loss": 4.303083138576003,
      "objective/train/tokens_used": 253112800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24473948776721954,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506447553634644,
      "objective/train/weighted_lm_loss": 1.7243413925170898,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9048373699188232,
      "theoretical_loss": 4.303083138576003,
      "tokens_seen": 232652800
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009388966900702107,
      "loss": 1.8242,
      "theoretical_loss": 4.303083138576003,
      "tokens_seen": 232652800
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000938876629889669,
      "loss": 1.7083,
      "theoretical_loss": 4.302936009747803,
      "tokens_seen": 232718336
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009388565697091274,
      "loss": 1.7864,
      "theoretical_loss": 4.302788933944375,
      "tokens_seen": 232783872
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009388365095285857,
      "loss": 1.8209,
      "theoretical_loss": 4.302641911131689,
      "tokens_seen": 232849408
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009388164493480441,
      "loss": 1.8624,
      "theoretical_loss": 4.3024949412757465,
      "tokens_seen": 232914944
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009387963891675026,
      "loss": 1.7925,
      "theoretical_loss": 4.302348024342581,
      "tokens_seen": 232980480
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009387763289869608,
      "loss": 1.8408,
      "theoretical_loss": 4.302201160298255,
      "tokens_seen": 233046016
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009387562688064193,
      "loss": 1.8639,
      "theoretical_loss": 4.302054349108865,
      "tokens_seen": 233111552
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009387362086258777,
      "loss": 1.7454,
      "theoretical_loss": 4.301907590740538,
      "tokens_seen": 233177088
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000938716148445336,
      "loss": 1.7967,
      "theoretical_loss": 4.301760885159431,
      "tokens_seen": 233242624
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009386960882647944,
      "loss": 1.8907,
      "theoretical_loss": 4.301614232331733,
      "tokens_seen": 233308160
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009386760280842527,
      "loss": 1.817,
      "theoretical_loss": 4.301467632223664,
      "tokens_seen": 233373696
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009386559679037111,
      "loss": 1.9417,
      "theoretical_loss": 4.301321084801478,
      "tokens_seen": 233439232
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009386359077231696,
      "loss": 1.744,
      "theoretical_loss": 4.301174590031453,
      "tokens_seen": 233504768
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009386158475426278,
      "loss": 1.7738,
      "theoretical_loss": 4.301028147879904,
      "tokens_seen": 233570304
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009385957873620863,
      "loss": 1.6949,
      "theoretical_loss": 4.300881758313178,
      "tokens_seen": 233635840
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009385757271815446,
      "loss": 1.7507,
      "theoretical_loss": 4.300735421297647,
      "tokens_seen": 233701376
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000938555667001003,
      "loss": 1.7811,
      "theoretical_loss": 4.300589136799719,
      "tokens_seen": 233766912
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009385356068204614,
      "loss": 1.8286,
      "theoretical_loss": 4.300442904785831,
      "tokens_seen": 233832448
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009385155466399198,
      "loss": 1.7496,
      "theoretical_loss": 4.300296725222449,
      "tokens_seen": 233897984
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009384954864593781,
      "loss": 1.8731,
      "theoretical_loss": 4.300150598076074,
      "tokens_seen": 233963520
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009384754262788365,
      "loss": 1.8883,
      "theoretical_loss": 4.300004523313237,
      "tokens_seen": 234029056
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009384553660982948,
      "loss": 1.8071,
      "theoretical_loss": 4.299858500900495,
      "tokens_seen": 234094592
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009384353059177533,
      "loss": 1.9614,
      "theoretical_loss": 4.299712530804441,
      "tokens_seen": 234160128
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009384152457372117,
      "loss": 1.6487,
      "theoretical_loss": 4.299566612991695,
      "tokens_seen": 234225664
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.00093839518555667,
      "loss": 1.8937,
      "theoretical_loss": 4.299420747428911,
      "tokens_seen": 234291200
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009383751253761285,
      "loss": 1.867,
      "theoretical_loss": 4.2992749340827725,
      "tokens_seen": 234356736
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009383550651955868,
      "loss": 1.828,
      "theoretical_loss": 4.299129172919991,
      "tokens_seen": 234422272
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009383350050150452,
      "loss": 1.8159,
      "theoretical_loss": 4.2989834639073115,
      "tokens_seen": 234487808
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009383149448345035,
      "loss": 1.8431,
      "theoretical_loss": 4.298837807011508,
      "tokens_seen": 234553344
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000938294884653962,
      "loss": 1.955,
      "theoretical_loss": 4.298692202199386,
      "tokens_seen": 234618880
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009382748244734203,
      "loss": 1.9081,
      "theoretical_loss": 4.29854664943778,
      "tokens_seen": 234684416
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009382547642928787,
      "loss": 1.7294,
      "theoretical_loss": 4.298401148693557,
      "tokens_seen": 234749952
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000938234704112337,
      "loss": 1.8228,
      "theoretical_loss": 4.298255699933612,
      "tokens_seen": 234815488
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009382146439317954,
      "loss": 1.8377,
      "theoretical_loss": 4.298110303124871,
      "tokens_seen": 234881024
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009381945837512538,
      "loss": 1.9015,
      "theoretical_loss": 4.297964958234292,
      "tokens_seen": 234946560
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009381745235707122,
      "loss": 1.8647,
      "theoretical_loss": 4.297819665228861,
      "tokens_seen": 235012096
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009381544633901705,
      "loss": 1.8087,
      "theoretical_loss": 4.297674424075594,
      "tokens_seen": 235077632
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000938134403209629,
      "loss": 1.812,
      "theoretical_loss": 4.29752923474154,
      "tokens_seen": 235143168
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009381143430290873,
      "loss": 1.823,
      "theoretical_loss": 4.297384097193775,
      "tokens_seen": 235208704
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009380942828485457,
      "loss": 1.8137,
      "theoretical_loss": 4.297239011399407,
      "tokens_seen": 235274240
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000938074222668004,
      "loss": 1.8754,
      "theoretical_loss": 4.297093977325573,
      "tokens_seen": 235339776
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009380541624874624,
      "loss": 1.7469,
      "theoretical_loss": 4.29694899493944,
      "tokens_seen": 235405312
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009380341023069208,
      "loss": 1.7643,
      "theoretical_loss": 4.296804064208207,
      "tokens_seen": 235470848
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009380140421263792,
      "loss": 1.8098,
      "theoretical_loss": 4.296659185099099,
      "tokens_seen": 235536384
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009379939819458375,
      "loss": 1.8432,
      "theoretical_loss": 4.296514357579375,
      "tokens_seen": 235601920
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000937973921765296,
      "loss": 1.9429,
      "theoretical_loss": 4.2963695816163225,
      "tokens_seen": 235667456
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009379538615847542,
      "loss": 1.7737,
      "theoretical_loss": 4.296224857177257,
      "tokens_seen": 235732992
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009379338014042127,
      "loss": 1.8518,
      "theoretical_loss": 4.296080184229526,
      "tokens_seen": 235798528
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009379137412236711,
      "loss": 1.7253,
      "theoretical_loss": 4.295935562740505,
      "tokens_seen": 235864064
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.4898257851600647,
      "objective/train/docs_used": 92471,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8431349992752075,
      "objective/train/original_loss": 1.843135118484497,
      "objective/train/theoretical_loss": 4.295790992677603,
      "objective/train/tokens_used": 256389600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24101541936397552,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502077341079712,
      "objective/train/weighted_lm_loss": 1.9356428384780884,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.295790992677603,
      "tokens_seen": 235929600
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009378936810431294,
      "loss": 1.856,
      "theoretical_loss": 4.295790992677603,
      "tokens_seen": 235929600
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009378736208625878,
      "loss": 1.8341,
      "theoretical_loss": 4.295646474008252,
      "tokens_seen": 235995136
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009378535606820461,
      "loss": 1.9499,
      "theoretical_loss": 4.29550200669992,
      "tokens_seen": 236060672
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009378335005015045,
      "loss": 1.7604,
      "theoretical_loss": 4.295357590720101,
      "tokens_seen": 236126208
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000937813440320963,
      "loss": 1.7573,
      "theoretical_loss": 4.2952132260363225,
      "tokens_seen": 236191744
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009377933801404212,
      "loss": 1.8687,
      "theoretical_loss": 4.295068912616135,
      "tokens_seen": 236257280
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009377733199598797,
      "loss": 1.7367,
      "theoretical_loss": 4.294924650427125,
      "tokens_seen": 236322816
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009377532597793381,
      "loss": 1.7755,
      "theoretical_loss": 4.294780439436904,
      "tokens_seen": 236388352
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009377331995987964,
      "loss": 1.7929,
      "theoretical_loss": 4.294636279613117,
      "tokens_seen": 236453888
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009377131394182548,
      "loss": 1.8914,
      "theoretical_loss": 4.2944921709234345,
      "tokens_seen": 236519424
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009376930792377131,
      "loss": 1.897,
      "theoretical_loss": 4.294348113335559,
      "tokens_seen": 236584960
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009376730190571715,
      "loss": 2.0073,
      "theoretical_loss": 4.29420410681722,
      "tokens_seen": 236650496
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.00093765295887663,
      "loss": 1.7469,
      "theoretical_loss": 4.294060151336178,
      "tokens_seen": 236716032
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009376328986960882,
      "loss": 1.8601,
      "theoretical_loss": 4.293916246860224,
      "tokens_seen": 236781568
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009376128385155467,
      "loss": 1.8224,
      "theoretical_loss": 4.293772393357176,
      "tokens_seen": 236847104
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000937592778335005,
      "loss": 1.9164,
      "theoretical_loss": 4.2936285907948815,
      "tokens_seen": 236912640
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009375727181544634,
      "loss": 1.8051,
      "theoretical_loss": 4.293484839141217,
      "tokens_seen": 236978176
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009375526579739218,
      "loss": 1.7483,
      "theoretical_loss": 4.293341138364091,
      "tokens_seen": 237043712
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009375325977933802,
      "loss": 1.7834,
      "theoretical_loss": 4.293197488431435,
      "tokens_seen": 237109248
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009375125376128385,
      "loss": 1.8088,
      "theoretical_loss": 4.293053889311217,
      "tokens_seen": 237174784
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009374924774322969,
      "loss": 1.8689,
      "theoretical_loss": 4.29291034097143,
      "tokens_seen": 237240320
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009374724172517552,
      "loss": 1.7728,
      "theoretical_loss": 4.292766843380094,
      "tokens_seen": 237305856
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009374523570712137,
      "loss": 1.7375,
      "theoretical_loss": 4.292623396505262,
      "tokens_seen": 237371392
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000937432296890672,
      "loss": 1.7408,
      "theoretical_loss": 4.292480000315014,
      "tokens_seen": 237436928
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009374122367101304,
      "loss": 1.7242,
      "theoretical_loss": 4.2923366547774595,
      "tokens_seen": 237502464
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009373921765295888,
      "loss": 1.7053,
      "theoretical_loss": 4.292193359860737,
      "tokens_seen": 237568000
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009373721163490472,
      "loss": 1.7752,
      "theoretical_loss": 4.292050115533012,
      "tokens_seen": 237633536
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009373520561685055,
      "loss": 1.7755,
      "theoretical_loss": 4.291906921762481,
      "tokens_seen": 237699072
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009373319959879639,
      "loss": 1.9598,
      "theoretical_loss": 4.2917637785173675,
      "tokens_seen": 237764608
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009373119358074222,
      "loss": 1.7152,
      "theoretical_loss": 4.291620685765926,
      "tokens_seen": 237830144
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009372918756268807,
      "loss": 1.884,
      "theoretical_loss": 4.291477643476438,
      "tokens_seen": 237895680
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000937271815446339,
      "loss": 1.8008,
      "theoretical_loss": 4.291334651617213,
      "tokens_seen": 237961216
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009372517552657974,
      "loss": 1.9097,
      "theoretical_loss": 4.291191710156591,
      "tokens_seen": 238026752
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009372316950852557,
      "loss": 1.8682,
      "theoretical_loss": 4.291048819062939,
      "tokens_seen": 238092288
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009372116349047142,
      "loss": 1.7768,
      "theoretical_loss": 4.290905978304655,
      "tokens_seen": 238157824
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009371915747241725,
      "loss": 1.7492,
      "theoretical_loss": 4.290763187850161,
      "tokens_seen": 238223360
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009371715145436309,
      "loss": 1.8498,
      "theoretical_loss": 4.290620447667912,
      "tokens_seen": 238288896
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009371514543630893,
      "loss": 1.7209,
      "theoretical_loss": 4.29047775772639,
      "tokens_seen": 238354432
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009371313941825476,
      "loss": 1.7331,
      "theoretical_loss": 4.290335117994104,
      "tokens_seen": 238419968
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000937111334002006,
      "loss": 1.738,
      "theoretical_loss": 4.290192528439594,
      "tokens_seen": 238485504
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009370912738214644,
      "loss": 1.9051,
      "theoretical_loss": 4.290049989031424,
      "tokens_seen": 238551040
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009370712136409227,
      "loss": 1.9187,
      "theoretical_loss": 4.289907499738192,
      "tokens_seen": 238616576
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009370511534603812,
      "loss": 1.8264,
      "theoretical_loss": 4.289765060528521,
      "tokens_seen": 238682112
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009370310932798395,
      "loss": 1.9482,
      "theoretical_loss": 4.2896226713710615,
      "tokens_seen": 238747648
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009370110330992979,
      "loss": 1.8173,
      "theoretical_loss": 4.289480332234493,
      "tokens_seen": 238813184
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009369909729187563,
      "loss": 1.7753,
      "theoretical_loss": 4.289338043087527,
      "tokens_seen": 238878720
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009369709127382146,
      "loss": 1.7691,
      "theoretical_loss": 4.289195803898897,
      "tokens_seen": 238944256
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000936950852557673,
      "loss": 1.9161,
      "theoretical_loss": 4.289053614637368,
      "tokens_seen": 239009792
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009369307923771315,
      "loss": 1.8554,
      "theoretical_loss": 4.288911475271731,
      "tokens_seen": 239075328
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009369107321965898,
      "loss": 1.8071,
      "theoretical_loss": 4.288769385770809,
      "tokens_seen": 239140864
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.49386152625083923,
      "objective/train/docs_used": 93570,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8087902069091797,
      "objective/train/original_loss": 1.8087904453277588,
      "objective/train/theoretical_loss": 4.28862734610345,
      "objective/train/tokens_used": 259666400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24422965943813324,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506278276443481,
      "objective/train/weighted_lm_loss": 1.9002867937088013,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.28862734610345,
      "tokens_seen": 239206400
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009368906720160482,
      "loss": 1.8793,
      "theoretical_loss": 4.28862734610345,
      "tokens_seen": 239206400
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009368706118355065,
      "loss": 1.8012,
      "theoretical_loss": 4.288485356238529,
      "tokens_seen": 239271936
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009368505516549649,
      "loss": 1.7934,
      "theoretical_loss": 4.288343416144952,
      "tokens_seen": 239337472
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009368304914744234,
      "loss": 1.9502,
      "theoretical_loss": 4.288201525791651,
      "tokens_seen": 239403008
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009368104312938816,
      "loss": 1.8676,
      "theoretical_loss": 4.288059685147586,
      "tokens_seen": 239468544
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009367903711133401,
      "loss": 1.8661,
      "theoretical_loss": 4.287917894181746,
      "tokens_seen": 239534080
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009367703109327985,
      "loss": 1.7173,
      "theoretical_loss": 4.287776152863146,
      "tokens_seen": 239599616
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009367502507522568,
      "loss": 1.8081,
      "theoretical_loss": 4.28763446116083,
      "tokens_seen": 239665152
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009367301905717152,
      "loss": 1.8705,
      "theoretical_loss": 4.28749281904387,
      "tokens_seen": 239730688
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009367101303911735,
      "loss": 1.8234,
      "theoretical_loss": 4.287351226481366,
      "tokens_seen": 239796224
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009366900702106319,
      "loss": 1.8895,
      "theoretical_loss": 4.287209683442444,
      "tokens_seen": 239861760
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009366700100300904,
      "loss": 1.8936,
      "theoretical_loss": 4.2870681898962575,
      "tokens_seen": 239927296
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009366499498495486,
      "loss": 1.8068,
      "theoretical_loss": 4.286926745811991,
      "tokens_seen": 239992832
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009366298896690071,
      "loss": 1.7208,
      "theoretical_loss": 4.286785351158852,
      "tokens_seen": 240058368
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009366098294884654,
      "loss": 1.8455,
      "theoretical_loss": 4.286644005906081,
      "tokens_seen": 240123904
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009365897693079238,
      "loss": 1.8656,
      "theoretical_loss": 4.2865027100229405,
      "tokens_seen": 240189440
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009365697091273822,
      "loss": 1.9201,
      "theoretical_loss": 4.286361463478723,
      "tokens_seen": 240254976
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009365496489468406,
      "loss": 1.8351,
      "theoretical_loss": 4.28622026624275,
      "tokens_seen": 240320512
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009365295887662989,
      "loss": 1.7223,
      "theoretical_loss": 4.286079118284368,
      "tokens_seen": 240386048
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009365095285857573,
      "loss": 1.7905,
      "theoretical_loss": 4.285938019572951,
      "tokens_seen": 240451584
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009364894684052156,
      "loss": 1.922,
      "theoretical_loss": 4.285796970077903,
      "tokens_seen": 240517120
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009364694082246741,
      "loss": 1.8986,
      "theoretical_loss": 4.285655969768651,
      "tokens_seen": 240582656
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009364493480441324,
      "loss": 1.9152,
      "theoretical_loss": 4.285515018614655,
      "tokens_seen": 240648192
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009364292878635908,
      "loss": 1.7653,
      "theoretical_loss": 4.2853741165853965,
      "tokens_seen": 240713728
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009364092276830492,
      "loss": 1.6739,
      "theoretical_loss": 4.285233263650387,
      "tokens_seen": 240779264
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009363891675025076,
      "loss": 1.7319,
      "theoretical_loss": 4.285092459779166,
      "tokens_seen": 240844800
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009363691073219659,
      "loss": 1.7374,
      "theoretical_loss": 4.2849517049412995,
      "tokens_seen": 240910336
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009363490471414243,
      "loss": 1.7947,
      "theoretical_loss": 4.28481099910638,
      "tokens_seen": 240975872
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009363289869608826,
      "loss": 1.7014,
      "theoretical_loss": 4.284670342244027,
      "tokens_seen": 241041408
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009363089267803411,
      "loss": 1.8506,
      "theoretical_loss": 4.284529734323887,
      "tokens_seen": 241106944
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009362888665997994,
      "loss": 1.7271,
      "theoretical_loss": 4.284389175315636,
      "tokens_seen": 241172480
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009362688064192578,
      "loss": 1.7538,
      "theoretical_loss": 4.284248665188974,
      "tokens_seen": 241238016
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009362487462387161,
      "loss": 1.7851,
      "theoretical_loss": 4.284108203913629,
      "tokens_seen": 241303552
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009362286860581746,
      "loss": 1.8025,
      "theoretical_loss": 4.283967791459356,
      "tokens_seen": 241369088
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009362086258776329,
      "loss": 1.8707,
      "theoretical_loss": 4.283827427795939,
      "tokens_seen": 241434624
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009361885656970913,
      "loss": 1.8208,
      "theoretical_loss": 4.283687112893184,
      "tokens_seen": 241500160
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009361685055165497,
      "loss": 1.7022,
      "theoretical_loss": 4.283546846720929,
      "tokens_seen": 241565696
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000936148445336008,
      "loss": 1.798,
      "theoretical_loss": 4.283406629249036,
      "tokens_seen": 241631232
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009361283851554664,
      "loss": 1.7899,
      "theoretical_loss": 4.283266460447394,
      "tokens_seen": 241696768
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009361083249749248,
      "loss": 1.7569,
      "theoretical_loss": 4.28312634028592,
      "tokens_seen": 241762304
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009360882647943831,
      "loss": 1.7866,
      "theoretical_loss": 4.282986268734556,
      "tokens_seen": 241827840
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009360682046138416,
      "loss": 1.6588,
      "theoretical_loss": 4.282846245763273,
      "tokens_seen": 241893376
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009360481444332999,
      "loss": 1.7699,
      "theoretical_loss": 4.282706271342066,
      "tokens_seen": 241958912
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009360280842527583,
      "loss": 1.8742,
      "theoretical_loss": 4.2825663454409595,
      "tokens_seen": 242024448
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009360080240722167,
      "loss": 1.8847,
      "theoretical_loss": 4.2824264680300015,
      "tokens_seen": 242089984
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000935987963891675,
      "loss": 1.9179,
      "theoretical_loss": 4.2822866390792695,
      "tokens_seen": 242155520
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009359679037111334,
      "loss": 1.8009,
      "theoretical_loss": 4.282146858558866,
      "tokens_seen": 242221056
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009359478435305919,
      "loss": 1.7736,
      "theoretical_loss": 4.28200712643892,
      "tokens_seen": 242286592
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009359277833500501,
      "loss": 1.627,
      "theoretical_loss": 4.281867442689587,
      "tokens_seen": 242352128
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009359077231695086,
      "loss": 1.87,
      "theoretical_loss": 4.281727807281051,
      "tokens_seen": 242417664
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.49168694019317627,
      "objective/train/docs_used": 94878,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7456352710723877,
      "objective/train/original_loss": 1.7456352710723877,
      "objective/train/theoretical_loss": 4.281588220183519,
      "objective/train/tokens_used": 262943200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24250559508800507,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504014492034912,
      "objective/train/weighted_lm_loss": 1.8333992958068848,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.281588220183519,
      "tokens_seen": 242483200
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009358876629889668,
      "loss": 1.7932,
      "theoretical_loss": 4.281588220183519,
      "tokens_seen": 242483200
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009358676028084253,
      "loss": 1.8316,
      "theoretical_loss": 4.281448681367226,
      "tokens_seen": 242548736
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009358475426278837,
      "loss": 1.7968,
      "theoretical_loss": 4.2813091908024345,
      "tokens_seen": 242614272
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000935827482447342,
      "loss": 1.7241,
      "theoretical_loss": 4.281169748459432,
      "tokens_seen": 242679808
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009358074222668004,
      "loss": 1.8417,
      "theoretical_loss": 4.281030354308533,
      "tokens_seen": 242745344
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009357873620862589,
      "loss": 1.811,
      "theoretical_loss": 4.2808910083200775,
      "tokens_seen": 242810880
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009357673019057171,
      "loss": 1.7931,
      "theoretical_loss": 4.280751710464432,
      "tokens_seen": 242876416
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009357472417251756,
      "loss": 1.7608,
      "theoretical_loss": 4.280612460711989,
      "tokens_seen": 242941952
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009357271815446338,
      "loss": 1.8087,
      "theoretical_loss": 4.280473259033169,
      "tokens_seen": 243007488
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009357071213640923,
      "loss": 1.8108,
      "theoretical_loss": 4.280334105398417,
      "tokens_seen": 243073024
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009356870611835507,
      "loss": 1.7234,
      "theoretical_loss": 4.280194999778203,
      "tokens_seen": 243138560
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000935667001003009,
      "loss": 1.7381,
      "theoretical_loss": 4.280055942143027,
      "tokens_seen": 243204096
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009356469408224674,
      "loss": 1.7984,
      "theoretical_loss": 4.27991693246341,
      "tokens_seen": 243269632
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009356268806419258,
      "loss": 1.8265,
      "theoretical_loss": 4.279777970709902,
      "tokens_seen": 243335168
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009356068204613841,
      "loss": 1.8135,
      "theoretical_loss": 4.27963905685308,
      "tokens_seen": 243400704
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009355867602808426,
      "loss": 1.7835,
      "theoretical_loss": 4.279500190863544,
      "tokens_seen": 243466240
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009355667001003008,
      "loss": 1.7395,
      "theoretical_loss": 4.279361372711923,
      "tokens_seen": 243531776
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009355466399197593,
      "loss": 1.899,
      "theoretical_loss": 4.27922260236887,
      "tokens_seen": 243597312
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009355265797392176,
      "loss": 1.8655,
      "theoretical_loss": 4.279083879805064,
      "tokens_seen": 243662848
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000935506519558676,
      "loss": 1.8957,
      "theoretical_loss": 4.278945204991212,
      "tokens_seen": 243728384
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009354864593781344,
      "loss": 1.8112,
      "theoretical_loss": 4.278806577898042,
      "tokens_seen": 243793920
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009354663991975928,
      "loss": 1.7751,
      "theoretical_loss": 4.278667998496313,
      "tokens_seen": 243859456
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009354463390170511,
      "loss": 1.8183,
      "theoretical_loss": 4.278529466756806,
      "tokens_seen": 243924992
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009354262788365096,
      "loss": 1.7717,
      "theoretical_loss": 4.278390982650333,
      "tokens_seen": 243990528
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000935406218655968,
      "loss": 1.7982,
      "theoretical_loss": 4.278252546147724,
      "tokens_seen": 244056064
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009353861584754263,
      "loss": 1.9031,
      "theoretical_loss": 4.278114157219839,
      "tokens_seen": 244121600
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009353660982948847,
      "loss": 1.7776,
      "theoretical_loss": 4.277975815837567,
      "tokens_seen": 244187136
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000935346038114343,
      "loss": 1.8307,
      "theoretical_loss": 4.277837521971816,
      "tokens_seen": 244252672
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009353259779338015,
      "loss": 1.7035,
      "theoretical_loss": 4.277699275593523,
      "tokens_seen": 244318208
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009353059177532598,
      "loss": 1.6069,
      "theoretical_loss": 4.277561076673652,
      "tokens_seen": 244383744
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009352858575727182,
      "loss": 1.7675,
      "theoretical_loss": 4.277422925183188,
      "tokens_seen": 244449280
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009352657973921765,
      "loss": 1.7285,
      "theoretical_loss": 4.277284821093147,
      "tokens_seen": 244514816
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000935245737211635,
      "loss": 1.7922,
      "theoretical_loss": 4.277146764374566,
      "tokens_seen": 244580352
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009352256770310933,
      "loss": 1.7639,
      "theoretical_loss": 4.27700875499851,
      "tokens_seen": 244645888
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009352056168505517,
      "loss": 1.8802,
      "theoretical_loss": 4.2768707929360685,
      "tokens_seen": 244711424
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.00093518555667001,
      "loss": 1.7328,
      "theoretical_loss": 4.276732878158356,
      "tokens_seen": 244776960
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009351654964894684,
      "loss": 1.8951,
      "theoretical_loss": 4.276595010636514,
      "tokens_seen": 244842496
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009351454363089268,
      "loss": 1.7935,
      "theoretical_loss": 4.276457190341707,
      "tokens_seen": 244908032
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009351253761283852,
      "loss": 1.8199,
      "theoretical_loss": 4.2763194172451255,
      "tokens_seen": 244973568
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009351053159478435,
      "loss": 1.7277,
      "theoretical_loss": 4.276181691317987,
      "tokens_seen": 245039104
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000935085255767302,
      "loss": 1.8886,
      "theoretical_loss": 4.276044012531534,
      "tokens_seen": 245104640
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009350651955867603,
      "loss": 2.0016,
      "theoretical_loss": 4.275906380857031,
      "tokens_seen": 245170176
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009350451354062187,
      "loss": 1.7677,
      "theoretical_loss": 4.275768796265771,
      "tokens_seen": 245235712
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009350250752256771,
      "loss": 1.7328,
      "theoretical_loss": 4.275631258729071,
      "tokens_seen": 245301248
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009350050150451354,
      "loss": 1.83,
      "theoretical_loss": 4.275493768218274,
      "tokens_seen": 245366784
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009349849548645938,
      "loss": 1.7851,
      "theoretical_loss": 4.275356324704745,
      "tokens_seen": 245432320
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009349648946840523,
      "loss": 1.8365,
      "theoretical_loss": 4.27521892815988,
      "tokens_seen": 245497856
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009349448345035105,
      "loss": 1.6734,
      "theoretical_loss": 4.275081578555094,
      "tokens_seen": 245563392
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000934924774322969,
      "loss": 1.7928,
      "theoretical_loss": 4.274944275861828,
      "tokens_seen": 245628928
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009349047141424272,
      "loss": 1.7817,
      "theoretical_loss": 4.274807020051552,
      "tokens_seen": 245694464
    },
    {
      "epoch": 0.07,
      "objective/train/advantage_avg": 0.4900068938732147,
      "objective/train/docs_used": 96014,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7985907793045044,
      "objective/train/original_loss": 1.798590898513794,
      "objective/train/theoretical_loss": 4.274669811095759,
      "objective/train/tokens_used": 266220000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24235199391841888,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502325296401978,
      "objective/train/weighted_lm_loss": 1.888869047164917,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.274669811095759,
      "tokens_seen": 245760000
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009348846539618857,
      "loss": 1.7632,
      "theoretical_loss": 4.274669811095759,
      "tokens_seen": 245760000
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009348645937813441,
      "loss": 1.7704,
      "theoretical_loss": 4.274532648965964,
      "tokens_seen": 245825536
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009348445336008024,
      "loss": 1.7771,
      "theoretical_loss": 4.274395533633712,
      "tokens_seen": 245891072
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009348244734202608,
      "loss": 1.7508,
      "theoretical_loss": 4.274258465070567,
      "tokens_seen": 245956608
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009348044132397193,
      "loss": 1.8499,
      "theoretical_loss": 4.274121443248122,
      "tokens_seen": 246022144
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009347843530591775,
      "loss": 1.7992,
      "theoretical_loss": 4.273984468137995,
      "tokens_seen": 246087680
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000934764292878636,
      "loss": 1.5857,
      "theoretical_loss": 4.273847539711825,
      "tokens_seen": 246153216
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009347442326980942,
      "loss": 1.8579,
      "theoretical_loss": 4.273710657941282,
      "tokens_seen": 246218752
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009347241725175527,
      "loss": 1.8407,
      "theoretical_loss": 4.273573822798054,
      "tokens_seen": 246284288
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009347041123370111,
      "loss": 1.839,
      "theoretical_loss": 4.273437034253859,
      "tokens_seen": 246349824
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009346840521564694,
      "loss": 1.8247,
      "theoretical_loss": 4.273300292280435,
      "tokens_seen": 246415360
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009346639919759278,
      "loss": 1.8677,
      "theoretical_loss": 4.273163596849549,
      "tokens_seen": 246480896
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009346439317953862,
      "loss": 1.8076,
      "theoretical_loss": 4.27302694793299,
      "tokens_seen": 246546432
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009346238716148445,
      "loss": 1.748,
      "theoretical_loss": 4.272890345502573,
      "tokens_seen": 246611968
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000934603811434303,
      "loss": 1.8006,
      "theoretical_loss": 4.272753789530134,
      "tokens_seen": 246677504
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009345837512537612,
      "loss": 1.6839,
      "theoretical_loss": 4.272617279987541,
      "tokens_seen": 246743040
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009345636910732197,
      "loss": 1.6944,
      "theoretical_loss": 4.272480816846678,
      "tokens_seen": 246808576
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000934543630892678,
      "loss": 1.7108,
      "theoretical_loss": 4.27234440007946,
      "tokens_seen": 246874112
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009345235707121364,
      "loss": 1.8806,
      "theoretical_loss": 4.272208029657822,
      "tokens_seen": 246939648
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009345035105315948,
      "loss": 1.8091,
      "theoretical_loss": 4.272071705553726,
      "tokens_seen": 247005184
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009344834503510532,
      "loss": 1.7406,
      "theoretical_loss": 4.271935427739158,
      "tokens_seen": 247070720
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009344633901705115,
      "loss": 1.7875,
      "theoretical_loss": 4.2717991961861275,
      "tokens_seen": 247136256
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.00093444332998997,
      "loss": 1.7592,
      "theoretical_loss": 4.271663010866669,
      "tokens_seen": 247201792
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009344232698094283,
      "loss": 1.9044,
      "theoretical_loss": 4.27152687175284,
      "tokens_seen": 247267328
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009344032096288867,
      "loss": 1.8186,
      "theoretical_loss": 4.271390778816727,
      "tokens_seen": 247332864
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.000934383149448345,
      "loss": 1.8104,
      "theoretical_loss": 4.2712547320304335,
      "tokens_seen": 247398400
    },
    {
      "epoch": 0.07,
      "learning_rate": 0.0009343630892678034,
      "loss": 1.7732,
      "theoretical_loss": 4.2711187313660925,
      "tokens_seen": 247463936
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009343430290872618,
      "loss": 1.7146,
      "theoretical_loss": 4.27098277679586,
      "tokens_seen": 247529472
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009343229689067202,
      "loss": 1.8197,
      "theoretical_loss": 4.270846868291914,
      "tokens_seen": 247595008
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009343029087261785,
      "loss": 1.8096,
      "theoretical_loss": 4.270711005826461,
      "tokens_seen": 247660544
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009342828485456369,
      "loss": 1.8219,
      "theoretical_loss": 4.270575189371727,
      "tokens_seen": 247726080
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009342627883650953,
      "loss": 1.8408,
      "theoretical_loss": 4.270439418899966,
      "tokens_seen": 247791616
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009342427281845537,
      "loss": 1.772,
      "theoretical_loss": 4.270303694383453,
      "tokens_seen": 247857152
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000934222668004012,
      "loss": 1.7264,
      "theoretical_loss": 4.270168015794489,
      "tokens_seen": 247922688
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009342026078234705,
      "loss": 1.8814,
      "theoretical_loss": 4.270032383105398,
      "tokens_seen": 247988224
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009341825476429287,
      "loss": 1.7944,
      "theoretical_loss": 4.269896796288528,
      "tokens_seen": 248053760
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009341624874623872,
      "loss": 1.7638,
      "theoretical_loss": 4.269761255316252,
      "tokens_seen": 248119296
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009341424272818455,
      "loss": 1.7488,
      "theoretical_loss": 4.269625760160965,
      "tokens_seen": 248184832
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009341223671013039,
      "loss": 1.7874,
      "theoretical_loss": 4.269490310795089,
      "tokens_seen": 248250368
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009341023069207623,
      "loss": 1.899,
      "theoretical_loss": 4.269354907191065,
      "tokens_seen": 248315904
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009340822467402207,
      "loss": 1.9713,
      "theoretical_loss": 4.269219549321364,
      "tokens_seen": 248381440
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000934062186559679,
      "loss": 1.8625,
      "theoretical_loss": 4.269084237158476,
      "tokens_seen": 248446976
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009340421263791375,
      "loss": 1.8055,
      "theoretical_loss": 4.268948970674917,
      "tokens_seen": 248512512
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009340220661985957,
      "loss": 2.1709,
      "theoretical_loss": 4.268813749843226,
      "tokens_seen": 248578048
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009340020060180542,
      "loss": 1.72,
      "theoretical_loss": 4.268678574635967,
      "tokens_seen": 248643584
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009339819458375125,
      "loss": 1.7354,
      "theoretical_loss": 4.268543445025724,
      "tokens_seen": 248709120
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009339618856569709,
      "loss": 1.8053,
      "theoretical_loss": 4.268408360985109,
      "tokens_seen": 248774656
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009339418254764293,
      "loss": 1.9108,
      "theoretical_loss": 4.268273322486757,
      "tokens_seen": 248840192
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009339217652958876,
      "loss": 1.775,
      "theoretical_loss": 4.268138329503325,
      "tokens_seen": 248905728
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009339017051153461,
      "loss": 1.8734,
      "theoretical_loss": 4.2680033820074925,
      "tokens_seen": 248971264
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.4903026223182678,
      "objective/train/docs_used": 97104,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.707381010055542,
      "objective/train/original_loss": 1.7073808908462524,
      "objective/train/theoretical_loss": 4.267868479971966,
      "objective/train/tokens_used": 269496800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24140579998493195,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502574443817139,
      "objective/train/weighted_lm_loss": 1.793129563331604,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.267868479971966,
      "tokens_seen": 249036800
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009338816449348045,
      "loss": 1.7747,
      "theoretical_loss": 4.267868479971966,
      "tokens_seen": 249036800
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009338615847542628,
      "loss": 1.7815,
      "theoretical_loss": 4.267733623369474,
      "tokens_seen": 249102336
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009338415245737212,
      "loss": 1.8381,
      "theoretical_loss": 4.267598812172769,
      "tokens_seen": 249167872
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009338214643931797,
      "loss": 1.966,
      "theoretical_loss": 4.2674640463546245,
      "tokens_seen": 249233408
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009338014042126379,
      "loss": 1.752,
      "theoretical_loss": 4.267329325887841,
      "tokens_seen": 249298944
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009337813440320964,
      "loss": 1.5597,
      "theoretical_loss": 4.26719465074524,
      "tokens_seen": 249364480
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009337612838515546,
      "loss": 1.719,
      "theoretical_loss": 4.267060020899668,
      "tokens_seen": 249430016
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009337412236710131,
      "loss": 1.804,
      "theoretical_loss": 4.2669254363239935,
      "tokens_seen": 249495552
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009337211634904715,
      "loss": 1.7289,
      "theoretical_loss": 4.266790896991109,
      "tokens_seen": 249561088
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009337011033099298,
      "loss": 1.8028,
      "theoretical_loss": 4.266656402873932,
      "tokens_seen": 249626624
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009336810431293882,
      "loss": 1.9201,
      "theoretical_loss": 4.2665219539454,
      "tokens_seen": 249692160
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009336609829488466,
      "loss": 1.9164,
      "theoretical_loss": 4.266387550178476,
      "tokens_seen": 249757696
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009336409227683049,
      "loss": 1.8046,
      "theoretical_loss": 4.266253191546146,
      "tokens_seen": 249823232
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009336208625877634,
      "loss": 1.8337,
      "theoretical_loss": 4.266118878021419,
      "tokens_seen": 249888768
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009336008024072216,
      "loss": 1.9622,
      "theoretical_loss": 4.265984609577327,
      "tokens_seen": 249954304
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009335807422266801,
      "loss": 1.8541,
      "theoretical_loss": 4.265850386186925,
      "tokens_seen": 250019840
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009335606820461384,
      "loss": 1.6784,
      "theoretical_loss": 4.265716207823292,
      "tokens_seen": 250085376
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009335406218655968,
      "loss": 1.7275,
      "theoretical_loss": 4.265582074459531,
      "tokens_seen": 250150912
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009335205616850552,
      "loss": 1.8329,
      "theoretical_loss": 4.265447986068765,
      "tokens_seen": 250216448
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009335005015045136,
      "loss": 1.8422,
      "theoretical_loss": 4.265313942624143,
      "tokens_seen": 250281984
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009334804413239719,
      "loss": 1.9325,
      "theoretical_loss": 4.2651799440988345,
      "tokens_seen": 250347520
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009334603811434304,
      "loss": 1.8524,
      "theoretical_loss": 4.265045990466035,
      "tokens_seen": 250413056
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009334403209628887,
      "loss": 1.8634,
      "theoretical_loss": 4.264912081698962,
      "tokens_seen": 250478592
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009334202607823471,
      "loss": 1.7535,
      "theoretical_loss": 4.264778217770854,
      "tokens_seen": 250544128
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009334002006018054,
      "loss": 1.7067,
      "theoretical_loss": 4.2646443986549745,
      "tokens_seen": 250609664
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009333801404212638,
      "loss": 1.767,
      "theoretical_loss": 4.264510624324609,
      "tokens_seen": 250675200
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009333600802407222,
      "loss": 1.9334,
      "theoretical_loss": 4.2643768947530685,
      "tokens_seen": 250740736
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009333400200601806,
      "loss": 1.8349,
      "theoretical_loss": 4.2642432099136816,
      "tokens_seen": 250806272
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009333199598796389,
      "loss": 1.751,
      "theoretical_loss": 4.264109569779803,
      "tokens_seen": 250871808
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009332998996990973,
      "loss": 1.9473,
      "theoretical_loss": 4.263975974324813,
      "tokens_seen": 250937344
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009332798395185557,
      "loss": 1.7429,
      "theoretical_loss": 4.263842423522109,
      "tokens_seen": 251002880
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009332597793380141,
      "loss": 1.8841,
      "theoretical_loss": 4.263708917345115,
      "tokens_seen": 251068416
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009332397191574724,
      "loss": 1.9271,
      "theoretical_loss": 4.263575455767277,
      "tokens_seen": 251133952
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009332196589769309,
      "loss": 1.9296,
      "theoretical_loss": 4.2634420387620615,
      "tokens_seen": 251199488
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009331995987963891,
      "loss": 1.8366,
      "theoretical_loss": 4.2633086663029625,
      "tokens_seen": 251265024
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009331795386158476,
      "loss": 1.8976,
      "theoretical_loss": 4.263175338363491,
      "tokens_seen": 251330560
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009331594784353059,
      "loss": 1.8301,
      "theoretical_loss": 4.263042054917186,
      "tokens_seen": 251396096
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009331394182547643,
      "loss": 2.0389,
      "theoretical_loss": 4.262908815937604,
      "tokens_seen": 251461632
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009331193580742227,
      "loss": 2.0028,
      "theoretical_loss": 4.262775621398329,
      "tokens_seen": 251527168
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009330992978936811,
      "loss": 2.0408,
      "theoretical_loss": 4.262642471272963,
      "tokens_seen": 251592704
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009330792377131394,
      "loss": 2.0631,
      "theoretical_loss": 4.262509365535134,
      "tokens_seen": 251658240
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009330591775325979,
      "loss": 1.8172,
      "theoretical_loss": 4.262376304158492,
      "tokens_seen": 251723776
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009330391173520561,
      "loss": 1.9694,
      "theoretical_loss": 4.262243287116707,
      "tokens_seen": 251789312
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009330190571715146,
      "loss": 1.8517,
      "theoretical_loss": 4.2621103143834755,
      "tokens_seen": 251854848
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000932998996990973,
      "loss": 2.0222,
      "theoretical_loss": 4.261977385932512,
      "tokens_seen": 251920384
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009329789368104313,
      "loss": 1.9252,
      "theoretical_loss": 4.261844501737556,
      "tokens_seen": 251985920
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009329588766298897,
      "loss": 1.9154,
      "theoretical_loss": 4.26171166177237,
      "tokens_seen": 252051456
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000932938816449348,
      "loss": 1.8763,
      "theoretical_loss": 4.261578866010739,
      "tokens_seen": 252116992
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009329187562688064,
      "loss": 1.9258,
      "theoretical_loss": 4.261446114426466,
      "tokens_seen": 252182528
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009328986960882649,
      "loss": 1.823,
      "theoretical_loss": 4.261313406993382,
      "tokens_seen": 252248064
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.49495235085487366,
      "objective/train/docs_used": 98257,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8033281564712524,
      "objective/train/original_loss": 1.803328037261963,
      "objective/train/theoretical_loss": 4.261180743685337,
      "objective/train/tokens_used": 272773600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24527713656425476,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507421493530273,
      "objective/train/weighted_lm_loss": 1.8945033550262451,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0253151655197144,
      "theoretical_loss": 4.261180743685337,
      "tokens_seen": 252313600
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009328786359077231,
      "loss": 1.8192,
      "theoretical_loss": 4.261180743685337,
      "tokens_seen": 252313600
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009328585757271816,
      "loss": 1.9158,
      "theoretical_loss": 4.261048124476204,
      "tokens_seen": 252379136
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009328385155466398,
      "loss": 1.9291,
      "theoretical_loss": 4.260915549339879,
      "tokens_seen": 252444672
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009328184553660983,
      "loss": 1.7672,
      "theoretical_loss": 4.2607830182502795,
      "tokens_seen": 252510208
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009327983951855567,
      "loss": 1.8606,
      "theoretical_loss": 4.260650531181345,
      "tokens_seen": 252575744
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000932778335005015,
      "loss": 1.8812,
      "theoretical_loss": 4.260518088107038,
      "tokens_seen": 252641280
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009327582748244734,
      "loss": 1.9065,
      "theoretical_loss": 4.2603856890013425,
      "tokens_seen": 252706816
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009327382146439319,
      "loss": 1.8938,
      "theoretical_loss": 4.260253333838264,
      "tokens_seen": 252772352
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009327181544633901,
      "loss": 1.9165,
      "theoretical_loss": 4.260121022591832,
      "tokens_seen": 252837888
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009326980942828486,
      "loss": 1.7946,
      "theoretical_loss": 4.259988755236097,
      "tokens_seen": 252903424
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009326780341023069,
      "loss": 1.9426,
      "theoretical_loss": 4.25985653174513,
      "tokens_seen": 252968960
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009326579739217653,
      "loss": 1.7934,
      "theoretical_loss": 4.259724352093027,
      "tokens_seen": 253034496
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009326379137412237,
      "loss": 1.7676,
      "theoretical_loss": 4.259592216253903,
      "tokens_seen": 253100032
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000932617853560682,
      "loss": 1.8153,
      "theoretical_loss": 4.259460124201899,
      "tokens_seen": 253165568
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009325977933801404,
      "loss": 1.7746,
      "theoretical_loss": 4.259328075911173,
      "tokens_seen": 253231104
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009325777331995988,
      "loss": 1.7761,
      "theoretical_loss": 4.259196071355909,
      "tokens_seen": 253296640
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009325576730190571,
      "loss": 1.9124,
      "theoretical_loss": 4.259064110510311,
      "tokens_seen": 253362176
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009325376128385156,
      "loss": 1.8265,
      "theoretical_loss": 4.258932193348604,
      "tokens_seen": 253427712
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009325175526579739,
      "loss": 1.9741,
      "theoretical_loss": 4.258800319845038,
      "tokens_seen": 253493248
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009324974924774323,
      "loss": 1.8248,
      "theoretical_loss": 4.258668489973882,
      "tokens_seen": 253558784
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009324774322968907,
      "loss": 1.8307,
      "theoretical_loss": 4.258536703709426,
      "tokens_seen": 253624320
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000932457372116349,
      "loss": 1.7819,
      "theoretical_loss": 4.258404961025986,
      "tokens_seen": 253689856
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009324373119358074,
      "loss": 1.9135,
      "theoretical_loss": 4.258273261897896,
      "tokens_seen": 253755392
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009324172517552658,
      "loss": 1.8765,
      "theoretical_loss": 4.258141606299514,
      "tokens_seen": 253820928
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009323971915747242,
      "loss": 1.7368,
      "theoretical_loss": 4.258009994205218,
      "tokens_seen": 253886464
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009323771313941826,
      "loss": 1.8066,
      "theoretical_loss": 4.257878425589407,
      "tokens_seen": 253952000
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000932357071213641,
      "loss": 1.8179,
      "theoretical_loss": 4.257746900426506,
      "tokens_seen": 254017536
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009323370110330993,
      "loss": 1.7851,
      "theoretical_loss": 4.257615418690956,
      "tokens_seen": 254083072
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009323169508525577,
      "loss": 1.7856,
      "theoretical_loss": 4.2574839803572235,
      "tokens_seen": 254148608
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009322968906720161,
      "loss": 1.8776,
      "theoretical_loss": 4.257352585399795,
      "tokens_seen": 254214144
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009322768304914745,
      "loss": 1.8318,
      "theoretical_loss": 4.25722123379318,
      "tokens_seen": 254279680
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009322567703109328,
      "loss": 1.952,
      "theoretical_loss": 4.257089925511907,
      "tokens_seen": 254345216
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009322367101303913,
      "loss": 2.0118,
      "theoretical_loss": 4.256958660530529,
      "tokens_seen": 254410752
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009322166499498495,
      "loss": 1.8912,
      "theoretical_loss": 4.256827438823617,
      "tokens_seen": 254476288
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000932196589769308,
      "loss": 1.7353,
      "theoretical_loss": 4.256696260365768,
      "tokens_seen": 254541824
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009321765295887663,
      "loss": 1.8728,
      "theoretical_loss": 4.256565125131597,
      "tokens_seen": 254607360
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009321564694082247,
      "loss": 1.8088,
      "theoretical_loss": 4.25643403309574,
      "tokens_seen": 254672896
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009321364092276831,
      "loss": 1.8988,
      "theoretical_loss": 4.256302984232858,
      "tokens_seen": 254738432
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009321163490471415,
      "loss": 1.8722,
      "theoretical_loss": 4.256171978517629,
      "tokens_seen": 254803968
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009320962888665998,
      "loss": 1.6762,
      "theoretical_loss": 4.256041015924756,
      "tokens_seen": 254869504
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009320762286860583,
      "loss": 1.8221,
      "theoretical_loss": 4.255910096428962,
      "tokens_seen": 254935040
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009320561685055165,
      "loss": 1.8804,
      "theoretical_loss": 4.255779220004991,
      "tokens_seen": 255000576
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000932036108324975,
      "loss": 1.6887,
      "theoretical_loss": 4.255648386627607,
      "tokens_seen": 255066112
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009320160481444333,
      "loss": 1.8639,
      "theoretical_loss": 4.255517596271598,
      "tokens_seen": 255131648
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009319959879638917,
      "loss": 1.8197,
      "theoretical_loss": 4.255386848911772,
      "tokens_seen": 255197184
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009319759277833501,
      "loss": 1.7893,
      "theoretical_loss": 4.255256144522958,
      "tokens_seen": 255262720
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009319558676028084,
      "loss": 1.8523,
      "theoretical_loss": 4.255125483080007,
      "tokens_seen": 255328256
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009319358074222668,
      "loss": 1.8729,
      "theoretical_loss": 4.254994864557789,
      "tokens_seen": 255393792
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009319157472417253,
      "loss": 1.7798,
      "theoretical_loss": 4.254864288931198,
      "tokens_seen": 255459328
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009318956870611835,
      "loss": 1.8593,
      "theoretical_loss": 4.254733756175147,
      "tokens_seen": 255524864
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.4922325015068054,
      "objective/train/docs_used": 99403,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7772718667984009,
      "objective/train/original_loss": 1.7772717475891113,
      "objective/train/theoretical_loss": 4.254603266264572,
      "objective/train/tokens_used": 276050400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24300870299339294,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050458550453186,
      "objective/train/weighted_lm_loss": 1.8668968677520752,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.254603266264572,
      "tokens_seen": 255590400
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000931875626880642,
      "loss": 1.7993,
      "theoretical_loss": 4.254603266264572,
      "tokens_seen": 255590400
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009318555667001002,
      "loss": 1.7996,
      "theoretical_loss": 4.254472819174428,
      "tokens_seen": 255655936
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009318355065195587,
      "loss": 1.6432,
      "theoretical_loss": 4.254342414879692,
      "tokens_seen": 255721472
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009318154463390171,
      "loss": 1.833,
      "theoretical_loss": 4.254212053355362,
      "tokens_seen": 255787008
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009317953861584754,
      "loss": 1.8269,
      "theoretical_loss": 4.254081734576458,
      "tokens_seen": 255852544
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009317753259779338,
      "loss": 1.7435,
      "theoretical_loss": 4.25395145851802,
      "tokens_seen": 255918080
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009317552657973923,
      "loss": 1.8863,
      "theoretical_loss": 4.253821225155107,
      "tokens_seen": 255983616
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009317352056168505,
      "loss": 1.7373,
      "theoretical_loss": 4.253691034462804,
      "tokens_seen": 256049152
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000931715145436309,
      "loss": 1.8167,
      "theoretical_loss": 4.253560886416212,
      "tokens_seen": 256114688
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009316950852557673,
      "loss": 1.8065,
      "theoretical_loss": 4.253430780990455,
      "tokens_seen": 256180224
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009316750250752257,
      "loss": 1.7302,
      "theoretical_loss": 4.253300718160678,
      "tokens_seen": 256245760
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009316549648946841,
      "loss": 1.8542,
      "theoretical_loss": 4.253170697902046,
      "tokens_seen": 256311296
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009316349047141424,
      "loss": 1.7814,
      "theoretical_loss": 4.253040720189746,
      "tokens_seen": 256376832
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009316148445336008,
      "loss": 1.9602,
      "theoretical_loss": 4.252910784998984,
      "tokens_seen": 256442368
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009315947843530592,
      "loss": 1.6887,
      "theoretical_loss": 4.2527808923049895,
      "tokens_seen": 256507904
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009315747241725175,
      "loss": 1.8397,
      "theoretical_loss": 4.252651042083009,
      "tokens_seen": 256573440
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000931554663991976,
      "loss": 1.7413,
      "theoretical_loss": 4.252521234308315,
      "tokens_seen": 256638976
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009315346038114343,
      "loss": 1.7299,
      "theoretical_loss": 4.252391468956194,
      "tokens_seen": 256704512
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009315145436308927,
      "loss": 1.7748,
      "theoretical_loss": 4.25226174600196,
      "tokens_seen": 256770048
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009314944834503511,
      "loss": 1.7904,
      "theoretical_loss": 4.252132065420942,
      "tokens_seen": 256835584
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009314744232698095,
      "loss": 1.8599,
      "theoretical_loss": 4.2520024271884935,
      "tokens_seen": 256901120
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009314543630892678,
      "loss": 1.6606,
      "theoretical_loss": 4.251872831279986,
      "tokens_seen": 256966656
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009314343029087262,
      "loss": 1.7963,
      "theoretical_loss": 4.251743277670815,
      "tokens_seen": 257032192
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009314142427281845,
      "loss": 1.9087,
      "theoretical_loss": 4.251613766336391,
      "tokens_seen": 257097728
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000931394182547643,
      "loss": 1.9016,
      "theoretical_loss": 4.251484297252151,
      "tokens_seen": 257163264
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009313741223671013,
      "loss": 1.9086,
      "theoretical_loss": 4.25135487039355,
      "tokens_seen": 257228800
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009313540621865597,
      "loss": 1.9209,
      "theoretical_loss": 4.251225485736063,
      "tokens_seen": 257294336
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000931334002006018,
      "loss": 1.7854,
      "theoretical_loss": 4.251096143255185,
      "tokens_seen": 257359872
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009313139418254765,
      "loss": 1.9528,
      "theoretical_loss": 4.250966842926434,
      "tokens_seen": 257425408
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009312938816449348,
      "loss": 1.808,
      "theoretical_loss": 4.250837584725346,
      "tokens_seen": 257490944
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009312738214643932,
      "loss": 1.7127,
      "theoretical_loss": 4.250708368627477,
      "tokens_seen": 257556480
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009312537612838515,
      "loss": 1.9612,
      "theoretical_loss": 4.250579194608408,
      "tokens_seen": 257622016
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009312337011033099,
      "loss": 1.8438,
      "theoretical_loss": 4.250450062643734,
      "tokens_seen": 257687552
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009312136409227683,
      "loss": 1.7651,
      "theoretical_loss": 4.2503209727090745,
      "tokens_seen": 257753088
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009311935807422267,
      "loss": 1.8181,
      "theoretical_loss": 4.250191924780069,
      "tokens_seen": 257818624
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000931173520561685,
      "loss": 1.781,
      "theoretical_loss": 4.250062918832374,
      "tokens_seen": 257884160
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009311534603811435,
      "loss": 1.6716,
      "theoretical_loss": 4.249933954841672,
      "tokens_seen": 257949696
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009311334002006018,
      "loss": 2.0471,
      "theoretical_loss": 4.249805032783661,
      "tokens_seen": 258015232
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009311133400200602,
      "loss": 1.8217,
      "theoretical_loss": 4.249676152634061,
      "tokens_seen": 258080768
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009310932798395186,
      "loss": 1.9101,
      "theoretical_loss": 4.249547314368613,
      "tokens_seen": 258146304
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009310732196589769,
      "loss": 1.8241,
      "theoretical_loss": 4.2494185179630755,
      "tokens_seen": 258211840
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009310531594784353,
      "loss": 1.8213,
      "theoretical_loss": 4.24928976339323,
      "tokens_seen": 258277376
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009310330992978937,
      "loss": 1.8407,
      "theoretical_loss": 4.249161050634876,
      "tokens_seen": 258342912
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000931013039117352,
      "loss": 1.9322,
      "theoretical_loss": 4.249032379663836,
      "tokens_seen": 258408448
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009309929789368105,
      "loss": 1.7703,
      "theoretical_loss": 4.24890375045595,
      "tokens_seen": 258473984
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009309729187562687,
      "loss": 1.8058,
      "theoretical_loss": 4.248775162987078,
      "tokens_seen": 258539520
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009309528585757272,
      "loss": 1.7423,
      "theoretical_loss": 4.248646617233103,
      "tokens_seen": 258605056
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009309327983951856,
      "loss": 1.7474,
      "theoretical_loss": 4.248518113169924,
      "tokens_seen": 258670592
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009309127382146439,
      "loss": 1.9037,
      "theoretical_loss": 4.248389650773463,
      "tokens_seen": 258736128
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009308926780341024,
      "loss": 1.7221,
      "theoretical_loss": 4.248261230019661,
      "tokens_seen": 258801664
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.4922111928462982,
      "objective/train/docs_used": 100739,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5991153717041016,
      "objective/train/original_loss": 1.5991151332855225,
      "objective/train/theoretical_loss": 4.24813285088448,
      "objective/train/tokens_used": 279327200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24256941676139832,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504542589187622,
      "objective/train/weighted_lm_loss": 1.679618239402771,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0253151655197144,
      "theoretical_loss": 4.24813285088448,
      "tokens_seen": 258867200
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009308726178535606,
      "loss": 1.7685,
      "theoretical_loss": 4.24813285088448,
      "tokens_seen": 258867200
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009308525576730191,
      "loss": 1.8756,
      "theoretical_loss": 4.248004513343899,
      "tokens_seen": 258932736
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009308324974924775,
      "loss": 1.9389,
      "theoretical_loss": 4.24787621737392,
      "tokens_seen": 258998272
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009308124373119358,
      "loss": 1.7622,
      "theoretical_loss": 4.247747962950564,
      "tokens_seen": 259063808
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009307923771313942,
      "loss": 1.8245,
      "theoretical_loss": 4.247619750049871,
      "tokens_seen": 259129344
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009307723169508527,
      "loss": 1.7996,
      "theoretical_loss": 4.247491578647903,
      "tokens_seen": 259194880
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009307522567703109,
      "loss": 1.7617,
      "theoretical_loss": 4.247363448720739,
      "tokens_seen": 259260416
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009307321965897694,
      "loss": 1.7627,
      "theoretical_loss": 4.24723536024448,
      "tokens_seen": 259325952
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009307121364092277,
      "loss": 1.8327,
      "theoretical_loss": 4.247107313195246,
      "tokens_seen": 259391488
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009306920762286861,
      "loss": 1.8597,
      "theoretical_loss": 4.246979307549177,
      "tokens_seen": 259457024
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009306720160481445,
      "loss": 1.8707,
      "theoretical_loss": 4.246851343282432,
      "tokens_seen": 259522560
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009306519558676028,
      "loss": 1.9826,
      "theoretical_loss": 4.246723420371191,
      "tokens_seen": 259588096
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009306318956870612,
      "loss": 1.828,
      "theoretical_loss": 4.246595538791653,
      "tokens_seen": 259653632
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009306118355065196,
      "loss": 1.8696,
      "theoretical_loss": 4.246467698520037,
      "tokens_seen": 259719168
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009305917753259779,
      "loss": 1.823,
      "theoretical_loss": 4.246339899532582,
      "tokens_seen": 259784704
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009305717151454364,
      "loss": 1.872,
      "theoretical_loss": 4.246212141805544,
      "tokens_seen": 259850240
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009305516549648947,
      "loss": 1.8403,
      "theoretical_loss": 4.246084425315203,
      "tokens_seen": 259915776
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009305315947843531,
      "loss": 1.6529,
      "theoretical_loss": 4.245956750037855,
      "tokens_seen": 259981312
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009305115346038115,
      "loss": 1.7805,
      "theoretical_loss": 4.245829115949818,
      "tokens_seen": 260046848
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009304914744232699,
      "loss": 1.9065,
      "theoretical_loss": 4.245701523027429,
      "tokens_seen": 260112384
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009304714142427282,
      "loss": 1.8109,
      "theoretical_loss": 4.245573971247042,
      "tokens_seen": 260177920
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009304513540621866,
      "loss": 1.8887,
      "theoretical_loss": 4.245446460585035,
      "tokens_seen": 260243456
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009304312938816449,
      "loss": 1.886,
      "theoretical_loss": 4.245318991017802,
      "tokens_seen": 260308992
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009304112337011034,
      "loss": 1.7386,
      "theoretical_loss": 4.245191562521757,
      "tokens_seen": 260374528
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009303911735205617,
      "loss": 1.8463,
      "theoretical_loss": 4.2450641750733356,
      "tokens_seen": 260440064
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009303711133400201,
      "loss": 1.7975,
      "theoretical_loss": 4.24493682864899,
      "tokens_seen": 260505600
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009303510531594784,
      "loss": 1.8787,
      "theoretical_loss": 4.244809523225195,
      "tokens_seen": 260571136
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009303309929789369,
      "loss": 1.7446,
      "theoretical_loss": 4.244682258778441,
      "tokens_seen": 260636672
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009303109327983952,
      "loss": 1.8998,
      "theoretical_loss": 4.2445550352852415,
      "tokens_seen": 260702208
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009302908726178536,
      "loss": 1.919,
      "theoretical_loss": 4.244427852722127,
      "tokens_seen": 260767744
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000930270812437312,
      "loss": 1.8169,
      "theoretical_loss": 4.244300711065646,
      "tokens_seen": 260833280
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009302507522567703,
      "loss": 1.785,
      "theoretical_loss": 4.244173610292372,
      "tokens_seen": 260898816
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009302306920762287,
      "loss": 1.685,
      "theoretical_loss": 4.244046550378892,
      "tokens_seen": 260964352
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009302106318956871,
      "loss": 1.6913,
      "theoretical_loss": 4.243919531301814,
      "tokens_seen": 261029888
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009301905717151454,
      "loss": 1.7615,
      "theoretical_loss": 4.243792553037767,
      "tokens_seen": 261095424
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009301705115346039,
      "loss": 1.8247,
      "theoretical_loss": 4.243665615563396,
      "tokens_seen": 261160960
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009301504513540622,
      "loss": 1.8357,
      "theoretical_loss": 4.243538718855369,
      "tokens_seen": 261226496
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009301303911735206,
      "loss": 1.8834,
      "theoretical_loss": 4.243411862890371,
      "tokens_seen": 261292032
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000930110330992979,
      "loss": 1.7083,
      "theoretical_loss": 4.243285047645106,
      "tokens_seen": 261357568
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009300902708124373,
      "loss": 1.8269,
      "theoretical_loss": 4.243158273096297,
      "tokens_seen": 261423104
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009300702106318957,
      "loss": 1.9182,
      "theoretical_loss": 4.243031539220689,
      "tokens_seen": 261488640
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009300501504513541,
      "loss": 1.9445,
      "theoretical_loss": 4.242904845995041,
      "tokens_seen": 261554176
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009300300902708124,
      "loss": 1.8878,
      "theoretical_loss": 4.242778193396136,
      "tokens_seen": 261619712
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009300100300902709,
      "loss": 1.8315,
      "theoretical_loss": 4.242651581400774,
      "tokens_seen": 261685248
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009299899699097291,
      "loss": 1.8104,
      "theoretical_loss": 4.242525009985773,
      "tokens_seen": 261750784
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009299699097291876,
      "loss": 1.8935,
      "theoretical_loss": 4.242398479127972,
      "tokens_seen": 261816320
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000929949849548646,
      "loss": 1.7999,
      "theoretical_loss": 4.242271988804228,
      "tokens_seen": 261881856
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009299297893681043,
      "loss": 1.7533,
      "theoretical_loss": 4.242145538991417,
      "tokens_seen": 261947392
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009299097291875627,
      "loss": 1.6952,
      "theoretical_loss": 4.242019129666434,
      "tokens_seen": 262012928
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000929889669007021,
      "loss": 1.688,
      "theoretical_loss": 4.241892760806194,
      "tokens_seen": 262078464
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.4959133267402649,
      "objective/train/docs_used": 101934,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8622653484344482,
      "objective/train/original_loss": 1.8622653484344482,
      "objective/train/theoretical_loss": 4.241766432387629,
      "objective/train/tokens_used": 282604000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24602378904819489,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050842046737671,
      "objective/train/weighted_lm_loss": 1.957066535949707,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0396549701690674,
      "theoretical_loss": 4.241766432387629,
      "tokens_seen": 262144000
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009298696088264794,
      "loss": 1.8692,
      "theoretical_loss": 4.241766432387629,
      "tokens_seen": 262144000
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009298495486459379,
      "loss": 1.8105,
      "theoretical_loss": 4.241640144387691,
      "tokens_seen": 262209536
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009298294884653961,
      "loss": 1.7178,
      "theoretical_loss": 4.241513896783352,
      "tokens_seen": 262275072
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009298094282848546,
      "loss": 1.8063,
      "theoretical_loss": 4.2413876895516,
      "tokens_seen": 262340608
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000929789368104313,
      "loss": 1.8252,
      "theoretical_loss": 4.241261522669445,
      "tokens_seen": 262406144
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009297693079237713,
      "loss": 1.8267,
      "theoretical_loss": 4.241135396113914,
      "tokens_seen": 262471680
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009297492477432297,
      "loss": 1.9232,
      "theoretical_loss": 4.2410093098620525,
      "tokens_seen": 262537216
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000929729187562688,
      "loss": 1.7269,
      "theoretical_loss": 4.240883263890925,
      "tokens_seen": 262602752
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009297091273821464,
      "loss": 1.9214,
      "theoretical_loss": 4.240757258177617,
      "tokens_seen": 262668288
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009296890672016049,
      "loss": 1.8207,
      "theoretical_loss": 4.240631292699231,
      "tokens_seen": 262733824
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009296690070210631,
      "loss": 1.7695,
      "theoretical_loss": 4.2405053674328865,
      "tokens_seen": 262799360
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009296489468405216,
      "loss": 1.789,
      "theoretical_loss": 4.240379482355724,
      "tokens_seen": 262864896
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009296288866599799,
      "loss": 1.6906,
      "theoretical_loss": 4.240253637444903,
      "tokens_seen": 262930432
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009296088264794383,
      "loss": 1.689,
      "theoretical_loss": 4.2401278326776,
      "tokens_seen": 262995968
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009295887662988967,
      "loss": 1.7909,
      "theoretical_loss": 4.2400020680310115,
      "tokens_seen": 263061504
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009295687061183551,
      "loss": 1.7541,
      "theoretical_loss": 4.239876343482352,
      "tokens_seen": 263127040
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009295486459378134,
      "loss": 1.8034,
      "theoretical_loss": 4.239750659008854,
      "tokens_seen": 263192576
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009295285857572719,
      "loss": 1.7027,
      "theoretical_loss": 4.23962501458777,
      "tokens_seen": 263258112
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009295085255767301,
      "loss": 1.923,
      "theoretical_loss": 4.2394994101963706,
      "tokens_seen": 263323648
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009294884653961886,
      "loss": 1.7502,
      "theoretical_loss": 4.239373845811944,
      "tokens_seen": 263389184
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009294684052156469,
      "loss": 1.8158,
      "theoretical_loss": 4.2392483214117975,
      "tokens_seen": 263454720
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009294483450351053,
      "loss": 1.9041,
      "theoretical_loss": 4.239122836973258,
      "tokens_seen": 263520256
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009294282848545638,
      "loss": 1.8612,
      "theoretical_loss": 4.238997392473669,
      "tokens_seen": 263585792
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009294082246740221,
      "loss": 1.8119,
      "theoretical_loss": 4.238871987890395,
      "tokens_seen": 263651328
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009293881644934805,
      "loss": 1.8197,
      "theoretical_loss": 4.238746623200815,
      "tokens_seen": 263716864
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009293681043129388,
      "loss": 1.8277,
      "theoretical_loss": 4.238621298382331,
      "tokens_seen": 263782400
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009293480441323973,
      "loss": 1.8537,
      "theoretical_loss": 4.23849601341236,
      "tokens_seen": 263847936
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009293279839518556,
      "loss": 1.7895,
      "theoretical_loss": 4.2383707682683385,
      "tokens_seen": 263913472
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000929307923771314,
      "loss": 1.8325,
      "theoretical_loss": 4.238245562927722,
      "tokens_seen": 263979008
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009292878635907723,
      "loss": 1.7732,
      "theoretical_loss": 4.238120397367984,
      "tokens_seen": 264044544
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009292678034102307,
      "loss": 1.8699,
      "theoretical_loss": 4.237995271566616,
      "tokens_seen": 264110080
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009292477432296891,
      "loss": 1.9023,
      "theoretical_loss": 4.237870185501127,
      "tokens_seen": 264175616
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009292276830491475,
      "loss": 1.7925,
      "theoretical_loss": 4.237745139149047,
      "tokens_seen": 264241152
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009292076228686058,
      "loss": 1.825,
      "theoretical_loss": 4.237620132487923,
      "tokens_seen": 264306688
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009291875626880643,
      "loss": 1.7765,
      "theoretical_loss": 4.237495165495317,
      "tokens_seen": 264372224
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009291675025075226,
      "loss": 1.7354,
      "theoretical_loss": 4.237370238148814,
      "tokens_seen": 264437760
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000929147442326981,
      "loss": 1.7925,
      "theoretical_loss": 4.237245350426015,
      "tokens_seen": 264503296
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009291273821464394,
      "loss": 1.6433,
      "theoretical_loss": 4.23712050230454,
      "tokens_seen": 264568832
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009291073219658977,
      "loss": 1.8064,
      "theoretical_loss": 4.236995693762026,
      "tokens_seen": 264634368
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009290872617853561,
      "loss": 1.6874,
      "theoretical_loss": 4.236870924776128,
      "tokens_seen": 264699904
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009290672016048145,
      "loss": 1.8004,
      "theoretical_loss": 4.236746195324523,
      "tokens_seen": 264765440
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009290471414242728,
      "loss": 1.8509,
      "theoretical_loss": 4.2366215053849,
      "tokens_seen": 264830976
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009290270812437313,
      "loss": 1.8856,
      "theoretical_loss": 4.23649685493497,
      "tokens_seen": 264896512
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009290070210631895,
      "loss": 1.8162,
      "theoretical_loss": 4.236372243952461,
      "tokens_seen": 264962048
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000928986960882648,
      "loss": 1.9025,
      "theoretical_loss": 4.2362476724151215,
      "tokens_seen": 265027584
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009289669007021064,
      "loss": 1.8805,
      "theoretical_loss": 4.236123140300713,
      "tokens_seen": 265093120
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009289468405215647,
      "loss": 1.7836,
      "theoretical_loss": 4.235998647587021,
      "tokens_seen": 265158656
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009289267803410231,
      "loss": 1.8182,
      "theoretical_loss": 4.235874194251843,
      "tokens_seen": 265224192
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009289067201604814,
      "loss": 1.737,
      "theoretical_loss": 4.235749780272998,
      "tokens_seen": 265289728
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009288866599799398,
      "loss": 1.8422,
      "theoretical_loss": 4.2356254056283245,
      "tokens_seen": 265355264
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.49507758021354675,
      "objective/train/docs_used": 103026,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5715179443359375,
      "objective/train/original_loss": 1.5715179443359375,
      "objective/train/theoretical_loss": 4.235501070295674,
      "objective/train/tokens_used": 285880800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24532651901245117,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507550239562988,
      "objective/train/weighted_lm_loss": 1.6512619256973267,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.235501070295674,
      "tokens_seen": 265420800
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009288665997993983,
      "loss": 1.8102,
      "theoretical_loss": 4.235501070295674,
      "tokens_seen": 265420800
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009288465396188565,
      "loss": 1.8941,
      "theoretical_loss": 4.235376774252922,
      "tokens_seen": 265486336
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000928826479438315,
      "loss": 1.9101,
      "theoretical_loss": 4.235252517477956,
      "tokens_seen": 265551872
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009288064192577734,
      "loss": 1.7015,
      "theoretical_loss": 4.235128299948684,
      "tokens_seen": 265617408
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009287863590772317,
      "loss": 1.7076,
      "theoretical_loss": 4.235004121643034,
      "tokens_seen": 265682944
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009287662988966901,
      "loss": 1.8181,
      "theoretical_loss": 4.234879982538949,
      "tokens_seen": 265748480
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009287462387161485,
      "loss": 1.7664,
      "theoretical_loss": 4.23475588261439,
      "tokens_seen": 265814016
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009287261785356068,
      "loss": 1.787,
      "theoretical_loss": 4.234631821847337,
      "tokens_seen": 265879552
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009287061183550653,
      "loss": 1.7446,
      "theoretical_loss": 4.234507800215788,
      "tokens_seen": 265945088
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009286860581745235,
      "loss": 1.8445,
      "theoretical_loss": 4.234383817697755,
      "tokens_seen": 266010624
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000928665997993982,
      "loss": 1.777,
      "theoretical_loss": 4.234259874271275,
      "tokens_seen": 266076160
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009286459378134403,
      "loss": 1.8054,
      "theoretical_loss": 4.234135969914396,
      "tokens_seen": 266141696
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009286258776328987,
      "loss": 1.9136,
      "theoretical_loss": 4.234012104605187,
      "tokens_seen": 266207232
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009286058174523571,
      "loss": 1.7999,
      "theoretical_loss": 4.233888278321734,
      "tokens_seen": 266272768
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009285857572718155,
      "loss": 1.6735,
      "theoretical_loss": 4.23376449104214,
      "tokens_seen": 266338304
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009285656970912738,
      "loss": 1.8224,
      "theoretical_loss": 4.233640742744528,
      "tokens_seen": 266403840
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009285456369107323,
      "loss": 1.9214,
      "theoretical_loss": 4.233517033407035,
      "tokens_seen": 266469376
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009285255767301905,
      "loss": 1.8953,
      "theoretical_loss": 4.23339336300782,
      "tokens_seen": 266534912
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000928505516549649,
      "loss": 1.8393,
      "theoretical_loss": 4.233269731525055,
      "tokens_seen": 266600448
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009284854563691073,
      "loss": 1.8253,
      "theoretical_loss": 4.233146138936933,
      "tokens_seen": 266665984
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009284653961885657,
      "loss": 1.8904,
      "theoretical_loss": 4.233022585221662,
      "tokens_seen": 266731520
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009284453360080241,
      "loss": 1.8195,
      "theoretical_loss": 4.232899070357472,
      "tokens_seen": 266797056
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009284252758274825,
      "loss": 1.7398,
      "theoretical_loss": 4.232775594322605,
      "tokens_seen": 266862592
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009284052156469408,
      "loss": 1.8531,
      "theoretical_loss": 4.232652157095323,
      "tokens_seen": 266928128
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009283851554663992,
      "loss": 1.8012,
      "theoretical_loss": 4.232528758653906,
      "tokens_seen": 266993664
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009283650952858576,
      "loss": 1.7835,
      "theoretical_loss": 4.232405398976653,
      "tokens_seen": 267059200
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000928345035105316,
      "loss": 1.7591,
      "theoretical_loss": 4.232282078041876,
      "tokens_seen": 267124736
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009283249749247743,
      "loss": 1.7493,
      "theoretical_loss": 4.232158795827908,
      "tokens_seen": 267190272
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009283049147442327,
      "loss": 1.8331,
      "theoretical_loss": 4.232035552313097,
      "tokens_seen": 267255808
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000928284854563691,
      "loss": 1.759,
      "theoretical_loss": 4.231912347475812,
      "tokens_seen": 267321344
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009282647943831495,
      "loss": 1.9049,
      "theoretical_loss": 4.231789181294436,
      "tokens_seen": 267386880
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009282447342026078,
      "loss": 1.8436,
      "theoretical_loss": 4.231666053747371,
      "tokens_seen": 267452416
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009282246740220662,
      "loss": 1.9009,
      "theoretical_loss": 4.231542964813034,
      "tokens_seen": 267517952
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009282046138415246,
      "loss": 1.7458,
      "theoretical_loss": 4.231419914469864,
      "tokens_seen": 267583488
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000928184553660983,
      "loss": 1.9132,
      "theoretical_loss": 4.231296902696314,
      "tokens_seen": 267649024
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009281644934804413,
      "loss": 1.738,
      "theoretical_loss": 4.231173929470853,
      "tokens_seen": 267714560
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009281444332998998,
      "loss": 1.8208,
      "theoretical_loss": 4.231050994771971,
      "tokens_seen": 267780096
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000928124373119358,
      "loss": 1.7115,
      "theoretical_loss": 4.230928098578174,
      "tokens_seen": 267845632
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009281043129388165,
      "loss": 1.7666,
      "theoretical_loss": 4.230805240867982,
      "tokens_seen": 267911168
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009280842527582748,
      "loss": 1.8364,
      "theoretical_loss": 4.230682421619936,
      "tokens_seen": 267976704
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009280641925777332,
      "loss": 1.7936,
      "theoretical_loss": 4.2305596408125945,
      "tokens_seen": 268042240
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009280441323971916,
      "loss": 1.7574,
      "theoretical_loss": 4.23043689842453,
      "tokens_seen": 268107776
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009280240722166499,
      "loss": 1.7779,
      "theoretical_loss": 4.230314194434336,
      "tokens_seen": 268173312
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009280040120361083,
      "loss": 1.7898,
      "theoretical_loss": 4.23019152882062,
      "tokens_seen": 268238848
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009279839518555668,
      "loss": 1.8302,
      "theoretical_loss": 4.2300689015620065,
      "tokens_seen": 268304384
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000927963891675025,
      "loss": 1.8138,
      "theoretical_loss": 4.229946312637141,
      "tokens_seen": 268369920
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009279438314944835,
      "loss": 1.7802,
      "theoretical_loss": 4.229823762024681,
      "tokens_seen": 268435456
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009279237713139418,
      "loss": 1.8906,
      "theoretical_loss": 4.2297012497033055,
      "tokens_seen": 268500992
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009279037111334002,
      "loss": 1.8555,
      "theoretical_loss": 4.229578775651707,
      "tokens_seen": 268566528
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009278836509528587,
      "loss": 1.7775,
      "theoretical_loss": 4.229456339848598,
      "tokens_seen": 268632064
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.49237704277038574,
      "objective/train/docs_used": 104173,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5822020769119263,
      "objective/train/original_loss": 1.5822019577026367,
      "objective/train/theoretical_loss": 4.2293339422727065,
      "objective/train/tokens_used": 289157600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24295735359191895,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504728555679321,
      "objective/train/weighted_lm_loss": 1.661965250968933,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.2293339422727065,
      "tokens_seen": 268697600
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009278635907723169,
      "loss": 1.677,
      "theoretical_loss": 4.2293339422727065,
      "tokens_seen": 268697600
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009278435305917754,
      "loss": 1.771,
      "theoretical_loss": 4.229211582902778,
      "tokens_seen": 268763136
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009278234704112338,
      "loss": 1.8272,
      "theoretical_loss": 4.229089261717574,
      "tokens_seen": 268828672
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009278034102306921,
      "loss": 1.8982,
      "theoretical_loss": 4.2289669786958735,
      "tokens_seen": 268894208
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009277833500501505,
      "loss": 1.8134,
      "theoretical_loss": 4.228844733816474,
      "tokens_seen": 268959744
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009277632898696089,
      "loss": 1.8508,
      "theoretical_loss": 4.228722527058188,
      "tokens_seen": 269025280
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009277432296890672,
      "loss": 1.7465,
      "theoretical_loss": 4.228600358399845,
      "tokens_seen": 269090816
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009277231695085257,
      "loss": 1.8031,
      "theoretical_loss": 4.228478227820292,
      "tokens_seen": 269156352
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009277031093279839,
      "loss": 1.6002,
      "theoretical_loss": 4.228356135298394,
      "tokens_seen": 269221888
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009276830491474424,
      "loss": 1.7915,
      "theoretical_loss": 4.2282340808130305,
      "tokens_seen": 269287424
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009276629889669007,
      "loss": 1.7165,
      "theoretical_loss": 4.2281120643431,
      "tokens_seen": 269352960
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009276429287863591,
      "loss": 1.6739,
      "theoretical_loss": 4.227990085867517,
      "tokens_seen": 269418496
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009276228686058175,
      "loss": 1.7462,
      "theoretical_loss": 4.227868145365211,
      "tokens_seen": 269484032
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009276028084252759,
      "loss": 1.7257,
      "theoretical_loss": 4.227746242815132,
      "tokens_seen": 269549568
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009275827482447342,
      "loss": 1.8566,
      "theoretical_loss": 4.227624378196245,
      "tokens_seen": 269615104
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009275626880641927,
      "loss": 1.8052,
      "theoretical_loss": 4.227502551487531,
      "tokens_seen": 269680640
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000927542627883651,
      "loss": 1.804,
      "theoretical_loss": 4.227380762667987,
      "tokens_seen": 269746176
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009275225677031094,
      "loss": 1.8553,
      "theoretical_loss": 4.2272590117166295,
      "tokens_seen": 269811712
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009275025075225677,
      "loss": 1.682,
      "theoretical_loss": 4.22713729861249,
      "tokens_seen": 269877248
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009274824473420261,
      "loss": 1.7293,
      "theoretical_loss": 4.2270156233346174,
      "tokens_seen": 269942784
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009274623871614845,
      "loss": 1.7395,
      "theoretical_loss": 4.226893985862076,
      "tokens_seen": 270008320
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009274423269809429,
      "loss": 1.8387,
      "theoretical_loss": 4.226772386173948,
      "tokens_seen": 270073856
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009274222668004012,
      "loss": 1.8202,
      "theoretical_loss": 4.226650824249333,
      "tokens_seen": 270139392
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009274022066198596,
      "loss": 1.8298,
      "theoretical_loss": 4.226529300067345,
      "tokens_seen": 270204928
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000927382146439318,
      "loss": 1.8401,
      "theoretical_loss": 4.226407813607116,
      "tokens_seen": 270270464
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009273620862587764,
      "loss": 1.831,
      "theoretical_loss": 4.226286364847795,
      "tokens_seen": 270336000
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009273420260782347,
      "loss": 1.7314,
      "theoretical_loss": 4.226164953768545,
      "tokens_seen": 270401536
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009273219658976931,
      "loss": 1.7272,
      "theoretical_loss": 4.226043580348549,
      "tokens_seen": 270467072
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009273019057171514,
      "loss": 1.9179,
      "theoretical_loss": 4.2259222445670055,
      "tokens_seen": 270532608
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009272818455366099,
      "loss": 1.8673,
      "theoretical_loss": 4.225800946403127,
      "tokens_seen": 270598144
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009272617853560682,
      "loss": 1.6766,
      "theoretical_loss": 4.2256796858361465,
      "tokens_seen": 270663680
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009272417251755266,
      "loss": 1.7084,
      "theoretical_loss": 4.225558462845311,
      "tokens_seen": 270729216
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000927221664994985,
      "loss": 1.6541,
      "theoretical_loss": 4.225437277409885,
      "tokens_seen": 270794752
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009272016048144434,
      "loss": 1.7796,
      "theoretical_loss": 4.225316129509147,
      "tokens_seen": 270860288
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009271815446339017,
      "loss": 1.6994,
      "theoretical_loss": 4.2251950191223955,
      "tokens_seen": 270925824
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009271614844533602,
      "loss": 1.8671,
      "theoretical_loss": 4.225073946228944,
      "tokens_seen": 270991360
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009271414242728184,
      "loss": 1.698,
      "theoretical_loss": 4.224952910808122,
      "tokens_seen": 271056896
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009271213640922769,
      "loss": 1.7853,
      "theoretical_loss": 4.224831912839275,
      "tokens_seen": 271122432
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009271013039117352,
      "loss": 1.8293,
      "theoretical_loss": 4.224710952301766,
      "tokens_seen": 271187968
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009270812437311936,
      "loss": 1.8736,
      "theoretical_loss": 4.2245900291749745,
      "tokens_seen": 271253504
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000927061183550652,
      "loss": 1.7211,
      "theoretical_loss": 4.224469143438294,
      "tokens_seen": 271319040
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009270411233701103,
      "loss": 1.9201,
      "theoretical_loss": 4.224348295071138,
      "tokens_seen": 271384576
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009270210631895687,
      "loss": 1.8416,
      "theoretical_loss": 4.224227484052934,
      "tokens_seen": 271450112
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009270010030090272,
      "loss": 1.7762,
      "theoretical_loss": 4.224106710363124,
      "tokens_seen": 271515648
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009269809428284854,
      "loss": 1.9026,
      "theoretical_loss": 4.223985973981171,
      "tokens_seen": 271581184
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009269608826479439,
      "loss": 1.9119,
      "theoretical_loss": 4.223865274886551,
      "tokens_seen": 271646720
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009269408224674021,
      "loss": 1.8214,
      "theoretical_loss": 4.223744613058755,
      "tokens_seen": 271712256
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009269207622868606,
      "loss": 1.7376,
      "theoretical_loss": 4.223623988477295,
      "tokens_seen": 271777792
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000926900702106319,
      "loss": 1.9123,
      "theoretical_loss": 4.223503401121693,
      "tokens_seen": 271843328
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009268806419257773,
      "loss": 1.741,
      "theoretical_loss": 4.223382850971494,
      "tokens_seen": 271908864
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.4953818619251251,
      "objective/train/docs_used": 105443,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6509478092193604,
      "objective/train/original_loss": 1.6509475708007812,
      "objective/train/theoretical_loss": 4.223262338006254,
      "objective/train/tokens_used": 292434400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24556824564933777,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507866144180298,
      "objective/train/weighted_lm_loss": 1.7348129749298096,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9753099083900452,
      "theoretical_loss": 4.223262338006254,
      "tokens_seen": 271974400
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009268605817452357,
      "loss": 1.7963,
      "theoretical_loss": 4.223262338006254,
      "tokens_seen": 271974400
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009268405215646942,
      "loss": 1.6685,
      "theoretical_loss": 4.223141862205546,
      "tokens_seen": 272039936
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009268204613841524,
      "loss": 1.7575,
      "theoretical_loss": 4.223021423548962,
      "tokens_seen": 272105472
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009268004012036109,
      "loss": 1.7426,
      "theoretical_loss": 4.2229010220161065,
      "tokens_seen": 272171008
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009267803410230691,
      "loss": 1.7231,
      "theoretical_loss": 4.222780657586601,
      "tokens_seen": 272236544
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009267602808425276,
      "loss": 1.7428,
      "theoretical_loss": 4.222660330240086,
      "tokens_seen": 272302080
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000926740220661986,
      "loss": 1.9007,
      "theoretical_loss": 4.222540039956215,
      "tokens_seen": 272367616
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009267201604814443,
      "loss": 1.795,
      "theoretical_loss": 4.222419786714657,
      "tokens_seen": 272433152
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009267001003009027,
      "loss": 1.768,
      "theoretical_loss": 4.2222995704951005,
      "tokens_seen": 272498688
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009266800401203611,
      "loss": 1.7197,
      "theoretical_loss": 4.222179391277247,
      "tokens_seen": 272564224
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009266599799398194,
      "loss": 1.76,
      "theoretical_loss": 4.222059249040814,
      "tokens_seen": 272629760
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009266399197592779,
      "loss": 1.7811,
      "theoretical_loss": 4.221939143765538,
      "tokens_seen": 272695296
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009266198595787362,
      "loss": 1.842,
      "theoretical_loss": 4.221819075431168,
      "tokens_seen": 272760832
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009265997993981946,
      "loss": 1.7936,
      "theoretical_loss": 4.221699044017471,
      "tokens_seen": 272826368
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009265797392176529,
      "loss": 1.904,
      "theoretical_loss": 4.2215790495042285,
      "tokens_seen": 272891904
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009265596790371113,
      "loss": 1.8888,
      "theoretical_loss": 4.22145909187124,
      "tokens_seen": 272957440
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009265396188565697,
      "loss": 1.8071,
      "theoretical_loss": 4.221339171098318,
      "tokens_seen": 273022976
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009265195586760281,
      "loss": 1.6943,
      "theoretical_loss": 4.221219287165294,
      "tokens_seen": 273088512
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009264994984954864,
      "loss": 1.9064,
      "theoretical_loss": 4.221099440052014,
      "tokens_seen": 273154048
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009264794383149449,
      "loss": 1.8892,
      "theoretical_loss": 4.220979629738339,
      "tokens_seen": 273219584
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009264593781344032,
      "loss": 1.7186,
      "theoretical_loss": 4.220859856204147,
      "tokens_seen": 273285120
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009264393179538616,
      "loss": 1.7486,
      "theoretical_loss": 4.22074011942933,
      "tokens_seen": 273350656
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.00092641925777332,
      "loss": 1.7085,
      "theoretical_loss": 4.220620419393799,
      "tokens_seen": 273416192
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009263991975927784,
      "loss": 1.7723,
      "theoretical_loss": 4.220500756077479,
      "tokens_seen": 273481728
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009263791374122368,
      "loss": 1.8846,
      "theoretical_loss": 4.22038112946031,
      "tokens_seen": 273547264
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009263590772316951,
      "loss": 1.875,
      "theoretical_loss": 4.220261539522249,
      "tokens_seen": 273612800
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009263390170511535,
      "loss": 1.7537,
      "theoretical_loss": 4.220141986243268,
      "tokens_seen": 273678336
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009263189568706118,
      "loss": 1.7963,
      "theoretical_loss": 4.220022469603356,
      "tokens_seen": 273743872
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009262988966900703,
      "loss": 1.7591,
      "theoretical_loss": 4.219902989582516,
      "tokens_seen": 273809408
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009262788365095286,
      "loss": 1.8612,
      "theoretical_loss": 4.219783546160768,
      "tokens_seen": 273874944
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000926258776328987,
      "loss": 1.8849,
      "theoretical_loss": 4.219664139318145,
      "tokens_seen": 273940480
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009262387161484454,
      "loss": 1.7991,
      "theoretical_loss": 4.219544769034701,
      "tokens_seen": 274006016
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009262186559679038,
      "loss": 1.7852,
      "theoretical_loss": 4.2194254352905,
      "tokens_seen": 274071552
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009261985957873621,
      "loss": 1.7718,
      "theoretical_loss": 4.219306138065626,
      "tokens_seen": 274137088
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009261785356068206,
      "loss": 1.8751,
      "theoretical_loss": 4.219186877340174,
      "tokens_seen": 274202624
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009261584754262788,
      "loss": 1.7946,
      "theoretical_loss": 4.219067653094261,
      "tokens_seen": 274268160
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009261384152457373,
      "loss": 1.6745,
      "theoretical_loss": 4.218948465308013,
      "tokens_seen": 274333696
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009261183550651956,
      "loss": 1.7044,
      "theoretical_loss": 4.218829313961575,
      "tokens_seen": 274399232
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000926098294884654,
      "loss": 1.8374,
      "theoretical_loss": 4.218710199035108,
      "tokens_seen": 274464768
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009260782347041124,
      "loss": 1.6713,
      "theoretical_loss": 4.218591120508787,
      "tokens_seen": 274530304
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009260581745235707,
      "loss": 1.7884,
      "theoretical_loss": 4.218472078362803,
      "tokens_seen": 274595840
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009260381143430291,
      "loss": 1.7148,
      "theoretical_loss": 4.2183530725773615,
      "tokens_seen": 274661376
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009260180541624876,
      "loss": 1.8029,
      "theoretical_loss": 4.218234103132686,
      "tokens_seen": 274726912
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009259979939819458,
      "loss": 1.9025,
      "theoretical_loss": 4.218115170009014,
      "tokens_seen": 274792448
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009259779338014043,
      "loss": 1.7295,
      "theoretical_loss": 4.217996273186598,
      "tokens_seen": 274857984
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009259578736208625,
      "loss": 1.7892,
      "theoretical_loss": 4.217877412645706,
      "tokens_seen": 274923520
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000925937813440321,
      "loss": 1.8208,
      "theoretical_loss": 4.217758588366623,
      "tokens_seen": 274989056
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009259177532597794,
      "loss": 1.8495,
      "theoretical_loss": 4.217639800329647,
      "tokens_seen": 275054592
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009258976930792377,
      "loss": 1.6701,
      "theoretical_loss": 4.217521048515094,
      "tokens_seen": 275120128
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009258776328986961,
      "loss": 1.7104,
      "theoretical_loss": 4.217402332903292,
      "tokens_seen": 275185664
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.4921969771385193,
      "objective/train/docs_used": 106703,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6392239332199097,
      "objective/train/original_loss": 1.6392236948013306,
      "objective/train/theoretical_loss": 4.217283653474588,
      "objective/train/tokens_used": 295711200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.243023082613945,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050455093383789,
      "objective/train/weighted_lm_loss": 1.7219140529632568,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.217283653474588,
      "tokens_seen": 275251200
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009258575727181546,
      "loss": 1.7057,
      "theoretical_loss": 4.217283653474588,
      "tokens_seen": 275251200
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009258375125376128,
      "loss": 1.752,
      "theoretical_loss": 4.217165010209343,
      "tokens_seen": 275316736
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009258174523570713,
      "loss": 1.8469,
      "theoretical_loss": 4.217046403087933,
      "tokens_seen": 275382272
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009257973921765295,
      "loss": 1.7028,
      "theoretical_loss": 4.2169278320907475,
      "tokens_seen": 275447808
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000925777331995988,
      "loss": 1.86,
      "theoretical_loss": 4.216809297198195,
      "tokens_seen": 275513344
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009257572718154464,
      "loss": 1.8148,
      "theoretical_loss": 4.216690798390697,
      "tokens_seen": 275578880
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009257372116349047,
      "loss": 1.7633,
      "theoretical_loss": 4.216572335648692,
      "tokens_seen": 275644416
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009257171514543631,
      "loss": 1.8091,
      "theoretical_loss": 4.21645390895263,
      "tokens_seen": 275709952
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009256970912738215,
      "loss": 1.8062,
      "theoretical_loss": 4.21633551828298,
      "tokens_seen": 275775488
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009256770310932798,
      "loss": 1.6688,
      "theoretical_loss": 4.216217163620225,
      "tokens_seen": 275841024
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009256569709127383,
      "loss": 1.8757,
      "theoretical_loss": 4.216098844944863,
      "tokens_seen": 275906560
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009256369107321966,
      "loss": 1.8344,
      "theoretical_loss": 4.215980562237409,
      "tokens_seen": 275972096
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000925616850551655,
      "loss": 1.904,
      "theoretical_loss": 4.215862315478388,
      "tokens_seen": 276037632
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009255967903711133,
      "loss": 1.8534,
      "theoretical_loss": 4.215744104648347,
      "tokens_seen": 276103168
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009255767301905717,
      "loss": 1.7778,
      "theoretical_loss": 4.2156259297278424,
      "tokens_seen": 276168704
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009255566700100301,
      "loss": 1.9598,
      "theoretical_loss": 4.215507790697449,
      "tokens_seen": 276234240
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009255366098294885,
      "loss": 1.7881,
      "theoretical_loss": 4.2153896875377574,
      "tokens_seen": 276299776
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009255165496489468,
      "loss": 1.8789,
      "theoretical_loss": 4.21527162022937,
      "tokens_seen": 276365312
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009254964894684053,
      "loss": 1.8463,
      "theoretical_loss": 4.215153588752907,
      "tokens_seen": 276430848
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009254764292878636,
      "loss": 1.9038,
      "theoretical_loss": 4.215035593089002,
      "tokens_seen": 276496384
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000925456369107322,
      "loss": 1.6358,
      "theoretical_loss": 4.214917633218304,
      "tokens_seen": 276561920
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009254363089267803,
      "loss": 1.8301,
      "theoretical_loss": 4.214799709121479,
      "tokens_seen": 276627456
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009254162487462388,
      "loss": 1.8829,
      "theoretical_loss": 4.214681820779205,
      "tokens_seen": 276692992
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009253961885656971,
      "loss": 1.7861,
      "theoretical_loss": 4.214563968172178,
      "tokens_seen": 276758528
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009253761283851555,
      "loss": 1.7876,
      "theoretical_loss": 4.214446151281106,
      "tokens_seen": 276824064
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009253560682046138,
      "loss": 1.8599,
      "theoretical_loss": 4.214328370086713,
      "tokens_seen": 276889600
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009253360080240722,
      "loss": 1.6949,
      "theoretical_loss": 4.21421062456974,
      "tokens_seen": 276955136
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009253159478435306,
      "loss": 1.7641,
      "theoretical_loss": 4.214092914710941,
      "tokens_seen": 277020672
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000925295887662989,
      "loss": 1.7158,
      "theoretical_loss": 4.213975240491084,
      "tokens_seen": 277086208
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009252758274824473,
      "loss": 1.8548,
      "theoretical_loss": 4.213857601890955,
      "tokens_seen": 277151744
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009252557673019058,
      "loss": 1.8456,
      "theoretical_loss": 4.213739998891352,
      "tokens_seen": 277217280
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000925235707121364,
      "loss": 1.7495,
      "theoretical_loss": 4.213622431473088,
      "tokens_seen": 277282816
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009252156469408225,
      "loss": 1.7643,
      "theoretical_loss": 4.213504899616995,
      "tokens_seen": 277348352
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009251955867602808,
      "loss": 1.7701,
      "theoretical_loss": 4.213387403303912,
      "tokens_seen": 277413888
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009251755265797392,
      "loss": 1.7438,
      "theoretical_loss": 4.213269942514701,
      "tokens_seen": 277479424
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009251554663991976,
      "loss": 1.8916,
      "theoretical_loss": 4.2131525172302355,
      "tokens_seen": 277544960
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000925135406218656,
      "loss": 1.8483,
      "theoretical_loss": 4.213035127431402,
      "tokens_seen": 277610496
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009251153460381143,
      "loss": 1.7789,
      "theoretical_loss": 4.212917773099105,
      "tokens_seen": 277676032
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009250952858575728,
      "loss": 1.7993,
      "theoretical_loss": 4.2128004542142605,
      "tokens_seen": 277741568
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000925075225677031,
      "loss": 1.8079,
      "theoretical_loss": 4.212683170757802,
      "tokens_seen": 277807104
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009250551654964895,
      "loss": 1.8363,
      "theoretical_loss": 4.212565922710677,
      "tokens_seen": 277872640
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009250351053159479,
      "loss": 1.8011,
      "theoretical_loss": 4.212448710053847,
      "tokens_seen": 277938176
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009250150451354062,
      "loss": 1.8875,
      "theoretical_loss": 4.212331532768289,
      "tokens_seen": 278003712
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009249949849548646,
      "loss": 1.7518,
      "theoretical_loss": 4.212214390834994,
      "tokens_seen": 278069248
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009249749247743229,
      "loss": 1.7445,
      "theoretical_loss": 4.21209728423497,
      "tokens_seen": 278134784
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009249548645937813,
      "loss": 1.9204,
      "theoretical_loss": 4.211980212949234,
      "tokens_seen": 278200320
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009249348044132398,
      "loss": 1.7544,
      "theoretical_loss": 4.2118631769588255,
      "tokens_seen": 278265856
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009249147442326981,
      "loss": 1.8568,
      "theoretical_loss": 4.211746176244792,
      "tokens_seen": 278331392
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009248946840521565,
      "loss": 1.8516,
      "theoretical_loss": 4.2116292107882,
      "tokens_seen": 278396928
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000924874623871615,
      "loss": 1.7939,
      "theoretical_loss": 4.211512280570127,
      "tokens_seen": 278462464
    },
    {
      "epoch": 0.08,
      "objective/train/advantage_avg": 0.4929666519165039,
      "objective/train/docs_used": 107929,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.715563178062439,
      "objective/train/original_loss": 1.7155632972717285,
      "objective/train/theoretical_loss": 4.211395385571668,
      "objective/train/tokens_used": 298988000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2441163808107376,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505377054214478,
      "objective/train/weighted_lm_loss": 1.8023045063018799,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.211395385571668,
      "tokens_seen": 278528000
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009248545636910732,
      "loss": 1.8125,
      "theoretical_loss": 4.211395385571668,
      "tokens_seen": 278528000
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009248345035105317,
      "loss": 1.8219,
      "theoretical_loss": 4.211278525773933,
      "tokens_seen": 278593536
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.00092481444332999,
      "loss": 1.838,
      "theoretical_loss": 4.211161701158042,
      "tokens_seen": 278659072
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009247943831494484,
      "loss": 1.8463,
      "theoretical_loss": 4.211044911705135,
      "tokens_seen": 278724608
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009247743229689068,
      "loss": 1.8635,
      "theoretical_loss": 4.2109281573963635,
      "tokens_seen": 278790144
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009247542627883651,
      "loss": 1.8983,
      "theoretical_loss": 4.210811438212894,
      "tokens_seen": 278855680
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009247342026078235,
      "loss": 1.739,
      "theoretical_loss": 4.2106947541359085,
      "tokens_seen": 278921216
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009247141424272819,
      "loss": 1.9169,
      "theoretical_loss": 4.2105781051466025,
      "tokens_seen": 278986752
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009246940822467402,
      "loss": 1.7906,
      "theoretical_loss": 4.2104614912261855,
      "tokens_seen": 279052288
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009246740220661987,
      "loss": 1.6945,
      "theoretical_loss": 4.210344912355883,
      "tokens_seen": 279117824
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000924653961885657,
      "loss": 1.7868,
      "theoretical_loss": 4.210228368516935,
      "tokens_seen": 279183360
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009246339017051154,
      "loss": 1.8477,
      "theoretical_loss": 4.210111859690594,
      "tokens_seen": 279248896
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009246138415245737,
      "loss": 1.9463,
      "theoretical_loss": 4.209995385858129,
      "tokens_seen": 279314432
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009245937813440321,
      "loss": 1.9702,
      "theoretical_loss": 4.20987894700082,
      "tokens_seen": 279379968
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009245737211634905,
      "loss": 1.8656,
      "theoretical_loss": 4.209762543099966,
      "tokens_seen": 279445504
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009245536609829489,
      "loss": 1.821,
      "theoretical_loss": 4.209646174136879,
      "tokens_seen": 279511040
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009245336008024072,
      "loss": 1.8142,
      "theoretical_loss": 4.209529840092884,
      "tokens_seen": 279576576
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009245135406218657,
      "loss": 1.911,
      "theoretical_loss": 4.209413540949319,
      "tokens_seen": 279642112
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000924493480441324,
      "loss": 1.7045,
      "theoretical_loss": 4.209297276687541,
      "tokens_seen": 279707648
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009244734202607824,
      "loss": 1.8239,
      "theoretical_loss": 4.209181047288917,
      "tokens_seen": 279773184
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009244533600802407,
      "loss": 1.7827,
      "theoretical_loss": 4.209064852734831,
      "tokens_seen": 279838720
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009244332998996992,
      "loss": 1.962,
      "theoretical_loss": 4.20894869300668,
      "tokens_seen": 279904256
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009244132397191575,
      "loss": 1.8998,
      "theoretical_loss": 4.2088325680858745,
      "tokens_seen": 279969792
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009243931795386159,
      "loss": 1.7249,
      "theoretical_loss": 4.208716477953843,
      "tokens_seen": 280035328
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009243731193580742,
      "loss": 1.8494,
      "theoretical_loss": 4.2086004225920215,
      "tokens_seen": 280100864
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009243530591775326,
      "loss": 1.8086,
      "theoretical_loss": 4.208484401981868,
      "tokens_seen": 280166400
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.000924332998996991,
      "loss": 1.7255,
      "theoretical_loss": 4.208368416104849,
      "tokens_seen": 280231936
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009243129388164494,
      "loss": 1.7742,
      "theoretical_loss": 4.208252464942449,
      "tokens_seen": 280297472
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009242928786359077,
      "loss": 1.8617,
      "theoretical_loss": 4.208136548476163,
      "tokens_seen": 280363008
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009242728184553662,
      "loss": 1.8351,
      "theoretical_loss": 4.208020666687503,
      "tokens_seen": 280428544
    },
    {
      "epoch": 0.08,
      "learning_rate": 0.0009242527582748244,
      "loss": 1.9316,
      "theoretical_loss": 4.207904819557995,
      "tokens_seen": 280494080
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009242326980942829,
      "loss": 1.7367,
      "theoretical_loss": 4.207789007069177,
      "tokens_seen": 280559616
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009242126379137412,
      "loss": 1.8257,
      "theoretical_loss": 4.207673229202604,
      "tokens_seen": 280625152
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009241925777331996,
      "loss": 1.7784,
      "theoretical_loss": 4.2075574859398435,
      "tokens_seen": 280690688
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000924172517552658,
      "loss": 1.8542,
      "theoretical_loss": 4.207441777262477,
      "tokens_seen": 280756224
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009241524573721164,
      "loss": 1.703,
      "theoretical_loss": 4.2073261031521,
      "tokens_seen": 280821760
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009241323971915747,
      "loss": 1.7389,
      "theoretical_loss": 4.207210463590323,
      "tokens_seen": 280887296
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009241123370110332,
      "loss": 1.9065,
      "theoretical_loss": 4.207094858558771,
      "tokens_seen": 280952832
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009240922768304914,
      "loss": 1.774,
      "theoretical_loss": 4.206979288039081,
      "tokens_seen": 281018368
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009240722166499499,
      "loss": 1.7895,
      "theoretical_loss": 4.2068637520129055,
      "tokens_seen": 281083904
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009240521564694083,
      "loss": 1.7552,
      "theoretical_loss": 4.206748250461911,
      "tokens_seen": 281149440
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009240320962888666,
      "loss": 1.771,
      "theoretical_loss": 4.206632783367777,
      "tokens_seen": 281214976
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000924012036108325,
      "loss": 1.7158,
      "theoretical_loss": 4.206517350712199,
      "tokens_seen": 281280512
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009239919759277833,
      "loss": 1.9191,
      "theoretical_loss": 4.2064019524768845,
      "tokens_seen": 281346048
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009239719157472417,
      "loss": 1.8433,
      "theoretical_loss": 4.206286588643556,
      "tokens_seen": 281411584
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009239518555667002,
      "loss": 1.7216,
      "theoretical_loss": 4.206171259193948,
      "tokens_seen": 281477120
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009239317953861584,
      "loss": 1.7115,
      "theoretical_loss": 4.206055964109813,
      "tokens_seen": 281542656
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009239117352056169,
      "loss": 1.705,
      "theoretical_loss": 4.205940703372915,
      "tokens_seen": 281608192
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009238916750250753,
      "loss": 1.7713,
      "theoretical_loss": 4.20582547696503,
      "tokens_seen": 281673728
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009238716148445336,
      "loss": 1.8652,
      "theoretical_loss": 4.205710284867951,
      "tokens_seen": 281739264
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.49427810311317444,
      "objective/train/docs_used": 109222,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6125668287277222,
      "objective/train/original_loss": 1.6125669479370117,
      "objective/train/theoretical_loss": 4.205595127063485,
      "objective/train/tokens_used": 302264800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24455593526363373,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050671100616455,
      "objective/train/weighted_lm_loss": 1.694186806678772,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.205595127063485,
      "tokens_seen": 281804800
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000923851554663992,
      "loss": 1.7848,
      "theoretical_loss": 4.205595127063485,
      "tokens_seen": 281804800
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009238314944834503,
      "loss": 1.7997,
      "theoretical_loss": 4.20548000353345,
      "tokens_seen": 281870336
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009238114343029087,
      "loss": 1.7206,
      "theoretical_loss": 4.205364914259679,
      "tokens_seen": 281935872
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009237913741223672,
      "loss": 1.7657,
      "theoretical_loss": 4.205249859224021,
      "tokens_seen": 282001408
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009237713139418254,
      "loss": 1.7049,
      "theoretical_loss": 4.205134838408337,
      "tokens_seen": 282066944
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009237512537612839,
      "loss": 1.7584,
      "theoretical_loss": 4.2050198517945,
      "tokens_seen": 282132480
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009237311935807422,
      "loss": 1.7654,
      "theoretical_loss": 4.204904899364402,
      "tokens_seen": 282198016
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009237111334002006,
      "loss": 1.7786,
      "theoretical_loss": 4.2047899810999425,
      "tokens_seen": 282263552
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000923691073219659,
      "loss": 1.8459,
      "theoretical_loss": 4.20467509698304,
      "tokens_seen": 282329088
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009236710130391174,
      "loss": 1.7859,
      "theoretical_loss": 4.204560246995625,
      "tokens_seen": 282394624
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009236509528585757,
      "loss": 1.7339,
      "theoretical_loss": 4.204445431119639,
      "tokens_seen": 282460160
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009236308926780341,
      "loss": 1.7904,
      "theoretical_loss": 4.204330649337042,
      "tokens_seen": 282525696
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009236108324974924,
      "loss": 1.7929,
      "theoretical_loss": 4.204215901629803,
      "tokens_seen": 282591232
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009235907723169509,
      "loss": 1.8564,
      "theoretical_loss": 4.2041011879799095,
      "tokens_seen": 282656768
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009235707121364092,
      "loss": 1.7214,
      "theoretical_loss": 4.2039865083693595,
      "tokens_seen": 282722304
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009235506519558676,
      "loss": 1.8249,
      "theoretical_loss": 4.203871862780165,
      "tokens_seen": 282787840
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000923530591775326,
      "loss": 1.7751,
      "theoretical_loss": 4.203757251194353,
      "tokens_seen": 282853376
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009235105315947844,
      "loss": 1.7531,
      "theoretical_loss": 4.203642673593962,
      "tokens_seen": 282918912
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009234904714142427,
      "loss": 1.7935,
      "theoretical_loss": 4.203528129961047,
      "tokens_seen": 282984448
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009234704112337011,
      "loss": 1.7967,
      "theoretical_loss": 4.203413620277673,
      "tokens_seen": 283049984
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009234503510531594,
      "loss": 1.7649,
      "theoretical_loss": 4.203299144525923,
      "tokens_seen": 283115520
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009234302908726179,
      "loss": 1.8882,
      "theoretical_loss": 4.20318470268789,
      "tokens_seen": 283181056
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009234102306920763,
      "loss": 1.7575,
      "theoretical_loss": 4.203070294745682,
      "tokens_seen": 283246592
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009233901705115346,
      "loss": 1.7621,
      "theoretical_loss": 4.202955920681421,
      "tokens_seen": 283312128
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000923370110330993,
      "loss": 1.7461,
      "theoretical_loss": 4.202841580477241,
      "tokens_seen": 283377664
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009233500501504514,
      "loss": 1.9422,
      "theoretical_loss": 4.202727274115292,
      "tokens_seen": 283443200
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009233299899699098,
      "loss": 1.798,
      "theoretical_loss": 4.202613001577735,
      "tokens_seen": 283508736
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009233099297893681,
      "loss": 1.8495,
      "theoretical_loss": 4.202498762846745,
      "tokens_seen": 283574272
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009232898696088266,
      "loss": 1.7852,
      "theoretical_loss": 4.202384557904513,
      "tokens_seen": 283639808
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009232698094282848,
      "loss": 1.7845,
      "theoretical_loss": 4.202270386733241,
      "tokens_seen": 283705344
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009232497492477433,
      "loss": 1.7888,
      "theoretical_loss": 4.202156249315145,
      "tokens_seen": 283770880
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009232296890672016,
      "loss": 1.8123,
      "theoretical_loss": 4.202042145632453,
      "tokens_seen": 283836416
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.00092320962888666,
      "loss": 1.7403,
      "theoretical_loss": 4.201928075667411,
      "tokens_seen": 283901952
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009231895687061184,
      "loss": 1.735,
      "theoretical_loss": 4.201814039402273,
      "tokens_seen": 283967488
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009231695085255768,
      "loss": 1.7603,
      "theoretical_loss": 4.201700036819311,
      "tokens_seen": 284033024
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009231494483450351,
      "loss": 1.9366,
      "theoretical_loss": 4.201586067900806,
      "tokens_seen": 284098560
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009231293881644936,
      "loss": 1.6707,
      "theoretical_loss": 4.201472132629057,
      "tokens_seen": 284164096
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009231093279839518,
      "loss": 1.9417,
      "theoretical_loss": 4.201358230986372,
      "tokens_seen": 284229632
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009230892678034103,
      "loss": 1.82,
      "theoretical_loss": 4.201244362955077,
      "tokens_seen": 284295168
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009230692076228687,
      "loss": 1.8958,
      "theoretical_loss": 4.201130528517506,
      "tokens_seen": 284360704
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000923049147442327,
      "loss": 1.8833,
      "theoretical_loss": 4.201016727656012,
      "tokens_seen": 284426240
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009230290872617854,
      "loss": 1.784,
      "theoretical_loss": 4.200902960352956,
      "tokens_seen": 284491776
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009230090270812437,
      "loss": 1.8496,
      "theoretical_loss": 4.200789226590717,
      "tokens_seen": 284557312
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009229889669007021,
      "loss": 1.7737,
      "theoretical_loss": 4.200675526351684,
      "tokens_seen": 284622848
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009229689067201606,
      "loss": 1.893,
      "theoretical_loss": 4.2005618596182615,
      "tokens_seen": 284688384
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009229488465396188,
      "loss": 1.8352,
      "theoretical_loss": 4.200448226372865,
      "tokens_seen": 284753920
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009229287863590773,
      "loss": 1.8389,
      "theoretical_loss": 4.200334626597925,
      "tokens_seen": 284819456
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009229087261785357,
      "loss": 1.8186,
      "theoretical_loss": 4.200221060275885,
      "tokens_seen": 284884992
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000922888665997994,
      "loss": 1.8161,
      "theoretical_loss": 4.200107527389202,
      "tokens_seen": 284950528
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009228686058174524,
      "loss": 1.7495,
      "theoretical_loss": 4.1999940279203445,
      "tokens_seen": 285016064
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.4956050515174866,
      "objective/train/docs_used": 110428,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.646780252456665,
      "objective/train/original_loss": 1.646780252456665,
      "objective/train/theoretical_loss": 4.1998805618517965,
      "objective/train/tokens_used": 305541600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24578233063220978,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0508100986480713,
      "objective/train/weighted_lm_loss": 1.730436086654663,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.1998805618517965,
      "tokens_seen": 285081600
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009228485456369107,
      "loss": 1.7924,
      "theoretical_loss": 4.1998805618517965,
      "tokens_seen": 285081600
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009228284854563691,
      "loss": 1.7654,
      "theoretical_loss": 4.199767129166054,
      "tokens_seen": 285147136
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009228084252758276,
      "loss": 1.871,
      "theoretical_loss": 4.199653729845626,
      "tokens_seen": 285212672
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009227883650952858,
      "loss": 1.8636,
      "theoretical_loss": 4.199540363873036,
      "tokens_seen": 285278208
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009227683049147443,
      "loss": 1.7923,
      "theoretical_loss": 4.19942703123082,
      "tokens_seen": 285343744
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009227482447342026,
      "loss": 1.7764,
      "theoretical_loss": 4.199313731901524,
      "tokens_seen": 285409280
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000922728184553661,
      "loss": 1.791,
      "theoretical_loss": 4.199200465867714,
      "tokens_seen": 285474816
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009227081243731194,
      "loss": 1.8081,
      "theoretical_loss": 4.199087233111962,
      "tokens_seen": 285540352
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009226880641925778,
      "loss": 1.6973,
      "theoretical_loss": 4.198974033616859,
      "tokens_seen": 285605888
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009226680040120361,
      "loss": 1.8117,
      "theoretical_loss": 4.1988608673650045,
      "tokens_seen": 285671424
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009226479438314945,
      "loss": 1.8261,
      "theoretical_loss": 4.198747734339013,
      "tokens_seen": 285736960
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009226278836509528,
      "loss": 1.8351,
      "theoretical_loss": 4.198634634521513,
      "tokens_seen": 285802496
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009226078234704113,
      "loss": 1.7645,
      "theoretical_loss": 4.198521567895144,
      "tokens_seen": 285868032
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009225877632898696,
      "loss": 1.7425,
      "theoretical_loss": 4.198408534442561,
      "tokens_seen": 285933568
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000922567703109328,
      "loss": 1.7141,
      "theoretical_loss": 4.198295534146429,
      "tokens_seen": 285999104
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009225476429287864,
      "loss": 1.7746,
      "theoretical_loss": 4.19818256698943,
      "tokens_seen": 286064640
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009225275827482448,
      "loss": 1.7737,
      "theoretical_loss": 4.198069632954255,
      "tokens_seen": 286130176
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009225075225677031,
      "loss": 1.7182,
      "theoretical_loss": 4.19795673202361,
      "tokens_seen": 286195712
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009224874623871615,
      "loss": 1.8036,
      "theoretical_loss": 4.197843864180214,
      "tokens_seen": 286261248
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009224674022066198,
      "loss": 1.8232,
      "theoretical_loss": 4.1977310294068,
      "tokens_seen": 286326784
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009224473420260783,
      "loss": 1.7982,
      "theoretical_loss": 4.19761822768611,
      "tokens_seen": 286392320
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009224272818455366,
      "loss": 1.735,
      "theoretical_loss": 4.1975054590009035,
      "tokens_seen": 286457856
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000922407221664995,
      "loss": 1.7124,
      "theoretical_loss": 4.197392723333951,
      "tokens_seen": 286523392
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009223871614844533,
      "loss": 1.8675,
      "theoretical_loss": 4.1972800206680345,
      "tokens_seen": 286588928
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009223671013039118,
      "loss": 1.7668,
      "theoretical_loss": 4.197167350985952,
      "tokens_seen": 286654464
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009223470411233701,
      "loss": 1.812,
      "theoretical_loss": 4.197054714270513,
      "tokens_seen": 286720000
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009223269809428285,
      "loss": 1.8564,
      "theoretical_loss": 4.196942110504538,
      "tokens_seen": 286785536
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009223069207622869,
      "loss": 1.7916,
      "theoretical_loss": 4.196829539670864,
      "tokens_seen": 286851072
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009222868605817452,
      "loss": 1.7991,
      "theoretical_loss": 4.1967170017523365,
      "tokens_seen": 286916608
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009222668004012036,
      "loss": 1.7301,
      "theoretical_loss": 4.196604496731818,
      "tokens_seen": 286982144
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000922246740220662,
      "loss": 1.8278,
      "theoretical_loss": 4.196492024592183,
      "tokens_seen": 287047680
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009222266800401203,
      "loss": 1.891,
      "theoretical_loss": 4.196379585316315,
      "tokens_seen": 287113216
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009222066198595788,
      "loss": 1.7694,
      "theoretical_loss": 4.1962671788871155,
      "tokens_seen": 287178752
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009221865596790371,
      "loss": 1.7081,
      "theoretical_loss": 4.196154805287497,
      "tokens_seen": 287244288
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009221664994984955,
      "loss": 1.7177,
      "theoretical_loss": 4.196042464500382,
      "tokens_seen": 287309824
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009221464393179539,
      "loss": 1.7366,
      "theoretical_loss": 4.195930156508711,
      "tokens_seen": 287375360
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009221263791374122,
      "loss": 1.7228,
      "theoretical_loss": 4.195817881295431,
      "tokens_seen": 287440896
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009221063189568706,
      "loss": 1.8829,
      "theoretical_loss": 4.195705638843508,
      "tokens_seen": 287506432
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000922086258776329,
      "loss": 1.7452,
      "theoretical_loss": 4.195593429135916,
      "tokens_seen": 287571968
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009220661985957873,
      "loss": 2.0069,
      "theoretical_loss": 4.195481252155645,
      "tokens_seen": 287637504
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009220461384152458,
      "loss": 1.9142,
      "theoretical_loss": 4.1953691078856945,
      "tokens_seen": 287703040
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000922026078234704,
      "loss": 1.8221,
      "theoretical_loss": 4.19525699630908,
      "tokens_seen": 287768576
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009220060180541625,
      "loss": 1.823,
      "theoretical_loss": 4.195144917408828,
      "tokens_seen": 287834112
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009219859578736209,
      "loss": 1.7773,
      "theoretical_loss": 4.195032871167977,
      "tokens_seen": 287899648
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009219658976930792,
      "loss": 1.706,
      "theoretical_loss": 4.1949208575695796,
      "tokens_seen": 287965184
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009219458375125376,
      "loss": 1.7919,
      "theoretical_loss": 4.1948088765967,
      "tokens_seen": 288030720
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009219257773319961,
      "loss": 1.7851,
      "theoretical_loss": 4.194696928232417,
      "tokens_seen": 288096256
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009219057171514544,
      "loss": 1.7673,
      "theoretical_loss": 4.194585012459818,
      "tokens_seen": 288161792
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009218856569709128,
      "loss": 1.7555,
      "theoretical_loss": 4.194473129262008,
      "tokens_seen": 288227328
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009218655967903711,
      "loss": 1.7409,
      "theoretical_loss": 4.194361278622099,
      "tokens_seen": 288292864
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.4943329095840454,
      "objective/train/docs_used": 111591,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6546730995178223,
      "objective/train/original_loss": 1.6546730995178223,
      "objective/train/theoretical_loss": 4.194249460523222,
      "objective/train/tokens_used": 308818400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2445804327726364,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050676703453064,
      "objective/train/weighted_lm_loss": 1.7384644746780396,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.194249460523222,
      "tokens_seen": 288358400
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009218455366098295,
      "loss": 1.713,
      "theoretical_loss": 4.194249460523222,
      "tokens_seen": 288358400
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000921825476429288,
      "loss": 1.8013,
      "theoretical_loss": 4.194137674948516,
      "tokens_seen": 288423936
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009218054162487462,
      "loss": 1.8124,
      "theoretical_loss": 4.194025921881134,
      "tokens_seen": 288489472
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009217853560682047,
      "loss": 1.7621,
      "theoretical_loss": 4.19391420130424,
      "tokens_seen": 288555008
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000921765295887663,
      "loss": 1.8619,
      "theoretical_loss": 4.193802513201015,
      "tokens_seen": 288620544
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009217452357071214,
      "loss": 1.8636,
      "theoretical_loss": 4.193690857554646,
      "tokens_seen": 288686080
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009217251755265798,
      "loss": 1.9187,
      "theoretical_loss": 4.193579234348338,
      "tokens_seen": 288751616
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009217051153460382,
      "loss": 1.8693,
      "theoretical_loss": 4.193467643565306,
      "tokens_seen": 288817152
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009216850551654965,
      "loss": 1.8022,
      "theoretical_loss": 4.193356085188778,
      "tokens_seen": 288882688
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009216649949849549,
      "loss": 1.7514,
      "theoretical_loss": 4.193244559201995,
      "tokens_seen": 288948224
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009216449348044132,
      "loss": 1.8518,
      "theoretical_loss": 4.193133065588209,
      "tokens_seen": 289013760
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009216248746238717,
      "loss": 2.0078,
      "theoretical_loss": 4.193021604330686,
      "tokens_seen": 289079296
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.00092160481444333,
      "loss": 1.8522,
      "theoretical_loss": 4.1929101754127025,
      "tokens_seen": 289144832
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009215847542627884,
      "loss": 1.7833,
      "theoretical_loss": 4.192798778817551,
      "tokens_seen": 289210368
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009215646940822468,
      "loss": 1.8018,
      "theoretical_loss": 4.192687414528532,
      "tokens_seen": 289275904
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009215446339017052,
      "loss": 1.8112,
      "theoretical_loss": 4.192576082528962,
      "tokens_seen": 289341440
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009215245737211635,
      "loss": 1.8178,
      "theoretical_loss": 4.192464782802167,
      "tokens_seen": 289406976
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009215045135406219,
      "loss": 1.8044,
      "theoretical_loss": 4.192353515331489,
      "tokens_seen": 289472512
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009214844533600802,
      "loss": 1.7338,
      "theoretical_loss": 4.192242280100279,
      "tokens_seen": 289538048
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009214643931795387,
      "loss": 1.8753,
      "theoretical_loss": 4.192131077091901,
      "tokens_seen": 289603584
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000921444332998997,
      "loss": 1.7773,
      "theoretical_loss": 4.192019906289733,
      "tokens_seen": 289669120
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009214242728184554,
      "loss": 1.9292,
      "theoretical_loss": 4.191908767677163,
      "tokens_seen": 289734656
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009214042126379137,
      "loss": 1.7483,
      "theoretical_loss": 4.191797661237594,
      "tokens_seen": 289800192
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009213841524573722,
      "loss": 1.8835,
      "theoretical_loss": 4.191686586954439,
      "tokens_seen": 289865728
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009213640922768305,
      "loss": 1.9106,
      "theoretical_loss": 4.1915755448111245,
      "tokens_seen": 289931264
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009213440320962889,
      "loss": 1.8306,
      "theoretical_loss": 4.191464534791089,
      "tokens_seen": 289996800
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009213239719157473,
      "loss": 1.8738,
      "theoretical_loss": 4.191353556877782,
      "tokens_seen": 290062336
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009213039117352056,
      "loss": 1.7921,
      "theoretical_loss": 4.191242611054668,
      "tokens_seen": 290127872
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000921283851554664,
      "loss": 1.8376,
      "theoretical_loss": 4.191131697305222,
      "tokens_seen": 290193408
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009212637913741224,
      "loss": 1.8509,
      "theoretical_loss": 4.191020815612932,
      "tokens_seen": 290258944
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009212437311935807,
      "loss": 1.8397,
      "theoretical_loss": 4.190909965961296,
      "tokens_seen": 290324480
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009212236710130392,
      "loss": 1.7438,
      "theoretical_loss": 4.190799148333827,
      "tokens_seen": 290390016
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009212036108324975,
      "loss": 1.9381,
      "theoretical_loss": 4.1906883627140505,
      "tokens_seen": 290455552
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009211835506519559,
      "loss": 1.7074,
      "theoretical_loss": 4.1905776090855,
      "tokens_seen": 290521088
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009211634904714143,
      "loss": 1.7709,
      "theoretical_loss": 4.190466887431725,
      "tokens_seen": 290586624
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009211434302908726,
      "loss": 1.8717,
      "theoretical_loss": 4.190356197736288,
      "tokens_seen": 290652160
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000921123370110331,
      "loss": 1.7057,
      "theoretical_loss": 4.19024553998276,
      "tokens_seen": 290717696
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009211033099297895,
      "loss": 1.8125,
      "theoretical_loss": 4.190134914154727,
      "tokens_seen": 290783232
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009210832497492477,
      "loss": 1.889,
      "theoretical_loss": 4.190024320235784,
      "tokens_seen": 290848768
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009210631895687062,
      "loss": 1.8423,
      "theoretical_loss": 4.189913758209543,
      "tokens_seen": 290914304
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009210431293881644,
      "loss": 1.7914,
      "theoretical_loss": 4.189803228059623,
      "tokens_seen": 290979840
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009210230692076229,
      "loss": 1.9284,
      "theoretical_loss": 4.189692729769659,
      "tokens_seen": 291045376
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009210030090270813,
      "loss": 1.9621,
      "theoretical_loss": 4.189582263323296,
      "tokens_seen": 291110912
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009209829488465396,
      "loss": 1.7024,
      "theoretical_loss": 4.1894718287041925,
      "tokens_seen": 291176448
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000920962888665998,
      "loss": 1.7627,
      "theoretical_loss": 4.189361425896016,
      "tokens_seen": 291241984
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009209428284854565,
      "loss": 1.7543,
      "theoretical_loss": 4.18925105488245,
      "tokens_seen": 291307520
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009209227683049147,
      "loss": 1.8712,
      "theoretical_loss": 4.189140715647189,
      "tokens_seen": 291373056
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009209027081243732,
      "loss": 1.7619,
      "theoretical_loss": 4.189030408173936,
      "tokens_seen": 291438592
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009208826479438314,
      "loss": 1.7172,
      "theoretical_loss": 4.188920132446411,
      "tokens_seen": 291504128
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009208625877632899,
      "loss": 1.8162,
      "theoretical_loss": 4.1888098884483425,
      "tokens_seen": 291569664
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.49275243282318115,
      "objective/train/docs_used": 112844,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.677362322807312,
      "objective/train/original_loss": 1.6773622035980225,
      "objective/train/theoretical_loss": 4.188699676163473,
      "objective/train/tokens_used": 312095200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24318408966064453,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505115985870361,
      "objective/train/weighted_lm_loss": 1.7620242834091187,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.188699676163473,
      "tokens_seen": 291635200
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009208425275827483,
      "loss": 1.865,
      "theoretical_loss": 4.188699676163473,
      "tokens_seen": 291635200
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009208224674022066,
      "loss": 1.8001,
      "theoretical_loss": 4.188589495575557,
      "tokens_seen": 291700736
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000920802407221665,
      "loss": 1.7084,
      "theoretical_loss": 4.188479346668359,
      "tokens_seen": 291766272
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009207823470411234,
      "loss": 1.6633,
      "theoretical_loss": 4.188369229425657,
      "tokens_seen": 291831808
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009207622868605817,
      "loss": 1.9012,
      "theoretical_loss": 4.188259143831241,
      "tokens_seen": 291897344
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009207422266800402,
      "loss": 1.8598,
      "theoretical_loss": 4.188149089868912,
      "tokens_seen": 291962880
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009207221664994984,
      "loss": 1.7914,
      "theoretical_loss": 4.188039067522484,
      "tokens_seen": 292028416
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009207021063189569,
      "loss": 1.818,
      "theoretical_loss": 4.1879290767757835,
      "tokens_seen": 292093952
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009206820461384152,
      "loss": 1.8285,
      "theoretical_loss": 4.187819117612645,
      "tokens_seen": 292159488
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009206619859578736,
      "loss": 1.7891,
      "theoretical_loss": 4.18770919001692,
      "tokens_seen": 292225024
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000920641925777332,
      "loss": 1.8172,
      "theoretical_loss": 4.18759929397247,
      "tokens_seen": 292290560
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009206218655967904,
      "loss": 1.8366,
      "theoretical_loss": 4.187489429463166,
      "tokens_seen": 292356096
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009206018054162487,
      "loss": 1.87,
      "theoretical_loss": 4.187379596472894,
      "tokens_seen": 292421632
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009205817452357072,
      "loss": 1.8121,
      "theoretical_loss": 4.18726979498555,
      "tokens_seen": 292487168
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009205616850551655,
      "loss": 1.8013,
      "theoretical_loss": 4.187160024985044,
      "tokens_seen": 292552704
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009205416248746239,
      "loss": 1.8087,
      "theoretical_loss": 4.187050286455294,
      "tokens_seen": 292618240
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009205215646940822,
      "loss": 1.9665,
      "theoretical_loss": 4.1869405793802335,
      "tokens_seen": 292683776
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009205015045135406,
      "loss": 1.8839,
      "theoretical_loss": 4.1868309037438065,
      "tokens_seen": 292749312
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000920481444332999,
      "loss": 1.7737,
      "theoretical_loss": 4.1867212595299685,
      "tokens_seen": 292814848
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009204613841524574,
      "loss": 1.7755,
      "theoretical_loss": 4.186611646722687,
      "tokens_seen": 292880384
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009204413239719157,
      "loss": 1.9252,
      "theoretical_loss": 4.186502065305942,
      "tokens_seen": 292945920
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009204212637913741,
      "loss": 1.7296,
      "theoretical_loss": 4.186392515263723,
      "tokens_seen": 293011456
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009204012036108326,
      "loss": 1.8694,
      "theoretical_loss": 4.186282996580034,
      "tokens_seen": 293076992
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009203811434302909,
      "loss": 1.9082,
      "theoretical_loss": 4.1861735092388885,
      "tokens_seen": 293142528
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009203610832497493,
      "loss": 1.9323,
      "theoretical_loss": 4.186064053224314,
      "tokens_seen": 293208064
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009203410230692077,
      "loss": 1.7652,
      "theoretical_loss": 4.185954628520347,
      "tokens_seen": 293273600
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000920320962888666,
      "loss": 1.8631,
      "theoretical_loss": 4.185845235111037,
      "tokens_seen": 293339136
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009203009027081244,
      "loss": 1.7549,
      "theoretical_loss": 4.185735872980447,
      "tokens_seen": 293404672
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009202808425275828,
      "loss": 1.927,
      "theoretical_loss": 4.185626542112649,
      "tokens_seen": 293470208
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009202607823470411,
      "loss": 1.7839,
      "theoretical_loss": 4.185517242491728,
      "tokens_seen": 293535744
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009202407221664996,
      "loss": 1.8171,
      "theoretical_loss": 4.185407974101779,
      "tokens_seen": 293601280
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009202206619859579,
      "loss": 1.8146,
      "theoretical_loss": 4.185298736926912,
      "tokens_seen": 293666816
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009202006018054163,
      "loss": 1.8778,
      "theoretical_loss": 4.185189530951245,
      "tokens_seen": 293732352
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009201805416248747,
      "loss": 1.6603,
      "theoretical_loss": 4.18508035615891,
      "tokens_seen": 293797888
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000920160481444333,
      "loss": 1.813,
      "theoretical_loss": 4.184971212534048,
      "tokens_seen": 293863424
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009201404212637914,
      "loss": 1.8153,
      "theoretical_loss": 4.184862100060816,
      "tokens_seen": 293928960
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009201203610832499,
      "loss": 1.8382,
      "theoretical_loss": 4.184753018723379,
      "tokens_seen": 293994496
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009201003009027081,
      "loss": 1.7472,
      "theoretical_loss": 4.184643968505914,
      "tokens_seen": 294060032
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009200802407221666,
      "loss": 1.8905,
      "theoretical_loss": 4.184534949392611,
      "tokens_seen": 294125568
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009200601805416248,
      "loss": 1.7706,
      "theoretical_loss": 4.18442596136767,
      "tokens_seen": 294191104
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009200401203610833,
      "loss": 1.8976,
      "theoretical_loss": 4.184317004415303,
      "tokens_seen": 294256640
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009200200601805417,
      "loss": 1.8011,
      "theoretical_loss": 4.184208078519735,
      "tokens_seen": 294322176
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.00092,
      "loss": 1.7645,
      "theoretical_loss": 4.184099183665199,
      "tokens_seen": 294387712
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009199799398194584,
      "loss": 1.9067,
      "theoretical_loss": 4.183990319835944,
      "tokens_seen": 294453248
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009199598796389169,
      "loss": 1.7968,
      "theoretical_loss": 4.183881487016227,
      "tokens_seen": 294518784
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009199398194583751,
      "loss": 1.856,
      "theoretical_loss": 4.183772685190318,
      "tokens_seen": 294584320
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009199197592778336,
      "loss": 1.925,
      "theoretical_loss": 4.1836639143425,
      "tokens_seen": 294649856
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009198996990972918,
      "loss": 1.787,
      "theoretical_loss": 4.183555174457063,
      "tokens_seen": 294715392
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009198796389167503,
      "loss": 1.7992,
      "theoretical_loss": 4.183446465518312,
      "tokens_seen": 294780928
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009198595787362087,
      "loss": 1.7961,
      "theoretical_loss": 4.183337787510563,
      "tokens_seen": 294846464
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.4940253496170044,
      "objective/train/docs_used": 114046,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7510039806365967,
      "objective/train/original_loss": 1.7510039806365967,
      "objective/train/theoretical_loss": 4.1832291404181445,
      "objective/train/tokens_used": 315372000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24425998330116272,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506443977355957,
      "objective/train/weighted_lm_loss": 1.83975088596344,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 4.1832291404181445,
      "tokens_seen": 294912000
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000919839518555667,
      "loss": 1.8329,
      "theoretical_loss": 4.1832291404181445,
      "tokens_seen": 294912000
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009198194583751254,
      "loss": 1.7224,
      "theoretical_loss": 4.183120524225393,
      "tokens_seen": 294977536
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009197993981945838,
      "loss": 1.9731,
      "theoretical_loss": 4.183011938916657,
      "tokens_seen": 295043072
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009197793380140421,
      "loss": 1.7446,
      "theoretical_loss": 4.182903384476301,
      "tokens_seen": 295108608
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009197592778335006,
      "loss": 1.7929,
      "theoretical_loss": 4.182794860888696,
      "tokens_seen": 295174144
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009197392176529588,
      "loss": 1.8788,
      "theoretical_loss": 4.182686368138226,
      "tokens_seen": 295239680
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009197191574724173,
      "loss": 1.8534,
      "theoretical_loss": 4.182577906209287,
      "tokens_seen": 295305216
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009196990972918756,
      "loss": 1.7757,
      "theoretical_loss": 4.182469475086286,
      "tokens_seen": 295370752
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000919679037111334,
      "loss": 1.8442,
      "theoretical_loss": 4.18236107475364,
      "tokens_seen": 295436288
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009196589769307924,
      "loss": 1.9694,
      "theoretical_loss": 4.182252705195778,
      "tokens_seen": 295501824
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009196389167502508,
      "loss": 1.6922,
      "theoretical_loss": 4.182144366397143,
      "tokens_seen": 295567360
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009196188565697091,
      "loss": 1.8983,
      "theoretical_loss": 4.182036058342186,
      "tokens_seen": 295632896
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009195987963891676,
      "loss": 1.8164,
      "theoretical_loss": 4.18192778101537,
      "tokens_seen": 295698432
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009195787362086259,
      "loss": 1.8424,
      "theoretical_loss": 4.1818195344011695,
      "tokens_seen": 295763968
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009195586760280843,
      "loss": 1.7505,
      "theoretical_loss": 4.181711318484071,
      "tokens_seen": 295829504
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009195386158475426,
      "loss": 1.769,
      "theoretical_loss": 4.181603133248572,
      "tokens_seen": 295895040
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000919518555667001,
      "loss": 1.7624,
      "theoretical_loss": 4.181494978679181,
      "tokens_seen": 295960576
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009194984954864594,
      "loss": 1.7339,
      "theoretical_loss": 4.181386854760417,
      "tokens_seen": 296026112
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009194784353059178,
      "loss": 1.8652,
      "theoretical_loss": 4.181278761476811,
      "tokens_seen": 296091648
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009194583751253761,
      "loss": 1.7627,
      "theoretical_loss": 4.181170698812906,
      "tokens_seen": 296157184
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009194383149448345,
      "loss": 1.8077,
      "theoretical_loss": 4.181062666753256,
      "tokens_seen": 296222720
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009194182547642929,
      "loss": 1.8082,
      "theoretical_loss": 4.180954665282423,
      "tokens_seen": 296288256
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009193981945837513,
      "loss": 1.7497,
      "theoretical_loss": 4.180846694384986,
      "tokens_seen": 296353792
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009193781344032096,
      "loss": 1.8343,
      "theoretical_loss": 4.18073875404553,
      "tokens_seen": 296419328
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000919358074222668,
      "loss": 1.7863,
      "theoretical_loss": 4.180630844248653,
      "tokens_seen": 296484864
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009193380140421263,
      "loss": 1.8319,
      "theoretical_loss": 4.180522964978966,
      "tokens_seen": 296550400
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009193179538615848,
      "loss": 1.7604,
      "theoretical_loss": 4.180415116221088,
      "tokens_seen": 296615936
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009192978936810431,
      "loss": 1.9332,
      "theoretical_loss": 4.180307297959651,
      "tokens_seen": 296681472
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009192778335005015,
      "loss": 1.8325,
      "theoretical_loss": 4.180199510179299,
      "tokens_seen": 296747008
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009192577733199599,
      "loss": 1.6463,
      "theoretical_loss": 4.180091752864684,
      "tokens_seen": 296812544
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009192377131394183,
      "loss": 1.8372,
      "theoretical_loss": 4.179984026000473,
      "tokens_seen": 296878080
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009192176529588766,
      "loss": 1.7951,
      "theoretical_loss": 4.179876329571341,
      "tokens_seen": 296943616
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009191975927783351,
      "loss": 1.867,
      "theoretical_loss": 4.179768663561975,
      "tokens_seen": 297009152
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009191775325977933,
      "loss": 1.7988,
      "theoretical_loss": 4.179661027957074,
      "tokens_seen": 297074688
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009191574724172518,
      "loss": 1.8474,
      "theoretical_loss": 4.179553422741347,
      "tokens_seen": 297140224
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009191374122367101,
      "loss": 1.725,
      "theoretical_loss": 4.179445847899515,
      "tokens_seen": 297205760
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009191173520561685,
      "loss": 1.603,
      "theoretical_loss": 4.1793383034163085,
      "tokens_seen": 297271296
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009190972918756269,
      "loss": 1.6909,
      "theoretical_loss": 4.17923078927647,
      "tokens_seen": 297336832
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009190772316950852,
      "loss": 1.7892,
      "theoretical_loss": 4.1791233054647545,
      "tokens_seen": 297402368
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009190571715145436,
      "loss": 1.8126,
      "theoretical_loss": 4.179015851965926,
      "tokens_seen": 297467904
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009190371113340021,
      "loss": 1.7874,
      "theoretical_loss": 4.178908428764759,
      "tokens_seen": 297533440
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009190170511534603,
      "loss": 1.7865,
      "theoretical_loss": 4.178801035846041,
      "tokens_seen": 297598976
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009189969909729188,
      "loss": 1.81,
      "theoretical_loss": 4.178693673194569,
      "tokens_seen": 297664512
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000918976930792377,
      "loss": 1.7445,
      "theoretical_loss": 4.178586340795152,
      "tokens_seen": 297730048
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009189568706118355,
      "loss": 1.8662,
      "theoretical_loss": 4.17847903863261,
      "tokens_seen": 297795584
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009189368104312939,
      "loss": 1.7746,
      "theoretical_loss": 4.178371766691773,
      "tokens_seen": 297861120
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009189167502507522,
      "loss": 1.7787,
      "theoretical_loss": 4.1782645249574815,
      "tokens_seen": 297926656
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009188966900702107,
      "loss": 1.8718,
      "theoretical_loss": 4.178157313414589,
      "tokens_seen": 297992192
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009188766298896691,
      "loss": 1.8258,
      "theoretical_loss": 4.178050132047958,
      "tokens_seen": 298057728
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009188565697091274,
      "loss": 1.7314,
      "theoretical_loss": 4.177942980842464,
      "tokens_seen": 298123264
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.4918665587902069,
      "objective/train/docs_used": 115247,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.66420578956604,
      "objective/train/original_loss": 1.66420578956604,
      "objective/train/theoretical_loss": 4.1778358597829905,
      "objective/train/tokens_used": 318648800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2422984093427658,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504183769226074,
      "objective/train/weighted_lm_loss": 1.7481639385223389,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.1778358597829905,
      "tokens_seen": 298188800
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009188365095285858,
      "loss": 1.825,
      "theoretical_loss": 4.1778358597829905,
      "tokens_seen": 298188800
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009188164493480442,
      "loss": 1.7152,
      "theoretical_loss": 4.1777287688544344,
      "tokens_seen": 298254336
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009187963891675025,
      "loss": 1.8607,
      "theoretical_loss": 4.177621708041703,
      "tokens_seen": 298319872
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000918776328986961,
      "loss": 1.8997,
      "theoretical_loss": 4.177514677329713,
      "tokens_seen": 298385408
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009187562688064192,
      "loss": 1.8691,
      "theoretical_loss": 4.177407676703393,
      "tokens_seen": 298450944
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009187362086258777,
      "loss": 1.7494,
      "theoretical_loss": 4.177300706147683,
      "tokens_seen": 298516480
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000918716148445336,
      "loss": 1.8863,
      "theoretical_loss": 4.177193765647534,
      "tokens_seen": 298582016
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009186960882647944,
      "loss": 1.7996,
      "theoretical_loss": 4.1770868551879055,
      "tokens_seen": 298647552
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009186760280842528,
      "loss": 1.8353,
      "theoretical_loss": 4.17697997475377,
      "tokens_seen": 298713088
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009186559679037112,
      "loss": 1.8755,
      "theoretical_loss": 4.176873124330111,
      "tokens_seen": 298778624
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009186359077231695,
      "loss": 2.0181,
      "theoretical_loss": 4.176766303901922,
      "tokens_seen": 298844160
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000918615847542628,
      "loss": 2.0517,
      "theoretical_loss": 4.176659513454207,
      "tokens_seen": 298909696
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009185957873620863,
      "loss": 1.8836,
      "theoretical_loss": 4.176552752971981,
      "tokens_seen": 298975232
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009185757271815447,
      "loss": 1.875,
      "theoretical_loss": 4.1764460224402695,
      "tokens_seen": 299040768
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000918555667001003,
      "loss": 1.8465,
      "theoretical_loss": 4.17633932184411,
      "tokens_seen": 299106304
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009185356068204614,
      "loss": 1.9421,
      "theoretical_loss": 4.176232651168551,
      "tokens_seen": 299171840
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009185155466399198,
      "loss": 1.8236,
      "theoretical_loss": 4.176126010398647,
      "tokens_seen": 299237376
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009184954864593782,
      "loss": 1.9052,
      "theoretical_loss": 4.176019399519471,
      "tokens_seen": 299302912
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009184754262788365,
      "loss": 1.8264,
      "theoretical_loss": 4.1759128185161005,
      "tokens_seen": 299368448
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009184553660982949,
      "loss": 1.7301,
      "theoretical_loss": 4.175806267373627,
      "tokens_seen": 299433984
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009184353059177533,
      "loss": 1.8618,
      "theoretical_loss": 4.17569974607715,
      "tokens_seen": 299499520
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009184152457372117,
      "loss": 1.7564,
      "theoretical_loss": 4.1755932546117815,
      "tokens_seen": 299565056
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.00091839518555667,
      "loss": 1.7976,
      "theoretical_loss": 4.175486792962646,
      "tokens_seen": 299630592
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009183751253761285,
      "loss": 1.8856,
      "theoretical_loss": 4.175380361114874,
      "tokens_seen": 299696128
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009183550651955867,
      "loss": 1.7783,
      "theoretical_loss": 4.1752739590536105,
      "tokens_seen": 299761664
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009183350050150452,
      "loss": 1.7971,
      "theoretical_loss": 4.17516758676401,
      "tokens_seen": 299827200
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009183149448345035,
      "loss": 1.924,
      "theoretical_loss": 4.175061244231237,
      "tokens_seen": 299892736
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009182948846539619,
      "loss": 1.9111,
      "theoretical_loss": 4.174954931440468,
      "tokens_seen": 299958272
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009182748244734203,
      "loss": 1.947,
      "theoretical_loss": 4.174848648376889,
      "tokens_seen": 300023808
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009182547642928787,
      "loss": 1.8443,
      "theoretical_loss": 4.174742395025696,
      "tokens_seen": 300089344
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000918234704112337,
      "loss": 1.8847,
      "theoretical_loss": 4.174636171372097,
      "tokens_seen": 300154880
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009182146439317955,
      "loss": 1.8018,
      "theoretical_loss": 4.174529977401312,
      "tokens_seen": 300220416
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009181945837512537,
      "loss": 1.7974,
      "theoretical_loss": 4.174423813098566,
      "tokens_seen": 300285952
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009181745235707122,
      "loss": 1.8399,
      "theoretical_loss": 4.174317678449101,
      "tokens_seen": 300351488
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009181544633901705,
      "loss": 1.7105,
      "theoretical_loss": 4.174211573438166,
      "tokens_seen": 300417024
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009181344032096289,
      "loss": 1.9141,
      "theoretical_loss": 4.174105498051022,
      "tokens_seen": 300482560
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009181143430290873,
      "loss": 1.9205,
      "theoretical_loss": 4.173999452272939,
      "tokens_seen": 300548096
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009180942828485456,
      "loss": 1.9612,
      "theoretical_loss": 4.173893436089199,
      "tokens_seen": 300613632
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000918074222668004,
      "loss": 1.92,
      "theoretical_loss": 4.173787449485094,
      "tokens_seen": 300679168
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009180541624874625,
      "loss": 2.085,
      "theoretical_loss": 4.173681492445926,
      "tokens_seen": 300744704
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009180341023069207,
      "loss": 1.9086,
      "theoretical_loss": 4.173575564957009,
      "tokens_seen": 300810240
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009180140421263792,
      "loss": 1.8754,
      "theoretical_loss": 4.1734696670036655,
      "tokens_seen": 300875776
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009179939819458374,
      "loss": 2.1293,
      "theoretical_loss": 4.17336379857123,
      "tokens_seen": 300941312
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009179739217652959,
      "loss": 1.9747,
      "theoretical_loss": 4.1732579596450465,
      "tokens_seen": 301006848
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009179538615847543,
      "loss": 1.8285,
      "theoretical_loss": 4.17315215021047,
      "tokens_seen": 301072384
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009179338014042126,
      "loss": 2.0572,
      "theoretical_loss": 4.173046370252866,
      "tokens_seen": 301137920
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000917913741223671,
      "loss": 1.832,
      "theoretical_loss": 4.172940619757611,
      "tokens_seen": 301203456
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009178936810431295,
      "loss": 1.8726,
      "theoretical_loss": 4.172834898710089,
      "tokens_seen": 301268992
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009178736208625877,
      "loss": 1.8577,
      "theoretical_loss": 4.1727292070957,
      "tokens_seen": 301334528
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009178535606820462,
      "loss": 1.934,
      "theoretical_loss": 4.172623544899849,
      "tokens_seen": 301400064
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.4897353947162628,
      "objective/train/docs_used": 116418,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8441132307052612,
      "objective/train/original_loss": 1.8441133499145508,
      "objective/train/theoretical_loss": 4.172517912107954,
      "objective/train/tokens_used": 321925600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24035406112670898,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050195336341858,
      "objective/train/weighted_lm_loss": 1.9364567995071411,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.172517912107954,
      "tokens_seen": 301465600
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009178335005015045,
      "loss": 2.0693,
      "theoretical_loss": 4.172517912107954,
      "tokens_seen": 301465600
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009178134403209629,
      "loss": 1.8463,
      "theoretical_loss": 4.172412308705443,
      "tokens_seen": 301531136
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009177933801404213,
      "loss": 1.8559,
      "theoretical_loss": 4.172306734677753,
      "tokens_seen": 301596672
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009177733199598796,
      "loss": 1.9022,
      "theoretical_loss": 4.172201190010334,
      "tokens_seen": 301662208
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000917753259779338,
      "loss": 1.814,
      "theoretical_loss": 4.172095674688645,
      "tokens_seen": 301727744
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009177331995987964,
      "loss": 2.0133,
      "theoretical_loss": 4.171990188698155,
      "tokens_seen": 301793280
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009177131394182547,
      "loss": 1.8285,
      "theoretical_loss": 4.171884732024344,
      "tokens_seen": 301858816
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009176930792377132,
      "loss": 1.8609,
      "theoretical_loss": 4.171779304652702,
      "tokens_seen": 301924352
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009176730190571715,
      "loss": 1.8516,
      "theoretical_loss": 4.171673906568729,
      "tokens_seen": 301989888
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009176529588766299,
      "loss": 1.8591,
      "theoretical_loss": 4.171568537757936,
      "tokens_seen": 302055424
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009176328986960883,
      "loss": 2.047,
      "theoretical_loss": 4.171463198205844,
      "tokens_seen": 302120960
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009176128385155467,
      "loss": 1.8739,
      "theoretical_loss": 4.171357887897985,
      "tokens_seen": 302186496
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000917592778335005,
      "loss": 1.8573,
      "theoretical_loss": 4.171252606819899,
      "tokens_seen": 302252032
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009175727181544634,
      "loss": 1.9282,
      "theoretical_loss": 4.17114735495714,
      "tokens_seen": 302317568
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009175526579739217,
      "loss": 1.8217,
      "theoretical_loss": 4.171042132295268,
      "tokens_seen": 302383104
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009175325977933802,
      "loss": 1.8833,
      "theoretical_loss": 4.170936938819857,
      "tokens_seen": 302448640
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009175125376128385,
      "loss": 1.779,
      "theoretical_loss": 4.170831774516489,
      "tokens_seen": 302514176
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009174924774322969,
      "loss": 1.8598,
      "theoretical_loss": 4.170726639370757,
      "tokens_seen": 302579712
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009174724172517552,
      "loss": 1.855,
      "theoretical_loss": 4.170621533368265,
      "tokens_seen": 302645248
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009174523570712137,
      "loss": 1.8664,
      "theoretical_loss": 4.170516456494624,
      "tokens_seen": 302710784
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000917432296890672,
      "loss": 1.853,
      "theoretical_loss": 4.170411408735461,
      "tokens_seen": 302776320
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009174122367101304,
      "loss": 1.7922,
      "theoretical_loss": 4.170306390076408,
      "tokens_seen": 302841856
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009173921765295889,
      "loss": 2.0018,
      "theoretical_loss": 4.1702014005031085,
      "tokens_seen": 302907392
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009173721163490471,
      "loss": 1.7791,
      "theoretical_loss": 4.170096440001217,
      "tokens_seen": 302972928
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009173520561685056,
      "loss": 1.9441,
      "theoretical_loss": 4.169991508556398,
      "tokens_seen": 303038464
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009173319959879639,
      "loss": 1.8292,
      "theoretical_loss": 4.169886606154328,
      "tokens_seen": 303104000
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009173119358074223,
      "loss": 1.8355,
      "theoretical_loss": 4.169781732780688,
      "tokens_seen": 303169536
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009172918756268807,
      "loss": 1.861,
      "theoretical_loss": 4.169676888421175,
      "tokens_seen": 303235072
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009172718154463391,
      "loss": 1.7574,
      "theoretical_loss": 4.169572073061493,
      "tokens_seen": 303300608
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009172517552657974,
      "loss": 1.9617,
      "theoretical_loss": 4.169467286687359,
      "tokens_seen": 303366144
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009172316950852559,
      "loss": 1.8277,
      "theoretical_loss": 4.169362529284497,
      "tokens_seen": 303431680
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009172116349047141,
      "loss": 1.8347,
      "theoretical_loss": 4.169257800838642,
      "tokens_seen": 303497216
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009171915747241726,
      "loss": 2.0021,
      "theoretical_loss": 4.16915310133554,
      "tokens_seen": 303562752
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000917171514543631,
      "loss": 1.9621,
      "theoretical_loss": 4.169048430760946,
      "tokens_seen": 303628288
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009171514543630893,
      "loss": 1.8753,
      "theoretical_loss": 4.168943789100626,
      "tokens_seen": 303693824
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009171313941825477,
      "loss": 1.8549,
      "theoretical_loss": 4.168839176340356,
      "tokens_seen": 303759360
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000917111334002006,
      "loss": 1.7648,
      "theoretical_loss": 4.1687345924659205,
      "tokens_seen": 303824896
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009170912738214644,
      "loss": 1.7129,
      "theoretical_loss": 4.168630037463117,
      "tokens_seen": 303890432
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009170712136409229,
      "loss": 1.8143,
      "theoretical_loss": 4.168525511317751,
      "tokens_seen": 303955968
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009170511534603811,
      "loss": 1.8852,
      "theoretical_loss": 4.168421014015637,
      "tokens_seen": 304021504
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009170310932798396,
      "loss": 1.8363,
      "theoretical_loss": 4.168316545542602,
      "tokens_seen": 304087040
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009170110330992978,
      "loss": 1.8495,
      "theoretical_loss": 4.168212105884482,
      "tokens_seen": 304152576
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009169909729187563,
      "loss": 1.8456,
      "theoretical_loss": 4.168107695027123,
      "tokens_seen": 304218112
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009169709127382147,
      "loss": 1.7831,
      "theoretical_loss": 4.1680033129563805,
      "tokens_seen": 304283648
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000916950852557673,
      "loss": 1.75,
      "theoretical_loss": 4.167898959658121,
      "tokens_seen": 304349184
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009169307923771314,
      "loss": 1.7698,
      "theoretical_loss": 4.167794635118219,
      "tokens_seen": 304414720
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009169107321965899,
      "loss": 1.8279,
      "theoretical_loss": 4.167690339322562,
      "tokens_seen": 304480256
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009168906720160481,
      "loss": 1.6521,
      "theoretical_loss": 4.167586072257047,
      "tokens_seen": 304545792
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009168706118355066,
      "loss": 1.7137,
      "theoretical_loss": 4.167481833907576,
      "tokens_seen": 304611328
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009168505516549649,
      "loss": 1.931,
      "theoretical_loss": 4.167377624260068,
      "tokens_seen": 304676864
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.4910045862197876,
      "objective/train/docs_used": 117554,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6303279399871826,
      "objective/train/original_loss": 1.6303277015686035,
      "objective/train/theoretical_loss": 4.167273443300447,
      "objective/train/tokens_used": 325202400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.241532102227211,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050328254699707,
      "objective/train/weighted_lm_loss": 1.7122821807861328,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.167273443300447,
      "tokens_seen": 304742400
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009168304914744233,
      "loss": 1.7162,
      "theoretical_loss": 4.167273443300447,
      "tokens_seen": 304742400
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009168104312938817,
      "loss": 1.9333,
      "theoretical_loss": 4.167169291014649,
      "tokens_seen": 304807936
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.00091679037111334,
      "loss": 2.0274,
      "theoretical_loss": 4.16706516738862,
      "tokens_seen": 304873472
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009167703109327984,
      "loss": 1.7796,
      "theoretical_loss": 4.1669610724083155,
      "tokens_seen": 304939008
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009167502507522568,
      "loss": 1.8187,
      "theoretical_loss": 4.1668570060597006,
      "tokens_seen": 305004544
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009167301905717151,
      "loss": 1.8278,
      "theoretical_loss": 4.16675296832875,
      "tokens_seen": 305070080
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009167101303911736,
      "loss": 1.7686,
      "theoretical_loss": 4.166648959201449,
      "tokens_seen": 305135616
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009166900702106319,
      "loss": 1.9126,
      "theoretical_loss": 4.1665449786637945,
      "tokens_seen": 305201152
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009166700100300903,
      "loss": 1.8431,
      "theoretical_loss": 4.166441026701788,
      "tokens_seen": 305266688
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009166499498495487,
      "loss": 1.8467,
      "theoretical_loss": 4.166337103301447,
      "tokens_seen": 305332224
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000916629889669007,
      "loss": 1.8731,
      "theoretical_loss": 4.166233208448794,
      "tokens_seen": 305397760
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009166098294884654,
      "loss": 1.8152,
      "theoretical_loss": 4.166129342129865,
      "tokens_seen": 305463296
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009165897693079238,
      "loss": 1.8298,
      "theoretical_loss": 4.166025504330703,
      "tokens_seen": 305528832
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009165697091273821,
      "loss": 1.8831,
      "theoretical_loss": 4.165921695037363,
      "tokens_seen": 305594368
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009165496489468406,
      "loss": 1.8718,
      "theoretical_loss": 4.165817914235908,
      "tokens_seen": 305659904
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009165295887662989,
      "loss": 1.8804,
      "theoretical_loss": 4.165714161912412,
      "tokens_seen": 305725440
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009165095285857573,
      "loss": 1.7289,
      "theoretical_loss": 4.165610438052958,
      "tokens_seen": 305790976
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009164894684052156,
      "loss": 1.7683,
      "theoretical_loss": 4.1655067426436405,
      "tokens_seen": 305856512
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009164694082246741,
      "loss": 1.8967,
      "theoretical_loss": 4.165403075670562,
      "tokens_seen": 305922048
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009164493480441324,
      "loss": 1.8074,
      "theoretical_loss": 4.165299437119834,
      "tokens_seen": 305987584
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009164292878635908,
      "loss": 1.8561,
      "theoretical_loss": 4.1651958269775795,
      "tokens_seen": 306053120
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009164092276830491,
      "loss": 1.7487,
      "theoretical_loss": 4.165092245229932,
      "tokens_seen": 306118656
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009163891675025075,
      "loss": 1.7637,
      "theoretical_loss": 4.164988691863032,
      "tokens_seen": 306184192
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009163691073219659,
      "loss": 1.9437,
      "theoretical_loss": 4.164885166863032,
      "tokens_seen": 306249728
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009163490471414243,
      "loss": 1.8181,
      "theoretical_loss": 4.164781670216092,
      "tokens_seen": 306315264
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009163289869608826,
      "loss": 1.7637,
      "theoretical_loss": 4.1646782019083854,
      "tokens_seen": 306380800
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009163089267803411,
      "loss": 1.6582,
      "theoretical_loss": 4.164574761926092,
      "tokens_seen": 306446336
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009162888665997994,
      "loss": 1.76,
      "theoretical_loss": 4.164471350255402,
      "tokens_seen": 306511872
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009162688064192578,
      "loss": 2.0577,
      "theoretical_loss": 4.164367966882517,
      "tokens_seen": 306577408
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009162487462387162,
      "loss": 1.574,
      "theoretical_loss": 4.164264611793644,
      "tokens_seen": 306642944
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009162286860581745,
      "loss": 1.8299,
      "theoretical_loss": 4.164161284975005,
      "tokens_seen": 306708480
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009162086258776329,
      "loss": 1.7307,
      "theoretical_loss": 4.164057986412828,
      "tokens_seen": 306774016
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009161885656970913,
      "loss": 1.8212,
      "theoretical_loss": 4.163954716093354,
      "tokens_seen": 306839552
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009161685055165496,
      "loss": 1.8538,
      "theoretical_loss": 4.163851474002829,
      "tokens_seen": 306905088
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009161484453360081,
      "loss": 1.7853,
      "theoretical_loss": 4.1637482601275115,
      "tokens_seen": 306970624
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009161283851554663,
      "loss": 1.8917,
      "theoretical_loss": 4.16364507445367,
      "tokens_seen": 307036160
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009161083249749248,
      "loss": 1.831,
      "theoretical_loss": 4.163541916967581,
      "tokens_seen": 307101696
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009160882647943832,
      "loss": 1.9265,
      "theoretical_loss": 4.163438787655533,
      "tokens_seen": 307167232
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009160682046138415,
      "loss": 1.8527,
      "theoretical_loss": 4.163335686503822,
      "tokens_seen": 307232768
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009160481444332999,
      "loss": 1.868,
      "theoretical_loss": 4.163232613498753,
      "tokens_seen": 307298304
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009160280842527582,
      "loss": 1.7818,
      "theoretical_loss": 4.163129568626642,
      "tokens_seen": 307363840
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009160080240722166,
      "loss": 1.7847,
      "theoretical_loss": 4.1630265518738145,
      "tokens_seen": 307429376
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009159879638916751,
      "loss": 1.8922,
      "theoretical_loss": 4.162923563226607,
      "tokens_seen": 307494912
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009159679037111333,
      "loss": 1.7713,
      "theoretical_loss": 4.162820602671362,
      "tokens_seen": 307560448
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009159478435305918,
      "loss": 1.8753,
      "theoretical_loss": 4.162717670194433,
      "tokens_seen": 307625984
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009159277833500502,
      "loss": 1.673,
      "theoretical_loss": 4.162614765782186,
      "tokens_seen": 307691520
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009159077231695085,
      "loss": 1.805,
      "theoretical_loss": 4.1625118894209905,
      "tokens_seen": 307757056
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000915887662988967,
      "loss": 1.7618,
      "theoretical_loss": 4.1624090410972325,
      "tokens_seen": 307822592
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009158676028084253,
      "loss": 1.7028,
      "theoretical_loss": 4.162306220797301,
      "tokens_seen": 307888128
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009158475426278837,
      "loss": 1.8686,
      "theoretical_loss": 4.162203428507601,
      "tokens_seen": 307953664
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.49156174063682556,
      "objective/train/docs_used": 118729,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7690374851226807,
      "objective/train/original_loss": 1.7690374851226807,
      "objective/train/theoretical_loss": 4.16210066421454,
      "objective/train/tokens_used": 328479200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2423403114080429,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050388216972351,
      "objective/train/weighted_lm_loss": 1.858426570892334,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.16210066421454,
      "tokens_seen": 308019200
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009158274824473421,
      "loss": 1.7856,
      "theoretical_loss": 4.16210066421454,
      "tokens_seen": 308019200
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009158074222668004,
      "loss": 1.8842,
      "theoretical_loss": 4.16199792790454,
      "tokens_seen": 308084736
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009157873620862588,
      "loss": 1.8463,
      "theoretical_loss": 4.161895219564032,
      "tokens_seen": 308150272
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009157673019057172,
      "loss": 1.7313,
      "theoretical_loss": 4.161792539179454,
      "tokens_seen": 308215808
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009157472417251755,
      "loss": 1.7317,
      "theoretical_loss": 4.161689886737255,
      "tokens_seen": 308281344
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000915727181544634,
      "loss": 1.9005,
      "theoretical_loss": 4.161587262223895,
      "tokens_seen": 308346880
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009157071213640923,
      "loss": 1.7685,
      "theoretical_loss": 4.161484665625839,
      "tokens_seen": 308412416
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009156870611835507,
      "loss": 1.8842,
      "theoretical_loss": 4.161382096929565,
      "tokens_seen": 308477952
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009156670010030091,
      "loss": 1.8179,
      "theoretical_loss": 4.161279556121562,
      "tokens_seen": 308543488
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009156469408224675,
      "loss": 1.8155,
      "theoretical_loss": 4.1611770431883235,
      "tokens_seen": 308609024
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009156268806419258,
      "loss": 1.8621,
      "theoretical_loss": 4.161074558116356,
      "tokens_seen": 308674560
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009156068204613842,
      "loss": 1.8453,
      "theoretical_loss": 4.160972100892174,
      "tokens_seen": 308740096
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009155867602808425,
      "loss": 1.737,
      "theoretical_loss": 4.160869671502302,
      "tokens_seen": 308805632
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000915566700100301,
      "loss": 1.767,
      "theoretical_loss": 4.160767269933274,
      "tokens_seen": 308871168
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009155466399197593,
      "loss": 1.814,
      "theoretical_loss": 4.160664896171632,
      "tokens_seen": 308936704
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009155265797392177,
      "loss": 1.8207,
      "theoretical_loss": 4.160562550203928,
      "tokens_seen": 309002240
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000915506519558676,
      "loss": 1.9213,
      "theoretical_loss": 4.160460232016725,
      "tokens_seen": 309067776
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009154864593781345,
      "loss": 1.8559,
      "theoretical_loss": 4.160357941596594,
      "tokens_seen": 309133312
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009154663991975928,
      "loss": 1.7862,
      "theoretical_loss": 4.160255678930115,
      "tokens_seen": 309198848
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009154463390170512,
      "loss": 1.8416,
      "theoretical_loss": 4.1601534440038765,
      "tokens_seen": 309264384
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009154262788365095,
      "loss": 1.8704,
      "theoretical_loss": 4.16005123680448,
      "tokens_seen": 309329920
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009154062186559679,
      "loss": 1.7641,
      "theoretical_loss": 4.159949057318532,
      "tokens_seen": 309395456
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009153861584754263,
      "loss": 1.7451,
      "theoretical_loss": 4.1598469055326515,
      "tokens_seen": 309460992
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009153660982948847,
      "loss": 1.7089,
      "theoretical_loss": 4.159744781433464,
      "tokens_seen": 309526528
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000915346038114343,
      "loss": 1.8765,
      "theoretical_loss": 4.159642685007606,
      "tokens_seen": 309592064
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009153259779338015,
      "loss": 1.8021,
      "theoretical_loss": 4.159540616241723,
      "tokens_seen": 309657600
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009153059177532598,
      "loss": 1.7355,
      "theoretical_loss": 4.159438575122471,
      "tokens_seen": 309723136
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009152858575727182,
      "loss": 1.7621,
      "theoretical_loss": 4.159336561636513,
      "tokens_seen": 309788672
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009152657973921766,
      "loss": 1.7003,
      "theoretical_loss": 4.1592345757705225,
      "tokens_seen": 309854208
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009152457372116349,
      "loss": 1.6826,
      "theoretical_loss": 4.159132617511181,
      "tokens_seen": 309919744
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009152256770310933,
      "loss": 1.7535,
      "theoretical_loss": 4.159030686845182,
      "tokens_seen": 309985280
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009152056168505517,
      "loss": 1.776,
      "theoretical_loss": 4.158928783759226,
      "tokens_seen": 310050816
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.00091518555667001,
      "loss": 1.7716,
      "theoretical_loss": 4.158826908240022,
      "tokens_seen": 310116352
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009151654964894685,
      "loss": 1.7732,
      "theoretical_loss": 4.158725060274291,
      "tokens_seen": 310181888
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009151454363089267,
      "loss": 1.7397,
      "theoretical_loss": 4.158623239848762,
      "tokens_seen": 310247424
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009151253761283852,
      "loss": 1.7299,
      "theoretical_loss": 4.15852144695017,
      "tokens_seen": 310312960
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009151053159478436,
      "loss": 1.8316,
      "theoretical_loss": 4.158419681565265,
      "tokens_seen": 310378496
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009150852557673019,
      "loss": 1.7404,
      "theoretical_loss": 4.158317943680801,
      "tokens_seen": 310444032
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009150651955867603,
      "loss": 1.6477,
      "theoretical_loss": 4.158216233283545,
      "tokens_seen": 310509568
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009150451354062186,
      "loss": 1.8772,
      "theoretical_loss": 4.158114550360271,
      "tokens_seen": 310575104
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000915025075225677,
      "loss": 1.925,
      "theoretical_loss": 4.1580128948977615,
      "tokens_seen": 310640640
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009150050150451355,
      "loss": 1.7451,
      "theoretical_loss": 4.157911266882811,
      "tokens_seen": 310706176
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009149849548645937,
      "loss": 1.7826,
      "theoretical_loss": 4.157809666302221,
      "tokens_seen": 310771712
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009149648946840522,
      "loss": 1.7177,
      "theoretical_loss": 4.157708093142801,
      "tokens_seen": 310837248
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009149448345035106,
      "loss": 1.9009,
      "theoretical_loss": 4.157606547391374,
      "tokens_seen": 310902784
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009149247743229689,
      "loss": 1.7678,
      "theoretical_loss": 4.157505029034767,
      "tokens_seen": 310968320
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009149047141424273,
      "loss": 1.9484,
      "theoretical_loss": 4.157403538059819,
      "tokens_seen": 311033856
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009148846539618857,
      "loss": 1.8541,
      "theoretical_loss": 4.157302074453379,
      "tokens_seen": 311099392
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000914864593781344,
      "loss": 1.8846,
      "theoretical_loss": 4.157200638202301,
      "tokens_seen": 311164928
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009148445336008025,
      "loss": 1.5425,
      "theoretical_loss": 4.157099229293453,
      "tokens_seen": 311230464
    },
    {
      "epoch": 0.09,
      "objective/train/advantage_avg": 0.4924384355545044,
      "objective/train/docs_used": 120076,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.819303274154663,
      "objective/train/original_loss": 1.8193035125732422,
      "objective/train/theoretical_loss": 4.156997847713709,
      "objective/train/tokens_used": 331756000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24328577518463135,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050480604171753,
      "objective/train/weighted_lm_loss": 1.911105751991272,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.156997847713709,
      "tokens_seen": 311296000
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009148244734202607,
      "loss": 1.7748,
      "theoretical_loss": 4.156997847713709,
      "tokens_seen": 311296000
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009148044132397192,
      "loss": 1.7985,
      "theoretical_loss": 4.156896493449951,
      "tokens_seen": 311361536
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009147843530591775,
      "loss": 1.7454,
      "theoretical_loss": 4.156795166489074,
      "tokens_seen": 311427072
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009147642928786359,
      "loss": 1.8097,
      "theoretical_loss": 4.15669386681798,
      "tokens_seen": 311492608
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009147442326980943,
      "loss": 1.8527,
      "theoretical_loss": 4.156592594423579,
      "tokens_seen": 311558144
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009147241725175527,
      "loss": 1.7922,
      "theoretical_loss": 4.156491349292789,
      "tokens_seen": 311623680
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000914704112337011,
      "loss": 1.7492,
      "theoretical_loss": 4.156390131412543,
      "tokens_seen": 311689216
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009146840521564695,
      "loss": 1.6228,
      "theoretical_loss": 4.156288940769778,
      "tokens_seen": 311754752
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009146639919759277,
      "loss": 1.8304,
      "theoretical_loss": 4.156187777351438,
      "tokens_seen": 311820288
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009146439317953862,
      "loss": 1.7774,
      "theoretical_loss": 4.156086641144483,
      "tokens_seen": 311885824
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009146238716148445,
      "loss": 1.8537,
      "theoretical_loss": 4.155985532135875,
      "tokens_seen": 311951360
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009146038114343029,
      "loss": 1.7901,
      "theoretical_loss": 4.155884450312589,
      "tokens_seen": 312016896
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009145837512537613,
      "loss": 1.8438,
      "theoretical_loss": 4.155783395661609,
      "tokens_seen": 312082432
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009145636910732197,
      "loss": 1.8641,
      "theoretical_loss": 4.155682368169925,
      "tokens_seen": 312147968
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000914543630892678,
      "loss": 1.8986,
      "theoretical_loss": 4.1555813678245395,
      "tokens_seen": 312213504
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009145235707121364,
      "loss": 1.9761,
      "theoretical_loss": 4.155480394612462,
      "tokens_seen": 312279040
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009145035105315948,
      "loss": 1.644,
      "theoretical_loss": 4.15537944852071,
      "tokens_seen": 312344576
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009144834503510532,
      "loss": 1.8129,
      "theoretical_loss": 4.155278529536313,
      "tokens_seen": 312410112
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009144633901705115,
      "loss": 1.6904,
      "theoretical_loss": 4.155177637646306,
      "tokens_seen": 312475648
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.00091444332998997,
      "loss": 1.8086,
      "theoretical_loss": 4.1550767728377345,
      "tokens_seen": 312541184
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009144232698094282,
      "loss": 1.619,
      "theoretical_loss": 4.154975935097655,
      "tokens_seen": 312606720
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009144032096288867,
      "loss": 1.7375,
      "theoretical_loss": 4.154875124413128,
      "tokens_seen": 312672256
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009143831494483451,
      "loss": 1.779,
      "theoretical_loss": 4.154774340771228,
      "tokens_seen": 312737792
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009143630892678034,
      "loss": 1.6979,
      "theoretical_loss": 4.154673584159036,
      "tokens_seen": 312803328
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009143430290872619,
      "loss": 1.8726,
      "theoretical_loss": 4.15457285456364,
      "tokens_seen": 312868864
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009143229689067202,
      "loss": 1.7921,
      "theoretical_loss": 4.154472151972142,
      "tokens_seen": 312934400
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009143029087261786,
      "loss": 1.8944,
      "theoretical_loss": 4.154371476371646,
      "tokens_seen": 312999936
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.000914282848545637,
      "loss": 1.9317,
      "theoretical_loss": 4.154270827749271,
      "tokens_seen": 313065472
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009142627883650953,
      "loss": 1.8379,
      "theoretical_loss": 4.154170206092143,
      "tokens_seen": 313131008
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009142427281845537,
      "loss": 1.6154,
      "theoretical_loss": 4.154069611387394,
      "tokens_seen": 313196544
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009142226680040121,
      "loss": 1.7201,
      "theoretical_loss": 4.153969043622169,
      "tokens_seen": 313262080
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009142026078234704,
      "loss": 1.9128,
      "theoretical_loss": 4.153868502783618,
      "tokens_seen": 313327616
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009141825476429289,
      "loss": 1.868,
      "theoretical_loss": 4.153767988858904,
      "tokens_seen": 313393152
    },
    {
      "epoch": 0.09,
      "learning_rate": 0.0009141624874623871,
      "loss": 1.7828,
      "theoretical_loss": 4.153667501835195,
      "tokens_seen": 313458688
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009141424272818456,
      "loss": 1.7955,
      "theoretical_loss": 4.15356704169967,
      "tokens_seen": 313524224
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000914122367101304,
      "loss": 1.8222,
      "theoretical_loss": 4.1534666084395155,
      "tokens_seen": 313589760
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009141023069207623,
      "loss": 1.7689,
      "theoretical_loss": 4.153366202041927,
      "tokens_seen": 313655296
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009140822467402207,
      "loss": 1.7823,
      "theoretical_loss": 4.153265822494111,
      "tokens_seen": 313720832
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000914062186559679,
      "loss": 2.0244,
      "theoretical_loss": 4.153165469783279,
      "tokens_seen": 313786368
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009140421263791374,
      "loss": 1.8247,
      "theoretical_loss": 4.153065143896655,
      "tokens_seen": 313851904
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009140220661985959,
      "loss": 1.9073,
      "theoretical_loss": 4.152964844821469,
      "tokens_seen": 313917440
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009140020060180541,
      "loss": 1.8277,
      "theoretical_loss": 4.15286457254496,
      "tokens_seen": 313982976
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009139819458375126,
      "loss": 1.8345,
      "theoretical_loss": 4.152764327054376,
      "tokens_seen": 314048512
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000913961885656971,
      "loss": 1.7599,
      "theoretical_loss": 4.152664108336976,
      "tokens_seen": 314114048
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009139418254764293,
      "loss": 1.8354,
      "theoretical_loss": 4.152563916380026,
      "tokens_seen": 314179584
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009139217652958877,
      "loss": 1.818,
      "theoretical_loss": 4.152463751170799,
      "tokens_seen": 314245120
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000913901705115346,
      "loss": 1.778,
      "theoretical_loss": 4.152363612696579,
      "tokens_seen": 314310656
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009138816449348044,
      "loss": 1.8639,
      "theoretical_loss": 4.152263500944658,
      "tokens_seen": 314376192
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009138615847542629,
      "loss": 1.7843,
      "theoretical_loss": 4.152163415902336,
      "tokens_seen": 314441728
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009138415245737211,
      "loss": 1.6987,
      "theoretical_loss": 4.152063357556923,
      "tokens_seen": 314507264
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.49289390444755554,
      "objective/train/docs_used": 121169,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7357332706451416,
      "objective/train/original_loss": 1.735733151435852,
      "objective/train/theoretical_loss": 4.151963325895737,
      "objective/train/tokens_used": 335032800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24352145195007324,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505274534225464,
      "objective/train/weighted_lm_loss": 1.8233038187026978,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.151963325895737,
      "tokens_seen": 314572800
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009138214643931796,
      "loss": 1.7403,
      "theoretical_loss": 4.151963325895737,
      "tokens_seen": 314572800
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009138014042126379,
      "loss": 1.7728,
      "theoretical_loss": 4.151863320906105,
      "tokens_seen": 314638336
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009137813440320963,
      "loss": 1.7417,
      "theoretical_loss": 4.151763342575362,
      "tokens_seen": 314703872
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009137612838515547,
      "loss": 1.7573,
      "theoretical_loss": 4.151663390890851,
      "tokens_seen": 314769408
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009137412236710131,
      "loss": 1.8931,
      "theoretical_loss": 4.151563465839927,
      "tokens_seen": 314834944
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009137211634904714,
      "loss": 1.7439,
      "theoretical_loss": 4.151463567409949,
      "tokens_seen": 314900480
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009137011033099299,
      "loss": 1.8137,
      "theoretical_loss": 4.151363695588287,
      "tokens_seen": 314966016
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009136810431293881,
      "loss": 1.7893,
      "theoretical_loss": 4.151263850362321,
      "tokens_seen": 315031552
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009136609829488466,
      "loss": 1.844,
      "theoretical_loss": 4.151164031719437,
      "tokens_seen": 315097088
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009136409227683049,
      "loss": 1.8051,
      "theoretical_loss": 4.151064239647032,
      "tokens_seen": 315162624
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009136208625877633,
      "loss": 1.7564,
      "theoretical_loss": 4.1509644741325085,
      "tokens_seen": 315228160
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009136008024072217,
      "loss": 1.8875,
      "theoretical_loss": 4.150864735163281,
      "tokens_seen": 315293696
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009135807422266801,
      "loss": 1.8228,
      "theoretical_loss": 4.15076502272677,
      "tokens_seen": 315359232
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009135606820461384,
      "loss": 1.7781,
      "theoretical_loss": 4.150665336810405,
      "tokens_seen": 315424768
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009135406218655968,
      "loss": 1.7661,
      "theoretical_loss": 4.150565677401627,
      "tokens_seen": 315490304
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009135205616850552,
      "loss": 1.7718,
      "theoretical_loss": 4.15046604448788,
      "tokens_seen": 315555840
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009135005015045136,
      "loss": 1.7745,
      "theoretical_loss": 4.150366438056622,
      "tokens_seen": 315621376
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009134804413239719,
      "loss": 1.7982,
      "theoretical_loss": 4.150266858095317,
      "tokens_seen": 315686912
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009134603811434303,
      "loss": 1.674,
      "theoretical_loss": 4.150167304591437,
      "tokens_seen": 315752448
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009134403209628886,
      "loss": 1.7676,
      "theoretical_loss": 4.150067777532464,
      "tokens_seen": 315817984
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009134202607823471,
      "loss": 1.7478,
      "theoretical_loss": 4.149968276905888,
      "tokens_seen": 315883520
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009134002006018054,
      "loss": 1.797,
      "theoretical_loss": 4.149868802699206,
      "tokens_seen": 315949056
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009133801404212638,
      "loss": 2.1356,
      "theoretical_loss": 4.149769354899927,
      "tokens_seen": 316014592
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009133600802407222,
      "loss": 1.8429,
      "theoretical_loss": 4.149669933495565,
      "tokens_seen": 316080128
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009133400200601806,
      "loss": 1.7527,
      "theoretical_loss": 4.149570538473644,
      "tokens_seen": 316145664
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009133199598796389,
      "loss": 1.731,
      "theoretical_loss": 4.149471169821696,
      "tokens_seen": 316211200
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009132998996990974,
      "loss": 1.7687,
      "theoretical_loss": 4.149371827527264,
      "tokens_seen": 316276736
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009132798395185556,
      "loss": 1.6522,
      "theoretical_loss": 4.149272511577895,
      "tokens_seen": 316342272
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009132597793380141,
      "loss": 1.7597,
      "theoretical_loss": 4.149173221961146,
      "tokens_seen": 316407808
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009132397191574724,
      "loss": 1.6395,
      "theoretical_loss": 4.1490739586645855,
      "tokens_seen": 316473344
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009132196589769308,
      "loss": 1.87,
      "theoretical_loss": 4.148974721675787,
      "tokens_seen": 316538880
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009131995987963892,
      "loss": 1.7709,
      "theoretical_loss": 4.148875510982333,
      "tokens_seen": 316604416
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009131795386158475,
      "loss": 1.8424,
      "theoretical_loss": 4.1487763265718165,
      "tokens_seen": 316669952
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009131594784353059,
      "loss": 1.7587,
      "theoretical_loss": 4.148677168431836,
      "tokens_seen": 316735488
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009131394182547644,
      "loss": 1.7732,
      "theoretical_loss": 4.148578036550001,
      "tokens_seen": 316801024
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009131193580742226,
      "loss": 1.8576,
      "theoretical_loss": 4.148478930913927,
      "tokens_seen": 316866560
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009130992978936811,
      "loss": 1.7591,
      "theoretical_loss": 4.148379851511241,
      "tokens_seen": 316932096
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009130792377131393,
      "loss": 1.8007,
      "theoretical_loss": 4.148280798329575,
      "tokens_seen": 316997632
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009130591775325978,
      "loss": 1.8282,
      "theoretical_loss": 4.148181771356571,
      "tokens_seen": 317063168
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009130391173520562,
      "loss": 1.7429,
      "theoretical_loss": 4.14808277057988,
      "tokens_seen": 317128704
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009130190571715145,
      "loss": 1.7629,
      "theoretical_loss": 4.147983795987161,
      "tokens_seen": 317194240
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009129989969909729,
      "loss": 1.7354,
      "theoretical_loss": 4.147884847566079,
      "tokens_seen": 317259776
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009129789368104314,
      "loss": 1.7621,
      "theoretical_loss": 4.147785925304311,
      "tokens_seen": 317325312
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009129588766298896,
      "loss": 1.7322,
      "theoretical_loss": 4.147687029189541,
      "tokens_seen": 317390848
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009129388164493481,
      "loss": 1.901,
      "theoretical_loss": 4.14758815920946,
      "tokens_seen": 317456384
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009129187562688063,
      "loss": 1.7886,
      "theoretical_loss": 4.147489315351769,
      "tokens_seen": 317521920
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009128986960882648,
      "loss": 1.6663,
      "theoretical_loss": 4.147390497604177,
      "tokens_seen": 317587456
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009128786359077233,
      "loss": 1.7448,
      "theoretical_loss": 4.147291705954401,
      "tokens_seen": 317652992
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009128585757271815,
      "loss": 1.717,
      "theoretical_loss": 4.147192940390165,
      "tokens_seen": 317718528
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.00091283851554664,
      "loss": 1.947,
      "theoretical_loss": 4.147094200899205,
      "tokens_seen": 317784064
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.49450305104255676,
      "objective/train/docs_used": 122440,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.54131281375885,
      "objective/train/original_loss": 1.5413126945495605,
      "objective/train/theoretical_loss": 4.146995487469262,
      "objective/train/tokens_used": 338309600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24474811553955078,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506945848464966,
      "objective/train/weighted_lm_loss": 1.619470477104187,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 4.146995487469262,
      "tokens_seen": 317849600
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009128184553660983,
      "loss": 1.7732,
      "theoretical_loss": 4.146995487469262,
      "tokens_seen": 317849600
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009127983951855567,
      "loss": 1.8489,
      "theoretical_loss": 4.146896800088085,
      "tokens_seen": 317915136
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009127783350050151,
      "loss": 1.7643,
      "theoretical_loss": 4.146798138743433,
      "tokens_seen": 317980672
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009127582748244735,
      "loss": 1.707,
      "theoretical_loss": 4.146699503423074,
      "tokens_seen": 318046208
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009127382146439318,
      "loss": 1.6486,
      "theoretical_loss": 4.1466008941147825,
      "tokens_seen": 318111744
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009127181544633903,
      "loss": 1.8341,
      "theoretical_loss": 4.146502310806342,
      "tokens_seen": 318177280
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009126980942828485,
      "loss": 1.6462,
      "theoretical_loss": 4.146403753485544,
      "tokens_seen": 318242816
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000912678034102307,
      "loss": 1.7509,
      "theoretical_loss": 4.146305222140187,
      "tokens_seen": 318308352
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009126579739217653,
      "loss": 1.7517,
      "theoretical_loss": 4.146206716758081,
      "tokens_seen": 318373888
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009126379137412237,
      "loss": 1.7286,
      "theoretical_loss": 4.1461082373270415,
      "tokens_seen": 318439424
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009126178535606821,
      "loss": 1.8237,
      "theoretical_loss": 4.146009783834892,
      "tokens_seen": 318504960
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009125977933801405,
      "loss": 1.8733,
      "theoretical_loss": 4.145911356269468,
      "tokens_seen": 318570496
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009125777331995988,
      "loss": 1.8255,
      "theoretical_loss": 4.1458129546186075,
      "tokens_seen": 318636032
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009125576730190572,
      "loss": 1.7995,
      "theoretical_loss": 4.1457145788701615,
      "tokens_seen": 318701568
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009125376128385156,
      "loss": 1.7193,
      "theoretical_loss": 4.145616229011987,
      "tokens_seen": 318767104
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000912517552657974,
      "loss": 1.828,
      "theoretical_loss": 4.145517905031948,
      "tokens_seen": 318832640
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009124974924774323,
      "loss": 1.7877,
      "theoretical_loss": 4.145419606917921,
      "tokens_seen": 318898176
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009124774322968907,
      "loss": 1.9088,
      "theoretical_loss": 4.145321334657785,
      "tokens_seen": 318963712
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000912457372116349,
      "loss": 1.7268,
      "theoretical_loss": 4.145223088239432,
      "tokens_seen": 319029248
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009124373119358075,
      "loss": 1.801,
      "theoretical_loss": 4.145124867650759,
      "tokens_seen": 319094784
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009124172517552658,
      "loss": 1.7628,
      "theoretical_loss": 4.145026672879674,
      "tokens_seen": 319160320
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009123971915747242,
      "loss": 1.7966,
      "theoretical_loss": 4.14492850391409,
      "tokens_seen": 319225856
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009123771313941826,
      "loss": 1.6315,
      "theoretical_loss": 4.14483036074193,
      "tokens_seen": 319291392
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000912357071213641,
      "loss": 1.7785,
      "theoretical_loss": 4.144732243351125,
      "tokens_seen": 319356928
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009123370110330993,
      "loss": 1.8006,
      "theoretical_loss": 4.144634151729615,
      "tokens_seen": 319422464
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009123169508525578,
      "loss": 1.7751,
      "theoretical_loss": 4.144536085865344,
      "tokens_seen": 319488000
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000912296890672016,
      "loss": 1.9455,
      "theoretical_loss": 4.14443804574627,
      "tokens_seen": 319553536
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009122768304914745,
      "loss": 1.7975,
      "theoretical_loss": 4.144340031360354,
      "tokens_seen": 319619072
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009122567703109328,
      "loss": 1.8911,
      "theoretical_loss": 4.14424204269557,
      "tokens_seen": 319684608
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009122367101303912,
      "loss": 1.6413,
      "theoretical_loss": 4.1441440797398945,
      "tokens_seen": 319750144
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009122166499498496,
      "loss": 1.6841,
      "theoretical_loss": 4.144046142481317,
      "tokens_seen": 319815680
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009121965897693079,
      "loss": 1.9622,
      "theoretical_loss": 4.143948230907832,
      "tokens_seen": 319881216
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009121765295887663,
      "loss": 1.7783,
      "theoretical_loss": 4.1438503450074435,
      "tokens_seen": 319946752
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009121564694082248,
      "loss": 1.7984,
      "theoretical_loss": 4.1437524847681635,
      "tokens_seen": 320012288
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000912136409227683,
      "loss": 1.7974,
      "theoretical_loss": 4.143654650178012,
      "tokens_seen": 320077824
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009121163490471415,
      "loss": 1.8931,
      "theoretical_loss": 4.143556841225015,
      "tokens_seen": 320143360
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009120962888665997,
      "loss": 1.8957,
      "theoretical_loss": 4.14345905789721,
      "tokens_seen": 320208896
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009120762286860582,
      "loss": 1.9032,
      "theoretical_loss": 4.14336130018264,
      "tokens_seen": 320274432
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009120561685055166,
      "loss": 1.6911,
      "theoretical_loss": 4.143263568069358,
      "tokens_seen": 320339968
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009120361083249749,
      "loss": 1.7567,
      "theoretical_loss": 4.143165861545421,
      "tokens_seen": 320405504
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009120160481444333,
      "loss": 1.6795,
      "theoretical_loss": 4.143068180598902,
      "tokens_seen": 320471040
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009119959879638918,
      "loss": 1.7473,
      "theoretical_loss": 4.142970525217872,
      "tokens_seen": 320536576
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.00091197592778335,
      "loss": 1.8411,
      "theoretical_loss": 4.142872895390417,
      "tokens_seen": 320602112
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009119558676028085,
      "loss": 1.8678,
      "theoretical_loss": 4.142775291104629,
      "tokens_seen": 320667648
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009119358074222667,
      "loss": 1.7909,
      "theoretical_loss": 4.142677712348608,
      "tokens_seen": 320733184
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009119157472417252,
      "loss": 1.748,
      "theoretical_loss": 4.14258015911046,
      "tokens_seen": 320798720
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009118956870611836,
      "loss": 1.7048,
      "theoretical_loss": 4.142482631378303,
      "tokens_seen": 320864256
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009118756268806419,
      "loss": 1.7556,
      "theoretical_loss": 4.142385129140259,
      "tokens_seen": 320929792
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009118555667001003,
      "loss": 1.6847,
      "theoretical_loss": 4.142287652384462,
      "tokens_seen": 320995328
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009118355065195587,
      "loss": 1.6888,
      "theoretical_loss": 4.142190201099049,
      "tokens_seen": 321060864
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.48887574672698975,
      "objective/train/docs_used": 123582,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6981501579284668,
      "objective/train/original_loss": 1.6981501579284668,
      "objective/train/theoretical_loss": 4.142092775272169,
      "objective/train/tokens_used": 341586400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.23958638310432434,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050105333328247,
      "objective/train/weighted_lm_loss": 1.7833449840545654,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.142092775272169,
      "tokens_seen": 321126400
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000911815446339017,
      "loss": 1.7979,
      "theoretical_loss": 4.142092775272169,
      "tokens_seen": 321126400
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009117953861584755,
      "loss": 1.783,
      "theoretical_loss": 4.141995374891978,
      "tokens_seen": 321191936
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009117753259779338,
      "loss": 1.5579,
      "theoretical_loss": 4.1418979999466385,
      "tokens_seen": 321257472
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009117552657973922,
      "loss": 1.656,
      "theoretical_loss": 4.141800650424322,
      "tokens_seen": 321323008
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009117352056168505,
      "loss": 1.7395,
      "theoretical_loss": 4.141703326313209,
      "tokens_seen": 321388544
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000911715145436309,
      "loss": 1.8497,
      "theoretical_loss": 4.141606027601485,
      "tokens_seen": 321454080
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009116950852557673,
      "loss": 1.7712,
      "theoretical_loss": 4.141508754277346,
      "tokens_seen": 321519616
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009116750250752257,
      "loss": 1.6483,
      "theoretical_loss": 4.141411506328995,
      "tokens_seen": 321585152
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000911654964894684,
      "loss": 1.8046,
      "theoretical_loss": 4.141314283744643,
      "tokens_seen": 321650688
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009116349047141425,
      "loss": 1.8253,
      "theoretical_loss": 4.141217086512508,
      "tokens_seen": 321716224
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009116148445336008,
      "loss": 1.8028,
      "theoretical_loss": 4.141119914620817,
      "tokens_seen": 321781760
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009115947843530592,
      "loss": 1.7437,
      "theoretical_loss": 4.141022768057805,
      "tokens_seen": 321847296
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009115747241725175,
      "loss": 1.6649,
      "theoretical_loss": 4.140925646811714,
      "tokens_seen": 321912832
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000911554663991976,
      "loss": 1.7697,
      "theoretical_loss": 4.140828550870793,
      "tokens_seen": 321978368
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009115346038114343,
      "loss": 1.9109,
      "theoretical_loss": 4.140731480223303,
      "tokens_seen": 322043904
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009115145436308927,
      "loss": 1.7703,
      "theoretical_loss": 4.140634434857506,
      "tokens_seen": 322109440
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000911494483450351,
      "loss": 1.7073,
      "theoretical_loss": 4.1405374147616785,
      "tokens_seen": 322174976
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009114744232698094,
      "loss": 1.7863,
      "theoretical_loss": 4.140440419924101,
      "tokens_seen": 322240512
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009114543630892678,
      "loss": 1.8862,
      "theoretical_loss": 4.140343450333064,
      "tokens_seen": 322306048
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009114343029087262,
      "loss": 1.5811,
      "theoretical_loss": 4.1402465059768625,
      "tokens_seen": 322371584
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009114142427281845,
      "loss": 1.7835,
      "theoretical_loss": 4.140149586843803,
      "tokens_seen": 322437120
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000911394182547643,
      "loss": 1.7408,
      "theoretical_loss": 4.140052692922199,
      "tokens_seen": 322502656
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009113741223671014,
      "loss": 1.8122,
      "theoretical_loss": 4.139955824200369,
      "tokens_seen": 322568192
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009113540621865597,
      "loss": 1.8127,
      "theoretical_loss": 4.1398589806666415,
      "tokens_seen": 322633728
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009113340020060182,
      "loss": 1.7441,
      "theoretical_loss": 4.139762162309355,
      "tokens_seen": 322699264
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009113139418254764,
      "loss": 1.8992,
      "theoretical_loss": 4.13966536911685,
      "tokens_seen": 322764800
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009112938816449349,
      "loss": 1.7507,
      "theoretical_loss": 4.139568601077481,
      "tokens_seen": 322830336
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009112738214643932,
      "loss": 1.7393,
      "theoretical_loss": 4.139471858179606,
      "tokens_seen": 322895872
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009112537612838516,
      "loss": 1.6676,
      "theoretical_loss": 4.139375140411592,
      "tokens_seen": 322961408
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.00091123370110331,
      "loss": 1.7302,
      "theoretical_loss": 4.139278447761814,
      "tokens_seen": 323026944
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009112136409227683,
      "loss": 1.6805,
      "theoretical_loss": 4.139181780218656,
      "tokens_seen": 323092480
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009111935807422267,
      "loss": 1.6964,
      "theoretical_loss": 4.139085137770507,
      "tokens_seen": 323158016
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009111735205616852,
      "loss": 1.7933,
      "theoretical_loss": 4.138988520405764,
      "tokens_seen": 323223552
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009111534603811434,
      "loss": 1.6671,
      "theoretical_loss": 4.138891928112835,
      "tokens_seen": 323289088
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009111334002006019,
      "loss": 1.8928,
      "theoretical_loss": 4.138795360880132,
      "tokens_seen": 323354624
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009111133400200601,
      "loss": 1.6507,
      "theoretical_loss": 4.138698818696076,
      "tokens_seen": 323420160
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009110932798395186,
      "loss": 1.7278,
      "theoretical_loss": 4.138602301549097,
      "tokens_seen": 323485696
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000911073219658977,
      "loss": 1.7415,
      "theoretical_loss": 4.138505809427632,
      "tokens_seen": 323551232
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009110531594784353,
      "loss": 1.8598,
      "theoretical_loss": 4.138409342320124,
      "tokens_seen": 323616768
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009110330992978937,
      "loss": 1.5982,
      "theoretical_loss": 4.138312900215024,
      "tokens_seen": 323682304
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009110130391173522,
      "loss": 1.7677,
      "theoretical_loss": 4.138216483100795,
      "tokens_seen": 323747840
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009109929789368104,
      "loss": 1.6368,
      "theoretical_loss": 4.1381200909659,
      "tokens_seen": 323813376
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009109729187562689,
      "loss": 1.7068,
      "theoretical_loss": 4.138023723798817,
      "tokens_seen": 323878912
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009109528585757271,
      "loss": 1.7321,
      "theoretical_loss": 4.137927381588026,
      "tokens_seen": 323944448
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009109327983951856,
      "loss": 1.7413,
      "theoretical_loss": 4.137831064322021,
      "tokens_seen": 324009984
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000910912738214644,
      "loss": 1.8037,
      "theoretical_loss": 4.1377347719892965,
      "tokens_seen": 324075520
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009108926780341023,
      "loss": 1.6356,
      "theoretical_loss": 4.137638504578359,
      "tokens_seen": 324141056
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009108726178535607,
      "loss": 1.7395,
      "theoretical_loss": 4.137542262077721,
      "tokens_seen": 324206592
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009108525576730191,
      "loss": 1.6488,
      "theoretical_loss": 4.1374460444759045,
      "tokens_seen": 324272128
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009108324974924774,
      "loss": 1.7526,
      "theoretical_loss": 4.137349851761437,
      "tokens_seen": 324337664
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.49557074904441833,
      "objective/train/docs_used": 124782,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.604373812675476,
      "objective/train/original_loss": 1.6043736934661865,
      "objective/train/theoretical_loss": 4.137253683922854,
      "objective/train/tokens_used": 344863200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2460704892873764,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0508081912994385,
      "objective/train/weighted_lm_loss": 1.6858175992965698,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9048373699188232,
      "theoretical_loss": 4.137253683922854,
      "tokens_seen": 324403200
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009108124373119359,
      "loss": 1.7217,
      "theoretical_loss": 4.137253683922854,
      "tokens_seen": 324403200
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009107923771313942,
      "loss": 1.7176,
      "theoretical_loss": 4.137157540948699,
      "tokens_seen": 324468736
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009107723169508526,
      "loss": 1.7777,
      "theoretical_loss": 4.137061422827525,
      "tokens_seen": 324534272
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009107522567703109,
      "loss": 1.817,
      "theoretical_loss": 4.136965329547888,
      "tokens_seen": 324599808
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009107321965897693,
      "loss": 1.6928,
      "theoretical_loss": 4.136869261098356,
      "tokens_seen": 324665344
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009107121364092277,
      "loss": 1.7311,
      "theoretical_loss": 4.136773217467502,
      "tokens_seen": 324730880
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009106920762286861,
      "loss": 1.863,
      "theoretical_loss": 4.136677198643908,
      "tokens_seen": 324796416
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009106720160481444,
      "loss": 1.7369,
      "theoretical_loss": 4.136581204616162,
      "tokens_seen": 324861952
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009106519558676029,
      "loss": 1.7076,
      "theoretical_loss": 4.136485235372861,
      "tokens_seen": 324927488
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009106318956870612,
      "loss": 1.7839,
      "theoretical_loss": 4.13638929090261,
      "tokens_seen": 324993024
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009106118355065196,
      "loss": 1.6871,
      "theoretical_loss": 4.13629337119402,
      "tokens_seen": 325058560
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009105917753259779,
      "loss": 1.6558,
      "theoretical_loss": 4.136197476235708,
      "tokens_seen": 325124096
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009105717151454364,
      "loss": 1.8245,
      "theoretical_loss": 4.136101606016304,
      "tokens_seen": 325189632
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009105516549648947,
      "loss": 1.7017,
      "theoretical_loss": 4.136005760524439,
      "tokens_seen": 325255168
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009105315947843531,
      "loss": 1.7789,
      "theoretical_loss": 4.135909939748757,
      "tokens_seen": 325320704
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009105115346038114,
      "loss": 1.7688,
      "theoretical_loss": 4.135814143677906,
      "tokens_seen": 325386240
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009104914744232698,
      "loss": 1.7062,
      "theoretical_loss": 4.135718372300543,
      "tokens_seen": 325451776
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009104714142427282,
      "loss": 1.7736,
      "theoretical_loss": 4.135622625605333,
      "tokens_seen": 325517312
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009104513540621866,
      "loss": 1.698,
      "theoretical_loss": 4.135526903580946,
      "tokens_seen": 325582848
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009104312938816449,
      "loss": 1.8322,
      "theoretical_loss": 4.135431206216061,
      "tokens_seen": 325648384
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009104112337011034,
      "loss": 1.7194,
      "theoretical_loss": 4.135335533499366,
      "tokens_seen": 325713920
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009103911735205616,
      "loss": 1.7395,
      "theoretical_loss": 4.135239885419554,
      "tokens_seen": 325779456
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009103711133400201,
      "loss": 1.7968,
      "theoretical_loss": 4.135144261965327,
      "tokens_seen": 325844992
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009103510531594784,
      "loss": 1.666,
      "theoretical_loss": 4.135048663125394,
      "tokens_seen": 325910528
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009103309929789368,
      "loss": 1.8125,
      "theoretical_loss": 4.134953088888471,
      "tokens_seen": 325976064
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009103109327983952,
      "loss": 1.8514,
      "theoretical_loss": 4.134857539243283,
      "tokens_seen": 326041600
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009102908726178536,
      "loss": 1.6628,
      "theoretical_loss": 4.134762014178559,
      "tokens_seen": 326107136
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009102708124373119,
      "loss": 1.8064,
      "theoretical_loss": 4.13466651368304,
      "tokens_seen": 326172672
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009102507522567704,
      "loss": 1.748,
      "theoretical_loss": 4.134571037745471,
      "tokens_seen": 326238208
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009102306920762286,
      "loss": 1.6509,
      "theoretical_loss": 4.134475586354606,
      "tokens_seen": 326303744
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009102106318956871,
      "loss": 1.7845,
      "theoretical_loss": 4.134380159499204,
      "tokens_seen": 326369280
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009101905717151455,
      "loss": 1.783,
      "theoretical_loss": 4.134284757168037,
      "tokens_seen": 326434816
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009101705115346038,
      "loss": 1.6616,
      "theoretical_loss": 4.134189379349877,
      "tokens_seen": 326500352
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009101504513540622,
      "loss": 1.7684,
      "theoretical_loss": 4.13409402603351,
      "tokens_seen": 326565888
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009101303911735205,
      "loss": 1.8304,
      "theoretical_loss": 4.1339986972077245,
      "tokens_seen": 326631424
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009101103309929789,
      "loss": 1.6475,
      "theoretical_loss": 4.13390339286132,
      "tokens_seen": 326696960
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009100902708124374,
      "loss": 1.7013,
      "theoretical_loss": 4.1338081129831,
      "tokens_seen": 326762496
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009100702106318956,
      "loss": 1.7019,
      "theoretical_loss": 4.1337128575618785,
      "tokens_seen": 326828032
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009100501504513541,
      "loss": 1.6178,
      "theoretical_loss": 4.133617626586475,
      "tokens_seen": 326893568
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009100300902708125,
      "loss": 1.6345,
      "theoretical_loss": 4.1335224200457175,
      "tokens_seen": 326959104
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009100100300902708,
      "loss": 1.7228,
      "theoretical_loss": 4.133427237928439,
      "tokens_seen": 327024640
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009099899699097292,
      "loss": 1.7379,
      "theoretical_loss": 4.133332080223483,
      "tokens_seen": 327090176
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009099699097291875,
      "loss": 1.8473,
      "theoretical_loss": 4.133236946919698,
      "tokens_seen": 327155712
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009099498495486459,
      "loss": 2.0025,
      "theoretical_loss": 4.133141838005942,
      "tokens_seen": 327221248
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009099297893681044,
      "loss": 1.7865,
      "theoretical_loss": 4.133046753471077,
      "tokens_seen": 327286784
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009099097291875627,
      "loss": 1.7868,
      "theoretical_loss": 4.132951693303976,
      "tokens_seen": 327352320
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009098896690070211,
      "loss": 1.7852,
      "theoretical_loss": 4.132856657493516,
      "tokens_seen": 327417856
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009098696088264795,
      "loss": 1.8734,
      "theoretical_loss": 4.132761646028585,
      "tokens_seen": 327483392
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009098495486459378,
      "loss": 1.7436,
      "theoretical_loss": 4.132666658898074,
      "tokens_seen": 327548928
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009098294884653963,
      "loss": 1.7704,
      "theoretical_loss": 4.132571696090885,
      "tokens_seen": 327614464
    },
    {
      "debugging/Compilability": 1.0,
      "debugging/distinct-1-grams": 0.7246169962973871,
      "debugging/entropy-1-grams": 5.13190351820591,
      "debugging/length": 472.8,
      "debugging/num_segments": 10,
      "debugging/raw_token_scores_avg": 0.008663635700941086,
      "debugging/raw_token_scores_std": 0.023182114586234093,
      "debugging/score": 0.0033104506818385365,
      "debugging/score_std": 0.0017105448100298692,
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.4913388192653656,
      "objective/train/docs_used": 126013,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.739636778831482,
      "objective/train/original_loss": 1.7396366596221924,
      "objective/train/theoretical_loss": 4.132476757595925,
      "objective/train/tokens_used": 348140000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24195149540901184,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503638982772827,
      "objective/train/weighted_lm_loss": 1.8274545669555664,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.132476757595925,
      "tokens_seen": 327680000
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009098094282848546,
      "loss": 1.8537,
      "theoretical_loss": 4.132476757595925,
      "tokens_seen": 327680000
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000909789368104313,
      "loss": 1.7564,
      "theoretical_loss": 4.13238184340211,
      "tokens_seen": 327745536
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009097693079237713,
      "loss": 1.8456,
      "theoretical_loss": 4.13228695349836,
      "tokens_seen": 327811072
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009097492477432297,
      "loss": 1.836,
      "theoretical_loss": 4.132192087873608,
      "tokens_seen": 327876608
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009097291875626881,
      "loss": 1.7369,
      "theoretical_loss": 4.132097246516788,
      "tokens_seen": 327942144
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009097091273821465,
      "loss": 1.8652,
      "theoretical_loss": 4.132002429416845,
      "tokens_seen": 328007680
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009096890672016048,
      "loss": 1.7203,
      "theoretical_loss": 4.131907636562731,
      "tokens_seen": 328073216
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009096690070210633,
      "loss": 1.8587,
      "theoretical_loss": 4.131812867943403,
      "tokens_seen": 328138752
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009096489468405216,
      "loss": 1.7171,
      "theoretical_loss": 4.131718123547829,
      "tokens_seen": 328204288
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.00090962888665998,
      "loss": 1.6613,
      "theoretical_loss": 4.131623403364981,
      "tokens_seen": 328269824
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009096088264794383,
      "loss": 1.9833,
      "theoretical_loss": 4.131528707383838,
      "tokens_seen": 328335360
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009095887662988968,
      "loss": 1.7528,
      "theoretical_loss": 4.131434035593388,
      "tokens_seen": 328400896
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009095687061183551,
      "loss": 1.6557,
      "theoretical_loss": 4.131339387982628,
      "tokens_seen": 328466432
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009095486459378135,
      "loss": 1.6133,
      "theoretical_loss": 4.131244764540557,
      "tokens_seen": 328531968
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009095285857572718,
      "loss": 1.7815,
      "theoretical_loss": 4.1311501652561855,
      "tokens_seen": 328597504
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009095085255767302,
      "loss": 1.909,
      "theoretical_loss": 4.131055590118528,
      "tokens_seen": 328663040
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009094884653961886,
      "loss": 1.6056,
      "theoretical_loss": 4.1309610391166105,
      "tokens_seen": 328728576
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000909468405215647,
      "loss": 1.7617,
      "theoretical_loss": 4.1308665122394626,
      "tokens_seen": 328794112
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009094483450351053,
      "loss": 1.6625,
      "theoretical_loss": 4.1307720094761216,
      "tokens_seen": 328859648
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009094282848545638,
      "loss": 1.6576,
      "theoretical_loss": 4.130677530815632,
      "tokens_seen": 328925184
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000909408224674022,
      "loss": 1.7591,
      "theoretical_loss": 4.1305830762470475,
      "tokens_seen": 328990720
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009093881644934805,
      "loss": 1.8302,
      "theoretical_loss": 4.130488645759426,
      "tokens_seen": 329056256
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009093681043129388,
      "loss": 1.7202,
      "theoretical_loss": 4.130394239341835,
      "tokens_seen": 329121792
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009093480441323972,
      "loss": 1.7775,
      "theoretical_loss": 4.1302998569833465,
      "tokens_seen": 329187328
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009093279839518556,
      "loss": 1.7116,
      "theoretical_loss": 4.1302054986730425,
      "tokens_seen": 329252864
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000909307923771314,
      "loss": 1.7708,
      "theoretical_loss": 4.130111164400011,
      "tokens_seen": 329318400
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009092878635907723,
      "loss": 1.7346,
      "theoretical_loss": 4.130016854153346,
      "tokens_seen": 329383936
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009092678034102308,
      "loss": 1.7316,
      "theoretical_loss": 4.129922567922149,
      "tokens_seen": 329449472
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000909247743229689,
      "loss": 1.7205,
      "theoretical_loss": 4.129828305695531,
      "tokens_seen": 329515008
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009092276830491475,
      "loss": 1.688,
      "theoretical_loss": 4.129734067462607,
      "tokens_seen": 329580544
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009092076228686059,
      "loss": 1.7054,
      "theoretical_loss": 4.129639853212501,
      "tokens_seen": 329646080
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009091875626880642,
      "loss": 1.8751,
      "theoretical_loss": 4.129545662934341,
      "tokens_seen": 329711616
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009091675025075226,
      "loss": 1.7017,
      "theoretical_loss": 4.129451496617269,
      "tokens_seen": 329777152
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009091474423269809,
      "loss": 1.7523,
      "theoretical_loss": 4.129357354250426,
      "tokens_seen": 329842688
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009091273821464393,
      "loss": 1.7279,
      "theoretical_loss": 4.1292632358229655,
      "tokens_seen": 329908224
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009091073219658978,
      "loss": 1.7621,
      "theoretical_loss": 4.129169141324045,
      "tokens_seen": 329973760
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000909087261785356,
      "loss": 1.8877,
      "theoretical_loss": 4.129075070742831,
      "tokens_seen": 330039296
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009090672016048145,
      "loss": 1.7733,
      "theoretical_loss": 4.128981024068497,
      "tokens_seen": 330104832
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009090471414242729,
      "loss": 1.7082,
      "theoretical_loss": 4.128887001290222,
      "tokens_seen": 330170368
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009090270812437312,
      "loss": 1.761,
      "theoretical_loss": 4.1287930023971935,
      "tokens_seen": 330235904
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009090070210631896,
      "loss": 1.83,
      "theoretical_loss": 4.128699027378604,
      "tokens_seen": 330301440
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009089869608826479,
      "loss": 1.8233,
      "theoretical_loss": 4.128605076223657,
      "tokens_seen": 330366976
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009089669007021063,
      "loss": 1.6636,
      "theoretical_loss": 4.128511148921558,
      "tokens_seen": 330432512
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009089468405215648,
      "loss": 1.7057,
      "theoretical_loss": 4.128417245461525,
      "tokens_seen": 330498048
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000908926780341023,
      "loss": 1.8225,
      "theoretical_loss": 4.128323365832777,
      "tokens_seen": 330563584
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009089067201604815,
      "loss": 1.8282,
      "theoretical_loss": 4.1282295100245445,
      "tokens_seen": 330629120
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009088866599799398,
      "loss": 1.8193,
      "theoretical_loss": 4.128135678026064,
      "tokens_seen": 330694656
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009088665997993982,
      "loss": 1.7495,
      "theoretical_loss": 4.128041869826578,
      "tokens_seen": 330760192
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009088465396188566,
      "loss": 1.7241,
      "theoretical_loss": 4.127948085415338,
      "tokens_seen": 330825728
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000908826479438315,
      "loss": 1.8882,
      "theoretical_loss": 4.127854324781599,
      "tokens_seen": 330891264
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.4939263164997101,
      "objective/train/docs_used": 127130,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7381356954574585,
      "objective/train/original_loss": 1.738135576248169,
      "objective/train/theoretical_loss": 4.127760587914626,
      "objective/train/tokens_used": 351416800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24445585906505585,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506354570388794,
      "objective/train/weighted_lm_loss": 1.8260606527328491,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.127760587914626,
      "tokens_seen": 330956800
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009088064192577733,
      "loss": 1.8542,
      "theoretical_loss": 4.127760587914626,
      "tokens_seen": 330956800
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009087863590772317,
      "loss": 1.715,
      "theoretical_loss": 4.12766687480369,
      "tokens_seen": 331022336
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.00090876629889669,
      "loss": 1.6645,
      "theoretical_loss": 4.127573185438068,
      "tokens_seen": 331087872
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009087462387161485,
      "loss": 1.9245,
      "theoretical_loss": 4.127479519807047,
      "tokens_seen": 331153408
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009087261785356068,
      "loss": 1.8333,
      "theoretical_loss": 4.1273858778999175,
      "tokens_seen": 331218944
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009087061183550652,
      "loss": 1.7053,
      "theoretical_loss": 4.127292259705978,
      "tokens_seen": 331284480
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009086860581745236,
      "loss": 1.8399,
      "theoretical_loss": 4.127198665214536,
      "tokens_seen": 331350016
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000908665997993982,
      "loss": 1.7369,
      "theoretical_loss": 4.127105094414903,
      "tokens_seen": 331415552
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009086459378134403,
      "loss": 1.7215,
      "theoretical_loss": 4.127011547296398,
      "tokens_seen": 331481088
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009086258776328987,
      "loss": 1.6885,
      "theoretical_loss": 4.126918023848349,
      "tokens_seen": 331546624
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000908605817452357,
      "loss": 1.6354,
      "theoretical_loss": 4.126824524060088,
      "tokens_seen": 331612160
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009085857572718155,
      "loss": 1.7601,
      "theoretical_loss": 4.1267310479209565,
      "tokens_seen": 331677696
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009085656970912738,
      "loss": 1.7374,
      "theoretical_loss": 4.126637595420302,
      "tokens_seen": 331743232
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009085456369107322,
      "loss": 1.75,
      "theoretical_loss": 4.126544166547479,
      "tokens_seen": 331808768
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009085255767301905,
      "loss": 1.7956,
      "theoretical_loss": 4.126450761291847,
      "tokens_seen": 331874304
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000908505516549649,
      "loss": 1.7369,
      "theoretical_loss": 4.126357379642775,
      "tokens_seen": 331939840
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009084854563691073,
      "loss": 1.8566,
      "theoretical_loss": 4.126264021589638,
      "tokens_seen": 332005376
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009084653961885657,
      "loss": 1.7816,
      "theoretical_loss": 4.126170687121817,
      "tokens_seen": 332070912
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000908445336008024,
      "loss": 1.8452,
      "theoretical_loss": 4.126077376228702,
      "tokens_seen": 332136448
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009084252758274824,
      "loss": 1.6802,
      "theoretical_loss": 4.125984088899688,
      "tokens_seen": 332201984
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009084052156469409,
      "loss": 1.64,
      "theoretical_loss": 4.125890825124177,
      "tokens_seen": 332267520
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009083851554663992,
      "loss": 1.6285,
      "theoretical_loss": 4.125797584891578,
      "tokens_seen": 332333056
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009083650952858576,
      "loss": 1.699,
      "theoretical_loss": 4.1257043681913075,
      "tokens_seen": 332398592
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000908345035105316,
      "loss": 1.6606,
      "theoretical_loss": 4.125611175012789,
      "tokens_seen": 332464128
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009083249749247744,
      "loss": 1.6762,
      "theoretical_loss": 4.125518005345452,
      "tokens_seen": 332529664
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009083049147442327,
      "loss": 1.8607,
      "theoretical_loss": 4.1254248591787315,
      "tokens_seen": 332595200
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009082848545636912,
      "loss": 1.7981,
      "theoretical_loss": 4.125331736502073,
      "tokens_seen": 332660736
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009082647943831494,
      "loss": 1.8264,
      "theoretical_loss": 4.125238637304925,
      "tokens_seen": 332726272
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009082447342026079,
      "loss": 1.8123,
      "theoretical_loss": 4.125145561576746,
      "tokens_seen": 332791808
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009082246740220663,
      "loss": 1.8439,
      "theoretical_loss": 4.125052509306999,
      "tokens_seen": 332857344
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009082046138415246,
      "loss": 1.6351,
      "theoretical_loss": 4.124959480485156,
      "tokens_seen": 332922880
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000908184553660983,
      "loss": 1.7231,
      "theoretical_loss": 4.124866475100693,
      "tokens_seen": 332988416
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009081644934804413,
      "loss": 1.7502,
      "theoretical_loss": 4.124773493143094,
      "tokens_seen": 333053952
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009081444332998997,
      "loss": 1.7452,
      "theoretical_loss": 4.124680534601851,
      "tokens_seen": 333119488
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009081243731193582,
      "loss": 1.6137,
      "theoretical_loss": 4.124587599466462,
      "tokens_seen": 333185024
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009081043129388164,
      "loss": 1.7891,
      "theoretical_loss": 4.124494687726431,
      "tokens_seen": 333250560
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009080842527582749,
      "loss": 1.6966,
      "theoretical_loss": 4.124401799371269,
      "tokens_seen": 333316096
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009080641925777333,
      "loss": 1.8163,
      "theoretical_loss": 4.124308934390496,
      "tokens_seen": 333381632
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009080441323971916,
      "loss": 1.8086,
      "theoretical_loss": 4.124216092773635,
      "tokens_seen": 333447168
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.00090802407221665,
      "loss": 1.712,
      "theoretical_loss": 4.124123274510217,
      "tokens_seen": 333512704
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009080040120361083,
      "loss": 1.6625,
      "theoretical_loss": 4.124030479589782,
      "tokens_seen": 333578240
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009079839518555667,
      "loss": 1.6341,
      "theoretical_loss": 4.1239377080018755,
      "tokens_seen": 333643776
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009079638916750252,
      "loss": 1.7989,
      "theoretical_loss": 4.123844959736049,
      "tokens_seen": 333709312
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009079438314944834,
      "loss": 1.7309,
      "theoretical_loss": 4.123752234781859,
      "tokens_seen": 333774848
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009079237713139419,
      "loss": 1.8247,
      "theoretical_loss": 4.1236595331288735,
      "tokens_seen": 333840384
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009079037111334002,
      "loss": 1.6904,
      "theoretical_loss": 4.123566854766663,
      "tokens_seen": 333905920
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009078836509528586,
      "loss": 1.6505,
      "theoretical_loss": 4.123474199684807,
      "tokens_seen": 333971456
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000907863590772317,
      "loss": 1.8607,
      "theoretical_loss": 4.123381567872891,
      "tokens_seen": 334036992
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009078435305917754,
      "loss": 1.8399,
      "theoretical_loss": 4.123288959320506,
      "tokens_seen": 334102528
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009078234704112337,
      "loss": 1.763,
      "theoretical_loss": 4.123196374017253,
      "tokens_seen": 334168064
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.4906438887119293,
      "objective/train/docs_used": 128373,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.830473780632019,
      "objective/train/original_loss": 1.8304736614227295,
      "objective/train/theoretical_loss": 4.123103811952736,
      "objective/train/tokens_used": 354693600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24106791615486145,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502898693084717,
      "objective/train/weighted_lm_loss": 1.9224073886871338,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.123103811952736,
      "tokens_seen": 334233600
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009078034102306921,
      "loss": 1.8137,
      "theoretical_loss": 4.123103811952736,
      "tokens_seen": 334233600
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009077833500501504,
      "loss": 1.6342,
      "theoretical_loss": 4.123011273116566,
      "tokens_seen": 334299136
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009077632898696089,
      "loss": 1.8001,
      "theoretical_loss": 4.122918757498365,
      "tokens_seen": 334364672
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009077432296890672,
      "loss": 1.6724,
      "theoretical_loss": 4.122826265087756,
      "tokens_seen": 334430208
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009077231695085256,
      "loss": 1.6699,
      "theoretical_loss": 4.122733795874372,
      "tokens_seen": 334495744
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000907703109327984,
      "loss": 1.8302,
      "theoretical_loss": 4.122641349847852,
      "tokens_seen": 334561280
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009076830491474424,
      "loss": 1.6683,
      "theoretical_loss": 4.122548926997842,
      "tokens_seen": 334626816
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009076629889669007,
      "loss": 1.8169,
      "theoretical_loss": 4.122456527313994,
      "tokens_seen": 334692352
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009076429287863591,
      "loss": 1.8627,
      "theoretical_loss": 4.122364150785966,
      "tokens_seen": 334757888
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009076228686058174,
      "loss": 1.7632,
      "theoretical_loss": 4.1222717974034255,
      "tokens_seen": 334823424
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009076028084252759,
      "loss": 1.7083,
      "theoretical_loss": 4.122179467156043,
      "tokens_seen": 334888960
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009075827482447342,
      "loss": 1.8839,
      "theoretical_loss": 4.122087160033496,
      "tokens_seen": 334954496
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009075626880641926,
      "loss": 1.8024,
      "theoretical_loss": 4.1219948760254725,
      "tokens_seen": 335020032
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009075426278836509,
      "loss": 1.6807,
      "theoretical_loss": 4.121902615121663,
      "tokens_seen": 335085568
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009075225677031094,
      "loss": 1.8905,
      "theoretical_loss": 4.121810377311768,
      "tokens_seen": 335151104
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009075025075225677,
      "loss": 1.7158,
      "theoretical_loss": 4.1217181625854895,
      "tokens_seen": 335216640
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009074824473420261,
      "loss": 1.7767,
      "theoretical_loss": 4.121625970932542,
      "tokens_seen": 335282176
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009074623871614845,
      "loss": 1.8351,
      "theoretical_loss": 4.121533802342643,
      "tokens_seen": 335347712
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009074423269809428,
      "loss": 1.6733,
      "theoretical_loss": 4.121441656805517,
      "tokens_seen": 335413248
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009074222668004012,
      "loss": 1.7529,
      "theoretical_loss": 4.121349534310896,
      "tokens_seen": 335478784
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009074022066198596,
      "loss": 1.7908,
      "theoretical_loss": 4.121257434848519,
      "tokens_seen": 335544320
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009073821464393179,
      "loss": 1.7714,
      "theoretical_loss": 4.121165358408129,
      "tokens_seen": 335609856
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009073620862587764,
      "loss": 1.8105,
      "theoretical_loss": 4.121073304979479,
      "tokens_seen": 335675392
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009073420260782347,
      "loss": 1.6567,
      "theoretical_loss": 4.120981274552326,
      "tokens_seen": 335740928
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009073219658976931,
      "loss": 1.793,
      "theoretical_loss": 4.120889267116435,
      "tokens_seen": 335806464
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009073019057171515,
      "loss": 1.7369,
      "theoretical_loss": 4.1207972826615755,
      "tokens_seen": 335872000
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009072818455366098,
      "loss": 1.8146,
      "theoretical_loss": 4.120705321177526,
      "tokens_seen": 335937536
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009072617853560682,
      "loss": 1.7543,
      "theoretical_loss": 4.120613382654072,
      "tokens_seen": 336003072
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009072417251755267,
      "loss": 1.7126,
      "theoretical_loss": 4.1205214670810015,
      "tokens_seen": 336068608
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009072216649949849,
      "loss": 1.7503,
      "theoretical_loss": 4.120429574448114,
      "tokens_seen": 336134144
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009072016048144434,
      "loss": 1.6967,
      "theoretical_loss": 4.120337704745211,
      "tokens_seen": 336199680
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009071815446339016,
      "loss": 1.7868,
      "theoretical_loss": 4.120245857962104,
      "tokens_seen": 336265216
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009071614844533601,
      "loss": 1.7849,
      "theoretical_loss": 4.120154034088609,
      "tokens_seen": 336330752
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009071414242728185,
      "loss": 1.7815,
      "theoretical_loss": 4.120062233114549,
      "tokens_seen": 336396288
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009071213640922768,
      "loss": 1.8475,
      "theoretical_loss": 4.119970455029755,
      "tokens_seen": 336461824
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009071013039117352,
      "loss": 1.6997,
      "theoretical_loss": 4.119878699824063,
      "tokens_seen": 336527360
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009070812437311937,
      "loss": 1.7022,
      "theoretical_loss": 4.119786967487314,
      "tokens_seen": 336592896
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009070611835506519,
      "loss": 1.8374,
      "theoretical_loss": 4.119695258009358,
      "tokens_seen": 336658432
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009070411233701104,
      "loss": 1.7614,
      "theoretical_loss": 4.119603571380052,
      "tokens_seen": 336723968
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009070210631895686,
      "loss": 1.7412,
      "theoretical_loss": 4.119511907589256,
      "tokens_seen": 336789504
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009070010030090271,
      "loss": 1.7961,
      "theoretical_loss": 4.11942026662684,
      "tokens_seen": 336855040
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009069809428284855,
      "loss": 1.7222,
      "theoretical_loss": 4.1193286484826785,
      "tokens_seen": 336920576
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009069608826479438,
      "loss": 1.7334,
      "theoretical_loss": 4.119237053146653,
      "tokens_seen": 336986112
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009069408224674022,
      "loss": 1.7046,
      "theoretical_loss": 4.119145480608653,
      "tokens_seen": 337051648
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009069207622868606,
      "loss": 1.6898,
      "theoretical_loss": 4.11905393085857,
      "tokens_seen": 337117184
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000906900702106319,
      "loss": 1.8415,
      "theoretical_loss": 4.118962403886306,
      "tokens_seen": 337182720
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009068806419257774,
      "loss": 1.7412,
      "theoretical_loss": 4.11887089968177,
      "tokens_seen": 337248256
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009068605817452358,
      "loss": 1.8208,
      "theoretical_loss": 4.118779418234874,
      "tokens_seen": 337313792
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009068405215646941,
      "loss": 1.6441,
      "theoretical_loss": 4.118687959535539,
      "tokens_seen": 337379328
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009068204613841525,
      "loss": 1.8413,
      "theoretical_loss": 4.118596523573689,
      "tokens_seen": 337444864
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.49176985025405884,
      "objective/train/docs_used": 129675,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5919711589813232,
      "objective/train/original_loss": 1.5919709205627441,
      "objective/train/theoretical_loss": 4.118505110339262,
      "objective/train/tokens_used": 357970400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24228759109973907,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504087209701538,
      "objective/train/weighted_lm_loss": 1.6722182035446167,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.118505110339262,
      "tokens_seen": 337510400
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009068004012036108,
      "loss": 1.6432,
      "theoretical_loss": 4.118505110339262,
      "tokens_seen": 337510400
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009067803410230693,
      "loss": 1.7769,
      "theoretical_loss": 4.118413719822192,
      "tokens_seen": 337575936
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009067602808425276,
      "loss": 1.6643,
      "theoretical_loss": 4.118322352012429,
      "tokens_seen": 337641472
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000906740220661986,
      "loss": 1.8076,
      "theoretical_loss": 4.118231006899924,
      "tokens_seen": 337707008
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009067201604814444,
      "loss": 1.6354,
      "theoretical_loss": 4.118139684474634,
      "tokens_seen": 337772544
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009067001003009028,
      "loss": 1.6821,
      "theoretical_loss": 4.118048384726525,
      "tokens_seen": 337838080
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009066800401203611,
      "loss": 1.7323,
      "theoretical_loss": 4.117957107645569,
      "tokens_seen": 337903616
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009066599799398195,
      "loss": 1.8233,
      "theoretical_loss": 4.117865853221743,
      "tokens_seen": 337969152
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009066399197592778,
      "loss": 1.7854,
      "theoretical_loss": 4.117774621445031,
      "tokens_seen": 338034688
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009066198595787363,
      "loss": 1.6559,
      "theoretical_loss": 4.117683412305425,
      "tokens_seen": 338100224
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009065997993981946,
      "loss": 1.807,
      "theoretical_loss": 4.1175922257929205,
      "tokens_seen": 338165760
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000906579739217653,
      "loss": 1.664,
      "theoretical_loss": 4.117501061897521,
      "tokens_seen": 338231296
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009065596790371113,
      "loss": 1.7741,
      "theoretical_loss": 4.117409920609235,
      "tokens_seen": 338296832
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009065396188565698,
      "loss": 1.9005,
      "theoretical_loss": 4.11731880191808,
      "tokens_seen": 338362368
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009065195586760281,
      "loss": 1.8514,
      "theoretical_loss": 4.117227705814078,
      "tokens_seen": 338427904
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009064994984954865,
      "loss": 1.8802,
      "theoretical_loss": 4.117136632287258,
      "tokens_seen": 338493440
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009064794383149449,
      "loss": 1.7452,
      "theoretical_loss": 4.117045581327653,
      "tokens_seen": 338558976
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009064593781344032,
      "loss": 1.8935,
      "theoretical_loss": 4.116954552925305,
      "tokens_seen": 338624512
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009064393179538616,
      "loss": 1.9351,
      "theoretical_loss": 4.116863547070264,
      "tokens_seen": 338690048
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.00090641925777332,
      "loss": 1.8515,
      "theoretical_loss": 4.116772563752581,
      "tokens_seen": 338755584
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009063991975927783,
      "loss": 1.7043,
      "theoretical_loss": 4.116681602962317,
      "tokens_seen": 338821120
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009063791374122368,
      "loss": 1.7002,
      "theoretical_loss": 4.116590664689539,
      "tokens_seen": 338886656
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009063590772316951,
      "loss": 1.8262,
      "theoretical_loss": 4.116499748924319,
      "tokens_seen": 338952192
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009063390170511535,
      "loss": 1.768,
      "theoretical_loss": 4.116408855656737,
      "tokens_seen": 339017728
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009063189568706119,
      "loss": 1.7185,
      "theoretical_loss": 4.116317984876878,
      "tokens_seen": 339083264
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009062988966900702,
      "loss": 1.7734,
      "theoretical_loss": 4.1162271365748335,
      "tokens_seen": 339148800
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009062788365095286,
      "loss": 1.8295,
      "theoretical_loss": 4.116136310740702,
      "tokens_seen": 339214336
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000906258776328987,
      "loss": 1.6748,
      "theoretical_loss": 4.1160455073645865,
      "tokens_seen": 339279872
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009062387161484453,
      "loss": 1.5677,
      "theoretical_loss": 4.115954726436599,
      "tokens_seen": 339345408
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009062186559679038,
      "loss": 1.6308,
      "theoretical_loss": 4.115863967946856,
      "tokens_seen": 339410944
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000906198595787362,
      "loss": 1.7707,
      "theoretical_loss": 4.115773231885479,
      "tokens_seen": 339476480
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009061785356068205,
      "loss": 1.7975,
      "theoretical_loss": 4.115682518242599,
      "tokens_seen": 339542016
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009061584754262789,
      "loss": 1.6861,
      "theoretical_loss": 4.115591827008351,
      "tokens_seen": 339607552
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009061384152457372,
      "loss": 1.6768,
      "theoretical_loss": 4.115501158172876,
      "tokens_seen": 339673088
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009061183550651956,
      "loss": 1.7246,
      "theoretical_loss": 4.115410511726323,
      "tokens_seen": 339738624
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009060982948846541,
      "loss": 1.8952,
      "theoretical_loss": 4.115319887658846,
      "tokens_seen": 339804160
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009060782347041123,
      "loss": 1.6856,
      "theoretical_loss": 4.1152292859606066,
      "tokens_seen": 339869696
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009060581745235708,
      "loss": 1.6744,
      "theoretical_loss": 4.115138706621769,
      "tokens_seen": 339935232
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000906038114343029,
      "loss": 1.7339,
      "theoretical_loss": 4.115048149632507,
      "tokens_seen": 340000768
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009060180541624875,
      "loss": 1.8115,
      "theoretical_loss": 4.114957614983001,
      "tokens_seen": 340066304
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009059979939819459,
      "loss": 1.6957,
      "theoretical_loss": 4.114867102663435,
      "tokens_seen": 340131840
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009059779338014042,
      "loss": 1.8678,
      "theoretical_loss": 4.114776612664,
      "tokens_seen": 340197376
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009059578736208626,
      "loss": 1.7384,
      "theoretical_loss": 4.114686144974897,
      "tokens_seen": 340262912
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000905937813440321,
      "loss": 1.7461,
      "theoretical_loss": 4.114595699586326,
      "tokens_seen": 340328448
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009059177532597793,
      "loss": 1.7942,
      "theoretical_loss": 4.114505276488499,
      "tokens_seen": 340393984
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009058976930792378,
      "loss": 1.7345,
      "theoretical_loss": 4.114414875671631,
      "tokens_seen": 340459520
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000905877632898696,
      "loss": 1.7869,
      "theoretical_loss": 4.114324497125947,
      "tokens_seen": 340525056
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009058575727181545,
      "loss": 1.6033,
      "theoretical_loss": 4.114234140841673,
      "tokens_seen": 340590592
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009058375125376128,
      "loss": 1.8136,
      "theoretical_loss": 4.114143806809045,
      "tokens_seen": 340656128
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009058174523570712,
      "loss": 1.8029,
      "theoretical_loss": 4.114053495018304,
      "tokens_seen": 340721664
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.49362632632255554,
      "objective/train/docs_used": 130755,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7257463932037354,
      "objective/train/original_loss": 1.7257463932037354,
      "objective/train/theoretical_loss": 4.113963205459697,
      "objective/train/tokens_used": 361247200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24409334361553192,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506036281585693,
      "objective/train/weighted_lm_loss": 1.8130598068237305,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.113963205459697,
      "tokens_seen": 340787200
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009057973921765296,
      "loss": 1.7073,
      "theoretical_loss": 4.113963205459697,
      "tokens_seen": 340787200
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000905777331995988,
      "loss": 1.6982,
      "theoretical_loss": 4.113872938123477,
      "tokens_seen": 340852736
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009057572718154463,
      "loss": 1.8377,
      "theoretical_loss": 4.113782692999903,
      "tokens_seen": 340918272
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009057372116349048,
      "loss": 1.5813,
      "theoretical_loss": 4.1136924700792425,
      "tokens_seen": 340983808
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000905717151454363,
      "loss": 1.8254,
      "theoretical_loss": 4.113602269351765,
      "tokens_seen": 341049344
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009056970912738215,
      "loss": 1.7014,
      "theoretical_loss": 4.113512090807749,
      "tokens_seen": 341114880
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009056770310932798,
      "loss": 1.6438,
      "theoretical_loss": 4.113421934437479,
      "tokens_seen": 341180416
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009056569709127382,
      "loss": 1.8645,
      "theoretical_loss": 4.1133318002312445,
      "tokens_seen": 341245952
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009056369107321966,
      "loss": 1.6754,
      "theoretical_loss": 4.113241688179341,
      "tokens_seen": 341311488
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000905616850551655,
      "loss": 1.8817,
      "theoretical_loss": 4.113151598272072,
      "tokens_seen": 341377024
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009055967903711133,
      "loss": 1.8039,
      "theoretical_loss": 4.113061530499746,
      "tokens_seen": 341442560
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009055767301905717,
      "loss": 1.6546,
      "theoretical_loss": 4.112971484852677,
      "tokens_seen": 341508096
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009055566700100301,
      "loss": 1.7393,
      "theoretical_loss": 4.1128814613211855,
      "tokens_seen": 341573632
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009055366098294885,
      "loss": 1.7555,
      "theoretical_loss": 4.112791459895599,
      "tokens_seen": 341639168
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009055165496489468,
      "loss": 1.6987,
      "theoretical_loss": 4.112701480566248,
      "tokens_seen": 341704704
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009054964894684053,
      "loss": 1.7557,
      "theoretical_loss": 4.112611523323474,
      "tokens_seen": 341770240
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009054764292878635,
      "loss": 1.6285,
      "theoretical_loss": 4.1125215881576205,
      "tokens_seen": 341835776
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000905456369107322,
      "loss": 1.7713,
      "theoretical_loss": 4.112431675059039,
      "tokens_seen": 341901312
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009054363089267803,
      "loss": 1.7883,
      "theoretical_loss": 4.112341784018086,
      "tokens_seen": 341966848
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009054162487462387,
      "loss": 1.552,
      "theoretical_loss": 4.112251915025125,
      "tokens_seen": 342032384
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009053961885656972,
      "loss": 1.7311,
      "theoretical_loss": 4.112162068070525,
      "tokens_seen": 342097920
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009053761283851555,
      "loss": 1.6875,
      "theoretical_loss": 4.112072243144662,
      "tokens_seen": 342163456
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009053560682046139,
      "loss": 1.7803,
      "theoretical_loss": 4.111982440237917,
      "tokens_seen": 342228992
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009053360080240723,
      "loss": 1.7886,
      "theoretical_loss": 4.1118926593406755,
      "tokens_seen": 342294528
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009053159478435306,
      "loss": 1.7322,
      "theoretical_loss": 4.111802900443333,
      "tokens_seen": 342360064
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000905295887662989,
      "loss": 1.5982,
      "theoretical_loss": 4.111713163536289,
      "tokens_seen": 342425600
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009052758274824475,
      "loss": 1.7256,
      "theoretical_loss": 4.111623448609948,
      "tokens_seen": 342491136
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009052557673019057,
      "loss": 1.8172,
      "theoretical_loss": 4.111533755654721,
      "tokens_seen": 342556672
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009052357071213642,
      "loss": 1.6669,
      "theoretical_loss": 4.111444084661026,
      "tokens_seen": 342622208
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009052156469408224,
      "loss": 1.7317,
      "theoretical_loss": 4.111354435619287,
      "tokens_seen": 342687744
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009051955867602809,
      "loss": 1.7397,
      "theoretical_loss": 4.111264808519933,
      "tokens_seen": 342753280
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009051755265797393,
      "loss": 1.6709,
      "theoretical_loss": 4.111175203353399,
      "tokens_seen": 342818816
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009051554663991976,
      "loss": 1.6312,
      "theoretical_loss": 4.111085620110127,
      "tokens_seen": 342884352
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000905135406218656,
      "loss": 1.8186,
      "theoretical_loss": 4.1109960587805645,
      "tokens_seen": 342949888
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009051153460381145,
      "loss": 1.7051,
      "theoretical_loss": 4.110906519355165,
      "tokens_seen": 343015424
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009050952858575727,
      "loss": 1.7722,
      "theoretical_loss": 4.110817001824387,
      "tokens_seen": 343080960
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009050752256770312,
      "loss": 1.6726,
      "theoretical_loss": 4.110727506178697,
      "tokens_seen": 343146496
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009050551654964894,
      "loss": 1.8694,
      "theoretical_loss": 4.110638032408565,
      "tokens_seen": 343212032
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009050351053159479,
      "loss": 1.6671,
      "theoretical_loss": 4.110548580504471,
      "tokens_seen": 343277568
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009050150451354063,
      "loss": 1.8713,
      "theoretical_loss": 4.1104591504568955,
      "tokens_seen": 343343104
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009049949849548646,
      "loss": 1.7105,
      "theoretical_loss": 4.110369742256329,
      "tokens_seen": 343408640
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000904974924774323,
      "loss": 1.7494,
      "theoretical_loss": 4.110280355893266,
      "tokens_seen": 343474176
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009049548645937814,
      "loss": 1.7071,
      "theoretical_loss": 4.110190991358209,
      "tokens_seen": 343539712
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009049348044132397,
      "loss": 1.7533,
      "theoretical_loss": 4.110101648641665,
      "tokens_seen": 343605248
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009049147442326982,
      "loss": 1.6532,
      "theoretical_loss": 4.110012327734145,
      "tokens_seen": 343670784
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009048946840521564,
      "loss": 1.7937,
      "theoretical_loss": 4.10992302862617,
      "tokens_seen": 343736320
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009048746238716149,
      "loss": 1.7139,
      "theoretical_loss": 4.109833751308264,
      "tokens_seen": 343801856
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009048545636910732,
      "loss": 1.683,
      "theoretical_loss": 4.109744495770959,
      "tokens_seen": 343867392
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009048345035105316,
      "loss": 1.746,
      "theoretical_loss": 4.1096552620047895,
      "tokens_seen": 343932928
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.00090481444332999,
      "loss": 1.7218,
      "theoretical_loss": 4.1095660500003,
      "tokens_seen": 343998464
    },
    {
      "epoch": 0.1,
      "objective/train/advantage_avg": 0.49333980679512024,
      "objective/train/docs_used": 131936,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7032973766326904,
      "objective/train/original_loss": 1.7032976150512695,
      "objective/train/theoretical_loss": 4.109476859748038,
      "objective/train/tokens_used": 364524000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24367128312587738,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505727529525757,
      "objective/train/weighted_lm_loss": 1.789353609085083,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0168063640594482,
      "theoretical_loss": 4.109476859748038,
      "tokens_seen": 344064000
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009047943831494484,
      "loss": 1.9069,
      "theoretical_loss": 4.109476859748038,
      "tokens_seen": 344064000
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009047743229689067,
      "loss": 1.6547,
      "theoretical_loss": 4.109387691238559,
      "tokens_seen": 344129536
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009047542627883652,
      "loss": 1.736,
      "theoretical_loss": 4.109298544462423,
      "tokens_seen": 344195072
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009047342026078235,
      "loss": 1.6227,
      "theoretical_loss": 4.109209419410195,
      "tokens_seen": 344260608
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009047141424272819,
      "loss": 1.6497,
      "theoretical_loss": 4.10912031607245,
      "tokens_seen": 344326144
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009046940822467402,
      "loss": 1.6495,
      "theoretical_loss": 4.1090312344397635,
      "tokens_seen": 344391680
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009046740220661986,
      "loss": 1.6644,
      "theoretical_loss": 4.108942174502721,
      "tokens_seen": 344457216
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.000904653961885657,
      "loss": 1.7992,
      "theoretical_loss": 4.10885313625191,
      "tokens_seen": 344522752
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009046339017051154,
      "loss": 1.7062,
      "theoretical_loss": 4.108764119677929,
      "tokens_seen": 344588288
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009046138415245737,
      "loss": 1.6949,
      "theoretical_loss": 4.108675124771377,
      "tokens_seen": 344653824
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009045937813440321,
      "loss": 1.7387,
      "theoretical_loss": 4.108586151522863,
      "tokens_seen": 344719360
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009045737211634905,
      "loss": 1.866,
      "theoretical_loss": 4.108497199923001,
      "tokens_seen": 344784896
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009045536609829489,
      "loss": 1.7702,
      "theoretical_loss": 4.108408269962408,
      "tokens_seen": 344850432
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009045336008024072,
      "loss": 1.7425,
      "theoretical_loss": 4.108319361631709,
      "tokens_seen": 344915968
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009045135406218657,
      "loss": 1.773,
      "theoretical_loss": 4.1082304749215375,
      "tokens_seen": 344981504
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009044934804413239,
      "loss": 1.9007,
      "theoretical_loss": 4.108141609822527,
      "tokens_seen": 345047040
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009044734202607824,
      "loss": 1.6881,
      "theoretical_loss": 4.108052766325322,
      "tokens_seen": 345112576
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009044533600802407,
      "loss": 1.686,
      "theoretical_loss": 4.10796394442057,
      "tokens_seen": 345178112
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009044332998996991,
      "loss": 1.6889,
      "theoretical_loss": 4.107875144098925,
      "tokens_seen": 345243648
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009044132397191575,
      "loss": 1.7391,
      "theoretical_loss": 4.107786365351047,
      "tokens_seen": 345309184
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009043931795386159,
      "loss": 1.7828,
      "theoretical_loss": 4.107697608167602,
      "tokens_seen": 345374720
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009043731193580742,
      "loss": 1.8004,
      "theoretical_loss": 4.107608872539261,
      "tokens_seen": 345440256
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009043530591775327,
      "loss": 1.9854,
      "theoretical_loss": 4.107520158456703,
      "tokens_seen": 345505792
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009043329989969909,
      "loss": 1.6611,
      "theoretical_loss": 4.107431465910609,
      "tokens_seen": 345571328
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009043129388164494,
      "loss": 1.6514,
      "theoretical_loss": 4.107342794891668,
      "tokens_seen": 345636864
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009042928786359077,
      "loss": 1.7724,
      "theoretical_loss": 4.107254145390577,
      "tokens_seen": 345702400
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009042728184553661,
      "loss": 1.8089,
      "theoretical_loss": 4.107165517398034,
      "tokens_seen": 345767936
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009042527582748245,
      "loss": 1.8605,
      "theoretical_loss": 4.107076910904747,
      "tokens_seen": 345833472
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009042326980942828,
      "loss": 1.7117,
      "theoretical_loss": 4.106988325901428,
      "tokens_seen": 345899008
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009042126379137412,
      "loss": 1.7907,
      "theoretical_loss": 4.106899762378793,
      "tokens_seen": 345964544
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009041925777331997,
      "loss": 1.7211,
      "theoretical_loss": 4.106811220327568,
      "tokens_seen": 346030080
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009041725175526579,
      "loss": 1.7923,
      "theoretical_loss": 4.106722699738481,
      "tokens_seen": 346095616
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009041524573721164,
      "loss": 1.8162,
      "theoretical_loss": 4.106634200602267,
      "tokens_seen": 346161152
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009041323971915746,
      "loss": 1.7074,
      "theoretical_loss": 4.106545722909668,
      "tokens_seen": 346226688
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009041123370110331,
      "loss": 1.6258,
      "theoretical_loss": 4.10645726665143,
      "tokens_seen": 346292224
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009040922768304915,
      "loss": 1.6748,
      "theoretical_loss": 4.106368831818305,
      "tokens_seen": 346357760
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009040722166499498,
      "loss": 1.8313,
      "theoretical_loss": 4.1062804184010515,
      "tokens_seen": 346423296
    },
    {
      "epoch": 0.1,
      "learning_rate": 0.0009040521564694082,
      "loss": 1.8808,
      "theoretical_loss": 4.106192026390433,
      "tokens_seen": 346488832
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009040320962888667,
      "loss": 1.7614,
      "theoretical_loss": 4.10610365577722,
      "tokens_seen": 346554368
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009040120361083249,
      "loss": 1.6404,
      "theoretical_loss": 4.106015306552187,
      "tokens_seen": 346619904
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009039919759277834,
      "loss": 1.8777,
      "theoretical_loss": 4.1059269787061154,
      "tokens_seen": 346685440
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009039719157472416,
      "loss": 1.6846,
      "theoretical_loss": 4.105838672229792,
      "tokens_seen": 346750976
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009039518555667001,
      "loss": 1.7727,
      "theoretical_loss": 4.105750387114009,
      "tokens_seen": 346816512
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009039317953861585,
      "loss": 1.6671,
      "theoretical_loss": 4.105662123349564,
      "tokens_seen": 346882048
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009039117352056168,
      "loss": 1.7073,
      "theoretical_loss": 4.105573880927262,
      "tokens_seen": 346947584
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009038916750250753,
      "loss": 1.7958,
      "theoretical_loss": 4.105485659837912,
      "tokens_seen": 347013120
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009038716148445336,
      "loss": 1.7587,
      "theoretical_loss": 4.105397460072329,
      "tokens_seen": 347078656
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000903851554663992,
      "loss": 1.6212,
      "theoretical_loss": 4.105309281621334,
      "tokens_seen": 347144192
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009038314944834504,
      "loss": 1.619,
      "theoretical_loss": 4.105221124475754,
      "tokens_seen": 347209728
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009038114343029088,
      "loss": 1.7397,
      "theoretical_loss": 4.105132988626422,
      "tokens_seen": 347275264
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.49424776434898376,
      "objective/train/docs_used": 133174,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6256812810897827,
      "objective/train/original_loss": 1.6256814002990723,
      "objective/train/theoretical_loss": 4.105044874064174,
      "objective/train/tokens_used": 367800800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24457493424415588,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506681203842163,
      "objective/train/weighted_lm_loss": 1.7079211473464966,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.105044874064174,
      "tokens_seen": 347340800
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009037913741223671,
      "loss": 1.6559,
      "theoretical_loss": 4.105044874064174,
      "tokens_seen": 347340800
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009037713139418256,
      "loss": 1.7065,
      "theoretical_loss": 4.1049567807798555,
      "tokens_seen": 347406336
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009037512537612839,
      "loss": 1.7464,
      "theoretical_loss": 4.1048687087643145,
      "tokens_seen": 347471872
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009037311935807423,
      "loss": 1.7268,
      "theoretical_loss": 4.104780658008407,
      "tokens_seen": 347537408
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009037111334002006,
      "loss": 1.7817,
      "theoretical_loss": 4.104692628502993,
      "tokens_seen": 347602944
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000903691073219659,
      "loss": 1.6799,
      "theoretical_loss": 4.104604620238939,
      "tokens_seen": 347668480
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009036710130391174,
      "loss": 1.7885,
      "theoretical_loss": 4.104516633207116,
      "tokens_seen": 347734016
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009036509528585758,
      "loss": 1.6622,
      "theoretical_loss": 4.104428667398404,
      "tokens_seen": 347799552
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009036308926780341,
      "loss": 1.7008,
      "theoretical_loss": 4.104340722803683,
      "tokens_seen": 347865088
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009036108324974925,
      "loss": 1.6566,
      "theoretical_loss": 4.104252799413844,
      "tokens_seen": 347930624
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009035907723169509,
      "loss": 1.7029,
      "theoretical_loss": 4.104164897219781,
      "tokens_seen": 347996160
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009035707121364093,
      "loss": 1.7565,
      "theoretical_loss": 4.104077016212393,
      "tokens_seen": 348061696
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009035506519558676,
      "loss": 1.7934,
      "theoretical_loss": 4.103989156382589,
      "tokens_seen": 348127232
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000903530591775326,
      "loss": 1.7331,
      "theoretical_loss": 4.103901317721276,
      "tokens_seen": 348192768
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009035105315947843,
      "loss": 1.6297,
      "theoretical_loss": 4.103813500219374,
      "tokens_seen": 348258304
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009034904714142428,
      "loss": 1.7116,
      "theoretical_loss": 4.103725703867804,
      "tokens_seen": 348323840
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009034704112337011,
      "loss": 1.5938,
      "theoretical_loss": 4.103637928657495,
      "tokens_seen": 348389376
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009034503510531595,
      "loss": 1.8078,
      "theoretical_loss": 4.10355017457938,
      "tokens_seen": 348454912
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009034302908726179,
      "loss": 1.8058,
      "theoretical_loss": 4.103462441624398,
      "tokens_seen": 348520448
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009034102306920763,
      "loss": 1.7124,
      "theoretical_loss": 4.1033747297834955,
      "tokens_seen": 348585984
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009033901705115346,
      "loss": 1.9676,
      "theoretical_loss": 4.103287039047622,
      "tokens_seen": 348651520
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009033701103309931,
      "loss": 1.8403,
      "theoretical_loss": 4.103199369407734,
      "tokens_seen": 348717056
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009033500501504513,
      "loss": 1.7325,
      "theoretical_loss": 4.103111720854792,
      "tokens_seen": 348782592
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009033299899699098,
      "loss": 1.6372,
      "theoretical_loss": 4.103024093379764,
      "tokens_seen": 348848128
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009033099297893681,
      "loss": 1.7101,
      "theoretical_loss": 4.102936486973624,
      "tokens_seen": 348913664
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009032898696088265,
      "loss": 1.639,
      "theoretical_loss": 4.102848901627348,
      "tokens_seen": 348979200
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009032698094282849,
      "loss": 1.7128,
      "theoretical_loss": 4.102761337331921,
      "tokens_seen": 349044736
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009032497492477432,
      "loss": 1.7724,
      "theoretical_loss": 4.102673794078333,
      "tokens_seen": 349110272
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009032296890672016,
      "loss": 1.7385,
      "theoretical_loss": 4.102586271857579,
      "tokens_seen": 349175808
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009032096288866601,
      "loss": 1.6569,
      "theoretical_loss": 4.102498770660658,
      "tokens_seen": 349241344
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009031895687061183,
      "loss": 1.7852,
      "theoretical_loss": 4.102411290478577,
      "tokens_seen": 349306880
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009031695085255768,
      "loss": 1.6594,
      "theoretical_loss": 4.102323831302349,
      "tokens_seen": 349372416
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000903149448345035,
      "loss": 1.6494,
      "theoretical_loss": 4.102236393122989,
      "tokens_seen": 349437952
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009031293881644935,
      "loss": 1.7227,
      "theoretical_loss": 4.102148975931521,
      "tokens_seen": 349503488
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009031093279839519,
      "loss": 1.89,
      "theoretical_loss": 4.102061579718972,
      "tokens_seen": 349569024
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009030892678034102,
      "loss": 1.669,
      "theoretical_loss": 4.101974204476377,
      "tokens_seen": 349634560
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009030692076228686,
      "loss": 1.7407,
      "theoretical_loss": 4.101886850194775,
      "tokens_seen": 349700096
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009030491474423271,
      "loss": 1.7601,
      "theoretical_loss": 4.10179951686521,
      "tokens_seen": 349765632
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009030290872617853,
      "loss": 1.7153,
      "theoretical_loss": 4.1017122044787335,
      "tokens_seen": 349831168
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009030090270812438,
      "loss": 1.5519,
      "theoretical_loss": 4.1016249130263995,
      "tokens_seen": 349896704
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000902988966900702,
      "loss": 1.704,
      "theoretical_loss": 4.10153764249927,
      "tokens_seen": 349962240
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009029689067201605,
      "loss": 1.6163,
      "theoretical_loss": 4.101450392888412,
      "tokens_seen": 350027776
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009029488465396189,
      "loss": 1.7973,
      "theoretical_loss": 4.101363164184898,
      "tokens_seen": 350093312
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009029287863590772,
      "loss": 1.7025,
      "theoretical_loss": 4.101275956379805,
      "tokens_seen": 350158848
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009029087261785356,
      "loss": 1.6458,
      "theoretical_loss": 4.1011887694642155,
      "tokens_seen": 350224384
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000902888665997994,
      "loss": 1.7947,
      "theoretical_loss": 4.10110160342922,
      "tokens_seen": 350289920
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009028686058174523,
      "loss": 1.6099,
      "theoretical_loss": 4.101014458265912,
      "tokens_seen": 350355456
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009028485456369108,
      "loss": 1.7333,
      "theoretical_loss": 4.100927333965389,
      "tokens_seen": 350420992
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009028284854563691,
      "loss": 1.738,
      "theoretical_loss": 4.100840230518759,
      "tokens_seen": 350486528
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009028084252758275,
      "loss": 1.7421,
      "theoretical_loss": 4.10075314791713,
      "tokens_seen": 350552064
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.4893075227737427,
      "objective/train/docs_used": 134382,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6907188892364502,
      "objective/train/original_loss": 1.6907188892364502,
      "objective/train/theoretical_loss": 4.10066608615162,
      "objective/train/tokens_used": 371077600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24007414281368256,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0501511096954346,
      "objective/train/weighted_lm_loss": 1.7755835056304932,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9048373699188232,
      "theoretical_loss": 4.10066608615162,
      "tokens_seen": 350617600
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009027883650952859,
      "loss": 1.7911,
      "theoretical_loss": 4.10066608615162,
      "tokens_seen": 350617600
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009027683049147443,
      "loss": 1.707,
      "theoretical_loss": 4.100579045213349,
      "tokens_seen": 350683136
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009027482447342026,
      "loss": 1.7838,
      "theoretical_loss": 4.100492025093445,
      "tokens_seen": 350748672
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000902728184553661,
      "loss": 1.8857,
      "theoretical_loss": 4.100405025783039,
      "tokens_seen": 350814208
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009027081243731193,
      "loss": 1.7437,
      "theoretical_loss": 4.100318047273269,
      "tokens_seen": 350879744
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009026880641925778,
      "loss": 1.7306,
      "theoretical_loss": 4.100231089555279,
      "tokens_seen": 350945280
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009026680040120361,
      "loss": 1.6636,
      "theoretical_loss": 4.100144152620215,
      "tokens_seen": 351010816
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009026479438314945,
      "loss": 1.675,
      "theoretical_loss": 4.100057236459235,
      "tokens_seen": 351076352
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009026278836509528,
      "loss": 1.7887,
      "theoretical_loss": 4.099970341063495,
      "tokens_seen": 351141888
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009026078234704113,
      "loss": 1.835,
      "theoretical_loss": 4.099883466424161,
      "tokens_seen": 351207424
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009025877632898696,
      "loss": 1.6373,
      "theoretical_loss": 4.099796612532403,
      "tokens_seen": 351272960
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000902567703109328,
      "loss": 1.6707,
      "theoretical_loss": 4.099709779379397,
      "tokens_seen": 351338496
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009025476429287863,
      "loss": 1.757,
      "theoretical_loss": 4.099622966956324,
      "tokens_seen": 351404032
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009025275827482447,
      "loss": 1.688,
      "theoretical_loss": 4.09953617525437,
      "tokens_seen": 351469568
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009025075225677031,
      "loss": 1.7404,
      "theoretical_loss": 4.0994494042647265,
      "tokens_seen": 351535104
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009024874623871615,
      "loss": 1.648,
      "theoretical_loss": 4.099362653978591,
      "tokens_seen": 351600640
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009024674022066198,
      "loss": 1.8621,
      "theoretical_loss": 4.0992759243871655,
      "tokens_seen": 351666176
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009024473420260783,
      "loss": 1.6942,
      "theoretical_loss": 4.099189215481659,
      "tokens_seen": 351731712
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009024272818455366,
      "loss": 1.7197,
      "theoretical_loss": 4.099102527253285,
      "tokens_seen": 351797248
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000902407221664995,
      "loss": 1.7993,
      "theoretical_loss": 4.09901585969326,
      "tokens_seen": 351862784
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009023871614844535,
      "loss": 1.7709,
      "theoretical_loss": 4.098929212792811,
      "tokens_seen": 351928320
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009023671013039117,
      "loss": 1.7283,
      "theoretical_loss": 4.098842586543165,
      "tokens_seen": 351993856
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009023470411233702,
      "loss": 1.7918,
      "theoretical_loss": 4.098755980935557,
      "tokens_seen": 352059392
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009023269809428285,
      "loss": 1.7396,
      "theoretical_loss": 4.098669395961229,
      "tokens_seen": 352124928
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009023069207622869,
      "loss": 1.7039,
      "theoretical_loss": 4.098582831611425,
      "tokens_seen": 352190464
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009022868605817453,
      "loss": 1.6457,
      "theoretical_loss": 4.098496287877395,
      "tokens_seen": 352256000
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009022668004012036,
      "loss": 1.7406,
      "theoretical_loss": 4.0984097647503965,
      "tokens_seen": 352321536
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000902246740220662,
      "loss": 1.6913,
      "theoretical_loss": 4.09832326222169,
      "tokens_seen": 352387072
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009022266800401205,
      "loss": 1.8149,
      "theoretical_loss": 4.098236780282543,
      "tokens_seen": 352452608
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009022066198595787,
      "loss": 1.6853,
      "theoretical_loss": 4.098150318924227,
      "tokens_seen": 352518144
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009021865596790372,
      "loss": 1.7264,
      "theoretical_loss": 4.09806387813802,
      "tokens_seen": 352583680
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009021664994984954,
      "loss": 1.7176,
      "theoretical_loss": 4.097977457915205,
      "tokens_seen": 352649216
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009021464393179539,
      "loss": 1.7579,
      "theoretical_loss": 4.097891058247068,
      "tokens_seen": 352714752
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009021263791374123,
      "loss": 1.6048,
      "theoretical_loss": 4.097804679124906,
      "tokens_seen": 352780288
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009021063189568706,
      "loss": 1.6215,
      "theoretical_loss": 4.0977183205400145,
      "tokens_seen": 352845824
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000902086258776329,
      "loss": 1.6109,
      "theoretical_loss": 4.097631982483699,
      "tokens_seen": 352911360
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009020661985957875,
      "loss": 1.7666,
      "theoretical_loss": 4.097545664947268,
      "tokens_seen": 352976896
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009020461384152457,
      "loss": 1.6963,
      "theoretical_loss": 4.097459367922037,
      "tokens_seen": 353042432
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009020260782347042,
      "loss": 1.7839,
      "theoretical_loss": 4.097373091399324,
      "tokens_seen": 353107968
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009020060180541625,
      "loss": 1.7953,
      "theoretical_loss": 4.097286835370456,
      "tokens_seen": 353173504
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009019859578736209,
      "loss": 1.5052,
      "theoretical_loss": 4.097200599826763,
      "tokens_seen": 353239040
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009019658976930793,
      "loss": 1.8183,
      "theoretical_loss": 4.09711438475958,
      "tokens_seen": 353304576
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009019458375125376,
      "loss": 1.7284,
      "theoretical_loss": 4.097028190160249,
      "tokens_seen": 353370112
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000901925777331996,
      "loss": 1.7696,
      "theoretical_loss": 4.096942016020115,
      "tokens_seen": 353435648
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009019057171514544,
      "loss": 1.8074,
      "theoretical_loss": 4.096855862330531,
      "tokens_seen": 353501184
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009018856569709127,
      "loss": 1.6989,
      "theoretical_loss": 4.096769729082853,
      "tokens_seen": 353566720
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009018655967903712,
      "loss": 1.7032,
      "theoretical_loss": 4.096683616268442,
      "tokens_seen": 353632256
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009018455366098295,
      "loss": 1.7195,
      "theoretical_loss": 4.096597523878667,
      "tokens_seen": 353697792
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009018254764292879,
      "loss": 1.8013,
      "theoretical_loss": 4.096511451904899,
      "tokens_seen": 353763328
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009018054162487463,
      "loss": 1.687,
      "theoretical_loss": 4.096425400338516,
      "tokens_seen": 353828864
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.49126100540161133,
      "objective/train/docs_used": 135614,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4353805780410767,
      "objective/train/original_loss": 1.435380458831787,
      "objective/train/theoretical_loss": 4.096339369170902,
      "objective/train/tokens_used": 374354400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24204039573669434,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503566265106201,
      "objective/train/weighted_lm_loss": 1.5078595876693726,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.096339369170902,
      "tokens_seen": 353894400
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009017853560682047,
      "loss": 1.669,
      "theoretical_loss": 4.096339369170902,
      "tokens_seen": 353894400
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000901765295887663,
      "loss": 1.7519,
      "theoretical_loss": 4.096253358393445,
      "tokens_seen": 353959936
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009017452357071214,
      "loss": 1.7505,
      "theoretical_loss": 4.096167367997538,
      "tokens_seen": 354025472
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009017251755265797,
      "loss": 1.731,
      "theoretical_loss": 4.0960813979745785,
      "tokens_seen": 354091008
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009017051153460382,
      "loss": 1.749,
      "theoretical_loss": 4.095995448315972,
      "tokens_seen": 354156544
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009016850551654965,
      "loss": 1.7191,
      "theoretical_loss": 4.095909519013127,
      "tokens_seen": 354222080
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009016649949849549,
      "loss": 1.7147,
      "theoretical_loss": 4.095823610057458,
      "tokens_seen": 354287616
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009016449348044132,
      "loss": 1.7708,
      "theoretical_loss": 4.0957377214403845,
      "tokens_seen": 354353152
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009016248746238717,
      "loss": 1.6917,
      "theoretical_loss": 4.095651853153331,
      "tokens_seen": 354418688
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.00090160481444333,
      "loss": 1.7711,
      "theoretical_loss": 4.095566005187729,
      "tokens_seen": 354484224
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009015847542627884,
      "loss": 1.7752,
      "theoretical_loss": 4.095480177535011,
      "tokens_seen": 354549760
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009015646940822467,
      "loss": 1.7135,
      "theoretical_loss": 4.095394370186619,
      "tokens_seen": 354615296
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009015446339017051,
      "loss": 1.7425,
      "theoretical_loss": 4.095308583133997,
      "tokens_seen": 354680832
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009015245737211635,
      "loss": 1.7808,
      "theoretical_loss": 4.095222816368597,
      "tokens_seen": 354746368
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009015045135406219,
      "loss": 1.8011,
      "theoretical_loss": 4.095137069881876,
      "tokens_seen": 354811904
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009014844533600802,
      "loss": 1.736,
      "theoretical_loss": 4.095051343665292,
      "tokens_seen": 354877440
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009014643931795387,
      "loss": 1.7026,
      "theoretical_loss": 4.094965637710314,
      "tokens_seen": 354942976
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000901444332998997,
      "loss": 1.7112,
      "theoretical_loss": 4.094879952008412,
      "tokens_seen": 355008512
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009014242728184554,
      "loss": 1.8059,
      "theoretical_loss": 4.094794286551062,
      "tokens_seen": 355074048
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009014042126379138,
      "loss": 1.6808,
      "theoretical_loss": 4.094708641329748,
      "tokens_seen": 355139584
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009013841524573721,
      "loss": 1.7377,
      "theoretical_loss": 4.094623016335954,
      "tokens_seen": 355205120
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009013640922768305,
      "loss": 1.7313,
      "theoretical_loss": 4.094537411561174,
      "tokens_seen": 355270656
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000901344032096289,
      "loss": 1.6661,
      "theoretical_loss": 4.094451826996904,
      "tokens_seen": 355336192
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009013239719157472,
      "loss": 1.702,
      "theoretical_loss": 4.094366262634647,
      "tokens_seen": 355401728
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009013039117352057,
      "loss": 1.8119,
      "theoretical_loss": 4.094280718465911,
      "tokens_seen": 355467264
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009012838515546639,
      "loss": 1.7697,
      "theoretical_loss": 4.094195194482207,
      "tokens_seen": 355532800
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009012637913741224,
      "loss": 1.7872,
      "theoretical_loss": 4.0941096906750545,
      "tokens_seen": 355598336
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009012437311935808,
      "loss": 1.7232,
      "theoretical_loss": 4.094024207035974,
      "tokens_seen": 355663872
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009012236710130391,
      "loss": 1.6203,
      "theoretical_loss": 4.093938743556496,
      "tokens_seen": 355729408
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009012036108324975,
      "loss": 1.6374,
      "theoretical_loss": 4.093853300228152,
      "tokens_seen": 355794944
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009011835506519558,
      "loss": 1.7938,
      "theoretical_loss": 4.09376787704248,
      "tokens_seen": 355860480
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009011634904714142,
      "loss": 1.7368,
      "theoretical_loss": 4.0936824739910245,
      "tokens_seen": 355926016
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009011434302908727,
      "loss": 1.6517,
      "theoretical_loss": 4.093597091065333,
      "tokens_seen": 355991552
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009011233701103309,
      "loss": 1.7932,
      "theoretical_loss": 4.09351172825696,
      "tokens_seen": 356057088
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009011033099297894,
      "loss": 1.6453,
      "theoretical_loss": 4.093426385557462,
      "tokens_seen": 356122624
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009010832497492478,
      "loss": 1.7598,
      "theoretical_loss": 4.093341062958405,
      "tokens_seen": 356188160
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009010631895687061,
      "loss": 1.6493,
      "theoretical_loss": 4.093255760451357,
      "tokens_seen": 356253696
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009010431293881645,
      "loss": 1.7055,
      "theoretical_loss": 4.093170478027892,
      "tokens_seen": 356319232
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009010230692076229,
      "loss": 1.698,
      "theoretical_loss": 4.093085215679588,
      "tokens_seen": 356384768
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009010030090270812,
      "loss": 1.7086,
      "theoretical_loss": 4.0929999733980305,
      "tokens_seen": 356450304
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009009829488465397,
      "loss": 1.6501,
      "theoretical_loss": 4.092914751174808,
      "tokens_seen": 356515840
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009009628886659979,
      "loss": 1.7705,
      "theoretical_loss": 4.092829549001514,
      "tokens_seen": 356581376
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009009428284854564,
      "loss": 1.7081,
      "theoretical_loss": 4.0927443668697485,
      "tokens_seen": 356646912
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009009227683049147,
      "loss": 1.7028,
      "theoretical_loss": 4.092659204771116,
      "tokens_seen": 356712448
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009009027081243731,
      "loss": 1.7112,
      "theoretical_loss": 4.092574062697225,
      "tokens_seen": 356777984
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009008826479438316,
      "loss": 1.6097,
      "theoretical_loss": 4.092488940639691,
      "tokens_seen": 356843520
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009008625877632899,
      "loss": 1.705,
      "theoretical_loss": 4.0924038385901325,
      "tokens_seen": 356909056
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009008425275827483,
      "loss": 1.7643,
      "theoretical_loss": 4.092318756540174,
      "tokens_seen": 356974592
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009008224674022067,
      "loss": 1.6911,
      "theoretical_loss": 4.092233694481447,
      "tokens_seen": 357040128
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000900802407221665,
      "loss": 1.6974,
      "theoretical_loss": 4.092148652405584,
      "tokens_seen": 357105664
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.4929529130458832,
      "objective/train/docs_used": 136331,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7153905630111694,
      "objective/train/original_loss": 1.715390682220459,
      "objective/train/theoretical_loss": 4.092063630304224,
      "objective/train/tokens_used": 377631200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24330100417137146,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505321025848389,
      "objective/train/weighted_lm_loss": 1.8019219636917114,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.092063630304224,
      "tokens_seen": 357171200
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009007823470411234,
      "loss": 1.803,
      "theoretical_loss": 4.092063630304224,
      "tokens_seen": 357171200
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009007622868605818,
      "loss": 1.8053,
      "theoretical_loss": 4.091978628169015,
      "tokens_seen": 357236736
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009007422266800401,
      "loss": 1.799,
      "theoretical_loss": 4.091893645991604,
      "tokens_seen": 357302272
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009007221664994986,
      "loss": 1.7151,
      "theoretical_loss": 4.091808683763646,
      "tokens_seen": 357367808
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009007021063189569,
      "loss": 1.8246,
      "theoretical_loss": 4.091723741476802,
      "tokens_seen": 357433344
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009006820461384153,
      "loss": 1.7746,
      "theoretical_loss": 4.091638819122735,
      "tokens_seen": 357498880
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009006619859578736,
      "loss": 1.6607,
      "theoretical_loss": 4.091553916693115,
      "tokens_seen": 357564416
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009006419257773321,
      "loss": 1.6609,
      "theoretical_loss": 4.091469034179619,
      "tokens_seen": 357629952
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009006218655967904,
      "loss": 1.5002,
      "theoretical_loss": 4.0913841715739245,
      "tokens_seen": 357695488
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009006018054162488,
      "loss": 1.6997,
      "theoretical_loss": 4.091299328867717,
      "tokens_seen": 357761024
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009005817452357071,
      "loss": 1.7176,
      "theoretical_loss": 4.091214506052687,
      "tokens_seen": 357826560
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009005616850551655,
      "loss": 1.5455,
      "theoretical_loss": 4.091129703120528,
      "tokens_seen": 357892096
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009005416248746239,
      "loss": 1.6436,
      "theoretical_loss": 4.09104492006294,
      "tokens_seen": 357957632
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009005215646940823,
      "loss": 1.6813,
      "theoretical_loss": 4.090960156871628,
      "tokens_seen": 358023168
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009005015045135406,
      "loss": 1.5725,
      "theoretical_loss": 4.090875413538302,
      "tokens_seen": 358088704
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009004814443329991,
      "loss": 1.8228,
      "theoretical_loss": 4.090790690054677,
      "tokens_seen": 358154240
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009004613841524574,
      "loss": 1.6769,
      "theoretical_loss": 4.0907059864124715,
      "tokens_seen": 358219776
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009004413239719158,
      "loss": 1.815,
      "theoretical_loss": 4.090621302603411,
      "tokens_seen": 358285312
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009004212637913742,
      "loss": 1.7456,
      "theoretical_loss": 4.090536638619224,
      "tokens_seen": 358350848
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009004012036108325,
      "loss": 1.6841,
      "theoretical_loss": 4.090451994451647,
      "tokens_seen": 358416384
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009003811434302909,
      "loss": 1.7235,
      "theoretical_loss": 4.090367370092419,
      "tokens_seen": 358481920
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009003610832497493,
      "loss": 1.6922,
      "theoretical_loss": 4.090282765533283,
      "tokens_seen": 358547456
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009003410230692076,
      "loss": 1.5614,
      "theoretical_loss": 4.090198180765989,
      "tokens_seen": 358612992
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009003209628886661,
      "loss": 1.738,
      "theoretical_loss": 4.090113615782292,
      "tokens_seen": 358678528
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009003009027081243,
      "loss": 1.6384,
      "theoretical_loss": 4.0900290705739515,
      "tokens_seen": 358744064
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009002808425275828,
      "loss": 1.8113,
      "theoretical_loss": 4.089944545132731,
      "tokens_seen": 358809600
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009002607823470412,
      "loss": 1.667,
      "theoretical_loss": 4.0898600394504,
      "tokens_seen": 358875136
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009002407221664995,
      "loss": 1.7378,
      "theoretical_loss": 4.089775553518733,
      "tokens_seen": 358940672
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009002206619859579,
      "loss": 1.7439,
      "theoretical_loss": 4.089691087329507,
      "tokens_seen": 359006208
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009002006018054162,
      "loss": 1.6211,
      "theoretical_loss": 4.089606640874509,
      "tokens_seen": 359071744
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009001805416248746,
      "loss": 1.688,
      "theoretical_loss": 4.089522214145525,
      "tokens_seen": 359137280
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009001604814443331,
      "loss": 1.7826,
      "theoretical_loss": 4.089437807134351,
      "tokens_seen": 359202816
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009001404212637913,
      "loss": 1.8018,
      "theoretical_loss": 4.0893534198327846,
      "tokens_seen": 359268352
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009001203610832498,
      "loss": 1.6606,
      "theoretical_loss": 4.0892690522326305,
      "tokens_seen": 359333888
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009001003009027082,
      "loss": 1.7576,
      "theoretical_loss": 4.089184704325695,
      "tokens_seen": 359399424
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009000802407221665,
      "loss": 1.6624,
      "theoretical_loss": 4.089100376103794,
      "tokens_seen": 359464960
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009000601805416249,
      "loss": 1.7172,
      "theoretical_loss": 4.0890160675587435,
      "tokens_seen": 359530496
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009000401203610833,
      "loss": 1.7355,
      "theoretical_loss": 4.088931778682369,
      "tokens_seen": 359596032
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009000200601805416,
      "loss": 1.6584,
      "theoretical_loss": 4.088847509466497,
      "tokens_seen": 359661568
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0009000000000000001,
      "loss": 1.6495,
      "theoretical_loss": 4.088763259902961,
      "tokens_seen": 359727104
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008999799398194583,
      "loss": 1.6659,
      "theoretical_loss": 4.088679029983599,
      "tokens_seen": 359792640
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008999598796389168,
      "loss": 1.6201,
      "theoretical_loss": 4.088594819700253,
      "tokens_seen": 359858176
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008999398194583751,
      "loss": 1.8343,
      "theoretical_loss": 4.088510629044771,
      "tokens_seen": 359923712
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008999197592778335,
      "loss": 1.6573,
      "theoretical_loss": 4.088426458009005,
      "tokens_seen": 359989248
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008998996990972919,
      "loss": 1.7296,
      "theoretical_loss": 4.0883423065848135,
      "tokens_seen": 360054784
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008998796389167503,
      "loss": 1.7597,
      "theoretical_loss": 4.088258174764058,
      "tokens_seen": 360120320
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008998595787362086,
      "loss": 1.62,
      "theoretical_loss": 4.088174062538605,
      "tokens_seen": 360185856
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008998395185556671,
      "loss": 1.7132,
      "theoretical_loss": 4.088089969900327,
      "tokens_seen": 360251392
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008998194583751253,
      "loss": 1.8231,
      "theoretical_loss": 4.088005896841102,
      "tokens_seen": 360316928
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008997993981945838,
      "loss": 1.77,
      "theoretical_loss": 4.087921843352808,
      "tokens_seen": 360382464
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.49181121587753296,
      "objective/train/docs_used": 137426,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.762984037399292,
      "objective/train/original_loss": 1.7629839181900024,
      "objective/train/theoretical_loss": 4.087837809427334,
      "objective/train/tokens_used": 380908000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2422368824481964,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504125356674194,
      "objective/train/weighted_lm_loss": 1.8516653776168823,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.087837809427334,
      "tokens_seen": 360448000
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008997793380140421,
      "loss": 1.8454,
      "theoretical_loss": 4.087837809427334,
      "tokens_seen": 360448000
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008997592778335005,
      "loss": 1.7136,
      "theoretical_loss": 4.087753795056573,
      "tokens_seen": 360513536
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008997392176529589,
      "loss": 1.6479,
      "theoretical_loss": 4.087669800232417,
      "tokens_seen": 360579072
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008997191574724173,
      "loss": 1.6049,
      "theoretical_loss": 4.08758582494677,
      "tokens_seen": 360644608
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008996990972918756,
      "loss": 1.7941,
      "theoretical_loss": 4.087501869191536,
      "tokens_seen": 360710144
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000899679037111334,
      "loss": 1.793,
      "theoretical_loss": 4.087417932958627,
      "tokens_seen": 360775680
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008996589769307924,
      "loss": 1.6668,
      "theoretical_loss": 4.087334016239957,
      "tokens_seen": 360841216
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008996389167502508,
      "loss": 1.5709,
      "theoretical_loss": 4.0872501190274475,
      "tokens_seen": 360906752
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008996188565697091,
      "loss": 1.6844,
      "theoretical_loss": 4.087166241313023,
      "tokens_seen": 360972288
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008995987963891675,
      "loss": 1.7546,
      "theoretical_loss": 4.087082383088613,
      "tokens_seen": 361037824
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008995787362086258,
      "loss": 1.7811,
      "theoretical_loss": 4.086998544346152,
      "tokens_seen": 361103360
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008995586760280843,
      "loss": 1.845,
      "theoretical_loss": 4.08691472507758,
      "tokens_seen": 361168896
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008995386158475426,
      "loss": 1.744,
      "theoretical_loss": 4.086830925274842,
      "tokens_seen": 361234432
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000899518555667001,
      "loss": 1.7291,
      "theoretical_loss": 4.086747144929885,
      "tokens_seen": 361299968
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008994984954864594,
      "loss": 1.6854,
      "theoretical_loss": 4.086663384034664,
      "tokens_seen": 361365504
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008994784353059178,
      "loss": 1.712,
      "theoretical_loss": 4.086579642581138,
      "tokens_seen": 361431040
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008994583751253761,
      "loss": 1.6976,
      "theoretical_loss": 4.08649592056127,
      "tokens_seen": 361496576
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008994383149448346,
      "loss": 1.8639,
      "theoretical_loss": 4.086412217967028,
      "tokens_seen": 361562112
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008994182547642928,
      "loss": 1.8171,
      "theoretical_loss": 4.086328534790384,
      "tokens_seen": 361627648
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008993981945837513,
      "loss": 1.6998,
      "theoretical_loss": 4.086244871023318,
      "tokens_seen": 361693184
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008993781344032097,
      "loss": 1.7485,
      "theoretical_loss": 4.086161226657811,
      "tokens_seen": 361758720
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000899358074222668,
      "loss": 1.6268,
      "theoretical_loss": 4.086077601685851,
      "tokens_seen": 361824256
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008993380140421265,
      "loss": 1.6268,
      "theoretical_loss": 4.085993996099428,
      "tokens_seen": 361889792
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008993179538615847,
      "loss": 1.5707,
      "theoretical_loss": 4.085910409890541,
      "tokens_seen": 361955328
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008992978936810432,
      "loss": 1.7354,
      "theoretical_loss": 4.085826843051191,
      "tokens_seen": 362020864
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008992778335005016,
      "loss": 1.7003,
      "theoretical_loss": 4.0857432955733834,
      "tokens_seen": 362086400
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008992577733199599,
      "loss": 1.794,
      "theoretical_loss": 4.085659767449131,
      "tokens_seen": 362151936
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008992377131394183,
      "loss": 1.832,
      "theoretical_loss": 4.0855762586704465,
      "tokens_seen": 362217472
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008992176529588766,
      "loss": 1.7118,
      "theoretical_loss": 4.0854927692293534,
      "tokens_seen": 362283008
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000899197592778335,
      "loss": 1.661,
      "theoretical_loss": 4.085409299117876,
      "tokens_seen": 362348544
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008991775325977935,
      "loss": 1.7882,
      "theoretical_loss": 4.0853258483280435,
      "tokens_seen": 362414080
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008991574724172517,
      "loss": 1.6931,
      "theoretical_loss": 4.085242416851891,
      "tokens_seen": 362479616
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008991374122367102,
      "loss": 1.6965,
      "theoretical_loss": 4.085159004681458,
      "tokens_seen": 362545152
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008991173520561686,
      "loss": 1.7359,
      "theoretical_loss": 4.085075611808788,
      "tokens_seen": 362610688
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008990972918756269,
      "loss": 1.6945,
      "theoretical_loss": 4.084992238225931,
      "tokens_seen": 362676224
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008990772316950853,
      "loss": 1.7613,
      "theoretical_loss": 4.08490888392494,
      "tokens_seen": 362741760
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008990571715145437,
      "loss": 1.7596,
      "theoretical_loss": 4.084825548897873,
      "tokens_seen": 362807296
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000899037111334002,
      "loss": 1.7875,
      "theoretical_loss": 4.084742233136794,
      "tokens_seen": 362872832
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008990170511534605,
      "loss": 1.7156,
      "theoretical_loss": 4.084658936633769,
      "tokens_seen": 362938368
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008989969909729187,
      "loss": 1.8888,
      "theoretical_loss": 4.084575659380871,
      "tokens_seen": 363003904
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008989769307923772,
      "loss": 1.7725,
      "theoretical_loss": 4.084492401370177,
      "tokens_seen": 363069440
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008989568706118355,
      "loss": 1.796,
      "theoretical_loss": 4.0844091625937695,
      "tokens_seen": 363134976
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008989368104312939,
      "loss": 1.7835,
      "theoretical_loss": 4.084325943043734,
      "tokens_seen": 363200512
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008989167502507523,
      "loss": 1.7005,
      "theoretical_loss": 4.084242742712163,
      "tokens_seen": 363266048
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008988966900702107,
      "loss": 1.7358,
      "theoretical_loss": 4.08415956159115,
      "tokens_seen": 363331584
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000898876629889669,
      "loss": 1.7673,
      "theoretical_loss": 4.084076399672798,
      "tokens_seen": 363397120
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008988565697091275,
      "loss": 1.6401,
      "theoretical_loss": 4.08399325694921,
      "tokens_seen": 363462656
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008988365095285857,
      "loss": 1.7859,
      "theoretical_loss": 4.0839101334124965,
      "tokens_seen": 363528192
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008988164493480442,
      "loss": 1.6849,
      "theoretical_loss": 4.083827029054773,
      "tokens_seen": 363593728
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008987963891675025,
      "loss": 1.7831,
      "theoretical_loss": 4.083743943868157,
      "tokens_seen": 363659264
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.49359139800071716,
      "objective/train/docs_used": 138686,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7015258073806763,
      "objective/train/original_loss": 1.7015259265899658,
      "objective/train/theoretical_loss": 4.083660877844774,
      "objective/train/tokens_used": 384184800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2438880205154419,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505989789962769,
      "objective/train/weighted_lm_loss": 1.7876434326171875,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.083660877844774,
      "tokens_seen": 363724800
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008987763289869609,
      "loss": 1.7741,
      "theoretical_loss": 4.083660877844774,
      "tokens_seen": 363724800
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008987562688064193,
      "loss": 1.7944,
      "theoretical_loss": 4.0835778309767505,
      "tokens_seen": 363790336
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008987362086258777,
      "loss": 1.6954,
      "theoretical_loss": 4.0834948032562215,
      "tokens_seen": 363855872
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000898716148445336,
      "loss": 1.8153,
      "theoretical_loss": 4.083411794675324,
      "tokens_seen": 363921408
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008986960882647944,
      "loss": 1.7651,
      "theoretical_loss": 4.083328805226199,
      "tokens_seen": 363986944
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008986760280842528,
      "loss": 1.6811,
      "theoretical_loss": 4.083245834900996,
      "tokens_seen": 364052480
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008986559679037112,
      "loss": 1.755,
      "theoretical_loss": 4.083162883691864,
      "tokens_seen": 364118016
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008986359077231695,
      "loss": 1.7445,
      "theoretical_loss": 4.083079951590962,
      "tokens_seen": 364183552
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008986158475426279,
      "loss": 1.8182,
      "theoretical_loss": 4.08299703859045,
      "tokens_seen": 364249088
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008985957873620862,
      "loss": 1.6772,
      "theoretical_loss": 4.082914144682492,
      "tokens_seen": 364314624
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008985757271815447,
      "loss": 1.7932,
      "theoretical_loss": 4.082831269859261,
      "tokens_seen": 364380160
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000898555667001003,
      "loss": 1.8315,
      "theoretical_loss": 4.082748414112929,
      "tokens_seen": 364445696
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008985356068204614,
      "loss": 1.6778,
      "theoretical_loss": 4.082665577435677,
      "tokens_seen": 364511232
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008985155466399198,
      "loss": 1.7543,
      "theoretical_loss": 4.082582759819688,
      "tokens_seen": 364576768
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008984954864593782,
      "loss": 1.6663,
      "theoretical_loss": 4.0824999612571515,
      "tokens_seen": 364642304
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008984754262788365,
      "loss": 1.7071,
      "theoretical_loss": 4.082417181740261,
      "tokens_seen": 364707840
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000898455366098295,
      "loss": 1.6863,
      "theoretical_loss": 4.082334421261213,
      "tokens_seen": 364773376
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008984353059177532,
      "loss": 1.6753,
      "theoretical_loss": 4.082251679812211,
      "tokens_seen": 364838912
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008984152457372117,
      "loss": 1.7689,
      "theoretical_loss": 4.082168957385462,
      "tokens_seen": 364904448
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.00089839518555667,
      "loss": 1.8623,
      "theoretical_loss": 4.082086253973177,
      "tokens_seen": 364969984
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008983751253761284,
      "loss": 1.78,
      "theoretical_loss": 4.082003569567573,
      "tokens_seen": 365035520
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008983550651955868,
      "loss": 1.6577,
      "theoretical_loss": 4.08192090416087,
      "tokens_seen": 365101056
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008983350050150451,
      "loss": 1.7802,
      "theoretical_loss": 4.081838257745293,
      "tokens_seen": 365166592
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008983149448345035,
      "loss": 1.7077,
      "theoretical_loss": 4.081755630313074,
      "tokens_seen": 365232128
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000898294884653962,
      "loss": 1.6648,
      "theoretical_loss": 4.081673021856445,
      "tokens_seen": 365297664
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008982748244734202,
      "loss": 1.7078,
      "theoretical_loss": 4.081590432367647,
      "tokens_seen": 365363200
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008982547642928787,
      "loss": 1.6293,
      "theoretical_loss": 4.081507861838922,
      "tokens_seen": 365428736
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008982347041123369,
      "loss": 1.6485,
      "theoretical_loss": 4.08142531026252,
      "tokens_seen": 365494272
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008982146439317954,
      "loss": 1.7936,
      "theoretical_loss": 4.081342777630692,
      "tokens_seen": 365559808
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008981945837512538,
      "loss": 1.6933,
      "theoretical_loss": 4.081260263935697,
      "tokens_seen": 365625344
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008981745235707121,
      "loss": 1.7872,
      "theoretical_loss": 4.081177769169795,
      "tokens_seen": 365690880
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008981544633901705,
      "loss": 1.6915,
      "theoretical_loss": 4.081095293325253,
      "tokens_seen": 365756416
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000898134403209629,
      "loss": 1.6843,
      "theoretical_loss": 4.081012836394344,
      "tokens_seen": 365821952
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008981143430290872,
      "loss": 1.7095,
      "theoretical_loss": 4.0809303983693415,
      "tokens_seen": 365887488
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008980942828485457,
      "loss": 1.6318,
      "theoretical_loss": 4.080847979242526,
      "tokens_seen": 365953024
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008980742226680039,
      "loss": 1.7328,
      "theoretical_loss": 4.080765579006181,
      "tokens_seen": 366018560
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008980541624874624,
      "loss": 1.7227,
      "theoretical_loss": 4.080683197652597,
      "tokens_seen": 366084096
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008980341023069208,
      "loss": 1.7973,
      "theoretical_loss": 4.080600835174067,
      "tokens_seen": 366149632
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008980140421263791,
      "loss": 1.7067,
      "theoretical_loss": 4.0805184915628905,
      "tokens_seen": 366215168
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008979939819458375,
      "loss": 1.724,
      "theoretical_loss": 4.080436166811368,
      "tokens_seen": 366280704
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008979739217652959,
      "loss": 1.6432,
      "theoretical_loss": 4.080353860911808,
      "tokens_seen": 366346240
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008979538615847542,
      "loss": 1.7096,
      "theoretical_loss": 4.080271573856523,
      "tokens_seen": 366411776
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008979338014042127,
      "loss": 1.7991,
      "theoretical_loss": 4.080189305637827,
      "tokens_seen": 366477312
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000897913741223671,
      "loss": 1.6541,
      "theoretical_loss": 4.080107056248043,
      "tokens_seen": 366542848
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008978936810431294,
      "loss": 1.7085,
      "theoretical_loss": 4.0800248256794935,
      "tokens_seen": 366608384
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008978736208625879,
      "loss": 1.7117,
      "theoretical_loss": 4.07994261392451,
      "tokens_seen": 366673920
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008978535606820461,
      "loss": 1.8144,
      "theoretical_loss": 4.079860420975429,
      "tokens_seen": 366739456
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008978335005015046,
      "loss": 1.8416,
      "theoretical_loss": 4.079778246824585,
      "tokens_seen": 366804992
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008978134403209629,
      "loss": 1.7117,
      "theoretical_loss": 4.0796960914643225,
      "tokens_seen": 366870528
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008977933801404213,
      "loss": 1.9746,
      "theoretical_loss": 4.079613954886991,
      "tokens_seen": 366936064
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.4951712489128113,
      "objective/train/docs_used": 139849,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7049927711486816,
      "objective/train/original_loss": 1.7049928903579712,
      "objective/train/theoretical_loss": 4.07953183708494,
      "objective/train/tokens_used": 387461600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2454361617565155,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507649183273315,
      "objective/train/weighted_lm_loss": 1.7913212776184082,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.07953183708494,
      "tokens_seen": 367001600
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008977733199598797,
      "loss": 1.7663,
      "theoretical_loss": 4.07953183708494,
      "tokens_seen": 367001600
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008977532597793381,
      "loss": 1.7864,
      "theoretical_loss": 4.0794497380505295,
      "tokens_seen": 367067136
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008977331995987964,
      "loss": 1.7711,
      "theoretical_loss": 4.0793676577761175,
      "tokens_seen": 367132672
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008977131394182548,
      "loss": 1.7033,
      "theoretical_loss": 4.079285596254071,
      "tokens_seen": 367198208
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008976930792377132,
      "loss": 1.7523,
      "theoretical_loss": 4.079203553476759,
      "tokens_seen": 367263744
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008976730190571716,
      "loss": 1.7355,
      "theoretical_loss": 4.0791215294365575,
      "tokens_seen": 367329280
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008976529588766299,
      "loss": 1.7067,
      "theoretical_loss": 4.079039524125845,
      "tokens_seen": 367394816
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008976328986960883,
      "loss": 1.8486,
      "theoretical_loss": 4.078957537537004,
      "tokens_seen": 367460352
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008976128385155466,
      "loss": 1.7477,
      "theoretical_loss": 4.078875569662424,
      "tokens_seen": 367525888
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008975927783350051,
      "loss": 1.6189,
      "theoretical_loss": 4.078793620494497,
      "tokens_seen": 367591424
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008975727181544634,
      "loss": 1.6504,
      "theoretical_loss": 4.078711690025619,
      "tokens_seen": 367656960
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008975526579739218,
      "loss": 1.8155,
      "theoretical_loss": 4.0786297782481915,
      "tokens_seen": 367722496
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008975325977933802,
      "loss": 1.7302,
      "theoretical_loss": 4.07854788515462,
      "tokens_seen": 367788032
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008975125376128386,
      "loss": 1.6967,
      "theoretical_loss": 4.078466010737316,
      "tokens_seen": 367853568
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008974924774322969,
      "loss": 1.6561,
      "theoretical_loss": 4.078384154988692,
      "tokens_seen": 367919104
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008974724172517554,
      "loss": 1.7334,
      "theoretical_loss": 4.078302317901168,
      "tokens_seen": 367984640
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008974523570712136,
      "loss": 1.7801,
      "theoretical_loss": 4.078220499467168,
      "tokens_seen": 368050176
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008974322968906721,
      "loss": 1.7574,
      "theoretical_loss": 4.078138699679119,
      "tokens_seen": 368115712
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008974122367101304,
      "loss": 1.7281,
      "theoretical_loss": 4.0780569185294535,
      "tokens_seen": 368181248
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008973921765295888,
      "loss": 1.5919,
      "theoretical_loss": 4.077975156010609,
      "tokens_seen": 368246784
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008973721163490472,
      "loss": 1.8051,
      "theoretical_loss": 4.077893412115025,
      "tokens_seen": 368312320
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008973520561685055,
      "loss": 1.7476,
      "theoretical_loss": 4.077811686835148,
      "tokens_seen": 368377856
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008973319959879639,
      "loss": 1.6414,
      "theoretical_loss": 4.077729980163428,
      "tokens_seen": 368443392
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008973119358074224,
      "loss": 1.6917,
      "theoretical_loss": 4.07764829209232,
      "tokens_seen": 368508928
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008972918756268806,
      "loss": 1.6509,
      "theoretical_loss": 4.077566622614281,
      "tokens_seen": 368574464
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008972718154463391,
      "loss": 1.5947,
      "theoretical_loss": 4.077484971721775,
      "tokens_seen": 368640000
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008972517552657973,
      "loss": 1.6623,
      "theoretical_loss": 4.07740333940727,
      "tokens_seen": 368705536
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008972316950852558,
      "loss": 1.8069,
      "theoretical_loss": 4.077321725663237,
      "tokens_seen": 368771072
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008972116349047142,
      "loss": 1.7939,
      "theoretical_loss": 4.077240130482153,
      "tokens_seen": 368836608
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008971915747241725,
      "loss": 1.7251,
      "theoretical_loss": 4.077158553856497,
      "tokens_seen": 368902144
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008971715145436309,
      "loss": 1.7161,
      "theoretical_loss": 4.077076995778757,
      "tokens_seen": 368967680
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008971514543630894,
      "loss": 1.8689,
      "theoretical_loss": 4.076995456241421,
      "tokens_seen": 369033216
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008971313941825476,
      "loss": 1.7164,
      "theoretical_loss": 4.076913935236982,
      "tokens_seen": 369098752
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008971113340020061,
      "loss": 1.8138,
      "theoretical_loss": 4.076832432757939,
      "tokens_seen": 369164288
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008970912738214643,
      "loss": 1.703,
      "theoretical_loss": 4.0767509487967954,
      "tokens_seen": 369229824
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008970712136409228,
      "loss": 1.6488,
      "theoretical_loss": 4.076669483346056,
      "tokens_seen": 369295360
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008970511534603812,
      "loss": 1.6958,
      "theoretical_loss": 4.076588036398235,
      "tokens_seen": 369360896
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008970310932798395,
      "loss": 1.6487,
      "theoretical_loss": 4.076506607945845,
      "tokens_seen": 369426432
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008970110330992979,
      "loss": 1.7663,
      "theoretical_loss": 4.076425197981408,
      "tokens_seen": 369491968
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008969909729187563,
      "loss": 1.7376,
      "theoretical_loss": 4.076343806497446,
      "tokens_seen": 369557504
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008969709127382146,
      "loss": 1.6712,
      "theoretical_loss": 4.076262433486491,
      "tokens_seen": 369623040
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008969508525576731,
      "loss": 1.79,
      "theoretical_loss": 4.076181078941074,
      "tokens_seen": 369688576
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008969307923771313,
      "loss": 1.8104,
      "theoretical_loss": 4.076099742853732,
      "tokens_seen": 369754112
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008969107321965898,
      "loss": 1.7615,
      "theoretical_loss": 4.076018425217009,
      "tokens_seen": 369819648
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008968906720160481,
      "loss": 1.6527,
      "theoretical_loss": 4.075937126023448,
      "tokens_seen": 369885184
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008968706118355065,
      "loss": 1.6824,
      "theoretical_loss": 4.075855845265602,
      "tokens_seen": 369950720
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008968505516549649,
      "loss": 1.6784,
      "theoretical_loss": 4.075774582936024,
      "tokens_seen": 370016256
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008968304914744233,
      "loss": 1.737,
      "theoretical_loss": 4.075693339027274,
      "tokens_seen": 370081792
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008968104312938816,
      "loss": 1.7294,
      "theoretical_loss": 4.075612113531915,
      "tokens_seen": 370147328
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008967903711133401,
      "loss": 1.7808,
      "theoretical_loss": 4.0755309064425145,
      "tokens_seen": 370212864
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.48967820405960083,
      "objective/train/docs_used": 141078,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6121673583984375,
      "objective/train/original_loss": 1.6121673583984375,
      "objective/train/theoretical_loss": 4.0754497177516456,
      "objective/train/tokens_used": 390738400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24023215472698212,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0501888990402222,
      "objective/train/weighted_lm_loss": 1.6929486989974976,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.0754497177516456,
      "tokens_seen": 370278400
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008967703109327984,
      "loss": 1.6428,
      "theoretical_loss": 4.0754497177516456,
      "tokens_seen": 370278400
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008967502507522568,
      "loss": 1.7115,
      "theoretical_loss": 4.075368547451882,
      "tokens_seen": 370343936
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008967301905717151,
      "loss": 1.6581,
      "theoretical_loss": 4.075287395535807,
      "tokens_seen": 370409472
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008967101303911736,
      "loss": 1.6987,
      "theoretical_loss": 4.075206261996005,
      "tokens_seen": 370475008
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008966900702106319,
      "loss": 1.8119,
      "theoretical_loss": 4.075125146825064,
      "tokens_seen": 370540544
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008966700100300903,
      "loss": 1.6419,
      "theoretical_loss": 4.0750440500155785,
      "tokens_seen": 370606080
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008966499498495486,
      "loss": 1.7411,
      "theoretical_loss": 4.074962971560145,
      "tokens_seen": 370671616
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000896629889669007,
      "loss": 1.6313,
      "theoretical_loss": 4.074881911451368,
      "tokens_seen": 370737152
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008966098294884654,
      "loss": 1.7241,
      "theoretical_loss": 4.074800869681852,
      "tokens_seen": 370802688
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008965897693079238,
      "loss": 1.7189,
      "theoretical_loss": 4.074719846244209,
      "tokens_seen": 370868224
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008965697091273821,
      "loss": 1.6872,
      "theoretical_loss": 4.074638841131052,
      "tokens_seen": 370933760
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008965496489468406,
      "loss": 1.7023,
      "theoretical_loss": 4.074557854335002,
      "tokens_seen": 370999296
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008965295887662988,
      "loss": 1.7682,
      "theoretical_loss": 4.074476885848682,
      "tokens_seen": 371064832
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008965095285857573,
      "loss": 1.7406,
      "theoretical_loss": 4.07439593566472,
      "tokens_seen": 371130368
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008964894684052156,
      "loss": 1.7485,
      "theoretical_loss": 4.074315003775746,
      "tokens_seen": 371195904
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000896469408224674,
      "loss": 1.7826,
      "theoretical_loss": 4.0742340901744,
      "tokens_seen": 371261440
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008964493480441324,
      "loss": 1.7554,
      "theoretical_loss": 4.07415319485332,
      "tokens_seen": 371326976
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008964292878635908,
      "loss": 1.701,
      "theoretical_loss": 4.074072317805152,
      "tokens_seen": 371392512
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008964092276830491,
      "loss": 1.7843,
      "theoretical_loss": 4.073991459022544,
      "tokens_seen": 371458048
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008963891675025076,
      "loss": 1.6254,
      "theoretical_loss": 4.07391061849815,
      "tokens_seen": 371523584
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008963691073219659,
      "loss": 1.6961,
      "theoretical_loss": 4.073829796224628,
      "tokens_seen": 371589120
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008963490471414243,
      "loss": 1.6637,
      "theoretical_loss": 4.0737489921946395,
      "tokens_seen": 371654656
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008963289869608828,
      "loss": 1.6752,
      "theoretical_loss": 4.073668206400851,
      "tokens_seen": 371720192
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000896308926780341,
      "loss": 1.8764,
      "theoretical_loss": 4.073587438835932,
      "tokens_seen": 371785728
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008962888665997995,
      "loss": 1.8139,
      "theoretical_loss": 4.073506689492557,
      "tokens_seen": 371851264
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008962688064192577,
      "loss": 1.7659,
      "theoretical_loss": 4.073425958363406,
      "tokens_seen": 371916800
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008962487462387162,
      "loss": 1.8909,
      "theoretical_loss": 4.0733452454411605,
      "tokens_seen": 371982336
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008962286860581746,
      "loss": 1.6557,
      "theoretical_loss": 4.073264550718509,
      "tokens_seen": 372047872
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008962086258776329,
      "loss": 1.6719,
      "theoretical_loss": 4.0731838741881425,
      "tokens_seen": 372113408
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008961885656970913,
      "loss": 1.7788,
      "theoretical_loss": 4.073103215842757,
      "tokens_seen": 372178944
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008961685055165498,
      "loss": 1.8669,
      "theoretical_loss": 4.0730225756750515,
      "tokens_seen": 372244480
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000896148445336008,
      "loss": 1.7435,
      "theoretical_loss": 4.072941953677732,
      "tokens_seen": 372310016
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008961283851554665,
      "loss": 1.5605,
      "theoretical_loss": 4.072861349843505,
      "tokens_seen": 372375552
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008961083249749247,
      "loss": 1.7418,
      "theoretical_loss": 4.072780764165083,
      "tokens_seen": 372441088
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008960882647943832,
      "loss": 1.7104,
      "theoretical_loss": 4.072700196635185,
      "tokens_seen": 372506624
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008960682046138416,
      "loss": 1.8206,
      "theoretical_loss": 4.07261964724653,
      "tokens_seen": 372572160
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008960481444332999,
      "loss": 1.7754,
      "theoretical_loss": 4.072539115991844,
      "tokens_seen": 372637696
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008960280842527583,
      "loss": 1.6862,
      "theoretical_loss": 4.072458602863856,
      "tokens_seen": 372703232
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008960080240722167,
      "loss": 1.7193,
      "theoretical_loss": 4.072378107855299,
      "tokens_seen": 372768768
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895987963891675,
      "loss": 1.6753,
      "theoretical_loss": 4.0722976309589125,
      "tokens_seen": 372834304
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008959679037111335,
      "loss": 1.705,
      "theoretical_loss": 4.072217172167436,
      "tokens_seen": 372899840
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008959478435305918,
      "loss": 1.5978,
      "theoretical_loss": 4.072136731473618,
      "tokens_seen": 372965376
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008959277833500502,
      "loss": 1.6881,
      "theoretical_loss": 4.0720563088702075,
      "tokens_seen": 373030912
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008959077231695085,
      "loss": 1.8104,
      "theoretical_loss": 4.0719759043499595,
      "tokens_seen": 373096448
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008958876629889669,
      "loss": 1.6941,
      "theoretical_loss": 4.071895517905633,
      "tokens_seen": 373161984
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008958676028084253,
      "loss": 1.6762,
      "theoretical_loss": 4.071815149529988,
      "tokens_seen": 373227520
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008958475426278837,
      "loss": 1.6635,
      "theoretical_loss": 4.071734799215796,
      "tokens_seen": 373293056
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895827482447342,
      "loss": 1.6936,
      "theoretical_loss": 4.071654466955825,
      "tokens_seen": 373358592
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008958074222668005,
      "loss": 1.698,
      "theoretical_loss": 4.071574152742851,
      "tokens_seen": 373424128
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008957873620862588,
      "loss": 1.5491,
      "theoretical_loss": 4.071493856569655,
      "tokens_seen": 373489664
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.4907689690589905,
      "objective/train/docs_used": 142333,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7129801511764526,
      "objective/train/original_loss": 1.7129802703857422,
      "objective/train/theoretical_loss": 4.071413578429017,
      "objective/train/tokens_used": 394015200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24147610366344452,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503044128417969,
      "objective/train/weighted_lm_loss": 1.7990201711654663,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.071413578429017,
      "tokens_seen": 373555200
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008957673019057172,
      "loss": 1.7972,
      "theoretical_loss": 4.071413578429017,
      "tokens_seen": 373555200
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008957472417251755,
      "loss": 1.6007,
      "theoretical_loss": 4.071333318313728,
      "tokens_seen": 373620736
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895727181544634,
      "loss": 1.7788,
      "theoretical_loss": 4.07125307621658,
      "tokens_seen": 373686272
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008957071213640923,
      "loss": 1.8137,
      "theoretical_loss": 4.071172852130368,
      "tokens_seen": 373751808
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008956870611835507,
      "loss": 1.7266,
      "theoretical_loss": 4.071092646047892,
      "tokens_seen": 373817344
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895667001003009,
      "loss": 1.7166,
      "theoretical_loss": 4.071012457961956,
      "tokens_seen": 373882880
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008956469408224674,
      "loss": 1.5693,
      "theoretical_loss": 4.07093228786537,
      "tokens_seen": 373948416
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008956268806419258,
      "loss": 1.7416,
      "theoretical_loss": 4.070852135750946,
      "tokens_seen": 374013952
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008956068204613842,
      "loss": 1.7768,
      "theoretical_loss": 4.0707720016115,
      "tokens_seen": 374079488
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008955867602808425,
      "loss": 1.6761,
      "theoretical_loss": 4.070691885439854,
      "tokens_seen": 374145024
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895566700100301,
      "loss": 1.6422,
      "theoretical_loss": 4.070611787228832,
      "tokens_seen": 374210560
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008955466399197592,
      "loss": 1.7156,
      "theoretical_loss": 4.0705317069712645,
      "tokens_seen": 374276096
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008955265797392177,
      "loss": 1.6769,
      "theoretical_loss": 4.0704516446599825,
      "tokens_seen": 374341632
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895506519558676,
      "loss": 1.7857,
      "theoretical_loss": 4.070371600287826,
      "tokens_seen": 374407168
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008954864593781344,
      "loss": 1.647,
      "theoretical_loss": 4.070291573847635,
      "tokens_seen": 374472704
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008954663991975928,
      "loss": 1.7466,
      "theoretical_loss": 4.070211565332255,
      "tokens_seen": 374538240
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008954463390170512,
      "loss": 1.8851,
      "theoretical_loss": 4.070131574734536,
      "tokens_seen": 374603776
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008954262788365095,
      "loss": 1.7502,
      "theoretical_loss": 4.070051602047332,
      "tokens_seen": 374669312
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895406218655968,
      "loss": 1.7069,
      "theoretical_loss": 4.069971647263501,
      "tokens_seen": 374734848
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008953861584754262,
      "loss": 1.743,
      "theoretical_loss": 4.069891710375904,
      "tokens_seen": 374800384
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008953660982948847,
      "loss": 1.7041,
      "theoretical_loss": 4.069811791377409,
      "tokens_seen": 374865920
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895346038114343,
      "loss": 1.6945,
      "theoretical_loss": 4.0697318902608846,
      "tokens_seen": 374931456
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008953259779338014,
      "loss": 1.685,
      "theoretical_loss": 4.069652007019204,
      "tokens_seen": 374996992
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008953059177532598,
      "loss": 1.6655,
      "theoretical_loss": 4.069572141645249,
      "tokens_seen": 375062528
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008952858575727181,
      "loss": 1.743,
      "theoretical_loss": 4.0694922941319,
      "tokens_seen": 375128064
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008952657973921765,
      "loss": 1.743,
      "theoretical_loss": 4.069412464472042,
      "tokens_seen": 375193600
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895245737211635,
      "loss": 1.7983,
      "theoretical_loss": 4.069332652658567,
      "tokens_seen": 375259136
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008952256770310932,
      "loss": 1.6777,
      "theoretical_loss": 4.069252858684371,
      "tokens_seen": 375324672
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008952056168505517,
      "loss": 1.8022,
      "theoretical_loss": 4.069173082542351,
      "tokens_seen": 375390208
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008951855566700101,
      "loss": 1.8357,
      "theoretical_loss": 4.069093324225411,
      "tokens_seen": 375455744
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008951654964894684,
      "loss": 1.7366,
      "theoretical_loss": 4.0690135837264565,
      "tokens_seen": 375521280
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008951454363089268,
      "loss": 1.8672,
      "theoretical_loss": 4.068933861038399,
      "tokens_seen": 375586816
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008951253761283851,
      "loss": 1.7984,
      "theoretical_loss": 4.068854156154154,
      "tokens_seen": 375652352
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008951053159478435,
      "loss": 1.8244,
      "theoretical_loss": 4.068774469066639,
      "tokens_seen": 375717888
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895085255767302,
      "loss": 1.6594,
      "theoretical_loss": 4.068694799768779,
      "tokens_seen": 375783424
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008950651955867602,
      "loss": 1.7258,
      "theoretical_loss": 4.0686151482535,
      "tokens_seen": 375848960
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008950451354062187,
      "loss": 1.7146,
      "theoretical_loss": 4.068535514513734,
      "tokens_seen": 375914496
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.000895025075225677,
      "loss": 1.6562,
      "theoretical_loss": 4.068455898542414,
      "tokens_seen": 375980032
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008950050150451354,
      "loss": 1.8663,
      "theoretical_loss": 4.068376300332482,
      "tokens_seen": 376045568
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008949849548645938,
      "loss": 1.8702,
      "theoretical_loss": 4.06829671987688,
      "tokens_seen": 376111104
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008949648946840522,
      "loss": 1.6686,
      "theoretical_loss": 4.068217157168556,
      "tokens_seen": 376176640
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008949448345035105,
      "loss": 1.7105,
      "theoretical_loss": 4.06813761220046,
      "tokens_seen": 376242176
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008949247743229689,
      "loss": 1.7407,
      "theoretical_loss": 4.068058084965549,
      "tokens_seen": 376307712
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008949047141424272,
      "loss": 1.6749,
      "theoretical_loss": 4.067978575456781,
      "tokens_seen": 376373248
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008948846539618857,
      "loss": 1.7597,
      "theoretical_loss": 4.06789908366712,
      "tokens_seen": 376438784
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008948645937813441,
      "loss": 1.7951,
      "theoretical_loss": 4.067819609589533,
      "tokens_seen": 376504320
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008948445336008024,
      "loss": 1.7017,
      "theoretical_loss": 4.067740153216993,
      "tokens_seen": 376569856
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008948244734202609,
      "loss": 1.6655,
      "theoretical_loss": 4.067660714542473,
      "tokens_seen": 376635392
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008948044132397192,
      "loss": 1.9055,
      "theoretical_loss": 4.067581293558955,
      "tokens_seen": 376700928
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008947843530591776,
      "loss": 1.7544,
      "theoretical_loss": 4.06750189025942,
      "tokens_seen": 376766464
    },
    {
      "epoch": 0.11,
      "objective/train/advantage_avg": 0.49255064129829407,
      "objective/train/docs_used": 143470,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6976510286331177,
      "objective/train/original_loss": 1.6976510286331177,
      "objective/train/theoretical_loss": 4.067422504636857,
      "objective/train/tokens_used": 397292000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24296031892299652,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050490140914917,
      "objective/train/weighted_lm_loss": 1.7832144498825073,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.067422504636857,
      "tokens_seen": 376832000
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008947642928786359,
      "loss": 1.7729,
      "theoretical_loss": 4.067422504636857,
      "tokens_seen": 376832000
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008947442326980944,
      "loss": 1.8275,
      "theoretical_loss": 4.067343136684259,
      "tokens_seen": 376897536
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008947241725175527,
      "loss": 1.6847,
      "theoretical_loss": 4.0672637863946175,
      "tokens_seen": 376963072
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008947041123370111,
      "loss": 1.6855,
      "theoretical_loss": 4.067184453760936,
      "tokens_seen": 377028608
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008946840521564694,
      "loss": 1.7641,
      "theoretical_loss": 4.067105138776214,
      "tokens_seen": 377094144
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008946639919759278,
      "loss": 1.6745,
      "theoretical_loss": 4.067025841433463,
      "tokens_seen": 377159680
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008946439317953862,
      "loss": 1.6787,
      "theoretical_loss": 4.0669465617256915,
      "tokens_seen": 377225216
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008946238716148446,
      "loss": 1.6579,
      "theoretical_loss": 4.066867299645916,
      "tokens_seen": 377290752
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008946038114343029,
      "loss": 1.6609,
      "theoretical_loss": 4.066788055187156,
      "tokens_seen": 377356288
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008945837512537614,
      "loss": 1.6764,
      "theoretical_loss": 4.066708828342435,
      "tokens_seen": 377421824
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008945636910732196,
      "loss": 1.7242,
      "theoretical_loss": 4.06662961910478,
      "tokens_seen": 377487360
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008945436308926781,
      "loss": 1.7348,
      "theoretical_loss": 4.066550427467221,
      "tokens_seen": 377552896
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008945235707121364,
      "loss": 1.5869,
      "theoretical_loss": 4.066471253422796,
      "tokens_seen": 377618432
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008945035105315948,
      "loss": 1.6576,
      "theoretical_loss": 4.066392096964542,
      "tokens_seen": 377683968
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008944834503510532,
      "loss": 1.7433,
      "theoretical_loss": 4.066312958085503,
      "tokens_seen": 377749504
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008944633901705116,
      "loss": 1.5885,
      "theoretical_loss": 4.066233836778727,
      "tokens_seen": 377815040
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008944433299899699,
      "loss": 1.656,
      "theoretical_loss": 4.066154733037264,
      "tokens_seen": 377880576
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008944232698094284,
      "loss": 1.6272,
      "theoretical_loss": 4.066075646854169,
      "tokens_seen": 377946112
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008944032096288866,
      "loss": 1.8592,
      "theoretical_loss": 4.065996578222502,
      "tokens_seen": 378011648
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008943831494483451,
      "loss": 1.7786,
      "theoretical_loss": 4.065917527135325,
      "tokens_seen": 378077184
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008943630892678035,
      "loss": 1.6858,
      "theoretical_loss": 4.065838493585705,
      "tokens_seen": 378142720
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008943430290872618,
      "loss": 1.9012,
      "theoretical_loss": 4.065759477566714,
      "tokens_seen": 378208256
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008943229689067202,
      "loss": 1.604,
      "theoretical_loss": 4.065680479071426,
      "tokens_seen": 378273792
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008943029087261785,
      "loss": 1.7117,
      "theoretical_loss": 4.065601498092919,
      "tokens_seen": 378339328
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008942828485456369,
      "loss": 1.6169,
      "theoretical_loss": 4.065522534624276,
      "tokens_seen": 378404864
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008942627883650954,
      "loss": 1.6878,
      "theoretical_loss": 4.065443588658584,
      "tokens_seen": 378470400
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008942427281845536,
      "loss": 1.6946,
      "theoretical_loss": 4.0653646601889335,
      "tokens_seen": 378535936
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008942226680040121,
      "loss": 1.8113,
      "theoretical_loss": 4.065285749208419,
      "tokens_seen": 378601472
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008942026078234705,
      "loss": 1.7916,
      "theoretical_loss": 4.065206855710139,
      "tokens_seen": 378667008
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008941825476429288,
      "loss": 1.6653,
      "theoretical_loss": 4.065127979687195,
      "tokens_seen": 378732544
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008941624874623872,
      "loss": 1.5783,
      "theoretical_loss": 4.065049121132693,
      "tokens_seen": 378798080
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008941424272818455,
      "loss": 1.6457,
      "theoretical_loss": 4.064970280039745,
      "tokens_seen": 378863616
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008941223671013039,
      "loss": 1.7093,
      "theoretical_loss": 4.0648914564014635,
      "tokens_seen": 378929152
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008941023069207624,
      "loss": 1.6528,
      "theoretical_loss": 4.064812650210966,
      "tokens_seen": 378994688
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008940822467402206,
      "loss": 1.753,
      "theoretical_loss": 4.0647338614613755,
      "tokens_seen": 379060224
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008940621865596791,
      "loss": 1.6148,
      "theoretical_loss": 4.064655090145817,
      "tokens_seen": 379125760
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008940421263791374,
      "loss": 1.6967,
      "theoretical_loss": 4.064576336257421,
      "tokens_seen": 379191296
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008940220661985958,
      "loss": 1.6283,
      "theoretical_loss": 4.064497599789319,
      "tokens_seen": 379256832
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008940020060180542,
      "loss": 1.6937,
      "theoretical_loss": 4.0644188807346495,
      "tokens_seen": 379322368
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008939819458375126,
      "loss": 1.6933,
      "theoretical_loss": 4.064340179086555,
      "tokens_seen": 379387904
    },
    {
      "epoch": 0.11,
      "learning_rate": 0.0008939618856569709,
      "loss": 1.7946,
      "theoretical_loss": 4.0642614948381794,
      "tokens_seen": 379453440
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008939418254764293,
      "loss": 1.8811,
      "theoretical_loss": 4.064182827982672,
      "tokens_seen": 379518976
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008939217652958876,
      "loss": 1.6261,
      "theoretical_loss": 4.064104178513186,
      "tokens_seen": 379584512
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008939017051153461,
      "loss": 1.624,
      "theoretical_loss": 4.064025546422878,
      "tokens_seen": 379650048
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008938816449348044,
      "loss": 1.717,
      "theoretical_loss": 4.063946931704908,
      "tokens_seen": 379715584
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008938615847542628,
      "loss": 1.8072,
      "theoretical_loss": 4.063868334352442,
      "tokens_seen": 379781120
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008938415245737212,
      "loss": 1.7027,
      "theoretical_loss": 4.0637897543586465,
      "tokens_seen": 379846656
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008938214643931796,
      "loss": 1.712,
      "theoretical_loss": 4.0637111917166955,
      "tokens_seen": 379912192
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008938014042126379,
      "loss": 1.644,
      "theoretical_loss": 4.063632646419765,
      "tokens_seen": 379977728
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008937813440320963,
      "loss": 1.7294,
      "theoretical_loss": 4.063554118461034,
      "tokens_seen": 380043264
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.4942775368690491,
      "objective/train/docs_used": 144698,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5595202445983887,
      "objective/train/original_loss": 1.5595201253890991,
      "objective/train/theoretical_loss": 4.063475607833687,
      "objective/train/tokens_used": 400568800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2447211593389511,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050671935081482,
      "objective/train/weighted_lm_loss": 1.6384665966033936,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.063475607833687,
      "tokens_seen": 380108800
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008937612838515546,
      "loss": 1.6146,
      "theoretical_loss": 4.063475607833687,
      "tokens_seen": 380108800
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008937412236710131,
      "loss": 1.7456,
      "theoretical_loss": 4.063397114530912,
      "tokens_seen": 380174336
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008937211634904714,
      "loss": 1.69,
      "theoretical_loss": 4.0633186385458995,
      "tokens_seen": 380239872
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008937011033099298,
      "loss": 1.687,
      "theoretical_loss": 4.063240179871846,
      "tokens_seen": 380305408
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008936810431293881,
      "loss": 1.7677,
      "theoretical_loss": 4.063161738501951,
      "tokens_seen": 380370944
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008936609829488466,
      "loss": 1.7485,
      "theoretical_loss": 4.0630833144294165,
      "tokens_seen": 380436480
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008936409227683049,
      "loss": 1.6952,
      "theoretical_loss": 4.06300490764745,
      "tokens_seen": 380502016
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008936208625877633,
      "loss": 1.7707,
      "theoretical_loss": 4.062926518149263,
      "tokens_seen": 380567552
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008936008024072217,
      "loss": 1.6512,
      "theoretical_loss": 4.0628481459280685,
      "tokens_seen": 380633088
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.00089358074222668,
      "loss": 1.5992,
      "theoretical_loss": 4.062769790977086,
      "tokens_seen": 380698624
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008935606820461384,
      "loss": 1.6339,
      "theoretical_loss": 4.062691453289538,
      "tokens_seen": 380764160
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008935406218655968,
      "loss": 1.7438,
      "theoretical_loss": 4.062613132858651,
      "tokens_seen": 380829696
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008935205616850551,
      "loss": 1.7412,
      "theoretical_loss": 4.062534829677653,
      "tokens_seen": 380895232
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008935005015045136,
      "loss": 1.6496,
      "theoretical_loss": 4.0624565437397795,
      "tokens_seen": 380960768
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008934804413239719,
      "loss": 1.8742,
      "theoretical_loss": 4.0623782750382675,
      "tokens_seen": 381026304
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008934603811434303,
      "loss": 1.761,
      "theoretical_loss": 4.062300023566358,
      "tokens_seen": 381091840
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008934403209628887,
      "loss": 1.5984,
      "theoretical_loss": 4.062221789317297,
      "tokens_seen": 381157376
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000893420260782347,
      "loss": 1.68,
      "theoretical_loss": 4.0621435722843335,
      "tokens_seen": 381222912
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008934002006018054,
      "loss": 1.6351,
      "theoretical_loss": 4.06206537246072,
      "tokens_seen": 381288448
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008933801404212639,
      "loss": 1.6759,
      "theoretical_loss": 4.061987189839712,
      "tokens_seen": 381353984
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008933600802407222,
      "loss": 1.7164,
      "theoretical_loss": 4.061909024414572,
      "tokens_seen": 381419520
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008933400200601806,
      "loss": 1.8483,
      "theoretical_loss": 4.061830876178562,
      "tokens_seen": 381485056
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008933199598796389,
      "loss": 1.6855,
      "theoretical_loss": 4.061752745124952,
      "tokens_seen": 381550592
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008932998996990973,
      "loss": 1.5888,
      "theoretical_loss": 4.061674631247014,
      "tokens_seen": 381616128
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008932798395185558,
      "loss": 1.6356,
      "theoretical_loss": 4.061596534538021,
      "tokens_seen": 381681664
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000893259779338014,
      "loss": 1.7905,
      "theoretical_loss": 4.061518454991255,
      "tokens_seen": 381747200
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008932397191574725,
      "loss": 1.7029,
      "theoretical_loss": 4.061440392599997,
      "tokens_seen": 381812736
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008932196589769309,
      "loss": 1.7284,
      "theoretical_loss": 4.061362347357536,
      "tokens_seen": 381878272
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008931995987963892,
      "loss": 1.8156,
      "theoretical_loss": 4.061284319257162,
      "tokens_seen": 381943808
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008931795386158476,
      "loss": 1.8333,
      "theoretical_loss": 4.061206308292169,
      "tokens_seen": 382009344
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008931594784353059,
      "loss": 1.7346,
      "theoretical_loss": 4.061128314455855,
      "tokens_seen": 382074880
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008931394182547643,
      "loss": 1.6804,
      "theoretical_loss": 4.0610503377415235,
      "tokens_seen": 382140416
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008931193580742228,
      "loss": 1.7114,
      "theoretical_loss": 4.060972378142479,
      "tokens_seen": 382205952
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000893099297893681,
      "loss": 1.6782,
      "theoretical_loss": 4.060894435652031,
      "tokens_seen": 382271488
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008930792377131395,
      "loss": 1.8034,
      "theoretical_loss": 4.060816510263494,
      "tokens_seen": 382337024
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008930591775325978,
      "loss": 1.79,
      "theoretical_loss": 4.0607386019701845,
      "tokens_seen": 382402560
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008930391173520562,
      "loss": 1.6879,
      "theoretical_loss": 4.060660710765423,
      "tokens_seen": 382468096
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008930190571715146,
      "loss": 1.8295,
      "theoretical_loss": 4.060582836642534,
      "tokens_seen": 382533632
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892998996990973,
      "loss": 1.6777,
      "theoretical_loss": 4.060504979594847,
      "tokens_seen": 382599168
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008929789368104313,
      "loss": 1.7917,
      "theoretical_loss": 4.060427139615693,
      "tokens_seen": 382664704
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008929588766298897,
      "loss": 1.7472,
      "theoretical_loss": 4.060349316698408,
      "tokens_seen": 382730240
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892938816449348,
      "loss": 1.613,
      "theoretical_loss": 4.060271510836332,
      "tokens_seen": 382795776
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008929187562688065,
      "loss": 1.7459,
      "theoretical_loss": 4.060193722022808,
      "tokens_seen": 382861312
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008928986960882648,
      "loss": 1.6282,
      "theoretical_loss": 4.060115950251183,
      "tokens_seen": 382926848
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008928786359077232,
      "loss": 1.7833,
      "theoretical_loss": 4.060038195514808,
      "tokens_seen": 382992384
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008928585757271816,
      "loss": 1.6383,
      "theoretical_loss": 4.059960457807038,
      "tokens_seen": 383057920
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.00089283851554664,
      "loss": 1.6525,
      "theoretical_loss": 4.05988273712123,
      "tokens_seen": 383123456
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008928184553660983,
      "loss": 1.7811,
      "theoretical_loss": 4.059805033450747,
      "tokens_seen": 383188992
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008927983951855567,
      "loss": 1.6016,
      "theoretical_loss": 4.059727346788955,
      "tokens_seen": 383254528
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892778335005015,
      "loss": 1.7204,
      "theoretical_loss": 4.059649677129222,
      "tokens_seen": 383320064
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.49355781078338623,
      "objective/train/docs_used": 145827,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6360312700271606,
      "objective/train/original_loss": 1.6360312700271606,
      "objective/train/theoretical_loss": 4.059572024464923,
      "objective/train/tokens_used": 403845600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24408653378486633,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505965948104858,
      "objective/train/weighted_lm_loss": 1.7188907861709595,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.059572024464923,
      "tokens_seen": 383385600
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008927582748244735,
      "loss": 1.6921,
      "theoretical_loss": 4.059572024464923,
      "tokens_seen": 383385600
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008927382146439318,
      "loss": 1.6756,
      "theoretical_loss": 4.0594943887894335,
      "tokens_seen": 383451136
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008927181544633902,
      "loss": 1.657,
      "theoretical_loss": 4.059416770096134,
      "tokens_seen": 383516672
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008926980942828485,
      "loss": 1.7426,
      "theoretical_loss": 4.0593391683784095,
      "tokens_seen": 383582208
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892678034102307,
      "loss": 1.6976,
      "theoretical_loss": 4.059261583629648,
      "tokens_seen": 383647744
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008926579739217653,
      "loss": 1.5842,
      "theoretical_loss": 4.059184015843241,
      "tokens_seen": 383713280
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008926379137412237,
      "loss": 1.7801,
      "theoretical_loss": 4.059106465012583,
      "tokens_seen": 383778816
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892617853560682,
      "loss": 1.7129,
      "theoretical_loss": 4.059028931131075,
      "tokens_seen": 383844352
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008925977933801404,
      "loss": 1.6377,
      "theoretical_loss": 4.058951414192118,
      "tokens_seen": 383909888
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008925777331995988,
      "loss": 1.7708,
      "theoretical_loss": 4.058873914189119,
      "tokens_seen": 383975424
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008925576730190572,
      "loss": 1.6659,
      "theoretical_loss": 4.058796431115489,
      "tokens_seen": 384040960
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008925376128385155,
      "loss": 1.7973,
      "theoretical_loss": 4.058718964964641,
      "tokens_seen": 384106496
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892517552657974,
      "loss": 1.699,
      "theoretical_loss": 4.0586415157299935,
      "tokens_seen": 384172032
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008924974924774323,
      "loss": 1.6343,
      "theoretical_loss": 4.058564083404967,
      "tokens_seen": 384237568
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008924774322968907,
      "loss": 1.7538,
      "theoretical_loss": 4.058486667982986,
      "tokens_seen": 384303104
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008924573721163491,
      "loss": 1.6043,
      "theoretical_loss": 4.05840926945748,
      "tokens_seen": 384368640
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008924373119358074,
      "loss": 1.7165,
      "theoretical_loss": 4.058331887821879,
      "tokens_seen": 384434176
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008924172517552658,
      "loss": 1.6977,
      "theoretical_loss": 4.058254523069623,
      "tokens_seen": 384499712
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008923971915747243,
      "loss": 1.7091,
      "theoretical_loss": 4.058177175194148,
      "tokens_seen": 384565248
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008923771313941825,
      "loss": 1.6789,
      "theoretical_loss": 4.058099844188899,
      "tokens_seen": 384630784
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892357071213641,
      "loss": 1.7138,
      "theoretical_loss": 4.058022530047323,
      "tokens_seen": 384696320
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008923370110330992,
      "loss": 1.6221,
      "theoretical_loss": 4.05794523276287,
      "tokens_seen": 384761856
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008923169508525577,
      "loss": 1.8069,
      "theoretical_loss": 4.057867952328994,
      "tokens_seen": 384827392
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008922968906720161,
      "loss": 1.7147,
      "theoretical_loss": 4.0577906887391535,
      "tokens_seen": 384892928
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008922768304914744,
      "loss": 1.83,
      "theoretical_loss": 4.05771344198681,
      "tokens_seen": 384958464
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008922567703109328,
      "loss": 1.6993,
      "theoretical_loss": 4.057636212065429,
      "tokens_seen": 385024000
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008922367101303913,
      "loss": 1.6941,
      "theoretical_loss": 4.057558998968479,
      "tokens_seen": 385089536
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008922166499498495,
      "loss": 1.7719,
      "theoretical_loss": 4.0574818026894315,
      "tokens_seen": 385155072
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892196589769308,
      "loss": 1.757,
      "theoretical_loss": 4.057404623221765,
      "tokens_seen": 385220608
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008921765295887662,
      "loss": 1.723,
      "theoretical_loss": 4.057327460558957,
      "tokens_seen": 385286144
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008921564694082247,
      "loss": 1.7299,
      "theoretical_loss": 4.0572503146944925,
      "tokens_seen": 385351680
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008921364092276831,
      "loss": 1.7218,
      "theoretical_loss": 4.057173185621857,
      "tokens_seen": 385417216
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008921163490471414,
      "loss": 1.706,
      "theoretical_loss": 4.057096073334543,
      "tokens_seen": 385482752
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008920962888665998,
      "loss": 1.7013,
      "theoretical_loss": 4.057018977826044,
      "tokens_seen": 385548288
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008920762286860582,
      "loss": 1.7012,
      "theoretical_loss": 4.056941899089858,
      "tokens_seen": 385613824
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008920561685055165,
      "loss": 1.646,
      "theoretical_loss": 4.056864837119486,
      "tokens_seen": 385679360
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000892036108324975,
      "loss": 1.7769,
      "theoretical_loss": 4.056787791908434,
      "tokens_seen": 385744896
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008920160481444332,
      "loss": 1.6899,
      "theoretical_loss": 4.05671076345021,
      "tokens_seen": 385810432
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008919959879638917,
      "loss": 1.6936,
      "theoretical_loss": 4.056633751738328,
      "tokens_seen": 385875968
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.00089197592778335,
      "loss": 1.8251,
      "theoretical_loss": 4.056556756766303,
      "tokens_seen": 385941504
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008919558676028084,
      "loss": 1.6149,
      "theoretical_loss": 4.056479778527653,
      "tokens_seen": 386007040
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008919358074222668,
      "loss": 1.669,
      "theoretical_loss": 4.056402817015905,
      "tokens_seen": 386072576
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008919157472417252,
      "loss": 1.6661,
      "theoretical_loss": 4.0563258722245825,
      "tokens_seen": 386138112
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008918956870611835,
      "loss": 1.6326,
      "theoretical_loss": 4.056248944147217,
      "tokens_seen": 386203648
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000891875626880642,
      "loss": 1.7272,
      "theoretical_loss": 4.0561720327773445,
      "tokens_seen": 386269184
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008918555667001004,
      "loss": 1.7083,
      "theoretical_loss": 4.0560951381085,
      "tokens_seen": 386334720
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008918355065195587,
      "loss": 1.6469,
      "theoretical_loss": 4.056018260134226,
      "tokens_seen": 386400256
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008918154463390171,
      "loss": 1.6153,
      "theoretical_loss": 4.055941398848067,
      "tokens_seen": 386465792
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008917953861584754,
      "loss": 1.6674,
      "theoretical_loss": 4.055864554243572,
      "tokens_seen": 386531328
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008917753259779339,
      "loss": 1.7016,
      "theoretical_loss": 4.055787726314292,
      "tokens_seen": 386596864
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.49315446615219116,
      "objective/train/docs_used": 146981,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7423198223114014,
      "objective/train/original_loss": 1.7423198223114014,
      "objective/train/theoretical_loss": 4.055710915053783,
      "objective/train/tokens_used": 407122400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24348418414592743,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505532026290894,
      "objective/train/weighted_lm_loss": 1.8302770853042603,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 4.055710915053783,
      "tokens_seen": 386662400
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008917552657973922,
      "loss": 1.8378,
      "theoretical_loss": 4.055710915053783,
      "tokens_seen": 386662400
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008917352056168506,
      "loss": 1.745,
      "theoretical_loss": 4.055634120455606,
      "tokens_seen": 386727936
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008917151454363089,
      "loss": 1.6814,
      "theoretical_loss": 4.055557342513321,
      "tokens_seen": 386793472
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008916950852557674,
      "loss": 1.6931,
      "theoretical_loss": 4.055480581220497,
      "tokens_seen": 386859008
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008916750250752257,
      "loss": 1.565,
      "theoretical_loss": 4.055403836570701,
      "tokens_seen": 386924544
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008916549648946841,
      "loss": 1.6658,
      "theoretical_loss": 4.055327108557509,
      "tokens_seen": 386990080
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008916349047141425,
      "loss": 1.7828,
      "theoretical_loss": 4.055250397174497,
      "tokens_seen": 387055616
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008916148445336008,
      "loss": 1.7989,
      "theoretical_loss": 4.055173702415246,
      "tokens_seen": 387121152
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008915947843530592,
      "loss": 1.8036,
      "theoretical_loss": 4.05509702427334,
      "tokens_seen": 387186688
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008915747241725176,
      "loss": 1.6286,
      "theoretical_loss": 4.055020362742367,
      "tokens_seen": 387252224
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008915546639919759,
      "loss": 1.6351,
      "theoretical_loss": 4.054943717815917,
      "tokens_seen": 387317760
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008915346038114344,
      "loss": 1.7932,
      "theoretical_loss": 4.054867089487587,
      "tokens_seen": 387383296
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008915145436308927,
      "loss": 1.6156,
      "theoretical_loss": 4.054790477750974,
      "tokens_seen": 387448832
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008914944834503511,
      "loss": 1.6245,
      "theoretical_loss": 4.054713882599681,
      "tokens_seen": 387514368
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008914744232698095,
      "loss": 1.7205,
      "theoretical_loss": 4.054637304027312,
      "tokens_seen": 387579904
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008914543630892678,
      "loss": 1.706,
      "theoretical_loss": 4.054560742027478,
      "tokens_seen": 387645440
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008914343029087262,
      "loss": 1.7275,
      "theoretical_loss": 4.054484196593791,
      "tokens_seen": 387710976
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008914142427281847,
      "loss": 1.8033,
      "theoretical_loss": 4.054407667719866,
      "tokens_seen": 387776512
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008913941825476429,
      "loss": 1.7001,
      "theoretical_loss": 4.054331155399325,
      "tokens_seen": 387842048
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008913741223671014,
      "loss": 1.664,
      "theoretical_loss": 4.054254659625788,
      "tokens_seen": 387907584
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008913540621865596,
      "loss": 1.6946,
      "theoretical_loss": 4.054178180392885,
      "tokens_seen": 387973120
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008913340020060181,
      "loss": 1.654,
      "theoretical_loss": 4.054101717694244,
      "tokens_seen": 388038656
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008913139418254765,
      "loss": 1.5095,
      "theoretical_loss": 4.0540252715235,
      "tokens_seen": 388104192
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008912938816449348,
      "loss": 1.5775,
      "theoretical_loss": 4.05394884187429,
      "tokens_seen": 388169728
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008912738214643932,
      "loss": 1.6788,
      "theoretical_loss": 4.053872428740256,
      "tokens_seen": 388235264
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008912537612838517,
      "loss": 1.8278,
      "theoretical_loss": 4.053796032115041,
      "tokens_seen": 388300800
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008912337011033099,
      "loss": 1.7206,
      "theoretical_loss": 4.053719651992293,
      "tokens_seen": 388366336
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008912136409227684,
      "loss": 1.7555,
      "theoretical_loss": 4.053643288365664,
      "tokens_seen": 388431872
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008911935807422266,
      "loss": 1.5701,
      "theoretical_loss": 4.053566941228809,
      "tokens_seen": 388497408
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008911735205616851,
      "loss": 1.6847,
      "theoretical_loss": 4.053490610575386,
      "tokens_seen": 388562944
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008911534603811435,
      "loss": 1.7782,
      "theoretical_loss": 4.0534142963990565,
      "tokens_seen": 388628480
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008911334002006018,
      "loss": 1.7133,
      "theoretical_loss": 4.053337998693488,
      "tokens_seen": 388694016
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008911133400200602,
      "loss": 1.7067,
      "theoretical_loss": 4.053261717452346,
      "tokens_seen": 388759552
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008910932798395186,
      "loss": 1.6116,
      "theoretical_loss": 4.053185452669307,
      "tokens_seen": 388825088
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008910732196589769,
      "loss": 1.7676,
      "theoretical_loss": 4.053109204338045,
      "tokens_seen": 388890624
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008910531594784354,
      "loss": 1.579,
      "theoretical_loss": 4.05303297245224,
      "tokens_seen": 388956160
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008910330992978936,
      "loss": 1.7036,
      "theoretical_loss": 4.052956757005573,
      "tokens_seen": 389021696
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008910130391173521,
      "loss": 1.7259,
      "theoretical_loss": 4.052880557991733,
      "tokens_seen": 389087232
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008909929789368104,
      "loss": 1.5456,
      "theoretical_loss": 4.052804375404408,
      "tokens_seen": 389152768
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008909729187562688,
      "loss": 1.7429,
      "theoretical_loss": 4.052728209237293,
      "tokens_seen": 389218304
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008909528585757272,
      "loss": 1.6288,
      "theoretical_loss": 4.0526520594840845,
      "tokens_seen": 389283840
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008909327983951856,
      "loss": 1.743,
      "theoretical_loss": 4.052575926138482,
      "tokens_seen": 389349376
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008909127382146439,
      "loss": 1.7862,
      "theoretical_loss": 4.052499809194191,
      "tokens_seen": 389414912
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008908926780341024,
      "loss": 1.7565,
      "theoretical_loss": 4.052423708644917,
      "tokens_seen": 389480448
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008908726178535606,
      "loss": 1.7019,
      "theoretical_loss": 4.052347624484373,
      "tokens_seen": 389545984
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008908525576730191,
      "loss": 1.5311,
      "theoretical_loss": 4.052271556706271,
      "tokens_seen": 389611520
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008908324974924774,
      "loss": 1.6859,
      "theoretical_loss": 4.052195505304329,
      "tokens_seen": 389677056
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008908124373119358,
      "loss": 1.7945,
      "theoretical_loss": 4.05211947027227,
      "tokens_seen": 389742592
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008907923771313942,
      "loss": 1.7262,
      "theoretical_loss": 4.052043451603818,
      "tokens_seen": 389808128
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008907723169508526,
      "loss": 1.6705,
      "theoretical_loss": 4.051967449292699,
      "tokens_seen": 389873664
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.4954656660556793,
      "objective/train/docs_used": 148229,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7292380332946777,
      "objective/train/original_loss": 1.7292380332946777,
      "objective/train/theoretical_loss": 4.051891463332648,
      "objective/train/tokens_used": 410399200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2457554042339325,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507960319519043,
      "objective/train/weighted_lm_loss": 1.817002534866333,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.051891463332648,
      "tokens_seen": 389939200
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008907522567703109,
      "loss": 1.7942,
      "theoretical_loss": 4.051891463332648,
      "tokens_seen": 389939200
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008907321965897693,
      "loss": 1.7186,
      "theoretical_loss": 4.051815493717399,
      "tokens_seen": 390004736
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008907121364092277,
      "loss": 1.7202,
      "theoretical_loss": 4.051739540440688,
      "tokens_seen": 390070272
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008906920762286861,
      "loss": 1.7049,
      "theoretical_loss": 4.0516636034962605,
      "tokens_seen": 390135808
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008906720160481444,
      "loss": 1.6774,
      "theoretical_loss": 4.051587682877859,
      "tokens_seen": 390201344
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008906519558676029,
      "loss": 1.6931,
      "theoretical_loss": 4.0515117785792345,
      "tokens_seen": 390266880
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008906318956870611,
      "loss": 1.5631,
      "theoretical_loss": 4.0514358905941386,
      "tokens_seen": 390332416
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008906118355065196,
      "loss": 1.6724,
      "theoretical_loss": 4.051360018916327,
      "tokens_seen": 390397952
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008905917753259779,
      "loss": 1.7544,
      "theoretical_loss": 4.051284163539558,
      "tokens_seen": 390463488
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008905717151454363,
      "loss": 1.7298,
      "theoretical_loss": 4.051208324457595,
      "tokens_seen": 390529024
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008905516549648947,
      "loss": 1.628,
      "theoretical_loss": 4.051132501664204,
      "tokens_seen": 390594560
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008905315947843531,
      "loss": 1.8071,
      "theoretical_loss": 4.0510566951531555,
      "tokens_seen": 390660096
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008905115346038114,
      "loss": 1.6248,
      "theoretical_loss": 4.05098090491822,
      "tokens_seen": 390725632
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008904914744232699,
      "loss": 1.6874,
      "theoretical_loss": 4.0509051309531765,
      "tokens_seen": 390791168
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008904714142427281,
      "loss": 1.8073,
      "theoretical_loss": 4.050829373251803,
      "tokens_seen": 390856704
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008904513540621866,
      "loss": 1.7927,
      "theoretical_loss": 4.050753631807883,
      "tokens_seen": 390922240
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008904312938816449,
      "loss": 1.7519,
      "theoretical_loss": 4.050677906615203,
      "tokens_seen": 390987776
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008904112337011033,
      "loss": 1.751,
      "theoretical_loss": 4.050602197667553,
      "tokens_seen": 391053312
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008903911735205617,
      "loss": 1.6699,
      "theoretical_loss": 4.050526504958727,
      "tokens_seen": 391118848
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.00089037111334002,
      "loss": 1.6841,
      "theoretical_loss": 4.050450828482521,
      "tokens_seen": 391184384
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008903510531594785,
      "loss": 1.4873,
      "theoretical_loss": 4.050375168232737,
      "tokens_seen": 391249920
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008903309929789369,
      "loss": 1.6055,
      "theoretical_loss": 4.050299524203176,
      "tokens_seen": 391315456
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008903109327983952,
      "loss": 1.772,
      "theoretical_loss": 4.050223896387647,
      "tokens_seen": 391380992
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008902908726178536,
      "loss": 1.7945,
      "theoretical_loss": 4.05014828477996,
      "tokens_seen": 391446528
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008902708124373121,
      "loss": 1.6458,
      "theoretical_loss": 4.050072689373928,
      "tokens_seen": 391512064
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008902507522567703,
      "loss": 1.6568,
      "theoretical_loss": 4.049997110163369,
      "tokens_seen": 391577600
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008902306920762288,
      "loss": 1.6862,
      "theoretical_loss": 4.0499215471421035,
      "tokens_seen": 391643136
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000890210631895687,
      "loss": 1.7335,
      "theoretical_loss": 4.049846000303956,
      "tokens_seen": 391708672
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008901905717151455,
      "loss": 1.6957,
      "theoretical_loss": 4.049770469642753,
      "tokens_seen": 391774208
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008901705115346039,
      "loss": 1.6915,
      "theoretical_loss": 4.049694955152326,
      "tokens_seen": 391839744
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008901504513540622,
      "loss": 1.717,
      "theoretical_loss": 4.049619456826508,
      "tokens_seen": 391905280
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008901303911735206,
      "loss": 1.664,
      "theoretical_loss": 4.049543974659139,
      "tokens_seen": 391970816
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000890110330992979,
      "loss": 1.55,
      "theoretical_loss": 4.049468508644058,
      "tokens_seen": 392036352
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008900902708124373,
      "loss": 1.7676,
      "theoretical_loss": 4.049393058775109,
      "tokens_seen": 392101888
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008900702106318958,
      "loss": 1.7747,
      "theoretical_loss": 4.0493176250461405,
      "tokens_seen": 392167424
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000890050150451354,
      "loss": 1.6398,
      "theoretical_loss": 4.0492422074510035,
      "tokens_seen": 392232960
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008900300902708125,
      "loss": 1.778,
      "theoretical_loss": 4.049166805983553,
      "tokens_seen": 392298496
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008900100300902708,
      "loss": 1.716,
      "theoretical_loss": 4.049091420637646,
      "tokens_seen": 392364032
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008899899699097292,
      "loss": 1.645,
      "theoretical_loss": 4.049016051407145,
      "tokens_seen": 392429568
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008899699097291876,
      "loss": 1.5746,
      "theoretical_loss": 4.048940698285913,
      "tokens_seen": 392495104
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000889949849548646,
      "loss": 1.625,
      "theoretical_loss": 4.048865361267817,
      "tokens_seen": 392560640
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008899297893681043,
      "loss": 1.7186,
      "theoretical_loss": 4.048790040346731,
      "tokens_seen": 392626176
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008899097291875628,
      "loss": 1.7445,
      "theoretical_loss": 4.048714735516527,
      "tokens_seen": 392691712
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000889889669007021,
      "loss": 1.8012,
      "theoretical_loss": 4.048639446771085,
      "tokens_seen": 392757248
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008898696088264795,
      "loss": 1.7188,
      "theoretical_loss": 4.048564174104285,
      "tokens_seen": 392822784
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008898495486459378,
      "loss": 1.7577,
      "theoretical_loss": 4.048488917510012,
      "tokens_seen": 392888320
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008898294884653962,
      "loss": 1.7189,
      "theoretical_loss": 4.048413676982155,
      "tokens_seen": 392953856
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008898094282848546,
      "loss": 1.6715,
      "theoretical_loss": 4.048338452514604,
      "tokens_seen": 393019392
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000889789368104313,
      "loss": 1.7221,
      "theoretical_loss": 4.048263244101253,
      "tokens_seen": 393084928
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008897693079237713,
      "loss": 1.6308,
      "theoretical_loss": 4.048188051736003,
      "tokens_seen": 393150464
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.49091511964797974,
      "objective/train/docs_used": 149327,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7502202987670898,
      "objective/train/original_loss": 1.7502202987670898,
      "objective/train/theoretical_loss": 4.048112875412752,
      "objective/train/tokens_used": 413676000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24176764488220215,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503205060958862,
      "objective/train/weighted_lm_loss": 1.838315725326538,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.048112875412752,
      "tokens_seen": 393216000
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008897492477432297,
      "loss": 1.7159,
      "theoretical_loss": 4.048112875412752,
      "tokens_seen": 393216000
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008897291875626881,
      "loss": 1.7339,
      "theoretical_loss": 4.048037715125407,
      "tokens_seen": 393281536
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008897091273821465,
      "loss": 1.6511,
      "theoretical_loss": 4.047962570867876,
      "tokens_seen": 393347072
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008896890672016048,
      "loss": 1.7161,
      "theoretical_loss": 4.047887442634067,
      "tokens_seen": 393412608
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008896690070210633,
      "loss": 1.7808,
      "theoretical_loss": 4.0478123304179,
      "tokens_seen": 393478144
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008896489468405215,
      "loss": 1.7958,
      "theoretical_loss": 4.0477372342132885,
      "tokens_seen": 393543680
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.00088962888665998,
      "loss": 1.7283,
      "theoretical_loss": 4.047662154014157,
      "tokens_seen": 393609216
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008896088264794383,
      "loss": 1.7392,
      "theoretical_loss": 4.047587089814427,
      "tokens_seen": 393674752
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008895887662988967,
      "loss": 1.7529,
      "theoretical_loss": 4.047512041608029,
      "tokens_seen": 393740288
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008895687061183551,
      "loss": 1.7174,
      "theoretical_loss": 4.047437009388894,
      "tokens_seen": 393805824
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008895486459378135,
      "loss": 1.7715,
      "theoretical_loss": 4.047361993150955,
      "tokens_seen": 393871360
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008895285857572718,
      "loss": 1.6531,
      "theoretical_loss": 4.047286992888151,
      "tokens_seen": 393936896
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008895085255767303,
      "loss": 1.7352,
      "theoretical_loss": 4.047212008594424,
      "tokens_seen": 394002432
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008894884653961885,
      "loss": 1.5426,
      "theoretical_loss": 4.047137040263717,
      "tokens_seen": 394067968
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000889468405215647,
      "loss": 1.7159,
      "theoretical_loss": 4.047062087889978,
      "tokens_seen": 394133504
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008894483450351053,
      "loss": 1.6867,
      "theoretical_loss": 4.046987151467159,
      "tokens_seen": 394199040
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008894282848545637,
      "loss": 1.7914,
      "theoretical_loss": 4.046912230989214,
      "tokens_seen": 394264576
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008894082246740221,
      "loss": 1.7054,
      "theoretical_loss": 4.046837326450102,
      "tokens_seen": 394330112
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008893881644934804,
      "loss": 1.7448,
      "theoretical_loss": 4.046762437843781,
      "tokens_seen": 394395648
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008893681043129388,
      "loss": 1.7828,
      "theoretical_loss": 4.046687565164217,
      "tokens_seen": 394461184
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008893480441323973,
      "loss": 1.6893,
      "theoretical_loss": 4.0466127084053785,
      "tokens_seen": 394526720
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008893279839518555,
      "loss": 1.7992,
      "theoretical_loss": 4.046537867561234,
      "tokens_seen": 394592256
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000889307923771314,
      "loss": 1.8085,
      "theoretical_loss": 4.04646304262576,
      "tokens_seen": 394657792
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008892878635907722,
      "loss": 1.6054,
      "theoretical_loss": 4.046388233592933,
      "tokens_seen": 394723328
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008892678034102307,
      "loss": 1.6152,
      "theoretical_loss": 4.046313440456733,
      "tokens_seen": 394788864
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008892477432296891,
      "loss": 1.6747,
      "theoretical_loss": 4.046238663211145,
      "tokens_seen": 394854400
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008892276830491474,
      "loss": 1.7601,
      "theoretical_loss": 4.046163901850156,
      "tokens_seen": 394919936
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008892076228686058,
      "loss": 1.7451,
      "theoretical_loss": 4.046089156367756,
      "tokens_seen": 394985472
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008891875626880643,
      "loss": 1.7268,
      "theoretical_loss": 4.0460144267579405,
      "tokens_seen": 395051008
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008891675025075225,
      "loss": 1.7339,
      "theoretical_loss": 4.045939713014705,
      "tokens_seen": 395116544
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000889147442326981,
      "loss": 1.7601,
      "theoretical_loss": 4.045865015132049,
      "tokens_seen": 395182080
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008891273821464392,
      "loss": 1.7139,
      "theoretical_loss": 4.045790333103978,
      "tokens_seen": 395247616
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008891073219658977,
      "loss": 1.7569,
      "theoretical_loss": 4.045715666924499,
      "tokens_seen": 395313152
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008890872617853561,
      "loss": 1.6522,
      "theoretical_loss": 4.04564101658762,
      "tokens_seen": 395378688
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008890672016048144,
      "loss": 1.7247,
      "theoretical_loss": 4.045566382087356,
      "tokens_seen": 395444224
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008890471414242728,
      "loss": 1.6179,
      "theoretical_loss": 4.045491763417724,
      "tokens_seen": 395509760
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008890270812437312,
      "loss": 1.8097,
      "theoretical_loss": 4.045417160572743,
      "tokens_seen": 395575296
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008890070210631895,
      "loss": 1.6343,
      "theoretical_loss": 4.045342573546435,
      "tokens_seen": 395640832
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000888986960882648,
      "loss": 1.7958,
      "theoretical_loss": 4.045268002332829,
      "tokens_seen": 395706368
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008889669007021063,
      "loss": 1.755,
      "theoretical_loss": 4.0451934469259525,
      "tokens_seen": 395771904
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008889468405215647,
      "loss": 1.787,
      "theoretical_loss": 4.045118907319839,
      "tokens_seen": 395837440
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000888926780341023,
      "loss": 1.6481,
      "theoretical_loss": 4.045044383508526,
      "tokens_seen": 395902976
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008889067201604815,
      "loss": 1.7161,
      "theoretical_loss": 4.044969875486051,
      "tokens_seen": 395968512
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008888866599799399,
      "loss": 1.6305,
      "theoretical_loss": 4.044895383246457,
      "tokens_seen": 396034048
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008888665997993982,
      "loss": 1.706,
      "theoretical_loss": 4.04482090678379,
      "tokens_seen": 396099584
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008888465396188566,
      "loss": 1.6584,
      "theoretical_loss": 4.044746446092098,
      "tokens_seen": 396165120
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000888826479438315,
      "loss": 1.7699,
      "theoretical_loss": 4.044672001165436,
      "tokens_seen": 396230656
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008888064192577734,
      "loss": 1.8087,
      "theoretical_loss": 4.0445975719978575,
      "tokens_seen": 396296192
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008887863590772317,
      "loss": 1.8132,
      "theoretical_loss": 4.044523158583421,
      "tokens_seen": 396361728
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008887662988966901,
      "loss": 1.8177,
      "theoretical_loss": 4.044448760916189,
      "tokens_seen": 396427264
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.4924962520599365,
      "objective/train/docs_used": 150450,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.605733036994934,
      "objective/train/original_loss": 1.6057329177856445,
      "objective/train/theoretical_loss": 4.044374378990227,
      "objective/train/tokens_used": 416952800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24324040114879608,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504863262176514,
      "objective/train/weighted_lm_loss": 1.6867058277130127,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.044374378990227,
      "tokens_seen": 396492800
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008887462387161485,
      "loss": 1.6627,
      "theoretical_loss": 4.044374378990227,
      "tokens_seen": 396492800
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008887261785356069,
      "loss": 1.7032,
      "theoretical_loss": 4.044300012799603,
      "tokens_seen": 396558336
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008887061183550652,
      "loss": 1.7416,
      "theoretical_loss": 4.044225662338388,
      "tokens_seen": 396623872
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008886860581745237,
      "loss": 1.8403,
      "theoretical_loss": 4.044151327600657,
      "tokens_seen": 396689408
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008886659979939819,
      "loss": 1.6319,
      "theoretical_loss": 4.04407700858049,
      "tokens_seen": 396754944
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008886459378134404,
      "loss": 1.759,
      "theoretical_loss": 4.044002705271966,
      "tokens_seen": 396820480
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008886258776328987,
      "loss": 1.6112,
      "theoretical_loss": 4.04392841766917,
      "tokens_seen": 396886016
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008886058174523571,
      "loss": 1.5804,
      "theoretical_loss": 4.04385414576619,
      "tokens_seen": 396951552
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008885857572718155,
      "loss": 1.5721,
      "theoretical_loss": 4.043779889557116,
      "tokens_seen": 397017088
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008885656970912739,
      "loss": 1.7787,
      "theoretical_loss": 4.043705649036043,
      "tokens_seen": 397082624
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008885456369107322,
      "loss": 1.7641,
      "theoretical_loss": 4.043631424197067,
      "tokens_seen": 397148160
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008885255767301907,
      "loss": 1.6464,
      "theoretical_loss": 4.043557215034291,
      "tokens_seen": 397213696
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008885055165496489,
      "loss": 1.6706,
      "theoretical_loss": 4.043483021541815,
      "tokens_seen": 397279232
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008884854563691074,
      "loss": 1.8047,
      "theoretical_loss": 4.043408843713749,
      "tokens_seen": 397344768
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008884653961885657,
      "loss": 1.8259,
      "theoretical_loss": 4.0433346815442,
      "tokens_seen": 397410304
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008884453360080241,
      "loss": 1.778,
      "theoretical_loss": 4.043260535027285,
      "tokens_seen": 397475840
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008884252758274825,
      "loss": 1.7848,
      "theoretical_loss": 4.043186404157117,
      "tokens_seen": 397541376
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008884052156469408,
      "loss": 1.7881,
      "theoretical_loss": 4.0431122889278175,
      "tokens_seen": 397606912
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008883851554663992,
      "loss": 1.7349,
      "theoretical_loss": 4.043038189333508,
      "tokens_seen": 397672448
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008883650952858577,
      "loss": 1.8538,
      "theoretical_loss": 4.0429641053683145,
      "tokens_seen": 397737984
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008883450351053159,
      "loss": 1.7809,
      "theoretical_loss": 4.042890037026368,
      "tokens_seen": 397803520
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008883249749247744,
      "loss": 1.7622,
      "theoretical_loss": 4.042815984301798,
      "tokens_seen": 397869056
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008883049147442326,
      "loss": 1.7507,
      "theoretical_loss": 4.042741947188741,
      "tokens_seen": 397934592
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008882848545636911,
      "loss": 1.7028,
      "theoretical_loss": 4.042667925681337,
      "tokens_seen": 398000128
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008882647943831495,
      "loss": 1.7455,
      "theoretical_loss": 4.042593919773726,
      "tokens_seen": 398065664
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008882447342026078,
      "loss": 1.59,
      "theoretical_loss": 4.0425199294600525,
      "tokens_seen": 398131200
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008882246740220662,
      "loss": 1.7077,
      "theoretical_loss": 4.042445954734466,
      "tokens_seen": 398196736
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008882046138415247,
      "loss": 1.8407,
      "theoretical_loss": 4.042371995591118,
      "tokens_seen": 398262272
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008881845536609829,
      "loss": 1.8003,
      "theoretical_loss": 4.042298052024162,
      "tokens_seen": 398327808
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008881644934804414,
      "loss": 1.8477,
      "theoretical_loss": 4.042224124027754,
      "tokens_seen": 398393344
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008881444332998996,
      "loss": 1.7051,
      "theoretical_loss": 4.0421502115960575,
      "tokens_seen": 398458880
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008881243731193581,
      "loss": 1.5807,
      "theoretical_loss": 4.042076314723236,
      "tokens_seen": 398524416
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008881043129388165,
      "loss": 1.6232,
      "theoretical_loss": 4.0420024334034546,
      "tokens_seen": 398589952
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008880842527582748,
      "loss": 1.7288,
      "theoretical_loss": 4.041928567630885,
      "tokens_seen": 398655488
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008880641925777332,
      "loss": 1.6554,
      "theoretical_loss": 4.0418547173997,
      "tokens_seen": 398721024
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008880441323971916,
      "loss": 1.7113,
      "theoretical_loss": 4.041780882704076,
      "tokens_seen": 398786560
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008880240722166499,
      "loss": 1.7606,
      "theoretical_loss": 4.041707063538193,
      "tokens_seen": 398852096
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008880040120361084,
      "loss": 1.7467,
      "theoretical_loss": 4.0416332598962335,
      "tokens_seen": 398917632
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008879839518555667,
      "loss": 1.6222,
      "theoretical_loss": 4.041559471772382,
      "tokens_seen": 398983168
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008879638916750251,
      "loss": 1.77,
      "theoretical_loss": 4.0414856991608294,
      "tokens_seen": 399048704
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008879438314944834,
      "loss": 1.6854,
      "theoretical_loss": 4.041411942055767,
      "tokens_seen": 399114240
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008879237713139419,
      "loss": 1.7659,
      "theoretical_loss": 4.04133820045139,
      "tokens_seen": 399179776
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008879037111334002,
      "loss": 1.8196,
      "theoretical_loss": 4.041264474341896,
      "tokens_seen": 399245312
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008878836509528586,
      "loss": 1.8015,
      "theoretical_loss": 4.041190763721488,
      "tokens_seen": 399310848
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008878635907723169,
      "loss": 1.7079,
      "theoretical_loss": 4.04111706858437,
      "tokens_seen": 399376384
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008878435305917754,
      "loss": 1.7024,
      "theoretical_loss": 4.04104338892475,
      "tokens_seen": 399441920
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008878234704112337,
      "loss": 1.6853,
      "theoretical_loss": 4.040969724736838,
      "tokens_seen": 399507456
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008878034102306921,
      "loss": 1.6133,
      "theoretical_loss": 4.040896076014848,
      "tokens_seen": 399572992
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008877833500501504,
      "loss": 1.7696,
      "theoretical_loss": 4.040822442752997,
      "tokens_seen": 399638528
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008877632898696089,
      "loss": 1.6849,
      "theoretical_loss": 4.040748824945507,
      "tokens_seen": 399704064
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.4920065701007843,
      "objective/train/docs_used": 151634,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6224642992019653,
      "objective/train/original_loss": 1.6224645376205444,
      "objective/train/theoretical_loss": 4.040675222586599,
      "objective/train/tokens_used": 420229600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24278749525547028,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050434947013855,
      "objective/train/weighted_lm_loss": 1.7043081521987915,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.040675222586599,
      "tokens_seen": 399769600
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008877432296890672,
      "loss": 1.6058,
      "theoretical_loss": 4.040675222586599,
      "tokens_seen": 399769600
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008877231695085256,
      "loss": 1.5935,
      "theoretical_loss": 4.040601635670501,
      "tokens_seen": 399835136
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008877031093279839,
      "loss": 1.6786,
      "theoretical_loss": 4.040528064191441,
      "tokens_seen": 399900672
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008876830491474423,
      "loss": 1.6391,
      "theoretical_loss": 4.040454508143652,
      "tokens_seen": 399966208
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008876629889669007,
      "loss": 1.6811,
      "theoretical_loss": 4.04038096752137,
      "tokens_seen": 400031744
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008876429287863591,
      "loss": 1.6204,
      "theoretical_loss": 4.040307442318834,
      "tokens_seen": 400097280
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008876228686058174,
      "loss": 1.7193,
      "theoretical_loss": 4.0402339325302865,
      "tokens_seen": 400162816
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008876028084252759,
      "loss": 1.772,
      "theoretical_loss": 4.040160438149971,
      "tokens_seen": 400228352
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008875827482447342,
      "loss": 1.7396,
      "theoretical_loss": 4.040086959172136,
      "tokens_seen": 400293888
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008875626880641926,
      "loss": 1.6233,
      "theoretical_loss": 4.040013495591033,
      "tokens_seen": 400359424
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000887542627883651,
      "loss": 1.6648,
      "theoretical_loss": 4.039940047400916,
      "tokens_seen": 400424960
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008875225677031093,
      "loss": 1.6596,
      "theoretical_loss": 4.0398666145960425,
      "tokens_seen": 400490496
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008875025075225677,
      "loss": 1.5847,
      "theoretical_loss": 4.039793197170672,
      "tokens_seen": 400556032
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008874824473420261,
      "loss": 1.6665,
      "theoretical_loss": 4.03971979511907,
      "tokens_seen": 400621568
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008874623871614844,
      "loss": 1.8048,
      "theoretical_loss": 4.039646408435502,
      "tokens_seen": 400687104
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008874423269809429,
      "loss": 1.7428,
      "theoretical_loss": 4.039573037114236,
      "tokens_seen": 400752640
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008874222668004011,
      "loss": 1.737,
      "theoretical_loss": 4.039499681149547,
      "tokens_seen": 400818176
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008874022066198596,
      "loss": 1.7766,
      "theoretical_loss": 4.0394263405357105,
      "tokens_seen": 400883712
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008873821464393181,
      "loss": 1.7353,
      "theoretical_loss": 4.039353015267004,
      "tokens_seen": 400949248
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008873620862587763,
      "loss": 1.6646,
      "theoretical_loss": 4.0392797053377105,
      "tokens_seen": 401014784
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008873420260782348,
      "loss": 1.7196,
      "theoretical_loss": 4.039206410742114,
      "tokens_seen": 401080320
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000887321965897693,
      "loss": 1.7357,
      "theoretical_loss": 4.039133131474505,
      "tokens_seen": 401145856
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008873019057171515,
      "loss": 1.5398,
      "theoretical_loss": 4.039059867529171,
      "tokens_seen": 401211392
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008872818455366099,
      "loss": 1.7319,
      "theoretical_loss": 4.0389866189004096,
      "tokens_seen": 401276928
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008872617853560682,
      "loss": 1.6942,
      "theoretical_loss": 4.038913385582515,
      "tokens_seen": 401342464
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008872417251755266,
      "loss": 1.6616,
      "theoretical_loss": 4.03884016756979,
      "tokens_seen": 401408000
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008872216649949851,
      "loss": 1.6534,
      "theoretical_loss": 4.038766964856537,
      "tokens_seen": 401473536
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008872016048144433,
      "loss": 1.675,
      "theoretical_loss": 4.038693777437061,
      "tokens_seen": 401539072
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008871815446339018,
      "loss": 1.7144,
      "theoretical_loss": 4.038620605305673,
      "tokens_seen": 401604608
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.00088716148445336,
      "loss": 1.6308,
      "theoretical_loss": 4.0385474484566855,
      "tokens_seen": 401670144
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008871414242728185,
      "loss": 1.6743,
      "theoretical_loss": 4.038474306884414,
      "tokens_seen": 401735680
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008871213640922769,
      "loss": 1.7202,
      "theoretical_loss": 4.038401180583175,
      "tokens_seen": 401801216
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008871013039117352,
      "loss": 1.5014,
      "theoretical_loss": 4.038328069547293,
      "tokens_seen": 401866752
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008870812437311936,
      "loss": 1.7556,
      "theoretical_loss": 4.0382549737710915,
      "tokens_seen": 401932288
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000887061183550652,
      "loss": 1.6331,
      "theoretical_loss": 4.038181893248898,
      "tokens_seen": 401997824
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008870411233701103,
      "loss": 1.7542,
      "theoretical_loss": 4.038108827975043,
      "tokens_seen": 402063360
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008870210631895688,
      "loss": 1.6324,
      "theoretical_loss": 4.03803577794386,
      "tokens_seen": 402128896
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008870010030090271,
      "loss": 1.6813,
      "theoretical_loss": 4.037962743149687,
      "tokens_seen": 402194432
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008869809428284855,
      "loss": 1.6299,
      "theoretical_loss": 4.037889723586863,
      "tokens_seen": 402259968
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008869608826479438,
      "loss": 1.8496,
      "theoretical_loss": 4.03781671924973,
      "tokens_seen": 402325504
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008869408224674023,
      "loss": 1.6649,
      "theoretical_loss": 4.037743730132635,
      "tokens_seen": 402391040
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008869207622868606,
      "loss": 1.6666,
      "theoretical_loss": 4.037670756229927,
      "tokens_seen": 402456576
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000886900702106319,
      "loss": 1.5544,
      "theoretical_loss": 4.037597797535957,
      "tokens_seen": 402522112
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008868806419257773,
      "loss": 1.6848,
      "theoretical_loss": 4.03752485404508,
      "tokens_seen": 402587648
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008868605817452358,
      "loss": 1.6837,
      "theoretical_loss": 4.037451925751654,
      "tokens_seen": 402653184
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008868405215646941,
      "loss": 1.7491,
      "theoretical_loss": 4.037379012650041,
      "tokens_seen": 402718720
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008868204613841525,
      "loss": 1.6797,
      "theoretical_loss": 4.037306114734603,
      "tokens_seen": 402784256
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008868004012036108,
      "loss": 1.7346,
      "theoretical_loss": 4.037233231999708,
      "tokens_seen": 402849792
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008867803410230693,
      "loss": 1.6458,
      "theoretical_loss": 4.0371603644397265,
      "tokens_seen": 402915328
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008867602808425276,
      "loss": 1.7762,
      "theoretical_loss": 4.03708751204903,
      "tokens_seen": 402980864
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.4884590804576874,
      "objective/train/docs_used": 152845,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.555906891822815,
      "objective/train/original_loss": 1.5559066534042358,
      "objective/train/theoretical_loss": 4.037014674821996,
      "objective/train/tokens_used": 423506400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.23926691710948944,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0500620603561401,
      "objective/train/weighted_lm_loss": 1.6336621046066284,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.037014674821996,
      "tokens_seen": 403046400
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000886740220661986,
      "loss": 1.7122,
      "theoretical_loss": 4.037014674821996,
      "tokens_seen": 403046400
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008867201604814443,
      "loss": 1.781,
      "theoretical_loss": 4.036941852753001,
      "tokens_seen": 403111936
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008867001003009027,
      "loss": 1.6895,
      "theoretical_loss": 4.03686904583643,
      "tokens_seen": 403177472
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008866800401203611,
      "loss": 1.7167,
      "theoretical_loss": 4.036796254066666,
      "tokens_seen": 403243008
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008866599799398195,
      "loss": 1.6275,
      "theoretical_loss": 4.036723477438097,
      "tokens_seen": 403308544
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008866399197592778,
      "loss": 1.6639,
      "theoretical_loss": 4.0366507159451155,
      "tokens_seen": 403374080
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008866198595787363,
      "loss": 1.7336,
      "theoretical_loss": 4.036577969582114,
      "tokens_seen": 403439616
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008865997993981946,
      "loss": 1.7358,
      "theoretical_loss": 4.036505238343488,
      "tokens_seen": 403505152
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000886579739217653,
      "loss": 1.6951,
      "theoretical_loss": 4.03643252222364,
      "tokens_seen": 403570688
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008865596790371114,
      "loss": 1.5281,
      "theoretical_loss": 4.036359821216973,
      "tokens_seen": 403636224
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008865396188565697,
      "loss": 1.6679,
      "theoretical_loss": 4.03628713531789,
      "tokens_seen": 403701760
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008865195586760281,
      "loss": 1.6987,
      "theoretical_loss": 4.036214464520802,
      "tokens_seen": 403767296
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008864994984954865,
      "loss": 1.7154,
      "theoretical_loss": 4.03614180882012,
      "tokens_seen": 403832832
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008864794383149448,
      "loss": 1.7357,
      "theoretical_loss": 4.036069168210259,
      "tokens_seen": 403898368
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008864593781344033,
      "loss": 1.7683,
      "theoretical_loss": 4.035996542685638,
      "tokens_seen": 403963904
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008864393179538615,
      "loss": 1.5992,
      "theoretical_loss": 4.035923932240674,
      "tokens_seen": 404029440
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.00088641925777332,
      "loss": 1.6875,
      "theoretical_loss": 4.035851336869795,
      "tokens_seen": 404094976
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008863991975927784,
      "loss": 1.6608,
      "theoretical_loss": 4.035778756567426,
      "tokens_seen": 404160512
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008863791374122367,
      "loss": 1.7379,
      "theoretical_loss": 4.0357061913279955,
      "tokens_seen": 404226048
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008863590772316951,
      "loss": 1.687,
      "theoretical_loss": 4.035633641145937,
      "tokens_seen": 404291584
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008863390170511534,
      "loss": 1.6801,
      "theoretical_loss": 4.035561106015686,
      "tokens_seen": 404357120
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008863189568706118,
      "loss": 1.7063,
      "theoretical_loss": 4.035488585931681,
      "tokens_seen": 404422656
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008862988966900703,
      "loss": 1.6407,
      "theoretical_loss": 4.035416080888364,
      "tokens_seen": 404488192
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008862788365095285,
      "loss": 1.7144,
      "theoretical_loss": 4.035343590880178,
      "tokens_seen": 404553728
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000886258776328987,
      "loss": 1.6793,
      "theoretical_loss": 4.035271115901571,
      "tokens_seen": 404619264
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008862387161484454,
      "loss": 1.8213,
      "theoretical_loss": 4.035198655946994,
      "tokens_seen": 404684800
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008862186559679037,
      "loss": 1.7371,
      "theoretical_loss": 4.035126211010899,
      "tokens_seen": 404750336
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008861985957873621,
      "loss": 1.5614,
      "theoretical_loss": 4.035053781087743,
      "tokens_seen": 404815872
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008861785356068204,
      "loss": 1.6326,
      "theoretical_loss": 4.034981366171985,
      "tokens_seen": 404881408
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008861584754262788,
      "loss": 1.6813,
      "theoretical_loss": 4.0349089662580875,
      "tokens_seen": 404946944
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008861384152457373,
      "loss": 1.7213,
      "theoretical_loss": 4.034836581340515,
      "tokens_seen": 405012480
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008861183550651955,
      "loss": 1.6701,
      "theoretical_loss": 4.034764211413735,
      "tokens_seen": 405078016
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000886098294884654,
      "loss": 1.6912,
      "theoretical_loss": 4.034691856472219,
      "tokens_seen": 405143552
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008860782347041123,
      "loss": 1.6378,
      "theoretical_loss": 4.034619516510441,
      "tokens_seen": 405209088
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008860581745235707,
      "loss": 1.7264,
      "theoretical_loss": 4.034547191522877,
      "tokens_seen": 405274624
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008860381143430291,
      "loss": 1.8048,
      "theoretical_loss": 4.034474881504007,
      "tokens_seen": 405340160
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008860180541624875,
      "loss": 1.743,
      "theoretical_loss": 4.034402586448315,
      "tokens_seen": 405405696
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008859979939819458,
      "loss": 1.7185,
      "theoretical_loss": 4.0343303063502844,
      "tokens_seen": 405471232
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008859779338014042,
      "loss": 1.9438,
      "theoretical_loss": 4.034258041204404,
      "tokens_seen": 405536768
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008859578736208625,
      "loss": 1.6551,
      "theoretical_loss": 4.034185791005166,
      "tokens_seen": 405602304
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000885937813440321,
      "loss": 1.703,
      "theoretical_loss": 4.034113555747065,
      "tokens_seen": 405667840
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008859177532597793,
      "loss": 1.73,
      "theoretical_loss": 4.034041335424597,
      "tokens_seen": 405733376
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008858976930792377,
      "loss": 1.6462,
      "theoretical_loss": 4.033969130032263,
      "tokens_seen": 405798912
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008858776328986962,
      "loss": 1.6174,
      "theoretical_loss": 4.033896939564565,
      "tokens_seen": 405864448
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008858575727181545,
      "loss": 1.6147,
      "theoretical_loss": 4.0338247640160105,
      "tokens_seen": 405929984
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008858375125376129,
      "loss": 1.8021,
      "theoretical_loss": 4.0337526033811075,
      "tokens_seen": 405995520
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008858174523570712,
      "loss": 1.6092,
      "theoretical_loss": 4.033680457654368,
      "tokens_seen": 406061056
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008857973921765297,
      "loss": 1.576,
      "theoretical_loss": 4.033608326830306,
      "tokens_seen": 406126592
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000885777331995988,
      "loss": 1.7152,
      "theoretical_loss": 4.033536210903439,
      "tokens_seen": 406192128
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008857572718154464,
      "loss": 1.7216,
      "theoretical_loss": 4.033464109868289,
      "tokens_seen": 406257664
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.49587690830230713,
      "objective/train/docs_used": 154148,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5138962268829346,
      "objective/train/original_loss": 1.5138959884643555,
      "objective/train/theoretical_loss": 4.0333920237193785,
      "objective/train/tokens_used": 426783200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24605120718479156,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0508387088775635,
      "objective/train/weighted_lm_loss": 1.5908278226852417,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.0333920237193785,
      "tokens_seen": 406323200
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008857372116349047,
      "loss": 1.614,
      "theoretical_loss": 4.0333920237193785,
      "tokens_seen": 406323200
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008857171514543631,
      "loss": 1.5859,
      "theoretical_loss": 4.033319952451233,
      "tokens_seen": 406388736
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008856970912738215,
      "loss": 1.5933,
      "theoretical_loss": 4.033247896058382,
      "tokens_seen": 406454272
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008856770310932799,
      "loss": 1.6813,
      "theoretical_loss": 4.03317585453536,
      "tokens_seen": 406519808
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008856569709127382,
      "loss": 1.4754,
      "theoretical_loss": 4.0331038278766975,
      "tokens_seen": 406585344
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008856369107321967,
      "loss": 1.6915,
      "theoretical_loss": 4.033031816076936,
      "tokens_seen": 406650880
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000885616850551655,
      "loss": 1.7038,
      "theoretical_loss": 4.032959819130615,
      "tokens_seen": 406716416
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008855967903711134,
      "loss": 1.7263,
      "theoretical_loss": 4.032887837032277,
      "tokens_seen": 406781952
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008855767301905718,
      "loss": 1.8097,
      "theoretical_loss": 4.032815869776471,
      "tokens_seen": 406847488
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008855566700100301,
      "loss": 1.7635,
      "theoretical_loss": 4.0327439173577435,
      "tokens_seen": 406913024
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008855366098294885,
      "loss": 1.7598,
      "theoretical_loss": 4.032671979770649,
      "tokens_seen": 406978560
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008855165496489469,
      "loss": 1.6579,
      "theoretical_loss": 4.032600057009741,
      "tokens_seen": 407044096
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008854964894684052,
      "loss": 1.6921,
      "theoretical_loss": 4.032528149069579,
      "tokens_seen": 407109632
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008854764292878637,
      "loss": 1.7444,
      "theoretical_loss": 4.032456255944723,
      "tokens_seen": 407175168
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008854563691073219,
      "loss": 1.6354,
      "theoretical_loss": 4.032384377629737,
      "tokens_seen": 407240704
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008854363089267804,
      "loss": 1.84,
      "theoretical_loss": 4.032312514119187,
      "tokens_seen": 407306240
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008854162487462388,
      "loss": 1.6797,
      "theoretical_loss": 4.0322406654076435,
      "tokens_seen": 407371776
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008853961885656971,
      "loss": 1.7291,
      "theoretical_loss": 4.032168831489678,
      "tokens_seen": 407437312
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008853761283851555,
      "loss": 1.7825,
      "theoretical_loss": 4.032097012359867,
      "tokens_seen": 407502848
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008853560682046138,
      "loss": 1.6212,
      "theoretical_loss": 4.0320252080127865,
      "tokens_seen": 407568384
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008853360080240722,
      "loss": 1.7734,
      "theoretical_loss": 4.0319534184430195,
      "tokens_seen": 407633920
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008853159478435307,
      "loss": 1.6859,
      "theoretical_loss": 4.031881643645149,
      "tokens_seen": 407699456
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008852958876629889,
      "loss": 1.7826,
      "theoretical_loss": 4.031809883613761,
      "tokens_seen": 407764992
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008852758274824474,
      "loss": 1.5211,
      "theoretical_loss": 4.031738138343446,
      "tokens_seen": 407830528
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008852557673019058,
      "loss": 1.7162,
      "theoretical_loss": 4.031666407828795,
      "tokens_seen": 407896064
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008852357071213641,
      "loss": 1.7394,
      "theoretical_loss": 4.031594692064405,
      "tokens_seen": 407961600
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008852156469408225,
      "loss": 1.6073,
      "theoretical_loss": 4.031522991044873,
      "tokens_seen": 408027136
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008851955867602808,
      "loss": 1.7767,
      "theoretical_loss": 4.0314513047648,
      "tokens_seen": 408092672
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008851755265797392,
      "loss": 1.5802,
      "theoretical_loss": 4.03137963321879,
      "tokens_seen": 408158208
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008851554663991977,
      "loss": 1.5961,
      "theoretical_loss": 4.031307976401449,
      "tokens_seen": 408223744
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008851354062186559,
      "loss": 1.7465,
      "theoretical_loss": 4.031236334307387,
      "tokens_seen": 408289280
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008851153460381144,
      "loss": 1.8301,
      "theoretical_loss": 4.031164706931215,
      "tokens_seen": 408354816
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008850952858575727,
      "loss": 1.6845,
      "theoretical_loss": 4.0310930942675505,
      "tokens_seen": 408420352
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008850752256770311,
      "loss": 1.7814,
      "theoretical_loss": 4.03102149631101,
      "tokens_seen": 408485888
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008850551654964895,
      "loss": 1.5873,
      "theoretical_loss": 4.030949913056213,
      "tokens_seen": 408551424
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008850351053159479,
      "loss": 1.7496,
      "theoretical_loss": 4.030878344497786,
      "tokens_seen": 408616960
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008850150451354062,
      "loss": 1.6609,
      "theoretical_loss": 4.030806790630353,
      "tokens_seen": 408682496
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008849949849548646,
      "loss": 1.6315,
      "theoretical_loss": 4.030735251448545,
      "tokens_seen": 408748032
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008849749247743229,
      "loss": 1.6608,
      "theoretical_loss": 4.030663726946992,
      "tokens_seen": 408813568
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008849548645937814,
      "loss": 1.7091,
      "theoretical_loss": 4.030592217120331,
      "tokens_seen": 408879104
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008849348044132397,
      "loss": 1.605,
      "theoretical_loss": 4.030520721963199,
      "tokens_seen": 408944640
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008849147442326981,
      "loss": 1.6562,
      "theoretical_loss": 4.030449241470235,
      "tokens_seen": 409010176
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008848946840521565,
      "loss": 1.6737,
      "theoretical_loss": 4.030377775636084,
      "tokens_seen": 409075712
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008848746238716149,
      "loss": 1.6781,
      "theoretical_loss": 4.030306324455392,
      "tokens_seen": 409141248
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008848545636910732,
      "loss": 1.5943,
      "theoretical_loss": 4.030234887922808,
      "tokens_seen": 409206784
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008848345035105316,
      "loss": 1.6954,
      "theoretical_loss": 4.030163466032984,
      "tokens_seen": 409272320
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.00088481444332999,
      "loss": 1.6496,
      "theoretical_loss": 4.030092058780573,
      "tokens_seen": 409337856
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008847943831494484,
      "loss": 1.8028,
      "theoretical_loss": 4.030020666160234,
      "tokens_seen": 409403392
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008847743229689067,
      "loss": 1.6431,
      "theoretical_loss": 4.029949288166627,
      "tokens_seen": 409468928
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008847542627883651,
      "loss": 1.5435,
      "theoretical_loss": 4.029877924794415,
      "tokens_seen": 409534464
    },
    {
      "epoch": 0.12,
      "objective/train/advantage_avg": 0.49289989471435547,
      "objective/train/docs_used": 155224,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.73809814453125,
      "objective/train/original_loss": 1.738098382949829,
      "objective/train/theoretical_loss": 4.029806576038263,
      "objective/train/tokens_used": 430060000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24331925809383392,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505269765853882,
      "objective/train/weighted_lm_loss": 1.8255553245544434,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.029806576038263,
      "tokens_seen": 409600000
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008847342026078234,
      "loss": 1.6939,
      "theoretical_loss": 4.029806576038263,
      "tokens_seen": 409600000
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008847141424272819,
      "loss": 1.5399,
      "theoretical_loss": 4.0297352418928405,
      "tokens_seen": 409665536
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008846940822467402,
      "loss": 1.5897,
      "theoretical_loss": 4.0296639223528175,
      "tokens_seen": 409731072
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008846740220661986,
      "loss": 1.7601,
      "theoretical_loss": 4.02959261741287,
      "tokens_seen": 409796608
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000884653961885657,
      "loss": 1.4896,
      "theoretical_loss": 4.029521327067675,
      "tokens_seen": 409862144
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008846339017051154,
      "loss": 1.6404,
      "theoretical_loss": 4.029450051311912,
      "tokens_seen": 409927680
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008846138415245737,
      "loss": 1.7851,
      "theoretical_loss": 4.029378790140261,
      "tokens_seen": 409993216
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008845937813440322,
      "loss": 1.6865,
      "theoretical_loss": 4.029307543547411,
      "tokens_seen": 410058752
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008845737211634904,
      "loss": 1.6179,
      "theoretical_loss": 4.029236311528049,
      "tokens_seen": 410124288
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008845536609829489,
      "loss": 1.6504,
      "theoretical_loss": 4.029165094076864,
      "tokens_seen": 410189824
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008845336008024072,
      "loss": 1.784,
      "theoretical_loss": 4.029093891188552,
      "tokens_seen": 410255360
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008845135406218656,
      "loss": 1.7336,
      "theoretical_loss": 4.029022702857809,
      "tokens_seen": 410320896
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000884493480441324,
      "loss": 1.7137,
      "theoretical_loss": 4.028951529079334,
      "tokens_seen": 410386432
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008844734202607823,
      "loss": 1.6833,
      "theoretical_loss": 4.028880369847829,
      "tokens_seen": 410451968
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008844533600802407,
      "loss": 1.6997,
      "theoretical_loss": 4.028809225158,
      "tokens_seen": 410517504
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008844332998996992,
      "loss": 1.7912,
      "theoretical_loss": 4.0287380950045515,
      "tokens_seen": 410583040
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008844132397191574,
      "loss": 1.5385,
      "theoretical_loss": 4.028666979382197,
      "tokens_seen": 410648576
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008843931795386159,
      "loss": 1.6839,
      "theoretical_loss": 4.028595878285648,
      "tokens_seen": 410714112
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008843731193580742,
      "loss": 1.6628,
      "theoretical_loss": 4.028524791709621,
      "tokens_seen": 410779648
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008843530591775326,
      "loss": 1.6706,
      "theoretical_loss": 4.0284537196488355,
      "tokens_seen": 410845184
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008843329989969911,
      "loss": 1.6575,
      "theoretical_loss": 4.028382662098011,
      "tokens_seen": 410910720
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008843129388164493,
      "loss": 1.6833,
      "theoretical_loss": 4.028311619051873,
      "tokens_seen": 410976256
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008842928786359078,
      "loss": 1.7105,
      "theoretical_loss": 4.028240590505148,
      "tokens_seen": 411041792
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008842728184553662,
      "loss": 1.6891,
      "theoretical_loss": 4.028169576452566,
      "tokens_seen": 411107328
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008842527582748245,
      "loss": 1.6028,
      "theoretical_loss": 4.028098576888859,
      "tokens_seen": 411172864
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008842326980942829,
      "loss": 1.8081,
      "theoretical_loss": 4.028027591808764,
      "tokens_seen": 411238400
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008842126379137413,
      "loss": 1.6304,
      "theoretical_loss": 4.027956621207015,
      "tokens_seen": 411303936
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008841925777331996,
      "loss": 1.7428,
      "theoretical_loss": 4.027885665078356,
      "tokens_seen": 411369472
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008841725175526581,
      "loss": 1.6705,
      "theoretical_loss": 4.0278147234175306,
      "tokens_seen": 411435008
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008841524573721163,
      "loss": 1.6065,
      "theoretical_loss": 4.0277437962192835,
      "tokens_seen": 411500544
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008841323971915748,
      "loss": 1.6634,
      "theoretical_loss": 4.027672883478364,
      "tokens_seen": 411566080
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008841123370110331,
      "loss": 1.8244,
      "theoretical_loss": 4.027601985189524,
      "tokens_seen": 411631616
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008840922768304915,
      "loss": 1.6679,
      "theoretical_loss": 4.027531101347519,
      "tokens_seen": 411697152
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008840722166499499,
      "loss": 1.7921,
      "theoretical_loss": 4.027460231947105,
      "tokens_seen": 411762688
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008840521564694083,
      "loss": 1.6681,
      "theoretical_loss": 4.027389376983041,
      "tokens_seen": 411828224
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008840320962888666,
      "loss": 1.5784,
      "theoretical_loss": 4.027318536450091,
      "tokens_seen": 411893760
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000884012036108325,
      "loss": 1.8254,
      "theoretical_loss": 4.02724771034302,
      "tokens_seen": 411959296
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008839919759277833,
      "loss": 1.8391,
      "theoretical_loss": 4.027176898656596,
      "tokens_seen": 412024832
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008839719157472418,
      "loss": 1.716,
      "theoretical_loss": 4.02710610138559,
      "tokens_seen": 412090368
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008839518555667001,
      "loss": 1.7174,
      "theoretical_loss": 4.027035318524776,
      "tokens_seen": 412155904
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008839317953861585,
      "loss": 1.6868,
      "theoretical_loss": 4.02696455006893,
      "tokens_seen": 412221440
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008839117352056169,
      "loss": 1.6135,
      "theoretical_loss": 4.026893796012831,
      "tokens_seen": 412286976
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008838916750250753,
      "loss": 1.6772,
      "theoretical_loss": 4.02682305635126,
      "tokens_seen": 412352512
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.0008838716148445336,
      "loss": 1.7418,
      "theoretical_loss": 4.0267523310790025,
      "tokens_seen": 412418048
    },
    {
      "epoch": 0.12,
      "learning_rate": 0.000883851554663992,
      "loss": 1.5971,
      "theoretical_loss": 4.026681620190845,
      "tokens_seen": 412483584
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008838314944834503,
      "loss": 1.7827,
      "theoretical_loss": 4.026610923681578,
      "tokens_seen": 412549120
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008838114343029088,
      "loss": 1.6737,
      "theoretical_loss": 4.026540241545994,
      "tokens_seen": 412614656
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008837913741223671,
      "loss": 1.663,
      "theoretical_loss": 4.026469573778888,
      "tokens_seen": 412680192
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008837713139418255,
      "loss": 1.8313,
      "theoretical_loss": 4.026398920375057,
      "tokens_seen": 412745728
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008837512537612838,
      "loss": 1.5836,
      "theoretical_loss": 4.026328281329304,
      "tokens_seen": 412811264
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.49095261096954346,
      "objective/train/docs_used": 156351,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5056899785995483,
      "objective/train/original_loss": 1.5056899785995483,
      "objective/train/theoretical_loss": 4.026257656636431,
      "objective/train/tokens_used": 433336800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24246512353420258,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503277778625488,
      "objective/train/weighted_lm_loss": 1.5813159942626953,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.026257656636431,
      "tokens_seen": 412876800
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008837311935807423,
      "loss": 1.5957,
      "theoretical_loss": 4.026257656636431,
      "tokens_seen": 412876800
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008837111334002006,
      "loss": 1.7631,
      "theoretical_loss": 4.026187046291244,
      "tokens_seen": 412942336
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000883691073219659,
      "loss": 1.6013,
      "theoretical_loss": 4.026116450288553,
      "tokens_seen": 413007872
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008836710130391174,
      "loss": 1.7958,
      "theoretical_loss": 4.026045868623169,
      "tokens_seen": 413073408
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008836509528585758,
      "loss": 1.6075,
      "theoretical_loss": 4.025975301289906,
      "tokens_seen": 413138944
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008836308926780341,
      "loss": 1.6469,
      "theoretical_loss": 4.02590474828358,
      "tokens_seen": 413204480
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008836108324974926,
      "loss": 1.6781,
      "theoretical_loss": 4.0258342095990125,
      "tokens_seen": 413270016
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008835907723169508,
      "loss": 1.645,
      "theoretical_loss": 4.025763685231025,
      "tokens_seen": 413335552
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008835707121364093,
      "loss": 1.769,
      "theoretical_loss": 4.025693175174443,
      "tokens_seen": 413401088
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008835506519558676,
      "loss": 1.631,
      "theoretical_loss": 4.025622679424094,
      "tokens_seen": 413466624
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000883530591775326,
      "loss": 1.5995,
      "theoretical_loss": 4.0255521979748075,
      "tokens_seen": 413532160
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008835105315947844,
      "loss": 1.612,
      "theoretical_loss": 4.025481730821418,
      "tokens_seen": 413597696
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008834904714142427,
      "loss": 1.6699,
      "theoretical_loss": 4.02541127795876,
      "tokens_seen": 413663232
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008834704112337011,
      "loss": 1.7716,
      "theoretical_loss": 4.025340839381673,
      "tokens_seen": 413728768
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008834503510531596,
      "loss": 1.6416,
      "theoretical_loss": 4.025270415084998,
      "tokens_seen": 413794304
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008834302908726178,
      "loss": 1.5975,
      "theoretical_loss": 4.025200005063578,
      "tokens_seen": 413859840
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008834102306920763,
      "loss": 1.6981,
      "theoretical_loss": 4.02512960931226,
      "tokens_seen": 413925376
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008833901705115345,
      "loss": 1.536,
      "theoretical_loss": 4.025059227825894,
      "tokens_seen": 413990912
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000883370110330993,
      "loss": 1.6279,
      "theoretical_loss": 4.024988860599331,
      "tokens_seen": 414056448
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008833500501504514,
      "loss": 1.5693,
      "theoretical_loss": 4.024918507627426,
      "tokens_seen": 414121984
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008833299899699097,
      "loss": 1.7256,
      "theoretical_loss": 4.024848168905035,
      "tokens_seen": 414187520
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008833099297893681,
      "loss": 1.7206,
      "theoretical_loss": 4.024777844427019,
      "tokens_seen": 414253056
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008832898696088266,
      "loss": 1.6334,
      "theoretical_loss": 4.024707534188241,
      "tokens_seen": 414318592
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008832698094282848,
      "loss": 1.6603,
      "theoretical_loss": 4.024637238183566,
      "tokens_seen": 414384128
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008832497492477433,
      "loss": 1.7197,
      "theoretical_loss": 4.02456695640786,
      "tokens_seen": 414449664
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008832296890672015,
      "loss": 1.5665,
      "theoretical_loss": 4.024496688855995,
      "tokens_seen": 414515200
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.00088320962888666,
      "loss": 1.6843,
      "theoretical_loss": 4.0244264355228445,
      "tokens_seen": 414580736
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008831895687061184,
      "loss": 1.7103,
      "theoretical_loss": 4.024356196403284,
      "tokens_seen": 414646272
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008831695085255767,
      "loss": 1.6974,
      "theoretical_loss": 4.0242859714921915,
      "tokens_seen": 414711808
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008831494483450351,
      "loss": 1.657,
      "theoretical_loss": 4.0242157607844495,
      "tokens_seen": 414777344
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008831293881644935,
      "loss": 1.657,
      "theoretical_loss": 4.02414556427494,
      "tokens_seen": 414842880
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008831093279839518,
      "loss": 1.8203,
      "theoretical_loss": 4.024075381958551,
      "tokens_seen": 414908416
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008830892678034103,
      "loss": 1.7671,
      "theoretical_loss": 4.024005213830171,
      "tokens_seen": 414973952
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008830692076228685,
      "loss": 1.6444,
      "theoretical_loss": 4.023935059884692,
      "tokens_seen": 415039488
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000883049147442327,
      "loss": 1.7232,
      "theoretical_loss": 4.023864920117007,
      "tokens_seen": 415105024
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008830290872617853,
      "loss": 1.6406,
      "theoretical_loss": 4.023794794522015,
      "tokens_seen": 415170560
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008830090270812437,
      "loss": 1.471,
      "theoretical_loss": 4.023724683094615,
      "tokens_seen": 415236096
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008829889669007021,
      "loss": 1.4989,
      "theoretical_loss": 4.023654585829709,
      "tokens_seen": 415301632
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008829689067201605,
      "loss": 1.6739,
      "theoretical_loss": 4.023584502722203,
      "tokens_seen": 415367168
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008829488465396188,
      "loss": 1.6447,
      "theoretical_loss": 4.023514433767002,
      "tokens_seen": 415432704
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008829287863590773,
      "loss": 1.6639,
      "theoretical_loss": 4.023444378959019,
      "tokens_seen": 415498240
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008829087261785356,
      "loss": 1.6446,
      "theoretical_loss": 4.023374338293166,
      "tokens_seen": 415563776
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000882888665997994,
      "loss": 1.5449,
      "theoretical_loss": 4.023304311764359,
      "tokens_seen": 415629312
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008828686058174524,
      "loss": 1.7832,
      "theoretical_loss": 4.023234299367514,
      "tokens_seen": 415694848
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008828485456369108,
      "loss": 1.6643,
      "theoretical_loss": 4.023164301097555,
      "tokens_seen": 415760384
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008828284854563692,
      "loss": 1.76,
      "theoretical_loss": 4.023094316949403,
      "tokens_seen": 415825920
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008828084252758275,
      "loss": 1.5454,
      "theoretical_loss": 4.0230243469179845,
      "tokens_seen": 415891456
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008827883650952859,
      "loss": 1.6855,
      "theoretical_loss": 4.022954390998228,
      "tokens_seen": 415956992
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008827683049147442,
      "loss": 1.666,
      "theoretical_loss": 4.0228844491850655,
      "tokens_seen": 416022528
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008827482447342027,
      "loss": 1.6823,
      "theoretical_loss": 4.02281452147343,
      "tokens_seen": 416088064
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.49329674243927,
      "objective/train/docs_used": 157543,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.461356520652771,
      "objective/train/original_loss": 1.461356520652771,
      "objective/train/theoretical_loss": 4.022744607858259,
      "objective/train/tokens_used": 436613600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2439008206129074,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505695343017578,
      "objective/train/weighted_lm_loss": 1.535351037979126,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.022744607858259,
      "tokens_seen": 416153600
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000882728184553661,
      "loss": 1.6574,
      "theoretical_loss": 4.022744607858259,
      "tokens_seen": 416153600
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008827081243731194,
      "loss": 1.6683,
      "theoretical_loss": 4.022674708334492,
      "tokens_seen": 416219136
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008826880641925778,
      "loss": 1.8037,
      "theoretical_loss": 4.022604822897068,
      "tokens_seen": 416284672
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008826680040120362,
      "loss": 1.7671,
      "theoretical_loss": 4.022534951540933,
      "tokens_seen": 416350208
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008826479438314945,
      "loss": 1.7737,
      "theoretical_loss": 4.022465094261035,
      "tokens_seen": 416415744
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000882627883650953,
      "loss": 1.5922,
      "theoretical_loss": 4.022395251052322,
      "tokens_seen": 416481280
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008826078234704112,
      "loss": 1.6788,
      "theoretical_loss": 4.0223254219097475,
      "tokens_seen": 416546816
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008825877632898697,
      "loss": 1.6164,
      "theoretical_loss": 4.022255606828264,
      "tokens_seen": 416612352
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000882567703109328,
      "loss": 1.7734,
      "theoretical_loss": 4.022185805802832,
      "tokens_seen": 416677888
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008825476429287864,
      "loss": 1.6331,
      "theoretical_loss": 4.022116018828409,
      "tokens_seen": 416743424
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008825275827482448,
      "loss": 1.7915,
      "theoretical_loss": 4.022046245899958,
      "tokens_seen": 416808960
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008825075225677031,
      "loss": 1.591,
      "theoretical_loss": 4.021976487012445,
      "tokens_seen": 416874496
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008824874623871615,
      "loss": 1.821,
      "theoretical_loss": 4.021906742160837,
      "tokens_seen": 416940032
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.00088246740220662,
      "loss": 1.7072,
      "theoretical_loss": 4.0218370113401045,
      "tokens_seen": 417005568
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008824473420260782,
      "loss": 1.5424,
      "theoretical_loss": 4.021767294545221,
      "tokens_seen": 417071104
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008824272818455367,
      "loss": 1.5978,
      "theoretical_loss": 4.0216975917711615,
      "tokens_seen": 417136640
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008824072216649949,
      "loss": 1.771,
      "theoretical_loss": 4.021627903012904,
      "tokens_seen": 417202176
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008823871614844534,
      "loss": 1.6915,
      "theoretical_loss": 4.021558228265429,
      "tokens_seen": 417267712
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008823671013039118,
      "loss": 1.7399,
      "theoretical_loss": 4.021488567523721,
      "tokens_seen": 417333248
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008823470411233701,
      "loss": 1.6824,
      "theoretical_loss": 4.021418920782764,
      "tokens_seen": 417398784
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008823269809428285,
      "loss": 1.7192,
      "theoretical_loss": 4.021349288037549,
      "tokens_seen": 417464320
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000882306920762287,
      "loss": 1.6438,
      "theoretical_loss": 4.021279669283064,
      "tokens_seen": 417529856
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008822868605817452,
      "loss": 1.7288,
      "theoretical_loss": 4.021210064514305,
      "tokens_seen": 417595392
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008822668004012037,
      "loss": 1.6427,
      "theoretical_loss": 4.021140473726268,
      "tokens_seen": 417660928
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008822467402206619,
      "loss": 1.6762,
      "theoretical_loss": 4.021070896913951,
      "tokens_seen": 417726464
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008822266800401204,
      "loss": 1.551,
      "theoretical_loss": 4.021001334072355,
      "tokens_seen": 417792000
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008822066198595788,
      "loss": 1.6138,
      "theoretical_loss": 4.020931785196484,
      "tokens_seen": 417857536
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008821865596790371,
      "loss": 1.673,
      "theoretical_loss": 4.020862250281345,
      "tokens_seen": 417923072
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008821664994984955,
      "loss": 1.5015,
      "theoretical_loss": 4.020792729321947,
      "tokens_seen": 417988608
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008821464393179539,
      "loss": 1.6645,
      "theoretical_loss": 4.020723222313302,
      "tokens_seen": 418054144
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008821263791374122,
      "loss": 1.7655,
      "theoretical_loss": 4.020653729250424,
      "tokens_seen": 418119680
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008821063189568707,
      "loss": 1.6516,
      "theoretical_loss": 4.020584250128329,
      "tokens_seen": 418185216
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000882086258776329,
      "loss": 1.7177,
      "theoretical_loss": 4.020514784942037,
      "tokens_seen": 418250752
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008820661985957874,
      "loss": 1.5534,
      "theoretical_loss": 4.0204453336865695,
      "tokens_seen": 418316288
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008820461384152457,
      "loss": 1.6524,
      "theoretical_loss": 4.020375896356951,
      "tokens_seen": 418381824
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008820260782347041,
      "loss": 1.7709,
      "theoretical_loss": 4.020306472948208,
      "tokens_seen": 418447360
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008820060180541625,
      "loss": 1.8081,
      "theoretical_loss": 4.020237063455372,
      "tokens_seen": 418512896
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008819859578736209,
      "loss": 1.765,
      "theoretical_loss": 4.020167667873473,
      "tokens_seen": 418578432
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008819658976930792,
      "loss": 1.73,
      "theoretical_loss": 4.0200982861975465,
      "tokens_seen": 418643968
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008819458375125377,
      "loss": 1.7852,
      "theoretical_loss": 4.020028918422629,
      "tokens_seen": 418709504
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000881925777331996,
      "loss": 1.7194,
      "theoretical_loss": 4.019959564543761,
      "tokens_seen": 418775040
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008819057171514544,
      "loss": 1.686,
      "theoretical_loss": 4.019890224555985,
      "tokens_seen": 418840576
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008818856569709127,
      "loss": 1.6681,
      "theoretical_loss": 4.019820898454345,
      "tokens_seen": 418906112
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008818655967903712,
      "loss": 1.6145,
      "theoretical_loss": 4.019751586233889,
      "tokens_seen": 418971648
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008818455366098295,
      "loss": 1.6892,
      "theoretical_loss": 4.019682287889666,
      "tokens_seen": 419037184
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008818254764292879,
      "loss": 1.5315,
      "theoretical_loss": 4.01961300341673,
      "tokens_seen": 419102720
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008818054162487462,
      "loss": 1.7103,
      "theoretical_loss": 4.019543732810134,
      "tokens_seen": 419168256
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008817853560682046,
      "loss": 1.654,
      "theoretical_loss": 4.0194744760649375,
      "tokens_seen": 419233792
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000881765295887663,
      "loss": 1.6612,
      "theoretical_loss": 4.019405233176199,
      "tokens_seen": 419299328
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008817452357071214,
      "loss": 1.7786,
      "theoretical_loss": 4.019336004138982,
      "tokens_seen": 419364864
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.4930272698402405,
      "objective/train/docs_used": 158751,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7433662414550781,
      "objective/train/original_loss": 1.7433662414550781,
      "objective/train/theoretical_loss": 4.019266788948352,
      "objective/train/tokens_used": 439890400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24367401003837585,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505414009094238,
      "objective/train/weighted_lm_loss": 1.8314003944396973,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.019266788948352,
      "tokens_seen": 419430400
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008817251755265797,
      "loss": 1.6892,
      "theoretical_loss": 4.019266788948352,
      "tokens_seen": 419430400
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008817051153460382,
      "loss": 1.6757,
      "theoretical_loss": 4.0191975875993755,
      "tokens_seen": 419495936
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008816850551654964,
      "loss": 1.8176,
      "theoretical_loss": 4.019128400087123,
      "tokens_seen": 419561472
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008816649949849549,
      "loss": 1.7537,
      "theoretical_loss": 4.019059226406667,
      "tokens_seen": 419627008
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008816449348044132,
      "loss": 1.7643,
      "theoretical_loss": 4.0189900665530836,
      "tokens_seen": 419692544
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008816248746238716,
      "loss": 1.7258,
      "theoretical_loss": 4.018920920521451,
      "tokens_seen": 419758080
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.00088160481444333,
      "loss": 1.7165,
      "theoretical_loss": 4.018851788306847,
      "tokens_seen": 419823616
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008815847542627884,
      "loss": 1.6832,
      "theoretical_loss": 4.018782669904357,
      "tokens_seen": 419889152
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008815646940822467,
      "loss": 1.7228,
      "theoretical_loss": 4.0187135653090635,
      "tokens_seen": 419954688
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008815446339017052,
      "loss": 1.5663,
      "theoretical_loss": 4.018644474516058,
      "tokens_seen": 420020224
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008815245737211634,
      "loss": 1.515,
      "theoretical_loss": 4.0185753975204275,
      "tokens_seen": 420085760
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008815045135406219,
      "loss": 1.6648,
      "theoretical_loss": 4.018506334317267,
      "tokens_seen": 420151296
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008814844533600802,
      "loss": 1.6975,
      "theoretical_loss": 4.018437284901671,
      "tokens_seen": 420216832
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008814643931795386,
      "loss": 1.7339,
      "theoretical_loss": 4.018368249268737,
      "tokens_seen": 420282368
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000881444332998997,
      "loss": 1.7528,
      "theoretical_loss": 4.018299227413566,
      "tokens_seen": 420347904
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008814242728184553,
      "loss": 1.7641,
      "theoretical_loss": 4.018230219331261,
      "tokens_seen": 420413440
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008814042126379137,
      "loss": 1.7406,
      "theoretical_loss": 4.018161225016926,
      "tokens_seen": 420478976
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008813841524573722,
      "loss": 1.5882,
      "theoretical_loss": 4.01809224446567,
      "tokens_seen": 420544512
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008813640922768305,
      "loss": 1.7368,
      "theoretical_loss": 4.018023277672604,
      "tokens_seen": 420610048
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008813440320962889,
      "loss": 1.7926,
      "theoretical_loss": 4.01795432463284,
      "tokens_seen": 420675584
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008813239719157474,
      "loss": 1.7206,
      "theoretical_loss": 4.0178853853414935,
      "tokens_seen": 420741120
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008813039117352056,
      "loss": 1.7873,
      "theoretical_loss": 4.017816459793683,
      "tokens_seen": 420806656
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008812838515546641,
      "loss": 1.6735,
      "theoretical_loss": 4.017747547984528,
      "tokens_seen": 420872192
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008812637913741223,
      "loss": 1.605,
      "theoretical_loss": 4.0176786499091515,
      "tokens_seen": 420937728
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008812437311935808,
      "loss": 1.7147,
      "theoretical_loss": 4.017609765562678,
      "tokens_seen": 421003264
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008812236710130392,
      "loss": 1.7636,
      "theoretical_loss": 4.017540894940238,
      "tokens_seen": 421068800
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008812036108324975,
      "loss": 1.7168,
      "theoretical_loss": 4.017472038036959,
      "tokens_seen": 421134336
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008811835506519559,
      "loss": 1.7088,
      "theoretical_loss": 4.017403194847975,
      "tokens_seen": 421199872
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008811634904714143,
      "loss": 1.6162,
      "theoretical_loss": 4.017334365368422,
      "tokens_seen": 421265408
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008811434302908726,
      "loss": 1.7104,
      "theoretical_loss": 4.017265549593437,
      "tokens_seen": 421330944
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008811233701103311,
      "loss": 1.7413,
      "theoretical_loss": 4.01719674751816,
      "tokens_seen": 421396480
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008811033099297893,
      "loss": 1.7278,
      "theoretical_loss": 4.017127959137733,
      "tokens_seen": 421462016
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008810832497492478,
      "loss": 1.7578,
      "theoretical_loss": 4.017059184447303,
      "tokens_seen": 421527552
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008810631895687061,
      "loss": 1.6661,
      "theoretical_loss": 4.016990423442016,
      "tokens_seen": 421593088
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008810431293881645,
      "loss": 1.7898,
      "theoretical_loss": 4.0169216761170246,
      "tokens_seen": 421658624
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008810230692076229,
      "loss": 1.6991,
      "theoretical_loss": 4.016852942467479,
      "tokens_seen": 421724160
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008810030090270813,
      "loss": 1.6974,
      "theoretical_loss": 4.0167842224885355,
      "tokens_seen": 421789696
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008809829488465396,
      "loss": 1.6438,
      "theoretical_loss": 4.016715516175351,
      "tokens_seen": 421855232
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008809628886659981,
      "loss": 1.6117,
      "theoretical_loss": 4.016646823523087,
      "tokens_seen": 421920768
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008809428284854564,
      "loss": 1.5419,
      "theoretical_loss": 4.016578144526903,
      "tokens_seen": 421986304
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008809227683049148,
      "loss": 1.6463,
      "theoretical_loss": 4.016509479181968,
      "tokens_seen": 422051840
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008809027081243731,
      "loss": 1.739,
      "theoretical_loss": 4.0164408274834456,
      "tokens_seen": 422117376
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008808826479438316,
      "loss": 1.6391,
      "theoretical_loss": 4.016372189426509,
      "tokens_seen": 422182912
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008808625877632899,
      "loss": 1.7629,
      "theoretical_loss": 4.016303565006328,
      "tokens_seen": 422248448
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008808425275827483,
      "loss": 1.5975,
      "theoretical_loss": 4.016234954218078,
      "tokens_seen": 422313984
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008808224674022066,
      "loss": 1.7056,
      "theoretical_loss": 4.016166357056937,
      "tokens_seen": 422379520
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000880802407221665,
      "loss": 1.7482,
      "theoretical_loss": 4.016097773518084,
      "tokens_seen": 422445056
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008807823470411234,
      "loss": 1.6291,
      "theoretical_loss": 4.016029203596702,
      "tokens_seen": 422510592
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008807622868605818,
      "loss": 1.5805,
      "theoretical_loss": 4.015960647287975,
      "tokens_seen": 422576128
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008807422266800401,
      "loss": 1.6762,
      "theoretical_loss": 4.01589210458709,
      "tokens_seen": 422641664
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.49287301301956177,
      "objective/train/docs_used": 159824,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5584911108016968,
      "objective/train/original_loss": 1.5584911108016968,
      "objective/train/theoretical_loss": 4.015823575489237,
      "objective/train/tokens_used": 443167200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24386560916900635,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505269765853882,
      "objective/train/weighted_lm_loss": 1.6371653079986572,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.015823575489237,
      "tokens_seen": 422707200
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008807221664994986,
      "loss": 1.6797,
      "theoretical_loss": 4.015823575489237,
      "tokens_seen": 422707200
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008807021063189568,
      "loss": 1.768,
      "theoretical_loss": 4.0157550599896075,
      "tokens_seen": 422772736
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008806820461384153,
      "loss": 1.7092,
      "theoretical_loss": 4.015686558083396,
      "tokens_seen": 422838272
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008806619859578736,
      "loss": 1.7596,
      "theoretical_loss": 4.015618069765799,
      "tokens_seen": 422903808
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000880641925777332,
      "loss": 1.6727,
      "theoretical_loss": 4.015549595032017,
      "tokens_seen": 422969344
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008806218655967904,
      "loss": 1.6833,
      "theoretical_loss": 4.015481133877251,
      "tokens_seen": 423034880
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008806018054162488,
      "loss": 1.7244,
      "theoretical_loss": 4.015412686296704,
      "tokens_seen": 423100416
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008805817452357071,
      "loss": 1.6397,
      "theoretical_loss": 4.015344252285585,
      "tokens_seen": 423165952
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008805616850551656,
      "loss": 1.665,
      "theoretical_loss": 4.0152758318391015,
      "tokens_seen": 423231488
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008805416248746238,
      "loss": 1.8013,
      "theoretical_loss": 4.0152074249524645,
      "tokens_seen": 423297024
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008805215646940823,
      "loss": 1.6566,
      "theoretical_loss": 4.0151390316208895,
      "tokens_seen": 423362560
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008805015045135407,
      "loss": 1.7786,
      "theoretical_loss": 4.015070651839592,
      "tokens_seen": 423428096
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000880481444332999,
      "loss": 1.5864,
      "theoretical_loss": 4.01500228560379,
      "tokens_seen": 423493632
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008804613841524574,
      "loss": 1.7001,
      "theoretical_loss": 4.014933932908706,
      "tokens_seen": 423559168
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008804413239719157,
      "loss": 1.7395,
      "theoretical_loss": 4.014865593749563,
      "tokens_seen": 423624704
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008804212637913741,
      "loss": 1.7334,
      "theoretical_loss": 4.014797268121587,
      "tokens_seen": 423690240
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008804012036108326,
      "loss": 1.6614,
      "theoretical_loss": 4.014728956020006,
      "tokens_seen": 423755776
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008803811434302908,
      "loss": 1.6281,
      "theoretical_loss": 4.014660657440052,
      "tokens_seen": 423821312
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008803610832497493,
      "loss": 1.6153,
      "theoretical_loss": 4.014592372376958,
      "tokens_seen": 423886848
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008803410230692077,
      "loss": 1.5827,
      "theoretical_loss": 4.014524100825959,
      "tokens_seen": 423952384
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000880320962888666,
      "loss": 1.7422,
      "theoretical_loss": 4.014455842782294,
      "tokens_seen": 424017920
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008803009027081244,
      "loss": 1.787,
      "theoretical_loss": 4.0143875982412025,
      "tokens_seen": 424083456
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008802808425275827,
      "loss": 1.6612,
      "theoretical_loss": 4.014319367197928,
      "tokens_seen": 424148992
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008802607823470411,
      "loss": 1.6953,
      "theoretical_loss": 4.014251149647716,
      "tokens_seen": 424214528
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008802407221664996,
      "loss": 1.7357,
      "theoretical_loss": 4.014182945585815,
      "tokens_seen": 424280064
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008802206619859578,
      "loss": 1.7909,
      "theoretical_loss": 4.014114755007474,
      "tokens_seen": 424345600
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008802006018054163,
      "loss": 1.8349,
      "theoretical_loss": 4.014046577907946,
      "tokens_seen": 424411136
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008801805416248746,
      "loss": 1.7697,
      "theoretical_loss": 4.0139784142824855,
      "tokens_seen": 424476672
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000880160481444333,
      "loss": 1.7789,
      "theoretical_loss": 4.01391026412635,
      "tokens_seen": 424542208
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008801404212637914,
      "loss": 1.7296,
      "theoretical_loss": 4.013842127434801,
      "tokens_seen": 424607744
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008801203610832497,
      "loss": 1.6642,
      "theoretical_loss": 4.013774004203099,
      "tokens_seen": 424673280
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008801003009027081,
      "loss": 1.7023,
      "theoretical_loss": 4.013705894426509,
      "tokens_seen": 424738816
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008800802407221665,
      "loss": 1.6199,
      "theoretical_loss": 4.013637798100298,
      "tokens_seen": 424804352
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008800601805416248,
      "loss": 1.7503,
      "theoretical_loss": 4.0135697152197345,
      "tokens_seen": 424869888
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008800401203610833,
      "loss": 1.7619,
      "theoretical_loss": 4.013501645780092,
      "tokens_seen": 424935424
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008800200601805416,
      "loss": 1.6706,
      "theoretical_loss": 4.0134335897766436,
      "tokens_seen": 425000960
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.00088,
      "loss": 1.5715,
      "theoretical_loss": 4.013365547204666,
      "tokens_seen": 425066496
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008799799398194584,
      "loss": 1.6421,
      "theoretical_loss": 4.013297518059439,
      "tokens_seen": 425132032
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008799598796389168,
      "loss": 1.6172,
      "theoretical_loss": 4.013229502336242,
      "tokens_seen": 425197568
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008799398194583751,
      "loss": 1.5202,
      "theoretical_loss": 4.01316150003036,
      "tokens_seen": 425263104
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008799197592778335,
      "loss": 1.7793,
      "theoretical_loss": 4.013093511137079,
      "tokens_seen": 425328640
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008798996990972918,
      "loss": 1.8113,
      "theoretical_loss": 4.013025535651688,
      "tokens_seen": 425394176
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008798796389167503,
      "loss": 1.6631,
      "theoretical_loss": 4.0129575735694765,
      "tokens_seen": 425459712
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008798595787362087,
      "loss": 1.7407,
      "theoretical_loss": 4.012889624885739,
      "tokens_seen": 425525248
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000879839518555667,
      "loss": 1.7129,
      "theoretical_loss": 4.012821689595771,
      "tokens_seen": 425590784
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008798194583751254,
      "loss": 1.8266,
      "theoretical_loss": 4.01275376769487,
      "tokens_seen": 425656320
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008797993981945838,
      "loss": 1.692,
      "theoretical_loss": 4.012685859178337,
      "tokens_seen": 425721856
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008797793380140422,
      "loss": 1.6876,
      "theoretical_loss": 4.012617964041474,
      "tokens_seen": 425787392
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008797592778335005,
      "loss": 1.6699,
      "theoretical_loss": 4.012550082279587,
      "tokens_seen": 425852928
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000879739217652959,
      "loss": 1.544,
      "theoretical_loss": 4.012482213887981,
      "tokens_seen": 425918464
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.4950158894062042,
      "objective/train/docs_used": 161088,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.462551474571228,
      "objective/train/original_loss": 1.4625513553619385,
      "objective/train/theoretical_loss": 4.012414358861969,
      "objective/train/tokens_used": 446444000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24526995420455933,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050748586654663,
      "objective/train/weighted_lm_loss": 1.5366774797439575,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.012414358861969,
      "tokens_seen": 425984000
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008797191574724172,
      "loss": 1.6405,
      "theoretical_loss": 4.012414358861969,
      "tokens_seen": 425984000
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008796990972918757,
      "loss": 1.7246,
      "theoretical_loss": 4.012346517196862,
      "tokens_seen": 426049536
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000879679037111334,
      "loss": 1.5951,
      "theoretical_loss": 4.012278688887975,
      "tokens_seen": 426115072
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008796589769307924,
      "loss": 1.7454,
      "theoretical_loss": 4.012210873930625,
      "tokens_seen": 426180608
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008796389167502508,
      "loss": 1.7153,
      "theoretical_loss": 4.01214307232013,
      "tokens_seen": 426246144
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008796188565697092,
      "loss": 1.6092,
      "theoretical_loss": 4.012075284051813,
      "tokens_seen": 426311680
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008795987963891675,
      "loss": 1.7099,
      "theoretical_loss": 4.012007509120997,
      "tokens_seen": 426377216
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000879578736208626,
      "loss": 1.7302,
      "theoretical_loss": 4.011939747523009,
      "tokens_seen": 426442752
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008795586760280842,
      "loss": 1.6965,
      "theoretical_loss": 4.011871999253178,
      "tokens_seen": 426508288
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008795386158475427,
      "loss": 1.8017,
      "theoretical_loss": 4.011804264306835,
      "tokens_seen": 426573824
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000879518555667001,
      "loss": 1.7581,
      "theoretical_loss": 4.011736542679313,
      "tokens_seen": 426639360
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008794984954864594,
      "loss": 1.7327,
      "theoretical_loss": 4.011668834365948,
      "tokens_seen": 426704896
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008794784353059178,
      "loss": 1.6182,
      "theoretical_loss": 4.011601139362078,
      "tokens_seen": 426770432
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008794583751253761,
      "loss": 1.691,
      "theoretical_loss": 4.011533457663044,
      "tokens_seen": 426835968
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008794383149448345,
      "loss": 1.6054,
      "theoretical_loss": 4.0114657892641885,
      "tokens_seen": 426901504
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000879418254764293,
      "loss": 1.698,
      "theoretical_loss": 4.011398134160857,
      "tokens_seen": 426967040
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008793981945837512,
      "loss": 1.6539,
      "theoretical_loss": 4.011330492348397,
      "tokens_seen": 427032576
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008793781344032097,
      "loss": 1.7655,
      "theoretical_loss": 4.011262863822157,
      "tokens_seen": 427098112
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008793580742226681,
      "loss": 1.6945,
      "theoretical_loss": 4.011195248577492,
      "tokens_seen": 427163648
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008793380140421264,
      "loss": 1.6915,
      "theoretical_loss": 4.011127646609753,
      "tokens_seen": 427229184
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008793179538615848,
      "loss": 1.8055,
      "theoretical_loss": 4.0110600579143,
      "tokens_seen": 427294720
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008792978936810431,
      "loss": 1.6929,
      "theoretical_loss": 4.0109924824864915,
      "tokens_seen": 427360256
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008792778335005015,
      "loss": 1.72,
      "theoretical_loss": 4.0109249203216875,
      "tokens_seen": 427425792
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.00087925777331996,
      "loss": 1.7089,
      "theoretical_loss": 4.010857371415254,
      "tokens_seen": 427491328
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008792377131394182,
      "loss": 1.6548,
      "theoretical_loss": 4.010789835762555,
      "tokens_seen": 427556864
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008792176529588767,
      "loss": 1.7562,
      "theoretical_loss": 4.010722313358961,
      "tokens_seen": 427622400
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000879197592778335,
      "loss": 1.7734,
      "theoretical_loss": 4.010654804199842,
      "tokens_seen": 427687936
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008791775325977934,
      "loss": 1.664,
      "theoretical_loss": 4.010587308280572,
      "tokens_seen": 427753472
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008791574724172518,
      "loss": 1.7622,
      "theoretical_loss": 4.010519825596525,
      "tokens_seen": 427819008
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008791374122367101,
      "loss": 1.7167,
      "theoretical_loss": 4.01045235614308,
      "tokens_seen": 427884544
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008791173520561685,
      "loss": 1.684,
      "theoretical_loss": 4.010384899915618,
      "tokens_seen": 427950080
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008790972918756269,
      "loss": 1.6483,
      "theoretical_loss": 4.010317456909519,
      "tokens_seen": 428015616
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008790772316950852,
      "loss": 1.6849,
      "theoretical_loss": 4.010250027120169,
      "tokens_seen": 428081152
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008790571715145437,
      "loss": 1.557,
      "theoretical_loss": 4.010182610542957,
      "tokens_seen": 428146688
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000879037111334002,
      "loss": 1.6306,
      "theoretical_loss": 4.0101152071732695,
      "tokens_seen": 428212224
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008790170511534604,
      "loss": 1.7604,
      "theoretical_loss": 4.0100478170065,
      "tokens_seen": 428277760
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008789969909729188,
      "loss": 1.6982,
      "theoretical_loss": 4.009980440038043,
      "tokens_seen": 428343296
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008789769307923772,
      "loss": 1.7026,
      "theoretical_loss": 4.0099130762632935,
      "tokens_seen": 428408832
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008789568706118355,
      "loss": 1.749,
      "theoretical_loss": 4.00984572567765,
      "tokens_seen": 428474368
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008789368104312939,
      "loss": 1.8499,
      "theoretical_loss": 4.009778388276516,
      "tokens_seen": 428539904
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008789167502507522,
      "loss": 1.6362,
      "theoretical_loss": 4.009711064055291,
      "tokens_seen": 428605440
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008788966900702107,
      "loss": 1.6866,
      "theoretical_loss": 4.009643753009383,
      "tokens_seen": 428670976
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000878876629889669,
      "loss": 1.5747,
      "theoretical_loss": 4.0095764551342015,
      "tokens_seen": 428736512
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008788565697091274,
      "loss": 1.6387,
      "theoretical_loss": 4.0095091704251535,
      "tokens_seen": 428802048
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008788365095285857,
      "loss": 1.7684,
      "theoretical_loss": 4.009441898877652,
      "tokens_seen": 428867584
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008788164493480442,
      "loss": 1.6421,
      "theoretical_loss": 4.009374640487114,
      "tokens_seen": 428933120
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008787963891675025,
      "loss": 1.6364,
      "theoretical_loss": 4.009307395248955,
      "tokens_seen": 428998656
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008787763289869609,
      "loss": 1.6676,
      "theoretical_loss": 4.009240163158595,
      "tokens_seen": 429064192
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008787562688064192,
      "loss": 1.73,
      "theoretical_loss": 4.009172944211455,
      "tokens_seen": 429129728
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008787362086258776,
      "loss": 1.597,
      "theoretical_loss": 4.00910573840296,
      "tokens_seen": 429195264
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.49417299032211304,
      "objective/train/docs_used": 162190,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6557865142822266,
      "objective/train/original_loss": 1.6557865142822266,
      "objective/train/theoretical_loss": 4.009038545728536,
      "objective/train/tokens_used": 449720800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24450218677520752,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506603717803955,
      "objective/train/weighted_lm_loss": 1.7396289110183716,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.009038545728536,
      "tokens_seen": 429260800
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000878716148445336,
      "loss": 1.7554,
      "theoretical_loss": 4.009038545728536,
      "tokens_seen": 429260800
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008786960882647944,
      "loss": 1.6362,
      "theoretical_loss": 4.00897136618361,
      "tokens_seen": 429326336
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008786760280842527,
      "loss": 1.8445,
      "theoretical_loss": 4.008904199763615,
      "tokens_seen": 429391872
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008786559679037112,
      "loss": 1.7453,
      "theoretical_loss": 4.0088370464639835,
      "tokens_seen": 429457408
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008786359077231695,
      "loss": 1.5949,
      "theoretical_loss": 4.0087699062801505,
      "tokens_seen": 429522944
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008786158475426279,
      "loss": 1.6461,
      "theoretical_loss": 4.008702779207555,
      "tokens_seen": 429588480
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008785957873620863,
      "loss": 1.7022,
      "theoretical_loss": 4.008635665241635,
      "tokens_seen": 429654016
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008785757271815446,
      "loss": 1.7671,
      "theoretical_loss": 4.008568564377834,
      "tokens_seen": 429719552
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000878555667001003,
      "loss": 1.6948,
      "theoretical_loss": 4.008501476611597,
      "tokens_seen": 429785088
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008785356068204615,
      "loss": 1.7342,
      "theoretical_loss": 4.00843440193837,
      "tokens_seen": 429850624
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008785155466399197,
      "loss": 1.7383,
      "theoretical_loss": 4.008367340353602,
      "tokens_seen": 429916160
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008784954864593782,
      "loss": 1.5975,
      "theoretical_loss": 4.008300291852745,
      "tokens_seen": 429981696
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008784754262788364,
      "loss": 1.7106,
      "theoretical_loss": 4.008233256431253,
      "tokens_seen": 430047232
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008784553660982949,
      "loss": 1.7632,
      "theoretical_loss": 4.00816623408458,
      "tokens_seen": 430112768
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008784353059177533,
      "loss": 1.7036,
      "theoretical_loss": 4.008099224808188,
      "tokens_seen": 430178304
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008784152457372116,
      "loss": 1.58,
      "theoretical_loss": 4.008032228597534,
      "tokens_seen": 430243840
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.00087839518555667,
      "loss": 1.7657,
      "theoretical_loss": 4.007965245448082,
      "tokens_seen": 430309376
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008783751253761285,
      "loss": 1.5189,
      "theoretical_loss": 4.007898275355297,
      "tokens_seen": 430374912
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008783550651955868,
      "loss": 1.6877,
      "theoretical_loss": 4.007831318314645,
      "tokens_seen": 430440448
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008783350050150452,
      "loss": 1.6408,
      "theoretical_loss": 4.007764374321599,
      "tokens_seen": 430505984
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008783149448345035,
      "loss": 1.604,
      "theoretical_loss": 4.007697443371628,
      "tokens_seen": 430571520
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008782948846539619,
      "loss": 1.7851,
      "theoretical_loss": 4.007630525460206,
      "tokens_seen": 430637056
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008782748244734204,
      "loss": 1.5834,
      "theoretical_loss": 4.00756362058281,
      "tokens_seen": 430702592
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008782547642928786,
      "loss": 1.7205,
      "theoretical_loss": 4.007496728734919,
      "tokens_seen": 430768128
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008782347041123371,
      "loss": 1.6197,
      "theoretical_loss": 4.007429849912014,
      "tokens_seen": 430833664
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008782146439317954,
      "loss": 1.5597,
      "theoretical_loss": 4.007362984109577,
      "tokens_seen": 430899200
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008781945837512538,
      "loss": 1.6667,
      "theoretical_loss": 4.007296131323094,
      "tokens_seen": 430964736
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008781745235707122,
      "loss": 1.657,
      "theoretical_loss": 4.007229291548052,
      "tokens_seen": 431030272
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008781544633901706,
      "loss": 1.6742,
      "theoretical_loss": 4.007162464779943,
      "tokens_seen": 431095808
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008781344032096289,
      "loss": 1.7575,
      "theoretical_loss": 4.0070956510142555,
      "tokens_seen": 431161344
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008781143430290873,
      "loss": 1.6987,
      "theoretical_loss": 4.007028850246487,
      "tokens_seen": 431226880
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008780942828485456,
      "loss": 1.8072,
      "theoretical_loss": 4.006962062472133,
      "tokens_seen": 431292416
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008780742226680041,
      "loss": 1.6421,
      "theoretical_loss": 4.006895287686692,
      "tokens_seen": 431357952
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008780541624874624,
      "loss": 1.6763,
      "theoretical_loss": 4.006828525885666,
      "tokens_seen": 431423488
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008780341023069208,
      "loss": 1.6193,
      "theoretical_loss": 4.006761777064557,
      "tokens_seen": 431489024
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008780140421263792,
      "loss": 1.6439,
      "theoretical_loss": 4.0066950412188715,
      "tokens_seen": 431554560
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008779939819458376,
      "loss": 1.6766,
      "theoretical_loss": 4.006628318344117,
      "tokens_seen": 431620096
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008779739217652959,
      "loss": 1.6931,
      "theoretical_loss": 4.0065616084358044,
      "tokens_seen": 431685632
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008779538615847543,
      "loss": 1.6771,
      "theoretical_loss": 4.006494911489444,
      "tokens_seen": 431751168
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008779338014042126,
      "loss": 1.6814,
      "theoretical_loss": 4.006428227500552,
      "tokens_seen": 431816704
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008779137412236711,
      "loss": 1.6386,
      "theoretical_loss": 4.006361556464646,
      "tokens_seen": 431882240
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008778936810431294,
      "loss": 1.7042,
      "theoretical_loss": 4.006294898377242,
      "tokens_seen": 431947776
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008778736208625878,
      "loss": 1.735,
      "theoretical_loss": 4.006228253233864,
      "tokens_seen": 432013312
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008778535606820461,
      "loss": 1.8019,
      "theoretical_loss": 4.006161621030033,
      "tokens_seen": 432078848
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008778335005015046,
      "loss": 1.6797,
      "theoretical_loss": 4.0060950017612775,
      "tokens_seen": 432144384
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008778134403209629,
      "loss": 1.6797,
      "theoretical_loss": 4.006028395423122,
      "tokens_seen": 432209920
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008777933801404213,
      "loss": 1.7321,
      "theoretical_loss": 4.0059618020111,
      "tokens_seen": 432275456
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008777733199598796,
      "loss": 1.6531,
      "theoretical_loss": 4.005895221520742,
      "tokens_seen": 432340992
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000877753259779338,
      "loss": 1.7324,
      "theoretical_loss": 4.005828653947583,
      "tokens_seen": 432406528
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008777331995987964,
      "loss": 1.6861,
      "theoretical_loss": 4.00576209928716,
      "tokens_seen": 432472064
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.4946593642234802,
      "objective/train/docs_used": 163281,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5753886699676514,
      "objective/train/original_loss": 1.5753885507583618,
      "objective/train/theoretical_loss": 4.00569555753501,
      "objective/train/tokens_used": 452997600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2450278401374817,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507116317749023,
      "objective/train/weighted_lm_loss": 1.6551384925842285,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.00569555753501,
      "tokens_seen": 432537600
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008777131394182548,
      "loss": 1.7133,
      "theoretical_loss": 4.00569555753501,
      "tokens_seen": 432537600
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008776930792377131,
      "loss": 1.7018,
      "theoretical_loss": 4.005629028686678,
      "tokens_seen": 432603136
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008776730190571716,
      "loss": 1.7188,
      "theoretical_loss": 4.005562512737704,
      "tokens_seen": 432668672
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008776529588766299,
      "loss": 1.6953,
      "theoretical_loss": 4.005496009683635,
      "tokens_seen": 432734208
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008776328986960883,
      "loss": 1.5622,
      "theoretical_loss": 4.0054295195200185,
      "tokens_seen": 432799744
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008776128385155467,
      "loss": 1.6209,
      "theoretical_loss": 4.005363042242405,
      "tokens_seen": 432865280
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000877592778335005,
      "loss": 1.7354,
      "theoretical_loss": 4.005296577846347,
      "tokens_seen": 432930816
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008775727181544634,
      "loss": 1.6755,
      "theoretical_loss": 4.005230126327398,
      "tokens_seen": 432996352
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008775526579739219,
      "loss": 1.6384,
      "theoretical_loss": 4.005163687681116,
      "tokens_seen": 433061888
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008775325977933801,
      "loss": 1.6189,
      "theoretical_loss": 4.0050972619030585,
      "tokens_seen": 433127424
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008775125376128386,
      "loss": 1.7411,
      "theoretical_loss": 4.005030848988788,
      "tokens_seen": 433192960
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008774924774322968,
      "loss": 1.6849,
      "theoretical_loss": 4.004964448933866,
      "tokens_seen": 433258496
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008774724172517553,
      "loss": 1.6512,
      "theoretical_loss": 4.00489806173386,
      "tokens_seen": 433324032
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008774523570712137,
      "loss": 1.7051,
      "theoretical_loss": 4.004831687384336,
      "tokens_seen": 433389568
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000877432296890672,
      "loss": 1.7895,
      "theoretical_loss": 4.004765325880865,
      "tokens_seen": 433455104
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008774122367101304,
      "loss": 1.8067,
      "theoretical_loss": 4.00469897721902,
      "tokens_seen": 433520640
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008773921765295889,
      "loss": 1.7438,
      "theoretical_loss": 4.004632641394372,
      "tokens_seen": 433586176
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008773721163490471,
      "loss": 1.8116,
      "theoretical_loss": 4.0045663184025,
      "tokens_seen": 433651712
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008773520561685056,
      "loss": 1.6301,
      "theoretical_loss": 4.004500008238983,
      "tokens_seen": 433717248
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008773319959879638,
      "loss": 1.6341,
      "theoretical_loss": 4.0044337108994,
      "tokens_seen": 433782784
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008773119358074223,
      "loss": 1.7843,
      "theoretical_loss": 4.0043674263793365,
      "tokens_seen": 433848320
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008772918756268807,
      "loss": 1.584,
      "theoretical_loss": 4.004301154674375,
      "tokens_seen": 433913856
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000877271815446339,
      "loss": 1.6833,
      "theoretical_loss": 4.0042348957801055,
      "tokens_seen": 433979392
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008772517552657974,
      "loss": 1.6795,
      "theoretical_loss": 4.004168649692115,
      "tokens_seen": 434044928
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008772316950852558,
      "loss": 1.6922,
      "theoretical_loss": 4.004102416405998,
      "tokens_seen": 434110464
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008772116349047141,
      "loss": 1.687,
      "theoretical_loss": 4.004036195917347,
      "tokens_seen": 434176000
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008771915747241726,
      "loss": 1.6465,
      "theoretical_loss": 4.003969988221758,
      "tokens_seen": 434241536
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008771715145436308,
      "loss": 1.7862,
      "theoretical_loss": 4.00390379331483,
      "tokens_seen": 434307072
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008771514543630893,
      "loss": 1.5721,
      "theoretical_loss": 4.0038376111921625,
      "tokens_seen": 434372608
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008771313941825476,
      "loss": 1.746,
      "theoretical_loss": 4.00377144184936,
      "tokens_seen": 434438144
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000877111334002006,
      "loss": 1.5794,
      "theoretical_loss": 4.003705285282026,
      "tokens_seen": 434503680
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008770912738214644,
      "loss": 1.8313,
      "theoretical_loss": 4.003639141485767,
      "tokens_seen": 434569216
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008770712136409228,
      "loss": 1.7165,
      "theoretical_loss": 4.0035730104561935,
      "tokens_seen": 434634752
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008770511534603811,
      "loss": 1.752,
      "theoretical_loss": 4.003506892188917,
      "tokens_seen": 434700288
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008770310932798396,
      "loss": 1.7615,
      "theoretical_loss": 4.00344078667955,
      "tokens_seen": 434765824
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008770110330992978,
      "loss": 1.7861,
      "theoretical_loss": 4.00337469392371,
      "tokens_seen": 434831360
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008769909729187563,
      "loss": 1.7035,
      "theoretical_loss": 4.003308613917012,
      "tokens_seen": 434896896
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008769709127382146,
      "loss": 1.713,
      "theoretical_loss": 4.003242546655079,
      "tokens_seen": 434962432
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000876950852557673,
      "loss": 1.7261,
      "theoretical_loss": 4.003176492133531,
      "tokens_seen": 435027968
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008769307923771314,
      "loss": 1.8141,
      "theoretical_loss": 4.003110450347994,
      "tokens_seen": 435093504
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008769107321965898,
      "loss": 1.6346,
      "theoretical_loss": 4.003044421294094,
      "tokens_seen": 435159040
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008768906720160481,
      "loss": 1.7725,
      "theoretical_loss": 4.002978404967459,
      "tokens_seen": 435224576
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008768706118355065,
      "loss": 1.5136,
      "theoretical_loss": 4.002912401363721,
      "tokens_seen": 435290112
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000876850551654965,
      "loss": 1.6824,
      "theoretical_loss": 4.0028464104785115,
      "tokens_seen": 435355648
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008768304914744233,
      "loss": 1.7163,
      "theoretical_loss": 4.002780432307468,
      "tokens_seen": 435421184
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008768104312938817,
      "loss": 1.795,
      "theoretical_loss": 4.002714466846227,
      "tokens_seen": 435486720
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.00087679037111334,
      "loss": 1.5681,
      "theoretical_loss": 4.002648514090426,
      "tokens_seen": 435552256
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008767703109327984,
      "loss": 1.6463,
      "theoretical_loss": 4.002582574035709,
      "tokens_seen": 435617792
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008767502507522568,
      "loss": 1.702,
      "theoretical_loss": 4.0025166466777184,
      "tokens_seen": 435683328
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008767301905717152,
      "loss": 1.7896,
      "theoretical_loss": 4.002450732012102,
      "tokens_seen": 435748864
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.49324312806129456,
      "objective/train/docs_used": 164592,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.526324987411499,
      "objective/train/original_loss": 1.526324987411499,
      "objective/train/theoretical_loss": 4.002384830034506,
      "objective/train/tokens_used": 456274400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24383056163787842,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505638122558594,
      "objective/train/weighted_lm_loss": 1.6033390760421753,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 4.002384830034506,
      "tokens_seen": 435814400
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008767101303911735,
      "loss": 1.7303,
      "theoretical_loss": 4.002384830034506,
      "tokens_seen": 435814400
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000876690070210632,
      "loss": 1.6407,
      "theoretical_loss": 4.002318940740581,
      "tokens_seen": 435879936
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008766700100300903,
      "loss": 1.6781,
      "theoretical_loss": 4.00225306412598,
      "tokens_seen": 435945472
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008766499498495487,
      "loss": 1.7347,
      "theoretical_loss": 4.002187200186357,
      "tokens_seen": 436011008
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008766298896690071,
      "loss": 1.776,
      "theoretical_loss": 4.002121348917369,
      "tokens_seen": 436076544
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008766098294884654,
      "loss": 1.6167,
      "theoretical_loss": 4.002055510314674,
      "tokens_seen": 436142080
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008765897693079238,
      "loss": 1.7234,
      "theoretical_loss": 4.001989684373934,
      "tokens_seen": 436207616
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008765697091273823,
      "loss": 1.6475,
      "theoretical_loss": 4.0019238710908125,
      "tokens_seen": 436273152
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008765496489468405,
      "loss": 1.759,
      "theoretical_loss": 4.001858070460973,
      "tokens_seen": 436338688
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000876529588766299,
      "loss": 1.6858,
      "theoretical_loss": 4.001792282480085,
      "tokens_seen": 436404224
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008765095285857572,
      "loss": 1.6897,
      "theoretical_loss": 4.0017265071438155,
      "tokens_seen": 436469760
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008764894684052157,
      "loss": 1.7926,
      "theoretical_loss": 4.001660744447838,
      "tokens_seen": 436535296
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008764694082246741,
      "loss": 1.5353,
      "theoretical_loss": 4.001594994387825,
      "tokens_seen": 436600832
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008764493480441324,
      "loss": 1.6745,
      "theoretical_loss": 4.001529256959454,
      "tokens_seen": 436666368
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008764292878635908,
      "loss": 1.744,
      "theoretical_loss": 4.001463532158402,
      "tokens_seen": 436731904
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008764092276830493,
      "loss": 1.7642,
      "theoretical_loss": 4.001397819980349,
      "tokens_seen": 436797440
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008763891675025075,
      "loss": 1.807,
      "theoretical_loss": 4.001332120420978,
      "tokens_seen": 436862976
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000876369107321966,
      "loss": 1.6098,
      "theoretical_loss": 4.001266433475973,
      "tokens_seen": 436928512
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008763490471414242,
      "loss": 1.616,
      "theoretical_loss": 4.001200759141019,
      "tokens_seen": 436994048
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008763289869608827,
      "loss": 1.7438,
      "theoretical_loss": 4.001135097411807,
      "tokens_seen": 437059584
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008763089267803411,
      "loss": 1.7516,
      "theoretical_loss": 4.001069448284027,
      "tokens_seen": 437125120
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008762888665997994,
      "loss": 1.6991,
      "theoretical_loss": 4.001003811753371,
      "tokens_seen": 437190656
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008762688064192578,
      "loss": 1.551,
      "theoretical_loss": 4.000938187815535,
      "tokens_seen": 437256192
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008762487462387162,
      "loss": 1.6185,
      "theoretical_loss": 4.000872576466216,
      "tokens_seen": 437321728
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008762286860581745,
      "loss": 1.6573,
      "theoretical_loss": 4.000806977701113,
      "tokens_seen": 437387264
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000876208625877633,
      "loss": 1.7303,
      "theoretical_loss": 4.000741391515927,
      "tokens_seen": 437452800
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008761885656970912,
      "loss": 1.6935,
      "theoretical_loss": 4.000675817906362,
      "tokens_seen": 437518336
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008761685055165497,
      "loss": 1.5171,
      "theoretical_loss": 4.000610256868123,
      "tokens_seen": 437583872
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000876148445336008,
      "loss": 1.8138,
      "theoretical_loss": 4.000544708396919,
      "tokens_seen": 437649408
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008761283851554664,
      "loss": 1.6239,
      "theoretical_loss": 4.000479172488458,
      "tokens_seen": 437714944
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008761083249749248,
      "loss": 1.7383,
      "theoretical_loss": 4.000413649138453,
      "tokens_seen": 437780480
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008760882647943832,
      "loss": 1.7627,
      "theoretical_loss": 4.000348138342618,
      "tokens_seen": 437846016
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008760682046138415,
      "loss": 1.7317,
      "theoretical_loss": 4.0002826400966685,
      "tokens_seen": 437911552
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008760481444333,
      "loss": 1.6734,
      "theoretical_loss": 4.0002171543963225,
      "tokens_seen": 437977088
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008760280842527582,
      "loss": 1.6857,
      "theoretical_loss": 4.000151681237301,
      "tokens_seen": 438042624
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008760080240722167,
      "loss": 1.5352,
      "theoretical_loss": 4.0000862206153265,
      "tokens_seen": 438108160
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000875987963891675,
      "loss": 1.6934,
      "theoretical_loss": 4.000020772526123,
      "tokens_seen": 438173696
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008759679037111334,
      "loss": 1.7256,
      "theoretical_loss": 3.999955336965418,
      "tokens_seen": 438239232
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008759478435305918,
      "loss": 1.718,
      "theoretical_loss": 3.9998899139289392,
      "tokens_seen": 438304768
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008759277833500502,
      "loss": 1.6726,
      "theoretical_loss": 3.999824503412417,
      "tokens_seen": 438370304
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008759077231695085,
      "loss": 1.5715,
      "theoretical_loss": 3.9997591054115853,
      "tokens_seen": 438435840
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008758876629889669,
      "loss": 1.6995,
      "theoretical_loss": 3.9996937199221785,
      "tokens_seen": 438501376
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008758676028084253,
      "loss": 1.7369,
      "theoretical_loss": 3.999628346939934,
      "tokens_seen": 438566912
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008758475426278837,
      "loss": 1.7391,
      "theoretical_loss": 3.999562986460591,
      "tokens_seen": 438632448
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000875827482447342,
      "loss": 1.7917,
      "theoretical_loss": 3.9994976384798906,
      "tokens_seen": 438697984
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008758074222668005,
      "loss": 1.6776,
      "theoretical_loss": 3.9994323029935757,
      "tokens_seen": 438763520
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008757873620862587,
      "loss": 1.7134,
      "theoretical_loss": 3.9993669799973928,
      "tokens_seen": 438829056
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008757673019057172,
      "loss": 1.6238,
      "theoretical_loss": 3.999301669487088,
      "tokens_seen": 438894592
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008757472417251755,
      "loss": 1.7006,
      "theoretical_loss": 3.9992363714584123,
      "tokens_seen": 438960128
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008757271815446339,
      "loss": 1.6622,
      "theoretical_loss": 3.9991710859071157,
      "tokens_seen": 439025664
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.49032777547836304,
      "objective/train/docs_used": 165818,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6073089838027954,
      "objective/train/original_loss": 1.6073087453842163,
      "objective/train/theoretical_loss": 3.9991058128289536,
      "objective/train/tokens_used": 459551200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24182981252670288,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502620935440063,
      "objective/train/weighted_lm_loss": 1.688266634941101,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9991058128289536,
      "tokens_seen": 439091200
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008757071213640923,
      "loss": 1.6988,
      "theoretical_loss": 3.9991058128289536,
      "tokens_seen": 439091200
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008756870611835507,
      "loss": 1.7678,
      "theoretical_loss": 3.9990405522196806,
      "tokens_seen": 439156736
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000875667001003009,
      "loss": 1.7156,
      "theoretical_loss": 3.9989753040750555,
      "tokens_seen": 439222272
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008756469408224675,
      "loss": 1.8251,
      "theoretical_loss": 3.9989100683908374,
      "tokens_seen": 439287808
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008756268806419257,
      "loss": 1.7089,
      "theoretical_loss": 3.998844845162789,
      "tokens_seen": 439353344
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008756068204613842,
      "loss": 1.7908,
      "theoretical_loss": 3.9987796343866746,
      "tokens_seen": 439418880
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008755867602808425,
      "loss": 1.6618,
      "theoretical_loss": 3.9987144360582594,
      "tokens_seen": 439484416
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008755667001003009,
      "loss": 1.6314,
      "theoretical_loss": 3.9986492501733126,
      "tokens_seen": 439549952
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008755466399197593,
      "loss": 1.7443,
      "theoretical_loss": 3.998584076727604,
      "tokens_seen": 439615488
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008755265797392176,
      "loss": 1.8115,
      "theoretical_loss": 3.9985189157169065,
      "tokens_seen": 439681024
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000875506519558676,
      "loss": 1.7254,
      "theoretical_loss": 3.998453767136994,
      "tokens_seen": 439746560
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008754864593781345,
      "loss": 1.6744,
      "theoretical_loss": 3.9983886309836434,
      "tokens_seen": 439812096
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008754663991975927,
      "loss": 1.7076,
      "theoretical_loss": 3.998323507252633,
      "tokens_seen": 439877632
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008754463390170512,
      "loss": 1.6958,
      "theoretical_loss": 3.998258395939744,
      "tokens_seen": 439943168
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008754262788365094,
      "loss": 1.6202,
      "theoretical_loss": 3.998193297040758,
      "tokens_seen": 440008704
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008754062186559679,
      "loss": 1.673,
      "theoretical_loss": 3.998128210551461,
      "tokens_seen": 440074240
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008753861584754263,
      "loss": 1.6711,
      "theoretical_loss": 3.998063136467639,
      "tokens_seen": 440139776
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008753660982948846,
      "loss": 1.7422,
      "theoretical_loss": 3.997998074785082,
      "tokens_seen": 440205312
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008753460381143431,
      "loss": 1.7045,
      "theoretical_loss": 3.99793302549958,
      "tokens_seen": 440270848
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008753259779338015,
      "loss": 1.6724,
      "theoretical_loss": 3.997867988606926,
      "tokens_seen": 440336384
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008753059177532598,
      "loss": 1.7573,
      "theoretical_loss": 3.9978029641029154,
      "tokens_seen": 440401920
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008752858575727182,
      "loss": 1.6944,
      "theoretical_loss": 3.997737951983345,
      "tokens_seen": 440467456
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008752657973921766,
      "loss": 1.6073,
      "theoretical_loss": 3.997672952244014,
      "tokens_seen": 440532992
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008752457372116349,
      "loss": 1.8349,
      "theoretical_loss": 3.997607964880724,
      "tokens_seen": 440598528
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008752256770310934,
      "loss": 1.5962,
      "theoretical_loss": 3.9975429898892783,
      "tokens_seen": 440664064
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008752056168505516,
      "loss": 1.6863,
      "theoretical_loss": 3.997478027265481,
      "tokens_seen": 440729600
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008751855566700101,
      "loss": 1.7402,
      "theoretical_loss": 3.997413077005141,
      "tokens_seen": 440795136
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008751654964894684,
      "loss": 1.6486,
      "theoretical_loss": 3.9973481391040675,
      "tokens_seen": 440860672
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008751454363089268,
      "loss": 1.7126,
      "theoretical_loss": 3.9972832135580707,
      "tokens_seen": 440926208
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008751253761283852,
      "loss": 1.643,
      "theoretical_loss": 3.9972183003629653,
      "tokens_seen": 440991744
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008751053159478436,
      "loss": 1.7164,
      "theoretical_loss": 3.9971533995145663,
      "tokens_seen": 441057280
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008750852557673019,
      "loss": 1.7811,
      "theoretical_loss": 3.997088511008691,
      "tokens_seen": 441122816
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008750651955867604,
      "loss": 1.6393,
      "theoretical_loss": 3.9970236348411605,
      "tokens_seen": 441188352
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008750451354062186,
      "loss": 1.6637,
      "theoretical_loss": 3.996958771007794,
      "tokens_seen": 441253888
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008750250752256771,
      "loss": 1.7993,
      "theoretical_loss": 3.996893919504417,
      "tokens_seen": 441319424
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008750050150451354,
      "loss": 1.6861,
      "theoretical_loss": 3.996829080326855,
      "tokens_seen": 441384960
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008749849548645938,
      "loss": 1.615,
      "theoretical_loss": 3.996764253470935,
      "tokens_seen": 441450496
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008749648946840522,
      "loss": 1.6733,
      "theoretical_loss": 3.9966994389324872,
      "tokens_seen": 441516032
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008749448345035106,
      "loss": 1.6116,
      "theoretical_loss": 3.996634636707344,
      "tokens_seen": 441581568
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008749247743229689,
      "loss": 1.6706,
      "theoretical_loss": 3.9965698467913375,
      "tokens_seen": 441647104
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008749047141424273,
      "loss": 1.4712,
      "theoretical_loss": 3.996505069180306,
      "tokens_seen": 441712640
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008748846539618857,
      "loss": 1.73,
      "theoretical_loss": 3.9964403038700844,
      "tokens_seen": 441778176
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008748645937813441,
      "loss": 1.7456,
      "theoretical_loss": 3.9963755508565155,
      "tokens_seen": 441843712
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008748445336008024,
      "loss": 1.6791,
      "theoretical_loss": 3.99631081013544,
      "tokens_seen": 441909248
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008748244734202609,
      "loss": 1.6616,
      "theoretical_loss": 3.9962460817027017,
      "tokens_seen": 441974784
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008748044132397191,
      "loss": 1.6458,
      "theoretical_loss": 3.996181365554146,
      "tokens_seen": 442040320
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008747843530591776,
      "loss": 1.8686,
      "theoretical_loss": 3.9961166616856225,
      "tokens_seen": 442105856
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008747642928786359,
      "loss": 1.6062,
      "theoretical_loss": 3.9960519700929797,
      "tokens_seen": 442171392
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008747442326980943,
      "loss": 1.6539,
      "theoretical_loss": 3.995987290772071,
      "tokens_seen": 442236928
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008747241725175527,
      "loss": 1.7101,
      "theoretical_loss": 3.995922623718749,
      "tokens_seen": 442302464
    },
    {
      "epoch": 0.13,
      "objective/train/advantage_avg": 0.4943990111351013,
      "objective/train/docs_used": 166951,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5713313817977905,
      "objective/train/original_loss": 1.57133150100708,
      "objective/train/theoretical_loss": 3.9958579689288705,
      "objective/train/tokens_used": 462828000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24468739330768585,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050683856010437,
      "objective/train/weighted_lm_loss": 1.6510241031646729,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9958579689288705,
      "tokens_seen": 442368000
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008747041123370111,
      "loss": 1.719,
      "theoretical_loss": 3.9958579689288705,
      "tokens_seen": 442368000
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008746840521564694,
      "loss": 1.5793,
      "theoretical_loss": 3.995793326398293,
      "tokens_seen": 442433536
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008746639919759279,
      "loss": 1.6893,
      "theoretical_loss": 3.9957286961228786,
      "tokens_seen": 442499072
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008746439317953861,
      "loss": 1.6669,
      "theoretical_loss": 3.995664078098487,
      "tokens_seen": 442564608
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008746238716148446,
      "loss": 1.5583,
      "theoretical_loss": 3.9955994723209836,
      "tokens_seen": 442630144
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008746038114343029,
      "loss": 1.717,
      "theoretical_loss": 3.9955348787862337,
      "tokens_seen": 442695680
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008745837512537613,
      "loss": 1.5958,
      "theoretical_loss": 3.995470297490106,
      "tokens_seen": 442761216
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008745636910732197,
      "loss": 1.7998,
      "theoretical_loss": 3.9954057284284703,
      "tokens_seen": 442826752
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000874543630892678,
      "loss": 1.6256,
      "theoretical_loss": 3.9953411715971994,
      "tokens_seen": 442892288
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008745235707121364,
      "loss": 1.6284,
      "theoretical_loss": 3.9952766269921667,
      "tokens_seen": 442957824
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008745035105315949,
      "loss": 1.4994,
      "theoretical_loss": 3.995212094609249,
      "tokens_seen": 443023360
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008744834503510531,
      "loss": 1.8477,
      "theoretical_loss": 3.9951475744443234,
      "tokens_seen": 443088896
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008744633901705116,
      "loss": 1.5417,
      "theoretical_loss": 3.9950830664932706,
      "tokens_seen": 443154432
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008744433299899698,
      "loss": 1.6993,
      "theoretical_loss": 3.9950185707519736,
      "tokens_seen": 443219968
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008744232698094283,
      "loss": 1.5878,
      "theoretical_loss": 3.994954087216315,
      "tokens_seen": 443285504
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008744032096288867,
      "loss": 1.7611,
      "theoretical_loss": 3.994889615882182,
      "tokens_seen": 443351040
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000874383149448345,
      "loss": 1.6522,
      "theoretical_loss": 3.9948251567454625,
      "tokens_seen": 443416576
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008743630892678034,
      "loss": 1.7121,
      "theoretical_loss": 3.994760709802046,
      "tokens_seen": 443482112
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008743430290872619,
      "loss": 1.7826,
      "theoretical_loss": 3.994696275047825,
      "tokens_seen": 443547648
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008743229689067201,
      "loss": 1.6826,
      "theoretical_loss": 3.994631852478695,
      "tokens_seen": 443613184
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008743029087261786,
      "loss": 1.6519,
      "theoretical_loss": 3.99456744209055,
      "tokens_seen": 443678720
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008742828485456368,
      "loss": 1.6761,
      "theoretical_loss": 3.9945030438792886,
      "tokens_seen": 443744256
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008742627883650953,
      "loss": 1.7019,
      "theoretical_loss": 3.9944386578408113,
      "tokens_seen": 443809792
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008742427281845537,
      "loss": 1.638,
      "theoretical_loss": 3.99437428397102,
      "tokens_seen": 443875328
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000874222668004012,
      "loss": 1.673,
      "theoretical_loss": 3.9943099222658187,
      "tokens_seen": 443940864
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008742026078234704,
      "loss": 1.6107,
      "theoretical_loss": 3.9942455727211135,
      "tokens_seen": 444006400
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008741825476429288,
      "loss": 1.6643,
      "theoretical_loss": 3.9941812353328126,
      "tokens_seen": 444071936
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008741624874623871,
      "loss": 1.7482,
      "theoretical_loss": 3.9941169100968255,
      "tokens_seen": 444137472
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008741424272818456,
      "loss": 1.7802,
      "theoretical_loss": 3.9940525970090643,
      "tokens_seen": 444203008
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008741223671013039,
      "loss": 1.5553,
      "theoretical_loss": 3.993988296065443,
      "tokens_seen": 444268544
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008741023069207623,
      "loss": 1.5713,
      "theoretical_loss": 3.993924007261878,
      "tokens_seen": 444334080
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008740822467402206,
      "loss": 1.6922,
      "theoretical_loss": 3.993859730594286,
      "tokens_seen": 444399616
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000874062186559679,
      "loss": 1.8085,
      "theoretical_loss": 3.9937954660585877,
      "tokens_seen": 444465152
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008740421263791374,
      "loss": 1.6667,
      "theoretical_loss": 3.9937312136507055,
      "tokens_seen": 444530688
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008740220661985958,
      "loss": 1.717,
      "theoretical_loss": 3.9936669733665617,
      "tokens_seen": 444596224
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008740020060180541,
      "loss": 1.6322,
      "theoretical_loss": 3.9936027452020832,
      "tokens_seen": 444661760
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008739819458375126,
      "loss": 1.6617,
      "theoretical_loss": 3.9935385291531977,
      "tokens_seen": 444727296
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008739618856569709,
      "loss": 1.6524,
      "theoretical_loss": 3.9934743252158347,
      "tokens_seen": 444792832
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008739418254764293,
      "loss": 1.7594,
      "theoretical_loss": 3.9934101333859253,
      "tokens_seen": 444858368
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008739217652958876,
      "loss": 1.7404,
      "theoretical_loss": 3.9933459536594036,
      "tokens_seen": 444923904
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008739017051153461,
      "loss": 1.8118,
      "theoretical_loss": 3.993281786032206,
      "tokens_seen": 444989440
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008738816449348044,
      "loss": 1.6814,
      "theoretical_loss": 3.9932176305002693,
      "tokens_seen": 445054976
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008738615847542628,
      "loss": 1.6673,
      "theoretical_loss": 3.9931534870595327,
      "tokens_seen": 445120512
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008738415245737213,
      "loss": 1.6706,
      "theoretical_loss": 3.993089355705939,
      "tokens_seen": 445186048
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008738214643931795,
      "loss": 1.6417,
      "theoretical_loss": 3.9930252364354297,
      "tokens_seen": 445251584
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.000873801404212638,
      "loss": 1.7252,
      "theoretical_loss": 3.9929611292439517,
      "tokens_seen": 445317120
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008737813440320963,
      "loss": 1.6898,
      "theoretical_loss": 3.9928970341274517,
      "tokens_seen": 445382656
    },
    {
      "epoch": 0.13,
      "learning_rate": 0.0008737612838515547,
      "loss": 1.6426,
      "theoretical_loss": 3.99283295108188,
      "tokens_seen": 445448192
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008737412236710131,
      "loss": 1.6633,
      "theoretical_loss": 3.9927688801031866,
      "tokens_seen": 445513728
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008737211634904715,
      "loss": 1.6642,
      "theoretical_loss": 3.992704821187325,
      "tokens_seen": 445579264
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.4929441809654236,
      "objective/train/docs_used": 168097,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7667591571807861,
      "objective/train/original_loss": 1.7667592763900757,
      "objective/train/theoretical_loss": 3.992640774330251,
      "objective/train/tokens_used": 466104800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24337506294250488,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505316257476807,
      "objective/train/weighted_lm_loss": 1.855811595916748,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.992640774330251,
      "tokens_seen": 445644800
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008737011033099298,
      "loss": 1.7735,
      "theoretical_loss": 3.992640774330251,
      "tokens_seen": 445644800
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008736810431293883,
      "loss": 1.6244,
      "theoretical_loss": 3.992576739527922,
      "tokens_seen": 445710336
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008736609829488465,
      "loss": 1.7317,
      "theoretical_loss": 3.9925127167762966,
      "tokens_seen": 445775872
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000873640922768305,
      "loss": 1.6294,
      "theoretical_loss": 3.992448706071335,
      "tokens_seen": 445841408
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008736208625877633,
      "loss": 1.582,
      "theoretical_loss": 3.9923847074090015,
      "tokens_seen": 445906944
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008736008024072217,
      "loss": 1.658,
      "theoretical_loss": 3.9923207207852602,
      "tokens_seen": 445972480
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008735807422266801,
      "loss": 1.6315,
      "theoretical_loss": 3.9922567461960785,
      "tokens_seen": 446038016
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008735606820461384,
      "loss": 1.6296,
      "theoretical_loss": 3.9921927836374245,
      "tokens_seen": 446103552
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008735406218655968,
      "loss": 1.6206,
      "theoretical_loss": 3.9921288331052702,
      "tokens_seen": 446169088
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008735205616850553,
      "loss": 1.6181,
      "theoretical_loss": 3.992064894595587,
      "tokens_seen": 446234624
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008735005015045135,
      "loss": 1.6894,
      "theoretical_loss": 3.9920009681043505,
      "tokens_seen": 446300160
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000873480441323972,
      "loss": 1.7492,
      "theoretical_loss": 3.9919370536275367,
      "tokens_seen": 446365696
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008734603811434302,
      "loss": 1.635,
      "theoretical_loss": 3.991873151161124,
      "tokens_seen": 446431232
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008734403209628887,
      "loss": 1.6543,
      "theoretical_loss": 3.9918092607010935,
      "tokens_seen": 446496768
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008734202607823471,
      "loss": 1.7162,
      "theoretical_loss": 3.991745382243427,
      "tokens_seen": 446562304
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008734002006018054,
      "loss": 1.8018,
      "theoretical_loss": 3.9916815157841095,
      "tokens_seen": 446627840
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008733801404212638,
      "loss": 1.7373,
      "theoretical_loss": 3.9916176613191263,
      "tokens_seen": 446693376
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008733600802407223,
      "loss": 1.6995,
      "theoretical_loss": 3.9915538188444666,
      "tokens_seen": 446758912
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008733400200601805,
      "loss": 1.6555,
      "theoretical_loss": 3.9914899883561197,
      "tokens_seen": 446824448
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000873319959879639,
      "loss": 1.693,
      "theoretical_loss": 3.9914261698500777,
      "tokens_seen": 446889984
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008732998996990972,
      "loss": 1.6775,
      "theoretical_loss": 3.9913623633223354,
      "tokens_seen": 446955520
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008732798395185557,
      "loss": 1.5977,
      "theoretical_loss": 3.9912985687688876,
      "tokens_seen": 447021056
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008732597793380141,
      "loss": 1.6131,
      "theoretical_loss": 3.9912347861857334,
      "tokens_seen": 447086592
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008732397191574724,
      "loss": 1.8031,
      "theoretical_loss": 3.9911710155688715,
      "tokens_seen": 447152128
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008732196589769308,
      "loss": 1.4963,
      "theoretical_loss": 3.9911072569143036,
      "tokens_seen": 447217664
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008731995987963892,
      "loss": 1.7504,
      "theoretical_loss": 3.9910435102180344,
      "tokens_seen": 447283200
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008731795386158475,
      "loss": 1.6181,
      "theoretical_loss": 3.9909797754760676,
      "tokens_seen": 447348736
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000873159478435306,
      "loss": 1.5899,
      "theoretical_loss": 3.990916052684412,
      "tokens_seen": 447414272
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008731394182547643,
      "loss": 1.6269,
      "theoretical_loss": 3.9908523418390764,
      "tokens_seen": 447479808
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008731193580742227,
      "loss": 1.7889,
      "theoretical_loss": 3.9907886429360726,
      "tokens_seen": 447545344
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000873099297893681,
      "loss": 1.6514,
      "theoretical_loss": 3.9907249559714133,
      "tokens_seen": 447610880
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008730792377131394,
      "loss": 1.5434,
      "theoretical_loss": 3.9906612809411137,
      "tokens_seen": 447676416
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008730591775325978,
      "loss": 1.5948,
      "theoretical_loss": 3.990597617841191,
      "tokens_seen": 447741952
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008730391173520562,
      "loss": 1.6374,
      "theoretical_loss": 3.9905339666676642,
      "tokens_seen": 447807488
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008730190571715145,
      "loss": 1.5835,
      "theoretical_loss": 3.9904703274165527,
      "tokens_seen": 447873024
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000872998996990973,
      "loss": 1.7278,
      "theoretical_loss": 3.9904067000838817,
      "tokens_seen": 447938560
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008729789368104313,
      "loss": 1.6106,
      "theoretical_loss": 3.9903430846656742,
      "tokens_seen": 448004096
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008729588766298897,
      "loss": 1.7808,
      "theoretical_loss": 3.990279481157957,
      "tokens_seen": 448069632
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000872938816449348,
      "loss": 1.6589,
      "theoretical_loss": 3.990215889556759,
      "tokens_seen": 448135168
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008729187562688065,
      "loss": 1.6781,
      "theoretical_loss": 3.990152309858111,
      "tokens_seen": 448200704
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008728986960882648,
      "loss": 1.6864,
      "theoretical_loss": 3.990088742058043,
      "tokens_seen": 448266240
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008728786359077232,
      "loss": 1.7106,
      "theoretical_loss": 3.990025186152592,
      "tokens_seen": 448331776
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008728585757271815,
      "loss": 1.5557,
      "theoretical_loss": 3.989961642137792,
      "tokens_seen": 448397312
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008728385155466399,
      "loss": 1.6223,
      "theoretical_loss": 3.9898981100096824,
      "tokens_seen": 448462848
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008728184553660983,
      "loss": 1.6308,
      "theoretical_loss": 3.9898345897643024,
      "tokens_seen": 448528384
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008727983951855567,
      "loss": 1.7198,
      "theoretical_loss": 3.9897710813976937,
      "tokens_seen": 448593920
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000872778335005015,
      "loss": 1.5826,
      "theoretical_loss": 3.9897075849059003,
      "tokens_seen": 448659456
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008727582748244735,
      "loss": 1.5786,
      "theoretical_loss": 3.989644100284967,
      "tokens_seen": 448724992
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008727382146439318,
      "loss": 1.617,
      "theoretical_loss": 3.989580627530943,
      "tokens_seen": 448790528
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008727181544633902,
      "loss": 1.6481,
      "theoretical_loss": 3.989517166639876,
      "tokens_seen": 448856064
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.49197646975517273,
      "objective/train/docs_used": 169309,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.678598403930664,
      "objective/train/original_loss": 1.678598403930664,
      "objective/train/theoretical_loss": 3.9894537176078178,
      "objective/train/tokens_used": 469381600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24267804622650146,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504313707351685,
      "objective/train/weighted_lm_loss": 1.7632415294647217,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9894537176078178,
      "tokens_seen": 448921600
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008726980942828485,
      "loss": 1.6911,
      "theoretical_loss": 3.9894537176078178,
      "tokens_seen": 448921600
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008726780341023069,
      "loss": 1.6666,
      "theoretical_loss": 3.989390280430821,
      "tokens_seen": 448987136
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008726579739217653,
      "loss": 1.5208,
      "theoretical_loss": 3.9893268551049417,
      "tokens_seen": 449052672
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008726379137412237,
      "loss": 1.755,
      "theoretical_loss": 3.989263441626236,
      "tokens_seen": 449118208
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000872617853560682,
      "loss": 1.6132,
      "theoretical_loss": 3.989200039990763,
      "tokens_seen": 449183744
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008725977933801405,
      "loss": 1.576,
      "theoretical_loss": 3.989136650194583,
      "tokens_seen": 449249280
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008725777331995987,
      "loss": 1.6667,
      "theoretical_loss": 3.9890732722337594,
      "tokens_seen": 449314816
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008725576730190572,
      "loss": 1.6623,
      "theoretical_loss": 3.9890099061043562,
      "tokens_seen": 449380352
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008725376128385156,
      "loss": 1.7523,
      "theoretical_loss": 3.9889465518024396,
      "tokens_seen": 449445888
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008725175526579739,
      "loss": 1.7429,
      "theoretical_loss": 3.9888832093240776,
      "tokens_seen": 449511424
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008724974924774323,
      "loss": 1.7226,
      "theoretical_loss": 3.988819878665341,
      "tokens_seen": 449576960
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008724774322968906,
      "loss": 1.59,
      "theoretical_loss": 3.988756559822301,
      "tokens_seen": 449642496
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000872457372116349,
      "loss": 1.6499,
      "theoretical_loss": 3.988693252791032,
      "tokens_seen": 449708032
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008724373119358075,
      "loss": 1.7629,
      "theoretical_loss": 3.9886299575676096,
      "tokens_seen": 449773568
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008724172517552657,
      "loss": 1.6231,
      "theoretical_loss": 3.988566674148111,
      "tokens_seen": 449839104
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008723971915747242,
      "loss": 1.6632,
      "theoretical_loss": 3.988503402528617,
      "tokens_seen": 449904640
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008723771313941826,
      "loss": 1.726,
      "theoretical_loss": 3.9884401427052074,
      "tokens_seen": 449970176
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008723570712136409,
      "loss": 1.6258,
      "theoretical_loss": 3.9883768946739657,
      "tokens_seen": 450035712
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008723370110330994,
      "loss": 1.6387,
      "theoretical_loss": 3.988313658430978,
      "tokens_seen": 450101248
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008723169508525576,
      "loss": 1.6668,
      "theoretical_loss": 3.988250433972331,
      "tokens_seen": 450166784
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008722968906720161,
      "loss": 1.7694,
      "theoretical_loss": 3.9881872212941123,
      "tokens_seen": 450232320
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008722768304914745,
      "loss": 1.7324,
      "theoretical_loss": 3.9881240203924135,
      "tokens_seen": 450297856
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008722567703109328,
      "loss": 1.6488,
      "theoretical_loss": 3.9880608312633274,
      "tokens_seen": 450363392
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008722367101303912,
      "loss": 1.6934,
      "theoretical_loss": 3.9879976539029482,
      "tokens_seen": 450428928
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008722166499498496,
      "loss": 1.7163,
      "theoretical_loss": 3.9879344883073724,
      "tokens_seen": 450494464
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008721965897693079,
      "loss": 1.7422,
      "theoretical_loss": 3.9878713344726973,
      "tokens_seen": 450560000
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008721765295887664,
      "loss": 1.7008,
      "theoretical_loss": 3.9878081923950237,
      "tokens_seen": 450625536
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008721564694082247,
      "loss": 1.6685,
      "theoretical_loss": 3.9877450620704535,
      "tokens_seen": 450691072
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008721364092276831,
      "loss": 1.6977,
      "theoretical_loss": 3.9876819434950903,
      "tokens_seen": 450756608
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008721163490471414,
      "loss": 1.6814,
      "theoretical_loss": 3.987618836665039,
      "tokens_seen": 450822144
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008720962888665998,
      "loss": 1.6058,
      "theoretical_loss": 3.9875557415764087,
      "tokens_seen": 450887680
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008720762286860582,
      "loss": 1.6729,
      "theoretical_loss": 3.987492658225307,
      "tokens_seen": 450953216
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008720561685055166,
      "loss": 1.6695,
      "theoretical_loss": 3.987429586607846,
      "tokens_seen": 451018752
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008720361083249749,
      "loss": 1.5177,
      "theoretical_loss": 3.9873665267201384,
      "tokens_seen": 451084288
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008720160481444334,
      "loss": 1.6466,
      "theoretical_loss": 3.9873034785582995,
      "tokens_seen": 451149824
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008719959879638917,
      "loss": 1.5169,
      "theoretical_loss": 3.9872404421184453,
      "tokens_seen": 451215360
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008719759277833501,
      "loss": 1.6245,
      "theoretical_loss": 3.987177417396695,
      "tokens_seen": 451280896
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008719558676028084,
      "loss": 1.6227,
      "theoretical_loss": 3.9871144043891684,
      "tokens_seen": 451346432
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008719358074222669,
      "loss": 1.6857,
      "theoretical_loss": 3.9870514030919884,
      "tokens_seen": 451411968
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008719157472417252,
      "loss": 1.672,
      "theoretical_loss": 3.9869884135012787,
      "tokens_seen": 451477504
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008718956870611836,
      "loss": 1.6045,
      "theoretical_loss": 3.9869254356131654,
      "tokens_seen": 451543040
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008718756268806419,
      "loss": 1.6892,
      "theoretical_loss": 3.9868624694237766,
      "tokens_seen": 451608576
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008718555667001003,
      "loss": 1.6541,
      "theoretical_loss": 3.986799514929242,
      "tokens_seen": 451674112
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008718355065195587,
      "loss": 1.6627,
      "theoretical_loss": 3.986736572125692,
      "tokens_seen": 451739648
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008718154463390171,
      "loss": 1.653,
      "theoretical_loss": 3.9866736410092614,
      "tokens_seen": 451805184
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008717953861584754,
      "loss": 1.7972,
      "theoretical_loss": 3.986610721576084,
      "tokens_seen": 451870720
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008717753259779339,
      "loss": 1.7353,
      "theoretical_loss": 3.9865478138222974,
      "tokens_seen": 451936256
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008717552657973922,
      "loss": 1.6426,
      "theoretical_loss": 3.9864849177440407,
      "tokens_seen": 452001792
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008717352056168506,
      "loss": 1.6807,
      "theoretical_loss": 3.9864220333374547,
      "tokens_seen": 452067328
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000871715145436309,
      "loss": 1.6885,
      "theoretical_loss": 3.9863591605986812,
      "tokens_seen": 452132864
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.49392029643058777,
      "objective/train/docs_used": 170598,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.682446002960205,
      "objective/train/original_loss": 1.682446002960205,
      "objective/train/theoretical_loss": 3.9862962995238647,
      "objective/train/tokens_used": 472658400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2446342259645462,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506356954574585,
      "objective/train/weighted_lm_loss": 1.7676743268966675,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9862962995238647,
      "tokens_seen": 452198400
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008716950852557673,
      "loss": 1.7119,
      "theoretical_loss": 3.9862962995238647,
      "tokens_seen": 452198400
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008716750250752257,
      "loss": 1.7363,
      "theoretical_loss": 3.986233450109152,
      "tokens_seen": 452263936
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008716549648946841,
      "loss": 1.7325,
      "theoretical_loss": 3.9861706123506906,
      "tokens_seen": 452329472
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008716349047141424,
      "loss": 1.7326,
      "theoretical_loss": 3.986107786244631,
      "tokens_seen": 452395008
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008716148445336009,
      "loss": 1.7056,
      "theoretical_loss": 3.9860449717871234,
      "tokens_seen": 452460544
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008715947843530591,
      "loss": 1.6295,
      "theoretical_loss": 3.985982168974323,
      "tokens_seen": 452526080
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008715747241725176,
      "loss": 1.6789,
      "theoretical_loss": 3.985919377802384,
      "tokens_seen": 452591616
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000871554663991976,
      "loss": 1.5917,
      "theoretical_loss": 3.9858565982674636,
      "tokens_seen": 452657152
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008715346038114343,
      "loss": 1.7548,
      "theoretical_loss": 3.9857938303657217,
      "tokens_seen": 452722688
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008715145436308927,
      "loss": 1.7352,
      "theoretical_loss": 3.985731074093318,
      "tokens_seen": 452788224
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000871494483450351,
      "loss": 1.6529,
      "theoretical_loss": 3.985668329446416,
      "tokens_seen": 452853760
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008714744232698094,
      "loss": 1.7179,
      "theoretical_loss": 3.9856055964211796,
      "tokens_seen": 452919296
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008714543630892679,
      "loss": 1.8021,
      "theoretical_loss": 3.9855428750137754,
      "tokens_seen": 452984832
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008714343029087261,
      "loss": 1.7151,
      "theoretical_loss": 3.9854801652203706,
      "tokens_seen": 453050368
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008714142427281846,
      "loss": 1.672,
      "theoretical_loss": 3.985417467037137,
      "tokens_seen": 453115904
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000871394182547643,
      "loss": 1.7477,
      "theoretical_loss": 3.985354780460244,
      "tokens_seen": 453181440
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008713741223671013,
      "loss": 1.8086,
      "theoretical_loss": 3.9852921054858665,
      "tokens_seen": 453246976
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008713540621865597,
      "loss": 1.7011,
      "theoretical_loss": 3.98522944211018,
      "tokens_seen": 453312512
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000871334002006018,
      "loss": 1.6457,
      "theoretical_loss": 3.9851667903293615,
      "tokens_seen": 453378048
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008713139418254764,
      "loss": 1.6447,
      "theoretical_loss": 3.9851041501395894,
      "tokens_seen": 453443584
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008712938816449349,
      "loss": 1.6089,
      "theoretical_loss": 3.9850415215370445,
      "tokens_seen": 453509120
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008712738214643931,
      "loss": 1.578,
      "theoretical_loss": 3.9849789045179103,
      "tokens_seen": 453574656
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008712537612838516,
      "loss": 1.5896,
      "theoretical_loss": 3.98491629907837,
      "tokens_seen": 453640192
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008712337011033099,
      "loss": 1.6224,
      "theoretical_loss": 3.984853705214611,
      "tokens_seen": 453705728
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008712136409227683,
      "loss": 1.5858,
      "theoretical_loss": 3.984791122922821,
      "tokens_seen": 453771264
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008711935807422267,
      "loss": 1.7106,
      "theoretical_loss": 3.984728552199189,
      "tokens_seen": 453836800
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008711735205616851,
      "loss": 1.6795,
      "theoretical_loss": 3.984665993039907,
      "tokens_seen": 453902336
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008711534603811434,
      "loss": 1.6176,
      "theoretical_loss": 3.984603445441169,
      "tokens_seen": 453967872
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008711334002006018,
      "loss": 1.6872,
      "theoretical_loss": 3.98454090939917,
      "tokens_seen": 454033408
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008711133400200601,
      "loss": 1.6369,
      "theoretical_loss": 3.9844783849101075,
      "tokens_seen": 454098944
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008710932798395186,
      "loss": 1.6765,
      "theoretical_loss": 3.984415871970179,
      "tokens_seen": 454164480
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008710732196589769,
      "loss": 1.6515,
      "theoretical_loss": 3.984353370575586,
      "tokens_seen": 454230016
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008710531594784353,
      "loss": 1.7684,
      "theoretical_loss": 3.984290880722531,
      "tokens_seen": 454295552
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008710330992978937,
      "loss": 1.6004,
      "theoretical_loss": 3.9842284024072177,
      "tokens_seen": 454361088
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008710130391173521,
      "loss": 1.659,
      "theoretical_loss": 3.9841659356258523,
      "tokens_seen": 454426624
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008709929789368104,
      "loss": 1.621,
      "theoretical_loss": 3.984103480374644,
      "tokens_seen": 454492160
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008709729187562688,
      "loss": 1.6278,
      "theoretical_loss": 3.9840410366498,
      "tokens_seen": 454557696
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008709528585757271,
      "loss": 1.8119,
      "theoretical_loss": 3.9839786044475334,
      "tokens_seen": 454623232
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008709327983951856,
      "loss": 1.6085,
      "theoretical_loss": 3.983916183764057,
      "tokens_seen": 454688768
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008709127382146439,
      "loss": 1.6838,
      "theoretical_loss": 3.9838537745955858,
      "tokens_seen": 454754304
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008708926780341023,
      "loss": 1.6876,
      "theoretical_loss": 3.983791376938336,
      "tokens_seen": 454819840
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008708726178535606,
      "loss": 1.5865,
      "theoretical_loss": 3.9837289907885274,
      "tokens_seen": 454885376
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008708525576730191,
      "loss": 1.6709,
      "theoretical_loss": 3.9836666161423793,
      "tokens_seen": 454950912
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008708324974924775,
      "loss": 1.616,
      "theoretical_loss": 3.9836042529961135,
      "tokens_seen": 455016448
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008708124373119358,
      "loss": 1.6237,
      "theoretical_loss": 3.983541901345955,
      "tokens_seen": 455081984
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008707923771313943,
      "loss": 1.7014,
      "theoretical_loss": 3.9834795611881297,
      "tokens_seen": 455147520
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008707723169508526,
      "loss": 1.7364,
      "theoretical_loss": 3.983417232518864,
      "tokens_seen": 455213056
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000870752256770311,
      "loss": 1.6464,
      "theoretical_loss": 3.9833549153343872,
      "tokens_seen": 455278592
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008707321965897693,
      "loss": 1.7088,
      "theoretical_loss": 3.983292609630931,
      "tokens_seen": 455344128
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008707121364092277,
      "loss": 1.681,
      "theoretical_loss": 3.983230315404728,
      "tokens_seen": 455409664
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.48992982506752014,
      "objective/train/docs_used": 171735,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6366153955459595,
      "objective/train/original_loss": 1.63661527633667,
      "objective/train/theoretical_loss": 3.983168032652013,
      "objective/train/tokens_used": 475935200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2419193685054779,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502227544784546,
      "objective/train/weighted_lm_loss": 1.718919277191162,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.983168032652013,
      "tokens_seen": 455475200
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008706920762286861,
      "loss": 1.7645,
      "theoretical_loss": 3.983168032652013,
      "tokens_seen": 455475200
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008706720160481445,
      "loss": 1.6411,
      "theoretical_loss": 3.983105761369022,
      "tokens_seen": 455540736
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008706519558676028,
      "loss": 1.7468,
      "theoretical_loss": 3.9830435015519936,
      "tokens_seen": 455606272
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008706318956870613,
      "loss": 1.6752,
      "theoretical_loss": 3.9829812531971673,
      "tokens_seen": 455671808
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008706118355065195,
      "loss": 1.6877,
      "theoretical_loss": 3.9829190163007846,
      "tokens_seen": 455737344
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000870591775325978,
      "loss": 1.6541,
      "theoretical_loss": 3.98285679085909,
      "tokens_seen": 455802880
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008705717151454364,
      "loss": 1.709,
      "theoretical_loss": 3.982794576868328,
      "tokens_seen": 455868416
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008705516549648947,
      "loss": 1.6451,
      "theoretical_loss": 3.9827323743247454,
      "tokens_seen": 455933952
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008705315947843531,
      "loss": 1.7059,
      "theoretical_loss": 3.9826701832245917,
      "tokens_seen": 455999488
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008705115346038114,
      "loss": 1.6,
      "theoretical_loss": 3.982608003564117,
      "tokens_seen": 456065024
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008704914744232698,
      "loss": 1.6583,
      "theoretical_loss": 3.982545835339573,
      "tokens_seen": 456130560
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008704714142427283,
      "loss": 1.6567,
      "theoretical_loss": 3.982483678547215,
      "tokens_seen": 456196096
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008704513540621865,
      "loss": 1.6338,
      "theoretical_loss": 3.982421533183299,
      "tokens_seen": 456261632
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000870431293881645,
      "loss": 1.7144,
      "theoretical_loss": 3.982359399244081,
      "tokens_seen": 456327168
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008704112337011034,
      "loss": 1.6345,
      "theoretical_loss": 3.982297276725822,
      "tokens_seen": 456392704
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008703911735205617,
      "loss": 1.6444,
      "theoretical_loss": 3.9822351656247816,
      "tokens_seen": 456458240
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008703711133400201,
      "loss": 1.5523,
      "theoretical_loss": 3.982173065937225,
      "tokens_seen": 456523776
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008703510531594784,
      "loss": 1.7206,
      "theoretical_loss": 3.9821109776594144,
      "tokens_seen": 456589312
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008703309929789368,
      "loss": 1.6512,
      "theoretical_loss": 3.9820489007876176,
      "tokens_seen": 456654848
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008703109327983953,
      "loss": 1.5267,
      "theoretical_loss": 3.9819868353181027,
      "tokens_seen": 456720384
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008702908726178535,
      "loss": 1.6197,
      "theoretical_loss": 3.981924781247139,
      "tokens_seen": 456785920
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000870270812437312,
      "loss": 1.5956,
      "theoretical_loss": 3.9818627385709986,
      "tokens_seen": 456851456
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008702507522567703,
      "loss": 1.5874,
      "theoretical_loss": 3.981800707285955,
      "tokens_seen": 456916992
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008702306920762287,
      "loss": 1.6294,
      "theoretical_loss": 3.9817386873882836,
      "tokens_seen": 456982528
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008702106318956871,
      "loss": 1.7186,
      "theoretical_loss": 3.981676678874261,
      "tokens_seen": 457048064
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008701905717151455,
      "loss": 1.6111,
      "theoretical_loss": 3.981614681740166,
      "tokens_seen": 457113600
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008701705115346038,
      "loss": 1.721,
      "theoretical_loss": 3.9815526959822787,
      "tokens_seen": 457179136
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008701504513540622,
      "loss": 1.7458,
      "theoretical_loss": 3.9814907215968818,
      "tokens_seen": 457244672
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008701303911735205,
      "loss": 1.6473,
      "theoretical_loss": 3.9814287585802592,
      "tokens_seen": 457310208
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000870110330992979,
      "loss": 1.6326,
      "theoretical_loss": 3.9813668069286967,
      "tokens_seen": 457375744
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008700902708124373,
      "loss": 1.4838,
      "theoretical_loss": 3.981304866638481,
      "tokens_seen": 457441280
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008700702106318957,
      "loss": 1.718,
      "theoretical_loss": 3.981242937705902,
      "tokens_seen": 457506816
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008700501504513541,
      "loss": 1.7773,
      "theoretical_loss": 3.9811810201272513,
      "tokens_seen": 457572352
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008700300902708125,
      "loss": 1.6464,
      "theoretical_loss": 3.98111911389882,
      "tokens_seen": 457637888
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008700100300902708,
      "loss": 1.6777,
      "theoretical_loss": 3.9810572190169027,
      "tokens_seen": 457703424
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008699899699097292,
      "loss": 1.6622,
      "theoretical_loss": 3.9809953354777967,
      "tokens_seen": 457768960
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008699699097291875,
      "loss": 1.7199,
      "theoretical_loss": 3.9809334632777986,
      "tokens_seen": 457834496
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000869949849548646,
      "loss": 1.7178,
      "theoretical_loss": 3.9808716024132096,
      "tokens_seen": 457900032
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008699297893681043,
      "loss": 1.6575,
      "theoretical_loss": 3.9808097528803295,
      "tokens_seen": 457965568
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008699097291875627,
      "loss": 1.6969,
      "theoretical_loss": 3.9807479146754625,
      "tokens_seen": 458031104
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000869889669007021,
      "loss": 1.5759,
      "theoretical_loss": 3.9806860877949126,
      "tokens_seen": 458096640
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008698696088264795,
      "loss": 1.6771,
      "theoretical_loss": 3.9806242722349867,
      "tokens_seen": 458162176
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008698495486459378,
      "loss": 1.7865,
      "theoretical_loss": 3.9805624679919935,
      "tokens_seen": 458227712
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008698294884653962,
      "loss": 1.8104,
      "theoretical_loss": 3.9805006750622427,
      "tokens_seen": 458293248
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008698094282848546,
      "loss": 1.695,
      "theoretical_loss": 3.980438893442046,
      "tokens_seen": 458358784
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000869789368104313,
      "loss": 1.7279,
      "theoretical_loss": 3.9803771231277167,
      "tokens_seen": 458424320
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008697693079237713,
      "loss": 1.6565,
      "theoretical_loss": 3.98031536411557,
      "tokens_seen": 458489856
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008697492477432297,
      "loss": 1.6428,
      "theoretical_loss": 3.9802536164019235,
      "tokens_seen": 458555392
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000869729187562688,
      "loss": 1.6321,
      "theoretical_loss": 3.9801918799830958,
      "tokens_seen": 458620928
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008697091273821465,
      "loss": 1.7232,
      "theoretical_loss": 3.9801301548554067,
      "tokens_seen": 458686464
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.49433034658432007,
      "objective/train/docs_used": 173009,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5577558279037476,
      "objective/train/original_loss": 1.557755708694458,
      "objective/train/theoretical_loss": 3.9800684410151783,
      "objective/train/tokens_used": 479212000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2445380836725235,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506762266159058,
      "objective/train/weighted_lm_loss": 1.6367053985595703,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9800684410151783,
      "tokens_seen": 458752000
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008696890672016048,
      "loss": 1.6053,
      "theoretical_loss": 3.9800684410151783,
      "tokens_seen": 458752000
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008696690070210632,
      "loss": 1.7288,
      "theoretical_loss": 3.9800067384587354,
      "tokens_seen": 458817536
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008696489468405216,
      "loss": 1.5646,
      "theoretical_loss": 3.9799450471824023,
      "tokens_seen": 458883072
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008696288866599799,
      "loss": 1.5993,
      "theoretical_loss": 3.9798833671825076,
      "tokens_seen": 458948608
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008696088264794383,
      "loss": 1.5431,
      "theoretical_loss": 3.979821698455379,
      "tokens_seen": 459014144
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008695887662988968,
      "loss": 1.6967,
      "theoretical_loss": 3.9797600409973484,
      "tokens_seen": 459079680
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000869568706118355,
      "loss": 1.7241,
      "theoretical_loss": 3.979698394804747,
      "tokens_seen": 459145216
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008695486459378135,
      "loss": 1.6775,
      "theoretical_loss": 3.9796367598739106,
      "tokens_seen": 459210752
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008695285857572717,
      "loss": 1.6272,
      "theoretical_loss": 3.9795751362011735,
      "tokens_seen": 459276288
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008695085255767302,
      "loss": 1.6908,
      "theoretical_loss": 3.9795135237828747,
      "tokens_seen": 459341824
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008694884653961886,
      "loss": 1.6438,
      "theoretical_loss": 3.979451922615352,
      "tokens_seen": 459407360
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008694684052156469,
      "loss": 1.7366,
      "theoretical_loss": 3.9793903326949476,
      "tokens_seen": 459472896
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008694483450351053,
      "loss": 1.5684,
      "theoretical_loss": 3.979328754018004,
      "tokens_seen": 459538432
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008694282848545638,
      "loss": 1.6673,
      "theoretical_loss": 3.9792671865808646,
      "tokens_seen": 459603968
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000869408224674022,
      "loss": 1.609,
      "theoretical_loss": 3.9792056303798775,
      "tokens_seen": 459669504
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008693881644934805,
      "loss": 1.6739,
      "theoretical_loss": 3.9791440854113893,
      "tokens_seen": 459735040
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008693681043129388,
      "loss": 1.7139,
      "theoretical_loss": 3.979082551671749,
      "tokens_seen": 459800576
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008693480441323972,
      "loss": 1.7229,
      "theoretical_loss": 3.97902102915731,
      "tokens_seen": 459866112
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008693279839518557,
      "loss": 1.6994,
      "theoretical_loss": 3.9789595178644226,
      "tokens_seen": 459931648
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008693079237713139,
      "loss": 1.7112,
      "theoretical_loss": 3.9788980177894433,
      "tokens_seen": 459997184
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008692878635907724,
      "loss": 1.7255,
      "theoretical_loss": 3.9788365289287286,
      "tokens_seen": 460062720
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008692678034102307,
      "loss": 1.6869,
      "theoretical_loss": 3.9787750512786353,
      "tokens_seen": 460128256
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008692477432296891,
      "loss": 1.6763,
      "theoretical_loss": 3.978713584835524,
      "tokens_seen": 460193792
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008692276830491475,
      "loss": 1.7387,
      "theoretical_loss": 3.9786521295957558,
      "tokens_seen": 460259328
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008692076228686059,
      "loss": 1.7619,
      "theoretical_loss": 3.9785906855556945,
      "tokens_seen": 460324864
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008691875626880642,
      "loss": 1.6247,
      "theoretical_loss": 3.978529252711705,
      "tokens_seen": 460390400
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008691675025075226,
      "loss": 1.6519,
      "theoretical_loss": 3.9784678310601533,
      "tokens_seen": 460455936
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008691474423269809,
      "loss": 1.6962,
      "theoretical_loss": 3.9784064205974077,
      "tokens_seen": 460521472
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008691273821464394,
      "loss": 1.6269,
      "theoretical_loss": 3.9783450213198384,
      "tokens_seen": 460587008
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008691073219658977,
      "loss": 1.5943,
      "theoretical_loss": 3.978283633223817,
      "tokens_seen": 460652544
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008690872617853561,
      "loss": 1.64,
      "theoretical_loss": 3.9782222563057177,
      "tokens_seen": 460718080
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008690672016048145,
      "loss": 1.673,
      "theoretical_loss": 3.978160890561914,
      "tokens_seen": 460783616
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008690471414242729,
      "loss": 1.6254,
      "theoretical_loss": 3.9780995359887843,
      "tokens_seen": 460849152
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008690270812437312,
      "loss": 1.6223,
      "theoretical_loss": 3.978038192582706,
      "tokens_seen": 460914688
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008690070210631896,
      "loss": 1.7393,
      "theoretical_loss": 3.9779768603400587,
      "tokens_seen": 460980224
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000868986960882648,
      "loss": 1.6811,
      "theoretical_loss": 3.977915539257226,
      "tokens_seen": 461045760
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008689669007021064,
      "loss": 1.7563,
      "theoretical_loss": 3.9778542293305894,
      "tokens_seen": 461111296
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008689468405215647,
      "loss": 1.8148,
      "theoretical_loss": 3.977792930556536,
      "tokens_seen": 461176832
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008689267803410231,
      "loss": 1.5884,
      "theoretical_loss": 3.9777316429314507,
      "tokens_seen": 461242368
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008689067201604814,
      "loss": 1.6516,
      "theoretical_loss": 3.9776703664517234,
      "tokens_seen": 461307904
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008688866599799399,
      "loss": 1.7036,
      "theoretical_loss": 3.977609101113744,
      "tokens_seen": 461373440
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008688665997993982,
      "loss": 1.6015,
      "theoretical_loss": 3.9775478469139047,
      "tokens_seen": 461438976
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008688465396188566,
      "loss": 1.5942,
      "theoretical_loss": 3.9774866038485985,
      "tokens_seen": 461504512
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000868826479438315,
      "loss": 1.6146,
      "theoretical_loss": 3.9774253719142214,
      "tokens_seen": 461570048
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008688064192577734,
      "loss": 1.517,
      "theoretical_loss": 3.97736415110717,
      "tokens_seen": 461635584
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008687863590772317,
      "loss": 1.8449,
      "theoretical_loss": 3.9773029414238428,
      "tokens_seen": 461701120
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008687662988966902,
      "loss": 1.7828,
      "theoretical_loss": 3.97724174286064,
      "tokens_seen": 461766656
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008687462387161484,
      "loss": 1.698,
      "theoretical_loss": 3.977180555413964,
      "tokens_seen": 461832192
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008687261785356069,
      "loss": 1.7773,
      "theoretical_loss": 3.977119379080218,
      "tokens_seen": 461897728
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008687061183550652,
      "loss": 1.6191,
      "theoretical_loss": 3.9770582138558077,
      "tokens_seen": 461963264
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.4929687976837158,
      "objective/train/docs_used": 174193,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7287741899490356,
      "objective/train/original_loss": 1.728774070739746,
      "objective/train/theoretical_loss": 3.9769970597371405,
      "objective/train/tokens_used": 482488800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2434602826833725,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505346059799194,
      "objective/train/weighted_lm_loss": 1.8161708116531372,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 3.9769970597371405,
      "tokens_seen": 462028800
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008686860581745236,
      "loss": 1.7777,
      "theoretical_loss": 3.9769970597371405,
      "tokens_seen": 462028800
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000868665997993982,
      "loss": 1.7605,
      "theoretical_loss": 3.976935916720624,
      "tokens_seen": 462094336
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008686459378134403,
      "loss": 1.808,
      "theoretical_loss": 3.9768747848026695,
      "tokens_seen": 462159872
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008686258776328987,
      "loss": 1.5267,
      "theoretical_loss": 3.976813663979689,
      "tokens_seen": 462225408
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008686058174523572,
      "loss": 1.6349,
      "theoretical_loss": 3.976752554248095,
      "tokens_seen": 462290944
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008685857572718154,
      "loss": 1.6544,
      "theoretical_loss": 3.9766914556043043,
      "tokens_seen": 462356480
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008685656970912739,
      "loss": 1.7583,
      "theoretical_loss": 3.9766303680447335,
      "tokens_seen": 462422016
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008685456369107321,
      "loss": 1.6585,
      "theoretical_loss": 3.976569291565801,
      "tokens_seen": 462487552
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008685255767301906,
      "loss": 1.6723,
      "theoretical_loss": 3.9765082261639275,
      "tokens_seen": 462553088
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000868505516549649,
      "loss": 1.6631,
      "theoretical_loss": 3.976447171835534,
      "tokens_seen": 462618624
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008684854563691073,
      "loss": 1.5455,
      "theoretical_loss": 3.9763861285770457,
      "tokens_seen": 462684160
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008684653961885657,
      "loss": 1.6465,
      "theoretical_loss": 3.9763250963848873,
      "tokens_seen": 462749696
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008684453360080242,
      "loss": 1.6204,
      "theoretical_loss": 3.976264075255485,
      "tokens_seen": 462815232
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008684252758274824,
      "loss": 1.6969,
      "theoretical_loss": 3.9762030651852687,
      "tokens_seen": 462880768
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008684052156469409,
      "loss": 1.8199,
      "theoretical_loss": 3.9761420661706683,
      "tokens_seen": 462946304
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008683851554663991,
      "loss": 1.6761,
      "theoretical_loss": 3.976081078208116,
      "tokens_seen": 463011840
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008683650952858576,
      "loss": 1.6621,
      "theoretical_loss": 3.976020101294045,
      "tokens_seen": 463077376
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000868345035105316,
      "loss": 1.6585,
      "theoretical_loss": 3.97595913542489,
      "tokens_seen": 463142912
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008683249749247743,
      "loss": 1.5819,
      "theoretical_loss": 3.975898180597089,
      "tokens_seen": 463208448
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008683049147442327,
      "loss": 1.5759,
      "theoretical_loss": 3.97583723680708,
      "tokens_seen": 463273984
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008682848545636911,
      "loss": 1.6291,
      "theoretical_loss": 3.975776304051304,
      "tokens_seen": 463339520
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008682647943831494,
      "loss": 1.774,
      "theoretical_loss": 3.975715382326203,
      "tokens_seen": 463405056
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008682447342026079,
      "loss": 1.7556,
      "theoretical_loss": 3.9756544716282187,
      "tokens_seen": 463470592
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008682246740220661,
      "loss": 1.5184,
      "theoretical_loss": 3.975593571953798,
      "tokens_seen": 463536128
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008682046138415246,
      "loss": 1.6399,
      "theoretical_loss": 3.975532683299387,
      "tokens_seen": 463601664
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008681845536609829,
      "loss": 1.6015,
      "theoretical_loss": 3.975471805661435,
      "tokens_seen": 463667200
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008681644934804413,
      "loss": 1.7948,
      "theoretical_loss": 3.975410939036392,
      "tokens_seen": 463732736
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008681444332998997,
      "loss": 1.6474,
      "theoretical_loss": 3.9753500834207083,
      "tokens_seen": 463798272
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008681243731193581,
      "loss": 1.6236,
      "theoretical_loss": 3.975289238810839,
      "tokens_seen": 463863808
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008681043129388164,
      "loss": 1.7626,
      "theoretical_loss": 3.975228405203239,
      "tokens_seen": 463929344
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008680842527582749,
      "loss": 1.6521,
      "theoretical_loss": 3.9751675825943638,
      "tokens_seen": 463994880
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008680641925777332,
      "loss": 1.8156,
      "theoretical_loss": 3.9751067709806733,
      "tokens_seen": 464060416
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008680441323971916,
      "loss": 1.8843,
      "theoretical_loss": 3.975045970358626,
      "tokens_seen": 464125952
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008680240722166499,
      "loss": 1.7733,
      "theoretical_loss": 3.974985180724685,
      "tokens_seen": 464191488
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008680040120361083,
      "loss": 1.5756,
      "theoretical_loss": 3.974924402075313,
      "tokens_seen": 464257024
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008679839518555667,
      "loss": 1.6463,
      "theoretical_loss": 3.9748636344069745,
      "tokens_seen": 464322560
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008679638916750251,
      "loss": 1.8061,
      "theoretical_loss": 3.9748028777161366,
      "tokens_seen": 464388096
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008679438314944834,
      "loss": 1.6805,
      "theoretical_loss": 3.9747421319992666,
      "tokens_seen": 464453632
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008679237713139418,
      "loss": 1.6201,
      "theoretical_loss": 3.9746813972528354,
      "tokens_seen": 464519168
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008679037111334002,
      "loss": 1.6282,
      "theoretical_loss": 3.9746206734733143,
      "tokens_seen": 464584704
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008678836509528586,
      "loss": 1.6675,
      "theoretical_loss": 3.974559960657176,
      "tokens_seen": 464650240
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000867863590772317,
      "loss": 1.7889,
      "theoretical_loss": 3.974499258800895,
      "tokens_seen": 464715776
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008678435305917754,
      "loss": 1.7932,
      "theoretical_loss": 3.9744385679009486,
      "tokens_seen": 464781312
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008678234704112338,
      "loss": 1.5776,
      "theoretical_loss": 3.974377887953813,
      "tokens_seen": 464846848
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008678034102306921,
      "loss": 1.6429,
      "theoretical_loss": 3.97431721895597,
      "tokens_seen": 464912384
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008677833500501506,
      "loss": 1.7277,
      "theoretical_loss": 3.9742565609039,
      "tokens_seen": 464977920
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008677632898696088,
      "loss": 1.5898,
      "theoretical_loss": 3.9741959137940848,
      "tokens_seen": 465043456
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008677432296890673,
      "loss": 1.5901,
      "theoretical_loss": 3.9741352776230103,
      "tokens_seen": 465108992
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008677231695085256,
      "loss": 1.6582,
      "theoretical_loss": 3.9740746523871615,
      "tokens_seen": 465174528
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000867703109327984,
      "loss": 1.6925,
      "theoretical_loss": 3.974014038083027,
      "tokens_seen": 465240064
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.49394601583480835,
      "objective/train/docs_used": 175471,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.3470487594604492,
      "objective/train/original_loss": 1.3470487594604492,
      "objective/train/theoretical_loss": 3.973953434707096,
      "objective/train/tokens_used": 485765600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2442466765642166,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506362915039062,
      "objective/train/weighted_lm_loss": 1.415244221687317,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.973953434707096,
      "tokens_seen": 465305600
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008676830491474424,
      "loss": 1.5362,
      "theoretical_loss": 3.973953434707096,
      "tokens_seen": 465305600
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008676629889669007,
      "loss": 1.6045,
      "theoretical_loss": 3.973892842255859,
      "tokens_seen": 465371136
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008676429287863591,
      "loss": 1.6473,
      "theoretical_loss": 3.9738322607258088,
      "tokens_seen": 465436672
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008676228686058176,
      "loss": 1.7065,
      "theoretical_loss": 3.97377169011344,
      "tokens_seen": 465502208
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008676028084252758,
      "loss": 1.6219,
      "theoretical_loss": 3.973711130415248,
      "tokens_seen": 465567744
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008675827482447343,
      "loss": 1.6834,
      "theoretical_loss": 3.9736505816277305,
      "tokens_seen": 465633280
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008675626880641925,
      "loss": 1.5651,
      "theoretical_loss": 3.973590043747387,
      "tokens_seen": 465698816
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000867542627883651,
      "loss": 1.7065,
      "theoretical_loss": 3.9735295167707165,
      "tokens_seen": 465764352
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008675225677031094,
      "loss": 1.7577,
      "theoretical_loss": 3.973469000694223,
      "tokens_seen": 465829888
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008675025075225677,
      "loss": 1.6708,
      "theoretical_loss": 3.9734084955144104,
      "tokens_seen": 465895424
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008674824473420261,
      "loss": 1.6733,
      "theoretical_loss": 3.9733480012277833,
      "tokens_seen": 465960960
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008674623871614846,
      "loss": 1.5407,
      "theoretical_loss": 3.9732875178308493,
      "tokens_seen": 466026496
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008674423269809428,
      "loss": 1.6824,
      "theoretical_loss": 3.973227045320117,
      "tokens_seen": 466092032
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008674222668004013,
      "loss": 1.4976,
      "theoretical_loss": 3.9731665836920973,
      "tokens_seen": 466157568
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008674022066198595,
      "loss": 1.6535,
      "theoretical_loss": 3.9731061329433013,
      "tokens_seen": 466223104
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000867382146439318,
      "loss": 1.8235,
      "theoretical_loss": 3.9730456930702434,
      "tokens_seen": 466288640
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008673620862587764,
      "loss": 1.7147,
      "theoretical_loss": 3.9729852640694383,
      "tokens_seen": 466354176
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008673420260782347,
      "loss": 1.736,
      "theoretical_loss": 3.9729248459374027,
      "tokens_seen": 466419712
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008673219658976931,
      "loss": 1.6526,
      "theoretical_loss": 3.9728644386706553,
      "tokens_seen": 466485248
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008673019057171515,
      "loss": 1.6138,
      "theoretical_loss": 3.9728040422657163,
      "tokens_seen": 466550784
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008672818455366098,
      "loss": 1.7119,
      "theoretical_loss": 3.972743656719107,
      "tokens_seen": 466616320
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008672617853560683,
      "loss": 1.5707,
      "theoretical_loss": 3.97268328202735,
      "tokens_seen": 466681856
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008672417251755265,
      "loss": 1.5306,
      "theoretical_loss": 3.972622918186971,
      "tokens_seen": 466747392
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000867221664994985,
      "loss": 1.6185,
      "theoretical_loss": 3.9725625651944965,
      "tokens_seen": 466812928
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008672016048144433,
      "loss": 1.6831,
      "theoretical_loss": 3.9725022230464537,
      "tokens_seen": 466878464
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008671815446339017,
      "loss": 1.7115,
      "theoretical_loss": 3.972441891739373,
      "tokens_seen": 466944000
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008671614844533601,
      "loss": 1.6376,
      "theoretical_loss": 3.9723815712697848,
      "tokens_seen": 467009536
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008671414242728185,
      "loss": 1.5643,
      "theoretical_loss": 3.9723212616342227,
      "tokens_seen": 467075072
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008671213640922768,
      "loss": 1.7625,
      "theoretical_loss": 3.9722609628292207,
      "tokens_seen": 467140608
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008671013039117353,
      "loss": 1.6438,
      "theoretical_loss": 3.9722006748513143,
      "tokens_seen": 467206144
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008670812437311936,
      "loss": 1.6106,
      "theoretical_loss": 3.9721403976970424,
      "tokens_seen": 467271680
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000867061183550652,
      "loss": 1.6831,
      "theoretical_loss": 3.972080131362943,
      "tokens_seen": 467337216
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008670411233701103,
      "loss": 1.6706,
      "theoretical_loss": 3.9720198758455574,
      "tokens_seen": 467402752
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008670210631895687,
      "loss": 1.6925,
      "theoretical_loss": 3.971959631141428,
      "tokens_seen": 467468288
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008670010030090271,
      "loss": 1.6552,
      "theoretical_loss": 3.971899397247099,
      "tokens_seen": 467533824
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008669809428284855,
      "loss": 1.6002,
      "theoretical_loss": 3.9718391741591144,
      "tokens_seen": 467599360
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008669608826479438,
      "loss": 1.7323,
      "theoretical_loss": 3.9717789618740227,
      "tokens_seen": 467664896
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008669408224674022,
      "loss": 1.7703,
      "theoretical_loss": 3.971718760388373,
      "tokens_seen": 467730432
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008669207622868606,
      "loss": 1.6947,
      "theoretical_loss": 3.971658569698714,
      "tokens_seen": 467795968
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000866900702106319,
      "loss": 1.5733,
      "theoretical_loss": 3.971598389801599,
      "tokens_seen": 467861504
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008668806419257773,
      "loss": 1.7538,
      "theoretical_loss": 3.9715382206935814,
      "tokens_seen": 467927040
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008668605817452358,
      "loss": 1.6245,
      "theoretical_loss": 3.971478062371215,
      "tokens_seen": 467992576
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000866840521564694,
      "loss": 1.5672,
      "theoretical_loss": 3.9714179148310587,
      "tokens_seen": 468058112
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008668204613841525,
      "loss": 1.59,
      "theoretical_loss": 3.971357778069668,
      "tokens_seen": 468123648
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008668004012036108,
      "loss": 1.6417,
      "theoretical_loss": 3.9712976520836043,
      "tokens_seen": 468189184
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008667803410230692,
      "loss": 1.6598,
      "theoretical_loss": 3.971237536869429,
      "tokens_seen": 468254720
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008667602808425276,
      "loss": 1.645,
      "theoretical_loss": 3.971177432423705,
      "tokens_seen": 468320256
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000866740220661986,
      "loss": 1.6404,
      "theoretical_loss": 3.9711173387429954,
      "tokens_seen": 468385792
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008667201604814443,
      "loss": 1.6616,
      "theoretical_loss": 3.971057255823868,
      "tokens_seen": 468451328
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008667001003009028,
      "loss": 1.5759,
      "theoretical_loss": 3.9709971836628903,
      "tokens_seen": 468516864
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.49590790271759033,
      "objective/train/docs_used": 176758,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7070120573043823,
      "objective/train/original_loss": 1.7070120573043823,
      "objective/train/theoretical_loss": 3.9709371222566308,
      "objective/train/tokens_used": 489042400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24609065055847168,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0508419275283813,
      "objective/train/weighted_lm_loss": 1.793755054473877,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9709371222566308,
      "tokens_seen": 468582400
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000866680040120361,
      "loss": 1.7491,
      "theoretical_loss": 3.9709371222566308,
      "tokens_seen": 468582400
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008666599799398195,
      "loss": 1.6753,
      "theoretical_loss": 3.9708770716016613,
      "tokens_seen": 468647936
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008666399197592778,
      "loss": 1.7408,
      "theoretical_loss": 3.9708170316945526,
      "tokens_seen": 468713472
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008666198595787362,
      "loss": 1.7188,
      "theoretical_loss": 3.9707570025318804,
      "tokens_seen": 468779008
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008665997993981946,
      "loss": 1.6429,
      "theoretical_loss": 3.9706969841102193,
      "tokens_seen": 468844544
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008665797392176529,
      "loss": 1.6667,
      "theoretical_loss": 3.970636976426147,
      "tokens_seen": 468910080
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008665596790371113,
      "loss": 1.6802,
      "theoretical_loss": 3.9705769794762418,
      "tokens_seen": 468975616
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008665396188565698,
      "loss": 1.719,
      "theoretical_loss": 3.9705169932570836,
      "tokens_seen": 469041152
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000866519558676028,
      "loss": 1.6971,
      "theoretical_loss": 3.9704570177652547,
      "tokens_seen": 469106688
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008664994984954865,
      "loss": 1.7641,
      "theoretical_loss": 3.9703970529973387,
      "tokens_seen": 469172224
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008664794383149449,
      "loss": 1.6095,
      "theoretical_loss": 3.97033709894992,
      "tokens_seen": 469237760
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008664593781344032,
      "loss": 1.678,
      "theoretical_loss": 3.970277155619586,
      "tokens_seen": 469303296
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008664393179538616,
      "loss": 1.6985,
      "theoretical_loss": 3.9702172230029236,
      "tokens_seen": 469368832
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008664192577733199,
      "loss": 1.686,
      "theoretical_loss": 3.970157301096523,
      "tokens_seen": 469434368
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008663991975927783,
      "loss": 1.5955,
      "theoretical_loss": 3.970097389896975,
      "tokens_seen": 469499904
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008663791374122368,
      "loss": 1.6344,
      "theoretical_loss": 3.9700374894008736,
      "tokens_seen": 469565440
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008663590772316951,
      "loss": 1.7155,
      "theoretical_loss": 3.969977599604812,
      "tokens_seen": 469630976
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008663390170511535,
      "loss": 1.7049,
      "theoretical_loss": 3.969917720505386,
      "tokens_seen": 469696512
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008663189568706119,
      "loss": 1.669,
      "theoretical_loss": 3.9698578520991936,
      "tokens_seen": 469762048
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008662988966900702,
      "loss": 1.7185,
      "theoretical_loss": 3.9697979943828337,
      "tokens_seen": 469827584
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008662788365095287,
      "loss": 1.6554,
      "theoretical_loss": 3.969738147352907,
      "tokens_seen": 469893120
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000866258776328987,
      "loss": 1.7524,
      "theoretical_loss": 3.9696783110060148,
      "tokens_seen": 469958656
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008662387161484454,
      "loss": 1.7266,
      "theoretical_loss": 3.9696184853387617,
      "tokens_seen": 470024192
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008662186559679037,
      "loss": 1.6859,
      "theoretical_loss": 3.9695586703477526,
      "tokens_seen": 470089728
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008661985957873621,
      "loss": 1.5995,
      "theoretical_loss": 3.9694988660295936,
      "tokens_seen": 470155264
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008661785356068205,
      "loss": 1.6525,
      "theoretical_loss": 3.9694390723808946,
      "tokens_seen": 470220800
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008661584754262789,
      "loss": 1.6127,
      "theoretical_loss": 3.9693792893982636,
      "tokens_seen": 470286336
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008661384152457372,
      "loss": 1.6209,
      "theoretical_loss": 3.969319517078313,
      "tokens_seen": 470351872
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008661183550651957,
      "loss": 1.7191,
      "theoretical_loss": 3.969259755417656,
      "tokens_seen": 470417408
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000866098294884654,
      "loss": 1.7615,
      "theoretical_loss": 3.969200004412907,
      "tokens_seen": 470482944
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008660782347041124,
      "loss": 1.6576,
      "theoretical_loss": 3.969140264060681,
      "tokens_seen": 470548480
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008660581745235707,
      "loss": 1.5596,
      "theoretical_loss": 3.9690805343575972,
      "tokens_seen": 470614016
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008660381143430291,
      "loss": 1.7647,
      "theoretical_loss": 3.969020815300274,
      "tokens_seen": 470679552
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008660180541624875,
      "loss": 1.6108,
      "theoretical_loss": 3.968961106885332,
      "tokens_seen": 470745088
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008659979939819459,
      "loss": 1.6862,
      "theoretical_loss": 3.9689014091093933,
      "tokens_seen": 470810624
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008659779338014042,
      "loss": 1.684,
      "theoretical_loss": 3.9688417219690817,
      "tokens_seen": 470876160
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008659578736208626,
      "loss": 1.6453,
      "theoretical_loss": 3.9687820454610234,
      "tokens_seen": 470941696
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000865937813440321,
      "loss": 1.6115,
      "theoretical_loss": 3.9687223795818443,
      "tokens_seen": 471007232
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008659177532597794,
      "loss": 1.6774,
      "theoretical_loss": 3.9686627243281727,
      "tokens_seen": 471072768
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008658976930792377,
      "loss": 1.7675,
      "theoretical_loss": 3.9686030796966394,
      "tokens_seen": 471138304
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008658776328986962,
      "loss": 1.7617,
      "theoretical_loss": 3.968543445683875,
      "tokens_seen": 471203840
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008658575727181544,
      "loss": 1.6725,
      "theoretical_loss": 3.9684838222865135,
      "tokens_seen": 471269376
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008658375125376129,
      "loss": 1.7021,
      "theoretical_loss": 3.9684242095011886,
      "tokens_seen": 471334912
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008658174523570712,
      "loss": 1.5613,
      "theoretical_loss": 3.9683646073245367,
      "tokens_seen": 471400448
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008657973921765296,
      "loss": 1.6433,
      "theoretical_loss": 3.968305015753195,
      "tokens_seen": 471465984
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000865777331995988,
      "loss": 1.7898,
      "theoretical_loss": 3.968245434783803,
      "tokens_seen": 471531520
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008657572718154464,
      "loss": 1.6126,
      "theoretical_loss": 3.9681858644130017,
      "tokens_seen": 471597056
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008657372116349047,
      "loss": 1.7728,
      "theoretical_loss": 3.9681263046374333,
      "tokens_seen": 471662592
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008657171514543632,
      "loss": 1.693,
      "theoretical_loss": 3.968066755453741,
      "tokens_seen": 471728128
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008656970912738214,
      "loss": 1.6555,
      "theoretical_loss": 3.9680072168585703,
      "tokens_seen": 471793664
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.4907188415527344,
      "objective/train/docs_used": 177948,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6511681079864502,
      "objective/train/original_loss": 1.6511683464050293,
      "objective/train/theoretical_loss": 3.967947688848568,
      "objective/train/tokens_used": 492319200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2415388971567154,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502997636795044,
      "objective/train/weighted_lm_loss": 1.7343614101409912,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.967947688848568,
      "tokens_seen": 471859200
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008656770310932799,
      "loss": 1.6961,
      "theoretical_loss": 3.967947688848568,
      "tokens_seen": 471859200
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008656569709127382,
      "loss": 1.5168,
      "theoretical_loss": 3.967888171420382,
      "tokens_seen": 471924736
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008656369107321966,
      "loss": 1.6529,
      "theoretical_loss": 3.9678286645706637,
      "tokens_seen": 471990272
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000865616850551655,
      "loss": 1.5362,
      "theoretical_loss": 3.967769168296063,
      "tokens_seen": 472055808
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008655967903711133,
      "loss": 1.6113,
      "theoretical_loss": 3.9677096825932328,
      "tokens_seen": 472121344
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008655767301905717,
      "loss": 1.5861,
      "theoretical_loss": 3.967650207458828,
      "tokens_seen": 472186880
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008655566700100302,
      "loss": 1.7183,
      "theoretical_loss": 3.9675907428895045,
      "tokens_seen": 472252416
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008655366098294884,
      "loss": 1.678,
      "theoretical_loss": 3.9675312888819194,
      "tokens_seen": 472317952
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008655165496489469,
      "loss": 1.605,
      "theoretical_loss": 3.9674718454327325,
      "tokens_seen": 472383488
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008654964894684053,
      "loss": 1.7314,
      "theoretical_loss": 3.967412412538604,
      "tokens_seen": 472449024
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008654764292878636,
      "loss": 1.688,
      "theoretical_loss": 3.9673529901961957,
      "tokens_seen": 472514560
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000865456369107322,
      "loss": 1.6936,
      "theoretical_loss": 3.9672935784021712,
      "tokens_seen": 472580096
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008654363089267803,
      "loss": 1.6403,
      "theoretical_loss": 3.9672341771531956,
      "tokens_seen": 472645632
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008654162487462387,
      "loss": 1.7436,
      "theoretical_loss": 3.967174786445935,
      "tokens_seen": 472711168
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008653961885656972,
      "loss": 1.6937,
      "theoretical_loss": 3.967115406277059,
      "tokens_seen": 472776704
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008653761283851554,
      "loss": 1.5322,
      "theoretical_loss": 3.9670560366432355,
      "tokens_seen": 472842240
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008653560682046139,
      "loss": 1.8065,
      "theoretical_loss": 3.9669966775411365,
      "tokens_seen": 472907776
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008653360080240722,
      "loss": 1.6547,
      "theoretical_loss": 3.9669373289674343,
      "tokens_seen": 472973312
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008653159478435306,
      "loss": 1.6371,
      "theoretical_loss": 3.966877990918804,
      "tokens_seen": 473038848
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000865295887662989,
      "loss": 1.5539,
      "theoretical_loss": 3.96681866339192,
      "tokens_seen": 473104384
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008652758274824473,
      "loss": 1.747,
      "theoretical_loss": 3.96675934638346,
      "tokens_seen": 473169920
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008652557673019057,
      "loss": 1.6029,
      "theoretical_loss": 3.9667000398901022,
      "tokens_seen": 473235456
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008652357071213641,
      "loss": 1.7309,
      "theoretical_loss": 3.966640743908528,
      "tokens_seen": 473300992
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008652156469408224,
      "loss": 1.7987,
      "theoretical_loss": 3.966581458435418,
      "tokens_seen": 473366528
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008651955867602809,
      "loss": 1.6473,
      "theoretical_loss": 3.9665221834674558,
      "tokens_seen": 473432064
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008651755265797392,
      "loss": 1.6844,
      "theoretical_loss": 3.9664629190013265,
      "tokens_seen": 473497600
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008651554663991976,
      "loss": 1.7004,
      "theoretical_loss": 3.9664036650337158,
      "tokens_seen": 473563136
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000865135406218656,
      "loss": 1.5744,
      "theoretical_loss": 3.9663444215613115,
      "tokens_seen": 473628672
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008651153460381144,
      "loss": 1.6554,
      "theoretical_loss": 3.9662851885808026,
      "tokens_seen": 473694208
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008650952858575727,
      "loss": 1.6671,
      "theoretical_loss": 3.9662259660888806,
      "tokens_seen": 473759744
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008650752256770311,
      "loss": 1.7138,
      "theoretical_loss": 3.9661667540822365,
      "tokens_seen": 473825280
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008650551654964894,
      "loss": 1.606,
      "theoretical_loss": 3.9661075525575655,
      "tokens_seen": 473890816
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008650351053159479,
      "loss": 1.6872,
      "theoretical_loss": 3.966048361511562,
      "tokens_seen": 473956352
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008650150451354062,
      "loss": 1.5797,
      "theoretical_loss": 3.9659891809409222,
      "tokens_seen": 474021888
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008649949849548646,
      "loss": 1.704,
      "theoretical_loss": 3.9659300108423454,
      "tokens_seen": 474087424
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008649749247743229,
      "loss": 1.6559,
      "theoretical_loss": 3.965870851212531,
      "tokens_seen": 474152960
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008649548645937814,
      "loss": 1.5749,
      "theoretical_loss": 3.96581170204818,
      "tokens_seen": 474218496
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008649348044132397,
      "loss": 1.7093,
      "theoretical_loss": 3.965752563345995,
      "tokens_seen": 474284032
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008649147442326981,
      "loss": 1.7509,
      "theoretical_loss": 3.965693435102681,
      "tokens_seen": 474349568
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008648946840521564,
      "loss": 1.6831,
      "theoretical_loss": 3.9656343173149424,
      "tokens_seen": 474415104
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008648746238716148,
      "loss": 1.7082,
      "theoretical_loss": 3.9655752099794874,
      "tokens_seen": 474480640
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008648545636910733,
      "loss": 1.7715,
      "theoretical_loss": 3.9655161130930248,
      "tokens_seen": 474546176
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008648345035105316,
      "loss": 1.6436,
      "theoretical_loss": 3.965457026652264,
      "tokens_seen": 474611712
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.00086481444332999,
      "loss": 1.8402,
      "theoretical_loss": 3.965397950653917,
      "tokens_seen": 474677248
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008647943831494484,
      "loss": 1.7173,
      "theoretical_loss": 3.9653388850946976,
      "tokens_seen": 474742784
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008647743229689068,
      "loss": 1.631,
      "theoretical_loss": 3.9652798299713194,
      "tokens_seen": 474808320
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008647542627883651,
      "loss": 1.5619,
      "theoretical_loss": 3.965220785280499,
      "tokens_seen": 474873856
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008647342026078236,
      "loss": 1.7494,
      "theoretical_loss": 3.9651617510189543,
      "tokens_seen": 474939392
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008647141424272818,
      "loss": 1.6467,
      "theoretical_loss": 3.965102727183404,
      "tokens_seen": 475004928
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008646940822467403,
      "loss": 1.7128,
      "theoretical_loss": 3.9650437137705685,
      "tokens_seen": 475070464
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.4953663647174835,
      "objective/train/docs_used": 179087,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4755969047546387,
      "objective/train/original_loss": 1.4755969047546387,
      "objective/train/theoretical_loss": 3.9649847107771707,
      "objective/train/tokens_used": 495596000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24567000567913055,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507856607437134,
      "objective/train/weighted_lm_loss": 1.550560712814331,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9649847107771707,
      "tokens_seen": 475136000
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008646740220661986,
      "loss": 1.556,
      "theoretical_loss": 3.9649847107771707,
      "tokens_seen": 475136000
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000864653961885657,
      "loss": 1.625,
      "theoretical_loss": 3.9649257181999333,
      "tokens_seen": 475201536
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008646339017051154,
      "loss": 1.643,
      "theoretical_loss": 3.9648667360355816,
      "tokens_seen": 475267072
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008646138415245737,
      "loss": 1.6444,
      "theoretical_loss": 3.9648077642808426,
      "tokens_seen": 475332608
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008645937813440321,
      "loss": 1.6659,
      "theoretical_loss": 3.9647488029324434,
      "tokens_seen": 475398144
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008645737211634906,
      "loss": 1.8248,
      "theoretical_loss": 3.9646898519871145,
      "tokens_seen": 475463680
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008645536609829488,
      "loss": 1.6969,
      "theoretical_loss": 3.9646309114415863,
      "tokens_seen": 475529216
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008645336008024073,
      "loss": 1.7494,
      "theoretical_loss": 3.9645719812925906,
      "tokens_seen": 475594752
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008645135406218657,
      "loss": 1.6244,
      "theoretical_loss": 3.964513061536863,
      "tokens_seen": 475660288
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000864493480441324,
      "loss": 1.6211,
      "theoretical_loss": 3.964454152171137,
      "tokens_seen": 475725824
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008644734202607824,
      "loss": 1.5774,
      "theoretical_loss": 3.9643952531921505,
      "tokens_seen": 475791360
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008644533600802407,
      "loss": 1.7316,
      "theoretical_loss": 3.964336364596642,
      "tokens_seen": 475856896
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008644332998996991,
      "loss": 1.5758,
      "theoretical_loss": 3.964277486381351,
      "tokens_seen": 475922432
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008644132397191576,
      "loss": 1.5914,
      "theoretical_loss": 3.9642186185430184,
      "tokens_seen": 475987968
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008643931795386158,
      "loss": 1.4938,
      "theoretical_loss": 3.9641597610783874,
      "tokens_seen": 476053504
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008643731193580743,
      "loss": 1.6912,
      "theoretical_loss": 3.9641009139842023,
      "tokens_seen": 476119040
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008643530591775326,
      "loss": 1.647,
      "theoretical_loss": 3.9640420772572087,
      "tokens_seen": 476184576
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000864332998996991,
      "loss": 1.7011,
      "theoretical_loss": 3.9639832508941533,
      "tokens_seen": 476250112
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008643129388164494,
      "loss": 1.6084,
      "theoretical_loss": 3.9639244348917853,
      "tokens_seen": 476315648
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008642928786359077,
      "loss": 1.5955,
      "theoretical_loss": 3.963865629246855,
      "tokens_seen": 476381184
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008642728184553661,
      "loss": 1.5804,
      "theoretical_loss": 3.963806833956113,
      "tokens_seen": 476446720
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008642527582748245,
      "loss": 1.6083,
      "theoretical_loss": 3.9637480490163135,
      "tokens_seen": 476512256
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008642326980942828,
      "loss": 1.6791,
      "theoretical_loss": 3.96368927442421,
      "tokens_seen": 476577792
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008642126379137413,
      "loss": 1.8064,
      "theoretical_loss": 3.9636305101765585,
      "tokens_seen": 476643328
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008641925777331996,
      "loss": 1.5492,
      "theoretical_loss": 3.9635717562701176,
      "tokens_seen": 476708864
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000864172517552658,
      "loss": 1.7523,
      "theoretical_loss": 3.963513012701645,
      "tokens_seen": 476774400
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008641524573721164,
      "loss": 1.581,
      "theoretical_loss": 3.9634542794679013,
      "tokens_seen": 476839936
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008641323971915748,
      "loss": 1.667,
      "theoretical_loss": 3.963395556565649,
      "tokens_seen": 476905472
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008641123370110331,
      "loss": 1.5052,
      "theoretical_loss": 3.96333684399165,
      "tokens_seen": 476971008
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008640922768304915,
      "loss": 1.6253,
      "theoretical_loss": 3.9632781417426703,
      "tokens_seen": 477036544
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008640722166499498,
      "loss": 1.7274,
      "theoretical_loss": 3.963219449815475,
      "tokens_seen": 477102080
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008640521564694083,
      "loss": 1.4946,
      "theoretical_loss": 3.9631607682068335,
      "tokens_seen": 477167616
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008640320962888666,
      "loss": 1.6131,
      "theoretical_loss": 3.963102096913513,
      "tokens_seen": 477233152
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000864012036108325,
      "loss": 1.6323,
      "theoretical_loss": 3.963043435932285,
      "tokens_seen": 477298688
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008639919759277833,
      "loss": 1.5629,
      "theoretical_loss": 3.9629847852599207,
      "tokens_seen": 477364224
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008639719157472418,
      "loss": 1.6746,
      "theoretical_loss": 3.9629261448931943,
      "tokens_seen": 477429760
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008639518555667001,
      "loss": 1.7605,
      "theoretical_loss": 3.962867514828881,
      "tokens_seen": 477495296
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008639317953861585,
      "loss": 1.7014,
      "theoretical_loss": 3.9628088950637563,
      "tokens_seen": 477560832
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008639117352056168,
      "loss": 1.7168,
      "theoretical_loss": 3.9627502855945984,
      "tokens_seen": 477626368
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008638916750250752,
      "loss": 1.674,
      "theoretical_loss": 3.9626916864181867,
      "tokens_seen": 477691904
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008638716148445336,
      "loss": 1.6337,
      "theoretical_loss": 3.962633097531302,
      "tokens_seen": 477757440
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000863851554663992,
      "loss": 1.5767,
      "theoretical_loss": 3.962574518930726,
      "tokens_seen": 477822976
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008638314944834503,
      "loss": 1.6741,
      "theoretical_loss": 3.962515950613242,
      "tokens_seen": 477888512
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008638114343029088,
      "loss": 1.7077,
      "theoretical_loss": 3.9624573925756366,
      "tokens_seen": 477954048
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008637913741223671,
      "loss": 1.7455,
      "theoretical_loss": 3.962398844814695,
      "tokens_seen": 478019584
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008637713139418255,
      "loss": 1.6796,
      "theoretical_loss": 3.9623403073272048,
      "tokens_seen": 478085120
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008637512537612839,
      "loss": 1.6243,
      "theoretical_loss": 3.962281780109957,
      "tokens_seen": 478150656
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008637311935807422,
      "loss": 1.6601,
      "theoretical_loss": 3.962223263159741,
      "tokens_seen": 478216192
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008637111334002006,
      "loss": 1.5535,
      "theoretical_loss": 3.9621647564733493,
      "tokens_seen": 478281728
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.000863691073219659,
      "loss": 1.6916,
      "theoretical_loss": 3.9621062600475763,
      "tokens_seen": 478347264
    },
    {
      "epoch": 0.14,
      "objective/train/advantage_avg": 0.49374765157699585,
      "objective/train/docs_used": 180329,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6273490190505981,
      "objective/train/original_loss": 1.627348780632019,
      "objective/train/theoretical_loss": 3.9620477738792164,
      "objective/train/tokens_used": 498872800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24403226375579834,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506154298782349,
      "objective/train/weighted_lm_loss": 1.7097764015197754,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 3.9620477738792164,
      "tokens_seen": 478412800
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008636710130391173,
      "loss": 1.7097,
      "theoretical_loss": 3.9620477738792164,
      "tokens_seen": 478412800
    },
    {
      "epoch": 0.14,
      "learning_rate": 0.0008636509528585758,
      "loss": 1.7818,
      "theoretical_loss": 3.9619892979650664,
      "tokens_seen": 478478336
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000863630892678034,
      "loss": 1.5919,
      "theoretical_loss": 3.9619308323019244,
      "tokens_seen": 478543872
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008636108324974925,
      "loss": 1.6456,
      "theoretical_loss": 3.96187237688659,
      "tokens_seen": 478609408
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008635907723169509,
      "loss": 1.6536,
      "theoretical_loss": 3.9618139317158647,
      "tokens_seen": 478674944
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008635707121364092,
      "loss": 1.7279,
      "theoretical_loss": 3.96175549678655,
      "tokens_seen": 478740480
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008635506519558676,
      "loss": 1.6284,
      "theoretical_loss": 3.9616970720954487,
      "tokens_seen": 478806016
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008635305917753261,
      "loss": 1.6787,
      "theoretical_loss": 3.9616386576393676,
      "tokens_seen": 478871552
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008635105315947843,
      "loss": 1.5603,
      "theoretical_loss": 3.961580253415114,
      "tokens_seen": 478937088
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008634904714142428,
      "loss": 1.5645,
      "theoretical_loss": 3.9615218594194936,
      "tokens_seen": 479002624
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000863470411233701,
      "loss": 1.7041,
      "theoretical_loss": 3.961463475649318,
      "tokens_seen": 479068160
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008634503510531595,
      "loss": 1.6872,
      "theoretical_loss": 3.9614051021013967,
      "tokens_seen": 479133696
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008634302908726179,
      "loss": 1.6363,
      "theoretical_loss": 3.9613467387725434,
      "tokens_seen": 479199232
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008634102306920762,
      "loss": 1.491,
      "theoretical_loss": 3.961288385659571,
      "tokens_seen": 479264768
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008633901705115346,
      "loss": 1.7415,
      "theoretical_loss": 3.961230042759295,
      "tokens_seen": 479330304
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000863370110330993,
      "loss": 1.6011,
      "theoretical_loss": 3.961171710068532,
      "tokens_seen": 479395840
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008633500501504514,
      "loss": 1.5091,
      "theoretical_loss": 3.9611133875841,
      "tokens_seen": 479461376
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008633299899699098,
      "loss": 1.669,
      "theoretical_loss": 3.961055075302819,
      "tokens_seen": 479526912
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008633099297893681,
      "loss": 1.644,
      "theoretical_loss": 3.9609967732215097,
      "tokens_seen": 479592448
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008632898696088265,
      "loss": 1.7018,
      "theoretical_loss": 3.960938481336994,
      "tokens_seen": 479657984
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008632698094282849,
      "loss": 1.6864,
      "theoretical_loss": 3.960880199646096,
      "tokens_seen": 479723520
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008632497492477432,
      "loss": 1.7271,
      "theoretical_loss": 3.9608219281456414,
      "tokens_seen": 479789056
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008632296890672017,
      "loss": 1.6468,
      "theoretical_loss": 3.960763666832456,
      "tokens_seen": 479854592
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.00086320962888666,
      "loss": 1.5897,
      "theoretical_loss": 3.9607054157033685,
      "tokens_seen": 479920128
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008631895687061184,
      "loss": 1.8161,
      "theoretical_loss": 3.9606471747552083,
      "tokens_seen": 479985664
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008631695085255768,
      "loss": 1.7091,
      "theoretical_loss": 3.960588943984806,
      "tokens_seen": 480051200
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008631494483450352,
      "loss": 1.5867,
      "theoretical_loss": 3.960530723388994,
      "tokens_seen": 480116736
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008631293881644935,
      "loss": 1.6212,
      "theoretical_loss": 3.960472512964606,
      "tokens_seen": 480182272
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008631093279839519,
      "loss": 1.578,
      "theoretical_loss": 3.9604143127084774,
      "tokens_seen": 480247808
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008630892678034102,
      "loss": 1.6746,
      "theoretical_loss": 3.960356122617445,
      "tokens_seen": 480313344
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008630692076228687,
      "loss": 1.6223,
      "theoretical_loss": 3.9602979426883462,
      "tokens_seen": 480378880
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000863049147442327,
      "loss": 1.7078,
      "theoretical_loss": 3.9602397729180208,
      "tokens_seen": 480444416
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008630290872617854,
      "loss": 1.6483,
      "theoretical_loss": 3.960181613303309,
      "tokens_seen": 480509952
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008630090270812437,
      "loss": 1.7556,
      "theoretical_loss": 3.960123463841054,
      "tokens_seen": 480575488
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008629889669007022,
      "loss": 1.7327,
      "theoretical_loss": 3.9600653245280983,
      "tokens_seen": 480641024
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008629689067201605,
      "loss": 1.7339,
      "theoretical_loss": 3.960007195361288,
      "tokens_seen": 480706560
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008629488465396189,
      "loss": 1.6611,
      "theoretical_loss": 3.9599490763374687,
      "tokens_seen": 480772096
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008629287863590772,
      "loss": 1.7667,
      "theoretical_loss": 3.9598909674534895,
      "tokens_seen": 480837632
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008629087261785356,
      "loss": 1.7056,
      "theoretical_loss": 3.9598328687061985,
      "tokens_seen": 480903168
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000862888665997994,
      "loss": 1.6263,
      "theoretical_loss": 3.959774780092447,
      "tokens_seen": 480968704
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008628686058174524,
      "loss": 1.628,
      "theoretical_loss": 3.9597167016090866,
      "tokens_seen": 481034240
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008628485456369107,
      "loss": 1.5508,
      "theoretical_loss": 3.959658633252971,
      "tokens_seen": 481099776
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008628284854563692,
      "loss": 1.6159,
      "theoretical_loss": 3.9596005750209553,
      "tokens_seen": 481165312
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008628084252758275,
      "loss": 1.6213,
      "theoretical_loss": 3.9595425269098956,
      "tokens_seen": 481230848
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008627883650952859,
      "loss": 1.7252,
      "theoretical_loss": 3.95948448891665,
      "tokens_seen": 481296384
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008627683049147443,
      "loss": 1.6368,
      "theoretical_loss": 3.959426461038077,
      "tokens_seen": 481361920
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008627482447342026,
      "loss": 1.689,
      "theoretical_loss": 3.9593684432710377,
      "tokens_seen": 481427456
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000862728184553661,
      "loss": 1.6588,
      "theoretical_loss": 3.9593104356123936,
      "tokens_seen": 481492992
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008627081243731195,
      "loss": 1.6145,
      "theoretical_loss": 3.959252438059009,
      "tokens_seen": 481558528
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008626880641925777,
      "loss": 1.609,
      "theoretical_loss": 3.9591944506077468,
      "tokens_seen": 481624064
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.492854505777359,
      "objective/train/docs_used": 181505,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5590598583221436,
      "objective/train/original_loss": 1.559059977531433,
      "objective/train/theoretical_loss": 3.959136473255475,
      "objective/train/tokens_used": 502149600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2436048984527588,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505238771438599,
      "objective/train/weighted_lm_loss": 1.6377798318862915,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.8607079386711121,
      "theoretical_loss": 3.959136473255475,
      "tokens_seen": 481689600
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008626680040120362,
      "loss": 1.6004,
      "theoretical_loss": 3.959136473255475,
      "tokens_seen": 481689600
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008626479438314944,
      "loss": 1.7456,
      "theoretical_loss": 3.95907850599906,
      "tokens_seen": 481755136
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008626278836509529,
      "loss": 1.6157,
      "theoretical_loss": 3.9590205488353707,
      "tokens_seen": 481820672
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008626078234704113,
      "loss": 1.673,
      "theoretical_loss": 3.9589626017612782,
      "tokens_seen": 481886208
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008625877632898696,
      "loss": 1.7322,
      "theoretical_loss": 3.9589046647736534,
      "tokens_seen": 481951744
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000862567703109328,
      "loss": 1.6756,
      "theoretical_loss": 3.9588467378693704,
      "tokens_seen": 482017280
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008625476429287865,
      "loss": 1.7392,
      "theoretical_loss": 3.958788821045302,
      "tokens_seen": 482082816
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008625275827482447,
      "loss": 1.6884,
      "theoretical_loss": 3.958730914298326,
      "tokens_seen": 482148352
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008625075225677032,
      "loss": 1.7627,
      "theoretical_loss": 3.958673017625319,
      "tokens_seen": 482213888
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008624874623871614,
      "loss": 1.6483,
      "theoretical_loss": 3.958615131023159,
      "tokens_seen": 482279424
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008624674022066199,
      "loss": 1.5586,
      "theoretical_loss": 3.958557254488727,
      "tokens_seen": 482344960
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008624473420260783,
      "loss": 1.6014,
      "theoretical_loss": 3.958499388018904,
      "tokens_seen": 482410496
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008624272818455366,
      "loss": 1.6735,
      "theoretical_loss": 3.958441531610573,
      "tokens_seen": 482476032
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000862407221664995,
      "loss": 1.6801,
      "theoretical_loss": 3.958383685260618,
      "tokens_seen": 482541568
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008623871614844534,
      "loss": 1.6305,
      "theoretical_loss": 3.958325848965925,
      "tokens_seen": 482607104
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008623671013039117,
      "loss": 1.683,
      "theoretical_loss": 3.958268022723381,
      "tokens_seen": 482672640
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008623470411233702,
      "loss": 1.6403,
      "theoretical_loss": 3.9582102065298734,
      "tokens_seen": 482738176
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008623269809428284,
      "loss": 1.5577,
      "theoretical_loss": 3.9581524003822937,
      "tokens_seen": 482803712
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008623069207622869,
      "loss": 1.64,
      "theoretical_loss": 3.958094604277532,
      "tokens_seen": 482869248
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008622868605817452,
      "loss": 1.6997,
      "theoretical_loss": 3.9580368182124808,
      "tokens_seen": 482934784
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008622668004012036,
      "loss": 1.7209,
      "theoretical_loss": 3.957979042184035,
      "tokens_seen": 483000320
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000862246740220662,
      "loss": 1.6709,
      "theoretical_loss": 3.9579212761890883,
      "tokens_seen": 483065856
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008622266800401204,
      "loss": 1.6907,
      "theoretical_loss": 3.9578635202245387,
      "tokens_seen": 483131392
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008622066198595787,
      "loss": 1.6686,
      "theoretical_loss": 3.957805774287284,
      "tokens_seen": 483196928
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008621865596790372,
      "loss": 1.4088,
      "theoretical_loss": 3.957748038374224,
      "tokens_seen": 483262464
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008621664994984954,
      "loss": 1.6271,
      "theoretical_loss": 3.957690312482258,
      "tokens_seen": 483328000
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008621464393179539,
      "loss": 1.6111,
      "theoretical_loss": 3.95763259660829,
      "tokens_seen": 483393536
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008621263791374122,
      "loss": 1.6555,
      "theoretical_loss": 3.957574890749223,
      "tokens_seen": 483459072
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008621063189568706,
      "loss": 1.5846,
      "theoretical_loss": 3.9575171949019623,
      "tokens_seen": 483524608
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000862086258776329,
      "loss": 1.5948,
      "theoretical_loss": 3.957459509063413,
      "tokens_seen": 483590144
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008620661985957874,
      "loss": 1.6619,
      "theoretical_loss": 3.957401833230484,
      "tokens_seen": 483655680
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008620461384152457,
      "loss": 1.6941,
      "theoretical_loss": 3.9573441674000844,
      "tokens_seen": 483721216
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008620260782347041,
      "loss": 1.6845,
      "theoretical_loss": 3.9572865115691243,
      "tokens_seen": 483786752
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008620060180541625,
      "loss": 1.5978,
      "theoretical_loss": 3.9572288657345154,
      "tokens_seen": 483852288
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008619859578736209,
      "loss": 1.5465,
      "theoretical_loss": 3.957171229893171,
      "tokens_seen": 483917824
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008619658976930792,
      "loss": 1.6417,
      "theoretical_loss": 3.9571136040420063,
      "tokens_seen": 483983360
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008619458375125376,
      "loss": 1.673,
      "theoretical_loss": 3.9570559881779364,
      "tokens_seen": 484048896
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008619257773319959,
      "loss": 1.7413,
      "theoretical_loss": 3.956998382297879,
      "tokens_seen": 484114432
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008619057171514544,
      "loss": 1.5772,
      "theoretical_loss": 3.956940786398753,
      "tokens_seen": 484179968
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008618856569709127,
      "loss": 1.6466,
      "theoretical_loss": 3.9568832004774785,
      "tokens_seen": 484245504
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008618655967903711,
      "loss": 1.5917,
      "theoretical_loss": 3.9568256245309765,
      "tokens_seen": 484311040
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008618455366098296,
      "loss": 1.6796,
      "theoretical_loss": 3.9567680585561695,
      "tokens_seen": 484376576
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008618254764292879,
      "loss": 1.7908,
      "theoretical_loss": 3.9567105025499827,
      "tokens_seen": 484442112
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008618054162487463,
      "loss": 1.5868,
      "theoretical_loss": 3.956652956509341,
      "tokens_seen": 484507648
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008617853560682047,
      "loss": 1.6694,
      "theoretical_loss": 3.9565954204311717,
      "tokens_seen": 484573184
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000861765295887663,
      "loss": 1.678,
      "theoretical_loss": 3.9565378943124023,
      "tokens_seen": 484638720
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008617452357071214,
      "loss": 1.6422,
      "theoretical_loss": 3.9564803781499633,
      "tokens_seen": 484704256
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008617251755265799,
      "loss": 1.6743,
      "theoretical_loss": 3.956422871940785,
      "tokens_seen": 484769792
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008617051153460381,
      "loss": 1.6426,
      "theoretical_loss": 3.9563653756818002,
      "tokens_seen": 484835328
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008616850551654966,
      "loss": 1.7416,
      "theoretical_loss": 3.956307889369942,
      "tokens_seen": 484900864
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.49089962244033813,
      "objective/train/docs_used": 182723,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.3967673778533936,
      "objective/train/original_loss": 1.3967671394348145,
      "objective/train/theoretical_loss": 3.956250413002146,
      "objective/train/tokens_used": 505426400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24230317771434784,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503216981887817,
      "objective/train/weighted_lm_loss": 1.4670437574386597,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.956250413002146,
      "tokens_seen": 484966400
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008616649949849548,
      "loss": 1.5712,
      "theoretical_loss": 3.956250413002146,
      "tokens_seen": 484966400
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008616449348044133,
      "loss": 1.6428,
      "theoretical_loss": 3.9561929465753485,
      "tokens_seen": 485031936
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008616248746238717,
      "loss": 1.7151,
      "theoretical_loss": 3.9561354900864876,
      "tokens_seen": 485097472
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.00086160481444333,
      "loss": 1.6437,
      "theoretical_loss": 3.956078043532502,
      "tokens_seen": 485163008
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008615847542627884,
      "loss": 1.6562,
      "theoretical_loss": 3.9560206069103314,
      "tokens_seen": 485228544
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008615646940822469,
      "loss": 1.7284,
      "theoretical_loss": 3.9559631802169193,
      "tokens_seen": 485294080
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008615446339017051,
      "loss": 1.6371,
      "theoretical_loss": 3.955905763449208,
      "tokens_seen": 485359616
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008615245737211636,
      "loss": 1.7452,
      "theoretical_loss": 3.9558483566041422,
      "tokens_seen": 485425152
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008615045135406218,
      "loss": 1.6399,
      "theoretical_loss": 3.9557909596786676,
      "tokens_seen": 485490688
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008614844533600803,
      "loss": 1.6872,
      "theoretical_loss": 3.9557335726697316,
      "tokens_seen": 485556224
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008614643931795387,
      "loss": 1.6266,
      "theoretical_loss": 3.9556761955742834,
      "tokens_seen": 485621760
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000861444332998997,
      "loss": 1.6871,
      "theoretical_loss": 3.955618828389272,
      "tokens_seen": 485687296
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008614242728184554,
      "loss": 1.61,
      "theoretical_loss": 3.9555614711116487,
      "tokens_seen": 485752832
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008614042126379138,
      "loss": 1.7205,
      "theoretical_loss": 3.9555041237383675,
      "tokens_seen": 485818368
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008613841524573721,
      "loss": 1.6493,
      "theoretical_loss": 3.9554467862663807,
      "tokens_seen": 485883904
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008613640922768306,
      "loss": 1.5156,
      "theoretical_loss": 3.955389458692645,
      "tokens_seen": 485949440
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008613440320962888,
      "loss": 1.6917,
      "theoretical_loss": 3.9553321410141162,
      "tokens_seen": 486014976
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008613239719157473,
      "loss": 1.6528,
      "theoretical_loss": 3.9552748332277528,
      "tokens_seen": 486080512
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008613039117352056,
      "loss": 1.6826,
      "theoretical_loss": 3.9552175353305135,
      "tokens_seen": 486146048
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000861283851554664,
      "loss": 1.6517,
      "theoretical_loss": 3.9551602473193603,
      "tokens_seen": 486211584
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008612637913741224,
      "loss": 1.5281,
      "theoretical_loss": 3.9551029691912545,
      "tokens_seen": 486277120
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008612437311935808,
      "loss": 1.8018,
      "theoretical_loss": 3.955045700943159,
      "tokens_seen": 486342656
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008612236710130391,
      "loss": 1.6226,
      "theoretical_loss": 3.95498844257204,
      "tokens_seen": 486408192
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008612036108324976,
      "loss": 1.6285,
      "theoretical_loss": 3.9549311940748626,
      "tokens_seen": 486473728
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008611835506519558,
      "loss": 1.5453,
      "theoretical_loss": 3.954873955448594,
      "tokens_seen": 486539264
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008611634904714143,
      "loss": 1.6237,
      "theoretical_loss": 3.9548167266902037,
      "tokens_seen": 486604800
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008611434302908726,
      "loss": 1.6427,
      "theoretical_loss": 3.9547595077966617,
      "tokens_seen": 486670336
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000861123370110331,
      "loss": 1.643,
      "theoretical_loss": 3.954702298764939,
      "tokens_seen": 486735872
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008611033099297894,
      "loss": 1.7013,
      "theoretical_loss": 3.9546450995920086,
      "tokens_seen": 486801408
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008610832497492478,
      "loss": 1.6646,
      "theoretical_loss": 3.9545879102748454,
      "tokens_seen": 486866944
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008610631895687061,
      "loss": 1.6002,
      "theoretical_loss": 3.954530730810424,
      "tokens_seen": 486932480
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008610431293881645,
      "loss": 1.5922,
      "theoretical_loss": 3.954473561195721,
      "tokens_seen": 486998016
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008610230692076229,
      "loss": 1.6364,
      "theoretical_loss": 3.954416401427715,
      "tokens_seen": 487063552
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008610030090270813,
      "loss": 1.6667,
      "theoretical_loss": 3.9543592515033854,
      "tokens_seen": 487129088
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008609829488465396,
      "loss": 1.5311,
      "theoretical_loss": 3.9543021114197137,
      "tokens_seen": 487194624
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000860962888665998,
      "loss": 1.5933,
      "theoretical_loss": 3.9542449811736806,
      "tokens_seen": 487260160
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008609428284854563,
      "loss": 1.682,
      "theoretical_loss": 3.9541878607622705,
      "tokens_seen": 487325696
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008609227683049148,
      "loss": 1.5848,
      "theoretical_loss": 3.9541307501824683,
      "tokens_seen": 487391232
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008609027081243731,
      "loss": 1.6445,
      "theoretical_loss": 3.95407364943126,
      "tokens_seen": 487456768
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008608826479438315,
      "loss": 1.5271,
      "theoretical_loss": 3.954016558505633,
      "tokens_seen": 487522304
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008608625877632899,
      "loss": 1.5393,
      "theoretical_loss": 3.9539594774025755,
      "tokens_seen": 487587840
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008608425275827483,
      "loss": 1.5841,
      "theoretical_loss": 3.9539024061190786,
      "tokens_seen": 487653376
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008608224674022066,
      "loss": 1.57,
      "theoretical_loss": 3.953845344652133,
      "tokens_seen": 487718912
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008608024072216651,
      "loss": 1.6846,
      "theoretical_loss": 3.9537882929987322,
      "tokens_seen": 487784448
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008607823470411233,
      "loss": 1.7473,
      "theoretical_loss": 3.9537312511558698,
      "tokens_seen": 487849984
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008607622868605818,
      "loss": 1.5851,
      "theoretical_loss": 3.9536742191205407,
      "tokens_seen": 487915520
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008607422266800401,
      "loss": 1.6263,
      "theoretical_loss": 3.9536171968897427,
      "tokens_seen": 487981056
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008607221664994985,
      "loss": 1.6595,
      "theoretical_loss": 3.953560184460473,
      "tokens_seen": 488046592
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008607021063189569,
      "loss": 1.6056,
      "theoretical_loss": 3.953503181829732,
      "tokens_seen": 488112128
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008606820461384152,
      "loss": 1.6239,
      "theoretical_loss": 3.95344618899452,
      "tokens_seen": 488177664
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.488285094499588,
      "objective/train/docs_used": 183865,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.634871244430542,
      "objective/train/original_loss": 1.634871244430542,
      "objective/train/theoretical_loss": 3.9533892059518383,
      "objective/train/tokens_used": 508703200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.240000382065773,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0500484704971313,
      "objective/train/weighted_lm_loss": 1.7165744304656982,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9533892059518383,
      "tokens_seen": 488243200
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008606619859578736,
      "loss": 1.7296,
      "theoretical_loss": 3.9533892059518383,
      "tokens_seen": 488243200
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008606419257773321,
      "loss": 1.5925,
      "theoretical_loss": 3.95333223269869,
      "tokens_seen": 488308736
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008606218655967903,
      "loss": 1.658,
      "theoretical_loss": 3.9532752692320816,
      "tokens_seen": 488374272
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008606018054162488,
      "loss": 1.6043,
      "theoretical_loss": 3.9532183155490177,
      "tokens_seen": 488439808
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000860581745235707,
      "loss": 1.7125,
      "theoretical_loss": 3.953161371646506,
      "tokens_seen": 488505344
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008605616850551655,
      "loss": 1.4775,
      "theoretical_loss": 3.9531044375215547,
      "tokens_seen": 488570880
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008605416248746239,
      "loss": 1.5956,
      "theoretical_loss": 3.9530475131711746,
      "tokens_seen": 488636416
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008605215646940822,
      "loss": 1.6439,
      "theoretical_loss": 3.9529905985923763,
      "tokens_seen": 488701952
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008605015045135406,
      "loss": 1.5719,
      "theoretical_loss": 3.952933693782172,
      "tokens_seen": 488767488
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008604814443329991,
      "loss": 1.5986,
      "theoretical_loss": 3.9528767987375764,
      "tokens_seen": 488833024
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008604613841524573,
      "loss": 1.6765,
      "theoretical_loss": 3.9528199134556044,
      "tokens_seen": 488898560
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008604413239719158,
      "loss": 1.6545,
      "theoretical_loss": 3.9527630379332725,
      "tokens_seen": 488964096
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000860421263791374,
      "loss": 1.6569,
      "theoretical_loss": 3.9527061721675985,
      "tokens_seen": 489029632
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008604012036108325,
      "loss": 1.6657,
      "theoretical_loss": 3.9526493161556013,
      "tokens_seen": 489095168
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008603811434302909,
      "loss": 1.6495,
      "theoretical_loss": 3.9525924698943022,
      "tokens_seen": 489160704
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008603610832497492,
      "loss": 1.687,
      "theoretical_loss": 3.952535633380722,
      "tokens_seen": 489226240
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008603410230692077,
      "loss": 1.5027,
      "theoretical_loss": 3.9524788066118837,
      "tokens_seen": 489291776
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000860320962888666,
      "loss": 1.6547,
      "theoretical_loss": 3.9524219895848116,
      "tokens_seen": 489357312
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008603009027081244,
      "loss": 1.6899,
      "theoretical_loss": 3.952365182296533,
      "tokens_seen": 489422848
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008602808425275828,
      "loss": 1.6411,
      "theoretical_loss": 3.952308384744073,
      "tokens_seen": 489488384
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008602607823470412,
      "loss": 1.6792,
      "theoretical_loss": 3.9522515969244605,
      "tokens_seen": 489553920
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008602407221664995,
      "loss": 1.5841,
      "theoretical_loss": 3.952194818834725,
      "tokens_seen": 489619456
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000860220661985958,
      "loss": 1.7696,
      "theoretical_loss": 3.9521380504718975,
      "tokens_seen": 489684992
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008602006018054162,
      "loss": 1.7318,
      "theoretical_loss": 3.952081291833011,
      "tokens_seen": 489750528
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008601805416248747,
      "loss": 1.5766,
      "theoretical_loss": 3.9520245429150975,
      "tokens_seen": 489816064
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000860160481444333,
      "loss": 1.6828,
      "theoretical_loss": 3.951967803715193,
      "tokens_seen": 489881600
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008601404212637914,
      "loss": 1.7221,
      "theoretical_loss": 3.9519110742303325,
      "tokens_seen": 489947136
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008601203610832498,
      "loss": 1.7698,
      "theoretical_loss": 3.951854354457554,
      "tokens_seen": 490012672
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008601003009027082,
      "loss": 1.6475,
      "theoretical_loss": 3.951797644393897,
      "tokens_seen": 490078208
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008600802407221665,
      "loss": 1.5595,
      "theoretical_loss": 3.9517409440364,
      "tokens_seen": 490143744
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008600601805416249,
      "loss": 1.5627,
      "theoretical_loss": 3.951684253382105,
      "tokens_seen": 490209280
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008600401203610833,
      "loss": 1.6962,
      "theoretical_loss": 3.9516275724280545,
      "tokens_seen": 490274816
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008600200601805417,
      "loss": 1.6326,
      "theoretical_loss": 3.9515709011712934,
      "tokens_seen": 490340352
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.00086,
      "loss": 1.6999,
      "theoretical_loss": 3.9515142396088647,
      "tokens_seen": 490405888
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008599799398194584,
      "loss": 1.5187,
      "theoretical_loss": 3.951457587737817,
      "tokens_seen": 490471424
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008599598796389167,
      "loss": 1.5951,
      "theoretical_loss": 3.9514009455551973,
      "tokens_seen": 490536960
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008599398194583752,
      "loss": 1.6903,
      "theoretical_loss": 3.951344313058054,
      "tokens_seen": 490602496
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008599197592778335,
      "loss": 1.6521,
      "theoretical_loss": 3.951287690243438,
      "tokens_seen": 490668032
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008598996990972919,
      "loss": 1.6147,
      "theoretical_loss": 3.9512310771084014,
      "tokens_seen": 490733568
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008598796389167503,
      "loss": 1.6771,
      "theoretical_loss": 3.9511744736499965,
      "tokens_seen": 490799104
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008598595787362087,
      "loss": 1.7336,
      "theoretical_loss": 3.951117879865278,
      "tokens_seen": 490864640
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000859839518555667,
      "loss": 1.5961,
      "theoretical_loss": 3.951061295751301,
      "tokens_seen": 490930176
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008598194583751255,
      "loss": 1.6592,
      "theoretical_loss": 3.951004721305123,
      "tokens_seen": 490995712
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008597993981945837,
      "loss": 1.6885,
      "theoretical_loss": 3.9509481565238014,
      "tokens_seen": 491061248
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008597793380140422,
      "loss": 1.655,
      "theoretical_loss": 3.950891601404395,
      "tokens_seen": 491126784
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008597592778335005,
      "loss": 1.5425,
      "theoretical_loss": 3.9508350559439664,
      "tokens_seen": 491192320
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008597392176529589,
      "loss": 1.6547,
      "theoretical_loss": 3.950778520139576,
      "tokens_seen": 491257856
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008597191574724173,
      "loss": 1.6784,
      "theoretical_loss": 3.9507219939882874,
      "tokens_seen": 491323392
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008596990972918756,
      "loss": 1.6538,
      "theoretical_loss": 3.9506654774871652,
      "tokens_seen": 491388928
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000859679037111334,
      "loss": 1.64,
      "theoretical_loss": 3.950608970633275,
      "tokens_seen": 491454464
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.4913667142391205,
      "objective/train/docs_used": 185162,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5327297449111938,
      "objective/train/original_loss": 1.5327297449111938,
      "objective/train/theoretical_loss": 3.9505524734236848,
      "objective/train/tokens_used": 511980000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24189111590385437,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503664016723633,
      "objective/train/weighted_lm_loss": 1.609784483909607,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9753099083900452,
      "theoretical_loss": 3.9505524734236848,
      "tokens_seen": 491520000
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008596589769307925,
      "loss": 1.6483,
      "theoretical_loss": 3.9505524734236848,
      "tokens_seen": 491520000
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008596389167502507,
      "loss": 1.5931,
      "theoretical_loss": 3.9504959858554622,
      "tokens_seen": 491585536
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008596188565697092,
      "loss": 1.5681,
      "theoretical_loss": 3.950439507925677,
      "tokens_seen": 491651072
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008595987963891674,
      "loss": 1.5339,
      "theoretical_loss": 3.9503830396314,
      "tokens_seen": 491716608
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008595787362086259,
      "loss": 1.7121,
      "theoretical_loss": 3.950326580969703,
      "tokens_seen": 491782144
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008595586760280843,
      "loss": 1.6072,
      "theoretical_loss": 3.950270131937661,
      "tokens_seen": 491847680
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008595386158475426,
      "loss": 1.7294,
      "theoretical_loss": 3.9502136925323477,
      "tokens_seen": 491913216
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000859518555667001,
      "loss": 1.7427,
      "theoretical_loss": 3.950157262750839,
      "tokens_seen": 491978752
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008594984954864595,
      "loss": 1.6756,
      "theoretical_loss": 3.950100842590212,
      "tokens_seen": 492044288
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008594784353059177,
      "loss": 1.6451,
      "theoretical_loss": 3.9500444320475467,
      "tokens_seen": 492109824
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008594583751253762,
      "loss": 1.6721,
      "theoretical_loss": 3.949988031119922,
      "tokens_seen": 492175360
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008594383149448344,
      "loss": 1.6257,
      "theoretical_loss": 3.9499316398044195,
      "tokens_seen": 492240896
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008594182547642929,
      "loss": 1.7382,
      "theoretical_loss": 3.949875258098121,
      "tokens_seen": 492306432
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008593981945837513,
      "loss": 1.6505,
      "theoretical_loss": 3.949818885998111,
      "tokens_seen": 492371968
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008593781344032096,
      "loss": 1.7274,
      "theoretical_loss": 3.9497625235014735,
      "tokens_seen": 492437504
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000859358074222668,
      "loss": 1.6178,
      "theoretical_loss": 3.949706170605296,
      "tokens_seen": 492503040
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008593380140421264,
      "loss": 1.7184,
      "theoretical_loss": 3.949649827306665,
      "tokens_seen": 492568576
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008593179538615847,
      "loss": 1.7109,
      "theoretical_loss": 3.9495934936026695,
      "tokens_seen": 492634112
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008592978936810432,
      "loss": 1.6992,
      "theoretical_loss": 3.9495371694904002,
      "tokens_seen": 492699648
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008592778335005015,
      "loss": 1.5218,
      "theoretical_loss": 3.9494808549669473,
      "tokens_seen": 492765184
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008592577733199599,
      "loss": 1.704,
      "theoretical_loss": 3.9494245500294047,
      "tokens_seen": 492830720
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008592377131394182,
      "loss": 1.7143,
      "theoretical_loss": 3.949368254674866,
      "tokens_seen": 492896256
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008592176529588766,
      "loss": 1.6393,
      "theoretical_loss": 3.9493119689004255,
      "tokens_seen": 492961792
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000859197592778335,
      "loss": 1.665,
      "theoretical_loss": 3.9492556927031806,
      "tokens_seen": 493027328
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008591775325977934,
      "loss": 1.5913,
      "theoretical_loss": 3.949199426080228,
      "tokens_seen": 493092864
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008591574724172517,
      "loss": 1.6989,
      "theoretical_loss": 3.949143169028668,
      "tokens_seen": 493158400
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008591374122367102,
      "loss": 1.6804,
      "theoretical_loss": 3.9490869215455993,
      "tokens_seen": 493223936
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008591173520561685,
      "loss": 1.6273,
      "theoretical_loss": 3.9490306836281244,
      "tokens_seen": 493289472
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008590972918756269,
      "loss": 1.6909,
      "theoretical_loss": 3.9489744552733455,
      "tokens_seen": 493355008
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008590772316950852,
      "loss": 1.677,
      "theoretical_loss": 3.948918236478367,
      "tokens_seen": 493420544
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008590571715145437,
      "loss": 1.8298,
      "theoretical_loss": 3.9488620272402937,
      "tokens_seen": 493486080
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000859037111334002,
      "loss": 1.7091,
      "theoretical_loss": 3.9488058275562326,
      "tokens_seen": 493551616
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008590170511534604,
      "loss": 1.8155,
      "theoretical_loss": 3.9487496374232913,
      "tokens_seen": 493617152
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008589969909729187,
      "loss": 1.6607,
      "theoretical_loss": 3.948693456838579,
      "tokens_seen": 493682688
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008589769307923771,
      "loss": 1.5868,
      "theoretical_loss": 3.9486372857992054,
      "tokens_seen": 493748224
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008589568706118355,
      "loss": 1.6415,
      "theoretical_loss": 3.948581124302283,
      "tokens_seen": 493813760
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008589368104312939,
      "loss": 1.6505,
      "theoretical_loss": 3.9485249723449236,
      "tokens_seen": 493879296
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008589167502507522,
      "loss": 1.5553,
      "theoretical_loss": 3.9484688299242423,
      "tokens_seen": 493944832
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008588966900702107,
      "loss": 1.5332,
      "theoretical_loss": 3.9484126970373534,
      "tokens_seen": 494010368
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000858876629889669,
      "loss": 1.7607,
      "theoretical_loss": 3.948356573681374,
      "tokens_seen": 494075904
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008588565697091274,
      "loss": 1.6188,
      "theoretical_loss": 3.9483004598534217,
      "tokens_seen": 494141440
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008588365095285859,
      "loss": 1.7941,
      "theoretical_loss": 3.9482443555506164,
      "tokens_seen": 494206976
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008588164493480441,
      "loss": 1.6809,
      "theoretical_loss": 3.9481882607700776,
      "tokens_seen": 494272512
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008587963891675026,
      "loss": 1.6599,
      "theoretical_loss": 3.948132175508927,
      "tokens_seen": 494338048
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008587763289869609,
      "loss": 1.6267,
      "theoretical_loss": 3.948076099764288,
      "tokens_seen": 494403584
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008587562688064193,
      "loss": 1.7205,
      "theoretical_loss": 3.948020033533284,
      "tokens_seen": 494469120
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008587362086258777,
      "loss": 1.7124,
      "theoretical_loss": 3.9479639768130403,
      "tokens_seen": 494534656
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000858716148445336,
      "loss": 1.74,
      "theoretical_loss": 3.9479079296006843,
      "tokens_seen": 494600192
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008586960882647944,
      "loss": 1.8365,
      "theoretical_loss": 3.947851891893343,
      "tokens_seen": 494665728
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008586760280842529,
      "loss": 1.6611,
      "theoretical_loss": 3.947795863688146,
      "tokens_seen": 494731264
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.4944104850292206,
      "objective/train/docs_used": 186433,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6233147382736206,
      "objective/train/original_loss": 1.6233150959014893,
      "objective/train/theoretical_loss": 3.947739844982224,
      "objective/train/tokens_used": 515256800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24478408694267273,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506855249404907,
      "objective/train/weighted_lm_loss": 1.7054861783981323,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.947739844982224,
      "tokens_seen": 494796800
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008586559679037111,
      "loss": 1.6021,
      "theoretical_loss": 3.947739844982224,
      "tokens_seen": 494796800
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008586359077231696,
      "loss": 1.5467,
      "theoretical_loss": 3.9476838357727075,
      "tokens_seen": 494862336
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008586158475426278,
      "loss": 1.5192,
      "theoretical_loss": 3.9476278360567303,
      "tokens_seen": 494927872
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008585957873620863,
      "loss": 1.6917,
      "theoretical_loss": 3.9475718458314257,
      "tokens_seen": 494993408
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008585757271815447,
      "loss": 1.6019,
      "theoretical_loss": 3.94751586509393,
      "tokens_seen": 495058944
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000858555667001003,
      "loss": 1.6493,
      "theoretical_loss": 3.9474598938413785,
      "tokens_seen": 495124480
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008585356068204614,
      "loss": 1.6765,
      "theoretical_loss": 3.9474039320709107,
      "tokens_seen": 495190016
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008585155466399199,
      "loss": 1.7863,
      "theoretical_loss": 3.9473479797796633,
      "tokens_seen": 495255552
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008584954864593781,
      "loss": 1.7943,
      "theoretical_loss": 3.947292036964779,
      "tokens_seen": 495321088
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008584754262788366,
      "loss": 1.6114,
      "theoretical_loss": 3.947236103623398,
      "tokens_seen": 495386624
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008584553660982948,
      "loss": 1.6652,
      "theoretical_loss": 3.9471801797526633,
      "tokens_seen": 495452160
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008584353059177533,
      "loss": 1.5683,
      "theoretical_loss": 3.947124265349719,
      "tokens_seen": 495517696
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008584152457372117,
      "loss": 1.5359,
      "theoretical_loss": 3.9470683604117105,
      "tokens_seen": 495583232
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.00085839518555667,
      "loss": 1.7465,
      "theoretical_loss": 3.947012464935784,
      "tokens_seen": 495648768
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008583751253761284,
      "loss": 1.6241,
      "theoretical_loss": 3.946956578919088,
      "tokens_seen": 495714304
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008583550651955868,
      "loss": 1.5361,
      "theoretical_loss": 3.9469007023587697,
      "tokens_seen": 495779840
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008583350050150451,
      "loss": 1.7957,
      "theoretical_loss": 3.946844835251981,
      "tokens_seen": 495845376
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008583149448345036,
      "loss": 1.7185,
      "theoretical_loss": 3.946788977595873,
      "tokens_seen": 495910912
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008582948846539619,
      "loss": 1.6026,
      "theoretical_loss": 3.9467331293875976,
      "tokens_seen": 495976448
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008582748244734203,
      "loss": 1.708,
      "theoretical_loss": 3.9466772906243097,
      "tokens_seen": 496041984
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008582547642928786,
      "loss": 1.6868,
      "theoretical_loss": 3.946621461303164,
      "tokens_seen": 496107520
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000858234704112337,
      "loss": 1.6382,
      "theoretical_loss": 3.9465656414213166,
      "tokens_seen": 496173056
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008582146439317954,
      "loss": 1.6916,
      "theoretical_loss": 3.9465098309759252,
      "tokens_seen": 496238592
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008581945837512538,
      "loss": 1.691,
      "theoretical_loss": 3.9464540299641495,
      "tokens_seen": 496304128
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008581745235707121,
      "loss": 1.5692,
      "theoretical_loss": 3.946398238383148,
      "tokens_seen": 496369664
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008581544633901706,
      "loss": 1.6496,
      "theoretical_loss": 3.9463424562300835,
      "tokens_seen": 496435200
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008581344032096289,
      "loss": 1.7202,
      "theoretical_loss": 3.9462866835021178,
      "tokens_seen": 496500736
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008581143430290873,
      "loss": 1.6107,
      "theoretical_loss": 3.946230920196414,
      "tokens_seen": 496566272
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008580942828485456,
      "loss": 1.6035,
      "theoretical_loss": 3.9461751663101388,
      "tokens_seen": 496631808
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008580742226680041,
      "loss": 1.7093,
      "theoretical_loss": 3.946119421840457,
      "tokens_seen": 496697344
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008580541624874624,
      "loss": 1.7058,
      "theoretical_loss": 3.9460636867845365,
      "tokens_seen": 496762880
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008580341023069208,
      "loss": 1.6845,
      "theoretical_loss": 3.9460079611395456,
      "tokens_seen": 496828416
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008580140421263791,
      "loss": 1.6648,
      "theoretical_loss": 3.9459522449026547,
      "tokens_seen": 496893952
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008579939819458375,
      "loss": 1.6558,
      "theoretical_loss": 3.945896538071035,
      "tokens_seen": 496959488
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008579739217652959,
      "loss": 1.6979,
      "theoretical_loss": 3.9458408406418584,
      "tokens_seen": 497025024
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008579538615847543,
      "loss": 1.6416,
      "theoretical_loss": 3.9457851526122987,
      "tokens_seen": 497090560
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008579338014042126,
      "loss": 1.6168,
      "theoretical_loss": 3.9457294739795303,
      "tokens_seen": 497156096
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008579137412236711,
      "loss": 1.7117,
      "theoretical_loss": 3.9456738047407294,
      "tokens_seen": 497221632
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008578936810431294,
      "loss": 1.5237,
      "theoretical_loss": 3.945618144893073,
      "tokens_seen": 497287168
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008578736208625878,
      "loss": 1.7463,
      "theoretical_loss": 3.945562494433741,
      "tokens_seen": 497352704
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008578535606820461,
      "loss": 1.7703,
      "theoretical_loss": 3.9455068533599107,
      "tokens_seen": 497418240
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008578335005015045,
      "loss": 1.6323,
      "theoretical_loss": 3.9454512216687645,
      "tokens_seen": 497483776
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008578134403209629,
      "loss": 1.6856,
      "theoretical_loss": 3.9453955993574845,
      "tokens_seen": 497549312
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008577933801404213,
      "loss": 1.6327,
      "theoretical_loss": 3.945339986423254,
      "tokens_seen": 497614848
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008577733199598796,
      "loss": 1.6471,
      "theoretical_loss": 3.9452843828632562,
      "tokens_seen": 497680384
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008577532597793381,
      "loss": 1.6732,
      "theoretical_loss": 3.945228788674678,
      "tokens_seen": 497745920
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008577331995987963,
      "loss": 1.6235,
      "theoretical_loss": 3.945173203854707,
      "tokens_seen": 497811456
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008577131394182548,
      "loss": 1.6917,
      "theoretical_loss": 3.9451176284005296,
      "tokens_seen": 497876992
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008576930792377132,
      "loss": 1.6517,
      "theoretical_loss": 3.945062062309337,
      "tokens_seen": 497942528
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008576730190571715,
      "loss": 1.7057,
      "theoretical_loss": 3.9450065055783186,
      "tokens_seen": 498008064
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.4921090602874756,
      "objective/train/docs_used": 187487,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.686283826828003,
      "objective/train/original_loss": 1.686283826828003,
      "objective/train/theoretical_loss": 3.9449509582046662,
      "objective/train/tokens_used": 518533600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2426682561635971,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050444483757019,
      "objective/train/weighted_lm_loss": 1.7715121507644653,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9449509582046662,
      "tokens_seen": 498073600
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008576529588766299,
      "loss": 1.7322,
      "theoretical_loss": 3.9449509582046662,
      "tokens_seen": 498073600
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008576328986960882,
      "loss": 1.622,
      "theoretical_loss": 3.944895420185574,
      "tokens_seen": 498139136
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008576128385155466,
      "loss": 1.7711,
      "theoretical_loss": 3.944839891518235,
      "tokens_seen": 498204672
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008575927783350051,
      "loss": 1.6856,
      "theoretical_loss": 3.9447843721998455,
      "tokens_seen": 498270208
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008575727181544633,
      "loss": 1.5376,
      "theoretical_loss": 3.944728862227601,
      "tokens_seen": 498335744
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008575526579739218,
      "loss": 1.5706,
      "theoretical_loss": 3.9446733615987,
      "tokens_seen": 498401280
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008575325977933802,
      "loss": 1.6647,
      "theoretical_loss": 3.944617870310343,
      "tokens_seen": 498466816
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008575125376128385,
      "loss": 1.6552,
      "theoretical_loss": 3.944562388359728,
      "tokens_seen": 498532352
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008574924774322969,
      "loss": 1.5662,
      "theoretical_loss": 3.9445069157440575,
      "tokens_seen": 498597888
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008574724172517552,
      "loss": 1.7122,
      "theoretical_loss": 3.9444514524605347,
      "tokens_seen": 498663424
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008574523570712136,
      "loss": 1.6933,
      "theoretical_loss": 3.9443959985063626,
      "tokens_seen": 498728960
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008574322968906721,
      "loss": 1.7744,
      "theoretical_loss": 3.9443405538787464,
      "tokens_seen": 498794496
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008574122367101303,
      "loss": 1.6706,
      "theoretical_loss": 3.944285118574893,
      "tokens_seen": 498860032
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008573921765295888,
      "loss": 1.5819,
      "theoretical_loss": 3.9442296925920095,
      "tokens_seen": 498925568
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008573721163490471,
      "loss": 1.8027,
      "theoretical_loss": 3.944174275927305,
      "tokens_seen": 498991104
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008573520561685055,
      "loss": 1.7501,
      "theoretical_loss": 3.9441188685779887,
      "tokens_seen": 499056640
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000857331995987964,
      "loss": 1.6712,
      "theoretical_loss": 3.9440634705412725,
      "tokens_seen": 499122176
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008573119358074223,
      "loss": 1.7242,
      "theoretical_loss": 3.9440080818143683,
      "tokens_seen": 499187712
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008572918756268807,
      "loss": 1.6799,
      "theoretical_loss": 3.9439527023944896,
      "tokens_seen": 499253248
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000857271815446339,
      "loss": 1.7857,
      "theoretical_loss": 3.9438973322788518,
      "tokens_seen": 499318784
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008572517552657974,
      "loss": 1.6998,
      "theoretical_loss": 3.9438419714646695,
      "tokens_seen": 499384320
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008572316950852558,
      "loss": 1.8287,
      "theoretical_loss": 3.943786619949161,
      "tokens_seen": 499449856
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008572116349047142,
      "loss": 1.6951,
      "theoretical_loss": 3.943731277729544,
      "tokens_seen": 499515392
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008571915747241725,
      "loss": 1.7141,
      "theoretical_loss": 3.9436759448030383,
      "tokens_seen": 499580928
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000857171514543631,
      "loss": 1.8381,
      "theoretical_loss": 3.9436206211668647,
      "tokens_seen": 499646464
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008571514543630893,
      "loss": 1.713,
      "theoretical_loss": 3.943565306818245,
      "tokens_seen": 499712000
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008571313941825477,
      "loss": 1.6078,
      "theoretical_loss": 3.943510001754402,
      "tokens_seen": 499777536
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000857111334002006,
      "loss": 1.6472,
      "theoretical_loss": 3.9434547059725604,
      "tokens_seen": 499843072
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008570912738214645,
      "loss": 1.7676,
      "theoretical_loss": 3.9433994194699453,
      "tokens_seen": 499908608
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008570712136409228,
      "loss": 1.844,
      "theoretical_loss": 3.9433441422437836,
      "tokens_seen": 499974144
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008570511534603812,
      "loss": 1.7302,
      "theoretical_loss": 3.9432888742913033,
      "tokens_seen": 500039680
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008570310932798395,
      "loss": 1.8328,
      "theoretical_loss": 3.943233615609733,
      "tokens_seen": 500105216
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008570110330992979,
      "loss": 1.8726,
      "theoretical_loss": 3.943178366196304,
      "tokens_seen": 500170752
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008569909729187563,
      "loss": 1.7599,
      "theoretical_loss": 3.9431231260482464,
      "tokens_seen": 500236288
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008569709127382147,
      "loss": 1.7863,
      "theoretical_loss": 3.9430678951627938,
      "tokens_seen": 500301824
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000856950852557673,
      "loss": 1.7858,
      "theoretical_loss": 3.94301267353718,
      "tokens_seen": 500367360
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008569307923771315,
      "loss": 1.7798,
      "theoretical_loss": 3.942957461168639,
      "tokens_seen": 500432896
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008569107321965898,
      "loss": 1.8921,
      "theoretical_loss": 3.942902258054408,
      "tokens_seen": 500498432
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008568906720160482,
      "loss": 1.7891,
      "theoretical_loss": 3.9428470641917244,
      "tokens_seen": 500563968
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008568706118355065,
      "loss": 1.8246,
      "theoretical_loss": 3.9427918795778267,
      "tokens_seen": 500629504
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008568505516549649,
      "loss": 1.7398,
      "theoretical_loss": 3.9427367042099544,
      "tokens_seen": 500695040
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008568304914744233,
      "loss": 1.6855,
      "theoretical_loss": 3.9426815380853477,
      "tokens_seen": 500760576
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008568104312938817,
      "loss": 1.6292,
      "theoretical_loss": 3.9426263812012503,
      "tokens_seen": 500826112
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.00085679037111334,
      "loss": 1.716,
      "theoretical_loss": 3.9425712335549044,
      "tokens_seen": 500891648
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008567703109327985,
      "loss": 1.7085,
      "theoretical_loss": 3.942516095143555,
      "tokens_seen": 500957184
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008567502507522567,
      "loss": 1.7111,
      "theoretical_loss": 3.9424609659644476,
      "tokens_seen": 501022720
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008567301905717152,
      "loss": 1.8806,
      "theoretical_loss": 3.9424058460148292,
      "tokens_seen": 501088256
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008567101303911736,
      "loss": 1.7525,
      "theoretical_loss": 3.9423507352919476,
      "tokens_seen": 501153792
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008566900702106319,
      "loss": 1.8146,
      "theoretical_loss": 3.9422956337930524,
      "tokens_seen": 501219328
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008566700100300903,
      "loss": 1.7736,
      "theoretical_loss": 3.9422405415153934,
      "tokens_seen": 501284864
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.49449148774147034,
      "objective/train/docs_used": 188727,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5148030519485474,
      "objective/train/original_loss": 1.5148030519485474,
      "objective/train/theoretical_loss": 3.9421854584562226,
      "objective/train/tokens_used": 521810400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24478769302368164,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506936311721802,
      "objective/train/weighted_lm_loss": 1.5914528369903564,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9421854584562226,
      "tokens_seen": 501350400
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008566499498495486,
      "loss": 1.7312,
      "theoretical_loss": 3.9421854584562226,
      "tokens_seen": 501350400
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000856629889669007,
      "loss": 1.6754,
      "theoretical_loss": 3.942130384612793,
      "tokens_seen": 501415936
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008566098294884655,
      "loss": 1.726,
      "theoretical_loss": 3.942075319982358,
      "tokens_seen": 501481472
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008565897693079237,
      "loss": 1.8364,
      "theoretical_loss": 3.9420202645621725,
      "tokens_seen": 501547008
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008565697091273822,
      "loss": 1.7069,
      "theoretical_loss": 3.9419652183494938,
      "tokens_seen": 501612544
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008565496489468406,
      "loss": 1.7698,
      "theoretical_loss": 3.9419101813415787,
      "tokens_seen": 501678080
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008565295887662989,
      "loss": 1.663,
      "theoretical_loss": 3.941855153535686,
      "tokens_seen": 501743616
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008565095285857573,
      "loss": 1.6876,
      "theoretical_loss": 3.941800134929075,
      "tokens_seen": 501809152
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008564894684052156,
      "loss": 1.7476,
      "theoretical_loss": 3.9417451255190077,
      "tokens_seen": 501874688
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000856469408224674,
      "loss": 1.808,
      "theoretical_loss": 3.9416901253027454,
      "tokens_seen": 501940224
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008564493480441325,
      "loss": 1.7675,
      "theoretical_loss": 3.9416351342775524,
      "tokens_seen": 502005760
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008564292878635907,
      "loss": 1.6907,
      "theoretical_loss": 3.9415801524406917,
      "tokens_seen": 502071296
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008564092276830492,
      "loss": 1.7417,
      "theoretical_loss": 3.94152517978943,
      "tokens_seen": 502136832
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008563891675025075,
      "loss": 1.588,
      "theoretical_loss": 3.941470216321034,
      "tokens_seen": 502202368
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008563691073219659,
      "loss": 1.8457,
      "theoretical_loss": 3.9414152620327716,
      "tokens_seen": 502267904
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008563490471414243,
      "loss": 1.6521,
      "theoretical_loss": 3.9413603169219122,
      "tokens_seen": 502333440
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008563289869608827,
      "loss": 1.6019,
      "theoretical_loss": 3.9413053809857264,
      "tokens_seen": 502398976
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000856308926780341,
      "loss": 1.6222,
      "theoretical_loss": 3.941250454221485,
      "tokens_seen": 502464512
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008562888665997994,
      "loss": 1.7308,
      "theoretical_loss": 3.941195536626461,
      "tokens_seen": 502530048
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008562688064192577,
      "loss": 1.7501,
      "theoretical_loss": 3.941140628197928,
      "tokens_seen": 502595584
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008562487462387162,
      "loss": 1.7315,
      "theoretical_loss": 3.941085728933162,
      "tokens_seen": 502661120
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008562286860581745,
      "loss": 1.7432,
      "theoretical_loss": 3.9410308388294384,
      "tokens_seen": 502726656
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008562086258776329,
      "loss": 1.6503,
      "theoretical_loss": 3.940975957884034,
      "tokens_seen": 502792192
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008561885656970913,
      "loss": 1.7376,
      "theoretical_loss": 3.9409210860942285,
      "tokens_seen": 502857728
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008561685055165497,
      "loss": 1.7768,
      "theoretical_loss": 3.9408662234573013,
      "tokens_seen": 502923264
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000856148445336008,
      "loss": 1.6886,
      "theoretical_loss": 3.9408113699705325,
      "tokens_seen": 502988800
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008561283851554664,
      "loss": 1.6498,
      "theoretical_loss": 3.9407565256312047,
      "tokens_seen": 503054336
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008561083249749247,
      "loss": 1.7782,
      "theoretical_loss": 3.940701690436601,
      "tokens_seen": 503119872
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008560882647943832,
      "loss": 1.7973,
      "theoretical_loss": 3.9406468643840054,
      "tokens_seen": 503185408
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008560682046138415,
      "loss": 1.7796,
      "theoretical_loss": 3.940592047470704,
      "tokens_seen": 503250944
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008560481444332999,
      "loss": 1.6904,
      "theoretical_loss": 3.940537239693983,
      "tokens_seen": 503316480
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008560280842527582,
      "loss": 1.7036,
      "theoretical_loss": 3.9404824410511305,
      "tokens_seen": 503382016
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008560080240722167,
      "loss": 1.7104,
      "theoretical_loss": 3.940427651539435,
      "tokens_seen": 503447552
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000855987963891675,
      "loss": 1.8244,
      "theoretical_loss": 3.940372871156187,
      "tokens_seen": 503513088
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008559679037111334,
      "loss": 1.7562,
      "theoretical_loss": 3.9403180998986778,
      "tokens_seen": 503578624
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008559478435305918,
      "loss": 1.6141,
      "theoretical_loss": 3.9402633377641996,
      "tokens_seen": 503644160
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008559277833500502,
      "loss": 1.7405,
      "theoretical_loss": 3.940208584750046,
      "tokens_seen": 503709696
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008559077231695085,
      "loss": 1.6818,
      "theoretical_loss": 3.940153840853512,
      "tokens_seen": 503775232
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000855887662988967,
      "loss": 1.6255,
      "theoretical_loss": 3.9400991060718935,
      "tokens_seen": 503840768
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008558676028084252,
      "loss": 1.7052,
      "theoretical_loss": 3.940044380402487,
      "tokens_seen": 503906304
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008558475426278837,
      "loss": 1.8162,
      "theoretical_loss": 3.9399896638425913,
      "tokens_seen": 503971840
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008558274824473421,
      "loss": 1.8732,
      "theoretical_loss": 3.9399349563895054,
      "tokens_seen": 504037376
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008558074222668004,
      "loss": 1.7249,
      "theoretical_loss": 3.93988025804053,
      "tokens_seen": 504102912
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008557873620862589,
      "loss": 1.7259,
      "theoretical_loss": 3.939825568792967,
      "tokens_seen": 504168448
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008557673019057171,
      "loss": 1.7787,
      "theoretical_loss": 3.9397708886441185,
      "tokens_seen": 504233984
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008557472417251756,
      "loss": 1.6362,
      "theoretical_loss": 3.939716217591289,
      "tokens_seen": 504299520
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000855727181544634,
      "loss": 1.7864,
      "theoretical_loss": 3.9396615556317838,
      "tokens_seen": 504365056
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008557071213640923,
      "loss": 1.664,
      "theoretical_loss": 3.9396069027629084,
      "tokens_seen": 504430592
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008556870611835507,
      "loss": 1.688,
      "theoretical_loss": 3.939552258981971,
      "tokens_seen": 504496128
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000855667001003009,
      "loss": 1.6894,
      "theoretical_loss": 3.93949762428628,
      "tokens_seen": 504561664
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.4919562339782715,
      "objective/train/docs_used": 189911,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.488186001777649,
      "objective/train/original_loss": 1.4881861209869385,
      "objective/train/theoretical_loss": 3.939442998673144,
      "objective/train/tokens_used": 525087200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24271796643733978,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504294633865356,
      "objective/train/weighted_lm_loss": 1.563111424446106,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.939442998673144,
      "tokens_seen": 504627200
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008556469408224674,
      "loss": 1.6751,
      "theoretical_loss": 3.939442998673144,
      "tokens_seen": 504627200
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008556268806419259,
      "loss": 1.6742,
      "theoretical_loss": 3.9393883821398754,
      "tokens_seen": 504692736
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008556068204613841,
      "loss": 1.6823,
      "theoretical_loss": 3.939333774683785,
      "tokens_seen": 504758272
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008555867602808426,
      "loss": 1.6976,
      "theoretical_loss": 3.939279176302187,
      "tokens_seen": 504823808
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000855566700100301,
      "loss": 1.6921,
      "theoretical_loss": 3.9392245869923954,
      "tokens_seen": 504889344
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008555466399197593,
      "loss": 1.7448,
      "theoretical_loss": 3.939170006751725,
      "tokens_seen": 504954880
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008555265797392177,
      "loss": 1.6712,
      "theoretical_loss": 3.9391154355774924,
      "tokens_seen": 505020416
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000855506519558676,
      "loss": 1.752,
      "theoretical_loss": 3.9390608734670156,
      "tokens_seen": 505085952
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008554864593781344,
      "loss": 1.87,
      "theoretical_loss": 3.939006320417614,
      "tokens_seen": 505151488
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008554663991975929,
      "loss": 1.7513,
      "theoretical_loss": 3.938951776426607,
      "tokens_seen": 505217024
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008554463390170511,
      "loss": 1.6259,
      "theoretical_loss": 3.9388972414913157,
      "tokens_seen": 505282560
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008554262788365096,
      "loss": 1.7146,
      "theoretical_loss": 3.938842715609062,
      "tokens_seen": 505348096
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008554062186559679,
      "loss": 1.6968,
      "theoretical_loss": 3.9387881987771705,
      "tokens_seen": 505413632
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008553861584754263,
      "loss": 1.6711,
      "theoretical_loss": 3.938733690992964,
      "tokens_seen": 505479168
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008553660982948847,
      "loss": 1.7096,
      "theoretical_loss": 3.93867919225377,
      "tokens_seen": 505544704
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008553460381143431,
      "loss": 1.7244,
      "theoretical_loss": 3.938624702556914,
      "tokens_seen": 505610240
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008553259779338014,
      "loss": 1.7492,
      "theoretical_loss": 3.9385702218997247,
      "tokens_seen": 505675776
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008553059177532598,
      "loss": 1.803,
      "theoretical_loss": 3.9385157502795316,
      "tokens_seen": 505741312
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008552858575727181,
      "loss": 1.6991,
      "theoretical_loss": 3.9384612876936633,
      "tokens_seen": 505806848
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008552657973921766,
      "loss": 1.7034,
      "theoretical_loss": 3.9384068341394523,
      "tokens_seen": 505872384
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008552457372116349,
      "loss": 1.6652,
      "theoretical_loss": 3.9383523896142316,
      "tokens_seen": 505937920
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008552256770310933,
      "loss": 1.6977,
      "theoretical_loss": 3.9382979541153333,
      "tokens_seen": 506003456
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008552056168505517,
      "loss": 1.7646,
      "theoretical_loss": 3.9382435276400938,
      "tokens_seen": 506068992
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008551855566700101,
      "loss": 1.613,
      "theoretical_loss": 3.9381891101858475,
      "tokens_seen": 506134528
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008551654964894684,
      "loss": 1.7528,
      "theoretical_loss": 3.9381347017499326,
      "tokens_seen": 506200064
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008551454363089268,
      "loss": 1.8456,
      "theoretical_loss": 3.938080302329687,
      "tokens_seen": 506265600
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008551253761283851,
      "loss": 1.7717,
      "theoretical_loss": 3.938025911922449,
      "tokens_seen": 506331136
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008551053159478436,
      "loss": 1.5916,
      "theoretical_loss": 3.9379715305255605,
      "tokens_seen": 506396672
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008550852557673019,
      "loss": 1.6501,
      "theoretical_loss": 3.9379171581363623,
      "tokens_seen": 506462208
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008550651955867603,
      "loss": 1.7902,
      "theoretical_loss": 3.9378627947521965,
      "tokens_seen": 506527744
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008550451354062186,
      "loss": 1.595,
      "theoretical_loss": 3.937808440370408,
      "tokens_seen": 506593280
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008550250752256771,
      "loss": 1.672,
      "theoretical_loss": 3.9377540949883416,
      "tokens_seen": 506658816
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008550050150451354,
      "loss": 1.7568,
      "theoretical_loss": 3.937699758603342,
      "tokens_seen": 506724352
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008549849548645938,
      "loss": 1.6198,
      "theoretical_loss": 3.9376454312127587,
      "tokens_seen": 506789888
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008549648946840522,
      "loss": 1.797,
      "theoretical_loss": 3.9375911128139376,
      "tokens_seen": 506855424
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008549448345035106,
      "loss": 1.7229,
      "theoretical_loss": 3.9375368034042295,
      "tokens_seen": 506920960
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008549247743229689,
      "loss": 1.6817,
      "theoretical_loss": 3.937482502980985,
      "tokens_seen": 506986496
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008549047141424273,
      "loss": 1.7819,
      "theoretical_loss": 3.9374282115415546,
      "tokens_seen": 507052032
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008548846539618856,
      "loss": 1.6991,
      "theoretical_loss": 3.9373739290832925,
      "tokens_seen": 507117568
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008548645937813441,
      "loss": 1.7333,
      "theoretical_loss": 3.937319655603552,
      "tokens_seen": 507183104
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008548445336008024,
      "loss": 1.8329,
      "theoretical_loss": 3.937265391099688,
      "tokens_seen": 507248640
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008548244734202608,
      "loss": 1.5592,
      "theoretical_loss": 3.937211135569057,
      "tokens_seen": 507314176
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008548044132397192,
      "loss": 1.6507,
      "theoretical_loss": 3.9371568890090165,
      "tokens_seen": 507379712
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008547843530591775,
      "loss": 1.6943,
      "theoretical_loss": 3.937102651416924,
      "tokens_seen": 507445248
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008547642928786359,
      "loss": 1.6051,
      "theoretical_loss": 3.9370484227901397,
      "tokens_seen": 507510784
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008547442326980944,
      "loss": 1.6238,
      "theoretical_loss": 3.936994203126024,
      "tokens_seen": 507576320
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008547241725175526,
      "loss": 1.6631,
      "theoretical_loss": 3.936939992421939,
      "tokens_seen": 507641856
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008547041123370111,
      "loss": 1.6778,
      "theoretical_loss": 3.936885790675247,
      "tokens_seen": 507707392
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008546840521564693,
      "loss": 1.5877,
      "theoretical_loss": 3.9368315978833124,
      "tokens_seen": 507772928
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008546639919759278,
      "loss": 1.7614,
      "theoretical_loss": 3.9367774140435,
      "tokens_seen": 507838464
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.49557170271873474,
      "objective/train/docs_used": 191028,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5263073444366455,
      "objective/train/original_loss": 1.5263073444366455,
      "objective/train/theoretical_loss": 3.936723239153177,
      "objective/train/tokens_used": 528364000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24599750339984894,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0508077144622803,
      "objective/train/weighted_lm_loss": 1.6038414239883423,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.936723239153177,
      "tokens_seen": 507904000
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008546439317953862,
      "loss": 1.5987,
      "theoretical_loss": 3.936723239153177,
      "tokens_seen": 507904000
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008546238716148445,
      "loss": 1.6881,
      "theoretical_loss": 3.9366690732097096,
      "tokens_seen": 507969536
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008546038114343029,
      "loss": 1.5569,
      "theoretical_loss": 3.936614916210466,
      "tokens_seen": 508035072
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008545837512537614,
      "loss": 1.627,
      "theoretical_loss": 3.936560768152817,
      "tokens_seen": 508100608
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008545636910732196,
      "loss": 1.5718,
      "theoretical_loss": 3.936506629034133,
      "tokens_seen": 508166144
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008545436308926781,
      "loss": 1.6788,
      "theoretical_loss": 3.9364524988517844,
      "tokens_seen": 508231680
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008545235707121363,
      "loss": 1.6036,
      "theoretical_loss": 3.9363983776031457,
      "tokens_seen": 508297216
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008545035105315948,
      "loss": 1.6642,
      "theoretical_loss": 3.936344265285591,
      "tokens_seen": 508362752
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008544834503510532,
      "loss": 1.6405,
      "theoretical_loss": 3.936290161896494,
      "tokens_seen": 508428288
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008544633901705115,
      "loss": 1.6672,
      "theoretical_loss": 3.9362360674332315,
      "tokens_seen": 508493824
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008544433299899699,
      "loss": 1.7934,
      "theoretical_loss": 3.936181981893182,
      "tokens_seen": 508559360
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008544232698094283,
      "loss": 1.7661,
      "theoretical_loss": 3.9361279052737217,
      "tokens_seen": 508624896
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008544032096288866,
      "loss": 1.6901,
      "theoretical_loss": 3.9360738375722324,
      "tokens_seen": 508690432
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008543831494483451,
      "loss": 1.8077,
      "theoretical_loss": 3.9360197787860938,
      "tokens_seen": 508755968
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008543630892678033,
      "loss": 1.6619,
      "theoretical_loss": 3.9359657289126875,
      "tokens_seen": 508821504
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008543430290872618,
      "loss": 1.6664,
      "theoretical_loss": 3.9359116879493965,
      "tokens_seen": 508887040
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008543229689067202,
      "loss": 1.6839,
      "theoretical_loss": 3.9358576558936047,
      "tokens_seen": 508952576
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008543029087261785,
      "loss": 1.71,
      "theoretical_loss": 3.9358036327426977,
      "tokens_seen": 509018112
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000854282848545637,
      "loss": 1.6584,
      "theoretical_loss": 3.935749618494061,
      "tokens_seen": 509083648
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008542627883650953,
      "loss": 1.5792,
      "theoretical_loss": 3.935695613145082,
      "tokens_seen": 509149184
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008542427281845537,
      "loss": 1.6885,
      "theoretical_loss": 3.9356416166931503,
      "tokens_seen": 509214720
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008542226680040121,
      "loss": 1.7117,
      "theoretical_loss": 3.9355876291356533,
      "tokens_seen": 509280256
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008542026078234705,
      "loss": 1.6857,
      "theoretical_loss": 3.935533650469983,
      "tokens_seen": 509345792
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008541825476429288,
      "loss": 1.595,
      "theoretical_loss": 3.9354796806935313,
      "tokens_seen": 509411328
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008541624874623872,
      "loss": 1.6484,
      "theoretical_loss": 3.93542571980369,
      "tokens_seen": 509476864
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008541424272818455,
      "loss": 1.6633,
      "theoretical_loss": 3.9353717677978537,
      "tokens_seen": 509542400
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000854122367101304,
      "loss": 1.6041,
      "theoretical_loss": 3.935317824673417,
      "tokens_seen": 509607936
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008541023069207623,
      "loss": 1.715,
      "theoretical_loss": 3.935263890427777,
      "tokens_seen": 509673472
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008540822467402207,
      "loss": 1.5171,
      "theoretical_loss": 3.935209965058329,
      "tokens_seen": 509739008
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000854062186559679,
      "loss": 1.6945,
      "theoretical_loss": 3.9351560485624733,
      "tokens_seen": 509804544
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008540421263791375,
      "loss": 1.7677,
      "theoretical_loss": 3.935102140937608,
      "tokens_seen": 509870080
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008540220661985958,
      "loss": 1.7678,
      "theoretical_loss": 3.935048242181134,
      "tokens_seen": 509935616
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008540020060180542,
      "loss": 1.5264,
      "theoretical_loss": 3.9349943522904525,
      "tokens_seen": 510001152
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008539819458375126,
      "loss": 1.7062,
      "theoretical_loss": 3.934940471262967,
      "tokens_seen": 510066688
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000853961885656971,
      "loss": 1.5977,
      "theoretical_loss": 3.934886599096081,
      "tokens_seen": 510132224
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008539418254764293,
      "loss": 1.6127,
      "theoretical_loss": 3.9348327357871984,
      "tokens_seen": 510197760
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008539217652958877,
      "loss": 1.5628,
      "theoretical_loss": 3.9347788813337266,
      "tokens_seen": 510263296
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000853901705115346,
      "loss": 1.7656,
      "theoretical_loss": 3.934725035733072,
      "tokens_seen": 510328832
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008538816449348045,
      "loss": 1.6011,
      "theoretical_loss": 3.9346711989826426,
      "tokens_seen": 510394368
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008538615847542628,
      "loss": 1.6009,
      "theoretical_loss": 3.934617371079848,
      "tokens_seen": 510459904
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008538415245737212,
      "loss": 1.6237,
      "theoretical_loss": 3.934563552022098,
      "tokens_seen": 510525440
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008538214643931796,
      "loss": 1.736,
      "theoretical_loss": 3.9345097418068047,
      "tokens_seen": 510590976
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008538014042126379,
      "loss": 1.6643,
      "theoretical_loss": 3.93445594043138,
      "tokens_seen": 510656512
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008537813440320963,
      "loss": 1.6711,
      "theoretical_loss": 3.9344021478932376,
      "tokens_seen": 510722048
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008537612838515548,
      "loss": 1.6318,
      "theoretical_loss": 3.934348364189793,
      "tokens_seen": 510787584
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.000853741223671013,
      "loss": 1.5485,
      "theoretical_loss": 3.93429458931846,
      "tokens_seen": 510853120
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008537211634904715,
      "loss": 1.6557,
      "theoretical_loss": 3.9342408232766584,
      "tokens_seen": 510918656
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008537011033099297,
      "loss": 1.4878,
      "theoretical_loss": 3.9341870660618037,
      "tokens_seen": 510984192
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008536810431293882,
      "loss": 1.7565,
      "theoretical_loss": 3.9341333176713156,
      "tokens_seen": 511049728
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008536609829488466,
      "loss": 1.6649,
      "theoretical_loss": 3.9340795781026148,
      "tokens_seen": 511115264
    },
    {
      "epoch": 0.15,
      "objective/train/advantage_avg": 0.4954502582550049,
      "objective/train/docs_used": 192198,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6345678567886353,
      "objective/train/original_loss": 1.6345679759979248,
      "objective/train/theoretical_loss": 3.934025847353122,
      "objective/train/tokens_used": 531640800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24563312530517578,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507938861846924,
      "objective/train/weighted_lm_loss": 1.7175711393356323,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0168063640594482,
      "theoretical_loss": 3.934025847353122,
      "tokens_seen": 511180800
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008536409227683049,
      "loss": 1.755,
      "theoretical_loss": 3.934025847353122,
      "tokens_seen": 511180800
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008536208625877633,
      "loss": 1.5519,
      "theoretical_loss": 3.933972125420259,
      "tokens_seen": 511246336
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008536008024072218,
      "loss": 1.6695,
      "theoretical_loss": 3.93391841230145,
      "tokens_seen": 511311872
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.00085358074222668,
      "loss": 1.6085,
      "theoretical_loss": 3.933864707994119,
      "tokens_seen": 511377408
    },
    {
      "epoch": 0.15,
      "learning_rate": 0.0008535606820461385,
      "loss": 1.6707,
      "theoretical_loss": 3.9338110124956924,
      "tokens_seen": 511442944
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008535406218655967,
      "loss": 1.65,
      "theoretical_loss": 3.933757325803596,
      "tokens_seen": 511508480
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008535205616850552,
      "loss": 1.5947,
      "theoretical_loss": 3.933703647915258,
      "tokens_seen": 511574016
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008535005015045136,
      "loss": 1.6617,
      "theoretical_loss": 3.9336499788281056,
      "tokens_seen": 511639552
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008534804413239719,
      "loss": 1.7443,
      "theoretical_loss": 3.9335963185395713,
      "tokens_seen": 511705088
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008534603811434303,
      "loss": 1.4573,
      "theoretical_loss": 3.9335426670470834,
      "tokens_seen": 511770624
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008534403209628887,
      "loss": 1.6172,
      "theoretical_loss": 3.933489024348076,
      "tokens_seen": 511836160
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000853420260782347,
      "loss": 1.7263,
      "theoretical_loss": 3.933435390439981,
      "tokens_seen": 511901696
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008534002006018055,
      "loss": 1.6474,
      "theoretical_loss": 3.933381765320233,
      "tokens_seen": 511967232
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008533801404212637,
      "loss": 1.7552,
      "theoretical_loss": 3.9333281489862673,
      "tokens_seen": 512032768
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008533600802407222,
      "loss": 1.5181,
      "theoretical_loss": 3.9332745414355204,
      "tokens_seen": 512098304
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008533400200601805,
      "loss": 1.6143,
      "theoretical_loss": 3.9332209426654288,
      "tokens_seen": 512163840
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008533199598796389,
      "loss": 1.5941,
      "theoretical_loss": 3.933167352673432,
      "tokens_seen": 512229376
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008532998996990973,
      "loss": 1.6446,
      "theoretical_loss": 3.933113771456969,
      "tokens_seen": 512294912
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008532798395185557,
      "loss": 1.8387,
      "theoretical_loss": 3.9330601990134806,
      "tokens_seen": 512360448
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000853259779338014,
      "loss": 1.5672,
      "theoretical_loss": 3.9330066353404085,
      "tokens_seen": 512425984
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008532397191574725,
      "loss": 1.602,
      "theoretical_loss": 3.9329530804351958,
      "tokens_seen": 512491520
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008532196589769308,
      "loss": 1.6394,
      "theoretical_loss": 3.932899534295286,
      "tokens_seen": 512557056
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008531995987963892,
      "loss": 1.7659,
      "theoretical_loss": 3.932845996918124,
      "tokens_seen": 512622592
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008531795386158475,
      "loss": 1.7233,
      "theoretical_loss": 3.9327924683011557,
      "tokens_seen": 512688128
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000853159478435306,
      "loss": 1.6986,
      "theoretical_loss": 3.9327389484418287,
      "tokens_seen": 512753664
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008531394182547643,
      "loss": 1.5626,
      "theoretical_loss": 3.9326854373375904,
      "tokens_seen": 512819200
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008531193580742227,
      "loss": 1.5647,
      "theoretical_loss": 3.9326319349858903,
      "tokens_seen": 512884736
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000853099297893681,
      "loss": 1.6313,
      "theoretical_loss": 3.9325784413841793,
      "tokens_seen": 512950272
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008530792377131394,
      "loss": 1.6366,
      "theoretical_loss": 3.9325249565299076,
      "tokens_seen": 513015808
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008530591775325978,
      "loss": 1.7395,
      "theoretical_loss": 3.932471480420528,
      "tokens_seen": 513081344
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008530391173520562,
      "loss": 1.7295,
      "theoretical_loss": 3.932418013053495,
      "tokens_seen": 513146880
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008530190571715145,
      "loss": 1.5878,
      "theoretical_loss": 3.932364554426262,
      "tokens_seen": 513212416
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000852998996990973,
      "loss": 1.6803,
      "theoretical_loss": 3.932311104536285,
      "tokens_seen": 513277952
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008529789368104312,
      "loss": 1.6529,
      "theoretical_loss": 3.9322576633810202,
      "tokens_seen": 513343488
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008529588766298897,
      "loss": 1.5544,
      "theoretical_loss": 3.9322042309579257,
      "tokens_seen": 513409024
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000852938816449348,
      "loss": 1.6762,
      "theoretical_loss": 3.9321508072644606,
      "tokens_seen": 513474560
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008529187562688064,
      "loss": 1.6738,
      "theoretical_loss": 3.9320973922980844,
      "tokens_seen": 513540096
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008528986960882648,
      "loss": 1.7436,
      "theoretical_loss": 3.932043986056258,
      "tokens_seen": 513605632
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008528786359077232,
      "loss": 1.7212,
      "theoretical_loss": 3.9319905885364443,
      "tokens_seen": 513671168
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008528585757271815,
      "loss": 1.5518,
      "theoretical_loss": 3.9319371997361054,
      "tokens_seen": 513736704
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.00085283851554664,
      "loss": 1.7026,
      "theoretical_loss": 3.931883819652705,
      "tokens_seen": 513802240
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008528184553660983,
      "loss": 1.6241,
      "theoretical_loss": 3.9318304482837094,
      "tokens_seen": 513867776
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008527983951855567,
      "loss": 1.5239,
      "theoretical_loss": 3.931777085626584,
      "tokens_seen": 513933312
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008527783350050152,
      "loss": 1.6441,
      "theoretical_loss": 3.9317237316787965,
      "tokens_seen": 513998848
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008527582748244734,
      "loss": 1.7269,
      "theoretical_loss": 3.9316703864378155,
      "tokens_seen": 514064384
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008527382146439319,
      "loss": 1.636,
      "theoretical_loss": 3.9316170499011096,
      "tokens_seen": 514129920
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008527181544633901,
      "loss": 1.5734,
      "theoretical_loss": 3.93156372206615,
      "tokens_seen": 514195456
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008526980942828486,
      "loss": 1.6645,
      "theoretical_loss": 3.9315104029304084,
      "tokens_seen": 514260992
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000852678034102307,
      "loss": 1.641,
      "theoretical_loss": 3.9314570924913568,
      "tokens_seen": 514326528
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008526579739217653,
      "loss": 1.6756,
      "theoretical_loss": 3.9314037907464687,
      "tokens_seen": 514392064
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.49453940987586975,
      "objective/train/docs_used": 193343,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4929040670394897,
      "objective/train/original_loss": 1.4929039478302002,
      "objective/train/theoretical_loss": 3.931350497693219,
      "objective/train/tokens_used": 534917600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24488162994384766,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506988763809204,
      "objective/train/weighted_lm_loss": 1.568619966506958,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.931350497693219,
      "tokens_seen": 514457600
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008526379137412237,
      "loss": 1.6691,
      "theoretical_loss": 3.931350497693219,
      "tokens_seen": 514457600
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008526178535606822,
      "loss": 1.681,
      "theoretical_loss": 3.9312972133290844,
      "tokens_seen": 514523136
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008525977933801404,
      "loss": 1.6421,
      "theoretical_loss": 3.9312439376515407,
      "tokens_seen": 514588672
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008525777331995989,
      "loss": 1.7608,
      "theoretical_loss": 3.931190670658066,
      "tokens_seen": 514654208
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008525576730190571,
      "loss": 1.584,
      "theoretical_loss": 3.9311374123461387,
      "tokens_seen": 514719744
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008525376128385156,
      "loss": 1.5738,
      "theoretical_loss": 3.9310841627132396,
      "tokens_seen": 514785280
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000852517552657974,
      "loss": 1.7031,
      "theoretical_loss": 3.9310309217568493,
      "tokens_seen": 514850816
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008524974924774323,
      "loss": 1.641,
      "theoretical_loss": 3.93097768947445,
      "tokens_seen": 514916352
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008524774322968907,
      "loss": 1.6874,
      "theoretical_loss": 3.9309244658635247,
      "tokens_seen": 514981888
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008524573721163491,
      "loss": 1.6679,
      "theoretical_loss": 3.930871250921558,
      "tokens_seen": 515047424
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008524373119358074,
      "loss": 1.7494,
      "theoretical_loss": 3.9308180446460343,
      "tokens_seen": 515112960
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008524172517552659,
      "loss": 1.6921,
      "theoretical_loss": 3.9307648470344407,
      "tokens_seen": 515178496
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008523971915747241,
      "loss": 1.5922,
      "theoretical_loss": 3.930711658084264,
      "tokens_seen": 515244032
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008523771313941826,
      "loss": 1.6567,
      "theoretical_loss": 3.930658477792993,
      "tokens_seen": 515309568
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008523570712136409,
      "loss": 1.7634,
      "theoretical_loss": 3.9306053061581165,
      "tokens_seen": 515375104
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008523370110330993,
      "loss": 1.6344,
      "theoretical_loss": 3.9305521431771253,
      "tokens_seen": 515440640
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008523169508525577,
      "loss": 1.6002,
      "theoretical_loss": 3.9304989888475115,
      "tokens_seen": 515506176
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008522968906720161,
      "loss": 1.7391,
      "theoretical_loss": 3.9304458431667664,
      "tokens_seen": 515571712
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008522768304914744,
      "loss": 1.6446,
      "theoretical_loss": 3.930392706132385,
      "tokens_seen": 515637248
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008522567703109329,
      "loss": 1.7342,
      "theoretical_loss": 3.9303395777418606,
      "tokens_seen": 515702784
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008522367101303912,
      "loss": 1.6347,
      "theoretical_loss": 3.93028645799269,
      "tokens_seen": 515768320
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008522166499498496,
      "loss": 1.6956,
      "theoretical_loss": 3.9302333468823694,
      "tokens_seen": 515833856
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008521965897693079,
      "loss": 1.654,
      "theoretical_loss": 3.9301802444083966,
      "tokens_seen": 515899392
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008521765295887663,
      "loss": 1.6685,
      "theoretical_loss": 3.930127150568271,
      "tokens_seen": 515964928
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008521564694082247,
      "loss": 1.7572,
      "theoretical_loss": 3.9300740653594914,
      "tokens_seen": 516030464
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008521364092276831,
      "loss": 1.6354,
      "theoretical_loss": 3.9300209887795594,
      "tokens_seen": 516096000
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008521163490471414,
      "loss": 1.6736,
      "theoretical_loss": 3.929967920825977,
      "tokens_seen": 516161536
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008520962888665998,
      "loss": 1.6179,
      "theoretical_loss": 3.929914861496247,
      "tokens_seen": 516227072
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008520762286860582,
      "loss": 1.57,
      "theoretical_loss": 3.9298618107878727,
      "tokens_seen": 516292608
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008520561685055166,
      "loss": 1.5583,
      "theoretical_loss": 3.9298087686983605,
      "tokens_seen": 516358144
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008520361083249749,
      "loss": 1.6439,
      "theoretical_loss": 3.929755735225216,
      "tokens_seen": 516423680
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008520160481444334,
      "loss": 1.6558,
      "theoretical_loss": 3.929702710365946,
      "tokens_seen": 516489216
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008519959879638916,
      "loss": 1.6313,
      "theoretical_loss": 3.9296496941180585,
      "tokens_seen": 516554752
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008519759277833501,
      "loss": 1.6295,
      "theoretical_loss": 3.929596686479064,
      "tokens_seen": 516620288
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008519558676028084,
      "loss": 1.683,
      "theoretical_loss": 3.9295436874464715,
      "tokens_seen": 516685824
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008519358074222668,
      "loss": 1.6023,
      "theoretical_loss": 3.9294906970177923,
      "tokens_seen": 516751360
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008519157472417252,
      "loss": 1.5942,
      "theoretical_loss": 3.9294377151905397,
      "tokens_seen": 516816896
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008518956870611836,
      "loss": 1.6842,
      "theoretical_loss": 3.929384741962226,
      "tokens_seen": 516882432
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008518756268806419,
      "loss": 1.7586,
      "theoretical_loss": 3.929331777330366,
      "tokens_seen": 516947968
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008518555667001004,
      "loss": 1.5151,
      "theoretical_loss": 3.929278821292476,
      "tokens_seen": 517013504
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008518355065195586,
      "loss": 1.5127,
      "theoretical_loss": 3.929225873846071,
      "tokens_seen": 517079040
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008518154463390171,
      "loss": 1.6306,
      "theoretical_loss": 3.929172934988669,
      "tokens_seen": 517144576
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008517953861584754,
      "loss": 1.6748,
      "theoretical_loss": 3.9291200047177886,
      "tokens_seen": 517210112
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008517753259779338,
      "loss": 1.574,
      "theoretical_loss": 3.9290670830309495,
      "tokens_seen": 517275648
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008517552657973922,
      "loss": 1.6791,
      "theoretical_loss": 3.929014169925672,
      "tokens_seen": 517341184
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008517352056168505,
      "loss": 1.6078,
      "theoretical_loss": 3.928961265399478,
      "tokens_seen": 517406720
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008517151454363089,
      "loss": 1.6236,
      "theoretical_loss": 3.9289083694498905,
      "tokens_seen": 517472256
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008516950852557674,
      "loss": 1.4969,
      "theoretical_loss": 3.9288554820744324,
      "tokens_seen": 517537792
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008516750250752256,
      "loss": 1.6019,
      "theoretical_loss": 3.9288026032706282,
      "tokens_seen": 517603328
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008516549648946841,
      "loss": 1.6628,
      "theoretical_loss": 3.928749733036005,
      "tokens_seen": 517668864
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.49043938517570496,
      "objective/train/docs_used": 194581,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5922307968139648,
      "objective/train/original_loss": 1.5922307968139648,
      "objective/train/theoretical_loss": 3.9286968713680883,
      "objective/train/tokens_used": 538194400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24175012111663818,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502729415893555,
      "objective/train/weighted_lm_loss": 1.672400951385498,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9286968713680883,
      "tokens_seen": 517734400
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008516349047141423,
      "loss": 1.6263,
      "theoretical_loss": 3.9286968713680883,
      "tokens_seen": 517734400
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008516148445336008,
      "loss": 1.64,
      "theoretical_loss": 3.9286440182644062,
      "tokens_seen": 517799936
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008515947843530592,
      "loss": 1.6413,
      "theoretical_loss": 3.928591173722488,
      "tokens_seen": 517865472
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008515747241725175,
      "loss": 1.6881,
      "theoretical_loss": 3.9285383377398624,
      "tokens_seen": 517931008
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008515546639919759,
      "loss": 1.5773,
      "theoretical_loss": 3.9284855103140615,
      "tokens_seen": 517996544
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008515346038114344,
      "loss": 1.5778,
      "theoretical_loss": 3.9284326914426164,
      "tokens_seen": 518062080
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008515145436308926,
      "loss": 1.6953,
      "theoretical_loss": 3.9283798811230604,
      "tokens_seen": 518127616
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008514944834503511,
      "loss": 1.6031,
      "theoretical_loss": 3.9283270793529272,
      "tokens_seen": 518193152
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008514744232698094,
      "loss": 1.6565,
      "theoretical_loss": 3.9282742861297524,
      "tokens_seen": 518258688
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008514543630892678,
      "loss": 1.5558,
      "theoretical_loss": 3.9282215014510706,
      "tokens_seen": 518324224
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008514343029087262,
      "loss": 1.5033,
      "theoretical_loss": 3.9281687253144204,
      "tokens_seen": 518389760
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008514142427281845,
      "loss": 1.6238,
      "theoretical_loss": 3.9281159577173392,
      "tokens_seen": 518455296
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008513941825476429,
      "loss": 1.579,
      "theoretical_loss": 3.928063198657365,
      "tokens_seen": 518520832
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008513741223671013,
      "loss": 1.6135,
      "theoretical_loss": 3.9280104481320395,
      "tokens_seen": 518586368
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008513540621865596,
      "loss": 1.6128,
      "theoretical_loss": 3.927957706138903,
      "tokens_seen": 518651904
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008513340020060181,
      "loss": 1.6363,
      "theoretical_loss": 3.9279049726754973,
      "tokens_seen": 518717440
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008513139418254765,
      "loss": 1.573,
      "theoretical_loss": 3.9278522477393656,
      "tokens_seen": 518782976
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008512938816449348,
      "loss": 1.7497,
      "theoretical_loss": 3.927799531328053,
      "tokens_seen": 518848512
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008512738214643933,
      "loss": 1.5822,
      "theoretical_loss": 3.9277468234391035,
      "tokens_seen": 518914048
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008512537612838516,
      "loss": 1.565,
      "theoretical_loss": 3.9276941240700642,
      "tokens_seen": 518979584
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.00085123370110331,
      "loss": 1.6436,
      "theoretical_loss": 3.9276414332184815,
      "tokens_seen": 519045120
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008512136409227683,
      "loss": 1.6952,
      "theoretical_loss": 3.9275887508819043,
      "tokens_seen": 519110656
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008511935807422267,
      "loss": 1.5753,
      "theoretical_loss": 3.927536077057881,
      "tokens_seen": 519176192
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008511735205616851,
      "loss": 1.5902,
      "theoretical_loss": 3.927483411743962,
      "tokens_seen": 519241728
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008511534603811435,
      "loss": 1.6483,
      "theoretical_loss": 3.927430754937699,
      "tokens_seen": 519307264
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008511334002006018,
      "loss": 1.5533,
      "theoretical_loss": 3.9273781066366444,
      "tokens_seen": 519372800
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008511133400200602,
      "loss": 1.6394,
      "theoretical_loss": 3.9273254668383513,
      "tokens_seen": 519438336
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008510932798395186,
      "loss": 1.5954,
      "theoretical_loss": 3.9272728355403737,
      "tokens_seen": 519503872
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000851073219658977,
      "loss": 1.5294,
      "theoretical_loss": 3.927220212740267,
      "tokens_seen": 519569408
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008510531594784353,
      "loss": 1.5487,
      "theoretical_loss": 3.927167598435587,
      "tokens_seen": 519634944
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008510330992978938,
      "loss": 1.5729,
      "theoretical_loss": 3.9271149926238924,
      "tokens_seen": 519700480
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000851013039117352,
      "loss": 1.5498,
      "theoretical_loss": 3.92706239530274,
      "tokens_seen": 519766016
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008509929789368105,
      "loss": 1.6295,
      "theoretical_loss": 3.9270098064696906,
      "tokens_seen": 519831552
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008509729187562688,
      "loss": 1.6667,
      "theoretical_loss": 3.9269572261223034,
      "tokens_seen": 519897088
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008509528585757272,
      "loss": 1.719,
      "theoretical_loss": 3.9269046542581396,
      "tokens_seen": 519962624
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008509327983951856,
      "loss": 1.5316,
      "theoretical_loss": 3.926852090874763,
      "tokens_seen": 520028160
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850912738214644,
      "loss": 1.5447,
      "theoretical_loss": 3.9267995359697356,
      "tokens_seen": 520093696
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008508926780341023,
      "loss": 1.5462,
      "theoretical_loss": 3.9267469895406224,
      "tokens_seen": 520159232
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008508726178535608,
      "loss": 1.713,
      "theoretical_loss": 3.9266944515849884,
      "tokens_seen": 520224768
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850852557673019,
      "loss": 1.619,
      "theoretical_loss": 3.9266419221004005,
      "tokens_seen": 520290304
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008508324974924775,
      "loss": 1.6332,
      "theoretical_loss": 3.926589401084426,
      "tokens_seen": 520355840
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008508124373119358,
      "loss": 1.5951,
      "theoretical_loss": 3.9265368885346326,
      "tokens_seen": 520421376
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008507923771313942,
      "loss": 1.5693,
      "theoretical_loss": 3.9264843844485906,
      "tokens_seen": 520486912
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008507723169508526,
      "loss": 1.674,
      "theoretical_loss": 3.9264318888238705,
      "tokens_seen": 520552448
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008507522567703109,
      "loss": 1.5264,
      "theoretical_loss": 3.9263794016580427,
      "tokens_seen": 520617984
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008507321965897693,
      "loss": 1.6321,
      "theoretical_loss": 3.9263269229486797,
      "tokens_seen": 520683520
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008507121364092278,
      "loss": 1.597,
      "theoretical_loss": 3.9262744526933564,
      "tokens_seen": 520749056
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850692076228686,
      "loss": 1.5582,
      "theoretical_loss": 3.926221990889646,
      "tokens_seen": 520814592
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008506720160481445,
      "loss": 1.5381,
      "theoretical_loss": 3.9261695375351238,
      "tokens_seen": 520880128
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008506519558676027,
      "loss": 1.6321,
      "theoretical_loss": 3.9261170926273667,
      "tokens_seen": 520945664
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.4877818822860718,
      "objective/train/docs_used": 195804,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6571053266525269,
      "objective/train/original_loss": 1.6571054458618164,
      "objective/train/theoretical_loss": 3.926064656163952,
      "objective/train/tokens_used": 541471200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2390846312046051,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0499935150146484,
      "objective/train/weighted_lm_loss": 1.7400037050247192,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.926064656163952,
      "tokens_seen": 521011200
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008506318956870612,
      "loss": 1.681,
      "theoretical_loss": 3.926064656163952,
      "tokens_seen": 521011200
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008506118355065196,
      "loss": 1.6877,
      "theoretical_loss": 3.926012228142458,
      "tokens_seen": 521076736
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008505917753259779,
      "loss": 1.6522,
      "theoretical_loss": 3.9259598085604646,
      "tokens_seen": 521142272
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008505717151454363,
      "loss": 1.5564,
      "theoretical_loss": 3.9259073974155516,
      "tokens_seen": 521207808
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008505516549648948,
      "loss": 1.6742,
      "theoretical_loss": 3.9258549947053005,
      "tokens_seen": 521273344
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850531594784353,
      "loss": 1.5631,
      "theoretical_loss": 3.9258026004272946,
      "tokens_seen": 521338880
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008505115346038115,
      "loss": 1.6367,
      "theoretical_loss": 3.925750214579116,
      "tokens_seen": 521404416
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008504914744232698,
      "loss": 1.7537,
      "theoretical_loss": 3.9256978371583493,
      "tokens_seen": 521469952
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008504714142427282,
      "loss": 1.5956,
      "theoretical_loss": 3.925645468162581,
      "tokens_seen": 521535488
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008504513540621866,
      "loss": 1.8158,
      "theoretical_loss": 3.925593107589397,
      "tokens_seen": 521601024
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850431293881645,
      "loss": 1.5445,
      "theoretical_loss": 3.9255407554363835,
      "tokens_seen": 521666560
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008504112337011033,
      "loss": 1.7284,
      "theoretical_loss": 3.9254884117011306,
      "tokens_seen": 521732096
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008503911735205617,
      "loss": 1.5485,
      "theoretical_loss": 3.925436076381227,
      "tokens_seen": 521797632
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.00085037111334002,
      "loss": 1.6211,
      "theoretical_loss": 3.925383749474263,
      "tokens_seen": 521863168
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008503510531594785,
      "loss": 1.5507,
      "theoretical_loss": 3.92533143097783,
      "tokens_seen": 521928704
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008503309929789368,
      "loss": 1.5523,
      "theoretical_loss": 3.9252791208895204,
      "tokens_seen": 521994240
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008503109327983952,
      "loss": 1.5814,
      "theoretical_loss": 3.9252268192069266,
      "tokens_seen": 522059776
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008502908726178536,
      "loss": 1.6998,
      "theoretical_loss": 3.9251745259276447,
      "tokens_seen": 522125312
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850270812437312,
      "loss": 1.7234,
      "theoretical_loss": 3.9251222410492694,
      "tokens_seen": 522190848
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008502507522567703,
      "loss": 1.7072,
      "theoretical_loss": 3.925069964569396,
      "tokens_seen": 522256384
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008502306920762287,
      "loss": 1.6071,
      "theoretical_loss": 3.9250176964856234,
      "tokens_seen": 522321920
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850210631895687,
      "loss": 1.7023,
      "theoretical_loss": 3.9249654367955484,
      "tokens_seen": 522387456
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008501905717151455,
      "loss": 1.5816,
      "theoretical_loss": 3.924913185496772,
      "tokens_seen": 522452992
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008501705115346038,
      "loss": 1.4997,
      "theoretical_loss": 3.9248609425868923,
      "tokens_seen": 522518528
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008501504513540622,
      "loss": 1.5808,
      "theoretical_loss": 3.9248087080635123,
      "tokens_seen": 522584064
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008501303911735205,
      "loss": 1.7183,
      "theoretical_loss": 3.9247564819242333,
      "tokens_seen": 522649600
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850110330992979,
      "loss": 1.6624,
      "theoretical_loss": 3.924704264166659,
      "tokens_seen": 522715136
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008500902708124373,
      "loss": 1.6772,
      "theoretical_loss": 3.9246520547883934,
      "tokens_seen": 522780672
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008500702106318957,
      "loss": 1.7125,
      "theoretical_loss": 3.9245998537870426,
      "tokens_seen": 522846208
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000850050150451354,
      "loss": 1.5976,
      "theoretical_loss": 3.924547661160211,
      "tokens_seen": 522911744
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008500300902708124,
      "loss": 1.4942,
      "theoretical_loss": 3.9244954769055074,
      "tokens_seen": 522977280
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008500100300902708,
      "loss": 1.6971,
      "theoretical_loss": 3.9244433010205393,
      "tokens_seen": 523042816
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008499899699097292,
      "loss": 1.6379,
      "theoretical_loss": 3.9243911335029154,
      "tokens_seen": 523108352
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008499699097291875,
      "loss": 1.6794,
      "theoretical_loss": 3.924338974350247,
      "tokens_seen": 523173888
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000849949849548646,
      "loss": 1.6712,
      "theoretical_loss": 3.924286823560144,
      "tokens_seen": 523239424
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008499297893681043,
      "loss": 1.5988,
      "theoretical_loss": 3.9242346811302187,
      "tokens_seen": 523304960
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008499097291875627,
      "loss": 1.5973,
      "theoretical_loss": 3.924182547058085,
      "tokens_seen": 523370496
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000849889669007021,
      "loss": 1.5487,
      "theoretical_loss": 3.9241304213413564,
      "tokens_seen": 523436032
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008498696088264794,
      "loss": 1.6642,
      "theoretical_loss": 3.9240783039776472,
      "tokens_seen": 523501568
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008498495486459379,
      "loss": 1.6403,
      "theoretical_loss": 3.9240261949645747,
      "tokens_seen": 523567104
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008498294884653962,
      "loss": 1.7722,
      "theoretical_loss": 3.9239740942997554,
      "tokens_seen": 523632640
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008498094282848546,
      "loss": 1.5607,
      "theoretical_loss": 3.9239220019808068,
      "tokens_seen": 523698176
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000849789368104313,
      "loss": 1.5919,
      "theoretical_loss": 3.9238699180053485,
      "tokens_seen": 523763712
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008497693079237713,
      "loss": 1.6647,
      "theoretical_loss": 3.923817842371,
      "tokens_seen": 523829248
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008497492477432297,
      "loss": 1.7784,
      "theoretical_loss": 3.923765775075382,
      "tokens_seen": 523894784
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008497291875626882,
      "loss": 1.5481,
      "theoretical_loss": 3.9237137161161173,
      "tokens_seen": 523960320
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008497091273821464,
      "loss": 1.6527,
      "theoretical_loss": 3.923661665490828,
      "tokens_seen": 524025856
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008496890672016049,
      "loss": 1.6881,
      "theoretical_loss": 3.9236096231971374,
      "tokens_seen": 524091392
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008496690070210631,
      "loss": 1.6147,
      "theoretical_loss": 3.9235575892326713,
      "tokens_seen": 524156928
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008496489468405216,
      "loss": 1.6174,
      "theoretical_loss": 3.9235055635950555,
      "tokens_seen": 524222464
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.49302032589912415,
      "objective/train/docs_used": 196964,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4459208250045776,
      "objective/train/original_loss": 1.4459209442138672,
      "objective/train/theoretical_loss": 3.9234535462819156,
      "objective/train/tokens_used": 544748000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24345248937606812,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505396127700806,
      "objective/train/weighted_lm_loss": 1.5191477537155151,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9234535462819156,
      "tokens_seen": 524288000
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.00084962888665998,
      "loss": 1.5534,
      "theoretical_loss": 3.9234535462819156,
      "tokens_seen": 524288000
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008496088264794383,
      "loss": 1.6074,
      "theoretical_loss": 3.9234015372908804,
      "tokens_seen": 524353536
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008495887662988967,
      "loss": 1.6905,
      "theoretical_loss": 3.923349536619578,
      "tokens_seen": 524419072
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008495687061183552,
      "loss": 1.6289,
      "theoretical_loss": 3.923297544265639,
      "tokens_seen": 524484608
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008495486459378134,
      "loss": 1.591,
      "theoretical_loss": 3.923245560226693,
      "tokens_seen": 524550144
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008495285857572719,
      "loss": 1.5203,
      "theoretical_loss": 3.923193584500372,
      "tokens_seen": 524615680
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008495085255767302,
      "loss": 1.6374,
      "theoretical_loss": 3.9231416170843083,
      "tokens_seen": 524681216
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008494884653961886,
      "loss": 1.6512,
      "theoretical_loss": 3.9230896579761354,
      "tokens_seen": 524746752
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000849468405215647,
      "loss": 1.4736,
      "theoretical_loss": 3.9230377071734885,
      "tokens_seen": 524812288
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008494483450351053,
      "loss": 1.7909,
      "theoretical_loss": 3.922985764674002,
      "tokens_seen": 524877824
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008494282848545637,
      "loss": 1.6283,
      "theoretical_loss": 3.922933830475314,
      "tokens_seen": 524943360
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008494082246740221,
      "loss": 1.4642,
      "theoretical_loss": 3.9228819045750596,
      "tokens_seen": 525008896
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008493881644934804,
      "loss": 1.6107,
      "theoretical_loss": 3.9228299869708794,
      "tokens_seen": 525074432
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008493681043129389,
      "loss": 1.5806,
      "theoretical_loss": 3.9227780776604115,
      "tokens_seen": 525139968
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008493480441323972,
      "loss": 1.6152,
      "theoretical_loss": 3.922726176641297,
      "tokens_seen": 525205504
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008493279839518556,
      "loss": 1.6536,
      "theoretical_loss": 3.922674283911176,
      "tokens_seen": 525271040
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000849307923771314,
      "loss": 1.6945,
      "theoretical_loss": 3.9226223994676923,
      "tokens_seen": 525336576
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008492878635907724,
      "loss": 1.6635,
      "theoretical_loss": 3.9225705233084875,
      "tokens_seen": 525402112
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008492678034102307,
      "loss": 1.7497,
      "theoretical_loss": 3.922518655431207,
      "tokens_seen": 525467648
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008492477432296891,
      "loss": 1.6683,
      "theoretical_loss": 3.922466795833495,
      "tokens_seen": 525533184
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008492276830491474,
      "loss": 1.6898,
      "theoretical_loss": 3.9224149445129983,
      "tokens_seen": 525598720
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008492076228686059,
      "loss": 1.6656,
      "theoretical_loss": 3.922363101467364,
      "tokens_seen": 525664256
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008491875626880642,
      "loss": 1.5702,
      "theoretical_loss": 3.9223112666942397,
      "tokens_seen": 525729792
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008491675025075226,
      "loss": 1.5852,
      "theoretical_loss": 3.9222594401912745,
      "tokens_seen": 525795328
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008491474423269809,
      "loss": 1.6441,
      "theoretical_loss": 3.922207621956119,
      "tokens_seen": 525860864
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008491273821464394,
      "loss": 1.5662,
      "theoretical_loss": 3.9221558119864226,
      "tokens_seen": 525926400
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008491073219658977,
      "loss": 1.64,
      "theoretical_loss": 3.922104010279839,
      "tokens_seen": 525991936
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008490872617853561,
      "loss": 1.6588,
      "theoretical_loss": 3.922052216834019,
      "tokens_seen": 526057472
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008490672016048144,
      "loss": 1.5135,
      "theoretical_loss": 3.9220004316466186,
      "tokens_seen": 526123008
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008490471414242728,
      "loss": 1.7018,
      "theoretical_loss": 3.921948654715291,
      "tokens_seen": 526188544
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008490270812437312,
      "loss": 1.6147,
      "theoretical_loss": 3.9218968860376924,
      "tokens_seen": 526254080
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008490070210631896,
      "loss": 1.6883,
      "theoretical_loss": 3.9218451256114792,
      "tokens_seen": 526319616
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008489869608826479,
      "loss": 1.6665,
      "theoretical_loss": 3.9217933734343093,
      "tokens_seen": 526385152
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008489669007021064,
      "loss": 1.6813,
      "theoretical_loss": 3.9217416295038414,
      "tokens_seen": 526450688
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008489468405215647,
      "loss": 1.5609,
      "theoretical_loss": 3.921689893817735,
      "tokens_seen": 526516224
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008489267803410231,
      "loss": 1.6527,
      "theoretical_loss": 3.9216381663736506,
      "tokens_seen": 526581760
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008489067201604815,
      "loss": 1.6266,
      "theoretical_loss": 3.9215864471692488,
      "tokens_seen": 526647296
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008488866599799398,
      "loss": 1.6327,
      "theoretical_loss": 3.921534736202193,
      "tokens_seen": 526712832
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008488665997993982,
      "loss": 1.7006,
      "theoretical_loss": 3.921483033470146,
      "tokens_seen": 526778368
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008488465396188566,
      "loss": 1.5604,
      "theoretical_loss": 3.921431338970773,
      "tokens_seen": 526843904
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008488264794383149,
      "loss": 1.5901,
      "theoretical_loss": 3.921379652701738,
      "tokens_seen": 526909440
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008488064192577734,
      "loss": 1.6418,
      "theoretical_loss": 3.9213279746607084,
      "tokens_seen": 526974976
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008487863590772316,
      "loss": 1.5446,
      "theoretical_loss": 3.9212763048453505,
      "tokens_seen": 527040512
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008487662988966901,
      "loss": 1.6844,
      "theoretical_loss": 3.921224643253332,
      "tokens_seen": 527106048
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008487462387161485,
      "loss": 1.561,
      "theoretical_loss": 3.9211729898823235,
      "tokens_seen": 527171584
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008487261785356068,
      "loss": 1.584,
      "theoretical_loss": 3.9211213447299937,
      "tokens_seen": 527237120
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008487061183550652,
      "loss": 1.6675,
      "theoretical_loss": 3.921069707794014,
      "tokens_seen": 527302656
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008486860581745235,
      "loss": 1.6663,
      "theoretical_loss": 3.9210180790720566,
      "tokens_seen": 527368192
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008486659979939819,
      "loss": 1.6013,
      "theoretical_loss": 3.920966458561794,
      "tokens_seen": 527433728
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008486459378134404,
      "loss": 1.6349,
      "theoretical_loss": 3.9209148462609003,
      "tokens_seen": 527499264
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.4926772713661194,
      "objective/train/docs_used": 198271,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.406691074371338,
      "objective/train/original_loss": 1.4066911935806274,
      "objective/train/theoretical_loss": 3.9208632421670497,
      "objective/train/tokens_used": 548024800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2433115839958191,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505046844482422,
      "objective/train/weighted_lm_loss": 1.4775863885879517,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9208632421670497,
      "tokens_seen": 527564800
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008486258776328986,
      "loss": 1.559,
      "theoretical_loss": 3.9208632421670497,
      "tokens_seen": 527564800
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008486058174523571,
      "loss": 1.6536,
      "theoretical_loss": 3.920811646277918,
      "tokens_seen": 527630336
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008485857572718155,
      "loss": 1.6171,
      "theoretical_loss": 3.920760058591182,
      "tokens_seen": 527695872
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008485656970912738,
      "loss": 1.6706,
      "theoretical_loss": 3.92070847910452,
      "tokens_seen": 527761408
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008485456369107322,
      "loss": 1.5576,
      "theoretical_loss": 3.9206569078156104,
      "tokens_seen": 527826944
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008485255767301906,
      "loss": 1.7217,
      "theoretical_loss": 3.920605344722132,
      "tokens_seen": 527892480
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008485055165496489,
      "loss": 1.6116,
      "theoretical_loss": 3.9205537898217644,
      "tokens_seen": 527958016
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008484854563691074,
      "loss": 1.6864,
      "theoretical_loss": 3.920502243112191,
      "tokens_seen": 528023552
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008484653961885656,
      "loss": 1.6173,
      "theoretical_loss": 3.920450704591093,
      "tokens_seen": 528089088
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008484453360080241,
      "loss": 1.657,
      "theoretical_loss": 3.920399174256154,
      "tokens_seen": 528154624
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008484252758274824,
      "loss": 1.5317,
      "theoretical_loss": 3.920347652105058,
      "tokens_seen": 528220160
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008484052156469408,
      "loss": 1.6393,
      "theoretical_loss": 3.92029613813549,
      "tokens_seen": 528285696
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008483851554663992,
      "loss": 1.6641,
      "theoretical_loss": 3.9202446323451365,
      "tokens_seen": 528351232
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008483650952858576,
      "loss": 1.6252,
      "theoretical_loss": 3.920193134731684,
      "tokens_seen": 528416768
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000848345035105316,
      "loss": 1.6232,
      "theoretical_loss": 3.920141645292821,
      "tokens_seen": 528482304
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008483249749247744,
      "loss": 1.5546,
      "theoretical_loss": 3.9200901640262362,
      "tokens_seen": 528547840
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008483049147442328,
      "loss": 1.6653,
      "theoretical_loss": 3.92003869092962,
      "tokens_seen": 528613376
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008482848545636911,
      "loss": 1.6492,
      "theoretical_loss": 3.919987226000662,
      "tokens_seen": 528678912
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008482647943831495,
      "loss": 1.5318,
      "theoretical_loss": 3.919935769237055,
      "tokens_seen": 528744448
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008482447342026078,
      "loss": 1.5988,
      "theoretical_loss": 3.9198843206364913,
      "tokens_seen": 528809984
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008482246740220663,
      "loss": 1.7496,
      "theoretical_loss": 3.919832880196664,
      "tokens_seen": 528875520
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008482046138415246,
      "loss": 1.6519,
      "theoretical_loss": 3.919781447915269,
      "tokens_seen": 528941056
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000848184553660983,
      "loss": 1.6635,
      "theoretical_loss": 3.91973002379,
      "tokens_seen": 529006592
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008481644934804413,
      "loss": 1.639,
      "theoretical_loss": 3.919678607818555,
      "tokens_seen": 529072128
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008481444332998998,
      "loss": 1.5742,
      "theoretical_loss": 3.9196271999986303,
      "tokens_seen": 529137664
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008481243731193581,
      "loss": 1.5977,
      "theoretical_loss": 3.919575800327925,
      "tokens_seen": 529203200
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008481043129388165,
      "loss": 1.6649,
      "theoretical_loss": 3.919524408804137,
      "tokens_seen": 529268736
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008480842527582748,
      "loss": 1.5606,
      "theoretical_loss": 3.919473025424969,
      "tokens_seen": 529334272
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008480641925777332,
      "loss": 1.6166,
      "theoretical_loss": 3.9194216501881187,
      "tokens_seen": 529399808
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008480441323971916,
      "loss": 1.5941,
      "theoretical_loss": 3.9193702830912907,
      "tokens_seen": 529465344
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.00084802407221665,
      "loss": 1.7999,
      "theoretical_loss": 3.9193189241321873,
      "tokens_seen": 529530880
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008480040120361083,
      "loss": 1.5353,
      "theoretical_loss": 3.9192675733085123,
      "tokens_seen": 529596416
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008479839518555668,
      "loss": 1.6426,
      "theoretical_loss": 3.91921623061797,
      "tokens_seen": 529661952
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008479638916750251,
      "loss": 1.6522,
      "theoretical_loss": 3.919164896058267,
      "tokens_seen": 529727488
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008479438314944835,
      "loss": 1.6843,
      "theoretical_loss": 3.9191135696271098,
      "tokens_seen": 529793024
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008479237713139419,
      "loss": 1.6878,
      "theoretical_loss": 3.9190622513222055,
      "tokens_seen": 529858560
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008479037111334002,
      "loss": 1.7081,
      "theoretical_loss": 3.9190109411412632,
      "tokens_seen": 529924096
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008478836509528586,
      "loss": 1.6211,
      "theoretical_loss": 3.918959639081992,
      "tokens_seen": 529989632
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847863590772317,
      "loss": 1.7312,
      "theoretical_loss": 3.9189083451421025,
      "tokens_seen": 530055168
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008478435305917753,
      "loss": 1.4901,
      "theoretical_loss": 3.918857059319306,
      "tokens_seen": 530120704
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008478234704112338,
      "loss": 1.4875,
      "theoretical_loss": 3.918805781611315,
      "tokens_seen": 530186240
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847803410230692,
      "loss": 1.583,
      "theoretical_loss": 3.918754512015842,
      "tokens_seen": 530251776
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008477833500501505,
      "loss": 1.6291,
      "theoretical_loss": 3.9187032505306023,
      "tokens_seen": 530317312
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008477632898696089,
      "loss": 1.6093,
      "theoretical_loss": 3.91865199715331,
      "tokens_seen": 530382848
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008477432296890672,
      "loss": 1.5456,
      "theoretical_loss": 3.918600751881681,
      "tokens_seen": 530448384
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008477231695085256,
      "loss": 1.6374,
      "theoretical_loss": 3.9185495147134324,
      "tokens_seen": 530513920
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847703109327984,
      "loss": 1.6139,
      "theoretical_loss": 3.918498285646282,
      "tokens_seen": 530579456
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008476830491474423,
      "loss": 1.6597,
      "theoretical_loss": 3.9184470646779492,
      "tokens_seen": 530644992
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008476629889669008,
      "loss": 1.6118,
      "theoretical_loss": 3.918395851806153,
      "tokens_seen": 530710528
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847642928786359,
      "loss": 1.6052,
      "theoretical_loss": 3.9183446470286136,
      "tokens_seen": 530776064
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.4919034540653229,
      "objective/train/docs_used": 199409,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5748982429504395,
      "objective/train/original_loss": 1.5748980045318604,
      "objective/train/theoretical_loss": 3.9182934503430538,
      "objective/train/tokens_used": 551301600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2425510436296463,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050423264503479,
      "objective/train/weighted_lm_loss": 1.654366374015808,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9182934503430538,
      "tokens_seen": 530841600
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008476228686058175,
      "loss": 1.5934,
      "theoretical_loss": 3.9182934503430538,
      "tokens_seen": 530841600
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008476028084252759,
      "loss": 1.6358,
      "theoretical_loss": 3.918242261747195,
      "tokens_seen": 530907136
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008475827482447342,
      "loss": 1.6104,
      "theoretical_loss": 3.9181910812387604,
      "tokens_seen": 530972672
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008475626880641926,
      "loss": 1.5103,
      "theoretical_loss": 3.9181399088154745,
      "tokens_seen": 531038208
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847542627883651,
      "loss": 1.5375,
      "theoretical_loss": 3.918088744475064,
      "tokens_seen": 531103744
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008475225677031093,
      "loss": 1.724,
      "theoretical_loss": 3.9180375882152525,
      "tokens_seen": 531169280
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008475025075225678,
      "loss": 1.6848,
      "theoretical_loss": 3.9179864400337685,
      "tokens_seen": 531234816
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847482447342026,
      "loss": 1.6819,
      "theoretical_loss": 3.91793529992834,
      "tokens_seen": 531300352
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008474623871614845,
      "loss": 1.5828,
      "theoretical_loss": 3.9178841678966956,
      "tokens_seen": 531365888
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008474423269809428,
      "loss": 1.5993,
      "theoretical_loss": 3.9178330439365654,
      "tokens_seen": 531431424
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008474222668004012,
      "loss": 1.6697,
      "theoretical_loss": 3.917781928045679,
      "tokens_seen": 531496960
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008474022066198596,
      "loss": 1.7663,
      "theoretical_loss": 3.9177308202217693,
      "tokens_seen": 531562496
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847382146439318,
      "loss": 1.6602,
      "theoretical_loss": 3.9176797204625693,
      "tokens_seen": 531628032
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008473620862587763,
      "loss": 1.6338,
      "theoretical_loss": 3.9176286287658106,
      "tokens_seen": 531693568
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008473420260782348,
      "loss": 1.716,
      "theoretical_loss": 3.9175775451292285,
      "tokens_seen": 531759104
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847321965897693,
      "loss": 1.7191,
      "theoretical_loss": 3.917526469550559,
      "tokens_seen": 531824640
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008473019057171515,
      "loss": 1.5278,
      "theoretical_loss": 3.917475402027537,
      "tokens_seen": 531890176
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008472818455366098,
      "loss": 1.5785,
      "theoretical_loss": 3.9174243425579007,
      "tokens_seen": 531955712
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008472617853560682,
      "loss": 1.642,
      "theoretical_loss": 3.917373291139388,
      "tokens_seen": 532021248
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008472417251755266,
      "loss": 1.5372,
      "theoretical_loss": 3.9173222477697376,
      "tokens_seen": 532086784
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847221664994985,
      "loss": 1.5666,
      "theoretical_loss": 3.917271212446689,
      "tokens_seen": 532152320
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008472016048144433,
      "loss": 1.5001,
      "theoretical_loss": 3.917220185167984,
      "tokens_seen": 532217856
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008471815446339017,
      "loss": 1.568,
      "theoretical_loss": 3.9171691659313637,
      "tokens_seen": 532283392
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.00084716148445336,
      "loss": 1.8325,
      "theoretical_loss": 3.9171181547345704,
      "tokens_seen": 532348928
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008471414242728185,
      "loss": 1.5937,
      "theoretical_loss": 3.917067151575348,
      "tokens_seen": 532414464
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008471213640922768,
      "loss": 1.6996,
      "theoretical_loss": 3.917016156451441,
      "tokens_seen": 532480000
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008471013039117352,
      "loss": 1.7402,
      "theoretical_loss": 3.9169651693605942,
      "tokens_seen": 532545536
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008470812437311935,
      "loss": 1.6802,
      "theoretical_loss": 3.916914190300555,
      "tokens_seen": 532611072
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000847061183550652,
      "loss": 1.6203,
      "theoretical_loss": 3.916863219269069,
      "tokens_seen": 532676608
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008470411233701103,
      "loss": 1.524,
      "theoretical_loss": 3.916812256263886,
      "tokens_seen": 532742144
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008470210631895687,
      "loss": 1.6564,
      "theoretical_loss": 3.9167613012827536,
      "tokens_seen": 532807680
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008470010030090271,
      "loss": 1.6716,
      "theoretical_loss": 3.9167103543234223,
      "tokens_seen": 532873216
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008469809428284855,
      "loss": 1.6233,
      "theoretical_loss": 3.9166594153836427,
      "tokens_seen": 532938752
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008469608826479438,
      "loss": 1.6648,
      "theoretical_loss": 3.9166084844611664,
      "tokens_seen": 533004288
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008469408224674023,
      "loss": 1.6156,
      "theoretical_loss": 3.9165575615537467,
      "tokens_seen": 533069824
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008469207622868605,
      "loss": 1.668,
      "theoretical_loss": 3.916506646659136,
      "tokens_seen": 533135360
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000846900702106319,
      "loss": 1.6171,
      "theoretical_loss": 3.9164557397750897,
      "tokens_seen": 533200896
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008468806419257773,
      "loss": 1.6386,
      "theoretical_loss": 3.916404840899363,
      "tokens_seen": 533266432
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008468605817452357,
      "loss": 1.6073,
      "theoretical_loss": 3.9163539500297118,
      "tokens_seen": 533331968
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008468405215646942,
      "loss": 1.6268,
      "theoretical_loss": 3.916303067163893,
      "tokens_seen": 533397504
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008468204613841524,
      "loss": 1.6233,
      "theoretical_loss": 3.916252192299665,
      "tokens_seen": 533463040
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008468004012036109,
      "loss": 1.6126,
      "theoretical_loss": 3.916201325434787,
      "tokens_seen": 533528576
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008467803410230693,
      "loss": 1.5906,
      "theoretical_loss": 3.916150466567019,
      "tokens_seen": 533594112
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008467602808425276,
      "loss": 1.6757,
      "theoretical_loss": 3.9160996156941206,
      "tokens_seen": 533659648
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000846740220661986,
      "loss": 1.6084,
      "theoretical_loss": 3.9160487728138538,
      "tokens_seen": 533725184
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008467201604814443,
      "loss": 1.63,
      "theoretical_loss": 3.915997937923982,
      "tokens_seen": 533790720
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008467001003009027,
      "loss": 1.5667,
      "theoretical_loss": 3.915947111022269,
      "tokens_seen": 533856256
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008466800401203612,
      "loss": 1.7042,
      "theoretical_loss": 3.9158962921064773,
      "tokens_seen": 533921792
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008466599799398194,
      "loss": 1.6378,
      "theoretical_loss": 3.9158454811743733,
      "tokens_seen": 533987328
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008466399197592779,
      "loss": 1.6598,
      "theoretical_loss": 3.9157946782237234,
      "tokens_seen": 534052864
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.49396708607673645,
      "objective/train/docs_used": 200516,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5956155061721802,
      "objective/train/original_loss": 1.5956156253814697,
      "objective/train/theoretical_loss": 3.9157438832522944,
      "objective/train/tokens_used": 554578400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24434596300125122,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050639033317566,
      "objective/train/weighted_lm_loss": 1.6763161420822144,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9157438832522944,
      "tokens_seen": 534118400
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008466198595787363,
      "loss": 1.6087,
      "theoretical_loss": 3.9157438832522944,
      "tokens_seen": 534118400
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008465997993981946,
      "loss": 1.7316,
      "theoretical_loss": 3.9156930962578533,
      "tokens_seen": 534183936
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000846579739217653,
      "loss": 1.7161,
      "theoretical_loss": 3.915642317238171,
      "tokens_seen": 534249472
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008465596790371114,
      "loss": 1.6406,
      "theoretical_loss": 3.915591546191015,
      "tokens_seen": 534315008
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008465396188565697,
      "loss": 1.6527,
      "theoretical_loss": 3.9155407831141575,
      "tokens_seen": 534380544
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008465195586760282,
      "loss": 1.6793,
      "theoretical_loss": 3.9154900280053697,
      "tokens_seen": 534446080
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008464994984954864,
      "loss": 1.6632,
      "theoretical_loss": 3.915439280862423,
      "tokens_seen": 534511616
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008464794383149449,
      "loss": 1.6393,
      "theoretical_loss": 3.9153885416830922,
      "tokens_seen": 534577152
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008464593781344032,
      "loss": 1.5558,
      "theoretical_loss": 3.915337810465151,
      "tokens_seen": 534642688
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008464393179538616,
      "loss": 1.6084,
      "theoretical_loss": 3.9152870872063747,
      "tokens_seen": 534708224
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.00084641925777332,
      "loss": 1.6117,
      "theoretical_loss": 3.915236371904539,
      "tokens_seen": 534773760
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008463991975927784,
      "loss": 1.7532,
      "theoretical_loss": 3.915185664557421,
      "tokens_seen": 534839296
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008463791374122367,
      "loss": 1.5398,
      "theoretical_loss": 3.915134965162798,
      "tokens_seen": 534904832
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008463590772316952,
      "loss": 1.6308,
      "theoretical_loss": 3.915084273718449,
      "tokens_seen": 534970368
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008463390170511534,
      "loss": 1.6037,
      "theoretical_loss": 3.915033590222153,
      "tokens_seen": 535035904
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008463189568706119,
      "loss": 1.5928,
      "theoretical_loss": 3.9149829146716923,
      "tokens_seen": 535101440
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008462988966900702,
      "loss": 1.6595,
      "theoretical_loss": 3.914932247064847,
      "tokens_seen": 535166976
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008462788365095286,
      "loss": 1.6967,
      "theoretical_loss": 3.914881587399399,
      "tokens_seen": 535232512
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000846258776328987,
      "loss": 1.5722,
      "theoretical_loss": 3.914830935673132,
      "tokens_seen": 535298048
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008462387161484454,
      "loss": 1.5449,
      "theoretical_loss": 3.9147802918838304,
      "tokens_seen": 535363584
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008462186559679037,
      "loss": 1.5417,
      "theoretical_loss": 3.914729656029279,
      "tokens_seen": 535429120
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008461985957873621,
      "loss": 1.6225,
      "theoretical_loss": 3.9146790281072628,
      "tokens_seen": 535494656
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008461785356068205,
      "loss": 1.5895,
      "theoretical_loss": 3.914628408115569,
      "tokens_seen": 535560192
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008461584754262789,
      "loss": 1.6551,
      "theoretical_loss": 3.9145777960519856,
      "tokens_seen": 535625728
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008461384152457372,
      "loss": 1.6158,
      "theoretical_loss": 3.9145271919143005,
      "tokens_seen": 535691264
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008461183550651956,
      "loss": 1.6353,
      "theoretical_loss": 3.9144765957003034,
      "tokens_seen": 535756800
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008460982948846539,
      "loss": 1.5529,
      "theoretical_loss": 3.9144260074077843,
      "tokens_seen": 535822336
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008460782347041124,
      "loss": 1.7125,
      "theoretical_loss": 3.914375427034535,
      "tokens_seen": 535887872
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008460581745235707,
      "loss": 1.6906,
      "theoretical_loss": 3.914324854578347,
      "tokens_seen": 535953408
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008460381143430291,
      "loss": 1.6646,
      "theoretical_loss": 3.914274290037013,
      "tokens_seen": 536018944
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008460180541624875,
      "loss": 1.6138,
      "theoretical_loss": 3.9142237334083276,
      "tokens_seen": 536084480
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008459979939819459,
      "loss": 1.6063,
      "theoretical_loss": 3.914173184690085,
      "tokens_seen": 536150016
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008459779338014042,
      "loss": 1.5708,
      "theoretical_loss": 3.9141226438800802,
      "tokens_seen": 536215552
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008459578736208627,
      "loss": 1.6743,
      "theoretical_loss": 3.9140721109761105,
      "tokens_seen": 536281088
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008459378134403209,
      "loss": 1.6015,
      "theoretical_loss": 3.914021585975973,
      "tokens_seen": 536346624
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008459177532597794,
      "loss": 1.7059,
      "theoretical_loss": 3.9139710688774656,
      "tokens_seen": 536412160
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008458976930792377,
      "loss": 1.6028,
      "theoretical_loss": 3.913920559678388,
      "tokens_seen": 536477696
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008458776328986961,
      "loss": 1.5366,
      "theoretical_loss": 3.9138700583765402,
      "tokens_seen": 536543232
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008458575727181545,
      "loss": 1.5262,
      "theoretical_loss": 3.9138195649697227,
      "tokens_seen": 536608768
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008458375125376128,
      "loss": 1.6228,
      "theoretical_loss": 3.9137690794557374,
      "tokens_seen": 536674304
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008458174523570712,
      "loss": 1.7002,
      "theoretical_loss": 3.913718601832387,
      "tokens_seen": 536739840
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008457973921765297,
      "loss": 1.5999,
      "theoretical_loss": 3.9136681320974738,
      "tokens_seen": 536805376
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008457773319959879,
      "loss": 1.6347,
      "theoretical_loss": 3.9136176702488044,
      "tokens_seen": 536870912
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008457572718154464,
      "loss": 1.5275,
      "theoretical_loss": 3.913567216284182,
      "tokens_seen": 536936448
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008457372116349046,
      "loss": 1.5859,
      "theoretical_loss": 3.913516770201414,
      "tokens_seen": 537001984
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008457171514543631,
      "loss": 1.5626,
      "theoretical_loss": 3.9134663319983076,
      "tokens_seen": 537067520
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008456970912738215,
      "loss": 1.5884,
      "theoretical_loss": 3.91341590167267,
      "tokens_seen": 537133056
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008456770310932798,
      "loss": 1.5622,
      "theoretical_loss": 3.91336547922231,
      "tokens_seen": 537198592
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008456569709127382,
      "loss": 1.6479,
      "theoretical_loss": 3.9133150646450376,
      "tokens_seen": 537264128
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008456369107321967,
      "loss": 1.6251,
      "theoretical_loss": 3.913264657938663,
      "tokens_seen": 537329664
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.4932006597518921,
      "objective/train/docs_used": 201641,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4497480392456055,
      "objective/train/original_loss": 1.449748158454895,
      "objective/train/theoretical_loss": 3.9132142591009975,
      "objective/train/tokens_used": 557855200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2437322437763214,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050559163093567,
      "objective/train/weighted_lm_loss": 1.5229429006576538,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9132142591009975,
      "tokens_seen": 537395200
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008456168505516549,
      "loss": 1.5278,
      "theoretical_loss": 3.9132142591009975,
      "tokens_seen": 537395200
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008455967903711134,
      "loss": 1.6539,
      "theoretical_loss": 3.913163868129854,
      "tokens_seen": 537460736
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008455767301905716,
      "loss": 1.6771,
      "theoretical_loss": 3.913113485023046,
      "tokens_seen": 537526272
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008455566700100301,
      "loss": 1.5341,
      "theoretical_loss": 3.9130631097783866,
      "tokens_seen": 537591808
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008455366098294885,
      "loss": 1.5754,
      "theoretical_loss": 3.9130127423936907,
      "tokens_seen": 537657344
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008455165496489468,
      "loss": 1.6758,
      "theoretical_loss": 3.912962382866774,
      "tokens_seen": 537722880
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008454964894684052,
      "loss": 1.6715,
      "theoretical_loss": 3.9129120311954537,
      "tokens_seen": 537788416
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008454764292878636,
      "loss": 1.6385,
      "theoretical_loss": 3.9128616873775472,
      "tokens_seen": 537853952
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008454563691073219,
      "loss": 1.6726,
      "theoretical_loss": 3.9128113514108733,
      "tokens_seen": 537919488
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008454363089267804,
      "loss": 1.6917,
      "theoretical_loss": 3.9127610232932506,
      "tokens_seen": 537985024
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008454162487462387,
      "loss": 1.6347,
      "theoretical_loss": 3.912710703022499,
      "tokens_seen": 538050560
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008453961885656971,
      "loss": 1.6559,
      "theoretical_loss": 3.9126603905964403,
      "tokens_seen": 538116096
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008453761283851554,
      "loss": 1.5037,
      "theoretical_loss": 3.9126100860128963,
      "tokens_seen": 538181632
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008453560682046138,
      "loss": 1.6297,
      "theoretical_loss": 3.9125597892696895,
      "tokens_seen": 538247168
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008453360080240723,
      "loss": 1.732,
      "theoretical_loss": 3.912509500364643,
      "tokens_seen": 538312704
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008453159478435306,
      "loss": 1.7173,
      "theoretical_loss": 3.9124592192955827,
      "tokens_seen": 538378240
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000845295887662989,
      "loss": 1.5541,
      "theoretical_loss": 3.9124089460603324,
      "tokens_seen": 538443776
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008452758274824474,
      "loss": 1.5886,
      "theoretical_loss": 3.9123586806567188,
      "tokens_seen": 538509312
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008452557673019058,
      "loss": 1.5959,
      "theoretical_loss": 3.9123084230825698,
      "tokens_seen": 538574848
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008452357071213641,
      "loss": 1.7745,
      "theoretical_loss": 3.9122581733357125,
      "tokens_seen": 538640384
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008452156469408225,
      "loss": 1.5802,
      "theoretical_loss": 3.9122079314139766,
      "tokens_seen": 538705920
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008451955867602809,
      "loss": 1.693,
      "theoretical_loss": 3.9121576973151906,
      "tokens_seen": 538771456
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008451755265797393,
      "loss": 1.5976,
      "theoretical_loss": 3.912107471037186,
      "tokens_seen": 538836992
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008451554663991976,
      "loss": 1.6027,
      "theoretical_loss": 3.9120572525777937,
      "tokens_seen": 538902528
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000845135406218656,
      "loss": 1.6657,
      "theoretical_loss": 3.9120070419348463,
      "tokens_seen": 538968064
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008451153460381143,
      "loss": 1.4962,
      "theoretical_loss": 3.9119568391061765,
      "tokens_seen": 539033600
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008450952858575728,
      "loss": 1.5403,
      "theoretical_loss": 3.911906644089619,
      "tokens_seen": 539099136
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008450752256770311,
      "loss": 1.5892,
      "theoretical_loss": 3.9118564568830085,
      "tokens_seen": 539164672
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008450551654964895,
      "loss": 1.6834,
      "theoretical_loss": 3.9118062774841804,
      "tokens_seen": 539230208
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008450351053159479,
      "loss": 1.7064,
      "theoretical_loss": 3.911756105890972,
      "tokens_seen": 539295744
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008450150451354063,
      "loss": 1.767,
      "theoretical_loss": 3.9117059421012197,
      "tokens_seen": 539361280
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008449949849548646,
      "loss": 1.6195,
      "theoretical_loss": 3.911655786112763,
      "tokens_seen": 539426816
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008449749247743231,
      "loss": 1.5392,
      "theoretical_loss": 3.91160563792344,
      "tokens_seen": 539492352
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008449548645937813,
      "loss": 1.6299,
      "theoretical_loss": 3.9115554975310918,
      "tokens_seen": 539557888
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008449348044132398,
      "loss": 1.6654,
      "theoretical_loss": 3.9115053649335585,
      "tokens_seen": 539623424
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008449147442326981,
      "loss": 1.6519,
      "theoretical_loss": 3.9114552401286824,
      "tokens_seen": 539688960
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008448946840521565,
      "loss": 1.612,
      "theoretical_loss": 3.911405123114305,
      "tokens_seen": 539754496
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008448746238716149,
      "loss": 1.7025,
      "theoretical_loss": 3.9113550138882722,
      "tokens_seen": 539820032
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008448545636910732,
      "loss": 1.6332,
      "theoretical_loss": 3.9113049124484256,
      "tokens_seen": 539885568
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008448345035105316,
      "loss": 1.7054,
      "theoretical_loss": 3.9112548187926124,
      "tokens_seen": 539951104
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008448144433299901,
      "loss": 1.6892,
      "theoretical_loss": 3.9112047329186783,
      "tokens_seen": 540016640
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008447943831494483,
      "loss": 1.64,
      "theoretical_loss": 3.911154654824469,
      "tokens_seen": 540082176
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008447743229689068,
      "loss": 1.5573,
      "theoretical_loss": 3.911104584507834,
      "tokens_seen": 540147712
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000844754262788365,
      "loss": 1.8344,
      "theoretical_loss": 3.91105452196662,
      "tokens_seen": 540213248
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008447342026078235,
      "loss": 1.5585,
      "theoretical_loss": 3.911004467198679,
      "tokens_seen": 540278784
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008447141424272819,
      "loss": 1.7415,
      "theoretical_loss": 3.9109544202018593,
      "tokens_seen": 540344320
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008446940822467402,
      "loss": 1.656,
      "theoretical_loss": 3.9109043809740127,
      "tokens_seen": 540409856
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008446740220661986,
      "loss": 1.5673,
      "theoretical_loss": 3.9108543495129915,
      "tokens_seen": 540475392
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008446539618856571,
      "loss": 1.7181,
      "theoretical_loss": 3.9108043258166485,
      "tokens_seen": 540540928
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008446339017051153,
      "loss": 1.5633,
      "theoretical_loss": 3.9107543098828375,
      "tokens_seen": 540606464
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.49212127923965454,
      "objective/train/docs_used": 202818,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4933364391326904,
      "objective/train/original_loss": 1.4933364391326904,
      "objective/train/theoretical_loss": 3.910704301709413,
      "objective/train/tokens_used": 561132000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24271930754184723,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504459142684937,
      "objective/train/weighted_lm_loss": 1.5686144828796387,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.910704301709413,
      "tokens_seen": 540672000
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008446138415245738,
      "loss": 1.6177,
      "theoretical_loss": 3.910704301709413,
      "tokens_seen": 540672000
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000844593781344032,
      "loss": 1.6962,
      "theoretical_loss": 3.91065430129423,
      "tokens_seen": 540737536
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008445737211634905,
      "loss": 1.5858,
      "theoretical_loss": 3.910604308635146,
      "tokens_seen": 540803072
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008445536609829489,
      "loss": 1.5983,
      "theoretical_loss": 3.9105543237300173,
      "tokens_seen": 540868608
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008445336008024072,
      "loss": 1.609,
      "theoretical_loss": 3.910504346576702,
      "tokens_seen": 540934144
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008445135406218656,
      "loss": 1.5867,
      "theoretical_loss": 3.9104543771730595,
      "tokens_seen": 540999680
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000844493480441324,
      "loss": 1.5341,
      "theoretical_loss": 3.9104044155169495,
      "tokens_seen": 541065216
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008444734202607823,
      "loss": 1.5828,
      "theoretical_loss": 3.9103544616062322,
      "tokens_seen": 541130752
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008444533600802408,
      "loss": 1.7507,
      "theoretical_loss": 3.910304515438769,
      "tokens_seen": 541196288
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000844433299899699,
      "loss": 1.7463,
      "theoretical_loss": 3.910254577012422,
      "tokens_seen": 541261824
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008444132397191575,
      "loss": 1.6767,
      "theoretical_loss": 3.910204646325055,
      "tokens_seen": 541327360
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008443931795386158,
      "loss": 1.656,
      "theoretical_loss": 3.9101547233745313,
      "tokens_seen": 541392896
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008443731193580742,
      "loss": 1.6428,
      "theoretical_loss": 3.9101048081587164,
      "tokens_seen": 541458432
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008443530591775326,
      "loss": 1.6341,
      "theoretical_loss": 3.9100549006754752,
      "tokens_seen": 541523968
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000844332998996991,
      "loss": 1.7341,
      "theoretical_loss": 3.9100050009226752,
      "tokens_seen": 541589504
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008443129388164493,
      "loss": 1.7356,
      "theoretical_loss": 3.909955108898183,
      "tokens_seen": 541655040
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008442928786359078,
      "loss": 1.6214,
      "theoretical_loss": 3.909905224599867,
      "tokens_seen": 541720576
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008442728184553661,
      "loss": 1.6616,
      "theoretical_loss": 3.909855348025596,
      "tokens_seen": 541786112
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008442527582748245,
      "loss": 1.5306,
      "theoretical_loss": 3.9098054791732406,
      "tokens_seen": 541851648
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008442326980942828,
      "loss": 1.7722,
      "theoretical_loss": 3.9097556180406716,
      "tokens_seen": 541917184
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008442126379137413,
      "loss": 1.6713,
      "theoretical_loss": 3.909705764625759,
      "tokens_seen": 541982720
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008441925777331996,
      "loss": 1.6604,
      "theoretical_loss": 3.9096559189263775,
      "tokens_seen": 542048256
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000844172517552658,
      "loss": 1.5467,
      "theoretical_loss": 3.909606080940399,
      "tokens_seen": 542113792
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008441524573721163,
      "loss": 1.6024,
      "theoretical_loss": 3.9095562506656982,
      "tokens_seen": 542179328
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008441323971915747,
      "loss": 1.6659,
      "theoretical_loss": 3.909506428100149,
      "tokens_seen": 542244864
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008441123370110331,
      "loss": 1.6401,
      "theoretical_loss": 3.9094566132416286,
      "tokens_seen": 542310400
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008440922768304915,
      "loss": 1.6249,
      "theoretical_loss": 3.909406806088013,
      "tokens_seen": 542375936
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008440722166499498,
      "loss": 1.5873,
      "theoretical_loss": 3.9093570066371797,
      "tokens_seen": 542441472
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008440521564694083,
      "loss": 1.7236,
      "theoretical_loss": 3.9093072148870074,
      "tokens_seen": 542507008
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008440320962888666,
      "loss": 1.6741,
      "theoretical_loss": 3.9092574308353747,
      "tokens_seen": 542572544
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000844012036108325,
      "loss": 1.657,
      "theoretical_loss": 3.909207654480162,
      "tokens_seen": 542638080
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008439919759277833,
      "loss": 1.6883,
      "theoretical_loss": 3.9091578858192504,
      "tokens_seen": 542703616
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008439719157472417,
      "loss": 1.675,
      "theoretical_loss": 3.909108124850521,
      "tokens_seen": 542769152
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008439518555667001,
      "loss": 1.6478,
      "theoretical_loss": 3.9090583715718568,
      "tokens_seen": 542834688
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008439317953861585,
      "loss": 1.4922,
      "theoretical_loss": 3.9090086259811403,
      "tokens_seen": 542900224
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008439117352056168,
      "loss": 1.8396,
      "theoretical_loss": 3.908958888076257,
      "tokens_seen": 542965760
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008438916750250753,
      "loss": 1.6117,
      "theoretical_loss": 3.908909157855091,
      "tokens_seen": 543031296
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008438716148445335,
      "loss": 1.5594,
      "theoretical_loss": 3.908859435315529,
      "tokens_seen": 543096832
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000843851554663992,
      "loss": 1.5383,
      "theoretical_loss": 3.908809720455457,
      "tokens_seen": 543162368
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008438314944834505,
      "loss": 1.6345,
      "theoretical_loss": 3.908760013272763,
      "tokens_seen": 543227904
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008438114343029087,
      "loss": 1.5347,
      "theoretical_loss": 3.908710313765335,
      "tokens_seen": 543293440
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008437913741223672,
      "loss": 1.677,
      "theoretical_loss": 3.908660621931063,
      "tokens_seen": 543358976
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008437713139418254,
      "loss": 1.659,
      "theoretical_loss": 3.908610937767836,
      "tokens_seen": 543424512
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008437512537612839,
      "loss": 1.6521,
      "theoretical_loss": 3.9085612612735456,
      "tokens_seen": 543490048
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008437311935807423,
      "loss": 1.6628,
      "theoretical_loss": 3.908511592446083,
      "tokens_seen": 543555584
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008437111334002006,
      "loss": 1.6616,
      "theoretical_loss": 3.908461931283342,
      "tokens_seen": 543621120
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000843691073219659,
      "loss": 1.6106,
      "theoretical_loss": 3.9084122777832144,
      "tokens_seen": 543686656
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008436710130391175,
      "loss": 1.6654,
      "theoretical_loss": 3.9083626319435956,
      "tokens_seen": 543752192
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008436509528585757,
      "loss": 1.6849,
      "theoretical_loss": 3.9083129937623804,
      "tokens_seen": 543817728
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008436308926780342,
      "loss": 1.5901,
      "theoretical_loss": 3.908263363237464,
      "tokens_seen": 543883264
    },
    {
      "epoch": 0.16,
      "objective/train/advantage_avg": 0.4907113313674927,
      "objective/train/docs_used": 204061,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5793161392211914,
      "objective/train/original_loss": 1.579316258430481,
      "objective/train/theoretical_loss": 3.908213740366744,
      "objective/train/tokens_used": 564408800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24126523733139038,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050297498703003,
      "objective/train/weighted_lm_loss": 1.6586852073669434,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.908213740366744,
      "tokens_seen": 543948800
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008436108324974924,
      "loss": 1.6774,
      "theoretical_loss": 3.908213740366744,
      "tokens_seen": 543948800
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008435907723169509,
      "loss": 1.6269,
      "theoretical_loss": 3.9081641251481174,
      "tokens_seen": 544014336
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008435707121364093,
      "loss": 1.6298,
      "theoretical_loss": 3.908114517579483,
      "tokens_seen": 544079872
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008435506519558676,
      "loss": 1.6212,
      "theoretical_loss": 3.9080649176587396,
      "tokens_seen": 544145408
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.000843530591775326,
      "loss": 1.6775,
      "theoretical_loss": 3.908015325383788,
      "tokens_seen": 544210944
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008435105315947844,
      "loss": 1.6693,
      "theoretical_loss": 3.9079657407525286,
      "tokens_seen": 544276480
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008434904714142427,
      "loss": 1.6731,
      "theoretical_loss": 3.907916163762863,
      "tokens_seen": 544342016
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008434704112337012,
      "loss": 1.4552,
      "theoretical_loss": 3.9078665944126936,
      "tokens_seen": 544407552
    },
    {
      "epoch": 0.16,
      "learning_rate": 0.0008434503510531595,
      "loss": 1.7013,
      "theoretical_loss": 3.907817032699924,
      "tokens_seen": 544473088
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008434302908726179,
      "loss": 1.6329,
      "theoretical_loss": 3.9077674786224588,
      "tokens_seen": 544538624
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008434102306920762,
      "loss": 1.5692,
      "theoretical_loss": 3.907717932178203,
      "tokens_seen": 544604160
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008433901705115346,
      "loss": 1.5835,
      "theoretical_loss": 3.9076683933650616,
      "tokens_seen": 544669696
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000843370110330993,
      "loss": 1.6368,
      "theoretical_loss": 3.9076188621809416,
      "tokens_seen": 544735232
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008433500501504514,
      "loss": 1.5922,
      "theoretical_loss": 3.907569338623751,
      "tokens_seen": 544800768
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008433299899699097,
      "loss": 1.6241,
      "theoretical_loss": 3.9075198226913974,
      "tokens_seen": 544866304
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008433099297893682,
      "loss": 1.6416,
      "theoretical_loss": 3.907470314381791,
      "tokens_seen": 544931840
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008432898696088265,
      "loss": 1.796,
      "theoretical_loss": 3.9074208136928408,
      "tokens_seen": 544997376
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008432698094282849,
      "loss": 1.6469,
      "theoretical_loss": 3.907371320622458,
      "tokens_seen": 545062912
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008432497492477432,
      "loss": 1.7268,
      "theoretical_loss": 3.9073218351685544,
      "tokens_seen": 545128448
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008432296890672017,
      "loss": 1.6564,
      "theoretical_loss": 3.9072723573290418,
      "tokens_seen": 545193984
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.00084320962888666,
      "loss": 1.5664,
      "theoretical_loss": 3.907222887101834,
      "tokens_seen": 545259520
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008431895687061184,
      "loss": 1.7664,
      "theoretical_loss": 3.907173424484845,
      "tokens_seen": 545325056
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008431695085255767,
      "loss": 1.5977,
      "theoretical_loss": 3.9071239694759896,
      "tokens_seen": 545390592
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008431494483450351,
      "loss": 1.7512,
      "theoretical_loss": 3.9070745220731844,
      "tokens_seen": 545456128
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008431293881644935,
      "loss": 1.6896,
      "theoretical_loss": 3.9070250822743446,
      "tokens_seen": 545521664
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008431093279839519,
      "loss": 1.659,
      "theoretical_loss": 3.9069756500773885,
      "tokens_seen": 545587200
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008430892678034102,
      "loss": 1.7349,
      "theoretical_loss": 3.9069262254802335,
      "tokens_seen": 545652736
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008430692076228687,
      "loss": 1.6861,
      "theoretical_loss": 3.906876808480799,
      "tokens_seen": 545718272
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000843049147442327,
      "loss": 1.6246,
      "theoretical_loss": 3.906827399077006,
      "tokens_seen": 545783808
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008430290872617854,
      "loss": 1.6346,
      "theoretical_loss": 3.906777997266773,
      "tokens_seen": 545849344
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008430090270812437,
      "loss": 1.6471,
      "theoretical_loss": 3.9067286030480237,
      "tokens_seen": 545914880
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008429889669007021,
      "loss": 1.689,
      "theoretical_loss": 3.9066792164186785,
      "tokens_seen": 545980416
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008429689067201605,
      "loss": 1.6054,
      "theoretical_loss": 3.9066298373766615,
      "tokens_seen": 546045952
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008429488465396189,
      "loss": 1.6965,
      "theoretical_loss": 3.9065804659198964,
      "tokens_seen": 546111488
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008429287863590772,
      "loss": 1.6824,
      "theoretical_loss": 3.9065311020463085,
      "tokens_seen": 546177024
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008429087261785357,
      "loss": 1.6068,
      "theoretical_loss": 3.9064817457538226,
      "tokens_seen": 546242560
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008428886659979939,
      "loss": 1.6234,
      "theoretical_loss": 3.9064323970403656,
      "tokens_seen": 546308096
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008428686058174524,
      "loss": 1.7045,
      "theoretical_loss": 3.9063830559038637,
      "tokens_seen": 546373632
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008428485456369108,
      "loss": 1.6106,
      "theoretical_loss": 3.9063337223422465,
      "tokens_seen": 546439168
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008428284854563691,
      "loss": 1.6302,
      "theoretical_loss": 3.906284396353441,
      "tokens_seen": 546504704
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008428084252758275,
      "loss": 1.6731,
      "theoretical_loss": 3.9062350779353787,
      "tokens_seen": 546570240
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008427883650952858,
      "loss": 1.6182,
      "theoretical_loss": 3.906185767085989,
      "tokens_seen": 546635776
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008427683049147442,
      "loss": 1.6749,
      "theoretical_loss": 3.906136463803204,
      "tokens_seen": 546701312
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008427482447342027,
      "loss": 1.5789,
      "theoretical_loss": 3.9060871680849543,
      "tokens_seen": 546766848
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008427281845536609,
      "loss": 1.6737,
      "theoretical_loss": 3.906037879929174,
      "tokens_seen": 546832384
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008427081243731194,
      "loss": 1.6823,
      "theoretical_loss": 3.9059885993337966,
      "tokens_seen": 546897920
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008426880641925778,
      "loss": 1.6538,
      "theoretical_loss": 3.905939326296756,
      "tokens_seen": 546963456
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008426680040120361,
      "loss": 1.4244,
      "theoretical_loss": 3.9058900608159886,
      "tokens_seen": 547028992
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008426479438314945,
      "loss": 1.7044,
      "theoretical_loss": 3.90584080288943,
      "tokens_seen": 547094528
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008426278836509528,
      "loss": 1.6694,
      "theoretical_loss": 3.9057915525150166,
      "tokens_seen": 547160064
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.49299463629722595,
      "objective/train/docs_used": 204823,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4490361213684082,
      "objective/train/original_loss": 1.4490363597869873,
      "objective/train/theoretical_loss": 3.9057423096906874,
      "objective/train/tokens_used": 567685600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.244096577167511,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505404472351074,
      "objective/train/weighted_lm_loss": 1.5221048593521118,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9057423096906874,
      "tokens_seen": 547225600
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008426078234704112,
      "loss": 1.5892,
      "theoretical_loss": 3.9057423096906874,
      "tokens_seen": 547225600
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008425877632898697,
      "loss": 1.61,
      "theoretical_loss": 3.90569307441438,
      "tokens_seen": 547291136
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008425677031093279,
      "loss": 1.7166,
      "theoretical_loss": 3.905643846684034,
      "tokens_seen": 547356672
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008425476429287864,
      "loss": 1.6075,
      "theoretical_loss": 3.9055946264975896,
      "tokens_seen": 547422208
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008425275827482447,
      "loss": 1.5245,
      "theoretical_loss": 3.9055454138529884,
      "tokens_seen": 547487744
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008425075225677031,
      "loss": 1.6103,
      "theoretical_loss": 3.905496208748171,
      "tokens_seen": 547553280
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008424874623871615,
      "loss": 1.5812,
      "theoretical_loss": 3.9054470111810815,
      "tokens_seen": 547618816
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008424674022066199,
      "loss": 1.5837,
      "theoretical_loss": 3.905397821149662,
      "tokens_seen": 547684352
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008424473420260782,
      "loss": 1.6127,
      "theoretical_loss": 3.9053486386518577,
      "tokens_seen": 547749888
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008424272818455366,
      "loss": 1.6257,
      "theoretical_loss": 3.905299463685613,
      "tokens_seen": 547815424
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008424072216649949,
      "loss": 1.6105,
      "theoretical_loss": 3.9052502962488735,
      "tokens_seen": 547880960
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008423871614844534,
      "loss": 1.6986,
      "theoretical_loss": 3.9052011363395867,
      "tokens_seen": 547946496
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008423671013039117,
      "loss": 1.5628,
      "theoretical_loss": 3.9051519839557,
      "tokens_seen": 548012032
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008423470411233701,
      "loss": 1.6445,
      "theoretical_loss": 3.905102839095161,
      "tokens_seen": 548077568
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008423269809428286,
      "loss": 1.6641,
      "theoretical_loss": 3.9050537017559197,
      "tokens_seen": 548143104
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008423069207622869,
      "loss": 1.6682,
      "theoretical_loss": 3.9050045719359248,
      "tokens_seen": 548208640
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008422868605817453,
      "loss": 1.7039,
      "theoretical_loss": 3.904955449633128,
      "tokens_seen": 548274176
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008422668004012036,
      "loss": 1.6177,
      "theoretical_loss": 3.9049063348454798,
      "tokens_seen": 548339712
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008422467402206621,
      "loss": 1.5919,
      "theoretical_loss": 3.904857227570934,
      "tokens_seen": 548405248
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008422266800401204,
      "loss": 1.6974,
      "theoretical_loss": 3.904808127807442,
      "tokens_seen": 548470784
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008422066198595788,
      "loss": 1.5215,
      "theoretical_loss": 3.9047590355529587,
      "tokens_seen": 548536320
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008421865596790371,
      "loss": 1.5496,
      "theoretical_loss": 3.9047099508054384,
      "tokens_seen": 548601856
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008421664994984955,
      "loss": 1.4681,
      "theoretical_loss": 3.904660873562837,
      "tokens_seen": 548667392
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008421464393179539,
      "loss": 1.4619,
      "theoretical_loss": 3.90461180382311,
      "tokens_seen": 548732928
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008421263791374123,
      "loss": 1.6295,
      "theoretical_loss": 3.9045627415842157,
      "tokens_seen": 548798464
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008421063189568706,
      "loss": 1.7297,
      "theoretical_loss": 3.9045136868441106,
      "tokens_seen": 548864000
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008420862587763291,
      "loss": 1.6703,
      "theoretical_loss": 3.9044646396007545,
      "tokens_seen": 548929536
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008420661985957874,
      "loss": 1.6258,
      "theoretical_loss": 3.9044155998521064,
      "tokens_seen": 548995072
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008420461384152458,
      "loss": 1.6198,
      "theoretical_loss": 3.904366567596127,
      "tokens_seen": 549060608
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008420260782347041,
      "loss": 1.6694,
      "theoretical_loss": 3.9043175428307766,
      "tokens_seen": 549126144
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008420060180541625,
      "loss": 1.6021,
      "theoretical_loss": 3.9042685255540177,
      "tokens_seen": 549191680
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008419859578736209,
      "loss": 1.655,
      "theoretical_loss": 3.9042195157638133,
      "tokens_seen": 549257216
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008419658976930793,
      "loss": 1.6636,
      "theoretical_loss": 3.904170513458126,
      "tokens_seen": 549322752
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008419458375125376,
      "loss": 1.6395,
      "theoretical_loss": 3.9041215186349203,
      "tokens_seen": 549388288
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008419257773319961,
      "loss": 1.6057,
      "theoretical_loss": 3.9040725312921616,
      "tokens_seen": 549453824
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008419057171514543,
      "loss": 1.6732,
      "theoretical_loss": 3.9040235514278163,
      "tokens_seen": 549519360
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008418856569709128,
      "loss": 1.7559,
      "theoretical_loss": 3.9039745790398497,
      "tokens_seen": 549584896
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008418655967903712,
      "loss": 1.4984,
      "theoretical_loss": 3.903925614126231,
      "tokens_seen": 549650432
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008418455366098295,
      "loss": 1.6358,
      "theoretical_loss": 3.9038766566849263,
      "tokens_seen": 549715968
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008418254764292879,
      "loss": 1.6838,
      "theoretical_loss": 3.903827706713906,
      "tokens_seen": 549781504
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008418054162487462,
      "loss": 1.6195,
      "theoretical_loss": 3.9037787642111406,
      "tokens_seen": 549847040
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008417853560682046,
      "loss": 1.5633,
      "theoretical_loss": 3.903729829174599,
      "tokens_seen": 549912576
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008417652958876631,
      "loss": 1.6828,
      "theoretical_loss": 3.903680901602254,
      "tokens_seen": 549978112
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008417452357071213,
      "loss": 1.5739,
      "theoretical_loss": 3.903631981492077,
      "tokens_seen": 550043648
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008417251755265798,
      "loss": 1.5889,
      "theoretical_loss": 3.9035830688420416,
      "tokens_seen": 550109184
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008417051153460382,
      "loss": 1.5274,
      "theoretical_loss": 3.9035341636501215,
      "tokens_seen": 550174720
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008416850551654965,
      "loss": 1.7497,
      "theoretical_loss": 3.9034852659142913,
      "tokens_seen": 550240256
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008416649949849549,
      "loss": 1.6579,
      "theoretical_loss": 3.903436375632526,
      "tokens_seen": 550305792
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008416449348044132,
      "loss": 1.4647,
      "theoretical_loss": 3.9033874928028025,
      "tokens_seen": 550371328
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008416248746238716,
      "loss": 1.6656,
      "theoretical_loss": 3.9033386174230973,
      "tokens_seen": 550436864
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.49245110154151917,
      "objective/train/docs_used": 206347,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6513526439666748,
      "objective/train/original_loss": 1.6513524055480957,
      "objective/train/theoretical_loss": 3.9032897494913876,
      "objective/train/tokens_used": 570962400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24299608170986176,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504804849624634,
      "objective/train/weighted_lm_loss": 1.73475980758667,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9032897494913876,
      "tokens_seen": 550502400
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008416048144433301,
      "loss": 1.6543,
      "theoretical_loss": 3.9032897494913876,
      "tokens_seen": 550502400
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008415847542627883,
      "loss": 1.6547,
      "theoretical_loss": 3.903240889005653,
      "tokens_seen": 550567936
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008415646940822468,
      "loss": 1.6404,
      "theoretical_loss": 3.9031920359638725,
      "tokens_seen": 550633472
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008415446339017051,
      "loss": 1.5895,
      "theoretical_loss": 3.903143190364026,
      "tokens_seen": 550699008
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008415245737211635,
      "loss": 1.5233,
      "theoretical_loss": 3.9030943522040946,
      "tokens_seen": 550764544
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008415045135406219,
      "loss": 1.6824,
      "theoretical_loss": 3.90304552148206,
      "tokens_seen": 550830080
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008414844533600803,
      "loss": 1.6634,
      "theoretical_loss": 3.9029966981959046,
      "tokens_seen": 550895616
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008414643931795386,
      "loss": 1.6046,
      "theoretical_loss": 3.9029478823436117,
      "tokens_seen": 550961152
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000841444332998997,
      "loss": 1.781,
      "theoretical_loss": 3.902899073923166,
      "tokens_seen": 551026688
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008414242728184553,
      "loss": 1.6708,
      "theoretical_loss": 3.902850272932551,
      "tokens_seen": 551092224
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008414042126379138,
      "loss": 1.6083,
      "theoretical_loss": 3.902801479369754,
      "tokens_seen": 551157760
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008413841524573721,
      "loss": 1.6505,
      "theoretical_loss": 3.9027526932327596,
      "tokens_seen": 551223296
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008413640922768305,
      "loss": 1.6389,
      "theoretical_loss": 3.902703914519557,
      "tokens_seen": 551288832
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008413440320962889,
      "loss": 1.6305,
      "theoretical_loss": 3.9026551432281322,
      "tokens_seen": 551354368
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008413239719157473,
      "loss": 1.6152,
      "theoretical_loss": 3.9026063793564756,
      "tokens_seen": 551419904
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008413039117352056,
      "loss": 1.5958,
      "theoretical_loss": 3.9025576229025756,
      "tokens_seen": 551485440
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000841283851554664,
      "loss": 1.7422,
      "theoretical_loss": 3.9025088738644236,
      "tokens_seen": 551550976
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008412637913741223,
      "loss": 1.5482,
      "theoretical_loss": 3.90246013224001,
      "tokens_seen": 551616512
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008412437311935808,
      "loss": 1.5257,
      "theoretical_loss": 3.902411398027327,
      "tokens_seen": 551682048
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008412236710130391,
      "loss": 1.6078,
      "theoretical_loss": 3.902362671224367,
      "tokens_seen": 551747584
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008412036108324975,
      "loss": 1.6325,
      "theoretical_loss": 3.9023139518291243,
      "tokens_seen": 551813120
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008411835506519558,
      "loss": 1.6692,
      "theoretical_loss": 3.902265239839592,
      "tokens_seen": 551878656
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008411634904714143,
      "loss": 1.5291,
      "theoretical_loss": 3.902216535253766,
      "tokens_seen": 551944192
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008411434302908726,
      "loss": 1.5662,
      "theoretical_loss": 3.9021678380696416,
      "tokens_seen": 552009728
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000841123370110331,
      "loss": 1.6418,
      "theoretical_loss": 3.902119148285216,
      "tokens_seen": 552075264
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008411033099297894,
      "loss": 1.5989,
      "theoretical_loss": 3.9020704658984857,
      "tokens_seen": 552140800
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008410832497492478,
      "loss": 1.5853,
      "theoretical_loss": 3.90202179090745,
      "tokens_seen": 552206336
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008410631895687061,
      "loss": 1.7523,
      "theoretical_loss": 3.901973123310107,
      "tokens_seen": 552271872
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008410431293881645,
      "loss": 1.6091,
      "theoretical_loss": 3.9019244631044563,
      "tokens_seen": 552337408
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008410230692076228,
      "loss": 1.4845,
      "theoretical_loss": 3.901875810288499,
      "tokens_seen": 552402944
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008410030090270813,
      "loss": 1.6014,
      "theoretical_loss": 3.9018271648602365,
      "tokens_seen": 552468480
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008409829488465396,
      "loss": 1.7345,
      "theoretical_loss": 3.9017785268176697,
      "tokens_seen": 552534016
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000840962888665998,
      "loss": 1.519,
      "theoretical_loss": 3.9017298961588027,
      "tokens_seen": 552599552
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008409428284854564,
      "loss": 1.7287,
      "theoretical_loss": 3.9016812728816386,
      "tokens_seen": 552665088
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008409227683049147,
      "loss": 1.656,
      "theoretical_loss": 3.901632656984182,
      "tokens_seen": 552730624
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008409027081243731,
      "loss": 1.568,
      "theoretical_loss": 3.901584048464438,
      "tokens_seen": 552796160
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008408826479438316,
      "loss": 1.5216,
      "theoretical_loss": 3.901535447320412,
      "tokens_seen": 552861696
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008408625877632898,
      "loss": 1.4595,
      "theoretical_loss": 3.9014868535501113,
      "tokens_seen": 552927232
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008408425275827483,
      "loss": 1.6847,
      "theoretical_loss": 3.9014382671515433,
      "tokens_seen": 552992768
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008408224674022066,
      "loss": 1.6083,
      "theoretical_loss": 3.901389688122716,
      "tokens_seen": 553058304
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000840802407221665,
      "loss": 1.6974,
      "theoretical_loss": 3.901341116461639,
      "tokens_seen": 553123840
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008407823470411235,
      "loss": 1.5456,
      "theoretical_loss": 3.9012925521663213,
      "tokens_seen": 553189376
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008407622868605817,
      "loss": 1.6157,
      "theoretical_loss": 3.901243995234774,
      "tokens_seen": 553254912
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008407422266800402,
      "loss": 1.6511,
      "theoretical_loss": 3.901195445665009,
      "tokens_seen": 553320448
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008407221664994986,
      "loss": 1.5942,
      "theoretical_loss": 3.9011469034550372,
      "tokens_seen": 553385984
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008407021063189569,
      "loss": 1.6496,
      "theoretical_loss": 3.901098368602872,
      "tokens_seen": 553451520
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008406820461384153,
      "loss": 1.5681,
      "theoretical_loss": 3.901049841106528,
      "tokens_seen": 553517056
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008406619859578736,
      "loss": 1.7091,
      "theoretical_loss": 3.901001320964018,
      "tokens_seen": 553582592
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000840641925777332,
      "loss": 1.5802,
      "theoretical_loss": 3.900952808173358,
      "tokens_seen": 553648128
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008406218655967905,
      "loss": 1.7343,
      "theoretical_loss": 3.900904302732564,
      "tokens_seen": 553713664
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.49144139885902405,
      "objective/train/docs_used": 207589,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7685081958770752,
      "objective/train/original_loss": 1.768507957458496,
      "objective/train/theoretical_loss": 3.9008558046396535,
      "objective/train/tokens_used": 574239200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24197053909301758,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503742694854736,
      "objective/train/weighted_lm_loss": 1.8574825525283813,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.9008558046396535,
      "tokens_seen": 553779200
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008406018054162487,
      "loss": 1.6778,
      "theoretical_loss": 3.9008558046396535,
      "tokens_seen": 553779200
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008405817452357072,
      "loss": 1.571,
      "theoretical_loss": 3.900807313892643,
      "tokens_seen": 553844736
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008405616850551655,
      "loss": 1.6578,
      "theoretical_loss": 3.900758830489551,
      "tokens_seen": 553910272
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008405416248746239,
      "loss": 1.7218,
      "theoretical_loss": 3.900710354428397,
      "tokens_seen": 553975808
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008405215646940823,
      "loss": 1.6324,
      "theoretical_loss": 3.9006618857072,
      "tokens_seen": 554041344
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008405015045135407,
      "loss": 1.5992,
      "theoretical_loss": 3.900613424323981,
      "tokens_seen": 554106880
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000840481444332999,
      "loss": 1.6166,
      "theoretical_loss": 3.900564970276762,
      "tokens_seen": 554172416
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008404613841524574,
      "loss": 1.5832,
      "theoretical_loss": 3.9005165235635646,
      "tokens_seen": 554237952
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008404413239719157,
      "loss": 1.5677,
      "theoretical_loss": 3.9004680841824113,
      "tokens_seen": 554303488
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008404212637913742,
      "loss": 1.7651,
      "theoretical_loss": 3.900419652131327,
      "tokens_seen": 554369024
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008404012036108325,
      "loss": 1.5291,
      "theoretical_loss": 3.9003712274083346,
      "tokens_seen": 554434560
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008403811434302909,
      "loss": 1.7042,
      "theoretical_loss": 3.9003228100114606,
      "tokens_seen": 554500096
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008403610832497493,
      "loss": 1.6628,
      "theoretical_loss": 3.9002743999387297,
      "tokens_seen": 554565632
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008403410230692077,
      "loss": 1.572,
      "theoretical_loss": 3.9002259971881696,
      "tokens_seen": 554631168
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000840320962888666,
      "loss": 1.5983,
      "theoretical_loss": 3.9001776017578074,
      "tokens_seen": 554696704
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008403009027081244,
      "loss": 1.6355,
      "theoretical_loss": 3.900129213645672,
      "tokens_seen": 554762240
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008402808425275827,
      "loss": 1.5939,
      "theoretical_loss": 3.9000808328497913,
      "tokens_seen": 554827776
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008402607823470412,
      "loss": 1.5598,
      "theoretical_loss": 3.9000324593681963,
      "tokens_seen": 554893312
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008402407221664995,
      "loss": 1.6882,
      "theoretical_loss": 3.899984093198916,
      "tokens_seen": 554958848
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008402206619859579,
      "loss": 1.5463,
      "theoretical_loss": 3.8999357343399836,
      "tokens_seen": 555024384
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008402006018054162,
      "loss": 1.6846,
      "theoretical_loss": 3.89988738278943,
      "tokens_seen": 555089920
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008401805416248747,
      "loss": 1.5995,
      "theoretical_loss": 3.8998390385452883,
      "tokens_seen": 555155456
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000840160481444333,
      "loss": 1.5115,
      "theoretical_loss": 3.899790701605592,
      "tokens_seen": 555220992
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008401404212637914,
      "loss": 1.5238,
      "theoretical_loss": 3.899742371968376,
      "tokens_seen": 555286528
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008401203610832498,
      "loss": 1.5713,
      "theoretical_loss": 3.899694049631674,
      "tokens_seen": 555352064
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008401003009027082,
      "loss": 1.6603,
      "theoretical_loss": 3.899645734593524,
      "tokens_seen": 555417600
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008400802407221665,
      "loss": 1.6839,
      "theoretical_loss": 3.899597426851961,
      "tokens_seen": 555483136
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000840060180541625,
      "loss": 1.5337,
      "theoretical_loss": 3.899549126405023,
      "tokens_seen": 555548672
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008400401203610832,
      "loss": 1.4908,
      "theoretical_loss": 3.8995008332507477,
      "tokens_seen": 555614208
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008400200601805417,
      "loss": 1.5943,
      "theoretical_loss": 3.899452547387175,
      "tokens_seen": 555679744
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.00084,
      "loss": 1.5668,
      "theoretical_loss": 3.899404268812343,
      "tokens_seen": 555745280
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008399799398194584,
      "loss": 1.5763,
      "theoretical_loss": 3.899355997524294,
      "tokens_seen": 555810816
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008399598796389168,
      "loss": 1.7422,
      "theoretical_loss": 3.899307733521068,
      "tokens_seen": 555876352
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008399398194583751,
      "loss": 1.6995,
      "theoretical_loss": 3.899259476800707,
      "tokens_seen": 555941888
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008399197592778335,
      "loss": 1.3834,
      "theoretical_loss": 3.8992112273612545,
      "tokens_seen": 556007424
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000839899699097292,
      "loss": 1.6003,
      "theoretical_loss": 3.8991629852007534,
      "tokens_seen": 556072960
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008398796389167502,
      "loss": 1.7357,
      "theoretical_loss": 3.899114750317248,
      "tokens_seen": 556138496
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008398595787362087,
      "loss": 1.7076,
      "theoretical_loss": 3.899066522708783,
      "tokens_seen": 556204032
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008398395185556669,
      "loss": 1.4848,
      "theoretical_loss": 3.8990183023734044,
      "tokens_seen": 556269568
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008398194583751254,
      "loss": 1.6815,
      "theoretical_loss": 3.8989700893091586,
      "tokens_seen": 556335104
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008397993981945838,
      "loss": 1.6465,
      "theoretical_loss": 3.898921883514093,
      "tokens_seen": 556400640
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008397793380140421,
      "loss": 1.5887,
      "theoretical_loss": 3.898873684986256,
      "tokens_seen": 556466176
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008397592778335005,
      "loss": 1.5622,
      "theoretical_loss": 3.8988254937236952,
      "tokens_seen": 556531712
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000839739217652959,
      "loss": 1.5486,
      "theoretical_loss": 3.898777309724461,
      "tokens_seen": 556597248
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008397191574724172,
      "loss": 1.5783,
      "theoretical_loss": 3.8987291329866034,
      "tokens_seen": 556662784
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008396990972918757,
      "loss": 1.735,
      "theoretical_loss": 3.898680963508174,
      "tokens_seen": 556728320
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008396790371113339,
      "loss": 1.6335,
      "theoretical_loss": 3.8986328012872233,
      "tokens_seen": 556793856
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008396589769307924,
      "loss": 1.6146,
      "theoretical_loss": 3.898584646321805,
      "tokens_seen": 556859392
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008396389167502508,
      "loss": 1.5333,
      "theoretical_loss": 3.8985364986099715,
      "tokens_seen": 556924928
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008396188565697091,
      "loss": 1.5544,
      "theoretical_loss": 3.898488358149778,
      "tokens_seen": 556990464
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.4942537546157837,
      "objective/train/docs_used": 208740,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5796339511871338,
      "objective/train/original_loss": 1.5796339511871338,
      "objective/train/theoretical_loss": 3.8984402249392778,
      "objective/train/tokens_used": 577516000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24452345073223114,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050668478012085,
      "objective/train/weighted_lm_loss": 1.6596615314483643,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8984402249392778,
      "tokens_seen": 557056000
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008395987963891675,
      "loss": 1.5798,
      "theoretical_loss": 3.8984402249392778,
      "tokens_seen": 557056000
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008395787362086259,
      "loss": 1.5101,
      "theoretical_loss": 3.8983920989765273,
      "tokens_seen": 557121536
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008395586760280842,
      "loss": 1.6682,
      "theoretical_loss": 3.8983439802595825,
      "tokens_seen": 557187072
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008395386158475427,
      "loss": 1.5552,
      "theoretical_loss": 3.8982958687865006,
      "tokens_seen": 557252608
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008395185556670009,
      "loss": 1.4999,
      "theoretical_loss": 3.8982477645553395,
      "tokens_seen": 557318144
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008394984954864594,
      "loss": 1.5518,
      "theoretical_loss": 3.8981996675641573,
      "tokens_seen": 557383680
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008394784353059177,
      "loss": 1.6179,
      "theoretical_loss": 3.8981515778110136,
      "tokens_seen": 557449216
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008394583751253761,
      "loss": 1.5914,
      "theoretical_loss": 3.8981034952939684,
      "tokens_seen": 557514752
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008394383149448345,
      "loss": 1.5776,
      "theoretical_loss": 3.898055420011082,
      "tokens_seen": 557580288
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008394182547642929,
      "loss": 1.5414,
      "theoretical_loss": 3.898007351960416,
      "tokens_seen": 557645824
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008393981945837512,
      "loss": 1.5064,
      "theoretical_loss": 3.897959291140034,
      "tokens_seen": 557711360
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008393781344032097,
      "loss": 1.5765,
      "theoretical_loss": 3.8979112375479974,
      "tokens_seen": 557776896
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000839358074222668,
      "loss": 1.5801,
      "theoretical_loss": 3.8978631911823705,
      "tokens_seen": 557842432
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008393380140421264,
      "loss": 1.5644,
      "theoretical_loss": 3.8978151520412174,
      "tokens_seen": 557907968
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008393179538615848,
      "loss": 1.5375,
      "theoretical_loss": 3.8977671201226043,
      "tokens_seen": 557973504
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008392978936810431,
      "loss": 1.566,
      "theoretical_loss": 3.8977190954245966,
      "tokens_seen": 558039040
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008392778335005016,
      "loss": 1.6901,
      "theoretical_loss": 3.8976710779452612,
      "tokens_seen": 558104576
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008392577733199599,
      "loss": 1.5803,
      "theoretical_loss": 3.897623067682665,
      "tokens_seen": 558170112
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008392377131394183,
      "loss": 1.6792,
      "theoretical_loss": 3.8975750646348772,
      "tokens_seen": 558235648
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008392176529588766,
      "loss": 1.6074,
      "theoretical_loss": 3.897527068799966,
      "tokens_seen": 558301184
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008391975927783351,
      "loss": 1.6105,
      "theoretical_loss": 3.8974790801760015,
      "tokens_seen": 558366720
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008391775325977934,
      "loss": 1.7481,
      "theoretical_loss": 3.8974310987610536,
      "tokens_seen": 558432256
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008391574724172518,
      "loss": 1.6748,
      "theoretical_loss": 3.8973831245531945,
      "tokens_seen": 558497792
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008391374122367102,
      "loss": 1.6032,
      "theoretical_loss": 3.8973351575504953,
      "tokens_seen": 558563328
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008391173520561686,
      "loss": 1.6181,
      "theoretical_loss": 3.897287197751029,
      "tokens_seen": 558628864
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008390972918756269,
      "loss": 1.6316,
      "theoretical_loss": 3.8972392451528686,
      "tokens_seen": 558694400
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008390772316950853,
      "loss": 1.6653,
      "theoretical_loss": 3.8971912997540885,
      "tokens_seen": 558759936
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008390571715145436,
      "loss": 1.6679,
      "theoretical_loss": 3.8971433615527644,
      "tokens_seen": 558825472
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008390371113340021,
      "loss": 1.629,
      "theoretical_loss": 3.897095430546971,
      "tokens_seen": 558891008
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008390170511534604,
      "loss": 1.6606,
      "theoretical_loss": 3.8970475067347854,
      "tokens_seen": 558956544
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008389969909729188,
      "loss": 1.6832,
      "theoretical_loss": 3.8969995901142838,
      "tokens_seen": 559022080
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008389769307923772,
      "loss": 1.5703,
      "theoretical_loss": 3.8969516806835443,
      "tokens_seen": 559087616
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008389568706118355,
      "loss": 1.6939,
      "theoretical_loss": 3.896903778440646,
      "tokens_seen": 559153152
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008389368104312939,
      "loss": 1.6771,
      "theoretical_loss": 3.8968558833836684,
      "tokens_seen": 559218688
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008389167502507524,
      "loss": 1.6217,
      "theoretical_loss": 3.8968079955106907,
      "tokens_seen": 559284224
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008388966900702106,
      "loss": 1.5746,
      "theoretical_loss": 3.8967601148197946,
      "tokens_seen": 559349760
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008388766298896691,
      "loss": 1.6534,
      "theoretical_loss": 3.896712241309061,
      "tokens_seen": 559415296
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008388565697091273,
      "loss": 1.6365,
      "theoretical_loss": 3.8966643749765724,
      "tokens_seen": 559480832
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008388365095285858,
      "loss": 1.6652,
      "theoretical_loss": 3.896616515820412,
      "tokens_seen": 559546368
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008388164493480442,
      "loss": 1.6126,
      "theoretical_loss": 3.8965686638386634,
      "tokens_seen": 559611904
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008387963891675025,
      "loss": 1.6858,
      "theoretical_loss": 3.896520819029411,
      "tokens_seen": 559677440
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008387763289869609,
      "loss": 1.6394,
      "theoretical_loss": 3.896472981390741,
      "tokens_seen": 559742976
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008387562688064194,
      "loss": 1.6711,
      "theoretical_loss": 3.896425150920738,
      "tokens_seen": 559808512
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008387362086258776,
      "loss": 1.6029,
      "theoretical_loss": 3.896377327617489,
      "tokens_seen": 559874048
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008387161484453361,
      "loss": 1.597,
      "theoretical_loss": 3.896329511479082,
      "tokens_seen": 559939584
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008386960882647943,
      "loss": 1.5347,
      "theoretical_loss": 3.8962817025036047,
      "tokens_seen": 560005120
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008386760280842528,
      "loss": 1.5564,
      "theoretical_loss": 3.8962339006891464,
      "tokens_seen": 560070656
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008386559679037112,
      "loss": 1.5286,
      "theoretical_loss": 3.896186106033796,
      "tokens_seen": 560136192
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008386359077231695,
      "loss": 1.6609,
      "theoretical_loss": 3.8961383185356455,
      "tokens_seen": 560201728
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008386158475426279,
      "loss": 1.5907,
      "theoretical_loss": 3.8960905381927837,
      "tokens_seen": 560267264
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.49280041456222534,
      "objective/train/docs_used": 209977,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4882906675338745,
      "objective/train/original_loss": 1.4882906675338745,
      "objective/train/theoretical_loss": 3.8960427650033047,
      "objective/train/tokens_used": 580792800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24333533644676208,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505170822143555,
      "objective/train/weighted_lm_loss": 1.5633784532546997,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8960427650033047,
      "tokens_seen": 560332800
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008385957873620863,
      "loss": 1.5401,
      "theoretical_loss": 3.8960427650033047,
      "tokens_seen": 560332800
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008385757271815446,
      "loss": 1.5833,
      "theoretical_loss": 3.8959949989652998,
      "tokens_seen": 560398336
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008385556670010031,
      "loss": 1.5688,
      "theoretical_loss": 3.895947240076862,
      "tokens_seen": 560463872
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008385356068204613,
      "loss": 1.5936,
      "theoretical_loss": 3.8958994883360862,
      "tokens_seen": 560529408
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008385155466399198,
      "loss": 1.4294,
      "theoretical_loss": 3.8958517437410674,
      "tokens_seen": 560594944
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008384954864593781,
      "loss": 1.554,
      "theoretical_loss": 3.8958040062899,
      "tokens_seen": 560660480
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008384754262788365,
      "loss": 1.6376,
      "theoretical_loss": 3.895756275980681,
      "tokens_seen": 560726016
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008384553660982949,
      "loss": 1.55,
      "theoretical_loss": 3.895708552811507,
      "tokens_seen": 560791552
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008384353059177533,
      "loss": 1.6216,
      "theoretical_loss": 3.895660836780476,
      "tokens_seen": 560857088
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008384152457372116,
      "loss": 1.6825,
      "theoretical_loss": 3.895613127885686,
      "tokens_seen": 560922624
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008383951855566701,
      "loss": 1.6211,
      "theoretical_loss": 3.895565426125237,
      "tokens_seen": 560988160
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008383751253761284,
      "loss": 1.7213,
      "theoretical_loss": 3.8955177314972276,
      "tokens_seen": 561053696
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008383550651955868,
      "loss": 1.5804,
      "theoretical_loss": 3.895470043999759,
      "tokens_seen": 561119232
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008383350050150451,
      "loss": 1.5192,
      "theoretical_loss": 3.8954223636309333,
      "tokens_seen": 561184768
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008383149448345035,
      "loss": 1.6263,
      "theoretical_loss": 3.8953746903888513,
      "tokens_seen": 561250304
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008382948846539619,
      "loss": 1.5549,
      "theoretical_loss": 3.8953270242716167,
      "tokens_seen": 561315840
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008382748244734203,
      "loss": 1.4904,
      "theoretical_loss": 3.8952793652773323,
      "tokens_seen": 561381376
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008382547642928786,
      "loss": 1.6546,
      "theoretical_loss": 3.895231713404103,
      "tokens_seen": 561446912
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000838234704112337,
      "loss": 1.5228,
      "theoretical_loss": 3.895184068650033,
      "tokens_seen": 561512448
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008382146439317954,
      "loss": 1.6943,
      "theoretical_loss": 3.895136431013229,
      "tokens_seen": 561577984
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008381945837512538,
      "loss": 1.5857,
      "theoretical_loss": 3.8950888004917967,
      "tokens_seen": 561643520
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008381745235707121,
      "loss": 1.6202,
      "theoretical_loss": 3.8950411770838427,
      "tokens_seen": 561709056
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008381544633901706,
      "loss": 1.5768,
      "theoretical_loss": 3.8949935607874764,
      "tokens_seen": 561774592
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008381344032096288,
      "loss": 1.6026,
      "theoretical_loss": 3.894945951600805,
      "tokens_seen": 561840128
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008381143430290873,
      "loss": 1.488,
      "theoretical_loss": 3.894898349521939,
      "tokens_seen": 561905664
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008380942828485456,
      "loss": 1.6545,
      "theoretical_loss": 3.8948507545489877,
      "tokens_seen": 561971200
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000838074222668004,
      "loss": 1.6421,
      "theoretical_loss": 3.8948031666800613,
      "tokens_seen": 562036736
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008380541624874624,
      "loss": 1.5853,
      "theoretical_loss": 3.894755585913272,
      "tokens_seen": 562102272
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008380341023069208,
      "loss": 1.6885,
      "theoretical_loss": 3.894708012246732,
      "tokens_seen": 562167808
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008380140421263791,
      "loss": 1.6247,
      "theoretical_loss": 3.8946604456785545,
      "tokens_seen": 562233344
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008379939819458376,
      "loss": 1.756,
      "theoretical_loss": 3.8946128862068528,
      "tokens_seen": 562298880
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008379739217652958,
      "loss": 1.6148,
      "theoretical_loss": 3.894565333829741,
      "tokens_seen": 562364416
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008379538615847543,
      "loss": 1.537,
      "theoretical_loss": 3.8945177885453344,
      "tokens_seen": 562429952
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008379338014042126,
      "loss": 1.5882,
      "theoretical_loss": 3.8944702503517483,
      "tokens_seen": 562495488
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837913741223671,
      "loss": 1.6343,
      "theoretical_loss": 3.8944227192471006,
      "tokens_seen": 562561024
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008378936810431294,
      "loss": 1.5708,
      "theoretical_loss": 3.894375195229507,
      "tokens_seen": 562626560
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008378736208625877,
      "loss": 1.6487,
      "theoretical_loss": 3.8943276782970866,
      "tokens_seen": 562692096
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008378535606820461,
      "loss": 1.6422,
      "theoretical_loss": 3.8942801684479575,
      "tokens_seen": 562757632
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008378335005015046,
      "loss": 1.6344,
      "theoretical_loss": 3.8942326656802395,
      "tokens_seen": 562823168
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008378134403209629,
      "loss": 1.6309,
      "theoretical_loss": 3.8941851699920518,
      "tokens_seen": 562888704
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008377933801404213,
      "loss": 1.7459,
      "theoretical_loss": 3.8941376813815167,
      "tokens_seen": 562954240
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008377733199598798,
      "loss": 1.6037,
      "theoretical_loss": 3.8940901998467545,
      "tokens_seen": 563019776
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837753259779338,
      "loss": 1.5997,
      "theoretical_loss": 3.894042725385888,
      "tokens_seen": 563085312
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008377331995987965,
      "loss": 1.7549,
      "theoretical_loss": 3.89399525799704,
      "tokens_seen": 563150848
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008377131394182547,
      "loss": 1.6288,
      "theoretical_loss": 3.8939477976783348,
      "tokens_seen": 563216384
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008376930792377132,
      "loss": 1.496,
      "theoretical_loss": 3.893900344427896,
      "tokens_seen": 563281920
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008376730190571716,
      "loss": 1.7518,
      "theoretical_loss": 3.893852898243849,
      "tokens_seen": 563347456
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008376529588766299,
      "loss": 1.6264,
      "theoretical_loss": 3.8938054591243194,
      "tokens_seen": 563412992
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008376328986960883,
      "loss": 1.5521,
      "theoretical_loss": 3.893758027067435,
      "tokens_seen": 563478528
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008376128385155467,
      "loss": 1.5585,
      "theoretical_loss": 3.8937106020713212,
      "tokens_seen": 563544064
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.49166640639305115,
      "objective/train/docs_used": 211161,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.651707410812378,
      "objective/train/original_loss": 1.651707410812378,
      "objective/train/theoretical_loss": 3.8936631841341076,
      "objective/train/tokens_used": 584069600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24235595762729645,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503987073898315,
      "objective/train/weighted_lm_loss": 1.734801173210144,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8936631841341076,
      "tokens_seen": 563609600
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837592778335005,
      "loss": 1.6154,
      "theoretical_loss": 3.8936631841341076,
      "tokens_seen": 563609600
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008375727181544635,
      "loss": 1.6302,
      "theoretical_loss": 3.8936157732539223,
      "tokens_seen": 563675136
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008375526579739217,
      "loss": 1.6018,
      "theoretical_loss": 3.8935683694288943,
      "tokens_seen": 563740672
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008375325977933802,
      "loss": 1.5432,
      "theoretical_loss": 3.8935209726571545,
      "tokens_seen": 563806208
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008375125376128385,
      "loss": 1.687,
      "theoretical_loss": 3.893473582936833,
      "tokens_seen": 563871744
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008374924774322969,
      "loss": 1.6945,
      "theoretical_loss": 3.8934262002660622,
      "tokens_seen": 563937280
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008374724172517553,
      "loss": 1.6496,
      "theoretical_loss": 3.8933788246429737,
      "tokens_seen": 564002816
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008374523570712137,
      "loss": 1.5673,
      "theoretical_loss": 3.893331456065701,
      "tokens_seen": 564068352
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837432296890672,
      "loss": 1.6722,
      "theoretical_loss": 3.8932840945323774,
      "tokens_seen": 564133888
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008374122367101305,
      "loss": 1.6569,
      "theoretical_loss": 3.8932367400411376,
      "tokens_seen": 564199424
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008373921765295888,
      "loss": 1.5352,
      "theoretical_loss": 3.8931893925901155,
      "tokens_seen": 564264960
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008373721163490472,
      "loss": 1.6271,
      "theoretical_loss": 3.893142052177449,
      "tokens_seen": 564330496
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008373520561685055,
      "loss": 1.6128,
      "theoretical_loss": 3.8930947188012737,
      "tokens_seen": 564396032
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837331995987964,
      "loss": 1.5328,
      "theoretical_loss": 3.893047392459726,
      "tokens_seen": 564461568
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008373119358074223,
      "loss": 1.5962,
      "theoretical_loss": 3.8930000731509455,
      "tokens_seen": 564527104
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008372918756268807,
      "loss": 1.5037,
      "theoretical_loss": 3.8929527608730696,
      "tokens_seen": 564592640
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837271815446339,
      "loss": 1.6041,
      "theoretical_loss": 3.8929054556242377,
      "tokens_seen": 564658176
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008372517552657974,
      "loss": 1.6258,
      "theoretical_loss": 3.8928581574025904,
      "tokens_seen": 564723712
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008372316950852558,
      "loss": 1.5506,
      "theoretical_loss": 3.8928108662062684,
      "tokens_seen": 564789248
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008372116349047142,
      "loss": 1.5471,
      "theoretical_loss": 3.8927635820334134,
      "tokens_seen": 564854784
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008371915747241725,
      "loss": 1.551,
      "theoretical_loss": 3.892716304882167,
      "tokens_seen": 564920320
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837171514543631,
      "loss": 1.6422,
      "theoretical_loss": 3.8926690347506727,
      "tokens_seen": 564985856
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008371514543630892,
      "loss": 1.6627,
      "theoretical_loss": 3.892621771637074,
      "tokens_seen": 565051392
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008371313941825477,
      "loss": 1.5411,
      "theoretical_loss": 3.892574515539515,
      "tokens_seen": 565116928
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837111334002006,
      "loss": 1.6011,
      "theoretical_loss": 3.892527266456141,
      "tokens_seen": 565182464
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008370912738214644,
      "loss": 1.7549,
      "theoretical_loss": 3.892480024385097,
      "tokens_seen": 565248000
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008370712136409228,
      "loss": 1.5504,
      "theoretical_loss": 3.8924327893245305,
      "tokens_seen": 565313536
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008370511534603812,
      "loss": 1.6712,
      "theoretical_loss": 3.892385561272588,
      "tokens_seen": 565379072
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008370310932798395,
      "loss": 1.6375,
      "theoretical_loss": 3.8923383402274174,
      "tokens_seen": 565444608
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000837011033099298,
      "loss": 1.6303,
      "theoretical_loss": 3.8922911261871675,
      "tokens_seen": 565510144
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008369909729187562,
      "loss": 1.6977,
      "theoretical_loss": 3.892243919149988,
      "tokens_seen": 565575680
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008369709127382147,
      "loss": 1.5744,
      "theoretical_loss": 3.8921967191140276,
      "tokens_seen": 565641216
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000836950852557673,
      "loss": 1.5975,
      "theoretical_loss": 3.8921495260774375,
      "tokens_seen": 565706752
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008369307923771314,
      "loss": 1.5785,
      "theoretical_loss": 3.89210234003837,
      "tokens_seen": 565772288
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008369107321965898,
      "loss": 1.5981,
      "theoretical_loss": 3.892055160994976,
      "tokens_seen": 565837824
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008368906720160481,
      "loss": 1.7243,
      "theoretical_loss": 3.8920079889454087,
      "tokens_seen": 565903360
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008368706118355065,
      "loss": 1.558,
      "theoretical_loss": 3.8919608238878216,
      "tokens_seen": 565968896
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000836850551654965,
      "loss": 1.5975,
      "theoretical_loss": 3.8919136658203684,
      "tokens_seen": 566034432
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008368304914744232,
      "loss": 1.5676,
      "theoretical_loss": 3.8918665147412055,
      "tokens_seen": 566099968
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008368104312938817,
      "loss": 1.3604,
      "theoretical_loss": 3.8918193706484865,
      "tokens_seen": 566165504
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008367903711133399,
      "loss": 1.6092,
      "theoretical_loss": 3.891772233540369,
      "tokens_seen": 566231040
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008367703109327984,
      "loss": 1.6393,
      "theoretical_loss": 3.891725103415009,
      "tokens_seen": 566296576
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008367502507522568,
      "loss": 1.6181,
      "theoretical_loss": 3.891677980270565,
      "tokens_seen": 566362112
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008367301905717151,
      "loss": 1.5997,
      "theoretical_loss": 3.8916308641051955,
      "tokens_seen": 566427648
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008367101303911735,
      "loss": 1.4944,
      "theoretical_loss": 3.8915837549170584,
      "tokens_seen": 566493184
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000836690070210632,
      "loss": 1.59,
      "theoretical_loss": 3.891536652704315,
      "tokens_seen": 566558720
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008366700100300902,
      "loss": 1.5597,
      "theoretical_loss": 3.8914895574651247,
      "tokens_seen": 566624256
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008366499498495487,
      "loss": 1.5383,
      "theoretical_loss": 3.891442469197649,
      "tokens_seen": 566689792
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000836629889669007,
      "loss": 1.6051,
      "theoretical_loss": 3.89139538790005,
      "tokens_seen": 566755328
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008366098294884654,
      "loss": 1.662,
      "theoretical_loss": 3.8913483135704894,
      "tokens_seen": 566820864
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.4925703704357147,
      "objective/train/docs_used": 212374,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.529457688331604,
      "objective/train/original_loss": 1.529457688331604,
      "objective/train/theoretical_loss": 3.8913012462071315,
      "objective/train/tokens_used": 587346400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24295853078365326,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504921674728394,
      "objective/train/weighted_lm_loss": 1.606675386428833,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0168063640594482,
      "theoretical_loss": 3.8913012462071315,
      "tokens_seen": 566886400
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008365897693079238,
      "loss": 1.6686,
      "theoretical_loss": 3.8913012462071315,
      "tokens_seen": 566886400
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008365697091273821,
      "loss": 1.6344,
      "theoretical_loss": 3.89125418580814,
      "tokens_seen": 566951936
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008365496489468405,
      "loss": 1.6334,
      "theoretical_loss": 3.8912071323716795,
      "tokens_seen": 567017472
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008365295887662989,
      "loss": 1.5775,
      "theoretical_loss": 3.8911600858959146,
      "tokens_seen": 567083008
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008365095285857572,
      "loss": 1.6099,
      "theoretical_loss": 3.8911130463790125,
      "tokens_seen": 567148544
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008364894684052157,
      "loss": 1.5115,
      "theoretical_loss": 3.891066013819139,
      "tokens_seen": 567214080
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000836469408224674,
      "loss": 1.7663,
      "theoretical_loss": 3.8910189882144626,
      "tokens_seen": 567279616
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008364493480441324,
      "loss": 1.6417,
      "theoretical_loss": 3.89097196956315,
      "tokens_seen": 567345152
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008364292878635908,
      "loss": 1.5897,
      "theoretical_loss": 3.890924957863372,
      "tokens_seen": 567410688
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008364092276830492,
      "loss": 1.7878,
      "theoretical_loss": 3.890877953113296,
      "tokens_seen": 567476224
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008363891675025075,
      "loss": 1.6121,
      "theoretical_loss": 3.8908309553110936,
      "tokens_seen": 567541760
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008363691073219659,
      "loss": 1.5549,
      "theoretical_loss": 3.890783964454935,
      "tokens_seen": 567607296
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008363490471414242,
      "loss": 1.5688,
      "theoretical_loss": 3.890736980542992,
      "tokens_seen": 567672832
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008363289869608827,
      "loss": 1.6342,
      "theoretical_loss": 3.890690003573437,
      "tokens_seen": 567738368
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008363089267803411,
      "loss": 1.6381,
      "theoretical_loss": 3.8906430335444426,
      "tokens_seen": 567803904
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008362888665997994,
      "loss": 1.4584,
      "theoretical_loss": 3.8905960704541833,
      "tokens_seen": 567869440
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008362688064192578,
      "loss": 1.6969,
      "theoretical_loss": 3.890549114300833,
      "tokens_seen": 567934976
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008362487462387162,
      "loss": 1.6092,
      "theoretical_loss": 3.890502165082566,
      "tokens_seen": 568000512
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008362286860581746,
      "loss": 1.7035,
      "theoretical_loss": 3.8904552227975593,
      "tokens_seen": 568066048
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008362086258776329,
      "loss": 1.6279,
      "theoretical_loss": 3.8904082874439885,
      "tokens_seen": 568131584
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008361885656970914,
      "loss": 1.6839,
      "theoretical_loss": 3.890361359020031,
      "tokens_seen": 568197120
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008361685055165496,
      "loss": 1.5912,
      "theoretical_loss": 3.890314437523865,
      "tokens_seen": 568262656
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008361484453360081,
      "loss": 1.6122,
      "theoretical_loss": 3.8902675229536685,
      "tokens_seen": 568328192
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008361283851554664,
      "loss": 1.6228,
      "theoretical_loss": 3.8902206153076206,
      "tokens_seen": 568393728
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008361083249749248,
      "loss": 1.6964,
      "theoretical_loss": 3.8901737145839017,
      "tokens_seen": 568459264
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008360882647943832,
      "loss": 1.6192,
      "theoretical_loss": 3.890126820780692,
      "tokens_seen": 568524800
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008360682046138416,
      "loss": 1.6188,
      "theoretical_loss": 3.8900799338961725,
      "tokens_seen": 568590336
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008360481444332999,
      "loss": 1.6078,
      "theoretical_loss": 3.8900330539285255,
      "tokens_seen": 568655872
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008360280842527584,
      "loss": 1.7185,
      "theoretical_loss": 3.8899861808759337,
      "tokens_seen": 568721408
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008360080240722166,
      "loss": 1.6761,
      "theoretical_loss": 3.889939314736581,
      "tokens_seen": 568786944
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008359879638916751,
      "loss": 1.6227,
      "theoretical_loss": 3.8898924555086496,
      "tokens_seen": 568852480
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008359679037111334,
      "loss": 1.6534,
      "theoretical_loss": 3.8898456031903255,
      "tokens_seen": 568918016
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008359478435305918,
      "loss": 1.5411,
      "theoretical_loss": 3.8897987577797943,
      "tokens_seen": 568983552
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008359277833500502,
      "loss": 1.5184,
      "theoretical_loss": 3.889751919275242,
      "tokens_seen": 569049088
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008359077231695085,
      "loss": 1.7104,
      "theoretical_loss": 3.8897050876748542,
      "tokens_seen": 569114624
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008358876629889669,
      "loss": 1.5735,
      "theoretical_loss": 3.8896582629768197,
      "tokens_seen": 569180160
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008358676028084254,
      "loss": 1.6771,
      "theoretical_loss": 3.8896114451793258,
      "tokens_seen": 569245696
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008358475426278836,
      "loss": 1.7303,
      "theoretical_loss": 3.8895646342805614,
      "tokens_seen": 569311232
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008358274824473421,
      "loss": 1.5807,
      "theoretical_loss": 3.8895178302787166,
      "tokens_seen": 569376768
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008358074222668003,
      "loss": 1.6122,
      "theoretical_loss": 3.8894710331719806,
      "tokens_seen": 569442304
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008357873620862588,
      "loss": 1.7613,
      "theoretical_loss": 3.889424242958545,
      "tokens_seen": 569507840
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008357673019057172,
      "loss": 1.5845,
      "theoretical_loss": 3.889377459636601,
      "tokens_seen": 569573376
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008357472417251755,
      "loss": 1.5114,
      "theoretical_loss": 3.8893306832043404,
      "tokens_seen": 569638912
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008357271815446339,
      "loss": 1.6762,
      "theoretical_loss": 3.8892839136599573,
      "tokens_seen": 569704448
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008357071213640924,
      "loss": 1.744,
      "theoretical_loss": 3.8892371510016446,
      "tokens_seen": 569769984
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008356870611835506,
      "loss": 1.6138,
      "theoretical_loss": 3.8891903952275966,
      "tokens_seen": 569835520
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008356670010030091,
      "loss": 1.6083,
      "theoretical_loss": 3.8891436463360076,
      "tokens_seen": 569901056
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008356469408224674,
      "loss": 1.5887,
      "theoretical_loss": 3.8890969043250743,
      "tokens_seen": 569966592
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008356268806419258,
      "loss": 1.7441,
      "theoretical_loss": 3.889050169192992,
      "tokens_seen": 570032128
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008356068204613842,
      "loss": 1.5472,
      "theoretical_loss": 3.889003440937959,
      "tokens_seen": 570097664
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.4926186800003052,
      "objective/train/docs_used": 213119,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4580154418945312,
      "objective/train/original_loss": 1.4580153226852417,
      "objective/train/theoretical_loss": 3.8889567195581716,
      "objective/train/tokens_used": 590623200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2429921180009842,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504971742630005,
      "objective/train/weighted_lm_loss": 1.5317919254302979,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0100501775741577,
      "theoretical_loss": 3.8889567195581716,
      "tokens_seen": 570163200
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008355867602808425,
      "loss": 1.6331,
      "theoretical_loss": 3.8889567195581716,
      "tokens_seen": 570163200
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008355667001003009,
      "loss": 1.5682,
      "theoretical_loss": 3.8889100050518284,
      "tokens_seen": 570228736
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008355466399197593,
      "loss": 1.544,
      "theoretical_loss": 3.8888632974171298,
      "tokens_seen": 570294272
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008355265797392176,
      "loss": 1.5229,
      "theoretical_loss": 3.8888165966522736,
      "tokens_seen": 570359808
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008355065195586761,
      "loss": 1.6094,
      "theoretical_loss": 3.8887699027554614,
      "tokens_seen": 570425344
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008354864593781344,
      "loss": 1.5273,
      "theoretical_loss": 3.8887232157248937,
      "tokens_seen": 570490880
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008354663991975928,
      "loss": 1.5283,
      "theoretical_loss": 3.8886765355587722,
      "tokens_seen": 570556416
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008354463390170512,
      "loss": 1.6902,
      "theoretical_loss": 3.8886298622553,
      "tokens_seen": 570621952
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008354262788365096,
      "loss": 1.6482,
      "theoretical_loss": 3.8885831958126786,
      "tokens_seen": 570687488
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008354062186559679,
      "loss": 1.6233,
      "theoretical_loss": 3.888536536229114,
      "tokens_seen": 570753024
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008353861584754263,
      "loss": 1.5259,
      "theoretical_loss": 3.8884898835028094,
      "tokens_seen": 570818560
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008353660982948846,
      "loss": 1.6376,
      "theoretical_loss": 3.8884432376319698,
      "tokens_seen": 570884096
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008353460381143431,
      "loss": 1.7704,
      "theoretical_loss": 3.8883965986148015,
      "tokens_seen": 570949632
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008353259779338014,
      "loss": 1.5043,
      "theoretical_loss": 3.8883499664495105,
      "tokens_seen": 571015168
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008353059177532598,
      "loss": 1.6525,
      "theoretical_loss": 3.8883033411343044,
      "tokens_seen": 571080704
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008352858575727181,
      "loss": 1.5998,
      "theoretical_loss": 3.88825672266739,
      "tokens_seen": 571146240
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008352657973921766,
      "loss": 1.5162,
      "theoretical_loss": 3.888210111046978,
      "tokens_seen": 571211776
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008352457372116349,
      "loss": 1.6696,
      "theoretical_loss": 3.888163506271275,
      "tokens_seen": 571277312
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008352256770310933,
      "loss": 1.7017,
      "theoretical_loss": 3.888116908338492,
      "tokens_seen": 571342848
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008352056168505516,
      "loss": 1.6289,
      "theoretical_loss": 3.88807031724684,
      "tokens_seen": 571408384
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.00083518555667001,
      "loss": 1.5372,
      "theoretical_loss": 3.8880237329945295,
      "tokens_seen": 571473920
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008351654964894684,
      "loss": 1.7852,
      "theoretical_loss": 3.8879771555797724,
      "tokens_seen": 571539456
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008351454363089268,
      "loss": 1.5828,
      "theoretical_loss": 3.8879305850007815,
      "tokens_seen": 571604992
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008351253761283851,
      "loss": 1.6125,
      "theoretical_loss": 3.88788402125577,
      "tokens_seen": 571670528
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008351053159478436,
      "loss": 1.6661,
      "theoretical_loss": 3.887837464342952,
      "tokens_seen": 571736064
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008350852557673019,
      "loss": 1.7204,
      "theoretical_loss": 3.887790914260541,
      "tokens_seen": 571801600
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008350651955867603,
      "loss": 1.5715,
      "theoretical_loss": 3.887744371006753,
      "tokens_seen": 571867136
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008350451354062187,
      "loss": 1.763,
      "theoretical_loss": 3.8876978345798046,
      "tokens_seen": 571932672
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000835025075225677,
      "loss": 1.5181,
      "theoretical_loss": 3.8876513049779113,
      "tokens_seen": 571998208
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008350050150451354,
      "loss": 1.6728,
      "theoretical_loss": 3.8876047821992903,
      "tokens_seen": 572063744
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008349849548645938,
      "loss": 1.6322,
      "theoretical_loss": 3.88755826624216,
      "tokens_seen": 572129280
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008349648946840521,
      "loss": 1.6106,
      "theoretical_loss": 3.887511757104739,
      "tokens_seen": 572194816
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008349448345035106,
      "loss": 1.6854,
      "theoretical_loss": 3.887465254785246,
      "tokens_seen": 572260352
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008349247743229688,
      "loss": 1.7422,
      "theoretical_loss": 3.8874187592819016,
      "tokens_seen": 572325888
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008349047141424273,
      "loss": 1.5888,
      "theoretical_loss": 3.887372270592926,
      "tokens_seen": 572391424
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008348846539618857,
      "loss": 1.7566,
      "theoretical_loss": 3.8873257887165407,
      "tokens_seen": 572456960
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834864593781344,
      "loss": 1.5776,
      "theoretical_loss": 3.887279313650967,
      "tokens_seen": 572522496
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008348445336008024,
      "loss": 1.5842,
      "theoretical_loss": 3.8872328453944283,
      "tokens_seen": 572588032
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008348244734202607,
      "loss": 1.7323,
      "theoretical_loss": 3.887186383945147,
      "tokens_seen": 572653568
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008348044132397192,
      "loss": 1.6703,
      "theoretical_loss": 3.8871399293013473,
      "tokens_seen": 572719104
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008347843530591776,
      "loss": 1.6304,
      "theoretical_loss": 3.8870934814612546,
      "tokens_seen": 572784640
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008347642928786359,
      "loss": 1.5136,
      "theoretical_loss": 3.8870470404230932,
      "tokens_seen": 572850176
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008347442326980943,
      "loss": 1.4513,
      "theoretical_loss": 3.8870006061850892,
      "tokens_seen": 572915712
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008347241725175528,
      "loss": 1.694,
      "theoretical_loss": 3.88695417874547,
      "tokens_seen": 572981248
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834704112337011,
      "loss": 1.5685,
      "theoretical_loss": 3.886907758102461,
      "tokens_seen": 573046784
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008346840521564695,
      "loss": 1.7508,
      "theoretical_loss": 3.8868613442542923,
      "tokens_seen": 573112320
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008346639919759278,
      "loss": 1.704,
      "theoretical_loss": 3.886814937199191,
      "tokens_seen": 573177856
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008346439317953862,
      "loss": 1.6086,
      "theoretical_loss": 3.886768536935387,
      "tokens_seen": 573243392
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008346238716148446,
      "loss": 1.5357,
      "theoretical_loss": 3.8867221434611094,
      "tokens_seen": 573308928
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834603811434303,
      "loss": 1.7271,
      "theoretical_loss": 3.88667575677459,
      "tokens_seen": 573374464
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.490614116191864,
      "objective/train/docs_used": 214228,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5626332759857178,
      "objective/train/original_loss": 1.5626335144042969,
      "objective/train/theoretical_loss": 3.8866293768740587,
      "objective/train/tokens_used": 593900000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24128560721874237,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502879619598389,
      "objective/train/weighted_lm_loss": 1.641161561012268,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 3.8866293768740587,
      "tokens_seen": 573440000
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008345837512537613,
      "loss": 1.5757,
      "theoretical_loss": 3.8866293768740587,
      "tokens_seen": 573440000
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008345636910732197,
      "loss": 1.5246,
      "theoretical_loss": 3.8865830037577487,
      "tokens_seen": 573505536
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834543630892678,
      "loss": 1.6753,
      "theoretical_loss": 3.8865366374238914,
      "tokens_seen": 573571072
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008345235707121365,
      "loss": 1.7239,
      "theoretical_loss": 3.8864902778707213,
      "tokens_seen": 573636608
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008345035105315948,
      "loss": 1.5884,
      "theoretical_loss": 3.8864439250964704,
      "tokens_seen": 573702144
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008344834503510532,
      "loss": 1.6644,
      "theoretical_loss": 3.886397579099375,
      "tokens_seen": 573767680
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008344633901705116,
      "loss": 1.5428,
      "theoretical_loss": 3.88635123987767,
      "tokens_seen": 573833216
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.00083444332998997,
      "loss": 1.5276,
      "theoretical_loss": 3.8863049074295906,
      "tokens_seen": 573898752
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008344232698094283,
      "loss": 1.6814,
      "theoretical_loss": 3.886258581753374,
      "tokens_seen": 573964288
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008344032096288867,
      "loss": 1.7092,
      "theoretical_loss": 3.8862122628472564,
      "tokens_seen": 574029824
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834383149448345,
      "loss": 1.6209,
      "theoretical_loss": 3.8861659507094766,
      "tokens_seen": 574095360
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008343630892678035,
      "loss": 1.5473,
      "theoretical_loss": 3.8861196453382734,
      "tokens_seen": 574160896
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008343430290872618,
      "loss": 1.629,
      "theoretical_loss": 3.8860733467318846,
      "tokens_seen": 574226432
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008343229689067202,
      "loss": 1.7029,
      "theoretical_loss": 3.886027054888551,
      "tokens_seen": 574291968
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008343029087261785,
      "loss": 1.5779,
      "theoretical_loss": 3.885980769806513,
      "tokens_seen": 574357504
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834282848545637,
      "loss": 1.6816,
      "theoretical_loss": 3.8859344914840115,
      "tokens_seen": 574423040
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008342627883650953,
      "loss": 1.6854,
      "theoretical_loss": 3.8858882199192886,
      "tokens_seen": 574488576
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008342427281845537,
      "loss": 1.6028,
      "theoretical_loss": 3.8858419551105863,
      "tokens_seen": 574554112
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834222668004012,
      "loss": 1.7471,
      "theoretical_loss": 3.8857956970561487,
      "tokens_seen": 574619648
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008342026078234704,
      "loss": 1.6562,
      "theoretical_loss": 3.885749445754218,
      "tokens_seen": 574685184
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008341825476429288,
      "loss": 1.5984,
      "theoretical_loss": 3.8857032012030404,
      "tokens_seen": 574750720
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008341624874623872,
      "loss": 1.4557,
      "theoretical_loss": 3.8856569634008595,
      "tokens_seen": 574816256
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008341424272818455,
      "loss": 1.541,
      "theoretical_loss": 3.8856107323459215,
      "tokens_seen": 574881792
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834122367101304,
      "loss": 1.7277,
      "theoretical_loss": 3.885564508036473,
      "tokens_seen": 574947328
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008341023069207623,
      "loss": 1.6655,
      "theoretical_loss": 3.8855182904707615,
      "tokens_seen": 575012864
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008340822467402207,
      "loss": 1.6032,
      "theoretical_loss": 3.885472079647034,
      "tokens_seen": 575078400
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000834062186559679,
      "loss": 1.5355,
      "theoretical_loss": 3.8854258755635387,
      "tokens_seen": 575143936
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008340421263791374,
      "loss": 1.6302,
      "theoretical_loss": 3.885379678218525,
      "tokens_seen": 575209472
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008340220661985958,
      "loss": 1.6603,
      "theoretical_loss": 3.885333487610243,
      "tokens_seen": 575275008
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008340020060180542,
      "loss": 1.6416,
      "theoretical_loss": 3.885287303736942,
      "tokens_seen": 575340544
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008339819458375125,
      "loss": 1.6654,
      "theoretical_loss": 3.885241126596874,
      "tokens_seen": 575406080
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000833961885656971,
      "loss": 1.5494,
      "theoretical_loss": 3.8851949561882897,
      "tokens_seen": 575471616
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008339418254764292,
      "loss": 1.4818,
      "theoretical_loss": 3.885148792509442,
      "tokens_seen": 575537152
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008339217652958877,
      "loss": 1.6263,
      "theoretical_loss": 3.8851026355585843,
      "tokens_seen": 575602688
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008339017051153461,
      "loss": 1.7115,
      "theoretical_loss": 3.885056485333969,
      "tokens_seen": 575668224
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008338816449348044,
      "loss": 1.6162,
      "theoretical_loss": 3.8850103418338513,
      "tokens_seen": 575733760
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008338615847542628,
      "loss": 1.7863,
      "theoretical_loss": 3.884964205056486,
      "tokens_seen": 575799296
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008338415245737211,
      "loss": 1.5634,
      "theoretical_loss": 3.884918075000128,
      "tokens_seen": 575864832
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008338214643931795,
      "loss": 1.6212,
      "theoretical_loss": 3.884871951663034,
      "tokens_seen": 575930368
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000833801404212638,
      "loss": 1.5676,
      "theoretical_loss": 3.8848258350434612,
      "tokens_seen": 575995904
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008337813440320962,
      "loss": 1.4362,
      "theoretical_loss": 3.884779725139666,
      "tokens_seen": 576061440
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008337612838515547,
      "loss": 1.5206,
      "theoretical_loss": 3.8847336219499073,
      "tokens_seen": 576126976
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008337412236710131,
      "loss": 1.651,
      "theoretical_loss": 3.8846875254724442,
      "tokens_seen": 576192512
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008337211634904714,
      "loss": 1.6546,
      "theoretical_loss": 3.884641435705535,
      "tokens_seen": 576258048
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008337011033099298,
      "loss": 1.6489,
      "theoretical_loss": 3.884595352647441,
      "tokens_seen": 576323584
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008336810431293882,
      "loss": 1.5317,
      "theoretical_loss": 3.884549276296423,
      "tokens_seen": 576389120
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008336609829488465,
      "loss": 1.5102,
      "theoretical_loss": 3.8845032066507414,
      "tokens_seen": 576454656
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000833640922768305,
      "loss": 1.5993,
      "theoretical_loss": 3.8844571437086586,
      "tokens_seen": 576520192
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008336208625877632,
      "loss": 1.4862,
      "theoretical_loss": 3.884411087468438,
      "tokens_seen": 576585728
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008336008024072217,
      "loss": 1.583,
      "theoretical_loss": 3.8843650379283416,
      "tokens_seen": 576651264
    },
    {
      "epoch": 0.17,
      "objective/train/advantage_avg": 0.49178701639175415,
      "objective/train/docs_used": 215374,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.578829288482666,
      "objective/train/original_loss": 1.5788294076919556,
      "objective/train/theoretical_loss": 3.884318995086635,
      "objective/train/tokens_used": 597176800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24241264164447784,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504109859466553,
      "objective/train/weighted_lm_loss": 1.6582890748977661,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.884318995086635,
      "tokens_seen": 576716800
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.00083358074222668,
      "loss": 1.6054,
      "theoretical_loss": 3.884318995086635,
      "tokens_seen": 576716800
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008335606820461384,
      "loss": 1.6457,
      "theoretical_loss": 3.884272958941581,
      "tokens_seen": 576782336
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008335406218655968,
      "loss": 1.5649,
      "theoretical_loss": 3.8842269294914464,
      "tokens_seen": 576847872
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008335205616850552,
      "loss": 1.6548,
      "theoretical_loss": 3.8841809067344966,
      "tokens_seen": 576913408
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008335005015045135,
      "loss": 1.6309,
      "theoretical_loss": 3.8841348906689985,
      "tokens_seen": 576978944
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000833480441323972,
      "loss": 1.6167,
      "theoretical_loss": 3.8840888812932186,
      "tokens_seen": 577044480
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008334603811434302,
      "loss": 1.648,
      "theoretical_loss": 3.8840428786054253,
      "tokens_seen": 577110016
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008334403209628887,
      "loss": 1.667,
      "theoretical_loss": 3.8839968826038866,
      "tokens_seen": 577175552
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.000833420260782347,
      "loss": 1.6751,
      "theoretical_loss": 3.8839508932868725,
      "tokens_seen": 577241088
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008334002006018054,
      "loss": 1.6029,
      "theoretical_loss": 3.883904910652652,
      "tokens_seen": 577306624
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008333801404212638,
      "loss": 1.7712,
      "theoretical_loss": 3.8838589346994965,
      "tokens_seen": 577372160
    },
    {
      "epoch": 0.17,
      "learning_rate": 0.0008333600802407222,
      "loss": 1.543,
      "theoretical_loss": 3.8838129654256757,
      "tokens_seen": 577437696
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008333400200601805,
      "loss": 1.4709,
      "theoretical_loss": 3.8837670028294626,
      "tokens_seen": 577503232
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008333199598796389,
      "loss": 1.5841,
      "theoretical_loss": 3.8837210469091286,
      "tokens_seen": 577568768
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008332998996990974,
      "loss": 1.625,
      "theoretical_loss": 3.8836750976629473,
      "tokens_seen": 577634304
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008332798395185557,
      "loss": 1.6486,
      "theoretical_loss": 3.883629155089192,
      "tokens_seen": 577699840
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008332597793380141,
      "loss": 1.6697,
      "theoretical_loss": 3.883583219186138,
      "tokens_seen": 577765376
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008332397191574724,
      "loss": 1.6512,
      "theoretical_loss": 3.8835372899520593,
      "tokens_seen": 577830912
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008332196589769308,
      "loss": 1.5178,
      "theoretical_loss": 3.883491367385231,
      "tokens_seen": 577896448
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008331995987963892,
      "loss": 1.6438,
      "theoretical_loss": 3.8834454514839303,
      "tokens_seen": 577961984
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008331795386158476,
      "loss": 1.7216,
      "theoretical_loss": 3.8833995422464342,
      "tokens_seen": 578027520
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008331594784353059,
      "loss": 1.5895,
      "theoretical_loss": 3.8833536396710198,
      "tokens_seen": 578093056
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008331394182547644,
      "loss": 1.6638,
      "theoretical_loss": 3.8833077437559647,
      "tokens_seen": 578158592
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008331193580742227,
      "loss": 1.6073,
      "theoretical_loss": 3.8832618544995485,
      "tokens_seen": 578224128
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008330992978936811,
      "loss": 1.6409,
      "theoretical_loss": 3.88321597190005,
      "tokens_seen": 578289664
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008330792377131395,
      "loss": 1.6481,
      "theoretical_loss": 3.88317009595575,
      "tokens_seen": 578355200
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008330591775325978,
      "loss": 1.7624,
      "theoretical_loss": 3.8831242266649286,
      "tokens_seen": 578420736
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008330391173520562,
      "loss": 1.5285,
      "theoretical_loss": 3.8830783640258675,
      "tokens_seen": 578486272
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008330190571715146,
      "loss": 1.6166,
      "theoretical_loss": 3.883032508036848,
      "tokens_seen": 578551808
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008329989969909729,
      "loss": 1.5131,
      "theoretical_loss": 3.8829866586961543,
      "tokens_seen": 578617344
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008329789368104314,
      "loss": 1.6345,
      "theoretical_loss": 3.882940816002068,
      "tokens_seen": 578682880
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008329588766298896,
      "loss": 1.6407,
      "theoretical_loss": 3.882894979952874,
      "tokens_seen": 578748416
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008329388164493481,
      "loss": 1.5498,
      "theoretical_loss": 3.882849150546856,
      "tokens_seen": 578813952
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008329187562688065,
      "loss": 1.681,
      "theoretical_loss": 3.8828033277823,
      "tokens_seen": 578879488
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008328986960882648,
      "loss": 1.707,
      "theoretical_loss": 3.8827575116574913,
      "tokens_seen": 578945024
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008328786359077232,
      "loss": 1.6368,
      "theoretical_loss": 3.8827117021707167,
      "tokens_seen": 579010560
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008328585757271815,
      "loss": 1.5782,
      "theoretical_loss": 3.8826658993202625,
      "tokens_seen": 579076096
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008328385155466399,
      "loss": 1.6314,
      "theoretical_loss": 3.8826201031044176,
      "tokens_seen": 579141632
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008328184553660984,
      "loss": 1.6671,
      "theoretical_loss": 3.8825743135214696,
      "tokens_seen": 579207168
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008327983951855566,
      "loss": 1.5509,
      "theoretical_loss": 3.882528530569708,
      "tokens_seen": 579272704
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008327783350050151,
      "loss": 1.6627,
      "theoretical_loss": 3.8824827542474214,
      "tokens_seen": 579338240
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008327582748244735,
      "loss": 1.601,
      "theoretical_loss": 3.8824369845529008,
      "tokens_seen": 579403776
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008327382146439318,
      "loss": 1.577,
      "theoretical_loss": 3.8823912214844376,
      "tokens_seen": 579469312
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008327181544633902,
      "loss": 1.744,
      "theoretical_loss": 3.882345465040322,
      "tokens_seen": 579534848
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008326980942828486,
      "loss": 1.6169,
      "theoretical_loss": 3.882299715218848,
      "tokens_seen": 579600384
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008326780341023069,
      "loss": 1.6419,
      "theoretical_loss": 3.8822539720183062,
      "tokens_seen": 579665920
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008326579739217654,
      "loss": 1.6029,
      "theoretical_loss": 3.882208235436992,
      "tokens_seen": 579731456
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008326379137412236,
      "loss": 1.6054,
      "theoretical_loss": 3.8821625054731976,
      "tokens_seen": 579796992
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008326178535606821,
      "loss": 1.669,
      "theoretical_loss": 3.8821167821252196,
      "tokens_seen": 579862528
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008325977933801404,
      "loss": 1.5815,
      "theoretical_loss": 3.8820710653913526,
      "tokens_seen": 579928064
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.4898017644882202,
      "objective/train/docs_used": 216668,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6693953275680542,
      "objective/train/original_loss": 1.6693953275680542,
      "objective/train/theoretical_loss": 3.8820253552698913,
      "objective/train/tokens_used": 600453600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24118079245090485,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0502060651779175,
      "objective/train/weighted_lm_loss": 1.7532109022140503,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8820253552698913,
      "tokens_seen": 579993600
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008325777331995988,
      "loss": 1.6935,
      "theoretical_loss": 3.8820253552698913,
      "tokens_seen": 579993600
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008325576730190572,
      "loss": 1.7155,
      "theoretical_loss": 3.881979651759134,
      "tokens_seen": 580059136
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008325376128385156,
      "loss": 1.6103,
      "theoretical_loss": 3.8819339548573772,
      "tokens_seen": 580124672
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008325175526579739,
      "loss": 1.5621,
      "theoretical_loss": 3.881888264562919,
      "tokens_seen": 580190208
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008324974924774324,
      "loss": 1.6421,
      "theoretical_loss": 3.8818425808740575,
      "tokens_seen": 580255744
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008324774322968906,
      "loss": 1.6178,
      "theoretical_loss": 3.881796903789092,
      "tokens_seen": 580321280
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008324573721163491,
      "loss": 1.676,
      "theoretical_loss": 3.881751233306322,
      "tokens_seen": 580386816
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008324373119358074,
      "loss": 1.6241,
      "theoretical_loss": 3.881705569424049,
      "tokens_seen": 580452352
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008324172517552658,
      "loss": 1.6982,
      "theoretical_loss": 3.8816599121405724,
      "tokens_seen": 580517888
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008323971915747242,
      "loss": 1.6278,
      "theoretical_loss": 3.881614261454195,
      "tokens_seen": 580583424
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008323771313941826,
      "loss": 1.5379,
      "theoretical_loss": 3.881568617363218,
      "tokens_seen": 580648960
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008323570712136409,
      "loss": 1.6818,
      "theoretical_loss": 3.881522979865945,
      "tokens_seen": 580714496
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008323370110330993,
      "loss": 1.69,
      "theoretical_loss": 3.88147734896068,
      "tokens_seen": 580780032
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008323169508525577,
      "loss": 1.5399,
      "theoretical_loss": 3.8814317246457266,
      "tokens_seen": 580845568
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008322968906720161,
      "loss": 1.6669,
      "theoretical_loss": 3.881386106919389,
      "tokens_seen": 580911104
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008322768304914744,
      "loss": 1.6094,
      "theoretical_loss": 3.8813404957799733,
      "tokens_seen": 580976640
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008322567703109328,
      "loss": 1.6382,
      "theoretical_loss": 3.881294891225785,
      "tokens_seen": 581042176
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008322367101303911,
      "loss": 1.5975,
      "theoretical_loss": 3.8812492932551317,
      "tokens_seen": 581107712
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008322166499498496,
      "loss": 1.7833,
      "theoretical_loss": 3.88120370186632,
      "tokens_seen": 581173248
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008321965897693079,
      "loss": 1.6112,
      "theoretical_loss": 3.8811581170576575,
      "tokens_seen": 581238784
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008321765295887663,
      "loss": 1.6593,
      "theoretical_loss": 3.881112538827453,
      "tokens_seen": 581304320
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008321564694082247,
      "loss": 1.5667,
      "theoretical_loss": 3.881066967174016,
      "tokens_seen": 581369856
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008321364092276831,
      "loss": 1.5722,
      "theoretical_loss": 3.881021402095657,
      "tokens_seen": 581435392
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008321163490471414,
      "loss": 1.6631,
      "theoretical_loss": 3.8809758435906843,
      "tokens_seen": 581500928
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008320962888665999,
      "loss": 1.5772,
      "theoretical_loss": 3.88093029165741,
      "tokens_seen": 581566464
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008320762286860581,
      "loss": 1.6153,
      "theoretical_loss": 3.8808847462941465,
      "tokens_seen": 581632000
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008320561685055166,
      "loss": 1.5789,
      "theoretical_loss": 3.880839207499205,
      "tokens_seen": 581697536
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008320361083249749,
      "loss": 1.5706,
      "theoretical_loss": 3.8807936752708985,
      "tokens_seen": 581763072
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008320160481444333,
      "loss": 1.697,
      "theoretical_loss": 3.880748149607541,
      "tokens_seen": 581828608
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008319959879638917,
      "loss": 1.6527,
      "theoretical_loss": 3.8807026305074475,
      "tokens_seen": 581894144
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.00083197592778335,
      "loss": 1.6531,
      "theoretical_loss": 3.880657117968931,
      "tokens_seen": 581959680
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008319558676028084,
      "loss": 1.5888,
      "theoretical_loss": 3.8806116119903074,
      "tokens_seen": 582025216
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008319358074222669,
      "loss": 1.6148,
      "theoretical_loss": 3.880566112569893,
      "tokens_seen": 582090752
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008319157472417251,
      "loss": 1.7515,
      "theoretical_loss": 3.8805206197060045,
      "tokens_seen": 582156288
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008318956870611836,
      "loss": 1.7871,
      "theoretical_loss": 3.880475133396959,
      "tokens_seen": 582221824
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008318756268806418,
      "loss": 1.5739,
      "theoretical_loss": 3.880429653641075,
      "tokens_seen": 582287360
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008318555667001003,
      "loss": 1.6752,
      "theoretical_loss": 3.8803841804366694,
      "tokens_seen": 582352896
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008318355065195587,
      "loss": 1.5701,
      "theoretical_loss": 3.880338713782063,
      "tokens_seen": 582418432
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000831815446339017,
      "loss": 1.6733,
      "theoretical_loss": 3.8802932536755748,
      "tokens_seen": 582483968
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008317953861584755,
      "loss": 1.5975,
      "theoretical_loss": 3.8802478001155247,
      "tokens_seen": 582549504
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008317753259779339,
      "loss": 1.8073,
      "theoretical_loss": 3.880202353100235,
      "tokens_seen": 582615040
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008317552657973922,
      "loss": 1.6272,
      "theoretical_loss": 3.880156912628026,
      "tokens_seen": 582680576
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008317352056168506,
      "loss": 1.5908,
      "theoretical_loss": 3.880111478697221,
      "tokens_seen": 582746112
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000831715145436309,
      "loss": 1.7125,
      "theoretical_loss": 3.880066051306142,
      "tokens_seen": 582811648
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008316950852557673,
      "loss": 1.7426,
      "theoretical_loss": 3.880020630453113,
      "tokens_seen": 582877184
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008316750250752258,
      "loss": 1.6432,
      "theoretical_loss": 3.879975216136457,
      "tokens_seen": 582942720
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000831654964894684,
      "loss": 1.6708,
      "theoretical_loss": 3.8799298083545004,
      "tokens_seen": 583008256
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008316349047141425,
      "loss": 1.4938,
      "theoretical_loss": 3.8798844071055667,
      "tokens_seen": 583073792
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008316148445336008,
      "loss": 1.6715,
      "theoretical_loss": 3.8798390123879836,
      "tokens_seen": 583139328
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008315947843530592,
      "loss": 1.668,
      "theoretical_loss": 3.879793624200077,
      "tokens_seen": 583204864
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.48885348439216614,
      "objective/train/docs_used": 217803,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.614006519317627,
      "objective/train/original_loss": 1.614006519317627,
      "objective/train/theoretical_loss": 3.879748242540173,
      "objective/train/tokens_used": 603730400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.23961599171161652,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0501033067703247,
      "objective/train/weighted_lm_loss": 1.6949542760849,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.879748242540173,
      "tokens_seen": 583270400
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008315747241725176,
      "loss": 1.6647,
      "theoretical_loss": 3.879748242540173,
      "tokens_seen": 583270400
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000831554663991976,
      "loss": 1.6274,
      "theoretical_loss": 3.8797028674066008,
      "tokens_seen": 583335936
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008315346038114343,
      "loss": 1.6726,
      "theoretical_loss": 3.8796574987976875,
      "tokens_seen": 583401472
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008315145436308928,
      "loss": 1.6305,
      "theoretical_loss": 3.8796121367117635,
      "tokens_seen": 583467008
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000831494483450351,
      "loss": 1.7564,
      "theoretical_loss": 3.8795667811471573,
      "tokens_seen": 583532544
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008314744232698095,
      "loss": 1.6404,
      "theoretical_loss": 3.8795214321022,
      "tokens_seen": 583598080
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008314543630892678,
      "loss": 1.6265,
      "theoretical_loss": 3.8794760895752223,
      "tokens_seen": 583663616
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008314343029087262,
      "loss": 1.7085,
      "theoretical_loss": 3.879430753564555,
      "tokens_seen": 583729152
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008314142427281846,
      "loss": 1.5586,
      "theoretical_loss": 3.8793854240685306,
      "tokens_seen": 583794688
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000831394182547643,
      "loss": 1.6759,
      "theoretical_loss": 3.8793401010854813,
      "tokens_seen": 583860224
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008313741223671013,
      "loss": 1.6092,
      "theoretical_loss": 3.8792947846137418,
      "tokens_seen": 583925760
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008313540621865597,
      "loss": 1.5635,
      "theoretical_loss": 3.879249474651645,
      "tokens_seen": 583991296
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000831334002006018,
      "loss": 1.6645,
      "theoretical_loss": 3.879204171197525,
      "tokens_seen": 584056832
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008313139418254765,
      "loss": 1.5576,
      "theoretical_loss": 3.8791588742497183,
      "tokens_seen": 584122368
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008312938816449348,
      "loss": 1.5826,
      "theoretical_loss": 3.8791135838065594,
      "tokens_seen": 584187904
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008312738214643932,
      "loss": 1.5733,
      "theoretical_loss": 3.8790682998663852,
      "tokens_seen": 584253440
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008312537612838515,
      "loss": 1.7476,
      "theoretical_loss": 3.879023022427533,
      "tokens_seen": 584318976
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.00083123370110331,
      "loss": 1.6421,
      "theoretical_loss": 3.87897775148834,
      "tokens_seen": 584384512
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008312136409227683,
      "loss": 1.65,
      "theoretical_loss": 3.878932487047144,
      "tokens_seen": 584450048
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008311935807422267,
      "loss": 1.6201,
      "theoretical_loss": 3.8788872291022845,
      "tokens_seen": 584515584
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008311735205616851,
      "loss": 1.5899,
      "theoretical_loss": 3.878841977652101,
      "tokens_seen": 584581120
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008311534603811435,
      "loss": 1.5468,
      "theoretical_loss": 3.8787967326949335,
      "tokens_seen": 584646656
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008311334002006018,
      "loss": 1.6185,
      "theoretical_loss": 3.878751494229122,
      "tokens_seen": 584712192
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008311133400200603,
      "loss": 1.7026,
      "theoretical_loss": 3.8787062622530084,
      "tokens_seen": 584777728
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008310932798395185,
      "loss": 1.6082,
      "theoretical_loss": 3.8786610367649343,
      "tokens_seen": 584843264
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000831073219658977,
      "loss": 1.6696,
      "theoretical_loss": 3.878615817763242,
      "tokens_seen": 584908800
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008310531594784353,
      "loss": 1.6589,
      "theoretical_loss": 3.8785706052462756,
      "tokens_seen": 584974336
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008310330992978937,
      "loss": 1.5331,
      "theoretical_loss": 3.8785253992123776,
      "tokens_seen": 585039872
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008310130391173521,
      "loss": 1.7219,
      "theoretical_loss": 3.8784801996598928,
      "tokens_seen": 585105408
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008309929789368104,
      "loss": 1.5674,
      "theoretical_loss": 3.878435006587166,
      "tokens_seen": 585170944
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008309729187562688,
      "loss": 1.6549,
      "theoretical_loss": 3.8783898199925426,
      "tokens_seen": 585236480
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008309528585757273,
      "loss": 1.6957,
      "theoretical_loss": 3.8783446398743693,
      "tokens_seen": 585302016
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008309327983951855,
      "loss": 1.6219,
      "theoretical_loss": 3.878299466230992,
      "tokens_seen": 585367552
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000830912738214644,
      "loss": 1.5805,
      "theoretical_loss": 3.8782542990607594,
      "tokens_seen": 585433088
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008308926780341022,
      "loss": 1.5979,
      "theoretical_loss": 3.8782091383620183,
      "tokens_seen": 585498624
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008308726178535607,
      "loss": 1.6933,
      "theoretical_loss": 3.878163984133117,
      "tokens_seen": 585564160
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008308525576730191,
      "loss": 1.5984,
      "theoretical_loss": 3.8781188363724057,
      "tokens_seen": 585629696
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008308324974924774,
      "loss": 1.6588,
      "theoretical_loss": 3.8780736950782337,
      "tokens_seen": 585695232
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008308124373119358,
      "loss": 1.547,
      "theoretical_loss": 3.8780285602489513,
      "tokens_seen": 585760768
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008307923771313943,
      "loss": 1.6371,
      "theoretical_loss": 3.87798343188291,
      "tokens_seen": 585826304
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008307723169508525,
      "loss": 1.5934,
      "theoretical_loss": 3.87793830997846,
      "tokens_seen": 585891840
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000830752256770311,
      "loss": 1.5106,
      "theoretical_loss": 3.8778931945339554,
      "tokens_seen": 585957376
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008307321965897692,
      "loss": 1.5949,
      "theoretical_loss": 3.8778480855477477,
      "tokens_seen": 586022912
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008307121364092277,
      "loss": 1.7127,
      "theoretical_loss": 3.8778029830181904,
      "tokens_seen": 586088448
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008306920762286861,
      "loss": 1.5959,
      "theoretical_loss": 3.8777578869436384,
      "tokens_seen": 586153984
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008306720160481444,
      "loss": 1.7218,
      "theoretical_loss": 3.8777127973224452,
      "tokens_seen": 586219520
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008306519558676028,
      "loss": 1.6409,
      "theoretical_loss": 3.8776677141529667,
      "tokens_seen": 586285056
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008306318956870612,
      "loss": 1.5809,
      "theoretical_loss": 3.877622637433558,
      "tokens_seen": 586350592
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008306118355065195,
      "loss": 1.5752,
      "theoretical_loss": 3.8775775671625765,
      "tokens_seen": 586416128
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000830591775325978,
      "loss": 1.6401,
      "theoretical_loss": 3.8775325033383785,
      "tokens_seen": 586481664
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.4921160936355591,
      "objective/train/docs_used": 218927,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.7031680345535278,
      "objective/train/original_loss": 1.7031681537628174,
      "objective/train/theoretical_loss": 3.8774874459593223,
      "objective/train/tokens_used": 607007200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.242936372756958,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050446629524231,
      "objective/train/weighted_lm_loss": 1.7889472246170044,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8774874459593223,
      "tokens_seen": 586547200
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008305717151454363,
      "loss": 1.7481,
      "theoretical_loss": 3.8774874459593223,
      "tokens_seen": 586547200
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008305516549648947,
      "loss": 1.5825,
      "theoretical_loss": 3.877442395023765,
      "tokens_seen": 586612736
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000830531594784353,
      "loss": 1.5842,
      "theoretical_loss": 3.8773973505300674,
      "tokens_seen": 586678272
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008305115346038114,
      "loss": 1.5653,
      "theoretical_loss": 3.8773523124765865,
      "tokens_seen": 586743808
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008304914744232698,
      "loss": 1.653,
      "theoretical_loss": 3.8773072808616837,
      "tokens_seen": 586809344
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008304714142427282,
      "loss": 1.5497,
      "theoretical_loss": 3.8772622556837195,
      "tokens_seen": 586874880
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008304513540621865,
      "loss": 1.6068,
      "theoretical_loss": 3.877217236941055,
      "tokens_seen": 586940416
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000830431293881645,
      "loss": 1.7406,
      "theoretical_loss": 3.8771722246320524,
      "tokens_seen": 587005952
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008304112337011033,
      "loss": 1.632,
      "theoretical_loss": 3.8771272187550734,
      "tokens_seen": 587071488
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008303911735205617,
      "loss": 1.6805,
      "theoretical_loss": 3.8770822193084817,
      "tokens_seen": 587137024
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.00083037111334002,
      "loss": 1.6109,
      "theoretical_loss": 3.877037226290641,
      "tokens_seen": 587202560
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008303510531594785,
      "loss": 1.6222,
      "theoretical_loss": 3.8769922396999146,
      "tokens_seen": 587268096
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008303309929789368,
      "loss": 1.5739,
      "theoretical_loss": 3.8769472595346683,
      "tokens_seen": 587333632
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008303109327983952,
      "loss": 1.5446,
      "theoretical_loss": 3.876902285793267,
      "tokens_seen": 587399168
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008302908726178536,
      "loss": 1.6361,
      "theoretical_loss": 3.8768573184740767,
      "tokens_seen": 587464704
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008302708124373119,
      "loss": 1.5991,
      "theoretical_loss": 3.8768123575754645,
      "tokens_seen": 587530240
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008302507522567704,
      "loss": 1.6164,
      "theoretical_loss": 3.8767674030957977,
      "tokens_seen": 587595776
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008302306920762287,
      "loss": 1.6073,
      "theoretical_loss": 3.876722455033443,
      "tokens_seen": 587661312
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008302106318956871,
      "loss": 1.5783,
      "theoretical_loss": 3.87667751338677,
      "tokens_seen": 587726848
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008301905717151455,
      "loss": 1.5493,
      "theoretical_loss": 3.8766325781541475,
      "tokens_seen": 587792384
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008301705115346039,
      "loss": 1.6647,
      "theoretical_loss": 3.876587649333944,
      "tokens_seen": 587857920
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008301504513540622,
      "loss": 1.6051,
      "theoretical_loss": 3.8765427269245314,
      "tokens_seen": 587923456
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008301303911735207,
      "loss": 1.5461,
      "theoretical_loss": 3.8764978109242794,
      "tokens_seen": 587988992
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008301103309929789,
      "loss": 1.5925,
      "theoretical_loss": 3.8764529013315596,
      "tokens_seen": 588054528
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008300902708124374,
      "loss": 1.6489,
      "theoretical_loss": 3.876407998144744,
      "tokens_seen": 588120064
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008300702106318957,
      "loss": 1.6229,
      "theoretical_loss": 3.8763631013622053,
      "tokens_seen": 588185600
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008300501504513541,
      "loss": 1.6815,
      "theoretical_loss": 3.8763182109823173,
      "tokens_seen": 588251136
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008300300902708125,
      "loss": 1.5642,
      "theoretical_loss": 3.876273327003452,
      "tokens_seen": 588316672
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008300100300902708,
      "loss": 1.4461,
      "theoretical_loss": 3.876228449423986,
      "tokens_seen": 588382208
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008299899699097292,
      "loss": 1.5003,
      "theoretical_loss": 3.8761835782422924,
      "tokens_seen": 588447744
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008299699097291877,
      "loss": 1.5458,
      "theoretical_loss": 3.8761387134567475,
      "tokens_seen": 588513280
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008299498495486459,
      "loss": 1.5513,
      "theoretical_loss": 3.8760938550657276,
      "tokens_seen": 588578816
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008299297893681044,
      "loss": 1.6935,
      "theoretical_loss": 3.8760490030676094,
      "tokens_seen": 588644352
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008299097291875626,
      "loss": 1.6617,
      "theoretical_loss": 3.8760041574607698,
      "tokens_seen": 588709888
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008298896690070211,
      "loss": 1.5717,
      "theoretical_loss": 3.8759593182435874,
      "tokens_seen": 588775424
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008298696088264795,
      "loss": 1.7028,
      "theoretical_loss": 3.87591448541444,
      "tokens_seen": 588840960
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008298495486459378,
      "loss": 1.6496,
      "theoretical_loss": 3.875869658971707,
      "tokens_seen": 588906496
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008298294884653962,
      "loss": 1.7078,
      "theoretical_loss": 3.875824838913769,
      "tokens_seen": 588972032
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008298094282848547,
      "loss": 1.6872,
      "theoretical_loss": 3.875780025239005,
      "tokens_seen": 589037568
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008297893681043129,
      "loss": 1.6225,
      "theoretical_loss": 3.8757352179457962,
      "tokens_seen": 589103104
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008297693079237714,
      "loss": 1.5302,
      "theoretical_loss": 3.8756904170325246,
      "tokens_seen": 589168640
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008297492477432296,
      "loss": 1.5347,
      "theoretical_loss": 3.8756456224975713,
      "tokens_seen": 589234176
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008297291875626881,
      "loss": 1.6227,
      "theoretical_loss": 3.8756008343393202,
      "tokens_seen": 589299712
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008297091273821465,
      "loss": 1.6826,
      "theoretical_loss": 3.8755560525561537,
      "tokens_seen": 589365248
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008296890672016048,
      "loss": 1.5985,
      "theoretical_loss": 3.8755112771464555,
      "tokens_seen": 589430784
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008296690070210632,
      "loss": 1.7784,
      "theoretical_loss": 3.8754665081086106,
      "tokens_seen": 589496320
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008296489468405216,
      "loss": 1.5733,
      "theoretical_loss": 3.8754217454410043,
      "tokens_seen": 589561856
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008296288866599799,
      "loss": 1.5693,
      "theoretical_loss": 3.875376989142021,
      "tokens_seen": 589627392
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008296088264794384,
      "loss": 1.6317,
      "theoretical_loss": 3.8753322392100475,
      "tokens_seen": 589692928
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008295887662988967,
      "loss": 1.6052,
      "theoretical_loss": 3.8752874956434713,
      "tokens_seen": 589758464
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.4909812808036804,
      "objective/train/docs_used": 220165,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6117116212844849,
      "objective/train/original_loss": 1.6117116212844849,
      "objective/train/theoretical_loss": 3.875242758440679,
      "objective/train/tokens_used": 610284000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24177321791648865,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503270626068115,
      "objective/train/weighted_lm_loss": 1.6927611827850342,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.875242758440679,
      "tokens_seen": 589824000
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008295687061183551,
      "loss": 1.7046,
      "theoretical_loss": 3.875242758440679,
      "tokens_seen": 589824000
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008295486459378134,
      "loss": 1.5758,
      "theoretical_loss": 3.8751980276000584,
      "tokens_seen": 589889536
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008295285857572718,
      "loss": 1.6406,
      "theoretical_loss": 3.8751533031199985,
      "tokens_seen": 589955072
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008295085255767302,
      "loss": 1.5666,
      "theoretical_loss": 3.875108584998888,
      "tokens_seen": 590020608
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008294884653961886,
      "loss": 1.6276,
      "theoretical_loss": 3.875063873235117,
      "tokens_seen": 590086144
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008294684052156469,
      "loss": 1.6714,
      "theoretical_loss": 3.875019167827076,
      "tokens_seen": 590151680
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008294483450351054,
      "loss": 1.8013,
      "theoretical_loss": 3.8749744687731553,
      "tokens_seen": 590217216
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008294282848545637,
      "loss": 1.632,
      "theoretical_loss": 3.8749297760717463,
      "tokens_seen": 590282752
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008294082246740221,
      "loss": 1.6734,
      "theoretical_loss": 3.874885089721242,
      "tokens_seen": 590348288
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008293881644934804,
      "loss": 1.6035,
      "theoretical_loss": 3.8748404097200346,
      "tokens_seen": 590413824
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008293681043129389,
      "loss": 1.6007,
      "theoretical_loss": 3.874795736066517,
      "tokens_seen": 590479360
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008293480441323972,
      "loss": 1.5973,
      "theoretical_loss": 3.8747510687590827,
      "tokens_seen": 590544896
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008293279839518556,
      "loss": 1.4097,
      "theoretical_loss": 3.8747064077961264,
      "tokens_seen": 590610432
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008293079237713139,
      "loss": 1.6399,
      "theoretical_loss": 3.874661753176044,
      "tokens_seen": 590675968
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008292878635907723,
      "loss": 1.4542,
      "theoretical_loss": 3.87461710489723,
      "tokens_seen": 590741504
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008292678034102307,
      "loss": 1.6091,
      "theoretical_loss": 3.8745724629580813,
      "tokens_seen": 590807040
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008292477432296891,
      "loss": 1.5679,
      "theoretical_loss": 3.874527827356994,
      "tokens_seen": 590872576
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008292276830491474,
      "loss": 1.6006,
      "theoretical_loss": 3.874483198092366,
      "tokens_seen": 590938112
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008292076228686059,
      "loss": 1.5959,
      "theoretical_loss": 3.8744385751625945,
      "tokens_seen": 591003648
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008291875626880642,
      "loss": 1.6411,
      "theoretical_loss": 3.8743939585660785,
      "tokens_seen": 591069184
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008291675025075226,
      "loss": 1.7404,
      "theoretical_loss": 3.8743493483012172,
      "tokens_seen": 591134720
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000829147442326981,
      "loss": 1.5292,
      "theoretical_loss": 3.87430474436641,
      "tokens_seen": 591200256
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008291273821464393,
      "loss": 1.6631,
      "theoretical_loss": 3.8742601467600566,
      "tokens_seen": 591265792
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008291073219658977,
      "loss": 1.5965,
      "theoretical_loss": 3.874215555480559,
      "tokens_seen": 591331328
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008290872617853561,
      "loss": 1.6225,
      "theoretical_loss": 3.874170970526317,
      "tokens_seen": 591396864
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008290672016048144,
      "loss": 1.5969,
      "theoretical_loss": 3.874126391895734,
      "tokens_seen": 591462400
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008290471414242729,
      "loss": 1.656,
      "theoretical_loss": 3.874081819587212,
      "tokens_seen": 591527936
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008290270812437311,
      "loss": 1.5633,
      "theoretical_loss": 3.8740372535991545,
      "tokens_seen": 591593472
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008290070210631896,
      "loss": 1.7388,
      "theoretical_loss": 3.873992693929965,
      "tokens_seen": 591659008
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000828986960882648,
      "loss": 1.4588,
      "theoretical_loss": 3.8739481405780474,
      "tokens_seen": 591724544
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008289669007021063,
      "loss": 1.599,
      "theoretical_loss": 3.873903593541807,
      "tokens_seen": 591790080
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008289468405215647,
      "loss": 1.6276,
      "theoretical_loss": 3.8738590528196486,
      "tokens_seen": 591855616
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000828926780341023,
      "loss": 1.5245,
      "theoretical_loss": 3.8738145184099797,
      "tokens_seen": 591921152
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008289067201604814,
      "loss": 1.6088,
      "theoretical_loss": 3.8737699903112057,
      "tokens_seen": 591986688
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008288866599799399,
      "loss": 1.5537,
      "theoretical_loss": 3.8737254685217337,
      "tokens_seen": 592052224
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008288665997993981,
      "loss": 1.5496,
      "theoretical_loss": 3.8736809530399725,
      "tokens_seen": 592117760
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008288465396188566,
      "loss": 1.6273,
      "theoretical_loss": 3.8736364438643296,
      "tokens_seen": 592183296
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008288264794383151,
      "loss": 1.5786,
      "theoretical_loss": 3.8735919409932134,
      "tokens_seen": 592248832
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008288064192577733,
      "loss": 1.6814,
      "theoretical_loss": 3.873547444425035,
      "tokens_seen": 592314368
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008287863590772318,
      "loss": 1.6656,
      "theoretical_loss": 3.8735029541582033,
      "tokens_seen": 592379904
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.00082876629889669,
      "loss": 1.6918,
      "theoretical_loss": 3.87345847019113,
      "tokens_seen": 592445440
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008287462387161485,
      "loss": 1.5025,
      "theoretical_loss": 3.8734139925222246,
      "tokens_seen": 592510976
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008287261785356069,
      "loss": 1.6032,
      "theoretical_loss": 3.8733695211499004,
      "tokens_seen": 592576512
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008287061183550652,
      "loss": 1.6024,
      "theoretical_loss": 3.8733250560725696,
      "tokens_seen": 592642048
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008286860581745236,
      "loss": 1.5592,
      "theoretical_loss": 3.8732805972886446,
      "tokens_seen": 592707584
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000828665997993982,
      "loss": 1.6073,
      "theoretical_loss": 3.873236144796539,
      "tokens_seen": 592773120
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008286459378134403,
      "loss": 1.6096,
      "theoretical_loss": 3.8731916985946677,
      "tokens_seen": 592838656
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008286258776328988,
      "loss": 1.5658,
      "theoretical_loss": 3.8731472586814446,
      "tokens_seen": 592904192
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000828605817452357,
      "loss": 1.5567,
      "theoretical_loss": 3.873102825055285,
      "tokens_seen": 592969728
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008285857572718155,
      "loss": 1.6636,
      "theoretical_loss": 3.8730583977146047,
      "tokens_seen": 593035264
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.4914178252220154,
      "objective/train/docs_used": 221319,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5754297971725464,
      "objective/train/original_loss": 1.5754297971725464,
      "objective/train/theoretical_loss": 3.873013976657821,
      "objective/train/tokens_used": 613560800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24230973422527313,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503736734390259,
      "objective/train/weighted_lm_loss": 1.6547985076904297,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9048373699188232,
      "theoretical_loss": 3.873013976657821,
      "tokens_seen": 593100800
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008285656970912738,
      "loss": 1.6206,
      "theoretical_loss": 3.873013976657821,
      "tokens_seen": 593100800
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008285456369107322,
      "loss": 1.6143,
      "theoretical_loss": 3.87296956188335,
      "tokens_seen": 593166336
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008285255767301906,
      "loss": 1.4283,
      "theoretical_loss": 3.87292515338961,
      "tokens_seen": 593231872
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000828505516549649,
      "loss": 1.6161,
      "theoretical_loss": 3.8728807511750176,
      "tokens_seen": 593297408
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008284854563691073,
      "loss": 1.6033,
      "theoretical_loss": 3.872836355237993,
      "tokens_seen": 593362944
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008284653961885658,
      "loss": 1.6209,
      "theoretical_loss": 3.8727919655769547,
      "tokens_seen": 593428480
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008284453360080241,
      "loss": 1.5363,
      "theoretical_loss": 3.872747582190324,
      "tokens_seen": 593494016
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008284252758274825,
      "loss": 1.6599,
      "theoretical_loss": 3.872703205076519,
      "tokens_seen": 593559552
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008284052156469408,
      "loss": 1.6903,
      "theoretical_loss": 3.8726588342339623,
      "tokens_seen": 593625088
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008283851554663993,
      "loss": 1.6047,
      "theoretical_loss": 3.872614469661075,
      "tokens_seen": 593690624
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008283650952858576,
      "loss": 1.538,
      "theoretical_loss": 3.8725701113562794,
      "tokens_seen": 593756160
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000828345035105316,
      "loss": 1.5646,
      "theoretical_loss": 3.8725257593179983,
      "tokens_seen": 593821696
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008283249749247743,
      "loss": 1.5392,
      "theoretical_loss": 3.872481413544654,
      "tokens_seen": 593887232
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008283049147442327,
      "loss": 1.6154,
      "theoretical_loss": 3.8724370740346723,
      "tokens_seen": 593952768
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008282848545636911,
      "loss": 1.6049,
      "theoretical_loss": 3.8723927407864758,
      "tokens_seen": 594018304
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008282647943831495,
      "loss": 1.6822,
      "theoretical_loss": 3.8723484137984903,
      "tokens_seen": 594083840
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008282447342026078,
      "loss": 1.4928,
      "theoretical_loss": 3.872304093069141,
      "tokens_seen": 594149376
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008282246740220663,
      "loss": 1.6281,
      "theoretical_loss": 3.8722597785968547,
      "tokens_seen": 594214912
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008282046138415246,
      "loss": 1.6902,
      "theoretical_loss": 3.8722154703800573,
      "tokens_seen": 594280448
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000828184553660983,
      "loss": 1.5848,
      "theoretical_loss": 3.8721711684171765,
      "tokens_seen": 594345984
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008281644934804413,
      "loss": 1.5481,
      "theoretical_loss": 3.87212687270664,
      "tokens_seen": 594411520
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008281444332998997,
      "loss": 1.7675,
      "theoretical_loss": 3.8720825832468764,
      "tokens_seen": 594477056
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008281243731193581,
      "loss": 1.6445,
      "theoretical_loss": 3.8720383000363148,
      "tokens_seen": 594542592
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008281043129388165,
      "loss": 1.6253,
      "theoretical_loss": 3.871994023073384,
      "tokens_seen": 594608128
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008280842527582748,
      "loss": 1.5153,
      "theoretical_loss": 3.8719497523565147,
      "tokens_seen": 594673664
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008280641925777333,
      "loss": 1.6036,
      "theoretical_loss": 3.871905487884138,
      "tokens_seen": 594739200
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008280441323971915,
      "loss": 1.7277,
      "theoretical_loss": 3.871861229654684,
      "tokens_seen": 594804736
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.00082802407221665,
      "loss": 1.4775,
      "theoretical_loss": 3.871816977666586,
      "tokens_seen": 594870272
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008280040120361084,
      "loss": 1.494,
      "theoretical_loss": 3.8717727319182744,
      "tokens_seen": 594935808
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008279839518555667,
      "loss": 1.6224,
      "theoretical_loss": 3.8717284924081836,
      "tokens_seen": 595001344
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008279638916750251,
      "loss": 1.5985,
      "theoretical_loss": 3.8716842591347476,
      "tokens_seen": 595066880
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008279438314944834,
      "loss": 1.5947,
      "theoretical_loss": 3.8716400320963986,
      "tokens_seen": 595132416
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008279237713139418,
      "loss": 1.6328,
      "theoretical_loss": 3.8715958112915727,
      "tokens_seen": 595197952
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008279037111334003,
      "loss": 1.592,
      "theoretical_loss": 3.871551596718705,
      "tokens_seen": 595263488
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008278836509528585,
      "loss": 1.7456,
      "theoretical_loss": 3.871507388376231,
      "tokens_seen": 595329024
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000827863590772317,
      "loss": 1.6105,
      "theoretical_loss": 3.8714631862625866,
      "tokens_seen": 595394560
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008278435305917754,
      "loss": 1.5579,
      "theoretical_loss": 3.8714189903762093,
      "tokens_seen": 595460096
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008278234704112337,
      "loss": 1.5667,
      "theoretical_loss": 3.8713748007155364,
      "tokens_seen": 595525632
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008278034102306921,
      "loss": 1.5983,
      "theoretical_loss": 3.871330617279006,
      "tokens_seen": 595591168
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008277833500501504,
      "loss": 1.6201,
      "theoretical_loss": 3.871286440065057,
      "tokens_seen": 595656704
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008277632898696088,
      "loss": 1.6441,
      "theoretical_loss": 3.8712422690721278,
      "tokens_seen": 595722240
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008277432296890673,
      "loss": 1.4712,
      "theoretical_loss": 3.8711981042986583,
      "tokens_seen": 595787776
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008277231695085255,
      "loss": 1.7621,
      "theoretical_loss": 3.8711539457430897,
      "tokens_seen": 595853312
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000827703109327984,
      "loss": 1.599,
      "theoretical_loss": 3.8711097934038614,
      "tokens_seen": 595918848
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008276830491474423,
      "loss": 1.5107,
      "theoretical_loss": 3.8710656472794156,
      "tokens_seen": 595984384
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008276629889669007,
      "loss": 1.6662,
      "theoretical_loss": 3.8710215073681944,
      "tokens_seen": 596049920
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008276429287863591,
      "loss": 1.5608,
      "theoretical_loss": 3.87097737366864,
      "tokens_seen": 596115456
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008276228686058175,
      "loss": 1.7127,
      "theoretical_loss": 3.870933246179196,
      "tokens_seen": 596180992
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008276028084252758,
      "loss": 1.5064,
      "theoretical_loss": 3.870889124898306,
      "tokens_seen": 596246528
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008275827482447342,
      "loss": 1.579,
      "theoretical_loss": 3.8708450098244134,
      "tokens_seen": 596312064
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.4931483566761017,
      "objective/train/docs_used": 222574,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4788110256195068,
      "objective/train/original_loss": 1.4788110256195068,
      "objective/train/theoretical_loss": 3.870800900955963,
      "objective/train/tokens_used": 616837600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2440672218799591,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505554676055908,
      "objective/train/weighted_lm_loss": 1.5536764860153198,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.870800900955963,
      "tokens_seen": 596377600
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008275626880641925,
      "loss": 1.5266,
      "theoretical_loss": 3.870800900955963,
      "tokens_seen": 596377600
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000827542627883651,
      "loss": 1.5448,
      "theoretical_loss": 3.8707567982914015,
      "tokens_seen": 596443136
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008275225677031093,
      "loss": 1.7643,
      "theoretical_loss": 3.8707127018291736,
      "tokens_seen": 596508672
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008275025075225677,
      "loss": 1.7255,
      "theoretical_loss": 3.8706686115677265,
      "tokens_seen": 596574208
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008274824473420261,
      "loss": 1.4608,
      "theoretical_loss": 3.8706245275055062,
      "tokens_seen": 596639744
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008274623871614845,
      "loss": 1.6486,
      "theoretical_loss": 3.8705804496409613,
      "tokens_seen": 596705280
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008274423269809428,
      "loss": 1.5271,
      "theoretical_loss": 3.87053637797254,
      "tokens_seen": 596770816
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008274222668004012,
      "loss": 1.5131,
      "theoretical_loss": 3.8704923124986896,
      "tokens_seen": 596836352
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008274022066198595,
      "loss": 1.6762,
      "theoretical_loss": 3.8704482532178606,
      "tokens_seen": 596901888
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000827382146439318,
      "loss": 1.6253,
      "theoretical_loss": 3.870404200128503,
      "tokens_seen": 596967424
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008273620862587763,
      "loss": 1.7444,
      "theoretical_loss": 3.870360153229066,
      "tokens_seen": 597032960
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008273420260782347,
      "loss": 1.6915,
      "theoretical_loss": 3.8703161125180015,
      "tokens_seen": 597098496
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008273219658976931,
      "loss": 1.6255,
      "theoretical_loss": 3.8702720779937607,
      "tokens_seen": 597164032
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008273019057171515,
      "loss": 1.6267,
      "theoretical_loss": 3.8702280496547954,
      "tokens_seen": 597229568
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008272818455366099,
      "loss": 1.6136,
      "theoretical_loss": 3.870184027499559,
      "tokens_seen": 597295104
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008272617853560682,
      "loss": 1.482,
      "theoretical_loss": 3.8701400115265034,
      "tokens_seen": 597360640
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008272417251755267,
      "loss": 1.6012,
      "theoretical_loss": 3.8700960017340833,
      "tokens_seen": 597426176
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000827221664994985,
      "loss": 1.5897,
      "theoretical_loss": 3.870051998120753,
      "tokens_seen": 597491712
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008272016048144434,
      "loss": 1.4931,
      "theoretical_loss": 3.8700080006849666,
      "tokens_seen": 597557248
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008271815446339017,
      "loss": 1.6767,
      "theoretical_loss": 3.86996400942518,
      "tokens_seen": 597622784
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008271614844533601,
      "loss": 1.6117,
      "theoretical_loss": 3.8699200243398493,
      "tokens_seen": 597688320
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008271414242728185,
      "loss": 1.7407,
      "theoretical_loss": 3.8698760454274304,
      "tokens_seen": 597753856
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008271213640922769,
      "loss": 1.6222,
      "theoretical_loss": 3.869832072686381,
      "tokens_seen": 597819392
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008271013039117352,
      "loss": 1.6342,
      "theoretical_loss": 3.869788106115158,
      "tokens_seen": 597884928
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008270812437311937,
      "loss": 1.7497,
      "theoretical_loss": 3.8697441457122204,
      "tokens_seen": 597950464
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008270611835506519,
      "loss": 1.527,
      "theoretical_loss": 3.869700191476026,
      "tokens_seen": 598016000
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008270411233701104,
      "loss": 1.6388,
      "theoretical_loss": 3.869656243405035,
      "tokens_seen": 598081536
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008270210631895688,
      "loss": 1.6888,
      "theoretical_loss": 3.869612301497707,
      "tokens_seen": 598147072
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008270010030090271,
      "loss": 1.6937,
      "theoretical_loss": 3.8695683657525013,
      "tokens_seen": 598212608
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008269809428284855,
      "loss": 1.7128,
      "theoretical_loss": 3.8695244361678798,
      "tokens_seen": 598278144
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008269608826479438,
      "loss": 1.5467,
      "theoretical_loss": 3.869480512742304,
      "tokens_seen": 598343680
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008269408224674022,
      "loss": 1.6887,
      "theoretical_loss": 3.8694365954742356,
      "tokens_seen": 598409216
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008269207622868607,
      "loss": 1.6296,
      "theoretical_loss": 3.8693926843621376,
      "tokens_seen": 598474752
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008269007021063189,
      "loss": 1.6146,
      "theoretical_loss": 3.869348779404473,
      "tokens_seen": 598540288
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008268806419257774,
      "loss": 1.5223,
      "theoretical_loss": 3.869304880599705,
      "tokens_seen": 598605824
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008268605817452358,
      "loss": 1.65,
      "theoretical_loss": 3.8692609879462987,
      "tokens_seen": 598671360
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008268405215646941,
      "loss": 1.635,
      "theoretical_loss": 3.8692171014427177,
      "tokens_seen": 598736896
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008268204613841525,
      "loss": 1.5668,
      "theoretical_loss": 3.8691732210874283,
      "tokens_seen": 598802432
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008268004012036108,
      "loss": 1.5304,
      "theoretical_loss": 3.8691293468788963,
      "tokens_seen": 598867968
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008267803410230692,
      "loss": 1.5948,
      "theoretical_loss": 3.869085478815588,
      "tokens_seen": 598933504
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008267602808425277,
      "loss": 1.6285,
      "theoretical_loss": 3.86904161689597,
      "tokens_seen": 598999040
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008267402206619859,
      "loss": 1.6064,
      "theoretical_loss": 3.8689977611185107,
      "tokens_seen": 599064576
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008267201604814444,
      "loss": 1.6543,
      "theoretical_loss": 3.868953911481678,
      "tokens_seen": 599130112
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008267001003009027,
      "loss": 1.6609,
      "theoretical_loss": 3.86891006798394,
      "tokens_seen": 599195648
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008266800401203611,
      "loss": 1.694,
      "theoretical_loss": 3.868866230623766,
      "tokens_seen": 599261184
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008266599799398195,
      "loss": 1.7556,
      "theoretical_loss": 3.868822399399626,
      "tokens_seen": 599326720
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008266399197592779,
      "loss": 1.5449,
      "theoretical_loss": 3.8687785743099896,
      "tokens_seen": 599392256
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008266198595787362,
      "loss": 1.5468,
      "theoretical_loss": 3.868734755353329,
      "tokens_seen": 599457792
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008265997993981946,
      "loss": 1.5997,
      "theoretical_loss": 3.8686909425281146,
      "tokens_seen": 599523328
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008265797392176529,
      "loss": 1.55,
      "theoretical_loss": 3.8686471358328185,
      "tokens_seen": 599588864
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.4909364879131317,
      "objective/train/docs_used": 223680,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6145431995391846,
      "objective/train/original_loss": 1.6145431995391846,
      "objective/train/theoretical_loss": 3.868603335265913,
      "objective/train/tokens_used": 620114400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24157437682151794,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503215789794922,
      "objective/train/weighted_lm_loss": 1.695879578590393,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.868603335265913,
      "tokens_seen": 599654400
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008265596790371114,
      "loss": 1.594,
      "theoretical_loss": 3.868603335265913,
      "tokens_seen": 599654400
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008265396188565697,
      "loss": 1.6526,
      "theoretical_loss": 3.8685595408258724,
      "tokens_seen": 599719936
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008265195586760281,
      "loss": 1.6839,
      "theoretical_loss": 3.8685157525111684,
      "tokens_seen": 599785472
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008264994984954865,
      "loss": 1.5806,
      "theoretical_loss": 3.868471970320276,
      "tokens_seen": 599851008
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008264794383149449,
      "loss": 1.6813,
      "theoretical_loss": 3.86842819425167,
      "tokens_seen": 599916544
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008264593781344032,
      "loss": 1.6634,
      "theoretical_loss": 3.8683844243038257,
      "tokens_seen": 599982080
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008264393179538616,
      "loss": 1.5757,
      "theoretical_loss": 3.8683406604752184,
      "tokens_seen": 600047616
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008264192577733199,
      "loss": 1.5867,
      "theoretical_loss": 3.8682969027643246,
      "tokens_seen": 600113152
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008263991975927784,
      "loss": 1.6753,
      "theoretical_loss": 3.8682531511696214,
      "tokens_seen": 600178688
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008263791374122367,
      "loss": 1.5887,
      "theoretical_loss": 3.868209405689586,
      "tokens_seen": 600244224
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008263590772316951,
      "loss": 1.7754,
      "theoretical_loss": 3.868165666322696,
      "tokens_seen": 600309760
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008263390170511534,
      "loss": 1.6739,
      "theoretical_loss": 3.868121933067431,
      "tokens_seen": 600375296
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008263189568706119,
      "loss": 1.589,
      "theoretical_loss": 3.8680782059222687,
      "tokens_seen": 600440832
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008262988966900702,
      "loss": 1.6292,
      "theoretical_loss": 3.8680344848856896,
      "tokens_seen": 600506368
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008262788365095286,
      "loss": 1.6832,
      "theoretical_loss": 3.8679907699561733,
      "tokens_seen": 600571904
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000826258776328987,
      "loss": 1.6069,
      "theoretical_loss": 3.8679470611322007,
      "tokens_seen": 600637440
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008262387161484454,
      "loss": 1.6778,
      "theoretical_loss": 3.8679033584122533,
      "tokens_seen": 600702976
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008262186559679037,
      "loss": 1.5571,
      "theoretical_loss": 3.867859661794812,
      "tokens_seen": 600768512
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008261985957873621,
      "loss": 1.7327,
      "theoretical_loss": 3.86781597127836,
      "tokens_seen": 600834048
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008261785356068204,
      "loss": 1.6719,
      "theoretical_loss": 3.8677722868613804,
      "tokens_seen": 600899584
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008261584754262789,
      "loss": 1.7021,
      "theoretical_loss": 3.867728608542355,
      "tokens_seen": 600965120
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008261384152457372,
      "loss": 1.5693,
      "theoretical_loss": 3.867684936319769,
      "tokens_seen": 601030656
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008261183550651956,
      "loss": 1.6478,
      "theoretical_loss": 3.867641270192107,
      "tokens_seen": 601096192
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000826098294884654,
      "loss": 1.6648,
      "theoretical_loss": 3.867597610157853,
      "tokens_seen": 601161728
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008260782347041123,
      "loss": 1.6999,
      "theoretical_loss": 3.8675539562154935,
      "tokens_seen": 601227264
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008260581745235707,
      "loss": 1.6054,
      "theoretical_loss": 3.8675103083635145,
      "tokens_seen": 601292800
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008260381143430292,
      "loss": 1.6909,
      "theoretical_loss": 3.867466666600402,
      "tokens_seen": 601358336
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008260180541624874,
      "loss": 1.7423,
      "theoretical_loss": 3.867423030924644,
      "tokens_seen": 601423872
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008259979939819459,
      "loss": 1.6517,
      "theoretical_loss": 3.8673794013347274,
      "tokens_seen": 601489408
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008259779338014041,
      "loss": 1.5173,
      "theoretical_loss": 3.8673357778291404,
      "tokens_seen": 601554944
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008259578736208626,
      "loss": 1.594,
      "theoretical_loss": 3.867292160406373,
      "tokens_seen": 601620480
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000825937813440321,
      "loss": 1.5296,
      "theoretical_loss": 3.8672485490649136,
      "tokens_seen": 601686016
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008259177532597793,
      "loss": 1.6407,
      "theoretical_loss": 3.8672049438032525,
      "tokens_seen": 601751552
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008258976930792377,
      "loss": 1.515,
      "theoretical_loss": 3.8671613446198796,
      "tokens_seen": 601817088
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008258776328986962,
      "loss": 1.694,
      "theoretical_loss": 3.8671177515132857,
      "tokens_seen": 601882624
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008258575727181544,
      "loss": 1.7133,
      "theoretical_loss": 3.8670741644819637,
      "tokens_seen": 601948160
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008258375125376129,
      "loss": 1.5727,
      "theoretical_loss": 3.867030583524404,
      "tokens_seen": 602013696
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008258174523570712,
      "loss": 1.6105,
      "theoretical_loss": 3.8669870086391,
      "tokens_seen": 602079232
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008257973921765296,
      "loss": 1.6828,
      "theoretical_loss": 3.866943439824545,
      "tokens_seen": 602144768
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008257773319959881,
      "loss": 1.4657,
      "theoretical_loss": 3.866899877079232,
      "tokens_seen": 602210304
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008257572718154463,
      "loss": 1.6577,
      "theoretical_loss": 3.8668563204016557,
      "tokens_seen": 602275840
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008257372116349048,
      "loss": 1.5897,
      "theoretical_loss": 3.86681276979031,
      "tokens_seen": 602341376
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008257171514543631,
      "loss": 1.6591,
      "theoretical_loss": 3.8667692252436914,
      "tokens_seen": 602406912
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008256970912738215,
      "loss": 1.7114,
      "theoretical_loss": 3.866725686760295,
      "tokens_seen": 602472448
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008256770310932799,
      "loss": 1.5512,
      "theoretical_loss": 3.8666821543386174,
      "tokens_seen": 602537984
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008256569709127383,
      "loss": 1.579,
      "theoretical_loss": 3.8666386279771547,
      "tokens_seen": 602603520
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008256369107321966,
      "loss": 1.6909,
      "theoretical_loss": 3.8665951076744056,
      "tokens_seen": 602669056
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000825616850551655,
      "loss": 1.5716,
      "theoretical_loss": 3.8665515934288672,
      "tokens_seen": 602734592
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008255967903711133,
      "loss": 1.5794,
      "theoretical_loss": 3.8665080852390377,
      "tokens_seen": 602800128
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008255767301905718,
      "loss": 1.6037,
      "theoretical_loss": 3.866464583103417,
      "tokens_seen": 602865664
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.49399736523628235,
      "objective/train/docs_used": 224736,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.500554084777832,
      "objective/train/original_loss": 1.5005537271499634,
      "objective/train/theoretical_loss": 3.866421087020504,
      "objective/train/tokens_used": 623391200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24457556009292603,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506432056427002,
      "objective/train/weighted_lm_loss": 1.5765460729599,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.866421087020504,
      "tokens_seen": 602931200
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008255566700100301,
      "loss": 1.5648,
      "theoretical_loss": 3.866421087020504,
      "tokens_seen": 602931200
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008255366098294885,
      "loss": 1.619,
      "theoretical_loss": 3.866377596988799,
      "tokens_seen": 602996736
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008255165496489469,
      "loss": 1.5137,
      "theoretical_loss": 3.8663341130068023,
      "tokens_seen": 603062272
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008254964894684053,
      "loss": 1.5574,
      "theoretical_loss": 3.8662906350730157,
      "tokens_seen": 603127808
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008254764292878636,
      "loss": 1.4186,
      "theoretical_loss": 3.8662471631859407,
      "tokens_seen": 603193344
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000825456369107322,
      "loss": 1.701,
      "theoretical_loss": 3.8662036973440794,
      "tokens_seen": 603258880
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008254363089267803,
      "loss": 1.6054,
      "theoretical_loss": 3.866160237545934,
      "tokens_seen": 603324416
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008254162487462388,
      "loss": 1.6217,
      "theoretical_loss": 3.866116783790009,
      "tokens_seen": 603389952
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008253961885656971,
      "loss": 1.612,
      "theoretical_loss": 3.866073336074807,
      "tokens_seen": 603455488
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008253761283851555,
      "loss": 1.5529,
      "theoretical_loss": 3.8660298943988334,
      "tokens_seen": 603521024
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008253560682046138,
      "loss": 1.5292,
      "theoretical_loss": 3.8659864587605917,
      "tokens_seen": 603586560
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008253360080240723,
      "loss": 1.6586,
      "theoretical_loss": 3.865943029158589,
      "tokens_seen": 603652096
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008253159478435306,
      "loss": 1.5184,
      "theoretical_loss": 3.8658996055913297,
      "tokens_seen": 603717632
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000825295887662989,
      "loss": 1.6554,
      "theoretical_loss": 3.8658561880573217,
      "tokens_seen": 603783168
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008252758274824474,
      "loss": 1.6835,
      "theoretical_loss": 3.865812776555071,
      "tokens_seen": 603848704
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008252557673019058,
      "loss": 1.6259,
      "theoretical_loss": 3.8657693710830863,
      "tokens_seen": 603914240
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008252357071213641,
      "loss": 1.6972,
      "theoretical_loss": 3.8657259716398737,
      "tokens_seen": 603979776
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008252156469408225,
      "loss": 1.5604,
      "theoretical_loss": 3.865682578223944,
      "tokens_seen": 604045312
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008251955867602808,
      "loss": 1.6027,
      "theoretical_loss": 3.8656391908338046,
      "tokens_seen": 604110848
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008251755265797393,
      "loss": 1.6397,
      "theoretical_loss": 3.8655958094679663,
      "tokens_seen": 604176384
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008251554663991976,
      "loss": 1.5769,
      "theoretical_loss": 3.8655524341249388,
      "tokens_seen": 604241920
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000825135406218656,
      "loss": 1.5215,
      "theoretical_loss": 3.8655090648032333,
      "tokens_seen": 604307456
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008251153460381144,
      "loss": 1.7109,
      "theoretical_loss": 3.8654657015013605,
      "tokens_seen": 604372992
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008250952858575727,
      "loss": 1.7039,
      "theoretical_loss": 3.8654223442178317,
      "tokens_seen": 604438528
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008250752256770311,
      "loss": 1.6434,
      "theoretical_loss": 3.865378992951161,
      "tokens_seen": 604504064
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008250551654964896,
      "loss": 1.6853,
      "theoretical_loss": 3.865335647699859,
      "tokens_seen": 604569600
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008250351053159478,
      "loss": 1.6419,
      "theoretical_loss": 3.8652923084624415,
      "tokens_seen": 604635136
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008250150451354063,
      "loss": 1.577,
      "theoretical_loss": 3.86524897523742,
      "tokens_seen": 604700672
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008249949849548645,
      "loss": 1.5877,
      "theoretical_loss": 3.865205648023311,
      "tokens_seen": 604766208
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000824974924774323,
      "loss": 1.6295,
      "theoretical_loss": 3.865162326818628,
      "tokens_seen": 604831744
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008249548645937814,
      "loss": 1.6224,
      "theoretical_loss": 3.865119011621887,
      "tokens_seen": 604897280
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008249348044132397,
      "loss": 1.588,
      "theoretical_loss": 3.8650757024316045,
      "tokens_seen": 604962816
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008249147442326981,
      "loss": 1.6898,
      "theoretical_loss": 3.8650323992462963,
      "tokens_seen": 605028352
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008248946840521566,
      "loss": 1.4756,
      "theoretical_loss": 3.8649891020644795,
      "tokens_seen": 605093888
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008248746238716148,
      "loss": 1.6003,
      "theoretical_loss": 3.8649458108846724,
      "tokens_seen": 605159424
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008248545636910733,
      "loss": 1.7556,
      "theoretical_loss": 3.864902525705393,
      "tokens_seen": 605224960
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008248345035105315,
      "loss": 1.6128,
      "theoretical_loss": 3.8648592465251586,
      "tokens_seen": 605290496
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.00082481444332999,
      "loss": 1.6823,
      "theoretical_loss": 3.8648159733424903,
      "tokens_seen": 605356032
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008247943831494484,
      "loss": 1.6702,
      "theoretical_loss": 3.8647727061559065,
      "tokens_seen": 605421568
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008247743229689067,
      "loss": 1.586,
      "theoretical_loss": 3.8647294449639276,
      "tokens_seen": 605487104
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008247542627883651,
      "loss": 1.5767,
      "theoretical_loss": 3.864686189765075,
      "tokens_seen": 605552640
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008247342026078235,
      "loss": 1.7416,
      "theoretical_loss": 3.8646429405578697,
      "tokens_seen": 605618176
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008247141424272818,
      "loss": 1.6241,
      "theoretical_loss": 3.864599697340833,
      "tokens_seen": 605683712
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008246940822467403,
      "loss": 1.5612,
      "theoretical_loss": 3.864556460112488,
      "tokens_seen": 605749248
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008246740220661985,
      "loss": 1.6083,
      "theoretical_loss": 3.864513228871357,
      "tokens_seen": 605814784
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000824653961885657,
      "loss": 1.6448,
      "theoretical_loss": 3.8644700036159634,
      "tokens_seen": 605880320
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008246339017051153,
      "loss": 1.5935,
      "theoretical_loss": 3.8644267843448317,
      "tokens_seen": 605945856
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008246138415245737,
      "loss": 1.5513,
      "theoretical_loss": 3.8643835710564858,
      "tokens_seen": 606011392
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008245937813440321,
      "loss": 1.6612,
      "theoretical_loss": 3.8643403637494504,
      "tokens_seen": 606076928
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008245737211634905,
      "loss": 1.5263,
      "theoretical_loss": 3.8642971624222513,
      "tokens_seen": 606142464
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.49447712302207947,
      "objective/train/docs_used": 225863,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5234355926513672,
      "objective/train/original_loss": 1.5234355926513672,
      "objective/train/theoretical_loss": 3.8642539670734144,
      "objective/train/tokens_used": 626668000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2446836531162262,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0506914854049683,
      "objective/train/weighted_lm_loss": 1.6006100177764893,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0293704271316528,
      "theoretical_loss": 3.8642539670734144,
      "tokens_seen": 606208000
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008245536609829488,
      "loss": 1.5679,
      "theoretical_loss": 3.8642539670734144,
      "tokens_seen": 606208000
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008245336008024073,
      "loss": 1.6517,
      "theoretical_loss": 3.8642107777014667,
      "tokens_seen": 606273536
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008245135406218656,
      "loss": 1.6136,
      "theoretical_loss": 3.8641675943049343,
      "tokens_seen": 606339072
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000824493480441324,
      "loss": 1.6884,
      "theoretical_loss": 3.8641244168823454,
      "tokens_seen": 606404608
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008244734202607823,
      "loss": 1.633,
      "theoretical_loss": 3.8640812454322284,
      "tokens_seen": 606470144
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008244533600802407,
      "loss": 1.6995,
      "theoretical_loss": 3.864038079953111,
      "tokens_seen": 606535680
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008244332998996991,
      "loss": 1.6062,
      "theoretical_loss": 3.863994920443523,
      "tokens_seen": 606601216
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008244132397191575,
      "loss": 1.6901,
      "theoretical_loss": 3.8639517669019945,
      "tokens_seen": 606666752
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008243931795386158,
      "loss": 1.5642,
      "theoretical_loss": 3.863908619327054,
      "tokens_seen": 606732288
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008243731193580742,
      "loss": 1.6432,
      "theoretical_loss": 3.863865477717234,
      "tokens_seen": 606797824
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008243530591775326,
      "loss": 1.6101,
      "theoretical_loss": 3.8638223420710647,
      "tokens_seen": 606863360
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000824332998996991,
      "loss": 1.5649,
      "theoretical_loss": 3.8637792123870778,
      "tokens_seen": 606928896
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008243129388164494,
      "loss": 1.4683,
      "theoretical_loss": 3.863736088663806,
      "tokens_seen": 606994432
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008242928786359078,
      "loss": 1.6181,
      "theoretical_loss": 3.8636929708997814,
      "tokens_seen": 607059968
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008242728184553662,
      "loss": 1.6326,
      "theoretical_loss": 3.863649859093538,
      "tokens_seen": 607125504
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008242527582748245,
      "loss": 1.5195,
      "theoretical_loss": 3.8636067532436105,
      "tokens_seen": 607191040
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000824232698094283,
      "loss": 1.5985,
      "theoretical_loss": 3.863563653348531,
      "tokens_seen": 607256576
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008242126379137412,
      "loss": 1.6593,
      "theoretical_loss": 3.8635205594068354,
      "tokens_seen": 607322112
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008241925777331997,
      "loss": 1.564,
      "theoretical_loss": 3.863477471417059,
      "tokens_seen": 607387648
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000824172517552658,
      "loss": 1.5371,
      "theoretical_loss": 3.8634343893777383,
      "tokens_seen": 607453184
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008241524573721164,
      "loss": 1.6972,
      "theoretical_loss": 3.8633913132874094,
      "tokens_seen": 607518720
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008241323971915748,
      "loss": 1.6091,
      "theoretical_loss": 3.8633482431446082,
      "tokens_seen": 607584256
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008241123370110331,
      "loss": 1.7134,
      "theoretical_loss": 3.8633051789478734,
      "tokens_seen": 607649792
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008240922768304915,
      "loss": 1.6356,
      "theoretical_loss": 3.8632621206957425,
      "tokens_seen": 607715328
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.00082407221664995,
      "loss": 1.7405,
      "theoretical_loss": 3.863219068386754,
      "tokens_seen": 607780864
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008240521564694082,
      "loss": 1.6688,
      "theoretical_loss": 3.863176022019447,
      "tokens_seen": 607846400
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008240320962888667,
      "loss": 1.6639,
      "theoretical_loss": 3.8631329815923605,
      "tokens_seen": 607911936
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008240120361083249,
      "loss": 1.5745,
      "theoretical_loss": 3.8630899471040347,
      "tokens_seen": 607977472
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008239919759277834,
      "loss": 1.6226,
      "theoretical_loss": 3.863046918553011,
      "tokens_seen": 608043008
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008239719157472418,
      "loss": 1.483,
      "theoretical_loss": 3.8630038959378292,
      "tokens_seen": 608108544
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008239518555667001,
      "loss": 1.6049,
      "theoretical_loss": 3.862960879257032,
      "tokens_seen": 608174080
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008239317953861585,
      "loss": 1.5995,
      "theoretical_loss": 3.86291786850916,
      "tokens_seen": 608239616
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000823911735205617,
      "loss": 1.7919,
      "theoretical_loss": 3.862874863692758,
      "tokens_seen": 608305152
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008238916750250752,
      "loss": 1.629,
      "theoretical_loss": 3.862831864806367,
      "tokens_seen": 608370688
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008238716148445337,
      "loss": 1.6048,
      "theoretical_loss": 3.8627888718485313,
      "tokens_seen": 608436224
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008238515546639919,
      "loss": 1.4945,
      "theoretical_loss": 3.862745884817796,
      "tokens_seen": 608501760
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008238314944834504,
      "loss": 1.7123,
      "theoretical_loss": 3.8627029037127043,
      "tokens_seen": 608567296
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008238114343029088,
      "loss": 1.5196,
      "theoretical_loss": 3.862659928531802,
      "tokens_seen": 608632832
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008237913741223671,
      "loss": 1.6041,
      "theoretical_loss": 3.862616959273635,
      "tokens_seen": 608698368
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008237713139418255,
      "loss": 1.5851,
      "theoretical_loss": 3.8625739959367493,
      "tokens_seen": 608763904
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008237512537612839,
      "loss": 1.6814,
      "theoretical_loss": 3.862531038519692,
      "tokens_seen": 608829440
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008237311935807422,
      "loss": 1.5637,
      "theoretical_loss": 3.8624880870210094,
      "tokens_seen": 608894976
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008237111334002007,
      "loss": 1.5645,
      "theoretical_loss": 3.8624451414392498,
      "tokens_seen": 608960512
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008236910732196589,
      "loss": 1.6494,
      "theoretical_loss": 3.8624022017729622,
      "tokens_seen": 609026048
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008236710130391174,
      "loss": 1.6423,
      "theoretical_loss": 3.862359268020694,
      "tokens_seen": 609091584
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008236509528585757,
      "loss": 1.5312,
      "theoretical_loss": 3.8623163401809952,
      "tokens_seen": 609157120
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008236308926780341,
      "loss": 1.4567,
      "theoretical_loss": 3.8622734182524154,
      "tokens_seen": 609222656
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008236108324974925,
      "loss": 1.6388,
      "theoretical_loss": 3.862230502233505,
      "tokens_seen": 609288192
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008235907723169509,
      "loss": 1.5897,
      "theoretical_loss": 3.8621875921228153,
      "tokens_seen": 609353728
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008235707121364092,
      "loss": 1.6345,
      "theoretical_loss": 3.8621446879188963,
      "tokens_seen": 609419264
    },
    {
      "epoch": 0.18,
      "objective/train/advantage_avg": 0.49629247188568115,
      "objective/train/docs_used": 227108,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4768539667129517,
      "objective/train/original_loss": 1.4768540859222412,
      "objective/train/theoretical_loss": 3.8621017896203007,
      "objective/train/tokens_used": 629944800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24641580879688263,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.05088210105896,
      "objective/train/weighted_lm_loss": 1.5519256591796875,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 3.8621017896203007,
      "tokens_seen": 609484800
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008235506519558677,
      "loss": 1.5544,
      "theoretical_loss": 3.8621017896203007,
      "tokens_seen": 609484800
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000823530591775326,
      "loss": 1.6336,
      "theoretical_loss": 3.862058897225581,
      "tokens_seen": 609550336
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008235105315947844,
      "loss": 1.5458,
      "theoretical_loss": 3.8620160107332904,
      "tokens_seen": 609615872
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008234904714142427,
      "loss": 1.5553,
      "theoretical_loss": 3.861973130141981,
      "tokens_seen": 609681408
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008234704112337011,
      "loss": 1.5939,
      "theoretical_loss": 3.8619302554502077,
      "tokens_seen": 609746944
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008234503510531595,
      "loss": 1.6292,
      "theoretical_loss": 3.861887386656524,
      "tokens_seen": 609812480
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008234302908726179,
      "loss": 1.4184,
      "theoretical_loss": 3.861844523759486,
      "tokens_seen": 609878016
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008234102306920762,
      "loss": 1.6175,
      "theoretical_loss": 3.861801666757648,
      "tokens_seen": 609943552
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008233901705115346,
      "loss": 1.6142,
      "theoretical_loss": 3.8617588156495666,
      "tokens_seen": 610009088
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.000823370110330993,
      "loss": 1.7761,
      "theoretical_loss": 3.8617159704337984,
      "tokens_seen": 610074624
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008233500501504514,
      "loss": 1.5401,
      "theoretical_loss": 3.8616731311088994,
      "tokens_seen": 610140160
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008233299899699097,
      "loss": 1.5921,
      "theoretical_loss": 3.8616302976734276,
      "tokens_seen": 610205696
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008233099297893682,
      "loss": 1.7097,
      "theoretical_loss": 3.861587470125941,
      "tokens_seen": 610271232
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008232898696088264,
      "loss": 1.5588,
      "theoretical_loss": 3.8615446484649976,
      "tokens_seen": 610336768
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008232698094282849,
      "loss": 1.6388,
      "theoretical_loss": 3.861501832689157,
      "tokens_seen": 610402304
    },
    {
      "epoch": 0.18,
      "learning_rate": 0.0008232497492477432,
      "loss": 1.5997,
      "theoretical_loss": 3.861459022796978,
      "tokens_seen": 610467840
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008232296890672016,
      "loss": 1.6778,
      "theoretical_loss": 3.8614162187870216,
      "tokens_seen": 610533376
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.00082320962888666,
      "loss": 1.4831,
      "theoretical_loss": 3.861373420657847,
      "tokens_seen": 610598912
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008231895687061184,
      "loss": 1.7374,
      "theoretical_loss": 3.861330628408016,
      "tokens_seen": 610664448
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008231695085255767,
      "loss": 1.6045,
      "theoretical_loss": 3.8612878420360905,
      "tokens_seen": 610729984
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008231494483450352,
      "loss": 1.602,
      "theoretical_loss": 3.861245061540631,
      "tokens_seen": 610795520
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008231293881644934,
      "loss": 1.6448,
      "theoretical_loss": 3.861202286920202,
      "tokens_seen": 610861056
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008231093279839519,
      "loss": 1.6841,
      "theoretical_loss": 3.8611595181733644,
      "tokens_seen": 610926592
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008230892678034102,
      "loss": 1.5038,
      "theoretical_loss": 3.861116755298683,
      "tokens_seen": 610992128
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008230692076228686,
      "loss": 1.5688,
      "theoretical_loss": 3.8610739982947218,
      "tokens_seen": 611057664
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000823049147442327,
      "loss": 1.5498,
      "theoretical_loss": 3.861031247160045,
      "tokens_seen": 611123200
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008230290872617853,
      "loss": 1.5346,
      "theoretical_loss": 3.8609885018932175,
      "tokens_seen": 611188736
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008230090270812437,
      "loss": 1.5068,
      "theoretical_loss": 3.860945762492806,
      "tokens_seen": 611254272
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008229889669007022,
      "loss": 1.5697,
      "theoretical_loss": 3.8609030289573747,
      "tokens_seen": 611319808
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008229689067201604,
      "loss": 1.6114,
      "theoretical_loss": 3.8608603012854914,
      "tokens_seen": 611385344
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008229488465396189,
      "loss": 1.6014,
      "theoretical_loss": 3.860817579475723,
      "tokens_seen": 611450880
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008229287863590771,
      "loss": 1.6972,
      "theoretical_loss": 3.860774863526637,
      "tokens_seen": 611516416
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008229087261785356,
      "loss": 1.6748,
      "theoretical_loss": 3.8607321534368007,
      "tokens_seen": 611581952
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000822888665997994,
      "loss": 1.6369,
      "theoretical_loss": 3.860689449204784,
      "tokens_seen": 611647488
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008228686058174523,
      "loss": 1.5479,
      "theoretical_loss": 3.860646750829155,
      "tokens_seen": 611713024
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008228485456369107,
      "loss": 1.4678,
      "theoretical_loss": 3.8606040583084833,
      "tokens_seen": 611778560
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008228284854563692,
      "loss": 1.5103,
      "theoretical_loss": 3.8605613716413396,
      "tokens_seen": 611844096
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008228084252758275,
      "loss": 1.5572,
      "theoretical_loss": 3.860518690826294,
      "tokens_seen": 611909632
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008227883650952859,
      "loss": 1.7006,
      "theoretical_loss": 3.8604760158619174,
      "tokens_seen": 611975168
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008227683049147443,
      "loss": 1.5919,
      "theoretical_loss": 3.860433346746782,
      "tokens_seen": 612040704
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008227482447342026,
      "loss": 1.7025,
      "theoretical_loss": 3.860390683479459,
      "tokens_seen": 612106240
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008227281845536611,
      "loss": 1.4717,
      "theoretical_loss": 3.860348026058522,
      "tokens_seen": 612171776
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008227081243731193,
      "loss": 1.5342,
      "theoretical_loss": 3.8603053744825435,
      "tokens_seen": 612237312
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008226880641925778,
      "loss": 1.5627,
      "theoretical_loss": 3.8602627287500972,
      "tokens_seen": 612302848
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008226680040120361,
      "loss": 1.6263,
      "theoretical_loss": 3.860220088859757,
      "tokens_seen": 612368384
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008226479438314945,
      "loss": 1.6842,
      "theoretical_loss": 3.8601774548100973,
      "tokens_seen": 612433920
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008226278836509529,
      "loss": 1.3856,
      "theoretical_loss": 3.860134826599694,
      "tokens_seen": 612499456
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008226078234704113,
      "loss": 1.5841,
      "theoretical_loss": 3.860092204227122,
      "tokens_seen": 612564992
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008225877632898696,
      "loss": 1.5106,
      "theoretical_loss": 3.860049587690958,
      "tokens_seen": 612630528
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008225677031093281,
      "loss": 1.6423,
      "theoretical_loss": 3.860006976989778,
      "tokens_seen": 612696064
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.4955649971961975,
      "objective/train/docs_used": 228263,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6900759935379028,
      "objective/train/original_loss": 1.6900761127471924,
      "objective/train/theoretical_loss": 3.8599643721221586,
      "objective/train/tokens_used": 633221600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24580927193164825,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0508062839508057,
      "objective/train/weighted_lm_loss": 1.7758455276489258,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0281672477722168,
      "theoretical_loss": 3.8599643721221586,
      "tokens_seen": 612761600
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008225476429287864,
      "loss": 1.6875,
      "theoretical_loss": 3.8599643721221586,
      "tokens_seen": 612761600
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008225275827482448,
      "loss": 1.4918,
      "theoretical_loss": 3.859921773086679,
      "tokens_seen": 612827136
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008225075225677031,
      "loss": 1.5538,
      "theoretical_loss": 3.8598791798819154,
      "tokens_seen": 612892672
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008224874623871615,
      "loss": 1.4891,
      "theoretical_loss": 3.859836592506448,
      "tokens_seen": 612958208
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008224674022066199,
      "loss": 1.5989,
      "theoretical_loss": 3.8597940109588547,
      "tokens_seen": 613023744
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008224473420260783,
      "loss": 1.6383,
      "theoretical_loss": 3.859751435237716,
      "tokens_seen": 613089280
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008224272818455366,
      "loss": 1.4104,
      "theoretical_loss": 3.859708865341611,
      "tokens_seen": 613154816
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000822407221664995,
      "loss": 1.6049,
      "theoretical_loss": 3.8596663012691206,
      "tokens_seen": 613220352
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008223871614844534,
      "loss": 1.6655,
      "theoretical_loss": 3.859623743018827,
      "tokens_seen": 613285888
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008223671013039118,
      "loss": 1.6336,
      "theoretical_loss": 3.85958119058931,
      "tokens_seen": 613351424
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008223470411233701,
      "loss": 1.6416,
      "theoretical_loss": 3.8595386439791532,
      "tokens_seen": 613416960
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008223269809428286,
      "loss": 1.5586,
      "theoretical_loss": 3.859496103186938,
      "tokens_seen": 613482496
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008223069207622868,
      "loss": 1.6274,
      "theoretical_loss": 3.859453568211248,
      "tokens_seen": 613548032
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008222868605817453,
      "loss": 1.6393,
      "theoretical_loss": 3.8594110390506664,
      "tokens_seen": 613613568
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008222668004012036,
      "loss": 1.5215,
      "theoretical_loss": 3.859368515703778,
      "tokens_seen": 613679104
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000822246740220662,
      "loss": 1.4824,
      "theoretical_loss": 3.8593259981691665,
      "tokens_seen": 613744640
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008222266800401204,
      "loss": 1.5944,
      "theoretical_loss": 3.8592834864454177,
      "tokens_seen": 613810176
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008222066198595788,
      "loss": 1.5089,
      "theoretical_loss": 3.8592409805311165,
      "tokens_seen": 613875712
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008221865596790371,
      "loss": 1.6413,
      "theoretical_loss": 3.859198480424849,
      "tokens_seen": 613941248
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008221664994984956,
      "loss": 1.5523,
      "theoretical_loss": 3.859155986125202,
      "tokens_seen": 614006784
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008221464393179538,
      "loss": 1.6317,
      "theoretical_loss": 3.859113497630762,
      "tokens_seen": 614072320
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008221263791374123,
      "loss": 1.6131,
      "theoretical_loss": 3.8590710149401177,
      "tokens_seen": 614137856
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008221063189568706,
      "loss": 1.4801,
      "theoretical_loss": 3.859028538051856,
      "tokens_seen": 614203392
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000822086258776329,
      "loss": 1.7205,
      "theoretical_loss": 3.8589860669645653,
      "tokens_seen": 614268928
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008220661985957874,
      "loss": 1.6196,
      "theoretical_loss": 3.8589436016768355,
      "tokens_seen": 614334464
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008220461384152457,
      "loss": 1.5722,
      "theoretical_loss": 3.858901142187256,
      "tokens_seen": 614400000
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008220260782347041,
      "loss": 1.4802,
      "theoretical_loss": 3.858858688494416,
      "tokens_seen": 614465536
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008220060180541626,
      "loss": 1.6181,
      "theoretical_loss": 3.8588162405969064,
      "tokens_seen": 614531072
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008219859578736208,
      "loss": 1.473,
      "theoretical_loss": 3.8587737984933184,
      "tokens_seen": 614596608
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008219658976930793,
      "loss": 1.6362,
      "theoretical_loss": 3.8587313621822434,
      "tokens_seen": 614662144
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008219458375125375,
      "loss": 1.5643,
      "theoretical_loss": 3.8586889316622726,
      "tokens_seen": 614727680
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000821925777331996,
      "loss": 1.5142,
      "theoretical_loss": 3.8586465069319997,
      "tokens_seen": 614793216
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008219057171514544,
      "loss": 1.4805,
      "theoretical_loss": 3.8586040879900168,
      "tokens_seen": 614858752
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008218856569709127,
      "loss": 1.4964,
      "theoretical_loss": 3.8585616748349167,
      "tokens_seen": 614924288
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008218655967903711,
      "loss": 1.5681,
      "theoretical_loss": 3.8585192674652955,
      "tokens_seen": 614989824
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008218455366098296,
      "loss": 1.4802,
      "theoretical_loss": 3.8584768658797453,
      "tokens_seen": 615055360
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008218254764292878,
      "loss": 1.6116,
      "theoretical_loss": 3.8584344700768622,
      "tokens_seen": 615120896
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008218054162487463,
      "loss": 1.5727,
      "theoretical_loss": 3.858392080055242,
      "tokens_seen": 615186432
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008217853560682046,
      "loss": 1.6323,
      "theoretical_loss": 3.8583496958134793,
      "tokens_seen": 615251968
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000821765295887663,
      "loss": 1.5314,
      "theoretical_loss": 3.858307317350171,
      "tokens_seen": 615317504
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008217452357071214,
      "loss": 1.4508,
      "theoretical_loss": 3.8582649446639143,
      "tokens_seen": 615383040
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008217251755265797,
      "loss": 1.541,
      "theoretical_loss": 3.858222577753307,
      "tokens_seen": 615448576
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008217051153460381,
      "loss": 1.6024,
      "theoretical_loss": 3.8581802166169457,
      "tokens_seen": 615514112
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008216850551654965,
      "loss": 1.6622,
      "theoretical_loss": 3.8581378612534296,
      "tokens_seen": 615579648
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008216649949849548,
      "loss": 1.5519,
      "theoretical_loss": 3.8580955116613573,
      "tokens_seen": 615645184
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008216449348044133,
      "loss": 1.5248,
      "theoretical_loss": 3.858053167839328,
      "tokens_seen": 615710720
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008216248746238716,
      "loss": 1.6547,
      "theoretical_loss": 3.8580108297859415,
      "tokens_seen": 615776256
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.00082160481444333,
      "loss": 1.5183,
      "theoretical_loss": 3.8579684974997983,
      "tokens_seen": 615841792
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008215847542627884,
      "loss": 1.4812,
      "theoretical_loss": 3.8579261709794994,
      "tokens_seen": 615907328
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008215646940822468,
      "loss": 1.5726,
      "theoretical_loss": 3.8578838502236454,
      "tokens_seen": 615972864
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.492830753326416,
      "objective/train/docs_used": 229528,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.466249704360962,
      "objective/train/original_loss": 1.466249704360962,
      "objective/train/theoretical_loss": 3.857841535230839,
      "objective/train/tokens_used": 636498400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24335187673568726,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505203008651733,
      "objective/train/weighted_lm_loss": 1.5402671098709106,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.9048373699188232,
      "theoretical_loss": 3.857841535230839,
      "tokens_seen": 616038400
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008215446339017051,
      "loss": 1.608,
      "theoretical_loss": 3.857841535230839,
      "tokens_seen": 616038400
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008215245737211635,
      "loss": 1.5664,
      "theoretical_loss": 3.8577992259996816,
      "tokens_seen": 616103936
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008215045135406218,
      "loss": 1.5859,
      "theoretical_loss": 3.8577569225287762,
      "tokens_seen": 616169472
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008214844533600803,
      "loss": 1.558,
      "theoretical_loss": 3.857714624816726,
      "tokens_seen": 616235008
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008214643931795386,
      "loss": 1.6013,
      "theoretical_loss": 3.8576723328621347,
      "tokens_seen": 616300544
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000821444332998997,
      "loss": 1.6573,
      "theoretical_loss": 3.8576300466636066,
      "tokens_seen": 616366080
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008214242728184553,
      "loss": 1.5305,
      "theoretical_loss": 3.857587766219747,
      "tokens_seen": 616431616
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008214042126379138,
      "loss": 1.5478,
      "theoretical_loss": 3.8575454915291596,
      "tokens_seen": 616497152
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008213841524573721,
      "loss": 1.6022,
      "theoretical_loss": 3.8575032225904513,
      "tokens_seen": 616562688
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008213640922768305,
      "loss": 1.5505,
      "theoretical_loss": 3.8574609594022276,
      "tokens_seen": 616628224
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008213440320962888,
      "loss": 1.7057,
      "theoretical_loss": 3.8574187019630957,
      "tokens_seen": 616693760
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008213239719157472,
      "loss": 1.4051,
      "theoretical_loss": 3.857376450271662,
      "tokens_seen": 616759296
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008213039117352057,
      "loss": 1.5181,
      "theoretical_loss": 3.8573342043265346,
      "tokens_seen": 616824832
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000821283851554664,
      "loss": 1.6967,
      "theoretical_loss": 3.857291964126322,
      "tokens_seen": 616890368
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008212637913741224,
      "loss": 1.7542,
      "theoretical_loss": 3.8572497296696313,
      "tokens_seen": 616955904
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008212437311935808,
      "loss": 1.5992,
      "theoretical_loss": 3.857207500955073,
      "tokens_seen": 617021440
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008212236710130392,
      "loss": 1.6053,
      "theoretical_loss": 3.857165277981256,
      "tokens_seen": 617086976
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008212036108324975,
      "loss": 1.6847,
      "theoretical_loss": 3.8571230607467912,
      "tokens_seen": 617152512
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000821183550651956,
      "loss": 1.5465,
      "theoretical_loss": 3.8570808492502877,
      "tokens_seen": 617218048
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008211634904714142,
      "loss": 1.6344,
      "theoretical_loss": 3.8570386434903567,
      "tokens_seen": 617283584
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008211434302908727,
      "loss": 1.5999,
      "theoretical_loss": 3.8569964434656105,
      "tokens_seen": 617349120
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000821123370110331,
      "loss": 1.6031,
      "theoretical_loss": 3.8569542491746613,
      "tokens_seen": 617414656
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008211033099297894,
      "loss": 1.5452,
      "theoretical_loss": 3.8569120606161205,
      "tokens_seen": 617480192
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008210832497492478,
      "loss": 1.6047,
      "theoretical_loss": 3.8568698777886015,
      "tokens_seen": 617545728
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008210631895687061,
      "loss": 1.7153,
      "theoretical_loss": 3.856827700690718,
      "tokens_seen": 617611264
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008210431293881645,
      "loss": 1.574,
      "theoretical_loss": 3.856785529321083,
      "tokens_seen": 617676800
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000821023069207623,
      "loss": 1.6403,
      "theoretical_loss": 3.8567433636783113,
      "tokens_seen": 617742336
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008210030090270812,
      "loss": 1.637,
      "theoretical_loss": 3.8567012037610184,
      "tokens_seen": 617807872
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008209829488465397,
      "loss": 1.493,
      "theoretical_loss": 3.8566590495678192,
      "tokens_seen": 617873408
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008209628886659979,
      "loss": 1.6298,
      "theoretical_loss": 3.8566169010973295,
      "tokens_seen": 617938944
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008209428284854564,
      "loss": 1.6373,
      "theoretical_loss": 3.856574758348165,
      "tokens_seen": 618004480
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008209227683049148,
      "loss": 1.6059,
      "theoretical_loss": 3.8565326213189435,
      "tokens_seen": 618070016
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008209027081243731,
      "loss": 1.6655,
      "theoretical_loss": 3.856490490008282,
      "tokens_seen": 618135552
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008208826479438315,
      "loss": 1.5592,
      "theoretical_loss": 3.8564483644147978,
      "tokens_seen": 618201088
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.00082086258776329,
      "loss": 1.7345,
      "theoretical_loss": 3.8564062445371095,
      "tokens_seen": 618266624
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008208425275827482,
      "loss": 1.6147,
      "theoretical_loss": 3.8563641303738354,
      "tokens_seen": 618332160
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008208224674022067,
      "loss": 1.5514,
      "theoretical_loss": 3.856322021923595,
      "tokens_seen": 618397696
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000820802407221665,
      "loss": 1.6845,
      "theoretical_loss": 3.856279919185008,
      "tokens_seen": 618463232
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008207823470411234,
      "loss": 1.5381,
      "theoretical_loss": 3.856237822156695,
      "tokens_seen": 618528768
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008207622868605818,
      "loss": 1.5259,
      "theoretical_loss": 3.8561957308372756,
      "tokens_seen": 618594304
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008207422266800401,
      "loss": 1.6203,
      "theoretical_loss": 3.8561536452253713,
      "tokens_seen": 618659840
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008207221664994985,
      "loss": 1.5652,
      "theoretical_loss": 3.856111565319604,
      "tokens_seen": 618725376
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008207021063189569,
      "loss": 1.6238,
      "theoretical_loss": 3.8560694911185953,
      "tokens_seen": 618790912
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008206820461384152,
      "loss": 1.4398,
      "theoretical_loss": 3.8560274226209685,
      "tokens_seen": 618856448
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008206619859578737,
      "loss": 1.5396,
      "theoretical_loss": 3.855985359825346,
      "tokens_seen": 618921984
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000820641925777332,
      "loss": 1.697,
      "theoretical_loss": 3.8559433027303514,
      "tokens_seen": 618987520
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008206218655967904,
      "loss": 1.6485,
      "theoretical_loss": 3.8559012513346085,
      "tokens_seen": 619053056
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008206018054162488,
      "loss": 1.6622,
      "theoretical_loss": 3.855859205636742,
      "tokens_seen": 619118592
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008205817452357072,
      "loss": 1.6001,
      "theoretical_loss": 3.855817165635377,
      "tokens_seen": 619184128
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008205616850551655,
      "loss": 1.532,
      "theoretical_loss": 3.855775131329138,
      "tokens_seen": 619249664
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.4949982166290283,
      "objective/train/docs_used": 230719,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6773806810379028,
      "objective/train/original_loss": 1.6773808002471924,
      "objective/train/theoretical_loss": 3.855733102716653,
      "objective/train/tokens_used": 639775200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24520695209503174,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507465600967407,
      "objective/train/weighted_lm_loss": 1.7623634338378906,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.855733102716653,
      "tokens_seen": 619315200
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008205416248746239,
      "loss": 1.7109,
      "theoretical_loss": 3.855733102716653,
      "tokens_seen": 619315200
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008205215646940822,
      "loss": 1.5925,
      "theoretical_loss": 3.8556910797965456,
      "tokens_seen": 619380736
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008205015045135407,
      "loss": 1.7159,
      "theoretical_loss": 3.8556490625674447,
      "tokens_seen": 619446272
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000820481444332999,
      "loss": 1.7182,
      "theoretical_loss": 3.8556070510279765,
      "tokens_seen": 619511808
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008204613841524574,
      "loss": 1.6026,
      "theoretical_loss": 3.8555650451767693,
      "tokens_seen": 619577344
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008204413239719157,
      "loss": 1.68,
      "theoretical_loss": 3.8555230450124514,
      "tokens_seen": 619642880
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008204212637913742,
      "loss": 1.5025,
      "theoretical_loss": 3.855481050533651,
      "tokens_seen": 619708416
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008204012036108325,
      "loss": 1.5698,
      "theoretical_loss": 3.855439061738998,
      "tokens_seen": 619773952
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008203811434302909,
      "loss": 1.6344,
      "theoretical_loss": 3.8553970786271217,
      "tokens_seen": 619839488
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008203610832497492,
      "loss": 1.6571,
      "theoretical_loss": 3.8553551011966523,
      "tokens_seen": 619905024
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008203410230692076,
      "loss": 1.6364,
      "theoretical_loss": 3.8553131294462206,
      "tokens_seen": 619970560
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000820320962888666,
      "loss": 1.6088,
      "theoretical_loss": 3.855271163374458,
      "tokens_seen": 620036096
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008203009027081244,
      "loss": 1.6577,
      "theoretical_loss": 3.855229202979995,
      "tokens_seen": 620101632
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008202808425275827,
      "loss": 1.6194,
      "theoretical_loss": 3.8551872482614655,
      "tokens_seen": 620167168
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008202607823470412,
      "loss": 1.6074,
      "theoretical_loss": 3.8551452992175,
      "tokens_seen": 620232704
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008202407221664995,
      "loss": 1.6324,
      "theoretical_loss": 3.8551033558467327,
      "tokens_seen": 620298240
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008202206619859579,
      "loss": 1.6571,
      "theoretical_loss": 3.8550614181477965,
      "tokens_seen": 620363776
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008202006018054163,
      "loss": 1.6409,
      "theoretical_loss": 3.8550194861193265,
      "tokens_seen": 620429312
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008201805416248746,
      "loss": 1.5578,
      "theoretical_loss": 3.8549775597599556,
      "tokens_seen": 620494848
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000820160481444333,
      "loss": 1.601,
      "theoretical_loss": 3.8549356390683203,
      "tokens_seen": 620560384
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008201404212637914,
      "loss": 1.5341,
      "theoretical_loss": 3.8548937240430545,
      "tokens_seen": 620625920
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008201203610832497,
      "loss": 1.613,
      "theoretical_loss": 3.854851814682795,
      "tokens_seen": 620691456
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008201003009027082,
      "loss": 1.5435,
      "theoretical_loss": 3.8548099109861775,
      "tokens_seen": 620756992
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008200802407221664,
      "loss": 1.5476,
      "theoretical_loss": 3.8547680129518396,
      "tokens_seen": 620822528
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008200601805416249,
      "loss": 1.6422,
      "theoretical_loss": 3.854726120578418,
      "tokens_seen": 620888064
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008200401203610833,
      "loss": 1.7115,
      "theoretical_loss": 3.8546842338645497,
      "tokens_seen": 620953600
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008200200601805416,
      "loss": 1.5755,
      "theoretical_loss": 3.8546423528088747,
      "tokens_seen": 621019136
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.00082,
      "loss": 1.7127,
      "theoretical_loss": 3.85460047741003,
      "tokens_seen": 621084672
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008199799398194583,
      "loss": 1.5826,
      "theoretical_loss": 3.8545586076666565,
      "tokens_seen": 621150208
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008199598796389167,
      "loss": 1.5554,
      "theoretical_loss": 3.854516743577392,
      "tokens_seen": 621215744
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008199398194583752,
      "loss": 1.5614,
      "theoretical_loss": 3.8544748851408777,
      "tokens_seen": 621281280
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008199197592778334,
      "loss": 1.6355,
      "theoretical_loss": 3.854433032355754,
      "tokens_seen": 621346816
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008198996990972919,
      "loss": 1.4199,
      "theoretical_loss": 3.8543911852206616,
      "tokens_seen": 621412352
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008198796389167503,
      "loss": 1.6398,
      "theoretical_loss": 3.8543493437342424,
      "tokens_seen": 621477888
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008198595787362086,
      "loss": 1.5345,
      "theoretical_loss": 3.8543075078951388,
      "tokens_seen": 621543424
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000819839518555667,
      "loss": 1.5795,
      "theoretical_loss": 3.8542656777019917,
      "tokens_seen": 621608960
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008198194583751254,
      "loss": 1.5518,
      "theoretical_loss": 3.8542238531534454,
      "tokens_seen": 621674496
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008197993981945838,
      "loss": 1.6004,
      "theoretical_loss": 3.8541820342481428,
      "tokens_seen": 621740032
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008197793380140422,
      "loss": 1.5146,
      "theoretical_loss": 3.8541402209847284,
      "tokens_seen": 621805568
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008197592778335005,
      "loss": 1.5916,
      "theoretical_loss": 3.8540984133618448,
      "tokens_seen": 621871104
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008197392176529589,
      "loss": 1.7015,
      "theoretical_loss": 3.854056611378139,
      "tokens_seen": 621936640
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008197191574724173,
      "loss": 1.5986,
      "theoretical_loss": 3.854014815032255,
      "tokens_seen": 622002176
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008196990972918756,
      "loss": 1.6376,
      "theoretical_loss": 3.8539730243228387,
      "tokens_seen": 622067712
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008196790371113341,
      "loss": 1.5778,
      "theoretical_loss": 3.853931239248536,
      "tokens_seen": 622133248
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008196589769307924,
      "loss": 1.6422,
      "theoretical_loss": 3.8538894598079945,
      "tokens_seen": 622198784
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008196389167502508,
      "loss": 1.5142,
      "theoretical_loss": 3.8538476859998605,
      "tokens_seen": 622264320
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008196188565697092,
      "loss": 1.4573,
      "theoretical_loss": 3.8538059178227817,
      "tokens_seen": 622329856
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008195987963891676,
      "loss": 1.7258,
      "theoretical_loss": 3.8537641552754067,
      "tokens_seen": 622395392
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008195787362086259,
      "loss": 1.7015,
      "theoretical_loss": 3.853722398356383,
      "tokens_seen": 622460928
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008195586760280843,
      "loss": 1.6101,
      "theoretical_loss": 3.853680647064361,
      "tokens_seen": 622526464
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.4887832999229431,
      "objective/train/docs_used": 231894,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.6082185506820679,
      "objective/train/original_loss": 1.6082186698913574,
      "objective/train/theoretical_loss": 3.8536389013979893,
      "objective/train/tokens_used": 643052000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.23972667753696442,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0500969886779785,
      "objective/train/weighted_lm_loss": 1.6888190507888794,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8536389013979893,
      "tokens_seen": 622592000
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008195386158475426,
      "loss": 1.6239,
      "theoretical_loss": 3.8536389013979893,
      "tokens_seen": 622592000
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008195185556670011,
      "loss": 1.6087,
      "theoretical_loss": 3.8535971613559177,
      "tokens_seen": 622657536
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008194984954864594,
      "loss": 1.6609,
      "theoretical_loss": 3.8535554269367966,
      "tokens_seen": 622723072
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008194784353059178,
      "loss": 1.4851,
      "theoretical_loss": 3.8535136981392775,
      "tokens_seen": 622788608
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008194583751253761,
      "loss": 1.5719,
      "theoretical_loss": 3.8534719749620114,
      "tokens_seen": 622854144
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008194383149448346,
      "loss": 1.6139,
      "theoretical_loss": 3.85343025740365,
      "tokens_seen": 622919680
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008194182547642929,
      "loss": 1.4939,
      "theoretical_loss": 3.8533885454628454,
      "tokens_seen": 622985216
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008193981945837513,
      "loss": 1.5971,
      "theoretical_loss": 3.8533468391382515,
      "tokens_seen": 623050752
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008193781344032096,
      "loss": 1.5851,
      "theoretical_loss": 3.8533051384285195,
      "tokens_seen": 623116288
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000819358074222668,
      "loss": 1.554,
      "theoretical_loss": 3.8532634433323043,
      "tokens_seen": 623181824
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008193380140421264,
      "loss": 1.8111,
      "theoretical_loss": 3.85322175384826,
      "tokens_seen": 623247360
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008193179538615848,
      "loss": 1.5707,
      "theoretical_loss": 3.853180069975041,
      "tokens_seen": 623312896
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008192978936810431,
      "loss": 1.5713,
      "theoretical_loss": 3.853138391711303,
      "tokens_seen": 623378432
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008192778335005016,
      "loss": 1.564,
      "theoretical_loss": 3.8530967190556997,
      "tokens_seen": 623443968
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008192577733199599,
      "loss": 1.5382,
      "theoretical_loss": 3.853055052006889,
      "tokens_seen": 623509504
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008192377131394183,
      "loss": 1.5818,
      "theoretical_loss": 3.853013390563526,
      "tokens_seen": 623575040
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008192176529588767,
      "loss": 1.6479,
      "theoretical_loss": 3.852971734724269,
      "tokens_seen": 623640576
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000819197592778335,
      "loss": 1.5934,
      "theoretical_loss": 3.852930084487774,
      "tokens_seen": 623706112
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008191775325977934,
      "loss": 1.5378,
      "theoretical_loss": 3.8528884398526992,
      "tokens_seen": 623771648
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008191574724172518,
      "loss": 1.5815,
      "theoretical_loss": 3.852846800817704,
      "tokens_seen": 623837184
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008191374122367101,
      "loss": 1.5563,
      "theoretical_loss": 3.8528051673814456,
      "tokens_seen": 623902720
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008191173520561686,
      "loss": 1.5153,
      "theoretical_loss": 3.852763539542584,
      "tokens_seen": 623968256
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008190972918756268,
      "loss": 1.6816,
      "theoretical_loss": 3.8527219172997786,
      "tokens_seen": 624033792
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008190772316950853,
      "loss": 1.6855,
      "theoretical_loss": 3.8526803006516896,
      "tokens_seen": 624099328
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008190571715145437,
      "loss": 1.5917,
      "theoretical_loss": 3.8526386895969775,
      "tokens_seen": 624164864
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000819037111334002,
      "loss": 1.6849,
      "theoretical_loss": 3.8525970841343042,
      "tokens_seen": 624230400
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008190170511534604,
      "loss": 1.5308,
      "theoretical_loss": 3.85255548426233,
      "tokens_seen": 624295936
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008189969909729187,
      "loss": 1.7186,
      "theoretical_loss": 3.8525138899797176,
      "tokens_seen": 624361472
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008189769307923771,
      "loss": 1.6299,
      "theoretical_loss": 3.8524723012851294,
      "tokens_seen": 624427008
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008189568706118356,
      "loss": 1.7516,
      "theoretical_loss": 3.8524307181772284,
      "tokens_seen": 624492544
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008189368104312938,
      "loss": 1.5709,
      "theoretical_loss": 3.8523891406546773,
      "tokens_seen": 624558080
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008189167502507523,
      "loss": 1.5958,
      "theoretical_loss": 3.8523475687161404,
      "tokens_seen": 624623616
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008188966900702107,
      "loss": 1.545,
      "theoretical_loss": 3.852306002360282,
      "tokens_seen": 624689152
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000818876629889669,
      "loss": 1.6013,
      "theoretical_loss": 3.852264441585767,
      "tokens_seen": 624754688
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008188565697091274,
      "loss": 1.6479,
      "theoretical_loss": 3.85222288639126,
      "tokens_seen": 624820224
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008188365095285858,
      "loss": 1.624,
      "theoretical_loss": 3.852181336775428,
      "tokens_seen": 624885760
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008188164493480441,
      "loss": 1.6374,
      "theoretical_loss": 3.852139792736936,
      "tokens_seen": 624951296
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008187963891675026,
      "loss": 1.4283,
      "theoretical_loss": 3.85209825427445,
      "tokens_seen": 625016832
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008187763289869608,
      "loss": 1.513,
      "theoretical_loss": 3.8520567213866386,
      "tokens_seen": 625082368
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008187562688064193,
      "loss": 1.7715,
      "theoretical_loss": 3.852015194072168,
      "tokens_seen": 625147904
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008187362086258776,
      "loss": 1.5732,
      "theoretical_loss": 3.8519736723297067,
      "tokens_seen": 625213440
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000818716148445336,
      "loss": 1.6342,
      "theoretical_loss": 3.8519321561579236,
      "tokens_seen": 625278976
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008186960882647944,
      "loss": 1.555,
      "theoretical_loss": 3.8518906455554864,
      "tokens_seen": 625344512
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008186760280842528,
      "loss": 1.54,
      "theoretical_loss": 3.8518491405210655,
      "tokens_seen": 625410048
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008186559679037111,
      "loss": 1.6034,
      "theoretical_loss": 3.8518076410533304,
      "tokens_seen": 625475584
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008186359077231696,
      "loss": 1.5722,
      "theoretical_loss": 3.8517661471509506,
      "tokens_seen": 625541120
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008186158475426278,
      "loss": 1.6391,
      "theoretical_loss": 3.851724658812598,
      "tokens_seen": 625606656
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008185957873620863,
      "loss": 1.5547,
      "theoretical_loss": 3.851683176036943,
      "tokens_seen": 625672192
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008185757271815446,
      "loss": 1.5753,
      "theoretical_loss": 3.8516416988226574,
      "tokens_seen": 625737728
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000818555667001003,
      "loss": 1.6727,
      "theoretical_loss": 3.8516002271684133,
      "tokens_seen": 625803264
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.4916975498199463,
      "objective/train/docs_used": 233149,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.3925726413726807,
      "objective/train/original_loss": 1.3925726413726807,
      "objective/train/theoretical_loss": 3.8515587610728836,
      "objective/train/tokens_used": 646328800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2422853410243988,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0504014492034912,
      "objective/train/weighted_lm_loss": 1.4627898931503296,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8515587610728836,
      "tokens_seen": 625868800
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008185356068204614,
      "loss": 1.5648,
      "theoretical_loss": 3.8515587610728836,
      "tokens_seen": 625868800
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008185155466399198,
      "loss": 1.5074,
      "theoretical_loss": 3.8515173005347405,
      "tokens_seen": 625934336
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008184954864593781,
      "loss": 1.6098,
      "theoretical_loss": 3.851475845552658,
      "tokens_seen": 625999872
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008184754262788365,
      "loss": 1.6126,
      "theoretical_loss": 3.8514343961253092,
      "tokens_seen": 626065408
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008184553660982949,
      "loss": 1.6273,
      "theoretical_loss": 3.851392952251369,
      "tokens_seen": 626130944
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008184353059177533,
      "loss": 1.6988,
      "theoretical_loss": 3.8513515139295134,
      "tokens_seen": 626196480
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008184152457372116,
      "loss": 1.6243,
      "theoretical_loss": 3.8513100811584158,
      "tokens_seen": 626262016
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.00081839518555667,
      "loss": 1.5949,
      "theoretical_loss": 3.851268653936753,
      "tokens_seen": 626327552
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008183751253761283,
      "loss": 1.6994,
      "theoretical_loss": 3.8512272322631995,
      "tokens_seen": 626393088
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008183550651955868,
      "loss": 1.5009,
      "theoretical_loss": 3.8511858161364345,
      "tokens_seen": 626458624
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008183350050150451,
      "loss": 1.775,
      "theoretical_loss": 3.851144405555134,
      "tokens_seen": 626524160
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008183149448345035,
      "loss": 1.5517,
      "theoretical_loss": 3.8511030005179743,
      "tokens_seen": 626589696
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000818294884653962,
      "loss": 1.7334,
      "theoretical_loss": 3.851061601023635,
      "tokens_seen": 626655232
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008182748244734203,
      "loss": 1.5323,
      "theoretical_loss": 3.851020207070794,
      "tokens_seen": 626720768
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008182547642928787,
      "loss": 1.531,
      "theoretical_loss": 3.85097881865813,
      "tokens_seen": 626786304
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000818234704112337,
      "loss": 1.5273,
      "theoretical_loss": 3.8509374357843225,
      "tokens_seen": 626851840
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008182146439317954,
      "loss": 1.6216,
      "theoretical_loss": 3.8508960584480514,
      "tokens_seen": 626917376
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008181945837512538,
      "loss": 1.5957,
      "theoretical_loss": 3.8508546866479962,
      "tokens_seen": 626982912
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008181745235707122,
      "loss": 1.7507,
      "theoretical_loss": 3.850813320382839,
      "tokens_seen": 627048448
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008181544633901705,
      "loss": 1.5464,
      "theoretical_loss": 3.85077195965126,
      "tokens_seen": 627113984
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000818134403209629,
      "loss": 1.6804,
      "theoretical_loss": 3.850730604451941,
      "tokens_seen": 627179520
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008181143430290872,
      "loss": 1.704,
      "theoretical_loss": 3.850689254783564,
      "tokens_seen": 627245056
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008180942828485457,
      "loss": 1.5882,
      "theoretical_loss": 3.8506479106448115,
      "tokens_seen": 627310592
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008180742226680041,
      "loss": 1.6426,
      "theoretical_loss": 3.8506065720343665,
      "tokens_seen": 627376128
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008180541624874624,
      "loss": 1.6072,
      "theoretical_loss": 3.850565238950913,
      "tokens_seen": 627441664
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008180341023069208,
      "loss": 1.734,
      "theoretical_loss": 3.8505239113931333,
      "tokens_seen": 627507200
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008180140421263791,
      "loss": 1.6028,
      "theoretical_loss": 3.8504825893597134,
      "tokens_seen": 627572736
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008179939819458375,
      "loss": 1.7099,
      "theoretical_loss": 3.850441272849337,
      "tokens_seen": 627638272
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000817973921765296,
      "loss": 1.4988,
      "theoretical_loss": 3.85039996186069,
      "tokens_seen": 627703808
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008179538615847542,
      "loss": 1.549,
      "theoretical_loss": 3.8503586563924577,
      "tokens_seen": 627769344
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008179338014042127,
      "loss": 1.6557,
      "theoretical_loss": 3.850317356443326,
      "tokens_seen": 627834880
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008179137412236711,
      "loss": 1.5299,
      "theoretical_loss": 3.8502760620119822,
      "tokens_seen": 627900416
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008178936810431294,
      "loss": 1.6339,
      "theoretical_loss": 3.850234773097112,
      "tokens_seen": 627965952
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008178736208625878,
      "loss": 1.5002,
      "theoretical_loss": 3.8501934896974044,
      "tokens_seen": 628031488
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008178535606820462,
      "loss": 1.6185,
      "theoretical_loss": 3.8501522118115465,
      "tokens_seen": 628097024
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008178335005015045,
      "loss": 1.7303,
      "theoretical_loss": 3.8501109394382267,
      "tokens_seen": 628162560
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000817813440320963,
      "loss": 1.6692,
      "theoretical_loss": 3.850069672576134,
      "tokens_seen": 628228096
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008177933801404212,
      "loss": 1.6016,
      "theoretical_loss": 3.850028411223958,
      "tokens_seen": 628293632
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008177733199598797,
      "loss": 1.6357,
      "theoretical_loss": 3.8499871553803873,
      "tokens_seen": 628359168
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000817753259779338,
      "loss": 1.516,
      "theoretical_loss": 3.8499459050441125,
      "tokens_seen": 628424704
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008177331995987964,
      "loss": 1.626,
      "theoretical_loss": 3.849904660213825,
      "tokens_seen": 628490240
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008177131394182548,
      "loss": 1.6112,
      "theoretical_loss": 3.849863420888215,
      "tokens_seen": 628555776
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008176930792377132,
      "loss": 1.6741,
      "theoretical_loss": 3.8498221870659743,
      "tokens_seen": 628621312
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008176730190571715,
      "loss": 1.5285,
      "theoretical_loss": 3.8497809587457947,
      "tokens_seen": 628686848
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.00081765295887663,
      "loss": 1.4918,
      "theoretical_loss": 3.849739735926369,
      "tokens_seen": 628752384
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008176328986960882,
      "loss": 1.5062,
      "theoretical_loss": 3.84969851860639,
      "tokens_seen": 628817920
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008176128385155467,
      "loss": 1.5491,
      "theoretical_loss": 3.8496573067845503,
      "tokens_seen": 628883456
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000817592778335005,
      "loss": 1.6805,
      "theoretical_loss": 3.849616100459544,
      "tokens_seen": 628948992
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008175727181544634,
      "loss": 1.6466,
      "theoretical_loss": 3.8495748996300656,
      "tokens_seen": 629014528
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008175526579739218,
      "loss": 1.6244,
      "theoretical_loss": 3.8495337042948092,
      "tokens_seen": 629080064
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.4909607768058777,
      "objective/train/docs_used": 234355,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4236242771148682,
      "objective/train/original_loss": 1.4236242771148682,
      "objective/train/theoretical_loss": 3.8494925144524705,
      "objective/train/tokens_used": 649605600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24220523238182068,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503273010253906,
      "objective/train/weighted_lm_loss": 1.4951283931732178,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8494925144524705,
      "tokens_seen": 629145600
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008175325977933802,
      "loss": 1.5261,
      "theoretical_loss": 3.8494925144524705,
      "tokens_seen": 629145600
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008175125376128385,
      "loss": 1.6114,
      "theoretical_loss": 3.8494513301017443,
      "tokens_seen": 629211136
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008174924774322969,
      "loss": 1.6353,
      "theoretical_loss": 3.8494101512413277,
      "tokens_seen": 629276672
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008174724172517553,
      "loss": 1.6749,
      "theoretical_loss": 3.849368977869916,
      "tokens_seen": 629342208
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008174523570712137,
      "loss": 1.5678,
      "theoretical_loss": 3.849327809986206,
      "tokens_seen": 629407744
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000817432296890672,
      "loss": 1.6866,
      "theoretical_loss": 3.8492866475888956,
      "tokens_seen": 629473280
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008174122367101304,
      "loss": 1.5259,
      "theoretical_loss": 3.849245490676682,
      "tokens_seen": 629538816
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008173921765295887,
      "loss": 1.4854,
      "theoretical_loss": 3.8492043392482644,
      "tokens_seen": 629604352
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008173721163490472,
      "loss": 1.4975,
      "theoretical_loss": 3.8491631933023407,
      "tokens_seen": 629669888
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008173520561685055,
      "loss": 1.6544,
      "theoretical_loss": 3.8491220528376098,
      "tokens_seen": 629735424
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008173319959879639,
      "loss": 1.6691,
      "theoretical_loss": 3.8490809178527714,
      "tokens_seen": 629800960
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008173119358074223,
      "loss": 1.7029,
      "theoretical_loss": 3.849039788346526,
      "tokens_seen": 629866496
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008172918756268807,
      "loss": 1.5522,
      "theoretical_loss": 3.8489986643175733,
      "tokens_seen": 629932032
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000817271815446339,
      "loss": 1.5926,
      "theoretical_loss": 3.8489575457646144,
      "tokens_seen": 629997568
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008172517552657975,
      "loss": 1.5624,
      "theoretical_loss": 3.8489164326863508,
      "tokens_seen": 630063104
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008172316950852557,
      "loss": 1.6351,
      "theoretical_loss": 3.848875325081484,
      "tokens_seen": 630128640
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008172116349047142,
      "loss": 1.6827,
      "theoretical_loss": 3.8488342229487156,
      "tokens_seen": 630194176
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008171915747241725,
      "loss": 1.6491,
      "theoretical_loss": 3.8487931262867496,
      "tokens_seen": 630259712
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008171715145436309,
      "loss": 1.6055,
      "theoretical_loss": 3.8487520350942876,
      "tokens_seen": 630325248
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008171514543630893,
      "loss": 1.686,
      "theoretical_loss": 3.8487109493700347,
      "tokens_seen": 630390784
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008171313941825476,
      "loss": 1.691,
      "theoretical_loss": 3.8486698691126935,
      "tokens_seen": 630456320
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000817111334002006,
      "loss": 1.5864,
      "theoretical_loss": 3.8486287943209687,
      "tokens_seen": 630521856
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008170912738214645,
      "loss": 1.6892,
      "theoretical_loss": 3.848587724993566,
      "tokens_seen": 630587392
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008170712136409227,
      "loss": 1.5977,
      "theoretical_loss": 3.8485466611291894,
      "tokens_seen": 630652928
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008170511534603812,
      "loss": 1.6303,
      "theoretical_loss": 3.8485056027265454,
      "tokens_seen": 630718464
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008170310932798394,
      "loss": 1.5844,
      "theoretical_loss": 3.8484645497843397,
      "tokens_seen": 630784000
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008170110330992979,
      "loss": 1.5034,
      "theoretical_loss": 3.848423502301279,
      "tokens_seen": 630849536
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008169909729187563,
      "loss": 1.7574,
      "theoretical_loss": 3.8483824602760714,
      "tokens_seen": 630915072
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008169709127382146,
      "loss": 1.5968,
      "theoretical_loss": 3.848341423707423,
      "tokens_seen": 630980608
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000816950852557673,
      "loss": 1.489,
      "theoretical_loss": 3.848300392594042,
      "tokens_seen": 631046144
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008169307923771315,
      "loss": 1.6075,
      "theoretical_loss": 3.8482593669346374,
      "tokens_seen": 631111680
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008169107321965897,
      "loss": 1.661,
      "theoretical_loss": 3.8482183467279176,
      "tokens_seen": 631177216
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008168906720160482,
      "loss": 1.5972,
      "theoretical_loss": 3.8481773319725914,
      "tokens_seen": 631242752
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008168706118355064,
      "loss": 1.6962,
      "theoretical_loss": 3.848136322667369,
      "tokens_seen": 631308288
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008168505516549649,
      "loss": 1.6175,
      "theoretical_loss": 3.8480953188109606,
      "tokens_seen": 631373824
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008168304914744233,
      "loss": 1.6909,
      "theoretical_loss": 3.8480543204020763,
      "tokens_seen": 631439360
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008168104312938816,
      "loss": 1.6689,
      "theoretical_loss": 3.8480133274394275,
      "tokens_seen": 631504896
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008167903711133401,
      "loss": 1.5611,
      "theoretical_loss": 3.847972339921726,
      "tokens_seen": 631570432
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008167703109327984,
      "loss": 1.4894,
      "theoretical_loss": 3.8479313578476826,
      "tokens_seen": 631635968
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008167502507522568,
      "loss": 1.5413,
      "theoretical_loss": 3.8478903812160103,
      "tokens_seen": 631701504
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008167301905717152,
      "loss": 1.5309,
      "theoretical_loss": 3.8478494100254217,
      "tokens_seen": 631767040
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008167101303911736,
      "loss": 1.6383,
      "theoretical_loss": 3.84780844427463,
      "tokens_seen": 631832576
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008166900702106319,
      "loss": 1.7528,
      "theoretical_loss": 3.8477674839623495,
      "tokens_seen": 631898112
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008166700100300904,
      "loss": 1.5873,
      "theoretical_loss": 3.847726529087293,
      "tokens_seen": 631963648
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008166499498495486,
      "loss": 1.6397,
      "theoretical_loss": 3.847685579648176,
      "tokens_seen": 632029184
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008166298896690071,
      "loss": 1.6154,
      "theoretical_loss": 3.8476446356437126,
      "tokens_seen": 632094720
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008166098294884654,
      "loss": 1.6617,
      "theoretical_loss": 3.8476036970726186,
      "tokens_seen": 632160256
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008165897693079238,
      "loss": 1.6953,
      "theoretical_loss": 3.8475627639336105,
      "tokens_seen": 632225792
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008165697091273822,
      "loss": 1.6312,
      "theoretical_loss": 3.847521836225404,
      "tokens_seen": 632291328
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008165496489468406,
      "loss": 1.5592,
      "theoretical_loss": 3.847480913946715,
      "tokens_seen": 632356864
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.49222224950790405,
      "objective/train/docs_used": 235545,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.8225845098495483,
      "objective/train/original_loss": 1.8225843906402588,
      "objective/train/theoretical_loss": 3.847439997096262,
      "objective/train/tokens_used": 652882400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2427038848400116,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050455927848816,
      "objective/train/weighted_lm_loss": 1.914535403251648,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.847439997096262,
      "tokens_seen": 632422400
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008165295887662989,
      "loss": 1.7776,
      "theoretical_loss": 3.847439997096262,
      "tokens_seen": 632422400
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008165095285857573,
      "loss": 1.6502,
      "theoretical_loss": 3.8473990856727616,
      "tokens_seen": 632487936
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008164894684052157,
      "loss": 1.6379,
      "theoretical_loss": 3.8473581796749317,
      "tokens_seen": 632553472
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008164694082246741,
      "loss": 1.5226,
      "theoretical_loss": 3.8473172791014916,
      "tokens_seen": 632619008
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008164493480441324,
      "loss": 1.5734,
      "theoretical_loss": 3.8472763839511597,
      "tokens_seen": 632684544
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008164292878635908,
      "loss": 1.5846,
      "theoretical_loss": 3.847235494222655,
      "tokens_seen": 632750080
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008164092276830491,
      "loss": 1.5095,
      "theoretical_loss": 3.8471946099146983,
      "tokens_seen": 632815616
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008163891675025076,
      "loss": 1.6385,
      "theoretical_loss": 3.847153731026008,
      "tokens_seen": 632881152
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008163691073219659,
      "loss": 1.5517,
      "theoretical_loss": 3.847112857555306,
      "tokens_seen": 632946688
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008163490471414243,
      "loss": 1.6175,
      "theoretical_loss": 3.8470719895013135,
      "tokens_seen": 633012224
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008163289869608827,
      "loss": 1.5434,
      "theoretical_loss": 3.847031126862751,
      "tokens_seen": 633077760
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008163089267803411,
      "loss": 1.6316,
      "theoretical_loss": 3.846990269638341,
      "tokens_seen": 633143296
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008162888665997994,
      "loss": 1.5763,
      "theoretical_loss": 3.8469494178268056,
      "tokens_seen": 633208832
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008162688064192579,
      "loss": 1.542,
      "theoretical_loss": 3.8469085714268676,
      "tokens_seen": 633274368
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008162487462387161,
      "loss": 1.5828,
      "theoretical_loss": 3.8468677304372507,
      "tokens_seen": 633339904
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008162286860581746,
      "loss": 1.5444,
      "theoretical_loss": 3.8468268948566786,
      "tokens_seen": 633405440
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008162086258776329,
      "loss": 1.5582,
      "theoretical_loss": 3.846786064683874,
      "tokens_seen": 633470976
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008161885656970913,
      "loss": 1.5706,
      "theoretical_loss": 3.8467452399175635,
      "tokens_seen": 633536512
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008161685055165497,
      "loss": 1.5734,
      "theoretical_loss": 3.8467044205564704,
      "tokens_seen": 633602048
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000816148445336008,
      "loss": 1.5057,
      "theoretical_loss": 3.8466636065993205,
      "tokens_seen": 633667584
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008161283851554664,
      "loss": 1.6447,
      "theoretical_loss": 3.8466227980448395,
      "tokens_seen": 633733120
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008161083249749249,
      "loss": 1.6633,
      "theoretical_loss": 3.8465819948917543,
      "tokens_seen": 633798656
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008160882647943831,
      "loss": 1.6112,
      "theoretical_loss": 3.846541197138791,
      "tokens_seen": 633864192
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008160682046138416,
      "loss": 1.625,
      "theoretical_loss": 3.8465004047846767,
      "tokens_seen": 633929728
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008160481444332998,
      "loss": 1.7111,
      "theoretical_loss": 3.846459617828139,
      "tokens_seen": 633995264
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008160280842527583,
      "loss": 1.6754,
      "theoretical_loss": 3.8464188362679055,
      "tokens_seen": 634060800
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008160080240722167,
      "loss": 1.6947,
      "theoretical_loss": 3.8463780601027056,
      "tokens_seen": 634126336
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000815987963891675,
      "loss": 1.5935,
      "theoretical_loss": 3.8463372893312675,
      "tokens_seen": 634191872
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008159679037111334,
      "loss": 1.5846,
      "theoretical_loss": 3.8462965239523204,
      "tokens_seen": 634257408
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008159478435305919,
      "loss": 1.5438,
      "theoretical_loss": 3.846255763964594,
      "tokens_seen": 634322944
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008159277833500501,
      "loss": 1.5796,
      "theoretical_loss": 3.846215009366819,
      "tokens_seen": 634388480
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008159077231695086,
      "loss": 1.6794,
      "theoretical_loss": 3.8461742601577247,
      "tokens_seen": 634454016
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008158876629889668,
      "loss": 1.5451,
      "theoretical_loss": 3.8461335163360433,
      "tokens_seen": 634519552
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008158676028084253,
      "loss": 1.6057,
      "theoretical_loss": 3.8460927779005054,
      "tokens_seen": 634585088
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008158475426278837,
      "loss": 1.664,
      "theoretical_loss": 3.846052044849843,
      "tokens_seen": 634650624
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000815827482447342,
      "loss": 1.6116,
      "theoretical_loss": 3.846011317182789,
      "tokens_seen": 634716160
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008158074222668004,
      "loss": 1.5671,
      "theoretical_loss": 3.845970594898076,
      "tokens_seen": 634781696
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008157873620862588,
      "loss": 1.6006,
      "theoretical_loss": 3.845929877994436,
      "tokens_seen": 634847232
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008157673019057171,
      "loss": 1.5838,
      "theoretical_loss": 3.845889166470604,
      "tokens_seen": 634912768
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008157472417251756,
      "loss": 1.5436,
      "theoretical_loss": 3.845848460325313,
      "tokens_seen": 634978304
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008157271815446339,
      "loss": 1.7046,
      "theoretical_loss": 3.8458077595572977,
      "tokens_seen": 635043840
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008157071213640923,
      "loss": 1.5955,
      "theoretical_loss": 3.8457670641652935,
      "tokens_seen": 635109376
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008156870611835506,
      "loss": 1.5783,
      "theoretical_loss": 3.8457263741480343,
      "tokens_seen": 635174912
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000815667001003009,
      "loss": 1.573,
      "theoretical_loss": 3.8456856895042573,
      "tokens_seen": 635240448
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008156469408224674,
      "loss": 1.5688,
      "theoretical_loss": 3.8456450102326976,
      "tokens_seen": 635305984
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008156268806419258,
      "loss": 1.6171,
      "theoretical_loss": 3.8456043363320926,
      "tokens_seen": 635371520
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008156068204613841,
      "loss": 1.5597,
      "theoretical_loss": 3.845563667801178,
      "tokens_seen": 635437056
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008155867602808426,
      "loss": 1.6134,
      "theoretical_loss": 3.8455230046386926,
      "tokens_seen": 635502592
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008155667001003009,
      "loss": 1.6806,
      "theoretical_loss": 3.845482346843373,
      "tokens_seen": 635568128
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008155466399197593,
      "loss": 1.5767,
      "theoretical_loss": 3.845441694413959,
      "tokens_seen": 635633664
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.49294331669807434,
      "objective/train/docs_used": 236526,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5541789531707764,
      "objective/train/original_loss": 1.5541788339614868,
      "objective/train/theoretical_loss": 3.8454010473491884,
      "objective/train/tokens_used": 656159200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24326615035533905,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0505311489105225,
      "objective/train/weighted_lm_loss": 1.6325575113296509,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0253151655197144,
      "theoretical_loss": 3.8454010473491884,
      "tokens_seen": 635699200
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008155265797392176,
      "loss": 1.5994,
      "theoretical_loss": 3.8454010473491884,
      "tokens_seen": 635699200
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000815506519558676,
      "loss": 1.6319,
      "theoretical_loss": 3.8453604056477997,
      "tokens_seen": 635764736
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008154864593781344,
      "loss": 1.579,
      "theoretical_loss": 3.8453197693085333,
      "tokens_seen": 635830272
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008154663991975928,
      "loss": 1.6322,
      "theoretical_loss": 3.845279138330129,
      "tokens_seen": 635895808
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008154463390170511,
      "loss": 1.5124,
      "theoretical_loss": 3.845238512711327,
      "tokens_seen": 635961344
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008154262788365095,
      "loss": 1.5947,
      "theoretical_loss": 3.845197892450868,
      "tokens_seen": 636026880
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008154062186559679,
      "loss": 1.6175,
      "theoretical_loss": 3.845157277547494,
      "tokens_seen": 636092416
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008153861584754263,
      "loss": 1.6552,
      "theoretical_loss": 3.845116667999946,
      "tokens_seen": 636157952
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008153660982948846,
      "loss": 1.6637,
      "theoretical_loss": 3.845076063806966,
      "tokens_seen": 636223488
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008153460381143431,
      "loss": 1.6709,
      "theoretical_loss": 3.845035464967297,
      "tokens_seen": 636289024
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008153259779338014,
      "loss": 1.6537,
      "theoretical_loss": 3.8449948714796816,
      "tokens_seen": 636354560
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008153059177532598,
      "loss": 1.623,
      "theoretical_loss": 3.8449542833428634,
      "tokens_seen": 636420096
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008152858575727183,
      "loss": 1.714,
      "theoretical_loss": 3.844913700555586,
      "tokens_seen": 636485632
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008152657973921765,
      "loss": 1.5764,
      "theoretical_loss": 3.844873123116594,
      "tokens_seen": 636551168
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000815245737211635,
      "loss": 1.5439,
      "theoretical_loss": 3.844832551024631,
      "tokens_seen": 636616704
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008152256770310933,
      "loss": 1.5508,
      "theoretical_loss": 3.8447919842784435,
      "tokens_seen": 636682240
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008152056168505517,
      "loss": 1.579,
      "theoretical_loss": 3.8447514228767763,
      "tokens_seen": 636747776
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008151855566700101,
      "loss": 1.5836,
      "theoretical_loss": 3.844710866818375,
      "tokens_seen": 636813312
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008151654964894684,
      "loss": 1.862,
      "theoretical_loss": 3.844670316101986,
      "tokens_seen": 636878848
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008151454363089268,
      "loss": 1.5483,
      "theoretical_loss": 3.844629770726357,
      "tokens_seen": 636944384
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008151253761283853,
      "loss": 1.6434,
      "theoretical_loss": 3.844589230690234,
      "tokens_seen": 637009920
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008151053159478435,
      "loss": 1.5362,
      "theoretical_loss": 3.844548695992365,
      "tokens_seen": 637075456
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000815085255767302,
      "loss": 1.6246,
      "theoretical_loss": 3.8445081666314986,
      "tokens_seen": 637140992
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008150651955867602,
      "loss": 1.5281,
      "theoretical_loss": 3.8444676426063826,
      "tokens_seen": 637206528
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008150451354062187,
      "loss": 1.6882,
      "theoretical_loss": 3.844427123915766,
      "tokens_seen": 637272064
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008150250752256771,
      "loss": 1.5928,
      "theoretical_loss": 3.844386610558398,
      "tokens_seen": 637337600
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008150050150451354,
      "loss": 1.568,
      "theoretical_loss": 3.844346102533029,
      "tokens_seen": 637403136
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008149849548645938,
      "loss": 1.5826,
      "theoretical_loss": 3.8443055998384077,
      "tokens_seen": 637468672
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008149648946840523,
      "loss": 1.516,
      "theoretical_loss": 3.8442651024732863,
      "tokens_seen": 637534208
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008149448345035105,
      "loss": 1.6055,
      "theoretical_loss": 3.844224610436415,
      "tokens_seen": 637599744
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000814924774322969,
      "loss": 1.455,
      "theoretical_loss": 3.844184123726545,
      "tokens_seen": 637665280
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008149047141424272,
      "loss": 1.5898,
      "theoretical_loss": 3.844143642342428,
      "tokens_seen": 637730816
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008148846539618857,
      "loss": 1.605,
      "theoretical_loss": 3.8441031662828173,
      "tokens_seen": 637796352
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008148645937813441,
      "loss": 1.6104,
      "theoretical_loss": 3.8440626955464645,
      "tokens_seen": 637861888
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008148445336008024,
      "loss": 1.5913,
      "theoretical_loss": 3.844022230132123,
      "tokens_seen": 637927424
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008148244734202608,
      "loss": 1.5559,
      "theoretical_loss": 3.8439817700385466,
      "tokens_seen": 637992960
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008148044132397192,
      "loss": 1.6045,
      "theoretical_loss": 3.843941315264489,
      "tokens_seen": 638058496
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008147843530591775,
      "loss": 1.7066,
      "theoretical_loss": 3.8439008658087044,
      "tokens_seen": 638124032
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000814764292878636,
      "loss": 1.5802,
      "theoretical_loss": 3.8438604216699477,
      "tokens_seen": 638189568
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008147442326980943,
      "loss": 1.5175,
      "theoretical_loss": 3.843819982846974,
      "tokens_seen": 638255104
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008147241725175527,
      "loss": 1.4899,
      "theoretical_loss": 3.8437795493385387,
      "tokens_seen": 638320640
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000814704112337011,
      "loss": 1.5917,
      "theoretical_loss": 3.8437391211433987,
      "tokens_seen": 638386176
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008146840521564694,
      "loss": 1.6376,
      "theoretical_loss": 3.84369869826031,
      "tokens_seen": 638451712
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008146639919759278,
      "loss": 1.5972,
      "theoretical_loss": 3.843658280688029,
      "tokens_seen": 638517248
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008146439317953862,
      "loss": 1.6802,
      "theoretical_loss": 3.8436178684253126,
      "tokens_seen": 638582784
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008146238716148445,
      "loss": 1.6295,
      "theoretical_loss": 3.84357746147092,
      "tokens_seen": 638648320
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000814603811434303,
      "loss": 1.7287,
      "theoretical_loss": 3.843537059823608,
      "tokens_seen": 638713856
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008145837512537613,
      "loss": 1.6448,
      "theoretical_loss": 3.843496663482136,
      "tokens_seen": 638779392
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008145636910732197,
      "loss": 1.5549,
      "theoretical_loss": 3.843456272445262,
      "tokens_seen": 638844928
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000814543630892678,
      "loss": 1.5987,
      "theoretical_loss": 3.8434158867117465,
      "tokens_seen": 638910464
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.49233630299568176,
      "objective/train/docs_used": 237858,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.3402748107910156,
      "objective/train/original_loss": 1.3402745723724365,
      "objective/train/theoretical_loss": 3.8433755062803483,
      "objective/train/tokens_used": 659436000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2428392469882965,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.05046808719635,
      "objective/train/weighted_lm_loss": 1.407797932624817,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8433755062803483,
      "tokens_seen": 638976000
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008145235707121365,
      "loss": 1.3713,
      "theoretical_loss": 3.8433755062803483,
      "tokens_seen": 638976000
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008145035105315948,
      "loss": 1.6112,
      "theoretical_loss": 3.8433351311498285,
      "tokens_seen": 639041536
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008144834503510532,
      "loss": 1.6616,
      "theoretical_loss": 3.8432947613189468,
      "tokens_seen": 639107072
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008144633901705115,
      "loss": 1.5815,
      "theoretical_loss": 3.843254396786465,
      "tokens_seen": 639172608
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008144433299899699,
      "loss": 1.5886,
      "theoretical_loss": 3.843214037551144,
      "tokens_seen": 639238144
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008144232698094283,
      "loss": 1.6515,
      "theoretical_loss": 3.843173683611746,
      "tokens_seen": 639303680
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008144032096288867,
      "loss": 1.7762,
      "theoretical_loss": 3.8431333349670336,
      "tokens_seen": 639369216
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000814383149448345,
      "loss": 1.5744,
      "theoretical_loss": 3.8430929916157686,
      "tokens_seen": 639434752
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008143630892678035,
      "loss": 1.4854,
      "theoretical_loss": 3.843052653556714,
      "tokens_seen": 639500288
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008143430290872618,
      "loss": 1.6078,
      "theoretical_loss": 3.8430123207886346,
      "tokens_seen": 639565824
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008143229689067202,
      "loss": 1.6266,
      "theoretical_loss": 3.842971993310294,
      "tokens_seen": 639631360
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008143029087261785,
      "loss": 1.5957,
      "theoretical_loss": 3.8429316711204557,
      "tokens_seen": 639696896
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008142828485456369,
      "loss": 1.5743,
      "theoretical_loss": 3.842891354217885,
      "tokens_seen": 639762432
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008142627883650953,
      "loss": 1.6018,
      "theoretical_loss": 3.842851042601348,
      "tokens_seen": 639827968
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008142427281845537,
      "loss": 1.5752,
      "theoretical_loss": 3.8428107362696085,
      "tokens_seen": 639893504
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000814222668004012,
      "loss": 1.5521,
      "theoretical_loss": 3.842770435221434,
      "tokens_seen": 639959040
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008142026078234705,
      "loss": 1.5399,
      "theoretical_loss": 3.8427301394555897,
      "tokens_seen": 640024576
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008141825476429287,
      "loss": 1.526,
      "theoretical_loss": 3.8426898489708434,
      "tokens_seen": 640090112
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008141624874623872,
      "loss": 1.6133,
      "theoretical_loss": 3.842649563765962,
      "tokens_seen": 640155648
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008141424272818456,
      "loss": 1.6805,
      "theoretical_loss": 3.8426092838397135,
      "tokens_seen": 640221184
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008141223671013039,
      "loss": 1.7152,
      "theoretical_loss": 3.842569009190866,
      "tokens_seen": 640286720
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008141023069207623,
      "loss": 1.5326,
      "theoretical_loss": 3.8425287398181873,
      "tokens_seen": 640352256
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008140822467402206,
      "loss": 1.6237,
      "theoretical_loss": 3.8424884757204474,
      "tokens_seen": 640417792
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000814062186559679,
      "loss": 1.6939,
      "theoretical_loss": 3.8424482168964142,
      "tokens_seen": 640483328
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008140421263791375,
      "loss": 1.5402,
      "theoretical_loss": 3.8424079633448587,
      "tokens_seen": 640548864
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008140220661985957,
      "loss": 1.5025,
      "theoretical_loss": 3.8423677150645505,
      "tokens_seen": 640614400
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008140020060180542,
      "loss": 1.6547,
      "theoretical_loss": 3.8423274720542606,
      "tokens_seen": 640679936
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008139819458375126,
      "loss": 1.5912,
      "theoretical_loss": 3.8422872343127596,
      "tokens_seen": 640745472
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008139618856569709,
      "loss": 1.5802,
      "theoretical_loss": 3.842247001838819,
      "tokens_seen": 640811008
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008139418254764293,
      "loss": 1.4553,
      "theoretical_loss": 3.8422067746312107,
      "tokens_seen": 640876544
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008139217652958876,
      "loss": 1.6781,
      "theoretical_loss": 3.842166552688706,
      "tokens_seen": 640942080
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000813901705115346,
      "loss": 1.5691,
      "theoretical_loss": 3.8421263360100797,
      "tokens_seen": 641007616
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008138816449348045,
      "loss": 1.6562,
      "theoretical_loss": 3.8420861245941027,
      "tokens_seen": 641073152
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008138615847542627,
      "loss": 1.6452,
      "theoretical_loss": 3.842045918439549,
      "tokens_seen": 641138688
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008138415245737212,
      "loss": 1.5871,
      "theoretical_loss": 3.8420057175451934,
      "tokens_seen": 641204224
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008138214643931795,
      "loss": 1.6131,
      "theoretical_loss": 3.8419655219098097,
      "tokens_seen": 641269760
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008138014042126379,
      "loss": 1.6284,
      "theoretical_loss": 3.8419253315321713,
      "tokens_seen": 641335296
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008137813440320964,
      "loss": 1.6019,
      "theoretical_loss": 3.8418851464110553,
      "tokens_seen": 641400832
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008137612838515547,
      "loss": 1.6425,
      "theoretical_loss": 3.841844966545236,
      "tokens_seen": 641466368
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008137412236710131,
      "loss": 1.5938,
      "theoretical_loss": 3.8418047919334892,
      "tokens_seen": 641531904
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008137211634904714,
      "loss": 1.5786,
      "theoretical_loss": 3.8417646225745923,
      "tokens_seen": 641597440
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008137011033099298,
      "loss": 1.5369,
      "theoretical_loss": 3.841724458467321,
      "tokens_seen": 641662976
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008136810431293882,
      "loss": 1.6833,
      "theoretical_loss": 3.841684299610453,
      "tokens_seen": 641728512
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008136609829488466,
      "loss": 1.5156,
      "theoretical_loss": 3.841644146002766,
      "tokens_seen": 641794048
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008136409227683049,
      "loss": 1.5341,
      "theoretical_loss": 3.841603997643037,
      "tokens_seen": 641859584
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008136208625877634,
      "loss": 1.687,
      "theoretical_loss": 3.8415638545300457,
      "tokens_seen": 641925120
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008136008024072217,
      "loss": 1.5458,
      "theoretical_loss": 3.8415237166625698,
      "tokens_seen": 641990656
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008135807422266801,
      "loss": 1.5651,
      "theoretical_loss": 3.8414835840393886,
      "tokens_seen": 642056192
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008135606820461384,
      "loss": 1.4712,
      "theoretical_loss": 3.8414434566592828,
      "tokens_seen": 642121728
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008135406218655969,
      "loss": 1.5997,
      "theoretical_loss": 3.8414033345210314,
      "tokens_seen": 642187264
    },
    {
      "epoch": 0.19,
      "objective/train/advantage_avg": 0.49311363697052,
      "objective/train/docs_used": 239059,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5280405282974243,
      "objective/train/original_loss": 1.5280404090881348,
      "objective/train/theoretical_loss": 3.8413632176234147,
      "objective/train/tokens_used": 662712800,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24341195821762085,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.050548791885376,
      "objective/train/weighted_lm_loss": 1.6052095890045166,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 1.0,
      "theoretical_loss": 3.8413632176234147,
      "tokens_seen": 642252800
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008135205616850552,
      "loss": 1.7193,
      "theoretical_loss": 3.8413632176234147,
      "tokens_seen": 642252800
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008135005015045136,
      "loss": 1.5913,
      "theoretical_loss": 3.8413231059652144,
      "tokens_seen": 642318336
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008134804413239719,
      "loss": 1.6132,
      "theoretical_loss": 3.8412829995452107,
      "tokens_seen": 642383872
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008134603811434303,
      "loss": 1.734,
      "theoretical_loss": 3.8412428983621867,
      "tokens_seen": 642449408
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008134403209628887,
      "loss": 1.7358,
      "theoretical_loss": 3.8412028024149225,
      "tokens_seen": 642514944
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008134202607823471,
      "loss": 1.5482,
      "theoretical_loss": 3.8411627117022022,
      "tokens_seen": 642580480
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008134002006018054,
      "loss": 1.5465,
      "theoretical_loss": 3.8411226262228078,
      "tokens_seen": 642646016
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008133801404212639,
      "loss": 1.6358,
      "theoretical_loss": 3.841082545975523,
      "tokens_seen": 642711552
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008133600802407222,
      "loss": 1.5882,
      "theoretical_loss": 3.841042470959131,
      "tokens_seen": 642777088
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008133400200601806,
      "loss": 1.5375,
      "theoretical_loss": 3.8410024011724166,
      "tokens_seen": 642842624
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008133199598796389,
      "loss": 1.5786,
      "theoretical_loss": 3.840962336614164,
      "tokens_seen": 642908160
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008132998996990973,
      "loss": 1.6254,
      "theoretical_loss": 3.8409222772831577,
      "tokens_seen": 642973696
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008132798395185557,
      "loss": 1.5837,
      "theoretical_loss": 3.840882223178183,
      "tokens_seen": 643039232
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008132597793380141,
      "loss": 1.6823,
      "theoretical_loss": 3.8408421742980265,
      "tokens_seen": 643104768
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008132397191574724,
      "loss": 1.5178,
      "theoretical_loss": 3.8408021306414737,
      "tokens_seen": 643170304
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008132196589769309,
      "loss": 1.6362,
      "theoretical_loss": 3.840762092207311,
      "tokens_seen": 643235840
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008131995987963891,
      "loss": 1.6533,
      "theoretical_loss": 3.8407220589943254,
      "tokens_seen": 643301376
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008131795386158476,
      "loss": 1.6076,
      "theoretical_loss": 3.840682031001305,
      "tokens_seen": 643366912
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.000813159478435306,
      "loss": 1.5621,
      "theoretical_loss": 3.840642008227036,
      "tokens_seen": 643432448
    },
    {
      "epoch": 0.19,
      "learning_rate": 0.0008131394182547643,
      "loss": 1.5643,
      "theoretical_loss": 3.840601990670308,
      "tokens_seen": 643497984
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008131193580742227,
      "loss": 1.4832,
      "theoretical_loss": 3.8405619783299083,
      "tokens_seen": 643563520
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000813099297893681,
      "loss": 1.4709,
      "theoretical_loss": 3.8405219712046272,
      "tokens_seen": 643629056
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008130792377131394,
      "loss": 1.5877,
      "theoretical_loss": 3.8404819692932533,
      "tokens_seen": 643694592
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008130591775325979,
      "loss": 1.5957,
      "theoretical_loss": 3.840441972594576,
      "tokens_seen": 643760128
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008130391173520561,
      "loss": 1.6079,
      "theoretical_loss": 3.8404019811073864,
      "tokens_seen": 643825664
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008130190571715146,
      "loss": 1.5431,
      "theoretical_loss": 3.8403619948304746,
      "tokens_seen": 643891200
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000812998996990973,
      "loss": 1.5753,
      "theoretical_loss": 3.8403220137626315,
      "tokens_seen": 643956736
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008129789368104313,
      "loss": 1.5984,
      "theoretical_loss": 3.8402820379026483,
      "tokens_seen": 644022272
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008129588766298897,
      "loss": 1.577,
      "theoretical_loss": 3.840242067249317,
      "tokens_seen": 644087808
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000812938816449348,
      "loss": 1.5498,
      "theoretical_loss": 3.84020210180143,
      "tokens_seen": 644153344
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008129187562688064,
      "loss": 1.5375,
      "theoretical_loss": 3.84016214155778,
      "tokens_seen": 644218880
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008128986960882649,
      "loss": 1.6124,
      "theoretical_loss": 3.8401221865171595,
      "tokens_seen": 644284416
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008128786359077231,
      "loss": 1.5779,
      "theoretical_loss": 3.840082236678362,
      "tokens_seen": 644349952
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008128585757271816,
      "loss": 1.534,
      "theoretical_loss": 3.8400422920401818,
      "tokens_seen": 644415488
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008128385155466399,
      "loss": 1.5585,
      "theoretical_loss": 3.840002352601412,
      "tokens_seen": 644481024
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008128184553660983,
      "loss": 1.6898,
      "theoretical_loss": 3.8399624183608485,
      "tokens_seen": 644546560
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008127983951855567,
      "loss": 1.6361,
      "theoretical_loss": 3.8399224893172854,
      "tokens_seen": 644612096
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000812778335005015,
      "loss": 1.6244,
      "theoretical_loss": 3.8398825654695186,
      "tokens_seen": 644677632
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008127582748244734,
      "loss": 1.6029,
      "theoretical_loss": 3.8398426468163436,
      "tokens_seen": 644743168
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008127382146439318,
      "loss": 1.5932,
      "theoretical_loss": 3.8398027333565565,
      "tokens_seen": 644808704
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008127181544633901,
      "loss": 1.5407,
      "theoretical_loss": 3.839762825088955,
      "tokens_seen": 644874240
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008126980942828486,
      "loss": 1.6714,
      "theoretical_loss": 3.839722922012334,
      "tokens_seen": 644939776
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008126780341023069,
      "loss": 1.681,
      "theoretical_loss": 3.8396830241254927,
      "tokens_seen": 645005312
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008126579739217653,
      "loss": 1.7181,
      "theoretical_loss": 3.839643131427229,
      "tokens_seen": 645070848
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008126379137412237,
      "loss": 1.6032,
      "theoretical_loss": 3.8396032439163394,
      "tokens_seen": 645136384
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008126178535606821,
      "loss": 1.5822,
      "theoretical_loss": 3.8395633615916243,
      "tokens_seen": 645201920
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008125977933801404,
      "loss": 1.6175,
      "theoretical_loss": 3.839523484451882,
      "tokens_seen": 645267456
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008125777331995988,
      "loss": 1.5985,
      "theoretical_loss": 3.8394836124959113,
      "tokens_seen": 645332992
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008125576730190571,
      "loss": 1.6482,
      "theoretical_loss": 3.8394437457225132,
      "tokens_seen": 645398528
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008125376128385156,
      "loss": 1.6193,
      "theoretical_loss": 3.8394038841304874,
      "tokens_seen": 645464064
    },
    {
      "epoch": 0.2,
      "objective/train/advantage_avg": 0.4884245991706848,
      "objective/train/docs_used": 240316,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5354140996932983,
      "objective/train/original_loss": 1.5354139804840088,
      "objective/train/theoretical_loss": 3.8393640277186343,
      "objective/train/tokens_used": 665989600,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.23942209780216217,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0500595569610596,
      "objective/train/weighted_lm_loss": 1.611963152885437,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8393640277186343,
      "tokens_seen": 645529600
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008125175526579739,
      "loss": 1.566,
      "theoretical_loss": 3.8393640277186343,
      "tokens_seen": 645529600
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008124974924774323,
      "loss": 1.6982,
      "theoretical_loss": 3.839324176485755,
      "tokens_seen": 645595136
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008124774322968906,
      "loss": 1.7881,
      "theoretical_loss": 3.839284330430651,
      "tokens_seen": 645660672
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008124573721163491,
      "loss": 1.5892,
      "theoretical_loss": 3.839244489552124,
      "tokens_seen": 645726208
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008124373119358074,
      "loss": 1.6584,
      "theoretical_loss": 3.8392046538489764,
      "tokens_seen": 645791744
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008124172517552658,
      "loss": 1.6323,
      "theoretical_loss": 3.839164823320011,
      "tokens_seen": 645857280
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008123971915747242,
      "loss": 1.6147,
      "theoretical_loss": 3.8391249979640305,
      "tokens_seen": 645922816
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008123771313941825,
      "loss": 1.6646,
      "theoretical_loss": 3.839085177779838,
      "tokens_seen": 645988352
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008123570712136409,
      "loss": 1.6631,
      "theoretical_loss": 3.839045362766238,
      "tokens_seen": 646053888
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008123370110330993,
      "loss": 1.4806,
      "theoretical_loss": 3.8390055529220337,
      "tokens_seen": 646119424
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008123169508525576,
      "loss": 1.5989,
      "theoretical_loss": 3.8389657482460313,
      "tokens_seen": 646184960
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008122968906720161,
      "loss": 1.5877,
      "theoretical_loss": 3.838925948737034,
      "tokens_seen": 646250496
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008122768304914745,
      "loss": 1.5171,
      "theoretical_loss": 3.8388861543938484,
      "tokens_seen": 646316032
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008122567703109328,
      "loss": 1.5722,
      "theoretical_loss": 3.8388463652152796,
      "tokens_seen": 646381568
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008122367101303913,
      "loss": 1.5325,
      "theoretical_loss": 3.838806581200134,
      "tokens_seen": 646447104
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008122166499498495,
      "loss": 1.5838,
      "theoretical_loss": 3.8387668023472195,
      "tokens_seen": 646512640
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000812196589769308,
      "loss": 1.6213,
      "theoretical_loss": 3.8387270286553408,
      "tokens_seen": 646578176
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008121765295887664,
      "loss": 1.699,
      "theoretical_loss": 3.838687260123306,
      "tokens_seen": 646643712
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008121564694082247,
      "loss": 1.5451,
      "theoretical_loss": 3.838647496749924,
      "tokens_seen": 646709248
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008121364092276831,
      "loss": 1.6904,
      "theoretical_loss": 3.838607738534002,
      "tokens_seen": 646774784
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008121163490471414,
      "loss": 1.7643,
      "theoretical_loss": 3.8385679854743486,
      "tokens_seen": 646840320
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008120962888665998,
      "loss": 1.6247,
      "theoretical_loss": 3.838528237569773,
      "tokens_seen": 646905856
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008120762286860583,
      "loss": 1.4815,
      "theoretical_loss": 3.8384884948190847,
      "tokens_seen": 646971392
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008120561685055165,
      "loss": 1.5986,
      "theoretical_loss": 3.8384487572210935,
      "tokens_seen": 647036928
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000812036108324975,
      "loss": 1.6638,
      "theoretical_loss": 3.8384090247746085,
      "tokens_seen": 647102464
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008120160481444334,
      "loss": 1.6351,
      "theoretical_loss": 3.8383692974784416,
      "tokens_seen": 647168000
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008119959879638917,
      "loss": 1.569,
      "theoretical_loss": 3.838329575331403,
      "tokens_seen": 647233536
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008119759277833501,
      "loss": 1.5627,
      "theoretical_loss": 3.8382898583323044,
      "tokens_seen": 647299072
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008119558676028084,
      "loss": 1.568,
      "theoretical_loss": 3.8382501464799565,
      "tokens_seen": 647364608
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008119358074222668,
      "loss": 1.5953,
      "theoretical_loss": 3.8382104397731736,
      "tokens_seen": 647430144
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008119157472417253,
      "loss": 1.6566,
      "theoretical_loss": 3.8381707382107657,
      "tokens_seen": 647495680
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008118956870611835,
      "loss": 1.649,
      "theoretical_loss": 3.8381310417915477,
      "tokens_seen": 647561216
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000811875626880642,
      "loss": 1.7117,
      "theoretical_loss": 3.8380913505143317,
      "tokens_seen": 647626752
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008118555667001003,
      "loss": 1.6518,
      "theoretical_loss": 3.8380516643779323,
      "tokens_seen": 647692288
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008118355065195587,
      "loss": 1.6882,
      "theoretical_loss": 3.838011983381162,
      "tokens_seen": 647757824
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008118154463390171,
      "loss": 1.6258,
      "theoretical_loss": 3.837972307522837,
      "tokens_seen": 647823360
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008117953861584755,
      "loss": 1.5789,
      "theoretical_loss": 3.837932636801772,
      "tokens_seen": 647888896
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008117753259779338,
      "loss": 1.5619,
      "theoretical_loss": 3.8378929712167817,
      "tokens_seen": 647954432
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008117552657973922,
      "loss": 1.5697,
      "theoretical_loss": 3.8378533107666817,
      "tokens_seen": 648019968
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008117352056168505,
      "loss": 1.612,
      "theoretical_loss": 3.8378136554502884,
      "tokens_seen": 648085504
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000811715145436309,
      "loss": 1.7064,
      "theoretical_loss": 3.837774005266418,
      "tokens_seen": 648151040
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008116950852557673,
      "loss": 1.6931,
      "theoretical_loss": 3.8377343602138874,
      "tokens_seen": 648216576
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008116750250752257,
      "loss": 1.6846,
      "theoretical_loss": 3.8376947202915144,
      "tokens_seen": 648282112
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008116549648946841,
      "loss": 1.612,
      "theoretical_loss": 3.8376550854981155,
      "tokens_seen": 648347648
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008116349047141425,
      "loss": 1.6302,
      "theoretical_loss": 3.83761545583251,
      "tokens_seen": 648413184
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008116148445336008,
      "loss": 1.5701,
      "theoretical_loss": 3.837575831293515,
      "tokens_seen": 648478720
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008115947843530592,
      "loss": 1.627,
      "theoretical_loss": 3.8375362118799505,
      "tokens_seen": 648544256
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008115747241725175,
      "loss": 1.5537,
      "theoretical_loss": 3.837496597590635,
      "tokens_seen": 648609792
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000811554663991976,
      "loss": 1.6683,
      "theoretical_loss": 3.837456988424388,
      "tokens_seen": 648675328
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008115346038114343,
      "loss": 1.5657,
      "theoretical_loss": 3.8374173843800303,
      "tokens_seen": 648740864
    },
    {
      "epoch": 0.2,
      "objective/train/advantage_avg": 0.4949766993522644,
      "objective/train/docs_used": 241544,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.4393831491470337,
      "objective/train/original_loss": 1.4393832683563232,
      "objective/train/theoretical_loss": 3.8373777854563813,
      "objective/train/tokens_used": 669266400,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.2453908920288086,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507452487945557,
      "objective/train/weighted_lm_loss": 1.512376308441162,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8373777854563813,
      "tokens_seen": 648806400
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008115145436308927,
      "loss": 1.6015,
      "theoretical_loss": 3.8373777854563813,
      "tokens_seen": 648806400
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000811494483450351,
      "loss": 1.5641,
      "theoretical_loss": 3.8373381916522624,
      "tokens_seen": 648871936
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008114744232698095,
      "loss": 1.6364,
      "theoretical_loss": 3.8372986029664946,
      "tokens_seen": 648937472
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008114543630892678,
      "loss": 1.6498,
      "theoretical_loss": 3.8372590193978993,
      "tokens_seen": 649003008
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008114343029087262,
      "loss": 1.6915,
      "theoretical_loss": 3.837219440945298,
      "tokens_seen": 649068544
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008114142427281846,
      "loss": 1.6399,
      "theoretical_loss": 3.8371798676075146,
      "tokens_seen": 649134080
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008113941825476429,
      "loss": 1.5462,
      "theoretical_loss": 3.83714029938337,
      "tokens_seen": 649199616
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008113741223671013,
      "loss": 1.6549,
      "theoretical_loss": 3.837100736271688,
      "tokens_seen": 649265152
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008113540621865597,
      "loss": 1.5815,
      "theoretical_loss": 3.8370611782712922,
      "tokens_seen": 649330688
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000811334002006018,
      "loss": 1.6956,
      "theoretical_loss": 3.8370216253810066,
      "tokens_seen": 649396224
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008113139418254765,
      "loss": 1.7814,
      "theoretical_loss": 3.8369820775996555,
      "tokens_seen": 649461760
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008112938816449348,
      "loss": 1.5998,
      "theoretical_loss": 3.836942534926063,
      "tokens_seen": 649527296
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008112738214643932,
      "loss": 1.5409,
      "theoretical_loss": 3.8369029973590543,
      "tokens_seen": 649592832
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008112537612838516,
      "loss": 1.5699,
      "theoretical_loss": 3.8368634648974558,
      "tokens_seen": 649658368
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008112337011033099,
      "loss": 1.6917,
      "theoretical_loss": 3.8368239375400917,
      "tokens_seen": 649723904
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008112136409227683,
      "loss": 1.6013,
      "theoretical_loss": 3.8367844152857895,
      "tokens_seen": 649789440
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008111935807422268,
      "loss": 1.5228,
      "theoretical_loss": 3.836744898133376,
      "tokens_seen": 649854976
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000811173520561685,
      "loss": 1.6308,
      "theoretical_loss": 3.836705386081676,
      "tokens_seen": 649920512
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008111534603811435,
      "loss": 1.6035,
      "theoretical_loss": 3.83666587912952,
      "tokens_seen": 649986048
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008111334002006017,
      "loss": 1.5487,
      "theoretical_loss": 3.8366263772757336,
      "tokens_seen": 650051584
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008111133400200602,
      "loss": 1.5378,
      "theoretical_loss": 3.8365868805191456,
      "tokens_seen": 650117120
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008110932798395186,
      "loss": 1.5322,
      "theoretical_loss": 3.836547388858585,
      "tokens_seen": 650182656
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008110732196589769,
      "loss": 1.5584,
      "theoretical_loss": 3.8365079022928796,
      "tokens_seen": 650248192
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008110531594784353,
      "loss": 1.7369,
      "theoretical_loss": 3.836468420820859,
      "tokens_seen": 650313728
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008110330992978938,
      "loss": 1.5655,
      "theoretical_loss": 3.836428944441354,
      "tokens_seen": 650379264
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000811013039117352,
      "loss": 1.6268,
      "theoretical_loss": 3.8363894731531936,
      "tokens_seen": 650444800
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008109929789368105,
      "loss": 1.6064,
      "theoretical_loss": 3.8363500069552092,
      "tokens_seen": 650510336
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008109729187562687,
      "loss": 1.6259,
      "theoretical_loss": 3.8363105458462305,
      "tokens_seen": 650575872
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008109528585757272,
      "loss": 1.5815,
      "theoretical_loss": 3.8362710898250896,
      "tokens_seen": 650641408
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008109327983951856,
      "loss": 1.6482,
      "theoretical_loss": 3.8362316388906184,
      "tokens_seen": 650706944
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008109127382146439,
      "loss": 1.5952,
      "theoretical_loss": 3.836192193041647,
      "tokens_seen": 650772480
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008108926780341023,
      "loss": 1.6838,
      "theoretical_loss": 3.8361527522770107,
      "tokens_seen": 650838016
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008108726178535607,
      "loss": 1.5973,
      "theoretical_loss": 3.83611331659554,
      "tokens_seen": 650903552
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000810852557673019,
      "loss": 1.575,
      "theoretical_loss": 3.8360738859960692,
      "tokens_seen": 650969088
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008108324974924775,
      "loss": 1.5135,
      "theoretical_loss": 3.836034460477431,
      "tokens_seen": 651034624
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008108124373119357,
      "loss": 1.4913,
      "theoretical_loss": 3.8359950400384606,
      "tokens_seen": 651100160
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008107923771313942,
      "loss": 1.5501,
      "theoretical_loss": 3.8359556246779913,
      "tokens_seen": 651165696
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008107723169508526,
      "loss": 1.6536,
      "theoretical_loss": 3.8359162143948584,
      "tokens_seen": 651231232
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008107522567703109,
      "loss": 1.7162,
      "theoretical_loss": 3.835876809187896,
      "tokens_seen": 651296768
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008107321965897694,
      "loss": 1.622,
      "theoretical_loss": 3.8358374090559413,
      "tokens_seen": 651362304
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008107121364092277,
      "loss": 1.6346,
      "theoretical_loss": 3.8357980139978283,
      "tokens_seen": 651427840
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008106920762286861,
      "loss": 1.64,
      "theoretical_loss": 3.835758624012395,
      "tokens_seen": 651493376
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008106720160481445,
      "loss": 1.7091,
      "theoretical_loss": 3.835719239098477,
      "tokens_seen": 651558912
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008106519558676029,
      "loss": 1.6619,
      "theoretical_loss": 3.835679859254911,
      "tokens_seen": 651624448
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008106318956870612,
      "loss": 1.6016,
      "theoretical_loss": 3.8356404844805354,
      "tokens_seen": 651689984
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008106118355065196,
      "loss": 1.6643,
      "theoretical_loss": 3.835601114774188,
      "tokens_seen": 651755520
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008105917753259779,
      "loss": 1.6405,
      "theoretical_loss": 3.8355617501347057,
      "tokens_seen": 651821056
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008105717151454364,
      "loss": 1.6398,
      "theoretical_loss": 3.8355223905609277,
      "tokens_seen": 651886592
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008105516549648947,
      "loss": 1.5885,
      "theoretical_loss": 3.835483036051694,
      "tokens_seen": 651952128
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008105315947843531,
      "loss": 1.6238,
      "theoretical_loss": 3.835443686605842,
      "tokens_seen": 652017664
    },
    {
      "epoch": 0.2,
      "objective/train/advantage_avg": 0.49520987272262573,
      "objective/train/docs_used": 242832,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.5519492626190186,
      "objective/train/original_loss": 1.5519492626190186,
      "objective/train/theoretical_loss": 3.8354043422222133,
      "objective/train/tokens_used": 672543200,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24547232687473297,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0507689714431763,
      "objective/train/weighted_lm_loss": 1.6308265924453735,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.8354043422222133,
      "tokens_seen": 652083200
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008105115346038114,
      "loss": 1.5766,
      "theoretical_loss": 3.8354043422222133,
      "tokens_seen": 652083200
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008104914744232699,
      "loss": 1.6648,
      "theoretical_loss": 3.835365002899647,
      "tokens_seen": 652148736
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008104714142427282,
      "loss": 1.5517,
      "theoretical_loss": 3.835325668636983,
      "tokens_seen": 652214272
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008104513540621866,
      "loss": 1.5991,
      "theoretical_loss": 3.835286339433064,
      "tokens_seen": 652279808
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000810431293881645,
      "loss": 1.7126,
      "theoretical_loss": 3.835247015286729,
      "tokens_seen": 652345344
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008104112337011033,
      "loss": 1.6634,
      "theoretical_loss": 3.8352076961968207,
      "tokens_seen": 652410880
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008103911735205617,
      "loss": 1.5613,
      "theoretical_loss": 3.8351683821621814,
      "tokens_seen": 652476416
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008103711133400201,
      "loss": 1.5562,
      "theoretical_loss": 3.8351290731816534,
      "tokens_seen": 652541952
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008103510531594784,
      "loss": 1.6435,
      "theoretical_loss": 3.835089769254079,
      "tokens_seen": 652607488
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008103309929789369,
      "loss": 1.7236,
      "theoretical_loss": 3.8350504703783015,
      "tokens_seen": 652673024
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008103109327983952,
      "loss": 1.6332,
      "theoretical_loss": 3.8350111765531647,
      "tokens_seen": 652738560
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008102908726178536,
      "loss": 1.6058,
      "theoretical_loss": 3.8349718877775123,
      "tokens_seen": 652804096
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000810270812437312,
      "loss": 1.5609,
      "theoretical_loss": 3.8349326040501888,
      "tokens_seen": 652869632
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008102507522567703,
      "loss": 1.6216,
      "theoretical_loss": 3.834893325370038,
      "tokens_seen": 652935168
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008102306920762287,
      "loss": 1.6131,
      "theoretical_loss": 3.834854051735906,
      "tokens_seen": 653000704
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008102106318956872,
      "loss": 1.678,
      "theoretical_loss": 3.834814783146638,
      "tokens_seen": 653066240
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008101905717151454,
      "loss": 1.5165,
      "theoretical_loss": 3.834775519601079,
      "tokens_seen": 653131776
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008101705115346039,
      "loss": 1.7688,
      "theoretical_loss": 3.8347362610980764,
      "tokens_seen": 653197312
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008101504513540621,
      "loss": 1.604,
      "theoretical_loss": 3.8346970076364757,
      "tokens_seen": 653262848
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008101303911735206,
      "loss": 1.5796,
      "theoretical_loss": 3.8346577592151245,
      "tokens_seen": 653328384
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000810110330992979,
      "loss": 1.5307,
      "theoretical_loss": 3.83461851583287,
      "tokens_seen": 653393920
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008100902708124373,
      "loss": 1.6857,
      "theoretical_loss": 3.83457927748856,
      "tokens_seen": 653459456
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008100702106318957,
      "loss": 1.6034,
      "theoretical_loss": 3.8345400441810424,
      "tokens_seen": 653524992
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008100501504513542,
      "loss": 1.619,
      "theoretical_loss": 3.8345008159091654,
      "tokens_seen": 653590528
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008100300902708124,
      "loss": 1.6036,
      "theoretical_loss": 3.8344615926717784,
      "tokens_seen": 653656064
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008100100300902709,
      "loss": 1.6723,
      "theoretical_loss": 3.8344223744677297,
      "tokens_seen": 653721600
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008099899699097291,
      "loss": 1.6536,
      "theoretical_loss": 3.8343831612958703,
      "tokens_seen": 653787136
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008099699097291876,
      "loss": 1.6551,
      "theoretical_loss": 3.834343953155049,
      "tokens_seen": 653852672
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.000809949849548646,
      "loss": 1.5267,
      "theoretical_loss": 3.834304750044117,
      "tokens_seen": 653918208
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008099297893681043,
      "loss": 1.7122,
      "theoretical_loss": 3.8342655519619244,
      "tokens_seen": 653983744
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008099097291875627,
      "loss": 1.6517,
      "theoretical_loss": 3.834226358907322,
      "tokens_seen": 654049280
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008098896690070211,
      "loss": 1.6659,
      "theoretical_loss": 3.8341871708791624,
      "tokens_seen": 654114816
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008098696088264794,
      "loss": 1.5852,
      "theoretical_loss": 3.8341479878762965,
      "tokens_seen": 654180352
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008098495486459379,
      "loss": 1.5873,
      "theoretical_loss": 3.834108809897577,
      "tokens_seen": 654245888
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008098294884653961,
      "loss": 1.6979,
      "theoretical_loss": 3.8340696369418565,
      "tokens_seen": 654311424
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008098094282848546,
      "loss": 1.583,
      "theoretical_loss": 3.834030469007988,
      "tokens_seen": 654376960
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008097893681043129,
      "loss": 1.5965,
      "theoretical_loss": 3.8339913060948243,
      "tokens_seen": 654442496
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008097693079237713,
      "loss": 1.5352,
      "theoretical_loss": 3.83395214820122,
      "tokens_seen": 654508032
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008097492477432297,
      "loss": 1.5438,
      "theoretical_loss": 3.833912995326029,
      "tokens_seen": 654573568
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008097291875626881,
      "loss": 1.6002,
      "theoretical_loss": 3.833873847468105,
      "tokens_seen": 654639104
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008097091273821464,
      "loss": 1.6113,
      "theoretical_loss": 3.8338347046263044,
      "tokens_seen": 654704640
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008096890672016049,
      "loss": 1.492,
      "theoretical_loss": 3.833795566799481,
      "tokens_seen": 654770176
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008096690070210631,
      "loss": 1.6333,
      "theoretical_loss": 3.8337564339864914,
      "tokens_seen": 654835712
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008096489468405216,
      "loss": 1.7008,
      "theoretical_loss": 3.83371730618619,
      "tokens_seen": 654901248
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008096288866599799,
      "loss": 1.6477,
      "theoretical_loss": 3.833678183397436,
      "tokens_seen": 654966784
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008096088264794383,
      "loss": 1.4679,
      "theoretical_loss": 3.833639065619084,
      "tokens_seen": 655032320
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008095887662988967,
      "loss": 1.5969,
      "theoretical_loss": 3.8335999528499913,
      "tokens_seen": 655097856
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008095687061183551,
      "loss": 1.7129,
      "theoretical_loss": 3.833560845089016,
      "tokens_seen": 655163392
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008095486459378134,
      "loss": 1.6872,
      "theoretical_loss": 3.833521742335016,
      "tokens_seen": 655228928
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008095285857572718,
      "loss": 1.5545,
      "theoretical_loss": 3.833482644586849,
      "tokens_seen": 655294464
    },
    {
      "debugging/Compilability": 0.9230769230769231,
      "debugging/distinct-1-grams": 0.7560193640250044,
      "debugging/entropy-1-grams": 5.2534808493234895,
      "debugging/length": 422.2307692307692,
      "debugging/num_segments": 13,
      "debugging/raw_token_scores_avg": 0.00857908371835947,
      "debugging/raw_token_scores_std": 0.02154596894979477,
      "debugging/score": 0.01638830111902192,
      "debugging/score_std": 0.019821935735086043,
      "epoch": 0.2,
      "objective/train/advantage_avg": 0.49142131209373474,
      "objective/train/docs_used": 243976,
      "objective/train/instantaneous_batch_size": 32,
      "objective/train/instantaneous_microbatch_size": 32768,
      "objective/train/lm_loss": 1.465399980545044,
      "objective/train/original_loss": 1.465399980545044,
      "objective/train/theoretical_loss": 3.833443551843374,
      "objective/train/tokens_used": 675820000,
      "objective/train/value_avg": -0.5,
      "objective/train/value_loss": 0.24195925891399384,
      "objective/train/value_max": -0.5,
      "objective/train/value_min": -0.5,
      "objective/train/value_reward_corr": NaN,
      "objective/train/value_std": 0.0,
      "objective/train/weight_avg": 1.0503722429275513,
      "objective/train/weighted_lm_loss": 1.538944125175476,
      "objective/train/weights_max": 1.0512710809707642,
      "objective/train/weights_min": 0.951229453086853,
      "theoretical_loss": 3.833443551843374,
      "tokens_seen": 655360000
    },
    {
      "epoch": 0.2,
      "learning_rate": 0.0008095085255767302,
      "loss": 1.6607,
      "theoretical_loss": 3.833443551843374,
      "tokens_seen": 655360000
    }
  ],
  "max_steps": 50354,
  "num_train_epochs": 9223372036854775807,
  "total_flos": 3.3445682085888e+17,
  "trial_name": null,
  "trial_params": null
}