diff --git "a/checkpoint-1120/trainer_state.json" "b/checkpoint-1120/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-1120/trainer_state.json"
@@ -0,0 +1,8640 @@
+{
+  "best_metric": 0.08840107917785645,
+  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-train-SF-RGBD-b5/checkpoint-1080",
+  "epoch": 101.81818181818181,
+  "global_step": 1120,
+  "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
+    },
+    {
+      "epoch": 18.27,
+      "learning_rate": 3.4411483253588513e-06,
+      "loss": 0.2356,
+      "step": 201
+    },
+    {
+      "epoch": 18.36,
+      "learning_rate": 3.437320574162679e-06,
+      "loss": 0.234,
+      "step": 202
+    },
+    {
+      "epoch": 18.45,
+      "learning_rate": 3.433492822966507e-06,
+      "loss": 0.2314,
+      "step": 203
+    },
+    {
+      "epoch": 18.55,
+      "learning_rate": 3.429665071770335e-06,
+      "loss": 0.2514,
+      "step": 204
+    },
+    {
+      "epoch": 18.64,
+      "learning_rate": 3.4258373205741627e-06,
+      "loss": 0.2345,
+      "step": 205
+    },
+    {
+      "epoch": 18.73,
+      "learning_rate": 3.4220095693779905e-06,
+      "loss": 0.223,
+      "step": 206
+    },
+    {
+      "epoch": 18.82,
+      "learning_rate": 3.418181818181818e-06,
+      "loss": 0.2321,
+      "step": 207
+    },
+    {
+      "epoch": 18.91,
+      "learning_rate": 3.4143540669856456e-06,
+      "loss": 0.2277,
+      "step": 208
+    },
+    {
+      "epoch": 19.0,
+      "learning_rate": 3.4105263157894733e-06,
+      "loss": 0.2287,
+      "step": 209
+    },
+    {
+      "epoch": 19.09,
+      "learning_rate": 3.4066985645933015e-06,
+      "loss": 0.2518,
+      "step": 210
+    },
+    {
+      "epoch": 19.09,
+      "eval_accuracy_safe": 0.04677507025801751,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9988634100888913,
+      "eval_iou_safe": 0.04509203479967148,
+      "eval_iou_unlabeled": 0.0,
+      "eval_iou_unsafe": 0.9706997641478909,
+      "eval_loss": 0.2025505006313324,
+      "eval_mean_accuracy": 0.5228192401734544,
+      "eval_mean_iou": 0.33859726631585413,
+      "eval_overall_accuracy": 0.9707402471286147,
+      "eval_runtime": 11.7249,
+      "eval_samples_per_second": 5.714,
+      "eval_steps_per_second": 0.426,
+      "step": 210
+    },
+    {
+      "epoch": 19.18,
+      "learning_rate": 3.4028708133971292e-06,
+      "loss": 0.2585,
+      "step": 211
+    },
+    {
+      "epoch": 19.27,
+      "learning_rate": 3.399043062200957e-06,
+      "loss": 0.2251,
+      "step": 212
+    },
+    {
+      "epoch": 19.36,
+      "learning_rate": 3.3952153110047848e-06,
+      "loss": 0.2321,
+      "step": 213
+    },
+    {
+      "epoch": 19.45,
+      "learning_rate": 3.391387559808612e-06,
+      "loss": 0.2208,
+      "step": 214
+    },
+    {
+      "epoch": 19.55,
+      "learning_rate": 3.38755980861244e-06,
+      "loss": 0.2056,
+      "step": 215
+    },
+    {
+      "epoch": 19.64,
+      "learning_rate": 3.3837320574162676e-06,
+      "loss": 0.2267,
+      "step": 216
+    },
+    {
+      "epoch": 19.73,
+      "learning_rate": 3.3799043062200958e-06,
+      "loss": 0.2294,
+      "step": 217
+    },
+    {
+      "epoch": 19.82,
+      "learning_rate": 3.3760765550239235e-06,
+      "loss": 0.1943,
+      "step": 218
+    },
+    {
+      "epoch": 19.91,
+      "learning_rate": 3.3722488038277513e-06,
+      "loss": 0.2065,
+      "step": 219
+    },
+    {
+      "epoch": 20.0,
+      "learning_rate": 3.3684210526315786e-06,
+      "loss": 0.218,
+      "step": 220
+    },
+    {
+      "epoch": 20.0,
+      "eval_accuracy_safe": 0.08791986152713367,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9983610294865674,
+      "eval_iou_safe": 0.08342752835910436,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9713937664812292,
+      "eval_loss": 0.18889041244983673,
+      "eval_mean_accuracy": 0.5431404455068505,
+      "eval_mean_iou": 0.5274106474201667,
+      "eval_overall_accuracy": 0.971468057205428,
+      "eval_runtime": 10.4226,
+      "eval_samples_per_second": 6.428,
+      "eval_steps_per_second": 0.48,
+      "step": 220
+    },
+    {
+      "epoch": 20.09,
+      "learning_rate": 3.3645933014354064e-06,
+      "loss": 0.268,
+      "step": 221
+    },
+    {
+      "epoch": 20.18,
+      "learning_rate": 3.360765550239234e-06,
+      "loss": 0.2117,
+      "step": 222
+    },
+    {
+      "epoch": 20.27,
+      "learning_rate": 3.356937799043062e-06,
+      "loss": 0.1985,
+      "step": 223
+    },
+    {
+      "epoch": 20.36,
+      "learning_rate": 3.35311004784689e-06,
+      "loss": 0.1991,
+      "step": 224
+    },
+    {
+      "epoch": 20.45,
+      "learning_rate": 3.349282296650718e-06,
+      "loss": 0.2118,
+      "step": 225
+    },
+    {
+      "epoch": 20.55,
+      "learning_rate": 3.345454545454545e-06,
+      "loss": 0.2384,
+      "step": 226
+    },
+    {
+      "epoch": 20.64,
+      "learning_rate": 3.341626794258373e-06,
+      "loss": 0.2055,
+      "step": 227
+    },
+    {
+      "epoch": 20.73,
+      "learning_rate": 3.3377990430622007e-06,
+      "loss": 0.2158,
+      "step": 228
+    },
+    {
+      "epoch": 20.82,
+      "learning_rate": 3.3339712918660284e-06,
+      "loss": 0.2031,
+      "step": 229
+    },
+    {
+      "epoch": 20.91,
+      "learning_rate": 3.330143540669856e-06,
+      "loss": 0.2046,
+      "step": 230
+    },
+    {
+      "epoch": 20.91,
+      "eval_accuracy_safe": 0.19307751319385816,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9968958358438674,
+      "eval_iou_safe": 0.17520884641217957,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9729983355550175,
+      "eval_loss": 0.18470104038715363,
+      "eval_mean_accuracy": 0.5949866745188628,
+      "eval_mean_iou": 0.5741035909835985,
+      "eval_overall_accuracy": 0.9731523314518715,
+      "eval_runtime": 11.9598,
+      "eval_samples_per_second": 5.602,
+      "eval_steps_per_second": 0.418,
+      "step": 230
+    },
+    {
+      "epoch": 21.0,
+      "learning_rate": 3.3263157894736843e-06,
+      "loss": 0.2316,
+      "step": 231
+    },
+    {
+      "epoch": 21.09,
+      "learning_rate": 3.322488038277512e-06,
+      "loss": 0.1925,
+      "step": 232
+    },
+    {
+      "epoch": 21.18,
+      "learning_rate": 3.3186602870813394e-06,
+      "loss": 0.2185,
+      "step": 233
+    },
+    {
+      "epoch": 21.27,
+      "learning_rate": 3.314832535885167e-06,
+      "loss": 0.2268,
+      "step": 234
+    },
+    {
+      "epoch": 21.36,
+      "learning_rate": 3.311004784688995e-06,
+      "loss": 0.2062,
+      "step": 235
+    },
+    {
+      "epoch": 21.45,
+      "learning_rate": 3.3071770334928227e-06,
+      "loss": 0.2012,
+      "step": 236
+    },
+    {
+      "epoch": 21.55,
+      "learning_rate": 3.3033492822966505e-06,
+      "loss": 0.2023,
+      "step": 237
+    },
+    {
+      "epoch": 21.64,
+      "learning_rate": 3.2995215311004786e-06,
+      "loss": 0.1949,
+      "step": 238
+    },
+    {
+      "epoch": 21.73,
+      "learning_rate": 3.295693779904306e-06,
+      "loss": 0.2129,
+      "step": 239
+    },
+    {
+      "epoch": 21.82,
+      "learning_rate": 3.2918660287081337e-06,
+      "loss": 0.2147,
+      "step": 240
+    },
+    {
+      "epoch": 21.82,
+      "eval_accuracy_safe": 0.20418001472623468,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9968364630575131,
+      "eval_iou_safe": 0.18495645371941807,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9732613980096713,
+      "eval_loss": 0.1766258329153061,
+      "eval_mean_accuracy": 0.6005082388918739,
+      "eval_mean_iou": 0.5791089258645447,
+      "eval_overall_accuracy": 0.9734226625357101,
+      "eval_runtime": 10.3683,
+      "eval_samples_per_second": 6.462,
+      "eval_steps_per_second": 0.482,
+      "step": 240
+    },
+    {
+      "epoch": 21.91,
+      "learning_rate": 3.2880382775119615e-06,
+      "loss": 0.1957,
+      "step": 241
+    },
+    {
+      "epoch": 22.0,
+      "learning_rate": 3.2842105263157892e-06,
+      "loss": 0.1871,
+      "step": 242
+    },
+    {
+      "epoch": 22.09,
+      "learning_rate": 3.280382775119617e-06,
+      "loss": 0.2137,
+      "step": 243
+    },
+    {
+      "epoch": 22.18,
+      "learning_rate": 3.276555023923445e-06,
+      "loss": 0.1965,
+      "step": 244
+    },
+    {
+      "epoch": 22.27,
+      "learning_rate": 3.272727272727273e-06,
+      "loss": 0.2014,
+      "step": 245
+    },
+    {
+      "epoch": 22.36,
+      "learning_rate": 3.2688995215311003e-06,
+      "loss": 0.1888,
+      "step": 246
+    },
+    {
+      "epoch": 22.45,
+      "learning_rate": 3.265071770334928e-06,
+      "loss": 0.2214,
+      "step": 247
+    },
+    {
+      "epoch": 22.55,
+      "learning_rate": 3.2612440191387558e-06,
+      "loss": 0.1947,
+      "step": 248
+    },
+    {
+      "epoch": 22.64,
+      "learning_rate": 3.2574162679425835e-06,
+      "loss": 0.1854,
+      "step": 249
+    },
+    {
+      "epoch": 22.73,
+      "learning_rate": 3.2535885167464113e-06,
+      "loss": 0.188,
+      "step": 250
+    },
+    {
+      "epoch": 22.73,
+      "eval_accuracy_safe": 0.20203854264247245,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.997185483518009,
+      "eval_iou_safe": 0.18493758546160294,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9735402085935885,
+      "eval_loss": 0.17257128655910492,
+      "eval_mean_accuracy": 0.5996120130802407,
+      "eval_mean_iou": 0.5792388970275957,
+      "eval_overall_accuracy": 0.9736981178397563,
+      "eval_runtime": 11.9222,
+      "eval_samples_per_second": 5.62,
+      "eval_steps_per_second": 0.419,
+      "step": 250
+    },
+    {
+      "epoch": 22.82,
+      "learning_rate": 3.2497607655502395e-06,
+      "loss": 0.1837,
+      "step": 251
+    },
+    {
+      "epoch": 22.91,
+      "learning_rate": 3.245933014354067e-06,
+      "loss": 0.22,
+      "step": 252
+    },
+    {
+      "epoch": 23.0,
+      "learning_rate": 3.2421052631578945e-06,
+      "loss": 0.2045,
+      "step": 253
+    },
+    {
+      "epoch": 23.09,
+      "learning_rate": 3.2382775119617223e-06,
+      "loss": 0.1979,
+      "step": 254
+    },
+    {
+      "epoch": 23.18,
+      "learning_rate": 3.23444976076555e-06,
+      "loss": 0.2015,
+      "step": 255
+    },
+    {
+      "epoch": 23.27,
+      "learning_rate": 3.230622009569378e-06,
+      "loss": 0.1811,
+      "step": 256
+    },
+    {
+      "epoch": 23.36,
+      "learning_rate": 3.2267942583732056e-06,
+      "loss": 0.1812,
+      "step": 257
+    },
+    {
+      "epoch": 23.45,
+      "learning_rate": 3.2229665071770337e-06,
+      "loss": 0.1955,
+      "step": 258
+    },
+    {
+      "epoch": 23.55,
+      "learning_rate": 3.219138755980861e-06,
+      "loss": 0.1956,
+      "step": 259
+    },
+    {
+      "epoch": 23.64,
+      "learning_rate": 3.215311004784689e-06,
+      "loss": 0.2175,
+      "step": 260
+    },
+    {
+      "epoch": 23.64,
+      "eval_accuracy_safe": 0.1897583278399081,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9973954590144141,
+      "eval_iou_safe": 0.1748006022770048,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9733899999204132,
+      "eval_loss": 0.17063008248806,
+      "eval_mean_accuracy": 0.593576893427161,
+      "eval_mean_iou": 0.574095301098709,
+      "eval_overall_accuracy": 0.9735391531417619,
+      "eval_runtime": 10.3294,
+      "eval_samples_per_second": 6.486,
+      "eval_steps_per_second": 0.484,
+      "step": 260
+    },
+    {
+      "epoch": 23.73,
+      "learning_rate": 3.2114832535885166e-06,
+      "loss": 0.2014,
+      "step": 261
+    },
+    {
+      "epoch": 23.82,
+      "learning_rate": 3.2076555023923443e-06,
+      "loss": 0.1943,
+      "step": 262
+    },
+    {
+      "epoch": 23.91,
+      "learning_rate": 3.203827751196172e-06,
+      "loss": 0.1902,
+      "step": 263
+    },
+    {
+      "epoch": 24.0,
+      "learning_rate": 3.2e-06,
+      "loss": 0.1767,
+      "step": 264
+    },
+    {
+      "epoch": 24.09,
+      "learning_rate": 3.1961722488038276e-06,
+      "loss": 0.1836,
+      "step": 265
+    },
+    {
+      "epoch": 24.18,
+      "learning_rate": 3.1923444976076554e-06,
+      "loss": 0.1745,
+      "step": 266
+    },
+    {
+      "epoch": 24.27,
+      "learning_rate": 3.188516746411483e-06,
+      "loss": 0.1805,
+      "step": 267
+    },
+    {
+      "epoch": 24.36,
+      "learning_rate": 3.184688995215311e-06,
+      "loss": 0.2039,
+      "step": 268
+    },
+    {
+      "epoch": 24.45,
+      "learning_rate": 3.1808612440191386e-06,
+      "loss": 0.1827,
+      "step": 269
+    },
+    {
+      "epoch": 24.55,
+      "learning_rate": 3.1770334928229664e-06,
+      "loss": 0.2059,
+      "step": 270
+    },
+    {
+      "epoch": 24.55,
+      "eval_accuracy_safe": 0.3006272142358742,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9961704552801475,
+      "eval_iou_safe": 0.2670303179723187,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9754068347526109,
+      "eval_loss": 0.16893711686134338,
+      "eval_mean_accuracy": 0.6483988347580109,
+      "eval_mean_iou": 0.6212185763624648,
+      "eval_overall_accuracy": 0.975625223188258,
+      "eval_runtime": 11.3681,
+      "eval_samples_per_second": 5.894,
+      "eval_steps_per_second": 0.44,
+      "step": 270
+    },
+    {
+      "epoch": 24.64,
+      "learning_rate": 3.173205741626794e-06,
+      "loss": 0.1668,
+      "step": 271
+    },
+    {
+      "epoch": 24.73,
+      "learning_rate": 3.169377990430622e-06,
+      "loss": 0.183,
+      "step": 272
+    },
+    {
+      "epoch": 24.82,
+      "learning_rate": 3.1655502392344497e-06,
+      "loss": 0.2111,
+      "step": 273
+    },
+    {
+      "epoch": 24.91,
+      "learning_rate": 3.1617224880382774e-06,
+      "loss": 0.1755,
+      "step": 274
+    },
+    {
+      "epoch": 25.0,
+      "learning_rate": 3.157894736842105e-06,
+      "loss": 0.1962,
+      "step": 275
+    },
+    {
+      "epoch": 25.09,
+      "learning_rate": 3.154066985645933e-06,
+      "loss": 0.1748,
+      "step": 276
+    },
+    {
+      "epoch": 25.18,
+      "learning_rate": 3.1502392344497607e-06,
+      "loss": 0.1813,
+      "step": 277
+    },
+    {
+      "epoch": 25.27,
+      "learning_rate": 3.146411483253588e-06,
+      "loss": 0.1882,
+      "step": 278
+    },
+    {
+      "epoch": 25.36,
+      "learning_rate": 3.142583732057416e-06,
+      "loss": 0.1855,
+      "step": 279
+    },
+    {
+      "epoch": 25.45,
+      "learning_rate": 3.138755980861244e-06,
+      "loss": 0.1776,
+      "step": 280
+    },
+    {
+      "epoch": 25.45,
+      "eval_accuracy_safe": 0.2869803894356614,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9966670276751107,
+      "eval_iou_safe": 0.2586568730325289,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.97549630595009,
+      "eval_loss": 0.16121132671833038,
+      "eval_mean_accuracy": 0.6418237085553861,
+      "eval_mean_iou": 0.6170765894913094,
+      "eval_overall_accuracy": 0.9757040223078941,
+      "eval_runtime": 10.9596,
+      "eval_samples_per_second": 6.113,
+      "eval_steps_per_second": 0.456,
+      "step": 280
+    },
+    {
+      "epoch": 25.55,
+      "learning_rate": 3.1349282296650717e-06,
+      "loss": 0.1784,
+      "step": 281
+    },
+    {
+      "epoch": 25.64,
+      "learning_rate": 3.1311004784688995e-06,
+      "loss": 0.1672,
+      "step": 282
+    },
+    {
+      "epoch": 25.73,
+      "learning_rate": 3.127272727272727e-06,
+      "loss": 0.1759,
+      "step": 283
+    },
+    {
+      "epoch": 25.82,
+      "learning_rate": 3.1234449760765545e-06,
+      "loss": 0.162,
+      "step": 284
+    },
+    {
+      "epoch": 25.91,
+      "learning_rate": 3.1196172248803823e-06,
+      "loss": 0.1923,
+      "step": 285
+    },
+    {
+      "epoch": 26.0,
+      "learning_rate": 3.1157894736842105e-06,
+      "loss": 0.2386,
+      "step": 286
+    },
+    {
+      "epoch": 26.09,
+      "learning_rate": 3.1119617224880382e-06,
+      "loss": 0.1797,
+      "step": 287
+    },
+    {
+      "epoch": 26.18,
+      "learning_rate": 3.108133971291866e-06,
+      "loss": 0.1789,
+      "step": 288
+    },
+    {
+      "epoch": 26.27,
+      "learning_rate": 3.1043062200956937e-06,
+      "loss": 0.1767,
+      "step": 289
+    },
+    {
+      "epoch": 26.36,
+      "learning_rate": 3.1004784688995215e-06,
+      "loss": 0.1585,
+      "step": 290
+    },
+    {
+      "epoch": 26.36,
+      "eval_accuracy_safe": 0.42542819804087106,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9944028241733601,
+      "eval_iou_safe": 0.35934744914157324,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.977311106799397,
+      "eval_loss": 0.1537064015865326,
+      "eval_mean_accuracy": 0.7099155111071156,
+      "eval_mean_iou": 0.6683292779704851,
+      "eval_overall_accuracy": 0.9775962260232043,
+      "eval_runtime": 11.3177,
+      "eval_samples_per_second": 5.92,
+      "eval_steps_per_second": 0.442,
+      "step": 290
+    },
+    {
+      "epoch": 26.45,
+      "learning_rate": 3.096650717703349e-06,
+      "loss": 0.1764,
+      "step": 291
+    },
+    {
+      "epoch": 26.55,
+      "learning_rate": 3.0928229665071766e-06,
+      "loss": 0.1974,
+      "step": 292
+    },
+    {
+      "epoch": 26.64,
+      "learning_rate": 3.0889952153110048e-06,
+      "loss": 0.1666,
+      "step": 293
+    },
+    {
+      "epoch": 26.73,
+      "learning_rate": 3.0851674641148325e-06,
+      "loss": 0.1899,
+      "step": 294
+    },
+    {
+      "epoch": 26.82,
+      "learning_rate": 3.0813397129186603e-06,
+      "loss": 0.1647,
+      "step": 295
+    },
+    {
+      "epoch": 26.91,
+      "learning_rate": 3.077511961722488e-06,
+      "loss": 0.16,
+      "step": 296
+    },
+    {
+      "epoch": 27.0,
+      "learning_rate": 3.0736842105263154e-06,
+      "loss": 0.1782,
+      "step": 297
+    },
+    {
+      "epoch": 27.09,
+      "learning_rate": 3.069856459330143e-06,
+      "loss": 0.1746,
+      "step": 298
+    },
+    {
+      "epoch": 27.18,
+      "learning_rate": 3.066028708133971e-06,
+      "loss": 0.1644,
+      "step": 299
+    },
+    {
+      "epoch": 27.27,
+      "learning_rate": 3.062200956937799e-06,
+      "loss": 0.1588,
+      "step": 300
+    },
+    {
+      "epoch": 27.27,
+      "eval_accuracy_safe": 0.2797734010277524,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.997013760053919,
+      "eval_iou_safe": 0.25477705888341623,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.975626203425886,
+      "eval_loss": 0.15269163250923157,
+      "eval_mean_accuracy": 0.6383935805408356,
+      "eval_mean_iou": 0.6152016311546511,
+      "eval_overall_accuracy": 0.9758276298864564,
+      "eval_runtime": 10.7571,
+      "eval_samples_per_second": 6.228,
+      "eval_steps_per_second": 0.465,
+      "step": 300
+    },
+    {
+      "epoch": 27.36,
+      "learning_rate": 3.058373205741627e-06,
+      "loss": 0.1755,
+      "step": 301
+    },
+    {
+      "epoch": 27.45,
+      "learning_rate": 3.0545454545454546e-06,
+      "loss": 0.1739,
+      "step": 302
+    },
+    {
+      "epoch": 27.55,
+      "learning_rate": 3.0507177033492823e-06,
+      "loss": 0.164,
+      "step": 303
+    },
+    {
+      "epoch": 27.64,
+      "learning_rate": 3.0468899521531096e-06,
+      "loss": 0.183,
+      "step": 304
+    },
+    {
+      "epoch": 27.73,
+      "learning_rate": 3.0430622009569374e-06,
+      "loss": 0.1602,
+      "step": 305
+    },
+    {
+      "epoch": 27.82,
+      "learning_rate": 3.0392344497607656e-06,
+      "loss": 0.1449,
+      "step": 306
+    },
+    {
+      "epoch": 27.91,
+      "learning_rate": 3.0354066985645933e-06,
+      "loss": 0.1753,
+      "step": 307
+    },
+    {
+      "epoch": 28.0,
+      "learning_rate": 3.031578947368421e-06,
+      "loss": 0.1904,
+      "step": 308
+    },
+    {
+      "epoch": 28.09,
+      "learning_rate": 3.027751196172249e-06,
+      "loss": 0.1588,
+      "step": 309
+    },
+    {
+      "epoch": 28.18,
+      "learning_rate": 3.023923444976076e-06,
+      "loss": 0.153,
+      "step": 310
+    },
+    {
+      "epoch": 28.18,
+      "eval_accuracy_safe": 0.4287859337473641,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.994641840706569,
+      "eval_iou_safe": 0.36460202808904973,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9776442139822791,
+      "eval_loss": 0.14515650272369385,
+      "eval_mean_accuracy": 0.7117138872269666,
+      "eval_mean_iou": 0.6711231210356644,
+      "eval_overall_accuracy": 0.9779273645201726,
+      "eval_runtime": 11.3084,
+      "eval_samples_per_second": 5.925,
+      "eval_steps_per_second": 0.442,
+      "step": 310
+    },
+    {
+      "epoch": 28.27,
+      "learning_rate": 3.020095693779904e-06,
+      "loss": 0.1594,
+      "step": 311
+    },
+    {
+      "epoch": 28.36,
+      "learning_rate": 3.0162679425837317e-06,
+      "loss": 0.1821,
+      "step": 312
+    },
+    {
+      "epoch": 28.45,
+      "learning_rate": 3.01244019138756e-06,
+      "loss": 0.1644,
+      "step": 313
+    },
+    {
+      "epoch": 28.55,
+      "learning_rate": 3.0086124401913876e-06,
+      "loss": 0.1711,
+      "step": 314
+    },
+    {
+      "epoch": 28.64,
+      "learning_rate": 3.0047846889952154e-06,
+      "loss": 0.1571,
+      "step": 315
+    },
+    {
+      "epoch": 28.73,
+      "learning_rate": 3.000956937799043e-06,
+      "loss": 0.1773,
+      "step": 316
+    },
+    {
+      "epoch": 28.82,
+      "learning_rate": 2.9971291866028705e-06,
+      "loss": 0.1596,
+      "step": 317
+    },
+    {
+      "epoch": 28.91,
+      "learning_rate": 2.9933014354066982e-06,
+      "loss": 0.1609,
+      "step": 318
+    },
+    {
+      "epoch": 29.0,
+      "learning_rate": 2.989473684210526e-06,
+      "loss": 0.1647,
+      "step": 319
+    },
+    {
+      "epoch": 29.09,
+      "learning_rate": 2.985645933014354e-06,
+      "loss": 0.1623,
+      "step": 320
+    },
+    {
+      "epoch": 29.09,
+      "eval_accuracy_safe": 0.4401370850536428,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9944793282379906,
+      "eval_iou_safe": 0.3725626161484481,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9778165413193792,
+      "eval_loss": 0.1442144215106964,
+      "eval_mean_accuracy": 0.7173082066458167,
+      "eval_mean_iou": 0.6751895787339137,
+      "eval_overall_accuracy": 0.9781049472182545,
+      "eval_runtime": 11.6309,
+      "eval_samples_per_second": 5.761,
+      "eval_steps_per_second": 0.43,
+      "step": 320
+    },
+    {
+      "epoch": 29.18,
+      "learning_rate": 2.981818181818182e-06,
+      "loss": 0.1641,
+      "step": 321
+    },
+    {
+      "epoch": 29.27,
+      "learning_rate": 2.9779904306220097e-06,
+      "loss": 0.1667,
+      "step": 322
+    },
+    {
+      "epoch": 29.36,
+      "learning_rate": 2.974162679425837e-06,
+      "loss": 0.1811,
+      "step": 323
+    },
+    {
+      "epoch": 29.45,
+      "learning_rate": 2.9703349282296648e-06,
+      "loss": 0.1639,
+      "step": 324
+    },
+    {
+      "epoch": 29.55,
+      "learning_rate": 2.9665071770334925e-06,
+      "loss": 0.1707,
+      "step": 325
+    },
+    {
+      "epoch": 29.64,
+      "learning_rate": 2.9626794258373203e-06,
+      "loss": 0.1458,
+      "step": 326
+    },
+    {
+      "epoch": 29.73,
+      "learning_rate": 2.9588516746411484e-06,
+      "loss": 0.1455,
+      "step": 327
+    },
+    {
+      "epoch": 29.82,
+      "learning_rate": 2.955023923444976e-06,
+      "loss": 0.1499,
+      "step": 328
+    },
+    {
+      "epoch": 29.91,
+      "learning_rate": 2.951196172248804e-06,
+      "loss": 0.1571,
+      "step": 329
+    },
+    {
+      "epoch": 30.0,
+      "learning_rate": 2.9473684210526313e-06,
+      "loss": 0.1603,
+      "step": 330
+    },
+    {
+      "epoch": 30.0,
+      "eval_accuracy_safe": 0.4049618158758062,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9958304698088795,
+      "eval_iou_safe": 0.3561710950360333,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9781153749897787,
+      "eval_loss": 0.14068810641765594,
+      "eval_mean_accuracy": 0.7003961428423429,
+      "eval_mean_iou": 0.6671432350129061,
+      "eval_overall_accuracy": 0.9783771571828358,
+      "eval_runtime": 13.5235,
+      "eval_samples_per_second": 4.954,
+      "eval_steps_per_second": 0.37,
+      "step": 330
+    },
+    {
+      "epoch": 30.09,
+      "learning_rate": 2.943540669856459e-06,
+      "loss": 0.1504,
+      "step": 331
+    },
+    {
+      "epoch": 30.18,
+      "learning_rate": 2.939712918660287e-06,
+      "loss": 0.1434,
+      "step": 332
+    },
+    {
+      "epoch": 30.27,
+      "learning_rate": 2.9358851674641146e-06,
+      "loss": 0.1525,
+      "step": 333
+    },
+    {
+      "epoch": 30.36,
+      "learning_rate": 2.9320574162679427e-06,
+      "loss": 0.1589,
+      "step": 334
+    },
+    {
+      "epoch": 30.45,
+      "learning_rate": 2.9282296650717705e-06,
+      "loss": 0.1517,
+      "step": 335
+    },
+    {
+      "epoch": 30.55,
+      "learning_rate": 2.924401913875598e-06,
+      "loss": 0.1638,
+      "step": 336
+    },
+    {
+      "epoch": 30.64,
+      "learning_rate": 2.9205741626794256e-06,
+      "loss": 0.1892,
+      "step": 337
+    },
+    {
+      "epoch": 30.73,
+      "learning_rate": 2.9167464114832533e-06,
+      "loss": 0.1542,
+      "step": 338
+    },
+    {
+      "epoch": 30.82,
+      "learning_rate": 2.912918660287081e-06,
+      "loss": 0.1422,
+      "step": 339
+    },
+    {
+      "epoch": 30.91,
+      "learning_rate": 2.909090909090909e-06,
+      "loss": 0.1694,
+      "step": 340
+    },
+    {
+      "epoch": 30.91,
+      "eval_accuracy_safe": 0.4584928354169799,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9947577115099778,
+      "eval_iou_safe": 0.39112826314300936,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9786278609893545,
+      "eval_loss": 0.1343463510274887,
+      "eval_mean_accuracy": 0.7266252734634788,
+      "eval_mean_iou": 0.684878062066182,
+      "eval_overall_accuracy": 0.9789173069284923,
+      "eval_runtime": 11.7988,
+      "eval_samples_per_second": 5.679,
+      "eval_steps_per_second": 0.424,
+      "step": 340
+    },
+    {
+      "epoch": 31.0,
+      "learning_rate": 2.905263157894737e-06,
+      "loss": 0.1586,
+      "step": 341
+    },
+    {
+      "epoch": 31.09,
+      "learning_rate": 2.9014354066985648e-06,
+      "loss": 0.1398,
+      "step": 342
+    },
+    {
+      "epoch": 31.18,
+      "learning_rate": 2.897607655502392e-06,
+      "loss": 0.1652,
+      "step": 343
+    },
+    {
+      "epoch": 31.27,
+      "learning_rate": 2.89377990430622e-06,
+      "loss": 0.1538,
+      "step": 344
+    },
+    {
+      "epoch": 31.36,
+      "learning_rate": 2.8899521531100476e-06,
+      "loss": 0.1564,
+      "step": 345
+    },
+    {
+      "epoch": 31.45,
+      "learning_rate": 2.8861244019138754e-06,
+      "loss": 0.1537,
+      "step": 346
+    },
+    {
+      "epoch": 31.55,
+      "learning_rate": 2.882296650717703e-06,
+      "loss": 0.1463,
+      "step": 347
+    },
+    {
+      "epoch": 31.64,
+      "learning_rate": 2.8784688995215313e-06,
+      "loss": 0.1446,
+      "step": 348
+    },
+    {
+      "epoch": 31.73,
+      "learning_rate": 2.8746411483253586e-06,
+      "loss": 0.1421,
+      "step": 349
+    },
+    {
+      "epoch": 31.82,
+      "learning_rate": 2.8708133971291864e-06,
+      "loss": 0.1585,
+      "step": 350
+    },
+    {
+      "epoch": 31.82,
+      "eval_accuracy_safe": 0.38612418610568194,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9962063605620138,
+      "eval_iou_safe": 0.34333223643618116,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9779338359482509,
+      "eval_loss": 0.1352754831314087,
+      "eval_mean_accuracy": 0.6911652733338479,
+      "eval_mean_iou": 0.660633036192216,
+      "eval_overall_accuracy": 0.9781855113470732,
+      "eval_runtime": 10.374,
+      "eval_samples_per_second": 6.458,
+      "eval_steps_per_second": 0.482,
+      "step": 350
+    },
+    {
+      "epoch": 31.91,
+      "learning_rate": 2.866985645933014e-06,
+      "loss": 0.1652,
+      "step": 351
+    },
+    {
+      "epoch": 32.0,
+      "learning_rate": 2.863157894736842e-06,
+      "loss": 0.1985,
+      "step": 352
+    },
+    {
+      "epoch": 32.09,
+      "learning_rate": 2.8593301435406697e-06,
+      "loss": 0.1471,
+      "step": 353
+    },
+    {
+      "epoch": 32.18,
+      "learning_rate": 2.855502392344498e-06,
+      "loss": 0.1474,
+      "step": 354
+    },
+    {
+      "epoch": 32.27,
+      "learning_rate": 2.851674641148325e-06,
+      "loss": 0.1454,
+      "step": 355
+    },
+    {
+      "epoch": 32.36,
+      "learning_rate": 2.847846889952153e-06,
+      "loss": 0.1432,
+      "step": 356
+    },
+    {
+      "epoch": 32.45,
+      "learning_rate": 2.8440191387559807e-06,
+      "loss": 0.1323,
+      "step": 357
+    },
+    {
+      "epoch": 32.55,
+      "learning_rate": 2.8401913875598084e-06,
+      "loss": 0.1604,
+      "step": 358
+    },
+    {
+      "epoch": 32.64,
+      "learning_rate": 2.836363636363636e-06,
+      "loss": 0.1515,
+      "step": 359
+    },
+    {
+      "epoch": 32.73,
+      "learning_rate": 2.832535885167464e-06,
+      "loss": 0.1342,
+      "step": 360
+    },
+    {
+      "epoch": 32.73,
+      "eval_accuracy_safe": 0.4963030211911288,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9938814935611622,
+      "eval_iou_safe": 0.4132349849379946,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9788741179805001,
+      "eval_loss": 0.1338294893503189,
+      "eval_mean_accuracy": 0.7450922573761455,
+      "eval_mean_iou": 0.6960545514592473,
+      "eval_overall_accuracy": 0.9791838233150653,
+      "eval_runtime": 12.0645,
+      "eval_samples_per_second": 5.553,
+      "eval_steps_per_second": 0.414,
+      "step": 360
+    },
+    {
+      "epoch": 32.82,
+      "learning_rate": 2.828708133971292e-06,
+      "loss": 0.1544,
+      "step": 361
+    },
+    {
+      "epoch": 32.91,
+      "learning_rate": 2.8248803827751195e-06,
+      "loss": 0.1796,
+      "step": 362
+    },
+    {
+      "epoch": 33.0,
+      "learning_rate": 2.8210526315789472e-06,
+      "loss": 0.15,
+      "step": 363
+    },
+    {
+      "epoch": 33.09,
+      "learning_rate": 2.817224880382775e-06,
+      "loss": 0.1424,
+      "step": 364
+    },
+    {
+      "epoch": 33.18,
+      "learning_rate": 2.8133971291866027e-06,
+      "loss": 0.1369,
+      "step": 365
+    },
+    {
+      "epoch": 33.27,
+      "learning_rate": 2.8095693779904305e-06,
+      "loss": 0.1645,
+      "step": 366
+    },
+    {
+      "epoch": 33.36,
+      "learning_rate": 2.8057416267942582e-06,
+      "loss": 0.1385,
+      "step": 367
+    },
+    {
+      "epoch": 33.45,
+      "learning_rate": 2.801913875598086e-06,
+      "loss": 0.1455,
+      "step": 368
+    },
+    {
+      "epoch": 33.55,
+      "learning_rate": 2.7980861244019138e-06,
+      "loss": 0.1577,
+      "step": 369
+    },
+    {
+      "epoch": 33.64,
+      "learning_rate": 2.7942583732057415e-06,
+      "loss": 0.1358,
+      "step": 370
+    },
+    {
+      "epoch": 33.64,
+      "eval_accuracy_safe": 0.5048110839973632,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.993699327057575,
+      "eval_iou_safe": 0.41823486576088,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9789443854555531,
+      "eval_loss": 0.13420753180980682,
+      "eval_mean_accuracy": 0.749255205527469,
+      "eval_mean_iou": 0.6985896256082165,
+      "eval_overall_accuracy": 0.9792583522511952,
+      "eval_runtime": 10.8902,
+      "eval_samples_per_second": 6.152,
+      "eval_steps_per_second": 0.459,
+      "step": 370
+    },
+    {
+      "epoch": 33.73,
+      "learning_rate": 2.7904306220095693e-06,
+      "loss": 0.1426,
+      "step": 371
+    },
+    {
+      "epoch": 33.82,
+      "learning_rate": 2.786602870813397e-06,
+      "loss": 0.1332,
+      "step": 372
+    },
+    {
+      "epoch": 33.91,
+      "learning_rate": 2.7827751196172248e-06,
+      "loss": 0.1415,
+      "step": 373
+    },
+    {
+      "epoch": 34.0,
+      "learning_rate": 2.7789473684210525e-06,
+      "loss": 0.1376,
+      "step": 374
+    },
+    {
+      "epoch": 34.09,
+      "learning_rate": 2.7751196172248803e-06,
+      "loss": 0.1378,
+      "step": 375
+    },
+    {
+      "epoch": 34.18,
+      "learning_rate": 2.771291866028708e-06,
+      "loss": 0.1321,
+      "step": 376
+    },
+    {
+      "epoch": 34.27,
+      "learning_rate": 2.767464114832536e-06,
+      "loss": 0.1382,
+      "step": 377
+    },
+    {
+      "epoch": 34.36,
+      "learning_rate": 2.7636363636363635e-06,
+      "loss": 0.1418,
+      "step": 378
+    },
+    {
+      "epoch": 34.45,
+      "learning_rate": 2.7598086124401913e-06,
+      "loss": 0.1481,
+      "step": 379
+    },
+    {
+      "epoch": 34.55,
+      "learning_rate": 2.755980861244019e-06,
+      "loss": 0.1493,
+      "step": 380
+    },
+    {
+      "epoch": 34.55,
+      "eval_accuracy_safe": 0.48086938755054914,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9945674487173425,
+      "eval_iou_safe": 0.40804123003145243,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9790967191500697,
+      "eval_loss": 0.12969090044498444,
+      "eval_mean_accuracy": 0.7377184181339458,
+      "eval_mean_iou": 0.6935689745907611,
+      "eval_overall_accuracy": 0.9793936316646746,
+      "eval_runtime": 10.4894,
+      "eval_samples_per_second": 6.387,
+      "eval_steps_per_second": 0.477,
+      "step": 380
+    },
+    {
+      "epoch": 34.64,
+      "learning_rate": 2.7521531100478464e-06,
+      "loss": 0.1437,
+      "step": 381
+    },
+    {
+      "epoch": 34.73,
+      "learning_rate": 2.7483253588516746e-06,
+      "loss": 0.1472,
+      "step": 382
+    },
+    {
+      "epoch": 34.82,
+      "learning_rate": 2.7444976076555023e-06,
+      "loss": 0.1412,
+      "step": 383
+    },
+    {
+      "epoch": 34.91,
+      "learning_rate": 2.74066985645933e-06,
+      "loss": 0.1484,
+      "step": 384
+    },
+    {
+      "epoch": 35.0,
+      "learning_rate": 2.736842105263158e-06,
+      "loss": 0.154,
+      "step": 385
+    },
+    {
+      "epoch": 35.09,
+      "learning_rate": 2.7330143540669856e-06,
+      "loss": 0.1348,
+      "step": 386
+    },
+    {
+      "epoch": 35.18,
+      "learning_rate": 2.7291866028708133e-06,
+      "loss": 0.139,
+      "step": 387
+    },
+    {
+      "epoch": 35.27,
+      "learning_rate": 2.7253588516746407e-06,
+      "loss": 0.1368,
+      "step": 388
+    },
+    {
+      "epoch": 35.36,
+      "learning_rate": 2.721531100478469e-06,
+      "loss": 0.1499,
+      "step": 389
+    },
+    {
+      "epoch": 35.45,
+      "learning_rate": 2.7177033492822966e-06,
+      "loss": 0.1435,
+      "step": 390
+    },
+    {
+      "epoch": 35.45,
+      "eval_accuracy_safe": 0.5658305095200096,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9923169150369561,
+      "eval_iou_safe": 0.4517890479393129,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.979374460845436,
+      "eval_loss": 0.12713803350925446,
+      "eval_mean_accuracy": 0.7790737122784828,
+      "eval_mean_iou": 0.7155817543923745,
+      "eval_overall_accuracy": 0.9797191904551947,
+      "eval_runtime": 11.095,
+      "eval_samples_per_second": 6.039,
+      "eval_steps_per_second": 0.451,
+      "step": 390
+    },
+    {
+      "epoch": 35.55,
+      "learning_rate": 2.7138755980861244e-06,
+      "loss": 0.1389,
+      "step": 391
+    },
+    {
+      "epoch": 35.64,
+      "learning_rate": 2.710047846889952e-06,
+      "loss": 0.1338,
+      "step": 392
+    },
+    {
+      "epoch": 35.73,
+      "learning_rate": 2.70622009569378e-06,
+      "loss": 0.1411,
+      "step": 393
+    },
+    {
+      "epoch": 35.82,
+      "learning_rate": 2.7023923444976072e-06,
+      "loss": 0.1556,
+      "step": 394
+    },
+    {
+      "epoch": 35.91,
+      "learning_rate": 2.698564593301435e-06,
+      "loss": 0.1377,
+      "step": 395
+    },
+    {
+      "epoch": 36.0,
+      "learning_rate": 2.694736842105263e-06,
+      "loss": 0.1369,
+      "step": 396
+    },
+    {
+      "epoch": 36.09,
+      "learning_rate": 2.690909090909091e-06,
+      "loss": 0.1292,
+      "step": 397
+    },
+    {
+      "epoch": 36.18,
+      "learning_rate": 2.6870813397129187e-06,
+      "loss": 0.1583,
+      "step": 398
+    },
+    {
+      "epoch": 36.27,
+      "learning_rate": 2.6832535885167464e-06,
+      "loss": 0.1229,
+      "step": 399
+    },
+    {
+      "epoch": 36.36,
+      "learning_rate": 2.679425837320574e-06,
+      "loss": 0.1305,
+      "step": 400
+    },
+    {
+      "epoch": 36.36,
+      "eval_accuracy_safe": 0.4157327843763131,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.996766177881572,
+      "eval_iou_safe": 0.3758054230365799,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9793497969358366,
+      "eval_loss": 0.12249775975942612,
+      "eval_mean_accuracy": 0.7062494811289426,
+      "eval_mean_iou": 0.6775776099862082,
+      "eval_overall_accuracy": 0.9796033830785039,
+      "eval_runtime": 11.1918,
+      "eval_samples_per_second": 5.987,
+      "eval_steps_per_second": 0.447,
+      "step": 400
+    },
+    {
+      "epoch": 36.45,
+      "learning_rate": 2.6755980861244015e-06,
+      "loss": 0.1488,
+      "step": 401
+    },
+    {
+      "epoch": 36.55,
+      "learning_rate": 2.6717703349282293e-06,
+      "loss": 0.1464,
+      "step": 402
+    },
+    {
+      "epoch": 36.64,
+      "learning_rate": 2.6679425837320574e-06,
+      "loss": 0.1271,
+      "step": 403
+    },
+    {
+      "epoch": 36.73,
+      "learning_rate": 2.664114832535885e-06,
+      "loss": 0.1289,
+      "step": 404
+    },
+    {
+      "epoch": 36.82,
+      "learning_rate": 2.660287081339713e-06,
+      "loss": 0.1291,
+      "step": 405
+    },
+    {
+      "epoch": 36.91,
+      "learning_rate": 2.6564593301435407e-06,
+      "loss": 0.1388,
+      "step": 406
+    },
+    {
+      "epoch": 37.0,
+      "learning_rate": 2.652631578947368e-06,
+      "loss": 0.1631,
+      "step": 407
+    },
+    {
+      "epoch": 37.09,
+      "learning_rate": 2.648803827751196e-06,
+      "loss": 0.1404,
+      "step": 408
+    },
+    {
+      "epoch": 37.18,
+      "learning_rate": 2.6449760765550235e-06,
+      "loss": 0.1367,
+      "step": 409
+    },
+    {
+      "epoch": 37.27,
+      "learning_rate": 2.6411483253588517e-06,
+      "loss": 0.1496,
+      "step": 410
+    },
+    {
+      "epoch": 37.27,
+      "eval_accuracy_safe": 0.5385040921199225,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9933532400351403,
+      "eval_iou_safe": 0.44198577745433837,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9795931225563403,
+      "eval_loss": 0.12370359897613525,
+      "eval_mean_accuracy": 0.7659286660775313,
+      "eval_mean_iou": 0.7107894500053393,
+      "eval_overall_accuracy": 0.9799177255203475,
+      "eval_runtime": 10.3073,
+      "eval_samples_per_second": 6.5,
+      "eval_steps_per_second": 0.485,
+      "step": 410
+    },
+    {
+      "epoch": 37.36,
+      "learning_rate": 2.6373205741626795e-06,
+      "loss": 0.13,
+      "step": 411
+    },
+    {
+      "epoch": 37.45,
+      "learning_rate": 2.6334928229665072e-06,
+      "loss": 0.1364,
+      "step": 412
+    },
+    {
+      "epoch": 37.55,
+      "learning_rate": 2.629665071770335e-06,
+      "loss": 0.127,
+      "step": 413
+    },
+    {
+      "epoch": 37.64,
+      "learning_rate": 2.6258373205741623e-06,
+      "loss": 0.1319,
+      "step": 414
+    },
+    {
+      "epoch": 37.73,
+      "learning_rate": 2.62200956937799e-06,
+      "loss": 0.1428,
+      "step": 415
+    },
+    {
+      "epoch": 37.82,
+      "learning_rate": 2.6181818181818183e-06,
+      "loss": 0.1363,
+      "step": 416
+    },
+    {
+      "epoch": 37.91,
+      "learning_rate": 2.614354066985646e-06,
+      "loss": 0.1275,
+      "step": 417
+    },
+    {
+      "epoch": 38.0,
+      "learning_rate": 2.6105263157894738e-06,
+      "loss": 0.1578,
+      "step": 418
+    },
+    {
+      "epoch": 38.09,
+      "learning_rate": 2.6066985645933015e-06,
+      "loss": 0.1307,
+      "step": 419
+    },
+    {
+      "epoch": 38.18,
+      "learning_rate": 2.602870813397129e-06,
+      "loss": 0.1445,
+      "step": 420
+    },
+    {
+      "epoch": 38.18,
+      "eval_accuracy_safe": 0.5762718725062741,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9924268602954817,
+      "eval_iou_safe": 0.4614568943562515,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9797902966849596,
+      "eval_loss": 0.12068379670381546,
+      "eval_mean_accuracy": 0.7843493664008778,
+      "eval_mean_iou": 0.7206235955206055,
+      "eval_overall_accuracy": 0.9801343092277869,
+      "eval_runtime": 10.7505,
+      "eval_samples_per_second": 6.232,
+      "eval_steps_per_second": 0.465,
+      "step": 420
+    },
+    {
+      "epoch": 38.27,
+      "learning_rate": 2.5990430622009566e-06,
+      "loss": 0.1382,
+      "step": 421
+    },
+    {
+      "epoch": 38.36,
+      "learning_rate": 2.5952153110047844e-06,
+      "loss": 0.133,
+      "step": 422
+    },
+    {
+      "epoch": 38.45,
+      "learning_rate": 2.5913875598086125e-06,
+      "loss": 0.136,
+      "step": 423
+    },
+    {
+      "epoch": 38.55,
+      "learning_rate": 2.5875598086124403e-06,
+      "loss": 0.1289,
+      "step": 424
+    },
+    {
+      "epoch": 38.64,
+      "learning_rate": 2.583732057416268e-06,
+      "loss": 0.1296,
+      "step": 425
+    },
+    {
+      "epoch": 38.73,
+      "learning_rate": 2.5799043062200954e-06,
+      "loss": 0.1263,
+      "step": 426
+    },
+    {
+      "epoch": 38.82,
+      "learning_rate": 2.576076555023923e-06,
+      "loss": 0.1309,
+      "step": 427
+    },
+    {
+      "epoch": 38.91,
+      "learning_rate": 2.572248803827751e-06,
+      "loss": 0.1328,
+      "step": 428
+    },
+    {
+      "epoch": 39.0,
+      "learning_rate": 2.5684210526315787e-06,
+      "loss": 0.1083,
+      "step": 429
+    },
+    {
+      "epoch": 39.09,
+      "learning_rate": 2.564593301435407e-06,
+      "loss": 0.1307,
+      "step": 430
+    },
+    {
+      "epoch": 39.09,
+      "eval_accuracy_safe": 0.4852602727051939,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9956334014399426,
+      "eval_iou_safe": 0.42437840298029433,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9802750640686686,
+      "eval_loss": 0.11942077428102493,
+      "eval_mean_accuracy": 0.7404468370725683,
+      "eval_mean_iou": 0.7023267335244815,
+      "eval_overall_accuracy": 0.9805577975600513,
+      "eval_runtime": 11.0705,
+      "eval_samples_per_second": 6.052,
+      "eval_steps_per_second": 0.452,
+      "step": 430
+    },
+    {
+      "epoch": 39.18,
+      "learning_rate": 2.5607655502392346e-06,
+      "loss": 0.1464,
+      "step": 431
+    },
+    {
+      "epoch": 39.27,
+      "learning_rate": 2.5569377990430623e-06,
+      "loss": 0.131,
+      "step": 432
+    },
+    {
+      "epoch": 39.36,
+      "learning_rate": 2.5531100478468897e-06,
+      "loss": 0.1256,
+      "step": 433
+    },
+    {
+      "epoch": 39.45,
+      "learning_rate": 2.5492822966507174e-06,
+      "loss": 0.1231,
+      "step": 434
+    },
+    {
+      "epoch": 39.55,
+      "learning_rate": 2.545454545454545e-06,
+      "loss": 0.1165,
+      "step": 435
+    },
+    {
+      "epoch": 39.64,
+      "learning_rate": 2.541626794258373e-06,
+      "loss": 0.1234,
+      "step": 436
+    },
+    {
+      "epoch": 39.73,
+      "learning_rate": 2.537799043062201e-06,
+      "loss": 0.1336,
+      "step": 437
+    },
+    {
+      "epoch": 39.82,
+      "learning_rate": 2.533971291866029e-06,
+      "loss": 0.1271,
+      "step": 438
+    },
+    {
+      "epoch": 39.91,
+      "learning_rate": 2.530143540669856e-06,
+      "loss": 0.1397,
+      "step": 439
+    },
+    {
+      "epoch": 40.0,
+      "learning_rate": 2.526315789473684e-06,
+      "loss": 0.1379,
+      "step": 440
+    },
+    {
+      "epoch": 40.0,
+      "eval_accuracy_safe": 0.5721855351367188,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9922185861931518,
+      "eval_iou_safe": 0.4556878276420518,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9794644021123315,
+      "eval_loss": 0.11739415675401688,
+      "eval_mean_accuracy": 0.7822020606649354,
+      "eval_mean_iou": 0.7175761148771916,
+      "eval_overall_accuracy": 0.9798114833547108,
+      "eval_runtime": 11.1185,
+      "eval_samples_per_second": 6.026,
+      "eval_steps_per_second": 0.45,
+      "step": 440
+    },
+    {
+      "epoch": 40.09,
+      "learning_rate": 2.5224880382775117e-06,
+      "loss": 0.1205,
+      "step": 441
+    },
+    {
+      "epoch": 40.18,
+      "learning_rate": 2.5186602870813395e-06,
+      "loss": 0.1271,
+      "step": 442
+    },
+    {
+      "epoch": 40.27,
+      "learning_rate": 2.5148325358851672e-06,
+      "loss": 0.1292,
+      "step": 443
+    },
+    {
+      "epoch": 40.36,
+      "learning_rate": 2.5110047846889954e-06,
+      "loss": 0.1333,
+      "step": 444
+    },
+    {
+      "epoch": 40.45,
+      "learning_rate": 2.507177033492823e-06,
+      "loss": 0.1326,
+      "step": 445
+    },
+    {
+      "epoch": 40.55,
+      "learning_rate": 2.5033492822966505e-06,
+      "loss": 0.1345,
+      "step": 446
+    },
+    {
+      "epoch": 40.64,
+      "learning_rate": 2.4995215311004783e-06,
+      "loss": 0.13,
+      "step": 447
+    },
+    {
+      "epoch": 40.73,
+      "learning_rate": 2.495693779904306e-06,
+      "loss": 0.151,
+      "step": 448
+    },
+    {
+      "epoch": 40.82,
+      "learning_rate": 2.4918660287081338e-06,
+      "loss": 0.1345,
+      "step": 449
+    },
+    {
+      "epoch": 40.91,
+      "learning_rate": 2.4880382775119615e-06,
+      "loss": 0.1202,
+      "step": 450
+    },
+    {
+      "epoch": 40.91,
+      "eval_accuracy_safe": 0.5398707021175708,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9942766276679765,
+      "eval_iou_safe": 0.4544224441718747,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9805439411158418,
+      "eval_loss": 0.11430685967206955,
+      "eval_mean_accuracy": 0.7670736648927736,
+      "eval_mean_iou": 0.7174831926438583,
+      "eval_overall_accuracy": 0.9808542052311684,
+      "eval_runtime": 10.6429,
+      "eval_samples_per_second": 6.295,
+      "eval_steps_per_second": 0.47,
+      "step": 450
+    },
+    {
+      "epoch": 41.0,
+      "learning_rate": 2.4842105263157897e-06,
+      "loss": 0.1246,
+      "step": 451
+    },
+    {
+      "epoch": 41.09,
+      "learning_rate": 2.480382775119617e-06,
+      "loss": 0.1379,
+      "step": 452
+    },
+    {
+      "epoch": 41.18,
+      "learning_rate": 2.4765550239234448e-06,
+      "loss": 0.1143,
+      "step": 453
+    },
+    {
+      "epoch": 41.27,
+      "learning_rate": 2.4727272727272725e-06,
+      "loss": 0.1425,
+      "step": 454
+    },
+    {
+      "epoch": 41.36,
+      "learning_rate": 2.4688995215311003e-06,
+      "loss": 0.1284,
+      "step": 455
+    },
+    {
+      "epoch": 41.45,
+      "learning_rate": 2.465071770334928e-06,
+      "loss": 0.1436,
+      "step": 456
+    },
+    {
+      "epoch": 41.55,
+      "learning_rate": 2.461244019138756e-06,
+      "loss": 0.1248,
+      "step": 457
+    },
+    {
+      "epoch": 41.64,
+      "learning_rate": 2.457416267942584e-06,
+      "loss": 0.1195,
+      "step": 458
+    },
+    {
+      "epoch": 41.73,
+      "learning_rate": 2.4535885167464113e-06,
+      "loss": 0.1224,
+      "step": 459
+    },
+    {
+      "epoch": 41.82,
+      "learning_rate": 2.449760765550239e-06,
+      "loss": 0.1239,
+      "step": 460
+    },
+    {
+      "epoch": 41.82,
+      "eval_accuracy_safe": 0.5579739476717515,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9931808712146769,
+      "eval_iou_safe": 0.45584717644594847,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9799958550830518,
+      "eval_loss": 0.11501991748809814,
+      "eval_mean_accuracy": 0.7755774094432142,
+      "eval_mean_iou": 0.7179215157645001,
+      "eval_overall_accuracy": 0.9803255565130888,
+      "eval_runtime": 10.3848,
+      "eval_samples_per_second": 6.452,
+      "eval_steps_per_second": 0.481,
+      "step": 460
+    },
+    {
+      "epoch": 41.91,
+      "learning_rate": 2.445933014354067e-06,
+      "loss": 0.1168,
+      "step": 461
+    },
+    {
+      "epoch": 42.0,
+      "learning_rate": 2.4421052631578946e-06,
+      "loss": 0.1028,
+      "step": 462
+    },
+    {
+      "epoch": 42.09,
+      "learning_rate": 2.4382775119617223e-06,
+      "loss": 0.1296,
+      "step": 463
+    },
+    {
+      "epoch": 42.18,
+      "learning_rate": 2.4344497607655505e-06,
+      "loss": 0.113,
+      "step": 464
+    },
+    {
+      "epoch": 42.27,
+      "learning_rate": 2.430622009569378e-06,
+      "loss": 0.1187,
+      "step": 465
+    },
+    {
+      "epoch": 42.36,
+      "learning_rate": 2.4267942583732056e-06,
+      "loss": 0.1362,
+      "step": 466
+    },
+    {
+      "epoch": 42.45,
+      "learning_rate": 2.4229665071770334e-06,
+      "loss": 0.1244,
+      "step": 467
+    },
+    {
+      "epoch": 42.55,
+      "learning_rate": 2.419138755980861e-06,
+      "loss": 0.1272,
+      "step": 468
+    },
+    {
+      "epoch": 42.64,
+      "learning_rate": 2.415311004784689e-06,
+      "loss": 0.1167,
+      "step": 469
+    },
+    {
+      "epoch": 42.73,
+      "learning_rate": 2.4114832535885166e-06,
+      "loss": 0.1183,
+      "step": 470
+    },
+    {
+      "epoch": 42.73,
+      "eval_accuracy_safe": 0.47769476601863525,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9961165386885865,
+      "eval_iou_safe": 0.42364293711912065,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9805284404970969,
+      "eval_loss": 0.1128791943192482,
+      "eval_mean_accuracy": 0.7369056523536108,
+      "eval_mean_iou": 0.7020856888081087,
+      "eval_overall_accuracy": 0.9808031907722131,
+      "eval_runtime": 10.8661,
+      "eval_samples_per_second": 6.166,
+      "eval_steps_per_second": 0.46,
+      "step": 470
+    },
+    {
+      "epoch": 42.82,
+      "learning_rate": 2.407655502392345e-06,
+      "loss": 0.1203,
+      "step": 471
+    },
+    {
+      "epoch": 42.91,
+      "learning_rate": 2.403827751196172e-06,
+      "loss": 0.1455,
+      "step": 472
+    },
+    {
+      "epoch": 43.0,
+      "learning_rate": 2.4e-06,
+      "loss": 0.1139,
+      "step": 473
+    },
+    {
+      "epoch": 43.09,
+      "learning_rate": 2.3961722488038276e-06,
+      "loss": 0.1254,
+      "step": 474
+    },
+    {
+      "epoch": 43.18,
+      "learning_rate": 2.3923444976076554e-06,
+      "loss": 0.1316,
+      "step": 475
+    },
+    {
+      "epoch": 43.27,
+      "learning_rate": 2.388516746411483e-06,
+      "loss": 0.1215,
+      "step": 476
+    },
+    {
+      "epoch": 43.36,
+      "learning_rate": 2.384688995215311e-06,
+      "loss": 0.1256,
+      "step": 477
+    },
+    {
+      "epoch": 43.45,
+      "learning_rate": 2.3808612440191387e-06,
+      "loss": 0.1339,
+      "step": 478
+    },
+    {
+      "epoch": 43.55,
+      "learning_rate": 2.3770334928229664e-06,
+      "loss": 0.1328,
+      "step": 479
+    },
+    {
+      "epoch": 43.64,
+      "learning_rate": 2.373205741626794e-06,
+      "loss": 0.1202,
+      "step": 480
+    },
+    {
+      "epoch": 43.64,
+      "eval_accuracy_safe": 0.593264867907217,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9927646163538233,
+      "eval_iou_safe": 0.47932343707173647,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9806244962218924,
+      "eval_loss": 0.11194637417793274,
+      "eval_mean_accuracy": 0.7930147421305201,
+      "eval_mean_iou": 0.7299739666468145,
+      "eval_overall_accuracy": 0.9809640343509504,
+      "eval_runtime": 10.421,
+      "eval_samples_per_second": 6.429,
+      "eval_steps_per_second": 0.48,
+      "step": 480
+    },
+    {
+      "epoch": 43.73,
+      "learning_rate": 2.369377990430622e-06,
+      "loss": 0.1129,
+      "step": 481
+    },
+    {
+      "epoch": 43.82,
+      "learning_rate": 2.3655502392344497e-06,
+      "loss": 0.1249,
+      "step": 482
+    },
+    {
+      "epoch": 43.91,
+      "learning_rate": 2.3617224880382774e-06,
+      "loss": 0.1222,
+      "step": 483
+    },
+    {
+      "epoch": 44.0,
+      "learning_rate": 2.3578947368421048e-06,
+      "loss": 0.1237,
+      "step": 484
+    },
+    {
+      "epoch": 44.09,
+      "learning_rate": 2.354066985645933e-06,
+      "loss": 0.1188,
+      "step": 485
+    },
+    {
+      "epoch": 44.18,
+      "learning_rate": 2.3502392344497607e-06,
+      "loss": 0.1145,
+      "step": 486
+    },
+    {
+      "epoch": 44.27,
+      "learning_rate": 2.3464114832535885e-06,
+      "loss": 0.117,
+      "step": 487
+    },
+    {
+      "epoch": 44.36,
+      "learning_rate": 2.3425837320574162e-06,
+      "loss": 0.1173,
+      "step": 488
+    },
+    {
+      "epoch": 44.45,
+      "learning_rate": 2.338755980861244e-06,
+      "loss": 0.1351,
+      "step": 489
+    },
+    {
+      "epoch": 44.55,
+      "learning_rate": 2.3349282296650717e-06,
+      "loss": 0.1276,
+      "step": 490
+    },
+    {
+      "epoch": 44.55,
+      "eval_accuracy_safe": 0.5424612858084588,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9942335647972413,
+      "eval_iou_safe": 0.4560598940187007,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9805777244564327,
+      "eval_loss": 0.11313802003860474,
+      "eval_mean_accuracy": 0.76834742530285,
+      "eval_mean_iou": 0.7183188092375666,
+      "eval_overall_accuracy": 0.9808889360570195,
+      "eval_runtime": 11.0555,
+      "eval_samples_per_second": 6.06,
+      "eval_steps_per_second": 0.452,
+      "step": 490
+    },
+    {
+      "epoch": 44.64,
+      "learning_rate": 2.331100478468899e-06,
+      "loss": 0.1254,
+      "step": 491
+    },
+    {
+      "epoch": 44.73,
+      "learning_rate": 2.3272727272727272e-06,
+      "loss": 0.1121,
+      "step": 492
+    },
+    {
+      "epoch": 44.82,
+      "learning_rate": 2.323444976076555e-06,
+      "loss": 0.1126,
+      "step": 493
+    },
+    {
+      "epoch": 44.91,
+      "learning_rate": 2.3196172248803828e-06,
+      "loss": 0.1346,
+      "step": 494
+    },
+    {
+      "epoch": 45.0,
+      "learning_rate": 2.3157894736842105e-06,
+      "loss": 0.1102,
+      "step": 495
+    },
+    {
+      "epoch": 45.09,
+      "learning_rate": 2.3119617224880383e-06,
+      "loss": 0.1278,
+      "step": 496
+    },
+    {
+      "epoch": 45.18,
+      "learning_rate": 2.3081339712918656e-06,
+      "loss": 0.1173,
+      "step": 497
+    },
+    {
+      "epoch": 45.27,
+      "learning_rate": 2.3043062200956934e-06,
+      "loss": 0.1239,
+      "step": 498
+    },
+    {
+      "epoch": 45.36,
+      "learning_rate": 2.3004784688995215e-06,
+      "loss": 0.1201,
+      "step": 499
+    },
+    {
+      "epoch": 45.45,
+      "learning_rate": 2.2966507177033493e-06,
+      "loss": 0.1172,
+      "step": 500
+    },
+    {
+      "epoch": 45.45,
+      "eval_accuracy_safe": 0.6271833955921526,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9898213219409551,
+      "eval_iou_safe": 0.4700070490079388,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.978715258102556,
+      "eval_loss": 0.11352450400590897,
+      "eval_mean_accuracy": 0.8085023587665539,
+      "eval_mean_iou": 0.7243611535552474,
+      "eval_overall_accuracy": 0.9791095790578358,
+      "eval_runtime": 10.616,
+      "eval_samples_per_second": 6.311,
+      "eval_steps_per_second": 0.471,
+      "step": 500
+    },
+    {
+      "epoch": 45.55,
+      "learning_rate": 2.292822966507177e-06,
+      "loss": 0.1216,
+      "step": 501
+    },
+    {
+      "epoch": 45.64,
+      "learning_rate": 2.288995215311005e-06,
+      "loss": 0.1183,
+      "step": 502
+    },
+    {
+      "epoch": 45.73,
+      "learning_rate": 2.2851674641148326e-06,
+      "loss": 0.1205,
+      "step": 503
+    },
+    {
+      "epoch": 45.82,
+      "learning_rate": 2.28133971291866e-06,
+      "loss": 0.152,
+      "step": 504
+    },
+    {
+      "epoch": 45.91,
+      "learning_rate": 2.2775119617224876e-06,
+      "loss": 0.1116,
+      "step": 505
+    },
+    {
+      "epoch": 46.0,
+      "learning_rate": 2.273684210526316e-06,
+      "loss": 0.143,
+      "step": 506
+    },
+    {
+      "epoch": 46.09,
+      "learning_rate": 2.2698564593301436e-06,
+      "loss": 0.1235,
+      "step": 507
+    },
+    {
+      "epoch": 46.18,
+      "learning_rate": 2.2660287081339713e-06,
+      "loss": 0.1128,
+      "step": 508
+    },
+    {
+      "epoch": 46.27,
+      "learning_rate": 2.262200956937799e-06,
+      "loss": 0.1084,
+      "step": 509
+    },
+    {
+      "epoch": 46.36,
+      "learning_rate": 2.2583732057416264e-06,
+      "loss": 0.1288,
+      "step": 510
+    },
+    {
+      "epoch": 46.36,
+      "eval_accuracy_safe": 0.42358163615406264,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9973644819084901,
+      "eval_iou_safe": 0.3898272913695354,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9801677166712226,
+      "eval_loss": 0.11046311259269714,
+      "eval_mean_accuracy": 0.7104730590312764,
+      "eval_mean_iou": 0.6849975040203791,
+      "eval_overall_accuracy": 0.980415856660302,
+      "eval_runtime": 10.6066,
+      "eval_samples_per_second": 6.317,
+      "eval_steps_per_second": 0.471,
+      "step": 510
+    },
+    {
+      "epoch": 46.45,
+      "learning_rate": 2.254545454545454e-06,
+      "loss": 0.1269,
+      "step": 511
+    },
+    {
+      "epoch": 46.55,
+      "learning_rate": 2.250717703349282e-06,
+      "loss": 0.1058,
+      "step": 512
+    },
+    {
+      "epoch": 46.64,
+      "learning_rate": 2.24688995215311e-06,
+      "loss": 0.1251,
+      "step": 513
+    },
+    {
+      "epoch": 46.73,
+      "learning_rate": 2.243062200956938e-06,
+      "loss": 0.1636,
+      "step": 514
+    },
+    {
+      "epoch": 46.82,
+      "learning_rate": 2.2392344497607656e-06,
+      "loss": 0.1107,
+      "step": 515
+    },
+    {
+      "epoch": 46.91,
+      "learning_rate": 2.2354066985645934e-06,
+      "loss": 0.118,
+      "step": 516
+    },
+    {
+      "epoch": 47.0,
+      "learning_rate": 2.2315789473684207e-06,
+      "loss": 0.1174,
+      "step": 517
+    },
+    {
+      "epoch": 47.09,
+      "learning_rate": 2.2277511961722485e-06,
+      "loss": 0.1267,
+      "step": 518
+    },
+    {
+      "epoch": 47.18,
+      "learning_rate": 2.2239234449760762e-06,
+      "loss": 0.1232,
+      "step": 519
+    },
+    {
+      "epoch": 47.27,
+      "learning_rate": 2.2200956937799044e-06,
+      "loss": 0.1185,
+      "step": 520
+    },
+    {
+      "epoch": 47.27,
+      "eval_accuracy_safe": 0.6035404643775467,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9914433958511564,
+      "eval_iou_safe": 0.471103501402246,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.979622075379713,
+      "eval_loss": 0.1129835844039917,
+      "eval_mean_accuracy": 0.7974919301143515,
+      "eval_mean_iou": 0.7253627883909795,
+      "eval_overall_accuracy": 0.9799853652270872,
+      "eval_runtime": 10.9811,
+      "eval_samples_per_second": 6.101,
+      "eval_steps_per_second": 0.455,
+      "step": 520
+    },
+    {
+      "epoch": 47.36,
+      "learning_rate": 2.216267942583732e-06,
+      "loss": 0.123,
+      "step": 521
+    },
+    {
+      "epoch": 47.45,
+      "learning_rate": 2.21244019138756e-06,
+      "loss": 0.1112,
+      "step": 522
+    },
+    {
+      "epoch": 47.55,
+      "learning_rate": 2.2086124401913872e-06,
+      "loss": 0.1046,
+      "step": 523
+    },
+    {
+      "epoch": 47.64,
+      "learning_rate": 2.204784688995215e-06,
+      "loss": 0.1181,
+      "step": 524
+    },
+    {
+      "epoch": 47.73,
+      "learning_rate": 2.2009569377990427e-06,
+      "loss": 0.1084,
+      "step": 525
+    },
+    {
+      "epoch": 47.82,
+      "learning_rate": 2.197129186602871e-06,
+      "loss": 0.1212,
+      "step": 526
+    },
+    {
+      "epoch": 47.91,
+      "learning_rate": 2.1933014354066987e-06,
+      "loss": 0.1342,
+      "step": 527
+    },
+    {
+      "epoch": 48.0,
+      "learning_rate": 2.1894736842105264e-06,
+      "loss": 0.1148,
+      "step": 528
+    },
+    {
+      "epoch": 48.09,
+      "learning_rate": 2.185645933014354e-06,
+      "loss": 0.1158,
+      "step": 529
+    },
+    {
+      "epoch": 48.18,
+      "learning_rate": 2.1818181818181815e-06,
+      "loss": 0.1045,
+      "step": 530
+    },
+    {
+      "epoch": 48.18,
+      "eval_accuracy_safe": 0.5750498263306618,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9930318525611789,
+      "eval_iou_safe": 0.4679260257256816,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9803515847470109,
+      "eval_loss": 0.11019009351730347,
+      "eval_mean_accuracy": 0.7840408394459204,
+      "eval_mean_iou": 0.7241388052363462,
+      "eval_overall_accuracy": 0.9806853337074394,
+      "eval_runtime": 10.6094,
+      "eval_samples_per_second": 6.315,
+      "eval_steps_per_second": 0.471,
+      "step": 530
+    },
+    {
+      "epoch": 48.27,
+      "learning_rate": 2.1779904306220093e-06,
+      "loss": 0.119,
+      "step": 531
+    },
+    {
+      "epoch": 48.36,
+      "learning_rate": 2.174162679425837e-06,
+      "loss": 0.1194,
+      "step": 532
+    },
+    {
+      "epoch": 48.45,
+      "learning_rate": 2.1703349282296652e-06,
+      "loss": 0.1058,
+      "step": 533
+    },
+    {
+      "epoch": 48.55,
+      "learning_rate": 2.166507177033493e-06,
+      "loss": 0.1141,
+      "step": 534
+    },
+    {
+      "epoch": 48.64,
+      "learning_rate": 2.1626794258373207e-06,
+      "loss": 0.1087,
+      "step": 535
+    },
+    {
+      "epoch": 48.73,
+      "learning_rate": 2.158851674641148e-06,
+      "loss": 0.124,
+      "step": 536
+    },
+    {
+      "epoch": 48.82,
+      "learning_rate": 2.155023923444976e-06,
+      "loss": 0.1151,
+      "step": 537
+    },
+    {
+      "epoch": 48.91,
+      "learning_rate": 2.1511961722488036e-06,
+      "loss": 0.1327,
+      "step": 538
+    },
+    {
+      "epoch": 49.0,
+      "learning_rate": 2.1473684210526313e-06,
+      "loss": 0.1204,
+      "step": 539
+    },
+    {
+      "epoch": 49.09,
+      "learning_rate": 2.1435406698564595e-06,
+      "loss": 0.1211,
+      "step": 540
+    },
+    {
+      "epoch": 49.09,
+      "eval_accuracy_safe": 0.5811735498321132,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9929194432146821,
+      "eval_iou_safe": 0.4714920819591893,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9804210186295622,
+      "eval_loss": 0.10691045969724655,
+      "eval_mean_accuracy": 0.7870464965233976,
+      "eval_mean_iou": 0.7259565502943758,
+      "eval_overall_accuracy": 0.9807571297261253,
+      "eval_runtime": 10.4844,
+      "eval_samples_per_second": 6.39,
+      "eval_steps_per_second": 0.477,
+      "step": 540
+    },
+    {
+      "epoch": 49.18,
+      "learning_rate": 2.1397129186602873e-06,
+      "loss": 0.1222,
+      "step": 541
+    },
+    {
+      "epoch": 49.27,
+      "learning_rate": 2.135885167464115e-06,
+      "loss": 0.1076,
+      "step": 542
+    },
+    {
+      "epoch": 49.36,
+      "learning_rate": 2.1320574162679423e-06,
+      "loss": 0.1193,
+      "step": 543
+    },
+    {
+      "epoch": 49.45,
+      "learning_rate": 2.12822966507177e-06,
+      "loss": 0.1208,
+      "step": 544
+    },
+    {
+      "epoch": 49.55,
+      "learning_rate": 2.124401913875598e-06,
+      "loss": 0.1147,
+      "step": 545
+    },
+    {
+      "epoch": 49.64,
+      "learning_rate": 2.1205741626794256e-06,
+      "loss": 0.1024,
+      "step": 546
+    },
+    {
+      "epoch": 49.73,
+      "learning_rate": 2.116746411483254e-06,
+      "loss": 0.0985,
+      "step": 547
+    },
+    {
+      "epoch": 49.82,
+      "learning_rate": 2.1129186602870815e-06,
+      "loss": 0.1168,
+      "step": 548
+    },
+    {
+      "epoch": 49.91,
+      "learning_rate": 2.109090909090909e-06,
+      "loss": 0.1315,
+      "step": 549
+    },
+    {
+      "epoch": 50.0,
+      "learning_rate": 2.1052631578947366e-06,
+      "loss": 0.1206,
+      "step": 550
+    },
+    {
+      "epoch": 50.0,
+      "eval_accuracy_safe": 0.5221317573949213,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9953372415333057,
+      "eval_iou_safe": 0.4527711197804365,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9810675158079116,
+      "eval_loss": 0.10707142949104309,
+      "eval_mean_accuracy": 0.7587344994641135,
+      "eval_mean_iou": 0.7169193177941741,
+      "eval_overall_accuracy": 0.9813595102794135,
+      "eval_runtime": 10.5803,
+      "eval_samples_per_second": 6.333,
+      "eval_steps_per_second": 0.473,
+      "step": 550
+    },
+    {
+      "epoch": 50.09,
+      "learning_rate": 2.1014354066985644e-06,
+      "loss": 0.1185,
+      "step": 551
+    },
+    {
+      "epoch": 50.18,
+      "learning_rate": 2.097607655502392e-06,
+      "loss": 0.1185,
+      "step": 552
+    },
+    {
+      "epoch": 50.27,
+      "learning_rate": 2.09377990430622e-06,
+      "loss": 0.1056,
+      "step": 553
+    },
+    {
+      "epoch": 50.36,
+      "learning_rate": 2.089952153110048e-06,
+      "loss": 0.1211,
+      "step": 554
+    },
+    {
+      "epoch": 50.45,
+      "learning_rate": 2.0861244019138754e-06,
+      "loss": 0.1243,
+      "step": 555
+    },
+    {
+      "epoch": 50.55,
+      "learning_rate": 2.082296650717703e-06,
+      "loss": 0.1082,
+      "step": 556
+    },
+    {
+      "epoch": 50.64,
+      "learning_rate": 2.078468899521531e-06,
+      "loss": 0.1183,
+      "step": 557
+    },
+    {
+      "epoch": 50.73,
+      "learning_rate": 2.0746411483253587e-06,
+      "loss": 0.1222,
+      "step": 558
+    },
+    {
+      "epoch": 50.82,
+      "learning_rate": 2.0708133971291864e-06,
+      "loss": 0.108,
+      "step": 559
+    },
+    {
+      "epoch": 50.91,
+      "learning_rate": 2.066985645933014e-06,
+      "loss": 0.1193,
+      "step": 560
+    },
+    {
+      "epoch": 50.91,
+      "eval_accuracy_safe": 0.4955589993870494,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9961437609937925,
+      "eval_iou_safe": 0.43983467087288786,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9810803435069183,
+      "eval_loss": 0.10526557266712189,
+      "eval_mean_accuracy": 0.745851380190421,
+      "eval_mean_iou": 0.7104575071899031,
+      "eval_overall_accuracy": 0.9813572897839902,
+      "eval_runtime": 11.1855,
+      "eval_samples_per_second": 5.99,
+      "eval_steps_per_second": 0.447,
+      "step": 560
+    },
+    {
+      "epoch": 51.0,
+      "learning_rate": 2.0631578947368424e-06,
+      "loss": 0.1199,
+      "step": 561
+    },
+    {
+      "epoch": 51.09,
+      "learning_rate": 2.0593301435406697e-06,
+      "loss": 0.1011,
+      "step": 562
+    },
+    {
+      "epoch": 51.18,
+      "learning_rate": 2.0555023923444975e-06,
+      "loss": 0.111,
+      "step": 563
+    },
+    {
+      "epoch": 51.27,
+      "learning_rate": 2.051674641148325e-06,
+      "loss": 0.1125,
+      "step": 564
+    },
+    {
+      "epoch": 51.36,
+      "learning_rate": 2.047846889952153e-06,
+      "loss": 0.1328,
+      "step": 565
+    },
+    {
+      "epoch": 51.45,
+      "learning_rate": 2.0440191387559807e-06,
+      "loss": 0.1076,
+      "step": 566
+    },
+    {
+      "epoch": 51.55,
+      "learning_rate": 2.0401913875598085e-06,
+      "loss": 0.1145,
+      "step": 567
+    },
+    {
+      "epoch": 51.64,
+      "learning_rate": 2.0363636363636362e-06,
+      "loss": 0.1035,
+      "step": 568
+    },
+    {
+      "epoch": 51.73,
+      "learning_rate": 2.032535885167464e-06,
+      "loss": 0.1105,
+      "step": 569
+    },
+    {
+      "epoch": 51.82,
+      "learning_rate": 2.0287081339712917e-06,
+      "loss": 0.1116,
+      "step": 570
+    },
+    {
+      "epoch": 51.82,
+      "eval_accuracy_safe": 0.5256976650051465,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9951027424947108,
+      "eval_iou_safe": 0.4528379892906064,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9809413209191069,
+      "eval_loss": 0.1042569950222969,
+      "eval_mean_accuracy": 0.7604002037499287,
+      "eval_mean_iou": 0.7168896551048567,
+      "eval_overall_accuracy": 0.9812372691595732,
+      "eval_runtime": 10.8585,
+      "eval_samples_per_second": 6.17,
+      "eval_steps_per_second": 0.46,
+      "step": 570
+    },
+    {
+      "epoch": 51.91,
+      "learning_rate": 2.0248803827751195e-06,
+      "loss": 0.1133,
+      "step": 571
+    },
+    {
+      "epoch": 52.0,
+      "learning_rate": 2.0210526315789473e-06,
+      "loss": 0.1099,
+      "step": 572
+    },
+    {
+      "epoch": 52.09,
+      "learning_rate": 2.017224880382775e-06,
+      "loss": 0.1218,
+      "step": 573
+    },
+    {
+      "epoch": 52.18,
+      "learning_rate": 2.013397129186603e-06,
+      "loss": 0.1045,
+      "step": 574
+    },
+    {
+      "epoch": 52.27,
+      "learning_rate": 2.0095693779904305e-06,
+      "loss": 0.1032,
+      "step": 575
+    },
+    {
+      "epoch": 52.36,
+      "learning_rate": 2.0057416267942583e-06,
+      "loss": 0.1136,
+      "step": 576
+    },
+    {
+      "epoch": 52.45,
+      "learning_rate": 2.001913875598086e-06,
+      "loss": 0.0962,
+      "step": 577
+    },
+    {
+      "epoch": 52.55,
+      "learning_rate": 1.9980861244019138e-06,
+      "loss": 0.1201,
+      "step": 578
+    },
+    {
+      "epoch": 52.64,
+      "learning_rate": 1.9942583732057415e-06,
+      "loss": 0.1166,
+      "step": 579
+    },
+    {
+      "epoch": 52.73,
+      "learning_rate": 1.9904306220095693e-06,
+      "loss": 0.1218,
+      "step": 580
+    },
+    {
+      "epoch": 52.73,
+      "eval_accuracy_safe": 0.5936118210801038,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9921905425252888,
+      "eval_iou_safe": 0.4724050601844111,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9800676658313282,
+      "eval_loss": 0.10777109861373901,
+      "eval_mean_accuracy": 0.7929011818026963,
+      "eval_mean_iou": 0.7262363630078696,
+      "eval_overall_accuracy": 0.9804171661832439,
+      "eval_runtime": 10.4208,
+      "eval_samples_per_second": 6.429,
+      "eval_steps_per_second": 0.48,
+      "step": 580
+    },
+    {
+      "epoch": 52.82,
+      "learning_rate": 1.986602870813397e-06,
+      "loss": 0.109,
+      "step": 581
+    },
+    {
+      "epoch": 52.91,
+      "learning_rate": 1.982775119617225e-06,
+      "loss": 0.0965,
+      "step": 582
+    },
+    {
+      "epoch": 53.0,
+      "learning_rate": 1.9789473684210526e-06,
+      "loss": 0.1157,
+      "step": 583
+    },
+    {
+      "epoch": 53.09,
+      "learning_rate": 1.9751196172248803e-06,
+      "loss": 0.1081,
+      "step": 584
+    },
+    {
+      "epoch": 53.18,
+      "learning_rate": 1.971291866028708e-06,
+      "loss": 0.1227,
+      "step": 585
+    },
+    {
+      "epoch": 53.27,
+      "learning_rate": 1.967464114832536e-06,
+      "loss": 0.1076,
+      "step": 586
+    },
+    {
+      "epoch": 53.36,
+      "learning_rate": 1.9636363636363636e-06,
+      "loss": 0.1168,
+      "step": 587
+    },
+    {
+      "epoch": 53.45,
+      "learning_rate": 1.9598086124401913e-06,
+      "loss": 0.1099,
+      "step": 588
+    },
+    {
+      "epoch": 53.55,
+      "learning_rate": 1.955980861244019e-06,
+      "loss": 0.0984,
+      "step": 589
+    },
+    {
+      "epoch": 53.64,
+      "learning_rate": 1.952153110047847e-06,
+      "loss": 0.1284,
+      "step": 590
+    },
+    {
+      "epoch": 53.64,
+      "eval_accuracy_safe": 0.58715849206441,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9923766398358776,
+      "eval_iou_safe": 0.46955410556755317,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9800613348526154,
+      "eval_loss": 0.10542566329240799,
+      "eval_mean_accuracy": 0.7897675659501437,
+      "eval_mean_iou": 0.7248077202100842,
+      "eval_overall_accuracy": 0.9804071454859492,
+      "eval_runtime": 10.3478,
+      "eval_samples_per_second": 6.475,
+      "eval_steps_per_second": 0.483,
+      "step": 590
+    },
+    {
+      "epoch": 53.73,
+      "learning_rate": 1.9483253588516746e-06,
+      "loss": 0.1142,
+      "step": 591
+    },
+    {
+      "epoch": 53.82,
+      "learning_rate": 1.9444976076555024e-06,
+      "loss": 0.1066,
+      "step": 592
+    },
+    {
+      "epoch": 53.91,
+      "learning_rate": 1.94066985645933e-06,
+      "loss": 0.0976,
+      "step": 593
+    },
+    {
+      "epoch": 54.0,
+      "learning_rate": 1.936842105263158e-06,
+      "loss": 0.0958,
+      "step": 594
+    },
+    {
+      "epoch": 54.09,
+      "learning_rate": 1.9330143540669856e-06,
+      "loss": 0.1366,
+      "step": 595
+    },
+    {
+      "epoch": 54.18,
+      "learning_rate": 1.9291866028708134e-06,
+      "loss": 0.1249,
+      "step": 596
+    },
+    {
+      "epoch": 54.27,
+      "learning_rate": 1.925358851674641e-06,
+      "loss": 0.1053,
+      "step": 597
+    },
+    {
+      "epoch": 54.36,
+      "learning_rate": 1.921531100478469e-06,
+      "loss": 0.1064,
+      "step": 598
+    },
+    {
+      "epoch": 54.45,
+      "learning_rate": 1.9177033492822967e-06,
+      "loss": 0.0949,
+      "step": 599
+    },
+    {
+      "epoch": 54.55,
+      "learning_rate": 1.9138755980861244e-06,
+      "loss": 0.096,
+      "step": 600
+    },
+    {
+      "epoch": 54.55,
+      "eval_accuracy_safe": 0.5451308977220597,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9942086892424842,
+      "eval_iou_safe": 0.45798961642974895,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9806317785722483,
+      "eval_loss": 0.10281717777252197,
+      "eval_mean_accuracy": 0.7696697934822719,
+      "eval_mean_iou": 0.7193106975009986,
+      "eval_overall_accuracy": 0.9809436513416803,
+      "eval_runtime": 10.3139,
+      "eval_samples_per_second": 6.496,
+      "eval_steps_per_second": 0.485,
+      "step": 600
+    },
+    {
+      "epoch": 54.64,
+      "learning_rate": 1.910047846889952e-06,
+      "loss": 0.1136,
+      "step": 601
+    },
+    {
+      "epoch": 54.73,
+      "learning_rate": 1.9062200956937797e-06,
+      "loss": 0.1175,
+      "step": 602
+    },
+    {
+      "epoch": 54.82,
+      "learning_rate": 1.9023923444976075e-06,
+      "loss": 0.1107,
+      "step": 603
+    },
+    {
+      "epoch": 54.91,
+      "learning_rate": 1.8985645933014354e-06,
+      "loss": 0.096,
+      "step": 604
+    },
+    {
+      "epoch": 55.0,
+      "learning_rate": 1.894736842105263e-06,
+      "loss": 0.1316,
+      "step": 605
+    },
+    {
+      "epoch": 55.09,
+      "learning_rate": 1.8909090909090907e-06,
+      "loss": 0.1177,
+      "step": 606
+    },
+    {
+      "epoch": 55.18,
+      "learning_rate": 1.8870813397129187e-06,
+      "loss": 0.1111,
+      "step": 607
+    },
+    {
+      "epoch": 55.27,
+      "learning_rate": 1.8832535885167464e-06,
+      "loss": 0.1068,
+      "step": 608
+    },
+    {
+      "epoch": 55.36,
+      "learning_rate": 1.879425837320574e-06,
+      "loss": 0.1087,
+      "step": 609
+    },
+    {
+      "epoch": 55.45,
+      "learning_rate": 1.8755980861244017e-06,
+      "loss": 0.1091,
+      "step": 610
+    },
+    {
+      "epoch": 55.45,
+      "eval_accuracy_safe": 0.6013989922937846,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9916947328242215,
+      "eval_iou_safe": 0.4724773003979623,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9798073122435136,
+      "eval_loss": 0.10218217223882675,
+      "eval_mean_accuracy": 0.7965468625590031,
+      "eval_mean_iou": 0.726142306320738,
+      "eval_overall_accuracy": 0.9801660224572936,
+      "eval_runtime": 11.0365,
+      "eval_samples_per_second": 6.071,
+      "eval_steps_per_second": 0.453,
+      "step": 610
+    },
+    {
+      "epoch": 55.55,
+      "learning_rate": 1.8717703349282297e-06,
+      "loss": 0.1026,
+      "step": 611
+    },
+    {
+      "epoch": 55.64,
+      "learning_rate": 1.8679425837320573e-06,
+      "loss": 0.0904,
+      "step": 612
+    },
+    {
+      "epoch": 55.73,
+      "learning_rate": 1.864114832535885e-06,
+      "loss": 0.1088,
+      "step": 613
+    },
+    {
+      "epoch": 55.82,
+      "learning_rate": 1.860287081339713e-06,
+      "loss": 0.1116,
+      "step": 614
+    },
+    {
+      "epoch": 55.91,
+      "learning_rate": 1.8564593301435405e-06,
+      "loss": 0.1021,
+      "step": 615
+    },
+    {
+      "epoch": 56.0,
+      "learning_rate": 1.8526315789473683e-06,
+      "loss": 0.1323,
+      "step": 616
+    },
+    {
+      "epoch": 56.09,
+      "learning_rate": 1.8488038277511962e-06,
+      "loss": 0.1097,
+      "step": 617
+    },
+    {
+      "epoch": 56.18,
+      "learning_rate": 1.8449760765550238e-06,
+      "loss": 0.0945,
+      "step": 618
+    },
+    {
+      "epoch": 56.27,
+      "learning_rate": 1.8411483253588515e-06,
+      "loss": 0.1039,
+      "step": 619
+    },
+    {
+      "epoch": 56.36,
+      "learning_rate": 1.8373205741626793e-06,
+      "loss": 0.1068,
+      "step": 620
+    },
+    {
+      "epoch": 56.36,
+      "eval_accuracy_safe": 0.492559781959206,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9961584281840974,
+      "eval_iou_safe": 0.4373597635384344,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.981006588415475,
+      "eval_loss": 0.10153985768556595,
+      "eval_mean_accuracy": 0.7443591050716517,
+      "eval_mean_iou": 0.7091831759769547,
+      "eval_overall_accuracy": 0.9812829316552005,
+      "eval_runtime": 10.1863,
+      "eval_samples_per_second": 6.577,
+      "eval_steps_per_second": 0.491,
+      "step": 620
+    },
+    {
+      "epoch": 56.45,
+      "learning_rate": 1.8334928229665073e-06,
+      "loss": 0.1031,
+      "step": 621
+    },
+    {
+      "epoch": 56.55,
+      "learning_rate": 1.8296650717703348e-06,
+      "loss": 0.0976,
+      "step": 622
+    },
+    {
+      "epoch": 56.64,
+      "learning_rate": 1.8258373205741626e-06,
+      "loss": 0.1113,
+      "step": 623
+    },
+    {
+      "epoch": 56.73,
+      "learning_rate": 1.8220095693779905e-06,
+      "loss": 0.1008,
+      "step": 624
+    },
+    {
+      "epoch": 56.82,
+      "learning_rate": 1.818181818181818e-06,
+      "loss": 0.102,
+      "step": 625
+    },
+    {
+      "epoch": 56.91,
+      "learning_rate": 1.8143540669856458e-06,
+      "loss": 0.1101,
+      "step": 626
+    },
+    {
+      "epoch": 57.0,
+      "learning_rate": 1.8105263157894736e-06,
+      "loss": 0.1363,
+      "step": 627
+    },
+    {
+      "epoch": 57.09,
+      "learning_rate": 1.8066985645933013e-06,
+      "loss": 0.1012,
+      "step": 628
+    },
+    {
+      "epoch": 57.18,
+      "learning_rate": 1.802870813397129e-06,
+      "loss": 0.1346,
+      "step": 629
+    },
+    {
+      "epoch": 57.27,
+      "learning_rate": 1.7990430622009569e-06,
+      "loss": 0.106,
+      "step": 630
+    },
+    {
+      "epoch": 57.27,
+      "eval_accuracy_safe": 0.5713085146163661,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9936834278232846,
+      "eval_iou_safe": 0.4731230854937291,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9808845669247197,
+      "eval_loss": 0.10105204582214355,
+      "eval_mean_accuracy": 0.7824959712198254,
+      "eval_mean_iou": 0.7270038262092244,
+      "eval_overall_accuracy": 0.9812071501319088,
+      "eval_runtime": 10.4732,
+      "eval_samples_per_second": 6.397,
+      "eval_steps_per_second": 0.477,
+      "step": 630
+    },
+    {
+      "epoch": 57.36,
+      "learning_rate": 1.7952153110047846e-06,
+      "loss": 0.1141,
+      "step": 631
+    },
+    {
+      "epoch": 57.45,
+      "learning_rate": 1.7913875598086124e-06,
+      "loss": 0.105,
+      "step": 632
+    },
+    {
+      "epoch": 57.55,
+      "learning_rate": 1.7875598086124401e-06,
+      "loss": 0.1156,
+      "step": 633
+    },
+    {
+      "epoch": 57.64,
+      "learning_rate": 1.7837320574162677e-06,
+      "loss": 0.0997,
+      "step": 634
+    },
+    {
+      "epoch": 57.73,
+      "learning_rate": 1.7799043062200956e-06,
+      "loss": 0.0994,
+      "step": 635
+    },
+    {
+      "epoch": 57.82,
+      "learning_rate": 1.7760765550239234e-06,
+      "loss": 0.103,
+      "step": 636
+    },
+    {
+      "epoch": 57.91,
+      "learning_rate": 1.7722488038277511e-06,
+      "loss": 0.1301,
+      "step": 637
+    },
+    {
+      "epoch": 58.0,
+      "learning_rate": 1.768421052631579e-06,
+      "loss": 0.1008,
+      "step": 638
+    },
+    {
+      "epoch": 58.09,
+      "learning_rate": 1.7645933014354067e-06,
+      "loss": 0.1128,
+      "step": 639
+    },
+    {
+      "epoch": 58.18,
+      "learning_rate": 1.7607655502392344e-06,
+      "loss": 0.1009,
+      "step": 640
+    },
+    {
+      "epoch": 58.18,
+      "eval_accuracy_safe": 0.45120489126873065,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9968523622918036,
+      "eval_iou_safe": 0.40891741330729336,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9804745786419761,
+      "eval_loss": 0.10278800129890442,
+      "eval_mean_accuracy": 0.7240286267802671,
+      "eval_mean_iou": 0.6946959959746347,
+      "eval_overall_accuracy": 0.9807348109003323,
+      "eval_runtime": 10.4926,
+      "eval_samples_per_second": 6.385,
+      "eval_steps_per_second": 0.477,
+      "step": 640
+    },
+    {
+      "epoch": 58.27,
+      "learning_rate": 1.756937799043062e-06,
+      "loss": 0.1043,
+      "step": 641
+    },
+    {
+      "epoch": 58.36,
+      "learning_rate": 1.75311004784689e-06,
+      "loss": 0.1104,
+      "step": 642
+    },
+    {
+      "epoch": 58.45,
+      "learning_rate": 1.7492822966507177e-06,
+      "loss": 0.0972,
+      "step": 643
+    },
+    {
+      "epoch": 58.55,
+      "learning_rate": 1.7454545454545452e-06,
+      "loss": 0.1032,
+      "step": 644
+    },
+    {
+      "epoch": 58.64,
+      "learning_rate": 1.7416267942583732e-06,
+      "loss": 0.1053,
+      "step": 645
+    },
+    {
+      "epoch": 58.73,
+      "learning_rate": 1.737799043062201e-06,
+      "loss": 0.1061,
+      "step": 646
+    },
+    {
+      "epoch": 58.82,
+      "learning_rate": 1.7339712918660285e-06,
+      "loss": 0.117,
+      "step": 647
+    },
+    {
+      "epoch": 58.91,
+      "learning_rate": 1.7301435406698565e-06,
+      "loss": 0.1165,
+      "step": 648
+    },
+    {
+      "epoch": 59.0,
+      "learning_rate": 1.7263157894736842e-06,
+      "loss": 0.0908,
+      "step": 649
+    },
+    {
+      "epoch": 59.09,
+      "learning_rate": 1.722488038277512e-06,
+      "loss": 0.1018,
+      "step": 650
+    },
+    {
+      "epoch": 59.09,
+      "eval_accuracy_safe": 0.6052867953477434,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9918838809103937,
+      "eval_iou_safe": 0.4778646863682016,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9801087838204473,
+      "eval_loss": 0.1022411584854126,
+      "eval_mean_accuracy": 0.7985853381290686,
+      "eval_mean_iou": 0.7289867350943244,
+      "eval_overall_accuracy": 0.9804644228807137,
+      "eval_runtime": 10.6805,
+      "eval_samples_per_second": 6.273,
+      "eval_steps_per_second": 0.468,
+      "step": 650
+    },
+    {
+      "epoch": 59.18,
+      "learning_rate": 1.7186602870813395e-06,
+      "loss": 0.0988,
+      "step": 651
+    },
+    {
+      "epoch": 59.27,
+      "learning_rate": 1.7148325358851675e-06,
+      "loss": 0.0946,
+      "step": 652
+    },
+    {
+      "epoch": 59.36,
+      "learning_rate": 1.7110047846889952e-06,
+      "loss": 0.1154,
+      "step": 653
+    },
+    {
+      "epoch": 59.45,
+      "learning_rate": 1.7071770334928228e-06,
+      "loss": 0.1091,
+      "step": 654
+    },
+    {
+      "epoch": 59.55,
+      "learning_rate": 1.7033492822966507e-06,
+      "loss": 0.1015,
+      "step": 655
+    },
+    {
+      "epoch": 59.64,
+      "learning_rate": 1.6995215311004785e-06,
+      "loss": 0.1099,
+      "step": 656
+    },
+    {
+      "epoch": 59.73,
+      "learning_rate": 1.695693779904306e-06,
+      "loss": 0.1035,
+      "step": 657
+    },
+    {
+      "epoch": 59.82,
+      "learning_rate": 1.6918660287081338e-06,
+      "loss": 0.0961,
+      "step": 658
+    },
+    {
+      "epoch": 59.91,
+      "learning_rate": 1.6880382775119618e-06,
+      "loss": 0.129,
+      "step": 659
+    },
+    {
+      "epoch": 60.0,
+      "learning_rate": 1.6842105263157893e-06,
+      "loss": 0.1012,
+      "step": 660
+    },
+    {
+      "epoch": 60.0,
+      "eval_accuracy_safe": 0.516669172439582,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9949175838843014,
+      "eval_iou_safe": 0.4427407912710119,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9804931878913242,
+      "eval_loss": 0.10163220763206482,
+      "eval_mean_accuracy": 0.7557933781619417,
+      "eval_mean_iou": 0.711616989581168,
+      "eval_overall_accuracy": 0.980790892643715,
+      "eval_runtime": 10.6583,
+      "eval_samples_per_second": 6.286,
+      "eval_steps_per_second": 0.469,
+      "step": 660
+    },
+    {
+      "epoch": 60.09,
+      "learning_rate": 1.680382775119617e-06,
+      "loss": 0.0998,
+      "step": 661
+    },
+    {
+      "epoch": 60.18,
+      "learning_rate": 1.676555023923445e-06,
+      "loss": 0.0925,
+      "step": 662
+    },
+    {
+      "epoch": 60.27,
+      "learning_rate": 1.6727272727272726e-06,
+      "loss": 0.1051,
+      "step": 663
+    },
+    {
+      "epoch": 60.36,
+      "learning_rate": 1.6688995215311003e-06,
+      "loss": 0.1022,
+      "step": 664
+    },
+    {
+      "epoch": 60.45,
+      "learning_rate": 1.665071770334928e-06,
+      "loss": 0.0945,
+      "step": 665
+    },
+    {
+      "epoch": 60.55,
+      "learning_rate": 1.661244019138756e-06,
+      "loss": 0.1073,
+      "step": 666
+    },
+    {
+      "epoch": 60.64,
+      "learning_rate": 1.6574162679425836e-06,
+      "loss": 0.095,
+      "step": 667
+    },
+    {
+      "epoch": 60.73,
+      "learning_rate": 1.6535885167464114e-06,
+      "loss": 0.1152,
+      "step": 668
+    },
+    {
+      "epoch": 60.82,
+      "learning_rate": 1.6497607655502393e-06,
+      "loss": 0.1185,
+      "step": 669
+    },
+    {
+      "epoch": 60.91,
+      "learning_rate": 1.6459330143540669e-06,
+      "loss": 0.1052,
+      "step": 670
+    },
+    {
+      "epoch": 60.91,
+      "eval_accuracy_safe": 0.5463510163800448,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9943183411572037,
+      "eval_iou_safe": 0.46040818978469733,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9807758590757052,
+      "eval_loss": 0.09994357079267502,
+      "eval_mean_accuracy": 0.7703346787686243,
+      "eval_mean_iou": 0.7205920244302013,
+      "eval_overall_accuracy": 0.9810861046634504,
+      "eval_runtime": 10.1258,
+      "eval_samples_per_second": 6.617,
+      "eval_steps_per_second": 0.494,
+      "step": 670
+    },
+    {
+      "epoch": 61.0,
+      "learning_rate": 1.6421052631578946e-06,
+      "loss": 0.0986,
+      "step": 671
+    },
+    {
+      "epoch": 61.09,
+      "learning_rate": 1.6382775119617226e-06,
+      "loss": 0.1114,
+      "step": 672
+    },
+    {
+      "epoch": 61.18,
+      "learning_rate": 1.6344497607655501e-06,
+      "loss": 0.1073,
+      "step": 673
+    },
+    {
+      "epoch": 61.27,
+      "learning_rate": 1.6306220095693779e-06,
+      "loss": 0.1028,
+      "step": 674
+    },
+    {
+      "epoch": 61.36,
+      "learning_rate": 1.6267942583732056e-06,
+      "loss": 0.0855,
+      "step": 675
+    },
+    {
+      "epoch": 61.45,
+      "learning_rate": 1.6229665071770334e-06,
+      "loss": 0.0954,
+      "step": 676
+    },
+    {
+      "epoch": 61.55,
+      "learning_rate": 1.6191387559808612e-06,
+      "loss": 0.105,
+      "step": 677
+    },
+    {
+      "epoch": 61.64,
+      "learning_rate": 1.615311004784689e-06,
+      "loss": 0.105,
+      "step": 678
+    },
+    {
+      "epoch": 61.73,
+      "learning_rate": 1.6114832535885169e-06,
+      "loss": 0.0932,
+      "step": 679
+    },
+    {
+      "epoch": 61.82,
+      "learning_rate": 1.6076555023923444e-06,
+      "loss": 0.1229,
+      "step": 680
+    },
+    {
+      "epoch": 61.82,
+      "eval_accuracy_safe": 0.5706261733763556,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.993868645102455,
+      "eval_iou_safe": 0.4749514687715583,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9810472858831734,
+      "eval_loss": 0.09930147975683212,
+      "eval_mean_accuracy": 0.7822474092394053,
+      "eval_mean_iou": 0.7279993773273659,
+      "eval_overall_accuracy": 0.9813667411234841,
+      "eval_runtime": 10.8251,
+      "eval_samples_per_second": 6.189,
+      "eval_steps_per_second": 0.462,
+      "step": 680
+    },
+    {
+      "epoch": 61.91,
+      "learning_rate": 1.6038277511961722e-06,
+      "loss": 0.1183,
+      "step": 681
+    },
+    {
+      "epoch": 62.0,
+      "learning_rate": 1.6e-06,
+      "loss": 0.1107,
+      "step": 682
+    },
+    {
+      "epoch": 62.09,
+      "learning_rate": 1.5961722488038277e-06,
+      "loss": 0.1028,
+      "step": 683
+    },
+    {
+      "epoch": 62.18,
+      "learning_rate": 1.5923444976076554e-06,
+      "loss": 0.097,
+      "step": 684
+    },
+    {
+      "epoch": 62.27,
+      "learning_rate": 1.5885167464114832e-06,
+      "loss": 0.0915,
+      "step": 685
+    },
+    {
+      "epoch": 62.36,
+      "learning_rate": 1.584688995215311e-06,
+      "loss": 0.0974,
+      "step": 686
+    },
+    {
+      "epoch": 62.45,
+      "learning_rate": 1.5808612440191387e-06,
+      "loss": 0.0997,
+      "step": 687
+    },
+    {
+      "epoch": 62.55,
+      "learning_rate": 1.5770334928229665e-06,
+      "loss": 0.1126,
+      "step": 688
+    },
+    {
+      "epoch": 62.64,
+      "learning_rate": 1.573205741626794e-06,
+      "loss": 0.1003,
+      "step": 689
+    },
+    {
+      "epoch": 62.73,
+      "learning_rate": 1.569377990430622e-06,
+      "loss": 0.0963,
+      "step": 690
+    },
+    {
+      "epoch": 62.73,
+      "eval_accuracy_safe": 0.5745833670648918,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9936493412730159,
+      "eval_iou_safe": 0.475394228193057,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9809474391002709,
+      "eval_loss": 0.09741559624671936,
+      "eval_mean_accuracy": 0.7841163541689538,
+      "eval_mean_iou": 0.7281708336466639,
+      "eval_overall_accuracy": 0.9812708043340427,
+      "eval_runtime": 10.6643,
+      "eval_samples_per_second": 6.283,
+      "eval_steps_per_second": 0.469,
+      "step": 690
+    },
+    {
+      "epoch": 62.82,
+      "learning_rate": 1.5655502392344497e-06,
+      "loss": 0.1206,
+      "step": 691
+    },
+    {
+      "epoch": 62.91,
+      "learning_rate": 1.5617224880382773e-06,
+      "loss": 0.1127,
+      "step": 692
+    },
+    {
+      "epoch": 63.0,
+      "learning_rate": 1.5578947368421052e-06,
+      "loss": 0.0828,
+      "step": 693
+    },
+    {
+      "epoch": 63.09,
+      "learning_rate": 1.554066985645933e-06,
+      "loss": 0.0967,
+      "step": 694
+    },
+    {
+      "epoch": 63.18,
+      "learning_rate": 1.5502392344497607e-06,
+      "loss": 0.1018,
+      "step": 695
+    },
+    {
+      "epoch": 63.27,
+      "learning_rate": 1.5464114832535883e-06,
+      "loss": 0.0982,
+      "step": 696
+    },
+    {
+      "epoch": 63.36,
+      "learning_rate": 1.5425837320574163e-06,
+      "loss": 0.1014,
+      "step": 697
+    },
+    {
+      "epoch": 63.45,
+      "learning_rate": 1.538755980861244e-06,
+      "loss": 0.0924,
+      "step": 698
+    },
+    {
+      "epoch": 63.55,
+      "learning_rate": 1.5349282296650716e-06,
+      "loss": 0.0919,
+      "step": 699
+    },
+    {
+      "epoch": 63.64,
+      "learning_rate": 1.5311004784688995e-06,
+      "loss": 0.1115,
+      "step": 700
+    },
+    {
+      "epoch": 63.64,
+      "eval_accuracy_safe": 0.5239243487881696,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9954837374300712,
+      "eval_iou_safe": 0.45622972199516265,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9812646835919959,
+      "eval_loss": 0.09740366041660309,
+      "eval_mean_accuracy": 0.7597040431091204,
+      "eval_mean_iou": 0.7187472027935793,
+      "eval_overall_accuracy": 0.9815546291977612,
+      "eval_runtime": 9.9953,
+      "eval_samples_per_second": 6.703,
+      "eval_steps_per_second": 0.5,
+      "step": 700
+    },
+    {
+      "epoch": 63.73,
+      "learning_rate": 1.5272727272727273e-06,
+      "loss": 0.0956,
+      "step": 701
+    },
+    {
+      "epoch": 63.82,
+      "learning_rate": 1.5234449760765548e-06,
+      "loss": 0.1093,
+      "step": 702
+    },
+    {
+      "epoch": 63.91,
+      "learning_rate": 1.5196172248803828e-06,
+      "loss": 0.1099,
+      "step": 703
+    },
+    {
+      "epoch": 64.0,
+      "learning_rate": 1.5157894736842105e-06,
+      "loss": 0.0971,
+      "step": 704
+    },
+    {
+      "epoch": 64.09,
+      "learning_rate": 1.511961722488038e-06,
+      "loss": 0.1028,
+      "step": 705
+    },
+    {
+      "epoch": 64.18,
+      "learning_rate": 1.5081339712918658e-06,
+      "loss": 0.0902,
+      "step": 706
+    },
+    {
+      "epoch": 64.27,
+      "learning_rate": 1.5043062200956938e-06,
+      "loss": 0.1065,
+      "step": 707
+    },
+    {
+      "epoch": 64.36,
+      "learning_rate": 1.5004784688995216e-06,
+      "loss": 0.0981,
+      "step": 708
+    },
+    {
+      "epoch": 64.45,
+      "learning_rate": 1.4966507177033491e-06,
+      "loss": 0.0957,
+      "step": 709
+    },
+    {
+      "epoch": 64.55,
+      "learning_rate": 1.492822966507177e-06,
+      "loss": 0.1025,
+      "step": 710
+    },
+    {
+      "epoch": 64.55,
+      "eval_accuracy_safe": 0.5844753875274189,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9934714575889979,
+      "eval_iou_safe": 0.4812515871000508,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9810634398778569,
+      "eval_loss": 0.09636449068784714,
+      "eval_mean_accuracy": 0.7889734225582083,
+      "eval_mean_iou": 0.7311575134889539,
+      "eval_overall_accuracy": 0.981390369472219,
+      "eval_runtime": 10.8756,
+      "eval_samples_per_second": 6.161,
+      "eval_steps_per_second": 0.46,
+      "step": 710
+    },
+    {
+      "epoch": 64.64,
+      "learning_rate": 1.4889952153110048e-06,
+      "loss": 0.1116,
+      "step": 711
+    },
+    {
+      "epoch": 64.73,
+      "learning_rate": 1.4851674641148324e-06,
+      "loss": 0.0946,
+      "step": 712
+    },
+    {
+      "epoch": 64.82,
+      "learning_rate": 1.4813397129186601e-06,
+      "loss": 0.1029,
+      "step": 713
+    },
+    {
+      "epoch": 64.91,
+      "learning_rate": 1.477511961722488e-06,
+      "loss": 0.1018,
+      "step": 714
+    },
+    {
+      "epoch": 65.0,
+      "learning_rate": 1.4736842105263156e-06,
+      "loss": 0.0875,
+      "step": 715
+    },
+    {
+      "epoch": 65.09,
+      "learning_rate": 1.4698564593301434e-06,
+      "loss": 0.114,
+      "step": 716
+    },
+    {
+      "epoch": 65.18,
+      "learning_rate": 1.4660287081339714e-06,
+      "loss": 0.0997,
+      "step": 717
+    },
+    {
+      "epoch": 65.27,
+      "learning_rate": 1.462200956937799e-06,
+      "loss": 0.1042,
+      "step": 718
+    },
+    {
+      "epoch": 65.36,
+      "learning_rate": 1.4583732057416267e-06,
+      "loss": 0.1002,
+      "step": 719
+    },
+    {
+      "epoch": 65.45,
+      "learning_rate": 1.4545454545454544e-06,
+      "loss": 0.0916,
+      "step": 720
+    },
+    {
+      "epoch": 65.45,
+      "eval_accuracy_safe": 0.5493309586316167,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9947474444767644,
+      "eval_iou_safe": 0.4684850846003245,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9812869104271664,
+      "eval_loss": 0.09620904177427292,
+      "eval_mean_accuracy": 0.7720392015541906,
+      "eval_mean_iou": 0.7248859975137455,
+      "eval_overall_accuracy": 0.9815905556749942,
+      "eval_runtime": 10.4119,
+      "eval_samples_per_second": 6.435,
+      "eval_steps_per_second": 0.48,
+      "step": 720
+    },
+    {
+      "epoch": 65.55,
+      "learning_rate": 1.4507177033492824e-06,
+      "loss": 0.0908,
+      "step": 721
+    },
+    {
+      "epoch": 65.64,
+      "learning_rate": 1.44688995215311e-06,
+      "loss": 0.1045,
+      "step": 722
+    },
+    {
+      "epoch": 65.73,
+      "learning_rate": 1.4430622009569377e-06,
+      "loss": 0.0971,
+      "step": 723
+    },
+    {
+      "epoch": 65.82,
+      "learning_rate": 1.4392344497607657e-06,
+      "loss": 0.0876,
+      "step": 724
+    },
+    {
+      "epoch": 65.91,
+      "learning_rate": 1.4354066985645932e-06,
+      "loss": 0.1065,
+      "step": 725
+    },
+    {
+      "epoch": 66.0,
+      "learning_rate": 1.431578947368421e-06,
+      "loss": 0.0998,
+      "step": 726
+    },
+    {
+      "epoch": 66.09,
+      "learning_rate": 1.427751196172249e-06,
+      "loss": 0.0952,
+      "step": 727
+    },
+    {
+      "epoch": 66.18,
+      "learning_rate": 1.4239234449760765e-06,
+      "loss": 0.1067,
+      "step": 728
+    },
+    {
+      "epoch": 66.27,
+      "learning_rate": 1.4200956937799042e-06,
+      "loss": 0.0991,
+      "step": 729
+    },
+    {
+      "epoch": 66.36,
+      "learning_rate": 1.416267942583732e-06,
+      "loss": 0.1055,
+      "step": 730
+    },
+    {
+      "epoch": 66.36,
+      "eval_accuracy_safe": 0.5273225623648329,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9953211662927315,
+      "eval_iou_safe": 0.45706304548309923,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9812044738737931,
+      "eval_loss": 0.09465456008911133,
+      "eval_mean_accuracy": 0.7613218643287822,
+      "eval_mean_iou": 0.7191337596784462,
+      "eval_overall_accuracy": 0.9814972379314366,
+      "eval_runtime": 10.9178,
+      "eval_samples_per_second": 6.137,
+      "eval_steps_per_second": 0.458,
+      "step": 730
+    },
+    {
+      "epoch": 66.45,
+      "learning_rate": 1.4124401913875597e-06,
+      "loss": 0.0913,
+      "step": 731
+    },
+    {
+      "epoch": 66.55,
+      "learning_rate": 1.4086124401913875e-06,
+      "loss": 0.0915,
+      "step": 732
+    },
+    {
+      "epoch": 66.64,
+      "learning_rate": 1.4047846889952152e-06,
+      "loss": 0.094,
+      "step": 733
+    },
+    {
+      "epoch": 66.73,
+      "learning_rate": 1.400956937799043e-06,
+      "loss": 0.0836,
+      "step": 734
+    },
+    {
+      "epoch": 66.82,
+      "learning_rate": 1.3971291866028708e-06,
+      "loss": 0.095,
+      "step": 735
+    },
+    {
+      "epoch": 66.91,
+      "learning_rate": 1.3933014354066985e-06,
+      "loss": 0.1146,
+      "step": 736
+    },
+    {
+      "epoch": 67.0,
+      "learning_rate": 1.3894736842105263e-06,
+      "loss": 0.085,
+      "step": 737
+    },
+    {
+      "epoch": 67.09,
+      "learning_rate": 1.385645933014354e-06,
+      "loss": 0.0946,
+      "step": 738
+    },
+    {
+      "epoch": 67.18,
+      "learning_rate": 1.3818181818181818e-06,
+      "loss": 0.1043,
+      "step": 739
+    },
+    {
+      "epoch": 67.27,
+      "learning_rate": 1.3779904306220095e-06,
+      "loss": 0.1081,
+      "step": 740
+    },
+    {
+      "epoch": 67.27,
+      "eval_accuracy_safe": 0.6093114521532299,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9919017748825657,
+      "eval_iou_safe": 0.4812654624068449,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9802451202305071,
+      "eval_loss": 0.09639958292245865,
+      "eval_mean_accuracy": 0.8006066135178977,
+      "eval_mean_iou": 0.730755291318676,
+      "eval_overall_accuracy": 0.9806006702024546,
+      "eval_runtime": 10.5244,
+      "eval_samples_per_second": 6.366,
+      "eval_steps_per_second": 0.475,
+      "step": 740
+    },
+    {
+      "epoch": 67.36,
+      "learning_rate": 1.3741626794258373e-06,
+      "loss": 0.0912,
+      "step": 741
+    },
+    {
+      "epoch": 67.45,
+      "learning_rate": 1.370334928229665e-06,
+      "loss": 0.0936,
+      "step": 742
+    },
+    {
+      "epoch": 67.55,
+      "learning_rate": 1.3665071770334928e-06,
+      "loss": 0.095,
+      "step": 743
+    },
+    {
+      "epoch": 67.64,
+      "learning_rate": 1.3626794258373203e-06,
+      "loss": 0.0994,
+      "step": 744
+    },
+    {
+      "epoch": 67.73,
+      "learning_rate": 1.3588516746411483e-06,
+      "loss": 0.0947,
+      "step": 745
+    },
+    {
+      "epoch": 67.82,
+      "learning_rate": 1.355023923444976e-06,
+      "loss": 0.1102,
+      "step": 746
+    },
+    {
+      "epoch": 67.91,
+      "learning_rate": 1.3511961722488036e-06,
+      "loss": 0.0852,
+      "step": 747
+    },
+    {
+      "epoch": 68.0,
+      "learning_rate": 1.3473684210526316e-06,
+      "loss": 0.1126,
+      "step": 748
+    },
+    {
+      "epoch": 68.09,
+      "learning_rate": 1.3435406698564593e-06,
+      "loss": 0.0905,
+      "step": 749
+    },
+    {
+      "epoch": 68.18,
+      "learning_rate": 1.339712918660287e-06,
+      "loss": 0.1039,
+      "step": 750
+    },
+    {
+      "epoch": 68.18,
+      "eval_accuracy_safe": 0.540481725205377,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9944637223475061,
+      "eval_iou_safe": 0.45730282288511936,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9807464396876859,
+      "eval_loss": 0.09501567482948303,
+      "eval_mean_accuracy": 0.7674727237764416,
+      "eval_mean_iou": 0.7190246312864026,
+      "eval_overall_accuracy": 0.9810538220761428,
+      "eval_runtime": 10.2387,
+      "eval_samples_per_second": 6.544,
+      "eval_steps_per_second": 0.488,
+      "step": 750
+    },
+    {
+      "epoch": 68.27,
+      "learning_rate": 1.3358851674641146e-06,
+      "loss": 0.0886,
+      "step": 751
+    },
+    {
+      "epoch": 68.36,
+      "learning_rate": 1.3320574162679426e-06,
+      "loss": 0.0977,
+      "step": 752
+    },
+    {
+      "epoch": 68.45,
+      "learning_rate": 1.3282296650717704e-06,
+      "loss": 0.0934,
+      "step": 753
+    },
+    {
+      "epoch": 68.55,
+      "learning_rate": 1.324401913875598e-06,
+      "loss": 0.0906,
+      "step": 754
+    },
+    {
+      "epoch": 68.64,
+      "learning_rate": 1.3205741626794259e-06,
+      "loss": 0.1029,
+      "step": 755
+    },
+    {
+      "epoch": 68.73,
+      "learning_rate": 1.3167464114832536e-06,
+      "loss": 0.1015,
+      "step": 756
+    },
+    {
+      "epoch": 68.82,
+      "learning_rate": 1.3129186602870812e-06,
+      "loss": 0.0967,
+      "step": 757
+    },
+    {
+      "epoch": 68.91,
+      "learning_rate": 1.3090909090909091e-06,
+      "loss": 0.0941,
+      "step": 758
+    },
+    {
+      "epoch": 69.0,
+      "learning_rate": 1.3052631578947369e-06,
+      "loss": 0.1303,
+      "step": 759
+    },
+    {
+      "epoch": 69.09,
+      "learning_rate": 1.3014354066985644e-06,
+      "loss": 0.106,
+      "step": 760
+    },
+    {
+      "epoch": 69.09,
+      "eval_accuracy_safe": 0.5563972382527438,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9942656566096285,
+      "eval_iou_safe": 0.46819118861133774,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9810197842105592,
+      "eval_loss": 0.09392161667346954,
+      "eval_mean_accuracy": 0.7753314474311861,
+      "eval_mean_iou": 0.7246054864109485,
+      "eval_overall_accuracy": 0.9813317256187325,
+      "eval_runtime": 11.1215,
+      "eval_samples_per_second": 6.024,
+      "eval_steps_per_second": 0.45,
+      "step": 760
+    },
+    {
+      "epoch": 69.18,
+      "learning_rate": 1.2976076555023922e-06,
+      "loss": 0.097,
+      "step": 761
+    },
+    {
+      "epoch": 69.27,
+      "learning_rate": 1.2937799043062201e-06,
+      "loss": 0.0903,
+      "step": 762
+    },
+    {
+      "epoch": 69.36,
+      "learning_rate": 1.2899521531100477e-06,
+      "loss": 0.1096,
+      "step": 763
+    },
+    {
+      "epoch": 69.45,
+      "learning_rate": 1.2861244019138754e-06,
+      "loss": 0.116,
+      "step": 764
+    },
+    {
+      "epoch": 69.55,
+      "learning_rate": 1.2822966507177034e-06,
+      "loss": 0.0892,
+      "step": 765
+    },
+    {
+      "epoch": 69.64,
+      "learning_rate": 1.2784688995215312e-06,
+      "loss": 0.0923,
+      "step": 766
+    },
+    {
+      "epoch": 69.73,
+      "learning_rate": 1.2746411483253587e-06,
+      "loss": 0.09,
+      "step": 767
+    },
+    {
+      "epoch": 69.82,
+      "learning_rate": 1.2708133971291865e-06,
+      "loss": 0.0939,
+      "step": 768
+    },
+    {
+      "epoch": 69.91,
+      "learning_rate": 1.2669856459330144e-06,
+      "loss": 0.0934,
+      "step": 769
+    },
+    {
+      "epoch": 70.0,
+      "learning_rate": 1.263157894736842e-06,
+      "loss": 0.0912,
+      "step": 770
+    },
+    {
+      "epoch": 70.0,
+      "eval_accuracy_safe": 0.5376868246460114,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9949495583591662,
+      "eval_iou_safe": 0.46116623628253917,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9811432599765284,
+      "eval_loss": 0.0935571938753128,
+      "eval_mean_accuracy": 0.7663181915025887,
+      "eval_mean_iou": 0.7211547481295337,
+      "eval_overall_accuracy": 0.9814427503898963,
+      "eval_runtime": 12.6589,
+      "eval_samples_per_second": 5.293,
+      "eval_steps_per_second": 0.395,
+      "step": 770
+    },
+    {
+      "epoch": 70.09,
+      "learning_rate": 1.2593301435406697e-06,
+      "loss": 0.1019,
+      "step": 771
+    },
+    {
+      "epoch": 70.18,
+      "learning_rate": 1.2555023923444977e-06,
+      "loss": 0.1146,
+      "step": 772
+    },
+    {
+      "epoch": 70.27,
+      "learning_rate": 1.2516746411483252e-06,
+      "loss": 0.0883,
+      "step": 773
+    },
+    {
+      "epoch": 70.36,
+      "learning_rate": 1.247846889952153e-06,
+      "loss": 0.0854,
+      "step": 774
+    },
+    {
+      "epoch": 70.45,
+      "learning_rate": 1.2440191387559808e-06,
+      "loss": 0.0956,
+      "step": 775
+    },
+    {
+      "epoch": 70.55,
+      "learning_rate": 1.2401913875598085e-06,
+      "loss": 0.101,
+      "step": 776
+    },
+    {
+      "epoch": 70.64,
+      "learning_rate": 1.2363636363636363e-06,
+      "loss": 0.1062,
+      "step": 777
+    },
+    {
+      "epoch": 70.73,
+      "learning_rate": 1.232535885167464e-06,
+      "loss": 0.0888,
+      "step": 778
+    },
+    {
+      "epoch": 70.82,
+      "learning_rate": 1.228708133971292e-06,
+      "loss": 0.0945,
+      "step": 779
+    },
+    {
+      "epoch": 70.91,
+      "learning_rate": 1.2248803827751195e-06,
+      "loss": 0.0951,
+      "step": 780
+    },
+    {
+      "epoch": 70.91,
+      "eval_accuracy_safe": 0.5600498841561906,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.994080145986652,
+      "eval_iou_safe": 0.46886018347439506,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9809443508467186,
+      "eval_loss": 0.09377636760473251,
+      "eval_mean_accuracy": 0.7770650150714213,
+      "eval_mean_iou": 0.7249022671605568,
+      "eval_overall_accuracy": 0.9812595879853662,
+      "eval_runtime": 10.9659,
+      "eval_samples_per_second": 6.11,
+      "eval_steps_per_second": 0.456,
+      "step": 780
+    },
+    {
+      "epoch": 71.0,
+      "learning_rate": 1.2210526315789473e-06,
+      "loss": 0.1009,
+      "step": 781
+    },
+    {
+      "epoch": 71.09,
+      "learning_rate": 1.2172248803827753e-06,
+      "loss": 0.089,
+      "step": 782
+    },
+    {
+      "epoch": 71.18,
+      "learning_rate": 1.2133971291866028e-06,
+      "loss": 0.1066,
+      "step": 783
+    },
+    {
+      "epoch": 71.27,
+      "learning_rate": 1.2095693779904306e-06,
+      "loss": 0.0972,
+      "step": 784
+    },
+    {
+      "epoch": 71.36,
+      "learning_rate": 1.2057416267942583e-06,
+      "loss": 0.0874,
+      "step": 785
+    },
+    {
+      "epoch": 71.45,
+      "learning_rate": 1.201913875598086e-06,
+      "loss": 0.0933,
+      "step": 786
+    },
+    {
+      "epoch": 71.55,
+      "learning_rate": 1.1980861244019138e-06,
+      "loss": 0.0909,
+      "step": 787
+    },
+    {
+      "epoch": 71.64,
+      "learning_rate": 1.1942583732057416e-06,
+      "loss": 0.1034,
+      "step": 788
+    },
+    {
+      "epoch": 71.73,
+      "learning_rate": 1.1904306220095693e-06,
+      "loss": 0.1078,
+      "step": 789
+    },
+    {
+      "epoch": 71.82,
+      "learning_rate": 1.186602870813397e-06,
+      "loss": 0.0998,
+      "step": 790
+    },
+    {
+      "epoch": 71.82,
+      "eval_accuracy_safe": 0.5573243742314024,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.994380119362768,
+      "eval_iou_safe": 0.4704600775799229,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9811600412098591,
+      "eval_loss": 0.09279777854681015,
+      "eval_mean_accuracy": 0.7758522467970852,
+      "eval_mean_iou": 0.7258100593948911,
+      "eval_overall_accuracy": 0.9814701934358967,
+      "eval_runtime": 10.7411,
+      "eval_samples_per_second": 6.238,
+      "eval_steps_per_second": 0.466,
+      "step": 790
+    },
+    {
+      "epoch": 71.91,
+      "learning_rate": 1.1827751196172248e-06,
+      "loss": 0.0833,
+      "step": 791
+    },
+    {
+      "epoch": 72.0,
+      "learning_rate": 1.1789473684210524e-06,
+      "loss": 0.1115,
+      "step": 792
+    },
+    {
+      "epoch": 72.09,
+      "learning_rate": 1.1751196172248804e-06,
+      "loss": 0.105,
+      "step": 793
+    },
+    {
+      "epoch": 72.18,
+      "learning_rate": 1.1712918660287081e-06,
+      "loss": 0.0938,
+      "step": 794
+    },
+    {
+      "epoch": 72.27,
+      "learning_rate": 1.1674641148325359e-06,
+      "loss": 0.0878,
+      "step": 795
+    },
+    {
+      "epoch": 72.36,
+      "learning_rate": 1.1636363636363636e-06,
+      "loss": 0.0952,
+      "step": 796
+    },
+    {
+      "epoch": 72.45,
+      "learning_rate": 1.1598086124401914e-06,
+      "loss": 0.0866,
+      "step": 797
+    },
+    {
+      "epoch": 72.55,
+      "learning_rate": 1.1559808612440191e-06,
+      "loss": 0.0936,
+      "step": 798
+    },
+    {
+      "epoch": 72.64,
+      "learning_rate": 1.1521531100478467e-06,
+      "loss": 0.0865,
+      "step": 799
+    },
+    {
+      "epoch": 72.73,
+      "learning_rate": 1.1483253588516746e-06,
+      "loss": 0.0889,
+      "step": 800
+    },
+    {
+      "epoch": 72.73,
+      "eval_accuracy_safe": 0.5398032390006207,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9949394086634752,
+      "eval_iou_safe": 0.4628490775279766,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9811955809893045,
+      "eval_loss": 0.09308235347270966,
+      "eval_mean_accuracy": 0.7673713238320479,
+      "eval_mean_iou": 0.7220223292586405,
+      "eval_overall_accuracy": 0.9814954159864738,
+      "eval_runtime": 10.6601,
+      "eval_samples_per_second": 6.285,
+      "eval_steps_per_second": 0.469,
+      "step": 800
+    },
+    {
+      "epoch": 72.82,
+      "learning_rate": 1.1444976076555024e-06,
+      "loss": 0.1,
+      "step": 801
+    },
+    {
+      "epoch": 72.91,
+      "learning_rate": 1.14066985645933e-06,
+      "loss": 0.091,
+      "step": 802
+    },
+    {
+      "epoch": 73.0,
+      "learning_rate": 1.136842105263158e-06,
+      "loss": 0.0853,
+      "step": 803
+    },
+    {
+      "epoch": 73.09,
+      "learning_rate": 1.1330143540669857e-06,
+      "loss": 0.0945,
+      "step": 804
+    },
+    {
+      "epoch": 73.18,
+      "learning_rate": 1.1291866028708132e-06,
+      "loss": 0.1075,
+      "step": 805
+    },
+    {
+      "epoch": 73.27,
+      "learning_rate": 1.125358851674641e-06,
+      "loss": 0.0914,
+      "step": 806
+    },
+    {
+      "epoch": 73.36,
+      "learning_rate": 1.121531100478469e-06,
+      "loss": 0.0907,
+      "step": 807
+    },
+    {
+      "epoch": 73.45,
+      "learning_rate": 1.1177033492822967e-06,
+      "loss": 0.1007,
+      "step": 808
+    },
+    {
+      "epoch": 73.55,
+      "learning_rate": 1.1138755980861242e-06,
+      "loss": 0.0926,
+      "step": 809
+    },
+    {
+      "epoch": 73.64,
+      "learning_rate": 1.1100478468899522e-06,
+      "loss": 0.0906,
+      "step": 810
+    },
+    {
+      "epoch": 73.64,
+      "eval_accuracy_safe": 0.5151445059965073,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9958092903860791,
+      "eval_iou_safe": 0.4528016102142886,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813270861723169,
+      "eval_loss": 0.09284604340791702,
+      "eval_mean_accuracy": 0.7554768981912932,
+      "eval_mean_iou": 0.7170643481933028,
+      "eval_overall_accuracy": 0.9816112233631646,
+      "eval_runtime": 10.255,
+      "eval_samples_per_second": 6.533,
+      "eval_steps_per_second": 0.488,
+      "step": 810
+    },
+    {
+      "epoch": 73.73,
+      "learning_rate": 1.10622009569378e-06,
+      "loss": 0.1001,
+      "step": 811
+    },
+    {
+      "epoch": 73.82,
+      "learning_rate": 1.1023923444976075e-06,
+      "loss": 0.0837,
+      "step": 812
+    },
+    {
+      "epoch": 73.91,
+      "learning_rate": 1.0985645933014355e-06,
+      "loss": 0.0877,
+      "step": 813
+    },
+    {
+      "epoch": 74.0,
+      "learning_rate": 1.0947368421052632e-06,
+      "loss": 0.214,
+      "step": 814
+    },
+    {
+      "epoch": 74.09,
+      "learning_rate": 1.0909090909090908e-06,
+      "loss": 0.0845,
+      "step": 815
+    },
+    {
+      "epoch": 74.18,
+      "learning_rate": 1.0870813397129185e-06,
+      "loss": 0.0832,
+      "step": 816
+    },
+    {
+      "epoch": 74.27,
+      "learning_rate": 1.0832535885167465e-06,
+      "loss": 0.1173,
+      "step": 817
+    },
+    {
+      "epoch": 74.36,
+      "learning_rate": 1.079425837320574e-06,
+      "loss": 0.0883,
+      "step": 818
+    },
+    {
+      "epoch": 74.45,
+      "learning_rate": 1.0755980861244018e-06,
+      "loss": 0.0978,
+      "step": 819
+    },
+    {
+      "epoch": 74.55,
+      "learning_rate": 1.0717703349282298e-06,
+      "loss": 0.0911,
+      "step": 820
+    },
+    {
+      "epoch": 74.55,
+      "eval_accuracy_safe": 0.5682264139305554,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9938084509534436,
+      "eval_iou_safe": 0.47217684067553367,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9809171438215548,
+      "eval_loss": 0.09244522452354431,
+      "eval_mean_accuracy": 0.7810174324419995,
+      "eval_mean_iou": 0.7265469922485442,
+      "eval_overall_accuracy": 0.9812374399669135,
+      "eval_runtime": 10.4305,
+      "eval_samples_per_second": 6.423,
+      "eval_steps_per_second": 0.479,
+      "step": 820
+    },
+    {
+      "epoch": 74.64,
+      "learning_rate": 1.0679425837320575e-06,
+      "loss": 0.086,
+      "step": 821
+    },
+    {
+      "epoch": 74.73,
+      "learning_rate": 1.064114832535885e-06,
+      "loss": 0.0892,
+      "step": 822
+    },
+    {
+      "epoch": 74.82,
+      "learning_rate": 1.0602870813397128e-06,
+      "loss": 0.1048,
+      "step": 823
+    },
+    {
+      "epoch": 74.91,
+      "learning_rate": 1.0564593301435408e-06,
+      "loss": 0.0882,
+      "step": 824
+    },
+    {
+      "epoch": 75.0,
+      "learning_rate": 1.0526315789473683e-06,
+      "loss": 0.182,
+      "step": 825
+    },
+    {
+      "epoch": 75.09,
+      "learning_rate": 1.048803827751196e-06,
+      "loss": 0.0897,
+      "step": 826
+    },
+    {
+      "epoch": 75.18,
+      "learning_rate": 1.044976076555024e-06,
+      "loss": 0.0971,
+      "step": 827
+    },
+    {
+      "epoch": 75.27,
+      "learning_rate": 1.0411483253588516e-06,
+      "loss": 0.0932,
+      "step": 828
+    },
+    {
+      "epoch": 75.36,
+      "learning_rate": 1.0373205741626793e-06,
+      "loss": 0.0946,
+      "step": 829
+    },
+    {
+      "epoch": 75.45,
+      "learning_rate": 1.033492822966507e-06,
+      "loss": 0.0907,
+      "step": 830
+    },
+    {
+      "epoch": 75.45,
+      "eval_accuracy_safe": 0.48640714569334736,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9965232892101226,
+      "eval_iou_safe": 0.43654313897125224,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.981184947472746,
+      "eval_loss": 0.09287872165441513,
+      "eval_mean_accuracy": 0.741465217451735,
+      "eval_mean_iou": 0.7088640432219991,
+      "eval_overall_accuracy": 0.9814552762615147,
+      "eval_runtime": 10.9268,
+      "eval_samples_per_second": 6.132,
+      "eval_steps_per_second": 0.458,
+      "step": 830
+    },
+    {
+      "epoch": 75.55,
+      "learning_rate": 1.0296650717703349e-06,
+      "loss": 0.0957,
+      "step": 831
+    },
+    {
+      "epoch": 75.64,
+      "learning_rate": 1.0258373205741626e-06,
+      "loss": 0.0908,
+      "step": 832
+    },
+    {
+      "epoch": 75.73,
+      "learning_rate": 1.0220095693779904e-06,
+      "loss": 0.113,
+      "step": 833
+    },
+    {
+      "epoch": 75.82,
+      "learning_rate": 1.0181818181818181e-06,
+      "loss": 0.0981,
+      "step": 834
+    },
+    {
+      "epoch": 75.91,
+      "learning_rate": 1.0143540669856459e-06,
+      "loss": 0.0899,
+      "step": 835
+    },
+    {
+      "epoch": 76.0,
+      "learning_rate": 1.0105263157894736e-06,
+      "loss": 0.0954,
+      "step": 836
+    },
+    {
+      "epoch": 76.09,
+      "learning_rate": 1.0066985645933016e-06,
+      "loss": 0.0948,
+      "step": 837
+    },
+    {
+      "epoch": 76.18,
+      "learning_rate": 1.0028708133971291e-06,
+      "loss": 0.086,
+      "step": 838
+    },
+    {
+      "epoch": 76.27,
+      "learning_rate": 9.990430622009569e-07,
+      "loss": 0.094,
+      "step": 839
+    },
+    {
+      "epoch": 76.36,
+      "learning_rate": 9.952153110047846e-07,
+      "loss": 0.1117,
+      "step": 840
+    },
+    {
+      "epoch": 76.36,
+      "eval_accuracy_safe": 0.5239320588586782,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9955866424372505,
+      "eval_iou_safe": 0.4575835778508384,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813663457617646,
+      "eval_loss": 0.09342558681964874,
+      "eval_mean_accuracy": 0.7597593506479643,
+      "eval_mean_iou": 0.7194749618063015,
+      "eval_overall_accuracy": 0.9816547222991487,
+      "eval_runtime": 10.5916,
+      "eval_samples_per_second": 6.326,
+      "eval_steps_per_second": 0.472,
+      "step": 840
+    },
+    {
+      "epoch": 76.45,
+      "learning_rate": 9.913875598086124e-07,
+      "loss": 0.0889,
+      "step": 841
+    },
+    {
+      "epoch": 76.55,
+      "learning_rate": 9.875598086124402e-07,
+      "loss": 0.0938,
+      "step": 842
+    },
+    {
+      "epoch": 76.64,
+      "learning_rate": 9.83732057416268e-07,
+      "loss": 0.0828,
+      "step": 843
+    },
+    {
+      "epoch": 76.73,
+      "learning_rate": 9.799043062200957e-07,
+      "loss": 0.0845,
+      "step": 844
+    },
+    {
+      "epoch": 76.82,
+      "learning_rate": 9.760765550239234e-07,
+      "loss": 0.1108,
+      "step": 845
+    },
+    {
+      "epoch": 76.91,
+      "learning_rate": 9.722488038277512e-07,
+      "loss": 0.1016,
+      "step": 846
+    },
+    {
+      "epoch": 77.0,
+      "learning_rate": 9.68421052631579e-07,
+      "loss": 0.1083,
+      "step": 847
+    },
+    {
+      "epoch": 77.09,
+      "learning_rate": 9.645933014354067e-07,
+      "loss": 0.0898,
+      "step": 848
+    },
+    {
+      "epoch": 77.18,
+      "learning_rate": 9.607655502392344e-07,
+      "loss": 0.0927,
+      "step": 849
+    },
+    {
+      "epoch": 77.27,
+      "learning_rate": 9.569377990430622e-07,
+      "loss": 0.0812,
+      "step": 850
+    },
+    {
+      "epoch": 77.27,
+      "eval_accuracy_safe": 0.5278603397828073,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9955512065054739,
+      "eval_iou_safe": 0.4605461201989133,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9814470884642749,
+      "eval_loss": 0.0914885550737381,
+      "eval_mean_accuracy": 0.7617057731441406,
+      "eval_mean_iou": 0.7209966043315941,
+      "eval_overall_accuracy": 0.9817363682077892,
+      "eval_runtime": 11.2224,
+      "eval_samples_per_second": 5.97,
+      "eval_steps_per_second": 0.446,
+      "step": 850
+    },
+    {
+      "epoch": 77.36,
+      "learning_rate": 9.531100478468899e-07,
+      "loss": 0.0892,
+      "step": 851
+    },
+    {
+      "epoch": 77.45,
+      "learning_rate": 9.492822966507177e-07,
+      "loss": 0.0849,
+      "step": 852
+    },
+    {
+      "epoch": 77.55,
+      "learning_rate": 9.454545454545454e-07,
+      "loss": 0.0954,
+      "step": 853
+    },
+    {
+      "epoch": 77.64,
+      "learning_rate": 9.416267942583732e-07,
+      "loss": 0.0844,
+      "step": 854
+    },
+    {
+      "epoch": 77.73,
+      "learning_rate": 9.377990430622009e-07,
+      "loss": 0.1121,
+      "step": 855
+    },
+    {
+      "epoch": 77.82,
+      "learning_rate": 9.339712918660286e-07,
+      "loss": 0.0919,
+      "step": 856
+    },
+    {
+      "epoch": 77.91,
+      "learning_rate": 9.301435406698565e-07,
+      "loss": 0.0836,
+      "step": 857
+    },
+    {
+      "epoch": 78.0,
+      "learning_rate": 9.263157894736841e-07,
+      "loss": 0.0807,
+      "step": 858
+    },
+    {
+      "epoch": 78.09,
+      "learning_rate": 9.224880382775119e-07,
+      "loss": 0.086,
+      "step": 859
+    },
+    {
+      "epoch": 78.18,
+      "learning_rate": 9.186602870813397e-07,
+      "loss": 0.0888,
+      "step": 860
+    },
+    {
+      "epoch": 78.18,
+      "eval_accuracy_safe": 0.5614569720240091,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9942326260970619,
+      "eval_iou_safe": 0.4720177732836822,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9811362811214656,
+      "eval_loss": 0.09153461456298828,
+      "eval_mean_accuracy": 0.7778447990605355,
+      "eval_mean_iou": 0.7265770272025739,
+      "eval_overall_accuracy": 0.9814491271972656,
+      "eval_runtime": 11.0853,
+      "eval_samples_per_second": 6.044,
+      "eval_steps_per_second": 0.451,
+      "step": 860
+    },
+    {
+      "epoch": 78.27,
+      "learning_rate": 9.148325358851674e-07,
+      "loss": 0.0814,
+      "step": 861
+    },
+    {
+      "epoch": 78.36,
+      "learning_rate": 9.110047846889953e-07,
+      "loss": 0.0927,
+      "step": 862
+    },
+    {
+      "epoch": 78.45,
+      "learning_rate": 9.071770334928229e-07,
+      "loss": 0.0965,
+      "step": 863
+    },
+    {
+      "epoch": 78.55,
+      "learning_rate": 9.033492822966507e-07,
+      "loss": 0.0922,
+      "step": 864
+    },
+    {
+      "epoch": 78.64,
+      "learning_rate": 8.995215311004784e-07,
+      "loss": 0.0814,
+      "step": 865
+    },
+    {
+      "epoch": 78.73,
+      "learning_rate": 8.956937799043062e-07,
+      "loss": 0.0859,
+      "step": 866
+    },
+    {
+      "epoch": 78.82,
+      "learning_rate": 8.918660287081338e-07,
+      "loss": 0.0967,
+      "step": 867
+    },
+    {
+      "epoch": 78.91,
+      "learning_rate": 8.880382775119617e-07,
+      "loss": 0.1038,
+      "step": 868
+    },
+    {
+      "epoch": 79.0,
+      "learning_rate": 8.842105263157895e-07,
+      "loss": 0.1504,
+      "step": 869
+    },
+    {
+      "epoch": 79.09,
+      "learning_rate": 8.803827751196172e-07,
+      "loss": 0.09,
+      "step": 870
+    },
+    {
+      "epoch": 79.09,
+      "eval_accuracy_safe": 0.5414493390542057,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.994825591266709,
+      "eval_iou_safe": 0.46277670968156454,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9811318145505028,
+      "eval_loss": 0.0920419991016388,
+      "eval_mean_accuracy": 0.7681374651604573,
+      "eval_mean_iou": 0.7219542621160336,
+      "eval_overall_accuracy": 0.9814335837293027,
+      "eval_runtime": 10.1364,
+      "eval_samples_per_second": 6.61,
+      "eval_steps_per_second": 0.493,
+      "step": 870
+    },
+    {
+      "epoch": 79.18,
+      "learning_rate": 8.76555023923445e-07,
+      "loss": 0.0865,
+      "step": 871
+    },
+    {
+      "epoch": 79.27,
+      "learning_rate": 8.727272727272726e-07,
+      "loss": 0.0851,
+      "step": 872
+    },
+    {
+      "epoch": 79.36,
+      "learning_rate": 8.688995215311005e-07,
+      "loss": 0.1074,
+      "step": 873
+    },
+    {
+      "epoch": 79.45,
+      "learning_rate": 8.650717703349282e-07,
+      "loss": 0.0888,
+      "step": 874
+    },
+    {
+      "epoch": 79.55,
+      "learning_rate": 8.61244019138756e-07,
+      "loss": 0.0903,
+      "step": 875
+    },
+    {
+      "epoch": 79.64,
+      "learning_rate": 8.574162679425837e-07,
+      "loss": 0.1036,
+      "step": 876
+    },
+    {
+      "epoch": 79.73,
+      "learning_rate": 8.535885167464114e-07,
+      "loss": 0.0874,
+      "step": 877
+    },
+    {
+      "epoch": 79.82,
+      "learning_rate": 8.497607655502392e-07,
+      "loss": 0.0953,
+      "step": 878
+    },
+    {
+      "epoch": 79.91,
+      "learning_rate": 8.459330143540669e-07,
+      "loss": 0.0881,
+      "step": 879
+    },
+    {
+      "epoch": 80.0,
+      "learning_rate": 8.421052631578947e-07,
+      "loss": 0.1052,
+      "step": 880
+    },
+    {
+      "epoch": 80.0,
+      "eval_accuracy_safe": 0.5866110770582997,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9931634465925946,
+      "eval_iou_safe": 0.4790187776431146,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9808222378260363,
+      "eval_loss": 0.09174863249063492,
+      "eval_mean_accuracy": 0.7898872618254471,
+      "eval_mean_iou": 0.7299205077345754,
+      "eval_overall_accuracy": 0.9811545414711113,
+      "eval_runtime": 11.2146,
+      "eval_samples_per_second": 5.974,
+      "eval_steps_per_second": 0.446,
+      "step": 880
+    },
+    {
+      "epoch": 80.09,
+      "learning_rate": 8.382775119617225e-07,
+      "loss": 0.0942,
+      "step": 881
+    },
+    {
+      "epoch": 80.18,
+      "learning_rate": 8.344497607655502e-07,
+      "loss": 0.0891,
+      "step": 882
+    },
+    {
+      "epoch": 80.27,
+      "learning_rate": 8.30622009569378e-07,
+      "loss": 0.0959,
+      "step": 883
+    },
+    {
+      "epoch": 80.36,
+      "learning_rate": 8.267942583732057e-07,
+      "loss": 0.0924,
+      "step": 884
+    },
+    {
+      "epoch": 80.45,
+      "learning_rate": 8.229665071770334e-07,
+      "loss": 0.1006,
+      "step": 885
+    },
+    {
+      "epoch": 80.55,
+      "learning_rate": 8.191387559808613e-07,
+      "loss": 0.0976,
+      "step": 886
+    },
+    {
+      "epoch": 80.64,
+      "learning_rate": 8.153110047846889e-07,
+      "loss": 0.0775,
+      "step": 887
+    },
+    {
+      "epoch": 80.73,
+      "learning_rate": 8.114832535885167e-07,
+      "loss": 0.09,
+      "step": 888
+    },
+    {
+      "epoch": 80.82,
+      "learning_rate": 8.076555023923445e-07,
+      "loss": 0.0844,
+      "step": 889
+    },
+    {
+      "epoch": 80.91,
+      "learning_rate": 8.038277511961722e-07,
+      "loss": 0.0867,
+      "step": 890
+    },
+    {
+      "epoch": 80.91,
+      "eval_accuracy_safe": 0.5251618151047991,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9954841481113997,
+      "eval_iou_safe": 0.45731267225491545,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813015215317995,
+      "eval_loss": 0.09118454903364182,
+      "eval_mean_accuracy": 0.7603229816080994,
+      "eval_mean_iou": 0.7193070968933575,
+      "eval_overall_accuracy": 0.9815915805190357,
+      "eval_runtime": 10.9147,
+      "eval_samples_per_second": 6.139,
+      "eval_steps_per_second": 0.458,
+      "step": 890
+    },
+    {
+      "epoch": 81.0,
+      "learning_rate": 8e-07,
+      "loss": 0.0841,
+      "step": 891
+    },
+    {
+      "epoch": 81.09,
+      "learning_rate": 7.961722488038277e-07,
+      "loss": 0.087,
+      "step": 892
+    },
+    {
+      "epoch": 81.18,
+      "learning_rate": 7.923444976076555e-07,
+      "loss": 0.088,
+      "step": 893
+    },
+    {
+      "epoch": 81.27,
+      "learning_rate": 7.885167464114832e-07,
+      "loss": 0.0903,
+      "step": 894
+    },
+    {
+      "epoch": 81.36,
+      "learning_rate": 7.84688995215311e-07,
+      "loss": 0.0988,
+      "step": 895
+    },
+    {
+      "epoch": 81.45,
+      "learning_rate": 7.808612440191386e-07,
+      "loss": 0.0994,
+      "step": 896
+    },
+    {
+      "epoch": 81.55,
+      "learning_rate": 7.770334928229665e-07,
+      "loss": 0.0927,
+      "step": 897
+    },
+    {
+      "epoch": 81.64,
+      "learning_rate": 7.732057416267941e-07,
+      "loss": 0.0986,
+      "step": 898
+    },
+    {
+      "epoch": 81.73,
+      "learning_rate": 7.69377990430622e-07,
+      "loss": 0.0919,
+      "step": 899
+    },
+    {
+      "epoch": 81.82,
+      "learning_rate": 7.655502392344498e-07,
+      "loss": 0.0942,
+      "step": 900
+    },
+    {
+      "epoch": 81.82,
+      "eval_accuracy_safe": 0.509124868446922,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9959278599525041,
+      "eval_iou_safe": 0.44904804568428675,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9812667558907583,
+      "eval_loss": 0.09251971542835236,
+      "eval_mean_accuracy": 0.7525263641997131,
+      "eval_mean_iou": 0.7151574007875225,
+      "eval_overall_accuracy": 0.9815484801335121,
+      "eval_runtime": 10.8337,
+      "eval_samples_per_second": 6.184,
+      "eval_steps_per_second": 0.462,
+      "step": 900
+    },
+    {
+      "epoch": 81.91,
+      "learning_rate": 7.617224880382774e-07,
+      "loss": 0.0811,
+      "step": 901
+    },
+    {
+      "epoch": 82.0,
+      "learning_rate": 7.578947368421053e-07,
+      "loss": 0.0841,
+      "step": 902
+    },
+    {
+      "epoch": 82.09,
+      "learning_rate": 7.540669856459329e-07,
+      "loss": 0.0979,
+      "step": 903
+    },
+    {
+      "epoch": 82.18,
+      "learning_rate": 7.502392344497608e-07,
+      "loss": 0.1012,
+      "step": 904
+    },
+    {
+      "epoch": 82.27,
+      "learning_rate": 7.464114832535885e-07,
+      "loss": 0.1066,
+      "step": 905
+    },
+    {
+      "epoch": 82.36,
+      "learning_rate": 7.425837320574162e-07,
+      "loss": 0.0867,
+      "step": 906
+    },
+    {
+      "epoch": 82.45,
+      "learning_rate": 7.38755980861244e-07,
+      "loss": 0.0837,
+      "step": 907
+    },
+    {
+      "epoch": 82.55,
+      "learning_rate": 7.349282296650717e-07,
+      "loss": 0.0795,
+      "step": 908
+    },
+    {
+      "epoch": 82.64,
+      "learning_rate": 7.311004784688995e-07,
+      "loss": 0.0891,
+      "step": 909
+    },
+    {
+      "epoch": 82.73,
+      "learning_rate": 7.272727272727272e-07,
+      "loss": 0.0917,
+      "step": 910
+    },
+    {
+      "epoch": 82.73,
+      "eval_accuracy_safe": 0.5453968951546062,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9949795967649107,
+      "eval_iou_safe": 0.4681753351401616,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813999957756329,
+      "eval_loss": 0.09079530835151672,
+      "eval_mean_accuracy": 0.7701882459597584,
+      "eval_mean_iou": 0.7247876654578972,
+      "eval_overall_accuracy": 0.981699644629635,
+      "eval_runtime": 10.5487,
+      "eval_samples_per_second": 6.352,
+      "eval_steps_per_second": 0.474,
+      "step": 910
+    },
+    {
+      "epoch": 82.82,
+      "learning_rate": 7.23444976076555e-07,
+      "loss": 0.0874,
+      "step": 911
+    },
+    {
+      "epoch": 82.91,
+      "learning_rate": 7.196172248803828e-07,
+      "loss": 0.0861,
+      "step": 912
+    },
+    {
+      "epoch": 83.0,
+      "learning_rate": 7.157894736842105e-07,
+      "loss": 0.089,
+      "step": 913
+    },
+    {
+      "epoch": 83.09,
+      "learning_rate": 7.119617224880382e-07,
+      "loss": 0.0823,
+      "step": 914
+    },
+    {
+      "epoch": 83.18,
+      "learning_rate": 7.08133971291866e-07,
+      "loss": 0.0857,
+      "step": 915
+    },
+    {
+      "epoch": 83.27,
+      "learning_rate": 7.043062200956937e-07,
+      "loss": 0.0969,
+      "step": 916
+    },
+    {
+      "epoch": 83.36,
+      "learning_rate": 7.004784688995215e-07,
+      "loss": 0.0893,
+      "step": 917
+    },
+    {
+      "epoch": 83.45,
+      "learning_rate": 6.966507177033493e-07,
+      "loss": 0.0935,
+      "step": 918
+    },
+    {
+      "epoch": 83.55,
+      "learning_rate": 6.92822966507177e-07,
+      "loss": 0.0798,
+      "step": 919
+    },
+    {
+      "epoch": 83.64,
+      "learning_rate": 6.889952153110048e-07,
+      "loss": 0.103,
+      "step": 920
+    },
+    {
+      "epoch": 83.64,
+      "eval_accuracy_safe": 0.5452176360152814,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9949149437900465,
+      "eval_iou_safe": 0.46716963182686017,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813309439361214,
+      "eval_loss": 0.09115264564752579,
+      "eval_mean_accuracy": 0.770066289902664,
+      "eval_mean_iou": 0.7242502878814908,
+      "eval_overall_accuracy": 0.9816316063724347,
+      "eval_runtime": 10.225,
+      "eval_samples_per_second": 6.553,
+      "eval_steps_per_second": 0.489,
+      "step": 920
+    },
+    {
+      "epoch": 83.73,
+      "learning_rate": 6.851674641148325e-07,
+      "loss": 0.1,
+      "step": 921
+    },
+    {
+      "epoch": 83.82,
+      "learning_rate": 6.813397129186602e-07,
+      "loss": 0.0907,
+      "step": 922
+    },
+    {
+      "epoch": 83.91,
+      "learning_rate": 6.77511961722488e-07,
+      "loss": 0.0999,
+      "step": 923
+    },
+    {
+      "epoch": 84.0,
+      "learning_rate": 6.736842105263158e-07,
+      "loss": 0.0872,
+      "step": 924
+    },
+    {
+      "epoch": 84.09,
+      "learning_rate": 6.698564593301435e-07,
+      "loss": 0.0826,
+      "step": 925
+    },
+    {
+      "epoch": 84.18,
+      "learning_rate": 6.660287081339713e-07,
+      "loss": 0.0842,
+      "step": 926
+    },
+    {
+      "epoch": 84.27,
+      "learning_rate": 6.62200956937799e-07,
+      "loss": 0.0913,
+      "step": 927
+    },
+    {
+      "epoch": 84.36,
+      "learning_rate": 6.583732057416268e-07,
+      "loss": 0.0941,
+      "step": 928
+    },
+    {
+      "epoch": 84.45,
+      "learning_rate": 6.545454545454546e-07,
+      "loss": 0.086,
+      "step": 929
+    },
+    {
+      "epoch": 84.55,
+      "learning_rate": 6.507177033492822e-07,
+      "loss": 0.0939,
+      "step": 930
+    },
+    {
+      "epoch": 84.55,
+      "eval_accuracy_safe": 0.5539396532781292,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9946924131787404,
+      "eval_iou_safe": 0.4716882034044076,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813684244098391,
+      "eval_loss": 0.09003055095672607,
+      "eval_mean_accuracy": 0.7743160332284348,
+      "eval_mean_iou": 0.7265283139071234,
+      "eval_overall_accuracy": 0.9816732833634562,
+      "eval_runtime": 10.5346,
+      "eval_samples_per_second": 6.36,
+      "eval_steps_per_second": 0.475,
+      "step": 930
+    },
+    {
+      "epoch": 84.64,
+      "learning_rate": 6.468899521531101e-07,
+      "loss": 0.0849,
+      "step": 931
+    },
+    {
+      "epoch": 84.73,
+      "learning_rate": 6.430622009569377e-07,
+      "loss": 0.081,
+      "step": 932
+    },
+    {
+      "epoch": 84.82,
+      "learning_rate": 6.392344497607656e-07,
+      "loss": 0.0905,
+      "step": 933
+    },
+    {
+      "epoch": 84.91,
+      "learning_rate": 6.354066985645932e-07,
+      "loss": 0.0945,
+      "step": 934
+    },
+    {
+      "epoch": 85.0,
+      "learning_rate": 6.31578947368421e-07,
+      "loss": 0.1062,
+      "step": 935
+    },
+    {
+      "epoch": 85.09,
+      "learning_rate": 6.277511961722489e-07,
+      "loss": 0.0886,
+      "step": 936
+    },
+    {
+      "epoch": 85.18,
+      "learning_rate": 6.239234449760765e-07,
+      "loss": 0.0845,
+      "step": 937
+    },
+    {
+      "epoch": 85.27,
+      "learning_rate": 6.200956937799043e-07,
+      "loss": 0.0821,
+      "step": 938
+    },
+    {
+      "epoch": 85.36,
+      "learning_rate": 6.16267942583732e-07,
+      "loss": 0.0914,
+      "step": 939
+    },
+    {
+      "epoch": 85.45,
+      "learning_rate": 6.124401913875598e-07,
+      "loss": 0.0892,
+      "step": 940
+    },
+    {
+      "epoch": 85.45,
+      "eval_accuracy_safe": 0.5329624789418699,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9954403812155299,
+      "eval_iou_safe": 0.4635251123181117,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9814880952587547,
+      "eval_loss": 0.08998803794384003,
+      "eval_mean_accuracy": 0.7642014300786999,
+      "eval_mean_iou": 0.7225066037884332,
+      "eval_overall_accuracy": 0.9817795255290929,
+      "eval_runtime": 10.498,
+      "eval_samples_per_second": 6.382,
+      "eval_steps_per_second": 0.476,
+      "step": 940
+    },
+    {
+      "epoch": 85.55,
+      "learning_rate": 6.086124401913876e-07,
+      "loss": 0.0924,
+      "step": 941
+    },
+    {
+      "epoch": 85.64,
+      "learning_rate": 6.047846889952153e-07,
+      "loss": 0.1009,
+      "step": 942
+    },
+    {
+      "epoch": 85.73,
+      "learning_rate": 6.00956937799043e-07,
+      "loss": 0.0842,
+      "step": 943
+    },
+    {
+      "epoch": 85.82,
+      "learning_rate": 5.971291866028708e-07,
+      "loss": 0.0874,
+      "step": 944
+    },
+    {
+      "epoch": 85.91,
+      "learning_rate": 5.933014354066985e-07,
+      "loss": 0.0853,
+      "step": 945
+    },
+    {
+      "epoch": 86.0,
+      "learning_rate": 5.894736842105262e-07,
+      "loss": 0.0871,
+      "step": 946
+    },
+    {
+      "epoch": 86.09,
+      "learning_rate": 5.856459330143541e-07,
+      "loss": 0.0878,
+      "step": 947
+    },
+    {
+      "epoch": 86.18,
+      "learning_rate": 5.818181818181818e-07,
+      "loss": 0.0898,
+      "step": 948
+    },
+    {
+      "epoch": 86.27,
+      "learning_rate": 5.779904306220096e-07,
+      "loss": 0.083,
+      "step": 949
+    },
+    {
+      "epoch": 86.36,
+      "learning_rate": 5.741626794258373e-07,
+      "loss": 0.0899,
+      "step": 950
+    },
+    {
+      "epoch": 86.36,
+      "eval_accuracy_safe": 0.5755625460194833,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9937747750845035,
+      "eval_iou_safe": 0.4778336080395577,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.98110013614213,
+      "eval_loss": 0.09046214818954468,
+      "eval_mean_accuracy": 0.7846686605519935,
+      "eval_mean_iou": 0.7294668720908438,
+      "eval_overall_accuracy": 0.9814214564081448,
+      "eval_runtime": 9.9923,
+      "eval_samples_per_second": 6.705,
+      "eval_steps_per_second": 0.5,
+      "step": 950
+    },
+    {
+      "epoch": 86.45,
+      "learning_rate": 5.70334928229665e-07,
+      "loss": 0.0982,
+      "step": 951
+    },
+    {
+      "epoch": 86.55,
+      "learning_rate": 5.665071770334928e-07,
+      "loss": 0.0898,
+      "step": 952
+    },
+    {
+      "epoch": 86.64,
+      "learning_rate": 5.626794258373205e-07,
+      "loss": 0.0858,
+      "step": 953
+    },
+    {
+      "epoch": 86.73,
+      "learning_rate": 5.588516746411483e-07,
+      "loss": 0.0815,
+      "step": 954
+    },
+    {
+      "epoch": 86.82,
+      "learning_rate": 5.550239234449761e-07,
+      "loss": 0.0875,
+      "step": 955
+    },
+    {
+      "epoch": 86.91,
+      "learning_rate": 5.511961722488037e-07,
+      "loss": 0.0944,
+      "step": 956
+    },
+    {
+      "epoch": 87.0,
+      "learning_rate": 5.473684210526316e-07,
+      "loss": 0.0853,
+      "step": 957
+    },
+    {
+      "epoch": 87.09,
+      "learning_rate": 5.435406698564593e-07,
+      "loss": 0.0874,
+      "step": 958
+    },
+    {
+      "epoch": 87.18,
+      "learning_rate": 5.39712918660287e-07,
+      "loss": 0.0943,
+      "step": 959
+    },
+    {
+      "epoch": 87.27,
+      "learning_rate": 5.358851674641149e-07,
+      "loss": 0.0877,
+      "step": 960
+    },
+    {
+      "epoch": 87.27,
+      "eval_accuracy_safe": 0.5771238352974738,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.993741803240698,
+      "eval_iou_safe": 0.478699286298987,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9811136143593967,
+      "eval_loss": 0.0893407091498375,
+      "eval_mean_accuracy": 0.7854328192690859,
+      "eval_mean_iou": 0.7299064503291919,
+      "eval_overall_accuracy": 0.9814355764816056,
+      "eval_runtime": 10.878,
+      "eval_samples_per_second": 6.159,
+      "eval_steps_per_second": 0.46,
+      "step": 960
+    },
+    {
+      "epoch": 87.36,
+      "learning_rate": 5.320574162679425e-07,
+      "loss": 0.0971,
+      "step": 961
+    },
+    {
+      "epoch": 87.45,
+      "learning_rate": 5.282296650717704e-07,
+      "loss": 0.0785,
+      "step": 962
+    },
+    {
+      "epoch": 87.55,
+      "learning_rate": 5.24401913875598e-07,
+      "loss": 0.0918,
+      "step": 963
+    },
+    {
+      "epoch": 87.64,
+      "learning_rate": 5.205741626794258e-07,
+      "loss": 0.0877,
+      "step": 964
+    },
+    {
+      "epoch": 87.73,
+      "learning_rate": 5.167464114832535e-07,
+      "loss": 0.0858,
+      "step": 965
+    },
+    {
+      "epoch": 87.82,
+      "learning_rate": 5.129186602870813e-07,
+      "loss": 0.0879,
+      "step": 966
+    },
+    {
+      "epoch": 87.91,
+      "learning_rate": 5.090909090909091e-07,
+      "loss": 0.0845,
+      "step": 967
+    },
+    {
+      "epoch": 88.0,
+      "learning_rate": 5.052631578947368e-07,
+      "loss": 0.0838,
+      "step": 968
+    },
+    {
+      "epoch": 88.09,
+      "learning_rate": 5.014354066985646e-07,
+      "loss": 0.087,
+      "step": 969
+    },
+    {
+      "epoch": 88.18,
+      "learning_rate": 4.976076555023923e-07,
+      "loss": 0.0851,
+      "step": 970
+    },
+    {
+      "epoch": 88.18,
+      "eval_accuracy_safe": 0.5086834669103049,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9961247523151573,
+      "eval_iou_safe": 0.45123321165075103,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9814477580070269,
+      "eval_loss": 0.08967232704162598,
+      "eval_mean_accuracy": 0.7524041096127311,
+      "eval_mean_iou": 0.716340484828889,
+      "eval_overall_accuracy": 0.9817265183178346,
+      "eval_runtime": 10.7386,
+      "eval_samples_per_second": 6.239,
+      "eval_steps_per_second": 0.466,
+      "step": 970
+    },
+    {
+      "epoch": 88.27,
+      "learning_rate": 4.937799043062201e-07,
+      "loss": 0.1024,
+      "step": 971
+    },
+    {
+      "epoch": 88.36,
+      "learning_rate": 4.899521531100478e-07,
+      "loss": 0.0857,
+      "step": 972
+    },
+    {
+      "epoch": 88.45,
+      "learning_rate": 4.861244019138756e-07,
+      "loss": 0.0818,
+      "step": 973
+    },
+    {
+      "epoch": 88.55,
+      "learning_rate": 4.822966507177033e-07,
+      "loss": 0.079,
+      "step": 974
+    },
+    {
+      "epoch": 88.64,
+      "learning_rate": 4.784688995215311e-07,
+      "loss": 0.1013,
+      "step": 975
+    },
+    {
+      "epoch": 88.73,
+      "learning_rate": 4.7464114832535886e-07,
+      "loss": 0.0876,
+      "step": 976
+    },
+    {
+      "epoch": 88.82,
+      "learning_rate": 4.708133971291866e-07,
+      "loss": 0.0865,
+      "step": 977
+    },
+    {
+      "epoch": 88.91,
+      "learning_rate": 4.669856459330143e-07,
+      "loss": 0.0826,
+      "step": 978
+    },
+    {
+      "epoch": 89.0,
+      "learning_rate": 4.6315789473684207e-07,
+      "loss": 0.0973,
+      "step": 979
+    },
+    {
+      "epoch": 89.09,
+      "learning_rate": 4.593301435406698e-07,
+      "loss": 0.0857,
+      "step": 980
+    },
+    {
+      "epoch": 89.09,
+      "eval_accuracy_safe": 0.5363144320954815,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9952871384112242,
+      "eval_iou_safe": 0.4644068418979457,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9814357278367817,
+      "eval_loss": 0.08939877152442932,
+      "eval_mean_accuracy": 0.7658007852533528,
+      "eval_mean_iou": 0.7229212848673636,
+      "eval_overall_accuracy": 0.9817298205930796,
+      "eval_runtime": 11.0847,
+      "eval_samples_per_second": 6.044,
+      "eval_steps_per_second": 0.451,
+      "step": 980
+    },
+    {
+      "epoch": 89.18,
+      "learning_rate": 4.5550239234449763e-07,
+      "loss": 0.0901,
+      "step": 981
+    },
+    {
+      "epoch": 89.27,
+      "learning_rate": 4.5167464114832534e-07,
+      "loss": 0.0903,
+      "step": 982
+    },
+    {
+      "epoch": 89.36,
+      "learning_rate": 4.478468899521531e-07,
+      "loss": 0.0868,
+      "step": 983
+    },
+    {
+      "epoch": 89.45,
+      "learning_rate": 4.4401913875598085e-07,
+      "loss": 0.0884,
+      "step": 984
+    },
+    {
+      "epoch": 89.55,
+      "learning_rate": 4.401913875598086e-07,
+      "loss": 0.0787,
+      "step": 985
+    },
+    {
+      "epoch": 89.64,
+      "learning_rate": 4.363636363636363e-07,
+      "loss": 0.0792,
+      "step": 986
+    },
+    {
+      "epoch": 89.73,
+      "learning_rate": 4.325358851674641e-07,
+      "loss": 0.0916,
+      "step": 987
+    },
+    {
+      "epoch": 89.82,
+      "learning_rate": 4.2870813397129187e-07,
+      "loss": 0.0968,
+      "step": 988
+    },
+    {
+      "epoch": 89.91,
+      "learning_rate": 4.248803827751196e-07,
+      "loss": 0.0855,
+      "step": 989
+    },
+    {
+      "epoch": 90.0,
+      "learning_rate": 4.2105263157894733e-07,
+      "loss": 0.0821,
+      "step": 990
+    },
+    {
+      "epoch": 90.0,
+      "eval_accuracy_safe": 0.5332997945266209,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9953243930745986,
+      "eval_iou_safe": 0.4622863606668037,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813836675031155,
+      "eval_loss": 0.08945173770189285,
+      "eval_mean_accuracy": 0.7643120938006098,
+      "eval_mean_iou": 0.7218350140849595,
+      "eval_overall_accuracy": 0.9816769272533815,
+      "eval_runtime": 10.507,
+      "eval_samples_per_second": 6.377,
+      "eval_steps_per_second": 0.476,
+      "step": 990
+    },
+    {
+      "epoch": 90.09,
+      "learning_rate": 4.172248803827751e-07,
+      "loss": 0.0901,
+      "step": 991
+    },
+    {
+      "epoch": 90.18,
+      "learning_rate": 4.1339712918660284e-07,
+      "loss": 0.0814,
+      "step": 992
+    },
+    {
+      "epoch": 90.27,
+      "learning_rate": 4.0956937799043065e-07,
+      "loss": 0.0967,
+      "step": 993
+    },
+    {
+      "epoch": 90.36,
+      "learning_rate": 4.0574162679425835e-07,
+      "loss": 0.084,
+      "step": 994
+    },
+    {
+      "epoch": 90.45,
+      "learning_rate": 4.019138755980861e-07,
+      "loss": 0.088,
+      "step": 995
+    },
+    {
+      "epoch": 90.55,
+      "learning_rate": 3.9808612440191386e-07,
+      "loss": 0.0876,
+      "step": 996
+    },
+    {
+      "epoch": 90.64,
+      "learning_rate": 3.942583732057416e-07,
+      "loss": 0.0835,
+      "step": 997
+    },
+    {
+      "epoch": 90.73,
+      "learning_rate": 3.904306220095693e-07,
+      "loss": 0.0895,
+      "step": 998
+    },
+    {
+      "epoch": 90.82,
+      "learning_rate": 3.8660287081339707e-07,
+      "loss": 0.0885,
+      "step": 999
+    },
+    {
+      "epoch": 90.91,
+      "learning_rate": 3.827751196172249e-07,
+      "loss": 0.0931,
+      "step": 1000
+    },
+    {
+      "epoch": 90.91,
+      "eval_accuracy_safe": 0.5581262215642963,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9944299878098047,
+      "eval_iou_safe": 0.47178944898670777,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9812328764506335,
+      "eval_loss": 0.08953512459993362,
+      "eval_mean_accuracy": 0.7762781046870505,
+      "eval_mean_iou": 0.7265111627186707,
+      "eval_overall_accuracy": 0.9815422741334829,
+      "eval_runtime": 12.228,
+      "eval_samples_per_second": 5.479,
+      "eval_steps_per_second": 0.409,
+      "step": 1000
+    },
+    {
+      "epoch": 91.0,
+      "learning_rate": 3.7894736842105264e-07,
+      "loss": 0.0787,
+      "step": 1001
+    },
+    {
+      "epoch": 91.09,
+      "learning_rate": 3.751196172248804e-07,
+      "loss": 0.0869,
+      "step": 1002
+    },
+    {
+      "epoch": 91.18,
+      "learning_rate": 3.712918660287081e-07,
+      "loss": 0.1074,
+      "step": 1003
+    },
+    {
+      "epoch": 91.27,
+      "learning_rate": 3.6746411483253585e-07,
+      "loss": 0.0836,
+      "step": 1004
+    },
+    {
+      "epoch": 91.36,
+      "learning_rate": 3.636363636363636e-07,
+      "loss": 0.0908,
+      "step": 1005
+    },
+    {
+      "epoch": 91.45,
+      "learning_rate": 3.598086124401914e-07,
+      "loss": 0.0813,
+      "step": 1006
+    },
+    {
+      "epoch": 91.55,
+      "learning_rate": 3.559808612440191e-07,
+      "loss": 0.0898,
+      "step": 1007
+    },
+    {
+      "epoch": 91.64,
+      "learning_rate": 3.5215311004784687e-07,
+      "loss": 0.1058,
+      "step": 1008
+    },
+    {
+      "epoch": 91.73,
+      "learning_rate": 3.4832535885167463e-07,
+      "loss": 0.0865,
+      "step": 1009
+    },
+    {
+      "epoch": 91.82,
+      "learning_rate": 3.444976076555024e-07,
+      "loss": 0.0787,
+      "step": 1010
+    },
+    {
+      "epoch": 91.82,
+      "eval_accuracy_safe": 0.552497870093022,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9945764250378091,
+      "eval_iou_safe": 0.46893885604159374,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9812115070135685,
+      "eval_loss": 0.08887456357479095,
+      "eval_mean_accuracy": 0.7735371475654156,
+      "eval_mean_iou": 0.7250751815275811,
+      "eval_overall_accuracy": 0.9815181333627274,
+      "eval_runtime": 10.9308,
+      "eval_samples_per_second": 6.129,
+      "eval_steps_per_second": 0.457,
+      "step": 1010
+    },
+    {
+      "epoch": 91.91,
+      "learning_rate": 3.406698564593301e-07,
+      "loss": 0.0804,
+      "step": 1011
+    },
+    {
+      "epoch": 92.0,
+      "learning_rate": 3.368421052631579e-07,
+      "loss": 0.0913,
+      "step": 1012
+    },
+    {
+      "epoch": 92.09,
+      "learning_rate": 3.3301435406698565e-07,
+      "loss": 0.0774,
+      "step": 1013
+    },
+    {
+      "epoch": 92.18,
+      "learning_rate": 3.291866028708134e-07,
+      "loss": 0.0797,
+      "step": 1014
+    },
+    {
+      "epoch": 92.27,
+      "learning_rate": 3.253588516746411e-07,
+      "loss": 0.0917,
+      "step": 1015
+    },
+    {
+      "epoch": 92.36,
+      "learning_rate": 3.2153110047846886e-07,
+      "loss": 0.0924,
+      "step": 1016
+    },
+    {
+      "epoch": 92.45,
+      "learning_rate": 3.177033492822966e-07,
+      "loss": 0.0841,
+      "step": 1017
+    },
+    {
+      "epoch": 92.55,
+      "learning_rate": 3.138755980861244e-07,
+      "loss": 0.0962,
+      "step": 1018
+    },
+    {
+      "epoch": 92.64,
+      "learning_rate": 3.1004784688995213e-07,
+      "loss": 0.0809,
+      "step": 1019
+    },
+    {
+      "epoch": 92.73,
+      "learning_rate": 3.062200956937799e-07,
+      "loss": 0.0865,
+      "step": 1020
+    },
+    {
+      "epoch": 92.73,
+      "eval_accuracy_safe": 0.5659114652603499,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9941407508169918,
+      "eval_iou_safe": 0.47455840897923296,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9811768951245775,
+      "eval_loss": 0.08833795040845871,
+      "eval_mean_accuracy": 0.7800261080386708,
+      "eval_mean_iou": 0.7278676520519052,
+      "eval_overall_accuracy": 0.9814915443534282,
+      "eval_runtime": 10.5901,
+      "eval_samples_per_second": 6.327,
+      "eval_steps_per_second": 0.472,
+      "step": 1020
+    },
+    {
+      "epoch": 92.82,
+      "learning_rate": 3.0239234449760764e-07,
+      "loss": 0.091,
+      "step": 1021
+    },
+    {
+      "epoch": 92.91,
+      "learning_rate": 2.985645933014354e-07,
+      "loss": 0.0795,
+      "step": 1022
+    },
+    {
+      "epoch": 93.0,
+      "learning_rate": 2.947368421052631e-07,
+      "loss": 0.0855,
+      "step": 1023
+    },
+    {
+      "epoch": 93.09,
+      "learning_rate": 2.909090909090909e-07,
+      "loss": 0.0879,
+      "step": 1024
+    },
+    {
+      "epoch": 93.18,
+      "learning_rate": 2.8708133971291866e-07,
+      "loss": 0.0927,
+      "step": 1025
+    },
+    {
+      "epoch": 93.27,
+      "learning_rate": 2.832535885167464e-07,
+      "loss": 0.0876,
+      "step": 1026
+    },
+    {
+      "epoch": 93.36,
+      "learning_rate": 2.7942583732057417e-07,
+      "loss": 0.0696,
+      "step": 1027
+    },
+    {
+      "epoch": 93.45,
+      "learning_rate": 2.755980861244019e-07,
+      "loss": 0.0943,
+      "step": 1028
+    },
+    {
+      "epoch": 93.55,
+      "learning_rate": 2.7177033492822963e-07,
+      "loss": 0.0897,
+      "step": 1029
+    },
+    {
+      "epoch": 93.64,
+      "learning_rate": 2.6794258373205744e-07,
+      "loss": 0.0939,
+      "step": 1030
+    },
+    {
+      "epoch": 93.64,
+      "eval_accuracy_safe": 0.5582669303510781,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9944564474211148,
+      "eval_iou_safe": 0.4722554219774952,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9812631317160837,
+      "eval_loss": 0.08905700594186783,
+      "eval_mean_accuracy": 0.7763616888860965,
+      "eval_mean_iou": 0.7267592768467894,
+      "eval_overall_accuracy": 0.981572108482247,
+      "eval_runtime": 10.3984,
+      "eval_samples_per_second": 6.443,
+      "eval_steps_per_second": 0.481,
+      "step": 1030
+    },
+    {
+      "epoch": 93.73,
+      "learning_rate": 2.641148325358852e-07,
+      "loss": 0.0816,
+      "step": 1031
+    },
+    {
+      "epoch": 93.82,
+      "learning_rate": 2.602870813397129e-07,
+      "loss": 0.0916,
+      "step": 1032
+    },
+    {
+      "epoch": 93.91,
+      "learning_rate": 2.5645933014354065e-07,
+      "loss": 0.082,
+      "step": 1033
+    },
+    {
+      "epoch": 94.0,
+      "learning_rate": 2.526315789473684e-07,
+      "loss": 0.0887,
+      "step": 1034
+    },
+    {
+      "epoch": 94.09,
+      "learning_rate": 2.4880382775119616e-07,
+      "loss": 0.0838,
+      "step": 1035
+    },
+    {
+      "epoch": 94.18,
+      "learning_rate": 2.449760765550239e-07,
+      "loss": 0.0787,
+      "step": 1036
+    },
+    {
+      "epoch": 94.27,
+      "learning_rate": 2.4114832535885167e-07,
+      "loss": 0.0885,
+      "step": 1037
+    },
+    {
+      "epoch": 94.36,
+      "learning_rate": 2.3732057416267943e-07,
+      "loss": 0.0833,
+      "step": 1038
+    },
+    {
+      "epoch": 94.45,
+      "learning_rate": 2.3349282296650716e-07,
+      "loss": 0.0853,
+      "step": 1039
+    },
+    {
+      "epoch": 94.55,
+      "learning_rate": 2.296650717703349e-07,
+      "loss": 0.0874,
+      "step": 1040
+    },
+    {
+      "epoch": 94.55,
+      "eval_accuracy_safe": 0.5258480113800641,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9954896043061932,
+      "eval_iou_safe": 0.4579817049729637,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813271040609544,
+      "eval_loss": 0.08934454619884491,
+      "eval_mean_accuracy": 0.7606688078431286,
+      "eval_mean_iou": 0.7196544045169591,
+      "eval_overall_accuracy": 0.9816171446842934,
+      "eval_runtime": 10.6394,
+      "eval_samples_per_second": 6.297,
+      "eval_steps_per_second": 0.47,
+      "step": 1040
+    },
+    {
+      "epoch": 94.64,
+      "learning_rate": 2.2583732057416267e-07,
+      "loss": 0.0841,
+      "step": 1041
+    },
+    {
+      "epoch": 94.73,
+      "learning_rate": 2.2200956937799042e-07,
+      "loss": 0.0897,
+      "step": 1042
+    },
+    {
+      "epoch": 94.82,
+      "learning_rate": 2.1818181818181815e-07,
+      "loss": 0.0924,
+      "step": 1043
+    },
+    {
+      "epoch": 94.91,
+      "learning_rate": 2.1435406698564593e-07,
+      "loss": 0.0838,
+      "step": 1044
+    },
+    {
+      "epoch": 95.0,
+      "learning_rate": 2.1052631578947366e-07,
+      "loss": 0.1452,
+      "step": 1045
+    },
+    {
+      "epoch": 95.09,
+      "learning_rate": 2.0669856459330142e-07,
+      "loss": 0.0802,
+      "step": 1046
+    },
+    {
+      "epoch": 95.18,
+      "learning_rate": 2.0287081339712917e-07,
+      "loss": 0.0833,
+      "step": 1047
+    },
+    {
+      "epoch": 95.27,
+      "learning_rate": 1.9904306220095693e-07,
+      "loss": 0.0935,
+      "step": 1048
+    },
+    {
+      "epoch": 95.36,
+      "learning_rate": 1.9521531100478466e-07,
+      "loss": 0.092,
+      "step": 1049
+    },
+    {
+      "epoch": 95.45,
+      "learning_rate": 1.9138755980861244e-07,
+      "loss": 0.0927,
+      "step": 1050
+    },
+    {
+      "epoch": 95.45,
+      "eval_accuracy_safe": 0.5318946341764296,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9953069684525164,
+      "eval_iou_safe": 0.460839620971486,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.981325103975659,
+      "eval_loss": 0.0893927589058876,
+      "eval_mean_accuracy": 0.763600801314473,
+      "eval_mean_iou": 0.7210823624735725,
+      "eval_overall_accuracy": 0.9816185111430153,
+      "eval_runtime": 10.7038,
+      "eval_samples_per_second": 6.259,
+      "eval_steps_per_second": 0.467,
+      "step": 1050
+    },
+    {
+      "epoch": 95.55,
+      "learning_rate": 1.875598086124402e-07,
+      "loss": 0.0898,
+      "step": 1051
+    },
+    {
+      "epoch": 95.64,
+      "learning_rate": 1.8373205741626793e-07,
+      "loss": 0.0822,
+      "step": 1052
+    },
+    {
+      "epoch": 95.73,
+      "learning_rate": 1.799043062200957e-07,
+      "loss": 0.0802,
+      "step": 1053
+    },
+    {
+      "epoch": 95.82,
+      "learning_rate": 1.7607655502392344e-07,
+      "loss": 0.0901,
+      "step": 1054
+    },
+    {
+      "epoch": 95.91,
+      "learning_rate": 1.722488038277512e-07,
+      "loss": 0.0844,
+      "step": 1055
+    },
+    {
+      "epoch": 96.0,
+      "learning_rate": 1.6842105263157895e-07,
+      "loss": 0.0983,
+      "step": 1056
+    },
+    {
+      "epoch": 96.09,
+      "learning_rate": 1.645933014354067e-07,
+      "loss": 0.0771,
+      "step": 1057
+    },
+    {
+      "epoch": 96.18,
+      "learning_rate": 1.6076555023923443e-07,
+      "loss": 0.0889,
+      "step": 1058
+    },
+    {
+      "epoch": 96.27,
+      "learning_rate": 1.569377990430622e-07,
+      "loss": 0.0886,
+      "step": 1059
+    },
+    {
+      "epoch": 96.36,
+      "learning_rate": 1.5311004784688994e-07,
+      "loss": 0.0808,
+      "step": 1060
+    },
+    {
+      "epoch": 96.36,
+      "eval_accuracy_safe": 0.5443714557769631,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9949228640728112,
+      "eval_iou_safe": 0.4665486064879134,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813138268480589,
+      "eval_loss": 0.08972165733575821,
+      "eval_mean_accuracy": 0.7696471599248871,
+      "eval_mean_iou": 0.7239312166679861,
+      "eval_overall_accuracy": 0.9816142978952892,
+      "eval_runtime": 10.3014,
+      "eval_samples_per_second": 6.504,
+      "eval_steps_per_second": 0.485,
+      "step": 1060
+    },
+    {
+      "epoch": 96.45,
+      "learning_rate": 1.492822966507177e-07,
+      "loss": 0.0941,
+      "step": 1061
+    },
+    {
+      "epoch": 96.55,
+      "learning_rate": 1.4545454545454545e-07,
+      "loss": 0.0778,
+      "step": 1062
+    },
+    {
+      "epoch": 96.64,
+      "learning_rate": 1.416267942583732e-07,
+      "loss": 0.0859,
+      "step": 1063
+    },
+    {
+      "epoch": 96.73,
+      "learning_rate": 1.3779904306220094e-07,
+      "loss": 0.088,
+      "step": 1064
+    },
+    {
+      "epoch": 96.82,
+      "learning_rate": 1.3397129186602872e-07,
+      "loss": 0.0965,
+      "step": 1065
+    },
+    {
+      "epoch": 96.91,
+      "learning_rate": 1.3014354066985645e-07,
+      "loss": 0.0851,
+      "step": 1066
+    },
+    {
+      "epoch": 97.0,
+      "learning_rate": 1.263157894736842e-07,
+      "loss": 0.0796,
+      "step": 1067
+    },
+    {
+      "epoch": 97.09,
+      "learning_rate": 1.2248803827751196e-07,
+      "loss": 0.0824,
+      "step": 1068
+    },
+    {
+      "epoch": 97.18,
+      "learning_rate": 1.1866028708133971e-07,
+      "loss": 0.0888,
+      "step": 1069
+    },
+    {
+      "epoch": 97.27,
+      "learning_rate": 1.1483253588516746e-07,
+      "loss": 0.0924,
+      "step": 1070
+    },
+    {
+      "epoch": 97.27,
+      "eval_accuracy_safe": 0.5445140920813721,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9949591800360061,
+      "eval_iou_safe": 0.467148540982468,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813538483225048,
+      "eval_loss": 0.08921775221824646,
+      "eval_mean_accuracy": 0.7697366360586891,
+      "eval_mean_iou": 0.7242511946524864,
+      "eval_overall_accuracy": 0.9816537543908873,
+      "eval_runtime": 10.3063,
+      "eval_samples_per_second": 6.501,
+      "eval_steps_per_second": 0.485,
+      "step": 1070
+    },
+    {
+      "epoch": 97.36,
+      "learning_rate": 1.1100478468899521e-07,
+      "loss": 0.0956,
+      "step": 1071
+    },
+    {
+      "epoch": 97.45,
+      "learning_rate": 1.0717703349282297e-07,
+      "loss": 0.0877,
+      "step": 1072
+    },
+    {
+      "epoch": 97.55,
+      "learning_rate": 1.0334928229665071e-07,
+      "loss": 0.0832,
+      "step": 1073
+    },
+    {
+      "epoch": 97.64,
+      "learning_rate": 9.952153110047846e-08,
+      "loss": 0.0835,
+      "step": 1074
+    },
+    {
+      "epoch": 97.73,
+      "learning_rate": 9.569377990430622e-08,
+      "loss": 0.0937,
+      "step": 1075
+    },
+    {
+      "epoch": 97.82,
+      "learning_rate": 9.186602870813396e-08,
+      "loss": 0.0927,
+      "step": 1076
+    },
+    {
+      "epoch": 97.91,
+      "learning_rate": 8.803827751196172e-08,
+      "loss": 0.0843,
+      "step": 1077
+    },
+    {
+      "epoch": 98.0,
+      "learning_rate": 8.421052631578947e-08,
+      "loss": 0.0938,
+      "step": 1078
+    },
+    {
+      "epoch": 98.09,
+      "learning_rate": 8.038277511961722e-08,
+      "loss": 0.076,
+      "step": 1079
+    },
+    {
+      "epoch": 98.18,
+      "learning_rate": 7.655502392344497e-08,
+      "loss": 0.08,
+      "step": 1080
+    },
+    {
+      "epoch": 98.18,
+      "eval_accuracy_safe": 0.5522068149313225,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9947004507990275,
+      "eval_iou_safe": 0.47031842030955523,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813252894196396,
+      "eval_loss": 0.08840107917785645,
+      "eval_mean_accuracy": 0.773453632865175,
+      "eval_mean_iou": 0.7258218548645974,
+      "eval_overall_accuracy": 0.9816298982990321,
+      "eval_runtime": 10.2975,
+      "eval_samples_per_second": 6.506,
+      "eval_steps_per_second": 0.486,
+      "step": 1080
+    },
+    {
+      "epoch": 98.27,
+      "learning_rate": 7.623604465709728e-07,
+      "loss": 0.0914,
+      "step": 1081
+    },
+    {
+      "epoch": 98.36,
+      "learning_rate": 7.59170653907496e-07,
+      "loss": 0.0874,
+      "step": 1082
+    },
+    {
+      "epoch": 98.45,
+      "learning_rate": 7.55980861244019e-07,
+      "loss": 0.0843,
+      "step": 1083
+    },
+    {
+      "epoch": 98.55,
+      "learning_rate": 7.527910685805422e-07,
+      "loss": 0.1002,
+      "step": 1084
+    },
+    {
+      "epoch": 98.64,
+      "learning_rate": 7.496012759170654e-07,
+      "loss": 0.0806,
+      "step": 1085
+    },
+    {
+      "epoch": 98.73,
+      "learning_rate": 7.464114832535885e-07,
+      "loss": 0.083,
+      "step": 1086
+    },
+    {
+      "epoch": 98.82,
+      "learning_rate": 7.432216905901116e-07,
+      "loss": 0.0939,
+      "step": 1087
+    },
+    {
+      "epoch": 98.91,
+      "learning_rate": 7.400318979266348e-07,
+      "loss": 0.0848,
+      "step": 1088
+    },
+    {
+      "epoch": 99.0,
+      "learning_rate": 7.368421052631578e-07,
+      "loss": 0.0988,
+      "step": 1089
+    },
+    {
+      "epoch": 99.09,
+      "learning_rate": 7.33652312599681e-07,
+      "loss": 0.0798,
+      "step": 1090
+    },
+    {
+      "epoch": 99.09,
+      "eval_accuracy_safe": 0.5744522958662457,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9939207429624181,
+      "eval_iou_safe": 0.47881822943276425,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.981211506959408,
+      "eval_loss": 0.08804704993963242,
+      "eval_mean_accuracy": 0.7841865194143318,
+      "eval_mean_iou": 0.7300148681960861,
+      "eval_overall_accuracy": 0.9815303176196654,
+      "eval_runtime": 11.4661,
+      "eval_samples_per_second": 5.843,
+      "eval_steps_per_second": 0.436,
+      "step": 1090
+    },
+    {
+      "epoch": 99.18,
+      "learning_rate": 7.304625199362042e-07,
+      "loss": 0.0884,
+      "step": 1091
+    },
+    {
+      "epoch": 99.27,
+      "learning_rate": 7.272727272727272e-07,
+      "loss": 0.0952,
+      "step": 1092
+    },
+    {
+      "epoch": 99.36,
+      "learning_rate": 7.240829346092504e-07,
+      "loss": 0.0898,
+      "step": 1093
+    },
+    {
+      "epoch": 99.45,
+      "learning_rate": 7.208931419457734e-07,
+      "loss": 0.0869,
+      "step": 1094
+    },
+    {
+      "epoch": 99.55,
+      "learning_rate": 7.177033492822966e-07,
+      "loss": 0.0888,
+      "step": 1095
+    },
+    {
+      "epoch": 99.64,
+      "learning_rate": 7.145135566188198e-07,
+      "loss": 0.0848,
+      "step": 1096
+    },
+    {
+      "epoch": 99.73,
+      "learning_rate": 7.113237639553429e-07,
+      "loss": 0.0948,
+      "step": 1097
+    },
+    {
+      "epoch": 99.82,
+      "learning_rate": 7.08133971291866e-07,
+      "loss": 0.0869,
+      "step": 1098
+    },
+    {
+      "epoch": 99.91,
+      "learning_rate": 7.049441786283892e-07,
+      "loss": 0.0823,
+      "step": 1099
+    },
+    {
+      "epoch": 100.0,
+      "learning_rate": 7.017543859649122e-07,
+      "loss": 0.0789,
+      "step": 1100
+    },
+    {
+      "epoch": 100.0,
+      "eval_accuracy_safe": 0.5383267604982248,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9951786598717289,
+      "eval_iou_safe": 0.46471519066326555,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813880326396756,
+      "eval_loss": 0.08767610043287277,
+      "eval_mean_accuracy": 0.7667527101849769,
+      "eval_mean_iou": 0.7230516116514706,
+      "eval_overall_accuracy": 0.981683987290112,
+      "eval_runtime": 11.409,
+      "eval_samples_per_second": 5.873,
+      "eval_steps_per_second": 0.438,
+      "step": 1100
+    },
+    {
+      "epoch": 100.09,
+      "learning_rate": 6.985645933014354e-07,
+      "loss": 0.0888,
+      "step": 1101
+    },
+    {
+      "epoch": 100.18,
+      "learning_rate": 6.953748006379585e-07,
+      "loss": 0.0914,
+      "step": 1102
+    },
+    {
+      "epoch": 100.27,
+      "learning_rate": 6.921850079744816e-07,
+      "loss": 0.09,
+      "step": 1103
+    },
+    {
+      "epoch": 100.36,
+      "learning_rate": 6.889952153110048e-07,
+      "loss": 0.0899,
+      "step": 1104
+    },
+    {
+      "epoch": 100.45,
+      "learning_rate": 6.858054226475278e-07,
+      "loss": 0.0832,
+      "step": 1105
+    },
+    {
+      "epoch": 100.55,
+      "learning_rate": 6.82615629984051e-07,
+      "loss": 0.0817,
+      "step": 1106
+    },
+    {
+      "epoch": 100.64,
+      "learning_rate": 6.794258373205742e-07,
+      "loss": 0.0927,
+      "step": 1107
+    },
+    {
+      "epoch": 100.73,
+      "learning_rate": 6.762360446570973e-07,
+      "loss": 0.0877,
+      "step": 1108
+    },
+    {
+      "epoch": 100.82,
+      "learning_rate": 6.730462519936204e-07,
+      "loss": 0.0781,
+      "step": 1109
+    },
+    {
+      "epoch": 100.91,
+      "learning_rate": 6.698564593301435e-07,
+      "loss": 0.0801,
+      "step": 1110
+    },
+    {
+      "epoch": 100.91,
+      "eval_accuracy_safe": 0.5403988419474096,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9950662505252321,
+      "eval_iou_safe": 0.4650213883963596,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813382152747787,
+      "eval_loss": 0.08850335329771042,
+      "eval_mean_accuracy": 0.7677325462363209,
+      "eval_mean_iou": 0.7231798018355692,
+      "eval_overall_accuracy": 0.9816361042990613,
+      "eval_runtime": 10.9824,
+      "eval_samples_per_second": 6.101,
+      "eval_steps_per_second": 0.455,
+      "step": 1110
+    },
+    {
+      "epoch": 101.0,
+      "learning_rate": 6.666666666666666e-07,
+      "loss": 0.0754,
+      "step": 1111
+    },
+    {
+      "epoch": 101.09,
+      "learning_rate": 6.634768740031898e-07,
+      "loss": 0.1021,
+      "step": 1112
+    },
+    {
+      "epoch": 101.18,
+      "learning_rate": 6.602870813397129e-07,
+      "loss": 0.0887,
+      "step": 1113
+    },
+    {
+      "epoch": 101.27,
+      "learning_rate": 6.57097288676236e-07,
+      "loss": 0.0767,
+      "step": 1114
+    },
+    {
+      "epoch": 101.36,
+      "learning_rate": 6.539074960127592e-07,
+      "loss": 0.0833,
+      "step": 1115
+    },
+    {
+      "epoch": 101.45,
+      "learning_rate": 6.507177033492822e-07,
+      "loss": 0.0823,
+      "step": 1116
+    },
+    {
+      "epoch": 101.55,
+      "learning_rate": 6.475279106858054e-07,
+      "loss": 0.0755,
+      "step": 1117
+    },
+    {
+      "epoch": 101.64,
+      "learning_rate": 6.443381180223285e-07,
+      "loss": 0.0854,
+      "step": 1118
+    },
+    {
+      "epoch": 101.73,
+      "learning_rate": 6.411483253588517e-07,
+      "loss": 0.0954,
+      "step": 1119
+    },
+    {
+      "epoch": 101.82,
+      "learning_rate": 6.379585326953748e-07,
+      "loss": 0.1043,
+      "step": 1120
+    },
+    {
+      "epoch": 101.82,
+      "eval_accuracy_safe": 0.5444524115173033,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_accuracy_unsafe": 0.9949533718286454,
+      "eval_iou_safe": 0.46701916765456225,
+      "eval_iou_unlabeled": NaN,
+      "eval_iou_unsafe": 0.9813463023513723,
+      "eval_loss": 0.08907562494277954,
+      "eval_mean_accuracy": 0.7697028916729743,
+      "eval_mean_iou": 0.7241827350029673,
+      "eval_overall_accuracy": 0.9816462958036963,
+      "eval_runtime": 11.3992,
+      "eval_samples_per_second": 5.878,
+      "eval_steps_per_second": 0.439,
+      "step": 1120
+    }
+  ],
+  "max_steps": 1320,
+  "num_train_epochs": 120,
+  "total_flos": 1.2526623324487287e+19,
+  "trial_name": null,
+  "trial_params": null
+}