trainer_state.json

{
  "best_metric": 1.527304768562317,
  "best_model_checkpoint": "van-base-Brain_Tumors_Image_Classification/checkpoint-180",
  "epoch": 3.0,
  "global_step": 540,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.01,
      "learning_rate": 0.00019962962962962963,
      "loss": 1.3357,
      "step": 1
    },
    {
      "epoch": 1.0,
      "eval_Macro F1": 0.6695349613078919,
      "eval_Macro Precision": 0.8420018091361375,
      "eval_Macro Recall": 0.7058227015835712,
      "eval_Micro F1": 0.7182741116751268,
      "eval_Micro Precision": 0.7182741116751269,
      "eval_Micro Recall": 0.7182741116751269,
      "eval_Weighted F1": 0.6630777960241693,
      "eval_Weighted Precision": 0.8218941288022279,
      "eval_Weighted Recall": 0.7182741116751269,
      "eval_accuracy": 0.7182741116751269,
      "eval_loss": 1.527304768562317,
      "eval_runtime": 70.7832,
      "eval_samples_per_second": 5.566,
      "eval_steps_per_second": 0.706,
      "step": 180
    },
    {
      "epoch": 2.0,
      "eval_Macro F1": 0.7411201610318727,
      "eval_Macro Precision": 0.8636309679372562,
      "eval_Macro Recall": 0.7763513513513514,
      "eval_Micro F1": 0.7791878172588832,
      "eval_Micro Precision": 0.7791878172588832,
      "eval_Micro Recall": 0.7791878172588832,
      "eval_Weighted F1": 0.7314045916031979,
      "eval_Weighted Precision": 0.8467143905892778,
      "eval_Weighted Recall": 0.7791878172588832,
      "eval_accuracy": 0.7791878172588832,
      "eval_loss": 1.9359180927276611,
      "eval_runtime": 126.0995,
      "eval_samples_per_second": 3.125,
      "eval_steps_per_second": 0.397,
      "step": 360
    },
    {
      "epoch": 2.78,
      "learning_rate": 1.4814814814814815e-05,
      "loss": 0.1229,
      "step": 500
    },
    {
      "epoch": 3.0,
      "eval_Macro F1": 0.7665103714400932,
      "eval_Macro Precision": 0.8675026896180742,
      "eval_Macro Recall": 0.786542303172738,
      "eval_Micro F1": 0.7918781725888325,
      "eval_Micro Precision": 0.7918781725888325,
      "eval_Micro Recall": 0.7918781725888325,
      "eval_Weighted F1": 0.7587838573985126,
      "eval_Weighted Precision": 0.8504992176897681,
      "eval_Weighted Recall": 0.7918781725888325,
      "eval_accuracy": 0.7918781725888325,
      "eval_loss": 1.7846513986587524,
      "eval_runtime": 125.5826,
      "eval_samples_per_second": 3.137,
      "eval_steps_per_second": 0.398,
      "step": 540
    },
    {
      "epoch": 3.0,
      "step": 540,
      "total_flos": 2.0271623668015104e+17,
      "train_loss": 0.11622388989285186,
      "train_runtime": 6860.1206,
      "train_samples_per_second": 1.255,
      "train_steps_per_second": 0.079
    }
  ],
  "max_steps": 540,
  "num_train_epochs": 3,
  "total_flos": 2.0271623668015104e+17,
  "trial_name": null,
  "trial_params": null
}