checkpoint-500/trainer_state.json

{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.08333333333333333,
  "eval_steps": 500,
  "global_step": 500,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.08333333333333333,
      "grad_norm": 10.64107608795166,
      "learning_rate": 2.4999999999999998e-06,
      "loss": 4.1627,
      "step": 500
    },
    {
      "epoch": 0.08333333333333333,
      "eval_accuracy_background": 0.014990974836900402,
      "eval_accuracy_double_plant": 0.09625850817128659,
      "eval_accuracy_drydown": 0.11748866381009407,
      "eval_accuracy_endrow": 0.019624906998737878,
      "eval_accuracy_nutrient_deficiency": 0.39043216558572647,
      "eval_accuracy_planter_skip": 0.03244089779472675,
      "eval_accuracy_water": 0.4902746667077927,
      "eval_accuracy_waterway": 0.37818186421979577,
      "eval_accuracy_weed_cluster": 0.5157577782392276,
      "eval_iou_background": 0.01485215844086368,
      "eval_iou_double_plant": 0.025830652956995166,
      "eval_iou_drydown": 0.09098524126821965,
      "eval_iou_endrow": 0.011190399377301194,
      "eval_iou_nutrient_deficiency": 0.16172838363740288,
      "eval_iou_planter_skip": 0.007145844854739052,
      "eval_iou_water": 0.052629358314533047,
      "eval_iou_waterway": 0.008009602169192399,
      "eval_iou_weed_cluster": 0.055193229922437775,
      "eval_loss": 3.9555599689483643,
      "eval_mean_accuracy": 0.22838338070714315,
      "eval_mean_iou": 0.047507207882409426,
      "eval_overall_accuracy": 0.09656532361323403,
      "eval_runtime": 348.036,
      "eval_samples_per_second": 8.166,
      "eval_steps_per_second": 2.724,
      "step": 500
    }
  ],
  "logging_steps": 500,
  "max_steps": 120000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 20,
  "save_steps": 500,
  "total_flos": 7.98417862262784e+17,
  "train_batch_size": 3,
  "trial_name": null,
  "trial_params": null
}