{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 7.020828457758015,
  "eval_steps": 500,
  "global_step": 60000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0,
      "loss_main": 1.3118659257888794,
      "step": 0
    },
    {
      "epoch": 0.0011701380762930026,
      "grad_norm": 7.4832658767700195,
      "learning_rate": 3.0000000000000004e-07,
      "loss": 0.8758,
      "step": 10
    },
    {
      "epoch": 0.0011701380762930026,
      "loss_main": 0.7866213917732239,
      "step": 10
    },
    {
      "epoch": 0.002340276152586005,
      "grad_norm": 7.719906806945801,
      "learning_rate": 6.333333333333333e-07,
      "loss": 0.9745,
      "step": 20
    },
    {
      "epoch": 0.002340276152586005,
      "loss_main": 0.8006845712661743,
      "step": 20
    },
    {
      "epoch": 0.0035104142288790077,
      "grad_norm": 6.75642728805542,
      "learning_rate": 9.666666666666668e-07,
      "loss": 0.8767,
      "step": 30
    },
    {
      "epoch": 0.0035104142288790077,
      "loss_main": 0.7680040597915649,
      "step": 30
    },
    {
      "epoch": 0.00468055230517201,
      "grad_norm": 5.108084678649902,
      "learning_rate": 1.3e-06,
      "loss": 0.7953,
      "step": 40
    },
    {
      "epoch": 0.00468055230517201,
      "loss_main": 0.6101213693618774,
      "step": 40
    },
    {
      "epoch": 0.005850690381465013,
      "grad_norm": 3.829056739807129,
      "learning_rate": 1.6333333333333333e-06,
      "loss": 0.738,
      "step": 50
    },
    {
      "epoch": 0.005850690381465013,
      "loss_main": 0.803771436214447,
      "step": 50
    },
    {
      "epoch": 0.007020828457758015,
      "grad_norm": 2.967066764831543,
      "learning_rate": 1.9666666666666668e-06,
      "loss": 0.6887,
      "step": 60
    },
    {
      "epoch": 0.007020828457758015,
      "loss_main": 0.4348171055316925,
      "step": 60
    },
    {
      "epoch": 0.008190966534051018,
      "grad_norm": 1.7008675336837769,
      "learning_rate": 2.3e-06,
      "loss": 0.5122,
      "step": 70
    },
    {
      "epoch": 0.008190966534051018,
      "loss_main": 0.6259973645210266,
      "step": 70
    },
    {
      "epoch": 0.00936110461034402,
      "grad_norm": 1.166891098022461,
      "learning_rate": 2.6333333333333337e-06,
      "loss": 0.5563,
      "step": 80
    },
    {
      "epoch": 0.00936110461034402,
      "loss_main": 0.48280858993530273,
      "step": 80
    },
    {
      "epoch": 0.010531242686637023,
      "grad_norm": 1.619026780128479,
      "learning_rate": 2.966666666666667e-06,
      "loss": 0.4647,
      "step": 90
    },
    {
      "epoch": 0.010531242686637023,
      "loss_main": 0.43995317816734314,
      "step": 90
    },
    {
      "epoch": 0.011701380762930026,
      "grad_norm": 1.6124939918518066,
      "learning_rate": 3.3e-06,
      "loss": 0.4197,
      "step": 100
    },
    {
      "epoch": 0.011701380762930026,
      "loss_main": 0.36913323402404785,
      "step": 100
    },
    {
      "epoch": 0.012871518839223028,
      "grad_norm": 1.6711795330047607,
      "learning_rate": 3.633333333333334e-06,
      "loss": 0.3771,
      "step": 110
    },
    {
      "epoch": 0.012871518839223028,
      "loss_main": 0.3178332448005676,
      "step": 110
    },
    {
      "epoch": 0.01404165691551603,
      "grad_norm": 1.6449681520462036,
      "learning_rate": 3.966666666666667e-06,
      "loss": 0.4124,
      "step": 120
    },
    {
      "epoch": 0.01404165691551603,
      "loss_main": 0.34763091802597046,
      "step": 120
    },
    {
      "epoch": 0.015211794991809033,
      "grad_norm": 1.548844575881958,
      "learning_rate": 4.2999999999999995e-06,
      "loss": 0.2955,
      "step": 130
    },
    {
      "epoch": 0.015211794991809033,
      "loss_main": 0.2956778109073639,
      "step": 130
    },
    {
      "epoch": 0.016381933068102036,
      "grad_norm": 1.4397625923156738,
      "learning_rate": 4.633333333333334e-06,
      "loss": 0.3184,
      "step": 140
    },
    {
      "epoch": 0.016381933068102036,
      "loss_main": 0.2914547622203827,
      "step": 140
    },
    {
      "epoch": 0.01755207114439504,
      "grad_norm": 1.0219942331314087,
      "learning_rate": 4.966666666666667e-06,
      "loss": 0.3224,
      "step": 150
    },
    {
      "epoch": 0.01755207114439504,
      "loss_main": 0.44315770268440247,
      "step": 150
    },
    {
      "epoch": 0.01872220922068804,
      "grad_norm": 1.4598621129989624,
      "learning_rate": 5.3e-06,
      "loss": 0.305,
      "step": 160
    },
    {
      "epoch": 0.01872220922068804,
      "loss_main": 0.3237486779689789,
      "step": 160
    },
    {
      "epoch": 0.019892347296981045,
      "grad_norm": 1.901322603225708,
      "learning_rate": 5.633333333333333e-06,
      "loss": 0.2737,
      "step": 170
    },
    {
      "epoch": 0.019892347296981045,
      "loss_main": 0.22261641919612885,
      "step": 170
    },
    {
      "epoch": 0.021062485373274046,
      "grad_norm": 1.4208427667617798,
      "learning_rate": 5.9666666666666666e-06,
      "loss": 0.2604,
      "step": 180
    },
    {
      "epoch": 0.021062485373274046,
      "loss_main": 0.3613899350166321,
      "step": 180
    },
    {
      "epoch": 0.02223262344956705,
      "grad_norm": 1.4195696115493774,
      "learning_rate": 6.300000000000001e-06,
      "loss": 0.251,
      "step": 190
    },
    {
      "epoch": 0.02223262344956705,
      "loss_main": 0.25192996859550476,
      "step": 190
    },
    {
      "epoch": 0.02340276152586005,
      "grad_norm": 1.0215144157409668,
      "learning_rate": 6.633333333333333e-06,
      "loss": 0.2672,
      "step": 200
    },
    {
      "epoch": 0.02340276152586005,
      "loss_main": 0.19020190834999084,
      "step": 200
    },
    {
      "epoch": 0.024572899602153055,
      "grad_norm": 1.3967993259429932,
      "learning_rate": 6.966666666666667e-06,
      "loss": 0.2499,
      "step": 210
    },
    {
      "epoch": 0.024572899602153055,
      "loss_main": 0.29027143120765686,
      "step": 210
    },
    {
      "epoch": 0.025743037678446056,
      "grad_norm": 2.4304120540618896,
      "learning_rate": 7.2999999999999996e-06,
      "loss": 0.2529,
      "step": 220
    },
    {
      "epoch": 0.025743037678446056,
      "loss_main": 0.24103735387325287,
      "step": 220
    },
    {
      "epoch": 0.02691317575473906,
      "grad_norm": 1.519933819770813,
      "learning_rate": 7.633333333333334e-06,
      "loss": 0.2383,
      "step": 230
    },
    {
      "epoch": 0.02691317575473906,
      "loss_main": 0.230159729719162,
      "step": 230
    },
    {
      "epoch": 0.02808331383103206,
      "grad_norm": 1.01149582862854,
      "learning_rate": 7.966666666666666e-06,
      "loss": 0.2386,
      "step": 240
    },
    {
      "epoch": 0.02808331383103206,
      "loss_main": 0.28280720114707947,
      "step": 240
    },
    {
      "epoch": 0.029253451907325066,
      "grad_norm": 1.6116044521331787,
      "learning_rate": 8.3e-06,
      "loss": 0.2289,
      "step": 250
    },
    {
      "epoch": 0.029253451907325066,
      "loss_main": 0.25328901410102844,
      "step": 250
    },
    {
      "epoch": 0.030423589983618066,
      "grad_norm": 1.4556052684783936,
      "learning_rate": 8.633333333333334e-06,
      "loss": 0.2387,
      "step": 260
    },
    {
      "epoch": 0.030423589983618066,
      "loss_main": 0.2230553925037384,
      "step": 260
    },
    {
      "epoch": 0.03159372805991107,
      "grad_norm": 1.0771762132644653,
      "learning_rate": 8.966666666666668e-06,
      "loss": 0.2188,
      "step": 270
    },
    {
      "epoch": 0.03159372805991107,
      "loss_main": 0.229118213057518,
      "step": 270
    },
    {
      "epoch": 0.03276386613620407,
      "grad_norm": 1.2231210470199585,
      "learning_rate": 9.3e-06,
      "loss": 0.2239,
      "step": 280
    },
    {
      "epoch": 0.03276386613620407,
      "loss_main": 0.2217153012752533,
      "step": 280
    },
    {
      "epoch": 0.03393400421249707,
      "grad_norm": 1.4304624795913696,
      "learning_rate": 9.633333333333335e-06,
      "loss": 0.2042,
      "step": 290
    },
    {
      "epoch": 0.03393400421249707,
      "loss_main": 0.22842714190483093,
      "step": 290
    },
    {
      "epoch": 0.03510414228879008,
      "grad_norm": 1.2511450052261353,
      "learning_rate": 9.966666666666667e-06,
      "loss": 0.2156,
      "step": 300
    },
    {
      "epoch": 0.03510414228879008,
      "loss_main": 0.2293686866760254,
      "step": 300
    },
    {
      "epoch": 0.03627428036508308,
      "grad_norm": 0.896489143371582,
      "learning_rate": 1.03e-05,
      "loss": 0.2162,
      "step": 310
    },
    {
      "epoch": 0.03627428036508308,
      "loss_main": 0.18258237838745117,
      "step": 310
    },
    {
      "epoch": 0.03744441844137608,
      "grad_norm": 1.1693602800369263,
      "learning_rate": 1.0633333333333334e-05,
      "loss": 0.2042,
      "step": 320
    },
    {
      "epoch": 0.03744441844137608,
      "loss_main": 0.19026343524456024,
      "step": 320
    },
    {
      "epoch": 0.03861455651766908,
      "grad_norm": 1.1336193084716797,
      "learning_rate": 1.0966666666666666e-05,
      "loss": 0.2076,
      "step": 330
    },
    {
      "epoch": 0.03861455651766908,
      "loss_main": 0.1302378922700882,
      "step": 330
    },
    {
      "epoch": 0.03978469459396209,
      "grad_norm": 1.2481406927108765,
      "learning_rate": 1.13e-05,
      "loss": 0.1773,
      "step": 340
    },
    {
      "epoch": 0.03978469459396209,
      "loss_main": 0.15751712024211884,
      "step": 340
    },
    {
      "epoch": 0.04095483267025509,
      "grad_norm": 1.286823034286499,
      "learning_rate": 1.1633333333333334e-05,
      "loss": 0.2016,
      "step": 350
    },
    {
      "epoch": 0.04095483267025509,
      "loss_main": 0.15316085517406464,
      "step": 350
    },
    {
      "epoch": 0.04212497074654809,
      "grad_norm": 1.5998200178146362,
      "learning_rate": 1.1966666666666668e-05,
      "loss": 0.1957,
      "step": 360
    },
    {
      "epoch": 0.04212497074654809,
      "loss_main": 0.13940615952014923,
      "step": 360
    },
    {
      "epoch": 0.04329510882284109,
      "grad_norm": 1.1124708652496338,
      "learning_rate": 1.23e-05,
      "loss": 0.1603,
      "step": 370
    },
    {
      "epoch": 0.04329510882284109,
      "loss_main": 0.15349875390529633,
      "step": 370
    },
    {
      "epoch": 0.0444652468991341,
      "grad_norm": 0.8221752047538757,
      "learning_rate": 1.2633333333333333e-05,
      "loss": 0.1711,
      "step": 380
    },
    {
      "epoch": 0.0444652468991341,
      "loss_main": 0.16528081893920898,
      "step": 380
    },
    {
      "epoch": 0.0456353849754271,
      "grad_norm": 1.2659629583358765,
      "learning_rate": 1.2966666666666669e-05,
      "loss": 0.2071,
      "step": 390
    },
    {
      "epoch": 0.0456353849754271,
      "loss_main": 0.13139010965824127,
      "step": 390
    },
    {
      "epoch": 0.0468055230517201,
      "grad_norm": 0.9057618975639343,
      "learning_rate": 1.3300000000000001e-05,
      "loss": 0.1632,
      "step": 400
    },
    {
      "epoch": 0.0468055230517201,
      "loss_main": 0.16351215541362762,
      "step": 400
    },
    {
      "epoch": 0.0479756611280131,
      "grad_norm": 0.9332668781280518,
      "learning_rate": 1.3633333333333334e-05,
      "loss": 0.1807,
      "step": 410
    },
    {
      "epoch": 0.0479756611280131,
      "loss_main": 0.14548268914222717,
      "step": 410
    },
    {
      "epoch": 0.04914579920430611,
      "grad_norm": 0.8934863209724426,
      "learning_rate": 1.3966666666666666e-05,
      "loss": 0.1754,
      "step": 420
    },
    {
      "epoch": 0.04914579920430611,
      "loss_main": 0.22217117249965668,
      "step": 420
    },
    {
      "epoch": 0.05031593728059911,
      "grad_norm": 0.7950248122215271,
      "learning_rate": 1.43e-05,
      "loss": 0.1834,
      "step": 430
    },
    {
      "epoch": 0.05031593728059911,
      "loss_main": 0.23015053570270538,
      "step": 430
    },
    {
      "epoch": 0.05148607535689211,
      "grad_norm": 1.0489100217819214,
      "learning_rate": 1.4633333333333334e-05,
      "loss": 0.1705,
      "step": 440
    },
    {
      "epoch": 0.05148607535689211,
      "loss_main": 0.1824006587266922,
      "step": 440
    },
    {
      "epoch": 0.05265621343318511,
      "grad_norm": 1.0346943140029907,
      "learning_rate": 1.4966666666666668e-05,
      "loss": 0.168,
      "step": 450
    },
    {
      "epoch": 0.05265621343318511,
      "loss_main": 0.1451272964477539,
      "step": 450
    },
    {
      "epoch": 0.05382635150947812,
      "grad_norm": 0.856285035610199,
      "learning_rate": 1.53e-05,
      "loss": 0.1607,
      "step": 460
    },
    {
      "epoch": 0.05382635150947812,
      "loss_main": 0.14127948880195618,
      "step": 460
    },
    {
      "epoch": 0.05499648958577112,
      "grad_norm": 1.1480826139450073,
      "learning_rate": 1.563333333333333e-05,
      "loss": 0.1573,
      "step": 470
    },
    {
      "epoch": 0.05499648958577112,
      "loss_main": 0.16239354014396667,
      "step": 470
    },
    {
      "epoch": 0.05616662766206412,
      "grad_norm": 0.7187125086784363,
      "learning_rate": 1.5966666666666667e-05,
      "loss": 0.1669,
      "step": 480
    },
    {
      "epoch": 0.05616662766206412,
      "loss_main": 0.1834668070077896,
      "step": 480
    },
    {
      "epoch": 0.05733676573835712,
      "grad_norm": 1.3918652534484863,
      "learning_rate": 1.63e-05,
      "loss": 0.1621,
      "step": 490
    },
    {
      "epoch": 0.05733676573835712,
      "loss_main": 0.16640540957450867,
      "step": 490
    },
    {
      "epoch": 0.05850690381465013,
      "grad_norm": 1.2490785121917725,
      "learning_rate": 1.6633333333333336e-05,
      "loss": 0.155,
      "step": 500
    },
    {
      "epoch": 0.05850690381465013,
      "loss_main": 0.137198805809021,
      "step": 500
    },
    {
      "epoch": 0.05967704189094313,
      "grad_norm": 0.6911506652832031,
      "learning_rate": 1.6966666666666668e-05,
      "loss": 0.1456,
      "step": 510
    },
    {
      "epoch": 0.05967704189094313,
      "loss_main": 0.11685174703598022,
      "step": 510
    },
    {
      "epoch": 0.06084717996723613,
      "grad_norm": 0.8802915215492249,
      "learning_rate": 1.73e-05,
      "loss": 0.1453,
      "step": 520
    },
    {
      "epoch": 0.06084717996723613,
      "loss_main": 0.15626992285251617,
      "step": 520
    },
    {
      "epoch": 0.062017318043529134,
      "grad_norm": 0.814007580280304,
      "learning_rate": 1.7633333333333336e-05,
      "loss": 0.1488,
      "step": 530
    },
    {
      "epoch": 0.062017318043529134,
      "loss_main": 0.183404341340065,
      "step": 530
    },
    {
      "epoch": 0.06318745611982214,
      "grad_norm": 1.0584241151809692,
      "learning_rate": 1.796666666666667e-05,
      "loss": 0.1668,
      "step": 540
    },
    {
      "epoch": 0.06318745611982214,
      "loss_main": 0.1694537103176117,
      "step": 540
    },
    {
      "epoch": 0.06435759419611514,
      "grad_norm": 1.0961863994598389,
      "learning_rate": 1.83e-05,
      "loss": 0.1565,
      "step": 550
    },
    {
      "epoch": 0.06435759419611514,
      "loss_main": 0.16789133846759796,
      "step": 550
    },
    {
      "epoch": 0.06552773227240814,
      "grad_norm": 0.8200808763504028,
      "learning_rate": 1.8633333333333333e-05,
      "loss": 0.1415,
      "step": 560
    },
    {
      "epoch": 0.06552773227240814,
      "loss_main": 0.1592937856912613,
      "step": 560
    },
    {
      "epoch": 0.06669787034870114,
      "grad_norm": 0.8449012637138367,
      "learning_rate": 1.896666666666667e-05,
      "loss": 0.1639,
      "step": 570
    },
    {
      "epoch": 0.06669787034870114,
      "loss_main": 0.11675786972045898,
      "step": 570
    },
    {
      "epoch": 0.06786800842499414,
      "grad_norm": 0.6697084903717041,
      "learning_rate": 1.93e-05,
      "loss": 0.1462,
      "step": 580
    },
    {
      "epoch": 0.06786800842499414,
      "loss_main": 0.15099918842315674,
      "step": 580
    },
    {
      "epoch": 0.06903814650128715,
      "grad_norm": 0.7039180994033813,
      "learning_rate": 1.9633333333333334e-05,
      "loss": 0.1344,
      "step": 590
    },
    {
      "epoch": 0.06903814650128715,
      "loss_main": 0.19685380160808563,
      "step": 590
    },
    {
      "epoch": 0.07020828457758016,
      "grad_norm": 0.6827440857887268,
      "learning_rate": 1.9966666666666666e-05,
      "loss": 0.1434,
      "step": 600
    },
    {
      "epoch": 0.07020828457758016,
      "loss_main": 0.17982080578804016,
      "step": 600
    },
    {
      "epoch": 0.07137842265387316,
      "grad_norm": 1.1890445947647095,
      "learning_rate": 2.0300000000000002e-05,
      "loss": 0.1551,
      "step": 610
    },
    {
      "epoch": 0.07137842265387316,
      "loss_main": 0.14751867949962616,
      "step": 610
    },
    {
      "epoch": 0.07254856073016616,
      "grad_norm": 1.0832898616790771,
      "learning_rate": 2.0633333333333335e-05,
      "loss": 0.1529,
      "step": 620
    },
    {
      "epoch": 0.07254856073016616,
      "loss_main": 0.15180251002311707,
      "step": 620
    },
    {
      "epoch": 0.07371869880645916,
      "grad_norm": 0.9993995428085327,
      "learning_rate": 2.0966666666666667e-05,
      "loss": 0.1454,
      "step": 630
    },
    {
      "epoch": 0.07371869880645916,
      "loss_main": 0.09265244007110596,
      "step": 630
    },
    {
      "epoch": 0.07488883688275216,
      "grad_norm": 1.14290452003479,
      "learning_rate": 2.13e-05,
      "loss": 0.1424,
      "step": 640
    },
    {
      "epoch": 0.07488883688275216,
      "loss_main": 0.1313002109527588,
      "step": 640
    },
    {
      "epoch": 0.07605897495904516,
      "grad_norm": 0.7458564639091492,
      "learning_rate": 2.1633333333333332e-05,
      "loss": 0.1349,
      "step": 650
    },
    {
      "epoch": 0.07605897495904516,
      "loss_main": 0.08957183361053467,
      "step": 650
    },
    {
      "epoch": 0.07722911303533816,
      "grad_norm": 0.6314276456832886,
      "learning_rate": 2.1966666666666668e-05,
      "loss": 0.135,
      "step": 660
    },
    {
      "epoch": 0.07722911303533816,
      "loss_main": 0.12112297117710114,
      "step": 660
    },
    {
      "epoch": 0.07839925111163117,
      "grad_norm": 0.6476473212242126,
      "learning_rate": 2.23e-05,
      "loss": 0.1285,
      "step": 670
    },
    {
      "epoch": 0.07839925111163117,
      "loss_main": 0.15594825148582458,
      "step": 670
    },
    {
      "epoch": 0.07956938918792418,
      "grad_norm": 0.6191851496696472,
      "learning_rate": 2.2633333333333336e-05,
      "loss": 0.137,
      "step": 680
    },
    {
      "epoch": 0.07956938918792418,
      "loss_main": 0.10984550416469574,
      "step": 680
    },
    {
      "epoch": 0.08073952726421718,
      "grad_norm": 0.6643352508544922,
      "learning_rate": 2.2966666666666668e-05,
      "loss": 0.1217,
      "step": 690
    },
    {
      "epoch": 0.08073952726421718,
      "loss_main": 0.10490590333938599,
      "step": 690
    },
    {
      "epoch": 0.08190966534051018,
      "grad_norm": 0.6610481142997742,
      "learning_rate": 2.3300000000000004e-05,
      "loss": 0.123,
      "step": 700
    },
    {
      "epoch": 0.08190966534051018,
      "loss_main": 0.10028455406427383,
      "step": 700
    },
    {
      "epoch": 0.08307980341680318,
      "grad_norm": 1.0067617893218994,
      "learning_rate": 2.3633333333333336e-05,
      "loss": 0.1321,
      "step": 710
    },
    {
      "epoch": 0.08307980341680318,
      "loss_main": 0.10476034134626389,
      "step": 710
    },
    {
      "epoch": 0.08424994149309618,
      "grad_norm": 0.9463035464286804,
      "learning_rate": 2.396666666666667e-05,
      "loss": 0.1261,
      "step": 720
    },
    {
      "epoch": 0.08424994149309618,
      "loss_main": 0.14547841250896454,
      "step": 720
    },
    {
      "epoch": 0.08542007956938918,
      "grad_norm": 0.7988779544830322,
      "learning_rate": 2.43e-05,
      "loss": 0.1412,
      "step": 730
    },
    {
      "epoch": 0.08542007956938918,
      "loss_main": 0.13443835079669952,
      "step": 730
    },
    {
      "epoch": 0.08659021764568219,
      "grad_norm": 0.9266788959503174,
      "learning_rate": 2.4633333333333334e-05,
      "loss": 0.1311,
      "step": 740
    },
    {
      "epoch": 0.08659021764568219,
      "loss_main": 0.09917716681957245,
      "step": 740
    },
    {
      "epoch": 0.08776035572197519,
      "grad_norm": 1.1143115758895874,
      "learning_rate": 2.496666666666667e-05,
      "loss": 0.1223,
      "step": 750
    },
    {
      "epoch": 0.08776035572197519,
      "loss_main": 0.1635911613702774,
      "step": 750
    },
    {
      "epoch": 0.0889304937982682,
      "grad_norm": 0.6741898059844971,
      "learning_rate": 2.5300000000000002e-05,
      "loss": 0.1293,
      "step": 760
    },
    {
      "epoch": 0.0889304937982682,
      "loss_main": 0.1354270875453949,
      "step": 760
    },
    {
      "epoch": 0.0901006318745612,
      "grad_norm": 0.6160279512405396,
      "learning_rate": 2.5633333333333338e-05,
      "loss": 0.1225,
      "step": 770
    },
    {
      "epoch": 0.0901006318745612,
      "loss_main": 0.12260179221630096,
      "step": 770
    },
    {
      "epoch": 0.0912707699508542,
      "grad_norm": 0.8080768585205078,
      "learning_rate": 2.5966666666666667e-05,
      "loss": 0.1278,
      "step": 780
    },
    {
      "epoch": 0.0912707699508542,
      "loss_main": 0.056064099073410034,
      "step": 780
    },
    {
      "epoch": 0.0924409080271472,
      "grad_norm": 0.918973445892334,
      "learning_rate": 2.6300000000000002e-05,
      "loss": 0.1134,
      "step": 790
    },
    {
      "epoch": 0.0924409080271472,
      "loss_main": 0.08106803148984909,
      "step": 790
    },
    {
      "epoch": 0.0936110461034402,
      "grad_norm": 0.8992382287979126,
      "learning_rate": 2.663333333333333e-05,
      "loss": 0.1309,
      "step": 800
    },
    {
      "epoch": 0.0936110461034402,
      "loss_main": 0.09109961986541748,
      "step": 800
    },
    {
      "epoch": 0.0947811841797332,
      "grad_norm": 0.85886549949646,
      "learning_rate": 2.6966666666666667e-05,
      "loss": 0.1197,
      "step": 810
    },
    {
      "epoch": 0.0947811841797332,
      "loss_main": 0.14947451651096344,
      "step": 810
    },
    {
      "epoch": 0.0959513222560262,
      "grad_norm": 0.8436309099197388,
      "learning_rate": 2.7300000000000003e-05,
      "loss": 0.1476,
      "step": 820
    },
    {
      "epoch": 0.0959513222560262,
      "loss_main": 0.08063036948442459,
      "step": 820
    },
    {
      "epoch": 0.09712146033231922,
      "grad_norm": 1.0066734552383423,
      "learning_rate": 2.7633333333333332e-05,
      "loss": 0.1297,
      "step": 830
    },
    {
      "epoch": 0.09712146033231922,
      "loss_main": 0.1502363234758377,
      "step": 830
    },
    {
      "epoch": 0.09829159840861222,
      "grad_norm": 0.7125228047370911,
      "learning_rate": 2.7966666666666668e-05,
      "loss": 0.1123,
      "step": 840
    },
    {
      "epoch": 0.09829159840861222,
      "loss_main": 0.12545007467269897,
      "step": 840
    },
    {
      "epoch": 0.09946173648490522,
      "grad_norm": 0.7807751297950745,
      "learning_rate": 2.83e-05,
      "loss": 0.1177,
      "step": 850
    },
    {
      "epoch": 0.09946173648490522,
      "loss_main": 0.14880172908306122,
      "step": 850
    },
    {
      "epoch": 0.10063187456119822,
      "grad_norm": 0.5642628073692322,
      "learning_rate": 2.8633333333333336e-05,
      "loss": 0.1157,
      "step": 860
    },
    {
      "epoch": 0.10063187456119822,
      "loss_main": 0.12849828600883484,
      "step": 860
    },
    {
      "epoch": 0.10180201263749122,
      "grad_norm": 1.0525195598602295,
      "learning_rate": 2.8966666666666668e-05,
      "loss": 0.1267,
      "step": 870
    },
    {
      "epoch": 0.10180201263749122,
      "loss_main": 0.1632884442806244,
      "step": 870
    },
    {
      "epoch": 0.10297215071378422,
      "grad_norm": 0.7371509075164795,
      "learning_rate": 2.93e-05,
      "loss": 0.1255,
      "step": 880
    },
    {
      "epoch": 0.10297215071378422,
      "loss_main": 0.16964833438396454,
      "step": 880
    },
    {
      "epoch": 0.10414228879007723,
      "grad_norm": 0.9985182881355286,
      "learning_rate": 2.9633333333333336e-05,
      "loss": 0.1249,
      "step": 890
    },
    {
      "epoch": 0.10414228879007723,
      "loss_main": 0.11398185789585114,
      "step": 890
    },
    {
      "epoch": 0.10531242686637023,
      "grad_norm": 0.7123116254806519,
      "learning_rate": 2.9966666666666672e-05,
      "loss": 0.1191,
      "step": 900
    },
    {
      "epoch": 0.10531242686637023,
      "loss_main": 0.11918755620718002,
      "step": 900
    },
    {
      "epoch": 0.10648256494266324,
      "grad_norm": 0.676751971244812,
      "learning_rate": 3.03e-05,
      "loss": 0.1239,
      "step": 910
    },
    {
      "epoch": 0.10648256494266324,
      "loss_main": 0.12953677773475647,
      "step": 910
    },
    {
      "epoch": 0.10765270301895624,
      "grad_norm": 0.614718496799469,
      "learning_rate": 3.063333333333334e-05,
      "loss": 0.1326,
      "step": 920
    },
    {
      "epoch": 0.10765270301895624,
      "loss_main": 0.1414894163608551,
      "step": 920
    },
    {
      "epoch": 0.10882284109524924,
      "grad_norm": 0.6427780389785767,
      "learning_rate": 3.096666666666666e-05,
      "loss": 0.1446,
      "step": 930
    },
    {
      "epoch": 0.10882284109524924,
      "loss_main": 0.11148513853549957,
      "step": 930
    },
    {
      "epoch": 0.10999297917154224,
      "grad_norm": 0.5846249461174011,
      "learning_rate": 3.13e-05,
      "loss": 0.1114,
      "step": 940
    },
    {
      "epoch": 0.10999297917154224,
      "loss_main": 0.13100917637348175,
      "step": 940
    },
    {
      "epoch": 0.11116311724783524,
      "grad_norm": 0.6403881907463074,
      "learning_rate": 3.1633333333333334e-05,
      "loss": 0.1186,
      "step": 950
    },
    {
      "epoch": 0.11116311724783524,
      "loss_main": 0.12420495599508286,
      "step": 950
    },
    {
      "epoch": 0.11233325532412824,
      "grad_norm": 0.717736542224884,
      "learning_rate": 3.196666666666667e-05,
      "loss": 0.1222,
      "step": 960
    },
    {
      "epoch": 0.11233325532412824,
      "loss_main": 0.1156066507101059,
      "step": 960
    },
    {
      "epoch": 0.11350339340042125,
      "grad_norm": 0.4944027364253998,
      "learning_rate": 3.2300000000000006e-05,
      "loss": 0.1122,
      "step": 970
    },
    {
      "epoch": 0.11350339340042125,
      "loss_main": 0.07993145287036896,
      "step": 970
    },
    {
      "epoch": 0.11467353147671425,
      "grad_norm": 0.810749888420105,
      "learning_rate": 3.263333333333333e-05,
      "loss": 0.1218,
      "step": 980
    },
    {
      "epoch": 0.11467353147671425,
      "loss_main": 0.10390225797891617,
      "step": 980
    },
    {
      "epoch": 0.11584366955300726,
      "grad_norm": 0.8177343010902405,
      "learning_rate": 3.296666666666667e-05,
      "loss": 0.1123,
      "step": 990
    },
    {
      "epoch": 0.11584366955300726,
      "loss_main": 0.07213989645242691,
      "step": 990
    },
    {
      "epoch": 0.11701380762930026,
      "grad_norm": 0.7034750580787659,
      "learning_rate": 3.33e-05,
      "loss": 0.1102,
      "step": 1000
    },
    {
      "epoch": 0.11701380762930026,
      "loss_main": 0.12191074341535568,
      "step": 1000
    },
    {
      "epoch": 0.11818394570559326,
      "grad_norm": 0.5590371489524841,
      "learning_rate": 3.3633333333333335e-05,
      "loss": 0.1357,
      "step": 1010
    },
    {
      "epoch": 0.11818394570559326,
      "loss_main": 0.059432175010442734,
      "step": 1010
    },
    {
      "epoch": 0.11935408378188626,
      "grad_norm": 0.5535507798194885,
      "learning_rate": 3.396666666666667e-05,
      "loss": 0.1194,
      "step": 1020
    },
    {
      "epoch": 0.11935408378188626,
      "loss_main": 0.09121101349592209,
      "step": 1020
    },
    {
      "epoch": 0.12052422185817926,
      "grad_norm": 0.703208863735199,
      "learning_rate": 3.430000000000001e-05,
      "loss": 0.1233,
      "step": 1030
    },
    {
      "epoch": 0.12052422185817926,
      "loss_main": 0.13843227922916412,
      "step": 1030
    },
    {
      "epoch": 0.12169435993447227,
      "grad_norm": 0.9112261533737183,
      "learning_rate": 3.463333333333333e-05,
      "loss": 0.1162,
      "step": 1040
    },
    {
      "epoch": 0.12169435993447227,
      "loss_main": 0.09944566339254379,
      "step": 1040
    },
    {
      "epoch": 0.12286449801076527,
      "grad_norm": 0.6699824333190918,
      "learning_rate": 3.496666666666667e-05,
      "loss": 0.1066,
      "step": 1050
    },
    {
      "epoch": 0.12286449801076527,
      "loss_main": 0.10237077623605728,
      "step": 1050
    },
    {
      "epoch": 0.12403463608705827,
      "grad_norm": 0.7975031733512878,
      "learning_rate": 3.53e-05,
      "loss": 0.1234,
      "step": 1060
    },
    {
      "epoch": 0.12403463608705827,
      "loss_main": 0.0827983021736145,
      "step": 1060
    },
    {
      "epoch": 0.12520477416335127,
      "grad_norm": 0.6100794672966003,
      "learning_rate": 3.563333333333334e-05,
      "loss": 0.0965,
      "step": 1070
    },
    {
      "epoch": 0.12520477416335127,
      "loss_main": 0.12455994635820389,
      "step": 1070
    },
    {
      "epoch": 0.12637491223964428,
      "grad_norm": 0.599738597869873,
      "learning_rate": 3.596666666666667e-05,
      "loss": 0.1074,
      "step": 1080
    },
    {
      "epoch": 0.12637491223964428,
      "loss_main": 0.10892277210950851,
      "step": 1080
    },
    {
      "epoch": 0.12754505031593727,
      "grad_norm": 0.6158097386360168,
      "learning_rate": 3.63e-05,
      "loss": 0.1142,
      "step": 1090
    },
    {
      "epoch": 0.12754505031593727,
      "loss_main": 0.1495901346206665,
      "step": 1090
    },
    {
      "epoch": 0.12871518839223028,
      "grad_norm": 0.8958266973495483,
      "learning_rate": 3.6633333333333334e-05,
      "loss": 0.1119,
      "step": 1100
    },
    {
      "epoch": 0.12871518839223028,
      "loss_main": 0.11430372297763824,
      "step": 1100
    },
    {
      "epoch": 0.1298853264685233,
      "grad_norm": 0.5079343914985657,
      "learning_rate": 3.6966666666666666e-05,
      "loss": 0.1118,
      "step": 1110
    },
    {
      "epoch": 0.1298853264685233,
      "loss_main": 0.12031161785125732,
      "step": 1110
    },
    {
      "epoch": 0.13105546454481629,
      "grad_norm": 0.8783027529716492,
      "learning_rate": 3.73e-05,
      "loss": 0.1124,
      "step": 1120
    },
    {
      "epoch": 0.13105546454481629,
      "loss_main": 0.17541396617889404,
      "step": 1120
    },
    {
      "epoch": 0.1322256026211093,
      "grad_norm": 0.7274705767631531,
      "learning_rate": 3.763333333333334e-05,
      "loss": 0.1163,
      "step": 1130
    },
    {
      "epoch": 0.1322256026211093,
      "loss_main": 0.10474052280187607,
      "step": 1130
    },
    {
      "epoch": 0.1333957406974023,
      "grad_norm": 0.7223723530769348,
      "learning_rate": 3.796666666666667e-05,
      "loss": 0.1106,
      "step": 1140
    },
    {
      "epoch": 0.1333957406974023,
      "loss_main": 0.16925454139709473,
      "step": 1140
    },
    {
      "epoch": 0.1345658787736953,
      "grad_norm": 0.6074584126472473,
      "learning_rate": 3.83e-05,
      "loss": 0.1145,
      "step": 1150
    },
    {
      "epoch": 0.1345658787736953,
      "loss_main": 0.12774138152599335,
      "step": 1150
    },
    {
      "epoch": 0.1357360168499883,
      "grad_norm": 1.0059070587158203,
      "learning_rate": 3.8633333333333335e-05,
      "loss": 0.1047,
      "step": 1160
    },
    {
      "epoch": 0.1357360168499883,
      "loss_main": 0.1720617562532425,
      "step": 1160
    },
    {
      "epoch": 0.1369061549262813,
      "grad_norm": 0.6709736585617065,
      "learning_rate": 3.896666666666667e-05,
      "loss": 0.1238,
      "step": 1170
    },
    {
      "epoch": 0.1369061549262813,
      "loss_main": 0.16238287091255188,
      "step": 1170
    },
    {
      "epoch": 0.1380762930025743,
      "grad_norm": 0.8064225912094116,
      "learning_rate": 3.9300000000000007e-05,
      "loss": 0.1287,
      "step": 1180
    },
    {
      "epoch": 0.1380762930025743,
      "loss_main": 0.10779585689306259,
      "step": 1180
    },
    {
      "epoch": 0.1392464310788673,
      "grad_norm": 0.5042034387588501,
      "learning_rate": 3.963333333333333e-05,
      "loss": 0.1156,
      "step": 1190
    },
    {
      "epoch": 0.1392464310788673,
      "loss_main": 0.1457672119140625,
      "step": 1190
    },
    {
      "epoch": 0.14041656915516032,
      "grad_norm": 0.6666590571403503,
      "learning_rate": 3.996666666666667e-05,
      "loss": 0.1237,
      "step": 1200
    },
    {
      "epoch": 0.14041656915516032,
      "loss_main": 0.13245782256126404,
      "step": 1200
    },
    {
      "epoch": 0.1415867072314533,
      "grad_norm": 0.6711907386779785,
      "learning_rate": 4.0300000000000004e-05,
      "loss": 0.1278,
      "step": 1210
    },
    {
      "epoch": 0.1415867072314533,
      "loss_main": 0.1356053203344345,
      "step": 1210
    },
    {
      "epoch": 0.14275684530774632,
      "grad_norm": 0.8520138263702393,
      "learning_rate": 4.0633333333333336e-05,
      "loss": 0.1229,
      "step": 1220
    },
    {
      "epoch": 0.14275684530774632,
      "loss_main": 0.1272735893726349,
      "step": 1220
    },
    {
      "epoch": 0.1439269833840393,
      "grad_norm": 0.5209691524505615,
      "learning_rate": 4.096666666666667e-05,
      "loss": 0.1104,
      "step": 1230
    },
    {
      "epoch": 0.1439269833840393,
      "loss_main": 0.09621458500623703,
      "step": 1230
    },
    {
      "epoch": 0.14509712146033232,
      "grad_norm": 0.8444809317588806,
      "learning_rate": 4.13e-05,
      "loss": 0.1213,
      "step": 1240
    },
    {
      "epoch": 0.14509712146033232,
      "loss_main": 0.12143150717020035,
      "step": 1240
    },
    {
      "epoch": 0.1462672595366253,
      "grad_norm": 0.6381915211677551,
      "learning_rate": 4.1633333333333333e-05,
      "loss": 0.098,
      "step": 1250
    },
    {
      "epoch": 0.1462672595366253,
      "loss_main": 0.08326148241758347,
      "step": 1250
    },
    {
      "epoch": 0.14743739761291833,
      "grad_norm": 0.425995409488678,
      "learning_rate": 4.196666666666667e-05,
      "loss": 0.0924,
      "step": 1260
    },
    {
      "epoch": 0.14743739761291833,
      "loss_main": 0.07839003950357437,
      "step": 1260
    },
    {
      "epoch": 0.14860753568921134,
      "grad_norm": 0.5994794964790344,
      "learning_rate": 4.23e-05,
      "loss": 0.1056,
      "step": 1270
    },
    {
      "epoch": 0.14860753568921134,
      "loss_main": 0.116432785987854,
      "step": 1270
    },
    {
      "epoch": 0.14977767376550433,
      "grad_norm": 0.5673111081123352,
      "learning_rate": 4.263333333333334e-05,
      "loss": 0.1143,
      "step": 1280
    },
    {
      "epoch": 0.14977767376550433,
      "loss_main": 0.0775914341211319,
      "step": 1280
    },
    {
      "epoch": 0.15094781184179734,
      "grad_norm": 0.7320958971977234,
      "learning_rate": 4.296666666666666e-05,
      "loss": 0.11,
      "step": 1290
    },
    {
      "epoch": 0.15094781184179734,
      "loss_main": 0.15974394977092743,
      "step": 1290
    },
    {
      "epoch": 0.15211794991809033,
      "grad_norm": 0.7012493014335632,
      "learning_rate": 4.33e-05,
      "loss": 0.1201,
      "step": 1300
    },
    {
      "epoch": 0.15211794991809033,
      "loss_main": 0.08645652234554291,
      "step": 1300
    },
    {
      "epoch": 0.15328808799438334,
      "grad_norm": 0.9551940560340881,
      "learning_rate": 4.3633333333333335e-05,
      "loss": 0.1087,
      "step": 1310
    },
    {
      "epoch": 0.15328808799438334,
      "loss_main": 0.12089089304208755,
      "step": 1310
    },
    {
      "epoch": 0.15445822607067633,
      "grad_norm": 0.6384814977645874,
      "learning_rate": 4.396666666666667e-05,
      "loss": 0.1123,
      "step": 1320
    },
    {
      "epoch": 0.15445822607067633,
      "loss_main": 0.13810673356056213,
      "step": 1320
    },
    {
      "epoch": 0.15562836414696934,
      "grad_norm": 0.6525187492370605,
      "learning_rate": 4.43e-05,
      "loss": 0.1137,
      "step": 1330
    },
    {
      "epoch": 0.15562836414696934,
      "loss_main": 0.158315971493721,
      "step": 1330
    },
    {
      "epoch": 0.15679850222326233,
      "grad_norm": 0.8275865912437439,
      "learning_rate": 4.463333333333334e-05,
      "loss": 0.1349,
      "step": 1340
    },
    {
      "epoch": 0.15679850222326233,
      "loss_main": 0.17283928394317627,
      "step": 1340
    },
    {
      "epoch": 0.15796864029955535,
      "grad_norm": 0.8050822615623474,
      "learning_rate": 4.496666666666667e-05,
      "loss": 0.1116,
      "step": 1350
    },
    {
      "epoch": 0.15796864029955535,
      "loss_main": 0.06593068689107895,
      "step": 1350
    },
    {
      "epoch": 0.15913877837584836,
      "grad_norm": 0.8809645175933838,
      "learning_rate": 4.53e-05,
      "loss": 0.1192,
      "step": 1360
    },
    {
      "epoch": 0.15913877837584836,
      "loss_main": 0.08210869878530502,
      "step": 1360
    },
    {
      "epoch": 0.16030891645214135,
      "grad_norm": 0.5857805609703064,
      "learning_rate": 4.5633333333333336e-05,
      "loss": 0.1128,
      "step": 1370
    },
    {
      "epoch": 0.16030891645214135,
      "loss_main": 0.08791517466306686,
      "step": 1370
    },
    {
      "epoch": 0.16147905452843436,
      "grad_norm": 1.0119092464447021,
      "learning_rate": 4.596666666666667e-05,
      "loss": 0.1139,
      "step": 1380
    },
    {
      "epoch": 0.16147905452843436,
      "loss_main": 0.08101890981197357,
      "step": 1380
    },
    {
      "epoch": 0.16264919260472735,
      "grad_norm": 0.5822144150733948,
      "learning_rate": 4.630000000000001e-05,
      "loss": 0.1146,
      "step": 1390
    },
    {
      "epoch": 0.16264919260472735,
      "loss_main": 0.16079626977443695,
      "step": 1390
    },
    {
      "epoch": 0.16381933068102036,
      "grad_norm": 0.8024778366088867,
      "learning_rate": 4.663333333333333e-05,
      "loss": 0.1145,
      "step": 1400
    },
    {
      "epoch": 0.16381933068102036,
      "loss_main": 0.12091147899627686,
      "step": 1400
    },
    {
      "epoch": 0.16498946875731335,
      "grad_norm": 0.7615772485733032,
      "learning_rate": 4.696666666666667e-05,
      "loss": 0.1302,
      "step": 1410
    },
    {
      "epoch": 0.16498946875731335,
      "loss_main": 0.13355378806591034,
      "step": 1410
    },
    {
      "epoch": 0.16615960683360637,
      "grad_norm": 0.47477859258651733,
      "learning_rate": 4.73e-05,
      "loss": 0.1181,
      "step": 1420
    },
    {
      "epoch": 0.16615960683360637,
      "loss_main": 0.16651436686515808,
      "step": 1420
    },
    {
      "epoch": 0.16732974490989938,
      "grad_norm": 0.5404967069625854,
      "learning_rate": 4.763333333333334e-05,
      "loss": 0.1163,
      "step": 1430
    },
    {
      "epoch": 0.16732974490989938,
      "loss_main": 0.12307556718587875,
      "step": 1430
    },
    {
      "epoch": 0.16849988298619237,
      "grad_norm": 0.6808135509490967,
      "learning_rate": 4.796666666666667e-05,
      "loss": 0.1048,
      "step": 1440
    },
    {
      "epoch": 0.16849988298619237,
      "loss_main": 0.17491303384304047,
      "step": 1440
    },
    {
      "epoch": 0.16967002106248538,
      "grad_norm": 0.7537589073181152,
      "learning_rate": 4.83e-05,
      "loss": 0.1106,
      "step": 1450
    },
    {
      "epoch": 0.16967002106248538,
      "loss_main": 0.12848827242851257,
      "step": 1450
    },
    {
      "epoch": 0.17084015913877837,
      "grad_norm": 0.7064046263694763,
      "learning_rate": 4.8633333333333334e-05,
      "loss": 0.1075,
      "step": 1460
    },
    {
      "epoch": 0.17084015913877837,
      "loss_main": 0.10712490975856781,
      "step": 1460
    },
    {
      "epoch": 0.17201029721507138,
      "grad_norm": 0.516555666923523,
      "learning_rate": 4.8966666666666667e-05,
      "loss": 0.0969,
      "step": 1470
    },
    {
      "epoch": 0.17201029721507138,
      "loss_main": 0.08735869824886322,
      "step": 1470
    },
    {
      "epoch": 0.17318043529136437,
      "grad_norm": 0.597628653049469,
      "learning_rate": 4.93e-05,
      "loss": 0.1027,
      "step": 1480
    },
    {
      "epoch": 0.17318043529136437,
      "loss_main": 0.06888429820537567,
      "step": 1480
    },
    {
      "epoch": 0.17435057336765739,
      "grad_norm": 0.8060656189918518,
      "learning_rate": 4.963333333333334e-05,
      "loss": 0.1111,
      "step": 1490
    },
    {
      "epoch": 0.17435057336765739,
      "loss_main": 0.12881258130073547,
      "step": 1490
    },
    {
      "epoch": 0.17552071144395037,
      "grad_norm": 0.5726759433746338,
      "learning_rate": 4.996666666666667e-05,
      "loss": 0.1155,
      "step": 1500
    },
    {
      "epoch": 0.17552071144395037,
      "loss_main": 0.10701531916856766,
      "step": 1500
    },
    {
      "epoch": 0.1766908495202434,
      "grad_norm": 0.6242645382881165,
      "learning_rate": 5.03e-05,
      "loss": 0.0884,
      "step": 1510
    },
    {
      "epoch": 0.1766908495202434,
      "loss_main": 0.09842284768819809,
      "step": 1510
    },
    {
      "epoch": 0.1778609875965364,
      "grad_norm": 0.49510473012924194,
      "learning_rate": 5.0633333333333335e-05,
      "loss": 0.1049,
      "step": 1520
    },
    {
      "epoch": 0.1778609875965364,
      "loss_main": 0.08731546252965927,
      "step": 1520
    },
    {
      "epoch": 0.1790311256728294,
      "grad_norm": 0.870434045791626,
      "learning_rate": 5.0966666666666674e-05,
      "loss": 0.1164,
      "step": 1530
    },
    {
      "epoch": 0.1790311256728294,
      "loss_main": 0.08948785811662674,
      "step": 1530
    },
    {
      "epoch": 0.1802012637491224,
      "grad_norm": 0.6957693696022034,
      "learning_rate": 5.130000000000001e-05,
      "loss": 0.1046,
      "step": 1540
    },
    {
      "epoch": 0.1802012637491224,
      "loss_main": 0.08836889266967773,
      "step": 1540
    },
    {
      "epoch": 0.1813714018254154,
      "grad_norm": 0.7772489190101624,
      "learning_rate": 5.163333333333333e-05,
      "loss": 0.1212,
      "step": 1550
    },
    {
      "epoch": 0.1813714018254154,
      "loss_main": 0.07521868497133255,
      "step": 1550
    },
    {
      "epoch": 0.1825415399017084,
      "grad_norm": 0.5978929996490479,
      "learning_rate": 5.196666666666667e-05,
      "loss": 0.095,
      "step": 1560
    },
    {
      "epoch": 0.1825415399017084,
      "loss_main": 0.10068494826555252,
      "step": 1560
    },
    {
      "epoch": 0.1837116779780014,
      "grad_norm": 0.5466823577880859,
      "learning_rate": 5.2300000000000004e-05,
      "loss": 0.0995,
      "step": 1570
    },
    {
      "epoch": 0.1837116779780014,
      "loss_main": 0.11040567606687546,
      "step": 1570
    },
    {
      "epoch": 0.1848818160542944,
      "grad_norm": 0.876833438873291,
      "learning_rate": 5.2633333333333336e-05,
      "loss": 0.1064,
      "step": 1580
    },
    {
      "epoch": 0.1848818160542944,
      "loss_main": 0.13484306633472443,
      "step": 1580
    },
    {
      "epoch": 0.18605195413058742,
      "grad_norm": 0.5120718479156494,
      "learning_rate": 5.296666666666666e-05,
      "loss": 0.1162,
      "step": 1590
    },
    {
      "epoch": 0.18605195413058742,
      "loss_main": 0.08490284532308578,
      "step": 1590
    },
    {
      "epoch": 0.1872220922068804,
      "grad_norm": 0.4015161097049713,
      "learning_rate": 5.330000000000001e-05,
      "loss": 0.1106,
      "step": 1600
    },
    {
      "epoch": 0.1872220922068804,
      "loss_main": 0.11179093271493912,
      "step": 1600
    },
    {
      "epoch": 0.18839223028317342,
      "grad_norm": 0.6660275459289551,
      "learning_rate": 5.3633333333333334e-05,
      "loss": 0.1071,
      "step": 1610
    },
    {
      "epoch": 0.18839223028317342,
      "loss_main": 0.09974606335163116,
      "step": 1610
    },
    {
      "epoch": 0.1895623683594664,
      "grad_norm": 0.42296460270881653,
      "learning_rate": 5.3966666666666666e-05,
      "loss": 0.1121,
      "step": 1620
    },
    {
      "epoch": 0.1895623683594664,
      "loss_main": 0.0834658294916153,
      "step": 1620
    },
    {
      "epoch": 0.19073250643575942,
      "grad_norm": 0.5009296536445618,
      "learning_rate": 5.4300000000000005e-05,
      "loss": 0.1037,
      "step": 1630
    },
    {
      "epoch": 0.19073250643575942,
      "loss_main": 0.06143859401345253,
      "step": 1630
    },
    {
      "epoch": 0.1919026445120524,
      "grad_norm": 0.7592416405677795,
      "learning_rate": 5.463333333333334e-05,
      "loss": 0.101,
      "step": 1640
    },
    {
      "epoch": 0.1919026445120524,
      "loss_main": 0.1388191133737564,
      "step": 1640
    },
    {
      "epoch": 0.19307278258834543,
      "grad_norm": 0.6232684254646301,
      "learning_rate": 5.496666666666666e-05,
      "loss": 0.0997,
      "step": 1650
    },
    {
      "epoch": 0.19307278258834543,
      "loss_main": 0.0809781551361084,
      "step": 1650
    },
    {
      "epoch": 0.19424292066463844,
      "grad_norm": 0.661689281463623,
      "learning_rate": 5.530000000000001e-05,
      "loss": 0.105,
      "step": 1660
    },
    {
      "epoch": 0.19424292066463844,
      "loss_main": 0.12127912044525146,
      "step": 1660
    },
    {
      "epoch": 0.19541305874093143,
      "grad_norm": 0.6466169953346252,
      "learning_rate": 5.5633333333333335e-05,
      "loss": 0.1043,
      "step": 1670
    },
    {
      "epoch": 0.19541305874093143,
      "loss_main": 0.09826819598674774,
      "step": 1670
    },
    {
      "epoch": 0.19658319681722444,
      "grad_norm": 0.621999204158783,
      "learning_rate": 5.596666666666667e-05,
      "loss": 0.1077,
      "step": 1680
    },
    {
      "epoch": 0.19658319681722444,
      "loss_main": 0.12383347749710083,
      "step": 1680
    },
    {
      "epoch": 0.19775333489351743,
      "grad_norm": 0.6962786316871643,
      "learning_rate": 5.63e-05,
      "loss": 0.1278,
      "step": 1690
    },
    {
      "epoch": 0.19775333489351743,
      "loss_main": 0.1273275762796402,
      "step": 1690
    },
    {
      "epoch": 0.19892347296981044,
      "grad_norm": 0.594748318195343,
      "learning_rate": 5.663333333333334e-05,
      "loss": 0.1157,
      "step": 1700
    },
    {
      "epoch": 0.19892347296981044,
      "loss_main": 0.17015300691127777,
      "step": 1700
    },
    {
      "epoch": 0.20009361104610343,
      "grad_norm": 0.410970002412796,
      "learning_rate": 5.696666666666667e-05,
      "loss": 0.108,
      "step": 1710
    },
    {
      "epoch": 0.20009361104610343,
      "loss_main": 0.09661001712083817,
      "step": 1710
    },
    {
      "epoch": 0.20126374912239645,
      "grad_norm": 0.6382223963737488,
      "learning_rate": 5.73e-05,
      "loss": 0.1088,
      "step": 1720
    },
    {
      "epoch": 0.20126374912239645,
      "loss_main": 0.09844643622636795,
      "step": 1720
    },
    {
      "epoch": 0.20243388719868943,
      "grad_norm": 0.689781665802002,
      "learning_rate": 5.7633333333333336e-05,
      "loss": 0.1007,
      "step": 1730
    },
    {
      "epoch": 0.20243388719868943,
      "loss_main": 0.17003627121448517,
      "step": 1730
    },
    {
      "epoch": 0.20360402527498245,
      "grad_norm": 0.5365110635757446,
      "learning_rate": 5.796666666666667e-05,
      "loss": 0.1092,
      "step": 1740
    },
    {
      "epoch": 0.20360402527498245,
      "loss_main": 0.10653622448444366,
      "step": 1740
    },
    {
      "epoch": 0.20477416335127546,
      "grad_norm": 0.5795477032661438,
      "learning_rate": 5.83e-05,
      "loss": 0.097,
      "step": 1750
    },
    {
      "epoch": 0.20477416335127546,
      "loss_main": 0.13888666033744812,
      "step": 1750
    },
    {
      "epoch": 0.20594430142756845,
      "grad_norm": 0.8692095279693604,
      "learning_rate": 5.863333333333334e-05,
      "loss": 0.1031,
      "step": 1760
    },
    {
      "epoch": 0.20594430142756845,
      "loss_main": 0.14612825214862823,
      "step": 1760
    },
    {
      "epoch": 0.20711443950386146,
      "grad_norm": 0.519339919090271,
      "learning_rate": 5.896666666666667e-05,
      "loss": 0.1038,
      "step": 1770
    },
    {
      "epoch": 0.20711443950386146,
      "loss_main": 0.08380313217639923,
      "step": 1770
    },
    {
      "epoch": 0.20828457758015445,
      "grad_norm": 0.6227666735649109,
      "learning_rate": 5.93e-05,
      "loss": 0.0852,
      "step": 1780
    },
    {
      "epoch": 0.20828457758015445,
      "loss_main": 0.09281033277511597,
      "step": 1780
    },
    {
      "epoch": 0.20945471565644747,
      "grad_norm": 0.8844098448753357,
      "learning_rate": 5.9633333333333344e-05,
      "loss": 0.1048,
      "step": 1790
    },
    {
      "epoch": 0.20945471565644747,
      "loss_main": 0.074513740837574,
      "step": 1790
    },
    {
      "epoch": 0.21062485373274045,
      "grad_norm": 0.563563346862793,
      "learning_rate": 5.996666666666667e-05,
      "loss": 0.1016,
      "step": 1800
    },
    {
      "epoch": 0.21062485373274045,
      "loss_main": 0.08865710347890854,
      "step": 1800
    },
    {
      "epoch": 0.21179499180903347,
      "grad_norm": 0.6576021313667297,
      "learning_rate": 6.03e-05,
      "loss": 0.1009,
      "step": 1810
    },
    {
      "epoch": 0.21179499180903347,
      "loss_main": 0.06159352511167526,
      "step": 1810
    },
    {
      "epoch": 0.21296512988532648,
      "grad_norm": 0.5296376943588257,
      "learning_rate": 6.063333333333333e-05,
      "loss": 0.0957,
      "step": 1820
    },
    {
      "epoch": 0.21296512988532648,
      "loss_main": 0.09408698230981827,
      "step": 1820
    },
    {
      "epoch": 0.21413526796161947,
      "grad_norm": 0.6130964159965515,
      "learning_rate": 6.0966666666666674e-05,
      "loss": 0.1091,
      "step": 1830
    },
    {
      "epoch": 0.21413526796161947,
      "loss_main": 0.1210596039891243,
      "step": 1830
    },
    {
      "epoch": 0.21530540603791248,
      "grad_norm": 0.49087435007095337,
      "learning_rate": 6.13e-05,
      "loss": 0.1189,
      "step": 1840
    },
    {
      "epoch": 0.21530540603791248,
      "loss_main": 0.0958041176199913,
      "step": 1840
    },
    {
      "epoch": 0.21647554411420547,
      "grad_norm": 0.4662105441093445,
      "learning_rate": 6.163333333333333e-05,
      "loss": 0.0984,
      "step": 1850
    },
    {
      "epoch": 0.21647554411420547,
      "loss_main": 0.07617761194705963,
      "step": 1850
    },
    {
      "epoch": 0.21764568219049849,
      "grad_norm": 0.7102389931678772,
      "learning_rate": 6.196666666666668e-05,
      "loss": 0.0977,
      "step": 1860
    },
    {
      "epoch": 0.21764568219049849,
      "loss_main": 0.08949533849954605,
      "step": 1860
    },
    {
      "epoch": 0.21881582026679147,
      "grad_norm": 0.43680480122566223,
      "learning_rate": 6.23e-05,
      "loss": 0.0912,
      "step": 1870
    },
    {
      "epoch": 0.21881582026679147,
      "loss_main": 0.08627164363861084,
      "step": 1870
    },
    {
      "epoch": 0.2199859583430845,
      "grad_norm": 0.8985812067985535,
      "learning_rate": 6.263333333333333e-05,
      "loss": 0.1029,
      "step": 1880
    },
    {
      "epoch": 0.2199859583430845,
      "loss_main": 0.10120543092489243,
      "step": 1880
    },
    {
      "epoch": 0.22115609641937747,
      "grad_norm": 0.4646529257297516,
      "learning_rate": 6.296666666666667e-05,
      "loss": 0.1117,
      "step": 1890
    },
    {
      "epoch": 0.22115609641937747,
      "loss_main": 0.11935917288064957,
      "step": 1890
    },
    {
      "epoch": 0.2223262344956705,
      "grad_norm": 0.5930261611938477,
      "learning_rate": 6.330000000000001e-05,
      "loss": 0.1068,
      "step": 1900
    },
    {
      "epoch": 0.2223262344956705,
      "loss_main": 0.08035849779844284,
      "step": 1900
    },
    {
      "epoch": 0.2234963725719635,
      "grad_norm": 0.5091857314109802,
      "learning_rate": 6.363333333333334e-05,
      "loss": 0.095,
      "step": 1910
    },
    {
      "epoch": 0.2234963725719635,
      "loss_main": 0.06812600046396255,
      "step": 1910
    },
    {
      "epoch": 0.2246665106482565,
      "grad_norm": 0.6287617683410645,
      "learning_rate": 6.396666666666667e-05,
      "loss": 0.0913,
      "step": 1920
    },
    {
      "epoch": 0.2246665106482565,
      "loss_main": 0.1331566870212555,
      "step": 1920
    },
    {
      "epoch": 0.2258366487245495,
      "grad_norm": 0.7181875109672546,
      "learning_rate": 6.43e-05,
      "loss": 0.1063,
      "step": 1930
    },
    {
      "epoch": 0.2258366487245495,
      "loss_main": 0.11203911155462265,
      "step": 1930
    },
    {
      "epoch": 0.2270067868008425,
      "grad_norm": 0.573650598526001,
      "learning_rate": 6.463333333333334e-05,
      "loss": 0.0944,
      "step": 1940
    },
    {
      "epoch": 0.2270067868008425,
      "loss_main": 0.10840259492397308,
      "step": 1940
    },
    {
      "epoch": 0.2281769248771355,
      "grad_norm": 0.5561143755912781,
      "learning_rate": 6.496666666666667e-05,
      "loss": 0.1163,
      "step": 1950
    },
    {
      "epoch": 0.2281769248771355,
      "loss_main": 0.1483452320098877,
      "step": 1950
    },
    {
      "epoch": 0.2293470629534285,
      "grad_norm": 0.6869015097618103,
      "learning_rate": 6.53e-05,
      "loss": 0.1001,
      "step": 1960
    },
    {
      "epoch": 0.2293470629534285,
      "loss_main": 0.11257942765951157,
      "step": 1960
    },
    {
      "epoch": 0.2305172010297215,
      "grad_norm": 0.4448278844356537,
      "learning_rate": 6.563333333333333e-05,
      "loss": 0.1014,
      "step": 1970
    },
    {
      "epoch": 0.2305172010297215,
      "loss_main": 0.1339060366153717,
      "step": 1970
    },
    {
      "epoch": 0.23168733910601452,
      "grad_norm": 0.549889326095581,
      "learning_rate": 6.596666666666667e-05,
      "loss": 0.11,
      "step": 1980
    },
    {
      "epoch": 0.23168733910601452,
      "loss_main": 0.07053396850824356,
      "step": 1980
    },
    {
      "epoch": 0.2328574771823075,
      "grad_norm": 0.48016950488090515,
      "learning_rate": 6.630000000000001e-05,
      "loss": 0.0952,
      "step": 1990
    },
    {
      "epoch": 0.2328574771823075,
      "loss_main": 0.08421308547258377,
      "step": 1990
    },
    {
      "epoch": 0.23402761525860052,
      "grad_norm": 0.37187063694000244,
      "learning_rate": 6.663333333333333e-05,
      "loss": 0.1085,
      "step": 2000
    },
    {
      "epoch": 0.23402761525860052,
      "loss_main": 0.08223439753055573,
      "step": 2000
    },
    {
      "epoch": 0.2351977533348935,
      "grad_norm": 0.5327889323234558,
      "learning_rate": 6.696666666666666e-05,
      "loss": 0.1045,
      "step": 2010
    },
    {
      "epoch": 0.2351977533348935,
      "loss_main": 0.10440925508737564,
      "step": 2010
    },
    {
      "epoch": 0.23636789141118653,
      "grad_norm": 0.6276612877845764,
      "learning_rate": 6.730000000000001e-05,
      "loss": 0.1,
      "step": 2020
    },
    {
      "epoch": 0.23636789141118653,
      "loss_main": 0.09120891988277435,
      "step": 2020
    },
    {
      "epoch": 0.2375380294874795,
      "grad_norm": 0.5454798340797424,
      "learning_rate": 6.763333333333334e-05,
      "loss": 0.1114,
      "step": 2030
    },
    {
      "epoch": 0.2375380294874795,
      "loss_main": 0.08867032825946808,
      "step": 2030
    },
    {
      "epoch": 0.23870816756377253,
      "grad_norm": 0.5995694398880005,
      "learning_rate": 6.796666666666666e-05,
      "loss": 0.0998,
      "step": 2040
    },
    {
      "epoch": 0.23870816756377253,
      "loss_main": 0.09142450243234634,
      "step": 2040
    },
    {
      "epoch": 0.23987830564006551,
      "grad_norm": 0.484831839799881,
      "learning_rate": 6.83e-05,
      "loss": 0.0963,
      "step": 2050
    },
    {
      "epoch": 0.23987830564006551,
      "loss_main": 0.0867501050233841,
      "step": 2050
    },
    {
      "epoch": 0.24104844371635853,
      "grad_norm": 0.425698846578598,
      "learning_rate": 6.863333333333334e-05,
      "loss": 0.1097,
      "step": 2060
    },
    {
      "epoch": 0.24104844371635853,
      "loss_main": 0.08215440064668655,
      "step": 2060
    },
    {
      "epoch": 0.24221858179265154,
      "grad_norm": 0.5377969145774841,
      "learning_rate": 6.896666666666667e-05,
      "loss": 0.105,
      "step": 2070
    },
    {
      "epoch": 0.24221858179265154,
      "loss_main": 0.11810234934091568,
      "step": 2070
    },
    {
      "epoch": 0.24338871986894453,
      "grad_norm": 0.5032196640968323,
      "learning_rate": 6.93e-05,
      "loss": 0.1039,
      "step": 2080
    },
    {
      "epoch": 0.24338871986894453,
      "loss_main": 0.11527891457080841,
      "step": 2080
    },
    {
      "epoch": 0.24455885794523755,
      "grad_norm": 0.4768078923225403,
      "learning_rate": 6.963333333333334e-05,
      "loss": 0.092,
      "step": 2090
    },
    {
      "epoch": 0.24455885794523755,
      "loss_main": 0.1508232057094574,
      "step": 2090
    },
    {
      "epoch": 0.24572899602153053,
      "grad_norm": 0.5450745224952698,
      "learning_rate": 6.996666666666667e-05,
      "loss": 0.102,
      "step": 2100
    },
    {
      "epoch": 0.24572899602153053,
      "loss_main": 0.09476476162672043,
      "step": 2100
    },
    {
      "epoch": 0.24689913409782355,
      "grad_norm": 0.3998696506023407,
      "learning_rate": 7.03e-05,
      "loss": 0.1137,
      "step": 2110
    },
    {
      "epoch": 0.24689913409782355,
      "loss_main": 0.09826956689357758,
      "step": 2110
    },
    {
      "epoch": 0.24806927217411653,
      "grad_norm": 0.5599823594093323,
      "learning_rate": 7.063333333333333e-05,
      "loss": 0.1145,
      "step": 2120
    },
    {
      "epoch": 0.24806927217411653,
      "loss_main": 0.09114454686641693,
      "step": 2120
    },
    {
      "epoch": 0.24923941025040955,
      "grad_norm": 0.677278459072113,
      "learning_rate": 7.096666666666667e-05,
      "loss": 0.1114,
      "step": 2130
    },
    {
      "epoch": 0.24923941025040955,
      "loss_main": 0.1277070939540863,
      "step": 2130
    },
    {
      "epoch": 0.25040954832670254,
      "grad_norm": 0.735707700252533,
      "learning_rate": 7.13e-05,
      "loss": 0.0999,
      "step": 2140
    },
    {
      "epoch": 0.25040954832670254,
      "loss_main": 0.09583481401205063,
      "step": 2140
    },
    {
      "epoch": 0.25157968640299555,
      "grad_norm": 0.6554832458496094,
      "learning_rate": 7.163333333333334e-05,
      "loss": 0.105,
      "step": 2150
    },
    {
      "epoch": 0.25157968640299555,
      "loss_main": 0.10701914131641388,
      "step": 2150
    },
    {
      "epoch": 0.25274982447928857,
      "grad_norm": 0.503365695476532,
      "learning_rate": 7.196666666666668e-05,
      "loss": 0.1068,
      "step": 2160
    },
    {
      "epoch": 0.25274982447928857,
      "loss_main": 0.07623835653066635,
      "step": 2160
    },
    {
      "epoch": 0.2539199625555816,
      "grad_norm": 0.5713593363761902,
      "learning_rate": 7.23e-05,
      "loss": 0.0967,
      "step": 2170
    },
    {
      "epoch": 0.2539199625555816,
      "loss_main": 0.10242123156785965,
      "step": 2170
    },
    {
      "epoch": 0.25509010063187454,
      "grad_norm": 0.5398196578025818,
      "learning_rate": 7.263333333333334e-05,
      "loss": 0.0989,
      "step": 2180
    },
    {
      "epoch": 0.25509010063187454,
      "loss_main": 0.10535743087530136,
      "step": 2180
    },
    {
      "epoch": 0.25626023870816755,
      "grad_norm": 0.6828546524047852,
      "learning_rate": 7.296666666666667e-05,
      "loss": 0.1112,
      "step": 2190
    },
    {
      "epoch": 0.25626023870816755,
      "loss_main": 0.09142204374074936,
      "step": 2190
    },
    {
      "epoch": 0.25743037678446057,
      "grad_norm": 0.5348831415176392,
      "learning_rate": 7.33e-05,
      "loss": 0.1092,
      "step": 2200
    },
    {
      "epoch": 0.25743037678446057,
      "loss_main": 0.12186127156019211,
      "step": 2200
    },
    {
      "epoch": 0.2586005148607536,
      "grad_norm": 0.5797008275985718,
      "learning_rate": 7.363333333333334e-05,
      "loss": 0.1066,
      "step": 2210
    },
    {
      "epoch": 0.2586005148607536,
      "loss_main": 0.07920967042446136,
      "step": 2210
    },
    {
      "epoch": 0.2597706529370466,
      "grad_norm": 0.5827093720436096,
      "learning_rate": 7.396666666666667e-05,
      "loss": 0.0985,
      "step": 2220
    },
    {
      "epoch": 0.2597706529370466,
      "loss_main": 0.09124619513750076,
      "step": 2220
    },
    {
      "epoch": 0.26094079101333956,
      "grad_norm": 0.7236694097518921,
      "learning_rate": 7.43e-05,
      "loss": 0.1207,
      "step": 2230
    },
    {
      "epoch": 0.26094079101333956,
      "loss_main": 0.11996611207723618,
      "step": 2230
    },
    {
      "epoch": 0.26211092908963257,
      "grad_norm": 0.5343965291976929,
      "learning_rate": 7.463333333333334e-05,
      "loss": 0.1001,
      "step": 2240
    },
    {
      "epoch": 0.26211092908963257,
      "loss_main": 0.11949585378170013,
      "step": 2240
    },
    {
      "epoch": 0.2632810671659256,
      "grad_norm": 0.4986639618873596,
      "learning_rate": 7.496666666666667e-05,
      "loss": 0.1109,
      "step": 2250
    },
    {
      "epoch": 0.2632810671659256,
      "loss_main": 0.1193009465932846,
      "step": 2250
    },
    {
      "epoch": 0.2644512052422186,
      "grad_norm": 0.478511780500412,
      "learning_rate": 7.53e-05,
      "loss": 0.1143,
      "step": 2260
    },
    {
      "epoch": 0.2644512052422186,
      "loss_main": 0.08516277372837067,
      "step": 2260
    },
    {
      "epoch": 0.26562134331851156,
      "grad_norm": 0.5392448306083679,
      "learning_rate": 7.563333333333333e-05,
      "loss": 0.1113,
      "step": 2270
    },
    {
      "epoch": 0.26562134331851156,
      "loss_main": 0.13216309249401093,
      "step": 2270
    },
    {
      "epoch": 0.2667914813948046,
      "grad_norm": 0.539665162563324,
      "learning_rate": 7.596666666666668e-05,
      "loss": 0.1112,
      "step": 2280
    },
    {
      "epoch": 0.2667914813948046,
      "loss_main": 0.09599894285202026,
      "step": 2280
    },
    {
      "epoch": 0.2679616194710976,
      "grad_norm": 0.5851640701293945,
      "learning_rate": 7.630000000000001e-05,
      "loss": 0.1206,
      "step": 2290
    },
    {
      "epoch": 0.2679616194710976,
      "loss_main": 0.09592794626951218,
      "step": 2290
    },
    {
      "epoch": 0.2691317575473906,
      "grad_norm": 0.47005677223205566,
      "learning_rate": 7.663333333333333e-05,
      "loss": 0.1015,
      "step": 2300
    },
    {
      "epoch": 0.2691317575473906,
      "loss_main": 0.061951782554388046,
      "step": 2300
    },
    {
      "epoch": 0.2703018956236836,
      "grad_norm": 0.5082420706748962,
      "learning_rate": 7.696666666666668e-05,
      "loss": 0.0968,
      "step": 2310
    },
    {
      "epoch": 0.2703018956236836,
      "loss_main": 0.08067576587200165,
      "step": 2310
    },
    {
      "epoch": 0.2714720336999766,
      "grad_norm": 0.37042713165283203,
      "learning_rate": 7.730000000000001e-05,
      "loss": 0.106,
      "step": 2320
    },
    {
      "epoch": 0.2714720336999766,
      "loss_main": 0.07465784251689911,
      "step": 2320
    },
    {
      "epoch": 0.2726421717762696,
      "grad_norm": 0.4228276014328003,
      "learning_rate": 7.763333333333334e-05,
      "loss": 0.0966,
      "step": 2330
    },
    {
      "epoch": 0.2726421717762696,
      "loss_main": 0.07154548913240433,
      "step": 2330
    },
    {
      "epoch": 0.2738123098525626,
      "grad_norm": 0.4700446128845215,
      "learning_rate": 7.796666666666666e-05,
      "loss": 0.0937,
      "step": 2340
    },
    {
      "epoch": 0.2738123098525626,
      "loss_main": 0.07182364910840988,
      "step": 2340
    },
    {
      "epoch": 0.2749824479288556,
      "grad_norm": 0.6366181373596191,
      "learning_rate": 7.83e-05,
      "loss": 0.0992,
      "step": 2350
    },
    {
      "epoch": 0.2749824479288556,
      "loss_main": 0.10893864184617996,
      "step": 2350
    },
    {
      "epoch": 0.2761525860051486,
      "grad_norm": 0.5424176454544067,
      "learning_rate": 7.863333333333334e-05,
      "loss": 0.1053,
      "step": 2360
    },
    {
      "epoch": 0.2761525860051486,
      "loss_main": 0.06606603413820267,
      "step": 2360
    },
    {
      "epoch": 0.2773227240814416,
      "grad_norm": 0.70343416929245,
      "learning_rate": 7.896666666666667e-05,
      "loss": 0.1059,
      "step": 2370
    },
    {
      "epoch": 0.2773227240814416,
      "loss_main": 0.09163417667150497,
      "step": 2370
    },
    {
      "epoch": 0.2784928621577346,
      "grad_norm": 0.7406337857246399,
      "learning_rate": 7.93e-05,
      "loss": 0.1089,
      "step": 2380
    },
    {
      "epoch": 0.2784928621577346,
      "loss_main": 0.11042772978544235,
      "step": 2380
    },
    {
      "epoch": 0.2796630002340276,
      "grad_norm": 0.6043682098388672,
      "learning_rate": 7.963333333333334e-05,
      "loss": 0.1079,
      "step": 2390
    },
    {
      "epoch": 0.2796630002340276,
      "loss_main": 0.12478935718536377,
      "step": 2390
    },
    {
      "epoch": 0.28083313831032064,
      "grad_norm": 0.49900171160697937,
      "learning_rate": 7.996666666666667e-05,
      "loss": 0.0954,
      "step": 2400
    },
    {
      "epoch": 0.28083313831032064,
      "loss_main": 0.06922612339258194,
      "step": 2400
    },
    {
      "epoch": 0.2820032763866136,
      "grad_norm": 0.4286934435367584,
      "learning_rate": 8.030000000000001e-05,
      "loss": 0.1009,
      "step": 2410
    },
    {
      "epoch": 0.2820032763866136,
      "loss_main": 0.13138197362422943,
      "step": 2410
    },
    {
      "epoch": 0.2831734144629066,
      "grad_norm": 0.5795945525169373,
      "learning_rate": 8.063333333333333e-05,
      "loss": 0.1061,
      "step": 2420
    },
    {
      "epoch": 0.2831734144629066,
      "loss_main": 0.12229681015014648,
      "step": 2420
    },
    {
      "epoch": 0.28434355253919963,
      "grad_norm": 0.6552769541740417,
      "learning_rate": 8.096666666666667e-05,
      "loss": 0.0984,
      "step": 2430
    },
    {
      "epoch": 0.28434355253919963,
      "loss_main": 0.15727755427360535,
      "step": 2430
    },
    {
      "epoch": 0.28551369061549264,
      "grad_norm": 0.4330909252166748,
      "learning_rate": 8.13e-05,
      "loss": 0.1093,
      "step": 2440
    },
    {
      "epoch": 0.28551369061549264,
      "loss_main": 0.0885615348815918,
      "step": 2440
    },
    {
      "epoch": 0.28668382869178566,
      "grad_norm": 0.2543618679046631,
      "learning_rate": 8.163333333333334e-05,
      "loss": 0.1036,
      "step": 2450
    },
    {
      "epoch": 0.28668382869178566,
      "loss_main": 0.06747984886169434,
      "step": 2450
    },
    {
      "epoch": 0.2878539667680786,
      "grad_norm": 0.46994611620903015,
      "learning_rate": 8.196666666666668e-05,
      "loss": 0.0977,
      "step": 2460
    },
    {
      "epoch": 0.2878539667680786,
      "loss_main": 0.11807603389024734,
      "step": 2460
    },
    {
      "epoch": 0.28902410484437163,
      "grad_norm": 0.4444718658924103,
      "learning_rate": 8.23e-05,
      "loss": 0.094,
      "step": 2470
    },
    {
      "epoch": 0.28902410484437163,
      "loss_main": 0.08751928061246872,
      "step": 2470
    },
    {
      "epoch": 0.29019424292066465,
      "grad_norm": 0.7152218818664551,
      "learning_rate": 8.263333333333334e-05,
      "loss": 0.1072,
      "step": 2480
    },
    {
      "epoch": 0.29019424292066465,
      "loss_main": 0.14982827007770538,
      "step": 2480
    },
    {
      "epoch": 0.29136438099695766,
      "grad_norm": 0.5862312912940979,
      "learning_rate": 8.296666666666667e-05,
      "loss": 0.1184,
      "step": 2490
    },
    {
      "epoch": 0.29136438099695766,
      "loss_main": 0.08168774098157883,
      "step": 2490
    },
    {
      "epoch": 0.2925345190732506,
      "grad_norm": 0.9449164271354675,
      "learning_rate": 8.33e-05,
      "loss": 0.1073,
      "step": 2500
    },
    {
      "epoch": 0.2925345190732506,
      "loss_main": 0.1017787978053093,
      "step": 2500
    },
    {
      "epoch": 0.29370465714954364,
      "grad_norm": 0.3774797320365906,
      "learning_rate": 8.363333333333334e-05,
      "loss": 0.1032,
      "step": 2510
    },
    {
      "epoch": 0.29370465714954364,
      "loss_main": 0.09769018739461899,
      "step": 2510
    },
    {
      "epoch": 0.29487479522583665,
      "grad_norm": 0.3862501382827759,
      "learning_rate": 8.396666666666667e-05,
      "loss": 0.0995,
      "step": 2520
    },
    {
      "epoch": 0.29487479522583665,
      "loss_main": 0.15956361591815948,
      "step": 2520
    },
    {
      "epoch": 0.29604493330212966,
      "grad_norm": 0.5975602269172668,
      "learning_rate": 8.43e-05,
      "loss": 0.0951,
      "step": 2530
    },
    {
      "epoch": 0.29604493330212966,
      "loss_main": 0.12117917090654373,
      "step": 2530
    },
    {
      "epoch": 0.2972150713784227,
      "grad_norm": 0.47991743683815,
      "learning_rate": 8.463333333333335e-05,
      "loss": 0.1132,
      "step": 2540
    },
    {
      "epoch": 0.2972150713784227,
      "loss_main": 0.07322447746992111,
      "step": 2540
    },
    {
      "epoch": 0.29838520945471564,
      "grad_norm": 0.5197693109512329,
      "learning_rate": 8.496666666666667e-05,
      "loss": 0.0909,
      "step": 2550
    },
    {
      "epoch": 0.29838520945471564,
      "loss_main": 0.06056847423315048,
      "step": 2550
    },
    {
      "epoch": 0.29955534753100865,
      "grad_norm": 0.4682486951351166,
      "learning_rate": 8.53e-05,
      "loss": 0.1093,
      "step": 2560
    },
    {
      "epoch": 0.29955534753100865,
      "loss_main": 0.08799250423908234,
      "step": 2560
    },
    {
      "epoch": 0.30072548560730167,
      "grad_norm": 0.42091065645217896,
      "learning_rate": 8.563333333333333e-05,
      "loss": 0.1043,
      "step": 2570
    },
    {
      "epoch": 0.30072548560730167,
      "loss_main": 0.07224845886230469,
      "step": 2570
    },
    {
      "epoch": 0.3018956236835947,
      "grad_norm": 0.45692315697669983,
      "learning_rate": 8.596666666666668e-05,
      "loss": 0.1048,
      "step": 2580
    },
    {
      "epoch": 0.3018956236835947,
      "loss_main": 0.133596271276474,
      "step": 2580
    },
    {
      "epoch": 0.30306576175988764,
      "grad_norm": 0.5643550753593445,
      "learning_rate": 8.63e-05,
      "loss": 0.0973,
      "step": 2590
    },
    {
      "epoch": 0.30306576175988764,
      "loss_main": 0.12293235212564468,
      "step": 2590
    },
    {
      "epoch": 0.30423589983618066,
      "grad_norm": 0.5669226050376892,
      "learning_rate": 8.663333333333333e-05,
      "loss": 0.1052,
      "step": 2600
    },
    {
      "epoch": 0.30423589983618066,
      "loss_main": 0.13594408333301544,
      "step": 2600
    },
    {
      "epoch": 0.30540603791247367,
      "grad_norm": 0.7140518426895142,
      "learning_rate": 8.696666666666668e-05,
      "loss": 0.1069,
      "step": 2610
    },
    {
      "epoch": 0.30540603791247367,
      "loss_main": 0.10760701447725296,
      "step": 2610
    },
    {
      "epoch": 0.3065761759887667,
      "grad_norm": 0.38238629698753357,
      "learning_rate": 8.730000000000001e-05,
      "loss": 0.0984,
      "step": 2620
    },
    {
      "epoch": 0.3065761759887667,
      "loss_main": 0.15872597694396973,
      "step": 2620
    },
    {
      "epoch": 0.3077463140650597,
      "grad_norm": 0.5739442110061646,
      "learning_rate": 8.763333333333334e-05,
      "loss": 0.1059,
      "step": 2630
    },
    {
      "epoch": 0.3077463140650597,
      "loss_main": 0.08514779061079025,
      "step": 2630
    },
    {
      "epoch": 0.30891645214135266,
      "grad_norm": 0.5772945880889893,
      "learning_rate": 8.796666666666667e-05,
      "loss": 0.1033,
      "step": 2640
    },
    {
      "epoch": 0.30891645214135266,
      "loss_main": 0.14932839572429657,
      "step": 2640
    },
    {
      "epoch": 0.3100865902176457,
      "grad_norm": 0.48025843501091003,
      "learning_rate": 8.83e-05,
      "loss": 0.1034,
      "step": 2650
    },
    {
      "epoch": 0.3100865902176457,
      "loss_main": 0.08597976714372635,
      "step": 2650
    },
    {
      "epoch": 0.3112567282939387,
      "grad_norm": 0.5860459208488464,
      "learning_rate": 8.863333333333334e-05,
      "loss": 0.0958,
      "step": 2660
    },
    {
      "epoch": 0.3112567282939387,
      "loss_main": 0.14503073692321777,
      "step": 2660
    },
    {
      "epoch": 0.3124268663702317,
      "grad_norm": 0.45188993215560913,
      "learning_rate": 8.896666666666667e-05,
      "loss": 0.0968,
      "step": 2670
    },
    {
      "epoch": 0.3124268663702317,
      "loss_main": 0.11113723367452621,
      "step": 2670
    },
    {
      "epoch": 0.31359700444652466,
      "grad_norm": 0.5160958170890808,
      "learning_rate": 8.93e-05,
      "loss": 0.0997,
      "step": 2680
    },
    {
      "epoch": 0.31359700444652466,
      "loss_main": 0.07100610435009003,
      "step": 2680
    },
    {
      "epoch": 0.3147671425228177,
      "grad_norm": 0.450196236371994,
      "learning_rate": 8.963333333333333e-05,
      "loss": 0.0985,
      "step": 2690
    },
    {
      "epoch": 0.3147671425228177,
      "loss_main": 0.13446423411369324,
      "step": 2690
    },
    {
      "epoch": 0.3159372805991107,
      "grad_norm": 0.700903594493866,
      "learning_rate": 8.996666666666667e-05,
      "loss": 0.0982,
      "step": 2700
    },
    {
      "epoch": 0.3159372805991107,
      "loss_main": 0.0813727006316185,
      "step": 2700
    },
    {
      "epoch": 0.3171074186754037,
      "grad_norm": 0.3953317403793335,
      "learning_rate": 9.030000000000001e-05,
      "loss": 0.0899,
      "step": 2710
    },
    {
      "epoch": 0.3171074186754037,
      "loss_main": 0.10271178185939789,
      "step": 2710
    },
    {
      "epoch": 0.3182775567516967,
      "grad_norm": 0.554831326007843,
      "learning_rate": 9.063333333333333e-05,
      "loss": 0.1007,
      "step": 2720
    },
    {
      "epoch": 0.3182775567516967,
      "loss_main": 0.04556824639439583,
      "step": 2720
    },
    {
      "epoch": 0.3194476948279897,
      "grad_norm": 0.3894742727279663,
      "learning_rate": 9.096666666666666e-05,
      "loss": 0.09,
      "step": 2730
    },
    {
      "epoch": 0.3194476948279897,
      "loss_main": 0.06998802721500397,
      "step": 2730
    },
    {
      "epoch": 0.3206178329042827,
      "grad_norm": 0.5323200821876526,
      "learning_rate": 9.130000000000001e-05,
      "loss": 0.0974,
      "step": 2740
    },
    {
      "epoch": 0.3206178329042827,
      "loss_main": 0.14149920642375946,
      "step": 2740
    },
    {
      "epoch": 0.3217879709805757,
      "grad_norm": 0.6280831098556519,
      "learning_rate": 9.163333333333334e-05,
      "loss": 0.1094,
      "step": 2750
    },
    {
      "epoch": 0.3217879709805757,
      "loss_main": 0.14370892941951752,
      "step": 2750
    },
    {
      "epoch": 0.3229581090568687,
      "grad_norm": 0.4483911395072937,
      "learning_rate": 9.196666666666666e-05,
      "loss": 0.0999,
      "step": 2760
    },
    {
      "epoch": 0.3229581090568687,
      "loss_main": 0.11850778013467789,
      "step": 2760
    },
    {
      "epoch": 0.32412824713316174,
      "grad_norm": 0.7152521014213562,
      "learning_rate": 9.230000000000001e-05,
      "loss": 0.0966,
      "step": 2770
    },
    {
      "epoch": 0.32412824713316174,
      "loss_main": 0.12676945328712463,
      "step": 2770
    },
    {
      "epoch": 0.3252983852094547,
      "grad_norm": 0.6317280530929565,
      "learning_rate": 9.263333333333334e-05,
      "loss": 0.1012,
      "step": 2780
    },
    {
      "epoch": 0.3252983852094547,
      "loss_main": 0.0785612091422081,
      "step": 2780
    },
    {
      "epoch": 0.3264685232857477,
      "grad_norm": 0.6086673736572266,
      "learning_rate": 9.296666666666667e-05,
      "loss": 0.1029,
      "step": 2790
    },
    {
      "epoch": 0.3264685232857477,
      "loss_main": 0.08893852680921555,
      "step": 2790
    },
    {
      "epoch": 0.32763866136204073,
      "grad_norm": 0.4204990565776825,
      "learning_rate": 9.33e-05,
      "loss": 0.1,
      "step": 2800
    },
    {
      "epoch": 0.32763866136204073,
      "loss_main": 0.09426934272050858,
      "step": 2800
    },
    {
      "epoch": 0.32880879943833374,
      "grad_norm": 0.350599467754364,
      "learning_rate": 9.363333333333334e-05,
      "loss": 0.1031,
      "step": 2810
    },
    {
      "epoch": 0.32880879943833374,
      "loss_main": 0.07319307327270508,
      "step": 2810
    },
    {
      "epoch": 0.3299789375146267,
      "grad_norm": 0.6340963840484619,
      "learning_rate": 9.396666666666667e-05,
      "loss": 0.0954,
      "step": 2820
    },
    {
      "epoch": 0.3299789375146267,
      "loss_main": 0.12979920208454132,
      "step": 2820
    },
    {
      "epoch": 0.3311490755909197,
      "grad_norm": 0.5475706458091736,
      "learning_rate": 9.43e-05,
      "loss": 0.1184,
      "step": 2830
    },
    {
      "epoch": 0.3311490755909197,
      "loss_main": 0.0737680047750473,
      "step": 2830
    },
    {
      "epoch": 0.33231921366721273,
      "grad_norm": 0.5146294236183167,
      "learning_rate": 9.463333333333333e-05,
      "loss": 0.1042,
      "step": 2840
    },
    {
      "epoch": 0.33231921366721273,
      "loss_main": 0.06743468344211578,
      "step": 2840
    },
    {
      "epoch": 0.33348935174350575,
      "grad_norm": 0.5341523885726929,
      "learning_rate": 9.496666666666667e-05,
      "loss": 0.1009,
      "step": 2850
    },
    {
      "epoch": 0.33348935174350575,
      "loss_main": 0.07580546289682388,
      "step": 2850
    },
    {
      "epoch": 0.33465948981979876,
      "grad_norm": 0.38051125407218933,
      "learning_rate": 9.53e-05,
      "loss": 0.1106,
      "step": 2860
    },
    {
      "epoch": 0.33465948981979876,
      "loss_main": 0.09902240335941315,
      "step": 2860
    },
    {
      "epoch": 0.3358296278960917,
      "grad_norm": 0.49452587962150574,
      "learning_rate": 9.563333333333334e-05,
      "loss": 0.1073,
      "step": 2870
    },
    {
      "epoch": 0.3358296278960917,
      "loss_main": 0.13598616421222687,
      "step": 2870
    },
    {
      "epoch": 0.33699976597238473,
      "grad_norm": 0.47150054574012756,
      "learning_rate": 9.596666666666668e-05,
      "loss": 0.0982,
      "step": 2880
    },
    {
      "epoch": 0.33699976597238473,
      "loss_main": 0.0901784896850586,
      "step": 2880
    },
    {
      "epoch": 0.33816990404867775,
      "grad_norm": 0.6280695796012878,
      "learning_rate": 9.63e-05,
      "loss": 0.0994,
      "step": 2890
    },
    {
      "epoch": 0.33816990404867775,
      "loss_main": 0.09328247606754303,
      "step": 2890
    },
    {
      "epoch": 0.33934004212497076,
      "grad_norm": 0.37997275590896606,
      "learning_rate": 9.663333333333334e-05,
      "loss": 0.1061,
      "step": 2900
    },
    {
      "epoch": 0.33934004212497076,
      "loss_main": 0.057282064110040665,
      "step": 2900
    },
    {
      "epoch": 0.3405101802012637,
      "grad_norm": 0.5887234210968018,
      "learning_rate": 9.696666666666667e-05,
      "loss": 0.0937,
      "step": 2910
    },
    {
      "epoch": 0.3405101802012637,
      "loss_main": 0.0867401510477066,
      "step": 2910
    },
    {
      "epoch": 0.34168031827755674,
      "grad_norm": 0.28726065158843994,
      "learning_rate": 9.730000000000001e-05,
      "loss": 0.0951,
      "step": 2920
    },
    {
      "epoch": 0.34168031827755674,
      "loss_main": 0.05197948217391968,
      "step": 2920
    },
    {
      "epoch": 0.34285045635384975,
      "grad_norm": 0.5868906378746033,
      "learning_rate": 9.763333333333334e-05,
      "loss": 0.0941,
      "step": 2930
    },
    {
      "epoch": 0.34285045635384975,
      "loss_main": 0.16154609620571136,
      "step": 2930
    },
    {
      "epoch": 0.34402059443014277,
      "grad_norm": 0.4812150001525879,
      "learning_rate": 9.796666666666667e-05,
      "loss": 0.1002,
      "step": 2940
    },
    {
      "epoch": 0.34402059443014277,
      "loss_main": 0.1464698612689972,
      "step": 2940
    },
    {
      "epoch": 0.3451907325064358,
      "grad_norm": 0.49590519070625305,
      "learning_rate": 9.83e-05,
      "loss": 0.106,
      "step": 2950
    },
    {
      "epoch": 0.3451907325064358,
      "loss_main": 0.12326163798570633,
      "step": 2950
    },
    {
      "epoch": 0.34636087058272874,
      "grad_norm": 0.468097448348999,
      "learning_rate": 9.863333333333334e-05,
      "loss": 0.0995,
      "step": 2960
    },
    {
      "epoch": 0.34636087058272874,
      "loss_main": 0.06952954083681107,
      "step": 2960
    },
    {
      "epoch": 0.34753100865902176,
      "grad_norm": 0.427296906709671,
      "learning_rate": 9.896666666666667e-05,
      "loss": 0.0987,
      "step": 2970
    },
    {
      "epoch": 0.34753100865902176,
      "loss_main": 0.12316039949655533,
      "step": 2970
    },
    {
      "epoch": 0.34870114673531477,
      "grad_norm": 0.5160908699035645,
      "learning_rate": 9.93e-05,
      "loss": 0.0915,
      "step": 2980
    },
    {
      "epoch": 0.34870114673531477,
      "loss_main": 0.11695647984743118,
      "step": 2980
    },
    {
      "epoch": 0.3498712848116078,
      "grad_norm": 0.4563028812408447,
      "learning_rate": 9.963333333333333e-05,
      "loss": 0.1054,
      "step": 2990
    },
    {
      "epoch": 0.3498712848116078,
      "loss_main": 0.12684008479118347,
      "step": 2990
    },
    {
      "epoch": 0.35104142288790074,
      "grad_norm": 0.29745933413505554,
      "learning_rate": 9.996666666666668e-05,
      "loss": 0.1187,
      "step": 3000
    },
    {
      "epoch": 0.35104142288790074,
      "loss_main": 0.08335602283477783,
      "step": 3000
    },
    {
      "epoch": 0.35221156096419376,
      "grad_norm": 0.45725277066230774,
      "learning_rate": 9.999999384858465e-05,
      "loss": 0.1082,
      "step": 3010
    },
    {
      "epoch": 0.35221156096419376,
      "loss_main": 0.09177880734205246,
      "step": 3010
    },
    {
      "epoch": 0.3533816990404868,
      "grad_norm": 0.463316410779953,
      "learning_rate": 9.999997258443473e-05,
      "loss": 0.1034,
      "step": 3020
    },
    {
      "epoch": 0.3533816990404868,
      "loss_main": 0.13021767139434814,
      "step": 3020
    },
    {
      "epoch": 0.3545518371167798,
      "grad_norm": 0.4410639703273773,
      "learning_rate": 9.999993613161331e-05,
      "loss": 0.0973,
      "step": 3030
    },
    {
      "epoch": 0.3545518371167798,
      "loss_main": 0.12459705024957657,
      "step": 3030
    },
    {
      "epoch": 0.3557219751930728,
      "grad_norm": 0.4924675226211548,
      "learning_rate": 9.999988449013146e-05,
      "loss": 0.1036,
      "step": 3040
    },
    {
      "epoch": 0.3557219751930728,
      "loss_main": 0.09098248183727264,
      "step": 3040
    },
    {
      "epoch": 0.35689211326936576,
      "grad_norm": 0.4165923595428467,
      "learning_rate": 9.99998176600049e-05,
      "loss": 0.111,
      "step": 3050
    },
    {
      "epoch": 0.35689211326936576,
      "loss_main": 0.07655839622020721,
      "step": 3050
    },
    {
      "epoch": 0.3580622513456588,
      "grad_norm": 0.3824102580547333,
      "learning_rate": 9.999973564125389e-05,
      "loss": 0.0905,
      "step": 3060
    },
    {
      "epoch": 0.3580622513456588,
      "loss_main": 0.06710273027420044,
      "step": 3060
    },
    {
      "epoch": 0.3592323894219518,
      "grad_norm": 0.5395789742469788,
      "learning_rate": 9.999963843390335e-05,
      "loss": 0.0878,
      "step": 3070
    },
    {
      "epoch": 0.3592323894219518,
      "loss_main": 0.1557062864303589,
      "step": 3070
    },
    {
      "epoch": 0.3604025274982448,
      "grad_norm": 0.5577585101127625,
      "learning_rate": 9.999952603798282e-05,
      "loss": 0.121,
      "step": 3080
    },
    {
      "epoch": 0.3604025274982448,
      "loss_main": 0.12019359320402145,
      "step": 3080
    },
    {
      "epoch": 0.3615726655745378,
      "grad_norm": 0.6044942736625671,
      "learning_rate": 9.999939845352646e-05,
      "loss": 0.0997,
      "step": 3090
    },
    {
      "epoch": 0.3615726655745378,
      "loss_main": 0.1326633095741272,
      "step": 3090
    },
    {
      "epoch": 0.3627428036508308,
      "grad_norm": 0.5628751516342163,
      "learning_rate": 9.999925568057298e-05,
      "loss": 0.1084,
      "step": 3100
    },
    {
      "epoch": 0.3627428036508308,
      "loss_main": 0.07526196539402008,
      "step": 3100
    },
    {
      "epoch": 0.3639129417271238,
      "grad_norm": 0.566059947013855,
      "learning_rate": 9.999909771916578e-05,
      "loss": 0.1015,
      "step": 3110
    },
    {
      "epoch": 0.3639129417271238,
      "loss_main": 0.10943132638931274,
      "step": 3110
    },
    {
      "epoch": 0.3650830798034168,
      "grad_norm": 0.6209599375724792,
      "learning_rate": 9.999892456935285e-05,
      "loss": 0.1067,
      "step": 3120
    },
    {
      "epoch": 0.3650830798034168,
      "loss_main": 0.1350456178188324,
      "step": 3120
    },
    {
      "epoch": 0.3662532178797098,
      "grad_norm": 0.456086128950119,
      "learning_rate": 9.999873623118679e-05,
      "loss": 0.0975,
      "step": 3130
    },
    {
      "epoch": 0.3662532178797098,
      "loss_main": 0.08446104824542999,
      "step": 3130
    },
    {
      "epoch": 0.3674233559560028,
      "grad_norm": 0.5891807675361633,
      "learning_rate": 9.999853270472479e-05,
      "loss": 0.0961,
      "step": 3140
    },
    {
      "epoch": 0.3674233559560028,
      "loss_main": 0.10360700637102127,
      "step": 3140
    },
    {
      "epoch": 0.3685934940322958,
      "grad_norm": 0.5978410840034485,
      "learning_rate": 9.999831399002871e-05,
      "loss": 0.1011,
      "step": 3150
    },
    {
      "epoch": 0.3685934940322958,
      "loss_main": 0.09930193424224854,
      "step": 3150
    },
    {
      "epoch": 0.3697636321085888,
      "grad_norm": 0.44969287514686584,
      "learning_rate": 9.999808008716494e-05,
      "loss": 0.0963,
      "step": 3160
    },
    {
      "epoch": 0.3697636321085888,
      "loss_main": 0.11609560996294022,
      "step": 3160
    },
    {
      "epoch": 0.37093377018488183,
      "grad_norm": 0.3896106779575348,
      "learning_rate": 9.999783099620459e-05,
      "loss": 0.0999,
      "step": 3170
    },
    {
      "epoch": 0.37093377018488183,
      "loss_main": 0.10263548791408539,
      "step": 3170
    },
    {
      "epoch": 0.37210390826117484,
      "grad_norm": 0.6703668236732483,
      "learning_rate": 9.999756671722328e-05,
      "loss": 0.1105,
      "step": 3180
    },
    {
      "epoch": 0.37210390826117484,
      "loss_main": 0.09715233743190765,
      "step": 3180
    },
    {
      "epoch": 0.3732740463374678,
      "grad_norm": 0.43745213747024536,
      "learning_rate": 9.99972872503013e-05,
      "loss": 0.0997,
      "step": 3190
    },
    {
      "epoch": 0.3732740463374678,
      "loss_main": 0.11972832679748535,
      "step": 3190
    },
    {
      "epoch": 0.3744441844137608,
      "grad_norm": 0.5290272831916809,
      "learning_rate": 9.999699259552359e-05,
      "loss": 0.1119,
      "step": 3200
    },
    {
      "epoch": 0.3744441844137608,
      "loss_main": 0.0877801775932312,
      "step": 3200
    },
    {
      "epoch": 0.37561432249005383,
      "grad_norm": 0.27732715010643005,
      "learning_rate": 9.99966827529796e-05,
      "loss": 0.1017,
      "step": 3210
    },
    {
      "epoch": 0.37561432249005383,
      "loss_main": 0.10583069175481796,
      "step": 3210
    },
    {
      "epoch": 0.37678446056634685,
      "grad_norm": 0.42592853307724,
      "learning_rate": 9.999635772276348e-05,
      "loss": 0.0991,
      "step": 3220
    },
    {
      "epoch": 0.37678446056634685,
      "loss_main": 0.07388588041067123,
      "step": 3220
    },
    {
      "epoch": 0.3779545986426398,
      "grad_norm": 0.371137410402298,
      "learning_rate": 9.999601750497396e-05,
      "loss": 0.0982,
      "step": 3230
    },
    {
      "epoch": 0.3779545986426398,
      "loss_main": 0.06996307522058487,
      "step": 3230
    },
    {
      "epoch": 0.3791247367189328,
      "grad_norm": 0.49171069264411926,
      "learning_rate": 9.99956620997144e-05,
      "loss": 0.1007,
      "step": 3240
    },
    {
      "epoch": 0.3791247367189328,
      "loss_main": 0.06894507259130478,
      "step": 3240
    },
    {
      "epoch": 0.38029487479522583,
      "grad_norm": 0.38170376420021057,
      "learning_rate": 9.999529150709275e-05,
      "loss": 0.0996,
      "step": 3250
    },
    {
      "epoch": 0.38029487479522583,
      "loss_main": 0.07376425713300705,
      "step": 3250
    },
    {
      "epoch": 0.38146501287151885,
      "grad_norm": 0.4756709039211273,
      "learning_rate": 9.999490572722158e-05,
      "loss": 0.0953,
      "step": 3260
    },
    {
      "epoch": 0.38146501287151885,
      "loss_main": 0.12079121172428131,
      "step": 3260
    },
    {
      "epoch": 0.38263515094781186,
      "grad_norm": 0.5873942971229553,
      "learning_rate": 9.99945047602181e-05,
      "loss": 0.0996,
      "step": 3270
    },
    {
      "epoch": 0.38263515094781186,
      "loss_main": 0.100596584379673,
      "step": 3270
    },
    {
      "epoch": 0.3838052890241048,
      "grad_norm": 0.3872354328632355,
      "learning_rate": 9.99940886062041e-05,
      "loss": 0.1131,
      "step": 3280
    },
    {
      "epoch": 0.3838052890241048,
      "loss_main": 0.11417724192142487,
      "step": 3280
    },
    {
      "epoch": 0.38497542710039784,
      "grad_norm": 0.5534605979919434,
      "learning_rate": 9.999365726530599e-05,
      "loss": 0.1091,
      "step": 3290
    },
    {
      "epoch": 0.38497542710039784,
      "loss_main": 0.14218683540821075,
      "step": 3290
    },
    {
      "epoch": 0.38614556517669085,
      "grad_norm": 0.5426556468009949,
      "learning_rate": 9.999321073765481e-05,
      "loss": 0.1134,
      "step": 3300
    },
    {
      "epoch": 0.38614556517669085,
      "loss_main": 0.09193885326385498,
      "step": 3300
    },
    {
      "epoch": 0.38731570325298387,
      "grad_norm": 0.4357473850250244,
      "learning_rate": 9.99927490233862e-05,
      "loss": 0.0874,
      "step": 3310
    },
    {
      "epoch": 0.38731570325298387,
      "loss_main": 0.08559874445199966,
      "step": 3310
    },
    {
      "epoch": 0.3884858413292769,
      "grad_norm": 0.5806319117546082,
      "learning_rate": 9.999227212264043e-05,
      "loss": 0.0951,
      "step": 3320
    },
    {
      "epoch": 0.3884858413292769,
      "loss_main": 0.06658326089382172,
      "step": 3320
    },
    {
      "epoch": 0.38965597940556984,
      "grad_norm": 0.5066753029823303,
      "learning_rate": 9.999178003556236e-05,
      "loss": 0.1078,
      "step": 3330
    },
    {
      "epoch": 0.38965597940556984,
      "loss_main": 0.10527319461107254,
      "step": 3330
    },
    {
      "epoch": 0.39082611748186286,
      "grad_norm": 0.3004814386367798,
      "learning_rate": 9.999127276230146e-05,
      "loss": 0.1079,
      "step": 3340
    },
    {
      "epoch": 0.39082611748186286,
      "loss_main": 0.09745492041110992,
      "step": 3340
    },
    {
      "epoch": 0.39199625555815587,
      "grad_norm": 0.6100115776062012,
      "learning_rate": 9.999075030301184e-05,
      "loss": 0.1082,
      "step": 3350
    },
    {
      "epoch": 0.39199625555815587,
      "loss_main": 0.08850966393947601,
      "step": 3350
    },
    {
      "epoch": 0.3931663936344489,
      "grad_norm": 0.8912506103515625,
      "learning_rate": 9.999021265785221e-05,
      "loss": 0.105,
      "step": 3360
    },
    {
      "epoch": 0.3931663936344489,
      "loss_main": 0.06888654083013535,
      "step": 3360
    },
    {
      "epoch": 0.39433653171074184,
      "grad_norm": 0.5572630763053894,
      "learning_rate": 9.998965982698589e-05,
      "loss": 0.1157,
      "step": 3370
    },
    {
      "epoch": 0.39433653171074184,
      "loss_main": 0.0873059630393982,
      "step": 3370
    },
    {
      "epoch": 0.39550666978703486,
      "grad_norm": 0.4885419011116028,
      "learning_rate": 9.998909181058082e-05,
      "loss": 0.0966,
      "step": 3380
    },
    {
      "epoch": 0.39550666978703486,
      "loss_main": 0.0840466096997261,
      "step": 3380
    },
    {
      "epoch": 0.3966768078633279,
      "grad_norm": 0.4264112114906311,
      "learning_rate": 9.998850860880953e-05,
      "loss": 0.1012,
      "step": 3390
    },
    {
      "epoch": 0.3966768078633279,
      "loss_main": 0.08837771415710449,
      "step": 3390
    },
    {
      "epoch": 0.3978469459396209,
      "grad_norm": 0.39200201630592346,
      "learning_rate": 9.998791022184922e-05,
      "loss": 0.1017,
      "step": 3400
    },
    {
      "epoch": 0.3978469459396209,
      "loss_main": 0.046622488647699356,
      "step": 3400
    },
    {
      "epoch": 0.3990170840159139,
      "grad_norm": 0.6601847410202026,
      "learning_rate": 9.99872966498816e-05,
      "loss": 0.0905,
      "step": 3410
    },
    {
      "epoch": 0.3990170840159139,
      "loss_main": 0.071583591401577,
      "step": 3410
    },
    {
      "epoch": 0.40018722209220686,
      "grad_norm": 0.39631393551826477,
      "learning_rate": 9.998666789309313e-05,
      "loss": 0.1071,
      "step": 3420
    },
    {
      "epoch": 0.40018722209220686,
      "loss_main": 0.09798603504896164,
      "step": 3420
    },
    {
      "epoch": 0.4013573601684999,
      "grad_norm": 0.4876513183116913,
      "learning_rate": 9.998602395167475e-05,
      "loss": 0.0996,
      "step": 3430
    },
    {
      "epoch": 0.4013573601684999,
      "loss_main": 0.122747503221035,
      "step": 3430
    },
    {
      "epoch": 0.4025274982447929,
      "grad_norm": 0.44517430663108826,
      "learning_rate": 9.998536482582213e-05,
      "loss": 0.1114,
      "step": 3440
    },
    {
      "epoch": 0.4025274982447929,
      "loss_main": 0.11068686097860336,
      "step": 3440
    },
    {
      "epoch": 0.4036976363210859,
      "grad_norm": 0.6135660409927368,
      "learning_rate": 9.998469051573544e-05,
      "loss": 0.1108,
      "step": 3450
    },
    {
      "epoch": 0.4036976363210859,
      "loss_main": 0.1077418178319931,
      "step": 3450
    },
    {
      "epoch": 0.40486777439737887,
      "grad_norm": 0.382659375667572,
      "learning_rate": 9.998400102161954e-05,
      "loss": 0.1016,
      "step": 3460
    },
    {
      "epoch": 0.40486777439737887,
      "loss_main": 0.12033494561910629,
      "step": 3460
    },
    {
      "epoch": 0.4060379124736719,
      "grad_norm": 0.3965802490711212,
      "learning_rate": 9.998329634368388e-05,
      "loss": 0.0948,
      "step": 3470
    },
    {
      "epoch": 0.4060379124736719,
      "loss_main": 0.07204610854387283,
      "step": 3470
    },
    {
      "epoch": 0.4072080505499649,
      "grad_norm": 0.45050016045570374,
      "learning_rate": 9.998257648214253e-05,
      "loss": 0.0935,
      "step": 3480
    },
    {
      "epoch": 0.4072080505499649,
      "loss_main": 0.15171422064304352,
      "step": 3480
    },
    {
      "epoch": 0.4083781886262579,
      "grad_norm": 0.5436944961547852,
      "learning_rate": 9.998184143721417e-05,
      "loss": 0.1018,
      "step": 3490
    },
    {
      "epoch": 0.4083781886262579,
      "loss_main": 0.07398683577775955,
      "step": 3490
    },
    {
      "epoch": 0.4095483267025509,
      "grad_norm": 0.4176469147205353,
      "learning_rate": 9.998109120912206e-05,
      "loss": 0.1024,
      "step": 3500
    },
    {
      "epoch": 0.4095483267025509,
      "loss_main": 0.12884780764579773,
      "step": 3500
    },
    {
      "epoch": 0.4107184647788439,
      "grad_norm": 0.5336498022079468,
      "learning_rate": 9.998032579809411e-05,
      "loss": 0.1111,
      "step": 3510
    },
    {
      "epoch": 0.4107184647788439,
      "loss_main": 0.12732240557670593,
      "step": 3510
    },
    {
      "epoch": 0.4118886028551369,
      "grad_norm": 0.5132617354393005,
      "learning_rate": 9.997954520436286e-05,
      "loss": 0.1066,
      "step": 3520
    },
    {
      "epoch": 0.4118886028551369,
      "loss_main": 0.07993578910827637,
      "step": 3520
    },
    {
      "epoch": 0.4130587409314299,
      "grad_norm": 0.5496331453323364,
      "learning_rate": 9.997874942816538e-05,
      "loss": 0.0894,
      "step": 3530
    },
    {
      "epoch": 0.4130587409314299,
      "loss_main": 0.10358471423387527,
      "step": 3530
    },
    {
      "epoch": 0.4142288790077229,
      "grad_norm": 0.40976718068122864,
      "learning_rate": 9.997793846974345e-05,
      "loss": 0.1026,
      "step": 3540
    },
    {
      "epoch": 0.4142288790077229,
      "loss_main": 0.06809455901384354,
      "step": 3540
    },
    {
      "epoch": 0.4153990170840159,
      "grad_norm": 0.36299803853034973,
      "learning_rate": 9.997711232934341e-05,
      "loss": 0.0998,
      "step": 3550
    },
    {
      "epoch": 0.4153990170840159,
      "loss_main": 0.13351045548915863,
      "step": 3550
    },
    {
      "epoch": 0.4165691551603089,
      "grad_norm": 0.4376254975795746,
      "learning_rate": 9.99762710072162e-05,
      "loss": 0.1012,
      "step": 3560
    },
    {
      "epoch": 0.4165691551603089,
      "loss_main": 0.07084742933511734,
      "step": 3560
    },
    {
      "epoch": 0.4177392932366019,
      "grad_norm": 0.372808039188385,
      "learning_rate": 9.997541450361743e-05,
      "loss": 0.1054,
      "step": 3570
    },
    {
      "epoch": 0.4177392932366019,
      "loss_main": 0.12981727719306946,
      "step": 3570
    },
    {
      "epoch": 0.41890943131289493,
      "grad_norm": 0.530205488204956,
      "learning_rate": 9.997454281880723e-05,
      "loss": 0.1121,
      "step": 3580
    },
    {
      "epoch": 0.41890943131289493,
      "loss_main": 0.07159893959760666,
      "step": 3580
    },
    {
      "epoch": 0.42007956938918795,
      "grad_norm": 0.48605969548225403,
      "learning_rate": 9.997365595305044e-05,
      "loss": 0.1031,
      "step": 3590
    },
    {
      "epoch": 0.42007956938918795,
      "loss_main": 0.11456155776977539,
      "step": 3590
    },
    {
      "epoch": 0.4212497074654809,
      "grad_norm": 0.5114377737045288,
      "learning_rate": 9.997275390661644e-05,
      "loss": 0.0984,
      "step": 3600
    },
    {
      "epoch": 0.4212497074654809,
      "loss_main": 0.1426023691892624,
      "step": 3600
    },
    {
      "epoch": 0.4224198455417739,
      "grad_norm": 0.4876452386379242,
      "learning_rate": 9.997183667977926e-05,
      "loss": 0.1066,
      "step": 3610
    },
    {
      "epoch": 0.4224198455417739,
      "loss_main": 0.09134469926357269,
      "step": 3610
    },
    {
      "epoch": 0.42358998361806693,
      "grad_norm": 0.516025185585022,
      "learning_rate": 9.997090427281752e-05,
      "loss": 0.1085,
      "step": 3620
    },
    {
      "epoch": 0.42358998361806693,
      "loss_main": 0.099266417324543,
      "step": 3620
    },
    {
      "epoch": 0.42476012169435995,
      "grad_norm": 0.45513996481895447,
      "learning_rate": 9.996995668601448e-05,
      "loss": 0.1011,
      "step": 3630
    },
    {
      "epoch": 0.42476012169435995,
      "loss_main": 0.1298159956932068,
      "step": 3630
    },
    {
      "epoch": 0.42593025977065296,
      "grad_norm": 0.5606253743171692,
      "learning_rate": 9.996899391965798e-05,
      "loss": 0.1027,
      "step": 3640
    },
    {
      "epoch": 0.42593025977065296,
      "loss_main": 0.07703876495361328,
      "step": 3640
    },
    {
      "epoch": 0.4271003978469459,
      "grad_norm": 0.6874237656593323,
      "learning_rate": 9.996801597404048e-05,
      "loss": 0.1019,
      "step": 3650
    },
    {
      "epoch": 0.4271003978469459,
      "loss_main": 0.13167209923267365,
      "step": 3650
    },
    {
      "epoch": 0.42827053592323894,
      "grad_norm": 0.4870017468929291,
      "learning_rate": 9.996702284945905e-05,
      "loss": 0.095,
      "step": 3660
    },
    {
      "epoch": 0.42827053592323894,
      "loss_main": 0.08140086382627487,
      "step": 3660
    },
    {
      "epoch": 0.42944067399953195,
      "grad_norm": 0.4046310484409332,
      "learning_rate": 9.996601454621539e-05,
      "loss": 0.1051,
      "step": 3670
    },
    {
      "epoch": 0.42944067399953195,
      "loss_main": 0.1155235543847084,
      "step": 3670
    },
    {
      "epoch": 0.43061081207582497,
      "grad_norm": 0.4837292730808258,
      "learning_rate": 9.996499106461577e-05,
      "loss": 0.0954,
      "step": 3680
    },
    {
      "epoch": 0.43061081207582497,
      "loss_main": 0.09344494342803955,
      "step": 3680
    },
    {
      "epoch": 0.4317809501521179,
      "grad_norm": 0.44275397062301636,
      "learning_rate": 9.996395240497112e-05,
      "loss": 0.0947,
      "step": 3690
    },
    {
      "epoch": 0.4317809501521179,
      "loss_main": 0.1107410341501236,
      "step": 3690
    },
    {
      "epoch": 0.43295108822841094,
      "grad_norm": 0.4148675203323364,
      "learning_rate": 9.996289856759696e-05,
      "loss": 0.0961,
      "step": 3700
    },
    {
      "epoch": 0.43295108822841094,
      "loss_main": 0.08532965928316116,
      "step": 3700
    },
    {
      "epoch": 0.43412122630470396,
      "grad_norm": 0.3873465657234192,
      "learning_rate": 9.996182955281342e-05,
      "loss": 0.0929,
      "step": 3710
    },
    {
      "epoch": 0.43412122630470396,
      "loss_main": 0.11149581521749496,
      "step": 3710
    },
    {
      "epoch": 0.43529136438099697,
      "grad_norm": 0.5543604493141174,
      "learning_rate": 9.996074536094519e-05,
      "loss": 0.0959,
      "step": 3720
    },
    {
      "epoch": 0.43529136438099697,
      "loss_main": 0.11689816415309906,
      "step": 3720
    },
    {
      "epoch": 0.43646150245729,
      "grad_norm": 0.4705604612827301,
      "learning_rate": 9.995964599232168e-05,
      "loss": 0.0933,
      "step": 3730
    },
    {
      "epoch": 0.43646150245729,
      "loss_main": 0.12103144824504852,
      "step": 3730
    },
    {
      "epoch": 0.43763164053358294,
      "grad_norm": 0.4693765938282013,
      "learning_rate": 9.995853144727683e-05,
      "loss": 0.0952,
      "step": 3740
    },
    {
      "epoch": 0.43763164053358294,
      "loss_main": 0.12166944891214371,
      "step": 3740
    },
    {
      "epoch": 0.43880177860987596,
      "grad_norm": 0.4579930305480957,
      "learning_rate": 9.99574017261492e-05,
      "loss": 0.0977,
      "step": 3750
    },
    {
      "epoch": 0.43880177860987596,
      "loss_main": 0.07520835846662521,
      "step": 3750
    },
    {
      "epoch": 0.439971916686169,
      "grad_norm": 0.36227431893348694,
      "learning_rate": 9.995625682928198e-05,
      "loss": 0.09,
      "step": 3760
    },
    {
      "epoch": 0.439971916686169,
      "loss_main": 0.05803636088967323,
      "step": 3760
    },
    {
      "epoch": 0.441142054762462,
      "grad_norm": 0.3341940641403198,
      "learning_rate": 9.995509675702295e-05,
      "loss": 0.0847,
      "step": 3770
    },
    {
      "epoch": 0.441142054762462,
      "loss_main": 0.0682251825928688,
      "step": 3770
    },
    {
      "epoch": 0.44231219283875495,
      "grad_norm": 0.40485838055610657,
      "learning_rate": 9.995392150972451e-05,
      "loss": 0.0928,
      "step": 3780
    },
    {
      "epoch": 0.44231219283875495,
      "loss_main": 0.11973050981760025,
      "step": 3780
    },
    {
      "epoch": 0.44348233091504796,
      "grad_norm": 0.3505423069000244,
      "learning_rate": 9.995273108774366e-05,
      "loss": 0.0993,
      "step": 3790
    },
    {
      "epoch": 0.44348233091504796,
      "loss_main": 0.07730156183242798,
      "step": 3790
    },
    {
      "epoch": 0.444652468991341,
      "grad_norm": 0.5070841908454895,
      "learning_rate": 9.995152549144205e-05,
      "loss": 0.1119,
      "step": 3800
    },
    {
      "epoch": 0.444652468991341,
      "loss_main": 0.15552592277526855,
      "step": 3800
    },
    {
      "epoch": 0.445822607067634,
      "grad_norm": 0.5504462122917175,
      "learning_rate": 9.995030472118587e-05,
      "loss": 0.1034,
      "step": 3810
    },
    {
      "epoch": 0.445822607067634,
      "loss_main": 0.11588384211063385,
      "step": 3810
    },
    {
      "epoch": 0.446992745143927,
      "grad_norm": 0.4038940966129303,
      "learning_rate": 9.9949068777346e-05,
      "loss": 0.1038,
      "step": 3820
    },
    {
      "epoch": 0.446992745143927,
      "loss_main": 0.12133939564228058,
      "step": 3820
    },
    {
      "epoch": 0.44816288322021997,
      "grad_norm": 0.3464316725730896,
      "learning_rate": 9.994781766029786e-05,
      "loss": 0.1031,
      "step": 3830
    },
    {
      "epoch": 0.44816288322021997,
      "loss_main": 0.12504377961158752,
      "step": 3830
    },
    {
      "epoch": 0.449333021296513,
      "grad_norm": 0.42756351828575134,
      "learning_rate": 9.994655137042151e-05,
      "loss": 0.1053,
      "step": 3840
    },
    {
      "epoch": 0.449333021296513,
      "loss_main": 0.10372311621904373,
      "step": 3840
    },
    {
      "epoch": 0.450503159372806,
      "grad_norm": 0.40676042437553406,
      "learning_rate": 9.99452699081016e-05,
      "loss": 0.0959,
      "step": 3850
    },
    {
      "epoch": 0.450503159372806,
      "loss_main": 0.13338978588581085,
      "step": 3850
    },
    {
      "epoch": 0.451673297449099,
      "grad_norm": 0.4370421767234802,
      "learning_rate": 9.994397327372743e-05,
      "loss": 0.1024,
      "step": 3860
    },
    {
      "epoch": 0.451673297449099,
      "loss_main": 0.10295724123716354,
      "step": 3860
    },
    {
      "epoch": 0.45284343552539197,
      "grad_norm": 0.2961949110031128,
      "learning_rate": 9.994266146769286e-05,
      "loss": 0.0974,
      "step": 3870
    },
    {
      "epoch": 0.45284343552539197,
      "loss_main": 0.0955820232629776,
      "step": 3870
    },
    {
      "epoch": 0.454013573601685,
      "grad_norm": 0.46605992317199707,
      "learning_rate": 9.994133449039642e-05,
      "loss": 0.0937,
      "step": 3880
    },
    {
      "epoch": 0.454013573601685,
      "loss_main": 0.08201747387647629,
      "step": 3880
    },
    {
      "epoch": 0.455183711677978,
      "grad_norm": 0.5537037253379822,
      "learning_rate": 9.993999234224118e-05,
      "loss": 0.0959,
      "step": 3890
    },
    {
      "epoch": 0.455183711677978,
      "loss_main": 0.14924634993076324,
      "step": 3890
    },
    {
      "epoch": 0.456353849754271,
      "grad_norm": 0.4471346437931061,
      "learning_rate": 9.993863502363485e-05,
      "loss": 0.0969,
      "step": 3900
    },
    {
      "epoch": 0.456353849754271,
      "loss_main": 0.0749816745519638,
      "step": 3900
    },
    {
      "epoch": 0.457523987830564,
      "grad_norm": 0.5134292840957642,
      "learning_rate": 9.993726253498976e-05,
      "loss": 0.0866,
      "step": 3910
    },
    {
      "epoch": 0.457523987830564,
      "loss_main": 0.056860458105802536,
      "step": 3910
    },
    {
      "epoch": 0.458694125906857,
      "grad_norm": 0.4411982297897339,
      "learning_rate": 9.993587487672282e-05,
      "loss": 0.0925,
      "step": 3920
    },
    {
      "epoch": 0.458694125906857,
      "loss_main": 0.12056776136159897,
      "step": 3920
    },
    {
      "epoch": 0.45986426398315,
      "grad_norm": 0.4535519480705261,
      "learning_rate": 9.993447204925558e-05,
      "loss": 0.0935,
      "step": 3930
    },
    {
      "epoch": 0.45986426398315,
      "loss_main": 0.09930424392223358,
      "step": 3930
    },
    {
      "epoch": 0.461034402059443,
      "grad_norm": 0.6544529795646667,
      "learning_rate": 9.993305405301416e-05,
      "loss": 0.1085,
      "step": 3940
    },
    {
      "epoch": 0.461034402059443,
      "loss_main": 0.14495719969272614,
      "step": 3940
    },
    {
      "epoch": 0.46220454013573603,
      "grad_norm": 0.36774253845214844,
      "learning_rate": 9.993162088842935e-05,
      "loss": 0.0992,
      "step": 3950
    },
    {
      "epoch": 0.46220454013573603,
      "loss_main": 0.13549605011940002,
      "step": 3950
    },
    {
      "epoch": 0.46337467821202905,
      "grad_norm": 0.4749549925327301,
      "learning_rate": 9.993017255593646e-05,
      "loss": 0.1155,
      "step": 3960
    },
    {
      "epoch": 0.46337467821202905,
      "loss_main": 0.06875137984752655,
      "step": 3960
    },
    {
      "epoch": 0.464544816288322,
      "grad_norm": 0.39194369316101074,
      "learning_rate": 9.992870905597548e-05,
      "loss": 0.0937,
      "step": 3970
    },
    {
      "epoch": 0.464544816288322,
      "loss_main": 0.09515629708766937,
      "step": 3970
    },
    {
      "epoch": 0.465714954364615,
      "grad_norm": 0.36752453446388245,
      "learning_rate": 9.9927230388991e-05,
      "loss": 0.0947,
      "step": 3980
    },
    {
      "epoch": 0.465714954364615,
      "loss_main": 0.13359056413173676,
      "step": 3980
    },
    {
      "epoch": 0.46688509244090803,
      "grad_norm": 0.5206919312477112,
      "learning_rate": 9.992573655543215e-05,
      "loss": 0.0957,
      "step": 3990
    },
    {
      "epoch": 0.46688509244090803,
      "loss_main": 0.06109964847564697,
      "step": 3990
    },
    {
      "epoch": 0.46805523051720105,
      "grad_norm": 0.44961994886398315,
      "learning_rate": 9.992422755575277e-05,
      "loss": 0.091,
      "step": 4000
    },
    {
      "epoch": 0.46805523051720105,
      "loss_main": 0.08978451788425446,
      "step": 4000
    },
    {
      "epoch": 0.469225368593494,
      "grad_norm": 0.4105890393257141,
      "learning_rate": 9.992270339041123e-05,
      "loss": 0.0891,
      "step": 4010
    },
    {
      "epoch": 0.469225368593494,
      "loss_main": 0.12050231546163559,
      "step": 4010
    },
    {
      "epoch": 0.470395506669787,
      "grad_norm": 0.3137156665325165,
      "learning_rate": 9.992116405987053e-05,
      "loss": 0.0908,
      "step": 4020
    },
    {
      "epoch": 0.470395506669787,
      "loss_main": 0.08744890242815018,
      "step": 4020
    },
    {
      "epoch": 0.47156564474608004,
      "grad_norm": 0.41784384846687317,
      "learning_rate": 9.991960956459828e-05,
      "loss": 0.0863,
      "step": 4030
    },
    {
      "epoch": 0.47156564474608004,
      "loss_main": 0.10532396286725998,
      "step": 4030
    },
    {
      "epoch": 0.47273578282237305,
      "grad_norm": 0.3649272322654724,
      "learning_rate": 9.991803990506669e-05,
      "loss": 0.0948,
      "step": 4040
    },
    {
      "epoch": 0.47273578282237305,
      "loss_main": 0.08440331369638443,
      "step": 4040
    },
    {
      "epoch": 0.47390592089866607,
      "grad_norm": 0.4426640272140503,
      "learning_rate": 9.991645508175258e-05,
      "loss": 0.0959,
      "step": 4050
    },
    {
      "epoch": 0.47390592089866607,
      "loss_main": 0.15854628384113312,
      "step": 4050
    },
    {
      "epoch": 0.475076058974959,
      "grad_norm": 0.3127923309803009,
      "learning_rate": 9.99148550951374e-05,
      "loss": 0.0928,
      "step": 4060
    },
    {
      "epoch": 0.475076058974959,
      "loss_main": 0.10989123582839966,
      "step": 4060
    },
    {
      "epoch": 0.47624619705125204,
      "grad_norm": 0.2785828709602356,
      "learning_rate": 9.991323994570716e-05,
      "loss": 0.1014,
      "step": 4070
    },
    {
      "epoch": 0.47624619705125204,
      "loss_main": 0.05988476797938347,
      "step": 4070
    },
    {
      "epoch": 0.47741633512754506,
      "grad_norm": 0.39819443225860596,
      "learning_rate": 9.99116096339525e-05,
      "loss": 0.1003,
      "step": 4080
    },
    {
      "epoch": 0.47741633512754506,
      "loss_main": 0.09917156398296356,
      "step": 4080
    },
    {
      "epoch": 0.47858647320383807,
      "grad_norm": 0.43889281153678894,
      "learning_rate": 9.990996416036869e-05,
      "loss": 0.0964,
      "step": 4090
    },
    {
      "epoch": 0.47858647320383807,
      "loss_main": 0.10502935945987701,
      "step": 4090
    },
    {
      "epoch": 0.47975661128013103,
      "grad_norm": 0.451034814119339,
      "learning_rate": 9.990830352545555e-05,
      "loss": 0.0965,
      "step": 4100
    },
    {
      "epoch": 0.47975661128013103,
      "loss_main": 0.08525709062814713,
      "step": 4100
    },
    {
      "epoch": 0.48092674935642404,
      "grad_norm": 0.32591333985328674,
      "learning_rate": 9.990662772971756e-05,
      "loss": 0.0901,
      "step": 4110
    },
    {
      "epoch": 0.48092674935642404,
      "loss_main": 0.13000896573066711,
      "step": 4110
    },
    {
      "epoch": 0.48209688743271706,
      "grad_norm": 0.30214744806289673,
      "learning_rate": 9.990493677366376e-05,
      "loss": 0.0909,
      "step": 4120
    },
    {
      "epoch": 0.48209688743271706,
      "loss_main": 0.08544492721557617,
      "step": 4120
    },
    {
      "epoch": 0.4832670255090101,
      "grad_norm": 0.44093167781829834,
      "learning_rate": 9.990323065780786e-05,
      "loss": 0.0796,
      "step": 4130
    },
    {
      "epoch": 0.4832670255090101,
      "loss_main": 0.11266531050205231,
      "step": 4130
    },
    {
      "epoch": 0.4844371635853031,
      "grad_norm": 0.5168373584747314,
      "learning_rate": 9.990150938266808e-05,
      "loss": 0.099,
      "step": 4140
    },
    {
      "epoch": 0.4844371635853031,
      "loss_main": 0.08589601516723633,
      "step": 4140
    },
    {
      "epoch": 0.48560730166159605,
      "grad_norm": 0.5366542935371399,
      "learning_rate": 9.989977294876733e-05,
      "loss": 0.1045,
      "step": 4150
    },
    {
      "epoch": 0.48560730166159605,
      "loss_main": 0.13189736008644104,
      "step": 4150
    },
    {
      "epoch": 0.48677743973788906,
      "grad_norm": 0.4223063588142395,
      "learning_rate": 9.989802135663308e-05,
      "loss": 0.0886,
      "step": 4160
    },
    {
      "epoch": 0.48677743973788906,
      "loss_main": 0.09620412439107895,
      "step": 4160
    },
    {
      "epoch": 0.4879475778141821,
      "grad_norm": 0.4322131872177124,
      "learning_rate": 9.989625460679743e-05,
      "loss": 0.0785,
      "step": 4170
    },
    {
      "epoch": 0.4879475778141821,
      "loss_main": 0.13368700444698334,
      "step": 4170
    },
    {
      "epoch": 0.4891177158904751,
      "grad_norm": 0.4326270520687103,
      "learning_rate": 9.989447269979706e-05,
      "loss": 0.0943,
      "step": 4180
    },
    {
      "epoch": 0.4891177158904751,
      "loss_main": 0.09939561784267426,
      "step": 4180
    },
    {
      "epoch": 0.4902878539667681,
      "grad_norm": 0.3737764358520508,
      "learning_rate": 9.989267563617328e-05,
      "loss": 0.0933,
      "step": 4190
    },
    {
      "epoch": 0.4902878539667681,
      "loss_main": 0.067879818379879,
      "step": 4190
    },
    {
      "epoch": 0.49145799204306106,
      "grad_norm": 0.38852304220199585,
      "learning_rate": 9.989086341647198e-05,
      "loss": 0.0865,
      "step": 4200
    },
    {
      "epoch": 0.49145799204306106,
      "loss_main": 0.052740681916475296,
      "step": 4200
    },
    {
      "epoch": 0.4926281301193541,
      "grad_norm": 0.35175463557243347,
      "learning_rate": 9.988903604124366e-05,
      "loss": 0.0893,
      "step": 4210
    },
    {
      "epoch": 0.4926281301193541,
      "loss_main": 0.06365234404802322,
      "step": 4210
    },
    {
      "epoch": 0.4937982681956471,
      "grad_norm": 0.4283531904220581,
      "learning_rate": 9.988719351104343e-05,
      "loss": 0.0879,
      "step": 4220
    },
    {
      "epoch": 0.4937982681956471,
      "loss_main": 0.06906317919492722,
      "step": 4220
    },
    {
      "epoch": 0.4949684062719401,
      "grad_norm": 0.4179007112979889,
      "learning_rate": 9.9885335826431e-05,
      "loss": 0.0813,
      "step": 4230
    },
    {
      "epoch": 0.4949684062719401,
      "loss_main": 0.07525471597909927,
      "step": 4230
    },
    {
      "epoch": 0.49613854434823307,
      "grad_norm": 0.37062740325927734,
      "learning_rate": 9.988346298797071e-05,
      "loss": 0.0925,
      "step": 4240
    },
    {
      "epoch": 0.49613854434823307,
      "loss_main": 0.09426421672105789,
      "step": 4240
    },
    {
      "epoch": 0.4973086824245261,
      "grad_norm": 0.46259480714797974,
      "learning_rate": 9.988157499623146e-05,
      "loss": 0.1022,
      "step": 4250
    },
    {
      "epoch": 0.4973086824245261,
      "loss_main": 0.10298141092061996,
      "step": 4250
    },
    {
      "epoch": 0.4984788205008191,
      "grad_norm": 0.2616429030895233,
      "learning_rate": 9.987967185178677e-05,
      "loss": 0.0951,
      "step": 4260
    },
    {
      "epoch": 0.4984788205008191,
      "loss_main": 0.10801894217729568,
      "step": 4260
    },
    {
      "epoch": 0.4996489585771121,
      "grad_norm": 0.5783679485321045,
      "learning_rate": 9.987775355521476e-05,
      "loss": 0.103,
      "step": 4270
    },
    {
      "epoch": 0.4996489585771121,
      "loss_main": 0.1216670423746109,
      "step": 4270
    },
    {
      "epoch": 0.5008190966534051,
      "grad_norm": 0.3307957351207733,
      "learning_rate": 9.987582010709817e-05,
      "loss": 0.0976,
      "step": 4280
    },
    {
      "epoch": 0.5008190966534051,
      "loss_main": 0.15248124301433563,
      "step": 4280
    },
    {
      "epoch": 0.5019892347296981,
      "grad_norm": 0.46286308765411377,
      "learning_rate": 9.987387150802431e-05,
      "loss": 0.1178,
      "step": 4290
    },
    {
      "epoch": 0.5019892347296981,
      "loss_main": 0.08290628343820572,
      "step": 4290
    },
    {
      "epoch": 0.5031593728059911,
      "grad_norm": 0.3866993188858032,
      "learning_rate": 9.987190775858517e-05,
      "loss": 0.0903,
      "step": 4300
    },
    {
      "epoch": 0.5031593728059911,
      "loss_main": 0.073097363114357,
      "step": 4300
    },
    {
      "epoch": 0.5043295108822841,
      "grad_norm": 0.3878912329673767,
      "learning_rate": 9.98699288593772e-05,
      "loss": 0.0941,
      "step": 4310
    },
    {
      "epoch": 0.5043295108822841,
      "loss_main": 0.0749451145529747,
      "step": 4310
    },
    {
      "epoch": 0.5054996489585771,
      "grad_norm": 0.4522467851638794,
      "learning_rate": 9.986793481100161e-05,
      "loss": 0.0965,
      "step": 4320
    },
    {
      "epoch": 0.5054996489585771,
      "loss_main": 0.10843467712402344,
      "step": 4320
    },
    {
      "epoch": 0.5066697870348701,
      "grad_norm": 0.4759817123413086,
      "learning_rate": 9.986592561406412e-05,
      "loss": 0.0994,
      "step": 4330
    },
    {
      "epoch": 0.5066697870348701,
      "loss_main": 0.14137311279773712,
      "step": 4330
    },
    {
      "epoch": 0.5078399251111632,
      "grad_norm": 0.4754449427127838,
      "learning_rate": 9.986390126917503e-05,
      "loss": 0.0804,
      "step": 4340
    },
    {
      "epoch": 0.5078399251111632,
      "loss_main": 0.10125239938497543,
      "step": 4340
    },
    {
      "epoch": 0.5090100631874561,
      "grad_norm": 0.3891715407371521,
      "learning_rate": 9.986186177694933e-05,
      "loss": 0.092,
      "step": 4350
    },
    {
      "epoch": 0.5090100631874561,
      "loss_main": 0.07734432071447372,
      "step": 4350
    },
    {
      "epoch": 0.5101802012637491,
      "grad_norm": 0.442106693983078,
      "learning_rate": 9.985980713800656e-05,
      "loss": 0.1055,
      "step": 4360
    },
    {
      "epoch": 0.5101802012637491,
      "loss_main": 0.06100134178996086,
      "step": 4360
    },
    {
      "epoch": 0.5113503393400421,
      "grad_norm": 0.4565401077270508,
      "learning_rate": 9.985773735297084e-05,
      "loss": 0.0848,
      "step": 4370
    },
    {
      "epoch": 0.5113503393400421,
      "loss_main": 0.08157026767730713,
      "step": 4370
    },
    {
      "epoch": 0.5125204774163351,
      "grad_norm": 0.39779913425445557,
      "learning_rate": 9.985565242247092e-05,
      "loss": 0.0867,
      "step": 4380
    },
    {
      "epoch": 0.5125204774163351,
      "loss_main": 0.1125522181391716,
      "step": 4380
    },
    {
      "epoch": 0.5136906154926282,
      "grad_norm": 0.2989782392978668,
      "learning_rate": 9.985355234714016e-05,
      "loss": 0.1026,
      "step": 4390
    },
    {
      "epoch": 0.5136906154926282,
      "loss_main": 0.07665766030550003,
      "step": 4390
    },
    {
      "epoch": 0.5148607535689211,
      "grad_norm": 0.4581146240234375,
      "learning_rate": 9.985143712761652e-05,
      "loss": 0.0809,
      "step": 4400
    },
    {
      "epoch": 0.5148607535689211,
      "loss_main": 0.06675618141889572,
      "step": 4400
    },
    {
      "epoch": 0.5160308916452141,
      "grad_norm": 0.3344420790672302,
      "learning_rate": 9.984930676454252e-05,
      "loss": 0.0967,
      "step": 4410
    },
    {
      "epoch": 0.5160308916452141,
      "loss_main": 0.08378122746944427,
      "step": 4410
    },
    {
      "epoch": 0.5172010297215072,
      "grad_norm": 0.41833433508872986,
      "learning_rate": 9.984716125856532e-05,
      "loss": 0.0893,
      "step": 4420
    },
    {
      "epoch": 0.5172010297215072,
      "loss_main": 0.06087222322821617,
      "step": 4420
    },
    {
      "epoch": 0.5183711677978001,
      "grad_norm": 0.5209783315658569,
      "learning_rate": 9.984500061033667e-05,
      "loss": 0.0828,
      "step": 4430
    },
    {
      "epoch": 0.5183711677978001,
      "loss_main": 0.10519123077392578,
      "step": 4430
    },
    {
      "epoch": 0.5195413058740932,
      "grad_norm": 0.5240378975868225,
      "learning_rate": 9.984282482051293e-05,
      "loss": 0.0971,
      "step": 4440
    },
    {
      "epoch": 0.5195413058740932,
      "loss_main": 0.10962153971195221,
      "step": 4440
    },
    {
      "epoch": 0.5207114439503862,
      "grad_norm": 0.3898724615573883,
      "learning_rate": 9.9840633889755e-05,
      "loss": 0.0959,
      "step": 4450
    },
    {
      "epoch": 0.5207114439503862,
      "loss_main": 0.07050704956054688,
      "step": 4450
    },
    {
      "epoch": 0.5218815820266791,
      "grad_norm": 0.3984258770942688,
      "learning_rate": 9.983842781872848e-05,
      "loss": 0.1004,
      "step": 4460
    },
    {
      "epoch": 0.5218815820266791,
      "loss_main": 0.08682557195425034,
      "step": 4460
    },
    {
      "epoch": 0.5230517201029722,
      "grad_norm": 0.39782950282096863,
      "learning_rate": 9.98362066081035e-05,
      "loss": 0.0836,
      "step": 4470
    },
    {
      "epoch": 0.5230517201029722,
      "loss_main": 0.08104906231164932,
      "step": 4470
    },
    {
      "epoch": 0.5242218581792651,
      "grad_norm": 0.3828965425491333,
      "learning_rate": 9.983397025855479e-05,
      "loss": 0.0833,
      "step": 4480
    },
    {
      "epoch": 0.5242218581792651,
      "loss_main": 0.07861430197954178,
      "step": 4480
    },
    {
      "epoch": 0.5253919962555581,
      "grad_norm": 0.4203605353832245,
      "learning_rate": 9.983171877076171e-05,
      "loss": 0.0833,
      "step": 4490
    },
    {
      "epoch": 0.5253919962555581,
      "loss_main": 0.14032962918281555,
      "step": 4490
    },
    {
      "epoch": 0.5265621343318512,
      "grad_norm": 0.33922868967056274,
      "learning_rate": 9.98294521454082e-05,
      "loss": 0.1064,
      "step": 4500
    },
    {
      "epoch": 0.5265621343318512,
      "loss_main": 0.04158056527376175,
      "step": 4500
    },
    {
      "epoch": 0.5277322724081441,
      "grad_norm": 0.5778353810310364,
      "learning_rate": 9.98271703831828e-05,
      "loss": 0.0935,
      "step": 4510
    },
    {
      "epoch": 0.5277322724081441,
      "loss_main": 0.17354623973369598,
      "step": 4510
    },
    {
      "epoch": 0.5289024104844372,
      "grad_norm": 0.3856670558452606,
      "learning_rate": 9.982487348477865e-05,
      "loss": 0.1019,
      "step": 4520
    },
    {
      "epoch": 0.5289024104844372,
      "loss_main": 0.08380848169326782,
      "step": 4520
    },
    {
      "epoch": 0.5300725485607302,
      "grad_norm": 0.3523010015487671,
      "learning_rate": 9.982256145089347e-05,
      "loss": 0.0936,
      "step": 4530
    },
    {
      "epoch": 0.5300725485607302,
      "loss_main": 0.06018541008234024,
      "step": 4530
    },
    {
      "epoch": 0.5312426866370231,
      "grad_norm": 0.4319969713687897,
      "learning_rate": 9.982023428222962e-05,
      "loss": 0.079,
      "step": 4540
    },
    {
      "epoch": 0.5312426866370231,
      "loss_main": 0.23128338158130646,
      "step": 4540
    },
    {
      "epoch": 0.5324128247133162,
      "grad_norm": 0.45102882385253906,
      "learning_rate": 9.981789197949403e-05,
      "loss": 0.0979,
      "step": 4550
    },
    {
      "epoch": 0.5324128247133162,
      "loss_main": 0.10115867108106613,
      "step": 4550
    },
    {
      "epoch": 0.5335829627896091,
      "grad_norm": 0.4583125710487366,
      "learning_rate": 9.98155345433982e-05,
      "loss": 0.097,
      "step": 4560
    },
    {
      "epoch": 0.5335829627896091,
      "loss_main": 0.1248629242181778,
      "step": 4560
    },
    {
      "epoch": 0.5347531008659022,
      "grad_norm": 0.35247334837913513,
      "learning_rate": 9.981316197465831e-05,
      "loss": 0.0875,
      "step": 4570
    },
    {
      "epoch": 0.5347531008659022,
      "loss_main": 0.0990460067987442,
      "step": 4570
    },
    {
      "epoch": 0.5359232389421952,
      "grad_norm": 0.5939956903457642,
      "learning_rate": 9.981077427399504e-05,
      "loss": 0.0909,
      "step": 4580
    },
    {
      "epoch": 0.5359232389421952,
      "loss_main": 0.06459303200244904,
      "step": 4580
    },
    {
      "epoch": 0.5370933770184881,
      "grad_norm": 0.38425105810165405,
      "learning_rate": 9.980837144213371e-05,
      "loss": 0.0836,
      "step": 4590
    },
    {
      "epoch": 0.5370933770184881,
      "loss_main": 0.1077013686299324,
      "step": 4590
    },
    {
      "epoch": 0.5382635150947812,
      "grad_norm": 0.3961838483810425,
      "learning_rate": 9.980595347980426e-05,
      "loss": 0.0824,
      "step": 4600
    },
    {
      "epoch": 0.5382635150947812,
      "loss_main": 0.07352396100759506,
      "step": 4600
    },
    {
      "epoch": 0.5394336531710742,
      "grad_norm": 0.5480282306671143,
      "learning_rate": 9.980352038774119e-05,
      "loss": 0.1144,
      "step": 4610
    },
    {
      "epoch": 0.5394336531710742,
      "loss_main": 0.10069189220666885,
      "step": 4610
    },
    {
      "epoch": 0.5406037912473672,
      "grad_norm": 0.6444381475448608,
      "learning_rate": 9.98010721666836e-05,
      "loss": 0.1012,
      "step": 4620
    },
    {
      "epoch": 0.5406037912473672,
      "loss_main": 0.11238549649715424,
      "step": 4620
    },
    {
      "epoch": 0.5417739293236602,
      "grad_norm": 0.27942153811454773,
      "learning_rate": 9.979860881737523e-05,
      "loss": 0.0905,
      "step": 4630
    },
    {
      "epoch": 0.5417739293236602,
      "loss_main": 0.08005116879940033,
      "step": 4630
    },
    {
      "epoch": 0.5429440673999532,
      "grad_norm": 0.3824757933616638,
      "learning_rate": 9.979613034056434e-05,
      "loss": 0.096,
      "step": 4640
    },
    {
      "epoch": 0.5429440673999532,
      "loss_main": 0.1416776329278946,
      "step": 4640
    },
    {
      "epoch": 0.5441142054762462,
      "grad_norm": 0.344586580991745,
      "learning_rate": 9.979363673700386e-05,
      "loss": 0.0984,
      "step": 4650
    },
    {
      "epoch": 0.5441142054762462,
      "loss_main": 0.07383923977613449,
      "step": 4650
    },
    {
      "epoch": 0.5452843435525392,
      "grad_norm": 0.4724901020526886,
      "learning_rate": 9.979112800745124e-05,
      "loss": 0.1062,
      "step": 4660
    },
    {
      "epoch": 0.5452843435525392,
      "loss_main": 0.10855363309383392,
      "step": 4660
    },
    {
      "epoch": 0.5464544816288323,
      "grad_norm": 0.2666434645652771,
      "learning_rate": 9.978860415266861e-05,
      "loss": 0.0915,
      "step": 4670
    },
    {
      "epoch": 0.5464544816288323,
      "loss_main": 0.08234106004238129,
      "step": 4670
    },
    {
      "epoch": 0.5476246197051252,
      "grad_norm": 0.3528188467025757,
      "learning_rate": 9.978606517342262e-05,
      "loss": 0.0935,
      "step": 4680
    },
    {
      "epoch": 0.5476246197051252,
      "loss_main": 0.09253811091184616,
      "step": 4680
    },
    {
      "epoch": 0.5487947577814182,
      "grad_norm": 0.47722119092941284,
      "learning_rate": 9.978351107048456e-05,
      "loss": 0.0953,
      "step": 4690
    },
    {
      "epoch": 0.5487947577814182,
      "loss_main": 0.0922643318772316,
      "step": 4690
    },
    {
      "epoch": 0.5499648958577112,
      "grad_norm": 0.3667944073677063,
      "learning_rate": 9.978094184463029e-05,
      "loss": 0.1002,
      "step": 4700
    },
    {
      "epoch": 0.5499648958577112,
      "loss_main": 0.1615992933511734,
      "step": 4700
    },
    {
      "epoch": 0.5511350339340042,
      "grad_norm": 0.3968888521194458,
      "learning_rate": 9.977835749664029e-05,
      "loss": 0.095,
      "step": 4710
    },
    {
      "epoch": 0.5511350339340042,
      "loss_main": 0.13532467186450958,
      "step": 4710
    },
    {
      "epoch": 0.5523051720102972,
      "grad_norm": 0.5065628290176392,
      "learning_rate": 9.97757580272996e-05,
      "loss": 0.0944,
      "step": 4720
    },
    {
      "epoch": 0.5523051720102972,
      "loss_main": 0.0738668367266655,
      "step": 4720
    },
    {
      "epoch": 0.5534753100865902,
      "grad_norm": 0.3995242416858673,
      "learning_rate": 9.977314343739786e-05,
      "loss": 0.0899,
      "step": 4730
    },
    {
      "epoch": 0.5534753100865902,
      "loss_main": 0.14857837557792664,
      "step": 4730
    },
    {
      "epoch": 0.5546454481628832,
      "grad_norm": 0.3966899812221527,
      "learning_rate": 9.977051372772934e-05,
      "loss": 0.1063,
      "step": 4740
    },
    {
      "epoch": 0.5546454481628832,
      "loss_main": 0.12024186551570892,
      "step": 4740
    },
    {
      "epoch": 0.5558155862391763,
      "grad_norm": 0.3133266866207123,
      "learning_rate": 9.976786889909286e-05,
      "loss": 0.0875,
      "step": 4750
    },
    {
      "epoch": 0.5558155862391763,
      "loss_main": 0.08006402105093002,
      "step": 4750
    },
    {
      "epoch": 0.5569857243154692,
      "grad_norm": 0.43677908182144165,
      "learning_rate": 9.976520895229185e-05,
      "loss": 0.096,
      "step": 4760
    },
    {
      "epoch": 0.5569857243154692,
      "loss_main": 0.08081678301095963,
      "step": 4760
    },
    {
      "epoch": 0.5581558623917622,
      "grad_norm": 0.48243770003318787,
      "learning_rate": 9.976253388813433e-05,
      "loss": 0.0874,
      "step": 4770
    },
    {
      "epoch": 0.5581558623917622,
      "loss_main": 0.08184916526079178,
      "step": 4770
    },
    {
      "epoch": 0.5593260004680553,
      "grad_norm": 0.34292280673980713,
      "learning_rate": 9.975984370743293e-05,
      "loss": 0.0935,
      "step": 4780
    },
    {
      "epoch": 0.5593260004680553,
      "loss_main": 0.07436998188495636,
      "step": 4780
    },
    {
      "epoch": 0.5604961385443482,
      "grad_norm": 0.35619524121284485,
      "learning_rate": 9.975713841100485e-05,
      "loss": 0.0972,
      "step": 4790
    },
    {
      "epoch": 0.5604961385443482,
      "loss_main": 0.10431044548749924,
      "step": 4790
    },
    {
      "epoch": 0.5616662766206413,
      "grad_norm": 0.3201838433742523,
      "learning_rate": 9.975441799967187e-05,
      "loss": 0.0897,
      "step": 4800
    },
    {
      "epoch": 0.5616662766206413,
      "loss_main": 0.07595283538103104,
      "step": 4800
    },
    {
      "epoch": 0.5628364146969342,
      "grad_norm": 0.4414823055267334,
      "learning_rate": 9.975168247426039e-05,
      "loss": 0.0938,
      "step": 4810
    },
    {
      "epoch": 0.5628364146969342,
      "loss_main": 0.1295914202928543,
      "step": 4810
    },
    {
      "epoch": 0.5640065527732272,
      "grad_norm": 0.479946494102478,
      "learning_rate": 9.974893183560139e-05,
      "loss": 0.0824,
      "step": 4820
    },
    {
      "epoch": 0.5640065527732272,
      "loss_main": 0.07139494270086288,
      "step": 4820
    },
    {
      "epoch": 0.5651766908495203,
      "grad_norm": 0.2531237304210663,
      "learning_rate": 9.974616608453045e-05,
      "loss": 0.0872,
      "step": 4830
    },
    {
      "epoch": 0.5651766908495203,
      "loss_main": 0.07109349220991135,
      "step": 4830
    },
    {
      "epoch": 0.5663468289258132,
      "grad_norm": 0.41787976026535034,
      "learning_rate": 9.974338522188772e-05,
      "loss": 0.0876,
      "step": 4840
    },
    {
      "epoch": 0.5663468289258132,
      "loss_main": 0.10302098095417023,
      "step": 4840
    },
    {
      "epoch": 0.5675169670021063,
      "grad_norm": 0.40898364782333374,
      "learning_rate": 9.974058924851797e-05,
      "loss": 0.0898,
      "step": 4850
    },
    {
      "epoch": 0.5675169670021063,
      "loss_main": 0.07288868725299835,
      "step": 4850
    },
    {
      "epoch": 0.5686871050783993,
      "grad_norm": 0.23549780249595642,
      "learning_rate": 9.973777816527051e-05,
      "loss": 0.0897,
      "step": 4860
    },
    {
      "epoch": 0.5686871050783993,
      "loss_main": 0.07818011939525604,
      "step": 4860
    },
    {
      "epoch": 0.5698572431546922,
      "grad_norm": 0.2901388704776764,
      "learning_rate": 9.973495197299931e-05,
      "loss": 0.098,
      "step": 4870
    },
    {
      "epoch": 0.5698572431546922,
      "loss_main": 0.09595994651317596,
      "step": 4870
    },
    {
      "epoch": 0.5710273812309853,
      "grad_norm": 0.4672869145870209,
      "learning_rate": 9.973211067256287e-05,
      "loss": 0.0898,
      "step": 4880
    },
    {
      "epoch": 0.5710273812309853,
      "loss_main": 0.08094171434640884,
      "step": 4880
    },
    {
      "epoch": 0.5721975193072782,
      "grad_norm": 0.311259388923645,
      "learning_rate": 9.97292542648243e-05,
      "loss": 0.089,
      "step": 4890
    },
    {
      "epoch": 0.5721975193072782,
      "loss_main": 0.059482641518116,
      "step": 4890
    },
    {
      "epoch": 0.5733676573835713,
      "grad_norm": 0.423209547996521,
      "learning_rate": 9.972638275065131e-05,
      "loss": 0.0867,
      "step": 4900
    },
    {
      "epoch": 0.5733676573835713,
      "loss_main": 0.09256965667009354,
      "step": 4900
    },
    {
      "epoch": 0.5745377954598643,
      "grad_norm": 0.4512397050857544,
      "learning_rate": 9.972349613091621e-05,
      "loss": 0.1007,
      "step": 4910
    },
    {
      "epoch": 0.5745377954598643,
      "loss_main": 0.12422966957092285,
      "step": 4910
    },
    {
      "epoch": 0.5757079335361572,
      "grad_norm": 0.3501798212528229,
      "learning_rate": 9.972059440649584e-05,
      "loss": 0.0978,
      "step": 4920
    },
    {
      "epoch": 0.5757079335361572,
      "loss_main": 0.07596435397863388,
      "step": 4920
    },
    {
      "epoch": 0.5768780716124503,
      "grad_norm": 0.32662445306777954,
      "learning_rate": 9.971767757827168e-05,
      "loss": 0.0879,
      "step": 4930
    },
    {
      "epoch": 0.5768780716124503,
      "loss_main": 0.10848076641559601,
      "step": 4930
    },
    {
      "epoch": 0.5780482096887433,
      "grad_norm": 0.4395288825035095,
      "learning_rate": 9.971474564712982e-05,
      "loss": 0.0904,
      "step": 4940
    },
    {
      "epoch": 0.5780482096887433,
      "loss_main": 0.05902222916483879,
      "step": 4940
    },
    {
      "epoch": 0.5792183477650362,
      "grad_norm": 0.4009506106376648,
      "learning_rate": 9.971179861396084e-05,
      "loss": 0.0826,
      "step": 4950
    },
    {
      "epoch": 0.5792183477650362,
      "loss_main": 0.09380058199167252,
      "step": 4950
    },
    {
      "epoch": 0.5803884858413293,
      "grad_norm": 0.3671059310436249,
      "learning_rate": 9.970883647966003e-05,
      "loss": 0.0874,
      "step": 4960
    },
    {
      "epoch": 0.5803884858413293,
      "loss_main": 0.1273370236158371,
      "step": 4960
    },
    {
      "epoch": 0.5815586239176223,
      "grad_norm": 0.35962021350860596,
      "learning_rate": 9.970585924512717e-05,
      "loss": 0.0878,
      "step": 4970
    },
    {
      "epoch": 0.5815586239176223,
      "loss_main": 0.06468655169010162,
      "step": 4970
    },
    {
      "epoch": 0.5827287619939153,
      "grad_norm": 0.3982914984226227,
      "learning_rate": 9.970286691126669e-05,
      "loss": 0.083,
      "step": 4980
    },
    {
      "epoch": 0.5827287619939153,
      "loss_main": 0.07408972829580307,
      "step": 4980
    },
    {
      "epoch": 0.5838989000702083,
      "grad_norm": 0.303981751203537,
      "learning_rate": 9.969985947898756e-05,
      "loss": 0.092,
      "step": 4990
    },
    {
      "epoch": 0.5838989000702083,
      "loss_main": 0.12198984622955322,
      "step": 4990
    },
    {
      "epoch": 0.5850690381465012,
      "grad_norm": 0.5101966261863708,
      "learning_rate": 9.969683694920337e-05,
      "loss": 0.1041,
      "step": 5000
    },
    {
      "epoch": 0.5850690381465012,
      "loss_main": 0.09475849568843842,
      "step": 5000
    },
    {
      "epoch": 0.5862391762227943,
      "grad_norm": 0.24164170026779175,
      "learning_rate": 9.969379932283228e-05,
      "loss": 0.0927,
      "step": 5010
    },
    {
      "epoch": 0.5862391762227943,
      "loss_main": 0.11732540279626846,
      "step": 5010
    },
    {
      "epoch": 0.5874093142990873,
      "grad_norm": 0.4613626003265381,
      "learning_rate": 9.969074660079704e-05,
      "loss": 0.0923,
      "step": 5020
    },
    {
      "epoch": 0.5874093142990873,
      "loss_main": 0.1239556074142456,
      "step": 5020
    },
    {
      "epoch": 0.5885794523753803,
      "grad_norm": 0.4329444169998169,
      "learning_rate": 9.968767878402501e-05,
      "loss": 0.0996,
      "step": 5030
    },
    {
      "epoch": 0.5885794523753803,
      "loss_main": 0.056440453976392746,
      "step": 5030
    },
    {
      "epoch": 0.5897495904516733,
      "grad_norm": 0.32458069920539856,
      "learning_rate": 9.968459587344808e-05,
      "loss": 0.0797,
      "step": 5040
    },
    {
      "epoch": 0.5897495904516733,
      "loss_main": 0.09997139126062393,
      "step": 5040
    },
    {
      "epoch": 0.5909197285279663,
      "grad_norm": 0.35954004526138306,
      "learning_rate": 9.968149787000278e-05,
      "loss": 0.0973,
      "step": 5050
    },
    {
      "epoch": 0.5909197285279663,
      "loss_main": 0.0667605847120285,
      "step": 5050
    },
    {
      "epoch": 0.5920898666042593,
      "grad_norm": 0.3652367889881134,
      "learning_rate": 9.967838477463018e-05,
      "loss": 0.0828,
      "step": 5060
    },
    {
      "epoch": 0.5920898666042593,
      "loss_main": 0.059936072677373886,
      "step": 5060
    },
    {
      "epoch": 0.5932600046805523,
      "grad_norm": 0.3399522304534912,
      "learning_rate": 9.967525658827597e-05,
      "loss": 0.0949,
      "step": 5070
    },
    {
      "epoch": 0.5932600046805523,
      "loss_main": 0.07504303008317947,
      "step": 5070
    },
    {
      "epoch": 0.5944301427568454,
      "grad_norm": 0.3825540542602539,
      "learning_rate": 9.967211331189042e-05,
      "loss": 0.0942,
      "step": 5080
    },
    {
      "epoch": 0.5944301427568454,
      "loss_main": 0.07795340567827225,
      "step": 5080
    },
    {
      "epoch": 0.5956002808331383,
      "grad_norm": 0.449230432510376,
      "learning_rate": 9.966895494642834e-05,
      "loss": 0.1008,
      "step": 5090
    },
    {
      "epoch": 0.5956002808331383,
      "loss_main": 0.07934226840734482,
      "step": 5090
    },
    {
      "epoch": 0.5967704189094313,
      "grad_norm": 0.4994509816169739,
      "learning_rate": 9.96657814928492e-05,
      "loss": 0.1025,
      "step": 5100
    },
    {
      "epoch": 0.5967704189094313,
      "loss_main": 0.09850890189409256,
      "step": 5100
    },
    {
      "epoch": 0.5979405569857243,
      "grad_norm": 0.4382489323616028,
      "learning_rate": 9.966259295211697e-05,
      "loss": 0.0935,
      "step": 5110
    },
    {
      "epoch": 0.5979405569857243,
      "loss_main": 0.15304671227931976,
      "step": 5110
    },
    {
      "epoch": 0.5991106950620173,
      "grad_norm": 0.573394238948822,
      "learning_rate": 9.965938932520028e-05,
      "loss": 0.096,
      "step": 5120
    },
    {
      "epoch": 0.5991106950620173,
      "loss_main": 0.11654885858297348,
      "step": 5120
    },
    {
      "epoch": 0.6002808331383103,
      "grad_norm": 0.29541364312171936,
      "learning_rate": 9.965617061307229e-05,
      "loss": 0.0943,
      "step": 5130
    },
    {
      "epoch": 0.6002808331383103,
      "loss_main": 0.07481139153242111,
      "step": 5130
    },
    {
      "epoch": 0.6014509712146033,
      "grad_norm": 0.31534433364868164,
      "learning_rate": 9.965293681671077e-05,
      "loss": 0.0819,
      "step": 5140
    },
    {
      "epoch": 0.6014509712146033,
      "loss_main": 0.16244010627269745,
      "step": 5140
    },
    {
      "epoch": 0.6026211092908963,
      "grad_norm": 0.3791547417640686,
      "learning_rate": 9.964968793709804e-05,
      "loss": 0.0935,
      "step": 5150
    },
    {
      "epoch": 0.6026211092908963,
      "loss_main": 0.09048562496900558,
      "step": 5150
    },
    {
      "epoch": 0.6037912473671894,
      "grad_norm": 0.40879830718040466,
      "learning_rate": 9.964642397522106e-05,
      "loss": 0.0816,
      "step": 5160
    },
    {
      "epoch": 0.6037912473671894,
      "loss_main": 0.10005742311477661,
      "step": 5160
    },
    {
      "epoch": 0.6049613854434823,
      "grad_norm": 0.5114991068840027,
      "learning_rate": 9.96431449320713e-05,
      "loss": 0.1018,
      "step": 5170
    },
    {
      "epoch": 0.6049613854434823,
      "loss_main": 0.09182985872030258,
      "step": 5170
    },
    {
      "epoch": 0.6061315235197753,
      "grad_norm": 0.34335094690322876,
      "learning_rate": 9.963985080864486e-05,
      "loss": 0.0951,
      "step": 5180
    },
    {
      "epoch": 0.6061315235197753,
      "loss_main": 0.09596016258001328,
      "step": 5180
    },
    {
      "epoch": 0.6073016615960684,
      "grad_norm": 0.3612913489341736,
      "learning_rate": 9.96365416059424e-05,
      "loss": 0.0845,
      "step": 5190
    },
    {
      "epoch": 0.6073016615960684,
      "loss_main": 0.1073107048869133,
      "step": 5190
    },
    {
      "epoch": 0.6084717996723613,
      "grad_norm": 0.3102925717830658,
      "learning_rate": 9.963321732496919e-05,
      "loss": 0.0928,
      "step": 5200
    },
    {
      "epoch": 0.6084717996723613,
      "loss_main": 0.07155179977416992,
      "step": 5200
    },
    {
      "epoch": 0.6096419377486544,
      "grad_norm": 0.329941064119339,
      "learning_rate": 9.962987796673506e-05,
      "loss": 0.0907,
      "step": 5210
    },
    {
      "epoch": 0.6096419377486544,
      "loss_main": 0.09188008308410645,
      "step": 5210
    },
    {
      "epoch": 0.6108120758249473,
      "grad_norm": 0.4794190526008606,
      "learning_rate": 9.962652353225438e-05,
      "loss": 0.0808,
      "step": 5220
    },
    {
      "epoch": 0.6108120758249473,
      "loss_main": 0.07889580726623535,
      "step": 5220
    },
    {
      "epoch": 0.6119822139012403,
      "grad_norm": 0.4961582124233246,
      "learning_rate": 9.962315402254619e-05,
      "loss": 0.0991,
      "step": 5230
    },
    {
      "epoch": 0.6119822139012403,
      "loss_main": 0.09665616601705551,
      "step": 5230
    },
    {
      "epoch": 0.6131523519775334,
      "grad_norm": 0.2964198589324951,
      "learning_rate": 9.9619769438634e-05,
      "loss": 0.0898,
      "step": 5240
    },
    {
      "epoch": 0.6131523519775334,
      "loss_main": 0.13118334114551544,
      "step": 5240
    },
    {
      "epoch": 0.6143224900538263,
      "grad_norm": 0.2849046289920807,
      "learning_rate": 9.9616369781546e-05,
      "loss": 0.0856,
      "step": 5250
    },
    {
      "epoch": 0.6143224900538263,
      "loss_main": 0.04908277466893196,
      "step": 5250
    },
    {
      "epoch": 0.6154926281301194,
      "grad_norm": 0.6137152314186096,
      "learning_rate": 9.961295505231491e-05,
      "loss": 0.0844,
      "step": 5260
    },
    {
      "epoch": 0.6154926281301194,
      "loss_main": 0.07957936823368073,
      "step": 5260
    },
    {
      "epoch": 0.6166627662064124,
      "grad_norm": 0.36211249232292175,
      "learning_rate": 9.960952525197804e-05,
      "loss": 0.0854,
      "step": 5270
    },
    {
      "epoch": 0.6166627662064124,
      "loss_main": 0.08667117357254028,
      "step": 5270
    },
    {
      "epoch": 0.6178329042827053,
      "grad_norm": 0.4228176474571228,
      "learning_rate": 9.960608038157724e-05,
      "loss": 0.0814,
      "step": 5280
    },
    {
      "epoch": 0.6178329042827053,
      "loss_main": 0.06619063019752502,
      "step": 5280
    },
    {
      "epoch": 0.6190030423589984,
      "grad_norm": 0.3562001883983612,
      "learning_rate": 9.960262044215901e-05,
      "loss": 0.0898,
      "step": 5290
    },
    {
      "epoch": 0.6190030423589984,
      "loss_main": 0.10868249088525772,
      "step": 5290
    },
    {
      "epoch": 0.6201731804352913,
      "grad_norm": 0.5002946853637695,
      "learning_rate": 9.959914543477435e-05,
      "loss": 0.0988,
      "step": 5300
    },
    {
      "epoch": 0.6201731804352913,
      "loss_main": 0.16579791903495789,
      "step": 5300
    },
    {
      "epoch": 0.6213433185115844,
      "grad_norm": 0.3485853970050812,
      "learning_rate": 9.959565536047892e-05,
      "loss": 0.1093,
      "step": 5310
    },
    {
      "epoch": 0.6213433185115844,
      "loss_main": 0.06556633114814758,
      "step": 5310
    },
    {
      "epoch": 0.6225134565878774,
      "grad_norm": 0.38559284806251526,
      "learning_rate": 9.959215022033288e-05,
      "loss": 0.0883,
      "step": 5320
    },
    {
      "epoch": 0.6225134565878774,
      "loss_main": 0.10036476701498032,
      "step": 5320
    },
    {
      "epoch": 0.6236835946641703,
      "grad_norm": 0.35497838258743286,
      "learning_rate": 9.9588630015401e-05,
      "loss": 0.102,
      "step": 5330
    },
    {
      "epoch": 0.6236835946641703,
      "loss_main": 0.09353574365377426,
      "step": 5330
    },
    {
      "epoch": 0.6248537327404634,
      "grad_norm": 0.40580058097839355,
      "learning_rate": 9.958509474675264e-05,
      "loss": 0.0966,
      "step": 5340
    },
    {
      "epoch": 0.6248537327404634,
      "loss_main": 0.09448961168527603,
      "step": 5340
    },
    {
      "epoch": 0.6260238708167564,
      "grad_norm": 0.2848566472530365,
      "learning_rate": 9.958154441546171e-05,
      "loss": 0.101,
      "step": 5350
    },
    {
      "epoch": 0.6260238708167564,
      "loss_main": 0.09758175909519196,
      "step": 5350
    },
    {
      "epoch": 0.6271940088930493,
      "grad_norm": 0.40449628233909607,
      "learning_rate": 9.957797902260673e-05,
      "loss": 0.0942,
      "step": 5360
    },
    {
      "epoch": 0.6271940088930493,
      "loss_main": 0.09774546325206757,
      "step": 5360
    },
    {
      "epoch": 0.6283641469693424,
      "grad_norm": 0.4624863564968109,
      "learning_rate": 9.957439856927073e-05,
      "loss": 0.086,
      "step": 5370
    },
    {
      "epoch": 0.6283641469693424,
      "loss_main": 0.06905891001224518,
      "step": 5370
    },
    {
      "epoch": 0.6295342850456354,
      "grad_norm": 0.2743745744228363,
      "learning_rate": 9.957080305654139e-05,
      "loss": 0.0931,
      "step": 5380
    },
    {
      "epoch": 0.6295342850456354,
      "loss_main": 0.10324390977621078,
      "step": 5380
    },
    {
      "epoch": 0.6307044231219284,
      "grad_norm": 0.6459489464759827,
      "learning_rate": 9.956719248551092e-05,
      "loss": 0.1074,
      "step": 5390
    },
    {
      "epoch": 0.6307044231219284,
      "loss_main": 0.08710391074419022,
      "step": 5390
    },
    {
      "epoch": 0.6318745611982214,
      "grad_norm": 0.4712410569190979,
      "learning_rate": 9.956356685727612e-05,
      "loss": 0.0949,
      "step": 5400
    },
    {
      "epoch": 0.6318745611982214,
      "loss_main": 0.11446253955364227,
      "step": 5400
    },
    {
      "epoch": 0.6330446992745143,
      "grad_norm": 0.43934085965156555,
      "learning_rate": 9.955992617293836e-05,
      "loss": 0.0954,
      "step": 5410
    },
    {
      "epoch": 0.6330446992745143,
      "loss_main": 0.07727725058794022,
      "step": 5410
    },
    {
      "epoch": 0.6342148373508074,
      "grad_norm": 0.5137950778007507,
      "learning_rate": 9.955627043360358e-05,
      "loss": 0.0995,
      "step": 5420
    },
    {
      "epoch": 0.6342148373508074,
      "loss_main": 0.09531164169311523,
      "step": 5420
    },
    {
      "epoch": 0.6353849754271004,
      "grad_norm": 0.3681325912475586,
      "learning_rate": 9.955259964038231e-05,
      "loss": 0.0856,
      "step": 5430
    },
    {
      "epoch": 0.6353849754271004,
      "loss_main": 0.07637076824903488,
      "step": 5430
    },
    {
      "epoch": 0.6365551135033934,
      "grad_norm": 0.3903835415840149,
      "learning_rate": 9.954891379438962e-05,
      "loss": 0.0947,
      "step": 5440
    },
    {
      "epoch": 0.6365551135033934,
      "loss_main": 0.09568679332733154,
      "step": 5440
    },
    {
      "epoch": 0.6377252515796864,
      "grad_norm": 0.49180373549461365,
      "learning_rate": 9.954521289674519e-05,
      "loss": 0.1008,
      "step": 5450
    },
    {
      "epoch": 0.6377252515796864,
      "loss_main": 0.0543401837348938,
      "step": 5450
    },
    {
      "epoch": 0.6388953896559794,
      "grad_norm": 0.39589568972587585,
      "learning_rate": 9.954149694857325e-05,
      "loss": 0.0902,
      "step": 5460
    },
    {
      "epoch": 0.6388953896559794,
      "loss_main": 0.06253540515899658,
      "step": 5460
    },
    {
      "epoch": 0.6400655277322724,
      "grad_norm": 0.44110777974128723,
      "learning_rate": 9.953776595100258e-05,
      "loss": 0.0996,
      "step": 5470
    },
    {
      "epoch": 0.6400655277322724,
      "loss_main": 0.07258490473031998,
      "step": 5470
    },
    {
      "epoch": 0.6412356658085654,
      "grad_norm": 0.4237578511238098,
      "learning_rate": 9.95340199051666e-05,
      "loss": 0.0865,
      "step": 5480
    },
    {
      "epoch": 0.6412356658085654,
      "loss_main": 0.051706649363040924,
      "step": 5480
    },
    {
      "epoch": 0.6424058038848585,
      "grad_norm": 0.4788060486316681,
      "learning_rate": 9.953025881220325e-05,
      "loss": 0.0839,
      "step": 5490
    },
    {
      "epoch": 0.6424058038848585,
      "loss_main": 0.08924007415771484,
      "step": 5490
    },
    {
      "epoch": 0.6435759419611514,
      "grad_norm": 0.3527761697769165,
      "learning_rate": 9.952648267325504e-05,
      "loss": 0.0837,
      "step": 5500
    },
    {
      "epoch": 0.6435759419611514,
      "loss_main": 0.0615590400993824,
      "step": 5500
    },
    {
      "epoch": 0.6447460800374444,
      "grad_norm": 0.298713743686676,
      "learning_rate": 9.952269148946905e-05,
      "loss": 0.0906,
      "step": 5510
    },
    {
      "epoch": 0.6447460800374444,
      "loss_main": 0.09354452043771744,
      "step": 5510
    },
    {
      "epoch": 0.6459162181137375,
      "grad_norm": 0.3985483646392822,
      "learning_rate": 9.951888526199697e-05,
      "loss": 0.0899,
      "step": 5520
    },
    {
      "epoch": 0.6459162181137375,
      "loss_main": 0.07567380368709564,
      "step": 5520
    },
    {
      "epoch": 0.6470863561900304,
      "grad_norm": 0.39340686798095703,
      "learning_rate": 9.951506399199501e-05,
      "loss": 0.0856,
      "step": 5530
    },
    {
      "epoch": 0.6470863561900304,
      "loss_main": 0.06186787411570549,
      "step": 5530
    },
    {
      "epoch": 0.6482564942663235,
      "grad_norm": 0.3552984893321991,
      "learning_rate": 9.951122768062399e-05,
      "loss": 0.0841,
      "step": 5540
    },
    {
      "epoch": 0.6482564942663235,
      "loss_main": 0.07361992448568344,
      "step": 5540
    },
    {
      "epoch": 0.6494266323426164,
      "grad_norm": 0.2768738567829132,
      "learning_rate": 9.950737632904927e-05,
      "loss": 0.0875,
      "step": 5550
    },
    {
      "epoch": 0.6494266323426164,
      "loss_main": 0.06520665436983109,
      "step": 5550
    },
    {
      "epoch": 0.6505967704189094,
      "grad_norm": 0.32619091868400574,
      "learning_rate": 9.950350993844077e-05,
      "loss": 0.0861,
      "step": 5560
    },
    {
      "epoch": 0.6505967704189094,
      "loss_main": 0.09078160673379898,
      "step": 5560
    },
    {
      "epoch": 0.6517669084952025,
      "grad_norm": 0.2818481922149658,
      "learning_rate": 9.949962850997303e-05,
      "loss": 0.087,
      "step": 5570
    },
    {
      "epoch": 0.6517669084952025,
      "loss_main": 0.05881946533918381,
      "step": 5570
    },
    {
      "epoch": 0.6529370465714954,
      "grad_norm": 0.3662341237068176,
      "learning_rate": 9.949573204482512e-05,
      "loss": 0.0733,
      "step": 5580
    },
    {
      "epoch": 0.6529370465714954,
      "loss_main": 0.09710519760847092,
      "step": 5580
    },
    {
      "epoch": 0.6541071846477884,
      "grad_norm": 0.4137779176235199,
      "learning_rate": 9.949182054418064e-05,
      "loss": 0.0906,
      "step": 5590
    },
    {
      "epoch": 0.6541071846477884,
      "loss_main": 0.17170776426792145,
      "step": 5590
    },
    {
      "epoch": 0.6552773227240815,
      "grad_norm": 0.359373539686203,
      "learning_rate": 9.948789400922787e-05,
      "loss": 0.0897,
      "step": 5600
    },
    {
      "epoch": 0.6552773227240815,
      "loss_main": 0.0811990350484848,
      "step": 5600
    },
    {
      "epoch": 0.6564474608003744,
      "grad_norm": 0.387029767036438,
      "learning_rate": 9.948395244115953e-05,
      "loss": 0.0893,
      "step": 5610
    },
    {
      "epoch": 0.6564474608003744,
      "loss_main": 0.12577752768993378,
      "step": 5610
    },
    {
      "epoch": 0.6576175988766675,
      "grad_norm": 0.4330533444881439,
      "learning_rate": 9.9479995841173e-05,
      "loss": 0.0906,
      "step": 5620
    },
    {
      "epoch": 0.6576175988766675,
      "loss_main": 0.06010807305574417,
      "step": 5620
    },
    {
      "epoch": 0.6587877369529604,
      "grad_norm": 0.4230320453643799,
      "learning_rate": 9.947602421047017e-05,
      "loss": 0.079,
      "step": 5630
    },
    {
      "epoch": 0.6587877369529604,
      "loss_main": 0.11390523612499237,
      "step": 5630
    },
    {
      "epoch": 0.6599578750292534,
      "grad_norm": 0.31902143359184265,
      "learning_rate": 9.947203755025753e-05,
      "loss": 0.0971,
      "step": 5640
    },
    {
      "epoch": 0.6599578750292534,
      "loss_main": 0.08600062131881714,
      "step": 5640
    },
    {
      "epoch": 0.6611280131055465,
      "grad_norm": 0.25721046328544617,
      "learning_rate": 9.946803586174611e-05,
      "loss": 0.0946,
      "step": 5650
    },
    {
      "epoch": 0.6611280131055465,
      "loss_main": 0.08060359954833984,
      "step": 5650
    },
    {
      "epoch": 0.6622981511818394,
      "grad_norm": 0.37414175271987915,
      "learning_rate": 9.946401914615151e-05,
      "loss": 0.092,
      "step": 5660
    },
    {
      "epoch": 0.6622981511818394,
      "loss_main": 0.07332120090723038,
      "step": 5660
    },
    {
      "epoch": 0.6634682892581325,
      "grad_norm": 0.38914790749549866,
      "learning_rate": 9.945998740469394e-05,
      "loss": 0.0829,
      "step": 5670
    },
    {
      "epoch": 0.6634682892581325,
      "loss_main": 0.1825312376022339,
      "step": 5670
    },
    {
      "epoch": 0.6646384273344255,
      "grad_norm": 0.34114280343055725,
      "learning_rate": 9.945594063859809e-05,
      "loss": 0.0951,
      "step": 5680
    },
    {
      "epoch": 0.6646384273344255,
      "loss_main": 0.08267152309417725,
      "step": 5680
    },
    {
      "epoch": 0.6658085654107184,
      "grad_norm": 0.4444347620010376,
      "learning_rate": 9.94518788490933e-05,
      "loss": 0.0761,
      "step": 5690
    },
    {
      "epoch": 0.6658085654107184,
      "loss_main": 0.1064717248082161,
      "step": 5690
    },
    {
      "epoch": 0.6669787034870115,
      "grad_norm": 0.3767355978488922,
      "learning_rate": 9.944780203741341e-05,
      "loss": 0.0916,
      "step": 5700
    },
    {
      "epoch": 0.6669787034870115,
      "loss_main": 0.07343504577875137,
      "step": 5700
    },
    {
      "epoch": 0.6681488415633045,
      "grad_norm": 0.38120177388191223,
      "learning_rate": 9.944371020479686e-05,
      "loss": 0.0852,
      "step": 5710
    },
    {
      "epoch": 0.6681488415633045,
      "loss_main": 0.05582055449485779,
      "step": 5710
    },
    {
      "epoch": 0.6693189796395975,
      "grad_norm": 0.258681058883667,
      "learning_rate": 9.943960335248662e-05,
      "loss": 0.078,
      "step": 5720
    },
    {
      "epoch": 0.6693189796395975,
      "loss_main": 0.04279813542962074,
      "step": 5720
    },
    {
      "epoch": 0.6704891177158905,
      "grad_norm": 0.4039822220802307,
      "learning_rate": 9.943548148173027e-05,
      "loss": 0.0908,
      "step": 5730
    },
    {
      "epoch": 0.6704891177158905,
      "loss_main": 0.05440540984272957,
      "step": 5730
    },
    {
      "epoch": 0.6716592557921834,
      "grad_norm": 0.529464602470398,
      "learning_rate": 9.943134459377992e-05,
      "loss": 0.0857,
      "step": 5740
    },
    {
      "epoch": 0.6716592557921834,
      "loss_main": 0.07892145216464996,
      "step": 5740
    },
    {
      "epoch": 0.6728293938684765,
      "grad_norm": 0.4623083770275116,
      "learning_rate": 9.942719268989222e-05,
      "loss": 0.0915,
      "step": 5750
    },
    {
      "epoch": 0.6728293938684765,
      "loss_main": 0.07927487790584564,
      "step": 5750
    },
    {
      "epoch": 0.6739995319447695,
      "grad_norm": 0.2733519375324249,
      "learning_rate": 9.942302577132844e-05,
      "loss": 0.086,
      "step": 5760
    },
    {
      "epoch": 0.6739995319447695,
      "loss_main": 0.1610502004623413,
      "step": 5760
    },
    {
      "epoch": 0.6751696700210625,
      "grad_norm": 0.3844568729400635,
      "learning_rate": 9.941884383935438e-05,
      "loss": 0.0925,
      "step": 5770
    },
    {
      "epoch": 0.6751696700210625,
      "loss_main": 0.11663450300693512,
      "step": 5770
    },
    {
      "epoch": 0.6763398080973555,
      "grad_norm": 0.3517434895038605,
      "learning_rate": 9.941464689524039e-05,
      "loss": 0.0922,
      "step": 5780
    },
    {
      "epoch": 0.6763398080973555,
      "loss_main": 0.0765460655093193,
      "step": 5780
    },
    {
      "epoch": 0.6775099461736485,
      "grad_norm": 0.375948429107666,
      "learning_rate": 9.941043494026139e-05,
      "loss": 0.0928,
      "step": 5790
    },
    {
      "epoch": 0.6775099461736485,
      "loss_main": 0.07783061265945435,
      "step": 5790
    },
    {
      "epoch": 0.6786800842499415,
      "grad_norm": 0.4217798709869385,
      "learning_rate": 9.940620797569685e-05,
      "loss": 0.089,
      "step": 5800
    },
    {
      "epoch": 0.6786800842499415,
      "loss_main": 0.11292129009962082,
      "step": 5800
    },
    {
      "epoch": 0.6798502223262345,
      "grad_norm": 0.4437747597694397,
      "learning_rate": 9.940196600283082e-05,
      "loss": 0.1142,
      "step": 5810
    },
    {
      "epoch": 0.6798502223262345,
      "loss_main": 0.10601228475570679,
      "step": 5810
    },
    {
      "epoch": 0.6810203604025274,
      "grad_norm": 0.24045659601688385,
      "learning_rate": 9.939770902295192e-05,
      "loss": 0.0932,
      "step": 5820
    },
    {
      "epoch": 0.6810203604025274,
      "loss_main": 0.0718572810292244,
      "step": 5820
    },
    {
      "epoch": 0.6821904984788205,
      "grad_norm": 0.32705098390579224,
      "learning_rate": 9.939343703735329e-05,
      "loss": 0.0984,
      "step": 5830
    },
    {
      "epoch": 0.6821904984788205,
      "loss_main": 0.0724664106965065,
      "step": 5830
    },
    {
      "epoch": 0.6833606365551135,
      "grad_norm": 0.39804837107658386,
      "learning_rate": 9.938915004733264e-05,
      "loss": 0.0881,
      "step": 5840
    },
    {
      "epoch": 0.6833606365551135,
      "loss_main": 0.0989592894911766,
      "step": 5840
    },
    {
      "epoch": 0.6845307746314065,
      "grad_norm": 0.3406358063220978,
      "learning_rate": 9.938484805419224e-05,
      "loss": 0.0921,
      "step": 5850
    },
    {
      "epoch": 0.6845307746314065,
      "loss_main": 0.08196114003658295,
      "step": 5850
    },
    {
      "epoch": 0.6857009127076995,
      "grad_norm": 0.41127514839172363,
      "learning_rate": 9.938053105923894e-05,
      "loss": 0.0924,
      "step": 5860
    },
    {
      "epoch": 0.6857009127076995,
      "loss_main": 0.08086608350276947,
      "step": 5860
    },
    {
      "epoch": 0.6868710507839925,
      "grad_norm": 0.3613940477371216,
      "learning_rate": 9.937619906378413e-05,
      "loss": 0.0784,
      "step": 5870
    },
    {
      "epoch": 0.6868710507839925,
      "loss_main": 0.08262719959020615,
      "step": 5870
    },
    {
      "epoch": 0.6880411888602855,
      "grad_norm": 0.3178706765174866,
      "learning_rate": 9.937185206914374e-05,
      "loss": 0.0937,
      "step": 5880
    },
    {
      "epoch": 0.6880411888602855,
      "loss_main": 0.08289293944835663,
      "step": 5880
    },
    {
      "epoch": 0.6892113269365785,
      "grad_norm": 0.3384135365486145,
      "learning_rate": 9.936749007663829e-05,
      "loss": 0.0893,
      "step": 5890
    },
    {
      "epoch": 0.6892113269365785,
      "loss_main": 0.09336503595113754,
      "step": 5890
    },
    {
      "epoch": 0.6903814650128716,
      "grad_norm": 0.3603559732437134,
      "learning_rate": 9.93631130875928e-05,
      "loss": 0.0912,
      "step": 5900
    },
    {
      "epoch": 0.6903814650128716,
      "loss_main": 0.10783442109823227,
      "step": 5900
    },
    {
      "epoch": 0.6915516030891645,
      "grad_norm": 0.28832143545150757,
      "learning_rate": 9.935872110333692e-05,
      "loss": 0.0989,
      "step": 5910
    },
    {
      "epoch": 0.6915516030891645,
      "loss_main": 0.08231884241104126,
      "step": 5910
    },
    {
      "epoch": 0.6927217411654575,
      "grad_norm": 0.30843573808670044,
      "learning_rate": 9.935431412520484e-05,
      "loss": 0.0911,
      "step": 5920
    },
    {
      "epoch": 0.6927217411654575,
      "loss_main": 0.06677283346652985,
      "step": 5920
    },
    {
      "epoch": 0.6938918792417506,
      "grad_norm": 0.34778842329978943,
      "learning_rate": 9.934989215453523e-05,
      "loss": 0.0877,
      "step": 5930
    },
    {
      "epoch": 0.6938918792417506,
      "loss_main": 0.07732988893985748,
      "step": 5930
    },
    {
      "epoch": 0.6950620173180435,
      "grad_norm": 0.42767444252967834,
      "learning_rate": 9.934545519267139e-05,
      "loss": 0.0929,
      "step": 5940
    },
    {
      "epoch": 0.6950620173180435,
      "loss_main": 0.08367674797773361,
      "step": 5940
    },
    {
      "epoch": 0.6962321553943366,
      "grad_norm": 0.285494327545166,
      "learning_rate": 9.934100324096117e-05,
      "loss": 0.0933,
      "step": 5950
    },
    {
      "epoch": 0.6962321553943366,
      "loss_main": 0.10658375173807144,
      "step": 5950
    },
    {
      "epoch": 0.6974022934706295,
      "grad_norm": 0.3479101061820984,
      "learning_rate": 9.933653630075692e-05,
      "loss": 0.0858,
      "step": 5960
    },
    {
      "epoch": 0.6974022934706295,
      "loss_main": 0.09228992462158203,
      "step": 5960
    },
    {
      "epoch": 0.6985724315469225,
      "grad_norm": 0.5783724784851074,
      "learning_rate": 9.93320543734156e-05,
      "loss": 0.0888,
      "step": 5970
    },
    {
      "epoch": 0.6985724315469225,
      "loss_main": 0.07936514168977737,
      "step": 5970
    },
    {
      "epoch": 0.6997425696232156,
      "grad_norm": 0.36595502495765686,
      "learning_rate": 9.932755746029871e-05,
      "loss": 0.0906,
      "step": 5980
    },
    {
      "epoch": 0.6997425696232156,
      "loss_main": 0.07353891432285309,
      "step": 5980
    },
    {
      "epoch": 0.7009127076995085,
      "grad_norm": 0.45816782116889954,
      "learning_rate": 9.932304556277228e-05,
      "loss": 0.0907,
      "step": 5990
    },
    {
      "epoch": 0.7009127076995085,
      "loss_main": 0.07714599370956421,
      "step": 5990
    },
    {
      "epoch": 0.7020828457758015,
      "grad_norm": 0.4202847480773926,
      "learning_rate": 9.93185186822069e-05,
      "loss": 0.0916,
      "step": 6000
    },
    {
      "epoch": 0.7020828457758015,
      "loss_main": 0.08376368135213852,
      "step": 6000
    },
    {
      "epoch": 0.7032529838520946,
      "grad_norm": 0.38510793447494507,
      "learning_rate": 9.931397681997773e-05,
      "loss": 0.0945,
      "step": 6010
    },
    {
      "epoch": 0.7032529838520946,
      "loss_main": 0.08249149471521378,
      "step": 6010
    },
    {
      "epoch": 0.7044231219283875,
      "grad_norm": 0.36409181356430054,
      "learning_rate": 9.930941997746446e-05,
      "loss": 0.0872,
      "step": 6020
    },
    {
      "epoch": 0.7044231219283875,
      "loss_main": 0.09909898042678833,
      "step": 6020
    },
    {
      "epoch": 0.7055932600046806,
      "grad_norm": 0.3120039999485016,
      "learning_rate": 9.930484815605134e-05,
      "loss": 0.0916,
      "step": 6030
    },
    {
      "epoch": 0.7055932600046806,
      "loss_main": 0.06644438952207565,
      "step": 6030
    },
    {
      "epoch": 0.7067633980809735,
      "grad_norm": 0.4405480921268463,
      "learning_rate": 9.930026135712717e-05,
      "loss": 0.1032,
      "step": 6040
    },
    {
      "epoch": 0.7067633980809735,
      "loss_main": 0.09863000363111496,
      "step": 6040
    },
    {
      "epoch": 0.7079335361572665,
      "grad_norm": 0.46898356080055237,
      "learning_rate": 9.92956595820853e-05,
      "loss": 0.0893,
      "step": 6050
    },
    {
      "epoch": 0.7079335361572665,
      "loss_main": 0.06088462099432945,
      "step": 6050
    },
    {
      "epoch": 0.7091036742335596,
      "grad_norm": 0.3434130549430847,
      "learning_rate": 9.929104283232362e-05,
      "loss": 0.0819,
      "step": 6060
    },
    {
      "epoch": 0.7091036742335596,
      "loss_main": 0.09165410697460175,
      "step": 6060
    },
    {
      "epoch": 0.7102738123098525,
      "grad_norm": 0.3222123980522156,
      "learning_rate": 9.92864111092446e-05,
      "loss": 0.0818,
      "step": 6070
    },
    {
      "epoch": 0.7102738123098525,
      "loss_main": 0.08784184604883194,
      "step": 6070
    },
    {
      "epoch": 0.7114439503861456,
      "grad_norm": 0.3326221704483032,
      "learning_rate": 9.92817644142552e-05,
      "loss": 0.0814,
      "step": 6080
    },
    {
      "epoch": 0.7114439503861456,
      "loss_main": 0.06771733611822128,
      "step": 6080
    },
    {
      "epoch": 0.7126140884624386,
      "grad_norm": 0.3498663008213043,
      "learning_rate": 9.927710274876698e-05,
      "loss": 0.0843,
      "step": 6090
    },
    {
      "epoch": 0.7126140884624386,
      "loss_main": 0.07874304056167603,
      "step": 6090
    },
    {
      "epoch": 0.7137842265387315,
      "grad_norm": 0.3359966576099396,
      "learning_rate": 9.927242611419603e-05,
      "loss": 0.0924,
      "step": 6100
    },
    {
      "epoch": 0.7137842265387315,
      "loss_main": 0.10613392293453217,
      "step": 6100
    },
    {
      "epoch": 0.7149543646150246,
      "grad_norm": 0.39612141251564026,
      "learning_rate": 9.926773451196301e-05,
      "loss": 0.0959,
      "step": 6110
    },
    {
      "epoch": 0.7149543646150246,
      "loss_main": 0.057467009872198105,
      "step": 6110
    },
    {
      "epoch": 0.7161245026913176,
      "grad_norm": 0.3003052771091461,
      "learning_rate": 9.926302794349306e-05,
      "loss": 0.0852,
      "step": 6120
    },
    {
      "epoch": 0.7161245026913176,
      "loss_main": 0.07379751652479172,
      "step": 6120
    },
    {
      "epoch": 0.7172946407676106,
      "grad_norm": 0.3545868396759033,
      "learning_rate": 9.925830641021594e-05,
      "loss": 0.0987,
      "step": 6130
    },
    {
      "epoch": 0.7172946407676106,
      "loss_main": 0.12402057647705078,
      "step": 6130
    },
    {
      "epoch": 0.7184647788439036,
      "grad_norm": 0.39563989639282227,
      "learning_rate": 9.925356991356593e-05,
      "loss": 0.0869,
      "step": 6140
    },
    {
      "epoch": 0.7184647788439036,
      "loss_main": 0.09442324936389923,
      "step": 6140
    },
    {
      "epoch": 0.7196349169201965,
      "grad_norm": 0.6724265217781067,
      "learning_rate": 9.924881845498184e-05,
      "loss": 0.0885,
      "step": 6150
    },
    {
      "epoch": 0.7196349169201965,
      "loss_main": 0.09464388340711594,
      "step": 6150
    },
    {
      "epoch": 0.7208050549964896,
      "grad_norm": 0.40351834893226624,
      "learning_rate": 9.924405203590705e-05,
      "loss": 0.0935,
      "step": 6160
    },
    {
      "epoch": 0.7208050549964896,
      "loss_main": 0.06512265652418137,
      "step": 6160
    },
    {
      "epoch": 0.7219751930727826,
      "grad_norm": 0.36923328042030334,
      "learning_rate": 9.923927065778946e-05,
      "loss": 0.0948,
      "step": 6170
    },
    {
      "epoch": 0.7219751930727826,
      "loss_main": 0.11781491339206696,
      "step": 6170
    },
    {
      "epoch": 0.7231453311490756,
      "grad_norm": 0.38895997405052185,
      "learning_rate": 9.923447432208154e-05,
      "loss": 0.0923,
      "step": 6180
    },
    {
      "epoch": 0.7231453311490756,
      "loss_main": 0.08395825326442719,
      "step": 6180
    },
    {
      "epoch": 0.7243154692253686,
      "grad_norm": 0.445708692073822,
      "learning_rate": 9.922966303024027e-05,
      "loss": 0.0859,
      "step": 6190
    },
    {
      "epoch": 0.7243154692253686,
      "loss_main": 0.08360065519809723,
      "step": 6190
    },
    {
      "epoch": 0.7254856073016616,
      "grad_norm": 0.46327903866767883,
      "learning_rate": 9.922483678372721e-05,
      "loss": 0.1007,
      "step": 6200
    },
    {
      "epoch": 0.7254856073016616,
      "loss_main": 0.10798627883195877,
      "step": 6200
    },
    {
      "epoch": 0.7266557453779546,
      "grad_norm": 0.41785895824432373,
      "learning_rate": 9.921999558400845e-05,
      "loss": 0.0972,
      "step": 6210
    },
    {
      "epoch": 0.7266557453779546,
      "loss_main": 0.10941873490810394,
      "step": 6210
    },
    {
      "epoch": 0.7278258834542476,
      "grad_norm": 0.42985740303993225,
      "learning_rate": 9.92151394325546e-05,
      "loss": 0.1075,
      "step": 6220
    },
    {
      "epoch": 0.7278258834542476,
      "loss_main": 0.11818645894527435,
      "step": 6220
    },
    {
      "epoch": 0.7289960215305405,
      "grad_norm": 0.35395652055740356,
      "learning_rate": 9.921026833084084e-05,
      "loss": 0.0883,
      "step": 6230
    },
    {
      "epoch": 0.7289960215305405,
      "loss_main": 0.1754162758588791,
      "step": 6230
    },
    {
      "epoch": 0.7301661596068336,
      "grad_norm": 0.3203878402709961,
      "learning_rate": 9.920538228034689e-05,
      "loss": 0.0862,
      "step": 6240
    },
    {
      "epoch": 0.7301661596068336,
      "loss_main": 0.10333810746669769,
      "step": 6240
    },
    {
      "epoch": 0.7313362976831266,
      "grad_norm": 0.2676195204257965,
      "learning_rate": 9.920048128255699e-05,
      "loss": 0.0845,
      "step": 6250
    },
    {
      "epoch": 0.7313362976831266,
      "loss_main": 0.07307592779397964,
      "step": 6250
    },
    {
      "epoch": 0.7325064357594196,
      "grad_norm": 0.33072805404663086,
      "learning_rate": 9.919556533895995e-05,
      "loss": 0.097,
      "step": 6260
    },
    {
      "epoch": 0.7325064357594196,
      "loss_main": 0.07592989504337311,
      "step": 6260
    },
    {
      "epoch": 0.7336765738357126,
      "grad_norm": 0.4232277274131775,
      "learning_rate": 9.919063445104907e-05,
      "loss": 0.0952,
      "step": 6270
    },
    {
      "epoch": 0.7336765738357126,
      "loss_main": 0.0711275115609169,
      "step": 6270
    },
    {
      "epoch": 0.7348467119120056,
      "grad_norm": 0.4128514230251312,
      "learning_rate": 9.918568862032227e-05,
      "loss": 0.0909,
      "step": 6280
    },
    {
      "epoch": 0.7348467119120056,
      "loss_main": 0.11476556211709976,
      "step": 6280
    },
    {
      "epoch": 0.7360168499882986,
      "grad_norm": 0.3992297947406769,
      "learning_rate": 9.918072784828194e-05,
      "loss": 0.0948,
      "step": 6290
    },
    {
      "epoch": 0.7360168499882986,
      "loss_main": 0.10500075668096542,
      "step": 6290
    },
    {
      "epoch": 0.7371869880645916,
      "grad_norm": 0.35043883323669434,
      "learning_rate": 9.917575213643501e-05,
      "loss": 0.0922,
      "step": 6300
    },
    {
      "epoch": 0.7371869880645916,
      "loss_main": 0.11081378906965256,
      "step": 6300
    },
    {
      "epoch": 0.7383571261408847,
      "grad_norm": 0.33423930406570435,
      "learning_rate": 9.917076148629302e-05,
      "loss": 0.0879,
      "step": 6310
    },
    {
      "epoch": 0.7383571261408847,
      "loss_main": 0.07087964564561844,
      "step": 6310
    },
    {
      "epoch": 0.7395272642171776,
      "grad_norm": 0.40930604934692383,
      "learning_rate": 9.916575589937196e-05,
      "loss": 0.0916,
      "step": 6320
    },
    {
      "epoch": 0.7395272642171776,
      "loss_main": 0.11114398390054703,
      "step": 6320
    },
    {
      "epoch": 0.7406974022934706,
      "grad_norm": 0.530158281326294,
      "learning_rate": 9.916073537719239e-05,
      "loss": 0.0964,
      "step": 6330
    },
    {
      "epoch": 0.7406974022934706,
      "loss_main": 0.08330947160720825,
      "step": 6330
    },
    {
      "epoch": 0.7418675403697637,
      "grad_norm": 0.5160695910453796,
      "learning_rate": 9.915569992127944e-05,
      "loss": 0.0968,
      "step": 6340
    },
    {
      "epoch": 0.7418675403697637,
      "loss_main": 0.09290075302124023,
      "step": 6340
    },
    {
      "epoch": 0.7430376784460566,
      "grad_norm": 0.3612249791622162,
      "learning_rate": 9.915064953316273e-05,
      "loss": 0.0895,
      "step": 6350
    },
    {
      "epoch": 0.7430376784460566,
      "loss_main": 0.06189427152276039,
      "step": 6350
    },
    {
      "epoch": 0.7442078165223497,
      "grad_norm": 0.2895497679710388,
      "learning_rate": 9.914558421437645e-05,
      "loss": 0.0836,
      "step": 6360
    },
    {
      "epoch": 0.7442078165223497,
      "loss_main": 0.0718512311577797,
      "step": 6360
    },
    {
      "epoch": 0.7453779545986426,
      "grad_norm": 0.2787635624408722,
      "learning_rate": 9.914050396645929e-05,
      "loss": 0.0864,
      "step": 6370
    },
    {
      "epoch": 0.7453779545986426,
      "loss_main": 0.08060795813798904,
      "step": 6370
    },
    {
      "epoch": 0.7465480926749356,
      "grad_norm": 0.43233799934387207,
      "learning_rate": 9.913540879095452e-05,
      "loss": 0.0897,
      "step": 6380
    },
    {
      "epoch": 0.7465480926749356,
      "loss_main": 0.12428727000951767,
      "step": 6380
    },
    {
      "epoch": 0.7477182307512287,
      "grad_norm": 0.38457152247428894,
      "learning_rate": 9.913029868940987e-05,
      "loss": 0.097,
      "step": 6390
    },
    {
      "epoch": 0.7477182307512287,
      "loss_main": 0.10829199105501175,
      "step": 6390
    },
    {
      "epoch": 0.7488883688275216,
      "grad_norm": 0.2947362959384918,
      "learning_rate": 9.912517366337772e-05,
      "loss": 0.0945,
      "step": 6400
    },
    {
      "epoch": 0.7488883688275216,
      "loss_main": 0.07787267118692398,
      "step": 6400
    },
    {
      "epoch": 0.7500585069038147,
      "grad_norm": 0.28372302651405334,
      "learning_rate": 9.912003371441487e-05,
      "loss": 0.0854,
      "step": 6410
    },
    {
      "epoch": 0.7500585069038147,
      "loss_main": 0.07175304740667343,
      "step": 6410
    },
    {
      "epoch": 0.7512286449801077,
      "grad_norm": 0.39722466468811035,
      "learning_rate": 9.911487884408271e-05,
      "loss": 0.0882,
      "step": 6420
    },
    {
      "epoch": 0.7512286449801077,
      "loss_main": 0.12091542780399323,
      "step": 6420
    },
    {
      "epoch": 0.7523987830564006,
      "grad_norm": 0.3909449279308319,
      "learning_rate": 9.910970905394719e-05,
      "loss": 0.0982,
      "step": 6430
    },
    {
      "epoch": 0.7523987830564006,
      "loss_main": 0.04740258678793907,
      "step": 6430
    },
    {
      "epoch": 0.7535689211326937,
      "grad_norm": 0.4157430827617645,
      "learning_rate": 9.91045243455787e-05,
      "loss": 0.0762,
      "step": 6440
    },
    {
      "epoch": 0.7535689211326937,
      "loss_main": 0.06602226197719574,
      "step": 6440
    },
    {
      "epoch": 0.7547390592089867,
      "grad_norm": 0.3769332468509674,
      "learning_rate": 9.909932472055225e-05,
      "loss": 0.0863,
      "step": 6450
    },
    {
      "epoch": 0.7547390592089867,
      "loss_main": 0.08644343912601471,
      "step": 6450
    },
    {
      "epoch": 0.7559091972852796,
      "grad_norm": 0.3162662386894226,
      "learning_rate": 9.909411018044734e-05,
      "loss": 0.0742,
      "step": 6460
    },
    {
      "epoch": 0.7559091972852796,
      "loss_main": 0.06842011958360672,
      "step": 6460
    },
    {
      "epoch": 0.7570793353615727,
      "grad_norm": 0.2765434682369232,
      "learning_rate": 9.908888072684802e-05,
      "loss": 0.0824,
      "step": 6470
    },
    {
      "epoch": 0.7570793353615727,
      "loss_main": 0.06351108103990555,
      "step": 6470
    },
    {
      "epoch": 0.7582494734378656,
      "grad_norm": 0.3108423948287964,
      "learning_rate": 9.908363636134285e-05,
      "loss": 0.0735,
      "step": 6480
    },
    {
      "epoch": 0.7582494734378656,
      "loss_main": 0.07290707528591156,
      "step": 6480
    },
    {
      "epoch": 0.7594196115141587,
      "grad_norm": 0.36459824442863464,
      "learning_rate": 9.907837708552493e-05,
      "loss": 0.0748,
      "step": 6490
    },
    {
      "epoch": 0.7594196115141587,
      "loss_main": 0.0785353034734726,
      "step": 6490
    },
    {
      "epoch": 0.7605897495904517,
      "grad_norm": 0.35935303568840027,
      "learning_rate": 9.90731029009919e-05,
      "loss": 0.0848,
      "step": 6500
    },
    {
      "epoch": 0.7605897495904517,
      "loss_main": 0.10280860960483551,
      "step": 6500
    },
    {
      "epoch": 0.7617598876667446,
      "grad_norm": 0.423700213432312,
      "learning_rate": 9.906781380934589e-05,
      "loss": 0.1039,
      "step": 6510
    },
    {
      "epoch": 0.7617598876667446,
      "loss_main": 0.14595140516757965,
      "step": 6510
    },
    {
      "epoch": 0.7629300257430377,
      "grad_norm": 0.3343035578727722,
      "learning_rate": 9.906250981219362e-05,
      "loss": 0.0877,
      "step": 6520
    },
    {
      "epoch": 0.7629300257430377,
      "loss_main": 0.0806494951248169,
      "step": 6520
    },
    {
      "epoch": 0.7641001638193307,
      "grad_norm": 0.3893333971500397,
      "learning_rate": 9.905719091114628e-05,
      "loss": 0.0787,
      "step": 6530
    },
    {
      "epoch": 0.7641001638193307,
      "loss_main": 0.10314243286848068,
      "step": 6530
    },
    {
      "epoch": 0.7652703018956237,
      "grad_norm": 0.4900369346141815,
      "learning_rate": 9.905185710781964e-05,
      "loss": 0.1021,
      "step": 6540
    },
    {
      "epoch": 0.7652703018956237,
      "loss_main": 0.12891355156898499,
      "step": 6540
    },
    {
      "epoch": 0.7664404399719167,
      "grad_norm": 0.25788724422454834,
      "learning_rate": 9.904650840383392e-05,
      "loss": 0.0838,
      "step": 6550
    },
    {
      "epoch": 0.7664404399719167,
      "loss_main": 0.06936277449131012,
      "step": 6550
    },
    {
      "epoch": 0.7676105780482096,
      "grad_norm": 0.2777283191680908,
      "learning_rate": 9.904114480081397e-05,
      "loss": 0.08,
      "step": 6560
    },
    {
      "epoch": 0.7676105780482096,
      "loss_main": 0.07167363166809082,
      "step": 6560
    },
    {
      "epoch": 0.7687807161245027,
      "grad_norm": 0.4218005836009979,
      "learning_rate": 9.903576630038906e-05,
      "loss": 0.0914,
      "step": 6570
    },
    {
      "epoch": 0.7687807161245027,
      "loss_main": 0.06352751702070236,
      "step": 6570
    },
    {
      "epoch": 0.7699508542007957,
      "grad_norm": 0.40839695930480957,
      "learning_rate": 9.903037290419309e-05,
      "loss": 0.0902,
      "step": 6580
    },
    {
      "epoch": 0.7699508542007957,
      "loss_main": 0.04897155985236168,
      "step": 6580
    },
    {
      "epoch": 0.7711209922770887,
      "grad_norm": 0.36739465594291687,
      "learning_rate": 9.902496461386439e-05,
      "loss": 0.0866,
      "step": 6590
    },
    {
      "epoch": 0.7711209922770887,
      "loss_main": 0.08238465338945389,
      "step": 6590
    },
    {
      "epoch": 0.7722911303533817,
      "grad_norm": 0.3511083722114563,
      "learning_rate": 9.901954143104588e-05,
      "loss": 0.085,
      "step": 6600
    },
    {
      "epoch": 0.7722911303533817,
      "loss_main": 0.04640374332666397,
      "step": 6600
    },
    {
      "epoch": 0.7734612684296747,
      "grad_norm": 0.5018561482429504,
      "learning_rate": 9.901410335738496e-05,
      "loss": 0.0714,
      "step": 6610
    },
    {
      "epoch": 0.7734612684296747,
      "loss_main": 0.08228472620248795,
      "step": 6610
    },
    {
      "epoch": 0.7746314065059677,
      "grad_norm": 0.334438681602478,
      "learning_rate": 9.900865039453358e-05,
      "loss": 0.0953,
      "step": 6620
    },
    {
      "epoch": 0.7746314065059677,
      "loss_main": 0.09707891941070557,
      "step": 6620
    },
    {
      "epoch": 0.7758015445822607,
      "grad_norm": 0.4423545300960541,
      "learning_rate": 9.900318254414821e-05,
      "loss": 0.0881,
      "step": 6630
    },
    {
      "epoch": 0.7758015445822607,
      "loss_main": 0.12551449239253998,
      "step": 6630
    },
    {
      "epoch": 0.7769716826585538,
      "grad_norm": 0.4552682638168335,
      "learning_rate": 9.899769980788985e-05,
      "loss": 0.0884,
      "step": 6640
    },
    {
      "epoch": 0.7769716826585538,
      "loss_main": 0.06748612225055695,
      "step": 6640
    },
    {
      "epoch": 0.7781418207348467,
      "grad_norm": 0.3686303496360779,
      "learning_rate": 9.899220218742398e-05,
      "loss": 0.0838,
      "step": 6650
    },
    {
      "epoch": 0.7781418207348467,
      "loss_main": 0.08634673058986664,
      "step": 6650
    },
    {
      "epoch": 0.7793119588111397,
      "grad_norm": 0.4563981294631958,
      "learning_rate": 9.898668968442066e-05,
      "loss": 0.0945,
      "step": 6660
    },
    {
      "epoch": 0.7793119588111397,
      "loss_main": 0.08134518563747406,
      "step": 6660
    },
    {
      "epoch": 0.7804820968874328,
      "grad_norm": 0.32341501116752625,
      "learning_rate": 9.898116230055443e-05,
      "loss": 0.0867,
      "step": 6670
    },
    {
      "epoch": 0.7804820968874328,
      "loss_main": 0.08451751619577408,
      "step": 6670
    },
    {
      "epoch": 0.7816522349637257,
      "grad_norm": 0.34433698654174805,
      "learning_rate": 9.897562003750437e-05,
      "loss": 0.0864,
      "step": 6680
    },
    {
      "epoch": 0.7816522349637257,
      "loss_main": 0.11543239653110504,
      "step": 6680
    },
    {
      "epoch": 0.7828223730400187,
      "grad_norm": 0.3580554723739624,
      "learning_rate": 9.897006289695407e-05,
      "loss": 0.0952,
      "step": 6690
    },
    {
      "epoch": 0.7828223730400187,
      "loss_main": 0.09218081086874008,
      "step": 6690
    },
    {
      "epoch": 0.7839925111163117,
      "grad_norm": 0.32895222306251526,
      "learning_rate": 9.896449088059164e-05,
      "loss": 0.0935,
      "step": 6700
    },
    {
      "epoch": 0.7839925111163117,
      "loss_main": 0.07770625501871109,
      "step": 6700
    },
    {
      "epoch": 0.7851626491926047,
      "grad_norm": 0.3161677420139313,
      "learning_rate": 9.89589039901097e-05,
      "loss": 0.0887,
      "step": 6710
    },
    {
      "epoch": 0.7851626491926047,
      "loss_main": 0.07013057917356491,
      "step": 6710
    },
    {
      "epoch": 0.7863327872688978,
      "grad_norm": 0.28014591336250305,
      "learning_rate": 9.895330222720542e-05,
      "loss": 0.0767,
      "step": 6720
    },
    {
      "epoch": 0.7863327872688978,
      "loss_main": 0.08248235285282135,
      "step": 6720
    },
    {
      "epoch": 0.7875029253451907,
      "grad_norm": 0.32325437664985657,
      "learning_rate": 9.894768559358047e-05,
      "loss": 0.0926,
      "step": 6730
    },
    {
      "epoch": 0.7875029253451907,
      "loss_main": 0.1262744963169098,
      "step": 6730
    },
    {
      "epoch": 0.7886730634214837,
      "grad_norm": 0.41219696402549744,
      "learning_rate": 9.894205409094101e-05,
      "loss": 0.0984,
      "step": 6740
    },
    {
      "epoch": 0.7886730634214837,
      "loss_main": 0.1439230591058731,
      "step": 6740
    },
    {
      "epoch": 0.7898432014977768,
      "grad_norm": 0.2971409559249878,
      "learning_rate": 9.893640772099777e-05,
      "loss": 0.0852,
      "step": 6750
    },
    {
      "epoch": 0.7898432014977768,
      "loss_main": 0.09616067260503769,
      "step": 6750
    },
    {
      "epoch": 0.7910133395740697,
      "grad_norm": 0.35284286737442017,
      "learning_rate": 9.893074648546595e-05,
      "loss": 0.0975,
      "step": 6760
    },
    {
      "epoch": 0.7910133395740697,
      "loss_main": 0.09931086003780365,
      "step": 6760
    },
    {
      "epoch": 0.7921834776503628,
      "grad_norm": 0.42449432611465454,
      "learning_rate": 9.892507038606528e-05,
      "loss": 0.0931,
      "step": 6770
    },
    {
      "epoch": 0.7921834776503628,
      "loss_main": 0.08961112797260284,
      "step": 6770
    },
    {
      "epoch": 0.7933536157266557,
      "grad_norm": 0.3756568729877472,
      "learning_rate": 9.891937942452003e-05,
      "loss": 0.0845,
      "step": 6780
    },
    {
      "epoch": 0.7933536157266557,
      "loss_main": 0.12429440021514893,
      "step": 6780
    },
    {
      "epoch": 0.7945237538029487,
      "grad_norm": 0.45263227820396423,
      "learning_rate": 9.891367360255895e-05,
      "loss": 0.0908,
      "step": 6790
    },
    {
      "epoch": 0.7945237538029487,
      "loss_main": 0.1397252231836319,
      "step": 6790
    },
    {
      "epoch": 0.7956938918792418,
      "grad_norm": 0.4246389865875244,
      "learning_rate": 9.890795292191532e-05,
      "loss": 0.1076,
      "step": 6800
    },
    {
      "epoch": 0.7956938918792418,
      "loss_main": 0.10676251351833344,
      "step": 6800
    },
    {
      "epoch": 0.7968640299555347,
      "grad_norm": 0.46237173676490784,
      "learning_rate": 9.890221738432694e-05,
      "loss": 0.09,
      "step": 6810
    },
    {
      "epoch": 0.7968640299555347,
      "loss_main": 0.07347900420427322,
      "step": 6810
    },
    {
      "epoch": 0.7980341680318278,
      "grad_norm": 0.29968929290771484,
      "learning_rate": 9.88964669915361e-05,
      "loss": 0.0881,
      "step": 6820
    },
    {
      "epoch": 0.7980341680318278,
      "loss_main": 0.0871758982539177,
      "step": 6820
    },
    {
      "epoch": 0.7992043061081208,
      "grad_norm": 0.38160908222198486,
      "learning_rate": 9.889070174528963e-05,
      "loss": 0.0956,
      "step": 6830
    },
    {
      "epoch": 0.7992043061081208,
      "loss_main": 0.0623694472014904,
      "step": 6830
    },
    {
      "epoch": 0.8003744441844137,
      "grad_norm": 0.2978762984275818,
      "learning_rate": 9.888492164733883e-05,
      "loss": 0.0807,
      "step": 6840
    },
    {
      "epoch": 0.8003744441844137,
      "loss_main": 0.09543197602033615,
      "step": 6840
    },
    {
      "epoch": 0.8015445822607068,
      "grad_norm": 0.30988818407058716,
      "learning_rate": 9.88791266994396e-05,
      "loss": 0.0876,
      "step": 6850
    },
    {
      "epoch": 0.8015445822607068,
      "loss_main": 0.0780365914106369,
      "step": 6850
    },
    {
      "epoch": 0.8027147203369998,
      "grad_norm": 0.39542627334594727,
      "learning_rate": 9.887331690335223e-05,
      "loss": 0.0872,
      "step": 6860
    },
    {
      "epoch": 0.8027147203369998,
      "loss_main": 0.15983958542346954,
      "step": 6860
    },
    {
      "epoch": 0.8038848584132927,
      "grad_norm": 0.4091472029685974,
      "learning_rate": 9.886749226084163e-05,
      "loss": 0.0797,
      "step": 6870
    },
    {
      "epoch": 0.8038848584132927,
      "loss_main": 0.052642565220594406,
      "step": 6870
    },
    {
      "epoch": 0.8050549964895858,
      "grad_norm": 0.4804246425628662,
      "learning_rate": 9.886165277367714e-05,
      "loss": 0.1005,
      "step": 6880
    },
    {
      "epoch": 0.8050549964895858,
      "loss_main": 0.04484792426228523,
      "step": 6880
    },
    {
      "epoch": 0.8062251345658787,
      "grad_norm": 0.24817273020744324,
      "learning_rate": 9.885579844363265e-05,
      "loss": 0.0857,
      "step": 6890
    },
    {
      "epoch": 0.8062251345658787,
      "loss_main": 0.13317112624645233,
      "step": 6890
    },
    {
      "epoch": 0.8073952726421718,
      "grad_norm": 0.3466450273990631,
      "learning_rate": 9.884992927248656e-05,
      "loss": 0.09,
      "step": 6900
    },
    {
      "epoch": 0.8073952726421718,
      "loss_main": 0.08222687989473343,
      "step": 6900
    },
    {
      "epoch": 0.8085654107184648,
      "grad_norm": 0.43617603182792664,
      "learning_rate": 9.884404526202178e-05,
      "loss": 0.0949,
      "step": 6910
    },
    {
      "epoch": 0.8085654107184648,
      "loss_main": 0.08959250897169113,
      "step": 6910
    },
    {
      "epoch": 0.8097355487947577,
      "grad_norm": 0.405300498008728,
      "learning_rate": 9.883814641402568e-05,
      "loss": 0.1013,
      "step": 6920
    },
    {
      "epoch": 0.8097355487947577,
      "loss_main": 0.0694580227136612,
      "step": 6920
    },
    {
      "epoch": 0.8109056868710508,
      "grad_norm": 0.22023607790470123,
      "learning_rate": 9.88322327302902e-05,
      "loss": 0.0817,
      "step": 6930
    },
    {
      "epoch": 0.8109056868710508,
      "loss_main": 0.06988134235143661,
      "step": 6930
    },
    {
      "epoch": 0.8120758249473438,
      "grad_norm": 0.27796855568885803,
      "learning_rate": 9.882630421261176e-05,
      "loss": 0.0821,
      "step": 6940
    },
    {
      "epoch": 0.8120758249473438,
      "loss_main": 0.06820011883974075,
      "step": 6940
    },
    {
      "epoch": 0.8132459630236368,
      "grad_norm": 0.2963891327381134,
      "learning_rate": 9.88203608627913e-05,
      "loss": 0.0924,
      "step": 6950
    },
    {
      "epoch": 0.8132459630236368,
      "loss_main": 0.10501110553741455,
      "step": 6950
    },
    {
      "epoch": 0.8144161010999298,
      "grad_norm": 0.3476846218109131,
      "learning_rate": 9.881440268263422e-05,
      "loss": 0.0889,
      "step": 6960
    },
    {
      "epoch": 0.8144161010999298,
      "loss_main": 0.04583153501152992,
      "step": 6960
    },
    {
      "epoch": 0.8155862391762227,
      "grad_norm": 0.375394344329834,
      "learning_rate": 9.880842967395048e-05,
      "loss": 0.0822,
      "step": 6970
    },
    {
      "epoch": 0.8155862391762227,
      "loss_main": 0.056429002434015274,
      "step": 6970
    },
    {
      "epoch": 0.8167563772525158,
      "grad_norm": 0.40629327297210693,
      "learning_rate": 9.880244183855452e-05,
      "loss": 0.0814,
      "step": 6980
    },
    {
      "epoch": 0.8167563772525158,
      "loss_main": 0.08544851839542389,
      "step": 6980
    },
    {
      "epoch": 0.8179265153288088,
      "grad_norm": 0.34903955459594727,
      "learning_rate": 9.879643917826527e-05,
      "loss": 0.098,
      "step": 6990
    },
    {
      "epoch": 0.8179265153288088,
      "loss_main": 0.16843053698539734,
      "step": 6990
    },
    {
      "epoch": 0.8190966534051018,
      "grad_norm": 0.422690212726593,
      "learning_rate": 9.87904216949062e-05,
      "loss": 0.087,
      "step": 7000
    },
    {
      "epoch": 0.8190966534051018,
      "loss_main": 0.123061902821064,
      "step": 7000
    },
    {
      "epoch": 0.8202667914813948,
      "grad_norm": 0.36990639567375183,
      "learning_rate": 9.878438939030526e-05,
      "loss": 0.098,
      "step": 7010
    },
    {
      "epoch": 0.8202667914813948,
      "loss_main": 0.056445688009262085,
      "step": 7010
    },
    {
      "epoch": 0.8214369295576878,
      "grad_norm": 0.3760337829589844,
      "learning_rate": 9.877834226629489e-05,
      "loss": 0.0827,
      "step": 7020
    },
    {
      "epoch": 0.8214369295576878,
      "loss_main": 0.09409775584936142,
      "step": 7020
    },
    {
      "epoch": 0.8226070676339808,
      "grad_norm": 0.2866468131542206,
      "learning_rate": 9.877228032471206e-05,
      "loss": 0.0767,
      "step": 7030
    },
    {
      "epoch": 0.8226070676339808,
      "loss_main": 0.09243357181549072,
      "step": 7030
    },
    {
      "epoch": 0.8237772057102738,
      "grad_norm": 0.38436466455459595,
      "learning_rate": 9.876620356739823e-05,
      "loss": 0.0876,
      "step": 7040
    },
    {
      "epoch": 0.8237772057102738,
      "loss_main": 0.0799662321805954,
      "step": 7040
    },
    {
      "epoch": 0.8249473437865669,
      "grad_norm": 0.3215489983558655,
      "learning_rate": 9.876011199619935e-05,
      "loss": 0.082,
      "step": 7050
    },
    {
      "epoch": 0.8249473437865669,
      "loss_main": 0.06772904843091965,
      "step": 7050
    },
    {
      "epoch": 0.8261174818628598,
      "grad_norm": 0.3549204468727112,
      "learning_rate": 9.875400561296589e-05,
      "loss": 0.0807,
      "step": 7060
    },
    {
      "epoch": 0.8261174818628598,
      "loss_main": 0.0911787748336792,
      "step": 7060
    },
    {
      "epoch": 0.8272876199391528,
      "grad_norm": 0.3689754605293274,
      "learning_rate": 9.874788441955278e-05,
      "loss": 0.0936,
      "step": 7070
    },
    {
      "epoch": 0.8272876199391528,
      "loss_main": 0.057989273220300674,
      "step": 7070
    },
    {
      "epoch": 0.8284577580154459,
      "grad_norm": 0.3419407308101654,
      "learning_rate": 9.874174841781951e-05,
      "loss": 0.0917,
      "step": 7080
    },
    {
      "epoch": 0.8284577580154459,
      "loss_main": 0.09711255133152008,
      "step": 7080
    },
    {
      "epoch": 0.8296278960917388,
      "grad_norm": 0.39506685733795166,
      "learning_rate": 9.873559760963003e-05,
      "loss": 0.0818,
      "step": 7090
    },
    {
      "epoch": 0.8296278960917388,
      "loss_main": 0.11509566754102707,
      "step": 7090
    },
    {
      "epoch": 0.8307980341680318,
      "grad_norm": 0.6023132801055908,
      "learning_rate": 9.872943199685278e-05,
      "loss": 0.0953,
      "step": 7100
    },
    {
      "epoch": 0.8307980341680318,
      "loss_main": 0.08945546299219131,
      "step": 7100
    },
    {
      "epoch": 0.8319681722443248,
      "grad_norm": 0.5427823662757874,
      "learning_rate": 9.872325158136071e-05,
      "loss": 0.0942,
      "step": 7110
    },
    {
      "epoch": 0.8319681722443248,
      "loss_main": 0.07644140720367432,
      "step": 7110
    },
    {
      "epoch": 0.8331383103206178,
      "grad_norm": 0.38256025314331055,
      "learning_rate": 9.871705636503128e-05,
      "loss": 0.0914,
      "step": 7120
    },
    {
      "epoch": 0.8331383103206178,
      "loss_main": 0.12201004475355148,
      "step": 7120
    },
    {
      "epoch": 0.8343084483969109,
      "grad_norm": 0.3821331858634949,
      "learning_rate": 9.871084634974641e-05,
      "loss": 0.0914,
      "step": 7130
    },
    {
      "epoch": 0.8343084483969109,
      "loss_main": 0.12225864827632904,
      "step": 7130
    },
    {
      "epoch": 0.8354785864732038,
      "grad_norm": 0.31747955083847046,
      "learning_rate": 9.870462153739257e-05,
      "loss": 0.0976,
      "step": 7140
    },
    {
      "epoch": 0.8354785864732038,
      "loss_main": 0.06732064485549927,
      "step": 7140
    },
    {
      "epoch": 0.8366487245494968,
      "grad_norm": 0.39962026476860046,
      "learning_rate": 9.869838192986067e-05,
      "loss": 0.0917,
      "step": 7150
    },
    {
      "epoch": 0.8366487245494968,
      "loss_main": 0.04726516082882881,
      "step": 7150
    },
    {
      "epoch": 0.8378188626257899,
      "grad_norm": 0.30288824439048767,
      "learning_rate": 9.869212752904616e-05,
      "loss": 0.0921,
      "step": 7160
    },
    {
      "epoch": 0.8378188626257899,
      "loss_main": 0.1143193393945694,
      "step": 7160
    },
    {
      "epoch": 0.8389890007020828,
      "grad_norm": 0.36818912625312805,
      "learning_rate": 9.868585833684894e-05,
      "loss": 0.0897,
      "step": 7170
    },
    {
      "epoch": 0.8389890007020828,
      "loss_main": 0.07705316692590714,
      "step": 7170
    },
    {
      "epoch": 0.8401591387783759,
      "grad_norm": 0.3880237638950348,
      "learning_rate": 9.867957435517342e-05,
      "loss": 0.0911,
      "step": 7180
    },
    {
      "epoch": 0.8401591387783759,
      "loss_main": 0.10081177204847336,
      "step": 7180
    },
    {
      "epoch": 0.8413292768546689,
      "grad_norm": 0.516001284122467,
      "learning_rate": 9.867327558592854e-05,
      "loss": 0.0934,
      "step": 7190
    },
    {
      "epoch": 0.8413292768546689,
      "loss_main": 0.07046687602996826,
      "step": 7190
    },
    {
      "epoch": 0.8424994149309618,
      "grad_norm": 0.3446457087993622,
      "learning_rate": 9.866696203102766e-05,
      "loss": 0.094,
      "step": 7200
    },
    {
      "epoch": 0.8424994149309618,
      "loss_main": 0.06390263885259628,
      "step": 7200
    },
    {
      "epoch": 0.8436695530072549,
      "grad_norm": 0.3758440911769867,
      "learning_rate": 9.86606336923887e-05,
      "loss": 0.0872,
      "step": 7210
    },
    {
      "epoch": 0.8436695530072549,
      "loss_main": 0.07956751435995102,
      "step": 7210
    },
    {
      "epoch": 0.8448396910835478,
      "grad_norm": 0.4561402499675751,
      "learning_rate": 9.865429057193403e-05,
      "loss": 0.0836,
      "step": 7220
    },
    {
      "epoch": 0.8448396910835478,
      "loss_main": 0.10153232514858246,
      "step": 7220
    },
    {
      "epoch": 0.8460098291598409,
      "grad_norm": 0.3895845413208008,
      "learning_rate": 9.864793267159053e-05,
      "loss": 0.0932,
      "step": 7230
    },
    {
      "epoch": 0.8460098291598409,
      "loss_main": 0.0891328975558281,
      "step": 7230
    },
    {
      "epoch": 0.8471799672361339,
      "grad_norm": 0.5239811539649963,
      "learning_rate": 9.864155999328957e-05,
      "loss": 0.0918,
      "step": 7240
    },
    {
      "epoch": 0.8471799672361339,
      "loss_main": 0.10368961095809937,
      "step": 7240
    },
    {
      "epoch": 0.8483501053124268,
      "grad_norm": 0.4063648581504822,
      "learning_rate": 9.8635172538967e-05,
      "loss": 0.0963,
      "step": 7250
    },
    {
      "epoch": 0.8483501053124268,
      "loss_main": 0.09573674201965332,
      "step": 7250
    },
    {
      "epoch": 0.8495202433887199,
      "grad_norm": 0.29836636781692505,
      "learning_rate": 9.862877031056312e-05,
      "loss": 0.0817,
      "step": 7260
    },
    {
      "epoch": 0.8495202433887199,
      "loss_main": 0.12280579656362534,
      "step": 7260
    },
    {
      "epoch": 0.8506903814650129,
      "grad_norm": 0.38391655683517456,
      "learning_rate": 9.862235331002279e-05,
      "loss": 0.1024,
      "step": 7270
    },
    {
      "epoch": 0.8506903814650129,
      "loss_main": 0.08294643461704254,
      "step": 7270
    },
    {
      "epoch": 0.8518605195413059,
      "grad_norm": 0.48701298236846924,
      "learning_rate": 9.861592153929533e-05,
      "loss": 0.0939,
      "step": 7280
    },
    {
      "epoch": 0.8518605195413059,
      "loss_main": 0.14840874075889587,
      "step": 7280
    },
    {
      "epoch": 0.8530306576175989,
      "grad_norm": 0.30925631523132324,
      "learning_rate": 9.860947500033455e-05,
      "loss": 0.0891,
      "step": 7290
    },
    {
      "epoch": 0.8530306576175989,
      "loss_main": 0.1253795623779297,
      "step": 7290
    },
    {
      "epoch": 0.8542007956938918,
      "grad_norm": 0.35936039686203003,
      "learning_rate": 9.86030136950987e-05,
      "loss": 0.0967,
      "step": 7300
    },
    {
      "epoch": 0.8542007956938918,
      "loss_main": 0.06903885304927826,
      "step": 7300
    },
    {
      "epoch": 0.8553709337701849,
      "grad_norm": 0.5421484112739563,
      "learning_rate": 9.85965376255506e-05,
      "loss": 0.0781,
      "step": 7310
    },
    {
      "epoch": 0.8553709337701849,
      "loss_main": 0.05861467868089676,
      "step": 7310
    },
    {
      "epoch": 0.8565410718464779,
      "grad_norm": 0.36903390288352966,
      "learning_rate": 9.859004679365747e-05,
      "loss": 0.0886,
      "step": 7320
    },
    {
      "epoch": 0.8565410718464779,
      "loss_main": 0.061355557292699814,
      "step": 7320
    },
    {
      "epoch": 0.8577112099227708,
      "grad_norm": 0.33556658029556274,
      "learning_rate": 9.858354120139108e-05,
      "loss": 0.094,
      "step": 7330
    },
    {
      "epoch": 0.8577112099227708,
      "loss_main": 0.15409885346889496,
      "step": 7330
    },
    {
      "epoch": 0.8588813479990639,
      "grad_norm": 0.49227339029312134,
      "learning_rate": 9.857702085072764e-05,
      "loss": 0.0914,
      "step": 7340
    },
    {
      "epoch": 0.8588813479990639,
      "loss_main": 0.10487763583660126,
      "step": 7340
    },
    {
      "epoch": 0.8600514860753569,
      "grad_norm": 0.3610241711139679,
      "learning_rate": 9.857048574364787e-05,
      "loss": 0.0934,
      "step": 7350
    },
    {
      "epoch": 0.8600514860753569,
      "loss_main": 0.0736924558877945,
      "step": 7350
    },
    {
      "epoch": 0.8612216241516499,
      "grad_norm": 0.33006227016448975,
      "learning_rate": 9.856393588213698e-05,
      "loss": 0.0977,
      "step": 7360
    },
    {
      "epoch": 0.8612216241516499,
      "loss_main": 0.07298728078603745,
      "step": 7360
    },
    {
      "epoch": 0.8623917622279429,
      "grad_norm": 0.377454549074173,
      "learning_rate": 9.855737126818458e-05,
      "loss": 0.0906,
      "step": 7370
    },
    {
      "epoch": 0.8623917622279429,
      "loss_main": 0.06704869121313095,
      "step": 7370
    },
    {
      "epoch": 0.8635619003042359,
      "grad_norm": 0.2714231014251709,
      "learning_rate": 9.855079190378491e-05,
      "loss": 0.0871,
      "step": 7380
    },
    {
      "epoch": 0.8635619003042359,
      "loss_main": 0.07165475934743881,
      "step": 7380
    },
    {
      "epoch": 0.8647320383805289,
      "grad_norm": 0.36389485001564026,
      "learning_rate": 9.854419779093655e-05,
      "loss": 0.0911,
      "step": 7390
    },
    {
      "epoch": 0.8647320383805289,
      "loss_main": 0.09440987557172775,
      "step": 7390
    },
    {
      "epoch": 0.8659021764568219,
      "grad_norm": 0.45096439123153687,
      "learning_rate": 9.853758893164264e-05,
      "loss": 0.0956,
      "step": 7400
    },
    {
      "epoch": 0.8659021764568219,
      "loss_main": 0.06482785195112228,
      "step": 7400
    },
    {
      "epoch": 0.867072314533115,
      "grad_norm": 0.30716586112976074,
      "learning_rate": 9.853096532791078e-05,
      "loss": 0.0834,
      "step": 7410
    },
    {
      "epoch": 0.867072314533115,
      "loss_main": 0.07516055554151535,
      "step": 7410
    },
    {
      "epoch": 0.8682424526094079,
      "grad_norm": 0.3791157603263855,
      "learning_rate": 9.852432698175304e-05,
      "loss": 0.0916,
      "step": 7420
    },
    {
      "epoch": 0.8682424526094079,
      "loss_main": 0.07598453760147095,
      "step": 7420
    },
    {
      "epoch": 0.8694125906857009,
      "grad_norm": 0.3894015848636627,
      "learning_rate": 9.851767389518597e-05,
      "loss": 0.0957,
      "step": 7430
    },
    {
      "epoch": 0.8694125906857009,
      "loss_main": 0.06420258432626724,
      "step": 7430
    },
    {
      "epoch": 0.8705827287619939,
      "grad_norm": 0.3849622905254364,
      "learning_rate": 9.85110060702306e-05,
      "loss": 0.0811,
      "step": 7440
    },
    {
      "epoch": 0.8705827287619939,
      "loss_main": 0.05812406912446022,
      "step": 7440
    },
    {
      "epoch": 0.8717528668382869,
      "grad_norm": 0.37889084219932556,
      "learning_rate": 9.850432350891245e-05,
      "loss": 0.093,
      "step": 7450
    },
    {
      "epoch": 0.8717528668382869,
      "loss_main": 0.10040603578090668,
      "step": 7450
    },
    {
      "epoch": 0.87292300491458,
      "grad_norm": 0.39627379179000854,
      "learning_rate": 9.84976262132615e-05,
      "loss": 0.0808,
      "step": 7460
    },
    {
      "epoch": 0.87292300491458,
      "loss_main": 0.0738673061132431,
      "step": 7460
    },
    {
      "epoch": 0.8740931429908729,
      "grad_norm": 0.3258231282234192,
      "learning_rate": 9.849091418531222e-05,
      "loss": 0.0927,
      "step": 7470
    },
    {
      "epoch": 0.8740931429908729,
      "loss_main": 0.12943363189697266,
      "step": 7470
    },
    {
      "epoch": 0.8752632810671659,
      "grad_norm": 0.23744964599609375,
      "learning_rate": 9.848418742710353e-05,
      "loss": 0.0963,
      "step": 7480
    },
    {
      "epoch": 0.8752632810671659,
      "loss_main": 0.04142702370882034,
      "step": 7480
    },
    {
      "epoch": 0.876433419143459,
      "grad_norm": 0.43082332611083984,
      "learning_rate": 9.847744594067885e-05,
      "loss": 0.0874,
      "step": 7490
    },
    {
      "epoch": 0.876433419143459,
      "loss_main": 0.07679935544729233,
      "step": 7490
    },
    {
      "epoch": 0.8776035572197519,
      "grad_norm": 0.43294256925582886,
      "learning_rate": 9.847068972808607e-05,
      "loss": 0.0924,
      "step": 7500
    },
    {
      "epoch": 0.8776035572197519,
      "loss_main": 0.09082496166229248,
      "step": 7500
    },
    {
      "epoch": 0.878773695296045,
      "grad_norm": 0.5246843695640564,
      "learning_rate": 9.846391879137756e-05,
      "loss": 0.0849,
      "step": 7510
    },
    {
      "epoch": 0.878773695296045,
      "loss_main": 0.0774611383676529,
      "step": 7510
    },
    {
      "epoch": 0.879943833372338,
      "grad_norm": 0.2827376127243042,
      "learning_rate": 9.845713313261012e-05,
      "loss": 0.0898,
      "step": 7520
    },
    {
      "epoch": 0.879943833372338,
      "loss_main": 0.08853746950626373,
      "step": 7520
    },
    {
      "epoch": 0.8811139714486309,
      "grad_norm": 0.30039921402931213,
      "learning_rate": 9.845033275384505e-05,
      "loss": 0.1004,
      "step": 7530
    },
    {
      "epoch": 0.8811139714486309,
      "loss_main": 0.06440375000238419,
      "step": 7530
    },
    {
      "epoch": 0.882284109524924,
      "grad_norm": 0.35137322545051575,
      "learning_rate": 9.844351765714818e-05,
      "loss": 0.0913,
      "step": 7540
    },
    {
      "epoch": 0.882284109524924,
      "loss_main": 0.09122628718614578,
      "step": 7540
    },
    {
      "epoch": 0.8834542476012169,
      "grad_norm": 0.43792423605918884,
      "learning_rate": 9.843668784458971e-05,
      "loss": 0.0878,
      "step": 7550
    },
    {
      "epoch": 0.8834542476012169,
      "loss_main": 0.0899164006114006,
      "step": 7550
    },
    {
      "epoch": 0.8846243856775099,
      "grad_norm": 0.37089353799819946,
      "learning_rate": 9.842984331824437e-05,
      "loss": 0.0854,
      "step": 7560
    },
    {
      "epoch": 0.8846243856775099,
      "loss_main": 0.07464047521352768,
      "step": 7560
    },
    {
      "epoch": 0.885794523753803,
      "grad_norm": 0.31781700253486633,
      "learning_rate": 9.842298408019133e-05,
      "loss": 0.083,
      "step": 7570
    },
    {
      "epoch": 0.885794523753803,
      "loss_main": 0.07577221840620041,
      "step": 7570
    },
    {
      "epoch": 0.8869646618300959,
      "grad_norm": 0.45997747778892517,
      "learning_rate": 9.841611013251429e-05,
      "loss": 0.0895,
      "step": 7580
    },
    {
      "epoch": 0.8869646618300959,
      "loss_main": 0.07512646168470383,
      "step": 7580
    },
    {
      "epoch": 0.888134799906389,
      "grad_norm": 0.31892645359039307,
      "learning_rate": 9.840922147730133e-05,
      "loss": 0.0846,
      "step": 7590
    },
    {
      "epoch": 0.888134799906389,
      "loss_main": 0.14613623917102814,
      "step": 7590
    },
    {
      "epoch": 0.889304937982682,
      "grad_norm": 0.47072407603263855,
      "learning_rate": 9.840231811664506e-05,
      "loss": 0.0935,
      "step": 7600
    },
    {
      "epoch": 0.889304937982682,
      "loss_main": 0.06335661560297012,
      "step": 7600
    },
    {
      "epoch": 0.8904750760589749,
      "grad_norm": 0.3021983504295349,
      "learning_rate": 9.839540005264252e-05,
      "loss": 0.079,
      "step": 7610
    },
    {
      "epoch": 0.8904750760589749,
      "loss_main": 0.11914283782243729,
      "step": 7610
    },
    {
      "epoch": 0.891645214135268,
      "grad_norm": 0.39036399126052856,
      "learning_rate": 9.838846728739527e-05,
      "loss": 0.0819,
      "step": 7620
    },
    {
      "epoch": 0.891645214135268,
      "loss_main": 0.11331018060445786,
      "step": 7620
    },
    {
      "epoch": 0.8928153522115609,
      "grad_norm": 0.4420784115791321,
      "learning_rate": 9.838151982300927e-05,
      "loss": 0.0935,
      "step": 7630
    },
    {
      "epoch": 0.8928153522115609,
      "loss_main": 0.09696732461452484,
      "step": 7630
    },
    {
      "epoch": 0.893985490287854,
      "grad_norm": 0.2958199381828308,
      "learning_rate": 9.8374557661595e-05,
      "loss": 0.0733,
      "step": 7640
    },
    {
      "epoch": 0.893985490287854,
      "loss_main": 0.10924552381038666,
      "step": 7640
    },
    {
      "epoch": 0.895155628364147,
      "grad_norm": 0.3820244073867798,
      "learning_rate": 9.836758080526735e-05,
      "loss": 0.0901,
      "step": 7650
    },
    {
      "epoch": 0.895155628364147,
      "loss_main": 0.10067474097013474,
      "step": 7650
    },
    {
      "epoch": 0.8963257664404399,
      "grad_norm": 0.4452815353870392,
      "learning_rate": 9.836058925614575e-05,
      "loss": 0.0847,
      "step": 7660
    },
    {
      "epoch": 0.8963257664404399,
      "loss_main": 0.08540499210357666,
      "step": 7660
    },
    {
      "epoch": 0.897495904516733,
      "grad_norm": 0.3487621545791626,
      "learning_rate": 9.8353583016354e-05,
      "loss": 0.0909,
      "step": 7670
    },
    {
      "epoch": 0.897495904516733,
      "loss_main": 0.10033775866031647,
      "step": 7670
    },
    {
      "epoch": 0.898666042593026,
      "grad_norm": 0.344835489988327,
      "learning_rate": 9.834656208802044e-05,
      "loss": 0.0863,
      "step": 7680
    },
    {
      "epoch": 0.898666042593026,
      "loss_main": 0.09465061873197556,
      "step": 7680
    },
    {
      "epoch": 0.899836180669319,
      "grad_norm": 0.3057320713996887,
      "learning_rate": 9.833952647327784e-05,
      "loss": 0.0861,
      "step": 7690
    },
    {
      "epoch": 0.899836180669319,
      "loss_main": 0.054390061646699905,
      "step": 7690
    },
    {
      "epoch": 0.901006318745612,
      "grad_norm": 0.432035356760025,
      "learning_rate": 9.833247617426342e-05,
      "loss": 0.0808,
      "step": 7700
    },
    {
      "epoch": 0.901006318745612,
      "loss_main": 0.08451402187347412,
      "step": 7700
    },
    {
      "epoch": 0.902176456821905,
      "grad_norm": 0.4065592885017395,
      "learning_rate": 9.832541119311889e-05,
      "loss": 0.0897,
      "step": 7710
    },
    {
      "epoch": 0.902176456821905,
      "loss_main": 0.08229128271341324,
      "step": 7710
    },
    {
      "epoch": 0.903346594898198,
      "grad_norm": 0.3261154294013977,
      "learning_rate": 9.83183315319904e-05,
      "loss": 0.1013,
      "step": 7720
    },
    {
      "epoch": 0.903346594898198,
      "loss_main": 0.11447473615407944,
      "step": 7720
    },
    {
      "epoch": 0.904516732974491,
      "grad_norm": 0.3598068654537201,
      "learning_rate": 9.831123719302855e-05,
      "loss": 0.098,
      "step": 7730
    },
    {
      "epoch": 0.904516732974491,
      "loss_main": 0.08325586467981339,
      "step": 7730
    },
    {
      "epoch": 0.9056868710507839,
      "grad_norm": 0.48939990997314453,
      "learning_rate": 9.830412817838842e-05,
      "loss": 0.0927,
      "step": 7740
    },
    {
      "epoch": 0.9056868710507839,
      "loss_main": 0.13301876187324524,
      "step": 7740
    },
    {
      "epoch": 0.906857009127077,
      "grad_norm": 0.49471256136894226,
      "learning_rate": 9.829700449022956e-05,
      "loss": 0.0889,
      "step": 7750
    },
    {
      "epoch": 0.906857009127077,
      "loss_main": 0.09366060793399811,
      "step": 7750
    },
    {
      "epoch": 0.90802714720337,
      "grad_norm": 0.41224604845046997,
      "learning_rate": 9.828986613071593e-05,
      "loss": 0.0854,
      "step": 7760
    },
    {
      "epoch": 0.90802714720337,
      "loss_main": 0.0688486248254776,
      "step": 7760
    },
    {
      "epoch": 0.909197285279663,
      "grad_norm": 0.41174909472465515,
      "learning_rate": 9.828271310201601e-05,
      "loss": 0.0838,
      "step": 7770
    },
    {
      "epoch": 0.909197285279663,
      "loss_main": 0.06494037061929703,
      "step": 7770
    },
    {
      "epoch": 0.910367423355956,
      "grad_norm": 0.3374776244163513,
      "learning_rate": 9.827554540630268e-05,
      "loss": 0.0799,
      "step": 7780
    },
    {
      "epoch": 0.910367423355956,
      "loss_main": 0.09126562625169754,
      "step": 7780
    },
    {
      "epoch": 0.911537561432249,
      "grad_norm": 0.3690633177757263,
      "learning_rate": 9.826836304575329e-05,
      "loss": 0.0986,
      "step": 7790
    },
    {
      "epoch": 0.911537561432249,
      "loss_main": 0.07388101518154144,
      "step": 7790
    },
    {
      "epoch": 0.912707699508542,
      "grad_norm": 0.30800768733024597,
      "learning_rate": 9.826116602254966e-05,
      "loss": 0.082,
      "step": 7800
    },
    {
      "epoch": 0.912707699508542,
      "loss_main": 0.10325915366411209,
      "step": 7800
    },
    {
      "epoch": 0.913877837584835,
      "grad_norm": 0.38037484884262085,
      "learning_rate": 9.825395433887805e-05,
      "loss": 0.0849,
      "step": 7810
    },
    {
      "epoch": 0.913877837584835,
      "loss_main": 0.10000339895486832,
      "step": 7810
    },
    {
      "epoch": 0.915047975661128,
      "grad_norm": 0.4264467656612396,
      "learning_rate": 9.824672799692917e-05,
      "loss": 0.0969,
      "step": 7820
    },
    {
      "epoch": 0.915047975661128,
      "loss_main": 0.10780651867389679,
      "step": 7820
    },
    {
      "epoch": 0.916218113737421,
      "grad_norm": 0.49468567967414856,
      "learning_rate": 9.823948699889823e-05,
      "loss": 0.101,
      "step": 7830
    },
    {
      "epoch": 0.916218113737421,
      "loss_main": 0.06106063723564148,
      "step": 7830
    },
    {
      "epoch": 0.917388251813714,
      "grad_norm": 0.5459566116333008,
      "learning_rate": 9.823223134698483e-05,
      "loss": 0.0886,
      "step": 7840
    },
    {
      "epoch": 0.917388251813714,
      "loss_main": 0.09823139756917953,
      "step": 7840
    },
    {
      "epoch": 0.918558389890007,
      "grad_norm": 0.2970658838748932,
      "learning_rate": 9.822496104339303e-05,
      "loss": 0.0927,
      "step": 7850
    },
    {
      "epoch": 0.918558389890007,
      "loss_main": 0.0968378335237503,
      "step": 7850
    },
    {
      "epoch": 0.9197285279663,
      "grad_norm": 0.30360448360443115,
      "learning_rate": 9.821767609033138e-05,
      "loss": 0.0852,
      "step": 7860
    },
    {
      "epoch": 0.9197285279663,
      "loss_main": 0.07162564247846603,
      "step": 7860
    },
    {
      "epoch": 0.9208986660425931,
      "grad_norm": 0.27964383363723755,
      "learning_rate": 9.821037649001284e-05,
      "loss": 0.0909,
      "step": 7870
    },
    {
      "epoch": 0.9208986660425931,
      "loss_main": 0.10837338119745255,
      "step": 7870
    },
    {
      "epoch": 0.922068804118886,
      "grad_norm": 0.49138781428337097,
      "learning_rate": 9.820306224465486e-05,
      "loss": 0.0908,
      "step": 7880
    },
    {
      "epoch": 0.922068804118886,
      "loss_main": 0.05624980852007866,
      "step": 7880
    },
    {
      "epoch": 0.923238942195179,
      "grad_norm": 0.3498728573322296,
      "learning_rate": 9.819573335647928e-05,
      "loss": 0.0879,
      "step": 7890
    },
    {
      "epoch": 0.923238942195179,
      "loss_main": 0.05157572031021118,
      "step": 7890
    },
    {
      "epoch": 0.9244090802714721,
      "grad_norm": 0.43073734641075134,
      "learning_rate": 9.818838982771246e-05,
      "loss": 0.0975,
      "step": 7900
    },
    {
      "epoch": 0.9244090802714721,
      "loss_main": 0.059510376304388046,
      "step": 7900
    },
    {
      "epoch": 0.925579218347765,
      "grad_norm": 0.3664515018463135,
      "learning_rate": 9.818103166058514e-05,
      "loss": 0.0893,
      "step": 7910
    },
    {
      "epoch": 0.925579218347765,
      "loss_main": 0.0820527896285057,
      "step": 7910
    },
    {
      "epoch": 0.9267493564240581,
      "grad_norm": 0.4191606342792511,
      "learning_rate": 9.817365885733254e-05,
      "loss": 0.0856,
      "step": 7920
    },
    {
      "epoch": 0.9267493564240581,
      "loss_main": 0.08348198235034943,
      "step": 7920
    },
    {
      "epoch": 0.927919494500351,
      "grad_norm": 0.25710082054138184,
      "learning_rate": 9.816627142019434e-05,
      "loss": 0.0841,
      "step": 7930
    },
    {
      "epoch": 0.927919494500351,
      "loss_main": 0.09972421079874039,
      "step": 7930
    },
    {
      "epoch": 0.929089632576644,
      "grad_norm": 0.41236576437950134,
      "learning_rate": 9.815886935141463e-05,
      "loss": 0.0951,
      "step": 7940
    },
    {
      "epoch": 0.929089632576644,
      "loss_main": 0.10397772490978241,
      "step": 7940
    },
    {
      "epoch": 0.9302597706529371,
      "grad_norm": 0.45824724435806274,
      "learning_rate": 9.8151452653242e-05,
      "loss": 0.0837,
      "step": 7950
    },
    {
      "epoch": 0.9302597706529371,
      "loss_main": 0.11850644648075104,
      "step": 7950
    },
    {
      "epoch": 0.93142990872923,
      "grad_norm": 0.3818645477294922,
      "learning_rate": 9.814402132792939e-05,
      "loss": 0.0877,
      "step": 7960
    },
    {
      "epoch": 0.93142990872923,
      "loss_main": 0.1002739667892456,
      "step": 7960
    },
    {
      "epoch": 0.932600046805523,
      "grad_norm": 0.21386310458183289,
      "learning_rate": 9.813657537773428e-05,
      "loss": 0.0856,
      "step": 7970
    },
    {
      "epoch": 0.932600046805523,
      "loss_main": 0.07670629024505615,
      "step": 7970
    },
    {
      "epoch": 0.9337701848818161,
      "grad_norm": 0.27512067556381226,
      "learning_rate": 9.812911480491854e-05,
      "loss": 0.0808,
      "step": 7980
    },
    {
      "epoch": 0.9337701848818161,
      "loss_main": 0.10063057392835617,
      "step": 7980
    },
    {
      "epoch": 0.934940322958109,
      "grad_norm": 0.4614672362804413,
      "learning_rate": 9.81216396117485e-05,
      "loss": 0.0886,
      "step": 7990
    },
    {
      "epoch": 0.934940322958109,
      "loss_main": 0.10861349105834961,
      "step": 7990
    },
    {
      "epoch": 0.9361104610344021,
      "grad_norm": 0.27288728952407837,
      "learning_rate": 9.811414980049491e-05,
      "loss": 0.093,
      "step": 8000
    },
    {
      "epoch": 0.9361104610344021,
      "loss_main": 0.10589984804391861,
      "step": 8000
    },
    {
      "epoch": 0.9372805991106951,
      "grad_norm": 0.31986740231513977,
      "learning_rate": 9.810664537343301e-05,
      "loss": 0.0828,
      "step": 8010
    },
    {
      "epoch": 0.9372805991106951,
      "loss_main": 0.08079470694065094,
      "step": 8010
    },
    {
      "epoch": 0.938450737186988,
      "grad_norm": 0.38221004605293274,
      "learning_rate": 9.809912633284243e-05,
      "loss": 0.0845,
      "step": 8020
    },
    {
      "epoch": 0.938450737186988,
      "loss_main": 0.1117132231593132,
      "step": 8020
    },
    {
      "epoch": 0.9396208752632811,
      "grad_norm": 0.3261908292770386,
      "learning_rate": 9.809159268100725e-05,
      "loss": 0.0842,
      "step": 8030
    },
    {
      "epoch": 0.9396208752632811,
      "loss_main": 0.07807248830795288,
      "step": 8030
    },
    {
      "epoch": 0.940791013339574,
      "grad_norm": 0.3975762724876404,
      "learning_rate": 9.808404442021599e-05,
      "loss": 0.0849,
      "step": 8040
    },
    {
      "epoch": 0.940791013339574,
      "loss_main": 0.092392198741436,
      "step": 8040
    },
    {
      "epoch": 0.9419611514158671,
      "grad_norm": 0.4296547472476959,
      "learning_rate": 9.807648155276163e-05,
      "loss": 0.09,
      "step": 8050
    },
    {
      "epoch": 0.9419611514158671,
      "loss_main": 0.1334933340549469,
      "step": 8050
    },
    {
      "epoch": 0.9431312894921601,
      "grad_norm": 0.2893369197845459,
      "learning_rate": 9.806890408094156e-05,
      "loss": 0.0936,
      "step": 8060
    },
    {
      "epoch": 0.9431312894921601,
      "loss_main": 0.05427458882331848,
      "step": 8060
    },
    {
      "epoch": 0.944301427568453,
      "grad_norm": 0.2960173189640045,
      "learning_rate": 9.806131200705761e-05,
      "loss": 0.0875,
      "step": 8070
    },
    {
      "epoch": 0.944301427568453,
      "loss_main": 0.09081625938415527,
      "step": 8070
    },
    {
      "epoch": 0.9454715656447461,
      "grad_norm": 0.30877628922462463,
      "learning_rate": 9.805370533341605e-05,
      "loss": 0.0925,
      "step": 8080
    },
    {
      "epoch": 0.9454715656447461,
      "loss_main": 0.08193708956241608,
      "step": 8080
    },
    {
      "epoch": 0.9466417037210391,
      "grad_norm": 0.2892475426197052,
      "learning_rate": 9.804608406232762e-05,
      "loss": 0.0853,
      "step": 8090
    },
    {
      "epoch": 0.9466417037210391,
      "loss_main": 0.09223634749650955,
      "step": 8090
    },
    {
      "epoch": 0.9478118417973321,
      "grad_norm": 0.3305436074733734,
      "learning_rate": 9.803844819610741e-05,
      "loss": 0.08,
      "step": 8100
    },
    {
      "epoch": 0.9478118417973321,
      "loss_main": 0.12511983513832092,
      "step": 8100
    },
    {
      "epoch": 0.9489819798736251,
      "grad_norm": 0.2884678840637207,
      "learning_rate": 9.803079773707504e-05,
      "loss": 0.0817,
      "step": 8110
    },
    {
      "epoch": 0.9489819798736251,
      "loss_main": 0.10501018911600113,
      "step": 8110
    },
    {
      "epoch": 0.950152117949918,
      "grad_norm": 0.3587099611759186,
      "learning_rate": 9.802313268755447e-05,
      "loss": 0.0945,
      "step": 8120
    },
    {
      "epoch": 0.950152117949918,
      "loss_main": 0.0817767009139061,
      "step": 8120
    },
    {
      "epoch": 0.9513222560262111,
      "grad_norm": 0.3435811698436737,
      "learning_rate": 9.801545304987419e-05,
      "loss": 0.0887,
      "step": 8130
    },
    {
      "epoch": 0.9513222560262111,
      "loss_main": 0.08150704950094223,
      "step": 8130
    },
    {
      "epoch": 0.9524923941025041,
      "grad_norm": 0.35590460896492004,
      "learning_rate": 9.800775882636704e-05,
      "loss": 0.0817,
      "step": 8140
    },
    {
      "epoch": 0.9524923941025041,
      "loss_main": 0.08284681290388107,
      "step": 8140
    },
    {
      "epoch": 0.9536625321787972,
      "grad_norm": 0.3192521631717682,
      "learning_rate": 9.800005001937034e-05,
      "loss": 0.0809,
      "step": 8150
    },
    {
      "epoch": 0.9536625321787972,
      "loss_main": 0.08603472262620926,
      "step": 8150
    },
    {
      "epoch": 0.9548326702550901,
      "grad_norm": 0.3514145314693451,
      "learning_rate": 9.79923266312258e-05,
      "loss": 0.0844,
      "step": 8160
    },
    {
      "epoch": 0.9548326702550901,
      "loss_main": 0.13286875188350677,
      "step": 8160
    },
    {
      "epoch": 0.9560028083313831,
      "grad_norm": 0.3474111258983612,
      "learning_rate": 9.79845886642796e-05,
      "loss": 0.0825,
      "step": 8170
    },
    {
      "epoch": 0.9560028083313831,
      "loss_main": 0.07603751868009567,
      "step": 8170
    },
    {
      "epoch": 0.9571729464076761,
      "grad_norm": 0.47056999802589417,
      "learning_rate": 9.797683612088233e-05,
      "loss": 0.0861,
      "step": 8180
    },
    {
      "epoch": 0.9571729464076761,
      "loss_main": 0.08353742212057114,
      "step": 8180
    },
    {
      "epoch": 0.9583430844839691,
      "grad_norm": 0.813140332698822,
      "learning_rate": 9.796906900338898e-05,
      "loss": 0.1011,
      "step": 8190
    },
    {
      "epoch": 0.9583430844839691,
      "loss_main": 0.0875173956155777,
      "step": 8190
    },
    {
      "epoch": 0.9595132225602621,
      "grad_norm": 0.2841300070285797,
      "learning_rate": 9.796128731415903e-05,
      "loss": 0.0948,
      "step": 8200
    },
    {
      "epoch": 0.9595132225602621,
      "loss_main": 0.07977358996868134,
      "step": 8200
    },
    {
      "epoch": 0.9606833606365551,
      "grad_norm": 0.3911629319190979,
      "learning_rate": 9.795349105555634e-05,
      "loss": 0.085,
      "step": 8210
    },
    {
      "epoch": 0.9606833606365551,
      "loss_main": 0.09417854249477386,
      "step": 8210
    },
    {
      "epoch": 0.9618534987128481,
      "grad_norm": 0.44994664192199707,
      "learning_rate": 9.794568022994922e-05,
      "loss": 0.0935,
      "step": 8220
    },
    {
      "epoch": 0.9618534987128481,
      "loss_main": 0.08815650641918182,
      "step": 8220
    },
    {
      "epoch": 0.9630236367891412,
      "grad_norm": 0.4616684913635254,
      "learning_rate": 9.793785483971034e-05,
      "loss": 0.0946,
      "step": 8230
    },
    {
      "epoch": 0.9630236367891412,
      "loss_main": 0.09026646614074707,
      "step": 8230
    },
    {
      "epoch": 0.9641937748654341,
      "grad_norm": 0.26224076747894287,
      "learning_rate": 9.793001488721691e-05,
      "loss": 0.0844,
      "step": 8240
    },
    {
      "epoch": 0.9641937748654341,
      "loss_main": 0.06458843499422073,
      "step": 8240
    },
    {
      "epoch": 0.9653639129417271,
      "grad_norm": 0.36409786343574524,
      "learning_rate": 9.792216037485047e-05,
      "loss": 0.0893,
      "step": 8250
    },
    {
      "epoch": 0.9653639129417271,
      "loss_main": 0.11291475594043732,
      "step": 8250
    },
    {
      "epoch": 0.9665340510180201,
      "grad_norm": 0.3375399708747864,
      "learning_rate": 9.791429130499704e-05,
      "loss": 0.0935,
      "step": 8260
    },
    {
      "epoch": 0.9665340510180201,
      "loss_main": 0.09253121167421341,
      "step": 8260
    },
    {
      "epoch": 0.9677041890943131,
      "grad_norm": 0.3984471261501312,
      "learning_rate": 9.790640768004698e-05,
      "loss": 0.0848,
      "step": 8270
    },
    {
      "epoch": 0.9677041890943131,
      "loss_main": 0.09223032742738724,
      "step": 8270
    },
    {
      "epoch": 0.9688743271706062,
      "grad_norm": 0.2784215807914734,
      "learning_rate": 9.789850950239518e-05,
      "loss": 0.0925,
      "step": 8280
    },
    {
      "epoch": 0.9688743271706062,
      "loss_main": 0.11223806440830231,
      "step": 8280
    },
    {
      "epoch": 0.9700444652468991,
      "grad_norm": 0.3077341318130493,
      "learning_rate": 9.789059677444089e-05,
      "loss": 0.0913,
      "step": 8290
    },
    {
      "epoch": 0.9700444652468991,
      "loss_main": 0.04274224117398262,
      "step": 8290
    },
    {
      "epoch": 0.9712146033231921,
      "grad_norm": 0.4875909388065338,
      "learning_rate": 9.788266949858776e-05,
      "loss": 0.0698,
      "step": 8300
    },
    {
      "epoch": 0.9712146033231921,
      "loss_main": 0.08859419077634811,
      "step": 8300
    },
    {
      "epoch": 0.9723847413994852,
      "grad_norm": 0.38979747891426086,
      "learning_rate": 9.787472767724392e-05,
      "loss": 0.0776,
      "step": 8310
    },
    {
      "epoch": 0.9723847413994852,
      "loss_main": 0.05509543418884277,
      "step": 8310
    },
    {
      "epoch": 0.9735548794757781,
      "grad_norm": 0.4457593858242035,
      "learning_rate": 9.786677131282185e-05,
      "loss": 0.0796,
      "step": 8320
    },
    {
      "epoch": 0.9735548794757781,
      "loss_main": 0.08258742839097977,
      "step": 8320
    },
    {
      "epoch": 0.9747250175520712,
      "grad_norm": 0.2588534355163574,
      "learning_rate": 9.785880040773853e-05,
      "loss": 0.0937,
      "step": 8330
    },
    {
      "epoch": 0.9747250175520712,
      "loss_main": 0.08534974604845047,
      "step": 8330
    },
    {
      "epoch": 0.9758951556283642,
      "grad_norm": 0.3475661873817444,
      "learning_rate": 9.785081496441527e-05,
      "loss": 0.0935,
      "step": 8340
    },
    {
      "epoch": 0.9758951556283642,
      "loss_main": 0.08469154685735703,
      "step": 8340
    },
    {
      "epoch": 0.9770652937046571,
      "grad_norm": 0.35264652967453003,
      "learning_rate": 9.784281498527785e-05,
      "loss": 0.083,
      "step": 8350
    },
    {
      "epoch": 0.9770652937046571,
      "loss_main": 0.06550076603889465,
      "step": 8350
    },
    {
      "epoch": 0.9782354317809502,
      "grad_norm": 0.3557131588459015,
      "learning_rate": 9.783480047275646e-05,
      "loss": 0.0969,
      "step": 8360
    },
    {
      "epoch": 0.9782354317809502,
      "loss_main": 0.0752192810177803,
      "step": 8360
    },
    {
      "epoch": 0.9794055698572431,
      "grad_norm": 0.39717862010002136,
      "learning_rate": 9.78267714292857e-05,
      "loss": 0.0898,
      "step": 8370
    },
    {
      "epoch": 0.9794055698572431,
      "loss_main": 0.061021752655506134,
      "step": 8370
    },
    {
      "epoch": 0.9805757079335362,
      "grad_norm": 0.46752047538757324,
      "learning_rate": 9.781872785730454e-05,
      "loss": 0.0879,
      "step": 8380
    },
    {
      "epoch": 0.9805757079335362,
      "loss_main": 0.10705920308828354,
      "step": 8380
    },
    {
      "epoch": 0.9817458460098292,
      "grad_norm": 0.3110704720020294,
      "learning_rate": 9.781066975925646e-05,
      "loss": 0.0832,
      "step": 8390
    },
    {
      "epoch": 0.9817458460098292,
      "loss_main": 0.09587482362985611,
      "step": 8390
    },
    {
      "epoch": 0.9829159840861221,
      "grad_norm": 0.3631443977355957,
      "learning_rate": 9.780259713758928e-05,
      "loss": 0.0878,
      "step": 8400
    },
    {
      "epoch": 0.9829159840861221,
      "loss_main": 0.080192431807518,
      "step": 8400
    },
    {
      "epoch": 0.9840861221624152,
      "grad_norm": 0.40187597274780273,
      "learning_rate": 9.779450999475524e-05,
      "loss": 0.079,
      "step": 8410
    },
    {
      "epoch": 0.9840861221624152,
      "loss_main": 0.10602223873138428,
      "step": 8410
    },
    {
      "epoch": 0.9852562602387082,
      "grad_norm": 0.31083500385284424,
      "learning_rate": 9.7786408333211e-05,
      "loss": 0.0879,
      "step": 8420
    },
    {
      "epoch": 0.9852562602387082,
      "loss_main": 0.053638748824596405,
      "step": 8420
    },
    {
      "epoch": 0.9864263983150011,
      "grad_norm": 0.3173442482948303,
      "learning_rate": 9.777829215541764e-05,
      "loss": 0.0949,
      "step": 8430
    },
    {
      "epoch": 0.9864263983150011,
      "loss_main": 0.11047661304473877,
      "step": 8430
    },
    {
      "epoch": 0.9875965363912942,
      "grad_norm": 0.36807137727737427,
      "learning_rate": 9.777016146384064e-05,
      "loss": 0.0808,
      "step": 8440
    },
    {
      "epoch": 0.9875965363912942,
      "loss_main": 0.11454733461141586,
      "step": 8440
    },
    {
      "epoch": 0.9887666744675871,
      "grad_norm": 0.4184601604938507,
      "learning_rate": 9.776201626094988e-05,
      "loss": 0.0865,
      "step": 8450
    },
    {
      "epoch": 0.9887666744675871,
      "loss_main": 0.10841960459947586,
      "step": 8450
    },
    {
      "epoch": 0.9899368125438802,
      "grad_norm": 0.4433445930480957,
      "learning_rate": 9.775385654921965e-05,
      "loss": 0.0875,
      "step": 8460
    },
    {
      "epoch": 0.9899368125438802,
      "loss_main": 0.1577654629945755,
      "step": 8460
    },
    {
      "epoch": 0.9911069506201732,
      "grad_norm": 0.32529857754707336,
      "learning_rate": 9.774568233112868e-05,
      "loss": 0.0843,
      "step": 8470
    },
    {
      "epoch": 0.9911069506201732,
      "loss_main": 0.10156912356615067,
      "step": 8470
    },
    {
      "epoch": 0.9922770886964661,
      "grad_norm": 0.4604734480381012,
      "learning_rate": 9.773749360916007e-05,
      "loss": 0.0964,
      "step": 8480
    },
    {
      "epoch": 0.9922770886964661,
      "loss_main": 0.19048680365085602,
      "step": 8480
    },
    {
      "epoch": 0.9934472267727592,
      "grad_norm": 0.2838863134384155,
      "learning_rate": 9.772929038580134e-05,
      "loss": 0.0926,
      "step": 8490
    },
    {
      "epoch": 0.9934472267727592,
      "loss_main": 0.07028589397668839,
      "step": 8490
    },
    {
      "epoch": 0.9946173648490522,
      "grad_norm": 0.48165008425712585,
      "learning_rate": 9.772107266354439e-05,
      "loss": 0.0949,
      "step": 8500
    },
    {
      "epoch": 0.9946173648490522,
      "loss_main": 0.12150418758392334,
      "step": 8500
    },
    {
      "epoch": 0.9957875029253452,
      "grad_norm": 0.3898032605648041,
      "learning_rate": 9.77128404448856e-05,
      "loss": 0.0941,
      "step": 8510
    },
    {
      "epoch": 0.9957875029253452,
      "loss_main": 0.06732659786939621,
      "step": 8510
    },
    {
      "epoch": 0.9969576410016382,
      "grad_norm": 0.3597699999809265,
      "learning_rate": 9.770459373232565e-05,
      "loss": 0.0762,
      "step": 8520
    },
    {
      "epoch": 0.9969576410016382,
      "loss_main": 0.10749723017215729,
      "step": 8520
    },
    {
      "epoch": 0.9981277790779312,
      "grad_norm": 0.415035605430603,
      "learning_rate": 9.769633252836969e-05,
      "loss": 0.0845,
      "step": 8530
    },
    {
      "epoch": 0.9981277790779312,
      "loss_main": 0.0734890028834343,
      "step": 8530
    },
    {
      "epoch": 0.9992979171542242,
      "grad_norm": 0.3575156629085541,
      "learning_rate": 9.768805683552724e-05,
      "loss": 0.0754,
      "step": 8540
    },
    {
      "epoch": 0.9992979171542242,
      "loss_main": 0.08243378251791,
      "step": 8540
    },
    {
      "epoch": 1.0004680552305172,
      "grad_norm": 0.3762432932853699,
      "learning_rate": 9.767976665631228e-05,
      "loss": 0.0875,
      "step": 8550
    },
    {
      "epoch": 1.0004680552305172,
      "loss_main": 0.07075261324644089,
      "step": 8550
    },
    {
      "epoch": 1.0016381933068101,
      "grad_norm": 0.480297327041626,
      "learning_rate": 9.767146199324311e-05,
      "loss": 0.0844,
      "step": 8560
    },
    {
      "epoch": 1.0016381933068101,
      "loss_main": 0.06088883429765701,
      "step": 8560
    },
    {
      "epoch": 1.002808331383103,
      "grad_norm": 0.42124447226524353,
      "learning_rate": 9.766314284884249e-05,
      "loss": 0.075,
      "step": 8570
    },
    {
      "epoch": 1.002808331383103,
      "loss_main": 0.11117255687713623,
      "step": 8570
    },
    {
      "epoch": 1.0039784694593963,
      "grad_norm": 0.36394187808036804,
      "learning_rate": 9.765480922563752e-05,
      "loss": 0.0891,
      "step": 8580
    },
    {
      "epoch": 1.0039784694593963,
      "loss_main": 0.11387993395328522,
      "step": 8580
    },
    {
      "epoch": 1.0051486075356892,
      "grad_norm": 0.4525715112686157,
      "learning_rate": 9.764646112615978e-05,
      "loss": 0.0885,
      "step": 8590
    },
    {
      "epoch": 1.0051486075356892,
      "loss_main": 0.0843527689576149,
      "step": 8590
    },
    {
      "epoch": 1.0063187456119822,
      "grad_norm": 0.34153392910957336,
      "learning_rate": 9.763809855294517e-05,
      "loss": 0.0905,
      "step": 8600
    },
    {
      "epoch": 1.0063187456119822,
      "loss_main": 0.12152622640132904,
      "step": 8600
    },
    {
      "epoch": 1.0074888836882752,
      "grad_norm": 0.3213471472263336,
      "learning_rate": 9.762972150853404e-05,
      "loss": 0.0859,
      "step": 8610
    },
    {
      "epoch": 1.0074888836882752,
      "loss_main": 0.10603805631399155,
      "step": 8610
    },
    {
      "epoch": 1.0086590217645681,
      "grad_norm": 0.40867048501968384,
      "learning_rate": 9.762132999547111e-05,
      "loss": 0.0846,
      "step": 8620
    },
    {
      "epoch": 1.0086590217645681,
      "loss_main": 0.0770873874425888,
      "step": 8620
    },
    {
      "epoch": 1.0098291598408613,
      "grad_norm": 0.3514220714569092,
      "learning_rate": 9.761292401630549e-05,
      "loss": 0.0893,
      "step": 8630
    },
    {
      "epoch": 1.0098291598408613,
      "loss_main": 0.09355168789625168,
      "step": 8630
    },
    {
      "epoch": 1.0109992979171543,
      "grad_norm": 0.23135598003864288,
      "learning_rate": 9.76045035735907e-05,
      "loss": 0.0799,
      "step": 8640
    },
    {
      "epoch": 1.0109992979171543,
      "loss_main": 0.0994039922952652,
      "step": 8640
    },
    {
      "epoch": 1.0121694359934472,
      "grad_norm": 0.3155389130115509,
      "learning_rate": 9.759606866988464e-05,
      "loss": 0.0898,
      "step": 8650
    },
    {
      "epoch": 1.0121694359934472,
      "loss_main": 0.0677046924829483,
      "step": 8650
    },
    {
      "epoch": 1.0133395740697402,
      "grad_norm": 0.4032255709171295,
      "learning_rate": 9.758761930774963e-05,
      "loss": 0.072,
      "step": 8660
    },
    {
      "epoch": 1.0133395740697402,
      "loss_main": 0.06561321020126343,
      "step": 8660
    },
    {
      "epoch": 1.0145097121460331,
      "grad_norm": 0.39943715929985046,
      "learning_rate": 9.757915548975235e-05,
      "loss": 0.0807,
      "step": 8670
    },
    {
      "epoch": 1.0145097121460331,
      "loss_main": 0.10326883941888809,
      "step": 8670
    },
    {
      "epoch": 1.0156798502223263,
      "grad_norm": 0.39283859729766846,
      "learning_rate": 9.757067721846389e-05,
      "loss": 0.0852,
      "step": 8680
    },
    {
      "epoch": 1.0156798502223263,
      "loss_main": 0.10390261560678482,
      "step": 8680
    },
    {
      "epoch": 1.0168499882986193,
      "grad_norm": 0.35515570640563965,
      "learning_rate": 9.756218449645971e-05,
      "loss": 0.0837,
      "step": 8690
    },
    {
      "epoch": 1.0168499882986193,
      "loss_main": 0.06523938477039337,
      "step": 8690
    },
    {
      "epoch": 1.0180201263749122,
      "grad_norm": 0.3498123586177826,
      "learning_rate": 9.75536773263197e-05,
      "loss": 0.0902,
      "step": 8700
    },
    {
      "epoch": 1.0180201263749122,
      "loss_main": 0.0668303593993187,
      "step": 8700
    },
    {
      "epoch": 1.0191902644512052,
      "grad_norm": 0.38915178179740906,
      "learning_rate": 9.75451557106281e-05,
      "loss": 0.077,
      "step": 8710
    },
    {
      "epoch": 1.0191902644512052,
      "loss_main": 0.08041548728942871,
      "step": 8710
    },
    {
      "epoch": 1.0203604025274982,
      "grad_norm": 0.503311812877655,
      "learning_rate": 9.753661965197354e-05,
      "loss": 0.0806,
      "step": 8720
    },
    {
      "epoch": 1.0203604025274982,
      "loss_main": 0.0851929560303688,
      "step": 8720
    },
    {
      "epoch": 1.0215305406037913,
      "grad_norm": 0.24264401197433472,
      "learning_rate": 9.752806915294908e-05,
      "loss": 0.0823,
      "step": 8730
    },
    {
      "epoch": 1.0215305406037913,
      "loss_main": 0.07543623447418213,
      "step": 8730
    },
    {
      "epoch": 1.0227006786800843,
      "grad_norm": 0.30645671486854553,
      "learning_rate": 9.75195042161521e-05,
      "loss": 0.076,
      "step": 8740
    },
    {
      "epoch": 1.0227006786800843,
      "loss_main": 0.05616099387407303,
      "step": 8740
    },
    {
      "epoch": 1.0238708167563773,
      "grad_norm": 0.3418373167514801,
      "learning_rate": 9.751092484418442e-05,
      "loss": 0.0738,
      "step": 8750
    },
    {
      "epoch": 1.0238708167563773,
      "loss_main": 0.0787932500243187,
      "step": 8750
    },
    {
      "epoch": 1.0250409548326702,
      "grad_norm": 0.3021087944507599,
      "learning_rate": 9.750233103965224e-05,
      "loss": 0.0885,
      "step": 8760
    },
    {
      "epoch": 1.0250409548326702,
      "loss_main": 0.09114436060190201,
      "step": 8760
    },
    {
      "epoch": 1.0262110929089632,
      "grad_norm": 0.37095025181770325,
      "learning_rate": 9.749372280516611e-05,
      "loss": 0.083,
      "step": 8770
    },
    {
      "epoch": 1.0262110929089632,
      "loss_main": 0.10049556195735931,
      "step": 8770
    },
    {
      "epoch": 1.0273812309852564,
      "grad_norm": 0.4571191370487213,
      "learning_rate": 9.748510014334097e-05,
      "loss": 0.0848,
      "step": 8780
    },
    {
      "epoch": 1.0273812309852564,
      "loss_main": 0.15044532716274261,
      "step": 8780
    },
    {
      "epoch": 1.0285513690615493,
      "grad_norm": 0.377799391746521,
      "learning_rate": 9.747646305679621e-05,
      "loss": 0.1015,
      "step": 8790
    },
    {
      "epoch": 1.0285513690615493,
      "loss_main": 0.050035323947668076,
      "step": 8790
    },
    {
      "epoch": 1.0297215071378423,
      "grad_norm": 0.29534539580345154,
      "learning_rate": 9.74678115481555e-05,
      "loss": 0.1034,
      "step": 8800
    },
    {
      "epoch": 1.0297215071378423,
      "loss_main": 0.08942333608865738,
      "step": 8800
    },
    {
      "epoch": 1.0308916452141352,
      "grad_norm": 0.45365020632743835,
      "learning_rate": 9.745914562004696e-05,
      "loss": 0.0919,
      "step": 8810
    },
    {
      "epoch": 1.0308916452141352,
      "loss_main": 0.13308848440647125,
      "step": 8810
    },
    {
      "epoch": 1.0320617832904282,
      "grad_norm": 0.3885847330093384,
      "learning_rate": 9.745046527510307e-05,
      "loss": 0.0971,
      "step": 8820
    },
    {
      "epoch": 1.0320617832904282,
      "loss_main": 0.07546687871217728,
      "step": 8820
    },
    {
      "epoch": 1.0332319213667214,
      "grad_norm": 0.5248426795005798,
      "learning_rate": 9.744177051596068e-05,
      "loss": 0.0946,
      "step": 8830
    },
    {
      "epoch": 1.0332319213667214,
      "loss_main": 0.15585963428020477,
      "step": 8830
    },
    {
      "epoch": 1.0344020594430143,
      "grad_norm": 0.32719480991363525,
      "learning_rate": 9.743306134526105e-05,
      "loss": 0.0929,
      "step": 8840
    },
    {
      "epoch": 1.0344020594430143,
      "loss_main": 0.10774023085832596,
      "step": 8840
    },
    {
      "epoch": 1.0355721975193073,
      "grad_norm": 0.37972038984298706,
      "learning_rate": 9.742433776564977e-05,
      "loss": 0.0879,
      "step": 8850
    },
    {
      "epoch": 1.0355721975193073,
      "loss_main": 0.08044027537107468,
      "step": 8850
    },
    {
      "epoch": 1.0367423355956003,
      "grad_norm": 0.46102580428123474,
      "learning_rate": 9.741559977977683e-05,
      "loss": 0.0936,
      "step": 8860
    },
    {
      "epoch": 1.0367423355956003,
      "loss_main": 0.08250948041677475,
      "step": 8860
    },
    {
      "epoch": 1.0379124736718932,
      "grad_norm": 0.2802013158798218,
      "learning_rate": 9.740684739029661e-05,
      "loss": 0.082,
      "step": 8870
    },
    {
      "epoch": 1.0379124736718932,
      "loss_main": 0.05843290314078331,
      "step": 8870
    },
    {
      "epoch": 1.0390826117481864,
      "grad_norm": 0.3906973600387573,
      "learning_rate": 9.739808059986789e-05,
      "loss": 0.1,
      "step": 8880
    },
    {
      "epoch": 1.0390826117481864,
      "loss_main": 0.07846423238515854,
      "step": 8880
    },
    {
      "epoch": 1.0402527498244794,
      "grad_norm": 0.24729585647583008,
      "learning_rate": 9.738929941115373e-05,
      "loss": 0.0886,
      "step": 8890
    },
    {
      "epoch": 1.0402527498244794,
      "loss_main": 0.06230468302965164,
      "step": 8890
    },
    {
      "epoch": 1.0414228879007723,
      "grad_norm": 0.27664393186569214,
      "learning_rate": 9.738050382682167e-05,
      "loss": 0.076,
      "step": 8900
    },
    {
      "epoch": 1.0414228879007723,
      "loss_main": 0.09101182967424393,
      "step": 8900
    },
    {
      "epoch": 1.0425930259770653,
      "grad_norm": 0.2601222097873688,
      "learning_rate": 9.737169384954355e-05,
      "loss": 0.0833,
      "step": 8910
    },
    {
      "epoch": 1.0425930259770653,
      "loss_main": 0.07952205091714859,
      "step": 8910
    },
    {
      "epoch": 1.0437631640533582,
      "grad_norm": 0.2965909540653229,
      "learning_rate": 9.736286948199562e-05,
      "loss": 0.075,
      "step": 8920
    },
    {
      "epoch": 1.0437631640533582,
      "loss_main": 0.07797686755657196,
      "step": 8920
    },
    {
      "epoch": 1.0449333021296514,
      "grad_norm": 0.39460328221321106,
      "learning_rate": 9.735403072685848e-05,
      "loss": 0.0837,
      "step": 8930
    },
    {
      "epoch": 1.0449333021296514,
      "loss_main": 0.08163167536258698,
      "step": 8930
    },
    {
      "epoch": 1.0461034402059444,
      "grad_norm": 0.3600192964076996,
      "learning_rate": 9.734517758681712e-05,
      "loss": 0.084,
      "step": 8940
    },
    {
      "epoch": 1.0461034402059444,
      "loss_main": 0.05031397566199303,
      "step": 8940
    },
    {
      "epoch": 1.0472735782822373,
      "grad_norm": 0.3029409646987915,
      "learning_rate": 9.733631006456088e-05,
      "loss": 0.0778,
      "step": 8950
    },
    {
      "epoch": 1.0472735782822373,
      "loss_main": 0.06935448199510574,
      "step": 8950
    },
    {
      "epoch": 1.0484437163585303,
      "grad_norm": 0.3379252254962921,
      "learning_rate": 9.732742816278348e-05,
      "loss": 0.0789,
      "step": 8960
    },
    {
      "epoch": 1.0484437163585303,
      "loss_main": 0.10512926429510117,
      "step": 8960
    },
    {
      "epoch": 1.0496138544348232,
      "grad_norm": 0.3167610168457031,
      "learning_rate": 9.731853188418302e-05,
      "loss": 0.0848,
      "step": 8970
    },
    {
      "epoch": 1.0496138544348232,
      "loss_main": 0.12176742404699326,
      "step": 8970
    },
    {
      "epoch": 1.0507839925111162,
      "grad_norm": 0.419312983751297,
      "learning_rate": 9.730962123146194e-05,
      "loss": 0.0887,
      "step": 8980
    },
    {
      "epoch": 1.0507839925111162,
      "loss_main": 0.11011854559183121,
      "step": 8980
    },
    {
      "epoch": 1.0519541305874094,
      "grad_norm": 0.3123132884502411,
      "learning_rate": 9.730069620732709e-05,
      "loss": 0.0828,
      "step": 8990
    },
    {
      "epoch": 1.0519541305874094,
      "loss_main": 0.06127830967307091,
      "step": 8990
    },
    {
      "epoch": 1.0531242686637023,
      "grad_norm": 0.4084487855434418,
      "learning_rate": 9.72917568144896e-05,
      "loss": 0.0836,
      "step": 9000
    },
    {
      "epoch": 1.0531242686637023,
      "loss_main": 0.07734733074903488,
      "step": 9000
    },
    {
      "epoch": 1.0542944067399953,
      "grad_norm": 0.3045320212841034,
      "learning_rate": 9.728280305566509e-05,
      "loss": 0.0948,
      "step": 9010
    },
    {
      "epoch": 1.0542944067399953,
      "loss_main": 0.07247180491685867,
      "step": 9010
    },
    {
      "epoch": 1.0554645448162883,
      "grad_norm": 0.3817307949066162,
      "learning_rate": 9.727383493357343e-05,
      "loss": 0.0925,
      "step": 9020
    },
    {
      "epoch": 1.0554645448162883,
      "loss_main": 0.10220245271921158,
      "step": 9020
    },
    {
      "epoch": 1.0566346828925812,
      "grad_norm": 0.32285943627357483,
      "learning_rate": 9.726485245093891e-05,
      "loss": 0.0821,
      "step": 9030
    },
    {
      "epoch": 1.0566346828925812,
      "loss_main": 0.07742845267057419,
      "step": 9030
    },
    {
      "epoch": 1.0578048209688744,
      "grad_norm": 0.4654102921485901,
      "learning_rate": 9.725585561049018e-05,
      "loss": 0.0934,
      "step": 9040
    },
    {
      "epoch": 1.0578048209688744,
      "loss_main": 0.09184898436069489,
      "step": 9040
    },
    {
      "epoch": 1.0589749590451674,
      "grad_norm": 0.2955380380153656,
      "learning_rate": 9.724684441496022e-05,
      "loss": 0.0925,
      "step": 9050
    },
    {
      "epoch": 1.0589749590451674,
      "loss_main": 0.06495784223079681,
      "step": 9050
    },
    {
      "epoch": 1.0601450971214603,
      "grad_norm": 0.34755074977874756,
      "learning_rate": 9.72378188670864e-05,
      "loss": 0.0878,
      "step": 9060
    },
    {
      "epoch": 1.0601450971214603,
      "loss_main": 0.08075602352619171,
      "step": 9060
    },
    {
      "epoch": 1.0613152351977533,
      "grad_norm": 0.3311648666858673,
      "learning_rate": 9.722877896961047e-05,
      "loss": 0.0765,
      "step": 9070
    },
    {
      "epoch": 1.0613152351977533,
      "loss_main": 0.09608496725559235,
      "step": 9070
    },
    {
      "epoch": 1.0624853732740462,
      "grad_norm": 0.38991275429725647,
      "learning_rate": 9.721972472527848e-05,
      "loss": 0.0884,
      "step": 9080
    },
    {
      "epoch": 1.0624853732740462,
      "loss_main": 0.07676111906766891,
      "step": 9080
    },
    {
      "epoch": 1.0636555113503394,
      "grad_norm": 0.3586535155773163,
      "learning_rate": 9.721065613684089e-05,
      "loss": 0.0828,
      "step": 9090
    },
    {
      "epoch": 1.0636555113503394,
      "loss_main": 0.08310902863740921,
      "step": 9090
    },
    {
      "epoch": 1.0648256494266324,
      "grad_norm": 0.38851937651634216,
      "learning_rate": 9.72015732070525e-05,
      "loss": 0.0829,
      "step": 9100
    },
    {
      "epoch": 1.0648256494266324,
      "loss_main": 0.09804785251617432,
      "step": 9100
    },
    {
      "epoch": 1.0659957875029253,
      "grad_norm": 0.436929851770401,
      "learning_rate": 9.719247593867244e-05,
      "loss": 0.0732,
      "step": 9110
    },
    {
      "epoch": 1.0659957875029253,
      "loss_main": 0.08881036937236786,
      "step": 9110
    },
    {
      "epoch": 1.0671659255792183,
      "grad_norm": 0.37946414947509766,
      "learning_rate": 9.718336433446423e-05,
      "loss": 0.0818,
      "step": 9120
    },
    {
      "epoch": 1.0671659255792183,
      "loss_main": 0.12216795980930328,
      "step": 9120
    },
    {
      "epoch": 1.0683360636555113,
      "grad_norm": 0.29482150077819824,
      "learning_rate": 9.717423839719574e-05,
      "loss": 0.0832,
      "step": 9130
    },
    {
      "epoch": 1.0683360636555113,
      "loss_main": 0.06276018172502518,
      "step": 9130
    },
    {
      "epoch": 1.0695062017318044,
      "grad_norm": 0.3328438997268677,
      "learning_rate": 9.71650981296392e-05,
      "loss": 0.0782,
      "step": 9140
    },
    {
      "epoch": 1.0695062017318044,
      "loss_main": 0.07683008164167404,
      "step": 9140
    },
    {
      "epoch": 1.0706763398080974,
      "grad_norm": 0.3635464608669281,
      "learning_rate": 9.715594353457118e-05,
      "loss": 0.0805,
      "step": 9150
    },
    {
      "epoch": 1.0706763398080974,
      "loss_main": 0.0828104019165039,
      "step": 9150
    },
    {
      "epoch": 1.0718464778843904,
      "grad_norm": 0.42968225479125977,
      "learning_rate": 9.714677461477257e-05,
      "loss": 0.0803,
      "step": 9160
    },
    {
      "epoch": 1.0718464778843904,
      "loss_main": 0.08205968141555786,
      "step": 9160
    },
    {
      "epoch": 1.0730166159606833,
      "grad_norm": 0.45708411931991577,
      "learning_rate": 9.713759137302869e-05,
      "loss": 0.0886,
      "step": 9170
    },
    {
      "epoch": 1.0730166159606833,
      "loss_main": 0.10358066111803055,
      "step": 9170
    },
    {
      "epoch": 1.0741867540369763,
      "grad_norm": 0.4337170124053955,
      "learning_rate": 9.712839381212914e-05,
      "loss": 0.0853,
      "step": 9180
    },
    {
      "epoch": 1.0741867540369763,
      "loss_main": 0.08680009841918945,
      "step": 9180
    },
    {
      "epoch": 1.0753568921132695,
      "grad_norm": 0.314998060464859,
      "learning_rate": 9.71191819348679e-05,
      "loss": 0.0856,
      "step": 9190
    },
    {
      "epoch": 1.0753568921132695,
      "loss_main": 0.11816387623548508,
      "step": 9190
    },
    {
      "epoch": 1.0765270301895624,
      "grad_norm": 0.31047961115837097,
      "learning_rate": 9.710995574404331e-05,
      "loss": 0.098,
      "step": 9200
    },
    {
      "epoch": 1.0765270301895624,
      "loss_main": 0.11178992688655853,
      "step": 9200
    },
    {
      "epoch": 1.0776971682658554,
      "grad_norm": 0.37209704518318176,
      "learning_rate": 9.710071524245802e-05,
      "loss": 0.0952,
      "step": 9210
    },
    {
      "epoch": 1.0776971682658554,
      "loss_main": 0.09022266417741776,
      "step": 9210
    },
    {
      "epoch": 1.0788673063421483,
      "grad_norm": 0.352431058883667,
      "learning_rate": 9.709146043291906e-05,
      "loss": 0.0789,
      "step": 9220
    },
    {
      "epoch": 1.0788673063421483,
      "loss_main": 0.07291321456432343,
      "step": 9220
    },
    {
      "epoch": 1.0800374444184413,
      "grad_norm": 0.32168126106262207,
      "learning_rate": 9.70821913182378e-05,
      "loss": 0.0871,
      "step": 9230
    },
    {
      "epoch": 1.0800374444184413,
      "loss_main": 0.06777383387088776,
      "step": 9230
    },
    {
      "epoch": 1.0812075824947345,
      "grad_norm": 0.3411179780960083,
      "learning_rate": 9.707290790122995e-05,
      "loss": 0.0779,
      "step": 9240
    },
    {
      "epoch": 1.0812075824947345,
      "loss_main": 0.06181322783231735,
      "step": 9240
    },
    {
      "epoch": 1.0823777205710274,
      "grad_norm": 0.30821239948272705,
      "learning_rate": 9.706361018471557e-05,
      "loss": 0.0824,
      "step": 9250
    },
    {
      "epoch": 1.0823777205710274,
      "loss_main": 0.09592918306589127,
      "step": 9250
    },
    {
      "epoch": 1.0835478586473204,
      "grad_norm": 0.29677602648735046,
      "learning_rate": 9.705429817151906e-05,
      "loss": 0.0894,
      "step": 9260
    },
    {
      "epoch": 1.0835478586473204,
      "loss_main": 0.07908674329519272,
      "step": 9260
    },
    {
      "epoch": 1.0847179967236134,
      "grad_norm": 0.2994176149368286,
      "learning_rate": 9.704497186446917e-05,
      "loss": 0.085,
      "step": 9270
    },
    {
      "epoch": 1.0847179967236134,
      "loss_main": 0.10970311611890793,
      "step": 9270
    },
    {
      "epoch": 1.0858881347999063,
      "grad_norm": 0.40132659673690796,
      "learning_rate": 9.703563126639896e-05,
      "loss": 0.0891,
      "step": 9280
    },
    {
      "epoch": 1.0858881347999063,
      "loss_main": 0.07145563513040543,
      "step": 9280
    },
    {
      "epoch": 1.0870582728761995,
      "grad_norm": 0.351555734872818,
      "learning_rate": 9.70262763801459e-05,
      "loss": 0.0812,
      "step": 9290
    },
    {
      "epoch": 1.0870582728761995,
      "loss_main": 0.07809943705797195,
      "step": 9290
    },
    {
      "epoch": 1.0882284109524925,
      "grad_norm": 0.26566338539123535,
      "learning_rate": 9.701690720855171e-05,
      "loss": 0.0882,
      "step": 9300
    },
    {
      "epoch": 1.0882284109524925,
      "loss_main": 0.07880742102861404,
      "step": 9300
    },
    {
      "epoch": 1.0893985490287854,
      "grad_norm": 0.3832605183124542,
      "learning_rate": 9.700752375446253e-05,
      "loss": 0.0865,
      "step": 9310
    },
    {
      "epoch": 1.0893985490287854,
      "loss_main": 0.14928504824638367,
      "step": 9310
    },
    {
      "epoch": 1.0905686871050784,
      "grad_norm": 0.30993467569351196,
      "learning_rate": 9.69981260207288e-05,
      "loss": 0.0974,
      "step": 9320
    },
    {
      "epoch": 1.0905686871050784,
      "loss_main": 0.10366805642843246,
      "step": 9320
    },
    {
      "epoch": 1.0917388251813713,
      "grad_norm": 0.30478230118751526,
      "learning_rate": 9.698871401020529e-05,
      "loss": 0.0865,
      "step": 9330
    },
    {
      "epoch": 1.0917388251813713,
      "loss_main": 0.07909311354160309,
      "step": 9330
    },
    {
      "epoch": 1.0929089632576643,
      "grad_norm": 0.25742337107658386,
      "learning_rate": 9.697928772575112e-05,
      "loss": 0.0863,
      "step": 9340
    },
    {
      "epoch": 1.0929089632576643,
      "loss_main": 0.0770229622721672,
      "step": 9340
    },
    {
      "epoch": 1.0940791013339575,
      "grad_norm": 0.34803199768066406,
      "learning_rate": 9.696984717022976e-05,
      "loss": 0.075,
      "step": 9350
    },
    {
      "epoch": 1.0940791013339575,
      "loss_main": 0.06988625228404999,
      "step": 9350
    },
    {
      "epoch": 1.0952492394102504,
      "grad_norm": 0.3594333827495575,
      "learning_rate": 9.6960392346509e-05,
      "loss": 0.0814,
      "step": 9360
    },
    {
      "epoch": 1.0952492394102504,
      "loss_main": 0.11414729058742523,
      "step": 9360
    },
    {
      "epoch": 1.0964193774865434,
      "grad_norm": 0.3730534613132477,
      "learning_rate": 9.695092325746097e-05,
      "loss": 0.0969,
      "step": 9370
    },
    {
      "epoch": 1.0964193774865434,
      "loss_main": 0.08792776614427567,
      "step": 9370
    },
    {
      "epoch": 1.0975895155628363,
      "grad_norm": 0.31370675563812256,
      "learning_rate": 9.694143990596211e-05,
      "loss": 0.0856,
      "step": 9380
    },
    {
      "epoch": 1.0975895155628363,
      "loss_main": 0.061580367386341095,
      "step": 9380
    },
    {
      "epoch": 1.0987596536391293,
      "grad_norm": 0.29128938913345337,
      "learning_rate": 9.693194229489325e-05,
      "loss": 0.0817,
      "step": 9390
    },
    {
      "epoch": 1.0987596536391293,
      "loss_main": 0.07178089767694473,
      "step": 9390
    },
    {
      "epoch": 1.0999297917154225,
      "grad_norm": 0.3143561780452728,
      "learning_rate": 9.692243042713944e-05,
      "loss": 0.0828,
      "step": 9400
    },
    {
      "epoch": 1.0999297917154225,
      "loss_main": 0.06612775474786758,
      "step": 9400
    },
    {
      "epoch": 1.1010999297917154,
      "grad_norm": 0.2699459195137024,
      "learning_rate": 9.691290430559022e-05,
      "loss": 0.079,
      "step": 9410
    },
    {
      "epoch": 1.1010999297917154,
      "loss_main": 0.07146750390529633,
      "step": 9410
    },
    {
      "epoch": 1.1022700678680084,
      "grad_norm": 0.3068464994430542,
      "learning_rate": 9.690336393313932e-05,
      "loss": 0.0838,
      "step": 9420
    },
    {
      "epoch": 1.1022700678680084,
      "loss_main": 0.07355327159166336,
      "step": 9420
    },
    {
      "epoch": 1.1034402059443014,
      "grad_norm": 0.30935925245285034,
      "learning_rate": 9.689380931268487e-05,
      "loss": 0.0844,
      "step": 9430
    },
    {
      "epoch": 1.1034402059443014,
      "loss_main": 0.07313091307878494,
      "step": 9430
    },
    {
      "epoch": 1.1046103440205943,
      "grad_norm": 0.3876843750476837,
      "learning_rate": 9.688424044712932e-05,
      "loss": 0.0912,
      "step": 9440
    },
    {
      "epoch": 1.1046103440205943,
      "loss_main": 0.0657576248049736,
      "step": 9440
    },
    {
      "epoch": 1.1057804820968875,
      "grad_norm": 0.34048980474472046,
      "learning_rate": 9.687465733937942e-05,
      "loss": 0.0852,
      "step": 9450
    },
    {
      "epoch": 1.1057804820968875,
      "loss_main": 0.13369500637054443,
      "step": 9450
    },
    {
      "epoch": 1.1069506201731805,
      "grad_norm": 0.2520897388458252,
      "learning_rate": 9.686505999234627e-05,
      "loss": 0.0862,
      "step": 9460
    },
    {
      "epoch": 1.1069506201731805,
      "loss_main": 0.10637475550174713,
      "step": 9460
    },
    {
      "epoch": 1.1081207582494734,
      "grad_norm": 0.2779518663883209,
      "learning_rate": 9.685544840894529e-05,
      "loss": 0.083,
      "step": 9470
    },
    {
      "epoch": 1.1081207582494734,
      "loss_main": 0.061118680983781815,
      "step": 9470
    },
    {
      "epoch": 1.1092908963257664,
      "grad_norm": 0.2845110595226288,
      "learning_rate": 9.684582259209624e-05,
      "loss": 0.0839,
      "step": 9480
    },
    {
      "epoch": 1.1092908963257664,
      "loss_main": 0.08607066422700882,
      "step": 9480
    },
    {
      "epoch": 1.1104610344020593,
      "grad_norm": 0.36044034361839294,
      "learning_rate": 9.683618254472317e-05,
      "loss": 0.0893,
      "step": 9490
    },
    {
      "epoch": 1.1104610344020593,
      "loss_main": 0.10867075622081757,
      "step": 9490
    },
    {
      "epoch": 1.1116311724783525,
      "grad_norm": 0.293749064207077,
      "learning_rate": 9.682652826975449e-05,
      "loss": 0.0808,
      "step": 9500
    },
    {
      "epoch": 1.1116311724783525,
      "loss_main": 0.12106312066316605,
      "step": 9500
    },
    {
      "epoch": 1.1128013105546455,
      "grad_norm": 0.39845889806747437,
      "learning_rate": 9.681685977012291e-05,
      "loss": 0.0805,
      "step": 9510
    },
    {
      "epoch": 1.1128013105546455,
      "loss_main": 0.06391268968582153,
      "step": 9510
    },
    {
      "epoch": 1.1139714486309384,
      "grad_norm": 0.35130026936531067,
      "learning_rate": 9.680717704876546e-05,
      "loss": 0.0756,
      "step": 9520
    },
    {
      "epoch": 1.1139714486309384,
      "loss_main": 0.10326217114925385,
      "step": 9520
    },
    {
      "epoch": 1.1151415867072314,
      "grad_norm": 0.2750699818134308,
      "learning_rate": 9.679748010862349e-05,
      "loss": 0.0906,
      "step": 9530
    },
    {
      "epoch": 1.1151415867072314,
      "loss_main": 0.08779019862413406,
      "step": 9530
    },
    {
      "epoch": 1.1163117247835244,
      "grad_norm": 0.43800127506256104,
      "learning_rate": 9.678776895264267e-05,
      "loss": 0.0909,
      "step": 9540
    },
    {
      "epoch": 1.1163117247835244,
      "loss_main": 0.1114356741309166,
      "step": 9540
    },
    {
      "epoch": 1.1174818628598175,
      "grad_norm": 0.40727144479751587,
      "learning_rate": 9.6778043583773e-05,
      "loss": 0.0969,
      "step": 9550
    },
    {
      "epoch": 1.1174818628598175,
      "loss_main": 0.046645961701869965,
      "step": 9550
    },
    {
      "epoch": 1.1186520009361105,
      "grad_norm": 0.41246598958969116,
      "learning_rate": 9.67683040049688e-05,
      "loss": 0.0854,
      "step": 9560
    },
    {
      "epoch": 1.1186520009361105,
      "loss_main": 0.07664389163255692,
      "step": 9560
    },
    {
      "epoch": 1.1198221390124035,
      "grad_norm": 0.3118351399898529,
      "learning_rate": 9.675855021918869e-05,
      "loss": 0.0856,
      "step": 9570
    },
    {
      "epoch": 1.1198221390124035,
      "loss_main": 0.07038198411464691,
      "step": 9570
    },
    {
      "epoch": 1.1209922770886964,
      "grad_norm": 0.3521362245082855,
      "learning_rate": 9.674878222939561e-05,
      "loss": 0.0785,
      "step": 9580
    },
    {
      "epoch": 1.1209922770886964,
      "loss_main": 0.07919580489397049,
      "step": 9580
    },
    {
      "epoch": 1.1221624151649894,
      "grad_norm": 0.3552834391593933,
      "learning_rate": 9.673900003855681e-05,
      "loss": 0.0763,
      "step": 9590
    },
    {
      "epoch": 1.1221624151649894,
      "loss_main": 0.05543297529220581,
      "step": 9590
    },
    {
      "epoch": 1.1233325532412826,
      "grad_norm": 0.293770432472229,
      "learning_rate": 9.672920364964389e-05,
      "loss": 0.0807,
      "step": 9600
    },
    {
      "epoch": 1.1233325532412826,
      "loss_main": 0.12451677769422531,
      "step": 9600
    },
    {
      "epoch": 1.1245026913175755,
      "grad_norm": 0.38549506664276123,
      "learning_rate": 9.671939306563269e-05,
      "loss": 0.0769,
      "step": 9610
    },
    {
      "epoch": 1.1245026913175755,
      "loss_main": 0.10320352762937546,
      "step": 9610
    },
    {
      "epoch": 1.1256728293938685,
      "grad_norm": 0.368572860956192,
      "learning_rate": 9.670956828950345e-05,
      "loss": 0.0796,
      "step": 9620
    },
    {
      "epoch": 1.1256728293938685,
      "loss_main": 0.10708945244550705,
      "step": 9620
    },
    {
      "epoch": 1.1268429674701614,
      "grad_norm": 0.2659781575202942,
      "learning_rate": 9.669972932424065e-05,
      "loss": 0.0813,
      "step": 9630
    },
    {
      "epoch": 1.1268429674701614,
      "loss_main": 0.06400176882743835,
      "step": 9630
    },
    {
      "epoch": 1.1280131055464544,
      "grad_norm": 0.33483272790908813,
      "learning_rate": 9.668987617283312e-05,
      "loss": 0.0789,
      "step": 9640
    },
    {
      "epoch": 1.1280131055464544,
      "loss_main": 0.04656602814793587,
      "step": 9640
    },
    {
      "epoch": 1.1291832436227476,
      "grad_norm": 0.3069669008255005,
      "learning_rate": 9.668000883827397e-05,
      "loss": 0.0949,
      "step": 9650
    },
    {
      "epoch": 1.1291832436227476,
      "loss_main": 0.08080130070447922,
      "step": 9650
    },
    {
      "epoch": 1.1303533816990405,
      "grad_norm": 0.27889207005500793,
      "learning_rate": 9.667012732356067e-05,
      "loss": 0.0839,
      "step": 9660
    },
    {
      "epoch": 1.1303533816990405,
      "loss_main": 0.08007659018039703,
      "step": 9660
    },
    {
      "epoch": 1.1315235197753335,
      "grad_norm": 0.34241750836372375,
      "learning_rate": 9.666023163169493e-05,
      "loss": 0.0994,
      "step": 9670
    },
    {
      "epoch": 1.1315235197753335,
      "loss_main": 0.11928671598434448,
      "step": 9670
    },
    {
      "epoch": 1.1326936578516265,
      "grad_norm": 0.3326161801815033,
      "learning_rate": 9.665032176568281e-05,
      "loss": 0.0846,
      "step": 9680
    },
    {
      "epoch": 1.1326936578516265,
      "loss_main": 0.07174257189035416,
      "step": 9680
    },
    {
      "epoch": 1.1338637959279194,
      "grad_norm": 0.2784976363182068,
      "learning_rate": 9.664039772853469e-05,
      "loss": 0.0828,
      "step": 9690
    },
    {
      "epoch": 1.1338637959279194,
      "loss_main": 0.06671866774559021,
      "step": 9690
    },
    {
      "epoch": 1.1350339340042126,
      "grad_norm": 0.35806357860565186,
      "learning_rate": 9.663045952326518e-05,
      "loss": 0.0874,
      "step": 9700
    },
    {
      "epoch": 1.1350339340042126,
      "loss_main": 0.089502714574337,
      "step": 9700
    },
    {
      "epoch": 1.1362040720805056,
      "grad_norm": 0.6273112893104553,
      "learning_rate": 9.662050715289328e-05,
      "loss": 0.0846,
      "step": 9710
    },
    {
      "epoch": 1.1362040720805056,
      "loss_main": 0.09305141866207123,
      "step": 9710
    },
    {
      "epoch": 1.1373742101567985,
      "grad_norm": 0.48186221718788147,
      "learning_rate": 9.661054062044226e-05,
      "loss": 0.0888,
      "step": 9720
    },
    {
      "epoch": 1.1373742101567985,
      "loss_main": 0.08328015357255936,
      "step": 9720
    },
    {
      "epoch": 1.1385443482330915,
      "grad_norm": 0.2505088746547699,
      "learning_rate": 9.660055992893968e-05,
      "loss": 0.0894,
      "step": 9730
    },
    {
      "epoch": 1.1385443482330915,
      "loss_main": 0.0990585908293724,
      "step": 9730
    },
    {
      "epoch": 1.1397144863093844,
      "grad_norm": 0.32118576765060425,
      "learning_rate": 9.659056508141739e-05,
      "loss": 0.0898,
      "step": 9740
    },
    {
      "epoch": 1.1397144863093844,
      "loss_main": 0.06285332143306732,
      "step": 9740
    },
    {
      "epoch": 1.1408846243856776,
      "grad_norm": 0.6330459713935852,
      "learning_rate": 9.658055608091161e-05,
      "loss": 0.0889,
      "step": 9750
    },
    {
      "epoch": 1.1408846243856776,
      "loss_main": 0.058521561324596405,
      "step": 9750
    },
    {
      "epoch": 1.1420547624619706,
      "grad_norm": 0.37538525462150574,
      "learning_rate": 9.657053293046276e-05,
      "loss": 0.0988,
      "step": 9760
    },
    {
      "epoch": 1.1420547624619706,
      "loss_main": 0.0778319463133812,
      "step": 9760
    },
    {
      "epoch": 1.1432249005382635,
      "grad_norm": 0.48692217469215393,
      "learning_rate": 9.656049563311564e-05,
      "loss": 0.0849,
      "step": 9770
    },
    {
      "epoch": 1.1432249005382635,
      "loss_main": 0.12825222313404083,
      "step": 9770
    },
    {
      "epoch": 1.1443950386145565,
      "grad_norm": 0.33315542340278625,
      "learning_rate": 9.655044419191929e-05,
      "loss": 0.0888,
      "step": 9780
    },
    {
      "epoch": 1.1443950386145565,
      "loss_main": 0.06266690790653229,
      "step": 9780
    },
    {
      "epoch": 1.1455651766908495,
      "grad_norm": 0.3039812743663788,
      "learning_rate": 9.654037860992711e-05,
      "loss": 0.0878,
      "step": 9790
    },
    {
      "epoch": 1.1455651766908495,
      "loss_main": 0.09676872193813324,
      "step": 9790
    },
    {
      "epoch": 1.1467353147671426,
      "grad_norm": 0.3559618294239044,
      "learning_rate": 9.653029889019672e-05,
      "loss": 0.0842,
      "step": 9800
    },
    {
      "epoch": 1.1467353147671426,
      "loss_main": 0.0673523023724556,
      "step": 9800
    },
    {
      "epoch": 1.1479054528434356,
      "grad_norm": 0.4121895432472229,
      "learning_rate": 9.65202050357901e-05,
      "loss": 0.0789,
      "step": 9810
    },
    {
      "epoch": 1.1479054528434356,
      "loss_main": 0.10926522314548492,
      "step": 9810
    },
    {
      "epoch": 1.1490755909197286,
      "grad_norm": 0.4127914011478424,
      "learning_rate": 9.651009704977347e-05,
      "loss": 0.0795,
      "step": 9820
    },
    {
      "epoch": 1.1490755909197286,
      "loss_main": 0.04418819025158882,
      "step": 9820
    },
    {
      "epoch": 1.1502457289960215,
      "grad_norm": 0.37538060545921326,
      "learning_rate": 9.649997493521738e-05,
      "loss": 0.0833,
      "step": 9830
    },
    {
      "epoch": 1.1502457289960215,
      "loss_main": 0.06684597581624985,
      "step": 9830
    },
    {
      "epoch": 1.1514158670723145,
      "grad_norm": 0.4499756097793579,
      "learning_rate": 9.64898386951967e-05,
      "loss": 0.085,
      "step": 9840
    },
    {
      "epoch": 1.1514158670723145,
      "loss_main": 0.08070119470357895,
      "step": 9840
    },
    {
      "epoch": 1.1525860051486077,
      "grad_norm": 0.2953965663909912,
      "learning_rate": 9.647968833279049e-05,
      "loss": 0.0851,
      "step": 9850
    },
    {
      "epoch": 1.1525860051486077,
      "loss_main": 0.10697858780622482,
      "step": 9850
    },
    {
      "epoch": 1.1537561432249006,
      "grad_norm": 0.4552677273750305,
      "learning_rate": 9.646952385108218e-05,
      "loss": 0.0931,
      "step": 9860
    },
    {
      "epoch": 1.1537561432249006,
      "loss_main": 0.05570054426789284,
      "step": 9860
    },
    {
      "epoch": 1.1549262813011936,
      "grad_norm": 0.3025417923927307,
      "learning_rate": 9.645934525315951e-05,
      "loss": 0.0788,
      "step": 9870
    },
    {
      "epoch": 1.1549262813011936,
      "loss_main": 0.06078498438000679,
      "step": 9870
    },
    {
      "epoch": 1.1560964193774865,
      "grad_norm": 0.33718791604042053,
      "learning_rate": 9.644915254211442e-05,
      "loss": 0.0802,
      "step": 9880
    },
    {
      "epoch": 1.1560964193774865,
      "loss_main": 0.07766187191009521,
      "step": 9880
    },
    {
      "epoch": 1.1572665574537795,
      "grad_norm": 0.3627527356147766,
      "learning_rate": 9.643894572104321e-05,
      "loss": 0.0927,
      "step": 9890
    },
    {
      "epoch": 1.1572665574537795,
      "loss_main": 0.051169853657484055,
      "step": 9890
    },
    {
      "epoch": 1.1584366955300727,
      "grad_norm": 0.2577114403247833,
      "learning_rate": 9.642872479304644e-05,
      "loss": 0.0793,
      "step": 9900
    },
    {
      "epoch": 1.1584366955300727,
      "loss_main": 0.10355670750141144,
      "step": 9900
    },
    {
      "epoch": 1.1596068336063656,
      "grad_norm": 0.33515360951423645,
      "learning_rate": 9.641848976122895e-05,
      "loss": 0.08,
      "step": 9910
    },
    {
      "epoch": 1.1596068336063656,
      "loss_main": 0.07499457895755768,
      "step": 9910
    },
    {
      "epoch": 1.1607769716826586,
      "grad_norm": 0.4053475856781006,
      "learning_rate": 9.64082406286999e-05,
      "loss": 0.0775,
      "step": 9920
    },
    {
      "epoch": 1.1607769716826586,
      "loss_main": 0.06869061291217804,
      "step": 9920
    },
    {
      "epoch": 1.1619471097589515,
      "grad_norm": 0.4461774528026581,
      "learning_rate": 9.639797739857269e-05,
      "loss": 0.0877,
      "step": 9930
    },
    {
      "epoch": 1.1619471097589515,
      "loss_main": 0.07801749557256699,
      "step": 9930
    },
    {
      "epoch": 1.1631172478352445,
      "grad_norm": 0.312921404838562,
      "learning_rate": 9.638770007396498e-05,
      "loss": 0.0839,
      "step": 9940
    },
    {
      "epoch": 1.1631172478352445,
      "loss_main": 0.0842323824763298,
      "step": 9940
    },
    {
      "epoch": 1.1642873859115375,
      "grad_norm": 0.5635484457015991,
      "learning_rate": 9.63774086579988e-05,
      "loss": 0.1003,
      "step": 9950
    },
    {
      "epoch": 1.1642873859115375,
      "loss_main": 0.12705978751182556,
      "step": 9950
    },
    {
      "epoch": 1.1654575239878306,
      "grad_norm": 0.45115649700164795,
      "learning_rate": 9.63671031538004e-05,
      "loss": 0.0808,
      "step": 9960
    },
    {
      "epoch": 1.1654575239878306,
      "loss_main": 0.09332474321126938,
      "step": 9960
    },
    {
      "epoch": 1.1666276620641236,
      "grad_norm": 0.2907744348049164,
      "learning_rate": 9.635678356450031e-05,
      "loss": 0.0805,
      "step": 9970
    },
    {
      "epoch": 1.1666276620641236,
      "loss_main": 0.07194839417934418,
      "step": 9970
    },
    {
      "epoch": 1.1677978001404166,
      "grad_norm": 0.27739790081977844,
      "learning_rate": 9.634644989323336e-05,
      "loss": 0.0835,
      "step": 9980
    },
    {
      "epoch": 1.1677978001404166,
      "loss_main": 0.1389172524213791,
      "step": 9980
    },
    {
      "epoch": 1.1689679382167095,
      "grad_norm": 0.33156347274780273,
      "learning_rate": 9.633610214313861e-05,
      "loss": 0.0874,
      "step": 9990
    },
    {
      "epoch": 1.1689679382167095,
      "loss_main": 0.06132805719971657,
      "step": 9990
    },
    {
      "epoch": 1.1701380762930025,
      "grad_norm": 0.3400235176086426,
      "learning_rate": 9.632574031735951e-05,
      "loss": 0.0898,
      "step": 10000
    },
    {
      "epoch": 1.1701380762930025,
      "loss_main": 0.06740453839302063,
      "step": 10000
    },
    {
      "epoch": 1.1713082143692957,
      "grad_norm": 0.29868924617767334,
      "learning_rate": 9.631536441904364e-05,
      "loss": 0.0777,
      "step": 10010
    },
    {
      "epoch": 1.1713082143692957,
      "loss_main": 0.048979103565216064,
      "step": 10010
    },
    {
      "epoch": 1.1724783524455886,
      "grad_norm": 0.3661920428276062,
      "learning_rate": 9.630497445134293e-05,
      "loss": 0.0941,
      "step": 10020
    },
    {
      "epoch": 1.1724783524455886,
      "loss_main": 0.04202335700392723,
      "step": 10020
    },
    {
      "epoch": 1.1736484905218816,
      "grad_norm": 0.5404410362243652,
      "learning_rate": 9.62945704174136e-05,
      "loss": 0.083,
      "step": 10030
    },
    {
      "epoch": 1.1736484905218816,
      "loss_main": 0.07010005414485931,
      "step": 10030
    },
    {
      "epoch": 1.1748186285981745,
      "grad_norm": 0.3461291790008545,
      "learning_rate": 9.628415232041612e-05,
      "loss": 0.0869,
      "step": 10040
    },
    {
      "epoch": 1.1748186285981745,
      "loss_main": 0.09305059164762497,
      "step": 10040
    },
    {
      "epoch": 1.1759887666744675,
      "grad_norm": 0.3868895173072815,
      "learning_rate": 9.627372016351524e-05,
      "loss": 0.081,
      "step": 10050
    },
    {
      "epoch": 1.1759887666744675,
      "loss_main": 0.11257672309875488,
      "step": 10050
    },
    {
      "epoch": 1.1771589047507607,
      "grad_norm": 0.3108636736869812,
      "learning_rate": 9.626327394987995e-05,
      "loss": 0.084,
      "step": 10060
    },
    {
      "epoch": 1.1771589047507607,
      "loss_main": 0.061874330043792725,
      "step": 10060
    },
    {
      "epoch": 1.1783290428270536,
      "grad_norm": 0.28093937039375305,
      "learning_rate": 9.625281368268355e-05,
      "loss": 0.0838,
      "step": 10070
    },
    {
      "epoch": 1.1783290428270536,
      "loss_main": 0.08952195197343826,
      "step": 10070
    },
    {
      "epoch": 1.1794991809033466,
      "grad_norm": 0.2716468870639801,
      "learning_rate": 9.624233936510357e-05,
      "loss": 0.0825,
      "step": 10080
    },
    {
      "epoch": 1.1794991809033466,
      "loss_main": 0.10018326342105865,
      "step": 10080
    },
    {
      "epoch": 1.1806693189796396,
      "grad_norm": 0.2848038077354431,
      "learning_rate": 9.623185100032187e-05,
      "loss": 0.0862,
      "step": 10090
    },
    {
      "epoch": 1.1806693189796396,
      "loss_main": 0.05538710579276085,
      "step": 10090
    },
    {
      "epoch": 1.1818394570559325,
      "grad_norm": 0.31887826323509216,
      "learning_rate": 9.62213485915245e-05,
      "loss": 0.0825,
      "step": 10100
    },
    {
      "epoch": 1.1818394570559325,
      "loss_main": 0.09295046329498291,
      "step": 10100
    },
    {
      "epoch": 1.1830095951322257,
      "grad_norm": 0.3517570197582245,
      "learning_rate": 9.621083214190186e-05,
      "loss": 0.0855,
      "step": 10110
    },
    {
      "epoch": 1.1830095951322257,
      "loss_main": 0.058417726308107376,
      "step": 10110
    },
    {
      "epoch": 1.1841797332085187,
      "grad_norm": 0.3126736283302307,
      "learning_rate": 9.62003016546485e-05,
      "loss": 0.0859,
      "step": 10120
    },
    {
      "epoch": 1.1841797332085187,
      "loss_main": 0.08627407252788544,
      "step": 10120
    },
    {
      "epoch": 1.1853498712848116,
      "grad_norm": 0.34822359681129456,
      "learning_rate": 9.618975713296339e-05,
      "loss": 0.0756,
      "step": 10130
    },
    {
      "epoch": 1.1853498712848116,
      "loss_main": 0.08403903990983963,
      "step": 10130
    },
    {
      "epoch": 1.1865200093611046,
      "grad_norm": 0.3628190755844116,
      "learning_rate": 9.61791985800496e-05,
      "loss": 0.0756,
      "step": 10140
    },
    {
      "epoch": 1.1865200093611046,
      "loss_main": 0.09259019792079926,
      "step": 10140
    },
    {
      "epoch": 1.1876901474373975,
      "grad_norm": 0.49441272020339966,
      "learning_rate": 9.616862599911458e-05,
      "loss": 0.0854,
      "step": 10150
    },
    {
      "epoch": 1.1876901474373975,
      "loss_main": 0.06771741807460785,
      "step": 10150
    },
    {
      "epoch": 1.1888602855136905,
      "grad_norm": 0.2746838629245758,
      "learning_rate": 9.615803939337e-05,
      "loss": 0.0845,
      "step": 10160
    },
    {
      "epoch": 1.1888602855136905,
      "loss_main": 0.1143883615732193,
      "step": 10160
    },
    {
      "epoch": 1.1900304235899837,
      "grad_norm": 0.2878350019454956,
      "learning_rate": 9.614743876603178e-05,
      "loss": 0.0823,
      "step": 10170
    },
    {
      "epoch": 1.1900304235899837,
      "loss_main": 0.08664257079362869,
      "step": 10170
    },
    {
      "epoch": 1.1912005616662766,
      "grad_norm": 0.38028404116630554,
      "learning_rate": 9.613682412032013e-05,
      "loss": 0.0864,
      "step": 10180
    },
    {
      "epoch": 1.1912005616662766,
      "loss_main": 0.08088161796331406,
      "step": 10180
    },
    {
      "epoch": 1.1923706997425696,
      "grad_norm": 0.41892048716545105,
      "learning_rate": 9.612619545945947e-05,
      "loss": 0.0921,
      "step": 10190
    },
    {
      "epoch": 1.1923706997425696,
      "loss_main": 0.08096504956483841,
      "step": 10190
    },
    {
      "epoch": 1.1935408378188626,
      "grad_norm": 0.39313462376594543,
      "learning_rate": 9.611555278667852e-05,
      "loss": 0.0735,
      "step": 10200
    },
    {
      "epoch": 1.1935408378188626,
      "loss_main": 0.07657206803560257,
      "step": 10200
    },
    {
      "epoch": 1.1947109758951555,
      "grad_norm": 0.38546431064605713,
      "learning_rate": 9.610489610521024e-05,
      "loss": 0.0847,
      "step": 10210
    },
    {
      "epoch": 1.1947109758951555,
      "loss_main": 0.0509195402264595,
      "step": 10210
    },
    {
      "epoch": 1.1958811139714487,
      "grad_norm": 0.2968984842300415,
      "learning_rate": 9.609422541829187e-05,
      "loss": 0.0725,
      "step": 10220
    },
    {
      "epoch": 1.1958811139714487,
      "loss_main": 0.08744243532419205,
      "step": 10220
    },
    {
      "epoch": 1.1970512520477417,
      "grad_norm": 0.41182491183280945,
      "learning_rate": 9.608354072916486e-05,
      "loss": 0.0922,
      "step": 10230
    },
    {
      "epoch": 1.1970512520477417,
      "loss_main": 0.054319802671670914,
      "step": 10230
    },
    {
      "epoch": 1.1982213901240346,
      "grad_norm": 0.2778118848800659,
      "learning_rate": 9.607284204107493e-05,
      "loss": 0.0855,
      "step": 10240
    },
    {
      "epoch": 1.1982213901240346,
      "loss_main": 0.10165173560380936,
      "step": 10240
    },
    {
      "epoch": 1.1993915282003276,
      "grad_norm": 0.5125450491905212,
      "learning_rate": 9.606212935727208e-05,
      "loss": 0.0797,
      "step": 10250
    },
    {
      "epoch": 1.1993915282003276,
      "loss_main": 0.06693320721387863,
      "step": 10250
    },
    {
      "epoch": 1.2005616662766205,
      "grad_norm": 0.36147835850715637,
      "learning_rate": 9.605140268101052e-05,
      "loss": 0.0795,
      "step": 10260
    },
    {
      "epoch": 1.2005616662766205,
      "loss_main": 0.07988124340772629,
      "step": 10260
    },
    {
      "epoch": 1.2017318043529137,
      "grad_norm": 0.4452548027038574,
      "learning_rate": 9.604066201554875e-05,
      "loss": 0.0786,
      "step": 10270
    },
    {
      "epoch": 1.2017318043529137,
      "loss_main": 0.08028792589902878,
      "step": 10270
    },
    {
      "epoch": 1.2029019424292067,
      "grad_norm": 0.3217140734195709,
      "learning_rate": 9.60299073641495e-05,
      "loss": 0.0813,
      "step": 10280
    },
    {
      "epoch": 1.2029019424292067,
      "loss_main": 0.05721786245703697,
      "step": 10280
    },
    {
      "epoch": 1.2040720805054996,
      "grad_norm": 0.36293867230415344,
      "learning_rate": 9.601913873007974e-05,
      "loss": 0.0726,
      "step": 10290
    },
    {
      "epoch": 1.2040720805054996,
      "loss_main": 0.07145268470048904,
      "step": 10290
    },
    {
      "epoch": 1.2052422185817926,
      "grad_norm": 0.2239101380109787,
      "learning_rate": 9.60083561166107e-05,
      "loss": 0.0841,
      "step": 10300
    },
    {
      "epoch": 1.2052422185817926,
      "loss_main": 0.05115760490298271,
      "step": 10300
    },
    {
      "epoch": 1.2064123566580855,
      "grad_norm": 0.3857554495334625,
      "learning_rate": 9.599755952701783e-05,
      "loss": 0.0907,
      "step": 10310
    },
    {
      "epoch": 1.2064123566580855,
      "loss_main": 0.08259725570678711,
      "step": 10310
    },
    {
      "epoch": 1.2075824947343787,
      "grad_norm": 0.2656293511390686,
      "learning_rate": 9.598674896458089e-05,
      "loss": 0.0888,
      "step": 10320
    },
    {
      "epoch": 1.2075824947343787,
      "loss_main": 0.08078434318304062,
      "step": 10320
    },
    {
      "epoch": 1.2087526328106717,
      "grad_norm": 0.3736938536167145,
      "learning_rate": 9.597592443258383e-05,
      "loss": 0.0727,
      "step": 10330
    },
    {
      "epoch": 1.2087526328106717,
      "loss_main": 0.04899736866354942,
      "step": 10330
    },
    {
      "epoch": 1.2099227708869646,
      "grad_norm": 0.2496660053730011,
      "learning_rate": 9.596508593431483e-05,
      "loss": 0.0729,
      "step": 10340
    },
    {
      "epoch": 1.2099227708869646,
      "loss_main": 0.05189812183380127,
      "step": 10340
    },
    {
      "epoch": 1.2110929089632576,
      "grad_norm": 0.3609190881252289,
      "learning_rate": 9.59542334730664e-05,
      "loss": 0.0799,
      "step": 10350
    },
    {
      "epoch": 1.2110929089632576,
      "loss_main": 0.058046869933605194,
      "step": 10350
    },
    {
      "epoch": 1.2122630470395506,
      "grad_norm": 0.3197472095489502,
      "learning_rate": 9.594336705213516e-05,
      "loss": 0.0849,
      "step": 10360
    },
    {
      "epoch": 1.2122630470395506,
      "loss_main": 0.07053236663341522,
      "step": 10360
    },
    {
      "epoch": 1.2134331851158437,
      "grad_norm": 0.2889004349708557,
      "learning_rate": 9.593248667482208e-05,
      "loss": 0.0732,
      "step": 10370
    },
    {
      "epoch": 1.2134331851158437,
      "loss_main": 0.07824545353651047,
      "step": 10370
    },
    {
      "epoch": 1.2146033231921367,
      "grad_norm": 0.36252066493034363,
      "learning_rate": 9.592159234443233e-05,
      "loss": 0.0866,
      "step": 10380
    },
    {
      "epoch": 1.2146033231921367,
      "loss_main": 0.05977489426732063,
      "step": 10380
    },
    {
      "epoch": 1.2157734612684297,
      "grad_norm": 0.48571041226387024,
      "learning_rate": 9.59106840642753e-05,
      "loss": 0.0977,
      "step": 10390
    },
    {
      "epoch": 1.2157734612684297,
      "loss_main": 0.06680970638990402,
      "step": 10390
    },
    {
      "epoch": 1.2169435993447226,
      "grad_norm": 0.4187976121902466,
      "learning_rate": 9.589976183766467e-05,
      "loss": 0.0812,
      "step": 10400
    },
    {
      "epoch": 1.2169435993447226,
      "loss_main": 0.04609579220414162,
      "step": 10400
    },
    {
      "epoch": 1.2181137374210156,
      "grad_norm": 0.4741189777851105,
      "learning_rate": 9.58888256679183e-05,
      "loss": 0.0824,
      "step": 10410
    },
    {
      "epoch": 1.2181137374210156,
      "loss_main": 0.08258768171072006,
      "step": 10410
    },
    {
      "epoch": 1.2192838754973088,
      "grad_norm": 0.29977333545684814,
      "learning_rate": 9.587787555835832e-05,
      "loss": 0.087,
      "step": 10420
    },
    {
      "epoch": 1.2192838754973088,
      "loss_main": 0.11378207057714462,
      "step": 10420
    },
    {
      "epoch": 1.2204540135736017,
      "grad_norm": 0.3810080885887146,
      "learning_rate": 9.586691151231107e-05,
      "loss": 0.0823,
      "step": 10430
    },
    {
      "epoch": 1.2204540135736017,
      "loss_main": 0.07980753481388092,
      "step": 10430
    },
    {
      "epoch": 1.2216241516498947,
      "grad_norm": 0.42899855971336365,
      "learning_rate": 9.585593353310715e-05,
      "loss": 0.0863,
      "step": 10440
    },
    {
      "epoch": 1.2216241516498947,
      "loss_main": 0.10695754736661911,
      "step": 10440
    },
    {
      "epoch": 1.2227942897261876,
      "grad_norm": 0.310057133436203,
      "learning_rate": 9.58449416240814e-05,
      "loss": 0.0837,
      "step": 10450
    },
    {
      "epoch": 1.2227942897261876,
      "loss_main": 0.0531502403318882,
      "step": 10450
    },
    {
      "epoch": 1.2239644278024806,
      "grad_norm": 0.2960018813610077,
      "learning_rate": 9.583393578857283e-05,
      "loss": 0.0888,
      "step": 10460
    },
    {
      "epoch": 1.2239644278024806,
      "loss_main": 0.10317040979862213,
      "step": 10460
    },
    {
      "epoch": 1.2251345658787738,
      "grad_norm": 0.5003922581672668,
      "learning_rate": 9.582291602992474e-05,
      "loss": 0.0879,
      "step": 10470
    },
    {
      "epoch": 1.2251345658787738,
      "loss_main": 0.09921520948410034,
      "step": 10470
    },
    {
      "epoch": 1.2263047039550667,
      "grad_norm": 0.40872055292129517,
      "learning_rate": 9.581188235148466e-05,
      "loss": 0.0888,
      "step": 10480
    },
    {
      "epoch": 1.2263047039550667,
      "loss_main": 0.08668389171361923,
      "step": 10480
    },
    {
      "epoch": 1.2274748420313597,
      "grad_norm": 0.3181769549846649,
      "learning_rate": 9.58008347566043e-05,
      "loss": 0.0791,
      "step": 10490
    },
    {
      "epoch": 1.2274748420313597,
      "loss_main": 0.11598646640777588,
      "step": 10490
    },
    {
      "epoch": 1.2286449801076527,
      "grad_norm": 0.229812353849411,
      "learning_rate": 9.578977324863965e-05,
      "loss": 0.0842,
      "step": 10500
    },
    {
      "epoch": 1.2286449801076527,
      "loss_main": 0.1133512407541275,
      "step": 10500
    },
    {
      "epoch": 1.2298151181839456,
      "grad_norm": 0.48630306124687195,
      "learning_rate": 9.577869783095089e-05,
      "loss": 0.0895,
      "step": 10510
    },
    {
      "epoch": 1.2298151181839456,
      "loss_main": 0.06137679144740105,
      "step": 10510
    },
    {
      "epoch": 1.2309852562602388,
      "grad_norm": 0.25313422083854675,
      "learning_rate": 9.576760850690245e-05,
      "loss": 0.0868,
      "step": 10520
    },
    {
      "epoch": 1.2309852562602388,
      "loss_main": 0.09884832799434662,
      "step": 10520
    },
    {
      "epoch": 1.2321553943365318,
      "grad_norm": 0.4034554064273834,
      "learning_rate": 9.575650527986298e-05,
      "loss": 0.1006,
      "step": 10530
    },
    {
      "epoch": 1.2321553943365318,
      "loss_main": 0.11072434484958649,
      "step": 10530
    },
    {
      "epoch": 1.2333255324128247,
      "grad_norm": 0.31794309616088867,
      "learning_rate": 9.574538815320531e-05,
      "loss": 0.0791,
      "step": 10540
    },
    {
      "epoch": 1.2333255324128247,
      "loss_main": 0.05025804787874222,
      "step": 10540
    },
    {
      "epoch": 1.2344956704891177,
      "grad_norm": 0.38042446970939636,
      "learning_rate": 9.573425713030656e-05,
      "loss": 0.0961,
      "step": 10550
    },
    {
      "epoch": 1.2344956704891177,
      "loss_main": 0.06769810616970062,
      "step": 10550
    },
    {
      "epoch": 1.2356658085654106,
      "grad_norm": 0.325001060962677,
      "learning_rate": 9.572311221454806e-05,
      "loss": 0.0736,
      "step": 10560
    },
    {
      "epoch": 1.2356658085654106,
      "loss_main": 0.09673672169446945,
      "step": 10560
    },
    {
      "epoch": 1.2368359466417038,
      "grad_norm": 0.43128299713134766,
      "learning_rate": 9.57119534093153e-05,
      "loss": 0.0842,
      "step": 10570
    },
    {
      "epoch": 1.2368359466417038,
      "loss_main": 0.07154726982116699,
      "step": 10570
    },
    {
      "epoch": 1.2380060847179968,
      "grad_norm": 0.39221444725990295,
      "learning_rate": 9.570078071799806e-05,
      "loss": 0.0794,
      "step": 10580
    },
    {
      "epoch": 1.2380060847179968,
      "loss_main": 0.08675821125507355,
      "step": 10580
    },
    {
      "epoch": 1.2391762227942897,
      "grad_norm": 0.4428459107875824,
      "learning_rate": 9.568959414399028e-05,
      "loss": 0.0802,
      "step": 10590
    },
    {
      "epoch": 1.2391762227942897,
      "loss_main": 0.10542934387922287,
      "step": 10590
    },
    {
      "epoch": 1.2403463608705827,
      "grad_norm": 0.3178199827671051,
      "learning_rate": 9.567839369069018e-05,
      "loss": 0.082,
      "step": 10600
    },
    {
      "epoch": 1.2403463608705827,
      "loss_main": 0.07800684869289398,
      "step": 10600
    },
    {
      "epoch": 1.2415164989468757,
      "grad_norm": 0.3512130677700043,
      "learning_rate": 9.566717936150013e-05,
      "loss": 0.0869,
      "step": 10610
    },
    {
      "epoch": 1.2415164989468757,
      "loss_main": 0.11889861524105072,
      "step": 10610
    },
    {
      "epoch": 1.2426866370231688,
      "grad_norm": 0.35488128662109375,
      "learning_rate": 9.565595115982678e-05,
      "loss": 0.0856,
      "step": 10620
    },
    {
      "epoch": 1.2426866370231688,
      "loss_main": 0.050773460417985916,
      "step": 10620
    },
    {
      "epoch": 1.2438567750994618,
      "grad_norm": 0.23416709899902344,
      "learning_rate": 9.564470908908094e-05,
      "loss": 0.0836,
      "step": 10630
    },
    {
      "epoch": 1.2438567750994618,
      "loss_main": 0.07860691100358963,
      "step": 10630
    },
    {
      "epoch": 1.2450269131757548,
      "grad_norm": 0.2692639231681824,
      "learning_rate": 9.563345315267764e-05,
      "loss": 0.0795,
      "step": 10640
    },
    {
      "epoch": 1.2450269131757548,
      "loss_main": 0.07477933913469315,
      "step": 10640
    },
    {
      "epoch": 1.2461970512520477,
      "grad_norm": 0.3506230115890503,
      "learning_rate": 9.562218335403616e-05,
      "loss": 0.0796,
      "step": 10650
    },
    {
      "epoch": 1.2461970512520477,
      "loss_main": 0.08625153452157974,
      "step": 10650
    },
    {
      "epoch": 1.2473671893283407,
      "grad_norm": 0.3659668266773224,
      "learning_rate": 9.561089969657999e-05,
      "loss": 0.0806,
      "step": 10660
    },
    {
      "epoch": 1.2473671893283407,
      "loss_main": 0.10906915366649628,
      "step": 10660
    },
    {
      "epoch": 1.2485373274046339,
      "grad_norm": 0.2687537670135498,
      "learning_rate": 9.559960218373673e-05,
      "loss": 0.0826,
      "step": 10670
    },
    {
      "epoch": 1.2485373274046339,
      "loss_main": 0.13035361468791962,
      "step": 10670
    },
    {
      "epoch": 1.2497074654809268,
      "grad_norm": 0.4440796375274658,
      "learning_rate": 9.558829081893836e-05,
      "loss": 0.0832,
      "step": 10680
    },
    {
      "epoch": 1.2497074654809268,
      "loss_main": 0.06341826915740967,
      "step": 10680
    },
    {
      "epoch": 1.2508776035572198,
      "grad_norm": 0.3147069811820984,
      "learning_rate": 9.55769656056209e-05,
      "loss": 0.0733,
      "step": 10690
    },
    {
      "epoch": 1.2508776035572198,
      "loss_main": 0.09111042320728302,
      "step": 10690
    },
    {
      "epoch": 1.2520477416335127,
      "grad_norm": 0.2729041576385498,
      "learning_rate": 9.556562654722469e-05,
      "loss": 0.0844,
      "step": 10700
    },
    {
      "epoch": 1.2520477416335127,
      "loss_main": 0.10279099643230438,
      "step": 10700
    },
    {
      "epoch": 1.2532178797098057,
      "grad_norm": 0.2412513643503189,
      "learning_rate": 9.555427364719422e-05,
      "loss": 0.0869,
      "step": 10710
    },
    {
      "epoch": 1.2532178797098057,
      "loss_main": 0.0887564867734909,
      "step": 10710
    },
    {
      "epoch": 1.2543880177860989,
      "grad_norm": 0.5306709408760071,
      "learning_rate": 9.55429069089782e-05,
      "loss": 0.0868,
      "step": 10720
    },
    {
      "epoch": 1.2543880177860989,
      "loss_main": 0.09197825938463211,
      "step": 10720
    },
    {
      "epoch": 1.2555581558623918,
      "grad_norm": 0.3324313759803772,
      "learning_rate": 9.553152633602956e-05,
      "loss": 0.0854,
      "step": 10730
    },
    {
      "epoch": 1.2555581558623918,
      "loss_main": 0.0978357121348381,
      "step": 10730
    },
    {
      "epoch": 1.2567282939386848,
      "grad_norm": 0.32552653551101685,
      "learning_rate": 9.552013193180543e-05,
      "loss": 0.0916,
      "step": 10740
    },
    {
      "epoch": 1.2567282939386848,
      "loss_main": 0.07184945046901703,
      "step": 10740
    },
    {
      "epoch": 1.2578984320149778,
      "grad_norm": 0.28507542610168457,
      "learning_rate": 9.550872369976707e-05,
      "loss": 0.0813,
      "step": 10750
    },
    {
      "epoch": 1.2578984320149778,
      "loss_main": 0.06891272217035294,
      "step": 10750
    },
    {
      "epoch": 1.2590685700912707,
      "grad_norm": 0.359051376581192,
      "learning_rate": 9.549730164338007e-05,
      "loss": 0.0821,
      "step": 10760
    },
    {
      "epoch": 1.2590685700912707,
      "loss_main": 0.09564195573329926,
      "step": 10760
    },
    {
      "epoch": 1.260238708167564,
      "grad_norm": 0.37988778948783875,
      "learning_rate": 9.548586576611408e-05,
      "loss": 0.0839,
      "step": 10770
    },
    {
      "epoch": 1.260238708167564,
      "loss_main": 0.09629391133785248,
      "step": 10770
    },
    {
      "epoch": 1.2614088462438569,
      "grad_norm": 0.3128255307674408,
      "learning_rate": 9.54744160714431e-05,
      "loss": 0.083,
      "step": 10780
    },
    {
      "epoch": 1.2614088462438569,
      "loss_main": 0.08201923221349716,
      "step": 10780
    },
    {
      "epoch": 1.2625789843201498,
      "grad_norm": 0.4504450559616089,
      "learning_rate": 9.546295256284516e-05,
      "loss": 0.0833,
      "step": 10790
    },
    {
      "epoch": 1.2625789843201498,
      "loss_main": 0.11473822593688965,
      "step": 10790
    },
    {
      "epoch": 1.2637491223964428,
      "grad_norm": 0.3097148835659027,
      "learning_rate": 9.545147524380265e-05,
      "loss": 0.087,
      "step": 10800
    },
    {
      "epoch": 1.2637491223964428,
      "loss_main": 0.07266078144311905,
      "step": 10800
    },
    {
      "epoch": 1.2649192604727357,
      "grad_norm": 0.355871319770813,
      "learning_rate": 9.543998411780201e-05,
      "loss": 0.0821,
      "step": 10810
    },
    {
      "epoch": 1.2649192604727357,
      "loss_main": 0.1095615029335022,
      "step": 10810
    },
    {
      "epoch": 1.266089398549029,
      "grad_norm": 0.30178552865982056,
      "learning_rate": 9.542847918833397e-05,
      "loss": 0.0788,
      "step": 10820
    },
    {
      "epoch": 1.266089398549029,
      "loss_main": 0.12185995280742645,
      "step": 10820
    },
    {
      "epoch": 1.2672595366253219,
      "grad_norm": 0.3307429850101471,
      "learning_rate": 9.541696045889343e-05,
      "loss": 0.0863,
      "step": 10830
    },
    {
      "epoch": 1.2672595366253219,
      "loss_main": 0.07807227224111557,
      "step": 10830
    },
    {
      "epoch": 1.2684296747016148,
      "grad_norm": 0.22929778695106506,
      "learning_rate": 9.540542793297947e-05,
      "loss": 0.08,
      "step": 10840
    },
    {
      "epoch": 1.2684296747016148,
      "loss_main": 0.09372995048761368,
      "step": 10840
    },
    {
      "epoch": 1.2695998127779078,
      "grad_norm": 0.400137722492218,
      "learning_rate": 9.539388161409537e-05,
      "loss": 0.0795,
      "step": 10850
    },
    {
      "epoch": 1.2695998127779078,
      "loss_main": 0.08653604984283447,
      "step": 10850
    },
    {
      "epoch": 1.2707699508542007,
      "grad_norm": 0.33959922194480896,
      "learning_rate": 9.538232150574857e-05,
      "loss": 0.0859,
      "step": 10860
    },
    {
      "epoch": 1.2707699508542007,
      "loss_main": 0.08019407838582993,
      "step": 10860
    },
    {
      "epoch": 1.271940088930494,
      "grad_norm": 0.37593504786491394,
      "learning_rate": 9.537074761145076e-05,
      "loss": 0.0798,
      "step": 10870
    },
    {
      "epoch": 1.271940088930494,
      "loss_main": 0.07168120890855789,
      "step": 10870
    },
    {
      "epoch": 1.2731102270067867,
      "grad_norm": 0.4649168848991394,
      "learning_rate": 9.535915993471778e-05,
      "loss": 0.0777,
      "step": 10880
    },
    {
      "epoch": 1.2731102270067867,
      "loss_main": 0.1309833973646164,
      "step": 10880
    },
    {
      "epoch": 1.2742803650830798,
      "grad_norm": 0.2549704313278198,
      "learning_rate": 9.534755847906964e-05,
      "loss": 0.0932,
      "step": 10890
    },
    {
      "epoch": 1.2742803650830798,
      "loss_main": 0.05581757053732872,
      "step": 10890
    },
    {
      "epoch": 1.2754505031593728,
      "grad_norm": 0.3945891857147217,
      "learning_rate": 9.533594324803057e-05,
      "loss": 0.0847,
      "step": 10900
    },
    {
      "epoch": 1.2754505031593728,
      "loss_main": 0.08787349611520767,
      "step": 10900
    },
    {
      "epoch": 1.2766206412356658,
      "grad_norm": 0.2799796462059021,
      "learning_rate": 9.532431424512895e-05,
      "loss": 0.0674,
      "step": 10910
    },
    {
      "epoch": 1.2766206412356658,
      "loss_main": 0.12090670317411423,
      "step": 10910
    },
    {
      "epoch": 1.277790779311959,
      "grad_norm": 0.5705267190933228,
      "learning_rate": 9.531267147389741e-05,
      "loss": 0.091,
      "step": 10920
    },
    {
      "epoch": 1.277790779311959,
      "loss_main": 0.08235417306423187,
      "step": 10920
    },
    {
      "epoch": 1.2789609173882517,
      "grad_norm": 0.43417027592658997,
      "learning_rate": 9.530101493787266e-05,
      "loss": 0.0849,
      "step": 10930
    },
    {
      "epoch": 1.2789609173882517,
      "loss_main": 0.07770509272813797,
      "step": 10930
    },
    {
      "epoch": 1.2801310554645449,
      "grad_norm": 0.2974376082420349,
      "learning_rate": 9.528934464059571e-05,
      "loss": 0.0706,
      "step": 10940
    },
    {
      "epoch": 1.2801310554645449,
      "loss_main": 0.0707746148109436,
      "step": 10940
    },
    {
      "epoch": 1.2813011935408378,
      "grad_norm": 0.36209821701049805,
      "learning_rate": 9.527766058561163e-05,
      "loss": 0.0746,
      "step": 10950
    },
    {
      "epoch": 1.2813011935408378,
      "loss_main": 0.10325489938259125,
      "step": 10950
    },
    {
      "epoch": 1.2824713316171308,
      "grad_norm": 0.38404756784439087,
      "learning_rate": 9.526596277646976e-05,
      "loss": 0.0843,
      "step": 10960
    },
    {
      "epoch": 1.2824713316171308,
      "loss_main": 0.11827593296766281,
      "step": 10960
    },
    {
      "epoch": 1.2836414696934237,
      "grad_norm": 0.33591723442077637,
      "learning_rate": 9.525425121672358e-05,
      "loss": 0.0872,
      "step": 10970
    },
    {
      "epoch": 1.2836414696934237,
      "loss_main": 0.06615785509347916,
      "step": 10970
    },
    {
      "epoch": 1.2848116077697167,
      "grad_norm": 0.3091685473918915,
      "learning_rate": 9.524252590993074e-05,
      "loss": 0.0804,
      "step": 10980
    },
    {
      "epoch": 1.2848116077697167,
      "loss_main": 0.07224734127521515,
      "step": 10980
    },
    {
      "epoch": 1.2859817458460099,
      "grad_norm": 0.4539334774017334,
      "learning_rate": 9.523078685965309e-05,
      "loss": 0.079,
      "step": 10990
    },
    {
      "epoch": 1.2859817458460099,
      "loss_main": 0.0709071233868599,
      "step": 10990
    },
    {
      "epoch": 1.2871518839223028,
      "grad_norm": 0.28127095103263855,
      "learning_rate": 9.521903406945664e-05,
      "loss": 0.0809,
      "step": 11000
    },
    {
      "epoch": 1.2871518839223028,
      "loss_main": 0.08074706047773361,
      "step": 11000
    },
    {
      "epoch": 1.2883220219985958,
      "grad_norm": 0.32596355676651,
      "learning_rate": 9.520726754291158e-05,
      "loss": 0.0781,
      "step": 11010
    },
    {
      "epoch": 1.2883220219985958,
      "loss_main": 0.10057078301906586,
      "step": 11010
    },
    {
      "epoch": 1.2894921600748888,
      "grad_norm": 0.32412660121917725,
      "learning_rate": 9.519548728359227e-05,
      "loss": 0.0811,
      "step": 11020
    },
    {
      "epoch": 1.2894921600748888,
      "loss_main": 0.08851717412471771,
      "step": 11020
    },
    {
      "epoch": 1.2906622981511817,
      "grad_norm": 0.32600000500679016,
      "learning_rate": 9.518369329507726e-05,
      "loss": 0.0783,
      "step": 11030
    },
    {
      "epoch": 1.2906622981511817,
      "loss_main": 0.09185101091861725,
      "step": 11030
    },
    {
      "epoch": 1.291832436227475,
      "grad_norm": 0.40586504340171814,
      "learning_rate": 9.51718855809492e-05,
      "loss": 0.0789,
      "step": 11040
    },
    {
      "epoch": 1.291832436227475,
      "loss_main": 0.06055183336138725,
      "step": 11040
    },
    {
      "epoch": 1.2930025743037679,
      "grad_norm": 0.43024030327796936,
      "learning_rate": 9.516006414479502e-05,
      "loss": 0.0829,
      "step": 11050
    },
    {
      "epoch": 1.2930025743037679,
      "loss_main": 0.09495624154806137,
      "step": 11050
    },
    {
      "epoch": 1.2941727123800608,
      "grad_norm": 0.36958813667297363,
      "learning_rate": 9.514822899020572e-05,
      "loss": 0.0904,
      "step": 11060
    },
    {
      "epoch": 1.2941727123800608,
      "loss_main": 0.09522069990634918,
      "step": 11060
    },
    {
      "epoch": 1.2953428504563538,
      "grad_norm": 0.22232066094875336,
      "learning_rate": 9.513638012077654e-05,
      "loss": 0.0844,
      "step": 11070
    },
    {
      "epoch": 1.2953428504563538,
      "loss_main": 0.09974052011966705,
      "step": 11070
    },
    {
      "epoch": 1.2965129885326467,
      "grad_norm": 0.3074275255203247,
      "learning_rate": 9.512451754010683e-05,
      "loss": 0.0749,
      "step": 11080
    },
    {
      "epoch": 1.2965129885326467,
      "loss_main": 0.08358482271432877,
      "step": 11080
    },
    {
      "epoch": 1.29768312660894,
      "grad_norm": 0.3926510810852051,
      "learning_rate": 9.511264125180013e-05,
      "loss": 0.071,
      "step": 11090
    },
    {
      "epoch": 1.29768312660894,
      "loss_main": 0.0515255331993103,
      "step": 11090
    },
    {
      "epoch": 1.2988532646852329,
      "grad_norm": 0.30901971459388733,
      "learning_rate": 9.510075125946414e-05,
      "loss": 0.0716,
      "step": 11100
    },
    {
      "epoch": 1.2988532646852329,
      "loss_main": 0.07859621942043304,
      "step": 11100
    },
    {
      "epoch": 1.3000234027615258,
      "grad_norm": 0.29949307441711426,
      "learning_rate": 9.508884756671075e-05,
      "loss": 0.084,
      "step": 11110
    },
    {
      "epoch": 1.3000234027615258,
      "loss_main": 0.11054503172636032,
      "step": 11110
    },
    {
      "epoch": 1.3011935408378188,
      "grad_norm": 0.5303064584732056,
      "learning_rate": 9.507693017715596e-05,
      "loss": 0.0853,
      "step": 11120
    },
    {
      "epoch": 1.3011935408378188,
      "loss_main": 0.05429154634475708,
      "step": 11120
    },
    {
      "epoch": 1.3023636789141118,
      "grad_norm": 0.4346771836280823,
      "learning_rate": 9.506499909441997e-05,
      "loss": 0.0748,
      "step": 11130
    },
    {
      "epoch": 1.3023636789141118,
      "loss_main": 0.06554598361253738,
      "step": 11130
    },
    {
      "epoch": 1.303533816990405,
      "grad_norm": 0.33902573585510254,
      "learning_rate": 9.505305432212713e-05,
      "loss": 0.0856,
      "step": 11140
    },
    {
      "epoch": 1.303533816990405,
      "loss_main": 0.06552087515592575,
      "step": 11140
    },
    {
      "epoch": 1.304703955066698,
      "grad_norm": 0.28870564699172974,
      "learning_rate": 9.504109586390595e-05,
      "loss": 0.079,
      "step": 11150
    },
    {
      "epoch": 1.304703955066698,
      "loss_main": 0.047415681183338165,
      "step": 11150
    },
    {
      "epoch": 1.3058740931429909,
      "grad_norm": 0.36936166882514954,
      "learning_rate": 9.502912372338908e-05,
      "loss": 0.0864,
      "step": 11160
    },
    {
      "epoch": 1.3058740931429909,
      "loss_main": 0.07291748374700546,
      "step": 11160
    },
    {
      "epoch": 1.3070442312192838,
      "grad_norm": 0.32840922474861145,
      "learning_rate": 9.501713790421335e-05,
      "loss": 0.0801,
      "step": 11170
    },
    {
      "epoch": 1.3070442312192838,
      "loss_main": 0.08027023077011108,
      "step": 11170
    },
    {
      "epoch": 1.3082143692955768,
      "grad_norm": 0.27313679456710815,
      "learning_rate": 9.500513841001974e-05,
      "loss": 0.0819,
      "step": 11180
    },
    {
      "epoch": 1.3082143692955768,
      "loss_main": 0.07003388553857803,
      "step": 11180
    },
    {
      "epoch": 1.30938450737187,
      "grad_norm": 0.3113323450088501,
      "learning_rate": 9.499312524445336e-05,
      "loss": 0.0875,
      "step": 11190
    },
    {
      "epoch": 1.30938450737187,
      "loss_main": 0.06567498296499252,
      "step": 11190
    },
    {
      "epoch": 1.310554645448163,
      "grad_norm": 0.312971293926239,
      "learning_rate": 9.498109841116351e-05,
      "loss": 0.0759,
      "step": 11200
    },
    {
      "epoch": 1.310554645448163,
      "loss_main": 0.060877420008182526,
      "step": 11200
    },
    {
      "epoch": 1.3117247835244559,
      "grad_norm": 0.4137791097164154,
      "learning_rate": 9.496905791380363e-05,
      "loss": 0.092,
      "step": 11210
    },
    {
      "epoch": 1.3117247835244559,
      "loss_main": 0.04602893069386482,
      "step": 11210
    },
    {
      "epoch": 1.3128949216007488,
      "grad_norm": 0.47215309739112854,
      "learning_rate": 9.495700375603129e-05,
      "loss": 0.074,
      "step": 11220
    },
    {
      "epoch": 1.3128949216007488,
      "loss_main": 0.08144427090883255,
      "step": 11220
    },
    {
      "epoch": 1.3140650596770418,
      "grad_norm": 0.3483511507511139,
      "learning_rate": 9.494493594150822e-05,
      "loss": 0.0899,
      "step": 11230
    },
    {
      "epoch": 1.3140650596770418,
      "loss_main": 0.03419310227036476,
      "step": 11230
    },
    {
      "epoch": 1.315235197753335,
      "grad_norm": 0.34147629141807556,
      "learning_rate": 9.493285447390032e-05,
      "loss": 0.0781,
      "step": 11240
    },
    {
      "epoch": 1.315235197753335,
      "loss_main": 0.10584381967782974,
      "step": 11240
    },
    {
      "epoch": 1.316405335829628,
      "grad_norm": 0.2985430359840393,
      "learning_rate": 9.492075935687761e-05,
      "loss": 0.0739,
      "step": 11250
    },
    {
      "epoch": 1.316405335829628,
      "loss_main": 0.06321863830089569,
      "step": 11250
    },
    {
      "epoch": 1.317575473905921,
      "grad_norm": 0.3065308928489685,
      "learning_rate": 9.490865059411427e-05,
      "loss": 0.0783,
      "step": 11260
    },
    {
      "epoch": 1.317575473905921,
      "loss_main": 0.12147264927625656,
      "step": 11260
    },
    {
      "epoch": 1.3187456119822139,
      "grad_norm": 0.21487261354923248,
      "learning_rate": 9.489652818928863e-05,
      "loss": 0.0752,
      "step": 11270
    },
    {
      "epoch": 1.3187456119822139,
      "loss_main": 0.11144177615642548,
      "step": 11270
    },
    {
      "epoch": 1.3199157500585068,
      "grad_norm": 0.2425507754087448,
      "learning_rate": 9.488439214608315e-05,
      "loss": 0.0888,
      "step": 11280
    },
    {
      "epoch": 1.3199157500585068,
      "loss_main": 0.06902331858873367,
      "step": 11280
    },
    {
      "epoch": 1.3210858881348,
      "grad_norm": 0.2606394290924072,
      "learning_rate": 9.487224246818444e-05,
      "loss": 0.0765,
      "step": 11290
    },
    {
      "epoch": 1.3210858881348,
      "loss_main": 0.07380832731723785,
      "step": 11290
    },
    {
      "epoch": 1.322256026211093,
      "grad_norm": 0.3599680960178375,
      "learning_rate": 9.486007915928325e-05,
      "loss": 0.0807,
      "step": 11300
    },
    {
      "epoch": 1.322256026211093,
      "loss_main": 0.07329720258712769,
      "step": 11300
    },
    {
      "epoch": 1.323426164287386,
      "grad_norm": 0.34438595175743103,
      "learning_rate": 9.484790222307448e-05,
      "loss": 0.0784,
      "step": 11310
    },
    {
      "epoch": 1.323426164287386,
      "loss_main": 0.07583785057067871,
      "step": 11310
    },
    {
      "epoch": 1.3245963023636789,
      "grad_norm": 0.3211688995361328,
      "learning_rate": 9.483571166325716e-05,
      "loss": 0.0949,
      "step": 11320
    },
    {
      "epoch": 1.3245963023636789,
      "loss_main": 0.12353110313415527,
      "step": 11320
    },
    {
      "epoch": 1.3257664404399718,
      "grad_norm": 0.4203760325908661,
      "learning_rate": 9.482350748353444e-05,
      "loss": 0.0878,
      "step": 11330
    },
    {
      "epoch": 1.3257664404399718,
      "loss_main": 0.09221092611551285,
      "step": 11330
    },
    {
      "epoch": 1.326936578516265,
      "grad_norm": 0.3741377890110016,
      "learning_rate": 9.481128968761363e-05,
      "loss": 0.0876,
      "step": 11340
    },
    {
      "epoch": 1.326936578516265,
      "loss_main": 0.0745023638010025,
      "step": 11340
    },
    {
      "epoch": 1.328106716592558,
      "grad_norm": 0.3552339971065521,
      "learning_rate": 9.479905827920621e-05,
      "loss": 0.08,
      "step": 11350
    },
    {
      "epoch": 1.328106716592558,
      "loss_main": 0.059879694133996964,
      "step": 11350
    },
    {
      "epoch": 1.329276854668851,
      "grad_norm": 0.28408268094062805,
      "learning_rate": 9.478681326202773e-05,
      "loss": 0.0879,
      "step": 11360
    },
    {
      "epoch": 1.329276854668851,
      "loss_main": 0.0778169110417366,
      "step": 11360
    },
    {
      "epoch": 1.3304469927451439,
      "grad_norm": 0.30329635739326477,
      "learning_rate": 9.477455463979791e-05,
      "loss": 0.0801,
      "step": 11370
    },
    {
      "epoch": 1.3304469927451439,
      "loss_main": 0.11988116800785065,
      "step": 11370
    },
    {
      "epoch": 1.3316171308214368,
      "grad_norm": 0.32524576783180237,
      "learning_rate": 9.476228241624059e-05,
      "loss": 0.098,
      "step": 11380
    },
    {
      "epoch": 1.3316171308214368,
      "loss_main": 0.047334857285022736,
      "step": 11380
    },
    {
      "epoch": 1.33278726889773,
      "grad_norm": 0.2163456529378891,
      "learning_rate": 9.474999659508374e-05,
      "loss": 0.0791,
      "step": 11390
    },
    {
      "epoch": 1.33278726889773,
      "loss_main": 0.04701199382543564,
      "step": 11390
    },
    {
      "epoch": 1.333957406974023,
      "grad_norm": 0.3466559946537018,
      "learning_rate": 9.47376971800595e-05,
      "loss": 0.0715,
      "step": 11400
    },
    {
      "epoch": 1.333957406974023,
      "loss_main": 0.06245904043316841,
      "step": 11400
    },
    {
      "epoch": 1.335127545050316,
      "grad_norm": 0.3550366461277008,
      "learning_rate": 9.472538417490409e-05,
      "loss": 0.0892,
      "step": 11410
    },
    {
      "epoch": 1.335127545050316,
      "loss_main": 0.08902279287576675,
      "step": 11410
    },
    {
      "epoch": 1.336297683126609,
      "grad_norm": 0.26390549540519714,
      "learning_rate": 9.471305758335784e-05,
      "loss": 0.0811,
      "step": 11420
    },
    {
      "epoch": 1.336297683126609,
      "loss_main": 0.05057279393076897,
      "step": 11420
    },
    {
      "epoch": 1.3374678212029019,
      "grad_norm": 0.44347888231277466,
      "learning_rate": 9.47007174091653e-05,
      "loss": 0.0841,
      "step": 11430
    },
    {
      "epoch": 1.3374678212029019,
      "loss_main": 0.08363992720842361,
      "step": 11430
    },
    {
      "epoch": 1.338637959279195,
      "grad_norm": 0.2708256244659424,
      "learning_rate": 9.468836365607507e-05,
      "loss": 0.0828,
      "step": 11440
    },
    {
      "epoch": 1.338637959279195,
      "loss_main": 0.12195420265197754,
      "step": 11440
    },
    {
      "epoch": 1.339808097355488,
      "grad_norm": 0.30860236287117004,
      "learning_rate": 9.467599632783988e-05,
      "loss": 0.0915,
      "step": 11450
    },
    {
      "epoch": 1.339808097355488,
      "loss_main": 0.10134709626436234,
      "step": 11450
    },
    {
      "epoch": 1.340978235431781,
      "grad_norm": 0.45158621668815613,
      "learning_rate": 9.466361542821662e-05,
      "loss": 0.0824,
      "step": 11460
    },
    {
      "epoch": 1.340978235431781,
      "loss_main": 0.07637546956539154,
      "step": 11460
    },
    {
      "epoch": 1.342148373508074,
      "grad_norm": 0.3456229269504547,
      "learning_rate": 9.465122096096625e-05,
      "loss": 0.0751,
      "step": 11470
    },
    {
      "epoch": 1.342148373508074,
      "loss_main": 0.07864809036254883,
      "step": 11470
    },
    {
      "epoch": 1.3433185115843669,
      "grad_norm": 0.3726896345615387,
      "learning_rate": 9.463881292985391e-05,
      "loss": 0.0834,
      "step": 11480
    },
    {
      "epoch": 1.3433185115843669,
      "loss_main": 0.08611298352479935,
      "step": 11480
    },
    {
      "epoch": 1.34448864966066,
      "grad_norm": 0.3101242184638977,
      "learning_rate": 9.462639133864881e-05,
      "loss": 0.0814,
      "step": 11490
    },
    {
      "epoch": 1.34448864966066,
      "loss_main": 0.08012209832668304,
      "step": 11490
    },
    {
      "epoch": 1.345658787736953,
      "grad_norm": 0.3620348572731018,
      "learning_rate": 9.461395619112432e-05,
      "loss": 0.0798,
      "step": 11500
    },
    {
      "epoch": 1.345658787736953,
      "loss_main": 0.07273471355438232,
      "step": 11500
    },
    {
      "epoch": 1.346828925813246,
      "grad_norm": 0.2701055407524109,
      "learning_rate": 9.460150749105791e-05,
      "loss": 0.0779,
      "step": 11510
    },
    {
      "epoch": 1.346828925813246,
      "loss_main": 0.0934501364827156,
      "step": 11510
    },
    {
      "epoch": 1.347999063889539,
      "grad_norm": 0.23590385913848877,
      "learning_rate": 9.458904524223116e-05,
      "loss": 0.0783,
      "step": 11520
    },
    {
      "epoch": 1.347999063889539,
      "loss_main": 0.09579426795244217,
      "step": 11520
    },
    {
      "epoch": 1.349169201965832,
      "grad_norm": 0.47681766748428345,
      "learning_rate": 9.457656944842976e-05,
      "loss": 0.09,
      "step": 11530
    },
    {
      "epoch": 1.349169201965832,
      "loss_main": 0.09866025298833847,
      "step": 11530
    },
    {
      "epoch": 1.350339340042125,
      "grad_norm": 0.4018368721008301,
      "learning_rate": 9.456408011344353e-05,
      "loss": 0.0814,
      "step": 11540
    },
    {
      "epoch": 1.350339340042125,
      "loss_main": 0.0741846039891243,
      "step": 11540
    },
    {
      "epoch": 1.351509478118418,
      "grad_norm": 0.2502049207687378,
      "learning_rate": 9.455157724106643e-05,
      "loss": 0.0804,
      "step": 11550
    },
    {
      "epoch": 1.351509478118418,
      "loss_main": 0.14263491332530975,
      "step": 11550
    },
    {
      "epoch": 1.352679616194711,
      "grad_norm": 0.3735528886318207,
      "learning_rate": 9.453906083509647e-05,
      "loss": 0.0944,
      "step": 11560
    },
    {
      "epoch": 1.352679616194711,
      "loss_main": 0.0687074288725853,
      "step": 11560
    },
    {
      "epoch": 1.353849754271004,
      "grad_norm": 0.23882652819156647,
      "learning_rate": 9.45265308993358e-05,
      "loss": 0.0715,
      "step": 11570
    },
    {
      "epoch": 1.353849754271004,
      "loss_main": 0.09470678865909576,
      "step": 11570
    },
    {
      "epoch": 1.355019892347297,
      "grad_norm": 0.3616156280040741,
      "learning_rate": 9.451398743759071e-05,
      "loss": 0.0849,
      "step": 11580
    },
    {
      "epoch": 1.355019892347297,
      "loss_main": 0.06724584847688675,
      "step": 11580
    },
    {
      "epoch": 1.35619003042359,
      "grad_norm": 0.33043062686920166,
      "learning_rate": 9.450143045367156e-05,
      "loss": 0.0777,
      "step": 11590
    },
    {
      "epoch": 1.35619003042359,
      "loss_main": 0.1148587092757225,
      "step": 11590
    },
    {
      "epoch": 1.357360168499883,
      "grad_norm": 0.2696019411087036,
      "learning_rate": 9.448885995139283e-05,
      "loss": 0.0802,
      "step": 11600
    },
    {
      "epoch": 1.357360168499883,
      "loss_main": 0.09607317298650742,
      "step": 11600
    },
    {
      "epoch": 1.358530306576176,
      "grad_norm": 0.2725173234939575,
      "learning_rate": 9.44762759345731e-05,
      "loss": 0.087,
      "step": 11610
    },
    {
      "epoch": 1.358530306576176,
      "loss_main": 0.07347021996974945,
      "step": 11610
    },
    {
      "epoch": 1.359700444652469,
      "grad_norm": 0.34243226051330566,
      "learning_rate": 9.446367840703509e-05,
      "loss": 0.074,
      "step": 11620
    },
    {
      "epoch": 1.359700444652469,
      "loss_main": 0.060120273381471634,
      "step": 11620
    },
    {
      "epoch": 1.360870582728762,
      "grad_norm": 0.30456167459487915,
      "learning_rate": 9.445106737260556e-05,
      "loss": 0.0725,
      "step": 11630
    },
    {
      "epoch": 1.360870582728762,
      "loss_main": 0.0830492228269577,
      "step": 11630
    },
    {
      "epoch": 1.3620407208050551,
      "grad_norm": 0.3318902552127838,
      "learning_rate": 9.443844283511543e-05,
      "loss": 0.0823,
      "step": 11640
    },
    {
      "epoch": 1.3620407208050551,
      "loss_main": 0.06565884500741959,
      "step": 11640
    },
    {
      "epoch": 1.363210858881348,
      "grad_norm": 0.37460243701934814,
      "learning_rate": 9.442580479839968e-05,
      "loss": 0.0922,
      "step": 11650
    },
    {
      "epoch": 1.363210858881348,
      "loss_main": 0.06689434498548508,
      "step": 11650
    },
    {
      "epoch": 1.364380996957641,
      "grad_norm": 0.3285318911075592,
      "learning_rate": 9.441315326629745e-05,
      "loss": 0.0879,
      "step": 11660
    },
    {
      "epoch": 1.364380996957641,
      "loss_main": 0.09641623497009277,
      "step": 11660
    },
    {
      "epoch": 1.365551135033934,
      "grad_norm": 0.4041946232318878,
      "learning_rate": 9.44004882426519e-05,
      "loss": 0.0811,
      "step": 11670
    },
    {
      "epoch": 1.365551135033934,
      "loss_main": 0.07195798307657242,
      "step": 11670
    },
    {
      "epoch": 1.366721273110227,
      "grad_norm": 0.3440130054950714,
      "learning_rate": 9.438780973131037e-05,
      "loss": 0.0742,
      "step": 11680
    },
    {
      "epoch": 1.366721273110227,
      "loss_main": 0.11863554269075394,
      "step": 11680
    },
    {
      "epoch": 1.3678914111865201,
      "grad_norm": 0.3473372161388397,
      "learning_rate": 9.437511773612423e-05,
      "loss": 0.0849,
      "step": 11690
    },
    {
      "epoch": 1.3678914111865201,
      "loss_main": 0.1066465750336647,
      "step": 11690
    },
    {
      "epoch": 1.369061549262813,
      "grad_norm": 0.35190823674201965,
      "learning_rate": 9.436241226094896e-05,
      "loss": 0.0966,
      "step": 11700
    },
    {
      "epoch": 1.369061549262813,
      "loss_main": 0.13100062310695648,
      "step": 11700
    },
    {
      "epoch": 1.370231687339106,
      "grad_norm": 0.40643310546875,
      "learning_rate": 9.434969330964418e-05,
      "loss": 0.0839,
      "step": 11710
    },
    {
      "epoch": 1.370231687339106,
      "loss_main": 0.10175817459821701,
      "step": 11710
    },
    {
      "epoch": 1.371401825415399,
      "grad_norm": 0.4275180399417877,
      "learning_rate": 9.433696088607356e-05,
      "loss": 0.088,
      "step": 11720
    },
    {
      "epoch": 1.371401825415399,
      "loss_main": 0.06539548188447952,
      "step": 11720
    },
    {
      "epoch": 1.372571963491692,
      "grad_norm": 0.3951192796230316,
      "learning_rate": 9.432421499410486e-05,
      "loss": 0.081,
      "step": 11730
    },
    {
      "epoch": 1.372571963491692,
      "loss_main": 0.09298598766326904,
      "step": 11730
    },
    {
      "epoch": 1.3737421015679852,
      "grad_norm": 0.38326767086982727,
      "learning_rate": 9.431145563760998e-05,
      "loss": 0.0781,
      "step": 11740
    },
    {
      "epoch": 1.3737421015679852,
      "loss_main": 0.09547094255685806,
      "step": 11740
    },
    {
      "epoch": 1.374912239644278,
      "grad_norm": 0.3679232597351074,
      "learning_rate": 9.429868282046484e-05,
      "loss": 0.0887,
      "step": 11750
    },
    {
      "epoch": 1.374912239644278,
      "loss_main": 0.06766226142644882,
      "step": 11750
    },
    {
      "epoch": 1.376082377720571,
      "grad_norm": 0.3883298337459564,
      "learning_rate": 9.428589654654951e-05,
      "loss": 0.0981,
      "step": 11760
    },
    {
      "epoch": 1.376082377720571,
      "loss_main": 0.09378385543823242,
      "step": 11760
    },
    {
      "epoch": 1.377252515796864,
      "grad_norm": 0.3751083016395569,
      "learning_rate": 9.42730968197481e-05,
      "loss": 0.0832,
      "step": 11770
    },
    {
      "epoch": 1.377252515796864,
      "loss_main": 0.0548376701772213,
      "step": 11770
    },
    {
      "epoch": 1.378422653873157,
      "grad_norm": 0.46906137466430664,
      "learning_rate": 9.426028364394883e-05,
      "loss": 0.0817,
      "step": 11780
    },
    {
      "epoch": 1.378422653873157,
      "loss_main": 0.0884624570608139,
      "step": 11780
    },
    {
      "epoch": 1.3795927919494502,
      "grad_norm": 0.3551075756549835,
      "learning_rate": 9.424745702304402e-05,
      "loss": 0.0774,
      "step": 11790
    },
    {
      "epoch": 1.3795927919494502,
      "loss_main": 0.08800764381885529,
      "step": 11790
    },
    {
      "epoch": 1.380762930025743,
      "grad_norm": 0.46627315878868103,
      "learning_rate": 9.423461696093006e-05,
      "loss": 0.0851,
      "step": 11800
    },
    {
      "epoch": 1.380762930025743,
      "loss_main": 0.11468875408172607,
      "step": 11800
    },
    {
      "epoch": 1.381933068102036,
      "grad_norm": 0.38040462136268616,
      "learning_rate": 9.422176346150741e-05,
      "loss": 0.0863,
      "step": 11810
    },
    {
      "epoch": 1.381933068102036,
      "loss_main": 0.08459067344665527,
      "step": 11810
    },
    {
      "epoch": 1.383103206178329,
      "grad_norm": 0.45756015181541443,
      "learning_rate": 9.420889652868063e-05,
      "loss": 0.0826,
      "step": 11820
    },
    {
      "epoch": 1.383103206178329,
      "loss_main": 0.0579722635447979,
      "step": 11820
    },
    {
      "epoch": 1.384273344254622,
      "grad_norm": 0.3830031156539917,
      "learning_rate": 9.419601616635836e-05,
      "loss": 0.0856,
      "step": 11830
    },
    {
      "epoch": 1.384273344254622,
      "loss_main": 0.12615028023719788,
      "step": 11830
    },
    {
      "epoch": 1.385443482330915,
      "grad_norm": 0.24847173690795898,
      "learning_rate": 9.418312237845331e-05,
      "loss": 0.0923,
      "step": 11840
    },
    {
      "epoch": 1.385443482330915,
      "loss_main": 0.05062909796833992,
      "step": 11840
    },
    {
      "epoch": 1.386613620407208,
      "grad_norm": 0.27792781591415405,
      "learning_rate": 9.417021516888225e-05,
      "loss": 0.0763,
      "step": 11850
    },
    {
      "epoch": 1.386613620407208,
      "loss_main": 0.07268667966127396,
      "step": 11850
    },
    {
      "epoch": 1.387783758483501,
      "grad_norm": 0.4312760829925537,
      "learning_rate": 9.415729454156608e-05,
      "loss": 0.0756,
      "step": 11860
    },
    {
      "epoch": 1.387783758483501,
      "loss_main": 0.07162361592054367,
      "step": 11860
    },
    {
      "epoch": 1.388953896559794,
      "grad_norm": 0.37530356645584106,
      "learning_rate": 9.414436050042973e-05,
      "loss": 0.0746,
      "step": 11870
    },
    {
      "epoch": 1.388953896559794,
      "loss_main": 0.0909799337387085,
      "step": 11870
    },
    {
      "epoch": 1.390124034636087,
      "grad_norm": 0.3097376525402069,
      "learning_rate": 9.413141304940223e-05,
      "loss": 0.0794,
      "step": 11880
    },
    {
      "epoch": 1.390124034636087,
      "loss_main": 0.06669473648071289,
      "step": 11880
    },
    {
      "epoch": 1.39129417271238,
      "grad_norm": 0.4286404848098755,
      "learning_rate": 9.411845219241666e-05,
      "loss": 0.0818,
      "step": 11890
    },
    {
      "epoch": 1.39129417271238,
      "loss_main": 0.03382161632180214,
      "step": 11890
    },
    {
      "epoch": 1.392464310788673,
      "grad_norm": 0.30609995126724243,
      "learning_rate": 9.410547793341021e-05,
      "loss": 0.0875,
      "step": 11900
    },
    {
      "epoch": 1.392464310788673,
      "loss_main": 0.07755698263645172,
      "step": 11900
    },
    {
      "epoch": 1.3936344488649661,
      "grad_norm": 0.3776889741420746,
      "learning_rate": 9.409249027632408e-05,
      "loss": 0.0728,
      "step": 11910
    },
    {
      "epoch": 1.3936344488649661,
      "loss_main": 0.0655476376414299,
      "step": 11910
    },
    {
      "epoch": 1.394804586941259,
      "grad_norm": 0.3511098027229309,
      "learning_rate": 9.407948922510362e-05,
      "loss": 0.0793,
      "step": 11920
    },
    {
      "epoch": 1.394804586941259,
      "loss_main": 0.0884392186999321,
      "step": 11920
    },
    {
      "epoch": 1.395974725017552,
      "grad_norm": 0.3372541666030884,
      "learning_rate": 9.406647478369817e-05,
      "loss": 0.0841,
      "step": 11930
    },
    {
      "epoch": 1.395974725017552,
      "loss_main": 0.0667743906378746,
      "step": 11930
    },
    {
      "epoch": 1.397144863093845,
      "grad_norm": 0.30123084783554077,
      "learning_rate": 9.405344695606118e-05,
      "loss": 0.0786,
      "step": 11940
    },
    {
      "epoch": 1.397144863093845,
      "loss_main": 0.07182254642248154,
      "step": 11940
    },
    {
      "epoch": 1.398315001170138,
      "grad_norm": 0.3094192445278168,
      "learning_rate": 9.404040574615018e-05,
      "loss": 0.0788,
      "step": 11950
    },
    {
      "epoch": 1.398315001170138,
      "loss_main": 0.10286872088909149,
      "step": 11950
    },
    {
      "epoch": 1.3994851392464311,
      "grad_norm": 0.37025749683380127,
      "learning_rate": 9.402735115792674e-05,
      "loss": 0.0837,
      "step": 11960
    },
    {
      "epoch": 1.3994851392464311,
      "loss_main": 0.06472192704677582,
      "step": 11960
    },
    {
      "epoch": 1.400655277322724,
      "grad_norm": 0.3810654282569885,
      "learning_rate": 9.401428319535649e-05,
      "loss": 0.0785,
      "step": 11970
    },
    {
      "epoch": 1.400655277322724,
      "loss_main": 0.0481087826192379,
      "step": 11970
    },
    {
      "epoch": 1.401825415399017,
      "grad_norm": 0.298183411359787,
      "learning_rate": 9.400120186240912e-05,
      "loss": 0.0797,
      "step": 11980
    },
    {
      "epoch": 1.401825415399017,
      "loss_main": 0.08828362077474594,
      "step": 11980
    },
    {
      "epoch": 1.40299555347531,
      "grad_norm": 0.4299525320529938,
      "learning_rate": 9.398810716305844e-05,
      "loss": 0.0773,
      "step": 11990
    },
    {
      "epoch": 1.40299555347531,
      "loss_main": 0.09055633842945099,
      "step": 11990
    },
    {
      "epoch": 1.404165691551603,
      "grad_norm": 0.31470170617103577,
      "learning_rate": 9.397499910128222e-05,
      "loss": 0.0805,
      "step": 12000
    },
    {
      "epoch": 1.404165691551603,
      "loss_main": 0.08575879782438278,
      "step": 12000
    },
    {
      "epoch": 1.4053358296278962,
      "grad_norm": 0.2841171622276306,
      "learning_rate": 9.396187768106237e-05,
      "loss": 0.0713,
      "step": 12010
    },
    {
      "epoch": 1.4053358296278962,
      "loss_main": 0.07273656129837036,
      "step": 12010
    },
    {
      "epoch": 1.4065059677041891,
      "grad_norm": 0.33079811930656433,
      "learning_rate": 9.394874290638482e-05,
      "loss": 0.0746,
      "step": 12020
    },
    {
      "epoch": 1.4065059677041891,
      "loss_main": 0.07148949056863785,
      "step": 12020
    },
    {
      "epoch": 1.407676105780482,
      "grad_norm": 0.36773425340652466,
      "learning_rate": 9.393559478123959e-05,
      "loss": 0.0857,
      "step": 12030
    },
    {
      "epoch": 1.407676105780482,
      "loss_main": 0.07410059869289398,
      "step": 12030
    },
    {
      "epoch": 1.408846243856775,
      "grad_norm": 0.3583264946937561,
      "learning_rate": 9.39224333096207e-05,
      "loss": 0.0777,
      "step": 12040
    },
    {
      "epoch": 1.408846243856775,
      "loss_main": 0.05386306717991829,
      "step": 12040
    },
    {
      "epoch": 1.410016381933068,
      "grad_norm": 0.237216979265213,
      "learning_rate": 9.390925849552629e-05,
      "loss": 0.0807,
      "step": 12050
    },
    {
      "epoch": 1.410016381933068,
      "loss_main": 0.07453637570142746,
      "step": 12050
    },
    {
      "epoch": 1.4111865200093612,
      "grad_norm": 0.3070715069770813,
      "learning_rate": 9.389607034295849e-05,
      "loss": 0.0763,
      "step": 12060
    },
    {
      "epoch": 1.4111865200093612,
      "loss_main": 0.07376269996166229,
      "step": 12060
    },
    {
      "epoch": 1.4123566580856541,
      "grad_norm": 0.2533242702484131,
      "learning_rate": 9.388286885592355e-05,
      "loss": 0.091,
      "step": 12070
    },
    {
      "epoch": 1.4123566580856541,
      "loss_main": 0.07017259299755096,
      "step": 12070
    },
    {
      "epoch": 1.413526796161947,
      "grad_norm": 0.332045316696167,
      "learning_rate": 9.386965403843168e-05,
      "loss": 0.0725,
      "step": 12080
    },
    {
      "epoch": 1.413526796161947,
      "loss_main": 0.06914065033197403,
      "step": 12080
    },
    {
      "epoch": 1.41469693423824,
      "grad_norm": 0.29487332701683044,
      "learning_rate": 9.385642589449726e-05,
      "loss": 0.0757,
      "step": 12090
    },
    {
      "epoch": 1.41469693423824,
      "loss_main": 0.04191816598176956,
      "step": 12090
    },
    {
      "epoch": 1.415867072314533,
      "grad_norm": 0.41278743743896484,
      "learning_rate": 9.38431844281386e-05,
      "loss": 0.072,
      "step": 12100
    },
    {
      "epoch": 1.415867072314533,
      "loss_main": 0.07141920179128647,
      "step": 12100
    },
    {
      "epoch": 1.4170372103908262,
      "grad_norm": 0.3175436556339264,
      "learning_rate": 9.38299296433781e-05,
      "loss": 0.0782,
      "step": 12110
    },
    {
      "epoch": 1.4170372103908262,
      "loss_main": 0.08978183567523956,
      "step": 12110
    },
    {
      "epoch": 1.4182073484671192,
      "grad_norm": 0.49427711963653564,
      "learning_rate": 9.381666154424226e-05,
      "loss": 0.0885,
      "step": 12120
    },
    {
      "epoch": 1.4182073484671192,
      "loss_main": 0.08815404027700424,
      "step": 12120
    },
    {
      "epoch": 1.4193774865434121,
      "grad_norm": 0.34293392300605774,
      "learning_rate": 9.380338013476157e-05,
      "loss": 0.0842,
      "step": 12130
    },
    {
      "epoch": 1.4193774865434121,
      "loss_main": 0.08873222768306732,
      "step": 12130
    },
    {
      "epoch": 1.420547624619705,
      "grad_norm": 0.34304529428482056,
      "learning_rate": 9.379008541897054e-05,
      "loss": 0.078,
      "step": 12140
    },
    {
      "epoch": 1.420547624619705,
      "loss_main": 0.12209022045135498,
      "step": 12140
    },
    {
      "epoch": 1.421717762695998,
      "grad_norm": 0.7356472611427307,
      "learning_rate": 9.377677740090777e-05,
      "loss": 0.0988,
      "step": 12150
    },
    {
      "epoch": 1.421717762695998,
      "loss_main": 0.04488757625222206,
      "step": 12150
    },
    {
      "epoch": 1.4228879007722912,
      "grad_norm": 0.44695961475372314,
      "learning_rate": 9.376345608461588e-05,
      "loss": 0.0829,
      "step": 12160
    },
    {
      "epoch": 1.4228879007722912,
      "loss_main": 0.0676250010728836,
      "step": 12160
    },
    {
      "epoch": 1.4240580388485842,
      "grad_norm": 0.47468841075897217,
      "learning_rate": 9.375012147414155e-05,
      "loss": 0.0773,
      "step": 12170
    },
    {
      "epoch": 1.4240580388485842,
      "loss_main": 0.11920314282178879,
      "step": 12170
    },
    {
      "epoch": 1.4252281769248771,
      "grad_norm": 0.3381766080856323,
      "learning_rate": 9.373677357353545e-05,
      "loss": 0.0886,
      "step": 12180
    },
    {
      "epoch": 1.4252281769248771,
      "loss_main": 0.09061522036790848,
      "step": 12180
    },
    {
      "epoch": 1.42639831500117,
      "grad_norm": 0.369912713766098,
      "learning_rate": 9.372341238685237e-05,
      "loss": 0.0708,
      "step": 12190
    },
    {
      "epoch": 1.42639831500117,
      "loss_main": 0.08149819821119308,
      "step": 12190
    },
    {
      "epoch": 1.427568453077463,
      "grad_norm": 0.3387299180030823,
      "learning_rate": 9.371003791815102e-05,
      "loss": 0.0787,
      "step": 12200
    },
    {
      "epoch": 1.427568453077463,
      "loss_main": 0.07422123104333878,
      "step": 12200
    },
    {
      "epoch": 1.4287385911537562,
      "grad_norm": 0.22646138072013855,
      "learning_rate": 9.369665017149429e-05,
      "loss": 0.0791,
      "step": 12210
    },
    {
      "epoch": 1.4287385911537562,
      "loss_main": 0.08122234791517258,
      "step": 12210
    },
    {
      "epoch": 1.4299087292300492,
      "grad_norm": 0.2487749457359314,
      "learning_rate": 9.368324915094895e-05,
      "loss": 0.0869,
      "step": 12220
    },
    {
      "epoch": 1.4299087292300492,
      "loss_main": 0.10393116623163223,
      "step": 12220
    },
    {
      "epoch": 1.4310788673063422,
      "grad_norm": 0.31621354818344116,
      "learning_rate": 9.366983486058591e-05,
      "loss": 0.0829,
      "step": 12230
    },
    {
      "epoch": 1.4310788673063422,
      "loss_main": 0.07502766698598862,
      "step": 12230
    },
    {
      "epoch": 1.432249005382635,
      "grad_norm": 0.3632882237434387,
      "learning_rate": 9.365640730448009e-05,
      "loss": 0.0835,
      "step": 12240
    },
    {
      "epoch": 1.432249005382635,
      "loss_main": 0.06813301146030426,
      "step": 12240
    },
    {
      "epoch": 1.433419143458928,
      "grad_norm": 0.33362212777137756,
      "learning_rate": 9.36429664867104e-05,
      "loss": 0.0744,
      "step": 12250
    },
    {
      "epoch": 1.433419143458928,
      "loss_main": 0.05784529075026512,
      "step": 12250
    },
    {
      "epoch": 1.4345892815352213,
      "grad_norm": 0.41398710012435913,
      "learning_rate": 9.362951241135982e-05,
      "loss": 0.0807,
      "step": 12260
    },
    {
      "epoch": 1.4345892815352213,
      "loss_main": 0.07423316687345505,
      "step": 12260
    },
    {
      "epoch": 1.4357594196115142,
      "grad_norm": 0.3085651397705078,
      "learning_rate": 9.361604508251534e-05,
      "loss": 0.065,
      "step": 12270
    },
    {
      "epoch": 1.4357594196115142,
      "loss_main": 0.10138886421918869,
      "step": 12270
    },
    {
      "epoch": 1.4369295576878072,
      "grad_norm": 0.34313300251960754,
      "learning_rate": 9.360256450426799e-05,
      "loss": 0.073,
      "step": 12280
    },
    {
      "epoch": 1.4369295576878072,
      "loss_main": 0.09761692583560944,
      "step": 12280
    },
    {
      "epoch": 1.4380996957641001,
      "grad_norm": 0.27617794275283813,
      "learning_rate": 9.358907068071279e-05,
      "loss": 0.0719,
      "step": 12290
    },
    {
      "epoch": 1.4380996957641001,
      "loss_main": 0.06681778281927109,
      "step": 12290
    },
    {
      "epoch": 1.439269833840393,
      "grad_norm": 0.2967992424964905,
      "learning_rate": 9.357556361594882e-05,
      "loss": 0.0745,
      "step": 12300
    },
    {
      "epoch": 1.439269833840393,
      "loss_main": 0.05987270548939705,
      "step": 12300
    },
    {
      "epoch": 1.4404399719166863,
      "grad_norm": 0.3536551296710968,
      "learning_rate": 9.356204331407917e-05,
      "loss": 0.0735,
      "step": 12310
    },
    {
      "epoch": 1.4404399719166863,
      "loss_main": 0.12469247728586197,
      "step": 12310
    },
    {
      "epoch": 1.4416101099929792,
      "grad_norm": 0.31777605414390564,
      "learning_rate": 9.354850977921094e-05,
      "loss": 0.0875,
      "step": 12320
    },
    {
      "epoch": 1.4416101099929792,
      "loss_main": 0.1082678884267807,
      "step": 12320
    },
    {
      "epoch": 1.4427802480692722,
      "grad_norm": 0.33780160546302795,
      "learning_rate": 9.353496301545529e-05,
      "loss": 0.0819,
      "step": 12330
    },
    {
      "epoch": 1.4427802480692722,
      "loss_main": 0.06991530954837799,
      "step": 12330
    },
    {
      "epoch": 1.4439503861455651,
      "grad_norm": 0.29260778427124023,
      "learning_rate": 9.352140302692733e-05,
      "loss": 0.0782,
      "step": 12340
    },
    {
      "epoch": 1.4439503861455651,
      "loss_main": 0.061759207397699356,
      "step": 12340
    },
    {
      "epoch": 1.445120524221858,
      "grad_norm": 0.4023406505584717,
      "learning_rate": 9.350782981774627e-05,
      "loss": 0.0925,
      "step": 12350
    },
    {
      "epoch": 1.445120524221858,
      "loss_main": 0.08941332995891571,
      "step": 12350
    },
    {
      "epoch": 1.4462906622981513,
      "grad_norm": 0.32195422053337097,
      "learning_rate": 9.349424339203526e-05,
      "loss": 0.0767,
      "step": 12360
    },
    {
      "epoch": 1.4462906622981513,
      "loss_main": 0.05726508051156998,
      "step": 12360
    },
    {
      "epoch": 1.4474608003744442,
      "grad_norm": 0.3905240595340729,
      "learning_rate": 9.34806437539215e-05,
      "loss": 0.0892,
      "step": 12370
    },
    {
      "epoch": 1.4474608003744442,
      "loss_main": 0.16201576590538025,
      "step": 12370
    },
    {
      "epoch": 1.4486309384507372,
      "grad_norm": 0.36837127804756165,
      "learning_rate": 9.346703090753622e-05,
      "loss": 0.0879,
      "step": 12380
    },
    {
      "epoch": 1.4486309384507372,
      "loss_main": 0.11764132976531982,
      "step": 12380
    },
    {
      "epoch": 1.4498010765270302,
      "grad_norm": 0.38683682680130005,
      "learning_rate": 9.345340485701461e-05,
      "loss": 0.079,
      "step": 12390
    },
    {
      "epoch": 1.4498010765270302,
      "loss_main": 0.06428495794534683,
      "step": 12390
    },
    {
      "epoch": 1.4509712146033231,
      "grad_norm": 0.2705724537372589,
      "learning_rate": 9.343976560649595e-05,
      "loss": 0.0711,
      "step": 12400
    },
    {
      "epoch": 1.4509712146033231,
      "loss_main": 0.08388055860996246,
      "step": 12400
    },
    {
      "epoch": 1.4521413526796163,
      "grad_norm": 0.3234003186225891,
      "learning_rate": 9.342611316012344e-05,
      "loss": 0.0771,
      "step": 12410
    },
    {
      "epoch": 1.4521413526796163,
      "loss_main": 0.12183088064193726,
      "step": 12410
    },
    {
      "epoch": 1.4533114907559093,
      "grad_norm": 0.2492019534111023,
      "learning_rate": 9.341244752204437e-05,
      "loss": 0.0769,
      "step": 12420
    },
    {
      "epoch": 1.4533114907559093,
      "loss_main": 0.09100677073001862,
      "step": 12420
    },
    {
      "epoch": 1.4544816288322022,
      "grad_norm": 0.3491683304309845,
      "learning_rate": 9.339876869640995e-05,
      "loss": 0.0767,
      "step": 12430
    },
    {
      "epoch": 1.4544816288322022,
      "loss_main": 0.0520174503326416,
      "step": 12430
    },
    {
      "epoch": 1.4556517669084952,
      "grad_norm": 0.3189852833747864,
      "learning_rate": 9.33850766873755e-05,
      "loss": 0.0679,
      "step": 12440
    },
    {
      "epoch": 1.4556517669084952,
      "loss_main": 0.11503381282091141,
      "step": 12440
    },
    {
      "epoch": 1.4568219049847881,
      "grad_norm": 0.2427692413330078,
      "learning_rate": 9.337137149910028e-05,
      "loss": 0.0821,
      "step": 12450
    },
    {
      "epoch": 1.4568219049847881,
      "loss_main": 0.05277804657816887,
      "step": 12450
    },
    {
      "epoch": 1.4579920430610813,
      "grad_norm": 0.2798173427581787,
      "learning_rate": 9.335765313574753e-05,
      "loss": 0.0742,
      "step": 12460
    },
    {
      "epoch": 1.4579920430610813,
      "loss_main": 0.09032826870679855,
      "step": 12460
    },
    {
      "epoch": 1.4591621811373743,
      "grad_norm": 0.43148648738861084,
      "learning_rate": 9.334392160148457e-05,
      "loss": 0.0815,
      "step": 12470
    },
    {
      "epoch": 1.4591621811373743,
      "loss_main": 0.07167644798755646,
      "step": 12470
    },
    {
      "epoch": 1.4603323192136672,
      "grad_norm": 0.2699102461338043,
      "learning_rate": 9.333017690048264e-05,
      "loss": 0.0776,
      "step": 12480
    },
    {
      "epoch": 1.4603323192136672,
      "loss_main": 0.057645320892333984,
      "step": 12480
    },
    {
      "epoch": 1.4615024572899602,
      "grad_norm": 0.29030653834342957,
      "learning_rate": 9.331641903691706e-05,
      "loss": 0.0794,
      "step": 12490
    },
    {
      "epoch": 1.4615024572899602,
      "loss_main": 0.08301271498203278,
      "step": 12490
    },
    {
      "epoch": 1.4626725953662532,
      "grad_norm": 0.39205417037010193,
      "learning_rate": 9.330264801496707e-05,
      "loss": 0.0845,
      "step": 12500
    },
    {
      "epoch": 1.4626725953662532,
      "loss_main": 0.11698492616415024,
      "step": 12500
    },
    {
      "epoch": 1.4638427334425463,
      "grad_norm": 0.29413995146751404,
      "learning_rate": 9.328886383881594e-05,
      "loss": 0.0926,
      "step": 12510
    },
    {
      "epoch": 1.4638427334425463,
      "loss_main": 0.08352258056402206,
      "step": 12510
    },
    {
      "epoch": 1.4650128715188393,
      "grad_norm": 0.2793509066104889,
      "learning_rate": 9.327506651265095e-05,
      "loss": 0.0784,
      "step": 12520
    },
    {
      "epoch": 1.4650128715188393,
      "loss_main": 0.07021457701921463,
      "step": 12520
    },
    {
      "epoch": 1.4661830095951323,
      "grad_norm": 0.38256406784057617,
      "learning_rate": 9.326125604066338e-05,
      "loss": 0.0843,
      "step": 12530
    },
    {
      "epoch": 1.4661830095951323,
      "loss_main": 0.06304769217967987,
      "step": 12530
    },
    {
      "epoch": 1.4673531476714252,
      "grad_norm": 0.43687573075294495,
      "learning_rate": 9.324743242704847e-05,
      "loss": 0.0827,
      "step": 12540
    },
    {
      "epoch": 1.4673531476714252,
      "loss_main": 0.07398814707994461,
      "step": 12540
    },
    {
      "epoch": 1.4685232857477182,
      "grad_norm": 0.3511740565299988,
      "learning_rate": 9.323359567600546e-05,
      "loss": 0.0883,
      "step": 12550
    },
    {
      "epoch": 1.4685232857477182,
      "loss_main": 0.11567701399326324,
      "step": 12550
    },
    {
      "epoch": 1.4696934238240114,
      "grad_norm": 0.48394837975502014,
      "learning_rate": 9.321974579173761e-05,
      "loss": 0.0811,
      "step": 12560
    },
    {
      "epoch": 1.4696934238240114,
      "loss_main": 0.10349713265895844,
      "step": 12560
    },
    {
      "epoch": 1.4708635619003043,
      "grad_norm": 0.2687028646469116,
      "learning_rate": 9.320588277845213e-05,
      "loss": 0.0918,
      "step": 12570
    },
    {
      "epoch": 1.4708635619003043,
      "loss_main": 0.08884406089782715,
      "step": 12570
    },
    {
      "epoch": 1.4720336999765973,
      "grad_norm": 0.38826823234558105,
      "learning_rate": 9.319200664036026e-05,
      "loss": 0.0828,
      "step": 12580
    },
    {
      "epoch": 1.4720336999765973,
      "loss_main": 0.0671427994966507,
      "step": 12580
    },
    {
      "epoch": 1.4732038380528902,
      "grad_norm": 0.35922902822494507,
      "learning_rate": 9.31781173816772e-05,
      "loss": 0.0736,
      "step": 12590
    },
    {
      "epoch": 1.4732038380528902,
      "loss_main": 0.11361323297023773,
      "step": 12590
    },
    {
      "epoch": 1.4743739761291832,
      "grad_norm": 0.25796282291412354,
      "learning_rate": 9.316421500662212e-05,
      "loss": 0.0743,
      "step": 12600
    },
    {
      "epoch": 1.4743739761291832,
      "loss_main": 0.0684276595711708,
      "step": 12600
    },
    {
      "epoch": 1.4755441142054764,
      "grad_norm": 0.27397772669792175,
      "learning_rate": 9.31502995194182e-05,
      "loss": 0.0777,
      "step": 12610
    },
    {
      "epoch": 1.4755441142054764,
      "loss_main": 0.06678258627653122,
      "step": 12610
    },
    {
      "epoch": 1.4767142522817691,
      "grad_norm": 0.3005768358707428,
      "learning_rate": 9.31363709242926e-05,
      "loss": 0.0802,
      "step": 12620
    },
    {
      "epoch": 1.4767142522817691,
      "loss_main": 0.1160443052649498,
      "step": 12620
    },
    {
      "epoch": 1.4778843903580623,
      "grad_norm": 0.26562613248825073,
      "learning_rate": 9.312242922547647e-05,
      "loss": 0.0726,
      "step": 12630
    },
    {
      "epoch": 1.4778843903580623,
      "loss_main": 0.07802682369947433,
      "step": 12630
    },
    {
      "epoch": 1.4790545284343553,
      "grad_norm": 0.31121405959129333,
      "learning_rate": 9.310847442720492e-05,
      "loss": 0.0806,
      "step": 12640
    },
    {
      "epoch": 1.4790545284343553,
      "loss_main": 0.09681806713342667,
      "step": 12640
    },
    {
      "epoch": 1.4802246665106482,
      "grad_norm": 0.23467929661273956,
      "learning_rate": 9.309450653371706e-05,
      "loss": 0.0784,
      "step": 12650
    },
    {
      "epoch": 1.4802246665106482,
      "loss_main": 0.07064413279294968,
      "step": 12650
    },
    {
      "epoch": 1.4813948045869414,
      "grad_norm": 0.3158050775527954,
      "learning_rate": 9.308052554925595e-05,
      "loss": 0.0767,
      "step": 12660
    },
    {
      "epoch": 1.4813948045869414,
      "loss_main": 0.10649202018976212,
      "step": 12660
    },
    {
      "epoch": 1.4825649426632341,
      "grad_norm": 0.4614030420780182,
      "learning_rate": 9.306653147806867e-05,
      "loss": 0.0834,
      "step": 12670
    },
    {
      "epoch": 1.4825649426632341,
      "loss_main": 0.0767083615064621,
      "step": 12670
    },
    {
      "epoch": 1.4837350807395273,
      "grad_norm": 0.28350046277046204,
      "learning_rate": 9.305252432440622e-05,
      "loss": 0.0731,
      "step": 12680
    },
    {
      "epoch": 1.4837350807395273,
      "loss_main": 0.11842063814401627,
      "step": 12680
    },
    {
      "epoch": 1.4849052188158203,
      "grad_norm": 0.3558371067047119,
      "learning_rate": 9.303850409252361e-05,
      "loss": 0.0804,
      "step": 12690
    },
    {
      "epoch": 1.4849052188158203,
      "loss_main": 0.06187772378325462,
      "step": 12690
    },
    {
      "epoch": 1.4860753568921132,
      "grad_norm": 0.2375917136669159,
      "learning_rate": 9.302447078667985e-05,
      "loss": 0.0776,
      "step": 12700
    },
    {
      "epoch": 1.4860753568921132,
      "loss_main": 0.057264186441898346,
      "step": 12700
    },
    {
      "epoch": 1.4872454949684062,
      "grad_norm": 0.30189016461372375,
      "learning_rate": 9.301042441113783e-05,
      "loss": 0.0723,
      "step": 12710
    },
    {
      "epoch": 1.4872454949684062,
      "loss_main": 0.08071615546941757,
      "step": 12710
    },
    {
      "epoch": 1.4884156330446991,
      "grad_norm": 0.3034808039665222,
      "learning_rate": 9.299636497016451e-05,
      "loss": 0.0819,
      "step": 12720
    },
    {
      "epoch": 1.4884156330446991,
      "loss_main": 0.10916981846094131,
      "step": 12720
    },
    {
      "epoch": 1.4895857711209923,
      "grad_norm": 0.2920791208744049,
      "learning_rate": 9.298229246803076e-05,
      "loss": 0.078,
      "step": 12730
    },
    {
      "epoch": 1.4895857711209923,
      "loss_main": 0.07158888131380081,
      "step": 12730
    },
    {
      "epoch": 1.4907559091972853,
      "grad_norm": 0.3596617877483368,
      "learning_rate": 9.296820690901144e-05,
      "loss": 0.0808,
      "step": 12740
    },
    {
      "epoch": 1.4907559091972853,
      "loss_main": 0.06483443826436996,
      "step": 12740
    },
    {
      "epoch": 1.4919260472735782,
      "grad_norm": 0.30908316373825073,
      "learning_rate": 9.295410829738539e-05,
      "loss": 0.0717,
      "step": 12750
    },
    {
      "epoch": 1.4919260472735782,
      "loss_main": 0.0902261734008789,
      "step": 12750
    },
    {
      "epoch": 1.4930961853498712,
      "grad_norm": 0.23670995235443115,
      "learning_rate": 9.293999663743535e-05,
      "loss": 0.075,
      "step": 12760
    },
    {
      "epoch": 1.4930961853498712,
      "loss_main": 0.0656767413020134,
      "step": 12760
    },
    {
      "epoch": 1.4942663234261642,
      "grad_norm": 0.20945221185684204,
      "learning_rate": 9.292587193344813e-05,
      "loss": 0.082,
      "step": 12770
    },
    {
      "epoch": 1.4942663234261642,
      "loss_main": 0.06226884946227074,
      "step": 12770
    },
    {
      "epoch": 1.4954364615024573,
      "grad_norm": 0.39521488547325134,
      "learning_rate": 9.291173418971437e-05,
      "loss": 0.0742,
      "step": 12780
    },
    {
      "epoch": 1.4954364615024573,
      "loss_main": 0.08504544198513031,
      "step": 12780
    },
    {
      "epoch": 1.4966065995787503,
      "grad_norm": 0.3391793668270111,
      "learning_rate": 9.28975834105288e-05,
      "loss": 0.0787,
      "step": 12790
    },
    {
      "epoch": 1.4966065995787503,
      "loss_main": 0.07307659089565277,
      "step": 12790
    },
    {
      "epoch": 1.4977767376550433,
      "grad_norm": 0.32654914259910583,
      "learning_rate": 9.288341960019004e-05,
      "loss": 0.0799,
      "step": 12800
    },
    {
      "epoch": 1.4977767376550433,
      "loss_main": 0.07283218204975128,
      "step": 12800
    },
    {
      "epoch": 1.4989468757313362,
      "grad_norm": 0.5309473872184753,
      "learning_rate": 9.286924276300067e-05,
      "loss": 0.0846,
      "step": 12810
    },
    {
      "epoch": 1.4989468757313362,
      "loss_main": 0.07837758958339691,
      "step": 12810
    },
    {
      "epoch": 1.5001170138076292,
      "grad_norm": 0.4690445363521576,
      "learning_rate": 9.285505290326726e-05,
      "loss": 0.0801,
      "step": 12820
    },
    {
      "epoch": 1.5001170138076292,
      "loss_main": 0.05846855416893959,
      "step": 12820
    },
    {
      "epoch": 1.5012871518839224,
      "grad_norm": 0.32706519961357117,
      "learning_rate": 9.284085002530027e-05,
      "loss": 0.075,
      "step": 12830
    },
    {
      "epoch": 1.5012871518839224,
      "loss_main": 0.11365098506212234,
      "step": 12830
    },
    {
      "epoch": 1.5024572899602153,
      "grad_norm": 0.3616355061531067,
      "learning_rate": 9.282663413341422e-05,
      "loss": 0.0885,
      "step": 12840
    },
    {
      "epoch": 1.5024572899602153,
      "loss_main": 0.08688824623823166,
      "step": 12840
    },
    {
      "epoch": 1.5036274280365083,
      "grad_norm": 0.44820383191108704,
      "learning_rate": 9.281240523192747e-05,
      "loss": 0.082,
      "step": 12850
    },
    {
      "epoch": 1.5036274280365083,
      "loss_main": 0.08382226526737213,
      "step": 12850
    },
    {
      "epoch": 1.5047975661128015,
      "grad_norm": 0.3814099431037903,
      "learning_rate": 9.279816332516242e-05,
      "loss": 0.0834,
      "step": 12860
    },
    {
      "epoch": 1.5047975661128015,
      "loss_main": 0.06202642247080803,
      "step": 12860
    },
    {
      "epoch": 1.5059677041890942,
      "grad_norm": 0.277367502450943,
      "learning_rate": 9.278390841744536e-05,
      "loss": 0.0834,
      "step": 12870
    },
    {
      "epoch": 1.5059677041890942,
      "loss_main": 0.05205431953072548,
      "step": 12870
    },
    {
      "epoch": 1.5071378422653874,
      "grad_norm": 0.40353718400001526,
      "learning_rate": 9.276964051310658e-05,
      "loss": 0.0793,
      "step": 12880
    },
    {
      "epoch": 1.5071378422653874,
      "loss_main": 0.07592378556728363,
      "step": 12880
    },
    {
      "epoch": 1.5083079803416803,
      "grad_norm": 0.28085842728614807,
      "learning_rate": 9.275535961648027e-05,
      "loss": 0.0742,
      "step": 12890
    },
    {
      "epoch": 1.5083079803416803,
      "loss_main": 0.10358435660600662,
      "step": 12890
    },
    {
      "epoch": 1.5094781184179733,
      "grad_norm": 0.567886471748352,
      "learning_rate": 9.274106573190459e-05,
      "loss": 0.0836,
      "step": 12900
    },
    {
      "epoch": 1.5094781184179733,
      "loss_main": 0.05674903467297554,
      "step": 12900
    },
    {
      "epoch": 1.5106482564942665,
      "grad_norm": 0.4031069278717041,
      "learning_rate": 9.272675886372168e-05,
      "loss": 0.073,
      "step": 12910
    },
    {
      "epoch": 1.5106482564942665,
      "loss_main": 0.09673731029033661,
      "step": 12910
    },
    {
      "epoch": 1.5118183945705592,
      "grad_norm": 0.3895335793495178,
      "learning_rate": 9.271243901627754e-05,
      "loss": 0.0838,
      "step": 12920
    },
    {
      "epoch": 1.5118183945705592,
      "loss_main": 0.06865201145410538,
      "step": 12920
    },
    {
      "epoch": 1.5129885326468524,
      "grad_norm": 0.35131746530532837,
      "learning_rate": 9.269810619392219e-05,
      "loss": 0.075,
      "step": 12930
    },
    {
      "epoch": 1.5129885326468524,
      "loss_main": 0.07805679738521576,
      "step": 12930
    },
    {
      "epoch": 1.5141586707231454,
      "grad_norm": 0.2578631639480591,
      "learning_rate": 9.268376040100955e-05,
      "loss": 0.0836,
      "step": 12940
    },
    {
      "epoch": 1.5141586707231454,
      "loss_main": 0.12034360319375992,
      "step": 12940
    },
    {
      "epoch": 1.5153288087994383,
      "grad_norm": 0.32329508662223816,
      "learning_rate": 9.266940164189752e-05,
      "loss": 0.0735,
      "step": 12950
    },
    {
      "epoch": 1.5153288087994383,
      "loss_main": 0.08857708424329758,
      "step": 12950
    },
    {
      "epoch": 1.5164989468757315,
      "grad_norm": 0.4324454963207245,
      "learning_rate": 9.265502992094787e-05,
      "loss": 0.0739,
      "step": 12960
    },
    {
      "epoch": 1.5164989468757315,
      "loss_main": 0.07457201927900314,
      "step": 12960
    },
    {
      "epoch": 1.5176690849520242,
      "grad_norm": 0.3776656687259674,
      "learning_rate": 9.264064524252638e-05,
      "loss": 0.0711,
      "step": 12970
    },
    {
      "epoch": 1.5176690849520242,
      "loss_main": 0.07516966760158539,
      "step": 12970
    },
    {
      "epoch": 1.5188392230283174,
      "grad_norm": 0.26097238063812256,
      "learning_rate": 9.262624761100271e-05,
      "loss": 0.0787,
      "step": 12980
    },
    {
      "epoch": 1.5188392230283174,
      "loss_main": 0.06265975534915924,
      "step": 12980
    },
    {
      "epoch": 1.5200093611046104,
      "grad_norm": 0.28198811411857605,
      "learning_rate": 9.261183703075051e-05,
      "loss": 0.0774,
      "step": 12990
    },
    {
      "epoch": 1.5200093611046104,
      "loss_main": 0.10287870466709137,
      "step": 12990
    },
    {
      "epoch": 1.5211794991809033,
      "grad_norm": 0.29702308773994446,
      "learning_rate": 9.259741350614733e-05,
      "loss": 0.0764,
      "step": 13000
    },
    {
      "epoch": 1.5211794991809033,
      "loss_main": 0.06037072464823723,
      "step": 13000
    },
    {
      "epoch": 1.5223496372571963,
      "grad_norm": 0.34218189120292664,
      "learning_rate": 9.258297704157464e-05,
      "loss": 0.0805,
      "step": 13010
    },
    {
      "epoch": 1.5223496372571963,
      "loss_main": 0.056067612022161484,
      "step": 13010
    },
    {
      "epoch": 1.5235197753334893,
      "grad_norm": 0.31046220660209656,
      "learning_rate": 9.256852764141786e-05,
      "loss": 0.0792,
      "step": 13020
    },
    {
      "epoch": 1.5235197753334893,
      "loss_main": 0.0790664330124855,
      "step": 13020
    },
    {
      "epoch": 1.5246899134097824,
      "grad_norm": 0.38848137855529785,
      "learning_rate": 9.255406531006634e-05,
      "loss": 0.0668,
      "step": 13030
    },
    {
      "epoch": 1.5246899134097824,
      "loss_main": 0.07076078653335571,
      "step": 13030
    },
    {
      "epoch": 1.5258600514860754,
      "grad_norm": 0.3066194951534271,
      "learning_rate": 9.253959005191335e-05,
      "loss": 0.0836,
      "step": 13040
    },
    {
      "epoch": 1.5258600514860754,
      "loss_main": 0.09412118047475815,
      "step": 13040
    },
    {
      "epoch": 1.5270301895623684,
      "grad_norm": 0.2770589590072632,
      "learning_rate": 9.25251018713561e-05,
      "loss": 0.0912,
      "step": 13050
    },
    {
      "epoch": 1.5270301895623684,
      "loss_main": 0.11279024928808212,
      "step": 13050
    },
    {
      "epoch": 1.5282003276386613,
      "grad_norm": 0.34929585456848145,
      "learning_rate": 9.251060077279571e-05,
      "loss": 0.0772,
      "step": 13060
    },
    {
      "epoch": 1.5282003276386613,
      "loss_main": 0.07527320832014084,
      "step": 13060
    },
    {
      "epoch": 1.5293704657149543,
      "grad_norm": 0.41738811135292053,
      "learning_rate": 9.249608676063724e-05,
      "loss": 0.0763,
      "step": 13070
    },
    {
      "epoch": 1.5293704657149543,
      "loss_main": 0.12026821076869965,
      "step": 13070
    },
    {
      "epoch": 1.5305406037912475,
      "grad_norm": 0.3192908763885498,
      "learning_rate": 9.248155983928964e-05,
      "loss": 0.078,
      "step": 13080
    },
    {
      "epoch": 1.5305406037912475,
      "loss_main": 0.07118839025497437,
      "step": 13080
    },
    {
      "epoch": 1.5317107418675404,
      "grad_norm": 0.44489654898643494,
      "learning_rate": 9.246702001316583e-05,
      "loss": 0.0759,
      "step": 13090
    },
    {
      "epoch": 1.5317107418675404,
      "loss_main": 0.11739242821931839,
      "step": 13090
    },
    {
      "epoch": 1.5328808799438334,
      "grad_norm": 0.281655490398407,
      "learning_rate": 9.245246728668262e-05,
      "loss": 0.0824,
      "step": 13100
    },
    {
      "epoch": 1.5328808799438334,
      "loss_main": 0.06402438879013062,
      "step": 13100
    },
    {
      "epoch": 1.5340510180201263,
      "grad_norm": 0.4806385040283203,
      "learning_rate": 9.243790166426073e-05,
      "loss": 0.0836,
      "step": 13110
    },
    {
      "epoch": 1.5340510180201263,
      "loss_main": 0.06749044358730316,
      "step": 13110
    },
    {
      "epoch": 1.5352211560964193,
      "grad_norm": 0.3027496337890625,
      "learning_rate": 9.242332315032484e-05,
      "loss": 0.0791,
      "step": 13120
    },
    {
      "epoch": 1.5352211560964193,
      "loss_main": 0.06659425795078278,
      "step": 13120
    },
    {
      "epoch": 1.5363912941727125,
      "grad_norm": 0.3802938163280487,
      "learning_rate": 9.240873174930349e-05,
      "loss": 0.0796,
      "step": 13130
    },
    {
      "epoch": 1.5363912941727125,
      "loss_main": 0.07363864034414291,
      "step": 13130
    },
    {
      "epoch": 1.5375614322490054,
      "grad_norm": 0.5051125288009644,
      "learning_rate": 9.239412746562917e-05,
      "loss": 0.0822,
      "step": 13140
    },
    {
      "epoch": 1.5375614322490054,
      "loss_main": 0.09543706476688385,
      "step": 13140
    },
    {
      "epoch": 1.5387315703252984,
      "grad_norm": 0.4067344069480896,
      "learning_rate": 9.237951030373828e-05,
      "loss": 0.0896,
      "step": 13150
    },
    {
      "epoch": 1.5387315703252984,
      "loss_main": 0.1294710338115692,
      "step": 13150
    },
    {
      "epoch": 1.5399017084015914,
      "grad_norm": 0.25606945157051086,
      "learning_rate": 9.236488026807113e-05,
      "loss": 0.0863,
      "step": 13160
    },
    {
      "epoch": 1.5399017084015914,
      "loss_main": 0.046846482902765274,
      "step": 13160
    },
    {
      "epoch": 1.5410718464778843,
      "grad_norm": 0.29537078738212585,
      "learning_rate": 9.235023736307193e-05,
      "loss": 0.0819,
      "step": 13170
    },
    {
      "epoch": 1.5410718464778843,
      "loss_main": 0.06212697550654411,
      "step": 13170
    },
    {
      "epoch": 1.5422419845541775,
      "grad_norm": 0.31403279304504395,
      "learning_rate": 9.233558159318881e-05,
      "loss": 0.0791,
      "step": 13180
    },
    {
      "epoch": 1.5422419845541775,
      "loss_main": 0.08253419399261475,
      "step": 13180
    },
    {
      "epoch": 1.5434121226304702,
      "grad_norm": 0.2909724712371826,
      "learning_rate": 9.232091296287382e-05,
      "loss": 0.0717,
      "step": 13190
    },
    {
      "epoch": 1.5434121226304702,
      "loss_main": 0.05831003561615944,
      "step": 13190
    },
    {
      "epoch": 1.5445822607067634,
      "grad_norm": 0.31076329946517944,
      "learning_rate": 9.230623147658288e-05,
      "loss": 0.0815,
      "step": 13200
    },
    {
      "epoch": 1.5445822607067634,
      "loss_main": 0.09241465479135513,
      "step": 13200
    },
    {
      "epoch": 1.5457523987830564,
      "grad_norm": 0.2331158071756363,
      "learning_rate": 9.229153713877586e-05,
      "loss": 0.0758,
      "step": 13210
    },
    {
      "epoch": 1.5457523987830564,
      "loss_main": 0.0898294448852539,
      "step": 13210
    },
    {
      "epoch": 1.5469225368593493,
      "grad_norm": 0.36535730957984924,
      "learning_rate": 9.227682995391649e-05,
      "loss": 0.0868,
      "step": 13220
    },
    {
      "epoch": 1.5469225368593493,
      "loss_main": 0.15113650262355804,
      "step": 13220
    },
    {
      "epoch": 1.5480926749356425,
      "grad_norm": 0.5328810811042786,
      "learning_rate": 9.226210992647243e-05,
      "loss": 0.0831,
      "step": 13230
    },
    {
      "epoch": 1.5480926749356425,
      "loss_main": 0.05913874879479408,
      "step": 13230
    },
    {
      "epoch": 1.5492628130119352,
      "grad_norm": 0.38476884365081787,
      "learning_rate": 9.224737706091525e-05,
      "loss": 0.076,
      "step": 13240
    },
    {
      "epoch": 1.5492628130119352,
      "loss_main": 0.07554265111684799,
      "step": 13240
    },
    {
      "epoch": 1.5504329510882284,
      "grad_norm": 0.31890788674354553,
      "learning_rate": 9.223263136172039e-05,
      "loss": 0.0849,
      "step": 13250
    },
    {
      "epoch": 1.5504329510882284,
      "loss_main": 0.13479918241500854,
      "step": 13250
    },
    {
      "epoch": 1.5516030891645214,
      "grad_norm": 0.2872370481491089,
      "learning_rate": 9.22178728333672e-05,
      "loss": 0.0862,
      "step": 13260
    },
    {
      "epoch": 1.5516030891645214,
      "loss_main": 0.0732385516166687,
      "step": 13260
    },
    {
      "epoch": 1.5527732272408143,
      "grad_norm": 0.35324424505233765,
      "learning_rate": 9.220310148033897e-05,
      "loss": 0.0851,
      "step": 13270
    },
    {
      "epoch": 1.5527732272408143,
      "loss_main": 0.050187867134809494,
      "step": 13270
    },
    {
      "epoch": 1.5539433653171075,
      "grad_norm": 0.4465695023536682,
      "learning_rate": 9.21883173071228e-05,
      "loss": 0.0719,
      "step": 13280
    },
    {
      "epoch": 1.5539433653171075,
      "loss_main": 0.07067488133907318,
      "step": 13280
    },
    {
      "epoch": 1.5551135033934003,
      "grad_norm": 0.2617020010948181,
      "learning_rate": 9.217352031820976e-05,
      "loss": 0.073,
      "step": 13290
    },
    {
      "epoch": 1.5551135033934003,
      "loss_main": 0.07961026579141617,
      "step": 13290
    },
    {
      "epoch": 1.5562836414696934,
      "grad_norm": 0.6096323132514954,
      "learning_rate": 9.215871051809477e-05,
      "loss": 0.0784,
      "step": 13300
    },
    {
      "epoch": 1.5562836414696934,
      "loss_main": 0.054321084171533585,
      "step": 13300
    },
    {
      "epoch": 1.5574537795459864,
      "grad_norm": 0.38356998562812805,
      "learning_rate": 9.214388791127666e-05,
      "loss": 0.0764,
      "step": 13310
    },
    {
      "epoch": 1.5574537795459864,
      "loss_main": 0.10656391084194183,
      "step": 13310
    },
    {
      "epoch": 1.5586239176222794,
      "grad_norm": 0.39801502227783203,
      "learning_rate": 9.212905250225814e-05,
      "loss": 0.0898,
      "step": 13320
    },
    {
      "epoch": 1.5586239176222794,
      "loss_main": 0.08071435242891312,
      "step": 13320
    },
    {
      "epoch": 1.5597940556985725,
      "grad_norm": 0.3033803105354309,
      "learning_rate": 9.211420429554583e-05,
      "loss": 0.0775,
      "step": 13330
    },
    {
      "epoch": 1.5597940556985725,
      "loss_main": 0.07113327831029892,
      "step": 13330
    },
    {
      "epoch": 1.5609641937748653,
      "grad_norm": 0.37986525893211365,
      "learning_rate": 9.209934329565022e-05,
      "loss": 0.0742,
      "step": 13340
    },
    {
      "epoch": 1.5609641937748653,
      "loss_main": 0.056282006204128265,
      "step": 13340
    },
    {
      "epoch": 1.5621343318511585,
      "grad_norm": 0.2536446452140808,
      "learning_rate": 9.208446950708568e-05,
      "loss": 0.0814,
      "step": 13350
    },
    {
      "epoch": 1.5621343318511585,
      "loss_main": 0.10738372057676315,
      "step": 13350
    },
    {
      "epoch": 1.5633044699274514,
      "grad_norm": 0.360251247882843,
      "learning_rate": 9.20695829343705e-05,
      "loss": 0.0821,
      "step": 13360
    },
    {
      "epoch": 1.5633044699274514,
      "loss_main": 0.05054163187742233,
      "step": 13360
    },
    {
      "epoch": 1.5644746080037444,
      "grad_norm": 0.2484610229730606,
      "learning_rate": 9.205468358202678e-05,
      "loss": 0.0644,
      "step": 13370
    },
    {
      "epoch": 1.5644746080037444,
      "loss_main": 0.05965084582567215,
      "step": 13370
    },
    {
      "epoch": 1.5656447460800376,
      "grad_norm": 0.2840879261493683,
      "learning_rate": 9.203977145458059e-05,
      "loss": 0.0836,
      "step": 13380
    },
    {
      "epoch": 1.5656447460800376,
      "loss_main": 0.05987027660012245,
      "step": 13380
    },
    {
      "epoch": 1.5668148841563303,
      "grad_norm": 0.3337753415107727,
      "learning_rate": 9.202484655656182e-05,
      "loss": 0.077,
      "step": 13390
    },
    {
      "epoch": 1.5668148841563303,
      "loss_main": 0.08051389455795288,
      "step": 13390
    },
    {
      "epoch": 1.5679850222326235,
      "grad_norm": 0.3700684607028961,
      "learning_rate": 9.200990889250427e-05,
      "loss": 0.0825,
      "step": 13400
    },
    {
      "epoch": 1.5679850222326235,
      "loss_main": 0.05470026656985283,
      "step": 13400
    },
    {
      "epoch": 1.5691551603089164,
      "grad_norm": 0.3258076310157776,
      "learning_rate": 9.19949584669456e-05,
      "loss": 0.0778,
      "step": 13410
    },
    {
      "epoch": 1.5691551603089164,
      "loss_main": 0.09075718373060226,
      "step": 13410
    },
    {
      "epoch": 1.5703252983852094,
      "grad_norm": 0.30918142199516296,
      "learning_rate": 9.197999528442738e-05,
      "loss": 0.0798,
      "step": 13420
    },
    {
      "epoch": 1.5703252983852094,
      "loss_main": 0.07369227707386017,
      "step": 13420
    },
    {
      "epoch": 1.5714954364615026,
      "grad_norm": 0.2519483268260956,
      "learning_rate": 9.196501934949499e-05,
      "loss": 0.0732,
      "step": 13430
    },
    {
      "epoch": 1.5714954364615026,
      "loss_main": 0.07740803062915802,
      "step": 13430
    },
    {
      "epoch": 1.5726655745377953,
      "grad_norm": 0.38922134041786194,
      "learning_rate": 9.195003066669776e-05,
      "loss": 0.0796,
      "step": 13440
    },
    {
      "epoch": 1.5726655745377953,
      "loss_main": 0.07556804269552231,
      "step": 13440
    },
    {
      "epoch": 1.5738357126140885,
      "grad_norm": 0.43634092807769775,
      "learning_rate": 9.193502924058884e-05,
      "loss": 0.0818,
      "step": 13450
    },
    {
      "epoch": 1.5738357126140885,
      "loss_main": 0.08552438765764236,
      "step": 13450
    },
    {
      "epoch": 1.5750058506903815,
      "grad_norm": 0.30001160502433777,
      "learning_rate": 9.192001507572526e-05,
      "loss": 0.088,
      "step": 13460
    },
    {
      "epoch": 1.5750058506903815,
      "loss_main": 0.07170342653989792,
      "step": 13460
    },
    {
      "epoch": 1.5761759887666744,
      "grad_norm": 0.33474451303482056,
      "learning_rate": 9.190498817666793e-05,
      "loss": 0.09,
      "step": 13470
    },
    {
      "epoch": 1.5761759887666744,
      "loss_main": 0.06779057532548904,
      "step": 13470
    },
    {
      "epoch": 1.5773461268429676,
      "grad_norm": 0.22680148482322693,
      "learning_rate": 9.188994854798163e-05,
      "loss": 0.07,
      "step": 13480
    },
    {
      "epoch": 1.5773461268429676,
      "loss_main": 0.08661352097988129,
      "step": 13480
    },
    {
      "epoch": 1.5785162649192603,
      "grad_norm": 0.17641900479793549,
      "learning_rate": 9.187489619423499e-05,
      "loss": 0.0725,
      "step": 13490
    },
    {
      "epoch": 1.5785162649192603,
      "loss_main": 0.07655288279056549,
      "step": 13490
    },
    {
      "epoch": 1.5796864029955535,
      "grad_norm": 0.26920291781425476,
      "learning_rate": 9.185983112000056e-05,
      "loss": 0.0817,
      "step": 13500
    },
    {
      "epoch": 1.5796864029955535,
      "loss_main": 0.05162472277879715,
      "step": 13500
    },
    {
      "epoch": 1.5808565410718465,
      "grad_norm": 0.47893571853637695,
      "learning_rate": 9.184475332985464e-05,
      "loss": 0.0743,
      "step": 13510
    },
    {
      "epoch": 1.5808565410718465,
      "loss_main": 0.06443627178668976,
      "step": 13510
    },
    {
      "epoch": 1.5820266791481394,
      "grad_norm": 0.23568925261497498,
      "learning_rate": 9.182966282837754e-05,
      "loss": 0.0824,
      "step": 13520
    },
    {
      "epoch": 1.5820266791481394,
      "loss_main": 0.03871018439531326,
      "step": 13520
    },
    {
      "epoch": 1.5831968172244326,
      "grad_norm": 0.320263534784317,
      "learning_rate": 9.18145596201533e-05,
      "loss": 0.0667,
      "step": 13530
    },
    {
      "epoch": 1.5831968172244326,
      "loss_main": 0.06915784627199173,
      "step": 13530
    },
    {
      "epoch": 1.5843669553007254,
      "grad_norm": 0.32315516471862793,
      "learning_rate": 9.179944370976991e-05,
      "loss": 0.0712,
      "step": 13540
    },
    {
      "epoch": 1.5843669553007254,
      "loss_main": 0.0960531234741211,
      "step": 13540
    },
    {
      "epoch": 1.5855370933770185,
      "grad_norm": 0.35415172576904297,
      "learning_rate": 9.178431510181918e-05,
      "loss": 0.0706,
      "step": 13550
    },
    {
      "epoch": 1.5855370933770185,
      "loss_main": 0.05764487013220787,
      "step": 13550
    },
    {
      "epoch": 1.5867072314533115,
      "grad_norm": 0.37420499324798584,
      "learning_rate": 9.176917380089675e-05,
      "loss": 0.0923,
      "step": 13560
    },
    {
      "epoch": 1.5867072314533115,
      "loss_main": 0.0848606675863266,
      "step": 13560
    },
    {
      "epoch": 1.5878773695296045,
      "grad_norm": 0.5688102841377258,
      "learning_rate": 9.175401981160219e-05,
      "loss": 0.0832,
      "step": 13570
    },
    {
      "epoch": 1.5878773695296045,
      "loss_main": 0.044194549322128296,
      "step": 13570
    },
    {
      "epoch": 1.5890475076058976,
      "grad_norm": 0.37909993529319763,
      "learning_rate": 9.173885313853885e-05,
      "loss": 0.0775,
      "step": 13580
    },
    {
      "epoch": 1.5890475076058976,
      "loss_main": 0.08121798932552338,
      "step": 13580
    },
    {
      "epoch": 1.5902176456821904,
      "grad_norm": 0.4014701843261719,
      "learning_rate": 9.172367378631398e-05,
      "loss": 0.0844,
      "step": 13590
    },
    {
      "epoch": 1.5902176456821904,
      "loss_main": 0.07181777060031891,
      "step": 13590
    },
    {
      "epoch": 1.5913877837584836,
      "grad_norm": 0.3357994556427002,
      "learning_rate": 9.170848175953866e-05,
      "loss": 0.0998,
      "step": 13600
    },
    {
      "epoch": 1.5913877837584836,
      "loss_main": 0.08087816089391708,
      "step": 13600
    },
    {
      "epoch": 1.5925579218347765,
      "grad_norm": 0.38579270243644714,
      "learning_rate": 9.169327706282784e-05,
      "loss": 0.0804,
      "step": 13610
    },
    {
      "epoch": 1.5925579218347765,
      "loss_main": 0.12104015797376633,
      "step": 13610
    },
    {
      "epoch": 1.5937280599110695,
      "grad_norm": 0.3729666769504547,
      "learning_rate": 9.167805970080029e-05,
      "loss": 0.0727,
      "step": 13620
    },
    {
      "epoch": 1.5937280599110695,
      "loss_main": 0.1253582388162613,
      "step": 13620
    },
    {
      "epoch": 1.5948981979873627,
      "grad_norm": 0.3230573236942291,
      "learning_rate": 9.166282967807864e-05,
      "loss": 0.0755,
      "step": 13630
    },
    {
      "epoch": 1.5948981979873627,
      "loss_main": 0.06148582696914673,
      "step": 13630
    },
    {
      "epoch": 1.5960683360636554,
      "grad_norm": 0.25052928924560547,
      "learning_rate": 9.16475869992894e-05,
      "loss": 0.071,
      "step": 13640
    },
    {
      "epoch": 1.5960683360636554,
      "loss_main": 0.08074556291103363,
      "step": 13640
    },
    {
      "epoch": 1.5972384741399486,
      "grad_norm": 0.30504387617111206,
      "learning_rate": 9.163233166906284e-05,
      "loss": 0.0684,
      "step": 13650
    },
    {
      "epoch": 1.5972384741399486,
      "loss_main": 0.07870148867368698,
      "step": 13650
    },
    {
      "epoch": 1.5984086122162415,
      "grad_norm": 0.3834797739982605,
      "learning_rate": 9.161706369203317e-05,
      "loss": 0.0787,
      "step": 13660
    },
    {
      "epoch": 1.5984086122162415,
      "loss_main": 0.06119387596845627,
      "step": 13660
    },
    {
      "epoch": 1.5995787502925345,
      "grad_norm": 0.40627405047416687,
      "learning_rate": 9.16017830728384e-05,
      "loss": 0.0704,
      "step": 13670
    },
    {
      "epoch": 1.5995787502925345,
      "loss_main": 0.09360580146312714,
      "step": 13670
    },
    {
      "epoch": 1.6007488883688277,
      "grad_norm": 0.23360468447208405,
      "learning_rate": 9.158648981612035e-05,
      "loss": 0.0768,
      "step": 13680
    },
    {
      "epoch": 1.6007488883688277,
      "loss_main": 0.04391556605696678,
      "step": 13680
    },
    {
      "epoch": 1.6019190264451204,
      "grad_norm": 0.29426977038383484,
      "learning_rate": 9.157118392652472e-05,
      "loss": 0.0702,
      "step": 13690
    },
    {
      "epoch": 1.6019190264451204,
      "loss_main": 0.08047784864902496,
      "step": 13690
    },
    {
      "epoch": 1.6030891645214136,
      "grad_norm": 0.41242891550064087,
      "learning_rate": 9.155586540870104e-05,
      "loss": 0.0714,
      "step": 13700
    },
    {
      "epoch": 1.6030891645214136,
      "loss_main": 0.06544408947229385,
      "step": 13700
    },
    {
      "epoch": 1.6042593025977065,
      "grad_norm": 0.2548418641090393,
      "learning_rate": 9.154053426730267e-05,
      "loss": 0.0714,
      "step": 13710
    },
    {
      "epoch": 1.6042593025977065,
      "loss_main": 0.04728793352842331,
      "step": 13710
    },
    {
      "epoch": 1.6054294406739995,
      "grad_norm": 0.39409276843070984,
      "learning_rate": 9.15251905069868e-05,
      "loss": 0.0803,
      "step": 13720
    },
    {
      "epoch": 1.6054294406739995,
      "loss_main": 0.09787225723266602,
      "step": 13720
    },
    {
      "epoch": 1.6065995787502927,
      "grad_norm": 0.3091713786125183,
      "learning_rate": 9.150983413241446e-05,
      "loss": 0.0827,
      "step": 13730
    },
    {
      "epoch": 1.6065995787502927,
      "loss_main": 0.08736748993396759,
      "step": 13730
    },
    {
      "epoch": 1.6077697168265854,
      "grad_norm": 0.3689734637737274,
      "learning_rate": 9.149446514825051e-05,
      "loss": 0.0863,
      "step": 13740
    },
    {
      "epoch": 1.6077697168265854,
      "loss_main": 0.08320406824350357,
      "step": 13740
    },
    {
      "epoch": 1.6089398549028786,
      "grad_norm": 0.32760095596313477,
      "learning_rate": 9.147908355916365e-05,
      "loss": 0.0888,
      "step": 13750
    },
    {
      "epoch": 1.6089398549028786,
      "loss_main": 0.1096971184015274,
      "step": 13750
    },
    {
      "epoch": 1.6101099929791716,
      "grad_norm": 0.26187172532081604,
      "learning_rate": 9.146368936982642e-05,
      "loss": 0.0959,
      "step": 13760
    },
    {
      "epoch": 1.6101099929791716,
      "loss_main": 0.07020460814237595,
      "step": 13760
    },
    {
      "epoch": 1.6112801310554645,
      "grad_norm": 0.21825242042541504,
      "learning_rate": 9.144828258491511e-05,
      "loss": 0.0828,
      "step": 13770
    },
    {
      "epoch": 1.6112801310554645,
      "loss_main": 0.09334401041269302,
      "step": 13770
    },
    {
      "epoch": 1.6124502691317577,
      "grad_norm": 0.47378596663475037,
      "learning_rate": 9.143286320910996e-05,
      "loss": 0.0919,
      "step": 13780
    },
    {
      "epoch": 1.6124502691317577,
      "loss_main": 0.06167949363589287,
      "step": 13780
    },
    {
      "epoch": 1.6136204072080504,
      "grad_norm": 0.37614893913269043,
      "learning_rate": 9.141743124709491e-05,
      "loss": 0.0747,
      "step": 13790
    },
    {
      "epoch": 1.6136204072080504,
      "loss_main": 0.08555563539266586,
      "step": 13790
    },
    {
      "epoch": 1.6147905452843436,
      "grad_norm": 0.3476293385028839,
      "learning_rate": 9.140198670355784e-05,
      "loss": 0.0742,
      "step": 13800
    },
    {
      "epoch": 1.6147905452843436,
      "loss_main": 0.10405532270669937,
      "step": 13800
    },
    {
      "epoch": 1.6159606833606366,
      "grad_norm": 0.3471044898033142,
      "learning_rate": 9.138652958319034e-05,
      "loss": 0.0886,
      "step": 13810
    },
    {
      "epoch": 1.6159606833606366,
      "loss_main": 0.04341468587517738,
      "step": 13810
    },
    {
      "epoch": 1.6171308214369295,
      "grad_norm": 0.3035576641559601,
      "learning_rate": 9.137105989068791e-05,
      "loss": 0.0764,
      "step": 13820
    },
    {
      "epoch": 1.6171308214369295,
      "loss_main": 0.07899243384599686,
      "step": 13820
    },
    {
      "epoch": 1.6183009595132227,
      "grad_norm": 0.40063607692718506,
      "learning_rate": 9.135557763074983e-05,
      "loss": 0.0785,
      "step": 13830
    },
    {
      "epoch": 1.6183009595132227,
      "loss_main": 0.08661212772130966,
      "step": 13830
    },
    {
      "epoch": 1.6194710975895155,
      "grad_norm": 0.45782360434532166,
      "learning_rate": 9.13400828080792e-05,
      "loss": 0.0886,
      "step": 13840
    },
    {
      "epoch": 1.6194710975895155,
      "loss_main": 0.0472542867064476,
      "step": 13840
    },
    {
      "epoch": 1.6206412356658086,
      "grad_norm": 0.3898983895778656,
      "learning_rate": 9.132457542738292e-05,
      "loss": 0.0751,
      "step": 13850
    },
    {
      "epoch": 1.6206412356658086,
      "loss_main": 0.14333213865756989,
      "step": 13850
    },
    {
      "epoch": 1.6218113737421016,
      "grad_norm": 0.32083019614219666,
      "learning_rate": 9.130905549337174e-05,
      "loss": 0.0746,
      "step": 13860
    },
    {
      "epoch": 1.6218113737421016,
      "loss_main": 0.0899166613817215,
      "step": 13860
    },
    {
      "epoch": 1.6229815118183946,
      "grad_norm": 0.3741950988769531,
      "learning_rate": 9.129352301076021e-05,
      "loss": 0.0792,
      "step": 13870
    },
    {
      "epoch": 1.6229815118183946,
      "loss_main": 0.09298332780599594,
      "step": 13870
    },
    {
      "epoch": 1.6241516498946875,
      "grad_norm": 0.31742405891418457,
      "learning_rate": 9.127797798426668e-05,
      "loss": 0.0795,
      "step": 13880
    },
    {
      "epoch": 1.6241516498946875,
      "loss_main": 0.12579329311847687,
      "step": 13880
    },
    {
      "epoch": 1.6253217879709805,
      "grad_norm": 0.2592814862728119,
      "learning_rate": 9.126242041861333e-05,
      "loss": 0.0871,
      "step": 13890
    },
    {
      "epoch": 1.6253217879709805,
      "loss_main": 0.06153884157538414,
      "step": 13890
    },
    {
      "epoch": 1.6264919260472737,
      "grad_norm": 0.244527667760849,
      "learning_rate": 9.124685031852611e-05,
      "loss": 0.0749,
      "step": 13900
    },
    {
      "epoch": 1.6264919260472737,
      "loss_main": 0.07109413295984268,
      "step": 13900
    },
    {
      "epoch": 1.6276620641235666,
      "grad_norm": 0.24457311630249023,
      "learning_rate": 9.123126768873482e-05,
      "loss": 0.078,
      "step": 13910
    },
    {
      "epoch": 1.6276620641235666,
      "loss_main": 0.07009454071521759,
      "step": 13910
    },
    {
      "epoch": 1.6288322021998596,
      "grad_norm": 0.3400014340877533,
      "learning_rate": 9.121567253397308e-05,
      "loss": 0.0824,
      "step": 13920
    },
    {
      "epoch": 1.6288322021998596,
      "loss_main": 0.03760344535112381,
      "step": 13920
    },
    {
      "epoch": 1.6300023402761525,
      "grad_norm": 0.47743353247642517,
      "learning_rate": 9.120006485897824e-05,
      "loss": 0.0729,
      "step": 13930
    },
    {
      "epoch": 1.6300023402761525,
      "loss_main": 0.04389933869242668,
      "step": 13930
    },
    {
      "epoch": 1.6311724783524455,
      "grad_norm": 0.3805467486381531,
      "learning_rate": 9.118444466849152e-05,
      "loss": 0.0791,
      "step": 13940
    },
    {
      "epoch": 1.6311724783524455,
      "loss_main": 0.09185489267110825,
      "step": 13940
    },
    {
      "epoch": 1.6323426164287387,
      "grad_norm": 0.5116806626319885,
      "learning_rate": 9.116881196725793e-05,
      "loss": 0.0772,
      "step": 13950
    },
    {
      "epoch": 1.6323426164287387,
      "loss_main": 0.08831635862588882,
      "step": 13950
    },
    {
      "epoch": 1.6335127545050316,
      "grad_norm": 0.39685630798339844,
      "learning_rate": 9.115316676002627e-05,
      "loss": 0.0757,
      "step": 13960
    },
    {
      "epoch": 1.6335127545050316,
      "loss_main": 0.10867532342672348,
      "step": 13960
    },
    {
      "epoch": 1.6346828925813246,
      "grad_norm": 0.29700374603271484,
      "learning_rate": 9.113750905154911e-05,
      "loss": 0.0841,
      "step": 13970
    },
    {
      "epoch": 1.6346828925813246,
      "loss_main": 0.08470217883586884,
      "step": 13970
    },
    {
      "epoch": 1.6358530306576176,
      "grad_norm": 0.3246474266052246,
      "learning_rate": 9.112183884658289e-05,
      "loss": 0.0899,
      "step": 13980
    },
    {
      "epoch": 1.6358530306576176,
      "loss_main": 0.06325317919254303,
      "step": 13980
    },
    {
      "epoch": 1.6370231687339105,
      "grad_norm": 0.2511451840400696,
      "learning_rate": 9.11061561498878e-05,
      "loss": 0.0726,
      "step": 13990
    },
    {
      "epoch": 1.6370231687339105,
      "loss_main": 0.08540232479572296,
      "step": 13990
    },
    {
      "epoch": 1.6381933068102037,
      "grad_norm": 0.3358708322048187,
      "learning_rate": 9.109046096622779e-05,
      "loss": 0.0896,
      "step": 14000
    },
    {
      "epoch": 1.6381933068102037,
      "loss_main": 0.08030863851308823,
      "step": 14000
    },
    {
      "epoch": 1.6393634448864967,
      "grad_norm": 0.4012344777584076,
      "learning_rate": 9.107475330037069e-05,
      "loss": 0.083,
      "step": 14010
    },
    {
      "epoch": 1.6393634448864967,
      "loss_main": 0.07893218100070953,
      "step": 14010
    },
    {
      "epoch": 1.6405335829627896,
      "grad_norm": 0.35836735367774963,
      "learning_rate": 9.105903315708806e-05,
      "loss": 0.0773,
      "step": 14020
    },
    {
      "epoch": 1.6405335829627896,
      "loss_main": 0.0633540004491806,
      "step": 14020
    },
    {
      "epoch": 1.6417037210390826,
      "grad_norm": 0.34190064668655396,
      "learning_rate": 9.104330054115524e-05,
      "loss": 0.0749,
      "step": 14030
    },
    {
      "epoch": 1.6417037210390826,
      "loss_main": 0.058220867067575455,
      "step": 14030
    },
    {
      "epoch": 1.6428738591153755,
      "grad_norm": 0.3155153691768646,
      "learning_rate": 9.102755545735141e-05,
      "loss": 0.0654,
      "step": 14040
    },
    {
      "epoch": 1.6428738591153755,
      "loss_main": 0.09084539115428925,
      "step": 14040
    },
    {
      "epoch": 1.6440439971916687,
      "grad_norm": 0.3000566363334656,
      "learning_rate": 9.10117979104595e-05,
      "loss": 0.0754,
      "step": 14050
    },
    {
      "epoch": 1.6440439971916687,
      "loss_main": 0.07544194161891937,
      "step": 14050
    },
    {
      "epoch": 1.6452141352679617,
      "grad_norm": 0.3609359562397003,
      "learning_rate": 9.099602790526624e-05,
      "loss": 0.0726,
      "step": 14060
    },
    {
      "epoch": 1.6452141352679617,
      "loss_main": 0.0761018618941307,
      "step": 14060
    },
    {
      "epoch": 1.6463842733442546,
      "grad_norm": 0.35953283309936523,
      "learning_rate": 9.098024544656212e-05,
      "loss": 0.0727,
      "step": 14070
    },
    {
      "epoch": 1.6463842733442546,
      "loss_main": 0.12095999717712402,
      "step": 14070
    },
    {
      "epoch": 1.6475544114205476,
      "grad_norm": 0.40341413021087646,
      "learning_rate": 9.096445053914148e-05,
      "loss": 0.0829,
      "step": 14080
    },
    {
      "epoch": 1.6475544114205476,
      "loss_main": 0.0631270781159401,
      "step": 14080
    },
    {
      "epoch": 1.6487245494968406,
      "grad_norm": 0.23419742286205292,
      "learning_rate": 9.094864318780236e-05,
      "loss": 0.0816,
      "step": 14090
    },
    {
      "epoch": 1.6487245494968406,
      "loss_main": 0.11687677353620529,
      "step": 14090
    },
    {
      "epoch": 1.6498946875731337,
      "grad_norm": 0.2718046307563782,
      "learning_rate": 9.093282339734663e-05,
      "loss": 0.0911,
      "step": 14100
    },
    {
      "epoch": 1.6498946875731337,
      "loss_main": 0.08409984409809113,
      "step": 14100
    },
    {
      "epoch": 1.6510648256494265,
      "grad_norm": 0.4866313934326172,
      "learning_rate": 9.091699117257992e-05,
      "loss": 0.0689,
      "step": 14110
    },
    {
      "epoch": 1.6510648256494265,
      "loss_main": 0.10749823600053787,
      "step": 14110
    },
    {
      "epoch": 1.6522349637257197,
      "grad_norm": 0.42372778058052063,
      "learning_rate": 9.090114651831163e-05,
      "loss": 0.0788,
      "step": 14120
    },
    {
      "epoch": 1.6522349637257197,
      "loss_main": 0.10447553545236588,
      "step": 14120
    },
    {
      "epoch": 1.6534051018020126,
      "grad_norm": 0.39330393075942993,
      "learning_rate": 9.088528943935497e-05,
      "loss": 0.0872,
      "step": 14130
    },
    {
      "epoch": 1.6534051018020126,
      "loss_main": 0.08001457899808884,
      "step": 14130
    },
    {
      "epoch": 1.6545752398783056,
      "grad_norm": 0.419142484664917,
      "learning_rate": 9.086941994052689e-05,
      "loss": 0.0769,
      "step": 14140
    },
    {
      "epoch": 1.6545752398783056,
      "loss_main": 0.07702092826366425,
      "step": 14140
    },
    {
      "epoch": 1.6557453779545988,
      "grad_norm": 0.34441685676574707,
      "learning_rate": 9.085353802664813e-05,
      "loss": 0.0698,
      "step": 14150
    },
    {
      "epoch": 1.6557453779545988,
      "loss_main": 0.061783190816640854,
      "step": 14150
    },
    {
      "epoch": 1.6569155160308915,
      "grad_norm": 0.3384557068347931,
      "learning_rate": 9.08376437025432e-05,
      "loss": 0.0795,
      "step": 14160
    },
    {
      "epoch": 1.6569155160308915,
      "loss_main": 0.043485336005687714,
      "step": 14160
    },
    {
      "epoch": 1.6580856541071847,
      "grad_norm": 0.24646133184432983,
      "learning_rate": 9.082173697304035e-05,
      "loss": 0.0808,
      "step": 14170
    },
    {
      "epoch": 1.6580856541071847,
      "loss_main": 0.08756452053785324,
      "step": 14170
    },
    {
      "epoch": 1.6592557921834776,
      "grad_norm": 0.3325705826282501,
      "learning_rate": 9.080581784297166e-05,
      "loss": 0.0762,
      "step": 14180
    },
    {
      "epoch": 1.6592557921834776,
      "loss_main": 0.10856673866510391,
      "step": 14180
    },
    {
      "epoch": 1.6604259302597706,
      "grad_norm": 0.4055921733379364,
      "learning_rate": 9.078988631717291e-05,
      "loss": 0.0835,
      "step": 14190
    },
    {
      "epoch": 1.6604259302597706,
      "loss_main": 0.07554423809051514,
      "step": 14190
    },
    {
      "epoch": 1.6615960683360638,
      "grad_norm": 0.4690997004508972,
      "learning_rate": 9.077394240048369e-05,
      "loss": 0.0751,
      "step": 14200
    },
    {
      "epoch": 1.6615960683360638,
      "loss_main": 0.0527375303208828,
      "step": 14200
    },
    {
      "epoch": 1.6627662064123565,
      "grad_norm": 0.27256903052330017,
      "learning_rate": 9.075798609774736e-05,
      "loss": 0.078,
      "step": 14210
    },
    {
      "epoch": 1.6627662064123565,
      "loss_main": 0.060610800981521606,
      "step": 14210
    },
    {
      "epoch": 1.6639363444886497,
      "grad_norm": 0.28712955117225647,
      "learning_rate": 9.0742017413811e-05,
      "loss": 0.0748,
      "step": 14220
    },
    {
      "epoch": 1.6639363444886497,
      "loss_main": 0.08073297888040543,
      "step": 14220
    },
    {
      "epoch": 1.6651064825649426,
      "grad_norm": 0.2505243420600891,
      "learning_rate": 9.072603635352548e-05,
      "loss": 0.0737,
      "step": 14230
    },
    {
      "epoch": 1.6651064825649426,
      "loss_main": 0.06861738115549088,
      "step": 14230
    },
    {
      "epoch": 1.6662766206412356,
      "grad_norm": 0.34316015243530273,
      "learning_rate": 9.071004292174541e-05,
      "loss": 0.0868,
      "step": 14240
    },
    {
      "epoch": 1.6662766206412356,
      "loss_main": 0.06530941277742386,
      "step": 14240
    },
    {
      "epoch": 1.6674467587175288,
      "grad_norm": 0.32026106119155884,
      "learning_rate": 9.06940371233292e-05,
      "loss": 0.0702,
      "step": 14250
    },
    {
      "epoch": 1.6674467587175288,
      "loss_main": 0.05608990043401718,
      "step": 14250
    },
    {
      "epoch": 1.6686168967938215,
      "grad_norm": 0.2651287615299225,
      "learning_rate": 9.067801896313898e-05,
      "loss": 0.0773,
      "step": 14260
    },
    {
      "epoch": 1.6686168967938215,
      "loss_main": 0.047781553119421005,
      "step": 14260
    },
    {
      "epoch": 1.6697870348701147,
      "grad_norm": 0.27155008912086487,
      "learning_rate": 9.066198844604064e-05,
      "loss": 0.0782,
      "step": 14270
    },
    {
      "epoch": 1.6697870348701147,
      "loss_main": 0.07653535157442093,
      "step": 14270
    },
    {
      "epoch": 1.6709571729464077,
      "grad_norm": 0.30029749870300293,
      "learning_rate": 9.06459455769038e-05,
      "loss": 0.0796,
      "step": 14280
    },
    {
      "epoch": 1.6709571729464077,
      "loss_main": 0.10474669933319092,
      "step": 14280
    },
    {
      "epoch": 1.6721273110227006,
      "grad_norm": 0.3094475567340851,
      "learning_rate": 9.062989036060193e-05,
      "loss": 0.0933,
      "step": 14290
    },
    {
      "epoch": 1.6721273110227006,
      "loss_main": 0.07936745136976242,
      "step": 14290
    },
    {
      "epoch": 1.6732974490989938,
      "grad_norm": 0.2804245352745056,
      "learning_rate": 9.061382280201212e-05,
      "loss": 0.0702,
      "step": 14300
    },
    {
      "epoch": 1.6732974490989938,
      "loss_main": 0.15286299586296082,
      "step": 14300
    },
    {
      "epoch": 1.6744675871752865,
      "grad_norm": 0.3061234951019287,
      "learning_rate": 9.059774290601528e-05,
      "loss": 0.0913,
      "step": 14310
    },
    {
      "epoch": 1.6744675871752865,
      "loss_main": 0.08382908999919891,
      "step": 14310
    },
    {
      "epoch": 1.6756377252515797,
      "grad_norm": 0.40740108489990234,
      "learning_rate": 9.058165067749606e-05,
      "loss": 0.0739,
      "step": 14320
    },
    {
      "epoch": 1.6756377252515797,
      "loss_main": 0.09408436715602875,
      "step": 14320
    },
    {
      "epoch": 1.6768078633278727,
      "grad_norm": 0.4024834632873535,
      "learning_rate": 9.056554612134288e-05,
      "loss": 0.0829,
      "step": 14330
    },
    {
      "epoch": 1.6768078633278727,
      "loss_main": 0.0693327859044075,
      "step": 14330
    },
    {
      "epoch": 1.6779780014041656,
      "grad_norm": 0.37730664014816284,
      "learning_rate": 9.054942924244785e-05,
      "loss": 0.0801,
      "step": 14340
    },
    {
      "epoch": 1.6779780014041656,
      "loss_main": 0.05450287461280823,
      "step": 14340
    },
    {
      "epoch": 1.6791481394804588,
      "grad_norm": 0.5350123047828674,
      "learning_rate": 9.053330004570686e-05,
      "loss": 0.0841,
      "step": 14350
    },
    {
      "epoch": 1.6791481394804588,
      "loss_main": 0.0770917534828186,
      "step": 14350
    },
    {
      "epoch": 1.6803182775567516,
      "grad_norm": 0.38933777809143066,
      "learning_rate": 9.051715853601955e-05,
      "loss": 0.0793,
      "step": 14360
    },
    {
      "epoch": 1.6803182775567516,
      "loss_main": 0.08783727884292603,
      "step": 14360
    },
    {
      "epoch": 1.6814884156330447,
      "grad_norm": 0.27043965458869934,
      "learning_rate": 9.050100471828926e-05,
      "loss": 0.084,
      "step": 14370
    },
    {
      "epoch": 1.6814884156330447,
      "loss_main": 0.08131217211484909,
      "step": 14370
    },
    {
      "epoch": 1.6826585537093377,
      "grad_norm": 0.2912229895591736,
      "learning_rate": 9.048483859742311e-05,
      "loss": 0.0837,
      "step": 14380
    },
    {
      "epoch": 1.6826585537093377,
      "loss_main": 0.0670272707939148,
      "step": 14380
    },
    {
      "epoch": 1.6838286917856307,
      "grad_norm": 0.3689119815826416,
      "learning_rate": 9.046866017833193e-05,
      "loss": 0.0758,
      "step": 14390
    },
    {
      "epoch": 1.6838286917856307,
      "loss_main": 0.06921248883008957,
      "step": 14390
    },
    {
      "epoch": 1.6849988298619238,
      "grad_norm": 0.4024425446987152,
      "learning_rate": 9.045246946593029e-05,
      "loss": 0.0797,
      "step": 14400
    },
    {
      "epoch": 1.6849988298619238,
      "loss_main": 0.06465626507997513,
      "step": 14400
    },
    {
      "epoch": 1.6861689679382166,
      "grad_norm": 0.4094381034374237,
      "learning_rate": 9.043626646513652e-05,
      "loss": 0.0737,
      "step": 14410
    },
    {
      "epoch": 1.6861689679382166,
      "loss_main": 0.07133354246616364,
      "step": 14410
    },
    {
      "epoch": 1.6873391060145098,
      "grad_norm": 0.2886447608470917,
      "learning_rate": 9.042005118087267e-05,
      "loss": 0.0744,
      "step": 14420
    },
    {
      "epoch": 1.6873391060145098,
      "loss_main": 0.08681833744049072,
      "step": 14420
    },
    {
      "epoch": 1.6885092440908027,
      "grad_norm": 0.4057755172252655,
      "learning_rate": 9.040382361806448e-05,
      "loss": 0.0881,
      "step": 14430
    },
    {
      "epoch": 1.6885092440908027,
      "loss_main": 0.10679329186677933,
      "step": 14430
    },
    {
      "epoch": 1.6896793821670957,
      "grad_norm": 0.2661786675453186,
      "learning_rate": 9.038758378164148e-05,
      "loss": 0.0868,
      "step": 14440
    },
    {
      "epoch": 1.6896793821670957,
      "loss_main": 0.06019771844148636,
      "step": 14440
    },
    {
      "epoch": 1.6908495202433889,
      "grad_norm": 0.39810529351234436,
      "learning_rate": 9.037133167653691e-05,
      "loss": 0.0774,
      "step": 14450
    },
    {
      "epoch": 1.6908495202433889,
      "loss_main": 0.08089765161275864,
      "step": 14450
    },
    {
      "epoch": 1.6920196583196816,
      "grad_norm": 0.32644885778427124,
      "learning_rate": 9.035506730768771e-05,
      "loss": 0.0734,
      "step": 14460
    },
    {
      "epoch": 1.6920196583196816,
      "loss_main": 0.06866186857223511,
      "step": 14460
    },
    {
      "epoch": 1.6931897963959748,
      "grad_norm": 0.5289462804794312,
      "learning_rate": 9.033879068003458e-05,
      "loss": 0.0675,
      "step": 14470
    },
    {
      "epoch": 1.6931897963959748,
      "loss_main": 0.0676201805472374,
      "step": 14470
    },
    {
      "epoch": 1.6943599344722677,
      "grad_norm": 0.32982292771339417,
      "learning_rate": 9.032250179852193e-05,
      "loss": 0.0783,
      "step": 14480
    },
    {
      "epoch": 1.6943599344722677,
      "loss_main": 0.1253172904253006,
      "step": 14480
    },
    {
      "epoch": 1.6955300725485607,
      "grad_norm": 0.5885764360427856,
      "learning_rate": 9.030620066809787e-05,
      "loss": 0.0757,
      "step": 14490
    },
    {
      "epoch": 1.6955300725485607,
      "loss_main": 0.10252535343170166,
      "step": 14490
    },
    {
      "epoch": 1.6967002106248539,
      "grad_norm": 0.31146976351737976,
      "learning_rate": 9.028988729371428e-05,
      "loss": 0.0772,
      "step": 14500
    },
    {
      "epoch": 1.6967002106248539,
      "loss_main": 0.06850182265043259,
      "step": 14500
    },
    {
      "epoch": 1.6978703487011466,
      "grad_norm": 0.3093491494655609,
      "learning_rate": 9.027356168032673e-05,
      "loss": 0.0749,
      "step": 14510
    },
    {
      "epoch": 1.6978703487011466,
      "loss_main": 0.06428109854459763,
      "step": 14510
    },
    {
      "epoch": 1.6990404867774398,
      "grad_norm": 0.35144466161727905,
      "learning_rate": 9.02572238328945e-05,
      "loss": 0.072,
      "step": 14520
    },
    {
      "epoch": 1.6990404867774398,
      "loss_main": 0.08521995693445206,
      "step": 14520
    },
    {
      "epoch": 1.7002106248537328,
      "grad_norm": 0.31012994050979614,
      "learning_rate": 9.02408737563806e-05,
      "loss": 0.078,
      "step": 14530
    },
    {
      "epoch": 1.7002106248537328,
      "loss_main": 0.09802629053592682,
      "step": 14530
    },
    {
      "epoch": 1.7013807629300257,
      "grad_norm": 0.2780067026615143,
      "learning_rate": 9.022451145575174e-05,
      "loss": 0.0769,
      "step": 14540
    },
    {
      "epoch": 1.7013807629300257,
      "loss_main": 0.11292976886034012,
      "step": 14540
    },
    {
      "epoch": 1.702550901006319,
      "grad_norm": 0.3281385004520416,
      "learning_rate": 9.02081369359784e-05,
      "loss": 0.0784,
      "step": 14550
    },
    {
      "epoch": 1.702550901006319,
      "loss_main": 0.048930663615465164,
      "step": 14550
    },
    {
      "epoch": 1.7037210390826116,
      "grad_norm": 0.36288538575172424,
      "learning_rate": 9.019175020203465e-05,
      "loss": 0.0791,
      "step": 14560
    },
    {
      "epoch": 1.7037210390826116,
      "loss_main": 0.06527875363826752,
      "step": 14560
    },
    {
      "epoch": 1.7048911771589048,
      "grad_norm": 0.4456743001937866,
      "learning_rate": 9.017535125889842e-05,
      "loss": 0.0754,
      "step": 14570
    },
    {
      "epoch": 1.7048911771589048,
      "loss_main": 0.07881396263837814,
      "step": 14570
    },
    {
      "epoch": 1.7060613152351978,
      "grad_norm": 0.285995751619339,
      "learning_rate": 9.015894011155124e-05,
      "loss": 0.0793,
      "step": 14580
    },
    {
      "epoch": 1.7060613152351978,
      "loss_main": 0.0982128158211708,
      "step": 14580
    },
    {
      "epoch": 1.7072314533114907,
      "grad_norm": 0.3261463940143585,
      "learning_rate": 9.014251676497838e-05,
      "loss": 0.0881,
      "step": 14590
    },
    {
      "epoch": 1.7072314533114907,
      "loss_main": 0.04029588773846626,
      "step": 14590
    },
    {
      "epoch": 1.708401591387784,
      "grad_norm": 0.3241274952888489,
      "learning_rate": 9.012608122416884e-05,
      "loss": 0.0735,
      "step": 14600
    },
    {
      "epoch": 1.708401591387784,
      "loss_main": 0.048880696296691895,
      "step": 14600
    },
    {
      "epoch": 1.7095717294640767,
      "grad_norm": 0.3248339891433716,
      "learning_rate": 9.010963349411529e-05,
      "loss": 0.0797,
      "step": 14610
    },
    {
      "epoch": 1.7095717294640767,
      "loss_main": 0.09209438413381577,
      "step": 14610
    },
    {
      "epoch": 1.7107418675403698,
      "grad_norm": 0.4395468235015869,
      "learning_rate": 9.00931735798141e-05,
      "loss": 0.0753,
      "step": 14620
    },
    {
      "epoch": 1.7107418675403698,
      "loss_main": 0.051277559250593185,
      "step": 14620
    },
    {
      "epoch": 1.7119120056166628,
      "grad_norm": 0.40779781341552734,
      "learning_rate": 9.00767014862654e-05,
      "loss": 0.0776,
      "step": 14630
    },
    {
      "epoch": 1.7119120056166628,
      "loss_main": 0.060133833438158035,
      "step": 14630
    },
    {
      "epoch": 1.7130821436929557,
      "grad_norm": 0.3123552203178406,
      "learning_rate": 9.006021721847295e-05,
      "loss": 0.0771,
      "step": 14640
    },
    {
      "epoch": 1.7130821436929557,
      "loss_main": 0.08852683007717133,
      "step": 14640
    },
    {
      "epoch": 1.714252281769249,
      "grad_norm": 0.3486958146095276,
      "learning_rate": 9.004372078144423e-05,
      "loss": 0.0782,
      "step": 14650
    },
    {
      "epoch": 1.714252281769249,
      "loss_main": 0.08992143720388412,
      "step": 14650
    },
    {
      "epoch": 1.7154224198455417,
      "grad_norm": 0.36490604281425476,
      "learning_rate": 9.002721218019043e-05,
      "loss": 0.0904,
      "step": 14660
    },
    {
      "epoch": 1.7154224198455417,
      "loss_main": 0.08368849009275436,
      "step": 14660
    },
    {
      "epoch": 1.7165925579218348,
      "grad_norm": 0.4069819152355194,
      "learning_rate": 9.001069141972642e-05,
      "loss": 0.0772,
      "step": 14670
    },
    {
      "epoch": 1.7165925579218348,
      "loss_main": 0.06611157208681107,
      "step": 14670
    },
    {
      "epoch": 1.7177626959981278,
      "grad_norm": 0.3784506618976593,
      "learning_rate": 8.99941585050708e-05,
      "loss": 0.079,
      "step": 14680
    },
    {
      "epoch": 1.7177626959981278,
      "loss_main": 0.06284813582897186,
      "step": 14680
    },
    {
      "epoch": 1.7189328340744208,
      "grad_norm": 0.36063310503959656,
      "learning_rate": 8.997761344124578e-05,
      "loss": 0.076,
      "step": 14690
    },
    {
      "epoch": 1.7189328340744208,
      "loss_main": 0.060709841549396515,
      "step": 14690
    },
    {
      "epoch": 1.720102972150714,
      "grad_norm": 0.2782735824584961,
      "learning_rate": 8.996105623327737e-05,
      "loss": 0.0791,
      "step": 14700
    },
    {
      "epoch": 1.720102972150714,
      "loss_main": 0.04606562852859497,
      "step": 14700
    },
    {
      "epoch": 1.7212731102270067,
      "grad_norm": 0.28183743357658386,
      "learning_rate": 8.994448688619517e-05,
      "loss": 0.0665,
      "step": 14710
    },
    {
      "epoch": 1.7212731102270067,
      "loss_main": 0.1329512894153595,
      "step": 14710
    },
    {
      "epoch": 1.7224432483032999,
      "grad_norm": 0.35149797797203064,
      "learning_rate": 8.992790540503253e-05,
      "loss": 0.076,
      "step": 14720
    },
    {
      "epoch": 1.7224432483032999,
      "loss_main": 0.08390495926141739,
      "step": 14720
    },
    {
      "epoch": 1.7236133863795928,
      "grad_norm": 0.2614721357822418,
      "learning_rate": 8.991131179482648e-05,
      "loss": 0.0741,
      "step": 14730
    },
    {
      "epoch": 1.7236133863795928,
      "loss_main": 0.05225716158747673,
      "step": 14730
    },
    {
      "epoch": 1.7247835244558858,
      "grad_norm": 0.5353416800498962,
      "learning_rate": 8.989470606061768e-05,
      "loss": 0.0832,
      "step": 14740
    },
    {
      "epoch": 1.7247835244558858,
      "loss_main": 0.10346544533967972,
      "step": 14740
    },
    {
      "epoch": 1.7259536625321787,
      "grad_norm": 0.2983313500881195,
      "learning_rate": 8.987808820745056e-05,
      "loss": 0.0867,
      "step": 14750
    },
    {
      "epoch": 1.7259536625321787,
      "loss_main": 0.047835495322942734,
      "step": 14750
    },
    {
      "epoch": 1.7271238006084717,
      "grad_norm": 0.4332236051559448,
      "learning_rate": 8.986145824037315e-05,
      "loss": 0.0706,
      "step": 14760
    },
    {
      "epoch": 1.7271238006084717,
      "loss_main": 0.08873607963323593,
      "step": 14760
    },
    {
      "epoch": 1.7282939386847649,
      "grad_norm": 0.3533863127231598,
      "learning_rate": 8.984481616443721e-05,
      "loss": 0.0835,
      "step": 14770
    },
    {
      "epoch": 1.7282939386847649,
      "loss_main": 0.07777874171733856,
      "step": 14770
    },
    {
      "epoch": 1.7294640767610578,
      "grad_norm": 0.35268011689186096,
      "learning_rate": 8.982816198469815e-05,
      "loss": 0.07,
      "step": 14780
    },
    {
      "epoch": 1.7294640767610578,
      "loss_main": 0.06578925251960754,
      "step": 14780
    },
    {
      "epoch": 1.7306342148373508,
      "grad_norm": 0.3308732211589813,
      "learning_rate": 8.98114957062151e-05,
      "loss": 0.082,
      "step": 14790
    },
    {
      "epoch": 1.7306342148373508,
      "loss_main": 0.08254612237215042,
      "step": 14790
    },
    {
      "epoch": 1.7318043529136438,
      "grad_norm": 0.44248396158218384,
      "learning_rate": 8.97948173340508e-05,
      "loss": 0.0847,
      "step": 14800
    },
    {
      "epoch": 1.7318043529136438,
      "loss_main": 0.0552581362426281,
      "step": 14800
    },
    {
      "epoch": 1.7329744909899367,
      "grad_norm": 0.40250182151794434,
      "learning_rate": 8.977812687327172e-05,
      "loss": 0.068,
      "step": 14810
    },
    {
      "epoch": 1.7329744909899367,
      "loss_main": 0.11595027893781662,
      "step": 14810
    },
    {
      "epoch": 1.73414462906623,
      "grad_norm": 0.273619681596756,
      "learning_rate": 8.976142432894798e-05,
      "loss": 0.075,
      "step": 14820
    },
    {
      "epoch": 1.73414462906623,
      "loss_main": 0.06790899485349655,
      "step": 14820
    },
    {
      "epoch": 1.7353147671425229,
      "grad_norm": 0.38443320989608765,
      "learning_rate": 8.974470970615336e-05,
      "loss": 0.083,
      "step": 14830
    },
    {
      "epoch": 1.7353147671425229,
      "loss_main": 0.11775029450654984,
      "step": 14830
    },
    {
      "epoch": 1.7364849052188158,
      "grad_norm": 0.3203258514404297,
      "learning_rate": 8.972798300996534e-05,
      "loss": 0.0781,
      "step": 14840
    },
    {
      "epoch": 1.7364849052188158,
      "loss_main": 0.05204855650663376,
      "step": 14840
    },
    {
      "epoch": 1.7376550432951088,
      "grad_norm": 0.35819193720817566,
      "learning_rate": 8.971124424546504e-05,
      "loss": 0.0713,
      "step": 14850
    },
    {
      "epoch": 1.7376550432951088,
      "loss_main": 0.05080868676304817,
      "step": 14850
    },
    {
      "epoch": 1.7388251813714017,
      "grad_norm": 0.3089061379432678,
      "learning_rate": 8.969449341773724e-05,
      "loss": 0.0799,
      "step": 14860
    },
    {
      "epoch": 1.7388251813714017,
      "loss_main": 0.08790486305952072,
      "step": 14860
    },
    {
      "epoch": 1.739995319447695,
      "grad_norm": 0.22207506000995636,
      "learning_rate": 8.967773053187042e-05,
      "loss": 0.0736,
      "step": 14870
    },
    {
      "epoch": 1.739995319447695,
      "loss_main": 0.0571795292198658,
      "step": 14870
    },
    {
      "epoch": 1.7411654575239879,
      "grad_norm": 0.3076718747615814,
      "learning_rate": 8.966095559295668e-05,
      "loss": 0.0757,
      "step": 14880
    },
    {
      "epoch": 1.7411654575239879,
      "loss_main": 0.07009682804346085,
      "step": 14880
    },
    {
      "epoch": 1.7423355956002808,
      "grad_norm": 0.40100955963134766,
      "learning_rate": 8.964416860609184e-05,
      "loss": 0.0833,
      "step": 14890
    },
    {
      "epoch": 1.7423355956002808,
      "loss_main": 0.07895665615797043,
      "step": 14890
    },
    {
      "epoch": 1.7435057336765738,
      "grad_norm": 0.34795573353767395,
      "learning_rate": 8.962736957637532e-05,
      "loss": 0.0668,
      "step": 14900
    },
    {
      "epoch": 1.7435057336765738,
      "loss_main": 0.10575810819864273,
      "step": 14900
    },
    {
      "epoch": 1.7446758717528668,
      "grad_norm": 0.23639950156211853,
      "learning_rate": 8.96105585089102e-05,
      "loss": 0.0673,
      "step": 14910
    },
    {
      "epoch": 1.7446758717528668,
      "loss_main": 0.14817452430725098,
      "step": 14910
    },
    {
      "epoch": 1.74584600982916,
      "grad_norm": 0.5600408911705017,
      "learning_rate": 8.959373540880329e-05,
      "loss": 0.0906,
      "step": 14920
    },
    {
      "epoch": 1.74584600982916,
      "loss_main": 0.11463238298892975,
      "step": 14920
    },
    {
      "epoch": 1.747016147905453,
      "grad_norm": 0.38130009174346924,
      "learning_rate": 8.957690028116495e-05,
      "loss": 0.0864,
      "step": 14930
    },
    {
      "epoch": 1.747016147905453,
      "loss_main": 0.05227337405085564,
      "step": 14930
    },
    {
      "epoch": 1.7481862859817459,
      "grad_norm": 0.366315096616745,
      "learning_rate": 8.956005313110928e-05,
      "loss": 0.0773,
      "step": 14940
    },
    {
      "epoch": 1.7481862859817459,
      "loss_main": 0.06985753029584885,
      "step": 14940
    },
    {
      "epoch": 1.7493564240580388,
      "grad_norm": 0.5307254195213318,
      "learning_rate": 8.9543193963754e-05,
      "loss": 0.0842,
      "step": 14950
    },
    {
      "epoch": 1.7493564240580388,
      "loss_main": 0.07249245792627335,
      "step": 14950
    },
    {
      "epoch": 1.7505265621343318,
      "grad_norm": 0.34202563762664795,
      "learning_rate": 8.952632278422048e-05,
      "loss": 0.0742,
      "step": 14960
    },
    {
      "epoch": 1.7505265621343318,
      "loss_main": 0.08371826261281967,
      "step": 14960
    },
    {
      "epoch": 1.751696700210625,
      "grad_norm": 0.28613924980163574,
      "learning_rate": 8.95094395976337e-05,
      "loss": 0.0906,
      "step": 14970
    },
    {
      "epoch": 1.751696700210625,
      "loss_main": 0.0926927849650383,
      "step": 14970
    },
    {
      "epoch": 1.7528668382869177,
      "grad_norm": 0.23909243941307068,
      "learning_rate": 8.949254440912239e-05,
      "loss": 0.0772,
      "step": 14980
    },
    {
      "epoch": 1.7528668382869177,
      "loss_main": 0.07420703023672104,
      "step": 14980
    },
    {
      "epoch": 1.7540369763632109,
      "grad_norm": 0.5279362201690674,
      "learning_rate": 8.94756372238188e-05,
      "loss": 0.0822,
      "step": 14990
    },
    {
      "epoch": 1.7540369763632109,
      "loss_main": 0.07278193533420563,
      "step": 14990
    },
    {
      "epoch": 1.7552071144395038,
      "grad_norm": 0.24671250581741333,
      "learning_rate": 8.945871804685892e-05,
      "loss": 0.0784,
      "step": 15000
    },
    {
      "epoch": 1.7552071144395038,
      "loss_main": 0.10003747045993805,
      "step": 15000
    },
    {
      "epoch": 1.7563772525157968,
      "grad_norm": 0.2837744355201721,
      "learning_rate": 8.944178688338236e-05,
      "loss": 0.0685,
      "step": 15010
    },
    {
      "epoch": 1.7563772525157968,
      "loss_main": 0.06999620050191879,
      "step": 15010
    },
    {
      "epoch": 1.75754739059209,
      "grad_norm": 0.4012641906738281,
      "learning_rate": 8.942484373853233e-05,
      "loss": 0.0681,
      "step": 15020
    },
    {
      "epoch": 1.75754739059209,
      "loss_main": 0.07281966507434845,
      "step": 15020
    },
    {
      "epoch": 1.7587175286683827,
      "grad_norm": 0.43079277873039246,
      "learning_rate": 8.940788861745572e-05,
      "loss": 0.0795,
      "step": 15030
    },
    {
      "epoch": 1.7587175286683827,
      "loss_main": 0.07409165799617767,
      "step": 15030
    },
    {
      "epoch": 1.759887666744676,
      "grad_norm": 0.42616456747055054,
      "learning_rate": 8.939092152530308e-05,
      "loss": 0.0715,
      "step": 15040
    },
    {
      "epoch": 1.759887666744676,
      "loss_main": 0.08446138352155685,
      "step": 15040
    },
    {
      "epoch": 1.7610578048209689,
      "grad_norm": 0.2820614278316498,
      "learning_rate": 8.937394246722853e-05,
      "loss": 0.0736,
      "step": 15050
    },
    {
      "epoch": 1.7610578048209689,
      "loss_main": 0.09418057650327682,
      "step": 15050
    },
    {
      "epoch": 1.7622279428972618,
      "grad_norm": 0.290821373462677,
      "learning_rate": 8.935695144838984e-05,
      "loss": 0.0861,
      "step": 15060
    },
    {
      "epoch": 1.7622279428972618,
      "loss_main": 0.08570253103971481,
      "step": 15060
    },
    {
      "epoch": 1.763398080973555,
      "grad_norm": 0.38326916098594666,
      "learning_rate": 8.933994847394849e-05,
      "loss": 0.0803,
      "step": 15070
    },
    {
      "epoch": 1.763398080973555,
      "loss_main": 0.0752754881978035,
      "step": 15070
    },
    {
      "epoch": 1.7645682190498477,
      "grad_norm": 0.3057630956172943,
      "learning_rate": 8.932293354906949e-05,
      "loss": 0.077,
      "step": 15080
    },
    {
      "epoch": 1.7645682190498477,
      "loss_main": 0.08142169564962387,
      "step": 15080
    },
    {
      "epoch": 1.765738357126141,
      "grad_norm": 0.43112698197364807,
      "learning_rate": 8.930590667892153e-05,
      "loss": 0.075,
      "step": 15090
    },
    {
      "epoch": 1.765738357126141,
      "loss_main": 0.07369117438793182,
      "step": 15090
    },
    {
      "epoch": 1.7669084952024339,
      "grad_norm": 0.33732253313064575,
      "learning_rate": 8.928886786867696e-05,
      "loss": 0.0747,
      "step": 15100
    },
    {
      "epoch": 1.7669084952024339,
      "loss_main": 0.05279666185379028,
      "step": 15100
    },
    {
      "epoch": 1.7680786332787268,
      "grad_norm": 0.32295408844947815,
      "learning_rate": 8.927181712351168e-05,
      "loss": 0.0686,
      "step": 15110
    },
    {
      "epoch": 1.7680786332787268,
      "loss_main": 0.040593989193439484,
      "step": 15110
    },
    {
      "epoch": 1.76924877135502,
      "grad_norm": 0.29997700452804565,
      "learning_rate": 8.925475444860527e-05,
      "loss": 0.0799,
      "step": 15120
    },
    {
      "epoch": 1.76924877135502,
      "loss_main": 0.061396487057209015,
      "step": 15120
    },
    {
      "epoch": 1.7704189094313127,
      "grad_norm": 0.316161185503006,
      "learning_rate": 8.923767984914092e-05,
      "loss": 0.0676,
      "step": 15130
    },
    {
      "epoch": 1.7704189094313127,
      "loss_main": 0.076495461165905,
      "step": 15130
    },
    {
      "epoch": 1.771589047507606,
      "grad_norm": 0.42953965067863464,
      "learning_rate": 8.922059333030545e-05,
      "loss": 0.0832,
      "step": 15140
    },
    {
      "epoch": 1.771589047507606,
      "loss_main": 0.039980556815862656,
      "step": 15140
    },
    {
      "epoch": 1.7727591855838989,
      "grad_norm": 0.508928656578064,
      "learning_rate": 8.920349489728928e-05,
      "loss": 0.0749,
      "step": 15150
    },
    {
      "epoch": 1.7727591855838989,
      "loss_main": 0.04733632132411003,
      "step": 15150
    },
    {
      "epoch": 1.7739293236601918,
      "grad_norm": 0.36751726269721985,
      "learning_rate": 8.918638455528646e-05,
      "loss": 0.0825,
      "step": 15160
    },
    {
      "epoch": 1.7739293236601918,
      "loss_main": 0.061351388692855835,
      "step": 15160
    },
    {
      "epoch": 1.775099461736485,
      "grad_norm": 0.3995213806629181,
      "learning_rate": 8.916926230949468e-05,
      "loss": 0.078,
      "step": 15170
    },
    {
      "epoch": 1.775099461736485,
      "loss_main": 0.07294193655252457,
      "step": 15170
    },
    {
      "epoch": 1.7762695998127778,
      "grad_norm": 0.6000197529792786,
      "learning_rate": 8.915212816511522e-05,
      "loss": 0.0843,
      "step": 15180
    },
    {
      "epoch": 1.7762695998127778,
      "loss_main": 0.05276079848408699,
      "step": 15180
    },
    {
      "epoch": 1.777439737889071,
      "grad_norm": 0.4817545413970947,
      "learning_rate": 8.913498212735296e-05,
      "loss": 0.0749,
      "step": 15190
    },
    {
      "epoch": 1.777439737889071,
      "loss_main": 0.0666772723197937,
      "step": 15190
    },
    {
      "epoch": 1.778609875965364,
      "grad_norm": 0.3817254900932312,
      "learning_rate": 8.911782420141643e-05,
      "loss": 0.0823,
      "step": 15200
    },
    {
      "epoch": 1.778609875965364,
      "loss_main": 0.06619484722614288,
      "step": 15200
    },
    {
      "epoch": 1.7797800140416569,
      "grad_norm": 0.4271787405014038,
      "learning_rate": 8.910065439251775e-05,
      "loss": 0.0731,
      "step": 15210
    },
    {
      "epoch": 1.7797800140416569,
      "loss_main": 0.07322365790605545,
      "step": 15210
    },
    {
      "epoch": 1.78095015211795,
      "grad_norm": 0.3632958233356476,
      "learning_rate": 8.908347270587268e-05,
      "loss": 0.077,
      "step": 15220
    },
    {
      "epoch": 1.78095015211795,
      "loss_main": 0.05500265955924988,
      "step": 15220
    },
    {
      "epoch": 1.7821202901942428,
      "grad_norm": 0.29034650325775146,
      "learning_rate": 8.906627914670054e-05,
      "loss": 0.075,
      "step": 15230
    },
    {
      "epoch": 1.7821202901942428,
      "loss_main": 0.05260007828474045,
      "step": 15230
    },
    {
      "epoch": 1.783290428270536,
      "grad_norm": 0.28262630105018616,
      "learning_rate": 8.904907372022427e-05,
      "loss": 0.0705,
      "step": 15240
    },
    {
      "epoch": 1.783290428270536,
      "loss_main": 0.09397274255752563,
      "step": 15240
    },
    {
      "epoch": 1.784460566346829,
      "grad_norm": 0.5144508481025696,
      "learning_rate": 8.903185643167042e-05,
      "loss": 0.0807,
      "step": 15250
    },
    {
      "epoch": 1.784460566346829,
      "loss_main": 0.05214962363243103,
      "step": 15250
    },
    {
      "epoch": 1.7856307044231219,
      "grad_norm": 0.3790561258792877,
      "learning_rate": 8.901462728626919e-05,
      "loss": 0.0724,
      "step": 15260
    },
    {
      "epoch": 1.7856307044231219,
      "loss_main": 0.09720546752214432,
      "step": 15260
    },
    {
      "epoch": 1.786800842499415,
      "grad_norm": 0.3303239345550537,
      "learning_rate": 8.899738628925429e-05,
      "loss": 0.0787,
      "step": 15270
    },
    {
      "epoch": 1.786800842499415,
      "loss_main": 0.0848962664604187,
      "step": 15270
    },
    {
      "epoch": 1.7879709805757078,
      "grad_norm": 0.34887948632240295,
      "learning_rate": 8.898013344586312e-05,
      "loss": 0.0714,
      "step": 15280
    },
    {
      "epoch": 1.7879709805757078,
      "loss_main": 0.12875573337078094,
      "step": 15280
    },
    {
      "epoch": 1.789141118652001,
      "grad_norm": 0.29979345202445984,
      "learning_rate": 8.896286876133661e-05,
      "loss": 0.0809,
      "step": 15290
    },
    {
      "epoch": 1.789141118652001,
      "loss_main": 0.05257514491677284,
      "step": 15290
    },
    {
      "epoch": 1.790311256728294,
      "grad_norm": 0.33228635787963867,
      "learning_rate": 8.894559224091933e-05,
      "loss": 0.0733,
      "step": 15300
    },
    {
      "epoch": 1.790311256728294,
      "loss_main": 0.11694718897342682,
      "step": 15300
    },
    {
      "epoch": 1.791481394804587,
      "grad_norm": 0.46157947182655334,
      "learning_rate": 8.892830388985942e-05,
      "loss": 0.08,
      "step": 15310
    },
    {
      "epoch": 1.791481394804587,
      "loss_main": 0.07575174421072006,
      "step": 15310
    },
    {
      "epoch": 1.79265153288088,
      "grad_norm": 0.25022196769714355,
      "learning_rate": 8.891100371340864e-05,
      "loss": 0.0708,
      "step": 15320
    },
    {
      "epoch": 1.79265153288088,
      "loss_main": 0.10970007628202438,
      "step": 15320
    },
    {
      "epoch": 1.7938216709571728,
      "grad_norm": 0.3364503085613251,
      "learning_rate": 8.889369171682231e-05,
      "loss": 0.0789,
      "step": 15330
    },
    {
      "epoch": 1.7938216709571728,
      "loss_main": 0.07896021753549576,
      "step": 15330
    },
    {
      "epoch": 1.794991809033466,
      "grad_norm": 0.2256230115890503,
      "learning_rate": 8.887636790535936e-05,
      "loss": 0.0672,
      "step": 15340
    },
    {
      "epoch": 1.794991809033466,
      "loss_main": 0.050887200981378555,
      "step": 15340
    },
    {
      "epoch": 1.796161947109759,
      "grad_norm": 0.3152996599674225,
      "learning_rate": 8.885903228428231e-05,
      "loss": 0.074,
      "step": 15350
    },
    {
      "epoch": 1.796161947109759,
      "loss_main": 0.07099544256925583,
      "step": 15350
    },
    {
      "epoch": 1.797332085186052,
      "grad_norm": 0.26219940185546875,
      "learning_rate": 8.884168485885727e-05,
      "loss": 0.0696,
      "step": 15360
    },
    {
      "epoch": 1.797332085186052,
      "loss_main": 0.0899985209107399,
      "step": 15360
    },
    {
      "epoch": 1.798502223262345,
      "grad_norm": 0.2692568004131317,
      "learning_rate": 8.882432563435393e-05,
      "loss": 0.0782,
      "step": 15370
    },
    {
      "epoch": 1.798502223262345,
      "loss_main": 0.07933259010314941,
      "step": 15370
    },
    {
      "epoch": 1.7996723613386378,
      "grad_norm": 0.3113274574279785,
      "learning_rate": 8.880695461604556e-05,
      "loss": 0.0675,
      "step": 15380
    },
    {
      "epoch": 1.7996723613386378,
      "loss_main": 0.09011590480804443,
      "step": 15380
    },
    {
      "epoch": 1.800842499414931,
      "grad_norm": 0.2645576596260071,
      "learning_rate": 8.878957180920901e-05,
      "loss": 0.0726,
      "step": 15390
    },
    {
      "epoch": 1.800842499414931,
      "loss_main": 0.08234257996082306,
      "step": 15390
    },
    {
      "epoch": 1.802012637491224,
      "grad_norm": 0.2962978184223175,
      "learning_rate": 8.877217721912473e-05,
      "loss": 0.0824,
      "step": 15400
    },
    {
      "epoch": 1.802012637491224,
      "loss_main": 0.07750222831964493,
      "step": 15400
    },
    {
      "epoch": 1.803182775567517,
      "grad_norm": 0.3069166839122772,
      "learning_rate": 8.875477085107673e-05,
      "loss": 0.0748,
      "step": 15410
    },
    {
      "epoch": 1.803182775567517,
      "loss_main": 0.05922112986445427,
      "step": 15410
    },
    {
      "epoch": 1.8043529136438101,
      "grad_norm": 0.3318839967250824,
      "learning_rate": 8.87373527103526e-05,
      "loss": 0.0735,
      "step": 15420
    },
    {
      "epoch": 1.8043529136438101,
      "loss_main": 0.08861499279737473,
      "step": 15420
    },
    {
      "epoch": 1.8055230517201029,
      "grad_norm": 0.2718890905380249,
      "learning_rate": 8.871992280224353e-05,
      "loss": 0.0727,
      "step": 15430
    },
    {
      "epoch": 1.8055230517201029,
      "loss_main": 0.06109059602022171,
      "step": 15430
    },
    {
      "epoch": 1.806693189796396,
      "grad_norm": 0.25386595726013184,
      "learning_rate": 8.870248113204422e-05,
      "loss": 0.0739,
      "step": 15440
    },
    {
      "epoch": 1.806693189796396,
      "loss_main": 0.07847339659929276,
      "step": 15440
    },
    {
      "epoch": 1.807863327872689,
      "grad_norm": 0.2983347475528717,
      "learning_rate": 8.868502770505306e-05,
      "loss": 0.0871,
      "step": 15450
    },
    {
      "epoch": 1.807863327872689,
      "loss_main": 0.07023435086011887,
      "step": 15450
    },
    {
      "epoch": 1.809033465948982,
      "grad_norm": 0.3979358375072479,
      "learning_rate": 8.86675625265719e-05,
      "loss": 0.0807,
      "step": 15460
    },
    {
      "epoch": 1.809033465948982,
      "loss_main": 0.07541503757238388,
      "step": 15460
    },
    {
      "epoch": 1.8102036040252751,
      "grad_norm": 0.37471139430999756,
      "learning_rate": 8.865008560190618e-05,
      "loss": 0.0672,
      "step": 15470
    },
    {
      "epoch": 1.8102036040252751,
      "loss_main": 0.12450876086950302,
      "step": 15470
    },
    {
      "epoch": 1.8113737421015679,
      "grad_norm": 0.28771889209747314,
      "learning_rate": 8.863259693636496e-05,
      "loss": 0.0735,
      "step": 15480
    },
    {
      "epoch": 1.8113737421015679,
      "loss_main": 0.060375213623046875,
      "step": 15480
    },
    {
      "epoch": 1.812543880177861,
      "grad_norm": 0.4249318838119507,
      "learning_rate": 8.861509653526083e-05,
      "loss": 0.0747,
      "step": 15490
    },
    {
      "epoch": 1.812543880177861,
      "loss_main": 0.08521335572004318,
      "step": 15490
    },
    {
      "epoch": 1.813714018254154,
      "grad_norm": 0.2523898482322693,
      "learning_rate": 8.859758440390993e-05,
      "loss": 0.0765,
      "step": 15500
    },
    {
      "epoch": 1.813714018254154,
      "loss_main": 0.06869926303625107,
      "step": 15500
    },
    {
      "epoch": 1.814884156330447,
      "grad_norm": 0.3427685499191284,
      "learning_rate": 8.858006054763202e-05,
      "loss": 0.075,
      "step": 15510
    },
    {
      "epoch": 1.814884156330447,
      "loss_main": 0.07297922670841217,
      "step": 15510
    },
    {
      "epoch": 1.8160542944067402,
      "grad_norm": 0.3273966312408447,
      "learning_rate": 8.856252497175035e-05,
      "loss": 0.0716,
      "step": 15520
    },
    {
      "epoch": 1.8160542944067402,
      "loss_main": 0.05761318653821945,
      "step": 15520
    },
    {
      "epoch": 1.817224432483033,
      "grad_norm": 0.3515723645687103,
      "learning_rate": 8.854497768159178e-05,
      "loss": 0.059,
      "step": 15530
    },
    {
      "epoch": 1.817224432483033,
      "loss_main": 0.06709285080432892,
      "step": 15530
    },
    {
      "epoch": 1.818394570559326,
      "grad_norm": 0.2660900950431824,
      "learning_rate": 8.852741868248671e-05,
      "loss": 0.0727,
      "step": 15540
    },
    {
      "epoch": 1.818394570559326,
      "loss_main": 0.12612520158290863,
      "step": 15540
    },
    {
      "epoch": 1.819564708635619,
      "grad_norm": 0.39942556619644165,
      "learning_rate": 8.85098479797691e-05,
      "loss": 0.0819,
      "step": 15550
    },
    {
      "epoch": 1.819564708635619,
      "loss_main": 0.08210713416337967,
      "step": 15550
    },
    {
      "epoch": 1.820734846711912,
      "grad_norm": 0.44942182302474976,
      "learning_rate": 8.849226557877646e-05,
      "loss": 0.0769,
      "step": 15560
    },
    {
      "epoch": 1.820734846711912,
      "loss_main": 0.09822438657283783,
      "step": 15560
    },
    {
      "epoch": 1.8219049847882052,
      "grad_norm": 0.28089967370033264,
      "learning_rate": 8.84746714848499e-05,
      "loss": 0.0688,
      "step": 15570
    },
    {
      "epoch": 1.8219049847882052,
      "loss_main": 0.05062168091535568,
      "step": 15570
    },
    {
      "epoch": 1.823075122864498,
      "grad_norm": 0.46840959787368774,
      "learning_rate": 8.845706570333397e-05,
      "loss": 0.0795,
      "step": 15580
    },
    {
      "epoch": 1.823075122864498,
      "loss_main": 0.07923641055822372,
      "step": 15580
    },
    {
      "epoch": 1.824245260940791,
      "grad_norm": 0.37616342306137085,
      "learning_rate": 8.84394482395769e-05,
      "loss": 0.0836,
      "step": 15590
    },
    {
      "epoch": 1.824245260940791,
      "loss_main": 0.05055531486868858,
      "step": 15590
    },
    {
      "epoch": 1.825415399017084,
      "grad_norm": 0.3255748152732849,
      "learning_rate": 8.842181909893038e-05,
      "loss": 0.0672,
      "step": 15600
    },
    {
      "epoch": 1.825415399017084,
      "loss_main": 0.08405602723360062,
      "step": 15600
    },
    {
      "epoch": 1.826585537093377,
      "grad_norm": 0.3208940923213959,
      "learning_rate": 8.840417828674969e-05,
      "loss": 0.0689,
      "step": 15610
    },
    {
      "epoch": 1.826585537093377,
      "loss_main": 0.0676688700914383,
      "step": 15610
    },
    {
      "epoch": 1.82775567516967,
      "grad_norm": 0.25032442808151245,
      "learning_rate": 8.838652580839364e-05,
      "loss": 0.0761,
      "step": 15620
    },
    {
      "epoch": 1.82775567516967,
      "loss_main": 0.0686279684305191,
      "step": 15620
    },
    {
      "epoch": 1.828925813245963,
      "grad_norm": 0.3723793923854828,
      "learning_rate": 8.836886166922458e-05,
      "loss": 0.0864,
      "step": 15630
    },
    {
      "epoch": 1.828925813245963,
      "loss_main": 0.04510677605867386,
      "step": 15630
    },
    {
      "epoch": 1.830095951322256,
      "grad_norm": 0.30155128240585327,
      "learning_rate": 8.835118587460844e-05,
      "loss": 0.0735,
      "step": 15640
    },
    {
      "epoch": 1.830095951322256,
      "loss_main": 0.07150519639253616,
      "step": 15640
    },
    {
      "epoch": 1.831266089398549,
      "grad_norm": 0.3579314053058624,
      "learning_rate": 8.83334984299146e-05,
      "loss": 0.081,
      "step": 15650
    },
    {
      "epoch": 1.831266089398549,
      "loss_main": 0.10379771143198013,
      "step": 15650
    },
    {
      "epoch": 1.832436227474842,
      "grad_norm": 0.2760959565639496,
      "learning_rate": 8.83157993405161e-05,
      "loss": 0.0852,
      "step": 15660
    },
    {
      "epoch": 1.832436227474842,
      "loss_main": 0.07233735173940659,
      "step": 15660
    },
    {
      "epoch": 1.833606365551135,
      "grad_norm": 0.2825183868408203,
      "learning_rate": 8.829808861178943e-05,
      "loss": 0.0764,
      "step": 15670
    },
    {
      "epoch": 1.833606365551135,
      "loss_main": 0.08227764815092087,
      "step": 15670
    },
    {
      "epoch": 1.834776503627428,
      "grad_norm": 0.33403024077415466,
      "learning_rate": 8.828036624911464e-05,
      "loss": 0.0784,
      "step": 15680
    },
    {
      "epoch": 1.834776503627428,
      "loss_main": 0.09366139024496078,
      "step": 15680
    },
    {
      "epoch": 1.8359466417037211,
      "grad_norm": 0.21883688867092133,
      "learning_rate": 8.826263225787532e-05,
      "loss": 0.0814,
      "step": 15690
    },
    {
      "epoch": 1.8359466417037211,
      "loss_main": 0.06688200682401657,
      "step": 15690
    },
    {
      "epoch": 1.837116779780014,
      "grad_norm": 0.2546077072620392,
      "learning_rate": 8.824488664345858e-05,
      "loss": 0.0709,
      "step": 15700
    },
    {
      "epoch": 1.837116779780014,
      "loss_main": 0.06843243539333344,
      "step": 15700
    },
    {
      "epoch": 1.838286917856307,
      "grad_norm": 0.2466701865196228,
      "learning_rate": 8.822712941125508e-05,
      "loss": 0.0682,
      "step": 15710
    },
    {
      "epoch": 1.838286917856307,
      "loss_main": 0.08761435747146606,
      "step": 15710
    },
    {
      "epoch": 1.8394570559326,
      "grad_norm": 0.3315846025943756,
      "learning_rate": 8.820936056665898e-05,
      "loss": 0.0749,
      "step": 15720
    },
    {
      "epoch": 1.8394570559326,
      "loss_main": 0.11477107554674149,
      "step": 15720
    },
    {
      "epoch": 1.840627194008893,
      "grad_norm": 0.32706645131111145,
      "learning_rate": 8.819158011506801e-05,
      "loss": 0.0715,
      "step": 15730
    },
    {
      "epoch": 1.840627194008893,
      "loss_main": 0.05720262974500656,
      "step": 15730
    },
    {
      "epoch": 1.8417973320851861,
      "grad_norm": 0.27720826864242554,
      "learning_rate": 8.81737880618834e-05,
      "loss": 0.0705,
      "step": 15740
    },
    {
      "epoch": 1.8417973320851861,
      "loss_main": 0.08053521066904068,
      "step": 15740
    },
    {
      "epoch": 1.842967470161479,
      "grad_norm": 0.30314281582832336,
      "learning_rate": 8.815598441250987e-05,
      "loss": 0.0812,
      "step": 15750
    },
    {
      "epoch": 1.842967470161479,
      "loss_main": 0.04184436798095703,
      "step": 15750
    },
    {
      "epoch": 1.844137608237772,
      "grad_norm": 0.3993373215198517,
      "learning_rate": 8.813816917235576e-05,
      "loss": 0.0712,
      "step": 15760
    },
    {
      "epoch": 1.844137608237772,
      "loss_main": 0.05922165885567665,
      "step": 15760
    },
    {
      "epoch": 1.845307746314065,
      "grad_norm": 0.4301815629005432,
      "learning_rate": 8.812034234683282e-05,
      "loss": 0.0742,
      "step": 15770
    },
    {
      "epoch": 1.845307746314065,
      "loss_main": 0.06714878976345062,
      "step": 15770
    },
    {
      "epoch": 1.846477884390358,
      "grad_norm": 0.34317120909690857,
      "learning_rate": 8.810250394135637e-05,
      "loss": 0.0757,
      "step": 15780
    },
    {
      "epoch": 1.846477884390358,
      "loss_main": 0.054553449153900146,
      "step": 15780
    },
    {
      "epoch": 1.8476480224666512,
      "grad_norm": 0.25522661209106445,
      "learning_rate": 8.808465396134529e-05,
      "loss": 0.0704,
      "step": 15790
    },
    {
      "epoch": 1.8476480224666512,
      "loss_main": 0.0770433098077774,
      "step": 15790
    },
    {
      "epoch": 1.8488181605429441,
      "grad_norm": 0.3683200776576996,
      "learning_rate": 8.806679241222189e-05,
      "loss": 0.0712,
      "step": 15800
    },
    {
      "epoch": 1.8488181605429441,
      "loss_main": 0.06553417444229126,
      "step": 15800
    },
    {
      "epoch": 1.849988298619237,
      "grad_norm": 0.28911125659942627,
      "learning_rate": 8.804891929941203e-05,
      "loss": 0.0879,
      "step": 15810
    },
    {
      "epoch": 1.849988298619237,
      "loss_main": 0.11273106187582016,
      "step": 15810
    },
    {
      "epoch": 1.85115843669553,
      "grad_norm": 0.24110057950019836,
      "learning_rate": 8.803103462834514e-05,
      "loss": 0.0785,
      "step": 15820
    },
    {
      "epoch": 1.85115843669553,
      "loss_main": 0.11937449872493744,
      "step": 15820
    },
    {
      "epoch": 1.852328574771823,
      "grad_norm": 0.34427955746650696,
      "learning_rate": 8.801313840445408e-05,
      "loss": 0.0846,
      "step": 15830
    },
    {
      "epoch": 1.852328574771823,
      "loss_main": 0.05817528814077377,
      "step": 15830
    },
    {
      "epoch": 1.8534987128481162,
      "grad_norm": 0.3242066204547882,
      "learning_rate": 8.799523063317524e-05,
      "loss": 0.0773,
      "step": 15840
    },
    {
      "epoch": 1.8534987128481162,
      "loss_main": 0.06945806741714478,
      "step": 15840
    },
    {
      "epoch": 1.854668850924409,
      "grad_norm": 0.31184160709381104,
      "learning_rate": 8.797731131994854e-05,
      "loss": 0.0718,
      "step": 15850
    },
    {
      "epoch": 1.854668850924409,
      "loss_main": 0.07260880619287491,
      "step": 15850
    },
    {
      "epoch": 1.855838989000702,
      "grad_norm": 0.3125391900539398,
      "learning_rate": 8.795938047021739e-05,
      "loss": 0.0812,
      "step": 15860
    },
    {
      "epoch": 1.855838989000702,
      "loss_main": 0.05249400809407234,
      "step": 15860
    },
    {
      "epoch": 1.857009127076995,
      "grad_norm": 0.29641541838645935,
      "learning_rate": 8.794143808942872e-05,
      "loss": 0.0807,
      "step": 15870
    },
    {
      "epoch": 1.857009127076995,
      "loss_main": 0.04956953600049019,
      "step": 15870
    },
    {
      "epoch": 1.858179265153288,
      "grad_norm": 0.2769124209880829,
      "learning_rate": 8.792348418303296e-05,
      "loss": 0.074,
      "step": 15880
    },
    {
      "epoch": 1.858179265153288,
      "loss_main": 0.0776997059583664,
      "step": 15880
    },
    {
      "epoch": 1.8593494032295812,
      "grad_norm": 0.4092090427875519,
      "learning_rate": 8.790551875648398e-05,
      "loss": 0.0791,
      "step": 15890
    },
    {
      "epoch": 1.8593494032295812,
      "loss_main": 0.11789832264184952,
      "step": 15890
    },
    {
      "epoch": 1.860519541305874,
      "grad_norm": 0.3488411009311676,
      "learning_rate": 8.788754181523926e-05,
      "loss": 0.0864,
      "step": 15900
    },
    {
      "epoch": 1.860519541305874,
      "loss_main": 0.10901577025651932,
      "step": 15900
    },
    {
      "epoch": 1.8616896793821671,
      "grad_norm": 0.34108108282089233,
      "learning_rate": 8.78695533647597e-05,
      "loss": 0.085,
      "step": 15910
    },
    {
      "epoch": 1.8616896793821671,
      "loss_main": 0.05231398716568947,
      "step": 15910
    },
    {
      "epoch": 1.86285981745846,
      "grad_norm": 0.3097239136695862,
      "learning_rate": 8.785155341050972e-05,
      "loss": 0.0844,
      "step": 15920
    },
    {
      "epoch": 1.86285981745846,
      "loss_main": 0.08071977645158768,
      "step": 15920
    },
    {
      "epoch": 1.864029955534753,
      "grad_norm": 0.32607200741767883,
      "learning_rate": 8.783354195795721e-05,
      "loss": 0.0747,
      "step": 15930
    },
    {
      "epoch": 1.864029955534753,
      "loss_main": 0.08237653970718384,
      "step": 15930
    },
    {
      "epoch": 1.8652000936110462,
      "grad_norm": 0.3725930154323578,
      "learning_rate": 8.78155190125736e-05,
      "loss": 0.0892,
      "step": 15940
    },
    {
      "epoch": 1.8652000936110462,
      "loss_main": 0.10404590517282486,
      "step": 15940
    },
    {
      "epoch": 1.866370231687339,
      "grad_norm": 0.2927340865135193,
      "learning_rate": 8.779748457983378e-05,
      "loss": 0.0848,
      "step": 15950
    },
    {
      "epoch": 1.866370231687339,
      "loss_main": 0.07636581361293793,
      "step": 15950
    },
    {
      "epoch": 1.8675403697636321,
      "grad_norm": 0.328457236289978,
      "learning_rate": 8.777943866521612e-05,
      "loss": 0.0909,
      "step": 15960
    },
    {
      "epoch": 1.8675403697636321,
      "loss_main": 0.05968128517270088,
      "step": 15960
    },
    {
      "epoch": 1.868710507839925,
      "grad_norm": 0.34938111901283264,
      "learning_rate": 8.77613812742025e-05,
      "loss": 0.0723,
      "step": 15970
    },
    {
      "epoch": 1.868710507839925,
      "loss_main": 0.09928351640701294,
      "step": 15970
    },
    {
      "epoch": 1.869880645916218,
      "grad_norm": 0.3259514272212982,
      "learning_rate": 8.774331241227829e-05,
      "loss": 0.0778,
      "step": 15980
    },
    {
      "epoch": 1.869880645916218,
      "loss_main": 0.05264196917414665,
      "step": 15980
    },
    {
      "epoch": 1.8710507839925112,
      "grad_norm": 0.4218841791152954,
      "learning_rate": 8.772523208493232e-05,
      "loss": 0.0683,
      "step": 15990
    },
    {
      "epoch": 1.8710507839925112,
      "loss_main": 0.06457554548978806,
      "step": 15990
    },
    {
      "epoch": 1.872220922068804,
      "grad_norm": 0.26700466871261597,
      "learning_rate": 8.770714029765692e-05,
      "loss": 0.0849,
      "step": 16000
    },
    {
      "epoch": 1.872220922068804,
      "loss_main": 0.13579407334327698,
      "step": 16000
    },
    {
      "epoch": 1.8733910601450972,
      "grad_norm": 0.3363242447376251,
      "learning_rate": 8.768903705594789e-05,
      "loss": 0.0811,
      "step": 16010
    },
    {
      "epoch": 1.8733910601450972,
      "loss_main": 0.09197282046079636,
      "step": 16010
    },
    {
      "epoch": 1.8745611982213901,
      "grad_norm": 0.2749835252761841,
      "learning_rate": 8.767092236530453e-05,
      "loss": 0.0852,
      "step": 16020
    },
    {
      "epoch": 1.8745611982213901,
      "loss_main": 0.09556357562541962,
      "step": 16020
    },
    {
      "epoch": 1.875731336297683,
      "grad_norm": 0.3491138219833374,
      "learning_rate": 8.76527962312296e-05,
      "loss": 0.0821,
      "step": 16030
    },
    {
      "epoch": 1.875731336297683,
      "loss_main": 0.1059446707367897,
      "step": 16030
    },
    {
      "epoch": 1.8769014743739763,
      "grad_norm": 0.31420430541038513,
      "learning_rate": 8.763465865922934e-05,
      "loss": 0.0741,
      "step": 16040
    },
    {
      "epoch": 1.8769014743739763,
      "loss_main": 0.16499921679496765,
      "step": 16040
    },
    {
      "epoch": 1.878071612450269,
      "grad_norm": 0.28066083788871765,
      "learning_rate": 8.761650965481347e-05,
      "loss": 0.0791,
      "step": 16050
    },
    {
      "epoch": 1.878071612450269,
      "loss_main": 0.06668667495250702,
      "step": 16050
    },
    {
      "epoch": 1.8792417505265622,
      "grad_norm": 0.30190277099609375,
      "learning_rate": 8.759834922349516e-05,
      "loss": 0.0722,
      "step": 16060
    },
    {
      "epoch": 1.8792417505265622,
      "loss_main": 0.045260511338710785,
      "step": 16060
    },
    {
      "epoch": 1.8804118886028551,
      "grad_norm": 0.35465729236602783,
      "learning_rate": 8.758017737079108e-05,
      "loss": 0.075,
      "step": 16070
    },
    {
      "epoch": 1.8804118886028551,
      "loss_main": 0.10920672118663788,
      "step": 16070
    },
    {
      "epoch": 1.881582026679148,
      "grad_norm": 0.2882536053657532,
      "learning_rate": 8.756199410222137e-05,
      "loss": 0.0693,
      "step": 16080
    },
    {
      "epoch": 1.881582026679148,
      "loss_main": 0.0633198693394661,
      "step": 16080
    },
    {
      "epoch": 1.8827521647554413,
      "grad_norm": 0.4027620851993561,
      "learning_rate": 8.754379942330963e-05,
      "loss": 0.0791,
      "step": 16090
    },
    {
      "epoch": 1.8827521647554413,
      "loss_main": 0.05932182818651199,
      "step": 16090
    },
    {
      "epoch": 1.883922302831734,
      "grad_norm": 0.20688630640506744,
      "learning_rate": 8.75255933395829e-05,
      "loss": 0.0668,
      "step": 16100
    },
    {
      "epoch": 1.883922302831734,
      "loss_main": 0.06321929395198822,
      "step": 16100
    },
    {
      "epoch": 1.8850924409080272,
      "grad_norm": 0.3832525312900543,
      "learning_rate": 8.750737585657171e-05,
      "loss": 0.0743,
      "step": 16110
    },
    {
      "epoch": 1.8850924409080272,
      "loss_main": 0.0841943547129631,
      "step": 16110
    },
    {
      "epoch": 1.8862625789843201,
      "grad_norm": 0.3341427147388458,
      "learning_rate": 8.748914697981008e-05,
      "loss": 0.0689,
      "step": 16120
    },
    {
      "epoch": 1.8862625789843201,
      "loss_main": 0.06810449063777924,
      "step": 16120
    },
    {
      "epoch": 1.887432717060613,
      "grad_norm": 0.34033265709877014,
      "learning_rate": 8.747090671483542e-05,
      "loss": 0.075,
      "step": 16130
    },
    {
      "epoch": 1.887432717060613,
      "loss_main": 0.030991151928901672,
      "step": 16130
    },
    {
      "epoch": 1.8886028551369063,
      "grad_norm": 0.27475523948669434,
      "learning_rate": 8.745265506718869e-05,
      "loss": 0.0679,
      "step": 16140
    },
    {
      "epoch": 1.8886028551369063,
      "loss_main": 0.06546726077795029,
      "step": 16140
    },
    {
      "epoch": 1.889772993213199,
      "grad_norm": 0.3053528070449829,
      "learning_rate": 8.74343920424142e-05,
      "loss": 0.0731,
      "step": 16150
    },
    {
      "epoch": 1.889772993213199,
      "loss_main": 0.08336304128170013,
      "step": 16150
    },
    {
      "epoch": 1.8909431312894922,
      "grad_norm": 0.3511712849140167,
      "learning_rate": 8.741611764605982e-05,
      "loss": 0.0766,
      "step": 16160
    },
    {
      "epoch": 1.8909431312894922,
      "loss_main": 0.09809862077236176,
      "step": 16160
    },
    {
      "epoch": 1.8921132693657852,
      "grad_norm": 0.21707890927791595,
      "learning_rate": 8.739783188367682e-05,
      "loss": 0.0708,
      "step": 16170
    },
    {
      "epoch": 1.8921132693657852,
      "loss_main": 0.04013872891664505,
      "step": 16170
    },
    {
      "epoch": 1.8932834074420781,
      "grad_norm": 0.3741755187511444,
      "learning_rate": 8.737953476081991e-05,
      "loss": 0.0631,
      "step": 16180
    },
    {
      "epoch": 1.8932834074420781,
      "loss_main": 0.05715441331267357,
      "step": 16180
    },
    {
      "epoch": 1.8944535455183713,
      "grad_norm": 0.33737504482269287,
      "learning_rate": 8.73612262830473e-05,
      "loss": 0.0676,
      "step": 16190
    },
    {
      "epoch": 1.8944535455183713,
      "loss_main": 0.049946632236242294,
      "step": 16190
    },
    {
      "epoch": 1.895623683594664,
      "grad_norm": 0.3332037627696991,
      "learning_rate": 8.734290645592061e-05,
      "loss": 0.0665,
      "step": 16200
    },
    {
      "epoch": 1.895623683594664,
      "loss_main": 0.08374714851379395,
      "step": 16200
    },
    {
      "epoch": 1.8967938216709572,
      "grad_norm": 0.37880924344062805,
      "learning_rate": 8.732457528500493e-05,
      "loss": 0.0774,
      "step": 16210
    },
    {
      "epoch": 1.8967938216709572,
      "loss_main": 0.08534115552902222,
      "step": 16210
    },
    {
      "epoch": 1.8979639597472502,
      "grad_norm": 0.3353297710418701,
      "learning_rate": 8.730623277586875e-05,
      "loss": 0.0757,
      "step": 16220
    },
    {
      "epoch": 1.8979639597472502,
      "loss_main": 0.058707159012556076,
      "step": 16220
    },
    {
      "epoch": 1.8991340978235431,
      "grad_norm": 0.32110702991485596,
      "learning_rate": 8.72878789340841e-05,
      "loss": 0.0756,
      "step": 16230
    },
    {
      "epoch": 1.8991340978235431,
      "loss_main": 0.052924852818250656,
      "step": 16230
    },
    {
      "epoch": 1.9003042358998363,
      "grad_norm": 0.2236434668302536,
      "learning_rate": 8.726951376522635e-05,
      "loss": 0.0809,
      "step": 16240
    },
    {
      "epoch": 1.9003042358998363,
      "loss_main": 0.052534349262714386,
      "step": 16240
    },
    {
      "epoch": 1.901474373976129,
      "grad_norm": 0.3063327670097351,
      "learning_rate": 8.725113727487435e-05,
      "loss": 0.0723,
      "step": 16250
    },
    {
      "epoch": 1.901474373976129,
      "loss_main": 0.06867650896310806,
      "step": 16250
    },
    {
      "epoch": 1.9026445120524222,
      "grad_norm": 0.3167077898979187,
      "learning_rate": 8.723274946861042e-05,
      "loss": 0.074,
      "step": 16260
    },
    {
      "epoch": 1.9026445120524222,
      "loss_main": 0.05226511508226395,
      "step": 16260
    },
    {
      "epoch": 1.9038146501287152,
      "grad_norm": 0.3069346845149994,
      "learning_rate": 8.721435035202026e-05,
      "loss": 0.0782,
      "step": 16270
    },
    {
      "epoch": 1.9038146501287152,
      "loss_main": 0.12009523808956146,
      "step": 16270
    },
    {
      "epoch": 1.9049847882050082,
      "grad_norm": 0.3777087926864624,
      "learning_rate": 8.719593993069306e-05,
      "loss": 0.0811,
      "step": 16280
    },
    {
      "epoch": 1.9049847882050082,
      "loss_main": 0.06602335721254349,
      "step": 16280
    },
    {
      "epoch": 1.9061549262813013,
      "grad_norm": 0.3966235816478729,
      "learning_rate": 8.717751821022139e-05,
      "loss": 0.0732,
      "step": 16290
    },
    {
      "epoch": 1.9061549262813013,
      "loss_main": 0.08033720403909683,
      "step": 16290
    },
    {
      "epoch": 1.907325064357594,
      "grad_norm": 0.4235604405403137,
      "learning_rate": 8.715908519620134e-05,
      "loss": 0.0729,
      "step": 16300
    },
    {
      "epoch": 1.907325064357594,
      "loss_main": 0.11843860149383545,
      "step": 16300
    },
    {
      "epoch": 1.9084952024338873,
      "grad_norm": 0.26987937092781067,
      "learning_rate": 8.71406408942323e-05,
      "loss": 0.0813,
      "step": 16310
    },
    {
      "epoch": 1.9084952024338873,
      "loss_main": 0.08508557826280594,
      "step": 16310
    },
    {
      "epoch": 1.9096653405101802,
      "grad_norm": 0.3059491515159607,
      "learning_rate": 8.712218530991723e-05,
      "loss": 0.0694,
      "step": 16320
    },
    {
      "epoch": 1.9096653405101802,
      "loss_main": 0.052830424159765244,
      "step": 16320
    },
    {
      "epoch": 1.9108354785864732,
      "grad_norm": 0.32423916459083557,
      "learning_rate": 8.710371844886241e-05,
      "loss": 0.071,
      "step": 16330
    },
    {
      "epoch": 1.9108354785864732,
      "loss_main": 0.1146896556019783,
      "step": 16330
    },
    {
      "epoch": 1.9120056166627664,
      "grad_norm": 0.35040685534477234,
      "learning_rate": 8.708524031667758e-05,
      "loss": 0.0745,
      "step": 16340
    },
    {
      "epoch": 1.9120056166627664,
      "loss_main": 0.08861254155635834,
      "step": 16340
    },
    {
      "epoch": 1.913175754739059,
      "grad_norm": 0.27120375633239746,
      "learning_rate": 8.706675091897592e-05,
      "loss": 0.0789,
      "step": 16350
    },
    {
      "epoch": 1.913175754739059,
      "loss_main": 0.09193646907806396,
      "step": 16350
    },
    {
      "epoch": 1.9143458928153523,
      "grad_norm": 0.30476802587509155,
      "learning_rate": 8.704825026137404e-05,
      "loss": 0.0727,
      "step": 16360
    },
    {
      "epoch": 1.9143458928153523,
      "loss_main": 0.060202453285455704,
      "step": 16360
    },
    {
      "epoch": 1.9155160308916452,
      "grad_norm": 0.40305161476135254,
      "learning_rate": 8.702973834949192e-05,
      "loss": 0.0828,
      "step": 16370
    },
    {
      "epoch": 1.9155160308916452,
      "loss_main": 0.05835001915693283,
      "step": 16370
    },
    {
      "epoch": 1.9166861689679382,
      "grad_norm": 0.2672179341316223,
      "learning_rate": 8.701121518895301e-05,
      "loss": 0.0744,
      "step": 16380
    },
    {
      "epoch": 1.9166861689679382,
      "loss_main": 0.07879584282636642,
      "step": 16380
    },
    {
      "epoch": 1.9178563070442314,
      "grad_norm": 0.27117857336997986,
      "learning_rate": 8.699268078538414e-05,
      "loss": 0.0891,
      "step": 16390
    },
    {
      "epoch": 1.9178563070442314,
      "loss_main": 0.06993158906698227,
      "step": 16390
    },
    {
      "epoch": 1.9190264451205241,
      "grad_norm": 0.3894394338130951,
      "learning_rate": 8.69741351444156e-05,
      "loss": 0.0731,
      "step": 16400
    },
    {
      "epoch": 1.9190264451205241,
      "loss_main": 0.06964781880378723,
      "step": 16400
    },
    {
      "epoch": 1.9201965831968173,
      "grad_norm": 0.3833184242248535,
      "learning_rate": 8.695557827168101e-05,
      "loss": 0.0722,
      "step": 16410
    },
    {
      "epoch": 1.9201965831968173,
      "loss_main": 0.05118824914097786,
      "step": 16410
    },
    {
      "epoch": 1.9213667212731103,
      "grad_norm": 0.3020156919956207,
      "learning_rate": 8.693701017281753e-05,
      "loss": 0.085,
      "step": 16420
    },
    {
      "epoch": 1.9213667212731103,
      "loss_main": 0.09310103952884674,
      "step": 16420
    },
    {
      "epoch": 1.9225368593494032,
      "grad_norm": 0.2826324701309204,
      "learning_rate": 8.691843085346563e-05,
      "loss": 0.0816,
      "step": 16430
    },
    {
      "epoch": 1.9225368593494032,
      "loss_main": 0.08248520642518997,
      "step": 16430
    },
    {
      "epoch": 1.9237069974256964,
      "grad_norm": 0.2882283627986908,
      "learning_rate": 8.689984031926919e-05,
      "loss": 0.0725,
      "step": 16440
    },
    {
      "epoch": 1.9237069974256964,
      "loss_main": 0.05811700224876404,
      "step": 16440
    },
    {
      "epoch": 1.9248771355019891,
      "grad_norm": 0.33455371856689453,
      "learning_rate": 8.688123857587555e-05,
      "loss": 0.0826,
      "step": 16450
    },
    {
      "epoch": 1.9248771355019891,
      "loss_main": 0.05503673106431961,
      "step": 16450
    },
    {
      "epoch": 1.9260472735782823,
      "grad_norm": 0.30649304389953613,
      "learning_rate": 8.686262562893544e-05,
      "loss": 0.0814,
      "step": 16460
    },
    {
      "epoch": 1.9260472735782823,
      "loss_main": 0.05827387049794197,
      "step": 16460
    },
    {
      "epoch": 1.9272174116545753,
      "grad_norm": 0.22498196363449097,
      "learning_rate": 8.684400148410294e-05,
      "loss": 0.0711,
      "step": 16470
    },
    {
      "epoch": 1.9272174116545753,
      "loss_main": 0.06911405175924301,
      "step": 16470
    },
    {
      "epoch": 1.9283875497308682,
      "grad_norm": 0.2629530429840088,
      "learning_rate": 8.682536614703562e-05,
      "loss": 0.0737,
      "step": 16480
    },
    {
      "epoch": 1.9283875497308682,
      "loss_main": 0.07147742807865143,
      "step": 16480
    },
    {
      "epoch": 1.9295576878071612,
      "grad_norm": 0.3370644152164459,
      "learning_rate": 8.680671962339437e-05,
      "loss": 0.0754,
      "step": 16490
    },
    {
      "epoch": 1.9295576878071612,
      "loss_main": 0.058498065918684006,
      "step": 16490
    },
    {
      "epoch": 1.9307278258834542,
      "grad_norm": 0.37092286348342896,
      "learning_rate": 8.678806191884352e-05,
      "loss": 0.0713,
      "step": 16500
    },
    {
      "epoch": 1.9307278258834542,
      "loss_main": 0.05873856693506241,
      "step": 16500
    },
    {
      "epoch": 1.9318979639597473,
      "grad_norm": 0.27297502756118774,
      "learning_rate": 8.67693930390508e-05,
      "loss": 0.0716,
      "step": 16510
    },
    {
      "epoch": 1.9318979639597473,
      "loss_main": 0.09231588989496231,
      "step": 16510
    },
    {
      "epoch": 1.9330681020360403,
      "grad_norm": 0.2733937203884125,
      "learning_rate": 8.67507129896873e-05,
      "loss": 0.0656,
      "step": 16520
    },
    {
      "epoch": 1.9330681020360403,
      "loss_main": 0.10160697996616364,
      "step": 16520
    },
    {
      "epoch": 1.9342382401123333,
      "grad_norm": 0.326345294713974,
      "learning_rate": 8.673202177642757e-05,
      "loss": 0.0857,
      "step": 16530
    },
    {
      "epoch": 1.9342382401123333,
      "loss_main": 0.10729773342609406,
      "step": 16530
    },
    {
      "epoch": 1.9354083781886262,
      "grad_norm": 0.26059308648109436,
      "learning_rate": 8.671331940494945e-05,
      "loss": 0.0814,
      "step": 16540
    },
    {
      "epoch": 1.9354083781886262,
      "loss_main": 0.08066270500421524,
      "step": 16540
    },
    {
      "epoch": 1.9365785162649192,
      "grad_norm": 0.3390713632106781,
      "learning_rate": 8.669460588093427e-05,
      "loss": 0.0815,
      "step": 16550
    },
    {
      "epoch": 1.9365785162649192,
      "loss_main": 0.10949452966451645,
      "step": 16550
    },
    {
      "epoch": 1.9377486543412124,
      "grad_norm": 0.29291442036628723,
      "learning_rate": 8.667588121006667e-05,
      "loss": 0.0693,
      "step": 16560
    },
    {
      "epoch": 1.9377486543412124,
      "loss_main": 0.16219905018806458,
      "step": 16560
    },
    {
      "epoch": 1.9389187924175053,
      "grad_norm": 0.30726152658462524,
      "learning_rate": 8.665714539803475e-05,
      "loss": 0.0851,
      "step": 16570
    },
    {
      "epoch": 1.9389187924175053,
      "loss_main": 0.04995763301849365,
      "step": 16570
    },
    {
      "epoch": 1.9400889304937983,
      "grad_norm": 0.2650098502635956,
      "learning_rate": 8.663839845052993e-05,
      "loss": 0.0741,
      "step": 16580
    },
    {
      "epoch": 1.9400889304937983,
      "loss_main": 0.06772994250059128,
      "step": 16580
    },
    {
      "epoch": 1.9412590685700912,
      "grad_norm": 0.3591599762439728,
      "learning_rate": 8.661964037324703e-05,
      "loss": 0.0714,
      "step": 16590
    },
    {
      "epoch": 1.9412590685700912,
      "loss_main": 0.07409529387950897,
      "step": 16590
    },
    {
      "epoch": 1.9424292066463842,
      "grad_norm": 0.2450903058052063,
      "learning_rate": 8.660087117188427e-05,
      "loss": 0.0667,
      "step": 16600
    },
    {
      "epoch": 1.9424292066463842,
      "loss_main": 0.057038988918066025,
      "step": 16600
    },
    {
      "epoch": 1.9435993447226774,
      "grad_norm": 0.3383219242095947,
      "learning_rate": 8.658209085214325e-05,
      "loss": 0.071,
      "step": 16610
    },
    {
      "epoch": 1.9435993447226774,
      "loss_main": 0.10330475121736526,
      "step": 16610
    },
    {
      "epoch": 1.9447694827989703,
      "grad_norm": 0.29931244254112244,
      "learning_rate": 8.656329941972891e-05,
      "loss": 0.0798,
      "step": 16620
    },
    {
      "epoch": 1.9447694827989703,
      "loss_main": 0.08805835247039795,
      "step": 16620
    },
    {
      "epoch": 1.9459396208752633,
      "grad_norm": 0.33232027292251587,
      "learning_rate": 8.654449688034963e-05,
      "loss": 0.079,
      "step": 16630
    },
    {
      "epoch": 1.9459396208752633,
      "loss_main": 0.08405323326587677,
      "step": 16630
    },
    {
      "epoch": 1.9471097589515562,
      "grad_norm": 0.31942468881607056,
      "learning_rate": 8.652568323971706e-05,
      "loss": 0.0763,
      "step": 16640
    },
    {
      "epoch": 1.9471097589515562,
      "loss_main": 0.06869488209486008,
      "step": 16640
    },
    {
      "epoch": 1.9482798970278492,
      "grad_norm": 0.3058222830295563,
      "learning_rate": 8.650685850354636e-05,
      "loss": 0.074,
      "step": 16650
    },
    {
      "epoch": 1.9482798970278492,
      "loss_main": 0.06538106501102448,
      "step": 16650
    },
    {
      "epoch": 1.9494500351041424,
      "grad_norm": 0.3018743097782135,
      "learning_rate": 8.648802267755593e-05,
      "loss": 0.0854,
      "step": 16660
    },
    {
      "epoch": 1.9494500351041424,
      "loss_main": 0.052008338272571564,
      "step": 16660
    },
    {
      "epoch": 1.9506201731804353,
      "grad_norm": 0.33171525597572327,
      "learning_rate": 8.646917576746764e-05,
      "loss": 0.079,
      "step": 16670
    },
    {
      "epoch": 1.9506201731804353,
      "loss_main": 0.04529546573758125,
      "step": 16670
    },
    {
      "epoch": 1.9517903112567283,
      "grad_norm": 0.33507874608039856,
      "learning_rate": 8.645031777900666e-05,
      "loss": 0.0711,
      "step": 16680
    },
    {
      "epoch": 1.9517903112567283,
      "loss_main": 0.08064735680818558,
      "step": 16680
    },
    {
      "epoch": 1.9529604493330213,
      "grad_norm": 0.37725338339805603,
      "learning_rate": 8.643144871790154e-05,
      "loss": 0.0825,
      "step": 16690
    },
    {
      "epoch": 1.9529604493330213,
      "loss_main": 0.056510355323553085,
      "step": 16690
    },
    {
      "epoch": 1.9541305874093142,
      "grad_norm": 0.3491358757019043,
      "learning_rate": 8.641256858988424e-05,
      "loss": 0.0732,
      "step": 16700
    },
    {
      "epoch": 1.9541305874093142,
      "loss_main": 0.044508397579193115,
      "step": 16700
    },
    {
      "epoch": 1.9553007254856074,
      "grad_norm": 0.466388463973999,
      "learning_rate": 8.639367740069e-05,
      "loss": 0.067,
      "step": 16710
    },
    {
      "epoch": 1.9553007254856074,
      "loss_main": 0.07081596553325653,
      "step": 16710
    },
    {
      "epoch": 1.9564708635619001,
      "grad_norm": 0.41360095143318176,
      "learning_rate": 8.63747751560575e-05,
      "loss": 0.082,
      "step": 16720
    },
    {
      "epoch": 1.9564708635619001,
      "loss_main": 0.05630258843302727,
      "step": 16720
    },
    {
      "epoch": 1.9576410016381933,
      "grad_norm": 0.34393128752708435,
      "learning_rate": 8.635586186172871e-05,
      "loss": 0.0689,
      "step": 16730
    },
    {
      "epoch": 1.9576410016381933,
      "loss_main": 0.06108919903635979,
      "step": 16730
    },
    {
      "epoch": 1.9588111397144863,
      "grad_norm": 0.3518317639827728,
      "learning_rate": 8.633693752344902e-05,
      "loss": 0.0718,
      "step": 16740
    },
    {
      "epoch": 1.9588111397144863,
      "loss_main": 0.049055177718400955,
      "step": 16740
    },
    {
      "epoch": 1.9599812777907792,
      "grad_norm": 0.2857545018196106,
      "learning_rate": 8.631800214696713e-05,
      "loss": 0.0708,
      "step": 16750
    },
    {
      "epoch": 1.9599812777907792,
      "loss_main": 0.07204748690128326,
      "step": 16750
    },
    {
      "epoch": 1.9611514158670724,
      "grad_norm": 0.21082563698291779,
      "learning_rate": 8.629905573803511e-05,
      "loss": 0.0719,
      "step": 16760
    },
    {
      "epoch": 1.9611514158670724,
      "loss_main": 0.11685250699520111,
      "step": 16760
    },
    {
      "epoch": 1.9623215539433652,
      "grad_norm": 0.27490752935409546,
      "learning_rate": 8.628009830240839e-05,
      "loss": 0.0795,
      "step": 16770
    },
    {
      "epoch": 1.9623215539433652,
      "loss_main": 0.05677191540598869,
      "step": 16770
    },
    {
      "epoch": 1.9634916920196583,
      "grad_norm": 0.23783136904239655,
      "learning_rate": 8.626112984584571e-05,
      "loss": 0.0789,
      "step": 16780
    },
    {
      "epoch": 1.9634916920196583,
      "loss_main": 0.04112616926431656,
      "step": 16780
    },
    {
      "epoch": 1.9646618300959513,
      "grad_norm": 0.2687076926231384,
      "learning_rate": 8.62421503741092e-05,
      "loss": 0.0744,
      "step": 16790
    },
    {
      "epoch": 1.9646618300959513,
      "loss_main": 0.04765245318412781,
      "step": 16790
    },
    {
      "epoch": 1.9658319681722443,
      "grad_norm": 0.3711089789867401,
      "learning_rate": 8.622315989296432e-05,
      "loss": 0.066,
      "step": 16800
    },
    {
      "epoch": 1.9658319681722443,
      "loss_main": 0.05524224787950516,
      "step": 16800
    },
    {
      "epoch": 1.9670021062485374,
      "grad_norm": 0.3638545870780945,
      "learning_rate": 8.62041584081799e-05,
      "loss": 0.0743,
      "step": 16810
    },
    {
      "epoch": 1.9670021062485374,
      "loss_main": 0.06507662683725357,
      "step": 16810
    },
    {
      "epoch": 1.9681722443248302,
      "grad_norm": 0.2919047474861145,
      "learning_rate": 8.618514592552807e-05,
      "loss": 0.0751,
      "step": 16820
    },
    {
      "epoch": 1.9681722443248302,
      "loss_main": 0.051371634006500244,
      "step": 16820
    },
    {
      "epoch": 1.9693423824011234,
      "grad_norm": 0.30674099922180176,
      "learning_rate": 8.616612245078431e-05,
      "loss": 0.0764,
      "step": 16830
    },
    {
      "epoch": 1.9693423824011234,
      "loss_main": 0.08568171411752701,
      "step": 16830
    },
    {
      "epoch": 1.9705125204774163,
      "grad_norm": 0.3242588937282562,
      "learning_rate": 8.614708798972746e-05,
      "loss": 0.0667,
      "step": 16840
    },
    {
      "epoch": 1.9705125204774163,
      "loss_main": 0.06450428068637848,
      "step": 16840
    },
    {
      "epoch": 1.9716826585537093,
      "grad_norm": 0.3673737049102783,
      "learning_rate": 8.61280425481397e-05,
      "loss": 0.0662,
      "step": 16850
    },
    {
      "epoch": 1.9716826585537093,
      "loss_main": 0.08802340179681778,
      "step": 16850
    },
    {
      "epoch": 1.9728527966300025,
      "grad_norm": 0.44329264760017395,
      "learning_rate": 8.61089861318065e-05,
      "loss": 0.0763,
      "step": 16860
    },
    {
      "epoch": 1.9728527966300025,
      "loss_main": 0.08749818056821823,
      "step": 16860
    },
    {
      "epoch": 1.9740229347062952,
      "grad_norm": 0.37840500473976135,
      "learning_rate": 8.608991874651673e-05,
      "loss": 0.0685,
      "step": 16870
    },
    {
      "epoch": 1.9740229347062952,
      "loss_main": 0.08955477178096771,
      "step": 16870
    },
    {
      "epoch": 1.9751930727825884,
      "grad_norm": 0.2763681411743164,
      "learning_rate": 8.607084039806255e-05,
      "loss": 0.0772,
      "step": 16880
    },
    {
      "epoch": 1.9751930727825884,
      "loss_main": 0.10106422007083893,
      "step": 16880
    },
    {
      "epoch": 1.9763632108588813,
      "grad_norm": 0.28364792466163635,
      "learning_rate": 8.605175109223944e-05,
      "loss": 0.0717,
      "step": 16890
    },
    {
      "epoch": 1.9763632108588813,
      "loss_main": 0.10291457176208496,
      "step": 16890
    },
    {
      "epoch": 1.9775333489351743,
      "grad_norm": 0.21235552430152893,
      "learning_rate": 8.603265083484624e-05,
      "loss": 0.0725,
      "step": 16900
    },
    {
      "epoch": 1.9775333489351743,
      "loss_main": 0.09732340276241302,
      "step": 16900
    },
    {
      "epoch": 1.9787034870114675,
      "grad_norm": 0.2869960069656372,
      "learning_rate": 8.60135396316851e-05,
      "loss": 0.0726,
      "step": 16910
    },
    {
      "epoch": 1.9787034870114675,
      "loss_main": 0.062253065407276154,
      "step": 16910
    },
    {
      "epoch": 1.9798736250877602,
      "grad_norm": 0.2514980137348175,
      "learning_rate": 8.599441748856152e-05,
      "loss": 0.0718,
      "step": 16920
    },
    {
      "epoch": 1.9798736250877602,
      "loss_main": 0.07786525785923004,
      "step": 16920
    },
    {
      "epoch": 1.9810437631640534,
      "grad_norm": 0.33734649419784546,
      "learning_rate": 8.597528441128427e-05,
      "loss": 0.0758,
      "step": 16930
    },
    {
      "epoch": 1.9810437631640534,
      "loss_main": 0.04520326107740402,
      "step": 16930
    },
    {
      "epoch": 1.9822139012403464,
      "grad_norm": 0.2742966413497925,
      "learning_rate": 8.595614040566549e-05,
      "loss": 0.0713,
      "step": 16940
    },
    {
      "epoch": 1.9822139012403464,
      "loss_main": 0.05884058400988579,
      "step": 16940
    },
    {
      "epoch": 1.9833840393166393,
      "grad_norm": 0.2128041535615921,
      "learning_rate": 8.593698547752063e-05,
      "loss": 0.068,
      "step": 16950
    },
    {
      "epoch": 1.9833840393166393,
      "loss_main": 0.07912708073854446,
      "step": 16950
    },
    {
      "epoch": 1.9845541773929325,
      "grad_norm": 0.3418958783149719,
      "learning_rate": 8.591781963266843e-05,
      "loss": 0.0938,
      "step": 16960
    },
    {
      "epoch": 1.9845541773929325,
      "loss_main": 0.048752643167972565,
      "step": 16960
    },
    {
      "epoch": 1.9857243154692252,
      "grad_norm": 0.32118499279022217,
      "learning_rate": 8.5898642876931e-05,
      "loss": 0.0681,
      "step": 16970
    },
    {
      "epoch": 1.9857243154692252,
      "loss_main": 0.11616071313619614,
      "step": 16970
    },
    {
      "epoch": 1.9868944535455184,
      "grad_norm": 0.304279625415802,
      "learning_rate": 8.587945521613369e-05,
      "loss": 0.0855,
      "step": 16980
    },
    {
      "epoch": 1.9868944535455184,
      "loss_main": 0.0667698010802269,
      "step": 16980
    },
    {
      "epoch": 1.9880645916218114,
      "grad_norm": 0.29060590267181396,
      "learning_rate": 8.586025665610524e-05,
      "loss": 0.0695,
      "step": 16990
    },
    {
      "epoch": 1.9880645916218114,
      "loss_main": 0.08428279310464859,
      "step": 16990
    },
    {
      "epoch": 1.9892347296981043,
      "grad_norm": 0.3155337870121002,
      "learning_rate": 8.584104720267765e-05,
      "loss": 0.0781,
      "step": 17000
    },
    {
      "epoch": 1.9892347296981043,
      "loss_main": 0.07834124565124512,
      "step": 17000
    },
    {
      "epoch": 1.9904048677743975,
      "grad_norm": 0.24871832132339478,
      "learning_rate": 8.582182686168625e-05,
      "loss": 0.0803,
      "step": 17010
    },
    {
      "epoch": 1.9904048677743975,
      "loss_main": 0.06273885816335678,
      "step": 17010
    },
    {
      "epoch": 1.9915750058506902,
      "grad_norm": 0.23726129531860352,
      "learning_rate": 8.580259563896967e-05,
      "loss": 0.0743,
      "step": 17020
    },
    {
      "epoch": 1.9915750058506902,
      "loss_main": 0.09056205302476883,
      "step": 17020
    },
    {
      "epoch": 1.9927451439269834,
      "grad_norm": 0.38992467522621155,
      "learning_rate": 8.578335354036983e-05,
      "loss": 0.0728,
      "step": 17030
    },
    {
      "epoch": 1.9927451439269834,
      "loss_main": 0.09667093306779861,
      "step": 17030
    },
    {
      "epoch": 1.9939152820032764,
      "grad_norm": 0.40923479199409485,
      "learning_rate": 8.576410057173201e-05,
      "loss": 0.0728,
      "step": 17040
    },
    {
      "epoch": 1.9939152820032764,
      "loss_main": 0.04981468245387077,
      "step": 17040
    },
    {
      "epoch": 1.9950854200795693,
      "grad_norm": 0.3180418312549591,
      "learning_rate": 8.574483673890474e-05,
      "loss": 0.0737,
      "step": 17050
    },
    {
      "epoch": 1.9950854200795693,
      "loss_main": 0.03892863169312477,
      "step": 17050
    },
    {
      "epoch": 1.9962555581558625,
      "grad_norm": 0.2915504574775696,
      "learning_rate": 8.572556204773983e-05,
      "loss": 0.0654,
      "step": 17060
    },
    {
      "epoch": 1.9962555581558625,
      "loss_main": 0.0629994124174118,
      "step": 17060
    },
    {
      "epoch": 1.9974256962321553,
      "grad_norm": 0.3241083323955536,
      "learning_rate": 8.570627650409246e-05,
      "loss": 0.0696,
      "step": 17070
    },
    {
      "epoch": 1.9974256962321553,
      "loss_main": 0.06399788707494736,
      "step": 17070
    },
    {
      "epoch": 1.9985958343084484,
      "grad_norm": 0.30875831842422485,
      "learning_rate": 8.568698011382107e-05,
      "loss": 0.0762,
      "step": 17080
    },
    {
      "epoch": 1.9985958343084484,
      "loss_main": 0.1242985725402832,
      "step": 17080
    },
    {
      "epoch": 1.9997659723847414,
      "grad_norm": 0.3918342888355255,
      "learning_rate": 8.566767288278738e-05,
      "loss": 0.0751,
      "step": 17090
    },
    {
      "epoch": 1.9997659723847414,
      "loss_main": 0.05791968107223511,
      "step": 17090
    },
    {
      "epoch": 2.0009361104610344,
      "grad_norm": 0.2921851575374603,
      "learning_rate": 8.56483548168564e-05,
      "loss": 0.0757,
      "step": 17100
    },
    {
      "epoch": 2.0009361104610344,
      "loss_main": 0.06738699227571487,
      "step": 17100
    },
    {
      "epoch": 2.0021062485373275,
      "grad_norm": 0.29935207962989807,
      "learning_rate": 8.562902592189648e-05,
      "loss": 0.0714,
      "step": 17110
    },
    {
      "epoch": 2.0021062485373275,
      "loss_main": 0.06488180160522461,
      "step": 17110
    },
    {
      "epoch": 2.0032763866136203,
      "grad_norm": 0.2258729189634323,
      "learning_rate": 8.560968620377921e-05,
      "loss": 0.0676,
      "step": 17120
    },
    {
      "epoch": 2.0032763866136203,
      "loss_main": 0.10998722165822983,
      "step": 17120
    },
    {
      "epoch": 2.0044465246899135,
      "grad_norm": 0.323182612657547,
      "learning_rate": 8.559033566837951e-05,
      "loss": 0.077,
      "step": 17130
    },
    {
      "epoch": 2.0044465246899135,
      "loss_main": 0.050165627151727676,
      "step": 17130
    },
    {
      "epoch": 2.005616662766206,
      "grad_norm": 0.33611878752708435,
      "learning_rate": 8.557097432157551e-05,
      "loss": 0.0805,
      "step": 17140
    },
    {
      "epoch": 2.005616662766206,
      "loss_main": 0.07452314347028732,
      "step": 17140
    },
    {
      "epoch": 2.0067868008424994,
      "grad_norm": 0.39647844433784485,
      "learning_rate": 8.555160216924872e-05,
      "loss": 0.0691,
      "step": 17150
    },
    {
      "epoch": 2.0067868008424994,
      "loss_main": 0.05393203720450401,
      "step": 17150
    },
    {
      "epoch": 2.0079569389187926,
      "grad_norm": 0.3120705485343933,
      "learning_rate": 8.55322192172839e-05,
      "loss": 0.0745,
      "step": 17160
    },
    {
      "epoch": 2.0079569389187926,
      "loss_main": 0.04391460865736008,
      "step": 17160
    },
    {
      "epoch": 2.0091270769950853,
      "grad_norm": 0.28913193941116333,
      "learning_rate": 8.551282547156902e-05,
      "loss": 0.0719,
      "step": 17170
    },
    {
      "epoch": 2.0091270769950853,
      "loss_main": 0.06282002478837967,
      "step": 17170
    },
    {
      "epoch": 2.0102972150713785,
      "grad_norm": 0.3287856876850128,
      "learning_rate": 8.549342093799544e-05,
      "loss": 0.0808,
      "step": 17180
    },
    {
      "epoch": 2.0102972150713785,
      "loss_main": 0.09049541503190994,
      "step": 17180
    },
    {
      "epoch": 2.0114673531476712,
      "grad_norm": 0.26825854182243347,
      "learning_rate": 8.547400562245773e-05,
      "loss": 0.085,
      "step": 17190
    },
    {
      "epoch": 2.0114673531476712,
      "loss_main": 0.07907702028751373,
      "step": 17190
    },
    {
      "epoch": 2.0126374912239644,
      "grad_norm": 0.31527388095855713,
      "learning_rate": 8.545457953085374e-05,
      "loss": 0.0741,
      "step": 17200
    },
    {
      "epoch": 2.0126374912239644,
      "loss_main": 0.0890299454331398,
      "step": 17200
    },
    {
      "epoch": 2.0138076293002576,
      "grad_norm": 0.4335208535194397,
      "learning_rate": 8.543514266908463e-05,
      "loss": 0.077,
      "step": 17210
    },
    {
      "epoch": 2.0138076293002576,
      "loss_main": 0.11462046951055527,
      "step": 17210
    },
    {
      "epoch": 2.0149777673765503,
      "grad_norm": 0.30495649576187134,
      "learning_rate": 8.541569504305478e-05,
      "loss": 0.0771,
      "step": 17220
    },
    {
      "epoch": 2.0149777673765503,
      "loss_main": 0.08283819258213043,
      "step": 17220
    },
    {
      "epoch": 2.0161479054528435,
      "grad_norm": 0.3267471194267273,
      "learning_rate": 8.539623665867187e-05,
      "loss": 0.078,
      "step": 17230
    },
    {
      "epoch": 2.0161479054528435,
      "loss_main": 0.11183477193117142,
      "step": 17230
    },
    {
      "epoch": 2.0173180435291362,
      "grad_norm": 0.2753521203994751,
      "learning_rate": 8.537676752184685e-05,
      "loss": 0.0768,
      "step": 17240
    },
    {
      "epoch": 2.0173180435291362,
      "loss_main": 0.08325718343257904,
      "step": 17240
    },
    {
      "epoch": 2.0184881816054294,
      "grad_norm": 0.2201641947031021,
      "learning_rate": 8.53572876384939e-05,
      "loss": 0.0716,
      "step": 17250
    },
    {
      "epoch": 2.0184881816054294,
      "loss_main": 0.06553046405315399,
      "step": 17250
    },
    {
      "epoch": 2.0196583196817226,
      "grad_norm": 0.30511289834976196,
      "learning_rate": 8.533779701453056e-05,
      "loss": 0.0766,
      "step": 17260
    },
    {
      "epoch": 2.0196583196817226,
      "loss_main": 0.09673730283975601,
      "step": 17260
    },
    {
      "epoch": 2.0208284577580153,
      "grad_norm": 0.2940027117729187,
      "learning_rate": 8.53182956558775e-05,
      "loss": 0.078,
      "step": 17270
    },
    {
      "epoch": 2.0208284577580153,
      "loss_main": 0.06904879212379456,
      "step": 17270
    },
    {
      "epoch": 2.0219985958343085,
      "grad_norm": 0.38083040714263916,
      "learning_rate": 8.529878356845877e-05,
      "loss": 0.0636,
      "step": 17280
    },
    {
      "epoch": 2.0219985958343085,
      "loss_main": 0.03842649236321449,
      "step": 17280
    },
    {
      "epoch": 2.0231687339106013,
      "grad_norm": 0.36780470609664917,
      "learning_rate": 8.527926075820158e-05,
      "loss": 0.0796,
      "step": 17290
    },
    {
      "epoch": 2.0231687339106013,
      "loss_main": 0.04078296199440956,
      "step": 17290
    },
    {
      "epoch": 2.0243388719868944,
      "grad_norm": 0.2984924912452698,
      "learning_rate": 8.525972723103648e-05,
      "loss": 0.0727,
      "step": 17300
    },
    {
      "epoch": 2.0243388719868944,
      "loss_main": 0.0867367833852768,
      "step": 17300
    },
    {
      "epoch": 2.0255090100631876,
      "grad_norm": 0.3696862459182739,
      "learning_rate": 8.524018299289722e-05,
      "loss": 0.0766,
      "step": 17310
    },
    {
      "epoch": 2.0255090100631876,
      "loss_main": 0.044016290456056595,
      "step": 17310
    },
    {
      "epoch": 2.0266791481394804,
      "grad_norm": 0.34281429648399353,
      "learning_rate": 8.522062804972083e-05,
      "loss": 0.0759,
      "step": 17320
    },
    {
      "epoch": 2.0266791481394804,
      "loss_main": 0.08748476952314377,
      "step": 17320
    },
    {
      "epoch": 2.0278492862157735,
      "grad_norm": 0.28886961936950684,
      "learning_rate": 8.520106240744759e-05,
      "loss": 0.0704,
      "step": 17330
    },
    {
      "epoch": 2.0278492862157735,
      "loss_main": 0.05875556915998459,
      "step": 17330
    },
    {
      "epoch": 2.0290194242920663,
      "grad_norm": 0.4140373766422272,
      "learning_rate": 8.518148607202102e-05,
      "loss": 0.0746,
      "step": 17340
    },
    {
      "epoch": 2.0290194242920663,
      "loss_main": 0.07726065069437027,
      "step": 17340
    },
    {
      "epoch": 2.0301895623683595,
      "grad_norm": 0.4482607841491699,
      "learning_rate": 8.51618990493879e-05,
      "loss": 0.0855,
      "step": 17350
    },
    {
      "epoch": 2.0301895623683595,
      "loss_main": 0.07745444029569626,
      "step": 17350
    },
    {
      "epoch": 2.0313597004446526,
      "grad_norm": 0.41290929913520813,
      "learning_rate": 8.514230134549823e-05,
      "loss": 0.0837,
      "step": 17360
    },
    {
      "epoch": 2.0313597004446526,
      "loss_main": 0.07883812487125397,
      "step": 17360
    },
    {
      "epoch": 2.0325298385209454,
      "grad_norm": 0.4010228216648102,
      "learning_rate": 8.51226929663053e-05,
      "loss": 0.0708,
      "step": 17370
    },
    {
      "epoch": 2.0325298385209454,
      "loss_main": 0.08870752900838852,
      "step": 17370
    },
    {
      "epoch": 2.0336999765972386,
      "grad_norm": 0.414371520280838,
      "learning_rate": 8.51030739177656e-05,
      "loss": 0.0752,
      "step": 17380
    },
    {
      "epoch": 2.0336999765972386,
      "loss_main": 0.05961182713508606,
      "step": 17380
    },
    {
      "epoch": 2.0348701146735313,
      "grad_norm": 0.46853572130203247,
      "learning_rate": 8.508344420583889e-05,
      "loss": 0.0756,
      "step": 17390
    },
    {
      "epoch": 2.0348701146735313,
      "loss_main": 0.05787023901939392,
      "step": 17390
    },
    {
      "epoch": 2.0360402527498245,
      "grad_norm": 0.21226570010185242,
      "learning_rate": 8.506380383648816e-05,
      "loss": 0.0684,
      "step": 17400
    },
    {
      "epoch": 2.0360402527498245,
      "loss_main": 0.059917617589235306,
      "step": 17400
    },
    {
      "epoch": 2.0372103908261177,
      "grad_norm": 0.24793218076229095,
      "learning_rate": 8.504415281567963e-05,
      "loss": 0.0658,
      "step": 17410
    },
    {
      "epoch": 2.0372103908261177,
      "loss_main": 0.02800310030579567,
      "step": 17410
    },
    {
      "epoch": 2.0383805289024104,
      "grad_norm": 0.379227876663208,
      "learning_rate": 8.502449114938275e-05,
      "loss": 0.0714,
      "step": 17420
    },
    {
      "epoch": 2.0383805289024104,
      "loss_main": 0.05328218266367912,
      "step": 17420
    },
    {
      "epoch": 2.0395506669787036,
      "grad_norm": 0.30168163776397705,
      "learning_rate": 8.500481884357025e-05,
      "loss": 0.0791,
      "step": 17430
    },
    {
      "epoch": 2.0395506669787036,
      "loss_main": 0.07051035016775131,
      "step": 17430
    },
    {
      "epoch": 2.0407208050549963,
      "grad_norm": 0.3076442778110504,
      "learning_rate": 8.498513590421801e-05,
      "loss": 0.0794,
      "step": 17440
    },
    {
      "epoch": 2.0407208050549963,
      "loss_main": 0.0996556282043457,
      "step": 17440
    },
    {
      "epoch": 2.0418909431312895,
      "grad_norm": 0.42824026942253113,
      "learning_rate": 8.496544233730522e-05,
      "loss": 0.0774,
      "step": 17450
    },
    {
      "epoch": 2.0418909431312895,
      "loss_main": 0.06360678374767303,
      "step": 17450
    },
    {
      "epoch": 2.0430610812075827,
      "grad_norm": 0.33863797783851624,
      "learning_rate": 8.494573814881426e-05,
      "loss": 0.0713,
      "step": 17460
    },
    {
      "epoch": 2.0430610812075827,
      "loss_main": 0.059016067534685135,
      "step": 17460
    },
    {
      "epoch": 2.0442312192838754,
      "grad_norm": 0.31485018134117126,
      "learning_rate": 8.492602334473074e-05,
      "loss": 0.0628,
      "step": 17470
    },
    {
      "epoch": 2.0442312192838754,
      "loss_main": 0.11269523203372955,
      "step": 17470
    },
    {
      "epoch": 2.0454013573601686,
      "grad_norm": 0.26922956109046936,
      "learning_rate": 8.49062979310435e-05,
      "loss": 0.0767,
      "step": 17480
    },
    {
      "epoch": 2.0454013573601686,
      "loss_main": 0.10456902533769608,
      "step": 17480
    },
    {
      "epoch": 2.0465714954364613,
      "grad_norm": 0.4173424243927002,
      "learning_rate": 8.488656191374458e-05,
      "loss": 0.0771,
      "step": 17490
    },
    {
      "epoch": 2.0465714954364613,
      "loss_main": 0.08290072530508041,
      "step": 17490
    },
    {
      "epoch": 2.0477416335127545,
      "grad_norm": 0.37760618329048157,
      "learning_rate": 8.48668152988293e-05,
      "loss": 0.0703,
      "step": 17500
    },
    {
      "epoch": 2.0477416335127545,
      "loss_main": 0.06317120045423508,
      "step": 17500
    },
    {
      "epoch": 2.0489117715890477,
      "grad_norm": 0.2797106206417084,
      "learning_rate": 8.484705809229612e-05,
      "loss": 0.0748,
      "step": 17510
    },
    {
      "epoch": 2.0489117715890477,
      "loss_main": 0.07371200621128082,
      "step": 17510
    },
    {
      "epoch": 2.0500819096653404,
      "grad_norm": 0.34859204292297363,
      "learning_rate": 8.482729030014677e-05,
      "loss": 0.0808,
      "step": 17520
    },
    {
      "epoch": 2.0500819096653404,
      "loss_main": 0.036985255777835846,
      "step": 17520
    },
    {
      "epoch": 2.0512520477416336,
      "grad_norm": 0.33218565583229065,
      "learning_rate": 8.48075119283862e-05,
      "loss": 0.0618,
      "step": 17530
    },
    {
      "epoch": 2.0512520477416336,
      "loss_main": 0.08715958148241043,
      "step": 17530
    },
    {
      "epoch": 2.0524221858179263,
      "grad_norm": 0.37605857849121094,
      "learning_rate": 8.478772298302254e-05,
      "loss": 0.0707,
      "step": 17540
    },
    {
      "epoch": 2.0524221858179263,
      "loss_main": 0.06596685945987701,
      "step": 17540
    },
    {
      "epoch": 2.0535923238942195,
      "grad_norm": 0.4269343614578247,
      "learning_rate": 8.476792347006716e-05,
      "loss": 0.0795,
      "step": 17550
    },
    {
      "epoch": 2.0535923238942195,
      "loss_main": 0.04121757671236992,
      "step": 17550
    },
    {
      "epoch": 2.0547624619705127,
      "grad_norm": 0.31250303983688354,
      "learning_rate": 8.474811339553462e-05,
      "loss": 0.0743,
      "step": 17560
    },
    {
      "epoch": 2.0547624619705127,
      "loss_main": 0.0854029506444931,
      "step": 17560
    },
    {
      "epoch": 2.0559326000468054,
      "grad_norm": 0.5129120349884033,
      "learning_rate": 8.47282927654427e-05,
      "loss": 0.0934,
      "step": 17570
    },
    {
      "epoch": 2.0559326000468054,
      "loss_main": 0.06443305313587189,
      "step": 17570
    },
    {
      "epoch": 2.0571027381230986,
      "grad_norm": 0.33714285492897034,
      "learning_rate": 8.470846158581238e-05,
      "loss": 0.0704,
      "step": 17580
    },
    {
      "epoch": 2.0571027381230986,
      "loss_main": 0.06974148005247116,
      "step": 17580
    },
    {
      "epoch": 2.0582728761993914,
      "grad_norm": 0.38793495297431946,
      "learning_rate": 8.468861986266787e-05,
      "loss": 0.0788,
      "step": 17590
    },
    {
      "epoch": 2.0582728761993914,
      "loss_main": 0.07781416922807693,
      "step": 17590
    },
    {
      "epoch": 2.0594430142756845,
      "grad_norm": 0.31390145421028137,
      "learning_rate": 8.466876760203654e-05,
      "loss": 0.069,
      "step": 17600
    },
    {
      "epoch": 2.0594430142756845,
      "loss_main": 0.1345416158437729,
      "step": 17600
    },
    {
      "epoch": 2.0606131523519777,
      "grad_norm": 0.42466968297958374,
      "learning_rate": 8.464890480994898e-05,
      "loss": 0.0799,
      "step": 17610
    },
    {
      "epoch": 2.0606131523519777,
      "loss_main": 0.04808247834444046,
      "step": 17610
    },
    {
      "epoch": 2.0617832904282705,
      "grad_norm": 0.42847275733947754,
      "learning_rate": 8.462903149243899e-05,
      "loss": 0.0739,
      "step": 17620
    },
    {
      "epoch": 2.0617832904282705,
      "loss_main": 0.07761193811893463,
      "step": 17620
    },
    {
      "epoch": 2.0629534285045636,
      "grad_norm": 0.234054297208786,
      "learning_rate": 8.460914765554357e-05,
      "loss": 0.0757,
      "step": 17630
    },
    {
      "epoch": 2.0629534285045636,
      "loss_main": 0.07497551292181015,
      "step": 17630
    },
    {
      "epoch": 2.0641235665808564,
      "grad_norm": 0.3308909237384796,
      "learning_rate": 8.458925330530288e-05,
      "loss": 0.0903,
      "step": 17640
    },
    {
      "epoch": 2.0641235665808564,
      "loss_main": 0.07460512965917587,
      "step": 17640
    },
    {
      "epoch": 2.0652937046571496,
      "grad_norm": 0.284758597612381,
      "learning_rate": 8.456934844776032e-05,
      "loss": 0.0734,
      "step": 17650
    },
    {
      "epoch": 2.0652937046571496,
      "loss_main": 0.0642753466963768,
      "step": 17650
    },
    {
      "epoch": 2.0664638427334427,
      "grad_norm": 0.2110336273908615,
      "learning_rate": 8.454943308896246e-05,
      "loss": 0.0724,
      "step": 17660
    },
    {
      "epoch": 2.0664638427334427,
      "loss_main": 0.10130124539136887,
      "step": 17660
    },
    {
      "epoch": 2.0676339808097355,
      "grad_norm": 0.2666301429271698,
      "learning_rate": 8.452950723495905e-05,
      "loss": 0.075,
      "step": 17670
    },
    {
      "epoch": 2.0676339808097355,
      "loss_main": 0.05761183053255081,
      "step": 17670
    },
    {
      "epoch": 2.0688041188860287,
      "grad_norm": 0.3568494915962219,
      "learning_rate": 8.450957089180303e-05,
      "loss": 0.0775,
      "step": 17680
    },
    {
      "epoch": 2.0688041188860287,
      "loss_main": 0.046990010887384415,
      "step": 17680
    },
    {
      "epoch": 2.0699742569623214,
      "grad_norm": 0.23347094655036926,
      "learning_rate": 8.448962406555055e-05,
      "loss": 0.0684,
      "step": 17690
    },
    {
      "epoch": 2.0699742569623214,
      "loss_main": 0.10926035791635513,
      "step": 17690
    },
    {
      "epoch": 2.0711443950386146,
      "grad_norm": 0.24379920959472656,
      "learning_rate": 8.446966676226093e-05,
      "loss": 0.079,
      "step": 17700
    },
    {
      "epoch": 2.0711443950386146,
      "loss_main": 0.03934146836400032,
      "step": 17700
    },
    {
      "epoch": 2.0723145331149078,
      "grad_norm": 0.27523860335350037,
      "learning_rate": 8.444969898799667e-05,
      "loss": 0.076,
      "step": 17710
    },
    {
      "epoch": 2.0723145331149078,
      "loss_main": 0.08531881868839264,
      "step": 17710
    },
    {
      "epoch": 2.0734846711912005,
      "grad_norm": 0.3530306816101074,
      "learning_rate": 8.442972074882343e-05,
      "loss": 0.0746,
      "step": 17720
    },
    {
      "epoch": 2.0734846711912005,
      "loss_main": 0.07534909248352051,
      "step": 17720
    },
    {
      "epoch": 2.0746548092674937,
      "grad_norm": 0.26332372426986694,
      "learning_rate": 8.44097320508101e-05,
      "loss": 0.0826,
      "step": 17730
    },
    {
      "epoch": 2.0746548092674937,
      "loss_main": 0.06662464886903763,
      "step": 17730
    },
    {
      "epoch": 2.0758249473437864,
      "grad_norm": 0.27206867933273315,
      "learning_rate": 8.43897329000287e-05,
      "loss": 0.0656,
      "step": 17740
    },
    {
      "epoch": 2.0758249473437864,
      "loss_main": 0.11100269854068756,
      "step": 17740
    },
    {
      "epoch": 2.0769950854200796,
      "grad_norm": 0.3232925832271576,
      "learning_rate": 8.436972330255448e-05,
      "loss": 0.0745,
      "step": 17750
    },
    {
      "epoch": 2.0769950854200796,
      "loss_main": 0.06104612722992897,
      "step": 17750
    },
    {
      "epoch": 2.078165223496373,
      "grad_norm": 0.32996174693107605,
      "learning_rate": 8.434970326446579e-05,
      "loss": 0.0835,
      "step": 17760
    },
    {
      "epoch": 2.078165223496373,
      "loss_main": 0.05993646755814552,
      "step": 17760
    },
    {
      "epoch": 2.0793353615726655,
      "grad_norm": 0.2809913754463196,
      "learning_rate": 8.432967279184418e-05,
      "loss": 0.071,
      "step": 17770
    },
    {
      "epoch": 2.0793353615726655,
      "loss_main": 0.08270611613988876,
      "step": 17770
    },
    {
      "epoch": 2.0805054996489587,
      "grad_norm": 0.3050394654273987,
      "learning_rate": 8.430963189077441e-05,
      "loss": 0.0661,
      "step": 17780
    },
    {
      "epoch": 2.0805054996489587,
      "loss_main": 0.03795776888728142,
      "step": 17780
    },
    {
      "epoch": 2.0816756377252514,
      "grad_norm": 0.41252070665359497,
      "learning_rate": 8.428958056734437e-05,
      "loss": 0.0641,
      "step": 17790
    },
    {
      "epoch": 2.0816756377252514,
      "loss_main": 0.07867373526096344,
      "step": 17790
    },
    {
      "epoch": 2.0828457758015446,
      "grad_norm": 0.41465649008750916,
      "learning_rate": 8.426951882764513e-05,
      "loss": 0.0826,
      "step": 17800
    },
    {
      "epoch": 2.0828457758015446,
      "loss_main": 0.10667823255062103,
      "step": 17800
    },
    {
      "epoch": 2.0840159138778374,
      "grad_norm": 0.39063340425491333,
      "learning_rate": 8.424944667777089e-05,
      "loss": 0.0721,
      "step": 17810
    },
    {
      "epoch": 2.0840159138778374,
      "loss_main": 0.05134940519928932,
      "step": 17810
    },
    {
      "epoch": 2.0851860519541305,
      "grad_norm": 0.2869962453842163,
      "learning_rate": 8.422936412381905e-05,
      "loss": 0.0697,
      "step": 17820
    },
    {
      "epoch": 2.0851860519541305,
      "loss_main": 0.09596198052167892,
      "step": 17820
    },
    {
      "epoch": 2.0863561900304237,
      "grad_norm": 0.29021281003952026,
      "learning_rate": 8.420927117189017e-05,
      "loss": 0.0729,
      "step": 17830
    },
    {
      "epoch": 2.0863561900304237,
      "loss_main": 0.10532908886671066,
      "step": 17830
    },
    {
      "epoch": 2.0875263281067165,
      "grad_norm": 0.3180733323097229,
      "learning_rate": 8.418916782808795e-05,
      "loss": 0.0735,
      "step": 17840
    },
    {
      "epoch": 2.0875263281067165,
      "loss_main": 0.059860289096832275,
      "step": 17840
    },
    {
      "epoch": 2.0886964661830096,
      "grad_norm": 0.4157228171825409,
      "learning_rate": 8.416905409851926e-05,
      "loss": 0.0695,
      "step": 17850
    },
    {
      "epoch": 2.0886964661830096,
      "loss_main": 0.04923495277762413,
      "step": 17850
    },
    {
      "epoch": 2.089866604259303,
      "grad_norm": 0.31424984335899353,
      "learning_rate": 8.41489299892941e-05,
      "loss": 0.0637,
      "step": 17860
    },
    {
      "epoch": 2.089866604259303,
      "loss_main": 0.08196251094341278,
      "step": 17860
    },
    {
      "epoch": 2.0910367423355956,
      "grad_norm": 0.2694312334060669,
      "learning_rate": 8.412879550652566e-05,
      "loss": 0.0674,
      "step": 17870
    },
    {
      "epoch": 2.0910367423355956,
      "loss_main": 0.08213334530591965,
      "step": 17870
    },
    {
      "epoch": 2.0922068804118887,
      "grad_norm": 0.3250337243080139,
      "learning_rate": 8.410865065633029e-05,
      "loss": 0.0745,
      "step": 17880
    },
    {
      "epoch": 2.0922068804118887,
      "loss_main": 0.04832305386662483,
      "step": 17880
    },
    {
      "epoch": 2.0933770184881815,
      "grad_norm": 0.33163806796073914,
      "learning_rate": 8.408849544482742e-05,
      "loss": 0.0773,
      "step": 17890
    },
    {
      "epoch": 2.0933770184881815,
      "loss_main": 0.08936323970556259,
      "step": 17890
    },
    {
      "epoch": 2.0945471565644747,
      "grad_norm": 0.30379992723464966,
      "learning_rate": 8.406832987813968e-05,
      "loss": 0.0701,
      "step": 17900
    },
    {
      "epoch": 2.0945471565644747,
      "loss_main": 0.09164601564407349,
      "step": 17900
    },
    {
      "epoch": 2.0957172946407674,
      "grad_norm": 0.3453783094882965,
      "learning_rate": 8.404815396239286e-05,
      "loss": 0.0742,
      "step": 17910
    },
    {
      "epoch": 2.0957172946407674,
      "loss_main": 0.07736248522996902,
      "step": 17910
    },
    {
      "epoch": 2.0968874327170606,
      "grad_norm": 0.34864458441734314,
      "learning_rate": 8.402796770371587e-05,
      "loss": 0.0808,
      "step": 17920
    },
    {
      "epoch": 2.0968874327170606,
      "loss_main": 0.08534479141235352,
      "step": 17920
    },
    {
      "epoch": 2.0980575707933538,
      "grad_norm": 0.2391037791967392,
      "learning_rate": 8.400777110824071e-05,
      "loss": 0.0664,
      "step": 17930
    },
    {
      "epoch": 2.0980575707933538,
      "loss_main": 0.06786523014307022,
      "step": 17930
    },
    {
      "epoch": 2.0992277088696465,
      "grad_norm": 0.27226653695106506,
      "learning_rate": 8.398756418210263e-05,
      "loss": 0.0645,
      "step": 17940
    },
    {
      "epoch": 2.0992277088696465,
      "loss_main": 0.0903364047408104,
      "step": 17940
    },
    {
      "epoch": 2.1003978469459397,
      "grad_norm": 0.31487706303596497,
      "learning_rate": 8.396734693143993e-05,
      "loss": 0.061,
      "step": 17950
    },
    {
      "epoch": 2.1003978469459397,
      "loss_main": 0.05443672463297844,
      "step": 17950
    },
    {
      "epoch": 2.1015679850222324,
      "grad_norm": 0.24150989949703217,
      "learning_rate": 8.39471193623941e-05,
      "loss": 0.0833,
      "step": 17960
    },
    {
      "epoch": 2.1015679850222324,
      "loss_main": 0.07645199447870255,
      "step": 17960
    },
    {
      "epoch": 2.1027381230985256,
      "grad_norm": 0.27042996883392334,
      "learning_rate": 8.392688148110974e-05,
      "loss": 0.0775,
      "step": 17970
    },
    {
      "epoch": 2.1027381230985256,
      "loss_main": 0.08424188196659088,
      "step": 17970
    },
    {
      "epoch": 2.1039082611748188,
      "grad_norm": 0.37920352816581726,
      "learning_rate": 8.390663329373456e-05,
      "loss": 0.0719,
      "step": 17980
    },
    {
      "epoch": 2.1039082611748188,
      "loss_main": 0.07338656485080719,
      "step": 17980
    },
    {
      "epoch": 2.1050783992511115,
      "grad_norm": 0.3272615671157837,
      "learning_rate": 8.388637480641944e-05,
      "loss": 0.08,
      "step": 17990
    },
    {
      "epoch": 2.1050783992511115,
      "loss_main": 0.07338842004537582,
      "step": 17990
    },
    {
      "epoch": 2.1062485373274047,
      "grad_norm": 0.2764270007610321,
      "learning_rate": 8.386610602531837e-05,
      "loss": 0.0745,
      "step": 18000
    },
    {
      "epoch": 2.1062485373274047,
      "loss_main": 0.04521599039435387,
      "step": 18000
    },
    {
      "epoch": 2.1074186754036974,
      "grad_norm": 0.3649463355541229,
      "learning_rate": 8.384582695658847e-05,
      "loss": 0.0657,
      "step": 18010
    },
    {
      "epoch": 2.1074186754036974,
      "loss_main": 0.11055077612400055,
      "step": 18010
    },
    {
      "epoch": 2.1085888134799906,
      "grad_norm": 0.34038692712783813,
      "learning_rate": 8.382553760638999e-05,
      "loss": 0.0729,
      "step": 18020
    },
    {
      "epoch": 2.1085888134799906,
      "loss_main": 0.0765339657664299,
      "step": 18020
    },
    {
      "epoch": 2.109758951556284,
      "grad_norm": 0.3224629759788513,
      "learning_rate": 8.380523798088631e-05,
      "loss": 0.0685,
      "step": 18030
    },
    {
      "epoch": 2.109758951556284,
      "loss_main": 0.06659315526485443,
      "step": 18030
    },
    {
      "epoch": 2.1109290896325765,
      "grad_norm": 0.32898247241973877,
      "learning_rate": 8.378492808624389e-05,
      "loss": 0.0707,
      "step": 18040
    },
    {
      "epoch": 2.1109290896325765,
      "loss_main": 0.07019486278295517,
      "step": 18040
    },
    {
      "epoch": 2.1120992277088697,
      "grad_norm": 0.2921055853366852,
      "learning_rate": 8.376460792863237e-05,
      "loss": 0.0713,
      "step": 18050
    },
    {
      "epoch": 2.1120992277088697,
      "loss_main": 0.06219128146767616,
      "step": 18050
    },
    {
      "epoch": 2.1132693657851624,
      "grad_norm": 0.28238406777381897,
      "learning_rate": 8.374427751422444e-05,
      "loss": 0.0582,
      "step": 18060
    },
    {
      "epoch": 2.1132693657851624,
      "loss_main": 0.06422721594572067,
      "step": 18060
    },
    {
      "epoch": 2.1144395038614556,
      "grad_norm": 0.4404115378856659,
      "learning_rate": 8.3723936849196e-05,
      "loss": 0.0696,
      "step": 18070
    },
    {
      "epoch": 2.1144395038614556,
      "loss_main": 0.05821692571043968,
      "step": 18070
    },
    {
      "epoch": 2.115609641937749,
      "grad_norm": 0.2947072684764862,
      "learning_rate": 8.370358593972595e-05,
      "loss": 0.073,
      "step": 18080
    },
    {
      "epoch": 2.115609641937749,
      "loss_main": 0.04648882895708084,
      "step": 18080
    },
    {
      "epoch": 2.1167797800140415,
      "grad_norm": 0.34441739320755005,
      "learning_rate": 8.36832247919964e-05,
      "loss": 0.0725,
      "step": 18090
    },
    {
      "epoch": 2.1167797800140415,
      "loss_main": 0.11336716264486313,
      "step": 18090
    },
    {
      "epoch": 2.1179499180903347,
      "grad_norm": 0.30609631538391113,
      "learning_rate": 8.36628534121925e-05,
      "loss": 0.0777,
      "step": 18100
    },
    {
      "epoch": 2.1179499180903347,
      "loss_main": 0.05762063339352608,
      "step": 18100
    },
    {
      "epoch": 2.1191200561666275,
      "grad_norm": 0.31476351618766785,
      "learning_rate": 8.364247180650254e-05,
      "loss": 0.0788,
      "step": 18110
    },
    {
      "epoch": 2.1191200561666275,
      "loss_main": 0.07013686746358871,
      "step": 18110
    },
    {
      "epoch": 2.1202901942429206,
      "grad_norm": 0.2991913855075836,
      "learning_rate": 8.362207998111794e-05,
      "loss": 0.0645,
      "step": 18120
    },
    {
      "epoch": 2.1202901942429206,
      "loss_main": 0.07920300960540771,
      "step": 18120
    },
    {
      "epoch": 2.121460332319214,
      "grad_norm": 0.36603614687919617,
      "learning_rate": 8.360167794223318e-05,
      "loss": 0.0729,
      "step": 18130
    },
    {
      "epoch": 2.121460332319214,
      "loss_main": 0.10144935548305511,
      "step": 18130
    },
    {
      "epoch": 2.1226304703955066,
      "grad_norm": 0.4463609457015991,
      "learning_rate": 8.358126569604586e-05,
      "loss": 0.0809,
      "step": 18140
    },
    {
      "epoch": 2.1226304703955066,
      "loss_main": 0.10338900238275528,
      "step": 18140
    },
    {
      "epoch": 2.1238006084717997,
      "grad_norm": 0.18167437613010406,
      "learning_rate": 8.356084324875668e-05,
      "loss": 0.0718,
      "step": 18150
    },
    {
      "epoch": 2.1238006084717997,
      "loss_main": 0.0645461305975914,
      "step": 18150
    },
    {
      "epoch": 2.1249707465480925,
      "grad_norm": 0.2997329533100128,
      "learning_rate": 8.354041060656945e-05,
      "loss": 0.0692,
      "step": 18160
    },
    {
      "epoch": 2.1249707465480925,
      "loss_main": 0.08732231706380844,
      "step": 18160
    },
    {
      "epoch": 2.1261408846243857,
      "grad_norm": 0.22088445723056793,
      "learning_rate": 8.351996777569106e-05,
      "loss": 0.0632,
      "step": 18170
    },
    {
      "epoch": 2.1261408846243857,
      "loss_main": 0.06527601927518845,
      "step": 18170
    },
    {
      "epoch": 2.127311022700679,
      "grad_norm": 0.32198572158813477,
      "learning_rate": 8.349951476233148e-05,
      "loss": 0.0771,
      "step": 18180
    },
    {
      "epoch": 2.127311022700679,
      "loss_main": 0.046829599887132645,
      "step": 18180
    },
    {
      "epoch": 2.1284811607769716,
      "grad_norm": 0.2326272875070572,
      "learning_rate": 8.347905157270386e-05,
      "loss": 0.0738,
      "step": 18190
    },
    {
      "epoch": 2.1284811607769716,
      "loss_main": 0.05283726006746292,
      "step": 18190
    },
    {
      "epoch": 2.1296512988532648,
      "grad_norm": 0.2881942689418793,
      "learning_rate": 8.345857821302432e-05,
      "loss": 0.0743,
      "step": 18200
    },
    {
      "epoch": 2.1296512988532648,
      "loss_main": 0.09773553907871246,
      "step": 18200
    },
    {
      "epoch": 2.1308214369295575,
      "grad_norm": 0.4268690049648285,
      "learning_rate": 8.343809468951213e-05,
      "loss": 0.0706,
      "step": 18210
    },
    {
      "epoch": 2.1308214369295575,
      "loss_main": 0.08758936822414398,
      "step": 18210
    },
    {
      "epoch": 2.1319915750058507,
      "grad_norm": 0.2891314625740051,
      "learning_rate": 8.341760100838965e-05,
      "loss": 0.0887,
      "step": 18220
    },
    {
      "epoch": 2.1319915750058507,
      "loss_main": 0.06758283823728561,
      "step": 18220
    },
    {
      "epoch": 2.133161713082144,
      "grad_norm": 0.20247933268547058,
      "learning_rate": 8.339709717588233e-05,
      "loss": 0.0784,
      "step": 18230
    },
    {
      "epoch": 2.133161713082144,
      "loss_main": 0.05070706084370613,
      "step": 18230
    },
    {
      "epoch": 2.1343318511584366,
      "grad_norm": 0.37176835536956787,
      "learning_rate": 8.33765831982187e-05,
      "loss": 0.0682,
      "step": 18240
    },
    {
      "epoch": 2.1343318511584366,
      "loss_main": 0.06934108585119247,
      "step": 18240
    },
    {
      "epoch": 2.13550198923473,
      "grad_norm": 0.28562793135643005,
      "learning_rate": 8.335605908163035e-05,
      "loss": 0.0669,
      "step": 18250
    },
    {
      "epoch": 2.13550198923473,
      "loss_main": 0.03740105405449867,
      "step": 18250
    },
    {
      "epoch": 2.1366721273110225,
      "grad_norm": 0.3559013605117798,
      "learning_rate": 8.333552483235196e-05,
      "loss": 0.0652,
      "step": 18260
    },
    {
      "epoch": 2.1366721273110225,
      "loss_main": 0.08042474836111069,
      "step": 18260
    },
    {
      "epoch": 2.1378422653873157,
      "grad_norm": 0.36058753728866577,
      "learning_rate": 8.33149804566213e-05,
      "loss": 0.0793,
      "step": 18270
    },
    {
      "epoch": 2.1378422653873157,
      "loss_main": 0.060005247592926025,
      "step": 18270
    },
    {
      "epoch": 2.139012403463609,
      "grad_norm": 0.38637736439704895,
      "learning_rate": 8.329442596067921e-05,
      "loss": 0.0811,
      "step": 18280
    },
    {
      "epoch": 2.139012403463609,
      "loss_main": 0.08133057504892349,
      "step": 18280
    },
    {
      "epoch": 2.1401825415399016,
      "grad_norm": 0.29928308725357056,
      "learning_rate": 8.32738613507696e-05,
      "loss": 0.0779,
      "step": 18290
    },
    {
      "epoch": 2.1401825415399016,
      "loss_main": 0.05531802028417587,
      "step": 18290
    },
    {
      "epoch": 2.141352679616195,
      "grad_norm": 0.3848152756690979,
      "learning_rate": 8.325328663313946e-05,
      "loss": 0.086,
      "step": 18300
    },
    {
      "epoch": 2.141352679616195,
      "loss_main": 0.09111867100000381,
      "step": 18300
    },
    {
      "epoch": 2.1425228176924875,
      "grad_norm": 0.2838610112667084,
      "learning_rate": 8.323270181403884e-05,
      "loss": 0.0718,
      "step": 18310
    },
    {
      "epoch": 2.1425228176924875,
      "loss_main": 0.08023345470428467,
      "step": 18310
    },
    {
      "epoch": 2.1436929557687807,
      "grad_norm": 0.3955132067203522,
      "learning_rate": 8.321210689972086e-05,
      "loss": 0.0711,
      "step": 18320
    },
    {
      "epoch": 2.1436929557687807,
      "loss_main": 0.05305337533354759,
      "step": 18320
    },
    {
      "epoch": 2.144863093845074,
      "grad_norm": 0.28254812955856323,
      "learning_rate": 8.319150189644174e-05,
      "loss": 0.0642,
      "step": 18330
    },
    {
      "epoch": 2.144863093845074,
      "loss_main": 0.04776826873421669,
      "step": 18330
    },
    {
      "epoch": 2.1460332319213666,
      "grad_norm": 0.3487183749675751,
      "learning_rate": 8.31708868104607e-05,
      "loss": 0.0827,
      "step": 18340
    },
    {
      "epoch": 2.1460332319213666,
      "loss_main": 0.08290211111307144,
      "step": 18340
    },
    {
      "epoch": 2.14720336999766,
      "grad_norm": 0.27432724833488464,
      "learning_rate": 8.315026164804007e-05,
      "loss": 0.072,
      "step": 18350
    },
    {
      "epoch": 2.14720336999766,
      "loss_main": 0.05100777745246887,
      "step": 18350
    },
    {
      "epoch": 2.1483735080739526,
      "grad_norm": 0.33799514174461365,
      "learning_rate": 8.312962641544524e-05,
      "loss": 0.0732,
      "step": 18360
    },
    {
      "epoch": 2.1483735080739526,
      "loss_main": 0.0477452278137207,
      "step": 18360
    },
    {
      "epoch": 2.1495436461502457,
      "grad_norm": 0.3055087625980377,
      "learning_rate": 8.310898111894465e-05,
      "loss": 0.0689,
      "step": 18370
    },
    {
      "epoch": 2.1495436461502457,
      "loss_main": 0.07889045029878616,
      "step": 18370
    },
    {
      "epoch": 2.150713784226539,
      "grad_norm": 0.30083441734313965,
      "learning_rate": 8.308832576480977e-05,
      "loss": 0.0694,
      "step": 18380
    },
    {
      "epoch": 2.150713784226539,
      "loss_main": 0.0493505485355854,
      "step": 18380
    },
    {
      "epoch": 2.1518839223028317,
      "grad_norm": 0.4094671308994293,
      "learning_rate": 8.306766035931519e-05,
      "loss": 0.0826,
      "step": 18390
    },
    {
      "epoch": 2.1518839223028317,
      "loss_main": 0.06592375785112381,
      "step": 18390
    },
    {
      "epoch": 2.153054060379125,
      "grad_norm": 0.4147386848926544,
      "learning_rate": 8.304698490873847e-05,
      "loss": 0.0715,
      "step": 18400
    },
    {
      "epoch": 2.153054060379125,
      "loss_main": 0.03918326646089554,
      "step": 18400
    },
    {
      "epoch": 2.1542241984554176,
      "grad_norm": 0.3140683174133301,
      "learning_rate": 8.30262994193603e-05,
      "loss": 0.0752,
      "step": 18410
    },
    {
      "epoch": 2.1542241984554176,
      "loss_main": 0.07410703599452972,
      "step": 18410
    },
    {
      "epoch": 2.1553943365317108,
      "grad_norm": 0.29923805594444275,
      "learning_rate": 8.300560389746438e-05,
      "loss": 0.0671,
      "step": 18420
    },
    {
      "epoch": 2.1553943365317108,
      "loss_main": 0.08262254297733307,
      "step": 18420
    },
    {
      "epoch": 2.156564474608004,
      "grad_norm": 0.3485985994338989,
      "learning_rate": 8.298489834933745e-05,
      "loss": 0.0762,
      "step": 18430
    },
    {
      "epoch": 2.156564474608004,
      "loss_main": 0.037497762590646744,
      "step": 18430
    },
    {
      "epoch": 2.1577346126842967,
      "grad_norm": 0.2621261775493622,
      "learning_rate": 8.296418278126934e-05,
      "loss": 0.0801,
      "step": 18440
    },
    {
      "epoch": 2.1577346126842967,
      "loss_main": 0.08614303171634674,
      "step": 18440
    },
    {
      "epoch": 2.15890475076059,
      "grad_norm": 0.28743284940719604,
      "learning_rate": 8.294345719955284e-05,
      "loss": 0.069,
      "step": 18450
    },
    {
      "epoch": 2.15890475076059,
      "loss_main": 0.0650038942694664,
      "step": 18450
    },
    {
      "epoch": 2.1600748888368826,
      "grad_norm": 0.28288397192955017,
      "learning_rate": 8.29227216104839e-05,
      "loss": 0.072,
      "step": 18460
    },
    {
      "epoch": 2.1600748888368826,
      "loss_main": 0.059267204254865646,
      "step": 18460
    },
    {
      "epoch": 2.1612450269131758,
      "grad_norm": 0.29641464352607727,
      "learning_rate": 8.290197602036137e-05,
      "loss": 0.0716,
      "step": 18470
    },
    {
      "epoch": 2.1612450269131758,
      "loss_main": 0.06548747420310974,
      "step": 18470
    },
    {
      "epoch": 2.162415164989469,
      "grad_norm": 0.29937681555747986,
      "learning_rate": 8.288122043548725e-05,
      "loss": 0.0709,
      "step": 18480
    },
    {
      "epoch": 2.162415164989469,
      "loss_main": 0.07979732006788254,
      "step": 18480
    },
    {
      "epoch": 2.1635853030657617,
      "grad_norm": 0.22297951579093933,
      "learning_rate": 8.286045486216657e-05,
      "loss": 0.0824,
      "step": 18490
    },
    {
      "epoch": 2.1635853030657617,
      "loss_main": 0.08758681267499924,
      "step": 18490
    },
    {
      "epoch": 2.164755441142055,
      "grad_norm": 0.42084741592407227,
      "learning_rate": 8.283967930670733e-05,
      "loss": 0.0749,
      "step": 18500
    },
    {
      "epoch": 2.164755441142055,
      "loss_main": 0.06717473268508911,
      "step": 18500
    },
    {
      "epoch": 2.1659255792183476,
      "grad_norm": 0.29329410195350647,
      "learning_rate": 8.281889377542058e-05,
      "loss": 0.0752,
      "step": 18510
    },
    {
      "epoch": 2.1659255792183476,
      "loss_main": 0.07306519895792007,
      "step": 18510
    },
    {
      "epoch": 2.167095717294641,
      "grad_norm": 0.2540392279624939,
      "learning_rate": 8.279809827462045e-05,
      "loss": 0.0715,
      "step": 18520
    },
    {
      "epoch": 2.167095717294641,
      "loss_main": 0.09308184683322906,
      "step": 18520
    },
    {
      "epoch": 2.168265855370934,
      "grad_norm": 0.2654740810394287,
      "learning_rate": 8.277729281062402e-05,
      "loss": 0.0788,
      "step": 18530
    },
    {
      "epoch": 2.168265855370934,
      "loss_main": 0.07896976917982101,
      "step": 18530
    },
    {
      "epoch": 2.1694359934472267,
      "grad_norm": 0.19606651365756989,
      "learning_rate": 8.27564773897515e-05,
      "loss": 0.0728,
      "step": 18540
    },
    {
      "epoch": 2.1694359934472267,
      "loss_main": 0.06467544287443161,
      "step": 18540
    },
    {
      "epoch": 2.17060613152352,
      "grad_norm": 0.2940122187137604,
      "learning_rate": 8.273565201832602e-05,
      "loss": 0.0686,
      "step": 18550
    },
    {
      "epoch": 2.17060613152352,
      "loss_main": 0.06363184750080109,
      "step": 18550
    },
    {
      "epoch": 2.1717762695998126,
      "grad_norm": 0.32760506868362427,
      "learning_rate": 8.27148167026738e-05,
      "loss": 0.0677,
      "step": 18560
    },
    {
      "epoch": 2.1717762695998126,
      "loss_main": 0.0781790092587471,
      "step": 18560
    },
    {
      "epoch": 2.172946407676106,
      "grad_norm": 0.3436425030231476,
      "learning_rate": 8.269397144912405e-05,
      "loss": 0.0763,
      "step": 18570
    },
    {
      "epoch": 2.172946407676106,
      "loss_main": 0.06798980385065079,
      "step": 18570
    },
    {
      "epoch": 2.174116545752399,
      "grad_norm": 0.28338560461997986,
      "learning_rate": 8.267311626400899e-05,
      "loss": 0.0644,
      "step": 18580
    },
    {
      "epoch": 2.174116545752399,
      "loss_main": 0.07974246889352798,
      "step": 18580
    },
    {
      "epoch": 2.1752866838286917,
      "grad_norm": 0.33483031392097473,
      "learning_rate": 8.26522511536639e-05,
      "loss": 0.0726,
      "step": 18590
    },
    {
      "epoch": 2.1752866838286917,
      "loss_main": 0.07117120921611786,
      "step": 18590
    },
    {
      "epoch": 2.176456821904985,
      "grad_norm": 0.4633052349090576,
      "learning_rate": 8.263137612442706e-05,
      "loss": 0.0702,
      "step": 18600
    },
    {
      "epoch": 2.176456821904985,
      "loss_main": 0.047351811081171036,
      "step": 18600
    },
    {
      "epoch": 2.1776269599812776,
      "grad_norm": 0.3418201208114624,
      "learning_rate": 8.261049118263971e-05,
      "loss": 0.0678,
      "step": 18610
    },
    {
      "epoch": 2.1776269599812776,
      "loss_main": 0.05330536141991615,
      "step": 18610
    },
    {
      "epoch": 2.178797098057571,
      "grad_norm": 0.2496597319841385,
      "learning_rate": 8.258959633464619e-05,
      "loss": 0.077,
      "step": 18620
    },
    {
      "epoch": 2.178797098057571,
      "loss_main": 0.0610896535217762,
      "step": 18620
    },
    {
      "epoch": 2.179967236133864,
      "grad_norm": 0.33835694193840027,
      "learning_rate": 8.256869158679377e-05,
      "loss": 0.0743,
      "step": 18630
    },
    {
      "epoch": 2.179967236133864,
      "loss_main": 0.08714135736227036,
      "step": 18630
    },
    {
      "epoch": 2.1811373742101567,
      "grad_norm": 0.24563032388687134,
      "learning_rate": 8.254777694543278e-05,
      "loss": 0.0745,
      "step": 18640
    },
    {
      "epoch": 2.1811373742101567,
      "loss_main": 0.08208947628736496,
      "step": 18640
    },
    {
      "epoch": 2.18230751228645,
      "grad_norm": 0.27250248193740845,
      "learning_rate": 8.252685241691651e-05,
      "loss": 0.0678,
      "step": 18650
    },
    {
      "epoch": 2.18230751228645,
      "loss_main": 0.10259748995304108,
      "step": 18650
    },
    {
      "epoch": 2.1834776503627427,
      "grad_norm": 0.2833857238292694,
      "learning_rate": 8.250591800760133e-05,
      "loss": 0.082,
      "step": 18660
    },
    {
      "epoch": 2.1834776503627427,
      "loss_main": 0.10006652772426605,
      "step": 18660
    },
    {
      "epoch": 2.184647788439036,
      "grad_norm": 0.34461531043052673,
      "learning_rate": 8.248497372384649e-05,
      "loss": 0.0805,
      "step": 18670
    },
    {
      "epoch": 2.184647788439036,
      "loss_main": 0.0917796716094017,
      "step": 18670
    },
    {
      "epoch": 2.1858179265153286,
      "grad_norm": 0.25778263807296753,
      "learning_rate": 8.246401957201437e-05,
      "loss": 0.0758,
      "step": 18680
    },
    {
      "epoch": 2.1858179265153286,
      "loss_main": 0.0647805854678154,
      "step": 18680
    },
    {
      "epoch": 2.1869880645916218,
      "grad_norm": 0.28191012144088745,
      "learning_rate": 8.244305555847027e-05,
      "loss": 0.0724,
      "step": 18690
    },
    {
      "epoch": 2.1869880645916218,
      "loss_main": 0.07838516682386398,
      "step": 18690
    },
    {
      "epoch": 2.188158202667915,
      "grad_norm": 0.4877074956893921,
      "learning_rate": 8.24220816895825e-05,
      "loss": 0.0686,
      "step": 18700
    },
    {
      "epoch": 2.188158202667915,
      "loss_main": 0.04934301599860191,
      "step": 18700
    },
    {
      "epoch": 2.1893283407442077,
      "grad_norm": 0.2859354615211487,
      "learning_rate": 8.240109797172237e-05,
      "loss": 0.0765,
      "step": 18710
    },
    {
      "epoch": 2.1893283407442077,
      "loss_main": 0.06442179530858994,
      "step": 18710
    },
    {
      "epoch": 2.190498478820501,
      "grad_norm": 0.3519039750099182,
      "learning_rate": 8.238010441126416e-05,
      "loss": 0.0714,
      "step": 18720
    },
    {
      "epoch": 2.190498478820501,
      "loss_main": 0.07855135947465897,
      "step": 18720
    },
    {
      "epoch": 2.191668616896794,
      "grad_norm": 0.3387995958328247,
      "learning_rate": 8.23591010145852e-05,
      "loss": 0.0753,
      "step": 18730
    },
    {
      "epoch": 2.191668616896794,
      "loss_main": 0.12461943924427032,
      "step": 18730
    },
    {
      "epoch": 2.192838754973087,
      "grad_norm": 0.3328387141227722,
      "learning_rate": 8.233808778806571e-05,
      "loss": 0.0787,
      "step": 18740
    },
    {
      "epoch": 2.192838754973087,
      "loss_main": 0.09109973907470703,
      "step": 18740
    },
    {
      "epoch": 2.19400889304938,
      "grad_norm": 0.3895794749259949,
      "learning_rate": 8.231706473808903e-05,
      "loss": 0.0697,
      "step": 18750
    },
    {
      "epoch": 2.19400889304938,
      "loss_main": 0.08224458247423172,
      "step": 18750
    },
    {
      "epoch": 2.1951790311256727,
      "grad_norm": 0.2580466866493225,
      "learning_rate": 8.229603187104133e-05,
      "loss": 0.0746,
      "step": 18760
    },
    {
      "epoch": 2.1951790311256727,
      "loss_main": 0.036666691303253174,
      "step": 18760
    },
    {
      "epoch": 2.196349169201966,
      "grad_norm": 0.40623605251312256,
      "learning_rate": 8.22749891933119e-05,
      "loss": 0.0756,
      "step": 18770
    },
    {
      "epoch": 2.196349169201966,
      "loss_main": 0.09983205795288086,
      "step": 18770
    },
    {
      "epoch": 2.1975193072782586,
      "grad_norm": 0.35222986340522766,
      "learning_rate": 8.225393671129291e-05,
      "loss": 0.0761,
      "step": 18780
    },
    {
      "epoch": 2.1975193072782586,
      "loss_main": 0.05467935651540756,
      "step": 18780
    },
    {
      "epoch": 2.198689445354552,
      "grad_norm": 0.21389402449131012,
      "learning_rate": 8.223287443137957e-05,
      "loss": 0.0743,
      "step": 18790
    },
    {
      "epoch": 2.198689445354552,
      "loss_main": 0.055680032819509506,
      "step": 18790
    },
    {
      "epoch": 2.199859583430845,
      "grad_norm": 0.26866209506988525,
      "learning_rate": 8.221180235997004e-05,
      "loss": 0.0764,
      "step": 18800
    },
    {
      "epoch": 2.199859583430845,
      "loss_main": 0.06414831429719925,
      "step": 18800
    },
    {
      "epoch": 2.2010297215071377,
      "grad_norm": 0.2919172942638397,
      "learning_rate": 8.219072050346544e-05,
      "loss": 0.0741,
      "step": 18810
    },
    {
      "epoch": 2.2010297215071377,
      "loss_main": 0.06833474338054657,
      "step": 18810
    },
    {
      "epoch": 2.202199859583431,
      "grad_norm": 0.2817119359970093,
      "learning_rate": 8.216962886826992e-05,
      "loss": 0.0741,
      "step": 18820
    },
    {
      "epoch": 2.202199859583431,
      "loss_main": 0.053084202110767365,
      "step": 18820
    },
    {
      "epoch": 2.203369997659724,
      "grad_norm": 0.3942645788192749,
      "learning_rate": 8.214852746079054e-05,
      "loss": 0.0777,
      "step": 18830
    },
    {
      "epoch": 2.203369997659724,
      "loss_main": 0.06583964079618454,
      "step": 18830
    },
    {
      "epoch": 2.204540135736017,
      "grad_norm": 0.35620585083961487,
      "learning_rate": 8.212741628743732e-05,
      "loss": 0.0678,
      "step": 18840
    },
    {
      "epoch": 2.204540135736017,
      "loss_main": 0.06887438148260117,
      "step": 18840
    },
    {
      "epoch": 2.20571027381231,
      "grad_norm": 0.39480629563331604,
      "learning_rate": 8.210629535462333e-05,
      "loss": 0.0681,
      "step": 18850
    },
    {
      "epoch": 2.20571027381231,
      "loss_main": 0.061646681278944016,
      "step": 18850
    },
    {
      "epoch": 2.2068804118886027,
      "grad_norm": 0.2696937024593353,
      "learning_rate": 8.208516466876453e-05,
      "loss": 0.0706,
      "step": 18860
    },
    {
      "epoch": 2.2068804118886027,
      "loss_main": 0.04951939359307289,
      "step": 18860
    },
    {
      "epoch": 2.208050549964896,
      "grad_norm": 0.5085792541503906,
      "learning_rate": 8.206402423627986e-05,
      "loss": 0.0695,
      "step": 18870
    },
    {
      "epoch": 2.208050549964896,
      "loss_main": 0.07803479582071304,
      "step": 18870
    },
    {
      "epoch": 2.2092206880411887,
      "grad_norm": 0.2688269019126892,
      "learning_rate": 8.204287406359124e-05,
      "loss": 0.0794,
      "step": 18880
    },
    {
      "epoch": 2.2092206880411887,
      "loss_main": 0.07179709523916245,
      "step": 18880
    },
    {
      "epoch": 2.210390826117482,
      "grad_norm": 0.2686591148376465,
      "learning_rate": 8.20217141571235e-05,
      "loss": 0.0737,
      "step": 18890
    },
    {
      "epoch": 2.210390826117482,
      "loss_main": 0.06615322083234787,
      "step": 18890
    },
    {
      "epoch": 2.211560964193775,
      "grad_norm": 0.4165855646133423,
      "learning_rate": 8.200054452330449e-05,
      "loss": 0.0718,
      "step": 18900
    },
    {
      "epoch": 2.211560964193775,
      "loss_main": 0.05468737334012985,
      "step": 18900
    },
    {
      "epoch": 2.2127311022700678,
      "grad_norm": 0.21569295227527618,
      "learning_rate": 8.197936516856499e-05,
      "loss": 0.0756,
      "step": 18910
    },
    {
      "epoch": 2.2127311022700678,
      "loss_main": 0.058453887701034546,
      "step": 18910
    },
    {
      "epoch": 2.213901240346361,
      "grad_norm": 0.24999882280826569,
      "learning_rate": 8.195817609933871e-05,
      "loss": 0.0722,
      "step": 18920
    },
    {
      "epoch": 2.213901240346361,
      "loss_main": 0.03876575827598572,
      "step": 18920
    },
    {
      "epoch": 2.2150713784226537,
      "grad_norm": 0.391131192445755,
      "learning_rate": 8.193697732206233e-05,
      "loss": 0.071,
      "step": 18930
    },
    {
      "epoch": 2.2150713784226537,
      "loss_main": 0.052139125764369965,
      "step": 18930
    },
    {
      "epoch": 2.216241516498947,
      "grad_norm": 0.25323548913002014,
      "learning_rate": 8.19157688431755e-05,
      "loss": 0.0748,
      "step": 18940
    },
    {
      "epoch": 2.216241516498947,
      "loss_main": 0.05083940550684929,
      "step": 18940
    },
    {
      "epoch": 2.21741165457524,
      "grad_norm": 0.396403968334198,
      "learning_rate": 8.189455066912077e-05,
      "loss": 0.0652,
      "step": 18950
    },
    {
      "epoch": 2.21741165457524,
      "loss_main": 0.060584716498851776,
      "step": 18950
    },
    {
      "epoch": 2.2185817926515328,
      "grad_norm": 0.2617599070072174,
      "learning_rate": 8.187332280634369e-05,
      "loss": 0.0655,
      "step": 18960
    },
    {
      "epoch": 2.2185817926515328,
      "loss_main": 0.11023770272731781,
      "step": 18960
    },
    {
      "epoch": 2.219751930727826,
      "grad_norm": 0.36777132749557495,
      "learning_rate": 8.18520852612927e-05,
      "loss": 0.0757,
      "step": 18970
    },
    {
      "epoch": 2.219751930727826,
      "loss_main": 0.06462448090314865,
      "step": 18970
    },
    {
      "epoch": 2.2209220688041187,
      "grad_norm": 0.361032098531723,
      "learning_rate": 8.183083804041921e-05,
      "loss": 0.0818,
      "step": 18980
    },
    {
      "epoch": 2.2209220688041187,
      "loss_main": 0.08843894302845001,
      "step": 18980
    },
    {
      "epoch": 2.222092206880412,
      "grad_norm": 0.4061737060546875,
      "learning_rate": 8.180958115017757e-05,
      "loss": 0.073,
      "step": 18990
    },
    {
      "epoch": 2.222092206880412,
      "loss_main": 0.04476355388760567,
      "step": 18990
    },
    {
      "epoch": 2.223262344956705,
      "grad_norm": 0.26400527358055115,
      "learning_rate": 8.178831459702505e-05,
      "loss": 0.0704,
      "step": 19000
    },
    {
      "epoch": 2.223262344956705,
      "loss_main": 0.07916911691427231,
      "step": 19000
    },
    {
      "epoch": 2.224432483032998,
      "grad_norm": 0.36751827597618103,
      "learning_rate": 8.17670383874219e-05,
      "loss": 0.0721,
      "step": 19010
    },
    {
      "epoch": 2.224432483032998,
      "loss_main": 0.08966844528913498,
      "step": 19010
    },
    {
      "epoch": 2.225602621109291,
      "grad_norm": 0.23756243288516998,
      "learning_rate": 8.174575252783124e-05,
      "loss": 0.0727,
      "step": 19020
    },
    {
      "epoch": 2.225602621109291,
      "loss_main": 0.1345159262418747,
      "step": 19020
    },
    {
      "epoch": 2.2267727591855837,
      "grad_norm": 0.4080379009246826,
      "learning_rate": 8.172445702471914e-05,
      "loss": 0.0767,
      "step": 19030
    },
    {
      "epoch": 2.2267727591855837,
      "loss_main": 0.05247202888131142,
      "step": 19030
    },
    {
      "epoch": 2.227942897261877,
      "grad_norm": 0.3281087279319763,
      "learning_rate": 8.170315188455466e-05,
      "loss": 0.0701,
      "step": 19040
    },
    {
      "epoch": 2.227942897261877,
      "loss_main": 0.06418043375015259,
      "step": 19040
    },
    {
      "epoch": 2.22911303533817,
      "grad_norm": 0.47277429699897766,
      "learning_rate": 8.168183711380969e-05,
      "loss": 0.0775,
      "step": 19050
    },
    {
      "epoch": 2.22911303533817,
      "loss_main": 0.07446833699941635,
      "step": 19050
    },
    {
      "epoch": 2.230283173414463,
      "grad_norm": 0.28131330013275146,
      "learning_rate": 8.166051271895913e-05,
      "loss": 0.0767,
      "step": 19060
    },
    {
      "epoch": 2.230283173414463,
      "loss_main": 0.0689031109213829,
      "step": 19060
    },
    {
      "epoch": 2.231453311490756,
      "grad_norm": 0.30771663784980774,
      "learning_rate": 8.163917870648075e-05,
      "loss": 0.0871,
      "step": 19070
    },
    {
      "epoch": 2.231453311490756,
      "loss_main": 0.06406349688768387,
      "step": 19070
    },
    {
      "epoch": 2.2326234495670487,
      "grad_norm": 0.3321216106414795,
      "learning_rate": 8.161783508285526e-05,
      "loss": 0.0773,
      "step": 19080
    },
    {
      "epoch": 2.2326234495670487,
      "loss_main": 0.09414434432983398,
      "step": 19080
    },
    {
      "epoch": 2.233793587643342,
      "grad_norm": 0.37754353880882263,
      "learning_rate": 8.159648185456628e-05,
      "loss": 0.0828,
      "step": 19090
    },
    {
      "epoch": 2.233793587643342,
      "loss_main": 0.06256256252527237,
      "step": 19090
    },
    {
      "epoch": 2.234963725719635,
      "grad_norm": 0.3716461956501007,
      "learning_rate": 8.157511902810038e-05,
      "loss": 0.083,
      "step": 19100
    },
    {
      "epoch": 2.234963725719635,
      "loss_main": 0.06060755252838135,
      "step": 19100
    },
    {
      "epoch": 2.236133863795928,
      "grad_norm": 0.38159680366516113,
      "learning_rate": 8.155374660994701e-05,
      "loss": 0.0719,
      "step": 19110
    },
    {
      "epoch": 2.236133863795928,
      "loss_main": 0.09515295922756195,
      "step": 19110
    },
    {
      "epoch": 2.237304001872221,
      "grad_norm": 0.41686108708381653,
      "learning_rate": 8.153236460659857e-05,
      "loss": 0.0825,
      "step": 19120
    },
    {
      "epoch": 2.237304001872221,
      "loss_main": 0.07330828160047531,
      "step": 19120
    },
    {
      "epoch": 2.2384741399485137,
      "grad_norm": 0.3230416774749756,
      "learning_rate": 8.151097302455031e-05,
      "loss": 0.0717,
      "step": 19130
    },
    {
      "epoch": 2.2384741399485137,
      "loss_main": 0.10139703750610352,
      "step": 19130
    },
    {
      "epoch": 2.239644278024807,
      "grad_norm": 0.3728787302970886,
      "learning_rate": 8.148957187030044e-05,
      "loss": 0.0788,
      "step": 19140
    },
    {
      "epoch": 2.239644278024807,
      "loss_main": 0.06889569014310837,
      "step": 19140
    },
    {
      "epoch": 2.2408144161011,
      "grad_norm": 0.3282225430011749,
      "learning_rate": 8.146816115035006e-05,
      "loss": 0.0764,
      "step": 19150
    },
    {
      "epoch": 2.2408144161011,
      "loss_main": 0.056459940969944,
      "step": 19150
    },
    {
      "epoch": 2.241984554177393,
      "grad_norm": 0.36646586656570435,
      "learning_rate": 8.14467408712032e-05,
      "loss": 0.0801,
      "step": 19160
    },
    {
      "epoch": 2.241984554177393,
      "loss_main": 0.09249770641326904,
      "step": 19160
    },
    {
      "epoch": 2.243154692253686,
      "grad_norm": 0.49683940410614014,
      "learning_rate": 8.142531103936678e-05,
      "loss": 0.0794,
      "step": 19170
    },
    {
      "epoch": 2.243154692253686,
      "loss_main": 0.09077138453722,
      "step": 19170
    },
    {
      "epoch": 2.2443248303299788,
      "grad_norm": 0.26674309372901917,
      "learning_rate": 8.14038716613506e-05,
      "loss": 0.0628,
      "step": 19180
    },
    {
      "epoch": 2.2443248303299788,
      "loss_main": 0.09439652413129807,
      "step": 19180
    },
    {
      "epoch": 2.245494968406272,
      "grad_norm": 0.23545652627944946,
      "learning_rate": 8.138242274366736e-05,
      "loss": 0.073,
      "step": 19190
    },
    {
      "epoch": 2.245494968406272,
      "loss_main": 0.0650240108370781,
      "step": 19190
    },
    {
      "epoch": 2.246665106482565,
      "grad_norm": 0.3433874845504761,
      "learning_rate": 8.136096429283271e-05,
      "loss": 0.0786,
      "step": 19200
    },
    {
      "epoch": 2.246665106482565,
      "loss_main": 0.07270949333906174,
      "step": 19200
    },
    {
      "epoch": 2.247835244558858,
      "grad_norm": 0.40979108214378357,
      "learning_rate": 8.133949631536515e-05,
      "loss": 0.0819,
      "step": 19210
    },
    {
      "epoch": 2.247835244558858,
      "loss_main": 0.07864822447299957,
      "step": 19210
    },
    {
      "epoch": 2.249005382635151,
      "grad_norm": 0.406818151473999,
      "learning_rate": 8.131801881778607e-05,
      "loss": 0.0805,
      "step": 19220
    },
    {
      "epoch": 2.249005382635151,
      "loss_main": 0.06175394728779793,
      "step": 19220
    },
    {
      "epoch": 2.2501755207114438,
      "grad_norm": 0.2528430223464966,
      "learning_rate": 8.129653180661978e-05,
      "loss": 0.0737,
      "step": 19230
    },
    {
      "epoch": 2.2501755207114438,
      "loss_main": 0.056971270591020584,
      "step": 19230
    },
    {
      "epoch": 2.251345658787737,
      "grad_norm": 0.41554680466651917,
      "learning_rate": 8.127503528839346e-05,
      "loss": 0.0791,
      "step": 19240
    },
    {
      "epoch": 2.251345658787737,
      "loss_main": 0.1214175820350647,
      "step": 19240
    },
    {
      "epoch": 2.25251579686403,
      "grad_norm": 0.6363333463668823,
      "learning_rate": 8.125352926963721e-05,
      "loss": 0.0797,
      "step": 19250
    },
    {
      "epoch": 2.25251579686403,
      "loss_main": 0.054315488785505295,
      "step": 19250
    },
    {
      "epoch": 2.253685934940323,
      "grad_norm": 0.27756693959236145,
      "learning_rate": 8.123201375688395e-05,
      "loss": 0.0796,
      "step": 19260
    },
    {
      "epoch": 2.253685934940323,
      "loss_main": 0.055169727653265,
      "step": 19260
    },
    {
      "epoch": 2.254856073016616,
      "grad_norm": 0.31970953941345215,
      "learning_rate": 8.121048875666954e-05,
      "loss": 0.083,
      "step": 19270
    },
    {
      "epoch": 2.254856073016616,
      "loss_main": 0.07737493515014648,
      "step": 19270
    },
    {
      "epoch": 2.256026211092909,
      "grad_norm": 0.25313547253608704,
      "learning_rate": 8.118895427553274e-05,
      "loss": 0.0648,
      "step": 19280
    },
    {
      "epoch": 2.256026211092909,
      "loss_main": 0.09579082578420639,
      "step": 19280
    },
    {
      "epoch": 2.257196349169202,
      "grad_norm": 0.2792856991291046,
      "learning_rate": 8.116741032001511e-05,
      "loss": 0.0784,
      "step": 19290
    },
    {
      "epoch": 2.257196349169202,
      "loss_main": 0.09162657707929611,
      "step": 19290
    },
    {
      "epoch": 2.258366487245495,
      "grad_norm": 0.3054519295692444,
      "learning_rate": 8.114585689666114e-05,
      "loss": 0.0745,
      "step": 19300
    },
    {
      "epoch": 2.258366487245495,
      "loss_main": 0.06392192095518112,
      "step": 19300
    },
    {
      "epoch": 2.259536625321788,
      "grad_norm": 0.4289669394493103,
      "learning_rate": 8.112429401201821e-05,
      "loss": 0.0721,
      "step": 19310
    },
    {
      "epoch": 2.259536625321788,
      "loss_main": 0.15686477720737457,
      "step": 19310
    },
    {
      "epoch": 2.260706763398081,
      "grad_norm": 0.4110340476036072,
      "learning_rate": 8.110272167263656e-05,
      "loss": 0.085,
      "step": 19320
    },
    {
      "epoch": 2.260706763398081,
      "loss_main": 0.08013970404863358,
      "step": 19320
    },
    {
      "epoch": 2.261876901474374,
      "grad_norm": 0.32314828038215637,
      "learning_rate": 8.108113988506929e-05,
      "loss": 0.0721,
      "step": 19330
    },
    {
      "epoch": 2.261876901474374,
      "loss_main": 0.10937709361314774,
      "step": 19330
    },
    {
      "epoch": 2.263047039550667,
      "grad_norm": 0.3833846151828766,
      "learning_rate": 8.105954865587235e-05,
      "loss": 0.0776,
      "step": 19340
    },
    {
      "epoch": 2.263047039550667,
      "loss_main": 0.06698544323444366,
      "step": 19340
    },
    {
      "epoch": 2.2642171776269597,
      "grad_norm": 0.3801696300506592,
      "learning_rate": 8.103794799160463e-05,
      "loss": 0.0678,
      "step": 19350
    },
    {
      "epoch": 2.2642171776269597,
      "loss_main": 0.1030011847615242,
      "step": 19350
    },
    {
      "epoch": 2.265387315703253,
      "grad_norm": 0.3007150888442993,
      "learning_rate": 8.101633789882781e-05,
      "loss": 0.0793,
      "step": 19360
    },
    {
      "epoch": 2.265387315703253,
      "loss_main": 0.05706091597676277,
      "step": 19360
    },
    {
      "epoch": 2.266557453779546,
      "grad_norm": 0.28378623723983765,
      "learning_rate": 8.099471838410648e-05,
      "loss": 0.0754,
      "step": 19370
    },
    {
      "epoch": 2.266557453779546,
      "loss_main": 0.06637462228536606,
      "step": 19370
    },
    {
      "epoch": 2.267727591855839,
      "grad_norm": 0.24560916423797607,
      "learning_rate": 8.097308945400806e-05,
      "loss": 0.076,
      "step": 19380
    },
    {
      "epoch": 2.267727591855839,
      "loss_main": 0.02925536222755909,
      "step": 19380
    },
    {
      "epoch": 2.268897729932132,
      "grad_norm": 0.20077462494373322,
      "learning_rate": 8.095145111510288e-05,
      "loss": 0.0626,
      "step": 19390
    },
    {
      "epoch": 2.268897729932132,
      "loss_main": 0.05536070093512535,
      "step": 19390
    },
    {
      "epoch": 2.270067868008425,
      "grad_norm": 0.3244616687297821,
      "learning_rate": 8.092980337396406e-05,
      "loss": 0.0642,
      "step": 19400
    },
    {
      "epoch": 2.270067868008425,
      "loss_main": 0.07858859747648239,
      "step": 19400
    },
    {
      "epoch": 2.271238006084718,
      "grad_norm": 0.30435341596603394,
      "learning_rate": 8.090814623716763e-05,
      "loss": 0.0746,
      "step": 19410
    },
    {
      "epoch": 2.271238006084718,
      "loss_main": 0.04997970536351204,
      "step": 19410
    },
    {
      "epoch": 2.272408144161011,
      "grad_norm": 0.4061035215854645,
      "learning_rate": 8.088647971129246e-05,
      "loss": 0.0704,
      "step": 19420
    },
    {
      "epoch": 2.272408144161011,
      "loss_main": 0.13841581344604492,
      "step": 19420
    },
    {
      "epoch": 2.273578282237304,
      "grad_norm": 0.2831445038318634,
      "learning_rate": 8.086480380292026e-05,
      "loss": 0.0674,
      "step": 19430
    },
    {
      "epoch": 2.273578282237304,
      "loss_main": 0.027427637949585915,
      "step": 19430
    },
    {
      "epoch": 2.274748420313597,
      "grad_norm": 0.2391502410173416,
      "learning_rate": 8.084311851863562e-05,
      "loss": 0.066,
      "step": 19440
    },
    {
      "epoch": 2.274748420313597,
      "loss_main": 0.07255268096923828,
      "step": 19440
    },
    {
      "epoch": 2.2759185583898898,
      "grad_norm": 0.3247147798538208,
      "learning_rate": 8.082142386502591e-05,
      "loss": 0.0741,
      "step": 19450
    },
    {
      "epoch": 2.2759185583898898,
      "loss_main": 0.12605604529380798,
      "step": 19450
    },
    {
      "epoch": 2.277088696466183,
      "grad_norm": 0.33593788743019104,
      "learning_rate": 8.079971984868145e-05,
      "loss": 0.0824,
      "step": 19460
    },
    {
      "epoch": 2.277088696466183,
      "loss_main": 0.09271209686994553,
      "step": 19460
    },
    {
      "epoch": 2.278258834542476,
      "grad_norm": 0.29758891463279724,
      "learning_rate": 8.077800647619532e-05,
      "loss": 0.0821,
      "step": 19470
    },
    {
      "epoch": 2.278258834542476,
      "loss_main": 0.08788840472698212,
      "step": 19470
    },
    {
      "epoch": 2.279428972618769,
      "grad_norm": 0.21261899173259735,
      "learning_rate": 8.075628375416345e-05,
      "loss": 0.0703,
      "step": 19480
    },
    {
      "epoch": 2.279428972618769,
      "loss_main": 0.05991785600781441,
      "step": 19480
    },
    {
      "epoch": 2.280599110695062,
      "grad_norm": 0.3249334692955017,
      "learning_rate": 8.073455168918464e-05,
      "loss": 0.0699,
      "step": 19490
    },
    {
      "epoch": 2.280599110695062,
      "loss_main": 0.05970633774995804,
      "step": 19490
    },
    {
      "epoch": 2.2817692487713552,
      "grad_norm": 0.3550071716308594,
      "learning_rate": 8.071281028786055e-05,
      "loss": 0.0629,
      "step": 19500
    },
    {
      "epoch": 2.2817692487713552,
      "loss_main": 0.06679404526948929,
      "step": 19500
    },
    {
      "epoch": 2.282939386847648,
      "grad_norm": 0.3460877537727356,
      "learning_rate": 8.069105955679562e-05,
      "loss": 0.0709,
      "step": 19510
    },
    {
      "epoch": 2.282939386847648,
      "loss_main": 0.06279666721820831,
      "step": 19510
    },
    {
      "epoch": 2.284109524923941,
      "grad_norm": 0.3121553957462311,
      "learning_rate": 8.066929950259713e-05,
      "loss": 0.0756,
      "step": 19520
    },
    {
      "epoch": 2.284109524923941,
      "loss_main": 0.06049403175711632,
      "step": 19520
    },
    {
      "epoch": 2.285279663000234,
      "grad_norm": 0.33092746138572693,
      "learning_rate": 8.064753013187522e-05,
      "loss": 0.0724,
      "step": 19530
    },
    {
      "epoch": 2.285279663000234,
      "loss_main": 0.0910472422838211,
      "step": 19530
    },
    {
      "epoch": 2.286449801076527,
      "grad_norm": 0.39661815762519836,
      "learning_rate": 8.062575145124289e-05,
      "loss": 0.0705,
      "step": 19540
    },
    {
      "epoch": 2.286449801076527,
      "loss_main": 0.10252052545547485,
      "step": 19540
    },
    {
      "epoch": 2.28761993915282,
      "grad_norm": 0.34405508637428284,
      "learning_rate": 8.060396346731587e-05,
      "loss": 0.0783,
      "step": 19550
    },
    {
      "epoch": 2.28761993915282,
      "loss_main": 0.034284837543964386,
      "step": 19550
    },
    {
      "epoch": 2.288790077229113,
      "grad_norm": 0.4047397971153259,
      "learning_rate": 8.058216618671281e-05,
      "loss": 0.0692,
      "step": 19560
    },
    {
      "epoch": 2.288790077229113,
      "loss_main": 0.06097688153386116,
      "step": 19560
    },
    {
      "epoch": 2.289960215305406,
      "grad_norm": 0.36139804124832153,
      "learning_rate": 8.056035961605514e-05,
      "loss": 0.0755,
      "step": 19570
    },
    {
      "epoch": 2.289960215305406,
      "loss_main": 0.04900091886520386,
      "step": 19570
    },
    {
      "epoch": 2.291130353381699,
      "grad_norm": 0.285096138715744,
      "learning_rate": 8.05385437619671e-05,
      "loss": 0.0833,
      "step": 19580
    },
    {
      "epoch": 2.291130353381699,
      "loss_main": 0.0720672607421875,
      "step": 19580
    },
    {
      "epoch": 2.292300491457992,
      "grad_norm": 0.18770428001880646,
      "learning_rate": 8.05167186310758e-05,
      "loss": 0.0659,
      "step": 19590
    },
    {
      "epoch": 2.292300491457992,
      "loss_main": 0.058614831417798996,
      "step": 19590
    },
    {
      "epoch": 2.2934706295342853,
      "grad_norm": 0.2057029902935028,
      "learning_rate": 8.049488423001113e-05,
      "loss": 0.0746,
      "step": 19600
    },
    {
      "epoch": 2.2934706295342853,
      "loss_main": 0.10195638984441757,
      "step": 19600
    },
    {
      "epoch": 2.294640767610578,
      "grad_norm": 0.3842117190361023,
      "learning_rate": 8.047304056540581e-05,
      "loss": 0.075,
      "step": 19610
    },
    {
      "epoch": 2.294640767610578,
      "loss_main": 0.07858388125896454,
      "step": 19610
    },
    {
      "epoch": 2.295810905686871,
      "grad_norm": 0.2628209590911865,
      "learning_rate": 8.045118764389534e-05,
      "loss": 0.0773,
      "step": 19620
    },
    {
      "epoch": 2.295810905686871,
      "loss_main": 0.03875424340367317,
      "step": 19620
    },
    {
      "epoch": 2.296981043763164,
      "grad_norm": 0.34225159883499146,
      "learning_rate": 8.042932547211809e-05,
      "loss": 0.0596,
      "step": 19630
    },
    {
      "epoch": 2.296981043763164,
      "loss_main": 0.06389207392930984,
      "step": 19630
    },
    {
      "epoch": 2.298151181839457,
      "grad_norm": 0.2618968188762665,
      "learning_rate": 8.04074540567152e-05,
      "loss": 0.0711,
      "step": 19640
    },
    {
      "epoch": 2.298151181839457,
      "loss_main": 0.07469669729471207,
      "step": 19640
    },
    {
      "epoch": 2.29932131991575,
      "grad_norm": 0.28998905420303345,
      "learning_rate": 8.038557340433063e-05,
      "loss": 0.0811,
      "step": 19650
    },
    {
      "epoch": 2.29932131991575,
      "loss_main": 0.06569015979766846,
      "step": 19650
    },
    {
      "epoch": 2.300491457992043,
      "grad_norm": 0.27928227186203003,
      "learning_rate": 8.036368352161115e-05,
      "loss": 0.073,
      "step": 19660
    },
    {
      "epoch": 2.300491457992043,
      "loss_main": 0.07991163432598114,
      "step": 19660
    },
    {
      "epoch": 2.301661596068336,
      "grad_norm": 0.37396323680877686,
      "learning_rate": 8.034178441520633e-05,
      "loss": 0.0657,
      "step": 19670
    },
    {
      "epoch": 2.301661596068336,
      "loss_main": 0.06902218610048294,
      "step": 19670
    },
    {
      "epoch": 2.302831734144629,
      "grad_norm": 0.25437912344932556,
      "learning_rate": 8.031987609176852e-05,
      "loss": 0.0762,
      "step": 19680
    },
    {
      "epoch": 2.302831734144629,
      "loss_main": 0.057081785053014755,
      "step": 19680
    },
    {
      "epoch": 2.304001872220922,
      "grad_norm": 0.27371662855148315,
      "learning_rate": 8.02979585579529e-05,
      "loss": 0.0709,
      "step": 19690
    },
    {
      "epoch": 2.304001872220922,
      "loss_main": 0.0988655537366867,
      "step": 19690
    },
    {
      "epoch": 2.3051720102972153,
      "grad_norm": 0.3748505115509033,
      "learning_rate": 8.027603182041745e-05,
      "loss": 0.0825,
      "step": 19700
    },
    {
      "epoch": 2.3051720102972153,
      "loss_main": 0.05981980636715889,
      "step": 19700
    },
    {
      "epoch": 2.306342148373508,
      "grad_norm": 0.2635452449321747,
      "learning_rate": 8.025409588582292e-05,
      "loss": 0.069,
      "step": 19710
    },
    {
      "epoch": 2.306342148373508,
      "loss_main": 0.07638085633516312,
      "step": 19710
    },
    {
      "epoch": 2.307512286449801,
      "grad_norm": 0.3484589159488678,
      "learning_rate": 8.023215076083288e-05,
      "loss": 0.0721,
      "step": 19720
    },
    {
      "epoch": 2.307512286449801,
      "loss_main": 0.06365082412958145,
      "step": 19720
    },
    {
      "epoch": 2.308682424526094,
      "grad_norm": 0.2507741451263428,
      "learning_rate": 8.021019645211367e-05,
      "loss": 0.0663,
      "step": 19730
    },
    {
      "epoch": 2.308682424526094,
      "loss_main": 0.06803135573863983,
      "step": 19730
    },
    {
      "epoch": 2.309852562602387,
      "grad_norm": 0.37320956587791443,
      "learning_rate": 8.018823296633441e-05,
      "loss": 0.0687,
      "step": 19740
    },
    {
      "epoch": 2.309852562602387,
      "loss_main": 0.07111118733882904,
      "step": 19740
    },
    {
      "epoch": 2.31102270067868,
      "grad_norm": 0.5493341088294983,
      "learning_rate": 8.016626031016708e-05,
      "loss": 0.0755,
      "step": 19750
    },
    {
      "epoch": 2.31102270067868,
      "loss_main": 0.06348567456007004,
      "step": 19750
    },
    {
      "epoch": 2.312192838754973,
      "grad_norm": 0.28179678320884705,
      "learning_rate": 8.014427849028636e-05,
      "loss": 0.0644,
      "step": 19760
    },
    {
      "epoch": 2.312192838754973,
      "loss_main": 0.05704805999994278,
      "step": 19760
    },
    {
      "epoch": 2.3133629768312662,
      "grad_norm": 0.3142430782318115,
      "learning_rate": 8.012228751336974e-05,
      "loss": 0.069,
      "step": 19770
    },
    {
      "epoch": 2.3133629768312662,
      "loss_main": 0.10708226263523102,
      "step": 19770
    },
    {
      "epoch": 2.314533114907559,
      "grad_norm": 0.3772107660770416,
      "learning_rate": 8.01002873860975e-05,
      "loss": 0.0855,
      "step": 19780
    },
    {
      "epoch": 2.314533114907559,
      "loss_main": 0.04741543531417847,
      "step": 19780
    },
    {
      "epoch": 2.315703252983852,
      "grad_norm": 0.30709078907966614,
      "learning_rate": 8.00782781151527e-05,
      "loss": 0.0766,
      "step": 19790
    },
    {
      "epoch": 2.315703252983852,
      "loss_main": 0.09171364456415176,
      "step": 19790
    },
    {
      "epoch": 2.3168733910601453,
      "grad_norm": 0.1734267771244049,
      "learning_rate": 8.005625970722119e-05,
      "loss": 0.0773,
      "step": 19800
    },
    {
      "epoch": 2.3168733910601453,
      "loss_main": 0.03723456338047981,
      "step": 19800
    },
    {
      "epoch": 2.318043529136438,
      "grad_norm": 0.4161030352115631,
      "learning_rate": 8.003423216899158e-05,
      "loss": 0.0668,
      "step": 19810
    },
    {
      "epoch": 2.318043529136438,
      "loss_main": 0.059375520795583725,
      "step": 19810
    },
    {
      "epoch": 2.3192136672127313,
      "grad_norm": 0.37296828627586365,
      "learning_rate": 8.001219550715522e-05,
      "loss": 0.0794,
      "step": 19820
    },
    {
      "epoch": 2.3192136672127313,
      "loss_main": 0.05288020893931389,
      "step": 19820
    },
    {
      "epoch": 2.320383805289024,
      "grad_norm": 0.3320542871952057,
      "learning_rate": 7.999014972840632e-05,
      "loss": 0.0783,
      "step": 19830
    },
    {
      "epoch": 2.320383805289024,
      "loss_main": 0.06951700150966644,
      "step": 19830
    },
    {
      "epoch": 2.321553943365317,
      "grad_norm": 0.336963415145874,
      "learning_rate": 7.996809483944174e-05,
      "loss": 0.0699,
      "step": 19840
    },
    {
      "epoch": 2.321553943365317,
      "loss_main": 0.07521182298660278,
      "step": 19840
    },
    {
      "epoch": 2.32272408144161,
      "grad_norm": 0.36402982473373413,
      "learning_rate": 7.994603084696124e-05,
      "loss": 0.0656,
      "step": 19850
    },
    {
      "epoch": 2.32272408144161,
      "loss_main": 0.07780016958713531,
      "step": 19850
    },
    {
      "epoch": 2.323894219517903,
      "grad_norm": 0.276295006275177,
      "learning_rate": 7.992395775766724e-05,
      "loss": 0.0619,
      "step": 19860
    },
    {
      "epoch": 2.323894219517903,
      "loss_main": 0.07914156466722488,
      "step": 19860
    },
    {
      "epoch": 2.3250643575941963,
      "grad_norm": 0.26542001962661743,
      "learning_rate": 7.990187557826497e-05,
      "loss": 0.0839,
      "step": 19870
    },
    {
      "epoch": 2.3250643575941963,
      "loss_main": 0.07135795056819916,
      "step": 19870
    },
    {
      "epoch": 2.326234495670489,
      "grad_norm": 0.1732954978942871,
      "learning_rate": 7.987978431546242e-05,
      "loss": 0.076,
      "step": 19880
    },
    {
      "epoch": 2.326234495670489,
      "loss_main": 0.06820006668567657,
      "step": 19880
    },
    {
      "epoch": 2.327404633746782,
      "grad_norm": 0.46394652128219604,
      "learning_rate": 7.985768397597031e-05,
      "loss": 0.0781,
      "step": 19890
    },
    {
      "epoch": 2.327404633746782,
      "loss_main": 0.06338765472173691,
      "step": 19890
    },
    {
      "epoch": 2.328574771823075,
      "grad_norm": 0.5106828808784485,
      "learning_rate": 7.983557456650216e-05,
      "loss": 0.0848,
      "step": 19900
    },
    {
      "epoch": 2.328574771823075,
      "loss_main": 0.08015595376491547,
      "step": 19900
    },
    {
      "epoch": 2.329744909899368,
      "grad_norm": 0.41432416439056396,
      "learning_rate": 7.981345609377422e-05,
      "loss": 0.073,
      "step": 19910
    },
    {
      "epoch": 2.329744909899368,
      "loss_main": 0.09755551815032959,
      "step": 19910
    },
    {
      "epoch": 2.3309150479756613,
      "grad_norm": 0.3082341253757477,
      "learning_rate": 7.97913285645055e-05,
      "loss": 0.0792,
      "step": 19920
    },
    {
      "epoch": 2.3309150479756613,
      "loss_main": 0.0627073347568512,
      "step": 19920
    },
    {
      "epoch": 2.332085186051954,
      "grad_norm": 0.31851688027381897,
      "learning_rate": 7.976919198541776e-05,
      "loss": 0.0692,
      "step": 19930
    },
    {
      "epoch": 2.332085186051954,
      "loss_main": 0.05044403672218323,
      "step": 19930
    },
    {
      "epoch": 2.333255324128247,
      "grad_norm": 0.2915154695510864,
      "learning_rate": 7.974704636323548e-05,
      "loss": 0.0764,
      "step": 19940
    },
    {
      "epoch": 2.333255324128247,
      "loss_main": 0.06630958616733551,
      "step": 19940
    },
    {
      "epoch": 2.33442546220454,
      "grad_norm": 0.2709543704986572,
      "learning_rate": 7.972489170468597e-05,
      "loss": 0.0663,
      "step": 19950
    },
    {
      "epoch": 2.33442546220454,
      "loss_main": 0.0824640616774559,
      "step": 19950
    },
    {
      "epoch": 2.335595600280833,
      "grad_norm": 0.2682894170284271,
      "learning_rate": 7.970272801649918e-05,
      "loss": 0.0812,
      "step": 19960
    },
    {
      "epoch": 2.335595600280833,
      "loss_main": 0.08768636733293533,
      "step": 19960
    },
    {
      "epoch": 2.3367657383571263,
      "grad_norm": 0.2790411412715912,
      "learning_rate": 7.96805553054079e-05,
      "loss": 0.0731,
      "step": 19970
    },
    {
      "epoch": 2.3367657383571263,
      "loss_main": 0.0656481385231018,
      "step": 19970
    },
    {
      "epoch": 2.337935876433419,
      "grad_norm": 0.41558972001075745,
      "learning_rate": 7.965837357814756e-05,
      "loss": 0.0685,
      "step": 19980
    },
    {
      "epoch": 2.337935876433419,
      "loss_main": 0.0752163678407669,
      "step": 19980
    },
    {
      "epoch": 2.3391060145097122,
      "grad_norm": 0.2929171621799469,
      "learning_rate": 7.963618284145643e-05,
      "loss": 0.0767,
      "step": 19990
    },
    {
      "epoch": 2.3391060145097122,
      "loss_main": 0.1367669254541397,
      "step": 19990
    },
    {
      "epoch": 2.340276152586005,
      "grad_norm": 0.2971945106983185,
      "learning_rate": 7.961398310207544e-05,
      "loss": 0.0756,
      "step": 20000
    },
    {
      "epoch": 2.340276152586005,
      "loss_main": 0.0450674369931221,
      "step": 20000
    },
    {
      "epoch": 2.341446290662298,
      "grad_norm": 0.3153098523616791,
      "learning_rate": 7.95917743667483e-05,
      "loss": 0.0686,
      "step": 20010
    },
    {
      "epoch": 2.341446290662298,
      "loss_main": 0.05134334787726402,
      "step": 20010
    },
    {
      "epoch": 2.3426164287385913,
      "grad_norm": 0.2995058000087738,
      "learning_rate": 7.956955664222144e-05,
      "loss": 0.0746,
      "step": 20020
    },
    {
      "epoch": 2.3426164287385913,
      "loss_main": 0.08526802062988281,
      "step": 20020
    },
    {
      "epoch": 2.343786566814884,
      "grad_norm": 0.30066990852355957,
      "learning_rate": 7.954732993524399e-05,
      "loss": 0.075,
      "step": 20030
    },
    {
      "epoch": 2.343786566814884,
      "loss_main": 0.0785590261220932,
      "step": 20030
    },
    {
      "epoch": 2.3449567048911772,
      "grad_norm": 0.244106262922287,
      "learning_rate": 7.952509425256786e-05,
      "loss": 0.0748,
      "step": 20040
    },
    {
      "epoch": 2.3449567048911772,
      "loss_main": 0.0606185607612133,
      "step": 20040
    },
    {
      "epoch": 2.34612684296747,
      "grad_norm": 0.36494702100753784,
      "learning_rate": 7.950284960094767e-05,
      "loss": 0.0726,
      "step": 20050
    },
    {
      "epoch": 2.34612684296747,
      "loss_main": 0.09165777266025543,
      "step": 20050
    },
    {
      "epoch": 2.347296981043763,
      "grad_norm": 0.44688549637794495,
      "learning_rate": 7.948059598714076e-05,
      "loss": 0.0708,
      "step": 20060
    },
    {
      "epoch": 2.347296981043763,
      "loss_main": 0.06621339172124863,
      "step": 20060
    },
    {
      "epoch": 2.3484671191200563,
      "grad_norm": 0.2906399071216583,
      "learning_rate": 7.945833341790717e-05,
      "loss": 0.0728,
      "step": 20070
    },
    {
      "epoch": 2.3484671191200563,
      "loss_main": 0.05592688173055649,
      "step": 20070
    },
    {
      "epoch": 2.349637257196349,
      "grad_norm": 0.4560497999191284,
      "learning_rate": 7.94360619000097e-05,
      "loss": 0.0779,
      "step": 20080
    },
    {
      "epoch": 2.349637257196349,
      "loss_main": 0.0456450879573822,
      "step": 20080
    },
    {
      "epoch": 2.3508073952726423,
      "grad_norm": 0.47557178139686584,
      "learning_rate": 7.941378144021381e-05,
      "loss": 0.0699,
      "step": 20090
    },
    {
      "epoch": 2.3508073952726423,
      "loss_main": 0.10805679857730865,
      "step": 20090
    },
    {
      "epoch": 2.351977533348935,
      "grad_norm": 0.2700778841972351,
      "learning_rate": 7.939149204528777e-05,
      "loss": 0.0716,
      "step": 20100
    },
    {
      "epoch": 2.351977533348935,
      "loss_main": 0.07297991216182709,
      "step": 20100
    },
    {
      "epoch": 2.353147671425228,
      "grad_norm": 0.28568243980407715,
      "learning_rate": 7.936919372200246e-05,
      "loss": 0.0727,
      "step": 20110
    },
    {
      "epoch": 2.353147671425228,
      "loss_main": 0.051301248371601105,
      "step": 20110
    },
    {
      "epoch": 2.3543178095015214,
      "grad_norm": 0.34929385781288147,
      "learning_rate": 7.934688647713158e-05,
      "loss": 0.0719,
      "step": 20120
    },
    {
      "epoch": 2.3543178095015214,
      "loss_main": 0.036450546234846115,
      "step": 20120
    },
    {
      "epoch": 2.355487947577814,
      "grad_norm": 0.27691006660461426,
      "learning_rate": 7.932457031745143e-05,
      "loss": 0.0678,
      "step": 20130
    },
    {
      "epoch": 2.355487947577814,
      "loss_main": 0.13598163425922394,
      "step": 20130
    },
    {
      "epoch": 2.3566580856541073,
      "grad_norm": 0.44668254256248474,
      "learning_rate": 7.930224524974108e-05,
      "loss": 0.0788,
      "step": 20140
    },
    {
      "epoch": 2.3566580856541073,
      "loss_main": 0.08226770907640457,
      "step": 20140
    },
    {
      "epoch": 2.3578282237304,
      "grad_norm": 0.34965646266937256,
      "learning_rate": 7.927991128078232e-05,
      "loss": 0.0631,
      "step": 20150
    },
    {
      "epoch": 2.3578282237304,
      "loss_main": 0.09239304065704346,
      "step": 20150
    },
    {
      "epoch": 2.358998361806693,
      "grad_norm": 0.29827481508255005,
      "learning_rate": 7.925756841735958e-05,
      "loss": 0.0665,
      "step": 20160
    },
    {
      "epoch": 2.358998361806693,
      "loss_main": 0.0948604866862297,
      "step": 20160
    },
    {
      "epoch": 2.3601684998829864,
      "grad_norm": 0.35148417949676514,
      "learning_rate": 7.923521666626008e-05,
      "loss": 0.0715,
      "step": 20170
    },
    {
      "epoch": 2.3601684998829864,
      "loss_main": 0.09511605650186539,
      "step": 20170
    },
    {
      "epoch": 2.361338637959279,
      "grad_norm": 0.433696985244751,
      "learning_rate": 7.921285603427366e-05,
      "loss": 0.0662,
      "step": 20180
    },
    {
      "epoch": 2.361338637959279,
      "loss_main": 0.0931628942489624,
      "step": 20180
    },
    {
      "epoch": 2.3625087760355723,
      "grad_norm": 0.3667084872722626,
      "learning_rate": 7.91904865281929e-05,
      "loss": 0.0735,
      "step": 20190
    },
    {
      "epoch": 2.3625087760355723,
      "loss_main": 0.07938943803310394,
      "step": 20190
    },
    {
      "epoch": 2.363678914111865,
      "grad_norm": 0.259941428899765,
      "learning_rate": 7.916810815481307e-05,
      "loss": 0.0716,
      "step": 20200
    },
    {
      "epoch": 2.363678914111865,
      "loss_main": 0.054017920047044754,
      "step": 20200
    },
    {
      "epoch": 2.364849052188158,
      "grad_norm": 0.4147595465183258,
      "learning_rate": 7.914572092093211e-05,
      "loss": 0.0728,
      "step": 20210
    },
    {
      "epoch": 2.364849052188158,
      "loss_main": 0.0792398452758789,
      "step": 20210
    },
    {
      "epoch": 2.3660191902644514,
      "grad_norm": 0.3364831507205963,
      "learning_rate": 7.912332483335068e-05,
      "loss": 0.0801,
      "step": 20220
    },
    {
      "epoch": 2.3660191902644514,
      "loss_main": 0.05464467406272888,
      "step": 20220
    },
    {
      "epoch": 2.367189328340744,
      "grad_norm": 0.3285196125507355,
      "learning_rate": 7.910091989887213e-05,
      "loss": 0.0629,
      "step": 20230
    },
    {
      "epoch": 2.367189328340744,
      "loss_main": 0.06917320936918259,
      "step": 20230
    },
    {
      "epoch": 2.3683594664170373,
      "grad_norm": 0.3575761616230011,
      "learning_rate": 7.907850612430248e-05,
      "loss": 0.0759,
      "step": 20240
    },
    {
      "epoch": 2.3683594664170373,
      "loss_main": 0.05158890411257744,
      "step": 20240
    },
    {
      "epoch": 2.36952960449333,
      "grad_norm": 0.3483484089374542,
      "learning_rate": 7.905608351645044e-05,
      "loss": 0.0668,
      "step": 20250
    },
    {
      "epoch": 2.36952960449333,
      "loss_main": 0.07469955831766129,
      "step": 20250
    },
    {
      "epoch": 2.3706997425696232,
      "grad_norm": 0.43430110812187195,
      "learning_rate": 7.90336520821274e-05,
      "loss": 0.0748,
      "step": 20260
    },
    {
      "epoch": 2.3706997425696232,
      "loss_main": 0.05194922909140587,
      "step": 20260
    },
    {
      "epoch": 2.3718698806459164,
      "grad_norm": 0.26968812942504883,
      "learning_rate": 7.901121182814746e-05,
      "loss": 0.0629,
      "step": 20270
    },
    {
      "epoch": 2.3718698806459164,
      "loss_main": 0.04169589653611183,
      "step": 20270
    },
    {
      "epoch": 2.373040018722209,
      "grad_norm": 0.2927400767803192,
      "learning_rate": 7.898876276132736e-05,
      "loss": 0.0735,
      "step": 20280
    },
    {
      "epoch": 2.373040018722209,
      "loss_main": 0.05916951969265938,
      "step": 20280
    },
    {
      "epoch": 2.3742101567985023,
      "grad_norm": 0.23718786239624023,
      "learning_rate": 7.896630488848654e-05,
      "loss": 0.0721,
      "step": 20290
    },
    {
      "epoch": 2.3742101567985023,
      "loss_main": 0.0785338506102562,
      "step": 20290
    },
    {
      "epoch": 2.375380294874795,
      "grad_norm": 0.2840544879436493,
      "learning_rate": 7.89438382164471e-05,
      "loss": 0.0716,
      "step": 20300
    },
    {
      "epoch": 2.375380294874795,
      "loss_main": 0.06742014735937119,
      "step": 20300
    },
    {
      "epoch": 2.3765504329510883,
      "grad_norm": 0.2946728467941284,
      "learning_rate": 7.892136275203383e-05,
      "loss": 0.0723,
      "step": 20310
    },
    {
      "epoch": 2.3765504329510883,
      "loss_main": 0.08828134089708328,
      "step": 20310
    },
    {
      "epoch": 2.377720571027381,
      "grad_norm": 0.32028621435165405,
      "learning_rate": 7.889887850207418e-05,
      "loss": 0.0743,
      "step": 20320
    },
    {
      "epoch": 2.377720571027381,
      "loss_main": 0.1275913566350937,
      "step": 20320
    },
    {
      "epoch": 2.378890709103674,
      "grad_norm": 0.26858091354370117,
      "learning_rate": 7.887638547339827e-05,
      "loss": 0.0866,
      "step": 20330
    },
    {
      "epoch": 2.378890709103674,
      "loss_main": 0.06803280115127563,
      "step": 20330
    },
    {
      "epoch": 2.3800608471799674,
      "grad_norm": 0.23646394908428192,
      "learning_rate": 7.885388367283891e-05,
      "loss": 0.0654,
      "step": 20340
    },
    {
      "epoch": 2.3800608471799674,
      "loss_main": 0.07642572373151779,
      "step": 20340
    },
    {
      "epoch": 2.38123098525626,
      "grad_norm": 0.29710498452186584,
      "learning_rate": 7.88313731072315e-05,
      "loss": 0.079,
      "step": 20350
    },
    {
      "epoch": 2.38123098525626,
      "loss_main": 0.08126314729452133,
      "step": 20350
    },
    {
      "epoch": 2.3824011233325533,
      "grad_norm": 0.28921544551849365,
      "learning_rate": 7.88088537834142e-05,
      "loss": 0.0698,
      "step": 20360
    },
    {
      "epoch": 2.3824011233325533,
      "loss_main": 0.09909182786941528,
      "step": 20360
    },
    {
      "epoch": 2.3835712614088465,
      "grad_norm": 0.48580116033554077,
      "learning_rate": 7.878632570822778e-05,
      "loss": 0.0747,
      "step": 20370
    },
    {
      "epoch": 2.3835712614088465,
      "loss_main": 0.060475535690784454,
      "step": 20370
    },
    {
      "epoch": 2.384741399485139,
      "grad_norm": 0.3941776156425476,
      "learning_rate": 7.876378888851567e-05,
      "loss": 0.0709,
      "step": 20380
    },
    {
      "epoch": 2.384741399485139,
      "loss_main": 0.06972803920507431,
      "step": 20380
    },
    {
      "epoch": 2.3859115375614324,
      "grad_norm": 0.27806589007377625,
      "learning_rate": 7.874124333112396e-05,
      "loss": 0.0715,
      "step": 20390
    },
    {
      "epoch": 2.3859115375614324,
      "loss_main": 0.0642547756433487,
      "step": 20390
    },
    {
      "epoch": 2.387081675637725,
      "grad_norm": 0.28776487708091736,
      "learning_rate": 7.871868904290138e-05,
      "loss": 0.0665,
      "step": 20400
    },
    {
      "epoch": 2.387081675637725,
      "loss_main": 0.10245926678180695,
      "step": 20400
    },
    {
      "epoch": 2.3882518137140183,
      "grad_norm": 0.3578380346298218,
      "learning_rate": 7.869612603069935e-05,
      "loss": 0.0697,
      "step": 20410
    },
    {
      "epoch": 2.3882518137140183,
      "loss_main": 0.047804493457078934,
      "step": 20410
    },
    {
      "epoch": 2.389421951790311,
      "grad_norm": 0.27758094668388367,
      "learning_rate": 7.867355430137192e-05,
      "loss": 0.0638,
      "step": 20420
    },
    {
      "epoch": 2.389421951790311,
      "loss_main": 0.07034717500209808,
      "step": 20420
    },
    {
      "epoch": 2.390592089866604,
      "grad_norm": 0.3139849603176117,
      "learning_rate": 7.865097386177577e-05,
      "loss": 0.0754,
      "step": 20430
    },
    {
      "epoch": 2.390592089866604,
      "loss_main": 0.13783100247383118,
      "step": 20430
    },
    {
      "epoch": 2.3917622279428974,
      "grad_norm": 0.42173340916633606,
      "learning_rate": 7.862838471877023e-05,
      "loss": 0.0768,
      "step": 20440
    },
    {
      "epoch": 2.3917622279428974,
      "loss_main": 0.09197268635034561,
      "step": 20440
    },
    {
      "epoch": 2.39293236601919,
      "grad_norm": 0.20088514685630798,
      "learning_rate": 7.860578687921731e-05,
      "loss": 0.075,
      "step": 20450
    },
    {
      "epoch": 2.39293236601919,
      "loss_main": 0.06135066971182823,
      "step": 20450
    },
    {
      "epoch": 2.3941025040954833,
      "grad_norm": 0.35170239210128784,
      "learning_rate": 7.858318034998164e-05,
      "loss": 0.0666,
      "step": 20460
    },
    {
      "epoch": 2.3941025040954833,
      "loss_main": 0.07695197314023972,
      "step": 20460
    },
    {
      "epoch": 2.3952726421717765,
      "grad_norm": 0.29142311215400696,
      "learning_rate": 7.856056513793046e-05,
      "loss": 0.0747,
      "step": 20470
    },
    {
      "epoch": 2.3952726421717765,
      "loss_main": 0.07946092635393143,
      "step": 20470
    },
    {
      "epoch": 2.3964427802480692,
      "grad_norm": 0.23438315093517303,
      "learning_rate": 7.85379412499337e-05,
      "loss": 0.0658,
      "step": 20480
    },
    {
      "epoch": 2.3964427802480692,
      "loss_main": 0.07144047319889069,
      "step": 20480
    },
    {
      "epoch": 2.3976129183243624,
      "grad_norm": 0.22924436628818512,
      "learning_rate": 7.851530869286389e-05,
      "loss": 0.0759,
      "step": 20490
    },
    {
      "epoch": 2.3976129183243624,
      "loss_main": 0.053362853825092316,
      "step": 20490
    },
    {
      "epoch": 2.398783056400655,
      "grad_norm": 0.3871263563632965,
      "learning_rate": 7.849266747359619e-05,
      "loss": 0.0677,
      "step": 20500
    },
    {
      "epoch": 2.398783056400655,
      "loss_main": 0.04015529155731201,
      "step": 20500
    },
    {
      "epoch": 2.3999531944769483,
      "grad_norm": 0.3518255650997162,
      "learning_rate": 7.847001759900843e-05,
      "loss": 0.0548,
      "step": 20510
    },
    {
      "epoch": 2.3999531944769483,
      "loss_main": 0.057390809059143066,
      "step": 20510
    },
    {
      "epoch": 2.401123332553241,
      "grad_norm": 0.26282525062561035,
      "learning_rate": 7.844735907598102e-05,
      "loss": 0.0713,
      "step": 20520
    },
    {
      "epoch": 2.401123332553241,
      "loss_main": 0.12953898310661316,
      "step": 20520
    },
    {
      "epoch": 2.4022934706295342,
      "grad_norm": 0.27998894453048706,
      "learning_rate": 7.842469191139703e-05,
      "loss": 0.0813,
      "step": 20530
    },
    {
      "epoch": 2.4022934706295342,
      "loss_main": 0.06853140145540237,
      "step": 20530
    },
    {
      "epoch": 2.4034636087058274,
      "grad_norm": 0.297843337059021,
      "learning_rate": 7.840201611214215e-05,
      "loss": 0.0707,
      "step": 20540
    },
    {
      "epoch": 2.4034636087058274,
      "loss_main": 0.09129071235656738,
      "step": 20540
    },
    {
      "epoch": 2.40463374678212,
      "grad_norm": 0.3827267289161682,
      "learning_rate": 7.837933168510469e-05,
      "loss": 0.083,
      "step": 20550
    },
    {
      "epoch": 2.40463374678212,
      "loss_main": 0.06943512707948685,
      "step": 20550
    },
    {
      "epoch": 2.4058038848584133,
      "grad_norm": 0.32068467140197754,
      "learning_rate": 7.835663863717559e-05,
      "loss": 0.0743,
      "step": 20560
    },
    {
      "epoch": 2.4058038848584133,
      "loss_main": 0.05716345086693764,
      "step": 20560
    },
    {
      "epoch": 2.4069740229347065,
      "grad_norm": 0.3739340901374817,
      "learning_rate": 7.833393697524838e-05,
      "loss": 0.0732,
      "step": 20570
    },
    {
      "epoch": 2.4069740229347065,
      "loss_main": 0.06742479652166367,
      "step": 20570
    },
    {
      "epoch": 2.4081441610109993,
      "grad_norm": 0.32607847452163696,
      "learning_rate": 7.831122670621922e-05,
      "loss": 0.074,
      "step": 20580
    },
    {
      "epoch": 2.4081441610109993,
      "loss_main": 0.044572681188583374,
      "step": 20580
    },
    {
      "epoch": 2.4093142990872924,
      "grad_norm": 0.33365902304649353,
      "learning_rate": 7.82885078369869e-05,
      "loss": 0.0733,
      "step": 20590
    },
    {
      "epoch": 2.4093142990872924,
      "loss_main": 0.05179215595126152,
      "step": 20590
    },
    {
      "epoch": 2.410484437163585,
      "grad_norm": 0.26518136262893677,
      "learning_rate": 7.826578037445283e-05,
      "loss": 0.0722,
      "step": 20600
    },
    {
      "epoch": 2.410484437163585,
      "loss_main": 0.07974036782979965,
      "step": 20600
    },
    {
      "epoch": 2.4116545752398784,
      "grad_norm": 0.22703786194324493,
      "learning_rate": 7.824304432552097e-05,
      "loss": 0.0641,
      "step": 20610
    },
    {
      "epoch": 2.4116545752398784,
      "loss_main": 0.07291389256715775,
      "step": 20610
    },
    {
      "epoch": 2.412824713316171,
      "grad_norm": 0.26543381810188293,
      "learning_rate": 7.822029969709798e-05,
      "loss": 0.079,
      "step": 20620
    },
    {
      "epoch": 2.412824713316171,
      "loss_main": 0.09196964651346207,
      "step": 20620
    },
    {
      "epoch": 2.4139948513924643,
      "grad_norm": 0.2837192416191101,
      "learning_rate": 7.819754649609306e-05,
      "loss": 0.0761,
      "step": 20630
    },
    {
      "epoch": 2.4139948513924643,
      "loss_main": 0.0448126494884491,
      "step": 20630
    },
    {
      "epoch": 2.4151649894687575,
      "grad_norm": 0.3688303232192993,
      "learning_rate": 7.817478472941802e-05,
      "loss": 0.0658,
      "step": 20640
    },
    {
      "epoch": 2.4151649894687575,
      "loss_main": 0.03832319751381874,
      "step": 20640
    },
    {
      "epoch": 2.41633512754505,
      "grad_norm": 0.2354583442211151,
      "learning_rate": 7.815201440398727e-05,
      "loss": 0.0564,
      "step": 20650
    },
    {
      "epoch": 2.41633512754505,
      "loss_main": 0.10925284773111343,
      "step": 20650
    },
    {
      "epoch": 2.4175052656213434,
      "grad_norm": 0.2827816605567932,
      "learning_rate": 7.812923552671789e-05,
      "loss": 0.0678,
      "step": 20660
    },
    {
      "epoch": 2.4175052656213434,
      "loss_main": 0.09583734720945358,
      "step": 20660
    },
    {
      "epoch": 2.4186754036976366,
      "grad_norm": 0.318173885345459,
      "learning_rate": 7.810644810452945e-05,
      "loss": 0.0758,
      "step": 20670
    },
    {
      "epoch": 2.4186754036976366,
      "loss_main": 0.06891829520463943,
      "step": 20670
    },
    {
      "epoch": 2.4198455417739293,
      "grad_norm": 0.3645920157432556,
      "learning_rate": 7.808365214434417e-05,
      "loss": 0.0739,
      "step": 20680
    },
    {
      "epoch": 2.4198455417739293,
      "loss_main": 0.0529792346060276,
      "step": 20680
    },
    {
      "epoch": 2.4210156798502225,
      "grad_norm": 0.3695385754108429,
      "learning_rate": 7.80608476530869e-05,
      "loss": 0.0749,
      "step": 20690
    },
    {
      "epoch": 2.4210156798502225,
      "loss_main": 0.07516548782587051,
      "step": 20690
    },
    {
      "epoch": 2.422185817926515,
      "grad_norm": 0.33829465508461,
      "learning_rate": 7.8038034637685e-05,
      "loss": 0.0701,
      "step": 20700
    },
    {
      "epoch": 2.422185817926515,
      "loss_main": 0.07481793314218521,
      "step": 20700
    },
    {
      "epoch": 2.4233559560028084,
      "grad_norm": 0.2914961874485016,
      "learning_rate": 7.801521310506848e-05,
      "loss": 0.0594,
      "step": 20710
    },
    {
      "epoch": 2.4233559560028084,
      "loss_main": 0.07210731506347656,
      "step": 20710
    },
    {
      "epoch": 2.424526094079101,
      "grad_norm": 0.3873600661754608,
      "learning_rate": 7.799238306216994e-05,
      "loss": 0.0714,
      "step": 20720
    },
    {
      "epoch": 2.424526094079101,
      "loss_main": 0.06958923488855362,
      "step": 20720
    },
    {
      "epoch": 2.4256962321553943,
      "grad_norm": 0.23645024001598358,
      "learning_rate": 7.796954451592448e-05,
      "loss": 0.0725,
      "step": 20730
    },
    {
      "epoch": 2.4256962321553943,
      "loss_main": 0.06366721540689468,
      "step": 20730
    },
    {
      "epoch": 2.4268663702316875,
      "grad_norm": 0.4076775014400482,
      "learning_rate": 7.794669747326992e-05,
      "loss": 0.0711,
      "step": 20740
    },
    {
      "epoch": 2.4268663702316875,
      "loss_main": 0.06889327615499496,
      "step": 20740
    },
    {
      "epoch": 2.4280365083079802,
      "grad_norm": 0.26394131779670715,
      "learning_rate": 7.792384194114654e-05,
      "loss": 0.0646,
      "step": 20750
    },
    {
      "epoch": 2.4280365083079802,
      "loss_main": 0.049373380839824677,
      "step": 20750
    },
    {
      "epoch": 2.4292066463842734,
      "grad_norm": 0.2991075813770294,
      "learning_rate": 7.790097792649729e-05,
      "loss": 0.0712,
      "step": 20760
    },
    {
      "epoch": 2.4292066463842734,
      "loss_main": 0.11705943197011948,
      "step": 20760
    },
    {
      "epoch": 2.430376784460566,
      "grad_norm": 0.28555968403816223,
      "learning_rate": 7.787810543626762e-05,
      "loss": 0.0767,
      "step": 20770
    },
    {
      "epoch": 2.430376784460566,
      "loss_main": 0.06443370133638382,
      "step": 20770
    },
    {
      "epoch": 2.4315469225368593,
      "grad_norm": 0.2327127903699875,
      "learning_rate": 7.785522447740558e-05,
      "loss": 0.0733,
      "step": 20780
    },
    {
      "epoch": 2.4315469225368593,
      "loss_main": 0.06972327083349228,
      "step": 20780
    },
    {
      "epoch": 2.4327170606131525,
      "grad_norm": 0.25204962491989136,
      "learning_rate": 7.783233505686182e-05,
      "loss": 0.0754,
      "step": 20790
    },
    {
      "epoch": 2.4327170606131525,
      "loss_main": 0.05498596653342247,
      "step": 20790
    },
    {
      "epoch": 2.4338871986894453,
      "grad_norm": 0.2740360498428345,
      "learning_rate": 7.780943718158955e-05,
      "loss": 0.0574,
      "step": 20800
    },
    {
      "epoch": 2.4338871986894453,
      "loss_main": 0.07199253141880035,
      "step": 20800
    },
    {
      "epoch": 2.4350573367657384,
      "grad_norm": 0.25155749917030334,
      "learning_rate": 7.778653085854453e-05,
      "loss": 0.068,
      "step": 20810
    },
    {
      "epoch": 2.4350573367657384,
      "loss_main": 0.07125819474458694,
      "step": 20810
    },
    {
      "epoch": 2.436227474842031,
      "grad_norm": 0.272645503282547,
      "learning_rate": 7.77636160946851e-05,
      "loss": 0.0639,
      "step": 20820
    },
    {
      "epoch": 2.436227474842031,
      "loss_main": 0.05634590610861778,
      "step": 20820
    },
    {
      "epoch": 2.4373976129183244,
      "grad_norm": 0.3230745792388916,
      "learning_rate": 7.774069289697215e-05,
      "loss": 0.0695,
      "step": 20830
    },
    {
      "epoch": 2.4373976129183244,
      "loss_main": 0.044886477291584015,
      "step": 20830
    },
    {
      "epoch": 2.4385677509946175,
      "grad_norm": 0.3536348044872284,
      "learning_rate": 7.771776127236913e-05,
      "loss": 0.0726,
      "step": 20840
    },
    {
      "epoch": 2.4385677509946175,
      "loss_main": 0.062474191188812256,
      "step": 20840
    },
    {
      "epoch": 2.4397378890709103,
      "grad_norm": 0.2900225818157196,
      "learning_rate": 7.769482122784212e-05,
      "loss": 0.0638,
      "step": 20850
    },
    {
      "epoch": 2.4397378890709103,
      "loss_main": 0.08442170917987823,
      "step": 20850
    },
    {
      "epoch": 2.4409080271472035,
      "grad_norm": 0.352429062128067,
      "learning_rate": 7.767187277035963e-05,
      "loss": 0.072,
      "step": 20860
    },
    {
      "epoch": 2.4409080271472035,
      "loss_main": 0.0778956264257431,
      "step": 20860
    },
    {
      "epoch": 2.442078165223496,
      "grad_norm": 0.27511122822761536,
      "learning_rate": 7.764891590689285e-05,
      "loss": 0.0758,
      "step": 20870
    },
    {
      "epoch": 2.442078165223496,
      "loss_main": 0.07978038489818573,
      "step": 20870
    },
    {
      "epoch": 2.4432483032997894,
      "grad_norm": 0.27850422263145447,
      "learning_rate": 7.762595064441542e-05,
      "loss": 0.0672,
      "step": 20880
    },
    {
      "epoch": 2.4432483032997894,
      "loss_main": 0.07358551025390625,
      "step": 20880
    },
    {
      "epoch": 2.4444184413760826,
      "grad_norm": 0.3242294490337372,
      "learning_rate": 7.760297698990362e-05,
      "loss": 0.07,
      "step": 20890
    },
    {
      "epoch": 2.4444184413760826,
      "loss_main": 0.07228141278028488,
      "step": 20890
    },
    {
      "epoch": 2.4455885794523753,
      "grad_norm": 0.31478795409202576,
      "learning_rate": 7.757999495033623e-05,
      "loss": 0.0698,
      "step": 20900
    },
    {
      "epoch": 2.4455885794523753,
      "loss_main": 0.05991249531507492,
      "step": 20900
    },
    {
      "epoch": 2.4467587175286685,
      "grad_norm": 0.22266191244125366,
      "learning_rate": 7.755700453269456e-05,
      "loss": 0.0642,
      "step": 20910
    },
    {
      "epoch": 2.4467587175286685,
      "loss_main": 0.04118669033050537,
      "step": 20910
    },
    {
      "epoch": 2.447928855604961,
      "grad_norm": 0.23743575811386108,
      "learning_rate": 7.753400574396254e-05,
      "loss": 0.0766,
      "step": 20920
    },
    {
      "epoch": 2.447928855604961,
      "loss_main": 0.0773337110877037,
      "step": 20920
    },
    {
      "epoch": 2.4490989936812544,
      "grad_norm": 0.34530845284461975,
      "learning_rate": 7.751099859112655e-05,
      "loss": 0.0748,
      "step": 20930
    },
    {
      "epoch": 2.4490989936812544,
      "loss_main": 0.04357241839170456,
      "step": 20930
    },
    {
      "epoch": 2.4502691317575476,
      "grad_norm": 0.3995584547519684,
      "learning_rate": 7.748798308117557e-05,
      "loss": 0.0602,
      "step": 20940
    },
    {
      "epoch": 2.4502691317575476,
      "loss_main": 0.054397743195295334,
      "step": 20940
    },
    {
      "epoch": 2.4514392698338403,
      "grad_norm": 0.29822465777397156,
      "learning_rate": 7.746495922110112e-05,
      "loss": 0.0698,
      "step": 20950
    },
    {
      "epoch": 2.4514392698338403,
      "loss_main": 0.10736081749200821,
      "step": 20950
    },
    {
      "epoch": 2.4526094079101335,
      "grad_norm": 0.2816818654537201,
      "learning_rate": 7.744192701789723e-05,
      "loss": 0.0691,
      "step": 20960
    },
    {
      "epoch": 2.4526094079101335,
      "loss_main": 0.06146598607301712,
      "step": 20960
    },
    {
      "epoch": 2.4537795459864262,
      "grad_norm": 0.32565852999687195,
      "learning_rate": 7.741888647856046e-05,
      "loss": 0.0632,
      "step": 20970
    },
    {
      "epoch": 2.4537795459864262,
      "loss_main": 0.0859430655837059,
      "step": 20970
    },
    {
      "epoch": 2.4549496840627194,
      "grad_norm": 0.2161569893360138,
      "learning_rate": 7.739583761008994e-05,
      "loss": 0.0699,
      "step": 20980
    },
    {
      "epoch": 2.4549496840627194,
      "loss_main": 0.0958796888589859,
      "step": 20980
    },
    {
      "epoch": 2.4561198221390126,
      "grad_norm": 0.239335834980011,
      "learning_rate": 7.73727804194873e-05,
      "loss": 0.0676,
      "step": 20990
    },
    {
      "epoch": 2.4561198221390126,
      "loss_main": 0.0844741240143776,
      "step": 20990
    },
    {
      "epoch": 2.4572899602153053,
      "grad_norm": 0.3096553385257721,
      "learning_rate": 7.734971491375671e-05,
      "loss": 0.0651,
      "step": 21000
    },
    {
      "epoch": 2.4572899602153053,
      "loss_main": 0.04029815271496773,
      "step": 21000
    },
    {
      "epoch": 2.4584600982915985,
      "grad_norm": 0.3115270137786865,
      "learning_rate": 7.732664109990485e-05,
      "loss": 0.0658,
      "step": 21010
    },
    {
      "epoch": 2.4584600982915985,
      "loss_main": 0.08490876853466034,
      "step": 21010
    },
    {
      "epoch": 2.4596302363678912,
      "grad_norm": 0.32411858439445496,
      "learning_rate": 7.730355898494095e-05,
      "loss": 0.0744,
      "step": 21020
    },
    {
      "epoch": 2.4596302363678912,
      "loss_main": 0.08642935752868652,
      "step": 21020
    },
    {
      "epoch": 2.4608003744441844,
      "grad_norm": 0.2588336169719696,
      "learning_rate": 7.728046857587673e-05,
      "loss": 0.0785,
      "step": 21030
    },
    {
      "epoch": 2.4608003744441844,
      "loss_main": 0.05089939758181572,
      "step": 21030
    },
    {
      "epoch": 2.4619705125204776,
      "grad_norm": 0.3426987826824188,
      "learning_rate": 7.725736987972647e-05,
      "loss": 0.0708,
      "step": 21040
    },
    {
      "epoch": 2.4619705125204776,
      "loss_main": 0.07485862821340561,
      "step": 21040
    },
    {
      "epoch": 2.4631406505967703,
      "grad_norm": 0.317832887172699,
      "learning_rate": 7.723426290350691e-05,
      "loss": 0.0792,
      "step": 21050
    },
    {
      "epoch": 2.4631406505967703,
      "loss_main": 0.14419297873973846,
      "step": 21050
    },
    {
      "epoch": 2.4643107886730635,
      "grad_norm": 0.23809948563575745,
      "learning_rate": 7.721114765423736e-05,
      "loss": 0.0798,
      "step": 21060
    },
    {
      "epoch": 2.4643107886730635,
      "loss_main": 0.07384148985147476,
      "step": 21060
    },
    {
      "epoch": 2.4654809267493563,
      "grad_norm": 0.28364071249961853,
      "learning_rate": 7.718802413893963e-05,
      "loss": 0.0646,
      "step": 21070
    },
    {
      "epoch": 2.4654809267493563,
      "loss_main": 0.07678637653589249,
      "step": 21070
    },
    {
      "epoch": 2.4666510648256494,
      "grad_norm": 0.25612884759902954,
      "learning_rate": 7.716489236463802e-05,
      "loss": 0.0714,
      "step": 21080
    },
    {
      "epoch": 2.4666510648256494,
      "loss_main": 0.06510098278522491,
      "step": 21080
    },
    {
      "epoch": 2.4678212029019426,
      "grad_norm": 0.34644731879234314,
      "learning_rate": 7.714175233835936e-05,
      "loss": 0.0683,
      "step": 21090
    },
    {
      "epoch": 2.4678212029019426,
      "loss_main": 0.05642935633659363,
      "step": 21090
    },
    {
      "epoch": 2.4689913409782354,
      "grad_norm": 0.3636082410812378,
      "learning_rate": 7.711860406713299e-05,
      "loss": 0.0671,
      "step": 21100
    },
    {
      "epoch": 2.4689913409782354,
      "loss_main": 0.0896291732788086,
      "step": 21100
    },
    {
      "epoch": 2.4701614790545285,
      "grad_norm": 0.2671927511692047,
      "learning_rate": 7.70954475579907e-05,
      "loss": 0.0676,
      "step": 21110
    },
    {
      "epoch": 2.4701614790545285,
      "loss_main": 0.055363979190588,
      "step": 21110
    },
    {
      "epoch": 2.4713316171308213,
      "grad_norm": 0.31496068835258484,
      "learning_rate": 7.707228281796688e-05,
      "loss": 0.0709,
      "step": 21120
    },
    {
      "epoch": 2.4713316171308213,
      "loss_main": 0.04920671135187149,
      "step": 21120
    },
    {
      "epoch": 2.4725017552071145,
      "grad_norm": 0.2742116153240204,
      "learning_rate": 7.704910985409833e-05,
      "loss": 0.0659,
      "step": 21130
    },
    {
      "epoch": 2.4725017552071145,
      "loss_main": 0.047764141112565994,
      "step": 21130
    },
    {
      "epoch": 2.4736718932834076,
      "grad_norm": 0.31412971019744873,
      "learning_rate": 7.702592867342439e-05,
      "loss": 0.0673,
      "step": 21140
    },
    {
      "epoch": 2.4736718932834076,
      "loss_main": 0.04345383867621422,
      "step": 21140
    },
    {
      "epoch": 2.4748420313597004,
      "grad_norm": 0.3153086304664612,
      "learning_rate": 7.700273928298691e-05,
      "loss": 0.0694,
      "step": 21150
    },
    {
      "epoch": 2.4748420313597004,
      "loss_main": 0.053667258471250534,
      "step": 21150
    },
    {
      "epoch": 2.4760121694359936,
      "grad_norm": 0.32229843735694885,
      "learning_rate": 7.697954168983021e-05,
      "loss": 0.0635,
      "step": 21160
    },
    {
      "epoch": 2.4760121694359936,
      "loss_main": 0.10551398247480392,
      "step": 21160
    },
    {
      "epoch": 2.4771823075122863,
      "grad_norm": 0.40532585978507996,
      "learning_rate": 7.695633590100109e-05,
      "loss": 0.0817,
      "step": 21170
    },
    {
      "epoch": 2.4771823075122863,
      "loss_main": 0.05606929212808609,
      "step": 21170
    },
    {
      "epoch": 2.4783524455885795,
      "grad_norm": 0.3178670406341553,
      "learning_rate": 7.693312192354886e-05,
      "loss": 0.0717,
      "step": 21180
    },
    {
      "epoch": 2.4783524455885795,
      "loss_main": 0.06386629492044449,
      "step": 21180
    },
    {
      "epoch": 2.479522583664872,
      "grad_norm": 0.2180577516555786,
      "learning_rate": 7.690989976452532e-05,
      "loss": 0.0681,
      "step": 21190
    },
    {
      "epoch": 2.479522583664872,
      "loss_main": 0.08093094825744629,
      "step": 21190
    },
    {
      "epoch": 2.4806927217411654,
      "grad_norm": 0.2549378275871277,
      "learning_rate": 7.688666943098475e-05,
      "loss": 0.0681,
      "step": 21200
    },
    {
      "epoch": 2.4806927217411654,
      "loss_main": 0.08671853691339493,
      "step": 21200
    },
    {
      "epoch": 2.4818628598174586,
      "grad_norm": 0.4120604693889618,
      "learning_rate": 7.686343092998389e-05,
      "loss": 0.0698,
      "step": 21210
    },
    {
      "epoch": 2.4818628598174586,
      "loss_main": 0.09313355386257172,
      "step": 21210
    },
    {
      "epoch": 2.4830329978937513,
      "grad_norm": 0.2666536271572113,
      "learning_rate": 7.684018426858202e-05,
      "loss": 0.0734,
      "step": 21220
    },
    {
      "epoch": 2.4830329978937513,
      "loss_main": 0.0472259558737278,
      "step": 21220
    },
    {
      "epoch": 2.4842031359700445,
      "grad_norm": 0.3309674561023712,
      "learning_rate": 7.681692945384084e-05,
      "loss": 0.0747,
      "step": 21230
    },
    {
      "epoch": 2.4842031359700445,
      "loss_main": 0.0623067282140255,
      "step": 21230
    },
    {
      "epoch": 2.4853732740463377,
      "grad_norm": 0.3472902178764343,
      "learning_rate": 7.679366649282456e-05,
      "loss": 0.0658,
      "step": 21240
    },
    {
      "epoch": 2.4853732740463377,
      "loss_main": 0.08914832025766373,
      "step": 21240
    },
    {
      "epoch": 2.4865434121226304,
      "grad_norm": 0.2540985345840454,
      "learning_rate": 7.677039539259983e-05,
      "loss": 0.0707,
      "step": 21250
    },
    {
      "epoch": 2.4865434121226304,
      "loss_main": 0.06938963383436203,
      "step": 21250
    },
    {
      "epoch": 2.4877135501989236,
      "grad_norm": 0.315728098154068,
      "learning_rate": 7.674711616023581e-05,
      "loss": 0.0728,
      "step": 21260
    },
    {
      "epoch": 2.4877135501989236,
      "loss_main": 0.07724200934171677,
      "step": 21260
    },
    {
      "epoch": 2.4888836882752163,
      "grad_norm": 0.3383145034313202,
      "learning_rate": 7.672382880280413e-05,
      "loss": 0.07,
      "step": 21270
    },
    {
      "epoch": 2.4888836882752163,
      "loss_main": 0.10191997140645981,
      "step": 21270
    },
    {
      "epoch": 2.4900538263515095,
      "grad_norm": 0.29590246081352234,
      "learning_rate": 7.670053332737885e-05,
      "loss": 0.0719,
      "step": 21280
    },
    {
      "epoch": 2.4900538263515095,
      "loss_main": 0.05451835319399834,
      "step": 21280
    },
    {
      "epoch": 2.4912239644278023,
      "grad_norm": 0.3803543150424957,
      "learning_rate": 7.667722974103654e-05,
      "loss": 0.066,
      "step": 21290
    },
    {
      "epoch": 2.4912239644278023,
      "loss_main": 0.021384326741099358,
      "step": 21290
    },
    {
      "epoch": 2.4923941025040954,
      "grad_norm": 0.3267247676849365,
      "learning_rate": 7.66539180508562e-05,
      "loss": 0.0614,
      "step": 21300
    },
    {
      "epoch": 2.4923941025040954,
      "loss_main": 0.052557315677404404,
      "step": 21300
    },
    {
      "epoch": 2.4935642405803886,
      "grad_norm": 0.3013059198856354,
      "learning_rate": 7.663059826391932e-05,
      "loss": 0.073,
      "step": 21310
    },
    {
      "epoch": 2.4935642405803886,
      "loss_main": 0.08082665503025055,
      "step": 21310
    },
    {
      "epoch": 2.4947343786566814,
      "grad_norm": 0.4113509953022003,
      "learning_rate": 7.660727038730981e-05,
      "loss": 0.0686,
      "step": 21320
    },
    {
      "epoch": 2.4947343786566814,
      "loss_main": 0.07942026108503342,
      "step": 21320
    },
    {
      "epoch": 2.4959045167329745,
      "grad_norm": 0.29987218976020813,
      "learning_rate": 7.65839344281141e-05,
      "loss": 0.0706,
      "step": 21330
    },
    {
      "epoch": 2.4959045167329745,
      "loss_main": 0.07844843715429306,
      "step": 21330
    },
    {
      "epoch": 2.4970746548092677,
      "grad_norm": 0.46324971318244934,
      "learning_rate": 7.656059039342101e-05,
      "loss": 0.0651,
      "step": 21340
    },
    {
      "epoch": 2.4970746548092677,
      "loss_main": 0.06714344769716263,
      "step": 21340
    },
    {
      "epoch": 2.4982447928855605,
      "grad_norm": 0.29787003993988037,
      "learning_rate": 7.653723829032187e-05,
      "loss": 0.0704,
      "step": 21350
    },
    {
      "epoch": 2.4982447928855605,
      "loss_main": 0.06495745480060577,
      "step": 21350
    },
    {
      "epoch": 2.4994149309618536,
      "grad_norm": 0.24418261647224426,
      "learning_rate": 7.65138781259104e-05,
      "loss": 0.0774,
      "step": 21360
    },
    {
      "epoch": 2.4994149309618536,
      "loss_main": 0.10423294454813004,
      "step": 21360
    },
    {
      "epoch": 2.5005850690381464,
      "grad_norm": 0.3631056547164917,
      "learning_rate": 7.649050990728279e-05,
      "loss": 0.0662,
      "step": 21370
    },
    {
      "epoch": 2.5005850690381464,
      "loss_main": 0.0642266497015953,
      "step": 21370
    },
    {
      "epoch": 2.5017552071144395,
      "grad_norm": 0.3292527496814728,
      "learning_rate": 7.646713364153774e-05,
      "loss": 0.0756,
      "step": 21380
    },
    {
      "epoch": 2.5017552071144395,
      "loss_main": 0.048406243324279785,
      "step": 21380
    },
    {
      "epoch": 2.5029253451907323,
      "grad_norm": 0.25468140840530396,
      "learning_rate": 7.64437493357763e-05,
      "loss": 0.0661,
      "step": 21390
    },
    {
      "epoch": 2.5029253451907323,
      "loss_main": 0.13430579006671906,
      "step": 21390
    },
    {
      "epoch": 2.5040954832670255,
      "grad_norm": 0.3802517056465149,
      "learning_rate": 7.642035699710202e-05,
      "loss": 0.079,
      "step": 21400
    },
    {
      "epoch": 2.5040954832670255,
      "loss_main": 0.049940016120672226,
      "step": 21400
    },
    {
      "epoch": 2.5052656213433186,
      "grad_norm": 0.28174901008605957,
      "learning_rate": 7.639695663262089e-05,
      "loss": 0.0691,
      "step": 21410
    },
    {
      "epoch": 2.5052656213433186,
      "loss_main": 0.10997884720563889,
      "step": 21410
    },
    {
      "epoch": 2.5064357594196114,
      "grad_norm": 0.2664678394794464,
      "learning_rate": 7.637354824944128e-05,
      "loss": 0.0853,
      "step": 21420
    },
    {
      "epoch": 2.5064357594196114,
      "loss_main": 0.09515666216611862,
      "step": 21420
    },
    {
      "epoch": 2.5076058974959046,
      "grad_norm": 0.16443687677383423,
      "learning_rate": 7.635013185467408e-05,
      "loss": 0.0644,
      "step": 21430
    },
    {
      "epoch": 2.5076058974959046,
      "loss_main": 0.06340418756008148,
      "step": 21430
    },
    {
      "epoch": 2.5087760355721977,
      "grad_norm": 0.2045264095067978,
      "learning_rate": 7.632670745543256e-05,
      "loss": 0.0696,
      "step": 21440
    },
    {
      "epoch": 2.5087760355721977,
      "loss_main": 0.07098889350891113,
      "step": 21440
    },
    {
      "epoch": 2.5099461736484905,
      "grad_norm": 0.2541070878505707,
      "learning_rate": 7.630327505883242e-05,
      "loss": 0.0666,
      "step": 21450
    },
    {
      "epoch": 2.5099461736484905,
      "loss_main": 0.040905650705099106,
      "step": 21450
    },
    {
      "epoch": 2.5111163117247837,
      "grad_norm": 0.29175692796707153,
      "learning_rate": 7.627983467199182e-05,
      "loss": 0.0645,
      "step": 21460
    },
    {
      "epoch": 2.5111163117247837,
      "loss_main": 0.046834684908390045,
      "step": 21460
    },
    {
      "epoch": 2.5122864498010764,
      "grad_norm": 0.2818804979324341,
      "learning_rate": 7.625638630203132e-05,
      "loss": 0.0724,
      "step": 21470
    },
    {
      "epoch": 2.5122864498010764,
      "loss_main": 0.08057039231061935,
      "step": 21470
    },
    {
      "epoch": 2.5134565878773696,
      "grad_norm": 0.23368683457374573,
      "learning_rate": 7.623292995607394e-05,
      "loss": 0.0681,
      "step": 21480
    },
    {
      "epoch": 2.5134565878773696,
      "loss_main": 0.05240604653954506,
      "step": 21480
    },
    {
      "epoch": 2.5146267259536623,
      "grad_norm": 0.2767779529094696,
      "learning_rate": 7.620946564124507e-05,
      "loss": 0.071,
      "step": 21490
    },
    {
      "epoch": 2.5146267259536623,
      "loss_main": 0.07624632865190506,
      "step": 21490
    },
    {
      "epoch": 2.5157968640299555,
      "grad_norm": 0.36873018741607666,
      "learning_rate": 7.618599336467256e-05,
      "loss": 0.0696,
      "step": 21500
    },
    {
      "epoch": 2.5157968640299555,
      "loss_main": 0.050182752311229706,
      "step": 21500
    },
    {
      "epoch": 2.5169670021062487,
      "grad_norm": 0.27579569816589355,
      "learning_rate": 7.616251313348666e-05,
      "loss": 0.0615,
      "step": 21510
    },
    {
      "epoch": 2.5169670021062487,
      "loss_main": 0.06235795468091965,
      "step": 21510
    },
    {
      "epoch": 2.5181371401825414,
      "grad_norm": 0.22328902781009674,
      "learning_rate": 7.613902495482005e-05,
      "loss": 0.0688,
      "step": 21520
    },
    {
      "epoch": 2.5181371401825414,
      "loss_main": 0.06044906750321388,
      "step": 21520
    },
    {
      "epoch": 2.5193072782588346,
      "grad_norm": 0.24410896003246307,
      "learning_rate": 7.611552883580784e-05,
      "loss": 0.0686,
      "step": 21530
    },
    {
      "epoch": 2.5193072782588346,
      "loss_main": 0.08617452532052994,
      "step": 21530
    },
    {
      "epoch": 2.520477416335128,
      "grad_norm": 0.22740411758422852,
      "learning_rate": 7.609202478358748e-05,
      "loss": 0.0663,
      "step": 21540
    },
    {
      "epoch": 2.520477416335128,
      "loss_main": 0.07319362461566925,
      "step": 21540
    },
    {
      "epoch": 2.5216475544114205,
      "grad_norm": 0.29100409150123596,
      "learning_rate": 7.606851280529895e-05,
      "loss": 0.0711,
      "step": 21550
    },
    {
      "epoch": 2.5216475544114205,
      "loss_main": 0.0669424906373024,
      "step": 21550
    },
    {
      "epoch": 2.5228176924877137,
      "grad_norm": 0.3806569576263428,
      "learning_rate": 7.604499290808449e-05,
      "loss": 0.0658,
      "step": 21560
    },
    {
      "epoch": 2.5228176924877137,
      "loss_main": 0.10703577101230621,
      "step": 21560
    },
    {
      "epoch": 2.5239878305640064,
      "grad_norm": 0.3748057782649994,
      "learning_rate": 7.602146509908888e-05,
      "loss": 0.0684,
      "step": 21570
    },
    {
      "epoch": 2.5239878305640064,
      "loss_main": 0.04661249369382858,
      "step": 21570
    },
    {
      "epoch": 2.5251579686402996,
      "grad_norm": 0.24713945388793945,
      "learning_rate": 7.599792938545921e-05,
      "loss": 0.0638,
      "step": 21580
    },
    {
      "epoch": 2.5251579686402996,
      "loss_main": 0.06269989162683487,
      "step": 21580
    },
    {
      "epoch": 2.5263281067165924,
      "grad_norm": 0.26881664991378784,
      "learning_rate": 7.597438577434506e-05,
      "loss": 0.0687,
      "step": 21590
    },
    {
      "epoch": 2.5263281067165924,
      "loss_main": 0.044777292758226395,
      "step": 21590
    },
    {
      "epoch": 2.5274982447928855,
      "grad_norm": 0.3171372413635254,
      "learning_rate": 7.595083427289831e-05,
      "loss": 0.0634,
      "step": 21600
    },
    {
      "epoch": 2.5274982447928855,
      "loss_main": 0.0469992570579052,
      "step": 21600
    },
    {
      "epoch": 2.5286683828691787,
      "grad_norm": 0.32132676243782043,
      "learning_rate": 7.59272748882733e-05,
      "loss": 0.0608,
      "step": 21610
    },
    {
      "epoch": 2.5286683828691787,
      "loss_main": 0.07440035790205002,
      "step": 21610
    },
    {
      "epoch": 2.5298385209454715,
      "grad_norm": 0.2938103973865509,
      "learning_rate": 7.590370762762675e-05,
      "loss": 0.0718,
      "step": 21620
    },
    {
      "epoch": 2.5298385209454715,
      "loss_main": 0.049563970416784286,
      "step": 21620
    },
    {
      "epoch": 2.5310086590217646,
      "grad_norm": 0.38464006781578064,
      "learning_rate": 7.588013249811777e-05,
      "loss": 0.0734,
      "step": 21630
    },
    {
      "epoch": 2.5310086590217646,
      "loss_main": 0.0727168396115303,
      "step": 21630
    },
    {
      "epoch": 2.532178797098058,
      "grad_norm": 0.28501096367836,
      "learning_rate": 7.585654950690786e-05,
      "loss": 0.0697,
      "step": 21640
    },
    {
      "epoch": 2.532178797098058,
      "loss_main": 0.0435931421816349,
      "step": 21640
    },
    {
      "epoch": 2.5333489351743506,
      "grad_norm": 0.27513745427131653,
      "learning_rate": 7.583295866116091e-05,
      "loss": 0.0742,
      "step": 21650
    },
    {
      "epoch": 2.5333489351743506,
      "loss_main": 0.07563800364732742,
      "step": 21650
    },
    {
      "epoch": 2.5345190732506437,
      "grad_norm": 0.32393988966941833,
      "learning_rate": 7.580935996804321e-05,
      "loss": 0.0758,
      "step": 21660
    },
    {
      "epoch": 2.5345190732506437,
      "loss_main": 0.07988549023866653,
      "step": 21660
    },
    {
      "epoch": 2.5356892113269365,
      "grad_norm": 0.26020607352256775,
      "learning_rate": 7.57857534347234e-05,
      "loss": 0.0717,
      "step": 21670
    },
    {
      "epoch": 2.5356892113269365,
      "loss_main": 0.04772242531180382,
      "step": 21670
    },
    {
      "epoch": 2.5368593494032297,
      "grad_norm": 0.3422619104385376,
      "learning_rate": 7.576213906837254e-05,
      "loss": 0.0675,
      "step": 21680
    },
    {
      "epoch": 2.5368593494032297,
      "loss_main": 0.11653552204370499,
      "step": 21680
    },
    {
      "epoch": 2.5380294874795224,
      "grad_norm": 0.3521164357662201,
      "learning_rate": 7.573851687616403e-05,
      "loss": 0.0702,
      "step": 21690
    },
    {
      "epoch": 2.5380294874795224,
      "loss_main": 0.08397793769836426,
      "step": 21690
    },
    {
      "epoch": 2.5391996255558156,
      "grad_norm": 0.24179132282733917,
      "learning_rate": 7.571488686527368e-05,
      "loss": 0.0664,
      "step": 21700
    },
    {
      "epoch": 2.5391996255558156,
      "loss_main": 0.03104843571782112,
      "step": 21700
    },
    {
      "epoch": 2.5403697636321088,
      "grad_norm": 0.3558522164821625,
      "learning_rate": 7.569124904287968e-05,
      "loss": 0.0606,
      "step": 21710
    },
    {
      "epoch": 2.5403697636321088,
      "loss_main": 0.09477351605892181,
      "step": 21710
    },
    {
      "epoch": 2.5415399017084015,
      "grad_norm": 0.3656998574733734,
      "learning_rate": 7.566760341616254e-05,
      "loss": 0.0724,
      "step": 21720
    },
    {
      "epoch": 2.5415399017084015,
      "loss_main": 0.07122555375099182,
      "step": 21720
    },
    {
      "epoch": 2.5427100397846947,
      "grad_norm": 0.2746495306491852,
      "learning_rate": 7.564394999230519e-05,
      "loss": 0.0748,
      "step": 21730
    },
    {
      "epoch": 2.5427100397846947,
      "loss_main": 0.0665384903550148,
      "step": 21730
    },
    {
      "epoch": 2.543880177860988,
      "grad_norm": 0.3637998402118683,
      "learning_rate": 7.562028877849294e-05,
      "loss": 0.0616,
      "step": 21740
    },
    {
      "epoch": 2.543880177860988,
      "loss_main": 0.07159265130758286,
      "step": 21740
    },
    {
      "epoch": 2.5450503159372806,
      "grad_norm": 0.40353623032569885,
      "learning_rate": 7.559661978191341e-05,
      "loss": 0.0795,
      "step": 21750
    },
    {
      "epoch": 2.5450503159372806,
      "loss_main": 0.07779278606176376,
      "step": 21750
    },
    {
      "epoch": 2.5462204540135733,
      "grad_norm": 0.3949243128299713,
      "learning_rate": 7.557294300975664e-05,
      "loss": 0.0679,
      "step": 21760
    },
    {
      "epoch": 2.5462204540135733,
      "loss_main": 0.057537443935871124,
      "step": 21760
    },
    {
      "epoch": 2.5473905920898665,
      "grad_norm": 0.3008716404438019,
      "learning_rate": 7.554925846921499e-05,
      "loss": 0.0652,
      "step": 21770
    },
    {
      "epoch": 2.5473905920898665,
      "loss_main": 0.029708724468946457,
      "step": 21770
    },
    {
      "epoch": 2.5485607301661597,
      "grad_norm": 0.2456457018852234,
      "learning_rate": 7.552556616748321e-05,
      "loss": 0.07,
      "step": 21780
    },
    {
      "epoch": 2.5485607301661597,
      "loss_main": 0.08046437799930573,
      "step": 21780
    },
    {
      "epoch": 2.5497308682424524,
      "grad_norm": 0.3089817762374878,
      "learning_rate": 7.550186611175838e-05,
      "loss": 0.0745,
      "step": 21790
    },
    {
      "epoch": 2.5497308682424524,
      "loss_main": 0.039162155240774155,
      "step": 21790
    },
    {
      "epoch": 2.5509010063187456,
      "grad_norm": 0.5068421959877014,
      "learning_rate": 7.547815830923998e-05,
      "loss": 0.0611,
      "step": 21800
    },
    {
      "epoch": 2.5509010063187456,
      "loss_main": 0.05817064270377159,
      "step": 21800
    },
    {
      "epoch": 2.552071144395039,
      "grad_norm": 0.4544600248336792,
      "learning_rate": 7.54544427671298e-05,
      "loss": 0.0697,
      "step": 21810
    },
    {
      "epoch": 2.552071144395039,
      "loss_main": 0.08378060907125473,
      "step": 21810
    },
    {
      "epoch": 2.5532412824713315,
      "grad_norm": 0.3325319290161133,
      "learning_rate": 7.543071949263198e-05,
      "loss": 0.0672,
      "step": 21820
    },
    {
      "epoch": 2.5532412824713315,
      "loss_main": 0.05167699232697487,
      "step": 21820
    },
    {
      "epoch": 2.5544114205476247,
      "grad_norm": 0.285578191280365,
      "learning_rate": 7.540698849295305e-05,
      "loss": 0.061,
      "step": 21830
    },
    {
      "epoch": 2.5544114205476247,
      "loss_main": 0.07590444386005402,
      "step": 21830
    },
    {
      "epoch": 2.555581558623918,
      "grad_norm": 0.2984165847301483,
      "learning_rate": 7.538324977530183e-05,
      "loss": 0.0646,
      "step": 21840
    },
    {
      "epoch": 2.555581558623918,
      "loss_main": 0.09946626424789429,
      "step": 21840
    },
    {
      "epoch": 2.5567516967002106,
      "grad_norm": 0.29671111702919006,
      "learning_rate": 7.535950334688955e-05,
      "loss": 0.0743,
      "step": 21850
    },
    {
      "epoch": 2.5567516967002106,
      "loss_main": 0.03768907114863396,
      "step": 21850
    },
    {
      "epoch": 2.5579218347765034,
      "grad_norm": 0.2489924281835556,
      "learning_rate": 7.533574921492972e-05,
      "loss": 0.0651,
      "step": 21860
    },
    {
      "epoch": 2.5579218347765034,
      "loss_main": 0.0424477681517601,
      "step": 21860
    },
    {
      "epoch": 2.5590919728527965,
      "grad_norm": 0.34608790278434753,
      "learning_rate": 7.531198738663824e-05,
      "loss": 0.0627,
      "step": 21870
    },
    {
      "epoch": 2.5590919728527965,
      "loss_main": 0.07983855158090591,
      "step": 21870
    },
    {
      "epoch": 2.5602621109290897,
      "grad_norm": 0.40902048349380493,
      "learning_rate": 7.528821786923333e-05,
      "loss": 0.0683,
      "step": 21880
    },
    {
      "epoch": 2.5602621109290897,
      "loss_main": 0.06813199073076248,
      "step": 21880
    },
    {
      "epoch": 2.5614322490053825,
      "grad_norm": 0.2421225905418396,
      "learning_rate": 7.52644406699355e-05,
      "loss": 0.0727,
      "step": 21890
    },
    {
      "epoch": 2.5614322490053825,
      "loss_main": 0.08132294565439224,
      "step": 21890
    },
    {
      "epoch": 2.5626023870816756,
      "grad_norm": 0.21594442427158356,
      "learning_rate": 7.524065579596766e-05,
      "loss": 0.0639,
      "step": 21900
    },
    {
      "epoch": 2.5626023870816756,
      "loss_main": 0.06389413774013519,
      "step": 21900
    },
    {
      "epoch": 2.563772525157969,
      "grad_norm": 0.20081423223018646,
      "learning_rate": 7.521686325455506e-05,
      "loss": 0.0585,
      "step": 21910
    },
    {
      "epoch": 2.563772525157969,
      "loss_main": 0.07058318704366684,
      "step": 21910
    },
    {
      "epoch": 2.5649426632342616,
      "grad_norm": 0.2228338122367859,
      "learning_rate": 7.51930630529252e-05,
      "loss": 0.0684,
      "step": 21920
    },
    {
      "epoch": 2.5649426632342616,
      "loss_main": 0.05459858849644661,
      "step": 21920
    },
    {
      "epoch": 2.5661128013105547,
      "grad_norm": 0.2790789008140564,
      "learning_rate": 7.516925519830797e-05,
      "loss": 0.0656,
      "step": 21930
    },
    {
      "epoch": 2.5661128013105547,
      "loss_main": 0.07306719571352005,
      "step": 21930
    },
    {
      "epoch": 2.5672829393868475,
      "grad_norm": 0.2946664094924927,
      "learning_rate": 7.514543969793557e-05,
      "loss": 0.0674,
      "step": 21940
    },
    {
      "epoch": 2.5672829393868475,
      "loss_main": 0.06656868010759354,
      "step": 21940
    },
    {
      "epoch": 2.5684530774631407,
      "grad_norm": 0.3480403423309326,
      "learning_rate": 7.512161655904251e-05,
      "loss": 0.0697,
      "step": 21950
    },
    {
      "epoch": 2.5684530774631407,
      "loss_main": 0.08656545728445053,
      "step": 21950
    },
    {
      "epoch": 2.5696232155394334,
      "grad_norm": 0.22364670038223267,
      "learning_rate": 7.509778578886563e-05,
      "loss": 0.063,
      "step": 21960
    },
    {
      "epoch": 2.5696232155394334,
      "loss_main": 0.043272633105516434,
      "step": 21960
    },
    {
      "epoch": 2.5707933536157266,
      "grad_norm": 0.21330058574676514,
      "learning_rate": 7.507394739464412e-05,
      "loss": 0.0728,
      "step": 21970
    },
    {
      "epoch": 2.5707933536157266,
      "loss_main": 0.045583855360746384,
      "step": 21970
    },
    {
      "epoch": 2.5719634916920198,
      "grad_norm": 0.3098517060279846,
      "learning_rate": 7.50501013836194e-05,
      "loss": 0.0737,
      "step": 21980
    },
    {
      "epoch": 2.5719634916920198,
      "loss_main": 0.07362707704305649,
      "step": 21980
    },
    {
      "epoch": 2.5731336297683125,
      "grad_norm": 0.2521144449710846,
      "learning_rate": 7.50262477630353e-05,
      "loss": 0.0827,
      "step": 21990
    },
    {
      "epoch": 2.5731336297683125,
      "loss_main": 0.11968664824962616,
      "step": 21990
    },
    {
      "epoch": 2.5743037678446057,
      "grad_norm": 0.23702403903007507,
      "learning_rate": 7.500238654013794e-05,
      "loss": 0.0797,
      "step": 22000
    },
    {
      "epoch": 2.5743037678446057,
      "loss_main": 0.041041646152734756,
      "step": 22000
    },
    {
      "epoch": 2.575473905920899,
      "grad_norm": 0.2847360670566559,
      "learning_rate": 7.497851772217566e-05,
      "loss": 0.0672,
      "step": 22010
    },
    {
      "epoch": 2.575473905920899,
      "loss_main": 0.07888894528150558,
      "step": 22010
    },
    {
      "epoch": 2.5766440439971916,
      "grad_norm": 0.29752156138420105,
      "learning_rate": 7.495464131639924e-05,
      "loss": 0.0665,
      "step": 22020
    },
    {
      "epoch": 2.5766440439971916,
      "loss_main": 0.05355948209762573,
      "step": 22020
    },
    {
      "epoch": 2.577814182073485,
      "grad_norm": 0.30537304282188416,
      "learning_rate": 7.493075733006166e-05,
      "loss": 0.0652,
      "step": 22030
    },
    {
      "epoch": 2.577814182073485,
      "loss_main": 0.036775462329387665,
      "step": 22030
    },
    {
      "epoch": 2.5789843201497775,
      "grad_norm": 0.3693614602088928,
      "learning_rate": 7.490686577041828e-05,
      "loss": 0.0699,
      "step": 22040
    },
    {
      "epoch": 2.5789843201497775,
      "loss_main": 0.09424548596143723,
      "step": 22040
    },
    {
      "epoch": 2.5801544582260707,
      "grad_norm": 0.3297884166240692,
      "learning_rate": 7.488296664472668e-05,
      "loss": 0.086,
      "step": 22050
    },
    {
      "epoch": 2.5801544582260707,
      "loss_main": 0.06515610963106155,
      "step": 22050
    },
    {
      "epoch": 2.5813245963023634,
      "grad_norm": 0.19107873737812042,
      "learning_rate": 7.485905996024682e-05,
      "loss": 0.0685,
      "step": 22060
    },
    {
      "epoch": 2.5813245963023634,
      "loss_main": 0.0883573368191719,
      "step": 22060
    },
    {
      "epoch": 2.5824947343786566,
      "grad_norm": 0.2545521557331085,
      "learning_rate": 7.483514572424093e-05,
      "loss": 0.0636,
      "step": 22070
    },
    {
      "epoch": 2.5824947343786566,
      "loss_main": 0.07092030346393585,
      "step": 22070
    },
    {
      "epoch": 2.58366487245495,
      "grad_norm": 0.21774190664291382,
      "learning_rate": 7.481122394397349e-05,
      "loss": 0.0649,
      "step": 22080
    },
    {
      "epoch": 2.58366487245495,
      "loss_main": 0.0709458515048027,
      "step": 22080
    },
    {
      "epoch": 2.5848350105312425,
      "grad_norm": 0.23219291865825653,
      "learning_rate": 7.478729462671131e-05,
      "loss": 0.0592,
      "step": 22090
    },
    {
      "epoch": 2.5848350105312425,
      "loss_main": 0.08605468273162842,
      "step": 22090
    },
    {
      "epoch": 2.5860051486075357,
      "grad_norm": 0.3528391718864441,
      "learning_rate": 7.47633577797235e-05,
      "loss": 0.0824,
      "step": 22100
    },
    {
      "epoch": 2.5860051486075357,
      "loss_main": 0.035752493888139725,
      "step": 22100
    },
    {
      "epoch": 2.587175286683829,
      "grad_norm": 0.3037314713001251,
      "learning_rate": 7.473941341028144e-05,
      "loss": 0.0666,
      "step": 22110
    },
    {
      "epoch": 2.587175286683829,
      "loss_main": 0.05768684297800064,
      "step": 22110
    },
    {
      "epoch": 2.5883454247601216,
      "grad_norm": 0.30095934867858887,
      "learning_rate": 7.471546152565879e-05,
      "loss": 0.0598,
      "step": 22120
    },
    {
      "epoch": 2.5883454247601216,
      "loss_main": 0.056900352239608765,
      "step": 22120
    },
    {
      "epoch": 2.589515562836415,
      "grad_norm": 0.2316785454750061,
      "learning_rate": 7.46915021331315e-05,
      "loss": 0.0731,
      "step": 22130
    },
    {
      "epoch": 2.589515562836415,
      "loss_main": 0.07797294110059738,
      "step": 22130
    },
    {
      "epoch": 2.5906857009127076,
      "grad_norm": 0.28355100750923157,
      "learning_rate": 7.466753523997778e-05,
      "loss": 0.0686,
      "step": 22140
    },
    {
      "epoch": 2.5906857009127076,
      "loss_main": 0.051619332283735275,
      "step": 22140
    },
    {
      "epoch": 2.5918558389890007,
      "grad_norm": 0.30834200978279114,
      "learning_rate": 7.464356085347819e-05,
      "loss": 0.0713,
      "step": 22150
    },
    {
      "epoch": 2.5918558389890007,
      "loss_main": 0.06761840730905533,
      "step": 22150
    },
    {
      "epoch": 2.5930259770652935,
      "grad_norm": 0.36090123653411865,
      "learning_rate": 7.461957898091548e-05,
      "loss": 0.0704,
      "step": 22160
    },
    {
      "epoch": 2.5930259770652935,
      "loss_main": 0.08949536830186844,
      "step": 22160
    },
    {
      "epoch": 2.5941961151415867,
      "grad_norm": 0.27994152903556824,
      "learning_rate": 7.459558962957473e-05,
      "loss": 0.0756,
      "step": 22170
    },
    {
      "epoch": 2.5941961151415867,
      "loss_main": 0.06009938567876816,
      "step": 22170
    },
    {
      "epoch": 2.59536625321788,
      "grad_norm": 0.26840466260910034,
      "learning_rate": 7.457159280674326e-05,
      "loss": 0.0749,
      "step": 22180
    },
    {
      "epoch": 2.59536625321788,
      "loss_main": 0.06589040905237198,
      "step": 22180
    },
    {
      "epoch": 2.5965363912941726,
      "grad_norm": 0.270915687084198,
      "learning_rate": 7.454758851971066e-05,
      "loss": 0.0743,
      "step": 22190
    },
    {
      "epoch": 2.5965363912941726,
      "loss_main": 0.10370521992444992,
      "step": 22190
    },
    {
      "epoch": 2.5977065293704658,
      "grad_norm": 0.3798902928829193,
      "learning_rate": 7.45235767757688e-05,
      "loss": 0.0709,
      "step": 22200
    },
    {
      "epoch": 2.5977065293704658,
      "loss_main": 0.12519873678684235,
      "step": 22200
    },
    {
      "epoch": 2.598876667446759,
      "grad_norm": 0.2519160211086273,
      "learning_rate": 7.449955758221183e-05,
      "loss": 0.0858,
      "step": 22210
    },
    {
      "epoch": 2.598876667446759,
      "loss_main": 0.1195635199546814,
      "step": 22210
    },
    {
      "epoch": 2.6000468055230517,
      "grad_norm": 0.32858389616012573,
      "learning_rate": 7.447553094633615e-05,
      "loss": 0.0731,
      "step": 22220
    },
    {
      "epoch": 2.6000468055230517,
      "loss_main": 0.11671806126832962,
      "step": 22220
    },
    {
      "epoch": 2.601216943599345,
      "grad_norm": 0.22905969619750977,
      "learning_rate": 7.445149687544039e-05,
      "loss": 0.0647,
      "step": 22230
    },
    {
      "epoch": 2.601216943599345,
      "loss_main": 0.08473597466945648,
      "step": 22230
    },
    {
      "epoch": 2.6023870816756376,
      "grad_norm": 0.29033854603767395,
      "learning_rate": 7.44274553768255e-05,
      "loss": 0.0693,
      "step": 22240
    },
    {
      "epoch": 2.6023870816756376,
      "loss_main": 0.07527758181095123,
      "step": 22240
    },
    {
      "epoch": 2.6035572197519308,
      "grad_norm": 0.22326649725437164,
      "learning_rate": 7.440340645779464e-05,
      "loss": 0.0633,
      "step": 22250
    },
    {
      "epoch": 2.6035572197519308,
      "loss_main": 0.05825253948569298,
      "step": 22250
    },
    {
      "epoch": 2.6047273578282235,
      "grad_norm": 0.2933562695980072,
      "learning_rate": 7.437935012565322e-05,
      "loss": 0.0635,
      "step": 22260
    },
    {
      "epoch": 2.6047273578282235,
      "loss_main": 0.04187759384512901,
      "step": 22260
    },
    {
      "epoch": 2.6058974959045167,
      "grad_norm": 0.23417426645755768,
      "learning_rate": 7.435528638770893e-05,
      "loss": 0.0661,
      "step": 22270
    },
    {
      "epoch": 2.6058974959045167,
      "loss_main": 0.056593555957078934,
      "step": 22270
    },
    {
      "epoch": 2.60706763398081,
      "grad_norm": 0.3274671733379364,
      "learning_rate": 7.433121525127171e-05,
      "loss": 0.0714,
      "step": 22280
    },
    {
      "epoch": 2.60706763398081,
      "loss_main": 0.08617193251848221,
      "step": 22280
    },
    {
      "epoch": 2.6082377720571026,
      "grad_norm": 0.3636036515235901,
      "learning_rate": 7.430713672365371e-05,
      "loss": 0.0932,
      "step": 22290
    },
    {
      "epoch": 2.6082377720571026,
      "loss_main": 0.07230905443429947,
      "step": 22290
    },
    {
      "epoch": 2.609407910133396,
      "grad_norm": 0.2783193588256836,
      "learning_rate": 7.428305081216938e-05,
      "loss": 0.0694,
      "step": 22300
    },
    {
      "epoch": 2.609407910133396,
      "loss_main": 0.05423707142472267,
      "step": 22300
    },
    {
      "epoch": 2.610578048209689,
      "grad_norm": 0.27406007051467896,
      "learning_rate": 7.425895752413536e-05,
      "loss": 0.0703,
      "step": 22310
    },
    {
      "epoch": 2.610578048209689,
      "loss_main": 0.08542037010192871,
      "step": 22310
    },
    {
      "epoch": 2.6117481862859817,
      "grad_norm": 0.3155767619609833,
      "learning_rate": 7.423485686687057e-05,
      "loss": 0.0752,
      "step": 22320
    },
    {
      "epoch": 2.6117481862859817,
      "loss_main": 0.08020919561386108,
      "step": 22320
    },
    {
      "epoch": 2.612918324362275,
      "grad_norm": 0.34570983052253723,
      "learning_rate": 7.421074884769616e-05,
      "loss": 0.0783,
      "step": 22330
    },
    {
      "epoch": 2.612918324362275,
      "loss_main": 0.10233701765537262,
      "step": 22330
    },
    {
      "epoch": 2.6140884624385676,
      "grad_norm": 0.24491672217845917,
      "learning_rate": 7.418663347393548e-05,
      "loss": 0.0696,
      "step": 22340
    },
    {
      "epoch": 2.6140884624385676,
      "loss_main": 0.08088982850313187,
      "step": 22340
    },
    {
      "epoch": 2.615258600514861,
      "grad_norm": 0.2864183187484741,
      "learning_rate": 7.416251075291418e-05,
      "loss": 0.0715,
      "step": 22350
    },
    {
      "epoch": 2.615258600514861,
      "loss_main": 0.057463936507701874,
      "step": 22350
    },
    {
      "epoch": 2.6164287385911535,
      "grad_norm": 0.31565046310424805,
      "learning_rate": 7.413838069196007e-05,
      "loss": 0.074,
      "step": 22360
    },
    {
      "epoch": 2.6164287385911535,
      "loss_main": 0.06482602655887604,
      "step": 22360
    },
    {
      "epoch": 2.6175988766674467,
      "grad_norm": 0.24184714257717133,
      "learning_rate": 7.411424329840324e-05,
      "loss": 0.069,
      "step": 22370
    },
    {
      "epoch": 2.6175988766674467,
      "loss_main": 0.08273618668317795,
      "step": 22370
    },
    {
      "epoch": 2.61876901474374,
      "grad_norm": 0.32597804069519043,
      "learning_rate": 7.409009857957601e-05,
      "loss": 0.0805,
      "step": 22380
    },
    {
      "epoch": 2.61876901474374,
      "loss_main": 0.09099296480417252,
      "step": 22380
    },
    {
      "epoch": 2.6199391528200326,
      "grad_norm": 0.3488982617855072,
      "learning_rate": 7.40659465428129e-05,
      "loss": 0.0721,
      "step": 22390
    },
    {
      "epoch": 2.6199391528200326,
      "loss_main": 0.08074861764907837,
      "step": 22390
    },
    {
      "epoch": 2.621109290896326,
      "grad_norm": 0.2697601020336151,
      "learning_rate": 7.404178719545063e-05,
      "loss": 0.0755,
      "step": 22400
    },
    {
      "epoch": 2.621109290896326,
      "loss_main": 0.07332796603441238,
      "step": 22400
    },
    {
      "epoch": 2.622279428972619,
      "grad_norm": 0.33080437779426575,
      "learning_rate": 7.401762054482822e-05,
      "loss": 0.0806,
      "step": 22410
    },
    {
      "epoch": 2.622279428972619,
      "loss_main": 0.06154799461364746,
      "step": 22410
    },
    {
      "epoch": 2.6234495670489117,
      "grad_norm": 0.3889002799987793,
      "learning_rate": 7.39934465982868e-05,
      "loss": 0.0786,
      "step": 22420
    },
    {
      "epoch": 2.6234495670489117,
      "loss_main": 0.09015752375125885,
      "step": 22420
    },
    {
      "epoch": 2.624619705125205,
      "grad_norm": 0.2594912350177765,
      "learning_rate": 7.396926536316984e-05,
      "loss": 0.0697,
      "step": 22430
    },
    {
      "epoch": 2.624619705125205,
      "loss_main": 0.07460461556911469,
      "step": 22430
    },
    {
      "epoch": 2.6257898432014977,
      "grad_norm": 0.2943812310695648,
      "learning_rate": 7.394507684682293e-05,
      "loss": 0.0718,
      "step": 22440
    },
    {
      "epoch": 2.6257898432014977,
      "loss_main": 0.052023377269506454,
      "step": 22440
    },
    {
      "epoch": 2.626959981277791,
      "grad_norm": 0.3099926710128784,
      "learning_rate": 7.392088105659393e-05,
      "loss": 0.0719,
      "step": 22450
    },
    {
      "epoch": 2.626959981277791,
      "loss_main": 0.06344673782587051,
      "step": 22450
    },
    {
      "epoch": 2.6281301193540836,
      "grad_norm": 0.26791590452194214,
      "learning_rate": 7.389667799983284e-05,
      "loss": 0.0748,
      "step": 22460
    },
    {
      "epoch": 2.6281301193540836,
      "loss_main": 0.05936086177825928,
      "step": 22460
    },
    {
      "epoch": 2.6293002574303768,
      "grad_norm": 0.3007056713104248,
      "learning_rate": 7.387246768389193e-05,
      "loss": 0.068,
      "step": 22470
    },
    {
      "epoch": 2.6293002574303768,
      "loss_main": 0.10813968628644943,
      "step": 22470
    },
    {
      "epoch": 2.63047039550667,
      "grad_norm": 0.32366329431533813,
      "learning_rate": 7.384825011612563e-05,
      "loss": 0.0695,
      "step": 22480
    },
    {
      "epoch": 2.63047039550667,
      "loss_main": 0.13213814795017242,
      "step": 22480
    },
    {
      "epoch": 2.6316405335829627,
      "grad_norm": 0.3212079703807831,
      "learning_rate": 7.382402530389066e-05,
      "loss": 0.0926,
      "step": 22490
    },
    {
      "epoch": 2.6316405335829627,
      "loss_main": 0.08597081154584885,
      "step": 22490
    },
    {
      "epoch": 2.632810671659256,
      "grad_norm": 0.29670870304107666,
      "learning_rate": 7.379979325454582e-05,
      "loss": 0.0738,
      "step": 22500
    },
    {
      "epoch": 2.632810671659256,
      "loss_main": 0.08863259851932526,
      "step": 22500
    },
    {
      "epoch": 2.633980809735549,
      "grad_norm": 0.3581444025039673,
      "learning_rate": 7.37755539754522e-05,
      "loss": 0.07,
      "step": 22510
    },
    {
      "epoch": 2.633980809735549,
      "loss_main": 0.05848727375268936,
      "step": 22510
    },
    {
      "epoch": 2.635150947811842,
      "grad_norm": 0.4464893937110901,
      "learning_rate": 7.375130747397302e-05,
      "loss": 0.0655,
      "step": 22520
    },
    {
      "epoch": 2.635150947811842,
      "loss_main": 0.07396858930587769,
      "step": 22520
    },
    {
      "epoch": 2.636321085888135,
      "grad_norm": 0.39677441120147705,
      "learning_rate": 7.372705375747377e-05,
      "loss": 0.0665,
      "step": 22530
    },
    {
      "epoch": 2.636321085888135,
      "loss_main": 0.0472988560795784,
      "step": 22530
    },
    {
      "epoch": 2.6374912239644277,
      "grad_norm": 0.37743595242500305,
      "learning_rate": 7.370279283332205e-05,
      "loss": 0.0661,
      "step": 22540
    },
    {
      "epoch": 2.6374912239644277,
      "loss_main": 0.07935317605733871,
      "step": 22540
    },
    {
      "epoch": 2.638661362040721,
      "grad_norm": 0.28010639548301697,
      "learning_rate": 7.36785247088877e-05,
      "loss": 0.0701,
      "step": 22550
    },
    {
      "epoch": 2.638661362040721,
      "loss_main": 0.07244366407394409,
      "step": 22550
    },
    {
      "epoch": 2.6398315001170136,
      "grad_norm": 0.3837538957595825,
      "learning_rate": 7.365424939154275e-05,
      "loss": 0.0717,
      "step": 22560
    },
    {
      "epoch": 2.6398315001170136,
      "loss_main": 0.06964947283267975,
      "step": 22560
    },
    {
      "epoch": 2.641001638193307,
      "grad_norm": 0.32725489139556885,
      "learning_rate": 7.362996688866138e-05,
      "loss": 0.0713,
      "step": 22570
    },
    {
      "epoch": 2.641001638193307,
      "loss_main": 0.08179774135351181,
      "step": 22570
    },
    {
      "epoch": 2.6421717762696,
      "grad_norm": 0.24307739734649658,
      "learning_rate": 7.360567720761999e-05,
      "loss": 0.0738,
      "step": 22580
    },
    {
      "epoch": 2.6421717762696,
      "loss_main": 0.07811156660318375,
      "step": 22580
    },
    {
      "epoch": 2.6433419143458927,
      "grad_norm": 0.2775734066963196,
      "learning_rate": 7.358138035579711e-05,
      "loss": 0.0688,
      "step": 22590
    },
    {
      "epoch": 2.6433419143458927,
      "loss_main": 0.045841243118047714,
      "step": 22590
    },
    {
      "epoch": 2.644512052422186,
      "grad_norm": 0.3040919005870819,
      "learning_rate": 7.355707634057354e-05,
      "loss": 0.0635,
      "step": 22600
    },
    {
      "epoch": 2.644512052422186,
      "loss_main": 0.051229365170001984,
      "step": 22600
    },
    {
      "epoch": 2.645682190498479,
      "grad_norm": 0.41995686292648315,
      "learning_rate": 7.353276516933215e-05,
      "loss": 0.0641,
      "step": 22610
    },
    {
      "epoch": 2.645682190498479,
      "loss_main": 0.08106695115566254,
      "step": 22610
    },
    {
      "epoch": 2.646852328574772,
      "grad_norm": 0.271867960691452,
      "learning_rate": 7.350844684945806e-05,
      "loss": 0.0704,
      "step": 22620
    },
    {
      "epoch": 2.646852328574772,
      "loss_main": 0.04313512519001961,
      "step": 22620
    },
    {
      "epoch": 2.6480224666510646,
      "grad_norm": 0.30306148529052734,
      "learning_rate": 7.348412138833851e-05,
      "loss": 0.0682,
      "step": 22630
    },
    {
      "epoch": 2.6480224666510646,
      "loss_main": 0.09808255732059479,
      "step": 22630
    },
    {
      "epoch": 2.6491926047273577,
      "grad_norm": 0.3463895916938782,
      "learning_rate": 7.345978879336295e-05,
      "loss": 0.0678,
      "step": 22640
    },
    {
      "epoch": 2.6491926047273577,
      "loss_main": 0.07629690319299698,
      "step": 22640
    },
    {
      "epoch": 2.650362742803651,
      "grad_norm": 0.3014281094074249,
      "learning_rate": 7.343544907192296e-05,
      "loss": 0.0718,
      "step": 22650
    },
    {
      "epoch": 2.650362742803651,
      "loss_main": 0.05771568790078163,
      "step": 22650
    },
    {
      "epoch": 2.6515328808799437,
      "grad_norm": 0.4293534755706787,
      "learning_rate": 7.341110223141235e-05,
      "loss": 0.0625,
      "step": 22660
    },
    {
      "epoch": 2.6515328808799437,
      "loss_main": 0.0644984021782875,
      "step": 22660
    },
    {
      "epoch": 2.652703018956237,
      "grad_norm": 0.2604033052921295,
      "learning_rate": 7.3386748279227e-05,
      "loss": 0.0655,
      "step": 22670
    },
    {
      "epoch": 2.652703018956237,
      "loss_main": 0.08582951128482819,
      "step": 22670
    },
    {
      "epoch": 2.65387315703253,
      "grad_norm": 0.44145792722702026,
      "learning_rate": 7.336238722276501e-05,
      "loss": 0.0822,
      "step": 22680
    },
    {
      "epoch": 2.65387315703253,
      "loss_main": 0.07397811859846115,
      "step": 22680
    },
    {
      "epoch": 2.6550432951088228,
      "grad_norm": 0.25101447105407715,
      "learning_rate": 7.333801906942663e-05,
      "loss": 0.0606,
      "step": 22690
    },
    {
      "epoch": 2.6550432951088228,
      "loss_main": 0.04802235588431358,
      "step": 22690
    },
    {
      "epoch": 2.656213433185116,
      "grad_norm": 0.23026683926582336,
      "learning_rate": 7.331364382661428e-05,
      "loss": 0.0665,
      "step": 22700
    },
    {
      "epoch": 2.656213433185116,
      "loss_main": 0.045722153037786484,
      "step": 22700
    },
    {
      "epoch": 2.657383571261409,
      "grad_norm": 0.24736936390399933,
      "learning_rate": 7.328926150173248e-05,
      "loss": 0.0666,
      "step": 22710
    },
    {
      "epoch": 2.657383571261409,
      "loss_main": 0.06080527976155281,
      "step": 22710
    },
    {
      "epoch": 2.658553709337702,
      "grad_norm": 0.20169329643249512,
      "learning_rate": 7.326487210218795e-05,
      "loss": 0.0694,
      "step": 22720
    },
    {
      "epoch": 2.658553709337702,
      "loss_main": 0.09247977286577225,
      "step": 22720
    },
    {
      "epoch": 2.6597238474139946,
      "grad_norm": 0.2586750090122223,
      "learning_rate": 7.324047563538955e-05,
      "loss": 0.0767,
      "step": 22730
    },
    {
      "epoch": 2.6597238474139946,
      "loss_main": 0.05935347080230713,
      "step": 22730
    },
    {
      "epoch": 2.6608939854902878,
      "grad_norm": 0.22445957362651825,
      "learning_rate": 7.321607210874828e-05,
      "loss": 0.0674,
      "step": 22740
    },
    {
      "epoch": 2.6608939854902878,
      "loss_main": 0.05229220539331436,
      "step": 22740
    },
    {
      "epoch": 2.662064123566581,
      "grad_norm": 0.24689479172229767,
      "learning_rate": 7.31916615296773e-05,
      "loss": 0.0615,
      "step": 22750
    },
    {
      "epoch": 2.662064123566581,
      "loss_main": 0.07373175770044327,
      "step": 22750
    },
    {
      "epoch": 2.6632342616428737,
      "grad_norm": 0.1985841691493988,
      "learning_rate": 7.316724390559188e-05,
      "loss": 0.0641,
      "step": 22760
    },
    {
      "epoch": 2.6632342616428737,
      "loss_main": 0.04292691498994827,
      "step": 22760
    },
    {
      "epoch": 2.664404399719167,
      "grad_norm": 0.4276215434074402,
      "learning_rate": 7.314281924390946e-05,
      "loss": 0.0589,
      "step": 22770
    },
    {
      "epoch": 2.664404399719167,
      "loss_main": 0.11112876981496811,
      "step": 22770
    },
    {
      "epoch": 2.66557453779546,
      "grad_norm": 0.2696966826915741,
      "learning_rate": 7.311838755204959e-05,
      "loss": 0.0721,
      "step": 22780
    },
    {
      "epoch": 2.66557453779546,
      "loss_main": 0.03848341479897499,
      "step": 22780
    },
    {
      "epoch": 2.666744675871753,
      "grad_norm": 0.2505573630332947,
      "learning_rate": 7.3093948837434e-05,
      "loss": 0.0624,
      "step": 22790
    },
    {
      "epoch": 2.666744675871753,
      "loss_main": 0.04525676742196083,
      "step": 22790
    },
    {
      "epoch": 2.667914813948046,
      "grad_norm": 0.29078224301338196,
      "learning_rate": 7.306950310748651e-05,
      "loss": 0.0588,
      "step": 22800
    },
    {
      "epoch": 2.667914813948046,
      "loss_main": 0.07965289056301117,
      "step": 22800
    },
    {
      "epoch": 2.6690849520243387,
      "grad_norm": 0.4356679618358612,
      "learning_rate": 7.304505036963311e-05,
      "loss": 0.0696,
      "step": 22810
    },
    {
      "epoch": 2.6690849520243387,
      "loss_main": 0.09336019307374954,
      "step": 22810
    },
    {
      "epoch": 2.670255090100632,
      "grad_norm": 0.3817191421985626,
      "learning_rate": 7.302059063130186e-05,
      "loss": 0.0663,
      "step": 22820
    },
    {
      "epoch": 2.670255090100632,
      "loss_main": 0.04931186884641647,
      "step": 22820
    },
    {
      "epoch": 2.6714252281769246,
      "grad_norm": 0.37821048498153687,
      "learning_rate": 7.2996123899923e-05,
      "loss": 0.0646,
      "step": 22830
    },
    {
      "epoch": 2.6714252281769246,
      "loss_main": 0.07613150030374527,
      "step": 22830
    },
    {
      "epoch": 2.672595366253218,
      "grad_norm": 0.36565855145454407,
      "learning_rate": 7.297165018292886e-05,
      "loss": 0.0756,
      "step": 22840
    },
    {
      "epoch": 2.672595366253218,
      "loss_main": 0.041537050157785416,
      "step": 22840
    },
    {
      "epoch": 2.673765504329511,
      "grad_norm": 0.21997663378715515,
      "learning_rate": 7.294716948775396e-05,
      "loss": 0.0704,
      "step": 22850
    },
    {
      "epoch": 2.673765504329511,
      "loss_main": 0.10185461491346359,
      "step": 22850
    },
    {
      "epoch": 2.6749356424058037,
      "grad_norm": 0.21269366145133972,
      "learning_rate": 7.292268182183484e-05,
      "loss": 0.0741,
      "step": 22860
    },
    {
      "epoch": 2.6749356424058037,
      "loss_main": 0.05069103464484215,
      "step": 22860
    },
    {
      "epoch": 2.676105780482097,
      "grad_norm": 0.22706910967826843,
      "learning_rate": 7.28981871926102e-05,
      "loss": 0.0616,
      "step": 22870
    },
    {
      "epoch": 2.676105780482097,
      "loss_main": 0.06210595741868019,
      "step": 22870
    },
    {
      "epoch": 2.67727591855839,
      "grad_norm": 0.18829679489135742,
      "learning_rate": 7.28736856075209e-05,
      "loss": 0.0716,
      "step": 22880
    },
    {
      "epoch": 2.67727591855839,
      "loss_main": 0.04948825761675835,
      "step": 22880
    },
    {
      "epoch": 2.678446056634683,
      "grad_norm": 0.25525814294815063,
      "learning_rate": 7.284917707400985e-05,
      "loss": 0.0622,
      "step": 22890
    },
    {
      "epoch": 2.678446056634683,
      "loss_main": 0.09853880852460861,
      "step": 22890
    },
    {
      "epoch": 2.679616194710976,
      "grad_norm": 0.2585681974887848,
      "learning_rate": 7.282466159952212e-05,
      "loss": 0.0743,
      "step": 22900
    },
    {
      "epoch": 2.679616194710976,
      "loss_main": 0.05473758652806282,
      "step": 22900
    },
    {
      "epoch": 2.6807863327872687,
      "grad_norm": 0.47103795409202576,
      "learning_rate": 7.280013919150483e-05,
      "loss": 0.0675,
      "step": 22910
    },
    {
      "epoch": 2.6807863327872687,
      "loss_main": 0.10354088246822357,
      "step": 22910
    },
    {
      "epoch": 2.681956470863562,
      "grad_norm": 0.3270910382270813,
      "learning_rate": 7.277560985740728e-05,
      "loss": 0.0855,
      "step": 22920
    },
    {
      "epoch": 2.681956470863562,
      "loss_main": 0.05994001775979996,
      "step": 22920
    },
    {
      "epoch": 2.6831266089398547,
      "grad_norm": 0.36126652359962463,
      "learning_rate": 7.275107360468079e-05,
      "loss": 0.0713,
      "step": 22930
    },
    {
      "epoch": 2.6831266089398547,
      "loss_main": 0.04724122956395149,
      "step": 22930
    },
    {
      "epoch": 2.684296747016148,
      "grad_norm": 0.38545018434524536,
      "learning_rate": 7.272653044077885e-05,
      "loss": 0.0752,
      "step": 22940
    },
    {
      "epoch": 2.684296747016148,
      "loss_main": 0.09458569437265396,
      "step": 22940
    },
    {
      "epoch": 2.685466885092441,
      "grad_norm": 0.4048512876033783,
      "learning_rate": 7.270198037315703e-05,
      "loss": 0.0745,
      "step": 22950
    },
    {
      "epoch": 2.685466885092441,
      "loss_main": 0.08430496603250504,
      "step": 22950
    },
    {
      "epoch": 2.6866370231687338,
      "grad_norm": 0.29372894763946533,
      "learning_rate": 7.267742340927297e-05,
      "loss": 0.0684,
      "step": 22960
    },
    {
      "epoch": 2.6866370231687338,
      "loss_main": 0.06899867206811905,
      "step": 22960
    },
    {
      "epoch": 2.687807161245027,
      "grad_norm": 0.2257160246372223,
      "learning_rate": 7.265285955658645e-05,
      "loss": 0.0647,
      "step": 22970
    },
    {
      "epoch": 2.687807161245027,
      "loss_main": 0.06616123765707016,
      "step": 22970
    },
    {
      "epoch": 2.68897729932132,
      "grad_norm": 0.23257863521575928,
      "learning_rate": 7.26282888225593e-05,
      "loss": 0.0678,
      "step": 22980
    },
    {
      "epoch": 2.68897729932132,
      "loss_main": 0.05793452262878418,
      "step": 22980
    },
    {
      "epoch": 2.690147437397613,
      "grad_norm": 0.3095945417881012,
      "learning_rate": 7.260371121465548e-05,
      "loss": 0.0709,
      "step": 22990
    },
    {
      "epoch": 2.690147437397613,
      "loss_main": 0.06614399701356888,
      "step": 22990
    },
    {
      "epoch": 2.691317575473906,
      "grad_norm": 0.22158649563789368,
      "learning_rate": 7.2579126740341e-05,
      "loss": 0.0747,
      "step": 23000
    },
    {
      "epoch": 2.691317575473906,
      "loss_main": 0.09491126239299774,
      "step": 23000
    },
    {
      "epoch": 2.692487713550199,
      "grad_norm": 0.32875972986221313,
      "learning_rate": 7.2554535407084e-05,
      "loss": 0.0705,
      "step": 23010
    },
    {
      "epoch": 2.692487713550199,
      "loss_main": 0.03815263882279396,
      "step": 23010
    },
    {
      "epoch": 2.693657851626492,
      "grad_norm": 0.3520798683166504,
      "learning_rate": 7.252993722235464e-05,
      "loss": 0.0663,
      "step": 23020
    },
    {
      "epoch": 2.693657851626492,
      "loss_main": 0.051494427025318146,
      "step": 23020
    },
    {
      "epoch": 2.6948279897027847,
      "grad_norm": 0.4028630256652832,
      "learning_rate": 7.250533219362523e-05,
      "loss": 0.0672,
      "step": 23030
    },
    {
      "epoch": 2.6948279897027847,
      "loss_main": 0.07999972254037857,
      "step": 23030
    },
    {
      "epoch": 2.695998127779078,
      "grad_norm": 0.31912344694137573,
      "learning_rate": 7.248072032837012e-05,
      "loss": 0.071,
      "step": 23040
    },
    {
      "epoch": 2.695998127779078,
      "loss_main": 0.07771378755569458,
      "step": 23040
    },
    {
      "epoch": 2.697168265855371,
      "grad_norm": 0.26331114768981934,
      "learning_rate": 7.245610163406575e-05,
      "loss": 0.0718,
      "step": 23050
    },
    {
      "epoch": 2.697168265855371,
      "loss_main": 0.06994162499904633,
      "step": 23050
    },
    {
      "epoch": 2.698338403931664,
      "grad_norm": 0.2193058878183365,
      "learning_rate": 7.243147611819061e-05,
      "loss": 0.0741,
      "step": 23060
    },
    {
      "epoch": 2.698338403931664,
      "loss_main": 0.056539375334978104,
      "step": 23060
    },
    {
      "epoch": 2.699508542007957,
      "grad_norm": 0.3327459692955017,
      "learning_rate": 7.240684378822531e-05,
      "loss": 0.0637,
      "step": 23070
    },
    {
      "epoch": 2.699508542007957,
      "loss_main": 0.04947492852807045,
      "step": 23070
    },
    {
      "epoch": 2.70067868008425,
      "grad_norm": 0.24042168259620667,
      "learning_rate": 7.238220465165248e-05,
      "loss": 0.0616,
      "step": 23080
    },
    {
      "epoch": 2.70067868008425,
      "loss_main": 0.05984632298350334,
      "step": 23080
    },
    {
      "epoch": 2.701848818160543,
      "grad_norm": 0.3528936505317688,
      "learning_rate": 7.235755871595684e-05,
      "loss": 0.0693,
      "step": 23090
    },
    {
      "epoch": 2.701848818160543,
      "loss_main": 0.05071733519434929,
      "step": 23090
    },
    {
      "epoch": 2.703018956236836,
      "grad_norm": 0.23736360669136047,
      "learning_rate": 7.233290598862517e-05,
      "loss": 0.0702,
      "step": 23100
    },
    {
      "epoch": 2.703018956236836,
      "loss_main": 0.08824712038040161,
      "step": 23100
    },
    {
      "epoch": 2.704189094313129,
      "grad_norm": 0.3101581037044525,
      "learning_rate": 7.230824647714635e-05,
      "loss": 0.0687,
      "step": 23110
    },
    {
      "epoch": 2.704189094313129,
      "loss_main": 0.06829248368740082,
      "step": 23110
    },
    {
      "epoch": 2.705359232389422,
      "grad_norm": 0.34928566217422485,
      "learning_rate": 7.228358018901124e-05,
      "loss": 0.0739,
      "step": 23120
    },
    {
      "epoch": 2.705359232389422,
      "loss_main": 0.07558334618806839,
      "step": 23120
    },
    {
      "epoch": 2.7065293704657147,
      "grad_norm": 0.22550459206104279,
      "learning_rate": 7.225890713171286e-05,
      "loss": 0.0744,
      "step": 23130
    },
    {
      "epoch": 2.7065293704657147,
      "loss_main": 0.08321472257375717,
      "step": 23130
    },
    {
      "epoch": 2.707699508542008,
      "grad_norm": 0.1778043806552887,
      "learning_rate": 7.223422731274618e-05,
      "loss": 0.0605,
      "step": 23140
    },
    {
      "epoch": 2.707699508542008,
      "loss_main": 0.04773992672562599,
      "step": 23140
    },
    {
      "epoch": 2.708869646618301,
      "grad_norm": 0.2990787625312805,
      "learning_rate": 7.220954073960832e-05,
      "loss": 0.0613,
      "step": 23150
    },
    {
      "epoch": 2.708869646618301,
      "loss_main": 0.12206239998340607,
      "step": 23150
    },
    {
      "epoch": 2.710039784694594,
      "grad_norm": 0.2945013642311096,
      "learning_rate": 7.218484741979838e-05,
      "loss": 0.0674,
      "step": 23160
    },
    {
      "epoch": 2.710039784694594,
      "loss_main": 0.04381543770432472,
      "step": 23160
    },
    {
      "epoch": 2.711209922770887,
      "grad_norm": 0.29903969168663025,
      "learning_rate": 7.216014736081756e-05,
      "loss": 0.071,
      "step": 23170
    },
    {
      "epoch": 2.711209922770887,
      "loss_main": 0.07213695347309113,
      "step": 23170
    },
    {
      "epoch": 2.71238006084718,
      "grad_norm": 0.27604201436042786,
      "learning_rate": 7.213544057016906e-05,
      "loss": 0.0645,
      "step": 23180
    },
    {
      "epoch": 2.71238006084718,
      "loss_main": 0.055348966270685196,
      "step": 23180
    },
    {
      "epoch": 2.713550198923473,
      "grad_norm": 0.3824429214000702,
      "learning_rate": 7.211072705535819e-05,
      "loss": 0.0671,
      "step": 23190
    },
    {
      "epoch": 2.713550198923473,
      "loss_main": 0.050349969416856766,
      "step": 23190
    },
    {
      "epoch": 2.714720336999766,
      "grad_norm": 0.30298471450805664,
      "learning_rate": 7.208600682389224e-05,
      "loss": 0.067,
      "step": 23200
    },
    {
      "epoch": 2.714720336999766,
      "loss_main": 0.06009816378355026,
      "step": 23200
    },
    {
      "epoch": 2.715890475076059,
      "grad_norm": 0.3149143159389496,
      "learning_rate": 7.206127988328055e-05,
      "loss": 0.0674,
      "step": 23210
    },
    {
      "epoch": 2.715890475076059,
      "loss_main": 0.04161551222205162,
      "step": 23210
    },
    {
      "epoch": 2.717060613152352,
      "grad_norm": 0.2527128756046295,
      "learning_rate": 7.203654624103453e-05,
      "loss": 0.0698,
      "step": 23220
    },
    {
      "epoch": 2.717060613152352,
      "loss_main": 0.03926769271492958,
      "step": 23220
    },
    {
      "epoch": 2.7182307512286448,
      "grad_norm": 0.3189641535282135,
      "learning_rate": 7.201180590466761e-05,
      "loss": 0.061,
      "step": 23230
    },
    {
      "epoch": 2.7182307512286448,
      "loss_main": 0.057495422661304474,
      "step": 23230
    },
    {
      "epoch": 2.719400889304938,
      "grad_norm": 0.4274697005748749,
      "learning_rate": 7.198705888169523e-05,
      "loss": 0.0683,
      "step": 23240
    },
    {
      "epoch": 2.719400889304938,
      "loss_main": 0.04922008514404297,
      "step": 23240
    },
    {
      "epoch": 2.720571027381231,
      "grad_norm": 0.32428810000419617,
      "learning_rate": 7.196230517963491e-05,
      "loss": 0.0601,
      "step": 23250
    },
    {
      "epoch": 2.720571027381231,
      "loss_main": 0.07527618110179901,
      "step": 23250
    },
    {
      "epoch": 2.721741165457524,
      "grad_norm": 0.27369409799575806,
      "learning_rate": 7.193754480600615e-05,
      "loss": 0.072,
      "step": 23260
    },
    {
      "epoch": 2.721741165457524,
      "loss_main": 0.053529851138591766,
      "step": 23260
    },
    {
      "epoch": 2.722911303533817,
      "grad_norm": 0.25448912382125854,
      "learning_rate": 7.19127777683305e-05,
      "loss": 0.0807,
      "step": 23270
    },
    {
      "epoch": 2.722911303533817,
      "loss_main": 0.040912117809057236,
      "step": 23270
    },
    {
      "epoch": 2.7240814416101102,
      "grad_norm": 0.26711663603782654,
      "learning_rate": 7.188800407413156e-05,
      "loss": 0.0579,
      "step": 23280
    },
    {
      "epoch": 2.7240814416101102,
      "loss_main": 0.03994370624423027,
      "step": 23280
    },
    {
      "epoch": 2.725251579686403,
      "grad_norm": 0.3856540322303772,
      "learning_rate": 7.186322373093489e-05,
      "loss": 0.0676,
      "step": 23290
    },
    {
      "epoch": 2.725251579686403,
      "loss_main": 0.04680878669023514,
      "step": 23290
    },
    {
      "epoch": 2.726421717762696,
      "grad_norm": 0.2569025456905365,
      "learning_rate": 7.18384367462681e-05,
      "loss": 0.064,
      "step": 23300
    },
    {
      "epoch": 2.726421717762696,
      "loss_main": 0.09625288099050522,
      "step": 23300
    },
    {
      "epoch": 2.727591855838989,
      "grad_norm": 0.35262224078178406,
      "learning_rate": 7.181364312766085e-05,
      "loss": 0.0726,
      "step": 23310
    },
    {
      "epoch": 2.727591855838989,
      "loss_main": 0.0627734512090683,
      "step": 23310
    },
    {
      "epoch": 2.728761993915282,
      "grad_norm": 0.24402527511119843,
      "learning_rate": 7.178884288264477e-05,
      "loss": 0.0681,
      "step": 23320
    },
    {
      "epoch": 2.728761993915282,
      "loss_main": 0.06165387108922005,
      "step": 23320
    },
    {
      "epoch": 2.729932131991575,
      "grad_norm": 0.33702829480171204,
      "learning_rate": 7.176403601875353e-05,
      "loss": 0.0667,
      "step": 23330
    },
    {
      "epoch": 2.729932131991575,
      "loss_main": 0.05022883415222168,
      "step": 23330
    },
    {
      "epoch": 2.731102270067868,
      "grad_norm": 0.24019628763198853,
      "learning_rate": 7.173922254352279e-05,
      "loss": 0.0683,
      "step": 23340
    },
    {
      "epoch": 2.731102270067868,
      "loss_main": 0.04465160146355629,
      "step": 23340
    },
    {
      "epoch": 2.732272408144161,
      "grad_norm": 0.33998116850852966,
      "learning_rate": 7.171440246449024e-05,
      "loss": 0.0656,
      "step": 23350
    },
    {
      "epoch": 2.732272408144161,
      "loss_main": 0.05739263445138931,
      "step": 23350
    },
    {
      "epoch": 2.733442546220454,
      "grad_norm": 0.444721519947052,
      "learning_rate": 7.168957578919555e-05,
      "loss": 0.0656,
      "step": 23360
    },
    {
      "epoch": 2.733442546220454,
      "loss_main": 0.06101798638701439,
      "step": 23360
    },
    {
      "epoch": 2.734612684296747,
      "grad_norm": 0.36340439319610596,
      "learning_rate": 7.16647425251804e-05,
      "loss": 0.0706,
      "step": 23370
    },
    {
      "epoch": 2.734612684296747,
      "loss_main": 0.053527314215898514,
      "step": 23370
    },
    {
      "epoch": 2.7357828223730403,
      "grad_norm": 0.26322004199028015,
      "learning_rate": 7.163990267998852e-05,
      "loss": 0.0726,
      "step": 23380
    },
    {
      "epoch": 2.7357828223730403,
      "loss_main": 0.0550859272480011,
      "step": 23380
    },
    {
      "epoch": 2.736952960449333,
      "grad_norm": 0.25792941451072693,
      "learning_rate": 7.161505626116556e-05,
      "loss": 0.0722,
      "step": 23390
    },
    {
      "epoch": 2.736952960449333,
      "loss_main": 0.06717916578054428,
      "step": 23390
    },
    {
      "epoch": 2.738123098525626,
      "grad_norm": 0.34860846400260925,
      "learning_rate": 7.159020327625923e-05,
      "loss": 0.071,
      "step": 23400
    },
    {
      "epoch": 2.738123098525626,
      "loss_main": 0.08593542128801346,
      "step": 23400
    },
    {
      "epoch": 2.739293236601919,
      "grad_norm": 0.2163257896900177,
      "learning_rate": 7.15653437328192e-05,
      "loss": 0.0796,
      "step": 23410
    },
    {
      "epoch": 2.739293236601919,
      "loss_main": 0.07316052913665771,
      "step": 23410
    },
    {
      "epoch": 2.740463374678212,
      "grad_norm": 0.25328606367111206,
      "learning_rate": 7.154047763839713e-05,
      "loss": 0.0649,
      "step": 23420
    },
    {
      "epoch": 2.740463374678212,
      "loss_main": 0.05853003263473511,
      "step": 23420
    },
    {
      "epoch": 2.741633512754505,
      "grad_norm": 0.24800768494606018,
      "learning_rate": 7.15156050005467e-05,
      "loss": 0.0594,
      "step": 23430
    },
    {
      "epoch": 2.741633512754505,
      "loss_main": 0.0755372866988182,
      "step": 23430
    },
    {
      "epoch": 2.742803650830798,
      "grad_norm": 0.30247512459754944,
      "learning_rate": 7.149072582682357e-05,
      "loss": 0.0667,
      "step": 23440
    },
    {
      "epoch": 2.742803650830798,
      "loss_main": 0.05659285932779312,
      "step": 23440
    },
    {
      "epoch": 2.743973788907091,
      "grad_norm": 0.2561779022216797,
      "learning_rate": 7.146584012478535e-05,
      "loss": 0.0605,
      "step": 23450
    },
    {
      "epoch": 2.743973788907091,
      "loss_main": 0.07964903116226196,
      "step": 23450
    },
    {
      "epoch": 2.745143926983384,
      "grad_norm": 0.22100837528705597,
      "learning_rate": 7.144094790199169e-05,
      "loss": 0.0661,
      "step": 23460
    },
    {
      "epoch": 2.745143926983384,
      "loss_main": 0.06690468639135361,
      "step": 23460
    },
    {
      "epoch": 2.746314065059677,
      "grad_norm": 0.18420551717281342,
      "learning_rate": 7.141604916600415e-05,
      "loss": 0.071,
      "step": 23470
    },
    {
      "epoch": 2.746314065059677,
      "loss_main": 0.05074656009674072,
      "step": 23470
    },
    {
      "epoch": 2.7474842031359703,
      "grad_norm": 0.23482786118984222,
      "learning_rate": 7.139114392438635e-05,
      "loss": 0.0604,
      "step": 23480
    },
    {
      "epoch": 2.7474842031359703,
      "loss_main": 0.06845761090517044,
      "step": 23480
    },
    {
      "epoch": 2.748654341212263,
      "grad_norm": 0.19316411018371582,
      "learning_rate": 7.136623218470382e-05,
      "loss": 0.0649,
      "step": 23490
    },
    {
      "epoch": 2.748654341212263,
      "loss_main": 0.059030670672655106,
      "step": 23490
    },
    {
      "epoch": 2.749824479288556,
      "grad_norm": 0.24721167981624603,
      "learning_rate": 7.13413139545241e-05,
      "loss": 0.0667,
      "step": 23500
    },
    {
      "epoch": 2.749824479288556,
      "loss_main": 0.0800863578915596,
      "step": 23500
    },
    {
      "epoch": 2.750994617364849,
      "grad_norm": 0.2362522929906845,
      "learning_rate": 7.131638924141668e-05,
      "loss": 0.0686,
      "step": 23510
    },
    {
      "epoch": 2.750994617364849,
      "loss_main": 0.03735736757516861,
      "step": 23510
    },
    {
      "epoch": 2.752164755441142,
      "grad_norm": 0.25895968079566956,
      "learning_rate": 7.129145805295304e-05,
      "loss": 0.0681,
      "step": 23520
    },
    {
      "epoch": 2.752164755441142,
      "loss_main": 0.03499867022037506,
      "step": 23520
    },
    {
      "epoch": 2.753334893517435,
      "grad_norm": 0.38579219579696655,
      "learning_rate": 7.126652039670661e-05,
      "loss": 0.0573,
      "step": 23530
    },
    {
      "epoch": 2.753334893517435,
      "loss_main": 0.06859222799539566,
      "step": 23530
    },
    {
      "epoch": 2.754505031593728,
      "grad_norm": 0.27252113819122314,
      "learning_rate": 7.124157628025278e-05,
      "loss": 0.0669,
      "step": 23540
    },
    {
      "epoch": 2.754505031593728,
      "loss_main": 0.06080948933959007,
      "step": 23540
    },
    {
      "epoch": 2.7556751696700212,
      "grad_norm": 0.3372681736946106,
      "learning_rate": 7.121662571116894e-05,
      "loss": 0.0667,
      "step": 23550
    },
    {
      "epoch": 2.7556751696700212,
      "loss_main": 0.05032151937484741,
      "step": 23550
    },
    {
      "epoch": 2.756845307746314,
      "grad_norm": 0.29340696334838867,
      "learning_rate": 7.119166869703441e-05,
      "loss": 0.056,
      "step": 23560
    },
    {
      "epoch": 2.756845307746314,
      "loss_main": 0.04167713597416878,
      "step": 23560
    },
    {
      "epoch": 2.758015445822607,
      "grad_norm": 0.2678355574607849,
      "learning_rate": 7.116670524543044e-05,
      "loss": 0.071,
      "step": 23570
    },
    {
      "epoch": 2.758015445822607,
      "loss_main": 0.04709475114941597,
      "step": 23570
    },
    {
      "epoch": 2.7591855838989003,
      "grad_norm": 0.3198591470718384,
      "learning_rate": 7.114173536394032e-05,
      "loss": 0.0677,
      "step": 23580
    },
    {
      "epoch": 2.7591855838989003,
      "loss_main": 0.055503733456134796,
      "step": 23580
    },
    {
      "epoch": 2.760355721975193,
      "grad_norm": 0.3037581741809845,
      "learning_rate": 7.111675906014917e-05,
      "loss": 0.0703,
      "step": 23590
    },
    {
      "epoch": 2.760355721975193,
      "loss_main": 0.07352820783853531,
      "step": 23590
    },
    {
      "epoch": 2.761525860051486,
      "grad_norm": 0.26043054461479187,
      "learning_rate": 7.109177634164421e-05,
      "loss": 0.0662,
      "step": 23600
    },
    {
      "epoch": 2.761525860051486,
      "loss_main": 0.07099103927612305,
      "step": 23600
    },
    {
      "epoch": 2.762695998127779,
      "grad_norm": 0.28445661067962646,
      "learning_rate": 7.106678721601449e-05,
      "loss": 0.0651,
      "step": 23610
    },
    {
      "epoch": 2.762695998127779,
      "loss_main": 0.09453771263360977,
      "step": 23610
    },
    {
      "epoch": 2.763866136204072,
      "grad_norm": 0.31862616539001465,
      "learning_rate": 7.104179169085103e-05,
      "loss": 0.0703,
      "step": 23620
    },
    {
      "epoch": 2.763866136204072,
      "loss_main": 0.06383176147937775,
      "step": 23620
    },
    {
      "epoch": 2.765036274280365,
      "grad_norm": 0.3036882281303406,
      "learning_rate": 7.101678977374683e-05,
      "loss": 0.0802,
      "step": 23630
    },
    {
      "epoch": 2.765036274280365,
      "loss_main": 0.055455684661865234,
      "step": 23630
    },
    {
      "epoch": 2.766206412356658,
      "grad_norm": 0.3897288739681244,
      "learning_rate": 7.099178147229685e-05,
      "loss": 0.0672,
      "step": 23640
    },
    {
      "epoch": 2.766206412356658,
      "loss_main": 0.07974941283464432,
      "step": 23640
    },
    {
      "epoch": 2.7673765504329513,
      "grad_norm": 0.19333340227603912,
      "learning_rate": 7.096676679409789e-05,
      "loss": 0.0667,
      "step": 23650
    },
    {
      "epoch": 2.7673765504329513,
      "loss_main": 0.05518332123756409,
      "step": 23650
    },
    {
      "epoch": 2.768546688509244,
      "grad_norm": 0.20206600427627563,
      "learning_rate": 7.094174574674877e-05,
      "loss": 0.0723,
      "step": 23660
    },
    {
      "epoch": 2.768546688509244,
      "loss_main": 0.10551273822784424,
      "step": 23660
    },
    {
      "epoch": 2.769716826585537,
      "grad_norm": 0.351557195186615,
      "learning_rate": 7.091671833785025e-05,
      "loss": 0.0623,
      "step": 23670
    },
    {
      "epoch": 2.769716826585537,
      "loss_main": 0.03333247825503349,
      "step": 23670
    },
    {
      "epoch": 2.77088696466183,
      "grad_norm": 0.2908381223678589,
      "learning_rate": 7.089168457500493e-05,
      "loss": 0.0637,
      "step": 23680
    },
    {
      "epoch": 2.77088696466183,
      "loss_main": 0.05194016918540001,
      "step": 23680
    },
    {
      "epoch": 2.772057102738123,
      "grad_norm": 0.2516986131668091,
      "learning_rate": 7.086664446581747e-05,
      "loss": 0.0668,
      "step": 23690
    },
    {
      "epoch": 2.772057102738123,
      "loss_main": 0.03332524746656418,
      "step": 23690
    },
    {
      "epoch": 2.773227240814416,
      "grad_norm": 0.27708789706230164,
      "learning_rate": 7.084159801789438e-05,
      "loss": 0.0607,
      "step": 23700
    },
    {
      "epoch": 2.773227240814416,
      "loss_main": 0.05463240295648575,
      "step": 23700
    },
    {
      "epoch": 2.774397378890709,
      "grad_norm": 0.4215051233768463,
      "learning_rate": 7.081654523884411e-05,
      "loss": 0.072,
      "step": 23710
    },
    {
      "epoch": 2.774397378890709,
      "loss_main": 0.05871450901031494,
      "step": 23710
    },
    {
      "epoch": 2.775567516967002,
      "grad_norm": 0.2737031579017639,
      "learning_rate": 7.0791486136277e-05,
      "loss": 0.0699,
      "step": 23720
    },
    {
      "epoch": 2.775567516967002,
      "loss_main": 0.0762096643447876,
      "step": 23720
    },
    {
      "epoch": 2.776737655043295,
      "grad_norm": 0.3655169904232025,
      "learning_rate": 7.07664207178054e-05,
      "loss": 0.0789,
      "step": 23730
    },
    {
      "epoch": 2.776737655043295,
      "loss_main": 0.04915882274508476,
      "step": 23730
    },
    {
      "epoch": 2.777907793119588,
      "grad_norm": 0.3427583575248718,
      "learning_rate": 7.074134899104345e-05,
      "loss": 0.0746,
      "step": 23740
    },
    {
      "epoch": 2.777907793119588,
      "loss_main": 0.06803499907255173,
      "step": 23740
    },
    {
      "epoch": 2.7790779311958813,
      "grad_norm": 0.21743229031562805,
      "learning_rate": 7.071627096360735e-05,
      "loss": 0.0577,
      "step": 23750
    },
    {
      "epoch": 2.7790779311958813,
      "loss_main": 0.06005415692925453,
      "step": 23750
    },
    {
      "epoch": 2.780248069272174,
      "grad_norm": 0.18891024589538574,
      "learning_rate": 7.069118664311511e-05,
      "loss": 0.0703,
      "step": 23760
    },
    {
      "epoch": 2.780248069272174,
      "loss_main": 0.050068557262420654,
      "step": 23760
    },
    {
      "epoch": 2.7814182073484672,
      "grad_norm": 0.2989934980869293,
      "learning_rate": 7.06660960371867e-05,
      "loss": 0.0557,
      "step": 23770
    },
    {
      "epoch": 2.7814182073484672,
      "loss_main": 0.06965897232294083,
      "step": 23770
    },
    {
      "epoch": 2.78258834542476,
      "grad_norm": 0.361079603433609,
      "learning_rate": 7.064099915344396e-05,
      "loss": 0.0788,
      "step": 23780
    },
    {
      "epoch": 2.78258834542476,
      "loss_main": 0.05141342431306839,
      "step": 23780
    },
    {
      "epoch": 2.783758483501053,
      "grad_norm": 0.3455384075641632,
      "learning_rate": 7.061589599951066e-05,
      "loss": 0.0656,
      "step": 23790
    },
    {
      "epoch": 2.783758483501053,
      "loss_main": 0.04808235168457031,
      "step": 23790
    },
    {
      "epoch": 2.784928621577346,
      "grad_norm": 0.25030165910720825,
      "learning_rate": 7.05907865830125e-05,
      "loss": 0.0704,
      "step": 23800
    },
    {
      "epoch": 2.784928621577346,
      "loss_main": 0.043057408183813095,
      "step": 23800
    },
    {
      "epoch": 2.786098759653639,
      "grad_norm": 0.32876724004745483,
      "learning_rate": 7.056567091157703e-05,
      "loss": 0.0633,
      "step": 23810
    },
    {
      "epoch": 2.786098759653639,
      "loss_main": 0.07822020351886749,
      "step": 23810
    },
    {
      "epoch": 2.7872688977299322,
      "grad_norm": 0.16651730239391327,
      "learning_rate": 7.054054899283375e-05,
      "loss": 0.0715,
      "step": 23820
    },
    {
      "epoch": 2.7872688977299322,
      "loss_main": 0.08381533622741699,
      "step": 23820
    },
    {
      "epoch": 2.788439035806225,
      "grad_norm": 0.3258245885372162,
      "learning_rate": 7.051542083441403e-05,
      "loss": 0.0658,
      "step": 23830
    },
    {
      "epoch": 2.788439035806225,
      "loss_main": 0.08313963562250137,
      "step": 23830
    },
    {
      "epoch": 2.789609173882518,
      "grad_norm": 0.3481309413909912,
      "learning_rate": 7.049028644395113e-05,
      "loss": 0.0652,
      "step": 23840
    },
    {
      "epoch": 2.789609173882518,
      "loss_main": 0.045839328318834305,
      "step": 23840
    },
    {
      "epoch": 2.7907793119588113,
      "grad_norm": 0.2211027443408966,
      "learning_rate": 7.046514582908024e-05,
      "loss": 0.059,
      "step": 23850
    },
    {
      "epoch": 2.7907793119588113,
      "loss_main": 0.05371808260679245,
      "step": 23850
    },
    {
      "epoch": 2.791949450035104,
      "grad_norm": 0.23271647095680237,
      "learning_rate": 7.043999899743838e-05,
      "loss": 0.0625,
      "step": 23860
    },
    {
      "epoch": 2.791949450035104,
      "loss_main": 0.06110069528222084,
      "step": 23860
    },
    {
      "epoch": 2.7931195881113973,
      "grad_norm": 0.22415469586849213,
      "learning_rate": 7.041484595666451e-05,
      "loss": 0.0712,
      "step": 23870
    },
    {
      "epoch": 2.7931195881113973,
      "loss_main": 0.038022398948669434,
      "step": 23870
    },
    {
      "epoch": 2.79428972618769,
      "grad_norm": 0.23709824681282043,
      "learning_rate": 7.038968671439948e-05,
      "loss": 0.0663,
      "step": 23880
    },
    {
      "epoch": 2.79428972618769,
      "loss_main": 0.1035255715250969,
      "step": 23880
    },
    {
      "epoch": 2.795459864263983,
      "grad_norm": 0.2168743908405304,
      "learning_rate": 7.036452127828596e-05,
      "loss": 0.0746,
      "step": 23890
    },
    {
      "epoch": 2.795459864263983,
      "loss_main": 0.03534390777349472,
      "step": 23890
    },
    {
      "epoch": 2.796630002340276,
      "grad_norm": 0.2799674868583679,
      "learning_rate": 7.033934965596859e-05,
      "loss": 0.063,
      "step": 23900
    },
    {
      "epoch": 2.796630002340276,
      "loss_main": 0.08666444569826126,
      "step": 23900
    },
    {
      "epoch": 2.797800140416569,
      "grad_norm": 0.3554934561252594,
      "learning_rate": 7.031417185509381e-05,
      "loss": 0.0615,
      "step": 23910
    },
    {
      "epoch": 2.797800140416569,
      "loss_main": 0.054760027676820755,
      "step": 23910
    },
    {
      "epoch": 2.7989702784928623,
      "grad_norm": 0.24146252870559692,
      "learning_rate": 7.028898788331e-05,
      "loss": 0.0663,
      "step": 23920
    },
    {
      "epoch": 2.7989702784928623,
      "loss_main": 0.14821062982082367,
      "step": 23920
    },
    {
      "epoch": 2.800140416569155,
      "grad_norm": 0.23012469708919525,
      "learning_rate": 7.026379774826736e-05,
      "loss": 0.0785,
      "step": 23930
    },
    {
      "epoch": 2.800140416569155,
      "loss_main": 0.08903708308935165,
      "step": 23930
    },
    {
      "epoch": 2.801310554645448,
      "grad_norm": 0.24027089774608612,
      "learning_rate": 7.0238601457618e-05,
      "loss": 0.0638,
      "step": 23940
    },
    {
      "epoch": 2.801310554645448,
      "loss_main": 0.08126337081193924,
      "step": 23940
    },
    {
      "epoch": 2.8024806927217414,
      "grad_norm": 0.3398570716381073,
      "learning_rate": 7.02133990190159e-05,
      "loss": 0.0705,
      "step": 23950
    },
    {
      "epoch": 2.8024806927217414,
      "loss_main": 0.0566103458404541,
      "step": 23950
    },
    {
      "epoch": 2.803650830798034,
      "grad_norm": 0.3250636160373688,
      "learning_rate": 7.018819044011687e-05,
      "loss": 0.068,
      "step": 23960
    },
    {
      "epoch": 2.803650830798034,
      "loss_main": 0.06625914573669434,
      "step": 23960
    },
    {
      "epoch": 2.8048209688743273,
      "grad_norm": 0.34008511900901794,
      "learning_rate": 7.016297572857863e-05,
      "loss": 0.0654,
      "step": 23970
    },
    {
      "epoch": 2.8048209688743273,
      "loss_main": 0.045649271458387375,
      "step": 23970
    },
    {
      "epoch": 2.80599110695062,
      "grad_norm": 0.3812340497970581,
      "learning_rate": 7.013775489206072e-05,
      "loss": 0.0635,
      "step": 23980
    },
    {
      "epoch": 2.80599110695062,
      "loss_main": 0.07094168663024902,
      "step": 23980
    },
    {
      "epoch": 2.807161245026913,
      "grad_norm": 0.33304473757743835,
      "learning_rate": 7.01125279382246e-05,
      "loss": 0.0564,
      "step": 23990
    },
    {
      "epoch": 2.807161245026913,
      "loss_main": 0.09781535714864731,
      "step": 23990
    },
    {
      "epoch": 2.808331383103206,
      "grad_norm": 0.24705025553703308,
      "learning_rate": 7.008729487473351e-05,
      "loss": 0.066,
      "step": 24000
    },
    {
      "epoch": 2.808331383103206,
      "loss_main": 0.05690925568342209,
      "step": 24000
    },
    {
      "epoch": 2.809501521179499,
      "grad_norm": 0.3468559682369232,
      "learning_rate": 7.006205570925263e-05,
      "loss": 0.0728,
      "step": 24010
    },
    {
      "epoch": 2.809501521179499,
      "loss_main": 0.0762936919927597,
      "step": 24010
    },
    {
      "epoch": 2.8106716592557923,
      "grad_norm": 0.32011809945106506,
      "learning_rate": 7.003681044944892e-05,
      "loss": 0.0699,
      "step": 24020
    },
    {
      "epoch": 2.8106716592557923,
      "loss_main": 0.05033205822110176,
      "step": 24020
    },
    {
      "epoch": 2.811841797332085,
      "grad_norm": 0.29116955399513245,
      "learning_rate": 7.001155910299126e-05,
      "loss": 0.0646,
      "step": 24030
    },
    {
      "epoch": 2.811841797332085,
      "loss_main": 0.09301559627056122,
      "step": 24030
    },
    {
      "epoch": 2.8130119354083782,
      "grad_norm": 0.25186920166015625,
      "learning_rate": 6.99863016775503e-05,
      "loss": 0.0763,
      "step": 24040
    },
    {
      "epoch": 2.8130119354083782,
      "loss_main": 0.07851159572601318,
      "step": 24040
    },
    {
      "epoch": 2.8141820734846714,
      "grad_norm": 0.27505916357040405,
      "learning_rate": 6.996103818079859e-05,
      "loss": 0.0663,
      "step": 24050
    },
    {
      "epoch": 2.8141820734846714,
      "loss_main": 0.08552849292755127,
      "step": 24050
    },
    {
      "epoch": 2.815352211560964,
      "grad_norm": 0.3560147285461426,
      "learning_rate": 6.993576862041054e-05,
      "loss": 0.0702,
      "step": 24060
    },
    {
      "epoch": 2.815352211560964,
      "loss_main": 0.08264541625976562,
      "step": 24060
    },
    {
      "epoch": 2.8165223496372573,
      "grad_norm": 0.2702954411506653,
      "learning_rate": 6.991049300406235e-05,
      "loss": 0.0562,
      "step": 24070
    },
    {
      "epoch": 2.8165223496372573,
      "loss_main": 0.06909497827291489,
      "step": 24070
    },
    {
      "epoch": 2.81769248771355,
      "grad_norm": 0.25214043259620667,
      "learning_rate": 6.988521133943209e-05,
      "loss": 0.0642,
      "step": 24080
    },
    {
      "epoch": 2.81769248771355,
      "loss_main": 0.08245086669921875,
      "step": 24080
    },
    {
      "epoch": 2.8188626257898433,
      "grad_norm": 0.44602885842323303,
      "learning_rate": 6.985992363419966e-05,
      "loss": 0.0668,
      "step": 24090
    },
    {
      "epoch": 2.8188626257898433,
      "loss_main": 0.06325764954090118,
      "step": 24090
    },
    {
      "epoch": 2.820032763866136,
      "grad_norm": 0.4558545649051666,
      "learning_rate": 6.983462989604682e-05,
      "loss": 0.073,
      "step": 24100
    },
    {
      "epoch": 2.820032763866136,
      "loss_main": 0.05565834790468216,
      "step": 24100
    },
    {
      "epoch": 2.821202901942429,
      "grad_norm": 0.2698618471622467,
      "learning_rate": 6.980933013265709e-05,
      "loss": 0.0561,
      "step": 24110
    },
    {
      "epoch": 2.821202901942429,
      "loss_main": 0.07969795167446136,
      "step": 24110
    },
    {
      "epoch": 2.8223730400187224,
      "grad_norm": 0.33446255326271057,
      "learning_rate": 6.978402435171592e-05,
      "loss": 0.0792,
      "step": 24120
    },
    {
      "epoch": 2.8223730400187224,
      "loss_main": 0.049048006534576416,
      "step": 24120
    },
    {
      "epoch": 2.823543178095015,
      "grad_norm": 0.3580068051815033,
      "learning_rate": 6.975871256091052e-05,
      "loss": 0.0705,
      "step": 24130
    },
    {
      "epoch": 2.823543178095015,
      "loss_main": 0.04462417587637901,
      "step": 24130
    },
    {
      "epoch": 2.8247133161713083,
      "grad_norm": 0.30950865149497986,
      "learning_rate": 6.973339476792995e-05,
      "loss": 0.064,
      "step": 24140
    },
    {
      "epoch": 2.8247133161713083,
      "loss_main": 0.07310213148593903,
      "step": 24140
    },
    {
      "epoch": 2.8258834542476015,
      "grad_norm": 0.37485575675964355,
      "learning_rate": 6.970807098046505e-05,
      "loss": 0.0621,
      "step": 24150
    },
    {
      "epoch": 2.8258834542476015,
      "loss_main": 0.04300787299871445,
      "step": 24150
    },
    {
      "epoch": 2.827053592323894,
      "grad_norm": 0.3287639319896698,
      "learning_rate": 6.968274120620858e-05,
      "loss": 0.065,
      "step": 24160
    },
    {
      "epoch": 2.827053592323894,
      "loss_main": 0.08966170251369476,
      "step": 24160
    },
    {
      "epoch": 2.8282237304001874,
      "grad_norm": 0.30828821659088135,
      "learning_rate": 6.965740545285499e-05,
      "loss": 0.0658,
      "step": 24170
    },
    {
      "epoch": 2.8282237304001874,
      "loss_main": 0.06475003063678741,
      "step": 24170
    },
    {
      "epoch": 2.82939386847648,
      "grad_norm": 0.18251878023147583,
      "learning_rate": 6.963206372810068e-05,
      "loss": 0.0591,
      "step": 24180
    },
    {
      "epoch": 2.82939386847648,
      "loss_main": 0.09465936571359634,
      "step": 24180
    },
    {
      "epoch": 2.8305640065527733,
      "grad_norm": 0.24210698902606964,
      "learning_rate": 6.960671603964375e-05,
      "loss": 0.0732,
      "step": 24190
    },
    {
      "epoch": 2.8305640065527733,
      "loss_main": 0.07394148409366608,
      "step": 24190
    },
    {
      "epoch": 2.831734144629066,
      "grad_norm": 0.22539447247982025,
      "learning_rate": 6.958136239518418e-05,
      "loss": 0.0662,
      "step": 24200
    },
    {
      "epoch": 2.831734144629066,
      "loss_main": 0.0707530602812767,
      "step": 24200
    },
    {
      "epoch": 2.832904282705359,
      "grad_norm": 0.24937817454338074,
      "learning_rate": 6.955600280242371e-05,
      "loss": 0.0661,
      "step": 24210
    },
    {
      "epoch": 2.832904282705359,
      "loss_main": 0.07232040911912918,
      "step": 24210
    },
    {
      "epoch": 2.8340744207816524,
      "grad_norm": 0.3744507133960724,
      "learning_rate": 6.953063726906596e-05,
      "loss": 0.0655,
      "step": 24220
    },
    {
      "epoch": 2.8340744207816524,
      "loss_main": 0.10186469554901123,
      "step": 24220
    },
    {
      "epoch": 2.835244558857945,
      "grad_norm": 0.19491389393806458,
      "learning_rate": 6.950526580281626e-05,
      "loss": 0.0682,
      "step": 24230
    },
    {
      "epoch": 2.835244558857945,
      "loss_main": 0.044486694037914276,
      "step": 24230
    },
    {
      "epoch": 2.8364146969342383,
      "grad_norm": 0.3618406355381012,
      "learning_rate": 6.947988841138184e-05,
      "loss": 0.0746,
      "step": 24240
    },
    {
      "epoch": 2.8364146969342383,
      "loss_main": 0.038731373846530914,
      "step": 24240
    },
    {
      "epoch": 2.8375848350105315,
      "grad_norm": 0.28299158811569214,
      "learning_rate": 6.945450510247165e-05,
      "loss": 0.0657,
      "step": 24250
    },
    {
      "epoch": 2.8375848350105315,
      "loss_main": 0.06575294584035873,
      "step": 24250
    },
    {
      "epoch": 2.8387549730868242,
      "grad_norm": 0.374537855386734,
      "learning_rate": 6.942911588379647e-05,
      "loss": 0.061,
      "step": 24260
    },
    {
      "epoch": 2.8387549730868242,
      "loss_main": 0.07948003709316254,
      "step": 24260
    },
    {
      "epoch": 2.8399251111631174,
      "grad_norm": 0.37056317925453186,
      "learning_rate": 6.940372076306888e-05,
      "loss": 0.0677,
      "step": 24270
    },
    {
      "epoch": 2.8399251111631174,
      "loss_main": 0.10629028081893921,
      "step": 24270
    },
    {
      "epoch": 2.84109524923941,
      "grad_norm": 0.3319720923900604,
      "learning_rate": 6.937831974800326e-05,
      "loss": 0.0682,
      "step": 24280
    },
    {
      "epoch": 2.84109524923941,
      "loss_main": 0.03654183819890022,
      "step": 24280
    },
    {
      "epoch": 2.8422653873157033,
      "grad_norm": 0.35690081119537354,
      "learning_rate": 6.935291284631574e-05,
      "loss": 0.0781,
      "step": 24290
    },
    {
      "epoch": 2.8422653873157033,
      "loss_main": 0.09822945296764374,
      "step": 24290
    },
    {
      "epoch": 2.843435525391996,
      "grad_norm": 0.3044922351837158,
      "learning_rate": 6.932750006572428e-05,
      "loss": 0.0622,
      "step": 24300
    },
    {
      "epoch": 2.843435525391996,
      "loss_main": 0.045263711363077164,
      "step": 24300
    },
    {
      "epoch": 2.8446056634682892,
      "grad_norm": 0.3732946813106537,
      "learning_rate": 6.930208141394863e-05,
      "loss": 0.0781,
      "step": 24310
    },
    {
      "epoch": 2.8446056634682892,
      "loss_main": 0.06337237358093262,
      "step": 24310
    },
    {
      "epoch": 2.8457758015445824,
      "grad_norm": 0.2616323232650757,
      "learning_rate": 6.927665689871026e-05,
      "loss": 0.0616,
      "step": 24320
    },
    {
      "epoch": 2.8457758015445824,
      "loss_main": 0.08870985358953476,
      "step": 24320
    },
    {
      "epoch": 2.846945939620875,
      "grad_norm": 0.26182273030281067,
      "learning_rate": 6.925122652773253e-05,
      "loss": 0.0684,
      "step": 24330
    },
    {
      "epoch": 2.846945939620875,
      "loss_main": 0.04907732829451561,
      "step": 24330
    },
    {
      "epoch": 2.8481160776971683,
      "grad_norm": 0.2388707548379898,
      "learning_rate": 6.922579030874046e-05,
      "loss": 0.0706,
      "step": 24340
    },
    {
      "epoch": 2.8481160776971683,
      "loss_main": 0.08056443929672241,
      "step": 24340
    },
    {
      "epoch": 2.8492862157734615,
      "grad_norm": 0.377887099981308,
      "learning_rate": 6.920034824946093e-05,
      "loss": 0.073,
      "step": 24350
    },
    {
      "epoch": 2.8492862157734615,
      "loss_main": 0.07236466556787491,
      "step": 24350
    },
    {
      "epoch": 2.8504563538497543,
      "grad_norm": 0.35496076941490173,
      "learning_rate": 6.917490035762255e-05,
      "loss": 0.0737,
      "step": 24360
    },
    {
      "epoch": 2.8504563538497543,
      "loss_main": 0.12170951068401337,
      "step": 24360
    },
    {
      "epoch": 2.851626491926047,
      "grad_norm": 0.23845899105072021,
      "learning_rate": 6.914944664095573e-05,
      "loss": 0.0788,
      "step": 24370
    },
    {
      "epoch": 2.851626491926047,
      "loss_main": 0.10829374939203262,
      "step": 24370
    },
    {
      "epoch": 2.85279663000234,
      "grad_norm": 0.3075593113899231,
      "learning_rate": 6.912398710719264e-05,
      "loss": 0.0741,
      "step": 24380
    },
    {
      "epoch": 2.85279663000234,
      "loss_main": 0.08141323179006577,
      "step": 24380
    },
    {
      "epoch": 2.8539667680786334,
      "grad_norm": 0.262274831533432,
      "learning_rate": 6.90985217640672e-05,
      "loss": 0.0606,
      "step": 24390
    },
    {
      "epoch": 2.8539667680786334,
      "loss_main": 0.041133470833301544,
      "step": 24390
    },
    {
      "epoch": 2.855136906154926,
      "grad_norm": 0.33367136120796204,
      "learning_rate": 6.90730506193151e-05,
      "loss": 0.0632,
      "step": 24400
    },
    {
      "epoch": 2.855136906154926,
      "loss_main": 0.048861462622880936,
      "step": 24400
    },
    {
      "epoch": 2.8563070442312193,
      "grad_norm": 0.25115546584129333,
      "learning_rate": 6.904757368067384e-05,
      "loss": 0.0666,
      "step": 24410
    },
    {
      "epoch": 2.8563070442312193,
      "loss_main": 0.05133857578039169,
      "step": 24410
    },
    {
      "epoch": 2.8574771823075125,
      "grad_norm": 0.2599662244319916,
      "learning_rate": 6.90220909558826e-05,
      "loss": 0.0743,
      "step": 24420
    },
    {
      "epoch": 2.8574771823075125,
      "loss_main": 0.05525459721684456,
      "step": 24420
    },
    {
      "epoch": 2.858647320383805,
      "grad_norm": 0.21664586663246155,
      "learning_rate": 6.899660245268237e-05,
      "loss": 0.0674,
      "step": 24430
    },
    {
      "epoch": 2.858647320383805,
      "loss_main": 0.05669227987527847,
      "step": 24430
    },
    {
      "epoch": 2.8598174584600984,
      "grad_norm": 0.2541535794734955,
      "learning_rate": 6.897110817881592e-05,
      "loss": 0.0664,
      "step": 24440
    },
    {
      "epoch": 2.8598174584600984,
      "loss_main": 0.088401660323143,
      "step": 24440
    },
    {
      "epoch": 2.8609875965363916,
      "grad_norm": 0.25891831517219543,
      "learning_rate": 6.894560814202769e-05,
      "loss": 0.0732,
      "step": 24450
    },
    {
      "epoch": 2.8609875965363916,
      "loss_main": 0.034178491681814194,
      "step": 24450
    },
    {
      "epoch": 2.8621577346126843,
      "grad_norm": 0.2306380569934845,
      "learning_rate": 6.892010235006394e-05,
      "loss": 0.0695,
      "step": 24460
    },
    {
      "epoch": 2.8621577346126843,
      "loss_main": 0.06369607895612717,
      "step": 24460
    },
    {
      "epoch": 2.863327872688977,
      "grad_norm": 0.2786383032798767,
      "learning_rate": 6.889459081067264e-05,
      "loss": 0.0621,
      "step": 24470
    },
    {
      "epoch": 2.863327872688977,
      "loss_main": 0.052605461329221725,
      "step": 24470
    },
    {
      "epoch": 2.86449801076527,
      "grad_norm": 0.2469841092824936,
      "learning_rate": 6.886907353160356e-05,
      "loss": 0.073,
      "step": 24480
    },
    {
      "epoch": 2.86449801076527,
      "loss_main": 0.09052525460720062,
      "step": 24480
    },
    {
      "epoch": 2.8656681488415634,
      "grad_norm": 0.24718107283115387,
      "learning_rate": 6.884355052060814e-05,
      "loss": 0.0693,
      "step": 24490
    },
    {
      "epoch": 2.8656681488415634,
      "loss_main": 0.04365519806742668,
      "step": 24490
    },
    {
      "epoch": 2.866838286917856,
      "grad_norm": 0.2736782431602478,
      "learning_rate": 6.88180217854396e-05,
      "loss": 0.0647,
      "step": 24500
    },
    {
      "epoch": 2.866838286917856,
      "loss_main": 0.0520983450114727,
      "step": 24500
    },
    {
      "epoch": 2.8680084249941493,
      "grad_norm": 0.28571024537086487,
      "learning_rate": 6.87924873338529e-05,
      "loss": 0.0708,
      "step": 24510
    },
    {
      "epoch": 2.8680084249941493,
      "loss_main": 0.05724068358540535,
      "step": 24510
    },
    {
      "epoch": 2.8691785630704425,
      "grad_norm": 0.3891964852809906,
      "learning_rate": 6.876694717360475e-05,
      "loss": 0.0661,
      "step": 24520
    },
    {
      "epoch": 2.8691785630704425,
      "loss_main": 0.06911448389291763,
      "step": 24520
    },
    {
      "epoch": 2.8703487011467352,
      "grad_norm": 0.26147040724754333,
      "learning_rate": 6.874140131245355e-05,
      "loss": 0.0658,
      "step": 24530
    },
    {
      "epoch": 2.8703487011467352,
      "loss_main": 0.11393686383962631,
      "step": 24530
    },
    {
      "epoch": 2.8715188392230284,
      "grad_norm": 0.2762574851512909,
      "learning_rate": 6.871584975815948e-05,
      "loss": 0.0772,
      "step": 24540
    },
    {
      "epoch": 2.8715188392230284,
      "loss_main": 0.05048954114317894,
      "step": 24540
    },
    {
      "epoch": 2.872688977299321,
      "grad_norm": 0.27883365750312805,
      "learning_rate": 6.86902925184844e-05,
      "loss": 0.069,
      "step": 24550
    },
    {
      "epoch": 2.872688977299321,
      "loss_main": 0.0234906617552042,
      "step": 24550
    },
    {
      "epoch": 2.8738591153756143,
      "grad_norm": 0.33380380272865295,
      "learning_rate": 6.866472960119195e-05,
      "loss": 0.0744,
      "step": 24560
    },
    {
      "epoch": 2.8738591153756143,
      "loss_main": 0.06514495611190796,
      "step": 24560
    },
    {
      "epoch": 2.875029253451907,
      "grad_norm": 0.4160383939743042,
      "learning_rate": 6.863916101404748e-05,
      "loss": 0.0707,
      "step": 24570
    },
    {
      "epoch": 2.875029253451907,
      "loss_main": 0.047146670520305634,
      "step": 24570
    },
    {
      "epoch": 2.8761993915282003,
      "grad_norm": 0.3321639597415924,
      "learning_rate": 6.8613586764818e-05,
      "loss": 0.0641,
      "step": 24580
    },
    {
      "epoch": 2.8761993915282003,
      "loss_main": 0.10543670505285263,
      "step": 24580
    },
    {
      "epoch": 2.8773695296044934,
      "grad_norm": 0.3300377130508423,
      "learning_rate": 6.858800686127233e-05,
      "loss": 0.0674,
      "step": 24590
    },
    {
      "epoch": 2.8773695296044934,
      "loss_main": 0.07381314784288406,
      "step": 24590
    },
    {
      "epoch": 2.878539667680786,
      "grad_norm": 0.3334120213985443,
      "learning_rate": 6.856242131118097e-05,
      "loss": 0.0661,
      "step": 24600
    },
    {
      "epoch": 2.878539667680786,
      "loss_main": 0.05463857203722,
      "step": 24600
    },
    {
      "epoch": 2.8797098057570794,
      "grad_norm": 0.2406378835439682,
      "learning_rate": 6.853683012231614e-05,
      "loss": 0.0633,
      "step": 24610
    },
    {
      "epoch": 2.8797098057570794,
      "loss_main": 0.04165752977132797,
      "step": 24610
    },
    {
      "epoch": 2.8808799438333725,
      "grad_norm": 0.29188790917396545,
      "learning_rate": 6.851123330245173e-05,
      "loss": 0.0673,
      "step": 24620
    },
    {
      "epoch": 2.8808799438333725,
      "loss_main": 0.060402072966098785,
      "step": 24620
    },
    {
      "epoch": 2.8820500819096653,
      "grad_norm": 0.4068327844142914,
      "learning_rate": 6.848563085936343e-05,
      "loss": 0.064,
      "step": 24630
    },
    {
      "epoch": 2.8820500819096653,
      "loss_main": 0.045003797858953476,
      "step": 24630
    },
    {
      "epoch": 2.8832202199859585,
      "grad_norm": 0.2605978548526764,
      "learning_rate": 6.846002280082853e-05,
      "loss": 0.0622,
      "step": 24640
    },
    {
      "epoch": 2.8832202199859585,
      "loss_main": 0.0592062808573246,
      "step": 24640
    },
    {
      "epoch": 2.884390358062251,
      "grad_norm": 0.30633461475372314,
      "learning_rate": 6.843440913462614e-05,
      "loss": 0.0641,
      "step": 24650
    },
    {
      "epoch": 2.884390358062251,
      "loss_main": 0.049662284553050995,
      "step": 24650
    },
    {
      "epoch": 2.8855604961385444,
      "grad_norm": 0.23077933490276337,
      "learning_rate": 6.840878986853698e-05,
      "loss": 0.0652,
      "step": 24660
    },
    {
      "epoch": 2.8855604961385444,
      "loss_main": 0.04318036511540413,
      "step": 24660
    },
    {
      "epoch": 2.886730634214837,
      "grad_norm": 0.23385974764823914,
      "learning_rate": 6.838316501034352e-05,
      "loss": 0.0614,
      "step": 24670
    },
    {
      "epoch": 2.886730634214837,
      "loss_main": 0.07313738763332367,
      "step": 24670
    },
    {
      "epoch": 2.8879007722911303,
      "grad_norm": 0.2876970171928406,
      "learning_rate": 6.83575345678299e-05,
      "loss": 0.0626,
      "step": 24680
    },
    {
      "epoch": 2.8879007722911303,
      "loss_main": 0.06926456838846207,
      "step": 24680
    },
    {
      "epoch": 2.8890709103674235,
      "grad_norm": 0.2703872621059418,
      "learning_rate": 6.833189854878196e-05,
      "loss": 0.0723,
      "step": 24690
    },
    {
      "epoch": 2.8890709103674235,
      "loss_main": 0.06859345734119415,
      "step": 24690
    },
    {
      "epoch": 2.890241048443716,
      "grad_norm": 0.338286817073822,
      "learning_rate": 6.83062569609873e-05,
      "loss": 0.0706,
      "step": 24700
    },
    {
      "epoch": 2.890241048443716,
      "loss_main": 0.08148805797100067,
      "step": 24700
    },
    {
      "epoch": 2.8914111865200094,
      "grad_norm": 0.28195539116859436,
      "learning_rate": 6.828060981223512e-05,
      "loss": 0.0607,
      "step": 24710
    },
    {
      "epoch": 2.8914111865200094,
      "loss_main": 0.08964027464389801,
      "step": 24710
    },
    {
      "epoch": 2.8925813245963026,
      "grad_norm": 0.3052160441875458,
      "learning_rate": 6.825495711031634e-05,
      "loss": 0.0647,
      "step": 24720
    },
    {
      "epoch": 2.8925813245963026,
      "loss_main": 0.09438025206327438,
      "step": 24720
    },
    {
      "epoch": 2.8937514626725953,
      "grad_norm": 0.2758478820323944,
      "learning_rate": 6.822929886302359e-05,
      "loss": 0.0588,
      "step": 24730
    },
    {
      "epoch": 2.8937514626725953,
      "loss_main": 0.047631118446588516,
      "step": 24730
    },
    {
      "epoch": 2.8949216007488885,
      "grad_norm": 0.3681251108646393,
      "learning_rate": 6.820363507815116e-05,
      "loss": 0.0626,
      "step": 24740
    },
    {
      "epoch": 2.8949216007488885,
      "loss_main": 0.0578465573489666,
      "step": 24740
    },
    {
      "epoch": 2.8960917388251812,
      "grad_norm": 0.3365331292152405,
      "learning_rate": 6.817796576349501e-05,
      "loss": 0.0666,
      "step": 24750
    },
    {
      "epoch": 2.8960917388251812,
      "loss_main": 0.051137976348400116,
      "step": 24750
    },
    {
      "epoch": 2.8972618769014744,
      "grad_norm": 0.26226168870925903,
      "learning_rate": 6.815229092685285e-05,
      "loss": 0.0776,
      "step": 24760
    },
    {
      "epoch": 2.8972618769014744,
      "loss_main": 0.06011441349983215,
      "step": 24760
    },
    {
      "epoch": 2.898432014977767,
      "grad_norm": 0.3327540159225464,
      "learning_rate": 6.812661057602399e-05,
      "loss": 0.0584,
      "step": 24770
    },
    {
      "epoch": 2.898432014977767,
      "loss_main": 0.05653451010584831,
      "step": 24770
    },
    {
      "epoch": 2.8996021530540603,
      "grad_norm": 0.24361401796340942,
      "learning_rate": 6.810092471880943e-05,
      "loss": 0.0636,
      "step": 24780
    },
    {
      "epoch": 2.8996021530540603,
      "loss_main": 0.10742602497339249,
      "step": 24780
    },
    {
      "epoch": 2.9007722911303535,
      "grad_norm": 0.38253864645957947,
      "learning_rate": 6.807523336301187e-05,
      "loss": 0.069,
      "step": 24790
    },
    {
      "epoch": 2.9007722911303535,
      "loss_main": 0.08124249428510666,
      "step": 24790
    },
    {
      "epoch": 2.9019424292066462,
      "grad_norm": 0.2551993131637573,
      "learning_rate": 6.804953651643566e-05,
      "loss": 0.0762,
      "step": 24800
    },
    {
      "epoch": 2.9019424292066462,
      "loss_main": 0.07940884679555893,
      "step": 24800
    },
    {
      "epoch": 2.9031125672829394,
      "grad_norm": 0.29017454385757446,
      "learning_rate": 6.802383418688685e-05,
      "loss": 0.0719,
      "step": 24810
    },
    {
      "epoch": 2.9031125672829394,
      "loss_main": 0.10302412509918213,
      "step": 24810
    },
    {
      "epoch": 2.9042827053592326,
      "grad_norm": 0.2188396006822586,
      "learning_rate": 6.799812638217309e-05,
      "loss": 0.0702,
      "step": 24820
    },
    {
      "epoch": 2.9042827053592326,
      "loss_main": 0.057304203510284424,
      "step": 24820
    },
    {
      "epoch": 2.9054528434355253,
      "grad_norm": 0.3154369294643402,
      "learning_rate": 6.797241311010373e-05,
      "loss": 0.0683,
      "step": 24830
    },
    {
      "epoch": 2.9054528434355253,
      "loss_main": 0.07930584251880646,
      "step": 24830
    },
    {
      "epoch": 2.9066229815118185,
      "grad_norm": 0.2811635136604309,
      "learning_rate": 6.794669437848982e-05,
      "loss": 0.0713,
      "step": 24840
    },
    {
      "epoch": 2.9066229815118185,
      "loss_main": 0.09254133701324463,
      "step": 24840
    },
    {
      "epoch": 2.9077931195881113,
      "grad_norm": 0.41111934185028076,
      "learning_rate": 6.792097019514402e-05,
      "loss": 0.0663,
      "step": 24850
    },
    {
      "epoch": 2.9077931195881113,
      "loss_main": 0.05817306786775589,
      "step": 24850
    },
    {
      "epoch": 2.9089632576644044,
      "grad_norm": 0.3310077488422394,
      "learning_rate": 6.789524056788064e-05,
      "loss": 0.0675,
      "step": 24860
    },
    {
      "epoch": 2.9089632576644044,
      "loss_main": 0.07931935042142868,
      "step": 24860
    },
    {
      "epoch": 2.910133395740697,
      "grad_norm": 0.19782869517803192,
      "learning_rate": 6.786950550451567e-05,
      "loss": 0.0567,
      "step": 24870
    },
    {
      "epoch": 2.910133395740697,
      "loss_main": 0.08303380012512207,
      "step": 24870
    },
    {
      "epoch": 2.9113035338169904,
      "grad_norm": 0.30190715193748474,
      "learning_rate": 6.784376501286676e-05,
      "loss": 0.0678,
      "step": 24880
    },
    {
      "epoch": 2.9113035338169904,
      "loss_main": 0.0505828857421875,
      "step": 24880
    },
    {
      "epoch": 2.9124736718932835,
      "grad_norm": 0.45097681879997253,
      "learning_rate": 6.781801910075316e-05,
      "loss": 0.0567,
      "step": 24890
    },
    {
      "epoch": 2.9124736718932835,
      "loss_main": 0.07285536825656891,
      "step": 24890
    },
    {
      "epoch": 2.9136438099695763,
      "grad_norm": 0.3048880696296692,
      "learning_rate": 6.779226777599581e-05,
      "loss": 0.069,
      "step": 24900
    },
    {
      "epoch": 2.9136438099695763,
      "loss_main": 0.05160345509648323,
      "step": 24900
    },
    {
      "epoch": 2.9148139480458695,
      "grad_norm": 0.2676968574523926,
      "learning_rate": 6.776651104641729e-05,
      "loss": 0.0725,
      "step": 24910
    },
    {
      "epoch": 2.9148139480458695,
      "loss_main": 0.058252573013305664,
      "step": 24910
    },
    {
      "epoch": 2.9159840861221626,
      "grad_norm": 0.29625025391578674,
      "learning_rate": 6.774074891984183e-05,
      "loss": 0.0655,
      "step": 24920
    },
    {
      "epoch": 2.9159840861221626,
      "loss_main": 0.0532645657658577,
      "step": 24920
    },
    {
      "epoch": 2.9171542241984554,
      "grad_norm": 0.3118816912174225,
      "learning_rate": 6.771498140409526e-05,
      "loss": 0.0697,
      "step": 24930
    },
    {
      "epoch": 2.9171542241984554,
      "loss_main": 0.06521396338939667,
      "step": 24930
    },
    {
      "epoch": 2.9183243622747486,
      "grad_norm": 0.28383246064186096,
      "learning_rate": 6.768920850700506e-05,
      "loss": 0.0637,
      "step": 24940
    },
    {
      "epoch": 2.9183243622747486,
      "loss_main": 0.08991294354200363,
      "step": 24940
    },
    {
      "epoch": 2.9194945003510413,
      "grad_norm": 0.2775345742702484,
      "learning_rate": 6.766343023640039e-05,
      "loss": 0.0704,
      "step": 24950
    },
    {
      "epoch": 2.9194945003510413,
      "loss_main": 0.04699410870671272,
      "step": 24950
    },
    {
      "epoch": 2.9206646384273345,
      "grad_norm": 0.4326521158218384,
      "learning_rate": 6.763764660011198e-05,
      "loss": 0.0637,
      "step": 24960
    },
    {
      "epoch": 2.9206646384273345,
      "loss_main": 0.06100881099700928,
      "step": 24960
    },
    {
      "epoch": 2.921834776503627,
      "grad_norm": 0.3362797200679779,
      "learning_rate": 6.761185760597223e-05,
      "loss": 0.0718,
      "step": 24970
    },
    {
      "epoch": 2.921834776503627,
      "loss_main": 0.03993046283721924,
      "step": 24970
    },
    {
      "epoch": 2.9230049145799204,
      "grad_norm": 0.3091294467449188,
      "learning_rate": 6.758606326181515e-05,
      "loss": 0.0636,
      "step": 24980
    },
    {
      "epoch": 2.9230049145799204,
      "loss_main": 0.06426219642162323,
      "step": 24980
    },
    {
      "epoch": 2.9241750526562136,
      "grad_norm": 0.3808053135871887,
      "learning_rate": 6.75602635754764e-05,
      "loss": 0.0692,
      "step": 24990
    },
    {
      "epoch": 2.9241750526562136,
      "loss_main": 0.05441160127520561,
      "step": 24990
    },
    {
      "epoch": 2.9253451907325063,
      "grad_norm": 0.3130556046962738,
      "learning_rate": 6.75344585547932e-05,
      "loss": 0.0662,
      "step": 25000
    },
    {
      "epoch": 2.9253451907325063,
      "loss_main": 0.10512929409742355,
      "step": 25000
    },
    {
      "epoch": 2.9265153288087995,
      "grad_norm": 0.438539057970047,
      "learning_rate": 6.750864820760449e-05,
      "loss": 0.0635,
      "step": 25010
    },
    {
      "epoch": 2.9265153288087995,
      "loss_main": 0.05864226073026657,
      "step": 25010
    },
    {
      "epoch": 2.9276854668850927,
      "grad_norm": 0.3278195261955261,
      "learning_rate": 6.748283254175072e-05,
      "loss": 0.0711,
      "step": 25020
    },
    {
      "epoch": 2.9276854668850927,
      "loss_main": 0.077920101583004,
      "step": 25020
    },
    {
      "epoch": 2.9288556049613854,
      "grad_norm": 0.400015264749527,
      "learning_rate": 6.745701156507404e-05,
      "loss": 0.0681,
      "step": 25030
    },
    {
      "epoch": 2.9288556049613854,
      "loss_main": 0.07379656285047531,
      "step": 25030
    },
    {
      "epoch": 2.9300257430376786,
      "grad_norm": 0.4467029273509979,
      "learning_rate": 6.743118528541818e-05,
      "loss": 0.0678,
      "step": 25040
    },
    {
      "epoch": 2.9300257430376786,
      "loss_main": 0.06308353692293167,
      "step": 25040
    },
    {
      "epoch": 2.9311958811139713,
      "grad_norm": 0.37923893332481384,
      "learning_rate": 6.740535371062846e-05,
      "loss": 0.0678,
      "step": 25050
    },
    {
      "epoch": 2.9311958811139713,
      "loss_main": 0.06868249177932739,
      "step": 25050
    },
    {
      "epoch": 2.9323660191902645,
      "grad_norm": 0.2743815779685974,
      "learning_rate": 6.737951684855185e-05,
      "loss": 0.0639,
      "step": 25060
    },
    {
      "epoch": 2.9323660191902645,
      "loss_main": 0.07134149223566055,
      "step": 25060
    },
    {
      "epoch": 2.9335361572665573,
      "grad_norm": 0.36868417263031006,
      "learning_rate": 6.735367470703691e-05,
      "loss": 0.0653,
      "step": 25070
    },
    {
      "epoch": 2.9335361572665573,
      "loss_main": 0.06370653212070465,
      "step": 25070
    },
    {
      "epoch": 2.9347062953428504,
      "grad_norm": 0.29650992155075073,
      "learning_rate": 6.732782729393379e-05,
      "loss": 0.0641,
      "step": 25080
    },
    {
      "epoch": 2.9347062953428504,
      "loss_main": 0.055727388709783554,
      "step": 25080
    },
    {
      "epoch": 2.9358764334191436,
      "grad_norm": 0.33387985825538635,
      "learning_rate": 6.730197461709425e-05,
      "loss": 0.0626,
      "step": 25090
    },
    {
      "epoch": 2.9358764334191436,
      "loss_main": 0.05704667046666145,
      "step": 25090
    },
    {
      "epoch": 2.9370465714954364,
      "grad_norm": 0.4409695863723755,
      "learning_rate": 6.727611668437164e-05,
      "loss": 0.0775,
      "step": 25100
    },
    {
      "epoch": 2.9370465714954364,
      "loss_main": 0.05185471847653389,
      "step": 25100
    },
    {
      "epoch": 2.9382167095717295,
      "grad_norm": 0.3246432840824127,
      "learning_rate": 6.725025350362094e-05,
      "loss": 0.0617,
      "step": 25110
    },
    {
      "epoch": 2.9382167095717295,
      "loss_main": 0.058130018413066864,
      "step": 25110
    },
    {
      "epoch": 2.9393868476480227,
      "grad_norm": 0.2339457869529724,
      "learning_rate": 6.72243850826987e-05,
      "loss": 0.0608,
      "step": 25120
    },
    {
      "epoch": 2.9393868476480227,
      "loss_main": 0.08209932595491409,
      "step": 25120
    },
    {
      "epoch": 2.9405569857243155,
      "grad_norm": 0.24662281572818756,
      "learning_rate": 6.719851142946305e-05,
      "loss": 0.0638,
      "step": 25130
    },
    {
      "epoch": 2.9405569857243155,
      "loss_main": 0.06881248205900192,
      "step": 25130
    },
    {
      "epoch": 2.9417271238006086,
      "grad_norm": 0.2403307557106018,
      "learning_rate": 6.717263255177372e-05,
      "loss": 0.0704,
      "step": 25140
    },
    {
      "epoch": 2.9417271238006086,
      "loss_main": 0.0666029155254364,
      "step": 25140
    },
    {
      "epoch": 2.9428972618769014,
      "grad_norm": 0.360845685005188,
      "learning_rate": 6.714674845749205e-05,
      "loss": 0.0664,
      "step": 25150
    },
    {
      "epoch": 2.9428972618769014,
      "loss_main": 0.06965409219264984,
      "step": 25150
    },
    {
      "epoch": 2.9440673999531946,
      "grad_norm": 0.2510961592197418,
      "learning_rate": 6.712085915448092e-05,
      "loss": 0.0669,
      "step": 25160
    },
    {
      "epoch": 2.9440673999531946,
      "loss_main": 0.03950171917676926,
      "step": 25160
    },
    {
      "epoch": 2.9452375380294873,
      "grad_norm": 0.26785773038864136,
      "learning_rate": 6.709496465060486e-05,
      "loss": 0.0664,
      "step": 25170
    },
    {
      "epoch": 2.9452375380294873,
      "loss_main": 0.060715191066265106,
      "step": 25170
    },
    {
      "epoch": 2.9464076761057805,
      "grad_norm": 0.27284690737724304,
      "learning_rate": 6.706906495372987e-05,
      "loss": 0.0605,
      "step": 25180
    },
    {
      "epoch": 2.9464076761057805,
      "loss_main": 0.044602103531360626,
      "step": 25180
    },
    {
      "epoch": 2.9475778141820737,
      "grad_norm": 0.28232598304748535,
      "learning_rate": 6.704316007172365e-05,
      "loss": 0.0691,
      "step": 25190
    },
    {
      "epoch": 2.9475778141820737,
      "loss_main": 0.057091712951660156,
      "step": 25190
    },
    {
      "epoch": 2.9487479522583664,
      "grad_norm": 0.26247307658195496,
      "learning_rate": 6.701725001245539e-05,
      "loss": 0.0762,
      "step": 25200
    },
    {
      "epoch": 2.9487479522583664,
      "loss_main": 0.07840069383382797,
      "step": 25200
    },
    {
      "epoch": 2.9499180903346596,
      "grad_norm": 0.27888986468315125,
      "learning_rate": 6.699133478379588e-05,
      "loss": 0.0637,
      "step": 25210
    },
    {
      "epoch": 2.9499180903346596,
      "loss_main": 0.06306816637516022,
      "step": 25210
    },
    {
      "epoch": 2.9510882284109528,
      "grad_norm": 0.3063541650772095,
      "learning_rate": 6.69654143936175e-05,
      "loss": 0.0608,
      "step": 25220
    },
    {
      "epoch": 2.9510882284109528,
      "loss_main": 0.05857056379318237,
      "step": 25220
    },
    {
      "epoch": 2.9522583664872455,
      "grad_norm": 0.30696189403533936,
      "learning_rate": 6.693948884979419e-05,
      "loss": 0.068,
      "step": 25230
    },
    {
      "epoch": 2.9522583664872455,
      "loss_main": 0.0612071231007576,
      "step": 25230
    },
    {
      "epoch": 2.9534285045635382,
      "grad_norm": 0.3136017918586731,
      "learning_rate": 6.691355816020142e-05,
      "loss": 0.0617,
      "step": 25240
    },
    {
      "epoch": 2.9534285045635382,
      "loss_main": 0.04130136966705322,
      "step": 25240
    },
    {
      "epoch": 2.9545986426398314,
      "grad_norm": 0.24870221316814423,
      "learning_rate": 6.688762233271624e-05,
      "loss": 0.0593,
      "step": 25250
    },
    {
      "epoch": 2.9545986426398314,
      "loss_main": 0.06466790288686752,
      "step": 25250
    },
    {
      "epoch": 2.9557687807161246,
      "grad_norm": 0.3256692588329315,
      "learning_rate": 6.68616813752173e-05,
      "loss": 0.0696,
      "step": 25260
    },
    {
      "epoch": 2.9557687807161246,
      "loss_main": 0.07262516021728516,
      "step": 25260
    },
    {
      "epoch": 2.9569389187924173,
      "grad_norm": 0.3650173842906952,
      "learning_rate": 6.683573529558477e-05,
      "loss": 0.0728,
      "step": 25270
    },
    {
      "epoch": 2.9569389187924173,
      "loss_main": 0.04785763472318649,
      "step": 25270
    },
    {
      "epoch": 2.9581090568687105,
      "grad_norm": 0.37481415271759033,
      "learning_rate": 6.680978410170037e-05,
      "loss": 0.0672,
      "step": 25280
    },
    {
      "epoch": 2.9581090568687105,
      "loss_main": 0.03980535268783569,
      "step": 25280
    },
    {
      "epoch": 2.9592791949450037,
      "grad_norm": 0.2990614175796509,
      "learning_rate": 6.678382780144741e-05,
      "loss": 0.0642,
      "step": 25290
    },
    {
      "epoch": 2.9592791949450037,
      "loss_main": 0.08406642824411392,
      "step": 25290
    },
    {
      "epoch": 2.9604493330212964,
      "grad_norm": 0.22011208534240723,
      "learning_rate": 6.675786640271071e-05,
      "loss": 0.0656,
      "step": 25300
    },
    {
      "epoch": 2.9604493330212964,
      "loss_main": 0.05611128732562065,
      "step": 25300
    },
    {
      "epoch": 2.9616194710975896,
      "grad_norm": 0.2449617236852646,
      "learning_rate": 6.673189991337665e-05,
      "loss": 0.0683,
      "step": 25310
    },
    {
      "epoch": 2.9616194710975896,
      "loss_main": 0.07309769839048386,
      "step": 25310
    },
    {
      "epoch": 2.962789609173883,
      "grad_norm": 0.3116636276245117,
      "learning_rate": 6.670592834133317e-05,
      "loss": 0.0709,
      "step": 25320
    },
    {
      "epoch": 2.962789609173883,
      "loss_main": 0.05692485347390175,
      "step": 25320
    },
    {
      "epoch": 2.9639597472501755,
      "grad_norm": 0.3151954412460327,
      "learning_rate": 6.667995169446979e-05,
      "loss": 0.064,
      "step": 25330
    },
    {
      "epoch": 2.9639597472501755,
      "loss_main": 0.09332867711782455,
      "step": 25330
    },
    {
      "epoch": 2.9651298853264683,
      "grad_norm": 0.22574274241924286,
      "learning_rate": 6.665396998067747e-05,
      "loss": 0.0778,
      "step": 25340
    },
    {
      "epoch": 2.9651298853264683,
      "loss_main": 0.07193031162023544,
      "step": 25340
    },
    {
      "epoch": 2.9663000234027614,
      "grad_norm": 0.2920016348361969,
      "learning_rate": 6.66279832078488e-05,
      "loss": 0.0665,
      "step": 25350
    },
    {
      "epoch": 2.9663000234027614,
      "loss_main": 0.06480129063129425,
      "step": 25350
    },
    {
      "epoch": 2.9674701614790546,
      "grad_norm": 0.31963658332824707,
      "learning_rate": 6.660199138387786e-05,
      "loss": 0.0644,
      "step": 25360
    },
    {
      "epoch": 2.9674701614790546,
      "loss_main": 0.05844356492161751,
      "step": 25360
    },
    {
      "epoch": 2.9686402995553474,
      "grad_norm": 0.2840099334716797,
      "learning_rate": 6.65759945166603e-05,
      "loss": 0.0686,
      "step": 25370
    },
    {
      "epoch": 2.9686402995553474,
      "loss_main": 0.0564645417034626,
      "step": 25370
    },
    {
      "epoch": 2.9698104376316405,
      "grad_norm": 0.4028671979904175,
      "learning_rate": 6.654999261409326e-05,
      "loss": 0.0753,
      "step": 25380
    },
    {
      "epoch": 2.9698104376316405,
      "loss_main": 0.07962082326412201,
      "step": 25380
    },
    {
      "epoch": 2.9709805757079337,
      "grad_norm": 0.24246904253959656,
      "learning_rate": 6.652398568407544e-05,
      "loss": 0.064,
      "step": 25390
    },
    {
      "epoch": 2.9709805757079337,
      "loss_main": 0.06602185219526291,
      "step": 25390
    },
    {
      "epoch": 2.9721507137842265,
      "grad_norm": 0.38009992241859436,
      "learning_rate": 6.649797373450707e-05,
      "loss": 0.0623,
      "step": 25400
    },
    {
      "epoch": 2.9721507137842265,
      "loss_main": 0.10115822404623032,
      "step": 25400
    },
    {
      "epoch": 2.9733208518605196,
      "grad_norm": 0.32263898849487305,
      "learning_rate": 6.647195677328988e-05,
      "loss": 0.061,
      "step": 25410
    },
    {
      "epoch": 2.9733208518605196,
      "loss_main": 0.061848729848861694,
      "step": 25410
    },
    {
      "epoch": 2.9744909899368124,
      "grad_norm": 0.2224603146314621,
      "learning_rate": 6.644593480832712e-05,
      "loss": 0.0647,
      "step": 25420
    },
    {
      "epoch": 2.9744909899368124,
      "loss_main": 0.058865394443273544,
      "step": 25420
    },
    {
      "epoch": 2.9756611280131056,
      "grad_norm": 0.38065922260284424,
      "learning_rate": 6.641990784752363e-05,
      "loss": 0.0657,
      "step": 25430
    },
    {
      "epoch": 2.9756611280131056,
      "loss_main": 0.07501647621393204,
      "step": 25430
    },
    {
      "epoch": 2.9768312660893983,
      "grad_norm": 0.26759374141693115,
      "learning_rate": 6.639387589878566e-05,
      "loss": 0.0727,
      "step": 25440
    },
    {
      "epoch": 2.9768312660893983,
      "loss_main": 0.03576862812042236,
      "step": 25440
    },
    {
      "epoch": 2.9780014041656915,
      "grad_norm": 0.27150627970695496,
      "learning_rate": 6.636783897002103e-05,
      "loss": 0.0655,
      "step": 25450
    },
    {
      "epoch": 2.9780014041656915,
      "loss_main": 0.08063919097185135,
      "step": 25450
    },
    {
      "epoch": 2.9791715422419847,
      "grad_norm": 0.24546074867248535,
      "learning_rate": 6.63417970691391e-05,
      "loss": 0.067,
      "step": 25460
    },
    {
      "epoch": 2.9791715422419847,
      "loss_main": 0.04895253852009773,
      "step": 25460
    },
    {
      "epoch": 2.9803416803182774,
      "grad_norm": 0.26613926887512207,
      "learning_rate": 6.63157502040507e-05,
      "loss": 0.0598,
      "step": 25470
    },
    {
      "epoch": 2.9803416803182774,
      "loss_main": 0.07032173126935959,
      "step": 25470
    },
    {
      "epoch": 2.9815118183945706,
      "grad_norm": 0.3325234353542328,
      "learning_rate": 6.628969838266819e-05,
      "loss": 0.0669,
      "step": 25480
    },
    {
      "epoch": 2.9815118183945706,
      "loss_main": 0.062489431351423264,
      "step": 25480
    },
    {
      "epoch": 2.9826819564708638,
      "grad_norm": 0.31276699900627136,
      "learning_rate": 6.626364161290541e-05,
      "loss": 0.0639,
      "step": 25490
    },
    {
      "epoch": 2.9826819564708638,
      "loss_main": 0.073361836373806,
      "step": 25490
    },
    {
      "epoch": 2.9838520945471565,
      "grad_norm": 0.275701642036438,
      "learning_rate": 6.623757990267774e-05,
      "loss": 0.0647,
      "step": 25500
    },
    {
      "epoch": 2.9838520945471565,
      "loss_main": 0.0813448578119278,
      "step": 25500
    },
    {
      "epoch": 2.9850222326234497,
      "grad_norm": 0.3077635169029236,
      "learning_rate": 6.621151325990201e-05,
      "loss": 0.0704,
      "step": 25510
    },
    {
      "epoch": 2.9850222326234497,
      "loss_main": 0.08862589299678802,
      "step": 25510
    },
    {
      "epoch": 2.9861923706997424,
      "grad_norm": 0.24590623378753662,
      "learning_rate": 6.618544169249657e-05,
      "loss": 0.0719,
      "step": 25520
    },
    {
      "epoch": 2.9861923706997424,
      "loss_main": 0.04865434393286705,
      "step": 25520
    },
    {
      "epoch": 2.9873625087760356,
      "grad_norm": 0.22856566309928894,
      "learning_rate": 6.615936520838133e-05,
      "loss": 0.0674,
      "step": 25530
    },
    {
      "epoch": 2.9873625087760356,
      "loss_main": 0.042278748005628586,
      "step": 25530
    },
    {
      "epoch": 2.9885326468523283,
      "grad_norm": 0.270879864692688,
      "learning_rate": 6.613328381547759e-05,
      "loss": 0.0582,
      "step": 25540
    },
    {
      "epoch": 2.9885326468523283,
      "loss_main": 0.0739508643746376,
      "step": 25540
    },
    {
      "epoch": 2.9897027849286215,
      "grad_norm": 0.28766289353370667,
      "learning_rate": 6.610719752170821e-05,
      "loss": 0.0897,
      "step": 25550
    },
    {
      "epoch": 2.9897027849286215,
      "loss_main": 0.05274577811360359,
      "step": 25550
    },
    {
      "epoch": 2.9908729230049147,
      "grad_norm": 0.25646093487739563,
      "learning_rate": 6.60811063349975e-05,
      "loss": 0.0609,
      "step": 25560
    },
    {
      "epoch": 2.9908729230049147,
      "loss_main": 0.06539206951856613,
      "step": 25560
    },
    {
      "epoch": 2.9920430610812074,
      "grad_norm": 0.334659606218338,
      "learning_rate": 6.605501026327127e-05,
      "loss": 0.0691,
      "step": 25570
    },
    {
      "epoch": 2.9920430610812074,
      "loss_main": 0.06328108161687851,
      "step": 25570
    },
    {
      "epoch": 2.9932131991575006,
      "grad_norm": 0.2561933994293213,
      "learning_rate": 6.602890931445685e-05,
      "loss": 0.0701,
      "step": 25580
    },
    {
      "epoch": 2.9932131991575006,
      "loss_main": 0.07651205360889435,
      "step": 25580
    },
    {
      "epoch": 2.994383337233794,
      "grad_norm": 0.33677250146865845,
      "learning_rate": 6.6002803496483e-05,
      "loss": 0.0698,
      "step": 25590
    },
    {
      "epoch": 2.994383337233794,
      "loss_main": 0.06097579747438431,
      "step": 25590
    },
    {
      "epoch": 2.9955534753100865,
      "grad_norm": 0.24492967128753662,
      "learning_rate": 6.597669281727997e-05,
      "loss": 0.0671,
      "step": 25600
    },
    {
      "epoch": 2.9955534753100865,
      "loss_main": 0.06794784963130951,
      "step": 25600
    },
    {
      "epoch": 2.9967236133863797,
      "grad_norm": 0.2895204424858093,
      "learning_rate": 6.595057728477949e-05,
      "loss": 0.071,
      "step": 25610
    },
    {
      "epoch": 2.9967236133863797,
      "loss_main": 0.06295377761125565,
      "step": 25610
    },
    {
      "epoch": 2.9978937514626725,
      "grad_norm": 0.2815292775630951,
      "learning_rate": 6.59244569069148e-05,
      "loss": 0.0715,
      "step": 25620
    },
    {
      "epoch": 2.9978937514626725,
      "loss_main": 0.07859107106924057,
      "step": 25620
    },
    {
      "epoch": 2.9990638895389656,
      "grad_norm": 0.2285180687904358,
      "learning_rate": 6.589833169162054e-05,
      "loss": 0.0648,
      "step": 25630
    },
    {
      "epoch": 2.9990638895389656,
      "loss_main": 0.06517436355352402,
      "step": 25630
    },
    {
      "epoch": 3.000234027615259,
      "grad_norm": 0.3766729533672333,
      "learning_rate": 6.587220164683291e-05,
      "loss": 0.0718,
      "step": 25640
    },
    {
      "epoch": 3.000234027615259,
      "loss_main": 0.09293246269226074,
      "step": 25640
    },
    {
      "epoch": 3.0014041656915516,
      "grad_norm": 0.2754901945590973,
      "learning_rate": 6.58460667804895e-05,
      "loss": 0.0668,
      "step": 25650
    },
    {
      "epoch": 3.0014041656915516,
      "loss_main": 0.07405946403741837,
      "step": 25650
    },
    {
      "epoch": 3.0025743037678447,
      "grad_norm": 0.21927006542682648,
      "learning_rate": 6.581992710052938e-05,
      "loss": 0.0755,
      "step": 25660
    },
    {
      "epoch": 3.0025743037678447,
      "loss_main": 0.07748489826917648,
      "step": 25660
    },
    {
      "epoch": 3.0037444418441375,
      "grad_norm": 0.3917349874973297,
      "learning_rate": 6.579378261489311e-05,
      "loss": 0.0726,
      "step": 25670
    },
    {
      "epoch": 3.0037444418441375,
      "loss_main": 0.08715168386697769,
      "step": 25670
    },
    {
      "epoch": 3.0049145799204307,
      "grad_norm": 0.22466638684272766,
      "learning_rate": 6.576763333152268e-05,
      "loss": 0.0709,
      "step": 25680
    },
    {
      "epoch": 3.0049145799204307,
      "loss_main": 0.06786797195672989,
      "step": 25680
    },
    {
      "epoch": 3.006084717996724,
      "grad_norm": 0.23888520896434784,
      "learning_rate": 6.574147925836159e-05,
      "loss": 0.0635,
      "step": 25690
    },
    {
      "epoch": 3.006084717996724,
      "loss_main": 0.09395115822553635,
      "step": 25690
    },
    {
      "epoch": 3.0072548560730166,
      "grad_norm": 0.3416005074977875,
      "learning_rate": 6.571532040335472e-05,
      "loss": 0.0683,
      "step": 25700
    },
    {
      "epoch": 3.0072548560730166,
      "loss_main": 0.0665402039885521,
      "step": 25700
    },
    {
      "epoch": 3.0084249941493098,
      "grad_norm": 0.30444851517677307,
      "learning_rate": 6.568915677444845e-05,
      "loss": 0.0661,
      "step": 25710
    },
    {
      "epoch": 3.0084249941493098,
      "loss_main": 0.07280044257640839,
      "step": 25710
    },
    {
      "epoch": 3.0095951322256025,
      "grad_norm": 0.2554144859313965,
      "learning_rate": 6.56629883795906e-05,
      "loss": 0.0793,
      "step": 25720
    },
    {
      "epoch": 3.0095951322256025,
      "loss_main": 0.07128618657588959,
      "step": 25720
    },
    {
      "epoch": 3.0107652703018957,
      "grad_norm": 0.33005431294441223,
      "learning_rate": 6.563681522673043e-05,
      "loss": 0.0668,
      "step": 25730
    },
    {
      "epoch": 3.0107652703018957,
      "loss_main": 0.09446149319410324,
      "step": 25730
    },
    {
      "epoch": 3.0119354083781884,
      "grad_norm": 0.2556845545768738,
      "learning_rate": 6.561063732381867e-05,
      "loss": 0.0715,
      "step": 25740
    },
    {
      "epoch": 3.0119354083781884,
      "loss_main": 0.060833048075437546,
      "step": 25740
    },
    {
      "epoch": 3.0131055464544816,
      "grad_norm": 0.30796563625335693,
      "learning_rate": 6.558445467880745e-05,
      "loss": 0.0653,
      "step": 25750
    },
    {
      "epoch": 3.0131055464544816,
      "loss_main": 0.07310247421264648,
      "step": 25750
    },
    {
      "epoch": 3.0142756845307748,
      "grad_norm": 0.2962873876094818,
      "learning_rate": 6.55582672996504e-05,
      "loss": 0.0722,
      "step": 25760
    },
    {
      "epoch": 3.0142756845307748,
      "loss_main": 0.08420991897583008,
      "step": 25760
    },
    {
      "epoch": 3.0154458226070675,
      "grad_norm": 0.3097603917121887,
      "learning_rate": 6.553207519430253e-05,
      "loss": 0.0611,
      "step": 25770
    },
    {
      "epoch": 3.0154458226070675,
      "loss_main": 0.06786467134952545,
      "step": 25770
    },
    {
      "epoch": 3.0166159606833607,
      "grad_norm": 0.33683347702026367,
      "learning_rate": 6.550587837072032e-05,
      "loss": 0.0676,
      "step": 25780
    },
    {
      "epoch": 3.0166159606833607,
      "loss_main": 0.08489877730607986,
      "step": 25780
    },
    {
      "epoch": 3.0177860987596534,
      "grad_norm": 0.2439068853855133,
      "learning_rate": 6.547967683686166e-05,
      "loss": 0.0722,
      "step": 25790
    },
    {
      "epoch": 3.0177860987596534,
      "loss_main": 0.06440631300210953,
      "step": 25790
    },
    {
      "epoch": 3.0189562368359466,
      "grad_norm": 0.26830771565437317,
      "learning_rate": 6.545347060068591e-05,
      "loss": 0.0586,
      "step": 25800
    },
    {
      "epoch": 3.0189562368359466,
      "loss_main": 0.05704784393310547,
      "step": 25800
    },
    {
      "epoch": 3.02012637491224,
      "grad_norm": 0.26065531373023987,
      "learning_rate": 6.542725967015382e-05,
      "loss": 0.0677,
      "step": 25810
    },
    {
      "epoch": 3.02012637491224,
      "loss_main": 0.046436164528131485,
      "step": 25810
    },
    {
      "epoch": 3.0212965129885325,
      "grad_norm": 0.3194819390773773,
      "learning_rate": 6.540104405322757e-05,
      "loss": 0.0658,
      "step": 25820
    },
    {
      "epoch": 3.0212965129885325,
      "loss_main": 0.04324209317564964,
      "step": 25820
    },
    {
      "epoch": 3.0224666510648257,
      "grad_norm": 0.26575136184692383,
      "learning_rate": 6.537482375787077e-05,
      "loss": 0.0707,
      "step": 25830
    },
    {
      "epoch": 3.0224666510648257,
      "loss_main": 0.05372396111488342,
      "step": 25830
    },
    {
      "epoch": 3.0236367891411184,
      "grad_norm": 0.2834387421607971,
      "learning_rate": 6.534859879204845e-05,
      "loss": 0.0686,
      "step": 25840
    },
    {
      "epoch": 3.0236367891411184,
      "loss_main": 0.10194220393896103,
      "step": 25840
    },
    {
      "epoch": 3.0248069272174116,
      "grad_norm": 0.2760162055492401,
      "learning_rate": 6.532236916372709e-05,
      "loss": 0.0697,
      "step": 25850
    },
    {
      "epoch": 3.0248069272174116,
      "loss_main": 0.0810077041387558,
      "step": 25850
    },
    {
      "epoch": 3.025977065293705,
      "grad_norm": 0.3359869122505188,
      "learning_rate": 6.529613488087454e-05,
      "loss": 0.0655,
      "step": 25860
    },
    {
      "epoch": 3.025977065293705,
      "loss_main": 0.046043865382671356,
      "step": 25860
    },
    {
      "epoch": 3.0271472033699975,
      "grad_norm": 0.26699766516685486,
      "learning_rate": 6.526989595146009e-05,
      "loss": 0.0623,
      "step": 25870
    },
    {
      "epoch": 3.0271472033699975,
      "loss_main": 0.06361573189496994,
      "step": 25870
    },
    {
      "epoch": 3.0283173414462907,
      "grad_norm": 0.26242029666900635,
      "learning_rate": 6.524365238345441e-05,
      "loss": 0.0674,
      "step": 25880
    },
    {
      "epoch": 3.0283173414462907,
      "loss_main": 0.09861474484205246,
      "step": 25880
    },
    {
      "epoch": 3.0294874795225835,
      "grad_norm": 0.33583706617355347,
      "learning_rate": 6.521740418482964e-05,
      "loss": 0.0703,
      "step": 25890
    },
    {
      "epoch": 3.0294874795225835,
      "loss_main": 0.07058510929346085,
      "step": 25890
    },
    {
      "epoch": 3.0306576175988766,
      "grad_norm": 0.2219882309436798,
      "learning_rate": 6.519115136355925e-05,
      "loss": 0.0649,
      "step": 25900
    },
    {
      "epoch": 3.0306576175988766,
      "loss_main": 0.09917240589857101,
      "step": 25900
    },
    {
      "epoch": 3.03182775567517,
      "grad_norm": 0.3586616814136505,
      "learning_rate": 6.51648939276182e-05,
      "loss": 0.0673,
      "step": 25910
    },
    {
      "epoch": 3.03182775567517,
      "loss_main": 0.07611958682537079,
      "step": 25910
    },
    {
      "epoch": 3.0329978937514626,
      "grad_norm": 0.4368586838245392,
      "learning_rate": 6.513863188498277e-05,
      "loss": 0.0669,
      "step": 25920
    },
    {
      "epoch": 3.0329978937514626,
      "loss_main": 0.05759270116686821,
      "step": 25920
    },
    {
      "epoch": 3.0341680318277557,
      "grad_norm": 0.24363991618156433,
      "learning_rate": 6.511236524363068e-05,
      "loss": 0.0627,
      "step": 25930
    },
    {
      "epoch": 3.0341680318277557,
      "loss_main": 0.0688251405954361,
      "step": 25930
    },
    {
      "epoch": 3.0353381699040485,
      "grad_norm": 0.3027186095714569,
      "learning_rate": 6.508609401154104e-05,
      "loss": 0.0711,
      "step": 25940
    },
    {
      "epoch": 3.0353381699040485,
      "loss_main": 0.05095406621694565,
      "step": 25940
    },
    {
      "epoch": 3.0365083079803417,
      "grad_norm": 0.3606421649456024,
      "learning_rate": 6.505981819669439e-05,
      "loss": 0.0625,
      "step": 25950
    },
    {
      "epoch": 3.0365083079803417,
      "loss_main": 0.09343210607767105,
      "step": 25950
    },
    {
      "epoch": 3.037678446056635,
      "grad_norm": 0.2896195650100708,
      "learning_rate": 6.503353780707258e-05,
      "loss": 0.0689,
      "step": 25960
    },
    {
      "epoch": 3.037678446056635,
      "loss_main": 0.04245223477482796,
      "step": 25960
    },
    {
      "epoch": 3.0388485841329276,
      "grad_norm": 0.3228442072868347,
      "learning_rate": 6.500725285065895e-05,
      "loss": 0.0639,
      "step": 25970
    },
    {
      "epoch": 3.0388485841329276,
      "loss_main": 0.08983087539672852,
      "step": 25970
    },
    {
      "epoch": 3.0400187222092208,
      "grad_norm": 0.3341597616672516,
      "learning_rate": 6.498096333543813e-05,
      "loss": 0.0717,
      "step": 25980
    },
    {
      "epoch": 3.0400187222092208,
      "loss_main": 0.06198746711015701,
      "step": 25980
    },
    {
      "epoch": 3.0411888602855135,
      "grad_norm": 0.40168190002441406,
      "learning_rate": 6.49546692693962e-05,
      "loss": 0.0621,
      "step": 25990
    },
    {
      "epoch": 3.0411888602855135,
      "loss_main": 0.06661628186702728,
      "step": 25990
    },
    {
      "epoch": 3.0423589983618067,
      "grad_norm": 0.26254451274871826,
      "learning_rate": 6.492837066052059e-05,
      "loss": 0.0665,
      "step": 26000
    },
    {
      "epoch": 3.0423589983618067,
      "loss_main": 0.05678952485322952,
      "step": 26000
    },
    {
      "epoch": 3.0435291364381,
      "grad_norm": 0.2543646991252899,
      "learning_rate": 6.490206751680014e-05,
      "loss": 0.0704,
      "step": 26010
    },
    {
      "epoch": 3.0435291364381,
      "loss_main": 0.04412988945841789,
      "step": 26010
    },
    {
      "epoch": 3.0446992745143926,
      "grad_norm": 0.28246748447418213,
      "learning_rate": 6.487575984622505e-05,
      "loss": 0.0651,
      "step": 26020
    },
    {
      "epoch": 3.0446992745143926,
      "loss_main": 0.07664685696363449,
      "step": 26020
    },
    {
      "epoch": 3.0458694125906858,
      "grad_norm": 0.3993590772151947,
      "learning_rate": 6.484944765678689e-05,
      "loss": 0.0672,
      "step": 26030
    },
    {
      "epoch": 3.0458694125906858,
      "loss_main": 0.059010181576013565,
      "step": 26030
    },
    {
      "epoch": 3.0470395506669785,
      "grad_norm": 0.2746794819831848,
      "learning_rate": 6.482313095647861e-05,
      "loss": 0.0643,
      "step": 26040
    },
    {
      "epoch": 3.0470395506669785,
      "loss_main": 0.04424596577882767,
      "step": 26040
    },
    {
      "epoch": 3.0482096887432717,
      "grad_norm": 0.28463563323020935,
      "learning_rate": 6.479680975329451e-05,
      "loss": 0.0646,
      "step": 26050
    },
    {
      "epoch": 3.0482096887432717,
      "loss_main": 0.10132437199354172,
      "step": 26050
    },
    {
      "epoch": 3.049379826819565,
      "grad_norm": 0.31822749972343445,
      "learning_rate": 6.477048405523031e-05,
      "loss": 0.0648,
      "step": 26060
    },
    {
      "epoch": 3.049379826819565,
      "loss_main": 0.04784448817372322,
      "step": 26060
    },
    {
      "epoch": 3.0505499648958576,
      "grad_norm": 0.20979823172092438,
      "learning_rate": 6.474415387028304e-05,
      "loss": 0.0615,
      "step": 26070
    },
    {
      "epoch": 3.0505499648958576,
      "loss_main": 0.04080828279256821,
      "step": 26070
    },
    {
      "epoch": 3.051720102972151,
      "grad_norm": 0.2486274540424347,
      "learning_rate": 6.471781920645114e-05,
      "loss": 0.0568,
      "step": 26080
    },
    {
      "epoch": 3.051720102972151,
      "loss_main": 0.05499781295657158,
      "step": 26080
    },
    {
      "epoch": 3.0528902410484435,
      "grad_norm": 0.29795220494270325,
      "learning_rate": 6.469148007173434e-05,
      "loss": 0.0661,
      "step": 26090
    },
    {
      "epoch": 3.0528902410484435,
      "loss_main": 0.05028173699975014,
      "step": 26090
    },
    {
      "epoch": 3.0540603791247367,
      "grad_norm": 0.2987719774246216,
      "learning_rate": 6.466513647413381e-05,
      "loss": 0.0656,
      "step": 26100
    },
    {
      "epoch": 3.0540603791247367,
      "loss_main": 0.07099272310733795,
      "step": 26100
    },
    {
      "epoch": 3.05523051720103,
      "grad_norm": 0.3051183223724365,
      "learning_rate": 6.463878842165203e-05,
      "loss": 0.065,
      "step": 26110
    },
    {
      "epoch": 3.05523051720103,
      "loss_main": 0.10223642736673355,
      "step": 26110
    },
    {
      "epoch": 3.0564006552773226,
      "grad_norm": 0.19416384398937225,
      "learning_rate": 6.461243592229286e-05,
      "loss": 0.0695,
      "step": 26120
    },
    {
      "epoch": 3.0564006552773226,
      "loss_main": 0.07583516836166382,
      "step": 26120
    },
    {
      "epoch": 3.057570793353616,
      "grad_norm": 0.24582836031913757,
      "learning_rate": 6.458607898406146e-05,
      "loss": 0.0622,
      "step": 26130
    },
    {
      "epoch": 3.057570793353616,
      "loss_main": 0.10638216882944107,
      "step": 26130
    },
    {
      "epoch": 3.0587409314299085,
      "grad_norm": 0.2859598398208618,
      "learning_rate": 6.455971761496439e-05,
      "loss": 0.0649,
      "step": 26140
    },
    {
      "epoch": 3.0587409314299085,
      "loss_main": 0.05232568457722664,
      "step": 26140
    },
    {
      "epoch": 3.0599110695062017,
      "grad_norm": 0.26982465386390686,
      "learning_rate": 6.453335182300953e-05,
      "loss": 0.0585,
      "step": 26150
    },
    {
      "epoch": 3.0599110695062017,
      "loss_main": 0.04324102774262428,
      "step": 26150
    },
    {
      "epoch": 3.061081207582495,
      "grad_norm": 0.2575530707836151,
      "learning_rate": 6.450698161620612e-05,
      "loss": 0.06,
      "step": 26160
    },
    {
      "epoch": 3.061081207582495,
      "loss_main": 0.04917741194367409,
      "step": 26160
    },
    {
      "epoch": 3.0622513456587876,
      "grad_norm": 0.375825971364975,
      "learning_rate": 6.448060700256473e-05,
      "loss": 0.0754,
      "step": 26170
    },
    {
      "epoch": 3.0622513456587876,
      "loss_main": 0.068979911506176,
      "step": 26170
    },
    {
      "epoch": 3.063421483735081,
      "grad_norm": 0.21451206505298615,
      "learning_rate": 6.445422799009726e-05,
      "loss": 0.0593,
      "step": 26180
    },
    {
      "epoch": 3.063421483735081,
      "loss_main": 0.06332278996706009,
      "step": 26180
    },
    {
      "epoch": 3.0645916218113736,
      "grad_norm": 0.3862006664276123,
      "learning_rate": 6.442784458681699e-05,
      "loss": 0.0731,
      "step": 26190
    },
    {
      "epoch": 3.0645916218113736,
      "loss_main": 0.04639187827706337,
      "step": 26190
    },
    {
      "epoch": 3.0657617598876667,
      "grad_norm": 0.3707364797592163,
      "learning_rate": 6.440145680073847e-05,
      "loss": 0.0622,
      "step": 26200
    },
    {
      "epoch": 3.0657617598876667,
      "loss_main": 0.059345655143260956,
      "step": 26200
    },
    {
      "epoch": 3.06693189796396,
      "grad_norm": 0.28423917293548584,
      "learning_rate": 6.437506463987762e-05,
      "loss": 0.072,
      "step": 26210
    },
    {
      "epoch": 3.06693189796396,
      "loss_main": 0.0617571584880352,
      "step": 26210
    },
    {
      "epoch": 3.0681020360402527,
      "grad_norm": 0.3263246417045593,
      "learning_rate": 6.434866811225168e-05,
      "loss": 0.0613,
      "step": 26220
    },
    {
      "epoch": 3.0681020360402527,
      "loss_main": 0.06998032331466675,
      "step": 26220
    },
    {
      "epoch": 3.069272174116546,
      "grad_norm": 0.32940995693206787,
      "learning_rate": 6.432226722587923e-05,
      "loss": 0.0691,
      "step": 26230
    },
    {
      "epoch": 3.069272174116546,
      "loss_main": 0.07005592435598373,
      "step": 26230
    },
    {
      "epoch": 3.0704423121928386,
      "grad_norm": 0.3341783881187439,
      "learning_rate": 6.429586198878015e-05,
      "loss": 0.0638,
      "step": 26240
    },
    {
      "epoch": 3.0704423121928386,
      "loss_main": 0.13070860505104065,
      "step": 26240
    },
    {
      "epoch": 3.0716124502691318,
      "grad_norm": 0.2549991309642792,
      "learning_rate": 6.426945240897566e-05,
      "loss": 0.0683,
      "step": 26250
    },
    {
      "epoch": 3.0716124502691318,
      "loss_main": 0.08859197050333023,
      "step": 26250
    },
    {
      "epoch": 3.072782588345425,
      "grad_norm": 0.25445279479026794,
      "learning_rate": 6.424303849448829e-05,
      "loss": 0.0651,
      "step": 26260
    },
    {
      "epoch": 3.072782588345425,
      "loss_main": 0.046258099377155304,
      "step": 26260
    },
    {
      "epoch": 3.0739527264217177,
      "grad_norm": 0.2877016067504883,
      "learning_rate": 6.42166202533419e-05,
      "loss": 0.065,
      "step": 26270
    },
    {
      "epoch": 3.0739527264217177,
      "loss_main": 0.06034250929951668,
      "step": 26270
    },
    {
      "epoch": 3.075122864498011,
      "grad_norm": 0.32856670022010803,
      "learning_rate": 6.419019769356164e-05,
      "loss": 0.0662,
      "step": 26280
    },
    {
      "epoch": 3.075122864498011,
      "loss_main": 0.06442330777645111,
      "step": 26280
    },
    {
      "epoch": 3.0762930025743036,
      "grad_norm": 0.2525203824043274,
      "learning_rate": 6.416377082317398e-05,
      "loss": 0.0667,
      "step": 26290
    },
    {
      "epoch": 3.0762930025743036,
      "loss_main": 0.06908677518367767,
      "step": 26290
    },
    {
      "epoch": 3.077463140650597,
      "grad_norm": 0.41992878913879395,
      "learning_rate": 6.413733965020674e-05,
      "loss": 0.0653,
      "step": 26300
    },
    {
      "epoch": 3.077463140650597,
      "loss_main": 0.065926693379879,
      "step": 26300
    },
    {
      "epoch": 3.07863327872689,
      "grad_norm": 0.34306663274765015,
      "learning_rate": 6.411090418268896e-05,
      "loss": 0.0584,
      "step": 26310
    },
    {
      "epoch": 3.07863327872689,
      "loss_main": 0.041859813034534454,
      "step": 26310
    },
    {
      "epoch": 3.0798034168031827,
      "grad_norm": 0.3078702390193939,
      "learning_rate": 6.408446442865109e-05,
      "loss": 0.0583,
      "step": 26320
    },
    {
      "epoch": 3.0798034168031827,
      "loss_main": 0.0614001490175724,
      "step": 26320
    },
    {
      "epoch": 3.080973554879476,
      "grad_norm": 0.40066367387771606,
      "learning_rate": 6.405802039612479e-05,
      "loss": 0.0657,
      "step": 26330
    },
    {
      "epoch": 3.080973554879476,
      "loss_main": 0.05178660899400711,
      "step": 26330
    },
    {
      "epoch": 3.0821436929557686,
      "grad_norm": 0.3181554675102234,
      "learning_rate": 6.403157209314308e-05,
      "loss": 0.0674,
      "step": 26340
    },
    {
      "epoch": 3.0821436929557686,
      "loss_main": 0.05683857202529907,
      "step": 26340
    },
    {
      "epoch": 3.083313831032062,
      "grad_norm": 0.2853957712650299,
      "learning_rate": 6.400511952774024e-05,
      "loss": 0.0676,
      "step": 26350
    },
    {
      "epoch": 3.083313831032062,
      "loss_main": 0.06683744490146637,
      "step": 26350
    },
    {
      "epoch": 3.084483969108355,
      "grad_norm": 0.5385624766349792,
      "learning_rate": 6.397866270795187e-05,
      "loss": 0.0687,
      "step": 26360
    },
    {
      "epoch": 3.084483969108355,
      "loss_main": 0.04516470059752464,
      "step": 26360
    },
    {
      "epoch": 3.0856541071846477,
      "grad_norm": 0.262287974357605,
      "learning_rate": 6.395220164181489e-05,
      "loss": 0.052,
      "step": 26370
    },
    {
      "epoch": 3.0856541071846477,
      "loss_main": 0.08714358508586884,
      "step": 26370
    },
    {
      "epoch": 3.086824245260941,
      "grad_norm": 0.23805804550647736,
      "learning_rate": 6.39257363373674e-05,
      "loss": 0.063,
      "step": 26380
    },
    {
      "epoch": 3.086824245260941,
      "loss_main": 0.07622243463993073,
      "step": 26380
    },
    {
      "epoch": 3.0879943833372336,
      "grad_norm": 0.23756499588489532,
      "learning_rate": 6.389926680264892e-05,
      "loss": 0.0657,
      "step": 26390
    },
    {
      "epoch": 3.0879943833372336,
      "loss_main": 0.060498062521219254,
      "step": 26390
    },
    {
      "epoch": 3.089164521413527,
      "grad_norm": 0.3023475408554077,
      "learning_rate": 6.387279304570017e-05,
      "loss": 0.0762,
      "step": 26400
    },
    {
      "epoch": 3.089164521413527,
      "loss_main": 0.04285474866628647,
      "step": 26400
    },
    {
      "epoch": 3.09033465948982,
      "grad_norm": 0.27250465750694275,
      "learning_rate": 6.384631507456319e-05,
      "loss": 0.0598,
      "step": 26410
    },
    {
      "epoch": 3.09033465948982,
      "loss_main": 0.07312403619289398,
      "step": 26410
    },
    {
      "epoch": 3.0915047975661127,
      "grad_norm": 0.2608753740787506,
      "learning_rate": 6.381983289728126e-05,
      "loss": 0.067,
      "step": 26420
    },
    {
      "epoch": 3.0915047975661127,
      "loss_main": 0.07215937227010727,
      "step": 26420
    },
    {
      "epoch": 3.092674935642406,
      "grad_norm": 0.2966424822807312,
      "learning_rate": 6.3793346521899e-05,
      "loss": 0.0694,
      "step": 26430
    },
    {
      "epoch": 3.092674935642406,
      "loss_main": 0.06160365790128708,
      "step": 26430
    },
    {
      "epoch": 3.0938450737186987,
      "grad_norm": 0.338503897190094,
      "learning_rate": 6.376685595646226e-05,
      "loss": 0.0721,
      "step": 26440
    },
    {
      "epoch": 3.0938450737186987,
      "loss_main": 0.06185328960418701,
      "step": 26440
    },
    {
      "epoch": 3.095015211794992,
      "grad_norm": 0.3489280045032501,
      "learning_rate": 6.374036120901816e-05,
      "loss": 0.0726,
      "step": 26450
    },
    {
      "epoch": 3.095015211794992,
      "loss_main": 0.1094575747847557,
      "step": 26450
    },
    {
      "epoch": 3.096185349871285,
      "grad_norm": 0.26758909225463867,
      "learning_rate": 6.371386228761514e-05,
      "loss": 0.0618,
      "step": 26460
    },
    {
      "epoch": 3.096185349871285,
      "loss_main": 0.06428767740726471,
      "step": 26460
    },
    {
      "epoch": 3.0973554879475778,
      "grad_norm": 0.2690761983394623,
      "learning_rate": 6.368735920030283e-05,
      "loss": 0.0736,
      "step": 26470
    },
    {
      "epoch": 3.0973554879475778,
      "loss_main": 0.0409223772585392,
      "step": 26470
    },
    {
      "epoch": 3.098525626023871,
      "grad_norm": 0.22543729841709137,
      "learning_rate": 6.366085195513218e-05,
      "loss": 0.0549,
      "step": 26480
    },
    {
      "epoch": 3.098525626023871,
      "loss_main": 0.055967867374420166,
      "step": 26480
    },
    {
      "epoch": 3.0996957641001637,
      "grad_norm": 0.3430536687374115,
      "learning_rate": 6.363434056015543e-05,
      "loss": 0.0748,
      "step": 26490
    },
    {
      "epoch": 3.0996957641001637,
      "loss_main": 0.04178812727332115,
      "step": 26490
    },
    {
      "epoch": 3.100865902176457,
      "grad_norm": 0.3796953558921814,
      "learning_rate": 6.360782502342599e-05,
      "loss": 0.0618,
      "step": 26500
    },
    {
      "epoch": 3.100865902176457,
      "loss_main": 0.06862372905015945,
      "step": 26500
    },
    {
      "epoch": 3.10203604025275,
      "grad_norm": 0.34248101711273193,
      "learning_rate": 6.358130535299862e-05,
      "loss": 0.0751,
      "step": 26510
    },
    {
      "epoch": 3.10203604025275,
      "loss_main": 0.0668548047542572,
      "step": 26510
    },
    {
      "epoch": 3.1032061783290428,
      "grad_norm": 0.37277841567993164,
      "learning_rate": 6.355478155692926e-05,
      "loss": 0.0665,
      "step": 26520
    },
    {
      "epoch": 3.1032061783290428,
      "loss_main": 0.07236242294311523,
      "step": 26520
    },
    {
      "epoch": 3.104376316405336,
      "grad_norm": 0.3231741189956665,
      "learning_rate": 6.352825364327517e-05,
      "loss": 0.0651,
      "step": 26530
    },
    {
      "epoch": 3.104376316405336,
      "loss_main": 0.10224004089832306,
      "step": 26530
    },
    {
      "epoch": 3.1055464544816287,
      "grad_norm": 0.367180734872818,
      "learning_rate": 6.350172162009482e-05,
      "loss": 0.0766,
      "step": 26540
    },
    {
      "epoch": 3.1055464544816287,
      "loss_main": 0.06707243621349335,
      "step": 26540
    },
    {
      "epoch": 3.106716592557922,
      "grad_norm": 0.30007997155189514,
      "learning_rate": 6.347518549544793e-05,
      "loss": 0.075,
      "step": 26550
    },
    {
      "epoch": 3.106716592557922,
      "loss_main": 0.08121461421251297,
      "step": 26550
    },
    {
      "epoch": 3.107886730634215,
      "grad_norm": 0.2688254415988922,
      "learning_rate": 6.344864527739547e-05,
      "loss": 0.0627,
      "step": 26560
    },
    {
      "epoch": 3.107886730634215,
      "loss_main": 0.11131589859724045,
      "step": 26560
    },
    {
      "epoch": 3.109056868710508,
      "grad_norm": 0.5400751233100891,
      "learning_rate": 6.342210097399966e-05,
      "loss": 0.0694,
      "step": 26570
    },
    {
      "epoch": 3.109056868710508,
      "loss_main": 0.06293971836566925,
      "step": 26570
    },
    {
      "epoch": 3.110227006786801,
      "grad_norm": 0.2468269020318985,
      "learning_rate": 6.339555259332398e-05,
      "loss": 0.0683,
      "step": 26580
    },
    {
      "epoch": 3.110227006786801,
      "loss_main": 0.06268011778593063,
      "step": 26580
    },
    {
      "epoch": 3.1113971448630937,
      "grad_norm": 0.37809988856315613,
      "learning_rate": 6.33690001434331e-05,
      "loss": 0.0728,
      "step": 26590
    },
    {
      "epoch": 3.1113971448630937,
      "loss_main": 0.09699901193380356,
      "step": 26590
    },
    {
      "epoch": 3.112567282939387,
      "grad_norm": 0.3572410047054291,
      "learning_rate": 6.334244363239296e-05,
      "loss": 0.0663,
      "step": 26600
    },
    {
      "epoch": 3.112567282939387,
      "loss_main": 0.04064834862947464,
      "step": 26600
    },
    {
      "epoch": 3.1137374210156796,
      "grad_norm": 0.31574109196662903,
      "learning_rate": 6.331588306827073e-05,
      "loss": 0.0589,
      "step": 26610
    },
    {
      "epoch": 3.1137374210156796,
      "loss_main": 0.04215506836771965,
      "step": 26610
    },
    {
      "epoch": 3.114907559091973,
      "grad_norm": 0.2797394096851349,
      "learning_rate": 6.328931845913483e-05,
      "loss": 0.066,
      "step": 26620
    },
    {
      "epoch": 3.114907559091973,
      "loss_main": 0.06790484488010406,
      "step": 26620
    },
    {
      "epoch": 3.116077697168266,
      "grad_norm": 0.3205795884132385,
      "learning_rate": 6.326274981305484e-05,
      "loss": 0.0605,
      "step": 26630
    },
    {
      "epoch": 3.116077697168266,
      "loss_main": 0.08284975588321686,
      "step": 26630
    },
    {
      "epoch": 3.1172478352445587,
      "grad_norm": 0.35351845622062683,
      "learning_rate": 6.323617713810166e-05,
      "loss": 0.081,
      "step": 26640
    },
    {
      "epoch": 3.1172478352445587,
      "loss_main": 0.058661215007305145,
      "step": 26640
    },
    {
      "epoch": 3.118417973320852,
      "grad_norm": 0.3566238284111023,
      "learning_rate": 6.320960044234734e-05,
      "loss": 0.0623,
      "step": 26650
    },
    {
      "epoch": 3.118417973320852,
      "loss_main": 0.07311011850833893,
      "step": 26650
    },
    {
      "epoch": 3.119588111397145,
      "grad_norm": 0.26354876160621643,
      "learning_rate": 6.318301973386518e-05,
      "loss": 0.0786,
      "step": 26660
    },
    {
      "epoch": 3.119588111397145,
      "loss_main": 0.06509917974472046,
      "step": 26660
    },
    {
      "epoch": 3.120758249473438,
      "grad_norm": 0.34747233986854553,
      "learning_rate": 6.315643502072971e-05,
      "loss": 0.0693,
      "step": 26670
    },
    {
      "epoch": 3.120758249473438,
      "loss_main": 0.037472885102033615,
      "step": 26670
    },
    {
      "epoch": 3.121928387549731,
      "grad_norm": 0.28381821513175964,
      "learning_rate": 6.312984631101667e-05,
      "loss": 0.0657,
      "step": 26680
    },
    {
      "epoch": 3.121928387549731,
      "loss_main": 0.06776988506317139,
      "step": 26680
    },
    {
      "epoch": 3.1230985256260237,
      "grad_norm": 0.4050109088420868,
      "learning_rate": 6.310325361280297e-05,
      "loss": 0.0636,
      "step": 26690
    },
    {
      "epoch": 3.1230985256260237,
      "loss_main": 0.07074101269245148,
      "step": 26690
    },
    {
      "epoch": 3.124268663702317,
      "grad_norm": 0.40756863355636597,
      "learning_rate": 6.30766569341668e-05,
      "loss": 0.0645,
      "step": 26700
    },
    {
      "epoch": 3.124268663702317,
      "loss_main": 0.0352761372923851,
      "step": 26700
    },
    {
      "epoch": 3.1254388017786097,
      "grad_norm": 0.17772111296653748,
      "learning_rate": 6.305005628318753e-05,
      "loss": 0.0566,
      "step": 26710
    },
    {
      "epoch": 3.1254388017786097,
      "loss_main": 0.07791460305452347,
      "step": 26710
    },
    {
      "epoch": 3.126608939854903,
      "grad_norm": 0.30807217955589294,
      "learning_rate": 6.302345166794572e-05,
      "loss": 0.0594,
      "step": 26720
    },
    {
      "epoch": 3.126608939854903,
      "loss_main": 0.04716912657022476,
      "step": 26720
    },
    {
      "epoch": 3.127779077931196,
      "grad_norm": 0.1851535439491272,
      "learning_rate": 6.299684309652316e-05,
      "loss": 0.0615,
      "step": 26730
    },
    {
      "epoch": 3.127779077931196,
      "loss_main": 0.037358980625867844,
      "step": 26730
    },
    {
      "epoch": 3.1289492160074888,
      "grad_norm": 0.3203834593296051,
      "learning_rate": 6.297023057700283e-05,
      "loss": 0.0583,
      "step": 26740
    },
    {
      "epoch": 3.1289492160074888,
      "loss_main": 0.11337362229824066,
      "step": 26740
    },
    {
      "epoch": 3.130119354083782,
      "grad_norm": 0.30555465817451477,
      "learning_rate": 6.294361411746891e-05,
      "loss": 0.0632,
      "step": 26750
    },
    {
      "epoch": 3.130119354083782,
      "loss_main": 0.06508263945579529,
      "step": 26750
    },
    {
      "epoch": 3.1312894921600747,
      "grad_norm": 0.3713401257991791,
      "learning_rate": 6.291699372600677e-05,
      "loss": 0.0642,
      "step": 26760
    },
    {
      "epoch": 3.1312894921600747,
      "loss_main": 0.06183934956789017,
      "step": 26760
    },
    {
      "epoch": 3.132459630236368,
      "grad_norm": 0.3341580331325531,
      "learning_rate": 6.2890369410703e-05,
      "loss": 0.064,
      "step": 26770
    },
    {
      "epoch": 3.132459630236368,
      "loss_main": 0.0664767250418663,
      "step": 26770
    },
    {
      "epoch": 3.133629768312661,
      "grad_norm": 0.2799316644668579,
      "learning_rate": 6.286374117964534e-05,
      "loss": 0.0672,
      "step": 26780
    },
    {
      "epoch": 3.133629768312661,
      "loss_main": 0.030750708654522896,
      "step": 26780
    },
    {
      "epoch": 3.134799906388954,
      "grad_norm": 0.2690753638744354,
      "learning_rate": 6.283710904092277e-05,
      "loss": 0.0536,
      "step": 26790
    },
    {
      "epoch": 3.134799906388954,
      "loss_main": 0.07544740289449692,
      "step": 26790
    },
    {
      "epoch": 3.135970044465247,
      "grad_norm": 0.3504592180252075,
      "learning_rate": 6.281047300262542e-05,
      "loss": 0.0599,
      "step": 26800
    },
    {
      "epoch": 3.135970044465247,
      "loss_main": 0.046511221677064896,
      "step": 26800
    },
    {
      "epoch": 3.1371401825415397,
      "grad_norm": 0.2477138340473175,
      "learning_rate": 6.278383307284461e-05,
      "loss": 0.064,
      "step": 26810
    },
    {
      "epoch": 3.1371401825415397,
      "loss_main": 0.07808076590299606,
      "step": 26810
    },
    {
      "epoch": 3.138310320617833,
      "grad_norm": 0.22838856279850006,
      "learning_rate": 6.275718925967284e-05,
      "loss": 0.0683,
      "step": 26820
    },
    {
      "epoch": 3.138310320617833,
      "loss_main": 0.08100210875272751,
      "step": 26820
    },
    {
      "epoch": 3.139480458694126,
      "grad_norm": 0.29342496395111084,
      "learning_rate": 6.273054157120382e-05,
      "loss": 0.0692,
      "step": 26830
    },
    {
      "epoch": 3.139480458694126,
      "loss_main": 0.046737972646951675,
      "step": 26830
    },
    {
      "epoch": 3.140650596770419,
      "grad_norm": 0.35925909876823425,
      "learning_rate": 6.270389001553238e-05,
      "loss": 0.06,
      "step": 26840
    },
    {
      "epoch": 3.140650596770419,
      "loss_main": 0.05536134913563728,
      "step": 26840
    },
    {
      "epoch": 3.141820734846712,
      "grad_norm": 0.40973949432373047,
      "learning_rate": 6.26772346007546e-05,
      "loss": 0.0757,
      "step": 26850
    },
    {
      "epoch": 3.141820734846712,
      "loss_main": 0.05910644307732582,
      "step": 26850
    },
    {
      "epoch": 3.1429908729230047,
      "grad_norm": 0.3462241590023041,
      "learning_rate": 6.265057533496767e-05,
      "loss": 0.0605,
      "step": 26860
    },
    {
      "epoch": 3.1429908729230047,
      "loss_main": 0.04917803034186363,
      "step": 26860
    },
    {
      "epoch": 3.144161010999298,
      "grad_norm": 0.3042606711387634,
      "learning_rate": 6.262391222626997e-05,
      "loss": 0.0704,
      "step": 26870
    },
    {
      "epoch": 3.144161010999298,
      "loss_main": 0.05897751823067665,
      "step": 26870
    },
    {
      "epoch": 3.145331149075591,
      "grad_norm": 0.22464561462402344,
      "learning_rate": 6.259724528276106e-05,
      "loss": 0.0628,
      "step": 26880
    },
    {
      "epoch": 3.145331149075591,
      "loss_main": 0.0648670643568039,
      "step": 26880
    },
    {
      "epoch": 3.146501287151884,
      "grad_norm": 0.2660921514034271,
      "learning_rate": 6.257057451254162e-05,
      "loss": 0.0586,
      "step": 26890
    },
    {
      "epoch": 3.146501287151884,
      "loss_main": 0.06176120787858963,
      "step": 26890
    },
    {
      "epoch": 3.147671425228177,
      "grad_norm": 0.2786407470703125,
      "learning_rate": 6.254389992371357e-05,
      "loss": 0.0636,
      "step": 26900
    },
    {
      "epoch": 3.147671425228177,
      "loss_main": 0.07138054817914963,
      "step": 26900
    },
    {
      "epoch": 3.1488415633044697,
      "grad_norm": 0.33505555987358093,
      "learning_rate": 6.25172215243799e-05,
      "loss": 0.066,
      "step": 26910
    },
    {
      "epoch": 3.1488415633044697,
      "loss_main": 0.067793108522892,
      "step": 26910
    },
    {
      "epoch": 3.150011701380763,
      "grad_norm": 0.26594966650009155,
      "learning_rate": 6.249053932264486e-05,
      "loss": 0.0656,
      "step": 26920
    },
    {
      "epoch": 3.150011701380763,
      "loss_main": 0.12113555520772934,
      "step": 26920
    },
    {
      "epoch": 3.151181839457056,
      "grad_norm": 0.32298457622528076,
      "learning_rate": 6.246385332661376e-05,
      "loss": 0.067,
      "step": 26930
    },
    {
      "epoch": 3.151181839457056,
      "loss_main": 0.0990060418844223,
      "step": 26930
    },
    {
      "epoch": 3.152351977533349,
      "grad_norm": 0.34965938329696655,
      "learning_rate": 6.24371635443931e-05,
      "loss": 0.072,
      "step": 26940
    },
    {
      "epoch": 3.152351977533349,
      "loss_main": 0.06349276006221771,
      "step": 26940
    },
    {
      "epoch": 3.153522115609642,
      "grad_norm": 0.2259288877248764,
      "learning_rate": 6.241046998409054e-05,
      "loss": 0.0532,
      "step": 26950
    },
    {
      "epoch": 3.153522115609642,
      "loss_main": 0.03775274008512497,
      "step": 26950
    },
    {
      "epoch": 3.1546922536859348,
      "grad_norm": 0.2721113860607147,
      "learning_rate": 6.238377265381489e-05,
      "loss": 0.0577,
      "step": 26960
    },
    {
      "epoch": 3.1546922536859348,
      "loss_main": 0.03670341521501541,
      "step": 26960
    },
    {
      "epoch": 3.155862391762228,
      "grad_norm": 0.39159709215164185,
      "learning_rate": 6.235707156167607e-05,
      "loss": 0.0744,
      "step": 26970
    },
    {
      "epoch": 3.155862391762228,
      "loss_main": 0.05489441752433777,
      "step": 26970
    },
    {
      "epoch": 3.157032529838521,
      "grad_norm": 0.3184000849723816,
      "learning_rate": 6.233036671578519e-05,
      "loss": 0.0757,
      "step": 26980
    },
    {
      "epoch": 3.157032529838521,
      "loss_main": 0.10858452320098877,
      "step": 26980
    },
    {
      "epoch": 3.158202667914814,
      "grad_norm": 0.5470442771911621,
      "learning_rate": 6.230365812425445e-05,
      "loss": 0.0651,
      "step": 26990
    },
    {
      "epoch": 3.158202667914814,
      "loss_main": 0.08760036528110504,
      "step": 26990
    },
    {
      "epoch": 3.159372805991107,
      "grad_norm": 0.31014418601989746,
      "learning_rate": 6.227694579519724e-05,
      "loss": 0.065,
      "step": 27000
    },
    {
      "epoch": 3.159372805991107,
      "loss_main": 0.08346683531999588,
      "step": 27000
    },
    {
      "epoch": 3.1605429440673998,
      "grad_norm": 0.23725053668022156,
      "learning_rate": 6.225022973672805e-05,
      "loss": 0.0689,
      "step": 27010
    },
    {
      "epoch": 3.1605429440673998,
      "loss_main": 0.11916305124759674,
      "step": 27010
    },
    {
      "epoch": 3.161713082143693,
      "grad_norm": 0.34657129645347595,
      "learning_rate": 6.222350995696253e-05,
      "loss": 0.0736,
      "step": 27020
    },
    {
      "epoch": 3.161713082143693,
      "loss_main": 0.06633199006319046,
      "step": 27020
    },
    {
      "epoch": 3.162883220219986,
      "grad_norm": 0.467415452003479,
      "learning_rate": 6.21967864640174e-05,
      "loss": 0.0703,
      "step": 27030
    },
    {
      "epoch": 3.162883220219986,
      "loss_main": 0.04135780781507492,
      "step": 27030
    },
    {
      "epoch": 3.164053358296279,
      "grad_norm": 0.28134873509407043,
      "learning_rate": 6.217005926601059e-05,
      "loss": 0.058,
      "step": 27040
    },
    {
      "epoch": 3.164053358296279,
      "loss_main": 0.0644959956407547,
      "step": 27040
    },
    {
      "epoch": 3.165223496372572,
      "grad_norm": 0.3383118510246277,
      "learning_rate": 6.214332837106111e-05,
      "loss": 0.063,
      "step": 27050
    },
    {
      "epoch": 3.165223496372572,
      "loss_main": 0.05230513587594032,
      "step": 27050
    },
    {
      "epoch": 3.166393634448865,
      "grad_norm": 0.4516579508781433,
      "learning_rate": 6.21165937872891e-05,
      "loss": 0.0638,
      "step": 27060
    },
    {
      "epoch": 3.166393634448865,
      "loss_main": 0.1134229525923729,
      "step": 27060
    },
    {
      "epoch": 3.167563772525158,
      "grad_norm": 0.27008581161499023,
      "learning_rate": 6.208985552281582e-05,
      "loss": 0.0711,
      "step": 27070
    },
    {
      "epoch": 3.167563772525158,
      "loss_main": 0.11247842013835907,
      "step": 27070
    },
    {
      "epoch": 3.168733910601451,
      "grad_norm": 0.32586729526519775,
      "learning_rate": 6.206311358576364e-05,
      "loss": 0.0775,
      "step": 27080
    },
    {
      "epoch": 3.168733910601451,
      "loss_main": 0.06777478009462357,
      "step": 27080
    },
    {
      "epoch": 3.169904048677744,
      "grad_norm": 0.2762793302536011,
      "learning_rate": 6.203636798425608e-05,
      "loss": 0.0662,
      "step": 27090
    },
    {
      "epoch": 3.169904048677744,
      "loss_main": 0.08699937164783478,
      "step": 27090
    },
    {
      "epoch": 3.171074186754037,
      "grad_norm": 0.24762371182441711,
      "learning_rate": 6.20096187264177e-05,
      "loss": 0.0692,
      "step": 27100
    },
    {
      "epoch": 3.171074186754037,
      "loss_main": 0.06742667406797409,
      "step": 27100
    },
    {
      "epoch": 3.17224432483033,
      "grad_norm": 0.27019253373146057,
      "learning_rate": 6.198286582037425e-05,
      "loss": 0.0639,
      "step": 27110
    },
    {
      "epoch": 3.17224432483033,
      "loss_main": 0.07507344335317612,
      "step": 27110
    },
    {
      "epoch": 3.173414462906623,
      "grad_norm": 0.370181143283844,
      "learning_rate": 6.195610927425256e-05,
      "loss": 0.0624,
      "step": 27120
    },
    {
      "epoch": 3.173414462906623,
      "loss_main": 0.037243228405714035,
      "step": 27120
    },
    {
      "epoch": 3.174584600982916,
      "grad_norm": 0.24066556990146637,
      "learning_rate": 6.192934909618056e-05,
      "loss": 0.056,
      "step": 27130
    },
    {
      "epoch": 3.174584600982916,
      "loss_main": 0.08139916509389877,
      "step": 27130
    },
    {
      "epoch": 3.175754739059209,
      "grad_norm": 0.19907401502132416,
      "learning_rate": 6.190258529428728e-05,
      "loss": 0.0686,
      "step": 27140
    },
    {
      "epoch": 3.175754739059209,
      "loss_main": 0.12169528007507324,
      "step": 27140
    },
    {
      "epoch": 3.176924877135502,
      "grad_norm": 0.24046990275382996,
      "learning_rate": 6.187581787670285e-05,
      "loss": 0.067,
      "step": 27150
    },
    {
      "epoch": 3.176924877135502,
      "loss_main": 0.061237458139657974,
      "step": 27150
    },
    {
      "epoch": 3.178095015211795,
      "grad_norm": 0.2578485310077667,
      "learning_rate": 6.184904685155852e-05,
      "loss": 0.0655,
      "step": 27160
    },
    {
      "epoch": 3.178095015211795,
      "loss_main": 0.09104108810424805,
      "step": 27160
    },
    {
      "epoch": 3.179265153288088,
      "grad_norm": 0.3893042504787445,
      "learning_rate": 6.18222722269866e-05,
      "loss": 0.0687,
      "step": 27170
    },
    {
      "epoch": 3.179265153288088,
      "loss_main": 0.05878269672393799,
      "step": 27170
    },
    {
      "epoch": 3.180435291364381,
      "grad_norm": 0.21734167635440826,
      "learning_rate": 6.179549401112053e-05,
      "loss": 0.059,
      "step": 27180
    },
    {
      "epoch": 3.180435291364381,
      "loss_main": 0.055405180901288986,
      "step": 27180
    },
    {
      "epoch": 3.181605429440674,
      "grad_norm": 0.23584413528442383,
      "learning_rate": 6.176871221209482e-05,
      "loss": 0.0685,
      "step": 27190
    },
    {
      "epoch": 3.181605429440674,
      "loss_main": 0.0633920431137085,
      "step": 27190
    },
    {
      "epoch": 3.182775567516967,
      "grad_norm": 0.37928491830825806,
      "learning_rate": 6.174192683804508e-05,
      "loss": 0.0611,
      "step": 27200
    },
    {
      "epoch": 3.182775567516967,
      "loss_main": 0.06524630635976791,
      "step": 27200
    },
    {
      "epoch": 3.18394570559326,
      "grad_norm": 0.3509206175804138,
      "learning_rate": 6.1715137897108e-05,
      "loss": 0.0644,
      "step": 27210
    },
    {
      "epoch": 3.18394570559326,
      "loss_main": 0.06041260436177254,
      "step": 27210
    },
    {
      "epoch": 3.185115843669553,
      "grad_norm": 0.442880243062973,
      "learning_rate": 6.168834539742134e-05,
      "loss": 0.0648,
      "step": 27220
    },
    {
      "epoch": 3.185115843669553,
      "loss_main": 0.062055803835392,
      "step": 27220
    },
    {
      "epoch": 3.186285981745846,
      "grad_norm": 0.2996866703033447,
      "learning_rate": 6.166154934712397e-05,
      "loss": 0.0699,
      "step": 27230
    },
    {
      "epoch": 3.186285981745846,
      "loss_main": 0.058508243411779404,
      "step": 27230
    },
    {
      "epoch": 3.187456119822139,
      "grad_norm": 0.26312634348869324,
      "learning_rate": 6.163474975435581e-05,
      "loss": 0.0659,
      "step": 27240
    },
    {
      "epoch": 3.187456119822139,
      "loss_main": 0.06652433425188065,
      "step": 27240
    },
    {
      "epoch": 3.188626257898432,
      "grad_norm": 0.3717149496078491,
      "learning_rate": 6.160794662725787e-05,
      "loss": 0.0766,
      "step": 27250
    },
    {
      "epoch": 3.188626257898432,
      "loss_main": 0.04331424832344055,
      "step": 27250
    },
    {
      "epoch": 3.189796395974725,
      "grad_norm": 0.26894405484199524,
      "learning_rate": 6.158113997397222e-05,
      "loss": 0.0611,
      "step": 27260
    },
    {
      "epoch": 3.189796395974725,
      "loss_main": 0.06955809146165848,
      "step": 27260
    },
    {
      "epoch": 3.190966534051018,
      "grad_norm": 0.30201733112335205,
      "learning_rate": 6.155432980264205e-05,
      "loss": 0.0676,
      "step": 27270
    },
    {
      "epoch": 3.190966534051018,
      "loss_main": 0.09931262582540512,
      "step": 27270
    },
    {
      "epoch": 3.1921366721273112,
      "grad_norm": 0.42845845222473145,
      "learning_rate": 6.152751612141156e-05,
      "loss": 0.0702,
      "step": 27280
    },
    {
      "epoch": 3.1921366721273112,
      "loss_main": 0.09040971845388412,
      "step": 27280
    },
    {
      "epoch": 3.193306810203604,
      "grad_norm": 0.39292487502098083,
      "learning_rate": 6.150069893842602e-05,
      "loss": 0.0646,
      "step": 27290
    },
    {
      "epoch": 3.193306810203604,
      "loss_main": 0.06176232546567917,
      "step": 27290
    },
    {
      "epoch": 3.194476948279897,
      "grad_norm": 0.31623461842536926,
      "learning_rate": 6.147387826183182e-05,
      "loss": 0.0597,
      "step": 27300
    },
    {
      "epoch": 3.194476948279897,
      "loss_main": 0.11511129885911942,
      "step": 27300
    },
    {
      "epoch": 3.19564708635619,
      "grad_norm": 0.22060444951057434,
      "learning_rate": 6.144705409977635e-05,
      "loss": 0.0696,
      "step": 27310
    },
    {
      "epoch": 3.19564708635619,
      "loss_main": 0.05309617146849632,
      "step": 27310
    },
    {
      "epoch": 3.196817224432483,
      "grad_norm": 0.32893097400665283,
      "learning_rate": 6.142022646040808e-05,
      "loss": 0.0615,
      "step": 27320
    },
    {
      "epoch": 3.196817224432483,
      "loss_main": 0.07494906336069107,
      "step": 27320
    },
    {
      "epoch": 3.1979873625087762,
      "grad_norm": 0.3260835111141205,
      "learning_rate": 6.139339535187653e-05,
      "loss": 0.0647,
      "step": 27330
    },
    {
      "epoch": 3.1979873625087762,
      "loss_main": 0.06407676637172699,
      "step": 27330
    },
    {
      "epoch": 3.199157500585069,
      "grad_norm": 0.2532532215118408,
      "learning_rate": 6.136656078233232e-05,
      "loss": 0.0607,
      "step": 27340
    },
    {
      "epoch": 3.199157500585069,
      "loss_main": 0.04740160331130028,
      "step": 27340
    },
    {
      "epoch": 3.200327638661362,
      "grad_norm": 0.355573832988739,
      "learning_rate": 6.133972275992707e-05,
      "loss": 0.0726,
      "step": 27350
    },
    {
      "epoch": 3.200327638661362,
      "loss_main": 0.07578680664300919,
      "step": 27350
    },
    {
      "epoch": 3.201497776737655,
      "grad_norm": 0.28169989585876465,
      "learning_rate": 6.131288129281342e-05,
      "loss": 0.0667,
      "step": 27360
    },
    {
      "epoch": 3.201497776737655,
      "loss_main": 0.08024334162473679,
      "step": 27360
    },
    {
      "epoch": 3.202667914813948,
      "grad_norm": 0.30223095417022705,
      "learning_rate": 6.128603638914516e-05,
      "loss": 0.0671,
      "step": 27370
    },
    {
      "epoch": 3.202667914813948,
      "loss_main": 0.06001096963882446,
      "step": 27370
    },
    {
      "epoch": 3.203838052890241,
      "grad_norm": 0.35819315910339355,
      "learning_rate": 6.125918805707704e-05,
      "loss": 0.0653,
      "step": 27380
    },
    {
      "epoch": 3.203838052890241,
      "loss_main": 0.07814167439937592,
      "step": 27380
    },
    {
      "epoch": 3.205008190966534,
      "grad_norm": 0.26489222049713135,
      "learning_rate": 6.123233630476485e-05,
      "loss": 0.0719,
      "step": 27390
    },
    {
      "epoch": 3.205008190966534,
      "loss_main": 0.039227429777383804,
      "step": 27390
    },
    {
      "epoch": 3.206178329042827,
      "grad_norm": 0.4066978394985199,
      "learning_rate": 6.120548114036547e-05,
      "loss": 0.0672,
      "step": 27400
    },
    {
      "epoch": 3.206178329042827,
      "loss_main": 0.06458892673254013,
      "step": 27400
    },
    {
      "epoch": 3.20734846711912,
      "grad_norm": 0.3315877914428711,
      "learning_rate": 6.117862257203679e-05,
      "loss": 0.0667,
      "step": 27410
    },
    {
      "epoch": 3.20734846711912,
      "loss_main": 0.05142189934849739,
      "step": 27410
    },
    {
      "epoch": 3.208518605195413,
      "grad_norm": 0.3189627230167389,
      "learning_rate": 6.115176060793771e-05,
      "loss": 0.0681,
      "step": 27420
    },
    {
      "epoch": 3.208518605195413,
      "loss_main": 0.0402967631816864,
      "step": 27420
    },
    {
      "epoch": 3.2096887432717063,
      "grad_norm": 0.23182366788387299,
      "learning_rate": 6.112489525622822e-05,
      "loss": 0.053,
      "step": 27430
    },
    {
      "epoch": 3.2096887432717063,
      "loss_main": 0.03552969545125961,
      "step": 27430
    },
    {
      "epoch": 3.210858881347999,
      "grad_norm": 0.4032752811908722,
      "learning_rate": 6.109802652506928e-05,
      "loss": 0.0679,
      "step": 27440
    },
    {
      "epoch": 3.210858881347999,
      "loss_main": 0.07489713281393051,
      "step": 27440
    },
    {
      "epoch": 3.212029019424292,
      "grad_norm": 0.24223043024539948,
      "learning_rate": 6.107115442262291e-05,
      "loss": 0.07,
      "step": 27450
    },
    {
      "epoch": 3.212029019424292,
      "loss_main": 0.059179987758398056,
      "step": 27450
    },
    {
      "epoch": 3.213199157500585,
      "grad_norm": 0.3162091076374054,
      "learning_rate": 6.104427895705214e-05,
      "loss": 0.0631,
      "step": 27460
    },
    {
      "epoch": 3.213199157500585,
      "loss_main": 0.04603932425379753,
      "step": 27460
    },
    {
      "epoch": 3.214369295576878,
      "grad_norm": 0.3277769684791565,
      "learning_rate": 6.101740013652103e-05,
      "loss": 0.0529,
      "step": 27470
    },
    {
      "epoch": 3.214369295576878,
      "loss_main": 0.07858791202306747,
      "step": 27470
    },
    {
      "epoch": 3.215539433653171,
      "grad_norm": 0.292562335729599,
      "learning_rate": 6.099051796919465e-05,
      "loss": 0.0741,
      "step": 27480
    },
    {
      "epoch": 3.215539433653171,
      "loss_main": 0.12435146421194077,
      "step": 27480
    },
    {
      "epoch": 3.216709571729464,
      "grad_norm": 0.26088765263557434,
      "learning_rate": 6.096363246323911e-05,
      "loss": 0.0674,
      "step": 27490
    },
    {
      "epoch": 3.216709571729464,
      "loss_main": 0.04283145070075989,
      "step": 27490
    },
    {
      "epoch": 3.217879709805757,
      "grad_norm": 0.19868522882461548,
      "learning_rate": 6.0936743626821504e-05,
      "loss": 0.0648,
      "step": 27500
    },
    {
      "epoch": 3.217879709805757,
      "loss_main": 0.04370544105768204,
      "step": 27500
    },
    {
      "epoch": 3.21904984788205,
      "grad_norm": 0.2337837964296341,
      "learning_rate": 6.090985146810996e-05,
      "loss": 0.0563,
      "step": 27510
    },
    {
      "epoch": 3.21904984788205,
      "loss_main": 0.06546954065561295,
      "step": 27510
    },
    {
      "epoch": 3.220219985958343,
      "grad_norm": 0.2946585416793823,
      "learning_rate": 6.088295599527357e-05,
      "loss": 0.0627,
      "step": 27520
    },
    {
      "epoch": 3.220219985958343,
      "loss_main": 0.05403217673301697,
      "step": 27520
    },
    {
      "epoch": 3.2213901240346363,
      "grad_norm": 0.2690736651420593,
      "learning_rate": 6.085605721648252e-05,
      "loss": 0.0673,
      "step": 27530
    },
    {
      "epoch": 3.2213901240346363,
      "loss_main": 0.09556933492422104,
      "step": 27530
    },
    {
      "epoch": 3.222560262110929,
      "grad_norm": 0.30918389558792114,
      "learning_rate": 6.082915513990792e-05,
      "loss": 0.0627,
      "step": 27540
    },
    {
      "epoch": 3.222560262110929,
      "loss_main": 0.05713125318288803,
      "step": 27540
    },
    {
      "epoch": 3.2237304001872222,
      "grad_norm": 0.23139896988868713,
      "learning_rate": 6.080224977372192e-05,
      "loss": 0.0664,
      "step": 27550
    },
    {
      "epoch": 3.2237304001872222,
      "loss_main": 0.12311410158872604,
      "step": 27550
    },
    {
      "epoch": 3.224900538263515,
      "grad_norm": 0.2920069098472595,
      "learning_rate": 6.0775341126097666e-05,
      "loss": 0.0618,
      "step": 27560
    },
    {
      "epoch": 3.224900538263515,
      "loss_main": 0.06041882559657097,
      "step": 27560
    },
    {
      "epoch": 3.226070676339808,
      "grad_norm": 0.24380770325660706,
      "learning_rate": 6.074842920520926e-05,
      "loss": 0.0721,
      "step": 27570
    },
    {
      "epoch": 3.226070676339808,
      "loss_main": 0.05746404081583023,
      "step": 27570
    },
    {
      "epoch": 3.227240814416101,
      "grad_norm": 0.34586790204048157,
      "learning_rate": 6.072151401923186e-05,
      "loss": 0.0661,
      "step": 27580
    },
    {
      "epoch": 3.227240814416101,
      "loss_main": 0.040295932441949844,
      "step": 27580
    },
    {
      "epoch": 3.228410952492394,
      "grad_norm": 0.3865742087364197,
      "learning_rate": 6.069459557634159e-05,
      "loss": 0.0689,
      "step": 27590
    },
    {
      "epoch": 3.228410952492394,
      "loss_main": 0.04953118786215782,
      "step": 27590
    },
    {
      "epoch": 3.2295810905686873,
      "grad_norm": 0.4086879789829254,
      "learning_rate": 6.066767388471557e-05,
      "loss": 0.0631,
      "step": 27600
    },
    {
      "epoch": 3.2295810905686873,
      "loss_main": 0.12128375470638275,
      "step": 27600
    },
    {
      "epoch": 3.23075122864498,
      "grad_norm": 0.3118717670440674,
      "learning_rate": 6.064074895253188e-05,
      "loss": 0.0697,
      "step": 27610
    },
    {
      "epoch": 3.23075122864498,
      "loss_main": 0.09220617264509201,
      "step": 27610
    },
    {
      "epoch": 3.231921366721273,
      "grad_norm": 0.37880897521972656,
      "learning_rate": 6.061382078796961e-05,
      "loss": 0.0765,
      "step": 27620
    },
    {
      "epoch": 3.231921366721273,
      "loss_main": 0.07066162675619125,
      "step": 27620
    },
    {
      "epoch": 3.233091504797566,
      "grad_norm": 0.3234718143939972,
      "learning_rate": 6.0586889399208814e-05,
      "loss": 0.0632,
      "step": 27630
    },
    {
      "epoch": 3.233091504797566,
      "loss_main": 0.07338857650756836,
      "step": 27630
    },
    {
      "epoch": 3.234261642873859,
      "grad_norm": 0.30108290910720825,
      "learning_rate": 6.0559954794430565e-05,
      "loss": 0.0714,
      "step": 27640
    },
    {
      "epoch": 3.234261642873859,
      "loss_main": 0.10536796599626541,
      "step": 27640
    },
    {
      "epoch": 3.2354317809501523,
      "grad_norm": 0.2852856516838074,
      "learning_rate": 6.053301698181687e-05,
      "loss": 0.0645,
      "step": 27650
    },
    {
      "epoch": 3.2354317809501523,
      "loss_main": 0.03447369486093521,
      "step": 27650
    },
    {
      "epoch": 3.236601919026445,
      "grad_norm": 0.3896460235118866,
      "learning_rate": 6.0506075969550725e-05,
      "loss": 0.0645,
      "step": 27660
    },
    {
      "epoch": 3.236601919026445,
      "loss_main": 0.06228658929467201,
      "step": 27660
    },
    {
      "epoch": 3.237772057102738,
      "grad_norm": 0.16206566989421844,
      "learning_rate": 6.047913176581609e-05,
      "loss": 0.0675,
      "step": 27670
    },
    {
      "epoch": 3.237772057102738,
      "loss_main": 0.07735176384449005,
      "step": 27670
    },
    {
      "epoch": 3.238942195179031,
      "grad_norm": 0.31700870394706726,
      "learning_rate": 6.0452184378797904e-05,
      "loss": 0.0688,
      "step": 27680
    },
    {
      "epoch": 3.238942195179031,
      "loss_main": 0.051516592502593994,
      "step": 27680
    },
    {
      "epoch": 3.240112333255324,
      "grad_norm": 0.3075876832008362,
      "learning_rate": 6.042523381668209e-05,
      "loss": 0.0621,
      "step": 27690
    },
    {
      "epoch": 3.240112333255324,
      "loss_main": 0.058536987751722336,
      "step": 27690
    },
    {
      "epoch": 3.2412824713316173,
      "grad_norm": 0.29370808601379395,
      "learning_rate": 6.03982800876555e-05,
      "loss": 0.0665,
      "step": 27700
    },
    {
      "epoch": 3.2412824713316173,
      "loss_main": 0.11941812932491302,
      "step": 27700
    },
    {
      "epoch": 3.24245260940791,
      "grad_norm": 0.3055298626422882,
      "learning_rate": 6.0371323199905975e-05,
      "loss": 0.0718,
      "step": 27710
    },
    {
      "epoch": 3.24245260940791,
      "loss_main": 0.051169101148843765,
      "step": 27710
    },
    {
      "epoch": 3.243622747484203,
      "grad_norm": 0.3121774196624756,
      "learning_rate": 6.03443631616223e-05,
      "loss": 0.0627,
      "step": 27720
    },
    {
      "epoch": 3.243622747484203,
      "loss_main": 0.038366176187992096,
      "step": 27720
    },
    {
      "epoch": 3.244792885560496,
      "grad_norm": 0.2317844182252884,
      "learning_rate": 6.031739998099421e-05,
      "loss": 0.0534,
      "step": 27730
    },
    {
      "epoch": 3.244792885560496,
      "loss_main": 0.07120432704687119,
      "step": 27730
    },
    {
      "epoch": 3.245963023636789,
      "grad_norm": 0.27420273423194885,
      "learning_rate": 6.029043366621243e-05,
      "loss": 0.0657,
      "step": 27740
    },
    {
      "epoch": 3.245963023636789,
      "loss_main": 0.03670808672904968,
      "step": 27740
    },
    {
      "epoch": 3.2471331617130823,
      "grad_norm": 0.32034286856651306,
      "learning_rate": 6.0263464225468615e-05,
      "loss": 0.0719,
      "step": 27750
    },
    {
      "epoch": 3.2471331617130823,
      "loss_main": 0.04522264376282692,
      "step": 27750
    },
    {
      "epoch": 3.248303299789375,
      "grad_norm": 0.3020245432853699,
      "learning_rate": 6.023649166695534e-05,
      "loss": 0.0588,
      "step": 27760
    },
    {
      "epoch": 3.248303299789375,
      "loss_main": 0.06447599828243256,
      "step": 27760
    },
    {
      "epoch": 3.2494734378656682,
      "grad_norm": 0.3430224359035492,
      "learning_rate": 6.0209515998866186e-05,
      "loss": 0.0683,
      "step": 27770
    },
    {
      "epoch": 3.2494734378656682,
      "loss_main": 0.041282039135694504,
      "step": 27770
    },
    {
      "epoch": 3.250643575941961,
      "grad_norm": 0.28095027804374695,
      "learning_rate": 6.018253722939563e-05,
      "loss": 0.0655,
      "step": 27780
    },
    {
      "epoch": 3.250643575941961,
      "loss_main": 0.0599195696413517,
      "step": 27780
    },
    {
      "epoch": 3.251813714018254,
      "grad_norm": 0.3489142954349518,
      "learning_rate": 6.015555536673914e-05,
      "loss": 0.0665,
      "step": 27790
    },
    {
      "epoch": 3.251813714018254,
      "loss_main": 0.04115390405058861,
      "step": 27790
    },
    {
      "epoch": 3.2529838520945473,
      "grad_norm": 0.33963248133659363,
      "learning_rate": 6.0128570419093054e-05,
      "loss": 0.0629,
      "step": 27800
    },
    {
      "epoch": 3.2529838520945473,
      "loss_main": 0.0520852729678154,
      "step": 27800
    },
    {
      "epoch": 3.25415399017084,
      "grad_norm": 0.34676307439804077,
      "learning_rate": 6.010158239465471e-05,
      "loss": 0.0581,
      "step": 27810
    },
    {
      "epoch": 3.25415399017084,
      "loss_main": 0.05610180273652077,
      "step": 27810
    },
    {
      "epoch": 3.2553241282471332,
      "grad_norm": 0.37891364097595215,
      "learning_rate": 6.007459130162235e-05,
      "loss": 0.0691,
      "step": 27820
    },
    {
      "epoch": 3.2553241282471332,
      "loss_main": 0.08478468656539917,
      "step": 27820
    },
    {
      "epoch": 3.256494266323426,
      "grad_norm": 0.2523857057094574,
      "learning_rate": 6.004759714819516e-05,
      "loss": 0.0722,
      "step": 27830
    },
    {
      "epoch": 3.256494266323426,
      "loss_main": 0.03412469103932381,
      "step": 27830
    },
    {
      "epoch": 3.257664404399719,
      "grad_norm": 0.3074226975440979,
      "learning_rate": 6.002059994257323e-05,
      "loss": 0.0698,
      "step": 27840
    },
    {
      "epoch": 3.257664404399719,
      "loss_main": 0.056292593479156494,
      "step": 27840
    },
    {
      "epoch": 3.2588345424760123,
      "grad_norm": 0.2934378981590271,
      "learning_rate": 5.999359969295764e-05,
      "loss": 0.0674,
      "step": 27850
    },
    {
      "epoch": 3.2588345424760123,
      "loss_main": 0.053006406873464584,
      "step": 27850
    },
    {
      "epoch": 3.260004680552305,
      "grad_norm": 0.2327415645122528,
      "learning_rate": 5.9966596407550314e-05,
      "loss": 0.061,
      "step": 27860
    },
    {
      "epoch": 3.260004680552305,
      "loss_main": 0.0718630999326706,
      "step": 27860
    },
    {
      "epoch": 3.2611748186285983,
      "grad_norm": 0.3555970788002014,
      "learning_rate": 5.993959009455416e-05,
      "loss": 0.0801,
      "step": 27870
    },
    {
      "epoch": 3.2611748186285983,
      "loss_main": 0.04929644614458084,
      "step": 27870
    },
    {
      "epoch": 3.262344956704891,
      "grad_norm": 0.3158498704433441,
      "learning_rate": 5.991258076217298e-05,
      "loss": 0.0596,
      "step": 27880
    },
    {
      "epoch": 3.262344956704891,
      "loss_main": 0.09254033118486404,
      "step": 27880
    },
    {
      "epoch": 3.263515094781184,
      "grad_norm": 0.31887125968933105,
      "learning_rate": 5.988556841861147e-05,
      "loss": 0.071,
      "step": 27890
    },
    {
      "epoch": 3.263515094781184,
      "loss_main": 0.08232225477695465,
      "step": 27890
    },
    {
      "epoch": 3.2646852328574774,
      "grad_norm": 0.24578556418418884,
      "learning_rate": 5.985855307207531e-05,
      "loss": 0.0681,
      "step": 27900
    },
    {
      "epoch": 3.2646852328574774,
      "loss_main": 0.051479775458574295,
      "step": 27900
    },
    {
      "epoch": 3.26585537093377,
      "grad_norm": 0.3887316584587097,
      "learning_rate": 5.9831534730771e-05,
      "loss": 0.0599,
      "step": 27910
    },
    {
      "epoch": 3.26585537093377,
      "loss_main": 0.05361708253622055,
      "step": 27910
    },
    {
      "epoch": 3.2670255090100633,
      "grad_norm": 0.3518938422203064,
      "learning_rate": 5.980451340290605e-05,
      "loss": 0.0725,
      "step": 27920
    },
    {
      "epoch": 3.2670255090100633,
      "loss_main": 0.06011491268873215,
      "step": 27920
    },
    {
      "epoch": 3.268195647086356,
      "grad_norm": 0.22318990528583527,
      "learning_rate": 5.97774890966888e-05,
      "loss": 0.0597,
      "step": 27930
    },
    {
      "epoch": 3.268195647086356,
      "loss_main": 0.048302359879016876,
      "step": 27930
    },
    {
      "epoch": 3.269365785162649,
      "grad_norm": 0.20229694247245789,
      "learning_rate": 5.975046182032851e-05,
      "loss": 0.0606,
      "step": 27940
    },
    {
      "epoch": 3.269365785162649,
      "loss_main": 0.06031502038240433,
      "step": 27940
    },
    {
      "epoch": 3.2705359232389424,
      "grad_norm": 0.3190080225467682,
      "learning_rate": 5.972343158203537e-05,
      "loss": 0.0603,
      "step": 27950
    },
    {
      "epoch": 3.2705359232389424,
      "loss_main": 0.06397543847560883,
      "step": 27950
    },
    {
      "epoch": 3.271706061315235,
      "grad_norm": 0.3480050265789032,
      "learning_rate": 5.969639839002045e-05,
      "loss": 0.0624,
      "step": 27960
    },
    {
      "epoch": 3.271706061315235,
      "loss_main": 0.11629630625247955,
      "step": 27960
    },
    {
      "epoch": 3.2728761993915283,
      "grad_norm": 0.18356548249721527,
      "learning_rate": 5.966936225249572e-05,
      "loss": 0.0588,
      "step": 27970
    },
    {
      "epoch": 3.2728761993915283,
      "loss_main": 0.09087119996547699,
      "step": 27970
    },
    {
      "epoch": 3.274046337467821,
      "grad_norm": 0.30523917078971863,
      "learning_rate": 5.9642323177674044e-05,
      "loss": 0.0586,
      "step": 27980
    },
    {
      "epoch": 3.274046337467821,
      "loss_main": 0.04081207513809204,
      "step": 27980
    },
    {
      "epoch": 3.275216475544114,
      "grad_norm": 0.30153295397758484,
      "learning_rate": 5.9615281173769154e-05,
      "loss": 0.0594,
      "step": 27990
    },
    {
      "epoch": 3.275216475544114,
      "loss_main": 0.07569585740566254,
      "step": 27990
    },
    {
      "epoch": 3.2763866136204074,
      "grad_norm": 0.2801727056503296,
      "learning_rate": 5.958823624899574e-05,
      "loss": 0.0596,
      "step": 28000
    },
    {
      "epoch": 3.2763866136204074,
      "loss_main": 0.04302772879600525,
      "step": 28000
    },
    {
      "epoch": 3.2775567516967,
      "grad_norm": 0.40253835916519165,
      "learning_rate": 5.956118841156933e-05,
      "loss": 0.0677,
      "step": 28010
    },
    {
      "epoch": 3.2775567516967,
      "loss_main": 0.09647643566131592,
      "step": 28010
    },
    {
      "epoch": 3.2787268897729933,
      "grad_norm": 0.26596447825431824,
      "learning_rate": 5.953413766970631e-05,
      "loss": 0.0687,
      "step": 28020
    },
    {
      "epoch": 3.2787268897729933,
      "loss_main": 0.05141056701540947,
      "step": 28020
    },
    {
      "epoch": 3.279897027849286,
      "grad_norm": 0.29571500420570374,
      "learning_rate": 5.9507084031624e-05,
      "loss": 0.0589,
      "step": 28030
    },
    {
      "epoch": 3.279897027849286,
      "loss_main": 0.042484719306230545,
      "step": 28030
    },
    {
      "epoch": 3.2810671659255792,
      "grad_norm": 0.2563614547252655,
      "learning_rate": 5.948002750554058e-05,
      "loss": 0.0638,
      "step": 28040
    },
    {
      "epoch": 3.2810671659255792,
      "loss_main": 0.10905991494655609,
      "step": 28040
    },
    {
      "epoch": 3.2822373040018724,
      "grad_norm": 0.2019539177417755,
      "learning_rate": 5.9452968099675124e-05,
      "loss": 0.0629,
      "step": 28050
    },
    {
      "epoch": 3.2822373040018724,
      "loss_main": 0.072043776512146,
      "step": 28050
    },
    {
      "epoch": 3.283407442078165,
      "grad_norm": 0.24393589794635773,
      "learning_rate": 5.9425905822247527e-05,
      "loss": 0.0673,
      "step": 28060
    },
    {
      "epoch": 3.283407442078165,
      "loss_main": 0.067064568400383,
      "step": 28060
    },
    {
      "epoch": 3.2845775801544583,
      "grad_norm": 0.37627580761909485,
      "learning_rate": 5.939884068147864e-05,
      "loss": 0.0657,
      "step": 28070
    },
    {
      "epoch": 3.2845775801544583,
      "loss_main": 0.04448642581701279,
      "step": 28070
    },
    {
      "epoch": 3.285747718230751,
      "grad_norm": 0.3719457983970642,
      "learning_rate": 5.937177268559011e-05,
      "loss": 0.0623,
      "step": 28080
    },
    {
      "epoch": 3.285747718230751,
      "loss_main": 0.0613502599298954,
      "step": 28080
    },
    {
      "epoch": 3.2869178563070443,
      "grad_norm": 0.22211986780166626,
      "learning_rate": 5.934470184280448e-05,
      "loss": 0.0599,
      "step": 28090
    },
    {
      "epoch": 3.2869178563070443,
      "loss_main": 0.04020863026380539,
      "step": 28090
    },
    {
      "epoch": 3.2880879943833374,
      "grad_norm": 0.2902105450630188,
      "learning_rate": 5.931762816134516e-05,
      "loss": 0.0607,
      "step": 28100
    },
    {
      "epoch": 3.2880879943833374,
      "loss_main": 0.10798779875040054,
      "step": 28100
    },
    {
      "epoch": 3.28925813245963,
      "grad_norm": 0.35144567489624023,
      "learning_rate": 5.9290551649436434e-05,
      "loss": 0.0646,
      "step": 28110
    },
    {
      "epoch": 3.28925813245963,
      "loss_main": 0.05808297172188759,
      "step": 28110
    },
    {
      "epoch": 3.2904282705359233,
      "grad_norm": 0.34068939089775085,
      "learning_rate": 5.9263472315303416e-05,
      "loss": 0.0704,
      "step": 28120
    },
    {
      "epoch": 3.2904282705359233,
      "loss_main": 0.05845474451780319,
      "step": 28120
    },
    {
      "epoch": 3.291598408612216,
      "grad_norm": 0.33088505268096924,
      "learning_rate": 5.9236390167172096e-05,
      "loss": 0.0628,
      "step": 28130
    },
    {
      "epoch": 3.291598408612216,
      "loss_main": 0.0645010769367218,
      "step": 28130
    },
    {
      "epoch": 3.2927685466885093,
      "grad_norm": 0.21777158975601196,
      "learning_rate": 5.920930521326932e-05,
      "loss": 0.0687,
      "step": 28140
    },
    {
      "epoch": 3.2927685466885093,
      "loss_main": 0.10680762678384781,
      "step": 28140
    },
    {
      "epoch": 3.293938684764802,
      "grad_norm": 0.2147134393453598,
      "learning_rate": 5.918221746182276e-05,
      "loss": 0.0629,
      "step": 28150
    },
    {
      "epoch": 3.293938684764802,
      "loss_main": 0.05305067077279091,
      "step": 28150
    },
    {
      "epoch": 3.295108822841095,
      "grad_norm": 0.3871883749961853,
      "learning_rate": 5.9155126921061e-05,
      "loss": 0.0608,
      "step": 28160
    },
    {
      "epoch": 3.295108822841095,
      "loss_main": 0.0975874811410904,
      "step": 28160
    },
    {
      "epoch": 3.2962789609173884,
      "grad_norm": 0.3280087113380432,
      "learning_rate": 5.91280335992134e-05,
      "loss": 0.0592,
      "step": 28170
    },
    {
      "epoch": 3.2962789609173884,
      "loss_main": 0.04314832016825676,
      "step": 28170
    },
    {
      "epoch": 3.297449098993681,
      "grad_norm": 0.23929813504219055,
      "learning_rate": 5.91009375045102e-05,
      "loss": 0.0663,
      "step": 28180
    },
    {
      "epoch": 3.297449098993681,
      "loss_main": 0.047859590500593185,
      "step": 28180
    },
    {
      "epoch": 3.2986192370699743,
      "grad_norm": 0.25181156396865845,
      "learning_rate": 5.9073838645182476e-05,
      "loss": 0.0584,
      "step": 28190
    },
    {
      "epoch": 3.2986192370699743,
      "loss_main": 0.0516282394528389,
      "step": 28190
    },
    {
      "epoch": 3.2997893751462675,
      "grad_norm": 0.2133183777332306,
      "learning_rate": 5.904673702946217e-05,
      "loss": 0.0702,
      "step": 28200
    },
    {
      "epoch": 3.2997893751462675,
      "loss_main": 0.0784788504242897,
      "step": 28200
    },
    {
      "epoch": 3.30095951322256,
      "grad_norm": 0.3896293640136719,
      "learning_rate": 5.9019632665582004e-05,
      "loss": 0.062,
      "step": 28210
    },
    {
      "epoch": 3.30095951322256,
      "loss_main": 0.02765180729329586,
      "step": 28210
    },
    {
      "epoch": 3.3021296512988534,
      "grad_norm": 0.27481546998023987,
      "learning_rate": 5.899252556177559e-05,
      "loss": 0.0632,
      "step": 28220
    },
    {
      "epoch": 3.3021296512988534,
      "loss_main": 0.044798221439123154,
      "step": 28220
    },
    {
      "epoch": 3.303299789375146,
      "grad_norm": 0.28185370564460754,
      "learning_rate": 5.896541572627735e-05,
      "loss": 0.0643,
      "step": 28230
    },
    {
      "epoch": 3.303299789375146,
      "loss_main": 0.04512569308280945,
      "step": 28230
    },
    {
      "epoch": 3.3044699274514393,
      "grad_norm": 0.31770527362823486,
      "learning_rate": 5.893830316732253e-05,
      "loss": 0.06,
      "step": 28240
    },
    {
      "epoch": 3.3044699274514393,
      "loss_main": 0.03722551092505455,
      "step": 28240
    },
    {
      "epoch": 3.305640065527732,
      "grad_norm": 0.28324824571609497,
      "learning_rate": 5.8911187893147214e-05,
      "loss": 0.0616,
      "step": 28250
    },
    {
      "epoch": 3.305640065527732,
      "loss_main": 0.0922873467206955,
      "step": 28250
    },
    {
      "epoch": 3.3068102036040252,
      "grad_norm": 0.25846800208091736,
      "learning_rate": 5.888406991198828e-05,
      "loss": 0.0616,
      "step": 28260
    },
    {
      "epoch": 3.3068102036040252,
      "loss_main": 0.07882034778594971,
      "step": 28260
    },
    {
      "epoch": 3.3079803416803184,
      "grad_norm": 0.35315442085266113,
      "learning_rate": 5.885694923208349e-05,
      "loss": 0.0671,
      "step": 28270
    },
    {
      "epoch": 3.3079803416803184,
      "loss_main": 0.07874911278486252,
      "step": 28270
    },
    {
      "epoch": 3.309150479756611,
      "grad_norm": 0.2956525683403015,
      "learning_rate": 5.882982586167138e-05,
      "loss": 0.0743,
      "step": 28280
    },
    {
      "epoch": 3.309150479756611,
      "loss_main": 0.05839588865637779,
      "step": 28280
    },
    {
      "epoch": 3.3103206178329043,
      "grad_norm": 0.21243683993816376,
      "learning_rate": 5.880269980899131e-05,
      "loss": 0.0616,
      "step": 28290
    },
    {
      "epoch": 3.3103206178329043,
      "loss_main": 0.03866193816065788,
      "step": 28290
    },
    {
      "epoch": 3.3114907559091975,
      "grad_norm": 0.2895989418029785,
      "learning_rate": 5.8775571082283465e-05,
      "loss": 0.0568,
      "step": 28300
    },
    {
      "epoch": 3.3114907559091975,
      "loss_main": 0.07269033044576645,
      "step": 28300
    },
    {
      "epoch": 3.3126608939854902,
      "grad_norm": 0.27239418029785156,
      "learning_rate": 5.8748439689788824e-05,
      "loss": 0.065,
      "step": 28310
    },
    {
      "epoch": 3.3126608939854902,
      "loss_main": 0.06389548629522324,
      "step": 28310
    },
    {
      "epoch": 3.3138310320617834,
      "grad_norm": 0.3120277225971222,
      "learning_rate": 5.87213056397492e-05,
      "loss": 0.0572,
      "step": 28320
    },
    {
      "epoch": 3.3138310320617834,
      "loss_main": 0.06161651015281677,
      "step": 28320
    },
    {
      "epoch": 3.315001170138076,
      "grad_norm": 0.28418856859207153,
      "learning_rate": 5.869416894040719e-05,
      "loss": 0.0646,
      "step": 28330
    },
    {
      "epoch": 3.315001170138076,
      "loss_main": 0.08031348139047623,
      "step": 28330
    },
    {
      "epoch": 3.3161713082143693,
      "grad_norm": 0.22516518831253052,
      "learning_rate": 5.866702960000621e-05,
      "loss": 0.0611,
      "step": 28340
    },
    {
      "epoch": 3.3161713082143693,
      "loss_main": 0.05435021594166756,
      "step": 28340
    },
    {
      "epoch": 3.317341446290662,
      "grad_norm": 0.2239789217710495,
      "learning_rate": 5.863988762679048e-05,
      "loss": 0.0648,
      "step": 28350
    },
    {
      "epoch": 3.317341446290662,
      "loss_main": 0.08029438555240631,
      "step": 28350
    },
    {
      "epoch": 3.3185115843669553,
      "grad_norm": 0.22585324943065643,
      "learning_rate": 5.8612743029005e-05,
      "loss": 0.056,
      "step": 28360
    },
    {
      "epoch": 3.3185115843669553,
      "loss_main": 0.049096785485744476,
      "step": 28360
    },
    {
      "epoch": 3.3196817224432484,
      "grad_norm": 0.279422789812088,
      "learning_rate": 5.858559581489561e-05,
      "loss": 0.0596,
      "step": 28370
    },
    {
      "epoch": 3.3196817224432484,
      "loss_main": 0.07963461428880692,
      "step": 28370
    },
    {
      "epoch": 3.320851860519541,
      "grad_norm": 0.28443723917007446,
      "learning_rate": 5.85584459927089e-05,
      "loss": 0.0594,
      "step": 28380
    },
    {
      "epoch": 3.320851860519541,
      "loss_main": 0.08439582586288452,
      "step": 28380
    },
    {
      "epoch": 3.3220219985958344,
      "grad_norm": 0.235215961933136,
      "learning_rate": 5.853129357069227e-05,
      "loss": 0.0634,
      "step": 28390
    },
    {
      "epoch": 3.3220219985958344,
      "loss_main": 0.04533209279179573,
      "step": 28390
    },
    {
      "epoch": 3.3231921366721275,
      "grad_norm": 0.3898649513721466,
      "learning_rate": 5.8504138557093913e-05,
      "loss": 0.0686,
      "step": 28400
    },
    {
      "epoch": 3.3231921366721275,
      "loss_main": 0.0760113000869751,
      "step": 28400
    },
    {
      "epoch": 3.3243622747484203,
      "grad_norm": 0.24851058423519135,
      "learning_rate": 5.8476980960162784e-05,
      "loss": 0.0652,
      "step": 28410
    },
    {
      "epoch": 3.3243622747484203,
      "loss_main": 0.07110267877578735,
      "step": 28410
    },
    {
      "epoch": 3.3255324128247135,
      "grad_norm": 0.26602160930633545,
      "learning_rate": 5.844982078814868e-05,
      "loss": 0.0621,
      "step": 28420
    },
    {
      "epoch": 3.3255324128247135,
      "loss_main": 0.04755750298500061,
      "step": 28420
    },
    {
      "epoch": 3.326702550901006,
      "grad_norm": 0.3348669409751892,
      "learning_rate": 5.842265804930211e-05,
      "loss": 0.071,
      "step": 28430
    },
    {
      "epoch": 3.326702550901006,
      "loss_main": 0.09111475944519043,
      "step": 28430
    },
    {
      "epoch": 3.3278726889772994,
      "grad_norm": 0.2929397225379944,
      "learning_rate": 5.839549275187444e-05,
      "loss": 0.0687,
      "step": 28440
    },
    {
      "epoch": 3.3278726889772994,
      "loss_main": 0.0909033790230751,
      "step": 28440
    },
    {
      "epoch": 3.329042827053592,
      "grad_norm": 0.3176235556602478,
      "learning_rate": 5.836832490411771e-05,
      "loss": 0.0706,
      "step": 28450
    },
    {
      "epoch": 3.329042827053592,
      "loss_main": 0.05303531512618065,
      "step": 28450
    },
    {
      "epoch": 3.3302129651298853,
      "grad_norm": 0.26251301169395447,
      "learning_rate": 5.834115451428485e-05,
      "loss": 0.0701,
      "step": 28460
    },
    {
      "epoch": 3.3302129651298853,
      "loss_main": 0.07091297209262848,
      "step": 28460
    },
    {
      "epoch": 3.3313831032061785,
      "grad_norm": 0.33111509680747986,
      "learning_rate": 5.831398159062946e-05,
      "loss": 0.0615,
      "step": 28470
    },
    {
      "epoch": 3.3313831032061785,
      "loss_main": 0.07356316596269608,
      "step": 28470
    },
    {
      "epoch": 3.332553241282471,
      "grad_norm": 0.2907370924949646,
      "learning_rate": 5.828680614140599e-05,
      "loss": 0.0605,
      "step": 28480
    },
    {
      "epoch": 3.332553241282471,
      "loss_main": 0.08164145052433014,
      "step": 28480
    },
    {
      "epoch": 3.3337233793587644,
      "grad_norm": 0.3417801856994629,
      "learning_rate": 5.825962817486962e-05,
      "loss": 0.0667,
      "step": 28490
    },
    {
      "epoch": 3.3337233793587644,
      "loss_main": 0.04143703356385231,
      "step": 28490
    },
    {
      "epoch": 3.3348935174350576,
      "grad_norm": 0.21075649559497833,
      "learning_rate": 5.823244769927629e-05,
      "loss": 0.0613,
      "step": 28500
    },
    {
      "epoch": 3.3348935174350576,
      "loss_main": 0.07555137574672699,
      "step": 28500
    },
    {
      "epoch": 3.3360636555113503,
      "grad_norm": 0.3208233714103699,
      "learning_rate": 5.8205264722882716e-05,
      "loss": 0.0722,
      "step": 28510
    },
    {
      "epoch": 3.3360636555113503,
      "loss_main": 0.045697230845689774,
      "step": 28510
    },
    {
      "epoch": 3.3372337935876435,
      "grad_norm": 0.2354031205177307,
      "learning_rate": 5.817807925394636e-05,
      "loss": 0.0578,
      "step": 28520
    },
    {
      "epoch": 3.3372337935876435,
      "loss_main": 0.0622004009783268,
      "step": 28520
    },
    {
      "epoch": 3.3384039316639362,
      "grad_norm": 0.2669195532798767,
      "learning_rate": 5.815089130072546e-05,
      "loss": 0.0583,
      "step": 28530
    },
    {
      "epoch": 3.3384039316639362,
      "loss_main": 0.059693992137908936,
      "step": 28530
    },
    {
      "epoch": 3.3395740697402294,
      "grad_norm": 0.386882483959198,
      "learning_rate": 5.8123700871479e-05,
      "loss": 0.0517,
      "step": 28540
    },
    {
      "epoch": 3.3395740697402294,
      "loss_main": 0.05572647601366043,
      "step": 28540
    },
    {
      "epoch": 3.340744207816522,
      "grad_norm": 0.24348406493663788,
      "learning_rate": 5.809650797446671e-05,
      "loss": 0.053,
      "step": 28550
    },
    {
      "epoch": 3.340744207816522,
      "loss_main": 0.035682130604982376,
      "step": 28550
    },
    {
      "epoch": 3.3419143458928153,
      "grad_norm": 0.2591341435909271,
      "learning_rate": 5.806931261794907e-05,
      "loss": 0.063,
      "step": 28560
    },
    {
      "epoch": 3.3419143458928153,
      "loss_main": 0.06067342311143875,
      "step": 28560
    },
    {
      "epoch": 3.3430844839691085,
      "grad_norm": 0.26765385270118713,
      "learning_rate": 5.804211481018731e-05,
      "loss": 0.0564,
      "step": 28570
    },
    {
      "epoch": 3.3430844839691085,
      "loss_main": 0.07614533603191376,
      "step": 28570
    },
    {
      "epoch": 3.3442546220454012,
      "grad_norm": 0.31923696398735046,
      "learning_rate": 5.801491455944341e-05,
      "loss": 0.0575,
      "step": 28580
    },
    {
      "epoch": 3.3442546220454012,
      "loss_main": 0.05656420439481735,
      "step": 28580
    },
    {
      "epoch": 3.3454247601216944,
      "grad_norm": 0.22379140555858612,
      "learning_rate": 5.79877118739801e-05,
      "loss": 0.0651,
      "step": 28590
    },
    {
      "epoch": 3.3454247601216944,
      "loss_main": 0.11622975021600723,
      "step": 28590
    },
    {
      "epoch": 3.3465948981979876,
      "grad_norm": 0.25337621569633484,
      "learning_rate": 5.7960506762060816e-05,
      "loss": 0.069,
      "step": 28600
    },
    {
      "epoch": 3.3465948981979876,
      "loss_main": 0.041806597262620926,
      "step": 28600
    },
    {
      "epoch": 3.3477650362742803,
      "grad_norm": 0.28739288449287415,
      "learning_rate": 5.793329923194977e-05,
      "loss": 0.057,
      "step": 28610
    },
    {
      "epoch": 3.3477650362742803,
      "loss_main": 0.05450530722737312,
      "step": 28610
    },
    {
      "epoch": 3.3489351743505735,
      "grad_norm": 0.27216628193855286,
      "learning_rate": 5.790608929191187e-05,
      "loss": 0.0591,
      "step": 28620
    },
    {
      "epoch": 3.3489351743505735,
      "loss_main": 0.08126760274171829,
      "step": 28620
    },
    {
      "epoch": 3.3501053124268663,
      "grad_norm": 0.3764667510986328,
      "learning_rate": 5.78788769502128e-05,
      "loss": 0.0764,
      "step": 28630
    },
    {
      "epoch": 3.3501053124268663,
      "loss_main": 0.05713989958167076,
      "step": 28630
    },
    {
      "epoch": 3.3512754505031594,
      "grad_norm": 0.2594020366668701,
      "learning_rate": 5.785166221511894e-05,
      "loss": 0.0639,
      "step": 28640
    },
    {
      "epoch": 3.3512754505031594,
      "loss_main": 0.046568579971790314,
      "step": 28640
    },
    {
      "epoch": 3.352445588579452,
      "grad_norm": 0.2895253598690033,
      "learning_rate": 5.7824445094897415e-05,
      "loss": 0.0652,
      "step": 28650
    },
    {
      "epoch": 3.352445588579452,
      "loss_main": 0.08624254167079926,
      "step": 28650
    },
    {
      "epoch": 3.3536157266557454,
      "grad_norm": 0.2926323413848877,
      "learning_rate": 5.7797225597816065e-05,
      "loss": 0.069,
      "step": 28660
    },
    {
      "epoch": 3.3536157266557454,
      "loss_main": 0.1279022991657257,
      "step": 28660
    },
    {
      "epoch": 3.3547858647320385,
      "grad_norm": 0.2529509365558624,
      "learning_rate": 5.777000373214345e-05,
      "loss": 0.065,
      "step": 28670
    },
    {
      "epoch": 3.3547858647320385,
      "loss_main": 0.08375991880893707,
      "step": 28670
    },
    {
      "epoch": 3.3559560028083313,
      "grad_norm": 0.328167200088501,
      "learning_rate": 5.774277950614885e-05,
      "loss": 0.0745,
      "step": 28680
    },
    {
      "epoch": 3.3559560028083313,
      "loss_main": 0.08807722479104996,
      "step": 28680
    },
    {
      "epoch": 3.3571261408846245,
      "grad_norm": 0.2339344471693039,
      "learning_rate": 5.771555292810227e-05,
      "loss": 0.0686,
      "step": 28690
    },
    {
      "epoch": 3.3571261408846245,
      "loss_main": 0.07048822194337845,
      "step": 28690
    },
    {
      "epoch": 3.358296278960917,
      "grad_norm": 0.3244026303291321,
      "learning_rate": 5.768832400627444e-05,
      "loss": 0.0576,
      "step": 28700
    },
    {
      "epoch": 3.358296278960917,
      "loss_main": 0.10488446801900864,
      "step": 28700
    },
    {
      "epoch": 3.3594664170372104,
      "grad_norm": 0.2775630056858063,
      "learning_rate": 5.7661092748936775e-05,
      "loss": 0.0608,
      "step": 28710
    },
    {
      "epoch": 3.3594664170372104,
      "loss_main": 0.061959702521562576,
      "step": 28710
    },
    {
      "epoch": 3.3606365551135036,
      "grad_norm": 0.298521488904953,
      "learning_rate": 5.76338591643614e-05,
      "loss": 0.064,
      "step": 28720
    },
    {
      "epoch": 3.3606365551135036,
      "loss_main": 0.0807960256934166,
      "step": 28720
    },
    {
      "epoch": 3.3618066931897963,
      "grad_norm": 0.4316326081752777,
      "learning_rate": 5.760662326082118e-05,
      "loss": 0.0711,
      "step": 28730
    },
    {
      "epoch": 3.3618066931897963,
      "loss_main": 0.06215415522456169,
      "step": 28730
    },
    {
      "epoch": 3.3629768312660895,
      "grad_norm": 0.4271678626537323,
      "learning_rate": 5.757938504658965e-05,
      "loss": 0.065,
      "step": 28740
    },
    {
      "epoch": 3.3629768312660895,
      "loss_main": 0.05866009742021561,
      "step": 28740
    },
    {
      "epoch": 3.364146969342382,
      "grad_norm": 0.34793493151664734,
      "learning_rate": 5.755214452994107e-05,
      "loss": 0.0687,
      "step": 28750
    },
    {
      "epoch": 3.364146969342382,
      "loss_main": 0.04508083313703537,
      "step": 28750
    },
    {
      "epoch": 3.3653171074186754,
      "grad_norm": 0.2759137451648712,
      "learning_rate": 5.752490171915039e-05,
      "loss": 0.0701,
      "step": 28760
    },
    {
      "epoch": 3.3653171074186754,
      "loss_main": 0.044621992856264114,
      "step": 28760
    },
    {
      "epoch": 3.3664872454949686,
      "grad_norm": 0.2561914920806885,
      "learning_rate": 5.749765662249324e-05,
      "loss": 0.0631,
      "step": 28770
    },
    {
      "epoch": 3.3664872454949686,
      "loss_main": 0.06479678303003311,
      "step": 28770
    },
    {
      "epoch": 3.3676573835712613,
      "grad_norm": 0.21024185419082642,
      "learning_rate": 5.747040924824596e-05,
      "loss": 0.0541,
      "step": 28780
    },
    {
      "epoch": 3.3676573835712613,
      "loss_main": 0.036039017140865326,
      "step": 28780
    },
    {
      "epoch": 3.3688275216475545,
      "grad_norm": 0.2615858316421509,
      "learning_rate": 5.7443159604685613e-05,
      "loss": 0.0597,
      "step": 28790
    },
    {
      "epoch": 3.3688275216475545,
      "loss_main": 0.062373317778110504,
      "step": 28790
    },
    {
      "epoch": 3.3699976597238472,
      "grad_norm": 0.30954089760780334,
      "learning_rate": 5.74159077000899e-05,
      "loss": 0.0693,
      "step": 28800
    },
    {
      "epoch": 3.3699976597238472,
      "loss_main": 0.04507491737604141,
      "step": 28800
    },
    {
      "epoch": 3.3711677978001404,
      "grad_norm": 0.22567956149578094,
      "learning_rate": 5.7388653542737235e-05,
      "loss": 0.0657,
      "step": 28810
    },
    {
      "epoch": 3.3711677978001404,
      "loss_main": 0.06333130598068237,
      "step": 28810
    },
    {
      "epoch": 3.3723379358764336,
      "grad_norm": 0.17385074496269226,
      "learning_rate": 5.736139714090672e-05,
      "loss": 0.0617,
      "step": 28820
    },
    {
      "epoch": 3.3723379358764336,
      "loss_main": 0.10425429046154022,
      "step": 28820
    },
    {
      "epoch": 3.3735080739527263,
      "grad_norm": 0.28725260496139526,
      "learning_rate": 5.73341385028781e-05,
      "loss": 0.0664,
      "step": 28830
    },
    {
      "epoch": 3.3735080739527263,
      "loss_main": 0.06927267462015152,
      "step": 28830
    },
    {
      "epoch": 3.3746782120290195,
      "grad_norm": 0.3459523618221283,
      "learning_rate": 5.7306877636931855e-05,
      "loss": 0.0584,
      "step": 28840
    },
    {
      "epoch": 3.3746782120290195,
      "loss_main": 0.05616152286529541,
      "step": 28840
    },
    {
      "epoch": 3.3758483501053123,
      "grad_norm": 0.2849443852901459,
      "learning_rate": 5.7279614551349125e-05,
      "loss": 0.0606,
      "step": 28850
    },
    {
      "epoch": 3.3758483501053123,
      "loss_main": 0.03805764764547348,
      "step": 28850
    },
    {
      "epoch": 3.3770184881816054,
      "grad_norm": 0.33427998423576355,
      "learning_rate": 5.725234925441169e-05,
      "loss": 0.0569,
      "step": 28860
    },
    {
      "epoch": 3.3770184881816054,
      "loss_main": 0.049086879938840866,
      "step": 28860
    },
    {
      "epoch": 3.3781886262578986,
      "grad_norm": 0.248911052942276,
      "learning_rate": 5.7225081754402044e-05,
      "loss": 0.0612,
      "step": 28870
    },
    {
      "epoch": 3.3781886262578986,
      "loss_main": 0.053883519023656845,
      "step": 28870
    },
    {
      "epoch": 3.3793587643341914,
      "grad_norm": 0.2543133497238159,
      "learning_rate": 5.7197812059603326e-05,
      "loss": 0.0598,
      "step": 28880
    },
    {
      "epoch": 3.3793587643341914,
      "loss_main": 0.07570596039295197,
      "step": 28880
    },
    {
      "epoch": 3.3805289024104845,
      "grad_norm": 0.2767349183559418,
      "learning_rate": 5.717054017829934e-05,
      "loss": 0.0672,
      "step": 28890
    },
    {
      "epoch": 3.3805289024104845,
      "loss_main": 0.06405707448720932,
      "step": 28890
    },
    {
      "epoch": 3.3816990404867773,
      "grad_norm": 0.25570738315582275,
      "learning_rate": 5.7143266118774584e-05,
      "loss": 0.0784,
      "step": 28900
    },
    {
      "epoch": 3.3816990404867773,
      "loss_main": 0.04363298416137695,
      "step": 28900
    },
    {
      "epoch": 3.3828691785630705,
      "grad_norm": 0.21352626383304596,
      "learning_rate": 5.711598988931418e-05,
      "loss": 0.0565,
      "step": 28910
    },
    {
      "epoch": 3.3828691785630705,
      "loss_main": 0.10958565026521683,
      "step": 28910
    },
    {
      "epoch": 3.3840393166393636,
      "grad_norm": 0.39310964941978455,
      "learning_rate": 5.7088711498203954e-05,
      "loss": 0.0675,
      "step": 28920
    },
    {
      "epoch": 3.3840393166393636,
      "loss_main": 0.05903524532914162,
      "step": 28920
    },
    {
      "epoch": 3.3852094547156564,
      "grad_norm": 0.2379816770553589,
      "learning_rate": 5.706143095373033e-05,
      "loss": 0.0729,
      "step": 28930
    },
    {
      "epoch": 3.3852094547156564,
      "loss_main": 0.04268563166260719,
      "step": 28930
    },
    {
      "epoch": 3.3863795927919496,
      "grad_norm": 0.31629177927970886,
      "learning_rate": 5.703414826418042e-05,
      "loss": 0.0633,
      "step": 28940
    },
    {
      "epoch": 3.3863795927919496,
      "loss_main": 0.09519098699092865,
      "step": 28940
    },
    {
      "epoch": 3.3875497308682423,
      "grad_norm": 0.26187649369239807,
      "learning_rate": 5.7006863437842007e-05,
      "loss": 0.0605,
      "step": 28950
    },
    {
      "epoch": 3.3875497308682423,
      "loss_main": 0.062483035027980804,
      "step": 28950
    },
    {
      "epoch": 3.3887198689445355,
      "grad_norm": 0.30771440267562866,
      "learning_rate": 5.697957648300348e-05,
      "loss": 0.0661,
      "step": 28960
    },
    {
      "epoch": 3.3887198689445355,
      "loss_main": 0.04026636853814125,
      "step": 28960
    },
    {
      "epoch": 3.3898900070208287,
      "grad_norm": 0.1987767517566681,
      "learning_rate": 5.695228740795391e-05,
      "loss": 0.065,
      "step": 28970
    },
    {
      "epoch": 3.3898900070208287,
      "loss_main": 0.029928157106041908,
      "step": 28970
    },
    {
      "epoch": 3.3910601450971214,
      "grad_norm": 0.2829575538635254,
      "learning_rate": 5.6924996220982985e-05,
      "loss": 0.0694,
      "step": 28980
    },
    {
      "epoch": 3.3910601450971214,
      "loss_main": 0.06388706713914871,
      "step": 28980
    },
    {
      "epoch": 3.3922302831734146,
      "grad_norm": 0.26483410596847534,
      "learning_rate": 5.6897702930381045e-05,
      "loss": 0.055,
      "step": 28990
    },
    {
      "epoch": 3.3922302831734146,
      "loss_main": 0.08011317253112793,
      "step": 28990
    },
    {
      "epoch": 3.3934004212497073,
      "grad_norm": 0.19794581830501556,
      "learning_rate": 5.687040754443908e-05,
      "loss": 0.0577,
      "step": 29000
    },
    {
      "epoch": 3.3934004212497073,
      "loss_main": 0.07496996968984604,
      "step": 29000
    },
    {
      "epoch": 3.3945705593260005,
      "grad_norm": 0.3391827642917633,
      "learning_rate": 5.6843110071448725e-05,
      "loss": 0.0656,
      "step": 29010
    },
    {
      "epoch": 3.3945705593260005,
      "loss_main": 0.06137104704976082,
      "step": 29010
    },
    {
      "epoch": 3.3957406974022932,
      "grad_norm": 0.3162307143211365,
      "learning_rate": 5.6815810519702194e-05,
      "loss": 0.066,
      "step": 29020
    },
    {
      "epoch": 3.3957406974022932,
      "loss_main": 0.07020102441310883,
      "step": 29020
    },
    {
      "epoch": 3.3969108354785864,
      "grad_norm": 0.30543646216392517,
      "learning_rate": 5.6788508897492396e-05,
      "loss": 0.0583,
      "step": 29030
    },
    {
      "epoch": 3.3969108354785864,
      "loss_main": 0.03676864132285118,
      "step": 29030
    },
    {
      "epoch": 3.3980809735548796,
      "grad_norm": 0.384393572807312,
      "learning_rate": 5.676120521311282e-05,
      "loss": 0.0592,
      "step": 29040
    },
    {
      "epoch": 3.3980809735548796,
      "loss_main": 0.07513832300901413,
      "step": 29040
    },
    {
      "epoch": 3.3992511116311723,
      "grad_norm": 0.16713324189186096,
      "learning_rate": 5.6733899474857634e-05,
      "loss": 0.0636,
      "step": 29050
    },
    {
      "epoch": 3.3992511116311723,
      "loss_main": 0.04051497206091881,
      "step": 29050
    },
    {
      "epoch": 3.4004212497074655,
      "grad_norm": 0.3254580795764923,
      "learning_rate": 5.670659169102157e-05,
      "loss": 0.0593,
      "step": 29060
    },
    {
      "epoch": 3.4004212497074655,
      "loss_main": 0.0726853758096695,
      "step": 29060
    },
    {
      "epoch": 3.4015913877837587,
      "grad_norm": 0.24214085936546326,
      "learning_rate": 5.6679281869900044e-05,
      "loss": 0.0686,
      "step": 29070
    },
    {
      "epoch": 3.4015913877837587,
      "loss_main": 0.0599205382168293,
      "step": 29070
    },
    {
      "epoch": 3.4027615258600514,
      "grad_norm": 0.29087862372398376,
      "learning_rate": 5.6651970019789045e-05,
      "loss": 0.0676,
      "step": 29080
    },
    {
      "epoch": 3.4027615258600514,
      "loss_main": 0.07297903299331665,
      "step": 29080
    },
    {
      "epoch": 3.4039316639363446,
      "grad_norm": 0.35957688093185425,
      "learning_rate": 5.662465614898519e-05,
      "loss": 0.0554,
      "step": 29090
    },
    {
      "epoch": 3.4039316639363446,
      "loss_main": 0.05210069939494133,
      "step": 29090
    },
    {
      "epoch": 3.4051018020126373,
      "grad_norm": 0.21925576031208038,
      "learning_rate": 5.6597340265785695e-05,
      "loss": 0.0722,
      "step": 29100
    },
    {
      "epoch": 3.4051018020126373,
      "loss_main": 0.04210969805717468,
      "step": 29100
    },
    {
      "epoch": 3.4062719400889305,
      "grad_norm": 0.23650656640529633,
      "learning_rate": 5.657002237848843e-05,
      "loss": 0.0639,
      "step": 29110
    },
    {
      "epoch": 3.4062719400889305,
      "loss_main": 0.03642553836107254,
      "step": 29110
    },
    {
      "epoch": 3.4074420781652233,
      "grad_norm": 0.3106103241443634,
      "learning_rate": 5.654270249539183e-05,
      "loss": 0.0584,
      "step": 29120
    },
    {
      "epoch": 3.4074420781652233,
      "loss_main": 0.06168545410037041,
      "step": 29120
    },
    {
      "epoch": 3.4086122162415164,
      "grad_norm": 0.19027850031852722,
      "learning_rate": 5.651538062479498e-05,
      "loss": 0.0667,
      "step": 29130
    },
    {
      "epoch": 3.4086122162415164,
      "loss_main": 0.10228143632411957,
      "step": 29130
    },
    {
      "epoch": 3.4097823543178096,
      "grad_norm": 0.3316936790943146,
      "learning_rate": 5.648805677499751e-05,
      "loss": 0.0722,
      "step": 29140
    },
    {
      "epoch": 3.4097823543178096,
      "loss_main": 0.0872945562005043,
      "step": 29140
    },
    {
      "epoch": 3.4109524923941024,
      "grad_norm": 0.34670498967170715,
      "learning_rate": 5.646073095429969e-05,
      "loss": 0.068,
      "step": 29150
    },
    {
      "epoch": 3.4109524923941024,
      "loss_main": 0.0691784992814064,
      "step": 29150
    },
    {
      "epoch": 3.4121226304703955,
      "grad_norm": 0.24193111062049866,
      "learning_rate": 5.643340317100241e-05,
      "loss": 0.0577,
      "step": 29160
    },
    {
      "epoch": 3.4121226304703955,
      "loss_main": 0.10745739936828613,
      "step": 29160
    },
    {
      "epoch": 3.4132927685466887,
      "grad_norm": 0.30562448501586914,
      "learning_rate": 5.64060734334071e-05,
      "loss": 0.0644,
      "step": 29170
    },
    {
      "epoch": 3.4132927685466887,
      "loss_main": 0.04899519309401512,
      "step": 29170
    },
    {
      "epoch": 3.4144629066229815,
      "grad_norm": 0.21474602818489075,
      "learning_rate": 5.637874174981583e-05,
      "loss": 0.0547,
      "step": 29180
    },
    {
      "epoch": 3.4144629066229815,
      "loss_main": 0.056739337742328644,
      "step": 29180
    },
    {
      "epoch": 3.4156330446992746,
      "grad_norm": 0.32218241691589355,
      "learning_rate": 5.635140812853124e-05,
      "loss": 0.0625,
      "step": 29190
    },
    {
      "epoch": 3.4156330446992746,
      "loss_main": 0.051304686814546585,
      "step": 29190
    },
    {
      "epoch": 3.4168031827755674,
      "grad_norm": 0.318539559841156,
      "learning_rate": 5.6324072577856544e-05,
      "loss": 0.0648,
      "step": 29200
    },
    {
      "epoch": 3.4168031827755674,
      "loss_main": 0.08793098479509354,
      "step": 29200
    },
    {
      "epoch": 3.4179733208518606,
      "grad_norm": 0.32367879152297974,
      "learning_rate": 5.629673510609559e-05,
      "loss": 0.0593,
      "step": 29210
    },
    {
      "epoch": 3.4179733208518606,
      "loss_main": 0.06036241725087166,
      "step": 29210
    },
    {
      "epoch": 3.4191434589281533,
      "grad_norm": 0.18777546286582947,
      "learning_rate": 5.626939572155276e-05,
      "loss": 0.0701,
      "step": 29220
    },
    {
      "epoch": 3.4191434589281533,
      "loss_main": 0.0955301970243454,
      "step": 29220
    },
    {
      "epoch": 3.4203135970044465,
      "grad_norm": 0.3073122501373291,
      "learning_rate": 5.6242054432533054e-05,
      "loss": 0.0627,
      "step": 29230
    },
    {
      "epoch": 3.4203135970044465,
      "loss_main": 0.06955637782812119,
      "step": 29230
    },
    {
      "epoch": 3.4214837350807397,
      "grad_norm": 0.31725114583969116,
      "learning_rate": 5.621471124734201e-05,
      "loss": 0.0654,
      "step": 29240
    },
    {
      "epoch": 3.4214837350807397,
      "loss_main": 0.09045039117336273,
      "step": 29240
    },
    {
      "epoch": 3.4226538731570324,
      "grad_norm": 0.3313106596469879,
      "learning_rate": 5.6187366174285794e-05,
      "loss": 0.0791,
      "step": 29250
    },
    {
      "epoch": 3.4226538731570324,
      "loss_main": 0.061082225292921066,
      "step": 29250
    },
    {
      "epoch": 3.4238240112333256,
      "grad_norm": 0.21417726576328278,
      "learning_rate": 5.616001922167109e-05,
      "loss": 0.0597,
      "step": 29260
    },
    {
      "epoch": 3.4238240112333256,
      "loss_main": 0.0513739176094532,
      "step": 29260
    },
    {
      "epoch": 3.4249941493096188,
      "grad_norm": 0.2753661274909973,
      "learning_rate": 5.61326703978052e-05,
      "loss": 0.0688,
      "step": 29270
    },
    {
      "epoch": 3.4249941493096188,
      "loss_main": 0.040594857186079025,
      "step": 29270
    },
    {
      "epoch": 3.4261642873859115,
      "grad_norm": 0.32589057087898254,
      "learning_rate": 5.6105319710995964e-05,
      "loss": 0.06,
      "step": 29280
    },
    {
      "epoch": 3.4261642873859115,
      "loss_main": 0.05163627862930298,
      "step": 29280
    },
    {
      "epoch": 3.4273344254622047,
      "grad_norm": 0.4126226603984833,
      "learning_rate": 5.60779671695518e-05,
      "loss": 0.067,
      "step": 29290
    },
    {
      "epoch": 3.4273344254622047,
      "loss_main": 0.052434954792261124,
      "step": 29290
    },
    {
      "epoch": 3.4285045635384974,
      "grad_norm": 0.24512144923210144,
      "learning_rate": 5.6050612781781684e-05,
      "loss": 0.062,
      "step": 29300
    },
    {
      "epoch": 3.4285045635384974,
      "loss_main": 0.03132458031177521,
      "step": 29300
    },
    {
      "epoch": 3.4296747016147906,
      "grad_norm": 0.2359691858291626,
      "learning_rate": 5.602325655599516e-05,
      "loss": 0.0624,
      "step": 29310
    },
    {
      "epoch": 3.4296747016147906,
      "loss_main": 0.08465010672807693,
      "step": 29310
    },
    {
      "epoch": 3.4308448396910833,
      "grad_norm": 0.2428569346666336,
      "learning_rate": 5.599589850050234e-05,
      "loss": 0.0577,
      "step": 29320
    },
    {
      "epoch": 3.4308448396910833,
      "loss_main": 0.07021724432706833,
      "step": 29320
    },
    {
      "epoch": 3.4320149777673765,
      "grad_norm": 0.3306892514228821,
      "learning_rate": 5.5968538623613874e-05,
      "loss": 0.0597,
      "step": 29330
    },
    {
      "epoch": 3.4320149777673765,
      "loss_main": 0.05280959606170654,
      "step": 29330
    },
    {
      "epoch": 3.4331851158436697,
      "grad_norm": 0.30820605158805847,
      "learning_rate": 5.594117693364095e-05,
      "loss": 0.0585,
      "step": 29340
    },
    {
      "epoch": 3.4331851158436697,
      "loss_main": 0.06955830752849579,
      "step": 29340
    },
    {
      "epoch": 3.4343552539199624,
      "grad_norm": 0.1948806345462799,
      "learning_rate": 5.591381343889535e-05,
      "loss": 0.0599,
      "step": 29350
    },
    {
      "epoch": 3.4343552539199624,
      "loss_main": 0.07460824400186539,
      "step": 29350
    },
    {
      "epoch": 3.4355253919962556,
      "grad_norm": 0.2231248915195465,
      "learning_rate": 5.5886448147689355e-05,
      "loss": 0.0586,
      "step": 29360
    },
    {
      "epoch": 3.4355253919962556,
      "loss_main": 0.06969796121120453,
      "step": 29360
    },
    {
      "epoch": 3.436695530072549,
      "grad_norm": 0.2848168909549713,
      "learning_rate": 5.585908106833585e-05,
      "loss": 0.0567,
      "step": 29370
    },
    {
      "epoch": 3.436695530072549,
      "loss_main": 0.036041200160980225,
      "step": 29370
    },
    {
      "epoch": 3.4378656681488415,
      "grad_norm": 0.22606134414672852,
      "learning_rate": 5.5831712209148226e-05,
      "loss": 0.0592,
      "step": 29380
    },
    {
      "epoch": 3.4378656681488415,
      "loss_main": 0.08942823857069016,
      "step": 29380
    },
    {
      "epoch": 3.4390358062251347,
      "grad_norm": 0.2229968011379242,
      "learning_rate": 5.58043415784404e-05,
      "loss": 0.0656,
      "step": 29390
    },
    {
      "epoch": 3.4390358062251347,
      "loss_main": 0.04624008387327194,
      "step": 29390
    },
    {
      "epoch": 3.4402059443014275,
      "grad_norm": 0.21359285712242126,
      "learning_rate": 5.577696918452686e-05,
      "loss": 0.0566,
      "step": 29400
    },
    {
      "epoch": 3.4402059443014275,
      "loss_main": 0.0617324598133564,
      "step": 29400
    },
    {
      "epoch": 3.4413760823777206,
      "grad_norm": 0.3588131368160248,
      "learning_rate": 5.5749595035722604e-05,
      "loss": 0.0779,
      "step": 29410
    },
    {
      "epoch": 3.4413760823777206,
      "loss_main": 0.04040857031941414,
      "step": 29410
    },
    {
      "epoch": 3.4425462204540134,
      "grad_norm": 0.2841182351112366,
      "learning_rate": 5.5722219140343193e-05,
      "loss": 0.0583,
      "step": 29420
    },
    {
      "epoch": 3.4425462204540134,
      "loss_main": 0.027271827682852745,
      "step": 29420
    },
    {
      "epoch": 3.4437163585303066,
      "grad_norm": 0.32474133372306824,
      "learning_rate": 5.56948415067047e-05,
      "loss": 0.0597,
      "step": 29430
    },
    {
      "epoch": 3.4437163585303066,
      "loss_main": 0.09442081302404404,
      "step": 29430
    },
    {
      "epoch": 3.4448864966065997,
      "grad_norm": 0.256403386592865,
      "learning_rate": 5.5667462143123704e-05,
      "loss": 0.0648,
      "step": 29440
    },
    {
      "epoch": 3.4448864966065997,
      "loss_main": 0.04895126074552536,
      "step": 29440
    },
    {
      "epoch": 3.4460566346828925,
      "grad_norm": 0.26836177706718445,
      "learning_rate": 5.564008105791737e-05,
      "loss": 0.0536,
      "step": 29450
    },
    {
      "epoch": 3.4460566346828925,
      "loss_main": 0.04968991130590439,
      "step": 29450
    },
    {
      "epoch": 3.4472267727591857,
      "grad_norm": 0.29745522141456604,
      "learning_rate": 5.5612698259403316e-05,
      "loss": 0.0676,
      "step": 29460
    },
    {
      "epoch": 3.4472267727591857,
      "loss_main": 0.06559436023235321,
      "step": 29460
    },
    {
      "epoch": 3.448396910835479,
      "grad_norm": 0.186002716422081,
      "learning_rate": 5.5585313755899724e-05,
      "loss": 0.0642,
      "step": 29470
    },
    {
      "epoch": 3.448396910835479,
      "loss_main": 0.08458036929368973,
      "step": 29470
    },
    {
      "epoch": 3.4495670489117716,
      "grad_norm": 0.3752639889717102,
      "learning_rate": 5.5557927555725285e-05,
      "loss": 0.0705,
      "step": 29480
    },
    {
      "epoch": 3.4495670489117716,
      "loss_main": 0.029774636030197144,
      "step": 29480
    },
    {
      "epoch": 3.4507371869880648,
      "grad_norm": 0.34161821007728577,
      "learning_rate": 5.55305396671992e-05,
      "loss": 0.0591,
      "step": 29490
    },
    {
      "epoch": 3.4507371869880648,
      "loss_main": 0.07034017890691757,
      "step": 29490
    },
    {
      "epoch": 3.4519073250643575,
      "grad_norm": 0.21844059228897095,
      "learning_rate": 5.55031500986412e-05,
      "loss": 0.0732,
      "step": 29500
    },
    {
      "epoch": 3.4519073250643575,
      "loss_main": 0.04986137896776199,
      "step": 29500
    },
    {
      "epoch": 3.4530774631406507,
      "grad_norm": 0.2966649532318115,
      "learning_rate": 5.547575885837149e-05,
      "loss": 0.0605,
      "step": 29510
    },
    {
      "epoch": 3.4530774631406507,
      "loss_main": 0.05665392428636551,
      "step": 29510
    },
    {
      "epoch": 3.4542476012169434,
      "grad_norm": 0.25286734104156494,
      "learning_rate": 5.5448365954710825e-05,
      "loss": 0.0661,
      "step": 29520
    },
    {
      "epoch": 3.4542476012169434,
      "loss_main": 0.08283400535583496,
      "step": 29520
    },
    {
      "epoch": 3.4554177392932366,
      "grad_norm": 0.19198259711265564,
      "learning_rate": 5.5420971395980446e-05,
      "loss": 0.0618,
      "step": 29530
    },
    {
      "epoch": 3.4554177392932366,
      "loss_main": 0.079255111515522,
      "step": 29530
    },
    {
      "epoch": 3.4565878773695298,
      "grad_norm": 0.2773069739341736,
      "learning_rate": 5.539357519050209e-05,
      "loss": 0.0705,
      "step": 29540
    },
    {
      "epoch": 3.4565878773695298,
      "loss_main": 0.11172240227460861,
      "step": 29540
    },
    {
      "epoch": 3.4577580154458225,
      "grad_norm": 0.4452371597290039,
      "learning_rate": 5.536617734659799e-05,
      "loss": 0.0666,
      "step": 29550
    },
    {
      "epoch": 3.4577580154458225,
      "loss_main": 0.05975360423326492,
      "step": 29550
    },
    {
      "epoch": 3.4589281535221157,
      "grad_norm": 0.30897024273872375,
      "learning_rate": 5.533877787259091e-05,
      "loss": 0.052,
      "step": 29560
    },
    {
      "epoch": 3.4589281535221157,
      "loss_main": 0.04955923184752464,
      "step": 29560
    },
    {
      "epoch": 3.4600982915984084,
      "grad_norm": 0.3644837737083435,
      "learning_rate": 5.5311376776804044e-05,
      "loss": 0.0615,
      "step": 29570
    },
    {
      "epoch": 3.4600982915984084,
      "loss_main": 0.03578319028019905,
      "step": 29570
    },
    {
      "epoch": 3.4612684296747016,
      "grad_norm": 0.2838604152202606,
      "learning_rate": 5.528397406756118e-05,
      "loss": 0.0631,
      "step": 29580
    },
    {
      "epoch": 3.4612684296747016,
      "loss_main": 0.06468801200389862,
      "step": 29580
    },
    {
      "epoch": 3.462438567750995,
      "grad_norm": 0.36172106862068176,
      "learning_rate": 5.525656975318652e-05,
      "loss": 0.055,
      "step": 29590
    },
    {
      "epoch": 3.462438567750995,
      "loss_main": 0.06806318461894989,
      "step": 29590
    },
    {
      "epoch": 3.4636087058272875,
      "grad_norm": 0.26283061504364014,
      "learning_rate": 5.522916384200474e-05,
      "loss": 0.0555,
      "step": 29600
    },
    {
      "epoch": 3.4636087058272875,
      "loss_main": 0.05012785643339157,
      "step": 29600
    },
    {
      "epoch": 3.4647788439035807,
      "grad_norm": 0.252401739358902,
      "learning_rate": 5.520175634234106e-05,
      "loss": 0.0559,
      "step": 29610
    },
    {
      "epoch": 3.4647788439035807,
      "loss_main": 0.0486929677426815,
      "step": 29610
    },
    {
      "epoch": 3.4659489819798734,
      "grad_norm": 0.2543348968029022,
      "learning_rate": 5.517434726252113e-05,
      "loss": 0.059,
      "step": 29620
    },
    {
      "epoch": 3.4659489819798734,
      "loss_main": 0.060196150094270706,
      "step": 29620
    },
    {
      "epoch": 3.4671191200561666,
      "grad_norm": 0.2869527339935303,
      "learning_rate": 5.514693661087113e-05,
      "loss": 0.0644,
      "step": 29630
    },
    {
      "epoch": 3.4671191200561666,
      "loss_main": 0.06880456954240799,
      "step": 29630
    },
    {
      "epoch": 3.46828925813246,
      "grad_norm": 0.35035306215286255,
      "learning_rate": 5.511952439571769e-05,
      "loss": 0.0639,
      "step": 29640
    },
    {
      "epoch": 3.46828925813246,
      "loss_main": 0.05367555841803551,
      "step": 29640
    },
    {
      "epoch": 3.4694593962087525,
      "grad_norm": 0.28621983528137207,
      "learning_rate": 5.509211062538791e-05,
      "loss": 0.0577,
      "step": 29650
    },
    {
      "epoch": 3.4694593962087525,
      "loss_main": 0.08747359365224838,
      "step": 29650
    },
    {
      "epoch": 3.4706295342850457,
      "grad_norm": 0.23798787593841553,
      "learning_rate": 5.506469530820939e-05,
      "loss": 0.0705,
      "step": 29660
    },
    {
      "epoch": 3.4706295342850457,
      "loss_main": 0.04325384646654129,
      "step": 29660
    },
    {
      "epoch": 3.4717996723613385,
      "grad_norm": 0.29509660601615906,
      "learning_rate": 5.503727845251014e-05,
      "loss": 0.0607,
      "step": 29670
    },
    {
      "epoch": 3.4717996723613385,
      "loss_main": 0.0835811123251915,
      "step": 29670
    },
    {
      "epoch": 3.4729698104376316,
      "grad_norm": 0.2127343863248825,
      "learning_rate": 5.50098600666187e-05,
      "loss": 0.0562,
      "step": 29680
    },
    {
      "epoch": 3.4729698104376316,
      "loss_main": 0.03795579820871353,
      "step": 29680
    },
    {
      "epoch": 3.474139948513925,
      "grad_norm": 0.3364923596382141,
      "learning_rate": 5.498244015886406e-05,
      "loss": 0.0584,
      "step": 29690
    },
    {
      "epoch": 3.474139948513925,
      "loss_main": 0.05007101967930794,
      "step": 29690
    },
    {
      "epoch": 3.4753100865902176,
      "grad_norm": 0.3068251311779022,
      "learning_rate": 5.495501873757565e-05,
      "loss": 0.0628,
      "step": 29700
    },
    {
      "epoch": 3.4753100865902176,
      "loss_main": 0.05510649457573891,
      "step": 29700
    },
    {
      "epoch": 3.4764802246665107,
      "grad_norm": 0.2825692296028137,
      "learning_rate": 5.492759581108336e-05,
      "loss": 0.0599,
      "step": 29710
    },
    {
      "epoch": 3.4764802246665107,
      "loss_main": 0.08404845744371414,
      "step": 29710
    },
    {
      "epoch": 3.4776503627428035,
      "grad_norm": 0.22573988139629364,
      "learning_rate": 5.490017138771759e-05,
      "loss": 0.0656,
      "step": 29720
    },
    {
      "epoch": 3.4776503627428035,
      "loss_main": 0.07727353274822235,
      "step": 29720
    },
    {
      "epoch": 3.4788205008190967,
      "grad_norm": 0.33289164304733276,
      "learning_rate": 5.487274547580912e-05,
      "loss": 0.0615,
      "step": 29730
    },
    {
      "epoch": 3.4788205008190967,
      "loss_main": 0.056644339114427567,
      "step": 29730
    },
    {
      "epoch": 3.47999063889539,
      "grad_norm": 0.3357650339603424,
      "learning_rate": 5.484531808368923e-05,
      "loss": 0.0629,
      "step": 29740
    },
    {
      "epoch": 3.47999063889539,
      "loss_main": 0.046496279537677765,
      "step": 29740
    },
    {
      "epoch": 3.4811607769716826,
      "grad_norm": 0.30950209498405457,
      "learning_rate": 5.4817889219689656e-05,
      "loss": 0.0597,
      "step": 29750
    },
    {
      "epoch": 3.4811607769716826,
      "loss_main": 0.056752342730760574,
      "step": 29750
    },
    {
      "epoch": 3.4823309150479758,
      "grad_norm": 0.3459513485431671,
      "learning_rate": 5.4790458892142536e-05,
      "loss": 0.0577,
      "step": 29760
    },
    {
      "epoch": 3.4823309150479758,
      "loss_main": 0.05220405384898186,
      "step": 29760
    },
    {
      "epoch": 3.4835010531242685,
      "grad_norm": 0.2958831787109375,
      "learning_rate": 5.476302710938048e-05,
      "loss": 0.0563,
      "step": 29770
    },
    {
      "epoch": 3.4835010531242685,
      "loss_main": 0.05134934186935425,
      "step": 29770
    },
    {
      "epoch": 3.4846711912005617,
      "grad_norm": 0.37020426988601685,
      "learning_rate": 5.473559387973657e-05,
      "loss": 0.0676,
      "step": 29780
    },
    {
      "epoch": 3.4846711912005617,
      "loss_main": 0.0872020348906517,
      "step": 29780
    },
    {
      "epoch": 3.485841329276855,
      "grad_norm": 0.26133087277412415,
      "learning_rate": 5.470815921154425e-05,
      "loss": 0.0679,
      "step": 29790
    },
    {
      "epoch": 3.485841329276855,
      "loss_main": 0.048526372760534286,
      "step": 29790
    },
    {
      "epoch": 3.4870114673531476,
      "grad_norm": 0.21253076195716858,
      "learning_rate": 5.468072311313749e-05,
      "loss": 0.051,
      "step": 29800
    },
    {
      "epoch": 3.4870114673531476,
      "loss_main": 0.046044833958148956,
      "step": 29800
    },
    {
      "epoch": 3.488181605429441,
      "grad_norm": 0.2663450241088867,
      "learning_rate": 5.465328559285063e-05,
      "loss": 0.0596,
      "step": 29810
    },
    {
      "epoch": 3.488181605429441,
      "loss_main": 0.04913065955042839,
      "step": 29810
    },
    {
      "epoch": 3.4893517435057335,
      "grad_norm": 0.3256794512271881,
      "learning_rate": 5.462584665901849e-05,
      "loss": 0.0602,
      "step": 29820
    },
    {
      "epoch": 3.4893517435057335,
      "loss_main": 0.04771256819367409,
      "step": 29820
    },
    {
      "epoch": 3.4905218815820267,
      "grad_norm": 0.2576957941055298,
      "learning_rate": 5.4598406319976235e-05,
      "loss": 0.0729,
      "step": 29830
    },
    {
      "epoch": 3.4905218815820267,
      "loss_main": 0.09766589105129242,
      "step": 29830
    },
    {
      "epoch": 3.49169201965832,
      "grad_norm": 0.34329190850257874,
      "learning_rate": 5.457096458405958e-05,
      "loss": 0.0732,
      "step": 29840
    },
    {
      "epoch": 3.49169201965832,
      "loss_main": 0.05038543790578842,
      "step": 29840
    },
    {
      "epoch": 3.4928621577346126,
      "grad_norm": 0.3560173511505127,
      "learning_rate": 5.454352145960457e-05,
      "loss": 0.0614,
      "step": 29850
    },
    {
      "epoch": 3.4928621577346126,
      "loss_main": 0.054692864418029785,
      "step": 29850
    },
    {
      "epoch": 3.494032295810906,
      "grad_norm": 0.2124066799879074,
      "learning_rate": 5.4516076954947715e-05,
      "loss": 0.0579,
      "step": 29860
    },
    {
      "epoch": 3.494032295810906,
      "loss_main": 0.0416986458003521,
      "step": 29860
    },
    {
      "epoch": 3.4952024338871985,
      "grad_norm": 0.23665516078472137,
      "learning_rate": 5.448863107842591e-05,
      "loss": 0.0614,
      "step": 29870
    },
    {
      "epoch": 3.4952024338871985,
      "loss_main": 0.055746130645275116,
      "step": 29870
    },
    {
      "epoch": 3.4963725719634917,
      "grad_norm": 0.2540479600429535,
      "learning_rate": 5.446118383837651e-05,
      "loss": 0.0623,
      "step": 29880
    },
    {
      "epoch": 3.4963725719634917,
      "loss_main": 0.07398871332406998,
      "step": 29880
    },
    {
      "epoch": 3.4975427100397845,
      "grad_norm": 0.3136679530143738,
      "learning_rate": 5.443373524313722e-05,
      "loss": 0.0643,
      "step": 29890
    },
    {
      "epoch": 3.4975427100397845,
      "loss_main": 0.049754995852708817,
      "step": 29890
    },
    {
      "epoch": 3.4987128481160776,
      "grad_norm": 0.3501001000404358,
      "learning_rate": 5.440628530104626e-05,
      "loss": 0.0574,
      "step": 29900
    },
    {
      "epoch": 3.4987128481160776,
      "loss_main": 0.03193560242652893,
      "step": 29900
    },
    {
      "epoch": 3.499882986192371,
      "grad_norm": 0.2529923617839813,
      "learning_rate": 5.4378834020442146e-05,
      "loss": 0.0619,
      "step": 29910
    },
    {
      "epoch": 3.499882986192371,
      "loss_main": 0.09275089204311371,
      "step": 29910
    },
    {
      "epoch": 3.5010531242686636,
      "grad_norm": 0.23210997879505157,
      "learning_rate": 5.4351381409663884e-05,
      "loss": 0.0582,
      "step": 29920
    },
    {
      "epoch": 3.5010531242686636,
      "loss_main": 0.05645924061536789,
      "step": 29920
    },
    {
      "epoch": 3.5022232623449567,
      "grad_norm": 0.3217625617980957,
      "learning_rate": 5.432392747705084e-05,
      "loss": 0.0592,
      "step": 29930
    },
    {
      "epoch": 3.5022232623449567,
      "loss_main": 0.04319153353571892,
      "step": 29930
    },
    {
      "epoch": 3.50339340042125,
      "grad_norm": 0.24912109971046448,
      "learning_rate": 5.429647223094278e-05,
      "loss": 0.0595,
      "step": 29940
    },
    {
      "epoch": 3.50339340042125,
      "loss_main": 0.04482930898666382,
      "step": 29940
    },
    {
      "epoch": 3.5045635384975427,
      "grad_norm": 0.2886543869972229,
      "learning_rate": 5.4269015679679924e-05,
      "loss": 0.0655,
      "step": 29950
    },
    {
      "epoch": 3.5045635384975427,
      "loss_main": 0.06835751980543137,
      "step": 29950
    },
    {
      "epoch": 3.505733676573836,
      "grad_norm": 0.26131823658943176,
      "learning_rate": 5.424155783160281e-05,
      "loss": 0.0587,
      "step": 29960
    },
    {
      "epoch": 3.505733676573836,
      "loss_main": 0.06910909712314606,
      "step": 29960
    },
    {
      "epoch": 3.5069038146501286,
      "grad_norm": 0.31591475009918213,
      "learning_rate": 5.4214098695052415e-05,
      "loss": 0.0693,
      "step": 29970
    },
    {
      "epoch": 3.5069038146501286,
      "loss_main": 0.08635596185922623,
      "step": 29970
    },
    {
      "epoch": 3.5080739527264218,
      "grad_norm": 0.43386736512184143,
      "learning_rate": 5.418663827837012e-05,
      "loss": 0.0676,
      "step": 29980
    },
    {
      "epoch": 3.5080739527264218,
      "loss_main": 0.07700785249471664,
      "step": 29980
    },
    {
      "epoch": 3.5092440908027145,
      "grad_norm": 0.33729782700538635,
      "learning_rate": 5.415917658989763e-05,
      "loss": 0.0628,
      "step": 29990
    },
    {
      "epoch": 3.5092440908027145,
      "loss_main": 0.030831443145871162,
      "step": 29990
    },
    {
      "epoch": 3.5104142288790077,
      "grad_norm": 0.4083704948425293,
      "learning_rate": 5.413171363797713e-05,
      "loss": 0.0609,
      "step": 30000
    },
    {
      "epoch": 3.5104142288790077,
      "loss_main": 0.08602285385131836,
      "step": 30000
    },
    {
      "epoch": 3.511584366955301,
      "grad_norm": 0.26302269101142883,
      "learning_rate": 5.4104249430951116e-05,
      "loss": 0.0572,
      "step": 30010
    },
    {
      "epoch": 3.511584366955301,
      "loss_main": 0.03665875270962715,
      "step": 30010
    },
    {
      "epoch": 3.5127545050315936,
      "grad_norm": 0.3174399137496948,
      "learning_rate": 5.4076783977162494e-05,
      "loss": 0.0573,
      "step": 30020
    },
    {
      "epoch": 3.5127545050315936,
      "loss_main": 0.06103399023413658,
      "step": 30020
    },
    {
      "epoch": 3.5139246431078868,
      "grad_norm": 0.27283069491386414,
      "learning_rate": 5.4049317284954525e-05,
      "loss": 0.052,
      "step": 30030
    },
    {
      "epoch": 3.5139246431078868,
      "loss_main": 0.05270949751138687,
      "step": 30030
    },
    {
      "epoch": 3.51509478118418,
      "grad_norm": 0.2641438841819763,
      "learning_rate": 5.4021849362670884e-05,
      "loss": 0.058,
      "step": 30040
    },
    {
      "epoch": 3.51509478118418,
      "loss_main": 0.044159628450870514,
      "step": 30040
    },
    {
      "epoch": 3.5162649192604727,
      "grad_norm": 0.27073198556900024,
      "learning_rate": 5.3994380218655604e-05,
      "loss": 0.0613,
      "step": 30050
    },
    {
      "epoch": 3.5162649192604727,
      "loss_main": 0.06674498319625854,
      "step": 30050
    },
    {
      "epoch": 3.517435057336766,
      "grad_norm": 0.27722829580307007,
      "learning_rate": 5.396690986125309e-05,
      "loss": 0.0578,
      "step": 30060
    },
    {
      "epoch": 3.517435057336766,
      "loss_main": 0.0596344955265522,
      "step": 30060
    },
    {
      "epoch": 3.5186051954130586,
      "grad_norm": 0.2881684899330139,
      "learning_rate": 5.3939438298808075e-05,
      "loss": 0.0589,
      "step": 30070
    },
    {
      "epoch": 3.5186051954130586,
      "loss_main": 0.08557888120412827,
      "step": 30070
    },
    {
      "epoch": 3.519775333489352,
      "grad_norm": 0.221141517162323,
      "learning_rate": 5.3911965539665744e-05,
      "loss": 0.0625,
      "step": 30080
    },
    {
      "epoch": 3.519775333489352,
      "loss_main": 0.05792059376835823,
      "step": 30080
    },
    {
      "epoch": 3.5209454715656445,
      "grad_norm": 0.26099157333374023,
      "learning_rate": 5.388449159217156e-05,
      "loss": 0.0643,
      "step": 30090
    },
    {
      "epoch": 3.5209454715656445,
      "loss_main": 0.07411090284585953,
      "step": 30090
    },
    {
      "epoch": 3.5221156096419377,
      "grad_norm": 0.24224171042442322,
      "learning_rate": 5.3857016464671385e-05,
      "loss": 0.0615,
      "step": 30100
    },
    {
      "epoch": 3.5221156096419377,
      "loss_main": 0.04952442646026611,
      "step": 30100
    },
    {
      "epoch": 3.523285747718231,
      "grad_norm": 0.32799404859542847,
      "learning_rate": 5.382954016551146e-05,
      "loss": 0.0583,
      "step": 30110
    },
    {
      "epoch": 3.523285747718231,
      "loss_main": 0.04477802291512489,
      "step": 30110
    },
    {
      "epoch": 3.5244558857945236,
      "grad_norm": 0.26925158500671387,
      "learning_rate": 5.380206270303835e-05,
      "loss": 0.0595,
      "step": 30120
    },
    {
      "epoch": 3.5244558857945236,
      "loss_main": 0.06264882534742355,
      "step": 30120
    },
    {
      "epoch": 3.525626023870817,
      "grad_norm": 0.2568187415599823,
      "learning_rate": 5.377458408559897e-05,
      "loss": 0.0655,
      "step": 30130
    },
    {
      "epoch": 3.525626023870817,
      "loss_main": 0.051378168165683746,
      "step": 30130
    },
    {
      "epoch": 3.52679616194711,
      "grad_norm": 0.30624091625213623,
      "learning_rate": 5.374710432154061e-05,
      "loss": 0.0612,
      "step": 30140
    },
    {
      "epoch": 3.52679616194711,
      "loss_main": 0.07348030060529709,
      "step": 30140
    },
    {
      "epoch": 3.5279663000234027,
      "grad_norm": 0.2400771975517273,
      "learning_rate": 5.3719623419210886e-05,
      "loss": 0.065,
      "step": 30150
    },
    {
      "epoch": 3.5279663000234027,
      "loss_main": 0.04088528826832771,
      "step": 30150
    },
    {
      "epoch": 3.529136438099696,
      "grad_norm": 0.21908730268478394,
      "learning_rate": 5.3692141386957786e-05,
      "loss": 0.0612,
      "step": 30160
    },
    {
      "epoch": 3.529136438099696,
      "loss_main": 0.0327746719121933,
      "step": 30160
    },
    {
      "epoch": 3.5303065761759886,
      "grad_norm": 0.24626250565052032,
      "learning_rate": 5.3664658233129616e-05,
      "loss": 0.0662,
      "step": 30170
    },
    {
      "epoch": 3.5303065761759886,
      "loss_main": 0.0539378896355629,
      "step": 30170
    },
    {
      "epoch": 3.531476714252282,
      "grad_norm": 0.3225553333759308,
      "learning_rate": 5.363717396607504e-05,
      "loss": 0.0631,
      "step": 30180
    },
    {
      "epoch": 3.531476714252282,
      "loss_main": 0.04613610729575157,
      "step": 30180
    },
    {
      "epoch": 3.5326468523285746,
      "grad_norm": 0.2826327681541443,
      "learning_rate": 5.360968859414305e-05,
      "loss": 0.0651,
      "step": 30190
    },
    {
      "epoch": 3.5326468523285746,
      "loss_main": 0.05300657078623772,
      "step": 30190
    },
    {
      "epoch": 3.5338169904048677,
      "grad_norm": 0.3162716329097748,
      "learning_rate": 5.358220212568295e-05,
      "loss": 0.057,
      "step": 30200
    },
    {
      "epoch": 3.5338169904048677,
      "loss_main": 0.06475012749433517,
      "step": 30200
    },
    {
      "epoch": 3.534987128481161,
      "grad_norm": 0.4446667730808258,
      "learning_rate": 5.355471456904444e-05,
      "loss": 0.0623,
      "step": 30210
    },
    {
      "epoch": 3.534987128481161,
      "loss_main": 0.0986703410744667,
      "step": 30210
    },
    {
      "epoch": 3.5361572665574537,
      "grad_norm": 0.31807971000671387,
      "learning_rate": 5.3527225932577495e-05,
      "loss": 0.0709,
      "step": 30220
    },
    {
      "epoch": 3.5361572665574537,
      "loss_main": 0.0910629853606224,
      "step": 30220
    },
    {
      "epoch": 3.537327404633747,
      "grad_norm": 0.22088900208473206,
      "learning_rate": 5.349973622463246e-05,
      "loss": 0.0701,
      "step": 30230
    },
    {
      "epoch": 3.537327404633747,
      "loss_main": 0.07504679262638092,
      "step": 30230
    },
    {
      "epoch": 3.53849754271004,
      "grad_norm": 0.48690786957740784,
      "learning_rate": 5.3472245453559956e-05,
      "loss": 0.0636,
      "step": 30240
    },
    {
      "epoch": 3.53849754271004,
      "loss_main": 0.10676383972167969,
      "step": 30240
    },
    {
      "epoch": 3.5396676807863328,
      "grad_norm": 0.18600472807884216,
      "learning_rate": 5.3444753627710955e-05,
      "loss": 0.0648,
      "step": 30250
    },
    {
      "epoch": 3.5396676807863328,
      "loss_main": 0.06863164901733398,
      "step": 30250
    },
    {
      "epoch": 3.540837818862626,
      "grad_norm": 0.3625544309616089,
      "learning_rate": 5.341726075543676e-05,
      "loss": 0.0631,
      "step": 30260
    },
    {
      "epoch": 3.540837818862626,
      "loss_main": 0.045101702213287354,
      "step": 30260
    },
    {
      "epoch": 3.5420079569389187,
      "grad_norm": 0.34894976019859314,
      "learning_rate": 5.338976684508898e-05,
      "loss": 0.0603,
      "step": 30270
    },
    {
      "epoch": 3.5420079569389187,
      "loss_main": 0.12576261162757874,
      "step": 30270
    },
    {
      "epoch": 3.543178095015212,
      "grad_norm": 0.2863130569458008,
      "learning_rate": 5.336227190501953e-05,
      "loss": 0.0611,
      "step": 30280
    },
    {
      "epoch": 3.543178095015212,
      "loss_main": 0.06736313551664352,
      "step": 30280
    },
    {
      "epoch": 3.5443482330915046,
      "grad_norm": 0.30538877844810486,
      "learning_rate": 5.3334775943580664e-05,
      "loss": 0.0671,
      "step": 30290
    },
    {
      "epoch": 3.5443482330915046,
      "loss_main": 0.0658157616853714,
      "step": 30290
    },
    {
      "epoch": 3.5455183711677978,
      "grad_norm": 0.2775125503540039,
      "learning_rate": 5.330727896912491e-05,
      "loss": 0.0608,
      "step": 30300
    },
    {
      "epoch": 3.5455183711677978,
      "loss_main": 0.049169812351465225,
      "step": 30300
    },
    {
      "epoch": 3.546688509244091,
      "grad_norm": 0.3159372806549072,
      "learning_rate": 5.327978099000511e-05,
      "loss": 0.0612,
      "step": 30310
    },
    {
      "epoch": 3.546688509244091,
      "loss_main": 0.05996384099125862,
      "step": 30310
    },
    {
      "epoch": 3.5478586473203837,
      "grad_norm": 0.24081869423389435,
      "learning_rate": 5.3252282014574465e-05,
      "loss": 0.061,
      "step": 30320
    },
    {
      "epoch": 3.5478586473203837,
      "loss_main": 0.07962174713611603,
      "step": 30320
    },
    {
      "epoch": 3.549028785396677,
      "grad_norm": 0.33886170387268066,
      "learning_rate": 5.322478205118641e-05,
      "loss": 0.0597,
      "step": 30330
    },
    {
      "epoch": 3.549028785396677,
      "loss_main": 0.06116057559847832,
      "step": 30330
    },
    {
      "epoch": 3.55019892347297,
      "grad_norm": 0.3370388150215149,
      "learning_rate": 5.3197281108194704e-05,
      "loss": 0.0675,
      "step": 30340
    },
    {
      "epoch": 3.55019892347297,
      "loss_main": 0.05779675766825676,
      "step": 30340
    },
    {
      "epoch": 3.551369061549263,
      "grad_norm": 0.3894088566303253,
      "learning_rate": 5.316977919395342e-05,
      "loss": 0.0608,
      "step": 30350
    },
    {
      "epoch": 3.551369061549263,
      "loss_main": 0.08565329015254974,
      "step": 30350
    },
    {
      "epoch": 3.552539199625556,
      "grad_norm": 0.3890315592288971,
      "learning_rate": 5.314227631681691e-05,
      "loss": 0.0627,
      "step": 30360
    },
    {
      "epoch": 3.552539199625556,
      "loss_main": 0.061455804854631424,
      "step": 30360
    },
    {
      "epoch": 3.5537093377018487,
      "grad_norm": 0.26768362522125244,
      "learning_rate": 5.311477248513982e-05,
      "loss": 0.0559,
      "step": 30370
    },
    {
      "epoch": 3.5537093377018487,
      "loss_main": 0.0757141262292862,
      "step": 30370
    },
    {
      "epoch": 3.554879475778142,
      "grad_norm": 0.2916207015514374,
      "learning_rate": 5.30872677072771e-05,
      "loss": 0.0634,
      "step": 30380
    },
    {
      "epoch": 3.554879475778142,
      "loss_main": 0.04172803834080696,
      "step": 30380
    },
    {
      "epoch": 3.5560496138544346,
      "grad_norm": 0.2676737308502197,
      "learning_rate": 5.3059761991583954e-05,
      "loss": 0.067,
      "step": 30390
    },
    {
      "epoch": 3.5560496138544346,
      "loss_main": 0.08341247588396072,
      "step": 30390
    },
    {
      "epoch": 3.557219751930728,
      "grad_norm": 0.22503939270973206,
      "learning_rate": 5.303225534641592e-05,
      "loss": 0.0601,
      "step": 30400
    },
    {
      "epoch": 3.557219751930728,
      "loss_main": 0.07888247072696686,
      "step": 30400
    },
    {
      "epoch": 3.558389890007021,
      "grad_norm": 0.36766505241394043,
      "learning_rate": 5.300474778012875e-05,
      "loss": 0.0593,
      "step": 30410
    },
    {
      "epoch": 3.558389890007021,
      "loss_main": 0.05340799316763878,
      "step": 30410
    },
    {
      "epoch": 3.5595600280833137,
      "grad_norm": 0.3301942050457001,
      "learning_rate": 5.297723930107855e-05,
      "loss": 0.0582,
      "step": 30420
    },
    {
      "epoch": 3.5595600280833137,
      "loss_main": 0.0653487965464592,
      "step": 30420
    },
    {
      "epoch": 3.560730166159607,
      "grad_norm": 0.26300889253616333,
      "learning_rate": 5.294972991762167e-05,
      "loss": 0.0632,
      "step": 30430
    },
    {
      "epoch": 3.560730166159607,
      "loss_main": 0.04598736763000488,
      "step": 30430
    },
    {
      "epoch": 3.5619003042359,
      "grad_norm": 0.22996786236763,
      "learning_rate": 5.292221963811472e-05,
      "loss": 0.0557,
      "step": 30440
    },
    {
      "epoch": 3.5619003042359,
      "loss_main": 0.05397329851984978,
      "step": 30440
    },
    {
      "epoch": 3.563070442312193,
      "grad_norm": 0.3631334602832794,
      "learning_rate": 5.28947084709146e-05,
      "loss": 0.0576,
      "step": 30450
    },
    {
      "epoch": 3.563070442312193,
      "loss_main": 0.04354057088494301,
      "step": 30450
    },
    {
      "epoch": 3.5642405803884856,
      "grad_norm": 0.21441055834293365,
      "learning_rate": 5.2867196424378465e-05,
      "loss": 0.0536,
      "step": 30460
    },
    {
      "epoch": 3.5642405803884856,
      "loss_main": 0.059561796486377716,
      "step": 30460
    },
    {
      "epoch": 3.5654107184647787,
      "grad_norm": 0.24836044013500214,
      "learning_rate": 5.2839683506863765e-05,
      "loss": 0.0663,
      "step": 30470
    },
    {
      "epoch": 3.5654107184647787,
      "loss_main": 0.04634493216872215,
      "step": 30470
    },
    {
      "epoch": 3.566580856541072,
      "grad_norm": 0.32698583602905273,
      "learning_rate": 5.281216972672821e-05,
      "loss": 0.0596,
      "step": 30480
    },
    {
      "epoch": 3.566580856541072,
      "loss_main": 0.058204445987939835,
      "step": 30480
    },
    {
      "epoch": 3.5677509946173647,
      "grad_norm": 0.24817931652069092,
      "learning_rate": 5.278465509232973e-05,
      "loss": 0.0724,
      "step": 30490
    },
    {
      "epoch": 3.5677509946173647,
      "loss_main": 0.03737803176045418,
      "step": 30490
    },
    {
      "epoch": 3.568921132693658,
      "grad_norm": 0.23015204071998596,
      "learning_rate": 5.275713961202655e-05,
      "loss": 0.0605,
      "step": 30500
    },
    {
      "epoch": 3.568921132693658,
      "loss_main": 0.058002423495054245,
      "step": 30500
    },
    {
      "epoch": 3.570091270769951,
      "grad_norm": 0.31576573848724365,
      "learning_rate": 5.2729623294177165e-05,
      "loss": 0.0552,
      "step": 30510
    },
    {
      "epoch": 3.570091270769951,
      "loss_main": 0.024205008521676064,
      "step": 30510
    },
    {
      "epoch": 3.5712614088462438,
      "grad_norm": 0.35384154319763184,
      "learning_rate": 5.270210614714028e-05,
      "loss": 0.0673,
      "step": 30520
    },
    {
      "epoch": 3.5712614088462438,
      "loss_main": 0.05230620875954628,
      "step": 30520
    },
    {
      "epoch": 3.572431546922537,
      "grad_norm": 0.26888006925582886,
      "learning_rate": 5.267458817927491e-05,
      "loss": 0.0696,
      "step": 30530
    },
    {
      "epoch": 3.572431546922537,
      "loss_main": 0.06963992118835449,
      "step": 30530
    },
    {
      "epoch": 3.57360168499883,
      "grad_norm": 0.41175347566604614,
      "learning_rate": 5.264706939894026e-05,
      "loss": 0.0595,
      "step": 30540
    },
    {
      "epoch": 3.57360168499883,
      "loss_main": 0.07125511020421982,
      "step": 30540
    },
    {
      "epoch": 3.574771823075123,
      "grad_norm": 0.29350247979164124,
      "learning_rate": 5.261954981449584e-05,
      "loss": 0.0636,
      "step": 30550
    },
    {
      "epoch": 3.574771823075123,
      "loss_main": 0.07305727899074554,
      "step": 30550
    },
    {
      "epoch": 3.5759419611514156,
      "grad_norm": 0.2302837073802948,
      "learning_rate": 5.2592029434301324e-05,
      "loss": 0.0671,
      "step": 30560
    },
    {
      "epoch": 3.5759419611514156,
      "loss_main": 0.04745492711663246,
      "step": 30560
    },
    {
      "epoch": 3.577112099227709,
      "grad_norm": 0.3253191411495209,
      "learning_rate": 5.256450826671672e-05,
      "loss": 0.06,
      "step": 30570
    },
    {
      "epoch": 3.577112099227709,
      "loss_main": 0.03895925357937813,
      "step": 30570
    },
    {
      "epoch": 3.578282237304002,
      "grad_norm": 0.31246471405029297,
      "learning_rate": 5.253698632010221e-05,
      "loss": 0.0561,
      "step": 30580
    },
    {
      "epoch": 3.578282237304002,
      "loss_main": 0.080652616918087,
      "step": 30580
    },
    {
      "epoch": 3.5794523753802947,
      "grad_norm": 0.22737012803554535,
      "learning_rate": 5.2509463602818246e-05,
      "loss": 0.0609,
      "step": 30590
    },
    {
      "epoch": 3.5794523753802947,
      "loss_main": 0.06134163588285446,
      "step": 30590
    },
    {
      "epoch": 3.580622513456588,
      "grad_norm": 0.26818346977233887,
      "learning_rate": 5.248194012322549e-05,
      "loss": 0.0561,
      "step": 30600
    },
    {
      "epoch": 3.580622513456588,
      "loss_main": 0.08702950924634933,
      "step": 30600
    },
    {
      "epoch": 3.581792651532881,
      "grad_norm": 0.35839152336120605,
      "learning_rate": 5.245441588968486e-05,
      "loss": 0.0642,
      "step": 30610
    },
    {
      "epoch": 3.581792651532881,
      "loss_main": 0.04930611327290535,
      "step": 30610
    },
    {
      "epoch": 3.582962789609174,
      "grad_norm": 0.4242972433567047,
      "learning_rate": 5.242689091055748e-05,
      "loss": 0.0619,
      "step": 30620
    },
    {
      "epoch": 3.582962789609174,
      "loss_main": 0.06644890457391739,
      "step": 30620
    },
    {
      "epoch": 3.584132927685467,
      "grad_norm": 0.27605515718460083,
      "learning_rate": 5.239936519420473e-05,
      "loss": 0.0682,
      "step": 30630
    },
    {
      "epoch": 3.584132927685467,
      "loss_main": 0.04991777613759041,
      "step": 30630
    },
    {
      "epoch": 3.5853030657617597,
      "grad_norm": 0.3278482258319855,
      "learning_rate": 5.2371838748988175e-05,
      "loss": 0.0635,
      "step": 30640
    },
    {
      "epoch": 3.5853030657617597,
      "loss_main": 0.05072254687547684,
      "step": 30640
    },
    {
      "epoch": 3.586473203838053,
      "grad_norm": 0.2970503568649292,
      "learning_rate": 5.234431158326965e-05,
      "loss": 0.0643,
      "step": 30650
    },
    {
      "epoch": 3.586473203838053,
      "loss_main": 0.07778586447238922,
      "step": 30650
    },
    {
      "epoch": 3.5876433419143456,
      "grad_norm": 0.29893508553504944,
      "learning_rate": 5.231678370541115e-05,
      "loss": 0.0596,
      "step": 30660
    },
    {
      "epoch": 3.5876433419143456,
      "loss_main": 0.08534740656614304,
      "step": 30660
    },
    {
      "epoch": 3.588813479990639,
      "grad_norm": 0.29719680547714233,
      "learning_rate": 5.228925512377495e-05,
      "loss": 0.0595,
      "step": 30670
    },
    {
      "epoch": 3.588813479990639,
      "loss_main": 0.05442241206765175,
      "step": 30670
    },
    {
      "epoch": 3.589983618066932,
      "grad_norm": 0.2681753933429718,
      "learning_rate": 5.2261725846723465e-05,
      "loss": 0.0639,
      "step": 30680
    },
    {
      "epoch": 3.589983618066932,
      "loss_main": 0.0372176431119442,
      "step": 30680
    },
    {
      "epoch": 3.5911537561432247,
      "grad_norm": 0.26628848910331726,
      "learning_rate": 5.22341958826194e-05,
      "loss": 0.0571,
      "step": 30690
    },
    {
      "epoch": 3.5911537561432247,
      "loss_main": 0.08384477347135544,
      "step": 30690
    },
    {
      "epoch": 3.592323894219518,
      "grad_norm": 0.2004813849925995,
      "learning_rate": 5.22066652398256e-05,
      "loss": 0.0577,
      "step": 30700
    },
    {
      "epoch": 3.592323894219518,
      "loss_main": 0.05811341479420662,
      "step": 30700
    },
    {
      "epoch": 3.593494032295811,
      "grad_norm": 0.25684645771980286,
      "learning_rate": 5.2179133926705185e-05,
      "loss": 0.0598,
      "step": 30710
    },
    {
      "epoch": 3.593494032295811,
      "loss_main": 0.047443900257349014,
      "step": 30710
    },
    {
      "epoch": 3.594664170372104,
      "grad_norm": 0.2557876706123352,
      "learning_rate": 5.215160195162141e-05,
      "loss": 0.058,
      "step": 30720
    },
    {
      "epoch": 3.594664170372104,
      "loss_main": 0.051302678883075714,
      "step": 30720
    },
    {
      "epoch": 3.595834308448397,
      "grad_norm": 0.28504878282546997,
      "learning_rate": 5.212406932293776e-05,
      "loss": 0.0605,
      "step": 30730
    },
    {
      "epoch": 3.595834308448397,
      "loss_main": 0.060922980308532715,
      "step": 30730
    },
    {
      "epoch": 3.5970044465246898,
      "grad_norm": 0.3253142237663269,
      "learning_rate": 5.209653604901795e-05,
      "loss": 0.0645,
      "step": 30740
    },
    {
      "epoch": 3.5970044465246898,
      "loss_main": 0.052745528519153595,
      "step": 30740
    },
    {
      "epoch": 3.598174584600983,
      "grad_norm": 0.26390182971954346,
      "learning_rate": 5.206900213822584e-05,
      "loss": 0.0624,
      "step": 30750
    },
    {
      "epoch": 3.598174584600983,
      "loss_main": 0.04913327842950821,
      "step": 30750
    },
    {
      "epoch": 3.5993447226772757,
      "grad_norm": 0.30469077825546265,
      "learning_rate": 5.204146759892551e-05,
      "loss": 0.0638,
      "step": 30760
    },
    {
      "epoch": 3.5993447226772757,
      "loss_main": 0.04977497458457947,
      "step": 30760
    },
    {
      "epoch": 3.600514860753569,
      "grad_norm": 0.2263498604297638,
      "learning_rate": 5.2013932439481216e-05,
      "loss": 0.0617,
      "step": 30770
    },
    {
      "epoch": 3.600514860753569,
      "loss_main": 0.10285468399524689,
      "step": 30770
    },
    {
      "epoch": 3.601684998829862,
      "grad_norm": 0.2612748444080353,
      "learning_rate": 5.198639666825743e-05,
      "loss": 0.0723,
      "step": 30780
    },
    {
      "epoch": 3.601684998829862,
      "loss_main": 0.06842455267906189,
      "step": 30780
    },
    {
      "epoch": 3.6028551369061548,
      "grad_norm": 0.3982399106025696,
      "learning_rate": 5.195886029361877e-05,
      "loss": 0.0641,
      "step": 30790
    },
    {
      "epoch": 3.6028551369061548,
      "loss_main": 0.07113621383905411,
      "step": 30790
    },
    {
      "epoch": 3.604025274982448,
      "grad_norm": 0.31593236327171326,
      "learning_rate": 5.193132332393009e-05,
      "loss": 0.0681,
      "step": 30800
    },
    {
      "epoch": 3.604025274982448,
      "loss_main": 0.10769133269786835,
      "step": 30800
    },
    {
      "epoch": 3.605195413058741,
      "grad_norm": 0.2583366334438324,
      "learning_rate": 5.1903785767556376e-05,
      "loss": 0.0585,
      "step": 30810
    },
    {
      "epoch": 3.605195413058741,
      "loss_main": 0.04071270301938057,
      "step": 30810
    },
    {
      "epoch": 3.606365551135034,
      "grad_norm": 0.23377157747745514,
      "learning_rate": 5.187624763286282e-05,
      "loss": 0.0533,
      "step": 30820
    },
    {
      "epoch": 3.606365551135034,
      "loss_main": 0.06097371131181717,
      "step": 30820
    },
    {
      "epoch": 3.607535689211327,
      "grad_norm": 0.3638167679309845,
      "learning_rate": 5.184870892821475e-05,
      "loss": 0.0698,
      "step": 30830
    },
    {
      "epoch": 3.607535689211327,
      "loss_main": 0.04559377208352089,
      "step": 30830
    },
    {
      "epoch": 3.60870582728762,
      "grad_norm": 0.2154771238565445,
      "learning_rate": 5.182116966197773e-05,
      "loss": 0.06,
      "step": 30840
    },
    {
      "epoch": 3.60870582728762,
      "loss_main": 0.03832527622580528,
      "step": 30840
    },
    {
      "epoch": 3.609875965363913,
      "grad_norm": 0.39832255244255066,
      "learning_rate": 5.1793629842517466e-05,
      "loss": 0.0553,
      "step": 30850
    },
    {
      "epoch": 3.609875965363913,
      "loss_main": 0.05582469701766968,
      "step": 30850
    },
    {
      "epoch": 3.6110461034402057,
      "grad_norm": 0.31484392285346985,
      "learning_rate": 5.17660894781998e-05,
      "loss": 0.0569,
      "step": 30860
    },
    {
      "epoch": 3.6110461034402057,
      "loss_main": 0.05359841138124466,
      "step": 30860
    },
    {
      "epoch": 3.612216241516499,
      "grad_norm": 0.27730610966682434,
      "learning_rate": 5.173854857739079e-05,
      "loss": 0.0637,
      "step": 30870
    },
    {
      "epoch": 3.612216241516499,
      "loss_main": 0.03825262561440468,
      "step": 30870
    },
    {
      "epoch": 3.613386379592792,
      "grad_norm": 0.23371869325637817,
      "learning_rate": 5.171100714845661e-05,
      "loss": 0.0553,
      "step": 30880
    },
    {
      "epoch": 3.613386379592792,
      "loss_main": 0.056082289665937424,
      "step": 30880
    },
    {
      "epoch": 3.614556517669085,
      "grad_norm": 0.3158968687057495,
      "learning_rate": 5.1683465199763646e-05,
      "loss": 0.0718,
      "step": 30890
    },
    {
      "epoch": 3.614556517669085,
      "loss_main": 0.05409478396177292,
      "step": 30890
    },
    {
      "epoch": 3.615726655745378,
      "grad_norm": 0.32922133803367615,
      "learning_rate": 5.16559227396784e-05,
      "loss": 0.0568,
      "step": 30900
    },
    {
      "epoch": 3.615726655745378,
      "loss_main": 0.06432871520519257,
      "step": 30900
    },
    {
      "epoch": 3.616896793821671,
      "grad_norm": 0.26289841532707214,
      "learning_rate": 5.1628379776567556e-05,
      "loss": 0.0648,
      "step": 30910
    },
    {
      "epoch": 3.616896793821671,
      "loss_main": 0.0388699509203434,
      "step": 30910
    },
    {
      "epoch": 3.618066931897964,
      "grad_norm": 0.21329255402088165,
      "learning_rate": 5.160083631879792e-05,
      "loss": 0.0521,
      "step": 30920
    },
    {
      "epoch": 3.618066931897964,
      "loss_main": 0.06398522108793259,
      "step": 30920
    },
    {
      "epoch": 3.619237069974257,
      "grad_norm": 0.37356022000312805,
      "learning_rate": 5.1573292374736484e-05,
      "loss": 0.0604,
      "step": 30930
    },
    {
      "epoch": 3.619237069974257,
      "loss_main": 0.054796744138002396,
      "step": 30930
    },
    {
      "epoch": 3.62040720805055,
      "grad_norm": 0.19438804686069489,
      "learning_rate": 5.1545747952750356e-05,
      "loss": 0.0599,
      "step": 30940
    },
    {
      "epoch": 3.62040720805055,
      "loss_main": 0.043281905353069305,
      "step": 30940
    },
    {
      "epoch": 3.621577346126843,
      "grad_norm": 0.2262945920228958,
      "learning_rate": 5.151820306120682e-05,
      "loss": 0.0586,
      "step": 30950
    },
    {
      "epoch": 3.621577346126843,
      "loss_main": 0.05464768037199974,
      "step": 30950
    },
    {
      "epoch": 3.6227474842031357,
      "grad_norm": 0.21685808897018433,
      "learning_rate": 5.149065770847328e-05,
      "loss": 0.05,
      "step": 30960
    },
    {
      "epoch": 3.6227474842031357,
      "loss_main": 0.06710758805274963,
      "step": 30960
    },
    {
      "epoch": 3.623917622279429,
      "grad_norm": 0.31276559829711914,
      "learning_rate": 5.1463111902917297e-05,
      "loss": 0.0598,
      "step": 30970
    },
    {
      "epoch": 3.623917622279429,
      "loss_main": 0.048375699669122696,
      "step": 30970
    },
    {
      "epoch": 3.625087760355722,
      "grad_norm": 0.263382226228714,
      "learning_rate": 5.143556565290654e-05,
      "loss": 0.0645,
      "step": 30980
    },
    {
      "epoch": 3.625087760355722,
      "loss_main": 0.09323499351739883,
      "step": 30980
    },
    {
      "epoch": 3.626257898432015,
      "grad_norm": 0.181178018450737,
      "learning_rate": 5.140801896680882e-05,
      "loss": 0.0573,
      "step": 30990
    },
    {
      "epoch": 3.626257898432015,
      "loss_main": 0.0691162645816803,
      "step": 30990
    },
    {
      "epoch": 3.627428036508308,
      "grad_norm": 0.28913792967796326,
      "learning_rate": 5.1380471852992144e-05,
      "loss": 0.0596,
      "step": 31000
    },
    {
      "epoch": 3.627428036508308,
      "loss_main": 0.03704138472676277,
      "step": 31000
    },
    {
      "epoch": 3.628598174584601,
      "grad_norm": 0.37990325689315796,
      "learning_rate": 5.135292431982457e-05,
      "loss": 0.0574,
      "step": 31010
    },
    {
      "epoch": 3.628598174584601,
      "loss_main": 0.04072282090783119,
      "step": 31010
    },
    {
      "epoch": 3.629768312660894,
      "grad_norm": 0.3411773443222046,
      "learning_rate": 5.1325376375674294e-05,
      "loss": 0.0587,
      "step": 31020
    },
    {
      "epoch": 3.629768312660894,
      "loss_main": 0.06490345299243927,
      "step": 31020
    },
    {
      "epoch": 3.630938450737187,
      "grad_norm": 0.33051320910453796,
      "learning_rate": 5.129782802890968e-05,
      "loss": 0.0642,
      "step": 31030
    },
    {
      "epoch": 3.630938450737187,
      "loss_main": 0.09227191656827927,
      "step": 31030
    },
    {
      "epoch": 3.63210858881348,
      "grad_norm": 0.26440590620040894,
      "learning_rate": 5.127027928789916e-05,
      "loss": 0.0679,
      "step": 31040
    },
    {
      "epoch": 3.63210858881348,
      "loss_main": 0.03262655809521675,
      "step": 31040
    },
    {
      "epoch": 3.633278726889773,
      "grad_norm": 0.31578412652015686,
      "learning_rate": 5.124273016101135e-05,
      "loss": 0.0717,
      "step": 31050
    },
    {
      "epoch": 3.633278726889773,
      "loss_main": 0.04756145924329758,
      "step": 31050
    },
    {
      "epoch": 3.634448864966066,
      "grad_norm": 0.25549545884132385,
      "learning_rate": 5.121518065661492e-05,
      "loss": 0.0634,
      "step": 31060
    },
    {
      "epoch": 3.634448864966066,
      "loss_main": 0.03219708427786827,
      "step": 31060
    },
    {
      "epoch": 3.635619003042359,
      "grad_norm": 0.2794785499572754,
      "learning_rate": 5.11876307830787e-05,
      "loss": 0.0629,
      "step": 31070
    },
    {
      "epoch": 3.635619003042359,
      "loss_main": 0.05559263750910759,
      "step": 31070
    },
    {
      "epoch": 3.636789141118652,
      "grad_norm": 0.2732904255390167,
      "learning_rate": 5.1160080548771596e-05,
      "loss": 0.0661,
      "step": 31080
    },
    {
      "epoch": 3.636789141118652,
      "loss_main": 0.06588108092546463,
      "step": 31080
    },
    {
      "epoch": 3.637959279194945,
      "grad_norm": 0.31482356786727905,
      "learning_rate": 5.1132529962062656e-05,
      "loss": 0.0574,
      "step": 31090
    },
    {
      "epoch": 3.637959279194945,
      "loss_main": 0.059465259313583374,
      "step": 31090
    },
    {
      "epoch": 3.639129417271238,
      "grad_norm": 0.2967311143875122,
      "learning_rate": 5.110497903132101e-05,
      "loss": 0.0646,
      "step": 31100
    },
    {
      "epoch": 3.639129417271238,
      "loss_main": 0.05969769507646561,
      "step": 31100
    },
    {
      "epoch": 3.6402995553475312,
      "grad_norm": 0.28699004650115967,
      "learning_rate": 5.107742776491592e-05,
      "loss": 0.0588,
      "step": 31110
    },
    {
      "epoch": 3.6402995553475312,
      "loss_main": 0.04313988238573074,
      "step": 31110
    },
    {
      "epoch": 3.641469693423824,
      "grad_norm": 0.2941208779811859,
      "learning_rate": 5.104987617121673e-05,
      "loss": 0.055,
      "step": 31120
    },
    {
      "epoch": 3.641469693423824,
      "loss_main": 0.05178377404808998,
      "step": 31120
    },
    {
      "epoch": 3.642639831500117,
      "grad_norm": 0.2208051234483719,
      "learning_rate": 5.102232425859287e-05,
      "loss": 0.0551,
      "step": 31130
    },
    {
      "epoch": 3.642639831500117,
      "loss_main": 0.03854471445083618,
      "step": 31130
    },
    {
      "epoch": 3.64380996957641,
      "grad_norm": 0.20316100120544434,
      "learning_rate": 5.09947720354139e-05,
      "loss": 0.0522,
      "step": 31140
    },
    {
      "epoch": 3.64380996957641,
      "loss_main": 0.06203296408057213,
      "step": 31140
    },
    {
      "epoch": 3.644980107652703,
      "grad_norm": 0.24838042259216309,
      "learning_rate": 5.096721951004942e-05,
      "loss": 0.059,
      "step": 31150
    },
    {
      "epoch": 3.644980107652703,
      "loss_main": 0.028466353192925453,
      "step": 31150
    },
    {
      "epoch": 3.646150245728996,
      "grad_norm": 0.23084081709384918,
      "learning_rate": 5.0939666690869227e-05,
      "loss": 0.0598,
      "step": 31160
    },
    {
      "epoch": 3.646150245728996,
      "loss_main": 0.06073993444442749,
      "step": 31160
    },
    {
      "epoch": 3.647320383805289,
      "grad_norm": 0.3434758484363556,
      "learning_rate": 5.0912113586243096e-05,
      "loss": 0.0615,
      "step": 31170
    },
    {
      "epoch": 3.647320383805289,
      "loss_main": 0.054600510746240616,
      "step": 31170
    },
    {
      "epoch": 3.648490521881582,
      "grad_norm": 0.27066344022750854,
      "learning_rate": 5.0884560204540935e-05,
      "loss": 0.0595,
      "step": 31180
    },
    {
      "epoch": 3.648490521881582,
      "loss_main": 0.06453351676464081,
      "step": 31180
    },
    {
      "epoch": 3.649660659957875,
      "grad_norm": 0.27698084712028503,
      "learning_rate": 5.0857006554132736e-05,
      "loss": 0.0674,
      "step": 31190
    },
    {
      "epoch": 3.649660659957875,
      "loss_main": 0.05735340714454651,
      "step": 31190
    },
    {
      "epoch": 3.650830798034168,
      "grad_norm": 0.25215691328048706,
      "learning_rate": 5.0829452643388575e-05,
      "loss": 0.0512,
      "step": 31200
    },
    {
      "epoch": 3.650830798034168,
      "loss_main": 0.08097636699676514,
      "step": 31200
    },
    {
      "epoch": 3.6520009361104613,
      "grad_norm": 0.2828115224838257,
      "learning_rate": 5.08018984806786e-05,
      "loss": 0.0594,
      "step": 31210
    },
    {
      "epoch": 3.6520009361104613,
      "loss_main": 0.056852661073207855,
      "step": 31210
    },
    {
      "epoch": 3.653171074186754,
      "grad_norm": 0.356723427772522,
      "learning_rate": 5.0774344074373036e-05,
      "loss": 0.0639,
      "step": 31220
    },
    {
      "epoch": 3.653171074186754,
      "loss_main": 0.058996062725782394,
      "step": 31220
    },
    {
      "epoch": 3.654341212263047,
      "grad_norm": 0.2569519579410553,
      "learning_rate": 5.07467894328422e-05,
      "loss": 0.0632,
      "step": 31230
    },
    {
      "epoch": 3.654341212263047,
      "loss_main": 0.06815213710069656,
      "step": 31230
    },
    {
      "epoch": 3.65551135033934,
      "grad_norm": 0.2616412937641144,
      "learning_rate": 5.0719234564456454e-05,
      "loss": 0.0581,
      "step": 31240
    },
    {
      "epoch": 3.65551135033934,
      "loss_main": 0.06806154549121857,
      "step": 31240
    },
    {
      "epoch": 3.656681488415633,
      "grad_norm": 0.2399865835905075,
      "learning_rate": 5.0691679477586216e-05,
      "loss": 0.0549,
      "step": 31250
    },
    {
      "epoch": 3.656681488415633,
      "loss_main": 0.03692034259438515,
      "step": 31250
    },
    {
      "epoch": 3.657851626491926,
      "grad_norm": 0.20638269186019897,
      "learning_rate": 5.0664124180602035e-05,
      "loss": 0.0591,
      "step": 31260
    },
    {
      "epoch": 3.657851626491926,
      "loss_main": 0.04895098879933357,
      "step": 31260
    },
    {
      "epoch": 3.659021764568219,
      "grad_norm": 0.30738309025764465,
      "learning_rate": 5.063656868187447e-05,
      "loss": 0.0555,
      "step": 31270
    },
    {
      "epoch": 3.659021764568219,
      "loss_main": 0.038509052246809006,
      "step": 31270
    },
    {
      "epoch": 3.660191902644512,
      "grad_norm": 0.2202012538909912,
      "learning_rate": 5.060901298977413e-05,
      "loss": 0.0587,
      "step": 31280
    },
    {
      "epoch": 3.660191902644512,
      "loss_main": 0.0775921419262886,
      "step": 31280
    },
    {
      "epoch": 3.661362040720805,
      "grad_norm": 0.3475134074687958,
      "learning_rate": 5.0581457112671725e-05,
      "loss": 0.0725,
      "step": 31290
    },
    {
      "epoch": 3.661362040720805,
      "loss_main": 0.04428611323237419,
      "step": 31290
    },
    {
      "epoch": 3.662532178797098,
      "grad_norm": 0.2197001874446869,
      "learning_rate": 5.0553901058938016e-05,
      "loss": 0.0565,
      "step": 31300
    },
    {
      "epoch": 3.662532178797098,
      "loss_main": 0.05475142225623131,
      "step": 31300
    },
    {
      "epoch": 3.6637023168733913,
      "grad_norm": 0.2895156741142273,
      "learning_rate": 5.052634483694377e-05,
      "loss": 0.0628,
      "step": 31310
    },
    {
      "epoch": 3.6637023168733913,
      "loss_main": 0.033902402967214584,
      "step": 31310
    },
    {
      "epoch": 3.664872454949684,
      "grad_norm": 0.23722894489765167,
      "learning_rate": 5.049878845505988e-05,
      "loss": 0.0615,
      "step": 31320
    },
    {
      "epoch": 3.664872454949684,
      "loss_main": 0.06862770020961761,
      "step": 31320
    },
    {
      "epoch": 3.666042593025977,
      "grad_norm": 0.21396134793758392,
      "learning_rate": 5.047123192165721e-05,
      "loss": 0.06,
      "step": 31330
    },
    {
      "epoch": 3.666042593025977,
      "loss_main": 0.05532034486532211,
      "step": 31330
    },
    {
      "epoch": 3.66721273110227,
      "grad_norm": 0.31398504972457886,
      "learning_rate": 5.0443675245106735e-05,
      "loss": 0.0645,
      "step": 31340
    },
    {
      "epoch": 3.66721273110227,
      "loss_main": 0.034235548228025436,
      "step": 31340
    },
    {
      "epoch": 3.668382869178563,
      "grad_norm": 0.2736360728740692,
      "learning_rate": 5.0416118433779426e-05,
      "loss": 0.0554,
      "step": 31350
    },
    {
      "epoch": 3.668382869178563,
      "loss_main": 0.09221148490905762,
      "step": 31350
    },
    {
      "epoch": 3.669553007254856,
      "grad_norm": 0.2888188362121582,
      "learning_rate": 5.038856149604633e-05,
      "loss": 0.061,
      "step": 31360
    },
    {
      "epoch": 3.669553007254856,
      "loss_main": 0.06487574428319931,
      "step": 31360
    },
    {
      "epoch": 3.670723145331149,
      "grad_norm": 0.2508819103240967,
      "learning_rate": 5.03610044402785e-05,
      "loss": 0.0585,
      "step": 31370
    },
    {
      "epoch": 3.670723145331149,
      "loss_main": 0.04199234023690224,
      "step": 31370
    },
    {
      "epoch": 3.6718932834074423,
      "grad_norm": 0.3361039161682129,
      "learning_rate": 5.033344727484707e-05,
      "loss": 0.068,
      "step": 31380
    },
    {
      "epoch": 3.6718932834074423,
      "loss_main": 0.09243428707122803,
      "step": 31380
    },
    {
      "epoch": 3.673063421483735,
      "grad_norm": 0.29223397374153137,
      "learning_rate": 5.030589000812315e-05,
      "loss": 0.0607,
      "step": 31390
    },
    {
      "epoch": 3.673063421483735,
      "loss_main": 0.055497847497463226,
      "step": 31390
    },
    {
      "epoch": 3.674233559560028,
      "grad_norm": 0.24644841253757477,
      "learning_rate": 5.027833264847793e-05,
      "loss": 0.0632,
      "step": 31400
    },
    {
      "epoch": 3.674233559560028,
      "loss_main": 0.05130988359451294,
      "step": 31400
    },
    {
      "epoch": 3.6754036976363214,
      "grad_norm": 0.264703631401062,
      "learning_rate": 5.025077520428258e-05,
      "loss": 0.0704,
      "step": 31410
    },
    {
      "epoch": 3.6754036976363214,
      "loss_main": 0.032189544290304184,
      "step": 31410
    },
    {
      "epoch": 3.676573835712614,
      "grad_norm": 0.3686770796775818,
      "learning_rate": 5.022321768390837e-05,
      "loss": 0.0551,
      "step": 31420
    },
    {
      "epoch": 3.676573835712614,
      "loss_main": 0.0811263918876648,
      "step": 31420
    },
    {
      "epoch": 3.677743973788907,
      "grad_norm": 0.4514395594596863,
      "learning_rate": 5.0195660095726516e-05,
      "loss": 0.0689,
      "step": 31430
    },
    {
      "epoch": 3.677743973788907,
      "loss_main": 0.05887746438384056,
      "step": 31430
    },
    {
      "epoch": 3.6789141118652,
      "grad_norm": 0.24578537046909332,
      "learning_rate": 5.016810244810829e-05,
      "loss": 0.0582,
      "step": 31440
    },
    {
      "epoch": 3.6789141118652,
      "loss_main": 0.09030653536319733,
      "step": 31440
    },
    {
      "epoch": 3.680084249941493,
      "grad_norm": 0.33408328890800476,
      "learning_rate": 5.0140544749424976e-05,
      "loss": 0.0559,
      "step": 31450
    },
    {
      "epoch": 3.680084249941493,
      "loss_main": 0.03728374466300011,
      "step": 31450
    },
    {
      "epoch": 3.681254388017786,
      "grad_norm": 0.34123530983924866,
      "learning_rate": 5.0112987008047874e-05,
      "loss": 0.0559,
      "step": 31460
    },
    {
      "epoch": 3.681254388017786,
      "loss_main": 0.079587422311306,
      "step": 31460
    },
    {
      "epoch": 3.682424526094079,
      "grad_norm": 0.23058538138866425,
      "learning_rate": 5.008542923234831e-05,
      "loss": 0.0662,
      "step": 31470
    },
    {
      "epoch": 3.682424526094079,
      "loss_main": 0.08133934438228607,
      "step": 31470
    },
    {
      "epoch": 3.6835946641703723,
      "grad_norm": 0.2808308005332947,
      "learning_rate": 5.00578714306976e-05,
      "loss": 0.0605,
      "step": 31480
    },
    {
      "epoch": 3.6835946641703723,
      "loss_main": 0.09482767432928085,
      "step": 31480
    },
    {
      "epoch": 3.684764802246665,
      "grad_norm": 0.23465724289417267,
      "learning_rate": 5.0030313611467084e-05,
      "loss": 0.0664,
      "step": 31490
    },
    {
      "epoch": 3.684764802246665,
      "loss_main": 0.06763606518507004,
      "step": 31490
    },
    {
      "epoch": 3.685934940322958,
      "grad_norm": 0.2637960910797119,
      "learning_rate": 5.0002755783028074e-05,
      "loss": 0.0629,
      "step": 31500
    },
    {
      "epoch": 3.685934940322958,
      "loss_main": 0.05740020424127579,
      "step": 31500
    },
    {
      "epoch": 3.687105078399251,
      "grad_norm": 0.3870232105255127,
      "learning_rate": 4.997519795375194e-05,
      "loss": 0.0624,
      "step": 31510
    },
    {
      "epoch": 3.687105078399251,
      "loss_main": 0.034587059170007706,
      "step": 31510
    },
    {
      "epoch": 3.688275216475544,
      "grad_norm": 0.23665033280849457,
      "learning_rate": 4.9947640132010016e-05,
      "loss": 0.0592,
      "step": 31520
    },
    {
      "epoch": 3.688275216475544,
      "loss_main": 0.11355619877576828,
      "step": 31520
    },
    {
      "epoch": 3.689445354551837,
      "grad_norm": 0.3447299003601074,
      "learning_rate": 4.9920082326173625e-05,
      "loss": 0.0607,
      "step": 31530
    },
    {
      "epoch": 3.689445354551837,
      "loss_main": 0.11761993169784546,
      "step": 31530
    },
    {
      "epoch": 3.69061549262813,
      "grad_norm": 0.3231283128261566,
      "learning_rate": 4.9892524544614114e-05,
      "loss": 0.0682,
      "step": 31540
    },
    {
      "epoch": 3.69061549262813,
      "loss_main": 0.06585436314344406,
      "step": 31540
    },
    {
      "epoch": 3.6917856307044232,
      "grad_norm": 0.2745772898197174,
      "learning_rate": 4.986496679570283e-05,
      "loss": 0.0634,
      "step": 31550
    },
    {
      "epoch": 3.6917856307044232,
      "loss_main": 0.042322881519794464,
      "step": 31550
    },
    {
      "epoch": 3.692955768780716,
      "grad_norm": 0.30238860845565796,
      "learning_rate": 4.983740908781105e-05,
      "loss": 0.0611,
      "step": 31560
    },
    {
      "epoch": 3.692955768780716,
      "loss_main": 0.026152240112423897,
      "step": 31560
    },
    {
      "epoch": 3.694125906857009,
      "grad_norm": 0.4502315819263458,
      "learning_rate": 4.9809851429310116e-05,
      "loss": 0.0547,
      "step": 31570
    },
    {
      "epoch": 3.694125906857009,
      "loss_main": 0.07719476521015167,
      "step": 31570
    },
    {
      "epoch": 3.6952960449333023,
      "grad_norm": 0.4366942346096039,
      "learning_rate": 4.9782293828571275e-05,
      "loss": 0.0646,
      "step": 31580
    },
    {
      "epoch": 3.6952960449333023,
      "loss_main": 0.05228758603334427,
      "step": 31580
    },
    {
      "epoch": 3.696466183009595,
      "grad_norm": 0.2057669311761856,
      "learning_rate": 4.9754736293965846e-05,
      "loss": 0.0591,
      "step": 31590
    },
    {
      "epoch": 3.696466183009595,
      "loss_main": 0.06947793066501617,
      "step": 31590
    },
    {
      "epoch": 3.6976363210858882,
      "grad_norm": 0.2980431914329529,
      "learning_rate": 4.972717883386502e-05,
      "loss": 0.0653,
      "step": 31600
    },
    {
      "epoch": 3.6976363210858882,
      "loss_main": 0.05978551506996155,
      "step": 31600
    },
    {
      "epoch": 3.698806459162181,
      "grad_norm": 0.35916760563850403,
      "learning_rate": 4.9699621456640075e-05,
      "loss": 0.0636,
      "step": 31610
    },
    {
      "epoch": 3.698806459162181,
      "loss_main": 0.033132582902908325,
      "step": 31610
    },
    {
      "epoch": 3.699976597238474,
      "grad_norm": 0.2875072956085205,
      "learning_rate": 4.9672064170662214e-05,
      "loss": 0.0632,
      "step": 31620
    },
    {
      "epoch": 3.699976597238474,
      "loss_main": 0.03469708189368248,
      "step": 31620
    },
    {
      "epoch": 3.701146735314767,
      "grad_norm": 0.26067042350769043,
      "learning_rate": 4.9644506984302583e-05,
      "loss": 0.0746,
      "step": 31630
    },
    {
      "epoch": 3.701146735314767,
      "loss_main": 0.033152345567941666,
      "step": 31630
    },
    {
      "epoch": 3.70231687339106,
      "grad_norm": 0.3226575255393982,
      "learning_rate": 4.9616949905932356e-05,
      "loss": 0.0592,
      "step": 31640
    },
    {
      "epoch": 3.70231687339106,
      "loss_main": 0.05444237217307091,
      "step": 31640
    },
    {
      "epoch": 3.7034870114673533,
      "grad_norm": 0.2929440438747406,
      "learning_rate": 4.9589392943922615e-05,
      "loss": 0.0641,
      "step": 31650
    },
    {
      "epoch": 3.7034870114673533,
      "loss_main": 0.04812541976571083,
      "step": 31650
    },
    {
      "epoch": 3.704657149543646,
      "grad_norm": 0.23940570652484894,
      "learning_rate": 4.956183610664447e-05,
      "loss": 0.0653,
      "step": 31660
    },
    {
      "epoch": 3.704657149543646,
      "loss_main": 0.057892654091119766,
      "step": 31660
    },
    {
      "epoch": 3.705827287619939,
      "grad_norm": 0.22166065871715546,
      "learning_rate": 4.9534279402468945e-05,
      "loss": 0.0563,
      "step": 31670
    },
    {
      "epoch": 3.705827287619939,
      "loss_main": 0.11840862035751343,
      "step": 31670
    },
    {
      "epoch": 3.7069974256962324,
      "grad_norm": 0.25275686383247375,
      "learning_rate": 4.9506722839767036e-05,
      "loss": 0.0614,
      "step": 31680
    },
    {
      "epoch": 3.7069974256962324,
      "loss_main": 0.04887789487838745,
      "step": 31680
    },
    {
      "epoch": 3.708167563772525,
      "grad_norm": 0.4262634813785553,
      "learning_rate": 4.947916642690972e-05,
      "loss": 0.0667,
      "step": 31690
    },
    {
      "epoch": 3.708167563772525,
      "loss_main": 0.06347815692424774,
      "step": 31690
    },
    {
      "epoch": 3.7093377018488183,
      "grad_norm": 0.24078865349292755,
      "learning_rate": 4.9451610172267874e-05,
      "loss": 0.0558,
      "step": 31700
    },
    {
      "epoch": 3.7093377018488183,
      "loss_main": 0.036515917629003525,
      "step": 31700
    },
    {
      "epoch": 3.710507839925111,
      "grad_norm": 0.4138122797012329,
      "learning_rate": 4.9424054084212376e-05,
      "loss": 0.058,
      "step": 31710
    },
    {
      "epoch": 3.710507839925111,
      "loss_main": 0.032917894423007965,
      "step": 31710
    },
    {
      "epoch": 3.711677978001404,
      "grad_norm": 0.2588628828525543,
      "learning_rate": 4.939649817111407e-05,
      "loss": 0.0541,
      "step": 31720
    },
    {
      "epoch": 3.711677978001404,
      "loss_main": 0.059371668845415115,
      "step": 31720
    },
    {
      "epoch": 3.712848116077697,
      "grad_norm": 0.28771793842315674,
      "learning_rate": 4.936894244134365e-05,
      "loss": 0.0555,
      "step": 31730
    },
    {
      "epoch": 3.712848116077697,
      "loss_main": 0.08470789343118668,
      "step": 31730
    },
    {
      "epoch": 3.71401825415399,
      "grad_norm": 0.26363489031791687,
      "learning_rate": 4.9341386903271886e-05,
      "loss": 0.0552,
      "step": 31740
    },
    {
      "epoch": 3.71401825415399,
      "loss_main": 0.11052966117858887,
      "step": 31740
    },
    {
      "epoch": 3.7151883922302833,
      "grad_norm": 0.3149210214614868,
      "learning_rate": 4.931383156526936e-05,
      "loss": 0.076,
      "step": 31750
    },
    {
      "epoch": 3.7151883922302833,
      "loss_main": 0.09774766117334366,
      "step": 31750
    },
    {
      "epoch": 3.716358530306576,
      "grad_norm": 0.27196061611175537,
      "learning_rate": 4.92862764357067e-05,
      "loss": 0.0593,
      "step": 31760
    },
    {
      "epoch": 3.716358530306576,
      "loss_main": 0.0793740376830101,
      "step": 31760
    },
    {
      "epoch": 3.717528668382869,
      "grad_norm": 0.32906827330589294,
      "learning_rate": 4.925872152295443e-05,
      "loss": 0.0547,
      "step": 31770
    },
    {
      "epoch": 3.717528668382869,
      "loss_main": 0.044165100902318954,
      "step": 31770
    },
    {
      "epoch": 3.7186988064591624,
      "grad_norm": 0.30220869183540344,
      "learning_rate": 4.923116683538296e-05,
      "loss": 0.0585,
      "step": 31780
    },
    {
      "epoch": 3.7186988064591624,
      "loss_main": 0.059480566531419754,
      "step": 31780
    },
    {
      "epoch": 3.719868944535455,
      "grad_norm": 0.20683836936950684,
      "learning_rate": 4.920361238136273e-05,
      "loss": 0.0643,
      "step": 31790
    },
    {
      "epoch": 3.719868944535455,
      "loss_main": 0.055793050676584244,
      "step": 31790
    },
    {
      "epoch": 3.7210390826117483,
      "grad_norm": 0.2691202163696289,
      "learning_rate": 4.9176058169264014e-05,
      "loss": 0.0594,
      "step": 31800
    },
    {
      "epoch": 3.7210390826117483,
      "loss_main": 0.06122555956244469,
      "step": 31800
    },
    {
      "epoch": 3.722209220688041,
      "grad_norm": 0.2581792175769806,
      "learning_rate": 4.9148504207457074e-05,
      "loss": 0.0646,
      "step": 31810
    },
    {
      "epoch": 3.722209220688041,
      "loss_main": 0.0706913024187088,
      "step": 31810
    },
    {
      "epoch": 3.7233793587643342,
      "grad_norm": 0.2697770893573761,
      "learning_rate": 4.912095050431208e-05,
      "loss": 0.064,
      "step": 31820
    },
    {
      "epoch": 3.7233793587643342,
      "loss_main": 0.06915630400180817,
      "step": 31820
    },
    {
      "epoch": 3.724549496840627,
      "grad_norm": 0.2820696532726288,
      "learning_rate": 4.909339706819911e-05,
      "loss": 0.0579,
      "step": 31830
    },
    {
      "epoch": 3.724549496840627,
      "loss_main": 0.061717092990875244,
      "step": 31830
    },
    {
      "epoch": 3.72571963491692,
      "grad_norm": 0.19547173380851746,
      "learning_rate": 4.906584390748819e-05,
      "loss": 0.0621,
      "step": 31840
    },
    {
      "epoch": 3.72571963491692,
      "loss_main": 0.10500387102365494,
      "step": 31840
    },
    {
      "epoch": 3.7268897729932133,
      "grad_norm": 0.32227376103401184,
      "learning_rate": 4.9038291030549195e-05,
      "loss": 0.0886,
      "step": 31850
    },
    {
      "epoch": 3.7268897729932133,
      "loss_main": 0.08202261477708817,
      "step": 31850
    },
    {
      "epoch": 3.728059911069506,
      "grad_norm": 0.2533752918243408,
      "learning_rate": 4.9010738445751995e-05,
      "loss": 0.0694,
      "step": 31860
    },
    {
      "epoch": 3.728059911069506,
      "loss_main": 0.0785241350531578,
      "step": 31860
    },
    {
      "epoch": 3.7292300491457993,
      "grad_norm": 0.24214652180671692,
      "learning_rate": 4.8983186161466364e-05,
      "loss": 0.0507,
      "step": 31870
    },
    {
      "epoch": 3.7292300491457993,
      "loss_main": 0.08025236427783966,
      "step": 31870
    },
    {
      "epoch": 3.7304001872220924,
      "grad_norm": 0.30095186829566956,
      "learning_rate": 4.89556341860619e-05,
      "loss": 0.0621,
      "step": 31880
    },
    {
      "epoch": 3.7304001872220924,
      "loss_main": 0.05580545589327812,
      "step": 31880
    },
    {
      "epoch": 3.731570325298385,
      "grad_norm": 0.3619765043258667,
      "learning_rate": 4.892808252790822e-05,
      "loss": 0.0637,
      "step": 31890
    },
    {
      "epoch": 3.731570325298385,
      "loss_main": 0.06083758547902107,
      "step": 31890
    },
    {
      "epoch": 3.7327404633746784,
      "grad_norm": 0.30763739347457886,
      "learning_rate": 4.890053119537475e-05,
      "loss": 0.0572,
      "step": 31900
    },
    {
      "epoch": 3.7327404633746784,
      "loss_main": 0.08101703226566315,
      "step": 31900
    },
    {
      "epoch": 3.733910601450971,
      "grad_norm": 0.2753641903400421,
      "learning_rate": 4.887298019683087e-05,
      "loss": 0.0651,
      "step": 31910
    },
    {
      "epoch": 3.733910601450971,
      "loss_main": 0.08093457669019699,
      "step": 31910
    },
    {
      "epoch": 3.7350807395272643,
      "grad_norm": 0.31474635004997253,
      "learning_rate": 4.884542954064587e-05,
      "loss": 0.0674,
      "step": 31920
    },
    {
      "epoch": 3.7350807395272643,
      "loss_main": 0.03721582144498825,
      "step": 31920
    },
    {
      "epoch": 3.736250877603557,
      "grad_norm": 0.25643762946128845,
      "learning_rate": 4.881787923518887e-05,
      "loss": 0.062,
      "step": 31930
    },
    {
      "epoch": 3.736250877603557,
      "loss_main": 0.0286259688436985,
      "step": 31930
    },
    {
      "epoch": 3.73742101567985,
      "grad_norm": 0.30310168862342834,
      "learning_rate": 4.879032928882896e-05,
      "loss": 0.0607,
      "step": 31940
    },
    {
      "epoch": 3.73742101567985,
      "loss_main": 0.05893809348344803,
      "step": 31940
    },
    {
      "epoch": 3.7385911537561434,
      "grad_norm": 0.27996423840522766,
      "learning_rate": 4.876277970993505e-05,
      "loss": 0.0526,
      "step": 31950
    },
    {
      "epoch": 3.7385911537561434,
      "loss_main": 0.07523071765899658,
      "step": 31950
    },
    {
      "epoch": 3.739761291832436,
      "grad_norm": 0.25926074385643005,
      "learning_rate": 4.873523050687602e-05,
      "loss": 0.0648,
      "step": 31960
    },
    {
      "epoch": 3.739761291832436,
      "loss_main": 0.036994609981775284,
      "step": 31960
    },
    {
      "epoch": 3.7409314299087293,
      "grad_norm": 0.26547956466674805,
      "learning_rate": 4.870768168802056e-05,
      "loss": 0.0571,
      "step": 31970
    },
    {
      "epoch": 3.7409314299087293,
      "loss_main": 0.05152076482772827,
      "step": 31970
    },
    {
      "epoch": 3.7421015679850225,
      "grad_norm": 0.25287193059921265,
      "learning_rate": 4.868013326173728e-05,
      "loss": 0.0696,
      "step": 31980
    },
    {
      "epoch": 3.7421015679850225,
      "loss_main": 0.050212644040584564,
      "step": 31980
    },
    {
      "epoch": 3.743271706061315,
      "grad_norm": 0.302528977394104,
      "learning_rate": 4.865258523639468e-05,
      "loss": 0.0575,
      "step": 31990
    },
    {
      "epoch": 3.743271706061315,
      "loss_main": 0.06340526789426804,
      "step": 31990
    },
    {
      "epoch": 3.7444418441376084,
      "grad_norm": 0.3623601794242859,
      "learning_rate": 4.862503762036109e-05,
      "loss": 0.0549,
      "step": 32000
    },
    {
      "epoch": 3.7444418441376084,
      "loss_main": 0.06042558699846268,
      "step": 32000
    },
    {
      "epoch": 3.745611982213901,
      "grad_norm": 0.2678550183773041,
      "learning_rate": 4.859749042200478e-05,
      "loss": 0.0536,
      "step": 32010
    },
    {
      "epoch": 3.745611982213901,
      "loss_main": 0.06665761023759842,
      "step": 32010
    },
    {
      "epoch": 3.7467821202901943,
      "grad_norm": 0.300060510635376,
      "learning_rate": 4.856994364969384e-05,
      "loss": 0.0627,
      "step": 32020
    },
    {
      "epoch": 3.7467821202901943,
      "loss_main": 0.05164707824587822,
      "step": 32020
    },
    {
      "epoch": 3.747952258366487,
      "grad_norm": 0.21860289573669434,
      "learning_rate": 4.854239731179625e-05,
      "loss": 0.0527,
      "step": 32030
    },
    {
      "epoch": 3.747952258366487,
      "loss_main": 0.06636013835668564,
      "step": 32030
    },
    {
      "epoch": 3.7491223964427802,
      "grad_norm": 0.24221520125865936,
      "learning_rate": 4.85148514166799e-05,
      "loss": 0.0614,
      "step": 32040
    },
    {
      "epoch": 3.7491223964427802,
      "loss_main": 0.05707937479019165,
      "step": 32040
    },
    {
      "epoch": 3.7502925345190734,
      "grad_norm": 0.25914397835731506,
      "learning_rate": 4.8487305972712456e-05,
      "loss": 0.0574,
      "step": 32050
    },
    {
      "epoch": 3.7502925345190734,
      "loss_main": 0.05120866745710373,
      "step": 32050
    },
    {
      "epoch": 3.751462672595366,
      "grad_norm": 0.2466535121202469,
      "learning_rate": 4.8459760988261526e-05,
      "loss": 0.0579,
      "step": 32060
    },
    {
      "epoch": 3.751462672595366,
      "loss_main": 0.06031494587659836,
      "step": 32060
    },
    {
      "epoch": 3.7526328106716593,
      "grad_norm": 0.28192850947380066,
      "learning_rate": 4.843221647169453e-05,
      "loss": 0.0652,
      "step": 32070
    },
    {
      "epoch": 3.7526328106716593,
      "loss_main": 0.05230535939335823,
      "step": 32070
    },
    {
      "epoch": 3.7538029487479525,
      "grad_norm": 0.3046380281448364,
      "learning_rate": 4.840467243137878e-05,
      "loss": 0.0604,
      "step": 32080
    },
    {
      "epoch": 3.7538029487479525,
      "loss_main": 0.045143239200115204,
      "step": 32080
    },
    {
      "epoch": 3.7549730868242452,
      "grad_norm": 0.35652264952659607,
      "learning_rate": 4.837712887568143e-05,
      "loss": 0.0587,
      "step": 32090
    },
    {
      "epoch": 3.7549730868242452,
      "loss_main": 0.05781049281358719,
      "step": 32090
    },
    {
      "epoch": 3.7561432249005384,
      "grad_norm": 0.21879535913467407,
      "learning_rate": 4.8349585812969464e-05,
      "loss": 0.0564,
      "step": 32100
    },
    {
      "epoch": 3.7561432249005384,
      "loss_main": 0.053655095398426056,
      "step": 32100
    },
    {
      "epoch": 3.757313362976831,
      "grad_norm": 0.18798774480819702,
      "learning_rate": 4.8322043251609775e-05,
      "loss": 0.0566,
      "step": 32110
    },
    {
      "epoch": 3.757313362976831,
      "loss_main": 0.06008250266313553,
      "step": 32110
    },
    {
      "epoch": 3.7584835010531243,
      "grad_norm": 0.24778559803962708,
      "learning_rate": 4.8294501199969015e-05,
      "loss": 0.0521,
      "step": 32120
    },
    {
      "epoch": 3.7584835010531243,
      "loss_main": 0.04877045378088951,
      "step": 32120
    },
    {
      "epoch": 3.759653639129417,
      "grad_norm": 0.3667694926261902,
      "learning_rate": 4.826695966641376e-05,
      "loss": 0.0544,
      "step": 32130
    },
    {
      "epoch": 3.759653639129417,
      "loss_main": 0.04770413786172867,
      "step": 32130
    },
    {
      "epoch": 3.7608237772057103,
      "grad_norm": 0.31090375781059265,
      "learning_rate": 4.823941865931043e-05,
      "loss": 0.0631,
      "step": 32140
    },
    {
      "epoch": 3.7608237772057103,
      "loss_main": 0.06422080099582672,
      "step": 32140
    },
    {
      "epoch": 3.7619939152820034,
      "grad_norm": 0.25619450211524963,
      "learning_rate": 4.82118781870252e-05,
      "loss": 0.0675,
      "step": 32150
    },
    {
      "epoch": 3.7619939152820034,
      "loss_main": 0.027723809704184532,
      "step": 32150
    },
    {
      "epoch": 3.763164053358296,
      "grad_norm": 0.26544827222824097,
      "learning_rate": 4.8184338257924185e-05,
      "loss": 0.0574,
      "step": 32160
    },
    {
      "epoch": 3.763164053358296,
      "loss_main": 0.10408405214548111,
      "step": 32160
    },
    {
      "epoch": 3.7643341914345894,
      "grad_norm": 0.40042251348495483,
      "learning_rate": 4.815679888037324e-05,
      "loss": 0.0645,
      "step": 32170
    },
    {
      "epoch": 3.7643341914345894,
      "loss_main": 0.056388795375823975,
      "step": 32170
    },
    {
      "epoch": 3.7655043295108825,
      "grad_norm": 0.23914362490177155,
      "learning_rate": 4.8129260062738135e-05,
      "loss": 0.0567,
      "step": 32180
    },
    {
      "epoch": 3.7655043295108825,
      "loss_main": 0.03876904770731926,
      "step": 32180
    },
    {
      "epoch": 3.7666744675871753,
      "grad_norm": 0.26428765058517456,
      "learning_rate": 4.810172181338445e-05,
      "loss": 0.0611,
      "step": 32190
    },
    {
      "epoch": 3.7666744675871753,
      "loss_main": 0.06487888842821121,
      "step": 32190
    },
    {
      "epoch": 3.767844605663468,
      "grad_norm": 0.3993188142776489,
      "learning_rate": 4.807418414067753e-05,
      "loss": 0.0543,
      "step": 32200
    },
    {
      "epoch": 3.767844605663468,
      "loss_main": 0.0440056137740612,
      "step": 32200
    },
    {
      "epoch": 3.769014743739761,
      "grad_norm": 0.29823175072669983,
      "learning_rate": 4.804664705298264e-05,
      "loss": 0.0552,
      "step": 32210
    },
    {
      "epoch": 3.769014743739761,
      "loss_main": 0.0561186857521534,
      "step": 32210
    },
    {
      "epoch": 3.7701848818160544,
      "grad_norm": 0.2627565264701843,
      "learning_rate": 4.80191105586648e-05,
      "loss": 0.0592,
      "step": 32220
    },
    {
      "epoch": 3.7701848818160544,
      "loss_main": 0.05788638815283775,
      "step": 32220
    },
    {
      "epoch": 3.771355019892347,
      "grad_norm": 0.228290855884552,
      "learning_rate": 4.799157466608886e-05,
      "loss": 0.0579,
      "step": 32230
    },
    {
      "epoch": 3.771355019892347,
      "loss_main": 0.053936030715703964,
      "step": 32230
    },
    {
      "epoch": 3.7725251579686403,
      "grad_norm": 0.30760458111763,
      "learning_rate": 4.796403938361951e-05,
      "loss": 0.0647,
      "step": 32240
    },
    {
      "epoch": 3.7725251579686403,
      "loss_main": 0.06385757774114609,
      "step": 32240
    },
    {
      "epoch": 3.7736952960449335,
      "grad_norm": 0.2581392526626587,
      "learning_rate": 4.793650471962123e-05,
      "loss": 0.0535,
      "step": 32250
    },
    {
      "epoch": 3.7736952960449335,
      "loss_main": 0.08269471675157547,
      "step": 32250
    },
    {
      "epoch": 3.774865434121226,
      "grad_norm": 0.36161747574806213,
      "learning_rate": 4.790897068245835e-05,
      "loss": 0.0594,
      "step": 32260
    },
    {
      "epoch": 3.774865434121226,
      "loss_main": 0.04081844910979271,
      "step": 32260
    },
    {
      "epoch": 3.7760355721975194,
      "grad_norm": 0.26657575368881226,
      "learning_rate": 4.7881437280494954e-05,
      "loss": 0.0603,
      "step": 32270
    },
    {
      "epoch": 3.7760355721975194,
      "loss_main": 0.07124852389097214,
      "step": 32270
    },
    {
      "epoch": 3.7772057102738126,
      "grad_norm": 0.2828451991081238,
      "learning_rate": 4.7853904522094965e-05,
      "loss": 0.0612,
      "step": 32280
    },
    {
      "epoch": 3.7772057102738126,
      "loss_main": 0.05322685092687607,
      "step": 32280
    },
    {
      "epoch": 3.7783758483501053,
      "grad_norm": 0.3254139721393585,
      "learning_rate": 4.782637241562215e-05,
      "loss": 0.0605,
      "step": 32290
    },
    {
      "epoch": 3.7783758483501053,
      "loss_main": 0.03202148526906967,
      "step": 32290
    },
    {
      "epoch": 3.779545986426398,
      "grad_norm": 0.25879380106925964,
      "learning_rate": 4.779884096943997e-05,
      "loss": 0.0493,
      "step": 32300
    },
    {
      "epoch": 3.779545986426398,
      "loss_main": 0.05220185965299606,
      "step": 32300
    },
    {
      "epoch": 3.7807161245026912,
      "grad_norm": 0.32471469044685364,
      "learning_rate": 4.777131019191182e-05,
      "loss": 0.0665,
      "step": 32310
    },
    {
      "epoch": 3.7807161245026912,
      "loss_main": 0.05875487998127937,
      "step": 32310
    },
    {
      "epoch": 3.7818862625789844,
      "grad_norm": 0.23041433095932007,
      "learning_rate": 4.774378009140076e-05,
      "loss": 0.0567,
      "step": 32320
    },
    {
      "epoch": 3.7818862625789844,
      "loss_main": 0.052143700420856476,
      "step": 32320
    },
    {
      "epoch": 3.783056400655277,
      "grad_norm": 0.4760643243789673,
      "learning_rate": 4.7716250676269735e-05,
      "loss": 0.0518,
      "step": 32330
    },
    {
      "epoch": 3.783056400655277,
      "loss_main": 0.043007466942071915,
      "step": 32330
    },
    {
      "epoch": 3.7842265387315703,
      "grad_norm": 0.28510087728500366,
      "learning_rate": 4.7688721954881485e-05,
      "loss": 0.0608,
      "step": 32340
    },
    {
      "epoch": 3.7842265387315703,
      "loss_main": 0.050423819571733475,
      "step": 32340
    },
    {
      "epoch": 3.7853966768078635,
      "grad_norm": 0.4150654077529907,
      "learning_rate": 4.7661193935598446e-05,
      "loss": 0.0704,
      "step": 32350
    },
    {
      "epoch": 3.7853966768078635,
      "loss_main": 0.060683924704790115,
      "step": 32350
    },
    {
      "epoch": 3.7865668148841563,
      "grad_norm": 0.344448983669281,
      "learning_rate": 4.763366662678296e-05,
      "loss": 0.0573,
      "step": 32360
    },
    {
      "epoch": 3.7865668148841563,
      "loss_main": 0.07330743223428726,
      "step": 32360
    },
    {
      "epoch": 3.7877369529604494,
      "grad_norm": 0.3431455194950104,
      "learning_rate": 4.7606140036797064e-05,
      "loss": 0.066,
      "step": 32370
    },
    {
      "epoch": 3.7877369529604494,
      "loss_main": 0.03224954754114151,
      "step": 32370
    },
    {
      "epoch": 3.788907091036742,
      "grad_norm": 0.3748330771923065,
      "learning_rate": 4.7578614174002614e-05,
      "loss": 0.0569,
      "step": 32380
    },
    {
      "epoch": 3.788907091036742,
      "loss_main": 0.05081688240170479,
      "step": 32380
    },
    {
      "epoch": 3.7900772291130354,
      "grad_norm": 0.2786608934402466,
      "learning_rate": 4.755108904676125e-05,
      "loss": 0.0549,
      "step": 32390
    },
    {
      "epoch": 3.7900772291130354,
      "loss_main": 0.06932483613491058,
      "step": 32390
    },
    {
      "epoch": 3.791247367189328,
      "grad_norm": 0.26935046911239624,
      "learning_rate": 4.752356466343436e-05,
      "loss": 0.0653,
      "step": 32400
    },
    {
      "epoch": 3.791247367189328,
      "loss_main": 0.05665689334273338,
      "step": 32400
    },
    {
      "epoch": 3.7924175052656213,
      "grad_norm": 0.30246245861053467,
      "learning_rate": 4.7496041032383174e-05,
      "loss": 0.0534,
      "step": 32410
    },
    {
      "epoch": 3.7924175052656213,
      "loss_main": 0.06037726625800133,
      "step": 32410
    },
    {
      "epoch": 3.7935876433419145,
      "grad_norm": 0.245808407664299,
      "learning_rate": 4.746851816196858e-05,
      "loss": 0.0619,
      "step": 32420
    },
    {
      "epoch": 3.7935876433419145,
      "loss_main": 0.06336301565170288,
      "step": 32420
    },
    {
      "epoch": 3.794757781418207,
      "grad_norm": 0.27323204278945923,
      "learning_rate": 4.744099606055135e-05,
      "loss": 0.0596,
      "step": 32430
    },
    {
      "epoch": 3.794757781418207,
      "loss_main": 0.07993847131729126,
      "step": 32430
    },
    {
      "epoch": 3.7959279194945004,
      "grad_norm": 0.2660634219646454,
      "learning_rate": 4.741347473649193e-05,
      "loss": 0.0602,
      "step": 32440
    },
    {
      "epoch": 3.7959279194945004,
      "loss_main": 0.07326427847146988,
      "step": 32440
    },
    {
      "epoch": 3.7970980575707936,
      "grad_norm": 0.321355402469635,
      "learning_rate": 4.738595419815058e-05,
      "loss": 0.0528,
      "step": 32450
    },
    {
      "epoch": 3.7970980575707936,
      "loss_main": 0.08342321962118149,
      "step": 32450
    },
    {
      "epoch": 3.7982681956470863,
      "grad_norm": 0.27965766191482544,
      "learning_rate": 4.7358434453887365e-05,
      "loss": 0.063,
      "step": 32460
    },
    {
      "epoch": 3.7982681956470863,
      "loss_main": 0.05464481934905052,
      "step": 32460
    },
    {
      "epoch": 3.7994383337233795,
      "grad_norm": 0.22544017434120178,
      "learning_rate": 4.7330915512061976e-05,
      "loss": 0.0591,
      "step": 32470
    },
    {
      "epoch": 3.7994383337233795,
      "loss_main": 0.06982444226741791,
      "step": 32470
    },
    {
      "epoch": 3.800608471799672,
      "grad_norm": 0.23095186054706573,
      "learning_rate": 4.730339738103402e-05,
      "loss": 0.0631,
      "step": 32480
    },
    {
      "epoch": 3.800608471799672,
      "loss_main": 0.08169946819543839,
      "step": 32480
    },
    {
      "epoch": 3.8017786098759654,
      "grad_norm": 0.2536209523677826,
      "learning_rate": 4.727588006916271e-05,
      "loss": 0.0609,
      "step": 32490
    },
    {
      "epoch": 3.8017786098759654,
      "loss_main": 0.05207306891679764,
      "step": 32490
    },
    {
      "epoch": 3.802948747952258,
      "grad_norm": 0.2232840657234192,
      "learning_rate": 4.724836358480711e-05,
      "loss": 0.0627,
      "step": 32500
    },
    {
      "epoch": 3.802948747952258,
      "loss_main": 0.09049931913614273,
      "step": 32500
    },
    {
      "epoch": 3.8041188860285513,
      "grad_norm": 0.3389546573162079,
      "learning_rate": 4.722084793632601e-05,
      "loss": 0.0671,
      "step": 32510
    },
    {
      "epoch": 3.8041188860285513,
      "loss_main": 0.02747482992708683,
      "step": 32510
    },
    {
      "epoch": 3.8052890241048445,
      "grad_norm": 0.223052978515625,
      "learning_rate": 4.719333313207792e-05,
      "loss": 0.0612,
      "step": 32520
    },
    {
      "epoch": 3.8052890241048445,
      "loss_main": 0.08022152632474899,
      "step": 32520
    },
    {
      "epoch": 3.8064591621811372,
      "grad_norm": 0.3266374468803406,
      "learning_rate": 4.716581918042114e-05,
      "loss": 0.0627,
      "step": 32530
    },
    {
      "epoch": 3.8064591621811372,
      "loss_main": 0.05141231045126915,
      "step": 32530
    },
    {
      "epoch": 3.8076293002574304,
      "grad_norm": 0.2522086203098297,
      "learning_rate": 4.7138306089713636e-05,
      "loss": 0.057,
      "step": 32540
    },
    {
      "epoch": 3.8076293002574304,
      "loss_main": 0.04370898753404617,
      "step": 32540
    },
    {
      "epoch": 3.8087994383337236,
      "grad_norm": 0.6639673709869385,
      "learning_rate": 4.7110793868313183e-05,
      "loss": 0.0544,
      "step": 32550
    },
    {
      "epoch": 3.8087994383337236,
      "loss_main": 0.04433133080601692,
      "step": 32550
    },
    {
      "epoch": 3.8099695764100163,
      "grad_norm": 0.3312356770038605,
      "learning_rate": 4.708328252457729e-05,
      "loss": 0.0573,
      "step": 32560
    },
    {
      "epoch": 3.8099695764100163,
      "loss_main": 0.07226961106061935,
      "step": 32560
    },
    {
      "epoch": 3.8111397144863095,
      "grad_norm": 0.25935760140419006,
      "learning_rate": 4.7055772066863135e-05,
      "loss": 0.0661,
      "step": 32570
    },
    {
      "epoch": 3.8111397144863095,
      "loss_main": 0.0803653821349144,
      "step": 32570
    },
    {
      "epoch": 3.8123098525626022,
      "grad_norm": 0.22702690958976746,
      "learning_rate": 4.702826250352771e-05,
      "loss": 0.0557,
      "step": 32580
    },
    {
      "epoch": 3.8123098525626022,
      "loss_main": 0.0944313183426857,
      "step": 32580
    },
    {
      "epoch": 3.8134799906388954,
      "grad_norm": 0.22450871765613556,
      "learning_rate": 4.7000753842927653e-05,
      "loss": 0.0639,
      "step": 32590
    },
    {
      "epoch": 3.8134799906388954,
      "loss_main": 0.04736461117863655,
      "step": 32590
    },
    {
      "epoch": 3.814650128715188,
      "grad_norm": 0.2798008918762207,
      "learning_rate": 4.6973246093419384e-05,
      "loss": 0.0601,
      "step": 32600
    },
    {
      "epoch": 3.814650128715188,
      "loss_main": 0.04800238087773323,
      "step": 32600
    },
    {
      "epoch": 3.8158202667914813,
      "grad_norm": 0.39650729298591614,
      "learning_rate": 4.694573926335906e-05,
      "loss": 0.0636,
      "step": 32610
    },
    {
      "epoch": 3.8158202667914813,
      "loss_main": 0.040932394564151764,
      "step": 32610
    },
    {
      "epoch": 3.8169904048677745,
      "grad_norm": 0.2769431173801422,
      "learning_rate": 4.6918233361102476e-05,
      "loss": 0.0539,
      "step": 32620
    },
    {
      "epoch": 3.8169904048677745,
      "loss_main": 0.04026505723595619,
      "step": 32620
    },
    {
      "epoch": 3.8181605429440673,
      "grad_norm": 0.33903127908706665,
      "learning_rate": 4.689072839500525e-05,
      "loss": 0.0595,
      "step": 32630
    },
    {
      "epoch": 3.8181605429440673,
      "loss_main": 0.04808437079191208,
      "step": 32630
    },
    {
      "epoch": 3.8193306810203604,
      "grad_norm": 0.27778202295303345,
      "learning_rate": 4.6863224373422635e-05,
      "loss": 0.0576,
      "step": 32640
    },
    {
      "epoch": 3.8193306810203604,
      "loss_main": 0.07257742434740067,
      "step": 32640
    },
    {
      "epoch": 3.8205008190966536,
      "grad_norm": 0.27730441093444824,
      "learning_rate": 4.683572130470962e-05,
      "loss": 0.0538,
      "step": 32650
    },
    {
      "epoch": 3.8205008190966536,
      "loss_main": 0.051745202392339706,
      "step": 32650
    },
    {
      "epoch": 3.8216709571729464,
      "grad_norm": 0.23468339443206787,
      "learning_rate": 4.680821919722094e-05,
      "loss": 0.0696,
      "step": 32660
    },
    {
      "epoch": 3.8216709571729464,
      "loss_main": 0.08353018015623093,
      "step": 32660
    },
    {
      "epoch": 3.8228410952492395,
      "grad_norm": 0.32641279697418213,
      "learning_rate": 4.6780718059310975e-05,
      "loss": 0.0648,
      "step": 32670
    },
    {
      "epoch": 3.8228410952492395,
      "loss_main": 0.047412335872650146,
      "step": 32670
    },
    {
      "epoch": 3.8240112333255323,
      "grad_norm": 0.23609036207199097,
      "learning_rate": 4.675321789933389e-05,
      "loss": 0.0533,
      "step": 32680
    },
    {
      "epoch": 3.8240112333255323,
      "loss_main": 0.04024593159556389,
      "step": 32680
    },
    {
      "epoch": 3.8251813714018255,
      "grad_norm": 0.4232819080352783,
      "learning_rate": 4.6725718725643464e-05,
      "loss": 0.0534,
      "step": 32690
    },
    {
      "epoch": 3.8251813714018255,
      "loss_main": 0.10213722288608551,
      "step": 32690
    },
    {
      "epoch": 3.826351509478118,
      "grad_norm": 0.3389211595058441,
      "learning_rate": 4.669822054659323e-05,
      "loss": 0.0603,
      "step": 32700
    },
    {
      "epoch": 3.826351509478118,
      "loss_main": 0.06362525373697281,
      "step": 32700
    },
    {
      "epoch": 3.8275216475544114,
      "grad_norm": 0.2970958948135376,
      "learning_rate": 4.667072337053644e-05,
      "loss": 0.0576,
      "step": 32710
    },
    {
      "epoch": 3.8275216475544114,
      "loss_main": 0.05405410751700401,
      "step": 32710
    },
    {
      "epoch": 3.8286917856307046,
      "grad_norm": 0.2935986816883087,
      "learning_rate": 4.6643227205825965e-05,
      "loss": 0.066,
      "step": 32720
    },
    {
      "epoch": 3.8286917856307046,
      "loss_main": 0.04528706148266792,
      "step": 32720
    },
    {
      "epoch": 3.8298619237069973,
      "grad_norm": 0.2167281061410904,
      "learning_rate": 4.6615732060814454e-05,
      "loss": 0.0616,
      "step": 32730
    },
    {
      "epoch": 3.8298619237069973,
      "loss_main": 0.047167472541332245,
      "step": 32730
    },
    {
      "epoch": 3.8310320617832905,
      "grad_norm": 0.2286626100540161,
      "learning_rate": 4.658823794385417e-05,
      "loss": 0.0632,
      "step": 32740
    },
    {
      "epoch": 3.8310320617832905,
      "loss_main": 0.05767172574996948,
      "step": 32740
    },
    {
      "epoch": 3.8322021998595837,
      "grad_norm": 0.3560206890106201,
      "learning_rate": 4.6560744863297115e-05,
      "loss": 0.0645,
      "step": 32750
    },
    {
      "epoch": 3.8322021998595837,
      "loss_main": 0.07399686425924301,
      "step": 32750
    },
    {
      "epoch": 3.8333723379358764,
      "grad_norm": 0.3175410032272339,
      "learning_rate": 4.653325282749498e-05,
      "loss": 0.0651,
      "step": 32760
    },
    {
      "epoch": 3.8333723379358764,
      "loss_main": 0.06378591805696487,
      "step": 32760
    },
    {
      "epoch": 3.8345424760121696,
      "grad_norm": 0.3821357786655426,
      "learning_rate": 4.6505761844799075e-05,
      "loss": 0.0602,
      "step": 32770
    },
    {
      "epoch": 3.8345424760121696,
      "loss_main": 0.031320612877607346,
      "step": 32770
    },
    {
      "epoch": 3.8357126140884623,
      "grad_norm": 0.20033885538578033,
      "learning_rate": 4.647827192356048e-05,
      "loss": 0.055,
      "step": 32780
    },
    {
      "epoch": 3.8357126140884623,
      "loss_main": 0.0739244893193245,
      "step": 32780
    },
    {
      "epoch": 3.8368827521647555,
      "grad_norm": 0.292438805103302,
      "learning_rate": 4.645078307212989e-05,
      "loss": 0.0638,
      "step": 32790
    },
    {
      "epoch": 3.8368827521647555,
      "loss_main": 0.0426764115691185,
      "step": 32790
    },
    {
      "epoch": 3.8380528902410482,
      "grad_norm": 0.2992100119590759,
      "learning_rate": 4.642329529885768e-05,
      "loss": 0.0629,
      "step": 32800
    },
    {
      "epoch": 3.8380528902410482,
      "loss_main": 0.029707079753279686,
      "step": 32800
    },
    {
      "epoch": 3.8392230283173414,
      "grad_norm": 0.25624552369117737,
      "learning_rate": 4.639580861209393e-05,
      "loss": 0.0594,
      "step": 32810
    },
    {
      "epoch": 3.8392230283173414,
      "loss_main": 0.052777133882045746,
      "step": 32810
    },
    {
      "epoch": 3.8403931663936346,
      "grad_norm": 0.1959867626428604,
      "learning_rate": 4.636832302018835e-05,
      "loss": 0.0611,
      "step": 32820
    },
    {
      "epoch": 3.8403931663936346,
      "loss_main": 0.03692331537604332,
      "step": 32820
    },
    {
      "epoch": 3.8415633044699273,
      "grad_norm": 0.2772231996059418,
      "learning_rate": 4.6340838531490365e-05,
      "loss": 0.05,
      "step": 32830
    },
    {
      "epoch": 3.8415633044699273,
      "loss_main": 0.036087214946746826,
      "step": 32830
    },
    {
      "epoch": 3.8427334425462205,
      "grad_norm": 0.18290024995803833,
      "learning_rate": 4.6313355154349e-05,
      "loss": 0.0582,
      "step": 32840
    },
    {
      "epoch": 3.8427334425462205,
      "loss_main": 0.07384998351335526,
      "step": 32840
    },
    {
      "epoch": 3.8439035806225137,
      "grad_norm": 0.1650337427854538,
      "learning_rate": 4.6285872897113025e-05,
      "loss": 0.0537,
      "step": 32850
    },
    {
      "epoch": 3.8439035806225137,
      "loss_main": 0.07067164033651352,
      "step": 32850
    },
    {
      "epoch": 3.8450737186988064,
      "grad_norm": 0.28142911195755005,
      "learning_rate": 4.625839176813077e-05,
      "loss": 0.0512,
      "step": 32860
    },
    {
      "epoch": 3.8450737186988064,
      "loss_main": 0.034957289695739746,
      "step": 32860
    },
    {
      "epoch": 3.8462438567750996,
      "grad_norm": 0.3523571491241455,
      "learning_rate": 4.623091177575031e-05,
      "loss": 0.059,
      "step": 32870
    },
    {
      "epoch": 3.8462438567750996,
      "loss_main": 0.056051526218652725,
      "step": 32870
    },
    {
      "epoch": 3.8474139948513923,
      "grad_norm": 0.24951788783073425,
      "learning_rate": 4.620343292831936e-05,
      "loss": 0.0557,
      "step": 32880
    },
    {
      "epoch": 3.8474139948513923,
      "loss_main": 0.05882422253489494,
      "step": 32880
    },
    {
      "epoch": 3.8485841329276855,
      "grad_norm": 0.2588433623313904,
      "learning_rate": 4.6175955234185206e-05,
      "loss": 0.0608,
      "step": 32890
    },
    {
      "epoch": 3.8485841329276855,
      "loss_main": 0.02412356622517109,
      "step": 32890
    },
    {
      "epoch": 3.8497542710039783,
      "grad_norm": 0.17728492617607117,
      "learning_rate": 4.614847870169492e-05,
      "loss": 0.0498,
      "step": 32900
    },
    {
      "epoch": 3.8497542710039783,
      "loss_main": 0.064230777323246,
      "step": 32900
    },
    {
      "epoch": 3.8509244090802714,
      "grad_norm": 0.3536585867404938,
      "learning_rate": 4.612100333919509e-05,
      "loss": 0.0564,
      "step": 32910
    },
    {
      "epoch": 3.8509244090802714,
      "loss_main": 0.04282776638865471,
      "step": 32910
    },
    {
      "epoch": 3.8520945471565646,
      "grad_norm": 0.21804825961589813,
      "learning_rate": 4.609352915503202e-05,
      "loss": 0.0694,
      "step": 32920
    },
    {
      "epoch": 3.8520945471565646,
      "loss_main": 0.024859990924596786,
      "step": 32920
    },
    {
      "epoch": 3.8532646852328574,
      "grad_norm": 0.2961057424545288,
      "learning_rate": 4.606605615755166e-05,
      "loss": 0.057,
      "step": 32930
    },
    {
      "epoch": 3.8532646852328574,
      "loss_main": 0.07820264250040054,
      "step": 32930
    },
    {
      "epoch": 3.8544348233091505,
      "grad_norm": 0.25183579325675964,
      "learning_rate": 4.6038584355099576e-05,
      "loss": 0.0644,
      "step": 32940
    },
    {
      "epoch": 3.8544348233091505,
      "loss_main": 0.02729472517967224,
      "step": 32940
    },
    {
      "epoch": 3.8556049613854437,
      "grad_norm": 0.2717636823654175,
      "learning_rate": 4.6011113756020964e-05,
      "loss": 0.0555,
      "step": 32950
    },
    {
      "epoch": 3.8556049613854437,
      "loss_main": 0.0862736850976944,
      "step": 32950
    },
    {
      "epoch": 3.8567750994617365,
      "grad_norm": 0.2245476394891739,
      "learning_rate": 4.598364436866066e-05,
      "loss": 0.0655,
      "step": 32960
    },
    {
      "epoch": 3.8567750994617365,
      "loss_main": 0.0712876245379448,
      "step": 32960
    },
    {
      "epoch": 3.8579452375380296,
      "grad_norm": 0.24843454360961914,
      "learning_rate": 4.595617620136316e-05,
      "loss": 0.0604,
      "step": 32970
    },
    {
      "epoch": 3.8579452375380296,
      "loss_main": 0.07676664739847183,
      "step": 32970
    },
    {
      "epoch": 3.8591153756143224,
      "grad_norm": 0.30005067586898804,
      "learning_rate": 4.592870926247257e-05,
      "loss": 0.0637,
      "step": 32980
    },
    {
      "epoch": 3.8591153756143224,
      "loss_main": 0.05594049021601677,
      "step": 32980
    },
    {
      "epoch": 3.8602855136906156,
      "grad_norm": 0.3264211416244507,
      "learning_rate": 4.5901243560332594e-05,
      "loss": 0.0681,
      "step": 32990
    },
    {
      "epoch": 3.8602855136906156,
      "loss_main": 0.08758626878261566,
      "step": 32990
    },
    {
      "epoch": 3.8614556517669083,
      "grad_norm": 0.31216946244239807,
      "learning_rate": 4.587377910328662e-05,
      "loss": 0.0548,
      "step": 33000
    },
    {
      "epoch": 3.8614556517669083,
      "loss_main": 0.04903959855437279,
      "step": 33000
    },
    {
      "epoch": 3.8626257898432015,
      "grad_norm": 0.2558853030204773,
      "learning_rate": 4.5846315899677586e-05,
      "loss": 0.0499,
      "step": 33010
    },
    {
      "epoch": 3.8626257898432015,
      "loss_main": 0.03658563643693924,
      "step": 33010
    },
    {
      "epoch": 3.8637959279194947,
      "grad_norm": 0.3896104097366333,
      "learning_rate": 4.5818853957848114e-05,
      "loss": 0.0622,
      "step": 33020
    },
    {
      "epoch": 3.8637959279194947,
      "loss_main": 0.05398146063089371,
      "step": 33020
    },
    {
      "epoch": 3.8649660659957874,
      "grad_norm": 0.2908857762813568,
      "learning_rate": 4.579139328614043e-05,
      "loss": 0.0601,
      "step": 33030
    },
    {
      "epoch": 3.8649660659957874,
      "loss_main": 0.03497052937746048,
      "step": 33030
    },
    {
      "epoch": 3.8661362040720806,
      "grad_norm": 0.22388237714767456,
      "learning_rate": 4.576393389289633e-05,
      "loss": 0.0556,
      "step": 33040
    },
    {
      "epoch": 3.8661362040720806,
      "loss_main": 0.07342518121004105,
      "step": 33040
    },
    {
      "epoch": 3.8673063421483738,
      "grad_norm": 0.20964577794075012,
      "learning_rate": 4.573647578645728e-05,
      "loss": 0.0621,
      "step": 33050
    },
    {
      "epoch": 3.8673063421483738,
      "loss_main": 0.10008209198713303,
      "step": 33050
    },
    {
      "epoch": 3.8684764802246665,
      "grad_norm": 0.3130273222923279,
      "learning_rate": 4.57090189751643e-05,
      "loss": 0.0554,
      "step": 33060
    },
    {
      "epoch": 3.8684764802246665,
      "loss_main": 0.06854776293039322,
      "step": 33060
    },
    {
      "epoch": 3.8696466183009592,
      "grad_norm": 0.29080602526664734,
      "learning_rate": 4.568156346735806e-05,
      "loss": 0.0554,
      "step": 33070
    },
    {
      "epoch": 3.8696466183009592,
      "loss_main": 0.05300680920481682,
      "step": 33070
    },
    {
      "epoch": 3.8708167563772524,
      "grad_norm": 0.16579583287239075,
      "learning_rate": 4.565410927137882e-05,
      "loss": 0.0563,
      "step": 33080
    },
    {
      "epoch": 3.8708167563772524,
      "loss_main": 0.06012926623225212,
      "step": 33080
    },
    {
      "epoch": 3.8719868944535456,
      "grad_norm": 0.2332119345664978,
      "learning_rate": 4.562665639556644e-05,
      "loss": 0.0793,
      "step": 33090
    },
    {
      "epoch": 3.8719868944535456,
      "loss_main": 0.03542884811758995,
      "step": 33090
    },
    {
      "epoch": 3.8731570325298383,
      "grad_norm": 0.2129896581172943,
      "learning_rate": 4.559920484826037e-05,
      "loss": 0.0562,
      "step": 33100
    },
    {
      "epoch": 3.8731570325298383,
      "loss_main": 0.048844628036022186,
      "step": 33100
    },
    {
      "epoch": 3.8743271706061315,
      "grad_norm": 0.26201197504997253,
      "learning_rate": 4.5571754637799665e-05,
      "loss": 0.0605,
      "step": 33110
    },
    {
      "epoch": 3.8743271706061315,
      "loss_main": 0.042891863733530045,
      "step": 33110
    },
    {
      "epoch": 3.8754973086824247,
      "grad_norm": 0.2526581287384033,
      "learning_rate": 4.554430577252298e-05,
      "loss": 0.0533,
      "step": 33120
    },
    {
      "epoch": 3.8754973086824247,
      "loss_main": 0.0451948381960392,
      "step": 33120
    },
    {
      "epoch": 3.8766674467587174,
      "grad_norm": 0.22050811350345612,
      "learning_rate": 4.551685826076858e-05,
      "loss": 0.0528,
      "step": 33130
    },
    {
      "epoch": 3.8766674467587174,
      "loss_main": 0.059737835079431534,
      "step": 33130
    },
    {
      "epoch": 3.8778375848350106,
      "grad_norm": 0.25158771872520447,
      "learning_rate": 4.5489412110874246e-05,
      "loss": 0.0604,
      "step": 33140
    },
    {
      "epoch": 3.8778375848350106,
      "loss_main": 0.06449966877698898,
      "step": 33140
    },
    {
      "epoch": 3.879007722911304,
      "grad_norm": 0.3790045976638794,
      "learning_rate": 4.5461967331177444e-05,
      "loss": 0.0615,
      "step": 33150
    },
    {
      "epoch": 3.879007722911304,
      "loss_main": 0.06252451986074448,
      "step": 33150
    },
    {
      "epoch": 3.8801778609875965,
      "grad_norm": 0.3386390507221222,
      "learning_rate": 4.5434523930015115e-05,
      "loss": 0.0647,
      "step": 33160
    },
    {
      "epoch": 3.8801778609875965,
      "loss_main": 0.05444687604904175,
      "step": 33160
    },
    {
      "epoch": 3.8813479990638893,
      "grad_norm": 0.3332926630973816,
      "learning_rate": 4.540708191572388e-05,
      "loss": 0.0594,
      "step": 33170
    },
    {
      "epoch": 3.8813479990638893,
      "loss_main": 0.050894446671009064,
      "step": 33170
    },
    {
      "epoch": 3.8825181371401825,
      "grad_norm": 0.2111625075340271,
      "learning_rate": 4.537964129663991e-05,
      "loss": 0.0514,
      "step": 33180
    },
    {
      "epoch": 3.8825181371401825,
      "loss_main": 0.057900574058294296,
      "step": 33180
    },
    {
      "epoch": 3.8836882752164756,
      "grad_norm": 0.2808193862438202,
      "learning_rate": 4.535220208109889e-05,
      "loss": 0.0634,
      "step": 33190
    },
    {
      "epoch": 3.8836882752164756,
      "loss_main": 0.05280493572354317,
      "step": 33190
    },
    {
      "epoch": 3.8848584132927684,
      "grad_norm": 0.21756140887737274,
      "learning_rate": 4.5324764277436194e-05,
      "loss": 0.0553,
      "step": 33200
    },
    {
      "epoch": 3.8848584132927684,
      "loss_main": 0.061830438673496246,
      "step": 33200
    },
    {
      "epoch": 3.8860285513690616,
      "grad_norm": 0.26434507966041565,
      "learning_rate": 4.529732789398664e-05,
      "loss": 0.0603,
      "step": 33210
    },
    {
      "epoch": 3.8860285513690616,
      "loss_main": 0.08874314278364182,
      "step": 33210
    },
    {
      "epoch": 3.8871986894453547,
      "grad_norm": 0.25422367453575134,
      "learning_rate": 4.526989293908472e-05,
      "loss": 0.0581,
      "step": 33220
    },
    {
      "epoch": 3.8871986894453547,
      "loss_main": 0.033567480742931366,
      "step": 33220
    },
    {
      "epoch": 3.8883688275216475,
      "grad_norm": 0.37797874212265015,
      "learning_rate": 4.524245942106442e-05,
      "loss": 0.0575,
      "step": 33230
    },
    {
      "epoch": 3.8883688275216475,
      "loss_main": 0.0659136027097702,
      "step": 33230
    },
    {
      "epoch": 3.8895389655979407,
      "grad_norm": 0.33224162459373474,
      "learning_rate": 4.5215027348259345e-05,
      "loss": 0.067,
      "step": 33240
    },
    {
      "epoch": 3.8895389655979407,
      "loss_main": 0.05718507617712021,
      "step": 33240
    },
    {
      "epoch": 3.8907091036742334,
      "grad_norm": 0.26708653569221497,
      "learning_rate": 4.5187596729002616e-05,
      "loss": 0.0621,
      "step": 33250
    },
    {
      "epoch": 3.8907091036742334,
      "loss_main": 0.07275781780481339,
      "step": 33250
    },
    {
      "epoch": 3.8918792417505266,
      "grad_norm": 0.25824037194252014,
      "learning_rate": 4.516016757162693e-05,
      "loss": 0.0544,
      "step": 33260
    },
    {
      "epoch": 3.8918792417505266,
      "loss_main": 0.07209478318691254,
      "step": 33260
    },
    {
      "epoch": 3.8930493798268193,
      "grad_norm": 0.1956573873758316,
      "learning_rate": 4.513273988446457e-05,
      "loss": 0.0545,
      "step": 33270
    },
    {
      "epoch": 3.8930493798268193,
      "loss_main": 0.06671291589736938,
      "step": 33270
    },
    {
      "epoch": 3.8942195179031125,
      "grad_norm": 0.24712945520877838,
      "learning_rate": 4.5105313675847296e-05,
      "loss": 0.0553,
      "step": 33280
    },
    {
      "epoch": 3.8942195179031125,
      "loss_main": 0.05649568513035774,
      "step": 33280
    },
    {
      "epoch": 3.8953896559794057,
      "grad_norm": 0.2790127694606781,
      "learning_rate": 4.5077888954106495e-05,
      "loss": 0.0554,
      "step": 33290
    },
    {
      "epoch": 3.8953896559794057,
      "loss_main": 0.045132577419281006,
      "step": 33290
    },
    {
      "epoch": 3.8965597940556984,
      "grad_norm": 0.2126699984073639,
      "learning_rate": 4.505046572757309e-05,
      "loss": 0.0624,
      "step": 33300
    },
    {
      "epoch": 3.8965597940556984,
      "loss_main": 0.11047735065221786,
      "step": 33300
    },
    {
      "epoch": 3.8977299321319916,
      "grad_norm": 0.27893513441085815,
      "learning_rate": 4.502304400457749e-05,
      "loss": 0.064,
      "step": 33310
    },
    {
      "epoch": 3.8977299321319916,
      "loss_main": 0.03523977845907211,
      "step": 33310
    },
    {
      "epoch": 3.8989000702082848,
      "grad_norm": 0.41763821244239807,
      "learning_rate": 4.499562379344973e-05,
      "loss": 0.0516,
      "step": 33320
    },
    {
      "epoch": 3.8989000702082848,
      "loss_main": 0.05867176502943039,
      "step": 33320
    },
    {
      "epoch": 3.9000702082845775,
      "grad_norm": 0.20030036568641663,
      "learning_rate": 4.4968205102519306e-05,
      "loss": 0.0578,
      "step": 33330
    },
    {
      "epoch": 3.9000702082845775,
      "loss_main": 0.04386523365974426,
      "step": 33330
    },
    {
      "epoch": 3.9012403463608707,
      "grad_norm": 0.27697548270225525,
      "learning_rate": 4.494078794011532e-05,
      "loss": 0.0551,
      "step": 33340
    },
    {
      "epoch": 3.9012403463608707,
      "loss_main": 0.047999780625104904,
      "step": 33340
    },
    {
      "epoch": 3.9024104844371634,
      "grad_norm": 0.20449401438236237,
      "learning_rate": 4.491337231456639e-05,
      "loss": 0.0595,
      "step": 33350
    },
    {
      "epoch": 3.9024104844371634,
      "loss_main": 0.061256565153598785,
      "step": 33350
    },
    {
      "epoch": 3.9035806225134566,
      "grad_norm": 0.3208346664905548,
      "learning_rate": 4.4885958234200634e-05,
      "loss": 0.0607,
      "step": 33360
    },
    {
      "epoch": 3.9035806225134566,
      "loss_main": 0.05493537336587906,
      "step": 33360
    },
    {
      "epoch": 3.9047507605897493,
      "grad_norm": 0.3055182695388794,
      "learning_rate": 4.485854570734575e-05,
      "loss": 0.0608,
      "step": 33370
    },
    {
      "epoch": 3.9047507605897493,
      "loss_main": 0.09503325074911118,
      "step": 33370
    },
    {
      "epoch": 3.9059208986660425,
      "grad_norm": 0.29576438665390015,
      "learning_rate": 4.483113474232891e-05,
      "loss": 0.0617,
      "step": 33380
    },
    {
      "epoch": 3.9059208986660425,
      "loss_main": 0.05243782326579094,
      "step": 33380
    },
    {
      "epoch": 3.9070910367423357,
      "grad_norm": 0.2467595934867859,
      "learning_rate": 4.480372534747688e-05,
      "loss": 0.0608,
      "step": 33390
    },
    {
      "epoch": 3.9070910367423357,
      "loss_main": 0.05711567401885986,
      "step": 33390
    },
    {
      "epoch": 3.9082611748186284,
      "grad_norm": 0.23167897760868073,
      "learning_rate": 4.477631753111588e-05,
      "loss": 0.0696,
      "step": 33400
    },
    {
      "epoch": 3.9082611748186284,
      "loss_main": 0.06211390718817711,
      "step": 33400
    },
    {
      "epoch": 3.9094313128949216,
      "grad_norm": 0.2874937355518341,
      "learning_rate": 4.4748911301571686e-05,
      "loss": 0.0551,
      "step": 33410
    },
    {
      "epoch": 3.9094313128949216,
      "loss_main": 0.046940214931964874,
      "step": 33410
    },
    {
      "epoch": 3.910601450971215,
      "grad_norm": 0.2628917098045349,
      "learning_rate": 4.472150666716961e-05,
      "loss": 0.0521,
      "step": 33420
    },
    {
      "epoch": 3.910601450971215,
      "loss_main": 0.06442355364561081,
      "step": 33420
    },
    {
      "epoch": 3.9117715890475075,
      "grad_norm": 0.1912509799003601,
      "learning_rate": 4.469410363623442e-05,
      "loss": 0.0478,
      "step": 33430
    },
    {
      "epoch": 3.9117715890475075,
      "loss_main": 0.06666853278875351,
      "step": 33430
    },
    {
      "epoch": 3.9129417271238007,
      "grad_norm": 0.2897901237010956,
      "learning_rate": 4.466670221709044e-05,
      "loss": 0.0653,
      "step": 33440
    },
    {
      "epoch": 3.9129417271238007,
      "loss_main": 0.05187749117612839,
      "step": 33440
    },
    {
      "epoch": 3.9141118652000935,
      "grad_norm": 0.31233981251716614,
      "learning_rate": 4.463930241806154e-05,
      "loss": 0.0559,
      "step": 33450
    },
    {
      "epoch": 3.9141118652000935,
      "loss_main": 0.029907407239079475,
      "step": 33450
    },
    {
      "epoch": 3.9152820032763866,
      "grad_norm": 0.248018279671669,
      "learning_rate": 4.4611904247471006e-05,
      "loss": 0.0554,
      "step": 33460
    },
    {
      "epoch": 3.9152820032763866,
      "loss_main": 0.06641747057437897,
      "step": 33460
    },
    {
      "epoch": 3.9164521413526794,
      "grad_norm": 0.3203284442424774,
      "learning_rate": 4.458450771364171e-05,
      "loss": 0.0534,
      "step": 33470
    },
    {
      "epoch": 3.9164521413526794,
      "loss_main": 0.02313503436744213,
      "step": 33470
    },
    {
      "epoch": 3.9176222794289726,
      "grad_norm": 0.2687617540359497,
      "learning_rate": 4.4557112824895965e-05,
      "loss": 0.0536,
      "step": 33480
    },
    {
      "epoch": 3.9176222794289726,
      "loss_main": 0.09618585556745529,
      "step": 33480
    },
    {
      "epoch": 3.9187924175052657,
      "grad_norm": 0.2594654858112335,
      "learning_rate": 4.452971958955563e-05,
      "loss": 0.0598,
      "step": 33490
    },
    {
      "epoch": 3.9187924175052657,
      "loss_main": 0.05287559702992439,
      "step": 33490
    },
    {
      "epoch": 3.9199625555815585,
      "grad_norm": 0.2947354316711426,
      "learning_rate": 4.450232801594208e-05,
      "loss": 0.0581,
      "step": 33500
    },
    {
      "epoch": 3.9199625555815585,
      "loss_main": 0.06544044613838196,
      "step": 33500
    },
    {
      "epoch": 3.9211326936578517,
      "grad_norm": 0.24591144919395447,
      "learning_rate": 4.447493811237609e-05,
      "loss": 0.0576,
      "step": 33510
    },
    {
      "epoch": 3.9211326936578517,
      "loss_main": 0.06449379026889801,
      "step": 33510
    },
    {
      "epoch": 3.922302831734145,
      "grad_norm": 0.2697240114212036,
      "learning_rate": 4.444754988717804e-05,
      "loss": 0.0635,
      "step": 33520
    },
    {
      "epoch": 3.922302831734145,
      "loss_main": 0.06421690434217453,
      "step": 33520
    },
    {
      "epoch": 3.9234729698104376,
      "grad_norm": 0.24969854950904846,
      "learning_rate": 4.442016334866771e-05,
      "loss": 0.073,
      "step": 33530
    },
    {
      "epoch": 3.9234729698104376,
      "loss_main": 0.06113621965050697,
      "step": 33530
    },
    {
      "epoch": 3.9246431078867308,
      "grad_norm": 0.19997520744800568,
      "learning_rate": 4.4392778505164445e-05,
      "loss": 0.054,
      "step": 33540
    },
    {
      "epoch": 3.9246431078867308,
      "loss_main": 0.05420999601483345,
      "step": 33540
    },
    {
      "epoch": 3.9258132459630235,
      "grad_norm": 0.31601670384407043,
      "learning_rate": 4.436539536498702e-05,
      "loss": 0.0554,
      "step": 33550
    },
    {
      "epoch": 3.9258132459630235,
      "loss_main": 0.053221844136714935,
      "step": 33550
    },
    {
      "epoch": 3.9269833840393167,
      "grad_norm": 0.20122165977954865,
      "learning_rate": 4.433801393645369e-05,
      "loss": 0.0568,
      "step": 33560
    },
    {
      "epoch": 3.9269833840393167,
      "loss_main": 0.06415366381406784,
      "step": 33560
    },
    {
      "epoch": 3.9281535221156094,
      "grad_norm": 0.4122769832611084,
      "learning_rate": 4.431063422788226e-05,
      "loss": 0.0623,
      "step": 33570
    },
    {
      "epoch": 3.9281535221156094,
      "loss_main": 0.06763985008001328,
      "step": 33570
    },
    {
      "epoch": 3.9293236601919026,
      "grad_norm": 0.24012695252895355,
      "learning_rate": 4.428325624758991e-05,
      "loss": 0.0605,
      "step": 33580
    },
    {
      "epoch": 3.9293236601919026,
      "loss_main": 0.05425577610731125,
      "step": 33580
    },
    {
      "epoch": 3.930493798268196,
      "grad_norm": 0.27198275923728943,
      "learning_rate": 4.4255880003893366e-05,
      "loss": 0.0617,
      "step": 33590
    },
    {
      "epoch": 3.930493798268196,
      "loss_main": 0.04827151075005531,
      "step": 33590
    },
    {
      "epoch": 3.9316639363444885,
      "grad_norm": 0.2875168025493622,
      "learning_rate": 4.422850550510884e-05,
      "loss": 0.0499,
      "step": 33600
    },
    {
      "epoch": 3.9316639363444885,
      "loss_main": 0.07987865060567856,
      "step": 33600
    },
    {
      "epoch": 3.9328340744207817,
      "grad_norm": 0.18043522536754608,
      "learning_rate": 4.4201132759551934e-05,
      "loss": 0.0612,
      "step": 33610
    },
    {
      "epoch": 3.9328340744207817,
      "loss_main": 0.0895964726805687,
      "step": 33610
    },
    {
      "epoch": 3.934004212497075,
      "grad_norm": 0.18773573637008667,
      "learning_rate": 4.4173761775537804e-05,
      "loss": 0.062,
      "step": 33620
    },
    {
      "epoch": 3.934004212497075,
      "loss_main": 0.06087706238031387,
      "step": 33620
    },
    {
      "epoch": 3.9351743505733676,
      "grad_norm": 0.28431469202041626,
      "learning_rate": 4.414639256138099e-05,
      "loss": 0.0622,
      "step": 33630
    },
    {
      "epoch": 3.9351743505733676,
      "loss_main": 0.03384556993842125,
      "step": 33630
    },
    {
      "epoch": 3.936344488649661,
      "grad_norm": 0.2649593949317932,
      "learning_rate": 4.411902512539557e-05,
      "loss": 0.0602,
      "step": 33640
    },
    {
      "epoch": 3.936344488649661,
      "loss_main": 0.05201014503836632,
      "step": 33640
    },
    {
      "epoch": 3.9375146267259535,
      "grad_norm": 0.20851024985313416,
      "learning_rate": 4.4091659475895044e-05,
      "loss": 0.054,
      "step": 33650
    },
    {
      "epoch": 3.9375146267259535,
      "loss_main": 0.04741065576672554,
      "step": 33650
    },
    {
      "epoch": 3.9386847648022467,
      "grad_norm": 0.3085896670818329,
      "learning_rate": 4.406429562119235e-05,
      "loss": 0.0586,
      "step": 33660
    },
    {
      "epoch": 3.9386847648022467,
      "loss_main": 0.05114292353391647,
      "step": 33660
    },
    {
      "epoch": 3.9398549028785395,
      "grad_norm": 0.3463015854358673,
      "learning_rate": 4.4036933569599945e-05,
      "loss": 0.0666,
      "step": 33670
    },
    {
      "epoch": 3.9398549028785395,
      "loss_main": 0.05092858523130417,
      "step": 33670
    },
    {
      "epoch": 3.9410250409548326,
      "grad_norm": 0.20124703645706177,
      "learning_rate": 4.400957332942965e-05,
      "loss": 0.0608,
      "step": 33680
    },
    {
      "epoch": 3.9410250409548326,
      "loss_main": 0.056899476796388626,
      "step": 33680
    },
    {
      "epoch": 3.942195179031126,
      "grad_norm": 0.2154158651828766,
      "learning_rate": 4.3982214908992844e-05,
      "loss": 0.0539,
      "step": 33690
    },
    {
      "epoch": 3.942195179031126,
      "loss_main": 0.0711805671453476,
      "step": 33690
    },
    {
      "epoch": 3.9433653171074186,
      "grad_norm": 0.2883232831954956,
      "learning_rate": 4.3954858316600235e-05,
      "loss": 0.0585,
      "step": 33700
    },
    {
      "epoch": 3.9433653171074186,
      "loss_main": 0.05889277532696724,
      "step": 33700
    },
    {
      "epoch": 3.9445354551837117,
      "grad_norm": 0.4216189980506897,
      "learning_rate": 4.392750356056205e-05,
      "loss": 0.0638,
      "step": 33710
    },
    {
      "epoch": 3.9445354551837117,
      "loss_main": 0.044365670531988144,
      "step": 33710
    },
    {
      "epoch": 3.945705593260005,
      "grad_norm": 0.33299922943115234,
      "learning_rate": 4.390015064918798e-05,
      "loss": 0.0534,
      "step": 33720
    },
    {
      "epoch": 3.945705593260005,
      "loss_main": 0.049111153930425644,
      "step": 33720
    },
    {
      "epoch": 3.9468757313362977,
      "grad_norm": 0.35113805532455444,
      "learning_rate": 4.387279959078705e-05,
      "loss": 0.0557,
      "step": 33730
    },
    {
      "epoch": 3.9468757313362977,
      "loss_main": 0.04323728755116463,
      "step": 33730
    },
    {
      "epoch": 3.948045869412591,
      "grad_norm": 0.19891144335269928,
      "learning_rate": 4.384545039366786e-05,
      "loss": 0.0539,
      "step": 33740
    },
    {
      "epoch": 3.948045869412591,
      "loss_main": 0.06699712574481964,
      "step": 33740
    },
    {
      "epoch": 3.9492160074888836,
      "grad_norm": 0.3027670979499817,
      "learning_rate": 4.381810306613831e-05,
      "loss": 0.0682,
      "step": 33750
    },
    {
      "epoch": 3.9492160074888836,
      "loss_main": 0.08435426652431488,
      "step": 33750
    },
    {
      "epoch": 3.9503861455651768,
      "grad_norm": 0.26002824306488037,
      "learning_rate": 4.3790757616505826e-05,
      "loss": 0.0636,
      "step": 33760
    },
    {
      "epoch": 3.9503861455651768,
      "loss_main": 0.03813064843416214,
      "step": 33760
    },
    {
      "epoch": 3.9515562836414695,
      "grad_norm": 0.1847323328256607,
      "learning_rate": 4.376341405307725e-05,
      "loss": 0.0712,
      "step": 33770
    },
    {
      "epoch": 3.9515562836414695,
      "loss_main": 0.040692079812288284,
      "step": 33770
    },
    {
      "epoch": 3.9527264217177627,
      "grad_norm": 0.2780677080154419,
      "learning_rate": 4.37360723841588e-05,
      "loss": 0.0603,
      "step": 33780
    },
    {
      "epoch": 3.9527264217177627,
      "loss_main": 0.08733247220516205,
      "step": 33780
    },
    {
      "epoch": 3.953896559794056,
      "grad_norm": 0.38708966970443726,
      "learning_rate": 4.370873261805619e-05,
      "loss": 0.0719,
      "step": 33790
    },
    {
      "epoch": 3.953896559794056,
      "loss_main": 0.07694351673126221,
      "step": 33790
    },
    {
      "epoch": 3.9550666978703486,
      "grad_norm": 0.3371647894382477,
      "learning_rate": 4.368139476307449e-05,
      "loss": 0.0629,
      "step": 33800
    },
    {
      "epoch": 3.9550666978703486,
      "loss_main": 0.06490927189588547,
      "step": 33800
    },
    {
      "epoch": 3.9562368359466418,
      "grad_norm": 0.2859461307525635,
      "learning_rate": 4.365405882751822e-05,
      "loss": 0.061,
      "step": 33810
    },
    {
      "epoch": 3.9562368359466418,
      "loss_main": 0.060116350650787354,
      "step": 33810
    },
    {
      "epoch": 3.957406974022935,
      "grad_norm": 0.21322917938232422,
      "learning_rate": 4.3626724819691326e-05,
      "loss": 0.0541,
      "step": 33820
    },
    {
      "epoch": 3.957406974022935,
      "loss_main": 0.04417577013373375,
      "step": 33820
    },
    {
      "epoch": 3.9585771120992277,
      "grad_norm": 0.3398202359676361,
      "learning_rate": 4.359939274789715e-05,
      "loss": 0.0589,
      "step": 33830
    },
    {
      "epoch": 3.9585771120992277,
      "loss_main": 0.04070094972848892,
      "step": 33830
    },
    {
      "epoch": 3.959747250175521,
      "grad_norm": 0.36837145686149597,
      "learning_rate": 4.357206262043848e-05,
      "loss": 0.0516,
      "step": 33840
    },
    {
      "epoch": 3.959747250175521,
      "loss_main": 0.042479317635297775,
      "step": 33840
    },
    {
      "epoch": 3.9609173882518136,
      "grad_norm": 0.2255343496799469,
      "learning_rate": 4.354473444561745e-05,
      "loss": 0.0491,
      "step": 33850
    },
    {
      "epoch": 3.9609173882518136,
      "loss_main": 0.041277576237916946,
      "step": 33850
    },
    {
      "epoch": 3.962087526328107,
      "grad_norm": 0.3257489800453186,
      "learning_rate": 4.3517408231735644e-05,
      "loss": 0.0629,
      "step": 33860
    },
    {
      "epoch": 3.962087526328107,
      "loss_main": 0.05727090314030647,
      "step": 33860
    },
    {
      "epoch": 3.9632576644043995,
      "grad_norm": 0.2592338025569916,
      "learning_rate": 4.3490083987094086e-05,
      "loss": 0.056,
      "step": 33870
    },
    {
      "epoch": 3.9632576644043995,
      "loss_main": 0.03389892354607582,
      "step": 33870
    },
    {
      "epoch": 3.9644278024806927,
      "grad_norm": 0.23151849210262299,
      "learning_rate": 4.34627617199931e-05,
      "loss": 0.0631,
      "step": 33880
    },
    {
      "epoch": 3.9644278024806927,
      "loss_main": 0.04989377036690712,
      "step": 33880
    },
    {
      "epoch": 3.965597940556986,
      "grad_norm": 0.30235686898231506,
      "learning_rate": 4.3435441438732526e-05,
      "loss": 0.058,
      "step": 33890
    },
    {
      "epoch": 3.965597940556986,
      "loss_main": 0.0773269385099411,
      "step": 33890
    },
    {
      "epoch": 3.9667680786332786,
      "grad_norm": 0.3892601728439331,
      "learning_rate": 4.340812315161149e-05,
      "loss": 0.0611,
      "step": 33900
    },
    {
      "epoch": 3.9667680786332786,
      "loss_main": 0.06310885399580002,
      "step": 33900
    },
    {
      "epoch": 3.967938216709572,
      "grad_norm": 0.43203842639923096,
      "learning_rate": 4.338080686692859e-05,
      "loss": 0.0623,
      "step": 33910
    },
    {
      "epoch": 3.967938216709572,
      "loss_main": 0.05806777626276016,
      "step": 33910
    },
    {
      "epoch": 3.969108354785865,
      "grad_norm": 0.27989667654037476,
      "learning_rate": 4.3353492592981816e-05,
      "loss": 0.0595,
      "step": 33920
    },
    {
      "epoch": 3.969108354785865,
      "loss_main": 0.04549519345164299,
      "step": 33920
    },
    {
      "epoch": 3.9702784928621577,
      "grad_norm": 0.18446087837219238,
      "learning_rate": 4.3326180338068485e-05,
      "loss": 0.0622,
      "step": 33930
    },
    {
      "epoch": 3.9702784928621577,
      "loss_main": 0.037031710147857666,
      "step": 33930
    },
    {
      "epoch": 3.9714486309384505,
      "grad_norm": 0.3351353108882904,
      "learning_rate": 4.3298870110485356e-05,
      "loss": 0.0549,
      "step": 33940
    },
    {
      "epoch": 3.9714486309384505,
      "loss_main": 0.03610642999410629,
      "step": 33940
    },
    {
      "epoch": 3.9726187690147436,
      "grad_norm": 0.27092716097831726,
      "learning_rate": 4.3271561918528567e-05,
      "loss": 0.0484,
      "step": 33950
    },
    {
      "epoch": 3.9726187690147436,
      "loss_main": 0.035853464156389236,
      "step": 33950
    },
    {
      "epoch": 3.973788907091037,
      "grad_norm": 0.25015223026275635,
      "learning_rate": 4.324425577049359e-05,
      "loss": 0.0619,
      "step": 33960
    },
    {
      "epoch": 3.973788907091037,
      "loss_main": 0.07891092449426651,
      "step": 33960
    },
    {
      "epoch": 3.9749590451673296,
      "grad_norm": 0.25182703137397766,
      "learning_rate": 4.321695167467535e-05,
      "loss": 0.0667,
      "step": 33970
    },
    {
      "epoch": 3.9749590451673296,
      "loss_main": 0.03909725695848465,
      "step": 33970
    },
    {
      "epoch": 3.9761291832436227,
      "grad_norm": 0.2198122888803482,
      "learning_rate": 4.3189649639368093e-05,
      "loss": 0.0676,
      "step": 33980
    },
    {
      "epoch": 3.9761291832436227,
      "loss_main": 0.09146388620138168,
      "step": 33980
    },
    {
      "epoch": 3.977299321319916,
      "grad_norm": 0.24560880661010742,
      "learning_rate": 4.316234967286547e-05,
      "loss": 0.0598,
      "step": 33990
    },
    {
      "epoch": 3.977299321319916,
      "loss_main": 0.0592280812561512,
      "step": 33990
    },
    {
      "epoch": 3.9784694593962087,
      "grad_norm": 0.24194517731666565,
      "learning_rate": 4.313505178346046e-05,
      "loss": 0.07,
      "step": 34000
    },
    {
      "epoch": 3.9784694593962087,
      "loss_main": 0.02813102863729,
      "step": 34000
    },
    {
      "epoch": 3.979639597472502,
      "grad_norm": 0.2111768126487732,
      "learning_rate": 4.3107755979445465e-05,
      "loss": 0.0634,
      "step": 34010
    },
    {
      "epoch": 3.979639597472502,
      "loss_main": 0.05257958173751831,
      "step": 34010
    },
    {
      "epoch": 3.980809735548795,
      "grad_norm": 0.23639436066150665,
      "learning_rate": 4.308046226911224e-05,
      "loss": 0.057,
      "step": 34020
    },
    {
      "epoch": 3.980809735548795,
      "loss_main": 0.042722053825855255,
      "step": 34020
    },
    {
      "epoch": 3.9819798736250878,
      "grad_norm": 0.30748945474624634,
      "learning_rate": 4.305317066075185e-05,
      "loss": 0.0544,
      "step": 34030
    },
    {
      "epoch": 3.9819798736250878,
      "loss_main": 0.03707028180360794,
      "step": 34030
    },
    {
      "epoch": 3.9831500117013805,
      "grad_norm": 0.3102835714817047,
      "learning_rate": 4.302588116265482e-05,
      "loss": 0.0606,
      "step": 34040
    },
    {
      "epoch": 3.9831500117013805,
      "loss_main": 0.06602811068296432,
      "step": 34040
    },
    {
      "epoch": 3.9843201497776737,
      "grad_norm": 0.3272586464881897,
      "learning_rate": 4.299859378311094e-05,
      "loss": 0.0594,
      "step": 34050
    },
    {
      "epoch": 3.9843201497776737,
      "loss_main": 0.046397123485803604,
      "step": 34050
    },
    {
      "epoch": 3.985490287853967,
      "grad_norm": 0.30810219049453735,
      "learning_rate": 4.2971308530409424e-05,
      "loss": 0.0536,
      "step": 34060
    },
    {
      "epoch": 3.985490287853967,
      "loss_main": 0.08652911335229874,
      "step": 34060
    },
    {
      "epoch": 3.9866604259302596,
      "grad_norm": 0.2925373315811157,
      "learning_rate": 4.2944025412838765e-05,
      "loss": 0.0605,
      "step": 34070
    },
    {
      "epoch": 3.9866604259302596,
      "loss_main": 0.04821982607245445,
      "step": 34070
    },
    {
      "epoch": 3.987830564006553,
      "grad_norm": 0.33162301778793335,
      "learning_rate": 4.291674443868689e-05,
      "loss": 0.0575,
      "step": 34080
    },
    {
      "epoch": 3.987830564006553,
      "loss_main": 0.0547998771071434,
      "step": 34080
    },
    {
      "epoch": 3.989000702082846,
      "grad_norm": 0.2743644118309021,
      "learning_rate": 4.288946561624104e-05,
      "loss": 0.0616,
      "step": 34090
    },
    {
      "epoch": 3.989000702082846,
      "loss_main": 0.05149993300437927,
      "step": 34090
    },
    {
      "epoch": 3.9901708401591387,
      "grad_norm": 0.2578289806842804,
      "learning_rate": 4.2862188953787794e-05,
      "loss": 0.0668,
      "step": 34100
    },
    {
      "epoch": 3.9901708401591387,
      "loss_main": 0.04214504733681679,
      "step": 34100
    },
    {
      "epoch": 3.991340978235432,
      "grad_norm": 0.3597632944583893,
      "learning_rate": 4.283491445961308e-05,
      "loss": 0.0535,
      "step": 34110
    },
    {
      "epoch": 3.991340978235432,
      "loss_main": 0.08497198671102524,
      "step": 34110
    },
    {
      "epoch": 3.9925111163117246,
      "grad_norm": 0.18746529519557953,
      "learning_rate": 4.2807642142002155e-05,
      "loss": 0.0602,
      "step": 34120
    },
    {
      "epoch": 3.9925111163117246,
      "loss_main": 0.05748111382126808,
      "step": 34120
    },
    {
      "epoch": 3.993681254388018,
      "grad_norm": 0.29905131459236145,
      "learning_rate": 4.278037200923966e-05,
      "loss": 0.0544,
      "step": 34130
    },
    {
      "epoch": 3.993681254388018,
      "loss_main": 0.06193988397717476,
      "step": 34130
    },
    {
      "epoch": 3.9948513924643105,
      "grad_norm": 0.3970341682434082,
      "learning_rate": 4.275310406960953e-05,
      "loss": 0.0716,
      "step": 34140
    },
    {
      "epoch": 3.9948513924643105,
      "loss_main": 0.08437754213809967,
      "step": 34140
    },
    {
      "epoch": 3.9960215305406037,
      "grad_norm": 0.2625718116760254,
      "learning_rate": 4.272583833139502e-05,
      "loss": 0.0583,
      "step": 34150
    },
    {
      "epoch": 3.9960215305406037,
      "loss_main": 0.04169028252363205,
      "step": 34150
    },
    {
      "epoch": 3.997191668616897,
      "grad_norm": 0.27210283279418945,
      "learning_rate": 4.2698574802878794e-05,
      "loss": 0.0606,
      "step": 34160
    },
    {
      "epoch": 3.997191668616897,
      "loss_main": 0.06488745659589767,
      "step": 34160
    },
    {
      "epoch": 3.9983618066931896,
      "grad_norm": 0.2938211262226105,
      "learning_rate": 4.2671313492342734e-05,
      "loss": 0.0568,
      "step": 34170
    },
    {
      "epoch": 3.9983618066931896,
      "loss_main": 0.04061981663107872,
      "step": 34170
    },
    {
      "epoch": 3.999531944769483,
      "grad_norm": 0.28909891843795776,
      "learning_rate": 4.264405440806813e-05,
      "loss": 0.0519,
      "step": 34180
    },
    {
      "epoch": 3.999531944769483,
      "loss_main": 0.07904958724975586,
      "step": 34180
    },
    {
      "epoch": 4.000702082845776,
      "grad_norm": 0.31913521885871887,
      "learning_rate": 4.26167975583356e-05,
      "loss": 0.0665,
      "step": 34190
    },
    {
      "epoch": 4.000702082845776,
      "loss_main": 0.05667046830058098,
      "step": 34190
    },
    {
      "epoch": 4.001872220922069,
      "grad_norm": 0.2772892713546753,
      "learning_rate": 4.2589542951425e-05,
      "loss": 0.0498,
      "step": 34200
    },
    {
      "epoch": 4.001872220922069,
      "loss_main": 0.06591711193323135,
      "step": 34200
    },
    {
      "epoch": 4.0030423589983615,
      "grad_norm": 0.2799738347530365,
      "learning_rate": 4.2562290595615615e-05,
      "loss": 0.0629,
      "step": 34210
    },
    {
      "epoch": 4.0030423589983615,
      "loss_main": 0.07080234587192535,
      "step": 34210
    },
    {
      "epoch": 4.004212497074655,
      "grad_norm": 0.32189324498176575,
      "learning_rate": 4.2535040499185946e-05,
      "loss": 0.0665,
      "step": 34220
    },
    {
      "epoch": 4.004212497074655,
      "loss_main": 0.04903555288910866,
      "step": 34220
    },
    {
      "epoch": 4.005382635150948,
      "grad_norm": 0.3308403789997101,
      "learning_rate": 4.250779267041387e-05,
      "loss": 0.0684,
      "step": 34230
    },
    {
      "epoch": 4.005382635150948,
      "loss_main": 0.07303052395582199,
      "step": 34230
    },
    {
      "epoch": 4.006552773227241,
      "grad_norm": 0.21899531781673431,
      "learning_rate": 4.248054711757657e-05,
      "loss": 0.0636,
      "step": 34240
    },
    {
      "epoch": 4.006552773227241,
      "loss_main": 0.04949931427836418,
      "step": 34240
    },
    {
      "epoch": 4.007722911303534,
      "grad_norm": 0.25123468041419983,
      "learning_rate": 4.245330384895052e-05,
      "loss": 0.0599,
      "step": 34250
    },
    {
      "epoch": 4.007722911303534,
      "loss_main": 0.10030221194028854,
      "step": 34250
    },
    {
      "epoch": 4.008893049379827,
      "grad_norm": 0.24657082557678223,
      "learning_rate": 4.242606287281151e-05,
      "loss": 0.0723,
      "step": 34260
    },
    {
      "epoch": 4.008893049379827,
      "loss_main": 0.06288108974695206,
      "step": 34260
    },
    {
      "epoch": 4.01006318745612,
      "grad_norm": 0.2732296586036682,
      "learning_rate": 4.2398824197434595e-05,
      "loss": 0.0721,
      "step": 34270
    },
    {
      "epoch": 4.01006318745612,
      "loss_main": 0.061635199934244156,
      "step": 34270
    },
    {
      "epoch": 4.011233325532412,
      "grad_norm": 0.22371749579906464,
      "learning_rate": 4.23715878310942e-05,
      "loss": 0.0581,
      "step": 34280
    },
    {
      "epoch": 4.011233325532412,
      "loss_main": 0.05332941189408302,
      "step": 34280
    },
    {
      "epoch": 4.012403463608706,
      "grad_norm": 0.3061496615409851,
      "learning_rate": 4.234435378206402e-05,
      "loss": 0.0579,
      "step": 34290
    },
    {
      "epoch": 4.012403463608706,
      "loss_main": 0.04685266688466072,
      "step": 34290
    },
    {
      "epoch": 4.013573601684999,
      "grad_norm": 0.30140799283981323,
      "learning_rate": 4.2317122058617006e-05,
      "loss": 0.0647,
      "step": 34300
    },
    {
      "epoch": 4.013573601684999,
      "loss_main": 0.05128436163067818,
      "step": 34300
    },
    {
      "epoch": 4.0147437397612915,
      "grad_norm": 0.22889673709869385,
      "learning_rate": 4.2289892669025485e-05,
      "loss": 0.0626,
      "step": 34310
    },
    {
      "epoch": 4.0147437397612915,
      "loss_main": 0.03896462917327881,
      "step": 34310
    },
    {
      "epoch": 4.015913877837585,
      "grad_norm": 0.5340273380279541,
      "learning_rate": 4.226266562156097e-05,
      "loss": 0.0638,
      "step": 34320
    },
    {
      "epoch": 4.015913877837585,
      "loss_main": 0.04337102919816971,
      "step": 34320
    },
    {
      "epoch": 4.017084015913878,
      "grad_norm": 0.22975216805934906,
      "learning_rate": 4.223544092449435e-05,
      "loss": 0.0545,
      "step": 34330
    },
    {
      "epoch": 4.017084015913878,
      "loss_main": 0.05025966838002205,
      "step": 34330
    },
    {
      "epoch": 4.018254153990171,
      "grad_norm": 0.2616274654865265,
      "learning_rate": 4.2208218586095784e-05,
      "loss": 0.0576,
      "step": 34340
    },
    {
      "epoch": 4.018254153990171,
      "loss_main": 0.05723262578248978,
      "step": 34340
    },
    {
      "epoch": 4.019424292066464,
      "grad_norm": 0.23932601511478424,
      "learning_rate": 4.218099861463466e-05,
      "loss": 0.0535,
      "step": 34350
    },
    {
      "epoch": 4.019424292066464,
      "loss_main": 0.031580425798892975,
      "step": 34350
    },
    {
      "epoch": 4.020594430142757,
      "grad_norm": 0.19405701756477356,
      "learning_rate": 4.215378101837972e-05,
      "loss": 0.0603,
      "step": 34360
    },
    {
      "epoch": 4.020594430142757,
      "loss_main": 0.06099080294370651,
      "step": 34360
    },
    {
      "epoch": 4.02176456821905,
      "grad_norm": 0.2830038368701935,
      "learning_rate": 4.2126565805598937e-05,
      "loss": 0.0546,
      "step": 34370
    },
    {
      "epoch": 4.02176456821905,
      "loss_main": 0.04994703084230423,
      "step": 34370
    },
    {
      "epoch": 4.0229347062953424,
      "grad_norm": 0.192203551530838,
      "learning_rate": 4.209935298455957e-05,
      "loss": 0.054,
      "step": 34380
    },
    {
      "epoch": 4.0229347062953424,
      "loss_main": 0.02675136737525463,
      "step": 34380
    },
    {
      "epoch": 4.024104844371636,
      "grad_norm": 0.2892991602420807,
      "learning_rate": 4.207214256352817e-05,
      "loss": 0.0525,
      "step": 34390
    },
    {
      "epoch": 4.024104844371636,
      "loss_main": 0.04088952764868736,
      "step": 34390
    },
    {
      "epoch": 4.025274982447929,
      "grad_norm": 0.26536017656326294,
      "learning_rate": 4.2044934550770524e-05,
      "loss": 0.062,
      "step": 34400
    },
    {
      "epoch": 4.025274982447929,
      "loss_main": 0.10965367406606674,
      "step": 34400
    },
    {
      "epoch": 4.0264451205242215,
      "grad_norm": 0.2910819947719574,
      "learning_rate": 4.201772895455174e-05,
      "loss": 0.0628,
      "step": 34410
    },
    {
      "epoch": 4.0264451205242215,
      "loss_main": 0.07964861392974854,
      "step": 34410
    },
    {
      "epoch": 4.027615258600515,
      "grad_norm": 0.22906839847564697,
      "learning_rate": 4.199052578313613e-05,
      "loss": 0.0606,
      "step": 34420
    },
    {
      "epoch": 4.027615258600515,
      "loss_main": 0.04954269528388977,
      "step": 34420
    },
    {
      "epoch": 4.028785396676808,
      "grad_norm": 0.28097355365753174,
      "learning_rate": 4.1963325044787294e-05,
      "loss": 0.0529,
      "step": 34430
    },
    {
      "epoch": 4.028785396676808,
      "loss_main": 0.029118161648511887,
      "step": 34430
    },
    {
      "epoch": 4.029955534753101,
      "grad_norm": 0.2565463185310364,
      "learning_rate": 4.193612674776814e-05,
      "loss": 0.0581,
      "step": 34440
    },
    {
      "epoch": 4.029955534753101,
      "loss_main": 0.06579163670539856,
      "step": 34440
    },
    {
      "epoch": 4.031125672829394,
      "grad_norm": 0.2034560739994049,
      "learning_rate": 4.1908930900340745e-05,
      "loss": 0.0602,
      "step": 34450
    },
    {
      "epoch": 4.031125672829394,
      "loss_main": 0.07387411594390869,
      "step": 34450
    },
    {
      "epoch": 4.032295810905687,
      "grad_norm": 0.2591146230697632,
      "learning_rate": 4.1881737510766536e-05,
      "loss": 0.0539,
      "step": 34460
    },
    {
      "epoch": 4.032295810905687,
      "loss_main": 0.04304995760321617,
      "step": 34460
    },
    {
      "epoch": 4.03346594898198,
      "grad_norm": 0.25681671500205994,
      "learning_rate": 4.185454658730609e-05,
      "loss": 0.0558,
      "step": 34470
    },
    {
      "epoch": 4.03346594898198,
      "loss_main": 0.044781651347875595,
      "step": 34470
    },
    {
      "epoch": 4.0346360870582725,
      "grad_norm": 0.2358609288930893,
      "learning_rate": 4.1827358138219355e-05,
      "loss": 0.0566,
      "step": 34480
    },
    {
      "epoch": 4.0346360870582725,
      "loss_main": 0.0289528276771307,
      "step": 34480
    },
    {
      "epoch": 4.035806225134566,
      "grad_norm": 0.2904971241950989,
      "learning_rate": 4.1800172171765404e-05,
      "loss": 0.0639,
      "step": 34490
    },
    {
      "epoch": 4.035806225134566,
      "loss_main": 0.08190988749265671,
      "step": 34490
    },
    {
      "epoch": 4.036976363210859,
      "grad_norm": 0.22578845918178558,
      "learning_rate": 4.177298869620264e-05,
      "loss": 0.0517,
      "step": 34500
    },
    {
      "epoch": 4.036976363210859,
      "loss_main": 0.052008021622896194,
      "step": 34500
    },
    {
      "epoch": 4.038146501287152,
      "grad_norm": 0.22348210215568542,
      "learning_rate": 4.1745807719788705e-05,
      "loss": 0.059,
      "step": 34510
    },
    {
      "epoch": 4.038146501287152,
      "loss_main": 0.07083607465028763,
      "step": 34510
    },
    {
      "epoch": 4.039316639363445,
      "grad_norm": 0.2678380608558655,
      "learning_rate": 4.1718629250780445e-05,
      "loss": 0.061,
      "step": 34520
    },
    {
      "epoch": 4.039316639363445,
      "loss_main": 0.048120755702257156,
      "step": 34520
    },
    {
      "epoch": 4.040486777439738,
      "grad_norm": 0.2836136519908905,
      "learning_rate": 4.1691453297433956e-05,
      "loss": 0.0552,
      "step": 34530
    },
    {
      "epoch": 4.040486777439738,
      "loss_main": 0.06039174646139145,
      "step": 34530
    },
    {
      "epoch": 4.041656915516031,
      "grad_norm": 0.3979605436325073,
      "learning_rate": 4.166427986800457e-05,
      "loss": 0.0572,
      "step": 34540
    },
    {
      "epoch": 4.041656915516031,
      "loss_main": 0.04220307618379593,
      "step": 34540
    },
    {
      "epoch": 4.042827053592324,
      "grad_norm": 0.30780696868896484,
      "learning_rate": 4.163710897074688e-05,
      "loss": 0.0529,
      "step": 34550
    },
    {
      "epoch": 4.042827053592324,
      "loss_main": 0.034689273685216904,
      "step": 34550
    },
    {
      "epoch": 4.043997191668617,
      "grad_norm": 0.25125935673713684,
      "learning_rate": 4.1609940613914686e-05,
      "loss": 0.0623,
      "step": 34560
    },
    {
      "epoch": 4.043997191668617,
      "loss_main": 0.08669233322143555,
      "step": 34560
    },
    {
      "epoch": 4.04516732974491,
      "grad_norm": 0.1814316362142563,
      "learning_rate": 4.1582774805760996e-05,
      "loss": 0.0583,
      "step": 34570
    },
    {
      "epoch": 4.04516732974491,
      "loss_main": 0.05944153293967247,
      "step": 34570
    },
    {
      "epoch": 4.0463374678212025,
      "grad_norm": 0.25118014216423035,
      "learning_rate": 4.155561155453809e-05,
      "loss": 0.0521,
      "step": 34580
    },
    {
      "epoch": 4.0463374678212025,
      "loss_main": 0.02871052548289299,
      "step": 34580
    },
    {
      "epoch": 4.047507605897496,
      "grad_norm": 0.2737228572368622,
      "learning_rate": 4.15284508684974e-05,
      "loss": 0.0556,
      "step": 34590
    },
    {
      "epoch": 4.047507605897496,
      "loss_main": 0.04817608743906021,
      "step": 34590
    },
    {
      "epoch": 4.048677743973789,
      "grad_norm": 0.2910214364528656,
      "learning_rate": 4.1501292755889675e-05,
      "loss": 0.0649,
      "step": 34600
    },
    {
      "epoch": 4.048677743973789,
      "loss_main": 0.039499640464782715,
      "step": 34600
    },
    {
      "epoch": 4.049847882050082,
      "grad_norm": 0.23391017317771912,
      "learning_rate": 4.1474137224964833e-05,
      "loss": 0.0608,
      "step": 34610
    },
    {
      "epoch": 4.049847882050082,
      "loss_main": 0.06947993487119675,
      "step": 34610
    },
    {
      "epoch": 4.051018020126375,
      "grad_norm": 0.2872798442840576,
      "learning_rate": 4.144698428397197e-05,
      "loss": 0.0614,
      "step": 34620
    },
    {
      "epoch": 4.051018020126375,
      "loss_main": 0.05446186661720276,
      "step": 34620
    },
    {
      "epoch": 4.052188158202668,
      "grad_norm": 0.22557714581489563,
      "learning_rate": 4.1419833941159466e-05,
      "loss": 0.0509,
      "step": 34630
    },
    {
      "epoch": 4.052188158202668,
      "loss_main": 0.046319685876369476,
      "step": 34630
    },
    {
      "epoch": 4.053358296278961,
      "grad_norm": 0.22215978801250458,
      "learning_rate": 4.1392686204774846e-05,
      "loss": 0.0539,
      "step": 34640
    },
    {
      "epoch": 4.053358296278961,
      "loss_main": 0.027146058157086372,
      "step": 34640
    },
    {
      "epoch": 4.054528434355254,
      "grad_norm": 0.23942650854587555,
      "learning_rate": 4.13655410830649e-05,
      "loss": 0.0568,
      "step": 34650
    },
    {
      "epoch": 4.054528434355254,
      "loss_main": 0.07093347609043121,
      "step": 34650
    },
    {
      "epoch": 4.055698572431547,
      "grad_norm": 0.22165288031101227,
      "learning_rate": 4.1338398584275594e-05,
      "loss": 0.0659,
      "step": 34660
    },
    {
      "epoch": 4.055698572431547,
      "loss_main": 0.09082718938589096,
      "step": 34660
    },
    {
      "epoch": 4.05686871050784,
      "grad_norm": 0.27929165959358215,
      "learning_rate": 4.1311258716652104e-05,
      "loss": 0.0642,
      "step": 34670
    },
    {
      "epoch": 4.05686871050784,
      "loss_main": 0.0655597597360611,
      "step": 34670
    },
    {
      "epoch": 4.0580388485841326,
      "grad_norm": 0.27453598380088806,
      "learning_rate": 4.128412148843881e-05,
      "loss": 0.0599,
      "step": 34680
    },
    {
      "epoch": 4.0580388485841326,
      "loss_main": 0.030005738139152527,
      "step": 34680
    },
    {
      "epoch": 4.059208986660426,
      "grad_norm": 0.253811776638031,
      "learning_rate": 4.125698690787926e-05,
      "loss": 0.0569,
      "step": 34690
    },
    {
      "epoch": 4.059208986660426,
      "loss_main": 0.08660787343978882,
      "step": 34690
    },
    {
      "epoch": 4.060379124736719,
      "grad_norm": 0.1911969929933548,
      "learning_rate": 4.1229854983216245e-05,
      "loss": 0.0593,
      "step": 34700
    },
    {
      "epoch": 4.060379124736719,
      "loss_main": 0.055370036512613297,
      "step": 34700
    },
    {
      "epoch": 4.061549262813012,
      "grad_norm": 0.3933829367160797,
      "learning_rate": 4.120272572269175e-05,
      "loss": 0.0572,
      "step": 34710
    },
    {
      "epoch": 4.061549262813012,
      "loss_main": 0.07451905310153961,
      "step": 34710
    },
    {
      "epoch": 4.062719400889305,
      "grad_norm": 0.18283085525035858,
      "learning_rate": 4.117559913454687e-05,
      "loss": 0.0594,
      "step": 34720
    },
    {
      "epoch": 4.062719400889305,
      "loss_main": 0.08380603790283203,
      "step": 34720
    },
    {
      "epoch": 4.063889538965598,
      "grad_norm": 0.2601676285266876,
      "learning_rate": 4.114847522702201e-05,
      "loss": 0.0612,
      "step": 34730
    },
    {
      "epoch": 4.063889538965598,
      "loss_main": 0.05424513667821884,
      "step": 34730
    },
    {
      "epoch": 4.065059677041891,
      "grad_norm": 0.20711536705493927,
      "learning_rate": 4.112135400835664e-05,
      "loss": 0.0525,
      "step": 34740
    },
    {
      "epoch": 4.065059677041891,
      "loss_main": 0.07673932611942291,
      "step": 34740
    },
    {
      "epoch": 4.066229815118184,
      "grad_norm": 0.30267465114593506,
      "learning_rate": 4.109423548678949e-05,
      "loss": 0.048,
      "step": 34750
    },
    {
      "epoch": 4.066229815118184,
      "loss_main": 0.052265364676713943,
      "step": 34750
    },
    {
      "epoch": 4.067399953194477,
      "grad_norm": 0.21786653995513916,
      "learning_rate": 4.106711967055848e-05,
      "loss": 0.0596,
      "step": 34760
    },
    {
      "epoch": 4.067399953194477,
      "loss_main": 0.06138582155108452,
      "step": 34760
    },
    {
      "epoch": 4.06857009127077,
      "grad_norm": 0.27404624223709106,
      "learning_rate": 4.1040006567900636e-05,
      "loss": 0.0571,
      "step": 34770
    },
    {
      "epoch": 4.06857009127077,
      "loss_main": 0.05867017060518265,
      "step": 34770
    },
    {
      "epoch": 4.069740229347063,
      "grad_norm": 0.32677358388900757,
      "learning_rate": 4.101289618705224e-05,
      "loss": 0.0521,
      "step": 34780
    },
    {
      "epoch": 4.069740229347063,
      "loss_main": 0.09025858342647552,
      "step": 34780
    },
    {
      "epoch": 4.070910367423356,
      "grad_norm": 0.3144257962703705,
      "learning_rate": 4.0985788536248675e-05,
      "loss": 0.0575,
      "step": 34790
    },
    {
      "epoch": 4.070910367423356,
      "loss_main": 0.043265726417303085,
      "step": 34790
    },
    {
      "epoch": 4.072080505499649,
      "grad_norm": 0.33765316009521484,
      "learning_rate": 4.095868362372454e-05,
      "loss": 0.0506,
      "step": 34800
    },
    {
      "epoch": 4.072080505499649,
      "loss_main": 0.0704200342297554,
      "step": 34800
    },
    {
      "epoch": 4.073250643575942,
      "grad_norm": 0.3714568614959717,
      "learning_rate": 4.0931581457713614e-05,
      "loss": 0.063,
      "step": 34810
    },
    {
      "epoch": 4.073250643575942,
      "loss_main": 0.0668581873178482,
      "step": 34810
    },
    {
      "epoch": 4.074420781652235,
      "grad_norm": 0.18861791491508484,
      "learning_rate": 4.09044820464488e-05,
      "loss": 0.0622,
      "step": 34820
    },
    {
      "epoch": 4.074420781652235,
      "loss_main": 0.036245040595531464,
      "step": 34820
    },
    {
      "epoch": 4.075590919728528,
      "grad_norm": 0.284932404756546,
      "learning_rate": 4.087738539816219e-05,
      "loss": 0.0642,
      "step": 34830
    },
    {
      "epoch": 4.075590919728528,
      "loss_main": 0.0677604153752327,
      "step": 34830
    },
    {
      "epoch": 4.076761057804821,
      "grad_norm": 0.3610853850841522,
      "learning_rate": 4.085029152108501e-05,
      "loss": 0.0698,
      "step": 34840
    },
    {
      "epoch": 4.076761057804821,
      "loss_main": 0.030948910862207413,
      "step": 34840
    },
    {
      "epoch": 4.0779311958811135,
      "grad_norm": 0.36611950397491455,
      "learning_rate": 4.0823200423447714e-05,
      "loss": 0.0637,
      "step": 34850
    },
    {
      "epoch": 4.0779311958811135,
      "loss_main": 0.05282718315720558,
      "step": 34850
    },
    {
      "epoch": 4.079101333957407,
      "grad_norm": 0.28116223216056824,
      "learning_rate": 4.079611211347981e-05,
      "loss": 0.0575,
      "step": 34860
    },
    {
      "epoch": 4.079101333957407,
      "loss_main": 0.09575562924146652,
      "step": 34860
    },
    {
      "epoch": 4.0802714720337,
      "grad_norm": 0.22940653562545776,
      "learning_rate": 4.076902659941002e-05,
      "loss": 0.0527,
      "step": 34870
    },
    {
      "epoch": 4.0802714720337,
      "loss_main": 0.046903155744075775,
      "step": 34870
    },
    {
      "epoch": 4.081441610109993,
      "grad_norm": 0.23135997354984283,
      "learning_rate": 4.074194388946624e-05,
      "loss": 0.0531,
      "step": 34880
    },
    {
      "epoch": 4.081441610109993,
      "loss_main": 0.04794689640402794,
      "step": 34880
    },
    {
      "epoch": 4.082611748186286,
      "grad_norm": 0.24067062139511108,
      "learning_rate": 4.071486399187545e-05,
      "loss": 0.0524,
      "step": 34890
    },
    {
      "epoch": 4.082611748186286,
      "loss_main": 0.058033108711242676,
      "step": 34890
    },
    {
      "epoch": 4.083781886262579,
      "grad_norm": 0.30242231488227844,
      "learning_rate": 4.0687786914863836e-05,
      "loss": 0.0562,
      "step": 34900
    },
    {
      "epoch": 4.083781886262579,
      "loss_main": 0.04018035903573036,
      "step": 34900
    },
    {
      "epoch": 4.084952024338872,
      "grad_norm": 0.3896670639514923,
      "learning_rate": 4.0660712666656666e-05,
      "loss": 0.0593,
      "step": 34910
    },
    {
      "epoch": 4.084952024338872,
      "loss_main": 0.07305993139743805,
      "step": 34910
    },
    {
      "epoch": 4.086122162415165,
      "grad_norm": 0.22444257140159607,
      "learning_rate": 4.0633641255478394e-05,
      "loss": 0.0596,
      "step": 34920
    },
    {
      "epoch": 4.086122162415165,
      "loss_main": 0.04416195675730705,
      "step": 34920
    },
    {
      "epoch": 4.087292300491458,
      "grad_norm": 0.1914241909980774,
      "learning_rate": 4.0606572689552624e-05,
      "loss": 0.0581,
      "step": 34930
    },
    {
      "epoch": 4.087292300491458,
      "loss_main": 0.06883155554533005,
      "step": 34930
    },
    {
      "epoch": 4.088462438567751,
      "grad_norm": 0.23830734193325043,
      "learning_rate": 4.0579506977102036e-05,
      "loss": 0.0621,
      "step": 34940
    },
    {
      "epoch": 4.088462438567751,
      "loss_main": 0.06415657699108124,
      "step": 34940
    },
    {
      "epoch": 4.089632576644044,
      "grad_norm": 0.3466104567050934,
      "learning_rate": 4.055244412634849e-05,
      "loss": 0.0577,
      "step": 34950
    },
    {
      "epoch": 4.089632576644044,
      "loss_main": 0.038041871041059494,
      "step": 34950
    },
    {
      "epoch": 4.090802714720337,
      "grad_norm": 0.24719102680683136,
      "learning_rate": 4.052538414551298e-05,
      "loss": 0.0548,
      "step": 34960
    },
    {
      "epoch": 4.090802714720337,
      "loss_main": 0.14149734377861023,
      "step": 34960
    },
    {
      "epoch": 4.09197285279663,
      "grad_norm": 0.17325855791568756,
      "learning_rate": 4.0498327042815596e-05,
      "loss": 0.059,
      "step": 34970
    },
    {
      "epoch": 4.09197285279663,
      "loss_main": 0.06511719524860382,
      "step": 34970
    },
    {
      "epoch": 4.093142990872923,
      "grad_norm": 0.272998183965683,
      "learning_rate": 4.047127282647559e-05,
      "loss": 0.062,
      "step": 34980
    },
    {
      "epoch": 4.093142990872923,
      "loss_main": 0.05986936762928963,
      "step": 34980
    },
    {
      "epoch": 4.094313128949216,
      "grad_norm": 0.24925725162029266,
      "learning_rate": 4.04442215047113e-05,
      "loss": 0.0521,
      "step": 34990
    },
    {
      "epoch": 4.094313128949216,
      "loss_main": 0.056429680436849594,
      "step": 34990
    },
    {
      "epoch": 4.095483267025509,
      "grad_norm": 0.25593093037605286,
      "learning_rate": 4.041717308574023e-05,
      "loss": 0.0525,
      "step": 35000
    },
    {
      "epoch": 4.095483267025509,
      "loss_main": 0.05670682713389397,
      "step": 35000
    },
    {
      "epoch": 4.096653405101802,
      "grad_norm": 0.22480551898479462,
      "learning_rate": 4.039012757777893e-05,
      "loss": 0.0568,
      "step": 35010
    },
    {
      "epoch": 4.096653405101802,
      "loss_main": 0.04286107048392296,
      "step": 35010
    },
    {
      "epoch": 4.097823543178095,
      "grad_norm": 0.1912001520395279,
      "learning_rate": 4.036308498904314e-05,
      "loss": 0.0504,
      "step": 35020
    },
    {
      "epoch": 4.097823543178095,
      "loss_main": 0.1132715493440628,
      "step": 35020
    },
    {
      "epoch": 4.098993681254388,
      "grad_norm": 0.2293396294116974,
      "learning_rate": 4.033604532774771e-05,
      "loss": 0.0598,
      "step": 35030
    },
    {
      "epoch": 4.098993681254388,
      "loss_main": 0.07159461081027985,
      "step": 35030
    },
    {
      "epoch": 4.100163819330681,
      "grad_norm": 0.230413556098938,
      "learning_rate": 4.030900860210652e-05,
      "loss": 0.0546,
      "step": 35040
    },
    {
      "epoch": 4.100163819330681,
      "loss_main": 0.09048370271921158,
      "step": 35040
    },
    {
      "epoch": 4.101333957406974,
      "grad_norm": 0.2342834174633026,
      "learning_rate": 4.028197482033266e-05,
      "loss": 0.0547,
      "step": 35050
    },
    {
      "epoch": 4.101333957406974,
      "loss_main": 0.04130849987268448,
      "step": 35050
    },
    {
      "epoch": 4.102504095483267,
      "grad_norm": 0.2883073389530182,
      "learning_rate": 4.0254943990638246e-05,
      "loss": 0.0606,
      "step": 35060
    },
    {
      "epoch": 4.102504095483267,
      "loss_main": 0.04177016019821167,
      "step": 35060
    },
    {
      "epoch": 4.10367423355956,
      "grad_norm": 0.23630750179290771,
      "learning_rate": 4.022791612123454e-05,
      "loss": 0.0658,
      "step": 35070
    },
    {
      "epoch": 4.10367423355956,
      "loss_main": 0.0414331816136837,
      "step": 35070
    },
    {
      "epoch": 4.104844371635853,
      "grad_norm": 0.19363291561603546,
      "learning_rate": 4.020089122033192e-05,
      "loss": 0.0541,
      "step": 35080
    },
    {
      "epoch": 4.104844371635853,
      "loss_main": 0.04998376965522766,
      "step": 35080
    },
    {
      "epoch": 4.106014509712146,
      "grad_norm": 0.24244530498981476,
      "learning_rate": 4.01738692961398e-05,
      "loss": 0.0682,
      "step": 35090
    },
    {
      "epoch": 4.106014509712146,
      "loss_main": 0.018550973385572433,
      "step": 35090
    },
    {
      "epoch": 4.107184647788439,
      "grad_norm": 0.2998017966747284,
      "learning_rate": 4.014685035686675e-05,
      "loss": 0.0475,
      "step": 35100
    },
    {
      "epoch": 4.107184647788439,
      "loss_main": 0.0774710401892662,
      "step": 35100
    },
    {
      "epoch": 4.108354785864732,
      "grad_norm": 0.27320465445518494,
      "learning_rate": 4.011983441072039e-05,
      "loss": 0.0553,
      "step": 35110
    },
    {
      "epoch": 4.108354785864732,
      "loss_main": 0.08031696081161499,
      "step": 35110
    },
    {
      "epoch": 4.109524923941025,
      "grad_norm": 0.22651268541812897,
      "learning_rate": 4.0092821465907485e-05,
      "loss": 0.066,
      "step": 35120
    },
    {
      "epoch": 4.109524923941025,
      "loss_main": 0.0376858226954937,
      "step": 35120
    },
    {
      "epoch": 4.110695062017318,
      "grad_norm": 0.28989794850349426,
      "learning_rate": 4.006581153063383e-05,
      "loss": 0.0508,
      "step": 35130
    },
    {
      "epoch": 4.110695062017318,
      "loss_main": 0.055361807346343994,
      "step": 35130
    },
    {
      "epoch": 4.111865200093611,
      "grad_norm": 0.24835896492004395,
      "learning_rate": 4.003880461310432e-05,
      "loss": 0.0505,
      "step": 35140
    },
    {
      "epoch": 4.111865200093611,
      "loss_main": 0.0629313662648201,
      "step": 35140
    },
    {
      "epoch": 4.113035338169904,
      "grad_norm": 0.2575612962245941,
      "learning_rate": 4.001180072152298e-05,
      "loss": 0.0585,
      "step": 35150
    },
    {
      "epoch": 4.113035338169904,
      "loss_main": 0.04440847784280777,
      "step": 35150
    },
    {
      "epoch": 4.114205476246197,
      "grad_norm": 0.2271091192960739,
      "learning_rate": 3.998479986409285e-05,
      "loss": 0.0589,
      "step": 35160
    },
    {
      "epoch": 4.114205476246197,
      "loss_main": 0.056931622326374054,
      "step": 35160
    },
    {
      "epoch": 4.11537561432249,
      "grad_norm": 0.28196990489959717,
      "learning_rate": 3.995780204901607e-05,
      "loss": 0.0557,
      "step": 35170
    },
    {
      "epoch": 4.11537561432249,
      "loss_main": 0.04491370543837547,
      "step": 35170
    },
    {
      "epoch": 4.116545752398783,
      "grad_norm": 0.2377714067697525,
      "learning_rate": 3.993080728449391e-05,
      "loss": 0.0505,
      "step": 35180
    },
    {
      "epoch": 4.116545752398783,
      "loss_main": 0.03474411740899086,
      "step": 35180
    },
    {
      "epoch": 4.117715890475076,
      "grad_norm": 0.2821425795555115,
      "learning_rate": 3.990381557872661e-05,
      "loss": 0.0555,
      "step": 35190
    },
    {
      "epoch": 4.117715890475076,
      "loss_main": 0.06333621591329575,
      "step": 35190
    },
    {
      "epoch": 4.118886028551369,
      "grad_norm": 0.19237762689590454,
      "learning_rate": 3.987682693991359e-05,
      "loss": 0.0563,
      "step": 35200
    },
    {
      "epoch": 4.118886028551369,
      "loss_main": 0.11020813882350922,
      "step": 35200
    },
    {
      "epoch": 4.120056166627662,
      "grad_norm": 0.21360869705677032,
      "learning_rate": 3.9849841376253226e-05,
      "loss": 0.0445,
      "step": 35210
    },
    {
      "epoch": 4.120056166627662,
      "loss_main": 0.04640192538499832,
      "step": 35210
    },
    {
      "epoch": 4.1212263047039555,
      "grad_norm": 0.2366233617067337,
      "learning_rate": 3.982285889594306e-05,
      "loss": 0.0519,
      "step": 35220
    },
    {
      "epoch": 4.1212263047039555,
      "loss_main": 0.0752435252070427,
      "step": 35220
    },
    {
      "epoch": 4.122396442780248,
      "grad_norm": 0.26386335492134094,
      "learning_rate": 3.9795879507179665e-05,
      "loss": 0.0607,
      "step": 35230
    },
    {
      "epoch": 4.122396442780248,
      "loss_main": 0.0672852024435997,
      "step": 35230
    },
    {
      "epoch": 4.123566580856541,
      "grad_norm": 0.2879360616207123,
      "learning_rate": 3.9768903218158634e-05,
      "loss": 0.0569,
      "step": 35240
    },
    {
      "epoch": 4.123566580856541,
      "loss_main": 0.04194760322570801,
      "step": 35240
    },
    {
      "epoch": 4.124736718932834,
      "grad_norm": 0.22603733837604523,
      "learning_rate": 3.974193003707468e-05,
      "loss": 0.0567,
      "step": 35250
    },
    {
      "epoch": 4.124736718932834,
      "loss_main": 0.06593571603298187,
      "step": 35250
    },
    {
      "epoch": 4.125906857009127,
      "grad_norm": 0.2244040071964264,
      "learning_rate": 3.971495997212152e-05,
      "loss": 0.0529,
      "step": 35260
    },
    {
      "epoch": 4.125906857009127,
      "loss_main": 0.05078619346022606,
      "step": 35260
    },
    {
      "epoch": 4.12707699508542,
      "grad_norm": 0.35893356800079346,
      "learning_rate": 3.9687993031491985e-05,
      "loss": 0.059,
      "step": 35270
    },
    {
      "epoch": 4.12707699508542,
      "loss_main": 0.07641662657260895,
      "step": 35270
    },
    {
      "epoch": 4.128247133161713,
      "grad_norm": 0.3065250515937805,
      "learning_rate": 3.966102922337787e-05,
      "loss": 0.0642,
      "step": 35280
    },
    {
      "epoch": 4.128247133161713,
      "loss_main": 0.0625324472784996,
      "step": 35280
    },
    {
      "epoch": 4.129417271238006,
      "grad_norm": 0.2414228916168213,
      "learning_rate": 3.963406855597009e-05,
      "loss": 0.0565,
      "step": 35290
    },
    {
      "epoch": 4.129417271238006,
      "loss_main": 0.07248417288064957,
      "step": 35290
    },
    {
      "epoch": 4.130587409314299,
      "grad_norm": 0.315246045589447,
      "learning_rate": 3.960711103745861e-05,
      "loss": 0.0548,
      "step": 35300
    },
    {
      "epoch": 4.130587409314299,
      "loss_main": 0.06289388984441757,
      "step": 35300
    },
    {
      "epoch": 4.131757547390592,
      "grad_norm": 0.31218892335891724,
      "learning_rate": 3.958015667603237e-05,
      "loss": 0.0609,
      "step": 35310
    },
    {
      "epoch": 4.131757547390592,
      "loss_main": 0.06890688836574554,
      "step": 35310
    },
    {
      "epoch": 4.1329276854668855,
      "grad_norm": 0.23270095884799957,
      "learning_rate": 3.955320547987943e-05,
      "loss": 0.0519,
      "step": 35320
    },
    {
      "epoch": 4.1329276854668855,
      "loss_main": 0.0627308115363121,
      "step": 35320
    },
    {
      "epoch": 4.134097823543178,
      "grad_norm": 0.2546514570713043,
      "learning_rate": 3.952625745718681e-05,
      "loss": 0.0666,
      "step": 35330
    },
    {
      "epoch": 4.134097823543178,
      "loss_main": 0.07376635074615479,
      "step": 35330
    },
    {
      "epoch": 4.135267961619471,
      "grad_norm": 0.24720405042171478,
      "learning_rate": 3.949931261614064e-05,
      "loss": 0.0622,
      "step": 35340
    },
    {
      "epoch": 4.135267961619471,
      "loss_main": 0.05486065149307251,
      "step": 35340
    },
    {
      "epoch": 4.136438099695764,
      "grad_norm": 0.347444623708725,
      "learning_rate": 3.947237096492605e-05,
      "loss": 0.048,
      "step": 35350
    },
    {
      "epoch": 4.136438099695764,
      "loss_main": 0.05583910271525383,
      "step": 35350
    },
    {
      "epoch": 4.137608237772057,
      "grad_norm": 0.30964362621307373,
      "learning_rate": 3.944543251172719e-05,
      "loss": 0.0629,
      "step": 35360
    },
    {
      "epoch": 4.137608237772057,
      "loss_main": 0.04062562808394432,
      "step": 35360
    },
    {
      "epoch": 4.13877837584835,
      "grad_norm": 0.3035465478897095,
      "learning_rate": 3.941849726472725e-05,
      "loss": 0.0569,
      "step": 35370
    },
    {
      "epoch": 4.13877837584835,
      "loss_main": 0.10504647344350815,
      "step": 35370
    },
    {
      "epoch": 4.139948513924643,
      "grad_norm": 0.2737233638763428,
      "learning_rate": 3.939156523210846e-05,
      "loss": 0.0562,
      "step": 35380
    },
    {
      "epoch": 4.139948513924643,
      "loss_main": 0.09944763034582138,
      "step": 35380
    },
    {
      "epoch": 4.141118652000936,
      "grad_norm": 0.2735597789287567,
      "learning_rate": 3.9364636422052046e-05,
      "loss": 0.0659,
      "step": 35390
    },
    {
      "epoch": 4.141118652000936,
      "loss_main": 0.047438476234674454,
      "step": 35390
    },
    {
      "epoch": 4.142288790077229,
      "grad_norm": 0.3300015330314636,
      "learning_rate": 3.933771084273828e-05,
      "loss": 0.0673,
      "step": 35400
    },
    {
      "epoch": 4.142288790077229,
      "loss_main": 0.07038001716136932,
      "step": 35400
    },
    {
      "epoch": 4.143458928153522,
      "grad_norm": 0.2613448202610016,
      "learning_rate": 3.931078850234643e-05,
      "loss": 0.0541,
      "step": 35410
    },
    {
      "epoch": 4.143458928153522,
      "loss_main": 0.03958001360297203,
      "step": 35410
    },
    {
      "epoch": 4.1446290662298155,
      "grad_norm": 0.28234368562698364,
      "learning_rate": 3.928386940905483e-05,
      "loss": 0.0543,
      "step": 35420
    },
    {
      "epoch": 4.1446290662298155,
      "loss_main": 0.04729918763041496,
      "step": 35420
    },
    {
      "epoch": 4.145799204306108,
      "grad_norm": 0.17791451513767242,
      "learning_rate": 3.925695357104073e-05,
      "loss": 0.051,
      "step": 35430
    },
    {
      "epoch": 4.145799204306108,
      "loss_main": 0.07832404226064682,
      "step": 35430
    },
    {
      "epoch": 4.146969342382401,
      "grad_norm": 0.25225916504859924,
      "learning_rate": 3.923004099648049e-05,
      "loss": 0.0618,
      "step": 35440
    },
    {
      "epoch": 4.146969342382401,
      "loss_main": 0.06585375219583511,
      "step": 35440
    },
    {
      "epoch": 4.148139480458694,
      "grad_norm": 0.21865540742874146,
      "learning_rate": 3.920313169354944e-05,
      "loss": 0.0544,
      "step": 35450
    },
    {
      "epoch": 4.148139480458694,
      "loss_main": 0.055365435779094696,
      "step": 35450
    },
    {
      "epoch": 4.149309618534987,
      "grad_norm": 0.21390533447265625,
      "learning_rate": 3.9176225670421897e-05,
      "loss": 0.0634,
      "step": 35460
    },
    {
      "epoch": 4.149309618534987,
      "loss_main": 0.0454922579228878,
      "step": 35460
    },
    {
      "epoch": 4.15047975661128,
      "grad_norm": 0.26491227746009827,
      "learning_rate": 3.9149322935271224e-05,
      "loss": 0.0604,
      "step": 35470
    },
    {
      "epoch": 4.15047975661128,
      "loss_main": 0.04484427720308304,
      "step": 35470
    },
    {
      "epoch": 4.151649894687573,
      "grad_norm": 0.2583899199962616,
      "learning_rate": 3.9122423496269725e-05,
      "loss": 0.0638,
      "step": 35480
    },
    {
      "epoch": 4.151649894687573,
      "loss_main": 0.03095192089676857,
      "step": 35480
    },
    {
      "epoch": 4.1528200327638665,
      "grad_norm": 0.22706715762615204,
      "learning_rate": 3.909552736158877e-05,
      "loss": 0.0511,
      "step": 35490
    },
    {
      "epoch": 4.1528200327638665,
      "loss_main": 0.062357570976018906,
      "step": 35490
    },
    {
      "epoch": 4.153990170840159,
      "grad_norm": 0.277823805809021,
      "learning_rate": 3.90686345393987e-05,
      "loss": 0.0503,
      "step": 35500
    },
    {
      "epoch": 4.153990170840159,
      "loss_main": 0.044633958488702774,
      "step": 35500
    },
    {
      "epoch": 4.155160308916452,
      "grad_norm": 0.21368032693862915,
      "learning_rate": 3.9041745037868816e-05,
      "loss": 0.0532,
      "step": 35510
    },
    {
      "epoch": 4.155160308916452,
      "loss_main": 0.04935331270098686,
      "step": 35510
    },
    {
      "epoch": 4.156330446992746,
      "grad_norm": 0.2523324489593506,
      "learning_rate": 3.9014858865167465e-05,
      "loss": 0.0619,
      "step": 35520
    },
    {
      "epoch": 4.156330446992746,
      "loss_main": 0.07794436067342758,
      "step": 35520
    },
    {
      "epoch": 4.157500585069038,
      "grad_norm": 0.3218735456466675,
      "learning_rate": 3.8987976029461935e-05,
      "loss": 0.0676,
      "step": 35530
    },
    {
      "epoch": 4.157500585069038,
      "loss_main": 0.038562413305044174,
      "step": 35530
    },
    {
      "epoch": 4.158670723145331,
      "grad_norm": 0.23056194186210632,
      "learning_rate": 3.896109653891853e-05,
      "loss": 0.0537,
      "step": 35540
    },
    {
      "epoch": 4.158670723145331,
      "loss_main": 0.08904916793107986,
      "step": 35540
    },
    {
      "epoch": 4.159840861221624,
      "grad_norm": 0.28673627972602844,
      "learning_rate": 3.893422040170254e-05,
      "loss": 0.0517,
      "step": 35550
    },
    {
      "epoch": 4.159840861221624,
      "loss_main": 0.08748217672109604,
      "step": 35550
    },
    {
      "epoch": 4.161010999297917,
      "grad_norm": 0.2758716940879822,
      "learning_rate": 3.8907347625978207e-05,
      "loss": 0.0599,
      "step": 35560
    },
    {
      "epoch": 4.161010999297917,
      "loss_main": 0.12117629498243332,
      "step": 35560
    },
    {
      "epoch": 4.16218113737421,
      "grad_norm": 0.29058465361595154,
      "learning_rate": 3.88804782199088e-05,
      "loss": 0.0595,
      "step": 35570
    },
    {
      "epoch": 4.16218113737421,
      "loss_main": 0.056644078344106674,
      "step": 35570
    },
    {
      "epoch": 4.163351275450503,
      "grad_norm": 0.23701246082782745,
      "learning_rate": 3.8853612191656495e-05,
      "loss": 0.0521,
      "step": 35580
    },
    {
      "epoch": 4.163351275450503,
      "loss_main": 0.05858878046274185,
      "step": 35580
    },
    {
      "epoch": 4.1645214135267965,
      "grad_norm": 0.3151393234729767,
      "learning_rate": 3.88267495493825e-05,
      "loss": 0.0565,
      "step": 35590
    },
    {
      "epoch": 4.1645214135267965,
      "loss_main": 0.05312692001461983,
      "step": 35590
    },
    {
      "epoch": 4.165691551603089,
      "grad_norm": 0.21150362491607666,
      "learning_rate": 3.8799890301247004e-05,
      "loss": 0.0576,
      "step": 35600
    },
    {
      "epoch": 4.165691551603089,
      "loss_main": 0.06842724978923798,
      "step": 35600
    },
    {
      "epoch": 4.166861689679382,
      "grad_norm": 0.29918649792671204,
      "learning_rate": 3.8773034455409096e-05,
      "loss": 0.059,
      "step": 35610
    },
    {
      "epoch": 4.166861689679382,
      "loss_main": 0.07104761153459549,
      "step": 35610
    },
    {
      "epoch": 4.168031827755675,
      "grad_norm": 0.37680545449256897,
      "learning_rate": 3.8746182020026904e-05,
      "loss": 0.0576,
      "step": 35620
    },
    {
      "epoch": 4.168031827755675,
      "loss_main": 0.059236329048871994,
      "step": 35620
    },
    {
      "epoch": 4.169201965831968,
      "grad_norm": 0.24084334075450897,
      "learning_rate": 3.871933300325745e-05,
      "loss": 0.0493,
      "step": 35630
    },
    {
      "epoch": 4.169201965831968,
      "loss_main": 0.04701637849211693,
      "step": 35630
    },
    {
      "epoch": 4.170372103908261,
      "grad_norm": 0.29208141565322876,
      "learning_rate": 3.869248741325679e-05,
      "loss": 0.0623,
      "step": 35640
    },
    {
      "epoch": 4.170372103908261,
      "loss_main": 0.048084504902362823,
      "step": 35640
    },
    {
      "epoch": 4.171542241984554,
      "grad_norm": 0.24232973158359528,
      "learning_rate": 3.866564525817992e-05,
      "loss": 0.0508,
      "step": 35650
    },
    {
      "epoch": 4.171542241984554,
      "loss_main": 0.05303511396050453,
      "step": 35650
    },
    {
      "epoch": 4.172712380060847,
      "grad_norm": 0.3425080180168152,
      "learning_rate": 3.8638806546180725e-05,
      "loss": 0.0574,
      "step": 35660
    },
    {
      "epoch": 4.172712380060847,
      "loss_main": 0.05004410073161125,
      "step": 35660
    },
    {
      "epoch": 4.17388251813714,
      "grad_norm": 0.24553456902503967,
      "learning_rate": 3.861197128541213e-05,
      "loss": 0.059,
      "step": 35670
    },
    {
      "epoch": 4.17388251813714,
      "loss_main": 0.05273532122373581,
      "step": 35670
    },
    {
      "epoch": 4.175052656213433,
      "grad_norm": 0.2643947899341583,
      "learning_rate": 3.858513948402599e-05,
      "loss": 0.0569,
      "step": 35680
    },
    {
      "epoch": 4.175052656213433,
      "loss_main": 0.04378928616642952,
      "step": 35680
    },
    {
      "epoch": 4.1762227942897265,
      "grad_norm": 0.22509904205799103,
      "learning_rate": 3.8558311150173077e-05,
      "loss": 0.073,
      "step": 35690
    },
    {
      "epoch": 4.1762227942897265,
      "loss_main": 0.053788188844919205,
      "step": 35690
    },
    {
      "epoch": 4.177392932366019,
      "grad_norm": 0.22793371975421906,
      "learning_rate": 3.853148629200312e-05,
      "loss": 0.0616,
      "step": 35700
    },
    {
      "epoch": 4.177392932366019,
      "loss_main": 0.04483144357800484,
      "step": 35700
    },
    {
      "epoch": 4.178563070442312,
      "grad_norm": 0.26122382283210754,
      "learning_rate": 3.850466491766482e-05,
      "loss": 0.0577,
      "step": 35710
    },
    {
      "epoch": 4.178563070442312,
      "loss_main": 0.03596010059118271,
      "step": 35710
    },
    {
      "epoch": 4.179733208518606,
      "grad_norm": 0.2899478077888489,
      "learning_rate": 3.847784703530583e-05,
      "loss": 0.0512,
      "step": 35720
    },
    {
      "epoch": 4.179733208518606,
      "loss_main": 0.04461653158068657,
      "step": 35720
    },
    {
      "epoch": 4.180903346594898,
      "grad_norm": 0.27677279710769653,
      "learning_rate": 3.845103265307266e-05,
      "loss": 0.0554,
      "step": 35730
    },
    {
      "epoch": 4.180903346594898,
      "loss_main": 0.052205853164196014,
      "step": 35730
    },
    {
      "epoch": 4.182073484671191,
      "grad_norm": 0.34392374753952026,
      "learning_rate": 3.842422177911086e-05,
      "loss": 0.0541,
      "step": 35740
    },
    {
      "epoch": 4.182073484671191,
      "loss_main": 0.06543735414743423,
      "step": 35740
    },
    {
      "epoch": 4.183243622747484,
      "grad_norm": 0.21861445903778076,
      "learning_rate": 3.8397414421564826e-05,
      "loss": 0.0544,
      "step": 35750
    },
    {
      "epoch": 4.183243622747484,
      "loss_main": 0.062276873737573624,
      "step": 35750
    },
    {
      "epoch": 4.1844137608237775,
      "grad_norm": 0.27685248851776123,
      "learning_rate": 3.8370610588577935e-05,
      "loss": 0.0648,
      "step": 35760
    },
    {
      "epoch": 4.1844137608237775,
      "loss_main": 0.04993024840950966,
      "step": 35760
    },
    {
      "epoch": 4.18558389890007,
      "grad_norm": 0.2918975055217743,
      "learning_rate": 3.834381028829251e-05,
      "loss": 0.0576,
      "step": 35770
    },
    {
      "epoch": 4.18558389890007,
      "loss_main": 0.05372852087020874,
      "step": 35770
    },
    {
      "epoch": 4.186754036976363,
      "grad_norm": 0.2812356948852539,
      "learning_rate": 3.8317013528849745e-05,
      "loss": 0.0581,
      "step": 35780
    },
    {
      "epoch": 4.186754036976363,
      "loss_main": 0.06264306604862213,
      "step": 35780
    },
    {
      "epoch": 4.187924175052657,
      "grad_norm": 0.2813474237918854,
      "learning_rate": 3.8290220318389815e-05,
      "loss": 0.0709,
      "step": 35790
    },
    {
      "epoch": 4.187924175052657,
      "loss_main": 0.05468312278389931,
      "step": 35790
    },
    {
      "epoch": 4.189094313128949,
      "grad_norm": 0.2481093406677246,
      "learning_rate": 3.8263430665051746e-05,
      "loss": 0.0554,
      "step": 35800
    },
    {
      "epoch": 4.189094313128949,
      "loss_main": 0.03478096053004265,
      "step": 35800
    },
    {
      "epoch": 4.190264451205242,
      "grad_norm": 0.2747422158718109,
      "learning_rate": 3.8236644576973554e-05,
      "loss": 0.0541,
      "step": 35810
    },
    {
      "epoch": 4.190264451205242,
      "loss_main": 0.05367159843444824,
      "step": 35810
    },
    {
      "epoch": 4.191434589281535,
      "grad_norm": 0.2371147722005844,
      "learning_rate": 3.820986206229217e-05,
      "loss": 0.0555,
      "step": 35820
    },
    {
      "epoch": 4.191434589281535,
      "loss_main": 0.10077862441539764,
      "step": 35820
    },
    {
      "epoch": 4.192604727357828,
      "grad_norm": 0.2944127023220062,
      "learning_rate": 3.8183083129143384e-05,
      "loss": 0.0601,
      "step": 35830
    },
    {
      "epoch": 4.192604727357828,
      "loss_main": 0.046315599232912064,
      "step": 35830
    },
    {
      "epoch": 4.193774865434121,
      "grad_norm": 0.2750868499279022,
      "learning_rate": 3.815630778566193e-05,
      "loss": 0.0664,
      "step": 35840
    },
    {
      "epoch": 4.193774865434121,
      "loss_main": 0.07640540599822998,
      "step": 35840
    },
    {
      "epoch": 4.194945003510414,
      "grad_norm": 0.17075449228286743,
      "learning_rate": 3.812953603998145e-05,
      "loss": 0.0567,
      "step": 35850
    },
    {
      "epoch": 4.194945003510414,
      "loss_main": 0.05073290318250656,
      "step": 35850
    },
    {
      "epoch": 4.1961151415867075,
      "grad_norm": 0.28320565819740295,
      "learning_rate": 3.8102767900234504e-05,
      "loss": 0.0568,
      "step": 35860
    },
    {
      "epoch": 4.1961151415867075,
      "loss_main": 0.047326236963272095,
      "step": 35860
    },
    {
      "epoch": 4.197285279663,
      "grad_norm": 0.21084195375442505,
      "learning_rate": 3.807600337455256e-05,
      "loss": 0.0574,
      "step": 35870
    },
    {
      "epoch": 4.197285279663,
      "loss_main": 0.054724641144275665,
      "step": 35870
    },
    {
      "epoch": 4.198455417739293,
      "grad_norm": 0.3036424517631531,
      "learning_rate": 3.804924247106593e-05,
      "loss": 0.0624,
      "step": 35880
    },
    {
      "epoch": 4.198455417739293,
      "loss_main": 0.037404078990221024,
      "step": 35880
    },
    {
      "epoch": 4.199625555815587,
      "grad_norm": 0.21695470809936523,
      "learning_rate": 3.8022485197903925e-05,
      "loss": 0.0551,
      "step": 35890
    },
    {
      "epoch": 4.199625555815587,
      "loss_main": 0.037870507687330246,
      "step": 35890
    },
    {
      "epoch": 4.200795693891879,
      "grad_norm": 0.21026720106601715,
      "learning_rate": 3.799573156319464e-05,
      "loss": 0.0583,
      "step": 35900
    },
    {
      "epoch": 4.200795693891879,
      "loss_main": 0.06826941668987274,
      "step": 35900
    },
    {
      "epoch": 4.201965831968172,
      "grad_norm": 0.24177762866020203,
      "learning_rate": 3.796898157506515e-05,
      "loss": 0.061,
      "step": 35910
    },
    {
      "epoch": 4.201965831968172,
      "loss_main": 0.04793819412589073,
      "step": 35910
    },
    {
      "epoch": 4.203135970044465,
      "grad_norm": 0.279865026473999,
      "learning_rate": 3.794223524164143e-05,
      "loss": 0.0627,
      "step": 35920
    },
    {
      "epoch": 4.203135970044465,
      "loss_main": 0.03191452845931053,
      "step": 35920
    },
    {
      "epoch": 4.2043061081207584,
      "grad_norm": 0.3110368847846985,
      "learning_rate": 3.7915492571048245e-05,
      "loss": 0.057,
      "step": 35930
    },
    {
      "epoch": 4.2043061081207584,
      "loss_main": 0.04970870539546013,
      "step": 35930
    },
    {
      "epoch": 4.205476246197051,
      "grad_norm": 0.22144436836242676,
      "learning_rate": 3.788875357140937e-05,
      "loss": 0.0655,
      "step": 35940
    },
    {
      "epoch": 4.205476246197051,
      "loss_main": 0.0581519678235054,
      "step": 35940
    },
    {
      "epoch": 4.206646384273344,
      "grad_norm": 0.3590497672557831,
      "learning_rate": 3.786201825084736e-05,
      "loss": 0.057,
      "step": 35950
    },
    {
      "epoch": 4.206646384273344,
      "loss_main": 0.038983602076768875,
      "step": 35950
    },
    {
      "epoch": 4.2078165223496375,
      "grad_norm": 0.2944691479206085,
      "learning_rate": 3.783528661748372e-05,
      "loss": 0.0585,
      "step": 35960
    },
    {
      "epoch": 4.2078165223496375,
      "loss_main": 0.03713923692703247,
      "step": 35960
    },
    {
      "epoch": 4.20898666042593,
      "grad_norm": 0.20368872582912445,
      "learning_rate": 3.780855867943882e-05,
      "loss": 0.0577,
      "step": 35970
    },
    {
      "epoch": 4.20898666042593,
      "loss_main": 0.04030665010213852,
      "step": 35970
    },
    {
      "epoch": 4.210156798502223,
      "grad_norm": 0.31863856315612793,
      "learning_rate": 3.778183444483189e-05,
      "loss": 0.0565,
      "step": 35980
    },
    {
      "epoch": 4.210156798502223,
      "loss_main": 0.08565271645784378,
      "step": 35980
    },
    {
      "epoch": 4.211326936578517,
      "grad_norm": 0.3145073354244232,
      "learning_rate": 3.775511392178108e-05,
      "loss": 0.0565,
      "step": 35990
    },
    {
      "epoch": 4.211326936578517,
      "loss_main": 0.04225269332528114,
      "step": 35990
    },
    {
      "epoch": 4.212497074654809,
      "grad_norm": 0.33794140815734863,
      "learning_rate": 3.772839711840332e-05,
      "loss": 0.0588,
      "step": 36000
    },
    {
      "epoch": 4.212497074654809,
      "loss_main": 0.06245040148496628,
      "step": 36000
    },
    {
      "epoch": 4.213667212731102,
      "grad_norm": 0.268431693315506,
      "learning_rate": 3.7701684042814515e-05,
      "loss": 0.0572,
      "step": 36010
    },
    {
      "epoch": 4.213667212731102,
      "loss_main": 0.05891711264848709,
      "step": 36010
    },
    {
      "epoch": 4.214837350807395,
      "grad_norm": 0.27607855200767517,
      "learning_rate": 3.76749747031294e-05,
      "loss": 0.0502,
      "step": 36020
    },
    {
      "epoch": 4.214837350807395,
      "loss_main": 0.10369808226823807,
      "step": 36020
    },
    {
      "epoch": 4.2160074888836885,
      "grad_norm": 0.28727778792381287,
      "learning_rate": 3.764826910746152e-05,
      "loss": 0.0633,
      "step": 36030
    },
    {
      "epoch": 4.2160074888836885,
      "loss_main": 0.05492863431572914,
      "step": 36030
    },
    {
      "epoch": 4.217177626959981,
      "grad_norm": 0.2880392074584961,
      "learning_rate": 3.762156726392338e-05,
      "loss": 0.0512,
      "step": 36040
    },
    {
      "epoch": 4.217177626959981,
      "loss_main": 0.05325574055314064,
      "step": 36040
    },
    {
      "epoch": 4.218347765036274,
      "grad_norm": 0.3221033215522766,
      "learning_rate": 3.759486918062625e-05,
      "loss": 0.0622,
      "step": 36050
    },
    {
      "epoch": 4.218347765036274,
      "loss_main": 0.05683436617255211,
      "step": 36050
    },
    {
      "epoch": 4.219517903112568,
      "grad_norm": 0.2506601810455322,
      "learning_rate": 3.756817486568033e-05,
      "loss": 0.0607,
      "step": 36060
    },
    {
      "epoch": 4.219517903112568,
      "loss_main": 0.05924871191382408,
      "step": 36060
    },
    {
      "epoch": 4.22068804118886,
      "grad_norm": 0.24060769379138947,
      "learning_rate": 3.7541484327194654e-05,
      "loss": 0.0589,
      "step": 36070
    },
    {
      "epoch": 4.22068804118886,
      "loss_main": 0.051859114319086075,
      "step": 36070
    },
    {
      "epoch": 4.221858179265153,
      "grad_norm": 0.29135218262672424,
      "learning_rate": 3.751479757327707e-05,
      "loss": 0.0622,
      "step": 36080
    },
    {
      "epoch": 4.221858179265153,
      "loss_main": 0.027741411700844765,
      "step": 36080
    },
    {
      "epoch": 4.223028317341447,
      "grad_norm": 0.23148466646671295,
      "learning_rate": 3.7488114612034345e-05,
      "loss": 0.0549,
      "step": 36090
    },
    {
      "epoch": 4.223028317341447,
      "loss_main": 0.031524818390607834,
      "step": 36090
    },
    {
      "epoch": 4.224198455417739,
      "grad_norm": 0.1845775693655014,
      "learning_rate": 3.7461435451572044e-05,
      "loss": 0.0646,
      "step": 36100
    },
    {
      "epoch": 4.224198455417739,
      "loss_main": 0.06851767003536224,
      "step": 36100
    },
    {
      "epoch": 4.225368593494032,
      "grad_norm": 0.34100157022476196,
      "learning_rate": 3.743476009999459e-05,
      "loss": 0.0621,
      "step": 36110
    },
    {
      "epoch": 4.225368593494032,
      "loss_main": 0.08424823731184006,
      "step": 36110
    },
    {
      "epoch": 4.226538731570325,
      "grad_norm": 0.2061561644077301,
      "learning_rate": 3.7408088565405245e-05,
      "loss": 0.0564,
      "step": 36120
    },
    {
      "epoch": 4.226538731570325,
      "loss_main": 0.04014940932393074,
      "step": 36120
    },
    {
      "epoch": 4.2277088696466185,
      "grad_norm": 0.21505111455917358,
      "learning_rate": 3.738142085590612e-05,
      "loss": 0.0578,
      "step": 36130
    },
    {
      "epoch": 4.2277088696466185,
      "loss_main": 0.048402007669210434,
      "step": 36130
    },
    {
      "epoch": 4.228879007722911,
      "grad_norm": 0.3297295570373535,
      "learning_rate": 3.7354756979598194e-05,
      "loss": 0.0519,
      "step": 36140
    },
    {
      "epoch": 4.228879007722911,
      "loss_main": 0.034279294312000275,
      "step": 36140
    },
    {
      "epoch": 4.230049145799204,
      "grad_norm": 0.37893688678741455,
      "learning_rate": 3.7328096944581187e-05,
      "loss": 0.0542,
      "step": 36150
    },
    {
      "epoch": 4.230049145799204,
      "loss_main": 0.03475373610854149,
      "step": 36150
    },
    {
      "epoch": 4.231219283875498,
      "grad_norm": 0.34565192461013794,
      "learning_rate": 3.730144075895377e-05,
      "loss": 0.0613,
      "step": 36160
    },
    {
      "epoch": 4.231219283875498,
      "loss_main": 0.08493045717477798,
      "step": 36160
    },
    {
      "epoch": 4.23238942195179,
      "grad_norm": 0.267691969871521,
      "learning_rate": 3.727478843081335e-05,
      "loss": 0.0565,
      "step": 36170
    },
    {
      "epoch": 4.23238942195179,
      "loss_main": 0.04573564976453781,
      "step": 36170
    },
    {
      "epoch": 4.233559560028083,
      "grad_norm": 0.24011746048927307,
      "learning_rate": 3.72481399682562e-05,
      "loss": 0.0529,
      "step": 36180
    },
    {
      "epoch": 4.233559560028083,
      "loss_main": 0.13472101092338562,
      "step": 36180
    },
    {
      "epoch": 4.234729698104377,
      "grad_norm": 0.24249598383903503,
      "learning_rate": 3.722149537937747e-05,
      "loss": 0.0571,
      "step": 36190
    },
    {
      "epoch": 4.234729698104377,
      "loss_main": 0.032734453678131104,
      "step": 36190
    },
    {
      "epoch": 4.2358998361806695,
      "grad_norm": 0.27030256390571594,
      "learning_rate": 3.7194854672271015e-05,
      "loss": 0.0523,
      "step": 36200
    },
    {
      "epoch": 4.2358998361806695,
      "loss_main": 0.0697612389922142,
      "step": 36200
    },
    {
      "epoch": 4.237069974256962,
      "grad_norm": 0.23913255333900452,
      "learning_rate": 3.7168217855029644e-05,
      "loss": 0.0596,
      "step": 36210
    },
    {
      "epoch": 4.237069974256962,
      "loss_main": 0.08241791278123856,
      "step": 36210
    },
    {
      "epoch": 4.238240112333255,
      "grad_norm": 0.2851456105709076,
      "learning_rate": 3.7141584935744856e-05,
      "loss": 0.0582,
      "step": 36220
    },
    {
      "epoch": 4.238240112333255,
      "loss_main": 0.0886189416050911,
      "step": 36220
    },
    {
      "epoch": 4.2394102504095486,
      "grad_norm": 0.26386210322380066,
      "learning_rate": 3.7114955922507055e-05,
      "loss": 0.0518,
      "step": 36230
    },
    {
      "epoch": 4.2394102504095486,
      "loss_main": 0.026372414082288742,
      "step": 36230
    },
    {
      "epoch": 4.240580388485841,
      "grad_norm": 0.2914215922355652,
      "learning_rate": 3.708833082340545e-05,
      "loss": 0.061,
      "step": 36240
    },
    {
      "epoch": 4.240580388485841,
      "loss_main": 0.03205587714910507,
      "step": 36240
    },
    {
      "epoch": 4.241750526562134,
      "grad_norm": 0.2244061976671219,
      "learning_rate": 3.7061709646528034e-05,
      "loss": 0.048,
      "step": 36250
    },
    {
      "epoch": 4.241750526562134,
      "loss_main": 0.10143137723207474,
      "step": 36250
    },
    {
      "epoch": 4.242920664638428,
      "grad_norm": 0.26004138588905334,
      "learning_rate": 3.7035092399961604e-05,
      "loss": 0.054,
      "step": 36260
    },
    {
      "epoch": 4.242920664638428,
      "loss_main": 0.037276022136211395,
      "step": 36260
    },
    {
      "epoch": 4.24409080271472,
      "grad_norm": 0.30658161640167236,
      "learning_rate": 3.700847909179177e-05,
      "loss": 0.0553,
      "step": 36270
    },
    {
      "epoch": 4.24409080271472,
      "loss_main": 0.10818009078502655,
      "step": 36270
    },
    {
      "epoch": 4.245260940791013,
      "grad_norm": 0.28170502185821533,
      "learning_rate": 3.698186973010297e-05,
      "loss": 0.0673,
      "step": 36280
    },
    {
      "epoch": 4.245260940791013,
      "loss_main": 0.08158445358276367,
      "step": 36280
    },
    {
      "epoch": 4.246431078867307,
      "grad_norm": 0.30901312828063965,
      "learning_rate": 3.695526432297844e-05,
      "loss": 0.0594,
      "step": 36290
    },
    {
      "epoch": 4.246431078867307,
      "loss_main": 0.05653373524546623,
      "step": 36290
    },
    {
      "epoch": 4.2476012169435995,
      "grad_norm": 0.26649489998817444,
      "learning_rate": 3.692866287850017e-05,
      "loss": 0.0453,
      "step": 36300
    },
    {
      "epoch": 4.2476012169435995,
      "loss_main": 0.06117416173219681,
      "step": 36300
    },
    {
      "epoch": 4.248771355019892,
      "grad_norm": 0.23023100197315216,
      "learning_rate": 3.6902065404749006e-05,
      "loss": 0.0546,
      "step": 36310
    },
    {
      "epoch": 4.248771355019892,
      "loss_main": 0.06942100077867508,
      "step": 36310
    },
    {
      "epoch": 4.249941493096185,
      "grad_norm": 0.30472564697265625,
      "learning_rate": 3.6875471909804516e-05,
      "loss": 0.0643,
      "step": 36320
    },
    {
      "epoch": 4.249941493096185,
      "loss_main": 0.039696693420410156,
      "step": 36320
    },
    {
      "epoch": 4.251111631172479,
      "grad_norm": 0.24333913624286652,
      "learning_rate": 3.6848882401745135e-05,
      "loss": 0.0445,
      "step": 36330
    },
    {
      "epoch": 4.251111631172479,
      "loss_main": 0.07575053721666336,
      "step": 36330
    },
    {
      "epoch": 4.252281769248771,
      "grad_norm": 0.26161885261535645,
      "learning_rate": 3.682229688864806e-05,
      "loss": 0.06,
      "step": 36340
    },
    {
      "epoch": 4.252281769248771,
      "loss_main": 0.031069615855813026,
      "step": 36340
    },
    {
      "epoch": 4.253451907325064,
      "grad_norm": 0.21679876744747162,
      "learning_rate": 3.6795715378589235e-05,
      "loss": 0.0557,
      "step": 36350
    },
    {
      "epoch": 4.253451907325064,
      "loss_main": 0.05227053165435791,
      "step": 36350
    },
    {
      "epoch": 4.254622045401358,
      "grad_norm": 0.22053611278533936,
      "learning_rate": 3.676913787964345e-05,
      "loss": 0.0589,
      "step": 36360
    },
    {
      "epoch": 4.254622045401358,
      "loss_main": 0.04315035790205002,
      "step": 36360
    },
    {
      "epoch": 4.25579218347765,
      "grad_norm": 0.26606932282447815,
      "learning_rate": 3.674256439988423e-05,
      "loss": 0.0586,
      "step": 36370
    },
    {
      "epoch": 4.25579218347765,
      "loss_main": 0.06401879340410233,
      "step": 36370
    },
    {
      "epoch": 4.256962321553943,
      "grad_norm": 0.2552952766418457,
      "learning_rate": 3.6715994947383904e-05,
      "loss": 0.0633,
      "step": 36380
    },
    {
      "epoch": 4.256962321553943,
      "loss_main": 0.03210395947098732,
      "step": 36380
    },
    {
      "epoch": 4.258132459630236,
      "grad_norm": 0.1874140053987503,
      "learning_rate": 3.668942953021357e-05,
      "loss": 0.0474,
      "step": 36390
    },
    {
      "epoch": 4.258132459630236,
      "loss_main": 0.08695660531520844,
      "step": 36390
    },
    {
      "epoch": 4.2593025977065295,
      "grad_norm": 0.2506163418292999,
      "learning_rate": 3.66628681564431e-05,
      "loss": 0.0566,
      "step": 36400
    },
    {
      "epoch": 4.2593025977065295,
      "loss_main": 0.028140366077423096,
      "step": 36400
    },
    {
      "epoch": 4.260472735782822,
      "grad_norm": 0.2813955247402191,
      "learning_rate": 3.663631083414114e-05,
      "loss": 0.0516,
      "step": 36410
    },
    {
      "epoch": 4.260472735782822,
      "loss_main": 0.05784507840871811,
      "step": 36410
    },
    {
      "epoch": 4.261642873859115,
      "grad_norm": 0.24034306406974792,
      "learning_rate": 3.660975757137509e-05,
      "loss": 0.0621,
      "step": 36420
    },
    {
      "epoch": 4.261642873859115,
      "loss_main": 0.058660298585891724,
      "step": 36420
    },
    {
      "epoch": 4.262813011935409,
      "grad_norm": 0.3272116184234619,
      "learning_rate": 3.658320837621114e-05,
      "loss": 0.0564,
      "step": 36430
    },
    {
      "epoch": 4.262813011935409,
      "loss_main": 0.04434264451265335,
      "step": 36430
    },
    {
      "epoch": 4.263983150011701,
      "grad_norm": 0.2795444130897522,
      "learning_rate": 3.655666325671426e-05,
      "loss": 0.0583,
      "step": 36440
    },
    {
      "epoch": 4.263983150011701,
      "loss_main": 0.04060891643166542,
      "step": 36440
    },
    {
      "epoch": 4.265153288087994,
      "grad_norm": 0.30296236276626587,
      "learning_rate": 3.65301222209481e-05,
      "loss": 0.0527,
      "step": 36450
    },
    {
      "epoch": 4.265153288087994,
      "loss_main": 0.05893469974398613,
      "step": 36450
    },
    {
      "epoch": 4.266323426164288,
      "grad_norm": 0.3277977406978607,
      "learning_rate": 3.650358527697519e-05,
      "loss": 0.0629,
      "step": 36460
    },
    {
      "epoch": 4.266323426164288,
      "loss_main": 0.04240592196583748,
      "step": 36460
    },
    {
      "epoch": 4.2674935642405805,
      "grad_norm": 0.2876778841018677,
      "learning_rate": 3.64770524328567e-05,
      "loss": 0.0603,
      "step": 36470
    },
    {
      "epoch": 4.2674935642405805,
      "loss_main": 0.05860671028494835,
      "step": 36470
    },
    {
      "epoch": 4.268663702316873,
      "grad_norm": 0.21077726781368256,
      "learning_rate": 3.645052369665265e-05,
      "loss": 0.0517,
      "step": 36480
    },
    {
      "epoch": 4.268663702316873,
      "loss_main": 0.07589453458786011,
      "step": 36480
    },
    {
      "epoch": 4.269833840393167,
      "grad_norm": 0.22271645069122314,
      "learning_rate": 3.6423999076421724e-05,
      "loss": 0.0556,
      "step": 36490
    },
    {
      "epoch": 4.269833840393167,
      "loss_main": 0.045801375061273575,
      "step": 36490
    },
    {
      "epoch": 4.27100397846946,
      "grad_norm": 0.32458117604255676,
      "learning_rate": 3.639747858022142e-05,
      "loss": 0.0641,
      "step": 36500
    },
    {
      "epoch": 4.27100397846946,
      "loss_main": 0.05220562964677811,
      "step": 36500
    },
    {
      "epoch": 4.272174116545752,
      "grad_norm": 0.28201377391815186,
      "learning_rate": 3.637096221610799e-05,
      "loss": 0.0592,
      "step": 36510
    },
    {
      "epoch": 4.272174116545752,
      "loss_main": 0.040434811264276505,
      "step": 36510
    },
    {
      "epoch": 4.273344254622045,
      "grad_norm": 0.23549304902553558,
      "learning_rate": 3.634444999213638e-05,
      "loss": 0.049,
      "step": 36520
    },
    {
      "epoch": 4.273344254622045,
      "loss_main": 0.058286964893341064,
      "step": 36520
    },
    {
      "epoch": 4.274514392698339,
      "grad_norm": 0.33496010303497314,
      "learning_rate": 3.6317941916360296e-05,
      "loss": 0.0563,
      "step": 36530
    },
    {
      "epoch": 4.274514392698339,
      "loss_main": 0.03805284947156906,
      "step": 36530
    },
    {
      "epoch": 4.275684530774631,
      "grad_norm": 0.21475106477737427,
      "learning_rate": 3.629143799683221e-05,
      "loss": 0.0532,
      "step": 36540
    },
    {
      "epoch": 4.275684530774631,
      "loss_main": 0.0664902850985527,
      "step": 36540
    },
    {
      "epoch": 4.276854668850924,
      "grad_norm": 0.1709621399641037,
      "learning_rate": 3.626493824160331e-05,
      "loss": 0.0555,
      "step": 36550
    },
    {
      "epoch": 4.276854668850924,
      "loss_main": 0.03631317615509033,
      "step": 36550
    },
    {
      "epoch": 4.278024806927218,
      "grad_norm": 0.2493562251329422,
      "learning_rate": 3.623844265872352e-05,
      "loss": 0.047,
      "step": 36560
    },
    {
      "epoch": 4.278024806927218,
      "loss_main": 0.04090522974729538,
      "step": 36560
    },
    {
      "epoch": 4.2791949450035105,
      "grad_norm": 0.23470863699913025,
      "learning_rate": 3.621195125624149e-05,
      "loss": 0.062,
      "step": 36570
    },
    {
      "epoch": 4.2791949450035105,
      "loss_main": 0.04270723834633827,
      "step": 36570
    },
    {
      "epoch": 4.280365083079803,
      "grad_norm": 0.2323354035615921,
      "learning_rate": 3.618546404220463e-05,
      "loss": 0.0526,
      "step": 36580
    },
    {
      "epoch": 4.280365083079803,
      "loss_main": 0.06553468108177185,
      "step": 36580
    },
    {
      "epoch": 4.281535221156096,
      "grad_norm": 0.30664536356925964,
      "learning_rate": 3.615898102465903e-05,
      "loss": 0.0576,
      "step": 36590
    },
    {
      "epoch": 4.281535221156096,
      "loss_main": 0.058259330689907074,
      "step": 36590
    },
    {
      "epoch": 4.28270535923239,
      "grad_norm": 0.2878600060939789,
      "learning_rate": 3.6132502211649544e-05,
      "loss": 0.0559,
      "step": 36600
    },
    {
      "epoch": 4.28270535923239,
      "loss_main": 0.040258146822452545,
      "step": 36600
    },
    {
      "epoch": 4.283875497308682,
      "grad_norm": 0.45860350131988525,
      "learning_rate": 3.610602761121975e-05,
      "loss": 0.0576,
      "step": 36610
    },
    {
      "epoch": 4.283875497308682,
      "loss_main": 0.05094681307673454,
      "step": 36610
    },
    {
      "epoch": 4.285045635384975,
      "grad_norm": 0.18224720656871796,
      "learning_rate": 3.6079557231411897e-05,
      "loss": 0.0498,
      "step": 36620
    },
    {
      "epoch": 4.285045635384975,
      "loss_main": 0.05297933891415596,
      "step": 36620
    },
    {
      "epoch": 4.286215773461269,
      "grad_norm": 0.31447646021842957,
      "learning_rate": 3.6053091080267035e-05,
      "loss": 0.0528,
      "step": 36630
    },
    {
      "epoch": 4.286215773461269,
      "loss_main": 0.052475716918706894,
      "step": 36630
    },
    {
      "epoch": 4.287385911537561,
      "grad_norm": 0.2579690217971802,
      "learning_rate": 3.602662916582483e-05,
      "loss": 0.0524,
      "step": 36640
    },
    {
      "epoch": 4.287385911537561,
      "loss_main": 0.07089828699827194,
      "step": 36640
    },
    {
      "epoch": 4.288556049613854,
      "grad_norm": 0.3017917573451996,
      "learning_rate": 3.600017149612375e-05,
      "loss": 0.0594,
      "step": 36650
    },
    {
      "epoch": 4.288556049613854,
      "loss_main": 0.03519027307629585,
      "step": 36650
    },
    {
      "epoch": 4.289726187690148,
      "grad_norm": 0.2774266004562378,
      "learning_rate": 3.5973718079200935e-05,
      "loss": 0.0573,
      "step": 36660
    },
    {
      "epoch": 4.289726187690148,
      "loss_main": 0.03242061287164688,
      "step": 36660
    },
    {
      "epoch": 4.2908963257664405,
      "grad_norm": 0.26286911964416504,
      "learning_rate": 3.5947268923092216e-05,
      "loss": 0.0557,
      "step": 36670
    },
    {
      "epoch": 4.2908963257664405,
      "loss_main": 0.05762259289622307,
      "step": 36670
    },
    {
      "epoch": 4.292066463842733,
      "grad_norm": 0.2725312411785126,
      "learning_rate": 3.592082403583216e-05,
      "loss": 0.0548,
      "step": 36680
    },
    {
      "epoch": 4.292066463842733,
      "loss_main": 0.047649163752794266,
      "step": 36680
    },
    {
      "epoch": 4.293236601919027,
      "grad_norm": 0.16416344046592712,
      "learning_rate": 3.5894383425454004e-05,
      "loss": 0.0522,
      "step": 36690
    },
    {
      "epoch": 4.293236601919027,
      "loss_main": 0.049061618745326996,
      "step": 36690
    },
    {
      "epoch": 4.29440673999532,
      "grad_norm": 0.29615363478660583,
      "learning_rate": 3.586794709998975e-05,
      "loss": 0.0454,
      "step": 36700
    },
    {
      "epoch": 4.29440673999532,
      "loss_main": 0.05912690982222557,
      "step": 36700
    },
    {
      "epoch": 4.295576878071612,
      "grad_norm": 0.2955731749534607,
      "learning_rate": 3.584151506747002e-05,
      "loss": 0.0524,
      "step": 36710
    },
    {
      "epoch": 4.295576878071612,
      "loss_main": 0.043523114174604416,
      "step": 36710
    },
    {
      "epoch": 4.296747016147905,
      "grad_norm": 0.2626206874847412,
      "learning_rate": 3.581508733592418e-05,
      "loss": 0.0567,
      "step": 36720
    },
    {
      "epoch": 4.296747016147905,
      "loss_main": 0.056022077798843384,
      "step": 36720
    },
    {
      "epoch": 4.297917154224199,
      "grad_norm": 0.25890085101127625,
      "learning_rate": 3.5788663913380297e-05,
      "loss": 0.0575,
      "step": 36730
    },
    {
      "epoch": 4.297917154224199,
      "loss_main": 0.0775274857878685,
      "step": 36730
    },
    {
      "epoch": 4.2990872923004915,
      "grad_norm": 0.22612538933753967,
      "learning_rate": 3.576224480786506e-05,
      "loss": 0.0622,
      "step": 36740
    },
    {
      "epoch": 4.2990872923004915,
      "loss_main": 0.04100993648171425,
      "step": 36740
    },
    {
      "epoch": 4.300257430376784,
      "grad_norm": 0.23401086032390594,
      "learning_rate": 3.573583002740393e-05,
      "loss": 0.0554,
      "step": 36750
    },
    {
      "epoch": 4.300257430376784,
      "loss_main": 0.06157129257917404,
      "step": 36750
    },
    {
      "epoch": 4.301427568453078,
      "grad_norm": 0.2942637503147125,
      "learning_rate": 3.570941958002103e-05,
      "loss": 0.0577,
      "step": 36760
    },
    {
      "epoch": 4.301427568453078,
      "loss_main": 0.030980626121163368,
      "step": 36760
    },
    {
      "epoch": 4.302597706529371,
      "grad_norm": 0.20887382328510284,
      "learning_rate": 3.568301347373912e-05,
      "loss": 0.0508,
      "step": 36770
    },
    {
      "epoch": 4.302597706529371,
      "loss_main": 0.07414158433675766,
      "step": 36770
    },
    {
      "epoch": 4.303767844605663,
      "grad_norm": 0.2464483082294464,
      "learning_rate": 3.5656611716579726e-05,
      "loss": 0.053,
      "step": 36780
    },
    {
      "epoch": 4.303767844605663,
      "loss_main": 0.04212329909205437,
      "step": 36780
    },
    {
      "epoch": 4.304937982681956,
      "grad_norm": 0.27927011251449585,
      "learning_rate": 3.5630214316562946e-05,
      "loss": 0.0665,
      "step": 36790
    },
    {
      "epoch": 4.304937982681956,
      "loss_main": 0.0821685940027237,
      "step": 36790
    },
    {
      "epoch": 4.30610812075825,
      "grad_norm": 0.16604837775230408,
      "learning_rate": 3.560382128170766e-05,
      "loss": 0.0506,
      "step": 36800
    },
    {
      "epoch": 4.30610812075825,
      "loss_main": 0.13093511760234833,
      "step": 36800
    },
    {
      "epoch": 4.307278258834542,
      "grad_norm": 0.18926021456718445,
      "learning_rate": 3.5577432620031374e-05,
      "loss": 0.0614,
      "step": 36810
    },
    {
      "epoch": 4.307278258834542,
      "loss_main": 0.04301571473479271,
      "step": 36810
    },
    {
      "epoch": 4.308448396910835,
      "grad_norm": 0.24135969579219818,
      "learning_rate": 3.5551048339550216e-05,
      "loss": 0.0592,
      "step": 36820
    },
    {
      "epoch": 4.308448396910835,
      "loss_main": 0.0603826642036438,
      "step": 36820
    },
    {
      "epoch": 4.309618534987129,
      "grad_norm": 0.20561327040195465,
      "learning_rate": 3.55246684482791e-05,
      "loss": 0.058,
      "step": 36830
    },
    {
      "epoch": 4.309618534987129,
      "loss_main": 0.05733225494623184,
      "step": 36830
    },
    {
      "epoch": 4.3107886730634215,
      "grad_norm": 0.29679277539253235,
      "learning_rate": 3.5498292954231496e-05,
      "loss": 0.0515,
      "step": 36840
    },
    {
      "epoch": 4.3107886730634215,
      "loss_main": 0.046876467764377594,
      "step": 36840
    },
    {
      "epoch": 4.311958811139714,
      "grad_norm": 0.31848660111427307,
      "learning_rate": 3.54719218654196e-05,
      "loss": 0.0549,
      "step": 36850
    },
    {
      "epoch": 4.311958811139714,
      "loss_main": 0.09367843717336655,
      "step": 36850
    },
    {
      "epoch": 4.313128949216008,
      "grad_norm": 0.2979915142059326,
      "learning_rate": 3.544555518985425e-05,
      "loss": 0.0543,
      "step": 36860
    },
    {
      "epoch": 4.313128949216008,
      "loss_main": 0.042235810309648514,
      "step": 36860
    },
    {
      "epoch": 4.314299087292301,
      "grad_norm": 0.25086915493011475,
      "learning_rate": 3.541919293554494e-05,
      "loss": 0.0529,
      "step": 36870
    },
    {
      "epoch": 4.314299087292301,
      "loss_main": 0.08848144859075546,
      "step": 36870
    },
    {
      "epoch": 4.315469225368593,
      "grad_norm": 0.27652791142463684,
      "learning_rate": 3.539283511049985e-05,
      "loss": 0.0631,
      "step": 36880
    },
    {
      "epoch": 4.315469225368593,
      "loss_main": 0.10324057191610336,
      "step": 36880
    },
    {
      "epoch": 4.316639363444887,
      "grad_norm": 0.24427561461925507,
      "learning_rate": 3.5366481722725755e-05,
      "loss": 0.058,
      "step": 36890
    },
    {
      "epoch": 4.316639363444887,
      "loss_main": 0.06600441783666611,
      "step": 36890
    },
    {
      "epoch": 4.31780950152118,
      "grad_norm": 0.235904261469841,
      "learning_rate": 3.534013278022816e-05,
      "loss": 0.0551,
      "step": 36900
    },
    {
      "epoch": 4.31780950152118,
      "loss_main": 0.09385322034358978,
      "step": 36900
    },
    {
      "epoch": 4.318979639597472,
      "grad_norm": 0.26814934611320496,
      "learning_rate": 3.531378829101113e-05,
      "loss": 0.0576,
      "step": 36910
    },
    {
      "epoch": 4.318979639597472,
      "loss_main": 0.045803725719451904,
      "step": 36910
    },
    {
      "epoch": 4.320149777673765,
      "grad_norm": 0.3576405942440033,
      "learning_rate": 3.528744826307746e-05,
      "loss": 0.0566,
      "step": 36920
    },
    {
      "epoch": 4.320149777673765,
      "loss_main": 0.05520788952708244,
      "step": 36920
    },
    {
      "epoch": 4.321319915750059,
      "grad_norm": 0.2317747175693512,
      "learning_rate": 3.5261112704428554e-05,
      "loss": 0.0544,
      "step": 36930
    },
    {
      "epoch": 4.321319915750059,
      "loss_main": 0.04441773146390915,
      "step": 36930
    },
    {
      "epoch": 4.3224900538263515,
      "grad_norm": 0.3666249215602875,
      "learning_rate": 3.523478162306443e-05,
      "loss": 0.051,
      "step": 36940
    },
    {
      "epoch": 4.3224900538263515,
      "loss_main": 0.04377176985144615,
      "step": 36940
    },
    {
      "epoch": 4.323660191902644,
      "grad_norm": 0.2041296362876892,
      "learning_rate": 3.520845502698381e-05,
      "loss": 0.0484,
      "step": 36950
    },
    {
      "epoch": 4.323660191902644,
      "loss_main": 0.04089759662747383,
      "step": 36950
    },
    {
      "epoch": 4.324830329978938,
      "grad_norm": 0.2752891778945923,
      "learning_rate": 3.5182132924184005e-05,
      "loss": 0.0605,
      "step": 36960
    },
    {
      "epoch": 4.324830329978938,
      "loss_main": 0.062458258122205734,
      "step": 36960
    },
    {
      "epoch": 4.326000468055231,
      "grad_norm": 0.27704140543937683,
      "learning_rate": 3.5155815322660966e-05,
      "loss": 0.0585,
      "step": 36970
    },
    {
      "epoch": 4.326000468055231,
      "loss_main": 0.06027781218290329,
      "step": 36970
    },
    {
      "epoch": 4.327170606131523,
      "grad_norm": 0.23461094498634338,
      "learning_rate": 3.512950223040931e-05,
      "loss": 0.0526,
      "step": 36980
    },
    {
      "epoch": 4.327170606131523,
      "loss_main": 0.07768718153238297,
      "step": 36980
    },
    {
      "epoch": 4.328340744207816,
      "grad_norm": 0.2482561618089676,
      "learning_rate": 3.5103193655422216e-05,
      "loss": 0.0611,
      "step": 36990
    },
    {
      "epoch": 4.328340744207816,
      "loss_main": 0.06528949737548828,
      "step": 36990
    },
    {
      "epoch": 4.32951088228411,
      "grad_norm": 0.4585546851158142,
      "learning_rate": 3.5076889605691596e-05,
      "loss": 0.0565,
      "step": 37000
    },
    {
      "epoch": 4.32951088228411,
      "loss_main": 0.05811292678117752,
      "step": 37000
    },
    {
      "epoch": 4.3306810203604025,
      "grad_norm": 0.22368428111076355,
      "learning_rate": 3.505059008920787e-05,
      "loss": 0.0595,
      "step": 37010
    },
    {
      "epoch": 4.3306810203604025,
      "loss_main": 0.05957436561584473,
      "step": 37010
    },
    {
      "epoch": 4.331851158436695,
      "grad_norm": 0.3200741112232208,
      "learning_rate": 3.502429511396016e-05,
      "loss": 0.0611,
      "step": 37020
    },
    {
      "epoch": 4.331851158436695,
      "loss_main": 0.03651101514697075,
      "step": 37020
    },
    {
      "epoch": 4.333021296512989,
      "grad_norm": 0.2546306252479553,
      "learning_rate": 3.4998004687936196e-05,
      "loss": 0.0615,
      "step": 37030
    },
    {
      "epoch": 4.333021296512989,
      "loss_main": 0.04732038825750351,
      "step": 37030
    },
    {
      "epoch": 4.334191434589282,
      "grad_norm": 0.27725955843925476,
      "learning_rate": 3.497171881912229e-05,
      "loss": 0.0539,
      "step": 37040
    },
    {
      "epoch": 4.334191434589282,
      "loss_main": 0.05673816055059433,
      "step": 37040
    },
    {
      "epoch": 4.335361572665574,
      "grad_norm": 0.2066662460565567,
      "learning_rate": 3.494543751550342e-05,
      "loss": 0.0533,
      "step": 37050
    },
    {
      "epoch": 4.335361572665574,
      "loss_main": 0.07951827347278595,
      "step": 37050
    },
    {
      "epoch": 4.336531710741868,
      "grad_norm": 0.3605417013168335,
      "learning_rate": 3.491916078506313e-05,
      "loss": 0.0599,
      "step": 37060
    },
    {
      "epoch": 4.336531710741868,
      "loss_main": 0.042176563292741776,
      "step": 37060
    },
    {
      "epoch": 4.337701848818161,
      "grad_norm": 0.2281881719827652,
      "learning_rate": 3.489288863578361e-05,
      "loss": 0.0616,
      "step": 37070
    },
    {
      "epoch": 4.337701848818161,
      "loss_main": 0.07653196901082993,
      "step": 37070
    },
    {
      "epoch": 4.338871986894453,
      "grad_norm": 0.20981431007385254,
      "learning_rate": 3.4866621075645646e-05,
      "loss": 0.0588,
      "step": 37080
    },
    {
      "epoch": 4.338871986894453,
      "loss_main": 0.04387606307864189,
      "step": 37080
    },
    {
      "epoch": 4.340042124970746,
      "grad_norm": 0.23014529049396515,
      "learning_rate": 3.4840358112628614e-05,
      "loss": 0.0558,
      "step": 37090
    },
    {
      "epoch": 4.340042124970746,
      "loss_main": 0.07653170078992844,
      "step": 37090
    },
    {
      "epoch": 4.34121226304704,
      "grad_norm": 0.20567896962165833,
      "learning_rate": 3.481409975471053e-05,
      "loss": 0.0479,
      "step": 37100
    },
    {
      "epoch": 4.34121226304704,
      "loss_main": 0.02639034204185009,
      "step": 37100
    },
    {
      "epoch": 4.3423824011233325,
      "grad_norm": 0.2905319631099701,
      "learning_rate": 3.4787846009867986e-05,
      "loss": 0.0574,
      "step": 37110
    },
    {
      "epoch": 4.3423824011233325,
      "loss_main": 0.06911132484674454,
      "step": 37110
    },
    {
      "epoch": 4.343552539199625,
      "grad_norm": 0.23546883463859558,
      "learning_rate": 3.476159688607615e-05,
      "loss": 0.0595,
      "step": 37120
    },
    {
      "epoch": 4.343552539199625,
      "loss_main": 0.03391169384121895,
      "step": 37120
    },
    {
      "epoch": 4.344722677275919,
      "grad_norm": 0.20814132690429688,
      "learning_rate": 3.4735352391308854e-05,
      "loss": 0.0624,
      "step": 37130
    },
    {
      "epoch": 4.344722677275919,
      "loss_main": 0.055202942341566086,
      "step": 37130
    },
    {
      "epoch": 4.345892815352212,
      "grad_norm": 0.24653784930706024,
      "learning_rate": 3.4709112533538446e-05,
      "loss": 0.057,
      "step": 37140
    },
    {
      "epoch": 4.345892815352212,
      "loss_main": 0.03252756968140602,
      "step": 37140
    },
    {
      "epoch": 4.347062953428504,
      "grad_norm": 0.2798886299133301,
      "learning_rate": 3.4682877320735934e-05,
      "loss": 0.0457,
      "step": 37150
    },
    {
      "epoch": 4.347062953428504,
      "loss_main": 0.05056043341755867,
      "step": 37150
    },
    {
      "epoch": 4.348233091504798,
      "grad_norm": 0.2062491923570633,
      "learning_rate": 3.465664676087085e-05,
      "loss": 0.0637,
      "step": 37160
    },
    {
      "epoch": 4.348233091504798,
      "loss_main": 0.06015565246343613,
      "step": 37160
    },
    {
      "epoch": 4.349403229581091,
      "grad_norm": 0.24630524218082428,
      "learning_rate": 3.463042086191136e-05,
      "loss": 0.0554,
      "step": 37170
    },
    {
      "epoch": 4.349403229581091,
      "loss_main": 0.05281141772866249,
      "step": 37170
    },
    {
      "epoch": 4.3505733676573835,
      "grad_norm": 0.23421385884284973,
      "learning_rate": 3.460419963182423e-05,
      "loss": 0.0608,
      "step": 37180
    },
    {
      "epoch": 4.3505733676573835,
      "loss_main": 0.05278636887669563,
      "step": 37180
    },
    {
      "epoch": 4.351743505733676,
      "grad_norm": 0.2761385440826416,
      "learning_rate": 3.457798307857473e-05,
      "loss": 0.0556,
      "step": 37190
    },
    {
      "epoch": 4.351743505733676,
      "loss_main": 0.06803803890943527,
      "step": 37190
    },
    {
      "epoch": 4.35291364380997,
      "grad_norm": 0.254238098859787,
      "learning_rate": 3.455177121012678e-05,
      "loss": 0.0542,
      "step": 37200
    },
    {
      "epoch": 4.35291364380997,
      "loss_main": 0.03689824789762497,
      "step": 37200
    },
    {
      "epoch": 4.3540837818862625,
      "grad_norm": 0.24552980065345764,
      "learning_rate": 3.452556403444285e-05,
      "loss": 0.0566,
      "step": 37210
    },
    {
      "epoch": 4.3540837818862625,
      "loss_main": 0.04938005656003952,
      "step": 37210
    },
    {
      "epoch": 4.355253919962555,
      "grad_norm": 0.3650680184364319,
      "learning_rate": 3.4499361559483975e-05,
      "loss": 0.0534,
      "step": 37220
    },
    {
      "epoch": 4.355253919962555,
      "loss_main": 0.05082855001091957,
      "step": 37220
    },
    {
      "epoch": 4.356424058038849,
      "grad_norm": 0.2629232704639435,
      "learning_rate": 3.44731637932098e-05,
      "loss": 0.0551,
      "step": 37230
    },
    {
      "epoch": 4.356424058038849,
      "loss_main": 0.04514608159661293,
      "step": 37230
    },
    {
      "epoch": 4.357594196115142,
      "grad_norm": 0.2681055963039398,
      "learning_rate": 3.44469707435785e-05,
      "loss": 0.0454,
      "step": 37240
    },
    {
      "epoch": 4.357594196115142,
      "loss_main": 0.05499350279569626,
      "step": 37240
    },
    {
      "epoch": 4.358764334191434,
      "grad_norm": 0.24828818440437317,
      "learning_rate": 3.4420782418546835e-05,
      "loss": 0.0524,
      "step": 37250
    },
    {
      "epoch": 4.358764334191434,
      "loss_main": 0.058114681392908096,
      "step": 37250
    },
    {
      "epoch": 4.359934472267728,
      "grad_norm": 0.2787177264690399,
      "learning_rate": 3.439459882607012e-05,
      "loss": 0.0533,
      "step": 37260
    },
    {
      "epoch": 4.359934472267728,
      "loss_main": 0.04002201929688454,
      "step": 37260
    },
    {
      "epoch": 4.361104610344021,
      "grad_norm": 0.26974886655807495,
      "learning_rate": 3.436841997410225e-05,
      "loss": 0.0499,
      "step": 37270
    },
    {
      "epoch": 4.361104610344021,
      "loss_main": 0.039368417114019394,
      "step": 37270
    },
    {
      "epoch": 4.3622747484203135,
      "grad_norm": 0.2648048996925354,
      "learning_rate": 3.434224587059567e-05,
      "loss": 0.0535,
      "step": 37280
    },
    {
      "epoch": 4.3622747484203135,
      "loss_main": 0.05329892411828041,
      "step": 37280
    },
    {
      "epoch": 4.363444886496606,
      "grad_norm": 0.33288076519966125,
      "learning_rate": 3.431607652350136e-05,
      "loss": 0.0548,
      "step": 37290
    },
    {
      "epoch": 4.363444886496606,
      "loss_main": 0.06187707558274269,
      "step": 37290
    },
    {
      "epoch": 4.3646150245729,
      "grad_norm": 0.280134916305542,
      "learning_rate": 3.428991194076891e-05,
      "loss": 0.0549,
      "step": 37300
    },
    {
      "epoch": 4.3646150245729,
      "loss_main": 0.04833400249481201,
      "step": 37300
    },
    {
      "epoch": 4.365785162649193,
      "grad_norm": 0.2671991288661957,
      "learning_rate": 3.4263752130346394e-05,
      "loss": 0.0551,
      "step": 37310
    },
    {
      "epoch": 4.365785162649193,
      "loss_main": 0.04613333195447922,
      "step": 37310
    },
    {
      "epoch": 4.366955300725485,
      "grad_norm": 0.3301287889480591,
      "learning_rate": 3.4237597100180515e-05,
      "loss": 0.0542,
      "step": 37320
    },
    {
      "epoch": 4.366955300725485,
      "loss_main": 0.025099368765950203,
      "step": 37320
    },
    {
      "epoch": 4.368125438801779,
      "grad_norm": 0.24095411598682404,
      "learning_rate": 3.4211446858216427e-05,
      "loss": 0.0603,
      "step": 37330
    },
    {
      "epoch": 4.368125438801779,
      "loss_main": 0.0608048178255558,
      "step": 37330
    },
    {
      "epoch": 4.369295576878072,
      "grad_norm": 0.26056933403015137,
      "learning_rate": 3.4185301412397915e-05,
      "loss": 0.0668,
      "step": 37340
    },
    {
      "epoch": 4.369295576878072,
      "loss_main": 0.06691478192806244,
      "step": 37340
    },
    {
      "epoch": 4.370465714954364,
      "grad_norm": 0.2728741765022278,
      "learning_rate": 3.415916077066729e-05,
      "loss": 0.0618,
      "step": 37350
    },
    {
      "epoch": 4.370465714954364,
      "loss_main": 0.025808559730648994,
      "step": 37350
    },
    {
      "epoch": 4.371635853030657,
      "grad_norm": 0.22738242149353027,
      "learning_rate": 3.413302494096535e-05,
      "loss": 0.0494,
      "step": 37360
    },
    {
      "epoch": 4.371635853030657,
      "loss_main": 0.07430495321750641,
      "step": 37360
    },
    {
      "epoch": 4.372805991106951,
      "grad_norm": 0.2748123109340668,
      "learning_rate": 3.410689393123151e-05,
      "loss": 0.0558,
      "step": 37370
    },
    {
      "epoch": 4.372805991106951,
      "loss_main": 0.06734324246644974,
      "step": 37370
    },
    {
      "epoch": 4.3739761291832435,
      "grad_norm": 0.3340875208377838,
      "learning_rate": 3.408076774940364e-05,
      "loss": 0.0573,
      "step": 37380
    },
    {
      "epoch": 4.3739761291832435,
      "loss_main": 0.06875162571668625,
      "step": 37380
    },
    {
      "epoch": 4.375146267259536,
      "grad_norm": 0.27147385478019714,
      "learning_rate": 3.40546464034182e-05,
      "loss": 0.059,
      "step": 37390
    },
    {
      "epoch": 4.375146267259536,
      "loss_main": 0.07437942177057266,
      "step": 37390
    },
    {
      "epoch": 4.37631640533583,
      "grad_norm": 0.520108699798584,
      "learning_rate": 3.4028529901210185e-05,
      "loss": 0.0682,
      "step": 37400
    },
    {
      "epoch": 4.37631640533583,
      "loss_main": 0.11118372529745102,
      "step": 37400
    },
    {
      "epoch": 4.377486543412123,
      "grad_norm": 0.19998350739479065,
      "learning_rate": 3.4002418250713086e-05,
      "loss": 0.068,
      "step": 37410
    },
    {
      "epoch": 4.377486543412123,
      "loss_main": 0.05008476972579956,
      "step": 37410
    },
    {
      "epoch": 4.378656681488415,
      "grad_norm": 0.25613829493522644,
      "learning_rate": 3.3976311459858936e-05,
      "loss": 0.056,
      "step": 37420
    },
    {
      "epoch": 4.378656681488415,
      "loss_main": 0.04895516112446785,
      "step": 37420
    },
    {
      "epoch": 4.379826819564709,
      "grad_norm": 0.33805620670318604,
      "learning_rate": 3.395020953657826e-05,
      "loss": 0.0632,
      "step": 37430
    },
    {
      "epoch": 4.379826819564709,
      "loss_main": 0.039348747581243515,
      "step": 37430
    },
    {
      "epoch": 4.380996957641002,
      "grad_norm": 0.22769507765769958,
      "learning_rate": 3.3924112488800165e-05,
      "loss": 0.0495,
      "step": 37440
    },
    {
      "epoch": 4.380996957641002,
      "loss_main": 0.04847462847828865,
      "step": 37440
    },
    {
      "epoch": 4.3821670957172945,
      "grad_norm": 0.25700831413269043,
      "learning_rate": 3.389802032445225e-05,
      "loss": 0.0513,
      "step": 37450
    },
    {
      "epoch": 4.3821670957172945,
      "loss_main": 0.03187980875372887,
      "step": 37450
    },
    {
      "epoch": 4.383337233793588,
      "grad_norm": 0.2487405687570572,
      "learning_rate": 3.38719330514606e-05,
      "loss": 0.0533,
      "step": 37460
    },
    {
      "epoch": 4.383337233793588,
      "loss_main": 0.027619851753115654,
      "step": 37460
    },
    {
      "epoch": 4.384507371869881,
      "grad_norm": 0.18377359211444855,
      "learning_rate": 3.3845850677749866e-05,
      "loss": 0.0577,
      "step": 37470
    },
    {
      "epoch": 4.384507371869881,
      "loss_main": 0.09715836495161057,
      "step": 37470
    },
    {
      "epoch": 4.385677509946174,
      "grad_norm": 0.310506671667099,
      "learning_rate": 3.3819773211243157e-05,
      "loss": 0.0533,
      "step": 37480
    },
    {
      "epoch": 4.385677509946174,
      "loss_main": 0.042649321258068085,
      "step": 37480
    },
    {
      "epoch": 4.386847648022466,
      "grad_norm": 0.19047750532627106,
      "learning_rate": 3.379370065986213e-05,
      "loss": 0.0472,
      "step": 37490
    },
    {
      "epoch": 4.386847648022466,
      "loss_main": 0.06350557506084442,
      "step": 37490
    },
    {
      "epoch": 4.38801778609876,
      "grad_norm": 0.2635789215564728,
      "learning_rate": 3.3767633031526955e-05,
      "loss": 0.0514,
      "step": 37500
    },
    {
      "epoch": 4.38801778609876,
      "loss_main": 0.07172032445669174,
      "step": 37500
    },
    {
      "epoch": 4.389187924175053,
      "grad_norm": 0.26511961221694946,
      "learning_rate": 3.374157033415626e-05,
      "loss": 0.0521,
      "step": 37510
    },
    {
      "epoch": 4.389187924175053,
      "loss_main": 0.06622758507728577,
      "step": 37510
    },
    {
      "epoch": 4.390358062251345,
      "grad_norm": 0.23277896642684937,
      "learning_rate": 3.371551257566723e-05,
      "loss": 0.0547,
      "step": 37520
    },
    {
      "epoch": 4.390358062251345,
      "loss_main": 0.10754282027482986,
      "step": 37520
    },
    {
      "epoch": 4.391528200327639,
      "grad_norm": 0.19822773337364197,
      "learning_rate": 3.36894597639755e-05,
      "loss": 0.0603,
      "step": 37530
    },
    {
      "epoch": 4.391528200327639,
      "loss_main": 0.07942627370357513,
      "step": 37530
    },
    {
      "epoch": 4.392698338403932,
      "grad_norm": 0.2114037126302719,
      "learning_rate": 3.366341190699523e-05,
      "loss": 0.058,
      "step": 37540
    },
    {
      "epoch": 4.392698338403932,
      "loss_main": 0.06609277427196503,
      "step": 37540
    },
    {
      "epoch": 4.3938684764802245,
      "grad_norm": 0.39282381534576416,
      "learning_rate": 3.36373690126391e-05,
      "loss": 0.0518,
      "step": 37550
    },
    {
      "epoch": 4.3938684764802245,
      "loss_main": 0.025770144537091255,
      "step": 37550
    },
    {
      "epoch": 4.395038614556517,
      "grad_norm": 0.274088054895401,
      "learning_rate": 3.3611331088818234e-05,
      "loss": 0.0574,
      "step": 37560
    },
    {
      "epoch": 4.395038614556517,
      "loss_main": 0.06493836641311646,
      "step": 37560
    },
    {
      "epoch": 4.396208752632811,
      "grad_norm": 0.23595605790615082,
      "learning_rate": 3.3585298143442265e-05,
      "loss": 0.0542,
      "step": 37570
    },
    {
      "epoch": 4.396208752632811,
      "loss_main": 0.05389075726270676,
      "step": 37570
    },
    {
      "epoch": 4.397378890709104,
      "grad_norm": 0.37371230125427246,
      "learning_rate": 3.35592701844193e-05,
      "loss": 0.0604,
      "step": 37580
    },
    {
      "epoch": 4.397378890709104,
      "loss_main": 0.03991037234663963,
      "step": 37580
    },
    {
      "epoch": 4.398549028785396,
      "grad_norm": 0.24673530459403992,
      "learning_rate": 3.353324721965596e-05,
      "loss": 0.0573,
      "step": 37590
    },
    {
      "epoch": 4.398549028785396,
      "loss_main": 0.06332585960626602,
      "step": 37590
    },
    {
      "epoch": 4.39971916686169,
      "grad_norm": 0.32989755272865295,
      "learning_rate": 3.350722925705736e-05,
      "loss": 0.0509,
      "step": 37600
    },
    {
      "epoch": 4.39971916686169,
      "loss_main": 0.04961025342345238,
      "step": 37600
    },
    {
      "epoch": 4.400889304937983,
      "grad_norm": 0.2270893007516861,
      "learning_rate": 3.348121630452703e-05,
      "loss": 0.0611,
      "step": 37610
    },
    {
      "epoch": 4.400889304937983,
      "loss_main": 0.0708189308643341,
      "step": 37610
    },
    {
      "epoch": 4.402059443014275,
      "grad_norm": 0.20685923099517822,
      "learning_rate": 3.3455208369967044e-05,
      "loss": 0.0514,
      "step": 37620
    },
    {
      "epoch": 4.402059443014275,
      "loss_main": 0.042610298842191696,
      "step": 37620
    },
    {
      "epoch": 4.403229581090569,
      "grad_norm": 0.20750613510608673,
      "learning_rate": 3.34292054612779e-05,
      "loss": 0.0511,
      "step": 37630
    },
    {
      "epoch": 4.403229581090569,
      "loss_main": 0.043859753757715225,
      "step": 37630
    },
    {
      "epoch": 4.404399719166862,
      "grad_norm": 0.20810016989707947,
      "learning_rate": 3.340320758635861e-05,
      "loss": 0.0608,
      "step": 37640
    },
    {
      "epoch": 4.404399719166862,
      "loss_main": 0.03134544938802719,
      "step": 37640
    },
    {
      "epoch": 4.4055698572431545,
      "grad_norm": 0.28522828221321106,
      "learning_rate": 3.337721475310666e-05,
      "loss": 0.0533,
      "step": 37650
    },
    {
      "epoch": 4.4055698572431545,
      "loss_main": 0.057346634566783905,
      "step": 37650
    },
    {
      "epoch": 4.406739995319448,
      "grad_norm": 0.29749399423599243,
      "learning_rate": 3.335122696941795e-05,
      "loss": 0.0552,
      "step": 37660
    },
    {
      "epoch": 4.406739995319448,
      "loss_main": 0.05993777513504028,
      "step": 37660
    },
    {
      "epoch": 4.407910133395741,
      "grad_norm": 0.30333542823791504,
      "learning_rate": 3.332524424318692e-05,
      "loss": 0.0554,
      "step": 37670
    },
    {
      "epoch": 4.407910133395741,
      "loss_main": 0.03288494795560837,
      "step": 37670
    },
    {
      "epoch": 4.409080271472034,
      "grad_norm": 0.2234320044517517,
      "learning_rate": 3.32992665823064e-05,
      "loss": 0.0492,
      "step": 37680
    },
    {
      "epoch": 4.409080271472034,
      "loss_main": 0.08007635176181793,
      "step": 37680
    },
    {
      "epoch": 4.410250409548326,
      "grad_norm": 0.31160596013069153,
      "learning_rate": 3.327329399466774e-05,
      "loss": 0.0659,
      "step": 37690
    },
    {
      "epoch": 4.410250409548326,
      "loss_main": 0.03413292020559311,
      "step": 37690
    },
    {
      "epoch": 4.41142054762462,
      "grad_norm": 0.23963606357574463,
      "learning_rate": 3.324732648816072e-05,
      "loss": 0.0542,
      "step": 37700
    },
    {
      "epoch": 4.41142054762462,
      "loss_main": 0.03863959386944771,
      "step": 37700
    },
    {
      "epoch": 4.412590685700913,
      "grad_norm": 0.2881587743759155,
      "learning_rate": 3.322136407067358e-05,
      "loss": 0.0531,
      "step": 37710
    },
    {
      "epoch": 4.412590685700913,
      "loss_main": 0.06352158635854721,
      "step": 37710
    },
    {
      "epoch": 4.4137608237772055,
      "grad_norm": 0.3669435977935791,
      "learning_rate": 3.3195406750093036e-05,
      "loss": 0.054,
      "step": 37720
    },
    {
      "epoch": 4.4137608237772055,
      "loss_main": 0.061877649277448654,
      "step": 37720
    },
    {
      "epoch": 4.414930961853499,
      "grad_norm": 0.34087827801704407,
      "learning_rate": 3.3169454534304205e-05,
      "loss": 0.0498,
      "step": 37730
    },
    {
      "epoch": 4.414930961853499,
      "loss_main": 0.03534875437617302,
      "step": 37730
    },
    {
      "epoch": 4.416101099929792,
      "grad_norm": 0.2577366530895233,
      "learning_rate": 3.3143507431190725e-05,
      "loss": 0.0575,
      "step": 37740
    },
    {
      "epoch": 4.416101099929792,
      "loss_main": 0.06680140644311905,
      "step": 37740
    },
    {
      "epoch": 4.417271238006085,
      "grad_norm": 0.3336004316806793,
      "learning_rate": 3.311756544863459e-05,
      "loss": 0.0488,
      "step": 37750
    },
    {
      "epoch": 4.417271238006085,
      "loss_main": 0.0347498320043087,
      "step": 37750
    },
    {
      "epoch": 4.418441376082377,
      "grad_norm": 0.257564514875412,
      "learning_rate": 3.309162859451633e-05,
      "loss": 0.0655,
      "step": 37760
    },
    {
      "epoch": 4.418441376082377,
      "loss_main": 0.037569351494312286,
      "step": 37760
    },
    {
      "epoch": 4.419611514158671,
      "grad_norm": 0.17491206526756287,
      "learning_rate": 3.306569687671487e-05,
      "loss": 0.0471,
      "step": 37770
    },
    {
      "epoch": 4.419611514158671,
      "loss_main": 0.08196764439344406,
      "step": 37770
    },
    {
      "epoch": 4.420781652234964,
      "grad_norm": 0.26714274287223816,
      "learning_rate": 3.303977030310756e-05,
      "loss": 0.0573,
      "step": 37780
    },
    {
      "epoch": 4.420781652234964,
      "loss_main": 0.07356606423854828,
      "step": 37780
    },
    {
      "epoch": 4.421951790311256,
      "grad_norm": 0.3170718848705292,
      "learning_rate": 3.3013848881570245e-05,
      "loss": 0.0574,
      "step": 37790
    },
    {
      "epoch": 4.421951790311256,
      "loss_main": 0.04168710112571716,
      "step": 37790
    },
    {
      "epoch": 4.42312192838755,
      "grad_norm": 0.2957955002784729,
      "learning_rate": 3.298793261997712e-05,
      "loss": 0.0547,
      "step": 37800
    },
    {
      "epoch": 4.42312192838755,
      "loss_main": 0.045611876994371414,
      "step": 37800
    },
    {
      "epoch": 4.424292066463843,
      "grad_norm": 0.28226354718208313,
      "learning_rate": 3.2962021526200893e-05,
      "loss": 0.0552,
      "step": 37810
    },
    {
      "epoch": 4.424292066463843,
      "loss_main": 0.04047570750117302,
      "step": 37810
    },
    {
      "epoch": 4.4254622045401355,
      "grad_norm": 0.2504005432128906,
      "learning_rate": 3.293611560811268e-05,
      "loss": 0.053,
      "step": 37820
    },
    {
      "epoch": 4.4254622045401355,
      "loss_main": 0.05827874690294266,
      "step": 37820
    },
    {
      "epoch": 4.426632342616429,
      "grad_norm": 0.36923569440841675,
      "learning_rate": 3.291021487358199e-05,
      "loss": 0.064,
      "step": 37830
    },
    {
      "epoch": 4.426632342616429,
      "loss_main": 0.06268500536680222,
      "step": 37830
    },
    {
      "epoch": 4.427802480692722,
      "grad_norm": 0.251625657081604,
      "learning_rate": 3.28843193304768e-05,
      "loss": 0.0647,
      "step": 37840
    },
    {
      "epoch": 4.427802480692722,
      "loss_main": 0.04603031277656555,
      "step": 37840
    },
    {
      "epoch": 4.428972618769015,
      "grad_norm": 0.2700379192829132,
      "learning_rate": 3.2858428986663456e-05,
      "loss": 0.0601,
      "step": 37850
    },
    {
      "epoch": 4.428972618769015,
      "loss_main": 0.04966280236840248,
      "step": 37850
    },
    {
      "epoch": 4.430142756845307,
      "grad_norm": 0.3044460713863373,
      "learning_rate": 3.283254385000681e-05,
      "loss": 0.0517,
      "step": 37860
    },
    {
      "epoch": 4.430142756845307,
      "loss_main": 0.06229588761925697,
      "step": 37860
    },
    {
      "epoch": 4.431312894921601,
      "grad_norm": 0.3011935353279114,
      "learning_rate": 3.2806663928370076e-05,
      "loss": 0.056,
      "step": 37870
    },
    {
      "epoch": 4.431312894921601,
      "loss_main": 0.05254372954368591,
      "step": 37870
    },
    {
      "epoch": 4.432483032997894,
      "grad_norm": 0.320764422416687,
      "learning_rate": 3.278078922961485e-05,
      "loss": 0.0521,
      "step": 37880
    },
    {
      "epoch": 4.432483032997894,
      "loss_main": 0.03714853152632713,
      "step": 37880
    },
    {
      "epoch": 4.433653171074186,
      "grad_norm": 0.381423681974411,
      "learning_rate": 3.275491976160123e-05,
      "loss": 0.0505,
      "step": 37890
    },
    {
      "epoch": 4.433653171074186,
      "loss_main": 0.07782776653766632,
      "step": 37890
    },
    {
      "epoch": 4.43482330915048,
      "grad_norm": 0.3263627886772156,
      "learning_rate": 3.2729055532187645e-05,
      "loss": 0.0505,
      "step": 37900
    },
    {
      "epoch": 4.43482330915048,
      "loss_main": 0.04156313091516495,
      "step": 37900
    },
    {
      "epoch": 4.435993447226773,
      "grad_norm": 0.2938850224018097,
      "learning_rate": 3.270319654923097e-05,
      "loss": 0.0558,
      "step": 37910
    },
    {
      "epoch": 4.435993447226773,
      "loss_main": 0.09461193531751633,
      "step": 37910
    },
    {
      "epoch": 4.4371635853030655,
      "grad_norm": 0.26264020800590515,
      "learning_rate": 3.2677342820586506e-05,
      "loss": 0.056,
      "step": 37920
    },
    {
      "epoch": 4.4371635853030655,
      "loss_main": 0.06420411169528961,
      "step": 37920
    },
    {
      "epoch": 4.438333723379359,
      "grad_norm": 0.23296406865119934,
      "learning_rate": 3.2651494354107905e-05,
      "loss": 0.0572,
      "step": 37930
    },
    {
      "epoch": 4.438333723379359,
      "loss_main": 0.04793833941221237,
      "step": 37930
    },
    {
      "epoch": 4.439503861455652,
      "grad_norm": 0.2359887808561325,
      "learning_rate": 3.2625651157647266e-05,
      "loss": 0.046,
      "step": 37940
    },
    {
      "epoch": 4.439503861455652,
      "loss_main": 0.027596473693847656,
      "step": 37940
    },
    {
      "epoch": 4.440673999531945,
      "grad_norm": 0.19040125608444214,
      "learning_rate": 3.259981323905505e-05,
      "loss": 0.0491,
      "step": 37950
    },
    {
      "epoch": 4.440673999531945,
      "loss_main": 0.07876519113779068,
      "step": 37950
    },
    {
      "epoch": 4.441844137608237,
      "grad_norm": 0.19506452977657318,
      "learning_rate": 3.257398060618014e-05,
      "loss": 0.0523,
      "step": 37960
    },
    {
      "epoch": 4.441844137608237,
      "loss_main": 0.09354522079229355,
      "step": 37960
    },
    {
      "epoch": 4.443014275684531,
      "grad_norm": 0.2679009735584259,
      "learning_rate": 3.254815326686983e-05,
      "loss": 0.0486,
      "step": 37970
    },
    {
      "epoch": 4.443014275684531,
      "loss_main": 0.04525882750749588,
      "step": 37970
    },
    {
      "epoch": 4.444184413760824,
      "grad_norm": 0.2954278886318207,
      "learning_rate": 3.2522331228969774e-05,
      "loss": 0.0529,
      "step": 37980
    },
    {
      "epoch": 4.444184413760824,
      "loss_main": 0.03339966759085655,
      "step": 37980
    },
    {
      "epoch": 4.4453545518371165,
      "grad_norm": 0.20786072313785553,
      "learning_rate": 3.2496514500324006e-05,
      "loss": 0.0538,
      "step": 37990
    },
    {
      "epoch": 4.4453545518371165,
      "loss_main": 0.050733186304569244,
      "step": 37990
    },
    {
      "epoch": 4.44652468991341,
      "grad_norm": 0.2874077260494232,
      "learning_rate": 3.247070308877498e-05,
      "loss": 0.0449,
      "step": 38000
    },
    {
      "epoch": 4.44652468991341,
      "loss_main": 0.06869761645793915,
      "step": 38000
    },
    {
      "epoch": 4.447694827989703,
      "grad_norm": 0.2113824188709259,
      "learning_rate": 3.2444897002163515e-05,
      "loss": 0.0525,
      "step": 38010
    },
    {
      "epoch": 4.447694827989703,
      "loss_main": 0.040881019085645676,
      "step": 38010
    },
    {
      "epoch": 4.448864966065996,
      "grad_norm": 0.4167325496673584,
      "learning_rate": 3.241909624832885e-05,
      "loss": 0.0581,
      "step": 38020
    },
    {
      "epoch": 4.448864966065996,
      "loss_main": 0.046651773154735565,
      "step": 38020
    },
    {
      "epoch": 4.450035104142289,
      "grad_norm": 0.3206314444541931,
      "learning_rate": 3.239330083510852e-05,
      "loss": 0.062,
      "step": 38030
    },
    {
      "epoch": 4.450035104142289,
      "loss_main": 0.038004644215106964,
      "step": 38030
    },
    {
      "epoch": 4.451205242218582,
      "grad_norm": 0.2660113275051117,
      "learning_rate": 3.236751077033855e-05,
      "loss": 0.0542,
      "step": 38040
    },
    {
      "epoch": 4.451205242218582,
      "loss_main": 0.060836318880319595,
      "step": 38040
    },
    {
      "epoch": 4.452375380294875,
      "grad_norm": 0.2991066575050354,
      "learning_rate": 3.234172606185322e-05,
      "loss": 0.0679,
      "step": 38050
    },
    {
      "epoch": 4.452375380294875,
      "loss_main": 0.04830366373062134,
      "step": 38050
    },
    {
      "epoch": 4.453545518371167,
      "grad_norm": 0.25371745228767395,
      "learning_rate": 3.231594671748528e-05,
      "loss": 0.0529,
      "step": 38060
    },
    {
      "epoch": 4.453545518371167,
      "loss_main": 0.05555172637104988,
      "step": 38060
    },
    {
      "epoch": 4.454715656447461,
      "grad_norm": 0.21718105673789978,
      "learning_rate": 3.2290172745065815e-05,
      "loss": 0.0561,
      "step": 38070
    },
    {
      "epoch": 4.454715656447461,
      "loss_main": 0.04861225560307503,
      "step": 38070
    },
    {
      "epoch": 4.455885794523754,
      "grad_norm": 0.29929450154304504,
      "learning_rate": 3.226440415242426e-05,
      "loss": 0.0548,
      "step": 38080
    },
    {
      "epoch": 4.455885794523754,
      "loss_main": 0.059500861912965775,
      "step": 38080
    },
    {
      "epoch": 4.4570559326000465,
      "grad_norm": 0.20579561591148376,
      "learning_rate": 3.223864094738846e-05,
      "loss": 0.054,
      "step": 38090
    },
    {
      "epoch": 4.4570559326000465,
      "loss_main": 0.04035510495305061,
      "step": 38090
    },
    {
      "epoch": 4.45822607067634,
      "grad_norm": 0.23447653651237488,
      "learning_rate": 3.221288313778456e-05,
      "loss": 0.0429,
      "step": 38100
    },
    {
      "epoch": 4.45822607067634,
      "loss_main": 0.03148449584841728,
      "step": 38100
    },
    {
      "epoch": 4.459396208752633,
      "grad_norm": 0.17249910533428192,
      "learning_rate": 3.2187130731437125e-05,
      "loss": 0.0476,
      "step": 38110
    },
    {
      "epoch": 4.459396208752633,
      "loss_main": 0.051307886838912964,
      "step": 38110
    },
    {
      "epoch": 4.460566346828926,
      "grad_norm": 0.3114623725414276,
      "learning_rate": 3.216138373616905e-05,
      "loss": 0.0522,
      "step": 38120
    },
    {
      "epoch": 4.460566346828926,
      "loss_main": 0.05815992131829262,
      "step": 38120
    },
    {
      "epoch": 4.461736484905218,
      "grad_norm": 0.25425130128860474,
      "learning_rate": 3.21356421598016e-05,
      "loss": 0.0557,
      "step": 38130
    },
    {
      "epoch": 4.461736484905218,
      "loss_main": 0.06086571887135506,
      "step": 38130
    },
    {
      "epoch": 4.462906622981512,
      "grad_norm": 0.23296299576759338,
      "learning_rate": 3.210990601015438e-05,
      "loss": 0.0558,
      "step": 38140
    },
    {
      "epoch": 4.462906622981512,
      "loss_main": 0.038514018058776855,
      "step": 38140
    },
    {
      "epoch": 4.464076761057805,
      "grad_norm": 0.28628504276275635,
      "learning_rate": 3.208417529504535e-05,
      "loss": 0.0543,
      "step": 38150
    },
    {
      "epoch": 4.464076761057805,
      "loss_main": 0.04055303335189819,
      "step": 38150
    },
    {
      "epoch": 4.4652468991340974,
      "grad_norm": 0.2823774814605713,
      "learning_rate": 3.205845002229084e-05,
      "loss": 0.0551,
      "step": 38160
    },
    {
      "epoch": 4.4652468991340974,
      "loss_main": 0.030771052464842796,
      "step": 38160
    },
    {
      "epoch": 4.466417037210391,
      "grad_norm": 0.28854310512542725,
      "learning_rate": 3.203273019970547e-05,
      "loss": 0.044,
      "step": 38170
    },
    {
      "epoch": 4.466417037210391,
      "loss_main": 0.08838474005460739,
      "step": 38170
    },
    {
      "epoch": 4.467587175286684,
      "grad_norm": 0.2265348732471466,
      "learning_rate": 3.200701583510227e-05,
      "loss": 0.0594,
      "step": 38180
    },
    {
      "epoch": 4.467587175286684,
      "loss_main": 0.05608966574072838,
      "step": 38180
    },
    {
      "epoch": 4.4687573133629765,
      "grad_norm": 0.24612411856651306,
      "learning_rate": 3.198130693629261e-05,
      "loss": 0.0554,
      "step": 38190
    },
    {
      "epoch": 4.4687573133629765,
      "loss_main": 0.060018058866262436,
      "step": 38190
    },
    {
      "epoch": 4.46992745143927,
      "grad_norm": 0.28140684962272644,
      "learning_rate": 3.195560351108612e-05,
      "loss": 0.0628,
      "step": 38200
    },
    {
      "epoch": 4.46992745143927,
      "loss_main": 0.03756777569651604,
      "step": 38200
    },
    {
      "epoch": 4.471097589515563,
      "grad_norm": 0.28325846791267395,
      "learning_rate": 3.1929905567290865e-05,
      "loss": 0.0508,
      "step": 38210
    },
    {
      "epoch": 4.471097589515563,
      "loss_main": 0.05014026537537575,
      "step": 38210
    },
    {
      "epoch": 4.472267727591856,
      "grad_norm": 0.23098339140415192,
      "learning_rate": 3.1904213112713164e-05,
      "loss": 0.0535,
      "step": 38220
    },
    {
      "epoch": 4.472267727591856,
      "loss_main": 0.06490444391965866,
      "step": 38220
    },
    {
      "epoch": 4.473437865668149,
      "grad_norm": 0.23116157948970795,
      "learning_rate": 3.187852615515774e-05,
      "loss": 0.0547,
      "step": 38230
    },
    {
      "epoch": 4.473437865668149,
      "loss_main": 0.03569107875227928,
      "step": 38230
    },
    {
      "epoch": 4.474608003744442,
      "grad_norm": 0.24176891148090363,
      "learning_rate": 3.1852844702427606e-05,
      "loss": 0.0468,
      "step": 38240
    },
    {
      "epoch": 4.474608003744442,
      "loss_main": 0.049815379083156586,
      "step": 38240
    },
    {
      "epoch": 4.475778141820735,
      "grad_norm": 0.18875367939472198,
      "learning_rate": 3.18271687623241e-05,
      "loss": 0.054,
      "step": 38250
    },
    {
      "epoch": 4.475778141820735,
      "loss_main": 0.034290824085474014,
      "step": 38250
    },
    {
      "epoch": 4.4769482798970275,
      "grad_norm": 0.22415372729301453,
      "learning_rate": 3.1801498342646896e-05,
      "loss": 0.0629,
      "step": 38260
    },
    {
      "epoch": 4.4769482798970275,
      "loss_main": 0.030614761635661125,
      "step": 38260
    },
    {
      "epoch": 4.478118417973321,
      "grad_norm": 0.275786429643631,
      "learning_rate": 3.177583345119398e-05,
      "loss": 0.0565,
      "step": 38270
    },
    {
      "epoch": 4.478118417973321,
      "loss_main": 0.06481820344924927,
      "step": 38270
    },
    {
      "epoch": 4.479288556049614,
      "grad_norm": 0.2404395341873169,
      "learning_rate": 3.17501740957617e-05,
      "loss": 0.0491,
      "step": 38280
    },
    {
      "epoch": 4.479288556049614,
      "loss_main": 0.05912092328071594,
      "step": 38280
    },
    {
      "epoch": 4.480458694125907,
      "grad_norm": 0.24188488721847534,
      "learning_rate": 3.172452028414467e-05,
      "loss": 0.052,
      "step": 38290
    },
    {
      "epoch": 4.480458694125907,
      "loss_main": 0.036657363176345825,
      "step": 38290
    },
    {
      "epoch": 4.4816288322022,
      "grad_norm": 0.344727098941803,
      "learning_rate": 3.169887202413583e-05,
      "loss": 0.0592,
      "step": 38300
    },
    {
      "epoch": 4.4816288322022,
      "loss_main": 0.05023447051644325,
      "step": 38300
    },
    {
      "epoch": 4.482798970278493,
      "grad_norm": 0.3067663311958313,
      "learning_rate": 3.167322932352646e-05,
      "loss": 0.0649,
      "step": 38310
    },
    {
      "epoch": 4.482798970278493,
      "loss_main": 0.041220176964998245,
      "step": 38310
    },
    {
      "epoch": 4.483969108354786,
      "grad_norm": 0.2789602279663086,
      "learning_rate": 3.164759219010613e-05,
      "loss": 0.0513,
      "step": 38320
    },
    {
      "epoch": 4.483969108354786,
      "loss_main": 0.04381043091416359,
      "step": 38320
    },
    {
      "epoch": 4.485139246431078,
      "grad_norm": 0.27772367000579834,
      "learning_rate": 3.1621960631662725e-05,
      "loss": 0.053,
      "step": 38330
    },
    {
      "epoch": 4.485139246431078,
      "loss_main": 0.043853361159563065,
      "step": 38330
    },
    {
      "epoch": 4.486309384507372,
      "grad_norm": 0.3160598874092102,
      "learning_rate": 3.159633465598245e-05,
      "loss": 0.0579,
      "step": 38340
    },
    {
      "epoch": 4.486309384507372,
      "loss_main": 0.05377513915300369,
      "step": 38340
    },
    {
      "epoch": 4.487479522583665,
      "grad_norm": 0.18990886211395264,
      "learning_rate": 3.1570714270849767e-05,
      "loss": 0.0505,
      "step": 38350
    },
    {
      "epoch": 4.487479522583665,
      "loss_main": 0.056019775569438934,
      "step": 38350
    },
    {
      "epoch": 4.4886496606599575,
      "grad_norm": 0.18384727835655212,
      "learning_rate": 3.1545099484047516e-05,
      "loss": 0.0503,
      "step": 38360
    },
    {
      "epoch": 4.4886496606599575,
      "loss_main": 0.05439659580588341,
      "step": 38360
    },
    {
      "epoch": 4.489819798736251,
      "grad_norm": 0.25454044342041016,
      "learning_rate": 3.151949030335674e-05,
      "loss": 0.0563,
      "step": 38370
    },
    {
      "epoch": 4.489819798736251,
      "loss_main": 0.04691959172487259,
      "step": 38370
    },
    {
      "epoch": 4.490989936812544,
      "grad_norm": 0.2545521557331085,
      "learning_rate": 3.149388673655687e-05,
      "loss": 0.0607,
      "step": 38380
    },
    {
      "epoch": 4.490989936812544,
      "loss_main": 0.04928344488143921,
      "step": 38380
    },
    {
      "epoch": 4.492160074888837,
      "grad_norm": 0.46959057450294495,
      "learning_rate": 3.146828879142559e-05,
      "loss": 0.0562,
      "step": 38390
    },
    {
      "epoch": 4.492160074888837,
      "loss_main": 0.044822998344898224,
      "step": 38390
    },
    {
      "epoch": 4.49333021296513,
      "grad_norm": 0.24913999438285828,
      "learning_rate": 3.1442696475738866e-05,
      "loss": 0.0512,
      "step": 38400
    },
    {
      "epoch": 4.49333021296513,
      "loss_main": 0.043329838663339615,
      "step": 38400
    },
    {
      "epoch": 4.494500351041423,
      "grad_norm": 0.3481089770793915,
      "learning_rate": 3.141710979727098e-05,
      "loss": 0.0473,
      "step": 38410
    },
    {
      "epoch": 4.494500351041423,
      "loss_main": 0.04451863095164299,
      "step": 38410
    },
    {
      "epoch": 4.495670489117716,
      "grad_norm": 0.1950022131204605,
      "learning_rate": 3.139152876379447e-05,
      "loss": 0.0561,
      "step": 38420
    },
    {
      "epoch": 4.495670489117716,
      "loss_main": 0.03421325609087944,
      "step": 38420
    },
    {
      "epoch": 4.496840627194009,
      "grad_norm": 0.24863159656524658,
      "learning_rate": 3.1365953383080214e-05,
      "loss": 0.0561,
      "step": 38430
    },
    {
      "epoch": 4.496840627194009,
      "loss_main": 0.050616223365068436,
      "step": 38430
    },
    {
      "epoch": 4.498010765270302,
      "grad_norm": 0.2538215219974518,
      "learning_rate": 3.134038366289731e-05,
      "loss": 0.0525,
      "step": 38440
    },
    {
      "epoch": 4.498010765270302,
      "loss_main": 0.03851620480418205,
      "step": 38440
    },
    {
      "epoch": 4.499180903346595,
      "grad_norm": 0.2362733781337738,
      "learning_rate": 3.131481961101317e-05,
      "loss": 0.0538,
      "step": 38450
    },
    {
      "epoch": 4.499180903346595,
      "loss_main": 0.050383467227220535,
      "step": 38450
    },
    {
      "epoch": 4.5003510414228876,
      "grad_norm": 0.22687751054763794,
      "learning_rate": 3.128926123519349e-05,
      "loss": 0.0622,
      "step": 38460
    },
    {
      "epoch": 4.5003510414228876,
      "loss_main": 0.027360688894987106,
      "step": 38460
    },
    {
      "epoch": 4.501521179499181,
      "grad_norm": 0.3034818470478058,
      "learning_rate": 3.1263708543202194e-05,
      "loss": 0.0479,
      "step": 38470
    },
    {
      "epoch": 4.501521179499181,
      "loss_main": 0.039295442402362823,
      "step": 38470
    },
    {
      "epoch": 4.502691317575474,
      "grad_norm": 0.21076315641403198,
      "learning_rate": 3.123816154280155e-05,
      "loss": 0.0578,
      "step": 38480
    },
    {
      "epoch": 4.502691317575474,
      "loss_main": 0.06654352694749832,
      "step": 38480
    },
    {
      "epoch": 4.503861455651767,
      "grad_norm": 0.2818489670753479,
      "learning_rate": 3.121262024175207e-05,
      "loss": 0.061,
      "step": 38490
    },
    {
      "epoch": 4.503861455651767,
      "loss_main": 0.0701446384191513,
      "step": 38490
    },
    {
      "epoch": 4.50503159372806,
      "grad_norm": 0.2751941680908203,
      "learning_rate": 3.118708464781248e-05,
      "loss": 0.0575,
      "step": 38500
    },
    {
      "epoch": 4.50503159372806,
      "loss_main": 0.054149869829416275,
      "step": 38500
    },
    {
      "epoch": 4.506201731804353,
      "grad_norm": 0.290889173746109,
      "learning_rate": 3.116155476873987e-05,
      "loss": 0.0571,
      "step": 38510
    },
    {
      "epoch": 4.506201731804353,
      "loss_main": 0.040643904358148575,
      "step": 38510
    },
    {
      "epoch": 4.507371869880646,
      "grad_norm": 0.23786617815494537,
      "learning_rate": 3.11360306122895e-05,
      "loss": 0.0581,
      "step": 38520
    },
    {
      "epoch": 4.507371869880646,
      "loss_main": 0.07308853417634964,
      "step": 38520
    },
    {
      "epoch": 4.5085420079569385,
      "grad_norm": 0.21966703236103058,
      "learning_rate": 3.1110512186214975e-05,
      "loss": 0.0569,
      "step": 38530
    },
    {
      "epoch": 4.5085420079569385,
      "loss_main": 0.05635714530944824,
      "step": 38530
    },
    {
      "epoch": 4.509712146033232,
      "grad_norm": 0.2967475950717926,
      "learning_rate": 3.1084999498268095e-05,
      "loss": 0.049,
      "step": 38540
    },
    {
      "epoch": 4.509712146033232,
      "loss_main": 0.059664011001586914,
      "step": 38540
    },
    {
      "epoch": 4.510882284109525,
      "grad_norm": 0.2693953514099121,
      "learning_rate": 3.1059492556198934e-05,
      "loss": 0.0522,
      "step": 38550
    },
    {
      "epoch": 4.510882284109525,
      "loss_main": 0.04707478731870651,
      "step": 38550
    },
    {
      "epoch": 4.512052422185818,
      "grad_norm": 0.3408527076244354,
      "learning_rate": 3.103399136775586e-05,
      "loss": 0.0579,
      "step": 38560
    },
    {
      "epoch": 4.512052422185818,
      "loss_main": 0.08012320846319199,
      "step": 38560
    },
    {
      "epoch": 4.513222560262111,
      "grad_norm": 0.22124747931957245,
      "learning_rate": 3.100849594068541e-05,
      "loss": 0.0599,
      "step": 38570
    },
    {
      "epoch": 4.513222560262111,
      "loss_main": 0.05131543427705765,
      "step": 38570
    },
    {
      "epoch": 4.514392698338404,
      "grad_norm": 0.19292578101158142,
      "learning_rate": 3.0983006282732484e-05,
      "loss": 0.0523,
      "step": 38580
    },
    {
      "epoch": 4.514392698338404,
      "loss_main": 0.05222319811582565,
      "step": 38580
    },
    {
      "epoch": 4.515562836414697,
      "grad_norm": 0.30675008893013,
      "learning_rate": 3.0957522401640116e-05,
      "loss": 0.0591,
      "step": 38590
    },
    {
      "epoch": 4.515562836414697,
      "loss_main": 0.030455678701400757,
      "step": 38590
    },
    {
      "epoch": 4.51673297449099,
      "grad_norm": 0.1857331246137619,
      "learning_rate": 3.0932044305149645e-05,
      "loss": 0.0479,
      "step": 38600
    },
    {
      "epoch": 4.51673297449099,
      "loss_main": 0.026849141344428062,
      "step": 38600
    },
    {
      "epoch": 4.517903112567283,
      "grad_norm": 0.20635434985160828,
      "learning_rate": 3.090657200100068e-05,
      "loss": 0.0478,
      "step": 38610
    },
    {
      "epoch": 4.517903112567283,
      "loss_main": 0.08654068410396576,
      "step": 38610
    },
    {
      "epoch": 4.519073250643576,
      "grad_norm": 0.3031258285045624,
      "learning_rate": 3.088110549693099e-05,
      "loss": 0.0564,
      "step": 38620
    },
    {
      "epoch": 4.519073250643576,
      "loss_main": 0.057685911655426025,
      "step": 38620
    },
    {
      "epoch": 4.520243388719869,
      "grad_norm": 0.24299858510494232,
      "learning_rate": 3.085564480067667e-05,
      "loss": 0.0523,
      "step": 38630
    },
    {
      "epoch": 4.520243388719869,
      "loss_main": 0.022429117932915688,
      "step": 38630
    },
    {
      "epoch": 4.521413526796162,
      "grad_norm": 0.24286718666553497,
      "learning_rate": 3.0830189919971955e-05,
      "loss": 0.0459,
      "step": 38640
    },
    {
      "epoch": 4.521413526796162,
      "loss_main": 0.040742915123701096,
      "step": 38640
    },
    {
      "epoch": 4.522583664872455,
      "grad_norm": 0.2383524477481842,
      "learning_rate": 3.080474086254939e-05,
      "loss": 0.0496,
      "step": 38650
    },
    {
      "epoch": 4.522583664872455,
      "loss_main": 0.04732893034815788,
      "step": 38650
    },
    {
      "epoch": 4.523753802948748,
      "grad_norm": 0.24911223351955414,
      "learning_rate": 3.077929763613975e-05,
      "loss": 0.054,
      "step": 38660
    },
    {
      "epoch": 4.523753802948748,
      "loss_main": 0.04922439903020859,
      "step": 38660
    },
    {
      "epoch": 4.524923941025041,
      "grad_norm": 0.23311227560043335,
      "learning_rate": 3.075386024847198e-05,
      "loss": 0.0511,
      "step": 38670
    },
    {
      "epoch": 4.524923941025041,
      "loss_main": 0.04573264345526695,
      "step": 38670
    },
    {
      "epoch": 4.526094079101334,
      "grad_norm": 0.2406875193119049,
      "learning_rate": 3.072842870727331e-05,
      "loss": 0.0518,
      "step": 38680
    },
    {
      "epoch": 4.526094079101334,
      "loss_main": 0.024885864928364754,
      "step": 38680
    },
    {
      "epoch": 4.527264217177627,
      "grad_norm": 0.26799654960632324,
      "learning_rate": 3.070300302026916e-05,
      "loss": 0.0531,
      "step": 38690
    },
    {
      "epoch": 4.527264217177627,
      "loss_main": 0.04430995136499405,
      "step": 38690
    },
    {
      "epoch": 4.5284343552539195,
      "grad_norm": 0.31574293971061707,
      "learning_rate": 3.067758319518318e-05,
      "loss": 0.0566,
      "step": 38700
    },
    {
      "epoch": 4.5284343552539195,
      "loss_main": 0.06404076516628265,
      "step": 38700
    },
    {
      "epoch": 4.529604493330213,
      "grad_norm": 0.30760201811790466,
      "learning_rate": 3.065216923973725e-05,
      "loss": 0.0524,
      "step": 38710
    },
    {
      "epoch": 4.529604493330213,
      "loss_main": 0.05202525481581688,
      "step": 38710
    },
    {
      "epoch": 4.530774631406506,
      "grad_norm": 0.21647576987743378,
      "learning_rate": 3.062676116165145e-05,
      "loss": 0.0525,
      "step": 38720
    },
    {
      "epoch": 4.530774631406506,
      "loss_main": 0.04022869095206261,
      "step": 38720
    },
    {
      "epoch": 4.531944769482799,
      "grad_norm": 0.2655732333660126,
      "learning_rate": 3.06013589686441e-05,
      "loss": 0.0507,
      "step": 38730
    },
    {
      "epoch": 4.531944769482799,
      "loss_main": 0.03892021253705025,
      "step": 38730
    },
    {
      "epoch": 4.533114907559092,
      "grad_norm": 0.24573256075382233,
      "learning_rate": 3.05759626684317e-05,
      "loss": 0.0544,
      "step": 38740
    },
    {
      "epoch": 4.533114907559092,
      "loss_main": 0.05984634906053543,
      "step": 38740
    },
    {
      "epoch": 4.534285045635385,
      "grad_norm": 0.2530142366886139,
      "learning_rate": 3.055057226872896e-05,
      "loss": 0.0531,
      "step": 38750
    },
    {
      "epoch": 4.534285045635385,
      "loss_main": 0.07910424470901489,
      "step": 38750
    },
    {
      "epoch": 4.535455183711678,
      "grad_norm": 0.2842549681663513,
      "learning_rate": 3.052518777724887e-05,
      "loss": 0.0525,
      "step": 38760
    },
    {
      "epoch": 4.535455183711678,
      "loss_main": 0.029902828857302666,
      "step": 38760
    },
    {
      "epoch": 4.536625321787971,
      "grad_norm": 0.21328960359096527,
      "learning_rate": 3.04998092017025e-05,
      "loss": 0.049,
      "step": 38770
    },
    {
      "epoch": 4.536625321787971,
      "loss_main": 0.08867727220058441,
      "step": 38770
    },
    {
      "epoch": 4.537795459864264,
      "grad_norm": 0.23867757618427277,
      "learning_rate": 3.0474436549799246e-05,
      "loss": 0.0597,
      "step": 38780
    },
    {
      "epoch": 4.537795459864264,
      "loss_main": 0.0495770238339901,
      "step": 38780
    },
    {
      "epoch": 4.538965597940557,
      "grad_norm": 0.31546512246131897,
      "learning_rate": 3.044906982924661e-05,
      "loss": 0.0586,
      "step": 38790
    },
    {
      "epoch": 4.538965597940557,
      "loss_main": 0.04264017567038536,
      "step": 38790
    },
    {
      "epoch": 4.54013573601685,
      "grad_norm": 0.2701174318790436,
      "learning_rate": 3.0423709047750337e-05,
      "loss": 0.0503,
      "step": 38800
    },
    {
      "epoch": 4.54013573601685,
      "loss_main": 0.029695967212319374,
      "step": 38800
    },
    {
      "epoch": 4.541305874093143,
      "grad_norm": 0.27960899472236633,
      "learning_rate": 3.03983542130144e-05,
      "loss": 0.0559,
      "step": 38810
    },
    {
      "epoch": 4.541305874093143,
      "loss_main": 0.04074588790535927,
      "step": 38810
    },
    {
      "epoch": 4.542476012169436,
      "grad_norm": 0.27918490767478943,
      "learning_rate": 3.0373005332740877e-05,
      "loss": 0.059,
      "step": 38820
    },
    {
      "epoch": 4.542476012169436,
      "loss_main": 0.05265870690345764,
      "step": 38820
    },
    {
      "epoch": 4.5436461502457295,
      "grad_norm": 0.2704511284828186,
      "learning_rate": 3.034766241463013e-05,
      "loss": 0.055,
      "step": 38830
    },
    {
      "epoch": 4.5436461502457295,
      "loss_main": 0.04462983086705208,
      "step": 38830
    },
    {
      "epoch": 4.544816288322022,
      "grad_norm": 0.2456730306148529,
      "learning_rate": 3.032232546638064e-05,
      "loss": 0.0518,
      "step": 38840
    },
    {
      "epoch": 4.544816288322022,
      "loss_main": 0.07547280937433243,
      "step": 38840
    },
    {
      "epoch": 4.545986426398315,
      "grad_norm": 0.22602252662181854,
      "learning_rate": 3.0296994495689114e-05,
      "loss": 0.05,
      "step": 38850
    },
    {
      "epoch": 4.545986426398315,
      "loss_main": 0.05063639208674431,
      "step": 38850
    },
    {
      "epoch": 4.547156564474608,
      "grad_norm": 0.2775900363922119,
      "learning_rate": 3.0271669510250444e-05,
      "loss": 0.0537,
      "step": 38860
    },
    {
      "epoch": 4.547156564474608,
      "loss_main": 0.05190616846084595,
      "step": 38860
    },
    {
      "epoch": 4.548326702550901,
      "grad_norm": 0.2054547369480133,
      "learning_rate": 3.024635051775766e-05,
      "loss": 0.0562,
      "step": 38870
    },
    {
      "epoch": 4.548326702550901,
      "loss_main": 0.03643620386719704,
      "step": 38870
    },
    {
      "epoch": 4.549496840627194,
      "grad_norm": 0.23682406544685364,
      "learning_rate": 3.022103752590205e-05,
      "loss": 0.0553,
      "step": 38880
    },
    {
      "epoch": 4.549496840627194,
      "loss_main": 0.05478304252028465,
      "step": 38880
    },
    {
      "epoch": 4.550666978703487,
      "grad_norm": 0.2262163609266281,
      "learning_rate": 3.0195730542372992e-05,
      "loss": 0.0554,
      "step": 38890
    },
    {
      "epoch": 4.550666978703487,
      "loss_main": 0.05685695633292198,
      "step": 38890
    },
    {
      "epoch": 4.5518371167797795,
      "grad_norm": 0.2051973193883896,
      "learning_rate": 3.0170429574858084e-05,
      "loss": 0.0512,
      "step": 38900
    },
    {
      "epoch": 4.5518371167797795,
      "loss_main": 0.07920428365468979,
      "step": 38900
    },
    {
      "epoch": 4.553007254856073,
      "grad_norm": 0.1810200959444046,
      "learning_rate": 3.0145134631043127e-05,
      "loss": 0.0496,
      "step": 38910
    },
    {
      "epoch": 4.553007254856073,
      "loss_main": 0.04244322329759598,
      "step": 38910
    },
    {
      "epoch": 4.554177392932366,
      "grad_norm": 0.2379302680492401,
      "learning_rate": 3.0119845718612018e-05,
      "loss": 0.0445,
      "step": 38920
    },
    {
      "epoch": 4.554177392932366,
      "loss_main": 0.04617396369576454,
      "step": 38920
    },
    {
      "epoch": 4.555347531008659,
      "grad_norm": 0.20419424772262573,
      "learning_rate": 3.009456284524688e-05,
      "loss": 0.0576,
      "step": 38930
    },
    {
      "epoch": 4.555347531008659,
      "loss_main": 0.070204958319664,
      "step": 38930
    },
    {
      "epoch": 4.556517669084952,
      "grad_norm": 0.2961868345737457,
      "learning_rate": 3.0069286018627967e-05,
      "loss": 0.0559,
      "step": 38940
    },
    {
      "epoch": 4.556517669084952,
      "loss_main": 0.051196031272411346,
      "step": 38940
    },
    {
      "epoch": 4.557687807161245,
      "grad_norm": 0.28324058651924133,
      "learning_rate": 3.0044015246433743e-05,
      "loss": 0.065,
      "step": 38950
    },
    {
      "epoch": 4.557687807161245,
      "loss_main": 0.0833495706319809,
      "step": 38950
    },
    {
      "epoch": 4.558857945237538,
      "grad_norm": 0.28172358870506287,
      "learning_rate": 3.0018750536340755e-05,
      "loss": 0.0526,
      "step": 38960
    },
    {
      "epoch": 4.558857945237538,
      "loss_main": 0.08688639849424362,
      "step": 38960
    },
    {
      "epoch": 4.560028083313831,
      "grad_norm": 0.2769036889076233,
      "learning_rate": 2.999349189602378e-05,
      "loss": 0.0626,
      "step": 38970
    },
    {
      "epoch": 4.560028083313831,
      "loss_main": 0.026267120614647865,
      "step": 38970
    },
    {
      "epoch": 4.561198221390124,
      "grad_norm": 0.3100900650024414,
      "learning_rate": 2.9968239333155733e-05,
      "loss": 0.0464,
      "step": 38980
    },
    {
      "epoch": 4.561198221390124,
      "loss_main": 0.061363335698843,
      "step": 38980
    },
    {
      "epoch": 4.562368359466417,
      "grad_norm": 0.3475107252597809,
      "learning_rate": 2.994299285540767e-05,
      "loss": 0.0484,
      "step": 38990
    },
    {
      "epoch": 4.562368359466417,
      "loss_main": 0.031455326825380325,
      "step": 38990
    },
    {
      "epoch": 4.5635384975427105,
      "grad_norm": 0.21515585482120514,
      "learning_rate": 2.9917752470448813e-05,
      "loss": 0.0484,
      "step": 39000
    },
    {
      "epoch": 4.5635384975427105,
      "loss_main": 0.041484005749225616,
      "step": 39000
    },
    {
      "epoch": 4.564708635619003,
      "grad_norm": 0.23003433644771576,
      "learning_rate": 2.9892518185946495e-05,
      "loss": 0.0474,
      "step": 39010
    },
    {
      "epoch": 4.564708635619003,
      "loss_main": 0.04338812828063965,
      "step": 39010
    },
    {
      "epoch": 4.565878773695296,
      "grad_norm": 0.25984084606170654,
      "learning_rate": 2.986729000956624e-05,
      "loss": 0.0543,
      "step": 39020
    },
    {
      "epoch": 4.565878773695296,
      "loss_main": 0.054902754724025726,
      "step": 39020
    },
    {
      "epoch": 4.567048911771589,
      "grad_norm": 0.17818446457386017,
      "learning_rate": 2.9842067948971736e-05,
      "loss": 0.0565,
      "step": 39030
    },
    {
      "epoch": 4.567048911771589,
      "loss_main": 0.06320241093635559,
      "step": 39030
    },
    {
      "epoch": 4.568219049847882,
      "grad_norm": 0.3082866668701172,
      "learning_rate": 2.9816852011824727e-05,
      "loss": 0.0457,
      "step": 39040
    },
    {
      "epoch": 4.568219049847882,
      "loss_main": 0.02613237500190735,
      "step": 39040
    },
    {
      "epoch": 4.569389187924175,
      "grad_norm": 0.2965254485607147,
      "learning_rate": 2.979164220578519e-05,
      "loss": 0.0615,
      "step": 39050
    },
    {
      "epoch": 4.569389187924175,
      "loss_main": 0.04508603364229202,
      "step": 39050
    },
    {
      "epoch": 4.570559326000468,
      "grad_norm": 0.33417370915412903,
      "learning_rate": 2.9766438538511165e-05,
      "loss": 0.0534,
      "step": 39060
    },
    {
      "epoch": 4.570559326000468,
      "loss_main": 0.05958241969347,
      "step": 39060
    },
    {
      "epoch": 4.571729464076761,
      "grad_norm": 0.2761639952659607,
      "learning_rate": 2.9741241017658873e-05,
      "loss": 0.0475,
      "step": 39070
    },
    {
      "epoch": 4.571729464076761,
      "loss_main": 0.06411975622177124,
      "step": 39070
    },
    {
      "epoch": 4.572899602153054,
      "grad_norm": 0.2931631803512573,
      "learning_rate": 2.971604965088267e-05,
      "loss": 0.0533,
      "step": 39080
    },
    {
      "epoch": 4.572899602153054,
      "loss_main": 0.07859187573194504,
      "step": 39080
    },
    {
      "epoch": 4.574069740229347,
      "grad_norm": 0.3133925497531891,
      "learning_rate": 2.9690864445835008e-05,
      "loss": 0.0549,
      "step": 39090
    },
    {
      "epoch": 4.574069740229347,
      "loss_main": 0.05124404653906822,
      "step": 39090
    },
    {
      "epoch": 4.57523987830564,
      "grad_norm": 0.24964796006679535,
      "learning_rate": 2.966568541016651e-05,
      "loss": 0.0683,
      "step": 39100
    },
    {
      "epoch": 4.57523987830564,
      "loss_main": 0.041290197521448135,
      "step": 39100
    },
    {
      "epoch": 4.576410016381933,
      "grad_norm": 0.26650917530059814,
      "learning_rate": 2.9640512551525867e-05,
      "loss": 0.057,
      "step": 39110
    },
    {
      "epoch": 4.576410016381933,
      "loss_main": 0.051250483840703964,
      "step": 39110
    },
    {
      "epoch": 4.577580154458226,
      "grad_norm": 0.2837759554386139,
      "learning_rate": 2.961534587755995e-05,
      "loss": 0.0526,
      "step": 39120
    },
    {
      "epoch": 4.577580154458226,
      "loss_main": 0.062045637518167496,
      "step": 39120
    },
    {
      "epoch": 4.578750292534519,
      "grad_norm": 0.2845636010169983,
      "learning_rate": 2.959018539591375e-05,
      "loss": 0.0561,
      "step": 39130
    },
    {
      "epoch": 4.578750292534519,
      "loss_main": 0.0636102482676506,
      "step": 39130
    },
    {
      "epoch": 4.579920430610812,
      "grad_norm": 0.23471692204475403,
      "learning_rate": 2.9565031114230325e-05,
      "loss": 0.0587,
      "step": 39140
    },
    {
      "epoch": 4.579920430610812,
      "loss_main": 0.03792927786707878,
      "step": 39140
    },
    {
      "epoch": 4.581090568687105,
      "grad_norm": 0.23243871331214905,
      "learning_rate": 2.9539883040150895e-05,
      "loss": 0.0561,
      "step": 39150
    },
    {
      "epoch": 4.581090568687105,
      "loss_main": 0.054355230182409286,
      "step": 39150
    },
    {
      "epoch": 4.582260706763398,
      "grad_norm": 0.2595846652984619,
      "learning_rate": 2.9514741181314774e-05,
      "loss": 0.0541,
      "step": 39160
    },
    {
      "epoch": 4.582260706763398,
      "loss_main": 0.04909275844693184,
      "step": 39160
    },
    {
      "epoch": 4.583430844839691,
      "grad_norm": 0.2629519999027252,
      "learning_rate": 2.94896055453594e-05,
      "loss": 0.0443,
      "step": 39170
    },
    {
      "epoch": 4.583430844839691,
      "loss_main": 0.02325538732111454,
      "step": 39170
    },
    {
      "epoch": 4.584600982915984,
      "grad_norm": 0.20078888535499573,
      "learning_rate": 2.9464476139920332e-05,
      "loss": 0.0521,
      "step": 39180
    },
    {
      "epoch": 4.584600982915984,
      "loss_main": 0.06310331076383591,
      "step": 39180
    },
    {
      "epoch": 4.585771120992277,
      "grad_norm": 0.34366440773010254,
      "learning_rate": 2.9439352972631186e-05,
      "loss": 0.0495,
      "step": 39190
    },
    {
      "epoch": 4.585771120992277,
      "loss_main": 0.06753836572170258,
      "step": 39190
    },
    {
      "epoch": 4.5869412590685705,
      "grad_norm": 0.24429942667484283,
      "learning_rate": 2.9414236051123757e-05,
      "loss": 0.0504,
      "step": 39200
    },
    {
      "epoch": 4.5869412590685705,
      "loss_main": 0.04886454716324806,
      "step": 39200
    },
    {
      "epoch": 4.588111397144863,
      "grad_norm": 0.24445198476314545,
      "learning_rate": 2.938912538302785e-05,
      "loss": 0.0489,
      "step": 39210
    },
    {
      "epoch": 4.588111397144863,
      "loss_main": 0.05019817128777504,
      "step": 39210
    },
    {
      "epoch": 4.589281535221156,
      "grad_norm": 0.25301286578178406,
      "learning_rate": 2.9364020975971464e-05,
      "loss": 0.0457,
      "step": 39220
    },
    {
      "epoch": 4.589281535221156,
      "loss_main": 0.036182183772325516,
      "step": 39220
    },
    {
      "epoch": 4.590451673297449,
      "grad_norm": 0.20760153234004974,
      "learning_rate": 2.9338922837580657e-05,
      "loss": 0.0476,
      "step": 39230
    },
    {
      "epoch": 4.590451673297449,
      "loss_main": 0.04045012965798378,
      "step": 39230
    },
    {
      "epoch": 4.591621811373742,
      "grad_norm": 0.21388697624206543,
      "learning_rate": 2.931383097547955e-05,
      "loss": 0.0537,
      "step": 39240
    },
    {
      "epoch": 4.591621811373742,
      "loss_main": 0.06340877711772919,
      "step": 39240
    },
    {
      "epoch": 4.592791949450035,
      "grad_norm": 0.3922765552997589,
      "learning_rate": 2.928874539729043e-05,
      "loss": 0.0576,
      "step": 39250
    },
    {
      "epoch": 4.592791949450035,
      "loss_main": 0.059198684990406036,
      "step": 39250
    },
    {
      "epoch": 4.593962087526328,
      "grad_norm": 0.2089555412530899,
      "learning_rate": 2.926366611063358e-05,
      "loss": 0.0532,
      "step": 39260
    },
    {
      "epoch": 4.593962087526328,
      "loss_main": 0.05396348237991333,
      "step": 39260
    },
    {
      "epoch": 4.5951322256026215,
      "grad_norm": 0.266213983297348,
      "learning_rate": 2.9238593123127463e-05,
      "loss": 0.0505,
      "step": 39270
    },
    {
      "epoch": 4.5951322256026215,
      "loss_main": 0.04215877503156662,
      "step": 39270
    },
    {
      "epoch": 4.596302363678914,
      "grad_norm": 0.3458203375339508,
      "learning_rate": 2.9213526442388583e-05,
      "loss": 0.0481,
      "step": 39280
    },
    {
      "epoch": 4.596302363678914,
      "loss_main": 0.05639573559165001,
      "step": 39280
    },
    {
      "epoch": 4.597472501755207,
      "grad_norm": 0.1600142866373062,
      "learning_rate": 2.9188466076031545e-05,
      "loss": 0.0455,
      "step": 39290
    },
    {
      "epoch": 4.597472501755207,
      "loss_main": 0.0404030866920948,
      "step": 39290
    },
    {
      "epoch": 4.5986426398315,
      "grad_norm": 0.25874122977256775,
      "learning_rate": 2.9163412031669012e-05,
      "loss": 0.0526,
      "step": 39300
    },
    {
      "epoch": 4.5986426398315,
      "loss_main": 0.061260052025318146,
      "step": 39300
    },
    {
      "epoch": 4.599812777907793,
      "grad_norm": 0.24371345341205597,
      "learning_rate": 2.913836431691175e-05,
      "loss": 0.0599,
      "step": 39310
    },
    {
      "epoch": 4.599812777907793,
      "loss_main": 0.031578179448843,
      "step": 39310
    },
    {
      "epoch": 4.600982915984086,
      "grad_norm": 0.21950547397136688,
      "learning_rate": 2.9113322939368583e-05,
      "loss": 0.0563,
      "step": 39320
    },
    {
      "epoch": 4.600982915984086,
      "loss_main": 0.04263349995017052,
      "step": 39320
    },
    {
      "epoch": 4.602153054060379,
      "grad_norm": 0.19663579761981964,
      "learning_rate": 2.9088287906646427e-05,
      "loss": 0.0598,
      "step": 39330
    },
    {
      "epoch": 4.602153054060379,
      "loss_main": 0.06309313327074051,
      "step": 39330
    },
    {
      "epoch": 4.603323192136672,
      "grad_norm": 0.4111316204071045,
      "learning_rate": 2.906325922635024e-05,
      "loss": 0.0533,
      "step": 39340
    },
    {
      "epoch": 4.603323192136672,
      "loss_main": 0.03915360942482948,
      "step": 39340
    },
    {
      "epoch": 4.604493330212965,
      "grad_norm": 0.2636898159980774,
      "learning_rate": 2.903823690608313e-05,
      "loss": 0.0432,
      "step": 39350
    },
    {
      "epoch": 4.604493330212965,
      "loss_main": 0.05862211436033249,
      "step": 39350
    },
    {
      "epoch": 4.605663468289258,
      "grad_norm": 0.2764129340648651,
      "learning_rate": 2.9013220953446174e-05,
      "loss": 0.0514,
      "step": 39360
    },
    {
      "epoch": 4.605663468289258,
      "loss_main": 0.04852226749062538,
      "step": 39360
    },
    {
      "epoch": 4.6068336063655515,
      "grad_norm": 0.22162504494190216,
      "learning_rate": 2.8988211376038564e-05,
      "loss": 0.0495,
      "step": 39370
    },
    {
      "epoch": 4.6068336063655515,
      "loss_main": 0.03373220935463905,
      "step": 39370
    },
    {
      "epoch": 4.608003744441844,
      "grad_norm": 0.204848513007164,
      "learning_rate": 2.8963208181457564e-05,
      "loss": 0.0522,
      "step": 39380
    },
    {
      "epoch": 4.608003744441844,
      "loss_main": 0.08333174884319305,
      "step": 39380
    },
    {
      "epoch": 4.609173882518137,
      "grad_norm": 0.2341001331806183,
      "learning_rate": 2.8938211377298453e-05,
      "loss": 0.0672,
      "step": 39390
    },
    {
      "epoch": 4.609173882518137,
      "loss_main": 0.04395139217376709,
      "step": 39390
    },
    {
      "epoch": 4.610344020594431,
      "grad_norm": 0.25140541791915894,
      "learning_rate": 2.8913220971154652e-05,
      "loss": 0.0615,
      "step": 39400
    },
    {
      "epoch": 4.610344020594431,
      "loss_main": 0.06630599498748779,
      "step": 39400
    },
    {
      "epoch": 4.611514158670723,
      "grad_norm": 0.22473680973052979,
      "learning_rate": 2.888823697061753e-05,
      "loss": 0.0505,
      "step": 39410
    },
    {
      "epoch": 4.611514158670723,
      "loss_main": 0.06396061182022095,
      "step": 39410
    },
    {
      "epoch": 4.612684296747016,
      "grad_norm": 0.290410578250885,
      "learning_rate": 2.8863259383276618e-05,
      "loss": 0.0606,
      "step": 39420
    },
    {
      "epoch": 4.612684296747016,
      "loss_main": 0.04189074784517288,
      "step": 39420
    },
    {
      "epoch": 4.613854434823309,
      "grad_norm": 0.20825444161891937,
      "learning_rate": 2.8838288216719395e-05,
      "loss": 0.0591,
      "step": 39430
    },
    {
      "epoch": 4.613854434823309,
      "loss_main": 0.06525983661413193,
      "step": 39430
    },
    {
      "epoch": 4.615024572899602,
      "grad_norm": 0.1954873502254486,
      "learning_rate": 2.8813323478531484e-05,
      "loss": 0.05,
      "step": 39440
    },
    {
      "epoch": 4.615024572899602,
      "loss_main": 0.04983227699995041,
      "step": 39440
    },
    {
      "epoch": 4.616194710975895,
      "grad_norm": 0.297748863697052,
      "learning_rate": 2.8788365176296496e-05,
      "loss": 0.0609,
      "step": 39450
    },
    {
      "epoch": 4.616194710975895,
      "loss_main": 0.05946802720427513,
      "step": 39450
    },
    {
      "epoch": 4.617364849052188,
      "grad_norm": 0.22365708649158478,
      "learning_rate": 2.876341331759611e-05,
      "loss": 0.0507,
      "step": 39460
    },
    {
      "epoch": 4.617364849052188,
      "loss_main": 0.0549718514084816,
      "step": 39460
    },
    {
      "epoch": 4.6185349871284815,
      "grad_norm": 0.43092185258865356,
      "learning_rate": 2.8738467910010036e-05,
      "loss": 0.0558,
      "step": 39470
    },
    {
      "epoch": 4.6185349871284815,
      "loss_main": 0.0784309059381485,
      "step": 39470
    },
    {
      "epoch": 4.619705125204774,
      "grad_norm": 0.2487635463476181,
      "learning_rate": 2.8713528961116032e-05,
      "loss": 0.0515,
      "step": 39480
    },
    {
      "epoch": 4.619705125204774,
      "loss_main": 0.04996823891997337,
      "step": 39480
    },
    {
      "epoch": 4.620875263281067,
      "grad_norm": 0.21421225368976593,
      "learning_rate": 2.8688596478489875e-05,
      "loss": 0.0616,
      "step": 39490
    },
    {
      "epoch": 4.620875263281067,
      "loss_main": 0.05964232608675957,
      "step": 39490
    },
    {
      "epoch": 4.62204540135736,
      "grad_norm": 0.25699758529663086,
      "learning_rate": 2.8663670469705434e-05,
      "loss": 0.0503,
      "step": 39500
    },
    {
      "epoch": 4.62204540135736,
      "loss_main": 0.09663919359445572,
      "step": 39500
    },
    {
      "epoch": 4.623215539433653,
      "grad_norm": 0.18652765452861786,
      "learning_rate": 2.8638750942334546e-05,
      "loss": 0.0574,
      "step": 39510
    },
    {
      "epoch": 4.623215539433653,
      "loss_main": 0.04955441132187843,
      "step": 39510
    },
    {
      "epoch": 4.624385677509946,
      "grad_norm": 0.2187562882900238,
      "learning_rate": 2.8613837903947115e-05,
      "loss": 0.0457,
      "step": 39520
    },
    {
      "epoch": 4.624385677509946,
      "loss_main": 0.11253325641155243,
      "step": 39520
    },
    {
      "epoch": 4.625555815586239,
      "grad_norm": 0.2264048010110855,
      "learning_rate": 2.858893136211106e-05,
      "loss": 0.0667,
      "step": 39530
    },
    {
      "epoch": 4.625555815586239,
      "loss_main": 0.04897085204720497,
      "step": 39530
    },
    {
      "epoch": 4.6267259536625325,
      "grad_norm": 0.4671773314476013,
      "learning_rate": 2.8564031324392315e-05,
      "loss": 0.0529,
      "step": 39540
    },
    {
      "epoch": 4.6267259536625325,
      "loss_main": 0.031022850424051285,
      "step": 39540
    },
    {
      "epoch": 4.627896091738825,
      "grad_norm": 0.2579321265220642,
      "learning_rate": 2.85391377983549e-05,
      "loss": 0.0596,
      "step": 39550
    },
    {
      "epoch": 4.627896091738825,
      "loss_main": 0.037955112755298615,
      "step": 39550
    },
    {
      "epoch": 4.629066229815118,
      "grad_norm": 0.29742953181266785,
      "learning_rate": 2.851425079156075e-05,
      "loss": 0.0514,
      "step": 39560
    },
    {
      "epoch": 4.629066229815118,
      "loss_main": 0.06104283779859543,
      "step": 39560
    },
    {
      "epoch": 4.630236367891412,
      "grad_norm": 0.25557801127433777,
      "learning_rate": 2.848937031156994e-05,
      "loss": 0.0549,
      "step": 39570
    },
    {
      "epoch": 4.630236367891412,
      "loss_main": 0.038204874843358994,
      "step": 39570
    },
    {
      "epoch": 4.631406505967704,
      "grad_norm": 0.2540440261363983,
      "learning_rate": 2.846449636594044e-05,
      "loss": 0.0606,
      "step": 39580
    },
    {
      "epoch": 4.631406505967704,
      "loss_main": 0.05533618852496147,
      "step": 39580
    },
    {
      "epoch": 4.632576644043997,
      "grad_norm": 0.34296879172325134,
      "learning_rate": 2.843962896222836e-05,
      "loss": 0.0519,
      "step": 39590
    },
    {
      "epoch": 4.632576644043997,
      "loss_main": 0.04320007562637329,
      "step": 39590
    },
    {
      "epoch": 4.633746782120291,
      "grad_norm": 0.24604365229606628,
      "learning_rate": 2.8414768107987722e-05,
      "loss": 0.0541,
      "step": 39600
    },
    {
      "epoch": 4.633746782120291,
      "loss_main": 0.07402829825878143,
      "step": 39600
    },
    {
      "epoch": 4.634916920196583,
      "grad_norm": 0.2831360399723053,
      "learning_rate": 2.838991381077061e-05,
      "loss": 0.0604,
      "step": 39610
    },
    {
      "epoch": 4.634916920196583,
      "loss_main": 0.05105513706803322,
      "step": 39610
    },
    {
      "epoch": 4.636087058272876,
      "grad_norm": 0.20862643420696259,
      "learning_rate": 2.83650660781271e-05,
      "loss": 0.0526,
      "step": 39620
    },
    {
      "epoch": 4.636087058272876,
      "loss_main": 0.04619726911187172,
      "step": 39620
    },
    {
      "epoch": 4.637257196349169,
      "grad_norm": 0.33830997347831726,
      "learning_rate": 2.8340224917605285e-05,
      "loss": 0.0496,
      "step": 39630
    },
    {
      "epoch": 4.637257196349169,
      "loss_main": 0.05353051424026489,
      "step": 39630
    },
    {
      "epoch": 4.6384273344254625,
      "grad_norm": 0.276129812002182,
      "learning_rate": 2.831539033675122e-05,
      "loss": 0.056,
      "step": 39640
    },
    {
      "epoch": 4.6384273344254625,
      "loss_main": 0.05995205417275429,
      "step": 39640
    },
    {
      "epoch": 4.639597472501755,
      "grad_norm": 0.24823589622974396,
      "learning_rate": 2.8290562343109038e-05,
      "loss": 0.064,
      "step": 39650
    },
    {
      "epoch": 4.639597472501755,
      "loss_main": 0.06256646662950516,
      "step": 39650
    },
    {
      "epoch": 4.640767610578048,
      "grad_norm": 0.2637379467487335,
      "learning_rate": 2.826574094422082e-05,
      "loss": 0.0467,
      "step": 39660
    },
    {
      "epoch": 4.640767610578048,
      "loss_main": 0.03875437751412392,
      "step": 39660
    },
    {
      "epoch": 4.641937748654341,
      "grad_norm": 0.29851898550987244,
      "learning_rate": 2.8240926147626645e-05,
      "loss": 0.0624,
      "step": 39670
    },
    {
      "epoch": 4.641937748654341,
      "loss_main": 0.02645573951303959,
      "step": 39670
    },
    {
      "epoch": 4.643107886730634,
      "grad_norm": 0.2352312207221985,
      "learning_rate": 2.8216117960864586e-05,
      "loss": 0.0524,
      "step": 39680
    },
    {
      "epoch": 4.643107886730634,
      "loss_main": 0.04899277165532112,
      "step": 39680
    },
    {
      "epoch": 4.644278024806927,
      "grad_norm": 0.27125129103660583,
      "learning_rate": 2.8191316391470703e-05,
      "loss": 0.0538,
      "step": 39690
    },
    {
      "epoch": 4.644278024806927,
      "loss_main": 0.043207842856645584,
      "step": 39690
    },
    {
      "epoch": 4.64544816288322,
      "grad_norm": 0.23241735994815826,
      "learning_rate": 2.816652144697911e-05,
      "loss": 0.0638,
      "step": 39700
    },
    {
      "epoch": 4.64544816288322,
      "loss_main": 0.034232597798109055,
      "step": 39700
    },
    {
      "epoch": 4.6466183009595134,
      "grad_norm": 0.2031330019235611,
      "learning_rate": 2.8141733134921783e-05,
      "loss": 0.061,
      "step": 39710
    },
    {
      "epoch": 4.6466183009595134,
      "loss_main": 0.027697911486029625,
      "step": 39710
    },
    {
      "epoch": 4.647788439035806,
      "grad_norm": 0.26984715461730957,
      "learning_rate": 2.811695146282884e-05,
      "loss": 0.0512,
      "step": 39720
    },
    {
      "epoch": 4.647788439035806,
      "loss_main": 0.032327983528375626,
      "step": 39720
    },
    {
      "epoch": 4.648958577112099,
      "grad_norm": 0.27632513642311096,
      "learning_rate": 2.8092176438228212e-05,
      "loss": 0.0511,
      "step": 39730
    },
    {
      "epoch": 4.648958577112099,
      "loss_main": 0.040838249027729034,
      "step": 39730
    },
    {
      "epoch": 4.6501287151883925,
      "grad_norm": 0.2580663859844208,
      "learning_rate": 2.806740806864598e-05,
      "loss": 0.0583,
      "step": 39740
    },
    {
      "epoch": 4.6501287151883925,
      "loss_main": 0.0874943733215332,
      "step": 39740
    },
    {
      "epoch": 4.651298853264685,
      "grad_norm": 0.23793329298496246,
      "learning_rate": 2.804264636160604e-05,
      "loss": 0.0529,
      "step": 39750
    },
    {
      "epoch": 4.651298853264685,
      "loss_main": 0.09032626450061798,
      "step": 39750
    },
    {
      "epoch": 4.652468991340978,
      "grad_norm": 0.22978509962558746,
      "learning_rate": 2.8017891324630402e-05,
      "loss": 0.0569,
      "step": 39760
    },
    {
      "epoch": 4.652468991340978,
      "loss_main": 0.07211071252822876,
      "step": 39760
    },
    {
      "epoch": 4.653639129417272,
      "grad_norm": 0.29188141226768494,
      "learning_rate": 2.7993142965238976e-05,
      "loss": 0.0523,
      "step": 39770
    },
    {
      "epoch": 4.653639129417272,
      "loss_main": 0.03958362340927124,
      "step": 39770
    },
    {
      "epoch": 4.654809267493564,
      "grad_norm": 0.22225651144981384,
      "learning_rate": 2.7968401290949665e-05,
      "loss": 0.055,
      "step": 39780
    },
    {
      "epoch": 4.654809267493564,
      "loss_main": 0.07236868143081665,
      "step": 39780
    },
    {
      "epoch": 4.655979405569857,
      "grad_norm": 0.302720844745636,
      "learning_rate": 2.7943666309278328e-05,
      "loss": 0.0511,
      "step": 39790
    },
    {
      "epoch": 4.655979405569857,
      "loss_main": 0.062157656997442245,
      "step": 39790
    },
    {
      "epoch": 4.65714954364615,
      "grad_norm": 0.17471009492874146,
      "learning_rate": 2.7918938027738783e-05,
      "loss": 0.0536,
      "step": 39800
    },
    {
      "epoch": 4.65714954364615,
      "loss_main": 0.04062335938215256,
      "step": 39800
    },
    {
      "epoch": 4.6583196817224435,
      "grad_norm": 0.25658294558525085,
      "learning_rate": 2.789421645384287e-05,
      "loss": 0.0475,
      "step": 39810
    },
    {
      "epoch": 4.6583196817224435,
      "loss_main": 0.12524345517158508,
      "step": 39810
    },
    {
      "epoch": 4.659489819798736,
      "grad_norm": 0.24423383176326752,
      "learning_rate": 2.786950159510032e-05,
      "loss": 0.0547,
      "step": 39820
    },
    {
      "epoch": 4.659489819798736,
      "loss_main": 0.032876044511795044,
      "step": 39820
    },
    {
      "epoch": 4.660659957875029,
      "grad_norm": 0.30468568205833435,
      "learning_rate": 2.7844793459018876e-05,
      "loss": 0.0528,
      "step": 39830
    },
    {
      "epoch": 4.660659957875029,
      "loss_main": 0.047162026166915894,
      "step": 39830
    },
    {
      "epoch": 4.661830095951323,
      "grad_norm": 0.3224888741970062,
      "learning_rate": 2.7820092053104195e-05,
      "loss": 0.0533,
      "step": 39840
    },
    {
      "epoch": 4.661830095951323,
      "loss_main": 0.04972979053854942,
      "step": 39840
    },
    {
      "epoch": 4.663000234027615,
      "grad_norm": 0.1914033144712448,
      "learning_rate": 2.7795397384859933e-05,
      "loss": 0.0506,
      "step": 39850
    },
    {
      "epoch": 4.663000234027615,
      "loss_main": 0.061032865196466446,
      "step": 39850
    },
    {
      "epoch": 4.664170372103908,
      "grad_norm": 0.23137818276882172,
      "learning_rate": 2.7770709461787638e-05,
      "loss": 0.0516,
      "step": 39860
    },
    {
      "epoch": 4.664170372103908,
      "loss_main": 0.033507004380226135,
      "step": 39860
    },
    {
      "epoch": 4.665340510180201,
      "grad_norm": 0.227654829621315,
      "learning_rate": 2.7746028291386915e-05,
      "loss": 0.0445,
      "step": 39870
    },
    {
      "epoch": 4.665340510180201,
      "loss_main": 0.03889324143528938,
      "step": 39870
    },
    {
      "epoch": 4.666510648256494,
      "grad_norm": 0.21852873265743256,
      "learning_rate": 2.772135388115519e-05,
      "loss": 0.0525,
      "step": 39880
    },
    {
      "epoch": 4.666510648256494,
      "loss_main": 0.0475710965692997,
      "step": 39880
    },
    {
      "epoch": 4.667680786332787,
      "grad_norm": 0.17736667394638062,
      "learning_rate": 2.7696686238587945e-05,
      "loss": 0.0444,
      "step": 39890
    },
    {
      "epoch": 4.667680786332787,
      "loss_main": 0.046272438019514084,
      "step": 39890
    },
    {
      "epoch": 4.66885092440908,
      "grad_norm": 0.23760758340358734,
      "learning_rate": 2.7672025371178505e-05,
      "loss": 0.0572,
      "step": 39900
    },
    {
      "epoch": 4.66885092440908,
      "loss_main": 0.05930901691317558,
      "step": 39900
    },
    {
      "epoch": 4.6700210624853735,
      "grad_norm": 0.3075883686542511,
      "learning_rate": 2.7647371286418238e-05,
      "loss": 0.0515,
      "step": 39910
    },
    {
      "epoch": 4.6700210624853735,
      "loss_main": 0.029948486015200615,
      "step": 39910
    },
    {
      "epoch": 4.671191200561666,
      "grad_norm": 0.3359912037849426,
      "learning_rate": 2.762272399179639e-05,
      "loss": 0.0546,
      "step": 39920
    },
    {
      "epoch": 4.671191200561666,
      "loss_main": 0.05988746136426926,
      "step": 39920
    },
    {
      "epoch": 4.672361338637959,
      "grad_norm": 0.27447301149368286,
      "learning_rate": 2.7598083494800154e-05,
      "loss": 0.0571,
      "step": 39930
    },
    {
      "epoch": 4.672361338637959,
      "loss_main": 0.05944044142961502,
      "step": 39930
    },
    {
      "epoch": 4.673531476714253,
      "grad_norm": 0.4127306640148163,
      "learning_rate": 2.7573449802914664e-05,
      "loss": 0.0588,
      "step": 39940
    },
    {
      "epoch": 4.673531476714253,
      "loss_main": 0.04787472262978554,
      "step": 39940
    },
    {
      "epoch": 4.674701614790545,
      "grad_norm": 0.2303021103143692,
      "learning_rate": 2.7548822923622964e-05,
      "loss": 0.0466,
      "step": 39950
    },
    {
      "epoch": 4.674701614790545,
      "loss_main": 0.041498392820358276,
      "step": 39950
    },
    {
      "epoch": 4.675871752866838,
      "grad_norm": 0.1912536472082138,
      "learning_rate": 2.752420286440609e-05,
      "loss": 0.0515,
      "step": 39960
    },
    {
      "epoch": 4.675871752866838,
      "loss_main": 0.04396182298660278,
      "step": 39960
    },
    {
      "epoch": 4.677041890943132,
      "grad_norm": 0.206567645072937,
      "learning_rate": 2.749958963274295e-05,
      "loss": 0.0536,
      "step": 39970
    },
    {
      "epoch": 4.677041890943132,
      "loss_main": 0.034694310277700424,
      "step": 39970
    },
    {
      "epoch": 4.6782120290194245,
      "grad_norm": 0.2701260447502136,
      "learning_rate": 2.747498323611039e-05,
      "loss": 0.053,
      "step": 39980
    },
    {
      "epoch": 4.6782120290194245,
      "loss_main": 0.028657495975494385,
      "step": 39980
    },
    {
      "epoch": 4.679382167095717,
      "grad_norm": 0.1815570443868637,
      "learning_rate": 2.7450383681983184e-05,
      "loss": 0.0503,
      "step": 39990
    },
    {
      "epoch": 4.679382167095717,
      "loss_main": 0.04884234070777893,
      "step": 39990
    },
    {
      "epoch": 4.68055230517201,
      "grad_norm": 0.25517573952674866,
      "learning_rate": 2.742579097783403e-05,
      "loss": 0.0515,
      "step": 40000
    },
    {
      "epoch": 4.68055230517201,
      "loss_main": 0.02322455681860447,
      "step": 40000
    },
    {
      "epoch": 4.6817224432483036,
      "grad_norm": 0.2288791686296463,
      "learning_rate": 2.7401205131133512e-05,
      "loss": 0.0534,
      "step": 40010
    },
    {
      "epoch": 4.6817224432483036,
      "loss_main": 0.04807059094309807,
      "step": 40010
    },
    {
      "epoch": 4.682892581324596,
      "grad_norm": 0.2158304750919342,
      "learning_rate": 2.7376626149350238e-05,
      "loss": 0.0476,
      "step": 40020
    },
    {
      "epoch": 4.682892581324596,
      "loss_main": 0.05299115180969238,
      "step": 40020
    },
    {
      "epoch": 4.684062719400889,
      "grad_norm": 0.21239794790744781,
      "learning_rate": 2.735205403995056e-05,
      "loss": 0.0538,
      "step": 40030
    },
    {
      "epoch": 4.684062719400889,
      "loss_main": 0.03395563364028931,
      "step": 40030
    },
    {
      "epoch": 4.685232857477183,
      "grad_norm": 0.1614907830953598,
      "learning_rate": 2.7327488810398917e-05,
      "loss": 0.051,
      "step": 40040
    },
    {
      "epoch": 4.685232857477183,
      "loss_main": 0.06374906748533249,
      "step": 40040
    },
    {
      "epoch": 4.686402995553475,
      "grad_norm": 0.2207844853401184,
      "learning_rate": 2.7302930468157507e-05,
      "loss": 0.0545,
      "step": 40050
    },
    {
      "epoch": 4.686402995553475,
      "loss_main": 0.05533643439412117,
      "step": 40050
    },
    {
      "epoch": 4.687573133629768,
      "grad_norm": 0.20461603999137878,
      "learning_rate": 2.727837902068655e-05,
      "loss": 0.0503,
      "step": 40060
    },
    {
      "epoch": 4.687573133629768,
      "loss_main": 0.052726466208696365,
      "step": 40060
    },
    {
      "epoch": 4.688743271706061,
      "grad_norm": 0.16474291682243347,
      "learning_rate": 2.7253834475444123e-05,
      "loss": 0.051,
      "step": 40070
    },
    {
      "epoch": 4.688743271706061,
      "loss_main": 0.04463529586791992,
      "step": 40070
    },
    {
      "epoch": 4.6899134097823545,
      "grad_norm": 0.19542774558067322,
      "learning_rate": 2.7229296839886204e-05,
      "loss": 0.0562,
      "step": 40080
    },
    {
      "epoch": 4.6899134097823545,
      "loss_main": 0.05863025411963463,
      "step": 40080
    },
    {
      "epoch": 4.691083547858647,
      "grad_norm": 0.24198263883590698,
      "learning_rate": 2.720476612146668e-05,
      "loss": 0.0511,
      "step": 40090
    },
    {
      "epoch": 4.691083547858647,
      "loss_main": 0.0451255738735199,
      "step": 40090
    },
    {
      "epoch": 4.69225368593494,
      "grad_norm": 0.20319516956806183,
      "learning_rate": 2.7180242327637317e-05,
      "loss": 0.0529,
      "step": 40100
    },
    {
      "epoch": 4.69225368593494,
      "loss_main": 0.044884443283081055,
      "step": 40100
    },
    {
      "epoch": 4.693423824011234,
      "grad_norm": 0.296874463558197,
      "learning_rate": 2.7155725465847826e-05,
      "loss": 0.0537,
      "step": 40110
    },
    {
      "epoch": 4.693423824011234,
      "loss_main": 0.06642811745405197,
      "step": 40110
    },
    {
      "epoch": 4.694593962087526,
      "grad_norm": 0.26912009716033936,
      "learning_rate": 2.713121554354578e-05,
      "loss": 0.0526,
      "step": 40120
    },
    {
      "epoch": 4.694593962087526,
      "loss_main": 0.022590667009353638,
      "step": 40120
    },
    {
      "epoch": 4.695764100163819,
      "grad_norm": 0.2381860762834549,
      "learning_rate": 2.7106712568176628e-05,
      "loss": 0.0542,
      "step": 40130
    },
    {
      "epoch": 4.695764100163819,
      "loss_main": 0.03619720786809921,
      "step": 40130
    },
    {
      "epoch": 4.696934238240113,
      "grad_norm": 0.25803402066230774,
      "learning_rate": 2.708221654718374e-05,
      "loss": 0.0524,
      "step": 40140
    },
    {
      "epoch": 4.696934238240113,
      "loss_main": 0.0674232691526413,
      "step": 40140
    },
    {
      "epoch": 4.698104376316405,
      "grad_norm": 0.22182907164096832,
      "learning_rate": 2.7057727488008357e-05,
      "loss": 0.0553,
      "step": 40150
    },
    {
      "epoch": 4.698104376316405,
      "loss_main": 0.048493485897779465,
      "step": 40150
    },
    {
      "epoch": 4.699274514392698,
      "grad_norm": 0.30704614520072937,
      "learning_rate": 2.703324539808961e-05,
      "loss": 0.0486,
      "step": 40160
    },
    {
      "epoch": 4.699274514392698,
      "loss_main": 0.033909622579813004,
      "step": 40160
    },
    {
      "epoch": 4.700444652468992,
      "grad_norm": 0.2688668966293335,
      "learning_rate": 2.7008770284864505e-05,
      "loss": 0.0473,
      "step": 40170
    },
    {
      "epoch": 4.700444652468992,
      "loss_main": 0.06906640529632568,
      "step": 40170
    },
    {
      "epoch": 4.7016147905452845,
      "grad_norm": 0.27308639883995056,
      "learning_rate": 2.6984302155767916e-05,
      "loss": 0.0527,
      "step": 40180
    },
    {
      "epoch": 4.7016147905452845,
      "loss_main": 0.0529121570289135,
      "step": 40180
    },
    {
      "epoch": 4.702784928621577,
      "grad_norm": 0.24907813966274261,
      "learning_rate": 2.6959841018232683e-05,
      "loss": 0.056,
      "step": 40190
    },
    {
      "epoch": 4.702784928621577,
      "loss_main": 0.03603914752602577,
      "step": 40190
    },
    {
      "epoch": 4.70395506669787,
      "grad_norm": 0.2387600690126419,
      "learning_rate": 2.693538687968937e-05,
      "loss": 0.0522,
      "step": 40200
    },
    {
      "epoch": 4.70395506669787,
      "loss_main": 0.04086820408701897,
      "step": 40200
    },
    {
      "epoch": 4.705125204774164,
      "grad_norm": 0.2896921932697296,
      "learning_rate": 2.6910939747566556e-05,
      "loss": 0.0581,
      "step": 40210
    },
    {
      "epoch": 4.705125204774164,
      "loss_main": 0.05775948241353035,
      "step": 40210
    },
    {
      "epoch": 4.706295342850456,
      "grad_norm": 0.32260259985923767,
      "learning_rate": 2.6886499629290607e-05,
      "loss": 0.0468,
      "step": 40220
    },
    {
      "epoch": 4.706295342850456,
      "loss_main": 0.042970240116119385,
      "step": 40220
    },
    {
      "epoch": 4.707465480926749,
      "grad_norm": 0.18971587717533112,
      "learning_rate": 2.6862066532285802e-05,
      "loss": 0.049,
      "step": 40230
    },
    {
      "epoch": 4.707465480926749,
      "loss_main": 0.0535106435418129,
      "step": 40230
    },
    {
      "epoch": 4.708635619003043,
      "grad_norm": 0.28667593002319336,
      "learning_rate": 2.6837640463974262e-05,
      "loss": 0.0525,
      "step": 40240
    },
    {
      "epoch": 4.708635619003043,
      "loss_main": 0.03407905623316765,
      "step": 40240
    },
    {
      "epoch": 4.7098057570793355,
      "grad_norm": 0.27768656611442566,
      "learning_rate": 2.681322143177596e-05,
      "loss": 0.045,
      "step": 40250
    },
    {
      "epoch": 4.7098057570793355,
      "loss_main": 0.05269347503781319,
      "step": 40250
    },
    {
      "epoch": 4.710975895155628,
      "grad_norm": 0.18809738755226135,
      "learning_rate": 2.678880944310882e-05,
      "loss": 0.0551,
      "step": 40260
    },
    {
      "epoch": 4.710975895155628,
      "loss_main": 0.0521133728325367,
      "step": 40260
    },
    {
      "epoch": 4.712146033231921,
      "grad_norm": 0.23455378413200378,
      "learning_rate": 2.6764404505388474e-05,
      "loss": 0.0511,
      "step": 40270
    },
    {
      "epoch": 4.712146033231921,
      "loss_main": 0.058607615530490875,
      "step": 40270
    },
    {
      "epoch": 4.713316171308215,
      "grad_norm": 0.31403160095214844,
      "learning_rate": 2.6740006626028558e-05,
      "loss": 0.057,
      "step": 40280
    },
    {
      "epoch": 4.713316171308215,
      "loss_main": 0.048557430505752563,
      "step": 40280
    },
    {
      "epoch": 4.714486309384507,
      "grad_norm": 0.2561230957508087,
      "learning_rate": 2.671561581244048e-05,
      "loss": 0.0502,
      "step": 40290
    },
    {
      "epoch": 4.714486309384507,
      "loss_main": 0.06624091416597366,
      "step": 40290
    },
    {
      "epoch": 4.7156564474608,
      "grad_norm": 0.2563839256763458,
      "learning_rate": 2.6691232072033536e-05,
      "loss": 0.0537,
      "step": 40300
    },
    {
      "epoch": 4.7156564474608,
      "loss_main": 0.07233178615570068,
      "step": 40300
    },
    {
      "epoch": 4.716826585537094,
      "grad_norm": 0.32202646136283875,
      "learning_rate": 2.6666855412214852e-05,
      "loss": 0.0595,
      "step": 40310
    },
    {
      "epoch": 4.716826585537094,
      "loss_main": 0.040334880352020264,
      "step": 40310
    },
    {
      "epoch": 4.717996723613386,
      "grad_norm": 0.3288048505783081,
      "learning_rate": 2.664248584038942e-05,
      "loss": 0.0522,
      "step": 40320
    },
    {
      "epoch": 4.717996723613386,
      "loss_main": 0.046123795211315155,
      "step": 40320
    },
    {
      "epoch": 4.719166861689679,
      "grad_norm": 0.25819680094718933,
      "learning_rate": 2.6618123363960047e-05,
      "loss": 0.0475,
      "step": 40330
    },
    {
      "epoch": 4.719166861689679,
      "loss_main": 0.06864288449287415,
      "step": 40330
    },
    {
      "epoch": 4.720336999765973,
      "grad_norm": 0.2767895460128784,
      "learning_rate": 2.659376799032748e-05,
      "loss": 0.0541,
      "step": 40340
    },
    {
      "epoch": 4.720336999765973,
      "loss_main": 0.044501516968011856,
      "step": 40340
    },
    {
      "epoch": 4.7215071378422655,
      "grad_norm": 0.18428058922290802,
      "learning_rate": 2.6569419726890145e-05,
      "loss": 0.0526,
      "step": 40350
    },
    {
      "epoch": 4.7215071378422655,
      "loss_main": 0.0770992636680603,
      "step": 40350
    },
    {
      "epoch": 4.722677275918558,
      "grad_norm": 0.29889026284217834,
      "learning_rate": 2.654507858104447e-05,
      "loss": 0.0522,
      "step": 40360
    },
    {
      "epoch": 4.722677275918558,
      "loss_main": 0.03221188485622406,
      "step": 40360
    },
    {
      "epoch": 4.723847413994852,
      "grad_norm": 0.33068379759788513,
      "learning_rate": 2.652074456018463e-05,
      "loss": 0.0514,
      "step": 40370
    },
    {
      "epoch": 4.723847413994852,
      "loss_main": 0.04975450038909912,
      "step": 40370
    },
    {
      "epoch": 4.725017552071145,
      "grad_norm": 0.26652973890304565,
      "learning_rate": 2.6496417671702646e-05,
      "loss": 0.0506,
      "step": 40380
    },
    {
      "epoch": 4.725017552071145,
      "loss_main": 0.04452968016266823,
      "step": 40380
    },
    {
      "epoch": 4.726187690147437,
      "grad_norm": 0.24549639225006104,
      "learning_rate": 2.6472097922988427e-05,
      "loss": 0.0458,
      "step": 40390
    },
    {
      "epoch": 4.726187690147437,
      "loss_main": 0.1007562205195427,
      "step": 40390
    },
    {
      "epoch": 4.72735782822373,
      "grad_norm": 0.23730362951755524,
      "learning_rate": 2.6447785321429607e-05,
      "loss": 0.0738,
      "step": 40400
    },
    {
      "epoch": 4.72735782822373,
      "loss_main": 0.040317732840776443,
      "step": 40400
    },
    {
      "epoch": 4.728527966300024,
      "grad_norm": 0.37085118889808655,
      "learning_rate": 2.6423479874411784e-05,
      "loss": 0.0569,
      "step": 40410
    },
    {
      "epoch": 4.728527966300024,
      "loss_main": 0.061579082161188126,
      "step": 40410
    },
    {
      "epoch": 4.729698104376316,
      "grad_norm": 0.2636024057865143,
      "learning_rate": 2.6399181589318234e-05,
      "loss": 0.0495,
      "step": 40420
    },
    {
      "epoch": 4.729698104376316,
      "loss_main": 0.039398662745952606,
      "step": 40420
    },
    {
      "epoch": 4.730868242452609,
      "grad_norm": 0.24100129306316376,
      "learning_rate": 2.6374890473530188e-05,
      "loss": 0.0497,
      "step": 40430
    },
    {
      "epoch": 4.730868242452609,
      "loss_main": 0.042976610362529755,
      "step": 40430
    },
    {
      "epoch": 4.732038380528903,
      "grad_norm": 0.1973344385623932,
      "learning_rate": 2.635060653442664e-05,
      "loss": 0.0517,
      "step": 40440
    },
    {
      "epoch": 4.732038380528903,
      "loss_main": 0.04304395988583565,
      "step": 40440
    },
    {
      "epoch": 4.7332085186051955,
      "grad_norm": 0.2473161220550537,
      "learning_rate": 2.6326329779384395e-05,
      "loss": 0.0489,
      "step": 40450
    },
    {
      "epoch": 4.7332085186051955,
      "loss_main": 0.04323211684823036,
      "step": 40450
    },
    {
      "epoch": 4.734378656681488,
      "grad_norm": 0.3074380159378052,
      "learning_rate": 2.63020602157781e-05,
      "loss": 0.0488,
      "step": 40460
    },
    {
      "epoch": 4.734378656681488,
      "loss_main": 0.02695813961327076,
      "step": 40460
    },
    {
      "epoch": 4.735548794757781,
      "grad_norm": 0.2912692427635193,
      "learning_rate": 2.62777978509802e-05,
      "loss": 0.0524,
      "step": 40470
    },
    {
      "epoch": 4.735548794757781,
      "loss_main": 0.06650283187627792,
      "step": 40470
    },
    {
      "epoch": 4.736718932834075,
      "grad_norm": 0.20315633714199066,
      "learning_rate": 2.6253542692360954e-05,
      "loss": 0.0534,
      "step": 40480
    },
    {
      "epoch": 4.736718932834075,
      "loss_main": 0.029782718047499657,
      "step": 40480
    },
    {
      "epoch": 4.737889070910367,
      "grad_norm": 0.18145079910755157,
      "learning_rate": 2.6229294747288458e-05,
      "loss": 0.0493,
      "step": 40490
    },
    {
      "epoch": 4.737889070910367,
      "loss_main": 0.0727214440703392,
      "step": 40490
    },
    {
      "epoch": 4.73905920898666,
      "grad_norm": 0.21957357227802277,
      "learning_rate": 2.6205054023128596e-05,
      "loss": 0.0484,
      "step": 40500
    },
    {
      "epoch": 4.73905920898666,
      "loss_main": 0.08233050256967545,
      "step": 40500
    },
    {
      "epoch": 4.740229347062954,
      "grad_norm": 0.2636874318122864,
      "learning_rate": 2.6180820527245043e-05,
      "loss": 0.057,
      "step": 40510
    },
    {
      "epoch": 4.740229347062954,
      "loss_main": 0.048150401562452316,
      "step": 40510
    },
    {
      "epoch": 4.7413994851392465,
      "grad_norm": 0.23373854160308838,
      "learning_rate": 2.6156594266999313e-05,
      "loss": 0.0503,
      "step": 40520
    },
    {
      "epoch": 4.7413994851392465,
      "loss_main": 0.0513739176094532,
      "step": 40520
    },
    {
      "epoch": 4.742569623215539,
      "grad_norm": 0.2973554730415344,
      "learning_rate": 2.6132375249750672e-05,
      "loss": 0.0601,
      "step": 40530
    },
    {
      "epoch": 4.742569623215539,
      "loss_main": 0.03118436224758625,
      "step": 40530
    },
    {
      "epoch": 4.743739761291833,
      "grad_norm": 0.22105416655540466,
      "learning_rate": 2.6108163482856286e-05,
      "loss": 0.0492,
      "step": 40540
    },
    {
      "epoch": 4.743739761291833,
      "loss_main": 0.0289128590375185,
      "step": 40540
    },
    {
      "epoch": 4.744909899368126,
      "grad_norm": 0.21134862303733826,
      "learning_rate": 2.6083958973670964e-05,
      "loss": 0.0455,
      "step": 40550
    },
    {
      "epoch": 4.744909899368126,
      "loss_main": 0.04933999851346016,
      "step": 40550
    },
    {
      "epoch": 4.746080037444418,
      "grad_norm": 0.2853824198246002,
      "learning_rate": 2.6059761729547483e-05,
      "loss": 0.053,
      "step": 40560
    },
    {
      "epoch": 4.746080037444418,
      "loss_main": 0.03861282393336296,
      "step": 40560
    },
    {
      "epoch": 4.747250175520712,
      "grad_norm": 0.2777717709541321,
      "learning_rate": 2.603557175783624e-05,
      "loss": 0.0435,
      "step": 40570
    },
    {
      "epoch": 4.747250175520712,
      "loss_main": 0.06013280898332596,
      "step": 40570
    },
    {
      "epoch": 4.748420313597005,
      "grad_norm": 0.15434639155864716,
      "learning_rate": 2.601138906588559e-05,
      "loss": 0.0518,
      "step": 40580
    },
    {
      "epoch": 4.748420313597005,
      "loss_main": 0.07109826058149338,
      "step": 40580
    },
    {
      "epoch": 4.749590451673297,
      "grad_norm": 0.289641797542572,
      "learning_rate": 2.598721366104152e-05,
      "loss": 0.0549,
      "step": 40590
    },
    {
      "epoch": 4.749590451673297,
      "loss_main": 0.046605657786130905,
      "step": 40590
    },
    {
      "epoch": 4.75076058974959,
      "grad_norm": 0.23565664887428284,
      "learning_rate": 2.5963045550647945e-05,
      "loss": 0.0462,
      "step": 40600
    },
    {
      "epoch": 4.75076058974959,
      "loss_main": 0.08011768013238907,
      "step": 40600
    },
    {
      "epoch": 4.751930727825884,
      "grad_norm": 0.2800872027873993,
      "learning_rate": 2.5938884742046466e-05,
      "loss": 0.0522,
      "step": 40610
    },
    {
      "epoch": 4.751930727825884,
      "loss_main": 0.07409477233886719,
      "step": 40610
    },
    {
      "epoch": 4.7531008659021765,
      "grad_norm": 0.20581889152526855,
      "learning_rate": 2.5914731242576507e-05,
      "loss": 0.0494,
      "step": 40620
    },
    {
      "epoch": 4.7531008659021765,
      "loss_main": 0.04397834092378616,
      "step": 40620
    },
    {
      "epoch": 4.754271003978469,
      "grad_norm": 0.23002608120441437,
      "learning_rate": 2.5890585059575268e-05,
      "loss": 0.0504,
      "step": 40630
    },
    {
      "epoch": 4.754271003978469,
      "loss_main": 0.059958960860967636,
      "step": 40630
    },
    {
      "epoch": 4.755441142054762,
      "grad_norm": 0.2549777328968048,
      "learning_rate": 2.5866446200377688e-05,
      "loss": 0.0464,
      "step": 40640
    },
    {
      "epoch": 4.755441142054762,
      "loss_main": 0.03982267901301384,
      "step": 40640
    },
    {
      "epoch": 4.756611280131056,
      "grad_norm": 0.2218366265296936,
      "learning_rate": 2.5842314672316566e-05,
      "loss": 0.0525,
      "step": 40650
    },
    {
      "epoch": 4.756611280131056,
      "loss_main": 0.04887847229838371,
      "step": 40650
    },
    {
      "epoch": 4.757781418207348,
      "grad_norm": 0.26233625411987305,
      "learning_rate": 2.581819048272239e-05,
      "loss": 0.0553,
      "step": 40660
    },
    {
      "epoch": 4.757781418207348,
      "loss_main": 0.09284056723117828,
      "step": 40660
    },
    {
      "epoch": 4.758951556283641,
      "grad_norm": 0.2285221666097641,
      "learning_rate": 2.5794073638923478e-05,
      "loss": 0.0551,
      "step": 40670
    },
    {
      "epoch": 4.758951556283641,
      "loss_main": 0.06204024329781532,
      "step": 40670
    },
    {
      "epoch": 4.760121694359935,
      "grad_norm": 0.3635985255241394,
      "learning_rate": 2.576996414824586e-05,
      "loss": 0.0638,
      "step": 40680
    },
    {
      "epoch": 4.760121694359935,
      "loss_main": 0.05041782185435295,
      "step": 40680
    },
    {
      "epoch": 4.7612918324362274,
      "grad_norm": 0.2762770354747772,
      "learning_rate": 2.574586201801339e-05,
      "loss": 0.0504,
      "step": 40690
    },
    {
      "epoch": 4.7612918324362274,
      "loss_main": 0.0448986291885376,
      "step": 40690
    },
    {
      "epoch": 4.76246197051252,
      "grad_norm": 0.2802804410457611,
      "learning_rate": 2.572176725554762e-05,
      "loss": 0.0529,
      "step": 40700
    },
    {
      "epoch": 4.76246197051252,
      "loss_main": 0.08271066099405289,
      "step": 40700
    },
    {
      "epoch": 4.763632108588814,
      "grad_norm": 0.2795085906982422,
      "learning_rate": 2.5697679868167966e-05,
      "loss": 0.0565,
      "step": 40710
    },
    {
      "epoch": 4.763632108588814,
      "loss_main": 0.03603621572256088,
      "step": 40710
    },
    {
      "epoch": 4.7648022466651065,
      "grad_norm": 0.3703453540802002,
      "learning_rate": 2.5673599863191468e-05,
      "loss": 0.053,
      "step": 40720
    },
    {
      "epoch": 4.7648022466651065,
      "loss_main": 0.08052033931016922,
      "step": 40720
    },
    {
      "epoch": 4.765972384741399,
      "grad_norm": 0.22842229902744293,
      "learning_rate": 2.564952724793306e-05,
      "loss": 0.0481,
      "step": 40730
    },
    {
      "epoch": 4.765972384741399,
      "loss_main": 0.05510907620191574,
      "step": 40730
    },
    {
      "epoch": 4.767142522817693,
      "grad_norm": 0.257192999124527,
      "learning_rate": 2.5625462029705306e-05,
      "loss": 0.049,
      "step": 40740
    },
    {
      "epoch": 4.767142522817693,
      "loss_main": 0.060184355825185776,
      "step": 40740
    },
    {
      "epoch": 4.768312660893986,
      "grad_norm": 0.20123693346977234,
      "learning_rate": 2.5601404215818624e-05,
      "loss": 0.0465,
      "step": 40750
    },
    {
      "epoch": 4.768312660893986,
      "loss_main": 0.04194159060716629,
      "step": 40750
    },
    {
      "epoch": 4.769482798970278,
      "grad_norm": 0.3194931149482727,
      "learning_rate": 2.5577353813581144e-05,
      "loss": 0.0503,
      "step": 40760
    },
    {
      "epoch": 4.769482798970278,
      "loss_main": 0.054190512746572495,
      "step": 40760
    },
    {
      "epoch": 4.770652937046571,
      "grad_norm": 0.2555525600910187,
      "learning_rate": 2.5553310830298733e-05,
      "loss": 0.0566,
      "step": 40770
    },
    {
      "epoch": 4.770652937046571,
      "loss_main": 0.07900267094373703,
      "step": 40770
    },
    {
      "epoch": 4.771823075122865,
      "grad_norm": 0.3083180785179138,
      "learning_rate": 2.5529275273275012e-05,
      "loss": 0.0538,
      "step": 40780
    },
    {
      "epoch": 4.771823075122865,
      "loss_main": 0.04241502657532692,
      "step": 40780
    },
    {
      "epoch": 4.7729932131991575,
      "grad_norm": 0.20112431049346924,
      "learning_rate": 2.550524714981133e-05,
      "loss": 0.0491,
      "step": 40790
    },
    {
      "epoch": 4.7729932131991575,
      "loss_main": 0.046057939529418945,
      "step": 40790
    },
    {
      "epoch": 4.77416335127545,
      "grad_norm": 0.22625944018363953,
      "learning_rate": 2.5481226467206837e-05,
      "loss": 0.0468,
      "step": 40800
    },
    {
      "epoch": 4.77416335127545,
      "loss_main": 0.0646696537733078,
      "step": 40800
    },
    {
      "epoch": 4.775333489351744,
      "grad_norm": 0.2872687578201294,
      "learning_rate": 2.5457213232758365e-05,
      "loss": 0.0487,
      "step": 40810
    },
    {
      "epoch": 4.775333489351744,
      "loss_main": 0.06065543368458748,
      "step": 40810
    },
    {
      "epoch": 4.776503627428037,
      "grad_norm": 0.21355071663856506,
      "learning_rate": 2.5433207453760498e-05,
      "loss": 0.062,
      "step": 40820
    },
    {
      "epoch": 4.776503627428037,
      "loss_main": 0.05155864730477333,
      "step": 40820
    },
    {
      "epoch": 4.777673765504329,
      "grad_norm": 0.29482579231262207,
      "learning_rate": 2.5409209137505552e-05,
      "loss": 0.0519,
      "step": 40830
    },
    {
      "epoch": 4.777673765504329,
      "loss_main": 0.061236441135406494,
      "step": 40830
    },
    {
      "epoch": 4.778843903580622,
      "grad_norm": 0.22104723751544952,
      "learning_rate": 2.5385218291283597e-05,
      "loss": 0.0403,
      "step": 40840
    },
    {
      "epoch": 4.778843903580622,
      "loss_main": 0.07094048708677292,
      "step": 40840
    },
    {
      "epoch": 4.780014041656916,
      "grad_norm": 0.25998038053512573,
      "learning_rate": 2.5361234922382383e-05,
      "loss": 0.052,
      "step": 40850
    },
    {
      "epoch": 4.780014041656916,
      "loss_main": 0.04641970992088318,
      "step": 40850
    },
    {
      "epoch": 4.781184179733208,
      "grad_norm": 0.21875624358654022,
      "learning_rate": 2.533725903808749e-05,
      "loss": 0.0528,
      "step": 40860
    },
    {
      "epoch": 4.781184179733208,
      "loss_main": 0.04674333333969116,
      "step": 40860
    },
    {
      "epoch": 4.782354317809501,
      "grad_norm": 0.2771938741207123,
      "learning_rate": 2.5313290645682085e-05,
      "loss": 0.0557,
      "step": 40870
    },
    {
      "epoch": 4.782354317809501,
      "loss_main": 0.0525534488260746,
      "step": 40870
    },
    {
      "epoch": 4.783524455885795,
      "grad_norm": 0.19424577057361603,
      "learning_rate": 2.52893297524472e-05,
      "loss": 0.057,
      "step": 40880
    },
    {
      "epoch": 4.783524455885795,
      "loss_main": 0.04481087252497673,
      "step": 40880
    },
    {
      "epoch": 4.7846945939620875,
      "grad_norm": 0.20991578698158264,
      "learning_rate": 2.526537636566145e-05,
      "loss": 0.0452,
      "step": 40890
    },
    {
      "epoch": 4.7846945939620875,
      "loss_main": 0.06266562640666962,
      "step": 40890
    },
    {
      "epoch": 4.78586473203838,
      "grad_norm": 0.24167759716510773,
      "learning_rate": 2.5241430492601305e-05,
      "loss": 0.0517,
      "step": 40900
    },
    {
      "epoch": 4.78586473203838,
      "loss_main": 0.06281791627407074,
      "step": 40900
    },
    {
      "epoch": 4.787034870114674,
      "grad_norm": 0.21010906994342804,
      "learning_rate": 2.5217492140540867e-05,
      "loss": 0.044,
      "step": 40910
    },
    {
      "epoch": 4.787034870114674,
      "loss_main": 0.03861396387219429,
      "step": 40910
    },
    {
      "epoch": 4.788205008190967,
      "grad_norm": 0.24253875017166138,
      "learning_rate": 2.5193561316751967e-05,
      "loss": 0.048,
      "step": 40920
    },
    {
      "epoch": 4.788205008190967,
      "loss_main": 0.03878252953290939,
      "step": 40920
    },
    {
      "epoch": 4.789375146267259,
      "grad_norm": 0.20610488951206207,
      "learning_rate": 2.516963802850416e-05,
      "loss": 0.0472,
      "step": 40930
    },
    {
      "epoch": 4.789375146267259,
      "loss_main": 0.04362231865525246,
      "step": 40930
    },
    {
      "epoch": 4.790545284343553,
      "grad_norm": 0.21980667114257812,
      "learning_rate": 2.5145722283064698e-05,
      "loss": 0.0469,
      "step": 40940
    },
    {
      "epoch": 4.790545284343553,
      "loss_main": 0.03789277747273445,
      "step": 40940
    },
    {
      "epoch": 4.791715422419846,
      "grad_norm": 0.19769933819770813,
      "learning_rate": 2.5121814087698602e-05,
      "loss": 0.0511,
      "step": 40950
    },
    {
      "epoch": 4.791715422419846,
      "loss_main": 0.042800482362508774,
      "step": 40950
    },
    {
      "epoch": 4.7928855604961385,
      "grad_norm": 0.24571415781974792,
      "learning_rate": 2.509791344966848e-05,
      "loss": 0.048,
      "step": 40960
    },
    {
      "epoch": 4.7928855604961385,
      "loss_main": 0.05837384983897209,
      "step": 40960
    },
    {
      "epoch": 4.794055698572431,
      "grad_norm": 0.42428624629974365,
      "learning_rate": 2.5074020376234768e-05,
      "loss": 0.0611,
      "step": 40970
    },
    {
      "epoch": 4.794055698572431,
      "loss_main": 0.03998694196343422,
      "step": 40970
    },
    {
      "epoch": 4.795225836648725,
      "grad_norm": 0.23001714050769806,
      "learning_rate": 2.5050134874655534e-05,
      "loss": 0.0483,
      "step": 40980
    },
    {
      "epoch": 4.795225836648725,
      "loss_main": 0.05670836940407753,
      "step": 40980
    },
    {
      "epoch": 4.7963959747250176,
      "grad_norm": 0.25844722986221313,
      "learning_rate": 2.5026256952186566e-05,
      "loss": 0.0485,
      "step": 40990
    },
    {
      "epoch": 4.7963959747250176,
      "loss_main": 0.04927825182676315,
      "step": 40990
    },
    {
      "epoch": 4.79756611280131,
      "grad_norm": 0.38904502987861633,
      "learning_rate": 2.5002386616081335e-05,
      "loss": 0.043,
      "step": 41000
    },
    {
      "epoch": 4.79756611280131,
      "loss_main": 0.05007736757397652,
      "step": 41000
    },
    {
      "epoch": 4.798736250877604,
      "grad_norm": 0.2271937131881714,
      "learning_rate": 2.497852387359103e-05,
      "loss": 0.063,
      "step": 41010
    },
    {
      "epoch": 4.798736250877604,
      "loss_main": 0.04878860339522362,
      "step": 41010
    },
    {
      "epoch": 4.799906388953897,
      "grad_norm": 0.30322664976119995,
      "learning_rate": 2.4954668731964496e-05,
      "loss": 0.0559,
      "step": 41020
    },
    {
      "epoch": 4.799906388953897,
      "loss_main": 0.05182470753788948,
      "step": 41020
    },
    {
      "epoch": 4.801076527030189,
      "grad_norm": 0.2772268056869507,
      "learning_rate": 2.4930821198448364e-05,
      "loss": 0.047,
      "step": 41030
    },
    {
      "epoch": 4.801076527030189,
      "loss_main": 0.07090753316879272,
      "step": 41030
    },
    {
      "epoch": 4.802246665106482,
      "grad_norm": 0.2630659341812134,
      "learning_rate": 2.4906981280286796e-05,
      "loss": 0.0601,
      "step": 41040
    },
    {
      "epoch": 4.802246665106482,
      "loss_main": 0.04828683286905289,
      "step": 41040
    },
    {
      "epoch": 4.803416803182776,
      "grad_norm": 0.23575249314308167,
      "learning_rate": 2.488314898472179e-05,
      "loss": 0.0501,
      "step": 41050
    },
    {
      "epoch": 4.803416803182776,
      "loss_main": 0.058291010558605194,
      "step": 41050
    },
    {
      "epoch": 4.8045869412590685,
      "grad_norm": 0.21065685153007507,
      "learning_rate": 2.485932431899295e-05,
      "loss": 0.0469,
      "step": 41060
    },
    {
      "epoch": 4.8045869412590685,
      "loss_main": 0.044309429824352264,
      "step": 41060
    },
    {
      "epoch": 4.805757079335361,
      "grad_norm": 0.31881529092788696,
      "learning_rate": 2.4835507290337584e-05,
      "loss": 0.0578,
      "step": 41070
    },
    {
      "epoch": 4.805757079335361,
      "loss_main": 0.03716852515935898,
      "step": 41070
    },
    {
      "epoch": 4.806927217411655,
      "grad_norm": 0.29830262064933777,
      "learning_rate": 2.4811697905990672e-05,
      "loss": 0.0488,
      "step": 41080
    },
    {
      "epoch": 4.806927217411655,
      "loss_main": 0.08450758457183838,
      "step": 41080
    },
    {
      "epoch": 4.808097355487948,
      "grad_norm": 0.2231834977865219,
      "learning_rate": 2.4787896173184854e-05,
      "loss": 0.0469,
      "step": 41090
    },
    {
      "epoch": 4.808097355487948,
      "loss_main": 0.04441255331039429,
      "step": 41090
    },
    {
      "epoch": 4.80926749356424,
      "grad_norm": 0.1728280931711197,
      "learning_rate": 2.4764102099150534e-05,
      "loss": 0.0553,
      "step": 41100
    },
    {
      "epoch": 4.80926749356424,
      "loss_main": 0.060340795665979385,
      "step": 41100
    },
    {
      "epoch": 4.810437631640534,
      "grad_norm": 0.2078304886817932,
      "learning_rate": 2.4740315691115644e-05,
      "loss": 0.0538,
      "step": 41110
    },
    {
      "epoch": 4.810437631640534,
      "loss_main": 0.04331653565168381,
      "step": 41110
    },
    {
      "epoch": 4.811607769716827,
      "grad_norm": 0.18594053387641907,
      "learning_rate": 2.4716536956305918e-05,
      "loss": 0.0549,
      "step": 41120
    },
    {
      "epoch": 4.811607769716827,
      "loss_main": 0.0668458342552185,
      "step": 41120
    },
    {
      "epoch": 4.812777907793119,
      "grad_norm": 0.21219711005687714,
      "learning_rate": 2.4692765901944697e-05,
      "loss": 0.0531,
      "step": 41130
    },
    {
      "epoch": 4.812777907793119,
      "loss_main": 0.051125671714544296,
      "step": 41130
    },
    {
      "epoch": 4.813948045869413,
      "grad_norm": 0.2828312814235687,
      "learning_rate": 2.4669002535253e-05,
      "loss": 0.0523,
      "step": 41140
    },
    {
      "epoch": 4.813948045869413,
      "loss_main": 0.0788174718618393,
      "step": 41140
    },
    {
      "epoch": 4.815118183945706,
      "grad_norm": 0.22447025775909424,
      "learning_rate": 2.46452468634495e-05,
      "loss": 0.0534,
      "step": 41150
    },
    {
      "epoch": 4.815118183945706,
      "loss_main": 0.04749852418899536,
      "step": 41150
    },
    {
      "epoch": 4.8162883220219985,
      "grad_norm": 0.1931218057870865,
      "learning_rate": 2.462149889375055e-05,
      "loss": 0.0528,
      "step": 41160
    },
    {
      "epoch": 4.8162883220219985,
      "loss_main": 0.04435231164097786,
      "step": 41160
    },
    {
      "epoch": 4.817458460098291,
      "grad_norm": 0.18305954337120056,
      "learning_rate": 2.459775863337014e-05,
      "loss": 0.0438,
      "step": 41170
    },
    {
      "epoch": 4.817458460098291,
      "loss_main": 0.051545899361371994,
      "step": 41170
    },
    {
      "epoch": 4.818628598174585,
      "grad_norm": 0.23219335079193115,
      "learning_rate": 2.4574026089519985e-05,
      "loss": 0.0499,
      "step": 41180
    },
    {
      "epoch": 4.818628598174585,
      "loss_main": 0.040540773421525955,
      "step": 41180
    },
    {
      "epoch": 4.819798736250878,
      "grad_norm": 0.19602112472057343,
      "learning_rate": 2.4550301269409333e-05,
      "loss": 0.0548,
      "step": 41190
    },
    {
      "epoch": 4.819798736250878,
      "loss_main": 0.05575791373848915,
      "step": 41190
    },
    {
      "epoch": 4.82096887432717,
      "grad_norm": 0.22193478047847748,
      "learning_rate": 2.4526584180245216e-05,
      "loss": 0.0565,
      "step": 41200
    },
    {
      "epoch": 4.82096887432717,
      "loss_main": 0.05161315202713013,
      "step": 41200
    },
    {
      "epoch": 4.822139012403464,
      "grad_norm": 0.2176767885684967,
      "learning_rate": 2.4502874829232236e-05,
      "loss": 0.0527,
      "step": 41210
    },
    {
      "epoch": 4.822139012403464,
      "loss_main": 0.03600715473294258,
      "step": 41210
    },
    {
      "epoch": 4.823309150479757,
      "grad_norm": 0.3349219858646393,
      "learning_rate": 2.447917322357267e-05,
      "loss": 0.0552,
      "step": 41220
    },
    {
      "epoch": 4.823309150479757,
      "loss_main": 0.06304584443569183,
      "step": 41220
    },
    {
      "epoch": 4.8244792885560495,
      "grad_norm": 0.28981345891952515,
      "learning_rate": 2.4455479370466443e-05,
      "loss": 0.0524,
      "step": 41230
    },
    {
      "epoch": 4.8244792885560495,
      "loss_main": 0.06867160648107529,
      "step": 41230
    },
    {
      "epoch": 4.825649426632342,
      "grad_norm": 0.22772769629955292,
      "learning_rate": 2.4431793277111097e-05,
      "loss": 0.0506,
      "step": 41240
    },
    {
      "epoch": 4.825649426632342,
      "loss_main": 0.02555627003312111,
      "step": 41240
    },
    {
      "epoch": 4.826819564708636,
      "grad_norm": 0.30338767170906067,
      "learning_rate": 2.4408114950701905e-05,
      "loss": 0.0441,
      "step": 41250
    },
    {
      "epoch": 4.826819564708636,
      "loss_main": 0.04212603345513344,
      "step": 41250
    },
    {
      "epoch": 4.827989702784929,
      "grad_norm": 0.1916525810956955,
      "learning_rate": 2.4384444398431634e-05,
      "loss": 0.0487,
      "step": 41260
    },
    {
      "epoch": 4.827989702784929,
      "loss_main": 0.0409177727997303,
      "step": 41260
    },
    {
      "epoch": 4.829159840861221,
      "grad_norm": 0.28391149640083313,
      "learning_rate": 2.4360781627490837e-05,
      "loss": 0.0525,
      "step": 41270
    },
    {
      "epoch": 4.829159840861221,
      "loss_main": 0.02174144610762596,
      "step": 41270
    },
    {
      "epoch": 4.830329978937515,
      "grad_norm": 0.19221436977386475,
      "learning_rate": 2.433712664506762e-05,
      "loss": 0.045,
      "step": 41280
    },
    {
      "epoch": 4.830329978937515,
      "loss_main": 0.05597039684653282,
      "step": 41280
    },
    {
      "epoch": 4.831500117013808,
      "grad_norm": 0.2883157432079315,
      "learning_rate": 2.431347945834774e-05,
      "loss": 0.0523,
      "step": 41290
    },
    {
      "epoch": 4.831500117013808,
      "loss_main": 0.05792936310172081,
      "step": 41290
    },
    {
      "epoch": 4.8326702550901,
      "grad_norm": 0.21733619272708893,
      "learning_rate": 2.428984007451458e-05,
      "loss": 0.0548,
      "step": 41300
    },
    {
      "epoch": 4.8326702550901,
      "loss_main": 0.03284561634063721,
      "step": 41300
    },
    {
      "epoch": 4.833840393166394,
      "grad_norm": 0.29089173674583435,
      "learning_rate": 2.426620850074917e-05,
      "loss": 0.0514,
      "step": 41310
    },
    {
      "epoch": 4.833840393166394,
      "loss_main": 0.06257523596286774,
      "step": 41310
    },
    {
      "epoch": 4.835010531242687,
      "grad_norm": 0.28946202993392944,
      "learning_rate": 2.424258474423014e-05,
      "loss": 0.0571,
      "step": 41320
    },
    {
      "epoch": 4.835010531242687,
      "loss_main": 0.05917265638709068,
      "step": 41320
    },
    {
      "epoch": 4.8361806693189795,
      "grad_norm": 0.2663279175758362,
      "learning_rate": 2.421896881213382e-05,
      "loss": 0.0552,
      "step": 41330
    },
    {
      "epoch": 4.8361806693189795,
      "loss_main": 0.07156364619731903,
      "step": 41330
    },
    {
      "epoch": 4.837350807395273,
      "grad_norm": 0.3678358197212219,
      "learning_rate": 2.419536071163402e-05,
      "loss": 0.0509,
      "step": 41340
    },
    {
      "epoch": 4.837350807395273,
      "loss_main": 0.03466759994626045,
      "step": 41340
    },
    {
      "epoch": 4.838520945471566,
      "grad_norm": 0.22683247923851013,
      "learning_rate": 2.417176044990233e-05,
      "loss": 0.0497,
      "step": 41350
    },
    {
      "epoch": 4.838520945471566,
      "loss_main": 0.049651000648736954,
      "step": 41350
    },
    {
      "epoch": 4.839691083547859,
      "grad_norm": 0.1820085197687149,
      "learning_rate": 2.4148168034107855e-05,
      "loss": 0.0482,
      "step": 41360
    },
    {
      "epoch": 4.839691083547859,
      "loss_main": 0.030914535745978355,
      "step": 41360
    },
    {
      "epoch": 4.840861221624151,
      "grad_norm": 0.21892374753952026,
      "learning_rate": 2.4124583471417355e-05,
      "loss": 0.0463,
      "step": 41370
    },
    {
      "epoch": 4.840861221624151,
      "loss_main": 0.0852532908320427,
      "step": 41370
    },
    {
      "epoch": 4.842031359700445,
      "grad_norm": 0.2470119297504425,
      "learning_rate": 2.41010067689952e-05,
      "loss": 0.0578,
      "step": 41380
    },
    {
      "epoch": 4.842031359700445,
      "loss_main": 0.040301401168107986,
      "step": 41380
    },
    {
      "epoch": 4.843201497776738,
      "grad_norm": 0.2551664710044861,
      "learning_rate": 2.4077437934003338e-05,
      "loss": 0.0485,
      "step": 41390
    },
    {
      "epoch": 4.843201497776738,
      "loss_main": 0.06547074019908905,
      "step": 41390
    },
    {
      "epoch": 4.84437163585303,
      "grad_norm": 0.2941076457500458,
      "learning_rate": 2.405387697360143e-05,
      "loss": 0.0528,
      "step": 41400
    },
    {
      "epoch": 4.84437163585303,
      "loss_main": 0.04774245619773865,
      "step": 41400
    },
    {
      "epoch": 4.845541773929323,
      "grad_norm": 0.2546171545982361,
      "learning_rate": 2.4030323894946595e-05,
      "loss": 0.0539,
      "step": 41410
    },
    {
      "epoch": 4.845541773929323,
      "loss_main": 0.05626269057393074,
      "step": 41410
    },
    {
      "epoch": 4.846711912005617,
      "grad_norm": 0.35411542654037476,
      "learning_rate": 2.40067787051937e-05,
      "loss": 0.0582,
      "step": 41420
    },
    {
      "epoch": 4.846711912005617,
      "loss_main": 0.044781215488910675,
      "step": 41420
    },
    {
      "epoch": 4.8478820500819095,
      "grad_norm": 0.19039063155651093,
      "learning_rate": 2.3983241411495087e-05,
      "loss": 0.0541,
      "step": 41430
    },
    {
      "epoch": 4.8478820500819095,
      "loss_main": 0.027653951197862625,
      "step": 41430
    },
    {
      "epoch": 4.849052188158202,
      "grad_norm": 0.34792065620422363,
      "learning_rate": 2.3959712021000823e-05,
      "loss": 0.0528,
      "step": 41440
    },
    {
      "epoch": 4.849052188158202,
      "loss_main": 0.0342191606760025,
      "step": 41440
    },
    {
      "epoch": 4.850222326234496,
      "grad_norm": 0.30313295125961304,
      "learning_rate": 2.3936190540858495e-05,
      "loss": 0.0504,
      "step": 41450
    },
    {
      "epoch": 4.850222326234496,
      "loss_main": 0.07504773885011673,
      "step": 41450
    },
    {
      "epoch": 4.851392464310789,
      "grad_norm": 0.3259706497192383,
      "learning_rate": 2.39126769782133e-05,
      "loss": 0.0573,
      "step": 41460
    },
    {
      "epoch": 4.851392464310789,
      "loss_main": 0.026948610320687294,
      "step": 41460
    },
    {
      "epoch": 4.852562602387081,
      "grad_norm": 0.2117774784564972,
      "learning_rate": 2.388917134020805e-05,
      "loss": 0.0515,
      "step": 41470
    },
    {
      "epoch": 4.852562602387081,
      "loss_main": 0.03612494841217995,
      "step": 41470
    },
    {
      "epoch": 4.853732740463375,
      "grad_norm": 0.29160651564598083,
      "learning_rate": 2.3865673633983128e-05,
      "loss": 0.0551,
      "step": 41480
    },
    {
      "epoch": 4.853732740463375,
      "loss_main": 0.03975823149085045,
      "step": 41480
    },
    {
      "epoch": 4.854902878539668,
      "grad_norm": 0.22464141249656677,
      "learning_rate": 2.3842183866676492e-05,
      "loss": 0.0457,
      "step": 41490
    },
    {
      "epoch": 4.854902878539668,
      "loss_main": 0.056095659732818604,
      "step": 41490
    },
    {
      "epoch": 4.8560730166159605,
      "grad_norm": 0.2805268466472626,
      "learning_rate": 2.381870204542377e-05,
      "loss": 0.0522,
      "step": 41500
    },
    {
      "epoch": 4.8560730166159605,
      "loss_main": 0.05791227146983147,
      "step": 41500
    },
    {
      "epoch": 4.857243154692254,
      "grad_norm": 0.24447977542877197,
      "learning_rate": 2.379522817735808e-05,
      "loss": 0.0467,
      "step": 41510
    },
    {
      "epoch": 4.857243154692254,
      "loss_main": 0.04161267727613449,
      "step": 41510
    },
    {
      "epoch": 4.858413292768547,
      "grad_norm": 0.31550365686416626,
      "learning_rate": 2.377176226961018e-05,
      "loss": 0.0515,
      "step": 41520
    },
    {
      "epoch": 4.858413292768547,
      "loss_main": 0.04532470181584358,
      "step": 41520
    },
    {
      "epoch": 4.85958343084484,
      "grad_norm": 0.1889663189649582,
      "learning_rate": 2.3748304329308384e-05,
      "loss": 0.0529,
      "step": 41530
    },
    {
      "epoch": 4.85958343084484,
      "loss_main": 0.031860701739788055,
      "step": 41530
    },
    {
      "epoch": 4.860753568921132,
      "grad_norm": 0.20971569418907166,
      "learning_rate": 2.372485436357858e-05,
      "loss": 0.0484,
      "step": 41540
    },
    {
      "epoch": 4.860753568921132,
      "loss_main": 0.06872779875993729,
      "step": 41540
    },
    {
      "epoch": 4.861923706997426,
      "grad_norm": 0.2825376093387604,
      "learning_rate": 2.3701412379544296e-05,
      "loss": 0.0601,
      "step": 41550
    },
    {
      "epoch": 4.861923706997426,
      "loss_main": 0.05544813349843025,
      "step": 41550
    },
    {
      "epoch": 4.863093845073719,
      "grad_norm": 0.23895983397960663,
      "learning_rate": 2.367797838432653e-05,
      "loss": 0.0519,
      "step": 41560
    },
    {
      "epoch": 4.863093845073719,
      "loss_main": 0.04279525950551033,
      "step": 41560
    },
    {
      "epoch": 4.864263983150011,
      "grad_norm": 0.2725798487663269,
      "learning_rate": 2.3654552385043967e-05,
      "loss": 0.0569,
      "step": 41570
    },
    {
      "epoch": 4.864263983150011,
      "loss_main": 0.09809231758117676,
      "step": 41570
    },
    {
      "epoch": 4.865434121226305,
      "grad_norm": 0.18717393279075623,
      "learning_rate": 2.3631134388812742e-05,
      "loss": 0.0543,
      "step": 41580
    },
    {
      "epoch": 4.865434121226305,
      "loss_main": 0.12961015105247498,
      "step": 41580
    },
    {
      "epoch": 4.866604259302598,
      "grad_norm": 0.20807211101055145,
      "learning_rate": 2.3607724402746684e-05,
      "loss": 0.0593,
      "step": 41590
    },
    {
      "epoch": 4.866604259302598,
      "loss_main": 0.05675183981657028,
      "step": 41590
    },
    {
      "epoch": 4.8677743973788905,
      "grad_norm": 0.25428617000579834,
      "learning_rate": 2.35843224339571e-05,
      "loss": 0.05,
      "step": 41600
    },
    {
      "epoch": 4.8677743973788905,
      "loss_main": 0.033066004514694214,
      "step": 41600
    },
    {
      "epoch": 4.868944535455183,
      "grad_norm": 0.20388206839561462,
      "learning_rate": 2.3560928489552897e-05,
      "loss": 0.0495,
      "step": 41610
    },
    {
      "epoch": 4.868944535455183,
      "loss_main": 0.028392110019922256,
      "step": 41610
    },
    {
      "epoch": 4.870114673531477,
      "grad_norm": 0.1954687088727951,
      "learning_rate": 2.353754257664053e-05,
      "loss": 0.0551,
      "step": 41620
    },
    {
      "epoch": 4.870114673531477,
      "loss_main": 0.05361606553196907,
      "step": 41620
    },
    {
      "epoch": 4.87128481160777,
      "grad_norm": 0.2700049877166748,
      "learning_rate": 2.3514164702324037e-05,
      "loss": 0.0519,
      "step": 41630
    },
    {
      "epoch": 4.87128481160777,
      "loss_main": 0.059369999915361404,
      "step": 41630
    },
    {
      "epoch": 4.872454949684062,
      "grad_norm": 0.27753692865371704,
      "learning_rate": 2.3490794873704963e-05,
      "loss": 0.0588,
      "step": 41640
    },
    {
      "epoch": 4.872454949684062,
      "loss_main": 0.031499236822128296,
      "step": 41640
    },
    {
      "epoch": 4.873625087760356,
      "grad_norm": 0.2793828547000885,
      "learning_rate": 2.3467433097882496e-05,
      "loss": 0.0492,
      "step": 41650
    },
    {
      "epoch": 4.873625087760356,
      "loss_main": 0.046453118324279785,
      "step": 41650
    },
    {
      "epoch": 4.874795225836649,
      "grad_norm": 0.25700506567955017,
      "learning_rate": 2.34440793819533e-05,
      "loss": 0.053,
      "step": 41660
    },
    {
      "epoch": 4.874795225836649,
      "loss_main": 0.04966651648283005,
      "step": 41660
    },
    {
      "epoch": 4.875965363912941,
      "grad_norm": 0.2190575748682022,
      "learning_rate": 2.3420733733011617e-05,
      "loss": 0.0472,
      "step": 41670
    },
    {
      "epoch": 4.875965363912941,
      "loss_main": 0.05998881906270981,
      "step": 41670
    },
    {
      "epoch": 4.877135501989235,
      "grad_norm": 0.2306007593870163,
      "learning_rate": 2.3397396158149243e-05,
      "loss": 0.0532,
      "step": 41680
    },
    {
      "epoch": 4.877135501989235,
      "loss_main": 0.07586321979761124,
      "step": 41680
    },
    {
      "epoch": 4.878305640065528,
      "grad_norm": 0.2203608602285385,
      "learning_rate": 2.3374066664455498e-05,
      "loss": 0.0503,
      "step": 41690
    },
    {
      "epoch": 4.878305640065528,
      "loss_main": 0.033811118453741074,
      "step": 41690
    },
    {
      "epoch": 4.8794757781418205,
      "grad_norm": 0.23560865223407745,
      "learning_rate": 2.3350745259017315e-05,
      "loss": 0.0462,
      "step": 41700
    },
    {
      "epoch": 4.8794757781418205,
      "loss_main": 0.06262174993753433,
      "step": 41700
    },
    {
      "epoch": 4.880645916218114,
      "grad_norm": 0.22863110899925232,
      "learning_rate": 2.332743194891906e-05,
      "loss": 0.0488,
      "step": 41710
    },
    {
      "epoch": 4.880645916218114,
      "loss_main": 0.029075857251882553,
      "step": 41710
    },
    {
      "epoch": 4.881816054294407,
      "grad_norm": 0.2520618140697479,
      "learning_rate": 2.330412674124276e-05,
      "loss": 0.0395,
      "step": 41720
    },
    {
      "epoch": 4.881816054294407,
      "loss_main": 0.03797311335802078,
      "step": 41720
    },
    {
      "epoch": 4.8829861923707,
      "grad_norm": 0.2142830640077591,
      "learning_rate": 2.328082964306786e-05,
      "loss": 0.048,
      "step": 41730
    },
    {
      "epoch": 4.8829861923707,
      "loss_main": 0.034865107387304306,
      "step": 41730
    },
    {
      "epoch": 4.884156330446992,
      "grad_norm": 0.3841436505317688,
      "learning_rate": 2.325754066147145e-05,
      "loss": 0.0505,
      "step": 41740
    },
    {
      "epoch": 4.884156330446992,
      "loss_main": 0.025794221088290215,
      "step": 41740
    },
    {
      "epoch": 4.885326468523286,
      "grad_norm": 0.30032840371131897,
      "learning_rate": 2.32342598035281e-05,
      "loss": 0.0525,
      "step": 41750
    },
    {
      "epoch": 4.885326468523286,
      "loss_main": 0.03539938107132912,
      "step": 41750
    },
    {
      "epoch": 4.886496606599579,
      "grad_norm": 0.2518666088581085,
      "learning_rate": 2.321098707630991e-05,
      "loss": 0.0481,
      "step": 41760
    },
    {
      "epoch": 4.886496606599579,
      "loss_main": 0.05383941903710365,
      "step": 41760
    },
    {
      "epoch": 4.8876667446758715,
      "grad_norm": 0.24331095814704895,
      "learning_rate": 2.318772248688652e-05,
      "loss": 0.042,
      "step": 41770
    },
    {
      "epoch": 4.8876667446758715,
      "loss_main": 0.025457292795181274,
      "step": 41770
    },
    {
      "epoch": 4.888836882752165,
      "grad_norm": 0.17002390325069427,
      "learning_rate": 2.3164466042325107e-05,
      "loss": 0.0522,
      "step": 41780
    },
    {
      "epoch": 4.888836882752165,
      "loss_main": 0.07101380079984665,
      "step": 41780
    },
    {
      "epoch": 4.890007020828458,
      "grad_norm": 0.22187721729278564,
      "learning_rate": 2.3141217749690353e-05,
      "loss": 0.0564,
      "step": 41790
    },
    {
      "epoch": 4.890007020828458,
      "loss_main": 0.052181463688611984,
      "step": 41790
    },
    {
      "epoch": 4.891177158904751,
      "grad_norm": 0.25463011860847473,
      "learning_rate": 2.3117977616044466e-05,
      "loss": 0.0528,
      "step": 41800
    },
    {
      "epoch": 4.891177158904751,
      "loss_main": 0.030092690140008926,
      "step": 41800
    },
    {
      "epoch": 4.892347296981043,
      "grad_norm": 0.3680768609046936,
      "learning_rate": 2.309474564844722e-05,
      "loss": 0.0525,
      "step": 41810
    },
    {
      "epoch": 4.892347296981043,
      "loss_main": 0.06960029900074005,
      "step": 41810
    },
    {
      "epoch": 4.893517435057337,
      "grad_norm": 0.2962423264980316,
      "learning_rate": 2.307152185395585e-05,
      "loss": 0.0444,
      "step": 41820
    },
    {
      "epoch": 4.893517435057337,
      "loss_main": 0.041865963488817215,
      "step": 41820
    },
    {
      "epoch": 4.89468757313363,
      "grad_norm": 0.2391926646232605,
      "learning_rate": 2.3048306239625144e-05,
      "loss": 0.0488,
      "step": 41830
    },
    {
      "epoch": 4.89468757313363,
      "loss_main": 0.045592401176691055,
      "step": 41830
    },
    {
      "epoch": 4.895857711209922,
      "grad_norm": 0.23433910310268402,
      "learning_rate": 2.3025098812507378e-05,
      "loss": 0.0572,
      "step": 41840
    },
    {
      "epoch": 4.895857711209922,
      "loss_main": 0.07489878684282303,
      "step": 41840
    },
    {
      "epoch": 4.897027849286216,
      "grad_norm": 0.18564362823963165,
      "learning_rate": 2.3001899579652366e-05,
      "loss": 0.0558,
      "step": 41850
    },
    {
      "epoch": 4.897027849286216,
      "loss_main": 0.047453317791223526,
      "step": 41850
    },
    {
      "epoch": 4.898197987362509,
      "grad_norm": 0.1901872456073761,
      "learning_rate": 2.2978708548107393e-05,
      "loss": 0.0452,
      "step": 41860
    },
    {
      "epoch": 4.898197987362509,
      "loss_main": 0.046897176653146744,
      "step": 41860
    },
    {
      "epoch": 4.8993681254388015,
      "grad_norm": 0.18650011718273163,
      "learning_rate": 2.2955525724917348e-05,
      "loss": 0.0477,
      "step": 41870
    },
    {
      "epoch": 4.8993681254388015,
      "loss_main": 0.06372430175542831,
      "step": 41870
    },
    {
      "epoch": 4.900538263515095,
      "grad_norm": 0.23345911502838135,
      "learning_rate": 2.2932351117124477e-05,
      "loss": 0.0545,
      "step": 41880
    },
    {
      "epoch": 4.900538263515095,
      "loss_main": 0.05756048113107681,
      "step": 41880
    },
    {
      "epoch": 4.901708401591388,
      "grad_norm": 0.21686328947544098,
      "learning_rate": 2.29091847317687e-05,
      "loss": 0.0515,
      "step": 41890
    },
    {
      "epoch": 4.901708401591388,
      "loss_main": 0.03291570022702217,
      "step": 41890
    },
    {
      "epoch": 4.902878539667681,
      "grad_norm": 0.2570917010307312,
      "learning_rate": 2.2886026575887277e-05,
      "loss": 0.0513,
      "step": 41900
    },
    {
      "epoch": 4.902878539667681,
      "loss_main": 0.034344471991062164,
      "step": 41900
    },
    {
      "epoch": 4.904048677743974,
      "grad_norm": 0.274728000164032,
      "learning_rate": 2.2862876656515094e-05,
      "loss": 0.0549,
      "step": 41910
    },
    {
      "epoch": 4.904048677743974,
      "loss_main": 0.08640459924936295,
      "step": 41910
    },
    {
      "epoch": 4.905218815820267,
      "grad_norm": 0.21834935247898102,
      "learning_rate": 2.2839734980684464e-05,
      "loss": 0.0514,
      "step": 41920
    },
    {
      "epoch": 4.905218815820267,
      "loss_main": 0.07508610188961029,
      "step": 41920
    },
    {
      "epoch": 4.90638895389656,
      "grad_norm": 0.2048812061548233,
      "learning_rate": 2.281660155542522e-05,
      "loss": 0.0554,
      "step": 41930
    },
    {
      "epoch": 4.90638895389656,
      "loss_main": 0.04541820287704468,
      "step": 41930
    },
    {
      "epoch": 4.9075590919728524,
      "grad_norm": 0.2007645219564438,
      "learning_rate": 2.279347638776469e-05,
      "loss": 0.0525,
      "step": 41940
    },
    {
      "epoch": 4.9075590919728524,
      "loss_main": 0.0937454029917717,
      "step": 41940
    },
    {
      "epoch": 4.908729230049146,
      "grad_norm": 0.2913353443145752,
      "learning_rate": 2.2770359484727665e-05,
      "loss": 0.056,
      "step": 41950
    },
    {
      "epoch": 4.908729230049146,
      "loss_main": 0.04008104279637337,
      "step": 41950
    },
    {
      "epoch": 4.909899368125439,
      "grad_norm": 0.2770039141178131,
      "learning_rate": 2.27472508533365e-05,
      "loss": 0.0537,
      "step": 41960
    },
    {
      "epoch": 4.909899368125439,
      "loss_main": 0.06365462392568588,
      "step": 41960
    },
    {
      "epoch": 4.9110695062017315,
      "grad_norm": 0.28379008173942566,
      "learning_rate": 2.2724150500610948e-05,
      "loss": 0.0555,
      "step": 41970
    },
    {
      "epoch": 4.9110695062017315,
      "loss_main": 0.02921326830983162,
      "step": 41970
    },
    {
      "epoch": 4.912239644278025,
      "grad_norm": 0.35529252886772156,
      "learning_rate": 2.2701058433568302e-05,
      "loss": 0.05,
      "step": 41980
    },
    {
      "epoch": 4.912239644278025,
      "loss_main": 0.0549006462097168,
      "step": 41980
    },
    {
      "epoch": 4.913409782354318,
      "grad_norm": 0.2360164076089859,
      "learning_rate": 2.2677974659223318e-05,
      "loss": 0.0593,
      "step": 41990
    },
    {
      "epoch": 4.913409782354318,
      "loss_main": 0.05036735534667969,
      "step": 41990
    },
    {
      "epoch": 4.914579920430611,
      "grad_norm": 0.25257962942123413,
      "learning_rate": 2.2654899184588235e-05,
      "loss": 0.0558,
      "step": 42000
    },
    {
      "epoch": 4.914579920430611,
      "loss_main": 0.03236496075987816,
      "step": 42000
    },
    {
      "epoch": 4.915750058506903,
      "grad_norm": 0.29748520255088806,
      "learning_rate": 2.2631832016672756e-05,
      "loss": 0.0521,
      "step": 42010
    },
    {
      "epoch": 4.915750058506903,
      "loss_main": 0.04750773310661316,
      "step": 42010
    },
    {
      "epoch": 4.916920196583197,
      "grad_norm": 0.21739189326763153,
      "learning_rate": 2.2608773162484127e-05,
      "loss": 0.0474,
      "step": 42020
    },
    {
      "epoch": 4.916920196583197,
      "loss_main": 0.05414062738418579,
      "step": 42020
    },
    {
      "epoch": 4.91809033465949,
      "grad_norm": 0.2421233355998993,
      "learning_rate": 2.2585722629026958e-05,
      "loss": 0.0512,
      "step": 42030
    },
    {
      "epoch": 4.91809033465949,
      "loss_main": 0.04245106130838394,
      "step": 42030
    },
    {
      "epoch": 4.9192604727357825,
      "grad_norm": 0.2791595160961151,
      "learning_rate": 2.2562680423303457e-05,
      "loss": 0.0459,
      "step": 42040
    },
    {
      "epoch": 4.9192604727357825,
      "loss_main": 0.025305794551968575,
      "step": 42040
    },
    {
      "epoch": 4.920430610812076,
      "grad_norm": 0.21950845420360565,
      "learning_rate": 2.2539646552313165e-05,
      "loss": 0.0532,
      "step": 42050
    },
    {
      "epoch": 4.920430610812076,
      "loss_main": 0.06925579160451889,
      "step": 42050
    },
    {
      "epoch": 4.921600748888369,
      "grad_norm": 0.2195734977722168,
      "learning_rate": 2.251662102305322e-05,
      "loss": 0.0507,
      "step": 42060
    },
    {
      "epoch": 4.921600748888369,
      "loss_main": 0.03454644978046417,
      "step": 42060
    },
    {
      "epoch": 4.922770886964662,
      "grad_norm": 0.2844189703464508,
      "learning_rate": 2.2493603842518152e-05,
      "loss": 0.05,
      "step": 42070
    },
    {
      "epoch": 4.922770886964662,
      "loss_main": 0.041966237127780914,
      "step": 42070
    },
    {
      "epoch": 4.923941025040955,
      "grad_norm": 0.4203420579433441,
      "learning_rate": 2.2470595017699974e-05,
      "loss": 0.0527,
      "step": 42080
    },
    {
      "epoch": 4.923941025040955,
      "loss_main": 0.03688768669962883,
      "step": 42080
    },
    {
      "epoch": 4.925111163117248,
      "grad_norm": 0.29709598422050476,
      "learning_rate": 2.244759455558816e-05,
      "loss": 0.0552,
      "step": 42090
    },
    {
      "epoch": 4.925111163117248,
      "loss_main": 0.03996141627430916,
      "step": 42090
    },
    {
      "epoch": 4.926281301193541,
      "grad_norm": 0.29140704870224,
      "learning_rate": 2.2424602463169614e-05,
      "loss": 0.0482,
      "step": 42100
    },
    {
      "epoch": 4.926281301193541,
      "loss_main": 0.06646600365638733,
      "step": 42100
    },
    {
      "epoch": 4.927451439269834,
      "grad_norm": 0.21071894466876984,
      "learning_rate": 2.2401618747428776e-05,
      "loss": 0.0535,
      "step": 42110
    },
    {
      "epoch": 4.927451439269834,
      "loss_main": 0.03746085241436958,
      "step": 42110
    },
    {
      "epoch": 4.928621577346127,
      "grad_norm": 0.21750031411647797,
      "learning_rate": 2.237864341534747e-05,
      "loss": 0.0531,
      "step": 42120
    },
    {
      "epoch": 4.928621577346127,
      "loss_main": 0.04559094458818436,
      "step": 42120
    },
    {
      "epoch": 4.92979171542242,
      "grad_norm": 0.25661423802375793,
      "learning_rate": 2.2355676473904998e-05,
      "loss": 0.0491,
      "step": 42130
    },
    {
      "epoch": 4.92979171542242,
      "loss_main": 0.027721738442778587,
      "step": 42130
    },
    {
      "epoch": 4.9309618534987125,
      "grad_norm": 0.19818928837776184,
      "learning_rate": 2.2332717930078108e-05,
      "loss": 0.0532,
      "step": 42140
    },
    {
      "epoch": 4.9309618534987125,
      "loss_main": 0.026796551421284676,
      "step": 42140
    },
    {
      "epoch": 4.932131991575006,
      "grad_norm": 0.24259072542190552,
      "learning_rate": 2.2309767790840992e-05,
      "loss": 0.0579,
      "step": 42150
    },
    {
      "epoch": 4.932131991575006,
      "loss_main": 0.06926929205656052,
      "step": 42150
    },
    {
      "epoch": 4.933302129651299,
      "grad_norm": 0.20595170557498932,
      "learning_rate": 2.228682606316529e-05,
      "loss": 0.0478,
      "step": 42160
    },
    {
      "epoch": 4.933302129651299,
      "loss_main": 0.042312245815992355,
      "step": 42160
    },
    {
      "epoch": 4.934472267727592,
      "grad_norm": 0.19425825774669647,
      "learning_rate": 2.2263892754020138e-05,
      "loss": 0.0526,
      "step": 42170
    },
    {
      "epoch": 4.934472267727592,
      "loss_main": 0.05197638273239136,
      "step": 42170
    },
    {
      "epoch": 4.935642405803885,
      "grad_norm": 0.2678694427013397,
      "learning_rate": 2.2240967870372004e-05,
      "loss": 0.0425,
      "step": 42180
    },
    {
      "epoch": 4.935642405803885,
      "loss_main": 0.04821617528796196,
      "step": 42180
    },
    {
      "epoch": 4.936812543880178,
      "grad_norm": 0.3168574571609497,
      "learning_rate": 2.2218051419184933e-05,
      "loss": 0.0535,
      "step": 42190
    },
    {
      "epoch": 4.936812543880178,
      "loss_main": 0.05850585177540779,
      "step": 42190
    },
    {
      "epoch": 4.937982681956471,
      "grad_norm": 0.29232487082481384,
      "learning_rate": 2.219514340742026e-05,
      "loss": 0.0575,
      "step": 42200
    },
    {
      "epoch": 4.937982681956471,
      "loss_main": 0.035644304007291794,
      "step": 42200
    },
    {
      "epoch": 4.9391528200327635,
      "grad_norm": 0.2271072417497635,
      "learning_rate": 2.2172243842036898e-05,
      "loss": 0.0573,
      "step": 42210
    },
    {
      "epoch": 4.9391528200327635,
      "loss_main": 0.05236448720097542,
      "step": 42210
    },
    {
      "epoch": 4.940322958109057,
      "grad_norm": 0.3036877512931824,
      "learning_rate": 2.2149352729991107e-05,
      "loss": 0.0535,
      "step": 42220
    },
    {
      "epoch": 4.940322958109057,
      "loss_main": 0.05806149169802666,
      "step": 42220
    },
    {
      "epoch": 4.94149309618535,
      "grad_norm": 0.22035931050777435,
      "learning_rate": 2.2126470078236605e-05,
      "loss": 0.0543,
      "step": 42230
    },
    {
      "epoch": 4.94149309618535,
      "loss_main": 0.02636154368519783,
      "step": 42230
    },
    {
      "epoch": 4.942663234261643,
      "grad_norm": 0.24757614731788635,
      "learning_rate": 2.2103595893724533e-05,
      "loss": 0.047,
      "step": 42240
    },
    {
      "epoch": 4.942663234261643,
      "loss_main": 0.05501110106706619,
      "step": 42240
    },
    {
      "epoch": 4.943833372337936,
      "grad_norm": 0.22128675878047943,
      "learning_rate": 2.208073018340345e-05,
      "loss": 0.0471,
      "step": 42250
    },
    {
      "epoch": 4.943833372337936,
      "loss_main": 0.029974844306707382,
      "step": 42250
    },
    {
      "epoch": 4.945003510414229,
      "grad_norm": 0.1786593496799469,
      "learning_rate": 2.2057872954219405e-05,
      "loss": 0.0536,
      "step": 42260
    },
    {
      "epoch": 4.945003510414229,
      "loss_main": 0.06061187759041786,
      "step": 42260
    },
    {
      "epoch": 4.946173648490522,
      "grad_norm": 0.2617979943752289,
      "learning_rate": 2.203502421311575e-05,
      "loss": 0.0557,
      "step": 42270
    },
    {
      "epoch": 4.946173648490522,
      "loss_main": 0.09997909516096115,
      "step": 42270
    },
    {
      "epoch": 4.947343786566815,
      "grad_norm": 0.23362678289413452,
      "learning_rate": 2.2012183967033388e-05,
      "loss": 0.0507,
      "step": 42280
    },
    {
      "epoch": 4.947343786566815,
      "loss_main": 0.04551517963409424,
      "step": 42280
    },
    {
      "epoch": 4.948513924643108,
      "grad_norm": 0.2601812481880188,
      "learning_rate": 2.198935222291056e-05,
      "loss": 0.0512,
      "step": 42290
    },
    {
      "epoch": 4.948513924643108,
      "loss_main": 0.07028766721487045,
      "step": 42290
    },
    {
      "epoch": 4.949684062719401,
      "grad_norm": 0.21880510449409485,
      "learning_rate": 2.1966528987682948e-05,
      "loss": 0.051,
      "step": 42300
    },
    {
      "epoch": 4.949684062719401,
      "loss_main": 0.020047005265951157,
      "step": 42300
    },
    {
      "epoch": 4.950854200795694,
      "grad_norm": 0.24457214772701263,
      "learning_rate": 2.194371426828365e-05,
      "loss": 0.0513,
      "step": 42310
    },
    {
      "epoch": 4.950854200795694,
      "loss_main": 0.0631144791841507,
      "step": 42310
    },
    {
      "epoch": 4.952024338871987,
      "grad_norm": 0.27440762519836426,
      "learning_rate": 2.192090807164317e-05,
      "loss": 0.0425,
      "step": 42320
    },
    {
      "epoch": 4.952024338871987,
      "loss_main": 0.06558262556791306,
      "step": 42320
    },
    {
      "epoch": 4.95319447694828,
      "grad_norm": 0.2538142204284668,
      "learning_rate": 2.1898110404689422e-05,
      "loss": 0.0528,
      "step": 42330
    },
    {
      "epoch": 4.95319447694828,
      "loss_main": 0.0880388468503952,
      "step": 42330
    },
    {
      "epoch": 4.954364615024573,
      "grad_norm": 0.2549414336681366,
      "learning_rate": 2.1875321274347776e-05,
      "loss": 0.0512,
      "step": 42340
    },
    {
      "epoch": 4.954364615024573,
      "loss_main": 0.02914990670979023,
      "step": 42340
    },
    {
      "epoch": 4.955534753100866,
      "grad_norm": 0.1825730949640274,
      "learning_rate": 2.18525406875409e-05,
      "loss": 0.0466,
      "step": 42350
    },
    {
      "epoch": 4.955534753100866,
      "loss_main": 0.07261604070663452,
      "step": 42350
    },
    {
      "epoch": 4.956704891177159,
      "grad_norm": 0.250318318605423,
      "learning_rate": 2.1829768651188997e-05,
      "loss": 0.0509,
      "step": 42360
    },
    {
      "epoch": 4.956704891177159,
      "loss_main": 0.035005100071430206,
      "step": 42360
    },
    {
      "epoch": 4.957875029253452,
      "grad_norm": 0.2739800810813904,
      "learning_rate": 2.180700517220958e-05,
      "loss": 0.041,
      "step": 42370
    },
    {
      "epoch": 4.957875029253452,
      "loss_main": 0.060594890266656876,
      "step": 42370
    },
    {
      "epoch": 4.959045167329744,
      "grad_norm": 0.1920781135559082,
      "learning_rate": 2.1784250257517603e-05,
      "loss": 0.0514,
      "step": 42380
    },
    {
      "epoch": 4.959045167329744,
      "loss_main": 0.09657837450504303,
      "step": 42380
    },
    {
      "epoch": 4.960215305406038,
      "grad_norm": 0.3885713815689087,
      "learning_rate": 2.1761503914025406e-05,
      "loss": 0.0553,
      "step": 42390
    },
    {
      "epoch": 4.960215305406038,
      "loss_main": 0.05094955116510391,
      "step": 42390
    },
    {
      "epoch": 4.961385443482331,
      "grad_norm": 0.3134958744049072,
      "learning_rate": 2.1738766148642705e-05,
      "loss": 0.0547,
      "step": 42400
    },
    {
      "epoch": 4.961385443482331,
      "loss_main": 0.04091065004467964,
      "step": 42400
    },
    {
      "epoch": 4.9625555815586235,
      "grad_norm": 0.34714818000793457,
      "learning_rate": 2.1716036968276683e-05,
      "loss": 0.0508,
      "step": 42410
    },
    {
      "epoch": 4.9625555815586235,
      "loss_main": 0.02821110375225544,
      "step": 42410
    },
    {
      "epoch": 4.963725719634917,
      "grad_norm": 0.26510536670684814,
      "learning_rate": 2.1693316379831808e-05,
      "loss": 0.0486,
      "step": 42420
    },
    {
      "epoch": 4.963725719634917,
      "loss_main": 0.04538285732269287,
      "step": 42420
    },
    {
      "epoch": 4.96489585771121,
      "grad_norm": 0.24655722081661224,
      "learning_rate": 2.1670604390210037e-05,
      "loss": 0.053,
      "step": 42430
    },
    {
      "epoch": 4.96489585771121,
      "loss_main": 0.045984022319316864,
      "step": 42430
    },
    {
      "epoch": 4.966065995787503,
      "grad_norm": 0.25827059149742126,
      "learning_rate": 2.1647901006310656e-05,
      "loss": 0.0518,
      "step": 42440
    },
    {
      "epoch": 4.966065995787503,
      "loss_main": 0.04209868982434273,
      "step": 42440
    },
    {
      "epoch": 4.967236133863796,
      "grad_norm": 0.2634910047054291,
      "learning_rate": 2.1625206235030353e-05,
      "loss": 0.051,
      "step": 42450
    },
    {
      "epoch": 4.967236133863796,
      "loss_main": 0.07816269248723984,
      "step": 42450
    },
    {
      "epoch": 4.968406271940089,
      "grad_norm": 0.2466811090707779,
      "learning_rate": 2.160252008326321e-05,
      "loss": 0.0513,
      "step": 42460
    },
    {
      "epoch": 4.968406271940089,
      "loss_main": 0.05977844074368477,
      "step": 42460
    },
    {
      "epoch": 4.969576410016382,
      "grad_norm": 0.19470509886741638,
      "learning_rate": 2.157984255790067e-05,
      "loss": 0.0537,
      "step": 42470
    },
    {
      "epoch": 4.969576410016382,
      "loss_main": 0.05571712926030159,
      "step": 42470
    },
    {
      "epoch": 4.970746548092675,
      "grad_norm": 0.2698816955089569,
      "learning_rate": 2.1557173665831553e-05,
      "loss": 0.0506,
      "step": 42480
    },
    {
      "epoch": 4.970746548092675,
      "loss_main": 0.04416078329086304,
      "step": 42480
    },
    {
      "epoch": 4.971916686168968,
      "grad_norm": 0.19586117565631866,
      "learning_rate": 2.153451341394212e-05,
      "loss": 0.0521,
      "step": 42490
    },
    {
      "epoch": 4.971916686168968,
      "loss_main": 0.04809047654271126,
      "step": 42490
    },
    {
      "epoch": 4.973086824245261,
      "grad_norm": 0.20308253169059753,
      "learning_rate": 2.151186180911589e-05,
      "loss": 0.0496,
      "step": 42500
    },
    {
      "epoch": 4.973086824245261,
      "loss_main": 0.04249321296811104,
      "step": 42500
    },
    {
      "epoch": 4.974256962321554,
      "grad_norm": 0.2701813280582428,
      "learning_rate": 2.1489218858233877e-05,
      "loss": 0.0551,
      "step": 42510
    },
    {
      "epoch": 4.974256962321554,
      "loss_main": 0.03972851485013962,
      "step": 42510
    },
    {
      "epoch": 4.975427100397847,
      "grad_norm": 0.235866516828537,
      "learning_rate": 2.1466584568174392e-05,
      "loss": 0.0605,
      "step": 42520
    },
    {
      "epoch": 4.975427100397847,
      "loss_main": 0.060809459537267685,
      "step": 42520
    },
    {
      "epoch": 4.97659723847414,
      "grad_norm": 0.20525327324867249,
      "learning_rate": 2.1443958945813132e-05,
      "loss": 0.0549,
      "step": 42530
    },
    {
      "epoch": 4.97659723847414,
      "loss_main": 0.03303796797990799,
      "step": 42530
    },
    {
      "epoch": 4.977767376550433,
      "grad_norm": 0.3085118532180786,
      "learning_rate": 2.1421341998023163e-05,
      "loss": 0.0485,
      "step": 42540
    },
    {
      "epoch": 4.977767376550433,
      "loss_main": 0.05492322891950607,
      "step": 42540
    },
    {
      "epoch": 4.978937514626726,
      "grad_norm": 0.27911919355392456,
      "learning_rate": 2.139873373167491e-05,
      "loss": 0.0568,
      "step": 42550
    },
    {
      "epoch": 4.978937514626726,
      "loss_main": 0.076470747590065,
      "step": 42550
    },
    {
      "epoch": 4.980107652703019,
      "grad_norm": 0.19813838601112366,
      "learning_rate": 2.13761341536362e-05,
      "loss": 0.0522,
      "step": 42560
    },
    {
      "epoch": 4.980107652703019,
      "loss_main": 0.0441003181040287,
      "step": 42560
    },
    {
      "epoch": 4.981277790779312,
      "grad_norm": 0.2657259404659271,
      "learning_rate": 2.1353543270772136e-05,
      "loss": 0.0475,
      "step": 42570
    },
    {
      "epoch": 4.981277790779312,
      "loss_main": 0.045861415565013885,
      "step": 42570
    },
    {
      "epoch": 4.9824479288556045,
      "grad_norm": 0.21272780001163483,
      "learning_rate": 2.1330961089945297e-05,
      "loss": 0.0506,
      "step": 42580
    },
    {
      "epoch": 4.9824479288556045,
      "loss_main": 0.05571490526199341,
      "step": 42580
    },
    {
      "epoch": 4.983618066931898,
      "grad_norm": 0.25797542929649353,
      "learning_rate": 2.130838761801548e-05,
      "loss": 0.0604,
      "step": 42590
    },
    {
      "epoch": 4.983618066931898,
      "loss_main": 0.05848544090986252,
      "step": 42590
    },
    {
      "epoch": 4.984788205008191,
      "grad_norm": 0.20231421291828156,
      "learning_rate": 2.1285822861839966e-05,
      "loss": 0.0516,
      "step": 42600
    },
    {
      "epoch": 4.984788205008191,
      "loss_main": 0.045019689947366714,
      "step": 42600
    },
    {
      "epoch": 4.985958343084484,
      "grad_norm": 0.2623257339000702,
      "learning_rate": 2.126326682827331e-05,
      "loss": 0.0442,
      "step": 42610
    },
    {
      "epoch": 4.985958343084484,
      "loss_main": 0.06866758316755295,
      "step": 42610
    },
    {
      "epoch": 4.987128481160777,
      "grad_norm": 0.27746203541755676,
      "learning_rate": 2.124071952416744e-05,
      "loss": 0.0513,
      "step": 42620
    },
    {
      "epoch": 4.987128481160777,
      "loss_main": 0.05372991785407066,
      "step": 42620
    },
    {
      "epoch": 4.98829861923707,
      "grad_norm": 0.18145304918289185,
      "learning_rate": 2.1218180956371634e-05,
      "loss": 0.0469,
      "step": 42630
    },
    {
      "epoch": 4.98829861923707,
      "loss_main": 0.050973303616046906,
      "step": 42630
    },
    {
      "epoch": 4.989468757313363,
      "grad_norm": 0.238094761967659,
      "learning_rate": 2.119565113173252e-05,
      "loss": 0.0505,
      "step": 42640
    },
    {
      "epoch": 4.989468757313363,
      "loss_main": 0.05648844689130783,
      "step": 42640
    },
    {
      "epoch": 4.990638895389656,
      "grad_norm": 0.24819296598434448,
      "learning_rate": 2.1173130057094033e-05,
      "loss": 0.0492,
      "step": 42650
    },
    {
      "epoch": 4.990638895389656,
      "loss_main": 0.046256694942712784,
      "step": 42650
    },
    {
      "epoch": 4.991809033465949,
      "grad_norm": 0.25814855098724365,
      "learning_rate": 2.115061773929753e-05,
      "loss": 0.0466,
      "step": 42660
    },
    {
      "epoch": 4.991809033465949,
      "loss_main": 0.05934811383485794,
      "step": 42660
    },
    {
      "epoch": 4.992979171542242,
      "grad_norm": 0.2239948809146881,
      "learning_rate": 2.1128114185181623e-05,
      "loss": 0.0529,
      "step": 42670
    },
    {
      "epoch": 4.992979171542242,
      "loss_main": 0.07870561629533768,
      "step": 42670
    },
    {
      "epoch": 4.994149309618535,
      "grad_norm": 0.28521960973739624,
      "learning_rate": 2.1105619401582317e-05,
      "loss": 0.0474,
      "step": 42680
    },
    {
      "epoch": 4.994149309618535,
      "loss_main": 0.031135842204093933,
      "step": 42680
    },
    {
      "epoch": 4.995319447694828,
      "grad_norm": 0.2591242790222168,
      "learning_rate": 2.1083133395332928e-05,
      "loss": 0.0468,
      "step": 42690
    },
    {
      "epoch": 4.995319447694828,
      "loss_main": 0.0452461838722229,
      "step": 42690
    },
    {
      "epoch": 4.996489585771121,
      "grad_norm": 0.237141951918602,
      "learning_rate": 2.1060656173264082e-05,
      "loss": 0.0437,
      "step": 42700
    },
    {
      "epoch": 4.996489585771121,
      "loss_main": 0.04854324832558632,
      "step": 42700
    },
    {
      "epoch": 4.997659723847414,
      "grad_norm": 0.2732025384902954,
      "learning_rate": 2.103818774220383e-05,
      "loss": 0.0566,
      "step": 42710
    },
    {
      "epoch": 4.997659723847414,
      "loss_main": 0.05452629178762436,
      "step": 42710
    },
    {
      "epoch": 4.998829861923707,
      "grad_norm": 0.3565099537372589,
      "learning_rate": 2.1015728108977412e-05,
      "loss": 0.0549,
      "step": 42720
    },
    {
      "epoch": 4.998829861923707,
      "loss_main": 0.06697720289230347,
      "step": 42720
    },
    {
      "epoch": 5.0,
      "grad_norm": 0.2500549256801605,
      "learning_rate": 2.0993277280407548e-05,
      "loss": 0.0558,
      "step": 42730
    },
    {
      "epoch": 5.0,
      "loss_main": 0.04717600345611572,
      "step": 42730
    },
    {
      "epoch": 5.001170138076293,
      "grad_norm": 0.3039185404777527,
      "learning_rate": 2.0970835263314132e-05,
      "loss": 0.0518,
      "step": 42740
    },
    {
      "epoch": 5.001170138076293,
      "loss_main": 0.03329845517873764,
      "step": 42740
    },
    {
      "epoch": 5.002340276152586,
      "grad_norm": 0.2922772467136383,
      "learning_rate": 2.094840206451451e-05,
      "loss": 0.0498,
      "step": 42750
    },
    {
      "epoch": 5.002340276152586,
      "loss_main": 0.07599391043186188,
      "step": 42750
    },
    {
      "epoch": 5.003510414228879,
      "grad_norm": 0.23218956589698792,
      "learning_rate": 2.0925977690823273e-05,
      "loss": 0.054,
      "step": 42760
    },
    {
      "epoch": 5.003510414228879,
      "loss_main": 0.04934762045741081,
      "step": 42760
    },
    {
      "epoch": 5.004680552305172,
      "grad_norm": 0.21242257952690125,
      "learning_rate": 2.0903562149052364e-05,
      "loss": 0.0446,
      "step": 42770
    },
    {
      "epoch": 5.004680552305172,
      "loss_main": 0.036504440009593964,
      "step": 42770
    },
    {
      "epoch": 5.005850690381465,
      "grad_norm": 0.29949939250946045,
      "learning_rate": 2.0881155446011025e-05,
      "loss": 0.0497,
      "step": 42780
    },
    {
      "epoch": 5.005850690381465,
      "loss_main": 0.08120269328355789,
      "step": 42780
    },
    {
      "epoch": 5.007020828457758,
      "grad_norm": 0.16127841174602509,
      "learning_rate": 2.0858757588505823e-05,
      "loss": 0.0586,
      "step": 42790
    },
    {
      "epoch": 5.007020828457758,
      "loss_main": 0.0669563040137291,
      "step": 42790
    },
    {
      "epoch": 5.008190966534051,
      "grad_norm": 0.27483320236206055,
      "learning_rate": 2.0836368583340622e-05,
      "loss": 0.0508,
      "step": 42800
    },
    {
      "epoch": 5.008190966534051,
      "loss_main": 0.031236156821250916,
      "step": 42800
    },
    {
      "epoch": 5.009361104610344,
      "grad_norm": 0.2619432508945465,
      "learning_rate": 2.081398843731664e-05,
      "loss": 0.0519,
      "step": 42810
    },
    {
      "epoch": 5.009361104610344,
      "loss_main": 0.06171678751707077,
      "step": 42810
    },
    {
      "epoch": 5.010531242686637,
      "grad_norm": 0.26457691192626953,
      "learning_rate": 2.0791617157232357e-05,
      "loss": 0.0567,
      "step": 42820
    },
    {
      "epoch": 5.010531242686637,
      "loss_main": 0.07728177309036255,
      "step": 42820
    },
    {
      "epoch": 5.01170138076293,
      "grad_norm": 0.26015791296958923,
      "learning_rate": 2.0769254749883576e-05,
      "loss": 0.057,
      "step": 42830
    },
    {
      "epoch": 5.01170138076293,
      "loss_main": 0.035843875259160995,
      "step": 42830
    },
    {
      "epoch": 5.012871518839223,
      "grad_norm": 0.16650819778442383,
      "learning_rate": 2.0746901222063415e-05,
      "loss": 0.0441,
      "step": 42840
    },
    {
      "epoch": 5.012871518839223,
      "loss_main": 0.049600739032030106,
      "step": 42840
    },
    {
      "epoch": 5.014041656915516,
      "grad_norm": 0.27939435839653015,
      "learning_rate": 2.072455658056226e-05,
      "loss": 0.0543,
      "step": 42850
    },
    {
      "epoch": 5.014041656915516,
      "loss_main": 0.06355737894773483,
      "step": 42850
    },
    {
      "epoch": 5.015211794991809,
      "grad_norm": 0.20773597061634064,
      "learning_rate": 2.0702220832167873e-05,
      "loss": 0.0524,
      "step": 42860
    },
    {
      "epoch": 5.015211794991809,
      "loss_main": 0.030851194635033607,
      "step": 42860
    },
    {
      "epoch": 5.016381933068102,
      "grad_norm": 0.1999526023864746,
      "learning_rate": 2.0679893983665205e-05,
      "loss": 0.0474,
      "step": 42870
    },
    {
      "epoch": 5.016381933068102,
      "loss_main": 0.04473046585917473,
      "step": 42870
    },
    {
      "epoch": 5.017552071144395,
      "grad_norm": 0.2869850993156433,
      "learning_rate": 2.0657576041836622e-05,
      "loss": 0.0571,
      "step": 42880
    },
    {
      "epoch": 5.017552071144395,
      "loss_main": 0.056088369339704514,
      "step": 42880
    },
    {
      "epoch": 5.018722209220688,
      "grad_norm": 0.26918700337409973,
      "learning_rate": 2.0635267013461666e-05,
      "loss": 0.0559,
      "step": 42890
    },
    {
      "epoch": 5.018722209220688,
      "loss_main": 0.04815075919032097,
      "step": 42890
    },
    {
      "epoch": 5.019892347296981,
      "grad_norm": 0.21392224729061127,
      "learning_rate": 2.061296690531728e-05,
      "loss": 0.0509,
      "step": 42900
    },
    {
      "epoch": 5.019892347296981,
      "loss_main": 0.04345979541540146,
      "step": 42900
    },
    {
      "epoch": 5.021062485373274,
      "grad_norm": 0.20472827553749084,
      "learning_rate": 2.0590675724177622e-05,
      "loss": 0.0503,
      "step": 42910
    },
    {
      "epoch": 5.021062485373274,
      "loss_main": 0.05961373820900917,
      "step": 42910
    },
    {
      "epoch": 5.022232623449567,
      "grad_norm": 0.252240926027298,
      "learning_rate": 2.0568393476814167e-05,
      "loss": 0.0491,
      "step": 42920
    },
    {
      "epoch": 5.022232623449567,
      "loss_main": 0.04660701006650925,
      "step": 42920
    },
    {
      "epoch": 5.02340276152586,
      "grad_norm": 0.2179299145936966,
      "learning_rate": 2.0546120169995685e-05,
      "loss": 0.0482,
      "step": 42930
    },
    {
      "epoch": 5.02340276152586,
      "loss_main": 0.07199716567993164,
      "step": 42930
    },
    {
      "epoch": 5.024572899602153,
      "grad_norm": 0.27670371532440186,
      "learning_rate": 2.0523855810488214e-05,
      "loss": 0.0588,
      "step": 42940
    },
    {
      "epoch": 5.024572899602153,
      "loss_main": 0.03684580326080322,
      "step": 42940
    },
    {
      "epoch": 5.025743037678446,
      "grad_norm": 0.2481183260679245,
      "learning_rate": 2.050160040505505e-05,
      "loss": 0.0456,
      "step": 42950
    },
    {
      "epoch": 5.025743037678446,
      "loss_main": 0.05681673437356949,
      "step": 42950
    },
    {
      "epoch": 5.026913175754739,
      "grad_norm": 0.26458069682121277,
      "learning_rate": 2.0479353960456843e-05,
      "loss": 0.0556,
      "step": 42960
    },
    {
      "epoch": 5.026913175754739,
      "loss_main": 0.0456048958003521,
      "step": 42960
    },
    {
      "epoch": 5.028083313831032,
      "grad_norm": 0.23371082544326782,
      "learning_rate": 2.0457116483451456e-05,
      "loss": 0.0544,
      "step": 42970
    },
    {
      "epoch": 5.028083313831032,
      "loss_main": 0.06450579315423965,
      "step": 42970
    },
    {
      "epoch": 5.029253451907325,
      "grad_norm": 0.27028539776802063,
      "learning_rate": 2.0434887980794043e-05,
      "loss": 0.0529,
      "step": 42980
    },
    {
      "epoch": 5.029253451907325,
      "loss_main": 0.040438003838062286,
      "step": 42980
    },
    {
      "epoch": 5.030423589983618,
      "grad_norm": 0.29753217101097107,
      "learning_rate": 2.0412668459237043e-05,
      "loss": 0.0567,
      "step": 42990
    },
    {
      "epoch": 5.030423589983618,
      "loss_main": 0.06436635553836823,
      "step": 42990
    },
    {
      "epoch": 5.031593728059911,
      "grad_norm": 0.30156654119491577,
      "learning_rate": 2.039045792553016e-05,
      "loss": 0.0485,
      "step": 43000
    },
    {
      "epoch": 5.031593728059911,
      "loss_main": 0.040662191808223724,
      "step": 43000
    },
    {
      "epoch": 5.032763866136204,
      "grad_norm": 0.21829892694950104,
      "learning_rate": 2.036825638642036e-05,
      "loss": 0.0515,
      "step": 43010
    },
    {
      "epoch": 5.032763866136204,
      "loss_main": 0.06155695393681526,
      "step": 43010
    },
    {
      "epoch": 5.033934004212497,
      "grad_norm": 0.21235685050487518,
      "learning_rate": 2.0346063848651868e-05,
      "loss": 0.0478,
      "step": 43020
    },
    {
      "epoch": 5.033934004212497,
      "loss_main": 0.024890897795557976,
      "step": 43020
    },
    {
      "epoch": 5.03510414228879,
      "grad_norm": 0.24652554094791412,
      "learning_rate": 2.0323880318966254e-05,
      "loss": 0.0508,
      "step": 43030
    },
    {
      "epoch": 5.03510414228879,
      "loss_main": 0.05531212314963341,
      "step": 43030
    },
    {
      "epoch": 5.036274280365083,
      "grad_norm": 0.23960906267166138,
      "learning_rate": 2.030170580410221e-05,
      "loss": 0.0513,
      "step": 43040
    },
    {
      "epoch": 5.036274280365083,
      "loss_main": 0.025998298078775406,
      "step": 43040
    },
    {
      "epoch": 5.0374444184413765,
      "grad_norm": 0.18346737325191498,
      "learning_rate": 2.0279540310795837e-05,
      "loss": 0.0442,
      "step": 43050
    },
    {
      "epoch": 5.0374444184413765,
      "loss_main": 0.0857105404138565,
      "step": 43050
    },
    {
      "epoch": 5.038614556517669,
      "grad_norm": 0.15386301279067993,
      "learning_rate": 2.0257383845780365e-05,
      "loss": 0.047,
      "step": 43060
    },
    {
      "epoch": 5.038614556517669,
      "loss_main": 0.05022035911679268,
      "step": 43060
    },
    {
      "epoch": 5.039784694593962,
      "grad_norm": 0.3492809236049652,
      "learning_rate": 2.0235236415786384e-05,
      "loss": 0.052,
      "step": 43070
    },
    {
      "epoch": 5.039784694593962,
      "loss_main": 0.04581088200211525,
      "step": 43070
    },
    {
      "epoch": 5.040954832670255,
      "grad_norm": 0.1852271854877472,
      "learning_rate": 2.021309802754169e-05,
      "loss": 0.0482,
      "step": 43080
    },
    {
      "epoch": 5.040954832670255,
      "loss_main": 0.05573936924338341,
      "step": 43080
    },
    {
      "epoch": 5.042124970746548,
      "grad_norm": 0.18636590242385864,
      "learning_rate": 2.0190968687771332e-05,
      "loss": 0.0401,
      "step": 43090
    },
    {
      "epoch": 5.042124970746548,
      "loss_main": 0.023205338045954704,
      "step": 43090
    },
    {
      "epoch": 5.043295108822841,
      "grad_norm": 0.3210602104663849,
      "learning_rate": 2.016884840319763e-05,
      "loss": 0.0474,
      "step": 43100
    },
    {
      "epoch": 5.043295108822841,
      "loss_main": 0.04680624231696129,
      "step": 43100
    },
    {
      "epoch": 5.044465246899134,
      "grad_norm": 0.23570764064788818,
      "learning_rate": 2.0146737180540122e-05,
      "loss": 0.0471,
      "step": 43110
    },
    {
      "epoch": 5.044465246899134,
      "loss_main": 0.08001904934644699,
      "step": 43110
    },
    {
      "epoch": 5.045635384975427,
      "grad_norm": 0.18172165751457214,
      "learning_rate": 2.012463502651564e-05,
      "loss": 0.0431,
      "step": 43120
    },
    {
      "epoch": 5.045635384975427,
      "loss_main": 0.04299454763531685,
      "step": 43120
    },
    {
      "epoch": 5.04680552305172,
      "grad_norm": 0.2622050940990448,
      "learning_rate": 2.0102541947838228e-05,
      "loss": 0.0505,
      "step": 43130
    },
    {
      "epoch": 5.04680552305172,
      "loss_main": 0.03602030500769615,
      "step": 43130
    },
    {
      "epoch": 5.047975661128013,
      "grad_norm": 0.2789149582386017,
      "learning_rate": 2.0080457951219173e-05,
      "loss": 0.0429,
      "step": 43140
    },
    {
      "epoch": 5.047975661128013,
      "loss_main": 0.025053909048438072,
      "step": 43140
    },
    {
      "epoch": 5.0491457992043065,
      "grad_norm": 0.24715211987495422,
      "learning_rate": 2.0058383043367017e-05,
      "loss": 0.0522,
      "step": 43150
    },
    {
      "epoch": 5.0491457992043065,
      "loss_main": 0.06125326827168465,
      "step": 43150
    },
    {
      "epoch": 5.050315937280599,
      "grad_norm": 0.18865840137004852,
      "learning_rate": 2.0036317230987528e-05,
      "loss": 0.0434,
      "step": 43160
    },
    {
      "epoch": 5.050315937280599,
      "loss_main": 0.049309875816106796,
      "step": 43160
    },
    {
      "epoch": 5.051486075356892,
      "grad_norm": 0.25187650322914124,
      "learning_rate": 2.0014260520783696e-05,
      "loss": 0.0562,
      "step": 43170
    },
    {
      "epoch": 5.051486075356892,
      "loss_main": 0.05187486857175827,
      "step": 43170
    },
    {
      "epoch": 5.052656213433185,
      "grad_norm": 0.3038800358772278,
      "learning_rate": 1.9992212919455834e-05,
      "loss": 0.0589,
      "step": 43180
    },
    {
      "epoch": 5.052656213433185,
      "loss_main": 0.0590018555521965,
      "step": 43180
    },
    {
      "epoch": 5.053826351509478,
      "grad_norm": 0.2880614101886749,
      "learning_rate": 1.9970174433701333e-05,
      "loss": 0.0483,
      "step": 43190
    },
    {
      "epoch": 5.053826351509478,
      "loss_main": 0.029169542714953423,
      "step": 43190
    },
    {
      "epoch": 5.054996489585771,
      "grad_norm": 0.22078725695610046,
      "learning_rate": 1.9948145070214992e-05,
      "loss": 0.0448,
      "step": 43200
    },
    {
      "epoch": 5.054996489585771,
      "loss_main": 0.047335583716630936,
      "step": 43200
    },
    {
      "epoch": 5.056166627662064,
      "grad_norm": 0.23935750126838684,
      "learning_rate": 1.9926124835688663e-05,
      "loss": 0.0557,
      "step": 43210
    },
    {
      "epoch": 5.056166627662064,
      "loss_main": 0.04003524035215378,
      "step": 43210
    },
    {
      "epoch": 5.057336765738357,
      "grad_norm": 0.33062201738357544,
      "learning_rate": 1.9904113736811576e-05,
      "loss": 0.0559,
      "step": 43220
    },
    {
      "epoch": 5.057336765738357,
      "loss_main": 0.043372247368097305,
      "step": 43220
    },
    {
      "epoch": 5.05850690381465,
      "grad_norm": 0.25601083040237427,
      "learning_rate": 1.9882111780270096e-05,
      "loss": 0.0517,
      "step": 43230
    },
    {
      "epoch": 5.05850690381465,
      "loss_main": 0.04516775161027908,
      "step": 43230
    },
    {
      "epoch": 5.059677041890943,
      "grad_norm": 0.23331758379936218,
      "learning_rate": 1.986011897274784e-05,
      "loss": 0.0492,
      "step": 43240
    },
    {
      "epoch": 5.059677041890943,
      "loss_main": 0.05015220865607262,
      "step": 43240
    },
    {
      "epoch": 5.0608471799672365,
      "grad_norm": 0.22524331510066986,
      "learning_rate": 1.983813532092565e-05,
      "loss": 0.0471,
      "step": 43250
    },
    {
      "epoch": 5.0608471799672365,
      "loss_main": 0.058125775307416916,
      "step": 43250
    },
    {
      "epoch": 5.062017318043529,
      "grad_norm": 0.2953704595565796,
      "learning_rate": 1.981616083148155e-05,
      "loss": 0.0463,
      "step": 43260
    },
    {
      "epoch": 5.062017318043529,
      "loss_main": 0.053599707782268524,
      "step": 43260
    },
    {
      "epoch": 5.063187456119822,
      "grad_norm": 0.4201298952102661,
      "learning_rate": 1.9794195511090845e-05,
      "loss": 0.0537,
      "step": 43270
    },
    {
      "epoch": 5.063187456119822,
      "loss_main": 0.04061632230877876,
      "step": 43270
    },
    {
      "epoch": 5.064357594196115,
      "grad_norm": 0.22762379050254822,
      "learning_rate": 1.977223936642601e-05,
      "loss": 0.0582,
      "step": 43280
    },
    {
      "epoch": 5.064357594196115,
      "loss_main": 0.04615391045808792,
      "step": 43280
    },
    {
      "epoch": 5.065527732272408,
      "grad_norm": 0.21336032450199127,
      "learning_rate": 1.975029240415674e-05,
      "loss": 0.045,
      "step": 43290
    },
    {
      "epoch": 5.065527732272408,
      "loss_main": 0.07138048112392426,
      "step": 43290
    },
    {
      "epoch": 5.066697870348701,
      "grad_norm": 0.20489417016506195,
      "learning_rate": 1.9728354630949936e-05,
      "loss": 0.0544,
      "step": 43300
    },
    {
      "epoch": 5.066697870348701,
      "loss_main": 0.052589964121580124,
      "step": 43300
    },
    {
      "epoch": 5.067868008424994,
      "grad_norm": 0.18905071914196014,
      "learning_rate": 1.9706426053469716e-05,
      "loss": 0.0462,
      "step": 43310
    },
    {
      "epoch": 5.067868008424994,
      "loss_main": 0.08564426749944687,
      "step": 43310
    },
    {
      "epoch": 5.0690381465012875,
      "grad_norm": 0.22237437963485718,
      "learning_rate": 1.9684506678377396e-05,
      "loss": 0.0469,
      "step": 43320
    },
    {
      "epoch": 5.0690381465012875,
      "loss_main": 0.03908788040280342,
      "step": 43320
    },
    {
      "epoch": 5.07020828457758,
      "grad_norm": 0.3486138880252838,
      "learning_rate": 1.9662596512331544e-05,
      "loss": 0.0476,
      "step": 43330
    },
    {
      "epoch": 5.07020828457758,
      "loss_main": 0.03414740413427353,
      "step": 43330
    },
    {
      "epoch": 5.071378422653873,
      "grad_norm": 0.2145288735628128,
      "learning_rate": 1.964069556198782e-05,
      "loss": 0.0542,
      "step": 43340
    },
    {
      "epoch": 5.071378422653873,
      "loss_main": 0.02574520744383335,
      "step": 43340
    },
    {
      "epoch": 5.072548560730167,
      "grad_norm": 0.2622199058532715,
      "learning_rate": 1.9618803833999232e-05,
      "loss": 0.0537,
      "step": 43350
    },
    {
      "epoch": 5.072548560730167,
      "loss_main": 0.05571426823735237,
      "step": 43350
    },
    {
      "epoch": 5.073718698806459,
      "grad_norm": 0.23767513036727905,
      "learning_rate": 1.9596921335015838e-05,
      "loss": 0.049,
      "step": 43360
    },
    {
      "epoch": 5.073718698806459,
      "loss_main": 0.06385017931461334,
      "step": 43360
    },
    {
      "epoch": 5.074888836882752,
      "grad_norm": 0.23710143566131592,
      "learning_rate": 1.957504807168501e-05,
      "loss": 0.0592,
      "step": 43370
    },
    {
      "epoch": 5.074888836882752,
      "loss_main": 0.05288258194923401,
      "step": 43370
    },
    {
      "epoch": 5.076058974959045,
      "grad_norm": 0.2618010640144348,
      "learning_rate": 1.9553184050651253e-05,
      "loss": 0.0465,
      "step": 43380
    },
    {
      "epoch": 5.076058974959045,
      "loss_main": 0.07658139616250992,
      "step": 43380
    },
    {
      "epoch": 5.077229113035338,
      "grad_norm": 0.2516014873981476,
      "learning_rate": 1.953132927855628e-05,
      "loss": 0.058,
      "step": 43390
    },
    {
      "epoch": 5.077229113035338,
      "loss_main": 0.05690421536564827,
      "step": 43390
    },
    {
      "epoch": 5.078399251111631,
      "grad_norm": 0.2903784513473511,
      "learning_rate": 1.9509483762038995e-05,
      "loss": 0.0512,
      "step": 43400
    },
    {
      "epoch": 5.078399251111631,
      "loss_main": 0.056882113218307495,
      "step": 43400
    },
    {
      "epoch": 5.079569389187924,
      "grad_norm": 0.2346152663230896,
      "learning_rate": 1.9487647507735467e-05,
      "loss": 0.0481,
      "step": 43410
    },
    {
      "epoch": 5.079569389187924,
      "loss_main": 0.040582261979579926,
      "step": 43410
    },
    {
      "epoch": 5.0807395272642175,
      "grad_norm": 0.2027130126953125,
      "learning_rate": 1.9465820522279032e-05,
      "loss": 0.0427,
      "step": 43420
    },
    {
      "epoch": 5.0807395272642175,
      "loss_main": 0.0598437525331974,
      "step": 43420
    },
    {
      "epoch": 5.08190966534051,
      "grad_norm": 0.28954043984413147,
      "learning_rate": 1.9444002812300078e-05,
      "loss": 0.0551,
      "step": 43430
    },
    {
      "epoch": 5.08190966534051,
      "loss_main": 0.0462033674120903,
      "step": 43430
    },
    {
      "epoch": 5.083079803416803,
      "grad_norm": 0.31659653782844543,
      "learning_rate": 1.94221943844263e-05,
      "loss": 0.0452,
      "step": 43440
    },
    {
      "epoch": 5.083079803416803,
      "loss_main": 0.04807763546705246,
      "step": 43440
    },
    {
      "epoch": 5.084249941493097,
      "grad_norm": 0.3241695463657379,
      "learning_rate": 1.9400395245282515e-05,
      "loss": 0.0548,
      "step": 43450
    },
    {
      "epoch": 5.084249941493097,
      "loss_main": 0.049192436039447784,
      "step": 43450
    },
    {
      "epoch": 5.085420079569389,
      "grad_norm": 0.22757120430469513,
      "learning_rate": 1.937860540149071e-05,
      "loss": 0.05,
      "step": 43460
    },
    {
      "epoch": 5.085420079569389,
      "loss_main": 0.040288712829351425,
      "step": 43460
    },
    {
      "epoch": 5.086590217645682,
      "grad_norm": 0.1992252618074417,
      "learning_rate": 1.9356824859670082e-05,
      "loss": 0.0437,
      "step": 43470
    },
    {
      "epoch": 5.086590217645682,
      "loss_main": 0.04453875496983528,
      "step": 43470
    },
    {
      "epoch": 5.087760355721975,
      "grad_norm": 0.31663668155670166,
      "learning_rate": 1.9335053626436967e-05,
      "loss": 0.0499,
      "step": 43480
    },
    {
      "epoch": 5.087760355721975,
      "loss_main": 0.05784233286976814,
      "step": 43480
    },
    {
      "epoch": 5.0889304937982685,
      "grad_norm": 0.3650382459163666,
      "learning_rate": 1.9313291708404885e-05,
      "loss": 0.05,
      "step": 43490
    },
    {
      "epoch": 5.0889304937982685,
      "loss_main": 0.041361626237630844,
      "step": 43490
    },
    {
      "epoch": 5.090100631874561,
      "grad_norm": 0.2185351848602295,
      "learning_rate": 1.9291539112184587e-05,
      "loss": 0.0465,
      "step": 43500
    },
    {
      "epoch": 5.090100631874561,
      "loss_main": 0.02621455118060112,
      "step": 43500
    },
    {
      "epoch": 5.091270769950854,
      "grad_norm": 0.4037860929965973,
      "learning_rate": 1.9269795844383854e-05,
      "loss": 0.0576,
      "step": 43510
    },
    {
      "epoch": 5.091270769950854,
      "loss_main": 0.0542449951171875,
      "step": 43510
    },
    {
      "epoch": 5.0924409080271476,
      "grad_norm": 0.2894892692565918,
      "learning_rate": 1.9248061911607777e-05,
      "loss": 0.0513,
      "step": 43520
    },
    {
      "epoch": 5.0924409080271476,
      "loss_main": 0.01266916561871767,
      "step": 43520
    },
    {
      "epoch": 5.09361104610344,
      "grad_norm": 0.19526590406894684,
      "learning_rate": 1.9226337320458538e-05,
      "loss": 0.0566,
      "step": 43530
    },
    {
      "epoch": 5.09361104610344,
      "loss_main": 0.0637553483247757,
      "step": 43530
    },
    {
      "epoch": 5.094781184179733,
      "grad_norm": 0.23253293335437775,
      "learning_rate": 1.9204622077535488e-05,
      "loss": 0.0435,
      "step": 43540
    },
    {
      "epoch": 5.094781184179733,
      "loss_main": 0.026212576776742935,
      "step": 43540
    },
    {
      "epoch": 5.095951322256026,
      "grad_norm": 0.2255406677722931,
      "learning_rate": 1.9182916189435147e-05,
      "loss": 0.0428,
      "step": 43550
    },
    {
      "epoch": 5.095951322256026,
      "loss_main": 0.06049028038978577,
      "step": 43550
    },
    {
      "epoch": 5.097121460332319,
      "grad_norm": 0.2568108141422272,
      "learning_rate": 1.916121966275117e-05,
      "loss": 0.0504,
      "step": 43560
    },
    {
      "epoch": 5.097121460332319,
      "loss_main": 0.047216951847076416,
      "step": 43560
    },
    {
      "epoch": 5.098291598408612,
      "grad_norm": 0.21372230350971222,
      "learning_rate": 1.9139532504074443e-05,
      "loss": 0.0515,
      "step": 43570
    },
    {
      "epoch": 5.098291598408612,
      "loss_main": 0.06163950636982918,
      "step": 43570
    },
    {
      "epoch": 5.099461736484905,
      "grad_norm": 0.2710046172142029,
      "learning_rate": 1.9117854719992885e-05,
      "loss": 0.0522,
      "step": 43580
    },
    {
      "epoch": 5.099461736484905,
      "loss_main": 0.025971844792366028,
      "step": 43580
    },
    {
      "epoch": 5.1006318745611985,
      "grad_norm": 0.1888965368270874,
      "learning_rate": 1.9096186317091687e-05,
      "loss": 0.0394,
      "step": 43590
    },
    {
      "epoch": 5.1006318745611985,
      "loss_main": 0.038638245314359665,
      "step": 43590
    },
    {
      "epoch": 5.101802012637491,
      "grad_norm": 0.19547387957572937,
      "learning_rate": 1.9074527301953116e-05,
      "loss": 0.052,
      "step": 43600
    },
    {
      "epoch": 5.101802012637491,
      "loss_main": 0.03868373855948448,
      "step": 43600
    },
    {
      "epoch": 5.102972150713784,
      "grad_norm": 0.23606708645820618,
      "learning_rate": 1.9052877681156607e-05,
      "loss": 0.0489,
      "step": 43610
    },
    {
      "epoch": 5.102972150713784,
      "loss_main": 0.09250897914171219,
      "step": 43610
    },
    {
      "epoch": 5.104142288790078,
      "grad_norm": 0.32803940773010254,
      "learning_rate": 1.903123746127875e-05,
      "loss": 0.0581,
      "step": 43620
    },
    {
      "epoch": 5.104142288790078,
      "loss_main": 0.06613220274448395,
      "step": 43620
    },
    {
      "epoch": 5.10531242686637,
      "grad_norm": 0.22691218554973602,
      "learning_rate": 1.900960664889327e-05,
      "loss": 0.0481,
      "step": 43630
    },
    {
      "epoch": 5.10531242686637,
      "loss_main": 0.024118969216942787,
      "step": 43630
    },
    {
      "epoch": 5.106482564942663,
      "grad_norm": 0.24941803514957428,
      "learning_rate": 1.8987985250571015e-05,
      "loss": 0.0417,
      "step": 43640
    },
    {
      "epoch": 5.106482564942663,
      "loss_main": 0.044468462467193604,
      "step": 43640
    },
    {
      "epoch": 5.107652703018957,
      "grad_norm": 0.2096201330423355,
      "learning_rate": 1.8966373272880054e-05,
      "loss": 0.0558,
      "step": 43650
    },
    {
      "epoch": 5.107652703018957,
      "loss_main": 0.07088100165128708,
      "step": 43650
    },
    {
      "epoch": 5.108822841095249,
      "grad_norm": 0.2175111025571823,
      "learning_rate": 1.8944770722385462e-05,
      "loss": 0.0528,
      "step": 43660
    },
    {
      "epoch": 5.108822841095249,
      "loss_main": 0.03680654987692833,
      "step": 43660
    },
    {
      "epoch": 5.109992979171542,
      "grad_norm": 0.2665199637413025,
      "learning_rate": 1.8923177605649576e-05,
      "loss": 0.0457,
      "step": 43670
    },
    {
      "epoch": 5.109992979171542,
      "loss_main": 0.027503278106451035,
      "step": 43670
    },
    {
      "epoch": 5.111163117247835,
      "grad_norm": 0.22901296615600586,
      "learning_rate": 1.8901593929231802e-05,
      "loss": 0.0455,
      "step": 43680
    },
    {
      "epoch": 5.111163117247835,
      "loss_main": 0.04608706012368202,
      "step": 43680
    },
    {
      "epoch": 5.1123332553241285,
      "grad_norm": 0.24244284629821777,
      "learning_rate": 1.8880019699688684e-05,
      "loss": 0.0469,
      "step": 43690
    },
    {
      "epoch": 5.1123332553241285,
      "loss_main": 0.057459354400634766,
      "step": 43690
    },
    {
      "epoch": 5.113503393400421,
      "grad_norm": 0.24703145027160645,
      "learning_rate": 1.8858454923573904e-05,
      "loss": 0.0537,
      "step": 43700
    },
    {
      "epoch": 5.113503393400421,
      "loss_main": 0.04270915314555168,
      "step": 43700
    },
    {
      "epoch": 5.114673531476714,
      "grad_norm": 0.3784754276275635,
      "learning_rate": 1.8836899607438253e-05,
      "loss": 0.0498,
      "step": 43710
    },
    {
      "epoch": 5.114673531476714,
      "loss_main": 0.04215369001030922,
      "step": 43710
    },
    {
      "epoch": 5.115843669553008,
      "grad_norm": 0.24357199668884277,
      "learning_rate": 1.8815353757829723e-05,
      "loss": 0.0482,
      "step": 43720
    },
    {
      "epoch": 5.115843669553008,
      "loss_main": 0.04800804331898689,
      "step": 43720
    },
    {
      "epoch": 5.1170138076293,
      "grad_norm": 0.2532426714897156,
      "learning_rate": 1.879381738129331e-05,
      "loss": 0.0523,
      "step": 43730
    },
    {
      "epoch": 5.1170138076293,
      "loss_main": 0.08177264034748077,
      "step": 43730
    },
    {
      "epoch": 5.118183945705593,
      "grad_norm": 0.3178964853286743,
      "learning_rate": 1.8772290484371236e-05,
      "loss": 0.0497,
      "step": 43740
    },
    {
      "epoch": 5.118183945705593,
      "loss_main": 0.02560923807322979,
      "step": 43740
    },
    {
      "epoch": 5.119354083781886,
      "grad_norm": 0.3055214583873749,
      "learning_rate": 1.8750773073602795e-05,
      "loss": 0.0508,
      "step": 43750
    },
    {
      "epoch": 5.119354083781886,
      "loss_main": 0.0396115705370903,
      "step": 43750
    },
    {
      "epoch": 5.1205242218581795,
      "grad_norm": 0.2953964173793793,
      "learning_rate": 1.8729265155524405e-05,
      "loss": 0.0485,
      "step": 43760
    },
    {
      "epoch": 5.1205242218581795,
      "loss_main": 0.038959894329309464,
      "step": 43760
    },
    {
      "epoch": 5.121694359934472,
      "grad_norm": 0.22866283357143402,
      "learning_rate": 1.8707766736669607e-05,
      "loss": 0.0519,
      "step": 43770
    },
    {
      "epoch": 5.121694359934472,
      "loss_main": 0.0666784942150116,
      "step": 43770
    },
    {
      "epoch": 5.122864498010765,
      "grad_norm": 0.23997962474822998,
      "learning_rate": 1.8686277823569055e-05,
      "loss": 0.05,
      "step": 43780
    },
    {
      "epoch": 5.122864498010765,
      "loss_main": 0.042993348091840744,
      "step": 43780
    },
    {
      "epoch": 5.124034636087059,
      "grad_norm": 0.18671195209026337,
      "learning_rate": 1.8664798422750484e-05,
      "loss": 0.0441,
      "step": 43790
    },
    {
      "epoch": 5.124034636087059,
      "loss_main": 0.04466959834098816,
      "step": 43790
    },
    {
      "epoch": 5.125204774163351,
      "grad_norm": 0.22701771557331085,
      "learning_rate": 1.8643328540738832e-05,
      "loss": 0.0513,
      "step": 43800
    },
    {
      "epoch": 5.125204774163351,
      "loss_main": 0.03552485629916191,
      "step": 43800
    },
    {
      "epoch": 5.126374912239644,
      "grad_norm": 0.17109708487987518,
      "learning_rate": 1.862186818405601e-05,
      "loss": 0.0465,
      "step": 43810
    },
    {
      "epoch": 5.126374912239644,
      "loss_main": 0.05926750227808952,
      "step": 43810
    },
    {
      "epoch": 5.127545050315938,
      "grad_norm": 0.19383221864700317,
      "learning_rate": 1.8600417359221156e-05,
      "loss": 0.0502,
      "step": 43820
    },
    {
      "epoch": 5.127545050315938,
      "loss_main": 0.04643040522933006,
      "step": 43820
    },
    {
      "epoch": 5.12871518839223,
      "grad_norm": 0.2337336242198944,
      "learning_rate": 1.8578976072750454e-05,
      "loss": 0.0489,
      "step": 43830
    },
    {
      "epoch": 5.12871518839223,
      "loss_main": 0.03962186351418495,
      "step": 43830
    },
    {
      "epoch": 5.129885326468523,
      "grad_norm": 0.2412223070859909,
      "learning_rate": 1.8557544331157194e-05,
      "loss": 0.0444,
      "step": 43840
    },
    {
      "epoch": 5.129885326468523,
      "loss_main": 0.034173060208559036,
      "step": 43840
    },
    {
      "epoch": 5.131055464544816,
      "grad_norm": 0.263458251953125,
      "learning_rate": 1.8536122140951785e-05,
      "loss": 0.0537,
      "step": 43850
    },
    {
      "epoch": 5.131055464544816,
      "loss_main": 0.07423876225948334,
      "step": 43850
    },
    {
      "epoch": 5.1322256026211095,
      "grad_norm": 0.23331642150878906,
      "learning_rate": 1.8514709508641688e-05,
      "loss": 0.055,
      "step": 43860
    },
    {
      "epoch": 5.1322256026211095,
      "loss_main": 0.09125101566314697,
      "step": 43860
    },
    {
      "epoch": 5.133395740697402,
      "grad_norm": 0.22952252626419067,
      "learning_rate": 1.8493306440731555e-05,
      "loss": 0.0548,
      "step": 43870
    },
    {
      "epoch": 5.133395740697402,
      "loss_main": 0.07158686220645905,
      "step": 43870
    },
    {
      "epoch": 5.134565878773695,
      "grad_norm": 0.2739162743091583,
      "learning_rate": 1.8471912943723013e-05,
      "loss": 0.0512,
      "step": 43880
    },
    {
      "epoch": 5.134565878773695,
      "loss_main": 0.04394860193133354,
      "step": 43880
    },
    {
      "epoch": 5.135736016849989,
      "grad_norm": 0.1967085897922516,
      "learning_rate": 1.8450529024114894e-05,
      "loss": 0.0477,
      "step": 43890
    },
    {
      "epoch": 5.135736016849989,
      "loss_main": 0.06692691892385483,
      "step": 43890
    },
    {
      "epoch": 5.136906154926281,
      "grad_norm": 0.22410394251346588,
      "learning_rate": 1.842915468840301e-05,
      "loss": 0.0457,
      "step": 43900
    },
    {
      "epoch": 5.136906154926281,
      "loss_main": 0.10501468926668167,
      "step": 43900
    },
    {
      "epoch": 5.138076293002574,
      "grad_norm": 0.1867143213748932,
      "learning_rate": 1.840778994308037e-05,
      "loss": 0.0553,
      "step": 43910
    },
    {
      "epoch": 5.138076293002574,
      "loss_main": 0.03454264998435974,
      "step": 43910
    },
    {
      "epoch": 5.139246431078868,
      "grad_norm": 0.21484296023845673,
      "learning_rate": 1.8386434794637004e-05,
      "loss": 0.0447,
      "step": 43920
    },
    {
      "epoch": 5.139246431078868,
      "loss_main": 0.03461966663599014,
      "step": 43920
    },
    {
      "epoch": 5.14041656915516,
      "grad_norm": 0.25317418575286865,
      "learning_rate": 1.8365089249560034e-05,
      "loss": 0.039,
      "step": 43930
    },
    {
      "epoch": 5.14041656915516,
      "loss_main": 0.045133065432310104,
      "step": 43930
    },
    {
      "epoch": 5.141586707231453,
      "grad_norm": 0.2866024971008301,
      "learning_rate": 1.8343753314333683e-05,
      "loss": 0.0515,
      "step": 43940
    },
    {
      "epoch": 5.141586707231453,
      "loss_main": 0.05360095947980881,
      "step": 43940
    },
    {
      "epoch": 5.142756845307746,
      "grad_norm": 0.23193812370300293,
      "learning_rate": 1.8322426995439236e-05,
      "loss": 0.0522,
      "step": 43950
    },
    {
      "epoch": 5.142756845307746,
      "loss_main": 0.07784958183765411,
      "step": 43950
    },
    {
      "epoch": 5.1439269833840395,
      "grad_norm": 0.22349917888641357,
      "learning_rate": 1.8301110299355058e-05,
      "loss": 0.0616,
      "step": 43960
    },
    {
      "epoch": 5.1439269833840395,
      "loss_main": 0.033059198409318924,
      "step": 43960
    },
    {
      "epoch": 5.145097121460332,
      "grad_norm": 0.18892881274223328,
      "learning_rate": 1.8279803232556625e-05,
      "loss": 0.0489,
      "step": 43970
    },
    {
      "epoch": 5.145097121460332,
      "loss_main": 0.041373174637556076,
      "step": 43970
    },
    {
      "epoch": 5.146267259536625,
      "grad_norm": 0.20469699800014496,
      "learning_rate": 1.8258505801516444e-05,
      "loss": 0.0452,
      "step": 43980
    },
    {
      "epoch": 5.146267259536625,
      "loss_main": 0.04165874049067497,
      "step": 43980
    },
    {
      "epoch": 5.147437397612919,
      "grad_norm": 0.23531630635261536,
      "learning_rate": 1.8237218012704117e-05,
      "loss": 0.0469,
      "step": 43990
    },
    {
      "epoch": 5.147437397612919,
      "loss_main": 0.05714189261198044,
      "step": 43990
    },
    {
      "epoch": 5.148607535689211,
      "grad_norm": 0.24116110801696777,
      "learning_rate": 1.821593987258631e-05,
      "loss": 0.0509,
      "step": 44000
    },
    {
      "epoch": 5.148607535689211,
      "loss_main": 0.0471077524125576,
      "step": 44000
    },
    {
      "epoch": 5.149777673765504,
      "grad_norm": 0.23133230209350586,
      "learning_rate": 1.8194671387626744e-05,
      "loss": 0.055,
      "step": 44010
    },
    {
      "epoch": 5.149777673765504,
      "loss_main": 0.04859597235918045,
      "step": 44010
    },
    {
      "epoch": 5.150947811841798,
      "grad_norm": 0.3331318497657776,
      "learning_rate": 1.8173412564286276e-05,
      "loss": 0.0486,
      "step": 44020
    },
    {
      "epoch": 5.150947811841798,
      "loss_main": 0.04043463245034218,
      "step": 44020
    },
    {
      "epoch": 5.1521179499180905,
      "grad_norm": 0.324619859457016,
      "learning_rate": 1.8152163409022697e-05,
      "loss": 0.0551,
      "step": 44030
    },
    {
      "epoch": 5.1521179499180905,
      "loss_main": 0.09127966314554214,
      "step": 44030
    },
    {
      "epoch": 5.153288087994383,
      "grad_norm": 0.2876479923725128,
      "learning_rate": 1.8130923928291023e-05,
      "loss": 0.0482,
      "step": 44040
    },
    {
      "epoch": 5.153288087994383,
      "loss_main": 0.0746752992272377,
      "step": 44040
    },
    {
      "epoch": 5.154458226070676,
      "grad_norm": 0.22614163160324097,
      "learning_rate": 1.8109694128543163e-05,
      "loss": 0.051,
      "step": 44050
    },
    {
      "epoch": 5.154458226070676,
      "loss_main": 0.04413517937064171,
      "step": 44050
    },
    {
      "epoch": 5.15562836414697,
      "grad_norm": 0.25638923048973083,
      "learning_rate": 1.8088474016228237e-05,
      "loss": 0.0472,
      "step": 44060
    },
    {
      "epoch": 5.15562836414697,
      "loss_main": 0.038813065737485886,
      "step": 44060
    },
    {
      "epoch": 5.156798502223262,
      "grad_norm": 0.3105698227882385,
      "learning_rate": 1.8067263597792328e-05,
      "loss": 0.0499,
      "step": 44070
    },
    {
      "epoch": 5.156798502223262,
      "loss_main": 0.04199272766709328,
      "step": 44070
    },
    {
      "epoch": 5.157968640299555,
      "grad_norm": 0.29720449447631836,
      "learning_rate": 1.80460628796786e-05,
      "loss": 0.0511,
      "step": 44080
    },
    {
      "epoch": 5.157968640299555,
      "loss_main": 0.025310592725872993,
      "step": 44080
    },
    {
      "epoch": 5.159138778375849,
      "grad_norm": 0.24864855408668518,
      "learning_rate": 1.8024871868327276e-05,
      "loss": 0.0492,
      "step": 44090
    },
    {
      "epoch": 5.159138778375849,
      "loss_main": 0.056098196655511856,
      "step": 44090
    },
    {
      "epoch": 5.160308916452141,
      "grad_norm": 0.23482736945152283,
      "learning_rate": 1.8003690570175608e-05,
      "loss": 0.0486,
      "step": 44100
    },
    {
      "epoch": 5.160308916452141,
      "loss_main": 0.06354285031557083,
      "step": 44100
    },
    {
      "epoch": 5.161479054528434,
      "grad_norm": 0.2399025410413742,
      "learning_rate": 1.7982518991657943e-05,
      "loss": 0.0508,
      "step": 44110
    },
    {
      "epoch": 5.161479054528434,
      "loss_main": 0.0380815789103508,
      "step": 44110
    },
    {
      "epoch": 5.162649192604728,
      "grad_norm": 0.2478620558977127,
      "learning_rate": 1.7961357139205643e-05,
      "loss": 0.0396,
      "step": 44120
    },
    {
      "epoch": 5.162649192604728,
      "loss_main": 0.05585592985153198,
      "step": 44120
    },
    {
      "epoch": 5.1638193306810205,
      "grad_norm": 0.24201716482639313,
      "learning_rate": 1.7940205019247108e-05,
      "loss": 0.0417,
      "step": 44130
    },
    {
      "epoch": 5.1638193306810205,
      "loss_main": 0.07597099989652634,
      "step": 44130
    },
    {
      "epoch": 5.164989468757313,
      "grad_norm": 0.2775195837020874,
      "learning_rate": 1.79190626382078e-05,
      "loss": 0.0502,
      "step": 44140
    },
    {
      "epoch": 5.164989468757313,
      "loss_main": 0.06452007591724396,
      "step": 44140
    },
    {
      "epoch": 5.166159606833606,
      "grad_norm": 0.27141499519348145,
      "learning_rate": 1.7897930002510215e-05,
      "loss": 0.0511,
      "step": 44150
    },
    {
      "epoch": 5.166159606833606,
      "loss_main": 0.035126447677612305,
      "step": 44150
    },
    {
      "epoch": 5.1673297449099,
      "grad_norm": 0.2428719699382782,
      "learning_rate": 1.787680711857387e-05,
      "loss": 0.0484,
      "step": 44160
    },
    {
      "epoch": 5.1673297449099,
      "loss_main": 0.07018207013607025,
      "step": 44160
    },
    {
      "epoch": 5.168499882986192,
      "grad_norm": 0.35037583112716675,
      "learning_rate": 1.7855693992815398e-05,
      "loss": 0.0554,
      "step": 44170
    },
    {
      "epoch": 5.168499882986192,
      "loss_main": 0.04983764514327049,
      "step": 44170
    },
    {
      "epoch": 5.169670021062485,
      "grad_norm": 0.20539160072803497,
      "learning_rate": 1.7834590631648328e-05,
      "loss": 0.0455,
      "step": 44180
    },
    {
      "epoch": 5.169670021062485,
      "loss_main": 0.02606634981930256,
      "step": 44180
    },
    {
      "epoch": 5.170840159138779,
      "grad_norm": 0.30626463890075684,
      "learning_rate": 1.7813497041483384e-05,
      "loss": 0.0484,
      "step": 44190
    },
    {
      "epoch": 5.170840159138779,
      "loss_main": 0.03642549738287926,
      "step": 44190
    },
    {
      "epoch": 5.172010297215071,
      "grad_norm": 0.27976205945014954,
      "learning_rate": 1.779241322872817e-05,
      "loss": 0.0519,
      "step": 44200
    },
    {
      "epoch": 5.172010297215071,
      "loss_main": 0.04955881088972092,
      "step": 44200
    },
    {
      "epoch": 5.173180435291364,
      "grad_norm": 0.20276351273059845,
      "learning_rate": 1.777133919978744e-05,
      "loss": 0.0459,
      "step": 44210
    },
    {
      "epoch": 5.173180435291364,
      "loss_main": 0.09171693027019501,
      "step": 44210
    },
    {
      "epoch": 5.174350573367658,
      "grad_norm": 0.23709119856357574,
      "learning_rate": 1.7750274961062912e-05,
      "loss": 0.0608,
      "step": 44220
    },
    {
      "epoch": 5.174350573367658,
      "loss_main": 0.050308823585510254,
      "step": 44220
    },
    {
      "epoch": 5.1755207114439505,
      "grad_norm": 0.2915528118610382,
      "learning_rate": 1.772922051895335e-05,
      "loss": 0.0464,
      "step": 44230
    },
    {
      "epoch": 5.1755207114439505,
      "loss_main": 0.040315892547369,
      "step": 44230
    },
    {
      "epoch": 5.176690849520243,
      "grad_norm": 0.25510597229003906,
      "learning_rate": 1.770817587985453e-05,
      "loss": 0.0444,
      "step": 44240
    },
    {
      "epoch": 5.176690849520243,
      "loss_main": 0.03944411873817444,
      "step": 44240
    },
    {
      "epoch": 5.177860987596536,
      "grad_norm": 0.3130427598953247,
      "learning_rate": 1.7687141050159246e-05,
      "loss": 0.0544,
      "step": 44250
    },
    {
      "epoch": 5.177860987596536,
      "loss_main": 0.070608951151371,
      "step": 44250
    },
    {
      "epoch": 5.17903112567283,
      "grad_norm": 0.18431192636489868,
      "learning_rate": 1.7666116036257375e-05,
      "loss": 0.0396,
      "step": 44260
    },
    {
      "epoch": 5.17903112567283,
      "loss_main": 0.08091019093990326,
      "step": 44260
    },
    {
      "epoch": 5.180201263749122,
      "grad_norm": 0.2712399661540985,
      "learning_rate": 1.764510084453569e-05,
      "loss": 0.045,
      "step": 44270
    },
    {
      "epoch": 5.180201263749122,
      "loss_main": 0.026660149917006493,
      "step": 44270
    },
    {
      "epoch": 5.181371401825415,
      "grad_norm": 0.1961536854505539,
      "learning_rate": 1.76240954813781e-05,
      "loss": 0.0522,
      "step": 44280
    },
    {
      "epoch": 5.181371401825415,
      "loss_main": 0.025461209937930107,
      "step": 44280
    },
    {
      "epoch": 5.182541539901709,
      "grad_norm": 0.30173733830451965,
      "learning_rate": 1.7603099953165476e-05,
      "loss": 0.0483,
      "step": 44290
    },
    {
      "epoch": 5.182541539901709,
      "loss_main": 0.0687607154250145,
      "step": 44290
    },
    {
      "epoch": 5.1837116779780015,
      "grad_norm": 0.24242237210273743,
      "learning_rate": 1.7582114266275683e-05,
      "loss": 0.049,
      "step": 44300
    },
    {
      "epoch": 5.1837116779780015,
      "loss_main": 0.11566444486379623,
      "step": 44300
    },
    {
      "epoch": 5.184881816054294,
      "grad_norm": 0.4028049111366272,
      "learning_rate": 1.756113842708364e-05,
      "loss": 0.0531,
      "step": 44310
    },
    {
      "epoch": 5.184881816054294,
      "loss_main": 0.062136925756931305,
      "step": 44310
    },
    {
      "epoch": 5.186051954130588,
      "grad_norm": 0.2202310562133789,
      "learning_rate": 1.7540172441961245e-05,
      "loss": 0.0529,
      "step": 44320
    },
    {
      "epoch": 5.186051954130588,
      "loss_main": 0.0599057637155056,
      "step": 44320
    },
    {
      "epoch": 5.187222092206881,
      "grad_norm": 0.21072377264499664,
      "learning_rate": 1.7519216317277387e-05,
      "loss": 0.044,
      "step": 44330
    },
    {
      "epoch": 5.187222092206881,
      "loss_main": 0.04333196207880974,
      "step": 44330
    },
    {
      "epoch": 5.188392230283173,
      "grad_norm": 0.21567049622535706,
      "learning_rate": 1.7498270059398046e-05,
      "loss": 0.0519,
      "step": 44340
    },
    {
      "epoch": 5.188392230283173,
      "loss_main": 0.03600697964429855,
      "step": 44340
    },
    {
      "epoch": 5.189562368359466,
      "grad_norm": 0.20108546316623688,
      "learning_rate": 1.7477333674686062e-05,
      "loss": 0.0468,
      "step": 44350
    },
    {
      "epoch": 5.189562368359466,
      "loss_main": 0.06865309923887253,
      "step": 44350
    },
    {
      "epoch": 5.19073250643576,
      "grad_norm": 0.2970103323459625,
      "learning_rate": 1.745640716950142e-05,
      "loss": 0.0563,
      "step": 44360
    },
    {
      "epoch": 5.19073250643576,
      "loss_main": 0.05222691223025322,
      "step": 44360
    },
    {
      "epoch": 5.191902644512052,
      "grad_norm": 0.24556578695774078,
      "learning_rate": 1.7435490550201017e-05,
      "loss": 0.058,
      "step": 44370
    },
    {
      "epoch": 5.191902644512052,
      "loss_main": 0.10289812088012695,
      "step": 44370
    },
    {
      "epoch": 5.193072782588345,
      "grad_norm": 0.2891627252101898,
      "learning_rate": 1.7414583823138762e-05,
      "loss": 0.0429,
      "step": 44380
    },
    {
      "epoch": 5.193072782588345,
      "loss_main": 0.03957970812916756,
      "step": 44380
    },
    {
      "epoch": 5.194242920664639,
      "grad_norm": 0.23578965663909912,
      "learning_rate": 1.739368699466558e-05,
      "loss": 0.0495,
      "step": 44390
    },
    {
      "epoch": 5.194242920664639,
      "loss_main": 0.03932015225291252,
      "step": 44390
    },
    {
      "epoch": 5.1954130587409315,
      "grad_norm": 0.26870572566986084,
      "learning_rate": 1.737280007112935e-05,
      "loss": 0.049,
      "step": 44400
    },
    {
      "epoch": 5.1954130587409315,
      "loss_main": 0.024655302986502647,
      "step": 44400
    },
    {
      "epoch": 5.196583196817224,
      "grad_norm": 0.3058909475803375,
      "learning_rate": 1.735192305887502e-05,
      "loss": 0.0421,
      "step": 44410
    },
    {
      "epoch": 5.196583196817224,
      "loss_main": 0.037678711116313934,
      "step": 44410
    },
    {
      "epoch": 5.197753334893518,
      "grad_norm": 0.20848633348941803,
      "learning_rate": 1.733105596424441e-05,
      "loss": 0.0435,
      "step": 44420
    },
    {
      "epoch": 5.197753334893518,
      "loss_main": 0.03395652398467064,
      "step": 44420
    },
    {
      "epoch": 5.198923472969811,
      "grad_norm": 0.21287564933300018,
      "learning_rate": 1.7310198793576437e-05,
      "loss": 0.0577,
      "step": 44430
    },
    {
      "epoch": 5.198923472969811,
      "loss_main": 0.05642110854387283,
      "step": 44430
    },
    {
      "epoch": 5.200093611046103,
      "grad_norm": 0.19244924187660217,
      "learning_rate": 1.7289351553206952e-05,
      "loss": 0.0552,
      "step": 44440
    },
    {
      "epoch": 5.200093611046103,
      "loss_main": 0.03406848758459091,
      "step": 44440
    },
    {
      "epoch": 5.201263749122396,
      "grad_norm": 0.2693368196487427,
      "learning_rate": 1.7268514249468788e-05,
      "loss": 0.0546,
      "step": 44450
    },
    {
      "epoch": 5.201263749122396,
      "loss_main": 0.03537723049521446,
      "step": 44450
    },
    {
      "epoch": 5.20243388719869,
      "grad_norm": 0.3315080404281616,
      "learning_rate": 1.7247686888691765e-05,
      "loss": 0.0523,
      "step": 44460
    },
    {
      "epoch": 5.20243388719869,
      "loss_main": 0.08024132251739502,
      "step": 44460
    },
    {
      "epoch": 5.2036040252749824,
      "grad_norm": 0.2567274272441864,
      "learning_rate": 1.7226869477202694e-05,
      "loss": 0.0442,
      "step": 44470
    },
    {
      "epoch": 5.2036040252749824,
      "loss_main": 0.03770569711923599,
      "step": 44470
    },
    {
      "epoch": 5.204774163351275,
      "grad_norm": 0.19158720970153809,
      "learning_rate": 1.7206062021325336e-05,
      "loss": 0.0544,
      "step": 44480
    },
    {
      "epoch": 5.204774163351275,
      "loss_main": 0.06000719591975212,
      "step": 44480
    },
    {
      "epoch": 5.205944301427569,
      "grad_norm": 0.35123953223228455,
      "learning_rate": 1.7185264527380502e-05,
      "loss": 0.0558,
      "step": 44490
    },
    {
      "epoch": 5.205944301427569,
      "loss_main": 0.02746247872710228,
      "step": 44490
    },
    {
      "epoch": 5.2071144395038615,
      "grad_norm": 0.19088730216026306,
      "learning_rate": 1.716447700168584e-05,
      "loss": 0.0471,
      "step": 44500
    },
    {
      "epoch": 5.2071144395038615,
      "loss_main": 0.07925163209438324,
      "step": 44500
    },
    {
      "epoch": 5.208284577580154,
      "grad_norm": 0.2825894057750702,
      "learning_rate": 1.714369945055611e-05,
      "loss": 0.0591,
      "step": 44510
    },
    {
      "epoch": 5.208284577580154,
      "loss_main": 0.06704585254192352,
      "step": 44510
    },
    {
      "epoch": 5.209454715656447,
      "grad_norm": 0.16183432936668396,
      "learning_rate": 1.7122931880302968e-05,
      "loss": 0.0451,
      "step": 44520
    },
    {
      "epoch": 5.209454715656447,
      "loss_main": 0.03110663592815399,
      "step": 44520
    },
    {
      "epoch": 5.210624853732741,
      "grad_norm": 0.22265231609344482,
      "learning_rate": 1.710217429723505e-05,
      "loss": 0.047,
      "step": 44530
    },
    {
      "epoch": 5.210624853732741,
      "loss_main": 0.0658433586359024,
      "step": 44530
    },
    {
      "epoch": 5.211794991809033,
      "grad_norm": 0.23237864673137665,
      "learning_rate": 1.7081426707657972e-05,
      "loss": 0.0463,
      "step": 44540
    },
    {
      "epoch": 5.211794991809033,
      "loss_main": 0.045613668859004974,
      "step": 44540
    },
    {
      "epoch": 5.212965129885326,
      "grad_norm": 0.22229963541030884,
      "learning_rate": 1.7060689117874275e-05,
      "loss": 0.0478,
      "step": 44550
    },
    {
      "epoch": 5.212965129885326,
      "loss_main": 0.03639044240117073,
      "step": 44550
    },
    {
      "epoch": 5.21413526796162,
      "grad_norm": 0.2512832581996918,
      "learning_rate": 1.703996153418354e-05,
      "loss": 0.0419,
      "step": 44560
    },
    {
      "epoch": 5.21413526796162,
      "loss_main": 0.04038556292653084,
      "step": 44560
    },
    {
      "epoch": 5.2153054060379125,
      "grad_norm": 0.25939691066741943,
      "learning_rate": 1.7019243962882205e-05,
      "loss": 0.0433,
      "step": 44570
    },
    {
      "epoch": 5.2153054060379125,
      "loss_main": 0.05037177726626396,
      "step": 44570
    },
    {
      "epoch": 5.216475544114205,
      "grad_norm": 0.17970819771289825,
      "learning_rate": 1.6998536410263754e-05,
      "loss": 0.047,
      "step": 44580
    },
    {
      "epoch": 5.216475544114205,
      "loss_main": 0.040190439671278,
      "step": 44580
    },
    {
      "epoch": 5.217645682190499,
      "grad_norm": 0.29015448689460754,
      "learning_rate": 1.6977838882618596e-05,
      "loss": 0.0467,
      "step": 44590
    },
    {
      "epoch": 5.217645682190499,
      "loss_main": 0.028382111340761185,
      "step": 44590
    },
    {
      "epoch": 5.218815820266792,
      "grad_norm": 0.2939574718475342,
      "learning_rate": 1.6957151386234088e-05,
      "loss": 0.0459,
      "step": 44600
    },
    {
      "epoch": 5.218815820266792,
      "loss_main": 0.041745226830244064,
      "step": 44600
    },
    {
      "epoch": 5.219985958343084,
      "grad_norm": 0.3659052550792694,
      "learning_rate": 1.6936473927394536e-05,
      "loss": 0.0444,
      "step": 44610
    },
    {
      "epoch": 5.219985958343084,
      "loss_main": 0.0543937087059021,
      "step": 44610
    },
    {
      "epoch": 5.221156096419377,
      "grad_norm": 0.2573823630809784,
      "learning_rate": 1.6915806512381222e-05,
      "loss": 0.0481,
      "step": 44620
    },
    {
      "epoch": 5.221156096419377,
      "loss_main": 0.05426861718297005,
      "step": 44620
    },
    {
      "epoch": 5.222326234495671,
      "grad_norm": 0.21570737659931183,
      "learning_rate": 1.6895149147472344e-05,
      "loss": 0.0527,
      "step": 44630
    },
    {
      "epoch": 5.222326234495671,
      "loss_main": 0.04522716999053955,
      "step": 44630
    },
    {
      "epoch": 5.223496372571963,
      "grad_norm": 0.29173219203948975,
      "learning_rate": 1.6874501838943073e-05,
      "loss": 0.0554,
      "step": 44640
    },
    {
      "epoch": 5.223496372571963,
      "loss_main": 0.04122146591544151,
      "step": 44640
    },
    {
      "epoch": 5.224666510648256,
      "grad_norm": 0.21863004565238953,
      "learning_rate": 1.6853864593065506e-05,
      "loss": 0.05,
      "step": 44650
    },
    {
      "epoch": 5.224666510648256,
      "loss_main": 0.03847404196858406,
      "step": 44650
    },
    {
      "epoch": 5.22583664872455,
      "grad_norm": 0.22216682136058807,
      "learning_rate": 1.683323741610871e-05,
      "loss": 0.0565,
      "step": 44660
    },
    {
      "epoch": 5.22583664872455,
      "loss_main": 0.037244733422994614,
      "step": 44660
    },
    {
      "epoch": 5.2270067868008425,
      "grad_norm": 0.32301703095436096,
      "learning_rate": 1.6812620314338674e-05,
      "loss": 0.0557,
      "step": 44670
    },
    {
      "epoch": 5.2270067868008425,
      "loss_main": 0.02660461701452732,
      "step": 44670
    },
    {
      "epoch": 5.228176924877135,
      "grad_norm": 0.27969470620155334,
      "learning_rate": 1.6792013294018326e-05,
      "loss": 0.0458,
      "step": 44680
    },
    {
      "epoch": 5.228176924877135,
      "loss_main": 0.06732462346553802,
      "step": 44680
    },
    {
      "epoch": 5.229347062953429,
      "grad_norm": 0.22578254342079163,
      "learning_rate": 1.6771416361407526e-05,
      "loss": 0.0499,
      "step": 44690
    },
    {
      "epoch": 5.229347062953429,
      "loss_main": 0.03845943138003349,
      "step": 44690
    },
    {
      "epoch": 5.230517201029722,
      "grad_norm": 0.2184699922800064,
      "learning_rate": 1.675082952276308e-05,
      "loss": 0.0431,
      "step": 44700
    },
    {
      "epoch": 5.230517201029722,
      "loss_main": 0.015585090033710003,
      "step": 44700
    },
    {
      "epoch": 5.231687339106014,
      "grad_norm": 0.267723023891449,
      "learning_rate": 1.6730252784338757e-05,
      "loss": 0.0449,
      "step": 44710
    },
    {
      "epoch": 5.231687339106014,
      "loss_main": 0.055844273418188095,
      "step": 44710
    },
    {
      "epoch": 5.232857477182307,
      "grad_norm": 0.28748154640197754,
      "learning_rate": 1.6709686152385166e-05,
      "loss": 0.0446,
      "step": 44720
    },
    {
      "epoch": 5.232857477182307,
      "loss_main": 0.04048150032758713,
      "step": 44720
    },
    {
      "epoch": 5.234027615258601,
      "grad_norm": 0.2254641205072403,
      "learning_rate": 1.668912963314998e-05,
      "loss": 0.0507,
      "step": 44730
    },
    {
      "epoch": 5.234027615258601,
      "loss_main": 0.05513809248805046,
      "step": 44730
    },
    {
      "epoch": 5.2351977533348935,
      "grad_norm": 0.23743321001529694,
      "learning_rate": 1.6668583232877653e-05,
      "loss": 0.0554,
      "step": 44740
    },
    {
      "epoch": 5.2351977533348935,
      "loss_main": 0.059788815677165985,
      "step": 44740
    },
    {
      "epoch": 5.236367891411186,
      "grad_norm": 0.30339857935905457,
      "learning_rate": 1.6648046957809698e-05,
      "loss": 0.0591,
      "step": 44750
    },
    {
      "epoch": 5.236367891411186,
      "loss_main": 0.040197260677814484,
      "step": 44750
    },
    {
      "epoch": 5.23753802948748,
      "grad_norm": 0.28999119997024536,
      "learning_rate": 1.6627520814184462e-05,
      "loss": 0.0477,
      "step": 44760
    },
    {
      "epoch": 5.23753802948748,
      "loss_main": 0.04580327123403549,
      "step": 44760
    },
    {
      "epoch": 5.2387081675637726,
      "grad_norm": 0.1887199878692627,
      "learning_rate": 1.660700480823726e-05,
      "loss": 0.0467,
      "step": 44770
    },
    {
      "epoch": 5.2387081675637726,
      "loss_main": 0.027723968029022217,
      "step": 44770
    },
    {
      "epoch": 5.239878305640065,
      "grad_norm": 0.1934043914079666,
      "learning_rate": 1.65864989462003e-05,
      "loss": 0.0477,
      "step": 44780
    },
    {
      "epoch": 5.239878305640065,
      "loss_main": 0.06296650320291519,
      "step": 44780
    },
    {
      "epoch": 5.241048443716359,
      "grad_norm": 0.17064613103866577,
      "learning_rate": 1.656600323430273e-05,
      "loss": 0.0563,
      "step": 44790
    },
    {
      "epoch": 5.241048443716359,
      "loss_main": 0.05263170227408409,
      "step": 44790
    },
    {
      "epoch": 5.242218581792652,
      "grad_norm": 0.19820420444011688,
      "learning_rate": 1.654551767877059e-05,
      "loss": 0.0505,
      "step": 44800
    },
    {
      "epoch": 5.242218581792652,
      "loss_main": 0.04375656321644783,
      "step": 44800
    },
    {
      "epoch": 5.243388719868944,
      "grad_norm": 0.2047962099313736,
      "learning_rate": 1.6525042285826874e-05,
      "loss": 0.0494,
      "step": 44810
    },
    {
      "epoch": 5.243388719868944,
      "loss_main": 0.05297011882066727,
      "step": 44810
    },
    {
      "epoch": 5.244558857945237,
      "grad_norm": 0.2043815702199936,
      "learning_rate": 1.6504577061691468e-05,
      "loss": 0.058,
      "step": 44820
    },
    {
      "epoch": 5.244558857945237,
      "loss_main": 0.07451265305280685,
      "step": 44820
    },
    {
      "epoch": 5.245728996021531,
      "grad_norm": 0.21721920371055603,
      "learning_rate": 1.6484122012581143e-05,
      "loss": 0.05,
      "step": 44830
    },
    {
      "epoch": 5.245728996021531,
      "loss_main": 0.042693234980106354,
      "step": 44830
    },
    {
      "epoch": 5.2468991340978235,
      "grad_norm": 0.1823236048221588,
      "learning_rate": 1.6463677144709623e-05,
      "loss": 0.0488,
      "step": 44840
    },
    {
      "epoch": 5.2468991340978235,
      "loss_main": 0.04138806834816933,
      "step": 44840
    },
    {
      "epoch": 5.248069272174116,
      "grad_norm": 0.25746995210647583,
      "learning_rate": 1.6443242464287493e-05,
      "loss": 0.0484,
      "step": 44850
    },
    {
      "epoch": 5.248069272174116,
      "loss_main": 0.04545130953192711,
      "step": 44850
    },
    {
      "epoch": 5.24923941025041,
      "grad_norm": 0.21748261153697968,
      "learning_rate": 1.642281797752232e-05,
      "loss": 0.0461,
      "step": 44860
    },
    {
      "epoch": 5.24923941025041,
      "loss_main": 0.07337994873523712,
      "step": 44860
    },
    {
      "epoch": 5.250409548326703,
      "grad_norm": 0.2569553554058075,
      "learning_rate": 1.6402403690618456e-05,
      "loss": 0.0468,
      "step": 44870
    },
    {
      "epoch": 5.250409548326703,
      "loss_main": 0.060563717037439346,
      "step": 44870
    },
    {
      "epoch": 5.251579686402995,
      "grad_norm": 0.26613715291023254,
      "learning_rate": 1.6381999609777295e-05,
      "loss": 0.0454,
      "step": 44880
    },
    {
      "epoch": 5.251579686402995,
      "loss_main": 0.05241851136088371,
      "step": 44880
    },
    {
      "epoch": 5.252749824479289,
      "grad_norm": 0.26755619049072266,
      "learning_rate": 1.6361605741196983e-05,
      "loss": 0.0588,
      "step": 44890
    },
    {
      "epoch": 5.252749824479289,
      "loss_main": 0.04045059159398079,
      "step": 44890
    },
    {
      "epoch": 5.253919962555582,
      "grad_norm": 0.2750217616558075,
      "learning_rate": 1.63412220910727e-05,
      "loss": 0.0606,
      "step": 44900
    },
    {
      "epoch": 5.253919962555582,
      "loss_main": 0.053794343024492264,
      "step": 44900
    },
    {
      "epoch": 5.255090100631874,
      "grad_norm": 0.25644823908805847,
      "learning_rate": 1.6320848665596433e-05,
      "loss": 0.0528,
      "step": 44910
    },
    {
      "epoch": 5.255090100631874,
      "loss_main": 0.030549267306923866,
      "step": 44910
    },
    {
      "epoch": 5.256260238708167,
      "grad_norm": 0.18711057305335999,
      "learning_rate": 1.6300485470957095e-05,
      "loss": 0.0483,
      "step": 44920
    },
    {
      "epoch": 5.256260238708167,
      "loss_main": 0.04415912553668022,
      "step": 44920
    },
    {
      "epoch": 5.257430376784461,
      "grad_norm": 0.21329718828201294,
      "learning_rate": 1.6280132513340483e-05,
      "loss": 0.0443,
      "step": 44930
    },
    {
      "epoch": 5.257430376784461,
      "loss_main": 0.03155536204576492,
      "step": 44930
    },
    {
      "epoch": 5.2586005148607535,
      "grad_norm": 0.2178858071565628,
      "learning_rate": 1.62597897989293e-05,
      "loss": 0.0508,
      "step": 44940
    },
    {
      "epoch": 5.2586005148607535,
      "loss_main": 0.045019738376140594,
      "step": 44940
    },
    {
      "epoch": 5.259770652937046,
      "grad_norm": 0.2749827802181244,
      "learning_rate": 1.623945733390309e-05,
      "loss": 0.0494,
      "step": 44950
    },
    {
      "epoch": 5.259770652937046,
      "loss_main": 0.055937759578228,
      "step": 44950
    },
    {
      "epoch": 5.26094079101334,
      "grad_norm": 0.223941370844841,
      "learning_rate": 1.6219135124438374e-05,
      "loss": 0.0539,
      "step": 44960
    },
    {
      "epoch": 5.26094079101334,
      "loss_main": 0.0380093976855278,
      "step": 44960
    },
    {
      "epoch": 5.262110929089633,
      "grad_norm": 0.24804526567459106,
      "learning_rate": 1.6198823176708465e-05,
      "loss": 0.0404,
      "step": 44970
    },
    {
      "epoch": 5.262110929089633,
      "loss_main": 0.024147706106305122,
      "step": 44970
    },
    {
      "epoch": 5.263281067165925,
      "grad_norm": 0.37615054845809937,
      "learning_rate": 1.6178521496883613e-05,
      "loss": 0.0632,
      "step": 44980
    },
    {
      "epoch": 5.263281067165925,
      "loss_main": 0.03492236137390137,
      "step": 44980
    },
    {
      "epoch": 5.264451205242219,
      "grad_norm": 0.2254646122455597,
      "learning_rate": 1.6158230091130926e-05,
      "loss": 0.0422,
      "step": 44990
    },
    {
      "epoch": 5.264451205242219,
      "loss_main": 0.03283269330859184,
      "step": 44990
    },
    {
      "epoch": 5.265621343318512,
      "grad_norm": 0.310080349445343,
      "learning_rate": 1.613794896561438e-05,
      "loss": 0.0529,
      "step": 45000
    },
    {
      "epoch": 5.265621343318512,
      "loss_main": 0.04296209290623665,
      "step": 45000
    },
    {
      "epoch": 5.2667914813948045,
      "grad_norm": 0.214653879404068,
      "learning_rate": 1.6117678126494894e-05,
      "loss": 0.0426,
      "step": 45010
    },
    {
      "epoch": 5.2667914813948045,
      "loss_main": 0.049861665815114975,
      "step": 45010
    },
    {
      "epoch": 5.267961619471097,
      "grad_norm": 0.23095382750034332,
      "learning_rate": 1.6097417579930153e-05,
      "loss": 0.0487,
      "step": 45020
    },
    {
      "epoch": 5.267961619471097,
      "loss_main": 0.04659442976117134,
      "step": 45020
    },
    {
      "epoch": 5.269131757547391,
      "grad_norm": 0.22690251469612122,
      "learning_rate": 1.6077167332074834e-05,
      "loss": 0.0438,
      "step": 45030
    },
    {
      "epoch": 5.269131757547391,
      "loss_main": 0.049025945365428925,
      "step": 45030
    },
    {
      "epoch": 5.270301895623684,
      "grad_norm": 0.19952455163002014,
      "learning_rate": 1.605692738908037e-05,
      "loss": 0.0472,
      "step": 45040
    },
    {
      "epoch": 5.270301895623684,
      "loss_main": 0.06091398000717163,
      "step": 45040
    },
    {
      "epoch": 5.271472033699976,
      "grad_norm": 0.14908385276794434,
      "learning_rate": 1.6036697757095176e-05,
      "loss": 0.0544,
      "step": 45050
    },
    {
      "epoch": 5.271472033699976,
      "loss_main": 0.06356702744960785,
      "step": 45050
    },
    {
      "epoch": 5.27264217177627,
      "grad_norm": 0.2841247320175171,
      "learning_rate": 1.6016478442264428e-05,
      "loss": 0.0472,
      "step": 45060
    },
    {
      "epoch": 5.27264217177627,
      "loss_main": 0.036331892013549805,
      "step": 45060
    },
    {
      "epoch": 5.273812309852563,
      "grad_norm": 0.19343656301498413,
      "learning_rate": 1.599626945073026e-05,
      "loss": 0.0427,
      "step": 45070
    },
    {
      "epoch": 5.273812309852563,
      "loss_main": 0.04616229981184006,
      "step": 45070
    },
    {
      "epoch": 5.274982447928855,
      "grad_norm": 0.2920691967010498,
      "learning_rate": 1.597607078863162e-05,
      "loss": 0.0468,
      "step": 45080
    },
    {
      "epoch": 5.274982447928855,
      "loss_main": 0.03193759545683861,
      "step": 45080
    },
    {
      "epoch": 5.276152586005148,
      "grad_norm": 0.2384359985589981,
      "learning_rate": 1.595588246210432e-05,
      "loss": 0.046,
      "step": 45090
    },
    {
      "epoch": 5.276152586005148,
      "loss_main": 0.028359459713101387,
      "step": 45090
    },
    {
      "epoch": 5.277322724081442,
      "grad_norm": 0.25989776849746704,
      "learning_rate": 1.5935704477281048e-05,
      "loss": 0.0449,
      "step": 45100
    },
    {
      "epoch": 5.277322724081442,
      "loss_main": 0.058070939034223557,
      "step": 45100
    },
    {
      "epoch": 5.2784928621577345,
      "grad_norm": 0.24053369462490082,
      "learning_rate": 1.5915536840291323e-05,
      "loss": 0.0495,
      "step": 45110
    },
    {
      "epoch": 5.2784928621577345,
      "loss_main": 0.057966914027929306,
      "step": 45110
    },
    {
      "epoch": 5.279663000234027,
      "grad_norm": 0.26461049914360046,
      "learning_rate": 1.5895379557261576e-05,
      "loss": 0.0404,
      "step": 45120
    },
    {
      "epoch": 5.279663000234027,
      "loss_main": 0.039318766444921494,
      "step": 45120
    },
    {
      "epoch": 5.280833138310321,
      "grad_norm": 0.18955926597118378,
      "learning_rate": 1.5875232634315033e-05,
      "loss": 0.0436,
      "step": 45130
    },
    {
      "epoch": 5.280833138310321,
      "loss_main": 0.01869657263159752,
      "step": 45130
    },
    {
      "epoch": 5.282003276386614,
      "grad_norm": 0.2783031463623047,
      "learning_rate": 1.5855096077571812e-05,
      "loss": 0.0528,
      "step": 45140
    },
    {
      "epoch": 5.282003276386614,
      "loss_main": 0.04172766953706741,
      "step": 45140
    },
    {
      "epoch": 5.283173414462906,
      "grad_norm": 0.23709604144096375,
      "learning_rate": 1.5834969893148855e-05,
      "loss": 0.0408,
      "step": 45150
    },
    {
      "epoch": 5.283173414462906,
      "loss_main": 0.09142980724573135,
      "step": 45150
    },
    {
      "epoch": 5.2843435525392,
      "grad_norm": 0.19569629430770874,
      "learning_rate": 1.581485408715997e-05,
      "loss": 0.0477,
      "step": 45160
    },
    {
      "epoch": 5.2843435525392,
      "loss_main": 0.0376681350171566,
      "step": 45160
    },
    {
      "epoch": 5.285513690615493,
      "grad_norm": 0.23019087314605713,
      "learning_rate": 1.5794748665715785e-05,
      "loss": 0.0475,
      "step": 45170
    },
    {
      "epoch": 5.285513690615493,
      "loss_main": 0.03162214905023575,
      "step": 45170
    },
    {
      "epoch": 5.286683828691785,
      "grad_norm": 0.19974862039089203,
      "learning_rate": 1.5774653634923857e-05,
      "loss": 0.0456,
      "step": 45180
    },
    {
      "epoch": 5.286683828691785,
      "loss_main": 0.03715598210692406,
      "step": 45180
    },
    {
      "epoch": 5.287853966768079,
      "grad_norm": 0.17779093980789185,
      "learning_rate": 1.575456900088845e-05,
      "loss": 0.0416,
      "step": 45190
    },
    {
      "epoch": 5.287853966768079,
      "loss_main": 0.08773519098758698,
      "step": 45190
    },
    {
      "epoch": 5.289024104844372,
      "grad_norm": 0.23049235343933105,
      "learning_rate": 1.5734494769710816e-05,
      "loss": 0.0535,
      "step": 45200
    },
    {
      "epoch": 5.289024104844372,
      "loss_main": 0.07734671980142593,
      "step": 45200
    },
    {
      "epoch": 5.2901942429206645,
      "grad_norm": 0.38836580514907837,
      "learning_rate": 1.5714430947488912e-05,
      "loss": 0.0531,
      "step": 45210
    },
    {
      "epoch": 5.2901942429206645,
      "loss_main": 0.041667427867650986,
      "step": 45210
    },
    {
      "epoch": 5.291364380996957,
      "grad_norm": 0.20883303880691528,
      "learning_rate": 1.5694377540317645e-05,
      "loss": 0.0562,
      "step": 45220
    },
    {
      "epoch": 5.291364380996957,
      "loss_main": 0.02458697371184826,
      "step": 45220
    },
    {
      "epoch": 5.292534519073251,
      "grad_norm": 0.4100537598133087,
      "learning_rate": 1.5674334554288694e-05,
      "loss": 0.0493,
      "step": 45230
    },
    {
      "epoch": 5.292534519073251,
      "loss_main": 0.06684256345033646,
      "step": 45230
    },
    {
      "epoch": 5.293704657149544,
      "grad_norm": 0.22921089828014374,
      "learning_rate": 1.5654301995490582e-05,
      "loss": 0.0438,
      "step": 45240
    },
    {
      "epoch": 5.293704657149544,
      "loss_main": 0.04685112088918686,
      "step": 45240
    },
    {
      "epoch": 5.294874795225836,
      "grad_norm": 0.21276092529296875,
      "learning_rate": 1.5634279870008685e-05,
      "loss": 0.0457,
      "step": 45250
    },
    {
      "epoch": 5.294874795225836,
      "loss_main": 0.07255560904741287,
      "step": 45250
    },
    {
      "epoch": 5.29604493330213,
      "grad_norm": 0.23224133253097534,
      "learning_rate": 1.5614268183925174e-05,
      "loss": 0.0403,
      "step": 45260
    },
    {
      "epoch": 5.29604493330213,
      "loss_main": 0.03967484459280968,
      "step": 45260
    },
    {
      "epoch": 5.297215071378423,
      "grad_norm": 0.27451297640800476,
      "learning_rate": 1.5594266943319097e-05,
      "loss": 0.0498,
      "step": 45270
    },
    {
      "epoch": 5.297215071378423,
      "loss_main": 0.04492509365081787,
      "step": 45270
    },
    {
      "epoch": 5.2983852094547155,
      "grad_norm": 0.2709178626537323,
      "learning_rate": 1.5574276154266294e-05,
      "loss": 0.0517,
      "step": 45280
    },
    {
      "epoch": 5.2983852094547155,
      "loss_main": 0.03551974147558212,
      "step": 45280
    },
    {
      "epoch": 5.299555347531008,
      "grad_norm": 0.22181378304958344,
      "learning_rate": 1.5554295822839437e-05,
      "loss": 0.0518,
      "step": 45290
    },
    {
      "epoch": 5.299555347531008,
      "loss_main": 0.04748460277915001,
      "step": 45290
    },
    {
      "epoch": 5.300725485607302,
      "grad_norm": 0.29884177446365356,
      "learning_rate": 1.5534325955108025e-05,
      "loss": 0.0487,
      "step": 45300
    },
    {
      "epoch": 5.300725485607302,
      "loss_main": 0.04849625378847122,
      "step": 45300
    },
    {
      "epoch": 5.301895623683595,
      "grad_norm": 0.4015991687774658,
      "learning_rate": 1.5514366557138373e-05,
      "loss": 0.0502,
      "step": 45310
    },
    {
      "epoch": 5.301895623683595,
      "loss_main": 0.03219854086637497,
      "step": 45310
    },
    {
      "epoch": 5.303065761759887,
      "grad_norm": 0.3058519661426544,
      "learning_rate": 1.5494417634993602e-05,
      "loss": 0.0466,
      "step": 45320
    },
    {
      "epoch": 5.303065761759887,
      "loss_main": 0.050353534519672394,
      "step": 45320
    },
    {
      "epoch": 5.304235899836181,
      "grad_norm": 0.2478761076927185,
      "learning_rate": 1.547447919473372e-05,
      "loss": 0.0534,
      "step": 45330
    },
    {
      "epoch": 5.304235899836181,
      "loss_main": 0.03417003154754639,
      "step": 45330
    },
    {
      "epoch": 5.305406037912474,
      "grad_norm": 0.2700400948524475,
      "learning_rate": 1.5454551242415434e-05,
      "loss": 0.0415,
      "step": 45340
    },
    {
      "epoch": 5.305406037912474,
      "loss_main": 0.059184618294239044,
      "step": 45340
    },
    {
      "epoch": 5.306576175988766,
      "grad_norm": 0.1972811073064804,
      "learning_rate": 1.543463378409239e-05,
      "loss": 0.0458,
      "step": 45350
    },
    {
      "epoch": 5.306576175988766,
      "loss_main": 0.05401505157351494,
      "step": 45350
    },
    {
      "epoch": 5.30774631406506,
      "grad_norm": 0.211857408285141,
      "learning_rate": 1.541472682581493e-05,
      "loss": 0.0508,
      "step": 45360
    },
    {
      "epoch": 5.30774631406506,
      "loss_main": 0.040773067623376846,
      "step": 45360
    },
    {
      "epoch": 5.308916452141353,
      "grad_norm": 0.2606152892112732,
      "learning_rate": 1.5394830373630298e-05,
      "loss": 0.0524,
      "step": 45370
    },
    {
      "epoch": 5.308916452141353,
      "loss_main": 0.07334066927433014,
      "step": 45370
    },
    {
      "epoch": 5.3100865902176455,
      "grad_norm": 0.2083461731672287,
      "learning_rate": 1.5374944433582506e-05,
      "loss": 0.054,
      "step": 45380
    },
    {
      "epoch": 5.3100865902176455,
      "loss_main": 0.046675022691488266,
      "step": 45380
    },
    {
      "epoch": 5.311256728293939,
      "grad_norm": 0.23195211589336395,
      "learning_rate": 1.5355069011712375e-05,
      "loss": 0.0449,
      "step": 45390
    },
    {
      "epoch": 5.311256728293939,
      "loss_main": 0.04411373659968376,
      "step": 45390
    },
    {
      "epoch": 5.312426866370232,
      "grad_norm": 0.27808117866516113,
      "learning_rate": 1.5335204114057526e-05,
      "loss": 0.0549,
      "step": 45400
    },
    {
      "epoch": 5.312426866370232,
      "loss_main": 0.05324104428291321,
      "step": 45400
    },
    {
      "epoch": 5.313597004446525,
      "grad_norm": 0.22796383500099182,
      "learning_rate": 1.5315349746652387e-05,
      "loss": 0.0423,
      "step": 45410
    },
    {
      "epoch": 5.313597004446525,
      "loss_main": 0.04768436402082443,
      "step": 45410
    },
    {
      "epoch": 5.314767142522817,
      "grad_norm": 0.21913647651672363,
      "learning_rate": 1.5295505915528212e-05,
      "loss": 0.0494,
      "step": 45420
    },
    {
      "epoch": 5.314767142522817,
      "loss_main": 0.04396403580904007,
      "step": 45420
    },
    {
      "epoch": 5.315937280599111,
      "grad_norm": 0.24480070173740387,
      "learning_rate": 1.5275672626713024e-05,
      "loss": 0.0491,
      "step": 45430
    },
    {
      "epoch": 5.315937280599111,
      "loss_main": 0.061003949493169785,
      "step": 45430
    },
    {
      "epoch": 5.317107418675404,
      "grad_norm": 0.2156757116317749,
      "learning_rate": 1.5255849886231643e-05,
      "loss": 0.0473,
      "step": 45440
    },
    {
      "epoch": 5.317107418675404,
      "loss_main": 0.06309541314840317,
      "step": 45440
    },
    {
      "epoch": 5.3182775567516964,
      "grad_norm": 0.22020243108272552,
      "learning_rate": 1.523603770010571e-05,
      "loss": 0.0535,
      "step": 45450
    },
    {
      "epoch": 5.3182775567516964,
      "loss_main": 0.03317281976342201,
      "step": 45450
    },
    {
      "epoch": 5.31944769482799,
      "grad_norm": 0.22224865853786469,
      "learning_rate": 1.521623607435363e-05,
      "loss": 0.0497,
      "step": 45460
    },
    {
      "epoch": 5.31944769482799,
      "loss_main": 0.06710486859083176,
      "step": 45460
    },
    {
      "epoch": 5.320617832904283,
      "grad_norm": 0.3105260729789734,
      "learning_rate": 1.5196445014990612e-05,
      "loss": 0.0521,
      "step": 45470
    },
    {
      "epoch": 5.320617832904283,
      "loss_main": 0.03499298170208931,
      "step": 45470
    },
    {
      "epoch": 5.3217879709805755,
      "grad_norm": 0.22633545100688934,
      "learning_rate": 1.5176664528028672e-05,
      "loss": 0.0421,
      "step": 45480
    },
    {
      "epoch": 5.3217879709805755,
      "loss_main": 0.05570521578192711,
      "step": 45480
    },
    {
      "epoch": 5.322958109056868,
      "grad_norm": 0.19867084920406342,
      "learning_rate": 1.5156894619476574e-05,
      "loss": 0.0476,
      "step": 45490
    },
    {
      "epoch": 5.322958109056868,
      "loss_main": 0.039769504219293594,
      "step": 45490
    },
    {
      "epoch": 5.324128247133162,
      "grad_norm": 0.33840465545654297,
      "learning_rate": 1.5137135295339938e-05,
      "loss": 0.0489,
      "step": 45500
    },
    {
      "epoch": 5.324128247133162,
      "loss_main": 0.049131300300359726,
      "step": 45500
    },
    {
      "epoch": 5.325298385209455,
      "grad_norm": 0.23615097999572754,
      "learning_rate": 1.5117386561621073e-05,
      "loss": 0.0521,
      "step": 45510
    },
    {
      "epoch": 5.325298385209455,
      "loss_main": 0.03847035393118858,
      "step": 45510
    },
    {
      "epoch": 5.326468523285747,
      "grad_norm": 0.24426834285259247,
      "learning_rate": 1.5097648424319167e-05,
      "loss": 0.044,
      "step": 45520
    },
    {
      "epoch": 5.326468523285747,
      "loss_main": 0.04315510764718056,
      "step": 45520
    },
    {
      "epoch": 5.327638661362041,
      "grad_norm": 0.23831817507743835,
      "learning_rate": 1.5077920889430119e-05,
      "loss": 0.0478,
      "step": 45530
    },
    {
      "epoch": 5.327638661362041,
      "loss_main": 0.040100786834955215,
      "step": 45530
    },
    {
      "epoch": 5.328808799438334,
      "grad_norm": 0.29171478748321533,
      "learning_rate": 1.5058203962946644e-05,
      "loss": 0.0517,
      "step": 45540
    },
    {
      "epoch": 5.328808799438334,
      "loss_main": 0.03763468191027641,
      "step": 45540
    },
    {
      "epoch": 5.3299789375146265,
      "grad_norm": 0.19235460460186005,
      "learning_rate": 1.503849765085822e-05,
      "loss": 0.0487,
      "step": 45550
    },
    {
      "epoch": 5.3299789375146265,
      "loss_main": 0.04363863915205002,
      "step": 45550
    },
    {
      "epoch": 5.33114907559092,
      "grad_norm": 0.23050203919410706,
      "learning_rate": 1.501880195915109e-05,
      "loss": 0.0535,
      "step": 45560
    },
    {
      "epoch": 5.33114907559092,
      "loss_main": 0.03881457448005676,
      "step": 45560
    },
    {
      "epoch": 5.332319213667213,
      "grad_norm": 0.19286078214645386,
      "learning_rate": 1.499911689380833e-05,
      "loss": 0.0575,
      "step": 45570
    },
    {
      "epoch": 5.332319213667213,
      "loss_main": 0.04827369377017021,
      "step": 45570
    },
    {
      "epoch": 5.333489351743506,
      "grad_norm": 0.22299335896968842,
      "learning_rate": 1.4979442460809683e-05,
      "loss": 0.0453,
      "step": 45580
    },
    {
      "epoch": 5.333489351743506,
      "loss_main": 0.042236775159835815,
      "step": 45580
    },
    {
      "epoch": 5.334659489819799,
      "grad_norm": 0.22179444134235382,
      "learning_rate": 1.4959778666131763e-05,
      "loss": 0.0462,
      "step": 45590
    },
    {
      "epoch": 5.334659489819799,
      "loss_main": 0.08104479312896729,
      "step": 45590
    },
    {
      "epoch": 5.335829627896092,
      "grad_norm": 0.2290511131286621,
      "learning_rate": 1.4940125515747905e-05,
      "loss": 0.0542,
      "step": 45600
    },
    {
      "epoch": 5.335829627896092,
      "loss_main": 0.06057175621390343,
      "step": 45600
    },
    {
      "epoch": 5.336999765972385,
      "grad_norm": 0.2401386797428131,
      "learning_rate": 1.4920483015628211e-05,
      "loss": 0.0488,
      "step": 45610
    },
    {
      "epoch": 5.336999765972385,
      "loss_main": 0.036698322743177414,
      "step": 45610
    },
    {
      "epoch": 5.338169904048677,
      "grad_norm": 0.22240133583545685,
      "learning_rate": 1.490085117173956e-05,
      "loss": 0.052,
      "step": 45620
    },
    {
      "epoch": 5.338169904048677,
      "loss_main": 0.0601809099316597,
      "step": 45620
    },
    {
      "epoch": 5.339340042124971,
      "grad_norm": 0.25119560956954956,
      "learning_rate": 1.488122999004558e-05,
      "loss": 0.0471,
      "step": 45630
    },
    {
      "epoch": 5.339340042124971,
      "loss_main": 0.04882577434182167,
      "step": 45630
    },
    {
      "epoch": 5.340510180201264,
      "grad_norm": 0.285236120223999,
      "learning_rate": 1.486161947650666e-05,
      "loss": 0.0411,
      "step": 45640
    },
    {
      "epoch": 5.340510180201264,
      "loss_main": 0.03453505039215088,
      "step": 45640
    },
    {
      "epoch": 5.3416803182775565,
      "grad_norm": 0.23226910829544067,
      "learning_rate": 1.4842019637079995e-05,
      "loss": 0.0415,
      "step": 45650
    },
    {
      "epoch": 5.3416803182775565,
      "loss_main": 0.03795039653778076,
      "step": 45650
    },
    {
      "epoch": 5.34285045635385,
      "grad_norm": 0.3341044783592224,
      "learning_rate": 1.482243047771944e-05,
      "loss": 0.0492,
      "step": 45660
    },
    {
      "epoch": 5.34285045635385,
      "loss_main": 0.03747224807739258,
      "step": 45660
    },
    {
      "epoch": 5.344020594430143,
      "grad_norm": 0.2011668086051941,
      "learning_rate": 1.4802852004375712e-05,
      "loss": 0.0429,
      "step": 45670
    },
    {
      "epoch": 5.344020594430143,
      "loss_main": 0.04034886881709099,
      "step": 45670
    },
    {
      "epoch": 5.345190732506436,
      "grad_norm": 0.1939401626586914,
      "learning_rate": 1.4783284222996218e-05,
      "loss": 0.0543,
      "step": 45680
    },
    {
      "epoch": 5.345190732506436,
      "loss_main": 0.03304479643702507,
      "step": 45680
    },
    {
      "epoch": 5.346360870582728,
      "grad_norm": 0.17778272926807404,
      "learning_rate": 1.4763727139525135e-05,
      "loss": 0.0472,
      "step": 45690
    },
    {
      "epoch": 5.346360870582728,
      "loss_main": 0.05299404636025429,
      "step": 45690
    },
    {
      "epoch": 5.347531008659022,
      "grad_norm": 0.2548135221004486,
      "learning_rate": 1.4744180759903392e-05,
      "loss": 0.0513,
      "step": 45700
    },
    {
      "epoch": 5.347531008659022,
      "loss_main": 0.03194079175591469,
      "step": 45700
    },
    {
      "epoch": 5.348701146735315,
      "grad_norm": 0.26197588443756104,
      "learning_rate": 1.4724645090068635e-05,
      "loss": 0.048,
      "step": 45710
    },
    {
      "epoch": 5.348701146735315,
      "loss_main": 0.05741019919514656,
      "step": 45710
    },
    {
      "epoch": 5.3498712848116075,
      "grad_norm": 0.22268696129322052,
      "learning_rate": 1.4705120135955341e-05,
      "loss": 0.0512,
      "step": 45720
    },
    {
      "epoch": 5.3498712848116075,
      "loss_main": 0.03246517851948738,
      "step": 45720
    },
    {
      "epoch": 5.351041422887901,
      "grad_norm": 0.16336481273174286,
      "learning_rate": 1.4685605903494614e-05,
      "loss": 0.0401,
      "step": 45730
    },
    {
      "epoch": 5.351041422887901,
      "loss_main": 0.059315431863069534,
      "step": 45730
    },
    {
      "epoch": 5.352211560964194,
      "grad_norm": 0.23248237371444702,
      "learning_rate": 1.46661023986144e-05,
      "loss": 0.0446,
      "step": 45740
    },
    {
      "epoch": 5.352211560964194,
      "loss_main": 0.07783442735671997,
      "step": 45740
    },
    {
      "epoch": 5.3533816990404866,
      "grad_norm": 0.20814765989780426,
      "learning_rate": 1.4646609627239344e-05,
      "loss": 0.0552,
      "step": 45750
    },
    {
      "epoch": 5.3533816990404866,
      "loss_main": 0.04472079128026962,
      "step": 45750
    },
    {
      "epoch": 5.35455183711678,
      "grad_norm": 0.2346809208393097,
      "learning_rate": 1.4627127595290835e-05,
      "loss": 0.0415,
      "step": 45760
    },
    {
      "epoch": 5.35455183711678,
      "loss_main": 0.052041955292224884,
      "step": 45760
    },
    {
      "epoch": 5.355721975193073,
      "grad_norm": 0.2797066271305084,
      "learning_rate": 1.460765630868699e-05,
      "loss": 0.0477,
      "step": 45770
    },
    {
      "epoch": 5.355721975193073,
      "loss_main": 0.0406893827021122,
      "step": 45770
    },
    {
      "epoch": 5.356892113269366,
      "grad_norm": 0.3655896782875061,
      "learning_rate": 1.4588195773342678e-05,
      "loss": 0.054,
      "step": 45780
    },
    {
      "epoch": 5.356892113269366,
      "loss_main": 0.03972044959664345,
      "step": 45780
    },
    {
      "epoch": 5.358062251345658,
      "grad_norm": 0.24266166985034943,
      "learning_rate": 1.4568745995169485e-05,
      "loss": 0.0409,
      "step": 45790
    },
    {
      "epoch": 5.358062251345658,
      "loss_main": 0.04525836557149887,
      "step": 45790
    },
    {
      "epoch": 5.359232389421952,
      "grad_norm": 0.30673947930336,
      "learning_rate": 1.4549306980075778e-05,
      "loss": 0.046,
      "step": 45800
    },
    {
      "epoch": 5.359232389421952,
      "loss_main": 0.042512595653533936,
      "step": 45800
    },
    {
      "epoch": 5.360402527498245,
      "grad_norm": 0.252470463514328,
      "learning_rate": 1.4529878733966557e-05,
      "loss": 0.0454,
      "step": 45810
    },
    {
      "epoch": 5.360402527498245,
      "loss_main": 0.028441444039344788,
      "step": 45810
    },
    {
      "epoch": 5.3615726655745375,
      "grad_norm": 0.19629552960395813,
      "learning_rate": 1.4510461262743658e-05,
      "loss": 0.052,
      "step": 45820
    },
    {
      "epoch": 5.3615726655745375,
      "loss_main": 0.07697049528360367,
      "step": 45820
    },
    {
      "epoch": 5.362742803650831,
      "grad_norm": 0.2625454366207123,
      "learning_rate": 1.4491054572305585e-05,
      "loss": 0.055,
      "step": 45830
    },
    {
      "epoch": 5.362742803650831,
      "loss_main": 0.05081918090581894,
      "step": 45830
    },
    {
      "epoch": 5.363912941727124,
      "grad_norm": 0.2551921010017395,
      "learning_rate": 1.4471658668547566e-05,
      "loss": 0.0483,
      "step": 45840
    },
    {
      "epoch": 5.363912941727124,
      "loss_main": 0.06608855724334717,
      "step": 45840
    },
    {
      "epoch": 5.365083079803417,
      "grad_norm": 0.26965171098709106,
      "learning_rate": 1.4452273557361579e-05,
      "loss": 0.0525,
      "step": 45850
    },
    {
      "epoch": 5.365083079803417,
      "loss_main": 0.07799903303384781,
      "step": 45850
    },
    {
      "epoch": 5.36625321787971,
      "grad_norm": 0.3775397837162018,
      "learning_rate": 1.4432899244636282e-05,
      "loss": 0.0457,
      "step": 45860
    },
    {
      "epoch": 5.36625321787971,
      "loss_main": 0.0600956492125988,
      "step": 45860
    },
    {
      "epoch": 5.367423355956003,
      "grad_norm": 0.21699483692646027,
      "learning_rate": 1.4413535736257134e-05,
      "loss": 0.0464,
      "step": 45870
    },
    {
      "epoch": 5.367423355956003,
      "loss_main": 0.0635027140378952,
      "step": 45870
    },
    {
      "epoch": 5.368593494032296,
      "grad_norm": 0.5754870176315308,
      "learning_rate": 1.439418303810619e-05,
      "loss": 0.0464,
      "step": 45880
    },
    {
      "epoch": 5.368593494032296,
      "loss_main": 0.05701582878828049,
      "step": 45880
    },
    {
      "epoch": 5.369763632108588,
      "grad_norm": 0.20939214527606964,
      "learning_rate": 1.4374841156062352e-05,
      "loss": 0.0552,
      "step": 45890
    },
    {
      "epoch": 5.369763632108588,
      "loss_main": 0.052336037158966064,
      "step": 45890
    },
    {
      "epoch": 5.370933770184882,
      "grad_norm": 0.28619131445884705,
      "learning_rate": 1.4355510096001112e-05,
      "loss": 0.0472,
      "step": 45900
    },
    {
      "epoch": 5.370933770184882,
      "loss_main": 0.032667018473148346,
      "step": 45900
    },
    {
      "epoch": 5.372103908261175,
      "grad_norm": 0.2593744993209839,
      "learning_rate": 1.4336189863794786e-05,
      "loss": 0.0478,
      "step": 45910
    },
    {
      "epoch": 5.372103908261175,
      "loss_main": 0.06885788589715958,
      "step": 45910
    },
    {
      "epoch": 5.3732740463374675,
      "grad_norm": 0.1831463724374771,
      "learning_rate": 1.4316880465312327e-05,
      "loss": 0.0446,
      "step": 45920
    },
    {
      "epoch": 5.3732740463374675,
      "loss_main": 0.03153777867555618,
      "step": 45920
    },
    {
      "epoch": 5.374444184413761,
      "grad_norm": 0.23114876449108124,
      "learning_rate": 1.4297581906419426e-05,
      "loss": 0.0455,
      "step": 45930
    },
    {
      "epoch": 5.374444184413761,
      "loss_main": 0.03336915001273155,
      "step": 45930
    },
    {
      "epoch": 5.375614322490054,
      "grad_norm": 0.20121726393699646,
      "learning_rate": 1.4278294192978475e-05,
      "loss": 0.0477,
      "step": 45940
    },
    {
      "epoch": 5.375614322490054,
      "loss_main": 0.0278784167021513,
      "step": 45940
    },
    {
      "epoch": 5.376784460566347,
      "grad_norm": 0.207087904214859,
      "learning_rate": 1.4259017330848574e-05,
      "loss": 0.052,
      "step": 45950
    },
    {
      "epoch": 5.376784460566347,
      "loss_main": 0.039890289306640625,
      "step": 45950
    },
    {
      "epoch": 5.37795459864264,
      "grad_norm": 0.22164694964885712,
      "learning_rate": 1.4239751325885498e-05,
      "loss": 0.0488,
      "step": 45960
    },
    {
      "epoch": 5.37795459864264,
      "loss_main": 0.04084863141179085,
      "step": 45960
    },
    {
      "epoch": 5.379124736718933,
      "grad_norm": 0.3169727325439453,
      "learning_rate": 1.4220496183941795e-05,
      "loss": 0.0437,
      "step": 45970
    },
    {
      "epoch": 5.379124736718933,
      "loss_main": 0.04202587157487869,
      "step": 45970
    },
    {
      "epoch": 5.380294874795226,
      "grad_norm": 0.2204863727092743,
      "learning_rate": 1.4201251910866648e-05,
      "loss": 0.0435,
      "step": 45980
    },
    {
      "epoch": 5.380294874795226,
      "loss_main": 0.04628748446702957,
      "step": 45980
    },
    {
      "epoch": 5.3814650128715185,
      "grad_norm": 0.17957693338394165,
      "learning_rate": 1.4182018512505957e-05,
      "loss": 0.0481,
      "step": 45990
    },
    {
      "epoch": 5.3814650128715185,
      "loss_main": 0.06172976642847061,
      "step": 45990
    },
    {
      "epoch": 5.382635150947812,
      "grad_norm": 0.45732712745666504,
      "learning_rate": 1.4162795994702327e-05,
      "loss": 0.048,
      "step": 46000
    },
    {
      "epoch": 5.382635150947812,
      "loss_main": 0.03394227474927902,
      "step": 46000
    },
    {
      "epoch": 5.383805289024105,
      "grad_norm": 0.191210076212883,
      "learning_rate": 1.4143584363295032e-05,
      "loss": 0.0485,
      "step": 46010
    },
    {
      "epoch": 5.383805289024105,
      "loss_main": 0.07079701125621796,
      "step": 46010
    },
    {
      "epoch": 5.384975427100398,
      "grad_norm": 0.23807986080646515,
      "learning_rate": 1.4124383624120101e-05,
      "loss": 0.0538,
      "step": 46020
    },
    {
      "epoch": 5.384975427100398,
      "loss_main": 0.04671346768736839,
      "step": 46020
    },
    {
      "epoch": 5.386145565176691,
      "grad_norm": 0.216008260846138,
      "learning_rate": 1.4105193783010151e-05,
      "loss": 0.0464,
      "step": 46030
    },
    {
      "epoch": 5.386145565176691,
      "loss_main": 0.04214731231331825,
      "step": 46030
    },
    {
      "epoch": 5.387315703252984,
      "grad_norm": 0.22079642117023468,
      "learning_rate": 1.4086014845794621e-05,
      "loss": 0.0454,
      "step": 46040
    },
    {
      "epoch": 5.387315703252984,
      "loss_main": 0.036470405757427216,
      "step": 46040
    },
    {
      "epoch": 5.388485841329277,
      "grad_norm": 0.28286564350128174,
      "learning_rate": 1.4066846818299489e-05,
      "loss": 0.0482,
      "step": 46050
    },
    {
      "epoch": 5.388485841329277,
      "loss_main": 0.04106965288519859,
      "step": 46050
    },
    {
      "epoch": 5.389655979405569,
      "grad_norm": 0.1855912059545517,
      "learning_rate": 1.4047689706347555e-05,
      "loss": 0.0467,
      "step": 46060
    },
    {
      "epoch": 5.389655979405569,
      "loss_main": 0.036940038204193115,
      "step": 46060
    },
    {
      "epoch": 5.390826117481863,
      "grad_norm": 0.2721445560455322,
      "learning_rate": 1.402854351575822e-05,
      "loss": 0.0518,
      "step": 46070
    },
    {
      "epoch": 5.390826117481863,
      "loss_main": 0.06046072393655777,
      "step": 46070
    },
    {
      "epoch": 5.391996255558156,
      "grad_norm": 0.2982903718948364,
      "learning_rate": 1.4009408252347588e-05,
      "loss": 0.0516,
      "step": 46080
    },
    {
      "epoch": 5.391996255558156,
      "loss_main": 0.03295820578932762,
      "step": 46080
    },
    {
      "epoch": 5.3931663936344485,
      "grad_norm": 0.2745293974876404,
      "learning_rate": 1.399028392192846e-05,
      "loss": 0.0484,
      "step": 46090
    },
    {
      "epoch": 5.3931663936344485,
      "loss_main": 0.0402260385453701,
      "step": 46090
    },
    {
      "epoch": 5.394336531710742,
      "grad_norm": 0.23623234033584595,
      "learning_rate": 1.397117053031029e-05,
      "loss": 0.0602,
      "step": 46100
    },
    {
      "epoch": 5.394336531710742,
      "loss_main": 0.03875051811337471,
      "step": 46100
    },
    {
      "epoch": 5.395506669787035,
      "grad_norm": 0.24777497351169586,
      "learning_rate": 1.3952068083299213e-05,
      "loss": 0.0464,
      "step": 46110
    },
    {
      "epoch": 5.395506669787035,
      "loss_main": 0.040881041437387466,
      "step": 46110
    },
    {
      "epoch": 5.396676807863328,
      "grad_norm": 0.45290690660476685,
      "learning_rate": 1.3932976586698082e-05,
      "loss": 0.055,
      "step": 46120
    },
    {
      "epoch": 5.396676807863328,
      "loss_main": 0.05960379168391228,
      "step": 46120
    },
    {
      "epoch": 5.397846945939621,
      "grad_norm": 0.2330474704504013,
      "learning_rate": 1.3913896046306363e-05,
      "loss": 0.0458,
      "step": 46130
    },
    {
      "epoch": 5.397846945939621,
      "loss_main": 0.07078489661216736,
      "step": 46130
    },
    {
      "epoch": 5.399017084015914,
      "grad_norm": 0.2760751247406006,
      "learning_rate": 1.389482646792023e-05,
      "loss": 0.0587,
      "step": 46140
    },
    {
      "epoch": 5.399017084015914,
      "loss_main": 0.044151127338409424,
      "step": 46140
    },
    {
      "epoch": 5.400187222092207,
      "grad_norm": 0.3172511160373688,
      "learning_rate": 1.387576785733251e-05,
      "loss": 0.0486,
      "step": 46150
    },
    {
      "epoch": 5.400187222092207,
      "loss_main": 0.06399596482515335,
      "step": 46150
    },
    {
      "epoch": 5.4013573601685,
      "grad_norm": 0.24011538922786713,
      "learning_rate": 1.3856720220332703e-05,
      "loss": 0.0472,
      "step": 46160
    },
    {
      "epoch": 5.4013573601685,
      "loss_main": 0.08258985728025436,
      "step": 46160
    },
    {
      "epoch": 5.402527498244793,
      "grad_norm": 0.2604790925979614,
      "learning_rate": 1.383768356270701e-05,
      "loss": 0.0481,
      "step": 46170
    },
    {
      "epoch": 5.402527498244793,
      "loss_main": 0.051935579627752304,
      "step": 46170
    },
    {
      "epoch": 5.403697636321086,
      "grad_norm": 0.33548596501350403,
      "learning_rate": 1.3818657890238207e-05,
      "loss": 0.055,
      "step": 46180
    },
    {
      "epoch": 5.403697636321086,
      "loss_main": 0.035291772335767746,
      "step": 46180
    },
    {
      "epoch": 5.4048677743973785,
      "grad_norm": 0.19998230040073395,
      "learning_rate": 1.3799643208705859e-05,
      "loss": 0.0483,
      "step": 46190
    },
    {
      "epoch": 5.4048677743973785,
      "loss_main": 0.055060021579265594,
      "step": 46190
    },
    {
      "epoch": 5.406037912473672,
      "grad_norm": 0.21831227838993073,
      "learning_rate": 1.3780639523886058e-05,
      "loss": 0.0562,
      "step": 46200
    },
    {
      "epoch": 5.406037912473672,
      "loss_main": 0.043820228427648544,
      "step": 46200
    },
    {
      "epoch": 5.407208050549965,
      "grad_norm": 0.24082380533218384,
      "learning_rate": 1.3761646841551668e-05,
      "loss": 0.046,
      "step": 46210
    },
    {
      "epoch": 5.407208050549965,
      "loss_main": 0.019961988553404808,
      "step": 46210
    },
    {
      "epoch": 5.408378188626258,
      "grad_norm": 0.21730488538742065,
      "learning_rate": 1.3742665167472146e-05,
      "loss": 0.0479,
      "step": 46220
    },
    {
      "epoch": 5.408378188626258,
      "loss_main": 0.05156046897172928,
      "step": 46220
    },
    {
      "epoch": 5.409548326702551,
      "grad_norm": 0.28667667508125305,
      "learning_rate": 1.372369450741363e-05,
      "loss": 0.0575,
      "step": 46230
    },
    {
      "epoch": 5.409548326702551,
      "loss_main": 0.057421375066041946,
      "step": 46230
    },
    {
      "epoch": 5.410718464778844,
      "grad_norm": 0.1908343881368637,
      "learning_rate": 1.3704734867138901e-05,
      "loss": 0.055,
      "step": 46240
    },
    {
      "epoch": 5.410718464778844,
      "loss_main": 0.04143170639872551,
      "step": 46240
    },
    {
      "epoch": 5.411888602855137,
      "grad_norm": 0.16876177489757538,
      "learning_rate": 1.36857862524074e-05,
      "loss": 0.0482,
      "step": 46250
    },
    {
      "epoch": 5.411888602855137,
      "loss_main": 0.04748274013400078,
      "step": 46250
    },
    {
      "epoch": 5.4130587409314295,
      "grad_norm": 0.20184111595153809,
      "learning_rate": 1.3666848668975213e-05,
      "loss": 0.0494,
      "step": 46260
    },
    {
      "epoch": 5.4130587409314295,
      "loss_main": 0.04067179188132286,
      "step": 46260
    },
    {
      "epoch": 5.414228879007723,
      "grad_norm": 0.2577230930328369,
      "learning_rate": 1.3647922122595063e-05,
      "loss": 0.041,
      "step": 46270
    },
    {
      "epoch": 5.414228879007723,
      "loss_main": 0.027699606493115425,
      "step": 46270
    },
    {
      "epoch": 5.415399017084016,
      "grad_norm": 0.2652057111263275,
      "learning_rate": 1.3629006619016366e-05,
      "loss": 0.052,
      "step": 46280
    },
    {
      "epoch": 5.415399017084016,
      "loss_main": 0.033492449671030045,
      "step": 46280
    },
    {
      "epoch": 5.416569155160309,
      "grad_norm": 0.21749407052993774,
      "learning_rate": 1.3610102163985139e-05,
      "loss": 0.0507,
      "step": 46290
    },
    {
      "epoch": 5.416569155160309,
      "loss_main": 0.03032025694847107,
      "step": 46290
    },
    {
      "epoch": 5.417739293236602,
      "grad_norm": 0.28802457451820374,
      "learning_rate": 1.3591208763244057e-05,
      "loss": 0.0476,
      "step": 46300
    },
    {
      "epoch": 5.417739293236602,
      "loss_main": 0.049949757754802704,
      "step": 46300
    },
    {
      "epoch": 5.418909431312895,
      "grad_norm": 0.2478986233472824,
      "learning_rate": 1.3572326422532428e-05,
      "loss": 0.0573,
      "step": 46310
    },
    {
      "epoch": 5.418909431312895,
      "loss_main": 0.04296886548399925,
      "step": 46310
    },
    {
      "epoch": 5.420079569389188,
      "grad_norm": 0.20009160041809082,
      "learning_rate": 1.355345514758622e-05,
      "loss": 0.0515,
      "step": 46320
    },
    {
      "epoch": 5.420079569389188,
      "loss_main": 0.05541567876935005,
      "step": 46320
    },
    {
      "epoch": 5.421249707465481,
      "grad_norm": 0.2326071709394455,
      "learning_rate": 1.3534594944138007e-05,
      "loss": 0.0514,
      "step": 46330
    },
    {
      "epoch": 5.421249707465481,
      "loss_main": 0.0646834596991539,
      "step": 46330
    },
    {
      "epoch": 5.422419845541774,
      "grad_norm": 0.2899201810359955,
      "learning_rate": 1.3515745817917069e-05,
      "loss": 0.0449,
      "step": 46340
    },
    {
      "epoch": 5.422419845541774,
      "loss_main": 0.06444971263408661,
      "step": 46340
    },
    {
      "epoch": 5.423589983618067,
      "grad_norm": 0.18875078856945038,
      "learning_rate": 1.3496907774649208e-05,
      "loss": 0.0482,
      "step": 46350
    },
    {
      "epoch": 5.423589983618067,
      "loss_main": 0.05070832744240761,
      "step": 46350
    },
    {
      "epoch": 5.42476012169436,
      "grad_norm": 0.2509639859199524,
      "learning_rate": 1.3478080820056987e-05,
      "loss": 0.0518,
      "step": 46360
    },
    {
      "epoch": 5.42476012169436,
      "loss_main": 0.04726452752947807,
      "step": 46360
    },
    {
      "epoch": 5.425930259770653,
      "grad_norm": 0.17357471585273743,
      "learning_rate": 1.3459264959859474e-05,
      "loss": 0.047,
      "step": 46370
    },
    {
      "epoch": 5.425930259770653,
      "loss_main": 0.041740782558918,
      "step": 46370
    },
    {
      "epoch": 5.427100397846946,
      "grad_norm": 0.2107822746038437,
      "learning_rate": 1.3440460199772487e-05,
      "loss": 0.041,
      "step": 46380
    },
    {
      "epoch": 5.427100397846946,
      "loss_main": 0.04452533647418022,
      "step": 46380
    },
    {
      "epoch": 5.428270535923239,
      "grad_norm": 0.25590434670448303,
      "learning_rate": 1.3421666545508382e-05,
      "loss": 0.0536,
      "step": 46390
    },
    {
      "epoch": 5.428270535923239,
      "loss_main": 0.03885491564869881,
      "step": 46390
    },
    {
      "epoch": 5.429440673999532,
      "grad_norm": 0.2060403823852539,
      "learning_rate": 1.3402884002776194e-05,
      "loss": 0.0473,
      "step": 46400
    },
    {
      "epoch": 5.429440673999532,
      "loss_main": 0.06777168810367584,
      "step": 46400
    },
    {
      "epoch": 5.430610812075825,
      "grad_norm": 0.3271436095237732,
      "learning_rate": 1.3384112577281555e-05,
      "loss": 0.0566,
      "step": 46410
    },
    {
      "epoch": 5.430610812075825,
      "loss_main": 0.07454044371843338,
      "step": 46410
    },
    {
      "epoch": 5.431780950152118,
      "grad_norm": 0.25254571437835693,
      "learning_rate": 1.3365352274726711e-05,
      "loss": 0.0525,
      "step": 46420
    },
    {
      "epoch": 5.431780950152118,
      "loss_main": 0.06285370886325836,
      "step": 46420
    },
    {
      "epoch": 5.432951088228411,
      "grad_norm": 0.2978552281856537,
      "learning_rate": 1.3346603100810578e-05,
      "loss": 0.0471,
      "step": 46430
    },
    {
      "epoch": 5.432951088228411,
      "loss_main": 0.03227010741829872,
      "step": 46430
    },
    {
      "epoch": 5.434121226304704,
      "grad_norm": 0.2506212890148163,
      "learning_rate": 1.3327865061228645e-05,
      "loss": 0.046,
      "step": 46440
    },
    {
      "epoch": 5.434121226304704,
      "loss_main": 0.06555651128292084,
      "step": 46440
    },
    {
      "epoch": 5.435291364380997,
      "grad_norm": 0.22204676270484924,
      "learning_rate": 1.330913816167304e-05,
      "loss": 0.0585,
      "step": 46450
    },
    {
      "epoch": 5.435291364380997,
      "loss_main": 0.02643798477947712,
      "step": 46450
    },
    {
      "epoch": 5.4364615024572895,
      "grad_norm": 0.18430745601654053,
      "learning_rate": 1.3290422407832492e-05,
      "loss": 0.051,
      "step": 46460
    },
    {
      "epoch": 5.4364615024572895,
      "loss_main": 0.04049784690141678,
      "step": 46460
    },
    {
      "epoch": 5.437631640533583,
      "grad_norm": 0.2462111860513687,
      "learning_rate": 1.3271717805392354e-05,
      "loss": 0.0427,
      "step": 46470
    },
    {
      "epoch": 5.437631640533583,
      "loss_main": 0.02956114150583744,
      "step": 46470
    },
    {
      "epoch": 5.438801778609876,
      "grad_norm": 0.24598266184329987,
      "learning_rate": 1.3253024360034582e-05,
      "loss": 0.0458,
      "step": 46480
    },
    {
      "epoch": 5.438801778609876,
      "loss_main": 0.0548366904258728,
      "step": 46480
    },
    {
      "epoch": 5.439971916686169,
      "grad_norm": 0.15583966672420502,
      "learning_rate": 1.323434207743779e-05,
      "loss": 0.0495,
      "step": 46490
    },
    {
      "epoch": 5.439971916686169,
      "loss_main": 0.04238453134894371,
      "step": 46490
    },
    {
      "epoch": 5.441142054762462,
      "grad_norm": 0.2902984619140625,
      "learning_rate": 1.3215670963277105e-05,
      "loss": 0.0588,
      "step": 46500
    },
    {
      "epoch": 5.441142054762462,
      "loss_main": 0.055050622671842575,
      "step": 46500
    },
    {
      "epoch": 5.442312192838755,
      "grad_norm": 0.17313796281814575,
      "learning_rate": 1.3197011023224376e-05,
      "loss": 0.0438,
      "step": 46510
    },
    {
      "epoch": 5.442312192838755,
      "loss_main": 0.029295792803168297,
      "step": 46510
    },
    {
      "epoch": 5.443482330915048,
      "grad_norm": 0.3052438795566559,
      "learning_rate": 1.3178362262947941e-05,
      "loss": 0.0605,
      "step": 46520
    },
    {
      "epoch": 5.443482330915048,
      "loss_main": 0.042151205241680145,
      "step": 46520
    },
    {
      "epoch": 5.444652468991341,
      "grad_norm": 0.27714163064956665,
      "learning_rate": 1.3159724688112845e-05,
      "loss": 0.0468,
      "step": 46530
    },
    {
      "epoch": 5.444652468991341,
      "loss_main": 0.044348519295454025,
      "step": 46530
    },
    {
      "epoch": 5.445822607067634,
      "grad_norm": 0.26811882853507996,
      "learning_rate": 1.3141098304380683e-05,
      "loss": 0.0512,
      "step": 46540
    },
    {
      "epoch": 5.445822607067634,
      "loss_main": 0.04236694797873497,
      "step": 46540
    },
    {
      "epoch": 5.446992745143927,
      "grad_norm": 0.24456925690174103,
      "learning_rate": 1.3122483117409651e-05,
      "loss": 0.0467,
      "step": 46550
    },
    {
      "epoch": 5.446992745143927,
      "loss_main": 0.025621766224503517,
      "step": 46550
    },
    {
      "epoch": 5.44816288322022,
      "grad_norm": 0.17440523207187653,
      "learning_rate": 1.3103879132854552e-05,
      "loss": 0.0514,
      "step": 46560
    },
    {
      "epoch": 5.44816288322022,
      "loss_main": 0.08259914070367813,
      "step": 46560
    },
    {
      "epoch": 5.449333021296513,
      "grad_norm": 0.2693122327327728,
      "learning_rate": 1.3085286356366771e-05,
      "loss": 0.0524,
      "step": 46570
    },
    {
      "epoch": 5.449333021296513,
      "loss_main": 0.02714080549776554,
      "step": 46570
    },
    {
      "epoch": 5.450503159372806,
      "grad_norm": 0.3090444803237915,
      "learning_rate": 1.3066704793594337e-05,
      "loss": 0.0444,
      "step": 46580
    },
    {
      "epoch": 5.450503159372806,
      "loss_main": 0.06480611860752106,
      "step": 46580
    },
    {
      "epoch": 5.451673297449099,
      "grad_norm": 0.2663383185863495,
      "learning_rate": 1.3048134450181816e-05,
      "loss": 0.0435,
      "step": 46590
    },
    {
      "epoch": 5.451673297449099,
      "loss_main": 0.06519728153944016,
      "step": 46590
    },
    {
      "epoch": 5.452843435525392,
      "grad_norm": 0.21879170835018158,
      "learning_rate": 1.3029575331770394e-05,
      "loss": 0.0494,
      "step": 46600
    },
    {
      "epoch": 5.452843435525392,
      "loss_main": 0.04553329572081566,
      "step": 46600
    },
    {
      "epoch": 5.454013573601685,
      "grad_norm": 0.20559710264205933,
      "learning_rate": 1.3011027443997837e-05,
      "loss": 0.0472,
      "step": 46610
    },
    {
      "epoch": 5.454013573601685,
      "loss_main": 0.07933628559112549,
      "step": 46610
    },
    {
      "epoch": 5.455183711677978,
      "grad_norm": 0.2562020719051361,
      "learning_rate": 1.2992490792498507e-05,
      "loss": 0.0443,
      "step": 46620
    },
    {
      "epoch": 5.455183711677978,
      "loss_main": 0.034015052020549774,
      "step": 46620
    },
    {
      "epoch": 5.456353849754271,
      "grad_norm": 0.1954527050256729,
      "learning_rate": 1.297396538290333e-05,
      "loss": 0.0545,
      "step": 46630
    },
    {
      "epoch": 5.456353849754271,
      "loss_main": 0.03728923201560974,
      "step": 46630
    },
    {
      "epoch": 5.457523987830564,
      "grad_norm": 0.27370786666870117,
      "learning_rate": 1.2955451220839888e-05,
      "loss": 0.0394,
      "step": 46640
    },
    {
      "epoch": 5.457523987830564,
      "loss_main": 0.04417327791452408,
      "step": 46640
    },
    {
      "epoch": 5.458694125906857,
      "grad_norm": 0.2203235626220703,
      "learning_rate": 1.2936948311932223e-05,
      "loss": 0.0504,
      "step": 46650
    },
    {
      "epoch": 5.458694125906857,
      "loss_main": 0.057606134563684464,
      "step": 46650
    },
    {
      "epoch": 5.45986426398315,
      "grad_norm": 0.1567775309085846,
      "learning_rate": 1.2918456661801104e-05,
      "loss": 0.0428,
      "step": 46660
    },
    {
      "epoch": 5.45986426398315,
      "loss_main": 0.05880201607942581,
      "step": 46660
    },
    {
      "epoch": 5.461034402059443,
      "grad_norm": 0.4836665093898773,
      "learning_rate": 1.2899976276063736e-05,
      "loss": 0.0492,
      "step": 46670
    },
    {
      "epoch": 5.461034402059443,
      "loss_main": 0.05770860239863396,
      "step": 46670
    },
    {
      "epoch": 5.462204540135736,
      "grad_norm": 0.16095291078090668,
      "learning_rate": 1.2881507160334022e-05,
      "loss": 0.0488,
      "step": 46680
    },
    {
      "epoch": 5.462204540135736,
      "loss_main": 0.015060914680361748,
      "step": 46680
    },
    {
      "epoch": 5.463374678212029,
      "grad_norm": 0.30894121527671814,
      "learning_rate": 1.286304932022238e-05,
      "loss": 0.0476,
      "step": 46690
    },
    {
      "epoch": 5.463374678212029,
      "loss_main": 0.025323165580630302,
      "step": 46690
    },
    {
      "epoch": 5.464544816288322,
      "grad_norm": 0.22111976146697998,
      "learning_rate": 1.2844602761335806e-05,
      "loss": 0.0539,
      "step": 46700
    },
    {
      "epoch": 5.464544816288322,
      "loss_main": 0.05087682232260704,
      "step": 46700
    },
    {
      "epoch": 5.465714954364615,
      "grad_norm": 0.2397622913122177,
      "learning_rate": 1.2826167489277885e-05,
      "loss": 0.0395,
      "step": 46710
    },
    {
      "epoch": 5.465714954364615,
      "loss_main": 0.034908588975667953,
      "step": 46710
    },
    {
      "epoch": 5.466885092440908,
      "grad_norm": 0.31099557876586914,
      "learning_rate": 1.2807743509648745e-05,
      "loss": 0.0416,
      "step": 46720
    },
    {
      "epoch": 5.466885092440908,
      "loss_main": 0.03771079704165459,
      "step": 46720
    },
    {
      "epoch": 5.468055230517201,
      "grad_norm": 0.26314541697502136,
      "learning_rate": 1.2789330828045149e-05,
      "loss": 0.0463,
      "step": 46730
    },
    {
      "epoch": 5.468055230517201,
      "loss_main": 0.06270503252744675,
      "step": 46730
    },
    {
      "epoch": 5.469225368593494,
      "grad_norm": 0.24544671177864075,
      "learning_rate": 1.2770929450060332e-05,
      "loss": 0.0515,
      "step": 46740
    },
    {
      "epoch": 5.469225368593494,
      "loss_main": 0.062043409794569016,
      "step": 46740
    },
    {
      "epoch": 5.470395506669787,
      "grad_norm": 0.23582935333251953,
      "learning_rate": 1.2752539381284184e-05,
      "loss": 0.0497,
      "step": 46750
    },
    {
      "epoch": 5.470395506669787,
      "loss_main": 0.028891904279589653,
      "step": 46750
    },
    {
      "epoch": 5.47156564474608,
      "grad_norm": 0.22208116948604584,
      "learning_rate": 1.273416062730311e-05,
      "loss": 0.0472,
      "step": 46760
    },
    {
      "epoch": 5.47156564474608,
      "loss_main": 0.05908889323472977,
      "step": 46760
    },
    {
      "epoch": 5.472735782822373,
      "grad_norm": 0.24082590639591217,
      "learning_rate": 1.2715793193700088e-05,
      "loss": 0.046,
      "step": 46770
    },
    {
      "epoch": 5.472735782822373,
      "loss_main": 0.04999425262212753,
      "step": 46770
    },
    {
      "epoch": 5.473905920898666,
      "grad_norm": 0.23131555318832397,
      "learning_rate": 1.2697437086054664e-05,
      "loss": 0.0458,
      "step": 46780
    },
    {
      "epoch": 5.473905920898666,
      "loss_main": 0.07953854650259018,
      "step": 46780
    },
    {
      "epoch": 5.475076058974959,
      "grad_norm": 0.297908216714859,
      "learning_rate": 1.2679092309942937e-05,
      "loss": 0.0521,
      "step": 46790
    },
    {
      "epoch": 5.475076058974959,
      "loss_main": 0.06212882325053215,
      "step": 46790
    },
    {
      "epoch": 5.476246197051252,
      "grad_norm": 0.23326188325881958,
      "learning_rate": 1.266075887093755e-05,
      "loss": 0.0456,
      "step": 46800
    },
    {
      "epoch": 5.476246197051252,
      "loss_main": 0.04821052774786949,
      "step": 46800
    },
    {
      "epoch": 5.477416335127545,
      "grad_norm": 0.19938212633132935,
      "learning_rate": 1.2642436774607757e-05,
      "loss": 0.0513,
      "step": 46810
    },
    {
      "epoch": 5.477416335127545,
      "loss_main": 0.044714927673339844,
      "step": 46810
    },
    {
      "epoch": 5.478586473203838,
      "grad_norm": 0.24227166175842285,
      "learning_rate": 1.2624126026519278e-05,
      "loss": 0.0432,
      "step": 46820
    },
    {
      "epoch": 5.478586473203838,
      "loss_main": 0.0608927346765995,
      "step": 46820
    },
    {
      "epoch": 5.479756611280131,
      "grad_norm": 0.24956943094730377,
      "learning_rate": 1.2605826632234474e-05,
      "loss": 0.0514,
      "step": 46830
    },
    {
      "epoch": 5.479756611280131,
      "loss_main": 0.03643682971596718,
      "step": 46830
    },
    {
      "epoch": 5.480926749356424,
      "grad_norm": 0.20187507569789886,
      "learning_rate": 1.2587538597312198e-05,
      "loss": 0.0433,
      "step": 46840
    },
    {
      "epoch": 5.480926749356424,
      "loss_main": 0.10057999193668365,
      "step": 46840
    },
    {
      "epoch": 5.482096887432717,
      "grad_norm": 0.21686074137687683,
      "learning_rate": 1.2569261927307884e-05,
      "loss": 0.0515,
      "step": 46850
    },
    {
      "epoch": 5.482096887432717,
      "loss_main": 0.029751667752861977,
      "step": 46850
    },
    {
      "epoch": 5.48326702550901,
      "grad_norm": 0.18122196197509766,
      "learning_rate": 1.2550996627773493e-05,
      "loss": 0.0376,
      "step": 46860
    },
    {
      "epoch": 5.48326702550901,
      "loss_main": 0.05142370983958244,
      "step": 46860
    },
    {
      "epoch": 5.484437163585303,
      "grad_norm": 0.3226533532142639,
      "learning_rate": 1.2532742704257527e-05,
      "loss": 0.0489,
      "step": 46870
    },
    {
      "epoch": 5.484437163585303,
      "loss_main": 0.04572128504514694,
      "step": 46870
    },
    {
      "epoch": 5.485607301661596,
      "grad_norm": 0.26476362347602844,
      "learning_rate": 1.2514500162305087e-05,
      "loss": 0.0409,
      "step": 46880
    },
    {
      "epoch": 5.485607301661596,
      "loss_main": 0.05889493227005005,
      "step": 46880
    },
    {
      "epoch": 5.486777439737889,
      "grad_norm": 0.31031864881515503,
      "learning_rate": 1.2496269007457728e-05,
      "loss": 0.0458,
      "step": 46890
    },
    {
      "epoch": 5.486777439737889,
      "loss_main": 0.052529461681842804,
      "step": 46890
    },
    {
      "epoch": 5.487947577814182,
      "grad_norm": 0.2704206109046936,
      "learning_rate": 1.2478049245253625e-05,
      "loss": 0.0436,
      "step": 46900
    },
    {
      "epoch": 5.487947577814182,
      "loss_main": 0.026361163705587387,
      "step": 46900
    },
    {
      "epoch": 5.489117715890475,
      "grad_norm": 0.22360268235206604,
      "learning_rate": 1.2459840881227459e-05,
      "loss": 0.0499,
      "step": 46910
    },
    {
      "epoch": 5.489117715890475,
      "loss_main": 0.06512591987848282,
      "step": 46910
    },
    {
      "epoch": 5.490287853966768,
      "grad_norm": 0.2622065246105194,
      "learning_rate": 1.2441643920910435e-05,
      "loss": 0.052,
      "step": 46920
    },
    {
      "epoch": 5.490287853966768,
      "loss_main": 0.031380485743284225,
      "step": 46920
    },
    {
      "epoch": 5.4914579920430615,
      "grad_norm": 0.23934485018253326,
      "learning_rate": 1.2423458369830322e-05,
      "loss": 0.0427,
      "step": 46930
    },
    {
      "epoch": 5.4914579920430615,
      "loss_main": 0.041248567402362823,
      "step": 46930
    },
    {
      "epoch": 5.492628130119354,
      "grad_norm": 0.20306672155857086,
      "learning_rate": 1.2405284233511406e-05,
      "loss": 0.0485,
      "step": 46940
    },
    {
      "epoch": 5.492628130119354,
      "loss_main": 0.07102646678686142,
      "step": 46940
    },
    {
      "epoch": 5.493798268195647,
      "grad_norm": 0.2715405821800232,
      "learning_rate": 1.2387121517474487e-05,
      "loss": 0.0486,
      "step": 46950
    },
    {
      "epoch": 5.493798268195647,
      "loss_main": 0.052633337676525116,
      "step": 46950
    },
    {
      "epoch": 5.49496840627194,
      "grad_norm": 0.17794553935527802,
      "learning_rate": 1.2368970227236975e-05,
      "loss": 0.0487,
      "step": 46960
    },
    {
      "epoch": 5.49496840627194,
      "loss_main": 0.04234621301293373,
      "step": 46960
    },
    {
      "epoch": 5.496138544348233,
      "grad_norm": 0.17709887027740479,
      "learning_rate": 1.2350830368312688e-05,
      "loss": 0.0476,
      "step": 46970
    },
    {
      "epoch": 5.496138544348233,
      "loss_main": 0.030196785926818848,
      "step": 46970
    },
    {
      "epoch": 5.497308682424526,
      "grad_norm": 0.2739419639110565,
      "learning_rate": 1.2332701946212083e-05,
      "loss": 0.052,
      "step": 46980
    },
    {
      "epoch": 5.497308682424526,
      "loss_main": 0.03767731785774231,
      "step": 46980
    },
    {
      "epoch": 5.498478820500819,
      "grad_norm": 0.19712460041046143,
      "learning_rate": 1.2314584966442077e-05,
      "loss": 0.0477,
      "step": 46990
    },
    {
      "epoch": 5.498478820500819,
      "loss_main": 0.06701089441776276,
      "step": 46990
    },
    {
      "epoch": 5.4996489585771124,
      "grad_norm": 0.20885440707206726,
      "learning_rate": 1.2296479434506136e-05,
      "loss": 0.0512,
      "step": 47000
    },
    {
      "epoch": 5.4996489585771124,
      "loss_main": 0.047806404531002045,
      "step": 47000
    },
    {
      "epoch": 5.500819096653405,
      "grad_norm": 0.1595827341079712,
      "learning_rate": 1.2278385355904232e-05,
      "loss": 0.0489,
      "step": 47010
    },
    {
      "epoch": 5.500819096653405,
      "loss_main": 0.04566550254821777,
      "step": 47010
    },
    {
      "epoch": 5.501989234729698,
      "grad_norm": 0.23315872251987457,
      "learning_rate": 1.2260302736132867e-05,
      "loss": 0.0464,
      "step": 47020
    },
    {
      "epoch": 5.501989234729698,
      "loss_main": 0.058226052671670914,
      "step": 47020
    },
    {
      "epoch": 5.503159372805991,
      "grad_norm": 0.2894911468029022,
      "learning_rate": 1.2242231580685098e-05,
      "loss": 0.0572,
      "step": 47030
    },
    {
      "epoch": 5.503159372805991,
      "loss_main": 0.04542726278305054,
      "step": 47030
    },
    {
      "epoch": 5.504329510882284,
      "grad_norm": 0.2096823751926422,
      "learning_rate": 1.2224171895050413e-05,
      "loss": 0.0534,
      "step": 47040
    },
    {
      "epoch": 5.504329510882284,
      "loss_main": 0.05224434286355972,
      "step": 47040
    },
    {
      "epoch": 5.505499648958577,
      "grad_norm": 0.24670082330703735,
      "learning_rate": 1.2206123684714903e-05,
      "loss": 0.0464,
      "step": 47050
    },
    {
      "epoch": 5.505499648958577,
      "loss_main": 0.033885408192873,
      "step": 47050
    },
    {
      "epoch": 5.50666978703487,
      "grad_norm": 0.2125113159418106,
      "learning_rate": 1.2188086955161132e-05,
      "loss": 0.0446,
      "step": 47060
    },
    {
      "epoch": 5.50666978703487,
      "loss_main": 0.049176305532455444,
      "step": 47060
    },
    {
      "epoch": 5.507839925111163,
      "grad_norm": 0.2411315143108368,
      "learning_rate": 1.2170061711868175e-05,
      "loss": 0.0513,
      "step": 47070
    },
    {
      "epoch": 5.507839925111163,
      "loss_main": 0.038760703057050705,
      "step": 47070
    },
    {
      "epoch": 5.509010063187456,
      "grad_norm": 0.22399142384529114,
      "learning_rate": 1.215204796031163e-05,
      "loss": 0.0495,
      "step": 47080
    },
    {
      "epoch": 5.509010063187456,
      "loss_main": 0.03499533236026764,
      "step": 47080
    },
    {
      "epoch": 5.510180201263749,
      "grad_norm": 0.26003772020339966,
      "learning_rate": 1.2134045705963599e-05,
      "loss": 0.0385,
      "step": 47090
    },
    {
      "epoch": 5.510180201263749,
      "loss_main": 0.024004409089684486,
      "step": 47090
    },
    {
      "epoch": 5.5113503393400425,
      "grad_norm": 0.2102273851633072,
      "learning_rate": 1.2116054954292689e-05,
      "loss": 0.0408,
      "step": 47100
    },
    {
      "epoch": 5.5113503393400425,
      "loss_main": 0.039517588913440704,
      "step": 47100
    },
    {
      "epoch": 5.512520477416335,
      "grad_norm": 0.22581152617931366,
      "learning_rate": 1.2098075710764011e-05,
      "loss": 0.0446,
      "step": 47110
    },
    {
      "epoch": 5.512520477416335,
      "loss_main": 0.02184459939599037,
      "step": 47110
    },
    {
      "epoch": 5.513690615492628,
      "grad_norm": 0.2876870334148407,
      "learning_rate": 1.2080107980839183e-05,
      "loss": 0.0409,
      "step": 47120
    },
    {
      "epoch": 5.513690615492628,
      "loss_main": 0.032389868050813675,
      "step": 47120
    },
    {
      "epoch": 5.514860753568922,
      "grad_norm": 0.20319120585918427,
      "learning_rate": 1.2062151769976343e-05,
      "loss": 0.0474,
      "step": 47130
    },
    {
      "epoch": 5.514860753568922,
      "loss_main": 0.04156848415732384,
      "step": 47130
    },
    {
      "epoch": 5.516030891645214,
      "grad_norm": 0.22414617240428925,
      "learning_rate": 1.204420708363011e-05,
      "loss": 0.0455,
      "step": 47140
    },
    {
      "epoch": 5.516030891645214,
      "loss_main": 0.037770092487335205,
      "step": 47140
    },
    {
      "epoch": 5.517201029721507,
      "grad_norm": 0.1533588320016861,
      "learning_rate": 1.2026273927251597e-05,
      "loss": 0.0476,
      "step": 47150
    },
    {
      "epoch": 5.517201029721507,
      "loss_main": 0.031418778002262115,
      "step": 47150
    },
    {
      "epoch": 5.5183711677978,
      "grad_norm": 0.23442678153514862,
      "learning_rate": 1.2008352306288424e-05,
      "loss": 0.0471,
      "step": 47160
    },
    {
      "epoch": 5.5183711677978,
      "loss_main": 0.0630965381860733,
      "step": 47160
    },
    {
      "epoch": 5.519541305874093,
      "grad_norm": 0.23722130060195923,
      "learning_rate": 1.1990442226184695e-05,
      "loss": 0.0458,
      "step": 47170
    },
    {
      "epoch": 5.519541305874093,
      "loss_main": 0.024663565680384636,
      "step": 47170
    },
    {
      "epoch": 5.520711443950386,
      "grad_norm": 0.27390411496162415,
      "learning_rate": 1.1972543692381066e-05,
      "loss": 0.0548,
      "step": 47180
    },
    {
      "epoch": 5.520711443950386,
      "loss_main": 0.06825936585664749,
      "step": 47180
    },
    {
      "epoch": 5.521881582026679,
      "grad_norm": 0.2114323377609253,
      "learning_rate": 1.1954656710314576e-05,
      "loss": 0.0536,
      "step": 47190
    },
    {
      "epoch": 5.521881582026679,
      "loss_main": 0.04901687055826187,
      "step": 47190
    },
    {
      "epoch": 5.5230517201029725,
      "grad_norm": 0.32150158286094666,
      "learning_rate": 1.1936781285418875e-05,
      "loss": 0.0459,
      "step": 47200
    },
    {
      "epoch": 5.5230517201029725,
      "loss_main": 0.045665595680475235,
      "step": 47200
    },
    {
      "epoch": 5.524221858179265,
      "grad_norm": 0.3386717140674591,
      "learning_rate": 1.1918917423123993e-05,
      "loss": 0.0442,
      "step": 47210
    },
    {
      "epoch": 5.524221858179265,
      "loss_main": 0.03511372581124306,
      "step": 47210
    },
    {
      "epoch": 5.525391996255558,
      "grad_norm": 0.35897698998451233,
      "learning_rate": 1.1901065128856537e-05,
      "loss": 0.0504,
      "step": 47220
    },
    {
      "epoch": 5.525391996255558,
      "loss_main": 0.02433175966143608,
      "step": 47220
    },
    {
      "epoch": 5.526562134331851,
      "grad_norm": 0.3010149598121643,
      "learning_rate": 1.1883224408039551e-05,
      "loss": 0.0462,
      "step": 47230
    },
    {
      "epoch": 5.526562134331851,
      "loss_main": 0.039212532341480255,
      "step": 47230
    },
    {
      "epoch": 5.527732272408144,
      "grad_norm": 0.1858038306236267,
      "learning_rate": 1.1865395266092578e-05,
      "loss": 0.049,
      "step": 47240
    },
    {
      "epoch": 5.527732272408144,
      "loss_main": 0.04667859897017479,
      "step": 47240
    },
    {
      "epoch": 5.528902410484437,
      "grad_norm": 0.22208866477012634,
      "learning_rate": 1.1847577708431633e-05,
      "loss": 0.0504,
      "step": 47250
    },
    {
      "epoch": 5.528902410484437,
      "loss_main": 0.024695660918951035,
      "step": 47250
    },
    {
      "epoch": 5.53007254856073,
      "grad_norm": 0.2525125741958618,
      "learning_rate": 1.1829771740469225e-05,
      "loss": 0.0431,
      "step": 47260
    },
    {
      "epoch": 5.53007254856073,
      "loss_main": 0.0326034314930439,
      "step": 47260
    },
    {
      "epoch": 5.5312426866370235,
      "grad_norm": 0.21376393735408783,
      "learning_rate": 1.1811977367614324e-05,
      "loss": 0.0457,
      "step": 47270
    },
    {
      "epoch": 5.5312426866370235,
      "loss_main": 0.04970843344926834,
      "step": 47270
    },
    {
      "epoch": 5.532412824713316,
      "grad_norm": 0.27544480562210083,
      "learning_rate": 1.1794194595272412e-05,
      "loss": 0.0603,
      "step": 47280
    },
    {
      "epoch": 5.532412824713316,
      "loss_main": 0.055047910660505295,
      "step": 47280
    },
    {
      "epoch": 5.533582962789609,
      "grad_norm": 0.24361328780651093,
      "learning_rate": 1.1776423428845423e-05,
      "loss": 0.0402,
      "step": 47290
    },
    {
      "epoch": 5.533582962789609,
      "loss_main": 0.05530223622918129,
      "step": 47290
    },
    {
      "epoch": 5.5347531008659026,
      "grad_norm": 0.20727239549160004,
      "learning_rate": 1.1758663873731756e-05,
      "loss": 0.0541,
      "step": 47300
    },
    {
      "epoch": 5.5347531008659026,
      "loss_main": 0.03702809289097786,
      "step": 47300
    },
    {
      "epoch": 5.535923238942195,
      "grad_norm": 0.3212704658508301,
      "learning_rate": 1.1740915935326302e-05,
      "loss": 0.0526,
      "step": 47310
    },
    {
      "epoch": 5.535923238942195,
      "loss_main": 0.025415685027837753,
      "step": 47310
    },
    {
      "epoch": 5.537093377018488,
      "grad_norm": 0.18369238078594208,
      "learning_rate": 1.1723179619020396e-05,
      "loss": 0.0447,
      "step": 47320
    },
    {
      "epoch": 5.537093377018488,
      "loss_main": 0.0620051808655262,
      "step": 47320
    },
    {
      "epoch": 5.538263515094782,
      "grad_norm": 0.17573091387748718,
      "learning_rate": 1.1705454930201914e-05,
      "loss": 0.0502,
      "step": 47330
    },
    {
      "epoch": 5.538263515094782,
      "loss_main": 0.038649048656225204,
      "step": 47330
    },
    {
      "epoch": 5.539433653171074,
      "grad_norm": 0.2703481912612915,
      "learning_rate": 1.1687741874255087e-05,
      "loss": 0.0542,
      "step": 47340
    },
    {
      "epoch": 5.539433653171074,
      "loss_main": 0.041218649595975876,
      "step": 47340
    },
    {
      "epoch": 5.540603791247367,
      "grad_norm": 0.2637597322463989,
      "learning_rate": 1.1670040456560728e-05,
      "loss": 0.0452,
      "step": 47350
    },
    {
      "epoch": 5.540603791247367,
      "loss_main": 0.11053977161645889,
      "step": 47350
    },
    {
      "epoch": 5.54177392932366,
      "grad_norm": 0.24198175966739655,
      "learning_rate": 1.1652350682496005e-05,
      "loss": 0.0546,
      "step": 47360
    },
    {
      "epoch": 5.54177392932366,
      "loss_main": 0.0892239436507225,
      "step": 47360
    },
    {
      "epoch": 5.5429440673999535,
      "grad_norm": 0.24699559807777405,
      "learning_rate": 1.163467255743465e-05,
      "loss": 0.0494,
      "step": 47370
    },
    {
      "epoch": 5.5429440673999535,
      "loss_main": 0.05603794381022453,
      "step": 47370
    },
    {
      "epoch": 5.544114205476246,
      "grad_norm": 0.23743171989917755,
      "learning_rate": 1.1617006086746796e-05,
      "loss": 0.051,
      "step": 47380
    },
    {
      "epoch": 5.544114205476246,
      "loss_main": 0.07809974253177643,
      "step": 47380
    },
    {
      "epoch": 5.545284343552539,
      "grad_norm": 0.18212300539016724,
      "learning_rate": 1.1599351275799047e-05,
      "loss": 0.0486,
      "step": 47390
    },
    {
      "epoch": 5.545284343552539,
      "loss_main": 0.04508589953184128,
      "step": 47390
    },
    {
      "epoch": 5.546454481628833,
      "grad_norm": 0.16686289012432098,
      "learning_rate": 1.1581708129954466e-05,
      "loss": 0.0404,
      "step": 47400
    },
    {
      "epoch": 5.546454481628833,
      "loss_main": 0.05349916219711304,
      "step": 47400
    },
    {
      "epoch": 5.547624619705125,
      "grad_norm": 0.2839333415031433,
      "learning_rate": 1.1564076654572587e-05,
      "loss": 0.0448,
      "step": 47410
    },
    {
      "epoch": 5.547624619705125,
      "loss_main": 0.0463414192199707,
      "step": 47410
    },
    {
      "epoch": 5.548794757781418,
      "grad_norm": 0.24664920568466187,
      "learning_rate": 1.1546456855009358e-05,
      "loss": 0.0479,
      "step": 47420
    },
    {
      "epoch": 5.548794757781418,
      "loss_main": 0.04424578323960304,
      "step": 47420
    },
    {
      "epoch": 5.549964895857711,
      "grad_norm": 0.21941404044628143,
      "learning_rate": 1.1528848736617248e-05,
      "loss": 0.0464,
      "step": 47430
    },
    {
      "epoch": 5.549964895857711,
      "loss_main": 0.03412752225995064,
      "step": 47430
    },
    {
      "epoch": 5.551135033934004,
      "grad_norm": 0.21939700841903687,
      "learning_rate": 1.1511252304745112e-05,
      "loss": 0.048,
      "step": 47440
    },
    {
      "epoch": 5.551135033934004,
      "loss_main": 0.030241308733820915,
      "step": 47440
    },
    {
      "epoch": 5.552305172010297,
      "grad_norm": 0.19799001514911652,
      "learning_rate": 1.1493667564738297e-05,
      "loss": 0.0585,
      "step": 47450
    },
    {
      "epoch": 5.552305172010297,
      "loss_main": 0.07699590176343918,
      "step": 47450
    },
    {
      "epoch": 5.55347531008659,
      "grad_norm": 0.14829741418361664,
      "learning_rate": 1.1476094521938574e-05,
      "loss": 0.0437,
      "step": 47460
    },
    {
      "epoch": 5.55347531008659,
      "loss_main": 0.021695775911211967,
      "step": 47460
    },
    {
      "epoch": 5.5546454481628835,
      "grad_norm": 0.15169064700603485,
      "learning_rate": 1.1458533181684167e-05,
      "loss": 0.0476,
      "step": 47470
    },
    {
      "epoch": 5.5546454481628835,
      "loss_main": 0.0443740151822567,
      "step": 47470
    },
    {
      "epoch": 5.555815586239176,
      "grad_norm": 0.19099664688110352,
      "learning_rate": 1.1440983549309753e-05,
      "loss": 0.0464,
      "step": 47480
    },
    {
      "epoch": 5.555815586239176,
      "loss_main": 0.0420432984828949,
      "step": 47480
    },
    {
      "epoch": 5.556985724315469,
      "grad_norm": 0.24441830813884735,
      "learning_rate": 1.1423445630146434e-05,
      "loss": 0.0443,
      "step": 47490
    },
    {
      "epoch": 5.556985724315469,
      "loss_main": 0.04888729378581047,
      "step": 47490
    },
    {
      "epoch": 5.558155862391763,
      "grad_norm": 0.22907021641731262,
      "learning_rate": 1.1405919429521799e-05,
      "loss": 0.0522,
      "step": 47500
    },
    {
      "epoch": 5.558155862391763,
      "loss_main": 0.03279678151011467,
      "step": 47500
    },
    {
      "epoch": 5.559326000468055,
      "grad_norm": 0.2457212209701538,
      "learning_rate": 1.1388404952759802e-05,
      "loss": 0.0454,
      "step": 47510
    },
    {
      "epoch": 5.559326000468055,
      "loss_main": 0.03556086868047714,
      "step": 47510
    },
    {
      "epoch": 5.560496138544348,
      "grad_norm": 0.21393051743507385,
      "learning_rate": 1.1370902205180923e-05,
      "loss": 0.0468,
      "step": 47520
    },
    {
      "epoch": 5.560496138544348,
      "loss_main": 0.05843386426568031,
      "step": 47520
    },
    {
      "epoch": 5.561666276620642,
      "grad_norm": 0.26866939663887024,
      "learning_rate": 1.1353411192101987e-05,
      "loss": 0.0491,
      "step": 47530
    },
    {
      "epoch": 5.561666276620642,
      "loss_main": 0.023676540702581406,
      "step": 47530
    },
    {
      "epoch": 5.5628364146969345,
      "grad_norm": 0.18667089939117432,
      "learning_rate": 1.133593191883634e-05,
      "loss": 0.0398,
      "step": 47540
    },
    {
      "epoch": 5.5628364146969345,
      "loss_main": 0.05624900013208389,
      "step": 47540
    },
    {
      "epoch": 5.564006552773227,
      "grad_norm": 0.17725639045238495,
      "learning_rate": 1.1318464390693711e-05,
      "loss": 0.0409,
      "step": 47550
    },
    {
      "epoch": 5.564006552773227,
      "loss_main": 0.03331718221306801,
      "step": 47550
    },
    {
      "epoch": 5.56517669084952,
      "grad_norm": 0.2232104241847992,
      "learning_rate": 1.1301008612980257e-05,
      "loss": 0.0499,
      "step": 47560
    },
    {
      "epoch": 5.56517669084952,
      "loss_main": 0.047236811369657516,
      "step": 47560
    },
    {
      "epoch": 5.566346828925814,
      "grad_norm": 0.2370040863752365,
      "learning_rate": 1.128356459099863e-05,
      "loss": 0.0485,
      "step": 47570
    },
    {
      "epoch": 5.566346828925814,
      "loss_main": 0.03666260465979576,
      "step": 47570
    },
    {
      "epoch": 5.567516967002106,
      "grad_norm": 0.3391278088092804,
      "learning_rate": 1.1266132330047802e-05,
      "loss": 0.0542,
      "step": 47580
    },
    {
      "epoch": 5.567516967002106,
      "loss_main": 0.0216154046356678,
      "step": 47580
    },
    {
      "epoch": 5.568687105078399,
      "grad_norm": 0.19332753121852875,
      "learning_rate": 1.1248711835423281e-05,
      "loss": 0.0512,
      "step": 47590
    },
    {
      "epoch": 5.568687105078399,
      "loss_main": 0.03571241348981857,
      "step": 47590
    },
    {
      "epoch": 5.569857243154692,
      "grad_norm": 0.24525517225265503,
      "learning_rate": 1.123130311241693e-05,
      "loss": 0.0449,
      "step": 47600
    },
    {
      "epoch": 5.569857243154692,
      "loss_main": 0.03389621526002884,
      "step": 47600
    },
    {
      "epoch": 5.571027381230985,
      "grad_norm": 0.23134475946426392,
      "learning_rate": 1.1213906166317068e-05,
      "loss": 0.0436,
      "step": 47610
    },
    {
      "epoch": 5.571027381230985,
      "loss_main": 0.0436326302587986,
      "step": 47610
    },
    {
      "epoch": 5.572197519307278,
      "grad_norm": 0.26342856884002686,
      "learning_rate": 1.1196521002408427e-05,
      "loss": 0.0558,
      "step": 47620
    },
    {
      "epoch": 5.572197519307278,
      "loss_main": 0.028559641912579536,
      "step": 47620
    },
    {
      "epoch": 5.573367657383571,
      "grad_norm": 0.26816877722740173,
      "learning_rate": 1.1179147625972159e-05,
      "loss": 0.0471,
      "step": 47630
    },
    {
      "epoch": 5.573367657383571,
      "loss_main": 0.03780429810285568,
      "step": 47630
    },
    {
      "epoch": 5.5745377954598645,
      "grad_norm": 0.26088184118270874,
      "learning_rate": 1.1161786042285822e-05,
      "loss": 0.0462,
      "step": 47640
    },
    {
      "epoch": 5.5745377954598645,
      "loss_main": 0.08628834784030914,
      "step": 47640
    },
    {
      "epoch": 5.575707933536157,
      "grad_norm": 0.30712026357650757,
      "learning_rate": 1.1144436256623447e-05,
      "loss": 0.052,
      "step": 47650
    },
    {
      "epoch": 5.575707933536157,
      "loss_main": 0.039118144661188126,
      "step": 47650
    },
    {
      "epoch": 5.57687807161245,
      "grad_norm": 0.2619384229183197,
      "learning_rate": 1.1127098274255392e-05,
      "loss": 0.0542,
      "step": 47660
    },
    {
      "epoch": 5.57687807161245,
      "loss_main": 0.035403456538915634,
      "step": 47660
    },
    {
      "epoch": 5.578048209688744,
      "grad_norm": 0.24577787518501282,
      "learning_rate": 1.1109772100448512e-05,
      "loss": 0.0455,
      "step": 47670
    },
    {
      "epoch": 5.578048209688744,
      "loss_main": 0.045095182955265045,
      "step": 47670
    },
    {
      "epoch": 5.579218347765036,
      "grad_norm": 0.23929020762443542,
      "learning_rate": 1.1092457740466033e-05,
      "loss": 0.0443,
      "step": 47680
    },
    {
      "epoch": 5.579218347765036,
      "loss_main": 0.058097176253795624,
      "step": 47680
    },
    {
      "epoch": 5.580388485841329,
      "grad_norm": 0.2785394489765167,
      "learning_rate": 1.10751551995676e-05,
      "loss": 0.0565,
      "step": 47690
    },
    {
      "epoch": 5.580388485841329,
      "loss_main": 0.02862297184765339,
      "step": 47690
    },
    {
      "epoch": 5.581558623917623,
      "grad_norm": 0.2873311936855316,
      "learning_rate": 1.1057864483009262e-05,
      "loss": 0.0474,
      "step": 47700
    },
    {
      "epoch": 5.581558623917623,
      "loss_main": 0.08766770362854004,
      "step": 47700
    },
    {
      "epoch": 5.582728761993915,
      "grad_norm": 0.2616598606109619,
      "learning_rate": 1.1040585596043473e-05,
      "loss": 0.0511,
      "step": 47710
    },
    {
      "epoch": 5.582728761993915,
      "loss_main": 0.04552235081791878,
      "step": 47710
    },
    {
      "epoch": 5.583898900070208,
      "grad_norm": 0.16995370388031006,
      "learning_rate": 1.1023318543919148e-05,
      "loss": 0.0519,
      "step": 47720
    },
    {
      "epoch": 5.583898900070208,
      "loss_main": 0.03546915575861931,
      "step": 47720
    },
    {
      "epoch": 5.585069038146501,
      "grad_norm": 0.2830187976360321,
      "learning_rate": 1.10060633318815e-05,
      "loss": 0.0474,
      "step": 47730
    },
    {
      "epoch": 5.585069038146501,
      "loss_main": 0.027398796752095222,
      "step": 47730
    },
    {
      "epoch": 5.5862391762227945,
      "grad_norm": 0.28962740302085876,
      "learning_rate": 1.0988819965172248e-05,
      "loss": 0.0509,
      "step": 47740
    },
    {
      "epoch": 5.5862391762227945,
      "loss_main": 0.07723566144704819,
      "step": 47740
    },
    {
      "epoch": 5.587409314299087,
      "grad_norm": 0.2366587370634079,
      "learning_rate": 1.0971588449029462e-05,
      "loss": 0.0506,
      "step": 47750
    },
    {
      "epoch": 5.587409314299087,
      "loss_main": 0.06655935943126678,
      "step": 47750
    },
    {
      "epoch": 5.58857945237538,
      "grad_norm": 0.31513896584510803,
      "learning_rate": 1.095436878868762e-05,
      "loss": 0.0557,
      "step": 47760
    },
    {
      "epoch": 5.58857945237538,
      "loss_main": 0.04624065011739731,
      "step": 47760
    },
    {
      "epoch": 5.589749590451674,
      "grad_norm": 0.1848800629377365,
      "learning_rate": 1.0937160989377598e-05,
      "loss": 0.0481,
      "step": 47770
    },
    {
      "epoch": 5.589749590451674,
      "loss_main": 0.03943558782339096,
      "step": 47770
    },
    {
      "epoch": 5.590919728527966,
      "grad_norm": 0.34513059258461,
      "learning_rate": 1.0919965056326676e-05,
      "loss": 0.047,
      "step": 47780
    },
    {
      "epoch": 5.590919728527966,
      "loss_main": 0.032457899302244186,
      "step": 47780
    },
    {
      "epoch": 5.592089866604259,
      "grad_norm": 0.2484450489282608,
      "learning_rate": 1.0902780994758504e-05,
      "loss": 0.0458,
      "step": 47790
    },
    {
      "epoch": 5.592089866604259,
      "loss_main": 0.07579454034566879,
      "step": 47790
    },
    {
      "epoch": 5.593260004680552,
      "grad_norm": 0.21820855140686035,
      "learning_rate": 1.0885608809893193e-05,
      "loss": 0.0437,
      "step": 47800
    },
    {
      "epoch": 5.593260004680552,
      "loss_main": 0.032489705830812454,
      "step": 47800
    },
    {
      "epoch": 5.5944301427568455,
      "grad_norm": 0.19064298272132874,
      "learning_rate": 1.0868448506947142e-05,
      "loss": 0.0472,
      "step": 47810
    },
    {
      "epoch": 5.5944301427568455,
      "loss_main": 0.04930320382118225,
      "step": 47810
    },
    {
      "epoch": 5.595600280833138,
      "grad_norm": 0.25806599855422974,
      "learning_rate": 1.0851300091133243e-05,
      "loss": 0.0512,
      "step": 47820
    },
    {
      "epoch": 5.595600280833138,
      "loss_main": 0.04303956776857376,
      "step": 47820
    },
    {
      "epoch": 5.596770418909431,
      "grad_norm": 0.24048563838005066,
      "learning_rate": 1.083416356766071e-05,
      "loss": 0.0471,
      "step": 47830
    },
    {
      "epoch": 5.596770418909431,
      "loss_main": 0.05801002308726311,
      "step": 47830
    },
    {
      "epoch": 5.597940556985725,
      "grad_norm": 0.19694602489471436,
      "learning_rate": 1.0817038941735175e-05,
      "loss": 0.055,
      "step": 47840
    },
    {
      "epoch": 5.597940556985725,
      "loss_main": 0.05131077021360397,
      "step": 47840
    },
    {
      "epoch": 5.599110695062017,
      "grad_norm": 0.18973703682422638,
      "learning_rate": 1.0799926218558642e-05,
      "loss": 0.0442,
      "step": 47850
    },
    {
      "epoch": 5.599110695062017,
      "loss_main": 0.05875571817159653,
      "step": 47850
    },
    {
      "epoch": 5.60028083313831,
      "grad_norm": 0.2770174443721771,
      "learning_rate": 1.0782825403329488e-05,
      "loss": 0.0523,
      "step": 47860
    },
    {
      "epoch": 5.60028083313831,
      "loss_main": 0.07306419312953949,
      "step": 47860
    },
    {
      "epoch": 5.601450971214604,
      "grad_norm": 0.2790743410587311,
      "learning_rate": 1.076573650124254e-05,
      "loss": 0.0439,
      "step": 47870
    },
    {
      "epoch": 5.601450971214604,
      "loss_main": 0.06790141761302948,
      "step": 47870
    },
    {
      "epoch": 5.602621109290896,
      "grad_norm": 0.2490013986825943,
      "learning_rate": 1.0748659517488891e-05,
      "loss": 0.0476,
      "step": 47880
    },
    {
      "epoch": 5.602621109290896,
      "loss_main": 0.043474383652210236,
      "step": 47880
    },
    {
      "epoch": 5.603791247367189,
      "grad_norm": 0.2509208917617798,
      "learning_rate": 1.0731594457256138e-05,
      "loss": 0.046,
      "step": 47890
    },
    {
      "epoch": 5.603791247367189,
      "loss_main": 0.06142500415444374,
      "step": 47890
    },
    {
      "epoch": 5.604961385443483,
      "grad_norm": 0.20846962928771973,
      "learning_rate": 1.0714541325728139e-05,
      "loss": 0.0486,
      "step": 47900
    },
    {
      "epoch": 5.604961385443483,
      "loss_main": 0.05377852916717529,
      "step": 47900
    },
    {
      "epoch": 5.6061315235197755,
      "grad_norm": 0.2970292270183563,
      "learning_rate": 1.0697500128085231e-05,
      "loss": 0.0406,
      "step": 47910
    },
    {
      "epoch": 5.6061315235197755,
      "loss_main": 0.04631870985031128,
      "step": 47910
    },
    {
      "epoch": 5.607301661596068,
      "grad_norm": 0.1988346427679062,
      "learning_rate": 1.0680470869504055e-05,
      "loss": 0.0499,
      "step": 47920
    },
    {
      "epoch": 5.607301661596068,
      "loss_main": 0.04568744823336601,
      "step": 47920
    },
    {
      "epoch": 5.608471799672361,
      "grad_norm": 0.27412131428718567,
      "learning_rate": 1.066345355515766e-05,
      "loss": 0.0506,
      "step": 47930
    },
    {
      "epoch": 5.608471799672361,
      "loss_main": 0.02697819471359253,
      "step": 47930
    },
    {
      "epoch": 5.609641937748655,
      "grad_norm": 0.2982269525527954,
      "learning_rate": 1.0646448190215453e-05,
      "loss": 0.0448,
      "step": 47940
    },
    {
      "epoch": 5.609641937748655,
      "loss_main": 0.04132915288209915,
      "step": 47940
    },
    {
      "epoch": 5.610812075824947,
      "grad_norm": 0.24331924319267273,
      "learning_rate": 1.0629454779843217e-05,
      "loss": 0.0435,
      "step": 47950
    },
    {
      "epoch": 5.610812075824947,
      "loss_main": 0.04526519402861595,
      "step": 47950
    },
    {
      "epoch": 5.61198221390124,
      "grad_norm": 0.2868945896625519,
      "learning_rate": 1.0612473329203082e-05,
      "loss": 0.051,
      "step": 47960
    },
    {
      "epoch": 5.61198221390124,
      "loss_main": 0.03658909723162651,
      "step": 47960
    },
    {
      "epoch": 5.613152351977534,
      "grad_norm": 0.20402128994464874,
      "learning_rate": 1.0595503843453596e-05,
      "loss": 0.0484,
      "step": 47970
    },
    {
      "epoch": 5.613152351977534,
      "loss_main": 0.10912307351827621,
      "step": 47970
    },
    {
      "epoch": 5.614322490053826,
      "grad_norm": 0.2997758686542511,
      "learning_rate": 1.0578546327749634e-05,
      "loss": 0.0542,
      "step": 47980
    },
    {
      "epoch": 5.614322490053826,
      "loss_main": 0.03645811602473259,
      "step": 47980
    },
    {
      "epoch": 5.615492628130119,
      "grad_norm": 0.19336707890033722,
      "learning_rate": 1.0561600787242425e-05,
      "loss": 0.0489,
      "step": 47990
    },
    {
      "epoch": 5.615492628130119,
      "loss_main": 0.04479489102959633,
      "step": 47990
    },
    {
      "epoch": 5.616662766206412,
      "grad_norm": 0.22689470648765564,
      "learning_rate": 1.0544667227079591e-05,
      "loss": 0.0465,
      "step": 48000
    },
    {
      "epoch": 5.616662766206412,
      "loss_main": 0.05972548946738243,
      "step": 48000
    },
    {
      "epoch": 5.6178329042827055,
      "grad_norm": 0.19549532234668732,
      "learning_rate": 1.0527745652405085e-05,
      "loss": 0.0497,
      "step": 48010
    },
    {
      "epoch": 5.6178329042827055,
      "loss_main": 0.035252995789051056,
      "step": 48010
    },
    {
      "epoch": 5.619003042358998,
      "grad_norm": 0.16393889486789703,
      "learning_rate": 1.051083606835927e-05,
      "loss": 0.0455,
      "step": 48020
    },
    {
      "epoch": 5.619003042358998,
      "loss_main": 0.04245573654770851,
      "step": 48020
    },
    {
      "epoch": 5.620173180435291,
      "grad_norm": 0.22301746904850006,
      "learning_rate": 1.049393848007878e-05,
      "loss": 0.0443,
      "step": 48030
    },
    {
      "epoch": 5.620173180435291,
      "loss_main": 0.039835598319768906,
      "step": 48030
    },
    {
      "epoch": 5.621343318511585,
      "grad_norm": 0.27701887488365173,
      "learning_rate": 1.0477052892696709e-05,
      "loss": 0.0564,
      "step": 48040
    },
    {
      "epoch": 5.621343318511585,
      "loss_main": 0.02466881275177002,
      "step": 48040
    },
    {
      "epoch": 5.622513456587877,
      "grad_norm": 0.22307977080345154,
      "learning_rate": 1.0460179311342394e-05,
      "loss": 0.0453,
      "step": 48050
    },
    {
      "epoch": 5.622513456587877,
      "loss_main": 0.05302487686276436,
      "step": 48050
    },
    {
      "epoch": 5.62368359466417,
      "grad_norm": 0.20655213296413422,
      "learning_rate": 1.0443317741141634e-05,
      "loss": 0.0364,
      "step": 48060
    },
    {
      "epoch": 5.62368359466417,
      "loss_main": 0.03074703924357891,
      "step": 48060
    },
    {
      "epoch": 5.624853732740464,
      "grad_norm": 0.26918765902519226,
      "learning_rate": 1.0426468187216514e-05,
      "loss": 0.0493,
      "step": 48070
    },
    {
      "epoch": 5.624853732740464,
      "loss_main": 0.04369582608342171,
      "step": 48070
    },
    {
      "epoch": 5.6260238708167565,
      "grad_norm": 0.20308366417884827,
      "learning_rate": 1.0409630654685477e-05,
      "loss": 0.0405,
      "step": 48080
    },
    {
      "epoch": 5.6260238708167565,
      "loss_main": 0.025892721489071846,
      "step": 48080
    },
    {
      "epoch": 5.627194008893049,
      "grad_norm": 0.2389388531446457,
      "learning_rate": 1.039280514866332e-05,
      "loss": 0.0448,
      "step": 48090
    },
    {
      "epoch": 5.627194008893049,
      "loss_main": 0.04131621867418289,
      "step": 48090
    },
    {
      "epoch": 5.628364146969343,
      "grad_norm": 0.23286943137645721,
      "learning_rate": 1.0375991674261198e-05,
      "loss": 0.0475,
      "step": 48100
    },
    {
      "epoch": 5.628364146969343,
      "loss_main": 0.02885025180876255,
      "step": 48100
    },
    {
      "epoch": 5.629534285045636,
      "grad_norm": 0.2374284863471985,
      "learning_rate": 1.0359190236586575e-05,
      "loss": 0.0487,
      "step": 48110
    },
    {
      "epoch": 5.629534285045636,
      "loss_main": 0.048050377517938614,
      "step": 48110
    },
    {
      "epoch": 5.630704423121928,
      "grad_norm": 0.627871036529541,
      "learning_rate": 1.0342400840743322e-05,
      "loss": 0.044,
      "step": 48120
    },
    {
      "epoch": 5.630704423121928,
      "loss_main": 0.03160979598760605,
      "step": 48120
    },
    {
      "epoch": 5.631874561198221,
      "grad_norm": 0.2154681384563446,
      "learning_rate": 1.0325623491831593e-05,
      "loss": 0.0453,
      "step": 48130
    },
    {
      "epoch": 5.631874561198221,
      "loss_main": 0.028493642807006836,
      "step": 48130
    },
    {
      "epoch": 5.633044699274515,
      "grad_norm": 0.17727307975292206,
      "learning_rate": 1.0308858194947906e-05,
      "loss": 0.0528,
      "step": 48140
    },
    {
      "epoch": 5.633044699274515,
      "loss_main": 0.03874307870864868,
      "step": 48140
    },
    {
      "epoch": 5.634214837350807,
      "grad_norm": 0.21581166982650757,
      "learning_rate": 1.0292104955185111e-05,
      "loss": 0.0525,
      "step": 48150
    },
    {
      "epoch": 5.634214837350807,
      "loss_main": 0.02757638692855835,
      "step": 48150
    },
    {
      "epoch": 5.6353849754271,
      "grad_norm": 0.2871497869491577,
      "learning_rate": 1.0275363777632396e-05,
      "loss": 0.0433,
      "step": 48160
    },
    {
      "epoch": 5.6353849754271,
      "loss_main": 0.03595453500747681,
      "step": 48160
    },
    {
      "epoch": 5.636555113503394,
      "grad_norm": 0.2303854078054428,
      "learning_rate": 1.0258634667375321e-05,
      "loss": 0.0561,
      "step": 48170
    },
    {
      "epoch": 5.636555113503394,
      "loss_main": 0.03695288673043251,
      "step": 48170
    },
    {
      "epoch": 5.6377252515796865,
      "grad_norm": 0.25550198554992676,
      "learning_rate": 1.02419176294957e-05,
      "loss": 0.0522,
      "step": 48180
    },
    {
      "epoch": 5.6377252515796865,
      "loss_main": 0.027317821979522705,
      "step": 48180
    },
    {
      "epoch": 5.638895389655979,
      "grad_norm": 0.21226420998573303,
      "learning_rate": 1.0225212669071782e-05,
      "loss": 0.0448,
      "step": 48190
    },
    {
      "epoch": 5.638895389655979,
      "loss_main": 0.03183307498693466,
      "step": 48190
    },
    {
      "epoch": 5.640065527732272,
      "grad_norm": 0.26354315876960754,
      "learning_rate": 1.0208519791178029e-05,
      "loss": 0.0438,
      "step": 48200
    },
    {
      "epoch": 5.640065527732272,
      "loss_main": 0.05897148326039314,
      "step": 48200
    },
    {
      "epoch": 5.641235665808566,
      "grad_norm": 0.23663438856601715,
      "learning_rate": 1.019183900088535e-05,
      "loss": 0.0483,
      "step": 48210
    },
    {
      "epoch": 5.641235665808566,
      "loss_main": 0.04260125383734703,
      "step": 48210
    },
    {
      "epoch": 5.642405803884858,
      "grad_norm": 0.26719963550567627,
      "learning_rate": 1.0175170303260906e-05,
      "loss": 0.0502,
      "step": 48220
    },
    {
      "epoch": 5.642405803884858,
      "loss_main": 0.03891468793153763,
      "step": 48220
    },
    {
      "epoch": 5.643575941961151,
      "grad_norm": 0.242527037858963,
      "learning_rate": 1.0158513703368206e-05,
      "loss": 0.0501,
      "step": 48230
    },
    {
      "epoch": 5.643575941961151,
      "loss_main": 0.01840418577194214,
      "step": 48230
    },
    {
      "epoch": 5.644746080037445,
      "grad_norm": 0.25500428676605225,
      "learning_rate": 1.0141869206267095e-05,
      "loss": 0.0437,
      "step": 48240
    },
    {
      "epoch": 5.644746080037445,
      "loss_main": 0.05223807319998741,
      "step": 48240
    },
    {
      "epoch": 5.6459162181137375,
      "grad_norm": 0.20978035032749176,
      "learning_rate": 1.0125236817013723e-05,
      "loss": 0.044,
      "step": 48250
    },
    {
      "epoch": 5.6459162181137375,
      "loss_main": 0.03557707741856575,
      "step": 48250
    },
    {
      "epoch": 5.64708635619003,
      "grad_norm": 0.21028514206409454,
      "learning_rate": 1.010861654066056e-05,
      "loss": 0.0445,
      "step": 48260
    },
    {
      "epoch": 5.64708635619003,
      "loss_main": 0.06594377011060715,
      "step": 48260
    },
    {
      "epoch": 5.648256494266324,
      "grad_norm": 0.25972291827201843,
      "learning_rate": 1.0092008382256434e-05,
      "loss": 0.053,
      "step": 48270
    },
    {
      "epoch": 5.648256494266324,
      "loss_main": 0.04605580121278763,
      "step": 48270
    },
    {
      "epoch": 5.6494266323426165,
      "grad_norm": 0.21901321411132812,
      "learning_rate": 1.0075412346846458e-05,
      "loss": 0.0464,
      "step": 48280
    },
    {
      "epoch": 5.6494266323426165,
      "loss_main": 0.05670415610074997,
      "step": 48280
    },
    {
      "epoch": 5.650596770418909,
      "grad_norm": 0.22395750880241394,
      "learning_rate": 1.0058828439472056e-05,
      "loss": 0.0446,
      "step": 48290
    },
    {
      "epoch": 5.650596770418909,
      "loss_main": 0.021839192137122154,
      "step": 48290
    },
    {
      "epoch": 5.651766908495203,
      "grad_norm": 0.3049758970737457,
      "learning_rate": 1.0042256665170996e-05,
      "loss": 0.0499,
      "step": 48300
    },
    {
      "epoch": 5.651766908495203,
      "loss_main": 0.06687074899673462,
      "step": 48300
    },
    {
      "epoch": 5.652937046571496,
      "grad_norm": 0.21737423539161682,
      "learning_rate": 1.0025697028977332e-05,
      "loss": 0.0449,
      "step": 48310
    },
    {
      "epoch": 5.652937046571496,
      "loss_main": 0.06430351734161377,
      "step": 48310
    },
    {
      "epoch": 5.654107184647788,
      "grad_norm": 0.20375436544418335,
      "learning_rate": 1.0009149535921454e-05,
      "loss": 0.0456,
      "step": 48320
    },
    {
      "epoch": 5.654107184647788,
      "loss_main": 0.036173589527606964,
      "step": 48320
    },
    {
      "epoch": 5.655277322724081,
      "grad_norm": 0.1925565004348755,
      "learning_rate": 9.992614191030031e-06,
      "loss": 0.0424,
      "step": 48330
    },
    {
      "epoch": 5.655277322724081,
      "loss_main": 0.03258350118994713,
      "step": 48330
    },
    {
      "epoch": 5.656447460800375,
      "grad_norm": 0.2508280873298645,
      "learning_rate": 9.976090999326115e-06,
      "loss": 0.055,
      "step": 48340
    },
    {
      "epoch": 5.656447460800375,
      "loss_main": 0.04647771641612053,
      "step": 48340
    },
    {
      "epoch": 5.6576175988766675,
      "grad_norm": 0.17634546756744385,
      "learning_rate": 9.959579965828952e-06,
      "loss": 0.0449,
      "step": 48350
    },
    {
      "epoch": 5.6576175988766675,
      "loss_main": 0.05372264236211777,
      "step": 48350
    },
    {
      "epoch": 5.65878773695296,
      "grad_norm": 0.20552511513233185,
      "learning_rate": 9.943081095554218e-06,
      "loss": 0.0463,
      "step": 48360
    },
    {
      "epoch": 5.65878773695296,
      "loss_main": 0.03820931166410446,
      "step": 48360
    },
    {
      "epoch": 5.659957875029253,
      "grad_norm": 0.30937492847442627,
      "learning_rate": 9.926594393513783e-06,
      "loss": 0.0502,
      "step": 48370
    },
    {
      "epoch": 5.659957875029253,
      "loss_main": 0.03690910339355469,
      "step": 48370
    },
    {
      "epoch": 5.661128013105547,
      "grad_norm": 0.2871842682361603,
      "learning_rate": 9.910119864715906e-06,
      "loss": 0.0566,
      "step": 48380
    },
    {
      "epoch": 5.661128013105547,
      "loss_main": 0.048132866621017456,
      "step": 48380
    },
    {
      "epoch": 5.662298151181839,
      "grad_norm": 0.25078386068344116,
      "learning_rate": 9.8936575141651e-06,
      "loss": 0.0464,
      "step": 48390
    },
    {
      "epoch": 5.662298151181839,
      "loss_main": 0.044875603169202805,
      "step": 48390
    },
    {
      "epoch": 5.663468289258132,
      "grad_norm": 0.18094368278980255,
      "learning_rate": 9.877207346862194e-06,
      "loss": 0.0506,
      "step": 48400
    },
    {
      "epoch": 5.663468289258132,
      "loss_main": 0.06853894889354706,
      "step": 48400
    },
    {
      "epoch": 5.664638427334426,
      "grad_norm": 0.236953467130661,
      "learning_rate": 9.860769367804312e-06,
      "loss": 0.0488,
      "step": 48410
    },
    {
      "epoch": 5.664638427334426,
      "loss_main": 0.04064704850316048,
      "step": 48410
    },
    {
      "epoch": 5.665808565410718,
      "grad_norm": 0.2182101160287857,
      "learning_rate": 9.844343581984877e-06,
      "loss": 0.0472,
      "step": 48420
    },
    {
      "epoch": 5.665808565410718,
      "loss_main": 0.03414037823677063,
      "step": 48420
    },
    {
      "epoch": 5.666978703487011,
      "grad_norm": 0.23643600940704346,
      "learning_rate": 9.82792999439362e-06,
      "loss": 0.0488,
      "step": 48430
    },
    {
      "epoch": 5.666978703487011,
      "loss_main": 0.03353778272867203,
      "step": 48430
    },
    {
      "epoch": 5.668148841563305,
      "grad_norm": 0.3180437684059143,
      "learning_rate": 9.811528610016546e-06,
      "loss": 0.0432,
      "step": 48440
    },
    {
      "epoch": 5.668148841563305,
      "loss_main": 0.03882845491170883,
      "step": 48440
    },
    {
      "epoch": 5.6693189796395975,
      "grad_norm": 0.20725351572036743,
      "learning_rate": 9.79513943383597e-06,
      "loss": 0.0447,
      "step": 48450
    },
    {
      "epoch": 5.6693189796395975,
      "loss_main": 0.039276327937841415,
      "step": 48450
    },
    {
      "epoch": 5.67048911771589,
      "grad_norm": 0.1661958396434784,
      "learning_rate": 9.778762470830489e-06,
      "loss": 0.0426,
      "step": 48460
    },
    {
      "epoch": 5.67048911771589,
      "loss_main": 0.051769744604825974,
      "step": 48460
    },
    {
      "epoch": 5.671659255792184,
      "grad_norm": 0.2352309674024582,
      "learning_rate": 9.762397725974982e-06,
      "loss": 0.0429,
      "step": 48470
    },
    {
      "epoch": 5.671659255792184,
      "loss_main": 0.05067703500390053,
      "step": 48470
    },
    {
      "epoch": 5.672829393868477,
      "grad_norm": 0.3107735216617584,
      "learning_rate": 9.746045204240622e-06,
      "loss": 0.0479,
      "step": 48480
    },
    {
      "epoch": 5.672829393868477,
      "loss_main": 0.03890976682305336,
      "step": 48480
    },
    {
      "epoch": 5.673999531944769,
      "grad_norm": 0.2029576152563095,
      "learning_rate": 9.729704910594917e-06,
      "loss": 0.0461,
      "step": 48490
    },
    {
      "epoch": 5.673999531944769,
      "loss_main": 0.03306692838668823,
      "step": 48490
    },
    {
      "epoch": 5.675169670021063,
      "grad_norm": 0.2498919814825058,
      "learning_rate": 9.713376850001554e-06,
      "loss": 0.0418,
      "step": 48500
    },
    {
      "epoch": 5.675169670021063,
      "loss_main": 0.09463449567556381,
      "step": 48500
    },
    {
      "epoch": 5.676339808097356,
      "grad_norm": 0.1588873267173767,
      "learning_rate": 9.697061027420622e-06,
      "loss": 0.0463,
      "step": 48510
    },
    {
      "epoch": 5.676339808097356,
      "loss_main": 0.04936569929122925,
      "step": 48510
    },
    {
      "epoch": 5.6775099461736485,
      "grad_norm": 0.21723337471485138,
      "learning_rate": 9.680757447808385e-06,
      "loss": 0.0522,
      "step": 48520
    },
    {
      "epoch": 5.6775099461736485,
      "loss_main": 0.035415228456258774,
      "step": 48520
    },
    {
      "epoch": 5.678680084249941,
      "grad_norm": 0.33548471331596375,
      "learning_rate": 9.664466116117488e-06,
      "loss": 0.0432,
      "step": 48530
    },
    {
      "epoch": 5.678680084249941,
      "loss_main": 0.04148051515221596,
      "step": 48530
    },
    {
      "epoch": 5.679850222326235,
      "grad_norm": 0.21204154193401337,
      "learning_rate": 9.64818703729678e-06,
      "loss": 0.0444,
      "step": 48540
    },
    {
      "epoch": 5.679850222326235,
      "loss_main": 0.042165517807006836,
      "step": 48540
    },
    {
      "epoch": 5.681020360402528,
      "grad_norm": 0.1646329164505005,
      "learning_rate": 9.631920216291423e-06,
      "loss": 0.0424,
      "step": 48550
    },
    {
      "epoch": 5.681020360402528,
      "loss_main": 0.06522376835346222,
      "step": 48550
    },
    {
      "epoch": 5.68219049847882,
      "grad_norm": 0.22401180863380432,
      "learning_rate": 9.615665658042849e-06,
      "loss": 0.0521,
      "step": 48560
    },
    {
      "epoch": 5.68219049847882,
      "loss_main": 0.03356301411986351,
      "step": 48560
    },
    {
      "epoch": 5.683360636555113,
      "grad_norm": 0.2822354733943939,
      "learning_rate": 9.599423367488747e-06,
      "loss": 0.0444,
      "step": 48570
    },
    {
      "epoch": 5.683360636555113,
      "loss_main": 0.07796651870012283,
      "step": 48570
    },
    {
      "epoch": 5.684530774631407,
      "grad_norm": 0.2592875361442566,
      "learning_rate": 9.583193349563124e-06,
      "loss": 0.0456,
      "step": 48580
    },
    {
      "epoch": 5.684530774631407,
      "loss_main": 0.02825196646153927,
      "step": 48580
    },
    {
      "epoch": 5.685700912707699,
      "grad_norm": 0.3029564917087555,
      "learning_rate": 9.566975609196216e-06,
      "loss": 0.0484,
      "step": 48590
    },
    {
      "epoch": 5.685700912707699,
      "loss_main": 0.09424686431884766,
      "step": 48590
    },
    {
      "epoch": 5.686871050783992,
      "grad_norm": 0.2001243680715561,
      "learning_rate": 9.550770151314548e-06,
      "loss": 0.0466,
      "step": 48600
    },
    {
      "epoch": 5.686871050783992,
      "loss_main": 0.05796432867646217,
      "step": 48600
    },
    {
      "epoch": 5.688041188860286,
      "grad_norm": 0.3075161278247833,
      "learning_rate": 9.53457698084091e-06,
      "loss": 0.044,
      "step": 48610
    },
    {
      "epoch": 5.688041188860286,
      "loss_main": 0.04317983239889145,
      "step": 48610
    },
    {
      "epoch": 5.6892113269365785,
      "grad_norm": 0.2520451843738556,
      "learning_rate": 9.518396102694355e-06,
      "loss": 0.0451,
      "step": 48620
    },
    {
      "epoch": 5.6892113269365785,
      "loss_main": 0.07370965927839279,
      "step": 48620
    },
    {
      "epoch": 5.690381465012871,
      "grad_norm": 0.1978195458650589,
      "learning_rate": 9.502227521790198e-06,
      "loss": 0.0538,
      "step": 48630
    },
    {
      "epoch": 5.690381465012871,
      "loss_main": 0.03197454661130905,
      "step": 48630
    },
    {
      "epoch": 5.691551603089165,
      "grad_norm": 0.21465040743350983,
      "learning_rate": 9.486071243040063e-06,
      "loss": 0.0492,
      "step": 48640
    },
    {
      "epoch": 5.691551603089165,
      "loss_main": 0.05031647905707359,
      "step": 48640
    },
    {
      "epoch": 5.692721741165458,
      "grad_norm": 0.17712265253067017,
      "learning_rate": 9.469927271351747e-06,
      "loss": 0.0489,
      "step": 48650
    },
    {
      "epoch": 5.692721741165458,
      "loss_main": 0.06954988092184067,
      "step": 48650
    },
    {
      "epoch": 5.69389187924175,
      "grad_norm": 0.1859876960515976,
      "learning_rate": 9.453795611629419e-06,
      "loss": 0.0468,
      "step": 48660
    },
    {
      "epoch": 5.69389187924175,
      "loss_main": 0.04988369718194008,
      "step": 48660
    },
    {
      "epoch": 5.695062017318044,
      "grad_norm": 0.2671433985233307,
      "learning_rate": 9.437676268773399e-06,
      "loss": 0.046,
      "step": 48670
    },
    {
      "epoch": 5.695062017318044,
      "loss_main": 0.048279549926519394,
      "step": 48670
    },
    {
      "epoch": 5.696232155394337,
      "grad_norm": 0.25896212458610535,
      "learning_rate": 9.421569247680357e-06,
      "loss": 0.0498,
      "step": 48680
    },
    {
      "epoch": 5.696232155394337,
      "loss_main": 0.04951527342200279,
      "step": 48680
    },
    {
      "epoch": 5.697402293470629,
      "grad_norm": 0.1745186597108841,
      "learning_rate": 9.40547455324316e-06,
      "loss": 0.0474,
      "step": 48690
    },
    {
      "epoch": 5.697402293470629,
      "loss_main": 0.07881660014390945,
      "step": 48690
    },
    {
      "epoch": 5.698572431546922,
      "grad_norm": 0.2538129687309265,
      "learning_rate": 9.389392190350965e-06,
      "loss": 0.0449,
      "step": 48700
    },
    {
      "epoch": 5.698572431546922,
      "loss_main": 0.053553614765405655,
      "step": 48700
    },
    {
      "epoch": 5.699742569623216,
      "grad_norm": 0.2933681309223175,
      "learning_rate": 9.373322163889153e-06,
      "loss": 0.0513,
      "step": 48710
    },
    {
      "epoch": 5.699742569623216,
      "loss_main": 0.0454065166413784,
      "step": 48710
    },
    {
      "epoch": 5.7009127076995085,
      "grad_norm": 0.2385723739862442,
      "learning_rate": 9.357264478739375e-06,
      "loss": 0.0469,
      "step": 48720
    },
    {
      "epoch": 5.7009127076995085,
      "loss_main": 0.04124719277024269,
      "step": 48720
    },
    {
      "epoch": 5.702082845775801,
      "grad_norm": 0.17999282479286194,
      "learning_rate": 9.341219139779567e-06,
      "loss": 0.0425,
      "step": 48730
    },
    {
      "epoch": 5.702082845775801,
      "loss_main": 0.04054178670048714,
      "step": 48730
    },
    {
      "epoch": 5.703252983852095,
      "grad_norm": 0.2124258577823639,
      "learning_rate": 9.325186151883824e-06,
      "loss": 0.0444,
      "step": 48740
    },
    {
      "epoch": 5.703252983852095,
      "loss_main": 0.05767956003546715,
      "step": 48740
    },
    {
      "epoch": 5.704423121928388,
      "grad_norm": 0.2678075432777405,
      "learning_rate": 9.30916551992258e-06,
      "loss": 0.0459,
      "step": 48750
    },
    {
      "epoch": 5.704423121928388,
      "loss_main": 0.08948910236358643,
      "step": 48750
    },
    {
      "epoch": 5.70559326000468,
      "grad_norm": 0.30619707703590393,
      "learning_rate": 9.293157248762479e-06,
      "loss": 0.0524,
      "step": 48760
    },
    {
      "epoch": 5.70559326000468,
      "loss_main": 0.046619705855846405,
      "step": 48760
    },
    {
      "epoch": 5.706763398080973,
      "grad_norm": 0.25282955169677734,
      "learning_rate": 9.2771613432664e-06,
      "loss": 0.0454,
      "step": 48770
    },
    {
      "epoch": 5.706763398080973,
      "loss_main": 0.10625944286584854,
      "step": 48770
    },
    {
      "epoch": 5.707933536157267,
      "grad_norm": 0.40843409299850464,
      "learning_rate": 9.261177808293481e-06,
      "loss": 0.0593,
      "step": 48780
    },
    {
      "epoch": 5.707933536157267,
      "loss_main": 0.026843326166272163,
      "step": 48780
    },
    {
      "epoch": 5.7091036742335595,
      "grad_norm": 0.2252713441848755,
      "learning_rate": 9.245206648699096e-06,
      "loss": 0.0464,
      "step": 48790
    },
    {
      "epoch": 5.7091036742335595,
      "loss_main": 0.042462535202503204,
      "step": 48790
    },
    {
      "epoch": 5.710273812309852,
      "grad_norm": 0.22139613330364227,
      "learning_rate": 9.22924786933485e-06,
      "loss": 0.0465,
      "step": 48800
    },
    {
      "epoch": 5.710273812309852,
      "loss_main": 0.06682677567005157,
      "step": 48800
    },
    {
      "epoch": 5.711443950386146,
      "grad_norm": 0.19388771057128906,
      "learning_rate": 9.213301475048642e-06,
      "loss": 0.0456,
      "step": 48810
    },
    {
      "epoch": 5.711443950386146,
      "loss_main": 0.0456244982779026,
      "step": 48810
    },
    {
      "epoch": 5.712614088462439,
      "grad_norm": 0.30262452363967896,
      "learning_rate": 9.197367470684504e-06,
      "loss": 0.0435,
      "step": 48820
    },
    {
      "epoch": 5.712614088462439,
      "loss_main": 0.04497641324996948,
      "step": 48820
    },
    {
      "epoch": 5.713784226538731,
      "grad_norm": 0.21967239677906036,
      "learning_rate": 9.181445861082816e-06,
      "loss": 0.0474,
      "step": 48830
    },
    {
      "epoch": 5.713784226538731,
      "loss_main": 0.0522589348256588,
      "step": 48830
    },
    {
      "epoch": 5.714954364615025,
      "grad_norm": 0.2652589678764343,
      "learning_rate": 9.16553665108012e-06,
      "loss": 0.0579,
      "step": 48840
    },
    {
      "epoch": 5.714954364615025,
      "loss_main": 0.04015318676829338,
      "step": 48840
    },
    {
      "epoch": 5.716124502691318,
      "grad_norm": 0.23037533462047577,
      "learning_rate": 9.149639845509223e-06,
      "loss": 0.0438,
      "step": 48850
    },
    {
      "epoch": 5.716124502691318,
      "loss_main": 0.0378984734416008,
      "step": 48850
    },
    {
      "epoch": 5.71729464076761,
      "grad_norm": 0.2392030507326126,
      "learning_rate": 9.133755449199144e-06,
      "loss": 0.0423,
      "step": 48860
    },
    {
      "epoch": 5.71729464076761,
      "loss_main": 0.047795575112104416,
      "step": 48860
    },
    {
      "epoch": 5.718464778843904,
      "grad_norm": 0.24354144930839539,
      "learning_rate": 9.117883466975135e-06,
      "loss": 0.0401,
      "step": 48870
    },
    {
      "epoch": 5.718464778843904,
      "loss_main": 0.024575021117925644,
      "step": 48870
    },
    {
      "epoch": 5.719634916920197,
      "grad_norm": 0.19255121052265167,
      "learning_rate": 9.10202390365873e-06,
      "loss": 0.0436,
      "step": 48880
    },
    {
      "epoch": 5.719634916920197,
      "loss_main": 0.04211845621466637,
      "step": 48880
    },
    {
      "epoch": 5.7208050549964895,
      "grad_norm": 0.1382124274969101,
      "learning_rate": 9.086176764067583e-06,
      "loss": 0.0433,
      "step": 48890
    },
    {
      "epoch": 5.7208050549964895,
      "loss_main": 0.041110653430223465,
      "step": 48890
    },
    {
      "epoch": 5.721975193072782,
      "grad_norm": 0.3477622866630554,
      "learning_rate": 9.070342053015684e-06,
      "loss": 0.0528,
      "step": 48900
    },
    {
      "epoch": 5.721975193072782,
      "loss_main": 0.029462307691574097,
      "step": 48900
    },
    {
      "epoch": 5.723145331149076,
      "grad_norm": 0.19695505499839783,
      "learning_rate": 9.054519775313187e-06,
      "loss": 0.0475,
      "step": 48910
    },
    {
      "epoch": 5.723145331149076,
      "loss_main": 0.0975937470793724,
      "step": 48910
    },
    {
      "epoch": 5.724315469225369,
      "grad_norm": 0.21088428795337677,
      "learning_rate": 9.038709935766476e-06,
      "loss": 0.0472,
      "step": 48920
    },
    {
      "epoch": 5.724315469225369,
      "loss_main": 0.028918812051415443,
      "step": 48920
    },
    {
      "epoch": 5.725485607301661,
      "grad_norm": 0.2304942011833191,
      "learning_rate": 9.02291253917817e-06,
      "loss": 0.0511,
      "step": 48930
    },
    {
      "epoch": 5.725485607301661,
      "loss_main": 0.059108588844537735,
      "step": 48930
    },
    {
      "epoch": 5.726655745377955,
      "grad_norm": 0.2377745807170868,
      "learning_rate": 9.007127590347091e-06,
      "loss": 0.0499,
      "step": 48940
    },
    {
      "epoch": 5.726655745377955,
      "loss_main": 0.03794478625059128,
      "step": 48940
    },
    {
      "epoch": 5.727825883454248,
      "grad_norm": 0.23937493562698364,
      "learning_rate": 8.991355094068288e-06,
      "loss": 0.0457,
      "step": 48950
    },
    {
      "epoch": 5.727825883454248,
      "loss_main": 0.07416147738695145,
      "step": 48950
    },
    {
      "epoch": 5.72899602153054,
      "grad_norm": 0.25207170844078064,
      "learning_rate": 8.975595055133062e-06,
      "loss": 0.0405,
      "step": 48960
    },
    {
      "epoch": 5.72899602153054,
      "loss_main": 0.04779031500220299,
      "step": 48960
    },
    {
      "epoch": 5.730166159606833,
      "grad_norm": 0.1778775006532669,
      "learning_rate": 8.959847478328848e-06,
      "loss": 0.0507,
      "step": 48970
    },
    {
      "epoch": 5.730166159606833,
      "loss_main": 0.02254371903836727,
      "step": 48970
    },
    {
      "epoch": 5.731336297683127,
      "grad_norm": 0.21659117937088013,
      "learning_rate": 8.944112368439378e-06,
      "loss": 0.0415,
      "step": 48980
    },
    {
      "epoch": 5.731336297683127,
      "loss_main": 0.05093182250857353,
      "step": 48980
    },
    {
      "epoch": 5.7325064357594195,
      "grad_norm": 0.22379375994205475,
      "learning_rate": 8.928389730244552e-06,
      "loss": 0.0445,
      "step": 48990
    },
    {
      "epoch": 5.7325064357594195,
      "loss_main": 0.030516190454363823,
      "step": 48990
    },
    {
      "epoch": 5.733676573835712,
      "grad_norm": 0.18326470255851746,
      "learning_rate": 8.912679568520494e-06,
      "loss": 0.0415,
      "step": 49000
    },
    {
      "epoch": 5.733676573835712,
      "loss_main": 0.04637020081281662,
      "step": 49000
    },
    {
      "epoch": 5.734846711912006,
      "grad_norm": 0.250236451625824,
      "learning_rate": 8.896981888039534e-06,
      "loss": 0.0468,
      "step": 49010
    },
    {
      "epoch": 5.734846711912006,
      "loss_main": 0.05071771889925003,
      "step": 49010
    },
    {
      "epoch": 5.736016849988299,
      "grad_norm": 0.20988158881664276,
      "learning_rate": 8.881296693570201e-06,
      "loss": 0.0442,
      "step": 49020
    },
    {
      "epoch": 5.736016849988299,
      "loss_main": 0.06962250173091888,
      "step": 49020
    },
    {
      "epoch": 5.737186988064591,
      "grad_norm": 0.2255777269601822,
      "learning_rate": 8.865623989877281e-06,
      "loss": 0.0505,
      "step": 49030
    },
    {
      "epoch": 5.737186988064591,
      "loss_main": 0.06471256911754608,
      "step": 49030
    },
    {
      "epoch": 5.738357126140885,
      "grad_norm": 0.1952390968799591,
      "learning_rate": 8.849963781721681e-06,
      "loss": 0.0468,
      "step": 49040
    },
    {
      "epoch": 5.738357126140885,
      "loss_main": 0.05153859406709671,
      "step": 49040
    },
    {
      "epoch": 5.739527264217178,
      "grad_norm": 0.23709316551685333,
      "learning_rate": 8.834316073860588e-06,
      "loss": 0.0504,
      "step": 49050
    },
    {
      "epoch": 5.739527264217178,
      "loss_main": 0.039404671639204025,
      "step": 49050
    },
    {
      "epoch": 5.7406974022934705,
      "grad_norm": 0.2577594518661499,
      "learning_rate": 8.818680871047357e-06,
      "loss": 0.0485,
      "step": 49060
    },
    {
      "epoch": 5.7406974022934705,
      "loss_main": 0.0495324470102787,
      "step": 49060
    },
    {
      "epoch": 5.741867540369764,
      "grad_norm": 0.2676173448562622,
      "learning_rate": 8.803058178031549e-06,
      "loss": 0.0519,
      "step": 49070
    },
    {
      "epoch": 5.741867540369764,
      "loss_main": 0.05405883118510246,
      "step": 49070
    },
    {
      "epoch": 5.743037678446057,
      "grad_norm": 0.2092466652393341,
      "learning_rate": 8.787447999558922e-06,
      "loss": 0.0493,
      "step": 49080
    },
    {
      "epoch": 5.743037678446057,
      "loss_main": 0.04269224777817726,
      "step": 49080
    },
    {
      "epoch": 5.74420781652235,
      "grad_norm": 0.24523361027240753,
      "learning_rate": 8.77185034037144e-06,
      "loss": 0.046,
      "step": 49090
    },
    {
      "epoch": 5.74420781652235,
      "loss_main": 0.032261837273836136,
      "step": 49090
    },
    {
      "epoch": 5.745377954598642,
      "grad_norm": 0.24091073870658875,
      "learning_rate": 8.756265205207259e-06,
      "loss": 0.0449,
      "step": 49100
    },
    {
      "epoch": 5.745377954598642,
      "loss_main": 0.06803876906633377,
      "step": 49100
    },
    {
      "epoch": 5.746548092674936,
      "grad_norm": 0.23931069672107697,
      "learning_rate": 8.740692598800732e-06,
      "loss": 0.0443,
      "step": 49110
    },
    {
      "epoch": 5.746548092674936,
      "loss_main": 0.03780899569392204,
      "step": 49110
    },
    {
      "epoch": 5.747718230751229,
      "grad_norm": 0.3507351577281952,
      "learning_rate": 8.72513252588239e-06,
      "loss": 0.0502,
      "step": 49120
    },
    {
      "epoch": 5.747718230751229,
      "loss_main": 0.034130074083805084,
      "step": 49120
    },
    {
      "epoch": 5.748888368827521,
      "grad_norm": 0.23793332278728485,
      "learning_rate": 8.709584991178998e-06,
      "loss": 0.0446,
      "step": 49130
    },
    {
      "epoch": 5.748888368827521,
      "loss_main": 0.030404776334762573,
      "step": 49130
    },
    {
      "epoch": 5.750058506903815,
      "grad_norm": 0.24100632965564728,
      "learning_rate": 8.694049999413479e-06,
      "loss": 0.0454,
      "step": 49140
    },
    {
      "epoch": 5.750058506903815,
      "loss_main": 0.024486029520630836,
      "step": 49140
    },
    {
      "epoch": 5.751228644980108,
      "grad_norm": 0.29146164655685425,
      "learning_rate": 8.678527555304945e-06,
      "loss": 0.0443,
      "step": 49150
    },
    {
      "epoch": 5.751228644980108,
      "loss_main": 0.05322406813502312,
      "step": 49150
    },
    {
      "epoch": 5.7523987830564005,
      "grad_norm": 0.21619093418121338,
      "learning_rate": 8.663017663568712e-06,
      "loss": 0.0404,
      "step": 49160
    },
    {
      "epoch": 5.7523987830564005,
      "loss_main": 0.02614550292491913,
      "step": 49160
    },
    {
      "epoch": 5.753568921132693,
      "grad_norm": 0.276271253824234,
      "learning_rate": 8.647520328916259e-06,
      "loss": 0.0537,
      "step": 49170
    },
    {
      "epoch": 5.753568921132693,
      "loss_main": 0.024475861340761185,
      "step": 49170
    },
    {
      "epoch": 5.754739059208987,
      "grad_norm": 0.22338682413101196,
      "learning_rate": 8.632035556055307e-06,
      "loss": 0.0481,
      "step": 49180
    },
    {
      "epoch": 5.754739059208987,
      "loss_main": 0.048076074570417404,
      "step": 49180
    },
    {
      "epoch": 5.75590919728528,
      "grad_norm": 0.2774009108543396,
      "learning_rate": 8.616563349689672e-06,
      "loss": 0.0515,
      "step": 49190
    },
    {
      "epoch": 5.75590919728528,
      "loss_main": 0.037571217864751816,
      "step": 49190
    },
    {
      "epoch": 5.757079335361572,
      "grad_norm": 0.14134787023067474,
      "learning_rate": 8.601103714519448e-06,
      "loss": 0.0473,
      "step": 49200
    },
    {
      "epoch": 5.757079335361572,
      "loss_main": 0.05786919593811035,
      "step": 49200
    },
    {
      "epoch": 5.758249473437866,
      "grad_norm": 0.21817578375339508,
      "learning_rate": 8.58565665524082e-06,
      "loss": 0.0453,
      "step": 49210
    },
    {
      "epoch": 5.758249473437866,
      "loss_main": 0.043552465736866,
      "step": 49210
    },
    {
      "epoch": 5.759419611514159,
      "grad_norm": 0.17350423336029053,
      "learning_rate": 8.570222176546222e-06,
      "loss": 0.0455,
      "step": 49220
    },
    {
      "epoch": 5.759419611514159,
      "loss_main": 0.044958505779504776,
      "step": 49220
    },
    {
      "epoch": 5.7605897495904514,
      "grad_norm": 0.24083365499973297,
      "learning_rate": 8.554800283124242e-06,
      "loss": 0.0472,
      "step": 49230
    },
    {
      "epoch": 5.7605897495904514,
      "loss_main": 0.058277081698179245,
      "step": 49230
    },
    {
      "epoch": 5.761759887666745,
      "grad_norm": 0.2966528534889221,
      "learning_rate": 8.539390979659639e-06,
      "loss": 0.053,
      "step": 49240
    },
    {
      "epoch": 5.761759887666745,
      "loss_main": 0.033960212022066116,
      "step": 49240
    },
    {
      "epoch": 5.762930025743038,
      "grad_norm": 0.1681622862815857,
      "learning_rate": 8.523994270833352e-06,
      "loss": 0.0419,
      "step": 49250
    },
    {
      "epoch": 5.762930025743038,
      "loss_main": 0.07076046615839005,
      "step": 49250
    },
    {
      "epoch": 5.7641001638193305,
      "grad_norm": 0.19996705651283264,
      "learning_rate": 8.5086101613225e-06,
      "loss": 0.0487,
      "step": 49260
    },
    {
      "epoch": 5.7641001638193305,
      "loss_main": 0.043630123138427734,
      "step": 49260
    },
    {
      "epoch": 5.765270301895624,
      "grad_norm": 0.18390852212905884,
      "learning_rate": 8.493238655800346e-06,
      "loss": 0.0398,
      "step": 49270
    },
    {
      "epoch": 5.765270301895624,
      "loss_main": 0.05533658713102341,
      "step": 49270
    },
    {
      "epoch": 5.766440439971917,
      "grad_norm": 0.2063005268573761,
      "learning_rate": 8.47787975893638e-06,
      "loss": 0.043,
      "step": 49280
    },
    {
      "epoch": 5.766440439971917,
      "loss_main": 0.04076966270804405,
      "step": 49280
    },
    {
      "epoch": 5.76761057804821,
      "grad_norm": 0.22709567844867706,
      "learning_rate": 8.462533475396211e-06,
      "loss": 0.0446,
      "step": 49290
    },
    {
      "epoch": 5.76761057804821,
      "loss_main": 0.03508353978395462,
      "step": 49290
    },
    {
      "epoch": 5.768780716124502,
      "grad_norm": 0.2143377959728241,
      "learning_rate": 8.447199809841643e-06,
      "loss": 0.0448,
      "step": 49300
    },
    {
      "epoch": 5.768780716124502,
      "loss_main": 0.03335658088326454,
      "step": 49300
    },
    {
      "epoch": 5.769950854200796,
      "grad_norm": 0.2238619029521942,
      "learning_rate": 8.431878766930635e-06,
      "loss": 0.0417,
      "step": 49310
    },
    {
      "epoch": 5.769950854200796,
      "loss_main": 0.02361191250383854,
      "step": 49310
    },
    {
      "epoch": 5.771120992277089,
      "grad_norm": 0.28967761993408203,
      "learning_rate": 8.416570351317304e-06,
      "loss": 0.0495,
      "step": 49320
    },
    {
      "epoch": 5.771120992277089,
      "loss_main": 0.055303167551755905,
      "step": 49320
    },
    {
      "epoch": 5.7722911303533815,
      "grad_norm": 0.2588192820549011,
      "learning_rate": 8.401274567651973e-06,
      "loss": 0.0454,
      "step": 49330
    },
    {
      "epoch": 5.7722911303533815,
      "loss_main": 0.047214098274707794,
      "step": 49330
    },
    {
      "epoch": 5.773461268429674,
      "grad_norm": 0.2574029266834259,
      "learning_rate": 8.385991420581058e-06,
      "loss": 0.0472,
      "step": 49340
    },
    {
      "epoch": 5.773461268429674,
      "loss_main": 0.057616155594587326,
      "step": 49340
    },
    {
      "epoch": 5.774631406505968,
      "grad_norm": 0.3234978914260864,
      "learning_rate": 8.370720914747215e-06,
      "loss": 0.0443,
      "step": 49350
    },
    {
      "epoch": 5.774631406505968,
      "loss_main": 0.03796996548771858,
      "step": 49350
    },
    {
      "epoch": 5.775801544582261,
      "grad_norm": 0.19845201075077057,
      "learning_rate": 8.355463054789181e-06,
      "loss": 0.0423,
      "step": 49360
    },
    {
      "epoch": 5.775801544582261,
      "loss_main": 0.08701096475124359,
      "step": 49360
    },
    {
      "epoch": 5.776971682658553,
      "grad_norm": 0.2263064980506897,
      "learning_rate": 8.340217845341919e-06,
      "loss": 0.0514,
      "step": 49370
    },
    {
      "epoch": 5.776971682658553,
      "loss_main": 0.05255891755223274,
      "step": 49370
    },
    {
      "epoch": 5.778141820734847,
      "grad_norm": 0.22738660871982574,
      "learning_rate": 8.324985291036514e-06,
      "loss": 0.0507,
      "step": 49380
    },
    {
      "epoch": 5.778141820734847,
      "loss_main": 0.09459482133388519,
      "step": 49380
    },
    {
      "epoch": 5.77931195881114,
      "grad_norm": 0.2518523931503296,
      "learning_rate": 8.309765396500213e-06,
      "loss": 0.0515,
      "step": 49390
    },
    {
      "epoch": 5.77931195881114,
      "loss_main": 0.04737743362784386,
      "step": 49390
    },
    {
      "epoch": 5.780482096887432,
      "grad_norm": 0.32981234788894653,
      "learning_rate": 8.294558166356419e-06,
      "loss": 0.0453,
      "step": 49400
    },
    {
      "epoch": 5.780482096887432,
      "loss_main": 0.05304393917322159,
      "step": 49400
    },
    {
      "epoch": 5.781652234963726,
      "grad_norm": 0.20458084344863892,
      "learning_rate": 8.279363605224683e-06,
      "loss": 0.0391,
      "step": 49410
    },
    {
      "epoch": 5.781652234963726,
      "loss_main": 0.05450041964650154,
      "step": 49410
    },
    {
      "epoch": 5.782822373040019,
      "grad_norm": 0.27907732129096985,
      "learning_rate": 8.264181717720704e-06,
      "loss": 0.04,
      "step": 49420
    },
    {
      "epoch": 5.782822373040019,
      "loss_main": 0.027479568496346474,
      "step": 49420
    },
    {
      "epoch": 5.7839925111163115,
      "grad_norm": 0.39382272958755493,
      "learning_rate": 8.249012508456361e-06,
      "loss": 0.0544,
      "step": 49430
    },
    {
      "epoch": 5.7839925111163115,
      "loss_main": 0.053169913589954376,
      "step": 49430
    },
    {
      "epoch": 5.785162649192605,
      "grad_norm": 0.27520912885665894,
      "learning_rate": 8.233855982039646e-06,
      "loss": 0.0475,
      "step": 49440
    },
    {
      "epoch": 5.785162649192605,
      "loss_main": 0.04301586374640465,
      "step": 49440
    },
    {
      "epoch": 5.786332787268898,
      "grad_norm": 0.2304208129644394,
      "learning_rate": 8.218712143074708e-06,
      "loss": 0.0478,
      "step": 49450
    },
    {
      "epoch": 5.786332787268898,
      "loss_main": 0.04542352631688118,
      "step": 49450
    },
    {
      "epoch": 5.787502925345191,
      "grad_norm": 0.20280824601650238,
      "learning_rate": 8.203580996161858e-06,
      "loss": 0.0382,
      "step": 49460
    },
    {
      "epoch": 5.787502925345191,
      "loss_main": 0.05108952149748802,
      "step": 49460
    },
    {
      "epoch": 5.788673063421483,
      "grad_norm": 0.2919418513774872,
      "learning_rate": 8.188462545897512e-06,
      "loss": 0.0418,
      "step": 49470
    },
    {
      "epoch": 5.788673063421483,
      "loss_main": 0.051390390843153,
      "step": 49470
    },
    {
      "epoch": 5.789843201497777,
      "grad_norm": 0.20127519965171814,
      "learning_rate": 8.173356796874304e-06,
      "loss": 0.0394,
      "step": 49480
    },
    {
      "epoch": 5.789843201497777,
      "loss_main": 0.06009405851364136,
      "step": 49480
    },
    {
      "epoch": 5.79101333957407,
      "grad_norm": 0.22873061895370483,
      "learning_rate": 8.158263753680906e-06,
      "loss": 0.0455,
      "step": 49490
    },
    {
      "epoch": 5.79101333957407,
      "loss_main": 0.03267819806933403,
      "step": 49490
    },
    {
      "epoch": 5.7921834776503625,
      "grad_norm": 0.2086755335330963,
      "learning_rate": 8.143183420902239e-06,
      "loss": 0.0464,
      "step": 49500
    },
    {
      "epoch": 5.7921834776503625,
      "loss_main": 0.06623577326536179,
      "step": 49500
    },
    {
      "epoch": 5.793353615726656,
      "grad_norm": 0.20090273022651672,
      "learning_rate": 8.128115803119258e-06,
      "loss": 0.0398,
      "step": 49510
    },
    {
      "epoch": 5.793353615726656,
      "loss_main": 0.031042734161019325,
      "step": 49510
    },
    {
      "epoch": 5.794523753802949,
      "grad_norm": 0.16892623901367188,
      "learning_rate": 8.11306090490916e-06,
      "loss": 0.0387,
      "step": 49520
    },
    {
      "epoch": 5.794523753802949,
      "loss_main": 0.03117208741605282,
      "step": 49520
    },
    {
      "epoch": 5.7956938918792416,
      "grad_norm": 0.25389420986175537,
      "learning_rate": 8.098018730845169e-06,
      "loss": 0.0424,
      "step": 49530
    },
    {
      "epoch": 5.7956938918792416,
      "loss_main": 0.03488508239388466,
      "step": 49530
    },
    {
      "epoch": 5.796864029955534,
      "grad_norm": 0.19915813207626343,
      "learning_rate": 8.082989285496745e-06,
      "loss": 0.0413,
      "step": 49540
    },
    {
      "epoch": 5.796864029955534,
      "loss_main": 0.04110303893685341,
      "step": 49540
    },
    {
      "epoch": 5.798034168031828,
      "grad_norm": 0.20787379145622253,
      "learning_rate": 8.067972573429416e-06,
      "loss": 0.0432,
      "step": 49550
    },
    {
      "epoch": 5.798034168031828,
      "loss_main": 0.03529082238674164,
      "step": 49550
    },
    {
      "epoch": 5.799204306108121,
      "grad_norm": 0.23506663739681244,
      "learning_rate": 8.052968599204874e-06,
      "loss": 0.0458,
      "step": 49560
    },
    {
      "epoch": 5.799204306108121,
      "loss_main": 0.044803064316511154,
      "step": 49560
    },
    {
      "epoch": 5.800374444184413,
      "grad_norm": 0.2149507999420166,
      "learning_rate": 8.037977367380922e-06,
      "loss": 0.0456,
      "step": 49570
    },
    {
      "epoch": 5.800374444184413,
      "loss_main": 0.0362451933324337,
      "step": 49570
    },
    {
      "epoch": 5.801544582260707,
      "grad_norm": 0.19695329666137695,
      "learning_rate": 8.022998882511495e-06,
      "loss": 0.0455,
      "step": 49580
    },
    {
      "epoch": 5.801544582260707,
      "loss_main": 0.04440085217356682,
      "step": 49580
    },
    {
      "epoch": 5.802714720337,
      "grad_norm": 0.2603764235973358,
      "learning_rate": 8.008033149146677e-06,
      "loss": 0.0489,
      "step": 49590
    },
    {
      "epoch": 5.802714720337,
      "loss_main": 0.029620233923196793,
      "step": 49590
    },
    {
      "epoch": 5.8038848584132925,
      "grad_norm": 0.21883179247379303,
      "learning_rate": 7.993080171832656e-06,
      "loss": 0.0392,
      "step": 49600
    },
    {
      "epoch": 5.8038848584132925,
      "loss_main": 0.023429270833730698,
      "step": 49600
    },
    {
      "epoch": 5.805054996489586,
      "grad_norm": 0.29017379879951477,
      "learning_rate": 7.978139955111752e-06,
      "loss": 0.0483,
      "step": 49610
    },
    {
      "epoch": 5.805054996489586,
      "loss_main": 0.04199483245611191,
      "step": 49610
    },
    {
      "epoch": 5.806225134565879,
      "grad_norm": 0.22562435269355774,
      "learning_rate": 7.9632125035224e-06,
      "loss": 0.0398,
      "step": 49620
    },
    {
      "epoch": 5.806225134565879,
      "loss_main": 0.04483911022543907,
      "step": 49620
    },
    {
      "epoch": 5.807395272642172,
      "grad_norm": 0.21252498030662537,
      "learning_rate": 7.948297821599177e-06,
      "loss": 0.0413,
      "step": 49630
    },
    {
      "epoch": 5.807395272642172,
      "loss_main": 0.05631060153245926,
      "step": 49630
    },
    {
      "epoch": 5.808565410718465,
      "grad_norm": 0.21143794059753418,
      "learning_rate": 7.933395913872755e-06,
      "loss": 0.0406,
      "step": 49640
    },
    {
      "epoch": 5.808565410718465,
      "loss_main": 0.0621979646384716,
      "step": 49640
    },
    {
      "epoch": 5.809735548794758,
      "grad_norm": 0.19657787680625916,
      "learning_rate": 7.918506784869972e-06,
      "loss": 0.0433,
      "step": 49650
    },
    {
      "epoch": 5.809735548794758,
      "loss_main": 0.06168437376618385,
      "step": 49650
    },
    {
      "epoch": 5.810905686871051,
      "grad_norm": 0.23431934416294098,
      "learning_rate": 7.903630439113707e-06,
      "loss": 0.0461,
      "step": 49660
    },
    {
      "epoch": 5.810905686871051,
      "loss_main": 0.02624967321753502,
      "step": 49660
    },
    {
      "epoch": 5.812075824947343,
      "grad_norm": 0.26675617694854736,
      "learning_rate": 7.888766881123044e-06,
      "loss": 0.0419,
      "step": 49670
    },
    {
      "epoch": 5.812075824947343,
      "loss_main": 0.04516660422086716,
      "step": 49670
    },
    {
      "epoch": 5.813245963023637,
      "grad_norm": 0.2623462378978729,
      "learning_rate": 7.873916115413099e-06,
      "loss": 0.0511,
      "step": 49680
    },
    {
      "epoch": 5.813245963023637,
      "loss_main": 0.048273712396621704,
      "step": 49680
    },
    {
      "epoch": 5.81441610109993,
      "grad_norm": 0.1876714825630188,
      "learning_rate": 7.85907814649518e-06,
      "loss": 0.0518,
      "step": 49690
    },
    {
      "epoch": 5.81441610109993,
      "loss_main": 0.03710440173745155,
      "step": 49690
    },
    {
      "epoch": 5.8155862391762225,
      "grad_norm": 0.27266815304756165,
      "learning_rate": 7.844252978876649e-06,
      "loss": 0.0497,
      "step": 49700
    },
    {
      "epoch": 5.8155862391762225,
      "loss_main": 0.03962196782231331,
      "step": 49700
    },
    {
      "epoch": 5.816756377252516,
      "grad_norm": 0.21361461281776428,
      "learning_rate": 7.829440617061001e-06,
      "loss": 0.041,
      "step": 49710
    },
    {
      "epoch": 5.816756377252516,
      "loss_main": 0.07265530526638031,
      "step": 49710
    },
    {
      "epoch": 5.817926515328809,
      "grad_norm": 0.16949555277824402,
      "learning_rate": 7.814641065547851e-06,
      "loss": 0.0497,
      "step": 49720
    },
    {
      "epoch": 5.817926515328809,
      "loss_main": 0.03571021929383278,
      "step": 49720
    },
    {
      "epoch": 5.819096653405102,
      "grad_norm": 0.20339758694171906,
      "learning_rate": 7.79985432883289e-06,
      "loss": 0.0408,
      "step": 49730
    },
    {
      "epoch": 5.819096653405102,
      "loss_main": 0.028653332963585854,
      "step": 49730
    },
    {
      "epoch": 5.820266791481394,
      "grad_norm": 0.21785154938697815,
      "learning_rate": 7.78508041140797e-06,
      "loss": 0.0407,
      "step": 49740
    },
    {
      "epoch": 5.820266791481394,
      "loss_main": 0.03028308041393757,
      "step": 49740
    },
    {
      "epoch": 5.821436929557688,
      "grad_norm": 0.2561100125312805,
      "learning_rate": 7.770319317760993e-06,
      "loss": 0.048,
      "step": 49750
    },
    {
      "epoch": 5.821436929557688,
      "loss_main": 0.06841503828763962,
      "step": 49750
    },
    {
      "epoch": 5.822607067633981,
      "grad_norm": 0.21777307987213135,
      "learning_rate": 7.755571052376004e-06,
      "loss": 0.0455,
      "step": 49760
    },
    {
      "epoch": 5.822607067633981,
      "loss_main": 0.05668174847960472,
      "step": 49760
    },
    {
      "epoch": 5.8237772057102735,
      "grad_norm": 0.31490880250930786,
      "learning_rate": 7.740835619733128e-06,
      "loss": 0.0427,
      "step": 49770
    },
    {
      "epoch": 5.8237772057102735,
      "loss_main": 0.043111324310302734,
      "step": 49770
    },
    {
      "epoch": 5.824947343786567,
      "grad_norm": 0.20731350779533386,
      "learning_rate": 7.726113024308601e-06,
      "loss": 0.047,
      "step": 49780
    },
    {
      "epoch": 5.824947343786567,
      "loss_main": 0.05745064094662666,
      "step": 49780
    },
    {
      "epoch": 5.82611748186286,
      "grad_norm": 0.20200859010219574,
      "learning_rate": 7.711403270574746e-06,
      "loss": 0.046,
      "step": 49790
    },
    {
      "epoch": 5.82611748186286,
      "loss_main": 0.05314319580793381,
      "step": 49790
    },
    {
      "epoch": 5.827287619939153,
      "grad_norm": 0.20881833136081696,
      "learning_rate": 7.696706363000039e-06,
      "loss": 0.0434,
      "step": 49800
    },
    {
      "epoch": 5.827287619939153,
      "loss_main": 0.01746060512959957,
      "step": 49800
    },
    {
      "epoch": 5.828457758015446,
      "grad_norm": 0.1898123025894165,
      "learning_rate": 7.682022306048959e-06,
      "loss": 0.0423,
      "step": 49810
    },
    {
      "epoch": 5.828457758015446,
      "loss_main": 0.03860051557421684,
      "step": 49810
    },
    {
      "epoch": 5.829627896091739,
      "grad_norm": 0.16581284999847412,
      "learning_rate": 7.667351104182186e-06,
      "loss": 0.0449,
      "step": 49820
    },
    {
      "epoch": 5.829627896091739,
      "loss_main": 0.08890929073095322,
      "step": 49820
    },
    {
      "epoch": 5.830798034168032,
      "grad_norm": 0.2143474668264389,
      "learning_rate": 7.652692761856395e-06,
      "loss": 0.0432,
      "step": 49830
    },
    {
      "epoch": 5.830798034168032,
      "loss_main": 0.03169689327478409,
      "step": 49830
    },
    {
      "epoch": 5.831968172244325,
      "grad_norm": 0.20651495456695557,
      "learning_rate": 7.63804728352444e-06,
      "loss": 0.0393,
      "step": 49840
    },
    {
      "epoch": 5.831968172244325,
      "loss_main": 0.033337995409965515,
      "step": 49840
    },
    {
      "epoch": 5.833138310320618,
      "grad_norm": 0.24849821627140045,
      "learning_rate": 7.623414673635215e-06,
      "loss": 0.0437,
      "step": 49850
    },
    {
      "epoch": 5.833138310320618,
      "loss_main": 0.03258247300982475,
      "step": 49850
    },
    {
      "epoch": 5.834308448396911,
      "grad_norm": 0.3832310140132904,
      "learning_rate": 7.608794936633723e-06,
      "loss": 0.0506,
      "step": 49860
    },
    {
      "epoch": 5.834308448396911,
      "loss_main": 0.12431152164936066,
      "step": 49860
    },
    {
      "epoch": 5.8354785864732035,
      "grad_norm": 0.20732712745666504,
      "learning_rate": 7.594188076961056e-06,
      "loss": 0.0502,
      "step": 49870
    },
    {
      "epoch": 5.8354785864732035,
      "loss_main": 0.03967249020934105,
      "step": 49870
    },
    {
      "epoch": 5.836648724549497,
      "grad_norm": 0.2385648488998413,
      "learning_rate": 7.579594099054382e-06,
      "loss": 0.0539,
      "step": 49880
    },
    {
      "epoch": 5.836648724549497,
      "loss_main": 0.05481650307774544,
      "step": 49880
    },
    {
      "epoch": 5.83781886262579,
      "grad_norm": 0.19573476910591125,
      "learning_rate": 7.565013007346983e-06,
      "loss": 0.0417,
      "step": 49890
    },
    {
      "epoch": 5.83781886262579,
      "loss_main": 0.0546831414103508,
      "step": 49890
    },
    {
      "epoch": 5.838989000702083,
      "grad_norm": 0.3242037296295166,
      "learning_rate": 7.5504448062682035e-06,
      "loss": 0.0484,
      "step": 49900
    },
    {
      "epoch": 5.838989000702083,
      "loss_main": 0.03555934503674507,
      "step": 49900
    },
    {
      "epoch": 5.840159138778376,
      "grad_norm": 0.2396342158317566,
      "learning_rate": 7.53588950024347e-06,
      "loss": 0.0448,
      "step": 49910
    },
    {
      "epoch": 5.840159138778376,
      "loss_main": 0.03507395461201668,
      "step": 49910
    },
    {
      "epoch": 5.841329276854669,
      "grad_norm": 0.2311677634716034,
      "learning_rate": 7.5213470936943145e-06,
      "loss": 0.038,
      "step": 49920
    },
    {
      "epoch": 5.841329276854669,
      "loss_main": 0.04162832349538803,
      "step": 49920
    },
    {
      "epoch": 5.842499414930962,
      "grad_norm": 0.20261886715888977,
      "learning_rate": 7.506817591038323e-06,
      "loss": 0.0459,
      "step": 49930
    },
    {
      "epoch": 5.842499414930962,
      "loss_main": 0.032561421394348145,
      "step": 49930
    },
    {
      "epoch": 5.843669553007254,
      "grad_norm": 0.20591652393341064,
      "learning_rate": 7.492300996689183e-06,
      "loss": 0.056,
      "step": 49940
    },
    {
      "epoch": 5.843669553007254,
      "loss_main": 0.022713927552103996,
      "step": 49940
    },
    {
      "epoch": 5.844839691083548,
      "grad_norm": 0.2724497616291046,
      "learning_rate": 7.477797315056645e-06,
      "loss": 0.0464,
      "step": 49950
    },
    {
      "epoch": 5.844839691083548,
      "loss_main": 0.05706243962049484,
      "step": 49950
    },
    {
      "epoch": 5.846009829159841,
      "grad_norm": 0.21502365171909332,
      "learning_rate": 7.463306550546539e-06,
      "loss": 0.043,
      "step": 49960
    },
    {
      "epoch": 5.846009829159841,
      "loss_main": 0.061929743736982346,
      "step": 49960
    },
    {
      "epoch": 5.8471799672361335,
      "grad_norm": 0.32378244400024414,
      "learning_rate": 7.448828707560812e-06,
      "loss": 0.0439,
      "step": 49970
    },
    {
      "epoch": 5.8471799672361335,
      "loss_main": 0.0621735043823719,
      "step": 49970
    },
    {
      "epoch": 5.848350105312427,
      "grad_norm": 0.24615812301635742,
      "learning_rate": 7.4343637904974e-06,
      "loss": 0.045,
      "step": 49980
    },
    {
      "epoch": 5.848350105312427,
      "loss_main": 0.04079508036375046,
      "step": 49980
    },
    {
      "epoch": 5.84952024338872,
      "grad_norm": 0.22226370871067047,
      "learning_rate": 7.419911803750401e-06,
      "loss": 0.0432,
      "step": 49990
    },
    {
      "epoch": 5.84952024338872,
      "loss_main": 0.04626365751028061,
      "step": 49990
    },
    {
      "epoch": 5.850690381465013,
      "grad_norm": 0.22253534197807312,
      "learning_rate": 7.405472751709935e-06,
      "loss": 0.0413,
      "step": 50000
    },
    {
      "epoch": 5.850690381465013,
      "loss_main": 0.09735783189535141,
      "step": 50000
    },
    {
      "epoch": 5.851860519541306,
      "grad_norm": 0.260042279958725,
      "learning_rate": 7.3910466387622e-06,
      "loss": 0.0622,
      "step": 50010
    },
    {
      "epoch": 5.851860519541306,
      "loss_main": 0.04923998564481735,
      "step": 50010
    },
    {
      "epoch": 5.853030657617599,
      "grad_norm": 0.2209751307964325,
      "learning_rate": 7.3766334692894735e-06,
      "loss": 0.042,
      "step": 50020
    },
    {
      "epoch": 5.853030657617599,
      "loss_main": 0.058930654078722,
      "step": 50020
    },
    {
      "epoch": 5.854200795693892,
      "grad_norm": 0.1915445178747177,
      "learning_rate": 7.3622332476700865e-06,
      "loss": 0.0424,
      "step": 50030
    },
    {
      "epoch": 5.854200795693892,
      "loss_main": 0.07018553465604782,
      "step": 50030
    },
    {
      "epoch": 5.855370933770185,
      "grad_norm": 0.20698997378349304,
      "learning_rate": 7.347845978278472e-06,
      "loss": 0.0524,
      "step": 50040
    },
    {
      "epoch": 5.855370933770185,
      "loss_main": 0.02440403588116169,
      "step": 50040
    },
    {
      "epoch": 5.856541071846478,
      "grad_norm": 0.19526338577270508,
      "learning_rate": 7.333471665485065e-06,
      "loss": 0.0413,
      "step": 50050
    },
    {
      "epoch": 5.856541071846478,
      "loss_main": 0.021961063146591187,
      "step": 50050
    },
    {
      "epoch": 5.857711209922771,
      "grad_norm": 0.29118403792381287,
      "learning_rate": 7.31911031365643e-06,
      "loss": 0.0449,
      "step": 50060
    },
    {
      "epoch": 5.857711209922771,
      "loss_main": 0.019126195460557938,
      "step": 50060
    },
    {
      "epoch": 5.858881347999064,
      "grad_norm": 0.28067657351493835,
      "learning_rate": 7.304761927155157e-06,
      "loss": 0.0435,
      "step": 50070
    },
    {
      "epoch": 5.858881347999064,
      "loss_main": 0.052194736897945404,
      "step": 50070
    },
    {
      "epoch": 5.860051486075357,
      "grad_norm": 0.2722361087799072,
      "learning_rate": 7.29042651033991e-06,
      "loss": 0.0443,
      "step": 50080
    },
    {
      "epoch": 5.860051486075357,
      "loss_main": 0.027624640613794327,
      "step": 50080
    },
    {
      "epoch": 5.86122162415165,
      "grad_norm": 0.31429222226142883,
      "learning_rate": 7.276104067565409e-06,
      "loss": 0.0407,
      "step": 50090
    },
    {
      "epoch": 5.86122162415165,
      "loss_main": 0.04797464236617088,
      "step": 50090
    },
    {
      "epoch": 5.862391762227943,
      "grad_norm": 0.31626948714256287,
      "learning_rate": 7.261794603182431e-06,
      "loss": 0.045,
      "step": 50100
    },
    {
      "epoch": 5.862391762227943,
      "loss_main": 0.05147740617394447,
      "step": 50100
    },
    {
      "epoch": 5.863561900304235,
      "grad_norm": 0.2617393136024475,
      "learning_rate": 7.24749812153781e-06,
      "loss": 0.0384,
      "step": 50110
    },
    {
      "epoch": 5.863561900304235,
      "loss_main": 0.06430993229150772,
      "step": 50110
    },
    {
      "epoch": 5.864732038380529,
      "grad_norm": 0.20402805507183075,
      "learning_rate": 7.2332146269744605e-06,
      "loss": 0.0572,
      "step": 50120
    },
    {
      "epoch": 5.864732038380529,
      "loss_main": 0.0344831608235836,
      "step": 50120
    },
    {
      "epoch": 5.865902176456822,
      "grad_norm": 0.22395619750022888,
      "learning_rate": 7.218944123831295e-06,
      "loss": 0.0408,
      "step": 50130
    },
    {
      "epoch": 5.865902176456822,
      "loss_main": 0.08128644526004791,
      "step": 50130
    },
    {
      "epoch": 5.8670723145331145,
      "grad_norm": 0.16060175001621246,
      "learning_rate": 7.204686616443351e-06,
      "loss": 0.0448,
      "step": 50140
    },
    {
      "epoch": 5.8670723145331145,
      "loss_main": 0.02463952638208866,
      "step": 50140
    },
    {
      "epoch": 5.868242452609408,
      "grad_norm": 0.2626860439777374,
      "learning_rate": 7.190442109141665e-06,
      "loss": 0.0462,
      "step": 50150
    },
    {
      "epoch": 5.868242452609408,
      "loss_main": 0.041264746338129044,
      "step": 50150
    },
    {
      "epoch": 5.869412590685701,
      "grad_norm": 0.25479552149772644,
      "learning_rate": 7.176210606253347e-06,
      "loss": 0.04,
      "step": 50160
    },
    {
      "epoch": 5.869412590685701,
      "loss_main": 0.049855224788188934,
      "step": 50160
    },
    {
      "epoch": 5.870582728761994,
      "grad_norm": 0.21198301017284393,
      "learning_rate": 7.161992112101551e-06,
      "loss": 0.0433,
      "step": 50170
    },
    {
      "epoch": 5.870582728761994,
      "loss_main": 0.057524051517248154,
      "step": 50170
    },
    {
      "epoch": 5.871752866838287,
      "grad_norm": 0.26858675479888916,
      "learning_rate": 7.147786631005465e-06,
      "loss": 0.0469,
      "step": 50180
    },
    {
      "epoch": 5.871752866838287,
      "loss_main": 0.053702231496572495,
      "step": 50180
    },
    {
      "epoch": 5.87292300491458,
      "grad_norm": 0.1983577013015747,
      "learning_rate": 7.1335941672803775e-06,
      "loss": 0.0488,
      "step": 50190
    },
    {
      "epoch": 5.87292300491458,
      "loss_main": 0.0505535714328289,
      "step": 50190
    },
    {
      "epoch": 5.874093142990873,
      "grad_norm": 0.19978484511375427,
      "learning_rate": 7.1194147252375384e-06,
      "loss": 0.0489,
      "step": 50200
    },
    {
      "epoch": 5.874093142990873,
      "loss_main": 0.052105989307165146,
      "step": 50200
    },
    {
      "epoch": 5.875263281067166,
      "grad_norm": 0.19088038802146912,
      "learning_rate": 7.10524830918432e-06,
      "loss": 0.0496,
      "step": 50210
    },
    {
      "epoch": 5.875263281067166,
      "loss_main": 0.051166705787181854,
      "step": 50210
    },
    {
      "epoch": 5.876433419143459,
      "grad_norm": 0.2827606797218323,
      "learning_rate": 7.091094923424097e-06,
      "loss": 0.0525,
      "step": 50220
    },
    {
      "epoch": 5.876433419143459,
      "loss_main": 0.07405520975589752,
      "step": 50220
    },
    {
      "epoch": 5.877603557219752,
      "grad_norm": 0.20501892268657684,
      "learning_rate": 7.0769545722562894e-06,
      "loss": 0.0467,
      "step": 50230
    },
    {
      "epoch": 5.877603557219752,
      "loss_main": 0.053744006901979446,
      "step": 50230
    },
    {
      "epoch": 5.878773695296045,
      "grad_norm": 0.22187182307243347,
      "learning_rate": 7.0628272599763675e-06,
      "loss": 0.0464,
      "step": 50240
    },
    {
      "epoch": 5.878773695296045,
      "loss_main": 0.0252270195633173,
      "step": 50240
    },
    {
      "epoch": 5.879943833372338,
      "grad_norm": 0.26052072644233704,
      "learning_rate": 7.048712990875828e-06,
      "loss": 0.044,
      "step": 50250
    },
    {
      "epoch": 5.879943833372338,
      "loss_main": 0.03114703670144081,
      "step": 50250
    },
    {
      "epoch": 5.881113971448631,
      "grad_norm": 0.3448467254638672,
      "learning_rate": 7.034611769242216e-06,
      "loss": 0.047,
      "step": 50260
    },
    {
      "epoch": 5.881113971448631,
      "loss_main": 0.04478094354271889,
      "step": 50260
    },
    {
      "epoch": 5.882284109524924,
      "grad_norm": 0.1648116409778595,
      "learning_rate": 7.02052359935913e-06,
      "loss": 0.042,
      "step": 50270
    },
    {
      "epoch": 5.882284109524924,
      "loss_main": 0.04111107811331749,
      "step": 50270
    },
    {
      "epoch": 5.883454247601217,
      "grad_norm": 0.25574466586112976,
      "learning_rate": 7.006448485506145e-06,
      "loss": 0.0487,
      "step": 50280
    },
    {
      "epoch": 5.883454247601217,
      "loss_main": 0.029641253873705864,
      "step": 50280
    },
    {
      "epoch": 5.88462438567751,
      "grad_norm": 0.2810150980949402,
      "learning_rate": 6.992386431958942e-06,
      "loss": 0.0509,
      "step": 50290
    },
    {
      "epoch": 5.88462438567751,
      "loss_main": 0.03082549199461937,
      "step": 50290
    },
    {
      "epoch": 5.885794523753803,
      "grad_norm": 0.23068471252918243,
      "learning_rate": 6.978337442989197e-06,
      "loss": 0.0516,
      "step": 50300
    },
    {
      "epoch": 5.885794523753803,
      "loss_main": 0.026487061753869057,
      "step": 50300
    },
    {
      "epoch": 5.8869646618300955,
      "grad_norm": 0.2810482084751129,
      "learning_rate": 6.964301522864608e-06,
      "loss": 0.0442,
      "step": 50310
    },
    {
      "epoch": 5.8869646618300955,
      "loss_main": 0.03147042542695999,
      "step": 50310
    },
    {
      "epoch": 5.888134799906389,
      "grad_norm": 0.3068719804286957,
      "learning_rate": 6.950278675848926e-06,
      "loss": 0.0485,
      "step": 50320
    },
    {
      "epoch": 5.888134799906389,
      "loss_main": 0.04141826555132866,
      "step": 50320
    },
    {
      "epoch": 5.889304937982682,
      "grad_norm": 0.25428634881973267,
      "learning_rate": 6.9362689062019145e-06,
      "loss": 0.0511,
      "step": 50330
    },
    {
      "epoch": 5.889304937982682,
      "loss_main": 0.04618880897760391,
      "step": 50330
    },
    {
      "epoch": 5.890475076058975,
      "grad_norm": 0.27067863941192627,
      "learning_rate": 6.922272218179393e-06,
      "loss": 0.0439,
      "step": 50340
    },
    {
      "epoch": 5.890475076058975,
      "loss_main": 0.0329342782497406,
      "step": 50340
    },
    {
      "epoch": 5.891645214135268,
      "grad_norm": 0.27099373936653137,
      "learning_rate": 6.908288616033148e-06,
      "loss": 0.0462,
      "step": 50350
    },
    {
      "epoch": 5.891645214135268,
      "loss_main": 0.02976216934621334,
      "step": 50350
    },
    {
      "epoch": 5.892815352211561,
      "grad_norm": 0.22017213702201843,
      "learning_rate": 6.894318104011077e-06,
      "loss": 0.0436,
      "step": 50360
    },
    {
      "epoch": 5.892815352211561,
      "loss_main": 0.04449499770998955,
      "step": 50360
    },
    {
      "epoch": 5.893985490287854,
      "grad_norm": 0.3191836476325989,
      "learning_rate": 6.880360686357007e-06,
      "loss": 0.0422,
      "step": 50370
    },
    {
      "epoch": 5.893985490287854,
      "loss_main": 0.03909524902701378,
      "step": 50370
    },
    {
      "epoch": 5.895155628364147,
      "grad_norm": 0.2105194479227066,
      "learning_rate": 6.8664163673108575e-06,
      "loss": 0.0456,
      "step": 50380
    },
    {
      "epoch": 5.895155628364147,
      "loss_main": 0.052999358624219894,
      "step": 50380
    },
    {
      "epoch": 5.89632576644044,
      "grad_norm": 0.2737829387187958,
      "learning_rate": 6.85248515110854e-06,
      "loss": 0.0462,
      "step": 50390
    },
    {
      "epoch": 5.89632576644044,
      "loss_main": 0.04876496642827988,
      "step": 50390
    },
    {
      "epoch": 5.897495904516733,
      "grad_norm": 0.216927170753479,
      "learning_rate": 6.838567041981992e-06,
      "loss": 0.0472,
      "step": 50400
    },
    {
      "epoch": 5.897495904516733,
      "loss_main": 0.04263204336166382,
      "step": 50400
    },
    {
      "epoch": 5.898666042593026,
      "grad_norm": 0.15446189045906067,
      "learning_rate": 6.8246620441591634e-06,
      "loss": 0.0378,
      "step": 50410
    },
    {
      "epoch": 5.898666042593026,
      "loss_main": 0.02001492865383625,
      "step": 50410
    },
    {
      "epoch": 5.899836180669319,
      "grad_norm": 0.2469388246536255,
      "learning_rate": 6.8107701618640275e-06,
      "loss": 0.0442,
      "step": 50420
    },
    {
      "epoch": 5.899836180669319,
      "loss_main": 0.0279291570186615,
      "step": 50420
    },
    {
      "epoch": 5.901006318745612,
      "grad_norm": 0.17877553403377533,
      "learning_rate": 6.796891399316557e-06,
      "loss": 0.0366,
      "step": 50430
    },
    {
      "epoch": 5.901006318745612,
      "loss_main": 0.04429231211543083,
      "step": 50430
    },
    {
      "epoch": 5.902176456821905,
      "grad_norm": 0.2521427273750305,
      "learning_rate": 6.7830257607327804e-06,
      "loss": 0.0436,
      "step": 50440
    },
    {
      "epoch": 5.902176456821905,
      "loss_main": 0.04780105873942375,
      "step": 50440
    },
    {
      "epoch": 5.903346594898198,
      "grad_norm": 0.26856595277786255,
      "learning_rate": 6.7691732503247e-06,
      "loss": 0.0499,
      "step": 50450
    },
    {
      "epoch": 5.903346594898198,
      "loss_main": 0.0820455402135849,
      "step": 50450
    },
    {
      "epoch": 5.904516732974491,
      "grad_norm": 0.22969701886177063,
      "learning_rate": 6.755333872300346e-06,
      "loss": 0.0491,
      "step": 50460
    },
    {
      "epoch": 5.904516732974491,
      "loss_main": 0.05000612512230873,
      "step": 50460
    },
    {
      "epoch": 5.905686871050784,
      "grad_norm": 0.25448134541511536,
      "learning_rate": 6.741507630863747e-06,
      "loss": 0.0456,
      "step": 50470
    },
    {
      "epoch": 5.905686871050784,
      "loss_main": 0.06257714331150055,
      "step": 50470
    },
    {
      "epoch": 5.906857009127077,
      "grad_norm": 0.30408498644828796,
      "learning_rate": 6.727694530214945e-06,
      "loss": 0.0456,
      "step": 50480
    },
    {
      "epoch": 5.906857009127077,
      "loss_main": 0.05343451723456383,
      "step": 50480
    },
    {
      "epoch": 5.90802714720337,
      "grad_norm": 0.35778748989105225,
      "learning_rate": 6.713894574550028e-06,
      "loss": 0.0527,
      "step": 50490
    },
    {
      "epoch": 5.90802714720337,
      "loss_main": 0.05458442494273186,
      "step": 50490
    },
    {
      "epoch": 5.909197285279663,
      "grad_norm": 0.2541503310203552,
      "learning_rate": 6.700107768061015e-06,
      "loss": 0.0455,
      "step": 50500
    },
    {
      "epoch": 5.909197285279663,
      "loss_main": 0.02958512306213379,
      "step": 50500
    },
    {
      "epoch": 5.9103674233559556,
      "grad_norm": 0.20072215795516968,
      "learning_rate": 6.686334114936016e-06,
      "loss": 0.0426,
      "step": 50510
    },
    {
      "epoch": 5.9103674233559556,
      "loss_main": 0.058054547756910324,
      "step": 50510
    },
    {
      "epoch": 5.911537561432249,
      "grad_norm": 0.17937928438186646,
      "learning_rate": 6.672573619359063e-06,
      "loss": 0.0413,
      "step": 50520
    },
    {
      "epoch": 5.911537561432249,
      "loss_main": 0.06464170664548874,
      "step": 50520
    },
    {
      "epoch": 5.912707699508542,
      "grad_norm": 0.23506410419940948,
      "learning_rate": 6.658826285510256e-06,
      "loss": 0.0458,
      "step": 50530
    },
    {
      "epoch": 5.912707699508542,
      "loss_main": 0.03775716572999954,
      "step": 50530
    },
    {
      "epoch": 5.913877837584835,
      "grad_norm": 0.31060245633125305,
      "learning_rate": 6.645092117565666e-06,
      "loss": 0.0544,
      "step": 50540
    },
    {
      "epoch": 5.913877837584835,
      "loss_main": 0.09581445157527924,
      "step": 50540
    },
    {
      "epoch": 5.915047975661128,
      "grad_norm": 0.17217165231704712,
      "learning_rate": 6.631371119697371e-06,
      "loss": 0.0469,
      "step": 50550
    },
    {
      "epoch": 5.915047975661128,
      "loss_main": 0.04490768536925316,
      "step": 50550
    },
    {
      "epoch": 5.916218113737421,
      "grad_norm": 0.36901816725730896,
      "learning_rate": 6.6176632960734505e-06,
      "loss": 0.0466,
      "step": 50560
    },
    {
      "epoch": 5.916218113737421,
      "loss_main": 0.027747193351387978,
      "step": 50560
    },
    {
      "epoch": 5.917388251813714,
      "grad_norm": 0.30778995156288147,
      "learning_rate": 6.603968650857978e-06,
      "loss": 0.0427,
      "step": 50570
    },
    {
      "epoch": 5.917388251813714,
      "loss_main": 0.05514729022979736,
      "step": 50570
    },
    {
      "epoch": 5.918558389890007,
      "grad_norm": 0.24364812672138214,
      "learning_rate": 6.5902871882110085e-06,
      "loss": 0.0454,
      "step": 50580
    },
    {
      "epoch": 5.918558389890007,
      "loss_main": 0.06546846032142639,
      "step": 50580
    },
    {
      "epoch": 5.9197285279663,
      "grad_norm": 0.2716433107852936,
      "learning_rate": 6.576618912288635e-06,
      "loss": 0.0427,
      "step": 50590
    },
    {
      "epoch": 5.9197285279663,
      "loss_main": 0.046794064342975616,
      "step": 50590
    },
    {
      "epoch": 5.920898666042593,
      "grad_norm": 0.222271129488945,
      "learning_rate": 6.562963827242913e-06,
      "loss": 0.0477,
      "step": 50600
    },
    {
      "epoch": 5.920898666042593,
      "loss_main": 0.04770369082689285,
      "step": 50600
    },
    {
      "epoch": 5.9220688041188865,
      "grad_norm": 0.20944981276988983,
      "learning_rate": 6.549321937221886e-06,
      "loss": 0.054,
      "step": 50610
    },
    {
      "epoch": 5.9220688041188865,
      "loss_main": 0.02915380336344242,
      "step": 50610
    },
    {
      "epoch": 5.923238942195179,
      "grad_norm": 0.23240211606025696,
      "learning_rate": 6.5356932463696064e-06,
      "loss": 0.0417,
      "step": 50620
    },
    {
      "epoch": 5.923238942195179,
      "loss_main": 0.03376295045018196,
      "step": 50620
    },
    {
      "epoch": 5.924409080271472,
      "grad_norm": 0.19890451431274414,
      "learning_rate": 6.522077758826101e-06,
      "loss": 0.0478,
      "step": 50630
    },
    {
      "epoch": 5.924409080271472,
      "loss_main": 0.0414072647690773,
      "step": 50630
    },
    {
      "epoch": 5.925579218347765,
      "grad_norm": 0.20754872262477875,
      "learning_rate": 6.5084754787274275e-06,
      "loss": 0.0445,
      "step": 50640
    },
    {
      "epoch": 5.925579218347765,
      "loss_main": 0.041142407804727554,
      "step": 50640
    },
    {
      "epoch": 5.926749356424058,
      "grad_norm": 0.2912697196006775,
      "learning_rate": 6.494886410205553e-06,
      "loss": 0.0482,
      "step": 50650
    },
    {
      "epoch": 5.926749356424058,
      "loss_main": 0.03930390626192093,
      "step": 50650
    },
    {
      "epoch": 5.927919494500351,
      "grad_norm": 0.2543610632419586,
      "learning_rate": 6.481310557388521e-06,
      "loss": 0.0446,
      "step": 50660
    },
    {
      "epoch": 5.927919494500351,
      "loss_main": 0.035928767174482346,
      "step": 50660
    },
    {
      "epoch": 5.929089632576644,
      "grad_norm": 0.1912160962820053,
      "learning_rate": 6.46774792440028e-06,
      "loss": 0.0381,
      "step": 50670
    },
    {
      "epoch": 5.929089632576644,
      "loss_main": 0.08055690675973892,
      "step": 50670
    },
    {
      "epoch": 5.930259770652937,
      "grad_norm": 0.26099276542663574,
      "learning_rate": 6.4541985153608206e-06,
      "loss": 0.0494,
      "step": 50680
    },
    {
      "epoch": 5.930259770652937,
      "loss_main": 0.06689776480197906,
      "step": 50680
    },
    {
      "epoch": 5.93142990872923,
      "grad_norm": 0.24590937793254852,
      "learning_rate": 6.4406623343861e-06,
      "loss": 0.0493,
      "step": 50690
    },
    {
      "epoch": 5.93142990872923,
      "loss_main": 0.07417052984237671,
      "step": 50690
    },
    {
      "epoch": 5.932600046805523,
      "grad_norm": 0.1470382958650589,
      "learning_rate": 6.427139385588038e-06,
      "loss": 0.0468,
      "step": 50700
    },
    {
      "epoch": 5.932600046805523,
      "loss_main": 0.029666731134057045,
      "step": 50700
    },
    {
      "epoch": 5.933770184881816,
      "grad_norm": 0.21000921726226807,
      "learning_rate": 6.413629673074561e-06,
      "loss": 0.0423,
      "step": 50710
    },
    {
      "epoch": 5.933770184881816,
      "loss_main": 0.04234257712960243,
      "step": 50710
    },
    {
      "epoch": 5.934940322958109,
      "grad_norm": 0.24644024670124054,
      "learning_rate": 6.400133200949554e-06,
      "loss": 0.0427,
      "step": 50720
    },
    {
      "epoch": 5.934940322958109,
      "loss_main": 0.02778976783156395,
      "step": 50720
    },
    {
      "epoch": 5.936110461034402,
      "grad_norm": 0.2614390254020691,
      "learning_rate": 6.386649973312897e-06,
      "loss": 0.0474,
      "step": 50730
    },
    {
      "epoch": 5.936110461034402,
      "loss_main": 0.06989675760269165,
      "step": 50730
    },
    {
      "epoch": 5.937280599110695,
      "grad_norm": 0.28011175990104675,
      "learning_rate": 6.37317999426042e-06,
      "loss": 0.0563,
      "step": 50740
    },
    {
      "epoch": 5.937280599110695,
      "loss_main": 0.024351047351956367,
      "step": 50740
    },
    {
      "epoch": 5.938450737186988,
      "grad_norm": 0.23077112436294556,
      "learning_rate": 6.359723267883977e-06,
      "loss": 0.0425,
      "step": 50750
    },
    {
      "epoch": 5.938450737186988,
      "loss_main": 0.03699618577957153,
      "step": 50750
    },
    {
      "epoch": 5.939620875263281,
      "grad_norm": 0.2223271131515503,
      "learning_rate": 6.346279798271343e-06,
      "loss": 0.0516,
      "step": 50760
    },
    {
      "epoch": 5.939620875263281,
      "loss_main": 0.03135766461491585,
      "step": 50760
    },
    {
      "epoch": 5.940791013339574,
      "grad_norm": 0.22726620733737946,
      "learning_rate": 6.332849589506301e-06,
      "loss": 0.044,
      "step": 50770
    },
    {
      "epoch": 5.940791013339574,
      "loss_main": 0.027626438066363335,
      "step": 50770
    },
    {
      "epoch": 5.9419611514158674,
      "grad_norm": 0.36028391122817993,
      "learning_rate": 6.319432645668588e-06,
      "loss": 0.0392,
      "step": 50780
    },
    {
      "epoch": 5.9419611514158674,
      "loss_main": 0.06234708055853844,
      "step": 50780
    },
    {
      "epoch": 5.94313128949216,
      "grad_norm": 0.3179047107696533,
      "learning_rate": 6.306028970833922e-06,
      "loss": 0.0429,
      "step": 50790
    },
    {
      "epoch": 5.94313128949216,
      "loss_main": 0.051641933619976044,
      "step": 50790
    },
    {
      "epoch": 5.944301427568453,
      "grad_norm": 0.21384015679359436,
      "learning_rate": 6.2926385690739665e-06,
      "loss": 0.048,
      "step": 50800
    },
    {
      "epoch": 5.944301427568453,
      "loss_main": 0.035256341099739075,
      "step": 50800
    },
    {
      "epoch": 5.9454715656447465,
      "grad_norm": 0.24490053951740265,
      "learning_rate": 6.279261444456413e-06,
      "loss": 0.0423,
      "step": 50810
    },
    {
      "epoch": 5.9454715656447465,
      "loss_main": 0.04855813458561897,
      "step": 50810
    },
    {
      "epoch": 5.946641703721039,
      "grad_norm": 0.18728779256343842,
      "learning_rate": 6.265897601044829e-06,
      "loss": 0.047,
      "step": 50820
    },
    {
      "epoch": 5.946641703721039,
      "loss_main": 0.045386821031570435,
      "step": 50820
    },
    {
      "epoch": 5.947811841797332,
      "grad_norm": 0.2301318347454071,
      "learning_rate": 6.2525470428988434e-06,
      "loss": 0.0435,
      "step": 50830
    },
    {
      "epoch": 5.947811841797332,
      "loss_main": 0.06687687337398529,
      "step": 50830
    },
    {
      "epoch": 5.948981979873625,
      "grad_norm": 0.17617246508598328,
      "learning_rate": 6.239209774073962e-06,
      "loss": 0.0513,
      "step": 50840
    },
    {
      "epoch": 5.948981979873625,
      "loss_main": 0.03398542478680611,
      "step": 50840
    },
    {
      "epoch": 5.950152117949918,
      "grad_norm": 0.34574654698371887,
      "learning_rate": 6.225885798621728e-06,
      "loss": 0.048,
      "step": 50850
    },
    {
      "epoch": 5.950152117949918,
      "loss_main": 0.03177042677998543,
      "step": 50850
    },
    {
      "epoch": 5.951322256026211,
      "grad_norm": 0.28413406014442444,
      "learning_rate": 6.2125751205895925e-06,
      "loss": 0.0457,
      "step": 50860
    },
    {
      "epoch": 5.951322256026211,
      "loss_main": 0.07960350811481476,
      "step": 50860
    },
    {
      "epoch": 5.952492394102504,
      "grad_norm": 0.2355150282382965,
      "learning_rate": 6.199277744020998e-06,
      "loss": 0.0517,
      "step": 50870
    },
    {
      "epoch": 5.952492394102504,
      "loss_main": 0.06851968914270401,
      "step": 50870
    },
    {
      "epoch": 5.9536625321787975,
      "grad_norm": 0.21080657839775085,
      "learning_rate": 6.185993672955337e-06,
      "loss": 0.049,
      "step": 50880
    },
    {
      "epoch": 5.9536625321787975,
      "loss_main": 0.025686876848340034,
      "step": 50880
    },
    {
      "epoch": 5.95483267025509,
      "grad_norm": 0.16809937357902527,
      "learning_rate": 6.172722911427947e-06,
      "loss": 0.0469,
      "step": 50890
    },
    {
      "epoch": 5.95483267025509,
      "loss_main": 0.044609349220991135,
      "step": 50890
    },
    {
      "epoch": 5.956002808331383,
      "grad_norm": 0.25428229570388794,
      "learning_rate": 6.159465463470149e-06,
      "loss": 0.0432,
      "step": 50900
    },
    {
      "epoch": 5.956002808331383,
      "loss_main": 0.02549091912806034,
      "step": 50900
    },
    {
      "epoch": 5.957172946407676,
      "grad_norm": 0.24725192785263062,
      "learning_rate": 6.146221333109204e-06,
      "loss": 0.0551,
      "step": 50910
    },
    {
      "epoch": 5.957172946407676,
      "loss_main": 0.09674616158008575,
      "step": 50910
    },
    {
      "epoch": 5.958343084483969,
      "grad_norm": 0.2628762125968933,
      "learning_rate": 6.132990524368326e-06,
      "loss": 0.0518,
      "step": 50920
    },
    {
      "epoch": 5.958343084483969,
      "loss_main": 0.053371164947748184,
      "step": 50920
    },
    {
      "epoch": 5.959513222560262,
      "grad_norm": 0.2913195788860321,
      "learning_rate": 6.119773041266685e-06,
      "loss": 0.0461,
      "step": 50930
    },
    {
      "epoch": 5.959513222560262,
      "loss_main": 0.02128039486706257,
      "step": 50930
    },
    {
      "epoch": 5.960683360636555,
      "grad_norm": 0.21635310351848602,
      "learning_rate": 6.106568887819402e-06,
      "loss": 0.0395,
      "step": 50940
    },
    {
      "epoch": 5.960683360636555,
      "loss_main": 0.044344257563352585,
      "step": 50940
    },
    {
      "epoch": 5.961853498712848,
      "grad_norm": 0.26899418234825134,
      "learning_rate": 6.093378068037548e-06,
      "loss": 0.0418,
      "step": 50950
    },
    {
      "epoch": 5.961853498712848,
      "loss_main": 0.05787721648812294,
      "step": 50950
    },
    {
      "epoch": 5.963023636789141,
      "grad_norm": 0.2881124019622803,
      "learning_rate": 6.080200585928164e-06,
      "loss": 0.0482,
      "step": 50960
    },
    {
      "epoch": 5.963023636789141,
      "loss_main": 0.046416204422712326,
      "step": 50960
    },
    {
      "epoch": 5.964193774865434,
      "grad_norm": 0.2714746296405792,
      "learning_rate": 6.06703644549419e-06,
      "loss": 0.0477,
      "step": 50970
    },
    {
      "epoch": 5.964193774865434,
      "loss_main": 0.03546607121825218,
      "step": 50970
    },
    {
      "epoch": 5.9653639129417275,
      "grad_norm": 0.22440306842327118,
      "learning_rate": 6.053885650734576e-06,
      "loss": 0.0454,
      "step": 50980
    },
    {
      "epoch": 5.9653639129417275,
      "loss_main": 0.04212220758199692,
      "step": 50980
    },
    {
      "epoch": 5.96653405101802,
      "grad_norm": 0.2702348828315735,
      "learning_rate": 6.040748205644153e-06,
      "loss": 0.0512,
      "step": 50990
    },
    {
      "epoch": 5.96653405101802,
      "loss_main": 0.037725988775491714,
      "step": 50990
    },
    {
      "epoch": 5.967704189094313,
      "grad_norm": 0.23481924831867218,
      "learning_rate": 6.0276241142137646e-06,
      "loss": 0.0449,
      "step": 51000
    },
    {
      "epoch": 5.967704189094313,
      "loss_main": 0.0698159709572792,
      "step": 51000
    },
    {
      "epoch": 5.968874327170607,
      "grad_norm": 0.26345962285995483,
      "learning_rate": 6.014513380430142e-06,
      "loss": 0.0598,
      "step": 51010
    },
    {
      "epoch": 5.968874327170607,
      "loss_main": 0.03783062472939491,
      "step": 51010
    },
    {
      "epoch": 5.970044465246899,
      "grad_norm": 0.18671369552612305,
      "learning_rate": 6.001416008275984e-06,
      "loss": 0.0486,
      "step": 51020
    },
    {
      "epoch": 5.970044465246899,
      "loss_main": 0.049765586853027344,
      "step": 51020
    },
    {
      "epoch": 5.971214603323192,
      "grad_norm": 0.25545811653137207,
      "learning_rate": 5.988332001729929e-06,
      "loss": 0.0515,
      "step": 51030
    },
    {
      "epoch": 5.971214603323192,
      "loss_main": 0.06226324662566185,
      "step": 51030
    },
    {
      "epoch": 5.972384741399485,
      "grad_norm": 0.24928976595401764,
      "learning_rate": 5.975261364766543e-06,
      "loss": 0.0385,
      "step": 51040
    },
    {
      "epoch": 5.972384741399485,
      "loss_main": 0.04526362195611,
      "step": 51040
    },
    {
      "epoch": 5.9735548794757785,
      "grad_norm": 0.2012772411108017,
      "learning_rate": 5.962204101356356e-06,
      "loss": 0.0507,
      "step": 51050
    },
    {
      "epoch": 5.9735548794757785,
      "loss_main": 0.05448555946350098,
      "step": 51050
    },
    {
      "epoch": 5.974725017552071,
      "grad_norm": 0.24163392186164856,
      "learning_rate": 5.94916021546581e-06,
      "loss": 0.0518,
      "step": 51060
    },
    {
      "epoch": 5.974725017552071,
      "loss_main": 0.02910882793366909,
      "step": 51060
    },
    {
      "epoch": 5.975895155628364,
      "grad_norm": 0.2650858461856842,
      "learning_rate": 5.936129711057298e-06,
      "loss": 0.0488,
      "step": 51070
    },
    {
      "epoch": 5.975895155628364,
      "loss_main": 0.036212120205163956,
      "step": 51070
    },
    {
      "epoch": 5.977065293704657,
      "grad_norm": 0.22242912650108337,
      "learning_rate": 5.923112592089142e-06,
      "loss": 0.0483,
      "step": 51080
    },
    {
      "epoch": 5.977065293704657,
      "loss_main": 0.05383912846446037,
      "step": 51080
    },
    {
      "epoch": 5.97823543178095,
      "grad_norm": 0.22431392967700958,
      "learning_rate": 5.9101088625155954e-06,
      "loss": 0.0439,
      "step": 51090
    },
    {
      "epoch": 5.97823543178095,
      "loss_main": 0.02578434720635414,
      "step": 51090
    },
    {
      "epoch": 5.979405569857243,
      "grad_norm": 0.2291882485151291,
      "learning_rate": 5.897118526286843e-06,
      "loss": 0.0415,
      "step": 51100
    },
    {
      "epoch": 5.979405569857243,
      "loss_main": 0.03798583522439003,
      "step": 51100
    },
    {
      "epoch": 5.980575707933536,
      "grad_norm": 0.23543372750282288,
      "learning_rate": 5.884141587349035e-06,
      "loss": 0.0461,
      "step": 51110
    },
    {
      "epoch": 5.980575707933536,
      "loss_main": 0.037982504814863205,
      "step": 51110
    },
    {
      "epoch": 5.981745846009829,
      "grad_norm": 0.21268044412136078,
      "learning_rate": 5.871178049644177e-06,
      "loss": 0.046,
      "step": 51120
    },
    {
      "epoch": 5.981745846009829,
      "loss_main": 0.030975358560681343,
      "step": 51120
    },
    {
      "epoch": 5.982915984086122,
      "grad_norm": 0.2788824141025543,
      "learning_rate": 5.858227917110293e-06,
      "loss": 0.0482,
      "step": 51130
    },
    {
      "epoch": 5.982915984086122,
      "loss_main": 0.034184426069259644,
      "step": 51130
    },
    {
      "epoch": 5.984086122162415,
      "grad_norm": 0.2132880985736847,
      "learning_rate": 5.845291193681252e-06,
      "loss": 0.0472,
      "step": 51140
    },
    {
      "epoch": 5.984086122162415,
      "loss_main": 0.03275404870510101,
      "step": 51140
    },
    {
      "epoch": 5.9852562602387085,
      "grad_norm": 0.23079298436641693,
      "learning_rate": 5.832367883286921e-06,
      "loss": 0.0468,
      "step": 51150
    },
    {
      "epoch": 5.9852562602387085,
      "loss_main": 0.04539722576737404,
      "step": 51150
    },
    {
      "epoch": 5.986426398315001,
      "grad_norm": 0.21475541591644287,
      "learning_rate": 5.81945798985305e-06,
      "loss": 0.0445,
      "step": 51160
    },
    {
      "epoch": 5.986426398315001,
      "loss_main": 0.04553694278001785,
      "step": 51160
    },
    {
      "epoch": 5.987596536391294,
      "grad_norm": 0.27872541546821594,
      "learning_rate": 5.806561517301306e-06,
      "loss": 0.0442,
      "step": 51170
    },
    {
      "epoch": 5.987596536391294,
      "loss_main": 0.04888525232672691,
      "step": 51170
    },
    {
      "epoch": 5.988766674467588,
      "grad_norm": 0.2710024416446686,
      "learning_rate": 5.793678469549335e-06,
      "loss": 0.0508,
      "step": 51180
    },
    {
      "epoch": 5.988766674467588,
      "loss_main": 0.08192194253206253,
      "step": 51180
    },
    {
      "epoch": 5.98993681254388,
      "grad_norm": 0.25315213203430176,
      "learning_rate": 5.780808850510627e-06,
      "loss": 0.0466,
      "step": 51190
    },
    {
      "epoch": 5.98993681254388,
      "loss_main": 0.0513017438352108,
      "step": 51190
    },
    {
      "epoch": 5.991106950620173,
      "grad_norm": 0.2359778881072998,
      "learning_rate": 5.767952664094673e-06,
      "loss": 0.0431,
      "step": 51200
    },
    {
      "epoch": 5.991106950620173,
      "loss_main": 0.057280924171209335,
      "step": 51200
    },
    {
      "epoch": 5.992277088696466,
      "grad_norm": 0.23994575440883636,
      "learning_rate": 5.755109914206791e-06,
      "loss": 0.0468,
      "step": 51210
    },
    {
      "epoch": 5.992277088696466,
      "loss_main": 0.056185267865657806,
      "step": 51210
    },
    {
      "epoch": 5.993447226772759,
      "grad_norm": 0.174892857670784,
      "learning_rate": 5.7422806047483125e-06,
      "loss": 0.0411,
      "step": 51220
    },
    {
      "epoch": 5.993447226772759,
      "loss_main": 0.03235592320561409,
      "step": 51220
    },
    {
      "epoch": 5.994617364849052,
      "grad_norm": 0.2024904489517212,
      "learning_rate": 5.72946473961643e-06,
      "loss": 0.0436,
      "step": 51230
    },
    {
      "epoch": 5.994617364849052,
      "loss_main": 0.04402059689164162,
      "step": 51230
    },
    {
      "epoch": 5.995787502925345,
      "grad_norm": 0.2864382565021515,
      "learning_rate": 5.716662322704264e-06,
      "loss": 0.0456,
      "step": 51240
    },
    {
      "epoch": 5.995787502925345,
      "loss_main": 0.03654518350958824,
      "step": 51240
    },
    {
      "epoch": 5.9969576410016385,
      "grad_norm": 0.22013124823570251,
      "learning_rate": 5.703873357900852e-06,
      "loss": 0.0503,
      "step": 51250
    },
    {
      "epoch": 5.9969576410016385,
      "loss_main": 0.04305038973689079,
      "step": 51250
    },
    {
      "epoch": 5.998127779077931,
      "grad_norm": 0.26885437965393066,
      "learning_rate": 5.691097849091143e-06,
      "loss": 0.0449,
      "step": 51260
    },
    {
      "epoch": 5.998127779077931,
      "loss_main": 0.06177343800663948,
      "step": 51260
    },
    {
      "epoch": 5.999297917154224,
      "grad_norm": 0.2256559580564499,
      "learning_rate": 5.678335800155982e-06,
      "loss": 0.0522,
      "step": 51270
    },
    {
      "epoch": 5.999297917154224,
      "loss_main": 0.037829022854566574,
      "step": 51270
    },
    {
      "epoch": 6.000468055230518,
      "grad_norm": 0.15948902070522308,
      "learning_rate": 5.665587214972174e-06,
      "loss": 0.0398,
      "step": 51280
    },
    {
      "epoch": 6.000468055230518,
      "loss_main": 0.05215173959732056,
      "step": 51280
    },
    {
      "epoch": 6.00163819330681,
      "grad_norm": 0.201999694108963,
      "learning_rate": 5.652852097412386e-06,
      "loss": 0.0434,
      "step": 51290
    },
    {
      "epoch": 6.00163819330681,
      "loss_main": 0.043294042348861694,
      "step": 51290
    },
    {
      "epoch": 6.002808331383103,
      "grad_norm": 0.18854683637619019,
      "learning_rate": 5.640130451345216e-06,
      "loss": 0.0413,
      "step": 51300
    },
    {
      "epoch": 6.002808331383103,
      "loss_main": 0.0653870552778244,
      "step": 51300
    },
    {
      "epoch": 6.003978469459396,
      "grad_norm": 0.22057387232780457,
      "learning_rate": 5.627422280635159e-06,
      "loss": 0.0416,
      "step": 51310
    },
    {
      "epoch": 6.003978469459396,
      "loss_main": 0.037608351558446884,
      "step": 51310
    },
    {
      "epoch": 6.0051486075356895,
      "grad_norm": 0.20786917209625244,
      "learning_rate": 5.61472758914261e-06,
      "loss": 0.0413,
      "step": 51320
    },
    {
      "epoch": 6.0051486075356895,
      "loss_main": 0.047047048807144165,
      "step": 51320
    },
    {
      "epoch": 6.006318745611982,
      "grad_norm": 0.2627561688423157,
      "learning_rate": 5.602046380723918e-06,
      "loss": 0.0409,
      "step": 51330
    },
    {
      "epoch": 6.006318745611982,
      "loss_main": 0.06853713095188141,
      "step": 51330
    },
    {
      "epoch": 6.007488883688275,
      "grad_norm": 0.2521172761917114,
      "learning_rate": 5.5893786592312535e-06,
      "loss": 0.0486,
      "step": 51340
    },
    {
      "epoch": 6.007488883688275,
      "loss_main": 0.039924256503582,
      "step": 51340
    },
    {
      "epoch": 6.008659021764569,
      "grad_norm": 0.22017724812030792,
      "learning_rate": 5.576724428512775e-06,
      "loss": 0.041,
      "step": 51350
    },
    {
      "epoch": 6.008659021764569,
      "loss_main": 0.06540004909038544,
      "step": 51350
    },
    {
      "epoch": 6.009829159840861,
      "grad_norm": 0.22222208976745605,
      "learning_rate": 5.56408369241247e-06,
      "loss": 0.0538,
      "step": 51360
    },
    {
      "epoch": 6.009829159840861,
      "loss_main": 0.03834955021739006,
      "step": 51360
    },
    {
      "epoch": 6.010999297917154,
      "grad_norm": 0.2215447574853897,
      "learning_rate": 5.5514564547702875e-06,
      "loss": 0.0433,
      "step": 51370
    },
    {
      "epoch": 6.010999297917154,
      "loss_main": 0.03409285843372345,
      "step": 51370
    },
    {
      "epoch": 6.012169435993448,
      "grad_norm": 0.2677093744277954,
      "learning_rate": 5.538842719422038e-06,
      "loss": 0.0481,
      "step": 51380
    },
    {
      "epoch": 6.012169435993448,
      "loss_main": 0.057515330612659454,
      "step": 51380
    },
    {
      "epoch": 6.01333957406974,
      "grad_norm": 0.2922666072845459,
      "learning_rate": 5.52624249019944e-06,
      "loss": 0.0455,
      "step": 51390
    },
    {
      "epoch": 6.01333957406974,
      "loss_main": 0.02873261086642742,
      "step": 51390
    },
    {
      "epoch": 6.014509712146033,
      "grad_norm": 0.2616964280605316,
      "learning_rate": 5.51365577093011e-06,
      "loss": 0.0549,
      "step": 51400
    },
    {
      "epoch": 6.014509712146033,
      "loss_main": 0.06306910514831543,
      "step": 51400
    },
    {
      "epoch": 6.015679850222326,
      "grad_norm": 0.23524272441864014,
      "learning_rate": 5.501082565437565e-06,
      "loss": 0.0451,
      "step": 51410
    },
    {
      "epoch": 6.015679850222326,
      "loss_main": 0.04289958253502846,
      "step": 51410
    },
    {
      "epoch": 6.0168499882986195,
      "grad_norm": 0.16810354590415955,
      "learning_rate": 5.488522877541202e-06,
      "loss": 0.0467,
      "step": 51420
    },
    {
      "epoch": 6.0168499882986195,
      "loss_main": 0.04099204018712044,
      "step": 51420
    },
    {
      "epoch": 6.018020126374912,
      "grad_norm": 0.263671875,
      "learning_rate": 5.475976711056341e-06,
      "loss": 0.0487,
      "step": 51430
    },
    {
      "epoch": 6.018020126374912,
      "loss_main": 0.06018102914094925,
      "step": 51430
    },
    {
      "epoch": 6.019190264451205,
      "grad_norm": 0.27283331751823425,
      "learning_rate": 5.463444069794166e-06,
      "loss": 0.0494,
      "step": 51440
    },
    {
      "epoch": 6.019190264451205,
      "loss_main": 0.05258646234869957,
      "step": 51440
    },
    {
      "epoch": 6.020360402527499,
      "grad_norm": 0.28583618998527527,
      "learning_rate": 5.4509249575617594e-06,
      "loss": 0.049,
      "step": 51450
    },
    {
      "epoch": 6.020360402527499,
      "loss_main": 0.02175794169306755,
      "step": 51450
    },
    {
      "epoch": 6.021530540603791,
      "grad_norm": 0.23085230588912964,
      "learning_rate": 5.438419378162107e-06,
      "loss": 0.0493,
      "step": 51460
    },
    {
      "epoch": 6.021530540603791,
      "loss_main": 0.06430646032094955,
      "step": 51460
    },
    {
      "epoch": 6.022700678680084,
      "grad_norm": 0.18491286039352417,
      "learning_rate": 5.425927335394054e-06,
      "loss": 0.0486,
      "step": 51470
    },
    {
      "epoch": 6.022700678680084,
      "loss_main": 0.0404277928173542,
      "step": 51470
    },
    {
      "epoch": 6.023870816756377,
      "grad_norm": 0.2851981520652771,
      "learning_rate": 5.413448833052387e-06,
      "loss": 0.0395,
      "step": 51480
    },
    {
      "epoch": 6.023870816756377,
      "loss_main": 0.03276766091585159,
      "step": 51480
    },
    {
      "epoch": 6.02504095483267,
      "grad_norm": 0.25482049584388733,
      "learning_rate": 5.400983874927701e-06,
      "loss": 0.0358,
      "step": 51490
    },
    {
      "epoch": 6.02504095483267,
      "loss_main": 0.0506725013256073,
      "step": 51490
    },
    {
      "epoch": 6.026211092908963,
      "grad_norm": 0.25469768047332764,
      "learning_rate": 5.388532464806567e-06,
      "loss": 0.044,
      "step": 51500
    },
    {
      "epoch": 6.026211092908963,
      "loss_main": 0.03467552736401558,
      "step": 51500
    },
    {
      "epoch": 6.027381230985256,
      "grad_norm": 0.2726966142654419,
      "learning_rate": 5.3760946064713546e-06,
      "loss": 0.0428,
      "step": 51510
    },
    {
      "epoch": 6.027381230985256,
      "loss_main": 0.03960634022951126,
      "step": 51510
    },
    {
      "epoch": 6.0285513690615495,
      "grad_norm": 0.19982914626598358,
      "learning_rate": 5.363670303700386e-06,
      "loss": 0.0462,
      "step": 51520
    },
    {
      "epoch": 6.0285513690615495,
      "loss_main": 0.0347733348608017,
      "step": 51520
    },
    {
      "epoch": 6.029721507137842,
      "grad_norm": 0.2535623610019684,
      "learning_rate": 5.351259560267824e-06,
      "loss": 0.0434,
      "step": 51530
    },
    {
      "epoch": 6.029721507137842,
      "loss_main": 0.04548116400837898,
      "step": 51530
    },
    {
      "epoch": 6.030891645214135,
      "grad_norm": 0.22193124890327454,
      "learning_rate": 5.338862379943721e-06,
      "loss": 0.0465,
      "step": 51540
    },
    {
      "epoch": 6.030891645214135,
      "loss_main": 0.041123874485492706,
      "step": 51540
    },
    {
      "epoch": 6.032061783290429,
      "grad_norm": 0.26519230008125305,
      "learning_rate": 5.326478766494025e-06,
      "loss": 0.0422,
      "step": 51550
    },
    {
      "epoch": 6.032061783290429,
      "loss_main": 0.039572034031152725,
      "step": 51550
    },
    {
      "epoch": 6.033231921366721,
      "grad_norm": 0.22074535489082336,
      "learning_rate": 5.3141087236805385e-06,
      "loss": 0.0612,
      "step": 51560
    },
    {
      "epoch": 6.033231921366721,
      "loss_main": 0.05821124091744423,
      "step": 51560
    },
    {
      "epoch": 6.034402059443014,
      "grad_norm": 0.23490574955940247,
      "learning_rate": 5.3017522552609615e-06,
      "loss": 0.0471,
      "step": 51570
    },
    {
      "epoch": 6.034402059443014,
      "loss_main": 0.03641122207045555,
      "step": 51570
    },
    {
      "epoch": 6.035572197519307,
      "grad_norm": 0.2780475318431854,
      "learning_rate": 5.289409364988851e-06,
      "loss": 0.0391,
      "step": 51580
    },
    {
      "epoch": 6.035572197519307,
      "loss_main": 0.017188552767038345,
      "step": 51580
    },
    {
      "epoch": 6.0367423355956005,
      "grad_norm": 0.2873583734035492,
      "learning_rate": 5.277080056613671e-06,
      "loss": 0.0487,
      "step": 51590
    },
    {
      "epoch": 6.0367423355956005,
      "loss_main": 0.048275332897901535,
      "step": 51590
    },
    {
      "epoch": 6.037912473671893,
      "grad_norm": 0.24512843787670135,
      "learning_rate": 5.264764333880729e-06,
      "loss": 0.046,
      "step": 51600
    },
    {
      "epoch": 6.037912473671893,
      "loss_main": 0.03928902745246887,
      "step": 51600
    },
    {
      "epoch": 6.039082611748186,
      "grad_norm": 0.19098763167858124,
      "learning_rate": 5.252462200531216e-06,
      "loss": 0.0416,
      "step": 51610
    },
    {
      "epoch": 6.039082611748186,
      "loss_main": 0.0577639602124691,
      "step": 51610
    },
    {
      "epoch": 6.04025274982448,
      "grad_norm": 0.25118768215179443,
      "learning_rate": 5.240173660302194e-06,
      "loss": 0.0425,
      "step": 51620
    },
    {
      "epoch": 6.04025274982448,
      "loss_main": 0.055475909262895584,
      "step": 51620
    },
    {
      "epoch": 6.041422887900772,
      "grad_norm": 0.21049551665782928,
      "learning_rate": 5.2278987169266044e-06,
      "loss": 0.0373,
      "step": 51630
    },
    {
      "epoch": 6.041422887900772,
      "loss_main": 0.06157488748431206,
      "step": 51630
    },
    {
      "epoch": 6.042593025977065,
      "grad_norm": 0.21460489928722382,
      "learning_rate": 5.215637374133231e-06,
      "loss": 0.0441,
      "step": 51640
    },
    {
      "epoch": 6.042593025977065,
      "loss_main": 0.04320681467652321,
      "step": 51640
    },
    {
      "epoch": 6.043763164053359,
      "grad_norm": 0.21575580537319183,
      "learning_rate": 5.203389635646782e-06,
      "loss": 0.0428,
      "step": 51650
    },
    {
      "epoch": 6.043763164053359,
      "loss_main": 0.025901509448885918,
      "step": 51650
    },
    {
      "epoch": 6.044933302129651,
      "grad_norm": 0.3205035626888275,
      "learning_rate": 5.191155505187756e-06,
      "loss": 0.0496,
      "step": 51660
    },
    {
      "epoch": 6.044933302129651,
      "loss_main": 0.048644810914993286,
      "step": 51660
    },
    {
      "epoch": 6.046103440205944,
      "grad_norm": 0.290522038936615,
      "learning_rate": 5.1789349864726e-06,
      "loss": 0.0466,
      "step": 51670
    },
    {
      "epoch": 6.046103440205944,
      "loss_main": 0.02184288017451763,
      "step": 51670
    },
    {
      "epoch": 6.047273578282237,
      "grad_norm": 0.2225378453731537,
      "learning_rate": 5.16672808321354e-06,
      "loss": 0.0414,
      "step": 51680
    },
    {
      "epoch": 6.047273578282237,
      "loss_main": 0.048519253730773926,
      "step": 51680
    },
    {
      "epoch": 6.0484437163585305,
      "grad_norm": 0.33491796255111694,
      "learning_rate": 5.154534799118749e-06,
      "loss": 0.0543,
      "step": 51690
    },
    {
      "epoch": 6.0484437163585305,
      "loss_main": 0.04945448413491249,
      "step": 51690
    },
    {
      "epoch": 6.049613854434823,
      "grad_norm": 0.17026366293430328,
      "learning_rate": 5.142355137892207e-06,
      "loss": 0.0493,
      "step": 51700
    },
    {
      "epoch": 6.049613854434823,
      "loss_main": 0.03899505361914635,
      "step": 51700
    },
    {
      "epoch": 6.050783992511116,
      "grad_norm": 0.2358725368976593,
      "learning_rate": 5.130189103233779e-06,
      "loss": 0.0472,
      "step": 51710
    },
    {
      "epoch": 6.050783992511116,
      "loss_main": 0.03470376878976822,
      "step": 51710
    },
    {
      "epoch": 6.05195413058741,
      "grad_norm": 0.15523633360862732,
      "learning_rate": 5.118036698839179e-06,
      "loss": 0.0432,
      "step": 51720
    },
    {
      "epoch": 6.05195413058741,
      "loss_main": 0.02912520244717598,
      "step": 51720
    },
    {
      "epoch": 6.053124268663702,
      "grad_norm": 0.17386296391487122,
      "learning_rate": 5.105897928399983e-06,
      "loss": 0.0428,
      "step": 51730
    },
    {
      "epoch": 6.053124268663702,
      "loss_main": 0.04293080419301987,
      "step": 51730
    },
    {
      "epoch": 6.054294406739995,
      "grad_norm": 0.19924819469451904,
      "learning_rate": 5.0937727956036464e-06,
      "loss": 0.0475,
      "step": 51740
    },
    {
      "epoch": 6.054294406739995,
      "loss_main": 0.0422988161444664,
      "step": 51740
    },
    {
      "epoch": 6.055464544816289,
      "grad_norm": 0.30890944600105286,
      "learning_rate": 5.081661304133456e-06,
      "loss": 0.0449,
      "step": 51750
    },
    {
      "epoch": 6.055464544816289,
      "loss_main": 0.06862302124500275,
      "step": 51750
    },
    {
      "epoch": 6.0566346828925814,
      "grad_norm": 0.2183355987071991,
      "learning_rate": 5.069563457668558e-06,
      "loss": 0.0481,
      "step": 51760
    },
    {
      "epoch": 6.0566346828925814,
      "loss_main": 0.048779044300317764,
      "step": 51760
    },
    {
      "epoch": 6.057804820968874,
      "grad_norm": 0.27101194858551025,
      "learning_rate": 5.0574792598839624e-06,
      "loss": 0.0512,
      "step": 51770
    },
    {
      "epoch": 6.057804820968874,
      "loss_main": 0.06185897812247276,
      "step": 51770
    },
    {
      "epoch": 6.058974959045167,
      "grad_norm": 0.19070111215114594,
      "learning_rate": 5.0454087144505276e-06,
      "loss": 0.0457,
      "step": 51780
    },
    {
      "epoch": 6.058974959045167,
      "loss_main": 0.07766050845384598,
      "step": 51780
    },
    {
      "epoch": 6.0601450971214605,
      "grad_norm": 0.20215138792991638,
      "learning_rate": 5.0333518250349655e-06,
      "loss": 0.0475,
      "step": 51790
    },
    {
      "epoch": 6.0601450971214605,
      "loss_main": 0.07138935476541519,
      "step": 51790
    },
    {
      "epoch": 6.061315235197753,
      "grad_norm": 0.17278869450092316,
      "learning_rate": 5.021308595299856e-06,
      "loss": 0.0499,
      "step": 51800
    },
    {
      "epoch": 6.061315235197753,
      "loss_main": 0.033961351960897446,
      "step": 51800
    },
    {
      "epoch": 6.062485373274046,
      "grad_norm": 0.19791670143604279,
      "learning_rate": 5.009279028903585e-06,
      "loss": 0.0412,
      "step": 51810
    },
    {
      "epoch": 6.062485373274046,
      "loss_main": 0.10287681967020035,
      "step": 51810
    },
    {
      "epoch": 6.06365551135034,
      "grad_norm": 0.24314825236797333,
      "learning_rate": 4.997263129500452e-06,
      "loss": 0.0495,
      "step": 51820
    },
    {
      "epoch": 6.06365551135034,
      "loss_main": 0.06923755258321762,
      "step": 51820
    },
    {
      "epoch": 6.064825649426632,
      "grad_norm": 0.23467253148555756,
      "learning_rate": 4.985260900740535e-06,
      "loss": 0.0452,
      "step": 51830
    },
    {
      "epoch": 6.064825649426632,
      "loss_main": 0.042349476367235184,
      "step": 51830
    },
    {
      "epoch": 6.065995787502925,
      "grad_norm": 0.19860206544399261,
      "learning_rate": 4.973272346269814e-06,
      "loss": 0.0392,
      "step": 51840
    },
    {
      "epoch": 6.065995787502925,
      "loss_main": 0.04787316545844078,
      "step": 51840
    },
    {
      "epoch": 6.067165925579219,
      "grad_norm": 0.2478487193584442,
      "learning_rate": 4.961297469730097e-06,
      "loss": 0.0473,
      "step": 51850
    },
    {
      "epoch": 6.067165925579219,
      "loss_main": 0.0456102192401886,
      "step": 51850
    },
    {
      "epoch": 6.0683360636555115,
      "grad_norm": 0.20138686895370483,
      "learning_rate": 4.949336274759031e-06,
      "loss": 0.0385,
      "step": 51860
    },
    {
      "epoch": 6.0683360636555115,
      "loss_main": 0.023242609575390816,
      "step": 51860
    },
    {
      "epoch": 6.069506201731804,
      "grad_norm": 0.2869415879249573,
      "learning_rate": 4.9373887649901144e-06,
      "loss": 0.0416,
      "step": 51870
    },
    {
      "epoch": 6.069506201731804,
      "loss_main": 0.05677096173167229,
      "step": 51870
    },
    {
      "epoch": 6.070676339808097,
      "grad_norm": 0.2399047613143921,
      "learning_rate": 4.925454944052666e-06,
      "loss": 0.0465,
      "step": 51880
    },
    {
      "epoch": 6.070676339808097,
      "loss_main": 0.03892337530851364,
      "step": 51880
    },
    {
      "epoch": 6.071846477884391,
      "grad_norm": 0.2823159396648407,
      "learning_rate": 4.913534815571891e-06,
      "loss": 0.0434,
      "step": 51890
    },
    {
      "epoch": 6.071846477884391,
      "loss_main": 0.04617660492658615,
      "step": 51890
    },
    {
      "epoch": 6.073016615960683,
      "grad_norm": 0.233501598238945,
      "learning_rate": 4.901628383168805e-06,
      "loss": 0.0451,
      "step": 51900
    },
    {
      "epoch": 6.073016615960683,
      "loss_main": 0.09078675508499146,
      "step": 51900
    },
    {
      "epoch": 6.074186754036976,
      "grad_norm": 0.27672630548477173,
      "learning_rate": 4.8897356504602585e-06,
      "loss": 0.0447,
      "step": 51910
    },
    {
      "epoch": 6.074186754036976,
      "loss_main": 0.025823941454291344,
      "step": 51910
    },
    {
      "epoch": 6.07535689211327,
      "grad_norm": 0.22194087505340576,
      "learning_rate": 4.877856621058957e-06,
      "loss": 0.0426,
      "step": 51920
    },
    {
      "epoch": 6.07535689211327,
      "loss_main": 0.06187563017010689,
      "step": 51920
    },
    {
      "epoch": 6.076527030189562,
      "grad_norm": 0.2775595784187317,
      "learning_rate": 4.86599129857343e-06,
      "loss": 0.0494,
      "step": 51930
    },
    {
      "epoch": 6.076527030189562,
      "loss_main": 0.04938831552863121,
      "step": 51930
    },
    {
      "epoch": 6.077697168265855,
      "grad_norm": 0.19529017806053162,
      "learning_rate": 4.85413968660805e-06,
      "loss": 0.0525,
      "step": 51940
    },
    {
      "epoch": 6.077697168265855,
      "loss_main": 0.04369325190782547,
      "step": 51940
    },
    {
      "epoch": 6.078867306342149,
      "grad_norm": 0.19053910672664642,
      "learning_rate": 4.842301788763031e-06,
      "loss": 0.0398,
      "step": 51950
    },
    {
      "epoch": 6.078867306342149,
      "loss_main": 0.029688874259591103,
      "step": 51950
    },
    {
      "epoch": 6.0800374444184415,
      "grad_norm": 0.2001488208770752,
      "learning_rate": 4.830477608634393e-06,
      "loss": 0.0447,
      "step": 51960
    },
    {
      "epoch": 6.0800374444184415,
      "loss_main": 0.04641661047935486,
      "step": 51960
    },
    {
      "epoch": 6.081207582494734,
      "grad_norm": 0.21081721782684326,
      "learning_rate": 4.818667149814049e-06,
      "loss": 0.0477,
      "step": 51970
    },
    {
      "epoch": 6.081207582494734,
      "loss_main": 0.02887067385017872,
      "step": 51970
    },
    {
      "epoch": 6.082377720571027,
      "grad_norm": 0.19983360171318054,
      "learning_rate": 4.806870415889664e-06,
      "loss": 0.0442,
      "step": 51980
    },
    {
      "epoch": 6.082377720571027,
      "loss_main": 0.055469926446676254,
      "step": 51980
    },
    {
      "epoch": 6.083547858647321,
      "grad_norm": 0.29199206829071045,
      "learning_rate": 4.795087410444798e-06,
      "loss": 0.0492,
      "step": 51990
    },
    {
      "epoch": 6.083547858647321,
      "loss_main": 0.05315762758255005,
      "step": 51990
    },
    {
      "epoch": 6.084717996723613,
      "grad_norm": 0.22360354661941528,
      "learning_rate": 4.783318137058807e-06,
      "loss": 0.0513,
      "step": 52000
    },
    {
      "epoch": 6.084717996723613,
      "loss_main": 0.06276578456163406,
      "step": 52000
    },
    {
      "epoch": 6.085888134799906,
      "grad_norm": 0.2660311162471771,
      "learning_rate": 4.771562599306895e-06,
      "loss": 0.0431,
      "step": 52010
    },
    {
      "epoch": 6.085888134799906,
      "loss_main": 0.05086822062730789,
      "step": 52010
    },
    {
      "epoch": 6.0870582728762,
      "grad_norm": 0.415300577878952,
      "learning_rate": 4.759820800760073e-06,
      "loss": 0.0466,
      "step": 52020
    },
    {
      "epoch": 6.0870582728762,
      "loss_main": 0.053558919578790665,
      "step": 52020
    },
    {
      "epoch": 6.0882284109524925,
      "grad_norm": 0.22321851551532745,
      "learning_rate": 4.7480927449851834e-06,
      "loss": 0.0459,
      "step": 52030
    },
    {
      "epoch": 6.0882284109524925,
      "loss_main": 0.04238341003656387,
      "step": 52030
    },
    {
      "epoch": 6.089398549028785,
      "grad_norm": 0.22164347767829895,
      "learning_rate": 4.73637843554493e-06,
      "loss": 0.043,
      "step": 52040
    },
    {
      "epoch": 6.089398549028785,
      "loss_main": 0.05159895494580269,
      "step": 52040
    },
    {
      "epoch": 6.090568687105079,
      "grad_norm": 0.1925133615732193,
      "learning_rate": 4.724677875997774e-06,
      "loss": 0.051,
      "step": 52050
    },
    {
      "epoch": 6.090568687105079,
      "loss_main": 0.07682304084300995,
      "step": 52050
    },
    {
      "epoch": 6.0917388251813716,
      "grad_norm": 0.21679110825061798,
      "learning_rate": 4.712991069898065e-06,
      "loss": 0.0494,
      "step": 52060
    },
    {
      "epoch": 6.0917388251813716,
      "loss_main": 0.035426389425992966,
      "step": 52060
    },
    {
      "epoch": 6.092908963257664,
      "grad_norm": 0.2481517195701599,
      "learning_rate": 4.7013180207959305e-06,
      "loss": 0.0463,
      "step": 52070
    },
    {
      "epoch": 6.092908963257664,
      "loss_main": 0.04276192560791969,
      "step": 52070
    },
    {
      "epoch": 6.094079101333957,
      "grad_norm": 0.16149477660655975,
      "learning_rate": 4.6896587322373395e-06,
      "loss": 0.043,
      "step": 52080
    },
    {
      "epoch": 6.094079101333957,
      "loss_main": 0.022611260414123535,
      "step": 52080
    },
    {
      "epoch": 6.095249239410251,
      "grad_norm": 0.2134123593568802,
      "learning_rate": 4.678013207764081e-06,
      "loss": 0.0412,
      "step": 52090
    },
    {
      "epoch": 6.095249239410251,
      "loss_main": 0.04509410634636879,
      "step": 52090
    },
    {
      "epoch": 6.096419377486543,
      "grad_norm": 0.19116690754890442,
      "learning_rate": 4.666381450913748e-06,
      "loss": 0.0376,
      "step": 52100
    },
    {
      "epoch": 6.096419377486543,
      "loss_main": 0.04231492802500725,
      "step": 52100
    },
    {
      "epoch": 6.097589515562836,
      "grad_norm": 0.19464460015296936,
      "learning_rate": 4.654763465219752e-06,
      "loss": 0.0473,
      "step": 52110
    },
    {
      "epoch": 6.097589515562836,
      "loss_main": 0.03969528153538704,
      "step": 52110
    },
    {
      "epoch": 6.09875965363913,
      "grad_norm": 0.34480777382850647,
      "learning_rate": 4.643159254211371e-06,
      "loss": 0.0457,
      "step": 52120
    },
    {
      "epoch": 6.09875965363913,
      "loss_main": 0.0299830324947834,
      "step": 52120
    },
    {
      "epoch": 6.0999297917154225,
      "grad_norm": 0.18806415796279907,
      "learning_rate": 4.631568821413606e-06,
      "loss": 0.0381,
      "step": 52130
    },
    {
      "epoch": 6.0999297917154225,
      "loss_main": 0.02941122092306614,
      "step": 52130
    },
    {
      "epoch": 6.101099929791715,
      "grad_norm": 0.30654266476631165,
      "learning_rate": 4.619992170347359e-06,
      "loss": 0.0411,
      "step": 52140
    },
    {
      "epoch": 6.101099929791715,
      "loss_main": 0.032120633870363235,
      "step": 52140
    },
    {
      "epoch": 6.102270067868009,
      "grad_norm": 0.16147807240486145,
      "learning_rate": 4.608429304529305e-06,
      "loss": 0.0465,
      "step": 52150
    },
    {
      "epoch": 6.102270067868009,
      "loss_main": 0.04295526072382927,
      "step": 52150
    },
    {
      "epoch": 6.103440205944302,
      "grad_norm": 0.24127881228923798,
      "learning_rate": 4.596880227471928e-06,
      "loss": 0.0457,
      "step": 52160
    },
    {
      "epoch": 6.103440205944302,
      "loss_main": 0.026519697159528732,
      "step": 52160
    },
    {
      "epoch": 6.104610344020594,
      "grad_norm": 0.3500726819038391,
      "learning_rate": 4.585344942683539e-06,
      "loss": 0.0424,
      "step": 52170
    },
    {
      "epoch": 6.104610344020594,
      "loss_main": 0.03662872686982155,
      "step": 52170
    },
    {
      "epoch": 6.105780482096887,
      "grad_norm": 0.25588279962539673,
      "learning_rate": 4.573823453668241e-06,
      "loss": 0.0459,
      "step": 52180
    },
    {
      "epoch": 6.105780482096887,
      "loss_main": 0.03525608032941818,
      "step": 52180
    },
    {
      "epoch": 6.106950620173181,
      "grad_norm": 0.17564307153224945,
      "learning_rate": 4.562315763925995e-06,
      "loss": 0.0462,
      "step": 52190
    },
    {
      "epoch": 6.106950620173181,
      "loss_main": 0.06105659157037735,
      "step": 52190
    },
    {
      "epoch": 6.108120758249473,
      "grad_norm": 0.21123813092708588,
      "learning_rate": 4.5508218769524825e-06,
      "loss": 0.0521,
      "step": 52200
    },
    {
      "epoch": 6.108120758249473,
      "loss_main": 0.045530226081609726,
      "step": 52200
    },
    {
      "epoch": 6.109290896325766,
      "grad_norm": 0.133493110537529,
      "learning_rate": 4.539341796239277e-06,
      "loss": 0.0396,
      "step": 52210
    },
    {
      "epoch": 6.109290896325766,
      "loss_main": 0.05108627304434776,
      "step": 52210
    },
    {
      "epoch": 6.11046103440206,
      "grad_norm": 0.24769903719425201,
      "learning_rate": 4.527875525273717e-06,
      "loss": 0.0513,
      "step": 52220
    },
    {
      "epoch": 6.11046103440206,
      "loss_main": 0.044912610203027725,
      "step": 52220
    },
    {
      "epoch": 6.1116311724783525,
      "grad_norm": 0.33422696590423584,
      "learning_rate": 4.51642306753895e-06,
      "loss": 0.0417,
      "step": 52230
    },
    {
      "epoch": 6.1116311724783525,
      "loss_main": 0.016631800681352615,
      "step": 52230
    },
    {
      "epoch": 6.112801310554645,
      "grad_norm": 0.2150377631187439,
      "learning_rate": 4.5049844265139306e-06,
      "loss": 0.0453,
      "step": 52240
    },
    {
      "epoch": 6.112801310554645,
      "loss_main": 0.03487960994243622,
      "step": 52240
    },
    {
      "epoch": 6.113971448630939,
      "grad_norm": 0.2961125373840332,
      "learning_rate": 4.4935596056734144e-06,
      "loss": 0.042,
      "step": 52250
    },
    {
      "epoch": 6.113971448630939,
      "loss_main": 0.029462680220603943,
      "step": 52250
    },
    {
      "epoch": 6.115141586707232,
      "grad_norm": 0.1783573180437088,
      "learning_rate": 4.482148608487957e-06,
      "loss": 0.0464,
      "step": 52260
    },
    {
      "epoch": 6.115141586707232,
      "loss_main": 0.04891057685017586,
      "step": 52260
    },
    {
      "epoch": 6.116311724783524,
      "grad_norm": 0.18672795593738556,
      "learning_rate": 4.4707514384239365e-06,
      "loss": 0.0388,
      "step": 52270
    },
    {
      "epoch": 6.116311724783524,
      "loss_main": 0.020858561620116234,
      "step": 52270
    },
    {
      "epoch": 6.117481862859817,
      "grad_norm": 0.2180829793214798,
      "learning_rate": 4.459368098943484e-06,
      "loss": 0.0424,
      "step": 52280
    },
    {
      "epoch": 6.117481862859817,
      "loss_main": 0.04379085823893547,
      "step": 52280
    },
    {
      "epoch": 6.118652000936111,
      "grad_norm": 0.21419553458690643,
      "learning_rate": 4.447998593504582e-06,
      "loss": 0.0524,
      "step": 52290
    },
    {
      "epoch": 6.118652000936111,
      "loss_main": 0.03289295360445976,
      "step": 52290
    },
    {
      "epoch": 6.1198221390124035,
      "grad_norm": 0.18965654075145721,
      "learning_rate": 4.4366429255609744e-06,
      "loss": 0.046,
      "step": 52300
    },
    {
      "epoch": 6.1198221390124035,
      "loss_main": 0.04281046614050865,
      "step": 52300
    },
    {
      "epoch": 6.120992277088696,
      "grad_norm": 0.252155601978302,
      "learning_rate": 4.425301098562212e-06,
      "loss": 0.0517,
      "step": 52310
    },
    {
      "epoch": 6.120992277088696,
      "loss_main": 0.07602189481258392,
      "step": 52310
    },
    {
      "epoch": 6.12216241516499,
      "grad_norm": 0.22507423162460327,
      "learning_rate": 4.413973115953651e-06,
      "loss": 0.0455,
      "step": 52320
    },
    {
      "epoch": 6.12216241516499,
      "loss_main": 0.04345638304948807,
      "step": 52320
    },
    {
      "epoch": 6.123332553241283,
      "grad_norm": 0.31774890422821045,
      "learning_rate": 4.402658981176416e-06,
      "loss": 0.0423,
      "step": 52330
    },
    {
      "epoch": 6.123332553241283,
      "loss_main": 0.04201927408576012,
      "step": 52330
    },
    {
      "epoch": 6.124502691317575,
      "grad_norm": 0.22213156521320343,
      "learning_rate": 4.391358697667475e-06,
      "loss": 0.0465,
      "step": 52340
    },
    {
      "epoch": 6.124502691317575,
      "loss_main": 0.022691765800118446,
      "step": 52340
    },
    {
      "epoch": 6.125672829393869,
      "grad_norm": 0.3362267315387726,
      "learning_rate": 4.3800722688595195e-06,
      "loss": 0.0519,
      "step": 52350
    },
    {
      "epoch": 6.125672829393869,
      "loss_main": 0.04150858148932457,
      "step": 52350
    },
    {
      "epoch": 6.126842967470162,
      "grad_norm": 0.24365636706352234,
      "learning_rate": 4.368799698181097e-06,
      "loss": 0.0461,
      "step": 52360
    },
    {
      "epoch": 6.126842967470162,
      "loss_main": 0.04948621988296509,
      "step": 52360
    },
    {
      "epoch": 6.128013105546454,
      "grad_norm": 0.22257962822914124,
      "learning_rate": 4.357540989056486e-06,
      "loss": 0.0412,
      "step": 52370
    },
    {
      "epoch": 6.128013105546454,
      "loss_main": 0.044394321739673615,
      "step": 52370
    },
    {
      "epoch": 6.129183243622747,
      "grad_norm": 0.24574784934520721,
      "learning_rate": 4.346296144905815e-06,
      "loss": 0.0473,
      "step": 52380
    },
    {
      "epoch": 6.129183243622747,
      "loss_main": 0.04180064797401428,
      "step": 52380
    },
    {
      "epoch": 6.130353381699041,
      "grad_norm": 0.3002969026565552,
      "learning_rate": 4.335065169144958e-06,
      "loss": 0.0488,
      "step": 52390
    },
    {
      "epoch": 6.130353381699041,
      "loss_main": 0.0515369288623333,
      "step": 52390
    },
    {
      "epoch": 6.1315235197753335,
      "grad_norm": 0.21911492943763733,
      "learning_rate": 4.323848065185593e-06,
      "loss": 0.0528,
      "step": 52400
    },
    {
      "epoch": 6.1315235197753335,
      "loss_main": 0.050526585429906845,
      "step": 52400
    },
    {
      "epoch": 6.132693657851626,
      "grad_norm": 0.1595878154039383,
      "learning_rate": 4.31264483643517e-06,
      "loss": 0.045,
      "step": 52410
    },
    {
      "epoch": 6.132693657851626,
      "loss_main": 0.06339741498231888,
      "step": 52410
    },
    {
      "epoch": 6.13386379592792,
      "grad_norm": 0.1838575005531311,
      "learning_rate": 4.301455486296946e-06,
      "loss": 0.0497,
      "step": 52420
    },
    {
      "epoch": 6.13386379592792,
      "loss_main": 0.0448714978992939,
      "step": 52420
    },
    {
      "epoch": 6.135033934004213,
      "grad_norm": 0.2243855744600296,
      "learning_rate": 4.290280018169935e-06,
      "loss": 0.0425,
      "step": 52430
    },
    {
      "epoch": 6.135033934004213,
      "loss_main": 0.054529059678316116,
      "step": 52430
    },
    {
      "epoch": 6.136204072080505,
      "grad_norm": 0.2258564978837967,
      "learning_rate": 4.27911843544897e-06,
      "loss": 0.0458,
      "step": 52440
    },
    {
      "epoch": 6.136204072080505,
      "loss_main": 0.04261919483542442,
      "step": 52440
    },
    {
      "epoch": 6.137374210156798,
      "grad_norm": 0.24523033201694489,
      "learning_rate": 4.2679707415246294e-06,
      "loss": 0.0466,
      "step": 52450
    },
    {
      "epoch": 6.137374210156798,
      "loss_main": 0.02598731778562069,
      "step": 52450
    },
    {
      "epoch": 6.138544348233092,
      "grad_norm": 0.36225998401641846,
      "learning_rate": 4.256836939783299e-06,
      "loss": 0.0496,
      "step": 52460
    },
    {
      "epoch": 6.138544348233092,
      "loss_main": 0.02867158316075802,
      "step": 52460
    },
    {
      "epoch": 6.139714486309384,
      "grad_norm": 0.20160214602947235,
      "learning_rate": 4.245717033607127e-06,
      "loss": 0.0411,
      "step": 52470
    },
    {
      "epoch": 6.139714486309384,
      "loss_main": 0.02556750178337097,
      "step": 52470
    },
    {
      "epoch": 6.140884624385677,
      "grad_norm": 0.2488873302936554,
      "learning_rate": 4.234611026374035e-06,
      "loss": 0.0392,
      "step": 52480
    },
    {
      "epoch": 6.140884624385677,
      "loss_main": 0.028368676081299782,
      "step": 52480
    },
    {
      "epoch": 6.142054762461971,
      "grad_norm": 0.27740946412086487,
      "learning_rate": 4.2235189214577694e-06,
      "loss": 0.0406,
      "step": 52490
    },
    {
      "epoch": 6.142054762461971,
      "loss_main": 0.05163593962788582,
      "step": 52490
    },
    {
      "epoch": 6.1432249005382635,
      "grad_norm": 0.218934565782547,
      "learning_rate": 4.212440722227779e-06,
      "loss": 0.0458,
      "step": 52500
    },
    {
      "epoch": 6.1432249005382635,
      "loss_main": 0.035834647715091705,
      "step": 52500
    },
    {
      "epoch": 6.144395038614556,
      "grad_norm": 0.2723400294780731,
      "learning_rate": 4.201376432049364e-06,
      "loss": 0.0428,
      "step": 52510
    },
    {
      "epoch": 6.144395038614556,
      "loss_main": 0.05123745650053024,
      "step": 52510
    },
    {
      "epoch": 6.14556517669085,
      "grad_norm": 0.23884718120098114,
      "learning_rate": 4.1903260542835275e-06,
      "loss": 0.0424,
      "step": 52520
    },
    {
      "epoch": 6.14556517669085,
      "loss_main": 0.04788847640156746,
      "step": 52520
    },
    {
      "epoch": 6.146735314767143,
      "grad_norm": 0.2532615661621094,
      "learning_rate": 4.1792895922871114e-06,
      "loss": 0.0455,
      "step": 52530
    },
    {
      "epoch": 6.146735314767143,
      "loss_main": 0.037858325988054276,
      "step": 52530
    },
    {
      "epoch": 6.147905452843435,
      "grad_norm": 0.2715631127357483,
      "learning_rate": 4.168267049412694e-06,
      "loss": 0.0393,
      "step": 52540
    },
    {
      "epoch": 6.147905452843435,
      "loss_main": 0.061582405120134354,
      "step": 52540
    },
    {
      "epoch": 6.149075590919728,
      "grad_norm": 0.19537584483623505,
      "learning_rate": 4.157258429008626e-06,
      "loss": 0.0413,
      "step": 52550
    },
    {
      "epoch": 6.149075590919728,
      "loss_main": 0.046710073947906494,
      "step": 52550
    },
    {
      "epoch": 6.150245728996022,
      "grad_norm": 0.19958922266960144,
      "learning_rate": 4.146263734419043e-06,
      "loss": 0.0428,
      "step": 52560
    },
    {
      "epoch": 6.150245728996022,
      "loss_main": 0.026683146134018898,
      "step": 52560
    },
    {
      "epoch": 6.1514158670723145,
      "grad_norm": 0.27398064732551575,
      "learning_rate": 4.135282968983839e-06,
      "loss": 0.0423,
      "step": 52570
    },
    {
      "epoch": 6.1514158670723145,
      "loss_main": 0.04275653138756752,
      "step": 52570
    },
    {
      "epoch": 6.152586005148607,
      "grad_norm": 0.26515793800354004,
      "learning_rate": 4.124316136038675e-06,
      "loss": 0.0465,
      "step": 52580
    },
    {
      "epoch": 6.152586005148607,
      "loss_main": 0.04811122640967369,
      "step": 52580
    },
    {
      "epoch": 6.153756143224901,
      "grad_norm": 0.21650774776935577,
      "learning_rate": 4.1133632389149965e-06,
      "loss": 0.0425,
      "step": 52590
    },
    {
      "epoch": 6.153756143224901,
      "loss_main": 0.05433645471930504,
      "step": 52590
    },
    {
      "epoch": 6.154926281301194,
      "grad_norm": 0.3466097116470337,
      "learning_rate": 4.102424280939998e-06,
      "loss": 0.039,
      "step": 52600
    },
    {
      "epoch": 6.154926281301194,
      "loss_main": 0.04702618718147278,
      "step": 52600
    },
    {
      "epoch": 6.156096419377486,
      "grad_norm": 0.1690015196800232,
      "learning_rate": 4.091499265436649e-06,
      "loss": 0.0434,
      "step": 52610
    },
    {
      "epoch": 6.156096419377486,
      "loss_main": 0.03021504171192646,
      "step": 52610
    },
    {
      "epoch": 6.15726655745378,
      "grad_norm": 0.21736294031143188,
      "learning_rate": 4.080588195723684e-06,
      "loss": 0.0473,
      "step": 52620
    },
    {
      "epoch": 6.15726655745378,
      "loss_main": 0.0469631552696228,
      "step": 52620
    },
    {
      "epoch": 6.158436695530073,
      "grad_norm": 0.21243298053741455,
      "learning_rate": 4.069691075115578e-06,
      "loss": 0.0434,
      "step": 52630
    },
    {
      "epoch": 6.158436695530073,
      "loss_main": 0.051727864891290665,
      "step": 52630
    },
    {
      "epoch": 6.159606833606365,
      "grad_norm": 0.18280810117721558,
      "learning_rate": 4.0588079069226235e-06,
      "loss": 0.036,
      "step": 52640
    },
    {
      "epoch": 6.159606833606365,
      "loss_main": 0.05071748420596123,
      "step": 52640
    },
    {
      "epoch": 6.160776971682658,
      "grad_norm": 0.16237306594848633,
      "learning_rate": 4.0479386944508034e-06,
      "loss": 0.0484,
      "step": 52650
    },
    {
      "epoch": 6.160776971682658,
      "loss_main": 0.07087955623865128,
      "step": 52650
    },
    {
      "epoch": 6.161947109758952,
      "grad_norm": 0.17015209794044495,
      "learning_rate": 4.037083441001932e-06,
      "loss": 0.053,
      "step": 52660
    },
    {
      "epoch": 6.161947109758952,
      "loss_main": 0.04173322394490242,
      "step": 52660
    },
    {
      "epoch": 6.1631172478352445,
      "grad_norm": 0.20510084927082062,
      "learning_rate": 4.026242149873516e-06,
      "loss": 0.0477,
      "step": 52670
    },
    {
      "epoch": 6.1631172478352445,
      "loss_main": 0.04435228183865547,
      "step": 52670
    },
    {
      "epoch": 6.164287385911537,
      "grad_norm": 0.17932114005088806,
      "learning_rate": 4.015414824358871e-06,
      "loss": 0.042,
      "step": 52680
    },
    {
      "epoch": 6.164287385911537,
      "loss_main": 0.04672173038125038,
      "step": 52680
    },
    {
      "epoch": 6.165457523987831,
      "grad_norm": 0.23710688948631287,
      "learning_rate": 4.004601467747054e-06,
      "loss": 0.0473,
      "step": 52690
    },
    {
      "epoch": 6.165457523987831,
      "loss_main": 0.048629965633153915,
      "step": 52690
    },
    {
      "epoch": 6.166627662064124,
      "grad_norm": 0.2700585722923279,
      "learning_rate": 3.993802083322873e-06,
      "loss": 0.0482,
      "step": 52700
    },
    {
      "epoch": 6.166627662064124,
      "loss_main": 0.057390980422496796,
      "step": 52700
    },
    {
      "epoch": 6.167797800140416,
      "grad_norm": 0.2299315631389618,
      "learning_rate": 3.9830166743668906e-06,
      "loss": 0.0442,
      "step": 52710
    },
    {
      "epoch": 6.167797800140416,
      "loss_main": 0.06039519980549812,
      "step": 52710
    },
    {
      "epoch": 6.16896793821671,
      "grad_norm": 0.1495126187801361,
      "learning_rate": 3.9722452441554425e-06,
      "loss": 0.0456,
      "step": 52720
    },
    {
      "epoch": 6.16896793821671,
      "loss_main": 0.02388121746480465,
      "step": 52720
    },
    {
      "epoch": 6.170138076293003,
      "grad_norm": 0.19949856400489807,
      "learning_rate": 3.961487795960584e-06,
      "loss": 0.0498,
      "step": 52730
    },
    {
      "epoch": 6.170138076293003,
      "loss_main": 0.09102310240268707,
      "step": 52730
    },
    {
      "epoch": 6.171308214369295,
      "grad_norm": 0.23175588250160217,
      "learning_rate": 3.95074433305016e-06,
      "loss": 0.0472,
      "step": 52740
    },
    {
      "epoch": 6.171308214369295,
      "loss_main": 0.06755249202251434,
      "step": 52740
    },
    {
      "epoch": 6.172478352445588,
      "grad_norm": 0.21325473487377167,
      "learning_rate": 3.940014858687752e-06,
      "loss": 0.0516,
      "step": 52750
    },
    {
      "epoch": 6.172478352445588,
      "loss_main": 0.03999372944235802,
      "step": 52750
    },
    {
      "epoch": 6.173648490521882,
      "grad_norm": 0.28504708409309387,
      "learning_rate": 3.929299376132689e-06,
      "loss": 0.0498,
      "step": 52760
    },
    {
      "epoch": 6.173648490521882,
      "loss_main": 0.02781979739665985,
      "step": 52760
    },
    {
      "epoch": 6.1748186285981745,
      "grad_norm": 0.2055540680885315,
      "learning_rate": 3.918597888640047e-06,
      "loss": 0.0373,
      "step": 52770
    },
    {
      "epoch": 6.1748186285981745,
      "loss_main": 0.032394472509622574,
      "step": 52770
    },
    {
      "epoch": 6.175988766674467,
      "grad_norm": 0.22815419733524323,
      "learning_rate": 3.907910399460657e-06,
      "loss": 0.0392,
      "step": 52780
    },
    {
      "epoch": 6.175988766674467,
      "loss_main": 0.05537427216768265,
      "step": 52780
    },
    {
      "epoch": 6.177158904750761,
      "grad_norm": 0.25081712007522583,
      "learning_rate": 3.8972369118410956e-06,
      "loss": 0.0427,
      "step": 52790
    },
    {
      "epoch": 6.177158904750761,
      "loss_main": 0.036401957273483276,
      "step": 52790
    },
    {
      "epoch": 6.178329042827054,
      "grad_norm": 0.2148778885602951,
      "learning_rate": 3.88657742902368e-06,
      "loss": 0.0441,
      "step": 52800
    },
    {
      "epoch": 6.178329042827054,
      "loss_main": 0.02573436126112938,
      "step": 52800
    },
    {
      "epoch": 6.179499180903346,
      "grad_norm": 0.1693536788225174,
      "learning_rate": 3.875931954246504e-06,
      "loss": 0.0413,
      "step": 52810
    },
    {
      "epoch": 6.179499180903346,
      "loss_main": 0.06272026151418686,
      "step": 52810
    },
    {
      "epoch": 6.18066931897964,
      "grad_norm": 0.21982523798942566,
      "learning_rate": 3.865300490743351e-06,
      "loss": 0.0461,
      "step": 52820
    },
    {
      "epoch": 6.18066931897964,
      "loss_main": 0.04096120595932007,
      "step": 52820
    },
    {
      "epoch": 6.181839457055933,
      "grad_norm": 0.21416428685188293,
      "learning_rate": 3.854683041743806e-06,
      "loss": 0.0437,
      "step": 52830
    },
    {
      "epoch": 6.181839457055933,
      "loss_main": 0.04165235161781311,
      "step": 52830
    },
    {
      "epoch": 6.1830095951322255,
      "grad_norm": 0.21470555663108826,
      "learning_rate": 3.844079610473139e-06,
      "loss": 0.0383,
      "step": 52840
    },
    {
      "epoch": 6.1830095951322255,
      "loss_main": 0.03720690682530403,
      "step": 52840
    },
    {
      "epoch": 6.184179733208518,
      "grad_norm": 0.24322552978992462,
      "learning_rate": 3.833490200152423e-06,
      "loss": 0.0519,
      "step": 52850
    },
    {
      "epoch": 6.184179733208518,
      "loss_main": 0.04472627118229866,
      "step": 52850
    },
    {
      "epoch": 6.185349871284812,
      "grad_norm": 0.2181672900915146,
      "learning_rate": 3.822914813998424e-06,
      "loss": 0.0421,
      "step": 52860
    },
    {
      "epoch": 6.185349871284812,
      "loss_main": 0.03294842690229416,
      "step": 52860
    },
    {
      "epoch": 6.186520009361105,
      "grad_norm": 0.20128555595874786,
      "learning_rate": 3.812353455223666e-06,
      "loss": 0.0386,
      "step": 52870
    },
    {
      "epoch": 6.186520009361105,
      "loss_main": 0.06685830652713776,
      "step": 52870
    },
    {
      "epoch": 6.187690147437397,
      "grad_norm": 0.23179103434085846,
      "learning_rate": 3.8018061270364225e-06,
      "loss": 0.0499,
      "step": 52880
    },
    {
      "epoch": 6.187690147437397,
      "loss_main": 0.03975444287061691,
      "step": 52880
    },
    {
      "epoch": 6.188860285513691,
      "grad_norm": 0.22102734446525574,
      "learning_rate": 3.7912728326406688e-06,
      "loss": 0.0472,
      "step": 52890
    },
    {
      "epoch": 6.188860285513691,
      "loss_main": 0.03962910175323486,
      "step": 52890
    },
    {
      "epoch": 6.190030423589984,
      "grad_norm": 0.19846484065055847,
      "learning_rate": 3.7807535752361732e-06,
      "loss": 0.0379,
      "step": 52900
    },
    {
      "epoch": 6.190030423589984,
      "loss_main": 0.06029490754008293,
      "step": 52900
    },
    {
      "epoch": 6.191200561666276,
      "grad_norm": 0.18926911056041718,
      "learning_rate": 3.7702483580183855e-06,
      "loss": 0.044,
      "step": 52910
    },
    {
      "epoch": 6.191200561666276,
      "loss_main": 0.03633024916052818,
      "step": 52910
    },
    {
      "epoch": 6.19237069974257,
      "grad_norm": 0.23927298188209534,
      "learning_rate": 3.759757184178525e-06,
      "loss": 0.0477,
      "step": 52920
    },
    {
      "epoch": 6.19237069974257,
      "loss_main": 0.030072305351495743,
      "step": 52920
    },
    {
      "epoch": 6.193540837818863,
      "grad_norm": 0.24495096504688263,
      "learning_rate": 3.7492800569035312e-06,
      "loss": 0.0378,
      "step": 52930
    },
    {
      "epoch": 6.193540837818863,
      "loss_main": 0.0578276701271534,
      "step": 52930
    },
    {
      "epoch": 6.1947109758951555,
      "grad_norm": 0.18021827936172485,
      "learning_rate": 3.7388169793760754e-06,
      "loss": 0.045,
      "step": 52940
    },
    {
      "epoch": 6.1947109758951555,
      "loss_main": 0.024853812530636787,
      "step": 52940
    },
    {
      "epoch": 6.195881113971448,
      "grad_norm": 0.29104042053222656,
      "learning_rate": 3.728367954774553e-06,
      "loss": 0.0446,
      "step": 52950
    },
    {
      "epoch": 6.195881113971448,
      "loss_main": 0.045469533652067184,
      "step": 52950
    },
    {
      "epoch": 6.197051252047742,
      "grad_norm": 0.22386372089385986,
      "learning_rate": 3.7179329862731317e-06,
      "loss": 0.0444,
      "step": 52960
    },
    {
      "epoch": 6.197051252047742,
      "loss_main": 0.03319167718291283,
      "step": 52960
    },
    {
      "epoch": 6.198221390124035,
      "grad_norm": 0.25969889760017395,
      "learning_rate": 3.707512077041647e-06,
      "loss": 0.0454,
      "step": 52970
    },
    {
      "epoch": 6.198221390124035,
      "loss_main": 0.03413673862814903,
      "step": 52970
    },
    {
      "epoch": 6.199391528200327,
      "grad_norm": 0.20358245074748993,
      "learning_rate": 3.6971052302457288e-06,
      "loss": 0.0428,
      "step": 52980
    },
    {
      "epoch": 6.199391528200327,
      "loss_main": 0.046343713998794556,
      "step": 52980
    },
    {
      "epoch": 6.200561666276621,
      "grad_norm": 0.23583640158176422,
      "learning_rate": 3.6867124490466698e-06,
      "loss": 0.0588,
      "step": 52990
    },
    {
      "epoch": 6.200561666276621,
      "loss_main": 0.059812337160110474,
      "step": 52990
    },
    {
      "epoch": 6.201731804352914,
      "grad_norm": 0.17087484896183014,
      "learning_rate": 3.6763337366015393e-06,
      "loss": 0.0472,
      "step": 53000
    },
    {
      "epoch": 6.201731804352914,
      "loss_main": 0.036012474447488785,
      "step": 53000
    },
    {
      "epoch": 6.2029019424292064,
      "grad_norm": 0.26087892055511475,
      "learning_rate": 3.665969096063121e-06,
      "loss": 0.0412,
      "step": 53010
    },
    {
      "epoch": 6.2029019424292064,
      "loss_main": 0.0488366074860096,
      "step": 53010
    },
    {
      "epoch": 6.2040720805055,
      "grad_norm": 0.23846875131130219,
      "learning_rate": 3.6556185305799074e-06,
      "loss": 0.0448,
      "step": 53020
    },
    {
      "epoch": 6.2040720805055,
      "loss_main": 0.028904275968670845,
      "step": 53020
    },
    {
      "epoch": 6.205242218581793,
      "grad_norm": 0.2526737153530121,
      "learning_rate": 3.645282043296133e-06,
      "loss": 0.0509,
      "step": 53030
    },
    {
      "epoch": 6.205242218581793,
      "loss_main": 0.03839835152029991,
      "step": 53030
    },
    {
      "epoch": 6.2064123566580855,
      "grad_norm": 0.2109609991312027,
      "learning_rate": 3.6349596373517427e-06,
      "loss": 0.0498,
      "step": 53040
    },
    {
      "epoch": 6.2064123566580855,
      "loss_main": 0.046688228845596313,
      "step": 53040
    },
    {
      "epoch": 6.207582494734378,
      "grad_norm": 0.2705515921115875,
      "learning_rate": 3.6246513158824215e-06,
      "loss": 0.0396,
      "step": 53050
    },
    {
      "epoch": 6.207582494734378,
      "loss_main": 0.046237725764513016,
      "step": 53050
    },
    {
      "epoch": 6.208752632810672,
      "grad_norm": 0.22292833030223846,
      "learning_rate": 3.6143570820195593e-06,
      "loss": 0.0368,
      "step": 53060
    },
    {
      "epoch": 6.208752632810672,
      "loss_main": 0.031294628977775574,
      "step": 53060
    },
    {
      "epoch": 6.209922770886965,
      "grad_norm": 0.2444227784872055,
      "learning_rate": 3.6040769388902773e-06,
      "loss": 0.0464,
      "step": 53070
    },
    {
      "epoch": 6.209922770886965,
      "loss_main": 0.034829918295145035,
      "step": 53070
    },
    {
      "epoch": 6.211092908963257,
      "grad_norm": 0.2041410654783249,
      "learning_rate": 3.593810889617405e-06,
      "loss": 0.0434,
      "step": 53080
    },
    {
      "epoch": 6.211092908963257,
      "loss_main": 0.032901834696531296,
      "step": 53080
    },
    {
      "epoch": 6.212263047039551,
      "grad_norm": 0.2573287785053253,
      "learning_rate": 3.5835589373194978e-06,
      "loss": 0.0429,
      "step": 53090
    },
    {
      "epoch": 6.212263047039551,
      "loss_main": 0.046777818351984024,
      "step": 53090
    },
    {
      "epoch": 6.213433185115844,
      "grad_norm": 0.23312921822071075,
      "learning_rate": 3.5733210851108257e-06,
      "loss": 0.0473,
      "step": 53100
    },
    {
      "epoch": 6.213433185115844,
      "loss_main": 0.03607473894953728,
      "step": 53100
    },
    {
      "epoch": 6.2146033231921365,
      "grad_norm": 0.21956372261047363,
      "learning_rate": 3.5630973361014008e-06,
      "loss": 0.0396,
      "step": 53110
    },
    {
      "epoch": 6.2146033231921365,
      "loss_main": 0.027299955487251282,
      "step": 53110
    },
    {
      "epoch": 6.21577346126843,
      "grad_norm": 0.2696469724178314,
      "learning_rate": 3.552887693396889e-06,
      "loss": 0.0424,
      "step": 53120
    },
    {
      "epoch": 6.21577346126843,
      "loss_main": 0.06318852305412292,
      "step": 53120
    },
    {
      "epoch": 6.216943599344723,
      "grad_norm": 0.1780545711517334,
      "learning_rate": 3.542692160098754e-06,
      "loss": 0.0441,
      "step": 53130
    },
    {
      "epoch": 6.216943599344723,
      "loss_main": 0.03938576951622963,
      "step": 53130
    },
    {
      "epoch": 6.218113737421016,
      "grad_norm": 0.1951216757297516,
      "learning_rate": 3.5325107393040846e-06,
      "loss": 0.0467,
      "step": 53140
    },
    {
      "epoch": 6.218113737421016,
      "loss_main": 0.0217000562697649,
      "step": 53140
    },
    {
      "epoch": 6.219283875497308,
      "grad_norm": 0.21400593221187592,
      "learning_rate": 3.522343434105757e-06,
      "loss": 0.0499,
      "step": 53150
    },
    {
      "epoch": 6.219283875497308,
      "loss_main": 0.033966366201639175,
      "step": 53150
    },
    {
      "epoch": 6.220454013573602,
      "grad_norm": 0.20059320330619812,
      "learning_rate": 3.512190247592323e-06,
      "loss": 0.043,
      "step": 53160
    },
    {
      "epoch": 6.220454013573602,
      "loss_main": 0.053873997181653976,
      "step": 53160
    },
    {
      "epoch": 6.221624151649895,
      "grad_norm": 0.27865877747535706,
      "learning_rate": 3.502051182848054e-06,
      "loss": 0.0475,
      "step": 53170
    },
    {
      "epoch": 6.221624151649895,
      "loss_main": 0.027569234371185303,
      "step": 53170
    },
    {
      "epoch": 6.222794289726187,
      "grad_norm": 0.224243625998497,
      "learning_rate": 3.4919262429529308e-06,
      "loss": 0.0417,
      "step": 53180
    },
    {
      "epoch": 6.222794289726187,
      "loss_main": 0.03902298957109451,
      "step": 53180
    },
    {
      "epoch": 6.223964427802481,
      "grad_norm": 0.3009334206581116,
      "learning_rate": 3.4818154309826325e-06,
      "loss": 0.0438,
      "step": 53190
    },
    {
      "epoch": 6.223964427802481,
      "loss_main": 0.03820493072271347,
      "step": 53190
    },
    {
      "epoch": 6.225134565878774,
      "grad_norm": 0.3011516034603119,
      "learning_rate": 3.4717187500085734e-06,
      "loss": 0.0523,
      "step": 53200
    },
    {
      "epoch": 6.225134565878774,
      "loss_main": 0.043500203639268875,
      "step": 53200
    },
    {
      "epoch": 6.2263047039550665,
      "grad_norm": 0.19187220931053162,
      "learning_rate": 3.46163620309784e-06,
      "loss": 0.0439,
      "step": 53210
    },
    {
      "epoch": 6.2263047039550665,
      "loss_main": 0.05350444093346596,
      "step": 53210
    },
    {
      "epoch": 6.227474842031359,
      "grad_norm": 0.1895330250263214,
      "learning_rate": 3.4515677933132595e-06,
      "loss": 0.0464,
      "step": 53220
    },
    {
      "epoch": 6.227474842031359,
      "loss_main": 0.019502969458699226,
      "step": 53220
    },
    {
      "epoch": 6.228644980107653,
      "grad_norm": 0.28809916973114014,
      "learning_rate": 3.4415135237133466e-06,
      "loss": 0.0389,
      "step": 53230
    },
    {
      "epoch": 6.228644980107653,
      "loss_main": 0.045585863292217255,
      "step": 53230
    },
    {
      "epoch": 6.229815118183946,
      "grad_norm": 0.26379895210266113,
      "learning_rate": 3.4314733973523196e-06,
      "loss": 0.0436,
      "step": 53240
    },
    {
      "epoch": 6.229815118183946,
      "loss_main": 0.020535562187433243,
      "step": 53240
    },
    {
      "epoch": 6.230985256260238,
      "grad_norm": 0.25877875089645386,
      "learning_rate": 3.4214474172800993e-06,
      "loss": 0.0422,
      "step": 53250
    },
    {
      "epoch": 6.230985256260238,
      "loss_main": 0.03678445890545845,
      "step": 53250
    },
    {
      "epoch": 6.232155394336532,
      "grad_norm": 0.2165629118680954,
      "learning_rate": 3.411435586542322e-06,
      "loss": 0.051,
      "step": 53260
    },
    {
      "epoch": 6.232155394336532,
      "loss_main": 0.05738157033920288,
      "step": 53260
    },
    {
      "epoch": 6.233325532412825,
      "grad_norm": 0.19785448908805847,
      "learning_rate": 3.4014379081802995e-06,
      "loss": 0.0431,
      "step": 53270
    },
    {
      "epoch": 6.233325532412825,
      "loss_main": 0.02367768995463848,
      "step": 53270
    },
    {
      "epoch": 6.2344956704891175,
      "grad_norm": 0.20247477293014526,
      "learning_rate": 3.391454385231102e-06,
      "loss": 0.0611,
      "step": 53280
    },
    {
      "epoch": 6.2344956704891175,
      "loss_main": 0.05241405963897705,
      "step": 53280
    },
    {
      "epoch": 6.235665808565411,
      "grad_norm": 0.22366076707839966,
      "learning_rate": 3.3814850207274095e-06,
      "loss": 0.0483,
      "step": 53290
    },
    {
      "epoch": 6.235665808565411,
      "loss_main": 0.03410184383392334,
      "step": 53290
    },
    {
      "epoch": 6.236835946641704,
      "grad_norm": 0.21130341291427612,
      "learning_rate": 3.371529817697694e-06,
      "loss": 0.0486,
      "step": 53300
    },
    {
      "epoch": 6.236835946641704,
      "loss_main": 0.04209451004862785,
      "step": 53300
    },
    {
      "epoch": 6.238006084717997,
      "grad_norm": 0.2860824763774872,
      "learning_rate": 3.3615887791660585e-06,
      "loss": 0.0413,
      "step": 53310
    },
    {
      "epoch": 6.238006084717997,
      "loss_main": 0.04387864097952843,
      "step": 53310
    },
    {
      "epoch": 6.23917622279429,
      "grad_norm": 0.21591337025165558,
      "learning_rate": 3.3516619081523426e-06,
      "loss": 0.0446,
      "step": 53320
    },
    {
      "epoch": 6.23917622279429,
      "loss_main": 0.0381181538105011,
      "step": 53320
    },
    {
      "epoch": 6.240346360870583,
      "grad_norm": 0.17364540696144104,
      "learning_rate": 3.3417492076720567e-06,
      "loss": 0.0412,
      "step": 53330
    },
    {
      "epoch": 6.240346360870583,
      "loss_main": 0.02713603712618351,
      "step": 53330
    },
    {
      "epoch": 6.241516498946876,
      "grad_norm": 0.21517018973827362,
      "learning_rate": 3.3318506807364147e-06,
      "loss": 0.0429,
      "step": 53340
    },
    {
      "epoch": 6.241516498946876,
      "loss_main": 0.03274068608880043,
      "step": 53340
    },
    {
      "epoch": 6.242686637023168,
      "grad_norm": 0.19219817221164703,
      "learning_rate": 3.3219663303523553e-06,
      "loss": 0.0372,
      "step": 53350
    },
    {
      "epoch": 6.242686637023168,
      "loss_main": 0.04831269755959511,
      "step": 53350
    },
    {
      "epoch": 6.243856775099462,
      "grad_norm": 0.23484759032726288,
      "learning_rate": 3.3120961595224374e-06,
      "loss": 0.0494,
      "step": 53360
    },
    {
      "epoch": 6.243856775099462,
      "loss_main": 0.04519229382276535,
      "step": 53360
    },
    {
      "epoch": 6.245026913175755,
      "grad_norm": 0.22154998779296875,
      "learning_rate": 3.302240171245002e-06,
      "loss": 0.0415,
      "step": 53370
    },
    {
      "epoch": 6.245026913175755,
      "loss_main": 0.05220368877053261,
      "step": 53370
    },
    {
      "epoch": 6.2461970512520475,
      "grad_norm": 0.23862019181251526,
      "learning_rate": 3.29239836851401e-06,
      "loss": 0.0448,
      "step": 53380
    },
    {
      "epoch": 6.2461970512520475,
      "loss_main": 0.033404745161533356,
      "step": 53380
    },
    {
      "epoch": 6.247367189328341,
      "grad_norm": 0.31920963525772095,
      "learning_rate": 3.2825707543191588e-06,
      "loss": 0.0459,
      "step": 53390
    },
    {
      "epoch": 6.247367189328341,
      "loss_main": 0.04005056247115135,
      "step": 53390
    },
    {
      "epoch": 6.248537327404634,
      "grad_norm": 0.1785118728876114,
      "learning_rate": 3.272757331645804e-06,
      "loss": 0.0455,
      "step": 53400
    },
    {
      "epoch": 6.248537327404634,
      "loss_main": 0.05797478184103966,
      "step": 53400
    },
    {
      "epoch": 6.249707465480927,
      "grad_norm": 0.2742840349674225,
      "learning_rate": 3.2629581034750166e-06,
      "loss": 0.0463,
      "step": 53410
    },
    {
      "epoch": 6.249707465480927,
      "loss_main": 0.03570176288485527,
      "step": 53410
    },
    {
      "epoch": 6.250877603557219,
      "grad_norm": 0.12582679092884064,
      "learning_rate": 3.2531730727835218e-06,
      "loss": 0.0492,
      "step": 53420
    },
    {
      "epoch": 6.250877603557219,
      "loss_main": 0.04494057968258858,
      "step": 53420
    },
    {
      "epoch": 6.252047741633513,
      "grad_norm": 0.2002810388803482,
      "learning_rate": 3.2434022425437914e-06,
      "loss": 0.0487,
      "step": 53430
    },
    {
      "epoch": 6.252047741633513,
      "loss_main": 0.03869502991437912,
      "step": 53430
    },
    {
      "epoch": 6.253217879709806,
      "grad_norm": 0.21061930060386658,
      "learning_rate": 3.233645615723907e-06,
      "loss": 0.0483,
      "step": 53440
    },
    {
      "epoch": 6.253217879709806,
      "loss_main": 0.045586638152599335,
      "step": 53440
    },
    {
      "epoch": 6.254388017786098,
      "grad_norm": 0.20488670468330383,
      "learning_rate": 3.2239031952876918e-06,
      "loss": 0.0423,
      "step": 53450
    },
    {
      "epoch": 6.254388017786098,
      "loss_main": 0.029568621888756752,
      "step": 53450
    },
    {
      "epoch": 6.255558155862392,
      "grad_norm": 0.15087658166885376,
      "learning_rate": 3.21417498419464e-06,
      "loss": 0.0416,
      "step": 53460
    },
    {
      "epoch": 6.255558155862392,
      "loss_main": 0.04722795635461807,
      "step": 53460
    },
    {
      "epoch": 6.256728293938685,
      "grad_norm": 0.2382214218378067,
      "learning_rate": 3.204460985399921e-06,
      "loss": 0.046,
      "step": 53470
    },
    {
      "epoch": 6.256728293938685,
      "loss_main": 0.032225362956523895,
      "step": 53470
    },
    {
      "epoch": 6.2578984320149775,
      "grad_norm": 0.25135737657546997,
      "learning_rate": 3.1947612018543903e-06,
      "loss": 0.0425,
      "step": 53480
    },
    {
      "epoch": 6.2578984320149775,
      "loss_main": 0.035105615854263306,
      "step": 53480
    },
    {
      "epoch": 6.259068570091271,
      "grad_norm": 0.1749807894229889,
      "learning_rate": 3.1850756365045753e-06,
      "loss": 0.0398,
      "step": 53490
    },
    {
      "epoch": 6.259068570091271,
      "loss_main": 0.040648091584444046,
      "step": 53490
    },
    {
      "epoch": 6.260238708167564,
      "grad_norm": 0.2203650325536728,
      "learning_rate": 3.175404292292722e-06,
      "loss": 0.0494,
      "step": 53500
    },
    {
      "epoch": 6.260238708167564,
      "loss_main": 0.049474913626909256,
      "step": 53500
    },
    {
      "epoch": 6.261408846243857,
      "grad_norm": 0.22302179038524628,
      "learning_rate": 3.1657471721566965e-06,
      "loss": 0.0485,
      "step": 53510
    },
    {
      "epoch": 6.261408846243857,
      "loss_main": 0.02927492745220661,
      "step": 53510
    },
    {
      "epoch": 6.262578984320149,
      "grad_norm": 0.257901668548584,
      "learning_rate": 3.1561042790300977e-06,
      "loss": 0.0403,
      "step": 53520
    },
    {
      "epoch": 6.262578984320149,
      "loss_main": 0.04625052958726883,
      "step": 53520
    },
    {
      "epoch": 6.263749122396443,
      "grad_norm": 0.25992971658706665,
      "learning_rate": 3.1464756158421816e-06,
      "loss": 0.0429,
      "step": 53530
    },
    {
      "epoch": 6.263749122396443,
      "loss_main": 0.04854002967476845,
      "step": 53530
    },
    {
      "epoch": 6.264919260472736,
      "grad_norm": 0.288892924785614,
      "learning_rate": 3.136861185517875e-06,
      "loss": 0.0421,
      "step": 53540
    },
    {
      "epoch": 6.264919260472736,
      "loss_main": 0.039362262934446335,
      "step": 53540
    },
    {
      "epoch": 6.2660893985490285,
      "grad_norm": 0.20831632614135742,
      "learning_rate": 3.127260990977798e-06,
      "loss": 0.0375,
      "step": 53550
    },
    {
      "epoch": 6.2660893985490285,
      "loss_main": 0.025331372395157814,
      "step": 53550
    },
    {
      "epoch": 6.267259536625322,
      "grad_norm": 0.25758859515190125,
      "learning_rate": 3.1176750351382235e-06,
      "loss": 0.0427,
      "step": 53560
    },
    {
      "epoch": 6.267259536625322,
      "loss_main": 0.021673139184713364,
      "step": 53560
    },
    {
      "epoch": 6.268429674701615,
      "grad_norm": 0.24116013944149017,
      "learning_rate": 3.1081033209111153e-06,
      "loss": 0.0508,
      "step": 53570
    },
    {
      "epoch": 6.268429674701615,
      "loss_main": 0.035139162093400955,
      "step": 53570
    },
    {
      "epoch": 6.269599812777908,
      "grad_norm": 0.2821608781814575,
      "learning_rate": 3.0985458512041155e-06,
      "loss": 0.0406,
      "step": 53580
    },
    {
      "epoch": 6.269599812777908,
      "loss_main": 0.031980592757463455,
      "step": 53580
    },
    {
      "epoch": 6.270769950854201,
      "grad_norm": 0.23148977756500244,
      "learning_rate": 3.089002628920512e-06,
      "loss": 0.0421,
      "step": 53590
    },
    {
      "epoch": 6.270769950854201,
      "loss_main": 0.036509186029434204,
      "step": 53590
    },
    {
      "epoch": 6.271940088930494,
      "grad_norm": 0.20406463742256165,
      "learning_rate": 3.079473656959303e-06,
      "loss": 0.0459,
      "step": 53600
    },
    {
      "epoch": 6.271940088930494,
      "loss_main": 0.09223056584596634,
      "step": 53600
    },
    {
      "epoch": 6.273110227006787,
      "grad_norm": 0.16262543201446533,
      "learning_rate": 3.0699589382151393e-06,
      "loss": 0.05,
      "step": 53610
    },
    {
      "epoch": 6.273110227006787,
      "loss_main": 0.042757805436849594,
      "step": 53610
    },
    {
      "epoch": 6.274280365083079,
      "grad_norm": 0.2648504674434662,
      "learning_rate": 3.060458475578326e-06,
      "loss": 0.0413,
      "step": 53620
    },
    {
      "epoch": 6.274280365083079,
      "loss_main": 0.041542381048202515,
      "step": 53620
    },
    {
      "epoch": 6.275450503159373,
      "grad_norm": 0.25700023770332336,
      "learning_rate": 3.0509722719348656e-06,
      "loss": 0.0422,
      "step": 53630
    },
    {
      "epoch": 6.275450503159373,
      "loss_main": 0.046471793204545975,
      "step": 53630
    },
    {
      "epoch": 6.276620641235666,
      "grad_norm": 0.23474609851837158,
      "learning_rate": 3.041500330166408e-06,
      "loss": 0.0451,
      "step": 53640
    },
    {
      "epoch": 6.276620641235666,
      "loss_main": 0.04257998242974281,
      "step": 53640
    },
    {
      "epoch": 6.2777907793119585,
      "grad_norm": 0.20600147545337677,
      "learning_rate": 3.032042653150291e-06,
      "loss": 0.0484,
      "step": 53650
    },
    {
      "epoch": 6.2777907793119585,
      "loss_main": 0.032707832753658295,
      "step": 53650
    },
    {
      "epoch": 6.278960917388252,
      "grad_norm": 0.27360478043556213,
      "learning_rate": 3.0225992437594887e-06,
      "loss": 0.0415,
      "step": 53660
    },
    {
      "epoch": 6.278960917388252,
      "loss_main": 0.039908554404973984,
      "step": 53660
    },
    {
      "epoch": 6.280131055464545,
      "grad_norm": 0.34042593836784363,
      "learning_rate": 3.0131701048626783e-06,
      "loss": 0.0556,
      "step": 53670
    },
    {
      "epoch": 6.280131055464545,
      "loss_main": 0.04848913475871086,
      "step": 53670
    },
    {
      "epoch": 6.281301193540838,
      "grad_norm": 0.20672312378883362,
      "learning_rate": 3.003755239324163e-06,
      "loss": 0.0415,
      "step": 53680
    },
    {
      "epoch": 6.281301193540838,
      "loss_main": 0.04029393941164017,
      "step": 53680
    },
    {
      "epoch": 6.282471331617131,
      "grad_norm": 0.18713977932929993,
      "learning_rate": 2.9943546500039553e-06,
      "loss": 0.0403,
      "step": 53690
    },
    {
      "epoch": 6.282471331617131,
      "loss_main": 0.061440739780664444,
      "step": 53690
    },
    {
      "epoch": 6.283641469693424,
      "grad_norm": 0.24398073554039001,
      "learning_rate": 2.9849683397576877e-06,
      "loss": 0.0503,
      "step": 53700
    },
    {
      "epoch": 6.283641469693424,
      "loss_main": 0.06298502534627914,
      "step": 53700
    },
    {
      "epoch": 6.284811607769717,
      "grad_norm": 0.2858975827693939,
      "learning_rate": 2.975596311436679e-06,
      "loss": 0.0488,
      "step": 53710
    },
    {
      "epoch": 6.284811607769717,
      "loss_main": 0.06957757472991943,
      "step": 53710
    },
    {
      "epoch": 6.285981745846009,
      "grad_norm": 0.22453251481056213,
      "learning_rate": 2.966238567887902e-06,
      "loss": 0.0574,
      "step": 53720
    },
    {
      "epoch": 6.285981745846009,
      "loss_main": 0.04490363597869873,
      "step": 53720
    },
    {
      "epoch": 6.287151883922303,
      "grad_norm": 0.23296698927879333,
      "learning_rate": 2.9568951119539943e-06,
      "loss": 0.0444,
      "step": 53730
    },
    {
      "epoch": 6.287151883922303,
      "loss_main": 0.05683225393295288,
      "step": 53730
    },
    {
      "epoch": 6.288322021998596,
      "grad_norm": 0.2283591777086258,
      "learning_rate": 2.947565946473241e-06,
      "loss": 0.0454,
      "step": 53740
    },
    {
      "epoch": 6.288322021998596,
      "loss_main": 0.06049088016152382,
      "step": 53740
    },
    {
      "epoch": 6.2894921600748885,
      "grad_norm": 0.2021632045507431,
      "learning_rate": 2.9382510742796188e-06,
      "loss": 0.0524,
      "step": 53750
    },
    {
      "epoch": 6.2894921600748885,
      "loss_main": 0.049219898879528046,
      "step": 53750
    },
    {
      "epoch": 6.290662298151182,
      "grad_norm": 0.24670033156871796,
      "learning_rate": 2.9289504982027204e-06,
      "loss": 0.0527,
      "step": 53760
    },
    {
      "epoch": 6.290662298151182,
      "loss_main": 0.038255833089351654,
      "step": 53760
    },
    {
      "epoch": 6.291832436227475,
      "grad_norm": 0.2123156189918518,
      "learning_rate": 2.9196642210678248e-06,
      "loss": 0.0457,
      "step": 53770
    },
    {
      "epoch": 6.291832436227475,
      "loss_main": 0.06256867200136185,
      "step": 53770
    },
    {
      "epoch": 6.293002574303768,
      "grad_norm": 0.2571590840816498,
      "learning_rate": 2.910392245695853e-06,
      "loss": 0.0424,
      "step": 53780
    },
    {
      "epoch": 6.293002574303768,
      "loss_main": 0.03108697198331356,
      "step": 53780
    },
    {
      "epoch": 6.294172712380061,
      "grad_norm": 0.3011976182460785,
      "learning_rate": 2.9011345749033804e-06,
      "loss": 0.0419,
      "step": 53790
    },
    {
      "epoch": 6.294172712380061,
      "loss_main": 0.04234466701745987,
      "step": 53790
    },
    {
      "epoch": 6.295342850456354,
      "grad_norm": 0.19918422400951385,
      "learning_rate": 2.8918912115026677e-06,
      "loss": 0.0443,
      "step": 53800
    },
    {
      "epoch": 6.295342850456354,
      "loss_main": 0.04694244638085365,
      "step": 53800
    },
    {
      "epoch": 6.296512988532647,
      "grad_norm": 0.2762314975261688,
      "learning_rate": 2.8826621583015743e-06,
      "loss": 0.0459,
      "step": 53810
    },
    {
      "epoch": 6.296512988532647,
      "loss_main": 0.03821125999093056,
      "step": 53810
    },
    {
      "epoch": 6.2976831266089395,
      "grad_norm": 0.17861957848072052,
      "learning_rate": 2.8734474181036643e-06,
      "loss": 0.0484,
      "step": 53820
    },
    {
      "epoch": 6.2976831266089395,
      "loss_main": 0.03275259584188461,
      "step": 53820
    },
    {
      "epoch": 6.298853264685233,
      "grad_norm": 0.2617127001285553,
      "learning_rate": 2.864246993708114e-06,
      "loss": 0.0408,
      "step": 53830
    },
    {
      "epoch": 6.298853264685233,
      "loss_main": 0.047317229211330414,
      "step": 53830
    },
    {
      "epoch": 6.300023402761526,
      "grad_norm": 0.18242277204990387,
      "learning_rate": 2.8550608879097884e-06,
      "loss": 0.0409,
      "step": 53840
    },
    {
      "epoch": 6.300023402761526,
      "loss_main": 0.028907213360071182,
      "step": 53840
    },
    {
      "epoch": 6.301193540837819,
      "grad_norm": 0.19242000579833984,
      "learning_rate": 2.845889103499172e-06,
      "loss": 0.0383,
      "step": 53850
    },
    {
      "epoch": 6.301193540837819,
      "loss_main": 0.030527202412486076,
      "step": 53850
    },
    {
      "epoch": 6.302363678914112,
      "grad_norm": 0.21295712888240814,
      "learning_rate": 2.8367316432624138e-06,
      "loss": 0.0378,
      "step": 53860
    },
    {
      "epoch": 6.302363678914112,
      "loss_main": 0.09030784666538239,
      "step": 53860
    },
    {
      "epoch": 6.303533816990405,
      "grad_norm": 0.22811228036880493,
      "learning_rate": 2.8275885099813105e-06,
      "loss": 0.0498,
      "step": 53870
    },
    {
      "epoch": 6.303533816990405,
      "loss_main": 0.06852097064256668,
      "step": 53870
    },
    {
      "epoch": 6.304703955066698,
      "grad_norm": 0.19672271609306335,
      "learning_rate": 2.8184597064332963e-06,
      "loss": 0.0542,
      "step": 53880
    },
    {
      "epoch": 6.304703955066698,
      "loss_main": 0.03721947222948074,
      "step": 53880
    },
    {
      "epoch": 6.305874093142991,
      "grad_norm": 0.22942999005317688,
      "learning_rate": 2.809345235391464e-06,
      "loss": 0.0526,
      "step": 53890
    },
    {
      "epoch": 6.305874093142991,
      "loss_main": 0.0435551293194294,
      "step": 53890
    },
    {
      "epoch": 6.307044231219284,
      "grad_norm": 0.16177447140216827,
      "learning_rate": 2.80024509962456e-06,
      "loss": 0.0345,
      "step": 53900
    },
    {
      "epoch": 6.307044231219284,
      "loss_main": 0.05635413900017738,
      "step": 53900
    },
    {
      "epoch": 6.308214369295577,
      "grad_norm": 0.2147049605846405,
      "learning_rate": 2.7911593018969563e-06,
      "loss": 0.0436,
      "step": 53910
    },
    {
      "epoch": 6.308214369295577,
      "loss_main": 0.05818886682391167,
      "step": 53910
    },
    {
      "epoch": 6.3093845073718695,
      "grad_norm": 0.18334269523620605,
      "learning_rate": 2.7820878449686838e-06,
      "loss": 0.0456,
      "step": 53920
    },
    {
      "epoch": 6.3093845073718695,
      "loss_main": 0.03168725594878197,
      "step": 53920
    },
    {
      "epoch": 6.310554645448163,
      "grad_norm": 0.32896268367767334,
      "learning_rate": 2.7730307315953995e-06,
      "loss": 0.0446,
      "step": 53930
    },
    {
      "epoch": 6.310554645448163,
      "loss_main": 0.05130019411444664,
      "step": 53930
    },
    {
      "epoch": 6.311724783524456,
      "grad_norm": 0.2503424882888794,
      "learning_rate": 2.763987964528425e-06,
      "loss": 0.0422,
      "step": 53940
    },
    {
      "epoch": 6.311724783524456,
      "loss_main": 0.04211539030075073,
      "step": 53940
    },
    {
      "epoch": 6.312894921600749,
      "grad_norm": 0.22714126110076904,
      "learning_rate": 2.754959546514718e-06,
      "loss": 0.0404,
      "step": 53950
    },
    {
      "epoch": 6.312894921600749,
      "loss_main": 0.03929995745420456,
      "step": 53950
    },
    {
      "epoch": 6.314065059677042,
      "grad_norm": 0.18590645492076874,
      "learning_rate": 2.7459454802968576e-06,
      "loss": 0.0441,
      "step": 53960
    },
    {
      "epoch": 6.314065059677042,
      "loss_main": 0.04661569371819496,
      "step": 53960
    },
    {
      "epoch": 6.315235197753335,
      "grad_norm": 0.23960986733436584,
      "learning_rate": 2.7369457686131028e-06,
      "loss": 0.0477,
      "step": 53970
    },
    {
      "epoch": 6.315235197753335,
      "loss_main": 0.03998643159866333,
      "step": 53970
    },
    {
      "epoch": 6.316405335829628,
      "grad_norm": 0.21856310963630676,
      "learning_rate": 2.7279604141973004e-06,
      "loss": 0.0439,
      "step": 53980
    },
    {
      "epoch": 6.316405335829628,
      "loss_main": 0.04812765493988991,
      "step": 53980
    },
    {
      "epoch": 6.3175754739059204,
      "grad_norm": 0.2121303528547287,
      "learning_rate": 2.7189894197789946e-06,
      "loss": 0.0438,
      "step": 53990
    },
    {
      "epoch": 6.3175754739059204,
      "loss_main": 0.04006102308630943,
      "step": 53990
    },
    {
      "epoch": 6.318745611982214,
      "grad_norm": 0.27039799094200134,
      "learning_rate": 2.7100327880833055e-06,
      "loss": 0.042,
      "step": 54000
    },
    {
      "epoch": 6.318745611982214,
      "loss_main": 0.08826946467161179,
      "step": 54000
    },
    {
      "epoch": 6.319915750058507,
      "grad_norm": 0.2015821635723114,
      "learning_rate": 2.70109052183104e-06,
      "loss": 0.0484,
      "step": 54010
    },
    {
      "epoch": 6.319915750058507,
      "loss_main": 0.06308857351541519,
      "step": 54010
    },
    {
      "epoch": 6.3210858881347995,
      "grad_norm": 0.25565052032470703,
      "learning_rate": 2.69216262373862e-06,
      "loss": 0.046,
      "step": 54020
    },
    {
      "epoch": 6.3210858881347995,
      "loss_main": 0.04157716780900955,
      "step": 54020
    },
    {
      "epoch": 6.322256026211093,
      "grad_norm": 0.23095840215682983,
      "learning_rate": 2.6832490965181036e-06,
      "loss": 0.0442,
      "step": 54030
    },
    {
      "epoch": 6.322256026211093,
      "loss_main": 0.08021331578493118,
      "step": 54030
    },
    {
      "epoch": 6.323426164287386,
      "grad_norm": 0.22153396904468536,
      "learning_rate": 2.6743499428771857e-06,
      "loss": 0.0458,
      "step": 54040
    },
    {
      "epoch": 6.323426164287386,
      "loss_main": 0.027434999123215675,
      "step": 54040
    },
    {
      "epoch": 6.324596302363679,
      "grad_norm": 0.2790329158306122,
      "learning_rate": 2.6654651655191875e-06,
      "loss": 0.0448,
      "step": 54050
    },
    {
      "epoch": 6.324596302363679,
      "loss_main": 0.03678930550813675,
      "step": 54050
    },
    {
      "epoch": 6.325766440439972,
      "grad_norm": 0.1703060418367386,
      "learning_rate": 2.6565947671430836e-06,
      "loss": 0.0413,
      "step": 54060
    },
    {
      "epoch": 6.325766440439972,
      "loss_main": 0.04936928302049637,
      "step": 54060
    },
    {
      "epoch": 6.326936578516265,
      "grad_norm": 0.17506879568099976,
      "learning_rate": 2.647738750443457e-06,
      "loss": 0.0378,
      "step": 54070
    },
    {
      "epoch": 6.326936578516265,
      "loss_main": 0.04520053043961525,
      "step": 54070
    },
    {
      "epoch": 6.328106716592558,
      "grad_norm": 0.1909317970275879,
      "learning_rate": 2.6388971181105393e-06,
      "loss": 0.0427,
      "step": 54080
    },
    {
      "epoch": 6.328106716592558,
      "loss_main": 0.0331367664039135,
      "step": 54080
    },
    {
      "epoch": 6.329276854668851,
      "grad_norm": 0.16392332315444946,
      "learning_rate": 2.630069872830171e-06,
      "loss": 0.043,
      "step": 54090
    },
    {
      "epoch": 6.329276854668851,
      "loss_main": 0.02899797447025776,
      "step": 54090
    },
    {
      "epoch": 6.330446992745144,
      "grad_norm": 0.25514400005340576,
      "learning_rate": 2.6212570172838514e-06,
      "loss": 0.0466,
      "step": 54100
    },
    {
      "epoch": 6.330446992745144,
      "loss_main": 0.031143013387918472,
      "step": 54100
    },
    {
      "epoch": 6.331617130821437,
      "grad_norm": 0.22927327454090118,
      "learning_rate": 2.6124585541486778e-06,
      "loss": 0.0477,
      "step": 54110
    },
    {
      "epoch": 6.331617130821437,
      "loss_main": 0.03951322287321091,
      "step": 54110
    },
    {
      "epoch": 6.33278726889773,
      "grad_norm": 0.22649791836738586,
      "learning_rate": 2.6036744860974127e-06,
      "loss": 0.0428,
      "step": 54120
    },
    {
      "epoch": 6.33278726889773,
      "loss_main": 0.06306358426809311,
      "step": 54120
    },
    {
      "epoch": 6.333957406974023,
      "grad_norm": 0.28065943717956543,
      "learning_rate": 2.594904815798399e-06,
      "loss": 0.0507,
      "step": 54130
    },
    {
      "epoch": 6.333957406974023,
      "loss_main": 0.05231240764260292,
      "step": 54130
    },
    {
      "epoch": 6.335127545050316,
      "grad_norm": 0.1873203068971634,
      "learning_rate": 2.586149545915656e-06,
      "loss": 0.0424,
      "step": 54140
    },
    {
      "epoch": 6.335127545050316,
      "loss_main": 0.03689092397689819,
      "step": 54140
    },
    {
      "epoch": 6.336297683126609,
      "grad_norm": 0.2063073366880417,
      "learning_rate": 2.577408679108778e-06,
      "loss": 0.0442,
      "step": 54150
    },
    {
      "epoch": 6.336297683126609,
      "loss_main": 0.03799017146229744,
      "step": 54150
    },
    {
      "epoch": 6.337467821202902,
      "grad_norm": 0.2855510711669922,
      "learning_rate": 2.5686822180330306e-06,
      "loss": 0.0426,
      "step": 54160
    },
    {
      "epoch": 6.337467821202902,
      "loss_main": 0.022587014362215996,
      "step": 54160
    },
    {
      "epoch": 6.338637959279195,
      "grad_norm": 0.2130175530910492,
      "learning_rate": 2.5599701653392703e-06,
      "loss": 0.0417,
      "step": 54170
    },
    {
      "epoch": 6.338637959279195,
      "loss_main": 0.06049450486898422,
      "step": 54170
    },
    {
      "epoch": 6.339808097355488,
      "grad_norm": 0.18265917897224426,
      "learning_rate": 2.551272523673992e-06,
      "loss": 0.0523,
      "step": 54180
    },
    {
      "epoch": 6.339808097355488,
      "loss_main": 0.04376162216067314,
      "step": 54180
    },
    {
      "epoch": 6.3409782354317805,
      "grad_norm": 0.20683026313781738,
      "learning_rate": 2.542589295679315e-06,
      "loss": 0.0452,
      "step": 54190
    },
    {
      "epoch": 6.3409782354317805,
      "loss_main": 0.101638063788414,
      "step": 54190
    },
    {
      "epoch": 6.342148373508074,
      "grad_norm": 0.20755337178707123,
      "learning_rate": 2.5339204839929575e-06,
      "loss": 0.043,
      "step": 54200
    },
    {
      "epoch": 6.342148373508074,
      "loss_main": 0.05825609713792801,
      "step": 54200
    },
    {
      "epoch": 6.343318511584367,
      "grad_norm": 0.2218703180551529,
      "learning_rate": 2.525266091248296e-06,
      "loss": 0.0523,
      "step": 54210
    },
    {
      "epoch": 6.343318511584367,
      "loss_main": 0.04018551856279373,
      "step": 54210
    },
    {
      "epoch": 6.34448864966066,
      "grad_norm": 0.24168576300144196,
      "learning_rate": 2.5166261200743e-06,
      "loss": 0.0461,
      "step": 54220
    },
    {
      "epoch": 6.34448864966066,
      "loss_main": 0.030114945024251938,
      "step": 54220
    },
    {
      "epoch": 6.345658787736953,
      "grad_norm": 0.16826076805591583,
      "learning_rate": 2.5080005730955646e-06,
      "loss": 0.045,
      "step": 54230
    },
    {
      "epoch": 6.345658787736953,
      "loss_main": 0.04841768741607666,
      "step": 54230
    },
    {
      "epoch": 6.346828925813246,
      "grad_norm": 0.24440589547157288,
      "learning_rate": 2.499389452932299e-06,
      "loss": 0.0448,
      "step": 54240
    },
    {
      "epoch": 6.346828925813246,
      "loss_main": 0.046447206288576126,
      "step": 54240
    },
    {
      "epoch": 6.347999063889539,
      "grad_norm": 0.1800905466079712,
      "learning_rate": 2.4907927622003336e-06,
      "loss": 0.0417,
      "step": 54250
    },
    {
      "epoch": 6.347999063889539,
      "loss_main": 0.0647745206952095,
      "step": 54250
    },
    {
      "epoch": 6.349169201965832,
      "grad_norm": 0.2468002587556839,
      "learning_rate": 2.4822105035111177e-06,
      "loss": 0.0473,
      "step": 54260
    },
    {
      "epoch": 6.349169201965832,
      "loss_main": 0.04677889868617058,
      "step": 54260
    },
    {
      "epoch": 6.350339340042125,
      "grad_norm": 0.22000013291835785,
      "learning_rate": 2.4736426794717273e-06,
      "loss": 0.0389,
      "step": 54270
    },
    {
      "epoch": 6.350339340042125,
      "loss_main": 0.034905802458524704,
      "step": 54270
    },
    {
      "epoch": 6.351509478118418,
      "grad_norm": 0.18858827650547028,
      "learning_rate": 2.4650892926848135e-06,
      "loss": 0.0415,
      "step": 54280
    },
    {
      "epoch": 6.351509478118418,
      "loss_main": 0.04784562811255455,
      "step": 54280
    },
    {
      "epoch": 6.352679616194711,
      "grad_norm": 0.20955370366573334,
      "learning_rate": 2.456550345748704e-06,
      "loss": 0.0494,
      "step": 54290
    },
    {
      "epoch": 6.352679616194711,
      "loss_main": 0.05721255764365196,
      "step": 54290
    },
    {
      "epoch": 6.353849754271004,
      "grad_norm": 0.257862389087677,
      "learning_rate": 2.4480258412572733e-06,
      "loss": 0.0418,
      "step": 54300
    },
    {
      "epoch": 6.353849754271004,
      "loss_main": 0.02555721625685692,
      "step": 54300
    },
    {
      "epoch": 6.355019892347297,
      "grad_norm": 0.2856454849243164,
      "learning_rate": 2.4395157818000612e-06,
      "loss": 0.0484,
      "step": 54310
    },
    {
      "epoch": 6.355019892347297,
      "loss_main": 0.027283387258648872,
      "step": 54310
    },
    {
      "epoch": 6.35619003042359,
      "grad_norm": 0.30754899978637695,
      "learning_rate": 2.431020169962189e-06,
      "loss": 0.0461,
      "step": 54320
    },
    {
      "epoch": 6.35619003042359,
      "loss_main": 0.03462537005543709,
      "step": 54320
    },
    {
      "epoch": 6.357360168499883,
      "grad_norm": 0.22248469293117523,
      "learning_rate": 2.422539008324409e-06,
      "loss": 0.038,
      "step": 54330
    },
    {
      "epoch": 6.357360168499883,
      "loss_main": 0.0289777722209692,
      "step": 54330
    },
    {
      "epoch": 6.358530306576176,
      "grad_norm": 0.21092718839645386,
      "learning_rate": 2.414072299463066e-06,
      "loss": 0.0416,
      "step": 54340
    },
    {
      "epoch": 6.358530306576176,
      "loss_main": 0.031044961884617805,
      "step": 54340
    },
    {
      "epoch": 6.359700444652469,
      "grad_norm": 0.2501760721206665,
      "learning_rate": 2.4056200459501186e-06,
      "loss": 0.0536,
      "step": 54350
    },
    {
      "epoch": 6.359700444652469,
      "loss_main": 0.049254246056079865,
      "step": 54350
    },
    {
      "epoch": 6.360870582728762,
      "grad_norm": 0.18963433802127838,
      "learning_rate": 2.397182250353147e-06,
      "loss": 0.0498,
      "step": 54360
    },
    {
      "epoch": 6.360870582728762,
      "loss_main": 0.06489875167608261,
      "step": 54360
    },
    {
      "epoch": 6.362040720805055,
      "grad_norm": 0.2799367308616638,
      "learning_rate": 2.388758915235334e-06,
      "loss": 0.046,
      "step": 54370
    },
    {
      "epoch": 6.362040720805055,
      "loss_main": 0.10369353741407394,
      "step": 54370
    },
    {
      "epoch": 6.363210858881348,
      "grad_norm": 0.2878340184688568,
      "learning_rate": 2.380350043155455e-06,
      "loss": 0.0463,
      "step": 54380
    },
    {
      "epoch": 6.363210858881348,
      "loss_main": 0.03305928781628609,
      "step": 54380
    },
    {
      "epoch": 6.364380996957641,
      "grad_norm": 0.3076952397823334,
      "learning_rate": 2.371955636667911e-06,
      "loss": 0.0436,
      "step": 54390
    },
    {
      "epoch": 6.364380996957641,
      "loss_main": 0.046857066452503204,
      "step": 54390
    },
    {
      "epoch": 6.365551135033934,
      "grad_norm": 0.26969829201698303,
      "learning_rate": 2.3635756983227008e-06,
      "loss": 0.0386,
      "step": 54400
    },
    {
      "epoch": 6.365551135033934,
      "loss_main": 0.05344480276107788,
      "step": 54400
    },
    {
      "epoch": 6.366721273110227,
      "grad_norm": 0.24889017641544342,
      "learning_rate": 2.3552102306654278e-06,
      "loss": 0.0407,
      "step": 54410
    },
    {
      "epoch": 6.366721273110227,
      "loss_main": 0.03156514838337898,
      "step": 54410
    },
    {
      "epoch": 6.36789141118652,
      "grad_norm": 0.23426087200641632,
      "learning_rate": 2.346859236237292e-06,
      "loss": 0.0442,
      "step": 54420
    },
    {
      "epoch": 6.36789141118652,
      "loss_main": 0.05477660521864891,
      "step": 54420
    },
    {
      "epoch": 6.369061549262813,
      "grad_norm": 0.3809016942977905,
      "learning_rate": 2.3385227175751145e-06,
      "loss": 0.0455,
      "step": 54430
    },
    {
      "epoch": 6.369061549262813,
      "loss_main": 0.034241266548633575,
      "step": 54430
    },
    {
      "epoch": 6.370231687339106,
      "grad_norm": 0.2698039710521698,
      "learning_rate": 2.330200677211314e-06,
      "loss": 0.0434,
      "step": 54440
    },
    {
      "epoch": 6.370231687339106,
      "loss_main": 0.04292527213692665,
      "step": 54440
    },
    {
      "epoch": 6.371401825415399,
      "grad_norm": 0.1669672131538391,
      "learning_rate": 2.3218931176738847e-06,
      "loss": 0.0409,
      "step": 54450
    },
    {
      "epoch": 6.371401825415399,
      "loss_main": 0.030276715755462646,
      "step": 54450
    },
    {
      "epoch": 6.372571963491692,
      "grad_norm": 0.17175033688545227,
      "learning_rate": 2.313600041486469e-06,
      "loss": 0.0461,
      "step": 54460
    },
    {
      "epoch": 6.372571963491692,
      "loss_main": 0.0645444467663765,
      "step": 54460
    },
    {
      "epoch": 6.373742101567985,
      "grad_norm": 0.2629293203353882,
      "learning_rate": 2.3053214511682743e-06,
      "loss": 0.0404,
      "step": 54470
    },
    {
      "epoch": 6.373742101567985,
      "loss_main": 0.052408743649721146,
      "step": 54470
    },
    {
      "epoch": 6.374912239644278,
      "grad_norm": 0.2758450508117676,
      "learning_rate": 2.2970573492341163e-06,
      "loss": 0.0456,
      "step": 54480
    },
    {
      "epoch": 6.374912239644278,
      "loss_main": 0.05277002602815628,
      "step": 54480
    },
    {
      "epoch": 6.376082377720571,
      "grad_norm": 0.2876652181148529,
      "learning_rate": 2.288807738194415e-06,
      "loss": 0.0457,
      "step": 54490
    },
    {
      "epoch": 6.376082377720571,
      "loss_main": 0.026922335848212242,
      "step": 54490
    },
    {
      "epoch": 6.377252515796864,
      "grad_norm": 0.4410552978515625,
      "learning_rate": 2.2805726205551768e-06,
      "loss": 0.0414,
      "step": 54500
    },
    {
      "epoch": 6.377252515796864,
      "loss_main": 0.028075721114873886,
      "step": 54500
    },
    {
      "epoch": 6.378422653873157,
      "grad_norm": 0.15565215051174164,
      "learning_rate": 2.272351998818023e-06,
      "loss": 0.044,
      "step": 54510
    },
    {
      "epoch": 6.378422653873157,
      "loss_main": 0.07825566083192825,
      "step": 54510
    },
    {
      "epoch": 6.37959279194945,
      "grad_norm": 0.22717434167861938,
      "learning_rate": 2.2641458754801505e-06,
      "loss": 0.0492,
      "step": 54520
    },
    {
      "epoch": 6.37959279194945,
      "loss_main": 0.03419579192996025,
      "step": 54520
    },
    {
      "epoch": 6.380762930025743,
      "grad_norm": 0.21446314454078674,
      "learning_rate": 2.2559542530343756e-06,
      "loss": 0.0334,
      "step": 54530
    },
    {
      "epoch": 6.380762930025743,
      "loss_main": 0.10962378978729248,
      "step": 54530
    },
    {
      "epoch": 6.381933068102036,
      "grad_norm": 0.16091428697109222,
      "learning_rate": 2.247777133969087e-06,
      "loss": 0.0503,
      "step": 54540
    },
    {
      "epoch": 6.381933068102036,
      "loss_main": 0.07261649519205093,
      "step": 54540
    },
    {
      "epoch": 6.383103206178329,
      "grad_norm": 0.20393016934394836,
      "learning_rate": 2.2396145207682795e-06,
      "loss": 0.0453,
      "step": 54550
    },
    {
      "epoch": 6.383103206178329,
      "loss_main": 0.05633014440536499,
      "step": 54550
    },
    {
      "epoch": 6.3842733442546225,
      "grad_norm": 0.25973957777023315,
      "learning_rate": 2.231466415911543e-06,
      "loss": 0.047,
      "step": 54560
    },
    {
      "epoch": 6.3842733442546225,
      "loss_main": 0.04236936569213867,
      "step": 54560
    },
    {
      "epoch": 6.385443482330915,
      "grad_norm": 0.248301699757576,
      "learning_rate": 2.2233328218740524e-06,
      "loss": 0.05,
      "step": 54570
    },
    {
      "epoch": 6.385443482330915,
      "loss_main": 0.047035831958055496,
      "step": 54570
    },
    {
      "epoch": 6.386613620407208,
      "grad_norm": 0.213282972574234,
      "learning_rate": 2.215213741126576e-06,
      "loss": 0.0443,
      "step": 54580
    },
    {
      "epoch": 6.386613620407208,
      "loss_main": 0.03152577951550484,
      "step": 54580
    },
    {
      "epoch": 6.387783758483501,
      "grad_norm": 0.27699899673461914,
      "learning_rate": 2.2071091761354912e-06,
      "loss": 0.0463,
      "step": 54590
    },
    {
      "epoch": 6.387783758483501,
      "loss_main": 0.03546437621116638,
      "step": 54590
    },
    {
      "epoch": 6.388953896559794,
      "grad_norm": 0.2729591429233551,
      "learning_rate": 2.1990191293627337e-06,
      "loss": 0.0463,
      "step": 54600
    },
    {
      "epoch": 6.388953896559794,
      "loss_main": 0.05333571508526802,
      "step": 54600
    },
    {
      "epoch": 6.390124034636087,
      "grad_norm": 0.2826339900493622,
      "learning_rate": 2.1909436032658548e-06,
      "loss": 0.0452,
      "step": 54610
    },
    {
      "epoch": 6.390124034636087,
      "loss_main": 0.04690219834446907,
      "step": 54610
    },
    {
      "epoch": 6.39129417271238,
      "grad_norm": 0.15130123496055603,
      "learning_rate": 2.1828826002979806e-06,
      "loss": 0.0462,
      "step": 54620
    },
    {
      "epoch": 6.39129417271238,
      "loss_main": 0.02162293531000614,
      "step": 54620
    },
    {
      "epoch": 6.392464310788673,
      "grad_norm": 0.28364071249961853,
      "learning_rate": 2.1748361229078362e-06,
      "loss": 0.0446,
      "step": 54630
    },
    {
      "epoch": 6.392464310788673,
      "loss_main": 0.05600779503583908,
      "step": 54630
    },
    {
      "epoch": 6.393634448864966,
      "grad_norm": 0.24080201983451843,
      "learning_rate": 2.1668041735397327e-06,
      "loss": 0.0427,
      "step": 54640
    },
    {
      "epoch": 6.393634448864966,
      "loss_main": 0.07366667687892914,
      "step": 54640
    },
    {
      "epoch": 6.394804586941259,
      "grad_norm": 0.2930483818054199,
      "learning_rate": 2.1587867546335514e-06,
      "loss": 0.0497,
      "step": 54650
    },
    {
      "epoch": 6.394804586941259,
      "loss_main": 0.05898124724626541,
      "step": 54650
    },
    {
      "epoch": 6.3959747250175525,
      "grad_norm": 0.18012595176696777,
      "learning_rate": 2.1507838686247894e-06,
      "loss": 0.0441,
      "step": 54660
    },
    {
      "epoch": 6.3959747250175525,
      "loss_main": 0.03556017577648163,
      "step": 54660
    },
    {
      "epoch": 6.397144863093845,
      "grad_norm": 0.23072604835033417,
      "learning_rate": 2.1427955179444957e-06,
      "loss": 0.0456,
      "step": 54670
    },
    {
      "epoch": 6.397144863093845,
      "loss_main": 0.05314091220498085,
      "step": 54670
    },
    {
      "epoch": 6.398315001170138,
      "grad_norm": 0.20563068985939026,
      "learning_rate": 2.13482170501933e-06,
      "loss": 0.0471,
      "step": 54680
    },
    {
      "epoch": 6.398315001170138,
      "loss_main": 0.0685545802116394,
      "step": 54680
    },
    {
      "epoch": 6.399485139246431,
      "grad_norm": 0.2110280990600586,
      "learning_rate": 2.1268624322715202e-06,
      "loss": 0.0501,
      "step": 54690
    },
    {
      "epoch": 6.399485139246431,
      "loss_main": 0.05280641093850136,
      "step": 54690
    },
    {
      "epoch": 6.400655277322724,
      "grad_norm": 0.2596302032470703,
      "learning_rate": 2.118917702118889e-06,
      "loss": 0.0508,
      "step": 54700
    },
    {
      "epoch": 6.400655277322724,
      "loss_main": 0.027328968048095703,
      "step": 54700
    },
    {
      "epoch": 6.401825415399017,
      "grad_norm": 0.22759497165679932,
      "learning_rate": 2.1109875169748327e-06,
      "loss": 0.0477,
      "step": 54710
    },
    {
      "epoch": 6.401825415399017,
      "loss_main": 0.04258354380726814,
      "step": 54710
    },
    {
      "epoch": 6.40299555347531,
      "grad_norm": 0.25958937406539917,
      "learning_rate": 2.103071879248336e-06,
      "loss": 0.0475,
      "step": 54720
    },
    {
      "epoch": 6.40299555347531,
      "loss_main": 0.05169430002570152,
      "step": 54720
    },
    {
      "epoch": 6.404165691551603,
      "grad_norm": 0.20255319774150848,
      "learning_rate": 2.0951707913439478e-06,
      "loss": 0.0493,
      "step": 54730
    },
    {
      "epoch": 6.404165691551603,
      "loss_main": 0.027670035138726234,
      "step": 54730
    },
    {
      "epoch": 6.405335829627896,
      "grad_norm": 0.20653554797172546,
      "learning_rate": 2.0872842556618255e-06,
      "loss": 0.0477,
      "step": 54740
    },
    {
      "epoch": 6.405335829627896,
      "loss_main": 0.050488729029893875,
      "step": 54740
    },
    {
      "epoch": 6.406505967704189,
      "grad_norm": 0.26704519987106323,
      "learning_rate": 2.079412274597686e-06,
      "loss": 0.0436,
      "step": 54750
    },
    {
      "epoch": 6.406505967704189,
      "loss_main": 0.01609363593161106,
      "step": 54750
    },
    {
      "epoch": 6.407676105780482,
      "grad_norm": 0.2666967213153839,
      "learning_rate": 2.0715548505428284e-06,
      "loss": 0.0425,
      "step": 54760
    },
    {
      "epoch": 6.407676105780482,
      "loss_main": 0.05458488687872887,
      "step": 54760
    },
    {
      "epoch": 6.408846243856775,
      "grad_norm": 0.21146899461746216,
      "learning_rate": 2.0637119858841258e-06,
      "loss": 0.0434,
      "step": 54770
    },
    {
      "epoch": 6.408846243856775,
      "loss_main": 0.060129012912511826,
      "step": 54770
    },
    {
      "epoch": 6.410016381933068,
      "grad_norm": 0.27064502239227295,
      "learning_rate": 2.055883683004034e-06,
      "loss": 0.0515,
      "step": 54780
    },
    {
      "epoch": 6.410016381933068,
      "loss_main": 0.02375342883169651,
      "step": 54780
    },
    {
      "epoch": 6.411186520009361,
      "grad_norm": 0.19390259683132172,
      "learning_rate": 2.0480699442806006e-06,
      "loss": 0.0473,
      "step": 54790
    },
    {
      "epoch": 6.411186520009361,
      "loss_main": 0.026517251506447792,
      "step": 54790
    },
    {
      "epoch": 6.412356658085654,
      "grad_norm": 0.1632353812456131,
      "learning_rate": 2.0402707720874105e-06,
      "loss": 0.0447,
      "step": 54800
    },
    {
      "epoch": 6.412356658085654,
      "loss_main": 0.04421567916870117,
      "step": 54800
    },
    {
      "epoch": 6.413526796161947,
      "grad_norm": 0.14041288197040558,
      "learning_rate": 2.032486168793668e-06,
      "loss": 0.0441,
      "step": 54810
    },
    {
      "epoch": 6.413526796161947,
      "loss_main": 0.03200644999742508,
      "step": 54810
    },
    {
      "epoch": 6.41469693423824,
      "grad_norm": 0.29117971658706665,
      "learning_rate": 2.024716136764104e-06,
      "loss": 0.0402,
      "step": 54820
    },
    {
      "epoch": 6.41469693423824,
      "loss_main": 0.03193570300936699,
      "step": 54820
    },
    {
      "epoch": 6.4158670723145335,
      "grad_norm": 0.20941852033138275,
      "learning_rate": 2.0169606783590757e-06,
      "loss": 0.0408,
      "step": 54830
    },
    {
      "epoch": 6.4158670723145335,
      "loss_main": 0.03821488097310066,
      "step": 54830
    },
    {
      "epoch": 6.417037210390826,
      "grad_norm": 0.1755857914686203,
      "learning_rate": 2.0092197959344638e-06,
      "loss": 0.0426,
      "step": 54840
    },
    {
      "epoch": 6.417037210390826,
      "loss_main": 0.03462938591837883,
      "step": 54840
    },
    {
      "epoch": 6.418207348467119,
      "grad_norm": 0.2136165052652359,
      "learning_rate": 2.0014934918417606e-06,
      "loss": 0.0453,
      "step": 54850
    },
    {
      "epoch": 6.418207348467119,
      "loss_main": 0.04537283629179001,
      "step": 54850
    },
    {
      "epoch": 6.419377486543413,
      "grad_norm": 0.2273552417755127,
      "learning_rate": 1.99378176842801e-06,
      "loss": 0.0458,
      "step": 54860
    },
    {
      "epoch": 6.419377486543413,
      "loss_main": 0.02251811884343624,
      "step": 54860
    },
    {
      "epoch": 6.420547624619705,
      "grad_norm": 0.2400992065668106,
      "learning_rate": 1.9860846280358224e-06,
      "loss": 0.0478,
      "step": 54870
    },
    {
      "epoch": 6.420547624619705,
      "loss_main": 0.03822179511189461,
      "step": 54870
    },
    {
      "epoch": 6.421717762695998,
      "grad_norm": 0.3152259290218353,
      "learning_rate": 1.978402073003394e-06,
      "loss": 0.0373,
      "step": 54880
    },
    {
      "epoch": 6.421717762695998,
      "loss_main": 0.05988507345318794,
      "step": 54880
    },
    {
      "epoch": 6.422887900772291,
      "grad_norm": 0.22919492423534393,
      "learning_rate": 1.9707341056644736e-06,
      "loss": 0.0436,
      "step": 54890
    },
    {
      "epoch": 6.422887900772291,
      "loss_main": 0.03758211061358452,
      "step": 54890
    },
    {
      "epoch": 6.424058038848584,
      "grad_norm": 0.3229612410068512,
      "learning_rate": 1.963080728348399e-06,
      "loss": 0.0415,
      "step": 54900
    },
    {
      "epoch": 6.424058038848584,
      "loss_main": 0.049533579498529434,
      "step": 54900
    },
    {
      "epoch": 6.425228176924877,
      "grad_norm": 0.20827238261699677,
      "learning_rate": 1.955441943380054e-06,
      "loss": 0.0469,
      "step": 54910
    },
    {
      "epoch": 6.425228176924877,
      "loss_main": 0.09039807319641113,
      "step": 54910
    },
    {
      "epoch": 6.42639831500117,
      "grad_norm": 0.17267800867557526,
      "learning_rate": 1.947817753079906e-06,
      "loss": 0.0555,
      "step": 54920
    },
    {
      "epoch": 6.42639831500117,
      "loss_main": 0.04201928898692131,
      "step": 54920
    },
    {
      "epoch": 6.4275684530774635,
      "grad_norm": 0.2764531075954437,
      "learning_rate": 1.9402081597639785e-06,
      "loss": 0.045,
      "step": 54930
    },
    {
      "epoch": 6.4275684530774635,
      "loss_main": 0.039529211819171906,
      "step": 54930
    },
    {
      "epoch": 6.428738591153756,
      "grad_norm": 0.19719728827476501,
      "learning_rate": 1.9326131657438683e-06,
      "loss": 0.0412,
      "step": 54940
    },
    {
      "epoch": 6.428738591153756,
      "loss_main": 0.034622009843587875,
      "step": 54940
    },
    {
      "epoch": 6.429908729230049,
      "grad_norm": 0.17507366836071014,
      "learning_rate": 1.925032773326724e-06,
      "loss": 0.0451,
      "step": 54950
    },
    {
      "epoch": 6.429908729230049,
      "loss_main": 0.03813744708895683,
      "step": 54950
    },
    {
      "epoch": 6.431078867306342,
      "grad_norm": 0.21579046547412872,
      "learning_rate": 1.9174669848152916e-06,
      "loss": 0.0431,
      "step": 54960
    },
    {
      "epoch": 6.431078867306342,
      "loss_main": 0.0342203751206398,
      "step": 54960
    },
    {
      "epoch": 6.432249005382635,
      "grad_norm": 0.2143688201904297,
      "learning_rate": 1.9099158025078336e-06,
      "loss": 0.0499,
      "step": 54970
    },
    {
      "epoch": 6.432249005382635,
      "loss_main": 0.05017044022679329,
      "step": 54970
    },
    {
      "epoch": 6.433419143458928,
      "grad_norm": 0.22035139799118042,
      "learning_rate": 1.90237922869822e-06,
      "loss": 0.0485,
      "step": 54980
    },
    {
      "epoch": 6.433419143458928,
      "loss_main": 0.044753577560186386,
      "step": 54980
    },
    {
      "epoch": 6.434589281535221,
      "grad_norm": 0.2682301998138428,
      "learning_rate": 1.8948572656758367e-06,
      "loss": 0.0534,
      "step": 54990
    },
    {
      "epoch": 6.434589281535221,
      "loss_main": 0.03539460152387619,
      "step": 54990
    },
    {
      "epoch": 6.435759419611514,
      "grad_norm": 0.25004634261131287,
      "learning_rate": 1.8873499157256834e-06,
      "loss": 0.0452,
      "step": 55000
    },
    {
      "epoch": 6.435759419611514,
      "loss_main": 0.09725601226091385,
      "step": 55000
    },
    {
      "epoch": 6.436929557687807,
      "grad_norm": 0.2474783957004547,
      "learning_rate": 1.879857181128286e-06,
      "loss": 0.045,
      "step": 55010
    },
    {
      "epoch": 6.436929557687807,
      "loss_main": 0.06383414566516876,
      "step": 55010
    },
    {
      "epoch": 6.4380996957641,
      "grad_norm": 0.24079115688800812,
      "learning_rate": 1.8723790641597349e-06,
      "loss": 0.0445,
      "step": 55020
    },
    {
      "epoch": 6.4380996957641,
      "loss_main": 0.03932584077119827,
      "step": 55020
    },
    {
      "epoch": 6.4392698338403935,
      "grad_norm": 0.21146468818187714,
      "learning_rate": 1.8649155670916906e-06,
      "loss": 0.045,
      "step": 55030
    },
    {
      "epoch": 6.4392698338403935,
      "loss_main": 0.04492725804448128,
      "step": 55030
    },
    {
      "epoch": 6.440439971916686,
      "grad_norm": 0.22635704278945923,
      "learning_rate": 1.8574666921913565e-06,
      "loss": 0.0442,
      "step": 55040
    },
    {
      "epoch": 6.440439971916686,
      "loss_main": 0.040695518255233765,
      "step": 55040
    },
    {
      "epoch": 6.441610109992979,
      "grad_norm": 0.2556118071079254,
      "learning_rate": 1.8500324417215166e-06,
      "loss": 0.0393,
      "step": 55050
    },
    {
      "epoch": 6.441610109992979,
      "loss_main": 0.03153783828020096,
      "step": 55050
    },
    {
      "epoch": 6.442780248069273,
      "grad_norm": 0.23169922828674316,
      "learning_rate": 1.8426128179404977e-06,
      "loss": 0.0386,
      "step": 55060
    },
    {
      "epoch": 6.442780248069273,
      "loss_main": 0.046899519860744476,
      "step": 55060
    },
    {
      "epoch": 6.443950386145565,
      "grad_norm": 0.19318585097789764,
      "learning_rate": 1.8352078231021807e-06,
      "loss": 0.0455,
      "step": 55070
    },
    {
      "epoch": 6.443950386145565,
      "loss_main": 0.022835347801446915,
      "step": 55070
    },
    {
      "epoch": 6.445120524221858,
      "grad_norm": 0.21913950145244598,
      "learning_rate": 1.8278174594560049e-06,
      "loss": 0.0409,
      "step": 55080
    },
    {
      "epoch": 6.445120524221858,
      "loss_main": 0.028099006041884422,
      "step": 55080
    },
    {
      "epoch": 6.446290662298151,
      "grad_norm": 0.19671490788459778,
      "learning_rate": 1.82044172924698e-06,
      "loss": 0.0367,
      "step": 55090
    },
    {
      "epoch": 6.446290662298151,
      "loss_main": 0.04317492991685867,
      "step": 55090
    },
    {
      "epoch": 6.4474608003744445,
      "grad_norm": 0.22585920989513397,
      "learning_rate": 1.813080634715636e-06,
      "loss": 0.0456,
      "step": 55100
    },
    {
      "epoch": 6.4474608003744445,
      "loss_main": 0.04786103963851929,
      "step": 55100
    },
    {
      "epoch": 6.448630938450737,
      "grad_norm": 0.3923230469226837,
      "learning_rate": 1.8057341780981119e-06,
      "loss": 0.0472,
      "step": 55110
    },
    {
      "epoch": 6.448630938450737,
      "loss_main": 0.04912310838699341,
      "step": 55110
    },
    {
      "epoch": 6.44980107652703,
      "grad_norm": 0.18443579971790314,
      "learning_rate": 1.7984023616260338e-06,
      "loss": 0.0423,
      "step": 55120
    },
    {
      "epoch": 6.44980107652703,
      "loss_main": 0.03892892971634865,
      "step": 55120
    },
    {
      "epoch": 6.450971214603324,
      "grad_norm": 0.323336660861969,
      "learning_rate": 1.7910851875266421e-06,
      "loss": 0.0474,
      "step": 55130
    },
    {
      "epoch": 6.450971214603324,
      "loss_main": 0.06937120854854584,
      "step": 55130
    },
    {
      "epoch": 6.452141352679616,
      "grad_norm": 0.25693535804748535,
      "learning_rate": 1.7837826580226757e-06,
      "loss": 0.0463,
      "step": 55140
    },
    {
      "epoch": 6.452141352679616,
      "loss_main": 0.06362442672252655,
      "step": 55140
    },
    {
      "epoch": 6.453311490755909,
      "grad_norm": 0.23723618686199188,
      "learning_rate": 1.7764947753324656e-06,
      "loss": 0.043,
      "step": 55150
    },
    {
      "epoch": 6.453311490755909,
      "loss_main": 0.04149533063173294,
      "step": 55150
    },
    {
      "epoch": 6.454481628832202,
      "grad_norm": 0.21533876657485962,
      "learning_rate": 1.7692215416698799e-06,
      "loss": 0.0426,
      "step": 55160
    },
    {
      "epoch": 6.454481628832202,
      "loss_main": 0.03184886649250984,
      "step": 55160
    },
    {
      "epoch": 6.455651766908495,
      "grad_norm": 0.2117060273885727,
      "learning_rate": 1.7619629592443233e-06,
      "loss": 0.0398,
      "step": 55170
    },
    {
      "epoch": 6.455651766908495,
      "loss_main": 0.054242294281721115,
      "step": 55170
    },
    {
      "epoch": 6.456821904984788,
      "grad_norm": 0.4020063281059265,
      "learning_rate": 1.7547190302607709e-06,
      "loss": 0.0396,
      "step": 55180
    },
    {
      "epoch": 6.456821904984788,
      "loss_main": 0.03917703032493591,
      "step": 55180
    },
    {
      "epoch": 6.457992043061081,
      "grad_norm": 0.24115625023841858,
      "learning_rate": 1.747489756919729e-06,
      "loss": 0.0452,
      "step": 55190
    },
    {
      "epoch": 6.457992043061081,
      "loss_main": 0.04302237555384636,
      "step": 55190
    },
    {
      "epoch": 6.4591621811373745,
      "grad_norm": 0.24118249118328094,
      "learning_rate": 1.7402751414172802e-06,
      "loss": 0.0427,
      "step": 55200
    },
    {
      "epoch": 6.4591621811373745,
      "loss_main": 0.0417894646525383,
      "step": 55200
    },
    {
      "epoch": 6.460332319213667,
      "grad_norm": 0.3236064612865448,
      "learning_rate": 1.7330751859450044e-06,
      "loss": 0.0461,
      "step": 55210
    },
    {
      "epoch": 6.460332319213667,
      "loss_main": 0.051480166614055634,
      "step": 55210
    },
    {
      "epoch": 6.46150245728996,
      "grad_norm": 0.25080156326293945,
      "learning_rate": 1.7258898926900801e-06,
      "loss": 0.0477,
      "step": 55220
    },
    {
      "epoch": 6.46150245728996,
      "loss_main": 0.03241259977221489,
      "step": 55220
    },
    {
      "epoch": 6.462672595366254,
      "grad_norm": 0.16442827880382538,
      "learning_rate": 1.7187192638352002e-06,
      "loss": 0.0409,
      "step": 55230
    },
    {
      "epoch": 6.462672595366254,
      "loss_main": 0.039907459169626236,
      "step": 55230
    },
    {
      "epoch": 6.463842733442546,
      "grad_norm": 0.2613483667373657,
      "learning_rate": 1.7115633015586163e-06,
      "loss": 0.0389,
      "step": 55240
    },
    {
      "epoch": 6.463842733442546,
      "loss_main": 0.031186411157250404,
      "step": 55240
    },
    {
      "epoch": 6.465012871518839,
      "grad_norm": 0.37866124510765076,
      "learning_rate": 1.7044220080341178e-06,
      "loss": 0.0453,
      "step": 55250
    },
    {
      "epoch": 6.465012871518839,
      "loss_main": 0.03808952122926712,
      "step": 55250
    },
    {
      "epoch": 6.466183009595132,
      "grad_norm": 0.1978127807378769,
      "learning_rate": 1.6972953854310414e-06,
      "loss": 0.0423,
      "step": 55260
    },
    {
      "epoch": 6.466183009595132,
      "loss_main": 0.027693776413798332,
      "step": 55260
    },
    {
      "epoch": 6.467353147671425,
      "grad_norm": 0.2156807780265808,
      "learning_rate": 1.690183435914261e-06,
      "loss": 0.0494,
      "step": 55270
    },
    {
      "epoch": 6.467353147671425,
      "loss_main": 0.02280217781662941,
      "step": 55270
    },
    {
      "epoch": 6.468523285747718,
      "grad_norm": 0.2729945480823517,
      "learning_rate": 1.6830861616442206e-06,
      "loss": 0.0438,
      "step": 55280
    },
    {
      "epoch": 6.468523285747718,
      "loss_main": 0.07538509368896484,
      "step": 55280
    },
    {
      "epoch": 6.469693423824011,
      "grad_norm": 0.17636138200759888,
      "learning_rate": 1.6760035647768568e-06,
      "loss": 0.0476,
      "step": 55290
    },
    {
      "epoch": 6.469693423824011,
      "loss_main": 0.07776734977960587,
      "step": 55290
    },
    {
      "epoch": 6.4708635619003045,
      "grad_norm": 0.20714718103408813,
      "learning_rate": 1.6689356474636985e-06,
      "loss": 0.0513,
      "step": 55300
    },
    {
      "epoch": 6.4708635619003045,
      "loss_main": 0.05905243009328842,
      "step": 55300
    },
    {
      "epoch": 6.472033699976597,
      "grad_norm": 0.23693446815013885,
      "learning_rate": 1.6618824118517784e-06,
      "loss": 0.0446,
      "step": 55310
    },
    {
      "epoch": 6.472033699976597,
      "loss_main": 0.053893353790044785,
      "step": 55310
    },
    {
      "epoch": 6.47320383805289,
      "grad_norm": 0.1841699481010437,
      "learning_rate": 1.654843860083688e-06,
      "loss": 0.0408,
      "step": 55320
    },
    {
      "epoch": 6.47320383805289,
      "loss_main": 0.049553874880075455,
      "step": 55320
    },
    {
      "epoch": 6.474373976129184,
      "grad_norm": 0.2366110384464264,
      "learning_rate": 1.647819994297556e-06,
      "loss": 0.045,
      "step": 55330
    },
    {
      "epoch": 6.474373976129184,
      "loss_main": 0.03930352255702019,
      "step": 55330
    },
    {
      "epoch": 6.475544114205476,
      "grad_norm": 0.18550509214401245,
      "learning_rate": 1.6408108166270363e-06,
      "loss": 0.0467,
      "step": 55340
    },
    {
      "epoch": 6.475544114205476,
      "loss_main": 0.038856301456689835,
      "step": 55340
    },
    {
      "epoch": 6.476714252281769,
      "grad_norm": 0.27301734685897827,
      "learning_rate": 1.6338163292013486e-06,
      "loss": 0.0492,
      "step": 55350
    },
    {
      "epoch": 6.476714252281769,
      "loss_main": 0.06423046439886093,
      "step": 55350
    },
    {
      "epoch": 6.477884390358062,
      "grad_norm": 0.28428134322166443,
      "learning_rate": 1.6268365341452208e-06,
      "loss": 0.0461,
      "step": 55360
    },
    {
      "epoch": 6.477884390358062,
      "loss_main": 0.039872895926237106,
      "step": 55360
    },
    {
      "epoch": 6.4790545284343555,
      "grad_norm": 0.19844374060630798,
      "learning_rate": 1.6198714335789345e-06,
      "loss": 0.0384,
      "step": 55370
    },
    {
      "epoch": 6.4790545284343555,
      "loss_main": 0.04377675801515579,
      "step": 55370
    },
    {
      "epoch": 6.480224666510648,
      "grad_norm": 0.2757773697376251,
      "learning_rate": 1.612921029618303e-06,
      "loss": 0.0438,
      "step": 55380
    },
    {
      "epoch": 6.480224666510648,
      "loss_main": 0.05838897451758385,
      "step": 55380
    },
    {
      "epoch": 6.481394804586941,
      "grad_norm": 0.2397848218679428,
      "learning_rate": 1.6059853243746815e-06,
      "loss": 0.0484,
      "step": 55390
    },
    {
      "epoch": 6.481394804586941,
      "loss_main": 0.033700764179229736,
      "step": 55390
    },
    {
      "epoch": 6.482564942663235,
      "grad_norm": 0.23086842894554138,
      "learning_rate": 1.5990643199549404e-06,
      "loss": 0.0415,
      "step": 55400
    },
    {
      "epoch": 6.482564942663235,
      "loss_main": 0.040554363280534744,
      "step": 55400
    },
    {
      "epoch": 6.483735080739527,
      "grad_norm": 0.18985143303871155,
      "learning_rate": 1.5921580184615147e-06,
      "loss": 0.0313,
      "step": 55410
    },
    {
      "epoch": 6.483735080739527,
      "loss_main": 0.035159915685653687,
      "step": 55410
    },
    {
      "epoch": 6.48490521881582,
      "grad_norm": 0.2525393068790436,
      "learning_rate": 1.5852664219923374e-06,
      "loss": 0.0435,
      "step": 55420
    },
    {
      "epoch": 6.48490521881582,
      "loss_main": 0.03585958108305931,
      "step": 55420
    },
    {
      "epoch": 6.486075356892114,
      "grad_norm": 0.30017369985580444,
      "learning_rate": 1.5783895326409172e-06,
      "loss": 0.0473,
      "step": 55430
    },
    {
      "epoch": 6.486075356892114,
      "loss_main": 0.020380815491080284,
      "step": 55430
    },
    {
      "epoch": 6.487245494968406,
      "grad_norm": 0.21244370937347412,
      "learning_rate": 1.5715273524962438e-06,
      "loss": 0.0376,
      "step": 55440
    },
    {
      "epoch": 6.487245494968406,
      "loss_main": 0.028947966173291206,
      "step": 55440
    },
    {
      "epoch": 6.488415633044699,
      "grad_norm": 0.27466535568237305,
      "learning_rate": 1.564679883642889e-06,
      "loss": 0.0429,
      "step": 55450
    },
    {
      "epoch": 6.488415633044699,
      "loss_main": 0.021133171394467354,
      "step": 55450
    },
    {
      "epoch": 6.489585771120992,
      "grad_norm": 0.2608717978000641,
      "learning_rate": 1.5578471281609276e-06,
      "loss": 0.0523,
      "step": 55460
    },
    {
      "epoch": 6.489585771120992,
      "loss_main": 0.025904208421707153,
      "step": 55460
    },
    {
      "epoch": 6.4907559091972855,
      "grad_norm": 0.1738034337759018,
      "learning_rate": 1.551029088125966e-06,
      "loss": 0.0516,
      "step": 55470
    },
    {
      "epoch": 6.4907559091972855,
      "loss_main": 0.056691110134124756,
      "step": 55470
    },
    {
      "epoch": 6.491926047273578,
      "grad_norm": 0.22769466042518616,
      "learning_rate": 1.544225765609142e-06,
      "loss": 0.0435,
      "step": 55480
    },
    {
      "epoch": 6.491926047273578,
      "loss_main": 0.03581295534968376,
      "step": 55480
    },
    {
      "epoch": 6.493096185349871,
      "grad_norm": 0.4325226843357086,
      "learning_rate": 1.53743716267713e-06,
      "loss": 0.0454,
      "step": 55490
    },
    {
      "epoch": 6.493096185349871,
      "loss_main": 0.05204338580369949,
      "step": 55490
    },
    {
      "epoch": 6.494266323426165,
      "grad_norm": 0.2605864107608795,
      "learning_rate": 1.5306632813921361e-06,
      "loss": 0.0446,
      "step": 55500
    },
    {
      "epoch": 6.494266323426165,
      "loss_main": 0.06221485882997513,
      "step": 55500
    },
    {
      "epoch": 6.495436461502457,
      "grad_norm": 0.26246729493141174,
      "learning_rate": 1.52390412381187e-06,
      "loss": 0.0442,
      "step": 55510
    },
    {
      "epoch": 6.495436461502457,
      "loss_main": 0.051985520869493484,
      "step": 55510
    },
    {
      "epoch": 6.49660659957875,
      "grad_norm": 0.2655155062675476,
      "learning_rate": 1.5171596919895948e-06,
      "loss": 0.044,
      "step": 55520
    },
    {
      "epoch": 6.49660659957875,
      "loss_main": 0.05889658257365227,
      "step": 55520
    },
    {
      "epoch": 6.497776737655044,
      "grad_norm": 0.20171873271465302,
      "learning_rate": 1.5104299879740936e-06,
      "loss": 0.0442,
      "step": 55530
    },
    {
      "epoch": 6.497776737655044,
      "loss_main": 0.035489391535520554,
      "step": 55530
    },
    {
      "epoch": 6.4989468757313364,
      "grad_norm": 0.20011861622333527,
      "learning_rate": 1.5037150138096701e-06,
      "loss": 0.0402,
      "step": 55540
    },
    {
      "epoch": 6.4989468757313364,
      "loss_main": 0.02882920205593109,
      "step": 55540
    },
    {
      "epoch": 6.500117013807629,
      "grad_norm": 0.4060772657394409,
      "learning_rate": 1.4970147715361538e-06,
      "loss": 0.0462,
      "step": 55550
    },
    {
      "epoch": 6.500117013807629,
      "loss_main": 0.05846511945128441,
      "step": 55550
    },
    {
      "epoch": 6.501287151883922,
      "grad_norm": 0.2215050607919693,
      "learning_rate": 1.4903292631889054e-06,
      "loss": 0.0473,
      "step": 55560
    },
    {
      "epoch": 6.501287151883922,
      "loss_main": 0.044529322534799576,
      "step": 55560
    },
    {
      "epoch": 6.5024572899602155,
      "grad_norm": 0.16189979016780853,
      "learning_rate": 1.483658490798795e-06,
      "loss": 0.0416,
      "step": 55570
    },
    {
      "epoch": 6.5024572899602155,
      "loss_main": 0.048056550323963165,
      "step": 55570
    },
    {
      "epoch": 6.503627428036508,
      "grad_norm": 0.3034301698207855,
      "learning_rate": 1.4770024563922457e-06,
      "loss": 0.0514,
      "step": 55580
    },
    {
      "epoch": 6.503627428036508,
      "loss_main": 0.03881445899605751,
      "step": 55580
    },
    {
      "epoch": 6.504797566112801,
      "grad_norm": 0.19693459570407867,
      "learning_rate": 1.4703611619911628e-06,
      "loss": 0.0468,
      "step": 55590
    },
    {
      "epoch": 6.504797566112801,
      "loss_main": 0.03786559775471687,
      "step": 55590
    },
    {
      "epoch": 6.505967704189095,
      "grad_norm": 0.24875985085964203,
      "learning_rate": 1.4637346096130155e-06,
      "loss": 0.0368,
      "step": 55600
    },
    {
      "epoch": 6.505967704189095,
      "loss_main": 0.03471767157316208,
      "step": 55600
    },
    {
      "epoch": 6.507137842265387,
      "grad_norm": 0.23770931363105774,
      "learning_rate": 1.4571228012707662e-06,
      "loss": 0.0536,
      "step": 55610
    },
    {
      "epoch": 6.507137842265387,
      "loss_main": 0.08159409463405609,
      "step": 55610
    },
    {
      "epoch": 6.50830798034168,
      "grad_norm": 0.1904362142086029,
      "learning_rate": 1.4505257389729132e-06,
      "loss": 0.0463,
      "step": 55620
    },
    {
      "epoch": 6.50830798034168,
      "loss_main": 0.03140983730554581,
      "step": 55620
    },
    {
      "epoch": 6.509478118417974,
      "grad_norm": 0.17025353014469147,
      "learning_rate": 1.4439434247234596e-06,
      "loss": 0.0414,
      "step": 55630
    },
    {
      "epoch": 6.509478118417974,
      "loss_main": 0.06743157655000687,
      "step": 55630
    },
    {
      "epoch": 6.5106482564942665,
      "grad_norm": 0.2319207787513733,
      "learning_rate": 1.4373758605219445e-06,
      "loss": 0.0459,
      "step": 55640
    },
    {
      "epoch": 6.5106482564942665,
      "loss_main": 0.05598612502217293,
      "step": 55640
    },
    {
      "epoch": 6.511818394570559,
      "grad_norm": 0.18246352672576904,
      "learning_rate": 1.4308230483634333e-06,
      "loss": 0.0424,
      "step": 55650
    },
    {
      "epoch": 6.511818394570559,
      "loss_main": 0.0691061019897461,
      "step": 55650
    },
    {
      "epoch": 6.512988532646852,
      "grad_norm": 0.19658373296260834,
      "learning_rate": 1.4242849902384724e-06,
      "loss": 0.0402,
      "step": 55660
    },
    {
      "epoch": 6.512988532646852,
      "loss_main": 0.0264821145683527,
      "step": 55660
    },
    {
      "epoch": 6.514158670723146,
      "grad_norm": 0.25833582878112793,
      "learning_rate": 1.4177616881331734e-06,
      "loss": 0.0459,
      "step": 55670
    },
    {
      "epoch": 6.514158670723146,
      "loss_main": 0.046417396515607834,
      "step": 55670
    },
    {
      "epoch": 6.515328808799438,
      "grad_norm": 0.2701236605644226,
      "learning_rate": 1.4112531440291233e-06,
      "loss": 0.0465,
      "step": 55680
    },
    {
      "epoch": 6.515328808799438,
      "loss_main": 0.03576967865228653,
      "step": 55680
    },
    {
      "epoch": 6.516498946875731,
      "grad_norm": 0.3056478798389435,
      "learning_rate": 1.4047593599034624e-06,
      "loss": 0.0434,
      "step": 55690
    },
    {
      "epoch": 6.516498946875731,
      "loss_main": 0.03501162678003311,
      "step": 55690
    },
    {
      "epoch": 6.517669084952025,
      "grad_norm": 0.16094815731048584,
      "learning_rate": 1.3982803377288246e-06,
      "loss": 0.0355,
      "step": 55700
    },
    {
      "epoch": 6.517669084952025,
      "loss_main": 0.055769264698028564,
      "step": 55700
    },
    {
      "epoch": 6.518839223028317,
      "grad_norm": 0.1946646273136139,
      "learning_rate": 1.3918160794733681e-06,
      "loss": 0.0371,
      "step": 55710
    },
    {
      "epoch": 6.518839223028317,
      "loss_main": 0.027687672525644302,
      "step": 55710
    },
    {
      "epoch": 6.52000936110461,
      "grad_norm": 0.24046330153942108,
      "learning_rate": 1.3853665871007615e-06,
      "loss": 0.0351,
      "step": 55720
    },
    {
      "epoch": 6.52000936110461,
      "loss_main": 0.0572095550596714,
      "step": 55720
    },
    {
      "epoch": 6.521179499180903,
      "grad_norm": 0.1903514266014099,
      "learning_rate": 1.378931862570193e-06,
      "loss": 0.0454,
      "step": 55730
    },
    {
      "epoch": 6.521179499180903,
      "loss_main": 0.03317682072520256,
      "step": 55730
    },
    {
      "epoch": 6.5223496372571965,
      "grad_norm": 0.26610106229782104,
      "learning_rate": 1.372511907836349e-06,
      "loss": 0.0386,
      "step": 55740
    },
    {
      "epoch": 6.5223496372571965,
      "loss_main": 0.047680553048849106,
      "step": 55740
    },
    {
      "epoch": 6.523519775333489,
      "grad_norm": 0.22781792283058167,
      "learning_rate": 1.3661067248494586e-06,
      "loss": 0.0382,
      "step": 55750
    },
    {
      "epoch": 6.523519775333489,
      "loss_main": 0.04424970597028732,
      "step": 55750
    },
    {
      "epoch": 6.524689913409782,
      "grad_norm": 0.1636022925376892,
      "learning_rate": 1.3597163155552429e-06,
      "loss": 0.0376,
      "step": 55760
    },
    {
      "epoch": 6.524689913409782,
      "loss_main": 0.05790344253182411,
      "step": 55760
    },
    {
      "epoch": 6.525860051486076,
      "grad_norm": 0.4118468463420868,
      "learning_rate": 1.3533406818949434e-06,
      "loss": 0.0465,
      "step": 55770
    },
    {
      "epoch": 6.525860051486076,
      "loss_main": 0.05907302722334862,
      "step": 55770
    },
    {
      "epoch": 6.527030189562368,
      "grad_norm": 0.277046263217926,
      "learning_rate": 1.3469798258053002e-06,
      "loss": 0.0453,
      "step": 55780
    },
    {
      "epoch": 6.527030189562368,
      "loss_main": 0.07525493204593658,
      "step": 55780
    },
    {
      "epoch": 6.528200327638661,
      "grad_norm": 0.344412237405777,
      "learning_rate": 1.3406337492185672e-06,
      "loss": 0.0491,
      "step": 55790
    },
    {
      "epoch": 6.528200327638661,
      "loss_main": 0.05545396730303764,
      "step": 55790
    },
    {
      "epoch": 6.529370465714955,
      "grad_norm": 0.2629719078540802,
      "learning_rate": 1.3343024540625414e-06,
      "loss": 0.0394,
      "step": 55800
    },
    {
      "epoch": 6.529370465714955,
      "loss_main": 0.03147829696536064,
      "step": 55800
    },
    {
      "epoch": 6.5305406037912475,
      "grad_norm": 0.23768408596515656,
      "learning_rate": 1.3279859422604735e-06,
      "loss": 0.0455,
      "step": 55810
    },
    {
      "epoch": 6.5305406037912475,
      "loss_main": 0.031323738396167755,
      "step": 55810
    },
    {
      "epoch": 6.53171074186754,
      "grad_norm": 0.21969205141067505,
      "learning_rate": 1.3216842157311781e-06,
      "loss": 0.0494,
      "step": 55820
    },
    {
      "epoch": 6.53171074186754,
      "loss_main": 0.1047804206609726,
      "step": 55820
    },
    {
      "epoch": 6.532880879943834,
      "grad_norm": 0.3053911328315735,
      "learning_rate": 1.3153972763889355e-06,
      "loss": 0.0491,
      "step": 55830
    },
    {
      "epoch": 6.532880879943834,
      "loss_main": 0.05053187161684036,
      "step": 55830
    },
    {
      "epoch": 6.5340510180201266,
      "grad_norm": 0.25396427512168884,
      "learning_rate": 1.3091251261435566e-06,
      "loss": 0.0475,
      "step": 55840
    },
    {
      "epoch": 6.5340510180201266,
      "loss_main": 0.02988516166806221,
      "step": 55840
    },
    {
      "epoch": 6.535221156096419,
      "grad_norm": 0.24341535568237305,
      "learning_rate": 1.3028677669003564e-06,
      "loss": 0.0532,
      "step": 55850
    },
    {
      "epoch": 6.535221156096419,
      "loss_main": 0.05460759624838829,
      "step": 55850
    },
    {
      "epoch": 6.536391294172712,
      "grad_norm": 0.23077784478664398,
      "learning_rate": 1.2966252005601587e-06,
      "loss": 0.0466,
      "step": 55860
    },
    {
      "epoch": 6.536391294172712,
      "loss_main": 0.03573177009820938,
      "step": 55860
    },
    {
      "epoch": 6.537561432249006,
      "grad_norm": 0.19634351134300232,
      "learning_rate": 1.2903974290192855e-06,
      "loss": 0.0422,
      "step": 55870
    },
    {
      "epoch": 6.537561432249006,
      "loss_main": 0.0767819732427597,
      "step": 55870
    },
    {
      "epoch": 6.538731570325298,
      "grad_norm": 0.17880940437316895,
      "learning_rate": 1.284184454169568e-06,
      "loss": 0.0537,
      "step": 55880
    },
    {
      "epoch": 6.538731570325298,
      "loss_main": 0.028460459783673286,
      "step": 55880
    },
    {
      "epoch": 6.539901708401591,
      "grad_norm": 0.1894674450159073,
      "learning_rate": 1.2779862778983464e-06,
      "loss": 0.048,
      "step": 55890
    },
    {
      "epoch": 6.539901708401591,
      "loss_main": 0.06548953801393509,
      "step": 55890
    },
    {
      "epoch": 6.541071846477885,
      "grad_norm": 0.1762550175189972,
      "learning_rate": 1.2718029020884647e-06,
      "loss": 0.0462,
      "step": 55900
    },
    {
      "epoch": 6.541071846477885,
      "loss_main": 0.03572533652186394,
      "step": 55900
    },
    {
      "epoch": 6.5422419845541775,
      "grad_norm": 0.2172519713640213,
      "learning_rate": 1.2656343286182703e-06,
      "loss": 0.0394,
      "step": 55910
    },
    {
      "epoch": 6.5422419845541775,
      "loss_main": 0.051118284463882446,
      "step": 55910
    },
    {
      "epoch": 6.54341212263047,
      "grad_norm": 0.2220430076122284,
      "learning_rate": 1.2594805593616088e-06,
      "loss": 0.0429,
      "step": 55920
    },
    {
      "epoch": 6.54341212263047,
      "loss_main": 0.04003065079450607,
      "step": 55920
    },
    {
      "epoch": 6.544582260706763,
      "grad_norm": 0.2754625380039215,
      "learning_rate": 1.2533415961878404e-06,
      "loss": 0.0472,
      "step": 55930
    },
    {
      "epoch": 6.544582260706763,
      "loss_main": 0.04417981579899788,
      "step": 55930
    },
    {
      "epoch": 6.545752398783057,
      "grad_norm": 0.173591747879982,
      "learning_rate": 1.2472174409618009e-06,
      "loss": 0.0394,
      "step": 55940
    },
    {
      "epoch": 6.545752398783057,
      "loss_main": 0.042413853108882904,
      "step": 55940
    },
    {
      "epoch": 6.546922536859349,
      "grad_norm": 0.28248870372772217,
      "learning_rate": 1.2411080955438747e-06,
      "loss": 0.0402,
      "step": 55950
    },
    {
      "epoch": 6.546922536859349,
      "loss_main": 0.04835566133260727,
      "step": 55950
    },
    {
      "epoch": 6.548092674935642,
      "grad_norm": 0.22760798037052155,
      "learning_rate": 1.235013561789894e-06,
      "loss": 0.0502,
      "step": 55960
    },
    {
      "epoch": 6.548092674935642,
      "loss_main": 0.047470007091760635,
      "step": 55960
    },
    {
      "epoch": 6.549262813011936,
      "grad_norm": 0.22517868876457214,
      "learning_rate": 1.2289338415512385e-06,
      "loss": 0.0421,
      "step": 55970
    },
    {
      "epoch": 6.549262813011936,
      "loss_main": 0.040258366614580154,
      "step": 55970
    },
    {
      "epoch": 6.550432951088228,
      "grad_norm": 0.2491413950920105,
      "learning_rate": 1.222868936674748e-06,
      "loss": 0.034,
      "step": 55980
    },
    {
      "epoch": 6.550432951088228,
      "loss_main": 0.0363449826836586,
      "step": 55980
    },
    {
      "epoch": 6.551603089164521,
      "grad_norm": 0.29741233587265015,
      "learning_rate": 1.2168188490027876e-06,
      "loss": 0.0425,
      "step": 55990
    },
    {
      "epoch": 6.551603089164521,
      "loss_main": 0.027787093073129654,
      "step": 55990
    },
    {
      "epoch": 6.552773227240815,
      "grad_norm": 0.19698527455329895,
      "learning_rate": 1.2107835803732204e-06,
      "loss": 0.0441,
      "step": 56000
    },
    {
      "epoch": 6.552773227240815,
      "loss_main": 0.029080752283334732,
      "step": 56000
    },
    {
      "epoch": 6.5539433653171075,
      "grad_norm": 0.19211606681346893,
      "learning_rate": 1.2047631326193964e-06,
      "loss": 0.0473,
      "step": 56010
    },
    {
      "epoch": 6.5539433653171075,
      "loss_main": 0.043958909809589386,
      "step": 56010
    },
    {
      "epoch": 6.5551135033934,
      "grad_norm": 0.20766417682170868,
      "learning_rate": 1.1987575075701696e-06,
      "loss": 0.0389,
      "step": 56020
    },
    {
      "epoch": 6.5551135033934,
      "loss_main": 0.04068014770746231,
      "step": 56020
    },
    {
      "epoch": 6.556283641469694,
      "grad_norm": 0.22358974814414978,
      "learning_rate": 1.1927667070498916e-06,
      "loss": 0.0517,
      "step": 56030
    },
    {
      "epoch": 6.556283641469694,
      "loss_main": 0.03702571988105774,
      "step": 56030
    },
    {
      "epoch": 6.557453779545987,
      "grad_norm": 0.2241072952747345,
      "learning_rate": 1.1867907328784067e-06,
      "loss": 0.046,
      "step": 56040
    },
    {
      "epoch": 6.557453779545987,
      "loss_main": 0.0453370176255703,
      "step": 56040
    },
    {
      "epoch": 6.558623917622279,
      "grad_norm": 0.3355643153190613,
      "learning_rate": 1.1808295868710518e-06,
      "loss": 0.047,
      "step": 56050
    },
    {
      "epoch": 6.558623917622279,
      "loss_main": 0.046686410903930664,
      "step": 56050
    },
    {
      "epoch": 6.559794055698572,
      "grad_norm": 0.20411643385887146,
      "learning_rate": 1.174883270838678e-06,
      "loss": 0.0365,
      "step": 56060
    },
    {
      "epoch": 6.559794055698572,
      "loss_main": 0.04040800407528877,
      "step": 56060
    },
    {
      "epoch": 6.560964193774866,
      "grad_norm": 0.19883139431476593,
      "learning_rate": 1.1689517865876187e-06,
      "loss": 0.0399,
      "step": 56070
    },
    {
      "epoch": 6.560964193774866,
      "loss_main": 0.04034223034977913,
      "step": 56070
    },
    {
      "epoch": 6.5621343318511585,
      "grad_norm": 0.16851934790611267,
      "learning_rate": 1.1630351359196933e-06,
      "loss": 0.0497,
      "step": 56080
    },
    {
      "epoch": 6.5621343318511585,
      "loss_main": 0.019303161650896072,
      "step": 56080
    },
    {
      "epoch": 6.563304469927451,
      "grad_norm": 0.22325631976127625,
      "learning_rate": 1.1571333206322255e-06,
      "loss": 0.0426,
      "step": 56090
    },
    {
      "epoch": 6.563304469927451,
      "loss_main": 0.04929700121283531,
      "step": 56090
    },
    {
      "epoch": 6.564474608003745,
      "grad_norm": 0.25970080494880676,
      "learning_rate": 1.1512463425180365e-06,
      "loss": 0.0382,
      "step": 56100
    },
    {
      "epoch": 6.564474608003745,
      "loss_main": 0.047810111194849014,
      "step": 56100
    },
    {
      "epoch": 6.565644746080038,
      "grad_norm": 0.21735085546970367,
      "learning_rate": 1.145374203365429e-06,
      "loss": 0.0397,
      "step": 56110
    },
    {
      "epoch": 6.565644746080038,
      "loss_main": 0.04448252171278,
      "step": 56110
    },
    {
      "epoch": 6.56681488415633,
      "grad_norm": 0.22911736369132996,
      "learning_rate": 1.1395169049582155e-06,
      "loss": 0.0463,
      "step": 56120
    },
    {
      "epoch": 6.56681488415633,
      "loss_main": 0.03193800151348114,
      "step": 56120
    },
    {
      "epoch": 6.567985022232623,
      "grad_norm": 0.17707885801792145,
      "learning_rate": 1.1336744490756722e-06,
      "loss": 0.0397,
      "step": 56130
    },
    {
      "epoch": 6.567985022232623,
      "loss_main": 0.034292738884687424,
      "step": 56130
    },
    {
      "epoch": 6.569155160308917,
      "grad_norm": 0.18118926882743835,
      "learning_rate": 1.1278468374925967e-06,
      "loss": 0.0397,
      "step": 56140
    },
    {
      "epoch": 6.569155160308917,
      "loss_main": 0.041450779885053635,
      "step": 56140
    },
    {
      "epoch": 6.570325298385209,
      "grad_norm": 0.24104450643062592,
      "learning_rate": 1.12203407197925e-06,
      "loss": 0.0464,
      "step": 56150
    },
    {
      "epoch": 6.570325298385209,
      "loss_main": 0.06549971550703049,
      "step": 56150
    },
    {
      "epoch": 6.571495436461502,
      "grad_norm": 0.16388021409511566,
      "learning_rate": 1.116236154301409e-06,
      "loss": 0.0397,
      "step": 56160
    },
    {
      "epoch": 6.571495436461502,
      "loss_main": 0.0449155829846859,
      "step": 56160
    },
    {
      "epoch": 6.572665574537796,
      "grad_norm": 0.3404686748981476,
      "learning_rate": 1.11045308622032e-06,
      "loss": 0.0433,
      "step": 56170
    },
    {
      "epoch": 6.572665574537796,
      "loss_main": 0.041299186646938324,
      "step": 56170
    },
    {
      "epoch": 6.5738357126140885,
      "grad_norm": 0.14933454990386963,
      "learning_rate": 1.1046848694927337e-06,
      "loss": 0.0477,
      "step": 56180
    },
    {
      "epoch": 6.5738357126140885,
      "loss_main": 0.04421865567564964,
      "step": 56180
    },
    {
      "epoch": 6.575005850690381,
      "grad_norm": 0.1879754364490509,
      "learning_rate": 1.098931505870876e-06,
      "loss": 0.0485,
      "step": 56190
    },
    {
      "epoch": 6.575005850690381,
      "loss_main": 0.029606325551867485,
      "step": 56190
    },
    {
      "epoch": 6.576175988766675,
      "grad_norm": 0.2389840930700302,
      "learning_rate": 1.0931929971024657e-06,
      "loss": 0.0493,
      "step": 56200
    },
    {
      "epoch": 6.576175988766675,
      "loss_main": 0.02919059619307518,
      "step": 56200
    },
    {
      "epoch": 6.577346126842968,
      "grad_norm": 0.20444680750370026,
      "learning_rate": 1.0874693449307193e-06,
      "loss": 0.0374,
      "step": 56210
    },
    {
      "epoch": 6.577346126842968,
      "loss_main": 0.0627269372344017,
      "step": 56210
    },
    {
      "epoch": 6.57851626491926,
      "grad_norm": 0.22010432183742523,
      "learning_rate": 1.081760551094324e-06,
      "loss": 0.0421,
      "step": 56220
    },
    {
      "epoch": 6.57851626491926,
      "loss_main": 0.0393623486161232,
      "step": 56220
    },
    {
      "epoch": 6.579686402995554,
      "grad_norm": 0.18241702020168304,
      "learning_rate": 1.0760666173274592e-06,
      "loss": 0.0509,
      "step": 56230
    },
    {
      "epoch": 6.579686402995554,
      "loss_main": 0.030417906120419502,
      "step": 56230
    },
    {
      "epoch": 6.580856541071847,
      "grad_norm": 0.25288906693458557,
      "learning_rate": 1.0703875453597967e-06,
      "loss": 0.0421,
      "step": 56240
    },
    {
      "epoch": 6.580856541071847,
      "loss_main": 0.027200715616345406,
      "step": 56240
    },
    {
      "epoch": 6.582026679148139,
      "grad_norm": 0.24156345427036285,
      "learning_rate": 1.0647233369164845e-06,
      "loss": 0.0402,
      "step": 56250
    },
    {
      "epoch": 6.582026679148139,
      "loss_main": 0.05594582483172417,
      "step": 56250
    },
    {
      "epoch": 6.583196817224432,
      "grad_norm": 0.1880263090133667,
      "learning_rate": 1.0590739937181625e-06,
      "loss": 0.0418,
      "step": 56260
    },
    {
      "epoch": 6.583196817224432,
      "loss_main": 0.03031686320900917,
      "step": 56260
    },
    {
      "epoch": 6.584366955300726,
      "grad_norm": 0.2220839411020279,
      "learning_rate": 1.053439517480953e-06,
      "loss": 0.0371,
      "step": 56270
    },
    {
      "epoch": 6.584366955300726,
      "loss_main": 0.021548792719841003,
      "step": 56270
    },
    {
      "epoch": 6.5855370933770185,
      "grad_norm": 0.25188833475112915,
      "learning_rate": 1.047819909916453e-06,
      "loss": 0.0349,
      "step": 56280
    },
    {
      "epoch": 6.5855370933770185,
      "loss_main": 0.04140103608369827,
      "step": 56280
    },
    {
      "epoch": 6.586707231453311,
      "grad_norm": 0.19453226029872894,
      "learning_rate": 1.0422151727317697e-06,
      "loss": 0.0441,
      "step": 56290
    },
    {
      "epoch": 6.586707231453311,
      "loss_main": 0.028326019644737244,
      "step": 56290
    },
    {
      "epoch": 6.587877369529604,
      "grad_norm": 0.1887352019548416,
      "learning_rate": 1.0366253076294462e-06,
      "loss": 0.0376,
      "step": 56300
    },
    {
      "epoch": 6.587877369529604,
      "loss_main": 0.017084242776036263,
      "step": 56300
    },
    {
      "epoch": 6.589047507605898,
      "grad_norm": 0.1848391443490982,
      "learning_rate": 1.031050316307558e-06,
      "loss": 0.0369,
      "step": 56310
    },
    {
      "epoch": 6.589047507605898,
      "loss_main": 0.10204476118087769,
      "step": 56310
    },
    {
      "epoch": 6.59021764568219,
      "grad_norm": 0.2194274216890335,
      "learning_rate": 1.0254902004596333e-06,
      "loss": 0.0446,
      "step": 56320
    },
    {
      "epoch": 6.59021764568219,
      "loss_main": 0.03914600983262062,
      "step": 56320
    },
    {
      "epoch": 6.591387783758483,
      "grad_norm": 0.15293772518634796,
      "learning_rate": 1.0199449617746882e-06,
      "loss": 0.0424,
      "step": 56330
    },
    {
      "epoch": 6.591387783758483,
      "loss_main": 0.04158870503306389,
      "step": 56330
    },
    {
      "epoch": 6.592557921834777,
      "grad_norm": 0.18535903096199036,
      "learning_rate": 1.0144146019372247e-06,
      "loss": 0.0415,
      "step": 56340
    },
    {
      "epoch": 6.592557921834777,
      "loss_main": 0.038470201194286346,
      "step": 56340
    },
    {
      "epoch": 6.5937280599110695,
      "grad_norm": 0.257140189409256,
      "learning_rate": 1.0088991226272048e-06,
      "loss": 0.0378,
      "step": 56350
    },
    {
      "epoch": 6.5937280599110695,
      "loss_main": 0.08446669578552246,
      "step": 56350
    },
    {
      "epoch": 6.594898197987362,
      "grad_norm": 0.21815547347068787,
      "learning_rate": 1.003398525520105e-06,
      "loss": 0.0501,
      "step": 56360
    },
    {
      "epoch": 6.594898197987362,
      "loss_main": 0.06393741816282272,
      "step": 56360
    },
    {
      "epoch": 6.596068336063656,
      "grad_norm": 0.23963108658790588,
      "learning_rate": 9.979128122868552e-07,
      "loss": 0.0383,
      "step": 56370
    },
    {
      "epoch": 6.596068336063656,
      "loss_main": 0.03602554276585579,
      "step": 56370
    },
    {
      "epoch": 6.597238474139949,
      "grad_norm": 0.23383381962776184,
      "learning_rate": 9.924419845938614e-07,
      "loss": 0.039,
      "step": 56380
    },
    {
      "epoch": 6.597238474139949,
      "loss_main": 0.03998811915516853,
      "step": 56380
    },
    {
      "epoch": 6.598408612216241,
      "grad_norm": 0.22453433275222778,
      "learning_rate": 9.869860441030276e-07,
      "loss": 0.0462,
      "step": 56390
    },
    {
      "epoch": 6.598408612216241,
      "loss_main": 0.029380472376942635,
      "step": 56390
    },
    {
      "epoch": 6.599578750292535,
      "grad_norm": 0.27150192856788635,
      "learning_rate": 9.815449924717169e-07,
      "loss": 0.0518,
      "step": 56400
    },
    {
      "epoch": 6.599578750292535,
      "loss_main": 0.05970773845911026,
      "step": 56400
    },
    {
      "epoch": 6.600748888368828,
      "grad_norm": 0.21636822819709778,
      "learning_rate": 9.761188313527791e-07,
      "loss": 0.0442,
      "step": 56410
    },
    {
      "epoch": 6.600748888368828,
      "loss_main": 0.033615194261074066,
      "step": 56410
    },
    {
      "epoch": 6.60191902644512,
      "grad_norm": 0.1888253539800644,
      "learning_rate": 9.70707562394546e-07,
      "loss": 0.0417,
      "step": 56420
    },
    {
      "epoch": 6.60191902644512,
      "loss_main": 0.04580644145607948,
      "step": 56420
    },
    {
      "epoch": 6.603089164521413,
      "grad_norm": 0.2044759839773178,
      "learning_rate": 9.653111872408027e-07,
      "loss": 0.0369,
      "step": 56430
    },
    {
      "epoch": 6.603089164521413,
      "loss_main": 0.048138100653886795,
      "step": 56430
    },
    {
      "epoch": 6.604259302597707,
      "grad_norm": 0.25733068585395813,
      "learning_rate": 9.599297075308434e-07,
      "loss": 0.0491,
      "step": 56440
    },
    {
      "epoch": 6.604259302597707,
      "loss_main": 0.04553009942173958,
      "step": 56440
    },
    {
      "epoch": 6.6054294406739995,
      "grad_norm": 0.2126634567975998,
      "learning_rate": 9.545631248994048e-07,
      "loss": 0.0394,
      "step": 56450
    },
    {
      "epoch": 6.6054294406739995,
      "loss_main": 0.07921723276376724,
      "step": 56450
    },
    {
      "epoch": 6.606599578750292,
      "grad_norm": 0.2653743028640747,
      "learning_rate": 9.492114409767217e-07,
      "loss": 0.0514,
      "step": 56460
    },
    {
      "epoch": 6.606599578750292,
      "loss_main": 0.04145940765738487,
      "step": 56460
    },
    {
      "epoch": 6.607769716826586,
      "grad_norm": 0.1928010880947113,
      "learning_rate": 9.438746573884938e-07,
      "loss": 0.0423,
      "step": 56470
    },
    {
      "epoch": 6.607769716826586,
      "loss_main": 0.033251289278268814,
      "step": 56470
    },
    {
      "epoch": 6.608939854902879,
      "grad_norm": 0.17882956564426422,
      "learning_rate": 9.385527757558909e-07,
      "loss": 0.0512,
      "step": 56480
    },
    {
      "epoch": 6.608939854902879,
      "loss_main": 0.051260821521282196,
      "step": 56480
    },
    {
      "epoch": 6.610109992979171,
      "grad_norm": 0.2218305915594101,
      "learning_rate": 9.332457976955644e-07,
      "loss": 0.0454,
      "step": 56490
    },
    {
      "epoch": 6.610109992979171,
      "loss_main": 0.04630887508392334,
      "step": 56490
    },
    {
      "epoch": 6.611280131055464,
      "grad_norm": 0.19828833639621735,
      "learning_rate": 9.279537248196247e-07,
      "loss": 0.0431,
      "step": 56500
    },
    {
      "epoch": 6.611280131055464,
      "loss_main": 0.04015880823135376,
      "step": 56500
    },
    {
      "epoch": 6.612450269131758,
      "grad_norm": 0.22227463126182556,
      "learning_rate": 9.2267655873568e-07,
      "loss": 0.0504,
      "step": 56510
    },
    {
      "epoch": 6.612450269131758,
      "loss_main": 0.06812296062707901,
      "step": 56510
    },
    {
      "epoch": 6.6136204072080504,
      "grad_norm": 0.21208487451076508,
      "learning_rate": 9.174143010467762e-07,
      "loss": 0.043,
      "step": 56520
    },
    {
      "epoch": 6.6136204072080504,
      "loss_main": 0.051758330315351486,
      "step": 56520
    },
    {
      "epoch": 6.614790545284343,
      "grad_norm": 0.18185488879680634,
      "learning_rate": 9.12166953351462e-07,
      "loss": 0.044,
      "step": 56530
    },
    {
      "epoch": 6.614790545284343,
      "loss_main": 0.07745493948459625,
      "step": 56530
    },
    {
      "epoch": 6.615960683360637,
      "grad_norm": 0.22530843317508698,
      "learning_rate": 9.069345172437404e-07,
      "loss": 0.0496,
      "step": 56540
    },
    {
      "epoch": 6.615960683360637,
      "loss_main": 0.028346985578536987,
      "step": 56540
    },
    {
      "epoch": 6.6171308214369295,
      "grad_norm": 0.19229933619499207,
      "learning_rate": 9.017169943130843e-07,
      "loss": 0.0415,
      "step": 56550
    },
    {
      "epoch": 6.6171308214369295,
      "loss_main": 0.019374670460820198,
      "step": 56550
    },
    {
      "epoch": 6.618300959513222,
      "grad_norm": 0.23718364536762238,
      "learning_rate": 8.965143861444425e-07,
      "loss": 0.0403,
      "step": 56560
    },
    {
      "epoch": 6.618300959513222,
      "loss_main": 0.06668516248464584,
      "step": 56560
    },
    {
      "epoch": 6.619471097589516,
      "grad_norm": 0.164542555809021,
      "learning_rate": 8.91326694318223e-07,
      "loss": 0.0357,
      "step": 56570
    },
    {
      "epoch": 6.619471097589516,
      "loss_main": 0.047157514840364456,
      "step": 56570
    },
    {
      "epoch": 6.620641235665809,
      "grad_norm": 0.2027062326669693,
      "learning_rate": 8.861539204103098e-07,
      "loss": 0.0406,
      "step": 56580
    },
    {
      "epoch": 6.620641235665809,
      "loss_main": 0.06537865102291107,
      "step": 56580
    },
    {
      "epoch": 6.621811373742101,
      "grad_norm": 0.2735450863838196,
      "learning_rate": 8.809960659920735e-07,
      "loss": 0.0427,
      "step": 56590
    },
    {
      "epoch": 6.621811373742101,
      "loss_main": 0.02971305511891842,
      "step": 56590
    },
    {
      "epoch": 6.622981511818395,
      "grad_norm": 0.1935201734304428,
      "learning_rate": 8.758531326303055e-07,
      "loss": 0.0432,
      "step": 56600
    },
    {
      "epoch": 6.622981511818395,
      "loss_main": 0.043480705469846725,
      "step": 56600
    },
    {
      "epoch": 6.624151649894688,
      "grad_norm": 0.21626292169094086,
      "learning_rate": 8.707251218873169e-07,
      "loss": 0.0366,
      "step": 56610
    },
    {
      "epoch": 6.624151649894688,
      "loss_main": 0.039593033492565155,
      "step": 56610
    },
    {
      "epoch": 6.6253217879709805,
      "grad_norm": 0.28062358498573303,
      "learning_rate": 8.656120353208507e-07,
      "loss": 0.0418,
      "step": 56620
    },
    {
      "epoch": 6.6253217879709805,
      "loss_main": 0.028548747301101685,
      "step": 56620
    },
    {
      "epoch": 6.626491926047273,
      "grad_norm": 0.2843966484069824,
      "learning_rate": 8.605138744841312e-07,
      "loss": 0.0399,
      "step": 56630
    },
    {
      "epoch": 6.626491926047273,
      "loss_main": 0.024801025167107582,
      "step": 56630
    },
    {
      "epoch": 6.627662064123567,
      "grad_norm": 0.23709920048713684,
      "learning_rate": 8.554306409258417e-07,
      "loss": 0.0394,
      "step": 56640
    },
    {
      "epoch": 6.627662064123567,
      "loss_main": 0.053000617772340775,
      "step": 56640
    },
    {
      "epoch": 6.62883220219986,
      "grad_norm": 0.34579014778137207,
      "learning_rate": 8.503623361901358e-07,
      "loss": 0.0572,
      "step": 56650
    },
    {
      "epoch": 6.62883220219986,
      "loss_main": 0.0204036645591259,
      "step": 56650
    },
    {
      "epoch": 6.630002340276152,
      "grad_norm": 0.2750664949417114,
      "learning_rate": 8.453089618166377e-07,
      "loss": 0.039,
      "step": 56660
    },
    {
      "epoch": 6.630002340276152,
      "loss_main": 0.035149574279785156,
      "step": 56660
    },
    {
      "epoch": 6.631172478352446,
      "grad_norm": 0.2563384771347046,
      "learning_rate": 8.402705193404137e-07,
      "loss": 0.0451,
      "step": 56670
    },
    {
      "epoch": 6.631172478352446,
      "loss_main": 0.02152050845324993,
      "step": 56670
    },
    {
      "epoch": 6.632342616428739,
      "grad_norm": 0.27917686104774475,
      "learning_rate": 8.352470102920174e-07,
      "loss": 0.0393,
      "step": 56680
    },
    {
      "epoch": 6.632342616428739,
      "loss_main": 0.03749025613069534,
      "step": 56680
    },
    {
      "epoch": 6.633512754505031,
      "grad_norm": 0.19674552977085114,
      "learning_rate": 8.302384361974669e-07,
      "loss": 0.0424,
      "step": 56690
    },
    {
      "epoch": 6.633512754505031,
      "loss_main": 0.06035660579800606,
      "step": 56690
    },
    {
      "epoch": 6.634682892581324,
      "grad_norm": 0.18830356001853943,
      "learning_rate": 8.252447985782231e-07,
      "loss": 0.0479,
      "step": 56700
    },
    {
      "epoch": 6.634682892581324,
      "loss_main": 0.035401783883571625,
      "step": 56700
    },
    {
      "epoch": 6.635853030657618,
      "grad_norm": 0.1672803908586502,
      "learning_rate": 8.202660989512279e-07,
      "loss": 0.0367,
      "step": 56710
    },
    {
      "epoch": 6.635853030657618,
      "loss_main": 0.05459937825798988,
      "step": 56710
    },
    {
      "epoch": 6.6370231687339105,
      "grad_norm": 0.20413585007190704,
      "learning_rate": 8.153023388288772e-07,
      "loss": 0.0402,
      "step": 56720
    },
    {
      "epoch": 6.6370231687339105,
      "loss_main": 0.0442078523337841,
      "step": 56720
    },
    {
      "epoch": 6.638193306810203,
      "grad_norm": 0.23738296329975128,
      "learning_rate": 8.103535197190204e-07,
      "loss": 0.0524,
      "step": 56730
    },
    {
      "epoch": 6.638193306810203,
      "loss_main": 0.03522903099656105,
      "step": 56730
    },
    {
      "epoch": 6.639363444886497,
      "grad_norm": 0.2282010018825531,
      "learning_rate": 8.054196431249993e-07,
      "loss": 0.039,
      "step": 56740
    },
    {
      "epoch": 6.639363444886497,
      "loss_main": 0.05168351158499718,
      "step": 56740
    },
    {
      "epoch": 6.64053358296279,
      "grad_norm": 0.20105288922786713,
      "learning_rate": 8.005007105455709e-07,
      "loss": 0.0459,
      "step": 56750
    },
    {
      "epoch": 6.64053358296279,
      "loss_main": 0.0542261004447937,
      "step": 56750
    },
    {
      "epoch": 6.641703721039082,
      "grad_norm": 0.3024175763130188,
      "learning_rate": 7.95596723474995e-07,
      "loss": 0.0472,
      "step": 56760
    },
    {
      "epoch": 6.641703721039082,
      "loss_main": 0.04287778586149216,
      "step": 56760
    },
    {
      "epoch": 6.642873859115376,
      "grad_norm": 0.2031564712524414,
      "learning_rate": 7.907076834029692e-07,
      "loss": 0.0396,
      "step": 56770
    },
    {
      "epoch": 6.642873859115376,
      "loss_main": 0.03494292497634888,
      "step": 56770
    },
    {
      "epoch": 6.644043997191669,
      "grad_norm": 0.24710844457149506,
      "learning_rate": 7.858335918146498e-07,
      "loss": 0.0394,
      "step": 56780
    },
    {
      "epoch": 6.644043997191669,
      "loss_main": 0.021753210574388504,
      "step": 56780
    },
    {
      "epoch": 6.6452141352679615,
      "grad_norm": 0.31354352831840515,
      "learning_rate": 7.809744501906635e-07,
      "loss": 0.0495,
      "step": 56790
    },
    {
      "epoch": 6.6452141352679615,
      "loss_main": 0.02380209043622017,
      "step": 56790
    },
    {
      "epoch": 6.646384273344255,
      "grad_norm": 0.22445674240589142,
      "learning_rate": 7.761302600070797e-07,
      "loss": 0.0351,
      "step": 56800
    },
    {
      "epoch": 6.646384273344255,
      "loss_main": 0.07104260474443436,
      "step": 56800
    },
    {
      "epoch": 6.647554411420548,
      "grad_norm": 0.16284853219985962,
      "learning_rate": 7.713010227354545e-07,
      "loss": 0.0463,
      "step": 56810
    },
    {
      "epoch": 6.647554411420548,
      "loss_main": 0.0395129919052124,
      "step": 56810
    },
    {
      "epoch": 6.6487245494968406,
      "grad_norm": 0.17261338233947754,
      "learning_rate": 7.664867398427589e-07,
      "loss": 0.0449,
      "step": 56820
    },
    {
      "epoch": 6.6487245494968406,
      "loss_main": 0.052179861813783646,
      "step": 56820
    },
    {
      "epoch": 6.649894687573133,
      "grad_norm": 0.2551567554473877,
      "learning_rate": 7.616874127914619e-07,
      "loss": 0.0497,
      "step": 56830
    },
    {
      "epoch": 6.649894687573133,
      "loss_main": 0.05627604201436043,
      "step": 56830
    },
    {
      "epoch": 6.651064825649427,
      "grad_norm": 0.1760953962802887,
      "learning_rate": 7.569030430394641e-07,
      "loss": 0.0438,
      "step": 56840
    },
    {
      "epoch": 6.651064825649427,
      "loss_main": 0.0389782190322876,
      "step": 56840
    },
    {
      "epoch": 6.65223496372572,
      "grad_norm": 0.20695747435092926,
      "learning_rate": 7.521336320401306e-07,
      "loss": 0.0424,
      "step": 56850
    },
    {
      "epoch": 6.65223496372572,
      "loss_main": 0.03506442904472351,
      "step": 56850
    },
    {
      "epoch": 6.653405101802012,
      "grad_norm": 0.21995025873184204,
      "learning_rate": 7.473791812422915e-07,
      "loss": 0.0418,
      "step": 56860
    },
    {
      "epoch": 6.653405101802012,
      "loss_main": 0.04387964680790901,
      "step": 56860
    },
    {
      "epoch": 6.654575239878306,
      "grad_norm": 0.23615624010562897,
      "learning_rate": 7.426396920902134e-07,
      "loss": 0.0402,
      "step": 56870
    },
    {
      "epoch": 6.654575239878306,
      "loss_main": 0.03738114610314369,
      "step": 56870
    },
    {
      "epoch": 6.655745377954599,
      "grad_norm": 0.18244272470474243,
      "learning_rate": 7.379151660236283e-07,
      "loss": 0.0516,
      "step": 56880
    },
    {
      "epoch": 6.655745377954599,
      "loss_main": 0.03851240873336792,
      "step": 56880
    },
    {
      "epoch": 6.6569155160308915,
      "grad_norm": 0.27331477403640747,
      "learning_rate": 7.332056044777324e-07,
      "loss": 0.0435,
      "step": 56890
    },
    {
      "epoch": 6.6569155160308915,
      "loss_main": 0.02644207701086998,
      "step": 56890
    },
    {
      "epoch": 6.658085654107184,
      "grad_norm": 0.18451955914497375,
      "learning_rate": 7.285110088831537e-07,
      "loss": 0.0428,
      "step": 56900
    },
    {
      "epoch": 6.658085654107184,
      "loss_main": 0.04246775433421135,
      "step": 56900
    },
    {
      "epoch": 6.659255792183478,
      "grad_norm": 0.24467071890830994,
      "learning_rate": 7.238313806659902e-07,
      "loss": 0.0487,
      "step": 56910
    },
    {
      "epoch": 6.659255792183478,
      "loss_main": 0.033078547567129135,
      "step": 56910
    },
    {
      "epoch": 6.660425930259771,
      "grad_norm": 0.278083473443985,
      "learning_rate": 7.191667212477993e-07,
      "loss": 0.0514,
      "step": 56920
    },
    {
      "epoch": 6.660425930259771,
      "loss_main": 0.02630365453660488,
      "step": 56920
    },
    {
      "epoch": 6.661596068336063,
      "grad_norm": 0.19360706210136414,
      "learning_rate": 7.145170320455697e-07,
      "loss": 0.0394,
      "step": 56930
    },
    {
      "epoch": 6.661596068336063,
      "loss_main": 0.060468949377536774,
      "step": 56930
    },
    {
      "epoch": 6.662766206412357,
      "grad_norm": 0.16308121383190155,
      "learning_rate": 7.098823144717604e-07,
      "loss": 0.0432,
      "step": 56940
    },
    {
      "epoch": 6.662766206412357,
      "loss_main": 0.043856825679540634,
      "step": 56940
    },
    {
      "epoch": 6.66393634448865,
      "grad_norm": 0.3411703407764435,
      "learning_rate": 7.052625699342674e-07,
      "loss": 0.051,
      "step": 56950
    },
    {
      "epoch": 6.66393634448865,
      "loss_main": 0.03378142789006233,
      "step": 56950
    },
    {
      "epoch": 6.665106482564942,
      "grad_norm": 0.22277122735977173,
      "learning_rate": 7.006577998364628e-07,
      "loss": 0.0442,
      "step": 56960
    },
    {
      "epoch": 6.665106482564942,
      "loss_main": 0.06302563846111298,
      "step": 56960
    },
    {
      "epoch": 6.666276620641236,
      "grad_norm": 0.24459969997406006,
      "learning_rate": 6.96068005577133e-07,
      "loss": 0.0456,
      "step": 56970
    },
    {
      "epoch": 6.666276620641236,
      "loss_main": 0.0728907436132431,
      "step": 56970
    },
    {
      "epoch": 6.667446758717529,
      "grad_norm": 0.28449195623397827,
      "learning_rate": 6.914931885505627e-07,
      "loss": 0.051,
      "step": 56980
    },
    {
      "epoch": 6.667446758717529,
      "loss_main": 0.03171950951218605,
      "step": 56980
    },
    {
      "epoch": 6.6686168967938215,
      "grad_norm": 0.17729033529758453,
      "learning_rate": 6.869333501464347e-07,
      "loss": 0.0391,
      "step": 56990
    },
    {
      "epoch": 6.6686168967938215,
      "loss_main": 0.028513047844171524,
      "step": 56990
    },
    {
      "epoch": 6.669787034870115,
      "grad_norm": 0.28568169474601746,
      "learning_rate": 6.823884917499246e-07,
      "loss": 0.0442,
      "step": 57000
    },
    {
      "epoch": 6.669787034870115,
      "loss_main": 0.07630898058414459,
      "step": 57000
    },
    {
      "epoch": 6.670957172946408,
      "grad_norm": 0.21218577027320862,
      "learning_rate": 6.778586147416278e-07,
      "loss": 0.0427,
      "step": 57010
    },
    {
      "epoch": 6.670957172946408,
      "loss_main": 0.0526629202067852,
      "step": 57010
    },
    {
      "epoch": 6.672127311022701,
      "grad_norm": 0.29159775376319885,
      "learning_rate": 6.733437204976156e-07,
      "loss": 0.049,
      "step": 57020
    },
    {
      "epoch": 6.672127311022701,
      "loss_main": 0.05723080411553383,
      "step": 57020
    },
    {
      "epoch": 6.673297449098993,
      "grad_norm": 0.29883238673210144,
      "learning_rate": 6.688438103893857e-07,
      "loss": 0.0445,
      "step": 57030
    },
    {
      "epoch": 6.673297449098993,
      "loss_main": 0.05996537208557129,
      "step": 57030
    },
    {
      "epoch": 6.674467587175287,
      "grad_norm": 0.2695520520210266,
      "learning_rate": 6.64358885783889e-07,
      "loss": 0.0446,
      "step": 57040
    },
    {
      "epoch": 6.674467587175287,
      "loss_main": 0.03454288840293884,
      "step": 57040
    },
    {
      "epoch": 6.67563772525158,
      "grad_norm": 0.2712351083755493,
      "learning_rate": 6.598889480435299e-07,
      "loss": 0.0432,
      "step": 57050
    },
    {
      "epoch": 6.67563772525158,
      "loss_main": 0.046354491263628006,
      "step": 57050
    },
    {
      "epoch": 6.6768078633278725,
      "grad_norm": 0.3092571198940277,
      "learning_rate": 6.554339985261615e-07,
      "loss": 0.0381,
      "step": 57060
    },
    {
      "epoch": 6.6768078633278725,
      "loss_main": 0.03200516849756241,
      "step": 57060
    },
    {
      "epoch": 6.677978001404165,
      "grad_norm": 0.17787641286849976,
      "learning_rate": 6.509940385850733e-07,
      "loss": 0.0455,
      "step": 57070
    },
    {
      "epoch": 6.677978001404165,
      "loss_main": 0.03540617600083351,
      "step": 57070
    },
    {
      "epoch": 6.679148139480459,
      "grad_norm": 0.3013744354248047,
      "learning_rate": 6.465690695690141e-07,
      "loss": 0.0491,
      "step": 57080
    },
    {
      "epoch": 6.679148139480459,
      "loss_main": 0.04092533141374588,
      "step": 57080
    },
    {
      "epoch": 6.680318277556752,
      "grad_norm": 0.20245379209518433,
      "learning_rate": 6.421590928221699e-07,
      "loss": 0.0431,
      "step": 57090
    },
    {
      "epoch": 6.680318277556752,
      "loss_main": 0.03475209325551987,
      "step": 57090
    },
    {
      "epoch": 6.681488415633044,
      "grad_norm": 0.27802374958992004,
      "learning_rate": 6.377641096841691e-07,
      "loss": 0.0454,
      "step": 57100
    },
    {
      "epoch": 6.681488415633044,
      "loss_main": 0.038217782974243164,
      "step": 57100
    },
    {
      "epoch": 6.682658553709338,
      "grad_norm": 0.3089686334133148,
      "learning_rate": 6.333841214901048e-07,
      "loss": 0.0411,
      "step": 57110
    },
    {
      "epoch": 6.682658553709338,
      "loss_main": 0.036375466734170914,
      "step": 57110
    },
    {
      "epoch": 6.683828691785631,
      "grad_norm": 0.25540247559547424,
      "learning_rate": 6.290191295704906e-07,
      "loss": 0.0399,
      "step": 57120
    },
    {
      "epoch": 6.683828691785631,
      "loss_main": 0.05570477247238159,
      "step": 57120
    },
    {
      "epoch": 6.684998829861923,
      "grad_norm": 0.19262994825839996,
      "learning_rate": 6.246691352513046e-07,
      "loss": 0.0446,
      "step": 57130
    },
    {
      "epoch": 6.684998829861923,
      "loss_main": 0.07123961299657822,
      "step": 57130
    },
    {
      "epoch": 6.686168967938217,
      "grad_norm": 0.16782136261463165,
      "learning_rate": 6.203341398539452e-07,
      "loss": 0.0363,
      "step": 57140
    },
    {
      "epoch": 6.686168967938217,
      "loss_main": 0.052784088999032974,
      "step": 57140
    },
    {
      "epoch": 6.68733910601451,
      "grad_norm": 0.18937693536281586,
      "learning_rate": 6.160141446952872e-07,
      "loss": 0.0491,
      "step": 57150
    },
    {
      "epoch": 6.68733910601451,
      "loss_main": 0.05365445837378502,
      "step": 57150
    },
    {
      "epoch": 6.6885092440908025,
      "grad_norm": 0.22678905725479126,
      "learning_rate": 6.11709151087625e-07,
      "loss": 0.0413,
      "step": 57160
    },
    {
      "epoch": 6.6885092440908025,
      "loss_main": 0.038074951618909836,
      "step": 57160
    },
    {
      "epoch": 6.689679382167096,
      "grad_norm": 0.1922735869884491,
      "learning_rate": 6.074191603386958e-07,
      "loss": 0.0479,
      "step": 57170
    },
    {
      "epoch": 6.689679382167096,
      "loss_main": 0.05898064374923706,
      "step": 57170
    },
    {
      "epoch": 6.690849520243389,
      "grad_norm": 0.163579061627388,
      "learning_rate": 6.031441737516907e-07,
      "loss": 0.0386,
      "step": 57180
    },
    {
      "epoch": 6.690849520243389,
      "loss_main": 0.04713091626763344,
      "step": 57180
    },
    {
      "epoch": 6.692019658319682,
      "grad_norm": 0.19842173159122467,
      "learning_rate": 5.988841926252431e-07,
      "loss": 0.049,
      "step": 57190
    },
    {
      "epoch": 6.692019658319682,
      "loss_main": 0.05726388096809387,
      "step": 57190
    },
    {
      "epoch": 6.693189796395975,
      "grad_norm": 0.1700318306684494,
      "learning_rate": 5.946392182534066e-07,
      "loss": 0.0419,
      "step": 57200
    },
    {
      "epoch": 6.693189796395975,
      "loss_main": 0.023530783131718636,
      "step": 57200
    },
    {
      "epoch": 6.694359934472268,
      "grad_norm": 0.21318094432353973,
      "learning_rate": 5.90409251925711e-07,
      "loss": 0.0463,
      "step": 57210
    },
    {
      "epoch": 6.694359934472268,
      "loss_main": 0.03308838605880737,
      "step": 57210
    },
    {
      "epoch": 6.695530072548561,
      "grad_norm": 0.24857081472873688,
      "learning_rate": 5.861942949270949e-07,
      "loss": 0.0367,
      "step": 57220
    },
    {
      "epoch": 6.695530072548561,
      "loss_main": 0.05315890535712242,
      "step": 57220
    },
    {
      "epoch": 6.696700210624853,
      "grad_norm": 0.2787034511566162,
      "learning_rate": 5.819943485379564e-07,
      "loss": 0.0524,
      "step": 57230
    },
    {
      "epoch": 6.696700210624853,
      "loss_main": 0.030389541760087013,
      "step": 57230
    },
    {
      "epoch": 6.697870348701147,
      "grad_norm": 0.13803422451019287,
      "learning_rate": 5.778094140341306e-07,
      "loss": 0.0476,
      "step": 57240
    },
    {
      "epoch": 6.697870348701147,
      "loss_main": 0.02495815046131611,
      "step": 57240
    },
    {
      "epoch": 6.69904048677744,
      "grad_norm": 0.23746877908706665,
      "learning_rate": 5.736394926868893e-07,
      "loss": 0.0427,
      "step": 57250
    },
    {
      "epoch": 6.69904048677744,
      "loss_main": 0.03971392288804054,
      "step": 57250
    },
    {
      "epoch": 6.7002106248537325,
      "grad_norm": 0.19901567697525024,
      "learning_rate": 5.694845857629416e-07,
      "loss": 0.0453,
      "step": 57260
    },
    {
      "epoch": 6.7002106248537325,
      "loss_main": 0.07200857251882553,
      "step": 57260
    },
    {
      "epoch": 6.701380762930025,
      "grad_norm": 0.2561838924884796,
      "learning_rate": 5.653446945244334e-07,
      "loss": 0.0433,
      "step": 57270
    },
    {
      "epoch": 6.701380762930025,
      "loss_main": 0.05793386697769165,
      "step": 57270
    },
    {
      "epoch": 6.702550901006319,
      "grad_norm": 0.35838550329208374,
      "learning_rate": 5.612198202289698e-07,
      "loss": 0.046,
      "step": 57280
    },
    {
      "epoch": 6.702550901006319,
      "loss_main": 0.03234110772609711,
      "step": 57280
    },
    {
      "epoch": 6.703721039082612,
      "grad_norm": 0.29081079363822937,
      "learning_rate": 5.571099641295596e-07,
      "loss": 0.0454,
      "step": 57290
    },
    {
      "epoch": 6.703721039082612,
      "loss_main": 0.03783627972006798,
      "step": 57290
    },
    {
      "epoch": 6.704891177158904,
      "grad_norm": 0.25714483857154846,
      "learning_rate": 5.530151274746875e-07,
      "loss": 0.0422,
      "step": 57300
    },
    {
      "epoch": 6.704891177158904,
      "loss_main": 0.04553220421075821,
      "step": 57300
    },
    {
      "epoch": 6.706061315235198,
      "grad_norm": 0.22395339608192444,
      "learning_rate": 5.489353115082418e-07,
      "loss": 0.0411,
      "step": 57310
    },
    {
      "epoch": 6.706061315235198,
      "loss_main": 0.04859942942857742,
      "step": 57310
    },
    {
      "epoch": 6.707231453311491,
      "grad_norm": 0.23110592365264893,
      "learning_rate": 5.4487051746957e-07,
      "loss": 0.0427,
      "step": 57320
    },
    {
      "epoch": 6.707231453311491,
      "loss_main": 0.026038357987999916,
      "step": 57320
    },
    {
      "epoch": 6.7084015913877835,
      "grad_norm": 0.36321648955345154,
      "learning_rate": 5.408207465934511e-07,
      "loss": 0.0515,
      "step": 57330
    },
    {
      "epoch": 6.7084015913877835,
      "loss_main": 0.026288026943802834,
      "step": 57330
    },
    {
      "epoch": 6.709571729464077,
      "grad_norm": 0.24264845252037048,
      "learning_rate": 5.3678600011009e-07,
      "loss": 0.0469,
      "step": 57340
    },
    {
      "epoch": 6.709571729464077,
      "loss_main": 0.041122931987047195,
      "step": 57340
    },
    {
      "epoch": 6.71074186754037,
      "grad_norm": 0.2443399578332901,
      "learning_rate": 5.327662792451449e-07,
      "loss": 0.0459,
      "step": 57350
    },
    {
      "epoch": 6.71074186754037,
      "loss_main": 0.03779765963554382,
      "step": 57350
    },
    {
      "epoch": 6.711912005616663,
      "grad_norm": 0.16292889416217804,
      "learning_rate": 5.287615852196947e-07,
      "loss": 0.0403,
      "step": 57360
    },
    {
      "epoch": 6.711912005616663,
      "loss_main": 0.018753772601485252,
      "step": 57360
    },
    {
      "epoch": 6.713082143692956,
      "grad_norm": 0.253409206867218,
      "learning_rate": 5.247719192502665e-07,
      "loss": 0.0426,
      "step": 57370
    },
    {
      "epoch": 6.713082143692956,
      "loss_main": 0.03433893993496895,
      "step": 57370
    },
    {
      "epoch": 6.714252281769249,
      "grad_norm": 0.24821527302265167,
      "learning_rate": 5.207972825488128e-07,
      "loss": 0.0418,
      "step": 57380
    },
    {
      "epoch": 6.714252281769249,
      "loss_main": 0.04393171891570091,
      "step": 57380
    },
    {
      "epoch": 6.715422419845542,
      "grad_norm": 0.2676764130592346,
      "learning_rate": 5.168376763227178e-07,
      "loss": 0.0446,
      "step": 57390
    },
    {
      "epoch": 6.715422419845542,
      "loss_main": 0.026791440322995186,
      "step": 57390
    },
    {
      "epoch": 6.716592557921834,
      "grad_norm": 0.21718235313892365,
      "learning_rate": 5.12893101774814e-07,
      "loss": 0.0395,
      "step": 57400
    },
    {
      "epoch": 6.716592557921834,
      "loss_main": 0.027051186189055443,
      "step": 57400
    },
    {
      "epoch": 6.717762695998128,
      "grad_norm": 0.21727769076824188,
      "learning_rate": 5.089635601033483e-07,
      "loss": 0.0422,
      "step": 57410
    },
    {
      "epoch": 6.717762695998128,
      "loss_main": 0.026456424966454506,
      "step": 57410
    },
    {
      "epoch": 6.718932834074421,
      "grad_norm": 0.2520313858985901,
      "learning_rate": 5.050490525020213e-07,
      "loss": 0.0389,
      "step": 57420
    },
    {
      "epoch": 6.718932834074421,
      "loss_main": 0.02752932906150818,
      "step": 57420
    },
    {
      "epoch": 6.7201029721507135,
      "grad_norm": 0.23319722712039948,
      "learning_rate": 5.011495801599541e-07,
      "loss": 0.0407,
      "step": 57430
    },
    {
      "epoch": 6.7201029721507135,
      "loss_main": 0.04247098043560982,
      "step": 57430
    },
    {
      "epoch": 6.721273110227007,
      "grad_norm": 0.17206749320030212,
      "learning_rate": 4.972651442616994e-07,
      "loss": 0.0422,
      "step": 57440
    },
    {
      "epoch": 6.721273110227007,
      "loss_main": 0.053608398884534836,
      "step": 57440
    },
    {
      "epoch": 6.7224432483033,
      "grad_norm": 0.25900885462760925,
      "learning_rate": 4.933957459872574e-07,
      "loss": 0.0426,
      "step": 57450
    },
    {
      "epoch": 6.7224432483033,
      "loss_main": 0.048105984926223755,
      "step": 57450
    },
    {
      "epoch": 6.723613386379593,
      "grad_norm": 0.23822885751724243,
      "learning_rate": 4.895413865120324e-07,
      "loss": 0.0435,
      "step": 57460
    },
    {
      "epoch": 6.723613386379593,
      "loss_main": 0.029178455471992493,
      "step": 57460
    },
    {
      "epoch": 6.724783524455885,
      "grad_norm": 0.2628805935382843,
      "learning_rate": 4.857020670068935e-07,
      "loss": 0.0355,
      "step": 57470
    },
    {
      "epoch": 6.724783524455885,
      "loss_main": 0.04648374766111374,
      "step": 57470
    },
    {
      "epoch": 6.725953662532179,
      "grad_norm": 0.18421821296215057,
      "learning_rate": 4.818777886381132e-07,
      "loss": 0.0373,
      "step": 57480
    },
    {
      "epoch": 6.725953662532179,
      "loss_main": 0.04251694306731224,
      "step": 57480
    },
    {
      "epoch": 6.727123800608472,
      "grad_norm": 0.19678804278373718,
      "learning_rate": 4.780685525674122e-07,
      "loss": 0.0414,
      "step": 57490
    },
    {
      "epoch": 6.727123800608472,
      "loss_main": 0.04681723937392235,
      "step": 57490
    },
    {
      "epoch": 6.728293938684764,
      "grad_norm": 0.2677786946296692,
      "learning_rate": 4.7427435995193724e-07,
      "loss": 0.0449,
      "step": 57500
    },
    {
      "epoch": 6.728293938684764,
      "loss_main": 0.04002923518419266,
      "step": 57500
    },
    {
      "epoch": 6.729464076761058,
      "grad_norm": 0.18717701733112335,
      "learning_rate": 4.7049521194425515e-07,
      "loss": 0.0428,
      "step": 57510
    },
    {
      "epoch": 6.729464076761058,
      "loss_main": 0.03959884122014046,
      "step": 57510
    },
    {
      "epoch": 6.730634214837351,
      "grad_norm": 0.20890219509601593,
      "learning_rate": 4.667311096923754e-07,
      "loss": 0.0494,
      "step": 57520
    },
    {
      "epoch": 6.730634214837351,
      "loss_main": 0.03396958112716675,
      "step": 57520
    },
    {
      "epoch": 6.7318043529136435,
      "grad_norm": 0.26898786425590515,
      "learning_rate": 4.6298205433973895e-07,
      "loss": 0.0431,
      "step": 57530
    },
    {
      "epoch": 6.7318043529136435,
      "loss_main": 0.041369933634996414,
      "step": 57530
    },
    {
      "epoch": 6.732974490989937,
      "grad_norm": 0.18525058031082153,
      "learning_rate": 4.5924804702520696e-07,
      "loss": 0.0441,
      "step": 57540
    },
    {
      "epoch": 6.732974490989937,
      "loss_main": 0.02046046033501625,
      "step": 57540
    },
    {
      "epoch": 6.73414462906623,
      "grad_norm": 0.22604703903198242,
      "learning_rate": 4.5552908888306655e-07,
      "loss": 0.0509,
      "step": 57550
    },
    {
      "epoch": 6.73414462906623,
      "loss_main": 0.02662920579314232,
      "step": 57550
    },
    {
      "epoch": 6.735314767142523,
      "grad_norm": 0.28085067868232727,
      "learning_rate": 4.5182518104304185e-07,
      "loss": 0.0549,
      "step": 57560
    },
    {
      "epoch": 6.735314767142523,
      "loss_main": 0.05521735921502113,
      "step": 57560
    },
    {
      "epoch": 6.736484905218816,
      "grad_norm": 0.2010784149169922,
      "learning_rate": 4.4813632463028277e-07,
      "loss": 0.0382,
      "step": 57570
    },
    {
      "epoch": 6.736484905218816,
      "loss_main": 0.035382144153118134,
      "step": 57570
    },
    {
      "epoch": 6.737655043295109,
      "grad_norm": 0.21802303194999695,
      "learning_rate": 4.444625207653763e-07,
      "loss": 0.0444,
      "step": 57580
    },
    {
      "epoch": 6.737655043295109,
      "loss_main": 0.035324856638908386,
      "step": 57580
    },
    {
      "epoch": 6.738825181371402,
      "grad_norm": 0.2032438963651657,
      "learning_rate": 4.4080377056430753e-07,
      "loss": 0.0428,
      "step": 57590
    },
    {
      "epoch": 6.738825181371402,
      "loss_main": 0.023672474548220634,
      "step": 57590
    },
    {
      "epoch": 6.7399953194476945,
      "grad_norm": 0.24850663542747498,
      "learning_rate": 4.371600751385263e-07,
      "loss": 0.0394,
      "step": 57600
    },
    {
      "epoch": 6.7399953194476945,
      "loss_main": 0.02369905821979046,
      "step": 57600
    },
    {
      "epoch": 6.741165457523988,
      "grad_norm": 0.19741247594356537,
      "learning_rate": 4.3353143559487495e-07,
      "loss": 0.0405,
      "step": 57610
    },
    {
      "epoch": 6.741165457523988,
      "loss_main": 0.029220368713140488,
      "step": 57610
    },
    {
      "epoch": 6.742335595600281,
      "grad_norm": 0.2947455644607544,
      "learning_rate": 4.2991785303565513e-07,
      "loss": 0.049,
      "step": 57620
    },
    {
      "epoch": 6.742335595600281,
      "loss_main": 0.06703145802021027,
      "step": 57620
    },
    {
      "epoch": 6.743505733676574,
      "grad_norm": 0.2126702517271042,
      "learning_rate": 4.2631932855856647e-07,
      "loss": 0.0388,
      "step": 57630
    },
    {
      "epoch": 6.743505733676574,
      "loss_main": 0.03357241675257683,
      "step": 57630
    },
    {
      "epoch": 6.744675871752867,
      "grad_norm": 0.2717527747154236,
      "learning_rate": 4.2273586325674576e-07,
      "loss": 0.0423,
      "step": 57640
    },
    {
      "epoch": 6.744675871752867,
      "loss_main": 0.036541491746902466,
      "step": 57640
    },
    {
      "epoch": 6.74584600982916,
      "grad_norm": 0.3021056652069092,
      "learning_rate": 4.1916745821876103e-07,
      "loss": 0.0393,
      "step": 57650
    },
    {
      "epoch": 6.74584600982916,
      "loss_main": 0.07346628606319427,
      "step": 57650
    },
    {
      "epoch": 6.747016147905453,
      "grad_norm": 0.20622092485427856,
      "learning_rate": 4.156141145285897e-07,
      "loss": 0.0387,
      "step": 57660
    },
    {
      "epoch": 6.747016147905453,
      "loss_main": 0.04389875382184982,
      "step": 57660
    },
    {
      "epoch": 6.748186285981745,
      "grad_norm": 0.20931081473827362,
      "learning_rate": 4.1207583326566267e-07,
      "loss": 0.0452,
      "step": 57670
    },
    {
      "epoch": 6.748186285981745,
      "loss_main": 0.037168193608522415,
      "step": 57670
    },
    {
      "epoch": 6.749356424058039,
      "grad_norm": 0.16459499299526215,
      "learning_rate": 4.085526155047925e-07,
      "loss": 0.0372,
      "step": 57680
    },
    {
      "epoch": 6.749356424058039,
      "loss_main": 0.039686501026153564,
      "step": 57680
    },
    {
      "epoch": 6.750526562134332,
      "grad_norm": 0.20810359716415405,
      "learning_rate": 4.050444623162564e-07,
      "loss": 0.0439,
      "step": 57690
    },
    {
      "epoch": 6.750526562134332,
      "loss_main": 0.052335407584905624,
      "step": 57690
    },
    {
      "epoch": 6.7516967002106245,
      "grad_norm": 0.21837499737739563,
      "learning_rate": 4.015513747657351e-07,
      "loss": 0.0513,
      "step": 57700
    },
    {
      "epoch": 6.7516967002106245,
      "loss_main": 0.04191117733716965,
      "step": 57700
    },
    {
      "epoch": 6.752866838286918,
      "grad_norm": 0.29630613327026367,
      "learning_rate": 3.9807335391433554e-07,
      "loss": 0.0469,
      "step": 57710
    },
    {
      "epoch": 6.752866838286918,
      "loss_main": 0.030304787680506706,
      "step": 57710
    },
    {
      "epoch": 6.754036976363211,
      "grad_norm": 0.20057906210422516,
      "learning_rate": 3.946104008185847e-07,
      "loss": 0.0404,
      "step": 57720
    },
    {
      "epoch": 6.754036976363211,
      "loss_main": 0.07274655252695084,
      "step": 57720
    },
    {
      "epoch": 6.755207114439504,
      "grad_norm": 0.2358003556728363,
      "learning_rate": 3.9116251653044113e-07,
      "loss": 0.0434,
      "step": 57730
    },
    {
      "epoch": 6.755207114439504,
      "loss_main": 0.057486142963171005,
      "step": 57730
    },
    {
      "epoch": 6.756377252515797,
      "grad_norm": 0.20866382122039795,
      "learning_rate": 3.877297020972781e-07,
      "loss": 0.049,
      "step": 57740
    },
    {
      "epoch": 6.756377252515797,
      "loss_main": 0.0335497185587883,
      "step": 57740
    },
    {
      "epoch": 6.75754739059209,
      "grad_norm": 0.20251643657684326,
      "learning_rate": 3.8431195856190036e-07,
      "loss": 0.0447,
      "step": 57750
    },
    {
      "epoch": 6.75754739059209,
      "loss_main": 0.030270015820860863,
      "step": 57750
    },
    {
      "epoch": 6.758717528668383,
      "grad_norm": 0.23336125910282135,
      "learning_rate": 3.8090928696251635e-07,
      "loss": 0.0369,
      "step": 57760
    },
    {
      "epoch": 6.758717528668383,
      "loss_main": 0.04815429076552391,
      "step": 57760
    },
    {
      "epoch": 6.759887666744676,
      "grad_norm": 0.19844022393226624,
      "learning_rate": 3.77521688332777e-07,
      "loss": 0.0394,
      "step": 57770
    },
    {
      "epoch": 6.759887666744676,
      "loss_main": 0.06325335055589676,
      "step": 57770
    },
    {
      "epoch": 6.761057804820969,
      "grad_norm": 0.20673733949661255,
      "learning_rate": 3.741491637017425e-07,
      "loss": 0.0479,
      "step": 57780
    },
    {
      "epoch": 6.761057804820969,
      "loss_main": 0.05478489026427269,
      "step": 57780
    },
    {
      "epoch": 6.762227942897262,
      "grad_norm": 0.18195274472236633,
      "learning_rate": 3.707917140939043e-07,
      "loss": 0.0448,
      "step": 57790
    },
    {
      "epoch": 6.762227942897262,
      "loss_main": 0.04782803729176521,
      "step": 57790
    },
    {
      "epoch": 6.7633980809735545,
      "grad_norm": 0.18094885349273682,
      "learning_rate": 3.6744934052915235e-07,
      "loss": 0.0432,
      "step": 57800
    },
    {
      "epoch": 6.7633980809735545,
      "loss_main": 0.08940186351537704,
      "step": 57800
    },
    {
      "epoch": 6.764568219049848,
      "grad_norm": 0.2787328064441681,
      "learning_rate": 3.641220440228188e-07,
      "loss": 0.0398,
      "step": 57810
    },
    {
      "epoch": 6.764568219049848,
      "loss_main": 0.026347815990447998,
      "step": 57810
    },
    {
      "epoch": 6.765738357126141,
      "grad_norm": 0.3147194981575012,
      "learning_rate": 3.608098255856562e-07,
      "loss": 0.0459,
      "step": 57820
    },
    {
      "epoch": 6.765738357126141,
      "loss_main": 0.03920043259859085,
      "step": 57820
    },
    {
      "epoch": 6.766908495202434,
      "grad_norm": 0.24364088475704193,
      "learning_rate": 3.575126862238154e-07,
      "loss": 0.0405,
      "step": 57830
    },
    {
      "epoch": 6.766908495202434,
      "loss_main": 0.057341478765010834,
      "step": 57830
    },
    {
      "epoch": 6.768078633278727,
      "grad_norm": 0.32644590735435486,
      "learning_rate": 3.5423062693888955e-07,
      "loss": 0.0498,
      "step": 57840
    },
    {
      "epoch": 6.768078633278727,
      "loss_main": 0.032519709318876266,
      "step": 57840
    },
    {
      "epoch": 6.76924877135502,
      "grad_norm": 0.2872423231601715,
      "learning_rate": 3.509636487278756e-07,
      "loss": 0.0533,
      "step": 57850
    },
    {
      "epoch": 6.76924877135502,
      "loss_main": 0.04419049248099327,
      "step": 57850
    },
    {
      "epoch": 6.770418909431313,
      "grad_norm": 0.5813325643539429,
      "learning_rate": 3.4771175258320186e-07,
      "loss": 0.0447,
      "step": 57860
    },
    {
      "epoch": 6.770418909431313,
      "loss_main": 0.04117153212428093,
      "step": 57860
    },
    {
      "epoch": 6.7715890475076055,
      "grad_norm": 0.2740628719329834,
      "learning_rate": 3.4447493949270047e-07,
      "loss": 0.0369,
      "step": 57870
    },
    {
      "epoch": 6.7715890475076055,
      "loss_main": 0.03582604601979256,
      "step": 57870
    },
    {
      "epoch": 6.772759185583899,
      "grad_norm": 0.1954294741153717,
      "learning_rate": 3.4125321043964045e-07,
      "loss": 0.0406,
      "step": 57880
    },
    {
      "epoch": 6.772759185583899,
      "loss_main": 0.0483248308300972,
      "step": 57880
    },
    {
      "epoch": 6.773929323660192,
      "grad_norm": 0.19537602365016937,
      "learning_rate": 3.380465664026833e-07,
      "loss": 0.0385,
      "step": 57890
    },
    {
      "epoch": 6.773929323660192,
      "loss_main": 0.07206448167562485,
      "step": 57890
    },
    {
      "epoch": 6.775099461736485,
      "grad_norm": 0.16174739599227905,
      "learning_rate": 3.348550083559388e-07,
      "loss": 0.0454,
      "step": 57900
    },
    {
      "epoch": 6.775099461736485,
      "loss_main": 0.0423399992287159,
      "step": 57900
    },
    {
      "epoch": 6.776269599812778,
      "grad_norm": 0.18782445788383484,
      "learning_rate": 3.316785372689091e-07,
      "loss": 0.0358,
      "step": 57910
    },
    {
      "epoch": 6.776269599812778,
      "loss_main": 0.061681170016527176,
      "step": 57910
    },
    {
      "epoch": 6.777439737889071,
      "grad_norm": 0.16065114736557007,
      "learning_rate": 3.285171541065224e-07,
      "loss": 0.0434,
      "step": 57920
    },
    {
      "epoch": 6.777439737889071,
      "loss_main": 0.03828030079603195,
      "step": 57920
    },
    {
      "epoch": 6.778609875965364,
      "grad_norm": 0.14010637998580933,
      "learning_rate": 3.253708598291272e-07,
      "loss": 0.0371,
      "step": 57930
    },
    {
      "epoch": 6.778609875965364,
      "loss_main": 0.04899626970291138,
      "step": 57930
    },
    {
      "epoch": 6.779780014041657,
      "grad_norm": 0.33456951379776,
      "learning_rate": 3.2223965539248116e-07,
      "loss": 0.0452,
      "step": 57940
    },
    {
      "epoch": 6.779780014041657,
      "loss_main": 0.035904765129089355,
      "step": 57940
    },
    {
      "epoch": 6.78095015211795,
      "grad_norm": 0.31966689229011536,
      "learning_rate": 3.1912354174776227e-07,
      "loss": 0.0424,
      "step": 57950
    },
    {
      "epoch": 6.78095015211795,
      "loss_main": 0.025654302909970284,
      "step": 57950
    },
    {
      "epoch": 6.782120290194243,
      "grad_norm": 0.28026124835014343,
      "learning_rate": 3.1602251984155786e-07,
      "loss": 0.0389,
      "step": 57960
    },
    {
      "epoch": 6.782120290194243,
      "loss_main": 0.056938376277685165,
      "step": 57960
    },
    {
      "epoch": 6.783290428270536,
      "grad_norm": 0.3159804046154022,
      "learning_rate": 3.1293659061589207e-07,
      "loss": 0.0443,
      "step": 57970
    },
    {
      "epoch": 6.783290428270536,
      "loss_main": 0.07204627245664597,
      "step": 57970
    },
    {
      "epoch": 6.784460566346829,
      "grad_norm": 0.33029937744140625,
      "learning_rate": 3.098657550081707e-07,
      "loss": 0.0596,
      "step": 57980
    },
    {
      "epoch": 6.784460566346829,
      "loss_main": 0.06435991078615189,
      "step": 57980
    },
    {
      "epoch": 6.785630704423122,
      "grad_norm": 0.1777835190296173,
      "learning_rate": 3.068100139512475e-07,
      "loss": 0.0382,
      "step": 57990
    },
    {
      "epoch": 6.785630704423122,
      "loss_main": 0.04697578400373459,
      "step": 57990
    },
    {
      "epoch": 6.786800842499415,
      "grad_norm": 0.21248705685138702,
      "learning_rate": 3.037693683733689e-07,
      "loss": 0.0387,
      "step": 58000
    },
    {
      "epoch": 6.786800842499415,
      "loss_main": 0.06472598761320114,
      "step": 58000
    },
    {
      "epoch": 6.787970980575708,
      "grad_norm": 0.24361269176006317,
      "learning_rate": 3.007438191982015e-07,
      "loss": 0.0563,
      "step": 58010
    },
    {
      "epoch": 6.787970980575708,
      "loss_main": 0.030688583850860596,
      "step": 58010
    },
    {
      "epoch": 6.789141118652001,
      "grad_norm": 0.2003895789384842,
      "learning_rate": 2.9773336734482684e-07,
      "loss": 0.0378,
      "step": 58020
    },
    {
      "epoch": 6.789141118652001,
      "loss_main": 0.04418399557471275,
      "step": 58020
    },
    {
      "epoch": 6.790311256728294,
      "grad_norm": 0.18794673681259155,
      "learning_rate": 2.9473801372774667e-07,
      "loss": 0.0456,
      "step": 58030
    },
    {
      "epoch": 6.790311256728294,
      "loss_main": 0.03634397312998772,
      "step": 58030
    },
    {
      "epoch": 6.7914813948045865,
      "grad_norm": 0.20183882117271423,
      "learning_rate": 2.91757759256861e-07,
      "loss": 0.0385,
      "step": 58040
    },
    {
      "epoch": 6.7914813948045865,
      "loss_main": 0.032847072929143906,
      "step": 58040
    },
    {
      "epoch": 6.79265153288088,
      "grad_norm": 0.25780799984931946,
      "learning_rate": 2.887926048375067e-07,
      "loss": 0.0403,
      "step": 58050
    },
    {
      "epoch": 6.79265153288088,
      "loss_main": 0.024770358577370644,
      "step": 58050
    },
    {
      "epoch": 6.793821670957173,
      "grad_norm": 0.287058025598526,
      "learning_rate": 2.858425513704022e-07,
      "loss": 0.0421,
      "step": 58060
    },
    {
      "epoch": 6.793821670957173,
      "loss_main": 0.04472574219107628,
      "step": 58060
    },
    {
      "epoch": 6.794991809033466,
      "grad_norm": 0.2829890847206116,
      "learning_rate": 2.8290759975170834e-07,
      "loss": 0.0467,
      "step": 58070
    },
    {
      "epoch": 6.794991809033466,
      "loss_main": 0.04454394802451134,
      "step": 58070
    },
    {
      "epoch": 6.796161947109759,
      "grad_norm": 0.31453368067741394,
      "learning_rate": 2.799877508729787e-07,
      "loss": 0.0468,
      "step": 58080
    },
    {
      "epoch": 6.796161947109759,
      "loss_main": 0.03338683024048805,
      "step": 58080
    },
    {
      "epoch": 6.797332085186052,
      "grad_norm": 0.34649720788002014,
      "learning_rate": 2.770830056211926e-07,
      "loss": 0.0431,
      "step": 58090
    },
    {
      "epoch": 6.797332085186052,
      "loss_main": 0.04024568572640419,
      "step": 58090
    },
    {
      "epoch": 6.798502223262345,
      "grad_norm": 0.14444532990455627,
      "learning_rate": 2.741933648787331e-07,
      "loss": 0.0382,
      "step": 58100
    },
    {
      "epoch": 6.798502223262345,
      "loss_main": 0.06446342915296555,
      "step": 58100
    },
    {
      "epoch": 6.799672361338638,
      "grad_norm": 0.19617542624473572,
      "learning_rate": 2.7131882952339263e-07,
      "loss": 0.0431,
      "step": 58110
    },
    {
      "epoch": 6.799672361338638,
      "loss_main": 0.07862614095211029,
      "step": 58110
    },
    {
      "epoch": 6.800842499414931,
      "grad_norm": 0.2312656193971634,
      "learning_rate": 2.684594004283836e-07,
      "loss": 0.044,
      "step": 58120
    },
    {
      "epoch": 6.800842499414931,
      "loss_main": 0.032888565212488174,
      "step": 58120
    },
    {
      "epoch": 6.802012637491224,
      "grad_norm": 0.15371491014957428,
      "learning_rate": 2.6561507846232234e-07,
      "loss": 0.0433,
      "step": 58130
    },
    {
      "epoch": 6.802012637491224,
      "loss_main": 0.033214639872312546,
      "step": 58130
    },
    {
      "epoch": 6.803182775567517,
      "grad_norm": 0.2193102240562439,
      "learning_rate": 2.6278586448924005e-07,
      "loss": 0.0397,
      "step": 58140
    },
    {
      "epoch": 6.803182775567517,
      "loss_main": 0.025894546881318092,
      "step": 58140
    },
    {
      "epoch": 6.80435291364381,
      "grad_norm": 0.2574045658111572,
      "learning_rate": 2.5997175936857685e-07,
      "loss": 0.0368,
      "step": 58150
    },
    {
      "epoch": 6.80435291364381,
      "loss_main": 0.03239430859684944,
      "step": 58150
    },
    {
      "epoch": 6.805523051720103,
      "grad_norm": 0.18252462148666382,
      "learning_rate": 2.57172763955188e-07,
      "loss": 0.0448,
      "step": 58160
    },
    {
      "epoch": 6.805523051720103,
      "loss_main": 0.03821846470236778,
      "step": 58160
    },
    {
      "epoch": 6.806693189796396,
      "grad_norm": 0.15802662074565887,
      "learning_rate": 2.543888790993265e-07,
      "loss": 0.0552,
      "step": 58170
    },
    {
      "epoch": 6.806693189796396,
      "loss_main": 0.06678114086389542,
      "step": 58170
    },
    {
      "epoch": 6.807863327872689,
      "grad_norm": 0.23265571892261505,
      "learning_rate": 2.5162010564666607e-07,
      "loss": 0.047,
      "step": 58180
    },
    {
      "epoch": 6.807863327872689,
      "loss_main": 0.052787814289331436,
      "step": 58180
    },
    {
      "epoch": 6.809033465948982,
      "grad_norm": 0.29654136300086975,
      "learning_rate": 2.488664444382893e-07,
      "loss": 0.0435,
      "step": 58190
    },
    {
      "epoch": 6.809033465948982,
      "loss_main": 0.08209448307752609,
      "step": 58190
    },
    {
      "epoch": 6.810203604025275,
      "grad_norm": 0.28578799962997437,
      "learning_rate": 2.461278963106828e-07,
      "loss": 0.0428,
      "step": 58200
    },
    {
      "epoch": 6.810203604025275,
      "loss_main": 0.04062135890126228,
      "step": 58200
    },
    {
      "epoch": 6.811373742101568,
      "grad_norm": 0.2626962959766388,
      "learning_rate": 2.434044620957421e-07,
      "loss": 0.048,
      "step": 58210
    },
    {
      "epoch": 6.811373742101568,
      "loss_main": 0.030882660299539566,
      "step": 58210
    },
    {
      "epoch": 6.812543880177861,
      "grad_norm": 0.29440838098526,
      "learning_rate": 2.406961426207832e-07,
      "loss": 0.042,
      "step": 58220
    },
    {
      "epoch": 6.812543880177861,
      "loss_main": 0.039530713111162186,
      "step": 58220
    },
    {
      "epoch": 6.813714018254154,
      "grad_norm": 0.33297479152679443,
      "learning_rate": 2.380029387085203e-07,
      "loss": 0.0421,
      "step": 58230
    },
    {
      "epoch": 6.813714018254154,
      "loss_main": 0.042465291917324066,
      "step": 58230
    },
    {
      "epoch": 6.8148841563304465,
      "grad_norm": 0.25439751148223877,
      "learning_rate": 2.353248511770767e-07,
      "loss": 0.0443,
      "step": 58240
    },
    {
      "epoch": 6.8148841563304465,
      "loss_main": 0.03339508920907974,
      "step": 58240
    },
    {
      "epoch": 6.81605429440674,
      "grad_norm": 0.20273202657699585,
      "learning_rate": 2.3266188083997942e-07,
      "loss": 0.0431,
      "step": 58250
    },
    {
      "epoch": 6.81605429440674,
      "loss_main": 0.0443122461438179,
      "step": 58250
    },
    {
      "epoch": 6.817224432483033,
      "grad_norm": 0.17333389818668365,
      "learning_rate": 2.3001402850617027e-07,
      "loss": 0.0346,
      "step": 58260
    },
    {
      "epoch": 6.817224432483033,
      "loss_main": 0.017864732071757317,
      "step": 58260
    },
    {
      "epoch": 6.818394570559326,
      "grad_norm": 0.19754402339458466,
      "learning_rate": 2.2738129498000581e-07,
      "loss": 0.0452,
      "step": 58270
    },
    {
      "epoch": 6.818394570559326,
      "loss_main": 0.03735210373997688,
      "step": 58270
    },
    {
      "epoch": 6.819564708635619,
      "grad_norm": 0.27920153737068176,
      "learning_rate": 2.2476368106122414e-07,
      "loss": 0.0421,
      "step": 58280
    },
    {
      "epoch": 6.819564708635619,
      "loss_main": 0.03367963060736656,
      "step": 58280
    },
    {
      "epoch": 6.820734846711912,
      "grad_norm": 0.2211390733718872,
      "learning_rate": 2.2216118754500582e-07,
      "loss": 0.0392,
      "step": 58290
    },
    {
      "epoch": 6.820734846711912,
      "loss_main": 0.041473567485809326,
      "step": 58290
    },
    {
      "epoch": 6.821904984788205,
      "grad_norm": 0.18321150541305542,
      "learning_rate": 2.1957381522190735e-07,
      "loss": 0.0429,
      "step": 58300
    },
    {
      "epoch": 6.821904984788205,
      "loss_main": 0.02056374028325081,
      "step": 58300
    },
    {
      "epoch": 6.823075122864498,
      "grad_norm": 0.2232523262500763,
      "learning_rate": 2.1700156487790558e-07,
      "loss": 0.0443,
      "step": 58310
    },
    {
      "epoch": 6.823075122864498,
      "loss_main": 0.04180116206407547,
      "step": 58310
    },
    {
      "epoch": 6.824245260940791,
      "grad_norm": 0.23167426884174347,
      "learning_rate": 2.1444443729439213e-07,
      "loss": 0.0402,
      "step": 58320
    },
    {
      "epoch": 6.824245260940791,
      "loss_main": 0.040251653641462326,
      "step": 58320
    },
    {
      "epoch": 6.825415399017084,
      "grad_norm": 0.24439255893230438,
      "learning_rate": 2.1190243324814007e-07,
      "loss": 0.0384,
      "step": 58330
    },
    {
      "epoch": 6.825415399017084,
      "loss_main": 0.04444345459342003,
      "step": 58330
    },
    {
      "epoch": 6.8265855370933775,
      "grad_norm": 0.20207059383392334,
      "learning_rate": 2.0937555351135395e-07,
      "loss": 0.0429,
      "step": 58340
    },
    {
      "epoch": 6.8265855370933775,
      "loss_main": 0.07940949499607086,
      "step": 58340
    },
    {
      "epoch": 6.82775567516967,
      "grad_norm": 0.2237802892923355,
      "learning_rate": 2.0686379885162532e-07,
      "loss": 0.0491,
      "step": 58350
    },
    {
      "epoch": 6.82775567516967,
      "loss_main": 0.03673083335161209,
      "step": 58350
    },
    {
      "epoch": 6.828925813245963,
      "grad_norm": 0.23809531331062317,
      "learning_rate": 2.0436717003196604e-07,
      "loss": 0.043,
      "step": 58360
    },
    {
      "epoch": 6.828925813245963,
      "loss_main": 0.025119204074144363,
      "step": 58360
    },
    {
      "epoch": 6.830095951322256,
      "grad_norm": 0.2635498642921448,
      "learning_rate": 2.0188566781078057e-07,
      "loss": 0.0377,
      "step": 58370
    },
    {
      "epoch": 6.830095951322256,
      "loss_main": 0.0482640266418457,
      "step": 58370
    },
    {
      "epoch": 6.831266089398549,
      "grad_norm": 0.1851283311843872,
      "learning_rate": 1.994192929418881e-07,
      "loss": 0.0438,
      "step": 58380
    },
    {
      "epoch": 6.831266089398549,
      "loss_main": 0.08032483607530594,
      "step": 58380
    },
    {
      "epoch": 6.832436227474842,
      "grad_norm": 0.14384712278842926,
      "learning_rate": 1.9696804617451158e-07,
      "loss": 0.0498,
      "step": 58390
    },
    {
      "epoch": 6.832436227474842,
      "loss_main": 0.040922973304986954,
      "step": 58390
    },
    {
      "epoch": 6.833606365551135,
      "grad_norm": 0.24547407031059265,
      "learning_rate": 1.9453192825326093e-07,
      "loss": 0.0438,
      "step": 58400
    },
    {
      "epoch": 6.833606365551135,
      "loss_main": 0.0413837656378746,
      "step": 58400
    },
    {
      "epoch": 6.834776503627428,
      "grad_norm": 0.1816708892583847,
      "learning_rate": 1.9211093991817753e-07,
      "loss": 0.0415,
      "step": 58410
    },
    {
      "epoch": 6.834776503627428,
      "loss_main": 0.04067031666636467,
      "step": 58410
    },
    {
      "epoch": 6.835946641703721,
      "grad_norm": 0.24364130198955536,
      "learning_rate": 1.8970508190468973e-07,
      "loss": 0.0406,
      "step": 58420
    },
    {
      "epoch": 6.835946641703721,
      "loss_main": 0.025909796357154846,
      "step": 58420
    },
    {
      "epoch": 6.837116779780014,
      "grad_norm": 0.2721261978149414,
      "learning_rate": 1.8731435494362958e-07,
      "loss": 0.0448,
      "step": 58430
    },
    {
      "epoch": 6.837116779780014,
      "loss_main": 0.03692946210503578,
      "step": 58430
    },
    {
      "epoch": 6.838286917856307,
      "grad_norm": 0.23637761175632477,
      "learning_rate": 1.849387597612495e-07,
      "loss": 0.0532,
      "step": 58440
    },
    {
      "epoch": 6.838286917856307,
      "loss_main": 0.03776165843009949,
      "step": 58440
    },
    {
      "epoch": 6.8394570559326,
      "grad_norm": 0.19011740386486053,
      "learning_rate": 1.8257829707917228e-07,
      "loss": 0.0428,
      "step": 58450
    },
    {
      "epoch": 6.8394570559326,
      "loss_main": 0.0418856143951416,
      "step": 58450
    },
    {
      "epoch": 6.840627194008893,
      "grad_norm": 0.238448828458786,
      "learning_rate": 1.8023296761446317e-07,
      "loss": 0.0408,
      "step": 58460
    },
    {
      "epoch": 6.840627194008893,
      "loss_main": 0.03340263292193413,
      "step": 58460
    },
    {
      "epoch": 6.841797332085186,
      "grad_norm": 0.21167027950286865,
      "learning_rate": 1.7790277207956341e-07,
      "loss": 0.0427,
      "step": 58470
    },
    {
      "epoch": 6.841797332085186,
      "loss_main": 0.03860281780362129,
      "step": 58470
    },
    {
      "epoch": 6.842967470161479,
      "grad_norm": 0.19213810563087463,
      "learning_rate": 1.7558771118232343e-07,
      "loss": 0.0474,
      "step": 58480
    },
    {
      "epoch": 6.842967470161479,
      "loss_main": 0.03764893487095833,
      "step": 58480
    },
    {
      "epoch": 6.844137608237772,
      "grad_norm": 0.20131345093250275,
      "learning_rate": 1.7328778562599734e-07,
      "loss": 0.0387,
      "step": 58490
    },
    {
      "epoch": 6.844137608237772,
      "loss_main": 0.057817522436380386,
      "step": 58490
    },
    {
      "epoch": 6.845307746314065,
      "grad_norm": 0.23955482244491577,
      "learning_rate": 1.7100299610924298e-07,
      "loss": 0.0404,
      "step": 58500
    },
    {
      "epoch": 6.845307746314065,
      "loss_main": 0.025594741106033325,
      "step": 58500
    },
    {
      "epoch": 6.846477884390358,
      "grad_norm": 0.43389034271240234,
      "learning_rate": 1.6873334332612733e-07,
      "loss": 0.0487,
      "step": 58510
    },
    {
      "epoch": 6.846477884390358,
      "loss_main": 0.03840181976556778,
      "step": 58510
    },
    {
      "epoch": 6.847648022466651,
      "grad_norm": 0.2719351053237915,
      "learning_rate": 1.6647882796609894e-07,
      "loss": 0.0427,
      "step": 58520
    },
    {
      "epoch": 6.847648022466651,
      "loss_main": 0.04049067944288254,
      "step": 58520
    },
    {
      "epoch": 6.848818160542944,
      "grad_norm": 0.17306387424468994,
      "learning_rate": 1.6423945071402102e-07,
      "loss": 0.0519,
      "step": 58530
    },
    {
      "epoch": 6.848818160542944,
      "loss_main": 0.02373923733830452,
      "step": 58530
    },
    {
      "epoch": 6.8499882986192375,
      "grad_norm": 0.23092710971832275,
      "learning_rate": 1.6201521225016614e-07,
      "loss": 0.0431,
      "step": 58540
    },
    {
      "epoch": 6.8499882986192375,
      "loss_main": 0.04983801394701004,
      "step": 58540
    },
    {
      "epoch": 6.85115843669553,
      "grad_norm": 0.3078177869319916,
      "learning_rate": 1.598061132501938e-07,
      "loss": 0.0457,
      "step": 58550
    },
    {
      "epoch": 6.85115843669553,
      "loss_main": 0.06369919329881668,
      "step": 58550
    },
    {
      "epoch": 6.852328574771823,
      "grad_norm": 0.2218804657459259,
      "learning_rate": 1.576121543851672e-07,
      "loss": 0.0416,
      "step": 58560
    },
    {
      "epoch": 6.852328574771823,
      "loss_main": 0.045933958142995834,
      "step": 58560
    },
    {
      "epoch": 6.853498712848116,
      "grad_norm": 0.1830814778804779,
      "learning_rate": 1.5543333632155876e-07,
      "loss": 0.0491,
      "step": 58570
    },
    {
      "epoch": 6.853498712848116,
      "loss_main": 0.036934249103069305,
      "step": 58570
    },
    {
      "epoch": 6.854668850924409,
      "grad_norm": 0.28874337673187256,
      "learning_rate": 1.5326965972123352e-07,
      "loss": 0.0423,
      "step": 58580
    },
    {
      "epoch": 6.854668850924409,
      "loss_main": 0.04062561318278313,
      "step": 58580
    },
    {
      "epoch": 6.855838989000702,
      "grad_norm": 0.25611183047294617,
      "learning_rate": 1.5112112524146016e-07,
      "loss": 0.0429,
      "step": 58590
    },
    {
      "epoch": 6.855838989000702,
      "loss_main": 0.03174929693341255,
      "step": 58590
    },
    {
      "epoch": 6.857009127076995,
      "grad_norm": 0.23558872938156128,
      "learning_rate": 1.4898773353489992e-07,
      "loss": 0.0387,
      "step": 58600
    },
    {
      "epoch": 6.857009127076995,
      "loss_main": 0.025441229343414307,
      "step": 58600
    },
    {
      "epoch": 6.8581792651532885,
      "grad_norm": 0.1604454219341278,
      "learning_rate": 1.4686948524962884e-07,
      "loss": 0.0501,
      "step": 58610
    },
    {
      "epoch": 6.8581792651532885,
      "loss_main": 0.04220980033278465,
      "step": 58610
    },
    {
      "epoch": 6.859349403229581,
      "grad_norm": 0.16815730929374695,
      "learning_rate": 1.4476638102911556e-07,
      "loss": 0.0374,
      "step": 58620
    },
    {
      "epoch": 6.859349403229581,
      "loss_main": 0.034285541623830795,
      "step": 58620
    },
    {
      "epoch": 6.860519541305874,
      "grad_norm": 0.4394691586494446,
      "learning_rate": 1.4267842151222123e-07,
      "loss": 0.0484,
      "step": 58630
    },
    {
      "epoch": 6.860519541305874,
      "loss_main": 0.03355156257748604,
      "step": 58630
    },
    {
      "epoch": 6.861689679382167,
      "grad_norm": 0.34306633472442627,
      "learning_rate": 1.4060560733321626e-07,
      "loss": 0.0374,
      "step": 58640
    },
    {
      "epoch": 6.861689679382167,
      "loss_main": 0.03652369976043701,
      "step": 58640
    },
    {
      "epoch": 6.86285981745846,
      "grad_norm": 0.2293664664030075,
      "learning_rate": 1.385479391217692e-07,
      "loss": 0.0528,
      "step": 58650
    },
    {
      "epoch": 6.86285981745846,
      "loss_main": 0.053571972995996475,
      "step": 58650
    },
    {
      "epoch": 6.864029955534753,
      "grad_norm": 0.25171035528182983,
      "learning_rate": 1.365054175029412e-07,
      "loss": 0.0451,
      "step": 58660
    },
    {
      "epoch": 6.864029955534753,
      "loss_main": 0.036821626126766205,
      "step": 58660
    },
    {
      "epoch": 6.865200093611046,
      "grad_norm": 0.23434467613697052,
      "learning_rate": 1.3447804309719702e-07,
      "loss": 0.0432,
      "step": 58670
    },
    {
      "epoch": 6.865200093611046,
      "loss_main": 0.05450088158249855,
      "step": 58670
    },
    {
      "epoch": 6.866370231687339,
      "grad_norm": 0.28648874163627625,
      "learning_rate": 1.3246581652040512e-07,
      "loss": 0.0449,
      "step": 58680
    },
    {
      "epoch": 6.866370231687339,
      "loss_main": 0.049131061881780624,
      "step": 58680
    },
    {
      "epoch": 6.867540369763632,
      "grad_norm": 0.18071229755878448,
      "learning_rate": 1.3046873838381546e-07,
      "loss": 0.0438,
      "step": 58690
    },
    {
      "epoch": 6.867540369763632,
      "loss_main": 0.05988316982984543,
      "step": 58690
    },
    {
      "epoch": 6.868710507839925,
      "grad_norm": 0.29385894536972046,
      "learning_rate": 1.2848680929409828e-07,
      "loss": 0.0505,
      "step": 58700
    },
    {
      "epoch": 6.868710507839925,
      "loss_main": 0.03153654560446739,
      "step": 58700
    },
    {
      "epoch": 6.8698806459162185,
      "grad_norm": 0.1502852588891983,
      "learning_rate": 1.2652002985331091e-07,
      "loss": 0.0391,
      "step": 58710
    },
    {
      "epoch": 6.8698806459162185,
      "loss_main": 0.0288422591984272,
      "step": 58710
    },
    {
      "epoch": 6.871050783992511,
      "grad_norm": 0.22470365464687347,
      "learning_rate": 1.2456840065889764e-07,
      "loss": 0.0406,
      "step": 58720
    },
    {
      "epoch": 6.871050783992511,
      "loss_main": 0.05785340070724487,
      "step": 58720
    },
    {
      "epoch": 6.872220922068804,
      "grad_norm": 0.21429425477981567,
      "learning_rate": 1.226319223037231e-07,
      "loss": 0.0388,
      "step": 58730
    },
    {
      "epoch": 6.872220922068804,
      "loss_main": 0.03663031384348869,
      "step": 58730
    },
    {
      "epoch": 6.873391060145098,
      "grad_norm": 0.22257955372333527,
      "learning_rate": 1.2071059537603902e-07,
      "loss": 0.0395,
      "step": 58740
    },
    {
      "epoch": 6.873391060145098,
      "loss_main": 0.06307495385408401,
      "step": 58740
    },
    {
      "epoch": 6.87456119822139,
      "grad_norm": 0.19148322939872742,
      "learning_rate": 1.1880442045948403e-07,
      "loss": 0.0407,
      "step": 58750
    },
    {
      "epoch": 6.87456119822139,
      "loss_main": 0.026787471026182175,
      "step": 58750
    },
    {
      "epoch": 6.875731336297683,
      "grad_norm": 0.21313023567199707,
      "learning_rate": 1.1691339813311164e-07,
      "loss": 0.0426,
      "step": 58760
    },
    {
      "epoch": 6.875731336297683,
      "loss_main": 0.041425563395023346,
      "step": 58760
    },
    {
      "epoch": 6.876901474373976,
      "grad_norm": 0.25114086270332336,
      "learning_rate": 1.1503752897136233e-07,
      "loss": 0.0471,
      "step": 58770
    },
    {
      "epoch": 6.876901474373976,
      "loss_main": 0.06560047715902328,
      "step": 58770
    },
    {
      "epoch": 6.878071612450269,
      "grad_norm": 0.29817458987236023,
      "learning_rate": 1.1317681354407472e-07,
      "loss": 0.0446,
      "step": 58780
    },
    {
      "epoch": 6.878071612450269,
      "loss_main": 0.032600678503513336,
      "step": 58780
    },
    {
      "epoch": 6.879241750526562,
      "grad_norm": 0.23644790053367615,
      "learning_rate": 1.1133125241649111e-07,
      "loss": 0.0487,
      "step": 58790
    },
    {
      "epoch": 6.879241750526562,
      "loss_main": 0.02671809308230877,
      "step": 58790
    },
    {
      "epoch": 6.880411888602855,
      "grad_norm": 0.23128913342952728,
      "learning_rate": 1.0950084614922973e-07,
      "loss": 0.0449,
      "step": 58800
    },
    {
      "epoch": 6.880411888602855,
      "loss_main": 0.02659730054438114,
      "step": 58800
    },
    {
      "epoch": 6.881582026679148,
      "grad_norm": 0.221648171544075,
      "learning_rate": 1.0768559529834021e-07,
      "loss": 0.0413,
      "step": 58810
    },
    {
      "epoch": 6.881582026679148,
      "loss_main": 0.03826596587896347,
      "step": 58810
    },
    {
      "epoch": 6.882752164755441,
      "grad_norm": 0.2991064190864563,
      "learning_rate": 1.0588550041522594e-07,
      "loss": 0.0425,
      "step": 58820
    },
    {
      "epoch": 6.882752164755441,
      "loss_main": 0.02982831373810768,
      "step": 58820
    },
    {
      "epoch": 6.883922302831734,
      "grad_norm": 0.23129777610301971,
      "learning_rate": 1.0410056204672169e-07,
      "loss": 0.0415,
      "step": 58830
    },
    {
      "epoch": 6.883922302831734,
      "loss_main": 0.05298510938882828,
      "step": 58830
    },
    {
      "epoch": 6.885092440908027,
      "grad_norm": 0.3322030305862427,
      "learning_rate": 1.0233078073504376e-07,
      "loss": 0.0416,
      "step": 58840
    },
    {
      "epoch": 6.885092440908027,
      "loss_main": 0.05125567689538002,
      "step": 58840
    },
    {
      "epoch": 6.88626257898432,
      "grad_norm": 0.13654178380966187,
      "learning_rate": 1.0057615701780654e-07,
      "loss": 0.0393,
      "step": 58850
    },
    {
      "epoch": 6.88626257898432,
      "loss_main": 0.03154322877526283,
      "step": 58850
    },
    {
      "epoch": 6.887432717060613,
      "grad_norm": 0.18607120215892792,
      "learning_rate": 9.883669142801144e-08,
      "loss": 0.0453,
      "step": 58860
    },
    {
      "epoch": 6.887432717060613,
      "loss_main": 0.029649877920746803,
      "step": 58860
    },
    {
      "epoch": 6.888602855136906,
      "grad_norm": 0.28653857111930847,
      "learning_rate": 9.711238449406356e-08,
      "loss": 0.0435,
      "step": 58870
    },
    {
      "epoch": 6.888602855136906,
      "loss_main": 0.0506863109767437,
      "step": 58870
    },
    {
      "epoch": 6.8897729932131995,
      "grad_norm": 0.18310239911079407,
      "learning_rate": 9.540323673976614e-08,
      "loss": 0.0402,
      "step": 58880
    },
    {
      "epoch": 6.8897729932131995,
      "loss_main": 0.04076741263270378,
      "step": 58880
    },
    {
      "epoch": 6.890943131289492,
      "grad_norm": 0.25813063979148865,
      "learning_rate": 9.370924868430942e-08,
      "loss": 0.0431,
      "step": 58890
    },
    {
      "epoch": 6.890943131289492,
      "loss_main": 0.07150758802890778,
      "step": 58890
    },
    {
      "epoch": 6.892113269365785,
      "grad_norm": 0.619204044342041,
      "learning_rate": 9.203042084228175e-08,
      "loss": 0.0464,
      "step": 58900
    },
    {
      "epoch": 6.892113269365785,
      "loss_main": 0.03727731108665466,
      "step": 58900
    },
    {
      "epoch": 6.893283407442079,
      "grad_norm": 0.1977115273475647,
      "learning_rate": 9.036675372366965e-08,
      "loss": 0.0465,
      "step": 58910
    },
    {
      "epoch": 6.893283407442079,
      "loss_main": 0.05136105790734291,
      "step": 58910
    },
    {
      "epoch": 6.894453545518371,
      "grad_norm": 0.2348070591688156,
      "learning_rate": 8.871824783385218e-08,
      "loss": 0.0477,
      "step": 58920
    },
    {
      "epoch": 6.894453545518371,
      "loss_main": 0.055946435779333115,
      "step": 58920
    },
    {
      "epoch": 6.895623683594664,
      "grad_norm": 0.26244479417800903,
      "learning_rate": 8.7084903673601e-08,
      "loss": 0.0439,
      "step": 58930
    },
    {
      "epoch": 6.895623683594664,
      "loss_main": 0.037355244159698486,
      "step": 58930
    },
    {
      "epoch": 6.896793821670958,
      "grad_norm": 0.24243827164173126,
      "learning_rate": 8.546672173908032e-08,
      "loss": 0.0373,
      "step": 58940
    },
    {
      "epoch": 6.896793821670958,
      "loss_main": 0.05006418749690056,
      "step": 58940
    },
    {
      "epoch": 6.89796395974725,
      "grad_norm": 0.1609877645969391,
      "learning_rate": 8.386370252185249e-08,
      "loss": 0.0417,
      "step": 58950
    },
    {
      "epoch": 6.89796395974725,
      "loss_main": 0.047027137130498886,
      "step": 58950
    },
    {
      "epoch": 6.899134097823543,
      "grad_norm": 0.16973085701465607,
      "learning_rate": 8.227584650887243e-08,
      "loss": 0.039,
      "step": 58960
    },
    {
      "epoch": 6.899134097823543,
      "loss_main": 0.062270671129226685,
      "step": 58960
    },
    {
      "epoch": 6.900304235899836,
      "grad_norm": 0.22160804271697998,
      "learning_rate": 8.070315418249319e-08,
      "loss": 0.039,
      "step": 58970
    },
    {
      "epoch": 6.900304235899836,
      "loss_main": 0.04892272874712944,
      "step": 58970
    },
    {
      "epoch": 6.9014743739761295,
      "grad_norm": 0.34856709837913513,
      "learning_rate": 7.914562602044929e-08,
      "loss": 0.0461,
      "step": 58980
    },
    {
      "epoch": 6.9014743739761295,
      "loss_main": 0.03199129179120064,
      "step": 58980
    },
    {
      "epoch": 6.902644512052422,
      "grad_norm": 0.27765804529190063,
      "learning_rate": 7.76032624958789e-08,
      "loss": 0.0456,
      "step": 58990
    },
    {
      "epoch": 6.902644512052422,
      "loss_main": 0.03259137645363808,
      "step": 58990
    },
    {
      "epoch": 6.903814650128715,
      "grad_norm": 0.24740096926689148,
      "learning_rate": 7.607606407731282e-08,
      "loss": 0.0398,
      "step": 59000
    },
    {
      "epoch": 6.903814650128715,
      "loss_main": 0.03413853049278259,
      "step": 59000
    },
    {
      "epoch": 6.904984788205008,
      "grad_norm": 0.1995217651128769,
      "learning_rate": 7.45640312286744e-08,
      "loss": 0.0471,
      "step": 59010
    },
    {
      "epoch": 6.904984788205008,
      "loss_main": 0.05723655968904495,
      "step": 59010
    },
    {
      "epoch": 6.906154926281301,
      "grad_norm": 0.2305024117231369,
      "learning_rate": 7.306716440927952e-08,
      "loss": 0.0481,
      "step": 59020
    },
    {
      "epoch": 6.906154926281301,
      "loss_main": 0.05149340629577637,
      "step": 59020
    },
    {
      "epoch": 6.907325064357594,
      "grad_norm": 0.16373401880264282,
      "learning_rate": 7.15854640738367e-08,
      "loss": 0.0455,
      "step": 59030
    },
    {
      "epoch": 6.907325064357594,
      "loss_main": 0.04871123656630516,
      "step": 59030
    },
    {
      "epoch": 6.908495202433887,
      "grad_norm": 0.2263702005147934,
      "learning_rate": 7.011893067244701e-08,
      "loss": 0.0462,
      "step": 59040
    },
    {
      "epoch": 6.908495202433887,
      "loss_main": 0.05074673146009445,
      "step": 59040
    },
    {
      "epoch": 6.90966534051018,
      "grad_norm": 0.23918813467025757,
      "learning_rate": 6.866756465060408e-08,
      "loss": 0.0417,
      "step": 59050
    },
    {
      "epoch": 6.90966534051018,
      "loss_main": 0.045946154743433,
      "step": 59050
    },
    {
      "epoch": 6.910835478586473,
      "grad_norm": 0.19257861375808716,
      "learning_rate": 6.723136644918859e-08,
      "loss": 0.0418,
      "step": 59060
    },
    {
      "epoch": 6.910835478586473,
      "loss_main": 0.022930482402443886,
      "step": 59060
    },
    {
      "epoch": 6.912005616662766,
      "grad_norm": 0.246445894241333,
      "learning_rate": 6.581033650449042e-08,
      "loss": 0.0408,
      "step": 59070
    },
    {
      "epoch": 6.912005616662766,
      "loss_main": 0.06005777791142464,
      "step": 59070
    },
    {
      "epoch": 6.9131757547390595,
      "grad_norm": 0.1591222584247589,
      "learning_rate": 6.440447524817539e-08,
      "loss": 0.0413,
      "step": 59080
    },
    {
      "epoch": 6.9131757547390595,
      "loss_main": 0.051075857132673264,
      "step": 59080
    },
    {
      "epoch": 6.914345892815352,
      "grad_norm": 0.18293295800685883,
      "learning_rate": 6.301378310730743e-08,
      "loss": 0.046,
      "step": 59090
    },
    {
      "epoch": 6.914345892815352,
      "loss_main": 0.06368109583854675,
      "step": 59090
    },
    {
      "epoch": 6.915516030891645,
      "grad_norm": 0.17004160583019257,
      "learning_rate": 6.163826050434307e-08,
      "loss": 0.039,
      "step": 59100
    },
    {
      "epoch": 6.915516030891645,
      "loss_main": 0.04450421407818794,
      "step": 59100
    },
    {
      "epoch": 6.916686168967939,
      "grad_norm": 0.20628759264945984,
      "learning_rate": 6.027790785713139e-08,
      "loss": 0.0528,
      "step": 59110
    },
    {
      "epoch": 6.916686168967939,
      "loss_main": 0.021814903244376183,
      "step": 59110
    },
    {
      "epoch": 6.917856307044231,
      "grad_norm": 0.29585298895835876,
      "learning_rate": 5.89327255789085e-08,
      "loss": 0.0453,
      "step": 59120
    },
    {
      "epoch": 6.917856307044231,
      "loss_main": 0.05444185063242912,
      "step": 59120
    },
    {
      "epoch": 6.919026445120524,
      "grad_norm": 0.17073015868663788,
      "learning_rate": 5.7602714078303085e-08,
      "loss": 0.0418,
      "step": 59130
    },
    {
      "epoch": 6.919026445120524,
      "loss_main": 0.03230910003185272,
      "step": 59130
    },
    {
      "epoch": 6.920196583196817,
      "grad_norm": 0.14053373038768768,
      "learning_rate": 5.628787375934197e-08,
      "loss": 0.0377,
      "step": 59140
    },
    {
      "epoch": 6.920196583196817,
      "loss_main": 0.036863021552562714,
      "step": 59140
    },
    {
      "epoch": 6.9213667212731105,
      "grad_norm": 0.2597382962703705,
      "learning_rate": 5.498820502143898e-08,
      "loss": 0.043,
      "step": 59150
    },
    {
      "epoch": 6.9213667212731105,
      "loss_main": 0.0439891554415226,
      "step": 59150
    },
    {
      "epoch": 6.922536859349403,
      "grad_norm": 0.2725977599620819,
      "learning_rate": 5.370370825939497e-08,
      "loss": 0.0448,
      "step": 59160
    },
    {
      "epoch": 6.922536859349403,
      "loss_main": 0.030533241108059883,
      "step": 59160
    },
    {
      "epoch": 6.923706997425696,
      "grad_norm": 0.15845714509487152,
      "learning_rate": 5.243438386340893e-08,
      "loss": 0.047,
      "step": 59170
    },
    {
      "epoch": 6.923706997425696,
      "loss_main": 0.050550736486911774,
      "step": 59170
    },
    {
      "epoch": 6.92487713550199,
      "grad_norm": 0.2492821216583252,
      "learning_rate": 5.118023221907242e-08,
      "loss": 0.0393,
      "step": 59180
    },
    {
      "epoch": 6.92487713550199,
      "loss_main": 0.062155067920684814,
      "step": 59180
    },
    {
      "epoch": 6.926047273578282,
      "grad_norm": 0.3210162818431854,
      "learning_rate": 4.994125370735292e-08,
      "loss": 0.05,
      "step": 59190
    },
    {
      "epoch": 6.926047273578282,
      "loss_main": 0.03338409587740898,
      "step": 59190
    },
    {
      "epoch": 6.927217411654575,
      "grad_norm": 0.1741465926170349,
      "learning_rate": 4.871744870462713e-08,
      "loss": 0.0441,
      "step": 59200
    },
    {
      "epoch": 6.927217411654575,
      "loss_main": 0.04767746850848198,
      "step": 59200
    },
    {
      "epoch": 6.928387549730868,
      "grad_norm": 0.16040389239788055,
      "learning_rate": 4.7508817582658794e-08,
      "loss": 0.0466,
      "step": 59210
    },
    {
      "epoch": 6.928387549730868,
      "loss_main": 0.04464300349354744,
      "step": 59210
    },
    {
      "epoch": 6.929557687807161,
      "grad_norm": 0.22963768243789673,
      "learning_rate": 4.631536070858755e-08,
      "loss": 0.0482,
      "step": 59220
    },
    {
      "epoch": 6.929557687807161,
      "loss_main": 0.028897006064653397,
      "step": 59220
    },
    {
      "epoch": 6.930727825883454,
      "grad_norm": 0.1965317577123642,
      "learning_rate": 4.513707844495674e-08,
      "loss": 0.0412,
      "step": 59230
    },
    {
      "epoch": 6.930727825883454,
      "loss_main": 0.04088568687438965,
      "step": 59230
    },
    {
      "epoch": 6.931897963959747,
      "grad_norm": 0.20933111011981964,
      "learning_rate": 4.3973971149702255e-08,
      "loss": 0.0394,
      "step": 59240
    },
    {
      "epoch": 6.931897963959747,
      "loss_main": 0.03887397050857544,
      "step": 59240
    },
    {
      "epoch": 6.9330681020360405,
      "grad_norm": 0.20766356587409973,
      "learning_rate": 4.2826039176147025e-08,
      "loss": 0.0398,
      "step": 59250
    },
    {
      "epoch": 6.9330681020360405,
      "loss_main": 0.05291416123509407,
      "step": 59250
    },
    {
      "epoch": 6.934238240112333,
      "grad_norm": 0.2551787197589874,
      "learning_rate": 4.169328287299545e-08,
      "loss": 0.0379,
      "step": 59260
    },
    {
      "epoch": 6.934238240112333,
      "loss_main": 0.033189915120601654,
      "step": 59260
    },
    {
      "epoch": 6.935408378188626,
      "grad_norm": 0.230485200881958,
      "learning_rate": 4.057570258435006e-08,
      "loss": 0.0416,
      "step": 59270
    },
    {
      "epoch": 6.935408378188626,
      "loss_main": 0.03549652174115181,
      "step": 59270
    },
    {
      "epoch": 6.93657851626492,
      "grad_norm": 0.24057896435260773,
      "learning_rate": 3.947329864970595e-08,
      "loss": 0.0418,
      "step": 59280
    },
    {
      "epoch": 6.93657851626492,
      "loss_main": 0.040745802223682404,
      "step": 59280
    },
    {
      "epoch": 6.937748654341212,
      "grad_norm": 0.24170301854610443,
      "learning_rate": 3.8386071403939686e-08,
      "loss": 0.0466,
      "step": 59290
    },
    {
      "epoch": 6.937748654341212,
      "loss_main": 0.08430586755275726,
      "step": 59290
    },
    {
      "epoch": 6.938918792417505,
      "grad_norm": 0.2817220389842987,
      "learning_rate": 3.731402117733152e-08,
      "loss": 0.0449,
      "step": 59300
    },
    {
      "epoch": 6.938918792417505,
      "loss_main": 0.049424510449171066,
      "step": 59300
    },
    {
      "epoch": 6.940088930493799,
      "grad_norm": 0.2623097002506256,
      "learning_rate": 3.625714829552651e-08,
      "loss": 0.0509,
      "step": 59310
    },
    {
      "epoch": 6.940088930493799,
      "loss_main": 0.03127920627593994,
      "step": 59310
    },
    {
      "epoch": 6.9412590685700915,
      "grad_norm": 0.20596720278263092,
      "learning_rate": 3.5215453079590065e-08,
      "loss": 0.0377,
      "step": 59320
    },
    {
      "epoch": 6.9412590685700915,
      "loss_main": 0.06884534657001495,
      "step": 59320
    },
    {
      "epoch": 6.942429206646384,
      "grad_norm": 0.22788475453853607,
      "learning_rate": 3.4188935845952396e-08,
      "loss": 0.0451,
      "step": 59330
    },
    {
      "epoch": 6.942429206646384,
      "loss_main": 0.035173624753952026,
      "step": 59330
    },
    {
      "epoch": 6.943599344722677,
      "grad_norm": 0.3256121277809143,
      "learning_rate": 3.3177596906447396e-08,
      "loss": 0.0437,
      "step": 59340
    },
    {
      "epoch": 6.943599344722677,
      "loss_main": 0.02860618568956852,
      "step": 59340
    },
    {
      "epoch": 6.9447694827989706,
      "grad_norm": 0.2742643356323242,
      "learning_rate": 3.218143656829043e-08,
      "loss": 0.0429,
      "step": 59350
    },
    {
      "epoch": 6.9447694827989706,
      "loss_main": 0.07583004981279373,
      "step": 59350
    },
    {
      "epoch": 6.945939620875263,
      "grad_norm": 0.15452416241168976,
      "learning_rate": 3.120045513408387e-08,
      "loss": 0.044,
      "step": 59360
    },
    {
      "epoch": 6.945939620875263,
      "loss_main": 0.034803345799446106,
      "step": 59360
    },
    {
      "epoch": 6.947109758951556,
      "grad_norm": 0.3065756857395172,
      "learning_rate": 3.02346529018338e-08,
      "loss": 0.0423,
      "step": 59370
    },
    {
      "epoch": 6.947109758951556,
      "loss_main": 0.0429927296936512,
      "step": 59370
    },
    {
      "epoch": 6.94827989702785,
      "grad_norm": 0.2046983391046524,
      "learning_rate": 2.9284030164922204e-08,
      "loss": 0.0448,
      "step": 59380
    },
    {
      "epoch": 6.94827989702785,
      "loss_main": 0.027423640713095665,
      "step": 59380
    },
    {
      "epoch": 6.949450035104142,
      "grad_norm": 0.2134322226047516,
      "learning_rate": 2.8348587212123634e-08,
      "loss": 0.0386,
      "step": 59390
    },
    {
      "epoch": 6.949450035104142,
      "loss_main": 0.03598484769463539,
      "step": 59390
    },
    {
      "epoch": 6.950620173180435,
      "grad_norm": 0.19798484444618225,
      "learning_rate": 2.7428324327594125e-08,
      "loss": 0.0379,
      "step": 59400
    },
    {
      "epoch": 6.950620173180435,
      "loss_main": 0.04768460988998413,
      "step": 59400
    },
    {
      "epoch": 6.951790311256728,
      "grad_norm": 0.228398397564888,
      "learning_rate": 2.6523241790893383e-08,
      "loss": 0.0449,
      "step": 59410
    },
    {
      "epoch": 6.951790311256728,
      "loss_main": 0.05361989513039589,
      "step": 59410
    },
    {
      "epoch": 6.9529604493330215,
      "grad_norm": 0.2763090133666992,
      "learning_rate": 2.563333987695704e-08,
      "loss": 0.0423,
      "step": 59420
    },
    {
      "epoch": 6.9529604493330215,
      "loss_main": 0.04096513241529465,
      "step": 59420
    },
    {
      "epoch": 6.954130587409314,
      "grad_norm": 0.16400651633739471,
      "learning_rate": 2.4758618856118852e-08,
      "loss": 0.041,
      "step": 59430
    },
    {
      "epoch": 6.954130587409314,
      "loss_main": 0.023035969585180283,
      "step": 59430
    },
    {
      "epoch": 6.955300725485607,
      "grad_norm": 0.28552356362342834,
      "learning_rate": 2.3899078994088497e-08,
      "loss": 0.0422,
      "step": 59440
    },
    {
      "epoch": 6.955300725485607,
      "loss_main": 0.06255987286567688,
      "step": 59440
    },
    {
      "epoch": 6.956470863561901,
      "grad_norm": 0.3662072420120239,
      "learning_rate": 2.3054720551973775e-08,
      "loss": 0.0437,
      "step": 59450
    },
    {
      "epoch": 6.956470863561901,
      "loss_main": 0.04613446816802025,
      "step": 59450
    },
    {
      "epoch": 6.957641001638193,
      "grad_norm": 0.2468271702528,
      "learning_rate": 2.222554378627506e-08,
      "loss": 0.0487,
      "step": 59460
    },
    {
      "epoch": 6.957641001638193,
      "loss_main": 0.041655804961919785,
      "step": 59460
    },
    {
      "epoch": 6.958811139714486,
      "grad_norm": 0.165910005569458,
      "learning_rate": 2.1411548948868653e-08,
      "loss": 0.0425,
      "step": 59470
    },
    {
      "epoch": 6.958811139714486,
      "loss_main": 0.059430453926324844,
      "step": 59470
    },
    {
      "epoch": 6.95998127779078,
      "grad_norm": 0.18215642869472504,
      "learning_rate": 2.0612736287023426e-08,
      "loss": 0.0496,
      "step": 59480
    },
    {
      "epoch": 6.95998127779078,
      "loss_main": 0.018320079892873764,
      "step": 59480
    },
    {
      "epoch": 6.961151415867072,
      "grad_norm": 0.23590131103992462,
      "learning_rate": 1.9829106043400826e-08,
      "loss": 0.0473,
      "step": 59490
    },
    {
      "epoch": 6.961151415867072,
      "loss_main": 0.04368990287184715,
      "step": 59490
    },
    {
      "epoch": 6.962321553943365,
      "grad_norm": 0.2276766151189804,
      "learning_rate": 1.9060658456043768e-08,
      "loss": 0.043,
      "step": 59500
    },
    {
      "epoch": 6.962321553943365,
      "loss_main": 0.05055830255150795,
      "step": 59500
    },
    {
      "epoch": 6.963491692019659,
      "grad_norm": 0.19825828075408936,
      "learning_rate": 1.830739375838775e-08,
      "loss": 0.0539,
      "step": 59510
    },
    {
      "epoch": 6.963491692019659,
      "loss_main": 0.05111071094870567,
      "step": 59510
    },
    {
      "epoch": 6.9646618300959515,
      "grad_norm": 0.20019593834877014,
      "learning_rate": 1.7569312179260832e-08,
      "loss": 0.0439,
      "step": 59520
    },
    {
      "epoch": 6.9646618300959515,
      "loss_main": 0.04729023203253746,
      "step": 59520
    },
    {
      "epoch": 6.965831968172244,
      "grad_norm": 0.19554530084133148,
      "learning_rate": 1.684641394286146e-08,
      "loss": 0.0412,
      "step": 59530
    },
    {
      "epoch": 6.965831968172244,
      "loss_main": 0.01907784678041935,
      "step": 59530
    },
    {
      "epoch": 6.967002106248537,
      "grad_norm": 0.27025309205055237,
      "learning_rate": 1.6138699268797296e-08,
      "loss": 0.042,
      "step": 59540
    },
    {
      "epoch": 6.967002106248537,
      "loss_main": 0.0315663181245327,
      "step": 59540
    },
    {
      "epoch": 6.968172244324831,
      "grad_norm": 0.2388489991426468,
      "learning_rate": 1.5446168372046376e-08,
      "loss": 0.0371,
      "step": 59550
    },
    {
      "epoch": 6.968172244324831,
      "loss_main": 0.0254262275993824,
      "step": 59550
    },
    {
      "epoch": 6.969342382401123,
      "grad_norm": 0.2589423656463623,
      "learning_rate": 1.4768821462984861e-08,
      "loss": 0.0482,
      "step": 59560
    },
    {
      "epoch": 6.969342382401123,
      "loss_main": 0.05051100254058838,
      "step": 59560
    },
    {
      "epoch": 6.970512520477416,
      "grad_norm": 0.30156904458999634,
      "learning_rate": 1.4106658747370383e-08,
      "loss": 0.0459,
      "step": 59570
    },
    {
      "epoch": 6.970512520477416,
      "loss_main": 0.037370603531599045,
      "step": 59570
    },
    {
      "epoch": 6.97168265855371,
      "grad_norm": 0.26887983083724976,
      "learning_rate": 1.3459680426353149e-08,
      "loss": 0.042,
      "step": 59580
    },
    {
      "epoch": 6.97168265855371,
      "loss_main": 0.03333086892962456,
      "step": 59580
    },
    {
      "epoch": 6.9728527966300025,
      "grad_norm": 0.14693638682365417,
      "learning_rate": 1.2827886696464841e-08,
      "loss": 0.0468,
      "step": 59590
    },
    {
      "epoch": 6.9728527966300025,
      "loss_main": 0.04833444580435753,
      "step": 59590
    },
    {
      "epoch": 6.974022934706295,
      "grad_norm": 0.253632515668869,
      "learning_rate": 1.2211277749635264e-08,
      "loss": 0.0462,
      "step": 59600
    },
    {
      "epoch": 6.974022934706295,
      "loss_main": 0.04159136861562729,
      "step": 59600
    },
    {
      "epoch": 6.975193072782588,
      "grad_norm": 0.21520042419433594,
      "learning_rate": 1.1609853773164592e-08,
      "loss": 0.0488,
      "step": 59610
    },
    {
      "epoch": 6.975193072782588,
      "loss_main": 0.03875075653195381,
      "step": 59610
    },
    {
      "epoch": 6.976363210858882,
      "grad_norm": 0.1901392936706543,
      "learning_rate": 1.1023614949751127e-08,
      "loss": 0.048,
      "step": 59620
    },
    {
      "epoch": 6.976363210858882,
      "loss_main": 0.060574404895305634,
      "step": 59620
    },
    {
      "epoch": 6.977533348935174,
      "grad_norm": 0.2016211748123169,
      "learning_rate": 1.0452561457485744e-08,
      "loss": 0.0454,
      "step": 59630
    },
    {
      "epoch": 6.977533348935174,
      "loss_main": 0.046936120837926865,
      "step": 59630
    },
    {
      "epoch": 6.978703487011467,
      "grad_norm": 0.2078363001346588,
      "learning_rate": 9.896693469829688e-09,
      "loss": 0.0432,
      "step": 59640
    },
    {
      "epoch": 6.978703487011467,
      "loss_main": 0.025368178263306618,
      "step": 59640
    },
    {
      "epoch": 6.979873625087761,
      "grad_norm": 0.20558284223079681,
      "learning_rate": 9.35601115564788e-09,
      "loss": 0.0388,
      "step": 59650
    },
    {
      "epoch": 6.979873625087761,
      "loss_main": 0.02865814045071602,
      "step": 59650
    },
    {
      "epoch": 6.981043763164053,
      "grad_norm": 0.2469612956047058,
      "learning_rate": 8.830514679186719e-09,
      "loss": 0.0386,
      "step": 59660
    },
    {
      "epoch": 6.981043763164053,
      "loss_main": 0.0399671271443367,
      "step": 59660
    },
    {
      "epoch": 6.982213901240346,
      "grad_norm": 0.22954770922660828,
      "learning_rate": 8.320204200074066e-09,
      "loss": 0.0406,
      "step": 59670
    },
    {
      "epoch": 6.982213901240346,
      "loss_main": 0.03237313777208328,
      "step": 59670
    },
    {
      "epoch": 6.98338403931664,
      "grad_norm": 0.24955056607723236,
      "learning_rate": 7.825079873324814e-09,
      "loss": 0.0443,
      "step": 59680
    },
    {
      "epoch": 6.98338403931664,
      "loss_main": 0.036235760897397995,
      "step": 59680
    },
    {
      "epoch": 6.9845541773929325,
      "grad_norm": 0.23681363463401794,
      "learning_rate": 7.345141849351977e-09,
      "loss": 0.0447,
      "step": 59690
    },
    {
      "epoch": 6.9845541773929325,
      "loss_main": 0.032045189291238785,
      "step": 59690
    },
    {
      "epoch": 6.985724315469225,
      "grad_norm": 0.25979122519493103,
      "learning_rate": 6.880390273944493e-09,
      "loss": 0.0471,
      "step": 59700
    },
    {
      "epoch": 6.985724315469225,
      "loss_main": 0.03349417820572853,
      "step": 59700
    },
    {
      "epoch": 6.986894453545519,
      "grad_norm": 0.21992523968219757,
      "learning_rate": 6.4308252882838704e-09,
      "loss": 0.0444,
      "step": 59710
    },
    {
      "epoch": 6.986894453545519,
      "loss_main": 0.03847375512123108,
      "step": 59710
    },
    {
      "epoch": 6.988064591621812,
      "grad_norm": 0.22831811010837555,
      "learning_rate": 5.9964470289386455e-09,
      "loss": 0.0532,
      "step": 59720
    },
    {
      "epoch": 6.988064591621812,
      "loss_main": 0.03066684864461422,
      "step": 59720
    },
    {
      "epoch": 6.989234729698104,
      "grad_norm": 0.28571009635925293,
      "learning_rate": 5.577255627853273e-09,
      "loss": 0.0562,
      "step": 59730
    },
    {
      "epoch": 6.989234729698104,
      "loss_main": 0.038282666355371475,
      "step": 59730
    },
    {
      "epoch": 6.990404867774397,
      "grad_norm": 0.14999879896640778,
      "learning_rate": 5.173251212370334e-09,
      "loss": 0.0477,
      "step": 59740
    },
    {
      "epoch": 6.990404867774397,
      "loss_main": 0.06453870236873627,
      "step": 59740
    },
    {
      "epoch": 6.991575005850691,
      "grad_norm": 0.161162331700325,
      "learning_rate": 4.784433905219432e-09,
      "loss": 0.0456,
      "step": 59750
    },
    {
      "epoch": 6.991575005850691,
      "loss_main": 0.04206085950136185,
      "step": 59750
    },
    {
      "epoch": 6.992745143926983,
      "grad_norm": 0.20665185153484344,
      "learning_rate": 4.4108038245060934e-09,
      "loss": 0.0415,
      "step": 59760
    },
    {
      "epoch": 6.992745143926983,
      "loss_main": 0.02929316647350788,
      "step": 59760
    },
    {
      "epoch": 6.993915282003276,
      "grad_norm": 0.17105358839035034,
      "learning_rate": 4.05236108373952e-09,
      "loss": 0.0359,
      "step": 59770
    },
    {
      "epoch": 6.993915282003276,
      "loss_main": 0.03836856037378311,
      "step": 59770
    },
    {
      "epoch": 6.995085420079569,
      "grad_norm": 0.1361159384250641,
      "learning_rate": 3.7091057917937324e-09,
      "loss": 0.0442,
      "step": 59780
    },
    {
      "epoch": 6.995085420079569,
      "loss_main": 0.036247167736291885,
      "step": 59780
    },
    {
      "epoch": 6.9962555581558625,
      "grad_norm": 0.29042139649391174,
      "learning_rate": 3.381038052946428e-09,
      "loss": 0.0413,
      "step": 59790
    },
    {
      "epoch": 6.9962555581558625,
      "loss_main": 0.03654700145125389,
      "step": 59790
    },
    {
      "epoch": 6.997425696232155,
      "grad_norm": 0.21030192077159882,
      "learning_rate": 3.0681579668623283e-09,
      "loss": 0.046,
      "step": 59800
    },
    {
      "epoch": 6.997425696232155,
      "loss_main": 0.053257960826158524,
      "step": 59800
    },
    {
      "epoch": 6.998595834308448,
      "grad_norm": 0.23277926445007324,
      "learning_rate": 2.7704656285709727e-09,
      "loss": 0.0378,
      "step": 59810
    },
    {
      "epoch": 6.998595834308448,
      "loss_main": 0.04619484394788742,
      "step": 59810
    },
    {
      "epoch": 6.999765972384742,
      "grad_norm": 0.17191848158836365,
      "learning_rate": 2.4879611285166803e-09,
      "loss": 0.0488,
      "step": 59820
    },
    {
      "epoch": 6.999765972384742,
      "loss_main": 0.027731215581297874,
      "step": 59820
    },
    {
      "epoch": 7.000936110461034,
      "grad_norm": 0.1748702973127365,
      "learning_rate": 2.2206445525085883e-09,
      "loss": 0.0367,
      "step": 59830
    },
    {
      "epoch": 7.000936110461034,
      "loss_main": 0.02425217069685459,
      "step": 59830
    },
    {
      "epoch": 7.002106248537327,
      "grad_norm": 0.19163377583026886,
      "learning_rate": 1.9685159817595112e-09,
      "loss": 0.0406,
      "step": 59840
    },
    {
      "epoch": 7.002106248537327,
      "loss_main": 0.024608194828033447,
      "step": 59840
    },
    {
      "epoch": 7.003276386613621,
      "grad_norm": 0.22812519967556,
      "learning_rate": 1.7315754928470817e-09,
      "loss": 0.042,
      "step": 59850
    },
    {
      "epoch": 7.003276386613621,
      "loss_main": 0.045473337173461914,
      "step": 59850
    },
    {
      "epoch": 7.0044465246899135,
      "grad_norm": 0.21162846684455872,
      "learning_rate": 1.5098231577581611e-09,
      "loss": 0.0424,
      "step": 59860
    },
    {
      "epoch": 7.0044465246899135,
      "loss_main": 0.03566332906484604,
      "step": 59860
    },
    {
      "epoch": 7.005616662766206,
      "grad_norm": 0.17175675928592682,
      "learning_rate": 1.3032590438499804e-09,
      "loss": 0.0465,
      "step": 59870
    },
    {
      "epoch": 7.005616662766206,
      "loss_main": 0.06599870324134827,
      "step": 59870
    },
    {
      "epoch": 7.0067868008425,
      "grad_norm": 0.2296479046344757,
      "learning_rate": 1.1118832138723444e-09,
      "loss": 0.0447,
      "step": 59880
    },
    {
      "epoch": 7.0067868008425,
      "loss_main": 0.040230292826890945,
      "step": 59880
    },
    {
      "epoch": 7.007956938918793,
      "grad_norm": 0.19158104062080383,
      "learning_rate": 9.356957259620825e-10,
      "loss": 0.047,
      "step": 59890
    },
    {
      "epoch": 7.007956938918793,
      "loss_main": 0.026889124885201454,
      "step": 59890
    },
    {
      "epoch": 7.009127076995085,
      "grad_norm": 0.21359451115131378,
      "learning_rate": 7.74696633637495e-10,
      "loss": 0.0445,
      "step": 59900
    },
    {
      "epoch": 7.009127076995085,
      "loss_main": 0.04644623026251793,
      "step": 59900
    },
    {
      "epoch": 7.010297215071378,
      "grad_norm": 0.22595590353012085,
      "learning_rate": 6.288859858039064e-10,
      "loss": 0.055,
      "step": 59910
    },
    {
      "epoch": 7.010297215071378,
      "loss_main": 0.04261469095945358,
      "step": 59910
    },
    {
      "epoch": 7.011467353147672,
      "grad_norm": 0.30006295442581177,
      "learning_rate": 4.982638267647665e-10,
      "loss": 0.0391,
      "step": 59920
    },
    {
      "epoch": 7.011467353147672,
      "loss_main": 0.034127507358789444,
      "step": 59920
    },
    {
      "epoch": 7.012637491223964,
      "grad_norm": 0.21636739373207092,
      "learning_rate": 3.8283019618834406e-10,
      "loss": 0.0452,
      "step": 59930
    },
    {
      "epoch": 7.012637491223964,
      "loss_main": 0.031238645315170288,
      "step": 59930
    },
    {
      "epoch": 7.013807629300257,
      "grad_norm": 0.26904597878456116,
      "learning_rate": 2.825851291410331e-10,
      "loss": 0.0451,
      "step": 59940
    },
    {
      "epoch": 7.013807629300257,
      "loss_main": 0.05585678294301033,
      "step": 59940
    },
    {
      "epoch": 7.014977767376551,
      "grad_norm": 0.23035645484924316,
      "learning_rate": 1.975286560873535e-10,
      "loss": 0.0431,
      "step": 59950
    },
    {
      "epoch": 7.014977767376551,
      "loss_main": 0.05442739278078079,
      "step": 59950
    },
    {
      "epoch": 7.0161479054528435,
      "grad_norm": 0.25404131412506104,
      "learning_rate": 1.2766080285109284e-10,
      "loss": 0.0428,
      "step": 59960
    },
    {
      "epoch": 7.0161479054528435,
      "loss_main": 0.061520177870988846,
      "step": 59960
    },
    {
      "epoch": 7.017318043529136,
      "grad_norm": 0.22515778243541718,
      "learning_rate": 7.298159065971533e-11,
      "loss": 0.0393,
      "step": 59970
    },
    {
      "epoch": 7.017318043529136,
      "loss_main": 0.05911855772137642,
      "step": 59970
    },
    {
      "epoch": 7.01848818160543,
      "grad_norm": 0.2407500147819519,
      "learning_rate": 3.349103612770854e-11,
      "loss": 0.0479,
      "step": 59980
    },
    {
      "epoch": 7.01848818160543,
      "loss_main": 0.03967353329062462,
      "step": 59980
    },
    {
      "epoch": 7.019658319681723,
      "grad_norm": 0.18318498134613037,
      "learning_rate": 9.189151245481142e-12,
      "loss": 0.042,
      "step": 59990
    },
    {
      "epoch": 7.019658319681723,
      "loss_main": 0.05218319222331047,
      "step": 59990
    },
    {
      "epoch": 7.020828457758015,
      "grad_norm": 0.21422041952610016,
      "learning_rate": 7.594340156735769e-14,
      "loss": 0.0408,
      "step": 60000
    }
  ],
  "logging_steps": 10,
  "max_steps": 60000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 8,
  "save_steps": 10000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 0.0,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}