{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.2,
  "eval_steps": 500,
  "global_step": 1000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.002,
      "grad_norm": 2.2951126098632812,
      "learning_rate": 5.399999999999999e-06,
      "loss": 12.198476409912109,
      "step": 10
    },
    {
      "epoch": 0.004,
      "grad_norm": 2.6434836387634277,
      "learning_rate": 1.14e-05,
      "loss": 11.658338928222657,
      "step": 20
    },
    {
      "epoch": 0.006,
      "grad_norm": 2.0430407524108887,
      "learning_rate": 1.74e-05,
      "loss": 10.702792358398437,
      "step": 30
    },
    {
      "epoch": 0.008,
      "grad_norm": 1.938822627067566,
      "learning_rate": 2.34e-05,
      "loss": 10.229283142089844,
      "step": 40
    },
    {
      "epoch": 0.01,
      "grad_norm": 1.3679683208465576,
      "learning_rate": 2.94e-05,
      "loss": 9.728135681152343,
      "step": 50
    },
    {
      "epoch": 0.012,
      "grad_norm": 2.546030044555664,
      "learning_rate": 3.539999999999999e-05,
      "loss": 9.183514404296876,
      "step": 60
    },
    {
      "epoch": 0.014,
      "grad_norm": 1.7231528759002686,
      "learning_rate": 4.14e-05,
      "loss": 8.578328704833984,
      "step": 70
    },
    {
      "epoch": 0.016,
      "grad_norm": 1.4366660118103027,
      "learning_rate": 4.7399999999999993e-05,
      "loss": 8.042523956298828,
      "step": 80
    },
    {
      "epoch": 0.018,
      "grad_norm": 1.4448844194412231,
      "learning_rate": 5.339999999999999e-05,
      "loss": 7.64561767578125,
      "step": 90
    },
    {
      "epoch": 0.02,
      "grad_norm": 1.3400251865386963,
      "learning_rate": 5.94e-05,
      "loss": 7.421743774414063,
      "step": 100
    },
    {
      "epoch": 0.022,
      "grad_norm": 0.8998181223869324,
      "learning_rate": 6.539999999999999e-05,
      "loss": 7.200779724121094,
      "step": 110
    },
    {
      "epoch": 0.024,
      "grad_norm": 0.8080795407295227,
      "learning_rate": 7.139999999999999e-05,
      "loss": 7.073184204101563,
      "step": 120
    },
    {
      "epoch": 0.026,
      "grad_norm": 0.970500648021698,
      "learning_rate": 7.74e-05,
      "loss": 6.967037963867187,
      "step": 130
    },
    {
      "epoch": 0.028,
      "grad_norm": 1.2163574695587158,
      "learning_rate": 8.34e-05,
      "loss": 6.91470947265625,
      "step": 140
    },
    {
      "epoch": 0.03,
      "grad_norm": 1.1663590669631958,
      "learning_rate": 8.939999999999999e-05,
      "loss": 6.799385070800781,
      "step": 150
    },
    {
      "epoch": 0.032,
      "grad_norm": 1.1311771869659424,
      "learning_rate": 9.539999999999999e-05,
      "loss": 6.620470428466797,
      "step": 160
    },
    {
      "epoch": 0.034,
      "grad_norm": 1.1079429388046265,
      "learning_rate": 0.0001014,
      "loss": 6.601376342773437,
      "step": 170
    },
    {
      "epoch": 0.036,
      "grad_norm": 1.3198282718658447,
      "learning_rate": 0.00010739999999999998,
      "loss": 6.559695434570313,
      "step": 180
    },
    {
      "epoch": 0.038,
      "grad_norm": 1.0375511646270752,
      "learning_rate": 0.00011339999999999999,
      "loss": 6.444459533691406,
      "step": 190
    },
    {
      "epoch": 0.04,
      "grad_norm": 1.1254252195358276,
      "learning_rate": 0.0001194,
      "loss": 6.329716491699219,
      "step": 200
    },
    {
      "epoch": 0.042,
      "grad_norm": 1.0709344148635864,
      "learning_rate": 0.00012539999999999999,
      "loss": 6.242716979980469,
      "step": 210
    },
    {
      "epoch": 0.044,
      "grad_norm": 1.0490570068359375,
      "learning_rate": 0.0001314,
      "loss": 6.146401214599609,
      "step": 220
    },
    {
      "epoch": 0.046,
      "grad_norm": 0.9572665095329285,
      "learning_rate": 0.0001374,
      "loss": 6.086616134643554,
      "step": 230
    },
    {
      "epoch": 0.048,
      "grad_norm": 1.1944881677627563,
      "learning_rate": 0.0001434,
      "loss": 6.035008239746094,
      "step": 240
    },
    {
      "epoch": 0.05,
      "grad_norm": 1.139950156211853,
      "learning_rate": 0.0001494,
      "loss": 5.863430786132812,
      "step": 250
    },
    {
      "epoch": 0.052,
      "grad_norm": 1.0705268383026123,
      "learning_rate": 0.00015539999999999998,
      "loss": 5.8445487976074215,
      "step": 260
    },
    {
      "epoch": 0.054,
      "grad_norm": 1.228391170501709,
      "learning_rate": 0.0001614,
      "loss": 5.702362442016602,
      "step": 270
    },
    {
      "epoch": 0.056,
      "grad_norm": 1.2548208236694336,
      "learning_rate": 0.0001674,
      "loss": 5.753414154052734,
      "step": 280
    },
    {
      "epoch": 0.058,
      "grad_norm": 1.0533807277679443,
      "learning_rate": 0.00017339999999999996,
      "loss": 5.655702209472656,
      "step": 290
    },
    {
      "epoch": 0.06,
      "grad_norm": 0.9166179895401001,
      "learning_rate": 0.00017939999999999997,
      "loss": 5.616793060302735,
      "step": 300
    },
    {
      "epoch": 0.062,
      "grad_norm": 0.8547247648239136,
      "learning_rate": 0.00018539999999999998,
      "loss": 5.597354888916016,
      "step": 310
    },
    {
      "epoch": 0.064,
      "grad_norm": 0.9937305450439453,
      "learning_rate": 0.0001914,
      "loss": 5.572035598754883,
      "step": 320
    },
    {
      "epoch": 0.066,
      "grad_norm": 0.9027020335197449,
      "learning_rate": 0.0001974,
      "loss": 5.49364013671875,
      "step": 330
    },
    {
      "epoch": 0.068,
      "grad_norm": 0.976992666721344,
      "learning_rate": 0.00020339999999999998,
      "loss": 5.4948478698730465,
      "step": 340
    },
    {
      "epoch": 0.07,
      "grad_norm": 0.8595808148384094,
      "learning_rate": 0.00020939999999999997,
      "loss": 5.504860687255859,
      "step": 350
    },
    {
      "epoch": 0.072,
      "grad_norm": 0.8206803798675537,
      "learning_rate": 0.00021539999999999998,
      "loss": 5.442459487915039,
      "step": 360
    },
    {
      "epoch": 0.074,
      "grad_norm": 0.8065920472145081,
      "learning_rate": 0.0002214,
      "loss": 5.3627159118652346,
      "step": 370
    },
    {
      "epoch": 0.076,
      "grad_norm": 1.3093833923339844,
      "learning_rate": 0.00022739999999999997,
      "loss": 5.357783508300781,
      "step": 380
    },
    {
      "epoch": 0.078,
      "grad_norm": 1.0391067266464233,
      "learning_rate": 0.00023339999999999998,
      "loss": 5.305191421508789,
      "step": 390
    },
    {
      "epoch": 0.08,
      "grad_norm": 1.1108404397964478,
      "learning_rate": 0.0002394,
      "loss": 5.3095954895019535,
      "step": 400
    },
    {
      "epoch": 0.082,
      "grad_norm": 0.7273285388946533,
      "learning_rate": 0.00024539999999999995,
      "loss": 5.245541381835937,
      "step": 410
    },
    {
      "epoch": 0.084,
      "grad_norm": 0.8222587704658508,
      "learning_rate": 0.0002514,
      "loss": 5.180028915405273,
      "step": 420
    },
    {
      "epoch": 0.086,
      "grad_norm": 0.9594866633415222,
      "learning_rate": 0.00025739999999999997,
      "loss": 5.281720733642578,
      "step": 430
    },
    {
      "epoch": 0.088,
      "grad_norm": 0.6788018345832825,
      "learning_rate": 0.00026339999999999995,
      "loss": 5.117179870605469,
      "step": 440
    },
    {
      "epoch": 0.09,
      "grad_norm": 0.9051905870437622,
      "learning_rate": 0.0002694,
      "loss": 5.1706890106201175,
      "step": 450
    },
    {
      "epoch": 0.092,
      "grad_norm": 0.7013774514198303,
      "learning_rate": 0.00027539999999999997,
      "loss": 5.212803649902344,
      "step": 460
    },
    {
      "epoch": 0.094,
      "grad_norm": 0.8330399990081787,
      "learning_rate": 0.00028139999999999996,
      "loss": 5.102211761474609,
      "step": 470
    },
    {
      "epoch": 0.096,
      "grad_norm": 0.7064741253852844,
      "learning_rate": 0.00028739999999999994,
      "loss": 5.0642646789550785,
      "step": 480
    },
    {
      "epoch": 0.098,
      "grad_norm": 0.7228646278381348,
      "learning_rate": 0.0002934,
      "loss": 5.086193084716797,
      "step": 490
    },
    {
      "epoch": 0.1,
      "grad_norm": 0.6500748991966248,
      "learning_rate": 0.00029939999999999996,
      "loss": 4.952143859863281,
      "step": 500
    },
    {
      "epoch": 0.102,
      "grad_norm": 0.6131930947303772,
      "learning_rate": 0.00029999703912842054,
      "loss": 5.013114547729492,
      "step": 510
    },
    {
      "epoch": 0.104,
      "grad_norm": 0.6693644523620605,
      "learning_rate": 0.0002999868041668571,
      "loss": 5.064251327514649,
      "step": 520
    },
    {
      "epoch": 0.106,
      "grad_norm": 0.8424168825149536,
      "learning_rate": 0.0002999692590600585,
      "loss": 4.953443145751953,
      "step": 530
    },
    {
      "epoch": 0.108,
      "grad_norm": 0.8079104423522949,
      "learning_rate": 0.00029994440466314827,
      "loss": 5.028744888305664,
      "step": 540
    },
    {
      "epoch": 0.11,
      "grad_norm": 0.6316964030265808,
      "learning_rate": 0.00029991224218749483,
      "loss": 4.919719696044922,
      "step": 550
    },
    {
      "epoch": 0.112,
      "grad_norm": 0.5195901393890381,
      "learning_rate": 0.0002998727732006518,
      "loss": 4.8272350311279295,
      "step": 560
    },
    {
      "epoch": 0.114,
      "grad_norm": 0.8154641389846802,
      "learning_rate": 0.00029982599962628185,
      "loss": 4.901306533813477,
      "step": 570
    },
    {
      "epoch": 0.116,
      "grad_norm": 0.6902443170547485,
      "learning_rate": 0.0002997719237440631,
      "loss": 4.899118041992187,
      "step": 580
    },
    {
      "epoch": 0.118,
      "grad_norm": 0.7940948009490967,
      "learning_rate": 0.0002997105481895778,
      "loss": 4.76171989440918,
      "step": 590
    },
    {
      "epoch": 0.12,
      "grad_norm": 0.6447590589523315,
      "learning_rate": 0.00029964187595418377,
      "loss": 4.846940231323242,
      "step": 600
    },
    {
      "epoch": 0.122,
      "grad_norm": 0.5544273853302002,
      "learning_rate": 0.00029956591038486895,
      "loss": 4.747989654541016,
      "step": 610
    },
    {
      "epoch": 0.124,
      "grad_norm": 0.5011790990829468,
      "learning_rate": 0.000299482655184088,
      "loss": 4.800208663940429,
      "step": 620
    },
    {
      "epoch": 0.126,
      "grad_norm": 0.5372884273529053,
      "learning_rate": 0.0002993921144095819,
      "loss": 4.803700256347656,
      "step": 630
    },
    {
      "epoch": 0.128,
      "grad_norm": 0.44695383310317993,
      "learning_rate": 0.00029929429247418037,
      "loss": 4.715145111083984,
      "step": 640
    },
    {
      "epoch": 0.13,
      "grad_norm": 0.5330514907836914,
      "learning_rate": 0.0002991891941455865,
      "loss": 4.6798866271972654,
      "step": 650
    },
    {
      "epoch": 0.132,
      "grad_norm": 0.7654022574424744,
      "learning_rate": 0.0002990768245461445,
      "loss": 4.705394744873047,
      "step": 660
    },
    {
      "epoch": 0.134,
      "grad_norm": 0.6759101748466492,
      "learning_rate": 0.0002989571891525901,
      "loss": 4.711079406738281,
      "step": 670
    },
    {
      "epoch": 0.136,
      "grad_norm": 0.5494880080223083,
      "learning_rate": 0.00029883029379578375,
      "loss": 4.6135414123535154,
      "step": 680
    },
    {
      "epoch": 0.138,
      "grad_norm": 0.7019897103309631,
      "learning_rate": 0.000298696144660426,
      "loss": 4.6716865539550785,
      "step": 690
    },
    {
      "epoch": 0.14,
      "grad_norm": 0.6551356911659241,
      "learning_rate": 0.0002985547482847564,
      "loss": 4.780976486206055,
      "step": 700
    },
    {
      "epoch": 0.142,
      "grad_norm": 0.505607545375824,
      "learning_rate": 0.000298406111560235,
      "loss": 4.607578277587891,
      "step": 710
    },
    {
      "epoch": 0.144,
      "grad_norm": 0.43605464696884155,
      "learning_rate": 0.00029825024173120587,
      "loss": 4.561976623535156,
      "step": 720
    },
    {
      "epoch": 0.146,
      "grad_norm": 0.48921728134155273,
      "learning_rate": 0.0002980871463945447,
      "loss": 4.422502899169922,
      "step": 730
    },
    {
      "epoch": 0.148,
      "grad_norm": 0.7877730131149292,
      "learning_rate": 0.00029791683349928805,
      "loss": 4.556784439086914,
      "step": 740
    },
    {
      "epoch": 0.15,
      "grad_norm": 0.5935314893722534,
      "learning_rate": 0.00029773931134624614,
      "loss": 4.396469497680664,
      "step": 750
    },
    {
      "epoch": 0.152,
      "grad_norm": 0.6277227401733398,
      "learning_rate": 0.0002975545885875983,
      "loss": 4.522803115844726,
      "step": 760
    },
    {
      "epoch": 0.154,
      "grad_norm": 0.6810743808746338,
      "learning_rate": 0.0002973626742264713,
      "loss": 4.3536125183105465,
      "step": 770
    },
    {
      "epoch": 0.156,
      "grad_norm": 0.4122457504272461,
      "learning_rate": 0.00029716357761650007,
      "loss": 4.4033355712890625,
      "step": 780
    },
    {
      "epoch": 0.158,
      "grad_norm": 0.4085232615470886,
      "learning_rate": 0.00029695730846137275,
      "loss": 4.320112609863282,
      "step": 790
    },
    {
      "epoch": 0.16,
      "grad_norm": 0.6114133596420288,
      "learning_rate": 0.0002967438768143568,
      "loss": 4.373843383789063,
      "step": 800
    },
    {
      "epoch": 0.162,
      "grad_norm": 0.5036770105361938,
      "learning_rate": 0.0002965232930778097,
      "loss": 4.2889457702636715,
      "step": 810
    },
    {
      "epoch": 0.164,
      "grad_norm": 0.4276726245880127,
      "learning_rate": 0.0002962955680026715,
      "loss": 4.212138366699219,
      "step": 820
    },
    {
      "epoch": 0.166,
      "grad_norm": 1.472080111503601,
      "learning_rate": 0.0002960607126879411,
      "loss": 4.297634124755859,
      "step": 830
    },
    {
      "epoch": 0.168,
      "grad_norm": 0.5528949499130249,
      "learning_rate": 0.00029581873858013525,
      "loss": 4.357148742675781,
      "step": 840
    },
    {
      "epoch": 0.17,
      "grad_norm": 0.5142295956611633,
      "learning_rate": 0.0002955696574727306,
      "loss": 4.17579231262207,
      "step": 850
    },
    {
      "epoch": 0.172,
      "grad_norm": 0.45028647780418396,
      "learning_rate": 0.00029531348150558883,
      "loss": 4.272770690917969,
      "step": 860
    },
    {
      "epoch": 0.174,
      "grad_norm": 0.38256195187568665,
      "learning_rate": 0.00029505022316436523,
      "loss": 4.236406707763672,
      "step": 870
    },
    {
      "epoch": 0.176,
      "grad_norm": 0.7847526669502258,
      "learning_rate": 0.00029477989527989997,
      "loss": 4.145956420898438,
      "step": 880
    },
    {
      "epoch": 0.178,
      "grad_norm": 1.6398742198944092,
      "learning_rate": 0.0002945025110275927,
      "loss": 4.173336029052734,
      "step": 890
    },
    {
      "epoch": 0.18,
      "grad_norm": 0.5771706700325012,
      "learning_rate": 0.0002942180839267606,
      "loss": 4.133999633789062,
      "step": 900
    },
    {
      "epoch": 0.182,
      "grad_norm": 0.4774192273616791,
      "learning_rate": 0.0002939266278399793,
      "loss": 4.1810859680175785,
      "step": 910
    },
    {
      "epoch": 0.184,
      "grad_norm": 0.4462798833847046,
      "learning_rate": 0.00029362815697240723,
      "loss": 4.148078155517578,
      "step": 920
    },
    {
      "epoch": 0.186,
      "grad_norm": 0.5258499383926392,
      "learning_rate": 0.0002933226858710934,
      "loss": 4.151172637939453,
      "step": 930
    },
    {
      "epoch": 0.188,
      "grad_norm": 0.4309222996234894,
      "learning_rate": 0.00029301022942426844,
      "loss": 4.16070442199707,
      "step": 940
    },
    {
      "epoch": 0.19,
      "grad_norm": 0.40204179286956787,
      "learning_rate": 0.00029269080286061866,
      "loss": 4.0530345916748045,
      "step": 950
    },
    {
      "epoch": 0.192,
      "grad_norm": 0.3796781301498413,
      "learning_rate": 0.00029236442174854416,
      "loss": 3.982533264160156,
      "step": 960
    },
    {
      "epoch": 0.194,
      "grad_norm": 0.3593500256538391,
      "learning_rate": 0.00029203110199539986,
      "loss": 4.06689338684082,
      "step": 970
    },
    {
      "epoch": 0.196,
      "grad_norm": 0.3557942807674408,
      "learning_rate": 0.0002916908598467203,
      "loss": 4.02539176940918,
      "step": 980
    },
    {
      "epoch": 0.198,
      "grad_norm": 0.3429299294948578,
      "learning_rate": 0.00029134371188542784,
      "loss": 4.072704315185547,
      "step": 990
    },
    {
      "epoch": 0.2,
      "grad_norm": 0.42700159549713135,
      "learning_rate": 0.0002909896750310243,
      "loss": 4.072047424316406,
      "step": 1000
    }
  ],
  "logging_steps": 10,
  "max_steps": 5000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 9223372036854775807,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 3.45182897700864e+17,
  "train_batch_size": 1,
  "trial_name": null,
  "trial_params": null
}