{
  "best_metric": 0.4145749807357788,
  "best_model_checkpoint": "Action_all_10_class/checkpoint-1400",
  "epoch": 5.0,
  "eval_steps": 100,
  "global_step": 1445,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.03,
      "grad_norm": 2.3712973594665527,
      "learning_rate": 0.0001986159169550173,
      "loss": 2.271,
      "step": 10
    },
    {
      "epoch": 0.07,
      "grad_norm": 1.8230082988739014,
      "learning_rate": 0.0001972318339100346,
      "loss": 2.0335,
      "step": 20
    },
    {
      "epoch": 0.1,
      "grad_norm": 2.621448040008545,
      "learning_rate": 0.0001958477508650519,
      "loss": 1.8885,
      "step": 30
    },
    {
      "epoch": 0.14,
      "grad_norm": 3.194303274154663,
      "learning_rate": 0.0001944636678200692,
      "loss": 1.6835,
      "step": 40
    },
    {
      "epoch": 0.17,
      "grad_norm": 2.9692325592041016,
      "learning_rate": 0.0001930795847750865,
      "loss": 1.5658,
      "step": 50
    },
    {
      "epoch": 0.21,
      "grad_norm": 3.6100592613220215,
      "learning_rate": 0.00019169550173010383,
      "loss": 1.4881,
      "step": 60
    },
    {
      "epoch": 0.24,
      "grad_norm": 2.926745653152466,
      "learning_rate": 0.00019031141868512113,
      "loss": 1.2581,
      "step": 70
    },
    {
      "epoch": 0.28,
      "grad_norm": 3.4733943939208984,
      "learning_rate": 0.00018892733564013843,
      "loss": 1.1786,
      "step": 80
    },
    {
      "epoch": 0.31,
      "grad_norm": 3.609900951385498,
      "learning_rate": 0.00018754325259515573,
      "loss": 1.1612,
      "step": 90
    },
    {
      "epoch": 0.35,
      "grad_norm": 3.1664867401123047,
      "learning_rate": 0.00018615916955017303,
      "loss": 1.1239,
      "step": 100
    },
    {
      "epoch": 0.35,
      "eval_accuracy": 0.7116564417177914,
      "eval_loss": 0.9933902025222778,
      "eval_runtime": 13.7579,
      "eval_samples_per_second": 59.239,
      "eval_steps_per_second": 7.414,
      "step": 100
    },
    {
      "epoch": 0.38,
      "grad_norm": 3.3514106273651123,
      "learning_rate": 0.00018477508650519033,
      "loss": 1.1763,
      "step": 110
    },
    {
      "epoch": 0.42,
      "grad_norm": 2.530064821243286,
      "learning_rate": 0.0001833910034602076,
      "loss": 1.1493,
      "step": 120
    },
    {
      "epoch": 0.45,
      "grad_norm": 4.615677833557129,
      "learning_rate": 0.0001820069204152249,
      "loss": 1.1661,
      "step": 130
    },
    {
      "epoch": 0.48,
      "grad_norm": 5.2710065841674805,
      "learning_rate": 0.0001807612456747405,
      "loss": 1.1174,
      "step": 140
    },
    {
      "epoch": 0.52,
      "grad_norm": 2.854151487350464,
      "learning_rate": 0.0001793771626297578,
      "loss": 1.0679,
      "step": 150
    },
    {
      "epoch": 0.55,
      "grad_norm": 4.68364143371582,
      "learning_rate": 0.0001779930795847751,
      "loss": 1.1566,
      "step": 160
    },
    {
      "epoch": 0.59,
      "grad_norm": 5.799492359161377,
      "learning_rate": 0.0001766089965397924,
      "loss": 0.9811,
      "step": 170
    },
    {
      "epoch": 0.62,
      "grad_norm": 7.8334760665893555,
      "learning_rate": 0.0001752249134948097,
      "loss": 1.0781,
      "step": 180
    },
    {
      "epoch": 0.66,
      "grad_norm": 4.817080020904541,
      "learning_rate": 0.000173840830449827,
      "loss": 1.0302,
      "step": 190
    },
    {
      "epoch": 0.69,
      "grad_norm": 4.084784507751465,
      "learning_rate": 0.0001724567474048443,
      "loss": 0.8878,
      "step": 200
    },
    {
      "epoch": 0.69,
      "eval_accuracy": 0.7705521472392638,
      "eval_loss": 0.7667088508605957,
      "eval_runtime": 10.9837,
      "eval_samples_per_second": 74.201,
      "eval_steps_per_second": 9.286,
      "step": 200
    },
    {
      "epoch": 0.73,
      "grad_norm": 3.8510003089904785,
      "learning_rate": 0.0001710726643598616,
      "loss": 1.0539,
      "step": 210
    },
    {
      "epoch": 0.76,
      "grad_norm": 4.4905619621276855,
      "learning_rate": 0.00016968858131487892,
      "loss": 0.9855,
      "step": 220
    },
    {
      "epoch": 0.8,
      "grad_norm": 3.903388738632202,
      "learning_rate": 0.00016830449826989622,
      "loss": 1.0442,
      "step": 230
    },
    {
      "epoch": 0.83,
      "grad_norm": 4.052041530609131,
      "learning_rate": 0.00016692041522491352,
      "loss": 1.004,
      "step": 240
    },
    {
      "epoch": 0.87,
      "grad_norm": 4.503437042236328,
      "learning_rate": 0.00016553633217993081,
      "loss": 0.8825,
      "step": 250
    },
    {
      "epoch": 0.9,
      "grad_norm": 4.2403459548950195,
      "learning_rate": 0.00016415224913494811,
      "loss": 0.9616,
      "step": 260
    },
    {
      "epoch": 0.93,
      "grad_norm": 6.690958499908447,
      "learning_rate": 0.00016276816608996541,
      "loss": 0.9127,
      "step": 270
    },
    {
      "epoch": 0.97,
      "grad_norm": 6.591899394989014,
      "learning_rate": 0.0001613840830449827,
      "loss": 1.0545,
      "step": 280
    },
    {
      "epoch": 1.0,
      "grad_norm": 2.814940929412842,
      "learning_rate": 0.0001601384083044983,
      "loss": 0.9937,
      "step": 290
    },
    {
      "epoch": 1.04,
      "grad_norm": 5.890912055969238,
      "learning_rate": 0.00015875432525951557,
      "loss": 0.9037,
      "step": 300
    },
    {
      "epoch": 1.04,
      "eval_accuracy": 0.8098159509202454,
      "eval_loss": 0.6369422674179077,
      "eval_runtime": 11.1962,
      "eval_samples_per_second": 72.793,
      "eval_steps_per_second": 9.11,
      "step": 300
    },
    {
      "epoch": 1.07,
      "grad_norm": 3.311278820037842,
      "learning_rate": 0.00015737024221453287,
      "loss": 0.825,
      "step": 310
    },
    {
      "epoch": 1.11,
      "grad_norm": 4.0047454833984375,
      "learning_rate": 0.00015598615916955017,
      "loss": 0.7347,
      "step": 320
    },
    {
      "epoch": 1.14,
      "grad_norm": 4.145818710327148,
      "learning_rate": 0.00015460207612456747,
      "loss": 0.8906,
      "step": 330
    },
    {
      "epoch": 1.18,
      "grad_norm": 5.543643951416016,
      "learning_rate": 0.00015321799307958477,
      "loss": 0.7669,
      "step": 340
    },
    {
      "epoch": 1.21,
      "grad_norm": 5.490930080413818,
      "learning_rate": 0.00015183391003460207,
      "loss": 0.774,
      "step": 350
    },
    {
      "epoch": 1.25,
      "grad_norm": 4.895139217376709,
      "learning_rate": 0.00015044982698961937,
      "loss": 0.6658,
      "step": 360
    },
    {
      "epoch": 1.28,
      "grad_norm": 4.527533054351807,
      "learning_rate": 0.00014906574394463667,
      "loss": 0.8603,
      "step": 370
    },
    {
      "epoch": 1.31,
      "grad_norm": 3.3609440326690674,
      "learning_rate": 0.00014768166089965397,
      "loss": 0.7099,
      "step": 380
    },
    {
      "epoch": 1.35,
      "grad_norm": 10.356605529785156,
      "learning_rate": 0.0001462975778546713,
      "loss": 0.7262,
      "step": 390
    },
    {
      "epoch": 1.38,
      "grad_norm": 3.6397433280944824,
      "learning_rate": 0.0001449134948096886,
      "loss": 0.7307,
      "step": 400
    },
    {
      "epoch": 1.38,
      "eval_accuracy": 0.8319018404907975,
      "eval_loss": 0.577174186706543,
      "eval_runtime": 11.1343,
      "eval_samples_per_second": 73.197,
      "eval_steps_per_second": 9.161,
      "step": 400
    },
    {
      "epoch": 1.42,
      "grad_norm": 5.693297386169434,
      "learning_rate": 0.0001435294117647059,
      "loss": 0.8061,
      "step": 410
    },
    {
      "epoch": 1.45,
      "grad_norm": 3.4219260215759277,
      "learning_rate": 0.0001421453287197232,
      "loss": 0.6227,
      "step": 420
    },
    {
      "epoch": 1.49,
      "grad_norm": 9.225582122802734,
      "learning_rate": 0.0001407612456747405,
      "loss": 0.6429,
      "step": 430
    },
    {
      "epoch": 1.52,
      "grad_norm": 5.8882293701171875,
      "learning_rate": 0.0001393771626297578,
      "loss": 0.686,
      "step": 440
    },
    {
      "epoch": 1.56,
      "grad_norm": 3.3229711055755615,
      "learning_rate": 0.0001379930795847751,
      "loss": 0.7574,
      "step": 450
    },
    {
      "epoch": 1.59,
      "grad_norm": 5.165830135345459,
      "learning_rate": 0.0001366089965397924,
      "loss": 0.6195,
      "step": 460
    },
    {
      "epoch": 1.63,
      "grad_norm": 6.676501274108887,
      "learning_rate": 0.0001352249134948097,
      "loss": 0.6789,
      "step": 470
    },
    {
      "epoch": 1.66,
      "grad_norm": 6.1190643310546875,
      "learning_rate": 0.000133840830449827,
      "loss": 0.7149,
      "step": 480
    },
    {
      "epoch": 1.7,
      "grad_norm": 2.723904848098755,
      "learning_rate": 0.0001324567474048443,
      "loss": 0.7014,
      "step": 490
    },
    {
      "epoch": 1.73,
      "grad_norm": 4.995339870452881,
      "learning_rate": 0.00013107266435986162,
      "loss": 0.6624,
      "step": 500
    },
    {
      "epoch": 1.73,
      "eval_accuracy": 0.7717791411042945,
      "eval_loss": 0.6924724578857422,
      "eval_runtime": 11.3854,
      "eval_samples_per_second": 71.583,
      "eval_steps_per_second": 8.959,
      "step": 500
    },
    {
      "epoch": 1.76,
      "grad_norm": 5.440706729888916,
      "learning_rate": 0.00012968858131487892,
      "loss": 0.8681,
      "step": 510
    },
    {
      "epoch": 1.8,
      "grad_norm": 4.539951801300049,
      "learning_rate": 0.0001283044982698962,
      "loss": 0.5997,
      "step": 520
    },
    {
      "epoch": 1.83,
      "grad_norm": 4.5558013916015625,
      "learning_rate": 0.0001269204152249135,
      "loss": 0.6839,
      "step": 530
    },
    {
      "epoch": 1.87,
      "grad_norm": 4.361762046813965,
      "learning_rate": 0.0001255363321799308,
      "loss": 0.6016,
      "step": 540
    },
    {
      "epoch": 1.9,
      "grad_norm": 4.278672695159912,
      "learning_rate": 0.0001241522491349481,
      "loss": 0.6111,
      "step": 550
    },
    {
      "epoch": 1.94,
      "grad_norm": 5.990556240081787,
      "learning_rate": 0.0001227681660899654,
      "loss": 0.9729,
      "step": 560
    },
    {
      "epoch": 1.97,
      "grad_norm": 4.511960983276367,
      "learning_rate": 0.0001213840830449827,
      "loss": 0.6829,
      "step": 570
    },
    {
      "epoch": 2.01,
      "grad_norm": 3.139665365219116,
      "learning_rate": 0.00012,
      "loss": 0.6516,
      "step": 580
    },
    {
      "epoch": 2.04,
      "grad_norm": 2.4773037433624268,
      "learning_rate": 0.0001186159169550173,
      "loss": 0.492,
      "step": 590
    },
    {
      "epoch": 2.08,
      "grad_norm": 1.613572597503662,
      "learning_rate": 0.0001172318339100346,
      "loss": 0.5781,
      "step": 600
    },
    {
      "epoch": 2.08,
      "eval_accuracy": 0.8404907975460123,
      "eval_loss": 0.5438538789749146,
      "eval_runtime": 11.1336,
      "eval_samples_per_second": 73.202,
      "eval_steps_per_second": 9.161,
      "step": 600
    },
    {
      "epoch": 2.11,
      "grad_norm": 2.5122666358947754,
      "learning_rate": 0.0001158477508650519,
      "loss": 0.55,
      "step": 610
    },
    {
      "epoch": 2.15,
      "grad_norm": 5.037026882171631,
      "learning_rate": 0.00011446366782006921,
      "loss": 0.5404,
      "step": 620
    },
    {
      "epoch": 2.18,
      "grad_norm": 6.795055866241455,
      "learning_rate": 0.00011307958477508651,
      "loss": 0.6455,
      "step": 630
    },
    {
      "epoch": 2.21,
      "grad_norm": 6.474560260772705,
      "learning_rate": 0.00011169550173010381,
      "loss": 0.5184,
      "step": 640
    },
    {
      "epoch": 2.25,
      "grad_norm": 8.110054969787598,
      "learning_rate": 0.00011031141868512111,
      "loss": 0.4706,
      "step": 650
    },
    {
      "epoch": 2.28,
      "grad_norm": 2.1118876934051514,
      "learning_rate": 0.00010892733564013841,
      "loss": 0.5291,
      "step": 660
    },
    {
      "epoch": 2.32,
      "grad_norm": 5.386865615844727,
      "learning_rate": 0.00010754325259515571,
      "loss": 0.605,
      "step": 670
    },
    {
      "epoch": 2.35,
      "grad_norm": 4.501631736755371,
      "learning_rate": 0.00010615916955017303,
      "loss": 0.4696,
      "step": 680
    },
    {
      "epoch": 2.39,
      "grad_norm": 4.06800651550293,
      "learning_rate": 0.00010477508650519033,
      "loss": 0.5214,
      "step": 690
    },
    {
      "epoch": 2.42,
      "grad_norm": 3.3574037551879883,
      "learning_rate": 0.00010339100346020762,
      "loss": 0.5537,
      "step": 700
    },
    {
      "epoch": 2.42,
      "eval_accuracy": 0.8331288343558282,
      "eval_loss": 0.5256926417350769,
      "eval_runtime": 11.3468,
      "eval_samples_per_second": 71.826,
      "eval_steps_per_second": 8.989,
      "step": 700
    },
    {
      "epoch": 2.46,
      "grad_norm": 6.940188884735107,
      "learning_rate": 0.00010200692041522492,
      "loss": 0.5371,
      "step": 710
    },
    {
      "epoch": 2.49,
      "grad_norm": 5.517166614532471,
      "learning_rate": 0.00010076124567474047,
      "loss": 0.5419,
      "step": 720
    },
    {
      "epoch": 2.53,
      "grad_norm": 3.2714200019836426,
      "learning_rate": 9.937716262975779e-05,
      "loss": 0.6657,
      "step": 730
    },
    {
      "epoch": 2.56,
      "grad_norm": 2.734272003173828,
      "learning_rate": 9.79930795847751e-05,
      "loss": 0.5712,
      "step": 740
    },
    {
      "epoch": 2.6,
      "grad_norm": 7.275644302368164,
      "learning_rate": 9.66089965397924e-05,
      "loss": 0.6513,
      "step": 750
    },
    {
      "epoch": 2.63,
      "grad_norm": 4.910625457763672,
      "learning_rate": 9.52249134948097e-05,
      "loss": 0.5641,
      "step": 760
    },
    {
      "epoch": 2.66,
      "grad_norm": 2.4771077632904053,
      "learning_rate": 9.384083044982698e-05,
      "loss": 0.4163,
      "step": 770
    },
    {
      "epoch": 2.7,
      "grad_norm": 4.3278303146362305,
      "learning_rate": 9.24567474048443e-05,
      "loss": 0.5759,
      "step": 780
    },
    {
      "epoch": 2.73,
      "grad_norm": 8.122814178466797,
      "learning_rate": 9.10726643598616e-05,
      "loss": 0.6389,
      "step": 790
    },
    {
      "epoch": 2.77,
      "grad_norm": 6.369782447814941,
      "learning_rate": 8.96885813148789e-05,
      "loss": 0.4112,
      "step": 800
    },
    {
      "epoch": 2.77,
      "eval_accuracy": 0.8564417177914111,
      "eval_loss": 0.4499729573726654,
      "eval_runtime": 11.5138,
      "eval_samples_per_second": 70.785,
      "eval_steps_per_second": 8.859,
      "step": 800
    },
    {
      "epoch": 2.8,
      "grad_norm": 7.130208969116211,
      "learning_rate": 8.83044982698962e-05,
      "loss": 0.4534,
      "step": 810
    },
    {
      "epoch": 2.84,
      "grad_norm": 5.094985008239746,
      "learning_rate": 8.69204152249135e-05,
      "loss": 0.4881,
      "step": 820
    },
    {
      "epoch": 2.87,
      "grad_norm": 2.527794361114502,
      "learning_rate": 8.55363321799308e-05,
      "loss": 0.2775,
      "step": 830
    },
    {
      "epoch": 2.91,
      "grad_norm": 9.366188049316406,
      "learning_rate": 8.415224913494811e-05,
      "loss": 0.6528,
      "step": 840
    },
    {
      "epoch": 2.94,
      "grad_norm": 3.863041400909424,
      "learning_rate": 8.276816608996541e-05,
      "loss": 0.5632,
      "step": 850
    },
    {
      "epoch": 2.98,
      "grad_norm": 3.3894691467285156,
      "learning_rate": 8.138408304498271e-05,
      "loss": 0.5885,
      "step": 860
    },
    {
      "epoch": 3.01,
      "grad_norm": 5.836617469787598,
      "learning_rate": 8e-05,
      "loss": 0.5864,
      "step": 870
    },
    {
      "epoch": 3.04,
      "grad_norm": 4.380435943603516,
      "learning_rate": 7.86159169550173e-05,
      "loss": 0.4523,
      "step": 880
    },
    {
      "epoch": 3.08,
      "grad_norm": 0.4667840898036957,
      "learning_rate": 7.72318339100346e-05,
      "loss": 0.2999,
      "step": 890
    },
    {
      "epoch": 3.11,
      "grad_norm": 3.003122091293335,
      "learning_rate": 7.58477508650519e-05,
      "loss": 0.3263,
      "step": 900
    },
    {
      "epoch": 3.11,
      "eval_accuracy": 0.841717791411043,
      "eval_loss": 0.4910809397697449,
      "eval_runtime": 11.4121,
      "eval_samples_per_second": 71.416,
      "eval_steps_per_second": 8.938,
      "step": 900
    },
    {
      "epoch": 3.15,
      "grad_norm": 4.519166469573975,
      "learning_rate": 7.44636678200692e-05,
      "loss": 0.417,
      "step": 910
    },
    {
      "epoch": 3.18,
      "grad_norm": 6.071402549743652,
      "learning_rate": 7.30795847750865e-05,
      "loss": 0.351,
      "step": 920
    },
    {
      "epoch": 3.22,
      "grad_norm": 6.0896687507629395,
      "learning_rate": 7.16955017301038e-05,
      "loss": 0.4687,
      "step": 930
    },
    {
      "epoch": 3.25,
      "grad_norm": 4.001879692077637,
      "learning_rate": 7.031141868512112e-05,
      "loss": 0.5083,
      "step": 940
    },
    {
      "epoch": 3.29,
      "grad_norm": 5.064093112945557,
      "learning_rate": 6.892733564013842e-05,
      "loss": 0.3407,
      "step": 950
    },
    {
      "epoch": 3.32,
      "grad_norm": 1.3715455532073975,
      "learning_rate": 6.754325259515572e-05,
      "loss": 0.4073,
      "step": 960
    },
    {
      "epoch": 3.36,
      "grad_norm": 6.380220413208008,
      "learning_rate": 6.615916955017302e-05,
      "loss": 0.4613,
      "step": 970
    },
    {
      "epoch": 3.39,
      "grad_norm": 2.1010074615478516,
      "learning_rate": 6.477508650519032e-05,
      "loss": 0.3706,
      "step": 980
    },
    {
      "epoch": 3.43,
      "grad_norm": 2.2640676498413086,
      "learning_rate": 6.339100346020761e-05,
      "loss": 0.3235,
      "step": 990
    },
    {
      "epoch": 3.46,
      "grad_norm": 2.6152210235595703,
      "learning_rate": 6.200692041522491e-05,
      "loss": 0.4592,
      "step": 1000
    },
    {
      "epoch": 3.46,
      "eval_accuracy": 0.8711656441717791,
      "eval_loss": 0.4550653100013733,
      "eval_runtime": 11.3924,
      "eval_samples_per_second": 71.539,
      "eval_steps_per_second": 8.953,
      "step": 1000
    },
    {
      "epoch": 3.49,
      "grad_norm": 4.868244647979736,
      "learning_rate": 6.0622837370242214e-05,
      "loss": 0.3891,
      "step": 1010
    },
    {
      "epoch": 3.53,
      "grad_norm": 4.125463962554932,
      "learning_rate": 5.9238754325259514e-05,
      "loss": 0.3462,
      "step": 1020
    },
    {
      "epoch": 3.56,
      "grad_norm": 5.510716915130615,
      "learning_rate": 5.785467128027682e-05,
      "loss": 0.3938,
      "step": 1030
    },
    {
      "epoch": 3.6,
      "grad_norm": 4.867416858673096,
      "learning_rate": 5.647058823529412e-05,
      "loss": 0.3851,
      "step": 1040
    },
    {
      "epoch": 3.63,
      "grad_norm": 5.599556922912598,
      "learning_rate": 5.508650519031142e-05,
      "loss": 0.3939,
      "step": 1050
    },
    {
      "epoch": 3.67,
      "grad_norm": 6.144674777984619,
      "learning_rate": 5.3702422145328725e-05,
      "loss": 0.5787,
      "step": 1060
    },
    {
      "epoch": 3.7,
      "grad_norm": 6.78063440322876,
      "learning_rate": 5.2318339100346025e-05,
      "loss": 0.4548,
      "step": 1070
    },
    {
      "epoch": 3.74,
      "grad_norm": 2.213515281677246,
      "learning_rate": 5.0934256055363325e-05,
      "loss": 0.418,
      "step": 1080
    },
    {
      "epoch": 3.77,
      "grad_norm": 3.3876266479492188,
      "learning_rate": 4.9550173010380624e-05,
      "loss": 0.432,
      "step": 1090
    },
    {
      "epoch": 3.81,
      "grad_norm": 6.500645160675049,
      "learning_rate": 4.8166089965397924e-05,
      "loss": 0.3204,
      "step": 1100
    },
    {
      "epoch": 3.81,
      "eval_accuracy": 0.8723926380368098,
      "eval_loss": 0.4324829876422882,
      "eval_runtime": 11.2965,
      "eval_samples_per_second": 72.146,
      "eval_steps_per_second": 9.029,
      "step": 1100
    },
    {
      "epoch": 3.84,
      "grad_norm": 2.518057107925415,
      "learning_rate": 4.678200692041523e-05,
      "loss": 0.3505,
      "step": 1110
    },
    {
      "epoch": 3.88,
      "grad_norm": 4.820311546325684,
      "learning_rate": 4.539792387543253e-05,
      "loss": 0.4641,
      "step": 1120
    },
    {
      "epoch": 3.91,
      "grad_norm": 1.1859557628631592,
      "learning_rate": 4.401384083044983e-05,
      "loss": 0.5202,
      "step": 1130
    },
    {
      "epoch": 3.94,
      "grad_norm": 2.3869946002960205,
      "learning_rate": 4.262975778546713e-05,
      "loss": 0.5007,
      "step": 1140
    },
    {
      "epoch": 3.98,
      "grad_norm": 6.64663553237915,
      "learning_rate": 4.124567474048443e-05,
      "loss": 0.3478,
      "step": 1150
    },
    {
      "epoch": 4.01,
      "grad_norm": 4.931898593902588,
      "learning_rate": 3.9861591695501735e-05,
      "loss": 0.3808,
      "step": 1160
    },
    {
      "epoch": 4.05,
      "grad_norm": 1.7521384954452515,
      "learning_rate": 3.8477508650519034e-05,
      "loss": 0.3247,
      "step": 1170
    },
    {
      "epoch": 4.08,
      "grad_norm": 1.5513068437576294,
      "learning_rate": 3.7093425605536334e-05,
      "loss": 0.393,
      "step": 1180
    },
    {
      "epoch": 4.12,
      "grad_norm": 5.2524094581604,
      "learning_rate": 3.570934256055363e-05,
      "loss": 0.3697,
      "step": 1190
    },
    {
      "epoch": 4.15,
      "grad_norm": 5.346503734588623,
      "learning_rate": 3.432525951557093e-05,
      "loss": 0.3268,
      "step": 1200
    },
    {
      "epoch": 4.15,
      "eval_accuracy": 0.8539877300613496,
      "eval_loss": 0.4529338777065277,
      "eval_runtime": 11.243,
      "eval_samples_per_second": 72.49,
      "eval_steps_per_second": 9.072,
      "step": 1200
    },
    {
      "epoch": 4.19,
      "grad_norm": 2.0786306858062744,
      "learning_rate": 3.294117647058824e-05,
      "loss": 0.2967,
      "step": 1210
    },
    {
      "epoch": 4.22,
      "grad_norm": 3.8141846656799316,
      "learning_rate": 3.155709342560554e-05,
      "loss": 0.2483,
      "step": 1220
    },
    {
      "epoch": 4.26,
      "grad_norm": 1.8864765167236328,
      "learning_rate": 3.0173010380622842e-05,
      "loss": 0.2443,
      "step": 1230
    },
    {
      "epoch": 4.29,
      "grad_norm": 0.9406078457832336,
      "learning_rate": 2.878892733564014e-05,
      "loss": 0.2637,
      "step": 1240
    },
    {
      "epoch": 4.33,
      "grad_norm": 3.418455123901367,
      "learning_rate": 2.7404844290657437e-05,
      "loss": 0.2866,
      "step": 1250
    },
    {
      "epoch": 4.36,
      "grad_norm": 1.8662397861480713,
      "learning_rate": 2.602076124567474e-05,
      "loss": 0.3381,
      "step": 1260
    },
    {
      "epoch": 4.39,
      "grad_norm": 3.9735002517700195,
      "learning_rate": 2.4636678200692043e-05,
      "loss": 0.3328,
      "step": 1270
    },
    {
      "epoch": 4.43,
      "grad_norm": 2.229581117630005,
      "learning_rate": 2.3252595155709346e-05,
      "loss": 0.4579,
      "step": 1280
    },
    {
      "epoch": 4.46,
      "grad_norm": 3.8186144828796387,
      "learning_rate": 2.1868512110726642e-05,
      "loss": 0.2986,
      "step": 1290
    },
    {
      "epoch": 4.5,
      "grad_norm": 2.310910701751709,
      "learning_rate": 2.0484429065743945e-05,
      "loss": 0.4267,
      "step": 1300
    },
    {
      "epoch": 4.5,
      "eval_accuracy": 0.8723926380368098,
      "eval_loss": 0.4355594515800476,
      "eval_runtime": 11.2323,
      "eval_samples_per_second": 72.559,
      "eval_steps_per_second": 9.081,
      "step": 1300
    },
    {
      "epoch": 4.53,
      "grad_norm": 5.4986042976379395,
      "learning_rate": 1.910034602076125e-05,
      "loss": 0.3289,
      "step": 1310
    },
    {
      "epoch": 4.57,
      "grad_norm": 9.17880916595459,
      "learning_rate": 1.7716262975778548e-05,
      "loss": 0.4384,
      "step": 1320
    },
    {
      "epoch": 4.6,
      "grad_norm": 0.5825958251953125,
      "learning_rate": 1.6332179930795848e-05,
      "loss": 0.1629,
      "step": 1330
    },
    {
      "epoch": 4.64,
      "grad_norm": 3.7861948013305664,
      "learning_rate": 1.4948096885813149e-05,
      "loss": 0.2607,
      "step": 1340
    },
    {
      "epoch": 4.67,
      "grad_norm": 0.08592710644006729,
      "learning_rate": 1.356401384083045e-05,
      "loss": 0.23,
      "step": 1350
    },
    {
      "epoch": 4.71,
      "grad_norm": 3.897566556930542,
      "learning_rate": 1.2179930795847751e-05,
      "loss": 0.2927,
      "step": 1360
    },
    {
      "epoch": 4.74,
      "grad_norm": 3.4900457859039307,
      "learning_rate": 1.0795847750865053e-05,
      "loss": 0.3149,
      "step": 1370
    },
    {
      "epoch": 4.78,
      "grad_norm": 0.48032164573669434,
      "learning_rate": 9.411764705882354e-06,
      "loss": 0.3346,
      "step": 1380
    },
    {
      "epoch": 4.81,
      "grad_norm": 3.148484945297241,
      "learning_rate": 8.027681660899653e-06,
      "loss": 0.3006,
      "step": 1390
    },
    {
      "epoch": 4.84,
      "grad_norm": 8.376299858093262,
      "learning_rate": 6.6435986159169555e-06,
      "loss": 0.2886,
      "step": 1400
    },
    {
      "epoch": 4.84,
      "eval_accuracy": 0.8785276073619632,
      "eval_loss": 0.4145749807357788,
      "eval_runtime": 11.483,
      "eval_samples_per_second": 70.974,
      "eval_steps_per_second": 8.883,
      "step": 1400
    },
    {
      "epoch": 4.88,
      "grad_norm": 10.592811584472656,
      "learning_rate": 5.259515570934257e-06,
      "loss": 0.2866,
      "step": 1410
    },
    {
      "epoch": 4.91,
      "grad_norm": 1.9005374908447266,
      "learning_rate": 3.875432525951557e-06,
      "loss": 0.367,
      "step": 1420
    },
    {
      "epoch": 4.95,
      "grad_norm": 6.674309730529785,
      "learning_rate": 2.4913494809688584e-06,
      "loss": 0.2715,
      "step": 1430
    },
    {
      "epoch": 4.98,
      "grad_norm": 7.622613430023193,
      "learning_rate": 1.1072664359861592e-06,
      "loss": 0.3306,
      "step": 1440
    },
    {
      "epoch": 5.0,
      "step": 1445,
      "total_flos": 1.789030847196795e+18,
      "train_loss": 0.6435174308433664,
      "train_runtime": 776.1745,
      "train_samples_per_second": 29.742,
      "train_steps_per_second": 1.862
    }
  ],
  "logging_steps": 10,
  "max_steps": 1445,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 100,
  "total_flos": 1.789030847196795e+18,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}