{
  "best_metric": 0.6476547574288157,
  "best_model_checkpoint": "./nlu_finetuned_models/cola/roberta-base_lr1e-05/checkpoint-1924",
  "epoch": 10.0,
  "eval_steps": 500,
  "global_step": 4810,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "eval_loss": 0.39855843782424927,
      "eval_matthews_correlation": 0.551445188783165,
      "eval_runtime": 0.4311,
      "eval_samples_per_second": 1985.797,
      "eval_steps_per_second": 125.272,
      "step": 481
    },
    {
      "epoch": 1.0395010395010396,
      "grad_norm": 23.592742919921875,
      "learning_rate": 9.533289095332891e-06,
      "loss": 0.537,
      "step": 500
    },
    {
      "epoch": 2.0,
      "eval_loss": 0.3867790997028351,
      "eval_matthews_correlation": 0.5922946478625156,
      "eval_runtime": 0.4252,
      "eval_samples_per_second": 2013.273,
      "eval_steps_per_second": 127.006,
      "step": 962
    },
    {
      "epoch": 2.079002079002079,
      "grad_norm": 24.047794342041016,
      "learning_rate": 8.427339084273391e-06,
      "loss": 0.3663,
      "step": 1000
    },
    {
      "epoch": 3.0,
      "eval_loss": 0.44944295287132263,
      "eval_matthews_correlation": 0.5697569536882547,
      "eval_runtime": 0.432,
      "eval_samples_per_second": 1981.347,
      "eval_steps_per_second": 124.992,
      "step": 1443
    },
    {
      "epoch": 3.1185031185031185,
      "grad_norm": 19.57173728942871,
      "learning_rate": 7.3213890732138915e-06,
      "loss": 0.2692,
      "step": 1500
    },
    {
      "epoch": 4.0,
      "eval_loss": 0.5625013709068298,
      "eval_matthews_correlation": 0.6476547574288157,
      "eval_runtime": 0.4183,
      "eval_samples_per_second": 2046.415,
      "eval_steps_per_second": 129.096,
      "step": 1924
    },
    {
      "epoch": 4.158004158004158,
      "grad_norm": 18.871004104614258,
      "learning_rate": 6.2154390621543915e-06,
      "loss": 0.2002,
      "step": 2000
    },
    {
      "epoch": 5.0,
      "eval_loss": 0.6681837439537048,
      "eval_matthews_correlation": 0.5953927103011889,
      "eval_runtime": 0.4264,
      "eval_samples_per_second": 2007.326,
      "eval_steps_per_second": 126.63,
      "step": 2405
    },
    {
      "epoch": 5.197505197505198,
      "grad_norm": 12.891629219055176,
      "learning_rate": 5.1094890510948916e-06,
      "loss": 0.1761,
      "step": 2500
    },
    {
      "epoch": 6.0,
      "eval_loss": 0.7049034237861633,
      "eval_matthews_correlation": 0.6376896027079079,
      "eval_runtime": 0.4161,
      "eval_samples_per_second": 2057.2,
      "eval_steps_per_second": 129.777,
      "step": 2886
    },
    {
      "epoch": 6.237006237006237,
      "grad_norm": 0.21968619525432587,
      "learning_rate": 4.003539040035391e-06,
      "loss": 0.1486,
      "step": 3000
    },
    {
      "epoch": 7.0,
      "eval_loss": 0.8099717497825623,
      "eval_matthews_correlation": 0.6372565507596143,
      "eval_runtime": 0.4311,
      "eval_samples_per_second": 1985.553,
      "eval_steps_per_second": 125.257,
      "step": 3367
    },
    {
      "epoch": 7.276507276507276,
      "grad_norm": 0.38824957609176636,
      "learning_rate": 2.897589028975891e-06,
      "loss": 0.1261,
      "step": 3500
    },
    {
      "epoch": 8.0,
      "eval_loss": 0.8209366202354431,
      "eval_matthews_correlation": 0.6409103987573772,
      "eval_runtime": 0.4313,
      "eval_samples_per_second": 1984.868,
      "eval_steps_per_second": 125.214,
      "step": 3848
    },
    {
      "epoch": 8.316008316008316,
      "grad_norm": 0.6946465969085693,
      "learning_rate": 1.7916390179163902e-06,
      "loss": 0.0963,
      "step": 4000
    },
    {
      "epoch": 9.0,
      "eval_loss": 0.866554319858551,
      "eval_matthews_correlation": 0.6465132548864322,
      "eval_runtime": 0.4204,
      "eval_samples_per_second": 2036.18,
      "eval_steps_per_second": 128.451,
      "step": 4329
    },
    {
      "epoch": 9.355509355509355,
      "grad_norm": 0.8278540968894958,
      "learning_rate": 6.856890068568902e-07,
      "loss": 0.0918,
      "step": 4500
    },
    {
      "epoch": 10.0,
      "eval_loss": 0.9010610580444336,
      "eval_matthews_correlation": 0.6344646755477841,
      "eval_runtime": 0.4206,
      "eval_samples_per_second": 2035.283,
      "eval_steps_per_second": 128.394,
      "step": 4810
    },
    {
      "epoch": 10.0,
      "step": 4810,
      "total_flos": 806520176327100.0,
      "train_loss": 0.2141543533103134,
      "train_runtime": 285.3143,
      "train_samples_per_second": 269.703,
      "train_steps_per_second": 16.859
    }
  ],
  "logging_steps": 500,
  "max_steps": 4810,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 10,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 806520176327100.0,
  "train_batch_size": 16,
  "trial_name": null,
  "trial_params": null
}