diff --git "a/trainer_state.json" "b/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/trainer_state.json"
@@ -0,0 +1,12736 @@
+{
+  "best_metric": 0.9547889828681946,
+  "best_model_checkpoint": "segformer-finetuned-segments-cmp-facade-outputs-baseline-0.0006/checkpoint-200",
+  "epoch": 50.0,
+  "global_step": 1600,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.03,
+      "learning_rate": 0.000599625,
+      "loss": 2.5686,
+      "step": 1
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 0.00059925,
+      "loss": 2.3903,
+      "step": 2
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 0.000598875,
+      "loss": 2.1973,
+      "step": 3
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 0.0005985,
+      "loss": 1.9997,
+      "step": 4
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 0.0005981249999999999,
+      "loss": 1.9846,
+      "step": 5
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 0.00059775,
+      "loss": 1.8746,
+      "step": 6
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 0.0005973749999999999,
+      "loss": 1.8332,
+      "step": 7
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 0.000597,
+      "loss": 1.8059,
+      "step": 8
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 0.0005966249999999999,
+      "loss": 1.6617,
+      "step": 9
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 0.00059625,
+      "loss": 1.6067,
+      "step": 10
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 0.0005958749999999999,
+      "loss": 1.5621,
+      "step": 11
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 0.0005955,
+      "loss": 1.4863,
+      "step": 12
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 0.000595125,
+      "loss": 1.4005,
+      "step": 13
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 0.0005947499999999999,
+      "loss": 1.4485,
+      "step": 14
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 0.000594375,
+      "loss": 1.3887,
+      "step": 15
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 0.0005939999999999999,
+      "loss": 1.5758,
+      "step": 16
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 0.000593625,
+      "loss": 1.3851,
+      "step": 17
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 0.0005932499999999999,
+      "loss": 1.3992,
+      "step": 18
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 0.000592875,
+      "loss": 1.3758,
+      "step": 19
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 0.0005924999999999999,
+      "loss": 1.3587,
+      "step": 20
+    },
+    {
+      "epoch": 0.62,
+      "eval_loss": 1.6255974769592285,
+      "eval_mean_accuracy": 0.3141481516866101,
+      "eval_mean_iou": 0.18315948364422277,
+      "eval_overall_accuracy": 0.5485008306670607,
+      "eval_per_category_accuracy": [
+        0.6488692748813196,
+        0.5616440349622522,
+        0.7688431729015686,
+        0.0,
+        0.39402196484247587,
+        0.003442343915017475,
+        0.2774663897762572,
+        6.214434416823545e-05,
+        0.0003349511548051072,
+        0.7398100827126451,
+        0.0,
+        0.37528346074881097
+      ],
+      "eval_per_category_iou": [
+        0.4772592315619872,
+        0.44317885541602603,
+        0.45610220048891476,
+        0.0,
+        0.16838871775158482,
+        0.00343126703650311,
+        0.1752659085145423,
+        6.214374491064158e-05,
+        0.0003344066828185833,
+        0.3009429801158851,
+        0.0,
+        0.172948092417501
+      ],
+      "eval_runtime": 121.8515,
+      "eval_samples_per_second": 0.936,
+      "eval_steps_per_second": 0.082,
+      "step": 20
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 0.0005921249999999999,
+      "loss": 1.3857,
+      "step": 21
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 0.0005917499999999999,
+      "loss": 1.2941,
+      "step": 22
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 0.0005913749999999999,
+      "loss": 1.2736,
+      "step": 23
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 0.0005909999999999999,
+      "loss": 1.2626,
+      "step": 24
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 0.0005906249999999999,
+      "loss": 1.2018,
+      "step": 25
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 0.00059025,
+      "loss": 1.2629,
+      "step": 26
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 0.000589875,
+      "loss": 1.1944,
+      "step": 27
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 0.0005895,
+      "loss": 1.1597,
+      "step": 28
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 0.000589125,
+      "loss": 1.1822,
+      "step": 29
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 0.00058875,
+      "loss": 1.1729,
+      "step": 30
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 0.0005883749999999999,
+      "loss": 1.0742,
+      "step": 31
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 0.000588,
+      "loss": 1.3135,
+      "step": 32
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 0.0005876249999999999,
+      "loss": 1.2218,
+      "step": 33
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 0.00058725,
+      "loss": 1.2487,
+      "step": 34
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 0.0005868749999999999,
+      "loss": 0.9779,
+      "step": 35
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 0.0005865,
+      "loss": 1.0848,
+      "step": 36
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 0.000586125,
+      "loss": 1.0201,
+      "step": 37
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 0.0005857499999999999,
+      "loss": 1.0673,
+      "step": 38
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 0.000585375,
+      "loss": 1.1126,
+      "step": 39
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 0.0005849999999999999,
+      "loss": 1.1272,
+      "step": 40
+    },
+    {
+      "epoch": 1.25,
+      "eval_loss": 1.1469159126281738,
+      "eval_mean_accuracy": 0.33231376608612845,
+      "eval_mean_iou": 0.23972121192053034,
+      "eval_overall_accuracy": 0.6103959334524054,
+      "eval_per_category_accuracy": [
+        0.7301639814634119,
+        0.725317008517752,
+        0.6197514067158824,
+        0.0,
+        0.41723881549491804,
+        0.32015068647269534,
+        0.43180497804452184,
+        0.00046179676442257726,
+        0.005066413493873278,
+        0.5026867658203124,
+        0.0,
+        0.23512334024575196
+      ],
+      "eval_per_category_iou": [
+        0.5087072250192229,
+        0.4908826565456839,
+        0.4917530621343207,
+        0.0,
+        0.2615037685634764,
+        0.2381227460211199,
+        0.32305258576467477,
+        0.00046177400501630137,
+        0.0050433565260017705,
+        0.36085494098019893,
+        0.0,
+        0.1962724274866488
+      ],
+      "eval_runtime": 120.0026,
+      "eval_samples_per_second": 0.95,
+      "eval_steps_per_second": 0.083,
+      "step": 40
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 0.000584625,
+      "loss": 1.0215,
+      "step": 41
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 0.0005842499999999999,
+      "loss": 1.0141,
+      "step": 42
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 0.000583875,
+      "loss": 1.1439,
+      "step": 43
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 0.0005834999999999999,
+      "loss": 1.1585,
+      "step": 44
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 0.000583125,
+      "loss": 0.9521,
+      "step": 45
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 0.0005827499999999999,
+      "loss": 1.046,
+      "step": 46
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 0.0005823749999999999,
+      "loss": 0.959,
+      "step": 47
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 0.0005819999999999999,
+      "loss": 1.0224,
+      "step": 48
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 0.0005816249999999999,
+      "loss": 1.0171,
+      "step": 49
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 0.00058125,
+      "loss": 0.8845,
+      "step": 50
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 0.000580875,
+      "loss": 1.0473,
+      "step": 51
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 0.0005805,
+      "loss": 0.8508,
+      "step": 52
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 0.000580125,
+      "loss": 0.9726,
+      "step": 53
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 0.00057975,
+      "loss": 1.0182,
+      "step": 54
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 0.0005793749999999999,
+      "loss": 1.0579,
+      "step": 55
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 0.000579,
+      "loss": 0.9741,
+      "step": 56
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 0.0005786249999999999,
+      "loss": 1.0505,
+      "step": 57
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 0.00057825,
+      "loss": 0.9772,
+      "step": 58
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 0.0005778749999999999,
+      "loss": 0.8138,
+      "step": 59
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 0.0005775,
+      "loss": 1.0172,
+      "step": 60
+    },
+    {
+      "epoch": 1.88,
+      "eval_loss": 1.1332030296325684,
+      "eval_mean_accuracy": 0.3450738777433666,
+      "eval_mean_iou": 0.24995460216123477,
+      "eval_overall_accuracy": 0.6090325472647684,
+      "eval_per_category_accuracy": [
+        0.7712100807932455,
+        0.7139090190445223,
+        0.4965342527006462,
+        0.0,
+        0.36147785586509107,
+        0.4176468666867485,
+        0.3385701622768831,
+        0.0021118362475102083,
+        0.05396040921714595,
+        0.5883826780663631,
+        0.0,
+        0.39708337202224353
+      ],
+      "eval_per_category_iou": [
+        0.5314603901384262,
+        0.48318539289384677,
+        0.4269313528486642,
+        0.0,
+        0.2673354998494226,
+        0.29045614876418957,
+        0.30780775407728883,
+        0.0021117366918267967,
+        0.05081000242290565,
+        0.3810971242949915,
+        0.0,
+        0.2582598239532546
+      ],
+      "eval_runtime": 121.0254,
+      "eval_samples_per_second": 0.942,
+      "eval_steps_per_second": 0.083,
+      "step": 60
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 0.000577125,
+      "loss": 0.9949,
+      "step": 61
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 0.00057675,
+      "loss": 1.0874,
+      "step": 62
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 0.000576375,
+      "loss": 0.9226,
+      "step": 63
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 0.0005759999999999999,
+      "loss": 1.0359,
+      "step": 64
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 0.000575625,
+      "loss": 1.0029,
+      "step": 65
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 0.0005752499999999999,
+      "loss": 0.9346,
+      "step": 66
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 0.000574875,
+      "loss": 0.8542,
+      "step": 67
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 0.0005744999999999999,
+      "loss": 0.8286,
+      "step": 68
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 0.000574125,
+      "loss": 0.8975,
+      "step": 69
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 0.0005737499999999999,
+      "loss": 0.8509,
+      "step": 70
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 0.0005733749999999999,
+      "loss": 1.1385,
+      "step": 71
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 0.0005729999999999999,
+      "loss": 0.9216,
+      "step": 72
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 0.0005726249999999999,
+      "loss": 0.8162,
+      "step": 73
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 0.00057225,
+      "loss": 0.8909,
+      "step": 74
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 0.0005718749999999999,
+      "loss": 0.8444,
+      "step": 75
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 0.0005715,
+      "loss": 0.8338,
+      "step": 76
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 0.000571125,
+      "loss": 0.9097,
+      "step": 77
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 0.00057075,
+      "loss": 0.8778,
+      "step": 78
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 0.000570375,
+      "loss": 0.826,
+      "step": 79
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 0.00057,
+      "loss": 0.8466,
+      "step": 80
+    },
+    {
+      "epoch": 2.5,
+      "eval_loss": 1.0760297775268555,
+      "eval_mean_accuracy": 0.3748513935505046,
+      "eval_mean_iou": 0.2710584212101326,
+      "eval_overall_accuracy": 0.629409488878752,
+      "eval_per_category_accuracy": [
+        0.5097210774762599,
+        0.8349397933225571,
+        0.7631626631252534,
+        0.01862582494222802,
+        0.5251667824807995,
+        0.4123753806176258,
+        0.5182476867632634,
+        0.0734942586126168,
+        0.21476358215841637,
+        0.4348659433460887,
+        6.121580714572117e-05,
+        0.1927925139538003
+      ],
+      "eval_per_category_iou": [
+        0.4536359079066772,
+        0.5416200722229393,
+        0.5074160392719198,
+        0.018448954341587477,
+        0.3023435103552763,
+        0.2943501851580806,
+        0.37784273527034967,
+        0.07253092131072308,
+        0.1504312441054501,
+        0.36153117043491534,
+        6.11807892321811e-05,
+        0.1724891333544405
+      ],
+      "eval_runtime": 121.6607,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 80
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 0.0005696249999999999,
+      "loss": 0.8429,
+      "step": 81
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 0.00056925,
+      "loss": 0.9725,
+      "step": 82
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 0.0005688749999999999,
+      "loss": 0.8059,
+      "step": 83
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 0.0005685,
+      "loss": 0.8261,
+      "step": 84
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 0.0005681249999999999,
+      "loss": 0.9101,
+      "step": 85
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 0.00056775,
+      "loss": 0.8387,
+      "step": 86
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 0.000567375,
+      "loss": 0.8772,
+      "step": 87
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 0.0005669999999999999,
+      "loss": 0.8313,
+      "step": 88
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 0.000566625,
+      "loss": 0.9802,
+      "step": 89
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 0.0005662499999999999,
+      "loss": 0.7859,
+      "step": 90
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 0.000565875,
+      "loss": 0.8676,
+      "step": 91
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 0.0005654999999999999,
+      "loss": 0.8232,
+      "step": 92
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 0.000565125,
+      "loss": 1.058,
+      "step": 93
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 0.0005647499999999999,
+      "loss": 0.8119,
+      "step": 94
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 0.000564375,
+      "loss": 0.8982,
+      "step": 95
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 0.0005639999999999999,
+      "loss": 0.8573,
+      "step": 96
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 0.0005636249999999999,
+      "loss": 0.9197,
+      "step": 97
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 0.00056325,
+      "loss": 0.7869,
+      "step": 98
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 0.0005628749999999999,
+      "loss": 0.8251,
+      "step": 99
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 0.0005625,
+      "loss": 0.783,
+      "step": 100
+    },
+    {
+      "epoch": 3.12,
+      "eval_loss": 1.0092662572860718,
+      "eval_mean_accuracy": 0.4518619118027695,
+      "eval_mean_iou": 0.33203451682727503,
+      "eval_overall_accuracy": 0.6477740438360917,
+      "eval_per_category_accuracy": [
+        0.5939408516675014,
+        0.7672641764327048,
+        0.7242871112894247,
+        0.10601938917842346,
+        0.43050848144373965,
+        0.378661459902228,
+        0.6076991383124275,
+        0.3051908742102043,
+        0.3373557048170856,
+        0.5822127392807093,
+        0.09827891758209804,
+        0.4909240975166878
+      ],
+      "eval_per_category_iou": [
+        0.5037467823030373,
+        0.538253245468685,
+        0.54696085927042,
+        0.0946461649020859,
+        0.32235906973357464,
+        0.2939849677023971,
+        0.4017418411370106,
+        0.2796837089281629,
+        0.21670437900040468,
+        0.38832214494282197,
+        0.07600716776305667,
+        0.3220038707756439
+      ],
+      "eval_runtime": 121.7376,
+      "eval_samples_per_second": 0.936,
+      "eval_steps_per_second": 0.082,
+      "step": 100
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 0.0005621249999999999,
+      "loss": 0.8053,
+      "step": 101
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 0.00056175,
+      "loss": 0.8289,
+      "step": 102
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 0.000561375,
+      "loss": 0.8684,
+      "step": 103
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 0.000561,
+      "loss": 0.824,
+      "step": 104
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 0.0005606249999999999,
+      "loss": 0.7919,
+      "step": 105
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 0.00056025,
+      "loss": 0.8915,
+      "step": 106
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 0.0005598749999999999,
+      "loss": 0.7168,
+      "step": 107
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 0.0005595,
+      "loss": 0.7627,
+      "step": 108
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 0.0005591249999999999,
+      "loss": 0.713,
+      "step": 109
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 0.00055875,
+      "loss": 0.7279,
+      "step": 110
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 0.000558375,
+      "loss": 0.7056,
+      "step": 111
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 0.000558,
+      "loss": 0.7856,
+      "step": 112
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 0.000557625,
+      "loss": 0.7288,
+      "step": 113
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 0.0005572499999999999,
+      "loss": 0.8074,
+      "step": 114
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 0.000556875,
+      "loss": 0.7689,
+      "step": 115
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 0.0005564999999999999,
+      "loss": 0.826,
+      "step": 116
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 0.000556125,
+      "loss": 0.731,
+      "step": 117
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 0.0005557499999999999,
+      "loss": 0.776,
+      "step": 118
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 0.000555375,
+      "loss": 0.7384,
+      "step": 119
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 0.0005549999999999999,
+      "loss": 0.6941,
+      "step": 120
+    },
+    {
+      "epoch": 3.75,
+      "eval_loss": 1.0103007555007935,
+      "eval_mean_accuracy": 0.4756915065746894,
+      "eval_mean_iou": 0.3391343323864871,
+      "eval_overall_accuracy": 0.6493041055244312,
+      "eval_per_category_accuracy": [
+        0.7335339765251577,
+        0.6849779444566702,
+        0.7191242307256597,
+        0.21663971797558657,
+        0.4872276932863594,
+        0.40480294985464854,
+        0.7152266633498596,
+        0.41164092140776226,
+        0.37097947703255457,
+        0.5582396846498912,
+        0.052777208130683505,
+        0.35312761150143923
+      ],
+      "eval_per_category_iou": [
+        0.5505120083172355,
+        0.5221275640424796,
+        0.5487694248053927,
+        0.15022159345180522,
+        0.3363354250080763,
+        0.3098512123030424,
+        0.37734158772772125,
+        0.3371687893170074,
+        0.1947684270226897,
+        0.3947398604106545,
+        0.04745601312244219,
+        0.30032008310929853
+      ],
+      "eval_runtime": 120.5562,
+      "eval_samples_per_second": 0.946,
+      "eval_steps_per_second": 0.083,
+      "step": 120
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 0.0005546249999999999,
+      "loss": 0.7105,
+      "step": 121
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 0.00055425,
+      "loss": 0.8228,
+      "step": 122
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 0.0005538749999999999,
+      "loss": 0.7552,
+      "step": 123
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 0.0005535,
+      "loss": 0.877,
+      "step": 124
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 0.0005531249999999999,
+      "loss": 0.8001,
+      "step": 125
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 0.00055275,
+      "loss": 0.828,
+      "step": 126
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 0.0005523749999999999,
+      "loss": 0.8079,
+      "step": 127
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 0.000552,
+      "loss": 0.855,
+      "step": 128
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 0.000551625,
+      "loss": 0.8755,
+      "step": 129
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 0.0005512499999999999,
+      "loss": 0.8063,
+      "step": 130
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 0.0005508749999999999,
+      "loss": 0.5529,
+      "step": 131
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 0.0005505,
+      "loss": 0.8243,
+      "step": 132
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 0.0005501249999999999,
+      "loss": 0.6746,
+      "step": 133
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 0.00054975,
+      "loss": 0.9177,
+      "step": 134
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 0.000549375,
+      "loss": 0.7125,
+      "step": 135
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 0.000549,
+      "loss": 0.8089,
+      "step": 136
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 0.000548625,
+      "loss": 0.8293,
+      "step": 137
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 0.0005482499999999999,
+      "loss": 0.7055,
+      "step": 138
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 0.000547875,
+      "loss": 0.6936,
+      "step": 139
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 0.0005474999999999999,
+      "loss": 0.7125,
+      "step": 140
+    },
+    {
+      "epoch": 4.38,
+      "eval_loss": 0.9963582158088684,
+      "eval_mean_accuracy": 0.48408195145164323,
+      "eval_mean_iou": 0.3450134720578326,
+      "eval_overall_accuracy": 0.6522797032406455,
+      "eval_per_category_accuracy": [
+        0.7072444908656982,
+        0.7263543410072708,
+        0.5920856376726774,
+        0.15527303766108494,
+        0.527573555480372,
+        0.5846359317555773,
+        0.6484084836786329,
+        0.4228011884570095,
+        0.22752056289539763,
+        0.5662632128222072,
+        0.07395481661274575,
+        0.5768681585110444
+      ],
+      "eval_per_category_iou": [
+        0.5488723081193907,
+        0.5312308684382402,
+        0.5141246717540758,
+        0.13968224240721666,
+        0.3491042953338633,
+        0.35387946042663687,
+        0.4041945008639172,
+        0.32553414229110106,
+        0.16599826506897644,
+        0.4057974643708637,
+        0.0649149673571371,
+        0.336828478262572
+      ],
+      "eval_runtime": 120.3417,
+      "eval_samples_per_second": 0.947,
+      "eval_steps_per_second": 0.083,
+      "step": 140
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 0.000547125,
+      "loss": 0.7107,
+      "step": 141
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 0.0005467499999999999,
+      "loss": 0.7305,
+      "step": 142
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 0.000546375,
+      "loss": 0.7128,
+      "step": 143
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 0.0005459999999999999,
+      "loss": 0.7267,
+      "step": 144
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 0.000545625,
+      "loss": 0.7552,
+      "step": 145
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 0.0005452499999999999,
+      "loss": 0.6716,
+      "step": 146
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 0.0005448749999999999,
+      "loss": 0.8395,
+      "step": 147
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 0.0005445,
+      "loss": 0.6889,
+      "step": 148
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 0.0005441249999999999,
+      "loss": 0.7029,
+      "step": 149
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 0.00054375,
+      "loss": 0.817,
+      "step": 150
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 0.0005433749999999999,
+      "loss": 0.6999,
+      "step": 151
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 0.000543,
+      "loss": 0.7728,
+      "step": 152
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 0.000542625,
+      "loss": 0.7724,
+      "step": 153
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 0.00054225,
+      "loss": 0.8145,
+      "step": 154
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 0.0005418749999999999,
+      "loss": 0.7165,
+      "step": 155
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 0.0005414999999999999,
+      "loss": 0.6526,
+      "step": 156
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 0.0005411249999999999,
+      "loss": 0.7235,
+      "step": 157
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 0.00054075,
+      "loss": 0.7316,
+      "step": 158
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 0.000540375,
+      "loss": 0.7012,
+      "step": 159
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 0.00054,
+      "loss": 0.7644,
+      "step": 160
+    },
+    {
+      "epoch": 5.0,
+      "eval_loss": 0.9798843860626221,
+      "eval_mean_accuracy": 0.4789239253628546,
+      "eval_mean_iou": 0.3599116146321528,
+      "eval_overall_accuracy": 0.6647834108586896,
+      "eval_per_category_accuracy": [
+        0.5731060591042803,
+        0.8157629264757552,
+        0.7355228611483893,
+        0.23873533146890027,
+        0.5446721784739682,
+        0.5222981138786161,
+        0.5353058888544263,
+        0.4231504825294034,
+        0.29340390229186447,
+        0.5527727955246821,
+        0.14528959667965463,
+        0.36706696792431415
+      ],
+      "eval_per_category_iou": [
+        0.5072616266412792,
+        0.5572558530924372,
+        0.5665812265091772,
+        0.19904648390941598,
+        0.34126626496389373,
+        0.3523336548678199,
+        0.4169797089718354,
+        0.36311888106744344,
+        0.19641239624014376,
+        0.4094302245894939,
+        0.11159881978158906,
+        0.2976542349513052
+      ],
+      "eval_runtime": 120.1425,
+      "eval_samples_per_second": 0.949,
+      "eval_steps_per_second": 0.083,
+      "step": 160
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 0.000539625,
+      "loss": 0.6204,
+      "step": 161
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 0.00053925,
+      "loss": 0.6766,
+      "step": 162
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 0.000538875,
+      "loss": 0.662,
+      "step": 163
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 0.0005384999999999999,
+      "loss": 0.7268,
+      "step": 164
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 0.000538125,
+      "loss": 0.5879,
+      "step": 165
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 0.0005377499999999999,
+      "loss": 0.7585,
+      "step": 166
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 0.000537375,
+      "loss": 0.6846,
+      "step": 167
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 0.0005369999999999999,
+      "loss": 0.6442,
+      "step": 168
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 0.000536625,
+      "loss": 0.6522,
+      "step": 169
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 0.0005362499999999999,
+      "loss": 0.6976,
+      "step": 170
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 0.0005358749999999999,
+      "loss": 0.7117,
+      "step": 171
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 0.0005355,
+      "loss": 0.7123,
+      "step": 172
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 0.0005351249999999999,
+      "loss": 0.7688,
+      "step": 173
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 0.00053475,
+      "loss": 0.6922,
+      "step": 174
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 0.0005343749999999999,
+      "loss": 0.6724,
+      "step": 175
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 0.000534,
+      "loss": 0.7095,
+      "step": 176
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 0.0005336249999999999,
+      "loss": 0.8182,
+      "step": 177
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 0.00053325,
+      "loss": 0.6316,
+      "step": 178
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 0.000532875,
+      "loss": 0.6601,
+      "step": 179
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 0.0005324999999999999,
+      "loss": 0.6728,
+      "step": 180
+    },
+    {
+      "epoch": 5.62,
+      "eval_loss": 0.9665025472640991,
+      "eval_mean_accuracy": 0.47597824341463396,
+      "eval_mean_iou": 0.35415269637901653,
+      "eval_overall_accuracy": 0.6696545450310958,
+      "eval_per_category_accuracy": [
+        0.658208933526517,
+        0.7669225220721168,
+        0.7575903018999617,
+        0.1655887190234627,
+        0.5932391558466553,
+        0.3657776213167646,
+        0.5939283538193401,
+        0.30520908893177084,
+        0.3045304916461851,
+        0.6047843885423767,
+        0.10483819131776208,
+        0.4911211530326947
+      ],
+      "eval_per_category_iou": [
+        0.5480879206551327,
+        0.5553860439192433,
+        0.5557114007818262,
+        0.15048023258013862,
+        0.3592312571075963,
+        0.2947221901368397,
+        0.4306816722276294,
+        0.28278529443006634,
+        0.21869046329512365,
+        0.4231944375509108,
+        0.09075084320494715,
+        0.34011060065874554
+      ],
+      "eval_runtime": 120.9628,
+      "eval_samples_per_second": 0.942,
+      "eval_steps_per_second": 0.083,
+      "step": 180
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 0.0005321249999999999,
+      "loss": 0.6799,
+      "step": 181
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 0.00053175,
+      "loss": 0.6823,
+      "step": 182
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 0.000531375,
+      "loss": 0.7729,
+      "step": 183
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 0.000531,
+      "loss": 0.638,
+      "step": 184
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 0.000530625,
+      "loss": 0.688,
+      "step": 185
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 0.00053025,
+      "loss": 0.6704,
+      "step": 186
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 0.000529875,
+      "loss": 0.5638,
+      "step": 187
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 0.0005294999999999999,
+      "loss": 0.6177,
+      "step": 188
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 0.000529125,
+      "loss": 0.672,
+      "step": 189
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 0.0005287499999999999,
+      "loss": 0.7459,
+      "step": 190
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 0.000528375,
+      "loss": 0.7019,
+      "step": 191
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 0.0005279999999999999,
+      "loss": 0.5613,
+      "step": 192
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 0.000527625,
+      "loss": 0.6292,
+      "step": 193
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 0.0005272499999999999,
+      "loss": 0.6222,
+      "step": 194
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 0.000526875,
+      "loss": 0.5749,
+      "step": 195
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 0.0005265,
+      "loss": 0.6755,
+      "step": 196
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 0.0005261249999999999,
+      "loss": 0.618,
+      "step": 197
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 0.00052575,
+      "loss": 0.6044,
+      "step": 198
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 0.0005253749999999999,
+      "loss": 0.6677,
+      "step": 199
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 0.000525,
+      "loss": 0.704,
+      "step": 200
+    },
+    {
+      "epoch": 6.25,
+      "eval_loss": 0.9547889828681946,
+      "eval_mean_accuracy": 0.5432725380717788,
+      "eval_mean_iou": 0.3815082719789144,
+      "eval_overall_accuracy": 0.6700487638774671,
+      "eval_per_category_accuracy": [
+        0.7256938555518796,
+        0.7108576559277031,
+        0.7035051829979063,
+        0.29511939542404286,
+        0.5846675594146761,
+        0.5089878383822254,
+        0.646818305625924,
+        0.43729474955293574,
+        0.4849826535229763,
+        0.5611655812337896,
+        0.293272688873721,
+        0.566904990353565
+      ],
+      "eval_per_category_iou": [
+        0.5812467494831517,
+        0.5482754384827201,
+        0.5501083753603029,
+        0.23335839200171204,
+        0.3699939230925049,
+        0.3534576834188965,
+        0.4296805995484874,
+        0.3685008401449329,
+        0.21395048439181916,
+        0.4146264956148413,
+        0.16240302785306165,
+        0.3524972543545424
+      ],
+      "eval_runtime": 120.5907,
+      "eval_samples_per_second": 0.945,
+      "eval_steps_per_second": 0.083,
+      "step": 200
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 0.0005246249999999999,
+      "loss": 0.6344,
+      "step": 201
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 0.00052425,
+      "loss": 0.6519,
+      "step": 202
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 0.0005238749999999999,
+      "loss": 0.6823,
+      "step": 203
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 0.0005235,
+      "loss": 0.6326,
+      "step": 204
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 0.0005231249999999999,
+      "loss": 0.6317,
+      "step": 205
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 0.0005227499999999999,
+      "loss": 0.6645,
+      "step": 206
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 0.0005223749999999999,
+      "loss": 0.5607,
+      "step": 207
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 0.000522,
+      "loss": 0.6568,
+      "step": 208
+    },
+    {
+      "epoch": 6.53,
+      "learning_rate": 0.000521625,
+      "loss": 0.6128,
+      "step": 209
+    },
+    {
+      "epoch": 6.56,
+      "learning_rate": 0.00052125,
+      "loss": 0.6798,
+      "step": 210
+    },
+    {
+      "epoch": 6.59,
+      "learning_rate": 0.000520875,
+      "loss": 0.541,
+      "step": 211
+    },
+    {
+      "epoch": 6.62,
+      "learning_rate": 0.0005205,
+      "loss": 0.7654,
+      "step": 212
+    },
+    {
+      "epoch": 6.66,
+      "learning_rate": 0.000520125,
+      "loss": 0.4971,
+      "step": 213
+    },
+    {
+      "epoch": 6.69,
+      "learning_rate": 0.0005197499999999999,
+      "loss": 0.5991,
+      "step": 214
+    },
+    {
+      "epoch": 6.72,
+      "learning_rate": 0.000519375,
+      "loss": 0.6807,
+      "step": 215
+    },
+    {
+      "epoch": 6.75,
+      "learning_rate": 0.0005189999999999999,
+      "loss": 0.6202,
+      "step": 216
+    },
+    {
+      "epoch": 6.78,
+      "learning_rate": 0.000518625,
+      "loss": 0.6998,
+      "step": 217
+    },
+    {
+      "epoch": 6.81,
+      "learning_rate": 0.0005182499999999999,
+      "loss": 0.6515,
+      "step": 218
+    },
+    {
+      "epoch": 6.84,
+      "learning_rate": 0.000517875,
+      "loss": 0.6392,
+      "step": 219
+    },
+    {
+      "epoch": 6.88,
+      "learning_rate": 0.0005175,
+      "loss": 0.5244,
+      "step": 220
+    },
+    {
+      "epoch": 6.88,
+      "eval_loss": 0.9732311367988586,
+      "eval_mean_accuracy": 0.5190252675707585,
+      "eval_mean_iou": 0.37072453530870725,
+      "eval_overall_accuracy": 0.6681206686454907,
+      "eval_per_category_accuracy": [
+        0.7338891842835178,
+        0.7097288868357648,
+        0.7076853099740072,
+        0.3548032282912679,
+        0.6012819915438801,
+        0.5101147206021652,
+        0.5778423645475314,
+        0.49800655944267236,
+        0.36301828700212063,
+        0.6331348139447919,
+        0.10758984184896224,
+        0.4312080225324213
+      ],
+      "eval_per_category_iou": [
+        0.5701632769051872,
+        0.5415899026405016,
+        0.5645307413899069,
+        0.2206872588205841,
+        0.3573009062629415,
+        0.35439027772524995,
+        0.3834653201038341,
+        0.3842033522268411,
+        0.21110496232687195,
+        0.42611516905105984,
+        0.09413286415527866,
+        0.3410103920962303
+      ],
+      "eval_runtime": 121.2246,
+      "eval_samples_per_second": 0.94,
+      "eval_steps_per_second": 0.082,
+      "step": 220
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 0.0005171249999999999,
+      "loss": 0.6746,
+      "step": 221
+    },
+    {
+      "epoch": 6.94,
+      "learning_rate": 0.00051675,
+      "loss": 0.6267,
+      "step": 222
+    },
+    {
+      "epoch": 6.97,
+      "learning_rate": 0.0005163749999999999,
+      "loss": 0.5898,
+      "step": 223
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 0.000516,
+      "loss": 0.5663,
+      "step": 224
+    },
+    {
+      "epoch": 7.03,
+      "learning_rate": 0.0005156249999999999,
+      "loss": 0.7091,
+      "step": 225
+    },
+    {
+      "epoch": 7.06,
+      "learning_rate": 0.00051525,
+      "loss": 0.5594,
+      "step": 226
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 0.0005148749999999999,
+      "loss": 0.6431,
+      "step": 227
+    },
+    {
+      "epoch": 7.12,
+      "learning_rate": 0.0005145,
+      "loss": 0.5699,
+      "step": 228
+    },
+    {
+      "epoch": 7.16,
+      "learning_rate": 0.0005141249999999999,
+      "loss": 0.5724,
+      "step": 229
+    },
+    {
+      "epoch": 7.19,
+      "learning_rate": 0.0005137499999999999,
+      "loss": 0.6459,
+      "step": 230
+    },
+    {
+      "epoch": 7.22,
+      "learning_rate": 0.0005133749999999999,
+      "loss": 0.6204,
+      "step": 231
+    },
+    {
+      "epoch": 7.25,
+      "learning_rate": 0.0005129999999999999,
+      "loss": 0.5749,
+      "step": 232
+    },
+    {
+      "epoch": 7.28,
+      "learning_rate": 0.000512625,
+      "loss": 0.5592,
+      "step": 233
+    },
+    {
+      "epoch": 7.31,
+      "learning_rate": 0.00051225,
+      "loss": 0.5161,
+      "step": 234
+    },
+    {
+      "epoch": 7.34,
+      "learning_rate": 0.000511875,
+      "loss": 0.6087,
+      "step": 235
+    },
+    {
+      "epoch": 7.38,
+      "learning_rate": 0.0005115,
+      "loss": 0.5574,
+      "step": 236
+    },
+    {
+      "epoch": 7.41,
+      "learning_rate": 0.000511125,
+      "loss": 0.5837,
+      "step": 237
+    },
+    {
+      "epoch": 7.44,
+      "learning_rate": 0.0005107499999999999,
+      "loss": 0.6824,
+      "step": 238
+    },
+    {
+      "epoch": 7.47,
+      "learning_rate": 0.000510375,
+      "loss": 0.6023,
+      "step": 239
+    },
+    {
+      "epoch": 7.5,
+      "learning_rate": 0.0005099999999999999,
+      "loss": 0.5475,
+      "step": 240
+    },
+    {
+      "epoch": 7.5,
+      "eval_loss": 1.0112754106521606,
+      "eval_mean_accuracy": 0.49948921620422193,
+      "eval_mean_iou": 0.37079604232386026,
+      "eval_overall_accuracy": 0.6726983388264974,
+      "eval_per_category_accuracy": [
+        0.583102389554102,
+        0.8353006889840965,
+        0.6556883997695494,
+        0.33851604083247167,
+        0.5812701197511381,
+        0.5531648877291417,
+        0.595624018930991,
+        0.5104322139136901,
+        0.3197519143234874,
+        0.5198379439239663,
+        0.07253154909660772,
+        0.42865042764142086
+      ],
+      "eval_per_category_iou": [
+        0.5216569916948373,
+        0.5698887704473066,
+        0.5512999870681475,
+        0.24047394276108386,
+        0.3687004459065267,
+        0.36736644211185965,
+        0.4299937429210045,
+        0.37814350961805593,
+        0.20937927588480856,
+        0.40461653626223026,
+        0.06787054274667048,
+        0.34016232046379147
+      ],
+      "eval_runtime": 121.1674,
+      "eval_samples_per_second": 0.941,
+      "eval_steps_per_second": 0.083,
+      "step": 240
+    },
+    {
+      "epoch": 7.53,
+      "learning_rate": 0.000509625,
+      "loss": 0.5618,
+      "step": 241
+    },
+    {
+      "epoch": 7.56,
+      "learning_rate": 0.0005092499999999999,
+      "loss": 0.5768,
+      "step": 242
+    },
+    {
+      "epoch": 7.59,
+      "learning_rate": 0.000508875,
+      "loss": 0.605,
+      "step": 243
+    },
+    {
+      "epoch": 7.62,
+      "learning_rate": 0.0005085,
+      "loss": 0.565,
+      "step": 244
+    },
+    {
+      "epoch": 7.66,
+      "learning_rate": 0.000508125,
+      "loss": 0.6199,
+      "step": 245
+    },
+    {
+      "epoch": 7.69,
+      "learning_rate": 0.00050775,
+      "loss": 0.4687,
+      "step": 246
+    },
+    {
+      "epoch": 7.72,
+      "learning_rate": 0.0005073749999999999,
+      "loss": 0.5892,
+      "step": 247
+    },
+    {
+      "epoch": 7.75,
+      "learning_rate": 0.000507,
+      "loss": 0.6484,
+      "step": 248
+    },
+    {
+      "epoch": 7.78,
+      "learning_rate": 0.0005066249999999999,
+      "loss": 0.596,
+      "step": 249
+    },
+    {
+      "epoch": 7.81,
+      "learning_rate": 0.00050625,
+      "loss": 0.5773,
+      "step": 250
+    },
+    {
+      "epoch": 7.84,
+      "learning_rate": 0.0005058749999999999,
+      "loss": 0.6027,
+      "step": 251
+    },
+    {
+      "epoch": 7.88,
+      "learning_rate": 0.0005055,
+      "loss": 0.5894,
+      "step": 252
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 0.0005051249999999999,
+      "loss": 0.6463,
+      "step": 253
+    },
+    {
+      "epoch": 7.94,
+      "learning_rate": 0.00050475,
+      "loss": 0.5688,
+      "step": 254
+    },
+    {
+      "epoch": 7.97,
+      "learning_rate": 0.0005043749999999999,
+      "loss": 0.5924,
+      "step": 255
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 0.0005039999999999999,
+      "loss": 0.6616,
+      "step": 256
+    },
+    {
+      "epoch": 8.03,
+      "learning_rate": 0.000503625,
+      "loss": 0.5987,
+      "step": 257
+    },
+    {
+      "epoch": 8.06,
+      "learning_rate": 0.0005032499999999999,
+      "loss": 0.5851,
+      "step": 258
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 0.000502875,
+      "loss": 0.5072,
+      "step": 259
+    },
+    {
+      "epoch": 8.12,
+      "learning_rate": 0.0005025,
+      "loss": 0.6294,
+      "step": 260
+    },
+    {
+      "epoch": 8.12,
+      "eval_loss": 0.9677308201789856,
+      "eval_mean_accuracy": 0.5352102294125518,
+      "eval_mean_iou": 0.3874779695183849,
+      "eval_overall_accuracy": 0.6757297515869141,
+      "eval_per_category_accuracy": [
+        0.7860712450729425,
+        0.6996335098089403,
+        0.65897109351882,
+        0.3466006481565013,
+        0.5820622479016707,
+        0.5249492812269769,
+        0.6516203284583617,
+        0.42754130188115214,
+        0.3181858513083059,
+        0.6402143362674009,
+        0.24809542320017874,
+        0.5385774861493702
+      ],
+      "eval_per_category_iou": [
+        0.5874043930268468,
+        0.5453910246161039,
+        0.5540919673215583,
+        0.24924583242883128,
+        0.36393645710753425,
+        0.36953703124141746,
+        0.43503432965720795,
+        0.37537864391855535,
+        0.235992267511208,
+        0.42864730414486657,
+        0.15718640675783643,
+        0.34788997648865216
+      ],
+      "eval_runtime": 121.2884,
+      "eval_samples_per_second": 0.94,
+      "eval_steps_per_second": 0.082,
+      "step": 260
+    },
+    {
+      "epoch": 8.16,
+      "learning_rate": 0.000502125,
+      "loss": 0.5728,
+      "step": 261
+    },
+    {
+      "epoch": 8.19,
+      "learning_rate": 0.00050175,
+      "loss": 0.4925,
+      "step": 262
+    },
+    {
+      "epoch": 8.22,
+      "learning_rate": 0.000501375,
+      "loss": 0.5615,
+      "step": 263
+    },
+    {
+      "epoch": 8.25,
+      "learning_rate": 0.0005009999999999999,
+      "loss": 0.5562,
+      "step": 264
+    },
+    {
+      "epoch": 8.28,
+      "learning_rate": 0.000500625,
+      "loss": 0.5429,
+      "step": 265
+    },
+    {
+      "epoch": 8.31,
+      "learning_rate": 0.0005002499999999999,
+      "loss": 0.5993,
+      "step": 266
+    },
+    {
+      "epoch": 8.34,
+      "learning_rate": 0.000499875,
+      "loss": 0.6164,
+      "step": 267
+    },
+    {
+      "epoch": 8.38,
+      "learning_rate": 0.0004994999999999999,
+      "loss": 0.5683,
+      "step": 268
+    },
+    {
+      "epoch": 8.41,
+      "learning_rate": 0.000499125,
+      "loss": 0.5876,
+      "step": 269
+    },
+    {
+      "epoch": 8.44,
+      "learning_rate": 0.00049875,
+      "loss": 0.648,
+      "step": 270
+    },
+    {
+      "epoch": 8.47,
+      "learning_rate": 0.0004983749999999999,
+      "loss": 0.5063,
+      "step": 271
+    },
+    {
+      "epoch": 8.5,
+      "learning_rate": 0.000498,
+      "loss": 0.5067,
+      "step": 272
+    },
+    {
+      "epoch": 8.53,
+      "learning_rate": 0.0004976249999999999,
+      "loss": 0.5348,
+      "step": 273
+    },
+    {
+      "epoch": 8.56,
+      "learning_rate": 0.00049725,
+      "loss": 0.5394,
+      "step": 274
+    },
+    {
+      "epoch": 8.59,
+      "learning_rate": 0.0004968749999999999,
+      "loss": 0.5167,
+      "step": 275
+    },
+    {
+      "epoch": 8.62,
+      "learning_rate": 0.0004965,
+      "loss": 0.5693,
+      "step": 276
+    },
+    {
+      "epoch": 8.66,
+      "learning_rate": 0.0004961249999999999,
+      "loss": 0.5314,
+      "step": 277
+    },
+    {
+      "epoch": 8.69,
+      "learning_rate": 0.00049575,
+      "loss": 0.5412,
+      "step": 278
+    },
+    {
+      "epoch": 8.72,
+      "learning_rate": 0.0004953749999999999,
+      "loss": 0.5591,
+      "step": 279
+    },
+    {
+      "epoch": 8.75,
+      "learning_rate": 0.0004949999999999999,
+      "loss": 0.6466,
+      "step": 280
+    },
+    {
+      "epoch": 8.75,
+      "eval_loss": 1.1633713245391846,
+      "eval_mean_accuracy": 0.45516446277688666,
+      "eval_mean_iou": 0.34044253625459714,
+      "eval_overall_accuracy": 0.6490158281828228,
+      "eval_per_category_accuracy": [
+        0.435275610530772,
+        0.8518805461230962,
+        0.77708300815052,
+        0.2383918224023428,
+        0.5511307578701274,
+        0.49349456882691745,
+        0.5473558172964139,
+        0.15364546223070338,
+        0.2011725508637747,
+        0.5975145003992467,
+        0.17590668262358705,
+        0.4391222260051379
+      ],
+      "eval_per_category_iou": [
+        0.41258905558282083,
+        0.5588823205403024,
+        0.5499089792040451,
+        0.21416770573566085,
+        0.3541706266029605,
+        0.3449148791262031,
+        0.4391789251215739,
+        0.1511643106374456,
+        0.17077354724513238,
+        0.41425552671677335,
+        0.13590669494952362,
+        0.33939786359272384
+      ],
+      "eval_runtime": 121.6491,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 280
+    },
+    {
+      "epoch": 8.78,
+      "learning_rate": 0.000494625,
+      "loss": 0.5825,
+      "step": 281
+    },
+    {
+      "epoch": 8.81,
+      "learning_rate": 0.0004942499999999999,
+      "loss": 0.6544,
+      "step": 282
+    },
+    {
+      "epoch": 8.84,
+      "learning_rate": 0.000493875,
+      "loss": 0.5803,
+      "step": 283
+    },
+    {
+      "epoch": 8.88,
+      "learning_rate": 0.0004935,
+      "loss": 0.526,
+      "step": 284
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 0.000493125,
+      "loss": 0.5914,
+      "step": 285
+    },
+    {
+      "epoch": 8.94,
+      "learning_rate": 0.00049275,
+      "loss": 0.4828,
+      "step": 286
+    },
+    {
+      "epoch": 8.97,
+      "learning_rate": 0.000492375,
+      "loss": 0.5737,
+      "step": 287
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 0.0004919999999999999,
+      "loss": 0.6182,
+      "step": 288
+    },
+    {
+      "epoch": 9.03,
+      "learning_rate": 0.000491625,
+      "loss": 0.5085,
+      "step": 289
+    },
+    {
+      "epoch": 9.06,
+      "learning_rate": 0.0004912499999999999,
+      "loss": 0.4522,
+      "step": 290
+    },
+    {
+      "epoch": 9.09,
+      "learning_rate": 0.000490875,
+      "loss": 0.5149,
+      "step": 291
+    },
+    {
+      "epoch": 9.12,
+      "learning_rate": 0.0004904999999999999,
+      "loss": 0.5657,
+      "step": 292
+    },
+    {
+      "epoch": 9.16,
+      "learning_rate": 0.000490125,
+      "loss": 0.5126,
+      "step": 293
+    },
+    {
+      "epoch": 9.19,
+      "learning_rate": 0.00048975,
+      "loss": 0.4972,
+      "step": 294
+    },
+    {
+      "epoch": 9.22,
+      "learning_rate": 0.000489375,
+      "loss": 0.5683,
+      "step": 295
+    },
+    {
+      "epoch": 9.25,
+      "learning_rate": 0.000489,
+      "loss": 0.4438,
+      "step": 296
+    },
+    {
+      "epoch": 9.28,
+      "learning_rate": 0.0004886249999999999,
+      "loss": 0.4399,
+      "step": 297
+    },
+    {
+      "epoch": 9.31,
+      "learning_rate": 0.0004882499999999999,
+      "loss": 0.5879,
+      "step": 298
+    },
+    {
+      "epoch": 9.34,
+      "learning_rate": 0.000487875,
+      "loss": 0.513,
+      "step": 299
+    },
+    {
+      "epoch": 9.38,
+      "learning_rate": 0.0004875,
+      "loss": 0.5057,
+      "step": 300
+    },
+    {
+      "epoch": 9.38,
+      "eval_loss": 1.0140376091003418,
+      "eval_mean_accuracy": 0.5164049216621516,
+      "eval_mean_iou": 0.3753016343967814,
+      "eval_overall_accuracy": 0.6761952450400904,
+      "eval_per_category_accuracy": [
+        0.7040612772295968,
+        0.7717356769130097,
+        0.6398490600064661,
+        0.3456325771507484,
+        0.6136622859886509,
+        0.6304842508682982,
+        0.5371849754939392,
+        0.5657203225934335,
+        0.31926612423804157,
+        0.5927235403244355,
+        0.12099304282351789,
+        0.3555459263156809
+      ],
+      "eval_per_category_iou": [
+        0.5693101894871927,
+        0.5607649636340032,
+        0.5462865963528141,
+        0.24402205732847634,
+        0.335674533957752,
+        0.376631147629836,
+        0.4208455085595592,
+        0.3879488841178381,
+        0.21433187793734848,
+        0.42881135039928037,
+        0.1030105772813859,
+        0.31598192607588993
+      ],
+      "eval_runtime": 120.959,
+      "eval_samples_per_second": 0.942,
+      "eval_steps_per_second": 0.083,
+      "step": 300
+    },
+    {
+      "epoch": 9.41,
+      "learning_rate": 0.000487125,
+      "loss": 0.5731,
+      "step": 301
+    },
+    {
+      "epoch": 9.44,
+      "learning_rate": 0.00048675,
+      "loss": 0.5922,
+      "step": 302
+    },
+    {
+      "epoch": 9.47,
+      "learning_rate": 0.000486375,
+      "loss": 0.5068,
+      "step": 303
+    },
+    {
+      "epoch": 9.5,
+      "learning_rate": 0.000486,
+      "loss": 0.4608,
+      "step": 304
+    },
+    {
+      "epoch": 9.53,
+      "learning_rate": 0.00048562499999999995,
+      "loss": 0.5201,
+      "step": 305
+    },
+    {
+      "epoch": 9.56,
+      "learning_rate": 0.00048524999999999995,
+      "loss": 0.4989,
+      "step": 306
+    },
+    {
+      "epoch": 9.59,
+      "learning_rate": 0.00048487499999999995,
+      "loss": 0.5214,
+      "step": 307
+    },
+    {
+      "epoch": 9.62,
+      "learning_rate": 0.00048449999999999996,
+      "loss": 0.5588,
+      "step": 308
+    },
+    {
+      "epoch": 9.66,
+      "learning_rate": 0.00048412499999999996,
+      "loss": 0.5217,
+      "step": 309
+    },
+    {
+      "epoch": 9.69,
+      "learning_rate": 0.00048374999999999997,
+      "loss": 0.5234,
+      "step": 310
+    },
+    {
+      "epoch": 9.72,
+      "learning_rate": 0.00048337499999999997,
+      "loss": 0.5274,
+      "step": 311
+    },
+    {
+      "epoch": 9.75,
+      "learning_rate": 0.000483,
+      "loss": 0.5443,
+      "step": 312
+    },
+    {
+      "epoch": 9.78,
+      "learning_rate": 0.0004826249999999999,
+      "loss": 0.479,
+      "step": 313
+    },
+    {
+      "epoch": 9.81,
+      "learning_rate": 0.00048224999999999993,
+      "loss": 0.4926,
+      "step": 314
+    },
+    {
+      "epoch": 9.84,
+      "learning_rate": 0.00048187499999999994,
+      "loss": 0.5416,
+      "step": 315
+    },
+    {
+      "epoch": 9.88,
+      "learning_rate": 0.00048149999999999994,
+      "loss": 0.5723,
+      "step": 316
+    },
+    {
+      "epoch": 9.91,
+      "learning_rate": 0.00048112499999999994,
+      "loss": 0.5107,
+      "step": 317
+    },
+    {
+      "epoch": 9.94,
+      "learning_rate": 0.00048074999999999995,
+      "loss": 0.6088,
+      "step": 318
+    },
+    {
+      "epoch": 9.97,
+      "learning_rate": 0.00048037499999999995,
+      "loss": 0.494,
+      "step": 319
+    },
+    {
+      "epoch": 10.0,
+      "learning_rate": 0.00047999999999999996,
+      "loss": 0.5537,
+      "step": 320
+    },
+    {
+      "epoch": 10.0,
+      "eval_loss": 1.002641201019287,
+      "eval_mean_accuracy": 0.49957440190713626,
+      "eval_mean_iou": 0.38251915719313995,
+      "eval_overall_accuracy": 0.6835857859829015,
+      "eval_per_category_accuracy": [
+        0.6994669240491218,
+        0.7697357134553124,
+        0.7671867023509327,
+        0.23985607317090096,
+        0.48212646903305667,
+        0.46076417494438177,
+        0.6212062051581597,
+        0.3814901999440701,
+        0.3827870597943267,
+        0.623428764849186,
+        0.26144046915794594,
+        0.30540406697824135
+      ],
+      "eval_per_category_iou": [
+        0.5761962288087772,
+        0.5649042718911467,
+        0.5704844183828389,
+        0.21412486254588256,
+        0.3832999932353424,
+        0.3660397268508708,
+        0.4478781724966386,
+        0.341642286423795,
+        0.24395053839748818,
+        0.424868727805109,
+        0.18011057645674572,
+        0.2767300830230447
+      ],
+      "eval_runtime": 121.7458,
+      "eval_samples_per_second": 0.936,
+      "eval_steps_per_second": 0.082,
+      "step": 320
+    },
+    {
+      "epoch": 10.03,
+      "learning_rate": 0.0004796249999999999,
+      "loss": 0.514,
+      "step": 321
+    },
+    {
+      "epoch": 10.06,
+      "learning_rate": 0.0004792499999999999,
+      "loss": 0.5937,
+      "step": 322
+    },
+    {
+      "epoch": 10.09,
+      "learning_rate": 0.0004788749999999999,
+      "loss": 0.5341,
+      "step": 323
+    },
+    {
+      "epoch": 10.12,
+      "learning_rate": 0.0004785,
+      "loss": 0.4831,
+      "step": 324
+    },
+    {
+      "epoch": 10.16,
+      "learning_rate": 0.000478125,
+      "loss": 0.5363,
+      "step": 325
+    },
+    {
+      "epoch": 10.19,
+      "learning_rate": 0.00047775,
+      "loss": 0.5426,
+      "step": 326
+    },
+    {
+      "epoch": 10.22,
+      "learning_rate": 0.000477375,
+      "loss": 0.4749,
+      "step": 327
+    },
+    {
+      "epoch": 10.25,
+      "learning_rate": 0.000477,
+      "loss": 0.5598,
+      "step": 328
+    },
+    {
+      "epoch": 10.28,
+      "learning_rate": 0.000476625,
+      "loss": 0.4966,
+      "step": 329
+    },
+    {
+      "epoch": 10.31,
+      "learning_rate": 0.00047624999999999995,
+      "loss": 0.4268,
+      "step": 330
+    },
+    {
+      "epoch": 10.34,
+      "learning_rate": 0.00047587499999999995,
+      "loss": 0.4425,
+      "step": 331
+    },
+    {
+      "epoch": 10.38,
+      "learning_rate": 0.00047549999999999996,
+      "loss": 0.4802,
+      "step": 332
+    },
+    {
+      "epoch": 10.41,
+      "learning_rate": 0.00047512499999999996,
+      "loss": 0.4799,
+      "step": 333
+    },
+    {
+      "epoch": 10.44,
+      "learning_rate": 0.00047474999999999997,
+      "loss": 0.4465,
+      "step": 334
+    },
+    {
+      "epoch": 10.47,
+      "learning_rate": 0.00047437499999999997,
+      "loss": 0.491,
+      "step": 335
+    },
+    {
+      "epoch": 10.5,
+      "learning_rate": 0.000474,
+      "loss": 0.4901,
+      "step": 336
+    },
+    {
+      "epoch": 10.53,
+      "learning_rate": 0.000473625,
+      "loss": 0.5444,
+      "step": 337
+    },
+    {
+      "epoch": 10.56,
+      "learning_rate": 0.00047324999999999993,
+      "loss": 0.5099,
+      "step": 338
+    },
+    {
+      "epoch": 10.59,
+      "learning_rate": 0.00047287499999999993,
+      "loss": 0.5143,
+      "step": 339
+    },
+    {
+      "epoch": 10.62,
+      "learning_rate": 0.00047249999999999994,
+      "loss": 0.5615,
+      "step": 340
+    },
+    {
+      "epoch": 10.62,
+      "eval_loss": 1.1075023412704468,
+      "eval_mean_accuracy": 0.5381353239414549,
+      "eval_mean_iou": 0.38556839144435723,
+      "eval_overall_accuracy": 0.6534654717696341,
+      "eval_per_category_accuracy": [
+        0.4403944788467833,
+        0.8219203044717592,
+        0.717188400197578,
+        0.3802853554104413,
+        0.4857811984027779,
+        0.5460551863801495,
+        0.6359948610483525,
+        0.5295748148259262,
+        0.526430086155648,
+        0.5662027129997951,
+        0.32515388123521255,
+        0.48264260732303693
+      ],
+      "eval_per_category_iou": [
+        0.41895938854417153,
+        0.5549934087655012,
+        0.5777242197305088,
+        0.2713095011176766,
+        0.3829874570948952,
+        0.37263111229988904,
+        0.44410674353714125,
+        0.41640556613811885,
+        0.22631118706068476,
+        0.4034299245288768,
+        0.18868883003760206,
+        0.3692733584772213
+      ],
+      "eval_runtime": 120.3448,
+      "eval_samples_per_second": 0.947,
+      "eval_steps_per_second": 0.083,
+      "step": 340
+    },
+    {
+      "epoch": 10.66,
+      "learning_rate": 0.00047212499999999994,
+      "loss": 0.583,
+      "step": 341
+    },
+    {
+      "epoch": 10.69,
+      "learning_rate": 0.00047174999999999995,
+      "loss": 0.4993,
+      "step": 342
+    },
+    {
+      "epoch": 10.72,
+      "learning_rate": 0.00047137499999999995,
+      "loss": 0.4513,
+      "step": 343
+    },
+    {
+      "epoch": 10.75,
+      "learning_rate": 0.00047099999999999996,
+      "loss": 0.4375,
+      "step": 344
+    },
+    {
+      "epoch": 10.78,
+      "learning_rate": 0.00047062499999999996,
+      "loss": 0.5255,
+      "step": 345
+    },
+    {
+      "epoch": 10.81,
+      "learning_rate": 0.0004702499999999999,
+      "loss": 0.4966,
+      "step": 346
+    },
+    {
+      "epoch": 10.84,
+      "learning_rate": 0.0004698749999999999,
+      "loss": 0.5821,
+      "step": 347
+    },
+    {
+      "epoch": 10.88,
+      "learning_rate": 0.0004694999999999999,
+      "loss": 0.4963,
+      "step": 348
+    },
+    {
+      "epoch": 10.91,
+      "learning_rate": 0.0004691249999999999,
+      "loss": 0.4858,
+      "step": 349
+    },
+    {
+      "epoch": 10.94,
+      "learning_rate": 0.00046875,
+      "loss": 0.4664,
+      "step": 350
+    },
+    {
+      "epoch": 10.97,
+      "learning_rate": 0.000468375,
+      "loss": 0.5505,
+      "step": 351
+    },
+    {
+      "epoch": 11.0,
+      "learning_rate": 0.000468,
+      "loss": 0.5578,
+      "step": 352
+    },
+    {
+      "epoch": 11.03,
+      "learning_rate": 0.000467625,
+      "loss": 0.4877,
+      "step": 353
+    },
+    {
+      "epoch": 11.06,
+      "learning_rate": 0.00046725,
+      "loss": 0.4719,
+      "step": 354
+    },
+    {
+      "epoch": 11.09,
+      "learning_rate": 0.00046687499999999995,
+      "loss": 0.481,
+      "step": 355
+    },
+    {
+      "epoch": 11.12,
+      "learning_rate": 0.00046649999999999996,
+      "loss": 0.4697,
+      "step": 356
+    },
+    {
+      "epoch": 11.16,
+      "learning_rate": 0.00046612499999999996,
+      "loss": 0.4387,
+      "step": 357
+    },
+    {
+      "epoch": 11.19,
+      "learning_rate": 0.00046574999999999996,
+      "loss": 0.5047,
+      "step": 358
+    },
+    {
+      "epoch": 11.22,
+      "learning_rate": 0.00046537499999999997,
+      "loss": 0.4715,
+      "step": 359
+    },
+    {
+      "epoch": 11.25,
+      "learning_rate": 0.00046499999999999997,
+      "loss": 0.3817,
+      "step": 360
+    },
+    {
+      "epoch": 11.25,
+      "eval_loss": 0.973293662071228,
+      "eval_mean_accuracy": 0.526337814551732,
+      "eval_mean_iou": 0.39394445021340596,
+      "eval_overall_accuracy": 0.6902817843253153,
+      "eval_per_category_accuracy": [
+        0.7100156693757316,
+        0.7872355149788677,
+        0.7002809864859968,
+        0.39814782687143047,
+        0.5660008873456008,
+        0.5177470339951876,
+        0.5890586303371335,
+        0.6255674689358638,
+        0.288426217580721,
+        0.5324119975320245,
+        0.2230336717547205,
+        0.3781278694275073
+      ],
+      "eval_per_category_iou": [
+        0.5918248396748393,
+        0.5740929031755979,
+        0.5702195309365028,
+        0.2694635246046004,
+        0.39050156266598085,
+        0.37028495637469355,
+        0.4510922837827899,
+        0.4202353926834219,
+        0.2081714990153856,
+        0.4102416337231896,
+        0.15852862600402043,
+        0.31267664991984906
+      ],
+      "eval_runtime": 121.3069,
+      "eval_samples_per_second": 0.94,
+      "eval_steps_per_second": 0.082,
+      "step": 360
+    },
+    {
+      "epoch": 11.28,
+      "learning_rate": 0.000464625,
+      "loss": 0.4344,
+      "step": 361
+    },
+    {
+      "epoch": 11.31,
+      "learning_rate": 0.00046425,
+      "loss": 0.5005,
+      "step": 362
+    },
+    {
+      "epoch": 11.34,
+      "learning_rate": 0.00046387499999999993,
+      "loss": 0.5012,
+      "step": 363
+    },
+    {
+      "epoch": 11.38,
+      "learning_rate": 0.00046349999999999994,
+      "loss": 0.487,
+      "step": 364
+    },
+    {
+      "epoch": 11.41,
+      "learning_rate": 0.00046312499999999994,
+      "loss": 0.4264,
+      "step": 365
+    },
+    {
+      "epoch": 11.44,
+      "learning_rate": 0.00046274999999999995,
+      "loss": 0.4818,
+      "step": 366
+    },
+    {
+      "epoch": 11.47,
+      "learning_rate": 0.00046237499999999995,
+      "loss": 0.4483,
+      "step": 367
+    },
+    {
+      "epoch": 11.5,
+      "learning_rate": 0.00046199999999999995,
+      "loss": 0.444,
+      "step": 368
+    },
+    {
+      "epoch": 11.53,
+      "learning_rate": 0.00046162499999999996,
+      "loss": 0.5017,
+      "step": 369
+    },
+    {
+      "epoch": 11.56,
+      "learning_rate": 0.00046124999999999996,
+      "loss": 0.4874,
+      "step": 370
+    },
+    {
+      "epoch": 11.59,
+      "learning_rate": 0.0004608749999999999,
+      "loss": 0.4134,
+      "step": 371
+    },
+    {
+      "epoch": 11.62,
+      "learning_rate": 0.0004604999999999999,
+      "loss": 0.4885,
+      "step": 372
+    },
+    {
+      "epoch": 11.66,
+      "learning_rate": 0.0004601249999999999,
+      "loss": 0.4436,
+      "step": 373
+    },
+    {
+      "epoch": 11.69,
+      "learning_rate": 0.0004597499999999999,
+      "loss": 0.4543,
+      "step": 374
+    },
+    {
+      "epoch": 11.72,
+      "learning_rate": 0.000459375,
+      "loss": 0.4744,
+      "step": 375
+    },
+    {
+      "epoch": 11.75,
+      "learning_rate": 0.000459,
+      "loss": 0.3958,
+      "step": 376
+    },
+    {
+      "epoch": 11.78,
+      "learning_rate": 0.000458625,
+      "loss": 0.4659,
+      "step": 377
+    },
+    {
+      "epoch": 11.81,
+      "learning_rate": 0.00045825,
+      "loss": 0.6125,
+      "step": 378
+    },
+    {
+      "epoch": 11.84,
+      "learning_rate": 0.000457875,
+      "loss": 0.4794,
+      "step": 379
+    },
+    {
+      "epoch": 11.88,
+      "learning_rate": 0.00045749999999999995,
+      "loss": 0.4716,
+      "step": 380
+    },
+    {
+      "epoch": 11.88,
+      "eval_loss": 1.0539450645446777,
+      "eval_mean_accuracy": 0.5215361286621208,
+      "eval_mean_iou": 0.38890202441878613,
+      "eval_overall_accuracy": 0.6796468098958334,
+      "eval_per_category_accuracy": [
+        0.606029809551961,
+        0.7980285104338783,
+        0.7272465007951344,
+        0.4234876926600093,
+        0.5754638811848293,
+        0.4046396335908891,
+        0.6712141560484233,
+        0.47858002316484,
+        0.3977023681712111,
+        0.6485998202738035,
+        0.15310685525216322,
+        0.37433429281830655
+      ],
+      "eval_per_category_iou": [
+        0.53324069408375,
+        0.5678157977237887,
+        0.5824690503659471,
+        0.2709843581744913,
+        0.38702904356261886,
+        0.33664562140879434,
+        0.4587162281029742,
+        0.40908241655584743,
+        0.24548972248237788,
+        0.42867887898198076,
+        0.12907470912905147,
+        0.3175977724538109
+      ],
+      "eval_runtime": 120.9367,
+      "eval_samples_per_second": 0.943,
+      "eval_steps_per_second": 0.083,
+      "step": 380
+    },
+    {
+      "epoch": 11.91,
+      "learning_rate": 0.00045712499999999996,
+      "loss": 0.5158,
+      "step": 381
+    },
+    {
+      "epoch": 11.94,
+      "learning_rate": 0.00045674999999999996,
+      "loss": 0.483,
+      "step": 382
+    },
+    {
+      "epoch": 11.97,
+      "learning_rate": 0.00045637499999999997,
+      "loss": 0.4933,
+      "step": 383
+    },
+    {
+      "epoch": 12.0,
+      "learning_rate": 0.00045599999999999997,
+      "loss": 0.4897,
+      "step": 384
+    },
+    {
+      "epoch": 12.03,
+      "learning_rate": 0.000455625,
+      "loss": 0.4114,
+      "step": 385
+    },
+    {
+      "epoch": 12.06,
+      "learning_rate": 0.00045525,
+      "loss": 0.4022,
+      "step": 386
+    },
+    {
+      "epoch": 12.09,
+      "learning_rate": 0.000454875,
+      "loss": 0.4279,
+      "step": 387
+    },
+    {
+      "epoch": 12.12,
+      "learning_rate": 0.00045449999999999993,
+      "loss": 0.4571,
+      "step": 388
+    },
+    {
+      "epoch": 12.16,
+      "learning_rate": 0.00045412499999999994,
+      "loss": 0.4438,
+      "step": 389
+    },
+    {
+      "epoch": 12.19,
+      "learning_rate": 0.00045374999999999994,
+      "loss": 0.4597,
+      "step": 390
+    },
+    {
+      "epoch": 12.22,
+      "learning_rate": 0.00045337499999999995,
+      "loss": 0.4246,
+      "step": 391
+    },
+    {
+      "epoch": 12.25,
+      "learning_rate": 0.00045299999999999995,
+      "loss": 0.4786,
+      "step": 392
+    },
+    {
+      "epoch": 12.28,
+      "learning_rate": 0.00045262499999999996,
+      "loss": 0.4771,
+      "step": 393
+    },
+    {
+      "epoch": 12.31,
+      "learning_rate": 0.00045224999999999996,
+      "loss": 0.4844,
+      "step": 394
+    },
+    {
+      "epoch": 12.34,
+      "learning_rate": 0.00045187499999999997,
+      "loss": 0.3906,
+      "step": 395
+    },
+    {
+      "epoch": 12.38,
+      "learning_rate": 0.0004514999999999999,
+      "loss": 0.5017,
+      "step": 396
+    },
+    {
+      "epoch": 12.41,
+      "learning_rate": 0.0004511249999999999,
+      "loss": 0.4675,
+      "step": 397
+    },
+    {
+      "epoch": 12.44,
+      "learning_rate": 0.0004507499999999999,
+      "loss": 0.4282,
+      "step": 398
+    },
+    {
+      "epoch": 12.47,
+      "learning_rate": 0.00045037499999999993,
+      "loss": 0.515,
+      "step": 399
+    },
+    {
+      "epoch": 12.5,
+      "learning_rate": 0.00045,
+      "loss": 0.423,
+      "step": 400
+    },
+    {
+      "epoch": 12.5,
+      "eval_loss": 0.9965826869010925,
+      "eval_mean_accuracy": 0.5294550265143403,
+      "eval_mean_iou": 0.40386816272224996,
+      "eval_overall_accuracy": 0.6948729397957785,
+      "eval_per_category_accuracy": [
+        0.7349185378949704,
+        0.7654971297314803,
+        0.7413103780662184,
+        0.35135425847148877,
+        0.5190039849515911,
+        0.5204308645963004,
+        0.6351895382028468,
+        0.46694831626328204,
+        0.25594926488203507,
+        0.5999741311104169,
+        0.30752678956760215,
+        0.4553571244338523
+      ],
+      "eval_per_category_iou": [
+        0.5994361438746969,
+        0.568817829265267,
+        0.5918675192878663,
+        0.2695900812276531,
+        0.3890955318046989,
+        0.3725834843554118,
+        0.4553506580428748,
+        0.408580980613854,
+        0.20422702235815526,
+        0.4312235983237876,
+        0.19979478246344562,
+        0.35585032104928777
+      ],
+      "eval_runtime": 121.0341,
+      "eval_samples_per_second": 0.942,
+      "eval_steps_per_second": 0.083,
+      "step": 400
+    },
+    {
+      "epoch": 12.53,
+      "learning_rate": 0.000449625,
+      "loss": 0.4788,
+      "step": 401
+    },
+    {
+      "epoch": 12.56,
+      "learning_rate": 0.00044925,
+      "loss": 0.4511,
+      "step": 402
+    },
+    {
+      "epoch": 12.59,
+      "learning_rate": 0.000448875,
+      "loss": 0.487,
+      "step": 403
+    },
+    {
+      "epoch": 12.62,
+      "learning_rate": 0.0004485,
+      "loss": 0.4686,
+      "step": 404
+    },
+    {
+      "epoch": 12.66,
+      "learning_rate": 0.00044812499999999996,
+      "loss": 0.4973,
+      "step": 405
+    },
+    {
+      "epoch": 12.69,
+      "learning_rate": 0.00044774999999999996,
+      "loss": 0.4654,
+      "step": 406
+    },
+    {
+      "epoch": 12.72,
+      "learning_rate": 0.00044737499999999996,
+      "loss": 0.3882,
+      "step": 407
+    },
+    {
+      "epoch": 12.75,
+      "learning_rate": 0.00044699999999999997,
+      "loss": 0.4508,
+      "step": 408
+    },
+    {
+      "epoch": 12.78,
+      "learning_rate": 0.000446625,
+      "loss": 0.4219,
+      "step": 409
+    },
+    {
+      "epoch": 12.81,
+      "learning_rate": 0.00044625,
+      "loss": 0.4973,
+      "step": 410
+    },
+    {
+      "epoch": 12.84,
+      "learning_rate": 0.000445875,
+      "loss": 0.3894,
+      "step": 411
+    },
+    {
+      "epoch": 12.88,
+      "learning_rate": 0.0004455,
+      "loss": 0.4357,
+      "step": 412
+    },
+    {
+      "epoch": 12.91,
+      "learning_rate": 0.00044512499999999994,
+      "loss": 0.4383,
+      "step": 413
+    },
+    {
+      "epoch": 12.94,
+      "learning_rate": 0.00044474999999999994,
+      "loss": 0.3998,
+      "step": 414
+    },
+    {
+      "epoch": 12.97,
+      "learning_rate": 0.00044437499999999995,
+      "loss": 0.4731,
+      "step": 415
+    },
+    {
+      "epoch": 13.0,
+      "learning_rate": 0.00044399999999999995,
+      "loss": 0.4363,
+      "step": 416
+    },
+    {
+      "epoch": 13.03,
+      "learning_rate": 0.00044362499999999995,
+      "loss": 0.3997,
+      "step": 417
+    },
+    {
+      "epoch": 13.06,
+      "learning_rate": 0.00044324999999999996,
+      "loss": 0.4673,
+      "step": 418
+    },
+    {
+      "epoch": 13.09,
+      "learning_rate": 0.00044287499999999996,
+      "loss": 0.4469,
+      "step": 419
+    },
+    {
+      "epoch": 13.12,
+      "learning_rate": 0.00044249999999999997,
+      "loss": 0.4181,
+      "step": 420
+    },
+    {
+      "epoch": 13.12,
+      "eval_loss": 1.0220005512237549,
+      "eval_mean_accuracy": 0.5241064268418345,
+      "eval_mean_iou": 0.3929816513248286,
+      "eval_overall_accuracy": 0.6911841610021758,
+      "eval_per_category_accuracy": [
+        0.7356569999936858,
+        0.7889698999759565,
+        0.7284989123794137,
+        0.36568448518747265,
+        0.6084800256074164,
+        0.58659572692069,
+        0.572704200116823,
+        0.44663354444552783,
+        0.38804202195150084,
+        0.4385538247622644,
+        0.2574981711777615,
+        0.371959309583501
+      ],
+      "eval_per_category_iou": [
+        0.5988537870212209,
+        0.5726384929814583,
+        0.583485471943568,
+        0.2549938544620477,
+        0.3744596921289283,
+        0.35425947539438574,
+        0.4622040405567757,
+        0.3917247341737655,
+        0.2406539452560217,
+        0.370119346197287,
+        0.1933110444743161,
+        0.319075931308168
+      ],
+      "eval_runtime": 119.9947,
+      "eval_samples_per_second": 0.95,
+      "eval_steps_per_second": 0.083,
+      "step": 420
+    },
+    {
+      "epoch": 13.16,
+      "learning_rate": 0.0004421249999999999,
+      "loss": 0.4524,
+      "step": 421
+    },
+    {
+      "epoch": 13.19,
+      "learning_rate": 0.0004417499999999999,
+      "loss": 0.4499,
+      "step": 422
+    },
+    {
+      "epoch": 13.22,
+      "learning_rate": 0.00044137499999999993,
+      "loss": 0.4485,
+      "step": 423
+    },
+    {
+      "epoch": 13.25,
+      "learning_rate": 0.00044099999999999993,
+      "loss": 0.4635,
+      "step": 424
+    },
+    {
+      "epoch": 13.28,
+      "learning_rate": 0.00044062499999999994,
+      "loss": 0.5295,
+      "step": 425
+    },
+    {
+      "epoch": 13.31,
+      "learning_rate": 0.00044025,
+      "loss": 0.4706,
+      "step": 426
+    },
+    {
+      "epoch": 13.34,
+      "learning_rate": 0.000439875,
+      "loss": 0.5182,
+      "step": 427
+    },
+    {
+      "epoch": 13.38,
+      "learning_rate": 0.0004395,
+      "loss": 0.4629,
+      "step": 428
+    },
+    {
+      "epoch": 13.41,
+      "learning_rate": 0.000439125,
+      "loss": 0.3929,
+      "step": 429
+    },
+    {
+      "epoch": 13.44,
+      "learning_rate": 0.00043874999999999996,
+      "loss": 0.409,
+      "step": 430
+    },
+    {
+      "epoch": 13.47,
+      "learning_rate": 0.00043837499999999996,
+      "loss": 0.4477,
+      "step": 431
+    },
+    {
+      "epoch": 13.5,
+      "learning_rate": 0.00043799999999999997,
+      "loss": 0.3867,
+      "step": 432
+    },
+    {
+      "epoch": 13.53,
+      "learning_rate": 0.00043762499999999997,
+      "loss": 0.4317,
+      "step": 433
+    },
+    {
+      "epoch": 13.56,
+      "learning_rate": 0.00043725,
+      "loss": 0.4167,
+      "step": 434
+    },
+    {
+      "epoch": 13.59,
+      "learning_rate": 0.000436875,
+      "loss": 0.4197,
+      "step": 435
+    },
+    {
+      "epoch": 13.62,
+      "learning_rate": 0.0004365,
+      "loss": 0.4011,
+      "step": 436
+    },
+    {
+      "epoch": 13.66,
+      "learning_rate": 0.000436125,
+      "loss": 0.4367,
+      "step": 437
+    },
+    {
+      "epoch": 13.69,
+      "learning_rate": 0.00043574999999999994,
+      "loss": 0.4304,
+      "step": 438
+    },
+    {
+      "epoch": 13.72,
+      "learning_rate": 0.00043537499999999994,
+      "loss": 0.4591,
+      "step": 439
+    },
+    {
+      "epoch": 13.75,
+      "learning_rate": 0.00043499999999999995,
+      "loss": 0.3603,
+      "step": 440
+    },
+    {
+      "epoch": 13.75,
+      "eval_loss": 1.0305607318878174,
+      "eval_mean_accuracy": 0.5256165387398323,
+      "eval_mean_iou": 0.3994621574277373,
+      "eval_overall_accuracy": 0.6892888587817811,
+      "eval_per_category_accuracy": [
+        0.6253831258547049,
+        0.8135350587013286,
+        0.7413722995321077,
+        0.3445257146029521,
+        0.5377294082134167,
+        0.5453928481993474,
+        0.6360350091080991,
+        0.4459103128539147,
+        0.40154654268298096,
+        0.5827718410878285,
+        0.1904331936592667,
+        0.4427631103820401
+      ],
+      "eval_per_category_iou": [
+        0.5500551744374971,
+        0.5795062724471024,
+        0.5796235126769651,
+        0.25914780557063516,
+        0.4032656025427154,
+        0.3794912069818887,
+        0.4722700884037064,
+        0.38460571659872633,
+        0.25955579771446596,
+        0.4305571236183654,
+        0.15461250577774685,
+        0.34085508236303275
+      ],
+      "eval_runtime": 122.784,
+      "eval_samples_per_second": 0.928,
+      "eval_steps_per_second": 0.081,
+      "step": 440
+    },
+    {
+      "epoch": 13.78,
+      "learning_rate": 0.00043462499999999995,
+      "loss": 0.3868,
+      "step": 441
+    },
+    {
+      "epoch": 13.81,
+      "learning_rate": 0.00043424999999999996,
+      "loss": 0.4209,
+      "step": 442
+    },
+    {
+      "epoch": 13.84,
+      "learning_rate": 0.00043387499999999996,
+      "loss": 0.4102,
+      "step": 443
+    },
+    {
+      "epoch": 13.88,
+      "learning_rate": 0.00043349999999999997,
+      "loss": 0.3974,
+      "step": 444
+    },
+    {
+      "epoch": 13.91,
+      "learning_rate": 0.00043312499999999997,
+      "loss": 0.4317,
+      "step": 445
+    },
+    {
+      "epoch": 13.94,
+      "learning_rate": 0.0004327499999999999,
+      "loss": 0.4768,
+      "step": 446
+    },
+    {
+      "epoch": 13.97,
+      "learning_rate": 0.0004323749999999999,
+      "loss": 0.4409,
+      "step": 447
+    },
+    {
+      "epoch": 14.0,
+      "learning_rate": 0.00043199999999999993,
+      "loss": 0.3988,
+      "step": 448
+    },
+    {
+      "epoch": 14.03,
+      "learning_rate": 0.00043162499999999993,
+      "loss": 0.3918,
+      "step": 449
+    },
+    {
+      "epoch": 14.06,
+      "learning_rate": 0.00043124999999999994,
+      "loss": 0.3668,
+      "step": 450
+    },
+    {
+      "epoch": 14.09,
+      "learning_rate": 0.000430875,
+      "loss": 0.4063,
+      "step": 451
+    },
+    {
+      "epoch": 14.12,
+      "learning_rate": 0.0004305,
+      "loss": 0.3424,
+      "step": 452
+    },
+    {
+      "epoch": 14.16,
+      "learning_rate": 0.000430125,
+      "loss": 0.3771,
+      "step": 453
+    },
+    {
+      "epoch": 14.19,
+      "learning_rate": 0.00042975,
+      "loss": 0.4562,
+      "step": 454
+    },
+    {
+      "epoch": 14.22,
+      "learning_rate": 0.00042937499999999996,
+      "loss": 0.4455,
+      "step": 455
+    },
+    {
+      "epoch": 14.25,
+      "learning_rate": 0.00042899999999999997,
+      "loss": 0.4262,
+      "step": 456
+    },
+    {
+      "epoch": 14.28,
+      "learning_rate": 0.00042862499999999997,
+      "loss": 0.4238,
+      "step": 457
+    },
+    {
+      "epoch": 14.31,
+      "learning_rate": 0.00042825,
+      "loss": 0.3942,
+      "step": 458
+    },
+    {
+      "epoch": 14.34,
+      "learning_rate": 0.000427875,
+      "loss": 0.379,
+      "step": 459
+    },
+    {
+      "epoch": 14.38,
+      "learning_rate": 0.0004275,
+      "loss": 0.3431,
+      "step": 460
+    },
+    {
+      "epoch": 14.38,
+      "eval_loss": 1.0286169052124023,
+      "eval_mean_accuracy": 0.5109097206693113,
+      "eval_mean_iou": 0.3859667110484661,
+      "eval_overall_accuracy": 0.6894934804815995,
+      "eval_per_category_accuracy": [
+        0.6452928217343096,
+        0.8196776419155087,
+        0.7718687641588283,
+        0.38626380108396197,
+        0.6219421523600759,
+        0.5779871451954169,
+        0.54301274189206,
+        0.3439871596927498,
+        0.33019972848992485,
+        0.5579069356266243,
+        0.18019178912378755,
+        0.35258596675848836
+      ],
+      "eval_per_category_iou": [
+        0.5685088458625193,
+        0.5804917965019951,
+        0.5817617509242167,
+        0.26922667646945014,
+        0.3822386851771151,
+        0.37215899796695723,
+        0.45551878734008194,
+        0.3229780859760508,
+        0.23288391317815874,
+        0.41768816601803266,
+        0.14884606438171907,
+        0.2992987627852963
+      ],
+      "eval_runtime": 120.5616,
+      "eval_samples_per_second": 0.946,
+      "eval_steps_per_second": 0.083,
+      "step": 460
+    },
+    {
+      "epoch": 14.41,
+      "learning_rate": 0.000427125,
+      "loss": 0.4144,
+      "step": 461
+    },
+    {
+      "epoch": 14.44,
+      "learning_rate": 0.00042675,
+      "loss": 0.3867,
+      "step": 462
+    },
+    {
+      "epoch": 14.47,
+      "learning_rate": 0.00042637499999999994,
+      "loss": 0.4027,
+      "step": 463
+    },
+    {
+      "epoch": 14.5,
+      "learning_rate": 0.00042599999999999995,
+      "loss": 0.4171,
+      "step": 464
+    },
+    {
+      "epoch": 14.53,
+      "learning_rate": 0.00042562499999999995,
+      "loss": 0.4234,
+      "step": 465
+    },
+    {
+      "epoch": 14.56,
+      "learning_rate": 0.00042524999999999996,
+      "loss": 0.369,
+      "step": 466
+    },
+    {
+      "epoch": 14.59,
+      "learning_rate": 0.00042487499999999996,
+      "loss": 0.4339,
+      "step": 467
+    },
+    {
+      "epoch": 14.62,
+      "learning_rate": 0.00042449999999999996,
+      "loss": 0.382,
+      "step": 468
+    },
+    {
+      "epoch": 14.66,
+      "learning_rate": 0.00042412499999999997,
+      "loss": 0.4579,
+      "step": 469
+    },
+    {
+      "epoch": 14.69,
+      "learning_rate": 0.00042375,
+      "loss": 0.3522,
+      "step": 470
+    },
+    {
+      "epoch": 14.72,
+      "learning_rate": 0.0004233749999999999,
+      "loss": 0.4585,
+      "step": 471
+    },
+    {
+      "epoch": 14.75,
+      "learning_rate": 0.00042299999999999993,
+      "loss": 0.3902,
+      "step": 472
+    },
+    {
+      "epoch": 14.78,
+      "learning_rate": 0.00042262499999999993,
+      "loss": 0.4327,
+      "step": 473
+    },
+    {
+      "epoch": 14.81,
+      "learning_rate": 0.00042224999999999994,
+      "loss": 0.4649,
+      "step": 474
+    },
+    {
+      "epoch": 14.84,
+      "learning_rate": 0.00042187499999999994,
+      "loss": 0.478,
+      "step": 475
+    },
+    {
+      "epoch": 14.88,
+      "learning_rate": 0.00042149999999999995,
+      "loss": 0.4444,
+      "step": 476
+    },
+    {
+      "epoch": 14.91,
+      "learning_rate": 0.000421125,
+      "loss": 0.3839,
+      "step": 477
+    },
+    {
+      "epoch": 14.94,
+      "learning_rate": 0.00042075,
+      "loss": 0.4462,
+      "step": 478
+    },
+    {
+      "epoch": 14.97,
+      "learning_rate": 0.000420375,
+      "loss": 0.4425,
+      "step": 479
+    },
+    {
+      "epoch": 15.0,
+      "learning_rate": 0.00041999999999999996,
+      "loss": 0.3941,
+      "step": 480
+    },
+    {
+      "epoch": 15.0,
+      "eval_loss": 1.0285805463790894,
+      "eval_mean_accuracy": 0.5197467910140181,
+      "eval_mean_iou": 0.3950472583408569,
+      "eval_overall_accuracy": 0.693745730216043,
+      "eval_per_category_accuracy": [
+        0.7170439078785785,
+        0.7963107796720668,
+        0.7134728268918363,
+        0.3132282218721591,
+        0.5066783898931742,
+        0.5751254813293218,
+        0.6022909584995016,
+        0.540462932505885,
+        0.4537634313194857,
+        0.5136252380224694,
+        0.16598360028526565,
+        0.3389757239984731
+      ],
+      "eval_per_category_iou": [
+        0.5938434922493059,
+        0.5763621233677051,
+        0.583965121686162,
+        0.24249660859871328,
+        0.3919245674491054,
+        0.3829421416506171,
+        0.4558428984661032,
+        0.41286712737231623,
+        0.26023646989677657,
+        0.41167628061899947,
+        0.13963015222361835,
+        0.28878011651086183
+      ],
+      "eval_runtime": 121.0924,
+      "eval_samples_per_second": 0.941,
+      "eval_steps_per_second": 0.083,
+      "step": 480
+    },
+    {
+      "epoch": 15.03,
+      "learning_rate": 0.00041962499999999997,
+      "loss": 0.3682,
+      "step": 481
+    },
+    {
+      "epoch": 15.06,
+      "learning_rate": 0.00041924999999999997,
+      "loss": 0.4215,
+      "step": 482
+    },
+    {
+      "epoch": 15.09,
+      "learning_rate": 0.000418875,
+      "loss": 0.4185,
+      "step": 483
+    },
+    {
+      "epoch": 15.12,
+      "learning_rate": 0.0004185,
+      "loss": 0.4211,
+      "step": 484
+    },
+    {
+      "epoch": 15.16,
+      "learning_rate": 0.000418125,
+      "loss": 0.3746,
+      "step": 485
+    },
+    {
+      "epoch": 15.19,
+      "learning_rate": 0.00041775,
+      "loss": 0.3681,
+      "step": 486
+    },
+    {
+      "epoch": 15.22,
+      "learning_rate": 0.000417375,
+      "loss": 0.4,
+      "step": 487
+    },
+    {
+      "epoch": 15.25,
+      "learning_rate": 0.00041699999999999994,
+      "loss": 0.3611,
+      "step": 488
+    },
+    {
+      "epoch": 15.28,
+      "learning_rate": 0.00041662499999999995,
+      "loss": 0.3832,
+      "step": 489
+    },
+    {
+      "epoch": 15.31,
+      "learning_rate": 0.00041624999999999995,
+      "loss": 0.3775,
+      "step": 490
+    },
+    {
+      "epoch": 15.34,
+      "learning_rate": 0.00041587499999999996,
+      "loss": 0.4411,
+      "step": 491
+    },
+    {
+      "epoch": 15.38,
+      "learning_rate": 0.00041549999999999996,
+      "loss": 0.386,
+      "step": 492
+    },
+    {
+      "epoch": 15.41,
+      "learning_rate": 0.00041512499999999997,
+      "loss": 0.357,
+      "step": 493
+    },
+    {
+      "epoch": 15.44,
+      "learning_rate": 0.00041474999999999997,
+      "loss": 0.3954,
+      "step": 494
+    },
+    {
+      "epoch": 15.47,
+      "learning_rate": 0.000414375,
+      "loss": 0.4039,
+      "step": 495
+    },
+    {
+      "epoch": 15.5,
+      "learning_rate": 0.0004139999999999999,
+      "loss": 0.3694,
+      "step": 496
+    },
+    {
+      "epoch": 15.53,
+      "learning_rate": 0.00041362499999999993,
+      "loss": 0.3414,
+      "step": 497
+    },
+    {
+      "epoch": 15.56,
+      "learning_rate": 0.00041324999999999993,
+      "loss": 0.4035,
+      "step": 498
+    },
+    {
+      "epoch": 15.59,
+      "learning_rate": 0.00041287499999999994,
+      "loss": 0.3496,
+      "step": 499
+    },
+    {
+      "epoch": 15.62,
+      "learning_rate": 0.00041249999999999994,
+      "loss": 0.3998,
+      "step": 500
+    },
+    {
+      "epoch": 15.62,
+      "eval_loss": 1.0187643766403198,
+      "eval_mean_accuracy": 0.5449461308223081,
+      "eval_mean_iou": 0.40555339117484207,
+      "eval_overall_accuracy": 0.690818753158837,
+      "eval_per_category_accuracy": [
+        0.6340379192725392,
+        0.7957837458867677,
+        0.7712004688910264,
+        0.42085759293828634,
+        0.5455190060108548,
+        0.5766679127092716,
+        0.6171118902808632,
+        0.42225046099317376,
+        0.3839294428719732,
+        0.5959274231613139,
+        0.26965256968654444,
+        0.506415137165083
+      ],
+      "eval_per_category_iou": [
+        0.5632719719341162,
+        0.5784232067182266,
+        0.5843311366525699,
+        0.25945201318522415,
+        0.39589450144892163,
+        0.39429823527441155,
+        0.46001328548056586,
+        0.3893771996213835,
+        0.2505569782608381,
+        0.43197369311482003,
+        0.20160876928005858,
+        0.35743970312696843
+      ],
+      "eval_runtime": 119.6818,
+      "eval_samples_per_second": 0.953,
+      "eval_steps_per_second": 0.084,
+      "step": 500
+    },
+    {
+      "epoch": 15.66,
+      "learning_rate": 0.00041212499999999995,
+      "loss": 0.3262,
+      "step": 501
+    },
+    {
+      "epoch": 15.69,
+      "learning_rate": 0.00041175,
+      "loss": 0.4409,
+      "step": 502
+    },
+    {
+      "epoch": 15.72,
+      "learning_rate": 0.000411375,
+      "loss": 0.3697,
+      "step": 503
+    },
+    {
+      "epoch": 15.75,
+      "learning_rate": 0.000411,
+      "loss": 0.3989,
+      "step": 504
+    },
+    {
+      "epoch": 15.78,
+      "learning_rate": 0.0004106249999999999,
+      "loss": 0.4078,
+      "step": 505
+    },
+    {
+      "epoch": 15.81,
+      "learning_rate": 0.00041024999999999997,
+      "loss": 0.4582,
+      "step": 506
+    },
+    {
+      "epoch": 15.84,
+      "learning_rate": 0.000409875,
+      "loss": 0.4895,
+      "step": 507
+    },
+    {
+      "epoch": 15.88,
+      "learning_rate": 0.0004095,
+      "loss": 0.3963,
+      "step": 508
+    },
+    {
+      "epoch": 15.91,
+      "learning_rate": 0.000409125,
+      "loss": 0.381,
+      "step": 509
+    },
+    {
+      "epoch": 15.94,
+      "learning_rate": 0.00040875,
+      "loss": 0.3515,
+      "step": 510
+    },
+    {
+      "epoch": 15.97,
+      "learning_rate": 0.000408375,
+      "loss": 0.4238,
+      "step": 511
+    },
+    {
+      "epoch": 16.0,
+      "learning_rate": 0.000408,
+      "loss": 0.4653,
+      "step": 512
+    },
+    {
+      "epoch": 16.03,
+      "learning_rate": 0.00040762499999999995,
+      "loss": 0.3632,
+      "step": 513
+    },
+    {
+      "epoch": 16.06,
+      "learning_rate": 0.00040724999999999995,
+      "loss": 0.3797,
+      "step": 514
+    },
+    {
+      "epoch": 16.09,
+      "learning_rate": 0.00040687499999999996,
+      "loss": 0.3571,
+      "step": 515
+    },
+    {
+      "epoch": 16.12,
+      "learning_rate": 0.00040649999999999996,
+      "loss": 0.3525,
+      "step": 516
+    },
+    {
+      "epoch": 16.16,
+      "learning_rate": 0.00040612499999999996,
+      "loss": 0.3836,
+      "step": 517
+    },
+    {
+      "epoch": 16.19,
+      "learning_rate": 0.00040574999999999997,
+      "loss": 0.3514,
+      "step": 518
+    },
+    {
+      "epoch": 16.22,
+      "learning_rate": 0.000405375,
+      "loss": 0.4191,
+      "step": 519
+    },
+    {
+      "epoch": 16.25,
+      "learning_rate": 0.000405,
+      "loss": 0.4009,
+      "step": 520
+    },
+    {
+      "epoch": 16.25,
+      "eval_loss": 1.114495038986206,
+      "eval_mean_accuracy": 0.5029868268736565,
+      "eval_mean_iou": 0.3926742613896017,
+      "eval_overall_accuracy": 0.6919309047230503,
+      "eval_per_category_accuracy": [
+        0.623267003901118,
+        0.8475502832594376,
+        0.750294412315352,
+        0.3323398172115391,
+        0.502444252208741,
+        0.44843924090600606,
+        0.5559498637327619,
+        0.49885086535999257,
+        0.41982467192532585,
+        0.5013834118874849,
+        0.20419144631526753,
+        0.351306653460852
+      ],
+      "eval_per_category_iou": [
+        0.5551447983466938,
+        0.5853227631905641,
+        0.5876847661686672,
+        0.24404481327377933,
+        0.3901884156471832,
+        0.3531714330221185,
+        0.4531918484479208,
+        0.41972168983814523,
+        0.25107187960987826,
+        0.40174525218856516,
+        0.16531161953944093,
+        0.3054918574022637
+      ],
+      "eval_runtime": 121.6784,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 520
+    },
+    {
+      "epoch": 16.28,
+      "learning_rate": 0.00040462499999999993,
+      "loss": 0.3151,
+      "step": 521
+    },
+    {
+      "epoch": 16.31,
+      "learning_rate": 0.00040424999999999993,
+      "loss": 0.3798,
+      "step": 522
+    },
+    {
+      "epoch": 16.34,
+      "learning_rate": 0.00040387499999999994,
+      "loss": 0.3264,
+      "step": 523
+    },
+    {
+      "epoch": 16.38,
+      "learning_rate": 0.00040349999999999994,
+      "loss": 0.337,
+      "step": 524
+    },
+    {
+      "epoch": 16.41,
+      "learning_rate": 0.00040312499999999995,
+      "loss": 0.3952,
+      "step": 525
+    },
+    {
+      "epoch": 16.44,
+      "learning_rate": 0.00040274999999999995,
+      "loss": 0.3929,
+      "step": 526
+    },
+    {
+      "epoch": 16.47,
+      "learning_rate": 0.00040237499999999995,
+      "loss": 0.3253,
+      "step": 527
+    },
+    {
+      "epoch": 16.5,
+      "learning_rate": 0.000402,
+      "loss": 0.3569,
+      "step": 528
+    },
+    {
+      "epoch": 16.53,
+      "learning_rate": 0.000401625,
+      "loss": 0.4118,
+      "step": 529
+    },
+    {
+      "epoch": 16.56,
+      "learning_rate": 0.0004012499999999999,
+      "loss": 0.3638,
+      "step": 530
+    },
+    {
+      "epoch": 16.59,
+      "learning_rate": 0.00040087499999999997,
+      "loss": 0.402,
+      "step": 531
+    },
+    {
+      "epoch": 16.62,
+      "learning_rate": 0.0004005,
+      "loss": 0.3996,
+      "step": 532
+    },
+    {
+      "epoch": 16.66,
+      "learning_rate": 0.000400125,
+      "loss": 0.4078,
+      "step": 533
+    },
+    {
+      "epoch": 16.69,
+      "learning_rate": 0.00039975,
+      "loss": 0.3869,
+      "step": 534
+    },
+    {
+      "epoch": 16.72,
+      "learning_rate": 0.000399375,
+      "loss": 0.4146,
+      "step": 535
+    },
+    {
+      "epoch": 16.75,
+      "learning_rate": 0.000399,
+      "loss": 0.3847,
+      "step": 536
+    },
+    {
+      "epoch": 16.78,
+      "learning_rate": 0.000398625,
+      "loss": 0.3444,
+      "step": 537
+    },
+    {
+      "epoch": 16.81,
+      "learning_rate": 0.00039824999999999995,
+      "loss": 0.3481,
+      "step": 538
+    },
+    {
+      "epoch": 16.84,
+      "learning_rate": 0.00039787499999999995,
+      "loss": 0.4012,
+      "step": 539
+    },
+    {
+      "epoch": 16.88,
+      "learning_rate": 0.00039749999999999996,
+      "loss": 0.3792,
+      "step": 540
+    },
+    {
+      "epoch": 16.88,
+      "eval_loss": 1.0373446941375732,
+      "eval_mean_accuracy": 0.5319010549581172,
+      "eval_mean_iou": 0.4043940051436823,
+      "eval_overall_accuracy": 0.6962674458821615,
+      "eval_per_category_accuracy": [
+        0.6613369350430882,
+        0.8117236276546326,
+        0.7599339009753059,
+        0.3705352495818905,
+        0.4854246394398783,
+        0.6103346531706946,
+        0.6136552210583658,
+        0.48955921445263156,
+        0.3992861769429385,
+        0.5614743389481689,
+        0.2739009467024575,
+        0.34564775552735566
+      ],
+      "eval_per_category_iou": [
+        0.5810657837731416,
+        0.5831931362481007,
+        0.5912954319035167,
+        0.26070132951843406,
+        0.3922197395667084,
+        0.387007915239396,
+        0.4639126789011901,
+        0.4238089761118109,
+        0.25894049071282255,
+        0.42081577082396004,
+        0.19016966873794,
+        0.2995971401871666
+      ],
+      "eval_runtime": 121.514,
+      "eval_samples_per_second": 0.938,
+      "eval_steps_per_second": 0.082,
+      "step": 540
+    },
+    {
+      "epoch": 16.91,
+      "learning_rate": 0.00039712499999999996,
+      "loss": 0.4063,
+      "step": 541
+    },
+    {
+      "epoch": 16.94,
+      "learning_rate": 0.00039674999999999997,
+      "loss": 0.3643,
+      "step": 542
+    },
+    {
+      "epoch": 16.97,
+      "learning_rate": 0.00039637499999999997,
+      "loss": 0.412,
+      "step": 543
+    },
+    {
+      "epoch": 17.0,
+      "learning_rate": 0.000396,
+      "loss": 0.2641,
+      "step": 544
+    },
+    {
+      "epoch": 17.03,
+      "learning_rate": 0.000395625,
+      "loss": 0.3285,
+      "step": 545
+    },
+    {
+      "epoch": 17.06,
+      "learning_rate": 0.00039524999999999993,
+      "loss": 0.3306,
+      "step": 546
+    },
+    {
+      "epoch": 17.09,
+      "learning_rate": 0.00039487499999999994,
+      "loss": 0.3111,
+      "step": 547
+    },
+    {
+      "epoch": 17.12,
+      "learning_rate": 0.00039449999999999994,
+      "loss": 0.346,
+      "step": 548
+    },
+    {
+      "epoch": 17.16,
+      "learning_rate": 0.00039412499999999994,
+      "loss": 0.3833,
+      "step": 549
+    },
+    {
+      "epoch": 17.19,
+      "learning_rate": 0.00039374999999999995,
+      "loss": 0.3654,
+      "step": 550
+    },
+    {
+      "epoch": 17.22,
+      "learning_rate": 0.00039337499999999995,
+      "loss": 0.3156,
+      "step": 551
+    },
+    {
+      "epoch": 17.25,
+      "learning_rate": 0.00039299999999999996,
+      "loss": 0.3768,
+      "step": 552
+    },
+    {
+      "epoch": 17.28,
+      "learning_rate": 0.000392625,
+      "loss": 0.3759,
+      "step": 553
+    },
+    {
+      "epoch": 17.31,
+      "learning_rate": 0.00039225,
+      "loss": 0.3784,
+      "step": 554
+    },
+    {
+      "epoch": 17.34,
+      "learning_rate": 0.0003918749999999999,
+      "loss": 0.3681,
+      "step": 555
+    },
+    {
+      "epoch": 17.38,
+      "learning_rate": 0.0003914999999999999,
+      "loss": 0.3638,
+      "step": 556
+    },
+    {
+      "epoch": 17.41,
+      "learning_rate": 0.000391125,
+      "loss": 0.3517,
+      "step": 557
+    },
+    {
+      "epoch": 17.44,
+      "learning_rate": 0.00039075,
+      "loss": 0.3914,
+      "step": 558
+    },
+    {
+      "epoch": 17.47,
+      "learning_rate": 0.000390375,
+      "loss": 0.374,
+      "step": 559
+    },
+    {
+      "epoch": 17.5,
+      "learning_rate": 0.00039,
+      "loss": 0.3532,
+      "step": 560
+    },
+    {
+      "epoch": 17.5,
+      "eval_loss": 1.0606136322021484,
+      "eval_mean_accuracy": 0.5364630707393,
+      "eval_mean_iou": 0.4027402888976221,
+      "eval_overall_accuracy": 0.692054413912589,
+      "eval_per_category_accuracy": [
+        0.6916576396319642,
+        0.785844562950018,
+        0.7279213591676157,
+        0.39280782229130956,
+        0.5252518704151278,
+        0.4992015649327318,
+        0.586307307419204,
+        0.45152687582167145,
+        0.4356427956664862,
+        0.6498259847780361,
+        0.35218066009004845,
+        0.33938840570738804
+      ],
+      "eval_per_category_iou": [
+        0.5901385584766586,
+        0.5730590545148666,
+        0.5915173876880175,
+        0.2577443047542723,
+        0.38687015793054874,
+        0.3762150792912744,
+        0.46806468824491687,
+        0.4018600863386738,
+        0.2583479349845038,
+        0.42879296228779334,
+        0.21245222400709024,
+        0.2878210282528486
+      ],
+      "eval_runtime": 121.6974,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 560
+    },
+    {
+      "epoch": 17.53,
+      "learning_rate": 0.000389625,
+      "loss": 0.3558,
+      "step": 561
+    },
+    {
+      "epoch": 17.56,
+      "learning_rate": 0.00038925,
+      "loss": 0.3454,
+      "step": 562
+    },
+    {
+      "epoch": 17.59,
+      "learning_rate": 0.00038887499999999995,
+      "loss": 0.3523,
+      "step": 563
+    },
+    {
+      "epoch": 17.62,
+      "learning_rate": 0.00038849999999999996,
+      "loss": 0.3781,
+      "step": 564
+    },
+    {
+      "epoch": 17.66,
+      "learning_rate": 0.00038812499999999996,
+      "loss": 0.3385,
+      "step": 565
+    },
+    {
+      "epoch": 17.69,
+      "learning_rate": 0.00038774999999999997,
+      "loss": 0.3942,
+      "step": 566
+    },
+    {
+      "epoch": 17.72,
+      "learning_rate": 0.00038737499999999997,
+      "loss": 0.4192,
+      "step": 567
+    },
+    {
+      "epoch": 17.75,
+      "learning_rate": 0.000387,
+      "loss": 0.3901,
+      "step": 568
+    },
+    {
+      "epoch": 17.78,
+      "learning_rate": 0.000386625,
+      "loss": 0.3555,
+      "step": 569
+    },
+    {
+      "epoch": 17.81,
+      "learning_rate": 0.00038625,
+      "loss": 0.3333,
+      "step": 570
+    },
+    {
+      "epoch": 17.84,
+      "learning_rate": 0.00038587499999999993,
+      "loss": 0.365,
+      "step": 571
+    },
+    {
+      "epoch": 17.88,
+      "learning_rate": 0.00038549999999999994,
+      "loss": 0.356,
+      "step": 572
+    },
+    {
+      "epoch": 17.91,
+      "learning_rate": 0.00038512499999999994,
+      "loss": 0.3568,
+      "step": 573
+    },
+    {
+      "epoch": 17.94,
+      "learning_rate": 0.00038474999999999995,
+      "loss": 0.3554,
+      "step": 574
+    },
+    {
+      "epoch": 17.97,
+      "learning_rate": 0.00038437499999999995,
+      "loss": 0.4282,
+      "step": 575
+    },
+    {
+      "epoch": 18.0,
+      "learning_rate": 0.00038399999999999996,
+      "loss": 0.3961,
+      "step": 576
+    },
+    {
+      "epoch": 18.03,
+      "learning_rate": 0.00038362499999999996,
+      "loss": 0.3528,
+      "step": 577
+    },
+    {
+      "epoch": 18.06,
+      "learning_rate": 0.00038324999999999996,
+      "loss": 0.3347,
+      "step": 578
+    },
+    {
+      "epoch": 18.09,
+      "learning_rate": 0.000382875,
+      "loss": 0.3684,
+      "step": 579
+    },
+    {
+      "epoch": 18.12,
+      "learning_rate": 0.0003824999999999999,
+      "loss": 0.3265,
+      "step": 580
+    },
+    {
+      "epoch": 18.12,
+      "eval_loss": 1.0410232543945312,
+      "eval_mean_accuracy": 0.5431988680928329,
+      "eval_mean_iou": 0.40551240287672713,
+      "eval_overall_accuracy": 0.6920698065506784,
+      "eval_per_category_accuracy": [
+        0.6552881242989272,
+        0.8041343896538069,
+        0.717873816608022,
+        0.38966419386402595,
+        0.5272575145814383,
+        0.5582077310290013,
+        0.6118446222854794,
+        0.5218807021453727,
+        0.3890490936354844,
+        0.5468698904379509,
+        0.2847973603743959,
+        0.5115189782000887
+      ],
+      "eval_per_category_iou": [
+        0.5724660939154029,
+        0.5789520770202325,
+        0.5912823548960956,
+        0.2570380949765168,
+        0.38400635048041737,
+        0.37546106319769756,
+        0.4716700600492167,
+        0.4283773134874427,
+        0.23948429522760076,
+        0.4078138849639129,
+        0.19265545964457492,
+        0.36694178666161426
+      ],
+      "eval_runtime": 120.5664,
+      "eval_samples_per_second": 0.946,
+      "eval_steps_per_second": 0.083,
+      "step": 580
+    },
+    {
+      "epoch": 18.16,
+      "learning_rate": 0.0003821249999999999,
+      "loss": 0.4198,
+      "step": 581
+    },
+    {
+      "epoch": 18.19,
+      "learning_rate": 0.00038175,
+      "loss": 0.4146,
+      "step": 582
+    },
+    {
+      "epoch": 18.22,
+      "learning_rate": 0.000381375,
+      "loss": 0.3704,
+      "step": 583
+    },
+    {
+      "epoch": 18.25,
+      "learning_rate": 0.000381,
+      "loss": 0.3398,
+      "step": 584
+    },
+    {
+      "epoch": 18.28,
+      "learning_rate": 0.000380625,
+      "loss": 0.3956,
+      "step": 585
+    },
+    {
+      "epoch": 18.31,
+      "learning_rate": 0.00038025,
+      "loss": 0.3259,
+      "step": 586
+    },
+    {
+      "epoch": 18.34,
+      "learning_rate": 0.000379875,
+      "loss": 0.3287,
+      "step": 587
+    },
+    {
+      "epoch": 18.38,
+      "learning_rate": 0.00037949999999999995,
+      "loss": 0.357,
+      "step": 588
+    },
+    {
+      "epoch": 18.41,
+      "learning_rate": 0.00037912499999999996,
+      "loss": 0.3232,
+      "step": 589
+    },
+    {
+      "epoch": 18.44,
+      "learning_rate": 0.00037874999999999996,
+      "loss": 0.322,
+      "step": 590
+    },
+    {
+      "epoch": 18.47,
+      "learning_rate": 0.00037837499999999997,
+      "loss": 0.4049,
+      "step": 591
+    },
+    {
+      "epoch": 18.5,
+      "learning_rate": 0.00037799999999999997,
+      "loss": 0.316,
+      "step": 592
+    },
+    {
+      "epoch": 18.53,
+      "learning_rate": 0.000377625,
+      "loss": 0.3508,
+      "step": 593
+    },
+    {
+      "epoch": 18.56,
+      "learning_rate": 0.00037725,
+      "loss": 0.3187,
+      "step": 594
+    },
+    {
+      "epoch": 18.59,
+      "learning_rate": 0.000376875,
+      "loss": 0.3259,
+      "step": 595
+    },
+    {
+      "epoch": 18.62,
+      "learning_rate": 0.00037649999999999994,
+      "loss": 0.3481,
+      "step": 596
+    },
+    {
+      "epoch": 18.66,
+      "learning_rate": 0.00037612499999999994,
+      "loss": 0.3535,
+      "step": 597
+    },
+    {
+      "epoch": 18.69,
+      "learning_rate": 0.00037574999999999994,
+      "loss": 0.3775,
+      "step": 598
+    },
+    {
+      "epoch": 18.72,
+      "learning_rate": 0.00037537499999999995,
+      "loss": 0.3696,
+      "step": 599
+    },
+    {
+      "epoch": 18.75,
+      "learning_rate": 0.00037499999999999995,
+      "loss": 0.3318,
+      "step": 600
+    },
+    {
+      "epoch": 18.75,
+      "eval_loss": 1.020988941192627,
+      "eval_mean_accuracy": 0.5447569844160918,
+      "eval_mean_iou": 0.4094633881675768,
+      "eval_overall_accuracy": 0.6994912331564385,
+      "eval_per_category_accuracy": [
+        0.7262703523105905,
+        0.7806017935156169,
+        0.7217685912067523,
+        0.40382093115245554,
+        0.563411783058182,
+        0.4817557587129227,
+        0.6108267508883742,
+        0.5442548089543571,
+        0.3793466012439775,
+        0.5605418072027125,
+        0.24519073315111428,
+        0.5192939015960465
+      ],
+      "eval_per_category_iou": [
+        0.6038722020699768,
+        0.5815322915495976,
+        0.5825486043697007,
+        0.2691492043375801,
+        0.39273292144748456,
+        0.36467870059904095,
+        0.4707391191189612,
+        0.4129633815000236,
+        0.25210178181566234,
+        0.4278171656124102,
+        0.1848999988459186,
+        0.3705252867445649
+      ],
+      "eval_runtime": 120.934,
+      "eval_samples_per_second": 0.943,
+      "eval_steps_per_second": 0.083,
+      "step": 600
+    },
+    {
+      "epoch": 18.78,
+      "learning_rate": 0.00037462499999999996,
+      "loss": 0.4432,
+      "step": 601
+    },
+    {
+      "epoch": 18.81,
+      "learning_rate": 0.00037424999999999996,
+      "loss": 0.3258,
+      "step": 602
+    },
+    {
+      "epoch": 18.84,
+      "learning_rate": 0.00037387499999999997,
+      "loss": 0.3648,
+      "step": 603
+    },
+    {
+      "epoch": 18.88,
+      "learning_rate": 0.0003735,
+      "loss": 0.2878,
+      "step": 604
+    },
+    {
+      "epoch": 18.91,
+      "learning_rate": 0.0003731249999999999,
+      "loss": 0.3666,
+      "step": 605
+    },
+    {
+      "epoch": 18.94,
+      "learning_rate": 0.0003727499999999999,
+      "loss": 0.2713,
+      "step": 606
+    },
+    {
+      "epoch": 18.97,
+      "learning_rate": 0.00037237499999999993,
+      "loss": 0.3096,
+      "step": 607
+    },
+    {
+      "epoch": 19.0,
+      "learning_rate": 0.000372,
+      "loss": 0.3383,
+      "step": 608
+    },
+    {
+      "epoch": 19.03,
+      "learning_rate": 0.000371625,
+      "loss": 0.3132,
+      "step": 609
+    },
+    {
+      "epoch": 19.06,
+      "learning_rate": 0.00037125,
+      "loss": 0.3102,
+      "step": 610
+    },
+    {
+      "epoch": 19.09,
+      "learning_rate": 0.000370875,
+      "loss": 0.3577,
+      "step": 611
+    },
+    {
+      "epoch": 19.12,
+      "learning_rate": 0.0003705,
+      "loss": 0.3291,
+      "step": 612
+    },
+    {
+      "epoch": 19.16,
+      "learning_rate": 0.00037012499999999996,
+      "loss": 0.2622,
+      "step": 613
+    },
+    {
+      "epoch": 19.19,
+      "learning_rate": 0.00036974999999999996,
+      "loss": 0.3496,
+      "step": 614
+    },
+    {
+      "epoch": 19.22,
+      "learning_rate": 0.00036937499999999997,
+      "loss": 0.3232,
+      "step": 615
+    },
+    {
+      "epoch": 19.25,
+      "learning_rate": 0.00036899999999999997,
+      "loss": 0.2898,
+      "step": 616
+    },
+    {
+      "epoch": 19.28,
+      "learning_rate": 0.000368625,
+      "loss": 0.2997,
+      "step": 617
+    },
+    {
+      "epoch": 19.31,
+      "learning_rate": 0.00036825,
+      "loss": 0.2981,
+      "step": 618
+    },
+    {
+      "epoch": 19.34,
+      "learning_rate": 0.000367875,
+      "loss": 0.3469,
+      "step": 619
+    },
+    {
+      "epoch": 19.38,
+      "learning_rate": 0.0003675,
+      "loss": 0.3084,
+      "step": 620
+    },
+    {
+      "epoch": 19.38,
+      "eval_loss": 1.059301495552063,
+      "eval_mean_accuracy": 0.5508115063471106,
+      "eval_mean_iou": 0.41124752889527083,
+      "eval_overall_accuracy": 0.6958786479213781,
+      "eval_per_category_accuracy": [
+        0.6755053046952768,
+        0.7849317329960662,
+        0.7434028670494764,
+        0.3357124516831944,
+        0.588747728456039,
+        0.6217468307571705,
+        0.612745198370776,
+        0.531245211939,
+        0.41738906683939203,
+        0.598762570011594,
+        0.2131779268042594,
+        0.48637118656308354
+      ],
+      "eval_per_category_iou": [
+        0.5804648202527455,
+        0.5775350994367053,
+        0.5956387213067603,
+        0.25720141635830035,
+        0.3986168225324602,
+        0.3902167636805523,
+        0.4700284540375506,
+        0.4395014781032412,
+        0.25725742832806736,
+        0.43049174672456336,
+        0.1754552141799091,
+        0.36256238180239475
+      ],
+      "eval_runtime": 120.6095,
+      "eval_samples_per_second": 0.945,
+      "eval_steps_per_second": 0.083,
+      "step": 620
+    },
+    {
+      "epoch": 19.41,
+      "learning_rate": 0.00036712499999999994,
+      "loss": 0.3575,
+      "step": 621
+    },
+    {
+      "epoch": 19.44,
+      "learning_rate": 0.00036674999999999994,
+      "loss": 0.3358,
+      "step": 622
+    },
+    {
+      "epoch": 19.47,
+      "learning_rate": 0.00036637499999999995,
+      "loss": 0.3695,
+      "step": 623
+    },
+    {
+      "epoch": 19.5,
+      "learning_rate": 0.00036599999999999995,
+      "loss": 0.3881,
+      "step": 624
+    },
+    {
+      "epoch": 19.53,
+      "learning_rate": 0.00036562499999999996,
+      "loss": 0.3451,
+      "step": 625
+    },
+    {
+      "epoch": 19.56,
+      "learning_rate": 0.00036524999999999996,
+      "loss": 0.4005,
+      "step": 626
+    },
+    {
+      "epoch": 19.59,
+      "learning_rate": 0.00036487499999999996,
+      "loss": 0.3328,
+      "step": 627
+    },
+    {
+      "epoch": 19.62,
+      "learning_rate": 0.00036449999999999997,
+      "loss": 0.3266,
+      "step": 628
+    },
+    {
+      "epoch": 19.66,
+      "learning_rate": 0.000364125,
+      "loss": 0.386,
+      "step": 629
+    },
+    {
+      "epoch": 19.69,
+      "learning_rate": 0.0003637499999999999,
+      "loss": 0.305,
+      "step": 630
+    },
+    {
+      "epoch": 19.72,
+      "learning_rate": 0.00036337499999999993,
+      "loss": 0.3171,
+      "step": 631
+    },
+    {
+      "epoch": 19.75,
+      "learning_rate": 0.00036299999999999993,
+      "loss": 0.3233,
+      "step": 632
+    },
+    {
+      "epoch": 19.78,
+      "learning_rate": 0.000362625,
+      "loss": 0.3538,
+      "step": 633
+    },
+    {
+      "epoch": 19.81,
+      "learning_rate": 0.00036225,
+      "loss": 0.3642,
+      "step": 634
+    },
+    {
+      "epoch": 19.84,
+      "learning_rate": 0.000361875,
+      "loss": 0.3367,
+      "step": 635
+    },
+    {
+      "epoch": 19.88,
+      "learning_rate": 0.0003615,
+      "loss": 0.3806,
+      "step": 636
+    },
+    {
+      "epoch": 19.91,
+      "learning_rate": 0.000361125,
+      "loss": 0.3214,
+      "step": 637
+    },
+    {
+      "epoch": 19.94,
+      "learning_rate": 0.00036074999999999996,
+      "loss": 0.4018,
+      "step": 638
+    },
+    {
+      "epoch": 19.97,
+      "learning_rate": 0.00036037499999999996,
+      "loss": 0.3683,
+      "step": 639
+    },
+    {
+      "epoch": 20.0,
+      "learning_rate": 0.00035999999999999997,
+      "loss": 0.341,
+      "step": 640
+    },
+    {
+      "epoch": 20.0,
+      "eval_loss": 1.1016860008239746,
+      "eval_mean_accuracy": 0.5467457135144904,
+      "eval_mean_iou": 0.3945612352121604,
+      "eval_overall_accuracy": 0.6805642111259594,
+      "eval_per_category_accuracy": [
+        0.5975676711895084,
+        0.7684589156993332,
+        0.7768818347244737,
+        0.41959458990569115,
+        0.5867744987409012,
+        0.5595941046458033,
+        0.7348700856019611,
+        0.48536768558390503,
+        0.4138266062127894,
+        0.665171556112803,
+        0.0968617716466746,
+        0.4559792421100416
+      ],
+      "eval_per_category_iou": [
+        0.5414112604633725,
+        0.5656503370160666,
+        0.5831030578837733,
+        0.2749207143100724,
+        0.38855618875902853,
+        0.39129750690271237,
+        0.4500853398648062,
+        0.4121934929699471,
+        0.2524171747266218,
+        0.43195502753202136,
+        0.09078521204540695,
+        0.3523595100720956
+      ],
+      "eval_runtime": 120.0985,
+      "eval_samples_per_second": 0.949,
+      "eval_steps_per_second": 0.083,
+      "step": 640
+    },
+    {
+      "epoch": 20.03,
+      "learning_rate": 0.000359625,
+      "loss": 0.3105,
+      "step": 641
+    },
+    {
+      "epoch": 20.06,
+      "learning_rate": 0.00035925,
+      "loss": 0.3282,
+      "step": 642
+    },
+    {
+      "epoch": 20.09,
+      "learning_rate": 0.000358875,
+      "loss": 0.3913,
+      "step": 643
+    },
+    {
+      "epoch": 20.12,
+      "learning_rate": 0.0003585,
+      "loss": 0.297,
+      "step": 644
+    },
+    {
+      "epoch": 20.16,
+      "learning_rate": 0.000358125,
+      "loss": 0.3105,
+      "step": 645
+    },
+    {
+      "epoch": 20.19,
+      "learning_rate": 0.00035774999999999994,
+      "loss": 0.3215,
+      "step": 646
+    },
+    {
+      "epoch": 20.22,
+      "learning_rate": 0.00035737499999999995,
+      "loss": 0.2984,
+      "step": 647
+    },
+    {
+      "epoch": 20.25,
+      "learning_rate": 0.00035699999999999995,
+      "loss": 0.3547,
+      "step": 648
+    },
+    {
+      "epoch": 20.28,
+      "learning_rate": 0.00035662499999999995,
+      "loss": 0.3373,
+      "step": 649
+    },
+    {
+      "epoch": 20.31,
+      "learning_rate": 0.00035624999999999996,
+      "loss": 0.3341,
+      "step": 650
+    },
+    {
+      "epoch": 20.34,
+      "learning_rate": 0.00035587499999999996,
+      "loss": 0.3409,
+      "step": 651
+    },
+    {
+      "epoch": 20.38,
+      "learning_rate": 0.00035549999999999997,
+      "loss": 0.3026,
+      "step": 652
+    },
+    {
+      "epoch": 20.41,
+      "learning_rate": 0.00035512499999999997,
+      "loss": 0.3091,
+      "step": 653
+    },
+    {
+      "epoch": 20.44,
+      "learning_rate": 0.00035475,
+      "loss": 0.2992,
+      "step": 654
+    },
+    {
+      "epoch": 20.47,
+      "learning_rate": 0.0003543749999999999,
+      "loss": 0.2883,
+      "step": 655
+    },
+    {
+      "epoch": 20.5,
+      "learning_rate": 0.00035399999999999993,
+      "loss": 0.3174,
+      "step": 656
+    },
+    {
+      "epoch": 20.53,
+      "learning_rate": 0.00035362499999999994,
+      "loss": 0.3571,
+      "step": 657
+    },
+    {
+      "epoch": 20.56,
+      "learning_rate": 0.00035324999999999994,
+      "loss": 0.2673,
+      "step": 658
+    },
+    {
+      "epoch": 20.59,
+      "learning_rate": 0.000352875,
+      "loss": 0.3882,
+      "step": 659
+    },
+    {
+      "epoch": 20.62,
+      "learning_rate": 0.0003525,
+      "loss": 0.3189,
+      "step": 660
+    },
+    {
+      "epoch": 20.62,
+      "eval_loss": 1.0981378555297852,
+      "eval_mean_accuracy": 0.5360663301848262,
+      "eval_mean_iou": 0.40852628331603835,
+      "eval_overall_accuracy": 0.700901801126045,
+      "eval_per_category_accuracy": [
+        0.7290374192798078,
+        0.780043876633833,
+        0.740563896247017,
+        0.3473848203690467,
+        0.5824512213157431,
+        0.5583347547897031,
+        0.6493192935830805,
+        0.49992981974925826,
+        0.300684098914847,
+        0.6107290176445646,
+        0.265603144043855,
+        0.36871459964715714
+      ],
+      "eval_per_category_iou": [
+        0.6096553067262501,
+        0.5804813346089412,
+        0.5902485731408187,
+        0.2537729966490416,
+        0.4024485499354687,
+        0.39114301078139363,
+        0.47082317523904227,
+        0.434038016814915,
+        0.23113951549234293,
+        0.4207477097579975,
+        0.2067764219215033,
+        0.3110407887247453
+      ],
+      "eval_runtime": 120.9446,
+      "eval_samples_per_second": 0.943,
+      "eval_steps_per_second": 0.083,
+      "step": 660
+    },
+    {
+      "epoch": 20.66,
+      "learning_rate": 0.000352125,
+      "loss": 0.3284,
+      "step": 661
+    },
+    {
+      "epoch": 20.69,
+      "learning_rate": 0.00035175,
+      "loss": 0.3129,
+      "step": 662
+    },
+    {
+      "epoch": 20.72,
+      "learning_rate": 0.00035137499999999996,
+      "loss": 0.2793,
+      "step": 663
+    },
+    {
+      "epoch": 20.75,
+      "learning_rate": 0.00035099999999999997,
+      "loss": 0.3778,
+      "step": 664
+    },
+    {
+      "epoch": 20.78,
+      "learning_rate": 0.00035062499999999997,
+      "loss": 0.3155,
+      "step": 665
+    },
+    {
+      "epoch": 20.81,
+      "learning_rate": 0.00035025,
+      "loss": 0.3472,
+      "step": 666
+    },
+    {
+      "epoch": 20.84,
+      "learning_rate": 0.000349875,
+      "loss": 0.3406,
+      "step": 667
+    },
+    {
+      "epoch": 20.88,
+      "learning_rate": 0.0003495,
+      "loss": 0.3245,
+      "step": 668
+    },
+    {
+      "epoch": 20.91,
+      "learning_rate": 0.000349125,
+      "loss": 0.3099,
+      "step": 669
+    },
+    {
+      "epoch": 20.94,
+      "learning_rate": 0.00034875,
+      "loss": 0.3515,
+      "step": 670
+    },
+    {
+      "epoch": 20.97,
+      "learning_rate": 0.00034837499999999994,
+      "loss": 0.3123,
+      "step": 671
+    },
+    {
+      "epoch": 21.0,
+      "learning_rate": 0.00034799999999999995,
+      "loss": 0.3299,
+      "step": 672
+    },
+    {
+      "epoch": 21.03,
+      "learning_rate": 0.00034762499999999995,
+      "loss": 0.3696,
+      "step": 673
+    },
+    {
+      "epoch": 21.06,
+      "learning_rate": 0.00034724999999999996,
+      "loss": 0.294,
+      "step": 674
+    },
+    {
+      "epoch": 21.09,
+      "learning_rate": 0.00034687499999999996,
+      "loss": 0.3386,
+      "step": 675
+    },
+    {
+      "epoch": 21.12,
+      "learning_rate": 0.00034649999999999997,
+      "loss": 0.3054,
+      "step": 676
+    },
+    {
+      "epoch": 21.16,
+      "learning_rate": 0.00034612499999999997,
+      "loss": 0.2931,
+      "step": 677
+    },
+    {
+      "epoch": 21.19,
+      "learning_rate": 0.00034575,
+      "loss": 0.3533,
+      "step": 678
+    },
+    {
+      "epoch": 21.22,
+      "learning_rate": 0.000345375,
+      "loss": 0.3265,
+      "step": 679
+    },
+    {
+      "epoch": 21.25,
+      "learning_rate": 0.00034499999999999993,
+      "loss": 0.306,
+      "step": 680
+    },
+    {
+      "epoch": 21.25,
+      "eval_loss": 1.079343318939209,
+      "eval_mean_accuracy": 0.5312425297991711,
+      "eval_mean_iou": 0.4073887918342692,
+      "eval_overall_accuracy": 0.7015380190129865,
+      "eval_per_category_accuracy": [
+        0.7030279589180716,
+        0.7993058741104738,
+        0.7431397721575408,
+        0.36075738544493097,
+        0.5635799330236403,
+        0.5524971056728811,
+        0.6310424797960793,
+        0.47471957364694084,
+        0.39750272841006895,
+        0.6082287060183241,
+        0.17880525109193696,
+        0.3623035892991633
+      ],
+      "eval_per_category_iou": [
+        0.5989279646164274,
+        0.5844779029725281,
+        0.5925695356010946,
+        0.2737699953920084,
+        0.40047996498875665,
+        0.38667738551533026,
+        0.47530902628924754,
+        0.41930446682319805,
+        0.2601354686238873,
+        0.4325911758661698,
+        0.1533369380909134,
+        0.3110856772316679
+      ],
+      "eval_runtime": 120.2944,
+      "eval_samples_per_second": 0.948,
+      "eval_steps_per_second": 0.083,
+      "step": 680
+    },
+    {
+      "epoch": 21.28,
+      "learning_rate": 0.00034462499999999993,
+      "loss": 0.2736,
+      "step": 681
+    },
+    {
+      "epoch": 21.31,
+      "learning_rate": 0.00034424999999999994,
+      "loss": 0.3526,
+      "step": 682
+    },
+    {
+      "epoch": 21.34,
+      "learning_rate": 0.00034387499999999994,
+      "loss": 0.3021,
+      "step": 683
+    },
+    {
+      "epoch": 21.38,
+      "learning_rate": 0.0003435,
+      "loss": 0.3016,
+      "step": 684
+    },
+    {
+      "epoch": 21.41,
+      "learning_rate": 0.000343125,
+      "loss": 0.3366,
+      "step": 685
+    },
+    {
+      "epoch": 21.44,
+      "learning_rate": 0.00034275,
+      "loss": 0.2822,
+      "step": 686
+    },
+    {
+      "epoch": 21.47,
+      "learning_rate": 0.000342375,
+      "loss": 0.3038,
+      "step": 687
+    },
+    {
+      "epoch": 21.5,
+      "learning_rate": 0.00034199999999999996,
+      "loss": 0.3254,
+      "step": 688
+    },
+    {
+      "epoch": 21.53,
+      "learning_rate": 0.00034162499999999997,
+      "loss": 0.2837,
+      "step": 689
+    },
+    {
+      "epoch": 21.56,
+      "learning_rate": 0.00034125,
+      "loss": 0.2991,
+      "step": 690
+    },
+    {
+      "epoch": 21.59,
+      "learning_rate": 0.000340875,
+      "loss": 0.359,
+      "step": 691
+    },
+    {
+      "epoch": 21.62,
+      "learning_rate": 0.0003405,
+      "loss": 0.3301,
+      "step": 692
+    },
+    {
+      "epoch": 21.66,
+      "learning_rate": 0.000340125,
+      "loss": 0.33,
+      "step": 693
+    },
+    {
+      "epoch": 21.69,
+      "learning_rate": 0.00033975,
+      "loss": 0.2849,
+      "step": 694
+    },
+    {
+      "epoch": 21.72,
+      "learning_rate": 0.000339375,
+      "loss": 0.3286,
+      "step": 695
+    },
+    {
+      "epoch": 21.75,
+      "learning_rate": 0.00033899999999999995,
+      "loss": 0.4095,
+      "step": 696
+    },
+    {
+      "epoch": 21.78,
+      "learning_rate": 0.00033862499999999995,
+      "loss": 0.3631,
+      "step": 697
+    },
+    {
+      "epoch": 21.81,
+      "learning_rate": 0.00033824999999999995,
+      "loss": 0.3216,
+      "step": 698
+    },
+    {
+      "epoch": 21.84,
+      "learning_rate": 0.00033787499999999996,
+      "loss": 0.3262,
+      "step": 699
+    },
+    {
+      "epoch": 21.88,
+      "learning_rate": 0.00033749999999999996,
+      "loss": 0.3057,
+      "step": 700
+    },
+    {
+      "epoch": 21.88,
+      "eval_loss": 1.0966832637786865,
+      "eval_mean_accuracy": 0.5436162135374909,
+      "eval_mean_iou": 0.4082507168437561,
+      "eval_overall_accuracy": 0.6936197448195073,
+      "eval_per_category_accuracy": [
+        0.6474889718930664,
+        0.7922275007344841,
+        0.7846337174081215,
+        0.42367159145321687,
+        0.5340949378756784,
+        0.4965993924634988,
+        0.6263183914325615,
+        0.49843621258080106,
+        0.4197692164361197,
+        0.6275072873600746,
+        0.3027978684655952,
+        0.36984947434667326
+      ],
+      "eval_per_category_iou": [
+        0.5704268488701265,
+        0.5789043669843227,
+        0.5858298969786,
+        0.2762748833610129,
+        0.3890980578526424,
+        0.36870745916349157,
+        0.4758416227673552,
+        0.431528172019072,
+        0.25716126103966874,
+        0.440727063864797,
+        0.21065362928640785,
+        0.31385533993757636
+      ],
+      "eval_runtime": 121.1302,
+      "eval_samples_per_second": 0.941,
+      "eval_steps_per_second": 0.083,
+      "step": 700
+    },
+    {
+      "epoch": 21.91,
+      "learning_rate": 0.00033712499999999997,
+      "loss": 0.2623,
+      "step": 701
+    },
+    {
+      "epoch": 21.94,
+      "learning_rate": 0.00033674999999999997,
+      "loss": 0.303,
+      "step": 702
+    },
+    {
+      "epoch": 21.97,
+      "learning_rate": 0.000336375,
+      "loss": 0.365,
+      "step": 703
+    },
+    {
+      "epoch": 22.0,
+      "learning_rate": 0.000336,
+      "loss": 0.2765,
+      "step": 704
+    },
+    {
+      "epoch": 22.03,
+      "learning_rate": 0.00033562499999999993,
+      "loss": 0.3511,
+      "step": 705
+    },
+    {
+      "epoch": 22.06,
+      "learning_rate": 0.00033524999999999994,
+      "loss": 0.3092,
+      "step": 706
+    },
+    {
+      "epoch": 22.09,
+      "learning_rate": 0.00033487499999999994,
+      "loss": 0.3209,
+      "step": 707
+    },
+    {
+      "epoch": 22.12,
+      "learning_rate": 0.00033449999999999994,
+      "loss": 0.3255,
+      "step": 708
+    },
+    {
+      "epoch": 22.16,
+      "learning_rate": 0.00033412499999999995,
+      "loss": 0.3228,
+      "step": 709
+    },
+    {
+      "epoch": 22.19,
+      "learning_rate": 0.00033375,
+      "loss": 0.2559,
+      "step": 710
+    },
+    {
+      "epoch": 22.22,
+      "learning_rate": 0.000333375,
+      "loss": 0.3944,
+      "step": 711
+    },
+    {
+      "epoch": 22.25,
+      "learning_rate": 0.000333,
+      "loss": 0.3088,
+      "step": 712
+    },
+    {
+      "epoch": 22.28,
+      "learning_rate": 0.00033262499999999997,
+      "loss": 0.2867,
+      "step": 713
+    },
+    {
+      "epoch": 22.31,
+      "learning_rate": 0.00033224999999999997,
+      "loss": 0.3286,
+      "step": 714
+    },
+    {
+      "epoch": 22.34,
+      "learning_rate": 0.000331875,
+      "loss": 0.3108,
+      "step": 715
+    },
+    {
+      "epoch": 22.38,
+      "learning_rate": 0.0003315,
+      "loss": 0.3645,
+      "step": 716
+    },
+    {
+      "epoch": 22.41,
+      "learning_rate": 0.000331125,
+      "loss": 0.3516,
+      "step": 717
+    },
+    {
+      "epoch": 22.44,
+      "learning_rate": 0.00033075,
+      "loss": 0.2856,
+      "step": 718
+    },
+    {
+      "epoch": 22.47,
+      "learning_rate": 0.000330375,
+      "loss": 0.3411,
+      "step": 719
+    },
+    {
+      "epoch": 22.5,
+      "learning_rate": 0.00033,
+      "loss": 0.2965,
+      "step": 720
+    },
+    {
+      "epoch": 22.5,
+      "eval_loss": 1.1567984819412231,
+      "eval_mean_accuracy": 0.5276929257850457,
+      "eval_mean_iou": 0.39465744241374917,
+      "eval_overall_accuracy": 0.6813106871487802,
+      "eval_per_category_accuracy": [
+        0.5635599423738187,
+        0.8229473695443837,
+        0.7641445606557854,
+        0.3653617948522217,
+        0.6130585668355595,
+        0.5156565858190674,
+        0.7143796180108919,
+        0.4005974428673829,
+        0.38591696760512145,
+        0.5533073844727205,
+        0.24381643828069285,
+        0.3895684381029022
+      ],
+      "eval_per_category_iou": [
+        0.5167840053968858,
+        0.5760424859842056,
+        0.5774014822404548,
+        0.2784320776768857,
+        0.3913638956104286,
+        0.3730870830182338,
+        0.4646286273391432,
+        0.37300618148587117,
+        0.250395074308512,
+        0.42088621405136056,
+        0.1926386302628715,
+        0.32122355159013727
+      ],
+      "eval_runtime": 122.064,
+      "eval_samples_per_second": 0.934,
+      "eval_steps_per_second": 0.082,
+      "step": 720
+    },
+    {
+      "epoch": 22.53,
+      "learning_rate": 0.00032962499999999995,
+      "loss": 0.3295,
+      "step": 721
+    },
+    {
+      "epoch": 22.56,
+      "learning_rate": 0.00032924999999999995,
+      "loss": 0.2803,
+      "step": 722
+    },
+    {
+      "epoch": 22.59,
+      "learning_rate": 0.00032887499999999996,
+      "loss": 0.2649,
+      "step": 723
+    },
+    {
+      "epoch": 22.62,
+      "learning_rate": 0.00032849999999999996,
+      "loss": 0.3229,
+      "step": 724
+    },
+    {
+      "epoch": 22.66,
+      "learning_rate": 0.00032812499999999997,
+      "loss": 0.3463,
+      "step": 725
+    },
+    {
+      "epoch": 22.69,
+      "learning_rate": 0.00032774999999999997,
+      "loss": 0.2763,
+      "step": 726
+    },
+    {
+      "epoch": 22.72,
+      "learning_rate": 0.000327375,
+      "loss": 0.2874,
+      "step": 727
+    },
+    {
+      "epoch": 22.75,
+      "learning_rate": 0.000327,
+      "loss": 0.3032,
+      "step": 728
+    },
+    {
+      "epoch": 22.78,
+      "learning_rate": 0.000326625,
+      "loss": 0.3154,
+      "step": 729
+    },
+    {
+      "epoch": 22.81,
+      "learning_rate": 0.00032624999999999993,
+      "loss": 0.353,
+      "step": 730
+    },
+    {
+      "epoch": 22.84,
+      "learning_rate": 0.00032587499999999994,
+      "loss": 0.2839,
+      "step": 731
+    },
+    {
+      "epoch": 22.88,
+      "learning_rate": 0.00032549999999999994,
+      "loss": 0.3356,
+      "step": 732
+    },
+    {
+      "epoch": 22.91,
+      "learning_rate": 0.00032512499999999995,
+      "loss": 0.3061,
+      "step": 733
+    },
+    {
+      "epoch": 22.94,
+      "learning_rate": 0.00032474999999999995,
+      "loss": 0.3054,
+      "step": 734
+    },
+    {
+      "epoch": 22.97,
+      "learning_rate": 0.000324375,
+      "loss": 0.2964,
+      "step": 735
+    },
+    {
+      "epoch": 23.0,
+      "learning_rate": 0.000324,
+      "loss": 0.3063,
+      "step": 736
+    },
+    {
+      "epoch": 23.03,
+      "learning_rate": 0.000323625,
+      "loss": 0.3541,
+      "step": 737
+    },
+    {
+      "epoch": 23.06,
+      "learning_rate": 0.0003232499999999999,
+      "loss": 0.3322,
+      "step": 738
+    },
+    {
+      "epoch": 23.09,
+      "learning_rate": 0.000322875,
+      "loss": 0.2977,
+      "step": 739
+    },
+    {
+      "epoch": 23.12,
+      "learning_rate": 0.0003225,
+      "loss": 0.3143,
+      "step": 740
+    },
+    {
+      "epoch": 23.12,
+      "eval_loss": 1.0905534029006958,
+      "eval_mean_accuracy": 0.5463207146573476,
+      "eval_mean_iou": 0.41371023617827113,
+      "eval_overall_accuracy": 0.69981106540613,
+      "eval_per_category_accuracy": [
+        0.7044612714147034,
+        0.7849399792679025,
+        0.7447528691468164,
+        0.3383147930965087,
+        0.5745846391967699,
+        0.4949244934473886,
+        0.6616045998655433,
+        0.4726130946704796,
+        0.41939877376822265,
+        0.5984454674941234,
+        0.33202535558731977,
+        0.4297832389323924
+      ],
+      "eval_per_category_iou": [
+        0.5971861818357448,
+        0.581877446089541,
+        0.5903583035659352,
+        0.27266092092237654,
+        0.39661556877518883,
+        0.37978681185554447,
+        0.47048401182318955,
+        0.4214648530063751,
+        0.2554613355816417,
+        0.4336517402781951,
+        0.22355882183221085,
+        0.34141683857331007
+      ],
+      "eval_runtime": 121.413,
+      "eval_samples_per_second": 0.939,
+      "eval_steps_per_second": 0.082,
+      "step": 740
+    },
+    {
+      "epoch": 23.16,
+      "learning_rate": 0.000322125,
+      "loss": 0.2742,
+      "step": 741
+    },
+    {
+      "epoch": 23.19,
+      "learning_rate": 0.00032175,
+      "loss": 0.2639,
+      "step": 742
+    },
+    {
+      "epoch": 23.22,
+      "learning_rate": 0.000321375,
+      "loss": 0.3433,
+      "step": 743
+    },
+    {
+      "epoch": 23.25,
+      "learning_rate": 0.000321,
+      "loss": 0.2833,
+      "step": 744
+    },
+    {
+      "epoch": 23.28,
+      "learning_rate": 0.000320625,
+      "loss": 0.2534,
+      "step": 745
+    },
+    {
+      "epoch": 23.31,
+      "learning_rate": 0.00032024999999999995,
+      "loss": 0.2722,
+      "step": 746
+    },
+    {
+      "epoch": 23.34,
+      "learning_rate": 0.00031987499999999996,
+      "loss": 0.3452,
+      "step": 747
+    },
+    {
+      "epoch": 23.38,
+      "learning_rate": 0.00031949999999999996,
+      "loss": 0.2878,
+      "step": 748
+    },
+    {
+      "epoch": 23.41,
+      "learning_rate": 0.00031912499999999996,
+      "loss": 0.3004,
+      "step": 749
+    },
+    {
+      "epoch": 23.44,
+      "learning_rate": 0.00031874999999999997,
+      "loss": 0.2635,
+      "step": 750
+    },
+    {
+      "epoch": 23.47,
+      "learning_rate": 0.000318375,
+      "loss": 0.3005,
+      "step": 751
+    },
+    {
+      "epoch": 23.5,
+      "learning_rate": 0.000318,
+      "loss": 0.2855,
+      "step": 752
+    },
+    {
+      "epoch": 23.53,
+      "learning_rate": 0.000317625,
+      "loss": 0.3189,
+      "step": 753
+    },
+    {
+      "epoch": 23.56,
+      "learning_rate": 0.00031725,
+      "loss": 0.2992,
+      "step": 754
+    },
+    {
+      "epoch": 23.59,
+      "learning_rate": 0.00031687499999999994,
+      "loss": 0.2889,
+      "step": 755
+    },
+    {
+      "epoch": 23.62,
+      "learning_rate": 0.00031649999999999994,
+      "loss": 0.31,
+      "step": 756
+    },
+    {
+      "epoch": 23.66,
+      "learning_rate": 0.00031612499999999995,
+      "loss": 0.2969,
+      "step": 757
+    },
+    {
+      "epoch": 23.69,
+      "learning_rate": 0.00031574999999999995,
+      "loss": 0.286,
+      "step": 758
+    },
+    {
+      "epoch": 23.72,
+      "learning_rate": 0.00031537499999999995,
+      "loss": 0.3337,
+      "step": 759
+    },
+    {
+      "epoch": 23.75,
+      "learning_rate": 0.00031499999999999996,
+      "loss": 0.3142,
+      "step": 760
+    },
+    {
+      "epoch": 23.75,
+      "eval_loss": 1.0847059488296509,
+      "eval_mean_accuracy": 0.5396379106302426,
+      "eval_mean_iou": 0.412021539622709,
+      "eval_overall_accuracy": 0.701039197152121,
+      "eval_per_category_accuracy": [
+        0.7171844344700422,
+        0.7857090653853361,
+        0.7497516720935933,
+        0.3166286146522231,
+        0.5432236576871884,
+        0.555117424393643,
+        0.5960939873950837,
+        0.5433580017807569,
+        0.4478452215114061,
+        0.5734204461236043,
+        0.21545821562043752,
+        0.43186418644959607
+      ],
+      "eval_per_category_iou": [
+        0.5979924064798365,
+        0.5815836208902815,
+        0.5944937368457016,
+        0.2574408540266658,
+        0.3991078312589492,
+        0.3835625541497348,
+        0.4710279525051257,
+        0.4474271627943089,
+        0.26428891210267125,
+        0.42654588130316284,
+        0.1764425740117356,
+        0.3443449891043343
+      ],
+      "eval_runtime": 120.5433,
+      "eval_samples_per_second": 0.946,
+      "eval_steps_per_second": 0.083,
+      "step": 760
+    },
+    {
+      "epoch": 23.78,
+      "learning_rate": 0.000314625,
+      "loss": 0.3012,
+      "step": 761
+    },
+    {
+      "epoch": 23.81,
+      "learning_rate": 0.00031425,
+      "loss": 0.3316,
+      "step": 762
+    },
+    {
+      "epoch": 23.84,
+      "learning_rate": 0.0003138749999999999,
+      "loss": 0.4011,
+      "step": 763
+    },
+    {
+      "epoch": 23.88,
+      "learning_rate": 0.0003135,
+      "loss": 0.2954,
+      "step": 764
+    },
+    {
+      "epoch": 23.91,
+      "learning_rate": 0.000313125,
+      "loss": 0.2984,
+      "step": 765
+    },
+    {
+      "epoch": 23.94,
+      "learning_rate": 0.00031275,
+      "loss": 0.2974,
+      "step": 766
+    },
+    {
+      "epoch": 23.97,
+      "learning_rate": 0.000312375,
+      "loss": 0.3273,
+      "step": 767
+    },
+    {
+      "epoch": 24.0,
+      "learning_rate": 0.000312,
+      "loss": 0.2832,
+      "step": 768
+    },
+    {
+      "epoch": 24.03,
+      "learning_rate": 0.000311625,
+      "loss": 0.3242,
+      "step": 769
+    },
+    {
+      "epoch": 24.06,
+      "learning_rate": 0.00031125,
+      "loss": 0.3003,
+      "step": 770
+    },
+    {
+      "epoch": 24.09,
+      "learning_rate": 0.00031087499999999995,
+      "loss": 0.2776,
+      "step": 771
+    },
+    {
+      "epoch": 24.12,
+      "learning_rate": 0.00031049999999999996,
+      "loss": 0.2718,
+      "step": 772
+    },
+    {
+      "epoch": 24.16,
+      "learning_rate": 0.00031012499999999996,
+      "loss": 0.3566,
+      "step": 773
+    },
+    {
+      "epoch": 24.19,
+      "learning_rate": 0.00030974999999999997,
+      "loss": 0.2627,
+      "step": 774
+    },
+    {
+      "epoch": 24.22,
+      "learning_rate": 0.00030937499999999997,
+      "loss": 0.2948,
+      "step": 775
+    },
+    {
+      "epoch": 24.25,
+      "learning_rate": 0.000309,
+      "loss": 0.2548,
+      "step": 776
+    },
+    {
+      "epoch": 24.28,
+      "learning_rate": 0.000308625,
+      "loss": 0.2471,
+      "step": 777
+    },
+    {
+      "epoch": 24.31,
+      "learning_rate": 0.00030825,
+      "loss": 0.2693,
+      "step": 778
+    },
+    {
+      "epoch": 24.34,
+      "learning_rate": 0.000307875,
+      "loss": 0.3797,
+      "step": 779
+    },
+    {
+      "epoch": 24.38,
+      "learning_rate": 0.00030749999999999994,
+      "loss": 0.2776,
+      "step": 780
+    },
+    {
+      "epoch": 24.38,
+      "eval_loss": 1.1065510511398315,
+      "eval_mean_accuracy": 0.5409000564293074,
+      "eval_mean_iou": 0.4130489167566249,
+      "eval_overall_accuracy": 0.6989034016927084,
+      "eval_per_category_accuracy": [
+        0.6585943904780239,
+        0.8114654061620347,
+        0.7382970569611855,
+        0.36781146556928823,
+        0.5205598786078803,
+        0.5318158228054831,
+        0.654185868197856,
+        0.45673092891865624,
+        0.4357226515709431,
+        0.5726850603511806,
+        0.23527071160315016,
+        0.5076614359260062
+      ],
+      "eval_per_category_iou": [
+        0.5767084069741052,
+        0.5852845346599131,
+        0.5944243532033295,
+        0.2747477470238944,
+        0.3942487238224414,
+        0.38964205611328484,
+        0.47090833672390403,
+        0.4136149683534171,
+        0.26310932158892425,
+        0.43047280328494825,
+        0.18921467025406352,
+        0.37421107907727336
+      ],
+      "eval_runtime": 120.6331,
+      "eval_samples_per_second": 0.945,
+      "eval_steps_per_second": 0.083,
+      "step": 780
+    },
+    {
+      "epoch": 24.41,
+      "learning_rate": 0.00030712499999999994,
+      "loss": 0.2481,
+      "step": 781
+    },
+    {
+      "epoch": 24.44,
+      "learning_rate": 0.00030674999999999995,
+      "loss": 0.2944,
+      "step": 782
+    },
+    {
+      "epoch": 24.47,
+      "learning_rate": 0.00030637499999999995,
+      "loss": 0.2631,
+      "step": 783
+    },
+    {
+      "epoch": 24.5,
+      "learning_rate": 0.00030599999999999996,
+      "loss": 0.276,
+      "step": 784
+    },
+    {
+      "epoch": 24.53,
+      "learning_rate": 0.00030562499999999996,
+      "loss": 0.2383,
+      "step": 785
+    },
+    {
+      "epoch": 24.56,
+      "learning_rate": 0.00030525,
+      "loss": 0.2755,
+      "step": 786
+    },
+    {
+      "epoch": 24.59,
+      "learning_rate": 0.000304875,
+      "loss": 0.3042,
+      "step": 787
+    },
+    {
+      "epoch": 24.62,
+      "learning_rate": 0.0003044999999999999,
+      "loss": 0.2983,
+      "step": 788
+    },
+    {
+      "epoch": 24.66,
+      "learning_rate": 0.000304125,
+      "loss": 0.2719,
+      "step": 789
+    },
+    {
+      "epoch": 24.69,
+      "learning_rate": 0.00030375,
+      "loss": 0.2943,
+      "step": 790
+    },
+    {
+      "epoch": 24.72,
+      "learning_rate": 0.000303375,
+      "loss": 0.32,
+      "step": 791
+    },
+    {
+      "epoch": 24.75,
+      "learning_rate": 0.000303,
+      "loss": 0.3395,
+      "step": 792
+    },
+    {
+      "epoch": 24.78,
+      "learning_rate": 0.000302625,
+      "loss": 0.3138,
+      "step": 793
+    },
+    {
+      "epoch": 24.81,
+      "learning_rate": 0.00030225,
+      "loss": 0.3289,
+      "step": 794
+    },
+    {
+      "epoch": 24.84,
+      "learning_rate": 0.000301875,
+      "loss": 0.3017,
+      "step": 795
+    },
+    {
+      "epoch": 24.88,
+      "learning_rate": 0.00030149999999999996,
+      "loss": 0.2854,
+      "step": 796
+    },
+    {
+      "epoch": 24.91,
+      "learning_rate": 0.00030112499999999996,
+      "loss": 0.358,
+      "step": 797
+    },
+    {
+      "epoch": 24.94,
+      "learning_rate": 0.00030074999999999996,
+      "loss": 0.2555,
+      "step": 798
+    },
+    {
+      "epoch": 24.97,
+      "learning_rate": 0.00030037499999999997,
+      "loss": 0.3597,
+      "step": 799
+    },
+    {
+      "epoch": 25.0,
+      "learning_rate": 0.0003,
+      "loss": 0.2831,
+      "step": 800
+    },
+    {
+      "epoch": 25.0,
+      "eval_loss": 1.0793243646621704,
+      "eval_mean_accuracy": 0.5555561395131962,
+      "eval_mean_iou": 0.4132143115540001,
+      "eval_overall_accuracy": 0.6999500341582716,
+      "eval_per_category_accuracy": [
+        0.7249887263019231,
+        0.7742597637071228,
+        0.7231768051890759,
+        0.4704408713333009,
+        0.5966588804453745,
+        0.6046893543200781,
+        0.6366584848594582,
+        0.5387593203123074,
+        0.36509675873756686,
+        0.6060694882184419,
+        0.21657234331048963,
+        0.4093028774232154
+      ],
+      "eval_per_category_iou": [
+        0.6036445508408057,
+        0.5797123082800245,
+        0.5924730622228696,
+        0.2904498714652956,
+        0.39978416826730556,
+        0.39527915476016784,
+        0.4738098306872107,
+        0.4425860694047625,
+        0.24609748803827752,
+        0.4288039545547971,
+        0.17714094016858645,
+        0.3287903399578989
+      ],
+      "eval_runtime": 120.153,
+      "eval_samples_per_second": 0.949,
+      "eval_steps_per_second": 0.083,
+      "step": 800
+    },
+    {
+      "epoch": 25.03,
+      "learning_rate": 0.000299625,
+      "loss": 0.308,
+      "step": 801
+    },
+    {
+      "epoch": 25.06,
+      "learning_rate": 0.00029925,
+      "loss": 0.2962,
+      "step": 802
+    },
+    {
+      "epoch": 25.09,
+      "learning_rate": 0.000298875,
+      "loss": 0.2647,
+      "step": 803
+    },
+    {
+      "epoch": 25.12,
+      "learning_rate": 0.0002985,
+      "loss": 0.302,
+      "step": 804
+    },
+    {
+      "epoch": 25.16,
+      "learning_rate": 0.000298125,
+      "loss": 0.2516,
+      "step": 805
+    },
+    {
+      "epoch": 25.19,
+      "learning_rate": 0.00029775,
+      "loss": 0.3046,
+      "step": 806
+    },
+    {
+      "epoch": 25.22,
+      "learning_rate": 0.00029737499999999995,
+      "loss": 0.2898,
+      "step": 807
+    },
+    {
+      "epoch": 25.25,
+      "learning_rate": 0.00029699999999999996,
+      "loss": 0.2943,
+      "step": 808
+    },
+    {
+      "epoch": 25.28,
+      "learning_rate": 0.00029662499999999996,
+      "loss": 0.2618,
+      "step": 809
+    },
+    {
+      "epoch": 25.31,
+      "learning_rate": 0.00029624999999999996,
+      "loss": 0.324,
+      "step": 810
+    },
+    {
+      "epoch": 25.34,
+      "learning_rate": 0.00029587499999999997,
+      "loss": 0.2513,
+      "step": 811
+    },
+    {
+      "epoch": 25.38,
+      "learning_rate": 0.00029549999999999997,
+      "loss": 0.2644,
+      "step": 812
+    },
+    {
+      "epoch": 25.41,
+      "learning_rate": 0.000295125,
+      "loss": 0.2757,
+      "step": 813
+    },
+    {
+      "epoch": 25.44,
+      "learning_rate": 0.00029475,
+      "loss": 0.2637,
+      "step": 814
+    },
+    {
+      "epoch": 25.47,
+      "learning_rate": 0.000294375,
+      "loss": 0.3461,
+      "step": 815
+    },
+    {
+      "epoch": 25.5,
+      "learning_rate": 0.000294,
+      "loss": 0.2798,
+      "step": 816
+    },
+    {
+      "epoch": 25.53,
+      "learning_rate": 0.000293625,
+      "loss": 0.2927,
+      "step": 817
+    },
+    {
+      "epoch": 25.56,
+      "learning_rate": 0.00029325,
+      "loss": 0.2956,
+      "step": 818
+    },
+    {
+      "epoch": 25.59,
+      "learning_rate": 0.00029287499999999995,
+      "loss": 0.286,
+      "step": 819
+    },
+    {
+      "epoch": 25.62,
+      "learning_rate": 0.00029249999999999995,
+      "loss": 0.2783,
+      "step": 820
+    },
+    {
+      "epoch": 25.62,
+      "eval_loss": 1.1129748821258545,
+      "eval_mean_accuracy": 0.5242069480648149,
+      "eval_mean_iou": 0.40439976233238045,
+      "eval_overall_accuracy": 0.6998102288497122,
+      "eval_per_category_accuracy": [
+        0.6928495459463953,
+        0.8052259696766798,
+        0.7618540371226322,
+        0.3347547900430948,
+        0.5257239058603301,
+        0.5060808088873081,
+        0.584639982114433,
+        0.4304599431486396,
+        0.44648101647693494,
+        0.5916913924920764,
+        0.22076562609997152,
+        0.38995635890928226
+      ],
+      "eval_per_category_iou": [
+        0.5929048312384559,
+        0.5857610162672414,
+        0.5929578522073312,
+        0.26841292584194637,
+        0.39631058412442427,
+        0.37693746587302873,
+        0.4692408348018164,
+        0.3940319440562184,
+        0.24970009291847117,
+        0.4318095918863708,
+        0.17983464398091126,
+        0.31489536479235
+      ],
+      "eval_runtime": 120.1309,
+      "eval_samples_per_second": 0.949,
+      "eval_steps_per_second": 0.083,
+      "step": 820
+    },
+    {
+      "epoch": 25.66,
+      "learning_rate": 0.00029212499999999996,
+      "loss": 0.2812,
+      "step": 821
+    },
+    {
+      "epoch": 25.69,
+      "learning_rate": 0.00029174999999999996,
+      "loss": 0.2671,
+      "step": 822
+    },
+    {
+      "epoch": 25.72,
+      "learning_rate": 0.00029137499999999997,
+      "loss": 0.2981,
+      "step": 823
+    },
+    {
+      "epoch": 25.75,
+      "learning_rate": 0.00029099999999999997,
+      "loss": 0.2718,
+      "step": 824
+    },
+    {
+      "epoch": 25.78,
+      "learning_rate": 0.000290625,
+      "loss": 0.2617,
+      "step": 825
+    },
+    {
+      "epoch": 25.81,
+      "learning_rate": 0.00029025,
+      "loss": 0.3141,
+      "step": 826
+    },
+    {
+      "epoch": 25.84,
+      "learning_rate": 0.000289875,
+      "loss": 0.2634,
+      "step": 827
+    },
+    {
+      "epoch": 25.88,
+      "learning_rate": 0.0002895,
+      "loss": 0.2708,
+      "step": 828
+    },
+    {
+      "epoch": 25.91,
+      "learning_rate": 0.000289125,
+      "loss": 0.325,
+      "step": 829
+    },
+    {
+      "epoch": 25.94,
+      "learning_rate": 0.00028875,
+      "loss": 0.3074,
+      "step": 830
+    },
+    {
+      "epoch": 25.97,
+      "learning_rate": 0.000288375,
+      "loss": 0.338,
+      "step": 831
+    },
+    {
+      "epoch": 26.0,
+      "learning_rate": 0.00028799999999999995,
+      "loss": 0.3272,
+      "step": 832
+    },
+    {
+      "epoch": 26.03,
+      "learning_rate": 0.00028762499999999996,
+      "loss": 0.2875,
+      "step": 833
+    },
+    {
+      "epoch": 26.06,
+      "learning_rate": 0.00028724999999999996,
+      "loss": 0.2664,
+      "step": 834
+    },
+    {
+      "epoch": 26.09,
+      "learning_rate": 0.00028687499999999997,
+      "loss": 0.2876,
+      "step": 835
+    },
+    {
+      "epoch": 26.12,
+      "learning_rate": 0.00028649999999999997,
+      "loss": 0.264,
+      "step": 836
+    },
+    {
+      "epoch": 26.16,
+      "learning_rate": 0.000286125,
+      "loss": 0.2825,
+      "step": 837
+    },
+    {
+      "epoch": 26.19,
+      "learning_rate": 0.00028575,
+      "loss": 0.2494,
+      "step": 838
+    },
+    {
+      "epoch": 26.22,
+      "learning_rate": 0.000285375,
+      "loss": 0.2277,
+      "step": 839
+    },
+    {
+      "epoch": 26.25,
+      "learning_rate": 0.000285,
+      "loss": 0.2966,
+      "step": 840
+    },
+    {
+      "epoch": 26.25,
+      "eval_loss": 1.1355547904968262,
+      "eval_mean_accuracy": 0.5296086582013562,
+      "eval_mean_iou": 0.40787758711413447,
+      "eval_overall_accuracy": 0.7007122709040057,
+      "eval_per_category_accuracy": [
+        0.6845148653492409,
+        0.8153546551740586,
+        0.7468978629675106,
+        0.39289109721653565,
+        0.5783123010816297,
+        0.48420005879385497,
+        0.5620169440556468,
+        0.4510479357898921,
+        0.44546063547554193,
+        0.5731023005057472,
+        0.19685473182885285,
+        0.4246505101777626
+      ],
+      "eval_per_category_iou": [
+        0.5913891081514681,
+        0.5869851217027395,
+        0.5930779050634145,
+        0.2991084754269412,
+        0.39366922071259536,
+        0.37145072534178974,
+        0.4626901171487159,
+        0.40810573165541625,
+        0.24822255842813845,
+        0.42864437477496875,
+        0.16783751480957626,
+        0.3433501921538499
+      ],
+      "eval_runtime": 120.5826,
+      "eval_samples_per_second": 0.945,
+      "eval_steps_per_second": 0.083,
+      "step": 840
+    },
+    {
+      "epoch": 26.28,
+      "learning_rate": 0.000284625,
+      "loss": 0.2934,
+      "step": 841
+    },
+    {
+      "epoch": 26.31,
+      "learning_rate": 0.00028425,
+      "loss": 0.2477,
+      "step": 842
+    },
+    {
+      "epoch": 26.34,
+      "learning_rate": 0.000283875,
+      "loss": 0.2836,
+      "step": 843
+    },
+    {
+      "epoch": 26.38,
+      "learning_rate": 0.00028349999999999995,
+      "loss": 0.2698,
+      "step": 844
+    },
+    {
+      "epoch": 26.41,
+      "learning_rate": 0.00028312499999999996,
+      "loss": 0.2764,
+      "step": 845
+    },
+    {
+      "epoch": 26.44,
+      "learning_rate": 0.00028274999999999996,
+      "loss": 0.2479,
+      "step": 846
+    },
+    {
+      "epoch": 26.47,
+      "learning_rate": 0.00028237499999999997,
+      "loss": 0.2842,
+      "step": 847
+    },
+    {
+      "epoch": 26.5,
+      "learning_rate": 0.00028199999999999997,
+      "loss": 0.3196,
+      "step": 848
+    },
+    {
+      "epoch": 26.53,
+      "learning_rate": 0.000281625,
+      "loss": 0.2755,
+      "step": 849
+    },
+    {
+      "epoch": 26.56,
+      "learning_rate": 0.00028125,
+      "loss": 0.2572,
+      "step": 850
+    },
+    {
+      "epoch": 26.59,
+      "learning_rate": 0.000280875,
+      "loss": 0.2665,
+      "step": 851
+    },
+    {
+      "epoch": 26.62,
+      "learning_rate": 0.0002805,
+      "loss": 0.2254,
+      "step": 852
+    },
+    {
+      "epoch": 26.66,
+      "learning_rate": 0.000280125,
+      "loss": 0.2048,
+      "step": 853
+    },
+    {
+      "epoch": 26.69,
+      "learning_rate": 0.00027975,
+      "loss": 0.3095,
+      "step": 854
+    },
+    {
+      "epoch": 26.72,
+      "learning_rate": 0.000279375,
+      "loss": 0.3002,
+      "step": 855
+    },
+    {
+      "epoch": 26.75,
+      "learning_rate": 0.000279,
+      "loss": 0.2936,
+      "step": 856
+    },
+    {
+      "epoch": 26.78,
+      "learning_rate": 0.00027862499999999996,
+      "loss": 0.2494,
+      "step": 857
+    },
+    {
+      "epoch": 26.81,
+      "learning_rate": 0.00027824999999999996,
+      "loss": 0.3019,
+      "step": 858
+    },
+    {
+      "epoch": 26.84,
+      "learning_rate": 0.00027787499999999996,
+      "loss": 0.3723,
+      "step": 859
+    },
+    {
+      "epoch": 26.88,
+      "learning_rate": 0.00027749999999999997,
+      "loss": 0.3107,
+      "step": 860
+    },
+    {
+      "epoch": 26.88,
+      "eval_loss": 1.1252777576446533,
+      "eval_mean_accuracy": 0.5371894166488177,
+      "eval_mean_iou": 0.40644828244848913,
+      "eval_overall_accuracy": 0.6969429484584875,
+      "eval_per_category_accuracy": [
+        0.6468952947673742,
+        0.8087662073601048,
+        0.7475841354350867,
+        0.4212253905247014,
+        0.5811931354296029,
+        0.5569574542986655,
+        0.6194302439270155,
+        0.5202328055707047,
+        0.34214705908449644,
+        0.6473480998101035,
+        0.18965881369887333,
+        0.36483435987908425
+      ],
+      "eval_per_category_iou": [
+        0.5736085592707574,
+        0.5845864037747086,
+        0.5940903513547896,
+        0.28328409615950045,
+        0.39303828164174565,
+        0.39455284730515444,
+        0.47445652468845856,
+        0.44227721221042504,
+        0.23241523859463548,
+        0.43479646852227,
+        0.16037518537977943,
+        0.309898220479645
+      ],
+      "eval_runtime": 121.618,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 860
+    },
+    {
+      "epoch": 26.91,
+      "learning_rate": 0.000277125,
+      "loss": 0.3372,
+      "step": 861
+    },
+    {
+      "epoch": 26.94,
+      "learning_rate": 0.00027675,
+      "loss": 0.2567,
+      "step": 862
+    },
+    {
+      "epoch": 26.97,
+      "learning_rate": 0.000276375,
+      "loss": 0.3728,
+      "step": 863
+    },
+    {
+      "epoch": 27.0,
+      "learning_rate": 0.000276,
+      "loss": 0.2861,
+      "step": 864
+    },
+    {
+      "epoch": 27.03,
+      "learning_rate": 0.00027562499999999994,
+      "loss": 0.2992,
+      "step": 865
+    },
+    {
+      "epoch": 27.06,
+      "learning_rate": 0.00027525,
+      "loss": 0.292,
+      "step": 866
+    },
+    {
+      "epoch": 27.09,
+      "learning_rate": 0.000274875,
+      "loss": 0.2634,
+      "step": 867
+    },
+    {
+      "epoch": 27.12,
+      "learning_rate": 0.0002745,
+      "loss": 0.264,
+      "step": 868
+    },
+    {
+      "epoch": 27.16,
+      "learning_rate": 0.00027412499999999995,
+      "loss": 0.2604,
+      "step": 869
+    },
+    {
+      "epoch": 27.19,
+      "learning_rate": 0.00027374999999999996,
+      "loss": 0.2821,
+      "step": 870
+    },
+    {
+      "epoch": 27.22,
+      "learning_rate": 0.00027337499999999996,
+      "loss": 0.2545,
+      "step": 871
+    },
+    {
+      "epoch": 27.25,
+      "learning_rate": 0.00027299999999999997,
+      "loss": 0.2885,
+      "step": 872
+    },
+    {
+      "epoch": 27.28,
+      "learning_rate": 0.00027262499999999997,
+      "loss": 0.2872,
+      "step": 873
+    },
+    {
+      "epoch": 27.31,
+      "learning_rate": 0.00027225,
+      "loss": 0.2851,
+      "step": 874
+    },
+    {
+      "epoch": 27.34,
+      "learning_rate": 0.000271875,
+      "loss": 0.2739,
+      "step": 875
+    },
+    {
+      "epoch": 27.38,
+      "learning_rate": 0.0002715,
+      "loss": 0.249,
+      "step": 876
+    },
+    {
+      "epoch": 27.41,
+      "learning_rate": 0.000271125,
+      "loss": 0.2513,
+      "step": 877
+    },
+    {
+      "epoch": 27.44,
+      "learning_rate": 0.00027074999999999994,
+      "loss": 0.3014,
+      "step": 878
+    },
+    {
+      "epoch": 27.47,
+      "learning_rate": 0.000270375,
+      "loss": 0.2851,
+      "step": 879
+    },
+    {
+      "epoch": 27.5,
+      "learning_rate": 0.00027,
+      "loss": 0.2487,
+      "step": 880
+    },
+    {
+      "epoch": 27.5,
+      "eval_loss": 1.104833960533142,
+      "eval_mean_accuracy": 0.5383438131308615,
+      "eval_mean_iou": 0.4136548750583668,
+      "eval_overall_accuracy": 0.7020032113058525,
+      "eval_per_category_accuracy": [
+        0.7062988364780013,
+        0.7964391628061449,
+        0.7500572844897572,
+        0.37996266507519033,
+        0.48858099662281934,
+        0.5463781896573625,
+        0.6165293498061085,
+        0.4301974368672393,
+        0.3860922069510128,
+        0.5806486102512777,
+        0.29522853391202675,
+        0.48371248465339894
+      ],
+      "eval_per_category_iou": [
+        0.5994901679158006,
+        0.5845418952301205,
+        0.5961975347621044,
+        0.28874737966222364,
+        0.3840471238705527,
+        0.3898276118774963,
+        0.47020173305147583,
+        0.3943195545210807,
+        0.24850338300402763,
+        0.42890566188928947,
+        0.22050990128435502,
+        0.3585665536318737
+      ],
+      "eval_runtime": 121.635,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 880
+    },
+    {
+      "epoch": 27.53,
+      "learning_rate": 0.000269625,
+      "loss": 0.2821,
+      "step": 881
+    },
+    {
+      "epoch": 27.56,
+      "learning_rate": 0.00026924999999999996,
+      "loss": 0.3248,
+      "step": 882
+    },
+    {
+      "epoch": 27.59,
+      "learning_rate": 0.00026887499999999996,
+      "loss": 0.2404,
+      "step": 883
+    },
+    {
+      "epoch": 27.62,
+      "learning_rate": 0.00026849999999999997,
+      "loss": 0.2778,
+      "step": 884
+    },
+    {
+      "epoch": 27.66,
+      "learning_rate": 0.00026812499999999997,
+      "loss": 0.2501,
+      "step": 885
+    },
+    {
+      "epoch": 27.69,
+      "learning_rate": 0.00026775,
+      "loss": 0.2917,
+      "step": 886
+    },
+    {
+      "epoch": 27.72,
+      "learning_rate": 0.000267375,
+      "loss": 0.2685,
+      "step": 887
+    },
+    {
+      "epoch": 27.75,
+      "learning_rate": 0.000267,
+      "loss": 0.2785,
+      "step": 888
+    },
+    {
+      "epoch": 27.78,
+      "learning_rate": 0.000266625,
+      "loss": 0.251,
+      "step": 889
+    },
+    {
+      "epoch": 27.81,
+      "learning_rate": 0.00026624999999999994,
+      "loss": 0.2734,
+      "step": 890
+    },
+    {
+      "epoch": 27.84,
+      "learning_rate": 0.000265875,
+      "loss": 0.2561,
+      "step": 891
+    },
+    {
+      "epoch": 27.88,
+      "learning_rate": 0.0002655,
+      "loss": 0.2527,
+      "step": 892
+    },
+    {
+      "epoch": 27.91,
+      "learning_rate": 0.000265125,
+      "loss": 0.3193,
+      "step": 893
+    },
+    {
+      "epoch": 27.94,
+      "learning_rate": 0.00026474999999999996,
+      "loss": 0.3094,
+      "step": 894
+    },
+    {
+      "epoch": 27.97,
+      "learning_rate": 0.00026437499999999996,
+      "loss": 0.2827,
+      "step": 895
+    },
+    {
+      "epoch": 28.0,
+      "learning_rate": 0.00026399999999999997,
+      "loss": 0.2405,
+      "step": 896
+    },
+    {
+      "epoch": 28.03,
+      "learning_rate": 0.00026362499999999997,
+      "loss": 0.2495,
+      "step": 897
+    },
+    {
+      "epoch": 28.06,
+      "learning_rate": 0.00026325,
+      "loss": 0.2883,
+      "step": 898
+    },
+    {
+      "epoch": 28.09,
+      "learning_rate": 0.000262875,
+      "loss": 0.2845,
+      "step": 899
+    },
+    {
+      "epoch": 28.12,
+      "learning_rate": 0.0002625,
+      "loss": 0.2757,
+      "step": 900
+    },
+    {
+      "epoch": 28.12,
+      "eval_loss": 1.1434400081634521,
+      "eval_mean_accuracy": 0.541031331711776,
+      "eval_mean_iou": 0.4109253831772188,
+      "eval_overall_accuracy": 0.6963558531644052,
+      "eval_per_category_accuracy": [
+        0.6428283935373039,
+        0.8076601559386173,
+        0.7724049412759075,
+        0.36275251386180524,
+        0.5600710686841961,
+        0.5238405452585659,
+        0.5975338070279572,
+        0.46645116150993615,
+        0.42154157387114805,
+        0.5871591213130964,
+        0.2668947975746297,
+        0.48323790068814676
+      ],
+      "eval_per_category_iou": [
+        0.5694416441741732,
+        0.5846775522171952,
+        0.5907702340495511,
+        0.2776358489263274,
+        0.4037566433525581,
+        0.3798271628979161,
+        0.4687103835596983,
+        0.4146082000889513,
+        0.250402874607502,
+        0.43084935182927925,
+        0.2028785213725326,
+        0.35754618105094094
+      ],
+      "eval_runtime": 120.892,
+      "eval_samples_per_second": 0.943,
+      "eval_steps_per_second": 0.083,
+      "step": 900
+    },
+    {
+      "epoch": 28.16,
+      "learning_rate": 0.000262125,
+      "loss": 0.2598,
+      "step": 901
+    },
+    {
+      "epoch": 28.19,
+      "learning_rate": 0.00026175,
+      "loss": 0.2775,
+      "step": 902
+    },
+    {
+      "epoch": 28.22,
+      "learning_rate": 0.00026137499999999994,
+      "loss": 0.2486,
+      "step": 903
+    },
+    {
+      "epoch": 28.25,
+      "learning_rate": 0.000261,
+      "loss": 0.2899,
+      "step": 904
+    },
+    {
+      "epoch": 28.28,
+      "learning_rate": 0.000260625,
+      "loss": 0.2669,
+      "step": 905
+    },
+    {
+      "epoch": 28.31,
+      "learning_rate": 0.00026025,
+      "loss": 0.2546,
+      "step": 906
+    },
+    {
+      "epoch": 28.34,
+      "learning_rate": 0.00025987499999999996,
+      "loss": 0.263,
+      "step": 907
+    },
+    {
+      "epoch": 28.38,
+      "learning_rate": 0.00025949999999999997,
+      "loss": 0.2662,
+      "step": 908
+    },
+    {
+      "epoch": 28.41,
+      "learning_rate": 0.00025912499999999997,
+      "loss": 0.2905,
+      "step": 909
+    },
+    {
+      "epoch": 28.44,
+      "learning_rate": 0.00025875,
+      "loss": 0.311,
+      "step": 910
+    },
+    {
+      "epoch": 28.47,
+      "learning_rate": 0.000258375,
+      "loss": 0.3024,
+      "step": 911
+    },
+    {
+      "epoch": 28.5,
+      "learning_rate": 0.000258,
+      "loss": 0.3055,
+      "step": 912
+    },
+    {
+      "epoch": 28.53,
+      "learning_rate": 0.000257625,
+      "loss": 0.2506,
+      "step": 913
+    },
+    {
+      "epoch": 28.56,
+      "learning_rate": 0.00025725,
+      "loss": 0.3058,
+      "step": 914
+    },
+    {
+      "epoch": 28.59,
+      "learning_rate": 0.00025687499999999994,
+      "loss": 0.279,
+      "step": 915
+    },
+    {
+      "epoch": 28.62,
+      "learning_rate": 0.00025649999999999995,
+      "loss": 0.2562,
+      "step": 916
+    },
+    {
+      "epoch": 28.66,
+      "learning_rate": 0.000256125,
+      "loss": 0.3,
+      "step": 917
+    },
+    {
+      "epoch": 28.69,
+      "learning_rate": 0.00025575,
+      "loss": 0.2449,
+      "step": 918
+    },
+    {
+      "epoch": 28.72,
+      "learning_rate": 0.00025537499999999996,
+      "loss": 0.2441,
+      "step": 919
+    },
+    {
+      "epoch": 28.75,
+      "learning_rate": 0.00025499999999999996,
+      "loss": 0.3006,
+      "step": 920
+    },
+    {
+      "epoch": 28.75,
+      "eval_loss": 1.1551755666732788,
+      "eval_mean_accuracy": 0.5317834096584441,
+      "eval_mean_iou": 0.4088655326412011,
+      "eval_overall_accuracy": 0.7012928745202851,
+      "eval_per_category_accuracy": [
+        0.6847293996753938,
+        0.8151594934072673,
+        0.7326114108420777,
+        0.44325507803554454,
+        0.553924478380574,
+        0.47528299079259195,
+        0.5780391687619756,
+        0.49798084454163727,
+        0.3804002555388943,
+        0.5856064663879155,
+        0.19763829416031808,
+        0.4367730353771395
+      ],
+      "eval_per_category_iou": [
+        0.5946446624900661,
+        0.5864312556841489,
+        0.5921941491294092,
+        0.2834809392038381,
+        0.40367233890517823,
+        0.3685974837456725,
+        0.4676389481527713,
+        0.4282021114776935,
+        0.24396694355964202,
+        0.428451065710109,
+        0.16577367122535064,
+        0.34333282241053376
+      ],
+      "eval_runtime": 121.5289,
+      "eval_samples_per_second": 0.938,
+      "eval_steps_per_second": 0.082,
+      "step": 920
+    },
+    {
+      "epoch": 28.78,
+      "learning_rate": 0.00025462499999999997,
+      "loss": 0.2727,
+      "step": 921
+    },
+    {
+      "epoch": 28.81,
+      "learning_rate": 0.00025425,
+      "loss": 0.2686,
+      "step": 922
+    },
+    {
+      "epoch": 28.84,
+      "learning_rate": 0.000253875,
+      "loss": 0.2407,
+      "step": 923
+    },
+    {
+      "epoch": 28.88,
+      "learning_rate": 0.0002535,
+      "loss": 0.276,
+      "step": 924
+    },
+    {
+      "epoch": 28.91,
+      "learning_rate": 0.000253125,
+      "loss": 0.2851,
+      "step": 925
+    },
+    {
+      "epoch": 28.94,
+      "learning_rate": 0.00025275,
+      "loss": 0.2783,
+      "step": 926
+    },
+    {
+      "epoch": 28.97,
+      "learning_rate": 0.000252375,
+      "loss": 0.2292,
+      "step": 927
+    },
+    {
+      "epoch": 29.0,
+      "learning_rate": 0.00025199999999999995,
+      "loss": 0.2581,
+      "step": 928
+    },
+    {
+      "epoch": 29.03,
+      "learning_rate": 0.00025162499999999995,
+      "loss": 0.249,
+      "step": 929
+    },
+    {
+      "epoch": 29.06,
+      "learning_rate": 0.00025125,
+      "loss": 0.3056,
+      "step": 930
+    },
+    {
+      "epoch": 29.09,
+      "learning_rate": 0.000250875,
+      "loss": 0.2546,
+      "step": 931
+    },
+    {
+      "epoch": 29.12,
+      "learning_rate": 0.00025049999999999996,
+      "loss": 0.2623,
+      "step": 932
+    },
+    {
+      "epoch": 29.16,
+      "learning_rate": 0.00025012499999999997,
+      "loss": 0.2781,
+      "step": 933
+    },
+    {
+      "epoch": 29.19,
+      "learning_rate": 0.00024974999999999997,
+      "loss": 0.2941,
+      "step": 934
+    },
+    {
+      "epoch": 29.22,
+      "learning_rate": 0.000249375,
+      "loss": 0.2379,
+      "step": 935
+    },
+    {
+      "epoch": 29.25,
+      "learning_rate": 0.000249,
+      "loss": 0.244,
+      "step": 936
+    },
+    {
+      "epoch": 29.28,
+      "learning_rate": 0.000248625,
+      "loss": 0.2522,
+      "step": 937
+    },
+    {
+      "epoch": 29.31,
+      "learning_rate": 0.00024825,
+      "loss": 0.2551,
+      "step": 938
+    },
+    {
+      "epoch": 29.34,
+      "learning_rate": 0.000247875,
+      "loss": 0.2351,
+      "step": 939
+    },
+    {
+      "epoch": 29.38,
+      "learning_rate": 0.00024749999999999994,
+      "loss": 0.2448,
+      "step": 940
+    },
+    {
+      "epoch": 29.38,
+      "eval_loss": 1.145183801651001,
+      "eval_mean_accuracy": 0.5433741077070523,
+      "eval_mean_iou": 0.40868834001996196,
+      "eval_overall_accuracy": 0.696986382467705,
+      "eval_per_category_accuracy": [
+        0.6598557524640978,
+        0.789122698542334,
+        0.7906669227228666,
+        0.4405902804283107,
+        0.6099184168782047,
+        0.5082692468216841,
+        0.6221579503392117,
+        0.44603460154225116,
+        0.3529897163340816,
+        0.6328604785431643,
+        0.20407819707204794,
+        0.4639450307963725
+      ],
+      "eval_per_category_iou": [
+        0.5793061475112379,
+        0.5826126060468451,
+        0.5842890156126345,
+        0.29182189026141914,
+        0.4037886875861736,
+        0.3810990667633601,
+        0.4806175151134428,
+        0.39815502919530005,
+        0.23608968120207943,
+        0.43974682580956126,
+        0.1657176375263769,
+        0.36101597761111354
+      ],
+      "eval_runtime": 121.7058,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 940
+    },
+    {
+      "epoch": 29.41,
+      "learning_rate": 0.00024712499999999995,
+      "loss": 0.2664,
+      "step": 941
+    },
+    {
+      "epoch": 29.44,
+      "learning_rate": 0.00024675,
+      "loss": 0.3154,
+      "step": 942
+    },
+    {
+      "epoch": 29.47,
+      "learning_rate": 0.000246375,
+      "loss": 0.2909,
+      "step": 943
+    },
+    {
+      "epoch": 29.5,
+      "learning_rate": 0.00024599999999999996,
+      "loss": 0.278,
+      "step": 944
+    },
+    {
+      "epoch": 29.53,
+      "learning_rate": 0.00024562499999999997,
+      "loss": 0.2459,
+      "step": 945
+    },
+    {
+      "epoch": 29.56,
+      "learning_rate": 0.00024524999999999997,
+      "loss": 0.2831,
+      "step": 946
+    },
+    {
+      "epoch": 29.59,
+      "learning_rate": 0.000244875,
+      "loss": 0.2497,
+      "step": 947
+    },
+    {
+      "epoch": 29.62,
+      "learning_rate": 0.0002445,
+      "loss": 0.2958,
+      "step": 948
+    },
+    {
+      "epoch": 29.66,
+      "learning_rate": 0.00024412499999999996,
+      "loss": 0.2965,
+      "step": 949
+    },
+    {
+      "epoch": 29.69,
+      "learning_rate": 0.00024375,
+      "loss": 0.2368,
+      "step": 950
+    },
+    {
+      "epoch": 29.72,
+      "learning_rate": 0.000243375,
+      "loss": 0.2432,
+      "step": 951
+    },
+    {
+      "epoch": 29.75,
+      "learning_rate": 0.000243,
+      "loss": 0.2763,
+      "step": 952
+    },
+    {
+      "epoch": 29.78,
+      "learning_rate": 0.00024262499999999998,
+      "loss": 0.265,
+      "step": 953
+    },
+    {
+      "epoch": 29.81,
+      "learning_rate": 0.00024224999999999998,
+      "loss": 0.2727,
+      "step": 954
+    },
+    {
+      "epoch": 29.84,
+      "learning_rate": 0.00024187499999999998,
+      "loss": 0.2872,
+      "step": 955
+    },
+    {
+      "epoch": 29.88,
+      "learning_rate": 0.0002415,
+      "loss": 0.2517,
+      "step": 956
+    },
+    {
+      "epoch": 29.91,
+      "learning_rate": 0.00024112499999999997,
+      "loss": 0.265,
+      "step": 957
+    },
+    {
+      "epoch": 29.94,
+      "learning_rate": 0.00024074999999999997,
+      "loss": 0.2555,
+      "step": 958
+    },
+    {
+      "epoch": 29.97,
+      "learning_rate": 0.00024037499999999997,
+      "loss": 0.2953,
+      "step": 959
+    },
+    {
+      "epoch": 30.0,
+      "learning_rate": 0.00023999999999999998,
+      "loss": 0.3151,
+      "step": 960
+    },
+    {
+      "epoch": 30.0,
+      "eval_loss": 1.0953387022018433,
+      "eval_mean_accuracy": 0.5538735449840179,
+      "eval_mean_iou": 0.4172366058027075,
+      "eval_overall_accuracy": 0.7053768760279605,
+      "eval_per_category_accuracy": [
+        0.7224544019110442,
+        0.7824803265782354,
+        0.7538798649370128,
+        0.393189499031929,
+        0.582250656899112,
+        0.5941627138082086,
+        0.6110833835840094,
+        0.48486624501372,
+        0.4447086590419066,
+        0.6095784779183472,
+        0.20903973824120864,
+        0.4587885728434801
+      ],
+      "eval_per_category_iou": [
+        0.6080449519519534,
+        0.5874099730246047,
+        0.5948282267183069,
+        0.2800942232636546,
+        0.4068426424394449,
+        0.40096447981033695,
+        0.4774651092699549,
+        0.42381419759908145,
+        0.25314633570426526,
+        0.446620261464882,
+        0.17328380627613363,
+        0.35432506210987114
+      ],
+      "eval_runtime": 121.2717,
+      "eval_samples_per_second": 0.94,
+      "eval_steps_per_second": 0.082,
+      "step": 960
+    },
+    {
+      "epoch": 30.03,
+      "learning_rate": 0.00023962499999999996,
+      "loss": 0.2554,
+      "step": 961
+    },
+    {
+      "epoch": 30.06,
+      "learning_rate": 0.00023925,
+      "loss": 0.273,
+      "step": 962
+    },
+    {
+      "epoch": 30.09,
+      "learning_rate": 0.000238875,
+      "loss": 0.3226,
+      "step": 963
+    },
+    {
+      "epoch": 30.12,
+      "learning_rate": 0.0002385,
+      "loss": 0.2494,
+      "step": 964
+    },
+    {
+      "epoch": 30.16,
+      "learning_rate": 0.00023812499999999997,
+      "loss": 0.3182,
+      "step": 965
+    },
+    {
+      "epoch": 30.19,
+      "learning_rate": 0.00023774999999999998,
+      "loss": 0.3089,
+      "step": 966
+    },
+    {
+      "epoch": 30.22,
+      "learning_rate": 0.00023737499999999998,
+      "loss": 0.2403,
+      "step": 967
+    },
+    {
+      "epoch": 30.25,
+      "learning_rate": 0.000237,
+      "loss": 0.2629,
+      "step": 968
+    },
+    {
+      "epoch": 30.28,
+      "learning_rate": 0.00023662499999999996,
+      "loss": 0.2556,
+      "step": 969
+    },
+    {
+      "epoch": 30.31,
+      "learning_rate": 0.00023624999999999997,
+      "loss": 0.2316,
+      "step": 970
+    },
+    {
+      "epoch": 30.34,
+      "learning_rate": 0.00023587499999999997,
+      "loss": 0.2726,
+      "step": 971
+    },
+    {
+      "epoch": 30.38,
+      "learning_rate": 0.00023549999999999998,
+      "loss": 0.3058,
+      "step": 972
+    },
+    {
+      "epoch": 30.41,
+      "learning_rate": 0.00023512499999999996,
+      "loss": 0.2135,
+      "step": 973
+    },
+    {
+      "epoch": 30.44,
+      "learning_rate": 0.00023474999999999996,
+      "loss": 0.2668,
+      "step": 974
+    },
+    {
+      "epoch": 30.47,
+      "learning_rate": 0.000234375,
+      "loss": 0.2747,
+      "step": 975
+    },
+    {
+      "epoch": 30.5,
+      "learning_rate": 0.000234,
+      "loss": 0.2319,
+      "step": 976
+    },
+    {
+      "epoch": 30.53,
+      "learning_rate": 0.000233625,
+      "loss": 0.2292,
+      "step": 977
+    },
+    {
+      "epoch": 30.56,
+      "learning_rate": 0.00023324999999999998,
+      "loss": 0.2578,
+      "step": 978
+    },
+    {
+      "epoch": 30.59,
+      "learning_rate": 0.00023287499999999998,
+      "loss": 0.2067,
+      "step": 979
+    },
+    {
+      "epoch": 30.62,
+      "learning_rate": 0.00023249999999999999,
+      "loss": 0.2548,
+      "step": 980
+    },
+    {
+      "epoch": 30.62,
+      "eval_loss": 1.1468321084976196,
+      "eval_mean_accuracy": 0.535823572920069,
+      "eval_mean_iou": 0.41210369379384004,
+      "eval_overall_accuracy": 0.7033535472133703,
+      "eval_per_category_accuracy": [
+        0.7125963954415347,
+        0.8034614291953271,
+        0.7514321122439284,
+        0.39369261837183644,
+        0.568446152505941,
+        0.5642195841605,
+        0.5853602855392986,
+        0.44800072001722896,
+        0.38008970479933984,
+        0.5576315571246103,
+        0.24405824071891843,
+        0.42089407492236425
+      ],
+      "eval_per_category_iou": [
+        0.6040713635052095,
+        0.5863880758702054,
+        0.5928431028454735,
+        0.2877551330952767,
+        0.4000327908584789,
+        0.3917724126721326,
+        0.4730418164838325,
+        0.40541937310500836,
+        0.24892678299297888,
+        0.4232548049497782,
+        0.1939322743756065,
+        0.3378063947720991
+      ],
+      "eval_runtime": 121.5815,
+      "eval_samples_per_second": 0.938,
+      "eval_steps_per_second": 0.082,
+      "step": 980
+    },
+    {
+      "epoch": 30.66,
+      "learning_rate": 0.000232125,
+      "loss": 0.2705,
+      "step": 981
+    },
+    {
+      "epoch": 30.69,
+      "learning_rate": 0.00023174999999999997,
+      "loss": 0.2559,
+      "step": 982
+    },
+    {
+      "epoch": 30.72,
+      "learning_rate": 0.00023137499999999997,
+      "loss": 0.2682,
+      "step": 983
+    },
+    {
+      "epoch": 30.75,
+      "learning_rate": 0.00023099999999999998,
+      "loss": 0.2325,
+      "step": 984
+    },
+    {
+      "epoch": 30.78,
+      "learning_rate": 0.00023062499999999998,
+      "loss": 0.2541,
+      "step": 985
+    },
+    {
+      "epoch": 30.81,
+      "learning_rate": 0.00023024999999999996,
+      "loss": 0.2479,
+      "step": 986
+    },
+    {
+      "epoch": 30.84,
+      "learning_rate": 0.00022987499999999996,
+      "loss": 0.2397,
+      "step": 987
+    },
+    {
+      "epoch": 30.88,
+      "learning_rate": 0.0002295,
+      "loss": 0.25,
+      "step": 988
+    },
+    {
+      "epoch": 30.91,
+      "learning_rate": 0.000229125,
+      "loss": 0.2854,
+      "step": 989
+    },
+    {
+      "epoch": 30.94,
+      "learning_rate": 0.00022874999999999998,
+      "loss": 0.2355,
+      "step": 990
+    },
+    {
+      "epoch": 30.97,
+      "learning_rate": 0.00022837499999999998,
+      "loss": 0.2723,
+      "step": 991
+    },
+    {
+      "epoch": 31.0,
+      "learning_rate": 0.00022799999999999999,
+      "loss": 0.2641,
+      "step": 992
+    },
+    {
+      "epoch": 31.03,
+      "learning_rate": 0.000227625,
+      "loss": 0.248,
+      "step": 993
+    },
+    {
+      "epoch": 31.06,
+      "learning_rate": 0.00022724999999999997,
+      "loss": 0.2873,
+      "step": 994
+    },
+    {
+      "epoch": 31.09,
+      "learning_rate": 0.00022687499999999997,
+      "loss": 0.2324,
+      "step": 995
+    },
+    {
+      "epoch": 31.12,
+      "learning_rate": 0.00022649999999999998,
+      "loss": 0.2885,
+      "step": 996
+    },
+    {
+      "epoch": 31.16,
+      "learning_rate": 0.00022612499999999998,
+      "loss": 0.241,
+      "step": 997
+    },
+    {
+      "epoch": 31.19,
+      "learning_rate": 0.00022574999999999996,
+      "loss": 0.3232,
+      "step": 998
+    },
+    {
+      "epoch": 31.22,
+      "learning_rate": 0.00022537499999999996,
+      "loss": 0.2455,
+      "step": 999
+    },
+    {
+      "epoch": 31.25,
+      "learning_rate": 0.000225,
+      "loss": 0.2763,
+      "step": 1000
+    },
+    {
+      "epoch": 31.25,
+      "eval_loss": 1.1431658267974854,
+      "eval_mean_accuracy": 0.5426868227912205,
+      "eval_mean_iou": 0.4177925601277734,
+      "eval_overall_accuracy": 0.7066975643760279,
+      "eval_per_category_accuracy": [
+        0.7353984134445035,
+        0.7787834905757239,
+        0.771003005045886,
+        0.41441766538747127,
+        0.5323688683507325,
+        0.5071351061011327,
+        0.597358257668673,
+        0.4926246449468612,
+        0.3493762366574093,
+        0.6489998492719942,
+        0.2640390801712818,
+        0.42073725587297656
+      ],
+      "eval_per_category_iou": [
+        0.6158041196029302,
+        0.5862280410550457,
+        0.59375,
+        0.29192320383247583,
+        0.40224742148107556,
+        0.3852280386894547,
+        0.4755897842800655,
+        0.43218721858649817,
+        0.24180229086035454,
+        0.4406818496521092,
+        0.20661681580409666,
+        0.3414519376891743
+      ],
+      "eval_runtime": 121.6147,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 1000
+    },
+    {
+      "epoch": 31.28,
+      "learning_rate": 0.000224625,
+      "loss": 0.2853,
+      "step": 1001
+    },
+    {
+      "epoch": 31.31,
+      "learning_rate": 0.00022425,
+      "loss": 0.2803,
+      "step": 1002
+    },
+    {
+      "epoch": 31.34,
+      "learning_rate": 0.00022387499999999998,
+      "loss": 0.2319,
+      "step": 1003
+    },
+    {
+      "epoch": 31.38,
+      "learning_rate": 0.00022349999999999998,
+      "loss": 0.2434,
+      "step": 1004
+    },
+    {
+      "epoch": 31.41,
+      "learning_rate": 0.000223125,
+      "loss": 0.269,
+      "step": 1005
+    },
+    {
+      "epoch": 31.44,
+      "learning_rate": 0.00022275,
+      "loss": 0.2882,
+      "step": 1006
+    },
+    {
+      "epoch": 31.47,
+      "learning_rate": 0.00022237499999999997,
+      "loss": 0.2216,
+      "step": 1007
+    },
+    {
+      "epoch": 31.5,
+      "learning_rate": 0.00022199999999999998,
+      "loss": 0.247,
+      "step": 1008
+    },
+    {
+      "epoch": 31.53,
+      "learning_rate": 0.00022162499999999998,
+      "loss": 0.2632,
+      "step": 1009
+    },
+    {
+      "epoch": 31.56,
+      "learning_rate": 0.00022124999999999998,
+      "loss": 0.2219,
+      "step": 1010
+    },
+    {
+      "epoch": 31.59,
+      "learning_rate": 0.00022087499999999996,
+      "loss": 0.2981,
+      "step": 1011
+    },
+    {
+      "epoch": 31.62,
+      "learning_rate": 0.00022049999999999997,
+      "loss": 0.218,
+      "step": 1012
+    },
+    {
+      "epoch": 31.66,
+      "learning_rate": 0.000220125,
+      "loss": 0.2882,
+      "step": 1013
+    },
+    {
+      "epoch": 31.69,
+      "learning_rate": 0.00021975,
+      "loss": 0.2499,
+      "step": 1014
+    },
+    {
+      "epoch": 31.72,
+      "learning_rate": 0.00021937499999999998,
+      "loss": 0.2722,
+      "step": 1015
+    },
+    {
+      "epoch": 31.75,
+      "learning_rate": 0.00021899999999999998,
+      "loss": 0.2294,
+      "step": 1016
+    },
+    {
+      "epoch": 31.78,
+      "learning_rate": 0.000218625,
+      "loss": 0.3083,
+      "step": 1017
+    },
+    {
+      "epoch": 31.81,
+      "learning_rate": 0.00021825,
+      "loss": 0.2963,
+      "step": 1018
+    },
+    {
+      "epoch": 31.84,
+      "learning_rate": 0.00021787499999999997,
+      "loss": 0.2566,
+      "step": 1019
+    },
+    {
+      "epoch": 31.88,
+      "learning_rate": 0.00021749999999999997,
+      "loss": 0.2657,
+      "step": 1020
+    },
+    {
+      "epoch": 31.88,
+      "eval_loss": 1.1816933155059814,
+      "eval_mean_accuracy": 0.5321881908963367,
+      "eval_mean_iou": 0.4105139952484221,
+      "eval_overall_accuracy": 0.6971298686245031,
+      "eval_per_category_accuracy": [
+        0.6392331447751347,
+        0.8221980098027152,
+        0.7585844695827206,
+        0.3918050534000458,
+        0.5576845547166065,
+        0.5289759344412225,
+        0.6227948087771531,
+        0.4863244941932539,
+        0.3432827875034382,
+        0.5762284723638376,
+        0.26690704073605886,
+        0.39223952046385424
+      ],
+      "eval_per_category_iou": [
+        0.5651671201298579,
+        0.5856504833280027,
+        0.5956935670832064,
+        0.29685216147722854,
+        0.40363255664597747,
+        0.38836427294358394,
+        0.4691390602809017,
+        0.42989402579788466,
+        0.23865746309627353,
+        0.42629906196596284,
+        0.20680005596753884,
+        0.3200181142646467
+      ],
+      "eval_runtime": 120.8064,
+      "eval_samples_per_second": 0.944,
+      "eval_steps_per_second": 0.083,
+      "step": 1020
+    },
+    {
+      "epoch": 31.91,
+      "learning_rate": 0.00021712499999999998,
+      "loss": 0.2544,
+      "step": 1021
+    },
+    {
+      "epoch": 31.94,
+      "learning_rate": 0.00021674999999999998,
+      "loss": 0.2061,
+      "step": 1022
+    },
+    {
+      "epoch": 31.97,
+      "learning_rate": 0.00021637499999999996,
+      "loss": 0.2565,
+      "step": 1023
+    },
+    {
+      "epoch": 32.0,
+      "learning_rate": 0.00021599999999999996,
+      "loss": 0.2837,
+      "step": 1024
+    },
+    {
+      "epoch": 32.03,
+      "learning_rate": 0.00021562499999999997,
+      "loss": 0.2679,
+      "step": 1025
+    },
+    {
+      "epoch": 32.06,
+      "learning_rate": 0.00021525,
+      "loss": 0.2731,
+      "step": 1026
+    },
+    {
+      "epoch": 32.09,
+      "learning_rate": 0.000214875,
+      "loss": 0.2612,
+      "step": 1027
+    },
+    {
+      "epoch": 32.12,
+      "learning_rate": 0.00021449999999999998,
+      "loss": 0.2192,
+      "step": 1028
+    },
+    {
+      "epoch": 32.16,
+      "learning_rate": 0.000214125,
+      "loss": 0.378,
+      "step": 1029
+    },
+    {
+      "epoch": 32.19,
+      "learning_rate": 0.00021375,
+      "loss": 0.2191,
+      "step": 1030
+    },
+    {
+      "epoch": 32.22,
+      "learning_rate": 0.000213375,
+      "loss": 0.3041,
+      "step": 1031
+    },
+    {
+      "epoch": 32.25,
+      "learning_rate": 0.00021299999999999997,
+      "loss": 0.3007,
+      "step": 1032
+    },
+    {
+      "epoch": 32.28,
+      "learning_rate": 0.00021262499999999998,
+      "loss": 0.2565,
+      "step": 1033
+    },
+    {
+      "epoch": 32.31,
+      "learning_rate": 0.00021224999999999998,
+      "loss": 0.3082,
+      "step": 1034
+    },
+    {
+      "epoch": 32.34,
+      "learning_rate": 0.000211875,
+      "loss": 0.2468,
+      "step": 1035
+    },
+    {
+      "epoch": 32.38,
+      "learning_rate": 0.00021149999999999996,
+      "loss": 0.2739,
+      "step": 1036
+    },
+    {
+      "epoch": 32.41,
+      "learning_rate": 0.00021112499999999997,
+      "loss": 0.2404,
+      "step": 1037
+    },
+    {
+      "epoch": 32.44,
+      "learning_rate": 0.00021074999999999997,
+      "loss": 0.2498,
+      "step": 1038
+    },
+    {
+      "epoch": 32.47,
+      "learning_rate": 0.000210375,
+      "loss": 0.2989,
+      "step": 1039
+    },
+    {
+      "epoch": 32.5,
+      "learning_rate": 0.00020999999999999998,
+      "loss": 0.3014,
+      "step": 1040
+    },
+    {
+      "epoch": 32.5,
+      "eval_loss": 1.1281774044036865,
+      "eval_mean_accuracy": 0.5466743341302135,
+      "eval_mean_iou": 0.4156135689975531,
+      "eval_overall_accuracy": 0.7047071289597896,
+      "eval_per_category_accuracy": [
+        0.7532959506168706,
+        0.7644732984909485,
+        0.7679936788741365,
+        0.40770709432967156,
+        0.5749553794820575,
+        0.524410337556571,
+        0.6095845226868026,
+        0.4903617336557696,
+        0.3954885850421018,
+        0.6112349213319767,
+        0.20299161649521139,
+        0.45759489100044365
+      ],
+      "eval_per_category_iou": [
+        0.6192489589596465,
+        0.5819954415006119,
+        0.5920585805132285,
+        0.29659265018060293,
+        0.4053515164153353,
+        0.38441974090100683,
+        0.4816514959824744,
+        0.427829182426154,
+        0.251820600897165,
+        0.4276636793604374,
+        0.16661307179033685,
+        0.3521179090436363
+      ],
+      "eval_runtime": 120.3512,
+      "eval_samples_per_second": 0.947,
+      "eval_steps_per_second": 0.083,
+      "step": 1040
+    },
+    {
+      "epoch": 32.53,
+      "learning_rate": 0.00020962499999999999,
+      "loss": 0.234,
+      "step": 1041
+    },
+    {
+      "epoch": 32.56,
+      "learning_rate": 0.00020925,
+      "loss": 0.2569,
+      "step": 1042
+    },
+    {
+      "epoch": 32.59,
+      "learning_rate": 0.000208875,
+      "loss": 0.2844,
+      "step": 1043
+    },
+    {
+      "epoch": 32.62,
+      "learning_rate": 0.00020849999999999997,
+      "loss": 0.2351,
+      "step": 1044
+    },
+    {
+      "epoch": 32.66,
+      "learning_rate": 0.00020812499999999998,
+      "loss": 0.2718,
+      "step": 1045
+    },
+    {
+      "epoch": 32.69,
+      "learning_rate": 0.00020774999999999998,
+      "loss": 0.2739,
+      "step": 1046
+    },
+    {
+      "epoch": 32.72,
+      "learning_rate": 0.00020737499999999999,
+      "loss": 0.2076,
+      "step": 1047
+    },
+    {
+      "epoch": 32.75,
+      "learning_rate": 0.00020699999999999996,
+      "loss": 0.2443,
+      "step": 1048
+    },
+    {
+      "epoch": 32.78,
+      "learning_rate": 0.00020662499999999997,
+      "loss": 0.2224,
+      "step": 1049
+    },
+    {
+      "epoch": 32.81,
+      "learning_rate": 0.00020624999999999997,
+      "loss": 0.2372,
+      "step": 1050
+    },
+    {
+      "epoch": 32.84,
+      "learning_rate": 0.000205875,
+      "loss": 0.2557,
+      "step": 1051
+    },
+    {
+      "epoch": 32.88,
+      "learning_rate": 0.0002055,
+      "loss": 0.2078,
+      "step": 1052
+    },
+    {
+      "epoch": 32.91,
+      "learning_rate": 0.00020512499999999999,
+      "loss": 0.2435,
+      "step": 1053
+    },
+    {
+      "epoch": 32.94,
+      "learning_rate": 0.00020475,
+      "loss": 0.2133,
+      "step": 1054
+    },
+    {
+      "epoch": 32.97,
+      "learning_rate": 0.000204375,
+      "loss": 0.2495,
+      "step": 1055
+    },
+    {
+      "epoch": 33.0,
+      "learning_rate": 0.000204,
+      "loss": 0.2682,
+      "step": 1056
+    },
+    {
+      "epoch": 33.03,
+      "learning_rate": 0.00020362499999999998,
+      "loss": 0.2392,
+      "step": 1057
+    },
+    {
+      "epoch": 33.06,
+      "learning_rate": 0.00020324999999999998,
+      "loss": 0.2775,
+      "step": 1058
+    },
+    {
+      "epoch": 33.09,
+      "learning_rate": 0.00020287499999999998,
+      "loss": 0.2146,
+      "step": 1059
+    },
+    {
+      "epoch": 33.12,
+      "learning_rate": 0.0002025,
+      "loss": 0.2202,
+      "step": 1060
+    },
+    {
+      "epoch": 33.12,
+      "eval_loss": 1.1772456169128418,
+      "eval_mean_accuracy": 0.5455600505697921,
+      "eval_mean_iou": 0.4153538119319073,
+      "eval_overall_accuracy": 0.6988246315403989,
+      "eval_per_category_accuracy": [
+        0.6312917036888891,
+        0.816038772353258,
+        0.7677979271432603,
+        0.42424063677559487,
+        0.5773621524816301,
+        0.5217718725842803,
+        0.6365994435951249,
+        0.5161773513866225,
+        0.33743999716067896,
+        0.5783579618027069,
+        0.2534211984218565,
+        0.4862215894436019
+      ],
+      "eval_per_category_iou": [
+        0.5644964927405993,
+        0.5872290170174665,
+        0.5960872745704526,
+        0.28939071900932073,
+        0.39981986430900285,
+        0.38982623443776665,
+        0.48157508749592814,
+        0.44699313208989705,
+        0.23746651597554808,
+        0.4299236550068545,
+        0.1990011008080604,
+        0.36243664972198936
+      ],
+      "eval_runtime": 120.8384,
+      "eval_samples_per_second": 0.943,
+      "eval_steps_per_second": 0.083,
+      "step": 1060
+    },
+    {
+      "epoch": 33.16,
+      "learning_rate": 0.00020212499999999997,
+      "loss": 0.2736,
+      "step": 1061
+    },
+    {
+      "epoch": 33.19,
+      "learning_rate": 0.00020174999999999997,
+      "loss": 0.2574,
+      "step": 1062
+    },
+    {
+      "epoch": 33.22,
+      "learning_rate": 0.00020137499999999998,
+      "loss": 0.1946,
+      "step": 1063
+    },
+    {
+      "epoch": 33.25,
+      "learning_rate": 0.000201,
+      "loss": 0.2931,
+      "step": 1064
+    },
+    {
+      "epoch": 33.28,
+      "learning_rate": 0.00020062499999999996,
+      "loss": 0.2585,
+      "step": 1065
+    },
+    {
+      "epoch": 33.31,
+      "learning_rate": 0.00020025,
+      "loss": 0.2526,
+      "step": 1066
+    },
+    {
+      "epoch": 33.34,
+      "learning_rate": 0.000199875,
+      "loss": 0.2499,
+      "step": 1067
+    },
+    {
+      "epoch": 33.38,
+      "learning_rate": 0.0001995,
+      "loss": 0.2608,
+      "step": 1068
+    },
+    {
+      "epoch": 33.41,
+      "learning_rate": 0.00019912499999999997,
+      "loss": 0.2576,
+      "step": 1069
+    },
+    {
+      "epoch": 33.44,
+      "learning_rate": 0.00019874999999999998,
+      "loss": 0.2493,
+      "step": 1070
+    },
+    {
+      "epoch": 33.47,
+      "learning_rate": 0.00019837499999999998,
+      "loss": 0.2435,
+      "step": 1071
+    },
+    {
+      "epoch": 33.5,
+      "learning_rate": 0.000198,
+      "loss": 0.2652,
+      "step": 1072
+    },
+    {
+      "epoch": 33.53,
+      "learning_rate": 0.00019762499999999997,
+      "loss": 0.2535,
+      "step": 1073
+    },
+    {
+      "epoch": 33.56,
+      "learning_rate": 0.00019724999999999997,
+      "loss": 0.2255,
+      "step": 1074
+    },
+    {
+      "epoch": 33.59,
+      "learning_rate": 0.00019687499999999997,
+      "loss": 0.2522,
+      "step": 1075
+    },
+    {
+      "epoch": 33.62,
+      "learning_rate": 0.00019649999999999998,
+      "loss": 0.207,
+      "step": 1076
+    },
+    {
+      "epoch": 33.66,
+      "learning_rate": 0.000196125,
+      "loss": 0.2401,
+      "step": 1077
+    },
+    {
+      "epoch": 33.69,
+      "learning_rate": 0.00019574999999999996,
+      "loss": 0.2141,
+      "step": 1078
+    },
+    {
+      "epoch": 33.72,
+      "learning_rate": 0.000195375,
+      "loss": 0.2697,
+      "step": 1079
+    },
+    {
+      "epoch": 33.75,
+      "learning_rate": 0.000195,
+      "loss": 0.284,
+      "step": 1080
+    },
+    {
+      "epoch": 33.75,
+      "eval_loss": 1.1533782482147217,
+      "eval_mean_accuracy": 0.5417320223218387,
+      "eval_mean_iou": 0.41501548178977404,
+      "eval_overall_accuracy": 0.7040917580587822,
+      "eval_per_category_accuracy": [
+        0.6810307750298491,
+        0.8023865842535337,
+        0.7727022784531276,
+        0.3999833450149548,
+        0.5764302370104152,
+        0.5228479452999394,
+        0.6035001235930467,
+        0.4833512087610668,
+        0.40330337258103155,
+        0.6129805498286447,
+        0.1784012267647752,
+        0.46386662127167866
+      ],
+      "eval_per_category_iou": [
+        0.5944681328862195,
+        0.5890893975968118,
+        0.5955567168072265,
+        0.29666879413641883,
+        0.40911136577026325,
+        0.38530974019374475,
+        0.47922598712895736,
+        0.4270812919692619,
+        0.25776532685330766,
+        0.4362124170622798,
+        0.15489196148827666,
+        0.35480464958452035
+      ],
+      "eval_runtime": 122.0109,
+      "eval_samples_per_second": 0.934,
+      "eval_steps_per_second": 0.082,
+      "step": 1080
+    },
+    {
+      "epoch": 33.78,
+      "learning_rate": 0.000194625,
+      "loss": 0.2171,
+      "step": 1081
+    },
+    {
+      "epoch": 33.81,
+      "learning_rate": 0.00019424999999999998,
+      "loss": 0.2444,
+      "step": 1082
+    },
+    {
+      "epoch": 33.84,
+      "learning_rate": 0.00019387499999999998,
+      "loss": 0.2264,
+      "step": 1083
+    },
+    {
+      "epoch": 33.88,
+      "learning_rate": 0.0001935,
+      "loss": 0.2576,
+      "step": 1084
+    },
+    {
+      "epoch": 33.91,
+      "learning_rate": 0.000193125,
+      "loss": 0.2581,
+      "step": 1085
+    },
+    {
+      "epoch": 33.94,
+      "learning_rate": 0.00019274999999999997,
+      "loss": 0.2462,
+      "step": 1086
+    },
+    {
+      "epoch": 33.97,
+      "learning_rate": 0.00019237499999999997,
+      "loss": 0.2761,
+      "step": 1087
+    },
+    {
+      "epoch": 34.0,
+      "learning_rate": 0.00019199999999999998,
+      "loss": 0.2616,
+      "step": 1088
+    },
+    {
+      "epoch": 34.03,
+      "learning_rate": 0.00019162499999999998,
+      "loss": 0.2064,
+      "step": 1089
+    },
+    {
+      "epoch": 34.06,
+      "learning_rate": 0.00019124999999999996,
+      "loss": 0.2012,
+      "step": 1090
+    },
+    {
+      "epoch": 34.09,
+      "learning_rate": 0.000190875,
+      "loss": 0.2801,
+      "step": 1091
+    },
+    {
+      "epoch": 34.12,
+      "learning_rate": 0.0001905,
+      "loss": 0.2011,
+      "step": 1092
+    },
+    {
+      "epoch": 34.16,
+      "learning_rate": 0.000190125,
+      "loss": 0.2571,
+      "step": 1093
+    },
+    {
+      "epoch": 34.19,
+      "learning_rate": 0.00018974999999999998,
+      "loss": 0.2309,
+      "step": 1094
+    },
+    {
+      "epoch": 34.22,
+      "learning_rate": 0.00018937499999999998,
+      "loss": 0.223,
+      "step": 1095
+    },
+    {
+      "epoch": 34.25,
+      "learning_rate": 0.00018899999999999999,
+      "loss": 0.249,
+      "step": 1096
+    },
+    {
+      "epoch": 34.28,
+      "learning_rate": 0.000188625,
+      "loss": 0.214,
+      "step": 1097
+    },
+    {
+      "epoch": 34.31,
+      "learning_rate": 0.00018824999999999997,
+      "loss": 0.2813,
+      "step": 1098
+    },
+    {
+      "epoch": 34.34,
+      "learning_rate": 0.00018787499999999997,
+      "loss": 0.2613,
+      "step": 1099
+    },
+    {
+      "epoch": 34.38,
+      "learning_rate": 0.00018749999999999998,
+      "loss": 0.2279,
+      "step": 1100
+    },
+    {
+      "epoch": 34.38,
+      "eval_loss": 1.192512035369873,
+      "eval_mean_accuracy": 0.5407574615787579,
+      "eval_mean_iou": 0.4143827356829877,
+      "eval_overall_accuracy": 0.7004951343201754,
+      "eval_per_category_accuracy": [
+        0.6698231552874635,
+        0.8025919325914168,
+        0.7661597191447761,
+        0.3873255563805942,
+        0.5848215280577463,
+        0.517186314822947,
+        0.5857877442930713,
+        0.44759035305487666,
+        0.3628630116323434,
+        0.6327759874118646,
+        0.270417767275866,
+        0.4617464689921281
+      ],
+      "eval_per_category_iou": [
+        0.585538330449646,
+        0.5855971099792416,
+        0.5937597026424801,
+        0.29552482209420533,
+        0.4122537055734624,
+        0.38106914892194643,
+        0.4730210350936634,
+        0.4056521599804234,
+        0.23906975248740223,
+        0.43819710960236985,
+        0.20950079675228592,
+        0.353409154618726
+      ],
+      "eval_runtime": 120.3836,
+      "eval_samples_per_second": 0.947,
+      "eval_steps_per_second": 0.083,
+      "step": 1100
+    },
+    {
+      "epoch": 34.41,
+      "learning_rate": 0.00018712499999999998,
+      "loss": 0.1984,
+      "step": 1101
+    },
+    {
+      "epoch": 34.44,
+      "learning_rate": 0.00018675,
+      "loss": 0.2619,
+      "step": 1102
+    },
+    {
+      "epoch": 34.47,
+      "learning_rate": 0.00018637499999999996,
+      "loss": 0.2283,
+      "step": 1103
+    },
+    {
+      "epoch": 34.5,
+      "learning_rate": 0.000186,
+      "loss": 0.2455,
+      "step": 1104
+    },
+    {
+      "epoch": 34.53,
+      "learning_rate": 0.000185625,
+      "loss": 0.232,
+      "step": 1105
+    },
+    {
+      "epoch": 34.56,
+      "learning_rate": 0.00018525,
+      "loss": 0.2437,
+      "step": 1106
+    },
+    {
+      "epoch": 34.59,
+      "learning_rate": 0.00018487499999999998,
+      "loss": 0.2105,
+      "step": 1107
+    },
+    {
+      "epoch": 34.62,
+      "learning_rate": 0.00018449999999999999,
+      "loss": 0.2573,
+      "step": 1108
+    },
+    {
+      "epoch": 34.66,
+      "learning_rate": 0.000184125,
+      "loss": 0.2402,
+      "step": 1109
+    },
+    {
+      "epoch": 34.69,
+      "learning_rate": 0.00018375,
+      "loss": 0.2487,
+      "step": 1110
+    },
+    {
+      "epoch": 34.72,
+      "learning_rate": 0.00018337499999999997,
+      "loss": 0.2527,
+      "step": 1111
+    },
+    {
+      "epoch": 34.75,
+      "learning_rate": 0.00018299999999999998,
+      "loss": 0.2162,
+      "step": 1112
+    },
+    {
+      "epoch": 34.78,
+      "learning_rate": 0.00018262499999999998,
+      "loss": 0.2017,
+      "step": 1113
+    },
+    {
+      "epoch": 34.81,
+      "learning_rate": 0.00018224999999999998,
+      "loss": 0.285,
+      "step": 1114
+    },
+    {
+      "epoch": 34.84,
+      "learning_rate": 0.00018187499999999996,
+      "loss": 0.2513,
+      "step": 1115
+    },
+    {
+      "epoch": 34.88,
+      "learning_rate": 0.00018149999999999997,
+      "loss": 0.2644,
+      "step": 1116
+    },
+    {
+      "epoch": 34.91,
+      "learning_rate": 0.000181125,
+      "loss": 0.2352,
+      "step": 1117
+    },
+    {
+      "epoch": 34.94,
+      "learning_rate": 0.00018075,
+      "loss": 0.2663,
+      "step": 1118
+    },
+    {
+      "epoch": 34.97,
+      "learning_rate": 0.00018037499999999998,
+      "loss": 0.2423,
+      "step": 1119
+    },
+    {
+      "epoch": 35.0,
+      "learning_rate": 0.00017999999999999998,
+      "loss": 0.1794,
+      "step": 1120
+    },
+    {
+      "epoch": 35.0,
+      "eval_loss": 1.206092119216919,
+      "eval_mean_accuracy": 0.5347156884237316,
+      "eval_mean_iou": 0.41176016396103804,
+      "eval_overall_accuracy": 0.6987620905825966,
+      "eval_per_category_accuracy": [
+        0.6473550531350573,
+        0.8230074379755047,
+        0.7621060032718504,
+        0.436999049277937,
+        0.5386957640390027,
+        0.49838498361393485,
+        0.5846100678738375,
+        0.4864723548742059,
+        0.3987116580747629,
+        0.5589208292022212,
+        0.2367337693939329,
+        0.44459129035253336
+      ],
+      "eval_per_category_iou": [
+        0.572379474871234,
+        0.5882666519040893,
+        0.5917878602959522,
+        0.308000068474893,
+        0.39821487405277506,
+        0.3732805451696248,
+        0.4741718598355213,
+        0.42927499865269503,
+        0.24856113217670764,
+        0.4172304215858955,
+        0.19014792615738635,
+        0.3498061543556824
+      ],
+      "eval_runtime": 120.9969,
+      "eval_samples_per_second": 0.942,
+      "eval_steps_per_second": 0.083,
+      "step": 1120
+    },
+    {
+      "epoch": 35.03,
+      "learning_rate": 0.000179625,
+      "loss": 0.2687,
+      "step": 1121
+    },
+    {
+      "epoch": 35.06,
+      "learning_rate": 0.00017925,
+      "loss": 0.2899,
+      "step": 1122
+    },
+    {
+      "epoch": 35.09,
+      "learning_rate": 0.00017887499999999997,
+      "loss": 0.2121,
+      "step": 1123
+    },
+    {
+      "epoch": 35.12,
+      "learning_rate": 0.00017849999999999997,
+      "loss": 0.2162,
+      "step": 1124
+    },
+    {
+      "epoch": 35.16,
+      "learning_rate": 0.00017812499999999998,
+      "loss": 0.2421,
+      "step": 1125
+    },
+    {
+      "epoch": 35.19,
+      "learning_rate": 0.00017774999999999998,
+      "loss": 0.2664,
+      "step": 1126
+    },
+    {
+      "epoch": 35.22,
+      "learning_rate": 0.000177375,
+      "loss": 0.2348,
+      "step": 1127
+    },
+    {
+      "epoch": 35.25,
+      "learning_rate": 0.00017699999999999997,
+      "loss": 0.241,
+      "step": 1128
+    },
+    {
+      "epoch": 35.28,
+      "learning_rate": 0.00017662499999999997,
+      "loss": 0.2226,
+      "step": 1129
+    },
+    {
+      "epoch": 35.31,
+      "learning_rate": 0.00017625,
+      "loss": 0.237,
+      "step": 1130
+    },
+    {
+      "epoch": 35.34,
+      "learning_rate": 0.000175875,
+      "loss": 0.3088,
+      "step": 1131
+    },
+    {
+      "epoch": 35.38,
+      "learning_rate": 0.00017549999999999998,
+      "loss": 0.2441,
+      "step": 1132
+    },
+    {
+      "epoch": 35.41,
+      "learning_rate": 0.000175125,
+      "loss": 0.2276,
+      "step": 1133
+    },
+    {
+      "epoch": 35.44,
+      "learning_rate": 0.00017475,
+      "loss": 0.2258,
+      "step": 1134
+    },
+    {
+      "epoch": 35.47,
+      "learning_rate": 0.000174375,
+      "loss": 0.2253,
+      "step": 1135
+    },
+    {
+      "epoch": 35.5,
+      "learning_rate": 0.00017399999999999997,
+      "loss": 0.2254,
+      "step": 1136
+    },
+    {
+      "epoch": 35.53,
+      "learning_rate": 0.00017362499999999998,
+      "loss": 0.271,
+      "step": 1137
+    },
+    {
+      "epoch": 35.56,
+      "learning_rate": 0.00017324999999999998,
+      "loss": 0.2412,
+      "step": 1138
+    },
+    {
+      "epoch": 35.59,
+      "learning_rate": 0.000172875,
+      "loss": 0.2575,
+      "step": 1139
+    },
+    {
+      "epoch": 35.62,
+      "learning_rate": 0.00017249999999999996,
+      "loss": 0.2021,
+      "step": 1140
+    },
+    {
+      "epoch": 35.62,
+      "eval_loss": 1.1695531606674194,
+      "eval_mean_accuracy": 0.5486888116143261,
+      "eval_mean_iou": 0.41512775111190786,
+      "eval_overall_accuracy": 0.7022644846062911,
+      "eval_per_category_accuracy": [
+        0.6808699844157919,
+        0.8069052986823912,
+        0.7333430543376558,
+        0.4240914358678982,
+        0.5945235784743733,
+        0.5769927306116376,
+        0.6103111238465305,
+        0.49709903772697417,
+        0.37890295733032836,
+        0.6123656421508521,
+        0.25974785209036677,
+        0.40911304383711455
+      ],
+      "eval_per_category_iou": [
+        0.5943270940074863,
+        0.5878542482460603,
+        0.5932866960452494,
+        0.2997890605837626,
+        0.39529808898150404,
+        0.3916421557759483,
+        0.4808459120062469,
+        0.43273402901881486,
+        0.24008804380826015,
+        0.43682411605625127,
+        0.19610711331105657,
+        0.3327364555022542
+      ],
+      "eval_runtime": 121.8454,
+      "eval_samples_per_second": 0.936,
+      "eval_steps_per_second": 0.082,
+      "step": 1140
+    },
+    {
+      "epoch": 35.66,
+      "learning_rate": 0.00017212499999999997,
+      "loss": 0.2405,
+      "step": 1141
+    },
+    {
+      "epoch": 35.69,
+      "learning_rate": 0.00017175,
+      "loss": 0.216,
+      "step": 1142
+    },
+    {
+      "epoch": 35.72,
+      "learning_rate": 0.000171375,
+      "loss": 0.2852,
+      "step": 1143
+    },
+    {
+      "epoch": 35.75,
+      "learning_rate": 0.00017099999999999998,
+      "loss": 0.1981,
+      "step": 1144
+    },
+    {
+      "epoch": 35.78,
+      "learning_rate": 0.000170625,
+      "loss": 0.2386,
+      "step": 1145
+    },
+    {
+      "epoch": 35.81,
+      "learning_rate": 0.00017025,
+      "loss": 0.2219,
+      "step": 1146
+    },
+    {
+      "epoch": 35.84,
+      "learning_rate": 0.000169875,
+      "loss": 0.2623,
+      "step": 1147
+    },
+    {
+      "epoch": 35.88,
+      "learning_rate": 0.00016949999999999997,
+      "loss": 0.2568,
+      "step": 1148
+    },
+    {
+      "epoch": 35.91,
+      "learning_rate": 0.00016912499999999998,
+      "loss": 0.2559,
+      "step": 1149
+    },
+    {
+      "epoch": 35.94,
+      "learning_rate": 0.00016874999999999998,
+      "loss": 0.2255,
+      "step": 1150
+    },
+    {
+      "epoch": 35.97,
+      "learning_rate": 0.00016837499999999999,
+      "loss": 0.2982,
+      "step": 1151
+    },
+    {
+      "epoch": 36.0,
+      "learning_rate": 0.000168,
+      "loss": 0.3005,
+      "step": 1152
+    },
+    {
+      "epoch": 36.03,
+      "learning_rate": 0.00016762499999999997,
+      "loss": 0.217,
+      "step": 1153
+    },
+    {
+      "epoch": 36.06,
+      "learning_rate": 0.00016724999999999997,
+      "loss": 0.2403,
+      "step": 1154
+    },
+    {
+      "epoch": 36.09,
+      "learning_rate": 0.000166875,
+      "loss": 0.2423,
+      "step": 1155
+    },
+    {
+      "epoch": 36.12,
+      "learning_rate": 0.0001665,
+      "loss": 0.2376,
+      "step": 1156
+    },
+    {
+      "epoch": 36.16,
+      "learning_rate": 0.00016612499999999999,
+      "loss": 0.2437,
+      "step": 1157
+    },
+    {
+      "epoch": 36.19,
+      "learning_rate": 0.00016575,
+      "loss": 0.2097,
+      "step": 1158
+    },
+    {
+      "epoch": 36.22,
+      "learning_rate": 0.000165375,
+      "loss": 0.2058,
+      "step": 1159
+    },
+    {
+      "epoch": 36.25,
+      "learning_rate": 0.000165,
+      "loss": 0.2455,
+      "step": 1160
+    },
+    {
+      "epoch": 36.25,
+      "eval_loss": 1.1927894353866577,
+      "eval_mean_accuracy": 0.5286617186936045,
+      "eval_mean_iou": 0.40662404560982357,
+      "eval_overall_accuracy": 0.701735212091814,
+      "eval_per_category_accuracy": [
+        0.6739947539676002,
+        0.8244083340380393,
+        0.7385555862150832,
+        0.42733568816316336,
+        0.5720137680381356,
+        0.5028671077415539,
+        0.6549384475138904,
+        0.4887502665242347,
+        0.36404310444265015,
+        0.5348565032875916,
+        0.1709114727604962,
+        0.3912655916308149
+      ],
+      "eval_per_category_iou": [
+        0.5895928623616953,
+        0.5895955023892184,
+        0.5940881841955401,
+        0.29616848747478003,
+        0.3984611932526203,
+        0.3716724785407725,
+        0.4835395335402019,
+        0.43027414936726055,
+        0.24190659809587822,
+        0.4155131309398784,
+        0.15100967896346656,
+        0.3176667481965697
+      ],
+      "eval_runtime": 120.7434,
+      "eval_samples_per_second": 0.944,
+      "eval_steps_per_second": 0.083,
+      "step": 1160
+    },
+    {
+      "epoch": 36.28,
+      "learning_rate": 0.00016462499999999998,
+      "loss": 0.2383,
+      "step": 1161
+    },
+    {
+      "epoch": 36.31,
+      "learning_rate": 0.00016424999999999998,
+      "loss": 0.2802,
+      "step": 1162
+    },
+    {
+      "epoch": 36.34,
+      "learning_rate": 0.00016387499999999999,
+      "loss": 0.1941,
+      "step": 1163
+    },
+    {
+      "epoch": 36.38,
+      "learning_rate": 0.0001635,
+      "loss": 0.2535,
+      "step": 1164
+    },
+    {
+      "epoch": 36.41,
+      "learning_rate": 0.00016312499999999997,
+      "loss": 0.2434,
+      "step": 1165
+    },
+    {
+      "epoch": 36.44,
+      "learning_rate": 0.00016274999999999997,
+      "loss": 0.2463,
+      "step": 1166
+    },
+    {
+      "epoch": 36.47,
+      "learning_rate": 0.00016237499999999998,
+      "loss": 0.2311,
+      "step": 1167
+    },
+    {
+      "epoch": 36.5,
+      "learning_rate": 0.000162,
+      "loss": 0.3015,
+      "step": 1168
+    },
+    {
+      "epoch": 36.53,
+      "learning_rate": 0.00016162499999999996,
+      "loss": 0.2338,
+      "step": 1169
+    },
+    {
+      "epoch": 36.56,
+      "learning_rate": 0.00016125,
+      "loss": 0.2364,
+      "step": 1170
+    },
+    {
+      "epoch": 36.59,
+      "learning_rate": 0.000160875,
+      "loss": 0.2417,
+      "step": 1171
+    },
+    {
+      "epoch": 36.62,
+      "learning_rate": 0.0001605,
+      "loss": 0.2409,
+      "step": 1172
+    },
+    {
+      "epoch": 36.66,
+      "learning_rate": 0.00016012499999999998,
+      "loss": 0.265,
+      "step": 1173
+    },
+    {
+      "epoch": 36.69,
+      "learning_rate": 0.00015974999999999998,
+      "loss": 0.2353,
+      "step": 1174
+    },
+    {
+      "epoch": 36.72,
+      "learning_rate": 0.00015937499999999998,
+      "loss": 0.276,
+      "step": 1175
+    },
+    {
+      "epoch": 36.75,
+      "learning_rate": 0.000159,
+      "loss": 0.2353,
+      "step": 1176
+    },
+    {
+      "epoch": 36.78,
+      "learning_rate": 0.000158625,
+      "loss": 0.2506,
+      "step": 1177
+    },
+    {
+      "epoch": 36.81,
+      "learning_rate": 0.00015824999999999997,
+      "loss": 0.2235,
+      "step": 1178
+    },
+    {
+      "epoch": 36.84,
+      "learning_rate": 0.00015787499999999997,
+      "loss": 0.2475,
+      "step": 1179
+    },
+    {
+      "epoch": 36.88,
+      "learning_rate": 0.00015749999999999998,
+      "loss": 0.2149,
+      "step": 1180
+    },
+    {
+      "epoch": 36.88,
+      "eval_loss": 1.1793041229248047,
+      "eval_mean_accuracy": 0.5447525649233147,
+      "eval_mean_iou": 0.41398307085453045,
+      "eval_overall_accuracy": 0.7034388759679961,
+      "eval_per_category_accuracy": [
+        0.690103624044232,
+        0.7961131116854038,
+        0.7614710942322009,
+        0.43997265806621744,
+        0.5809419234330145,
+        0.5098515999549973,
+        0.6271134804589159,
+        0.5144319524788629,
+        0.36333992883951627,
+        0.6188657222088068,
+        0.20328545236951084,
+        0.4315402313080978
+      ],
+      "eval_per_category_iou": [
+        0.598477367963551,
+        0.587562916284497,
+        0.594311420254118,
+        0.2903604746486162,
+        0.4041882322429327,
+        0.3847148130702917,
+        0.4809337088453609,
+        0.44490416201414984,
+        0.2329717344089799,
+        0.4366151525186739,
+        0.1705475723368633,
+        0.34220929566633124
+      ],
+      "eval_runtime": 119.8592,
+      "eval_samples_per_second": 0.951,
+      "eval_steps_per_second": 0.083,
+      "step": 1180
+    },
+    {
+      "epoch": 36.91,
+      "learning_rate": 0.000157125,
+      "loss": 0.1864,
+      "step": 1181
+    },
+    {
+      "epoch": 36.94,
+      "learning_rate": 0.00015675,
+      "loss": 0.2087,
+      "step": 1182
+    },
+    {
+      "epoch": 36.97,
+      "learning_rate": 0.000156375,
+      "loss": 0.2452,
+      "step": 1183
+    },
+    {
+      "epoch": 37.0,
+      "learning_rate": 0.000156,
+      "loss": 0.2422,
+      "step": 1184
+    },
+    {
+      "epoch": 37.03,
+      "learning_rate": 0.000155625,
+      "loss": 0.2206,
+      "step": 1185
+    },
+    {
+      "epoch": 37.06,
+      "learning_rate": 0.00015524999999999998,
+      "loss": 0.241,
+      "step": 1186
+    },
+    {
+      "epoch": 37.09,
+      "learning_rate": 0.00015487499999999998,
+      "loss": 0.2382,
+      "step": 1187
+    },
+    {
+      "epoch": 37.12,
+      "learning_rate": 0.0001545,
+      "loss": 0.2637,
+      "step": 1188
+    },
+    {
+      "epoch": 37.16,
+      "learning_rate": 0.000154125,
+      "loss": 0.214,
+      "step": 1189
+    },
+    {
+      "epoch": 37.19,
+      "learning_rate": 0.00015374999999999997,
+      "loss": 0.2231,
+      "step": 1190
+    },
+    {
+      "epoch": 37.22,
+      "learning_rate": 0.00015337499999999997,
+      "loss": 0.2303,
+      "step": 1191
+    },
+    {
+      "epoch": 37.25,
+      "learning_rate": 0.00015299999999999998,
+      "loss": 0.2353,
+      "step": 1192
+    },
+    {
+      "epoch": 37.28,
+      "learning_rate": 0.000152625,
+      "loss": 0.2069,
+      "step": 1193
+    },
+    {
+      "epoch": 37.31,
+      "learning_rate": 0.00015224999999999996,
+      "loss": 0.2343,
+      "step": 1194
+    },
+    {
+      "epoch": 37.34,
+      "learning_rate": 0.000151875,
+      "loss": 0.2249,
+      "step": 1195
+    },
+    {
+      "epoch": 37.38,
+      "learning_rate": 0.0001515,
+      "loss": 0.2138,
+      "step": 1196
+    },
+    {
+      "epoch": 37.41,
+      "learning_rate": 0.000151125,
+      "loss": 0.1916,
+      "step": 1197
+    },
+    {
+      "epoch": 37.44,
+      "learning_rate": 0.00015074999999999998,
+      "loss": 0.2689,
+      "step": 1198
+    },
+    {
+      "epoch": 37.47,
+      "learning_rate": 0.00015037499999999998,
+      "loss": 0.2109,
+      "step": 1199
+    },
+    {
+      "epoch": 37.5,
+      "learning_rate": 0.00015,
+      "loss": 0.2594,
+      "step": 1200
+    },
+    {
+      "epoch": 37.5,
+      "eval_loss": 1.2089279890060425,
+      "eval_mean_accuracy": 0.5263138758818015,
+      "eval_mean_iou": 0.40955180839829874,
+      "eval_overall_accuracy": 0.7039898320248252,
+      "eval_per_category_accuracy": [
+        0.6883679664568753,
+        0.8146846051644622,
+        0.7600189359837624,
+        0.40760994025024117,
+        0.5038927729955207,
+        0.5025005534606717,
+        0.6290705015673488,
+        0.4880666787383841,
+        0.34420113040469197,
+        0.5878971148364862,
+        0.2201381640767279,
+        0.36931814664644524
+      ],
+      "eval_per_category_iou": [
+        0.5975281600925688,
+        0.5897601388874746,
+        0.5966038398543478,
+        0.2958968287952444,
+        0.3947270277519008,
+        0.37816552500191186,
+        0.4780340178254698,
+        0.43296197974141165,
+        0.23256685341558678,
+        0.4289049105175803,
+        0.17959218326283355,
+        0.30988023563325356
+      ],
+      "eval_runtime": 121.6039,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 1200
+    },
+    {
+      "epoch": 37.53,
+      "learning_rate": 0.000149625,
+      "loss": 0.2088,
+      "step": 1201
+    },
+    {
+      "epoch": 37.56,
+      "learning_rate": 0.00014925,
+      "loss": 0.1996,
+      "step": 1202
+    },
+    {
+      "epoch": 37.59,
+      "learning_rate": 0.000148875,
+      "loss": 0.2592,
+      "step": 1203
+    },
+    {
+      "epoch": 37.62,
+      "learning_rate": 0.00014849999999999998,
+      "loss": 0.2244,
+      "step": 1204
+    },
+    {
+      "epoch": 37.66,
+      "learning_rate": 0.00014812499999999998,
+      "loss": 0.2332,
+      "step": 1205
+    },
+    {
+      "epoch": 37.69,
+      "learning_rate": 0.00014774999999999999,
+      "loss": 0.2074,
+      "step": 1206
+    },
+    {
+      "epoch": 37.72,
+      "learning_rate": 0.000147375,
+      "loss": 0.2358,
+      "step": 1207
+    },
+    {
+      "epoch": 37.75,
+      "learning_rate": 0.000147,
+      "loss": 0.2484,
+      "step": 1208
+    },
+    {
+      "epoch": 37.78,
+      "learning_rate": 0.000146625,
+      "loss": 0.2339,
+      "step": 1209
+    },
+    {
+      "epoch": 37.81,
+      "learning_rate": 0.00014624999999999998,
+      "loss": 0.2141,
+      "step": 1210
+    },
+    {
+      "epoch": 37.84,
+      "learning_rate": 0.00014587499999999998,
+      "loss": 0.2366,
+      "step": 1211
+    },
+    {
+      "epoch": 37.88,
+      "learning_rate": 0.00014549999999999999,
+      "loss": 0.2304,
+      "step": 1212
+    },
+    {
+      "epoch": 37.91,
+      "learning_rate": 0.000145125,
+      "loss": 0.2272,
+      "step": 1213
+    },
+    {
+      "epoch": 37.94,
+      "learning_rate": 0.00014475,
+      "loss": 0.2436,
+      "step": 1214
+    },
+    {
+      "epoch": 37.97,
+      "learning_rate": 0.000144375,
+      "loss": 0.1956,
+      "step": 1215
+    },
+    {
+      "epoch": 38.0,
+      "learning_rate": 0.00014399999999999998,
+      "loss": 0.2949,
+      "step": 1216
+    },
+    {
+      "epoch": 38.03,
+      "learning_rate": 0.00014362499999999998,
+      "loss": 0.194,
+      "step": 1217
+    },
+    {
+      "epoch": 38.06,
+      "learning_rate": 0.00014324999999999999,
+      "loss": 0.2139,
+      "step": 1218
+    },
+    {
+      "epoch": 38.09,
+      "learning_rate": 0.000142875,
+      "loss": 0.2356,
+      "step": 1219
+    },
+    {
+      "epoch": 38.12,
+      "learning_rate": 0.0001425,
+      "loss": 0.2511,
+      "step": 1220
+    },
+    {
+      "epoch": 38.12,
+      "eval_loss": 1.2089366912841797,
+      "eval_mean_accuracy": 0.5381798652178589,
+      "eval_mean_iou": 0.415049205374649,
+      "eval_overall_accuracy": 0.7004293809857285,
+      "eval_per_category_accuracy": [
+        0.6432255977482852,
+        0.8263136270612244,
+        0.7586329794868737,
+        0.4105141532674999,
+        0.5319839467430567,
+        0.5391396499225155,
+        0.5871496294570251,
+        0.5152741154877635,
+        0.39026689617845134,
+        0.5752662122573683,
+        0.26634079451996096,
+        0.414050780484282
+      ],
+      "eval_per_category_iou": [
+        0.5731277864458117,
+        0.5890083915684147,
+        0.5958862828503297,
+        0.30327755390814853,
+        0.39756757820661504,
+        0.3867994412324132,
+        0.4744467740252867,
+        0.4434946839953632,
+        0.24839754450167165,
+        0.43026452269006366,
+        0.20763865524161315,
+        0.33068124983005626
+      ],
+      "eval_runtime": 120.0903,
+      "eval_samples_per_second": 0.949,
+      "eval_steps_per_second": 0.083,
+      "step": 1220
+    },
+    {
+      "epoch": 38.16,
+      "learning_rate": 0.000142125,
+      "loss": 0.2539,
+      "step": 1221
+    },
+    {
+      "epoch": 38.19,
+      "learning_rate": 0.00014174999999999998,
+      "loss": 0.2664,
+      "step": 1222
+    },
+    {
+      "epoch": 38.22,
+      "learning_rate": 0.00014137499999999998,
+      "loss": 0.2463,
+      "step": 1223
+    },
+    {
+      "epoch": 38.25,
+      "learning_rate": 0.00014099999999999998,
+      "loss": 0.227,
+      "step": 1224
+    },
+    {
+      "epoch": 38.28,
+      "learning_rate": 0.000140625,
+      "loss": 0.2247,
+      "step": 1225
+    },
+    {
+      "epoch": 38.31,
+      "learning_rate": 0.00014025,
+      "loss": 0.2054,
+      "step": 1226
+    },
+    {
+      "epoch": 38.34,
+      "learning_rate": 0.000139875,
+      "loss": 0.2078,
+      "step": 1227
+    },
+    {
+      "epoch": 38.38,
+      "learning_rate": 0.0001395,
+      "loss": 0.2359,
+      "step": 1228
+    },
+    {
+      "epoch": 38.41,
+      "learning_rate": 0.00013912499999999998,
+      "loss": 0.2243,
+      "step": 1229
+    },
+    {
+      "epoch": 38.44,
+      "learning_rate": 0.00013874999999999998,
+      "loss": 0.2125,
+      "step": 1230
+    },
+    {
+      "epoch": 38.47,
+      "learning_rate": 0.000138375,
+      "loss": 0.248,
+      "step": 1231
+    },
+    {
+      "epoch": 38.5,
+      "learning_rate": 0.000138,
+      "loss": 0.2346,
+      "step": 1232
+    },
+    {
+      "epoch": 38.53,
+      "learning_rate": 0.000137625,
+      "loss": 0.2263,
+      "step": 1233
+    },
+    {
+      "epoch": 38.56,
+      "learning_rate": 0.00013725,
+      "loss": 0.215,
+      "step": 1234
+    },
+    {
+      "epoch": 38.59,
+      "learning_rate": 0.00013687499999999998,
+      "loss": 0.2269,
+      "step": 1235
+    },
+    {
+      "epoch": 38.62,
+      "learning_rate": 0.00013649999999999998,
+      "loss": 0.2232,
+      "step": 1236
+    },
+    {
+      "epoch": 38.66,
+      "learning_rate": 0.000136125,
+      "loss": 0.2086,
+      "step": 1237
+    },
+    {
+      "epoch": 38.69,
+      "learning_rate": 0.00013575,
+      "loss": 0.1898,
+      "step": 1238
+    },
+    {
+      "epoch": 38.72,
+      "learning_rate": 0.00013537499999999997,
+      "loss": 0.2351,
+      "step": 1239
+    },
+    {
+      "epoch": 38.75,
+      "learning_rate": 0.000135,
+      "loss": 0.1822,
+      "step": 1240
+    },
+    {
+      "epoch": 38.75,
+      "eval_loss": 1.2078841924667358,
+      "eval_mean_accuracy": 0.5374575435555918,
+      "eval_mean_iou": 0.4111054433363706,
+      "eval_overall_accuracy": 0.7002429962158203,
+      "eval_per_category_accuracy": [
+        0.6592882129907363,
+        0.802385371566499,
+        0.7735372195093138,
+        0.4195356035003227,
+        0.5506870850697012,
+        0.5083109832002003,
+        0.6184155213973415,
+        0.46772940638222416,
+        0.36316025305448835,
+        0.6486592769958291,
+        0.1863500993226471,
+        0.4514314896777988
+      ],
+      "eval_per_category_iou": [
+        0.5817238536919455,
+        0.586264443093182,
+        0.5935163252272255,
+        0.2958494314775711,
+        0.40589922664671685,
+        0.3814028026643284,
+        0.47835308173029206,
+        0.4151528897993265,
+        0.24335850402830217,
+        0.4402164352462257,
+        0.15835483468236958,
+        0.3531734917489618
+      ],
+      "eval_runtime": 122.8719,
+      "eval_samples_per_second": 0.928,
+      "eval_steps_per_second": 0.081,
+      "step": 1240
+    },
+    {
+      "epoch": 38.78,
+      "learning_rate": 0.00013462499999999998,
+      "loss": 0.2076,
+      "step": 1241
+    },
+    {
+      "epoch": 38.81,
+      "learning_rate": 0.00013424999999999998,
+      "loss": 0.2237,
+      "step": 1242
+    },
+    {
+      "epoch": 38.84,
+      "learning_rate": 0.000133875,
+      "loss": 0.2621,
+      "step": 1243
+    },
+    {
+      "epoch": 38.88,
+      "learning_rate": 0.0001335,
+      "loss": 0.2507,
+      "step": 1244
+    },
+    {
+      "epoch": 38.91,
+      "learning_rate": 0.00013312499999999997,
+      "loss": 0.2172,
+      "step": 1245
+    },
+    {
+      "epoch": 38.94,
+      "learning_rate": 0.00013275,
+      "loss": 0.2163,
+      "step": 1246
+    },
+    {
+      "epoch": 38.97,
+      "learning_rate": 0.00013237499999999998,
+      "loss": 0.1983,
+      "step": 1247
+    },
+    {
+      "epoch": 39.0,
+      "learning_rate": 0.00013199999999999998,
+      "loss": 0.268,
+      "step": 1248
+    },
+    {
+      "epoch": 39.03,
+      "learning_rate": 0.000131625,
+      "loss": 0.2105,
+      "step": 1249
+    },
+    {
+      "epoch": 39.06,
+      "learning_rate": 0.00013125,
+      "loss": 0.2187,
+      "step": 1250
+    },
+    {
+      "epoch": 39.09,
+      "learning_rate": 0.000130875,
+      "loss": 0.2175,
+      "step": 1251
+    },
+    {
+      "epoch": 39.12,
+      "learning_rate": 0.0001305,
+      "loss": 0.1908,
+      "step": 1252
+    },
+    {
+      "epoch": 39.16,
+      "learning_rate": 0.000130125,
+      "loss": 0.2414,
+      "step": 1253
+    },
+    {
+      "epoch": 39.19,
+      "learning_rate": 0.00012974999999999998,
+      "loss": 0.2304,
+      "step": 1254
+    },
+    {
+      "epoch": 39.22,
+      "learning_rate": 0.000129375,
+      "loss": 0.1962,
+      "step": 1255
+    },
+    {
+      "epoch": 39.25,
+      "learning_rate": 0.000129,
+      "loss": 0.2076,
+      "step": 1256
+    },
+    {
+      "epoch": 39.28,
+      "learning_rate": 0.000128625,
+      "loss": 0.2357,
+      "step": 1257
+    },
+    {
+      "epoch": 39.31,
+      "learning_rate": 0.00012824999999999997,
+      "loss": 0.2407,
+      "step": 1258
+    },
+    {
+      "epoch": 39.34,
+      "learning_rate": 0.000127875,
+      "loss": 0.2051,
+      "step": 1259
+    },
+    {
+      "epoch": 39.38,
+      "learning_rate": 0.00012749999999999998,
+      "loss": 0.202,
+      "step": 1260
+    },
+    {
+      "epoch": 39.38,
+      "eval_loss": 1.191946029663086,
+      "eval_mean_accuracy": 0.5415699730514477,
+      "eval_mean_iou": 0.41463356090068654,
+      "eval_overall_accuracy": 0.7053561628910533,
+      "eval_per_category_accuracy": [
+        0.7035946173469725,
+        0.8030445073927827,
+        0.7471572482785405,
+        0.40887294328283635,
+        0.5578608082948581,
+        0.5694638508523294,
+        0.6320902654337801,
+        0.48683986366816634,
+        0.39428852825568084,
+        0.5843928190883199,
+        0.18492989259686637,
+        0.42630433212623936
+      ],
+      "eval_per_category_iou": [
+        0.6032537220304772,
+        0.5895821545050566,
+        0.5971628931474557,
+        0.30526634837325817,
+        0.4037920946703913,
+        0.3923213189599239,
+        0.47881924214767163,
+        0.4279964356629629,
+        0.24973726656063663,
+        0.4333172070892052,
+        0.15773466095101843,
+        0.3366193867101802
+      ],
+      "eval_runtime": 121.1835,
+      "eval_samples_per_second": 0.941,
+      "eval_steps_per_second": 0.083,
+      "step": 1260
+    },
+    {
+      "epoch": 39.41,
+      "learning_rate": 0.000127125,
+      "loss": 0.2142,
+      "step": 1261
+    },
+    {
+      "epoch": 39.44,
+      "learning_rate": 0.00012675,
+      "loss": 0.2486,
+      "step": 1262
+    },
+    {
+      "epoch": 39.47,
+      "learning_rate": 0.000126375,
+      "loss": 0.2134,
+      "step": 1263
+    },
+    {
+      "epoch": 39.5,
+      "learning_rate": 0.00012599999999999997,
+      "loss": 0.2855,
+      "step": 1264
+    },
+    {
+      "epoch": 39.53,
+      "learning_rate": 0.000125625,
+      "loss": 0.1992,
+      "step": 1265
+    },
+    {
+      "epoch": 39.56,
+      "learning_rate": 0.00012524999999999998,
+      "loss": 0.1953,
+      "step": 1266
+    },
+    {
+      "epoch": 39.59,
+      "learning_rate": 0.00012487499999999999,
+      "loss": 0.2632,
+      "step": 1267
+    },
+    {
+      "epoch": 39.62,
+      "learning_rate": 0.0001245,
+      "loss": 0.2671,
+      "step": 1268
+    },
+    {
+      "epoch": 39.66,
+      "learning_rate": 0.000124125,
+      "loss": 0.2133,
+      "step": 1269
+    },
+    {
+      "epoch": 39.69,
+      "learning_rate": 0.00012374999999999997,
+      "loss": 0.2342,
+      "step": 1270
+    },
+    {
+      "epoch": 39.72,
+      "learning_rate": 0.000123375,
+      "loss": 0.2251,
+      "step": 1271
+    },
+    {
+      "epoch": 39.75,
+      "learning_rate": 0.00012299999999999998,
+      "loss": 0.2148,
+      "step": 1272
+    },
+    {
+      "epoch": 39.78,
+      "learning_rate": 0.00012262499999999999,
+      "loss": 0.227,
+      "step": 1273
+    },
+    {
+      "epoch": 39.81,
+      "learning_rate": 0.00012225,
+      "loss": 0.2002,
+      "step": 1274
+    },
+    {
+      "epoch": 39.84,
+      "learning_rate": 0.000121875,
+      "loss": 0.2177,
+      "step": 1275
+    },
+    {
+      "epoch": 39.88,
+      "learning_rate": 0.0001215,
+      "loss": 0.2427,
+      "step": 1276
+    },
+    {
+      "epoch": 39.91,
+      "learning_rate": 0.00012112499999999999,
+      "loss": 0.2332,
+      "step": 1277
+    },
+    {
+      "epoch": 39.94,
+      "learning_rate": 0.00012075,
+      "loss": 0.2661,
+      "step": 1278
+    },
+    {
+      "epoch": 39.97,
+      "learning_rate": 0.00012037499999999999,
+      "loss": 0.2777,
+      "step": 1279
+    },
+    {
+      "epoch": 40.0,
+      "learning_rate": 0.00011999999999999999,
+      "loss": 0.2239,
+      "step": 1280
+    },
+    {
+      "epoch": 40.0,
+      "eval_loss": 1.205686330795288,
+      "eval_mean_accuracy": 0.5402591155031062,
+      "eval_mean_iou": 0.41392497699356817,
+      "eval_overall_accuracy": 0.7030476018002159,
+      "eval_per_category_accuracy": [
+        0.6794993729166052,
+        0.8039176420577974,
+        0.7643471608437185,
+        0.40916787530967863,
+        0.5661994258590336,
+        0.5326868428788665,
+        0.6147691329121199,
+        0.45756023447703925,
+        0.3673438151601998,
+        0.6279798118351213,
+        0.2298837205743267,
+        0.4297543512127684
+      ],
+      "eval_per_category_iou": [
+        0.5944679486051297,
+        0.5886137271158206,
+        0.5934208596579145,
+        0.30317435424117073,
+        0.40594125558844635,
+        0.3887621938460642,
+        0.4767783182901575,
+        0.40792180971685443,
+        0.24012023193802226,
+        0.43946154952329713,
+        0.18589535744252184,
+        0.34254211795741785
+      ],
+      "eval_runtime": 120.8929,
+      "eval_samples_per_second": 0.943,
+      "eval_steps_per_second": 0.083,
+      "step": 1280
+    },
+    {
+      "epoch": 40.03,
+      "learning_rate": 0.000119625,
+      "loss": 0.2149,
+      "step": 1281
+    },
+    {
+      "epoch": 40.06,
+      "learning_rate": 0.00011925,
+      "loss": 0.2077,
+      "step": 1282
+    },
+    {
+      "epoch": 40.09,
+      "learning_rate": 0.00011887499999999999,
+      "loss": 0.2276,
+      "step": 1283
+    },
+    {
+      "epoch": 40.12,
+      "learning_rate": 0.0001185,
+      "loss": 0.2491,
+      "step": 1284
+    },
+    {
+      "epoch": 40.16,
+      "learning_rate": 0.00011812499999999998,
+      "loss": 0.2104,
+      "step": 1285
+    },
+    {
+      "epoch": 40.19,
+      "learning_rate": 0.00011774999999999999,
+      "loss": 0.184,
+      "step": 1286
+    },
+    {
+      "epoch": 40.22,
+      "learning_rate": 0.00011737499999999998,
+      "loss": 0.2601,
+      "step": 1287
+    },
+    {
+      "epoch": 40.25,
+      "learning_rate": 0.000117,
+      "loss": 0.1699,
+      "step": 1288
+    },
+    {
+      "epoch": 40.28,
+      "learning_rate": 0.00011662499999999999,
+      "loss": 0.263,
+      "step": 1289
+    },
+    {
+      "epoch": 40.31,
+      "learning_rate": 0.00011624999999999999,
+      "loss": 0.2149,
+      "step": 1290
+    },
+    {
+      "epoch": 40.34,
+      "learning_rate": 0.00011587499999999998,
+      "loss": 0.2045,
+      "step": 1291
+    },
+    {
+      "epoch": 40.38,
+      "learning_rate": 0.00011549999999999999,
+      "loss": 0.1905,
+      "step": 1292
+    },
+    {
+      "epoch": 40.41,
+      "learning_rate": 0.00011512499999999998,
+      "loss": 0.1969,
+      "step": 1293
+    },
+    {
+      "epoch": 40.44,
+      "learning_rate": 0.00011475,
+      "loss": 0.1929,
+      "step": 1294
+    },
+    {
+      "epoch": 40.47,
+      "learning_rate": 0.00011437499999999999,
+      "loss": 0.1865,
+      "step": 1295
+    },
+    {
+      "epoch": 40.5,
+      "learning_rate": 0.00011399999999999999,
+      "loss": 0.2141,
+      "step": 1296
+    },
+    {
+      "epoch": 40.53,
+      "learning_rate": 0.00011362499999999998,
+      "loss": 0.2481,
+      "step": 1297
+    },
+    {
+      "epoch": 40.56,
+      "learning_rate": 0.00011324999999999999,
+      "loss": 0.2185,
+      "step": 1298
+    },
+    {
+      "epoch": 40.59,
+      "learning_rate": 0.00011287499999999998,
+      "loss": 0.2283,
+      "step": 1299
+    },
+    {
+      "epoch": 40.62,
+      "learning_rate": 0.0001125,
+      "loss": 0.205,
+      "step": 1300
+    },
+    {
+      "epoch": 40.62,
+      "eval_loss": 1.2278581857681274,
+      "eval_mean_accuracy": 0.5400553056088017,
+      "eval_mean_iou": 0.41229328192756104,
+      "eval_overall_accuracy": 0.7019564310709635,
+      "eval_per_category_accuracy": [
+        0.6716197517833441,
+        0.8134176705963655,
+        0.7499810954049992,
+        0.4063434674290949,
+        0.6060023459959036,
+        0.5324092052304755,
+        0.6147856644661331,
+        0.45189974188668086,
+        0.3578032527971749,
+        0.601534609288496,
+        0.2598947700275165,
+        0.4149720923994346
+      ],
+      "eval_per_category_iou": [
+        0.5893931584061195,
+        0.5893101847046803,
+        0.5909261886084519,
+        0.2929218901639836,
+        0.3994303523643181,
+        0.3864072171737126,
+        0.4753690533292104,
+        0.4069755436224773,
+        0.24030057310901584,
+        0.4364868935742486,
+        0.2014099240244507,
+        0.33858840405006346
+      ],
+      "eval_runtime": 121.5488,
+      "eval_samples_per_second": 0.938,
+      "eval_steps_per_second": 0.082,
+      "step": 1300
+    },
+    {
+      "epoch": 40.66,
+      "learning_rate": 0.000112125,
+      "loss": 0.2945,
+      "step": 1301
+    },
+    {
+      "epoch": 40.69,
+      "learning_rate": 0.00011174999999999999,
+      "loss": 0.2677,
+      "step": 1302
+    },
+    {
+      "epoch": 40.72,
+      "learning_rate": 0.000111375,
+      "loss": 0.2366,
+      "step": 1303
+    },
+    {
+      "epoch": 40.75,
+      "learning_rate": 0.00011099999999999999,
+      "loss": 0.2419,
+      "step": 1304
+    },
+    {
+      "epoch": 40.78,
+      "learning_rate": 0.00011062499999999999,
+      "loss": 0.2268,
+      "step": 1305
+    },
+    {
+      "epoch": 40.81,
+      "learning_rate": 0.00011024999999999998,
+      "loss": 0.2102,
+      "step": 1306
+    },
+    {
+      "epoch": 40.84,
+      "learning_rate": 0.000109875,
+      "loss": 0.2075,
+      "step": 1307
+    },
+    {
+      "epoch": 40.88,
+      "learning_rate": 0.00010949999999999999,
+      "loss": 0.2442,
+      "step": 1308
+    },
+    {
+      "epoch": 40.91,
+      "learning_rate": 0.000109125,
+      "loss": 0.2181,
+      "step": 1309
+    },
+    {
+      "epoch": 40.94,
+      "learning_rate": 0.00010874999999999999,
+      "loss": 0.2153,
+      "step": 1310
+    },
+    {
+      "epoch": 40.97,
+      "learning_rate": 0.00010837499999999999,
+      "loss": 0.2156,
+      "step": 1311
+    },
+    {
+      "epoch": 41.0,
+      "learning_rate": 0.00010799999999999998,
+      "loss": 0.2362,
+      "step": 1312
+    },
+    {
+      "epoch": 41.03,
+      "learning_rate": 0.000107625,
+      "loss": 0.221,
+      "step": 1313
+    },
+    {
+      "epoch": 41.06,
+      "learning_rate": 0.00010724999999999999,
+      "loss": 0.2362,
+      "step": 1314
+    },
+    {
+      "epoch": 41.09,
+      "learning_rate": 0.000106875,
+      "loss": 0.2304,
+      "step": 1315
+    },
+    {
+      "epoch": 41.12,
+      "learning_rate": 0.00010649999999999999,
+      "loss": 0.227,
+      "step": 1316
+    },
+    {
+      "epoch": 41.16,
+      "learning_rate": 0.00010612499999999999,
+      "loss": 0.201,
+      "step": 1317
+    },
+    {
+      "epoch": 41.19,
+      "learning_rate": 0.00010574999999999998,
+      "loss": 0.1784,
+      "step": 1318
+    },
+    {
+      "epoch": 41.22,
+      "learning_rate": 0.00010537499999999999,
+      "loss": 0.1798,
+      "step": 1319
+    },
+    {
+      "epoch": 41.25,
+      "learning_rate": 0.00010499999999999999,
+      "loss": 0.2058,
+      "step": 1320
+    },
+    {
+      "epoch": 41.25,
+      "eval_loss": 1.1916838884353638,
+      "eval_mean_accuracy": 0.5428442314377016,
+      "eval_mean_iou": 0.41712092361496583,
+      "eval_overall_accuracy": 0.7062055353532758,
+      "eval_per_category_accuracy": [
+        0.727733179796652,
+        0.7940247837657748,
+        0.7426212868878586,
+        0.3889251289026447,
+        0.5820926364496452,
+        0.5376171794192474,
+        0.6216982156942702,
+        0.5008351985565369,
+        0.37881422854759855,
+        0.5879346664503972,
+        0.24598347785365138,
+        0.4058507949281418
+      ],
+      "eval_per_category_iou": [
+        0.6118488881327784,
+        0.5872956844721544,
+        0.5945713715206621,
+        0.29506190063782795,
+        0.40417361213681346,
+        0.38640237786881293,
+        0.4833673433163957,
+        0.43740537161700604,
+        0.24862637762604278,
+        0.43074469938740323,
+        0.1945945684177902,
+        0.33135888824590265
+      ],
+      "eval_runtime": 121.5819,
+      "eval_samples_per_second": 0.938,
+      "eval_steps_per_second": 0.082,
+      "step": 1320
+    },
+    {
+      "epoch": 41.28,
+      "learning_rate": 0.000104625,
+      "loss": 0.2284,
+      "step": 1321
+    },
+    {
+      "epoch": 41.31,
+      "learning_rate": 0.00010424999999999999,
+      "loss": 0.2351,
+      "step": 1322
+    },
+    {
+      "epoch": 41.34,
+      "learning_rate": 0.00010387499999999999,
+      "loss": 0.2076,
+      "step": 1323
+    },
+    {
+      "epoch": 41.38,
+      "learning_rate": 0.00010349999999999998,
+      "loss": 0.215,
+      "step": 1324
+    },
+    {
+      "epoch": 41.41,
+      "learning_rate": 0.00010312499999999999,
+      "loss": 0.2599,
+      "step": 1325
+    },
+    {
+      "epoch": 41.44,
+      "learning_rate": 0.00010275,
+      "loss": 0.2101,
+      "step": 1326
+    },
+    {
+      "epoch": 41.47,
+      "learning_rate": 0.000102375,
+      "loss": 0.2202,
+      "step": 1327
+    },
+    {
+      "epoch": 41.5,
+      "learning_rate": 0.000102,
+      "loss": 0.2204,
+      "step": 1328
+    },
+    {
+      "epoch": 41.53,
+      "learning_rate": 0.00010162499999999999,
+      "loss": 0.1776,
+      "step": 1329
+    },
+    {
+      "epoch": 41.56,
+      "learning_rate": 0.00010125,
+      "loss": 0.2226,
+      "step": 1330
+    },
+    {
+      "epoch": 41.59,
+      "learning_rate": 0.00010087499999999999,
+      "loss": 0.2157,
+      "step": 1331
+    },
+    {
+      "epoch": 41.62,
+      "learning_rate": 0.0001005,
+      "loss": 0.1871,
+      "step": 1332
+    },
+    {
+      "epoch": 41.66,
+      "learning_rate": 0.000100125,
+      "loss": 0.1964,
+      "step": 1333
+    },
+    {
+      "epoch": 41.69,
+      "learning_rate": 9.975e-05,
+      "loss": 0.1925,
+      "step": 1334
+    },
+    {
+      "epoch": 41.72,
+      "learning_rate": 9.937499999999999e-05,
+      "loss": 0.218,
+      "step": 1335
+    },
+    {
+      "epoch": 41.75,
+      "learning_rate": 9.9e-05,
+      "loss": 0.2306,
+      "step": 1336
+    },
+    {
+      "epoch": 41.78,
+      "learning_rate": 9.862499999999998e-05,
+      "loss": 0.2378,
+      "step": 1337
+    },
+    {
+      "epoch": 41.81,
+      "learning_rate": 9.824999999999999e-05,
+      "loss": 0.2471,
+      "step": 1338
+    },
+    {
+      "epoch": 41.84,
+      "learning_rate": 9.787499999999998e-05,
+      "loss": 0.2303,
+      "step": 1339
+    },
+    {
+      "epoch": 41.88,
+      "learning_rate": 9.75e-05,
+      "loss": 0.2189,
+      "step": 1340
+    },
+    {
+      "epoch": 41.88,
+      "eval_loss": 1.1985454559326172,
+      "eval_mean_accuracy": 0.5384366691500525,
+      "eval_mean_iou": 0.41643004818995394,
+      "eval_overall_accuracy": 0.7061509584125719,
+      "eval_per_category_accuracy": [
+        0.7031471936017376,
+        0.8041545202585837,
+        0.7472445661060159,
+        0.38468157750466686,
+        0.5514022288986987,
+        0.539517091954315,
+        0.5980943054306942,
+        0.49206213148671774,
+        0.3565876684737762,
+        0.6189976744076885,
+        0.22239090577969042,
+        0.442960165898047
+      ],
+      "eval_per_category_iou": [
+        0.603842294138224,
+        0.5894939159875506,
+        0.597915323339282,
+        0.2947145124288574,
+        0.4044935137119342,
+        0.39095628573382213,
+        0.47929178484551793,
+        0.43116520988048407,
+        0.2413364980280709,
+        0.4381123592187494,
+        0.18196571457193303,
+        0.34387316639502147
+      ],
+      "eval_runtime": 120.8889,
+      "eval_samples_per_second": 0.943,
+      "eval_steps_per_second": 0.083,
+      "step": 1340
+    },
+    {
+      "epoch": 41.91,
+      "learning_rate": 9.712499999999999e-05,
+      "loss": 0.2059,
+      "step": 1341
+    },
+    {
+      "epoch": 41.94,
+      "learning_rate": 9.675e-05,
+      "loss": 0.224,
+      "step": 1342
+    },
+    {
+      "epoch": 41.97,
+      "learning_rate": 9.637499999999998e-05,
+      "loss": 0.187,
+      "step": 1343
+    },
+    {
+      "epoch": 42.0,
+      "learning_rate": 9.599999999999999e-05,
+      "loss": 0.2338,
+      "step": 1344
+    },
+    {
+      "epoch": 42.03,
+      "learning_rate": 9.562499999999998e-05,
+      "loss": 0.2027,
+      "step": 1345
+    },
+    {
+      "epoch": 42.06,
+      "learning_rate": 9.525e-05,
+      "loss": 0.2356,
+      "step": 1346
+    },
+    {
+      "epoch": 42.09,
+      "learning_rate": 9.487499999999999e-05,
+      "loss": 0.2426,
+      "step": 1347
+    },
+    {
+      "epoch": 42.12,
+      "learning_rate": 9.449999999999999e-05,
+      "loss": 0.2095,
+      "step": 1348
+    },
+    {
+      "epoch": 42.16,
+      "learning_rate": 9.412499999999998e-05,
+      "loss": 0.2229,
+      "step": 1349
+    },
+    {
+      "epoch": 42.19,
+      "learning_rate": 9.374999999999999e-05,
+      "loss": 0.2428,
+      "step": 1350
+    },
+    {
+      "epoch": 42.22,
+      "learning_rate": 9.3375e-05,
+      "loss": 0.2567,
+      "step": 1351
+    },
+    {
+      "epoch": 42.25,
+      "learning_rate": 9.3e-05,
+      "loss": 0.2076,
+      "step": 1352
+    },
+    {
+      "epoch": 42.28,
+      "learning_rate": 9.2625e-05,
+      "loss": 0.2191,
+      "step": 1353
+    },
+    {
+      "epoch": 42.31,
+      "learning_rate": 9.224999999999999e-05,
+      "loss": 0.2117,
+      "step": 1354
+    },
+    {
+      "epoch": 42.34,
+      "learning_rate": 9.1875e-05,
+      "loss": 0.1953,
+      "step": 1355
+    },
+    {
+      "epoch": 42.38,
+      "learning_rate": 9.149999999999999e-05,
+      "loss": 0.2186,
+      "step": 1356
+    },
+    {
+      "epoch": 42.41,
+      "learning_rate": 9.112499999999999e-05,
+      "loss": 0.1736,
+      "step": 1357
+    },
+    {
+      "epoch": 42.44,
+      "learning_rate": 9.074999999999998e-05,
+      "loss": 0.2272,
+      "step": 1358
+    },
+    {
+      "epoch": 42.47,
+      "learning_rate": 9.0375e-05,
+      "loss": 0.1994,
+      "step": 1359
+    },
+    {
+      "epoch": 42.5,
+      "learning_rate": 8.999999999999999e-05,
+      "loss": 0.2062,
+      "step": 1360
+    },
+    {
+      "epoch": 42.5,
+      "eval_loss": 1.2215803861618042,
+      "eval_mean_accuracy": 0.5385122849354729,
+      "eval_mean_iou": 0.4144403878109735,
+      "eval_overall_accuracy": 0.7025854880349678,
+      "eval_per_category_accuracy": [
+        0.6718564590617552,
+        0.8138226272201672,
+        0.759146899118518,
+        0.41468483910590487,
+        0.5515703788641572,
+        0.5251906263723103,
+        0.6183249914586971,
+        0.48296441379133,
+        0.3684573613834592,
+        0.5922817873107881,
+        0.2620281409065449,
+        0.4018188946320427
+      ],
+      "eval_per_category_iou": [
+        0.5886245196867331,
+        0.5886885271786583,
+        0.594060078047334,
+        0.30461123600087675,
+        0.39737485842392256,
+        0.3834340426039297,
+        0.4841315726539136,
+        0.42507949800264805,
+        0.2404092478789335,
+        0.4352542705486317,
+        0.20264502796760822,
+        0.3289717747384923
+      ],
+      "eval_runtime": 119.7705,
+      "eval_samples_per_second": 0.952,
+      "eval_steps_per_second": 0.083,
+      "step": 1360
+    },
+    {
+      "epoch": 42.53,
+      "learning_rate": 8.9625e-05,
+      "loss": 0.2136,
+      "step": 1361
+    },
+    {
+      "epoch": 42.56,
+      "learning_rate": 8.924999999999999e-05,
+      "loss": 0.1984,
+      "step": 1362
+    },
+    {
+      "epoch": 42.59,
+      "learning_rate": 8.887499999999999e-05,
+      "loss": 0.2049,
+      "step": 1363
+    },
+    {
+      "epoch": 42.62,
+      "learning_rate": 8.849999999999998e-05,
+      "loss": 0.2546,
+      "step": 1364
+    },
+    {
+      "epoch": 42.66,
+      "learning_rate": 8.8125e-05,
+      "loss": 0.1899,
+      "step": 1365
+    },
+    {
+      "epoch": 42.69,
+      "learning_rate": 8.774999999999999e-05,
+      "loss": 0.2047,
+      "step": 1366
+    },
+    {
+      "epoch": 42.72,
+      "learning_rate": 8.7375e-05,
+      "loss": 0.2338,
+      "step": 1367
+    },
+    {
+      "epoch": 42.75,
+      "learning_rate": 8.699999999999999e-05,
+      "loss": 0.2044,
+      "step": 1368
+    },
+    {
+      "epoch": 42.78,
+      "learning_rate": 8.662499999999999e-05,
+      "loss": 0.2675,
+      "step": 1369
+    },
+    {
+      "epoch": 42.81,
+      "learning_rate": 8.624999999999998e-05,
+      "loss": 0.2391,
+      "step": 1370
+    },
+    {
+      "epoch": 42.84,
+      "learning_rate": 8.5875e-05,
+      "loss": 0.2164,
+      "step": 1371
+    },
+    {
+      "epoch": 42.88,
+      "learning_rate": 8.549999999999999e-05,
+      "loss": 0.2141,
+      "step": 1372
+    },
+    {
+      "epoch": 42.91,
+      "learning_rate": 8.5125e-05,
+      "loss": 0.2197,
+      "step": 1373
+    },
+    {
+      "epoch": 42.94,
+      "learning_rate": 8.474999999999999e-05,
+      "loss": 0.2225,
+      "step": 1374
+    },
+    {
+      "epoch": 42.97,
+      "learning_rate": 8.437499999999999e-05,
+      "loss": 0.1972,
+      "step": 1375
+    },
+    {
+      "epoch": 43.0,
+      "learning_rate": 8.4e-05,
+      "loss": 0.1842,
+      "step": 1376
+    },
+    {
+      "epoch": 43.03,
+      "learning_rate": 8.362499999999999e-05,
+      "loss": 0.2227,
+      "step": 1377
+    },
+    {
+      "epoch": 43.06,
+      "learning_rate": 8.325e-05,
+      "loss": 0.2243,
+      "step": 1378
+    },
+    {
+      "epoch": 43.09,
+      "learning_rate": 8.2875e-05,
+      "loss": 0.1999,
+      "step": 1379
+    },
+    {
+      "epoch": 43.12,
+      "learning_rate": 8.25e-05,
+      "loss": 0.2333,
+      "step": 1380
+    },
+    {
+      "epoch": 43.12,
+      "eval_loss": 1.2233190536499023,
+      "eval_mean_accuracy": 0.5419502357871261,
+      "eval_mean_iou": 0.41555464038015716,
+      "eval_overall_accuracy": 0.7028216981051261,
+      "eval_per_category_accuracy": [
+        0.6753528840035953,
+        0.8059138866085881,
+        0.764177376179183,
+        0.4105072136903977,
+        0.5650345315200149,
+        0.5352128010916785,
+        0.6400277730107423,
+        0.4669654595306388,
+        0.38071524271758517,
+        0.6113517485752552,
+        0.24183916770988606,
+        0.40630474480794826
+      ],
+      "eval_per_category_iou": [
+        0.5887592431675703,
+        0.5878009496418574,
+        0.5979673673955744,
+        0.29830210182346295,
+        0.40551307607195874,
+        0.39438344826110733,
+        0.4878772676866356,
+        0.41865737307445505,
+        0.24463496678924712,
+        0.43755979584211024,
+        0.19373331142926498,
+        0.3314667833786434
+      ],
+      "eval_runtime": 118.4439,
+      "eval_samples_per_second": 0.962,
+      "eval_steps_per_second": 0.084,
+      "step": 1380
+    },
+    {
+      "epoch": 43.16,
+      "learning_rate": 8.212499999999999e-05,
+      "loss": 0.1933,
+      "step": 1381
+    },
+    {
+      "epoch": 43.19,
+      "learning_rate": 8.175e-05,
+      "loss": 0.247,
+      "step": 1382
+    },
+    {
+      "epoch": 43.22,
+      "learning_rate": 8.137499999999999e-05,
+      "loss": 0.1892,
+      "step": 1383
+    },
+    {
+      "epoch": 43.25,
+      "learning_rate": 8.1e-05,
+      "loss": 0.1881,
+      "step": 1384
+    },
+    {
+      "epoch": 43.28,
+      "learning_rate": 8.0625e-05,
+      "loss": 0.2508,
+      "step": 1385
+    },
+    {
+      "epoch": 43.31,
+      "learning_rate": 8.025e-05,
+      "loss": 0.2118,
+      "step": 1386
+    },
+    {
+      "epoch": 43.34,
+      "learning_rate": 7.987499999999999e-05,
+      "loss": 0.2577,
+      "step": 1387
+    },
+    {
+      "epoch": 43.38,
+      "learning_rate": 7.95e-05,
+      "loss": 0.2003,
+      "step": 1388
+    },
+    {
+      "epoch": 43.41,
+      "learning_rate": 7.912499999999999e-05,
+      "loss": 0.2151,
+      "step": 1389
+    },
+    {
+      "epoch": 43.44,
+      "learning_rate": 7.874999999999999e-05,
+      "loss": 0.1922,
+      "step": 1390
+    },
+    {
+      "epoch": 43.47,
+      "learning_rate": 7.8375e-05,
+      "loss": 0.2273,
+      "step": 1391
+    },
+    {
+      "epoch": 43.5,
+      "learning_rate": 7.8e-05,
+      "loss": 0.2456,
+      "step": 1392
+    },
+    {
+      "epoch": 43.53,
+      "learning_rate": 7.762499999999999e-05,
+      "loss": 0.2563,
+      "step": 1393
+    },
+    {
+      "epoch": 43.56,
+      "learning_rate": 7.725e-05,
+      "loss": 0.1909,
+      "step": 1394
+    },
+    {
+      "epoch": 43.59,
+      "learning_rate": 7.687499999999998e-05,
+      "loss": 0.1989,
+      "step": 1395
+    },
+    {
+      "epoch": 43.62,
+      "learning_rate": 7.649999999999999e-05,
+      "loss": 0.2266,
+      "step": 1396
+    },
+    {
+      "epoch": 43.66,
+      "learning_rate": 7.612499999999998e-05,
+      "loss": 0.2547,
+      "step": 1397
+    },
+    {
+      "epoch": 43.69,
+      "learning_rate": 7.575e-05,
+      "loss": 0.1986,
+      "step": 1398
+    },
+    {
+      "epoch": 43.72,
+      "learning_rate": 7.537499999999999e-05,
+      "loss": 0.2222,
+      "step": 1399
+    },
+    {
+      "epoch": 43.75,
+      "learning_rate": 7.5e-05,
+      "loss": 0.2484,
+      "step": 1400
+    },
+    {
+      "epoch": 43.75,
+      "eval_loss": 1.201867938041687,
+      "eval_mean_accuracy": 0.544322121292772,
+      "eval_mean_iou": 0.41711693892539214,
+      "eval_overall_accuracy": 0.7045823482044956,
+      "eval_per_category_accuracy": [
+        0.6799161069464629,
+        0.8108142740701803,
+        0.7512329362845237,
+        0.4081824553611703,
+        0.5735939725328044,
+        0.5372433666377536,
+        0.6177298555142179,
+        0.4925871440495183,
+        0.364628714408667,
+        0.6073159931802097,
+        0.2701882079990695,
+        0.41843242852868656
+      ],
+      "eval_per_category_iou": [
+        0.5946168145087501,
+        0.5902119335726044,
+        0.5968576131907426,
+        0.29379929371687735,
+        0.40266003364862923,
+        0.39009347075053297,
+        0.48381828136522037,
+        0.43044762200808206,
+        0.23968487036553696,
+        0.43856714034446787,
+        0.20797806045156075,
+        0.3366681331817012
+      ],
+      "eval_runtime": 119.4683,
+      "eval_samples_per_second": 0.954,
+      "eval_steps_per_second": 0.084,
+      "step": 1400
+    },
+    {
+      "epoch": 43.78,
+      "learning_rate": 7.4625e-05,
+      "loss": 0.1947,
+      "step": 1401
+    },
+    {
+      "epoch": 43.81,
+      "learning_rate": 7.424999999999999e-05,
+      "loss": 0.2332,
+      "step": 1402
+    },
+    {
+      "epoch": 43.84,
+      "learning_rate": 7.387499999999999e-05,
+      "loss": 0.267,
+      "step": 1403
+    },
+    {
+      "epoch": 43.88,
+      "learning_rate": 7.35e-05,
+      "loss": 0.2099,
+      "step": 1404
+    },
+    {
+      "epoch": 43.91,
+      "learning_rate": 7.312499999999999e-05,
+      "loss": 0.2204,
+      "step": 1405
+    },
+    {
+      "epoch": 43.94,
+      "learning_rate": 7.274999999999999e-05,
+      "loss": 0.2214,
+      "step": 1406
+    },
+    {
+      "epoch": 43.97,
+      "learning_rate": 7.2375e-05,
+      "loss": 0.203,
+      "step": 1407
+    },
+    {
+      "epoch": 44.0,
+      "learning_rate": 7.199999999999999e-05,
+      "loss": 0.1898,
+      "step": 1408
+    },
+    {
+      "epoch": 44.03,
+      "learning_rate": 7.162499999999999e-05,
+      "loss": 0.2071,
+      "step": 1409
+    },
+    {
+      "epoch": 44.06,
+      "learning_rate": 7.125e-05,
+      "loss": 0.2148,
+      "step": 1410
+    },
+    {
+      "epoch": 44.09,
+      "learning_rate": 7.087499999999999e-05,
+      "loss": 0.1788,
+      "step": 1411
+    },
+    {
+      "epoch": 44.12,
+      "learning_rate": 7.049999999999999e-05,
+      "loss": 0.1988,
+      "step": 1412
+    },
+    {
+      "epoch": 44.16,
+      "learning_rate": 7.0125e-05,
+      "loss": 0.2435,
+      "step": 1413
+    },
+    {
+      "epoch": 44.19,
+      "learning_rate": 6.975e-05,
+      "loss": 0.2119,
+      "step": 1414
+    },
+    {
+      "epoch": 44.22,
+      "learning_rate": 6.937499999999999e-05,
+      "loss": 0.2299,
+      "step": 1415
+    },
+    {
+      "epoch": 44.25,
+      "learning_rate": 6.9e-05,
+      "loss": 0.2722,
+      "step": 1416
+    },
+    {
+      "epoch": 44.28,
+      "learning_rate": 6.8625e-05,
+      "loss": 0.2116,
+      "step": 1417
+    },
+    {
+      "epoch": 44.31,
+      "learning_rate": 6.824999999999999e-05,
+      "loss": 0.2225,
+      "step": 1418
+    },
+    {
+      "epoch": 44.34,
+      "learning_rate": 6.7875e-05,
+      "loss": 0.2139,
+      "step": 1419
+    },
+    {
+      "epoch": 44.38,
+      "learning_rate": 6.75e-05,
+      "loss": 0.2065,
+      "step": 1420
+    },
+    {
+      "epoch": 44.38,
+      "eval_loss": 1.2391397953033447,
+      "eval_mean_accuracy": 0.5367092637127736,
+      "eval_mean_iou": 0.4139386094862935,
+      "eval_overall_accuracy": 0.7044861776786938,
+      "eval_per_category_accuracy": [
+        0.6754433379015489,
+        0.8167576532274533,
+        0.7571100538488471,
+        0.39708607157479825,
+        0.567514237034726,
+        0.5687434446666352,
+        0.5982619826214006,
+        0.48236439943384357,
+        0.3556183065224528,
+        0.604266489200521,
+        0.21481544964540744,
+        0.4025297388756487
+      ],
+      "eval_per_category_iou": [
+        0.5911919793466145,
+        0.590792266050845,
+        0.5983501230191215,
+        0.29512217136225904,
+        0.4010275894452334,
+        0.3950325683633137,
+        0.47928132806356016,
+        0.4294050655368374,
+        0.2392712478097734,
+        0.4385312759983558,
+        0.17995456456856854,
+        0.3293031342710404
+      ],
+      "eval_runtime": 121.3502,
+      "eval_samples_per_second": 0.939,
+      "eval_steps_per_second": 0.082,
+      "step": 1420
+    },
+    {
+      "epoch": 44.41,
+      "learning_rate": 6.712499999999999e-05,
+      "loss": 0.2315,
+      "step": 1421
+    },
+    {
+      "epoch": 44.44,
+      "learning_rate": 6.675e-05,
+      "loss": 0.2123,
+      "step": 1422
+    },
+    {
+      "epoch": 44.47,
+      "learning_rate": 6.6375e-05,
+      "loss": 0.2273,
+      "step": 1423
+    },
+    {
+      "epoch": 44.5,
+      "learning_rate": 6.599999999999999e-05,
+      "loss": 0.2051,
+      "step": 1424
+    },
+    {
+      "epoch": 44.53,
+      "learning_rate": 6.5625e-05,
+      "loss": 0.2092,
+      "step": 1425
+    },
+    {
+      "epoch": 44.56,
+      "learning_rate": 6.525e-05,
+      "loss": 0.2265,
+      "step": 1426
+    },
+    {
+      "epoch": 44.59,
+      "learning_rate": 6.487499999999999e-05,
+      "loss": 0.1972,
+      "step": 1427
+    },
+    {
+      "epoch": 44.62,
+      "learning_rate": 6.45e-05,
+      "loss": 0.1769,
+      "step": 1428
+    },
+    {
+      "epoch": 44.66,
+      "learning_rate": 6.412499999999999e-05,
+      "loss": 0.2191,
+      "step": 1429
+    },
+    {
+      "epoch": 44.69,
+      "learning_rate": 6.374999999999999e-05,
+      "loss": 0.19,
+      "step": 1430
+    },
+    {
+      "epoch": 44.72,
+      "learning_rate": 6.3375e-05,
+      "loss": 0.2425,
+      "step": 1431
+    },
+    {
+      "epoch": 44.75,
+      "learning_rate": 6.299999999999999e-05,
+      "loss": 0.2113,
+      "step": 1432
+    },
+    {
+      "epoch": 44.78,
+      "learning_rate": 6.262499999999999e-05,
+      "loss": 0.2118,
+      "step": 1433
+    },
+    {
+      "epoch": 44.81,
+      "learning_rate": 6.225e-05,
+      "loss": 0.2026,
+      "step": 1434
+    },
+    {
+      "epoch": 44.84,
+      "learning_rate": 6.187499999999999e-05,
+      "loss": 0.2428,
+      "step": 1435
+    },
+    {
+      "epoch": 44.88,
+      "learning_rate": 6.149999999999999e-05,
+      "loss": 0.1812,
+      "step": 1436
+    },
+    {
+      "epoch": 44.91,
+      "learning_rate": 6.1125e-05,
+      "loss": 0.2303,
+      "step": 1437
+    },
+    {
+      "epoch": 44.94,
+      "learning_rate": 6.075e-05,
+      "loss": 0.1923,
+      "step": 1438
+    },
+    {
+      "epoch": 44.97,
+      "learning_rate": 6.0375e-05,
+      "loss": 0.2639,
+      "step": 1439
+    },
+    {
+      "epoch": 45.0,
+      "learning_rate": 5.9999999999999995e-05,
+      "loss": 0.2072,
+      "step": 1440
+    },
+    {
+      "epoch": 45.0,
+      "eval_loss": 1.2111115455627441,
+      "eval_mean_accuracy": 0.5407650388177646,
+      "eval_mean_iou": 0.4161584309777115,
+      "eval_overall_accuracy": 0.7053119257876748,
+      "eval_per_category_accuracy": [
+        0.693233975012698,
+        0.8080079545801815,
+        0.7471352761454829,
+        0.4088139568774679,
+        0.5584280611903802,
+        0.5118113951201101,
+        0.6278636981243771,
+        0.4885220467775479,
+        0.37354373885344666,
+        0.6002218674521907,
+        0.24139535310807958,
+        0.4302031425712134
+      ],
+      "eval_per_category_iou": [
+        0.6005611522652172,
+        0.5897708054265532,
+        0.5951666816313217,
+        0.29419652220812814,
+        0.4009845537940525,
+        0.38115830783701093,
+        0.4853986167879391,
+        0.42906184626937727,
+        0.24387199175112018,
+        0.4366663403551029,
+        0.195869862163169,
+        0.34119449124354506
+      ],
+      "eval_runtime": 120.1593,
+      "eval_samples_per_second": 0.949,
+      "eval_steps_per_second": 0.083,
+      "step": 1440
+    },
+    {
+      "epoch": 45.03,
+      "learning_rate": 5.9625e-05,
+      "loss": 0.1773,
+      "step": 1441
+    },
+    {
+      "epoch": 45.06,
+      "learning_rate": 5.925e-05,
+      "loss": 0.2137,
+      "step": 1442
+    },
+    {
+      "epoch": 45.09,
+      "learning_rate": 5.8874999999999995e-05,
+      "loss": 0.2357,
+      "step": 1443
+    },
+    {
+      "epoch": 45.12,
+      "learning_rate": 5.85e-05,
+      "loss": 0.1948,
+      "step": 1444
+    },
+    {
+      "epoch": 45.16,
+      "learning_rate": 5.8124999999999997e-05,
+      "loss": 0.1802,
+      "step": 1445
+    },
+    {
+      "epoch": 45.19,
+      "learning_rate": 5.7749999999999994e-05,
+      "loss": 0.2139,
+      "step": 1446
+    },
+    {
+      "epoch": 45.22,
+      "learning_rate": 5.7375e-05,
+      "loss": 0.2451,
+      "step": 1447
+    },
+    {
+      "epoch": 45.25,
+      "learning_rate": 5.6999999999999996e-05,
+      "loss": 0.194,
+      "step": 1448
+    },
+    {
+      "epoch": 45.28,
+      "learning_rate": 5.6624999999999994e-05,
+      "loss": 0.2362,
+      "step": 1449
+    },
+    {
+      "epoch": 45.31,
+      "learning_rate": 5.625e-05,
+      "loss": 0.2293,
+      "step": 1450
+    },
+    {
+      "epoch": 45.34,
+      "learning_rate": 5.5874999999999996e-05,
+      "loss": 0.2094,
+      "step": 1451
+    },
+    {
+      "epoch": 45.38,
+      "learning_rate": 5.5499999999999994e-05,
+      "loss": 0.1936,
+      "step": 1452
+    },
+    {
+      "epoch": 45.41,
+      "learning_rate": 5.512499999999999e-05,
+      "loss": 0.2013,
+      "step": 1453
+    },
+    {
+      "epoch": 45.44,
+      "learning_rate": 5.4749999999999996e-05,
+      "loss": 0.2197,
+      "step": 1454
+    },
+    {
+      "epoch": 45.47,
+      "learning_rate": 5.4374999999999994e-05,
+      "loss": 0.2146,
+      "step": 1455
+    },
+    {
+      "epoch": 45.5,
+      "learning_rate": 5.399999999999999e-05,
+      "loss": 0.212,
+      "step": 1456
+    },
+    {
+      "epoch": 45.53,
+      "learning_rate": 5.3624999999999996e-05,
+      "loss": 0.1773,
+      "step": 1457
+    },
+    {
+      "epoch": 45.56,
+      "learning_rate": 5.324999999999999e-05,
+      "loss": 0.2205,
+      "step": 1458
+    },
+    {
+      "epoch": 45.59,
+      "learning_rate": 5.287499999999999e-05,
+      "loss": 0.2059,
+      "step": 1459
+    },
+    {
+      "epoch": 45.62,
+      "learning_rate": 5.2499999999999995e-05,
+      "loss": 0.1899,
+      "step": 1460
+    },
+    {
+      "epoch": 45.62,
+      "eval_loss": 1.2186626195907593,
+      "eval_mean_accuracy": 0.5382333206115961,
+      "eval_mean_iou": 0.4155721113962822,
+      "eval_overall_accuracy": 0.7060486642937911,
+      "eval_per_category_accuracy": [
+        0.6909288690223152,
+        0.811623783088772,
+        0.7568092924430986,
+        0.4024330157320213,
+        0.5395587988014756,
+        0.5541411560613924,
+        0.6225366016478023,
+        0.4974772610630326,
+        0.364628714408667,
+        0.5905147752561984,
+        0.22572410647877494,
+        0.4024234733356031
+      ],
+      "eval_per_category_iou": [
+        0.6002947575662394,
+        0.5911024089561174,
+        0.5966849933788959,
+        0.2917610005936748,
+        0.4011368531399016,
+        0.3944847702719636,
+        0.4833263658677306,
+        0.43479005950140187,
+        0.24372737004735795,
+        0.4341372699386503,
+        0.18557606411756564,
+        0.32984342337588535
+      ],
+      "eval_runtime": 120.6325,
+      "eval_samples_per_second": 0.945,
+      "eval_steps_per_second": 0.083,
+      "step": 1460
+    },
+    {
+      "epoch": 45.66,
+      "learning_rate": 5.212499999999999e-05,
+      "loss": 0.2081,
+      "step": 1461
+    },
+    {
+      "epoch": 45.69,
+      "learning_rate": 5.174999999999999e-05,
+      "loss": 0.2006,
+      "step": 1462
+    },
+    {
+      "epoch": 45.72,
+      "learning_rate": 5.1375e-05,
+      "loss": 0.2027,
+      "step": 1463
+    },
+    {
+      "epoch": 45.75,
+      "learning_rate": 5.1e-05,
+      "loss": 0.2181,
+      "step": 1464
+    },
+    {
+      "epoch": 45.78,
+      "learning_rate": 5.0625e-05,
+      "loss": 0.183,
+      "step": 1465
+    },
+    {
+      "epoch": 45.81,
+      "learning_rate": 5.025e-05,
+      "loss": 0.2126,
+      "step": 1466
+    },
+    {
+      "epoch": 45.84,
+      "learning_rate": 4.9875e-05,
+      "loss": 0.2874,
+      "step": 1467
+    },
+    {
+      "epoch": 45.88,
+      "learning_rate": 4.95e-05,
+      "loss": 0.2159,
+      "step": 1468
+    },
+    {
+      "epoch": 45.91,
+      "learning_rate": 4.9124999999999995e-05,
+      "loss": 0.2796,
+      "step": 1469
+    },
+    {
+      "epoch": 45.94,
+      "learning_rate": 4.875e-05,
+      "loss": 0.206,
+      "step": 1470
+    },
+    {
+      "epoch": 45.97,
+      "learning_rate": 4.8375e-05,
+      "loss": 0.2062,
+      "step": 1471
+    },
+    {
+      "epoch": 46.0,
+      "learning_rate": 4.7999999999999994e-05,
+      "loss": 0.1917,
+      "step": 1472
+    },
+    {
+      "epoch": 46.03,
+      "learning_rate": 4.7625e-05,
+      "loss": 0.2048,
+      "step": 1473
+    },
+    {
+      "epoch": 46.06,
+      "learning_rate": 4.7249999999999997e-05,
+      "loss": 0.1561,
+      "step": 1474
+    },
+    {
+      "epoch": 46.09,
+      "learning_rate": 4.6874999999999994e-05,
+      "loss": 0.2362,
+      "step": 1475
+    },
+    {
+      "epoch": 46.12,
+      "learning_rate": 4.65e-05,
+      "loss": 0.1971,
+      "step": 1476
+    },
+    {
+      "epoch": 46.16,
+      "learning_rate": 4.6124999999999996e-05,
+      "loss": 0.2101,
+      "step": 1477
+    },
+    {
+      "epoch": 46.19,
+      "learning_rate": 4.5749999999999994e-05,
+      "loss": 0.2068,
+      "step": 1478
+    },
+    {
+      "epoch": 46.22,
+      "learning_rate": 4.537499999999999e-05,
+      "loss": 0.1963,
+      "step": 1479
+    },
+    {
+      "epoch": 46.25,
+      "learning_rate": 4.4999999999999996e-05,
+      "loss": 0.1993,
+      "step": 1480
+    },
+    {
+      "epoch": 46.25,
+      "eval_loss": 1.2132420539855957,
+      "eval_mean_accuracy": 0.542613118394146,
+      "eval_mean_iou": 0.41735489371775647,
+      "eval_overall_accuracy": 0.7050070846289919,
+      "eval_per_category_accuracy": [
+        0.6835652462614714,
+        0.8096551069565711,
+        0.7524941937925015,
+        0.4019923525860334,
+        0.5629032813554102,
+        0.5304221906880696,
+        0.6222461186272827,
+        0.49411503775268906,
+        0.36087548689919524,
+        0.6064872499232017,
+        0.2598304934300135,
+        0.4267706624573132
+      ],
+      "eval_per_category_iou": [
+        0.5963839392150861,
+        0.5900616733858792,
+        0.5971839572158605,
+        0.29516445440880484,
+        0.401339565572847,
+        0.3897724469454019,
+        0.4813518614036359,
+        0.4352635150631947,
+        0.24202099963998916,
+        0.436904019417038,
+        0.2033395691759346,
+        0.33947272316940563
+      ],
+      "eval_runtime": 120.7706,
+      "eval_samples_per_second": 0.944,
+      "eval_steps_per_second": 0.083,
+      "step": 1480
+    },
+    {
+      "epoch": 46.28,
+      "learning_rate": 4.4624999999999994e-05,
+      "loss": 0.2167,
+      "step": 1481
+    },
+    {
+      "epoch": 46.31,
+      "learning_rate": 4.424999999999999e-05,
+      "loss": 0.2552,
+      "step": 1482
+    },
+    {
+      "epoch": 46.34,
+      "learning_rate": 4.3874999999999996e-05,
+      "loss": 0.2103,
+      "step": 1483
+    },
+    {
+      "epoch": 46.38,
+      "learning_rate": 4.3499999999999993e-05,
+      "loss": 0.1995,
+      "step": 1484
+    },
+    {
+      "epoch": 46.41,
+      "learning_rate": 4.312499999999999e-05,
+      "loss": 0.1848,
+      "step": 1485
+    },
+    {
+      "epoch": 46.44,
+      "learning_rate": 4.2749999999999996e-05,
+      "loss": 0.245,
+      "step": 1486
+    },
+    {
+      "epoch": 46.47,
+      "learning_rate": 4.237499999999999e-05,
+      "loss": 0.1811,
+      "step": 1487
+    },
+    {
+      "epoch": 46.5,
+      "learning_rate": 4.2e-05,
+      "loss": 0.2135,
+      "step": 1488
+    },
+    {
+      "epoch": 46.53,
+      "learning_rate": 4.1625e-05,
+      "loss": 0.2282,
+      "step": 1489
+    },
+    {
+      "epoch": 46.56,
+      "learning_rate": 4.125e-05,
+      "loss": 0.2299,
+      "step": 1490
+    },
+    {
+      "epoch": 46.59,
+      "learning_rate": 4.0875e-05,
+      "loss": 0.1918,
+      "step": 1491
+    },
+    {
+      "epoch": 46.62,
+      "learning_rate": 4.05e-05,
+      "loss": 0.2055,
+      "step": 1492
+    },
+    {
+      "epoch": 46.66,
+      "learning_rate": 4.0125e-05,
+      "loss": 0.1748,
+      "step": 1493
+    },
+    {
+      "epoch": 46.69,
+      "learning_rate": 3.975e-05,
+      "loss": 0.1756,
+      "step": 1494
+    },
+    {
+      "epoch": 46.72,
+      "learning_rate": 3.9374999999999995e-05,
+      "loss": 0.1932,
+      "step": 1495
+    },
+    {
+      "epoch": 46.75,
+      "learning_rate": 3.9e-05,
+      "loss": 0.225,
+      "step": 1496
+    },
+    {
+      "epoch": 46.78,
+      "learning_rate": 3.8625e-05,
+      "loss": 0.2625,
+      "step": 1497
+    },
+    {
+      "epoch": 46.81,
+      "learning_rate": 3.8249999999999995e-05,
+      "loss": 0.2108,
+      "step": 1498
+    },
+    {
+      "epoch": 46.84,
+      "learning_rate": 3.7875e-05,
+      "loss": 0.271,
+      "step": 1499
+    },
+    {
+      "epoch": 46.88,
+      "learning_rate": 3.75e-05,
+      "loss": 0.2307,
+      "step": 1500
+    },
+    {
+      "epoch": 46.88,
+      "eval_loss": 1.2265039682388306,
+      "eval_mean_accuracy": 0.5389079411622628,
+      "eval_mean_iou": 0.4159692011117362,
+      "eval_overall_accuracy": 0.7059000584117153,
+      "eval_per_category_accuracy": [
+        0.6889055504185769,
+        0.8124674090359413,
+        0.7571462936007732,
+        0.39943164863533215,
+        0.5560577544483769,
+        0.5379710313240594,
+        0.6159302777773404,
+        0.5005137622935978,
+        0.3676965120715509,
+        0.5859486033146601,
+        0.23176610664405764,
+        0.4130603443828861
+      ],
+      "eval_per_category_iou": [
+        0.5990133388729494,
+        0.5912773584964496,
+        0.5971065220610309,
+        0.29478856973114437,
+        0.40079728979877927,
+        0.3909545989228651,
+        0.4814390468871046,
+        0.436967220092064,
+        0.24266565947579227,
+        0.4335934952195591,
+        0.18898311852968483,
+        0.33404419525341145
+      ],
+      "eval_runtime": 120.8078,
+      "eval_samples_per_second": 0.944,
+      "eval_steps_per_second": 0.083,
+      "step": 1500
+    },
+    {
+      "epoch": 46.91,
+      "learning_rate": 3.7124999999999994e-05,
+      "loss": 0.2222,
+      "step": 1501
+    },
+    {
+      "epoch": 46.94,
+      "learning_rate": 3.675e-05,
+      "loss": 0.1913,
+      "step": 1502
+    },
+    {
+      "epoch": 46.97,
+      "learning_rate": 3.6374999999999996e-05,
+      "loss": 0.2306,
+      "step": 1503
+    },
+    {
+      "epoch": 47.0,
+      "learning_rate": 3.5999999999999994e-05,
+      "loss": 0.2099,
+      "step": 1504
+    },
+    {
+      "epoch": 47.03,
+      "learning_rate": 3.5625e-05,
+      "loss": 0.1849,
+      "step": 1505
+    },
+    {
+      "epoch": 47.06,
+      "learning_rate": 3.5249999999999996e-05,
+      "loss": 0.2198,
+      "step": 1506
+    },
+    {
+      "epoch": 47.09,
+      "learning_rate": 3.4875e-05,
+      "loss": 0.2451,
+      "step": 1507
+    },
+    {
+      "epoch": 47.12,
+      "learning_rate": 3.45e-05,
+      "loss": 0.2289,
+      "step": 1508
+    },
+    {
+      "epoch": 47.16,
+      "learning_rate": 3.4124999999999996e-05,
+      "loss": 0.2109,
+      "step": 1509
+    },
+    {
+      "epoch": 47.19,
+      "learning_rate": 3.375e-05,
+      "loss": 0.1748,
+      "step": 1510
+    },
+    {
+      "epoch": 47.22,
+      "learning_rate": 3.3375e-05,
+      "loss": 0.195,
+      "step": 1511
+    },
+    {
+      "epoch": 47.25,
+      "learning_rate": 3.2999999999999996e-05,
+      "loss": 0.2413,
+      "step": 1512
+    },
+    {
+      "epoch": 47.28,
+      "learning_rate": 3.2625e-05,
+      "loss": 0.1991,
+      "step": 1513
+    },
+    {
+      "epoch": 47.31,
+      "learning_rate": 3.225e-05,
+      "loss": 0.2139,
+      "step": 1514
+    },
+    {
+      "epoch": 47.34,
+      "learning_rate": 3.1874999999999996e-05,
+      "loss": 0.2668,
+      "step": 1515
+    },
+    {
+      "epoch": 47.38,
+      "learning_rate": 3.149999999999999e-05,
+      "loss": 0.2191,
+      "step": 1516
+    },
+    {
+      "epoch": 47.41,
+      "learning_rate": 3.1125e-05,
+      "loss": 0.2043,
+      "step": 1517
+    },
+    {
+      "epoch": 47.44,
+      "learning_rate": 3.0749999999999995e-05,
+      "loss": 0.1925,
+      "step": 1518
+    },
+    {
+      "epoch": 47.47,
+      "learning_rate": 3.0375e-05,
+      "loss": 0.2236,
+      "step": 1519
+    },
+    {
+      "epoch": 47.5,
+      "learning_rate": 2.9999999999999997e-05,
+      "loss": 0.2274,
+      "step": 1520
+    },
+    {
+      "epoch": 47.5,
+      "eval_loss": 1.228417158126831,
+      "eval_mean_accuracy": 0.5417850315296816,
+      "eval_mean_iou": 0.4162704929646823,
+      "eval_overall_accuracy": 0.7056037166662383,
+      "eval_per_category_accuracy": [
+        0.6935326490848371,
+        0.8064145646623322,
+        0.7524120123078187,
+        0.40547255050277237,
+        0.5592242411473095,
+        0.5387694663913275,
+        0.6155697324564787,
+        0.48629663638379916,
+        0.38145169161424275,
+        0.6083450117114095,
+        0.22693005787954565,
+        0.42700176421430563
+      ],
+      "eval_per_category_iou": [
+        0.6012910052936997,
+        0.5902791035178767,
+        0.5959089289092157,
+        0.2969911531413845,
+        0.4017976579500877,
+        0.3901580059501908,
+        0.48168540024898715,
+        0.4292969086226189,
+        0.24509351189101825,
+        0.4384137950994141,
+        0.18475665977223454,
+        0.3395737851794602
+      ],
+      "eval_runtime": 121.0337,
+      "eval_samples_per_second": 0.942,
+      "eval_steps_per_second": 0.083,
+      "step": 1520
+    },
+    {
+      "epoch": 47.53,
+      "learning_rate": 2.9625e-05,
+      "loss": 0.201,
+      "step": 1521
+    },
+    {
+      "epoch": 47.56,
+      "learning_rate": 2.925e-05,
+      "loss": 0.1997,
+      "step": 1522
+    },
+    {
+      "epoch": 47.59,
+      "learning_rate": 2.8874999999999997e-05,
+      "loss": 0.1815,
+      "step": 1523
+    },
+    {
+      "epoch": 47.62,
+      "learning_rate": 2.8499999999999998e-05,
+      "loss": 0.2201,
+      "step": 1524
+    },
+    {
+      "epoch": 47.66,
+      "learning_rate": 2.8125e-05,
+      "loss": 0.1775,
+      "step": 1525
+    },
+    {
+      "epoch": 47.69,
+      "learning_rate": 2.7749999999999997e-05,
+      "loss": 0.175,
+      "step": 1526
+    },
+    {
+      "epoch": 47.72,
+      "learning_rate": 2.7374999999999998e-05,
+      "loss": 0.2122,
+      "step": 1527
+    },
+    {
+      "epoch": 47.75,
+      "learning_rate": 2.6999999999999996e-05,
+      "loss": 0.22,
+      "step": 1528
+    },
+    {
+      "epoch": 47.78,
+      "learning_rate": 2.6624999999999997e-05,
+      "loss": 0.2389,
+      "step": 1529
+    },
+    {
+      "epoch": 47.81,
+      "learning_rate": 2.6249999999999998e-05,
+      "loss": 0.2338,
+      "step": 1530
+    },
+    {
+      "epoch": 47.84,
+      "learning_rate": 2.5874999999999995e-05,
+      "loss": 0.2138,
+      "step": 1531
+    },
+    {
+      "epoch": 47.88,
+      "learning_rate": 2.55e-05,
+      "loss": 0.2325,
+      "step": 1532
+    },
+    {
+      "epoch": 47.91,
+      "learning_rate": 2.5125e-05,
+      "loss": 0.207,
+      "step": 1533
+    },
+    {
+      "epoch": 47.94,
+      "learning_rate": 2.475e-05,
+      "loss": 0.2198,
+      "step": 1534
+    },
+    {
+      "epoch": 47.97,
+      "learning_rate": 2.4375e-05,
+      "loss": 0.258,
+      "step": 1535
+    },
+    {
+      "epoch": 48.0,
+      "learning_rate": 2.3999999999999997e-05,
+      "loss": 0.2143,
+      "step": 1536
+    },
+    {
+      "epoch": 48.03,
+      "learning_rate": 2.3624999999999998e-05,
+      "loss": 0.2141,
+      "step": 1537
+    },
+    {
+      "epoch": 48.06,
+      "learning_rate": 2.325e-05,
+      "loss": 0.2572,
+      "step": 1538
+    },
+    {
+      "epoch": 48.09,
+      "learning_rate": 2.2874999999999997e-05,
+      "loss": 0.1879,
+      "step": 1539
+    },
+    {
+      "epoch": 48.12,
+      "learning_rate": 2.2499999999999998e-05,
+      "loss": 0.2162,
+      "step": 1540
+    },
+    {
+      "epoch": 48.12,
+      "eval_loss": 1.238689661026001,
+      "eval_mean_accuracy": 0.5359523890899606,
+      "eval_mean_iou": 0.41398860077625316,
+      "eval_overall_accuracy": 0.7056294491416529,
+      "eval_per_category_accuracy": [
+        0.6929335388516379,
+        0.8110093549911692,
+        0.7534561166566175,
+        0.39843928910972165,
+        0.5536307224168214,
+        0.5392086056783251,
+        0.6108369847075253,
+        0.48347014017835427,
+        0.3579252548734284,
+        0.6028045840089582,
+        0.2204044528378118,
+        0.40730962476915616
+      ],
+      "eval_per_category_iou": [
+        0.6014395714550632,
+        0.590602100980571,
+        0.5963685863817585,
+        0.29323170031026163,
+        0.40124627790470263,
+        0.3876715778047695,
+        0.4818528555531268,
+        0.4285345256591918,
+        0.24055739508589472,
+        0.4356924352800369,
+        0.18182437960185438,
+        0.3288418032978053
+      ],
+      "eval_runtime": 121.637,
+      "eval_samples_per_second": 0.937,
+      "eval_steps_per_second": 0.082,
+      "step": 1540
+    },
+    {
+      "epoch": 48.16,
+      "learning_rate": 2.2124999999999996e-05,
+      "loss": 0.1866,
+      "step": 1541
+    },
+    {
+      "epoch": 48.19,
+      "learning_rate": 2.1749999999999997e-05,
+      "loss": 0.1975,
+      "step": 1542
+    },
+    {
+      "epoch": 48.22,
+      "learning_rate": 2.1374999999999998e-05,
+      "loss": 0.2603,
+      "step": 1543
+    },
+    {
+      "epoch": 48.25,
+      "learning_rate": 2.1e-05,
+      "loss": 0.1973,
+      "step": 1544
+    },
+    {
+      "epoch": 48.28,
+      "learning_rate": 2.0625e-05,
+      "loss": 0.1866,
+      "step": 1545
+    },
+    {
+      "epoch": 48.31,
+      "learning_rate": 2.025e-05,
+      "loss": 0.2273,
+      "step": 1546
+    },
+    {
+      "epoch": 48.34,
+      "learning_rate": 1.9875e-05,
+      "loss": 0.2041,
+      "step": 1547
+    },
+    {
+      "epoch": 48.38,
+      "learning_rate": 1.95e-05,
+      "loss": 0.2279,
+      "step": 1548
+    },
+    {
+      "epoch": 48.41,
+      "learning_rate": 1.9124999999999997e-05,
+      "loss": 0.23,
+      "step": 1549
+    },
+    {
+      "epoch": 48.44,
+      "learning_rate": 1.875e-05,
+      "loss": 0.1929,
+      "step": 1550
+    },
+    {
+      "epoch": 48.47,
+      "learning_rate": 1.8375e-05,
+      "loss": 0.2324,
+      "step": 1551
+    },
+    {
+      "epoch": 48.5,
+      "learning_rate": 1.7999999999999997e-05,
+      "loss": 0.1811,
+      "step": 1552
+    },
+    {
+      "epoch": 48.53,
+      "learning_rate": 1.7624999999999998e-05,
+      "loss": 0.1918,
+      "step": 1553
+    },
+    {
+      "epoch": 48.56,
+      "learning_rate": 1.725e-05,
+      "loss": 0.2154,
+      "step": 1554
+    },
+    {
+      "epoch": 48.59,
+      "learning_rate": 1.6875e-05,
+      "loss": 0.1683,
+      "step": 1555
+    },
+    {
+      "epoch": 48.62,
+      "learning_rate": 1.6499999999999998e-05,
+      "loss": 0.2156,
+      "step": 1556
+    },
+    {
+      "epoch": 48.66,
+      "learning_rate": 1.6125e-05,
+      "loss": 0.2146,
+      "step": 1557
+    },
+    {
+      "epoch": 48.69,
+      "learning_rate": 1.5749999999999997e-05,
+      "loss": 0.1849,
+      "step": 1558
+    },
+    {
+      "epoch": 48.72,
+      "learning_rate": 1.5374999999999998e-05,
+      "loss": 0.2477,
+      "step": 1559
+    },
+    {
+      "epoch": 48.75,
+      "learning_rate": 1.4999999999999999e-05,
+      "loss": 0.1964,
+      "step": 1560
+    },
+    {
+      "epoch": 48.75,
+      "eval_loss": 1.2338758707046509,
+      "eval_mean_accuracy": 0.5383149903358363,
+      "eval_mean_iou": 0.41519941353660733,
+      "eval_overall_accuracy": 0.7056235263222143,
+      "eval_per_category_accuracy": [
+        0.6928750962357523,
+        0.8105594481012798,
+        0.7520747257977668,
+        0.4031408525964428,
+        0.5630126801281181,
+        0.5324237222316986,
+        0.610259954750775,
+        0.4977269098939153,
+        0.35636141007781513,
+        0.5972552899532222,
+        0.23397599728201815,
+        0.4101137969812333
+      ],
+      "eval_per_category_iou": [
+        0.6005822858729871,
+        0.590471088009995,
+        0.5970726765397967,
+        0.2950545738969673,
+        0.40192845593361043,
+        0.38627175030641653,
+        0.48217857747081594,
+        0.43427583144413023,
+        0.23918969822124883,
+        0.43495438720699603,
+        0.18943056591523538,
+        0.3309830716210894
+      ],
+      "eval_runtime": 121.5564,
+      "eval_samples_per_second": 0.938,
+      "eval_steps_per_second": 0.082,
+      "step": 1560
+    },
+    {
+      "epoch": 48.78,
+      "learning_rate": 1.4625e-05,
+      "loss": 0.2095,
+      "step": 1561
+    },
+    {
+      "epoch": 48.81,
+      "learning_rate": 1.4249999999999999e-05,
+      "loss": 0.1985,
+      "step": 1562
+    },
+    {
+      "epoch": 48.84,
+      "learning_rate": 1.3874999999999998e-05,
+      "loss": 0.2149,
+      "step": 1563
+    },
+    {
+      "epoch": 48.88,
+      "learning_rate": 1.3499999999999998e-05,
+      "loss": 0.2221,
+      "step": 1564
+    },
+    {
+      "epoch": 48.91,
+      "learning_rate": 1.3124999999999999e-05,
+      "loss": 0.2061,
+      "step": 1565
+    },
+    {
+      "epoch": 48.94,
+      "learning_rate": 1.275e-05,
+      "loss": 0.2173,
+      "step": 1566
+    },
+    {
+      "epoch": 48.97,
+      "learning_rate": 1.2375e-05,
+      "loss": 0.2114,
+      "step": 1567
+    },
+    {
+      "epoch": 49.0,
+      "learning_rate": 1.1999999999999999e-05,
+      "loss": 0.2642,
+      "step": 1568
+    },
+    {
+      "epoch": 49.03,
+      "learning_rate": 1.1625e-05,
+      "loss": 0.2254,
+      "step": 1569
+    },
+    {
+      "epoch": 49.06,
+      "learning_rate": 1.1249999999999999e-05,
+      "loss": 0.2391,
+      "step": 1570
+    },
+    {
+      "epoch": 49.09,
+      "learning_rate": 1.0874999999999998e-05,
+      "loss": 0.1892,
+      "step": 1571
+    },
+    {
+      "epoch": 49.12,
+      "learning_rate": 1.05e-05,
+      "loss": 0.1954,
+      "step": 1572
+    },
+    {
+      "epoch": 49.16,
+      "learning_rate": 1.0125e-05,
+      "loss": 0.1884,
+      "step": 1573
+    },
+    {
+      "epoch": 49.19,
+      "learning_rate": 9.75e-06,
+      "loss": 0.1996,
+      "step": 1574
+    },
+    {
+      "epoch": 49.22,
+      "learning_rate": 9.375e-06,
+      "loss": 0.2168,
+      "step": 1575
+    },
+    {
+      "epoch": 49.25,
+      "learning_rate": 8.999999999999999e-06,
+      "loss": 0.2033,
+      "step": 1576
+    },
+    {
+      "epoch": 49.28,
+      "learning_rate": 8.625e-06,
+      "loss": 0.1984,
+      "step": 1577
+    },
+    {
+      "epoch": 49.31,
+      "learning_rate": 8.249999999999999e-06,
+      "loss": 0.1963,
+      "step": 1578
+    },
+    {
+      "epoch": 49.34,
+      "learning_rate": 7.874999999999998e-06,
+      "loss": 0.1986,
+      "step": 1579
+    },
+    {
+      "epoch": 49.38,
+      "learning_rate": 7.499999999999999e-06,
+      "loss": 0.2199,
+      "step": 1580
+    },
+    {
+      "epoch": 49.38,
+      "eval_loss": 1.2340189218521118,
+      "eval_mean_accuracy": 0.5405919375104974,
+      "eval_mean_iou": 0.4152986757418806,
+      "eval_overall_accuracy": 0.7049936662640488,
+      "eval_per_category_accuracy": [
+        0.6868409695659343,
+        0.8073299816819581,
+        0.7543455600168815,
+        0.40291184655207113,
+        0.5607922902227885,
+        0.5419142117812723,
+        0.6187343442247409,
+        0.49715261043746406,
+        0.35784096252983505,
+        0.6150245050358278,
+        0.22603630709521813,
+        0.4181796609819761
+      ],
+      "eval_per_category_iou": [
+        0.5980365545449745,
+        0.5901125587428404,
+        0.5967057419281152,
+        0.2953850536994358,
+        0.401904031802587,
+        0.38961172813010847,
+        0.48294437559179315,
+        0.4336928450653673,
+        0.23825833436719068,
+        0.4381406450404767,
+        0.18425858928615985,
+        0.3345336507035186
+      ],
+      "eval_runtime": 120.7449,
+      "eval_samples_per_second": 0.944,
+      "eval_steps_per_second": 0.083,
+      "step": 1580
+    },
+    {
+      "epoch": 49.41,
+      "learning_rate": 7.1249999999999995e-06,
+      "loss": 0.238,
+      "step": 1581
+    },
+    {
+      "epoch": 49.44,
+      "learning_rate": 6.749999999999999e-06,
+      "loss": 0.1921,
+      "step": 1582
+    },
+    {
+      "epoch": 49.47,
+      "learning_rate": 6.375e-06,
+      "loss": 0.2344,
+      "step": 1583
+    },
+    {
+      "epoch": 49.5,
+      "learning_rate": 5.999999999999999e-06,
+      "loss": 0.2126,
+      "step": 1584
+    },
+    {
+      "epoch": 49.53,
+      "learning_rate": 5.6249999999999995e-06,
+      "loss": 0.1864,
+      "step": 1585
+    },
+    {
+      "epoch": 49.56,
+      "learning_rate": 5.25e-06,
+      "loss": 0.1963,
+      "step": 1586
+    },
+    {
+      "epoch": 49.59,
+      "learning_rate": 4.875e-06,
+      "loss": 0.1962,
+      "step": 1587
+    },
+    {
+      "epoch": 49.62,
+      "learning_rate": 4.499999999999999e-06,
+      "loss": 0.2059,
+      "step": 1588
+    },
+    {
+      "epoch": 49.66,
+      "learning_rate": 4.1249999999999995e-06,
+      "loss": 0.1905,
+      "step": 1589
+    },
+    {
+      "epoch": 49.69,
+      "learning_rate": 3.7499999999999997e-06,
+      "loss": 0.203,
+      "step": 1590
+    },
+    {
+      "epoch": 49.72,
+      "learning_rate": 3.3749999999999995e-06,
+      "loss": 0.1977,
+      "step": 1591
+    },
+    {
+      "epoch": 49.75,
+      "learning_rate": 2.9999999999999997e-06,
+      "loss": 0.2377,
+      "step": 1592
+    },
+    {
+      "epoch": 49.78,
+      "learning_rate": 2.625e-06,
+      "loss": 0.212,
+      "step": 1593
+    },
+    {
+      "epoch": 49.81,
+      "learning_rate": 2.2499999999999996e-06,
+      "loss": 0.1973,
+      "step": 1594
+    },
+    {
+      "epoch": 49.84,
+      "learning_rate": 1.8749999999999998e-06,
+      "loss": 0.2026,
+      "step": 1595
+    },
+    {
+      "epoch": 49.88,
+      "learning_rate": 1.4999999999999998e-06,
+      "loss": 0.2187,
+      "step": 1596
+    },
+    {
+      "epoch": 49.91,
+      "learning_rate": 1.1249999999999998e-06,
+      "loss": 0.1904,
+      "step": 1597
+    },
+    {
+      "epoch": 49.94,
+      "learning_rate": 7.499999999999999e-07,
+      "loss": 0.2054,
+      "step": 1598
+    },
+    {
+      "epoch": 49.97,
+      "learning_rate": 3.7499999999999996e-07,
+      "loss": 0.2161,
+      "step": 1599
+    },
+    {
+      "epoch": 50.0,
+      "learning_rate": 0.0,
+      "loss": 0.2031,
+      "step": 1600
+    },
+    {
+      "epoch": 50.0,
+      "eval_loss": 1.229608178138733,
+      "eval_mean_accuracy": 0.5404716490223255,
+      "eval_mean_iou": 0.4154950472956,
+      "eval_overall_accuracy": 0.7054718753747773,
+      "eval_per_category_accuracy": [
+        0.6946913693912526,
+        0.8044984383016367,
+        0.7569579610317086,
+        0.3999798752264037,
+        0.5655491109323815,
+        0.5352581667205005,
+        0.623003421244464,
+        0.48813096599097194,
+        0.35956230091479374,
+        0.6079757541746181,
+        0.22768301230743804,
+        0.4223694120317352
+      ],
+      "eval_per_category_iou": [
+        0.601148619354301,
+        0.5898795376412639,
+        0.5965897373214338,
+        0.2955056986264849,
+        0.40248939921941346,
+        0.38841403757346077,
+        0.4836819971164877,
+        0.42991142792973874,
+        0.23916953526494633,
+        0.4374508636182315,
+        0.18560603225218886,
+        0.3360936816292486
+      ],
+      "eval_runtime": 121.227,
+      "eval_samples_per_second": 0.94,
+      "eval_steps_per_second": 0.082,
+      "step": 1600
+    }
+  ],
+  "max_steps": 1600,
+  "num_train_epochs": 50,
+  "total_flos": 3.31530475143168e+17,
+  "trial_name": null,
+  "trial_params": null
+}