{
  "best_metric": 0.31975847482681274,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGB-b0_7/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": 5e-06,
      "loss": 1.2178,
      "step": 1
    },
    {
      "epoch": 0.67,
      "learning_rate": 1e-05,
      "loss": 1.2142,
      "step": 2
    },
    {
      "epoch": 1.0,
      "learning_rate": 1.5e-05,
      "loss": 1.2223,
      "step": 3
    },
    {
      "epoch": 1.33,
      "learning_rate": 2e-05,
      "loss": 1.208,
      "step": 4
    },
    {
      "epoch": 1.67,
      "learning_rate": 2.5e-05,
      "loss": 1.2072,
      "step": 5
    },
    {
      "epoch": 2.0,
      "learning_rate": 3e-05,
      "loss": 1.2015,
      "step": 6
    },
    {
      "epoch": 2.33,
      "learning_rate": 3.5000000000000004e-05,
      "loss": 1.1888,
      "step": 7
    },
    {
      "epoch": 2.67,
      "learning_rate": 4e-05,
      "loss": 1.1768,
      "step": 8
    },
    {
      "epoch": 3.0,
      "learning_rate": 4.5e-05,
      "loss": 1.1543,
      "step": 9
    },
    {
      "epoch": 3.33,
      "learning_rate": 5e-05,
      "loss": 1.1505,
      "step": 10
    },
    {
      "epoch": 3.33,
      "eval_accuracy_dropoff": 0.9456199600248122,
      "eval_accuracy_undropoff": 0.2616499186280552,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.07032648592680048,
      "eval_iou_undropoff": 0.2615922132145178,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.1103304624557495,
      "eval_mean_accuracy": 0.6036349393264338,
      "eval_mean_iou": 0.11063956638043942,
      "eval_overall_accuracy": 0.2919347127278646,
      "eval_runtime": 1.7484,
      "eval_samples_per_second": 8.579,
      "eval_steps_per_second": 0.572,
      "step": 10
    },
    {
      "epoch": 3.67,
      "learning_rate": 5.500000000000001e-05,
      "loss": 1.1405,
      "step": 11
    },
    {
      "epoch": 4.0,
      "learning_rate": 6e-05,
      "loss": 1.1039,
      "step": 12
    },
    {
      "epoch": 4.33,
      "learning_rate": 6.500000000000001e-05,
      "loss": 1.1211,
      "step": 13
    },
    {
      "epoch": 4.67,
      "learning_rate": 7.000000000000001e-05,
      "loss": 1.0914,
      "step": 14
    },
    {
      "epoch": 5.0,
      "learning_rate": 7.500000000000001e-05,
      "loss": 1.0583,
      "step": 15
    },
    {
      "epoch": 5.33,
      "learning_rate": 8e-05,
      "loss": 1.0527,
      "step": 16
    },
    {
      "epoch": 5.67,
      "learning_rate": 8.5e-05,
      "loss": 1.0227,
      "step": 17
    },
    {
      "epoch": 6.0,
      "learning_rate": 9e-05,
      "loss": 1.1078,
      "step": 18
    },
    {
      "epoch": 6.33,
      "learning_rate": 8.973684210526316e-05,
      "loss": 1.0001,
      "step": 19
    },
    {
      "epoch": 6.67,
      "learning_rate": 8.947368421052632e-05,
      "loss": 0.9635,
      "step": 20
    },
    {
      "epoch": 6.67,
      "eval_accuracy_dropoff": 0.8177051025800078,
      "eval_accuracy_undropoff": 0.8763050644323176,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.2434656953279978,
      "eval_iou_undropoff": 0.8694137027963167,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.0114336013793945,
      "eval_mean_accuracy": 0.8470050835061627,
      "eval_mean_iou": 0.3709597993747715,
      "eval_overall_accuracy": 0.8737103780110677,
      "eval_runtime": 1.7139,
      "eval_samples_per_second": 8.752,
      "eval_steps_per_second": 0.583,
      "step": 20
    },
    {
      "epoch": 7.0,
      "learning_rate": 8.921052631578948e-05,
      "loss": 1.0405,
      "step": 21
    },
    {
      "epoch": 7.33,
      "learning_rate": 8.894736842105263e-05,
      "loss": 0.9095,
      "step": 22
    },
    {
      "epoch": 7.67,
      "learning_rate": 8.86842105263158e-05,
      "loss": 0.9104,
      "step": 23
    },
    {
      "epoch": 8.0,
      "learning_rate": 8.842105263157894e-05,
      "loss": 0.8736,
      "step": 24
    },
    {
      "epoch": 8.33,
      "learning_rate": 8.815789473684211e-05,
      "loss": 0.8849,
      "step": 25
    },
    {
      "epoch": 8.67,
      "learning_rate": 8.789473684210527e-05,
      "loss": 0.8143,
      "step": 26
    },
    {
      "epoch": 9.0,
      "learning_rate": 8.763157894736843e-05,
      "loss": 0.7846,
      "step": 27
    },
    {
      "epoch": 9.33,
      "learning_rate": 8.736842105263158e-05,
      "loss": 0.7748,
      "step": 28
    },
    {
      "epoch": 9.67,
      "learning_rate": 8.710526315789474e-05,
      "loss": 0.7513,
      "step": 29
    },
    {
      "epoch": 10.0,
      "learning_rate": 8.68421052631579e-05,
      "loss": 0.9358,
      "step": 30
    },
    {
      "epoch": 10.0,
      "eval_accuracy_dropoff": 0.5848209157534404,
      "eval_accuracy_undropoff": 0.9605968730608304,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.3193744373731639,
      "eval_iou_undropoff": 0.9425019881457949,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.8242360353469849,
      "eval_mean_accuracy": 0.7727088944071354,
      "eval_mean_iou": 0.42062547517298626,
      "eval_overall_accuracy": 0.9439582824707031,
      "eval_runtime": 1.7189,
      "eval_samples_per_second": 8.726,
      "eval_steps_per_second": 0.582,
      "step": 30
    },
    {
      "epoch": 10.33,
      "learning_rate": 8.657894736842107e-05,
      "loss": 0.7191,
      "step": 31
    },
    {
      "epoch": 10.67,
      "learning_rate": 8.631578947368421e-05,
      "loss": 0.73,
      "step": 32
    },
    {
      "epoch": 11.0,
      "learning_rate": 8.605263157894737e-05,
      "loss": 0.6168,
      "step": 33
    },
    {
      "epoch": 11.33,
      "learning_rate": 8.578947368421054e-05,
      "loss": 0.6559,
      "step": 34
    },
    {
      "epoch": 11.67,
      "learning_rate": 8.552631578947369e-05,
      "loss": 0.6277,
      "step": 35
    },
    {
      "epoch": 12.0,
      "learning_rate": 8.526315789473684e-05,
      "loss": 0.6168,
      "step": 36
    },
    {
      "epoch": 12.33,
      "learning_rate": 8.5e-05,
      "loss": 0.6095,
      "step": 37
    },
    {
      "epoch": 12.67,
      "learning_rate": 8.473684210526316e-05,
      "loss": 0.5927,
      "step": 38
    },
    {
      "epoch": 13.0,
      "learning_rate": 8.447368421052632e-05,
      "loss": 0.8162,
      "step": 39
    },
    {
      "epoch": 13.33,
      "learning_rate": 8.421052631578948e-05,
      "loss": 0.579,
      "step": 40
    },
    {
      "epoch": 13.33,
      "eval_accuracy_dropoff": 0.5401589817814231,
      "eval_accuracy_undropoff": 0.982923599779886,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.3951496002117638,
      "eval_iou_undropoff": 0.9624271055937778,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.5702608823776245,
      "eval_mean_accuracy": 0.7615412907806545,
      "eval_mean_iou": 0.4525255686018472,
      "eval_overall_accuracy": 0.9633188883463542,
      "eval_runtime": 1.8153,
      "eval_samples_per_second": 8.263,
      "eval_steps_per_second": 0.551,
      "step": 40
    },
    {
      "epoch": 13.67,
      "learning_rate": 8.394736842105263e-05,
      "loss": 0.5659,
      "step": 41
    },
    {
      "epoch": 14.0,
      "learning_rate": 8.368421052631579e-05,
      "loss": 0.5447,
      "step": 42
    },
    {
      "epoch": 14.33,
      "learning_rate": 8.342105263157896e-05,
      "loss": 0.5265,
      "step": 43
    },
    {
      "epoch": 14.67,
      "learning_rate": 8.31578947368421e-05,
      "loss": 0.5263,
      "step": 44
    },
    {
      "epoch": 15.0,
      "learning_rate": 8.289473684210527e-05,
      "loss": 0.4922,
      "step": 45
    },
    {
      "epoch": 15.33,
      "learning_rate": 8.263157894736843e-05,
      "loss": 0.5099,
      "step": 46
    },
    {
      "epoch": 15.67,
      "learning_rate": 8.236842105263158e-05,
      "loss": 0.4706,
      "step": 47
    },
    {
      "epoch": 16.0,
      "learning_rate": 8.210526315789474e-05,
      "loss": 0.4809,
      "step": 48
    },
    {
      "epoch": 16.33,
      "learning_rate": 8.18421052631579e-05,
      "loss": 0.4888,
      "step": 49
    },
    {
      "epoch": 16.67,
      "learning_rate": 8.157894736842105e-05,
      "loss": 0.4411,
      "step": 50
    },
    {
      "epoch": 16.67,
      "eval_accuracy_dropoff": 0.48720334505019874,
      "eval_accuracy_undropoff": 0.9888639114094216,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.39281844198905264,
      "eval_iou_undropoff": 0.9659161382921264,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.4165554940700531,
      "eval_mean_accuracy": 0.7380336282298102,
      "eval_mean_iou": 0.452911526760393,
      "eval_overall_accuracy": 0.9666514078776042,
      "eval_runtime": 1.7457,
      "eval_samples_per_second": 8.593,
      "eval_steps_per_second": 0.573,
      "step": 50
    },
    {
      "epoch": 17.0,
      "learning_rate": 8.131578947368422e-05,
      "loss": 0.4717,
      "step": 51
    },
    {
      "epoch": 17.33,
      "learning_rate": 8.105263157894737e-05,
      "loss": 0.4523,
      "step": 52
    },
    {
      "epoch": 17.67,
      "learning_rate": 8.078947368421052e-05,
      "loss": 0.4215,
      "step": 53
    },
    {
      "epoch": 18.0,
      "learning_rate": 8.05263157894737e-05,
      "loss": 0.4007,
      "step": 54
    },
    {
      "epoch": 18.33,
      "learning_rate": 8.026315789473685e-05,
      "loss": 0.4253,
      "step": 55
    },
    {
      "epoch": 18.67,
      "learning_rate": 8e-05,
      "loss": 0.4153,
      "step": 56
    },
    {
      "epoch": 19.0,
      "learning_rate": 7.973684210526316e-05,
      "loss": 0.6574,
      "step": 57
    },
    {
      "epoch": 19.33,
      "learning_rate": 7.947368421052632e-05,
      "loss": 0.3899,
      "step": 58
    },
    {
      "epoch": 19.67,
      "learning_rate": 7.921052631578948e-05,
      "loss": 0.4108,
      "step": 59
    },
    {
      "epoch": 20.0,
      "learning_rate": 7.894736842105263e-05,
      "loss": 0.4311,
      "step": 60
    },
    {
      "epoch": 20.0,
      "eval_accuracy_dropoff": 0.44003721827831,
      "eval_accuracy_undropoff": 0.9911496700950386,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.3694591258053316,
      "eval_iou_undropoff": 0.966086774739557,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.38426053524017334,
      "eval_mean_accuracy": 0.7155934441866743,
      "eval_mean_iou": 0.6677729502724443,
      "eval_overall_accuracy": 0.966747538248698,
      "eval_runtime": 1.8089,
      "eval_samples_per_second": 8.292,
      "eval_steps_per_second": 0.553,
      "step": 60
    },
    {
      "epoch": 20.33,
      "learning_rate": 7.868421052631579e-05,
      "loss": 0.3803,
      "step": 61
    },
    {
      "epoch": 20.67,
      "learning_rate": 7.842105263157896e-05,
      "loss": 0.3828,
      "step": 62
    },
    {
      "epoch": 21.0,
      "learning_rate": 7.815789473684212e-05,
      "loss": 0.5699,
      "step": 63
    },
    {
      "epoch": 21.33,
      "learning_rate": 7.789473684210526e-05,
      "loss": 0.3803,
      "step": 64
    },
    {
      "epoch": 21.67,
      "learning_rate": 7.763157894736843e-05,
      "loss": 0.3709,
      "step": 65
    },
    {
      "epoch": 22.0,
      "learning_rate": 7.736842105263159e-05,
      "loss": 0.4738,
      "step": 66
    },
    {
      "epoch": 22.33,
      "learning_rate": 7.710526315789474e-05,
      "loss": 0.3891,
      "step": 67
    },
    {
      "epoch": 22.67,
      "learning_rate": 7.68421052631579e-05,
      "loss": 0.3576,
      "step": 68
    },
    {
      "epoch": 23.0,
      "learning_rate": 7.657894736842105e-05,
      "loss": 0.5494,
      "step": 69
    },
    {
      "epoch": 23.33,
      "learning_rate": 7.631578947368421e-05,
      "loss": 0.3437,
      "step": 70
    },
    {
      "epoch": 23.33,
      "eval_accuracy_dropoff": 0.39948767431709054,
      "eval_accuracy_undropoff": 0.9917728653036201,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.33924477871099273,
      "eval_iou_undropoff": 0.9649275553674269,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 0.35898175835609436,
      "eval_mean_accuracy": 0.6956302698103554,
      "eval_mean_iou": 0.43472411135947325,
      "eval_overall_accuracy": 0.9655476888020833,
      "eval_runtime": 1.8305,
      "eval_samples_per_second": 8.195,
      "eval_steps_per_second": 0.546,
      "step": 70
    },
    {
      "epoch": 23.67,
      "learning_rate": 7.605263157894738e-05,
      "loss": 0.3447,
      "step": 71
    },
    {
      "epoch": 24.0,
      "learning_rate": 7.578947368421052e-05,
      "loss": 0.3733,
      "step": 72
    },
    {
      "epoch": 24.33,
      "learning_rate": 7.55263157894737e-05,
      "loss": 0.3481,
      "step": 73
    },
    {
      "epoch": 24.67,
      "learning_rate": 7.526315789473685e-05,
      "loss": 0.3377,
      "step": 74
    },
    {
      "epoch": 25.0,
      "learning_rate": 7.500000000000001e-05,
      "loss": 0.5311,
      "step": 75
    },
    {
      "epoch": 25.33,
      "learning_rate": 7.473684210526316e-05,
      "loss": 0.3228,
      "step": 76
    },
    {
      "epoch": 25.67,
      "learning_rate": 7.447368421052632e-05,
      "loss": 0.3248,
      "step": 77
    },
    {
      "epoch": 26.0,
      "learning_rate": 7.421052631578948e-05,
      "loss": 0.5215,
      "step": 78
    },
    {
      "epoch": 26.33,
      "learning_rate": 7.394736842105263e-05,
      "loss": 0.3395,
      "step": 79
    },
    {
      "epoch": 26.67,
      "learning_rate": 7.368421052631579e-05,
      "loss": 0.3136,
      "step": 80
    },
    {
      "epoch": 26.67,
      "eval_accuracy_dropoff": 0.33123693339766125,
      "eval_accuracy_undropoff": 0.9931389986088537,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_dropoff": 0.2885107958297481,
      "eval_iou_undropoff": 0.9632929658594979,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.31975847482681274,
      "eval_mean_accuracy": 0.6621879660032575,
      "eval_mean_iou": 0.625901880844623,
      "eval_overall_accuracy": 0.9638313293457031,
      "eval_runtime": 1.7647,
      "eval_samples_per_second": 8.5,
      "eval_steps_per_second": 0.567,
      "step": 80
    }
  ],
  "max_steps": 360,
  "num_train_epochs": 120,
  "total_flos": 1.605671807090688e+16,
  "trial_name": null,
  "trial_params": null
}