diff --git "a/checkpoint-1240/trainer_state.json" "b/checkpoint-1240/trainer_state.json"
deleted file mode 100644--- "a/checkpoint-1240/trainer_state.json"
+++ /dev/null
@@ -1,9564 +0,0 @@
-{
-  "best_metric": 0.22187493741512299,
-  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-train-SF-RGB-b5/checkpoint-1240",
-  "epoch": 112.72727272727273,
-  "global_step": 1240,
-  "is_hyper_param_search": false,
-  "is_local_process_zero": true,
-  "is_world_process_zero": true,
-  "log_history": [
-    {
-      "epoch": 0.09,
-      "learning_rate": 1.764705882352941e-07,
-      "loss": 1.2293,
-      "step": 1
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 3.529411764705882e-07,
-      "loss": 1.2349,
-      "step": 2
-    },
-    {
-      "epoch": 0.27,
-      "learning_rate": 5.294117647058824e-07,
-      "loss": 1.2357,
-      "step": 3
-    },
-    {
-      "epoch": 0.36,
-      "learning_rate": 7.058823529411765e-07,
-      "loss": 1.2242,
-      "step": 4
-    },
-    {
-      "epoch": 0.45,
-      "learning_rate": 8.823529411764706e-07,
-      "loss": 1.2247,
-      "step": 5
-    },
-    {
-      "epoch": 0.55,
-      "learning_rate": 1.0588235294117648e-06,
-      "loss": 1.227,
-      "step": 6
-    },
-    {
-      "epoch": 0.64,
-      "learning_rate": 1.2352941176470588e-06,
-      "loss": 1.2292,
-      "step": 7
-    },
-    {
-      "epoch": 0.73,
-      "learning_rate": 1.411764705882353e-06,
-      "loss": 1.2257,
-      "step": 8
-    },
-    {
-      "epoch": 0.82,
-      "learning_rate": 1.5882352941176472e-06,
-      "loss": 1.2173,
-      "step": 9
-    },
-    {
-      "epoch": 0.91,
-      "learning_rate": 1.7647058823529412e-06,
-      "loss": 1.2239,
-      "step": 10
-    },
-    {
-      "epoch": 0.91,
-      "eval_accuracy_safe": 0.39920432072351303,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.29510386893492613,
-      "eval_iou_safe": 0.031418797454432365,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.29392505413655395,
-      "eval_loss": 1.1102662086486816,
-      "eval_mean_accuracy": 0.34715409482921955,
-      "eval_mean_iou": 0.10844795053032878,
-      "eval_overall_accuracy": 0.2981788293639226,
-      "eval_runtime": 9.3061,
-      "eval_samples_per_second": 7.2,
-      "eval_steps_per_second": 0.537,
-      "step": 10
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 1.9411764705882353e-06,
-      "loss": 1.221,
-      "step": 11
-    },
-    {
-      "epoch": 1.09,
-      "learning_rate": 2.1176470588235296e-06,
-      "loss": 1.2149,
-      "step": 12
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 2.2941176470588234e-06,
-      "loss": 1.2197,
-      "step": 13
-    },
-    {
-      "epoch": 1.27,
-      "learning_rate": 2.4705882352941177e-06,
-      "loss": 1.2064,
-      "step": 14
-    },
-    {
-      "epoch": 1.36,
-      "learning_rate": 2.647058823529412e-06,
-      "loss": 1.2042,
-      "step": 15
-    },
-    {
-      "epoch": 1.45,
-      "learning_rate": 2.823529411764706e-06,
-      "loss": 1.2046,
-      "step": 16
-    },
-    {
-      "epoch": 1.55,
-      "learning_rate": 3e-06,
-      "loss": 1.2054,
-      "step": 17
-    },
-    {
-      "epoch": 1.64,
-      "learning_rate": 2.990415335463259e-06,
-      "loss": 1.2002,
-      "step": 18
-    },
-    {
-      "epoch": 1.73,
-      "learning_rate": 2.9808306709265177e-06,
-      "loss": 1.1982,
-      "step": 19
-    },
-    {
-      "epoch": 1.82,
-      "learning_rate": 2.9712460063897764e-06,
-      "loss": 1.1948,
-      "step": 20
-    },
-    {
-      "epoch": 1.82,
-      "eval_accuracy_safe": 0.5218599774094934,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.3705393407485172,
-      "eval_iou_safe": 0.0439841259320199,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.36893471456827454,
-      "eval_loss": 1.0963102579116821,
-      "eval_mean_accuracy": 0.4461996590790053,
-      "eval_mean_iou": 0.13763961350009815,
-      "eval_overall_accuracy": 0.37500910972481344,
-      "eval_runtime": 8.3942,
-      "eval_samples_per_second": 7.982,
-      "eval_steps_per_second": 0.596,
-      "step": 20
-    },
-    {
-      "epoch": 1.91,
-      "learning_rate": 2.9616613418530352e-06,
-      "loss": 1.1776,
-      "step": 21
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 2.952076677316294e-06,
-      "loss": 1.1936,
-      "step": 22
-    },
-    {
-      "epoch": 2.09,
-      "learning_rate": 2.942492012779553e-06,
-      "loss": 1.1664,
-      "step": 23
-    },
-    {
-      "epoch": 2.18,
-      "learning_rate": 2.9329073482428116e-06,
-      "loss": 1.1809,
-      "step": 24
-    },
-    {
-      "epoch": 2.27,
-      "learning_rate": 2.9233226837060704e-06,
-      "loss": 1.1731,
-      "step": 25
-    },
-    {
-      "epoch": 2.36,
-      "learning_rate": 2.913738019169329e-06,
-      "loss": 1.1794,
-      "step": 26
-    },
-    {
-      "epoch": 2.45,
-      "learning_rate": 2.904153354632588e-06,
-      "loss": 1.1684,
-      "step": 27
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 2.8945686900958468e-06,
-      "loss": 1.1594,
-      "step": 28
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 2.884984025559105e-06,
-      "loss": 1.146,
-      "step": 29
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 2.8753993610223644e-06,
-      "loss": 1.1661,
-      "step": 30
-    },
-    {
-      "epoch": 2.73,
-      "eval_accuracy_safe": 0.5863200218966003,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.49882181393718666,
-      "eval_iou_safe": 0.06472096392925002,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.49613792944888824,
-      "eval_loss": 1.0516184568405151,
-      "eval_mean_accuracy": 0.5425709179168935,
-      "eval_mean_iou": 0.1869529644593794,
-      "eval_overall_accuracy": 0.5014063707038537,
-      "eval_runtime": 9.3615,
-      "eval_samples_per_second": 7.157,
-      "eval_steps_per_second": 0.534,
-      "step": 30
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 2.865814696485623e-06,
-      "loss": 1.153,
-      "step": 31
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 2.856230031948882e-06,
-      "loss": 1.149,
-      "step": 32
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 2.8466453674121407e-06,
-      "loss": 1.164,
-      "step": 33
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 2.8370607028753995e-06,
-      "loss": 1.1479,
-      "step": 34
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 2.8274760383386583e-06,
-      "loss": 1.1428,
-      "step": 35
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 2.817891373801917e-06,
-      "loss": 1.1398,
-      "step": 36
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 2.8083067092651755e-06,
-      "loss": 1.1415,
-      "step": 37
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 2.7987220447284347e-06,
-      "loss": 1.1146,
-      "step": 38
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 2.7891373801916935e-06,
-      "loss": 1.1234,
-      "step": 39
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 2.7795527156549523e-06,
-      "loss": 1.1112,
-      "step": 40
-    },
-    {
-      "epoch": 3.64,
-      "eval_accuracy_safe": 0.5458517893146133,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.5794390867479824,
-      "eval_iou_safe": 0.0899865428032596,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.5753581085678546,
-      "eval_loss": 1.0047743320465088,
-      "eval_mean_accuracy": 0.5626454380312979,
-      "eval_mean_iou": 0.22178155045703807,
-      "eval_overall_accuracy": 0.5784469718363747,
-      "eval_runtime": 9.38,
-      "eval_samples_per_second": 7.143,
-      "eval_steps_per_second": 0.533,
-      "step": 40
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 2.769968051118211e-06,
-      "loss": 1.1212,
-      "step": 41
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 2.7603833865814694e-06,
-      "loss": 1.1117,
-      "step": 42
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 2.7507987220447286e-06,
-      "loss": 1.1051,
-      "step": 43
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 2.7412140575079874e-06,
-      "loss": 1.1382,
-      "step": 44
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 2.731629392971246e-06,
-      "loss": 1.1075,
-      "step": 45
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 2.722044728434505e-06,
-      "loss": 1.1096,
-      "step": 46
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 2.7124600638977634e-06,
-      "loss": 1.0974,
-      "step": 47
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 2.7028753993610226e-06,
-      "loss": 1.0851,
-      "step": 48
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 2.6932907348242814e-06,
-      "loss": 1.0851,
-      "step": 49
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 2.6837060702875398e-06,
-      "loss": 1.0907,
-      "step": 50
-    },
-    {
-      "epoch": 4.55,
-      "eval_accuracy_safe": 0.5993018531154467,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.6366664738420048,
-      "eval_iou_safe": 0.10939424133801141,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.6320960572783577,
-      "eval_loss": 0.9689871668815613,
-      "eval_mean_accuracy": 0.6179841634787258,
-      "eval_mean_iou": 0.24716343287212306,
-      "eval_overall_accuracy": 0.6355627828569554,
-      "eval_runtime": 9.8119,
-      "eval_samples_per_second": 6.828,
-      "eval_steps_per_second": 0.51,
-      "step": 50
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 2.674121405750799e-06,
-      "loss": 1.0789,
-      "step": 51
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 2.6645367412140573e-06,
-      "loss": 1.0566,
-      "step": 52
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 2.654952076677316e-06,
-      "loss": 1.0718,
-      "step": 53
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 2.6453674121405753e-06,
-      "loss": 1.0808,
-      "step": 54
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 2.6357827476038337e-06,
-      "loss": 1.0808,
-      "step": 55
-    },
-    {
-      "epoch": 5.09,
-      "learning_rate": 2.626198083067093e-06,
-      "loss": 1.0625,
-      "step": 56
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 2.6166134185303517e-06,
-      "loss": 1.0508,
-      "step": 57
-    },
-    {
-      "epoch": 5.27,
-      "learning_rate": 2.60702875399361e-06,
-      "loss": 1.0521,
-      "step": 58
-    },
-    {
-      "epoch": 5.36,
-      "learning_rate": 2.5974440894568693e-06,
-      "loss": 1.052,
-      "step": 59
-    },
-    {
-      "epoch": 5.45,
-      "learning_rate": 2.5878594249201277e-06,
-      "loss": 1.047,
-      "step": 60
-    },
-    {
-      "epoch": 5.45,
-      "eval_accuracy_safe": 0.6691570194409427,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.669918812994849,
-      "eval_iou_safe": 0.11592410370525706,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.6655783206367077,
-      "eval_loss": 0.9437160491943359,
-      "eval_mean_accuracy": 0.6695379162178958,
-      "eval_mean_iou": 0.2605008081139883,
-      "eval_overall_accuracy": 0.6698963108347423,
-      "eval_runtime": 9.3506,
-      "eval_samples_per_second": 7.165,
-      "eval_steps_per_second": 0.535,
-      "step": 60
-    },
-    {
-      "epoch": 5.55,
-      "learning_rate": 2.5782747603833865e-06,
-      "loss": 1.0456,
-      "step": 61
-    },
-    {
-      "epoch": 5.64,
-      "learning_rate": 2.5686900958466457e-06,
-      "loss": 1.0296,
-      "step": 62
-    },
-    {
-      "epoch": 5.73,
-      "learning_rate": 2.559105431309904e-06,
-      "loss": 1.0334,
-      "step": 63
-    },
-    {
-      "epoch": 5.82,
-      "learning_rate": 2.5495207667731633e-06,
-      "loss": 1.0525,
-      "step": 64
-    },
-    {
-      "epoch": 5.91,
-      "learning_rate": 2.5399361022364216e-06,
-      "loss": 1.0397,
-      "step": 65
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 2.5303514376996804e-06,
-      "loss": 1.0575,
-      "step": 66
-    },
-    {
-      "epoch": 6.09,
-      "learning_rate": 2.5207667731629396e-06,
-      "loss": 1.0245,
-      "step": 67
-    },
-    {
-      "epoch": 6.18,
-      "learning_rate": 2.511182108626198e-06,
-      "loss": 1.0284,
-      "step": 68
-    },
-    {
-      "epoch": 6.27,
-      "learning_rate": 2.501597444089457e-06,
-      "loss": 1.0127,
-      "step": 69
-    },
-    {
-      "epoch": 6.36,
-      "learning_rate": 2.4920127795527156e-06,
-      "loss": 1.0112,
-      "step": 70
-    },
-    {
-      "epoch": 6.36,
-      "eval_accuracy_safe": 0.6673046750012529,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.718865515124044,
-      "eval_iou_safe": 0.13494764234639625,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.7136723036690169,
-      "eval_loss": 0.9084014892578125,
-      "eval_mean_accuracy": 0.6930850950626484,
-      "eval_mean_iou": 0.28287331533847104,
-      "eval_overall_accuracy": 0.7173424905805446,
-      "eval_runtime": 9.0845,
-      "eval_samples_per_second": 7.375,
-      "eval_steps_per_second": 0.55,
-      "step": 70
-    },
-    {
-      "epoch": 6.45,
-      "learning_rate": 2.4824281150159744e-06,
-      "loss": 1.0107,
-      "step": 71
-    },
-    {
-      "epoch": 6.55,
-      "learning_rate": 2.4728434504792336e-06,
-      "loss": 1.0122,
-      "step": 72
-    },
-    {
-      "epoch": 6.64,
-      "learning_rate": 2.463258785942492e-06,
-      "loss": 1.0087,
-      "step": 73
-    },
-    {
-      "epoch": 6.73,
-      "learning_rate": 2.4536741214057507e-06,
-      "loss": 0.991,
-      "step": 74
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 2.44408945686901e-06,
-      "loss": 1.0042,
-      "step": 75
-    },
-    {
-      "epoch": 6.91,
-      "learning_rate": 2.4345047923322683e-06,
-      "loss": 0.9788,
-      "step": 76
-    },
-    {
-      "epoch": 7.0,
-      "learning_rate": 2.424920127795527e-06,
-      "loss": 1.0371,
-      "step": 77
-    },
-    {
-      "epoch": 7.09,
-      "learning_rate": 2.415335463258786e-06,
-      "loss": 0.9884,
-      "step": 78
-    },
-    {
-      "epoch": 7.18,
-      "learning_rate": 2.4057507987220447e-06,
-      "loss": 0.9759,
-      "step": 79
-    },
-    {
-      "epoch": 7.27,
-      "learning_rate": 2.396166134185304e-06,
-      "loss": 0.9925,
-      "step": 80
-    },
-    {
-      "epoch": 7.27,
-      "eval_accuracy_safe": 0.6842282797676185,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.7664939888573942,
-      "eval_iou_safe": 0.14521835177647324,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.7605469001045981,
-      "eval_loss": 0.8647122979164124,
-      "eval_mean_accuracy": 0.7253611343125064,
-      "eval_mean_iou": 0.3019217506270238,
-      "eval_overall_accuracy": 0.7640639917174382,
-      "eval_runtime": 9.9787,
-      "eval_samples_per_second": 6.714,
-      "eval_steps_per_second": 0.501,
-      "step": 80
-    },
-    {
-      "epoch": 7.36,
-      "learning_rate": 2.3865814696485623e-06,
-      "loss": 0.9784,
-      "step": 81
-    },
-    {
-      "epoch": 7.45,
-      "learning_rate": 2.376996805111821e-06,
-      "loss": 0.9806,
-      "step": 82
-    },
-    {
-      "epoch": 7.55,
-      "learning_rate": 2.36741214057508e-06,
-      "loss": 0.973,
-      "step": 83
-    },
-    {
-      "epoch": 7.64,
-      "learning_rate": 2.3578274760383387e-06,
-      "loss": 0.9551,
-      "step": 84
-    },
-    {
-      "epoch": 7.73,
-      "learning_rate": 2.3482428115015974e-06,
-      "loss": 0.9484,
-      "step": 85
-    },
-    {
-      "epoch": 7.82,
-      "learning_rate": 2.3386581469648562e-06,
-      "loss": 0.939,
-      "step": 86
-    },
-    {
-      "epoch": 7.91,
-      "learning_rate": 2.329073482428115e-06,
-      "loss": 0.9533,
-      "step": 87
-    },
-    {
-      "epoch": 8.0,
-      "learning_rate": 2.319488817891374e-06,
-      "loss": 0.9533,
-      "step": 88
-    },
-    {
-      "epoch": 8.09,
-      "learning_rate": 2.3099041533546326e-06,
-      "loss": 0.9469,
-      "step": 89
-    },
-    {
-      "epoch": 8.18,
-      "learning_rate": 2.3003194888178914e-06,
-      "loss": 0.9395,
-      "step": 90
-    },
-    {
-      "epoch": 8.18,
-      "eval_accuracy_safe": 0.6817822598987667,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.7920860651953089,
-      "eval_iou_safe": 0.16203325137990066,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.785611270080723,
-      "eval_loss": 0.8319001197814941,
-      "eval_mean_accuracy": 0.7369341625470378,
-      "eval_mean_iou": 0.3158815071535412,
-      "eval_overall_accuracy": 0.788827867650274,
-      "eval_runtime": 9.682,
-      "eval_samples_per_second": 6.92,
-      "eval_steps_per_second": 0.516,
-      "step": 90
-    },
-    {
-      "epoch": 8.27,
-      "learning_rate": 2.29073482428115e-06,
-      "loss": 0.9422,
-      "step": 91
-    },
-    {
-      "epoch": 8.36,
-      "learning_rate": 2.281150159744409e-06,
-      "loss": 0.9012,
-      "step": 92
-    },
-    {
-      "epoch": 8.45,
-      "learning_rate": 2.2715654952076678e-06,
-      "loss": 0.937,
-      "step": 93
-    },
-    {
-      "epoch": 8.55,
-      "learning_rate": 2.2619808306709266e-06,
-      "loss": 0.9122,
-      "step": 94
-    },
-    {
-      "epoch": 8.64,
-      "learning_rate": 2.2523961661341854e-06,
-      "loss": 0.9272,
-      "step": 95
-    },
-    {
-      "epoch": 8.73,
-      "learning_rate": 2.242811501597444e-06,
-      "loss": 0.9122,
-      "step": 96
-    },
-    {
-      "epoch": 8.82,
-      "learning_rate": 2.233226837060703e-06,
-      "loss": 0.9261,
-      "step": 97
-    },
-    {
-      "epoch": 8.91,
-      "learning_rate": 2.2236421725239617e-06,
-      "loss": 0.9004,
-      "step": 98
-    },
-    {
-      "epoch": 9.0,
-      "learning_rate": 2.2140575079872205e-06,
-      "loss": 0.8981,
-      "step": 99
-    },
-    {
-      "epoch": 9.09,
-      "learning_rate": 2.2044728434504793e-06,
-      "loss": 0.8902,
-      "step": 100
-    },
-    {
-      "epoch": 9.09,
-      "eval_accuracy_safe": 0.6805698513112902,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8141590718977455,
-      "eval_iou_safe": 0.1769983948474704,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8072069422555583,
-      "eval_loss": 0.8013909459114075,
-      "eval_mean_accuracy": 0.7473644616045179,
-      "eval_mean_iou": 0.32806844570100957,
-      "eval_overall_accuracy": 0.8102130605213678,
-      "eval_runtime": 9.6492,
-      "eval_samples_per_second": 6.944,
-      "eval_steps_per_second": 0.518,
-      "step": 100
-    },
-    {
-      "epoch": 9.18,
-      "learning_rate": 2.194888178913738e-06,
-      "loss": 0.9085,
-      "step": 101
-    },
-    {
-      "epoch": 9.27,
-      "learning_rate": 2.185303514376997e-06,
-      "loss": 0.8976,
-      "step": 102
-    },
-    {
-      "epoch": 9.36,
-      "learning_rate": 2.1757188498402557e-06,
-      "loss": 0.8909,
-      "step": 103
-    },
-    {
-      "epoch": 9.45,
-      "learning_rate": 2.1661341853035145e-06,
-      "loss": 0.8878,
-      "step": 104
-    },
-    {
-      "epoch": 9.55,
-      "learning_rate": 2.1565495207667733e-06,
-      "loss": 0.8909,
-      "step": 105
-    },
-    {
-      "epoch": 9.64,
-      "learning_rate": 2.146964856230032e-06,
-      "loss": 0.8529,
-      "step": 106
-    },
-    {
-      "epoch": 9.73,
-      "learning_rate": 2.137380191693291e-06,
-      "loss": 0.9097,
-      "step": 107
-    },
-    {
-      "epoch": 9.82,
-      "learning_rate": 2.1277955271565496e-06,
-      "loss": 0.8778,
-      "step": 108
-    },
-    {
-      "epoch": 9.91,
-      "learning_rate": 2.1182108626198084e-06,
-      "loss": 0.8819,
-      "step": 109
-    },
-    {
-      "epoch": 10.0,
-      "learning_rate": 2.1086261980830672e-06,
-      "loss": 0.9057,
-      "step": 110
-    },
-    {
-      "epoch": 10.0,
-      "eval_accuracy_safe": 0.6984244470915687,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8178638868312451,
-      "eval_iou_safe": 0.17330681673606707,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8109252159162895,
-      "eval_loss": 0.7867299914360046,
-      "eval_mean_accuracy": 0.7581441669614069,
-      "eval_mean_iou": 0.3280773442174522,
-      "eval_overall_accuracy": 0.814335837293027,
-      "eval_runtime": 9.8362,
-      "eval_samples_per_second": 6.812,
-      "eval_steps_per_second": 0.508,
-      "step": 110
-    },
-    {
-      "epoch": 10.09,
-      "learning_rate": 2.099041533546326e-06,
-      "loss": 0.8509,
-      "step": 111
-    },
-    {
-      "epoch": 10.18,
-      "learning_rate": 2.089456869009585e-06,
-      "loss": 0.8746,
-      "step": 112
-    },
-    {
-      "epoch": 10.27,
-      "learning_rate": 2.0798722044728436e-06,
-      "loss": 0.8532,
-      "step": 113
-    },
-    {
-      "epoch": 10.36,
-      "learning_rate": 2.0702875399361024e-06,
-      "loss": 0.8509,
-      "step": 114
-    },
-    {
-      "epoch": 10.45,
-      "learning_rate": 2.060702875399361e-06,
-      "loss": 0.8559,
-      "step": 115
-    },
-    {
-      "epoch": 10.55,
-      "learning_rate": 2.05111821086262e-06,
-      "loss": 0.8435,
-      "step": 116
-    },
-    {
-      "epoch": 10.64,
-      "learning_rate": 2.0415335463258783e-06,
-      "loss": 0.8465,
-      "step": 117
-    },
-    {
-      "epoch": 10.73,
-      "learning_rate": 2.0319488817891376e-06,
-      "loss": 0.8345,
-      "step": 118
-    },
-    {
-      "epoch": 10.82,
-      "learning_rate": 2.0223642172523963e-06,
-      "loss": 0.8403,
-      "step": 119
-    },
-    {
-      "epoch": 10.91,
-      "learning_rate": 2.012779552715655e-06,
-      "loss": 0.8321,
-      "step": 120
-    },
-    {
-      "epoch": 10.91,
-      "eval_accuracy_safe": 0.6743748096576343,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8493713583566551,
-      "eval_iou_safe": 0.18617264978095113,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8413265356526651,
-      "eval_loss": 0.7439588308334351,
-      "eval_mean_accuracy": 0.7618730840071447,
-      "eval_mean_iou": 0.3424997284778721,
-      "eval_overall_accuracy": 0.8442022409012069,
-      "eval_runtime": 9.4999,
-      "eval_samples_per_second": 7.053,
-      "eval_steps_per_second": 0.526,
-      "step": 120
-    },
-    {
-      "epoch": 11.0,
-      "learning_rate": 2.003194888178914e-06,
-      "loss": 0.8343,
-      "step": 121
-    },
-    {
-      "epoch": 11.09,
-      "learning_rate": 1.9936102236421723e-06,
-      "loss": 0.8413,
-      "step": 122
-    },
-    {
-      "epoch": 11.18,
-      "learning_rate": 1.9840255591054315e-06,
-      "loss": 0.8037,
-      "step": 123
-    },
-    {
-      "epoch": 11.27,
-      "learning_rate": 1.9744408945686903e-06,
-      "loss": 0.8284,
-      "step": 124
-    },
-    {
-      "epoch": 11.36,
-      "learning_rate": 1.9648562300319487e-06,
-      "loss": 0.8023,
-      "step": 125
-    },
-    {
-      "epoch": 11.45,
-      "learning_rate": 1.955271565495208e-06,
-      "loss": 0.8176,
-      "step": 126
-    },
-    {
-      "epoch": 11.55,
-      "learning_rate": 1.9456869009584667e-06,
-      "loss": 0.8078,
-      "step": 127
-    },
-    {
-      "epoch": 11.64,
-      "learning_rate": 1.936102236421725e-06,
-      "loss": 0.8074,
-      "step": 128
-    },
-    {
-      "epoch": 11.73,
-      "learning_rate": 1.9265175718849843e-06,
-      "loss": 0.8139,
-      "step": 129
-    },
-    {
-      "epoch": 11.82,
-      "learning_rate": 1.9169329073482426e-06,
-      "loss": 0.8152,
-      "step": 130
-    },
-    {
-      "epoch": 11.82,
-      "eval_accuracy_safe": 0.6688004286799203,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8589712690862681,
-      "eval_iou_safe": 0.20063653201077156,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8506617582762998,
-      "eval_loss": 0.7270171046257019,
-      "eval_mean_accuracy": 0.7638858488830942,
-      "eval_mean_iou": 0.3504327634290238,
-      "eval_overall_accuracy": 0.8533539273845616,
-      "eval_runtime": 11.0671,
-      "eval_samples_per_second": 6.054,
-      "eval_steps_per_second": 0.452,
-      "step": 130
-    },
-    {
-      "epoch": 11.91,
-      "learning_rate": 1.9073482428115018e-06,
-      "loss": 0.8102,
-      "step": 131
-    },
-    {
-      "epoch": 12.0,
-      "learning_rate": 1.8977635782747604e-06,
-      "loss": 0.8192,
-      "step": 132
-    },
-    {
-      "epoch": 12.09,
-      "learning_rate": 1.888178913738019e-06,
-      "loss": 0.8246,
-      "step": 133
-    },
-    {
-      "epoch": 12.18,
-      "learning_rate": 1.878594249201278e-06,
-      "loss": 0.7864,
-      "step": 134
-    },
-    {
-      "epoch": 12.27,
-      "learning_rate": 1.8690095846645368e-06,
-      "loss": 0.7718,
-      "step": 135
-    },
-    {
-      "epoch": 12.36,
-      "learning_rate": 1.8594249201277954e-06,
-      "loss": 0.8074,
-      "step": 136
-    },
-    {
-      "epoch": 12.45,
-      "learning_rate": 1.8498402555910544e-06,
-      "loss": 0.7569,
-      "step": 137
-    },
-    {
-      "epoch": 12.55,
-      "learning_rate": 1.8402555910543132e-06,
-      "loss": 0.7989,
-      "step": 138
-    },
-    {
-      "epoch": 12.64,
-      "learning_rate": 1.8306709265175722e-06,
-      "loss": 0.7987,
-      "step": 139
-    },
-    {
-      "epoch": 12.73,
-      "learning_rate": 1.8210862619808307e-06,
-      "loss": 0.7929,
-      "step": 140
-    },
-    {
-      "epoch": 12.73,
-      "eval_accuracy_safe": 0.6659920354971646,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8656634973410731,
-      "eval_iou_safe": 0.20852726465106847,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8572303764577973,
-      "eval_loss": 0.7045122981071472,
-      "eval_mean_accuracy": 0.7658277664191189,
-      "eval_mean_iou": 0.3552525470362886,
-      "eval_overall_accuracy": 0.8597655225155959,
-      "eval_runtime": 9.4502,
-      "eval_samples_per_second": 7.09,
-      "eval_steps_per_second": 0.529,
-      "step": 140
-    },
-    {
-      "epoch": 12.82,
-      "learning_rate": 1.8115015974440893e-06,
-      "loss": 0.7703,
-      "step": 141
-    },
-    {
-      "epoch": 12.91,
-      "learning_rate": 1.8019169329073483e-06,
-      "loss": 0.768,
-      "step": 142
-    },
-    {
-      "epoch": 13.0,
-      "learning_rate": 1.7923322683706071e-06,
-      "loss": 0.8397,
-      "step": 143
-    },
-    {
-      "epoch": 13.09,
-      "learning_rate": 1.7827476038338657e-06,
-      "loss": 0.775,
-      "step": 144
-    },
-    {
-      "epoch": 13.18,
-      "learning_rate": 1.7731629392971247e-06,
-      "loss": 0.7718,
-      "step": 145
-    },
-    {
-      "epoch": 13.27,
-      "learning_rate": 1.7635782747603833e-06,
-      "loss": 0.7544,
-      "step": 146
-    },
-    {
-      "epoch": 13.36,
-      "learning_rate": 1.7539936102236423e-06,
-      "loss": 0.7585,
-      "step": 147
-    },
-    {
-      "epoch": 13.45,
-      "learning_rate": 1.744408945686901e-06,
-      "loss": 0.742,
-      "step": 148
-    },
-    {
-      "epoch": 13.55,
-      "learning_rate": 1.7348242811501597e-06,
-      "loss": 0.7629,
-      "step": 149
-    },
-    {
-      "epoch": 13.64,
-      "learning_rate": 1.7252396166134187e-06,
-      "loss": 0.7568,
-      "step": 150
-    },
-    {
-      "epoch": 13.64,
-      "eval_accuracy_safe": 0.657079193989229,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8838084544735693,
-      "eval_iou_safe": 0.21846087789622765,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.874847055157359,
-      "eval_loss": 0.6744459867477417,
-      "eval_mean_accuracy": 0.7704438242313991,
-      "eval_mean_iou": 0.3644359776845289,
-      "eval_overall_accuracy": 0.8771112356612931,
-      "eval_runtime": 10.8117,
-      "eval_samples_per_second": 6.197,
-      "eval_steps_per_second": 0.462,
-      "step": 150
-    },
-    {
-      "epoch": 13.73,
-      "learning_rate": 1.7156549520766772e-06,
-      "loss": 0.7311,
-      "step": 151
-    },
-    {
-      "epoch": 13.82,
-      "learning_rate": 1.706070287539936e-06,
-      "loss": 0.7352,
-      "step": 152
-    },
-    {
-      "epoch": 13.91,
-      "learning_rate": 1.696485623003195e-06,
-      "loss": 0.7661,
-      "step": 153
-    },
-    {
-      "epoch": 14.0,
-      "learning_rate": 1.6869009584664536e-06,
-      "loss": 0.7335,
-      "step": 154
-    },
-    {
-      "epoch": 14.09,
-      "learning_rate": 1.6773162939297126e-06,
-      "loss": 0.7491,
-      "step": 155
-    },
-    {
-      "epoch": 14.18,
-      "learning_rate": 1.6677316293929714e-06,
-      "loss": 0.715,
-      "step": 156
-    },
-    {
-      "epoch": 14.27,
-      "learning_rate": 1.65814696485623e-06,
-      "loss": 0.7349,
-      "step": 157
-    },
-    {
-      "epoch": 14.36,
-      "learning_rate": 1.648562300319489e-06,
-      "loss": 0.7127,
-      "step": 158
-    },
-    {
-      "epoch": 14.45,
-      "learning_rate": 1.6389776357827476e-06,
-      "loss": 0.718,
-      "step": 159
-    },
-    {
-      "epoch": 14.55,
-      "learning_rate": 1.6293929712460064e-06,
-      "loss": 0.7085,
-      "step": 160
-    },
-    {
-      "epoch": 14.55,
-      "eval_accuracy_safe": 0.6519423595128777,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8934174588611713,
-      "eval_iou_safe": 0.2259934305828019,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8842020655315154,
-      "eval_loss": 0.65559983253479,
-      "eval_mean_accuracy": 0.7726799091870244,
-      "eval_mean_iou": 0.3700651653714391,
-      "eval_overall_accuracy": 0.8862846716126399,
-      "eval_runtime": 9.325,
-      "eval_samples_per_second": 7.185,
-      "eval_steps_per_second": 0.536,
-      "step": 160
-    },
-    {
-      "epoch": 14.64,
-      "learning_rate": 1.6198083067092654e-06,
-      "loss": 0.753,
-      "step": 161
-    },
-    {
-      "epoch": 14.73,
-      "learning_rate": 1.610223642172524e-06,
-      "loss": 0.7376,
-      "step": 162
-    },
-    {
-      "epoch": 14.82,
-      "learning_rate": 1.600638977635783e-06,
-      "loss": 0.7256,
-      "step": 163
-    },
-    {
-      "epoch": 14.91,
-      "learning_rate": 1.5910543130990415e-06,
-      "loss": 0.7132,
-      "step": 164
-    },
-    {
-      "epoch": 15.0,
-      "learning_rate": 1.5814696485623003e-06,
-      "loss": 0.722,
-      "step": 165
-    },
-    {
-      "epoch": 15.09,
-      "learning_rate": 1.5718849840255593e-06,
-      "loss": 0.7015,
-      "step": 166
-    },
-    {
-      "epoch": 15.18,
-      "learning_rate": 1.562300319488818e-06,
-      "loss": 0.7176,
-      "step": 167
-    },
-    {
-      "epoch": 15.27,
-      "learning_rate": 1.5527156549520765e-06,
-      "loss": 0.7147,
-      "step": 168
-    },
-    {
-      "epoch": 15.36,
-      "learning_rate": 1.5431309904153355e-06,
-      "loss": 0.7217,
-      "step": 169
-    },
-    {
-      "epoch": 15.45,
-      "learning_rate": 1.5335463258785943e-06,
-      "loss": 0.7147,
-      "step": 170
-    },
-    {
-      "epoch": 15.45,
-      "eval_accuracy_safe": 0.6561000150346374,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8963761831582403,
-      "eval_iou_safe": 0.22827906452703248,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8872234626234047,
-      "eval_loss": 0.6508952975273132,
-      "eval_mean_accuracy": 0.7762380990964388,
-      "eval_mean_iou": 0.3718341757168124,
-      "eval_overall_accuracy": 0.889278810415695,
-      "eval_runtime": 10.5537,
-      "eval_samples_per_second": 6.348,
-      "eval_steps_per_second": 0.474,
-      "step": 170
-    },
-    {
-      "epoch": 15.55,
-      "learning_rate": 1.5239616613418533e-06,
-      "loss": 0.7212,
-      "step": 171
-    },
-    {
-      "epoch": 15.64,
-      "learning_rate": 1.5143769968051119e-06,
-      "loss": 0.7196,
-      "step": 172
-    },
-    {
-      "epoch": 15.73,
-      "learning_rate": 1.5047923322683706e-06,
-      "loss": 0.6797,
-      "step": 173
-    },
-    {
-      "epoch": 15.82,
-      "learning_rate": 1.4952076677316294e-06,
-      "loss": 0.6626,
-      "step": 174
-    },
-    {
-      "epoch": 15.91,
-      "learning_rate": 1.4856230031948882e-06,
-      "loss": 0.6783,
-      "step": 175
-    },
-    {
-      "epoch": 16.0,
-      "learning_rate": 1.476038338658147e-06,
-      "loss": 0.7436,
-      "step": 176
-    },
-    {
-      "epoch": 16.09,
-      "learning_rate": 1.4664536741214058e-06,
-      "loss": 0.7123,
-      "step": 177
-    },
-    {
-      "epoch": 16.18,
-      "learning_rate": 1.4568690095846646e-06,
-      "loss": 0.7241,
-      "step": 178
-    },
-    {
-      "epoch": 16.27,
-      "learning_rate": 1.4472843450479234e-06,
-      "loss": 0.6914,
-      "step": 179
-    },
-    {
-      "epoch": 16.36,
-      "learning_rate": 1.4376996805111822e-06,
-      "loss": 0.6991,
-      "step": 180
-    },
-    {
-      "epoch": 16.36,
-      "eval_accuracy_safe": 0.6619847263503225,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.8964291023808605,
-      "eval_iou_safe": 0.22672413565462626,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.8874142561414083,
-      "eval_loss": 0.6502141952514648,
-      "eval_mean_accuracy": 0.7792069143655915,
-      "eval_mean_iou": 0.37137946393201154,
-      "eval_overall_accuracy": 0.889503991425927,
-      "eval_runtime": 9.3648,
-      "eval_samples_per_second": 7.154,
-      "eval_steps_per_second": 0.534,
-      "step": 180
-    },
-    {
-      "epoch": 16.45,
-      "learning_rate": 1.428115015974441e-06,
-      "loss": 0.6552,
-      "step": 181
-    },
-    {
-      "epoch": 16.55,
-      "learning_rate": 1.4185303514376998e-06,
-      "loss": 0.6621,
-      "step": 182
-    },
-    {
-      "epoch": 16.64,
-      "learning_rate": 1.4089456869009586e-06,
-      "loss": 0.6742,
-      "step": 183
-    },
-    {
-      "epoch": 16.73,
-      "learning_rate": 1.3993610223642173e-06,
-      "loss": 0.6462,
-      "step": 184
-    },
-    {
-      "epoch": 16.82,
-      "learning_rate": 1.3897763578274761e-06,
-      "loss": 0.6759,
-      "step": 185
-    },
-    {
-      "epoch": 16.91,
-      "learning_rate": 1.3801916932907347e-06,
-      "loss": 0.6773,
-      "step": 186
-    },
-    {
-      "epoch": 17.0,
-      "learning_rate": 1.3706070287539937e-06,
-      "loss": 0.7158,
-      "step": 187
-    },
-    {
-      "epoch": 17.09,
-      "learning_rate": 1.3610223642172525e-06,
-      "loss": 0.6958,
-      "step": 188
-    },
-    {
-      "epoch": 17.18,
-      "learning_rate": 1.3514376996805113e-06,
-      "loss": 0.6747,
-      "step": 189
-    },
-    {
-      "epoch": 17.27,
-      "learning_rate": 1.3418530351437699e-06,
-      "loss": 0.6357,
-      "step": 190
-    },
-    {
-      "epoch": 17.27,
-      "eval_accuracy_safe": 0.6611674588764114,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9051011079830231,
-      "eval_iou_safe": 0.24110413837507133,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.895975219611748,
-      "eval_loss": 0.6229549646377563,
-      "eval_mean_accuracy": 0.7831342834297172,
-      "eval_mean_iou": 0.3790264526622731,
-      "eval_overall_accuracy": 0.8978956991167211,
-      "eval_runtime": 10.2719,
-      "eval_samples_per_second": 6.523,
-      "eval_steps_per_second": 0.487,
-      "step": 190
-    },
-    {
-      "epoch": 17.36,
-      "learning_rate": 1.3322683706070287e-06,
-      "loss": 0.6769,
-      "step": 191
-    },
-    {
-      "epoch": 17.45,
-      "learning_rate": 1.3226837060702877e-06,
-      "loss": 0.6396,
-      "step": 192
-    },
-    {
-      "epoch": 17.55,
-      "learning_rate": 1.3130990415335465e-06,
-      "loss": 0.6567,
-      "step": 193
-    },
-    {
-      "epoch": 17.64,
-      "learning_rate": 1.303514376996805e-06,
-      "loss": 0.6538,
-      "step": 194
-    },
-    {
-      "epoch": 17.73,
-      "learning_rate": 1.2939297124600638e-06,
-      "loss": 0.6654,
-      "step": 195
-    },
-    {
-      "epoch": 17.82,
-      "learning_rate": 1.2843450479233228e-06,
-      "loss": 0.6637,
-      "step": 196
-    },
-    {
-      "epoch": 17.91,
-      "learning_rate": 1.2747603833865816e-06,
-      "loss": 0.6417,
-      "step": 197
-    },
-    {
-      "epoch": 18.0,
-      "learning_rate": 1.2651757188498402e-06,
-      "loss": 0.6707,
-      "step": 198
-    },
-    {
-      "epoch": 18.09,
-      "learning_rate": 1.255591054313099e-06,
-      "loss": 0.6229,
-      "step": 199
-    },
-    {
-      "epoch": 18.18,
-      "learning_rate": 1.2460063897763578e-06,
-      "loss": 0.6815,
-      "step": 200
-    },
-    {
-      "epoch": 18.18,
-      "eval_accuracy_safe": 0.6484207848080771,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9178010760554833,
-      "eval_iou_safe": 0.2593923131195133,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.908185465668198,
-      "eval_loss": 0.5992904305458069,
-      "eval_mean_accuracy": 0.7831109304317803,
-      "eval_mean_iou": 0.3891925929292371,
-      "eval_overall_accuracy": 0.9098440141820195,
-      "eval_runtime": 9.7351,
-      "eval_samples_per_second": 6.882,
-      "eval_steps_per_second": 0.514,
-      "step": 200
-    },
-    {
-      "epoch": 18.27,
-      "learning_rate": 1.2364217252396168e-06,
-      "loss": 0.6677,
-      "step": 201
-    },
-    {
-      "epoch": 18.36,
-      "learning_rate": 1.2268370607028754e-06,
-      "loss": 0.6658,
-      "step": 202
-    },
-    {
-      "epoch": 18.45,
-      "learning_rate": 1.2172523961661342e-06,
-      "loss": 0.649,
-      "step": 203
-    },
-    {
-      "epoch": 18.55,
-      "learning_rate": 1.207667731629393e-06,
-      "loss": 0.6342,
-      "step": 204
-    },
-    {
-      "epoch": 18.64,
-      "learning_rate": 1.198083067092652e-06,
-      "loss": 0.6441,
-      "step": 205
-    },
-    {
-      "epoch": 18.73,
-      "learning_rate": 1.1884984025559105e-06,
-      "loss": 0.6429,
-      "step": 206
-    },
-    {
-      "epoch": 18.82,
-      "learning_rate": 1.1789137380191693e-06,
-      "loss": 0.6155,
-      "step": 207
-    },
-    {
-      "epoch": 18.91,
-      "learning_rate": 1.1693290734824281e-06,
-      "loss": 0.6347,
-      "step": 208
-    },
-    {
-      "epoch": 19.0,
-      "learning_rate": 1.159744408945687e-06,
-      "loss": 0.6611,
-      "step": 209
-    },
-    {
-      "epoch": 19.09,
-      "learning_rate": 1.1501597444089457e-06,
-      "loss": 0.6398,
-      "step": 210
-    },
-    {
-      "epoch": 19.09,
-      "eval_accuracy_safe": 0.6413602877398313,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9258418644556835,
-      "eval_iou_safe": 0.26823363821482293,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9159351131912487,
-      "eval_loss": 0.5784907341003418,
-      "eval_mean_accuracy": 0.7836010760977574,
-      "eval_mean_iou": 0.39472291713535723,
-      "eval_overall_accuracy": 0.9174387348231985,
-      "eval_runtime": 11.2723,
-      "eval_samples_per_second": 5.944,
-      "eval_steps_per_second": 0.444,
-      "step": 210
-    },
-    {
-      "epoch": 19.18,
-      "learning_rate": 1.1405750798722045e-06,
-      "loss": 0.6403,
-      "step": 211
-    },
-    {
-      "epoch": 19.27,
-      "learning_rate": 1.1309904153354633e-06,
-      "loss": 0.6271,
-      "step": 212
-    },
-    {
-      "epoch": 19.36,
-      "learning_rate": 1.121405750798722e-06,
-      "loss": 0.679,
-      "step": 213
-    },
-    {
-      "epoch": 19.45,
-      "learning_rate": 1.1118210862619809e-06,
-      "loss": 0.6239,
-      "step": 214
-    },
-    {
-      "epoch": 19.55,
-      "learning_rate": 1.1022364217252397e-06,
-      "loss": 0.6527,
-      "step": 215
-    },
-    {
-      "epoch": 19.64,
-      "learning_rate": 1.0926517571884984e-06,
-      "loss": 0.6228,
-      "step": 216
-    },
-    {
-      "epoch": 19.73,
-      "learning_rate": 1.0830670926517572e-06,
-      "loss": 0.6227,
-      "step": 217
-    },
-    {
-      "epoch": 19.82,
-      "learning_rate": 1.073482428115016e-06,
-      "loss": 0.6347,
-      "step": 218
-    },
-    {
-      "epoch": 19.91,
-      "learning_rate": 1.0638977635782748e-06,
-      "loss": 0.6374,
-      "step": 219
-    },
-    {
-      "epoch": 20.0,
-      "learning_rate": 1.0543130990415336e-06,
-      "loss": 0.5845,
-      "step": 220
-    },
-    {
-      "epoch": 20.0,
-      "eval_accuracy_safe": 0.6426151017151052,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9286315640516787,
-      "eval_iou_safe": 0.26982054857603477,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9187221572756564,
-      "eval_loss": 0.5641139149665833,
-      "eval_mean_accuracy": 0.7856233328833919,
-      "eval_mean_iou": 0.39618090195056377,
-      "eval_overall_accuracy": 0.9201830963590252,
-      "eval_runtime": 9.4874,
-      "eval_samples_per_second": 7.062,
-      "eval_steps_per_second": 0.527,
-      "step": 220
-    },
-    {
-      "epoch": 20.09,
-      "learning_rate": 1.0447284345047924e-06,
-      "loss": 0.606,
-      "step": 221
-    },
-    {
-      "epoch": 20.18,
-      "learning_rate": 1.0351437699680512e-06,
-      "loss": 0.6491,
-      "step": 222
-    },
-    {
-      "epoch": 20.27,
-      "learning_rate": 1.02555910543131e-06,
-      "loss": 0.6396,
-      "step": 223
-    },
-    {
-      "epoch": 20.36,
-      "learning_rate": 1.0159744408945688e-06,
-      "loss": 0.617,
-      "step": 224
-    },
-    {
-      "epoch": 20.45,
-      "learning_rate": 1.0063897763578276e-06,
-      "loss": 0.614,
-      "step": 225
-    },
-    {
-      "epoch": 20.55,
-      "learning_rate": 9.968051118210861e-07,
-      "loss": 0.6281,
-      "step": 226
-    },
-    {
-      "epoch": 20.64,
-      "learning_rate": 9.872204472843452e-07,
-      "loss": 0.6374,
-      "step": 227
-    },
-    {
-      "epoch": 20.73,
-      "learning_rate": 9.77635782747604e-07,
-      "loss": 0.6394,
-      "step": 228
-    },
-    {
-      "epoch": 20.82,
-      "learning_rate": 9.680511182108625e-07,
-      "loss": 0.6351,
-      "step": 229
-    },
-    {
-      "epoch": 20.91,
-      "learning_rate": 9.584664536741213e-07,
-      "loss": 0.6062,
-      "step": 230
-    },
-    {
-      "epoch": 20.91,
-      "eval_accuracy_safe": 0.6519963300064379,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9252195649054266,
-      "eval_iou_safe": 0.2640576051310034,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9156001170929594,
-      "eval_loss": 0.5693235397338867,
-      "eval_mean_accuracy": 0.7886079474559322,
-      "eval_mean_iou": 0.3932192407413209,
-      "eval_overall_accuracy": 0.9171489886383513,
-      "eval_runtime": 11.0762,
-      "eval_samples_per_second": 6.049,
-      "eval_steps_per_second": 0.451,
-      "step": 230
-    },
-    {
-      "epoch": 21.0,
-      "learning_rate": 9.488817891373802e-07,
-      "loss": 0.6456,
-      "step": 231
-    },
-    {
-      "epoch": 21.09,
-      "learning_rate": 9.39297124600639e-07,
-      "loss": 0.6055,
-      "step": 232
-    },
-    {
-      "epoch": 21.18,
-      "learning_rate": 9.297124600638977e-07,
-      "loss": 0.6134,
-      "step": 233
-    },
-    {
-      "epoch": 21.27,
-      "learning_rate": 9.201277955271566e-07,
-      "loss": 0.5718,
-      "step": 234
-    },
-    {
-      "epoch": 21.36,
-      "learning_rate": 9.105431309904154e-07,
-      "loss": 0.5939,
-      "step": 235
-    },
-    {
-      "epoch": 21.45,
-      "learning_rate": 9.009584664536742e-07,
-      "loss": 0.5917,
-      "step": 236
-    },
-    {
-      "epoch": 21.55,
-      "learning_rate": 8.913738019169329e-07,
-      "loss": 0.599,
-      "step": 237
-    },
-    {
-      "epoch": 21.64,
-      "learning_rate": 8.817891373801916e-07,
-      "loss": 0.6239,
-      "step": 238
-    },
-    {
-      "epoch": 21.73,
-      "learning_rate": 8.722044728434505e-07,
-      "loss": 0.5946,
-      "step": 239
-    },
-    {
-      "epoch": 21.82,
-      "learning_rate": 8.626198083067093e-07,
-      "loss": 0.6071,
-      "step": 240
-    },
-    {
-      "epoch": 21.82,
-      "eval_accuracy_safe": 0.6591840432380754,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9282916959179331,
-      "eval_iou_safe": 0.267531459702011,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9188227497665283,
-      "eval_loss": 0.5627052187919617,
-      "eval_mean_accuracy": 0.7937378695780043,
-      "eval_mean_iou": 0.39545140315617977,
-      "eval_overall_accuracy": 0.9203426873506005,
-      "eval_runtime": 9.7383,
-      "eval_samples_per_second": 6.88,
-      "eval_steps_per_second": 0.513,
-      "step": 240
-    },
-    {
-      "epoch": 21.91,
-      "learning_rate": 8.53035143769968e-07,
-      "loss": 0.5784,
-      "step": 241
-    },
-    {
-      "epoch": 22.0,
-      "learning_rate": 8.434504792332268e-07,
-      "loss": 0.6049,
-      "step": 242
-    },
-    {
-      "epoch": 22.09,
-      "learning_rate": 8.338658146964857e-07,
-      "loss": 0.5871,
-      "step": 243
-    },
-    {
-      "epoch": 22.18,
-      "learning_rate": 8.242811501597445e-07,
-      "loss": 0.5894,
-      "step": 244
-    },
-    {
-      "epoch": 22.27,
-      "learning_rate": 8.146964856230032e-07,
-      "loss": 0.5801,
-      "step": 245
-    },
-    {
-      "epoch": 22.36,
-      "learning_rate": 8.05111821086262e-07,
-      "loss": 0.5719,
-      "step": 246
-    },
-    {
-      "epoch": 22.45,
-      "learning_rate": 7.955271565495208e-07,
-      "loss": 0.5979,
-      "step": 247
-    },
-    {
-      "epoch": 22.55,
-      "learning_rate": 7.859424920127797e-07,
-      "loss": 0.5855,
-      "step": 248
-    },
-    {
-      "epoch": 22.64,
-      "learning_rate": 7.763578274760382e-07,
-      "loss": 0.5975,
-      "step": 249
-    },
-    {
-      "epoch": 22.73,
-      "learning_rate": 7.667731629392971e-07,
-      "loss": 0.6209,
-      "step": 250
-    },
-    {
-      "epoch": 22.73,
-      "eval_accuracy_safe": 0.6618690752926936,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9299528432231069,
-      "eval_iou_safe": 0.2724302598559697,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9205411497943506,
-      "eval_loss": 0.5631816983222961,
-      "eval_mean_accuracy": 0.7959109592579002,
-      "eval_mean_iou": 0.39765713655010676,
-      "eval_overall_accuracy": 0.9220340785695545,
-      "eval_runtime": 10.1606,
-      "eval_samples_per_second": 6.594,
-      "eval_steps_per_second": 0.492,
-      "step": 250
-    },
-    {
-      "epoch": 22.82,
-      "learning_rate": 7.571884984025559e-07,
-      "loss": 0.5852,
-      "step": 251
-    },
-    {
-      "epoch": 22.91,
-      "learning_rate": 7.476038338658147e-07,
-      "loss": 0.5775,
-      "step": 252
-    },
-    {
-      "epoch": 23.0,
-      "learning_rate": 7.380191693290735e-07,
-      "loss": 0.6365,
-      "step": 253
-    },
-    {
-      "epoch": 23.09,
-      "learning_rate": 7.284345047923323e-07,
-      "loss": 0.6243,
-      "step": 254
-    },
-    {
-      "epoch": 23.18,
-      "learning_rate": 7.188498402555911e-07,
-      "loss": 0.5995,
-      "step": 255
-    },
-    {
-      "epoch": 23.27,
-      "learning_rate": 7.092651757188499e-07,
-      "loss": 0.6019,
-      "step": 256
-    },
-    {
-      "epoch": 23.36,
-      "learning_rate": 6.996805111821087e-07,
-      "loss": 0.5883,
-      "step": 257
-    },
-    {
-      "epoch": 23.45,
-      "learning_rate": 6.900958466453674e-07,
-      "loss": 0.589,
-      "step": 258
-    },
-    {
-      "epoch": 23.55,
-      "learning_rate": 6.805111821086263e-07,
-      "loss": 0.5717,
-      "step": 259
-    },
-    {
-      "epoch": 23.64,
-      "learning_rate": 6.709265175718849e-07,
-      "loss": 0.5609,
-      "step": 260
-    },
-    {
-      "epoch": 23.64,
-      "eval_accuracy_safe": 0.6504639534928547,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9378781128324656,
-      "eval_iou_safe": 0.2868443063281154,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9280609954701114,
-      "eval_loss": 0.5416346192359924,
-      "eval_mean_accuracy": 0.7941710331626601,
-      "eval_mean_iou": 0.40496843393274223,
-      "eval_overall_accuracy": 0.9293883594114389,
-      "eval_runtime": 9.5225,
-      "eval_samples_per_second": 7.036,
-      "eval_steps_per_second": 0.525,
-      "step": 260
-    },
-    {
-      "epoch": 23.73,
-      "learning_rate": 6.613418530351438e-07,
-      "loss": 0.5557,
-      "step": 261
-    },
-    {
-      "epoch": 23.82,
-      "learning_rate": 6.517571884984025e-07,
-      "loss": 0.5968,
-      "step": 262
-    },
-    {
-      "epoch": 23.91,
-      "learning_rate": 6.421725239616614e-07,
-      "loss": 0.5673,
-      "step": 263
-    },
-    {
-      "epoch": 24.0,
-      "learning_rate": 6.325878594249201e-07,
-      "loss": 0.6091,
-      "step": 264
-    },
-    {
-      "epoch": 24.09,
-      "learning_rate": 6.230031948881789e-07,
-      "loss": 0.5776,
-      "step": 265
-    },
-    {
-      "epoch": 24.18,
-      "learning_rate": 6.134185303514377e-07,
-      "loss": 0.5562,
-      "step": 266
-    },
-    {
-      "epoch": 24.27,
-      "learning_rate": 6.038338658146965e-07,
-      "loss": 0.5863,
-      "step": 267
-    },
-    {
-      "epoch": 24.36,
-      "learning_rate": 5.942492012779553e-07,
-      "loss": 0.5565,
-      "step": 268
-    },
-    {
-      "epoch": 24.45,
-      "learning_rate": 5.846645367412141e-07,
-      "loss": 0.5858,
-      "step": 269
-    },
-    {
-      "epoch": 24.55,
-      "learning_rate": 5.750798722044729e-07,
-      "loss": 0.5752,
-      "step": 270
-    },
-    {
-      "epoch": 24.55,
-      "eval_accuracy_safe": 0.6412195789530495,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.945148932410419,
-      "eval_iou_safe": 0.2982960328042603,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9349957034087992,
-      "eval_loss": 0.5140887498855591,
-      "eval_mean_accuracy": 0.7931842556817342,
-      "eval_mean_iou": 0.41109724540435316,
-      "eval_overall_accuracy": 0.9361713466359608,
-      "eval_runtime": 10.0336,
-      "eval_samples_per_second": 6.678,
-      "eval_steps_per_second": 0.498,
-      "step": 270
-    },
-    {
-      "epoch": 24.64,
-      "learning_rate": 5.654952076677316e-07,
-      "loss": 0.5881,
-      "step": 271
-    },
-    {
-      "epoch": 24.73,
-      "learning_rate": 5.559105431309904e-07,
-      "loss": 0.6124,
-      "step": 272
-    },
-    {
-      "epoch": 24.82,
-      "learning_rate": 5.463258785942492e-07,
-      "loss": 0.5648,
-      "step": 273
-    },
-    {
-      "epoch": 24.91,
-      "learning_rate": 5.36741214057508e-07,
-      "loss": 0.578,
-      "step": 274
-    },
-    {
-      "epoch": 25.0,
-      "learning_rate": 5.271565495207668e-07,
-      "loss": 0.5605,
-      "step": 275
-    },
-    {
-      "epoch": 25.09,
-      "learning_rate": 5.175718849840256e-07,
-      "loss": 0.588,
-      "step": 276
-    },
-    {
-      "epoch": 25.18,
-      "learning_rate": 5.079872204472844e-07,
-      "loss": 0.5814,
-      "step": 277
-    },
-    {
-      "epoch": 25.27,
-      "learning_rate": 4.984025559105431e-07,
-      "loss": 0.5671,
-      "step": 278
-    },
-    {
-      "epoch": 25.36,
-      "learning_rate": 4.88817891373802e-07,
-      "loss": 0.5237,
-      "step": 279
-    },
-    {
-      "epoch": 25.45,
-      "learning_rate": 4.792332268370607e-07,
-      "loss": 0.6004,
-      "step": 280
-    },
-    {
-      "epoch": 25.45,
-      "eval_accuracy_safe": 0.6491917918589365,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9412261630289883,
-      "eval_iou_safe": 0.2907094372975922,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9313366391036164,
-      "eval_loss": 0.5255351662635803,
-      "eval_mean_accuracy": 0.7952089774439624,
-      "eval_mean_iou": 0.4073486921337362,
-      "eval_overall_accuracy": 0.9325999359586346,
-      "eval_runtime": 9.8685,
-      "eval_samples_per_second": 6.789,
-      "eval_steps_per_second": 0.507,
-      "step": 280
-    },
-    {
-      "epoch": 25.55,
-      "learning_rate": 4.696485623003195e-07,
-      "loss": 0.5482,
-      "step": 281
-    },
-    {
-      "epoch": 25.64,
-      "learning_rate": 4.600638977635783e-07,
-      "loss": 0.5764,
-      "step": 282
-    },
-    {
-      "epoch": 25.73,
-      "learning_rate": 4.504792332268371e-07,
-      "loss": 0.5608,
-      "step": 283
-    },
-    {
-      "epoch": 25.82,
-      "learning_rate": 4.408945686900958e-07,
-      "loss": 0.5698,
-      "step": 284
-    },
-    {
-      "epoch": 25.91,
-      "learning_rate": 4.3130990415335466e-07,
-      "loss": 0.5502,
-      "step": 285
-    },
-    {
-      "epoch": 26.0,
-      "learning_rate": 4.217252396166134e-07,
-      "loss": 0.5975,
-      "step": 286
-    },
-    {
-      "epoch": 26.09,
-      "learning_rate": 4.1214057507987225e-07,
-      "loss": 0.5736,
-      "step": 287
-    },
-    {
-      "epoch": 26.18,
-      "learning_rate": 4.02555910543131e-07,
-      "loss": 0.5498,
-      "step": 288
-    },
-    {
-      "epoch": 26.27,
-      "learning_rate": 3.9297124600638983e-07,
-      "loss": 0.5682,
-      "step": 289
-    },
-    {
-      "epoch": 26.36,
-      "learning_rate": 3.8338658146964857e-07,
-      "loss": 0.5524,
-      "step": 290
-    },
-    {
-      "epoch": 26.36,
-      "eval_accuracy_safe": 0.6587715544658655,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9386905578378355,
-      "eval_iou_safe": 0.28676309565062713,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9290998927343179,
-      "eval_loss": 0.5313962697982788,
-      "eval_mean_accuracy": 0.7987310561518506,
-      "eval_mean_iou": 0.4052876627949817,
-      "eval_overall_accuracy": 0.9304221993062034,
-      "eval_runtime": 9.5415,
-      "eval_samples_per_second": 7.022,
-      "eval_steps_per_second": 0.524,
-      "step": 290
-    },
-    {
-      "epoch": 26.45,
-      "learning_rate": 3.7380191693290736e-07,
-      "loss": 0.5519,
-      "step": 291
-    },
-    {
-      "epoch": 26.55,
-      "learning_rate": 3.6421725239616615e-07,
-      "loss": 0.5424,
-      "step": 292
-    },
-    {
-      "epoch": 26.64,
-      "learning_rate": 3.5463258785942494e-07,
-      "loss": 0.583,
-      "step": 293
-    },
-    {
-      "epoch": 26.73,
-      "learning_rate": 3.450479233226837e-07,
-      "loss": 0.5798,
-      "step": 294
-    },
-    {
-      "epoch": 26.82,
-      "learning_rate": 3.3546325878594247e-07,
-      "loss": 0.5284,
-      "step": 295
-    },
-    {
-      "epoch": 26.91,
-      "learning_rate": 3.2587859424920126e-07,
-      "loss": 0.5681,
-      "step": 296
-    },
-    {
-      "epoch": 27.0,
-      "learning_rate": 3.1629392971246005e-07,
-      "loss": 0.6352,
-      "step": 297
-    },
-    {
-      "epoch": 27.09,
-      "learning_rate": 3.0670926517571884e-07,
-      "loss": 0.5651,
-      "step": 298
-    },
-    {
-      "epoch": 27.18,
-      "learning_rate": 2.9712460063897763e-07,
-      "loss": 0.5707,
-      "step": 299
-    },
-    {
-      "epoch": 27.27,
-      "learning_rate": 2.875399361022364e-07,
-      "loss": 0.5758,
-      "step": 300
-    },
-    {
-      "epoch": 27.27,
-      "eval_accuracy_safe": 0.6544115095932552,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9423355893036522,
-      "eval_iou_safe": 0.291325116977104,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9325746804551616,
-      "eval_loss": 0.526845395565033,
-      "eval_mean_accuracy": 0.7983735494484537,
-      "eval_mean_iou": 0.40796659914408856,
-      "eval_overall_accuracy": 0.9338307736524895,
-      "eval_runtime": 10.1793,
-      "eval_samples_per_second": 6.582,
-      "eval_steps_per_second": 0.491,
-      "step": 300
-    },
-    {
-      "epoch": 27.36,
-      "learning_rate": 2.779552715654952e-07,
-      "loss": 0.5541,
-      "step": 301
-    },
-    {
-      "epoch": 27.45,
-      "learning_rate": 2.68370607028754e-07,
-      "loss": 0.5143,
-      "step": 302
-    },
-    {
-      "epoch": 27.55,
-      "learning_rate": 2.587859424920128e-07,
-      "loss": 0.5392,
-      "step": 303
-    },
-    {
-      "epoch": 27.64,
-      "learning_rate": 2.4920127795527154e-07,
-      "loss": 0.5803,
-      "step": 304
-    },
-    {
-      "epoch": 27.73,
-      "learning_rate": 2.3961661341853033e-07,
-      "loss": 0.5733,
-      "step": 305
-    },
-    {
-      "epoch": 27.82,
-      "learning_rate": 2.3003194888178915e-07,
-      "loss": 0.5799,
-      "step": 306
-    },
-    {
-      "epoch": 27.91,
-      "learning_rate": 2.204472843450479e-07,
-      "loss": 0.5737,
-      "step": 307
-    },
-    {
-      "epoch": 28.0,
-      "learning_rate": 2.108626198083067e-07,
-      "loss": 0.507,
-      "step": 308
-    },
-    {
-      "epoch": 28.09,
-      "learning_rate": 2.012779552715655e-07,
-      "loss": 0.5913,
-      "step": 309
-    },
-    {
-      "epoch": 28.18,
-      "learning_rate": 1.9169329073482428e-07,
-      "loss": 0.5598,
-      "step": 310
-    },
-    {
-      "epoch": 28.18,
-      "eval_accuracy_safe": 0.6604831901187737,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9407892567641855,
-      "eval_iou_safe": 0.2896690945241332,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9312167321135931,
-      "eval_loss": 0.5239936709403992,
-      "eval_mean_accuracy": 0.8006362234414796,
-      "eval_mean_iou": 0.4069619422125754,
-      "eval_overall_accuracy": 0.9325094650040812,
-      "eval_runtime": 9.6243,
-      "eval_samples_per_second": 6.962,
-      "eval_steps_per_second": 0.52,
-      "step": 310
-    },
-    {
-      "epoch": 28.27,
-      "learning_rate": 1.8210862619808307e-07,
-      "loss": 0.5394,
-      "step": 311
-    },
-    {
-      "epoch": 28.36,
-      "learning_rate": 1.7252396166134184e-07,
-      "loss": 0.5312,
-      "step": 312
-    },
-    {
-      "epoch": 28.45,
-      "learning_rate": 1.6293929712460063e-07,
-      "loss": 0.5313,
-      "step": 313
-    },
-    {
-      "epoch": 28.55,
-      "learning_rate": 1.5335463258785942e-07,
-      "loss": 0.5188,
-      "step": 314
-    },
-    {
-      "epoch": 28.64,
-      "learning_rate": 1.437699680511182e-07,
-      "loss": 0.5497,
-      "step": 315
-    },
-    {
-      "epoch": 28.73,
-      "learning_rate": 1.34185303514377e-07,
-      "loss": 0.5908,
-      "step": 316
-    },
-    {
-      "epoch": 28.82,
-      "learning_rate": 1.2460063897763577e-07,
-      "loss": 0.56,
-      "step": 317
-    },
-    {
-      "epoch": 28.91,
-      "learning_rate": 1.1501597444089457e-07,
-      "loss": 0.5792,
-      "step": 318
-    },
-    {
-      "epoch": 29.0,
-      "learning_rate": 1.0543130990415335e-07,
-      "loss": 0.6318,
-      "step": 319
-    },
-    {
-      "epoch": 29.09,
-      "learning_rate": 9.584664536741214e-08,
-      "loss": 0.5505,
-      "step": 320
-    },
-    {
-      "epoch": 29.09,
-      "eval_accuracy_safe": 0.6582414871183997,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9420817295738547,
-      "eval_iou_safe": 0.29588640634788055,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9324414181256692,
-      "eval_loss": 0.5165032744407654,
-      "eval_mean_accuracy": 0.8001616083461272,
-      "eval_mean_iou": 0.4094426081578499,
-      "eval_overall_accuracy": 0.933697543927093,
-      "eval_runtime": 11.4418,
-      "eval_samples_per_second": 5.856,
-      "eval_steps_per_second": 0.437,
-      "step": 320
-    },
-    {
-      "epoch": 29.18,
-      "learning_rate": 1.3160919540229885e-06,
-      "loss": 0.5725,
-      "step": 321
-    },
-    {
-      "epoch": 29.27,
-      "learning_rate": 1.310344827586207e-06,
-      "loss": 0.5195,
-      "step": 322
-    },
-    {
-      "epoch": 29.36,
-      "learning_rate": 1.3045977011494253e-06,
-      "loss": 0.5441,
-      "step": 323
-    },
-    {
-      "epoch": 29.45,
-      "learning_rate": 1.2988505747126436e-06,
-      "loss": 0.5896,
-      "step": 324
-    },
-    {
-      "epoch": 29.55,
-      "learning_rate": 1.293103448275862e-06,
-      "loss": 0.5837,
-      "step": 325
-    },
-    {
-      "epoch": 29.64,
-      "learning_rate": 1.2873563218390806e-06,
-      "loss": 0.5317,
-      "step": 326
-    },
-    {
-      "epoch": 29.73,
-      "learning_rate": 1.2816091954022989e-06,
-      "loss": 0.5395,
-      "step": 327
-    },
-    {
-      "epoch": 29.82,
-      "learning_rate": 1.2758620689655172e-06,
-      "loss": 0.5642,
-      "step": 328
-    },
-    {
-      "epoch": 29.91,
-      "learning_rate": 1.2701149425287357e-06,
-      "loss": 0.5314,
-      "step": 329
-    },
-    {
-      "epoch": 30.0,
-      "learning_rate": 1.2643678160919542e-06,
-      "loss": 0.5754,
-      "step": 330
-    },
-    {
-      "epoch": 30.0,
-      "eval_accuracy_safe": 0.657778882887884,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9432738201330771,
-      "eval_iou_safe": 0.2958553359937371,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9336061303908306,
-      "eval_loss": 0.5145326852798462,
-      "eval_mean_accuracy": 0.8005263515104806,
-      "eval_mean_iou": 0.40982048879485594,
-      "eval_overall_accuracy": 0.9348407574553987,
-      "eval_runtime": 8.5617,
-      "eval_samples_per_second": 7.826,
-      "eval_steps_per_second": 0.584,
-      "step": 330
-    },
-    {
-      "epoch": 30.09,
-      "learning_rate": 1.2586206896551725e-06,
-      "loss": 0.5332,
-      "step": 331
-    },
-    {
-      "epoch": 30.18,
-      "learning_rate": 1.252873563218391e-06,
-      "loss": 0.5386,
-      "step": 332
-    },
-    {
-      "epoch": 30.27,
-      "learning_rate": 1.2471264367816092e-06,
-      "loss": 0.5446,
-      "step": 333
-    },
-    {
-      "epoch": 30.36,
-      "learning_rate": 1.2413793103448275e-06,
-      "loss": 0.524,
-      "step": 334
-    },
-    {
-      "epoch": 30.45,
-      "learning_rate": 1.235632183908046e-06,
-      "loss": 0.5337,
-      "step": 335
-    },
-    {
-      "epoch": 30.55,
-      "learning_rate": 1.2298850574712645e-06,
-      "loss": 0.5228,
-      "step": 336
-    },
-    {
-      "epoch": 30.64,
-      "learning_rate": 1.2241379310344828e-06,
-      "loss": 0.5856,
-      "step": 337
-    },
-    {
-      "epoch": 30.73,
-      "learning_rate": 1.2183908045977011e-06,
-      "loss": 0.5221,
-      "step": 338
-    },
-    {
-      "epoch": 30.82,
-      "learning_rate": 1.2126436781609196e-06,
-      "loss": 0.5505,
-      "step": 339
-    },
-    {
-      "epoch": 30.91,
-      "learning_rate": 1.206896551724138e-06,
-      "loss": 0.5284,
-      "step": 340
-    },
-    {
-      "epoch": 30.91,
-      "eval_accuracy_safe": 0.6718902394362396,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9410974437668724,
-      "eval_iou_safe": 0.29405467602594526,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9318466838959976,
-      "eval_loss": 0.5174660086631775,
-      "eval_mean_accuracy": 0.806493841601556,
-      "eval_mean_iou": 0.4086337866406476,
-      "eval_overall_accuracy": 0.933145494603399,
-      "eval_runtime": 8.5465,
-      "eval_samples_per_second": 7.839,
-      "eval_steps_per_second": 0.585,
-      "step": 340
-    },
-    {
-      "epoch": 31.0,
-      "learning_rate": 1.2011494252873564e-06,
-      "loss": 0.5315,
-      "step": 341
-    },
-    {
-      "epoch": 31.09,
-      "learning_rate": 1.1954022988505747e-06,
-      "loss": 0.511,
-      "step": 342
-    },
-    {
-      "epoch": 31.18,
-      "learning_rate": 1.1896551724137932e-06,
-      "loss": 0.5607,
-      "step": 343
-    },
-    {
-      "epoch": 31.27,
-      "learning_rate": 1.1839080459770115e-06,
-      "loss": 0.5254,
-      "step": 344
-    },
-    {
-      "epoch": 31.36,
-      "learning_rate": 1.1781609195402298e-06,
-      "loss": 0.4814,
-      "step": 345
-    },
-    {
-      "epoch": 31.45,
-      "learning_rate": 1.1724137931034483e-06,
-      "loss": 0.5376,
-      "step": 346
-    },
-    {
-      "epoch": 31.55,
-      "learning_rate": 1.1666666666666668e-06,
-      "loss": 0.5253,
-      "step": 347
-    },
-    {
-      "epoch": 31.64,
-      "learning_rate": 1.160919540229885e-06,
-      "loss": 0.5591,
-      "step": 348
-    },
-    {
-      "epoch": 31.73,
-      "learning_rate": 1.1551724137931034e-06,
-      "loss": 0.5161,
-      "step": 349
-    },
-    {
-      "epoch": 31.82,
-      "learning_rate": 1.1494252873563219e-06,
-      "loss": 0.5463,
-      "step": 350
-    },
-    {
-      "epoch": 31.82,
-      "eval_accuracy_safe": 0.6683590271433032,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9448280729553086,
-      "eval_iou_safe": 0.3019912053572595,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9354427723232134,
-      "eval_loss": 0.5016354918479919,
-      "eval_mean_accuracy": 0.8065935500493059,
-      "eval_mean_iou": 0.4124779925601576,
-      "eval_overall_accuracy": 0.9366616206382637,
-      "eval_runtime": 8.7504,
-      "eval_samples_per_second": 7.657,
-      "eval_steps_per_second": 0.571,
-      "step": 350
-    },
-    {
-      "epoch": 31.91,
-      "learning_rate": 1.1436781609195402e-06,
-      "loss": 0.4994,
-      "step": 351
-    },
-    {
-      "epoch": 32.0,
-      "learning_rate": 1.1379310344827587e-06,
-      "loss": 0.5251,
-      "step": 352
-    },
-    {
-      "epoch": 32.09,
-      "learning_rate": 1.1321839080459772e-06,
-      "loss": 0.5379,
-      "step": 353
-    },
-    {
-      "epoch": 32.18,
-      "learning_rate": 1.1264367816091955e-06,
-      "loss": 0.4885,
-      "step": 354
-    },
-    {
-      "epoch": 32.27,
-      "learning_rate": 1.1206896551724137e-06,
-      "loss": 0.5029,
-      "step": 355
-    },
-    {
-      "epoch": 32.36,
-      "learning_rate": 1.1149425287356322e-06,
-      "loss": 0.5273,
-      "step": 356
-    },
-    {
-      "epoch": 32.45,
-      "learning_rate": 1.1091954022988505e-06,
-      "loss": 0.5111,
-      "step": 357
-    },
-    {
-      "epoch": 32.55,
-      "learning_rate": 1.103448275862069e-06,
-      "loss": 0.5102,
-      "step": 358
-    },
-    {
-      "epoch": 32.64,
-      "learning_rate": 1.0977011494252873e-06,
-      "loss": 0.5197,
-      "step": 359
-    },
-    {
-      "epoch": 32.73,
-      "learning_rate": 1.0919540229885058e-06,
-      "loss": 0.4923,
-      "step": 360
-    },
-    {
-      "epoch": 32.73,
-      "eval_accuracy_safe": 0.6687907910917845,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9462536065154241,
-      "eval_iou_safe": 0.3065972713134455,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.936872865769379,
-      "eval_loss": 0.4946950674057007,
-      "eval_mean_accuracy": 0.8075221988036043,
-      "eval_mean_iou": 0.41449004569427483,
-      "eval_overall_accuracy": 0.9380577998374825,
-      "eval_runtime": 9.7999,
-      "eval_samples_per_second": 6.837,
-      "eval_steps_per_second": 0.51,
-      "step": 360
-    },
-    {
-      "epoch": 32.82,
-      "learning_rate": 1.0862068965517241e-06,
-      "loss": 0.4919,
-      "step": 361
-    },
-    {
-      "epoch": 32.91,
-      "learning_rate": 1.0804597701149424e-06,
-      "loss": 0.5359,
-      "step": 362
-    },
-    {
-      "epoch": 33.0,
-      "learning_rate": 1.0747126436781611e-06,
-      "loss": 0.5646,
-      "step": 363
-    },
-    {
-      "epoch": 33.09,
-      "learning_rate": 1.0689655172413794e-06,
-      "loss": 0.5186,
-      "step": 364
-    },
-    {
-      "epoch": 33.18,
-      "learning_rate": 1.0632183908045977e-06,
-      "loss": 0.4801,
-      "step": 365
-    },
-    {
-      "epoch": 33.27,
-      "learning_rate": 1.057471264367816e-06,
-      "loss": 0.4874,
-      "step": 366
-    },
-    {
-      "epoch": 33.36,
-      "learning_rate": 1.0517241379310345e-06,
-      "loss": 0.5092,
-      "step": 367
-    },
-    {
-      "epoch": 33.45,
-      "learning_rate": 1.045977011494253e-06,
-      "loss": 0.503,
-      "step": 368
-    },
-    {
-      "epoch": 33.55,
-      "learning_rate": 1.0402298850574713e-06,
-      "loss": 0.5327,
-      "step": 369
-    },
-    {
-      "epoch": 33.64,
-      "learning_rate": 1.0344827586206898e-06,
-      "loss": 0.4922,
-      "step": 370
-    },
-    {
-      "epoch": 33.64,
-      "eval_accuracy_safe": 0.6685016634477122,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9503706868340143,
-      "eval_iou_safe": 0.3165045602136191,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9409432282460215,
-      "eval_loss": 0.4738222360610962,
-      "eval_mean_accuracy": 0.8094361751408632,
-      "eval_mean_iou": 0.41914926281988024,
-      "eval_overall_accuracy": 0.9420447278378615,
-      "eval_runtime": 10.1885,
-      "eval_samples_per_second": 6.576,
-      "eval_steps_per_second": 0.491,
-      "step": 370
-    },
-    {
-      "epoch": 33.73,
-      "learning_rate": 1.028735632183908e-06,
-      "loss": 0.5358,
-      "step": 371
-    },
-    {
-      "epoch": 33.82,
-      "learning_rate": 1.0229885057471264e-06,
-      "loss": 0.4633,
-      "step": 372
-    },
-    {
-      "epoch": 33.91,
-      "learning_rate": 1.0172413793103449e-06,
-      "loss": 0.4886,
-      "step": 373
-    },
-    {
-      "epoch": 34.0,
-      "learning_rate": 1.0114942528735634e-06,
-      "loss": 0.531,
-      "step": 374
-    },
-    {
-      "epoch": 34.09,
-      "learning_rate": 1.0057471264367817e-06,
-      "loss": 0.5054,
-      "step": 375
-    },
-    {
-      "epoch": 34.18,
-      "learning_rate": 1e-06,
-      "loss": 0.498,
-      "step": 376
-    },
-    {
-      "epoch": 34.27,
-      "learning_rate": 9.942528735632185e-07,
-      "loss": 0.5071,
-      "step": 377
-    },
-    {
-      "epoch": 34.36,
-      "learning_rate": 9.885057471264367e-07,
-      "loss": 0.4907,
-      "step": 378
-    },
-    {
-      "epoch": 34.45,
-      "learning_rate": 9.827586206896552e-07,
-      "loss": 0.4967,
-      "step": 379
-    },
-    {
-      "epoch": 34.55,
-      "learning_rate": 9.770114942528735e-07,
-      "loss": 0.4976,
-      "step": 380
-    },
-    {
-      "epoch": 34.55,
-      "eval_accuracy_safe": 0.6747757333240813,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9535474829165368,
-      "eval_iou_safe": 0.3232939737984814,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.944264577460996,
-      "eval_loss": 0.46629098057746887,
-      "eval_mean_accuracy": 0.814161608120309,
-      "eval_mean_iou": 0.42251951708649244,
-      "eval_overall_accuracy": 0.9453130124220207,
-      "eval_runtime": 8.7895,
-      "eval_samples_per_second": 7.623,
-      "eval_steps_per_second": 0.569,
-      "step": 380
-    },
-    {
-      "epoch": 34.64,
-      "learning_rate": 9.71264367816092e-07,
-      "loss": 0.4647,
-      "step": 381
-    },
-    {
-      "epoch": 34.73,
-      "learning_rate": 9.655172413793103e-07,
-      "loss": 0.4782,
-      "step": 382
-    },
-    {
-      "epoch": 34.82,
-      "learning_rate": 9.597701149425286e-07,
-      "loss": 0.4785,
-      "step": 383
-    },
-    {
-      "epoch": 34.91,
-      "learning_rate": 9.540229885057471e-07,
-      "loss": 0.4996,
-      "step": 384
-    },
-    {
-      "epoch": 35.0,
-      "learning_rate": 9.482758620689655e-07,
-      "loss": 0.437,
-      "step": 385
-    },
-    {
-      "epoch": 35.09,
-      "learning_rate": 9.425287356321839e-07,
-      "loss": 0.4617,
-      "step": 386
-    },
-    {
-      "epoch": 35.18,
-      "learning_rate": 9.367816091954024e-07,
-      "loss": 0.4486,
-      "step": 387
-    },
-    {
-      "epoch": 35.27,
-      "learning_rate": 9.310344827586207e-07,
-      "loss": 0.5012,
-      "step": 388
-    },
-    {
-      "epoch": 35.36,
-      "learning_rate": 9.252873563218391e-07,
-      "loss": 0.4752,
-      "step": 389
-    },
-    {
-      "epoch": 35.45,
-      "learning_rate": 9.195402298850574e-07,
-      "loss": 0.4922,
-      "step": 390
-    },
-    {
-      "epoch": 35.45,
-      "eval_accuracy_safe": 0.650909210064726,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9652786537349766,
-      "eval_iou_safe": 0.3483788236580638,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9551696231808672,
-      "eval_loss": 0.4295278489589691,
-      "eval_mean_accuracy": 0.8080939318998512,
-      "eval_mean_iou": 0.4345161489463103,
-      "eval_overall_accuracy": 0.9559926844354886,
-      "eval_runtime": 9.5593,
-      "eval_samples_per_second": 7.009,
-      "eval_steps_per_second": 0.523,
-      "step": 390
-    },
-    {
-      "epoch": 35.55,
-      "learning_rate": 9.137931034482759e-07,
-      "loss": 0.4963,
-      "step": 391
-    },
-    {
-      "epoch": 35.64,
-      "learning_rate": 9.080459770114943e-07,
-      "loss": 0.4951,
-      "step": 392
-    },
-    {
-      "epoch": 35.73,
-      "learning_rate": 9.022988505747126e-07,
-      "loss": 0.4667,
-      "step": 393
-    },
-    {
-      "epoch": 35.82,
-      "learning_rate": 8.965517241379311e-07,
-      "loss": 0.4694,
-      "step": 394
-    },
-    {
-      "epoch": 35.91,
-      "learning_rate": 8.908045977011495e-07,
-      "loss": 0.471,
-      "step": 395
-    },
-    {
-      "epoch": 36.0,
-      "learning_rate": 8.850574712643678e-07,
-      "loss": 0.5268,
-      "step": 396
-    },
-    {
-      "epoch": 36.09,
-      "learning_rate": 8.793103448275862e-07,
-      "loss": 0.4383,
-      "step": 397
-    },
-    {
-      "epoch": 36.18,
-      "learning_rate": 8.735632183908047e-07,
-      "loss": 0.48,
-      "step": 398
-    },
-    {
-      "epoch": 36.27,
-      "learning_rate": 8.67816091954023e-07,
-      "loss": 0.4727,
-      "step": 399
-    },
-    {
-      "epoch": 36.36,
-      "learning_rate": 8.620689655172414e-07,
-      "loss": 0.4608,
-      "step": 400
-    },
-    {
-      "epoch": 36.36,
-      "eval_accuracy_safe": 0.6580178950736505,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9636894929998194,
-      "eval_iou_safe": 0.35071600798452407,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9538031459361976,
-      "eval_loss": 0.44339337944984436,
-      "eval_mean_accuracy": 0.810853694036735,
-      "eval_mean_iou": 0.4348397179735739,
-      "eval_overall_accuracy": 0.9546604441173041,
-      "eval_runtime": 9.0957,
-      "eval_samples_per_second": 7.366,
-      "eval_steps_per_second": 0.55,
-      "step": 400
-    },
-    {
-      "epoch": 36.45,
-      "learning_rate": 8.563218390804599e-07,
-      "loss": 0.5072,
-      "step": 401
-    },
-    {
-      "epoch": 36.55,
-      "learning_rate": 8.505747126436782e-07,
-      "loss": 0.4273,
-      "step": 402
-    },
-    {
-      "epoch": 36.64,
-      "learning_rate": 8.448275862068965e-07,
-      "loss": 0.4543,
-      "step": 403
-    },
-    {
-      "epoch": 36.73,
-      "learning_rate": 8.390804597701149e-07,
-      "loss": 0.4856,
-      "step": 404
-    },
-    {
-      "epoch": 36.82,
-      "learning_rate": 8.333333333333334e-07,
-      "loss": 0.4673,
-      "step": 405
-    },
-    {
-      "epoch": 36.91,
-      "learning_rate": 8.275862068965517e-07,
-      "loss": 0.4909,
-      "step": 406
-    },
-    {
-      "epoch": 37.0,
-      "learning_rate": 8.218390804597701e-07,
-      "loss": 0.5135,
-      "step": 407
-    },
-    {
-      "epoch": 37.09,
-      "learning_rate": 8.160919540229886e-07,
-      "loss": 0.4513,
-      "step": 408
-    },
-    {
-      "epoch": 37.18,
-      "learning_rate": 8.103448275862069e-07,
-      "loss": 0.4915,
-      "step": 409
-    },
-    {
-      "epoch": 37.27,
-      "learning_rate": 8.045977011494253e-07,
-      "loss": 0.4836,
-      "step": 410
-    },
-    {
-      "epoch": 37.27,
-      "eval_accuracy_safe": 0.6521813716986442,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9662073802250839,
-      "eval_iou_safe": 0.3588467371591596,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.956125924319263,
-      "eval_loss": 0.4327642321586609,
-      "eval_mean_accuracy": 0.809194375961864,
-      "eval_mean_iou": 0.43832422049280756,
-      "eval_overall_accuracy": 0.956931555449073,
-      "eval_runtime": 10.5295,
-      "eval_samples_per_second": 6.363,
-      "eval_steps_per_second": 0.475,
-      "step": 410
-    },
-    {
-      "epoch": 37.36,
-      "learning_rate": 7.988505747126438e-07,
-      "loss": 0.4822,
-      "step": 411
-    },
-    {
-      "epoch": 37.45,
-      "learning_rate": 7.931034482758621e-07,
-      "loss": 0.4407,
-      "step": 412
-    },
-    {
-      "epoch": 37.55,
-      "learning_rate": 7.873563218390805e-07,
-      "loss": 0.471,
-      "step": 413
-    },
-    {
-      "epoch": 37.64,
-      "learning_rate": 7.816091954022988e-07,
-      "loss": 0.4699,
-      "step": 414
-    },
-    {
-      "epoch": 37.73,
-      "learning_rate": 7.758620689655173e-07,
-      "loss": 0.4575,
-      "step": 415
-    },
-    {
-      "epoch": 37.82,
-      "learning_rate": 7.701149425287357e-07,
-      "loss": 0.4705,
-      "step": 416
-    },
-    {
-      "epoch": 37.91,
-      "learning_rate": 7.64367816091954e-07,
-      "loss": 0.435,
-      "step": 417
-    },
-    {
-      "epoch": 38.0,
-      "learning_rate": 7.586206896551725e-07,
-      "loss": 0.4484,
-      "step": 418
-    },
-    {
-      "epoch": 38.09,
-      "learning_rate": 7.528735632183909e-07,
-      "loss": 0.447,
-      "step": 419
-    },
-    {
-      "epoch": 38.18,
-      "learning_rate": 7.471264367816092e-07,
-      "loss": 0.459,
-      "step": 420
-    },
-    {
-      "epoch": 38.18,
-      "eval_accuracy_safe": 0.6477345885328122,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9690508790751174,
-      "eval_iou_safe": 0.36317283147412854,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9587996460225279,
-      "eval_loss": 0.42110592126846313,
-      "eval_mean_accuracy": 0.8083927338039648,
-      "eval_mean_iou": 0.4406574924988855,
-      "eval_overall_accuracy": 0.9595597110577484,
-      "eval_runtime": 9.3518,
-      "eval_samples_per_second": 7.164,
-      "eval_steps_per_second": 0.535,
-      "step": 420
-    },
-    {
-      "epoch": 38.27,
-      "learning_rate": 7.413793103448277e-07,
-      "loss": 0.4621,
-      "step": 421
-    },
-    {
-      "epoch": 38.36,
-      "learning_rate": 7.35632183908046e-07,
-      "loss": 0.4535,
-      "step": 422
-    },
-    {
-      "epoch": 38.45,
-      "learning_rate": 7.298850574712644e-07,
-      "loss": 0.494,
-      "step": 423
-    },
-    {
-      "epoch": 38.55,
-      "learning_rate": 7.241379310344829e-07,
-      "loss": 0.4429,
-      "step": 424
-    },
-    {
-      "epoch": 38.64,
-      "learning_rate": 7.183908045977011e-07,
-      "loss": 0.4367,
-      "step": 425
-    },
-    {
-      "epoch": 38.73,
-      "learning_rate": 7.126436781609195e-07,
-      "loss": 0.425,
-      "step": 426
-    },
-    {
-      "epoch": 38.82,
-      "learning_rate": 7.068965517241379e-07,
-      "loss": 0.4453,
-      "step": 427
-    },
-    {
-      "epoch": 38.91,
-      "learning_rate": 7.011494252873563e-07,
-      "loss": 0.4533,
-      "step": 428
-    },
-    {
-      "epoch": 39.0,
-      "learning_rate": 6.954022988505747e-07,
-      "loss": 0.476,
-      "step": 429
-    },
-    {
-      "epoch": 39.09,
-      "learning_rate": 6.896551724137931e-07,
-      "loss": 0.4528,
-      "step": 430
-    },
-    {
-      "epoch": 39.09,
-      "eval_accuracy_safe": 0.6592746365665514,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9668307944818041,
-      "eval_iou_safe": 0.357427398033294,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9569448389901313,
-      "eval_loss": 0.4239484369754791,
-      "eval_mean_accuracy": 0.8130527155241778,
-      "eval_mean_iou": 0.4381240790078084,
-      "eval_overall_accuracy": 0.957746078718954,
-      "eval_runtime": 9.9636,
-      "eval_samples_per_second": 6.724,
-      "eval_steps_per_second": 0.502,
-      "step": 430
-    },
-    {
-      "epoch": 39.18,
-      "learning_rate": 6.839080459770115e-07,
-      "loss": 0.4644,
-      "step": 431
-    },
-    {
-      "epoch": 39.27,
-      "learning_rate": 6.781609195402299e-07,
-      "loss": 0.4788,
-      "step": 432
-    },
-    {
-      "epoch": 39.36,
-      "learning_rate": 6.724137931034483e-07,
-      "loss": 0.4112,
-      "step": 433
-    },
-    {
-      "epoch": 39.45,
-      "learning_rate": 6.666666666666666e-07,
-      "loss": 0.4425,
-      "step": 434
-    },
-    {
-      "epoch": 39.55,
-      "learning_rate": 6.609195402298851e-07,
-      "loss": 0.4941,
-      "step": 435
-    },
-    {
-      "epoch": 39.64,
-      "learning_rate": 6.551724137931035e-07,
-      "loss": 0.4367,
-      "step": 436
-    },
-    {
-      "epoch": 39.73,
-      "learning_rate": 6.494252873563218e-07,
-      "loss": 0.4364,
-      "step": 437
-    },
-    {
-      "epoch": 39.82,
-      "learning_rate": 6.436781609195403e-07,
-      "loss": 0.4365,
-      "step": 438
-    },
-    {
-      "epoch": 39.91,
-      "learning_rate": 6.379310344827586e-07,
-      "loss": 0.4137,
-      "step": 439
-    },
-    {
-      "epoch": 40.0,
-      "learning_rate": 6.321839080459771e-07,
-      "loss": 0.4202,
-      "step": 440
-    },
-    {
-      "epoch": 40.0,
-      "eval_accuracy_safe": 0.657156294694315,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9689412271603979,
-      "eval_iou_safe": 0.3649963867997752,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9589753883685492,
-      "eval_loss": 0.41405442357063293,
-      "eval_mean_accuracy": 0.8130487609273565,
-      "eval_mean_iou": 0.4413239250561081,
-      "eval_overall_accuracy": 0.9597316001778218,
-      "eval_runtime": 9.3639,
-      "eval_samples_per_second": 7.155,
-      "eval_steps_per_second": 0.534,
-      "step": 440
-    },
-    {
-      "epoch": 40.09,
-      "learning_rate": 6.264367816091955e-07,
-      "loss": 0.4279,
-      "step": 441
-    },
-    {
-      "epoch": 40.18,
-      "learning_rate": 6.206896551724138e-07,
-      "loss": 0.4672,
-      "step": 442
-    },
-    {
-      "epoch": 40.27,
-      "learning_rate": 6.149425287356323e-07,
-      "loss": 0.4416,
-      "step": 443
-    },
-    {
-      "epoch": 40.36,
-      "learning_rate": 6.091954022988506e-07,
-      "loss": 0.4589,
-      "step": 444
-    },
-    {
-      "epoch": 40.45,
-      "learning_rate": 6.03448275862069e-07,
-      "loss": 0.4292,
-      "step": 445
-    },
-    {
-      "epoch": 40.55,
-      "learning_rate": 5.977011494252874e-07,
-      "loss": 0.4247,
-      "step": 446
-    },
-    {
-      "epoch": 40.64,
-      "learning_rate": 5.919540229885057e-07,
-      "loss": 0.4454,
-      "step": 447
-    },
-    {
-      "epoch": 40.73,
-      "learning_rate": 5.862068965517241e-07,
-      "loss": 0.4544,
-      "step": 448
-    },
-    {
-      "epoch": 40.82,
-      "learning_rate": 5.804597701149425e-07,
-      "loss": 0.4194,
-      "step": 449
-    },
-    {
-      "epoch": 40.91,
-      "learning_rate": 5.747126436781609e-07,
-      "loss": 0.4805,
-      "step": 450
-    },
-    {
-      "epoch": 40.91,
-      "eval_accuracy_safe": 0.6469790016229698,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9723798619242439,
-      "eval_iou_safe": 0.37541578774301354,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9620762119956281,
-      "eval_loss": 0.40121129155158997,
-      "eval_mean_accuracy": 0.8096794317736069,
-      "eval_mean_iou": 0.44583066657954723,
-      "eval_overall_accuracy": 0.9627680422654793,
-      "eval_runtime": 10.0604,
-      "eval_samples_per_second": 6.66,
-      "eval_steps_per_second": 0.497,
-      "step": 450
-    },
-    {
-      "epoch": 41.0,
-      "learning_rate": 5.689655172413793e-07,
-      "loss": 0.5147,
-      "step": 451
-    },
-    {
-      "epoch": 41.09,
-      "learning_rate": 5.632183908045977e-07,
-      "loss": 0.455,
-      "step": 452
-    },
-    {
-      "epoch": 41.18,
-      "learning_rate": 5.574712643678161e-07,
-      "loss": 0.4551,
-      "step": 453
-    },
-    {
-      "epoch": 41.27,
-      "learning_rate": 5.517241379310345e-07,
-      "loss": 0.4154,
-      "step": 454
-    },
-    {
-      "epoch": 41.36,
-      "learning_rate": 5.459770114942529e-07,
-      "loss": 0.4427,
-      "step": 455
-    },
-    {
-      "epoch": 41.45,
-      "learning_rate": 5.402298850574712e-07,
-      "loss": 0.4494,
-      "step": 456
-    },
-    {
-      "epoch": 41.55,
-      "learning_rate": 5.344827586206897e-07,
-      "loss": 0.417,
-      "step": 457
-    },
-    {
-      "epoch": 41.64,
-      "learning_rate": 5.28735632183908e-07,
-      "loss": 0.4539,
-      "step": 458
-    },
-    {
-      "epoch": 41.73,
-      "learning_rate": 5.229885057471265e-07,
-      "loss": 0.4351,
-      "step": 459
-    },
-    {
-      "epoch": 41.82,
-      "learning_rate": 5.172413793103449e-07,
-      "loss": 0.4611,
-      "step": 460
-    },
-    {
-      "epoch": 41.82,
-      "eval_accuracy_safe": 0.6525167597657681,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9718386426019924,
-      "eval_iou_safe": 0.37157282960052246,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9616940748738755,
-      "eval_loss": 0.4025232493877411,
-      "eval_mean_accuracy": 0.8121777011838802,
-      "eval_mean_iou": 0.44442230149146594,
-      "eval_overall_accuracy": 0.962406386190386,
-      "eval_runtime": 9.1949,
-      "eval_samples_per_second": 7.287,
-      "eval_steps_per_second": 0.544,
-      "step": 460
-    },
-    {
-      "epoch": 41.91,
-      "learning_rate": 5.114942528735632e-07,
-      "loss": 0.4298,
-      "step": 461
-    },
-    {
-      "epoch": 42.0,
-      "learning_rate": 5.057471264367817e-07,
-      "loss": 0.4254,
-      "step": 462
-    },
-    {
-      "epoch": 42.09,
-      "learning_rate": 5e-07,
-      "loss": 0.4288,
-      "step": 463
-    },
-    {
-      "epoch": 42.18,
-      "learning_rate": 4.942528735632184e-07,
-      "loss": 0.423,
-      "step": 464
-    },
-    {
-      "epoch": 42.27,
-      "learning_rate": 4.885057471264368e-07,
-      "loss": 0.4418,
-      "step": 465
-    },
-    {
-      "epoch": 42.36,
-      "learning_rate": 4.827586206896552e-07,
-      "loss": 0.4314,
-      "step": 466
-    },
-    {
-      "epoch": 42.45,
-      "learning_rate": 4.770114942528736e-07,
-      "loss": 0.4333,
-      "step": 467
-    },
-    {
-      "epoch": 42.55,
-      "learning_rate": 4.7126436781609196e-07,
-      "loss": 0.4233,
-      "step": 468
-    },
-    {
-      "epoch": 42.64,
-      "learning_rate": 4.6551724137931035e-07,
-      "loss": 0.4519,
-      "step": 469
-    },
-    {
-      "epoch": 42.73,
-      "learning_rate": 4.597701149425287e-07,
-      "loss": 0.4339,
-      "step": 470
-    },
-    {
-      "epoch": 42.73,
-      "eval_accuracy_safe": 0.6487465352870652,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9726293214969499,
-      "eval_iou_safe": 0.3744354059959327,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9623660982402702,
-      "eval_loss": 0.39508846402168274,
-      "eval_mean_accuracy": 0.8106879283920075,
-      "eval_mean_iou": 0.44560050141206764,
-      "eval_overall_accuracy": 0.9630623433127332,
-      "eval_runtime": 9.952,
-      "eval_samples_per_second": 6.732,
-      "eval_steps_per_second": 0.502,
-      "step": 470
-    },
-    {
-      "epoch": 42.82,
-      "learning_rate": 4.5402298850574714e-07,
-      "loss": 0.4261,
-      "step": 471
-    },
-    {
-      "epoch": 42.91,
-      "learning_rate": 4.4827586206896554e-07,
-      "loss": 0.4671,
-      "step": 472
-    },
-    {
-      "epoch": 43.0,
-      "learning_rate": 4.425287356321839e-07,
-      "loss": 0.4605,
-      "step": 473
-    },
-    {
-      "epoch": 43.09,
-      "learning_rate": 4.3678160919540233e-07,
-      "loss": 0.416,
-      "step": 474
-    },
-    {
-      "epoch": 43.18,
-      "learning_rate": 4.310344827586207e-07,
-      "loss": 0.4621,
-      "step": 475
-    },
-    {
-      "epoch": 43.27,
-      "learning_rate": 4.252873563218391e-07,
-      "loss": 0.3928,
-      "step": 476
-    },
-    {
-      "epoch": 43.36,
-      "learning_rate": 4.1954022988505747e-07,
-      "loss": 0.4083,
-      "step": 477
-    },
-    {
-      "epoch": 43.45,
-      "learning_rate": 4.1379310344827586e-07,
-      "loss": 0.4524,
-      "step": 478
-    },
-    {
-      "epoch": 43.55,
-      "learning_rate": 4.080459770114943e-07,
-      "loss": 0.4233,
-      "step": 479
-    },
-    {
-      "epoch": 43.64,
-      "learning_rate": 4.0229885057471266e-07,
-      "loss": 0.4361,
-      "step": 480
-    },
-    {
-      "epoch": 43.64,
-      "eval_accuracy_safe": 0.6447700664222574,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9739781749861512,
-      "eval_iou_safe": 0.3768989563134417,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9635797860139171,
-      "eval_loss": 0.3945651650428772,
-      "eval_mean_accuracy": 0.8093741207042043,
-      "eval_mean_iou": 0.4468262474424529,
-      "eval_overall_accuracy": 0.9642538953183303,
-      "eval_runtime": 9.4487,
-      "eval_samples_per_second": 7.091,
-      "eval_steps_per_second": 0.529,
-      "step": 480
-    },
-    {
-      "epoch": 43.73,
-      "learning_rate": 3.9655172413793105e-07,
-      "loss": 0.4263,
-      "step": 481
-    },
-    {
-      "epoch": 43.82,
-      "learning_rate": 3.908045977011494e-07,
-      "loss": 0.3853,
-      "step": 482
-    },
-    {
-      "epoch": 43.91,
-      "learning_rate": 3.8505747126436784e-07,
-      "loss": 0.4272,
-      "step": 483
-    },
-    {
-      "epoch": 44.0,
-      "learning_rate": 3.7931034482758624e-07,
-      "loss": 0.4668,
-      "step": 484
-    },
-    {
-      "epoch": 44.09,
-      "learning_rate": 3.735632183908046e-07,
-      "loss": 0.4326,
-      "step": 485
-    },
-    {
-      "epoch": 44.18,
-      "learning_rate": 3.67816091954023e-07,
-      "loss": 0.4139,
-      "step": 486
-    },
-    {
-      "epoch": 44.27,
-      "learning_rate": 3.6206896551724143e-07,
-      "loss": 0.4106,
-      "step": 487
-    },
-    {
-      "epoch": 44.36,
-      "learning_rate": 3.5632183908045977e-07,
-      "loss": 0.4596,
-      "step": 488
-    },
-    {
-      "epoch": 44.45,
-      "learning_rate": 3.5057471264367817e-07,
-      "loss": 0.4191,
-      "step": 489
-    },
-    {
-      "epoch": 44.55,
-      "learning_rate": 3.4482758620689656e-07,
-      "loss": 0.4416,
-      "step": 490
-    },
-    {
-      "epoch": 44.55,
-      "eval_accuracy_safe": 0.6447026033053073,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9746319796611832,
-      "eval_iou_safe": 0.37830962816343844,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9642232525873471,
-      "eval_loss": 0.3870936930179596,
-      "eval_mean_accuracy": 0.8096672914832452,
-      "eval_mean_iou": 0.4475109602502618,
-      "eval_overall_accuracy": 0.9648863948992829,
-      "eval_runtime": 10.7764,
-      "eval_samples_per_second": 6.217,
-      "eval_steps_per_second": 0.464,
-      "step": 490
-    },
-    {
-      "epoch": 44.64,
-      "learning_rate": 3.3908045977011496e-07,
-      "loss": 0.3936,
-      "step": 491
-    },
-    {
-      "epoch": 44.73,
-      "learning_rate": 3.333333333333333e-07,
-      "loss": 0.4229,
-      "step": 492
-    },
-    {
-      "epoch": 44.82,
-      "learning_rate": 3.2758620689655175e-07,
-      "loss": 0.4332,
-      "step": 493
-    },
-    {
-      "epoch": 44.91,
-      "learning_rate": 3.2183908045977015e-07,
-      "loss": 0.4375,
-      "step": 494
-    },
-    {
-      "epoch": 45.0,
-      "learning_rate": 3.1609195402298854e-07,
-      "loss": 0.4713,
-      "step": 495
-    },
-    {
-      "epoch": 45.09,
-      "learning_rate": 3.103448275862069e-07,
-      "loss": 0.4549,
-      "step": 496
-    },
-    {
-      "epoch": 45.18,
-      "learning_rate": 3.045977011494253e-07,
-      "loss": 0.3869,
-      "step": 497
-    },
-    {
-      "epoch": 45.27,
-      "learning_rate": 2.988505747126437e-07,
-      "loss": 0.4323,
-      "step": 498
-    },
-    {
-      "epoch": 45.36,
-      "learning_rate": 2.9310344827586207e-07,
-      "loss": 0.4158,
-      "step": 499
-    },
-    {
-      "epoch": 45.45,
-      "learning_rate": 2.8735632183908047e-07,
-      "loss": 0.4524,
-      "step": 500
-    },
-    {
-      "epoch": 45.45,
-      "eval_accuracy_safe": 0.6588968431116302,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9712046679682527,
-      "eval_iou_safe": 0.3701253070960435,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9612466938703839,
-      "eval_loss": 0.40250322222709656,
-      "eval_mean_accuracy": 0.8150507555399414,
-      "eval_mean_iou": 0.44379066698880915,
-      "eval_overall_accuracy": 0.9619795955828766,
-      "eval_runtime": 9.5625,
-      "eval_samples_per_second": 7.007,
-      "eval_steps_per_second": 0.523,
-      "step": 500
-    },
-    {
-      "epoch": 45.55,
-      "learning_rate": 2.8160919540229886e-07,
-      "loss": 0.4077,
-      "step": 501
-    },
-    {
-      "epoch": 45.64,
-      "learning_rate": 2.7586206896551726e-07,
-      "loss": 0.4103,
-      "step": 502
-    },
-    {
-      "epoch": 45.73,
-      "learning_rate": 2.701149425287356e-07,
-      "loss": 0.4242,
-      "step": 503
-    },
-    {
-      "epoch": 45.82,
-      "learning_rate": 2.64367816091954e-07,
-      "loss": 0.3912,
-      "step": 504
-    },
-    {
-      "epoch": 45.91,
-      "learning_rate": 2.5862068965517245e-07,
-      "loss": 0.4475,
-      "step": 505
-    },
-    {
-      "epoch": 46.0,
-      "learning_rate": 2.5287356321839084e-07,
-      "loss": 0.4558,
-      "step": 506
-    },
-    {
-      "epoch": 46.09,
-      "learning_rate": 2.471264367816092e-07,
-      "loss": 0.4301,
-      "step": 507
-    },
-    {
-      "epoch": 46.18,
-      "learning_rate": 2.413793103448276e-07,
-      "loss": 0.4541,
-      "step": 508
-    },
-    {
-      "epoch": 46.27,
-      "learning_rate": 2.3563218390804598e-07,
-      "loss": 0.4438,
-      "step": 509
-    },
-    {
-      "epoch": 46.36,
-      "learning_rate": 2.2988505747126435e-07,
-      "loss": 0.4319,
-      "step": 510
-    },
-    {
-      "epoch": 46.36,
-      "eval_accuracy_safe": 0.6730409674596474,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.967265588671203,
-      "eval_iou_safe": 0.3593928995264361,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9577608849356718,
-      "eval_loss": 0.4168514609336853,
-      "eval_mean_accuracy": 0.8201532780654253,
-      "eval_mean_iou": 0.43905126148736934,
-      "eval_overall_accuracy": 0.9585746651265159,
-      "eval_runtime": 10.2031,
-      "eval_samples_per_second": 6.567,
-      "eval_steps_per_second": 0.49,
-      "step": 510
-    },
-    {
-      "epoch": 46.45,
-      "learning_rate": 2.2413793103448277e-07,
-      "loss": 0.4356,
-      "step": 511
-    },
-    {
-      "epoch": 46.55,
-      "learning_rate": 2.1839080459770117e-07,
-      "loss": 0.3902,
-      "step": 512
-    },
-    {
-      "epoch": 46.64,
-      "learning_rate": 2.1264367816091956e-07,
-      "loss": 0.4067,
-      "step": 513
-    },
-    {
-      "epoch": 46.73,
-      "learning_rate": 2.0689655172413793e-07,
-      "loss": 0.3947,
-      "step": 514
-    },
-    {
-      "epoch": 46.82,
-      "learning_rate": 2.0114942528735633e-07,
-      "loss": 0.4378,
-      "step": 515
-    },
-    {
-      "epoch": 46.91,
-      "learning_rate": 1.954022988505747e-07,
-      "loss": 0.3984,
-      "step": 516
-    },
-    {
-      "epoch": 47.0,
-      "learning_rate": 1.8965517241379312e-07,
-      "loss": 0.4473,
-      "step": 517
-    },
-    {
-      "epoch": 47.09,
-      "learning_rate": 1.839080459770115e-07,
-      "loss": 0.4235,
-      "step": 518
-    },
-    {
-      "epoch": 47.18,
-      "learning_rate": 1.7816091954022988e-07,
-      "loss": 0.4192,
-      "step": 519
-    },
-    {
-      "epoch": 47.27,
-      "learning_rate": 1.7241379310344828e-07,
-      "loss": 0.4224,
-      "step": 520
-    },
-    {
-      "epoch": 47.27,
-      "eval_accuracy_safe": 0.6603020034618217,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9711994464485041,
-      "eval_iou_safe": 0.37161152869427305,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9612871860526603,
-      "eval_loss": 0.39859944581985474,
-      "eval_mean_accuracy": 0.8157507249551629,
-      "eval_mean_iou": 0.4442995715823111,
-      "eval_overall_accuracy": 0.9620160344821304,
-      "eval_runtime": 9.2903,
-      "eval_samples_per_second": 7.212,
-      "eval_steps_per_second": 0.538,
-      "step": 520
-    },
-    {
-      "epoch": 47.36,
-      "learning_rate": 1.0218878248974007e-06,
-      "loss": 0.4479,
-      "step": 521
-    },
-    {
-      "epoch": 47.45,
-      "learning_rate": 1.0177838577291383e-06,
-      "loss": 0.4097,
-      "step": 522
-    },
-    {
-      "epoch": 47.55,
-      "learning_rate": 1.0136798905608756e-06,
-      "loss": 0.4073,
-      "step": 523
-    },
-    {
-      "epoch": 47.64,
-      "learning_rate": 1.009575923392613e-06,
-      "loss": 0.4305,
-      "step": 524
-    },
-    {
-      "epoch": 47.73,
-      "learning_rate": 1.00547195622435e-06,
-      "loss": 0.4537,
-      "step": 525
-    },
-    {
-      "epoch": 47.82,
-      "learning_rate": 1.0013679890560876e-06,
-      "loss": 0.4054,
-      "step": 526
-    },
-    {
-      "epoch": 47.91,
-      "learning_rate": 9.97264021887825e-07,
-      "loss": 0.3876,
-      "step": 527
-    },
-    {
-      "epoch": 48.0,
-      "learning_rate": 9.931600547195623e-07,
-      "loss": 0.4008,
-      "step": 528
-    },
-    {
-      "epoch": 48.09,
-      "learning_rate": 9.890560875512995e-07,
-      "loss": 0.4183,
-      "step": 529
-    },
-    {
-      "epoch": 48.18,
-      "learning_rate": 9.84952120383037e-07,
-      "loss": 0.4333,
-      "step": 530
-    },
-    {
-      "epoch": 48.18,
-      "eval_accuracy_safe": 0.6649550310137586,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9702578714996897,
-      "eval_iou_safe": 0.3723805021059608,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9604921504914465,
-      "eval_loss": 0.4038476347923279,
-      "eval_mean_accuracy": 0.8176064512567242,
-      "eval_mean_iou": 0.44429088419913576,
-      "eval_overall_accuracy": 0.9612397151206856,
-      "eval_runtime": 8.6363,
-      "eval_samples_per_second": 7.758,
-      "eval_steps_per_second": 0.579,
-      "step": 530
-    },
-    {
-      "epoch": 48.27,
-      "learning_rate": 9.808481532147744e-07,
-      "loss": 0.414,
-      "step": 531
-    },
-    {
-      "epoch": 48.36,
-      "learning_rate": 9.767441860465117e-07,
-      "loss": 0.4344,
-      "step": 532
-    },
-    {
-      "epoch": 48.45,
-      "learning_rate": 9.726402188782489e-07,
-      "loss": 0.4448,
-      "step": 533
-    },
-    {
-      "epoch": 48.55,
-      "learning_rate": 9.685362517099864e-07,
-      "loss": 0.4283,
-      "step": 534
-    },
-    {
-      "epoch": 48.64,
-      "learning_rate": 9.644322845417238e-07,
-      "loss": 0.3871,
-      "step": 535
-    },
-    {
-      "epoch": 48.73,
-      "learning_rate": 9.603283173734611e-07,
-      "loss": 0.3983,
-      "step": 536
-    },
-    {
-      "epoch": 48.82,
-      "learning_rate": 9.562243502051983e-07,
-      "loss": 0.3946,
-      "step": 537
-    },
-    {
-      "epoch": 48.91,
-      "learning_rate": 9.521203830369357e-07,
-      "loss": 0.4091,
-      "step": 538
-    },
-    {
-      "epoch": 49.0,
-      "learning_rate": 9.480164158686731e-07,
-      "loss": 0.438,
-      "step": 539
-    },
-    {
-      "epoch": 49.09,
-      "learning_rate": 9.439124487004105e-07,
-      "loss": 0.3916,
-      "step": 540
-    },
-    {
-      "epoch": 49.09,
-      "eval_accuracy_safe": 0.6624396205103296,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9724313730965947,
-      "eval_iou_safe": 0.3781414596199825,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9625620999872877,
-      "eval_loss": 0.39684730768203735,
-      "eval_mean_accuracy": 0.8174354968034621,
-      "eval_mean_iou": 0.4469011865357568,
-      "eval_overall_accuracy": 0.9632747137724463,
-      "eval_runtime": 8.8446,
-      "eval_samples_per_second": 7.575,
-      "eval_steps_per_second": 0.565,
-      "step": 540
-    },
-    {
-      "epoch": 49.18,
-      "learning_rate": 9.398084815321478e-07,
-      "loss": 0.4174,
-      "step": 541
-    },
-    {
-      "epoch": 49.27,
-      "learning_rate": 9.357045143638851e-07,
-      "loss": 0.3837,
-      "step": 542
-    },
-    {
-      "epoch": 49.36,
-      "learning_rate": 9.316005471956224e-07,
-      "loss": 0.4472,
-      "step": 543
-    },
-    {
-      "epoch": 49.45,
-      "learning_rate": 9.274965800273599e-07,
-      "loss": 0.3876,
-      "step": 544
-    },
-    {
-      "epoch": 49.55,
-      "learning_rate": 9.233926128590971e-07,
-      "loss": 0.3935,
-      "step": 545
-    },
-    {
-      "epoch": 49.64,
-      "learning_rate": 9.192886456908345e-07,
-      "loss": 0.4331,
-      "step": 546
-    },
-    {
-      "epoch": 49.73,
-      "learning_rate": 9.151846785225718e-07,
-      "loss": 0.3941,
-      "step": 547
-    },
-    {
-      "epoch": 49.82,
-      "learning_rate": 9.110807113543093e-07,
-      "loss": 0.3938,
-      "step": 548
-    },
-    {
-      "epoch": 49.91,
-      "learning_rate": 9.069767441860465e-07,
-      "loss": 0.4009,
-      "step": 549
-    },
-    {
-      "epoch": 50.0,
-      "learning_rate": 9.028727770177839e-07,
-      "loss": 0.4803,
-      "step": 550
-    },
-    {
-      "epoch": 50.0,
-      "eval_accuracy_safe": 0.6680390592171965,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9725606203775616,
-      "eval_iou_safe": 0.38091072150808025,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.962854988299414,
-      "eval_loss": 0.39420995116233826,
-      "eval_mean_accuracy": 0.8202998397973791,
-      "eval_mean_iou": 0.44792190326916476,
-      "eval_overall_accuracy": 0.9635655417371152,
-      "eval_runtime": 8.6063,
-      "eval_samples_per_second": 7.785,
-      "eval_steps_per_second": 0.581,
-      "step": 550
-    },
-    {
-      "epoch": 50.09,
-      "learning_rate": 8.987688098495212e-07,
-      "loss": 0.4009,
-      "step": 551
-    },
-    {
-      "epoch": 50.18,
-      "learning_rate": 8.946648426812587e-07,
-      "loss": 0.4059,
-      "step": 552
-    },
-    {
-      "epoch": 50.27,
-      "learning_rate": 8.905608755129958e-07,
-      "loss": 0.412,
-      "step": 553
-    },
-    {
-      "epoch": 50.36,
-      "learning_rate": 8.864569083447333e-07,
-      "loss": 0.3805,
-      "step": 554
-    },
-    {
-      "epoch": 50.45,
-      "learning_rate": 8.823529411764706e-07,
-      "loss": 0.407,
-      "step": 555
-    },
-    {
-      "epoch": 50.55,
-      "learning_rate": 8.782489740082081e-07,
-      "loss": 0.3856,
-      "step": 556
-    },
-    {
-      "epoch": 50.64,
-      "learning_rate": 8.741450068399452e-07,
-      "loss": 0.413,
-      "step": 557
-    },
-    {
-      "epoch": 50.73,
-      "learning_rate": 8.700410396716827e-07,
-      "loss": 0.3965,
-      "step": 558
-    },
-    {
-      "epoch": 50.82,
-      "learning_rate": 8.6593707250342e-07,
-      "loss": 0.4005,
-      "step": 559
-    },
-    {
-      "epoch": 50.91,
-      "learning_rate": 8.618331053351575e-07,
-      "loss": 0.3543,
-      "step": 560
-    },
-    {
-      "epoch": 50.91,
-      "eval_accuracy_safe": 0.6472642742317879,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9777157857571726,
-      "eval_iou_safe": 0.3951853111506791,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9673436905478155,
-      "eval_loss": 0.3696909546852112,
-      "eval_mean_accuracy": 0.8124900299944802,
-      "eval_mean_iou": 0.4541763338994982,
-      "eval_overall_accuracy": 0.9679547779595674,
-      "eval_runtime": 9.4902,
-      "eval_samples_per_second": 7.06,
-      "eval_steps_per_second": 0.527,
-      "step": 560
-    },
-    {
-      "epoch": 51.0,
-      "learning_rate": 8.577291381668946e-07,
-      "loss": 0.3975,
-      "step": 561
-    },
-    {
-      "epoch": 51.09,
-      "learning_rate": 8.53625170998632e-07,
-      "loss": 0.3652,
-      "step": 562
-    },
-    {
-      "epoch": 51.18,
-      "learning_rate": 8.495212038303694e-07,
-      "loss": 0.3648,
-      "step": 563
-    },
-    {
-      "epoch": 51.27,
-      "learning_rate": 8.454172366621067e-07,
-      "loss": 0.4468,
-      "step": 564
-    },
-    {
-      "epoch": 51.36,
-      "learning_rate": 8.41313269493844e-07,
-      "loss": 0.4287,
-      "step": 565
-    },
-    {
-      "epoch": 51.45,
-      "learning_rate": 8.372093023255814e-07,
-      "loss": 0.3904,
-      "step": 566
-    },
-    {
-      "epoch": 51.55,
-      "learning_rate": 8.331053351573188e-07,
-      "loss": 0.3856,
-      "step": 567
-    },
-    {
-      "epoch": 51.64,
-      "learning_rate": 8.290013679890561e-07,
-      "loss": 0.407,
-      "step": 568
-    },
-    {
-      "epoch": 51.73,
-      "learning_rate": 8.248974008207934e-07,
-      "loss": 0.3889,
-      "step": 569
-    },
-    {
-      "epoch": 51.82,
-      "learning_rate": 8.207934336525308e-07,
-      "loss": 0.3684,
-      "step": 570
-    },
-    {
-      "epoch": 51.82,
-      "eval_accuracy_safe": 0.6514681901765992,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9772261362760333,
-      "eval_iou_safe": 0.3951423109973169,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9669824966129977,
-      "eval_loss": 0.3708346486091614,
-      "eval_mean_accuracy": 0.8143471632263162,
-      "eval_mean_iou": 0.45404160253677156,
-      "eval_overall_accuracy": 0.9676037688753498,
-      "eval_runtime": 10.3386,
-      "eval_samples_per_second": 6.481,
-      "eval_steps_per_second": 0.484,
-      "step": 570
-    },
-    {
-      "epoch": 51.91,
-      "learning_rate": 8.166894664842682e-07,
-      "loss": 0.413,
-      "step": 571
-    },
-    {
-      "epoch": 52.0,
-      "learning_rate": 8.125854993160055e-07,
-      "loss": 0.3516,
-      "step": 572
-    },
-    {
-      "epoch": 52.09,
-      "learning_rate": 8.084815321477428e-07,
-      "loss": 0.3804,
-      "step": 573
-    },
-    {
-      "epoch": 52.18,
-      "learning_rate": 8.043775649794802e-07,
-      "loss": 0.3756,
-      "step": 574
-    },
-    {
-      "epoch": 52.27,
-      "learning_rate": 8.002735978112176e-07,
-      "loss": 0.412,
-      "step": 575
-    },
-    {
-      "epoch": 52.36,
-      "learning_rate": 7.961696306429549e-07,
-      "loss": 0.4109,
-      "step": 576
-    },
-    {
-      "epoch": 52.45,
-      "learning_rate": 7.920656634746922e-07,
-      "loss": 0.3654,
-      "step": 577
-    },
-    {
-      "epoch": 52.55,
-      "learning_rate": 7.879616963064296e-07,
-      "loss": 0.3624,
-      "step": 578
-    },
-    {
-      "epoch": 52.64,
-      "learning_rate": 7.83857729138167e-07,
-      "loss": 0.3876,
-      "step": 579
-    },
-    {
-      "epoch": 52.73,
-      "learning_rate": 7.797537619699043e-07,
-      "loss": 0.4004,
-      "step": 580
-    },
-    {
-      "epoch": 52.73,
-      "eval_accuracy_safe": 0.64372727938597,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9793208457266203,
-      "eval_iou_safe": 0.401423873377919,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.968824985839696,
-      "eval_loss": 0.3584851026535034,
-      "eval_mean_accuracy": 0.8115240625562952,
-      "eval_mean_iou": 0.45674961973920497,
-      "eval_overall_accuracy": 0.9694079498746502,
-      "eval_runtime": 9.5054,
-      "eval_samples_per_second": 7.049,
-      "eval_steps_per_second": 0.526,
-      "step": 580
-    },
-    {
-      "epoch": 52.82,
-      "learning_rate": 7.756497948016415e-07,
-      "loss": 0.3849,
-      "step": 581
-    },
-    {
-      "epoch": 52.91,
-      "learning_rate": 7.715458276333789e-07,
-      "loss": 0.3474,
-      "step": 582
-    },
-    {
-      "epoch": 53.0,
-      "learning_rate": 7.674418604651163e-07,
-      "loss": 0.3701,
-      "step": 583
-    },
-    {
-      "epoch": 53.09,
-      "learning_rate": 7.633378932968537e-07,
-      "loss": 0.3447,
-      "step": 584
-    },
-    {
-      "epoch": 53.18,
-      "learning_rate": 7.592339261285909e-07,
-      "loss": 0.4049,
-      "step": 585
-    },
-    {
-      "epoch": 53.27,
-      "learning_rate": 7.551299589603283e-07,
-      "loss": 0.4026,
-      "step": 586
-    },
-    {
-      "epoch": 53.36,
-      "learning_rate": 7.510259917920657e-07,
-      "loss": 0.3855,
-      "step": 587
-    },
-    {
-      "epoch": 53.45,
-      "learning_rate": 7.46922024623803e-07,
-      "loss": 0.372,
-      "step": 588
-    },
-    {
-      "epoch": 53.55,
-      "learning_rate": 7.428180574555404e-07,
-      "loss": 0.358,
-      "step": 589
-    },
-    {
-      "epoch": 53.64,
-      "learning_rate": 7.387140902872777e-07,
-      "loss": 0.3656,
-      "step": 590
-    },
-    {
-      "epoch": 53.64,
-      "eval_accuracy_safe": 0.6558667854017525,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9780078388505241,
-      "eval_iou_safe": 0.40104638732360337,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9678817565011693,
-      "eval_loss": 0.3653949499130249,
-      "eval_mean_accuracy": 0.8169373121261383,
-      "eval_mean_iou": 0.4563093812749242,
-      "eval_overall_accuracy": 0.96849230865934,
-      "eval_runtime": 10.8561,
-      "eval_samples_per_second": 6.172,
-      "eval_steps_per_second": 0.461,
-      "step": 590
-    },
-    {
-      "epoch": 53.73,
-      "learning_rate": 7.346101231190151e-07,
-      "loss": 0.4163,
-      "step": 591
-    },
-    {
-      "epoch": 53.82,
-      "learning_rate": 7.305061559507524e-07,
-      "loss": 0.3652,
-      "step": 592
-    },
-    {
-      "epoch": 53.91,
-      "learning_rate": 7.264021887824898e-07,
-      "loss": 0.3957,
-      "step": 593
-    },
-    {
-      "epoch": 54.0,
-      "learning_rate": 7.222982216142271e-07,
-      "loss": 0.4365,
-      "step": 594
-    },
-    {
-      "epoch": 54.09,
-      "learning_rate": 7.181942544459645e-07,
-      "loss": 0.4133,
-      "step": 595
-    },
-    {
-      "epoch": 54.18,
-      "learning_rate": 7.140902872777018e-07,
-      "loss": 0.4297,
-      "step": 596
-    },
-    {
-      "epoch": 54.27,
-      "learning_rate": 7.099863201094392e-07,
-      "loss": 0.327,
-      "step": 597
-    },
-    {
-      "epoch": 54.36,
-      "learning_rate": 7.058823529411765e-07,
-      "loss": 0.3749,
-      "step": 598
-    },
-    {
-      "epoch": 54.45,
-      "learning_rate": 7.017783857729139e-07,
-      "loss": 0.3516,
-      "step": 599
-    },
-    {
-      "epoch": 54.55,
-      "learning_rate": 6.976744186046511e-07,
-      "loss": 0.3918,
-      "step": 600
-    },
-    {
-      "epoch": 54.55,
-      "eval_accuracy_safe": 0.6432222697676571,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.980886949638618,
-      "eval_iou_safe": 0.41152118741259774,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9703598872160661,
-      "eval_loss": 0.3526640236377716,
-      "eval_mean_accuracy": 0.8120546097031376,
-      "eval_mean_iou": 0.46062702487622126,
-      "eval_overall_accuracy": 0.9709128764138293,
-      "eval_runtime": 9.0586,
-      "eval_samples_per_second": 7.396,
-      "eval_steps_per_second": 0.552,
-      "step": 600
-    },
-    {
-      "epoch": 54.64,
-      "learning_rate": 6.935704514363886e-07,
-      "loss": 0.3599,
-      "step": 601
-    },
-    {
-      "epoch": 54.73,
-      "learning_rate": 6.894664842681258e-07,
-      "loss": 0.3603,
-      "step": 602
-    },
-    {
-      "epoch": 54.82,
-      "learning_rate": 6.853625170998633e-07,
-      "loss": 0.3782,
-      "step": 603
-    },
-    {
-      "epoch": 54.91,
-      "learning_rate": 6.812585499316005e-07,
-      "loss": 0.3599,
-      "step": 604
-    },
-    {
-      "epoch": 55.0,
-      "learning_rate": 6.77154582763338e-07,
-      "loss": 0.4199,
-      "step": 605
-    },
-    {
-      "epoch": 55.09,
-      "learning_rate": 6.730506155950752e-07,
-      "loss": 0.3504,
-      "step": 606
-    },
-    {
-      "epoch": 55.18,
-      "learning_rate": 6.689466484268127e-07,
-      "loss": 0.3583,
-      "step": 607
-    },
-    {
-      "epoch": 55.27,
-      "learning_rate": 6.648426812585499e-07,
-      "loss": 0.3537,
-      "step": 608
-    },
-    {
-      "epoch": 55.36,
-      "learning_rate": 6.607387140902874e-07,
-      "loss": 0.4038,
-      "step": 609
-    },
-    {
-      "epoch": 55.45,
-      "learning_rate": 6.566347469220246e-07,
-      "loss": 0.3741,
-      "step": 610
-    },
-    {
-      "epoch": 55.45,
-      "eval_accuracy_safe": 0.6392901338082737,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9826956488782591,
-      "eval_iou_safe": 0.418450346516478,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9720286712517688,
-      "eval_loss": 0.3361377418041229,
-      "eval_mean_accuracy": 0.8109928913432665,
-      "eval_mean_iou": 0.46349300592274895,
-      "eval_overall_accuracy": 0.9725520005866662,
-      "eval_runtime": 9.4221,
-      "eval_samples_per_second": 7.111,
-      "eval_steps_per_second": 0.531,
-      "step": 610
-    },
-    {
-      "epoch": 55.55,
-      "learning_rate": 6.52530779753762e-07,
-      "loss": 0.3866,
-      "step": 611
-    },
-    {
-      "epoch": 55.64,
-      "learning_rate": 6.484268125854993e-07,
-      "loss": 0.3333,
-      "step": 612
-    },
-    {
-      "epoch": 55.73,
-      "learning_rate": 6.443228454172367e-07,
-      "loss": 0.3843,
-      "step": 613
-    },
-    {
-      "epoch": 55.82,
-      "learning_rate": 6.40218878248974e-07,
-      "loss": 0.3664,
-      "step": 614
-    },
-    {
-      "epoch": 55.91,
-      "learning_rate": 6.361149110807114e-07,
-      "loss": 0.3787,
-      "step": 615
-    },
-    {
-      "epoch": 56.0,
-      "learning_rate": 6.320109439124487e-07,
-      "loss": 0.4437,
-      "step": 616
-    },
-    {
-      "epoch": 56.09,
-      "learning_rate": 6.27906976744186e-07,
-      "loss": 0.3822,
-      "step": 617
-    },
-    {
-      "epoch": 56.18,
-      "learning_rate": 6.238030095759234e-07,
-      "loss": 0.3639,
-      "step": 618
-    },
-    {
-      "epoch": 56.27,
-      "learning_rate": 6.196990424076607e-07,
-      "loss": 0.3971,
-      "step": 619
-    },
-    {
-      "epoch": 56.36,
-      "learning_rate": 6.155950752393981e-07,
-      "loss": 0.3656,
-      "step": 620
-    },
-    {
-      "epoch": 56.36,
-      "eval_accuracy_safe": 0.6540472087617242,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9806777955048699,
-      "eval_iou_safe": 0.41467641232391395,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9704675533352136,
-      "eval_loss": 0.3472718894481659,
-      "eval_mean_accuracy": 0.8173625021332971,
-      "eval_mean_iou": 0.46171465521970917,
-      "eval_overall_accuracy": 0.9710296516987815,
-      "eval_runtime": 9.6643,
-      "eval_samples_per_second": 6.933,
-      "eval_steps_per_second": 0.517,
-      "step": 620
-    },
-    {
-      "epoch": 56.45,
-      "learning_rate": 6.114911080711354e-07,
-      "loss": 0.3344,
-      "step": 621
-    },
-    {
-      "epoch": 56.55,
-      "learning_rate": 6.073871409028728e-07,
-      "loss": 0.3539,
-      "step": 622
-    },
-    {
-      "epoch": 56.64,
-      "learning_rate": 6.032831737346101e-07,
-      "loss": 0.3563,
-      "step": 623
-    },
-    {
-      "epoch": 56.73,
-      "learning_rate": 5.991792065663475e-07,
-      "loss": 0.3685,
-      "step": 624
-    },
-    {
-      "epoch": 56.82,
-      "learning_rate": 5.950752393980848e-07,
-      "loss": 0.3558,
-      "step": 625
-    },
-    {
-      "epoch": 56.91,
-      "learning_rate": 5.909712722298222e-07,
-      "loss": 0.3618,
-      "step": 626
-    },
-    {
-      "epoch": 57.0,
-      "learning_rate": 5.868673050615595e-07,
-      "loss": 0.422,
-      "step": 627
-    },
-    {
-      "epoch": 57.09,
-      "learning_rate": 5.827633378932969e-07,
-      "loss": 0.3629,
-      "step": 628
-    },
-    {
-      "epoch": 57.18,
-      "learning_rate": 5.786593707250342e-07,
-      "loss": 0.348,
-      "step": 629
-    },
-    {
-      "epoch": 57.27,
-      "learning_rate": 5.745554035567716e-07,
-      "loss": 0.3341,
-      "step": 630
-    },
-    {
-      "epoch": 57.27,
-      "eval_accuracy_safe": 0.6257917278653513,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9845255862094618,
-      "eval_iou_safe": 0.4246501164098674,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9734443258687906,
-      "eval_loss": 0.3334999680519104,
-      "eval_mean_accuracy": 0.8051586570374065,
-      "eval_mean_iou": 0.4660314807595527,
-      "eval_overall_accuracy": 0.973929163235337,
-      "eval_runtime": 9.6696,
-      "eval_samples_per_second": 6.929,
-      "eval_steps_per_second": 0.517,
-      "step": 630
-    },
-    {
-      "epoch": 57.36,
-      "learning_rate": 5.704514363885089e-07,
-      "loss": 0.3766,
-      "step": 631
-    },
-    {
-      "epoch": 57.45,
-      "learning_rate": 5.663474692202463e-07,
-      "loss": 0.3456,
-      "step": 632
-    },
-    {
-      "epoch": 57.55,
-      "learning_rate": 5.622435020519836e-07,
-      "loss": 0.3709,
-      "step": 633
-    },
-    {
-      "epoch": 57.64,
-      "learning_rate": 5.58139534883721e-07,
-      "loss": 0.355,
-      "step": 634
-    },
-    {
-      "epoch": 57.73,
-      "learning_rate": 5.540355677154583e-07,
-      "loss": 0.3679,
-      "step": 635
-    },
-    {
-      "epoch": 57.82,
-      "learning_rate": 5.499316005471957e-07,
-      "loss": 0.3541,
-      "step": 636
-    },
-    {
-      "epoch": 57.91,
-      "learning_rate": 5.45827633378933e-07,
-      "loss": 0.3495,
-      "step": 637
-    },
-    {
-      "epoch": 58.0,
-      "learning_rate": 5.417236662106703e-07,
-      "loss": 0.384,
-      "step": 638
-    },
-    {
-      "epoch": 58.09,
-      "learning_rate": 5.376196990424077e-07,
-      "loss": 0.393,
-      "step": 639
-    },
-    {
-      "epoch": 58.18,
-      "learning_rate": 5.33515731874145e-07,
-      "loss": 0.3669,
-      "step": 640
-    },
-    {
-      "epoch": 58.18,
-      "eval_accuracy_safe": 0.6495040497145347,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9815480879088024,
-      "eval_iou_safe": 0.41900261749180867,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.971195827459642,
-      "eval_loss": 0.33950886130332947,
-      "eval_mean_accuracy": 0.8155260688116686,
-      "eval_mean_iou": 0.46339948165048356,
-      "eval_overall_accuracy": 0.971740039426889,
-      "eval_runtime": 9.4613,
-      "eval_samples_per_second": 7.082,
-      "eval_steps_per_second": 0.528,
-      "step": 640
-    },
-    {
-      "epoch": 58.27,
-      "learning_rate": 5.294117647058824e-07,
-      "loss": 0.3638,
-      "step": 641
-    },
-    {
-      "epoch": 58.36,
-      "learning_rate": 5.253077975376197e-07,
-      "loss": 0.3462,
-      "step": 642
-    },
-    {
-      "epoch": 58.45,
-      "learning_rate": 5.212038303693571e-07,
-      "loss": 0.3637,
-      "step": 643
-    },
-    {
-      "epoch": 58.55,
-      "learning_rate": 5.170998632010944e-07,
-      "loss": 0.3705,
-      "step": 644
-    },
-    {
-      "epoch": 58.64,
-      "learning_rate": 5.129958960328318e-07,
-      "loss": 0.3323,
-      "step": 645
-    },
-    {
-      "epoch": 58.73,
-      "learning_rate": 5.088919288645691e-07,
-      "loss": 0.336,
-      "step": 646
-    },
-    {
-      "epoch": 58.82,
-      "learning_rate": 5.047879616963065e-07,
-      "loss": 0.3403,
-      "step": 647
-    },
-    {
-      "epoch": 58.91,
-      "learning_rate": 5.006839945280438e-07,
-      "loss": 0.3547,
-      "step": 648
-    },
-    {
-      "epoch": 59.0,
-      "learning_rate": 4.965800273597812e-07,
-      "loss": 0.3592,
-      "step": 649
-    },
-    {
-      "epoch": 59.09,
-      "learning_rate": 4.924760601915185e-07,
-      "loss": 0.3347,
-      "step": 650
-    },
-    {
-      "epoch": 59.09,
-      "eval_accuracy_safe": 0.66117516894692,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9799751784205032,
-      "eval_iou_safe": 0.41742094387885487,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9699836589351479,
-      "eval_loss": 0.3415713906288147,
-      "eval_mean_accuracy": 0.8205751736837116,
-      "eval_mean_iou": 0.46246820093800095,
-      "eval_overall_accuracy": 0.9705583373112465,
-      "eval_runtime": 11.7726,
-      "eval_samples_per_second": 5.691,
-      "eval_steps_per_second": 0.425,
-      "step": 650
-    },
-    {
-      "epoch": 59.18,
-      "learning_rate": 4.883720930232559e-07,
-      "loss": 0.3449,
-      "step": 651
-    },
-    {
-      "epoch": 59.27,
-      "learning_rate": 4.842681258549932e-07,
-      "loss": 0.3679,
-      "step": 652
-    },
-    {
-      "epoch": 59.36,
-      "learning_rate": 4.801641586867306e-07,
-      "loss": 0.3559,
-      "step": 653
-    },
-    {
-      "epoch": 59.45,
-      "learning_rate": 4.7606019151846786e-07,
-      "loss": 0.3403,
-      "step": 654
-    },
-    {
-      "epoch": 59.55,
-      "learning_rate": 4.7195622435020526e-07,
-      "loss": 0.353,
-      "step": 655
-    },
-    {
-      "epoch": 59.64,
-      "learning_rate": 4.6785225718194255e-07,
-      "loss": 0.3617,
-      "step": 656
-    },
-    {
-      "epoch": 59.73,
-      "learning_rate": 4.6374829001367995e-07,
-      "loss": 0.3504,
-      "step": 657
-    },
-    {
-      "epoch": 59.82,
-      "learning_rate": 4.5964432284541724e-07,
-      "loss": 0.3331,
-      "step": 658
-    },
-    {
-      "epoch": 59.91,
-      "learning_rate": 4.5554035567715465e-07,
-      "loss": 0.38,
-      "step": 659
-    },
-    {
-      "epoch": 60.0,
-      "learning_rate": 4.5143638850889194e-07,
-      "loss": 0.4287,
-      "step": 660
-    },
-    {
-      "epoch": 60.0,
-      "eval_accuracy_safe": 0.6673470803890501,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9796976751799341,
-      "eval_iou_safe": 0.4184647553664668,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9698893282294503,
-      "eval_loss": 0.34193894267082214,
-      "eval_mean_accuracy": 0.8235223777844921,
-      "eval_mean_iou": 0.4627846945319724,
-      "eval_overall_accuracy": 0.9704713394392782,
-      "eval_runtime": 9.3579,
-      "eval_samples_per_second": 7.16,
-      "eval_steps_per_second": 0.534,
-      "step": 660
-    },
-    {
-      "epoch": 60.09,
-      "learning_rate": 4.4733242134062934e-07,
-      "loss": 0.3861,
-      "step": 661
-    },
-    {
-      "epoch": 60.18,
-      "learning_rate": 4.4322845417236663e-07,
-      "loss": 0.37,
-      "step": 662
-    },
-    {
-      "epoch": 60.27,
-      "learning_rate": 4.3912448700410404e-07,
-      "loss": 0.3201,
-      "step": 663
-    },
-    {
-      "epoch": 60.36,
-      "learning_rate": 4.3502051983584133e-07,
-      "loss": 0.3348,
-      "step": 664
-    },
-    {
-      "epoch": 60.45,
-      "learning_rate": 4.3091655266757873e-07,
-      "loss": 0.3363,
-      "step": 665
-    },
-    {
-      "epoch": 60.55,
-      "learning_rate": 4.26812585499316e-07,
-      "loss": 0.3438,
-      "step": 666
-    },
-    {
-      "epoch": 60.64,
-      "learning_rate": 4.2270861833105337e-07,
-      "loss": 0.3656,
-      "step": 667
-    },
-    {
-      "epoch": 60.73,
-      "learning_rate": 4.186046511627907e-07,
-      "loss": 0.3284,
-      "step": 668
-    },
-    {
-      "epoch": 60.82,
-      "learning_rate": 4.1450068399452807e-07,
-      "loss": 0.3491,
-      "step": 669
-    },
-    {
-      "epoch": 60.91,
-      "learning_rate": 4.103967168262654e-07,
-      "loss": 0.3838,
-      "step": 670
-    },
-    {
-      "epoch": 60.91,
-      "eval_accuracy_safe": 0.661071082995054,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9812073984123999,
-      "eval_iou_safe": 0.42273980606218975,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9711982397733865,
-      "eval_loss": 0.3380735516548157,
-      "eval_mean_accuracy": 0.821139240703727,
-      "eval_mean_iou": 0.4646460152785254,
-      "eval_overall_accuracy": 0.9717510849682253,
-      "eval_runtime": 10.5907,
-      "eval_samples_per_second": 6.326,
-      "eval_steps_per_second": 0.472,
-      "step": 670
-    },
-    {
-      "epoch": 61.0,
-      "learning_rate": 4.0629274965800276e-07,
-      "loss": 0.3348,
-      "step": 671
-    },
-    {
-      "epoch": 61.09,
-      "learning_rate": 4.021887824897401e-07,
-      "loss": 0.3757,
-      "step": 672
-    },
-    {
-      "epoch": 61.18,
-      "learning_rate": 3.9808481532147746e-07,
-      "loss": 0.3252,
-      "step": 673
-    },
-    {
-      "epoch": 61.27,
-      "learning_rate": 3.939808481532148e-07,
-      "loss": 0.3364,
-      "step": 674
-    },
-    {
-      "epoch": 61.36,
-      "learning_rate": 3.8987688098495215e-07,
-      "loss": 0.377,
-      "step": 675
-    },
-    {
-      "epoch": 61.45,
-      "learning_rate": 3.8577291381668945e-07,
-      "loss": 0.3288,
-      "step": 676
-    },
-    {
-      "epoch": 61.55,
-      "learning_rate": 3.8166894664842685e-07,
-      "loss": 0.362,
-      "step": 677
-    },
-    {
-      "epoch": 61.64,
-      "learning_rate": 3.7756497948016414e-07,
-      "loss": 0.3206,
-      "step": 678
-    },
-    {
-      "epoch": 61.73,
-      "learning_rate": 3.734610123119015e-07,
-      "loss": 0.3152,
-      "step": 679
-    },
-    {
-      "epoch": 61.82,
-      "learning_rate": 3.6935704514363884e-07,
-      "loss": 0.352,
-      "step": 680
-    },
-    {
-      "epoch": 61.82,
-      "eval_accuracy_safe": 0.6406548162882949,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9844651573854055,
-      "eval_iou_safe": 0.4317581022923749,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9738170049692355,
-      "eval_loss": 0.3216310143470764,
-      "eval_mean_accuracy": 0.8125599868368503,
-      "eval_mean_iou": 0.46852503575387017,
-      "eval_overall_accuracy": 0.9743095511820778,
-      "eval_runtime": 9.4908,
-      "eval_samples_per_second": 7.059,
-      "eval_steps_per_second": 0.527,
-      "step": 680
-    },
-    {
-      "epoch": 61.91,
-      "learning_rate": 3.652530779753762e-07,
-      "loss": 0.3738,
-      "step": 681
-    },
-    {
-      "epoch": 62.0,
-      "learning_rate": 3.6114911080711353e-07,
-      "loss": 0.3422,
-      "step": 682
-    },
-    {
-      "epoch": 62.09,
-      "learning_rate": 3.570451436388509e-07,
-      "loss": 0.3377,
-      "step": 683
-    },
-    {
-      "epoch": 62.18,
-      "learning_rate": 3.529411764705882e-07,
-      "loss": 0.3689,
-      "step": 684
-    },
-    {
-      "epoch": 62.27,
-      "learning_rate": 3.4883720930232557e-07,
-      "loss": 0.3431,
-      "step": 685
-    },
-    {
-      "epoch": 62.36,
-      "learning_rate": 3.447332421340629e-07,
-      "loss": 0.3775,
-      "step": 686
-    },
-    {
-      "epoch": 62.45,
-      "learning_rate": 3.4062927496580027e-07,
-      "loss": 0.3317,
-      "step": 687
-    },
-    {
-      "epoch": 62.55,
-      "learning_rate": 3.365253077975376e-07,
-      "loss": 0.3552,
-      "step": 688
-    },
-    {
-      "epoch": 62.64,
-      "learning_rate": 3.3242134062927496e-07,
-      "loss": 0.3333,
-      "step": 689
-    },
-    {
-      "epoch": 62.73,
-      "learning_rate": 3.283173734610123e-07,
-      "loss": 0.3343,
-      "step": 690
-    },
-    {
-      "epoch": 62.73,
-      "eval_accuracy_safe": 0.6499146109691173,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9836740091403583,
-      "eval_iou_safe": 0.4311119919192952,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9733058809214871,
-      "eval_loss": 0.3275364339351654,
-      "eval_mean_accuracy": 0.8167943100547378,
-      "eval_mean_iou": 0.46813929094692747,
-      "eval_overall_accuracy": 0.9738152916751691,
-      "eval_runtime": 9.716,
-      "eval_samples_per_second": 6.896,
-      "eval_steps_per_second": 0.515,
-      "step": 690
-    },
-    {
-      "epoch": 62.82,
-      "learning_rate": 3.2421340629274966e-07,
-      "loss": 0.3756,
-      "step": 691
-    },
-    {
-      "epoch": 62.91,
-      "learning_rate": 3.20109439124487e-07,
-      "loss": 0.3231,
-      "step": 692
-    },
-    {
-      "epoch": 63.0,
-      "learning_rate": 3.1600547195622435e-07,
-      "loss": 0.3676,
-      "step": 693
-    },
-    {
-      "epoch": 63.09,
-      "learning_rate": 3.119015047879617e-07,
-      "loss": 0.297,
-      "step": 694
-    },
-    {
-      "epoch": 63.18,
-      "learning_rate": 3.0779753761969905e-07,
-      "loss": 0.3796,
-      "step": 695
-    },
-    {
-      "epoch": 63.27,
-      "learning_rate": 3.036935704514364e-07,
-      "loss": 0.3462,
-      "step": 696
-    },
-    {
-      "epoch": 63.36,
-      "learning_rate": 2.9958960328317374e-07,
-      "loss": 0.3536,
-      "step": 697
-    },
-    {
-      "epoch": 63.45,
-      "learning_rate": 2.954856361149111e-07,
-      "loss": 0.3426,
-      "step": 698
-    },
-    {
-      "epoch": 63.55,
-      "learning_rate": 2.9138166894664844e-07,
-      "loss": 0.332,
-      "step": 699
-    },
-    {
-      "epoch": 63.64,
-      "learning_rate": 2.872777017783858e-07,
-      "loss": 0.3443,
-      "step": 700
-    },
-    {
-      "epoch": 63.64,
-      "eval_accuracy_safe": 0.6527538444339074,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.983601025201401,
-      "eval_iou_safe": 0.4323750817126982,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9733172883777333,
-      "eval_loss": 0.3273077607154846,
-      "eval_mean_accuracy": 0.8181774348176543,
-      "eval_mean_iou": 0.4685641233634772,
-      "eval_overall_accuracy": 0.9738283299688083,
-      "eval_runtime": 9.4474,
-      "eval_samples_per_second": 7.092,
-      "eval_steps_per_second": 0.529,
-      "step": 700
-    },
-    {
-      "epoch": 63.73,
-      "learning_rate": 2.8317373461012313e-07,
-      "loss": 0.3492,
-      "step": 701
-    },
-    {
-      "epoch": 63.82,
-      "learning_rate": 2.790697674418605e-07,
-      "loss": 0.3385,
-      "step": 702
-    },
-    {
-      "epoch": 63.91,
-      "learning_rate": 2.7496580027359783e-07,
-      "loss": 0.3446,
-      "step": 703
-    },
-    {
-      "epoch": 64.0,
-      "learning_rate": 2.708618331053352e-07,
-      "loss": 0.3284,
-      "step": 704
-    },
-    {
-      "epoch": 64.09,
-      "learning_rate": 2.667578659370725e-07,
-      "loss": 0.3588,
-      "step": 705
-    },
-    {
-      "epoch": 64.18,
-      "learning_rate": 2.6265389876880987e-07,
-      "loss": 0.3259,
-      "step": 706
-    },
-    {
-      "epoch": 64.27,
-      "learning_rate": 2.585499316005472e-07,
-      "loss": 0.3423,
-      "step": 707
-    },
-    {
-      "epoch": 64.36,
-      "learning_rate": 2.5444596443228456e-07,
-      "loss": 0.3579,
-      "step": 708
-    },
-    {
-      "epoch": 64.45,
-      "learning_rate": 2.503419972640219e-07,
-      "loss": 0.3768,
-      "step": 709
-    },
-    {
-      "epoch": 64.55,
-      "learning_rate": 2.4623803009575926e-07,
-      "loss": 0.3183,
-      "step": 710
-    },
-    {
-      "epoch": 64.55,
-      "eval_accuracy_safe": 0.6455776963080327,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9848008600371044,
-      "eval_iou_safe": 0.4367295259604276,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9742930333126004,
-      "eval_loss": 0.315491646528244,
-      "eval_mean_accuracy": 0.8151892781725686,
-      "eval_mean_iou": 0.4703408530910093,
-      "eval_overall_accuracy": 0.9747807516980527,
-      "eval_runtime": 9.6459,
-      "eval_samples_per_second": 6.946,
-      "eval_steps_per_second": 0.518,
-      "step": 710
-    },
-    {
-      "epoch": 64.64,
-      "learning_rate": 2.421340629274966e-07,
-      "loss": 0.3502,
-      "step": 711
-    },
-    {
-      "epoch": 64.73,
-      "learning_rate": 2.3803009575923393e-07,
-      "loss": 0.3373,
-      "step": 712
-    },
-    {
-      "epoch": 64.82,
-      "learning_rate": 2.3392612859097128e-07,
-      "loss": 0.3172,
-      "step": 713
-    },
-    {
-      "epoch": 64.91,
-      "learning_rate": 2.2982216142270862e-07,
-      "loss": 0.3584,
-      "step": 714
-    },
-    {
-      "epoch": 65.0,
-      "learning_rate": 2.2571819425444597e-07,
-      "loss": 0.3384,
-      "step": 715
-    },
-    {
-      "epoch": 65.09,
-      "learning_rate": 2.2161422708618332e-07,
-      "loss": 0.3099,
-      "step": 716
-    },
-    {
-      "epoch": 65.18,
-      "learning_rate": 2.1751025991792066e-07,
-      "loss": 0.356,
-      "step": 717
-    },
-    {
-      "epoch": 65.27,
-      "learning_rate": 2.13406292749658e-07,
-      "loss": 0.3414,
-      "step": 718
-    },
-    {
-      "epoch": 65.36,
-      "learning_rate": 2.0930232558139536e-07,
-      "loss": 0.3399,
-      "step": 719
-    },
-    {
-      "epoch": 65.45,
-      "learning_rate": 2.051983584131327e-07,
-      "loss": 0.3346,
-      "step": 720
-    },
-    {
-      "epoch": 65.45,
-      "eval_accuracy_safe": 0.6517303325738913,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9841023497660231,
-      "eval_iou_safe": 0.4356108967463073,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9737832392255908,
-      "eval_loss": 0.3212249279022217,
-      "eval_mean_accuracy": 0.8179163411699573,
-      "eval_mean_iou": 0.4697980453239661,
-      "eval_overall_accuracy": 0.9742846133104012,
-      "eval_runtime": 9.9806,
-      "eval_samples_per_second": 6.713,
-      "eval_steps_per_second": 0.501,
-      "step": 720
-    },
-    {
-      "epoch": 65.55,
-      "learning_rate": 2.0109439124487005e-07,
-      "loss": 0.337,
-      "step": 721
-    },
-    {
-      "epoch": 65.64,
-      "learning_rate": 1.969904240766074e-07,
-      "loss": 0.348,
-      "step": 722
-    },
-    {
-      "epoch": 65.73,
-      "learning_rate": 1.9288645690834472e-07,
-      "loss": 0.3375,
-      "step": 723
-    },
-    {
-      "epoch": 65.82,
-      "learning_rate": 1.8878248974008207e-07,
-      "loss": 0.3443,
-      "step": 724
-    },
-    {
-      "epoch": 65.91,
-      "learning_rate": 1.8467852257181942e-07,
-      "loss": 0.3469,
-      "step": 725
-    },
-    {
-      "epoch": 66.0,
-      "learning_rate": 1.8057455540355677e-07,
-      "loss": 0.3453,
-      "step": 726
-    },
-    {
-      "epoch": 66.09,
-      "learning_rate": 1.764705882352941e-07,
-      "loss": 0.3513,
-      "step": 727
-    },
-    {
-      "epoch": 66.18,
-      "learning_rate": 1.7236662106703146e-07,
-      "loss": 0.3488,
-      "step": 728
-    },
-    {
-      "epoch": 66.27,
-      "learning_rate": 1.682626538987688e-07,
-      "loss": 0.3507,
-      "step": 729
-    },
-    {
-      "epoch": 66.36,
-      "learning_rate": 1.6415868673050616e-07,
-      "loss": 0.3225,
-      "step": 730
-    },
-    {
-      "epoch": 66.36,
-      "eval_accuracy_safe": 0.6367342454346745,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9862625922228925,
-      "eval_iou_safe": 0.4432164889806017,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9754777569208625,
-      "eval_loss": 0.30519598722457886,
-      "eval_mean_accuracy": 0.8114984188287835,
-      "eval_mean_iou": 0.4728980819671547,
-      "eval_overall_accuracy": 0.9759380852998193,
-      "eval_runtime": 10.6181,
-      "eval_samples_per_second": 6.31,
-      "eval_steps_per_second": 0.471,
-      "step": 730
-    },
-    {
-      "epoch": 66.45,
-      "learning_rate": 1.600547195622435e-07,
-      "loss": 0.3436,
-      "step": 731
-    },
-    {
-      "epoch": 66.55,
-      "learning_rate": 1.5595075239398085e-07,
-      "loss": 0.3231,
-      "step": 732
-    },
-    {
-      "epoch": 66.64,
-      "learning_rate": 1.518467852257182e-07,
-      "loss": 0.3333,
-      "step": 733
-    },
-    {
-      "epoch": 66.73,
-      "learning_rate": 1.4774281805745554e-07,
-      "loss": 0.3432,
-      "step": 734
-    },
-    {
-      "epoch": 66.82,
-      "learning_rate": 1.436388508891929e-07,
-      "loss": 0.3159,
-      "step": 735
-    },
-    {
-      "epoch": 66.91,
-      "learning_rate": 1.3953488372093024e-07,
-      "loss": 0.3653,
-      "step": 736
-    },
-    {
-      "epoch": 67.0,
-      "learning_rate": 1.354309165526676e-07,
-      "loss": 0.3593,
-      "step": 737
-    },
-    {
-      "epoch": 67.09,
-      "learning_rate": 1.3132694938440493e-07,
-      "loss": 0.3153,
-      "step": 738
-    },
-    {
-      "epoch": 67.18,
-      "learning_rate": 1.2722298221614228e-07,
-      "loss": 0.3259,
-      "step": 739
-    },
-    {
-      "epoch": 67.27,
-      "learning_rate": 1.2311901504787963e-07,
-      "loss": 0.3792,
-      "step": 740
-    },
-    {
-      "epoch": 67.27,
-      "eval_accuracy_safe": 0.6381124205380858,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.986087524639413,
-      "eval_iou_safe": 0.44292960173610785,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9753461456304894,
-      "eval_loss": 0.3036666810512543,
-      "eval_mean_accuracy": 0.8120999725887494,
-      "eval_mean_iou": 0.47275858245553243,
-      "eval_overall_accuracy": 0.9758088980148087,
-      "eval_runtime": 9.8604,
-      "eval_samples_per_second": 6.795,
-      "eval_steps_per_second": 0.507,
-      "step": 740
-    },
-    {
-      "epoch": 67.36,
-      "learning_rate": 1.1901504787961696e-07,
-      "loss": 0.3021,
-      "step": 741
-    },
-    {
-      "epoch": 67.45,
-      "learning_rate": 1.1491108071135431e-07,
-      "loss": 0.353,
-      "step": 742
-    },
-    {
-      "epoch": 67.55,
-      "learning_rate": 1.1080711354309166e-07,
-      "loss": 0.3839,
-      "step": 743
-    },
-    {
-      "epoch": 67.64,
-      "learning_rate": 1.06703146374829e-07,
-      "loss": 0.3346,
-      "step": 744
-    },
-    {
-      "epoch": 67.73,
-      "learning_rate": 1.0259917920656635e-07,
-      "loss": 0.3334,
-      "step": 745
-    },
-    {
-      "epoch": 67.82,
-      "learning_rate": 9.84952120383037e-08,
-      "loss": 0.3281,
-      "step": 746
-    },
-    {
-      "epoch": 67.91,
-      "learning_rate": 9.439124487004104e-08,
-      "loss": 0.327,
-      "step": 747
-    },
-    {
-      "epoch": 68.0,
-      "learning_rate": 9.028727770177838e-08,
-      "loss": 0.3521,
-      "step": 748
-    },
-    {
-      "epoch": 68.09,
-      "learning_rate": 8.618331053351573e-08,
-      "loss": 0.3113,
-      "step": 749
-    },
-    {
-      "epoch": 68.18,
-      "learning_rate": 8.207934336525308e-08,
-      "loss": 0.3177,
-      "step": 750
-    },
-    {
-      "epoch": 68.18,
-      "eval_accuracy_safe": 0.6345195276810807,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9864928084419184,
-      "eval_iou_safe": 0.444597209973218,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9756419916569254,
-      "eval_loss": 0.2988866865634918,
-      "eval_mean_accuracy": 0.8105061680614996,
-      "eval_mean_iou": 0.47341306721004783,
-      "eval_overall_accuracy": 0.9760960820895522,
-      "eval_runtime": 10.085,
-      "eval_samples_per_second": 6.644,
-      "eval_steps_per_second": 0.496,
-      "step": 750
-    },
-    {
-      "epoch": 68.27,
-      "learning_rate": 7.797537619699043e-08,
-      "loss": 0.3472,
-      "step": 751
-    },
-    {
-      "epoch": 68.36,
-      "learning_rate": 7.387140902872777e-08,
-      "loss": 0.3095,
-      "step": 752
-    },
-    {
-      "epoch": 68.45,
-      "learning_rate": 6.976744186046512e-08,
-      "loss": 0.3655,
-      "step": 753
-    },
-    {
-      "epoch": 68.55,
-      "learning_rate": 6.566347469220247e-08,
-      "loss": 0.3232,
-      "step": 754
-    },
-    {
-      "epoch": 68.64,
-      "learning_rate": 6.155950752393981e-08,
-      "loss": 0.3631,
-      "step": 755
-    },
-    {
-      "epoch": 68.73,
-      "learning_rate": 5.7455540355677156e-08,
-      "loss": 0.3515,
-      "step": 756
-    },
-    {
-      "epoch": 68.82,
-      "learning_rate": 5.33515731874145e-08,
-      "loss": 0.3406,
-      "step": 757
-    },
-    {
-      "epoch": 68.91,
-      "learning_rate": 4.924760601915185e-08,
-      "loss": 0.3381,
-      "step": 758
-    },
-    {
-      "epoch": 69.0,
-      "learning_rate": 4.514363885088919e-08,
-      "loss": 0.3706,
-      "step": 759
-    },
-    {
-      "epoch": 69.09,
-      "learning_rate": 4.103967168262654e-08,
-      "loss": 0.3295,
-      "step": 760
-    },
-    {
-      "epoch": 69.09,
-      "eval_accuracy_safe": 0.6403599060913412,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9858564283889687,
-      "eval_iou_safe": 0.44260650493805614,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.975184116097936,
-      "eval_loss": 0.3063713610172272,
-      "eval_mean_accuracy": 0.813108167240155,
-      "eval_mean_iou": 0.47259687367866404,
-      "eval_overall_accuracy": 0.975651015096636,
-      "eval_runtime": 9.6213,
-      "eval_samples_per_second": 6.964,
-      "eval_steps_per_second": 0.52,
-      "step": 760
-    },
-    {
-      "epoch": 69.18,
-      "learning_rate": 5.878378378378378e-07,
-      "loss": 0.3508,
-      "step": 761
-    },
-    {
-      "epoch": 69.27,
-      "learning_rate": 5.844594594594596e-07,
-      "loss": 0.3553,
-      "step": 762
-    },
-    {
-      "epoch": 69.36,
-      "learning_rate": 5.810810810810811e-07,
-      "loss": 0.3406,
-      "step": 763
-    },
-    {
-      "epoch": 69.45,
-      "learning_rate": 5.777027027027027e-07,
-      "loss": 0.3433,
-      "step": 764
-    },
-    {
-      "epoch": 69.55,
-      "learning_rate": 5.743243243243243e-07,
-      "loss": 0.3606,
-      "step": 765
-    },
-    {
-      "epoch": 69.64,
-      "learning_rate": 5.70945945945946e-07,
-      "loss": 0.323,
-      "step": 766
-    },
-    {
-      "epoch": 69.73,
-      "learning_rate": 5.675675675675676e-07,
-      "loss": 0.3178,
-      "step": 767
-    },
-    {
-      "epoch": 69.82,
-      "learning_rate": 5.641891891891891e-07,
-      "loss": 0.3124,
-      "step": 768
-    },
-    {
-      "epoch": 69.91,
-      "learning_rate": 5.608108108108109e-07,
-      "loss": 0.3003,
-      "step": 769
-    },
-    {
-      "epoch": 70.0,
-      "learning_rate": 5.574324324324324e-07,
-      "loss": 0.3847,
-      "step": 770
-    },
-    {
-      "epoch": 70.0,
-      "eval_accuracy_safe": 0.6429312146059576,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9856813608054892,
-      "eval_iou_safe": 0.44386336810476407,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9750875071024252,
-      "eval_loss": 0.30536746978759766,
-      "eval_mean_accuracy": 0.8143062877057234,
-      "eval_mean_iou": 0.4729836250690631,
-      "eval_overall_accuracy": 0.9755570710594974,
-      "eval_runtime": 8.333,
-      "eval_samples_per_second": 8.04,
-      "eval_steps_per_second": 0.6,
-      "step": 770
-    },
-    {
-      "epoch": 70.09,
-      "learning_rate": 5.54054054054054e-07,
-      "loss": 0.3266,
-      "step": 771
-    },
-    {
-      "epoch": 70.18,
-      "learning_rate": 5.506756756756757e-07,
-      "loss": 0.328,
-      "step": 772
-    },
-    {
-      "epoch": 70.27,
-      "learning_rate": 5.472972972972973e-07,
-      "loss": 0.3455,
-      "step": 773
-    },
-    {
-      "epoch": 70.36,
-      "learning_rate": 5.43918918918919e-07,
-      "loss": 0.3372,
-      "step": 774
-    },
-    {
-      "epoch": 70.45,
-      "learning_rate": 5.405405405405405e-07,
-      "loss": 0.3266,
-      "step": 775
-    },
-    {
-      "epoch": 70.55,
-      "learning_rate": 5.371621621621622e-07,
-      "loss": 0.362,
-      "step": 776
-    },
-    {
-      "epoch": 70.64,
-      "learning_rate": 5.337837837837837e-07,
-      "loss": 0.3509,
-      "step": 777
-    },
-    {
-      "epoch": 70.73,
-      "learning_rate": 5.304054054054054e-07,
-      "loss": 0.2946,
-      "step": 778
-    },
-    {
-      "epoch": 70.82,
-      "learning_rate": 5.270270270270271e-07,
-      "loss": 0.3261,
-      "step": 779
-    },
-    {
-      "epoch": 70.91,
-      "learning_rate": 5.236486486486486e-07,
-      "loss": 0.3406,
-      "step": 780
-    },
-    {
-      "epoch": 70.91,
-      "eval_accuracy_safe": 0.6442862594978431,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9861615646160722,
-      "eval_iou_safe": 0.4475671202721901,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9756000336170195,
-      "eval_loss": 0.30749550461769104,
-      "eval_mean_accuracy": 0.8152239120569577,
-      "eval_mean_iou": 0.47438905129640324,
-      "eval_overall_accuracy": 0.9760631162728837,
-      "eval_runtime": 9.0467,
-      "eval_samples_per_second": 7.406,
-      "eval_steps_per_second": 0.553,
-      "step": 780
-    },
-    {
-      "epoch": 71.0,
-      "learning_rate": 5.202702702702703e-07,
-      "loss": 0.2752,
-      "step": 781
-    },
-    {
-      "epoch": 71.09,
-      "learning_rate": 5.168918918918919e-07,
-      "loss": 0.3096,
-      "step": 782
-    },
-    {
-      "epoch": 71.18,
-      "learning_rate": 5.135135135135135e-07,
-      "loss": 0.3476,
-      "step": 783
-    },
-    {
-      "epoch": 71.27,
-      "learning_rate": 5.101351351351352e-07,
-      "loss": 0.3235,
-      "step": 784
-    },
-    {
-      "epoch": 71.36,
-      "learning_rate": 5.067567567567567e-07,
-      "loss": 0.3134,
-      "step": 785
-    },
-    {
-      "epoch": 71.45,
-      "learning_rate": 5.033783783783784e-07,
-      "loss": 0.3702,
-      "step": 786
-    },
-    {
-      "epoch": 71.55,
-      "learning_rate": 5e-07,
-      "loss": 0.3547,
-      "step": 787
-    },
-    {
-      "epoch": 71.64,
-      "learning_rate": 4.966216216216216e-07,
-      "loss": 0.2856,
-      "step": 788
-    },
-    {
-      "epoch": 71.73,
-      "learning_rate": 4.932432432432432e-07,
-      "loss": 0.3265,
-      "step": 789
-    },
-    {
-      "epoch": 71.82,
-      "learning_rate": 4.898648648648649e-07,
-      "loss": 0.3847,
-      "step": 790
-    },
-    {
-      "epoch": 71.82,
-      "eval_accuracy_safe": 0.6342573852837884,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9877470878880337,
-      "eval_iou_safe": 0.45462382943213125,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9768733490672094,
-      "eval_loss": 0.29106298089027405,
-      "eval_mean_accuracy": 0.8110022365859111,
-      "eval_mean_iou": 0.47716572616644687,
-      "eval_overall_accuracy": 0.9773055688658757,
-      "eval_runtime": 9.0041,
-      "eval_samples_per_second": 7.441,
-      "eval_steps_per_second": 0.555,
-      "step": 790
-    },
-    {
-      "epoch": 71.91,
-      "learning_rate": 4.864864864864865e-07,
-      "loss": 0.3186,
-      "step": 791
-    },
-    {
-      "epoch": 72.0,
-      "learning_rate": 4.831081081081081e-07,
-      "loss": 0.369,
-      "step": 792
-    },
-    {
-      "epoch": 72.09,
-      "learning_rate": 4.797297297297298e-07,
-      "loss": 0.3192,
-      "step": 793
-    },
-    {
-      "epoch": 72.18,
-      "learning_rate": 4.763513513513513e-07,
-      "loss": 0.315,
-      "step": 794
-    },
-    {
-      "epoch": 72.27,
-      "learning_rate": 4.72972972972973e-07,
-      "loss": 0.3422,
-      "step": 795
-    },
-    {
-      "epoch": 72.36,
-      "learning_rate": 4.6959459459459463e-07,
-      "loss": 0.3231,
-      "step": 796
-    },
-    {
-      "epoch": 72.45,
-      "learning_rate": 4.662162162162162e-07,
-      "loss": 0.3281,
-      "step": 797
-    },
-    {
-      "epoch": 72.55,
-      "learning_rate": 4.6283783783783784e-07,
-      "loss": 0.3152,
-      "step": 798
-    },
-    {
-      "epoch": 72.64,
-      "learning_rate": 4.594594594594594e-07,
-      "loss": 0.3132,
-      "step": 799
-    },
-    {
-      "epoch": 72.73,
-      "learning_rate": 4.560810810810811e-07,
-      "loss": 0.3292,
-      "step": 800
-    },
-    {
-      "epoch": 72.73,
-      "eval_accuracy_safe": 0.6328483698983427,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9880534561591228,
-      "eval_iou_safe": 0.45670444681845496,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9771349564164951,
-      "eval_loss": 0.29046520590782166,
-      "eval_mean_accuracy": 0.8104509130287327,
-      "eval_mean_iou": 0.47794646774498334,
-      "eval_overall_accuracy": 0.9775612674542328,
-      "eval_runtime": 9.9385,
-      "eval_samples_per_second": 6.741,
-      "eval_steps_per_second": 0.503,
-      "step": 800
-    },
-    {
-      "epoch": 72.82,
-      "learning_rate": 4.5270270270270275e-07,
-      "loss": 0.3247,
-      "step": 801
-    },
-    {
-      "epoch": 72.91,
-      "learning_rate": 4.4932432432432433e-07,
-      "loss": 0.3317,
-      "step": 802
-    },
-    {
-      "epoch": 73.0,
-      "learning_rate": 4.4594594594594596e-07,
-      "loss": 0.3452,
-      "step": 803
-    },
-    {
-      "epoch": 73.09,
-      "learning_rate": 4.4256756756756754e-07,
-      "loss": 0.34,
-      "step": 804
-    },
-    {
-      "epoch": 73.18,
-      "learning_rate": 4.3918918918918923e-07,
-      "loss": 0.3101,
-      "step": 805
-    },
-    {
-      "epoch": 73.27,
-      "learning_rate": 4.3581081081081076e-07,
-      "loss": 0.3173,
-      "step": 806
-    },
-    {
-      "epoch": 73.36,
-      "learning_rate": 4.3243243243243244e-07,
-      "loss": 0.3542,
-      "step": 807
-    },
-    {
-      "epoch": 73.45,
-      "learning_rate": 4.290540540540541e-07,
-      "loss": 0.3141,
-      "step": 808
-    },
-    {
-      "epoch": 73.55,
-      "learning_rate": 4.2567567567567566e-07,
-      "loss": 0.3388,
-      "step": 809
-    },
-    {
-      "epoch": 73.64,
-      "learning_rate": 4.2229729729729735e-07,
-      "loss": 0.3156,
-      "step": 810
-    },
-    {
-      "epoch": 73.64,
-      "eval_accuracy_safe": 0.6318094378973096,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9882225395289579,
-      "eval_iou_safe": 0.4578583531101107,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.977271212196021,
-      "eval_loss": 0.28654035925865173,
-      "eval_mean_accuracy": 0.8100159887131337,
-      "eval_mean_iou": 0.47837652176871054,
-      "eval_overall_accuracy": 0.9776946679869695,
-      "eval_runtime": 9.2746,
-      "eval_samples_per_second": 7.224,
-      "eval_steps_per_second": 0.539,
-      "step": 810
-    },
-    {
-      "epoch": 73.73,
-      "learning_rate": 4.189189189189189e-07,
-      "loss": 0.3512,
-      "step": 811
-    },
-    {
-      "epoch": 73.82,
-      "learning_rate": 4.1554054054054056e-07,
-      "loss": 0.2992,
-      "step": 812
-    },
-    {
-      "epoch": 73.91,
-      "learning_rate": 4.121621621621622e-07,
-      "loss": 0.34,
-      "step": 813
-    },
-    {
-      "epoch": 74.0,
-      "learning_rate": 4.087837837837838e-07,
-      "loss": 0.3609,
-      "step": 814
-    },
-    {
-      "epoch": 74.09,
-      "learning_rate": 4.0540540540540546e-07,
-      "loss": 0.2993,
-      "step": 815
-    },
-    {
-      "epoch": 74.18,
-      "learning_rate": 4.02027027027027e-07,
-      "loss": 0.3078,
-      "step": 816
-    },
-    {
-      "epoch": 74.27,
-      "learning_rate": 3.986486486486487e-07,
-      "loss": 0.3215,
-      "step": 817
-    },
-    {
-      "epoch": 74.36,
-      "learning_rate": 3.9527027027027026e-07,
-      "loss": 0.2968,
-      "step": 818
-    },
-    {
-      "epoch": 74.45,
-      "learning_rate": 3.918918918918919e-07,
-      "loss": 0.3119,
-      "step": 819
-    },
-    {
-      "epoch": 74.55,
-      "learning_rate": 3.8851351351351353e-07,
-      "loss": 0.3106,
-      "step": 820
-    },
-    {
-      "epoch": 74.55,
-      "eval_accuracy_safe": 0.6333418144108928,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9883734942515761,
-      "eval_iou_safe": 0.460039398466344,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9774654093148158,
-      "eval_loss": 0.2812236547470093,
-      "eval_mean_accuracy": 0.8108576543312345,
-      "eval_mean_iou": 0.47916826926038664,
-      "eval_overall_accuracy": 0.9778864276942922,
-      "eval_runtime": 9.2069,
-      "eval_samples_per_second": 7.277,
-      "eval_steps_per_second": 0.543,
-      "step": 820
-    },
-    {
-      "epoch": 74.64,
-      "learning_rate": 3.851351351351351e-07,
-      "loss": 0.327,
-      "step": 821
-    },
-    {
-      "epoch": 74.73,
-      "learning_rate": 3.817567567567568e-07,
-      "loss": 0.3093,
-      "step": 822
-    },
-    {
-      "epoch": 74.82,
-      "learning_rate": 3.783783783783784e-07,
-      "loss": 0.326,
-      "step": 823
-    },
-    {
-      "epoch": 74.91,
-      "learning_rate": 3.75e-07,
-      "loss": 0.3625,
-      "step": 824
-    },
-    {
-      "epoch": 75.0,
-      "learning_rate": 3.7162162162162164e-07,
-      "loss": 0.3852,
-      "step": 825
-    },
-    {
-      "epoch": 75.09,
-      "learning_rate": 3.682432432432432e-07,
-      "loss": 0.3337,
-      "step": 826
-    },
-    {
-      "epoch": 75.18,
-      "learning_rate": 3.648648648648649e-07,
-      "loss": 0.3197,
-      "step": 827
-    },
-    {
-      "epoch": 75.27,
-      "learning_rate": 3.614864864864865e-07,
-      "loss": 0.3149,
-      "step": 828
-    },
-    {
-      "epoch": 75.36,
-      "learning_rate": 3.581081081081081e-07,
-      "loss": 0.3312,
-      "step": 829
-    },
-    {
-      "epoch": 75.45,
-      "learning_rate": 3.5472972972972976e-07,
-      "loss": 0.3004,
-      "step": 830
-    },
-    {
-      "epoch": 75.45,
-      "eval_accuracy_safe": 0.6232358394917521,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.989330147072024,
-      "eval_iou_safe": 0.4632075377093191,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9781148642197699,
-      "eval_loss": 0.27982932329177856,
-      "eval_mean_accuracy": 0.806282993281888,
-      "eval_mean_iou": 0.4804408006430297,
-      "eval_overall_accuracy": 0.978516308229361,
-      "eval_runtime": 9.385,
-      "eval_samples_per_second": 7.139,
-      "eval_steps_per_second": 0.533,
-      "step": 830
-    },
-    {
-      "epoch": 75.55,
-      "learning_rate": 3.5135135135135134e-07,
-      "loss": 0.2947,
-      "step": 831
-    },
-    {
-      "epoch": 75.64,
-      "learning_rate": 3.47972972972973e-07,
-      "loss": 0.3226,
-      "step": 832
-    },
-    {
-      "epoch": 75.73,
-      "learning_rate": 3.445945945945946e-07,
-      "loss": 0.3221,
-      "step": 833
-    },
-    {
-      "epoch": 75.82,
-      "learning_rate": 3.4121621621621624e-07,
-      "loss": 0.3153,
-      "step": 834
-    },
-    {
-      "epoch": 75.91,
-      "learning_rate": 3.378378378378379e-07,
-      "loss": 0.3358,
-      "step": 835
-    },
-    {
-      "epoch": 76.0,
-      "learning_rate": 3.3445945945945946e-07,
-      "loss": 0.3165,
-      "step": 836
-    },
-    {
-      "epoch": 76.09,
-      "learning_rate": 3.310810810810811e-07,
-      "loss": 0.3004,
-      "step": 837
-    },
-    {
-      "epoch": 76.18,
-      "learning_rate": 3.2770270270270267e-07,
-      "loss": 0.2937,
-      "step": 838
-    },
-    {
-      "epoch": 76.27,
-      "learning_rate": 3.2432432432432436e-07,
-      "loss": 0.3298,
-      "step": 839
-    },
-    {
-      "epoch": 76.36,
-      "learning_rate": 3.2094594594594594e-07,
-      "loss": 0.3336,
-      "step": 840
-    },
-    {
-      "epoch": 76.36,
-      "eval_accuracy_safe": 0.6484612626782472,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.987219421049624,
-      "eval_iou_safe": 0.4592642445455861,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9767689022192217,
-      "eval_loss": 0.29541948437690735,
-      "eval_mean_accuracy": 0.8178403418639356,
-      "eval_mean_iou": 0.4786777155882693,
-      "eval_overall_accuracy": 0.9772130482232393,
-      "eval_runtime": 10.9829,
-      "eval_samples_per_second": 6.1,
-      "eval_steps_per_second": 0.455,
-      "step": 840
-    },
-    {
-      "epoch": 76.45,
-      "learning_rate": 3.175675675675676e-07,
-      "loss": 0.329,
-      "step": 841
-    },
-    {
-      "epoch": 76.55,
-      "learning_rate": 3.141891891891892e-07,
-      "loss": 0.3153,
-      "step": 842
-    },
-    {
-      "epoch": 76.64,
-      "learning_rate": 3.108108108108108e-07,
-      "loss": 0.2911,
-      "step": 843
-    },
-    {
-      "epoch": 76.73,
-      "learning_rate": 3.074324324324324e-07,
-      "loss": 0.3222,
-      "step": 844
-    },
-    {
-      "epoch": 76.82,
-      "learning_rate": 3.0405405405405406e-07,
-      "loss": 0.2909,
-      "step": 845
-    },
-    {
-      "epoch": 76.91,
-      "learning_rate": 3.006756756756757e-07,
-      "loss": 0.3212,
-      "step": 846
-    },
-    {
-      "epoch": 77.0,
-      "learning_rate": 2.972972972972973e-07,
-      "loss": 0.3321,
-      "step": 847
-    },
-    {
-      "epoch": 77.09,
-      "learning_rate": 2.939189189189189e-07,
-      "loss": 0.3263,
-      "step": 848
-    },
-    {
-      "epoch": 77.18,
-      "learning_rate": 2.9054054054054054e-07,
-      "loss": 0.3369,
-      "step": 849
-    },
-    {
-      "epoch": 77.27,
-      "learning_rate": 2.871621621621622e-07,
-      "loss": 0.299,
-      "step": 850
-    },
-    {
-      "epoch": 77.27,
-      "eval_accuracy_safe": 0.6490125327196117,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9873693784033015,
-      "eval_iou_safe": 0.4612546833154106,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9769332073058408,
-      "eval_loss": 0.2908638119697571,
-      "eval_mean_accuracy": 0.8181909555614566,
-      "eval_mean_iou": 0.4793959635404171,
-      "eval_overall_accuracy": 0.9773748597102379,
-      "eval_runtime": 10.9162,
-      "eval_samples_per_second": 6.138,
-      "eval_steps_per_second": 0.458,
-      "step": 850
-    },
-    {
-      "epoch": 77.36,
-      "learning_rate": 2.837837837837838e-07,
-      "loss": 0.2851,
-      "step": 851
-    },
-    {
-      "epoch": 77.45,
-      "learning_rate": 2.8040540540540544e-07,
-      "loss": 0.2774,
-      "step": 852
-    },
-    {
-      "epoch": 77.55,
-      "learning_rate": 2.77027027027027e-07,
-      "loss": 0.3091,
-      "step": 853
-    },
-    {
-      "epoch": 77.64,
-      "learning_rate": 2.7364864864864866e-07,
-      "loss": 0.3395,
-      "step": 854
-    },
-    {
-      "epoch": 77.73,
-      "learning_rate": 2.7027027027027024e-07,
-      "loss": 0.3364,
-      "step": 855
-    },
-    {
-      "epoch": 77.82,
-      "learning_rate": 2.6689189189189187e-07,
-      "loss": 0.3395,
-      "step": 856
-    },
-    {
-      "epoch": 77.91,
-      "learning_rate": 2.6351351351351356e-07,
-      "loss": 0.2895,
-      "step": 857
-    },
-    {
-      "epoch": 78.0,
-      "learning_rate": 2.6013513513513514e-07,
-      "loss": 0.3205,
-      "step": 858
-    },
-    {
-      "epoch": 78.09,
-      "learning_rate": 2.567567567567568e-07,
-      "loss": 0.3041,
-      "step": 859
-    },
-    {
-      "epoch": 78.18,
-      "learning_rate": 2.5337837837837835e-07,
-      "loss": 0.292,
-      "step": 860
-    },
-    {
-      "epoch": 78.18,
-      "eval_accuracy_safe": 0.6497314967945382,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9875053725918087,
-      "eval_iou_safe": 0.462851791638289,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9770889201532796,
-      "eval_loss": 0.2852531969547272,
-      "eval_mean_accuracy": 0.8186184346931735,
-      "eval_mean_iou": 0.4799802372638562,
-      "eval_overall_accuracy": 0.9775280738944438,
-      "eval_runtime": 10.619,
-      "eval_samples_per_second": 6.309,
-      "eval_steps_per_second": 0.471,
-      "step": 860
-    },
-    {
-      "epoch": 78.27,
-      "learning_rate": 2.5e-07,
-      "loss": 0.3158,
-      "step": 861
-    },
-    {
-      "epoch": 78.36,
-      "learning_rate": 2.466216216216216e-07,
-      "loss": 0.3292,
-      "step": 862
-    },
-    {
-      "epoch": 78.45,
-      "learning_rate": 2.4324324324324326e-07,
-      "loss": 0.3412,
-      "step": 863
-    },
-    {
-      "epoch": 78.55,
-      "learning_rate": 2.398648648648649e-07,
-      "loss": 0.2824,
-      "step": 864
-    },
-    {
-      "epoch": 78.64,
-      "learning_rate": 2.364864864864865e-07,
-      "loss": 0.3077,
-      "step": 865
-    },
-    {
-      "epoch": 78.73,
-      "learning_rate": 2.331081081081081e-07,
-      "loss": 0.3257,
-      "step": 866
-    },
-    {
-      "epoch": 78.82,
-      "learning_rate": 2.297297297297297e-07,
-      "loss": 0.3032,
-      "step": 867
-    },
-    {
-      "epoch": 78.91,
-      "learning_rate": 2.2635135135135137e-07,
-      "loss": 0.3132,
-      "step": 868
-    },
-    {
-      "epoch": 79.0,
-      "learning_rate": 2.2297297297297298e-07,
-      "loss": 0.3469,
-      "step": 869
-    },
-    {
-      "epoch": 79.09,
-      "learning_rate": 2.1959459459459461e-07,
-      "loss": 0.2922,
-      "step": 870
-    },
-    {
-      "epoch": 79.09,
-      "eval_accuracy_safe": 0.6586057879499309,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.986641475082849,
-      "eval_iou_safe": 0.4601317293824394,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.976495115071741,
-      "eval_loss": 0.2916642725467682,
-      "eval_mean_accuracy": 0.82262363151639,
-      "eval_mean_iou": 0.4788756148180601,
-      "eval_overall_accuracy": 0.9769518268642141,
-      "eval_runtime": 9.4334,
-      "eval_samples_per_second": 7.102,
-      "eval_steps_per_second": 0.53,
-      "step": 870
-    },
-    {
-      "epoch": 79.18,
-      "learning_rate": 2.1621621621621622e-07,
-      "loss": 0.2991,
-      "step": 871
-    },
-    {
-      "epoch": 79.27,
-      "learning_rate": 2.1283783783783783e-07,
-      "loss": 0.3065,
-      "step": 872
-    },
-    {
-      "epoch": 79.36,
-      "learning_rate": 2.0945945945945944e-07,
-      "loss": 0.3218,
-      "step": 873
-    },
-    {
-      "epoch": 79.45,
-      "learning_rate": 2.060810810810811e-07,
-      "loss": 0.2962,
-      "step": 874
-    },
-    {
-      "epoch": 79.55,
-      "learning_rate": 2.0270270270270273e-07,
-      "loss": 0.3047,
-      "step": 875
-    },
-    {
-      "epoch": 79.64,
-      "learning_rate": 1.9932432432432434e-07,
-      "loss": 0.3408,
-      "step": 876
-    },
-    {
-      "epoch": 79.73,
-      "learning_rate": 1.9594594594594595e-07,
-      "loss": 0.2935,
-      "step": 877
-    },
-    {
-      "epoch": 79.82,
-      "learning_rate": 1.9256756756756755e-07,
-      "loss": 0.3327,
-      "step": 878
-    },
-    {
-      "epoch": 79.91,
-      "learning_rate": 1.891891891891892e-07,
-      "loss": 0.3071,
-      "step": 879
-    },
-    {
-      "epoch": 80.0,
-      "learning_rate": 1.8581081081081082e-07,
-      "loss": 0.3583,
-      "step": 880
-    },
-    {
-      "epoch": 80.0,
-      "eval_accuracy_safe": 0.6515260157054136,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9875674441411791,
-      "eval_iou_safe": 0.4643852215774202,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9772031493777439,
-      "eval_loss": 0.2875562906265259,
-      "eval_mean_accuracy": 0.8195467299232964,
-      "eval_mean_iou": 0.48052945698505467,
-      "eval_overall_accuracy": 0.9776413191610308,
-      "eval_runtime": 10.47,
-      "eval_samples_per_second": 6.399,
-      "eval_steps_per_second": 0.478,
-      "step": 880
-    },
-    {
-      "epoch": 80.09,
-      "learning_rate": 1.8243243243243246e-07,
-      "loss": 0.3193,
-      "step": 881
-    },
-    {
-      "epoch": 80.18,
-      "learning_rate": 1.7905405405405406e-07,
-      "loss": 0.3032,
-      "step": 882
-    },
-    {
-      "epoch": 80.27,
-      "learning_rate": 1.7567567567567567e-07,
-      "loss": 0.2931,
-      "step": 883
-    },
-    {
-      "epoch": 80.36,
-      "learning_rate": 1.722972972972973e-07,
-      "loss": 0.3185,
-      "step": 884
-    },
-    {
-      "epoch": 80.45,
-      "learning_rate": 1.6891891891891894e-07,
-      "loss": 0.3118,
-      "step": 885
-    },
-    {
-      "epoch": 80.55,
-      "learning_rate": 1.6554054054054055e-07,
-      "loss": 0.2959,
-      "step": 886
-    },
-    {
-      "epoch": 80.64,
-      "learning_rate": 1.6216216216216218e-07,
-      "loss": 0.3005,
-      "step": 887
-    },
-    {
-      "epoch": 80.73,
-      "learning_rate": 1.587837837837838e-07,
-      "loss": 0.3257,
-      "step": 888
-    },
-    {
-      "epoch": 80.82,
-      "learning_rate": 1.554054054054054e-07,
-      "loss": 0.287,
-      "step": 889
-    },
-    {
-      "epoch": 80.91,
-      "learning_rate": 1.5202702702702703e-07,
-      "loss": 0.293,
-      "step": 890
-    },
-    {
-      "epoch": 80.91,
-      "eval_accuracy_safe": 0.6464797745575384,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9881851088592998,
-      "eval_iou_safe": 0.46743123613294535,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9776656866480414,
-      "eval_loss": 0.27672818303108215,
-      "eval_mean_accuracy": 0.8173324417084191,
-      "eval_mean_iou": 0.48169897426032887,
-      "eval_overall_accuracy": 0.9780916811814949,
-      "eval_runtime": 10.4393,
-      "eval_samples_per_second": 6.418,
-      "eval_steps_per_second": 0.479,
-      "step": 890
-    },
-    {
-      "epoch": 81.0,
-      "learning_rate": 1.4864864864864866e-07,
-      "loss": 0.3292,
-      "step": 891
-    },
-    {
-      "epoch": 81.09,
-      "learning_rate": 1.4527027027027027e-07,
-      "loss": 0.3271,
-      "step": 892
-    },
-    {
-      "epoch": 81.18,
-      "learning_rate": 1.418918918918919e-07,
-      "loss": 0.3114,
-      "step": 893
-    },
-    {
-      "epoch": 81.27,
-      "learning_rate": 1.385135135135135e-07,
-      "loss": 0.298,
-      "step": 894
-    },
-    {
-      "epoch": 81.36,
-      "learning_rate": 1.3513513513513512e-07,
-      "loss": 0.292,
-      "step": 895
-    },
-    {
-      "epoch": 81.45,
-      "learning_rate": 1.3175675675675678e-07,
-      "loss": 0.2806,
-      "step": 896
-    },
-    {
-      "epoch": 81.55,
-      "learning_rate": 1.283783783783784e-07,
-      "loss": 0.2994,
-      "step": 897
-    },
-    {
-      "epoch": 81.64,
-      "learning_rate": 1.25e-07,
-      "loss": 0.3274,
-      "step": 898
-    },
-    {
-      "epoch": 81.73,
-      "learning_rate": 1.2162162162162163e-07,
-      "loss": 0.3284,
-      "step": 899
-    },
-    {
-      "epoch": 81.82,
-      "learning_rate": 1.1824324324324325e-07,
-      "loss": 0.3287,
-      "step": 900
-    },
-    {
-      "epoch": 81.82,
-      "eval_accuracy_safe": 0.651817070867113,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9876163152192751,
-      "eval_iou_safe": 0.4652088022143105,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9772601877034197,
-      "eval_loss": 0.28578364849090576,
-      "eval_mean_accuracy": 0.8197166930431941,
-      "eval_mean_iou": 0.48082299663924344,
-      "eval_overall_accuracy": 0.9776973439686334,
-      "eval_runtime": 9.8853,
-      "eval_samples_per_second": 6.778,
-      "eval_steps_per_second": 0.506,
-      "step": 900
-    },
-    {
-      "epoch": 81.91,
-      "learning_rate": 1.1486486486486486e-07,
-      "loss": 0.2998,
-      "step": 901
-    },
-    {
-      "epoch": 82.0,
-      "learning_rate": 1.1148648648648649e-07,
-      "loss": 0.3326,
-      "step": 902
-    },
-    {
-      "epoch": 82.09,
-      "learning_rate": 1.0810810810810811e-07,
-      "loss": 0.287,
-      "step": 903
-    },
-    {
-      "epoch": 82.18,
-      "learning_rate": 1.0472972972972972e-07,
-      "loss": 0.2957,
-      "step": 904
-    },
-    {
-      "epoch": 82.27,
-      "learning_rate": 1.0135135135135137e-07,
-      "loss": 0.3262,
-      "step": 905
-    },
-    {
-      "epoch": 82.36,
-      "learning_rate": 9.797297297297297e-08,
-      "loss": 0.3166,
-      "step": 906
-    },
-    {
-      "epoch": 82.45,
-      "learning_rate": 9.45945945945946e-08,
-      "loss": 0.2848,
-      "step": 907
-    },
-    {
-      "epoch": 82.55,
-      "learning_rate": 9.121621621621623e-08,
-      "loss": 0.2995,
-      "step": 908
-    },
-    {
-      "epoch": 82.64,
-      "learning_rate": 8.783783783783784e-08,
-      "loss": 0.324,
-      "step": 909
-    },
-    {
-      "epoch": 82.73,
-      "learning_rate": 8.445945945945947e-08,
-      "loss": 0.3067,
-      "step": 910
-    },
-    {
-      "epoch": 82.73,
-      "eval_accuracy_safe": 0.6528463652800105,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.987525202633101,
-      "eval_iou_safe": 0.4653979333846323,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.977200381052192,
-      "eval_loss": 0.2860513925552368,
-      "eval_mean_accuracy": 0.8201857839565557,
-      "eval_mean_iou": 0.4808661048122748,
-      "eval_overall_accuracy": 0.9776393264087279,
-      "eval_runtime": 9.3628,
-      "eval_samples_per_second": 7.156,
-      "eval_steps_per_second": 0.534,
-      "step": 910
-    },
-    {
-      "epoch": 82.82,
-      "learning_rate": 8.108108108108109e-08,
-      "loss": 0.2896,
-      "step": 911
-    },
-    {
-      "epoch": 82.91,
-      "learning_rate": 7.77027027027027e-08,
-      "loss": 0.3404,
-      "step": 912
-    },
-    {
-      "epoch": 83.0,
-      "learning_rate": 7.432432432432433e-08,
-      "loss": 0.3558,
-      "step": 913
-    },
-    {
-      "epoch": 83.09,
-      "learning_rate": 7.094594594594595e-08,
-      "loss": 0.301,
-      "step": 914
-    },
-    {
-      "epoch": 83.18,
-      "learning_rate": 6.756756756756756e-08,
-      "loss": 0.3109,
-      "step": 915
-    },
-    {
-      "epoch": 83.27,
-      "learning_rate": 6.41891891891892e-08,
-      "loss": 0.2861,
-      "step": 916
-    },
-    {
-      "epoch": 83.36,
-      "learning_rate": 6.081081081081081e-08,
-      "loss": 0.2973,
-      "step": 917
-    },
-    {
-      "epoch": 83.45,
-      "learning_rate": 5.743243243243243e-08,
-      "loss": 0.3144,
-      "step": 918
-    },
-    {
-      "epoch": 83.55,
-      "learning_rate": 5.4054054054054056e-08,
-      "loss": 0.3096,
-      "step": 919
-    },
-    {
-      "epoch": 83.64,
-      "learning_rate": 5.067567567567568e-08,
-      "loss": 0.3374,
-      "step": 920
-    },
-    {
-      "epoch": 83.64,
-      "eval_accuracy_safe": 0.6577017821827981,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9870246994311359,
-      "eval_iou_safe": 0.46313936379874804,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9768479096925932,
-      "eval_loss": 0.28693443536758423,
-      "eval_mean_accuracy": 0.822363240806967,
-      "eval_mean_iou": 0.47999575783044707,
-      "eval_overall_accuracy": 0.9772970284988631,
-      "eval_runtime": 11.1318,
-      "eval_samples_per_second": 6.019,
-      "eval_steps_per_second": 0.449,
-      "step": 920
-    },
-    {
-      "epoch": 83.73,
-      "learning_rate": 7.545691906005222e-07,
-      "loss": 0.3124,
-      "step": 921
-    },
-    {
-      "epoch": 83.82,
-      "learning_rate": 7.519582245430809e-07,
-      "loss": 0.2825,
-      "step": 922
-    },
-    {
-      "epoch": 83.91,
-      "learning_rate": 7.493472584856398e-07,
-      "loss": 0.2987,
-      "step": 923
-    },
-    {
-      "epoch": 84.0,
-      "learning_rate": 7.467362924281985e-07,
-      "loss": 0.3105,
-      "step": 924
-    },
-    {
-      "epoch": 84.09,
-      "learning_rate": 7.441253263707572e-07,
-      "loss": 0.2889,
-      "step": 925
-    },
-    {
-      "epoch": 84.18,
-      "learning_rate": 7.415143603133159e-07,
-      "loss": 0.3116,
-      "step": 926
-    },
-    {
-      "epoch": 84.27,
-      "learning_rate": 7.389033942558747e-07,
-      "loss": 0.3094,
-      "step": 927
-    },
-    {
-      "epoch": 84.36,
-      "learning_rate": 7.362924281984335e-07,
-      "loss": 0.3213,
-      "step": 928
-    },
-    {
-      "epoch": 84.45,
-      "learning_rate": 7.336814621409922e-07,
-      "loss": 0.3122,
-      "step": 929
-    },
-    {
-      "epoch": 84.55,
-      "learning_rate": 7.310704960835509e-07,
-      "loss": 0.3171,
-      "step": 930
-    },
-    {
-      "epoch": 84.55,
-      "eval_accuracy_safe": 0.6441841010636042,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9887108982973504,
-      "eval_iou_safe": 0.47137576040519213,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9781178068263727,
-      "eval_loss": 0.27060091495513916,
-      "eval_mean_accuracy": 0.8164474996804774,
-      "eval_mean_iou": 0.4831645224105216,
-      "eval_overall_accuracy": 0.9785341291285273,
-      "eval_runtime": 10.3282,
-      "eval_samples_per_second": 6.487,
-      "eval_steps_per_second": 0.484,
-      "step": 930
-    },
-    {
-      "epoch": 84.64,
-      "learning_rate": 7.284595300261096e-07,
-      "loss": 0.2831,
-      "step": 931
-    },
-    {
-      "epoch": 84.73,
-      "learning_rate": 7.258485639686684e-07,
-      "loss": 0.3347,
-      "step": 932
-    },
-    {
-      "epoch": 84.82,
-      "learning_rate": 7.232375979112272e-07,
-      "loss": 0.3065,
-      "step": 933
-    },
-    {
-      "epoch": 84.91,
-      "learning_rate": 7.206266318537859e-07,
-      "loss": 0.3036,
-      "step": 934
-    },
-    {
-      "epoch": 85.0,
-      "learning_rate": 7.180156657963447e-07,
-      "loss": 0.3049,
-      "step": 935
-    },
-    {
-      "epoch": 85.09,
-      "learning_rate": 7.154046997389033e-07,
-      "loss": 0.3003,
-      "step": 936
-    },
-    {
-      "epoch": 85.18,
-      "learning_rate": 7.127937336814622e-07,
-      "loss": 0.3006,
-      "step": 937
-    },
-    {
-      "epoch": 85.27,
-      "learning_rate": 7.101827676240209e-07,
-      "loss": 0.284,
-      "step": 938
-    },
-    {
-      "epoch": 85.36,
-      "learning_rate": 7.075718015665797e-07,
-      "loss": 0.3199,
-      "step": 939
-    },
-    {
-      "epoch": 85.45,
-      "learning_rate": 7.049608355091384e-07,
-      "loss": 0.3156,
-      "step": 940
-    },
-    {
-      "epoch": 85.45,
-      "eval_accuracy_safe": 0.6320773628474833,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9898702516878123,
-      "eval_iou_safe": 0.4757064376163252,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9789079344104102,
-      "eval_loss": 0.2708066403865814,
-      "eval_mean_accuracy": 0.8109738072676478,
-      "eval_mean_iou": 0.4848714573422452,
-      "eval_overall_accuracy": 0.979301623444059,
-      "eval_runtime": 9.3749,
-      "eval_samples_per_second": 7.147,
-      "eval_steps_per_second": 0.533,
-      "step": 940
-    },
-    {
-      "epoch": 85.55,
-      "learning_rate": 7.023498694516971e-07,
-      "loss": 0.2918,
-      "step": 941
-    },
-    {
-      "epoch": 85.64,
-      "learning_rate": 6.997389033942559e-07,
-      "loss": 0.2911,
-      "step": 942
-    },
-    {
-      "epoch": 85.73,
-      "learning_rate": 6.971279373368146e-07,
-      "loss": 0.315,
-      "step": 943
-    },
-    {
-      "epoch": 85.82,
-      "learning_rate": 6.945169712793734e-07,
-      "loss": 0.2923,
-      "step": 944
-    },
-    {
-      "epoch": 85.91,
-      "learning_rate": 6.919060052219321e-07,
-      "loss": 0.3048,
-      "step": 945
-    },
-    {
-      "epoch": 86.0,
-      "learning_rate": 6.89295039164491e-07,
-      "loss": 0.2949,
-      "step": 946
-    },
-    {
-      "epoch": 86.09,
-      "learning_rate": 6.866840731070496e-07,
-      "loss": 0.2835,
-      "step": 947
-    },
-    {
-      "epoch": 86.18,
-      "learning_rate": 6.840731070496084e-07,
-      "loss": 0.3182,
-      "step": 948
-    },
-    {
-      "epoch": 86.27,
-      "learning_rate": 6.814621409921671e-07,
-      "loss": 0.2974,
-      "step": 949
-    },
-    {
-      "epoch": 86.36,
-      "learning_rate": 6.788511749347258e-07,
-      "loss": 0.2749,
-      "step": 950
-    },
-    {
-      "epoch": 86.36,
-      "eval_accuracy_safe": 0.6518093607966045,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9886810358978896,
-      "eval_iou_safe": 0.4765481543905536,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9783128350267929,
-      "eval_loss": 0.2760426700115204,
-      "eval_mean_accuracy": 0.820245198347247,
-      "eval_mean_iou": 0.48495366313911553,
-      "eval_overall_accuracy": 0.9787303867624767,
-      "eval_runtime": 9.6435,
-      "eval_samples_per_second": 6.948,
-      "eval_steps_per_second": 0.518,
-      "step": 950
-    },
-    {
-      "epoch": 86.45,
-      "learning_rate": 6.762402088772847e-07,
-      "loss": 0.319,
-      "step": 951
-    },
-    {
-      "epoch": 86.55,
-      "learning_rate": 6.736292428198434e-07,
-      "loss": 0.2734,
-      "step": 952
-    },
-    {
-      "epoch": 86.64,
-      "learning_rate": 6.710182767624021e-07,
-      "loss": 0.2848,
-      "step": 953
-    },
-    {
-      "epoch": 86.73,
-      "learning_rate": 6.684073107049608e-07,
-      "loss": 0.2765,
-      "step": 954
-    },
-    {
-      "epoch": 86.82,
-      "learning_rate": 6.657963446475195e-07,
-      "loss": 0.2782,
-      "step": 955
-    },
-    {
-      "epoch": 86.91,
-      "learning_rate": 6.631853785900784e-07,
-      "loss": 0.2816,
-      "step": 956
-    },
-    {
-      "epoch": 87.0,
-      "learning_rate": 6.605744125326371e-07,
-      "loss": 0.3259,
-      "step": 957
-    },
-    {
-      "epoch": 87.09,
-      "learning_rate": 6.579634464751959e-07,
-      "loss": 0.2913,
-      "step": 958
-    },
-    {
-      "epoch": 87.18,
-      "learning_rate": 6.553524804177545e-07,
-      "loss": 0.2994,
-      "step": 959
-    },
-    {
-      "epoch": 87.27,
-      "learning_rate": 6.527415143603134e-07,
-      "loss": 0.2725,
-      "step": 960
-    },
-    {
-      "epoch": 87.27,
-      "eval_accuracy_safe": 0.6680833921226209,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9876196006699034,
-      "eval_iou_safe": 0.4764950831797042,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9777417669524591,
-      "eval_loss": 0.2780168652534485,
-      "eval_mean_accuracy": 0.8278514963962622,
-      "eval_mean_iou": 0.4847456167107211,
-      "eval_overall_accuracy": 0.9781810134204466,
-      "eval_runtime": 10.2148,
-      "eval_samples_per_second": 6.559,
-      "eval_steps_per_second": 0.489,
-      "step": 960
-    },
-    {
-      "epoch": 87.36,
-      "learning_rate": 6.501305483028721e-07,
-      "loss": 0.2836,
-      "step": 961
-    },
-    {
-      "epoch": 87.45,
-      "learning_rate": 6.475195822454308e-07,
-      "loss": 0.2821,
-      "step": 962
-    },
-    {
-      "epoch": 87.55,
-      "learning_rate": 6.449086161879896e-07,
-      "loss": 0.3042,
-      "step": 963
-    },
-    {
-      "epoch": 87.64,
-      "learning_rate": 6.422976501305482e-07,
-      "loss": 0.3242,
-      "step": 964
-    },
-    {
-      "epoch": 87.73,
-      "learning_rate": 6.396866840731071e-07,
-      "loss": 0.2735,
-      "step": 965
-    },
-    {
-      "epoch": 87.82,
-      "learning_rate": 6.370757180156658e-07,
-      "loss": 0.3072,
-      "step": 966
-    },
-    {
-      "epoch": 87.91,
-      "learning_rate": 6.344647519582246e-07,
-      "loss": 0.2815,
-      "step": 967
-    },
-    {
-      "epoch": 88.0,
-      "learning_rate": 6.318537859007833e-07,
-      "loss": 0.3286,
-      "step": 968
-    },
-    {
-      "epoch": 88.09,
-      "learning_rate": 6.29242819843342e-07,
-      "loss": 0.2896,
-      "step": 969
-    },
-    {
-      "epoch": 88.18,
-      "learning_rate": 6.266318537859008e-07,
-      "loss": 0.2948,
-      "step": 970
-    },
-    {
-      "epoch": 88.18,
-      "eval_accuracy_safe": 0.6564527507604057,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.989077226042406,
-      "eval_iou_safe": 0.4845336330525837,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9788417622656205,
-      "eval_loss": 0.26358866691589355,
-      "eval_mean_accuracy": 0.8227649884014059,
-      "eval_mean_iou": 0.48779179843940135,
-      "eval_overall_accuracy": 0.9792520323796059,
-      "eval_runtime": 10.7789,
-      "eval_samples_per_second": 6.216,
-      "eval_steps_per_second": 0.464,
-      "step": 970
-    },
-    {
-      "epoch": 88.27,
-      "learning_rate": 6.240208877284595e-07,
-      "loss": 0.3121,
-      "step": 971
-    },
-    {
-      "epoch": 88.36,
-      "learning_rate": 6.214099216710183e-07,
-      "loss": 0.2941,
-      "step": 972
-    },
-    {
-      "epoch": 88.45,
-      "learning_rate": 6.18798955613577e-07,
-      "loss": 0.2852,
-      "step": 973
-    },
-    {
-      "epoch": 88.55,
-      "learning_rate": 6.161879895561358e-07,
-      "loss": 0.2894,
-      "step": 974
-    },
-    {
-      "epoch": 88.64,
-      "learning_rate": 6.135770234986946e-07,
-      "loss": 0.3066,
-      "step": 975
-    },
-    {
-      "epoch": 88.73,
-      "learning_rate": 6.109660574412532e-07,
-      "loss": 0.2867,
-      "step": 976
-    },
-    {
-      "epoch": 88.82,
-      "learning_rate": 6.08355091383812e-07,
-      "loss": 0.2914,
-      "step": 977
-    },
-    {
-      "epoch": 88.91,
-      "learning_rate": 6.057441253263707e-07,
-      "loss": 0.2741,
-      "step": 978
-    },
-    {
-      "epoch": 89.0,
-      "learning_rate": 6.031331592689296e-07,
-      "loss": 0.3029,
-      "step": 979
-    },
-    {
-      "epoch": 89.09,
-      "learning_rate": 6.005221932114883e-07,
-      "loss": 0.2972,
-      "step": 980
-    },
-    {
-      "epoch": 89.09,
-      "eval_accuracy_safe": 0.6722256275033636,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9879379373682813,
-      "eval_iou_safe": 0.48294905349452316,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9781790142676435,
-      "eval_loss": 0.2769739031791687,
-      "eval_mean_accuracy": 0.8300817824358224,
-      "eval_mean_iou": 0.4870426892540556,
-      "eval_overall_accuracy": 0.9786123019545826,
-      "eval_runtime": 10.5252,
-      "eval_samples_per_second": 6.366,
-      "eval_steps_per_second": 0.475,
-      "step": 980
-    },
-    {
-      "epoch": 89.18,
-      "learning_rate": 5.97911227154047e-07,
-      "loss": 0.2912,
-      "step": 981
-    },
-    {
-      "epoch": 89.27,
-      "learning_rate": 5.953002610966057e-07,
-      "loss": 0.3064,
-      "step": 982
-    },
-    {
-      "epoch": 89.36,
-      "learning_rate": 5.926892950391645e-07,
-      "loss": 0.2837,
-      "step": 983
-    },
-    {
-      "epoch": 89.45,
-      "learning_rate": 5.900783289817233e-07,
-      "loss": 0.2857,
-      "step": 984
-    },
-    {
-      "epoch": 89.55,
-      "learning_rate": 5.87467362924282e-07,
-      "loss": 0.2545,
-      "step": 985
-    },
-    {
-      "epoch": 89.64,
-      "learning_rate": 5.848563968668408e-07,
-      "loss": 0.2777,
-      "step": 986
-    },
-    {
-      "epoch": 89.73,
-      "learning_rate": 5.822454308093994e-07,
-      "loss": 0.292,
-      "step": 987
-    },
-    {
-      "epoch": 89.82,
-      "learning_rate": 5.796344647519583e-07,
-      "loss": 0.3179,
-      "step": 988
-    },
-    {
-      "epoch": 89.91,
-      "learning_rate": 5.77023498694517e-07,
-      "loss": 0.2939,
-      "step": 989
-    },
-    {
-      "epoch": 90.0,
-      "learning_rate": 5.744125326370757e-07,
-      "loss": 0.3101,
-      "step": 990
-    },
-    {
-      "epoch": 90.0,
-      "eval_accuracy_safe": 0.6711462176321602,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9881590599293182,
-      "eval_iou_safe": 0.4859155224409652,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9783666378224161,
-      "eval_loss": 0.27649641036987305,
-      "eval_mean_accuracy": 0.8296526387807392,
-      "eval_mean_iou": 0.4880940534211271,
-      "eval_overall_accuracy": 0.978795008872872,
-      "eval_runtime": 9.7226,
-      "eval_samples_per_second": 6.891,
-      "eval_steps_per_second": 0.514,
-      "step": 990
-    },
-    {
-      "epoch": 90.09,
-      "learning_rate": 5.718015665796345e-07,
-      "loss": 0.297,
-      "step": 991
-    },
-    {
-      "epoch": 90.18,
-      "learning_rate": 5.691906005221933e-07,
-      "loss": 0.288,
-      "step": 992
-    },
-    {
-      "epoch": 90.27,
-      "learning_rate": 5.66579634464752e-07,
-      "loss": 0.2584,
-      "step": 993
-    },
-    {
-      "epoch": 90.36,
-      "learning_rate": 5.639686684073107e-07,
-      "loss": 0.2859,
-      "step": 994
-    },
-    {
-      "epoch": 90.45,
-      "learning_rate": 5.613577023498694e-07,
-      "loss": 0.279,
-      "step": 995
-    },
-    {
-      "epoch": 90.55,
-      "learning_rate": 5.587467362924282e-07,
-      "loss": 0.2944,
-      "step": 996
-    },
-    {
-      "epoch": 90.64,
-      "learning_rate": 5.56135770234987e-07,
-      "loss": 0.2943,
-      "step": 997
-    },
-    {
-      "epoch": 90.73,
-      "learning_rate": 5.535248041775457e-07,
-      "loss": 0.2939,
-      "step": 998
-    },
-    {
-      "epoch": 90.82,
-      "learning_rate": 5.509138381201044e-07,
-      "loss": 0.2731,
-      "step": 999
-    },
-    {
-      "epoch": 90.91,
-      "learning_rate": 5.483028720626632e-07,
-      "loss": 0.2874,
-      "step": 1000
-    },
-    {
-      "epoch": 90.91,
-      "eval_accuracy_safe": 0.6689314998785664,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9887533744804734,
-      "eval_iou_safe": 0.4898762051296529,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9788892177058136,
-      "eval_loss": 0.2690369188785553,
-      "eval_mean_accuracy": 0.8288424371795199,
-      "eval_mean_iou": 0.48958847427848884,
-      "eval_overall_accuracy": 0.979306349113806,
-      "eval_runtime": 12.0357,
-      "eval_samples_per_second": 5.567,
-      "eval_steps_per_second": 0.415,
-      "step": 1000
-    },
-    {
-      "epoch": 91.0,
-      "learning_rate": 5.45691906005222e-07,
-      "loss": 0.2744,
-      "step": 1001
-    },
-    {
-      "epoch": 91.09,
-      "learning_rate": 5.430809399477807e-07,
-      "loss": 0.2804,
-      "step": 1002
-    },
-    {
-      "epoch": 91.18,
-      "learning_rate": 5.404699738903395e-07,
-      "loss": 0.3046,
-      "step": 1003
-    },
-    {
-      "epoch": 91.27,
-      "learning_rate": 5.378590078328981e-07,
-      "loss": 0.2818,
-      "step": 1004
-    },
-    {
-      "epoch": 91.36,
-      "learning_rate": 5.352480417754569e-07,
-      "loss": 0.2463,
-      "step": 1005
-    },
-    {
-      "epoch": 91.45,
-      "learning_rate": 5.326370757180157e-07,
-      "loss": 0.2533,
-      "step": 1006
-    },
-    {
-      "epoch": 91.55,
-      "learning_rate": 5.300261096605745e-07,
-      "loss": 0.2837,
-      "step": 1007
-    },
-    {
-      "epoch": 91.64,
-      "learning_rate": 5.274151436031332e-07,
-      "loss": 0.276,
-      "step": 1008
-    },
-    {
-      "epoch": 91.73,
-      "learning_rate": 5.248041775456919e-07,
-      "loss": 0.2684,
-      "step": 1009
-    },
-    {
-      "epoch": 91.82,
-      "learning_rate": 5.221932114882506e-07,
-      "loss": 0.275,
-      "step": 1010
-    },
-    {
-      "epoch": 91.82,
-      "eval_accuracy_safe": 0.6541802074779974,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9901128470154555,
-      "eval_iou_safe": 0.4947347245788654,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9797995942698954,
-      "eval_loss": 0.2592916190624237,
-      "eval_mean_accuracy": 0.8221465272467265,
-      "eval_mean_iou": 0.4915114396162536,
-      "eval_overall_accuracy": 0.9801899354849288,
-      "eval_runtime": 12.6805,
-      "eval_samples_per_second": 5.284,
-      "eval_steps_per_second": 0.394,
-      "step": 1010
-    },
-    {
-      "epoch": 91.91,
-      "learning_rate": 5.195822454308094e-07,
-      "loss": 0.2927,
-      "step": 1011
-    },
-    {
-      "epoch": 92.0,
-      "learning_rate": 5.169712793733682e-07,
-      "loss": 0.3249,
-      "step": 1012
-    },
-    {
-      "epoch": 92.09,
-      "learning_rate": 5.143603133159269e-07,
-      "loss": 0.2613,
-      "step": 1013
-    },
-    {
-      "epoch": 92.18,
-      "learning_rate": 5.117493472584857e-07,
-      "loss": 0.2562,
-      "step": 1014
-    },
-    {
-      "epoch": 92.27,
-      "learning_rate": 5.091383812010444e-07,
-      "loss": 0.283,
-      "step": 1015
-    },
-    {
-      "epoch": 92.36,
-      "learning_rate": 5.065274151436031e-07,
-      "loss": 0.2957,
-      "step": 1016
-    },
-    {
-      "epoch": 92.45,
-      "learning_rate": 5.039164490861619e-07,
-      "loss": 0.2785,
-      "step": 1017
-    },
-    {
-      "epoch": 92.55,
-      "learning_rate": 5.013054830287206e-07,
-      "loss": 0.2555,
-      "step": 1018
-    },
-    {
-      "epoch": 92.64,
-      "learning_rate": 4.986945169712794e-07,
-      "loss": 0.2776,
-      "step": 1019
-    },
-    {
-      "epoch": 92.73,
-      "learning_rate": 4.960835509138382e-07,
-      "loss": 0.2711,
-      "step": 1020
-    },
-    {
-      "epoch": 92.73,
-      "eval_accuracy_safe": 0.6672584145782013,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9893351339167277,
-      "eval_iou_safe": 0.49565376756999,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9794157946407118,
-      "eval_loss": 0.26075512170791626,
-      "eval_mean_accuracy": 0.8282967742474645,
-      "eval_mean_iou": 0.49168985407023397,
-      "eval_overall_accuracy": 0.9798215040520056,
-      "eval_runtime": 11.1156,
-      "eval_samples_per_second": 6.028,
-      "eval_steps_per_second": 0.45,
-      "step": 1020
-    },
-    {
-      "epoch": 92.82,
-      "learning_rate": 4.934725848563969e-07,
-      "loss": 0.2631,
-      "step": 1021
-    },
-    {
-      "epoch": 92.91,
-      "learning_rate": 4.908616187989556e-07,
-      "loss": 0.2891,
-      "step": 1022
-    },
-    {
-      "epoch": 93.0,
-      "learning_rate": 4.882506527415143e-07,
-      "loss": 0.3368,
-      "step": 1023
-    },
-    {
-      "epoch": 93.09,
-      "learning_rate": 4.856396866840732e-07,
-      "loss": 0.2506,
-      "step": 1024
-    },
-    {
-      "epoch": 93.18,
-      "learning_rate": 4.830287206266319e-07,
-      "loss": 0.2542,
-      "step": 1025
-    },
-    {
-      "epoch": 93.27,
-      "learning_rate": 4.804177545691907e-07,
-      "loss": 0.2926,
-      "step": 1026
-    },
-    {
-      "epoch": 93.36,
-      "learning_rate": 4.778067885117493e-07,
-      "loss": 0.2756,
-      "step": 1027
-    },
-    {
-      "epoch": 93.45,
-      "learning_rate": 4.751958224543081e-07,
-      "loss": 0.2448,
-      "step": 1028
-    },
-    {
-      "epoch": 93.55,
-      "learning_rate": 4.7258485639686686e-07,
-      "loss": 0.3061,
-      "step": 1029
-    },
-    {
-      "epoch": 93.64,
-      "learning_rate": 4.6997389033942563e-07,
-      "loss": 0.2691,
-      "step": 1030
-    },
-    {
-      "epoch": 93.64,
-      "eval_accuracy_safe": 0.6818844183330056,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9884273521743757,
-      "eval_iou_safe": 0.4954192037733101,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9789485519513931,
-      "eval_loss": 0.2608976364135742,
-      "eval_mean_accuracy": 0.8351558852536907,
-      "eval_mean_iou": 0.49145591857490106,
-      "eval_overall_accuracy": 0.9793725654260436,
-      "eval_runtime": 9.5589,
-      "eval_samples_per_second": 7.009,
-      "eval_steps_per_second": 0.523,
-      "step": 1030
-    },
-    {
-      "epoch": 93.73,
-      "learning_rate": 4.6736292428198435e-07,
-      "loss": 0.2734,
-      "step": 1031
-    },
-    {
-      "epoch": 93.82,
-      "learning_rate": 4.6475195822454307e-07,
-      "loss": 0.2837,
-      "step": 1032
-    },
-    {
-      "epoch": 93.91,
-      "learning_rate": 4.621409921671018e-07,
-      "loss": 0.2912,
-      "step": 1033
-    },
-    {
-      "epoch": 94.0,
-      "learning_rate": 4.5953002610966057e-07,
-      "loss": 0.2975,
-      "step": 1034
-    },
-    {
-      "epoch": 94.09,
-      "learning_rate": 4.5691906005221934e-07,
-      "loss": 0.2788,
-      "step": 1035
-    },
-    {
-      "epoch": 94.18,
-      "learning_rate": 4.543080939947781e-07,
-      "loss": 0.3211,
-      "step": 1036
-    },
-    {
-      "epoch": 94.27,
-      "learning_rate": 4.516971279373369e-07,
-      "loss": 0.3114,
-      "step": 1037
-    },
-    {
-      "epoch": 94.36,
-      "learning_rate": 4.4908616187989555e-07,
-      "loss": 0.2451,
-      "step": 1038
-    },
-    {
-      "epoch": 94.45,
-      "learning_rate": 4.464751958224543e-07,
-      "loss": 0.3075,
-      "step": 1039
-    },
-    {
-      "epoch": 94.55,
-      "learning_rate": 4.4386422976501305e-07,
-      "loss": 0.274,
-      "step": 1040
-    },
-    {
-      "epoch": 94.55,
-      "eval_accuracy_safe": 0.6722159899152278,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9895092041312664,
-      "eval_iou_safe": 0.5006991902601504,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9797344753826932,
-      "eval_loss": 0.2541664242744446,
-      "eval_mean_accuracy": 0.830862597023247,
-      "eval_mean_iou": 0.49347788854761454,
-      "eval_overall_accuracy": 0.9801368713378906,
-      "eval_runtime": 10.6958,
-      "eval_samples_per_second": 6.264,
-      "eval_steps_per_second": 0.467,
-      "step": 1040
-    },
-    {
-      "epoch": 94.64,
-      "learning_rate": 4.412532637075718e-07,
-      "loss": 0.24,
-      "step": 1041
-    },
-    {
-      "epoch": 94.73,
-      "learning_rate": 4.386422976501306e-07,
-      "loss": 0.2447,
-      "step": 1042
-    },
-    {
-      "epoch": 94.82,
-      "learning_rate": 4.3603133159268937e-07,
-      "loss": 0.2805,
-      "step": 1043
-    },
-    {
-      "epoch": 94.91,
-      "learning_rate": 4.3342036553524803e-07,
-      "loss": 0.2669,
-      "step": 1044
-    },
-    {
-      "epoch": 95.0,
-      "learning_rate": 4.308093994778068e-07,
-      "loss": 0.303,
-      "step": 1045
-    },
-    {
-      "epoch": 95.09,
-      "learning_rate": 4.2819843342036553e-07,
-      "loss": 0.2528,
-      "step": 1046
-    },
-    {
-      "epoch": 95.18,
-      "learning_rate": 4.255874673629243e-07,
-      "loss": 0.2796,
-      "step": 1047
-    },
-    {
-      "epoch": 95.27,
-      "learning_rate": 4.229765013054831e-07,
-      "loss": 0.2867,
-      "step": 1048
-    },
-    {
-      "epoch": 95.36,
-      "learning_rate": 4.2036553524804174e-07,
-      "loss": 0.2571,
-      "step": 1049
-    },
-    {
-      "epoch": 95.45,
-      "learning_rate": 4.177545691906005e-07,
-      "loss": 0.27,
-      "step": 1050
-    },
-    {
-      "epoch": 95.45,
-      "eval_accuracy_safe": 0.6435942806696967,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9918861689920813,
-      "eval_iou_safe": 0.5086651838761769,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9812415734412862,
-      "eval_loss": 0.23565810918807983,
-      "eval_mean_accuracy": 0.8177402248308889,
-      "eval_mean_iou": 0.4966355857724876,
-      "eval_overall_accuracy": 0.9815981850695255,
-      "eval_runtime": 9.9504,
-      "eval_samples_per_second": 6.733,
-      "eval_steps_per_second": 0.502,
-      "step": 1050
-    },
-    {
-      "epoch": 95.55,
-      "learning_rate": 4.151436031331593e-07,
-      "loss": 0.2617,
-      "step": 1051
-    },
-    {
-      "epoch": 95.64,
-      "learning_rate": 4.12532637075718e-07,
-      "loss": 0.2838,
-      "step": 1052
-    },
-    {
-      "epoch": 95.73,
-      "learning_rate": 4.099216710182768e-07,
-      "loss": 0.2938,
-      "step": 1053
-    },
-    {
-      "epoch": 95.82,
-      "learning_rate": 4.0731070496083555e-07,
-      "loss": 0.2417,
-      "step": 1054
-    },
-    {
-      "epoch": 95.91,
-      "learning_rate": 4.046997389033942e-07,
-      "loss": 0.2567,
-      "step": 1055
-    },
-    {
-      "epoch": 96.0,
-      "learning_rate": 4.02088772845953e-07,
-      "loss": 0.3307,
-      "step": 1056
-    },
-    {
-      "epoch": 96.09,
-      "learning_rate": 3.9947780678851177e-07,
-      "loss": 0.2799,
-      "step": 1057
-    },
-    {
-      "epoch": 96.18,
-      "learning_rate": 3.9686684073107054e-07,
-      "loss": 0.2739,
-      "step": 1058
-    },
-    {
-      "epoch": 96.27,
-      "learning_rate": 3.9425587467362926e-07,
-      "loss": 0.2708,
-      "step": 1059
-    },
-    {
-      "epoch": 96.36,
-      "learning_rate": 3.9164490861618804e-07,
-      "loss": 0.255,
-      "step": 1060
-    },
-    {
-      "epoch": 96.36,
-      "eval_accuracy_safe": 0.6671022856504023,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9903374310334044,
-      "eval_iou_safe": 0.5074312624715747,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.980403430127183,
-      "eval_loss": 0.2459607720375061,
-      "eval_mean_accuracy": 0.8287198583419033,
-      "eval_mean_iou": 0.49594489753291926,
-      "eval_overall_accuracy": 0.9807895831207731,
-      "eval_runtime": 11.3645,
-      "eval_samples_per_second": 5.896,
-      "eval_steps_per_second": 0.44,
-      "step": 1060
-    },
-    {
-      "epoch": 96.45,
-      "learning_rate": 3.890339425587467e-07,
-      "loss": 0.2801,
-      "step": 1061
-    },
-    {
-      "epoch": 96.55,
-      "learning_rate": 3.864229765013055e-07,
-      "loss": 0.273,
-      "step": 1062
-    },
-    {
-      "epoch": 96.64,
-      "learning_rate": 3.8381201044386425e-07,
-      "loss": 0.2592,
-      "step": 1063
-    },
-    {
-      "epoch": 96.73,
-      "learning_rate": 3.81201044386423e-07,
-      "loss": 0.2829,
-      "step": 1064
-    },
-    {
-      "epoch": 96.82,
-      "learning_rate": 3.7859007832898174e-07,
-      "loss": 0.2865,
-      "step": 1065
-    },
-    {
-      "epoch": 96.91,
-      "learning_rate": 3.7597911227154046e-07,
-      "loss": 0.2452,
-      "step": 1066
-    },
-    {
-      "epoch": 97.0,
-      "learning_rate": 3.7336814621409924e-07,
-      "loss": 0.2683,
-      "step": 1067
-    },
-    {
-      "epoch": 97.09,
-      "learning_rate": 3.7075718015665796e-07,
-      "loss": 0.2888,
-      "step": 1068
-    },
-    {
-      "epoch": 97.18,
-      "learning_rate": 3.6814621409921673e-07,
-      "loss": 0.265,
-      "step": 1069
-    },
-    {
-      "epoch": 97.27,
-      "learning_rate": 3.6553524804177545e-07,
-      "loss": 0.2756,
-      "step": 1070
-    },
-    {
-      "epoch": 97.27,
-      "eval_accuracy_safe": 0.6633551913832252,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9909028805540396,
-      "eval_iou_safe": 0.5112561019451947,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9808524615525958,
-      "eval_loss": 0.2411348670721054,
-      "eval_mean_accuracy": 0.8271290359686324,
-      "eval_mean_iou": 0.49736952116593014,
-      "eval_overall_accuracy": 0.981227647012739,
-      "eval_runtime": 11.7692,
-      "eval_samples_per_second": 5.693,
-      "eval_steps_per_second": 0.425,
-      "step": 1070
-    },
-    {
-      "epoch": 97.36,
-      "learning_rate": 3.629242819843342e-07,
-      "loss": 0.2541,
-      "step": 1071
-    },
-    {
-      "epoch": 97.45,
-      "learning_rate": 3.6031331592689294e-07,
-      "loss": 0.3003,
-      "step": 1072
-    },
-    {
-      "epoch": 97.55,
-      "learning_rate": 3.5770234986945166e-07,
-      "loss": 0.2535,
-      "step": 1073
-    },
-    {
-      "epoch": 97.64,
-      "learning_rate": 3.5509138381201044e-07,
-      "loss": 0.2349,
-      "step": 1074
-    },
-    {
-      "epoch": 97.73,
-      "learning_rate": 3.524804177545692e-07,
-      "loss": 0.3017,
-      "step": 1075
-    },
-    {
-      "epoch": 97.82,
-      "learning_rate": 3.4986945169712793e-07,
-      "loss": 0.2767,
-      "step": 1076
-    },
-    {
-      "epoch": 97.91,
-      "learning_rate": 3.472584856396867e-07,
-      "loss": 0.2357,
-      "step": 1077
-    },
-    {
-      "epoch": 98.0,
-      "learning_rate": 3.446475195822455e-07,
-      "loss": 0.2671,
-      "step": 1078
-    },
-    {
-      "epoch": 98.09,
-      "learning_rate": 3.420365535248042e-07,
-      "loss": 0.2846,
-      "step": 1079
-    },
-    {
-      "epoch": 98.18,
-      "learning_rate": 3.394255874673629e-07,
-      "loss": 0.2473,
-      "step": 1080
-    },
-    {
-      "epoch": 98.18,
-      "eval_accuracy_safe": 0.6724203067837056,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9903802005603336,
-      "eval_iou_safe": 0.5115604181605686,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9806029058144462,
-      "eval_loss": 0.2446996420621872,
-      "eval_mean_accuracy": 0.8314002536720195,
-      "eval_mean_iou": 0.49738777465833833,
-      "eval_overall_accuracy": 0.9809881751217059,
-      "eval_runtime": 10.9453,
-      "eval_samples_per_second": 6.121,
-      "eval_steps_per_second": 0.457,
-      "step": 1080
-    },
-    {
-      "epoch": 98.27,
-      "learning_rate": 3.368146214099217e-07,
-      "loss": 0.26,
-      "step": 1081
-    },
-    {
-      "epoch": 98.36,
-      "learning_rate": 3.342036553524804e-07,
-      "loss": 0.2765,
-      "step": 1082
-    },
-    {
-      "epoch": 98.45,
-      "learning_rate": 3.315926892950392e-07,
-      "loss": 0.2686,
-      "step": 1083
-    },
-    {
-      "epoch": 98.55,
-      "learning_rate": 3.2898172323759796e-07,
-      "loss": 0.2977,
-      "step": 1084
-    },
-    {
-      "epoch": 98.64,
-      "learning_rate": 3.263707571801567e-07,
-      "loss": 0.2444,
-      "step": 1085
-    },
-    {
-      "epoch": 98.73,
-      "learning_rate": 3.237597911227154e-07,
-      "loss": 0.262,
-      "step": 1086
-    },
-    {
-      "epoch": 98.82,
-      "learning_rate": 3.211488250652741e-07,
-      "loss": 0.2618,
-      "step": 1087
-    },
-    {
-      "epoch": 98.91,
-      "learning_rate": 3.185378590078329e-07,
-      "loss": 0.246,
-      "step": 1088
-    },
-    {
-      "epoch": 99.0,
-      "learning_rate": 3.1592689295039167e-07,
-      "loss": 0.285,
-      "step": 1089
-    },
-    {
-      "epoch": 99.09,
-      "learning_rate": 3.133159268929504e-07,
-      "loss": 0.266,
-      "step": 1090
-    },
-    {
-      "epoch": 99.09,
-      "eval_accuracy_safe": 0.6777942259281962,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9900612771743434,
-      "eval_iou_safe": 0.511777068191146,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9804459747457676,
-      "eval_loss": 0.24548684060573578,
-      "eval_mean_accuracy": 0.8339277515512697,
-      "eval_mean_iou": 0.4974076809789712,
-      "eval_overall_accuracy": 0.9808374091760436,
-      "eval_runtime": 11.7993,
-      "eval_samples_per_second": 5.678,
-      "eval_steps_per_second": 0.424,
-      "step": 1090
-    },
-    {
-      "epoch": 99.18,
-      "learning_rate": 3.1070496083550916e-07,
-      "loss": 0.2705,
-      "step": 1091
-    },
-    {
-      "epoch": 99.27,
-      "learning_rate": 3.080939947780679e-07,
-      "loss": 0.298,
-      "step": 1092
-    },
-    {
-      "epoch": 99.36,
-      "learning_rate": 3.054830287206266e-07,
-      "loss": 0.2351,
-      "step": 1093
-    },
-    {
-      "epoch": 99.45,
-      "learning_rate": 3.0287206266318537e-07,
-      "loss": 0.2721,
-      "step": 1094
-    },
-    {
-      "epoch": 99.55,
-      "learning_rate": 3.0026109660574415e-07,
-      "loss": 0.2686,
-      "step": 1095
-    },
-    {
-      "epoch": 99.64,
-      "learning_rate": 2.9765013054830287e-07,
-      "loss": 0.2837,
-      "step": 1096
-    },
-    {
-      "epoch": 99.73,
-      "learning_rate": 2.9503916449086164e-07,
-      "loss": 0.28,
-      "step": 1097
-    },
-    {
-      "epoch": 99.82,
-      "learning_rate": 2.924281984334204e-07,
-      "loss": 0.2622,
-      "step": 1098
-    },
-    {
-      "epoch": 99.91,
-      "learning_rate": 2.8981723237597913e-07,
-      "loss": 0.2638,
-      "step": 1099
-    },
-    {
-      "epoch": 100.0,
-      "learning_rate": 2.8720626631853785e-07,
-      "loss": 0.2682,
-      "step": 1100
-    },
-    {
-      "epoch": 100.0,
-      "eval_accuracy_safe": 0.6651053773886763,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9911382596240529,
-      "eval_iou_safe": 0.5156260740404243,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.981137303476917,
-      "eval_loss": 0.2358776330947876,
-      "eval_mean_accuracy": 0.8281218185063646,
-      "eval_mean_iou": 0.49892112583911374,
-      "eval_overall_accuracy": 0.9815077710507522,
-      "eval_runtime": 9.6315,
-      "eval_samples_per_second": 6.956,
-      "eval_steps_per_second": 0.519,
-      "step": 1100
-    },
-    {
-      "epoch": 100.09,
-      "learning_rate": 2.8459530026109663e-07,
-      "loss": 0.2801,
-      "step": 1101
-    },
-    {
-      "epoch": 100.18,
-      "learning_rate": 2.8198433420365535e-07,
-      "loss": 0.286,
-      "step": 1102
-    },
-    {
-      "epoch": 100.27,
-      "learning_rate": 2.793733681462141e-07,
-      "loss": 0.2534,
-      "step": 1103
-    },
-    {
-      "epoch": 100.36,
-      "learning_rate": 2.7676240208877284e-07,
-      "loss": 0.2624,
-      "step": 1104
-    },
-    {
-      "epoch": 100.45,
-      "learning_rate": 2.741514360313316e-07,
-      "loss": 0.2565,
-      "step": 1105
-    },
-    {
-      "epoch": 100.55,
-      "learning_rate": 2.7154046997389033e-07,
-      "loss": 0.2646,
-      "step": 1106
-    },
-    {
-      "epoch": 100.64,
-      "learning_rate": 2.6892950391644905e-07,
-      "loss": 0.2764,
-      "step": 1107
-    },
-    {
-      "epoch": 100.73,
-      "learning_rate": 2.6631853785900783e-07,
-      "loss": 0.3186,
-      "step": 1108
-    },
-    {
-      "epoch": 100.82,
-      "learning_rate": 2.637075718015666e-07,
-      "loss": 0.2657,
-      "step": 1109
-    },
-    {
-      "epoch": 100.91,
-      "learning_rate": 2.610966057441253e-07,
-      "loss": 0.2607,
-      "step": 1110
-    },
-    {
-      "epoch": 100.91,
-      "eval_accuracy_safe": 0.6738698000393214,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9905278698323235,
-      "eval_iou_safe": 0.5144442580542631,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9807920754923263,
-      "eval_loss": 0.23757004737854004,
-      "eval_mean_accuracy": 0.8321988349358225,
-      "eval_mean_iou": 0.4984121111821964,
-      "eval_overall_accuracy": 0.9811742981868004,
-      "eval_runtime": 10.0274,
-      "eval_samples_per_second": 6.682,
-      "eval_steps_per_second": 0.499,
-      "step": 1110
-    },
-    {
-      "epoch": 101.0,
-      "learning_rate": 2.584856396866841e-07,
-      "loss": 0.2811,
-      "step": 1111
-    },
-    {
-      "epoch": 101.09,
-      "learning_rate": 2.5587467362924287e-07,
-      "loss": 0.2518,
-      "step": 1112
-    },
-    {
-      "epoch": 101.18,
-      "learning_rate": 2.5326370757180154e-07,
-      "loss": 0.2703,
-      "step": 1113
-    },
-    {
-      "epoch": 101.27,
-      "learning_rate": 2.506527415143603e-07,
-      "loss": 0.2491,
-      "step": 1114
-    },
-    {
-      "epoch": 101.36,
-      "learning_rate": 2.480417754569191e-07,
-      "loss": 0.2838,
-      "step": 1115
-    },
-    {
-      "epoch": 101.45,
-      "learning_rate": 2.454308093994778e-07,
-      "loss": 0.2518,
-      "step": 1116
-    },
-    {
-      "epoch": 101.55,
-      "learning_rate": 2.428198433420366e-07,
-      "loss": 0.2552,
-      "step": 1117
-    },
-    {
-      "epoch": 101.64,
-      "learning_rate": 2.4020887728459535e-07,
-      "loss": 0.2887,
-      "step": 1118
-    },
-    {
-      "epoch": 101.73,
-      "learning_rate": 2.3759791122715404e-07,
-      "loss": 0.277,
-      "step": 1119
-    },
-    {
-      "epoch": 101.82,
-      "learning_rate": 2.3498694516971282e-07,
-      "loss": 0.2506,
-      "step": 1120
-    },
-    {
-      "epoch": 101.82,
-      "eval_accuracy_safe": 0.6726824491809977,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9906665041150856,
-      "eval_iou_safe": 0.5153487204477325,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9808942461066753,
-      "eval_loss": 0.23799440264701843,
-      "eval_mean_accuracy": 0.8316744766480417,
-      "eval_mean_iou": 0.4987476555181359,
-      "eval_overall_accuracy": 0.9812737649946071,
-      "eval_runtime": 10.407,
-      "eval_samples_per_second": 6.438,
-      "eval_steps_per_second": 0.48,
-      "step": 1120
-    },
-    {
-      "epoch": 101.91,
-      "learning_rate": 2.3237597911227154e-07,
-      "loss": 0.2496,
-      "step": 1121
-    },
-    {
-      "epoch": 102.0,
-      "learning_rate": 2.2976501305483028e-07,
-      "loss": 0.273,
-      "step": 1122
-    },
-    {
-      "epoch": 102.09,
-      "learning_rate": 2.2715404699738906e-07,
-      "loss": 0.2565,
-      "step": 1123
-    },
-    {
-      "epoch": 102.18,
-      "learning_rate": 2.2454308093994778e-07,
-      "loss": 0.2824,
-      "step": 1124
-    },
-    {
-      "epoch": 102.27,
-      "learning_rate": 2.2193211488250652e-07,
-      "loss": 0.2724,
-      "step": 1125
-    },
-    {
-      "epoch": 102.36,
-      "learning_rate": 2.193211488250653e-07,
-      "loss": 0.2681,
-      "step": 1126
-    },
-    {
-      "epoch": 102.45,
-      "learning_rate": 2.1671018276762402e-07,
-      "loss": 0.2424,
-      "step": 1127
-    },
-    {
-      "epoch": 102.55,
-      "learning_rate": 2.1409921671018276e-07,
-      "loss": 0.2529,
-      "step": 1128
-    },
-    {
-      "epoch": 102.64,
-      "learning_rate": 2.1148825065274154e-07,
-      "loss": 0.2577,
-      "step": 1129
-    },
-    {
-      "epoch": 102.73,
-      "learning_rate": 2.0887728459530026e-07,
-      "loss": 0.2729,
-      "step": 1130
-    },
-    {
-      "epoch": 102.73,
-      "eval_accuracy_safe": 0.6802209706207764,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9901609553996558,
-      "eval_iou_safe": 0.5145597517748594,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9806164706631543,
-      "eval_loss": 0.23748475313186646,
-      "eval_mean_accuracy": 0.8351909630102161,
-      "eval_mean_iou": 0.49839207414600456,
-      "eval_overall_accuracy": 0.981005825213532,
-      "eval_runtime": 10.7695,
-      "eval_samples_per_second": 6.221,
-      "eval_steps_per_second": 0.464,
-      "step": 1130
-    },
-    {
-      "epoch": 102.82,
-      "learning_rate": 2.06266318537859e-07,
-      "loss": 0.2593,
-      "step": 1131
-    },
-    {
-      "epoch": 102.91,
-      "learning_rate": 2.0365535248041778e-07,
-      "loss": 0.267,
-      "step": 1132
-    },
-    {
-      "epoch": 103.0,
-      "learning_rate": 2.010443864229765e-07,
-      "loss": 0.2543,
-      "step": 1133
-    },
-    {
-      "epoch": 103.09,
-      "learning_rate": 1.9843342036553527e-07,
-      "loss": 0.2673,
-      "step": 1134
-    },
-    {
-      "epoch": 103.18,
-      "learning_rate": 1.9582245430809402e-07,
-      "loss": 0.2496,
-      "step": 1135
-    },
-    {
-      "epoch": 103.27,
-      "learning_rate": 1.9321148825065274e-07,
-      "loss": 0.2757,
-      "step": 1136
-    },
-    {
-      "epoch": 103.36,
-      "learning_rate": 1.906005221932115e-07,
-      "loss": 0.2433,
-      "step": 1137
-    },
-    {
-      "epoch": 103.45,
-      "learning_rate": 1.8798955613577023e-07,
-      "loss": 0.2917,
-      "step": 1138
-    },
-    {
-      "epoch": 103.55,
-      "learning_rate": 1.8537859007832898e-07,
-      "loss": 0.2546,
-      "step": 1139
-    },
-    {
-      "epoch": 103.64,
-      "learning_rate": 1.8276762402088773e-07,
-      "loss": 0.2348,
-      "step": 1140
-    },
-    {
-      "epoch": 103.64,
-      "eval_accuracy_safe": 0.6730949379532076,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9907904712075427,
-      "eval_iou_safe": 0.5171906657661821,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.981029185783124,
-      "eval_loss": 0.23353296518325806,
-      "eval_mean_accuracy": 0.8319427045803751,
-      "eval_mean_iou": 0.49940661718310203,
-      "eval_overall_accuracy": 0.9814062545548624,
-      "eval_runtime": 10.7501,
-      "eval_samples_per_second": 6.232,
-      "eval_steps_per_second": 0.465,
-      "step": 1140
-    },
-    {
-      "epoch": 103.73,
-      "learning_rate": 1.8015665796344647e-07,
-      "loss": 0.2609,
-      "step": 1141
-    },
-    {
-      "epoch": 103.82,
-      "learning_rate": 1.7754569190600522e-07,
-      "loss": 0.2461,
-      "step": 1142
-    },
-    {
-      "epoch": 103.91,
-      "learning_rate": 1.7493472584856397e-07,
-      "loss": 0.3057,
-      "step": 1143
-    },
-    {
-      "epoch": 104.0,
-      "learning_rate": 1.7232375979112274e-07,
-      "loss": 0.2862,
-      "step": 1144
-    },
-    {
-      "epoch": 104.09,
-      "learning_rate": 1.6971279373368146e-07,
-      "loss": 0.2562,
-      "step": 1145
-    },
-    {
-      "epoch": 104.18,
-      "learning_rate": 1.671018276762402e-07,
-      "loss": 0.241,
-      "step": 1146
-    },
-    {
-      "epoch": 104.27,
-      "learning_rate": 1.6449086161879898e-07,
-      "loss": 0.2644,
-      "step": 1147
-    },
-    {
-      "epoch": 104.36,
-      "learning_rate": 1.618798955613577e-07,
-      "loss": 0.2752,
-      "step": 1148
-    },
-    {
-      "epoch": 104.45,
-      "learning_rate": 1.5926892950391645e-07,
-      "loss": 0.2514,
-      "step": 1149
-    },
-    {
-      "epoch": 104.55,
-      "learning_rate": 1.566579634464752e-07,
-      "loss": 0.2409,
-      "step": 1150
-    },
-    {
-      "epoch": 104.55,
-      "eval_accuracy_safe": 0.6781007012309127,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9904347625082679,
-      "eval_iou_safe": 0.5164347505684745,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.980824950444014,
-      "eval_loss": 0.2385135442018509,
-      "eval_mean_accuracy": 0.8342677318695904,
-      "eval_mean_iou": 0.4990865670041629,
-      "eval_overall_accuracy": 0.9812089151410914,
-      "eval_runtime": 9.9272,
-      "eval_samples_per_second": 6.749,
-      "eval_steps_per_second": 0.504,
-      "step": 1150
-    },
-    {
-      "epoch": 104.64,
-      "learning_rate": 1.5404699738903394e-07,
-      "loss": 0.2578,
-      "step": 1151
-    },
-    {
-      "epoch": 104.73,
-      "learning_rate": 1.5143603133159269e-07,
-      "loss": 0.2695,
-      "step": 1152
-    },
-    {
-      "epoch": 104.82,
-      "learning_rate": 1.4882506527415143e-07,
-      "loss": 0.2637,
-      "step": 1153
-    },
-    {
-      "epoch": 104.91,
-      "learning_rate": 1.462140992167102e-07,
-      "loss": 0.2437,
-      "step": 1154
-    },
-    {
-      "epoch": 105.0,
-      "learning_rate": 1.4360313315926893e-07,
-      "loss": 0.2994,
-      "step": 1155
-    },
-    {
-      "epoch": 105.09,
-      "learning_rate": 1.4099216710182767e-07,
-      "loss": 0.2443,
-      "step": 1156
-    },
-    {
-      "epoch": 105.18,
-      "learning_rate": 1.3838120104438642e-07,
-      "loss": 0.2707,
-      "step": 1157
-    },
-    {
-      "epoch": 105.27,
-      "learning_rate": 1.3577023498694517e-07,
-      "loss": 0.2544,
-      "step": 1158
-    },
-    {
-      "epoch": 105.36,
-      "learning_rate": 1.3315926892950391e-07,
-      "loss": 0.2371,
-      "step": 1159
-    },
-    {
-      "epoch": 105.45,
-      "learning_rate": 1.3054830287206266e-07,
-      "loss": 0.2737,
-      "step": 1160
-    },
-    {
-      "epoch": 105.45,
-      "eval_accuracy_safe": 0.677416432473275,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9905725754283728,
-      "eval_iou_safe": 0.517675149619011,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9809411939821061,
-      "eval_loss": 0.23569297790527344,
-      "eval_mean_accuracy": 0.8339945039508239,
-      "eval_mean_iou": 0.4995387812003724,
-      "eval_overall_accuracy": 0.9813224450865788,
-      "eval_runtime": 10.724,
-      "eval_samples_per_second": 6.248,
-      "eval_steps_per_second": 0.466,
-      "step": 1160
-    },
-    {
-      "epoch": 105.55,
-      "learning_rate": 1.2793733681462143e-07,
-      "loss": 0.272,
-      "step": 1161
-    },
-    {
-      "epoch": 105.64,
-      "learning_rate": 1.2532637075718015e-07,
-      "loss": 0.2526,
-      "step": 1162
-    },
-    {
-      "epoch": 105.73,
-      "learning_rate": 1.227154046997389e-07,
-      "loss": 0.2923,
-      "step": 1163
-    },
-    {
-      "epoch": 105.82,
-      "learning_rate": 1.2010443864229767e-07,
-      "loss": 0.2535,
-      "step": 1164
-    },
-    {
-      "epoch": 105.91,
-      "learning_rate": 1.1749347258485641e-07,
-      "loss": 0.257,
-      "step": 1165
-    },
-    {
-      "epoch": 106.0,
-      "learning_rate": 1.1488250652741514e-07,
-      "loss": 0.2114,
-      "step": 1166
-    },
-    {
-      "epoch": 106.09,
-      "learning_rate": 1.1227154046997389e-07,
-      "loss": 0.2743,
-      "step": 1167
-    },
-    {
-      "epoch": 106.18,
-      "learning_rate": 1.0966057441253265e-07,
-      "loss": 0.2485,
-      "step": 1168
-    },
-    {
-      "epoch": 106.27,
-      "learning_rate": 1.0704960835509138e-07,
-      "loss": 0.2781,
-      "step": 1169
-    },
-    {
-      "epoch": 106.36,
-      "learning_rate": 1.0443864229765013e-07,
-      "loss": 0.2857,
-      "step": 1170
-    },
-    {
-      "epoch": 106.36,
-      "eval_accuracy_safe": 0.6658706018866543,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9914503187649804,
-      "eval_iou_safe": 0.5202995990686102,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9814688430393181,
-      "eval_loss": 0.23036494851112366,
-      "eval_mean_accuracy": 0.8286604603258173,
-      "eval_mean_iou": 0.5005894807026428,
-      "eval_overall_accuracy": 0.981833215969712,
-      "eval_runtime": 10.0808,
-      "eval_samples_per_second": 6.646,
-      "eval_steps_per_second": 0.496,
-      "step": 1170
-    },
-    {
-      "epoch": 106.45,
-      "learning_rate": 1.0182767624020889e-07,
-      "loss": 0.2707,
-      "step": 1171
-    },
-    {
-      "epoch": 106.55,
-      "learning_rate": 9.921671018276764e-08,
-      "loss": 0.2403,
-      "step": 1172
-    },
-    {
-      "epoch": 106.64,
-      "learning_rate": 9.660574412532637e-08,
-      "loss": 0.2787,
-      "step": 1173
-    },
-    {
-      "epoch": 106.73,
-      "learning_rate": 9.399477806788512e-08,
-      "loss": 0.2574,
-      "step": 1174
-    },
-    {
-      "epoch": 106.82,
-      "learning_rate": 9.138381201044386e-08,
-      "loss": 0.226,
-      "step": 1175
-    },
-    {
-      "epoch": 106.91,
-      "learning_rate": 8.877284595300261e-08,
-      "loss": 0.2629,
-      "step": 1176
-    },
-    {
-      "epoch": 107.0,
-      "learning_rate": 8.616187989556137e-08,
-      "loss": 0.2674,
-      "step": 1177
-    },
-    {
-      "epoch": 107.09,
-      "learning_rate": 8.35509138381201e-08,
-      "loss": 0.2729,
-      "step": 1178
-    },
-    {
-      "epoch": 107.18,
-      "learning_rate": 8.093994778067885e-08,
-      "loss": 0.2824,
-      "step": 1179
-    },
-    {
-      "epoch": 107.27,
-      "learning_rate": 7.83289817232376e-08,
-      "loss": 0.2503,
-      "step": 1180
-    },
-    {
-      "epoch": 107.27,
-      "eval_accuracy_safe": 0.682225588953011,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9902289524939093,
-      "eval_iou_safe": 0.5171544669118453,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9807430760293058,
-      "eval_loss": 0.23972667753696442,
-      "eval_mean_accuracy": 0.8362272707234601,
-      "eval_mean_iou": 0.49929918098038373,
-      "eval_overall_accuracy": 0.9811310269939366,
-      "eval_runtime": 10.3086,
-      "eval_samples_per_second": 6.499,
-      "eval_steps_per_second": 0.485,
-      "step": 1180
-    },
-    {
-      "epoch": 107.36,
-      "learning_rate": 7.571801566579634e-08,
-      "loss": 0.2528,
-      "step": 1181
-    },
-    {
-      "epoch": 107.45,
-      "learning_rate": 7.31070496083551e-08,
-      "loss": 0.2749,
-      "step": 1182
-    },
-    {
-      "epoch": 107.55,
-      "learning_rate": 7.049608355091384e-08,
-      "loss": 0.2361,
-      "step": 1183
-    },
-    {
-      "epoch": 107.64,
-      "learning_rate": 6.788511749347258e-08,
-      "loss": 0.2599,
-      "step": 1184
-    },
-    {
-      "epoch": 107.73,
-      "learning_rate": 6.527415143603133e-08,
-      "loss": 0.2649,
-      "step": 1185
-    },
-    {
-      "epoch": 107.82,
-      "learning_rate": 6.266318537859008e-08,
-      "loss": 0.2451,
-      "step": 1186
-    },
-    {
-      "epoch": 107.91,
-      "learning_rate": 6.005221932114884e-08,
-      "loss": 0.265,
-      "step": 1187
-    },
-    {
-      "epoch": 108.0,
-      "learning_rate": 5.744125326370757e-08,
-      "loss": 0.2867,
-      "step": 1188
-    },
-    {
-      "epoch": 108.09,
-      "learning_rate": 5.4830287206266324e-08,
-      "loss": 0.2771,
-      "step": 1189
-    },
-    {
-      "epoch": 108.18,
-      "learning_rate": 5.2219321148825064e-08,
-      "loss": 0.2524,
-      "step": 1190
-    },
-    {
-      "epoch": 108.18,
-      "eval_accuracy_safe": 0.6740837544959348,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9909394311922795,
-      "eval_iou_safe": 0.5199116919028611,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9812059186949175,
-      "eval_loss": 0.22940021753311157,
-      "eval_mean_accuracy": 0.8325115928441071,
-      "eval_mean_iou": 0.5003725368659262,
-      "eval_overall_accuracy": 0.9815800225556787,
-      "eval_runtime": 9.8553,
-      "eval_samples_per_second": 6.798,
-      "eval_steps_per_second": 0.507,
-      "step": 1190
-    },
-    {
-      "epoch": 108.27,
-      "learning_rate": 4.960835509138382e-08,
-      "loss": 0.2572,
-      "step": 1191
-    },
-    {
-      "epoch": 108.36,
-      "learning_rate": 4.699738903394256e-08,
-      "loss": 0.2375,
-      "step": 1192
-    },
-    {
-      "epoch": 108.45,
-      "learning_rate": 4.4386422976501305e-08,
-      "loss": 0.2592,
-      "step": 1193
-    },
-    {
-      "epoch": 108.55,
-      "learning_rate": 4.177545691906005e-08,
-      "loss": 0.2463,
-      "step": 1194
-    },
-    {
-      "epoch": 108.64,
-      "learning_rate": 3.91644908616188e-08,
-      "loss": 0.2377,
-      "step": 1195
-    },
-    {
-      "epoch": 108.73,
-      "learning_rate": 3.655352480417755e-08,
-      "loss": 0.2416,
-      "step": 1196
-    },
-    {
-      "epoch": 108.82,
-      "learning_rate": 3.394255874673629e-08,
-      "loss": 0.2558,
-      "step": 1197
-    },
-    {
-      "epoch": 108.91,
-      "learning_rate": 3.133159268929504e-08,
-      "loss": 0.2902,
-      "step": 1198
-    },
-    {
-      "epoch": 109.0,
-      "learning_rate": 2.8720626631853785e-08,
-      "loss": 0.2877,
-      "step": 1199
-    },
-    {
-      "epoch": 109.09,
-      "learning_rate": 2.6109660574412532e-08,
-      "loss": 0.2722,
-      "step": 1200
-    },
-    {
-      "epoch": 109.09,
-      "eval_accuracy_safe": 0.6927633278206329,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9893742073117,
-      "eval_iou_safe": 0.5144704521640547,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9802079010876331,
-      "eval_loss": 0.2461329847574234,
-      "eval_mean_accuracy": 0.8410687675661664,
-      "eval_mean_iou": 0.49822611775056264,
-      "eval_overall_accuracy": 0.9806127975236124,
-      "eval_runtime": 10.7536,
-      "eval_samples_per_second": 6.23,
-      "eval_steps_per_second": 0.465,
-      "step": 1200
-    },
-    {
-      "epoch": 109.18,
-      "learning_rate": 5.058910162002946e-07,
-      "loss": 0.2403,
-      "step": 1201
-    },
-    {
-      "epoch": 109.27,
-      "learning_rate": 5.03681885125184e-07,
-      "loss": 0.2633,
-      "step": 1202
-    },
-    {
-      "epoch": 109.36,
-      "learning_rate": 5.014727540500737e-07,
-      "loss": 0.2314,
-      "step": 1203
-    },
-    {
-      "epoch": 109.45,
-      "learning_rate": 4.992636229749632e-07,
-      "loss": 0.2628,
-      "step": 1204
-    },
-    {
-      "epoch": 109.55,
-      "learning_rate": 4.970544918998527e-07,
-      "loss": 0.2858,
-      "step": 1205
-    },
-    {
-      "epoch": 109.64,
-      "learning_rate": 4.948453608247422e-07,
-      "loss": 0.2463,
-      "step": 1206
-    },
-    {
-      "epoch": 109.73,
-      "learning_rate": 4.926362297496319e-07,
-      "loss": 0.2232,
-      "step": 1207
-    },
-    {
-      "epoch": 109.82,
-      "learning_rate": 4.904270986745214e-07,
-      "loss": 0.2384,
-      "step": 1208
-    },
-    {
-      "epoch": 109.91,
-      "learning_rate": 4.882179675994109e-07,
-      "loss": 0.2542,
-      "step": 1209
-    },
-    {
-      "epoch": 110.0,
-      "learning_rate": 4.860088365243004e-07,
-      "loss": 0.2761,
-      "step": 1210
-    },
-    {
-      "epoch": 110.0,
-      "eval_accuracy_safe": 0.6981218268241063,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.988942346560362,
-      "eval_iou_safe": 0.5136464391621959,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9799386156578034,
-      "eval_loss": 0.2529892325401306,
-      "eval_mean_accuracy": 0.8435320866922341,
-      "eval_mean_iou": 0.4978616849399997,
-      "eval_overall_accuracy": 0.9803519747150478,
-      "eval_runtime": 11.8704,
-      "eval_samples_per_second": 5.644,
-      "eval_steps_per_second": 0.421,
-      "step": 1210
-    },
-    {
-      "epoch": 110.09,
-      "learning_rate": 4.837997054491901e-07,
-      "loss": 0.245,
-      "step": 1211
-    },
-    {
-      "epoch": 110.18,
-      "learning_rate": 4.815905743740796e-07,
-      "loss": 0.269,
-      "step": 1212
-    },
-    {
-      "epoch": 110.27,
-      "learning_rate": 4.79381443298969e-07,
-      "loss": 0.2476,
-      "step": 1213
-    },
-    {
-      "epoch": 110.36,
-      "learning_rate": 4.771723122238586e-07,
-      "loss": 0.2507,
-      "step": 1214
-    },
-    {
-      "epoch": 110.45,
-      "learning_rate": 4.7496318114874815e-07,
-      "loss": 0.2477,
-      "step": 1215
-    },
-    {
-      "epoch": 110.55,
-      "learning_rate": 4.727540500736377e-07,
-      "loss": 0.2485,
-      "step": 1216
-    },
-    {
-      "epoch": 110.64,
-      "learning_rate": 4.7054491899852725e-07,
-      "loss": 0.2638,
-      "step": 1217
-    },
-    {
-      "epoch": 110.73,
-      "learning_rate": 4.683357879234168e-07,
-      "loss": 0.2381,
-      "step": 1218
-    },
-    {
-      "epoch": 110.82,
-      "learning_rate": 4.6612665684830635e-07,
-      "loss": 0.2491,
-      "step": 1219
-    },
-    {
-      "epoch": 110.91,
-      "learning_rate": 4.6391752577319593e-07,
-      "loss": 0.2595,
-      "step": 1220
-    },
-    {
-      "epoch": 110.91,
-      "eval_accuracy_safe": 0.6890586389412531,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9900403324265881,
-      "eval_iou_safe": 0.5196969785076293,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9807582876854402,
-      "eval_loss": 0.23863691091537476,
-      "eval_mean_accuracy": 0.8395494856839205,
-      "eval_mean_iou": 0.5001517553976899,
-      "eval_overall_accuracy": 0.9811498158013643,
-      "eval_runtime": 11.7338,
-      "eval_samples_per_second": 5.71,
-      "eval_steps_per_second": 0.426,
-      "step": 1220
-    },
-    {
-      "epoch": 111.0,
-      "learning_rate": 4.6170839469808545e-07,
-      "loss": 0.2617,
-      "step": 1221
-    },
-    {
-      "epoch": 111.09,
-      "learning_rate": 4.5949926362297503e-07,
-      "loss": 0.2358,
-      "step": 1222
-    },
-    {
-      "epoch": 111.18,
-      "learning_rate": 4.5729013254786455e-07,
-      "loss": 0.251,
-      "step": 1223
-    },
-    {
-      "epoch": 111.27,
-      "learning_rate": 4.55081001472754e-07,
-      "loss": 0.2633,
-      "step": 1224
-    },
-    {
-      "epoch": 111.36,
-      "learning_rate": 4.528718703976436e-07,
-      "loss": 0.2447,
-      "step": 1225
-    },
-    {
-      "epoch": 111.45,
-      "learning_rate": 4.506627393225331e-07,
-      "loss": 0.2509,
-      "step": 1226
-    },
-    {
-      "epoch": 111.55,
-      "learning_rate": 4.484536082474227e-07,
-      "loss": 0.2705,
-      "step": 1227
-    },
-    {
-      "epoch": 111.64,
-      "learning_rate": 4.462444771723122e-07,
-      "loss": 0.2494,
-      "step": 1228
-    },
-    {
-      "epoch": 111.73,
-      "learning_rate": 4.440353460972018e-07,
-      "loss": 0.2625,
-      "step": 1229
-    },
-    {
-      "epoch": 111.82,
-      "learning_rate": 4.418262150220913e-07,
-      "loss": 0.2877,
-      "step": 1230
-    },
-    {
-      "epoch": 111.82,
-      "eval_accuracy_safe": 0.6779060219505707,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9911065784929943,
-      "eval_iou_safe": 0.5250508331865317,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.981484545675428,
-      "eval_loss": 0.22597333788871765,
-      "eval_mean_accuracy": 0.8345063002217825,
-      "eval_mean_iou": 0.5021784596206532,
-      "eval_overall_accuracy": 0.9818551362450443,
-      "eval_runtime": 10.6701,
-      "eval_samples_per_second": 6.279,
-      "eval_steps_per_second": 0.469,
-      "step": 1230
-    },
-    {
-      "epoch": 111.91,
-      "learning_rate": 4.396170839469809e-07,
-      "loss": 0.2425,
-      "step": 1231
-    },
-    {
-      "epoch": 112.0,
-      "learning_rate": 4.374079528718704e-07,
-      "loss": 0.274,
-      "step": 1232
-    },
-    {
-      "epoch": 112.09,
-      "learning_rate": 4.3519882179676e-07,
-      "loss": 0.2509,
-      "step": 1233
-    },
-    {
-      "epoch": 112.18,
-      "learning_rate": 4.3298969072164947e-07,
-      "loss": 0.2546,
-      "step": 1234
-    },
-    {
-      "epoch": 112.27,
-      "learning_rate": 4.30780559646539e-07,
-      "loss": 0.2595,
-      "step": 1235
-    },
-    {
-      "epoch": 112.36,
-      "learning_rate": 4.2857142857142857e-07,
-      "loss": 0.2535,
-      "step": 1236
-    },
-    {
-      "epoch": 112.45,
-      "learning_rate": 4.263622974963181e-07,
-      "loss": 0.2524,
-      "step": 1237
-    },
-    {
-      "epoch": 112.55,
-      "learning_rate": 4.2415316642120767e-07,
-      "loss": 0.2372,
-      "step": 1238
-    },
-    {
-      "epoch": 112.64,
-      "learning_rate": 4.219440353460972e-07,
-      "loss": 0.2495,
-      "step": 1239
-    },
-    {
-      "epoch": 112.73,
-      "learning_rate": 4.1973490427098677e-07,
-      "loss": 0.2307,
-      "step": 1240
-    },
-    {
-      "epoch": 112.73,
-      "eval_accuracy_safe": 0.6717495306494577,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_accuracy_unsafe": 0.9916446297021398,
-      "eval_iou_safe": 0.5274904076829351,
-      "eval_iou_unlabeled": 0.0,
-      "eval_iou_unsafe": 0.9818352893830999,
-      "eval_loss": 0.22187493741512299,
-      "eval_mean_accuracy": 0.8316970801757988,
-      "eval_mean_iou": 0.5031085656886783,
-      "eval_overall_accuracy": 0.9821954414026061,
-      "eval_runtime": 10.6767,
-      "eval_samples_per_second": 6.275,
-      "eval_steps_per_second": 0.468,
-      "step": 1240
-    }
-  ],
-  "max_steps": 1430,
-  "num_train_epochs": 130,
-  "total_flos": 6.932825583969632e+18,
-  "trial_name": null,
-  "trial_params": null
-}