trainer_state.json

{
  "best_metric": 0.9409828782081604,
  "best_model_checkpoint": "efficientnet-b5-Brain_Tumors_Image_Classification/checkpoint-540",
  "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.3872,
      "step": 1
    },
    {
      "epoch": 1.0,
      "eval_Macro F1": 0.6549508293588824,
      "eval_Macro Precision": 0.8329646017699115,
      "eval_Macro Recall": 0.6801586368977673,
      "eval_Micro F1": 0.6852791878172588,
      "eval_Micro Precision": 0.6852791878172588,
      "eval_Micro Recall": 0.6852791878172588,
      "eval_Weighted F1": 0.64850645556817,
      "eval_Weighted Precision": 0.8176744081577646,
      "eval_Weighted Recall": 0.6852791878172588,
      "eval_accuracy": 0.6852791878172588,
      "eval_loss": 1.0601236820220947,
      "eval_runtime": 157.183,
      "eval_samples_per_second": 2.507,
      "eval_steps_per_second": 0.318,
      "step": 180
    },
    {
      "epoch": 2.0,
      "eval_Macro F1": 0.754761660411291,
      "eval_Macro Precision": 0.847140909975985,
      "eval_Macro Recall": 0.7818963684181075,
      "eval_Micro F1": 0.7842639593908629,
      "eval_Micro Precision": 0.7842639593908629,
      "eval_Micro Recall": 0.7842639593908629,
      "eval_Weighted F1": 0.7482587054309443,
      "eval_Weighted Precision": 0.8353897743757452,
      "eval_Weighted Recall": 0.7842639593908629,
      "eval_accuracy": 0.7842639593908629,
      "eval_loss": 0.9532977342605591,
      "eval_runtime": 156.7569,
      "eval_samples_per_second": 2.513,
      "eval_steps_per_second": 0.319,
      "step": 360
    },
    {
      "epoch": 2.78,
      "learning_rate": 1.4814814814814815e-05,
      "loss": 0.8186,
      "step": 500
    },
    {
      "epoch": 3.0,
      "eval_Macro F1": 0.7802118787775556,
      "eval_Macro Precision": 0.8682432432432432,
      "eval_Macro Recall": 0.7977467685076381,
      "eval_Micro F1": 0.8020304568527918,
      "eval_Micro Precision": 0.8020304568527918,
      "eval_Micro Recall": 0.8020304568527918,
      "eval_Weighted F1": 0.7736247112111547,
      "eval_Weighted Precision": 0.8535464398408561,
      "eval_Weighted Recall": 0.8020304568527918,
      "eval_accuracy": 0.8020304568527918,
      "eval_loss": 0.9409828782081604,
      "eval_runtime": 156.1926,
      "eval_samples_per_second": 2.523,
      "eval_steps_per_second": 0.32,
      "step": 540
    },
    {
      "epoch": 3.0,
      "step": 540,
      "total_flos": 9.135719978482944e+17,
      "train_loss": 0.8154728790124257,
      "train_runtime": 21763.4535,
      "train_samples_per_second": 0.396,
      "train_steps_per_second": 0.025
    }
  ],
  "max_steps": 540,
  "num_train_epochs": 3,
  "total_flos": 9.135719978482944e+17,
  "trial_name": null,
  "trial_params": null
}