checkpoint-1971/trainer_state.json

{
  "best_metric": 0.5755290389060974,
  "best_model_checkpoint": "/content/drive/MyDrive/Models/huBERT_Plain2/checkpoint-1971",
  "epoch": 3.0,
  "global_step": 1971,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "learning_rate": 4.5e-06,
      "loss": 0.6481,
      "step": 657
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.6730182926829268,
      "eval_loss": 0.6048818230628967,
      "eval_runtime": 5.483,
      "eval_samples_per_second": 239.285,
      "eval_steps_per_second": 14.955,
      "step": 657
    },
    {
      "epoch": 2.0,
      "learning_rate": 4.000000000000001e-06,
      "loss": 0.5526,
      "step": 1314
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.6821646341463414,
      "eval_loss": 0.6087087988853455,
      "eval_runtime": 4.5344,
      "eval_samples_per_second": 289.344,
      "eval_steps_per_second": 18.084,
      "step": 1314
    },
    {
      "epoch": 3.0,
      "learning_rate": 3.5e-06,
      "loss": 0.4988,
      "step": 1971
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.7103658536585366,
      "eval_loss": 0.5755290389060974,
      "eval_runtime": 4.5357,
      "eval_samples_per_second": 289.264,
      "eval_steps_per_second": 18.079,
      "step": 1971
    }
  ],
  "max_steps": 6570,
  "num_train_epochs": 10,
  "total_flos": 517950556917120.0,
  "trial_name": null,
  "trial_params": null
}