{
  "best_metric": 0.6206496953964233,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b5_5/checkpoint-40",
  "epoch": 20.0,
  "global_step": 40,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.5,
      "learning_rate": 7.5e-07,
      "loss": 1.1306,
      "step": 1
    },
    {
      "epoch": 1.0,
      "learning_rate": 1.5e-06,
      "loss": 1.1189,
      "step": 2
    },
    {
      "epoch": 1.5,
      "learning_rate": 2.25e-06,
      "loss": 1.1168,
      "step": 3
    },
    {
      "epoch": 2.0,
      "learning_rate": 3e-06,
      "loss": 1.1199,
      "step": 4
    },
    {
      "epoch": 2.5,
      "learning_rate": 3.75e-06,
      "loss": 1.1175,
      "step": 5
    },
    {
      "epoch": 3.0,
      "learning_rate": 4.5e-06,
      "loss": 1.1046,
      "step": 6
    },
    {
      "epoch": 3.5,
      "learning_rate": 5.2500000000000006e-06,
      "loss": 1.0996,
      "step": 7
    },
    {
      "epoch": 4.0,
      "learning_rate": 6e-06,
      "loss": 1.0903,
      "step": 8
    },
    {
      "epoch": 4.5,
      "learning_rate": 6.75e-06,
      "loss": 1.0751,
      "step": 9
    },
    {
      "epoch": 5.0,
      "learning_rate": 7.5e-06,
      "loss": 1.0685,
      "step": 10
    },
    {
      "epoch": 5.0,
      "eval_accuracy_dropoff": 0.12556710016284778,
      "eval_accuracy_undropoff": 0.6194458181630027,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.04967425823428001,
      "eval_iou_undropoff": 0.6069927004338553,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.0222245454788208,
      "eval_mean_accuracy": 0.37250645916292524,
      "eval_mean_iou": 0.21888898622271177,
      "eval_overall_accuracy": 0.5989107131958008,
      "eval_runtime": 2.0623,
      "eval_samples_per_second": 9.698,
      "eval_steps_per_second": 0.97,
      "step": 10
    },
    {
      "epoch": 5.5,
      "learning_rate": 8.25e-06,
      "loss": 1.0536,
      "step": 11
    },
    {
      "epoch": 6.0,
      "learning_rate": 9e-06,
      "loss": 1.0449,
      "step": 12
    },
    {
      "epoch": 6.5,
      "learning_rate": 8.960526315789474e-06,
      "loss": 1.0298,
      "step": 13
    },
    {
      "epoch": 7.0,
      "learning_rate": 8.921052631578947e-06,
      "loss": 1.0194,
      "step": 14
    },
    {
      "epoch": 7.5,
      "learning_rate": 8.881578947368421e-06,
      "loss": 1.0003,
      "step": 15
    },
    {
      "epoch": 8.0,
      "learning_rate": 8.842105263157895e-06,
      "loss": 0.9971,
      "step": 16
    },
    {
      "epoch": 8.5,
      "learning_rate": 8.80263157894737e-06,
      "loss": 0.9839,
      "step": 17
    },
    {
      "epoch": 9.0,
      "learning_rate": 8.763157894736842e-06,
      "loss": 0.9637,
      "step": 18
    },
    {
      "epoch": 9.5,
      "learning_rate": 8.723684210526316e-06,
      "loss": 0.9503,
      "step": 19
    },
    {
      "epoch": 10.0,
      "learning_rate": 8.68421052631579e-06,
      "loss": 0.9481,
      "step": 20
    },
    {
      "epoch": 10.0,
      "eval_accuracy_dropoff": 0.41593614532443407,
      "eval_accuracy_undropoff": 0.8637103137683748,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.26333490163277395,
      "eval_iou_undropoff": 0.8475766394676805,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.8418876528739929,
      "eval_mean_accuracy": 0.6398232295464045,
      "eval_mean_iou": 0.3703038470334848,
      "eval_overall_accuracy": 0.8450922012329102,
      "eval_runtime": 2.1115,
      "eval_samples_per_second": 9.472,
      "eval_steps_per_second": 0.947,
      "step": 20
    },
    {
      "epoch": 10.5,
      "learning_rate": 8.644736842105263e-06,
      "loss": 0.9183,
      "step": 21
    },
    {
      "epoch": 11.0,
      "learning_rate": 8.605263157894737e-06,
      "loss": 0.9305,
      "step": 22
    },
    {
      "epoch": 11.5,
      "learning_rate": 8.56578947368421e-06,
      "loss": 0.9155,
      "step": 23
    },
    {
      "epoch": 12.0,
      "learning_rate": 8.526315789473683e-06,
      "loss": 0.8859,
      "step": 24
    },
    {
      "epoch": 12.5,
      "learning_rate": 8.486842105263159e-06,
      "loss": 0.8853,
      "step": 25
    },
    {
      "epoch": 13.0,
      "learning_rate": 8.447368421052632e-06,
      "loss": 0.8673,
      "step": 26
    },
    {
      "epoch": 13.5,
      "learning_rate": 8.407894736842106e-06,
      "loss": 0.8644,
      "step": 27
    },
    {
      "epoch": 14.0,
      "learning_rate": 8.36842105263158e-06,
      "loss": 0.8401,
      "step": 28
    },
    {
      "epoch": 14.5,
      "learning_rate": 8.328947368421052e-06,
      "loss": 0.836,
      "step": 29
    },
    {
      "epoch": 15.0,
      "learning_rate": 8.289473684210526e-06,
      "loss": 0.8268,
      "step": 30
    },
    {
      "epoch": 15.0,
      "eval_accuracy_dropoff": 0.5023463840913782,
      "eval_accuracy_undropoff": 0.8852725982783686,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.31364905698205614,
      "eval_iou_undropoff": 0.8711414077560768,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.7165183424949646,
      "eval_mean_accuracy": 0.6938094911848733,
      "eval_mean_iou": 0.39493015491271094,
      "eval_overall_accuracy": 0.869350814819336,
      "eval_runtime": 2.2715,
      "eval_samples_per_second": 8.805,
      "eval_steps_per_second": 0.88,
      "step": 30
    },
    {
      "epoch": 15.5,
      "learning_rate": 8.25e-06,
      "loss": 0.8126,
      "step": 31
    },
    {
      "epoch": 16.0,
      "learning_rate": 8.210526315789473e-06,
      "loss": 0.8039,
      "step": 32
    },
    {
      "epoch": 16.5,
      "learning_rate": 8.171052631578947e-06,
      "loss": 0.8044,
      "step": 33
    },
    {
      "epoch": 17.0,
      "learning_rate": 8.131578947368421e-06,
      "loss": 0.7729,
      "step": 34
    },
    {
      "epoch": 17.5,
      "learning_rate": 8.092105263157896e-06,
      "loss": 0.7518,
      "step": 35
    },
    {
      "epoch": 18.0,
      "learning_rate": 8.052631578947368e-06,
      "loss": 0.7796,
      "step": 36
    },
    {
      "epoch": 18.5,
      "learning_rate": 8.013157894736842e-06,
      "loss": 0.7496,
      "step": 37
    },
    {
      "epoch": 19.0,
      "learning_rate": 7.973684210526316e-06,
      "loss": 0.7309,
      "step": 38
    },
    {
      "epoch": 19.5,
      "learning_rate": 7.934210526315789e-06,
      "loss": 0.7012,
      "step": 39
    },
    {
      "epoch": 20.0,
      "learning_rate": 7.894736842105263e-06,
      "loss": 0.7573,
      "step": 40
    },
    {
      "epoch": 20.0,
      "eval_accuracy_dropoff": 0.5213972797541228,
      "eval_accuracy_undropoff": 0.915831108572634,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.324261253088216,
      "eval_iou_undropoff": 0.900986633054498,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.6206496953964233,
      "eval_mean_accuracy": 0.7186141941633784,
      "eval_mean_iou": 0.40841596204757136,
      "eval_overall_accuracy": 0.8994308471679687,
      "eval_runtime": 12.3072,
      "eval_samples_per_second": 1.625,
      "eval_steps_per_second": 0.163,
      "step": 40
    }
  ],
  "max_steps": 240,
  "num_train_epochs": 120,
  "total_flos": 2.3151996998516736e+17,
  "trial_name": null,
  "trial_params": null
}