{
  "best_metric": 0.49757373332977295,
  "best_model_checkpoint": "../../experiments_checkpoints/MAdAiLab/google_t5/t5_small_twitter/checkpoint-200",
  "epoch": 3.0,
  "eval_steps": 50,
  "global_step": 816,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.04,
      "grad_norm": 0.9986526370048523,
      "learning_rate": 0.0004938725490196079,
      "loss": 0.6031,
      "step": 10
    },
    {
      "epoch": 0.07,
      "grad_norm": 0.9707651734352112,
      "learning_rate": 0.0004877450980392157,
      "loss": 0.5921,
      "step": 20
    },
    {
      "epoch": 0.11,
      "grad_norm": 1.3216536045074463,
      "learning_rate": 0.00048161764705882356,
      "loss": 0.5945,
      "step": 30
    },
    {
      "epoch": 0.15,
      "grad_norm": 0.8769760131835938,
      "learning_rate": 0.00047549019607843134,
      "loss": 0.5244,
      "step": 40
    },
    {
      "epoch": 0.18,
      "grad_norm": 1.3491202592849731,
      "learning_rate": 0.0004693627450980392,
      "loss": 0.4779,
      "step": 50
    },
    {
      "epoch": 0.18,
      "eval_accuracy": 0.7564338235294118,
      "eval_f1_macro": 0.7177028105371891,
      "eval_f1_micro": 0.7564338235294118,
      "eval_loss": 0.5150883793830872,
      "eval_runtime": 0.9275,
      "eval_samples_per_second": 1173.066,
      "eval_steps_per_second": 36.658,
      "step": 50
    },
    {
      "epoch": 0.22,
      "grad_norm": 2.29738450050354,
      "learning_rate": 0.0004632352941176471,
      "loss": 0.4934,
      "step": 60
    },
    {
      "epoch": 0.26,
      "grad_norm": 3.7542002201080322,
      "learning_rate": 0.0004571078431372549,
      "loss": 0.5007,
      "step": 70
    },
    {
      "epoch": 0.29,
      "grad_norm": 1.893402099609375,
      "learning_rate": 0.0004509803921568628,
      "loss": 0.4738,
      "step": 80
    },
    {
      "epoch": 0.33,
      "grad_norm": 2.0344367027282715,
      "learning_rate": 0.00044485294117647056,
      "loss": 0.4594,
      "step": 90
    },
    {
      "epoch": 0.37,
      "grad_norm": 2.5279059410095215,
      "learning_rate": 0.00043872549019607844,
      "loss": 0.5123,
      "step": 100
    },
    {
      "epoch": 0.37,
      "eval_accuracy": 0.7527573529411765,
      "eval_f1_macro": 0.6986562303405088,
      "eval_f1_micro": 0.7527573529411765,
      "eval_loss": 0.5059758424758911,
      "eval_runtime": 0.8718,
      "eval_samples_per_second": 1248.006,
      "eval_steps_per_second": 39.0,
      "step": 100
    },
    {
      "epoch": 0.4,
      "grad_norm": 1.4078208208084106,
      "learning_rate": 0.0004325980392156863,
      "loss": 0.516,
      "step": 110
    },
    {
      "epoch": 0.44,
      "grad_norm": 2.224971055984497,
      "learning_rate": 0.0004264705882352941,
      "loss": 0.5001,
      "step": 120
    },
    {
      "epoch": 0.48,
      "grad_norm": 1.39370858669281,
      "learning_rate": 0.000420343137254902,
      "loss": 0.4835,
      "step": 130
    },
    {
      "epoch": 0.51,
      "grad_norm": 0.9512380361557007,
      "learning_rate": 0.0004142156862745098,
      "loss": 0.5218,
      "step": 140
    },
    {
      "epoch": 0.55,
      "grad_norm": 1.359387755393982,
      "learning_rate": 0.00040808823529411766,
      "loss": 0.4617,
      "step": 150
    },
    {
      "epoch": 0.55,
      "eval_accuracy": 0.7270220588235294,
      "eval_f1_macro": 0.614936097953347,
      "eval_f1_micro": 0.7270220588235294,
      "eval_loss": 0.5286622047424316,
      "eval_runtime": 0.9283,
      "eval_samples_per_second": 1172.017,
      "eval_steps_per_second": 36.626,
      "step": 150
    },
    {
      "epoch": 0.59,
      "grad_norm": 1.1093153953552246,
      "learning_rate": 0.0004019607843137255,
      "loss": 0.4961,
      "step": 160
    },
    {
      "epoch": 0.62,
      "grad_norm": 2.4549243450164795,
      "learning_rate": 0.0003958333333333333,
      "loss": 0.5351,
      "step": 170
    },
    {
      "epoch": 0.66,
      "grad_norm": 1.2358920574188232,
      "learning_rate": 0.0003897058823529412,
      "loss": 0.5129,
      "step": 180
    },
    {
      "epoch": 0.7,
      "grad_norm": 1.4391891956329346,
      "learning_rate": 0.00038357843137254904,
      "loss": 0.4945,
      "step": 190
    },
    {
      "epoch": 0.74,
      "grad_norm": 1.3604400157928467,
      "learning_rate": 0.0003774509803921569,
      "loss": 0.4942,
      "step": 200
    },
    {
      "epoch": 0.74,
      "eval_accuracy": 0.7444852941176471,
      "eval_f1_macro": 0.6956024668695554,
      "eval_f1_micro": 0.7444852941176471,
      "eval_loss": 0.49757373332977295,
      "eval_runtime": 0.9302,
      "eval_samples_per_second": 1169.596,
      "eval_steps_per_second": 36.55,
      "step": 200
    },
    {
      "epoch": 0.77,
      "grad_norm": 1.0586875677108765,
      "learning_rate": 0.0003713235294117647,
      "loss": 0.5041,
      "step": 210
    },
    {
      "epoch": 0.81,
      "grad_norm": 5.99719762802124,
      "learning_rate": 0.00036519607843137254,
      "loss": 0.5088,
      "step": 220
    },
    {
      "epoch": 0.85,
      "grad_norm": 1.8691691160202026,
      "learning_rate": 0.0003590686274509804,
      "loss": 0.4577,
      "step": 230
    },
    {
      "epoch": 0.88,
      "grad_norm": 0.8101117610931396,
      "learning_rate": 0.00035294117647058826,
      "loss": 0.4562,
      "step": 240
    },
    {
      "epoch": 0.92,
      "grad_norm": 1.3150193691253662,
      "learning_rate": 0.0003468137254901961,
      "loss": 0.4783,
      "step": 250
    },
    {
      "epoch": 0.92,
      "eval_accuracy": 0.7573529411764706,
      "eval_f1_macro": 0.7124355755253073,
      "eval_f1_micro": 0.7573529411764706,
      "eval_loss": 0.49782079458236694,
      "eval_runtime": 0.8783,
      "eval_samples_per_second": 1238.797,
      "eval_steps_per_second": 38.712,
      "step": 250
    },
    {
      "epoch": 0.96,
      "grad_norm": 1.0855648517608643,
      "learning_rate": 0.0003406862745098039,
      "loss": 0.4206,
      "step": 260
    },
    {
      "epoch": 0.99,
      "grad_norm": 1.4224942922592163,
      "learning_rate": 0.00033455882352941176,
      "loss": 0.532,
      "step": 270
    },
    {
      "epoch": 1.03,
      "grad_norm": 2.6312692165374756,
      "learning_rate": 0.0003284313725490196,
      "loss": 0.465,
      "step": 280
    },
    {
      "epoch": 1.07,
      "grad_norm": 1.7025282382965088,
      "learning_rate": 0.0003223039215686275,
      "loss": 0.4146,
      "step": 290
    },
    {
      "epoch": 1.1,
      "grad_norm": 1.1280652284622192,
      "learning_rate": 0.0003161764705882353,
      "loss": 0.4369,
      "step": 300
    },
    {
      "epoch": 1.1,
      "eval_accuracy": 0.7601102941176471,
      "eval_f1_macro": 0.7124360123342397,
      "eval_f1_micro": 0.7601102941176471,
      "eval_loss": 0.5052167773246765,
      "eval_runtime": 0.8783,
      "eval_samples_per_second": 1238.751,
      "eval_steps_per_second": 38.711,
      "step": 300
    },
    {
      "epoch": 1.14,
      "grad_norm": 1.285010814666748,
      "learning_rate": 0.00031004901960784314,
      "loss": 0.448,
      "step": 310
    },
    {
      "epoch": 1.18,
      "grad_norm": 0.8714994788169861,
      "learning_rate": 0.00030392156862745097,
      "loss": 0.4204,
      "step": 320
    },
    {
      "epoch": 1.21,
      "grad_norm": 1.7784018516540527,
      "learning_rate": 0.0002977941176470588,
      "loss": 0.4939,
      "step": 330
    },
    {
      "epoch": 1.25,
      "grad_norm": 1.4992843866348267,
      "learning_rate": 0.0002916666666666667,
      "loss": 0.4632,
      "step": 340
    },
    {
      "epoch": 1.29,
      "grad_norm": 1.7776957750320435,
      "learning_rate": 0.0002855392156862745,
      "loss": 0.439,
      "step": 350
    },
    {
      "epoch": 1.29,
      "eval_accuracy": 0.7564338235294118,
      "eval_f1_macro": 0.7223909884910498,
      "eval_f1_micro": 0.7564338235294118,
      "eval_loss": 0.5092098116874695,
      "eval_runtime": 0.9311,
      "eval_samples_per_second": 1168.532,
      "eval_steps_per_second": 36.517,
      "step": 350
    },
    {
      "epoch": 1.32,
      "grad_norm": 1.2894947528839111,
      "learning_rate": 0.00027941176470588236,
      "loss": 0.4339,
      "step": 360
    },
    {
      "epoch": 1.36,
      "grad_norm": 1.262589693069458,
      "learning_rate": 0.0002732843137254902,
      "loss": 0.3804,
      "step": 370
    },
    {
      "epoch": 1.4,
      "grad_norm": 2.4415295124053955,
      "learning_rate": 0.000267156862745098,
      "loss": 0.5278,
      "step": 380
    },
    {
      "epoch": 1.43,
      "grad_norm": 0.8975765109062195,
      "learning_rate": 0.0002610294117647059,
      "loss": 0.4198,
      "step": 390
    },
    {
      "epoch": 1.47,
      "grad_norm": 1.8955622911453247,
      "learning_rate": 0.00025490196078431374,
      "loss": 0.4417,
      "step": 400
    },
    {
      "epoch": 1.47,
      "eval_accuracy": 0.7545955882352942,
      "eval_f1_macro": 0.6808490301741896,
      "eval_f1_micro": 0.7545955882352942,
      "eval_loss": 0.5228371620178223,
      "eval_runtime": 0.8846,
      "eval_samples_per_second": 1229.999,
      "eval_steps_per_second": 38.437,
      "step": 400
    },
    {
      "epoch": 1.51,
      "grad_norm": 1.5438319444656372,
      "learning_rate": 0.00024877450980392157,
      "loss": 0.4897,
      "step": 410
    },
    {
      "epoch": 1.54,
      "grad_norm": 2.117488384246826,
      "learning_rate": 0.0002426470588235294,
      "loss": 0.4292,
      "step": 420
    },
    {
      "epoch": 1.58,
      "grad_norm": 1.4254924058914185,
      "learning_rate": 0.00023651960784313726,
      "loss": 0.4201,
      "step": 430
    },
    {
      "epoch": 1.62,
      "grad_norm": 1.8626124858856201,
      "learning_rate": 0.0002303921568627451,
      "loss": 0.4013,
      "step": 440
    },
    {
      "epoch": 1.65,
      "grad_norm": 1.8252973556518555,
      "learning_rate": 0.00022426470588235296,
      "loss": 0.47,
      "step": 450
    },
    {
      "epoch": 1.65,
      "eval_accuracy": 0.7693014705882353,
      "eval_f1_macro": 0.7234537896394413,
      "eval_f1_micro": 0.7693014705882353,
      "eval_loss": 0.5087092518806458,
      "eval_runtime": 0.9392,
      "eval_samples_per_second": 1158.396,
      "eval_steps_per_second": 36.2,
      "step": 450
    },
    {
      "epoch": 1.69,
      "grad_norm": 1.173530101776123,
      "learning_rate": 0.0002181372549019608,
      "loss": 0.4267,
      "step": 460
    },
    {
      "epoch": 1.73,
      "grad_norm": 1.2974258661270142,
      "learning_rate": 0.00021200980392156862,
      "loss": 0.429,
      "step": 470
    },
    {
      "epoch": 1.76,
      "grad_norm": 1.577986717224121,
      "learning_rate": 0.00020588235294117645,
      "loss": 0.3808,
      "step": 480
    },
    {
      "epoch": 1.8,
      "grad_norm": 1.0223047733306885,
      "learning_rate": 0.00019975490196078434,
      "loss": 0.4225,
      "step": 490
    },
    {
      "epoch": 1.84,
      "grad_norm": 1.9677762985229492,
      "learning_rate": 0.00019362745098039217,
      "loss": 0.4415,
      "step": 500
    },
    {
      "epoch": 1.84,
      "eval_accuracy": 0.7647058823529411,
      "eval_f1_macro": 0.7262032085561497,
      "eval_f1_micro": 0.7647058823529411,
      "eval_loss": 0.510553240776062,
      "eval_runtime": 0.9392,
      "eval_samples_per_second": 1158.407,
      "eval_steps_per_second": 36.2,
      "step": 500
    },
    {
      "epoch": 1.88,
      "grad_norm": 1.1366698741912842,
      "learning_rate": 0.0001875,
      "loss": 0.4622,
      "step": 510
    },
    {
      "epoch": 1.91,
      "grad_norm": 1.5312350988388062,
      "learning_rate": 0.00018137254901960784,
      "loss": 0.4507,
      "step": 520
    },
    {
      "epoch": 1.95,
      "grad_norm": 1.2124278545379639,
      "learning_rate": 0.00017524509803921567,
      "loss": 0.4495,
      "step": 530
    },
    {
      "epoch": 1.99,
      "grad_norm": 1.3845272064208984,
      "learning_rate": 0.00016911764705882356,
      "loss": 0.4161,
      "step": 540
    },
    {
      "epoch": 2.02,
      "grad_norm": 1.2308210134506226,
      "learning_rate": 0.0001629901960784314,
      "loss": 0.4297,
      "step": 550
    },
    {
      "epoch": 2.02,
      "eval_accuracy": 0.7628676470588235,
      "eval_f1_macro": 0.7291254940711462,
      "eval_f1_micro": 0.7628676470588235,
      "eval_loss": 0.5022936463356018,
      "eval_runtime": 0.9417,
      "eval_samples_per_second": 1155.3,
      "eval_steps_per_second": 36.103,
      "step": 550
    },
    {
      "epoch": 2.06,
      "grad_norm": 1.803513765335083,
      "learning_rate": 0.00015686274509803922,
      "loss": 0.4231,
      "step": 560
    },
    {
      "epoch": 2.1,
      "grad_norm": 1.3074411153793335,
      "learning_rate": 0.00015073529411764705,
      "loss": 0.3723,
      "step": 570
    },
    {
      "epoch": 2.13,
      "grad_norm": 1.8602393865585327,
      "learning_rate": 0.0001446078431372549,
      "loss": 0.4003,
      "step": 580
    },
    {
      "epoch": 2.17,
      "grad_norm": 1.486925721168518,
      "learning_rate": 0.00013848039215686274,
      "loss": 0.3796,
      "step": 590
    },
    {
      "epoch": 2.21,
      "grad_norm": 1.5388500690460205,
      "learning_rate": 0.0001323529411764706,
      "loss": 0.4366,
      "step": 600
    },
    {
      "epoch": 2.21,
      "eval_accuracy": 0.7555147058823529,
      "eval_f1_macro": 0.7127019171206659,
      "eval_f1_micro": 0.7555147058823529,
      "eval_loss": 0.5225424766540527,
      "eval_runtime": 0.9394,
      "eval_samples_per_second": 1158.157,
      "eval_steps_per_second": 36.192,
      "step": 600
    },
    {
      "epoch": 2.24,
      "grad_norm": 1.38102388381958,
      "learning_rate": 0.00012622549019607844,
      "loss": 0.4028,
      "step": 610
    },
    {
      "epoch": 2.28,
      "grad_norm": 1.0879701375961304,
      "learning_rate": 0.00012009803921568628,
      "loss": 0.3619,
      "step": 620
    },
    {
      "epoch": 2.32,
      "grad_norm": 2.56364107131958,
      "learning_rate": 0.00011397058823529411,
      "loss": 0.3883,
      "step": 630
    },
    {
      "epoch": 2.35,
      "grad_norm": 1.285326600074768,
      "learning_rate": 0.00010784313725490197,
      "loss": 0.4058,
      "step": 640
    },
    {
      "epoch": 2.39,
      "grad_norm": 1.1956150531768799,
      "learning_rate": 0.0001017156862745098,
      "loss": 0.3623,
      "step": 650
    },
    {
      "epoch": 2.39,
      "eval_accuracy": 0.7582720588235294,
      "eval_f1_macro": 0.7157047718685389,
      "eval_f1_micro": 0.7582720588235294,
      "eval_loss": 0.522619366645813,
      "eval_runtime": 0.9391,
      "eval_samples_per_second": 1158.576,
      "eval_steps_per_second": 36.206,
      "step": 650
    },
    {
      "epoch": 2.43,
      "grad_norm": 1.5235141515731812,
      "learning_rate": 9.558823529411764e-05,
      "loss": 0.4147,
      "step": 660
    },
    {
      "epoch": 2.46,
      "grad_norm": 1.760062575340271,
      "learning_rate": 8.94607843137255e-05,
      "loss": 0.4255,
      "step": 670
    },
    {
      "epoch": 2.5,
      "grad_norm": 1.636400580406189,
      "learning_rate": 8.333333333333333e-05,
      "loss": 0.3687,
      "step": 680
    },
    {
      "epoch": 2.54,
      "grad_norm": 1.4576902389526367,
      "learning_rate": 7.720588235294118e-05,
      "loss": 0.3764,
      "step": 690
    },
    {
      "epoch": 2.57,
      "grad_norm": 1.5375081300735474,
      "learning_rate": 7.107843137254902e-05,
      "loss": 0.3337,
      "step": 700
    },
    {
      "epoch": 2.57,
      "eval_accuracy": 0.7573529411764706,
      "eval_f1_macro": 0.7143845832587554,
      "eval_f1_micro": 0.7573529411764706,
      "eval_loss": 0.5312902927398682,
      "eval_runtime": 0.9369,
      "eval_samples_per_second": 1161.294,
      "eval_steps_per_second": 36.29,
      "step": 700
    },
    {
      "epoch": 2.61,
      "grad_norm": 1.7449744939804077,
      "learning_rate": 6.495098039215687e-05,
      "loss": 0.3967,
      "step": 710
    },
    {
      "epoch": 2.65,
      "grad_norm": 1.1091108322143555,
      "learning_rate": 5.882352941176471e-05,
      "loss": 0.3403,
      "step": 720
    },
    {
      "epoch": 2.68,
      "grad_norm": 1.650840163230896,
      "learning_rate": 5.2696078431372553e-05,
      "loss": 0.3709,
      "step": 730
    },
    {
      "epoch": 2.72,
      "grad_norm": 1.3335641622543335,
      "learning_rate": 4.656862745098039e-05,
      "loss": 0.3852,
      "step": 740
    },
    {
      "epoch": 2.76,
      "grad_norm": 1.7249501943588257,
      "learning_rate": 4.044117647058824e-05,
      "loss": 0.4158,
      "step": 750
    },
    {
      "epoch": 2.76,
      "eval_accuracy": 0.7601102941176471,
      "eval_f1_macro": 0.7210790741531978,
      "eval_f1_micro": 0.7601102941176471,
      "eval_loss": 0.5384809374809265,
      "eval_runtime": 0.9393,
      "eval_samples_per_second": 1158.291,
      "eval_steps_per_second": 36.197,
      "step": 750
    },
    {
      "epoch": 2.79,
      "grad_norm": 1.5633273124694824,
      "learning_rate": 3.4313725490196084e-05,
      "loss": 0.3583,
      "step": 760
    },
    {
      "epoch": 2.83,
      "grad_norm": 1.3377939462661743,
      "learning_rate": 2.8186274509803924e-05,
      "loss": 0.3841,
      "step": 770
    },
    {
      "epoch": 2.87,
      "grad_norm": 1.5654672384262085,
      "learning_rate": 2.2058823529411766e-05,
      "loss": 0.3559,
      "step": 780
    },
    {
      "epoch": 2.9,
      "grad_norm": 1.3034334182739258,
      "learning_rate": 1.593137254901961e-05,
      "loss": 0.3877,
      "step": 790
    },
    {
      "epoch": 2.94,
      "grad_norm": 1.5207332372665405,
      "learning_rate": 9.803921568627451e-06,
      "loss": 0.4003,
      "step": 800
    },
    {
      "epoch": 2.94,
      "eval_accuracy": 0.7545955882352942,
      "eval_f1_macro": 0.7058253459511188,
      "eval_f1_micro": 0.7545955882352942,
      "eval_loss": 0.5349971055984497,
      "eval_runtime": 0.9377,
      "eval_samples_per_second": 1160.255,
      "eval_steps_per_second": 36.258,
      "step": 800
    },
    {
      "epoch": 2.98,
      "grad_norm": 2.046905279159546,
      "learning_rate": 3.6764705882352942e-06,
      "loss": 0.3939,
      "step": 810
    },
    {
      "epoch": 3.0,
      "step": 816,
      "total_flos": 888799123472384.0,
      "train_loss": 0.44335933526357013,
      "train_runtime": 92.5,
      "train_samples_per_second": 282.162,
      "train_steps_per_second": 8.822
    }
  ],
  "logging_steps": 10,
  "max_steps": 816,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 50,
  "total_flos": 888799123472384.0,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}