{
  "best_metric": 1.6402901411056519,
  "best_model_checkpoint": "../../experiments_checkpoints/MAdAiLab/Qwen/Qwen1.5_1.8B_scotus/checkpoint-300",
  "epoch": 3.0,
  "eval_steps": 50,
  "global_step": 471,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.06,
      "grad_norm": 206.6490020751953,
      "learning_rate": 4.893842887473461e-06,
      "loss": 5.8906,
      "step": 10
    },
    {
      "epoch": 0.13,
      "grad_norm": 142.15260314941406,
      "learning_rate": 4.787685774946922e-06,
      "loss": 4.4914,
      "step": 20
    },
    {
      "epoch": 0.19,
      "grad_norm": 47.5128173828125,
      "learning_rate": 4.6815286624203824e-06,
      "loss": 3.1742,
      "step": 30
    },
    {
      "epoch": 0.25,
      "grad_norm": 90.23841094970703,
      "learning_rate": 4.575371549893844e-06,
      "loss": 2.3301,
      "step": 40
    },
    {
      "epoch": 0.32,
      "grad_norm": 106.74323272705078,
      "learning_rate": 4.469214437367304e-06,
      "loss": 2.1746,
      "step": 50
    },
    {
      "epoch": 0.32,
      "eval_accuracy": 0.2692857142857143,
      "eval_f1_macro": 0.11482650807382964,
      "eval_f1_micro": 0.2692857142857143,
      "eval_loss": 2.2791965007781982,
      "eval_runtime": 3.6276,
      "eval_samples_per_second": 385.929,
      "eval_steps_per_second": 12.129,
      "step": 50
    },
    {
      "epoch": 0.38,
      "grad_norm": 87.4228515625,
      "learning_rate": 4.3630573248407645e-06,
      "loss": 1.9594,
      "step": 60
    },
    {
      "epoch": 0.45,
      "grad_norm": 61.52133560180664,
      "learning_rate": 4.256900212314226e-06,
      "loss": 1.8109,
      "step": 70
    },
    {
      "epoch": 0.51,
      "grad_norm": 55.21061706542969,
      "learning_rate": 4.150743099787686e-06,
      "loss": 1.7418,
      "step": 80
    },
    {
      "epoch": 0.57,
      "grad_norm": 71.5464096069336,
      "learning_rate": 4.044585987261147e-06,
      "loss": 1.6996,
      "step": 90
    },
    {
      "epoch": 0.64,
      "grad_norm": 50.51081848144531,
      "learning_rate": 3.938428874734608e-06,
      "loss": 1.5758,
      "step": 100
    },
    {
      "epoch": 0.64,
      "eval_accuracy": 0.41,
      "eval_f1_macro": 0.22906680413417876,
      "eval_f1_micro": 0.41,
      "eval_loss": 1.930892825126648,
      "eval_runtime": 3.6465,
      "eval_samples_per_second": 383.927,
      "eval_steps_per_second": 12.066,
      "step": 100
    },
    {
      "epoch": 0.7,
      "grad_norm": 108.21426391601562,
      "learning_rate": 3.832271762208068e-06,
      "loss": 1.6871,
      "step": 110
    },
    {
      "epoch": 0.76,
      "grad_norm": 62.29310607910156,
      "learning_rate": 3.7261146496815285e-06,
      "loss": 1.6055,
      "step": 120
    },
    {
      "epoch": 0.83,
      "grad_norm": 79.33343505859375,
      "learning_rate": 3.6199575371549893e-06,
      "loss": 1.6043,
      "step": 130
    },
    {
      "epoch": 0.89,
      "grad_norm": 77.69229888916016,
      "learning_rate": 3.51380042462845e-06,
      "loss": 1.5793,
      "step": 140
    },
    {
      "epoch": 0.96,
      "grad_norm": 67.4636001586914,
      "learning_rate": 3.407643312101911e-06,
      "loss": 1.7312,
      "step": 150
    },
    {
      "epoch": 0.96,
      "eval_accuracy": 0.4714285714285714,
      "eval_f1_macro": 0.2546511715546434,
      "eval_f1_micro": 0.4714285714285714,
      "eval_loss": 1.7475446462631226,
      "eval_runtime": 3.6415,
      "eval_samples_per_second": 384.454,
      "eval_steps_per_second": 12.083,
      "step": 150
    },
    {
      "epoch": 1.02,
      "grad_norm": 72.03321838378906,
      "learning_rate": 3.3014861995753718e-06,
      "loss": 1.5695,
      "step": 160
    },
    {
      "epoch": 1.08,
      "grad_norm": 59.265869140625,
      "learning_rate": 3.195329087048832e-06,
      "loss": 1.2879,
      "step": 170
    },
    {
      "epoch": 1.15,
      "grad_norm": 78.44482421875,
      "learning_rate": 3.089171974522293e-06,
      "loss": 1.3846,
      "step": 180
    },
    {
      "epoch": 1.21,
      "grad_norm": 46.30598068237305,
      "learning_rate": 2.983014861995754e-06,
      "loss": 1.365,
      "step": 190
    },
    {
      "epoch": 1.27,
      "grad_norm": 62.2653923034668,
      "learning_rate": 2.8768577494692146e-06,
      "loss": 1.2859,
      "step": 200
    },
    {
      "epoch": 1.27,
      "eval_accuracy": 0.4685714285714286,
      "eval_f1_macro": 0.30187587586686543,
      "eval_f1_micro": 0.4685714285714286,
      "eval_loss": 1.793928623199463,
      "eval_runtime": 3.6406,
      "eval_samples_per_second": 384.55,
      "eval_steps_per_second": 12.086,
      "step": 200
    },
    {
      "epoch": 1.34,
      "grad_norm": 38.229000091552734,
      "learning_rate": 2.7707006369426754e-06,
      "loss": 1.3562,
      "step": 210
    },
    {
      "epoch": 1.4,
      "grad_norm": 64.51058197021484,
      "learning_rate": 2.6645435244161363e-06,
      "loss": 1.2486,
      "step": 220
    },
    {
      "epoch": 1.46,
      "grad_norm": 45.04010009765625,
      "learning_rate": 2.5583864118895966e-06,
      "loss": 1.3775,
      "step": 230
    },
    {
      "epoch": 1.53,
      "grad_norm": 36.0023307800293,
      "learning_rate": 2.4522292993630575e-06,
      "loss": 1.2434,
      "step": 240
    },
    {
      "epoch": 1.59,
      "grad_norm": 42.964847564697266,
      "learning_rate": 2.3460721868365183e-06,
      "loss": 1.1971,
      "step": 250
    },
    {
      "epoch": 1.59,
      "eval_accuracy": 0.4957142857142857,
      "eval_f1_macro": 0.33073042661457275,
      "eval_f1_micro": 0.4957142857142857,
      "eval_loss": 1.7145981788635254,
      "eval_runtime": 3.8826,
      "eval_samples_per_second": 360.584,
      "eval_steps_per_second": 11.333,
      "step": 250
    },
    {
      "epoch": 1.66,
      "grad_norm": 43.22111511230469,
      "learning_rate": 2.239915074309979e-06,
      "loss": 1.2471,
      "step": 260
    },
    {
      "epoch": 1.72,
      "grad_norm": 49.360374450683594,
      "learning_rate": 2.13375796178344e-06,
      "loss": 1.2869,
      "step": 270
    },
    {
      "epoch": 1.78,
      "grad_norm": 54.94723129272461,
      "learning_rate": 2.0276008492569003e-06,
      "loss": 1.1953,
      "step": 280
    },
    {
      "epoch": 1.85,
      "grad_norm": 58.562225341796875,
      "learning_rate": 1.921443736730361e-06,
      "loss": 1.1973,
      "step": 290
    },
    {
      "epoch": 1.91,
      "grad_norm": 47.98759460449219,
      "learning_rate": 1.8152866242038217e-06,
      "loss": 1.1271,
      "step": 300
    },
    {
      "epoch": 1.91,
      "eval_accuracy": 0.515,
      "eval_f1_macro": 0.35688242427831446,
      "eval_f1_micro": 0.515,
      "eval_loss": 1.6402901411056519,
      "eval_runtime": 3.6453,
      "eval_samples_per_second": 384.061,
      "eval_steps_per_second": 12.071,
      "step": 300
    },
    {
      "epoch": 1.97,
      "grad_norm": 49.362281799316406,
      "learning_rate": 1.7091295116772823e-06,
      "loss": 1.2777,
      "step": 310
    },
    {
      "epoch": 2.04,
      "grad_norm": 39.05293273925781,
      "learning_rate": 1.6029723991507432e-06,
      "loss": 0.9527,
      "step": 320
    },
    {
      "epoch": 2.1,
      "grad_norm": 44.27582550048828,
      "learning_rate": 1.496815286624204e-06,
      "loss": 0.8315,
      "step": 330
    },
    {
      "epoch": 2.17,
      "grad_norm": 37.02938461303711,
      "learning_rate": 1.3906581740976646e-06,
      "loss": 0.8729,
      "step": 340
    },
    {
      "epoch": 2.23,
      "grad_norm": 45.579402923583984,
      "learning_rate": 1.2845010615711254e-06,
      "loss": 0.8436,
      "step": 350
    },
    {
      "epoch": 2.23,
      "eval_accuracy": 0.51,
      "eval_f1_macro": 0.3717426255688204,
      "eval_f1_micro": 0.51,
      "eval_loss": 1.7994866371154785,
      "eval_runtime": 3.644,
      "eval_samples_per_second": 384.193,
      "eval_steps_per_second": 12.075,
      "step": 350
    },
    {
      "epoch": 2.29,
      "grad_norm": 28.59382438659668,
      "learning_rate": 1.178343949044586e-06,
      "loss": 0.7411,
      "step": 360
    },
    {
      "epoch": 2.36,
      "grad_norm": 57.68552780151367,
      "learning_rate": 1.0721868365180468e-06,
      "loss": 0.8667,
      "step": 370
    },
    {
      "epoch": 2.42,
      "grad_norm": 60.32491683959961,
      "learning_rate": 9.660297239915076e-07,
      "loss": 0.7807,
      "step": 380
    },
    {
      "epoch": 2.48,
      "grad_norm": 52.709407806396484,
      "learning_rate": 8.598726114649681e-07,
      "loss": 0.7298,
      "step": 390
    },
    {
      "epoch": 2.55,
      "grad_norm": 61.55146408081055,
      "learning_rate": 7.537154989384289e-07,
      "loss": 0.7723,
      "step": 400
    },
    {
      "epoch": 2.55,
      "eval_accuracy": 0.5028571428571429,
      "eval_f1_macro": 0.37231960236652384,
      "eval_f1_micro": 0.5028571428571429,
      "eval_loss": 1.8401116132736206,
      "eval_runtime": 3.6434,
      "eval_samples_per_second": 384.253,
      "eval_steps_per_second": 12.077,
      "step": 400
    },
    {
      "epoch": 2.61,
      "grad_norm": 46.969879150390625,
      "learning_rate": 6.475583864118897e-07,
      "loss": 0.8337,
      "step": 410
    },
    {
      "epoch": 2.68,
      "grad_norm": 52.0188102722168,
      "learning_rate": 5.414012738853504e-07,
      "loss": 0.8848,
      "step": 420
    },
    {
      "epoch": 2.74,
      "grad_norm": 53.10190963745117,
      "learning_rate": 4.35244161358811e-07,
      "loss": 0.718,
      "step": 430
    },
    {
      "epoch": 2.8,
      "grad_norm": 58.61328125,
      "learning_rate": 3.290870488322718e-07,
      "loss": 0.7417,
      "step": 440
    },
    {
      "epoch": 2.87,
      "grad_norm": 63.344459533691406,
      "learning_rate": 2.229299363057325e-07,
      "loss": 0.7211,
      "step": 450
    },
    {
      "epoch": 2.87,
      "eval_accuracy": 0.5178571428571429,
      "eval_f1_macro": 0.375345717505455,
      "eval_f1_micro": 0.5178571428571429,
      "eval_loss": 1.8231920003890991,
      "eval_runtime": 3.6452,
      "eval_samples_per_second": 384.067,
      "eval_steps_per_second": 12.071,
      "step": 450
    },
    {
      "epoch": 2.93,
      "grad_norm": 42.954627990722656,
      "learning_rate": 1.1677282377919321e-07,
      "loss": 0.7757,
      "step": 460
    },
    {
      "epoch": 2.99,
      "grad_norm": 49.02348327636719,
      "learning_rate": 1.0615711252653928e-08,
      "loss": 0.8159,
      "step": 470
    },
    {
      "epoch": 3.0,
      "step": 471,
      "total_flos": 1.405849146753024e+16,
      "train_loss": 1.457581732683121,
      "train_runtime": 520.0867,
      "train_samples_per_second": 28.841,
      "train_steps_per_second": 0.906
    }
  ],
  "logging_steps": 10,
  "max_steps": 471,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 50,
  "total_flos": 1.405849146753024e+16,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}