{
  "best_metric": 0.20112648606300354,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-train-SF-RGBD-b5/checkpoint-200",
  "epoch": 18.181818181818183,
  "global_step": 200,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.09,
      "learning_rate": 7.272727272727273e-08,
      "loss": 0.8005,
      "step": 1
    },
    {
      "epoch": 0.18,
      "learning_rate": 1.4545454545454545e-07,
      "loss": 0.7952,
      "step": 2
    },
    {
      "epoch": 0.27,
      "learning_rate": 2.1818181818181815e-07,
      "loss": 0.7806,
      "step": 3
    },
    {
      "epoch": 0.36,
      "learning_rate": 2.909090909090909e-07,
      "loss": 0.7985,
      "step": 4
    },
    {
      "epoch": 0.45,
      "learning_rate": 3.636363636363636e-07,
      "loss": 0.7916,
      "step": 5
    },
    {
      "epoch": 0.55,
      "learning_rate": 4.363636363636363e-07,
      "loss": 0.7885,
      "step": 6
    },
    {
      "epoch": 0.64,
      "learning_rate": 5.090909090909091e-07,
      "loss": 0.7869,
      "step": 7
    },
    {
      "epoch": 0.73,
      "learning_rate": 5.818181818181818e-07,
      "loss": 0.7871,
      "step": 8
    },
    {
      "epoch": 0.82,
      "learning_rate": 6.545454545454546e-07,
      "loss": 0.7895,
      "step": 9
    },
    {
      "epoch": 0.91,
      "learning_rate": 7.272727272727272e-07,
      "loss": 0.789,
      "step": 10
    },
    {
      "epoch": 0.91,
      "eval_accuracy_safe": 0.020290978060994367,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.8956518586322223,
      "eval_iou_safe": 0.009542102362547317,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.872225496966486,
      "eval_loss": 0.9554787278175354,
      "eval_mean_accuracy": 0.4579714183466083,
      "eval_mean_iou": 0.29392253310967775,
      "eval_overall_accuracy": 0.8697951017920651,
      "eval_runtime": 10.4712,
      "eval_samples_per_second": 6.399,
      "eval_steps_per_second": 0.478,
      "step": 10
    },
    {
      "epoch": 1.0,
      "learning_rate": 8e-07,
      "loss": 0.7808,
      "step": 11
    },
    {
      "epoch": 1.09,
      "learning_rate": 8.727272727272726e-07,
      "loss": 0.7854,
      "step": 12
    },
    {
      "epoch": 1.18,
      "learning_rate": 9.454545454545454e-07,
      "loss": 0.7698,
      "step": 13
    },
    {
      "epoch": 1.27,
      "learning_rate": 1.0181818181818181e-06,
      "loss": 0.7788,
      "step": 14
    },
    {
      "epoch": 1.36,
      "learning_rate": 1.0909090909090908e-06,
      "loss": 0.7783,
      "step": 15
    },
    {
      "epoch": 1.45,
      "learning_rate": 1.1636363636363636e-06,
      "loss": 0.7696,
      "step": 16
    },
    {
      "epoch": 1.55,
      "learning_rate": 1.2363636363636363e-06,
      "loss": 0.7638,
      "step": 17
    },
    {
      "epoch": 1.64,
      "learning_rate": 1.3090909090909091e-06,
      "loss": 0.7611,
      "step": 18
    },
    {
      "epoch": 1.73,
      "learning_rate": 1.3818181818181818e-06,
      "loss": 0.7597,
      "step": 19
    },
    {
      "epoch": 1.82,
      "learning_rate": 1.4545454545454544e-06,
      "loss": 0.7579,
      "step": 20
    },
    {
      "epoch": 1.82,
      "eval_accuracy_safe": 0.011667264197131082,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9614483463212281,
      "eval_iou_safe": 0.006852036536648151,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9338335778122896,
      "eval_loss": 0.8321800827980042,
      "eval_mean_accuracy": 0.4865578052591796,
      "eval_mean_iou": 0.3135618714496459,
      "eval_overall_accuracy": 0.9333933360541045,
      "eval_runtime": 11.8686,
      "eval_samples_per_second": 5.645,
      "eval_steps_per_second": 0.421,
      "step": 20
    },
    {
      "epoch": 1.91,
      "learning_rate": 1.5272727272727273e-06,
      "loss": 0.7626,
      "step": 21
    },
    {
      "epoch": 2.0,
      "learning_rate": 1.6e-06,
      "loss": 0.7399,
      "step": 22
    },
    {
      "epoch": 2.09,
      "learning_rate": 1.6727272727272726e-06,
      "loss": 0.7658,
      "step": 23
    },
    {
      "epoch": 2.18,
      "learning_rate": 1.7454545454545452e-06,
      "loss": 0.7404,
      "step": 24
    },
    {
      "epoch": 2.27,
      "learning_rate": 1.818181818181818e-06,
      "loss": 0.742,
      "step": 25
    },
    {
      "epoch": 2.36,
      "learning_rate": 1.8909090909090907e-06,
      "loss": 0.721,
      "step": 26
    },
    {
      "epoch": 2.45,
      "learning_rate": 1.9636363636363636e-06,
      "loss": 0.7222,
      "step": 27
    },
    {
      "epoch": 2.55,
      "learning_rate": 2.0363636363636362e-06,
      "loss": 0.7358,
      "step": 28
    },
    {
      "epoch": 2.64,
      "learning_rate": 2.109090909090909e-06,
      "loss": 0.7427,
      "step": 29
    },
    {
      "epoch": 2.73,
      "learning_rate": 2.1818181818181815e-06,
      "loss": 0.7103,
      "step": 30
    },
    {
      "epoch": 2.73,
      "eval_accuracy_safe": 0.005111776747198353,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9893036287919527,
      "eval_iou_safe": 0.004283838202726343,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9604181920132638,
      "eval_loss": 0.6728952527046204,
      "eval_mean_accuracy": 0.4972077027695755,
      "eval_mean_iou": 0.32156734340533005,
      "eval_overall_accuracy": 0.96023217955632,
      "eval_runtime": 12.232,
      "eval_samples_per_second": 5.477,
      "eval_steps_per_second": 0.409,
      "step": 30
    },
    {
      "epoch": 2.82,
      "learning_rate": 2.254545454545454e-06,
      "loss": 0.7102,
      "step": 31
    },
    {
      "epoch": 2.91,
      "learning_rate": 2.3272727272727272e-06,
      "loss": 0.7035,
      "step": 32
    },
    {
      "epoch": 3.0,
      "learning_rate": 2.4e-06,
      "loss": 0.6845,
      "step": 33
    },
    {
      "epoch": 3.09,
      "learning_rate": 2.4727272727272725e-06,
      "loss": 0.6932,
      "step": 34
    },
    {
      "epoch": 3.18,
      "learning_rate": 2.545454545454545e-06,
      "loss": 0.6848,
      "step": 35
    },
    {
      "epoch": 3.27,
      "learning_rate": 2.6181818181818183e-06,
      "loss": 0.6816,
      "step": 36
    },
    {
      "epoch": 3.36,
      "learning_rate": 2.690909090909091e-06,
      "loss": 0.6617,
      "step": 37
    },
    {
      "epoch": 3.45,
      "learning_rate": 2.7636363636363635e-06,
      "loss": 0.6924,
      "step": 38
    },
    {
      "epoch": 3.55,
      "learning_rate": 2.836363636363636e-06,
      "loss": 0.6775,
      "step": 39
    },
    {
      "epoch": 3.64,
      "learning_rate": 2.909090909090909e-06,
      "loss": 0.676,
      "step": 40
    },
    {
      "epoch": 3.64,
      "eval_accuracy_safe": 0.0021453271190165035,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9968958945126286,
      "eval_iou_safe": 0.0020376330502982305,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9675401930395611,
      "eval_loss": 0.5335736274719238,
      "eval_mean_accuracy": 0.49952061081582255,
      "eval_mean_iou": 0.32319260869661975,
      "eval_overall_accuracy": 0.9675125577556554,
      "eval_runtime": 10.8655,
      "eval_samples_per_second": 6.166,
      "eval_steps_per_second": 0.46,
      "step": 40
    },
    {
      "epoch": 3.73,
      "learning_rate": 2.981818181818182e-06,
      "loss": 0.6708,
      "step": 41
    },
    {
      "epoch": 3.82,
      "learning_rate": 3.0545454545454546e-06,
      "loss": 0.6587,
      "step": 42
    },
    {
      "epoch": 3.91,
      "learning_rate": 3.127272727272727e-06,
      "loss": 0.6511,
      "step": 43
    },
    {
      "epoch": 4.0,
      "learning_rate": 3.2e-06,
      "loss": 0.6533,
      "step": 44
    },
    {
      "epoch": 4.09,
      "learning_rate": 3.272727272727273e-06,
      "loss": 0.6371,
      "step": 45
    },
    {
      "epoch": 4.18,
      "learning_rate": 3.345454545454545e-06,
      "loss": 0.6289,
      "step": 46
    },
    {
      "epoch": 4.27,
      "learning_rate": 3.418181818181818e-06,
      "loss": 0.6143,
      "step": 47
    },
    {
      "epoch": 4.36,
      "learning_rate": 3.4909090909090904e-06,
      "loss": 0.6047,
      "step": 48
    },
    {
      "epoch": 4.45,
      "learning_rate": 3.5636363636363635e-06,
      "loss": 0.6173,
      "step": 49
    },
    {
      "epoch": 4.55,
      "learning_rate": 3.636363636363636e-06,
      "loss": 0.5955,
      "step": 50
    },
    {
      "epoch": 4.55,
      "eval_accuracy_safe": 0.00010023091661173242,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9992954468465131,
      "eval_iou_safe": 9.918496661090691e-05,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9697964694697941,
      "eval_loss": 0.4440336227416992,
      "eval_mean_accuracy": 0.4996978388815624,
      "eval_mean_iou": 0.32329855147880165,
      "eval_overall_accuracy": 0.96978082229842,
      "eval_runtime": 10.7685,
      "eval_samples_per_second": 6.222,
      "eval_steps_per_second": 0.464,
      "step": 50
    },
    {
      "epoch": 4.64,
      "learning_rate": 3.709090909090909e-06,
      "loss": 0.6184,
      "step": 51
    },
    {
      "epoch": 4.73,
      "learning_rate": 3.7818181818181815e-06,
      "loss": 0.6059,
      "step": 52
    },
    {
      "epoch": 4.82,
      "learning_rate": 3.8545454545454545e-06,
      "loss": 0.5942,
      "step": 53
    },
    {
      "epoch": 4.91,
      "learning_rate": 3.927272727272727e-06,
      "loss": 0.5764,
      "step": 54
    },
    {
      "epoch": 5.0,
      "learning_rate": 4e-06,
      "loss": 0.5745,
      "step": 55
    },
    {
      "epoch": 5.09,
      "learning_rate": 3.9961722488038276e-06,
      "loss": 0.5496,
      "step": 56
    },
    {
      "epoch": 5.18,
      "learning_rate": 3.992344497607655e-06,
      "loss": 0.5785,
      "step": 57
    },
    {
      "epoch": 5.27,
      "learning_rate": 3.988516746411483e-06,
      "loss": 0.5468,
      "step": 58
    },
    {
      "epoch": 5.36,
      "learning_rate": 3.984688995215311e-06,
      "loss": 0.5697,
      "step": 59
    },
    {
      "epoch": 5.45,
      "learning_rate": 3.980861244019139e-06,
      "loss": 0.5691,
      "step": 60
    },
    {
      "epoch": 5.45,
      "eval_accuracy_safe": 1.9275176271487e-05,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9997433241696639,
      "eval_iou_safe": 1.9217760285345306e-05,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9702130617813918,
      "eval_loss": 0.38122960925102234,
      "eval_mean_accuracy": 0.49988129967296774,
      "eval_mean_iou": 0.3234107598472257,
      "eval_overall_accuracy": 0.9702130787408174,
      "eval_runtime": 11.2754,
      "eval_samples_per_second": 5.942,
      "eval_steps_per_second": 0.443,
      "step": 60
    },
    {
      "epoch": 5.55,
      "learning_rate": 3.977033492822966e-06,
      "loss": 0.5238,
      "step": 61
    },
    {
      "epoch": 5.64,
      "learning_rate": 3.973205741626794e-06,
      "loss": 0.5514,
      "step": 62
    },
    {
      "epoch": 5.73,
      "learning_rate": 3.969377990430622e-06,
      "loss": 0.5261,
      "step": 63
    },
    {
      "epoch": 5.82,
      "learning_rate": 3.96555023923445e-06,
      "loss": 0.5348,
      "step": 64
    },
    {
      "epoch": 5.91,
      "learning_rate": 3.961722488038277e-06,
      "loss": 0.5614,
      "step": 65
    },
    {
      "epoch": 6.0,
      "learning_rate": 3.957894736842105e-06,
      "loss": 0.5475,
      "step": 66
    },
    {
      "epoch": 6.09,
      "learning_rate": 3.954066985645933e-06,
      "loss": 0.5134,
      "step": 67
    },
    {
      "epoch": 6.18,
      "learning_rate": 3.950239234449761e-06,
      "loss": 0.5047,
      "step": 68
    },
    {
      "epoch": 6.27,
      "learning_rate": 3.946411483253588e-06,
      "loss": 0.4845,
      "step": 69
    },
    {
      "epoch": 6.36,
      "learning_rate": 3.942583732057416e-06,
      "loss": 0.5067,
      "step": 70
    },
    {
      "epoch": 6.36,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.999641064518858,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9701133255524892,
      "eval_loss": 0.3590196967124939,
      "eval_mean_accuracy": 0.499820532259429,
      "eval_mean_iou": 0.3233711085174964,
      "eval_overall_accuracy": 0.9701132703183303,
      "eval_runtime": 10.8376,
      "eval_samples_per_second": 6.182,
      "eval_steps_per_second": 0.461,
      "step": 70
    },
    {
      "epoch": 6.45,
      "learning_rate": 3.938755980861244e-06,
      "loss": 0.4808,
      "step": 71
    },
    {
      "epoch": 6.55,
      "learning_rate": 3.934928229665072e-06,
      "loss": 0.5075,
      "step": 72
    },
    {
      "epoch": 6.64,
      "learning_rate": 3.931100478468899e-06,
      "loss": 0.5245,
      "step": 73
    },
    {
      "epoch": 6.73,
      "learning_rate": 3.927272727272727e-06,
      "loss": 0.4984,
      "step": 74
    },
    {
      "epoch": 6.82,
      "learning_rate": 3.923444976076555e-06,
      "loss": 0.483,
      "step": 75
    },
    {
      "epoch": 6.91,
      "learning_rate": 3.919617224880383e-06,
      "loss": 0.5161,
      "step": 76
    },
    {
      "epoch": 7.0,
      "learning_rate": 3.9157894736842104e-06,
      "loss": 0.4869,
      "step": 77
    },
    {
      "epoch": 7.09,
      "learning_rate": 3.911961722488038e-06,
      "loss": 0.4928,
      "step": 78
    },
    {
      "epoch": 7.18,
      "learning_rate": 3.908133971291866e-06,
      "loss": 0.4828,
      "step": 79
    },
    {
      "epoch": 7.27,
      "learning_rate": 3.904306220095694e-06,
      "loss": 0.4656,
      "step": 80
    },
    {
      "epoch": 7.27,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.999869403337525,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.970334864374417,
      "eval_loss": 0.32469046115875244,
      "eval_mean_accuracy": 0.4999347016687625,
      "eval_mean_iou": 0.32344495479147234,
      "eval_overall_accuracy": 0.970334864374417,
      "eval_runtime": 11.3252,
      "eval_samples_per_second": 5.916,
      "eval_steps_per_second": 0.441,
      "step": 80
    },
    {
      "epoch": 7.36,
      "learning_rate": 3.9004784688995215e-06,
      "loss": 0.4827,
      "step": 81
    },
    {
      "epoch": 7.45,
      "learning_rate": 3.896650717703349e-06,
      "loss": 0.4507,
      "step": 82
    },
    {
      "epoch": 7.55,
      "learning_rate": 3.892822966507177e-06,
      "loss": 0.4652,
      "step": 83
    },
    {
      "epoch": 7.64,
      "learning_rate": 3.888995215311005e-06,
      "loss": 0.4509,
      "step": 84
    },
    {
      "epoch": 7.73,
      "learning_rate": 3.8851674641148325e-06,
      "loss": 0.4557,
      "step": 85
    },
    {
      "epoch": 7.82,
      "learning_rate": 3.88133971291866e-06,
      "loss": 0.4792,
      "step": 86
    },
    {
      "epoch": 7.91,
      "learning_rate": 3.877511961722488e-06,
      "loss": 0.4415,
      "step": 87
    },
    {
      "epoch": 8.0,
      "learning_rate": 3.873684210526316e-06,
      "loss": 0.4496,
      "step": 88
    },
    {
      "epoch": 8.09,
      "learning_rate": 3.8698564593301435e-06,
      "loss": 0.4443,
      "step": 89
    },
    {
      "epoch": 8.18,
      "learning_rate": 3.866028708133971e-06,
      "loss": 0.4227,
      "step": 90
    },
    {
      "epoch": 8.18,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9997710158249596,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9702393830712162,
      "eval_loss": 0.3170846998691559,
      "eval_mean_accuracy": 0.4998855079124798,
      "eval_mean_iou": 0.32341312769040537,
      "eval_overall_accuracy": 0.9702393830712162,
      "eval_runtime": 10.5487,
      "eval_samples_per_second": 6.351,
      "eval_steps_per_second": 0.474,
      "step": 90
    },
    {
      "epoch": 8.27,
      "learning_rate": 3.862200956937799e-06,
      "loss": 0.4439,
      "step": 91
    },
    {
      "epoch": 8.36,
      "learning_rate": 3.858373205741627e-06,
      "loss": 0.462,
      "step": 92
    },
    {
      "epoch": 8.45,
      "learning_rate": 3.8545454545454545e-06,
      "loss": 0.4303,
      "step": 93
    },
    {
      "epoch": 8.55,
      "learning_rate": 3.850717703349282e-06,
      "loss": 0.4348,
      "step": 94
    },
    {
      "epoch": 8.64,
      "learning_rate": 3.84688995215311e-06,
      "loss": 0.4616,
      "step": 95
    },
    {
      "epoch": 8.73,
      "learning_rate": 3.843062200956938e-06,
      "loss": 0.4138,
      "step": 96
    },
    {
      "epoch": 8.82,
      "learning_rate": 3.8392344497607655e-06,
      "loss": 0.4202,
      "step": 97
    },
    {
      "epoch": 8.91,
      "learning_rate": 3.835406698564593e-06,
      "loss": 0.3929,
      "step": 98
    },
    {
      "epoch": 9.0,
      "learning_rate": 3.831578947368421e-06,
      "loss": 0.399,
      "step": 99
    },
    {
      "epoch": 9.09,
      "learning_rate": 3.827751196172249e-06,
      "loss": 0.3898,
      "step": 100
    },
    {
      "epoch": 9.09,
      "eval_accuracy_safe": 0.00038357600780259137,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9996028711553041,
      "eval_iou_safe": 0.0003819299168198865,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9700920569815766,
      "eval_loss": 0.3121778070926666,
      "eval_mean_accuracy": 0.49999322358155335,
      "eval_mean_iou": 0.32349132896613214,
      "eval_overall_accuracy": 0.9700875353457322,
      "eval_runtime": 10.9682,
      "eval_samples_per_second": 6.109,
      "eval_steps_per_second": 0.456,
      "step": 100
    },
    {
      "epoch": 9.18,
      "learning_rate": 3.8239234449760766e-06,
      "loss": 0.3893,
      "step": 101
    },
    {
      "epoch": 9.27,
      "learning_rate": 3.820095693779904e-06,
      "loss": 0.4505,
      "step": 102
    },
    {
      "epoch": 9.36,
      "learning_rate": 3.816267942583732e-06,
      "loss": 0.4265,
      "step": 103
    },
    {
      "epoch": 9.45,
      "learning_rate": 3.8124401913875594e-06,
      "loss": 0.403,
      "step": 104
    },
    {
      "epoch": 9.55,
      "learning_rate": 3.808612440191387e-06,
      "loss": 0.3879,
      "step": 105
    },
    {
      "epoch": 9.64,
      "learning_rate": 3.804784688995215e-06,
      "loss": 0.405,
      "step": 106
    },
    {
      "epoch": 9.73,
      "learning_rate": 3.800956937799043e-06,
      "loss": 0.3982,
      "step": 107
    },
    {
      "epoch": 9.82,
      "learning_rate": 3.797129186602871e-06,
      "loss": 0.3909,
      "step": 108
    },
    {
      "epoch": 9.91,
      "learning_rate": 3.7933014354066986e-06,
      "loss": 0.3877,
      "step": 109
    },
    {
      "epoch": 10.0,
      "learning_rate": 3.789473684210526e-06,
      "loss": 0.3513,
      "step": 110
    },
    {
      "epoch": 10.0,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9998844812091585,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9703494968698986,
      "eval_loss": 0.2876318097114563,
      "eval_mean_accuracy": 0.49994224060457926,
      "eval_mean_iou": 0.3234498322899662,
      "eval_overall_accuracy": 0.9703494968698986,
      "eval_runtime": 10.9657,
      "eval_samples_per_second": 6.11,
      "eval_steps_per_second": 0.456,
      "step": 110
    },
    {
      "epoch": 10.09,
      "learning_rate": 3.7856459330143537e-06,
      "loss": 0.4009,
      "step": 111
    },
    {
      "epoch": 10.18,
      "learning_rate": 3.7818181818181815e-06,
      "loss": 0.3629,
      "step": 112
    },
    {
      "epoch": 10.27,
      "learning_rate": 3.777990430622009e-06,
      "loss": 0.3697,
      "step": 113
    },
    {
      "epoch": 10.36,
      "learning_rate": 3.7741626794258374e-06,
      "loss": 0.3489,
      "step": 114
    },
    {
      "epoch": 10.45,
      "learning_rate": 3.770334928229665e-06,
      "loss": 0.369,
      "step": 115
    },
    {
      "epoch": 10.55,
      "learning_rate": 3.766507177033493e-06,
      "loss": 0.3628,
      "step": 116
    },
    {
      "epoch": 10.64,
      "learning_rate": 3.7626794258373202e-06,
      "loss": 0.365,
      "step": 117
    },
    {
      "epoch": 10.73,
      "learning_rate": 3.758851674641148e-06,
      "loss": 0.3425,
      "step": 118
    },
    {
      "epoch": 10.82,
      "learning_rate": 3.7550239234449757e-06,
      "loss": 0.3448,
      "step": 119
    },
    {
      "epoch": 10.91,
      "learning_rate": 3.7511961722488035e-06,
      "loss": 0.4157,
      "step": 120
    },
    {
      "epoch": 10.91,
      "eval_accuracy_safe": 1.9275176271487003e-06,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.999816777458711,
      "eval_iou_safe": 1.924298112263552e-06,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.970290201310306,
      "eval_loss": 0.2819870412349701,
      "eval_mean_accuracy": 0.4999093524881691,
      "eval_mean_iou": 0.3234307085361394,
      "eval_overall_accuracy": 0.9702838499154618,
      "eval_runtime": 10.472,
      "eval_samples_per_second": 6.398,
      "eval_steps_per_second": 0.477,
      "step": 120
    },
    {
      "epoch": 11.0,
      "learning_rate": 3.7473684210526317e-06,
      "loss": 0.3494,
      "step": 121
    },
    {
      "epoch": 11.09,
      "learning_rate": 3.7435406698564594e-06,
      "loss": 0.3522,
      "step": 122
    },
    {
      "epoch": 11.18,
      "learning_rate": 3.7397129186602868e-06,
      "loss": 0.3473,
      "step": 123
    },
    {
      "epoch": 11.27,
      "learning_rate": 3.7358851674641145e-06,
      "loss": 0.3539,
      "step": 124
    },
    {
      "epoch": 11.36,
      "learning_rate": 3.7320574162679423e-06,
      "loss": 0.3872,
      "step": 125
    },
    {
      "epoch": 11.45,
      "learning_rate": 3.72822966507177e-06,
      "loss": 0.3386,
      "step": 126
    },
    {
      "epoch": 11.55,
      "learning_rate": 3.7244019138755978e-06,
      "loss": 0.347,
      "step": 127
    },
    {
      "epoch": 11.64,
      "learning_rate": 3.720574162679426e-06,
      "loss": 0.356,
      "step": 128
    },
    {
      "epoch": 11.73,
      "learning_rate": 3.7167464114832537e-06,
      "loss": 0.3492,
      "step": 129
    },
    {
      "epoch": 11.82,
      "learning_rate": 3.712918660287081e-06,
      "loss": 0.3317,
      "step": 130
    },
    {
      "epoch": 11.82,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9998623044174174,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9703347152502294,
      "eval_loss": 0.26932939887046814,
      "eval_mean_accuracy": 0.4999311522087087,
      "eval_mean_iou": 0.3234449050834098,
      "eval_overall_accuracy": 0.9703279751450268,
      "eval_runtime": 11.1056,
      "eval_samples_per_second": 6.033,
      "eval_steps_per_second": 0.45,
      "step": 130
    },
    {
      "epoch": 11.91,
      "learning_rate": 3.709090909090909e-06,
      "loss": 0.3109,
      "step": 131
    },
    {
      "epoch": 12.0,
      "learning_rate": 3.7052631578947366e-06,
      "loss": 0.3266,
      "step": 132
    },
    {
      "epoch": 12.09,
      "learning_rate": 3.7014354066985643e-06,
      "loss": 0.3358,
      "step": 133
    },
    {
      "epoch": 12.18,
      "learning_rate": 3.6976076555023925e-06,
      "loss": 0.3508,
      "step": 134
    },
    {
      "epoch": 12.27,
      "learning_rate": 3.6937799043062203e-06,
      "loss": 0.3609,
      "step": 135
    },
    {
      "epoch": 12.36,
      "learning_rate": 3.6899521531100476e-06,
      "loss": 0.3139,
      "step": 136
    },
    {
      "epoch": 12.45,
      "learning_rate": 3.6861244019138753e-06,
      "loss": 0.3304,
      "step": 137
    },
    {
      "epoch": 12.55,
      "learning_rate": 3.682296650717703e-06,
      "loss": 0.3292,
      "step": 138
    },
    {
      "epoch": 12.64,
      "learning_rate": 3.678468899521531e-06,
      "loss": 0.3122,
      "step": 139
    },
    {
      "epoch": 12.73,
      "learning_rate": 3.6746411483253586e-06,
      "loss": 0.321,
      "step": 140
    },
    {
      "epoch": 12.73,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9998858892594278,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.970356332880103,
      "eval_loss": 0.2647153437137604,
      "eval_mean_accuracy": 0.4999429446297139,
      "eval_mean_iou": 0.32345211096003434,
      "eval_overall_accuracy": 0.9703508633286205,
      "eval_runtime": 11.6782,
      "eval_samples_per_second": 5.737,
      "eval_steps_per_second": 0.428,
      "step": 140
    },
    {
      "epoch": 12.82,
      "learning_rate": 3.6708133971291868e-06,
      "loss": 0.3236,
      "step": 141
    },
    {
      "epoch": 12.91,
      "learning_rate": 3.6669856459330145e-06,
      "loss": 0.3252,
      "step": 142
    },
    {
      "epoch": 13.0,
      "learning_rate": 3.663157894736842e-06,
      "loss": 0.3171,
      "step": 143
    },
    {
      "epoch": 13.09,
      "learning_rate": 3.6593301435406696e-06,
      "loss": 0.3249,
      "step": 144
    },
    {
      "epoch": 13.18,
      "learning_rate": 3.6555023923444974e-06,
      "loss": 0.3035,
      "step": 145
    },
    {
      "epoch": 13.27,
      "learning_rate": 3.651674641148325e-06,
      "loss": 0.3212,
      "step": 146
    },
    {
      "epoch": 13.36,
      "learning_rate": 3.647846889952153e-06,
      "loss": 0.3009,
      "step": 147
    },
    {
      "epoch": 13.45,
      "learning_rate": 3.644019138755981e-06,
      "loss": 0.3335,
      "step": 148
    },
    {
      "epoch": 13.55,
      "learning_rate": 3.6401913875598084e-06,
      "loss": 0.2966,
      "step": 149
    },
    {
      "epoch": 13.64,
      "learning_rate": 3.636363636363636e-06,
      "loss": 0.2887,
      "step": 150
    },
    {
      "epoch": 13.64,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9999037832315998,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9703687812285815,
      "eval_loss": 0.2538875639438629,
      "eval_mean_accuracy": 0.4999518916157999,
      "eval_mean_iou": 0.3234562604095272,
      "eval_overall_accuracy": 0.9703682287415462,
      "eval_runtime": 10.4703,
      "eval_samples_per_second": 6.399,
      "eval_steps_per_second": 0.478,
      "step": 150
    },
    {
      "epoch": 13.73,
      "learning_rate": 3.632535885167464e-06,
      "loss": 0.3162,
      "step": 151
    },
    {
      "epoch": 13.82,
      "learning_rate": 3.6287081339712917e-06,
      "loss": 0.3097,
      "step": 152
    },
    {
      "epoch": 13.91,
      "learning_rate": 3.6248803827751194e-06,
      "loss": 0.3177,
      "step": 153
    },
    {
      "epoch": 14.0,
      "learning_rate": 3.621052631578947e-06,
      "loss": 0.2942,
      "step": 154
    },
    {
      "epoch": 14.09,
      "learning_rate": 3.617224880382775e-06,
      "loss": 0.2976,
      "step": 155
    },
    {
      "epoch": 14.18,
      "learning_rate": 3.6133971291866027e-06,
      "loss": 0.2883,
      "step": 156
    },
    {
      "epoch": 14.27,
      "learning_rate": 3.6095693779904304e-06,
      "loss": 0.2901,
      "step": 157
    },
    {
      "epoch": 14.36,
      "learning_rate": 3.605741626794258e-06,
      "loss": 0.2725,
      "step": 158
    },
    {
      "epoch": 14.45,
      "learning_rate": 3.601913875598086e-06,
      "loss": 0.2876,
      "step": 159
    },
    {
      "epoch": 14.55,
      "learning_rate": 3.5980861244019137e-06,
      "loss": 0.3008,
      "step": 160
    },
    {
      "epoch": 14.55,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9999237892791757,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9703876438425548,
      "eval_loss": 0.2536369860172272,
      "eval_mean_accuracy": 0.49996189463958784,
      "eval_mean_iou": 0.3234625479475183,
      "eval_overall_accuracy": 0.9703876438425548,
      "eval_runtime": 11.5733,
      "eval_samples_per_second": 5.789,
      "eval_steps_per_second": 0.432,
      "step": 160
    },
    {
      "epoch": 14.64,
      "learning_rate": 3.5942583732057415e-06,
      "loss": 0.3092,
      "step": 161
    },
    {
      "epoch": 14.73,
      "learning_rate": 3.5904306220095692e-06,
      "loss": 0.2576,
      "step": 162
    },
    {
      "epoch": 14.82,
      "learning_rate": 3.586602870813397e-06,
      "loss": 0.2861,
      "step": 163
    },
    {
      "epoch": 14.91,
      "learning_rate": 3.5827751196172247e-06,
      "loss": 0.3094,
      "step": 164
    },
    {
      "epoch": 15.0,
      "learning_rate": 3.5789473684210525e-06,
      "loss": 0.2665,
      "step": 165
    },
    {
      "epoch": 15.09,
      "learning_rate": 3.5751196172248802e-06,
      "loss": 0.2972,
      "step": 166
    },
    {
      "epoch": 15.18,
      "learning_rate": 3.571291866028708e-06,
      "loss": 0.2658,
      "step": 167
    },
    {
      "epoch": 15.27,
      "learning_rate": 3.5674641148325353e-06,
      "loss": 0.2905,
      "step": 168
    },
    {
      "epoch": 15.36,
      "learning_rate": 3.5636363636363635e-06,
      "loss": 0.2572,
      "step": 169
    },
    {
      "epoch": 15.45,
      "learning_rate": 3.5598086124401913e-06,
      "loss": 0.2853,
      "step": 170
    },
    {
      "epoch": 15.45,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.999935112350091,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.970398632448111,
      "eval_loss": 0.23974378407001495,
      "eval_mean_accuracy": 0.4999675561750455,
      "eval_mean_iou": 0.323466210816037,
      "eval_overall_accuracy": 0.970398632448111,
      "eval_runtime": 10.444,
      "eval_samples_per_second": 6.415,
      "eval_steps_per_second": 0.479,
      "step": 170
    },
    {
      "epoch": 15.55,
      "learning_rate": 3.555980861244019e-06,
      "loss": 0.2842,
      "step": 171
    },
    {
      "epoch": 15.64,
      "learning_rate": 3.5521531100478468e-06,
      "loss": 0.2953,
      "step": 172
    },
    {
      "epoch": 15.73,
      "learning_rate": 3.5483253588516745e-06,
      "loss": 0.2651,
      "step": 173
    },
    {
      "epoch": 15.82,
      "learning_rate": 3.5444976076555023e-06,
      "loss": 0.2886,
      "step": 174
    },
    {
      "epoch": 15.91,
      "learning_rate": 3.5406698564593296e-06,
      "loss": 0.2602,
      "step": 175
    },
    {
      "epoch": 16.0,
      "learning_rate": 3.536842105263158e-06,
      "loss": 0.2527,
      "step": 176
    },
    {
      "epoch": 16.09,
      "learning_rate": 3.5330143540669856e-06,
      "loss": 0.2378,
      "step": 177
    },
    {
      "epoch": 16.18,
      "learning_rate": 3.5291866028708133e-06,
      "loss": 0.2776,
      "step": 178
    },
    {
      "epoch": 16.27,
      "learning_rate": 3.525358851674641e-06,
      "loss": 0.2698,
      "step": 179
    },
    {
      "epoch": 16.36,
      "learning_rate": 3.521531100478469e-06,
      "loss": 0.2684,
      "step": 180
    },
    {
      "epoch": 16.36,
      "eval_accuracy_safe": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9999444406831249,
      "eval_iou_safe": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9704076852371444,
      "eval_loss": 0.2320707142353058,
      "eval_mean_accuracy": 0.49997222034156247,
      "eval_mean_iou": 0.32346922841238146,
      "eval_overall_accuracy": 0.9704076852371444,
      "eval_runtime": 10.8231,
      "eval_samples_per_second": 6.19,
      "eval_steps_per_second": 0.462,
      "step": 180
    },
    {
      "epoch": 16.45,
      "learning_rate": 3.517703349282296e-06,
      "loss": 0.2772,
      "step": 181
    },
    {
      "epoch": 16.55,
      "learning_rate": 3.513875598086124e-06,
      "loss": 0.2784,
      "step": 182
    },
    {
      "epoch": 16.64,
      "learning_rate": 3.510047846889952e-06,
      "loss": 0.2508,
      "step": 183
    },
    {
      "epoch": 16.73,
      "learning_rate": 3.50622009569378e-06,
      "loss": 0.2542,
      "step": 184
    },
    {
      "epoch": 16.82,
      "learning_rate": 3.5023923444976076e-06,
      "loss": 0.2578,
      "step": 185
    },
    {
      "epoch": 16.91,
      "learning_rate": 3.4985645933014354e-06,
      "loss": 0.2542,
      "step": 186
    },
    {
      "epoch": 17.0,
      "learning_rate": 3.494736842105263e-06,
      "loss": 0.2606,
      "step": 187
    },
    {
      "epoch": 17.09,
      "learning_rate": 3.4909090909090904e-06,
      "loss": 0.2607,
      "step": 188
    },
    {
      "epoch": 17.18,
      "learning_rate": 3.487081339712918e-06,
      "loss": 0.2417,
      "step": 189
    },
    {
      "epoch": 17.27,
      "learning_rate": 3.4832535885167464e-06,
      "loss": 0.2585,
      "step": 190
    },
    {
      "epoch": 17.27,
      "eval_accuracy_safe": 7.710070508594801e-06,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9999450860394984,
      "eval_iou_safe": 7.69974090371859e-06,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9704085325345925,
      "eval_loss": 0.2208096832036972,
      "eval_mean_accuracy": 0.4999763980550035,
      "eval_mean_iou": 0.3234720774251654,
      "eval_overall_accuracy": 0.9704085392738456,
      "eval_runtime": 11.7648,
      "eval_samples_per_second": 5.695,
      "eval_steps_per_second": 0.425,
      "step": 190
    },
    {
      "epoch": 17.36,
      "learning_rate": 3.479425837320574e-06,
      "loss": 0.2631,
      "step": 191
    },
    {
      "epoch": 17.45,
      "learning_rate": 3.475598086124402e-06,
      "loss": 0.2473,
      "step": 192
    },
    {
      "epoch": 17.55,
      "learning_rate": 3.4717703349282296e-06,
      "loss": 0.2549,
      "step": 193
    },
    {
      "epoch": 17.64,
      "learning_rate": 3.467942583732057e-06,
      "loss": 0.2364,
      "step": 194
    },
    {
      "epoch": 17.73,
      "learning_rate": 3.4641148325358847e-06,
      "loss": 0.2363,
      "step": 195
    },
    {
      "epoch": 17.82,
      "learning_rate": 3.460287081339713e-06,
      "loss": 0.2579,
      "step": 196
    },
    {
      "epoch": 17.91,
      "learning_rate": 3.4564593301435407e-06,
      "loss": 0.2345,
      "step": 197
    },
    {
      "epoch": 18.0,
      "learning_rate": 3.4526315789473684e-06,
      "loss": 0.2844,
      "step": 198
    },
    {
      "epoch": 18.09,
      "learning_rate": 3.448803827751196e-06,
      "loss": 0.2902,
      "step": 199
    },
    {
      "epoch": 18.18,
      "learning_rate": 3.444976076555024e-06,
      "loss": 0.2088,
      "step": 200
    },
    {
      "epoch": 18.18,
      "eval_accuracy_safe": 0.008429034583521266,
      "eval_accuracy_unlabeled": NaN,
      "eval_accuracy_unsafe": 0.9996887035529685,
      "eval_iou_safe": 0.00834470322205154,
      "eval_iou_unlabeled": 0.0,
      "eval_iou_unsafe": 0.9704011128022085,
      "eval_loss": 0.20112648606300354,
      "eval_mean_accuracy": 0.5040588690682448,
      "eval_mean_iou": 0.32624860534142003,
      "eval_overall_accuracy": 0.9704084823380655,
      "eval_runtime": 10.8546,
      "eval_samples_per_second": 6.173,
      "eval_steps_per_second": 0.461,
      "step": 200
    }
  ],
  "max_steps": 1100,
  "num_train_epochs": 100,
  "total_flos": 2.2372106925370245e+18,
  "trial_name": null,
  "trial_params": null
}