{
  "best_metric": 0.7645955681800842,
  "best_model_checkpoint": "email-classifier/checkpoint-27",
  "epoch": 3.0,
  "eval_steps": 500,
  "global_step": 27,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.1111111111111111,
      "grad_norm": 12.661477088928223,
      "learning_rate": 1.6666666666666667e-05,
      "loss": 1.0907,
      "step": 1
    },
    {
      "epoch": 0.2222222222222222,
      "grad_norm": 9.262643814086914,
      "learning_rate": 3.3333333333333335e-05,
      "loss": 1.1452,
      "step": 2
    },
    {
      "epoch": 0.3333333333333333,
      "grad_norm": 4.253861427307129,
      "learning_rate": 5e-05,
      "loss": 1.0398,
      "step": 3
    },
    {
      "epoch": 0.4444444444444444,
      "grad_norm": 5.083256721496582,
      "learning_rate": 4.791666666666667e-05,
      "loss": 1.0236,
      "step": 4
    },
    {
      "epoch": 0.5555555555555556,
      "grad_norm": 6.380894184112549,
      "learning_rate": 4.5833333333333334e-05,
      "loss": 0.8063,
      "step": 5
    },
    {
      "epoch": 0.6666666666666666,
      "grad_norm": 7.080038070678711,
      "learning_rate": 4.375e-05,
      "loss": 0.8933,
      "step": 6
    },
    {
      "epoch": 0.7777777777777778,
      "grad_norm": 5.052453994750977,
      "learning_rate": 4.166666666666667e-05,
      "loss": 0.5092,
      "step": 7
    },
    {
      "epoch": 0.8888888888888888,
      "grad_norm": 4.828819751739502,
      "learning_rate": 3.958333333333333e-05,
      "loss": 0.9788,
      "step": 8
    },
    {
      "epoch": 1.0,
      "grad_norm": 5.715775012969971,
      "learning_rate": 3.7500000000000003e-05,
      "loss": 0.7505,
      "step": 9
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.7058823529411765,
      "eval_f1_macro": 0.27586206896551724,
      "eval_f1_micro": 0.7058823529411765,
      "eval_f1_weighted": 0.5841784989858012,
      "eval_loss": 0.8175652027130127,
      "eval_precision_macro": 0.23529411764705885,
      "eval_precision_micro": 0.7058823529411765,
      "eval_precision_weighted": 0.49826989619377166,
      "eval_recall_macro": 0.3333333333333333,
      "eval_recall_micro": 0.7058823529411765,
      "eval_recall_weighted": 0.7058823529411765,
      "eval_runtime": 4.292,
      "eval_samples_per_second": 3.961,
      "eval_steps_per_second": 0.466,
      "step": 9
    },
    {
      "epoch": 1.1111111111111112,
      "grad_norm": 5.281602382659912,
      "learning_rate": 3.541666666666667e-05,
      "loss": 1.0502,
      "step": 10
    },
    {
      "epoch": 1.2222222222222223,
      "grad_norm": 4.289942741394043,
      "learning_rate": 3.3333333333333335e-05,
      "loss": 0.7533,
      "step": 11
    },
    {
      "epoch": 1.3333333333333333,
      "grad_norm": 4.913335800170898,
      "learning_rate": 3.125e-05,
      "loss": 1.0854,
      "step": 12
    },
    {
      "epoch": 1.4444444444444444,
      "grad_norm": 4.09973669052124,
      "learning_rate": 2.916666666666667e-05,
      "loss": 0.4934,
      "step": 13
    },
    {
      "epoch": 1.5555555555555556,
      "grad_norm": 6.382025718688965,
      "learning_rate": 2.7083333333333332e-05,
      "loss": 0.8873,
      "step": 14
    },
    {
      "epoch": 1.6666666666666665,
      "grad_norm": 2.496077299118042,
      "learning_rate": 2.5e-05,
      "loss": 0.7701,
      "step": 15
    },
    {
      "epoch": 1.7777777777777777,
      "grad_norm": 3.6502137184143066,
      "learning_rate": 2.2916666666666667e-05,
      "loss": 0.653,
      "step": 16
    },
    {
      "epoch": 1.8888888888888888,
      "grad_norm": 3.0348548889160156,
      "learning_rate": 2.0833333333333336e-05,
      "loss": 0.8641,
      "step": 17
    },
    {
      "epoch": 2.0,
      "grad_norm": 5.570982456207275,
      "learning_rate": 1.8750000000000002e-05,
      "loss": 0.835,
      "step": 18
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.7058823529411765,
      "eval_f1_macro": 0.27586206896551724,
      "eval_f1_micro": 0.7058823529411765,
      "eval_f1_weighted": 0.5841784989858012,
      "eval_loss": 0.7890189290046692,
      "eval_precision_macro": 0.23529411764705885,
      "eval_precision_micro": 0.7058823529411765,
      "eval_precision_weighted": 0.49826989619377166,
      "eval_recall_macro": 0.3333333333333333,
      "eval_recall_micro": 0.7058823529411765,
      "eval_recall_weighted": 0.7058823529411765,
      "eval_runtime": 4.8269,
      "eval_samples_per_second": 3.522,
      "eval_steps_per_second": 0.414,
      "step": 18
    },
    {
      "epoch": 2.111111111111111,
      "grad_norm": 2.422776699066162,
      "learning_rate": 1.6666666666666667e-05,
      "loss": 0.875,
      "step": 19
    },
    {
      "epoch": 2.2222222222222223,
      "grad_norm": 6.083028316497803,
      "learning_rate": 1.4583333333333335e-05,
      "loss": 0.5999,
      "step": 20
    },
    {
      "epoch": 2.3333333333333335,
      "grad_norm": 4.942434310913086,
      "learning_rate": 1.25e-05,
      "loss": 0.9778,
      "step": 21
    },
    {
      "epoch": 2.4444444444444446,
      "grad_norm": 5.678409099578857,
      "learning_rate": 1.0416666666666668e-05,
      "loss": 0.5364,
      "step": 22
    },
    {
      "epoch": 2.5555555555555554,
      "grad_norm": 5.16926383972168,
      "learning_rate": 8.333333333333334e-06,
      "loss": 0.7531,
      "step": 23
    },
    {
      "epoch": 2.6666666666666665,
      "grad_norm": 4.587949275970459,
      "learning_rate": 6.25e-06,
      "loss": 0.7907,
      "step": 24
    },
    {
      "epoch": 2.7777777777777777,
      "grad_norm": 5.132978916168213,
      "learning_rate": 4.166666666666667e-06,
      "loss": 0.7656,
      "step": 25
    },
    {
      "epoch": 2.888888888888889,
      "grad_norm": 2.9670441150665283,
      "learning_rate": 2.0833333333333334e-06,
      "loss": 0.8906,
      "step": 26
    },
    {
      "epoch": 3.0,
      "grad_norm": 10.066261291503906,
      "learning_rate": 0.0,
      "loss": 0.3755,
      "step": 27
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.7058823529411765,
      "eval_f1_macro": 0.27586206896551724,
      "eval_f1_micro": 0.7058823529411765,
      "eval_f1_weighted": 0.5841784989858012,
      "eval_loss": 0.7645955681800842,
      "eval_precision_macro": 0.23529411764705885,
      "eval_precision_micro": 0.7058823529411765,
      "eval_precision_weighted": 0.49826989619377166,
      "eval_recall_macro": 0.3333333333333333,
      "eval_recall_micro": 0.7058823529411765,
      "eval_recall_weighted": 0.7058823529411765,
      "eval_runtime": 4.2687,
      "eval_samples_per_second": 3.982,
      "eval_steps_per_second": 0.469,
      "step": 27
    }
  ],
  "logging_steps": 1,
  "max_steps": 27,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 3,
  "save_steps": 500,
  "stateful_callbacks": {
    "EarlyStoppingCallback": {
      "args": {
        "early_stopping_patience": 5,
        "early_stopping_threshold": 0.01
      },
      "attributes": {
        "early_stopping_patience_counter": 0
      }
    },
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 13418784304128.0,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}