diff --git "a/trainer_state.json" "b/trainer_state.json"
--- "a/trainer_state.json"
+++ "b/trainer_state.json"
@@ -1,8 +1,8 @@
 {
-  "best_metric": 0.9547889828681946,
-  "best_model_checkpoint": "segformer-finetuned-segments-cmp-facade-outputs-baseline-0.0006/checkpoint-200",
-  "epoch": 50.0,
-  "global_step": 1600,
+  "best_metric": 0.9291033148765564,
+  "best_model_checkpoint": "segformer-finetuned-segments-cmp-facade-outputs-transforms/checkpoint-360",
+  "epoch": 11.25,
+  "global_step": 360,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -10,508 +10,508 @@
     {
       "epoch": 0.03,
       "learning_rate": 0.000599625,
-      "loss": 2.5686,
+      "loss": 1.4974,
       "step": 1
     },
     {
       "epoch": 0.06,
       "learning_rate": 0.00059925,
-      "loss": 2.3903,
+      "loss": 1.5572,
       "step": 2
     },
     {
       "epoch": 0.09,
       "learning_rate": 0.000598875,
-      "loss": 2.1973,
+      "loss": 1.5276,
       "step": 3
     },
     {
       "epoch": 0.12,
       "learning_rate": 0.0005985,
-      "loss": 1.9997,
+      "loss": 1.5646,
       "step": 4
     },
     {
       "epoch": 0.16,
       "learning_rate": 0.0005981249999999999,
-      "loss": 1.9846,
+      "loss": 1.3389,
       "step": 5
     },
     {
       "epoch": 0.19,
       "learning_rate": 0.00059775,
-      "loss": 1.8746,
+      "loss": 1.5391,
       "step": 6
     },
     {
       "epoch": 0.22,
       "learning_rate": 0.0005973749999999999,
-      "loss": 1.8332,
+      "loss": 1.3854,
       "step": 7
     },
     {
       "epoch": 0.25,
       "learning_rate": 0.000597,
-      "loss": 1.8059,
+      "loss": 1.4288,
       "step": 8
     },
     {
       "epoch": 0.28,
       "learning_rate": 0.0005966249999999999,
-      "loss": 1.6617,
+      "loss": 1.3674,
       "step": 9
     },
     {
       "epoch": 0.31,
       "learning_rate": 0.00059625,
-      "loss": 1.6067,
+      "loss": 1.2815,
       "step": 10
     },
     {
       "epoch": 0.34,
       "learning_rate": 0.0005958749999999999,
-      "loss": 1.5621,
+      "loss": 1.3141,
       "step": 11
     },
     {
       "epoch": 0.38,
       "learning_rate": 0.0005955,
-      "loss": 1.4863,
+      "loss": 1.2826,
       "step": 12
     },
     {
       "epoch": 0.41,
       "learning_rate": 0.000595125,
-      "loss": 1.4005,
+      "loss": 1.2023,
       "step": 13
     },
     {
       "epoch": 0.44,
       "learning_rate": 0.0005947499999999999,
-      "loss": 1.4485,
+      "loss": 1.2487,
       "step": 14
     },
     {
       "epoch": 0.47,
       "learning_rate": 0.000594375,
-      "loss": 1.3887,
+      "loss": 1.2041,
       "step": 15
     },
     {
       "epoch": 0.5,
       "learning_rate": 0.0005939999999999999,
-      "loss": 1.5758,
+      "loss": 1.3763,
       "step": 16
     },
     {
       "epoch": 0.53,
       "learning_rate": 0.000593625,
-      "loss": 1.3851,
+      "loss": 1.2913,
       "step": 17
     },
     {
       "epoch": 0.56,
       "learning_rate": 0.0005932499999999999,
-      "loss": 1.3992,
+      "loss": 1.2048,
       "step": 18
     },
     {
       "epoch": 0.59,
       "learning_rate": 0.000592875,
-      "loss": 1.3758,
+      "loss": 1.2608,
       "step": 19
     },
     {
       "epoch": 0.62,
       "learning_rate": 0.0005924999999999999,
-      "loss": 1.3587,
+      "loss": 1.2328,
       "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,
+      "eval_loss": 1.2321703433990479,
+      "eval_mean_accuracy": 0.27640981040002394,
+      "eval_mean_iou": 0.19381676955791272,
+      "eval_overall_accuracy": 0.5843182279352557,
+      "eval_per_category_accuracy": [
+        0.6484014402727478,
+        0.7268698138428387,
+        0.7396621827345261,
+        0.0,
+        0.314071721025026,
+        0.16580774409430243,
+        0.18421110637031626,
+        0.0,
         0.0,
-        0.39402196484247587,
-        0.003442343915017475,
-        0.2774663897762572,
-        6.214434416823545e-05,
-        0.0003349511548051072,
-        0.7398100827126451,
+        0.48692030347962645,
         0.0,
-        0.37528346074881097
+        0.05097341298090315
       ],
       "eval_per_category_iou": [
-        0.4772592315619872,
-        0.44317885541602603,
-        0.45610220048891476,
+        0.46588897486611575,
+        0.4794379279754499,
+        0.47516306513696516,
         0.0,
-        0.16838871775158482,
-        0.00343126703650311,
-        0.1752659085145423,
-        6.214374491064158e-05,
-        0.0003344066828185833,
-        0.3009429801158851,
+        0.21479697731322273,
+        0.1468532929608522,
+        0.15399766900533918,
         0.0,
-        0.172948092417501
+        0.0,
+        0.34140124873016997,
+        0.0,
+        0.04826207870683769
       ],
-      "eval_runtime": 121.8515,
-      "eval_samples_per_second": 0.936,
+      "eval_runtime": 121.98,
+      "eval_samples_per_second": 0.935,
       "eval_steps_per_second": 0.082,
       "step": 20
     },
     {
       "epoch": 0.66,
       "learning_rate": 0.0005921249999999999,
-      "loss": 1.3857,
+      "loss": 1.2252,
       "step": 21
     },
     {
       "epoch": 0.69,
       "learning_rate": 0.0005917499999999999,
-      "loss": 1.2941,
+      "loss": 1.1954,
       "step": 22
     },
     {
       "epoch": 0.72,
       "learning_rate": 0.0005913749999999999,
-      "loss": 1.2736,
+      "loss": 1.2127,
       "step": 23
     },
     {
       "epoch": 0.75,
       "learning_rate": 0.0005909999999999999,
-      "loss": 1.2626,
+      "loss": 1.1908,
       "step": 24
     },
     {
       "epoch": 0.78,
       "learning_rate": 0.0005906249999999999,
-      "loss": 1.2018,
+      "loss": 1.1458,
       "step": 25
     },
     {
       "epoch": 0.81,
       "learning_rate": 0.00059025,
-      "loss": 1.2629,
+      "loss": 1.0854,
       "step": 26
     },
     {
       "epoch": 0.84,
       "learning_rate": 0.000589875,
-      "loss": 1.1944,
+      "loss": 1.1372,
       "step": 27
     },
     {
       "epoch": 0.88,
       "learning_rate": 0.0005895,
-      "loss": 1.1597,
+      "loss": 1.0638,
       "step": 28
     },
     {
       "epoch": 0.91,
       "learning_rate": 0.000589125,
-      "loss": 1.1822,
+      "loss": 1.1055,
       "step": 29
     },
     {
       "epoch": 0.94,
       "learning_rate": 0.00058875,
-      "loss": 1.1729,
+      "loss": 1.1073,
       "step": 30
     },
     {
       "epoch": 0.97,
       "learning_rate": 0.0005883749999999999,
-      "loss": 1.0742,
+      "loss": 1.0367,
       "step": 31
     },
     {
       "epoch": 1.0,
       "learning_rate": 0.000588,
-      "loss": 1.3135,
+      "loss": 1.2498,
       "step": 32
     },
     {
       "epoch": 1.03,
       "learning_rate": 0.0005876249999999999,
-      "loss": 1.2218,
+      "loss": 1.2009,
       "step": 33
     },
     {
       "epoch": 1.06,
       "learning_rate": 0.00058725,
-      "loss": 1.2487,
+      "loss": 1.2267,
       "step": 34
     },
     {
       "epoch": 1.09,
       "learning_rate": 0.0005868749999999999,
-      "loss": 0.9779,
+      "loss": 0.9791,
       "step": 35
     },
     {
       "epoch": 1.12,
       "learning_rate": 0.0005865,
-      "loss": 1.0848,
+      "loss": 1.0073,
       "step": 36
     },
     {
       "epoch": 1.16,
       "learning_rate": 0.000586125,
-      "loss": 1.0201,
+      "loss": 1.0052,
       "step": 37
     },
     {
       "epoch": 1.19,
       "learning_rate": 0.0005857499999999999,
-      "loss": 1.0673,
+      "loss": 1.0387,
       "step": 38
     },
     {
       "epoch": 1.22,
       "learning_rate": 0.000585375,
-      "loss": 1.1126,
+      "loss": 1.076,
       "step": 39
     },
     {
       "epoch": 1.25,
       "learning_rate": 0.0005849999999999999,
-      "loss": 1.1272,
+      "loss": 1.1158,
       "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,
+      "eval_loss": 1.1340055465698242,
+      "eval_mean_accuracy": 0.3584914050139854,
+      "eval_mean_iou": 0.25033676612937344,
+      "eval_overall_accuracy": 0.6050458539996231,
+      "eval_per_category_accuracy": [
+        0.8000948297521079,
+        0.6337371473364622,
+        0.6603131057495936,
+        0.003965968313890952,
+        0.34629573729708046,
+        0.3680114248799625,
+        0.4865078902745655,
+        0.04561502007369462,
+        0.0118031463226356,
+        0.5511836320859765,
         0.0,
-        0.41723881549491804,
-        0.32015068647269534,
-        0.43180497804452184,
-        0.00046179676442257726,
-        0.005066413493873278,
-        0.5026867658203124,
+        0.3943689580818554
+      ],
+      "eval_per_category_iou": [
+        0.5176739126303989,
+        0.47108775141725884,
+        0.49329266238923963,
+        0.003963836494345551,
+        0.26672845921223864,
+        0.2701529378697397,
+        0.31332750965950773,
+        0.04532147556410988,
+        0.011623004855842847,
+        0.3621998398780441,
         0.0,
-        0.23512334024575196
+        0.24866980358175594
       ],
-      "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,
+      "eval_runtime": 124.8448,
+      "eval_samples_per_second": 0.913,
+      "eval_steps_per_second": 0.08,
       "step": 40
     },
     {
       "epoch": 1.28,
       "learning_rate": 0.000584625,
-      "loss": 1.0215,
+      "loss": 1.0057,
       "step": 41
     },
     {
       "epoch": 1.31,
       "learning_rate": 0.0005842499999999999,
-      "loss": 1.0141,
+      "loss": 0.9904,
       "step": 42
     },
     {
       "epoch": 1.34,
       "learning_rate": 0.000583875,
-      "loss": 1.1439,
+      "loss": 1.1023,
       "step": 43
     },
     {
       "epoch": 1.38,
       "learning_rate": 0.0005834999999999999,
-      "loss": 1.1585,
+      "loss": 1.1263,
       "step": 44
     },
     {
       "epoch": 1.41,
       "learning_rate": 0.000583125,
-      "loss": 0.9521,
+      "loss": 0.9288,
       "step": 45
     },
     {
       "epoch": 1.44,
       "learning_rate": 0.0005827499999999999,
-      "loss": 1.046,
+      "loss": 0.9942,
       "step": 46
     },
     {
       "epoch": 1.47,
       "learning_rate": 0.0005823749999999999,
-      "loss": 0.959,
+      "loss": 0.9388,
       "step": 47
     },
     {
       "epoch": 1.5,
       "learning_rate": 0.0005819999999999999,
-      "loss": 1.0224,
+      "loss": 1.0192,
       "step": 48
     },
     {
       "epoch": 1.53,
       "learning_rate": 0.0005816249999999999,
-      "loss": 1.0171,
+      "loss": 1.0228,
       "step": 49
     },
     {
       "epoch": 1.56,
       "learning_rate": 0.00058125,
-      "loss": 0.8845,
+      "loss": 0.9034,
       "step": 50
     },
     {
       "epoch": 1.59,
       "learning_rate": 0.000580875,
-      "loss": 1.0473,
+      "loss": 1.0346,
       "step": 51
     },
     {
       "epoch": 1.62,
       "learning_rate": 0.0005805,
-      "loss": 0.8508,
+      "loss": 0.8394,
       "step": 52
     },
     {
       "epoch": 1.66,
       "learning_rate": 0.000580125,
-      "loss": 0.9726,
+      "loss": 0.9892,
       "step": 53
     },
     {
       "epoch": 1.69,
       "learning_rate": 0.00057975,
-      "loss": 1.0182,
+      "loss": 1.0063,
       "step": 54
     },
     {
       "epoch": 1.72,
       "learning_rate": 0.0005793749999999999,
-      "loss": 1.0579,
+      "loss": 1.0106,
       "step": 55
     },
     {
       "epoch": 1.75,
       "learning_rate": 0.000579,
-      "loss": 0.9741,
+      "loss": 0.9481,
       "step": 56
     },
     {
       "epoch": 1.78,
       "learning_rate": 0.0005786249999999999,
-      "loss": 1.0505,
+      "loss": 1.0124,
       "step": 57
     },
     {
       "epoch": 1.81,
       "learning_rate": 0.00057825,
-      "loss": 0.9772,
+      "loss": 0.9676,
       "step": 58
     },
     {
       "epoch": 1.84,
       "learning_rate": 0.0005778749999999999,
-      "loss": 0.8138,
+      "loss": 0.7901,
       "step": 59
     },
     {
       "epoch": 1.88,
       "learning_rate": 0.0005775,
-      "loss": 1.0172,
+      "loss": 0.9771,
       "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,
+      "eval_loss": 1.0994304418563843,
+      "eval_mean_accuracy": 0.3909737030282887,
+      "eval_mean_iou": 0.2835007332832486,
+      "eval_overall_accuracy": 0.6179555591784025,
+      "eval_per_category_accuracy": [
+        0.7168020611741474,
+        0.7219130768570321,
+        0.5340207091634352,
+        0.0431815185182615,
+        0.45474233550172505,
+        0.5430138746239189,
+        0.469268628306114,
+        0.217241626853214,
+        0.09605778018331367,
+        0.5683827928074056,
+        0.0005080911993094857,
+        0.3265519411515883
+      ],
+      "eval_per_category_iou": [
+        0.5113476068813879,
+        0.49334140108090285,
+        0.4746839761440892,
+        0.040214174001835405,
+        0.31186037529280675,
+        0.31181019822943934,
+        0.35578648191496853,
+        0.20097078621392386,
+        0.08022661461990363,
+        0.38374995642386217,
+        0.0005073985432159898,
+        0.23750983005264706
+      ],
+      "eval_runtime": 125.5081,
+      "eval_samples_per_second": 0.908,
+      "eval_steps_per_second": 0.08,
       "step": 60
     },
     {
       "epoch": 1.91,
       "learning_rate": 0.000577125,
-      "loss": 0.9949,
+      "loss": 0.9391,
       "step": 61
     },
     {
       "epoch": 1.94,
       "learning_rate": 0.00057675,
-      "loss": 1.0874,
+      "loss": 1.0473,
       "step": 62
     },
     {
       "epoch": 1.97,
       "learning_rate": 0.000576375,
-      "loss": 0.9226,
+      "loss": 0.918,
       "step": 63
     },
     {
       "epoch": 2.0,
       "learning_rate": 0.0005759999999999999,
-      "loss": 1.0359,
+      "loss": 1.0355,
       "step": 64
     },
     {
       "epoch": 2.03,
       "learning_rate": 0.000575625,
-      "loss": 1.0029,
+      "loss": 1.015,
       "step": 65
     },
     {
@@ -523,379 +523,379 @@
     {
       "epoch": 2.09,
       "learning_rate": 0.000574875,
-      "loss": 0.8542,
+      "loss": 0.8417,
       "step": 67
     },
     {
       "epoch": 2.12,
       "learning_rate": 0.0005744999999999999,
-      "loss": 0.8286,
+      "loss": 0.8185,
       "step": 68
     },
     {
       "epoch": 2.16,
       "learning_rate": 0.000574125,
-      "loss": 0.8975,
+      "loss": 0.9263,
       "step": 69
     },
     {
       "epoch": 2.19,
       "learning_rate": 0.0005737499999999999,
-      "loss": 0.8509,
+      "loss": 0.8448,
       "step": 70
     },
     {
       "epoch": 2.22,
       "learning_rate": 0.0005733749999999999,
-      "loss": 1.1385,
+      "loss": 1.1335,
       "step": 71
     },
     {
       "epoch": 2.25,
       "learning_rate": 0.0005729999999999999,
-      "loss": 0.9216,
+      "loss": 0.9229,
       "step": 72
     },
     {
       "epoch": 2.28,
       "learning_rate": 0.0005726249999999999,
-      "loss": 0.8162,
+      "loss": 0.8549,
       "step": 73
     },
     {
       "epoch": 2.31,
       "learning_rate": 0.00057225,
-      "loss": 0.8909,
+      "loss": 0.8818,
       "step": 74
     },
     {
       "epoch": 2.34,
       "learning_rate": 0.0005718749999999999,
-      "loss": 0.8444,
+      "loss": 0.831,
       "step": 75
     },
     {
       "epoch": 2.38,
       "learning_rate": 0.0005715,
-      "loss": 0.8338,
+      "loss": 0.8845,
       "step": 76
     },
     {
       "epoch": 2.41,
       "learning_rate": 0.000571125,
-      "loss": 0.9097,
+      "loss": 0.9197,
       "step": 77
     },
     {
       "epoch": 2.44,
       "learning_rate": 0.00057075,
-      "loss": 0.8778,
+      "loss": 0.8579,
       "step": 78
     },
     {
       "epoch": 2.47,
       "learning_rate": 0.000570375,
-      "loss": 0.826,
+      "loss": 0.8167,
       "step": 79
     },
     {
       "epoch": 2.5,
       "learning_rate": 0.00057,
-      "loss": 0.8466,
+      "loss": 0.8566,
       "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,
+      "eval_loss": 1.0861966609954834,
+      "eval_mean_accuracy": 0.3778088961525232,
+      "eval_mean_iou": 0.26404526643863496,
+      "eval_overall_accuracy": 0.6308857432582922,
+      "eval_per_category_accuracy": [
+        0.6129614268456303,
+        0.7713753471720866,
+        0.7994049832227069,
+        0.019437755463182074,
+        0.5688067632752372,
+        0.4619781591716599,
+        0.3827566445038881,
+        0.08782710157707345,
+        0.15580996069314926,
+        0.5327405741537456,
+        0.0009059939457566733,
+        0.1397020438061634
+      ],
+      "eval_per_category_iou": [
+        0.5101947204251175,
+        0.5344538212406682,
+        0.49436541874132667,
+        0.018884843581445522,
+        0.270377214768106,
+        0.28449876963719384,
+        0.32160160861987425,
+        0.086283029811171,
+        0.11932598089180631,
+        0.39498832385101396,
+        0.0009054175167701066,
+        0.13266404817912572
+      ],
+      "eval_runtime": 123.9,
+      "eval_samples_per_second": 0.92,
+      "eval_steps_per_second": 0.081,
       "step": 80
     },
     {
       "epoch": 2.53,
       "learning_rate": 0.0005696249999999999,
-      "loss": 0.8429,
+      "loss": 0.827,
       "step": 81
     },
     {
       "epoch": 2.56,
       "learning_rate": 0.00056925,
-      "loss": 0.9725,
+      "loss": 0.9862,
       "step": 82
     },
     {
       "epoch": 2.59,
       "learning_rate": 0.0005688749999999999,
-      "loss": 0.8059,
+      "loss": 0.8236,
       "step": 83
     },
     {
       "epoch": 2.62,
       "learning_rate": 0.0005685,
-      "loss": 0.8261,
+      "loss": 0.8522,
       "step": 84
     },
     {
       "epoch": 2.66,
       "learning_rate": 0.0005681249999999999,
-      "loss": 0.9101,
+      "loss": 0.9125,
       "step": 85
     },
     {
       "epoch": 2.69,
       "learning_rate": 0.00056775,
-      "loss": 0.8387,
+      "loss": 0.8568,
       "step": 86
     },
     {
       "epoch": 2.72,
       "learning_rate": 0.000567375,
-      "loss": 0.8772,
+      "loss": 0.8791,
       "step": 87
     },
     {
       "epoch": 2.75,
       "learning_rate": 0.0005669999999999999,
-      "loss": 0.8313,
+      "loss": 0.8188,
       "step": 88
     },
     {
       "epoch": 2.78,
       "learning_rate": 0.000566625,
-      "loss": 0.9802,
+      "loss": 0.992,
       "step": 89
     },
     {
       "epoch": 2.81,
       "learning_rate": 0.0005662499999999999,
-      "loss": 0.7859,
+      "loss": 0.8239,
       "step": 90
     },
     {
       "epoch": 2.84,
       "learning_rate": 0.000565875,
-      "loss": 0.8676,
+      "loss": 0.8892,
       "step": 91
     },
     {
       "epoch": 2.88,
       "learning_rate": 0.0005654999999999999,
-      "loss": 0.8232,
+      "loss": 0.8093,
       "step": 92
     },
     {
       "epoch": 2.91,
       "learning_rate": 0.000565125,
-      "loss": 1.058,
+      "loss": 1.0857,
       "step": 93
     },
     {
       "epoch": 2.94,
       "learning_rate": 0.0005647499999999999,
-      "loss": 0.8119,
+      "loss": 0.8162,
       "step": 94
     },
     {
       "epoch": 2.97,
       "learning_rate": 0.000564375,
-      "loss": 0.8982,
+      "loss": 0.9017,
       "step": 95
     },
     {
       "epoch": 3.0,
       "learning_rate": 0.0005639999999999999,
-      "loss": 0.8573,
+      "loss": 0.8848,
       "step": 96
     },
     {
       "epoch": 3.03,
       "learning_rate": 0.0005636249999999999,
-      "loss": 0.9197,
+      "loss": 0.9075,
       "step": 97
     },
     {
       "epoch": 3.06,
       "learning_rate": 0.00056325,
-      "loss": 0.7869,
+      "loss": 0.8108,
       "step": 98
     },
     {
       "epoch": 3.09,
       "learning_rate": 0.0005628749999999999,
-      "loss": 0.8251,
+      "loss": 0.8238,
       "step": 99
     },
     {
       "epoch": 3.12,
       "learning_rate": 0.0005625,
-      "loss": 0.783,
+      "loss": 0.7854,
       "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,
+      "eval_loss": 1.0004723072052002,
+      "eval_mean_accuracy": 0.4409761597550485,
+      "eval_mean_iou": 0.3278147775750712,
+      "eval_overall_accuracy": 0.6476280145477831,
+      "eval_per_category_accuracy": [
+        0.7394164167070112,
+        0.7120692919670155,
+        0.65725755249271,
+        0.17395090943157923,
+        0.46355298851110294,
+        0.31981316619425926,
+        0.6107480292025965,
+        0.35138126519456003,
+        0.1554151176100015,
+        0.5705962518273815,
+        0.05702864593695384,
+        0.4804842819854117
+      ],
+      "eval_per_category_iou": [
+        0.5451244129938286,
+        0.5178187584431643,
+        0.5318888189962618,
+        0.14778251064896017,
+        0.3169269253463421,
+        0.2626134690690619,
+        0.39125714035001896,
+        0.3081966042250305,
+        0.12383697911511432,
+        0.3977603328560864,
+        0.04942253722870921,
+        0.34114884162827647
+      ],
+      "eval_runtime": 124.3127,
+      "eval_samples_per_second": 0.917,
+      "eval_steps_per_second": 0.08,
       "step": 100
     },
     {
       "epoch": 3.16,
       "learning_rate": 0.0005621249999999999,
-      "loss": 0.8053,
+      "loss": 0.8351,
       "step": 101
     },
     {
       "epoch": 3.19,
       "learning_rate": 0.00056175,
-      "loss": 0.8289,
+      "loss": 0.8611,
       "step": 102
     },
     {
       "epoch": 3.22,
       "learning_rate": 0.000561375,
-      "loss": 0.8684,
+      "loss": 0.9022,
       "step": 103
     },
     {
       "epoch": 3.25,
       "learning_rate": 0.000561,
-      "loss": 0.824,
+      "loss": 0.8189,
       "step": 104
     },
     {
       "epoch": 3.28,
       "learning_rate": 0.0005606249999999999,
-      "loss": 0.7919,
+      "loss": 0.8431,
       "step": 105
     },
     {
       "epoch": 3.31,
       "learning_rate": 0.00056025,
-      "loss": 0.8915,
+      "loss": 0.9712,
       "step": 106
     },
     {
       "epoch": 3.34,
       "learning_rate": 0.0005598749999999999,
-      "loss": 0.7168,
+      "loss": 0.7789,
       "step": 107
     },
     {
       "epoch": 3.38,
       "learning_rate": 0.0005595,
-      "loss": 0.7627,
+      "loss": 0.7784,
       "step": 108
     },
     {
       "epoch": 3.41,
       "learning_rate": 0.0005591249999999999,
-      "loss": 0.713,
+      "loss": 0.7459,
       "step": 109
     },
     {
       "epoch": 3.44,
       "learning_rate": 0.00055875,
-      "loss": 0.7279,
+      "loss": 0.7602,
       "step": 110
     },
     {
       "epoch": 3.47,
       "learning_rate": 0.000558375,
-      "loss": 0.7056,
+      "loss": 0.7709,
       "step": 111
     },
     {
       "epoch": 3.5,
       "learning_rate": 0.000558,
-      "loss": 0.7856,
+      "loss": 0.792,
       "step": 112
     },
     {
       "epoch": 3.53,
       "learning_rate": 0.000557625,
-      "loss": 0.7288,
+      "loss": 0.7701,
       "step": 113
     },
     {
       "epoch": 3.56,
       "learning_rate": 0.0005572499999999999,
-      "loss": 0.8074,
+      "loss": 0.8632,
       "step": 114
     },
     {
       "epoch": 3.59,
       "learning_rate": 0.000556875,
-      "loss": 0.7689,
+      "loss": 0.776,
       "step": 115
     },
     {
       "epoch": 3.62,
       "learning_rate": 0.0005564999999999999,
-      "loss": 0.826,
+      "loss": 0.8366,
       "step": 116
     },
     {
@@ -907,11830 +907,1972 @@
     {
       "epoch": 3.69,
       "learning_rate": 0.0005557499999999999,
-      "loss": 0.776,
+      "loss": 0.8111,
       "step": 118
     },
     {
       "epoch": 3.72,
       "learning_rate": 0.000555375,
-      "loss": 0.7384,
+      "loss": 0.7547,
       "step": 119
     },
     {
       "epoch": 3.75,
       "learning_rate": 0.0005549999999999999,
-      "loss": 0.6941,
+      "loss": 0.7275,
       "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,
+      "eval_loss": 1.0055418014526367,
+      "eval_mean_accuracy": 0.46582571564323233,
+      "eval_mean_iou": 0.34071242663610873,
+      "eval_overall_accuracy": 0.6500220717045299,
+      "eval_per_category_accuracy": [
+        0.6790541517825218,
+        0.7248947508922547,
+        0.702760127940592,
+        0.22049812284439385,
+        0.49308863123902213,
+        0.336202860575091,
+        0.6458925386011786,
+        0.40734546148068546,
+        0.2508473598750699,
+        0.5798344703996796,
+        0.12197249573784942,
+        0.4275176163504493
+      ],
+      "eval_per_category_iou": [
+        0.5331521735997542,
+        0.5305308321791462,
+        0.5588320252288795,
+        0.16697584226263984,
+        0.3296551780182928,
+        0.28048869107094193,
+        0.35449032046775947,
+        0.34190939733724723,
+        0.16527723782325413,
+        0.39484470653333187,
+        0.0880505368099854,
+        0.3443421783020718
+      ],
+      "eval_runtime": 124.4108,
+      "eval_samples_per_second": 0.916,
+      "eval_steps_per_second": 0.08,
       "step": 120
     },
     {
       "epoch": 3.78,
       "learning_rate": 0.0005546249999999999,
-      "loss": 0.7105,
+      "loss": 0.7266,
       "step": 121
     },
     {
       "epoch": 3.81,
       "learning_rate": 0.00055425,
-      "loss": 0.8228,
+      "loss": 0.8377,
       "step": 122
     },
     {
       "epoch": 3.84,
       "learning_rate": 0.0005538749999999999,
-      "loss": 0.7552,
+      "loss": 0.7775,
       "step": 123
     },
     {
       "epoch": 3.88,
       "learning_rate": 0.0005535,
-      "loss": 0.877,
+      "loss": 0.8343,
       "step": 124
     },
     {
       "epoch": 3.91,
       "learning_rate": 0.0005531249999999999,
-      "loss": 0.8001,
+      "loss": 0.8143,
       "step": 125
     },
     {
       "epoch": 3.94,
       "learning_rate": 0.00055275,
-      "loss": 0.828,
+      "loss": 0.8377,
       "step": 126
     },
     {
       "epoch": 3.97,
       "learning_rate": 0.0005523749999999999,
-      "loss": 0.8079,
+      "loss": 0.8807,
       "step": 127
     },
     {
       "epoch": 4.0,
       "learning_rate": 0.000552,
-      "loss": 0.855,
+      "loss": 0.8941,
       "step": 128
     },
     {
       "epoch": 4.03,
       "learning_rate": 0.000551625,
-      "loss": 0.8755,
+      "loss": 0.9077,
       "step": 129
     },
     {
       "epoch": 4.06,
       "learning_rate": 0.0005512499999999999,
-      "loss": 0.8063,
+      "loss": 0.8361,
       "step": 130
     },
     {
       "epoch": 4.09,
       "learning_rate": 0.0005508749999999999,
-      "loss": 0.5529,
+      "loss": 0.6058,
       "step": 131
     },
     {
       "epoch": 4.12,
       "learning_rate": 0.0005505,
-      "loss": 0.8243,
+      "loss": 0.8466,
       "step": 132
     },
     {
       "epoch": 4.16,
       "learning_rate": 0.0005501249999999999,
-      "loss": 0.6746,
+      "loss": 0.6832,
       "step": 133
     },
     {
       "epoch": 4.19,
       "learning_rate": 0.00054975,
-      "loss": 0.9177,
+      "loss": 0.9442,
       "step": 134
     },
     {
       "epoch": 4.22,
       "learning_rate": 0.000549375,
-      "loss": 0.7125,
+      "loss": 0.7119,
       "step": 135
     },
     {
       "epoch": 4.25,
       "learning_rate": 0.000549,
-      "loss": 0.8089,
+      "loss": 0.8412,
       "step": 136
     },
     {
       "epoch": 4.28,
       "learning_rate": 0.000548625,
-      "loss": 0.8293,
+      "loss": 0.8589,
       "step": 137
     },
     {
       "epoch": 4.31,
       "learning_rate": 0.0005482499999999999,
-      "loss": 0.7055,
+      "loss": 0.7529,
       "step": 138
     },
     {
       "epoch": 4.34,
       "learning_rate": 0.000547875,
-      "loss": 0.6936,
+      "loss": 0.7525,
       "step": 139
     },
     {
       "epoch": 4.38,
       "learning_rate": 0.0005474999999999999,
-      "loss": 0.7125,
+      "loss": 0.762,
       "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,
+      "eval_loss": 1.009921669960022,
+      "eval_mean_accuracy": 0.4630782466624949,
+      "eval_mean_iou": 0.3357803602667826,
+      "eval_overall_accuracy": 0.6476533454761171,
+      "eval_per_category_accuracy": [
+        0.6388404926271997,
+        0.7725403351834625,
+        0.6020424381762673,
+        0.21688954275126474,
+        0.5520221552773765,
+        0.4518525508185774,
+        0.6697900807527053,
+        0.46810120099302377,
+        0.1897420654286044,
+        0.45603045018648025,
+        0.024152696709344286,
+        0.5149349510456322
+      ],
+      "eval_per_category_iou": [
+        0.5210483275809908,
+        0.5385459417489167,
+        0.519285461653176,
+        0.19235183880209375,
+        0.3408901260438495,
+        0.3345069902216156,
+        0.36090171027181106,
+        0.33539228285452394,
+        0.1422865859237117,
+        0.3681340241516971,
+        0.023384620855608634,
+        0.35263641309339605
+      ],
+      "eval_runtime": 123.6747,
+      "eval_samples_per_second": 0.922,
+      "eval_steps_per_second": 0.081,
       "step": 140
     },
     {
       "epoch": 4.41,
       "learning_rate": 0.000547125,
-      "loss": 0.7107,
+      "loss": 0.7219,
       "step": 141
     },
     {
       "epoch": 4.44,
       "learning_rate": 0.0005467499999999999,
-      "loss": 0.7305,
+      "loss": 0.7571,
       "step": 142
     },
     {
       "epoch": 4.47,
       "learning_rate": 0.000546375,
-      "loss": 0.7128,
+      "loss": 0.7736,
       "step": 143
     },
     {
       "epoch": 4.5,
       "learning_rate": 0.0005459999999999999,
-      "loss": 0.7267,
+      "loss": 0.7782,
       "step": 144
     },
     {
       "epoch": 4.53,
       "learning_rate": 0.000545625,
-      "loss": 0.7552,
+      "loss": 0.7814,
       "step": 145
     },
     {
       "epoch": 4.56,
       "learning_rate": 0.0005452499999999999,
-      "loss": 0.6716,
+      "loss": 0.6899,
       "step": 146
     },
     {
       "epoch": 4.59,
       "learning_rate": 0.0005448749999999999,
-      "loss": 0.8395,
+      "loss": 0.842,
       "step": 147
     },
     {
       "epoch": 4.62,
       "learning_rate": 0.0005445,
-      "loss": 0.6889,
+      "loss": 0.7052,
       "step": 148
     },
     {
       "epoch": 4.66,
       "learning_rate": 0.0005441249999999999,
-      "loss": 0.7029,
+      "loss": 0.7245,
       "step": 149
     },
     {
       "epoch": 4.69,
       "learning_rate": 0.00054375,
-      "loss": 0.817,
+      "loss": 0.8852,
       "step": 150
     },
     {
       "epoch": 4.72,
       "learning_rate": 0.0005433749999999999,
-      "loss": 0.6999,
+      "loss": 0.6868,
       "step": 151
     },
     {
       "epoch": 4.75,
       "learning_rate": 0.000543,
-      "loss": 0.7728,
+      "loss": 0.8091,
       "step": 152
     },
     {
       "epoch": 4.78,
       "learning_rate": 0.000542625,
-      "loss": 0.7724,
+      "loss": 0.7912,
       "step": 153
     },
     {
       "epoch": 4.81,
       "learning_rate": 0.00054225,
-      "loss": 0.8145,
+      "loss": 0.8428,
       "step": 154
     },
     {
       "epoch": 4.84,
       "learning_rate": 0.0005418749999999999,
-      "loss": 0.7165,
+      "loss": 0.7627,
       "step": 155
     },
     {
       "epoch": 4.88,
       "learning_rate": 0.0005414999999999999,
-      "loss": 0.6526,
+      "loss": 0.6705,
       "step": 156
     },
     {
       "epoch": 4.91,
       "learning_rate": 0.0005411249999999999,
-      "loss": 0.7235,
+      "loss": 0.7269,
       "step": 157
     },
     {
       "epoch": 4.94,
       "learning_rate": 0.00054075,
-      "loss": 0.7316,
+      "loss": 0.7688,
       "step": 158
     },
     {
       "epoch": 4.97,
       "learning_rate": 0.000540375,
-      "loss": 0.7012,
+      "loss": 0.757,
       "step": 159
     },
     {
       "epoch": 5.0,
       "learning_rate": 0.00054,
-      "loss": 0.7644,
+      "loss": 0.7543,
       "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,
+      "eval_loss": 0.9964900612831116,
+      "eval_mean_accuracy": 0.47220927209560704,
+      "eval_mean_iou": 0.35076273333971525,
+      "eval_overall_accuracy": 0.6553197827255517,
+      "eval_per_category_accuracy": [
+        0.5974941039770494,
+        0.7656300402304882,
+        0.7736861734503012,
+        0.26844019125474494,
+        0.47782547654307983,
+        0.5049230780397693,
+        0.5503440924885342,
+        0.38577494532904893,
+        0.29696414469889887,
+        0.6231179209340338,
+        0.08607248563724125,
+        0.33623861256409465
+      ],
+      "eval_per_category_iou": [
+        0.5029245432853789,
+        0.5438264868522664,
+        0.5504928848830769,
+        0.20921503460100002,
+        0.3484194245550177,
+        0.34286066341367244,
+        0.4044913460614946,
+        0.3442055933600121,
+        0.19153018129377833,
+        0.4114387452368694,
+        0.07501013614442405,
+        0.28473776038959164
+      ],
+      "eval_runtime": 123.8413,
+      "eval_samples_per_second": 0.921,
+      "eval_steps_per_second": 0.081,
       "step": 160
     },
     {
       "epoch": 5.03,
       "learning_rate": 0.000539625,
-      "loss": 0.6204,
+      "loss": 0.639,
       "step": 161
     },
     {
       "epoch": 5.06,
       "learning_rate": 0.00053925,
-      "loss": 0.6766,
+      "loss": 0.7108,
       "step": 162
     },
     {
       "epoch": 5.09,
       "learning_rate": 0.000538875,
-      "loss": 0.662,
+      "loss": 0.6735,
       "step": 163
     },
     {
       "epoch": 5.12,
       "learning_rate": 0.0005384999999999999,
-      "loss": 0.7268,
+      "loss": 0.7478,
       "step": 164
     },
     {
       "epoch": 5.16,
       "learning_rate": 0.000538125,
-      "loss": 0.5879,
+      "loss": 0.6023,
       "step": 165
     },
     {
       "epoch": 5.19,
       "learning_rate": 0.0005377499999999999,
-      "loss": 0.7585,
+      "loss": 0.7976,
       "step": 166
     },
     {
       "epoch": 5.22,
       "learning_rate": 0.000537375,
-      "loss": 0.6846,
+      "loss": 0.7348,
       "step": 167
     },
     {
       "epoch": 5.25,
       "learning_rate": 0.0005369999999999999,
-      "loss": 0.6442,
+      "loss": 0.7037,
       "step": 168
     },
     {
       "epoch": 5.28,
       "learning_rate": 0.000536625,
-      "loss": 0.6522,
+      "loss": 0.694,
       "step": 169
     },
     {
       "epoch": 5.31,
       "learning_rate": 0.0005362499999999999,
-      "loss": 0.6976,
+      "loss": 0.7427,
       "step": 170
     },
     {
       "epoch": 5.34,
       "learning_rate": 0.0005358749999999999,
-      "loss": 0.7117,
+      "loss": 0.7319,
       "step": 171
     },
     {
       "epoch": 5.38,
       "learning_rate": 0.0005355,
-      "loss": 0.7123,
+      "loss": 0.7365,
       "step": 172
     },
     {
       "epoch": 5.41,
       "learning_rate": 0.0005351249999999999,
-      "loss": 0.7688,
+      "loss": 0.823,
       "step": 173
     },
     {
       "epoch": 5.44,
       "learning_rate": 0.00053475,
-      "loss": 0.6922,
+      "loss": 0.6932,
       "step": 174
     },
     {
       "epoch": 5.47,
       "learning_rate": 0.0005343749999999999,
-      "loss": 0.6724,
+      "loss": 0.6911,
       "step": 175
     },
     {
       "epoch": 5.5,
       "learning_rate": 0.000534,
-      "loss": 0.7095,
+      "loss": 0.7337,
       "step": 176
     },
     {
       "epoch": 5.53,
       "learning_rate": 0.0005336249999999999,
-      "loss": 0.8182,
+      "loss": 0.8752,
       "step": 177
     },
     {
       "epoch": 5.56,
       "learning_rate": 0.00053325,
-      "loss": 0.6316,
+      "loss": 0.669,
       "step": 178
     },
     {
       "epoch": 5.59,
       "learning_rate": 0.000532875,
-      "loss": 0.6601,
+      "loss": 0.7305,
       "step": 179
     },
     {
       "epoch": 5.62,
       "learning_rate": 0.0005324999999999999,
-      "loss": 0.6728,
+      "loss": 0.7142,
       "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,
+      "eval_loss": 0.9814519286155701,
+      "eval_mean_accuracy": 0.497081755026592,
+      "eval_mean_iou": 0.35939949129333554,
+      "eval_overall_accuracy": 0.6597169909560889,
+      "eval_per_category_accuracy": [
+        0.6908915714734837,
+        0.7305051261897875,
+        0.7036370157962516,
+        0.2776663590120818,
+        0.5628060380018922,
+        0.4295562515651142,
+        0.5859168478577468,
+        0.2814967358147499,
+        0.3033792356902656,
+        0.6404991026728926,
+        0.30056349150477635,
+        0.4580632847400621
+      ],
+      "eval_per_category_iou": [
+        0.5553982956913027,
+        0.5378390398854487,
+        0.5530532106585494,
+        0.21904999137749334,
+        0.3437066198831567,
+        0.32814516520304693,
+        0.4234087095030862,
+        0.2666791179990438,
+        0.18338491161074402,
+        0.41252681863698343,
+        0.14896654252535288,
+        0.34063547254581833
+      ],
+      "eval_runtime": 124.8537,
+      "eval_samples_per_second": 0.913,
+      "eval_steps_per_second": 0.08,
       "step": 180
     },
     {
       "epoch": 5.66,
       "learning_rate": 0.0005321249999999999,
-      "loss": 0.6799,
+      "loss": 0.7298,
       "step": 181
     },
     {
       "epoch": 5.69,
       "learning_rate": 0.00053175,
-      "loss": 0.6823,
+      "loss": 0.6999,
       "step": 182
     },
     {
       "epoch": 5.72,
       "learning_rate": 0.000531375,
-      "loss": 0.7729,
+      "loss": 0.8023,
       "step": 183
     },
     {
       "epoch": 5.75,
       "learning_rate": 0.000531,
-      "loss": 0.638,
+      "loss": 0.6922,
       "step": 184
     },
     {
       "epoch": 5.78,
       "learning_rate": 0.000530625,
-      "loss": 0.688,
+      "loss": 0.7184,
       "step": 185
     },
     {
       "epoch": 5.81,
       "learning_rate": 0.00053025,
-      "loss": 0.6704,
+      "loss": 0.7189,
       "step": 186
     },
     {
       "epoch": 5.84,
       "learning_rate": 0.000529875,
-      "loss": 0.5638,
+      "loss": 0.614,
       "step": 187
     },
     {
       "epoch": 5.88,
       "learning_rate": 0.0005294999999999999,
-      "loss": 0.6177,
+      "loss": 0.6966,
       "step": 188
     },
     {
       "epoch": 5.91,
       "learning_rate": 0.000529125,
-      "loss": 0.672,
+      "loss": 0.7259,
       "step": 189
     },
     {
       "epoch": 5.94,
       "learning_rate": 0.0005287499999999999,
-      "loss": 0.7459,
+      "loss": 0.7856,
       "step": 190
     },
     {
       "epoch": 5.97,
       "learning_rate": 0.000528375,
-      "loss": 0.7019,
+      "loss": 0.7678,
       "step": 191
     },
     {
       "epoch": 6.0,
       "learning_rate": 0.0005279999999999999,
-      "loss": 0.5613,
+      "loss": 0.623,
       "step": 192
     },
     {
       "epoch": 6.03,
       "learning_rate": 0.000527625,
-      "loss": 0.6292,
+      "loss": 0.6785,
       "step": 193
     },
     {
       "epoch": 6.06,
       "learning_rate": 0.0005272499999999999,
-      "loss": 0.6222,
+      "loss": 0.6884,
       "step": 194
     },
     {
       "epoch": 6.09,
       "learning_rate": 0.000526875,
-      "loss": 0.5749,
+      "loss": 0.6096,
       "step": 195
     },
     {
       "epoch": 6.12,
       "learning_rate": 0.0005265,
-      "loss": 0.6755,
+      "loss": 0.739,
       "step": 196
     },
     {
       "epoch": 6.16,
       "learning_rate": 0.0005261249999999999,
-      "loss": 0.618,
+      "loss": 0.6649,
       "step": 197
     },
     {
       "epoch": 6.19,
       "learning_rate": 0.00052575,
-      "loss": 0.6044,
+      "loss": 0.6959,
       "step": 198
     },
     {
       "epoch": 6.22,
       "learning_rate": 0.0005253749999999999,
-      "loss": 0.6677,
+      "loss": 0.6718,
       "step": 199
     },
     {
       "epoch": 6.25,
       "learning_rate": 0.000525,
-      "loss": 0.704,
+      "loss": 0.7132,
       "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,
+      "eval_loss": 0.9577063918113708,
+      "eval_mean_accuracy": 0.49358942386599547,
+      "eval_mean_iou": 0.36445855775079217,
+      "eval_overall_accuracy": 0.6702703174791838,
+      "eval_per_category_accuracy": [
+        0.757485170186219,
+        0.7039742826269,
+        0.7879783326232815,
+        0.3002650918453029,
+        0.530024898350307,
+        0.5117678441164408,
+        0.5200409667652787,
+        0.282272468662643,
+        0.18195167830492534,
+        0.6273017965839506,
+        0.20833269566867557,
+        0.511677860658021
+      ],
+      "eval_per_category_iou": [
+        0.5717339461326115,
+        0.5411086678260705,
+        0.5550507730528956,
+        0.23120184241180683,
+        0.33923086199004965,
+        0.35084943122443496,
+        0.4183980452164044,
+        0.2680651542162202,
+        0.15420450059218319,
+        0.4336026908109024,
+        0.15032155761093297,
+        0.3597352219249941
+      ],
+      "eval_runtime": 123.8598,
+      "eval_samples_per_second": 0.92,
+      "eval_steps_per_second": 0.081,
       "step": 200
     },
     {
       "epoch": 6.28,
       "learning_rate": 0.0005246249999999999,
-      "loss": 0.6344,
+      "loss": 0.661,
       "step": 201
     },
     {
       "epoch": 6.31,
       "learning_rate": 0.00052425,
-      "loss": 0.6519,
+      "loss": 0.7055,
       "step": 202
     },
     {
       "epoch": 6.34,
       "learning_rate": 0.0005238749999999999,
-      "loss": 0.6823,
+      "loss": 0.6852,
       "step": 203
     },
     {
       "epoch": 6.38,
       "learning_rate": 0.0005235,
-      "loss": 0.6326,
+      "loss": 0.6802,
       "step": 204
     },
     {
       "epoch": 6.41,
       "learning_rate": 0.0005231249999999999,
-      "loss": 0.6317,
+      "loss": 0.6662,
       "step": 205
     },
     {
       "epoch": 6.44,
       "learning_rate": 0.0005227499999999999,
-      "loss": 0.6645,
+      "loss": 0.7165,
       "step": 206
     },
     {
       "epoch": 6.47,
       "learning_rate": 0.0005223749999999999,
-      "loss": 0.5607,
+      "loss": 0.6282,
       "step": 207
     },
     {
       "epoch": 6.5,
       "learning_rate": 0.000522,
-      "loss": 0.6568,
+      "loss": 0.693,
       "step": 208
     },
     {
       "epoch": 6.53,
       "learning_rate": 0.000521625,
-      "loss": 0.6128,
+      "loss": 0.6498,
       "step": 209
     },
     {
       "epoch": 6.56,
       "learning_rate": 0.00052125,
-      "loss": 0.6798,
+      "loss": 0.7024,
       "step": 210
     },
     {
       "epoch": 6.59,
       "learning_rate": 0.000520875,
-      "loss": 0.541,
+      "loss": 0.573,
       "step": 211
     },
     {
       "epoch": 6.62,
       "learning_rate": 0.0005205,
-      "loss": 0.7654,
+      "loss": 0.8226,
       "step": 212
     },
     {
       "epoch": 6.66,
       "learning_rate": 0.000520125,
-      "loss": 0.4971,
+      "loss": 0.5298,
       "step": 213
     },
     {
       "epoch": 6.69,
       "learning_rate": 0.0005197499999999999,
-      "loss": 0.5991,
+      "loss": 0.6492,
       "step": 214
     },
     {
       "epoch": 6.72,
       "learning_rate": 0.000519375,
-      "loss": 0.6807,
+      "loss": 0.769,
       "step": 215
     },
     {
       "epoch": 6.75,
       "learning_rate": 0.0005189999999999999,
-      "loss": 0.6202,
+      "loss": 0.664,
       "step": 216
     },
     {
       "epoch": 6.78,
       "learning_rate": 0.000518625,
-      "loss": 0.6998,
+      "loss": 0.7115,
       "step": 217
     },
     {
       "epoch": 6.81,
       "learning_rate": 0.0005182499999999999,
-      "loss": 0.6515,
+      "loss": 0.6526,
       "step": 218
     },
     {
       "epoch": 6.84,
       "learning_rate": 0.000517875,
-      "loss": 0.6392,
+      "loss": 0.679,
       "step": 219
     },
     {
       "epoch": 6.88,
       "learning_rate": 0.0005175,
-      "loss": 0.5244,
+      "loss": 0.5572,
       "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,
+      "eval_loss": 0.9686026573181152,
+      "eval_mean_accuracy": 0.49162090929743196,
+      "eval_mean_iou": 0.3674592922271513,
+      "eval_overall_accuracy": 0.668428052935684,
+      "eval_per_category_accuracy": [
+        0.8315143436240795,
+        0.67630998091554,
+        0.7426204308307264,
+        0.32498386548323743,
+        0.49111945333028095,
+        0.5001669455140652,
+        0.5374337360209966,
+        0.43089816792044666,
+        0.34422109438080617,
+        0.6128100029154656,
+        0.11920860204522012,
+        0.28816428858831905
+      ],
+      "eval_per_category_iou": [
+        0.5778473653822723,
+        0.5269478191058978,
+        0.574435506709114,
+        0.23668203929496492,
+        0.3570071601716556,
+        0.358769802685785,
+        0.41459541328306615,
+        0.37143503659747396,
+        0.2224686465997259,
+        0.4152571514805167,
+        0.09885903570114096,
+        0.25520652971420166
+      ],
+      "eval_runtime": 124.6105,
+      "eval_samples_per_second": 0.915,
+      "eval_steps_per_second": 0.08,
       "step": 220
     },
     {
       "epoch": 6.91,
       "learning_rate": 0.0005171249999999999,
-      "loss": 0.6746,
+      "loss": 0.7046,
       "step": 221
     },
     {
       "epoch": 6.94,
       "learning_rate": 0.00051675,
-      "loss": 0.6267,
+      "loss": 0.6436,
       "step": 222
     },
     {
       "epoch": 6.97,
       "learning_rate": 0.0005163749999999999,
-      "loss": 0.5898,
+      "loss": 0.6286,
       "step": 223
     },
     {
       "epoch": 7.0,
       "learning_rate": 0.000516,
-      "loss": 0.5663,
+      "loss": 0.621,
       "step": 224
     },
     {
       "epoch": 7.03,
       "learning_rate": 0.0005156249999999999,
-      "loss": 0.7091,
+      "loss": 0.7192,
       "step": 225
     },
     {
       "epoch": 7.06,
       "learning_rate": 0.00051525,
-      "loss": 0.5594,
+      "loss": 0.5768,
       "step": 226
     },
     {
       "epoch": 7.09,
       "learning_rate": 0.0005148749999999999,
-      "loss": 0.6431,
+      "loss": 0.6557,
       "step": 227
     },
     {
       "epoch": 7.12,
       "learning_rate": 0.0005145,
-      "loss": 0.5699,
+      "loss": 0.6194,
       "step": 228
     },
     {
       "epoch": 7.16,
       "learning_rate": 0.0005141249999999999,
-      "loss": 0.5724,
+      "loss": 0.624,
       "step": 229
     },
     {
       "epoch": 7.19,
       "learning_rate": 0.0005137499999999999,
-      "loss": 0.6459,
+      "loss": 0.6787,
       "step": 230
     },
     {
       "epoch": 7.22,
       "learning_rate": 0.0005133749999999999,
-      "loss": 0.6204,
+      "loss": 0.6515,
       "step": 231
     },
     {
       "epoch": 7.25,
       "learning_rate": 0.0005129999999999999,
-      "loss": 0.5749,
+      "loss": 0.618,
       "step": 232
     },
     {
       "epoch": 7.28,
       "learning_rate": 0.000512625,
-      "loss": 0.5592,
+      "loss": 0.5743,
       "step": 233
     },
     {
       "epoch": 7.31,
       "learning_rate": 0.00051225,
-      "loss": 0.5161,
+      "loss": 0.561,
       "step": 234
     },
     {
       "epoch": 7.34,
       "learning_rate": 0.000511875,
-      "loss": 0.6087,
+      "loss": 0.656,
       "step": 235
     },
     {
       "epoch": 7.38,
       "learning_rate": 0.0005115,
-      "loss": 0.5574,
+      "loss": 0.5951,
       "step": 236
     },
     {
       "epoch": 7.41,
       "learning_rate": 0.000511125,
-      "loss": 0.5837,
+      "loss": 0.6536,
       "step": 237
     },
     {
       "epoch": 7.44,
       "learning_rate": 0.0005107499999999999,
-      "loss": 0.6824,
+      "loss": 0.734,
       "step": 238
     },
     {
       "epoch": 7.47,
       "learning_rate": 0.000510375,
-      "loss": 0.6023,
+      "loss": 0.6387,
       "step": 239
     },
     {
       "epoch": 7.5,
       "learning_rate": 0.0005099999999999999,
-      "loss": 0.5475,
+      "loss": 0.5676,
       "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,
+      "eval_loss": 0.9805742502212524,
+      "eval_mean_accuracy": 0.5024018526714576,
+      "eval_mean_iou": 0.371506102641304,
+      "eval_overall_accuracy": 0.6800810161389803,
+      "eval_per_category_accuracy": [
+        0.7315930355784842,
+        0.7541007820537842,
+        0.6881292554956727,
+        0.30834622938078154,
+        0.5766004128790718,
+        0.4276563390300466,
+        0.6505111399057544,
+        0.4819251032078268,
+        0.27926275254429783,
+        0.5722067988240086,
+        0.07342223909057798,
+        0.48506814406718457
+      ],
+      "eval_per_category_iou": [
+        0.5874312271992882,
+        0.5604905946583346,
+        0.5587517085148045,
+        0.2254583833345596,
+        0.36177477005679304,
+        0.33678157470165493,
+        0.4185254872962371,
+        0.3728270419920094,
+        0.20258071367884453,
+        0.42075932706243124,
+        0.0692206473059698,
+        0.34347175589472057
+      ],
+      "eval_runtime": 123.7241,
+      "eval_samples_per_second": 0.921,
+      "eval_steps_per_second": 0.081,
       "step": 240
     },
     {
       "epoch": 7.53,
       "learning_rate": 0.000509625,
-      "loss": 0.5618,
+      "loss": 0.6097,
       "step": 241
     },
     {
       "epoch": 7.56,
       "learning_rate": 0.0005092499999999999,
-      "loss": 0.5768,
+      "loss": 0.6319,
       "step": 242
     },
     {
       "epoch": 7.59,
       "learning_rate": 0.000508875,
-      "loss": 0.605,
+      "loss": 0.6219,
       "step": 243
     },
     {
       "epoch": 7.62,
       "learning_rate": 0.0005085,
-      "loss": 0.565,
+      "loss": 0.5975,
       "step": 244
     },
     {
       "epoch": 7.66,
       "learning_rate": 0.000508125,
-      "loss": 0.6199,
+      "loss": 0.651,
       "step": 245
     },
     {
       "epoch": 7.69,
       "learning_rate": 0.00050775,
-      "loss": 0.4687,
+      "loss": 0.512,
       "step": 246
     },
     {
       "epoch": 7.72,
       "learning_rate": 0.0005073749999999999,
-      "loss": 0.5892,
+      "loss": 0.5962,
       "step": 247
     },
     {
       "epoch": 7.75,
       "learning_rate": 0.000507,
-      "loss": 0.6484,
+      "loss": 0.7017,
       "step": 248
     },
     {
       "epoch": 7.78,
       "learning_rate": 0.0005066249999999999,
-      "loss": 0.596,
+      "loss": 0.6477,
       "step": 249
     },
     {
       "epoch": 7.81,
       "learning_rate": 0.00050625,
-      "loss": 0.5773,
+      "loss": 0.6546,
       "step": 250
     },
     {
       "epoch": 7.84,
       "learning_rate": 0.0005058749999999999,
-      "loss": 0.6027,
+      "loss": 0.6881,
       "step": 251
     },
     {
       "epoch": 7.88,
       "learning_rate": 0.0005055,
-      "loss": 0.5894,
+      "loss": 0.6078,
       "step": 252
     },
     {
       "epoch": 7.91,
       "learning_rate": 0.0005051249999999999,
-      "loss": 0.6463,
+      "loss": 0.6891,
       "step": 253
     },
     {
       "epoch": 7.94,
       "learning_rate": 0.00050475,
-      "loss": 0.5688,
+      "loss": 0.598,
       "step": 254
     },
     {
       "epoch": 7.97,
       "learning_rate": 0.0005043749999999999,
-      "loss": 0.5924,
+      "loss": 0.638,
       "step": 255
     },
     {
       "epoch": 8.0,
       "learning_rate": 0.0005039999999999999,
-      "loss": 0.6616,
+      "loss": 0.6779,
       "step": 256
     },
     {
       "epoch": 8.03,
       "learning_rate": 0.000503625,
-      "loss": 0.5987,
+      "loss": 0.628,
       "step": 257
     },
     {
       "epoch": 8.06,
       "learning_rate": 0.0005032499999999999,
-      "loss": 0.5851,
+      "loss": 0.5842,
       "step": 258
     },
     {
       "epoch": 8.09,
       "learning_rate": 0.000502875,
-      "loss": 0.5072,
+      "loss": 0.5126,
       "step": 259
     },
     {
       "epoch": 8.12,
       "learning_rate": 0.0005025,
-      "loss": 0.6294,
+      "loss": 0.6533,
       "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,
+      "eval_loss": 0.9647707939147949,
+      "eval_mean_accuracy": 0.5290347832309982,
+      "eval_mean_iou": 0.3826944916955128,
+      "eval_overall_accuracy": 0.6695300654361123,
+      "eval_per_category_accuracy": [
+        0.6390227219897978,
+        0.7622451073333125,
+        0.6798905616562309,
+        0.39068084190949404,
+        0.5202418118057482,
+        0.5864632592845296,
+        0.7008623173460086,
+        0.5284176442793452,
+        0.26142604899603383,
+        0.5910629245092603,
+        0.23104988170045268,
+        0.45705427796176507
+      ],
+      "eval_per_category_iou": [
+        0.539363765599418,
+        0.5541112560719536,
+        0.564224130990146,
+        0.28007313069001544,
+        0.35043846601035505,
+        0.37497345958257,
+        0.4145645950481569,
+        0.40213470319634703,
+        0.1947903409748277,
+        0.4211198117656152,
+        0.15867080332782618,
+        0.33786943708892114
+      ],
+      "eval_runtime": 125.2253,
+      "eval_samples_per_second": 0.91,
+      "eval_steps_per_second": 0.08,
       "step": 260
     },
     {
       "epoch": 8.16,
       "learning_rate": 0.000502125,
-      "loss": 0.5728,
+      "loss": 0.5969,
       "step": 261
     },
     {
       "epoch": 8.19,
       "learning_rate": 0.00050175,
-      "loss": 0.4925,
+      "loss": 0.5633,
       "step": 262
     },
     {
       "epoch": 8.22,
       "learning_rate": 0.000501375,
-      "loss": 0.5615,
+      "loss": 0.6202,
       "step": 263
     },
     {
       "epoch": 8.25,
       "learning_rate": 0.0005009999999999999,
-      "loss": 0.5562,
+      "loss": 0.5568,
       "step": 264
     },
     {
       "epoch": 8.28,
       "learning_rate": 0.000500625,
-      "loss": 0.5429,
+      "loss": 0.6021,
       "step": 265
     },
     {
       "epoch": 8.31,
       "learning_rate": 0.0005002499999999999,
-      "loss": 0.5993,
+      "loss": 0.6637,
       "step": 266
     },
     {
       "epoch": 8.34,
       "learning_rate": 0.000499875,
-      "loss": 0.6164,
+      "loss": 0.6756,
       "step": 267
     },
     {
       "epoch": 8.38,
       "learning_rate": 0.0004994999999999999,
-      "loss": 0.5683,
+      "loss": 0.6323,
       "step": 268
     },
     {
       "epoch": 8.41,
       "learning_rate": 0.000499125,
-      "loss": 0.5876,
+      "loss": 0.6591,
       "step": 269
     },
     {
       "epoch": 8.44,
       "learning_rate": 0.00049875,
-      "loss": 0.648,
+      "loss": 0.7191,
       "step": 270
     },
     {
       "epoch": 8.47,
       "learning_rate": 0.0004983749999999999,
-      "loss": 0.5063,
+      "loss": 0.5911,
       "step": 271
     },
     {
       "epoch": 8.5,
       "learning_rate": 0.000498,
-      "loss": 0.5067,
+      "loss": 0.5841,
       "step": 272
     },
     {
       "epoch": 8.53,
       "learning_rate": 0.0004976249999999999,
-      "loss": 0.5348,
+      "loss": 0.5837,
       "step": 273
     },
     {
       "epoch": 8.56,
       "learning_rate": 0.00049725,
-      "loss": 0.5394,
+      "loss": 0.6009,
       "step": 274
     },
     {
       "epoch": 8.59,
       "learning_rate": 0.0004968749999999999,
-      "loss": 0.5167,
+      "loss": 0.5528,
       "step": 275
     },
     {
       "epoch": 8.62,
       "learning_rate": 0.0004965,
-      "loss": 0.5693,
+      "loss": 0.5907,
       "step": 276
     },
     {
       "epoch": 8.66,
       "learning_rate": 0.0004961249999999999,
-      "loss": 0.5314,
+      "loss": 0.5582,
       "step": 277
     },
     {
       "epoch": 8.69,
       "learning_rate": 0.00049575,
-      "loss": 0.5412,
+      "loss": 0.5714,
       "step": 278
     },
     {
       "epoch": 8.72,
       "learning_rate": 0.0004953749999999999,
-      "loss": 0.5591,
+      "loss": 0.6407,
       "step": 279
     },
     {
       "epoch": 8.75,
       "learning_rate": 0.0004949999999999999,
-      "loss": 0.6466,
+      "loss": 0.67,
       "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,
+      "eval_loss": 1.0987780094146729,
+      "eval_mean_accuracy": 0.4580998870390869,
+      "eval_mean_iou": 0.3479111276469824,
+      "eval_overall_accuracy": 0.6546705814830044,
+      "eval_per_category_accuracy": [
+        0.48066672745873446,
+        0.8305757369135304,
+        0.7676997659254449,
+        0.21412412127604943,
+        0.47398841588551216,
+        0.4325884901955803,
+        0.5711691272441585,
+        0.2749094353329169,
+        0.21508300577624376,
+        0.6848772870627909,
+        0.20816741298938213,
+        0.3433491184086993
+      ],
+      "eval_per_category_iou": [
+        0.4455015326675329,
+        0.5573106019694799,
+        0.5610375857814821,
+        0.19477330471696624,
+        0.35352567686407715,
+        0.32306020230054533,
+        0.4240965616003975,
+        0.26065154759824616,
+        0.18418258317583033,
+        0.4215337886509578,
+        0.15863326887955384,
+        0.2906268775587196
+      ],
+      "eval_runtime": 123.9472,
+      "eval_samples_per_second": 0.92,
+      "eval_steps_per_second": 0.081,
       "step": 280
     },
     {
       "epoch": 8.78,
       "learning_rate": 0.000494625,
-      "loss": 0.5825,
+      "loss": 0.6267,
       "step": 281
     },
     {
       "epoch": 8.81,
       "learning_rate": 0.0004942499999999999,
-      "loss": 0.6544,
+      "loss": 0.7083,
       "step": 282
     },
     {
       "epoch": 8.84,
       "learning_rate": 0.000493875,
-      "loss": 0.5803,
+      "loss": 0.6574,
       "step": 283
     },
     {
       "epoch": 8.88,
       "learning_rate": 0.0004935,
-      "loss": 0.526,
+      "loss": 0.5851,
       "step": 284
     },
     {
       "epoch": 8.91,
       "learning_rate": 0.000493125,
-      "loss": 0.5914,
+      "loss": 0.5928,
       "step": 285
     },
     {
       "epoch": 8.94,
       "learning_rate": 0.00049275,
-      "loss": 0.4828,
+      "loss": 0.5435,
       "step": 286
     },
     {
       "epoch": 8.97,
       "learning_rate": 0.000492375,
-      "loss": 0.5737,
+      "loss": 0.6518,
       "step": 287
     },
     {
       "epoch": 9.0,
       "learning_rate": 0.0004919999999999999,
-      "loss": 0.6182,
+      "loss": 0.654,
       "step": 288
     },
     {
       "epoch": 9.03,
       "learning_rate": 0.000491625,
-      "loss": 0.5085,
+      "loss": 0.5646,
       "step": 289
     },
     {
       "epoch": 9.06,
       "learning_rate": 0.0004912499999999999,
-      "loss": 0.4522,
+      "loss": 0.5048,
       "step": 290
     },
     {
       "epoch": 9.09,
       "learning_rate": 0.000490875,
-      "loss": 0.5149,
+      "loss": 0.5574,
       "step": 291
     },
     {
       "epoch": 9.12,
       "learning_rate": 0.0004904999999999999,
-      "loss": 0.5657,
+      "loss": 0.5995,
       "step": 292
     },
     {
       "epoch": 9.16,
       "learning_rate": 0.000490125,
-      "loss": 0.5126,
+      "loss": 0.5645,
       "step": 293
     },
     {
       "epoch": 9.19,
       "learning_rate": 0.00048975,
-      "loss": 0.4972,
+      "loss": 0.5597,
       "step": 294
     },
     {
       "epoch": 9.22,
       "learning_rate": 0.000489375,
-      "loss": 0.5683,
+      "loss": 0.6295,
       "step": 295
     },
     {
       "epoch": 9.25,
       "learning_rate": 0.000489,
-      "loss": 0.4438,
+      "loss": 0.5053,
       "step": 296
     },
     {
       "epoch": 9.28,
       "learning_rate": 0.0004886249999999999,
-      "loss": 0.4399,
+      "loss": 0.4657,
       "step": 297
     },
     {
       "epoch": 9.31,
       "learning_rate": 0.0004882499999999999,
-      "loss": 0.5879,
+      "loss": 0.6421,
       "step": 298
     },
     {
       "epoch": 9.34,
       "learning_rate": 0.000487875,
-      "loss": 0.513,
+      "loss": 0.5838,
       "step": 299
     },
     {
       "epoch": 9.38,
       "learning_rate": 0.0004875,
-      "loss": 0.5057,
+      "loss": 0.5644,
       "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,
+      "eval_loss": 0.983669638633728,
+      "eval_mean_accuracy": 0.5204264301169051,
+      "eval_mean_iou": 0.36998021455594227,
+      "eval_overall_accuracy": 0.6649547710753324,
+      "eval_per_category_accuracy": [
+        0.5940242572097704,
+        0.8056204971919827,
+        0.652415122648732,
+        0.3024337096897315,
+        0.6671805707779671,
+        0.7223677954844868,
+        0.5593419811729217,
+        0.49315930059755003,
+        0.35564048871813525,
+        0.4758290170708594,
+        0.1260647724455409,
+        0.491039648395184
+      ],
+      "eval_per_category_iou": [
+        0.5120521418371993,
+        0.5615051709697225,
+        0.5605287190343511,
+        0.24633585240536524,
+        0.31312694322687384,
+        0.3423289627909097,
+        0.42465141853084787,
+        0.4075806091690605,
+        0.2081047227792813,
+        0.3872001100058186,
+        0.10902319327019286,
+        0.36732473065168403
+      ],
+      "eval_runtime": 123.8999,
+      "eval_samples_per_second": 0.92,
+      "eval_steps_per_second": 0.081,
       "step": 300
     },
     {
       "epoch": 9.41,
       "learning_rate": 0.000487125,
-      "loss": 0.5731,
+      "loss": 0.6492,
       "step": 301
     },
     {
       "epoch": 9.44,
       "learning_rate": 0.00048675,
-      "loss": 0.5922,
+      "loss": 0.6658,
       "step": 302
     },
     {
       "epoch": 9.47,
       "learning_rate": 0.000486375,
-      "loss": 0.5068,
+      "loss": 0.5424,
       "step": 303
     },
     {
       "epoch": 9.5,
       "learning_rate": 0.000486,
-      "loss": 0.4608,
+      "loss": 0.4951,
       "step": 304
     },
     {
       "epoch": 9.53,
       "learning_rate": 0.00048562499999999995,
-      "loss": 0.5201,
+      "loss": 0.5663,
       "step": 305
     },
     {
       "epoch": 9.56,
       "learning_rate": 0.00048524999999999995,
-      "loss": 0.4989,
+      "loss": 0.5105,
       "step": 306
     },
     {
       "epoch": 9.59,
       "learning_rate": 0.00048487499999999995,
-      "loss": 0.5214,
+      "loss": 0.5594,
       "step": 307
     },
     {
       "epoch": 9.62,
       "learning_rate": 0.00048449999999999996,
-      "loss": 0.5588,
+      "loss": 0.5925,
       "step": 308
     },
     {
       "epoch": 9.66,
       "learning_rate": 0.00048412499999999996,
-      "loss": 0.5217,
+      "loss": 0.5676,
       "step": 309
     },
     {
       "epoch": 9.69,
       "learning_rate": 0.00048374999999999997,
-      "loss": 0.5234,
+      "loss": 0.552,
       "step": 310
     },
     {
       "epoch": 9.72,
       "learning_rate": 0.00048337499999999997,
-      "loss": 0.5274,
+      "loss": 0.6232,
       "step": 311
     },
     {
       "epoch": 9.75,
       "learning_rate": 0.000483,
-      "loss": 0.5443,
+      "loss": 0.5877,
       "step": 312
     },
     {
       "epoch": 9.78,
       "learning_rate": 0.0004826249999999999,
-      "loss": 0.479,
+      "loss": 0.5313,
       "step": 313
     },
     {
       "epoch": 9.81,
       "learning_rate": 0.00048224999999999993,
-      "loss": 0.4926,
+      "loss": 0.5456,
       "step": 314
     },
     {
       "epoch": 9.84,
       "learning_rate": 0.00048187499999999994,
-      "loss": 0.5416,
+      "loss": 0.5813,
       "step": 315
     },
     {
       "epoch": 9.88,
       "learning_rate": 0.00048149999999999994,
-      "loss": 0.5723,
+      "loss": 0.6503,
       "step": 316
     },
     {
       "epoch": 9.91,
       "learning_rate": 0.00048112499999999994,
-      "loss": 0.5107,
+      "loss": 0.5662,
       "step": 317
     },
     {
       "epoch": 9.94,
       "learning_rate": 0.00048074999999999995,
-      "loss": 0.6088,
+      "loss": 0.6718,
       "step": 318
     },
     {
       "epoch": 9.97,
       "learning_rate": 0.00048037499999999995,
-      "loss": 0.494,
+      "loss": 0.5636,
       "step": 319
     },
     {
       "epoch": 10.0,
       "learning_rate": 0.00047999999999999996,
-      "loss": 0.5537,
+      "loss": 0.5678,
       "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,
+      "eval_loss": 0.9780566692352295,
+      "eval_mean_accuracy": 0.5350595191545178,
+      "eval_mean_iou": 0.3906962436607775,
+      "eval_overall_accuracy": 0.6771719413891173,
+      "eval_per_category_accuracy": [
+        0.6554152883827387,
+        0.7588004293882252,
+        0.7401698246138683,
+        0.3209831992838356,
+        0.5589183297643672,
+        0.5685601675261941,
+        0.5751910181705395,
+        0.45440587328339643,
+        0.4784721790901751,
+        0.6453562996222412,
+        0.21756097859589305,
+        0.4468806421327391
+      ],
+      "eval_per_category_iou": [
+        0.5545401246305556,
+        0.559287046180608,
+        0.5841953654092598,
+        0.24684268808453297,
+        0.3812876262500622,
+        0.3824948025455624,
+        0.4319588439462186,
+        0.384631151257913,
+        0.2309813395356408,
+        0.4207958788984324,
+        0.15715858689932388,
+        0.3541814702912209
+      ],
+      "eval_runtime": 123.6926,
+      "eval_samples_per_second": 0.922,
+      "eval_steps_per_second": 0.081,
       "step": 320
     },
     {
       "epoch": 10.03,
       "learning_rate": 0.0004796249999999999,
-      "loss": 0.514,
+      "loss": 0.5511,
       "step": 321
     },
     {
       "epoch": 10.06,
       "learning_rate": 0.0004792499999999999,
-      "loss": 0.5937,
+      "loss": 0.611,
       "step": 322
     },
     {
       "epoch": 10.09,
       "learning_rate": 0.0004788749999999999,
-      "loss": 0.5341,
+      "loss": 0.5803,
       "step": 323
     },
     {
       "epoch": 10.12,
       "learning_rate": 0.0004785,
-      "loss": 0.4831,
+      "loss": 0.5232,
       "step": 324
     },
     {
       "epoch": 10.16,
       "learning_rate": 0.000478125,
-      "loss": 0.5363,
+      "loss": 0.5558,
       "step": 325
     },
     {
       "epoch": 10.19,
       "learning_rate": 0.00047775,
-      "loss": 0.5426,
+      "loss": 0.5829,
       "step": 326
     },
     {
       "epoch": 10.22,
       "learning_rate": 0.000477375,
-      "loss": 0.4749,
+      "loss": 0.5082,
       "step": 327
     },
     {
       "epoch": 10.25,
       "learning_rate": 0.000477,
-      "loss": 0.5598,
+      "loss": 0.5775,
       "step": 328
     },
     {
       "epoch": 10.28,
       "learning_rate": 0.000476625,
-      "loss": 0.4966,
+      "loss": 0.5354,
       "step": 329
     },
     {
       "epoch": 10.31,
       "learning_rate": 0.00047624999999999995,
-      "loss": 0.4268,
+      "loss": 0.4546,
       "step": 330
     },
     {
       "epoch": 10.34,
       "learning_rate": 0.00047587499999999995,
-      "loss": 0.4425,
+      "loss": 0.5009,
       "step": 331
     },
     {
       "epoch": 10.38,
       "learning_rate": 0.00047549999999999996,
-      "loss": 0.4802,
+      "loss": 0.5521,
       "step": 332
     },
     {
       "epoch": 10.41,
       "learning_rate": 0.00047512499999999996,
-      "loss": 0.4799,
+      "loss": 0.5354,
       "step": 333
     },
     {
       "epoch": 10.44,
       "learning_rate": 0.00047474999999999997,
-      "loss": 0.4465,
+      "loss": 0.4855,
       "step": 334
     },
     {
       "epoch": 10.47,
       "learning_rate": 0.00047437499999999997,
-      "loss": 0.491,
+      "loss": 0.5356,
       "step": 335
     },
     {
       "epoch": 10.5,
       "learning_rate": 0.000474,
-      "loss": 0.4901,
+      "loss": 0.5868,
       "step": 336
     },
     {
       "epoch": 10.53,
       "learning_rate": 0.000473625,
-      "loss": 0.5444,
+      "loss": 0.5931,
       "step": 337
     },
     {
       "epoch": 10.56,
       "learning_rate": 0.00047324999999999993,
-      "loss": 0.5099,
+      "loss": 0.5786,
       "step": 338
     },
     {
       "epoch": 10.59,
       "learning_rate": 0.00047287499999999993,
-      "loss": 0.5143,
+      "loss": 0.5718,
       "step": 339
     },
     {
       "epoch": 10.62,
       "learning_rate": 0.00047249999999999994,
-      "loss": 0.5615,
+      "loss": 0.6153,
       "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,
+      "eval_loss": 0.9987176060676575,
+      "eval_mean_accuracy": 0.5497999939840363,
+      "eval_mean_iou": 0.3857005670859029,
+      "eval_overall_accuracy": 0.660497130009166,
+      "eval_per_category_accuracy": [
+        0.5231205889958428,
+        0.7747069218397336,
+        0.7498118814452185,
+        0.3063129332898453,
+        0.546321263677379,
+        0.6574151027622224,
+        0.6809347098082498,
+        0.45994636300225755,
+        0.4333802117068756,
+        0.5861572233919434,
+        0.342909526097829,
+        0.5365832017910386
+      ],
+      "eval_per_category_iou": [
+        0.4772630084705572,
+        0.5541882497823591,
+        0.5825233033041953,
+        0.24382429625701532,
+        0.38442721103951644,
+        0.38358193468602964,
+        0.42849690958511605,
+        0.3960627429416036,
+        0.21535871992663122,
+        0.4152440328818222,
+        0.1834708686588136,
+        0.36396552749717453
+      ],
+      "eval_runtime": 123.7757,
+      "eval_samples_per_second": 0.921,
+      "eval_steps_per_second": 0.081,
       "step": 340
     },
     {
       "epoch": 10.66,
       "learning_rate": 0.00047212499999999994,
-      "loss": 0.583,
+      "loss": 0.6764,
       "step": 341
     },
     {
       "epoch": 10.69,
       "learning_rate": 0.00047174999999999995,
-      "loss": 0.4993,
+      "loss": 0.541,
       "step": 342
     },
     {
       "epoch": 10.72,
       "learning_rate": 0.00047137499999999995,
-      "loss": 0.4513,
+      "loss": 0.5097,
       "step": 343
     },
     {
       "epoch": 10.75,
       "learning_rate": 0.00047099999999999996,
-      "loss": 0.4375,
+      "loss": 0.4856,
       "step": 344
     },
     {
       "epoch": 10.78,
       "learning_rate": 0.00047062499999999996,
-      "loss": 0.5255,
+      "loss": 0.5468,
       "step": 345
     },
     {
       "epoch": 10.81,
       "learning_rate": 0.0004702499999999999,
-      "loss": 0.4966,
+      "loss": 0.5513,
       "step": 346
     },
     {
       "epoch": 10.84,
       "learning_rate": 0.0004698749999999999,
-      "loss": 0.5821,
+      "loss": 0.6115,
       "step": 347
     },
     {
       "epoch": 10.88,
       "learning_rate": 0.0004694999999999999,
-      "loss": 0.4963,
+      "loss": 0.5045,
       "step": 348
     },
     {
       "epoch": 10.91,
       "learning_rate": 0.0004691249999999999,
-      "loss": 0.4858,
+      "loss": 0.5671,
       "step": 349
     },
     {
       "epoch": 10.94,
       "learning_rate": 0.00046875,
-      "loss": 0.4664,
+      "loss": 0.5062,
       "step": 350
     },
     {
       "epoch": 10.97,
       "learning_rate": 0.000468375,
-      "loss": 0.5505,
+      "loss": 0.6014,
       "step": 351
     },
     {
       "epoch": 11.0,
       "learning_rate": 0.000468,
-      "loss": 0.5578,
+      "loss": 0.5981,
       "step": 352
     },
     {
       "epoch": 11.03,
       "learning_rate": 0.000467625,
-      "loss": 0.4877,
+      "loss": 0.5369,
       "step": 353
     },
     {
       "epoch": 11.06,
       "learning_rate": 0.00046725,
-      "loss": 0.4719,
+      "loss": 0.4975,
       "step": 354
     },
     {
       "epoch": 11.09,
       "learning_rate": 0.00046687499999999995,
-      "loss": 0.481,
+      "loss": 0.5137,
       "step": 355
     },
     {
       "epoch": 11.12,
       "learning_rate": 0.00046649999999999996,
-      "loss": 0.4697,
+      "loss": 0.4978,
       "step": 356
     },
     {
       "epoch": 11.16,
       "learning_rate": 0.00046612499999999996,
-      "loss": 0.4387,
+      "loss": 0.4579,
       "step": 357
     },
     {
       "epoch": 11.19,
       "learning_rate": 0.00046574999999999996,
-      "loss": 0.5047,
+      "loss": 0.5321,
       "step": 358
     },
     {
       "epoch": 11.22,
       "learning_rate": 0.00046537499999999997,
-      "loss": 0.4715,
+      "loss": 0.5201,
       "step": 359
     },
     {
       "epoch": 11.25,
       "learning_rate": 0.00046499999999999997,
-      "loss": 0.3817,
+      "loss": 0.4252,
       "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,
+      "eval_loss": 0.9291033148765564,
+      "eval_mean_accuracy": 0.528009973276733,
+      "eval_mean_iou": 0.39591471843488896,
+      "eval_overall_accuracy": 0.6944302341394257,
+      "eval_per_category_accuracy": [
+        0.7758503951264145,
+        0.7639770669563318,
+        0.6693553518837109,
+        0.38660037057341723,
+        0.5337667415575549,
+        0.5057342154831077,
+        0.5763875877943601,
+        0.6287229015837165,
+        0.3423444806260703,
+        0.5546978372878139,
+        0.15132853605458002,
+        0.447354194393719
+      ],
+      "eval_per_category_iou": [
+        0.6138349391900487,
+        0.5748497846975844,
+        0.5686854787211605,
+        0.28537363740676996,
+        0.3701880790533382,
+        0.3687162885337687,
+        0.4387556373721067,
+        0.4152303639246888,
+        0.22665730662981398,
+        0.4070483981278673,
+        0.12507399555773677,
+        0.35656271200378264
+      ],
+      "eval_runtime": 123.9322,
+      "eval_samples_per_second": 0.92,
+      "eval_steps_per_second": 0.081,
       "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,
+  "total_flos": 7.462066884968448e+16,
   "trial_name": null,
   "trial_params": null
 }