{
  "best_metric": 0.6250978708267212,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGBD-b5_1/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": 2.5e-07,
      "loss": 0.8297,
      "step": 1
    },
    {
      "epoch": 1.0,
      "learning_rate": 5e-07,
      "loss": 0.8296,
      "step": 2
    },
    {
      "epoch": 1.5,
      "learning_rate": 7.5e-07,
      "loss": 0.8276,
      "step": 3
    },
    {
      "epoch": 2.0,
      "learning_rate": 1e-06,
      "loss": 0.8294,
      "step": 4
    },
    {
      "epoch": 2.5,
      "learning_rate": 1.25e-06,
      "loss": 0.8244,
      "step": 5
    },
    {
      "epoch": 3.0,
      "learning_rate": 1.5e-06,
      "loss": 0.822,
      "step": 6
    },
    {
      "epoch": 3.5,
      "learning_rate": 1.7500000000000002e-06,
      "loss": 0.8221,
      "step": 7
    },
    {
      "epoch": 4.0,
      "learning_rate": 2e-06,
      "loss": 0.8133,
      "step": 8
    },
    {
      "epoch": 4.5,
      "learning_rate": 2.25e-06,
      "loss": 0.8076,
      "step": 9
    },
    {
      "epoch": 5.0,
      "learning_rate": 2.5e-06,
      "loss": 0.8047,
      "step": 10
    },
    {
      "epoch": 5.0,
      "eval_accuracy_dropoff": 0.5048739649991972,
      "eval_accuracy_undropoff": 0.7580744633956797,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.08117782339287308,
      "eval_iou_undropoff": 0.7421699512154445,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.986731231212616,
      "eval_mean_accuracy": 0.6314742141974385,
      "eval_mean_iou": 0.2744492582027725,
      "eval_overall_accuracy": 0.7475465774536133,
      "eval_runtime": 3.2198,
      "eval_samples_per_second": 6.212,
      "eval_steps_per_second": 0.621,
      "step": 10
    },
    {
      "epoch": 5.5,
      "learning_rate": 2.75e-06,
      "loss": 0.7953,
      "step": 11
    },
    {
      "epoch": 6.0,
      "learning_rate": 3e-06,
      "loss": 0.7953,
      "step": 12
    },
    {
      "epoch": 6.5,
      "learning_rate": 2.986842105263158e-06,
      "loss": 0.7885,
      "step": 13
    },
    {
      "epoch": 7.0,
      "learning_rate": 2.973684210526316e-06,
      "loss": 0.7867,
      "step": 14
    },
    {
      "epoch": 7.5,
      "learning_rate": 2.960526315789474e-06,
      "loss": 0.7783,
      "step": 15
    },
    {
      "epoch": 8.0,
      "learning_rate": 2.9473684210526313e-06,
      "loss": 0.7697,
      "step": 16
    },
    {
      "epoch": 8.5,
      "learning_rate": 2.9342105263157896e-06,
      "loss": 0.768,
      "step": 17
    },
    {
      "epoch": 9.0,
      "learning_rate": 2.9210526315789475e-06,
      "loss": 0.7614,
      "step": 18
    },
    {
      "epoch": 9.5,
      "learning_rate": 2.9078947368421054e-06,
      "loss": 0.7604,
      "step": 19
    },
    {
      "epoch": 10.0,
      "learning_rate": 2.8947368421052634e-06,
      "loss": 0.7528,
      "step": 20
    },
    {
      "epoch": 10.0,
      "eval_accuracy_dropoff": 0.2405926741439024,
      "eval_accuracy_undropoff": 0.9508346957194045,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.11781963585717334,
      "eval_iou_undropoff": 0.9205125917871181,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.8525959253311157,
      "eval_mean_accuracy": 0.5957136849316534,
      "eval_mean_iou": 0.34611074254809715,
      "eval_overall_accuracy": 0.9213033676147461,
      "eval_runtime": 3.4854,
      "eval_samples_per_second": 5.738,
      "eval_steps_per_second": 0.574,
      "step": 20
    },
    {
      "epoch": 10.5,
      "learning_rate": 2.8815789473684213e-06,
      "loss": 0.7504,
      "step": 21
    },
    {
      "epoch": 11.0,
      "learning_rate": 2.8684210526315787e-06,
      "loss": 0.7408,
      "step": 22
    },
    {
      "epoch": 11.5,
      "learning_rate": 2.8552631578947367e-06,
      "loss": 0.7363,
      "step": 23
    },
    {
      "epoch": 12.0,
      "learning_rate": 2.8421052631578946e-06,
      "loss": 0.7381,
      "step": 24
    },
    {
      "epoch": 12.5,
      "learning_rate": 2.828947368421053e-06,
      "loss": 0.73,
      "step": 25
    },
    {
      "epoch": 13.0,
      "learning_rate": 2.815789473684211e-06,
      "loss": 0.7244,
      "step": 26
    },
    {
      "epoch": 13.5,
      "learning_rate": 2.8026315789473687e-06,
      "loss": 0.7226,
      "step": 27
    },
    {
      "epoch": 14.0,
      "learning_rate": 2.789473684210526e-06,
      "loss": 0.7193,
      "step": 28
    },
    {
      "epoch": 14.5,
      "learning_rate": 2.776315789473684e-06,
      "loss": 0.7159,
      "step": 29
    },
    {
      "epoch": 15.0,
      "learning_rate": 2.763157894736842e-06,
      "loss": 0.7087,
      "step": 30
    },
    {
      "epoch": 15.0,
      "eval_accuracy_dropoff": 0.124291841555999,
      "eval_accuracy_undropoff": 0.9823552180796178,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.08867354586184664,
      "eval_iou_undropoff": 0.9464005897482354,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.7022603154182434,
      "eval_mean_accuracy": 0.5533235298178084,
      "eval_mean_iou": 0.34502471187002737,
      "eval_overall_accuracy": 0.9466775894165039,
      "eval_runtime": 3.2954,
      "eval_samples_per_second": 6.069,
      "eval_steps_per_second": 0.607,
      "step": 30
    },
    {
      "epoch": 15.5,
      "learning_rate": 2.75e-06,
      "loss": 0.7036,
      "step": 31
    },
    {
      "epoch": 16.0,
      "learning_rate": 2.736842105263158e-06,
      "loss": 0.6985,
      "step": 32
    },
    {
      "epoch": 16.5,
      "learning_rate": 2.723684210526316e-06,
      "loss": 0.6966,
      "step": 33
    },
    {
      "epoch": 17.0,
      "learning_rate": 2.7105263157894737e-06,
      "loss": 0.699,
      "step": 34
    },
    {
      "epoch": 17.5,
      "learning_rate": 2.6973684210526316e-06,
      "loss": 0.6958,
      "step": 35
    },
    {
      "epoch": 18.0,
      "learning_rate": 2.6842105263157895e-06,
      "loss": 0.6815,
      "step": 36
    },
    {
      "epoch": 18.5,
      "learning_rate": 2.6710526315789474e-06,
      "loss": 0.678,
      "step": 37
    },
    {
      "epoch": 19.0,
      "learning_rate": 2.6578947368421053e-06,
      "loss": 0.6791,
      "step": 38
    },
    {
      "epoch": 19.5,
      "learning_rate": 2.644736842105263e-06,
      "loss": 0.6869,
      "step": 39
    },
    {
      "epoch": 20.0,
      "learning_rate": 2.631578947368421e-06,
      "loss": 0.6601,
      "step": 40
    },
    {
      "epoch": 20.0,
      "eval_accuracy_dropoff": 0.09481410124085415,
      "eval_accuracy_undropoff": 0.9831825006940458,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.06841004322585342,
      "eval_iou_undropoff": 0.9460320542390953,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.6250978708267212,
      "eval_mean_accuracy": 0.5389983009674499,
      "eval_mean_iou": 0.3381473658216496,
      "eval_overall_accuracy": 0.9462448120117187,
      "eval_runtime": 3.2662,
      "eval_samples_per_second": 6.123,
      "eval_steps_per_second": 0.612,
      "step": 40
    }
  ],
  "max_steps": 240,
  "num_train_epochs": 120,
  "total_flos": 4.630914352860365e+17,
  "trial_name": null,
  "trial_params": null
}