trainer_state.json

{
  "best_metric": 0.5237217871948411,
  "best_model_checkpoint": "videomae-base-finetuned-ucf101-subset/checkpoint-7",
  "epoch": 1.4166666666666667,
  "global_step": 12,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.58,
      "eval_accuracy": 0.5237217871948411,
      "eval_loss": 1.0619391202926636,
      "eval_runtime": 12138.9686,
      "eval_samples_per_second": 0.358,
      "eval_steps_per_second": 0.045,
      "step": 7
    },
    {
      "epoch": 1.25,
      "learning_rate": 1e-05,
      "loss": 1.1329,
      "step": 10
    },
    {
      "epoch": 1.42,
      "eval_accuracy": 0.49700598802395207,
      "eval_loss": 1.0687167644500732,
      "eval_runtime": 11836.9614,
      "eval_samples_per_second": 0.367,
      "eval_steps_per_second": 0.046,
      "step": 12
    },
    {
      "epoch": 1.42,
      "step": 12,
      "total_flos": 1.1214681318457344e+17,
      "train_loss": 1.1226235230763753,
      "train_runtime": 25193.7599,
      "train_samples_per_second": 0.004,
      "train_steps_per_second": 0.0
    },
    {
      "epoch": 1.42,
      "eval_accuracy": 0.2545671845654772,
      "eval_loss": 1.1779148578643799,
      "eval_runtime": 18429.4117,
      "eval_samples_per_second": 0.318,
      "eval_steps_per_second": 0.04,
      "step": 12
    },
    {
      "epoch": 1.42,
      "eval_accuracy": 0.2545671845654772,
      "eval_loss": 1.1779147386550903,
      "eval_runtime": 18593.8969,
      "eval_samples_per_second": 0.315,
      "eval_steps_per_second": 0.039,
      "step": 12
    }
  ],
  "max_steps": 12,
  "num_train_epochs": 9223372036854775807,
  "total_flos": 1.1214681318457344e+17,
  "trial_name": null,
  "trial_params": null
}