trainer_state.json

{
  "best_metric": 1.0963644981384277,
  "best_model_checkpoint": "regnet-y-064-Brain_Tumors_Image_Classification/checkpoint-360",
  "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.288,
      "step": 1
    },
    {
      "epoch": 1.0,
      "eval_Macro F1": 0.5868468449113611,
      "eval_Macro Precision": 0.8284871470712178,
      "eval_Macro Recall": 0.617642689273124,
      "eval_Micro F1": 0.6548223350253807,
      "eval_Micro Precision": 0.6548223350253807,
      "eval_Micro Recall": 0.6548223350253807,
      "eval_Weighted F1": 0.5991176009188128,
      "eval_Weighted Precision": 0.8045909887246755,
      "eval_Weighted Recall": 0.6548223350253807,
      "eval_accuracy": 0.6548223350253807,
      "eval_loss": 1.3795779943466187,
      "eval_runtime": 78.5841,
      "eval_samples_per_second": 5.014,
      "eval_steps_per_second": 0.636,
      "step": 180
    },
    {
      "epoch": 2.0,
      "eval_Macro F1": 0.7754730215471972,
      "eval_Macro Precision": 0.8727018662232077,
      "eval_Macro Recall": 0.7871975546975547,
      "eval_Micro F1": 0.7944162436548223,
      "eval_Micro Precision": 0.7944162436548223,
      "eval_Micro Recall": 0.7944162436548223,
      "eval_Weighted F1": 0.7686714219002021,
      "eval_Weighted Precision": 0.8555047581797922,
      "eval_Weighted Recall": 0.7944162436548223,
      "eval_accuracy": 0.7944162436548223,
      "eval_loss": 1.0963644981384277,
      "eval_runtime": 45.8042,
      "eval_samples_per_second": 8.602,
      "eval_steps_per_second": 1.092,
      "step": 360
    },
    {
      "epoch": 2.78,
      "learning_rate": 1.4814814814814815e-05,
      "loss": 0.1498,
      "step": 500
    },
    {
      "epoch": 3.0,
      "eval_Macro F1": 0.7838766405792781,
      "eval_Macro Precision": 0.8735744504736503,
      "eval_Macro Recall": 0.7978378378378378,
      "eval_Micro F1": 0.8045685279187818,
      "eval_Micro Precision": 0.8045685279187818,
      "eval_Micro Recall": 0.8045685279187818,
      "eval_Weighted F1": 0.7775760915891737,
      "eval_Weighted Precision": 0.8574339594982402,
      "eval_Weighted Recall": 0.8045685279187818,
      "eval_accuracy": 0.8045685279187818,
      "eval_loss": 1.1560683250427246,
      "eval_runtime": 88.1188,
      "eval_samples_per_second": 4.471,
      "eval_steps_per_second": 0.567,
      "step": 540
    },
    {
      "epoch": 3.0,
      "step": 540,
      "total_flos": 2.277783359373312e+17,
      "train_loss": 0.1418286879857381,
      "train_runtime": 5110.9896,
      "train_samples_per_second": 1.685,
      "train_steps_per_second": 0.106
    }
  ],
  "max_steps": 540,
  "num_train_epochs": 3,
  "total_flos": 2.277783359373312e+17,
  "trial_name": null,
  "trial_params": null
}