{
  "best_metric": 0.3731442093849182,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b0_6/checkpoint-80",
  "epoch": 26.666666666666668,
  "global_step": 80,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.33,
      "learning_rate": 3.888888888888889e-06,
      "loss": 1.1755,
      "step": 1
    },
    {
      "epoch": 0.67,
      "learning_rate": 7.777777777777777e-06,
      "loss": 1.1749,
      "step": 2
    },
    {
      "epoch": 1.0,
      "learning_rate": 1.1666666666666665e-05,
      "loss": 1.183,
      "step": 3
    },
    {
      "epoch": 1.33,
      "learning_rate": 1.5555555555555555e-05,
      "loss": 1.1718,
      "step": 4
    },
    {
      "epoch": 1.67,
      "learning_rate": 1.9444444444444442e-05,
      "loss": 1.1706,
      "step": 5
    },
    {
      "epoch": 2.0,
      "learning_rate": 2.333333333333333e-05,
      "loss": 1.1696,
      "step": 6
    },
    {
      "epoch": 2.33,
      "learning_rate": 2.722222222222222e-05,
      "loss": 1.1558,
      "step": 7
    },
    {
      "epoch": 2.67,
      "learning_rate": 3.111111111111111e-05,
      "loss": 1.1489,
      "step": 8
    },
    {
      "epoch": 3.0,
      "learning_rate": 3.5e-05,
      "loss": 1.135,
      "step": 9
    },
    {
      "epoch": 3.33,
      "learning_rate": 3.8888888888888884e-05,
      "loss": 1.1234,
      "step": 10
    },
    {
      "epoch": 3.33,
      "eval_accuracy_dropoff": 0.772015071105291,
      "eval_accuracy_undropoff": 0.353825865102452,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.1801195043477445,
      "eval_iou_undropoff": 0.3536296682807036,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.097259283065796,
      "eval_mean_accuracy": 0.5629204681038715,
      "eval_mean_iou": 0.17791639087614938,
      "eval_overall_accuracy": 0.37234242757161456,
      "eval_runtime": 1.873,
      "eval_samples_per_second": 8.008,
      "eval_steps_per_second": 0.534,
      "step": 10
    },
    {
      "epoch": 3.67,
      "learning_rate": 4.277777777777778e-05,
      "loss": 1.1168,
      "step": 11
    },
    {
      "epoch": 4.0,
      "learning_rate": 4.666666666666666e-05,
      "loss": 1.0895,
      "step": 12
    },
    {
      "epoch": 4.33,
      "learning_rate": 5.055555555555555e-05,
      "loss": 1.1027,
      "step": 13
    },
    {
      "epoch": 4.67,
      "learning_rate": 5.444444444444444e-05,
      "loss": 1.0795,
      "step": 14
    },
    {
      "epoch": 5.0,
      "learning_rate": 5.833333333333333e-05,
      "loss": 1.0598,
      "step": 15
    },
    {
      "epoch": 5.33,
      "learning_rate": 6.222222222222222e-05,
      "loss": 1.0449,
      "step": 16
    },
    {
      "epoch": 5.67,
      "learning_rate": 6.611111111111111e-05,
      "loss": 1.0268,
      "step": 17
    },
    {
      "epoch": 6.0,
      "learning_rate": 7e-05,
      "loss": 1.0843,
      "step": 18
    },
    {
      "epoch": 6.33,
      "learning_rate": 6.97953216374269e-05,
      "loss": 1.0029,
      "step": 19
    },
    {
      "epoch": 6.67,
      "learning_rate": 6.95906432748538e-05,
      "loss": 0.975,
      "step": 20
    },
    {
      "epoch": 6.67,
      "eval_accuracy_dropoff": 0.8258609598639924,
      "eval_accuracy_undropoff": 0.8102080545984994,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.24279987572018127,
      "eval_iou_undropoff": 0.8068537963656117,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.0260422229766846,
      "eval_mean_accuracy": 0.8180345072312459,
      "eval_mean_iou": 0.34988455736193097,
      "eval_overall_accuracy": 0.810901133219401,
      "eval_runtime": 1.6566,
      "eval_samples_per_second": 9.055,
      "eval_steps_per_second": 0.604,
      "step": 20
    },
    {
      "epoch": 7.0,
      "learning_rate": 6.938596491228069e-05,
      "loss": 1.0358,
      "step": 21
    },
    {
      "epoch": 7.33,
      "learning_rate": 6.91812865497076e-05,
      "loss": 0.9361,
      "step": 22
    },
    {
      "epoch": 7.67,
      "learning_rate": 6.89766081871345e-05,
      "loss": 0.9311,
      "step": 23
    },
    {
      "epoch": 8.0,
      "learning_rate": 6.877192982456139e-05,
      "loss": 0.9149,
      "step": 24
    },
    {
      "epoch": 8.33,
      "learning_rate": 6.85672514619883e-05,
      "loss": 0.9056,
      "step": 25
    },
    {
      "epoch": 8.67,
      "learning_rate": 6.83625730994152e-05,
      "loss": 0.8637,
      "step": 26
    },
    {
      "epoch": 9.0,
      "learning_rate": 6.81578947368421e-05,
      "loss": 0.7991,
      "step": 27
    },
    {
      "epoch": 9.33,
      "learning_rate": 6.7953216374269e-05,
      "loss": 0.8143,
      "step": 28
    },
    {
      "epoch": 9.67,
      "learning_rate": 6.774853801169589e-05,
      "loss": 0.7999,
      "step": 29
    },
    {
      "epoch": 10.0,
      "learning_rate": 6.75438596491228e-05,
      "loss": 0.9464,
      "step": 30
    },
    {
      "epoch": 10.0,
      "eval_accuracy_dropoff": 0.5212052289383601,
      "eval_accuracy_undropoff": 0.9700488444545206,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.3384996437670423,
      "eval_iou_undropoff": 0.9507046257182781,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.8130465149879456,
      "eval_mean_accuracy": 0.7456270366964404,
      "eval_mean_iou": 0.4297347564951068,
      "eval_overall_accuracy": 0.9501749674479166,
      "eval_runtime": 1.7097,
      "eval_samples_per_second": 8.773,
      "eval_steps_per_second": 0.585,
      "step": 30
    },
    {
      "epoch": 10.33,
      "learning_rate": 6.733918128654971e-05,
      "loss": 0.763,
      "step": 31
    },
    {
      "epoch": 10.67,
      "learning_rate": 6.71345029239766e-05,
      "loss": 0.7591,
      "step": 32
    },
    {
      "epoch": 11.0,
      "learning_rate": 6.69298245614035e-05,
      "loss": 0.6623,
      "step": 33
    },
    {
      "epoch": 11.33,
      "learning_rate": 6.672514619883041e-05,
      "loss": 0.7138,
      "step": 34
    },
    {
      "epoch": 11.67,
      "learning_rate": 6.65204678362573e-05,
      "loss": 0.6695,
      "step": 35
    },
    {
      "epoch": 12.0,
      "learning_rate": 6.63157894736842e-05,
      "loss": 0.6987,
      "step": 36
    },
    {
      "epoch": 12.33,
      "learning_rate": 6.611111111111111e-05,
      "loss": 0.6494,
      "step": 37
    },
    {
      "epoch": 12.67,
      "learning_rate": 6.5906432748538e-05,
      "loss": 0.6317,
      "step": 38
    },
    {
      "epoch": 13.0,
      "learning_rate": 6.570175438596491e-05,
      "loss": 0.834,
      "step": 39
    },
    {
      "epoch": 13.33,
      "learning_rate": 6.54970760233918e-05,
      "loss": 0.6167,
      "step": 40
    },
    {
      "epoch": 13.33,
      "eval_accuracy_dropoff": 0.5047671560181037,
      "eval_accuracy_undropoff": 0.9829004494881923,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.3743041743152479,
      "eval_iou_undropoff": 0.961015278770039,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.6001297235488892,
      "eval_mean_accuracy": 0.743833802753148,
      "eval_mean_iou": 0.44510648436176226,
      "eval_overall_accuracy": 0.9617296854654948,
      "eval_runtime": 1.7423,
      "eval_samples_per_second": 8.609,
      "eval_steps_per_second": 0.574,
      "step": 40
    },
    {
      "epoch": 13.67,
      "learning_rate": 6.52923976608187e-05,
      "loss": 0.6201,
      "step": 41
    },
    {
      "epoch": 14.0,
      "learning_rate": 6.508771929824561e-05,
      "loss": 0.6009,
      "step": 42
    },
    {
      "epoch": 14.33,
      "learning_rate": 6.488304093567252e-05,
      "loss": 0.5716,
      "step": 43
    },
    {
      "epoch": 14.67,
      "learning_rate": 6.467836257309941e-05,
      "loss": 0.5779,
      "step": 44
    },
    {
      "epoch": 15.0,
      "learning_rate": 6.447368421052631e-05,
      "loss": 0.5397,
      "step": 45
    },
    {
      "epoch": 15.33,
      "learning_rate": 6.426900584795322e-05,
      "loss": 0.5569,
      "step": 46
    },
    {
      "epoch": 15.67,
      "learning_rate": 6.406432748538011e-05,
      "loss": 0.5152,
      "step": 47
    },
    {
      "epoch": 16.0,
      "learning_rate": 6.3859649122807e-05,
      "loss": 0.5266,
      "step": 48
    },
    {
      "epoch": 16.33,
      "learning_rate": 6.365497076023391e-05,
      "loss": 0.5389,
      "step": 49
    },
    {
      "epoch": 16.67,
      "learning_rate": 6.345029239766081e-05,
      "loss": 0.4818,
      "step": 50
    },
    {
      "epoch": 16.67,
      "eval_accuracy_dropoff": 0.4572736462425621,
      "eval_accuracy_undropoff": 0.9901589972677334,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.38151532722193204,
      "eval_iou_undropoff": 0.9659037872797944,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.46292775869369507,
      "eval_mean_accuracy": 0.7237163217551478,
      "eval_mean_iou": 0.4491397048339088,
      "eval_overall_accuracy": 0.9665639241536458,
      "eval_runtime": 1.7358,
      "eval_samples_per_second": 8.641,
      "eval_steps_per_second": 0.576,
      "step": 50
    },
    {
      "epoch": 17.0,
      "learning_rate": 6.324561403508772e-05,
      "loss": 0.5289,
      "step": 51
    },
    {
      "epoch": 17.33,
      "learning_rate": 6.304093567251461e-05,
      "loss": 0.4991,
      "step": 52
    },
    {
      "epoch": 17.67,
      "learning_rate": 6.283625730994151e-05,
      "loss": 0.4643,
      "step": 53
    },
    {
      "epoch": 18.0,
      "learning_rate": 6.263157894736842e-05,
      "loss": 0.4359,
      "step": 54
    },
    {
      "epoch": 18.33,
      "learning_rate": 6.242690058479532e-05,
      "loss": 0.4679,
      "step": 55
    },
    {
      "epoch": 18.67,
      "learning_rate": 6.222222222222222e-05,
      "loss": 0.4619,
      "step": 56
    },
    {
      "epoch": 19.0,
      "learning_rate": 6.201754385964911e-05,
      "loss": 0.6641,
      "step": 57
    },
    {
      "epoch": 19.33,
      "learning_rate": 6.181286549707602e-05,
      "loss": 0.4316,
      "step": 58
    },
    {
      "epoch": 19.67,
      "learning_rate": 6.160818713450292e-05,
      "loss": 0.4475,
      "step": 59
    },
    {
      "epoch": 20.0,
      "learning_rate": 6.140350877192981e-05,
      "loss": 0.4733,
      "step": 60
    },
    {
      "epoch": 20.0,
      "eval_accuracy_dropoff": 0.42558641762584143,
      "eval_accuracy_undropoff": 0.987872440296196,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.33826825716385683,
      "eval_iou_undropoff": 0.9622649725509712,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.43790048360824585,
      "eval_mean_accuracy": 0.7067294289610188,
      "eval_mean_iou": 0.4335110765716093,
      "eval_overall_accuracy": 0.9629755655924479,
      "eval_runtime": 1.7742,
      "eval_samples_per_second": 8.455,
      "eval_steps_per_second": 0.564,
      "step": 60
    },
    {
      "epoch": 20.33,
      "learning_rate": 6.119883040935672e-05,
      "loss": 0.423,
      "step": 61
    },
    {
      "epoch": 20.67,
      "learning_rate": 6.0994152046783624e-05,
      "loss": 0.4219,
      "step": 62
    },
    {
      "epoch": 21.0,
      "learning_rate": 6.0789473684210525e-05,
      "loss": 0.6326,
      "step": 63
    },
    {
      "epoch": 21.33,
      "learning_rate": 6.058479532163742e-05,
      "loss": 0.4299,
      "step": 64
    },
    {
      "epoch": 21.67,
      "learning_rate": 6.038011695906432e-05,
      "loss": 0.4147,
      "step": 65
    },
    {
      "epoch": 22.0,
      "learning_rate": 6.0175438596491224e-05,
      "loss": 0.4815,
      "step": 66
    },
    {
      "epoch": 22.33,
      "learning_rate": 5.9970760233918126e-05,
      "loss": 0.4477,
      "step": 67
    },
    {
      "epoch": 22.67,
      "learning_rate": 5.976608187134502e-05,
      "loss": 0.4031,
      "step": 68
    },
    {
      "epoch": 23.0,
      "learning_rate": 5.956140350877192e-05,
      "loss": 0.6048,
      "step": 69
    },
    {
      "epoch": 23.33,
      "learning_rate": 5.9356725146198824e-05,
      "loss": 0.3843,
      "step": 70
    },
    {
      "epoch": 23.33,
      "eval_accuracy_dropoff": 0.3821076573161486,
      "eval_accuracy_undropoff": 0.992239862567096,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.328346864743479,
      "eval_iou_undropoff": 0.9646259654307497,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.4072829484939575,
      "eval_mean_accuracy": 0.6871737599416223,
      "eval_mean_iou": 0.4309909433914096,
      "eval_overall_accuracy": 0.9652244567871093,
      "eval_runtime": 1.8122,
      "eval_samples_per_second": 8.277,
      "eval_steps_per_second": 0.552,
      "step": 70
    },
    {
      "epoch": 23.67,
      "learning_rate": 5.9152046783625726e-05,
      "loss": 0.3967,
      "step": 71
    },
    {
      "epoch": 24.0,
      "learning_rate": 5.894736842105262e-05,
      "loss": 0.4051,
      "step": 72
    },
    {
      "epoch": 24.33,
      "learning_rate": 5.874269005847952e-05,
      "loss": 0.3836,
      "step": 73
    },
    {
      "epoch": 24.67,
      "learning_rate": 5.853801169590643e-05,
      "loss": 0.3896,
      "step": 74
    },
    {
      "epoch": 25.0,
      "learning_rate": 5.833333333333333e-05,
      "loss": 0.5942,
      "step": 75
    },
    {
      "epoch": 25.33,
      "learning_rate": 5.812865497076023e-05,
      "loss": 0.3688,
      "step": 76
    },
    {
      "epoch": 25.67,
      "learning_rate": 5.792397660818713e-05,
      "loss": 0.3721,
      "step": 77
    },
    {
      "epoch": 26.0,
      "learning_rate": 5.771929824561403e-05,
      "loss": 0.5951,
      "step": 78
    },
    {
      "epoch": 26.33,
      "learning_rate": 5.751461988304093e-05,
      "loss": 0.3973,
      "step": 79
    },
    {
      "epoch": 26.67,
      "learning_rate": 5.730994152046783e-05,
      "loss": 0.3579,
      "step": 80
    },
    {
      "epoch": 26.67,
      "eval_accuracy_dropoff": 0.40901049923036276,
      "eval_accuracy_undropoff": 0.9908359437282933,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.34181486548107615,
      "eval_iou_undropoff": 0.9644297203691943,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.3731442093849182,
      "eval_mean_accuracy": 0.6999232214793281,
      "eval_mean_iou": 0.43541486195009016,
      "eval_overall_accuracy": 0.9650739034016927,
      "eval_runtime": 1.7608,
      "eval_samples_per_second": 8.519,
      "eval_steps_per_second": 0.568,
      "step": 80
    }
  ],
  "max_steps": 360,
  "num_train_epochs": 120,
  "total_flos": 1.605671807090688e+16,
  "trial_name": null,
  "trial_params": null
}