diff --git "a/checkpoint-2560/trainer_state.json" "b/checkpoint-2560/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/checkpoint-2560/trainer_state.json"
@@ -0,0 +1,20368 @@
+{
+  "best_metric": 0.08097036182880402,
+  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0-optim/checkpoint-2560",
+  "epoch": 232.72727272727272,
+  "global_step": 2560,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.09,
+      "learning_rate": 3.6363636363636366e-07,
+      "loss": 2.757,
+      "step": 1
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 7.272727272727273e-07,
+      "loss": 2.7552,
+      "step": 2
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 1.090909090909091e-06,
+      "loss": 2.7635,
+      "step": 3
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 1.4545454545454546e-06,
+      "loss": 2.7567,
+      "step": 4
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 1.818181818181818e-06,
+      "loss": 2.7588,
+      "step": 5
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 2.181818181818182e-06,
+      "loss": 2.7621,
+      "step": 6
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 2.5454545454545456e-06,
+      "loss": 2.7553,
+      "step": 7
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 2.9090909090909093e-06,
+      "loss": 2.7371,
+      "step": 8
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 3.2727272727272733e-06,
+      "loss": 2.7444,
+      "step": 9
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 3.636363636363636e-06,
+      "loss": 2.7363,
+      "step": 10
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 4.000000000000001e-06,
+      "loss": 2.7364,
+      "step": 11
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 4.363636363636364e-06,
+      "loss": 2.7248,
+      "step": 12
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 4.727272727272727e-06,
+      "loss": 2.7344,
+      "step": 13
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 5.090909090909091e-06,
+      "loss": 2.7259,
+      "step": 14
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 5.4545454545454545e-06,
+      "loss": 2.7086,
+      "step": 15
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 5.8181818181818185e-06,
+      "loss": 2.7087,
+      "step": 16
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 6.181818181818183e-06,
+      "loss": 2.7055,
+      "step": 17
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 6.545454545454547e-06,
+      "loss": 2.6959,
+      "step": 18
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 6.909090909090909e-06,
+      "loss": 2.6983,
+      "step": 19
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 7.272727272727272e-06,
+      "loss": 2.6746,
+      "step": 20
+    },
+    {
+      "epoch": 1.82,
+      "eval_accuracy_NAT": 0.019056367662015652,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.0010381346838920125,
+      "eval_accuracy_concrete tiles": 0.09181801504311135,
+      "eval_accuracy_dark bricks": 0.09326045956660636,
+      "eval_accuracy_grass": 0.007958282800392846,
+      "eval_accuracy_red bricks": 0.00870028409090909,
+      "eval_accuracy_road": 0.05730198902763889,
+      "eval_accuracy_rough concrete": 0.03398947127489128,
+      "eval_accuracy_rough red sidewalk": 0.06001709260331376,
+      "eval_accuracy_speedway bricks": 0.03133919121631325,
+      "eval_accuracy_steel": 0.00842622518303542,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.018937014149192417,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.0009231352727125269,
+      "eval_iou_concrete tiles": 0.002698064724802789,
+      "eval_iou_dark bricks": 0.0018963436490879863,
+      "eval_iou_grass": 0.006090841001497967,
+      "eval_iou_red bricks": 1.3048864535742704e-05,
+      "eval_iou_road": 0.0465812133841859,
+      "eval_iou_rough concrete": 0.004291938525567382,
+      "eval_iou_rough red sidewalk": 0.0013826848729701939,
+      "eval_iou_speedway bricks": 0.030977499755458056,
+      "eval_iou_steel": 0.0016645601782480097,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 2.647437810897827,
+      "eval_mean_accuracy": 0.03176196255016307,
+      "eval_mean_iou": 0.008246881741304212,
+      "eval_overall_accuracy": 0.02327410380045573,
+      "eval_runtime": 3.4336,
+      "eval_samples_per_second": 10.485,
+      "eval_steps_per_second": 0.582,
+      "step": 20
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 7.636363636363638e-06,
+      "loss": 2.6783,
+      "step": 21
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 8.000000000000001e-06,
+      "loss": 2.6636,
+      "step": 22
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 8.363636363636365e-06,
+      "loss": 2.6626,
+      "step": 23
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 8.727272727272728e-06,
+      "loss": 2.6414,
+      "step": 24
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 9.090909090909091e-06,
+      "loss": 2.6365,
+      "step": 25
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 9.454545454545454e-06,
+      "loss": 2.6263,
+      "step": 26
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 9.818181818181818e-06,
+      "loss": 2.6182,
+      "step": 27
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 1.0181818181818182e-05,
+      "loss": 2.6144,
+      "step": 28
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 1.0545454545454546e-05,
+      "loss": 2.5878,
+      "step": 29
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 1.0909090909090909e-05,
+      "loss": 2.5814,
+      "step": 30
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 1.1272727272727274e-05,
+      "loss": 2.5739,
+      "step": 31
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 1.1636363636363637e-05,
+      "loss": 2.5536,
+      "step": 32
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 1.2e-05,
+      "loss": 2.539,
+      "step": 33
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 1.2363636363636365e-05,
+      "loss": 2.5256,
+      "step": 34
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 1.2727272727272727e-05,
+      "loss": 2.4979,
+      "step": 35
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 1.3090909090909093e-05,
+      "loss": 2.4936,
+      "step": 36
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 1.3454545454545457e-05,
+      "loss": 2.4862,
+      "step": 37
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 1.3818181818181818e-05,
+      "loss": 2.4491,
+      "step": 38
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 1.4181818181818181e-05,
+      "loss": 2.4468,
+      "step": 39
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 1.4545454545454545e-05,
+      "loss": 2.4385,
+      "step": 40
+    },
+    {
+      "epoch": 3.64,
+      "eval_accuracy_NAT": 0.447892573737157,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.04360568050130907,
+      "eval_accuracy_concrete tiles": 0.10279459426404941,
+      "eval_accuracy_dark bricks": 0.07266848851083484,
+      "eval_accuracy_grass": 0.017545269942016727,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.37485716220448445,
+      "eval_accuracy_rough concrete": 0.2949759670405127,
+      "eval_accuracy_rough red sidewalk": 0.05480660546411932,
+      "eval_accuracy_speedway bricks": 0.7969013802350152,
+      "eval_accuracy_steel": 0.0050211067871512435,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.4472536503231145,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.039962583094048695,
+      "eval_iou_concrete tiles": 0.0041023463419931645,
+      "eval_iou_dark bricks": 0.007329448622040366,
+      "eval_iou_grass": 0.010573184550513284,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.2852347789122688,
+      "eval_iou_rough concrete": 0.048997709321445884,
+      "eval_iou_rough red sidewalk": 0.003679975417468036,
+      "eval_iou_speedway bricks": 0.7166237181229831,
+      "eval_iou_steel": 0.0015635711039736272,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 2.4373269081115723,
+      "eval_mean_accuracy": 0.17008221759128078,
+      "eval_mean_iou": 0.11180864041498925,
+      "eval_overall_accuracy": 0.4471186531914605,
+      "eval_runtime": 3.2402,
+      "eval_samples_per_second": 11.111,
+      "eval_steps_per_second": 0.617,
+      "step": 40
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 1.4909090909090908e-05,
+      "loss": 2.4028,
+      "step": 41
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 1.5272727272727276e-05,
+      "loss": 2.4083,
+      "step": 42
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 1.563636363636364e-05,
+      "loss": 2.3613,
+      "step": 43
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 1.6000000000000003e-05,
+      "loss": 2.3597,
+      "step": 44
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 1.6363636363636366e-05,
+      "loss": 2.3331,
+      "step": 45
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 1.672727272727273e-05,
+      "loss": 2.3068,
+      "step": 46
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 1.7090909090909092e-05,
+      "loss": 2.2492,
+      "step": 47
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 1.7454545454545456e-05,
+      "loss": 2.2469,
+      "step": 48
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 1.781818181818182e-05,
+      "loss": 2.2194,
+      "step": 49
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 1.8181818181818182e-05,
+      "loss": 2.2206,
+      "step": 50
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 1.8545454545454545e-05,
+      "loss": 2.1524,
+      "step": 51
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 1.890909090909091e-05,
+      "loss": 2.1454,
+      "step": 52
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 1.9272727272727272e-05,
+      "loss": 2.1673,
+      "step": 53
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 1.9636363636363635e-05,
+      "loss": 2.1274,
+      "step": 54
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 2e-05,
+      "loss": 2.0958,
+      "step": 55
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 2.0363636363636365e-05,
+      "loss": 2.0027,
+      "step": 56
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 2.0727272727272728e-05,
+      "loss": 2.0081,
+      "step": 57
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 2.109090909090909e-05,
+      "loss": 2.016,
+      "step": 58
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 2.1454545454545455e-05,
+      "loss": 1.9957,
+      "step": 59
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 2.1818181818181818e-05,
+      "loss": 1.9773,
+      "step": 60
+    },
+    {
+      "epoch": 5.45,
+      "eval_accuracy_NAT": 0.898956856290447,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.24511379243744366,
+      "eval_accuracy_concrete tiles": 0.007154650522839846,
+      "eval_accuracy_dark bricks": 0.009314499057003194,
+      "eval_accuracy_grass": 0.07155464372966304,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.7720496210944066,
+      "eval_accuracy_rough concrete": 0.0036621652552071412,
+      "eval_accuracy_rough red sidewalk": 0.010007443553055993,
+      "eval_accuracy_speedway bricks": 0.9787968531076106,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.0009587727708533077,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.89400043378223,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.19197997304386832,
+      "eval_iou_concrete tiles": 0.0013976657786909726,
+      "eval_iou_dark bricks": 0.005338039042682255,
+      "eval_iou_grass": 0.07039652848458174,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.5405843619796408,
+      "eval_iou_rough concrete": 0.0013066024253807521,
+      "eval_iou_rough red sidewalk": 0.003800927719548077,
+      "eval_iou_speedway bricks": 0.6998719868699171,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.00034965034965034965,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 1.916090488433838,
+      "eval_mean_accuracy": 0.23058225367834848,
+      "eval_mean_iou": 0.1720732978197279,
+      "eval_overall_accuracy": 0.8010207282172309,
+      "eval_runtime": 3.5175,
+      "eval_samples_per_second": 10.235,
+      "eval_steps_per_second": 0.569,
+      "step": 60
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 2.2181818181818184e-05,
+      "loss": 1.9281,
+      "step": 61
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 2.2545454545454548e-05,
+      "loss": 1.8834,
+      "step": 62
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 2.290909090909091e-05,
+      "loss": 1.8846,
+      "step": 63
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 2.3272727272727274e-05,
+      "loss": 1.8532,
+      "step": 64
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 2.3636363636363637e-05,
+      "loss": 1.8357,
+      "step": 65
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 2.4e-05,
+      "loss": 1.8409,
+      "step": 66
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 2.4363636363636364e-05,
+      "loss": 1.7691,
+      "step": 67
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 2.472727272727273e-05,
+      "loss": 1.7549,
+      "step": 68
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 2.5090909090909094e-05,
+      "loss": 1.8554,
+      "step": 69
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 2.5454545454545454e-05,
+      "loss": 1.7217,
+      "step": 70
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 2.581818181818182e-05,
+      "loss": 1.7575,
+      "step": 71
+    },
+    {
+      "epoch": 6.55,
+      "learning_rate": 2.6181818181818187e-05,
+      "loss": 1.7587,
+      "step": 72
+    },
+    {
+      "epoch": 6.64,
+      "learning_rate": 2.6545454545454547e-05,
+      "loss": 1.6981,
+      "step": 73
+    },
+    {
+      "epoch": 6.73,
+      "learning_rate": 2.6909090909090913e-05,
+      "loss": 1.6573,
+      "step": 74
+    },
+    {
+      "epoch": 6.82,
+      "learning_rate": 2.7272727272727273e-05,
+      "loss": 1.6428,
+      "step": 75
+    },
+    {
+      "epoch": 6.91,
+      "learning_rate": 2.7636363636363636e-05,
+      "loss": 1.6877,
+      "step": 76
+    },
+    {
+      "epoch": 7.0,
+      "learning_rate": 2.8000000000000003e-05,
+      "loss": 1.7226,
+      "step": 77
+    },
+    {
+      "epoch": 7.09,
+      "learning_rate": 2.8363636363636363e-05,
+      "loss": 1.6332,
+      "step": 78
+    },
+    {
+      "epoch": 7.18,
+      "learning_rate": 2.872727272727273e-05,
+      "loss": 1.6322,
+      "step": 79
+    },
+    {
+      "epoch": 7.27,
+      "learning_rate": 2.909090909090909e-05,
+      "loss": 1.6038,
+      "step": 80
+    },
+    {
+      "epoch": 7.27,
+      "eval_accuracy_NAT": 0.9470286690715778,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.7309205867204601,
+      "eval_accuracy_concrete tiles": 0.0014064697609001407,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.029533372711163616,
+      "eval_accuracy_red bricks": 0.015447443181818182,
+      "eval_accuracy_road": 0.8218001008480181,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0020400849116422684,
+      "eval_accuracy_speedway bricks": 0.9799789289994171,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.937594640128183,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.4290493301504721,
+      "eval_iou_concrete tiles": 0.0011021395883748233,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.029093488223162548,
+      "eval_iou_red bricks": 0.002611828279795857,
+      "eval_iou_road": 0.7049759862244951,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0009831927190593237,
+      "eval_iou_speedway bricks": 0.7986861621141007,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 1.4904648065567017,
+      "eval_mean_accuracy": 0.27139658893884594,
+      "eval_mean_iou": 0.2074354833876888,
+      "eval_overall_accuracy": 0.8714875115288628,
+      "eval_runtime": 3.8608,
+      "eval_samples_per_second": 9.325,
+      "eval_steps_per_second": 0.518,
+      "step": 80
+    },
+    {
+      "epoch": 7.36,
+      "learning_rate": 2.9454545454545456e-05,
+      "loss": 1.6326,
+      "step": 81
+    },
+    {
+      "epoch": 7.45,
+      "learning_rate": 2.9818181818181816e-05,
+      "loss": 1.5668,
+      "step": 82
+    },
+    {
+      "epoch": 7.55,
+      "learning_rate": 3.0181818181818182e-05,
+      "loss": 1.5495,
+      "step": 83
+    },
+    {
+      "epoch": 7.64,
+      "learning_rate": 3.054545454545455e-05,
+      "loss": 1.5698,
+      "step": 84
+    },
+    {
+      "epoch": 7.73,
+      "learning_rate": 3.090909090909091e-05,
+      "loss": 1.4842,
+      "step": 85
+    },
+    {
+      "epoch": 7.82,
+      "learning_rate": 3.127272727272728e-05,
+      "loss": 1.4979,
+      "step": 86
+    },
+    {
+      "epoch": 7.91,
+      "learning_rate": 3.1636363636363635e-05,
+      "loss": 1.5206,
+      "step": 87
+    },
+    {
+      "epoch": 8.0,
+      "learning_rate": 3.2000000000000005e-05,
+      "loss": 1.555,
+      "step": 88
+    },
+    {
+      "epoch": 8.09,
+      "learning_rate": 3.236363636363636e-05,
+      "loss": 1.47,
+      "step": 89
+    },
+    {
+      "epoch": 8.18,
+      "learning_rate": 3.272727272727273e-05,
+      "loss": 1.5321,
+      "step": 90
+    },
+    {
+      "epoch": 8.27,
+      "learning_rate": 3.3090909090909095e-05,
+      "loss": 1.5188,
+      "step": 91
+    },
+    {
+      "epoch": 8.36,
+      "learning_rate": 3.345454545454546e-05,
+      "loss": 1.4527,
+      "step": 92
+    },
+    {
+      "epoch": 8.45,
+      "learning_rate": 3.381818181818182e-05,
+      "loss": 1.4205,
+      "step": 93
+    },
+    {
+      "epoch": 8.55,
+      "learning_rate": 3.4181818181818185e-05,
+      "loss": 1.4396,
+      "step": 94
+    },
+    {
+      "epoch": 8.64,
+      "learning_rate": 3.454545454545455e-05,
+      "loss": 1.4194,
+      "step": 95
+    },
+    {
+      "epoch": 8.73,
+      "learning_rate": 3.490909090909091e-05,
+      "loss": 1.4087,
+      "step": 96
+    },
+    {
+      "epoch": 8.82,
+      "learning_rate": 3.5272727272727274e-05,
+      "loss": 1.4411,
+      "step": 97
+    },
+    {
+      "epoch": 8.91,
+      "learning_rate": 3.563636363636364e-05,
+      "loss": 1.3689,
+      "step": 98
+    },
+    {
+      "epoch": 9.0,
+      "learning_rate": 3.6e-05,
+      "loss": 1.3591,
+      "step": 99
+    },
+    {
+      "epoch": 9.09,
+      "learning_rate": 3.6363636363636364e-05,
+      "loss": 1.4392,
+      "step": 100
+    },
+    {
+      "epoch": 9.09,
+      "eval_accuracy_NAT": 0.9504804058093353,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.76187819219709,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.07590950555192456,
+      "eval_accuracy_red bricks": 0.0005326704545454545,
+      "eval_accuracy_road": 0.8848434134998494,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0008821988807101701,
+      "eval_accuracy_speedway bricks": 0.9805213926169435,
+      "eval_accuracy_steel": 0.0003380383879691313,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9410302391336439,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.44589299171062546,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.06707307657862148,
+      "eval_iou_red bricks": 0.0003791229622140781,
+      "eval_iou_road": 0.7530169393154503,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0006904289289721239,
+      "eval_iou_speedway bricks": 0.8219766311954673,
+      "eval_iou_steel": 0.00033548260399960723,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 1.2885687351226807,
+      "eval_mean_accuracy": 0.281183524415259,
+      "eval_mean_iou": 0.21645677945921385,
+      "eval_overall_accuracy": 0.883018069797092,
+      "eval_runtime": 3.8233,
+      "eval_samples_per_second": 9.416,
+      "eval_steps_per_second": 0.523,
+      "step": 100
+    },
+    {
+      "epoch": 9.18,
+      "learning_rate": 3.672727272727273e-05,
+      "loss": 1.3807,
+      "step": 101
+    },
+    {
+      "epoch": 9.27,
+      "learning_rate": 3.709090909090909e-05,
+      "loss": 1.3464,
+      "step": 102
+    },
+    {
+      "epoch": 9.36,
+      "learning_rate": 3.745454545454546e-05,
+      "loss": 1.377,
+      "step": 103
+    },
+    {
+      "epoch": 9.45,
+      "learning_rate": 3.781818181818182e-05,
+      "loss": 1.314,
+      "step": 104
+    },
+    {
+      "epoch": 9.55,
+      "learning_rate": 3.818181818181819e-05,
+      "loss": 1.3202,
+      "step": 105
+    },
+    {
+      "epoch": 9.64,
+      "learning_rate": 3.8545454545454544e-05,
+      "loss": 1.2905,
+      "step": 106
+    },
+    {
+      "epoch": 9.73,
+      "learning_rate": 3.8909090909090914e-05,
+      "loss": 1.3153,
+      "step": 107
+    },
+    {
+      "epoch": 9.82,
+      "learning_rate": 3.927272727272727e-05,
+      "loss": 1.3037,
+      "step": 108
+    },
+    {
+      "epoch": 9.91,
+      "learning_rate": 3.963636363636364e-05,
+      "loss": 1.3246,
+      "step": 109
+    },
+    {
+      "epoch": 10.0,
+      "learning_rate": 4e-05,
+      "loss": 1.3242,
+      "step": 110
+    },
+    {
+      "epoch": 10.09,
+      "learning_rate": 4.0363636363636367e-05,
+      "loss": 1.2856,
+      "step": 111
+    },
+    {
+      "epoch": 10.18,
+      "learning_rate": 4.072727272727273e-05,
+      "loss": 1.2349,
+      "step": 112
+    },
+    {
+      "epoch": 10.27,
+      "learning_rate": 4.109090909090909e-05,
+      "loss": 1.2786,
+      "step": 113
+    },
+    {
+      "epoch": 10.36,
+      "learning_rate": 4.1454545454545456e-05,
+      "loss": 1.2887,
+      "step": 114
+    },
+    {
+      "epoch": 10.45,
+      "learning_rate": 4.181818181818182e-05,
+      "loss": 1.2777,
+      "step": 115
+    },
+    {
+      "epoch": 10.55,
+      "learning_rate": 4.218181818181818e-05,
+      "loss": 1.2586,
+      "step": 116
+    },
+    {
+      "epoch": 10.64,
+      "learning_rate": 4.254545454545455e-05,
+      "loss": 1.2073,
+      "step": 117
+    },
+    {
+      "epoch": 10.73,
+      "learning_rate": 4.290909090909091e-05,
+      "loss": 1.2722,
+      "step": 118
+    },
+    {
+      "epoch": 10.82,
+      "learning_rate": 4.327272727272728e-05,
+      "loss": 1.2427,
+      "step": 119
+    },
+    {
+      "epoch": 10.91,
+      "learning_rate": 4.3636363636363636e-05,
+      "loss": 1.1951,
+      "step": 120
+    },
+    {
+      "epoch": 10.91,
+      "eval_accuracy_NAT": 0.9539229670890436,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8772452680372548,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.14066972602117317,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.8862182839633513,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.981366766953044,
+      "eval_accuracy_steel": 0.00023085548446672384,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9437268461219142,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.5137735030918505,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.1162642232152916,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8261045014174364,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.844953008348876,
+      "eval_iou_steel": 0.0002300380384327837,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 1.1216386556625366,
+      "eval_mean_accuracy": 0.2953579898114103,
+      "eval_mean_iou": 0.24961939386413856,
+      "eval_overall_accuracy": 0.8964270485772027,
+      "eval_runtime": 3.543,
+      "eval_samples_per_second": 10.161,
+      "eval_steps_per_second": 0.564,
+      "step": 120
+    },
+    {
+      "epoch": 11.0,
+      "learning_rate": 4.4000000000000006e-05,
+      "loss": 1.1897,
+      "step": 121
+    },
+    {
+      "epoch": 11.09,
+      "learning_rate": 4.436363636363637e-05,
+      "loss": 1.163,
+      "step": 122
+    },
+    {
+      "epoch": 11.18,
+      "learning_rate": 4.472727272727273e-05,
+      "loss": 1.1332,
+      "step": 123
+    },
+    {
+      "epoch": 11.27,
+      "learning_rate": 4.5090909090909095e-05,
+      "loss": 1.2278,
+      "step": 124
+    },
+    {
+      "epoch": 11.36,
+      "learning_rate": 4.545454545454546e-05,
+      "loss": 1.1608,
+      "step": 125
+    },
+    {
+      "epoch": 11.45,
+      "learning_rate": 4.581818181818182e-05,
+      "loss": 1.2089,
+      "step": 126
+    },
+    {
+      "epoch": 11.55,
+      "learning_rate": 4.618181818181818e-05,
+      "loss": 1.1246,
+      "step": 127
+    },
+    {
+      "epoch": 11.64,
+      "learning_rate": 4.654545454545455e-05,
+      "loss": 1.1045,
+      "step": 128
+    },
+    {
+      "epoch": 11.73,
+      "learning_rate": 4.690909090909091e-05,
+      "loss": 1.1625,
+      "step": 129
+    },
+    {
+      "epoch": 11.82,
+      "learning_rate": 4.7272727272727275e-05,
+      "loss": 1.2094,
+      "step": 130
+    },
+    {
+      "epoch": 11.91,
+      "learning_rate": 4.763636363636364e-05,
+      "loss": 1.1249,
+      "step": 131
+    },
+    {
+      "epoch": 12.0,
+      "learning_rate": 4.8e-05,
+      "loss": 1.0953,
+      "step": 132
+    },
+    {
+      "epoch": 12.09,
+      "learning_rate": 4.8363636363636364e-05,
+      "loss": 1.1209,
+      "step": 133
+    },
+    {
+      "epoch": 12.18,
+      "learning_rate": 4.872727272727273e-05,
+      "loss": 1.1054,
+      "step": 134
+    },
+    {
+      "epoch": 12.27,
+      "learning_rate": 4.909090909090909e-05,
+      "loss": 1.0896,
+      "step": 135
+    },
+    {
+      "epoch": 12.36,
+      "learning_rate": 4.945454545454546e-05,
+      "loss": 1.0754,
+      "step": 136
+    },
+    {
+      "epoch": 12.45,
+      "learning_rate": 4.981818181818182e-05,
+      "loss": 1.0641,
+      "step": 137
+    },
+    {
+      "epoch": 12.55,
+      "learning_rate": 5.018181818181819e-05,
+      "loss": 1.0827,
+      "step": 138
+    },
+    {
+      "epoch": 12.64,
+      "learning_rate": 5.054545454545455e-05,
+      "loss": 1.1269,
+      "step": 139
+    },
+    {
+      "epoch": 12.73,
+      "learning_rate": 5.090909090909091e-05,
+      "loss": 1.0666,
+      "step": 140
+    },
+    {
+      "epoch": 12.73,
+      "eval_accuracy_NAT": 0.9561173131717935,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8869868341988927,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.3516009185053667,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9451321483892734,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.975032795214252,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9454440385038502,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.60397101197788,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.27263931573897915,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.805971688307403,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8606970792130627,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.9632436037063599,
+      "eval_mean_accuracy": 0.31652846226765985,
+      "eval_mean_iou": 0.2683633179800904,
+      "eval_overall_accuracy": 0.9089568456013998,
+      "eval_runtime": 3.8233,
+      "eval_samples_per_second": 9.416,
+      "eval_steps_per_second": 0.523,
+      "step": 140
+    },
+    {
+      "epoch": 12.82,
+      "learning_rate": 5.127272727272727e-05,
+      "loss": 1.025,
+      "step": 141
+    },
+    {
+      "epoch": 12.91,
+      "learning_rate": 5.163636363636364e-05,
+      "loss": 1.057,
+      "step": 142
+    },
+    {
+      "epoch": 13.0,
+      "learning_rate": 5.2000000000000004e-05,
+      "loss": 1.0304,
+      "step": 143
+    },
+    {
+      "epoch": 13.09,
+      "learning_rate": 5.2363636363636374e-05,
+      "loss": 1.0345,
+      "step": 144
+    },
+    {
+      "epoch": 13.18,
+      "learning_rate": 5.272727272727272e-05,
+      "loss": 1.0249,
+      "step": 145
+    },
+    {
+      "epoch": 13.27,
+      "learning_rate": 5.309090909090909e-05,
+      "loss": 1.0423,
+      "step": 146
+    },
+    {
+      "epoch": 13.36,
+      "learning_rate": 5.3454545454545457e-05,
+      "loss": 0.961,
+      "step": 147
+    },
+    {
+      "epoch": 13.45,
+      "learning_rate": 5.3818181818181827e-05,
+      "loss": 1.0416,
+      "step": 148
+    },
+    {
+      "epoch": 13.55,
+      "learning_rate": 5.418181818181819e-05,
+      "loss": 1.0241,
+      "step": 149
+    },
+    {
+      "epoch": 13.64,
+      "learning_rate": 5.4545454545454546e-05,
+      "loss": 0.9978,
+      "step": 150
+    },
+    {
+      "epoch": 13.73,
+      "learning_rate": 5.490909090909091e-05,
+      "loss": 1.0115,
+      "step": 151
+    },
+    {
+      "epoch": 13.82,
+      "learning_rate": 5.527272727272727e-05,
+      "loss": 0.9345,
+      "step": 152
+    },
+    {
+      "epoch": 13.91,
+      "learning_rate": 5.563636363636364e-05,
+      "loss": 0.9835,
+      "step": 153
+    },
+    {
+      "epoch": 14.0,
+      "learning_rate": 5.6000000000000006e-05,
+      "loss": 0.9658,
+      "step": 154
+    },
+    {
+      "epoch": 14.09,
+      "learning_rate": 5.636363636363636e-05,
+      "loss": 0.8986,
+      "step": 155
+    },
+    {
+      "epoch": 14.18,
+      "learning_rate": 5.6727272727272726e-05,
+      "loss": 0.9494,
+      "step": 156
+    },
+    {
+      "epoch": 14.27,
+      "learning_rate": 5.7090909090909096e-05,
+      "loss": 0.9374,
+      "step": 157
+    },
+    {
+      "epoch": 14.36,
+      "learning_rate": 5.745454545454546e-05,
+      "loss": 0.9526,
+      "step": 158
+    },
+    {
+      "epoch": 14.45,
+      "learning_rate": 5.7818181818181815e-05,
+      "loss": 0.9791,
+      "step": 159
+    },
+    {
+      "epoch": 14.55,
+      "learning_rate": 5.818181818181818e-05,
+      "loss": 0.9858,
+      "step": 160
+    },
+    {
+      "epoch": 14.55,
+      "eval_accuracy_NAT": 0.9542294626781186,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9107794325936736,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.4873705512080722,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.8653860338796135,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9734096282163145,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9443239861289295,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.5657941221562315,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.35750086526259117,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8179847069984342,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8611517704146181,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.8469406962394714,
+      "eval_mean_accuracy": 0.3223980852750609,
+      "eval_mean_iou": 0.27282734238160034,
+      "eval_overall_accuracy": 0.9065458509657118,
+      "eval_runtime": 3.5282,
+      "eval_samples_per_second": 10.203,
+      "eval_steps_per_second": 0.567,
+      "step": 160
+    },
+    {
+      "epoch": 14.64,
+      "learning_rate": 5.854545454545455e-05,
+      "loss": 0.9323,
+      "step": 161
+    },
+    {
+      "epoch": 14.73,
+      "learning_rate": 5.890909090909091e-05,
+      "loss": 0.8957,
+      "step": 162
+    },
+    {
+      "epoch": 14.82,
+      "learning_rate": 5.927272727272728e-05,
+      "loss": 0.9079,
+      "step": 163
+    },
+    {
+      "epoch": 14.91,
+      "learning_rate": 5.963636363636363e-05,
+      "loss": 0.8696,
+      "step": 164
+    },
+    {
+      "epoch": 15.0,
+      "learning_rate": 6e-05,
+      "loss": 0.8968,
+      "step": 165
+    },
+    {
+      "epoch": 15.09,
+      "learning_rate": 6.0363636363636365e-05,
+      "loss": 0.8978,
+      "step": 166
+    },
+    {
+      "epoch": 15.18,
+      "learning_rate": 6.0727272727272735e-05,
+      "loss": 0.9059,
+      "step": 167
+    },
+    {
+      "epoch": 15.27,
+      "learning_rate": 6.10909090909091e-05,
+      "loss": 0.8761,
+      "step": 168
+    },
+    {
+      "epoch": 15.36,
+      "learning_rate": 6.145454545454545e-05,
+      "loss": 0.905,
+      "step": 169
+    },
+    {
+      "epoch": 15.45,
+      "learning_rate": 6.181818181818182e-05,
+      "loss": 0.8571,
+      "step": 170
+    },
+    {
+      "epoch": 15.55,
+      "learning_rate": 6.218181818181819e-05,
+      "loss": 0.8471,
+      "step": 171
+    },
+    {
+      "epoch": 15.64,
+      "learning_rate": 6.254545454545456e-05,
+      "loss": 0.8319,
+      "step": 172
+    },
+    {
+      "epoch": 15.73,
+      "learning_rate": 6.290909090909091e-05,
+      "loss": 0.861,
+      "step": 173
+    },
+    {
+      "epoch": 15.82,
+      "learning_rate": 6.327272727272727e-05,
+      "loss": 0.8161,
+      "step": 174
+    },
+    {
+      "epoch": 15.91,
+      "learning_rate": 6.363636363636364e-05,
+      "loss": 0.8789,
+      "step": 175
+    },
+    {
+      "epoch": 16.0,
+      "learning_rate": 6.400000000000001e-05,
+      "loss": 0.7966,
+      "step": 176
+    },
+    {
+      "epoch": 16.09,
+      "learning_rate": 6.436363636363637e-05,
+      "loss": 0.8516,
+      "step": 177
+    },
+    {
+      "epoch": 16.18,
+      "learning_rate": 6.472727272727272e-05,
+      "loss": 0.8139,
+      "step": 178
+    },
+    {
+      "epoch": 16.27,
+      "learning_rate": 6.50909090909091e-05,
+      "loss": 0.8312,
+      "step": 179
+    },
+    {
+      "epoch": 16.36,
+      "learning_rate": 6.545454545454546e-05,
+      "loss": 0.8171,
+      "step": 180
+    },
+    {
+      "epoch": 16.36,
+      "eval_accuracy_NAT": 0.9580538641735771,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8946829799562213,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7175526095967734,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9461273272735847,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9772442848229662,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9459735019378309,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.6848531827515401,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.48368507602636773,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8659028381838987,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8638598572414595,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.7246155738830566,
+      "eval_mean_accuracy": 0.34566623583254785,
+      "eval_mean_iou": 0.2957134197031613,
+      "eval_overall_accuracy": 0.9222983254326714,
+      "eval_runtime": 3.4993,
+      "eval_samples_per_second": 10.288,
+      "eval_steps_per_second": 0.572,
+      "step": 180
+    },
+    {
+      "epoch": 16.45,
+      "learning_rate": 6.581818181818182e-05,
+      "loss": 0.8185,
+      "step": 181
+    },
+    {
+      "epoch": 16.55,
+      "learning_rate": 6.618181818181819e-05,
+      "loss": 0.8349,
+      "step": 182
+    },
+    {
+      "epoch": 16.64,
+      "learning_rate": 6.654545454545455e-05,
+      "loss": 0.7596,
+      "step": 183
+    },
+    {
+      "epoch": 16.73,
+      "learning_rate": 6.690909090909092e-05,
+      "loss": 0.7586,
+      "step": 184
+    },
+    {
+      "epoch": 16.82,
+      "learning_rate": 6.727272727272727e-05,
+      "loss": 0.7928,
+      "step": 185
+    },
+    {
+      "epoch": 16.91,
+      "learning_rate": 6.763636363636364e-05,
+      "loss": 0.7646,
+      "step": 186
+    },
+    {
+      "epoch": 17.0,
+      "learning_rate": 6.800000000000001e-05,
+      "loss": 0.7323,
+      "step": 187
+    },
+    {
+      "epoch": 17.09,
+      "learning_rate": 6.836363636363637e-05,
+      "loss": 0.7875,
+      "step": 188
+    },
+    {
+      "epoch": 17.18,
+      "learning_rate": 6.872727272727273e-05,
+      "loss": 0.8039,
+      "step": 189
+    },
+    {
+      "epoch": 17.27,
+      "learning_rate": 6.90909090909091e-05,
+      "loss": 0.8027,
+      "step": 190
+    },
+    {
+      "epoch": 17.36,
+      "learning_rate": 6.945454545454547e-05,
+      "loss": 0.7244,
+      "step": 191
+    },
+    {
+      "epoch": 17.45,
+      "learning_rate": 6.981818181818182e-05,
+      "loss": 0.7035,
+      "step": 192
+    },
+    {
+      "epoch": 17.55,
+      "learning_rate": 7.018181818181818e-05,
+      "loss": 0.6842,
+      "step": 193
+    },
+    {
+      "epoch": 17.64,
+      "learning_rate": 7.054545454545455e-05,
+      "loss": 0.6885,
+      "step": 194
+    },
+    {
+      "epoch": 17.73,
+      "learning_rate": 7.090909090909092e-05,
+      "loss": 0.7147,
+      "step": 195
+    },
+    {
+      "epoch": 17.82,
+      "learning_rate": 7.127272727272728e-05,
+      "loss": 0.7371,
+      "step": 196
+    },
+    {
+      "epoch": 17.91,
+      "learning_rate": 7.163636363636363e-05,
+      "loss": 0.6814,
+      "step": 197
+    },
+    {
+      "epoch": 18.0,
+      "learning_rate": 7.2e-05,
+      "loss": 0.6467,
+      "step": 198
+    },
+    {
+      "epoch": 18.09,
+      "learning_rate": 7.236363636363637e-05,
+      "loss": 0.639,
+      "step": 199
+    },
+    {
+      "epoch": 18.18,
+      "learning_rate": 7.272727272727273e-05,
+      "loss": 0.7362,
+      "step": 200
+    },
+    {
+      "epoch": 18.18,
+      "eval_accuracy_NAT": 0.9589253562371588,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8930412785956479,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7201005183194288,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9467536974583639,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9858386222027523,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9477355622967787,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.6862279892606659,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.5313936273928227,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.861003861003861,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8618114030437445,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.6129858493804932,
+      "eval_mean_accuracy": 0.3465122671394886,
+      "eval_mean_iou": 0.29909018792291325,
+      "eval_overall_accuracy": 0.9243333604600694,
+      "eval_runtime": 3.4286,
+      "eval_samples_per_second": 10.5,
+      "eval_steps_per_second": 0.583,
+      "step": 200
+    },
+    {
+      "epoch": 18.27,
+      "learning_rate": 7.30909090909091e-05,
+      "loss": 0.7156,
+      "step": 201
+    },
+    {
+      "epoch": 18.36,
+      "learning_rate": 7.345454545454545e-05,
+      "loss": 0.6647,
+      "step": 202
+    },
+    {
+      "epoch": 18.45,
+      "learning_rate": 7.381818181818182e-05,
+      "loss": 0.7283,
+      "step": 203
+    },
+    {
+      "epoch": 18.55,
+      "learning_rate": 7.418181818181818e-05,
+      "loss": 0.6373,
+      "step": 204
+    },
+    {
+      "epoch": 18.64,
+      "learning_rate": 7.454545454545455e-05,
+      "loss": 0.661,
+      "step": 205
+    },
+    {
+      "epoch": 18.73,
+      "learning_rate": 7.490909090909092e-05,
+      "loss": 0.6538,
+      "step": 206
+    },
+    {
+      "epoch": 18.82,
+      "learning_rate": 7.527272727272728e-05,
+      "loss": 0.6686,
+      "step": 207
+    },
+    {
+      "epoch": 18.91,
+      "learning_rate": 7.563636363636363e-05,
+      "loss": 0.6689,
+      "step": 208
+    },
+    {
+      "epoch": 19.0,
+      "learning_rate": 7.6e-05,
+      "loss": 0.5993,
+      "step": 209
+    },
+    {
+      "epoch": 19.09,
+      "learning_rate": 7.636363636363637e-05,
+      "loss": 0.6439,
+      "step": 210
+    },
+    {
+      "epoch": 19.18,
+      "learning_rate": 7.672727272727273e-05,
+      "loss": 0.6698,
+      "step": 211
+    },
+    {
+      "epoch": 19.27,
+      "learning_rate": 7.709090909090909e-05,
+      "loss": 0.6703,
+      "step": 212
+    },
+    {
+      "epoch": 19.36,
+      "learning_rate": 7.745454545454546e-05,
+      "loss": 0.6178,
+      "step": 213
+    },
+    {
+      "epoch": 19.45,
+      "learning_rate": 7.781818181818183e-05,
+      "loss": 0.6013,
+      "step": 214
+    },
+    {
+      "epoch": 19.55,
+      "learning_rate": 7.818181818181818e-05,
+      "loss": 0.6247,
+      "step": 215
+    },
+    {
+      "epoch": 19.64,
+      "learning_rate": 7.854545454545454e-05,
+      "loss": 0.624,
+      "step": 216
+    },
+    {
+      "epoch": 19.73,
+      "learning_rate": 7.890909090909091e-05,
+      "loss": 0.5774,
+      "step": 217
+    },
+    {
+      "epoch": 19.82,
+      "learning_rate": 7.927272727272728e-05,
+      "loss": 0.6187,
+      "step": 218
+    },
+    {
+      "epoch": 19.91,
+      "learning_rate": 7.963636363636364e-05,
+      "loss": 0.6401,
+      "step": 219
+    },
+    {
+      "epoch": 20.0,
+      "learning_rate": 8e-05,
+      "loss": 0.608,
+      "step": 220
+    },
+    {
+      "epoch": 20.0,
+      "eval_accuracy_NAT": 0.9587116739352763,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8941330636508005,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7906241153094713,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9608881397168275,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9820896495045418,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9479116908257735,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7102665429327287,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.5579340180148183,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8665623415357604,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8661181583018343,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.5255590081214905,
+      "eval_mean_accuracy": 0.3528035878551475,
+      "eval_mean_iou": 0.3037532885854551,
+      "eval_overall_accuracy": 0.9270096884833442,
+      "eval_runtime": 3.8134,
+      "eval_samples_per_second": 9.44,
+      "eval_steps_per_second": 0.524,
+      "step": 220
+    },
+    {
+      "epoch": 20.09,
+      "learning_rate": 8.036363636363636e-05,
+      "loss": 0.5913,
+      "step": 221
+    },
+    {
+      "epoch": 20.18,
+      "learning_rate": 8.072727272727273e-05,
+      "loss": 0.6022,
+      "step": 222
+    },
+    {
+      "epoch": 20.27,
+      "learning_rate": 8.109090909090909e-05,
+      "loss": 0.5652,
+      "step": 223
+    },
+    {
+      "epoch": 20.36,
+      "learning_rate": 8.145454545454546e-05,
+      "loss": 0.5683,
+      "step": 224
+    },
+    {
+      "epoch": 20.45,
+      "learning_rate": 8.181818181818183e-05,
+      "loss": 0.6305,
+      "step": 225
+    },
+    {
+      "epoch": 20.55,
+      "learning_rate": 8.218181818181819e-05,
+      "loss": 0.6003,
+      "step": 226
+    },
+    {
+      "epoch": 20.64,
+      "learning_rate": 8.254545454545454e-05,
+      "loss": 0.5246,
+      "step": 227
+    },
+    {
+      "epoch": 20.73,
+      "learning_rate": 8.290909090909091e-05,
+      "loss": 0.5921,
+      "step": 228
+    },
+    {
+      "epoch": 20.82,
+      "learning_rate": 8.327272727272728e-05,
+      "loss": 0.603,
+      "step": 229
+    },
+    {
+      "epoch": 20.91,
+      "learning_rate": 8.363636363636364e-05,
+      "loss": 0.5426,
+      "step": 230
+    },
+    {
+      "epoch": 21.0,
+      "learning_rate": 8.4e-05,
+      "loss": 0.5409,
+      "step": 231
+    },
+    {
+      "epoch": 21.09,
+      "learning_rate": 8.436363636363637e-05,
+      "loss": 0.4839,
+      "step": 232
+    },
+    {
+      "epoch": 21.18,
+      "learning_rate": 8.472727272727274e-05,
+      "loss": 0.5619,
+      "step": 233
+    },
+    {
+      "epoch": 21.27,
+      "learning_rate": 8.50909090909091e-05,
+      "loss": 0.5996,
+      "step": 234
+    },
+    {
+      "epoch": 21.36,
+      "learning_rate": 8.545454545454545e-05,
+      "loss": 0.5703,
+      "step": 235
+    },
+    {
+      "epoch": 21.45,
+      "learning_rate": 8.581818181818182e-05,
+      "loss": 0.5253,
+      "step": 236
+    },
+    {
+      "epoch": 21.55,
+      "learning_rate": 8.618181818181819e-05,
+      "loss": 0.5508,
+      "step": 237
+    },
+    {
+      "epoch": 21.64,
+      "learning_rate": 8.654545454545456e-05,
+      "loss": 0.5668,
+      "step": 238
+    },
+    {
+      "epoch": 21.73,
+      "learning_rate": 8.690909090909091e-05,
+      "loss": 0.5009,
+      "step": 239
+    },
+    {
+      "epoch": 21.82,
+      "learning_rate": 8.727272727272727e-05,
+      "loss": 0.5261,
+      "step": 240
+    },
+    {
+      "epoch": 21.82,
+      "eval_accuracy_NAT": 0.9583323040351547,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9321604468002919,
+      "eval_accuracy_concrete tiles": 0.0,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7454677631877868,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9518480943473114,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9563380145710917,
+      "eval_accuracy_steel": 0.0,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.948486096354183,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.6478410704181764,
+      "eval_iou_concrete tiles": 0.0,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.5856762745792591,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8490727427463814,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.871423285203022,
+      "eval_iou_steel": 0.0,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.47437721490859985,
+      "eval_mean_accuracy": 0.3495497402262797,
+      "eval_mean_iou": 0.3001922668693094,
+      "eval_overall_accuracy": 0.9231026967366537,
+      "eval_runtime": 3.2633,
+      "eval_samples_per_second": 11.032,
+      "eval_steps_per_second": 0.613,
+      "step": 240
+    },
+    {
+      "epoch": 21.91,
+      "learning_rate": 8.763636363636364e-05,
+      "loss": 0.5156,
+      "step": 241
+    },
+    {
+      "epoch": 22.0,
+      "learning_rate": 8.800000000000001e-05,
+      "loss": 0.5294,
+      "step": 242
+    },
+    {
+      "epoch": 22.09,
+      "learning_rate": 8.836363636363637e-05,
+      "loss": 0.505,
+      "step": 243
+    },
+    {
+      "epoch": 22.18,
+      "learning_rate": 8.872727272727274e-05,
+      "loss": 0.5347,
+      "step": 244
+    },
+    {
+      "epoch": 22.27,
+      "learning_rate": 8.90909090909091e-05,
+      "loss": 0.5135,
+      "step": 245
+    },
+    {
+      "epoch": 22.36,
+      "learning_rate": 8.945454545454546e-05,
+      "loss": 0.4949,
+      "step": 246
+    },
+    {
+      "epoch": 22.45,
+      "learning_rate": 8.981818181818182e-05,
+      "loss": 0.5015,
+      "step": 247
+    },
+    {
+      "epoch": 22.55,
+      "learning_rate": 9.018181818181819e-05,
+      "loss": 0.4432,
+      "step": 248
+    },
+    {
+      "epoch": 22.64,
+      "learning_rate": 9.054545454545455e-05,
+      "loss": 0.5089,
+      "step": 249
+    },
+    {
+      "epoch": 22.73,
+      "learning_rate": 9.090909090909092e-05,
+      "loss": 0.5342,
+      "step": 250
+    },
+    {
+      "epoch": 22.82,
+      "learning_rate": 9.127272727272727e-05,
+      "loss": 0.4953,
+      "step": 251
+    },
+    {
+      "epoch": 22.91,
+      "learning_rate": 9.163636363636364e-05,
+      "loss": 0.5519,
+      "step": 252
+    },
+    {
+      "epoch": 23.0,
+      "learning_rate": 9.200000000000001e-05,
+      "loss": 0.4735,
+      "step": 253
+    },
+    {
+      "epoch": 23.09,
+      "learning_rate": 9.236363636363636e-05,
+      "loss": 0.4638,
+      "step": 254
+    },
+    {
+      "epoch": 23.18,
+      "learning_rate": 9.272727272727273e-05,
+      "loss": 0.488,
+      "step": 255
+    },
+    {
+      "epoch": 23.27,
+      "learning_rate": 9.30909090909091e-05,
+      "loss": 0.4371,
+      "step": 256
+    },
+    {
+      "epoch": 23.36,
+      "learning_rate": 9.345454545454547e-05,
+      "loss": 0.4664,
+      "step": 257
+    },
+    {
+      "epoch": 23.45,
+      "learning_rate": 9.381818181818182e-05,
+      "loss": 0.4603,
+      "step": 258
+    },
+    {
+      "epoch": 23.55,
+      "learning_rate": 9.418181818181818e-05,
+      "loss": 0.4394,
+      "step": 259
+    },
+    {
+      "epoch": 23.64,
+      "learning_rate": 9.454545454545455e-05,
+      "loss": 0.4777,
+      "step": 260
+    },
+    {
+      "epoch": 23.64,
+      "eval_accuracy_NAT": 0.9674894111685087,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8888887398600799,
+      "eval_accuracy_concrete tiles": 0.10248883996820156,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7898062680651622,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9544164539466763,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9888592816792116,
+      "eval_accuracy_steel": 0.0009646461315216675,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9545549900462824,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.725248497197943,
+      "eval_iou_concrete tiles": 0.10248883996820156,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.6226222227120437,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8777720566238544,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8638473573125913,
+      "eval_iou_steel": 0.0009646461315216675,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.4128577411174774,
+      "eval_mean_accuracy": 0.36099335698610474,
+      "eval_mean_iou": 0.31903835461480295,
+      "eval_overall_accuracy": 0.9328311284383138,
+      "eval_runtime": 4.1127,
+      "eval_samples_per_second": 8.753,
+      "eval_steps_per_second": 0.486,
+      "step": 260
+    },
+    {
+      "epoch": 23.73,
+      "learning_rate": 9.490909090909092e-05,
+      "loss": 0.4488,
+      "step": 261
+    },
+    {
+      "epoch": 23.82,
+      "learning_rate": 9.527272727272728e-05,
+      "loss": 0.4711,
+      "step": 262
+    },
+    {
+      "epoch": 23.91,
+      "learning_rate": 9.563636363636365e-05,
+      "loss": 0.4346,
+      "step": 263
+    },
+    {
+      "epoch": 24.0,
+      "learning_rate": 9.6e-05,
+      "loss": 0.5418,
+      "step": 264
+    },
+    {
+      "epoch": 24.09,
+      "learning_rate": 9.636363636363637e-05,
+      "loss": 0.4004,
+      "step": 265
+    },
+    {
+      "epoch": 24.18,
+      "learning_rate": 9.672727272727273e-05,
+      "loss": 0.5282,
+      "step": 266
+    },
+    {
+      "epoch": 24.27,
+      "learning_rate": 9.70909090909091e-05,
+      "loss": 0.4103,
+      "step": 267
+    },
+    {
+      "epoch": 24.36,
+      "learning_rate": 9.745454545454546e-05,
+      "loss": 0.4233,
+      "step": 268
+    },
+    {
+      "epoch": 24.45,
+      "learning_rate": 9.781818181818183e-05,
+      "loss": 0.4435,
+      "step": 269
+    },
+    {
+      "epoch": 24.55,
+      "learning_rate": 9.818181818181818e-05,
+      "loss": 0.4471,
+      "step": 270
+    },
+    {
+      "epoch": 24.64,
+      "learning_rate": 9.854545454545455e-05,
+      "loss": 0.4428,
+      "step": 271
+    },
+    {
+      "epoch": 24.73,
+      "learning_rate": 9.890909090909092e-05,
+      "loss": 0.4204,
+      "step": 272
+    },
+    {
+      "epoch": 24.82,
+      "learning_rate": 9.927272727272728e-05,
+      "loss": 0.4329,
+      "step": 273
+    },
+    {
+      "epoch": 24.91,
+      "learning_rate": 9.963636363636363e-05,
+      "loss": 0.4094,
+      "step": 274
+    },
+    {
+      "epoch": 25.0,
+      "learning_rate": 0.0001,
+      "loss": 0.3999,
+      "step": 275
+    },
+    {
+      "epoch": 25.09,
+      "learning_rate": 9.998086124401914e-05,
+      "loss": 0.3871,
+      "step": 276
+    },
+    {
+      "epoch": 25.18,
+      "learning_rate": 9.996172248803828e-05,
+      "loss": 0.4378,
+      "step": 277
+    },
+    {
+      "epoch": 25.27,
+      "learning_rate": 9.994258373205743e-05,
+      "loss": 0.3492,
+      "step": 278
+    },
+    {
+      "epoch": 25.36,
+      "learning_rate": 9.992344497607656e-05,
+      "loss": 0.408,
+      "step": 279
+    },
+    {
+      "epoch": 25.45,
+      "learning_rate": 9.99043062200957e-05,
+      "loss": 0.3775,
+      "step": 280
+    },
+    {
+      "epoch": 25.45,
+      "eval_accuracy_NAT": 0.9729683947818464,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8722732198806816,
+      "eval_accuracy_concrete tiles": 0.7616339509570109,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.8074948360286177,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9740443320148926,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.986779334972189,
+      "eval_accuracy_steel": 0.014296550359474969,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9569670815303015,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7644293101036613,
+      "eval_iou_concrete tiles": 0.7351335399144164,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.6186385763477944,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.8826613018104198,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8804635809724196,
+      "eval_iou_steel": 0.014296550359474969,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.3502545952796936,
+      "eval_mean_accuracy": 0.41457620146113183,
+      "eval_mean_iou": 0.37327614931065295,
+      "eval_overall_accuracy": 0.9391554726494683,
+      "eval_runtime": 3.892,
+      "eval_samples_per_second": 9.25,
+      "eval_steps_per_second": 0.514,
+      "step": 280
+    },
+    {
+      "epoch": 25.55,
+      "learning_rate": 9.988516746411484e-05,
+      "loss": 0.4176,
+      "step": 281
+    },
+    {
+      "epoch": 25.64,
+      "learning_rate": 9.986602870813397e-05,
+      "loss": 0.3613,
+      "step": 282
+    },
+    {
+      "epoch": 25.73,
+      "learning_rate": 9.984688995215312e-05,
+      "loss": 0.4248,
+      "step": 283
+    },
+    {
+      "epoch": 25.82,
+      "learning_rate": 9.982775119617226e-05,
+      "loss": 0.4028,
+      "step": 284
+    },
+    {
+      "epoch": 25.91,
+      "learning_rate": 9.980861244019139e-05,
+      "loss": 0.414,
+      "step": 285
+    },
+    {
+      "epoch": 26.0,
+      "learning_rate": 9.978947368421054e-05,
+      "loss": 0.3797,
+      "step": 286
+    },
+    {
+      "epoch": 26.09,
+      "learning_rate": 9.977033492822967e-05,
+      "loss": 0.3478,
+      "step": 287
+    },
+    {
+      "epoch": 26.18,
+      "learning_rate": 9.97511961722488e-05,
+      "loss": 0.4247,
+      "step": 288
+    },
+    {
+      "epoch": 26.27,
+      "learning_rate": 9.973205741626795e-05,
+      "loss": 0.3849,
+      "step": 289
+    },
+    {
+      "epoch": 26.36,
+      "learning_rate": 9.97129186602871e-05,
+      "loss": 0.3291,
+      "step": 290
+    },
+    {
+      "epoch": 26.45,
+      "learning_rate": 9.969377990430623e-05,
+      "loss": 0.3979,
+      "step": 291
+    },
+    {
+      "epoch": 26.55,
+      "learning_rate": 9.967464114832537e-05,
+      "loss": 0.3622,
+      "step": 292
+    },
+    {
+      "epoch": 26.64,
+      "learning_rate": 9.96555023923445e-05,
+      "loss": 0.3409,
+      "step": 293
+    },
+    {
+      "epoch": 26.73,
+      "learning_rate": 9.963636363636363e-05,
+      "loss": 0.3457,
+      "step": 294
+    },
+    {
+      "epoch": 26.82,
+      "learning_rate": 9.961722488038278e-05,
+      "loss": 0.3679,
+      "step": 295
+    },
+    {
+      "epoch": 26.91,
+      "learning_rate": 9.959808612440193e-05,
+      "loss": 0.3988,
+      "step": 296
+    },
+    {
+      "epoch": 27.0,
+      "learning_rate": 9.957894736842106e-05,
+      "loss": 0.3584,
+      "step": 297
+    },
+    {
+      "epoch": 27.09,
+      "learning_rate": 9.95598086124402e-05,
+      "loss": 0.3431,
+      "step": 298
+    },
+    {
+      "epoch": 27.18,
+      "learning_rate": 9.954066985645934e-05,
+      "loss": 0.3898,
+      "step": 299
+    },
+    {
+      "epoch": 27.27,
+      "learning_rate": 9.952153110047847e-05,
+      "loss": 0.3272,
+      "step": 300
+    },
+    {
+      "epoch": 27.27,
+      "eval_accuracy_NAT": 0.9755313472466215,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8942215867633804,
+      "eval_accuracy_concrete tiles": 0.7856356631810677,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7729774882303393,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9490753784521401,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9889141917895506,
+      "eval_accuracy_steel": 0.12353241870589012,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9595498328439508,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7432702405611682,
+      "eval_iou_concrete tiles": 0.6156998059089929,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.6411552053527684,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9023240039777659,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.8827876336194475,
+      "eval_iou_steel": 0.12300000820930443,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.31202512979507446,
+      "eval_mean_accuracy": 0.4222990826437684,
+      "eval_mean_iou": 0.3744451331133383,
+      "eval_overall_accuracy": 0.941056039598253,
+      "eval_runtime": 3.7112,
+      "eval_samples_per_second": 9.7,
+      "eval_steps_per_second": 0.539,
+      "step": 300
+    },
+    {
+      "epoch": 27.36,
+      "learning_rate": 9.950239234449761e-05,
+      "loss": 0.3692,
+      "step": 301
+    },
+    {
+      "epoch": 27.45,
+      "learning_rate": 9.948325358851676e-05,
+      "loss": 0.312,
+      "step": 302
+    },
+    {
+      "epoch": 27.55,
+      "learning_rate": 9.946411483253589e-05,
+      "loss": 0.3441,
+      "step": 303
+    },
+    {
+      "epoch": 27.64,
+      "learning_rate": 9.944497607655504e-05,
+      "loss": 0.3033,
+      "step": 304
+    },
+    {
+      "epoch": 27.73,
+      "learning_rate": 9.942583732057417e-05,
+      "loss": 0.3007,
+      "step": 305
+    },
+    {
+      "epoch": 27.82,
+      "learning_rate": 9.94066985645933e-05,
+      "loss": 0.3405,
+      "step": 306
+    },
+    {
+      "epoch": 27.91,
+      "learning_rate": 9.938755980861245e-05,
+      "loss": 0.3206,
+      "step": 307
+    },
+    {
+      "epoch": 28.0,
+      "learning_rate": 9.936842105263159e-05,
+      "loss": 0.3518,
+      "step": 308
+    },
+    {
+      "epoch": 28.09,
+      "learning_rate": 9.934928229665072e-05,
+      "loss": 0.3508,
+      "step": 309
+    },
+    {
+      "epoch": 28.18,
+      "learning_rate": 9.933014354066987e-05,
+      "loss": 0.2985,
+      "step": 310
+    },
+    {
+      "epoch": 28.27,
+      "learning_rate": 9.9311004784689e-05,
+      "loss": 0.3159,
+      "step": 311
+    },
+    {
+      "epoch": 28.36,
+      "learning_rate": 9.929186602870813e-05,
+      "loss": 0.3215,
+      "step": 312
+    },
+    {
+      "epoch": 28.45,
+      "learning_rate": 9.927272727272728e-05,
+      "loss": 0.3105,
+      "step": 313
+    },
+    {
+      "epoch": 28.55,
+      "learning_rate": 9.925358851674642e-05,
+      "loss": 0.3098,
+      "step": 314
+    },
+    {
+      "epoch": 28.64,
+      "learning_rate": 9.923444976076556e-05,
+      "loss": 0.3491,
+      "step": 315
+    },
+    {
+      "epoch": 28.73,
+      "learning_rate": 9.921531100478469e-05,
+      "loss": 0.3527,
+      "step": 316
+    },
+    {
+      "epoch": 28.82,
+      "learning_rate": 9.919617224880383e-05,
+      "loss": 0.2871,
+      "step": 317
+    },
+    {
+      "epoch": 28.91,
+      "learning_rate": 9.917703349282297e-05,
+      "loss": 0.3123,
+      "step": 318
+    },
+    {
+      "epoch": 29.0,
+      "learning_rate": 9.915789473684211e-05,
+      "loss": 0.3469,
+      "step": 319
+    },
+    {
+      "epoch": 29.09,
+      "learning_rate": 9.913875598086126e-05,
+      "loss": 0.2718,
+      "step": 320
+    },
+    {
+      "epoch": 29.09,
+      "eval_accuracy_NAT": 0.9825863922310691,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8046442443881712,
+      "eval_accuracy_concrete tiles": 0.8758637558857701,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.7965098194095422,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9697637326826177,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0,
+      "eval_accuracy_speedway bricks": 0.9828523569712307,
+      "eval_accuracy_steel": 0.39059923487896575,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9586311515822584,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7106968754627573,
+      "eval_iou_concrete tiles": 0.30282784502352134,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.644390092178929,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9171158004759503,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0,
+      "eval_iou_speedway bricks": 0.9192782290288645,
+      "eval_iou_steel": 0.3835287069719245,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.28215399384498596,
+      "eval_mean_accuracy": 0.4463707335728744,
+      "eval_mean_iou": 0.3720360539018619,
+      "eval_overall_accuracy": 0.9434228473239474,
+      "eval_runtime": 4.2101,
+      "eval_samples_per_second": 8.551,
+      "eval_steps_per_second": 0.475,
+      "step": 320
+    },
+    {
+      "epoch": 29.18,
+      "learning_rate": 9.911961722488039e-05,
+      "loss": 0.3449,
+      "step": 321
+    },
+    {
+      "epoch": 29.27,
+      "learning_rate": 9.910047846889952e-05,
+      "loss": 0.2981,
+      "step": 322
+    },
+    {
+      "epoch": 29.36,
+      "learning_rate": 9.908133971291867e-05,
+      "loss": 0.3002,
+      "step": 323
+    },
+    {
+      "epoch": 29.45,
+      "learning_rate": 9.90622009569378e-05,
+      "loss": 0.3164,
+      "step": 324
+    },
+    {
+      "epoch": 29.55,
+      "learning_rate": 9.904306220095694e-05,
+      "loss": 0.2772,
+      "step": 325
+    },
+    {
+      "epoch": 29.64,
+      "learning_rate": 9.902392344497609e-05,
+      "loss": 0.2679,
+      "step": 326
+    },
+    {
+      "epoch": 29.73,
+      "learning_rate": 9.900478468899522e-05,
+      "loss": 0.2789,
+      "step": 327
+    },
+    {
+      "epoch": 29.82,
+      "learning_rate": 9.898564593301435e-05,
+      "loss": 0.3047,
+      "step": 328
+    },
+    {
+      "epoch": 29.91,
+      "learning_rate": 9.89665071770335e-05,
+      "loss": 0.3267,
+      "step": 329
+    },
+    {
+      "epoch": 30.0,
+      "learning_rate": 9.894736842105263e-05,
+      "loss": 0.3023,
+      "step": 330
+    },
+    {
+      "epoch": 30.09,
+      "learning_rate": 9.892822966507178e-05,
+      "loss": 0.2884,
+      "step": 331
+    },
+    {
+      "epoch": 30.18,
+      "learning_rate": 9.890909090909092e-05,
+      "loss": 0.291,
+      "step": 332
+    },
+    {
+      "epoch": 30.27,
+      "learning_rate": 9.888995215311005e-05,
+      "loss": 0.2564,
+      "step": 333
+    },
+    {
+      "epoch": 30.36,
+      "learning_rate": 9.887081339712919e-05,
+      "loss": 0.2616,
+      "step": 334
+    },
+    {
+      "epoch": 30.45,
+      "learning_rate": 9.885167464114833e-05,
+      "loss": 0.234,
+      "step": 335
+    },
+    {
+      "epoch": 30.55,
+      "learning_rate": 9.883253588516746e-05,
+      "loss": 0.2786,
+      "step": 336
+    },
+    {
+      "epoch": 30.64,
+      "learning_rate": 9.881339712918661e-05,
+      "loss": 0.2706,
+      "step": 337
+    },
+    {
+      "epoch": 30.73,
+      "learning_rate": 9.879425837320575e-05,
+      "loss": 0.2773,
+      "step": 338
+    },
+    {
+      "epoch": 30.82,
+      "learning_rate": 9.877511961722489e-05,
+      "loss": 0.2576,
+      "step": 339
+    },
+    {
+      "epoch": 30.91,
+      "learning_rate": 9.875598086124402e-05,
+      "loss": 0.286,
+      "step": 340
+    },
+    {
+      "epoch": 30.91,
+      "eval_accuracy_NAT": 0.977698166955223,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9108478368170307,
+      "eval_accuracy_concrete tiles": 0.8354736134042683,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.8607387886773593,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9504163910678161,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.0005789430154660492,
+      "eval_accuracy_speedway bricks": 0.9874183438386632,
+      "eval_accuracy_steel": 0.47383088186795064,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9621091350951422,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7653984394459729,
+      "eval_iou_concrete tiles": 0.8104701171585348,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.6783810659754455,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9239618416180283,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.0005789430154660492,
+      "eval_iou_speedway bricks": 0.9187664611554629,
+      "eval_iou_steel": 0.45589763523429505,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.23237627744674683,
+      "eval_mean_accuracy": 0.4613079204341367,
+      "eval_mean_iou": 0.424274126053719,
+      "eval_overall_accuracy": 0.950863414340549,
+      "eval_runtime": 3.4041,
+      "eval_samples_per_second": 10.576,
+      "eval_steps_per_second": 0.588,
+      "step": 340
+    },
+    {
+      "epoch": 31.0,
+      "learning_rate": 9.873684210526316e-05,
+      "loss": 0.2626,
+      "step": 341
+    },
+    {
+      "epoch": 31.09,
+      "learning_rate": 9.87177033492823e-05,
+      "loss": 0.2658,
+      "step": 342
+    },
+    {
+      "epoch": 31.18,
+      "learning_rate": 9.869856459330143e-05,
+      "loss": 0.2426,
+      "step": 343
+    },
+    {
+      "epoch": 31.27,
+      "learning_rate": 9.867942583732059e-05,
+      "loss": 0.2669,
+      "step": 344
+    },
+    {
+      "epoch": 31.36,
+      "learning_rate": 9.866028708133972e-05,
+      "loss": 0.2462,
+      "step": 345
+    },
+    {
+      "epoch": 31.45,
+      "learning_rate": 9.864114832535885e-05,
+      "loss": 0.2497,
+      "step": 346
+    },
+    {
+      "epoch": 31.55,
+      "learning_rate": 9.8622009569378e-05,
+      "loss": 0.2362,
+      "step": 347
+    },
+    {
+      "epoch": 31.64,
+      "learning_rate": 9.860287081339713e-05,
+      "loss": 0.2559,
+      "step": 348
+    },
+    {
+      "epoch": 31.73,
+      "learning_rate": 9.858373205741627e-05,
+      "loss": 0.2661,
+      "step": 349
+    },
+    {
+      "epoch": 31.82,
+      "learning_rate": 9.856459330143542e-05,
+      "loss": 0.2389,
+      "step": 350
+    },
+    {
+      "epoch": 31.91,
+      "learning_rate": 9.854545454545455e-05,
+      "loss": 0.2693,
+      "step": 351
+    },
+    {
+      "epoch": 32.0,
+      "learning_rate": 9.852631578947368e-05,
+      "loss": 0.2688,
+      "step": 352
+    },
+    {
+      "epoch": 32.09,
+      "learning_rate": 9.850717703349283e-05,
+      "loss": 0.2313,
+      "step": 353
+    },
+    {
+      "epoch": 32.18,
+      "learning_rate": 9.848803827751196e-05,
+      "loss": 0.1979,
+      "step": 354
+    },
+    {
+      "epoch": 32.27,
+      "learning_rate": 9.846889952153111e-05,
+      "loss": 0.2052,
+      "step": 355
+    },
+    {
+      "epoch": 32.36,
+      "learning_rate": 9.844976076555025e-05,
+      "loss": 0.2669,
+      "step": 356
+    },
+    {
+      "epoch": 32.45,
+      "learning_rate": 9.843062200956938e-05,
+      "loss": 0.2743,
+      "step": 357
+    },
+    {
+      "epoch": 32.55,
+      "learning_rate": 9.841148325358852e-05,
+      "loss": 0.2396,
+      "step": 358
+    },
+    {
+      "epoch": 32.64,
+      "learning_rate": 9.839234449760766e-05,
+      "loss": 0.2322,
+      "step": 359
+    },
+    {
+      "epoch": 32.73,
+      "learning_rate": 9.83732057416268e-05,
+      "loss": 0.2553,
+      "step": 360
+    },
+    {
+      "epoch": 32.73,
+      "eval_accuracy_NAT": 0.986369504165305,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9002438409373793,
+      "eval_accuracy_concrete tiles": 0.9165902280927047,
+      "eval_accuracy_dark bricks": 0.0,
+      "eval_accuracy_grass": 0.8603193798341238,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9630828537812566,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.026521103851349488,
+      "eval_accuracy_speedway bricks": 0.9764043411571189,
+      "eval_accuracy_steel": 0.6370292197084625,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.964050345451529,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.802261703556906,
+      "eval_iou_concrete tiles": 0.8505603631720812,
+      "eval_iou_dark bricks": 0.0,
+      "eval_iou_grass": 0.6887008099827931,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9318697956093869,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.026312910284463895,
+      "eval_iou_speedway bricks": 0.9343460443813649,
+      "eval_iou_steel": 0.5962418489794343,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.21057012677192688,
+      "eval_mean_accuracy": 0.4820431131944385,
+      "eval_mean_iou": 0.4457187554936892,
+      "eval_overall_accuracy": 0.9568744235568576,
+      "eval_runtime": 3.4643,
+      "eval_samples_per_second": 10.392,
+      "eval_steps_per_second": 0.577,
+      "step": 360
+    },
+    {
+      "epoch": 32.82,
+      "learning_rate": 9.835406698564594e-05,
+      "loss": 0.2174,
+      "step": 361
+    },
+    {
+      "epoch": 32.91,
+      "learning_rate": 9.833492822966509e-05,
+      "loss": 0.3019,
+      "step": 362
+    },
+    {
+      "epoch": 33.0,
+      "learning_rate": 9.831578947368422e-05,
+      "loss": 0.2601,
+      "step": 363
+    },
+    {
+      "epoch": 33.09,
+      "learning_rate": 9.829665071770335e-05,
+      "loss": 0.2548,
+      "step": 364
+    },
+    {
+      "epoch": 33.18,
+      "learning_rate": 9.82775119617225e-05,
+      "loss": 0.2142,
+      "step": 365
+    },
+    {
+      "epoch": 33.27,
+      "learning_rate": 9.825837320574163e-05,
+      "loss": 0.2233,
+      "step": 366
+    },
+    {
+      "epoch": 33.36,
+      "learning_rate": 9.823923444976077e-05,
+      "loss": 0.2116,
+      "step": 367
+    },
+    {
+      "epoch": 33.45,
+      "learning_rate": 9.822009569377992e-05,
+      "loss": 0.2108,
+      "step": 368
+    },
+    {
+      "epoch": 33.55,
+      "learning_rate": 9.820095693779905e-05,
+      "loss": 0.1959,
+      "step": 369
+    },
+    {
+      "epoch": 33.64,
+      "learning_rate": 9.818181818181818e-05,
+      "loss": 0.263,
+      "step": 370
+    },
+    {
+      "epoch": 33.73,
+      "learning_rate": 9.816267942583733e-05,
+      "loss": 0.2373,
+      "step": 371
+    },
+    {
+      "epoch": 33.82,
+      "learning_rate": 9.814354066985646e-05,
+      "loss": 0.1877,
+      "step": 372
+    },
+    {
+      "epoch": 33.91,
+      "learning_rate": 9.81244019138756e-05,
+      "loss": 0.2542,
+      "step": 373
+    },
+    {
+      "epoch": 34.0,
+      "learning_rate": 9.810526315789475e-05,
+      "loss": 0.2463,
+      "step": 374
+    },
+    {
+      "epoch": 34.09,
+      "learning_rate": 9.808612440191388e-05,
+      "loss": 0.2061,
+      "step": 375
+    },
+    {
+      "epoch": 34.18,
+      "learning_rate": 9.806698564593301e-05,
+      "loss": 0.2288,
+      "step": 376
+    },
+    {
+      "epoch": 34.27,
+      "learning_rate": 9.804784688995216e-05,
+      "loss": 0.2206,
+      "step": 377
+    },
+    {
+      "epoch": 34.36,
+      "learning_rate": 9.802870813397129e-05,
+      "loss": 0.1926,
+      "step": 378
+    },
+    {
+      "epoch": 34.45,
+      "learning_rate": 9.800956937799044e-05,
+      "loss": 0.199,
+      "step": 379
+    },
+    {
+      "epoch": 34.55,
+      "learning_rate": 9.799043062200958e-05,
+      "loss": 0.2305,
+      "step": 380
+    },
+    {
+      "epoch": 34.55,
+      "eval_accuracy_NAT": 0.985468897667528,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.8977423923773552,
+      "eval_accuracy_concrete tiles": 0.9639515685195377,
+      "eval_accuracy_dark bricks": 0.04491743966744929,
+      "eval_accuracy_grass": 0.8741843371767494,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9721228991507727,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.14093127119344967,
+      "eval_accuracy_speedway bricks": 0.9746079961188809,
+      "eval_accuracy_steel": 0.7391003231976783,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.965356884261195,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8089298835905171,
+      "eval_iou_concrete tiles": 0.8073495518565941,
+      "eval_iou_dark bricks": 0.04491225369458128,
+      "eval_iou_grass": 0.7313335516543666,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9402549192061903,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.11247029833670685,
+      "eval_iou_speedway bricks": 0.9375066387515026,
+      "eval_iou_steel": 0.6287497808171137,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.2024267166852951,
+      "eval_mean_accuracy": 0.5071559326976464,
+      "eval_mean_iou": 0.459758750936059,
+      "eval_overall_accuracy": 0.9590724309285482,
+      "eval_runtime": 4.213,
+      "eval_samples_per_second": 8.545,
+      "eval_steps_per_second": 0.475,
+      "step": 380
+    },
+    {
+      "epoch": 34.64,
+      "learning_rate": 9.797129186602872e-05,
+      "loss": 0.2066,
+      "step": 381
+    },
+    {
+      "epoch": 34.73,
+      "learning_rate": 9.795215311004785e-05,
+      "loss": 0.2228,
+      "step": 382
+    },
+    {
+      "epoch": 34.82,
+      "learning_rate": 9.793301435406699e-05,
+      "loss": 0.204,
+      "step": 383
+    },
+    {
+      "epoch": 34.91,
+      "learning_rate": 9.791387559808612e-05,
+      "loss": 0.1985,
+      "step": 384
+    },
+    {
+      "epoch": 35.0,
+      "learning_rate": 9.789473684210527e-05,
+      "loss": 0.2294,
+      "step": 385
+    },
+    {
+      "epoch": 35.09,
+      "learning_rate": 9.787559808612442e-05,
+      "loss": 0.2086,
+      "step": 386
+    },
+    {
+      "epoch": 35.18,
+      "learning_rate": 9.785645933014355e-05,
+      "loss": 0.187,
+      "step": 387
+    },
+    {
+      "epoch": 35.27,
+      "learning_rate": 9.783732057416268e-05,
+      "loss": 0.1915,
+      "step": 388
+    },
+    {
+      "epoch": 35.36,
+      "learning_rate": 9.781818181818183e-05,
+      "loss": 0.2294,
+      "step": 389
+    },
+    {
+      "epoch": 35.45,
+      "learning_rate": 9.779904306220096e-05,
+      "loss": 0.2292,
+      "step": 390
+    },
+    {
+      "epoch": 35.55,
+      "learning_rate": 9.77799043062201e-05,
+      "loss": 0.1793,
+      "step": 391
+    },
+    {
+      "epoch": 35.64,
+      "learning_rate": 9.776076555023923e-05,
+      "loss": 0.2091,
+      "step": 392
+    },
+    {
+      "epoch": 35.73,
+      "learning_rate": 9.774162679425838e-05,
+      "loss": 0.1883,
+      "step": 393
+    },
+    {
+      "epoch": 35.82,
+      "learning_rate": 9.772248803827751e-05,
+      "loss": 0.2175,
+      "step": 394
+    },
+    {
+      "epoch": 35.91,
+      "learning_rate": 9.770334928229666e-05,
+      "loss": 0.2277,
+      "step": 395
+    },
+    {
+      "epoch": 36.0,
+      "learning_rate": 9.768421052631579e-05,
+      "loss": 0.1767,
+      "step": 396
+    },
+    {
+      "epoch": 36.09,
+      "learning_rate": 9.766507177033494e-05,
+      "loss": 0.2035,
+      "step": 397
+    },
+    {
+      "epoch": 36.18,
+      "learning_rate": 9.764593301435407e-05,
+      "loss": 0.229,
+      "step": 398
+    },
+    {
+      "epoch": 36.27,
+      "learning_rate": 9.762679425837321e-05,
+      "loss": 0.1671,
+      "step": 399
+    },
+    {
+      "epoch": 36.36,
+      "learning_rate": 9.760765550239235e-05,
+      "loss": 0.2019,
+      "step": 400
+    },
+    {
+      "epoch": 36.36,
+      "eval_accuracy_NAT": 0.9836902704148648,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9404172389372935,
+      "eval_accuracy_concrete tiles": 0.8294196783464808,
+      "eval_accuracy_dark bricks": 0.19722104614910896,
+      "eval_accuracy_grass": 0.8577819563325493,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9300545232270883,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.12643012709177626,
+      "eval_accuracy_speedway bricks": 0.9749531453838699,
+      "eval_accuracy_steel": 0.7743717432887013,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9671312829982205,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.7636547208281054,
+      "eval_iou_concrete tiles": 0.8252060961883613,
+      "eval_iou_dark bricks": 0.19474004256612953,
+      "eval_iou_grass": 0.7586279376717473,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9198591645308932,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.09111499642374632,
+      "eval_iou_speedway bricks": 0.9437740799671028,
+      "eval_iou_steel": 0.6910348379501895,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.18835988640785217,
+      "eval_mean_accuracy": 0.5087953637824411,
+      "eval_mean_iou": 0.47347255070188426,
+      "eval_overall_accuracy": 0.9576032426622179,
+      "eval_runtime": 3.4211,
+      "eval_samples_per_second": 10.523,
+      "eval_steps_per_second": 0.585,
+      "step": 400
+    },
+    {
+      "epoch": 36.45,
+      "learning_rate": 9.758851674641149e-05,
+      "loss": 0.1615,
+      "step": 401
+    },
+    {
+      "epoch": 36.55,
+      "learning_rate": 9.756937799043062e-05,
+      "loss": 0.1832,
+      "step": 402
+    },
+    {
+      "epoch": 36.64,
+      "learning_rate": 9.755023923444977e-05,
+      "loss": 0.1985,
+      "step": 403
+    },
+    {
+      "epoch": 36.73,
+      "learning_rate": 9.75311004784689e-05,
+      "loss": 0.1902,
+      "step": 404
+    },
+    {
+      "epoch": 36.82,
+      "learning_rate": 9.751196172248805e-05,
+      "loss": 0.2116,
+      "step": 405
+    },
+    {
+      "epoch": 36.91,
+      "learning_rate": 9.749282296650718e-05,
+      "loss": 0.1628,
+      "step": 406
+    },
+    {
+      "epoch": 37.0,
+      "learning_rate": 9.747368421052632e-05,
+      "loss": 0.1925,
+      "step": 407
+    },
+    {
+      "epoch": 37.09,
+      "learning_rate": 9.745454545454546e-05,
+      "loss": 0.1796,
+      "step": 408
+    },
+    {
+      "epoch": 37.18,
+      "learning_rate": 9.74354066985646e-05,
+      "loss": 0.1916,
+      "step": 409
+    },
+    {
+      "epoch": 37.27,
+      "learning_rate": 9.741626794258373e-05,
+      "loss": 0.1789,
+      "step": 410
+    },
+    {
+      "epoch": 37.36,
+      "learning_rate": 9.739712918660288e-05,
+      "loss": 0.2041,
+      "step": 411
+    },
+    {
+      "epoch": 37.45,
+      "learning_rate": 9.737799043062201e-05,
+      "loss": 0.1686,
+      "step": 412
+    },
+    {
+      "epoch": 37.55,
+      "learning_rate": 9.735885167464116e-05,
+      "loss": 0.1806,
+      "step": 413
+    },
+    {
+      "epoch": 37.64,
+      "learning_rate": 9.733971291866029e-05,
+      "loss": 0.1512,
+      "step": 414
+    },
+    {
+      "epoch": 37.73,
+      "learning_rate": 9.732057416267943e-05,
+      "loss": 0.2371,
+      "step": 415
+    },
+    {
+      "epoch": 37.82,
+      "learning_rate": 9.730143540669857e-05,
+      "loss": 0.1742,
+      "step": 416
+    },
+    {
+      "epoch": 37.91,
+      "learning_rate": 9.728229665071771e-05,
+      "loss": 0.1657,
+      "step": 417
+    },
+    {
+      "epoch": 38.0,
+      "learning_rate": 9.726315789473684e-05,
+      "loss": 0.1894,
+      "step": 418
+    },
+    {
+      "epoch": 38.09,
+      "learning_rate": 9.724401913875598e-05,
+      "loss": 0.184,
+      "step": 419
+    },
+    {
+      "epoch": 38.18,
+      "learning_rate": 9.722488038277513e-05,
+      "loss": 0.1776,
+      "step": 420
+    },
+    {
+      "epoch": 38.18,
+      "eval_accuracy_NAT": 0.9882026548129159,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9232866753937937,
+      "eval_accuracy_concrete tiles": 0.9740720357121018,
+      "eval_accuracy_dark bricks": 0.48146722604980563,
+      "eval_accuracy_grass": 0.8350744625450428,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9704227515063719,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.29145645521462243,
+      "eval_accuracy_speedway bricks": 0.9685033193463404,
+      "eval_accuracy_steel": 0.7703400171492646,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9676635307001212,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8076784084215862,
+      "eval_iou_concrete tiles": 0.8866685221263568,
+      "eval_iou_dark bricks": 0.47503132951050014,
+      "eval_iou_grass": 0.7490336475611553,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9414358613408411,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.21556153657939808,
+      "eval_iou_speedway bricks": 0.9483383387700864,
+      "eval_iou_steel": 0.6777481176282841,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.16935062408447266,
+      "eval_mean_accuracy": 0.5540635075177123,
+      "eval_mean_iou": 0.5130122532798714,
+      "eval_overall_accuracy": 0.9625424279106988,
+      "eval_runtime": 4.1696,
+      "eval_samples_per_second": 8.634,
+      "eval_steps_per_second": 0.48,
+      "step": 420
+    },
+    {
+      "epoch": 38.27,
+      "learning_rate": 9.720574162679427e-05,
+      "loss": 0.1633,
+      "step": 421
+    },
+    {
+      "epoch": 38.36,
+      "learning_rate": 9.71866028708134e-05,
+      "loss": 0.1644,
+      "step": 422
+    },
+    {
+      "epoch": 38.45,
+      "learning_rate": 9.716746411483254e-05,
+      "loss": 0.2091,
+      "step": 423
+    },
+    {
+      "epoch": 38.55,
+      "learning_rate": 9.714832535885168e-05,
+      "loss": 0.1725,
+      "step": 424
+    },
+    {
+      "epoch": 38.64,
+      "learning_rate": 9.712918660287081e-05,
+      "loss": 0.1785,
+      "step": 425
+    },
+    {
+      "epoch": 38.73,
+      "learning_rate": 9.711004784688997e-05,
+      "loss": 0.1577,
+      "step": 426
+    },
+    {
+      "epoch": 38.82,
+      "learning_rate": 9.70909090909091e-05,
+      "loss": 0.1615,
+      "step": 427
+    },
+    {
+      "epoch": 38.91,
+      "learning_rate": 9.707177033492823e-05,
+      "loss": 0.1651,
+      "step": 428
+    },
+    {
+      "epoch": 39.0,
+      "learning_rate": 9.705263157894738e-05,
+      "loss": 0.1627,
+      "step": 429
+    },
+    {
+      "epoch": 39.09,
+      "learning_rate": 9.703349282296651e-05,
+      "loss": 0.1565,
+      "step": 430
+    },
+    {
+      "epoch": 39.18,
+      "learning_rate": 9.701435406698564e-05,
+      "loss": 0.1748,
+      "step": 431
+    },
+    {
+      "epoch": 39.27,
+      "learning_rate": 9.69952153110048e-05,
+      "loss": 0.1529,
+      "step": 432
+    },
+    {
+      "epoch": 39.36,
+      "learning_rate": 9.697607655502393e-05,
+      "loss": 0.1678,
+      "step": 433
+    },
+    {
+      "epoch": 39.45,
+      "learning_rate": 9.695693779904306e-05,
+      "loss": 0.1408,
+      "step": 434
+    },
+    {
+      "epoch": 39.55,
+      "learning_rate": 9.693779904306221e-05,
+      "loss": 0.163,
+      "step": 435
+    },
+    {
+      "epoch": 39.64,
+      "learning_rate": 9.691866028708134e-05,
+      "loss": 0.1518,
+      "step": 436
+    },
+    {
+      "epoch": 39.73,
+      "learning_rate": 9.689952153110047e-05,
+      "loss": 0.189,
+      "step": 437
+    },
+    {
+      "epoch": 39.82,
+      "learning_rate": 9.688038277511963e-05,
+      "loss": 0.1797,
+      "step": 438
+    },
+    {
+      "epoch": 39.91,
+      "learning_rate": 9.686124401913876e-05,
+      "loss": 0.1552,
+      "step": 439
+    },
+    {
+      "epoch": 40.0,
+      "learning_rate": 9.68421052631579e-05,
+      "loss": 0.1435,
+      "step": 440
+    },
+    {
+      "epoch": 40.0,
+      "eval_accuracy_NAT": 0.9841262811233301,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.930624704922958,
+      "eval_accuracy_concrete tiles": 0.9227358894392467,
+      "eval_accuracy_dark bricks": 0.10492282822062277,
+      "eval_accuracy_grass": 0.8747260735992618,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9651179522773635,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.38513494886003363,
+      "eval_accuracy_speedway bricks": 0.98022873983107,
+      "eval_accuracy_steel": 0.8193390937273267,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9689827518449886,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8176077357896933,
+      "eval_iou_concrete tiles": 0.8828140997513529,
+      "eval_iou_dark bricks": 0.10472531694199001,
+      "eval_iou_grass": 0.770671412910934,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9476733831229741,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.2691559254763694,
+      "eval_iou_speedway bricks": 0.9438091010711107,
+      "eval_iou_steel": 0.70446457686473,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.15863868594169617,
+      "eval_mean_accuracy": 0.5359197316924011,
+      "eval_mean_iou": 0.49306956182878026,
+      "eval_overall_accuracy": 0.9632458157009549,
+      "eval_runtime": 4.1178,
+      "eval_samples_per_second": 8.743,
+      "eval_steps_per_second": 0.486,
+      "step": 440
+    },
+    {
+      "epoch": 40.09,
+      "learning_rate": 9.682296650717704e-05,
+      "loss": 0.1606,
+      "step": 441
+    },
+    {
+      "epoch": 40.18,
+      "learning_rate": 9.680382775119617e-05,
+      "loss": 0.1511,
+      "step": 442
+    },
+    {
+      "epoch": 40.27,
+      "learning_rate": 9.67846889952153e-05,
+      "loss": 0.1478,
+      "step": 443
+    },
+    {
+      "epoch": 40.36,
+      "learning_rate": 9.676555023923446e-05,
+      "loss": 0.1711,
+      "step": 444
+    },
+    {
+      "epoch": 40.45,
+      "learning_rate": 9.67464114832536e-05,
+      "loss": 0.1528,
+      "step": 445
+    },
+    {
+      "epoch": 40.55,
+      "learning_rate": 9.672727272727273e-05,
+      "loss": 0.1651,
+      "step": 446
+    },
+    {
+      "epoch": 40.64,
+      "learning_rate": 9.670813397129187e-05,
+      "loss": 0.1367,
+      "step": 447
+    },
+    {
+      "epoch": 40.73,
+      "learning_rate": 9.6688995215311e-05,
+      "loss": 0.1446,
+      "step": 448
+    },
+    {
+      "epoch": 40.82,
+      "learning_rate": 9.666985645933014e-05,
+      "loss": 0.1597,
+      "step": 449
+    },
+    {
+      "epoch": 40.91,
+      "learning_rate": 9.66507177033493e-05,
+      "loss": 0.1932,
+      "step": 450
+    },
+    {
+      "epoch": 41.0,
+      "learning_rate": 9.663157894736843e-05,
+      "loss": 0.1348,
+      "step": 451
+    },
+    {
+      "epoch": 41.09,
+      "learning_rate": 9.661244019138756e-05,
+      "loss": 0.1724,
+      "step": 452
+    },
+    {
+      "epoch": 41.18,
+      "learning_rate": 9.659330143540671e-05,
+      "loss": 0.1336,
+      "step": 453
+    },
+    {
+      "epoch": 41.27,
+      "learning_rate": 9.657416267942584e-05,
+      "loss": 0.152,
+      "step": 454
+    },
+    {
+      "epoch": 41.36,
+      "learning_rate": 9.655502392344497e-05,
+      "loss": 0.1477,
+      "step": 455
+    },
+    {
+      "epoch": 41.45,
+      "learning_rate": 9.653588516746413e-05,
+      "loss": 0.1708,
+      "step": 456
+    },
+    {
+      "epoch": 41.55,
+      "learning_rate": 9.651674641148326e-05,
+      "loss": 0.1433,
+      "step": 457
+    },
+    {
+      "epoch": 41.64,
+      "learning_rate": 9.64976076555024e-05,
+      "loss": 0.1567,
+      "step": 458
+    },
+    {
+      "epoch": 41.73,
+      "learning_rate": 9.647846889952154e-05,
+      "loss": 0.1316,
+      "step": 459
+    },
+    {
+      "epoch": 41.82,
+      "learning_rate": 9.645933014354067e-05,
+      "loss": 0.1526,
+      "step": 460
+    },
+    {
+      "epoch": 41.82,
+      "eval_accuracy_NAT": 0.98626380994662,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9280843598437701,
+      "eval_accuracy_concrete tiles": 0.9928147740475753,
+      "eval_accuracy_dark bricks": 0.18301835957045534,
+      "eval_accuracy_grass": 0.8375454796464383,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9556401733038126,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.366884459515342,
+      "eval_accuracy_speedway bricks": 0.9842570903434236,
+      "eval_accuracy_steel": 0.7994113185146099,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.969852297129785,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8065399732377226,
+      "eval_iou_concrete tiles": 0.8172917191039517,
+      "eval_iou_dark bricks": 0.18194688910997167,
+      "eval_iou_grass": 0.7651311000140487,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9412503126377094,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.2536064792758456,
+      "eval_iou_speedway bricks": 0.9485585220955278,
+      "eval_iou_steel": 0.7195206114800935,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1500294953584671,
+      "eval_mean_accuracy": 0.5410707557486191,
+      "eval_mean_iou": 0.4925921464680504,
+      "eval_overall_accuracy": 0.9632099999321831,
+      "eval_runtime": 3.8948,
+      "eval_samples_per_second": 9.243,
+      "eval_steps_per_second": 0.514,
+      "step": 460
+    },
+    {
+      "epoch": 41.91,
+      "learning_rate": 9.64401913875598e-05,
+      "loss": 0.133,
+      "step": 461
+    },
+    {
+      "epoch": 42.0,
+      "learning_rate": 9.642105263157896e-05,
+      "loss": 0.1788,
+      "step": 462
+    },
+    {
+      "epoch": 42.09,
+      "learning_rate": 9.64019138755981e-05,
+      "loss": 0.1193,
+      "step": 463
+    },
+    {
+      "epoch": 42.18,
+      "learning_rate": 9.638277511961723e-05,
+      "loss": 0.1273,
+      "step": 464
+    },
+    {
+      "epoch": 42.27,
+      "learning_rate": 9.636363636363637e-05,
+      "loss": 0.1813,
+      "step": 465
+    },
+    {
+      "epoch": 42.36,
+      "learning_rate": 9.63444976076555e-05,
+      "loss": 0.1501,
+      "step": 466
+    },
+    {
+      "epoch": 42.45,
+      "learning_rate": 9.632535885167464e-05,
+      "loss": 0.1675,
+      "step": 467
+    },
+    {
+      "epoch": 42.55,
+      "learning_rate": 9.63062200956938e-05,
+      "loss": 0.1266,
+      "step": 468
+    },
+    {
+      "epoch": 42.64,
+      "learning_rate": 9.628708133971293e-05,
+      "loss": 0.1474,
+      "step": 469
+    },
+    {
+      "epoch": 42.73,
+      "learning_rate": 9.626794258373206e-05,
+      "loss": 0.1492,
+      "step": 470
+    },
+    {
+      "epoch": 42.82,
+      "learning_rate": 9.62488038277512e-05,
+      "loss": 0.1898,
+      "step": 471
+    },
+    {
+      "epoch": 42.91,
+      "learning_rate": 9.622966507177034e-05,
+      "loss": 0.1311,
+      "step": 472
+    },
+    {
+      "epoch": 43.0,
+      "learning_rate": 9.621052631578947e-05,
+      "loss": 0.128,
+      "step": 473
+    },
+    {
+      "epoch": 43.09,
+      "learning_rate": 9.619138755980861e-05,
+      "loss": 0.1338,
+      "step": 474
+    },
+    {
+      "epoch": 43.18,
+      "learning_rate": 9.617224880382776e-05,
+      "loss": 0.1197,
+      "step": 475
+    },
+    {
+      "epoch": 43.27,
+      "learning_rate": 9.615311004784689e-05,
+      "loss": 0.1256,
+      "step": 476
+    },
+    {
+      "epoch": 43.36,
+      "learning_rate": 9.613397129186604e-05,
+      "loss": 0.142,
+      "step": 477
+    },
+    {
+      "epoch": 43.45,
+      "learning_rate": 9.611483253588517e-05,
+      "loss": 0.1206,
+      "step": 478
+    },
+    {
+      "epoch": 43.55,
+      "learning_rate": 9.60956937799043e-05,
+      "loss": 0.1462,
+      "step": 479
+    },
+    {
+      "epoch": 43.64,
+      "learning_rate": 9.607655502392345e-05,
+      "loss": 0.1364,
+      "step": 480
+    },
+    {
+      "epoch": 43.64,
+      "eval_accuracy_NAT": 0.9884720956675604,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9232786278381047,
+      "eval_accuracy_concrete tiles": 0.9190362624594876,
+      "eval_accuracy_dark bricks": 0.1722797428890343,
+      "eval_accuracy_grass": 0.8367171471810483,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9589751713146639,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.43765335097731095,
+      "eval_accuracy_speedway bricks": 0.9872934384228368,
+      "eval_accuracy_steel": 0.7097899215091353,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9701265728085884,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8276357506729972,
+      "eval_iou_concrete tiles": 0.8994553833263511,
+      "eval_iou_dark bricks": 0.16926334896384812,
+      "eval_iou_grass": 0.772446704504022,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9352166755503564,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3003329675734988,
+      "eval_iou_speedway bricks": 0.9432432728465528,
+      "eval_iou_steel": 0.6637957622675261,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14121562242507935,
+      "eval_mean_accuracy": 0.5333458275583985,
+      "eval_mean_iou": 0.49857818757798006,
+      "eval_overall_accuracy": 0.963791741265191,
+      "eval_runtime": 3.2569,
+      "eval_samples_per_second": 11.054,
+      "eval_steps_per_second": 0.614,
+      "step": 480
+    },
+    {
+      "epoch": 43.73,
+      "learning_rate": 9.605741626794259e-05,
+      "loss": 0.1608,
+      "step": 481
+    },
+    {
+      "epoch": 43.82,
+      "learning_rate": 9.603827751196172e-05,
+      "loss": 0.1309,
+      "step": 482
+    },
+    {
+      "epoch": 43.91,
+      "learning_rate": 9.601913875598087e-05,
+      "loss": 0.1342,
+      "step": 483
+    },
+    {
+      "epoch": 44.0,
+      "learning_rate": 9.6e-05,
+      "loss": 0.1406,
+      "step": 484
+    },
+    {
+      "epoch": 44.09,
+      "learning_rate": 9.598086124401913e-05,
+      "loss": 0.148,
+      "step": 485
+    },
+    {
+      "epoch": 44.18,
+      "learning_rate": 9.596172248803828e-05,
+      "loss": 0.1433,
+      "step": 486
+    },
+    {
+      "epoch": 44.27,
+      "learning_rate": 9.594258373205743e-05,
+      "loss": 0.1346,
+      "step": 487
+    },
+    {
+      "epoch": 44.36,
+      "learning_rate": 9.592344497607656e-05,
+      "loss": 0.134,
+      "step": 488
+    },
+    {
+      "epoch": 44.45,
+      "learning_rate": 9.59043062200957e-05,
+      "loss": 0.1316,
+      "step": 489
+    },
+    {
+      "epoch": 44.55,
+      "learning_rate": 9.588516746411484e-05,
+      "loss": 0.1333,
+      "step": 490
+    },
+    {
+      "epoch": 44.64,
+      "learning_rate": 9.586602870813398e-05,
+      "loss": 0.1439,
+      "step": 491
+    },
+    {
+      "epoch": 44.73,
+      "learning_rate": 9.584688995215311e-05,
+      "loss": 0.1429,
+      "step": 492
+    },
+    {
+      "epoch": 44.82,
+      "learning_rate": 9.582775119617226e-05,
+      "loss": 0.1169,
+      "step": 493
+    },
+    {
+      "epoch": 44.91,
+      "learning_rate": 9.580861244019139e-05,
+      "loss": 0.1286,
+      "step": 494
+    },
+    {
+      "epoch": 45.0,
+      "learning_rate": 9.578947368421052e-05,
+      "loss": 0.134,
+      "step": 495
+    },
+    {
+      "epoch": 45.09,
+      "learning_rate": 9.577033492822967e-05,
+      "loss": 0.1088,
+      "step": 496
+    },
+    {
+      "epoch": 45.18,
+      "learning_rate": 9.575119617224881e-05,
+      "loss": 0.1463,
+      "step": 497
+    },
+    {
+      "epoch": 45.27,
+      "learning_rate": 9.573205741626795e-05,
+      "loss": 0.1278,
+      "step": 498
+    },
+    {
+      "epoch": 45.36,
+      "learning_rate": 9.571291866028709e-05,
+      "loss": 0.1499,
+      "step": 499
+    },
+    {
+      "epoch": 45.45,
+      "learning_rate": 9.569377990430622e-05,
+      "loss": 0.1252,
+      "step": 500
+    },
+    {
+      "epoch": 45.45,
+      "eval_accuracy_NAT": 0.9867451685919836,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9337914180866131,
+      "eval_accuracy_concrete tiles": 0.9220938054179662,
+      "eval_accuracy_dark bricks": 0.4072206612524537,
+      "eval_accuracy_grass": 0.8675926281905654,
+      "eval_accuracy_red bricks": 0.0,
+      "eval_accuracy_road": 0.9799597817136182,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5415047004659113,
+      "eval_accuracy_speedway bricks": 0.9817058821399737,
+      "eval_accuracy_steel": 0.8060154343381043,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9714276730892634,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8489828619370133,
+      "eval_iou_concrete tiles": 0.8891181933429642,
+      "eval_iou_dark bricks": 0.3961063272182703,
+      "eval_iou_grass": 0.7790292614955876,
+      "eval_iou_red bricks": 0.0,
+      "eval_iou_road": 0.9523537561739167,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3482253660958054,
+      "eval_iou_speedway bricks": 0.953560313450594,
+      "eval_iou_steel": 0.7239282884457313,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.13200995326042175,
+      "eval_mean_accuracy": 0.5712791907843993,
+      "eval_mean_iou": 0.5279024647114728,
+      "eval_overall_accuracy": 0.967672242058648,
+      "eval_runtime": 4.0122,
+      "eval_samples_per_second": 8.973,
+      "eval_steps_per_second": 0.498,
+      "step": 500
+    },
+    {
+      "epoch": 45.55,
+      "learning_rate": 9.567464114832535e-05,
+      "loss": 0.125,
+      "step": 501
+    },
+    {
+      "epoch": 45.64,
+      "learning_rate": 9.56555023923445e-05,
+      "loss": 0.1376,
+      "step": 502
+    },
+    {
+      "epoch": 45.73,
+      "learning_rate": 9.563636363636365e-05,
+      "loss": 0.1213,
+      "step": 503
+    },
+    {
+      "epoch": 45.82,
+      "learning_rate": 9.561722488038278e-05,
+      "loss": 0.1391,
+      "step": 504
+    },
+    {
+      "epoch": 45.91,
+      "learning_rate": 9.559808612440192e-05,
+      "loss": 0.136,
+      "step": 505
+    },
+    {
+      "epoch": 46.0,
+      "learning_rate": 9.557894736842106e-05,
+      "loss": 0.129,
+      "step": 506
+    },
+    {
+      "epoch": 46.09,
+      "learning_rate": 9.555980861244019e-05,
+      "loss": 0.1192,
+      "step": 507
+    },
+    {
+      "epoch": 46.18,
+      "learning_rate": 9.554066985645933e-05,
+      "loss": 0.1359,
+      "step": 508
+    },
+    {
+      "epoch": 46.27,
+      "learning_rate": 9.552153110047848e-05,
+      "loss": 0.1543,
+      "step": 509
+    },
+    {
+      "epoch": 46.36,
+      "learning_rate": 9.550239234449761e-05,
+      "loss": 0.138,
+      "step": 510
+    },
+    {
+      "epoch": 46.45,
+      "learning_rate": 9.548325358851676e-05,
+      "loss": 0.1424,
+      "step": 511
+    },
+    {
+      "epoch": 46.55,
+      "learning_rate": 9.546411483253589e-05,
+      "loss": 0.1281,
+      "step": 512
+    },
+    {
+      "epoch": 46.64,
+      "learning_rate": 9.544497607655502e-05,
+      "loss": 0.1156,
+      "step": 513
+    },
+    {
+      "epoch": 46.73,
+      "learning_rate": 9.542583732057417e-05,
+      "loss": 0.1226,
+      "step": 514
+    },
+    {
+      "epoch": 46.82,
+      "learning_rate": 9.540669856459331e-05,
+      "loss": 0.1352,
+      "step": 515
+    },
+    {
+      "epoch": 46.91,
+      "learning_rate": 9.538755980861244e-05,
+      "loss": 0.1239,
+      "step": 516
+    },
+    {
+      "epoch": 47.0,
+      "learning_rate": 9.536842105263159e-05,
+      "loss": 0.1336,
+      "step": 517
+    },
+    {
+      "epoch": 47.09,
+      "learning_rate": 9.534928229665072e-05,
+      "loss": 0.1247,
+      "step": 518
+    },
+    {
+      "epoch": 47.18,
+      "learning_rate": 9.533014354066985e-05,
+      "loss": 0.1408,
+      "step": 519
+    },
+    {
+      "epoch": 47.27,
+      "learning_rate": 9.5311004784689e-05,
+      "loss": 0.1178,
+      "step": 520
+    },
+    {
+      "epoch": 47.27,
+      "eval_accuracy_NAT": 0.9902731322119982,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9302679299540753,
+      "eval_accuracy_concrete tiles": 0.9653886137100226,
+      "eval_accuracy_dark bricks": 0.7099033909395327,
+      "eval_accuracy_grass": 0.8182002467522028,
+      "eval_accuracy_red bricks": 0.0234375,
+      "eval_accuracy_road": 0.9629184013775307,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5105450334959888,
+      "eval_accuracy_speedway bricks": 0.9873513655722056,
+      "eval_accuracy_steel": 0.792337246883451,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9714994904626413,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8438971863117871,
+      "eval_iou_concrete tiles": 0.9061011306893187,
+      "eval_iou_dark bricks": 0.6812691611568722,
+      "eval_iou_grass": 0.7603841856355888,
+      "eval_iou_red bricks": 0.0234375,
+      "eval_iou_road": 0.9529383185264609,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.34072343244038855,
+      "eval_iou_speedway bricks": 0.9570195914208796,
+      "eval_iou_steel": 0.7326614163623473,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12958088517189026,
+      "eval_mean_accuracy": 0.5915863739151543,
+      "eval_mean_iou": 0.5515331856158681,
+      "eval_overall_accuracy": 0.9682156244913737,
+      "eval_runtime": 4.1158,
+      "eval_samples_per_second": 8.747,
+      "eval_steps_per_second": 0.486,
+      "step": 520
+    },
+    {
+      "epoch": 47.36,
+      "learning_rate": 9.529186602870814e-05,
+      "loss": 0.1102,
+      "step": 521
+    },
+    {
+      "epoch": 47.45,
+      "learning_rate": 9.527272727272728e-05,
+      "loss": 0.1083,
+      "step": 522
+    },
+    {
+      "epoch": 47.55,
+      "learning_rate": 9.525358851674642e-05,
+      "loss": 0.1129,
+      "step": 523
+    },
+    {
+      "epoch": 47.64,
+      "learning_rate": 9.523444976076555e-05,
+      "loss": 0.1056,
+      "step": 524
+    },
+    {
+      "epoch": 47.73,
+      "learning_rate": 9.521531100478469e-05,
+      "loss": 0.1407,
+      "step": 525
+    },
+    {
+      "epoch": 47.82,
+      "learning_rate": 9.519617224880383e-05,
+      "loss": 0.1192,
+      "step": 526
+    },
+    {
+      "epoch": 47.91,
+      "learning_rate": 9.517703349282298e-05,
+      "loss": 0.1211,
+      "step": 527
+    },
+    {
+      "epoch": 48.0,
+      "learning_rate": 9.515789473684211e-05,
+      "loss": 0.1165,
+      "step": 528
+    },
+    {
+      "epoch": 48.09,
+      "learning_rate": 9.513875598086125e-05,
+      "loss": 0.1535,
+      "step": 529
+    },
+    {
+      "epoch": 48.18,
+      "learning_rate": 9.511961722488039e-05,
+      "loss": 0.1237,
+      "step": 530
+    },
+    {
+      "epoch": 48.27,
+      "learning_rate": 9.510047846889952e-05,
+      "loss": 0.1065,
+      "step": 531
+    },
+    {
+      "epoch": 48.36,
+      "learning_rate": 9.508133971291866e-05,
+      "loss": 0.1436,
+      "step": 532
+    },
+    {
+      "epoch": 48.45,
+      "learning_rate": 9.506220095693781e-05,
+      "loss": 0.1079,
+      "step": 533
+    },
+    {
+      "epoch": 48.55,
+      "learning_rate": 9.504306220095694e-05,
+      "loss": 0.1295,
+      "step": 534
+    },
+    {
+      "epoch": 48.64,
+      "learning_rate": 9.502392344497609e-05,
+      "loss": 0.118,
+      "step": 535
+    },
+    {
+      "epoch": 48.73,
+      "learning_rate": 9.500478468899522e-05,
+      "loss": 0.1279,
+      "step": 536
+    },
+    {
+      "epoch": 48.82,
+      "learning_rate": 9.498564593301435e-05,
+      "loss": 0.1042,
+      "step": 537
+    },
+    {
+      "epoch": 48.91,
+      "learning_rate": 9.49665071770335e-05,
+      "loss": 0.1316,
+      "step": 538
+    },
+    {
+      "epoch": 49.0,
+      "learning_rate": 9.494736842105264e-05,
+      "loss": 0.1238,
+      "step": 539
+    },
+    {
+      "epoch": 49.09,
+      "learning_rate": 9.492822966507177e-05,
+      "loss": 0.1182,
+      "step": 540
+    },
+    {
+      "epoch": 49.09,
+      "eval_accuracy_NAT": 0.9888795212951794,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9476318726125584,
+      "eval_accuracy_concrete tiles": 0.9935485843576102,
+      "eval_accuracy_dark bricks": 0.7152534544474809,
+      "eval_accuracy_grass": 0.8474714889363443,
+      "eval_accuracy_red bricks": 0.06711647727272728,
+      "eval_accuracy_road": 0.9674009385879101,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5158933642102942,
+      "eval_accuracy_speedway bricks": 0.9691055203366533,
+      "eval_accuracy_steel": 0.8630284941626541,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9726659956912479,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8319524843506398,
+      "eval_iou_concrete tiles": 0.8837367419091651,
+      "eval_iou_dark bricks": 0.6892804154302671,
+      "eval_iou_grass": 0.7758611311095113,
+      "eval_iou_red bricks": 0.06711647727272728,
+      "eval_iou_road": 0.9536358478241176,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3404963790530951,
+      "eval_iou_speedway bricks": 0.9559566417408064,
+      "eval_iou_steel": 0.7409465428393454,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1293036937713623,
+      "eval_mean_accuracy": 0.6057945935553394,
+      "eval_mean_iou": 0.5547422044016095,
+      "eval_overall_accuracy": 0.9678941302829318,
+      "eval_runtime": 4.4972,
+      "eval_samples_per_second": 8.005,
+      "eval_steps_per_second": 0.445,
+      "step": 540
+    },
+    {
+      "epoch": 49.18,
+      "learning_rate": 9.490909090909092e-05,
+      "loss": 0.1147,
+      "step": 541
+    },
+    {
+      "epoch": 49.27,
+      "learning_rate": 9.488995215311005e-05,
+      "loss": 0.1217,
+      "step": 542
+    },
+    {
+      "epoch": 49.36,
+      "learning_rate": 9.487081339712918e-05,
+      "loss": 0.1194,
+      "step": 543
+    },
+    {
+      "epoch": 49.45,
+      "learning_rate": 9.485167464114833e-05,
+      "loss": 0.1149,
+      "step": 544
+    },
+    {
+      "epoch": 49.55,
+      "learning_rate": 9.483253588516747e-05,
+      "loss": 0.108,
+      "step": 545
+    },
+    {
+      "epoch": 49.64,
+      "learning_rate": 9.48133971291866e-05,
+      "loss": 0.105,
+      "step": 546
+    },
+    {
+      "epoch": 49.73,
+      "learning_rate": 9.479425837320575e-05,
+      "loss": 0.1405,
+      "step": 547
+    },
+    {
+      "epoch": 49.82,
+      "learning_rate": 9.477511961722488e-05,
+      "loss": 0.1161,
+      "step": 548
+    },
+    {
+      "epoch": 49.91,
+      "learning_rate": 9.475598086124402e-05,
+      "loss": 0.1092,
+      "step": 549
+    },
+    {
+      "epoch": 50.0,
+      "learning_rate": 9.473684210526316e-05,
+      "loss": 0.1408,
+      "step": 550
+    },
+    {
+      "epoch": 50.09,
+      "learning_rate": 9.471770334928231e-05,
+      "loss": 0.1119,
+      "step": 551
+    },
+    {
+      "epoch": 50.18,
+      "learning_rate": 9.469856459330144e-05,
+      "loss": 0.111,
+      "step": 552
+    },
+    {
+      "epoch": 50.27,
+      "learning_rate": 9.467942583732058e-05,
+      "loss": 0.1117,
+      "step": 553
+    },
+    {
+      "epoch": 50.36,
+      "learning_rate": 9.466028708133972e-05,
+      "loss": 0.1068,
+      "step": 554
+    },
+    {
+      "epoch": 50.45,
+      "learning_rate": 9.464114832535885e-05,
+      "loss": 0.1142,
+      "step": 555
+    },
+    {
+      "epoch": 50.55,
+      "learning_rate": 9.4622009569378e-05,
+      "loss": 0.0968,
+      "step": 556
+    },
+    {
+      "epoch": 50.64,
+      "learning_rate": 9.460287081339714e-05,
+      "loss": 0.1139,
+      "step": 557
+    },
+    {
+      "epoch": 50.73,
+      "learning_rate": 9.458373205741627e-05,
+      "loss": 0.1156,
+      "step": 558
+    },
+    {
+      "epoch": 50.82,
+      "learning_rate": 9.456459330143542e-05,
+      "loss": 0.1214,
+      "step": 559
+    },
+    {
+      "epoch": 50.91,
+      "learning_rate": 9.454545454545455e-05,
+      "loss": 0.1342,
+      "step": 560
+    },
+    {
+      "epoch": 50.91,
+      "eval_accuracy_NAT": 0.9915176419355135,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9200985020816344,
+      "eval_accuracy_concrete tiles": 0.9832752400171222,
+      "eval_accuracy_dark bricks": 0.7222200839074708,
+      "eval_accuracy_grass": 0.8231073302180576,
+      "eval_accuracy_red bricks": 0.017933238636363636,
+      "eval_accuracy_road": 0.9656354936655595,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.35271413999393486,
+      "eval_accuracy_speedway bricks": 0.9884079326403798,
+      "eval_accuracy_steel": 0.8092968801530243,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.971782404665203,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8448726033065994,
+      "eval_iou_concrete tiles": 0.9157412153311693,
+      "eval_iou_dark bricks": 0.6926541159099299,
+      "eval_iou_grass": 0.7688728987035544,
+      "eval_iou_red bricks": 0.017933238636363636,
+      "eval_iou_road": 0.9543721100882345,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.24622312888512538,
+      "eval_iou_speedway bricks": 0.955861148182015,
+      "eval_iou_steel": 0.7430189164843649,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12546773254871368,
+      "eval_mean_accuracy": 0.5826312679422355,
+      "eval_mean_iou": 0.5470255215532738,
+      "eval_overall_accuracy": 0.9684359232584635,
+      "eval_runtime": 3.4215,
+      "eval_samples_per_second": 10.522,
+      "eval_steps_per_second": 0.585,
+      "step": 560
+    },
+    {
+      "epoch": 51.0,
+      "learning_rate": 9.452631578947368e-05,
+      "loss": 0.1262,
+      "step": 561
+    },
+    {
+      "epoch": 51.09,
+      "learning_rate": 9.450717703349283e-05,
+      "loss": 0.12,
+      "step": 562
+    },
+    {
+      "epoch": 51.18,
+      "learning_rate": 9.448803827751197e-05,
+      "loss": 0.1003,
+      "step": 563
+    },
+    {
+      "epoch": 51.27,
+      "learning_rate": 9.44688995215311e-05,
+      "loss": 0.129,
+      "step": 564
+    },
+    {
+      "epoch": 51.36,
+      "learning_rate": 9.444976076555025e-05,
+      "loss": 0.1304,
+      "step": 565
+    },
+    {
+      "epoch": 51.45,
+      "learning_rate": 9.443062200956938e-05,
+      "loss": 0.1054,
+      "step": 566
+    },
+    {
+      "epoch": 51.55,
+      "learning_rate": 9.441148325358851e-05,
+      "loss": 0.0973,
+      "step": 567
+    },
+    {
+      "epoch": 51.64,
+      "learning_rate": 9.439234449760766e-05,
+      "loss": 0.109,
+      "step": 568
+    },
+    {
+      "epoch": 51.73,
+      "learning_rate": 9.43732057416268e-05,
+      "loss": 0.1078,
+      "step": 569
+    },
+    {
+      "epoch": 51.82,
+      "learning_rate": 9.435406698564594e-05,
+      "loss": 0.0951,
+      "step": 570
+    },
+    {
+      "epoch": 51.91,
+      "learning_rate": 9.433492822966507e-05,
+      "loss": 0.1112,
+      "step": 571
+    },
+    {
+      "epoch": 52.0,
+      "learning_rate": 9.431578947368421e-05,
+      "loss": 0.1046,
+      "step": 572
+    },
+    {
+      "epoch": 52.09,
+      "learning_rate": 9.429665071770335e-05,
+      "loss": 0.1108,
+      "step": 573
+    },
+    {
+      "epoch": 52.18,
+      "learning_rate": 9.427751196172249e-05,
+      "loss": 0.1173,
+      "step": 574
+    },
+    {
+      "epoch": 52.27,
+      "learning_rate": 9.425837320574164e-05,
+      "loss": 0.0919,
+      "step": 575
+    },
+    {
+      "epoch": 52.36,
+      "learning_rate": 9.423923444976077e-05,
+      "loss": 0.1083,
+      "step": 576
+    },
+    {
+      "epoch": 52.45,
+      "learning_rate": 9.42200956937799e-05,
+      "loss": 0.1198,
+      "step": 577
+    },
+    {
+      "epoch": 52.55,
+      "learning_rate": 9.420095693779905e-05,
+      "loss": 0.1068,
+      "step": 578
+    },
+    {
+      "epoch": 52.64,
+      "learning_rate": 9.418181818181818e-05,
+      "loss": 0.1001,
+      "step": 579
+    },
+    {
+      "epoch": 52.73,
+      "learning_rate": 9.416267942583733e-05,
+      "loss": 0.1106,
+      "step": 580
+    },
+    {
+      "epoch": 52.73,
+      "eval_accuracy_NAT": 0.990988465037623,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9138535988669042,
+      "eval_accuracy_concrete tiles": 0.9930288020546688,
+      "eval_accuracy_dark bricks": 0.7633655363534891,
+      "eval_accuracy_grass": 0.8616649832061709,
+      "eval_accuracy_red bricks": 0.375,
+      "eval_accuracy_road": 0.9735981339489013,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5821134176936013,
+      "eval_accuracy_speedway bricks": 0.9852738324964067,
+      "eval_accuracy_steel": 0.8296533869797507,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9728656348044922,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8589113279378766,
+      "eval_iou_concrete tiles": 0.8547067028079687,
+      "eval_iou_dark bricks": 0.7323043975925858,
+      "eval_iou_grass": 0.7874795174257596,
+      "eval_iou_red bricks": 0.37202747930244845,
+      "eval_iou_road": 0.9544895755166009,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.38177807510803335,
+      "eval_iou_speedway bricks": 0.9600984294794527,
+      "eval_iou_steel": 0.748139446703791,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11939381062984467,
+      "eval_mean_accuracy": 0.6360415505105781,
+      "eval_mean_iou": 0.5863692758983853,
+      "eval_overall_accuracy": 0.9704446792602539,
+      "eval_runtime": 4.6156,
+      "eval_samples_per_second": 7.8,
+      "eval_steps_per_second": 0.433,
+      "step": 580
+    },
+    {
+      "epoch": 52.82,
+      "learning_rate": 9.414354066985647e-05,
+      "loss": 0.136,
+      "step": 581
+    },
+    {
+      "epoch": 52.91,
+      "learning_rate": 9.41244019138756e-05,
+      "loss": 0.096,
+      "step": 582
+    },
+    {
+      "epoch": 53.0,
+      "learning_rate": 9.410526315789473e-05,
+      "loss": 0.1119,
+      "step": 583
+    },
+    {
+      "epoch": 53.09,
+      "learning_rate": 9.408612440191388e-05,
+      "loss": 0.1004,
+      "step": 584
+    },
+    {
+      "epoch": 53.18,
+      "learning_rate": 9.406698564593301e-05,
+      "loss": 0.1061,
+      "step": 585
+    },
+    {
+      "epoch": 53.27,
+      "learning_rate": 9.404784688995216e-05,
+      "loss": 0.1312,
+      "step": 586
+    },
+    {
+      "epoch": 53.36,
+      "learning_rate": 9.40287081339713e-05,
+      "loss": 0.092,
+      "step": 587
+    },
+    {
+      "epoch": 53.45,
+      "learning_rate": 9.400956937799044e-05,
+      "loss": 0.1053,
+      "step": 588
+    },
+    {
+      "epoch": 53.55,
+      "learning_rate": 9.399043062200957e-05,
+      "loss": 0.1041,
+      "step": 589
+    },
+    {
+      "epoch": 53.64,
+      "learning_rate": 9.397129186602871e-05,
+      "loss": 0.1127,
+      "step": 590
+    },
+    {
+      "epoch": 53.73,
+      "learning_rate": 9.395215311004784e-05,
+      "loss": 0.1054,
+      "step": 591
+    },
+    {
+      "epoch": 53.82,
+      "learning_rate": 9.393301435406699e-05,
+      "loss": 0.13,
+      "step": 592
+    },
+    {
+      "epoch": 53.91,
+      "learning_rate": 9.391387559808614e-05,
+      "loss": 0.0898,
+      "step": 593
+    },
+    {
+      "epoch": 54.0,
+      "learning_rate": 9.389473684210527e-05,
+      "loss": 0.0978,
+      "step": 594
+    },
+    {
+      "epoch": 54.09,
+      "learning_rate": 9.38755980861244e-05,
+      "loss": 0.1383,
+      "step": 595
+    },
+    {
+      "epoch": 54.18,
+      "learning_rate": 9.385645933014355e-05,
+      "loss": 0.0953,
+      "step": 596
+    },
+    {
+      "epoch": 54.27,
+      "learning_rate": 9.383732057416268e-05,
+      "loss": 0.1027,
+      "step": 597
+    },
+    {
+      "epoch": 54.36,
+      "learning_rate": 9.381818181818182e-05,
+      "loss": 0.1041,
+      "step": 598
+    },
+    {
+      "epoch": 54.45,
+      "learning_rate": 9.379904306220097e-05,
+      "loss": 0.118,
+      "step": 599
+    },
+    {
+      "epoch": 54.55,
+      "learning_rate": 9.37799043062201e-05,
+      "loss": 0.0981,
+      "step": 600
+    },
+    {
+      "epoch": 54.55,
+      "eval_accuracy_NAT": 0.9892555386240907,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.921518895660758,
+      "eval_accuracy_concrete tiles": 0.9933957072096863,
+      "eval_accuracy_dark bricks": 0.7974673800084677,
+      "eval_accuracy_grass": 0.8811814747113943,
+      "eval_accuracy_red bricks": 0.6079545454545454,
+      "eval_accuracy_road": 0.9667661039411745,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5978551539712734,
+      "eval_accuracy_speedway bricks": 0.982337650112777,
+      "eval_accuracy_steel": 0.8642652199722973,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9732483430700409,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8550894909575614,
+      "eval_iou_concrete tiles": 0.8616665782634063,
+      "eval_iou_dark bricks": 0.7314739629302736,
+      "eval_iou_grass": 0.7812012951802563,
+      "eval_iou_red bricks": 0.6022867194371152,
+      "eval_iou_road": 0.9531634296587104,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3826108435222922,
+      "eval_iou_speedway bricks": 0.9613461585021267,
+      "eval_iou_steel": 0.7556370923560451,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11864423751831055,
+      "eval_mean_accuracy": 0.6616921284358818,
+      "eval_mean_iou": 0.6044403010675253,
+      "eval_overall_accuracy": 0.9702264997694228,
+      "eval_runtime": 3.9263,
+      "eval_samples_per_second": 9.169,
+      "eval_steps_per_second": 0.509,
+      "step": 600
+    },
+    {
+      "epoch": 54.64,
+      "learning_rate": 9.376076555023923e-05,
+      "loss": 0.1038,
+      "step": 601
+    },
+    {
+      "epoch": 54.73,
+      "learning_rate": 9.374162679425838e-05,
+      "loss": 0.1029,
+      "step": 602
+    },
+    {
+      "epoch": 54.82,
+      "learning_rate": 9.372248803827751e-05,
+      "loss": 0.0943,
+      "step": 603
+    },
+    {
+      "epoch": 54.91,
+      "learning_rate": 9.370334928229666e-05,
+      "loss": 0.0935,
+      "step": 604
+    },
+    {
+      "epoch": 55.0,
+      "learning_rate": 9.36842105263158e-05,
+      "loss": 0.0993,
+      "step": 605
+    },
+    {
+      "epoch": 55.09,
+      "learning_rate": 9.366507177033493e-05,
+      "loss": 0.1025,
+      "step": 606
+    },
+    {
+      "epoch": 55.18,
+      "learning_rate": 9.364593301435407e-05,
+      "loss": 0.1031,
+      "step": 607
+    },
+    {
+      "epoch": 55.27,
+      "learning_rate": 9.362679425837321e-05,
+      "loss": 0.0963,
+      "step": 608
+    },
+    {
+      "epoch": 55.36,
+      "learning_rate": 9.360765550239234e-05,
+      "loss": 0.1222,
+      "step": 609
+    },
+    {
+      "epoch": 55.45,
+      "learning_rate": 9.358851674641149e-05,
+      "loss": 0.1241,
+      "step": 610
+    },
+    {
+      "epoch": 55.55,
+      "learning_rate": 9.356937799043063e-05,
+      "loss": 0.0804,
+      "step": 611
+    },
+    {
+      "epoch": 55.64,
+      "learning_rate": 9.355023923444977e-05,
+      "loss": 0.1012,
+      "step": 612
+    },
+    {
+      "epoch": 55.73,
+      "learning_rate": 9.35311004784689e-05,
+      "loss": 0.0894,
+      "step": 613
+    },
+    {
+      "epoch": 55.82,
+      "learning_rate": 9.351196172248804e-05,
+      "loss": 0.0969,
+      "step": 614
+    },
+    {
+      "epoch": 55.91,
+      "learning_rate": 9.349282296650718e-05,
+      "loss": 0.0957,
+      "step": 615
+    },
+    {
+      "epoch": 56.0,
+      "learning_rate": 9.347368421052632e-05,
+      "loss": 0.1112,
+      "step": 616
+    },
+    {
+      "epoch": 56.09,
+      "learning_rate": 9.345454545454547e-05,
+      "loss": 0.088,
+      "step": 617
+    },
+    {
+      "epoch": 56.18,
+      "learning_rate": 9.34354066985646e-05,
+      "loss": 0.095,
+      "step": 618
+    },
+    {
+      "epoch": 56.27,
+      "learning_rate": 9.341626794258373e-05,
+      "loss": 0.1222,
+      "step": 619
+    },
+    {
+      "epoch": 56.36,
+      "learning_rate": 9.339712918660288e-05,
+      "loss": 0.0855,
+      "step": 620
+    },
+    {
+      "epoch": 56.36,
+      "eval_accuracy_NAT": 0.9907137306494883,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9243140800034336,
+      "eval_accuracy_concrete tiles": 0.988595364764875,
+      "eval_accuracy_dark bricks": 0.7492013394403603,
+      "eval_accuracy_grass": 0.8642024067077454,
+      "eval_accuracy_red bricks": 0.5704900568181818,
+      "eval_accuracy_road": 0.9746235430544857,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6228875472114245,
+      "eval_accuracy_speedway bricks": 0.9847446438506107,
+      "eval_accuracy_steel": 0.8373045973220764,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9733679941265038,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8603332775706791,
+      "eval_iou_concrete tiles": 0.8990879261442634,
+      "eval_iou_dark bricks": 0.7226925076111977,
+      "eval_iou_grass": 0.7875295009443487,
+      "eval_iou_red bricks": 0.5659679408138101,
+      "eval_iou_road": 0.9618276612222834,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4005886316088082,
+      "eval_iou_speedway bricks": 0.9632398467053032,
+      "eval_iou_steel": 0.7601763552255341,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11297067254781723,
+      "eval_mean_accuracy": 0.6543905622940525,
+      "eval_mean_iou": 0.6072932032286716,
+      "eval_overall_accuracy": 0.9714973237779405,
+      "eval_runtime": 4.2977,
+      "eval_samples_per_second": 8.376,
+      "eval_steps_per_second": 0.465,
+      "step": 620
+    },
+    {
+      "epoch": 56.45,
+      "learning_rate": 9.337799043062201e-05,
+      "loss": 0.0827,
+      "step": 621
+    },
+    {
+      "epoch": 56.55,
+      "learning_rate": 9.335885167464115e-05,
+      "loss": 0.1103,
+      "step": 622
+    },
+    {
+      "epoch": 56.64,
+      "learning_rate": 9.33397129186603e-05,
+      "loss": 0.109,
+      "step": 623
+    },
+    {
+      "epoch": 56.73,
+      "learning_rate": 9.332057416267943e-05,
+      "loss": 0.1005,
+      "step": 624
+    },
+    {
+      "epoch": 56.82,
+      "learning_rate": 9.330143540669856e-05,
+      "loss": 0.0995,
+      "step": 625
+    },
+    {
+      "epoch": 56.91,
+      "learning_rate": 9.328229665071771e-05,
+      "loss": 0.0944,
+      "step": 626
+    },
+    {
+      "epoch": 57.0,
+      "learning_rate": 9.326315789473684e-05,
+      "loss": 0.1083,
+      "step": 627
+    },
+    {
+      "epoch": 57.09,
+      "learning_rate": 9.324401913875599e-05,
+      "loss": 0.0935,
+      "step": 628
+    },
+    {
+      "epoch": 57.18,
+      "learning_rate": 9.322488038277513e-05,
+      "loss": 0.1,
+      "step": 629
+    },
+    {
+      "epoch": 57.27,
+      "learning_rate": 9.320574162679426e-05,
+      "loss": 0.0901,
+      "step": 630
+    },
+    {
+      "epoch": 57.36,
+      "learning_rate": 9.31866028708134e-05,
+      "loss": 0.0952,
+      "step": 631
+    },
+    {
+      "epoch": 57.45,
+      "learning_rate": 9.316746411483254e-05,
+      "loss": 0.0998,
+      "step": 632
+    },
+    {
+      "epoch": 57.55,
+      "learning_rate": 9.314832535885169e-05,
+      "loss": 0.0855,
+      "step": 633
+    },
+    {
+      "epoch": 57.64,
+      "learning_rate": 9.312918660287082e-05,
+      "loss": 0.1009,
+      "step": 634
+    },
+    {
+      "epoch": 57.73,
+      "learning_rate": 9.311004784688996e-05,
+      "loss": 0.0942,
+      "step": 635
+    },
+    {
+      "epoch": 57.82,
+      "learning_rate": 9.30909090909091e-05,
+      "loss": 0.1119,
+      "step": 636
+    },
+    {
+      "epoch": 57.91,
+      "learning_rate": 9.307177033492823e-05,
+      "loss": 0.1179,
+      "step": 637
+    },
+    {
+      "epoch": 58.0,
+      "learning_rate": 9.305263157894737e-05,
+      "loss": 0.0923,
+      "step": 638
+    },
+    {
+      "epoch": 58.09,
+      "learning_rate": 9.303349282296652e-05,
+      "loss": 0.0918,
+      "step": 639
+    },
+    {
+      "epoch": 58.18,
+      "learning_rate": 9.301435406698565e-05,
+      "loss": 0.0847,
+      "step": 640
+    },
+    {
+      "epoch": 58.18,
+      "eval_accuracy_NAT": 0.9867044083841101,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.929539626164213,
+      "eval_accuracy_concrete tiles": 0.9964838255977496,
+      "eval_accuracy_dark bricks": 0.7726415457449675,
+      "eval_accuracy_grass": 0.8989748948856586,
+      "eval_accuracy_red bricks": 0.7141335227272727,
+      "eval_accuracy_road": 0.9702232320459693,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5952361260441651,
+      "eval_accuracy_speedway bricks": 0.9871739636772637,
+      "eval_accuracy_steel": 0.8400171492645604,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9737533146614082,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8514518794267426,
+      "eval_iou_concrete tiles": 0.8494318181818182,
+      "eval_iou_dark bricks": 0.7351497839302717,
+      "eval_iou_grass": 0.7935188698745299,
+      "eval_iou_red bricks": 0.7004528039010798,
+      "eval_iou_road": 0.9553543575223431,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.38963077922546646,
+      "eval_iou_speedway bricks": 0.9620433449276555,
+      "eval_iou_steel": 0.7626961312732063,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11372082680463791,
+      "eval_mean_accuracy": 0.6685483303489176,
+      "eval_mean_iou": 0.6133448525326556,
+      "eval_overall_accuracy": 0.9707038667466905,
+      "eval_runtime": 4.0488,
+      "eval_samples_per_second": 8.891,
+      "eval_steps_per_second": 0.494,
+      "step": 640
+    },
+    {
+      "epoch": 58.27,
+      "learning_rate": 9.29952153110048e-05,
+      "loss": 0.1006,
+      "step": 641
+    },
+    {
+      "epoch": 58.36,
+      "learning_rate": 9.297607655502393e-05,
+      "loss": 0.0896,
+      "step": 642
+    },
+    {
+      "epoch": 58.45,
+      "learning_rate": 9.295693779904306e-05,
+      "loss": 0.1217,
+      "step": 643
+    },
+    {
+      "epoch": 58.55,
+      "learning_rate": 9.29377990430622e-05,
+      "loss": 0.0789,
+      "step": 644
+    },
+    {
+      "epoch": 58.64,
+      "learning_rate": 9.291866028708135e-05,
+      "loss": 0.0953,
+      "step": 645
+    },
+    {
+      "epoch": 58.73,
+      "learning_rate": 9.289952153110048e-05,
+      "loss": 0.1015,
+      "step": 646
+    },
+    {
+      "epoch": 58.82,
+      "learning_rate": 9.288038277511963e-05,
+      "loss": 0.0959,
+      "step": 647
+    },
+    {
+      "epoch": 58.91,
+      "learning_rate": 9.286124401913876e-05,
+      "loss": 0.0949,
+      "step": 648
+    },
+    {
+      "epoch": 59.0,
+      "learning_rate": 9.28421052631579e-05,
+      "loss": 0.0946,
+      "step": 649
+    },
+    {
+      "epoch": 59.09,
+      "learning_rate": 9.282296650717704e-05,
+      "loss": 0.1096,
+      "step": 650
+    },
+    {
+      "epoch": 59.18,
+      "learning_rate": 9.280382775119618e-05,
+      "loss": 0.0929,
+      "step": 651
+    },
+    {
+      "epoch": 59.27,
+      "learning_rate": 9.278468899521532e-05,
+      "loss": 0.0974,
+      "step": 652
+    },
+    {
+      "epoch": 59.36,
+      "learning_rate": 9.276555023923445e-05,
+      "loss": 0.0871,
+      "step": 653
+    },
+    {
+      "epoch": 59.45,
+      "learning_rate": 9.27464114832536e-05,
+      "loss": 0.085,
+      "step": 654
+    },
+    {
+      "epoch": 59.55,
+      "learning_rate": 9.272727272727273e-05,
+      "loss": 0.0844,
+      "step": 655
+    },
+    {
+      "epoch": 59.64,
+      "learning_rate": 9.270813397129187e-05,
+      "loss": 0.0938,
+      "step": 656
+    },
+    {
+      "epoch": 59.73,
+      "learning_rate": 9.268899521531102e-05,
+      "loss": 0.1136,
+      "step": 657
+    },
+    {
+      "epoch": 59.82,
+      "learning_rate": 9.266985645933015e-05,
+      "loss": 0.1001,
+      "step": 658
+    },
+    {
+      "epoch": 59.91,
+      "learning_rate": 9.265071770334928e-05,
+      "loss": 0.0971,
+      "step": 659
+    },
+    {
+      "epoch": 60.0,
+      "learning_rate": 9.263157894736843e-05,
+      "loss": 0.0838,
+      "step": 660
+    },
+    {
+      "epoch": 60.0,
+      "eval_accuracy_NAT": 0.990589509063588,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9347812674363707,
+      "eval_accuracy_concrete tiles": 0.9723903870849385,
+      "eval_accuracy_dark bricks": 0.7611331357530503,
+      "eval_accuracy_grass": 0.8680050468864136,
+      "eval_accuracy_red bricks": 0.6459517045454546,
+      "eval_accuracy_road": 0.9749935609628688,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5930030601273675,
+      "eval_accuracy_speedway bricks": 0.9821904186081313,
+      "eval_accuracy_steel": 0.8720318580568565,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9744622757692344,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8627101712811613,
+      "eval_iou_concrete tiles": 0.9480132351626077,
+      "eval_iou_dark bricks": 0.7324074074074074,
+      "eval_iou_grass": 0.7981687347943603,
+      "eval_iou_red bricks": 0.6398170946183609,
+      "eval_iou_road": 0.9634851499345772,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3859265106932683,
+      "eval_iou_speedway bricks": 0.9634396522537922,
+      "eval_iou_steel": 0.766105549841371,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11036855727434158,
+      "eval_mean_accuracy": 0.6611592268096185,
+      "eval_mean_iou": 0.6180412139812416,
+      "eval_overall_accuracy": 0.9723020129733615,
+      "eval_runtime": 3.425,
+      "eval_samples_per_second": 10.511,
+      "eval_steps_per_second": 0.584,
+      "step": 660
+    },
+    {
+      "epoch": 60.09,
+      "learning_rate": 9.261244019138756e-05,
+      "loss": 0.0843,
+      "step": 661
+    },
+    {
+      "epoch": 60.18,
+      "learning_rate": 9.25933014354067e-05,
+      "loss": 0.0813,
+      "step": 662
+    },
+    {
+      "epoch": 60.27,
+      "learning_rate": 9.257416267942585e-05,
+      "loss": 0.0905,
+      "step": 663
+    },
+    {
+      "epoch": 60.36,
+      "learning_rate": 9.255502392344498e-05,
+      "loss": 0.0953,
+      "step": 664
+    },
+    {
+      "epoch": 60.45,
+      "learning_rate": 9.253588516746411e-05,
+      "loss": 0.0846,
+      "step": 665
+    },
+    {
+      "epoch": 60.55,
+      "learning_rate": 9.251674641148326e-05,
+      "loss": 0.0923,
+      "step": 666
+    },
+    {
+      "epoch": 60.64,
+      "learning_rate": 9.249760765550239e-05,
+      "loss": 0.1094,
+      "step": 667
+    },
+    {
+      "epoch": 60.73,
+      "learning_rate": 9.247846889952154e-05,
+      "loss": 0.0775,
+      "step": 668
+    },
+    {
+      "epoch": 60.82,
+      "learning_rate": 9.245933014354068e-05,
+      "loss": 0.095,
+      "step": 669
+    },
+    {
+      "epoch": 60.91,
+      "learning_rate": 9.244019138755981e-05,
+      "loss": 0.0894,
+      "step": 670
+    },
+    {
+      "epoch": 61.0,
+      "learning_rate": 9.242105263157895e-05,
+      "loss": 0.1015,
+      "step": 671
+    },
+    {
+      "epoch": 61.09,
+      "learning_rate": 9.240191387559809e-05,
+      "loss": 0.0972,
+      "step": 672
+    },
+    {
+      "epoch": 61.18,
+      "learning_rate": 9.238277511961722e-05,
+      "loss": 0.101,
+      "step": 673
+    },
+    {
+      "epoch": 61.27,
+      "learning_rate": 9.236363636363636e-05,
+      "loss": 0.0883,
+      "step": 674
+    },
+    {
+      "epoch": 61.36,
+      "learning_rate": 9.234449760765552e-05,
+      "loss": 0.0853,
+      "step": 675
+    },
+    {
+      "epoch": 61.45,
+      "learning_rate": 9.232535885167465e-05,
+      "loss": 0.0965,
+      "step": 676
+    },
+    {
+      "epoch": 61.55,
+      "learning_rate": 9.230622009569378e-05,
+      "loss": 0.1064,
+      "step": 677
+    },
+    {
+      "epoch": 61.64,
+      "learning_rate": 9.228708133971293e-05,
+      "loss": 0.0883,
+      "step": 678
+    },
+    {
+      "epoch": 61.73,
+      "learning_rate": 9.226794258373206e-05,
+      "loss": 0.1067,
+      "step": 679
+    },
+    {
+      "epoch": 61.82,
+      "learning_rate": 9.224880382775119e-05,
+      "loss": 0.0823,
+      "step": 680
+    },
+    {
+      "epoch": 61.82,
+      "eval_accuracy_NAT": 0.9878933360059796,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9405781900510751,
+      "eval_accuracy_concrete tiles": 0.9945575735339082,
+      "eval_accuracy_dark bricks": 0.814133405180709,
+      "eval_accuracy_grass": 0.8956126339923878,
+      "eval_accuracy_red bricks": 0.6695667613636364,
+      "eval_accuracy_road": 0.9646717542113721,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5996195517326938,
+      "eval_accuracy_speedway bricks": 0.9835161055577479,
+      "eval_accuracy_steel": 0.8631769012598114,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9747093810737841,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8535087740650856,
+      "eval_iou_concrete tiles": 0.885766413419383,
+      "eval_iou_dark bricks": 0.7665990142070165,
+      "eval_iou_grass": 0.8000237275329922,
+      "eval_iou_red bricks": 0.6571976298361799,
+      "eval_iou_road": 0.9536201948190889,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.39167311951882733,
+      "eval_iou_speedway bricks": 0.9642857142857143,
+      "eval_iou_steel": 0.7689533602644143,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11057078093290329,
+      "eval_mean_accuracy": 0.6702558625299478,
+      "eval_mean_iou": 0.6166413330017296,
+      "eval_overall_accuracy": 0.9714546203613281,
+      "eval_runtime": 3.6282,
+      "eval_samples_per_second": 9.922,
+      "eval_steps_per_second": 0.551,
+      "step": 680
+    },
+    {
+      "epoch": 61.91,
+      "learning_rate": 9.222966507177035e-05,
+      "loss": 0.0907,
+      "step": 681
+    },
+    {
+      "epoch": 62.0,
+      "learning_rate": 9.221052631578948e-05,
+      "loss": 0.091,
+      "step": 682
+    },
+    {
+      "epoch": 62.09,
+      "learning_rate": 9.219138755980861e-05,
+      "loss": 0.0752,
+      "step": 683
+    },
+    {
+      "epoch": 62.18,
+      "learning_rate": 9.217224880382776e-05,
+      "loss": 0.0887,
+      "step": 684
+    },
+    {
+      "epoch": 62.27,
+      "learning_rate": 9.215311004784689e-05,
+      "loss": 0.0938,
+      "step": 685
+    },
+    {
+      "epoch": 62.36,
+      "learning_rate": 9.213397129186602e-05,
+      "loss": 0.081,
+      "step": 686
+    },
+    {
+      "epoch": 62.45,
+      "learning_rate": 9.211483253588518e-05,
+      "loss": 0.0943,
+      "step": 687
+    },
+    {
+      "epoch": 62.55,
+      "learning_rate": 9.209569377990431e-05,
+      "loss": 0.0869,
+      "step": 688
+    },
+    {
+      "epoch": 62.64,
+      "learning_rate": 9.207655502392344e-05,
+      "loss": 0.1057,
+      "step": 689
+    },
+    {
+      "epoch": 62.73,
+      "learning_rate": 9.205741626794259e-05,
+      "loss": 0.0922,
+      "step": 690
+    },
+    {
+      "epoch": 62.82,
+      "learning_rate": 9.203827751196172e-05,
+      "loss": 0.0889,
+      "step": 691
+    },
+    {
+      "epoch": 62.91,
+      "learning_rate": 9.201913875598085e-05,
+      "loss": 0.0726,
+      "step": 692
+    },
+    {
+      "epoch": 63.0,
+      "learning_rate": 9.200000000000001e-05,
+      "loss": 0.0962,
+      "step": 693
+    },
+    {
+      "epoch": 63.09,
+      "learning_rate": 9.198086124401915e-05,
+      "loss": 0.0973,
+      "step": 694
+    },
+    {
+      "epoch": 63.18,
+      "learning_rate": 9.196172248803828e-05,
+      "loss": 0.0767,
+      "step": 695
+    },
+    {
+      "epoch": 63.27,
+      "learning_rate": 9.194258373205742e-05,
+      "loss": 0.0916,
+      "step": 696
+    },
+    {
+      "epoch": 63.36,
+      "learning_rate": 9.192344497607656e-05,
+      "loss": 0.0858,
+      "step": 697
+    },
+    {
+      "epoch": 63.45,
+      "learning_rate": 9.190430622009569e-05,
+      "loss": 0.1245,
+      "step": 698
+    },
+    {
+      "epoch": 63.55,
+      "learning_rate": 9.188516746411485e-05,
+      "loss": 0.0889,
+      "step": 699
+    },
+    {
+      "epoch": 63.64,
+      "learning_rate": 9.186602870813398e-05,
+      "loss": 0.101,
+      "step": 700
+    },
+    {
+      "epoch": 63.64,
+      "eval_accuracy_NAT": 0.9882303576381807,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9374825636293403,
+      "eval_accuracy_concrete tiles": 0.992723047758821,
+      "eval_accuracy_dark bricks": 0.8190985720334091,
+      "eval_accuracy_grass": 0.8962766979941772,
+      "eval_accuracy_red bricks": 0.7782315340909091,
+      "eval_accuracy_road": 0.9652920783518967,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5543241529512309,
+      "eval_accuracy_speedway bricks": 0.9859792162215327,
+      "eval_accuracy_steel": 0.8542807202691115,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9748528388379446,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8564411028797306,
+      "eval_iou_concrete tiles": 0.8934015739364922,
+      "eval_iou_dark bricks": 0.7666894837338329,
+      "eval_iou_grass": 0.800481962310797,
+      "eval_iou_red bricks": 0.7569948186528498,
+      "eval_iou_road": 0.9565534149269899,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.36519006883524946,
+      "eval_iou_speedway bricks": 0.9626962267319062,
+      "eval_iou_steel": 0.7745219692326093,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10857050120830536,
+      "eval_mean_accuracy": 0.6747629954568161,
+      "eval_mean_iou": 0.6236787276983388,
+      "eval_overall_accuracy": 0.9717032114664713,
+      "eval_runtime": 4.5864,
+      "eval_samples_per_second": 7.849,
+      "eval_steps_per_second": 0.436,
+      "step": 700
+    },
+    {
+      "epoch": 63.73,
+      "learning_rate": 9.184688995215311e-05,
+      "loss": 0.0974,
+      "step": 701
+    },
+    {
+      "epoch": 63.82,
+      "learning_rate": 9.182775119617226e-05,
+      "loss": 0.0957,
+      "step": 702
+    },
+    {
+      "epoch": 63.91,
+      "learning_rate": 9.180861244019139e-05,
+      "loss": 0.0837,
+      "step": 703
+    },
+    {
+      "epoch": 64.0,
+      "learning_rate": 9.178947368421052e-05,
+      "loss": 0.0861,
+      "step": 704
+    },
+    {
+      "epoch": 64.09,
+      "learning_rate": 9.177033492822968e-05,
+      "loss": 0.0912,
+      "step": 705
+    },
+    {
+      "epoch": 64.18,
+      "learning_rate": 9.175119617224881e-05,
+      "loss": 0.0967,
+      "step": 706
+    },
+    {
+      "epoch": 64.27,
+      "learning_rate": 9.173205741626794e-05,
+      "loss": 0.0947,
+      "step": 707
+    },
+    {
+      "epoch": 64.36,
+      "learning_rate": 9.171291866028709e-05,
+      "loss": 0.1158,
+      "step": 708
+    },
+    {
+      "epoch": 64.45,
+      "learning_rate": 9.169377990430622e-05,
+      "loss": 0.0964,
+      "step": 709
+    },
+    {
+      "epoch": 64.55,
+      "learning_rate": 9.167464114832537e-05,
+      "loss": 0.0853,
+      "step": 710
+    },
+    {
+      "epoch": 64.64,
+      "learning_rate": 9.165550239234451e-05,
+      "loss": 0.0927,
+      "step": 711
+    },
+    {
+      "epoch": 64.73,
+      "learning_rate": 9.163636363636364e-05,
+      "loss": 0.083,
+      "step": 712
+    },
+    {
+      "epoch": 64.82,
+      "learning_rate": 9.161722488038278e-05,
+      "loss": 0.0918,
+      "step": 713
+    },
+    {
+      "epoch": 64.91,
+      "learning_rate": 9.159808612440192e-05,
+      "loss": 0.0837,
+      "step": 714
+    },
+    {
+      "epoch": 65.0,
+      "learning_rate": 9.157894736842105e-05,
+      "loss": 0.0882,
+      "step": 715
+    },
+    {
+      "epoch": 65.09,
+      "learning_rate": 9.15598086124402e-05,
+      "loss": 0.0932,
+      "step": 716
+    },
+    {
+      "epoch": 65.18,
+      "learning_rate": 9.154066985645934e-05,
+      "loss": 0.0778,
+      "step": 717
+    },
+    {
+      "epoch": 65.27,
+      "learning_rate": 9.152153110047848e-05,
+      "loss": 0.0758,
+      "step": 718
+    },
+    {
+      "epoch": 65.36,
+      "learning_rate": 9.150239234449761e-05,
+      "loss": 0.0803,
+      "step": 719
+    },
+    {
+      "epoch": 65.45,
+      "learning_rate": 9.148325358851675e-05,
+      "loss": 0.1006,
+      "step": 720
+    },
+    {
+      "epoch": 65.45,
+      "eval_accuracy_NAT": 0.9898000667691024,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9418644577020473,
+      "eval_accuracy_concrete tiles": 0.9800648199107197,
+      "eval_accuracy_dark bricks": 0.7655209576228783,
+      "eval_accuracy_grass": 0.8681098990972225,
+      "eval_accuracy_red bricks": 0.7223011363636364,
+      "eval_accuracy_road": 0.9716174498510859,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5196702781683346,
+      "eval_accuracy_speedway bricks": 0.9854500275757367,
+      "eval_accuracy_steel": 0.8342540069916232,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9743002589166629,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.85825470545099,
+      "eval_iou_concrete tiles": 0.9227093468436052,
+      "eval_iou_dark bricks": 0.7324789157735794,
+      "eval_iou_grass": 0.7957639309388106,
+      "eval_iou_red bricks": 0.7145617424907782,
+      "eval_iou_road": 0.9585647684217559,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.35379793164286116,
+      "eval_iou_speedway bricks": 0.9652743997404091,
+      "eval_iou_steel": 0.7741538131962297,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10650945454835892,
+      "eval_mean_accuracy": 0.659896392311722,
+      "eval_mean_iou": 0.6192199856473601,
+      "eval_overall_accuracy": 0.9719841215345595,
+      "eval_runtime": 4.4686,
+      "eval_samples_per_second": 8.056,
+      "eval_steps_per_second": 0.448,
+      "step": 720
+    },
+    {
+      "epoch": 65.55,
+      "learning_rate": 9.146411483253589e-05,
+      "loss": 0.0804,
+      "step": 721
+    },
+    {
+      "epoch": 65.64,
+      "learning_rate": 9.144497607655503e-05,
+      "loss": 0.1121,
+      "step": 722
+    },
+    {
+      "epoch": 65.73,
+      "learning_rate": 9.142583732057418e-05,
+      "loss": 0.0875,
+      "step": 723
+    },
+    {
+      "epoch": 65.82,
+      "learning_rate": 9.140669856459331e-05,
+      "loss": 0.079,
+      "step": 724
+    },
+    {
+      "epoch": 65.91,
+      "learning_rate": 9.138755980861244e-05,
+      "loss": 0.0836,
+      "step": 725
+    },
+    {
+      "epoch": 66.0,
+      "learning_rate": 9.136842105263159e-05,
+      "loss": 0.0858,
+      "step": 726
+    },
+    {
+      "epoch": 66.09,
+      "learning_rate": 9.134928229665072e-05,
+      "loss": 0.0749,
+      "step": 727
+    },
+    {
+      "epoch": 66.18,
+      "learning_rate": 9.133014354066986e-05,
+      "loss": 0.0956,
+      "step": 728
+    },
+    {
+      "epoch": 66.27,
+      "learning_rate": 9.1311004784689e-05,
+      "loss": 0.0911,
+      "step": 729
+    },
+    {
+      "epoch": 66.36,
+      "learning_rate": 9.129186602870814e-05,
+      "loss": 0.081,
+      "step": 730
+    },
+    {
+      "epoch": 66.45,
+      "learning_rate": 9.127272727272727e-05,
+      "loss": 0.0742,
+      "step": 731
+    },
+    {
+      "epoch": 66.55,
+      "learning_rate": 9.125358851674642e-05,
+      "loss": 0.0907,
+      "step": 732
+    },
+    {
+      "epoch": 66.64,
+      "learning_rate": 9.123444976076555e-05,
+      "loss": 0.0807,
+      "step": 733
+    },
+    {
+      "epoch": 66.73,
+      "learning_rate": 9.12153110047847e-05,
+      "loss": 0.0779,
+      "step": 734
+    },
+    {
+      "epoch": 66.82,
+      "learning_rate": 9.119617224880383e-05,
+      "loss": 0.0876,
+      "step": 735
+    },
+    {
+      "epoch": 66.91,
+      "learning_rate": 9.117703349282297e-05,
+      "loss": 0.0731,
+      "step": 736
+    },
+    {
+      "epoch": 67.0,
+      "learning_rate": 9.11578947368421e-05,
+      "loss": 0.0801,
+      "step": 737
+    },
+    {
+      "epoch": 67.09,
+      "learning_rate": 9.113875598086125e-05,
+      "loss": 0.0736,
+      "step": 738
+    },
+    {
+      "epoch": 67.18,
+      "learning_rate": 9.111961722488038e-05,
+      "loss": 0.0838,
+      "step": 739
+    },
+    {
+      "epoch": 67.27,
+      "learning_rate": 9.110047846889953e-05,
+      "loss": 0.0717,
+      "step": 740
+    },
+    {
+      "epoch": 67.27,
+      "eval_accuracy_NAT": 0.990716200965117,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9394153182540023,
+      "eval_accuracy_concrete tiles": 0.9859964532501682,
+      "eval_accuracy_dark bricks": 0.7996997806089066,
+      "eval_accuracy_grass": 0.8864869965783229,
+      "eval_accuracy_red bricks": 0.78515625,
+      "eval_accuracy_road": 0.9727359680381916,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5673090177266837,
+      "eval_accuracy_speedway bricks": 0.9853124505959859,
+      "eval_accuracy_steel": 0.8620885825473253,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9755388923532303,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8705982085812093,
+      "eval_iou_concrete tiles": 0.9317269076305221,
+      "eval_iou_dark bricks": 0.7594210314704485,
+      "eval_iou_grass": 0.8046105725307076,
+      "eval_iou_red bricks": 0.7669094693028096,
+      "eval_iou_road": 0.9656992147781607,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3812647064272877,
+      "eval_iou_speedway bricks": 0.9662504408419668,
+      "eval_iou_steel": 0.7817001966193434,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10277089476585388,
+      "eval_mean_accuracy": 0.6749936168126696,
+      "eval_mean_iou": 0.6310553569642837,
+      "eval_overall_accuracy": 0.973664813571506,
+      "eval_runtime": 4.0339,
+      "eval_samples_per_second": 8.924,
+      "eval_steps_per_second": 0.496,
+      "step": 740
+    },
+    {
+      "epoch": 67.36,
+      "learning_rate": 9.108133971291866e-05,
+      "loss": 0.0908,
+      "step": 741
+    },
+    {
+      "epoch": 67.45,
+      "learning_rate": 9.106220095693781e-05,
+      "loss": 0.0722,
+      "step": 742
+    },
+    {
+      "epoch": 67.55,
+      "learning_rate": 9.104306220095694e-05,
+      "loss": 0.0711,
+      "step": 743
+    },
+    {
+      "epoch": 67.64,
+      "learning_rate": 9.102392344497608e-05,
+      "loss": 0.0755,
+      "step": 744
+    },
+    {
+      "epoch": 67.73,
+      "learning_rate": 9.100478468899522e-05,
+      "loss": 0.0821,
+      "step": 745
+    },
+    {
+      "epoch": 67.82,
+      "learning_rate": 9.098564593301436e-05,
+      "loss": 0.087,
+      "step": 746
+    },
+    {
+      "epoch": 67.91,
+      "learning_rate": 9.09665071770335e-05,
+      "loss": 0.0858,
+      "step": 747
+    },
+    {
+      "epoch": 68.0,
+      "learning_rate": 9.094736842105264e-05,
+      "loss": 0.091,
+      "step": 748
+    },
+    {
+      "epoch": 68.09,
+      "learning_rate": 9.092822966507177e-05,
+      "loss": 0.0764,
+      "step": 749
+    },
+    {
+      "epoch": 68.18,
+      "learning_rate": 9.090909090909092e-05,
+      "loss": 0.0764,
+      "step": 750
+    },
+    {
+      "epoch": 68.27,
+      "learning_rate": 9.088995215311005e-05,
+      "loss": 0.0768,
+      "step": 751
+    },
+    {
+      "epoch": 68.36,
+      "learning_rate": 9.08708133971292e-05,
+      "loss": 0.0831,
+      "step": 752
+    },
+    {
+      "epoch": 68.45,
+      "learning_rate": 9.085167464114833e-05,
+      "loss": 0.1088,
+      "step": 753
+    },
+    {
+      "epoch": 68.55,
+      "learning_rate": 9.083253588516747e-05,
+      "loss": 0.0728,
+      "step": 754
+    },
+    {
+      "epoch": 68.64,
+      "learning_rate": 9.08133971291866e-05,
+      "loss": 0.0831,
+      "step": 755
+    },
+    {
+      "epoch": 68.73,
+      "learning_rate": 9.079425837320574e-05,
+      "loss": 0.0762,
+      "step": 756
+    },
+    {
+      "epoch": 68.82,
+      "learning_rate": 9.077511961722488e-05,
+      "loss": 0.0727,
+      "step": 757
+    },
+    {
+      "epoch": 68.91,
+      "learning_rate": 9.075598086124403e-05,
+      "loss": 0.0829,
+      "step": 758
+    },
+    {
+      "epoch": 69.0,
+      "learning_rate": 9.073684210526316e-05,
+      "loss": 0.0778,
+      "step": 759
+    },
+    {
+      "epoch": 69.09,
+      "learning_rate": 9.07177033492823e-05,
+      "loss": 0.0757,
+      "step": 760
+    },
+    {
+      "epoch": 69.09,
+      "eval_accuracy_NAT": 0.9892010152291429,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9321282565775355,
+      "eval_accuracy_concrete tiles": 0.9883507613281967,
+      "eval_accuracy_dark bricks": 0.7977752973326662,
+      "eval_accuracy_grass": 0.8776374699860546,
+      "eval_accuracy_red bricks": 0.7638494318181818,
+      "eval_accuracy_road": 0.9708616943192577,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6121081796377471,
+      "eval_accuracy_speedway bricks": 0.9885008574424922,
+      "eval_accuracy_steel": 0.8403634324912604,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9752679812332677,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8614699028163427,
+      "eval_iou_concrete tiles": 0.9113335212855934,
+      "eval_iou_dark bricks": 0.7579536312440577,
+      "eval_iou_grass": 0.784861441904369,
+      "eval_iou_red bricks": 0.7424922333448395,
+      "eval_iou_road": 0.957650084208417,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3882729434806939,
+      "eval_iou_speedway bricks": 0.9649309226704992,
+      "eval_iou_steel": 0.7797932812583678,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10357774794101715,
+      "eval_mean_accuracy": 0.6739058766278874,
+      "eval_mean_iou": 0.6249250725728038,
+      "eval_overall_accuracy": 0.9721896913316515,
+      "eval_runtime": 4.2296,
+      "eval_samples_per_second": 8.511,
+      "eval_steps_per_second": 0.473,
+      "step": 760
+    },
+    {
+      "epoch": 69.18,
+      "learning_rate": 9.069856459330144e-05,
+      "loss": 0.0923,
+      "step": 761
+    },
+    {
+      "epoch": 69.27,
+      "learning_rate": 9.067942583732057e-05,
+      "loss": 0.0806,
+      "step": 762
+    },
+    {
+      "epoch": 69.36,
+      "learning_rate": 9.066028708133971e-05,
+      "loss": 0.0754,
+      "step": 763
+    },
+    {
+      "epoch": 69.45,
+      "learning_rate": 9.064114832535886e-05,
+      "loss": 0.0766,
+      "step": 764
+    },
+    {
+      "epoch": 69.55,
+      "learning_rate": 9.062200956937799e-05,
+      "loss": 0.0759,
+      "step": 765
+    },
+    {
+      "epoch": 69.64,
+      "learning_rate": 9.060287081339714e-05,
+      "loss": 0.0758,
+      "step": 766
+    },
+    {
+      "epoch": 69.73,
+      "learning_rate": 9.058373205741627e-05,
+      "loss": 0.0848,
+      "step": 767
+    },
+    {
+      "epoch": 69.82,
+      "learning_rate": 9.05645933014354e-05,
+      "loss": 0.0877,
+      "step": 768
+    },
+    {
+      "epoch": 69.91,
+      "learning_rate": 9.054545454545455e-05,
+      "loss": 0.0952,
+      "step": 769
+    },
+    {
+      "epoch": 70.0,
+      "learning_rate": 9.052631578947369e-05,
+      "loss": 0.084,
+      "step": 770
+    },
+    {
+      "epoch": 70.09,
+      "learning_rate": 9.050717703349282e-05,
+      "loss": 0.0773,
+      "step": 771
+    },
+    {
+      "epoch": 70.18,
+      "learning_rate": 9.048803827751197e-05,
+      "loss": 0.0775,
+      "step": 772
+    },
+    {
+      "epoch": 70.27,
+      "learning_rate": 9.04688995215311e-05,
+      "loss": 0.0769,
+      "step": 773
+    },
+    {
+      "epoch": 70.36,
+      "learning_rate": 9.044976076555023e-05,
+      "loss": 0.0834,
+      "step": 774
+    },
+    {
+      "epoch": 70.45,
+      "learning_rate": 9.043062200956938e-05,
+      "loss": 0.0762,
+      "step": 775
+    },
+    {
+      "epoch": 70.55,
+      "learning_rate": 9.041148325358853e-05,
+      "loss": 0.0788,
+      "step": 776
+    },
+    {
+      "epoch": 70.64,
+      "learning_rate": 9.039234449760766e-05,
+      "loss": 0.0826,
+      "step": 777
+    },
+    {
+      "epoch": 70.73,
+      "learning_rate": 9.03732057416268e-05,
+      "loss": 0.0709,
+      "step": 778
+    },
+    {
+      "epoch": 70.82,
+      "learning_rate": 9.035406698564593e-05,
+      "loss": 0.0808,
+      "step": 779
+    },
+    {
+      "epoch": 70.91,
+      "learning_rate": 9.033492822966507e-05,
+      "loss": 0.0728,
+      "step": 780
+    },
+    {
+      "epoch": 70.91,
+      "eval_accuracy_NAT": 0.9914297692795784,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9278979248036396,
+      "eval_accuracy_concrete tiles": 0.9925090197517276,
+      "eval_accuracy_dark bricks": 0.787460066972018,
+      "eval_accuracy_grass": 0.8662679952606801,
+      "eval_accuracy_red bricks": 0.7787642045454546,
+      "eval_accuracy_road": 0.9702994122035775,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5826923607090674,
+      "eval_accuracy_speedway bricks": 0.9879246029878341,
+      "eval_accuracy_steel": 0.8526235076841897,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9755445497280615,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.860988316233317,
+      "eval_iou_concrete tiles": 0.8804176837537293,
+      "eval_iou_dark bricks": 0.7544713648264926,
+      "eval_iou_grass": 0.7960520821958286,
+      "eval_iou_red bricks": 0.7597436341590161,
+      "eval_iou_road": 0.9619511271164903,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.38639147364764814,
+      "eval_iou_speedway bricks": 0.9645996531018165,
+      "eval_iou_steel": 0.7835327276997796,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10247956216335297,
+      "eval_mean_accuracy": 0.6721437587844437,
+      "eval_mean_iou": 0.6248994317278599,
+      "eval_overall_accuracy": 0.972738160027398,
+      "eval_runtime": 3.4385,
+      "eval_samples_per_second": 10.47,
+      "eval_steps_per_second": 0.582,
+      "step": 780
+    },
+    {
+      "epoch": 71.0,
+      "learning_rate": 9.031578947368423e-05,
+      "loss": 0.0747,
+      "step": 781
+    },
+    {
+      "epoch": 71.09,
+      "learning_rate": 9.029665071770336e-05,
+      "loss": 0.0715,
+      "step": 782
+    },
+    {
+      "epoch": 71.18,
+      "learning_rate": 9.027751196172249e-05,
+      "loss": 0.0775,
+      "step": 783
+    },
+    {
+      "epoch": 71.27,
+      "learning_rate": 9.025837320574164e-05,
+      "loss": 0.0886,
+      "step": 784
+    },
+    {
+      "epoch": 71.36,
+      "learning_rate": 9.023923444976077e-05,
+      "loss": 0.0747,
+      "step": 785
+    },
+    {
+      "epoch": 71.45,
+      "learning_rate": 9.02200956937799e-05,
+      "loss": 0.0824,
+      "step": 786
+    },
+    {
+      "epoch": 71.55,
+      "learning_rate": 9.020095693779906e-05,
+      "loss": 0.0712,
+      "step": 787
+    },
+    {
+      "epoch": 71.64,
+      "learning_rate": 9.018181818181819e-05,
+      "loss": 0.073,
+      "step": 788
+    },
+    {
+      "epoch": 71.73,
+      "learning_rate": 9.016267942583732e-05,
+      "loss": 0.0713,
+      "step": 789
+    },
+    {
+      "epoch": 71.82,
+      "learning_rate": 9.014354066985647e-05,
+      "loss": 0.0835,
+      "step": 790
+    },
+    {
+      "epoch": 71.91,
+      "learning_rate": 9.01244019138756e-05,
+      "loss": 0.0729,
+      "step": 791
+    },
+    {
+      "epoch": 72.0,
+      "learning_rate": 9.010526315789473e-05,
+      "loss": 0.0644,
+      "step": 792
+    },
+    {
+      "epoch": 72.09,
+      "learning_rate": 9.008612440191389e-05,
+      "loss": 0.0713,
+      "step": 793
+    },
+    {
+      "epoch": 72.18,
+      "learning_rate": 9.006698564593302e-05,
+      "loss": 0.0795,
+      "step": 794
+    },
+    {
+      "epoch": 72.27,
+      "learning_rate": 9.004784688995216e-05,
+      "loss": 0.0766,
+      "step": 795
+    },
+    {
+      "epoch": 72.36,
+      "learning_rate": 9.00287081339713e-05,
+      "loss": 0.0775,
+      "step": 796
+    },
+    {
+      "epoch": 72.45,
+      "learning_rate": 9.000956937799043e-05,
+      "loss": 0.0774,
+      "step": 797
+    },
+    {
+      "epoch": 72.55,
+      "learning_rate": 8.999043062200956e-05,
+      "loss": 0.0702,
+      "step": 798
+    },
+    {
+      "epoch": 72.64,
+      "learning_rate": 8.997129186602872e-05,
+      "loss": 0.0783,
+      "step": 799
+    },
+    {
+      "epoch": 72.73,
+      "learning_rate": 8.995215311004786e-05,
+      "loss": 0.0814,
+      "step": 800
+    },
+    {
+      "epoch": 72.73,
+      "eval_accuracy_NAT": 0.9897047831662812,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9412997875445298,
+      "eval_accuracy_concrete tiles": 0.993579159787195,
+      "eval_accuracy_dark bricks": 0.8406912743928255,
+      "eval_accuracy_grass": 0.8837713243183732,
+      "eval_accuracy_red bricks": 0.8765980113636364,
+      "eval_accuracy_road": 0.9714638803270185,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6378573594684752,
+      "eval_accuracy_speedway bricks": 0.9855863977398758,
+      "eval_accuracy_steel": 0.85418178220434,
+      "eval_accuracy_tiles": 0.018079715107519517,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9761939694684522,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8644086376446012,
+      "eval_iou_concrete tiles": 0.8763281376409039,
+      "eval_iou_dark bricks": 0.7828673835125448,
+      "eval_iou_grass": 0.8048136937556296,
+      "eval_iou_red bricks": 0.8284947138781675,
+      "eval_iou_road": 0.9588010153847255,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.41281424518707516,
+      "eval_iou_speedway bricks": 0.9677520077971554,
+      "eval_iou_steel": 0.7899986274420094,
+      "eval_iou_tiles": 0.018079715107519517,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09859243035316467,
+      "eval_mean_accuracy": 0.6917548827246207,
+      "eval_mean_iou": 0.636965549755291,
+      "eval_overall_accuracy": 0.9734518263075087,
+      "eval_runtime": 4.0936,
+      "eval_samples_per_second": 8.794,
+      "eval_steps_per_second": 0.489,
+      "step": 800
+    },
+    {
+      "epoch": 72.82,
+      "learning_rate": 8.993301435406699e-05,
+      "loss": 0.0714,
+      "step": 801
+    },
+    {
+      "epoch": 72.91,
+      "learning_rate": 8.991387559808613e-05,
+      "loss": 0.073,
+      "step": 802
+    },
+    {
+      "epoch": 73.0,
+      "learning_rate": 8.989473684210527e-05,
+      "loss": 0.0769,
+      "step": 803
+    },
+    {
+      "epoch": 73.09,
+      "learning_rate": 8.98755980861244e-05,
+      "loss": 0.0658,
+      "step": 804
+    },
+    {
+      "epoch": 73.18,
+      "learning_rate": 8.985645933014354e-05,
+      "loss": 0.0759,
+      "step": 805
+    },
+    {
+      "epoch": 73.27,
+      "learning_rate": 8.983732057416269e-05,
+      "loss": 0.081,
+      "step": 806
+    },
+    {
+      "epoch": 73.36,
+      "learning_rate": 8.981818181818182e-05,
+      "loss": 0.0683,
+      "step": 807
+    },
+    {
+      "epoch": 73.45,
+      "learning_rate": 8.979904306220097e-05,
+      "loss": 0.0789,
+      "step": 808
+    },
+    {
+      "epoch": 73.55,
+      "learning_rate": 8.97799043062201e-05,
+      "loss": 0.0757,
+      "step": 809
+    },
+    {
+      "epoch": 73.64,
+      "learning_rate": 8.976076555023923e-05,
+      "loss": 0.0751,
+      "step": 810
+    },
+    {
+      "epoch": 73.73,
+      "learning_rate": 8.974162679425838e-05,
+      "loss": 0.0795,
+      "step": 811
+    },
+    {
+      "epoch": 73.82,
+      "learning_rate": 8.972248803827752e-05,
+      "loss": 0.0758,
+      "step": 812
+    },
+    {
+      "epoch": 73.91,
+      "learning_rate": 8.970334928229665e-05,
+      "loss": 0.0815,
+      "step": 813
+    },
+    {
+      "epoch": 74.0,
+      "learning_rate": 8.96842105263158e-05,
+      "loss": 0.0814,
+      "step": 814
+    },
+    {
+      "epoch": 74.09,
+      "learning_rate": 8.966507177033493e-05,
+      "loss": 0.0739,
+      "step": 815
+    },
+    {
+      "epoch": 74.18,
+      "learning_rate": 8.964593301435406e-05,
+      "loss": 0.069,
+      "step": 816
+    },
+    {
+      "epoch": 74.27,
+      "learning_rate": 8.962679425837321e-05,
+      "loss": 0.0644,
+      "step": 817
+    },
+    {
+      "epoch": 74.36,
+      "learning_rate": 8.960765550239235e-05,
+      "loss": 0.0761,
+      "step": 818
+    },
+    {
+      "epoch": 74.45,
+      "learning_rate": 8.958851674641149e-05,
+      "loss": 0.069,
+      "step": 819
+    },
+    {
+      "epoch": 74.55,
+      "learning_rate": 8.956937799043063e-05,
+      "loss": 0.076,
+      "step": 820
+    },
+    {
+      "epoch": 74.55,
+      "eval_accuracy_NAT": 0.9920632287872232,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9309680672990257,
+      "eval_accuracy_concrete tiles": 0.9874029230110684,
+      "eval_accuracy_dark bricks": 0.8415380470343713,
+      "eval_accuracy_grass": 0.8731043594054181,
+      "eval_accuracy_red bricks": 0.8909801136363636,
+      "eval_accuracy_road": 0.967015200964465,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6033137595456676,
+      "eval_accuracy_speedway bricks": 0.9881376059433256,
+      "eval_accuracy_steel": 0.8546434931732735,
+      "eval_accuracy_tiles": 0.020613614573346116,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9760405921470416,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.869267916361924,
+      "eval_iou_concrete tiles": 0.907797829875752,
+      "eval_iou_dark bricks": 0.7832628788421581,
+      "eval_iou_grass": 0.8008219476697591,
+      "eval_iou_red bricks": 0.8301075268817204,
+      "eval_iou_road": 0.9597675077289398,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.40376383763837637,
+      "eval_iou_speedway bricks": 0.9662878054767491,
+      "eval_iou_steel": 0.7895466455426239,
+      "eval_iou_tiles": 0.020613614573346116,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09899807721376419,
+      "eval_mean_accuracy": 0.6884446471825806,
+      "eval_mean_iou": 0.6390213925183377,
+      "eval_overall_accuracy": 0.9736532635158963,
+      "eval_runtime": 4.519,
+      "eval_samples_per_second": 7.966,
+      "eval_steps_per_second": 0.443,
+      "step": 820
+    },
+    {
+      "epoch": 74.64,
+      "learning_rate": 8.955023923444976e-05,
+      "loss": 0.08,
+      "step": 821
+    },
+    {
+      "epoch": 74.73,
+      "learning_rate": 8.95311004784689e-05,
+      "loss": 0.0846,
+      "step": 822
+    },
+    {
+      "epoch": 74.82,
+      "learning_rate": 8.951196172248804e-05,
+      "loss": 0.085,
+      "step": 823
+    },
+    {
+      "epoch": 74.91,
+      "learning_rate": 8.949282296650719e-05,
+      "loss": 0.078,
+      "step": 824
+    },
+    {
+      "epoch": 75.0,
+      "learning_rate": 8.947368421052632e-05,
+      "loss": 0.067,
+      "step": 825
+    },
+    {
+      "epoch": 75.09,
+      "learning_rate": 8.945454545454546e-05,
+      "loss": 0.0705,
+      "step": 826
+    },
+    {
+      "epoch": 75.18,
+      "learning_rate": 8.94354066985646e-05,
+      "loss": 0.0717,
+      "step": 827
+    },
+    {
+      "epoch": 75.27,
+      "learning_rate": 8.941626794258373e-05,
+      "loss": 0.0756,
+      "step": 828
+    },
+    {
+      "epoch": 75.36,
+      "learning_rate": 8.939712918660287e-05,
+      "loss": 0.0814,
+      "step": 829
+    },
+    {
+      "epoch": 75.45,
+      "learning_rate": 8.937799043062202e-05,
+      "loss": 0.0649,
+      "step": 830
+    },
+    {
+      "epoch": 75.55,
+      "learning_rate": 8.935885167464115e-05,
+      "loss": 0.0727,
+      "step": 831
+    },
+    {
+      "epoch": 75.64,
+      "learning_rate": 8.933971291866028e-05,
+      "loss": 0.0818,
+      "step": 832
+    },
+    {
+      "epoch": 75.73,
+      "learning_rate": 8.932057416267943e-05,
+      "loss": 0.0755,
+      "step": 833
+    },
+    {
+      "epoch": 75.82,
+      "learning_rate": 8.930143540669856e-05,
+      "loss": 0.0735,
+      "step": 834
+    },
+    {
+      "epoch": 75.91,
+      "learning_rate": 8.92822966507177e-05,
+      "loss": 0.0729,
+      "step": 835
+    },
+    {
+      "epoch": 76.0,
+      "learning_rate": 8.926315789473685e-05,
+      "loss": 0.066,
+      "step": 836
+    },
+    {
+      "epoch": 76.09,
+      "learning_rate": 8.924401913875598e-05,
+      "loss": 0.0733,
+      "step": 837
+    },
+    {
+      "epoch": 76.18,
+      "learning_rate": 8.922488038277512e-05,
+      "loss": 0.0885,
+      "step": 838
+    },
+    {
+      "epoch": 76.27,
+      "learning_rate": 8.920574162679426e-05,
+      "loss": 0.0833,
+      "step": 839
+    },
+    {
+      "epoch": 76.36,
+      "learning_rate": 8.91866028708134e-05,
+      "loss": 0.0734,
+      "step": 840
+    },
+    {
+      "epoch": 76.36,
+      "eval_accuracy_NAT": 0.9922469143993287,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9355873642645607,
+      "eval_accuracy_concrete tiles": 0.9832446645875375,
+      "eval_accuracy_dark bricks": 0.7988145183018359,
+      "eval_accuracy_grass": 0.8709409087890618,
+      "eval_accuracy_red bricks": 0.7869318181818182,
+      "eval_accuracy_road": 0.981028722337836,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5984616657017616,
+      "eval_accuracy_speedway bricks": 0.9853311562379695,
+      "eval_accuracy_steel": 0.8601098212518963,
+      "eval_accuracy_tiles": 0.006779893165319819,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9757802020847034,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8791178876287243,
+      "eval_iou_concrete tiles": 0.9262897139729815,
+      "eval_iou_dark bricks": 0.7648988316809788,
+      "eval_iou_grass": 0.7994424233091438,
+      "eval_iou_red bricks": 0.7649292371418709,
+      "eval_iou_road": 0.9694780364227331,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4033669658286415,
+      "eval_iou_speedway bricks": 0.9682873534251449,
+      "eval_iou_steel": 0.7923575296789434,
+      "eval_iou_tiles": 0.006779893165319819,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09609613567590714,
+      "eval_mean_accuracy": 0.676882880555302,
+      "eval_mean_iou": 0.6346713903337835,
+      "eval_overall_accuracy": 0.9746341705322266,
+      "eval_runtime": 3.8707,
+      "eval_samples_per_second": 9.301,
+      "eval_steps_per_second": 0.517,
+      "step": 840
+    },
+    {
+      "epoch": 76.45,
+      "learning_rate": 8.916746411483254e-05,
+      "loss": 0.0737,
+      "step": 841
+    },
+    {
+      "epoch": 76.55,
+      "learning_rate": 8.914832535885168e-05,
+      "loss": 0.0663,
+      "step": 842
+    },
+    {
+      "epoch": 76.64,
+      "learning_rate": 8.912918660287082e-05,
+      "loss": 0.0661,
+      "step": 843
+    },
+    {
+      "epoch": 76.73,
+      "learning_rate": 8.911004784688995e-05,
+      "loss": 0.0681,
+      "step": 844
+    },
+    {
+      "epoch": 76.82,
+      "learning_rate": 8.90909090909091e-05,
+      "loss": 0.0715,
+      "step": 845
+    },
+    {
+      "epoch": 76.91,
+      "learning_rate": 8.907177033492823e-05,
+      "loss": 0.0728,
+      "step": 846
+    },
+    {
+      "epoch": 77.0,
+      "learning_rate": 8.905263157894737e-05,
+      "loss": 0.0797,
+      "step": 847
+    },
+    {
+      "epoch": 77.09,
+      "learning_rate": 8.903349282296652e-05,
+      "loss": 0.0609,
+      "step": 848
+    },
+    {
+      "epoch": 77.18,
+      "learning_rate": 8.901435406698565e-05,
+      "loss": 0.0694,
+      "step": 849
+    },
+    {
+      "epoch": 77.27,
+      "learning_rate": 8.899521531100478e-05,
+      "loss": 0.0691,
+      "step": 850
+    },
+    {
+      "epoch": 77.36,
+      "learning_rate": 8.897607655502393e-05,
+      "loss": 0.079,
+      "step": 851
+    },
+    {
+      "epoch": 77.45,
+      "learning_rate": 8.895693779904307e-05,
+      "loss": 0.0731,
+      "step": 852
+    },
+    {
+      "epoch": 77.55,
+      "learning_rate": 8.89377990430622e-05,
+      "loss": 0.0867,
+      "step": 853
+    },
+    {
+      "epoch": 77.64,
+      "learning_rate": 8.891866028708135e-05,
+      "loss": 0.0568,
+      "step": 854
+    },
+    {
+      "epoch": 77.73,
+      "learning_rate": 8.889952153110048e-05,
+      "loss": 0.0748,
+      "step": 855
+    },
+    {
+      "epoch": 77.82,
+      "learning_rate": 8.888038277511961e-05,
+      "loss": 0.0721,
+      "step": 856
+    },
+    {
+      "epoch": 77.91,
+      "learning_rate": 8.886124401913876e-05,
+      "loss": 0.0788,
+      "step": 857
+    },
+    {
+      "epoch": 78.0,
+      "learning_rate": 8.88421052631579e-05,
+      "loss": 0.074,
+      "step": 858
+    },
+    {
+      "epoch": 78.09,
+      "learning_rate": 8.882296650717704e-05,
+      "loss": 0.0705,
+      "step": 859
+    },
+    {
+      "epoch": 78.18,
+      "learning_rate": 8.880382775119618e-05,
+      "loss": 0.0727,
+      "step": 860
+    },
+    {
+      "epoch": 78.18,
+      "eval_accuracy_NAT": 0.9905180463614721,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9205961092750762,
+      "eval_accuracy_concrete tiles": 0.9839784748975723,
+      "eval_accuracy_dark bricks": 0.8286440090835611,
+      "eval_accuracy_grass": 0.8898946934296109,
+      "eval_accuracy_red bricks": 0.8694957386363636,
+      "eval_accuracy_road": 0.9745667102384923,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5958702064896755,
+      "eval_accuracy_speedway bricks": 0.990947676095517,
+      "eval_accuracy_steel": 0.848418639931403,
+      "eval_accuracy_tiles": 0.011162854403506369,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9756243169740805,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8678358062701117,
+      "eval_iou_concrete tiles": 0.9213019953622856,
+      "eval_iou_dark bricks": 0.7832429875941355,
+      "eval_iou_grass": 0.7956737229607685,
+      "eval_iou_red bricks": 0.8242720080794479,
+      "eval_iou_road": 0.9661064288103147,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3970971890501562,
+      "eval_iou_speedway bricks": 0.9640022141646665,
+      "eval_iou_steel": 0.7857830110876936,
+      "eval_iou_tiles": 0.011162854403506369,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09707868099212646,
+      "eval_mean_accuracy": 0.6849302429878654,
+      "eval_mean_iou": 0.6378540411351666,
+      "eval_overall_accuracy": 0.9733867645263672,
+      "eval_runtime": 3.5722,
+      "eval_samples_per_second": 10.078,
+      "eval_steps_per_second": 0.56,
+      "step": 860
+    },
+    {
+      "epoch": 78.27,
+      "learning_rate": 8.878468899521531e-05,
+      "loss": 0.1273,
+      "step": 861
+    },
+    {
+      "epoch": 78.36,
+      "learning_rate": 8.876555023923445e-05,
+      "loss": 0.0748,
+      "step": 862
+    },
+    {
+      "epoch": 78.45,
+      "learning_rate": 8.874641148325359e-05,
+      "loss": 0.0689,
+      "step": 863
+    },
+    {
+      "epoch": 78.55,
+      "learning_rate": 8.872727272727274e-05,
+      "loss": 0.0696,
+      "step": 864
+    },
+    {
+      "epoch": 78.64,
+      "learning_rate": 8.870813397129187e-05,
+      "loss": 0.0689,
+      "step": 865
+    },
+    {
+      "epoch": 78.73,
+      "learning_rate": 8.868899521531102e-05,
+      "loss": 0.0772,
+      "step": 866
+    },
+    {
+      "epoch": 78.82,
+      "learning_rate": 8.866985645933015e-05,
+      "loss": 0.0825,
+      "step": 867
+    },
+    {
+      "epoch": 78.91,
+      "learning_rate": 8.865071770334928e-05,
+      "loss": 0.0744,
+      "step": 868
+    },
+    {
+      "epoch": 79.0,
+      "learning_rate": 8.863157894736842e-05,
+      "loss": 0.0696,
+      "step": 869
+    },
+    {
+      "epoch": 79.09,
+      "learning_rate": 8.861244019138757e-05,
+      "loss": 0.0756,
+      "step": 870
+    },
+    {
+      "epoch": 79.18,
+      "learning_rate": 8.85933014354067e-05,
+      "loss": 0.0675,
+      "step": 871
+    },
+    {
+      "epoch": 79.27,
+      "learning_rate": 8.857416267942585e-05,
+      "loss": 0.0758,
+      "step": 872
+    },
+    {
+      "epoch": 79.36,
+      "learning_rate": 8.855502392344498e-05,
+      "loss": 0.0709,
+      "step": 873
+    },
+    {
+      "epoch": 79.45,
+      "learning_rate": 8.853588516746411e-05,
+      "loss": 0.0687,
+      "step": 874
+    },
+    {
+      "epoch": 79.55,
+      "learning_rate": 8.851674641148326e-05,
+      "loss": 0.0783,
+      "step": 875
+    },
+    {
+      "epoch": 79.64,
+      "learning_rate": 8.84976076555024e-05,
+      "loss": 0.0705,
+      "step": 876
+    },
+    {
+      "epoch": 79.73,
+      "learning_rate": 8.847846889952154e-05,
+      "loss": 0.0739,
+      "step": 877
+    },
+    {
+      "epoch": 79.82,
+      "learning_rate": 8.845933014354068e-05,
+      "loss": 0.0705,
+      "step": 878
+    },
+    {
+      "epoch": 79.91,
+      "learning_rate": 8.844019138755981e-05,
+      "loss": 0.0687,
+      "step": 879
+    },
+    {
+      "epoch": 80.0,
+      "learning_rate": 8.842105263157894e-05,
+      "loss": 0.0824,
+      "step": 880
+    },
+    {
+      "epoch": 80.0,
+      "eval_accuracy_NAT": 0.9920424075554957,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9387017683162368,
+      "eval_accuracy_concrete tiles": 0.9839784748975723,
+      "eval_accuracy_dark bricks": 0.7994303529502329,
+      "eval_accuracy_grass": 0.8628113673776812,
+      "eval_accuracy_red bricks": 0.8721590909090909,
+      "eval_accuracy_road": 0.965464995217579,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5129710804179417,
+      "eval_accuracy_speedway bricks": 0.9872554237310636,
+      "eval_accuracy_steel": 0.8576858386649957,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9764542318069598,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.858586961455559,
+      "eval_iou_concrete tiles": 0.9162396082450746,
+      "eval_iou_dark bricks": 0.7641648270787343,
+      "eval_iou_grass": 0.7941100781677228,
+      "eval_iou_red bricks": 0.8196228933756049,
+      "eval_iou_road": 0.9587881538908622,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3452261679468626,
+      "eval_iou_speedway bricks": 0.9684824829019911,
+      "eval_iou_steel": 0.7989294052592775,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10084852576255798,
+      "eval_mean_accuracy": 0.6748077538490684,
+      "eval_mean_iou": 0.6308157546252807,
+      "eval_overall_accuracy": 0.9731698566012912,
+      "eval_runtime": 3.6857,
+      "eval_samples_per_second": 9.768,
+      "eval_steps_per_second": 0.543,
+      "step": 880
+    },
+    {
+      "epoch": 80.09,
+      "learning_rate": 8.840191387559809e-05,
+      "loss": 0.0718,
+      "step": 881
+    },
+    {
+      "epoch": 80.18,
+      "learning_rate": 8.838277511961724e-05,
+      "loss": 0.0792,
+      "step": 882
+    },
+    {
+      "epoch": 80.27,
+      "learning_rate": 8.836363636363637e-05,
+      "loss": 0.0629,
+      "step": 883
+    },
+    {
+      "epoch": 80.36,
+      "learning_rate": 8.834449760765551e-05,
+      "loss": 0.0737,
+      "step": 884
+    },
+    {
+      "epoch": 80.45,
+      "learning_rate": 8.832535885167465e-05,
+      "loss": 0.0712,
+      "step": 885
+    },
+    {
+      "epoch": 80.55,
+      "learning_rate": 8.830622009569378e-05,
+      "loss": 0.073,
+      "step": 886
+    },
+    {
+      "epoch": 80.64,
+      "learning_rate": 8.828708133971292e-05,
+      "loss": 0.0632,
+      "step": 887
+    },
+    {
+      "epoch": 80.73,
+      "learning_rate": 8.826794258373207e-05,
+      "loss": 0.0798,
+      "step": 888
+    },
+    {
+      "epoch": 80.82,
+      "learning_rate": 8.82488038277512e-05,
+      "loss": 0.0607,
+      "step": 889
+    },
+    {
+      "epoch": 80.91,
+      "learning_rate": 8.822966507177035e-05,
+      "loss": 0.0702,
+      "step": 890
+    },
+    {
+      "epoch": 81.0,
+      "learning_rate": 8.821052631578948e-05,
+      "loss": 0.0775,
+      "step": 891
+    },
+    {
+      "epoch": 81.09,
+      "learning_rate": 8.819138755980861e-05,
+      "loss": 0.0633,
+      "step": 892
+    },
+    {
+      "epoch": 81.18,
+      "learning_rate": 8.817224880382776e-05,
+      "loss": 0.0677,
+      "step": 893
+    },
+    {
+      "epoch": 81.27,
+      "learning_rate": 8.81531100478469e-05,
+      "loss": 0.0709,
+      "step": 894
+    },
+    {
+      "epoch": 81.36,
+      "learning_rate": 8.813397129186603e-05,
+      "loss": 0.0666,
+      "step": 895
+    },
+    {
+      "epoch": 81.45,
+      "learning_rate": 8.811483253588518e-05,
+      "loss": 0.0729,
+      "step": 896
+    },
+    {
+      "epoch": 81.55,
+      "learning_rate": 8.809569377990431e-05,
+      "loss": 0.0734,
+      "step": 897
+    },
+    {
+      "epoch": 81.64,
+      "learning_rate": 8.807655502392344e-05,
+      "loss": 0.076,
+      "step": 898
+    },
+    {
+      "epoch": 81.73,
+      "learning_rate": 8.805741626794259e-05,
+      "loss": 0.0694,
+      "step": 899
+    },
+    {
+      "epoch": 81.82,
+      "learning_rate": 8.803827751196173e-05,
+      "loss": 0.0862,
+      "step": 900
+    },
+    {
+      "epoch": 81.82,
+      "eval_accuracy_NAT": 0.9912538475165917,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.937121764882613,
+      "eval_accuracy_concrete tiles": 0.9933957072096863,
+      "eval_accuracy_dark bricks": 0.8434625303106116,
+      "eval_accuracy_grass": 0.8783469699458613,
+      "eval_accuracy_red bricks": 0.9119318181818182,
+      "eval_accuracy_road": 0.9753260933968733,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5548479585366526,
+      "eval_accuracy_speedway bricks": 0.9829911407665934,
+      "eval_accuracy_steel": 0.8830799419563353,
+      "eval_accuracy_tiles": 0.0350636899054924,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.976762757958059,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8632790587909684,
+      "eval_iou_concrete tiles": 0.8622611464968153,
+      "eval_iou_dark bricks": 0.7884719173892707,
+      "eval_iou_grass": 0.80727644182894,
+      "eval_iou_red bricks": 0.7954158277838005,
+      "eval_iou_road": 0.963723933410041,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.37113668215681933,
+      "eval_iou_speedway bricks": 0.9686281831435795,
+      "eval_iou_steel": 0.7952820802209715,
+      "eval_iou_tiles": 0.0350636899054924,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09674377739429474,
+      "eval_mean_accuracy": 0.6912939586622406,
+      "eval_mean_iou": 0.6328693630065197,
+      "eval_overall_accuracy": 0.973876953125,
+      "eval_runtime": 3.8507,
+      "eval_samples_per_second": 9.349,
+      "eval_steps_per_second": 0.519,
+      "step": 900
+    },
+    {
+      "epoch": 81.91,
+      "learning_rate": 8.801913875598087e-05,
+      "loss": 0.0721,
+      "step": 901
+    },
+    {
+      "epoch": 82.0,
+      "learning_rate": 8.800000000000001e-05,
+      "loss": 0.0683,
+      "step": 902
+    },
+    {
+      "epoch": 82.09,
+      "learning_rate": 8.798086124401914e-05,
+      "loss": 0.0716,
+      "step": 903
+    },
+    {
+      "epoch": 82.18,
+      "learning_rate": 8.796172248803828e-05,
+      "loss": 0.065,
+      "step": 904
+    },
+    {
+      "epoch": 82.27,
+      "learning_rate": 8.794258373205742e-05,
+      "loss": 0.0718,
+      "step": 905
+    },
+    {
+      "epoch": 82.36,
+      "learning_rate": 8.792344497607657e-05,
+      "loss": 0.0672,
+      "step": 906
+    },
+    {
+      "epoch": 82.45,
+      "learning_rate": 8.79043062200957e-05,
+      "loss": 0.0699,
+      "step": 907
+    },
+    {
+      "epoch": 82.55,
+      "learning_rate": 8.788516746411483e-05,
+      "loss": 0.0682,
+      "step": 908
+    },
+    {
+      "epoch": 82.64,
+      "learning_rate": 8.786602870813398e-05,
+      "loss": 0.0635,
+      "step": 909
+    },
+    {
+      "epoch": 82.73,
+      "learning_rate": 8.784688995215311e-05,
+      "loss": 0.0697,
+      "step": 910
+    },
+    {
+      "epoch": 82.82,
+      "learning_rate": 8.782775119617225e-05,
+      "loss": 0.0765,
+      "step": 911
+    },
+    {
+      "epoch": 82.91,
+      "learning_rate": 8.78086124401914e-05,
+      "loss": 0.0677,
+      "step": 912
+    },
+    {
+      "epoch": 83.0,
+      "learning_rate": 8.778947368421053e-05,
+      "loss": 0.0827,
+      "step": 913
+    },
+    {
+      "epoch": 83.09,
+      "learning_rate": 8.777033492822966e-05,
+      "loss": 0.0622,
+      "step": 914
+    },
+    {
+      "epoch": 83.18,
+      "learning_rate": 8.775119617224881e-05,
+      "loss": 0.0673,
+      "step": 915
+    },
+    {
+      "epoch": 83.27,
+      "learning_rate": 8.773205741626794e-05,
+      "loss": 0.0701,
+      "step": 916
+    },
+    {
+      "epoch": 83.36,
+      "learning_rate": 8.771291866028709e-05,
+      "loss": 0.0848,
+      "step": 917
+    },
+    {
+      "epoch": 83.45,
+      "learning_rate": 8.769377990430623e-05,
+      "loss": 0.0809,
+      "step": 918
+    },
+    {
+      "epoch": 83.55,
+      "learning_rate": 8.767464114832536e-05,
+      "loss": 0.0685,
+      "step": 919
+    },
+    {
+      "epoch": 83.64,
+      "learning_rate": 8.76555023923445e-05,
+      "loss": 0.0613,
+      "step": 920
+    },
+    {
+      "epoch": 83.64,
+      "eval_accuracy_NAT": 0.991557872790038,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9380123610455384,
+      "eval_accuracy_concrete tiles": 0.9919892374487862,
+      "eval_accuracy_dark bricks": 0.8314152650013471,
+      "eval_accuracy_grass": 0.8772704872482237,
+      "eval_accuracy_red bricks": 0.8993252840909091,
+      "eval_accuracy_road": 0.976588507437239,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5793289774763598,
+      "eval_accuracy_speedway bricks": 0.982833651329247,
+      "eval_accuracy_steel": 0.8828903106655234,
+      "eval_accuracy_tiles": 0.07156553896726475,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.976197888148002,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8712386789749723,
+      "eval_iou_concrete tiles": 0.886448087431694,
+      "eval_iou_dark bricks": 0.7909846570727599,
+      "eval_iou_grass": 0.7987538266686185,
+      "eval_iou_red bricks": 0.7956330505812127,
+      "eval_iou_road": 0.9661603540081324,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3915263079445521,
+      "eval_iou_speedway bricks": 0.9692579320259975,
+      "eval_iou_steel": 0.7988124188759753,
+      "eval_iou_tiles": 0.07156553896726475,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09457966685295105,
+      "eval_mean_accuracy": 0.6940598071923444,
+      "eval_mean_iou": 0.6397368262076294,
+      "eval_overall_accuracy": 0.9742827945285373,
+      "eval_runtime": 3.8335,
+      "eval_samples_per_second": 9.391,
+      "eval_steps_per_second": 0.522,
+      "step": 920
+    },
+    {
+      "epoch": 83.73,
+      "learning_rate": 8.763636363636364e-05,
+      "loss": 0.08,
+      "step": 921
+    },
+    {
+      "epoch": 83.82,
+      "learning_rate": 8.761722488038277e-05,
+      "loss": 0.0705,
+      "step": 922
+    },
+    {
+      "epoch": 83.91,
+      "learning_rate": 8.759808612440192e-05,
+      "loss": 0.0677,
+      "step": 923
+    },
+    {
+      "epoch": 84.0,
+      "learning_rate": 8.757894736842106e-05,
+      "loss": 0.0698,
+      "step": 924
+    },
+    {
+      "epoch": 84.09,
+      "learning_rate": 8.75598086124402e-05,
+      "loss": 0.0679,
+      "step": 925
+    },
+    {
+      "epoch": 84.18,
+      "learning_rate": 8.754066985645933e-05,
+      "loss": 0.0622,
+      "step": 926
+    },
+    {
+      "epoch": 84.27,
+      "learning_rate": 8.752153110047847e-05,
+      "loss": 0.0731,
+      "step": 927
+    },
+    {
+      "epoch": 84.36,
+      "learning_rate": 8.75023923444976e-05,
+      "loss": 0.0697,
+      "step": 928
+    },
+    {
+      "epoch": 84.45,
+      "learning_rate": 8.748325358851675e-05,
+      "loss": 0.0728,
+      "step": 929
+    },
+    {
+      "epoch": 84.55,
+      "learning_rate": 8.74641148325359e-05,
+      "loss": 0.0588,
+      "step": 930
+    },
+    {
+      "epoch": 84.64,
+      "learning_rate": 8.744497607655503e-05,
+      "loss": 0.0713,
+      "step": 931
+    },
+    {
+      "epoch": 84.73,
+      "learning_rate": 8.742583732057416e-05,
+      "loss": 0.0965,
+      "step": 932
+    },
+    {
+      "epoch": 84.82,
+      "learning_rate": 8.74066985645933e-05,
+      "loss": 0.0643,
+      "step": 933
+    },
+    {
+      "epoch": 84.91,
+      "learning_rate": 8.738755980861244e-05,
+      "loss": 0.0687,
+      "step": 934
+    },
+    {
+      "epoch": 85.0,
+      "learning_rate": 8.736842105263158e-05,
+      "loss": 0.0686,
+      "step": 935
+    },
+    {
+      "epoch": 85.09,
+      "learning_rate": 8.734928229665073e-05,
+      "loss": 0.0666,
+      "step": 936
+    },
+    {
+      "epoch": 85.18,
+      "learning_rate": 8.733014354066986e-05,
+      "loss": 0.0684,
+      "step": 937
+    },
+    {
+      "epoch": 85.27,
+      "learning_rate": 8.7311004784689e-05,
+      "loss": 0.0603,
+      "step": 938
+    },
+    {
+      "epoch": 85.36,
+      "learning_rate": 8.729186602870814e-05,
+      "loss": 0.0692,
+      "step": 939
+    },
+    {
+      "epoch": 85.45,
+      "learning_rate": 8.727272727272727e-05,
+      "loss": 0.0795,
+      "step": 940
+    },
+    {
+      "epoch": 85.45,
+      "eval_accuracy_NAT": 0.989249362835019,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9451921219794841,
+      "eval_accuracy_concrete tiles": 0.9900629853849446,
+      "eval_accuracy_dark bricks": 0.8005465532504522,
+      "eval_accuracy_grass": 0.865722763764474,
+      "eval_accuracy_red bricks": 0.8662997159090909,
+      "eval_accuracy_road": 0.9801230249084931,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5751385327929865,
+      "eval_accuracy_speedway bricks": 0.9864534947569896,
+      "eval_accuracy_steel": 0.8651474177165095,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9762791454160705,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8654417688438262,
+      "eval_iou_concrete tiles": 0.8984240608179346,
+      "eval_iou_dark bricks": 0.7636303557660535,
+      "eval_iou_grass": 0.7902035659938557,
+      "eval_iou_red bricks": 0.8131666666666667,
+      "eval_iou_road": 0.9638457797284747,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3788751066959664,
+      "eval_iou_speedway bricks": 0.9689283434554526,
+      "eval_iou_steel": 0.7981622764648163,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09722045809030533,
+      "eval_mean_accuracy": 0.6818412287152648,
+      "eval_mean_iou": 0.6320736207576244,
+      "eval_overall_accuracy": 0.9735927581787109,
+      "eval_runtime": 4.312,
+      "eval_samples_per_second": 8.349,
+      "eval_steps_per_second": 0.464,
+      "step": 940
+    },
+    {
+      "epoch": 85.55,
+      "learning_rate": 8.725358851674642e-05,
+      "loss": 0.0835,
+      "step": 941
+    },
+    {
+      "epoch": 85.64,
+      "learning_rate": 8.723444976076556e-05,
+      "loss": 0.0705,
+      "step": 942
+    },
+    {
+      "epoch": 85.73,
+      "learning_rate": 8.72153110047847e-05,
+      "loss": 0.0643,
+      "step": 943
+    },
+    {
+      "epoch": 85.82,
+      "learning_rate": 8.719617224880383e-05,
+      "loss": 0.074,
+      "step": 944
+    },
+    {
+      "epoch": 85.91,
+      "learning_rate": 8.717703349282297e-05,
+      "loss": 0.0636,
+      "step": 945
+    },
+    {
+      "epoch": 86.0,
+      "learning_rate": 8.71578947368421e-05,
+      "loss": 0.0661,
+      "step": 946
+    },
+    {
+      "epoch": 86.09,
+      "learning_rate": 8.713875598086125e-05,
+      "loss": 0.0681,
+      "step": 947
+    },
+    {
+      "epoch": 86.18,
+      "learning_rate": 8.71196172248804e-05,
+      "loss": 0.0629,
+      "step": 948
+    },
+    {
+      "epoch": 86.27,
+      "learning_rate": 8.710047846889953e-05,
+      "loss": 0.077,
+      "step": 949
+    },
+    {
+      "epoch": 86.36,
+      "learning_rate": 8.708133971291866e-05,
+      "loss": 0.069,
+      "step": 950
+    },
+    {
+      "epoch": 86.45,
+      "learning_rate": 8.70622009569378e-05,
+      "loss": 0.0621,
+      "step": 951
+    },
+    {
+      "epoch": 86.55,
+      "learning_rate": 8.704306220095694e-05,
+      "loss": 0.0712,
+      "step": 952
+    },
+    {
+      "epoch": 86.64,
+      "learning_rate": 8.702392344497608e-05,
+      "loss": 0.0639,
+      "step": 953
+    },
+    {
+      "epoch": 86.73,
+      "learning_rate": 8.700478468899523e-05,
+      "loss": 0.0675,
+      "step": 954
+    },
+    {
+      "epoch": 86.82,
+      "learning_rate": 8.698564593301436e-05,
+      "loss": 0.0609,
+      "step": 955
+    },
+    {
+      "epoch": 86.91,
+      "learning_rate": 8.696650717703349e-05,
+      "loss": 0.0615,
+      "step": 956
+    },
+    {
+      "epoch": 87.0,
+      "learning_rate": 8.694736842105264e-05,
+      "loss": 0.0697,
+      "step": 957
+    },
+    {
+      "epoch": 87.09,
+      "learning_rate": 8.692822966507177e-05,
+      "loss": 0.0715,
+      "step": 958
+    },
+    {
+      "epoch": 87.18,
+      "learning_rate": 8.690909090909091e-05,
+      "loss": 0.0834,
+      "step": 959
+    },
+    {
+      "epoch": 87.27,
+      "learning_rate": 8.688995215311006e-05,
+      "loss": 0.074,
+      "step": 960
+    },
+    {
+      "epoch": 87.27,
+      "eval_accuracy_NAT": 0.9924157781176618,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9454536675393794,
+      "eval_accuracy_concrete tiles": 0.9829083348621048,
+      "eval_accuracy_dark bricks": 0.8169816404295447,
+      "eval_accuracy_grass": 0.8133805401286887,
+      "eval_accuracy_red bricks": 0.8787286931818182,
+      "eval_accuracy_road": 0.9724784065529446,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6270228544647534,
+      "eval_accuracy_speedway bricks": 0.9862640247059292,
+      "eval_accuracy_steel": 0.8651639073939714,
+      "eval_accuracy_tiles": 0.0018490617723599507,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9761661279304197,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8633602829773865,
+      "eval_iou_concrete tiles": 0.92185707731131,
+      "eval_iou_dark bricks": 0.7807982343203973,
+      "eval_iou_grass": 0.768820614469772,
+      "eval_iou_red bricks": 0.8052391799544419,
+      "eval_iou_road": 0.9640951931122614,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4098904267589389,
+      "eval_iou_speedway bricks": 0.9696364597404122,
+      "eval_iou_steel": 0.7993692437781383,
+      "eval_iou_tiles": 0.0018490617723599507,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09748014807701111,
+      "eval_mean_accuracy": 0.6832805314730119,
+      "eval_mean_iou": 0.6354678386250645,
+      "eval_overall_accuracy": 0.9734552171495225,
+      "eval_runtime": 4.4458,
+      "eval_samples_per_second": 8.098,
+      "eval_steps_per_second": 0.45,
+      "step": 960
+    },
+    {
+      "epoch": 87.36,
+      "learning_rate": 8.687081339712919e-05,
+      "loss": 0.0732,
+      "step": 961
+    },
+    {
+      "epoch": 87.45,
+      "learning_rate": 8.685167464114832e-05,
+      "loss": 0.0759,
+      "step": 962
+    },
+    {
+      "epoch": 87.55,
+      "learning_rate": 8.683253588516747e-05,
+      "loss": 0.0662,
+      "step": 963
+    },
+    {
+      "epoch": 87.64,
+      "learning_rate": 8.68133971291866e-05,
+      "loss": 0.061,
+      "step": 964
+    },
+    {
+      "epoch": 87.73,
+      "learning_rate": 8.679425837320575e-05,
+      "loss": 0.0618,
+      "step": 965
+    },
+    {
+      "epoch": 87.82,
+      "learning_rate": 8.677511961722489e-05,
+      "loss": 0.065,
+      "step": 966
+    },
+    {
+      "epoch": 87.91,
+      "learning_rate": 8.675598086124403e-05,
+      "loss": 0.0621,
+      "step": 967
+    },
+    {
+      "epoch": 88.0,
+      "learning_rate": 8.673684210526316e-05,
+      "loss": 0.0634,
+      "step": 968
+    },
+    {
+      "epoch": 88.09,
+      "learning_rate": 8.67177033492823e-05,
+      "loss": 0.0725,
+      "step": 969
+    },
+    {
+      "epoch": 88.18,
+      "learning_rate": 8.669856459330143e-05,
+      "loss": 0.0678,
+      "step": 970
+    },
+    {
+      "epoch": 88.27,
+      "learning_rate": 8.667942583732058e-05,
+      "loss": 0.0611,
+      "step": 971
+    },
+    {
+      "epoch": 88.36,
+      "learning_rate": 8.666028708133973e-05,
+      "loss": 0.0768,
+      "step": 972
+    },
+    {
+      "epoch": 88.45,
+      "learning_rate": 8.664114832535886e-05,
+      "loss": 0.0686,
+      "step": 973
+    },
+    {
+      "epoch": 88.55,
+      "learning_rate": 8.662200956937799e-05,
+      "loss": 0.0655,
+      "step": 974
+    },
+    {
+      "epoch": 88.64,
+      "learning_rate": 8.660287081339714e-05,
+      "loss": 0.0647,
+      "step": 975
+    },
+    {
+      "epoch": 88.73,
+      "learning_rate": 8.658373205741627e-05,
+      "loss": 0.0771,
+      "step": 976
+    },
+    {
+      "epoch": 88.82,
+      "learning_rate": 8.656459330143541e-05,
+      "loss": 0.0618,
+      "step": 977
+    },
+    {
+      "epoch": 88.91,
+      "learning_rate": 8.654545454545456e-05,
+      "loss": 0.0696,
+      "step": 978
+    },
+    {
+      "epoch": 89.0,
+      "learning_rate": 8.652631578947369e-05,
+      "loss": 0.0574,
+      "step": 979
+    },
+    {
+      "epoch": 89.09,
+      "learning_rate": 8.650717703349282e-05,
+      "loss": 0.0621,
+      "step": 980
+    },
+    {
+      "epoch": 89.09,
+      "eval_accuracy_NAT": 0.991732912297443,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9362660414610069,
+      "eval_accuracy_concrete tiles": 0.9804317250657372,
+      "eval_accuracy_dark bricks": 0.82687348446942,
+      "eval_accuracy_grass": 0.8733455194902785,
+      "eval_accuracy_red bricks": 0.8986150568181818,
+      "eval_accuracy_road": 0.9806538675940493,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6488021393322857,
+      "eval_accuracy_speedway bricks": 0.9834527477381259,
+      "eval_accuracy_steel": 0.8916710639139899,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9761750308070268,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8781883944016354,
+      "eval_iou_concrete tiles": 0.9332363213038417,
+      "eval_iou_dark bricks": 0.7838508410260152,
+      "eval_iou_grass": 0.8029556650246306,
+      "eval_iou_red bricks": 0.8210577547047372,
+      "eval_iou_road": 0.9702159629805093,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.40471891176116526,
+      "eval_iou_speedway bricks": 0.9694461925365127,
+      "eval_iou_steel": 0.7955057006252299,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09414172172546387,
+      "eval_mean_accuracy": 0.6932188121677322,
+      "eval_mean_iou": 0.641180828859331,
+      "eval_overall_accuracy": 0.9748120837741427,
+      "eval_runtime": 3.788,
+      "eval_samples_per_second": 9.504,
+      "eval_steps_per_second": 0.528,
+      "step": 980
+    },
+    {
+      "epoch": 89.18,
+      "learning_rate": 8.648803827751197e-05,
+      "loss": 0.0617,
+      "step": 981
+    },
+    {
+      "epoch": 89.27,
+      "learning_rate": 8.64688995215311e-05,
+      "loss": 0.0653,
+      "step": 982
+    },
+    {
+      "epoch": 89.36,
+      "learning_rate": 8.644976076555025e-05,
+      "loss": 0.0625,
+      "step": 983
+    },
+    {
+      "epoch": 89.45,
+      "learning_rate": 8.643062200956938e-05,
+      "loss": 0.0688,
+      "step": 984
+    },
+    {
+      "epoch": 89.55,
+      "learning_rate": 8.641148325358852e-05,
+      "loss": 0.0678,
+      "step": 985
+    },
+    {
+      "epoch": 89.64,
+      "learning_rate": 8.639234449760765e-05,
+      "loss": 0.0696,
+      "step": 986
+    },
+    {
+      "epoch": 89.73,
+      "learning_rate": 8.63732057416268e-05,
+      "loss": 0.0659,
+      "step": 987
+    },
+    {
+      "epoch": 89.82,
+      "learning_rate": 8.635406698564593e-05,
+      "loss": 0.0632,
+      "step": 988
+    },
+    {
+      "epoch": 89.91,
+      "learning_rate": 8.633492822966508e-05,
+      "loss": 0.0776,
+      "step": 989
+    },
+    {
+      "epoch": 90.0,
+      "learning_rate": 8.631578947368421e-05,
+      "loss": 0.0722,
+      "step": 990
+    },
+    {
+      "epoch": 90.09,
+      "learning_rate": 8.629665071770336e-05,
+      "loss": 0.0761,
+      "step": 991
+    },
+    {
+      "epoch": 90.18,
+      "learning_rate": 8.627751196172249e-05,
+      "loss": 0.0594,
+      "step": 992
+    },
+    {
+      "epoch": 90.27,
+      "learning_rate": 8.625837320574163e-05,
+      "loss": 0.0682,
+      "step": 993
+    },
+    {
+      "epoch": 90.36,
+      "learning_rate": 8.623923444976078e-05,
+      "loss": 0.0899,
+      "step": 994
+    },
+    {
+      "epoch": 90.45,
+      "learning_rate": 8.622009569377991e-05,
+      "loss": 0.0641,
+      "step": 995
+    },
+    {
+      "epoch": 90.55,
+      "learning_rate": 8.620095693779904e-05,
+      "loss": 0.0696,
+      "step": 996
+    },
+    {
+      "epoch": 90.64,
+      "learning_rate": 8.618181818181819e-05,
+      "loss": 0.0639,
+      "step": 997
+    },
+    {
+      "epoch": 90.73,
+      "learning_rate": 8.616267942583732e-05,
+      "loss": 0.0751,
+      "step": 998
+    },
+    {
+      "epoch": 90.82,
+      "learning_rate": 8.614354066985647e-05,
+      "loss": 0.0676,
+      "step": 999
+    },
+    {
+      "epoch": 90.91,
+      "learning_rate": 8.612440191387561e-05,
+      "loss": 0.061,
+      "step": 1000
+    },
+    {
+      "epoch": 90.91,
+      "eval_accuracy_NAT": 0.9902636038517161,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9429347826086957,
+      "eval_accuracy_concrete tiles": 0.9466152999449642,
+      "eval_accuracy_dark bricks": 0.8102074592971787,
+      "eval_accuracy_grass": 0.8856866247024818,
+      "eval_accuracy_red bricks": 0.8501420454545454,
+      "eval_accuracy_road": 0.9734518196779394,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6191382019684063,
+      "eval_accuracy_speedway bricks": 0.9875553174106082,
+      "eval_accuracy_steel": 0.8819916232438494,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9769699465932689,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8708710791898056,
+      "eval_iou_concrete tiles": 0.9226917804136615,
+      "eval_iou_dark bricks": 0.7768100966861023,
+      "eval_iou_grass": 0.8107459592275502,
+      "eval_iou_red bricks": 0.8043003527633126,
+      "eval_iou_road": 0.9634859330661212,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4054449278763698,
+      "eval_iou_speedway bricks": 0.9680546633874313,
+      "eval_iou_steel": 0.80170120283284,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09617039561271667,
+      "eval_mean_accuracy": 0.6836912906277219,
+      "eval_mean_iou": 0.638544303233574,
+      "eval_overall_accuracy": 0.9744896358913846,
+      "eval_runtime": 3.3638,
+      "eval_samples_per_second": 10.702,
+      "eval_steps_per_second": 0.595,
+      "step": 1000
+    },
+    {
+      "epoch": 91.0,
+      "learning_rate": 8.610526315789474e-05,
+      "loss": 0.0685,
+      "step": 1001
+    },
+    {
+      "epoch": 91.09,
+      "learning_rate": 8.608612440191388e-05,
+      "loss": 0.0658,
+      "step": 1002
+    },
+    {
+      "epoch": 91.18,
+      "learning_rate": 8.606698564593302e-05,
+      "loss": 0.0771,
+      "step": 1003
+    },
+    {
+      "epoch": 91.27,
+      "learning_rate": 8.604784688995215e-05,
+      "loss": 0.0658,
+      "step": 1004
+    },
+    {
+      "epoch": 91.36,
+      "learning_rate": 8.60287081339713e-05,
+      "loss": 0.0644,
+      "step": 1005
+    },
+    {
+      "epoch": 91.45,
+      "learning_rate": 8.600956937799044e-05,
+      "loss": 0.0661,
+      "step": 1006
+    },
+    {
+      "epoch": 91.55,
+      "learning_rate": 8.599043062200958e-05,
+      "loss": 0.0611,
+      "step": 1007
+    },
+    {
+      "epoch": 91.64,
+      "learning_rate": 8.597129186602871e-05,
+      "loss": 0.0698,
+      "step": 1008
+    },
+    {
+      "epoch": 91.73,
+      "learning_rate": 8.595215311004785e-05,
+      "loss": 0.0575,
+      "step": 1009
+    },
+    {
+      "epoch": 91.82,
+      "learning_rate": 8.593301435406699e-05,
+      "loss": 0.0581,
+      "step": 1010
+    },
+    {
+      "epoch": 91.91,
+      "learning_rate": 8.591387559808612e-05,
+      "loss": 0.0573,
+      "step": 1011
+    },
+    {
+      "epoch": 92.0,
+      "learning_rate": 8.589473684210528e-05,
+      "loss": 0.0737,
+      "step": 1012
+    },
+    {
+      "epoch": 92.09,
+      "learning_rate": 8.587559808612441e-05,
+      "loss": 0.0622,
+      "step": 1013
+    },
+    {
+      "epoch": 92.18,
+      "learning_rate": 8.585645933014354e-05,
+      "loss": 0.0664,
+      "step": 1014
+    },
+    {
+      "epoch": 92.27,
+      "learning_rate": 8.583732057416269e-05,
+      "loss": 0.0682,
+      "step": 1015
+    },
+    {
+      "epoch": 92.36,
+      "learning_rate": 8.581818181818182e-05,
+      "loss": 0.0619,
+      "step": 1016
+    },
+    {
+      "epoch": 92.45,
+      "learning_rate": 8.579904306220095e-05,
+      "loss": 0.078,
+      "step": 1017
+    },
+    {
+      "epoch": 92.55,
+      "learning_rate": 8.577990430622011e-05,
+      "loss": 0.0652,
+      "step": 1018
+    },
+    {
+      "epoch": 92.64,
+      "learning_rate": 8.576076555023924e-05,
+      "loss": 0.0603,
+      "step": 1019
+    },
+    {
+      "epoch": 92.73,
+      "learning_rate": 8.574162679425837e-05,
+      "loss": 0.0694,
+      "step": 1020
+    },
+    {
+      "epoch": 92.73,
+      "eval_accuracy_NAT": 0.9927873841686646,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9322449461350273,
+      "eval_accuracy_concrete tiles": 0.9890539962086468,
+      "eval_accuracy_dark bricks": 0.8446172202763558,
+      "eval_accuracy_grass": 0.851277624188706,
+      "eval_accuracy_red bricks": 0.9103338068181818,
+      "eval_accuracy_road": 0.9790456198223189,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.598213547266562,
+      "eval_accuracy_speedway bricks": 0.9847011984885842,
+      "eval_accuracy_steel": 0.8924790581096234,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9764802969835654,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8730821891997136,
+      "eval_iou_concrete tiles": 0.8749087171719904,
+      "eval_iou_dark bricks": 0.8022227096585508,
+      "eval_iou_grass": 0.7914635731461623,
+      "eval_iou_red bricks": 0.8057520037718058,
+      "eval_iou_road": 0.967173870974678,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.39416893732970026,
+      "eval_iou_speedway bricks": 0.9696323667906708,
+      "eval_iou_steel": 0.8037705867502265,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0933745801448822,
+      "eval_mean_accuracy": 0.6903657231909747,
+      "eval_mean_iou": 0.6352811732136203,
+      "eval_overall_accuracy": 0.9744385613335503,
+      "eval_runtime": 3.7417,
+      "eval_samples_per_second": 9.621,
+      "eval_steps_per_second": 0.535,
+      "step": 1020
+    },
+    {
+      "epoch": 92.82,
+      "learning_rate": 8.572248803827752e-05,
+      "loss": 0.0682,
+      "step": 1021
+    },
+    {
+      "epoch": 92.91,
+      "learning_rate": 8.570334928229665e-05,
+      "loss": 0.0638,
+      "step": 1022
+    },
+    {
+      "epoch": 93.0,
+      "learning_rate": 8.568421052631578e-05,
+      "loss": 0.0622,
+      "step": 1023
+    },
+    {
+      "epoch": 93.09,
+      "learning_rate": 8.566507177033494e-05,
+      "loss": 0.0676,
+      "step": 1024
+    },
+    {
+      "epoch": 93.18,
+      "learning_rate": 8.564593301435407e-05,
+      "loss": 0.0546,
+      "step": 1025
+    },
+    {
+      "epoch": 93.27,
+      "learning_rate": 8.56267942583732e-05,
+      "loss": 0.0677,
+      "step": 1026
+    },
+    {
+      "epoch": 93.36,
+      "learning_rate": 8.560765550239235e-05,
+      "loss": 0.0666,
+      "step": 1027
+    },
+    {
+      "epoch": 93.45,
+      "learning_rate": 8.558851674641148e-05,
+      "loss": 0.0612,
+      "step": 1028
+    },
+    {
+      "epoch": 93.55,
+      "learning_rate": 8.556937799043062e-05,
+      "loss": 0.0621,
+      "step": 1029
+    },
+    {
+      "epoch": 93.64,
+      "learning_rate": 8.555023923444977e-05,
+      "loss": 0.0631,
+      "step": 1030
+    },
+    {
+      "epoch": 93.73,
+      "learning_rate": 8.55311004784689e-05,
+      "loss": 0.0655,
+      "step": 1031
+    },
+    {
+      "epoch": 93.82,
+      "learning_rate": 8.551196172248804e-05,
+      "loss": 0.0629,
+      "step": 1032
+    },
+    {
+      "epoch": 93.91,
+      "learning_rate": 8.549282296650718e-05,
+      "loss": 0.0595,
+      "step": 1033
+    },
+    {
+      "epoch": 94.0,
+      "learning_rate": 8.547368421052632e-05,
+      "loss": 0.0649,
+      "step": 1034
+    },
+    {
+      "epoch": 94.09,
+      "learning_rate": 8.545454545454545e-05,
+      "loss": 0.0641,
+      "step": 1035
+    },
+    {
+      "epoch": 94.18,
+      "learning_rate": 8.543540669856461e-05,
+      "loss": 0.0688,
+      "step": 1036
+    },
+    {
+      "epoch": 94.27,
+      "learning_rate": 8.541626794258374e-05,
+      "loss": 0.0588,
+      "step": 1037
+    },
+    {
+      "epoch": 94.36,
+      "learning_rate": 8.539712918660287e-05,
+      "loss": 0.0716,
+      "step": 1038
+    },
+    {
+      "epoch": 94.45,
+      "learning_rate": 8.537799043062202e-05,
+      "loss": 0.0667,
+      "step": 1039
+    },
+    {
+      "epoch": 94.55,
+      "learning_rate": 8.535885167464115e-05,
+      "loss": 0.0581,
+      "step": 1040
+    },
+    {
+      "epoch": 94.55,
+      "eval_accuracy_NAT": 0.9923660189028551,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9378742113395425,
+      "eval_accuracy_concrete tiles": 0.982021647404146,
+      "eval_accuracy_dark bricks": 0.8086293830106617,
+      "eval_accuracy_grass": 0.8861235089141855,
+      "eval_accuracy_red bricks": 0.9076704545454546,
+      "eval_accuracy_road": 0.975686437634449,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4666280704656356,
+      "eval_accuracy_speedway bricks": 0.9827847752969672,
+      "eval_accuracy_steel": 0.8853637622848097,
+      "eval_accuracy_tiles": 0.010341049171346392,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9768645444401677,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8697505488423003,
+      "eval_iou_concrete tiles": 0.9239930955120829,
+      "eval_iou_dark bricks": 0.7740687520725102,
+      "eval_iou_grass": 0.7970743393213071,
+      "eval_iou_red bricks": 0.8142720611659764,
+      "eval_iou_road": 0.9671132589731972,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.317984557290199,
+      "eval_iou_speedway bricks": 0.9701022085626474,
+      "eval_iou_steel": 0.8058897252512214,
+      "eval_iou_tiles": 0.010341049171346392,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0951547846198082,
+      "eval_mean_accuracy": 0.679653024536158,
+      "eval_mean_iou": 0.6328810877386888,
+      "eval_overall_accuracy": 0.9743495517306857,
+      "eval_runtime": 4.3441,
+      "eval_samples_per_second": 8.287,
+      "eval_steps_per_second": 0.46,
+      "step": 1040
+    },
+    {
+      "epoch": 94.64,
+      "learning_rate": 8.533971291866028e-05,
+      "loss": 0.0604,
+      "step": 1041
+    },
+    {
+      "epoch": 94.73,
+      "learning_rate": 8.532057416267944e-05,
+      "loss": 0.0624,
+      "step": 1042
+    },
+    {
+      "epoch": 94.82,
+      "learning_rate": 8.530143540669857e-05,
+      "loss": 0.0684,
+      "step": 1043
+    },
+    {
+      "epoch": 94.91,
+      "learning_rate": 8.52822966507177e-05,
+      "loss": 0.063,
+      "step": 1044
+    },
+    {
+      "epoch": 95.0,
+      "learning_rate": 8.526315789473685e-05,
+      "loss": 0.0631,
+      "step": 1045
+    },
+    {
+      "epoch": 95.09,
+      "learning_rate": 8.524401913875598e-05,
+      "loss": 0.0668,
+      "step": 1046
+    },
+    {
+      "epoch": 95.18,
+      "learning_rate": 8.522488038277511e-05,
+      "loss": 0.062,
+      "step": 1047
+    },
+    {
+      "epoch": 95.27,
+      "learning_rate": 8.520574162679427e-05,
+      "loss": 0.0556,
+      "step": 1048
+    },
+    {
+      "epoch": 95.36,
+      "learning_rate": 8.51866028708134e-05,
+      "loss": 0.0571,
+      "step": 1049
+    },
+    {
+      "epoch": 95.45,
+      "learning_rate": 8.516746411483254e-05,
+      "loss": 0.0541,
+      "step": 1050
+    },
+    {
+      "epoch": 95.55,
+      "learning_rate": 8.514832535885168e-05,
+      "loss": 0.0621,
+      "step": 1051
+    },
+    {
+      "epoch": 95.64,
+      "learning_rate": 8.512918660287081e-05,
+      "loss": 0.0641,
+      "step": 1052
+    },
+    {
+      "epoch": 95.73,
+      "learning_rate": 8.511004784688995e-05,
+      "loss": 0.0646,
+      "step": 1053
+    },
+    {
+      "epoch": 95.82,
+      "learning_rate": 8.50909090909091e-05,
+      "loss": 0.0706,
+      "step": 1054
+    },
+    {
+      "epoch": 95.91,
+      "learning_rate": 8.507177033492824e-05,
+      "loss": 0.0593,
+      "step": 1055
+    },
+    {
+      "epoch": 96.0,
+      "learning_rate": 8.505263157894737e-05,
+      "loss": 0.0681,
+      "step": 1056
+    },
+    {
+      "epoch": 96.09,
+      "learning_rate": 8.503349282296651e-05,
+      "loss": 0.0658,
+      "step": 1057
+    },
+    {
+      "epoch": 96.18,
+      "learning_rate": 8.501435406698565e-05,
+      "loss": 0.0624,
+      "step": 1058
+    },
+    {
+      "epoch": 96.27,
+      "learning_rate": 8.499521531100478e-05,
+      "loss": 0.0587,
+      "step": 1059
+    },
+    {
+      "epoch": 96.36,
+      "learning_rate": 8.497607655502394e-05,
+      "loss": 0.0767,
+      "step": 1060
+    },
+    {
+      "epoch": 96.36,
+      "eval_accuracy_NAT": 0.9909325300337445,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.946840529636465,
+      "eval_accuracy_concrete tiles": 0.9704029841619275,
+      "eval_accuracy_dark bricks": 0.8423463300103922,
+      "eval_accuracy_grass": 0.8539863063012684,
+      "eval_accuracy_red bricks": 0.93359375,
+      "eval_accuracy_road": 0.978220939385988,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6411104678410939,
+      "eval_accuracy_speedway bricks": 0.982465572567633,
+      "eval_accuracy_steel": 0.892355385528659,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9771728111633462,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8732797193367897,
+      "eval_iou_concrete tiles": 0.9176014802821788,
+      "eval_iou_dark bricks": 0.7935385619493093,
+      "eval_iou_grass": 0.7785346953601447,
+      "eval_iou_red bricks": 0.4663828277452546,
+      "eval_iou_road": 0.9669621427273684,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4152901048270443,
+      "eval_iou_speedway bricks": 0.9699529319392625,
+      "eval_iou_steel": 0.8042982306211776,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09629139304161072,
+      "eval_mean_accuracy": 0.6947888304205516,
+      "eval_mean_iou": 0.6125395004578367,
+      "eval_overall_accuracy": 0.9742013083563911,
+      "eval_runtime": 4.0715,
+      "eval_samples_per_second": 8.842,
+      "eval_steps_per_second": 0.491,
+      "step": 1060
+    },
+    {
+      "epoch": 96.45,
+      "learning_rate": 8.495693779904307e-05,
+      "loss": 0.0595,
+      "step": 1061
+    },
+    {
+      "epoch": 96.55,
+      "learning_rate": 8.49377990430622e-05,
+      "loss": 0.0662,
+      "step": 1062
+    },
+    {
+      "epoch": 96.64,
+      "learning_rate": 8.491866028708135e-05,
+      "loss": 0.0592,
+      "step": 1063
+    },
+    {
+      "epoch": 96.73,
+      "learning_rate": 8.489952153110048e-05,
+      "loss": 0.0694,
+      "step": 1064
+    },
+    {
+      "epoch": 96.82,
+      "learning_rate": 8.488038277511963e-05,
+      "loss": 0.0573,
+      "step": 1065
+    },
+    {
+      "epoch": 96.91,
+      "learning_rate": 8.486124401913876e-05,
+      "loss": 0.0653,
+      "step": 1066
+    },
+    {
+      "epoch": 97.0,
+      "learning_rate": 8.48421052631579e-05,
+      "loss": 0.0629,
+      "step": 1067
+    },
+    {
+      "epoch": 97.09,
+      "learning_rate": 8.482296650717703e-05,
+      "loss": 0.0636,
+      "step": 1068
+    },
+    {
+      "epoch": 97.18,
+      "learning_rate": 8.480382775119618e-05,
+      "loss": 0.0713,
+      "step": 1069
+    },
+    {
+      "epoch": 97.27,
+      "learning_rate": 8.478468899521531e-05,
+      "loss": 0.0616,
+      "step": 1070
+    },
+    {
+      "epoch": 97.36,
+      "learning_rate": 8.476555023923446e-05,
+      "loss": 0.0586,
+      "step": 1071
+    },
+    {
+      "epoch": 97.45,
+      "learning_rate": 8.474641148325359e-05,
+      "loss": 0.0644,
+      "step": 1072
+    },
+    {
+      "epoch": 97.55,
+      "learning_rate": 8.472727272727274e-05,
+      "loss": 0.0621,
+      "step": 1073
+    },
+    {
+      "epoch": 97.64,
+      "learning_rate": 8.470813397129187e-05,
+      "loss": 0.0621,
+      "step": 1074
+    },
+    {
+      "epoch": 97.73,
+      "learning_rate": 8.468899521531101e-05,
+      "loss": 0.06,
+      "step": 1075
+    },
+    {
+      "epoch": 97.82,
+      "learning_rate": 8.466985645933014e-05,
+      "loss": 0.0566,
+      "step": 1076
+    },
+    {
+      "epoch": 97.91,
+      "learning_rate": 8.465071770334929e-05,
+      "loss": 0.0475,
+      "step": 1077
+    },
+    {
+      "epoch": 98.0,
+      "learning_rate": 8.463157894736842e-05,
+      "loss": 0.0665,
+      "step": 1078
+    },
+    {
+      "epoch": 98.09,
+      "learning_rate": 8.461244019138757e-05,
+      "loss": 0.0626,
+      "step": 1079
+    },
+    {
+      "epoch": 98.18,
+      "learning_rate": 8.45933014354067e-05,
+      "loss": 0.0578,
+      "step": 1080
+    },
+    {
+      "epoch": 98.18,
+      "eval_accuracy_NAT": 0.990769842104483,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9453168590926649,
+      "eval_accuracy_concrete tiles": 0.9643796245337247,
+      "eval_accuracy_dark bricks": 0.8405373157307263,
+      "eval_accuracy_grass": 0.8714966255063488,
+      "eval_accuracy_red bricks": 0.9161931818181818,
+      "eval_accuracy_road": 0.9812367062601951,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6312132991481267,
+      "eval_accuracy_speedway bricks": 0.985648548743886,
+      "eval_accuracy_steel": 0.8800128619484203,
+      "eval_accuracy_tiles": 0.002944802081906588,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9774220866766411,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.881339018869812,
+      "eval_iou_concrete tiles": 0.9314845988009804,
+      "eval_iou_dark bricks": 0.8000732734933138,
+      "eval_iou_grass": 0.7942474000222969,
+      "eval_iou_red bricks": 0.42867824208689875,
+      "eval_iou_road": 0.968219972437821,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.41174672253493266,
+      "eval_iou_speedway bricks": 0.9709009662826381,
+      "eval_iou_steel": 0.8064784241425949,
+      "eval_iou_tiles": 0.002944802081906588,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09231866151094437,
+      "eval_mean_accuracy": 0.6930576666898973,
+      "eval_mean_iou": 0.6133488851869106,
+      "eval_overall_accuracy": 0.9751089943779839,
+      "eval_runtime": 4.5422,
+      "eval_samples_per_second": 7.926,
+      "eval_steps_per_second": 0.44,
+      "step": 1080
+    },
+    {
+      "epoch": 98.27,
+      "learning_rate": 8.457416267942585e-05,
+      "loss": 0.0667,
+      "step": 1081
+    },
+    {
+      "epoch": 98.36,
+      "learning_rate": 8.455502392344498e-05,
+      "loss": 0.0604,
+      "step": 1082
+    },
+    {
+      "epoch": 98.45,
+      "learning_rate": 8.453588516746412e-05,
+      "loss": 0.0692,
+      "step": 1083
+    },
+    {
+      "epoch": 98.55,
+      "learning_rate": 8.451674641148326e-05,
+      "loss": 0.0739,
+      "step": 1084
+    },
+    {
+      "epoch": 98.64,
+      "learning_rate": 8.44976076555024e-05,
+      "loss": 0.0521,
+      "step": 1085
+    },
+    {
+      "epoch": 98.73,
+      "learning_rate": 8.447846889952153e-05,
+      "loss": 0.0617,
+      "step": 1086
+    },
+    {
+      "epoch": 98.82,
+      "learning_rate": 8.445933014354066e-05,
+      "loss": 0.0658,
+      "step": 1087
+    },
+    {
+      "epoch": 98.91,
+      "learning_rate": 8.444019138755981e-05,
+      "loss": 0.0542,
+      "step": 1088
+    },
+    {
+      "epoch": 99.0,
+      "learning_rate": 8.442105263157896e-05,
+      "loss": 0.0674,
+      "step": 1089
+    },
+    {
+      "epoch": 99.09,
+      "learning_rate": 8.440191387559809e-05,
+      "loss": 0.0654,
+      "step": 1090
+    },
+    {
+      "epoch": 99.18,
+      "learning_rate": 8.438277511961723e-05,
+      "loss": 0.064,
+      "step": 1091
+    },
+    {
+      "epoch": 99.27,
+      "learning_rate": 8.436363636363637e-05,
+      "loss": 0.0546,
+      "step": 1092
+    },
+    {
+      "epoch": 99.36,
+      "learning_rate": 8.43444976076555e-05,
+      "loss": 0.0614,
+      "step": 1093
+    },
+    {
+      "epoch": 99.45,
+      "learning_rate": 8.432535885167464e-05,
+      "loss": 0.0603,
+      "step": 1094
+    },
+    {
+      "epoch": 99.55,
+      "learning_rate": 8.430622009569379e-05,
+      "loss": 0.0656,
+      "step": 1095
+    },
+    {
+      "epoch": 99.64,
+      "learning_rate": 8.428708133971292e-05,
+      "loss": 0.062,
+      "step": 1096
+    },
+    {
+      "epoch": 99.73,
+      "learning_rate": 8.426794258373207e-05,
+      "loss": 0.055,
+      "step": 1097
+    },
+    {
+      "epoch": 99.82,
+      "learning_rate": 8.42488038277512e-05,
+      "loss": 0.0607,
+      "step": 1098
+    },
+    {
+      "epoch": 99.91,
+      "learning_rate": 8.422966507177033e-05,
+      "loss": 0.0692,
+      "step": 1099
+    },
+    {
+      "epoch": 100.0,
+      "learning_rate": 8.421052631578948e-05,
+      "loss": 0.0625,
+      "step": 1100
+    },
+    {
+      "epoch": 100.0,
+      "eval_accuracy_NAT": 0.992329140619541,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9524536997296021,
+      "eval_accuracy_concrete tiles": 0.9807680547911698,
+      "eval_accuracy_dark bricks": 0.8104383972903275,
+      "eval_accuracy_grass": 0.8593792050105377,
+      "eval_accuracy_red bricks": 0.8751775568181818,
+      "eval_accuracy_road": 0.9742607803992082,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6041683897113556,
+      "eval_accuracy_speedway bricks": 0.9823412705596125,
+      "eval_accuracy_steel": 0.8810022425961348,
+      "eval_accuracy_tiles": 0.025270510888919327,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9767176299038048,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8767247594966715,
+      "eval_iou_concrete tiles": 0.9214086691753081,
+      "eval_iou_dark bricks": 0.7708585026542193,
+      "eval_iou_grass": 0.7946294973677492,
+      "eval_iou_red bricks": 0.8010726474890297,
+      "eval_iou_road": 0.9666455509404897,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4063299588385805,
+      "eval_iou_speedway bricks": 0.9709338144299154,
+      "eval_iou_steel": 0.8104100778897712,
+      "eval_iou_tiles": 0.025270510888919327,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0945933535695076,
+      "eval_mean_accuracy": 0.6875068652626608,
+      "eval_mean_iou": 0.6400770476211123,
+      "eval_overall_accuracy": 0.9749425252278646,
+      "eval_runtime": 3.5491,
+      "eval_samples_per_second": 10.143,
+      "eval_steps_per_second": 0.564,
+      "step": 1100
+    },
+    {
+      "epoch": 100.09,
+      "learning_rate": 8.419138755980862e-05,
+      "loss": 0.0684,
+      "step": 1101
+    },
+    {
+      "epoch": 100.18,
+      "learning_rate": 8.417224880382775e-05,
+      "loss": 0.0601,
+      "step": 1102
+    },
+    {
+      "epoch": 100.27,
+      "learning_rate": 8.41531100478469e-05,
+      "loss": 0.0634,
+      "step": 1103
+    },
+    {
+      "epoch": 100.36,
+      "learning_rate": 8.413397129186603e-05,
+      "loss": 0.0567,
+      "step": 1104
+    },
+    {
+      "epoch": 100.45,
+      "learning_rate": 8.411483253588516e-05,
+      "loss": 0.057,
+      "step": 1105
+    },
+    {
+      "epoch": 100.55,
+      "learning_rate": 8.409569377990431e-05,
+      "loss": 0.067,
+      "step": 1106
+    },
+    {
+      "epoch": 100.64,
+      "learning_rate": 8.407655502392345e-05,
+      "loss": 0.0509,
+      "step": 1107
+    },
+    {
+      "epoch": 100.73,
+      "learning_rate": 8.405741626794259e-05,
+      "loss": 0.0558,
+      "step": 1108
+    },
+    {
+      "epoch": 100.82,
+      "learning_rate": 8.403827751196173e-05,
+      "loss": 0.0533,
+      "step": 1109
+    },
+    {
+      "epoch": 100.91,
+      "learning_rate": 8.401913875598086e-05,
+      "loss": 0.0582,
+      "step": 1110
+    },
+    {
+      "epoch": 101.0,
+      "learning_rate": 8.4e-05,
+      "loss": 0.0587,
+      "step": 1111
+    },
+    {
+      "epoch": 101.09,
+      "learning_rate": 8.398086124401914e-05,
+      "loss": 0.0571,
+      "step": 1112
+    },
+    {
+      "epoch": 101.18,
+      "learning_rate": 8.396172248803829e-05,
+      "loss": 0.0516,
+      "step": 1113
+    },
+    {
+      "epoch": 101.27,
+      "learning_rate": 8.394258373205742e-05,
+      "loss": 0.0559,
+      "step": 1114
+    },
+    {
+      "epoch": 101.36,
+      "learning_rate": 8.392344497607656e-05,
+      "loss": 0.0635,
+      "step": 1115
+    },
+    {
+      "epoch": 101.45,
+      "learning_rate": 8.39043062200957e-05,
+      "loss": 0.0751,
+      "step": 1116
+    },
+    {
+      "epoch": 101.55,
+      "learning_rate": 8.388516746411483e-05,
+      "loss": 0.0638,
+      "step": 1117
+    },
+    {
+      "epoch": 101.64,
+      "learning_rate": 8.386602870813397e-05,
+      "loss": 0.0635,
+      "step": 1118
+    },
+    {
+      "epoch": 101.73,
+      "learning_rate": 8.384688995215312e-05,
+      "loss": 0.0553,
+      "step": 1119
+    },
+    {
+      "epoch": 101.82,
+      "learning_rate": 8.382775119617225e-05,
+      "loss": 0.0625,
+      "step": 1120
+    },
+    {
+      "epoch": 101.82,
+      "eval_accuracy_NAT": 0.99186066290567,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9373242950341216,
+      "eval_accuracy_concrete tiles": 0.979178132452761,
+      "eval_accuracy_dark bricks": 0.8209075863130749,
+      "eval_accuracy_grass": 0.8786021103254962,
+      "eval_accuracy_red bricks": 0.8895596590909091,
+      "eval_accuracy_road": 0.9782281946390935,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.648691864472197,
+      "eval_accuracy_speedway bricks": 0.9877954737173662,
+      "eval_accuracy_steel": 0.8742002506430975,
+      "eval_accuracy_tiles": 0.08272839337077113,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9770328373224345,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8803062004713683,
+      "eval_iou_concrete tiles": 0.9378569127595396,
+      "eval_iou_dark bricks": 0.7759022118742724,
+      "eval_iou_grass": 0.8100897475146223,
+      "eval_iou_red bricks": 0.8296075509190264,
+      "eval_iou_road": 0.9676586684529078,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.42287439570116636,
+      "eval_iou_speedway bricks": 0.9699808792012282,
+      "eval_iou_steel": 0.8078353091762411,
+      "eval_iou_tiles": 0.08272839337077113,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09188192337751389,
+      "eval_mean_accuracy": 0.6976212786895813,
+      "eval_mean_iou": 0.6509133159048905,
+      "eval_overall_accuracy": 0.975558704800076,
+      "eval_runtime": 3.5384,
+      "eval_samples_per_second": 10.174,
+      "eval_steps_per_second": 0.565,
+      "step": 1120
+    },
+    {
+      "epoch": 101.91,
+      "learning_rate": 8.38086124401914e-05,
+      "loss": 0.0598,
+      "step": 1121
+    },
+    {
+      "epoch": 102.0,
+      "learning_rate": 8.378947368421053e-05,
+      "loss": 0.0662,
+      "step": 1122
+    },
+    {
+      "epoch": 102.09,
+      "learning_rate": 8.377033492822966e-05,
+      "loss": 0.0581,
+      "step": 1123
+    },
+    {
+      "epoch": 102.18,
+      "learning_rate": 8.37511961722488e-05,
+      "loss": 0.061,
+      "step": 1124
+    },
+    {
+      "epoch": 102.27,
+      "learning_rate": 8.373205741626795e-05,
+      "loss": 0.0738,
+      "step": 1125
+    },
+    {
+      "epoch": 102.36,
+      "learning_rate": 8.371291866028708e-05,
+      "loss": 0.0568,
+      "step": 1126
+    },
+    {
+      "epoch": 102.45,
+      "learning_rate": 8.369377990430623e-05,
+      "loss": 0.0518,
+      "step": 1127
+    },
+    {
+      "epoch": 102.55,
+      "learning_rate": 8.367464114832536e-05,
+      "loss": 0.055,
+      "step": 1128
+    },
+    {
+      "epoch": 102.64,
+      "learning_rate": 8.36555023923445e-05,
+      "loss": 0.0558,
+      "step": 1129
+    },
+    {
+      "epoch": 102.73,
+      "learning_rate": 8.363636363636364e-05,
+      "loss": 0.0635,
+      "step": 1130
+    },
+    {
+      "epoch": 102.82,
+      "learning_rate": 8.361722488038278e-05,
+      "loss": 0.0709,
+      "step": 1131
+    },
+    {
+      "epoch": 102.91,
+      "learning_rate": 8.359808612440192e-05,
+      "loss": 0.0509,
+      "step": 1132
+    },
+    {
+      "epoch": 103.0,
+      "learning_rate": 8.357894736842106e-05,
+      "loss": 0.0677,
+      "step": 1133
+    },
+    {
+      "epoch": 103.09,
+      "learning_rate": 8.35598086124402e-05,
+      "loss": 0.0639,
+      "step": 1134
+    },
+    {
+      "epoch": 103.18,
+      "learning_rate": 8.354066985645933e-05,
+      "loss": 0.0705,
+      "step": 1135
+    },
+    {
+      "epoch": 103.27,
+      "learning_rate": 8.352153110047849e-05,
+      "loss": 0.0612,
+      "step": 1136
+    },
+    {
+      "epoch": 103.36,
+      "learning_rate": 8.350239234449762e-05,
+      "loss": 0.0551,
+      "step": 1137
+    },
+    {
+      "epoch": 103.45,
+      "learning_rate": 8.348325358851675e-05,
+      "loss": 0.0666,
+      "step": 1138
+    },
+    {
+      "epoch": 103.55,
+      "learning_rate": 8.34641148325359e-05,
+      "loss": 0.0587,
+      "step": 1139
+    },
+    {
+      "epoch": 103.64,
+      "learning_rate": 8.344497607655503e-05,
+      "loss": 0.0601,
+      "step": 1140
+    },
+    {
+      "epoch": 103.64,
+      "eval_accuracy_NAT": 0.9897266631047068,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9467613953388557,
+      "eval_accuracy_concrete tiles": 0.9738580077050083,
+      "eval_accuracy_dark bricks": 0.8004695739194027,
+      "eval_accuracy_grass": 0.9090442021970033,
+      "eval_accuracy_red bricks": 0.9341264204545454,
+      "eval_accuracy_road": 0.9836019187726046,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6284564276459075,
+      "eval_accuracy_speedway bricks": 0.9853450346175058,
+      "eval_accuracy_steel": 0.88078787678913,
+      "eval_accuracy_tiles": 0.01712094233666621,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9776361335661923,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8834126793420038,
+      "eval_iou_concrete tiles": 0.9350340535462659,
+      "eval_iou_dark bricks": 0.7719461044504659,
+      "eval_iou_grass": 0.8132099363734425,
+      "eval_iou_red bricks": 0.8064071122011036,
+      "eval_iou_road": 0.9689629664042426,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.41631967272992915,
+      "eval_iou_speedway bricks": 0.9704030017138389,
+      "eval_iou_steel": 0.8075853101706959,
+      "eval_iou_tiles": 0.01712094233666621,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0895157977938652,
+      "eval_mean_accuracy": 0.6960998817601027,
+      "eval_mean_iou": 0.6436952240642189,
+      "eval_overall_accuracy": 0.9758435355292426,
+      "eval_runtime": 3.2851,
+      "eval_samples_per_second": 10.959,
+      "eval_steps_per_second": 0.609,
+      "step": 1140
+    },
+    {
+      "epoch": 103.73,
+      "learning_rate": 8.342583732057416e-05,
+      "loss": 0.0527,
+      "step": 1141
+    },
+    {
+      "epoch": 103.82,
+      "learning_rate": 8.34066985645933e-05,
+      "loss": 0.0631,
+      "step": 1142
+    },
+    {
+      "epoch": 103.91,
+      "learning_rate": 8.338755980861245e-05,
+      "loss": 0.0611,
+      "step": 1143
+    },
+    {
+      "epoch": 104.0,
+      "learning_rate": 8.336842105263158e-05,
+      "loss": 0.062,
+      "step": 1144
+    },
+    {
+      "epoch": 104.09,
+      "learning_rate": 8.334928229665073e-05,
+      "loss": 0.0572,
+      "step": 1145
+    },
+    {
+      "epoch": 104.18,
+      "learning_rate": 8.333014354066986e-05,
+      "loss": 0.0595,
+      "step": 1146
+    },
+    {
+      "epoch": 104.27,
+      "learning_rate": 8.331100478468899e-05,
+      "loss": 0.0612,
+      "step": 1147
+    },
+    {
+      "epoch": 104.36,
+      "learning_rate": 8.329186602870814e-05,
+      "loss": 0.0564,
+      "step": 1148
+    },
+    {
+      "epoch": 104.45,
+      "learning_rate": 8.327272727272728e-05,
+      "loss": 0.0527,
+      "step": 1149
+    },
+    {
+      "epoch": 104.55,
+      "learning_rate": 8.325358851674641e-05,
+      "loss": 0.0598,
+      "step": 1150
+    },
+    {
+      "epoch": 104.64,
+      "learning_rate": 8.323444976076556e-05,
+      "loss": 0.0531,
+      "step": 1151
+    },
+    {
+      "epoch": 104.73,
+      "learning_rate": 8.321531100478469e-05,
+      "loss": 0.0583,
+      "step": 1152
+    },
+    {
+      "epoch": 104.82,
+      "learning_rate": 8.319617224880382e-05,
+      "loss": 0.0552,
+      "step": 1153
+    },
+    {
+      "epoch": 104.91,
+      "learning_rate": 8.317703349282297e-05,
+      "loss": 0.056,
+      "step": 1154
+    },
+    {
+      "epoch": 105.0,
+      "learning_rate": 8.315789473684212e-05,
+      "loss": 0.0685,
+      "step": 1155
+    },
+    {
+      "epoch": 105.09,
+      "learning_rate": 8.313875598086125e-05,
+      "loss": 0.0551,
+      "step": 1156
+    },
+    {
+      "epoch": 105.18,
+      "learning_rate": 8.311961722488039e-05,
+      "loss": 0.0618,
+      "step": 1157
+    },
+    {
+      "epoch": 105.27,
+      "learning_rate": 8.310047846889952e-05,
+      "loss": 0.0552,
+      "step": 1158
+    },
+    {
+      "epoch": 105.36,
+      "learning_rate": 8.308133971291866e-05,
+      "loss": 0.0721,
+      "step": 1159
+    },
+    {
+      "epoch": 105.45,
+      "learning_rate": 8.30622009569378e-05,
+      "loss": 0.0605,
+      "step": 1160
+    },
+    {
+      "epoch": 105.45,
+      "eval_accuracy_NAT": 0.9920129402190676,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9460867419202541,
+      "eval_accuracy_concrete tiles": 0.9788112272977435,
+      "eval_accuracy_dark bricks": 0.8182902890573881,
+      "eval_accuracy_grass": 0.885676139481401,
+      "eval_accuracy_red bricks": 0.9195667613636364,
+      "eval_accuracy_road": 0.9817699673634531,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6502632812284619,
+      "eval_accuracy_speedway bricks": 0.9806113003800262,
+      "eval_accuracy_steel": 0.8977475100587032,
+      "eval_accuracy_tiles": 0.07711272428434461,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9779814330542569,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8784537404884367,
+      "eval_iou_concrete tiles": 0.9327525421753445,
+      "eval_iou_dark bricks": 0.7845018450184502,
+      "eval_iou_grass": 0.8142950790172173,
+      "eval_iou_red bricks": 0.7904456654456654,
+      "eval_iou_road": 0.9699497649525425,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.42373886174187986,
+      "eval_iou_speedway bricks": 0.9703953123572506,
+      "eval_iou_steel": 0.8112622748066578,
+      "eval_iou_tiles": 0.07711272428434461,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08989302068948746,
+      "eval_mean_accuracy": 0.7021499140503445,
+      "eval_mean_iou": 0.6485299417955421,
+      "eval_overall_accuracy": 0.9759150611029731,
+      "eval_runtime": 3.5167,
+      "eval_samples_per_second": 10.237,
+      "eval_steps_per_second": 0.569,
+      "step": 1160
+    },
+    {
+      "epoch": 105.55,
+      "learning_rate": 8.304306220095695e-05,
+      "loss": 0.0579,
+      "step": 1161
+    },
+    {
+      "epoch": 105.64,
+      "learning_rate": 8.302392344497608e-05,
+      "loss": 0.0454,
+      "step": 1162
+    },
+    {
+      "epoch": 105.73,
+      "learning_rate": 8.300478468899521e-05,
+      "loss": 0.0557,
+      "step": 1163
+    },
+    {
+      "epoch": 105.82,
+      "learning_rate": 8.298564593301436e-05,
+      "loss": 0.0675,
+      "step": 1164
+    },
+    {
+      "epoch": 105.91,
+      "learning_rate": 8.296650717703349e-05,
+      "loss": 0.0599,
+      "step": 1165
+    },
+    {
+      "epoch": 106.0,
+      "learning_rate": 8.294736842105263e-05,
+      "loss": 0.0604,
+      "step": 1166
+    },
+    {
+      "epoch": 106.09,
+      "learning_rate": 8.292822966507178e-05,
+      "loss": 0.0588,
+      "step": 1167
+    },
+    {
+      "epoch": 106.18,
+      "learning_rate": 8.290909090909091e-05,
+      "loss": 0.0541,
+      "step": 1168
+    },
+    {
+      "epoch": 106.27,
+      "learning_rate": 8.288995215311004e-05,
+      "loss": 0.0636,
+      "step": 1169
+    },
+    {
+      "epoch": 106.36,
+      "learning_rate": 8.287081339712919e-05,
+      "loss": 0.0566,
+      "step": 1170
+    },
+    {
+      "epoch": 106.45,
+      "learning_rate": 8.285167464114832e-05,
+      "loss": 0.0545,
+      "step": 1171
+    },
+    {
+      "epoch": 106.55,
+      "learning_rate": 8.283253588516747e-05,
+      "loss": 0.0589,
+      "step": 1172
+    },
+    {
+      "epoch": 106.64,
+      "learning_rate": 8.281339712918661e-05,
+      "loss": 0.0617,
+      "step": 1173
+    },
+    {
+      "epoch": 106.73,
+      "learning_rate": 8.279425837320575e-05,
+      "loss": 0.0569,
+      "step": 1174
+    },
+    {
+      "epoch": 106.82,
+      "learning_rate": 8.277511961722488e-05,
+      "loss": 0.0579,
+      "step": 1175
+    },
+    {
+      "epoch": 106.91,
+      "learning_rate": 8.275598086124402e-05,
+      "loss": 0.0579,
+      "step": 1176
+    },
+    {
+      "epoch": 107.0,
+      "learning_rate": 8.273684210526315e-05,
+      "loss": 0.0648,
+      "step": 1177
+    },
+    {
+      "epoch": 107.09,
+      "learning_rate": 8.27177033492823e-05,
+      "loss": 0.0582,
+      "step": 1178
+    },
+    {
+      "epoch": 107.18,
+      "learning_rate": 8.269856459330145e-05,
+      "loss": 0.053,
+      "step": 1179
+    },
+    {
+      "epoch": 107.27,
+      "learning_rate": 8.267942583732058e-05,
+      "loss": 0.0549,
+      "step": 1180
+    },
+    {
+      "epoch": 107.27,
+      "eval_accuracy_NAT": 0.9917161494413911,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.948535881368299,
+      "eval_accuracy_concrete tiles": 0.9822662508408243,
+      "eval_accuracy_dark bricks": 0.8331472999499634,
+      "eval_accuracy_grass": 0.8989014983380924,
+      "eval_accuracy_red bricks": 0.8638139204545454,
+      "eval_accuracy_road": 0.9702812740708137,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6363962175722989,
+      "eval_accuracy_speedway bricks": 0.9841551144242222,
+      "eval_accuracy_steel": 0.8858172284150122,
+      "eval_accuracy_tiles": 0.08211203944665114,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9783798293441271,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8787983095014122,
+      "eval_iou_concrete tiles": 0.9191988555078684,
+      "eval_iou_dark bricks": 0.7879008481054126,
+      "eval_iou_grass": 0.8144729175111471,
+      "eval_iou_red bricks": 0.8186101295641932,
+      "eval_iou_road": 0.9622134470771212,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4237851333737218,
+      "eval_iou_speedway bricks": 0.9716192194138574,
+      "eval_iou_steel": 0.8131494698282713,
+      "eval_iou_tiles": 0.08210641648976237,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09069426357746124,
+      "eval_mean_accuracy": 0.6982417595632395,
+      "eval_mean_iou": 0.6500180442859149,
+      "eval_overall_accuracy": 0.975767559475369,
+      "eval_runtime": 4.0341,
+      "eval_samples_per_second": 8.924,
+      "eval_steps_per_second": 0.496,
+      "step": 1180
+    },
+    {
+      "epoch": 107.36,
+      "learning_rate": 8.266028708133971e-05,
+      "loss": 0.0549,
+      "step": 1181
+    },
+    {
+      "epoch": 107.45,
+      "learning_rate": 8.264114832535886e-05,
+      "loss": 0.0645,
+      "step": 1182
+    },
+    {
+      "epoch": 107.55,
+      "learning_rate": 8.262200956937799e-05,
+      "loss": 0.0513,
+      "step": 1183
+    },
+    {
+      "epoch": 107.64,
+      "learning_rate": 8.260287081339713e-05,
+      "loss": 0.0588,
+      "step": 1184
+    },
+    {
+      "epoch": 107.73,
+      "learning_rate": 8.258373205741628e-05,
+      "loss": 0.062,
+      "step": 1185
+    },
+    {
+      "epoch": 107.82,
+      "learning_rate": 8.256459330143541e-05,
+      "loss": 0.0669,
+      "step": 1186
+    },
+    {
+      "epoch": 107.91,
+      "learning_rate": 8.254545454545454e-05,
+      "loss": 0.0531,
+      "step": 1187
+    },
+    {
+      "epoch": 108.0,
+      "learning_rate": 8.252631578947369e-05,
+      "loss": 0.0615,
+      "step": 1188
+    },
+    {
+      "epoch": 108.09,
+      "learning_rate": 8.250717703349282e-05,
+      "loss": 0.0643,
+      "step": 1189
+    },
+    {
+      "epoch": 108.18,
+      "learning_rate": 8.248803827751197e-05,
+      "loss": 0.0635,
+      "step": 1190
+    },
+    {
+      "epoch": 108.27,
+      "learning_rate": 8.246889952153111e-05,
+      "loss": 0.0511,
+      "step": 1191
+    },
+    {
+      "epoch": 108.36,
+      "learning_rate": 8.244976076555024e-05,
+      "loss": 0.0619,
+      "step": 1192
+    },
+    {
+      "epoch": 108.45,
+      "learning_rate": 8.243062200956938e-05,
+      "loss": 0.0541,
+      "step": 1193
+    },
+    {
+      "epoch": 108.55,
+      "learning_rate": 8.241148325358852e-05,
+      "loss": 0.0551,
+      "step": 1194
+    },
+    {
+      "epoch": 108.64,
+      "learning_rate": 8.239234449760765e-05,
+      "loss": 0.0574,
+      "step": 1195
+    },
+    {
+      "epoch": 108.73,
+      "learning_rate": 8.23732057416268e-05,
+      "loss": 0.0571,
+      "step": 1196
+    },
+    {
+      "epoch": 108.82,
+      "learning_rate": 8.235406698564594e-05,
+      "loss": 0.0665,
+      "step": 1197
+    },
+    {
+      "epoch": 108.91,
+      "learning_rate": 8.233492822966508e-05,
+      "loss": 0.0666,
+      "step": 1198
+    },
+    {
+      "epoch": 109.0,
+      "learning_rate": 8.231578947368421e-05,
+      "loss": 0.0532,
+      "step": 1199
+    },
+    {
+      "epoch": 109.09,
+      "learning_rate": 8.229665071770335e-05,
+      "loss": 0.0582,
+      "step": 1200
+    },
+    {
+      "epoch": 109.09,
+      "eval_accuracy_NAT": 0.9919693567933326,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9445952615992103,
+      "eval_accuracy_concrete tiles": 0.9728490185287103,
+      "eval_accuracy_dark bricks": 0.8003541049228282,
+      "eval_accuracy_grass": 0.8876298856761394,
+      "eval_accuracy_red bricks": 0.8927556818181818,
+      "eval_accuracy_road": 0.9797360780761971,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.643260827612825,
+      "eval_accuracy_speedway bricks": 0.9875450594779074,
+      "eval_accuracy_steel": 0.8584443638282435,
+      "eval_accuracy_tiles": 0.07745514313107793,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9777362381108743,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8830137769508153,
+      "eval_iou_concrete tiles": 0.9505571654766528,
+      "eval_iou_dark bricks": 0.7727238944630249,
+      "eval_iou_grass": 0.8183660777294068,
+      "eval_iou_red bricks": 0.8218372016999019,
+      "eval_iou_road": 0.9688338895578463,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.42423636363636363,
+      "eval_iou_speedway bricks": 0.9712502477659523,
+      "eval_iou_steel": 0.8091940623299915,
+      "eval_iou_tiles": 0.07745514313107793,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08953683823347092,
+      "eval_mean_accuracy": 0.695122675497281,
+      "eval_mean_iou": 0.6519387739116851,
+      "eval_overall_accuracy": 0.9762521319919162,
+      "eval_runtime": 3.4888,
+      "eval_samples_per_second": 10.319,
+      "eval_steps_per_second": 0.573,
+      "step": 1200
+    },
+    {
+      "epoch": 109.18,
+      "learning_rate": 8.227751196172249e-05,
+      "loss": 0.0597,
+      "step": 1201
+    },
+    {
+      "epoch": 109.27,
+      "learning_rate": 8.225837320574163e-05,
+      "loss": 0.0808,
+      "step": 1202
+    },
+    {
+      "epoch": 109.36,
+      "learning_rate": 8.223923444976078e-05,
+      "loss": 0.055,
+      "step": 1203
+    },
+    {
+      "epoch": 109.45,
+      "learning_rate": 8.222009569377991e-05,
+      "loss": 0.0633,
+      "step": 1204
+    },
+    {
+      "epoch": 109.55,
+      "learning_rate": 8.220095693779904e-05,
+      "loss": 0.0619,
+      "step": 1205
+    },
+    {
+      "epoch": 109.64,
+      "learning_rate": 8.218181818181819e-05,
+      "loss": 0.0572,
+      "step": 1206
+    },
+    {
+      "epoch": 109.73,
+      "learning_rate": 8.216267942583733e-05,
+      "loss": 0.0663,
+      "step": 1207
+    },
+    {
+      "epoch": 109.82,
+      "learning_rate": 8.214354066985646e-05,
+      "loss": 0.0461,
+      "step": 1208
+    },
+    {
+      "epoch": 109.91,
+      "learning_rate": 8.212440191387561e-05,
+      "loss": 0.0556,
+      "step": 1209
+    },
+    {
+      "epoch": 110.0,
+      "learning_rate": 8.210526315789474e-05,
+      "loss": 0.0599,
+      "step": 1210
+    },
+    {
+      "epoch": 110.09,
+      "learning_rate": 8.208612440191387e-05,
+      "loss": 0.057,
+      "step": 1211
+    },
+    {
+      "epoch": 110.18,
+      "learning_rate": 8.206698564593302e-05,
+      "loss": 0.0506,
+      "step": 1212
+    },
+    {
+      "epoch": 110.27,
+      "learning_rate": 8.204784688995216e-05,
+      "loss": 0.0665,
+      "step": 1213
+    },
+    {
+      "epoch": 110.36,
+      "learning_rate": 8.20287081339713e-05,
+      "loss": 0.0583,
+      "step": 1214
+    },
+    {
+      "epoch": 110.45,
+      "learning_rate": 8.200956937799044e-05,
+      "loss": 0.0584,
+      "step": 1215
+    },
+    {
+      "epoch": 110.55,
+      "learning_rate": 8.199043062200957e-05,
+      "loss": 0.0596,
+      "step": 1216
+    },
+    {
+      "epoch": 110.64,
+      "learning_rate": 8.19712918660287e-05,
+      "loss": 0.0583,
+      "step": 1217
+    },
+    {
+      "epoch": 110.73,
+      "learning_rate": 8.195215311004785e-05,
+      "loss": 0.0563,
+      "step": 1218
+    },
+    {
+      "epoch": 110.82,
+      "learning_rate": 8.1933014354067e-05,
+      "loss": 0.0564,
+      "step": 1219
+    },
+    {
+      "epoch": 110.91,
+      "learning_rate": 8.191387559808613e-05,
+      "loss": 0.0601,
+      "step": 1220
+    },
+    {
+      "epoch": 110.91,
+      "eval_accuracy_NAT": 0.9917304419818143,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9413091763595004,
+      "eval_accuracy_concrete tiles": 0.9817770439674678,
+      "eval_accuracy_dark bricks": 0.8577421962203149,
+      "eval_accuracy_grass": 0.8822055313036276,
+      "eval_accuracy_red bricks": 0.9138849431818182,
+      "eval_accuracy_road": 0.9710648414062132,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6408072119758498,
+      "eval_accuracy_speedway bricks": 0.9857831086846072,
+      "eval_accuracy_steel": 0.8868230987401886,
+      "eval_accuracy_tiles": 0.06293658402958499,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9770247956008707,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8746821875467371,
+      "eval_iou_concrete tiles": 0.9266687830077054,
+      "eval_iou_dark bricks": 0.798259125264176,
+      "eval_iou_grass": 0.8055414604256623,
+      "eval_iou_red bricks": 0.7878463186897291,
+      "eval_iou_road": 0.9624815128205743,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4185091825711199,
+      "eval_iou_speedway bricks": 0.9719422388857093,
+      "eval_iou_steel": 0.8182157038750019,
+      "eval_iou_tiles": 0.06293658402958499,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09376034885644913,
+      "eval_mean_accuracy": 0.7012357059885374,
+      "eval_mean_iou": 0.6464698379012978,
+      "eval_overall_accuracy": 0.9751496844821506,
+      "eval_runtime": 3.306,
+      "eval_samples_per_second": 10.889,
+      "eval_steps_per_second": 0.605,
+      "step": 1220
+    },
+    {
+      "epoch": 111.0,
+      "learning_rate": 8.189473684210527e-05,
+      "loss": 0.0571,
+      "step": 1221
+    },
+    {
+      "epoch": 111.09,
+      "learning_rate": 8.18755980861244e-05,
+      "loss": 0.0637,
+      "step": 1222
+    },
+    {
+      "epoch": 111.18,
+      "learning_rate": 8.185645933014354e-05,
+      "loss": 0.0565,
+      "step": 1223
+    },
+    {
+      "epoch": 111.27,
+      "learning_rate": 8.183732057416268e-05,
+      "loss": 0.0557,
+      "step": 1224
+    },
+    {
+      "epoch": 111.36,
+      "learning_rate": 8.181818181818183e-05,
+      "loss": 0.0616,
+      "step": 1225
+    },
+    {
+      "epoch": 111.45,
+      "learning_rate": 8.179904306220096e-05,
+      "loss": 0.0798,
+      "step": 1226
+    },
+    {
+      "epoch": 111.55,
+      "learning_rate": 8.177990430622011e-05,
+      "loss": 0.0587,
+      "step": 1227
+    },
+    {
+      "epoch": 111.64,
+      "learning_rate": 8.176076555023924e-05,
+      "loss": 0.075,
+      "step": 1228
+    },
+    {
+      "epoch": 111.73,
+      "learning_rate": 8.174162679425837e-05,
+      "loss": 0.0546,
+      "step": 1229
+    },
+    {
+      "epoch": 111.82,
+      "learning_rate": 8.172248803827752e-05,
+      "loss": 0.0501,
+      "step": 1230
+    },
+    {
+      "epoch": 111.91,
+      "learning_rate": 8.170334928229666e-05,
+      "loss": 0.0576,
+      "step": 1231
+    },
+    {
+      "epoch": 112.0,
+      "learning_rate": 8.16842105263158e-05,
+      "loss": 0.0588,
+      "step": 1232
+    },
+    {
+      "epoch": 112.09,
+      "learning_rate": 8.166507177033494e-05,
+      "loss": 0.0512,
+      "step": 1233
+    },
+    {
+      "epoch": 112.18,
+      "learning_rate": 8.164593301435407e-05,
+      "loss": 0.0578,
+      "step": 1234
+    },
+    {
+      "epoch": 112.27,
+      "learning_rate": 8.16267942583732e-05,
+      "loss": 0.0766,
+      "step": 1235
+    },
+    {
+      "epoch": 112.36,
+      "learning_rate": 8.160765550239235e-05,
+      "loss": 0.0635,
+      "step": 1236
+    },
+    {
+      "epoch": 112.45,
+      "learning_rate": 8.15885167464115e-05,
+      "loss": 0.0529,
+      "step": 1237
+    },
+    {
+      "epoch": 112.55,
+      "learning_rate": 8.156937799043063e-05,
+      "loss": 0.0548,
+      "step": 1238
+    },
+    {
+      "epoch": 112.64,
+      "learning_rate": 8.155023923444977e-05,
+      "loss": 0.0478,
+      "step": 1239
+    },
+    {
+      "epoch": 112.73,
+      "learning_rate": 8.15311004784689e-05,
+      "loss": 0.0507,
+      "step": 1240
+    },
+    {
+      "epoch": 112.73,
+      "eval_accuracy_NAT": 0.9924701250614932,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9471973046053479,
+      "eval_accuracy_concrete tiles": 0.9834586925946309,
+      "eval_accuracy_dark bricks": 0.8411146607135984,
+      "eval_accuracy_grass": 0.861109266488884,
+      "eval_accuracy_red bricks": 0.9334161931818182,
+      "eval_accuracy_road": 0.9792886708013548,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.641551567281449,
+      "eval_accuracy_speedway bricks": 0.9833972342199807,
+      "eval_accuracy_steel": 0.8931221555306378,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.977311285315044,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8787877656615734,
+      "eval_iou_concrete tiles": 0.9250258828942828,
+      "eval_iou_dark bricks": 0.8051656165948197,
+      "eval_iou_grass": 0.8029945506218549,
+      "eval_iou_red bricks": 0.7785841232227488,
+      "eval_iou_road": 0.9697493300380065,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4215226329994385,
+      "eval_iou_speedway bricks": 0.9716657435357484,
+      "eval_iou_steel": 0.8161120444802724,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09003829210996628,
+      "eval_mean_accuracy": 0.696625066959938,
+      "eval_mean_iou": 0.6420706904125991,
+      "eval_overall_accuracy": 0.9756792916191949,
+      "eval_runtime": 3.5573,
+      "eval_samples_per_second": 10.12,
+      "eval_steps_per_second": 0.562,
+      "step": 1240
+    },
+    {
+      "epoch": 112.82,
+      "learning_rate": 8.151196172248804e-05,
+      "loss": 0.0568,
+      "step": 1241
+    },
+    {
+      "epoch": 112.91,
+      "learning_rate": 8.149282296650718e-05,
+      "loss": 0.061,
+      "step": 1242
+    },
+    {
+      "epoch": 113.0,
+      "learning_rate": 8.147368421052633e-05,
+      "loss": 0.0603,
+      "step": 1243
+    },
+    {
+      "epoch": 113.09,
+      "learning_rate": 8.145454545454546e-05,
+      "loss": 0.0561,
+      "step": 1244
+    },
+    {
+      "epoch": 113.18,
+      "learning_rate": 8.143540669856459e-05,
+      "loss": 0.0598,
+      "step": 1245
+    },
+    {
+      "epoch": 113.27,
+      "learning_rate": 8.141626794258374e-05,
+      "loss": 0.0523,
+      "step": 1246
+    },
+    {
+      "epoch": 113.36,
+      "learning_rate": 8.139712918660287e-05,
+      "loss": 0.0549,
+      "step": 1247
+    },
+    {
+      "epoch": 113.45,
+      "learning_rate": 8.137799043062201e-05,
+      "loss": 0.0573,
+      "step": 1248
+    },
+    {
+      "epoch": 113.55,
+      "learning_rate": 8.135885167464116e-05,
+      "loss": 0.0554,
+      "step": 1249
+    },
+    {
+      "epoch": 113.64,
+      "learning_rate": 8.133971291866029e-05,
+      "loss": 0.0575,
+      "step": 1250
+    },
+    {
+      "epoch": 113.73,
+      "learning_rate": 8.132057416267942e-05,
+      "loss": 0.0528,
+      "step": 1251
+    },
+    {
+      "epoch": 113.82,
+      "learning_rate": 8.130143540669857e-05,
+      "loss": 0.0548,
+      "step": 1252
+    },
+    {
+      "epoch": 113.91,
+      "learning_rate": 8.12822966507177e-05,
+      "loss": 0.0565,
+      "step": 1253
+    },
+    {
+      "epoch": 114.0,
+      "learning_rate": 8.126315789473685e-05,
+      "loss": 0.0686,
+      "step": 1254
+    },
+    {
+      "epoch": 114.09,
+      "learning_rate": 8.124401913875599e-05,
+      "loss": 0.0624,
+      "step": 1255
+    },
+    {
+      "epoch": 114.18,
+      "learning_rate": 8.122488038277512e-05,
+      "loss": 0.0592,
+      "step": 1256
+    },
+    {
+      "epoch": 114.27,
+      "learning_rate": 8.120574162679426e-05,
+      "loss": 0.0512,
+      "step": 1257
+    },
+    {
+      "epoch": 114.36,
+      "learning_rate": 8.11866028708134e-05,
+      "loss": 0.0507,
+      "step": 1258
+    },
+    {
+      "epoch": 114.45,
+      "learning_rate": 8.116746411483253e-05,
+      "loss": 0.0495,
+      "step": 1259
+    },
+    {
+      "epoch": 114.55,
+      "learning_rate": 8.114832535885168e-05,
+      "loss": 0.0481,
+      "step": 1260
+    },
+    {
+      "epoch": 114.55,
+      "eval_accuracy_NAT": 0.992069404576295,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9439903536632474,
+      "eval_accuracy_concrete tiles": 0.9854766709472268,
+      "eval_accuracy_dark bricks": 0.802971402178515,
+      "eval_accuracy_grass": 0.8811360387533771,
+      "eval_accuracy_red bricks": 0.9060724431818182,
+      "eval_accuracy_road": 0.9731192872439349,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6327020097593251,
+      "eval_accuracy_speedway bricks": 0.9849570433982963,
+      "eval_accuracy_steel": 0.9022821713607282,
+      "eval_accuracy_tiles": 0.07149705519791809,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9773199965721165,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8786311107338186,
+      "eval_iou_concrete tiles": 0.9205700902547698,
+      "eval_iou_dark bricks": 0.784226749868431,
+      "eval_iou_grass": 0.8095667783090513,
+      "eval_iou_red bricks": 0.8196273690973338,
+      "eval_iou_road": 0.9654365330722868,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4137446141087815,
+      "eval_iou_speedway bricks": 0.9721045876968613,
+      "eval_iou_steel": 0.8166378126679005,
+      "eval_iou_tiles": 0.07149705519791809,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09019369632005692,
+      "eval_mean_accuracy": 0.6981749138662063,
+      "eval_mean_iou": 0.6484125151984053,
+      "eval_overall_accuracy": 0.9756057527330186,
+      "eval_runtime": 3.3345,
+      "eval_samples_per_second": 10.796,
+      "eval_steps_per_second": 0.6,
+      "step": 1260
+    },
+    {
+      "epoch": 114.64,
+      "learning_rate": 8.112918660287083e-05,
+      "loss": 0.0483,
+      "step": 1261
+    },
+    {
+      "epoch": 114.73,
+      "learning_rate": 8.111004784688996e-05,
+      "loss": 0.0652,
+      "step": 1262
+    },
+    {
+      "epoch": 114.82,
+      "learning_rate": 8.109090909090909e-05,
+      "loss": 0.0526,
+      "step": 1263
+    },
+    {
+      "epoch": 114.91,
+      "learning_rate": 8.107177033492824e-05,
+      "loss": 0.0651,
+      "step": 1264
+    },
+    {
+      "epoch": 115.0,
+      "learning_rate": 8.105263157894737e-05,
+      "loss": 0.0655,
+      "step": 1265
+    },
+    {
+      "epoch": 115.09,
+      "learning_rate": 8.10334928229665e-05,
+      "loss": 0.0576,
+      "step": 1266
+    },
+    {
+      "epoch": 115.18,
+      "learning_rate": 8.101435406698566e-05,
+      "loss": 0.0587,
+      "step": 1267
+    },
+    {
+      "epoch": 115.27,
+      "learning_rate": 8.099521531100479e-05,
+      "loss": 0.0621,
+      "step": 1268
+    },
+    {
+      "epoch": 115.36,
+      "learning_rate": 8.097607655502392e-05,
+      "loss": 0.0656,
+      "step": 1269
+    },
+    {
+      "epoch": 115.45,
+      "learning_rate": 8.095693779904307e-05,
+      "loss": 0.0636,
+      "step": 1270
+    },
+    {
+      "epoch": 115.55,
+      "learning_rate": 8.09377990430622e-05,
+      "loss": 0.0473,
+      "step": 1271
+    },
+    {
+      "epoch": 115.64,
+      "learning_rate": 8.091866028708133e-05,
+      "loss": 0.0697,
+      "step": 1272
+    },
+    {
+      "epoch": 115.73,
+      "learning_rate": 8.089952153110049e-05,
+      "loss": 0.0512,
+      "step": 1273
+    },
+    {
+      "epoch": 115.82,
+      "learning_rate": 8.088038277511962e-05,
+      "loss": 0.0677,
+      "step": 1274
+    },
+    {
+      "epoch": 115.91,
+      "learning_rate": 8.086124401913875e-05,
+      "loss": 0.0528,
+      "step": 1275
+    },
+    {
+      "epoch": 116.0,
+      "learning_rate": 8.08421052631579e-05,
+      "loss": 0.0449,
+      "step": 1276
+    },
+    {
+      "epoch": 116.09,
+      "learning_rate": 8.082296650717703e-05,
+      "loss": 0.0578,
+      "step": 1277
+    },
+    {
+      "epoch": 116.18,
+      "learning_rate": 8.080382775119618e-05,
+      "loss": 0.0555,
+      "step": 1278
+    },
+    {
+      "epoch": 116.27,
+      "learning_rate": 8.078468899521532e-05,
+      "loss": 0.0515,
+      "step": 1279
+    },
+    {
+      "epoch": 116.36,
+      "learning_rate": 8.076555023923446e-05,
+      "loss": 0.0489,
+      "step": 1280
+    },
+    {
+      "epoch": 116.36,
+      "eval_accuracy_NAT": 0.9922377389412792,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9396500386282673,
+      "eval_accuracy_concrete tiles": 0.9819604965449764,
+      "eval_accuracy_dark bricks": 0.8691351372156576,
+      "eval_accuracy_grass": 0.8745618051356613,
+      "eval_accuracy_red bricks": 0.9115767045454546,
+      "eval_accuracy_road": 0.9742341778044878,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6309927494279491,
+      "eval_accuracy_speedway bricks": 0.9871709466382341,
+      "eval_accuracy_steel": 0.8815628916298397,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9778551912758353,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8761072128437315,
+      "eval_iou_concrete tiles": 0.924680409996545,
+      "eval_iou_dark bricks": 0.8086302596239928,
+      "eval_iou_grass": 0.8036142682343269,
+      "eval_iou_red bricks": 0.7027101012866137,
+      "eval_iou_road": 0.963644409891457,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4167895839023946,
+      "eval_iou_speedway bricks": 0.9717086643403371,
+      "eval_iou_steel": 0.8185743486881895,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0926472619175911,
+      "eval_mean_accuracy": 0.6956217451162929,
+      "eval_mean_iou": 0.6357164961602634,
+      "eval_overall_accuracy": 0.9754408730400933,
+      "eval_runtime": 3.5915,
+      "eval_samples_per_second": 10.024,
+      "eval_steps_per_second": 0.557,
+      "step": 1280
+    },
+    {
+      "epoch": 116.45,
+      "learning_rate": 8.074641148325359e-05,
+      "loss": 0.0575,
+      "step": 1281
+    },
+    {
+      "epoch": 116.55,
+      "learning_rate": 8.072727272727273e-05,
+      "loss": 0.0626,
+      "step": 1282
+    },
+    {
+      "epoch": 116.64,
+      "learning_rate": 8.070813397129187e-05,
+      "loss": 0.0571,
+      "step": 1283
+    },
+    {
+      "epoch": 116.73,
+      "learning_rate": 8.068899521531101e-05,
+      "loss": 0.0516,
+      "step": 1284
+    },
+    {
+      "epoch": 116.82,
+      "learning_rate": 8.066985645933016e-05,
+      "loss": 0.0543,
+      "step": 1285
+    },
+    {
+      "epoch": 116.91,
+      "learning_rate": 8.065071770334929e-05,
+      "loss": 0.0652,
+      "step": 1286
+    },
+    {
+      "epoch": 117.0,
+      "learning_rate": 8.063157894736842e-05,
+      "loss": 0.0626,
+      "step": 1287
+    },
+    {
+      "epoch": 117.09,
+      "learning_rate": 8.061244019138757e-05,
+      "loss": 0.058,
+      "step": 1288
+    },
+    {
+      "epoch": 117.18,
+      "learning_rate": 8.05933014354067e-05,
+      "loss": 0.0603,
+      "step": 1289
+    },
+    {
+      "epoch": 117.27,
+      "learning_rate": 8.057416267942584e-05,
+      "loss": 0.0541,
+      "step": 1290
+    },
+    {
+      "epoch": 117.36,
+      "learning_rate": 8.055502392344499e-05,
+      "loss": 0.0586,
+      "step": 1291
+    },
+    {
+      "epoch": 117.45,
+      "learning_rate": 8.053588516746412e-05,
+      "loss": 0.0501,
+      "step": 1292
+    },
+    {
+      "epoch": 117.55,
+      "learning_rate": 8.051674641148325e-05,
+      "loss": 0.0671,
+      "step": 1293
+    },
+    {
+      "epoch": 117.64,
+      "learning_rate": 8.04976076555024e-05,
+      "loss": 0.0539,
+      "step": 1294
+    },
+    {
+      "epoch": 117.73,
+      "learning_rate": 8.047846889952153e-05,
+      "loss": 0.0602,
+      "step": 1295
+    },
+    {
+      "epoch": 117.82,
+      "learning_rate": 8.045933014354068e-05,
+      "loss": 0.053,
+      "step": 1296
+    },
+    {
+      "epoch": 117.91,
+      "learning_rate": 8.044019138755982e-05,
+      "loss": 0.0466,
+      "step": 1297
+    },
+    {
+      "epoch": 118.0,
+      "learning_rate": 8.042105263157895e-05,
+      "loss": 0.0532,
+      "step": 1298
+    },
+    {
+      "epoch": 118.09,
+      "learning_rate": 8.040191387559809e-05,
+      "loss": 0.0602,
+      "step": 1299
+    },
+    {
+      "epoch": 118.18,
+      "learning_rate": 8.038277511961723e-05,
+      "loss": 0.0496,
+      "step": 1300
+    },
+    {
+      "epoch": 118.18,
+      "eval_accuracy_NAT": 0.9919492413660704,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9472013283831924,
+      "eval_accuracy_concrete tiles": 0.9865773864122791,
+      "eval_accuracy_dark bricks": 0.8211770139717486,
+      "eval_accuracy_grass": 0.8729855268998347,
+      "eval_accuracy_red bricks": 0.9156605113636364,
+      "eval_accuracy_road": 0.9680381916523476,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.622115623190803,
+      "eval_accuracy_speedway bricks": 0.9866562397797803,
+      "eval_accuracy_steel": 0.88294802453664,
+      "eval_accuracy_tiles": 0.06834680180797151,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9779473853226809,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8720170401926283,
+      "eval_iou_concrete tiles": 0.9236824778862394,
+      "eval_iou_dark bricks": 0.7957851547929877,
+      "eval_iou_grass": 0.8000435614007553,
+      "eval_iou_red bricks": 0.7025885558583106,
+      "eval_iou_road": 0.9614359178830407,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.406323712119848,
+      "eval_iou_speedway bricks": 0.9726902666132873,
+      "eval_iou_steel": 0.8193273453399232,
+      "eval_iou_tiles": 0.06834680180797151,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09199370443820953,
+      "eval_mean_accuracy": 0.6972042991818694,
+      "eval_mean_iou": 0.638476016862898,
+      "eval_overall_accuracy": 0.9751589033338759,
+      "eval_runtime": 3.3973,
+      "eval_samples_per_second": 10.597,
+      "eval_steps_per_second": 0.589,
+      "step": 1300
+    },
+    {
+      "epoch": 118.27,
+      "learning_rate": 8.036363636363636e-05,
+      "loss": 0.0516,
+      "step": 1301
+    },
+    {
+      "epoch": 118.36,
+      "learning_rate": 8.034449760765551e-05,
+      "loss": 0.0539,
+      "step": 1302
+    },
+    {
+      "epoch": 118.45,
+      "learning_rate": 8.032535885167465e-05,
+      "loss": 0.0486,
+      "step": 1303
+    },
+    {
+      "epoch": 118.55,
+      "learning_rate": 8.030622009569379e-05,
+      "loss": 0.0624,
+      "step": 1304
+    },
+    {
+      "epoch": 118.64,
+      "learning_rate": 8.028708133971292e-05,
+      "loss": 0.0509,
+      "step": 1305
+    },
+    {
+      "epoch": 118.73,
+      "learning_rate": 8.026794258373206e-05,
+      "loss": 0.0649,
+      "step": 1306
+    },
+    {
+      "epoch": 118.82,
+      "learning_rate": 8.02488038277512e-05,
+      "loss": 0.0546,
+      "step": 1307
+    },
+    {
+      "epoch": 118.91,
+      "learning_rate": 8.022966507177034e-05,
+      "loss": 0.0537,
+      "step": 1308
+    },
+    {
+      "epoch": 119.0,
+      "learning_rate": 8.021052631578949e-05,
+      "loss": 0.0566,
+      "step": 1309
+    },
+    {
+      "epoch": 119.09,
+      "learning_rate": 8.019138755980862e-05,
+      "loss": 0.0535,
+      "step": 1310
+    },
+    {
+      "epoch": 119.18,
+      "learning_rate": 8.017224880382775e-05,
+      "loss": 0.0586,
+      "step": 1311
+    },
+    {
+      "epoch": 119.27,
+      "learning_rate": 8.01531100478469e-05,
+      "loss": 0.0643,
+      "step": 1312
+    },
+    {
+      "epoch": 119.36,
+      "learning_rate": 8.013397129186603e-05,
+      "loss": 0.0517,
+      "step": 1313
+    },
+    {
+      "epoch": 119.45,
+      "learning_rate": 8.011483253588517e-05,
+      "loss": 0.0604,
+      "step": 1314
+    },
+    {
+      "epoch": 119.55,
+      "learning_rate": 8.009569377990432e-05,
+      "loss": 0.0555,
+      "step": 1315
+    },
+    {
+      "epoch": 119.64,
+      "learning_rate": 8.007655502392345e-05,
+      "loss": 0.0546,
+      "step": 1316
+    },
+    {
+      "epoch": 119.73,
+      "learning_rate": 8.005741626794258e-05,
+      "loss": 0.0598,
+      "step": 1317
+    },
+    {
+      "epoch": 119.82,
+      "learning_rate": 8.003827751196173e-05,
+      "loss": 0.0559,
+      "step": 1318
+    },
+    {
+      "epoch": 119.91,
+      "learning_rate": 8.001913875598086e-05,
+      "loss": 0.0465,
+      "step": 1319
+    },
+    {
+      "epoch": 120.0,
+      "learning_rate": 8e-05,
+      "loss": 0.0466,
+      "step": 1320
+    },
+    {
+      "epoch": 120.0,
+      "eval_accuracy_NAT": 0.9926375771708957,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9490334885617409,
+      "eval_accuracy_concrete tiles": 0.9778633889806152,
+      "eval_accuracy_dark bricks": 0.82887494707671,
+      "eval_accuracy_grass": 0.860336855202592,
+      "eval_accuracy_red bricks": 0.9073153409090909,
+      "eval_accuracy_road": 0.9775607113533828,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6184765528078736,
+      "eval_accuracy_speedway bricks": 0.9882371682313031,
+      "eval_accuracy_steel": 0.8696573445023416,
+      "eval_accuracy_tiles": 0.0299958909738392,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9779262994287574,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8820642404266406,
+      "eval_iou_concrete tiles": 0.9411453122241187,
+      "eval_iou_dark bricks": 0.7891458096668987,
+      "eval_iou_grass": 0.805745952563322,
+      "eval_iou_red bricks": 0.7945887109314259,
+      "eval_iou_road": 0.9708247020324837,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.40774263904034896,
+      "eval_iou_speedway bricks": 0.9721565152044885,
+      "eval_iou_steel": 0.8146102993420037,
+      "eval_iou_tiles": 0.0299958909738392,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08918214589357376,
+      "eval_mean_accuracy": 0.692306866597722,
+      "eval_mean_iou": 0.6450727978334098,
+      "eval_overall_accuracy": 0.9761874940660265,
+      "eval_runtime": 3.2662,
+      "eval_samples_per_second": 11.022,
+      "eval_steps_per_second": 0.612,
+      "step": 1320
+    },
+    {
+      "epoch": 120.09,
+      "learning_rate": 7.998086124401914e-05,
+      "loss": 0.0478,
+      "step": 1321
+    },
+    {
+      "epoch": 120.18,
+      "learning_rate": 7.996172248803828e-05,
+      "loss": 0.0523,
+      "step": 1322
+    },
+    {
+      "epoch": 120.27,
+      "learning_rate": 7.994258373205742e-05,
+      "loss": 0.0566,
+      "step": 1323
+    },
+    {
+      "epoch": 120.36,
+      "learning_rate": 7.992344497607656e-05,
+      "loss": 0.0475,
+      "step": 1324
+    },
+    {
+      "epoch": 120.45,
+      "learning_rate": 7.99043062200957e-05,
+      "loss": 0.0669,
+      "step": 1325
+    },
+    {
+      "epoch": 120.55,
+      "learning_rate": 7.988516746411484e-05,
+      "loss": 0.0499,
+      "step": 1326
+    },
+    {
+      "epoch": 120.64,
+      "learning_rate": 7.986602870813397e-05,
+      "loss": 0.0574,
+      "step": 1327
+    },
+    {
+      "epoch": 120.73,
+      "learning_rate": 7.984688995215312e-05,
+      "loss": 0.0507,
+      "step": 1328
+    },
+    {
+      "epoch": 120.82,
+      "learning_rate": 7.982775119617225e-05,
+      "loss": 0.0517,
+      "step": 1329
+    },
+    {
+      "epoch": 120.91,
+      "learning_rate": 7.98086124401914e-05,
+      "loss": 0.064,
+      "step": 1330
+    },
+    {
+      "epoch": 121.0,
+      "learning_rate": 7.978947368421053e-05,
+      "loss": 0.0567,
+      "step": 1331
+    },
+    {
+      "epoch": 121.09,
+      "learning_rate": 7.977033492822967e-05,
+      "loss": 0.0629,
+      "step": 1332
+    },
+    {
+      "epoch": 121.18,
+      "learning_rate": 7.97511961722488e-05,
+      "loss": 0.0545,
+      "step": 1333
+    },
+    {
+      "epoch": 121.27,
+      "learning_rate": 7.973205741626795e-05,
+      "loss": 0.0457,
+      "step": 1334
+    },
+    {
+      "epoch": 121.36,
+      "learning_rate": 7.971291866028708e-05,
+      "loss": 0.0565,
+      "step": 1335
+    },
+    {
+      "epoch": 121.45,
+      "learning_rate": 7.969377990430623e-05,
+      "loss": 0.0571,
+      "step": 1336
+    },
+    {
+      "epoch": 121.55,
+      "learning_rate": 7.967464114832536e-05,
+      "loss": 0.0566,
+      "step": 1337
+    },
+    {
+      "epoch": 121.64,
+      "learning_rate": 7.96555023923445e-05,
+      "loss": 0.0549,
+      "step": 1338
+    },
+    {
+      "epoch": 121.73,
+      "learning_rate": 7.963636363636364e-05,
+      "loss": 0.0538,
+      "step": 1339
+    },
+    {
+      "epoch": 121.82,
+      "learning_rate": 7.961722488038278e-05,
+      "loss": 0.0515,
+      "step": 1340
+    },
+    {
+      "epoch": 121.82,
+      "eval_accuracy_NAT": 0.9915778117661839,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9511741383750375,
+      "eval_accuracy_concrete tiles": 0.9872194704335596,
+      "eval_accuracy_dark bricks": 0.8690966475501328,
+      "eval_accuracy_grass": 0.8571213874044534,
+      "eval_accuracy_red bricks": 0.9002130681818182,
+      "eval_accuracy_road": 0.9829138789364282,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.548617428941637,
+      "eval_accuracy_speedway bricks": 0.9876820330498524,
+      "eval_accuracy_steel": 0.8805157971110085,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9786767639632121,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8783947818892117,
+      "eval_iou_concrete tiles": 0.9319132969665483,
+      "eval_iou_dark bricks": 0.8018750665861714,
+      "eval_iou_grass": 0.7992341285360448,
+      "eval_iou_red bricks": 0.5739839239216574,
+      "eval_iou_road": 0.9724542459342347,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3640687888766923,
+      "eval_iou_speedway bricks": 0.9723647942210817,
+      "eval_iou_steel": 0.8198301936038567,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08886505663394928,
+      "eval_mean_accuracy": 0.6889332047500086,
+      "eval_mean_iou": 0.6225227680383624,
+      "eval_overall_accuracy": 0.9759578704833984,
+      "eval_runtime": 3.4556,
+      "eval_samples_per_second": 10.418,
+      "eval_steps_per_second": 0.579,
+      "step": 1340
+    },
+    {
+      "epoch": 121.91,
+      "learning_rate": 7.959808612440191e-05,
+      "loss": 0.0558,
+      "step": 1341
+    },
+    {
+      "epoch": 122.0,
+      "learning_rate": 7.957894736842106e-05,
+      "loss": 0.0579,
+      "step": 1342
+    },
+    {
+      "epoch": 122.09,
+      "learning_rate": 7.955980861244019e-05,
+      "loss": 0.048,
+      "step": 1343
+    },
+    {
+      "epoch": 122.18,
+      "learning_rate": 7.954066985645934e-05,
+      "loss": 0.0467,
+      "step": 1344
+    },
+    {
+      "epoch": 122.27,
+      "learning_rate": 7.952153110047847e-05,
+      "loss": 0.0599,
+      "step": 1345
+    },
+    {
+      "epoch": 122.36,
+      "learning_rate": 7.950239234449761e-05,
+      "loss": 0.0475,
+      "step": 1346
+    },
+    {
+      "epoch": 122.45,
+      "learning_rate": 7.948325358851675e-05,
+      "loss": 0.0636,
+      "step": 1347
+    },
+    {
+      "epoch": 122.55,
+      "learning_rate": 7.946411483253588e-05,
+      "loss": 0.0563,
+      "step": 1348
+    },
+    {
+      "epoch": 122.64,
+      "learning_rate": 7.944497607655504e-05,
+      "loss": 0.0539,
+      "step": 1349
+    },
+    {
+      "epoch": 122.73,
+      "learning_rate": 7.942583732057417e-05,
+      "loss": 0.0521,
+      "step": 1350
+    },
+    {
+      "epoch": 122.82,
+      "learning_rate": 7.94066985645933e-05,
+      "loss": 0.0526,
+      "step": 1351
+    },
+    {
+      "epoch": 122.91,
+      "learning_rate": 7.938755980861245e-05,
+      "loss": 0.0519,
+      "step": 1352
+    },
+    {
+      "epoch": 123.0,
+      "learning_rate": 7.936842105263158e-05,
+      "loss": 0.0534,
+      "step": 1353
+    },
+    {
+      "epoch": 123.09,
+      "learning_rate": 7.934928229665071e-05,
+      "loss": 0.0536,
+      "step": 1354
+    },
+    {
+      "epoch": 123.18,
+      "learning_rate": 7.933014354066987e-05,
+      "loss": 0.0538,
+      "step": 1355
+    },
+    {
+      "epoch": 123.27,
+      "learning_rate": 7.9311004784689e-05,
+      "loss": 0.0577,
+      "step": 1356
+    },
+    {
+      "epoch": 123.36,
+      "learning_rate": 7.929186602870813e-05,
+      "loss": 0.0533,
+      "step": 1357
+    },
+    {
+      "epoch": 123.45,
+      "learning_rate": 7.927272727272728e-05,
+      "loss": 0.0552,
+      "step": 1358
+    },
+    {
+      "epoch": 123.55,
+      "learning_rate": 7.925358851674641e-05,
+      "loss": 0.0639,
+      "step": 1359
+    },
+    {
+      "epoch": 123.64,
+      "learning_rate": 7.923444976076554e-05,
+      "loss": 0.0646,
+      "step": 1360
+    },
+    {
+      "epoch": 123.64,
+      "eval_accuracy_NAT": 0.9908981220660591,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9483212798832568,
+      "eval_accuracy_concrete tiles": 0.986883140708127,
+      "eval_accuracy_dark bricks": 0.8176359647434663,
+      "eval_accuracy_grass": 0.9133780935771031,
+      "eval_accuracy_red bricks": 0.8758877840909091,
+      "eval_accuracy_road": 0.9790722224170392,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6465415047004659,
+      "eval_accuracy_speedway bricks": 0.9862217861595145,
+      "eval_accuracy_steel": 0.887655827452015,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9783620262969326,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8847109640704027,
+      "eval_iou_concrete tiles": 0.9354838709677419,
+      "eval_iou_dark bricks": 0.7862827108857386,
+      "eval_iou_grass": 0.8193849588321241,
+      "eval_iou_red bricks": 0.834545762138386,
+      "eval_iou_road": 0.971793643878101,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4218061475925826,
+      "eval_iou_speedway bricks": 0.9732722604126471,
+      "eval_iou_steel": 0.8218536019358926,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08792630583047867,
+      "eval_mean_accuracy": 0.6948073635229195,
+      "eval_mean_iou": 0.6482689190008115,
+      "eval_overall_accuracy": 0.9767476187811958,
+      "eval_runtime": 3.9763,
+      "eval_samples_per_second": 9.054,
+      "eval_steps_per_second": 0.503,
+      "step": 1360
+    },
+    {
+      "epoch": 123.73,
+      "learning_rate": 7.92153110047847e-05,
+      "loss": 0.0572,
+      "step": 1361
+    },
+    {
+      "epoch": 123.82,
+      "learning_rate": 7.919617224880384e-05,
+      "loss": 0.0468,
+      "step": 1362
+    },
+    {
+      "epoch": 123.91,
+      "learning_rate": 7.917703349282297e-05,
+      "loss": 0.0469,
+      "step": 1363
+    },
+    {
+      "epoch": 124.0,
+      "learning_rate": 7.915789473684211e-05,
+      "loss": 0.0556,
+      "step": 1364
+    },
+    {
+      "epoch": 124.09,
+      "learning_rate": 7.913875598086124e-05,
+      "loss": 0.0537,
+      "step": 1365
+    },
+    {
+      "epoch": 124.18,
+      "learning_rate": 7.911961722488038e-05,
+      "loss": 0.0503,
+      "step": 1366
+    },
+    {
+      "epoch": 124.27,
+      "learning_rate": 7.910047846889954e-05,
+      "loss": 0.0506,
+      "step": 1367
+    },
+    {
+      "epoch": 124.36,
+      "learning_rate": 7.908133971291867e-05,
+      "loss": 0.054,
+      "step": 1368
+    },
+    {
+      "epoch": 124.45,
+      "learning_rate": 7.90622009569378e-05,
+      "loss": 0.0518,
+      "step": 1369
+    },
+    {
+      "epoch": 124.55,
+      "learning_rate": 7.904306220095695e-05,
+      "loss": 0.0504,
+      "step": 1370
+    },
+    {
+      "epoch": 124.64,
+      "learning_rate": 7.902392344497608e-05,
+      "loss": 0.0539,
+      "step": 1371
+    },
+    {
+      "epoch": 124.73,
+      "learning_rate": 7.900478468899521e-05,
+      "loss": 0.0585,
+      "step": 1372
+    },
+    {
+      "epoch": 124.82,
+      "learning_rate": 7.898564593301437e-05,
+      "loss": 0.0499,
+      "step": 1373
+    },
+    {
+      "epoch": 124.91,
+      "learning_rate": 7.89665071770335e-05,
+      "loss": 0.0543,
+      "step": 1374
+    },
+    {
+      "epoch": 125.0,
+      "learning_rate": 7.894736842105263e-05,
+      "loss": 0.049,
+      "step": 1375
+    },
+    {
+      "epoch": 125.09,
+      "learning_rate": 7.892822966507178e-05,
+      "loss": 0.0565,
+      "step": 1376
+    },
+    {
+      "epoch": 125.18,
+      "learning_rate": 7.890909090909091e-05,
+      "loss": 0.0492,
+      "step": 1377
+    },
+    {
+      "epoch": 125.27,
+      "learning_rate": 7.888995215311004e-05,
+      "loss": 0.0647,
+      "step": 1378
+    },
+    {
+      "epoch": 125.36,
+      "learning_rate": 7.88708133971292e-05,
+      "loss": 0.0491,
+      "step": 1379
+    },
+    {
+      "epoch": 125.45,
+      "learning_rate": 7.885167464114833e-05,
+      "loss": 0.052,
+      "step": 1380
+    },
+    {
+      "epoch": 125.45,
+      "eval_accuracy_NAT": 0.9909690554148259,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9478411090604747,
+      "eval_accuracy_concrete tiles": 0.9907050694062252,
+      "eval_accuracy_dark bricks": 0.8246795735345059,
+      "eval_accuracy_grass": 0.9098305937780699,
+      "eval_accuracy_red bricks": 0.8794389204545454,
+      "eval_accuracy_road": 0.98304689191003,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6542883136217021,
+      "eval_accuracy_speedway bricks": 0.9831088052887488,
+      "eval_accuracy_steel": 0.8939054152100785,
+      "eval_accuracy_tiles": 0.0763594028215313,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.978022955918437,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8868029891389598,
+      "eval_iou_concrete tiles": 0.9183459456395431,
+      "eval_iou_dark bricks": 0.7907732053884481,
+      "eval_iou_grass": 0.8143080580580581,
+      "eval_iou_red bricks": 0.8250874562718641,
+      "eval_iou_road": 0.9734208210353643,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.42807669414332356,
+      "eval_iou_speedway bricks": 0.972496893164245,
+      "eval_iou_steel": 0.8232596282347224,
+      "eval_iou_tiles": 0.0763594028215313,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08597618341445923,
+      "eval_mean_accuracy": 0.702628703884672,
+      "eval_mean_iou": 0.6528426192164998,
+      "eval_overall_accuracy": 0.9767093658447266,
+      "eval_runtime": 3.9369,
+      "eval_samples_per_second": 9.144,
+      "eval_steps_per_second": 0.508,
+      "step": 1380
+    },
+    {
+      "epoch": 125.55,
+      "learning_rate": 7.883253588516747e-05,
+      "loss": 0.0544,
+      "step": 1381
+    },
+    {
+      "epoch": 125.64,
+      "learning_rate": 7.881339712918661e-05,
+      "loss": 0.0526,
+      "step": 1382
+    },
+    {
+      "epoch": 125.73,
+      "learning_rate": 7.879425837320574e-05,
+      "loss": 0.0495,
+      "step": 1383
+    },
+    {
+      "epoch": 125.82,
+      "learning_rate": 7.877511961722487e-05,
+      "loss": 0.048,
+      "step": 1384
+    },
+    {
+      "epoch": 125.91,
+      "learning_rate": 7.875598086124403e-05,
+      "loss": 0.0544,
+      "step": 1385
+    },
+    {
+      "epoch": 126.0,
+      "learning_rate": 7.873684210526317e-05,
+      "loss": 0.0467,
+      "step": 1386
+    },
+    {
+      "epoch": 126.09,
+      "learning_rate": 7.87177033492823e-05,
+      "loss": 0.0505,
+      "step": 1387
+    },
+    {
+      "epoch": 126.18,
+      "learning_rate": 7.869856459330144e-05,
+      "loss": 0.0445,
+      "step": 1388
+    },
+    {
+      "epoch": 126.27,
+      "learning_rate": 7.867942583732058e-05,
+      "loss": 0.0562,
+      "step": 1389
+    },
+    {
+      "epoch": 126.36,
+      "learning_rate": 7.866028708133971e-05,
+      "loss": 0.0555,
+      "step": 1390
+    },
+    {
+      "epoch": 126.45,
+      "learning_rate": 7.864114832535887e-05,
+      "loss": 0.0505,
+      "step": 1391
+    },
+    {
+      "epoch": 126.55,
+      "learning_rate": 7.8622009569378e-05,
+      "loss": 0.0584,
+      "step": 1392
+    },
+    {
+      "epoch": 126.64,
+      "learning_rate": 7.860287081339713e-05,
+      "loss": 0.0551,
+      "step": 1393
+    },
+    {
+      "epoch": 126.73,
+      "learning_rate": 7.858373205741628e-05,
+      "loss": 0.0555,
+      "step": 1394
+    },
+    {
+      "epoch": 126.82,
+      "learning_rate": 7.856459330143541e-05,
+      "loss": 0.0516,
+      "step": 1395
+    },
+    {
+      "epoch": 126.91,
+      "learning_rate": 7.854545454545454e-05,
+      "loss": 0.0519,
+      "step": 1396
+    },
+    {
+      "epoch": 127.0,
+      "learning_rate": 7.852631578947369e-05,
+      "loss": 0.051,
+      "step": 1397
+    },
+    {
+      "epoch": 127.09,
+      "learning_rate": 7.850717703349283e-05,
+      "loss": 0.0536,
+      "step": 1398
+    },
+    {
+      "epoch": 127.18,
+      "learning_rate": 7.848803827751196e-05,
+      "loss": 0.0627,
+      "step": 1399
+    },
+    {
+      "epoch": 127.27,
+      "learning_rate": 7.846889952153111e-05,
+      "loss": 0.0498,
+      "step": 1400
+    },
+    {
+      "epoch": 127.27,
+      "eval_accuracy_NAT": 0.9916681547377477,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9511365831151551,
+      "eval_accuracy_concrete tiles": 0.983733871460894,
+      "eval_accuracy_dark bricks": 0.8251414495208037,
+      "eval_accuracy_grass": 0.8927781292268547,
+      "eval_accuracy_red bricks": 0.8561789772727273,
+      "eval_accuracy_road": 0.9820045538805325,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6401455628153172,
+      "eval_accuracy_speedway bricks": 0.9827847752969672,
+      "eval_accuracy_steel": 0.8927346481102829,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.978557847339994,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8815090408697533,
+      "eval_iou_concrete tiles": 0.9445993952027245,
+      "eval_iou_dark bricks": 0.7955321359655633,
+      "eval_iou_grass": 0.8056614782924099,
+      "eval_iou_red bricks": 0.8106926698049765,
+      "eval_iou_road": 0.9734505318524095,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4227890970666958,
+      "eval_iou_speedway bricks": 0.9725478862457589,
+      "eval_iou_steel": 0.8246045587126549,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08908680081367493,
+      "eval_mean_accuracy": 0.692177438879791,
+      "eval_mean_iou": 0.6469188185656107,
+      "eval_overall_accuracy": 0.976499981350369,
+      "eval_runtime": 3.6222,
+      "eval_samples_per_second": 9.939,
+      "eval_steps_per_second": 0.552,
+      "step": 1400
+    },
+    {
+      "epoch": 127.36,
+      "learning_rate": 7.844976076555024e-05,
+      "loss": 0.0469,
+      "step": 1401
+    },
+    {
+      "epoch": 127.45,
+      "learning_rate": 7.843062200956937e-05,
+      "loss": 0.0542,
+      "step": 1402
+    },
+    {
+      "epoch": 127.55,
+      "learning_rate": 7.841148325358852e-05,
+      "loss": 0.0456,
+      "step": 1403
+    },
+    {
+      "epoch": 127.64,
+      "learning_rate": 7.839234449760766e-05,
+      "loss": 0.056,
+      "step": 1404
+    },
+    {
+      "epoch": 127.73,
+      "learning_rate": 7.83732057416268e-05,
+      "loss": 0.0612,
+      "step": 1405
+    },
+    {
+      "epoch": 127.82,
+      "learning_rate": 7.835406698564594e-05,
+      "loss": 0.0497,
+      "step": 1406
+    },
+    {
+      "epoch": 127.91,
+      "learning_rate": 7.833492822966507e-05,
+      "loss": 0.0594,
+      "step": 1407
+    },
+    {
+      "epoch": 128.0,
+      "learning_rate": 7.83157894736842e-05,
+      "loss": 0.0533,
+      "step": 1408
+    },
+    {
+      "epoch": 128.09,
+      "learning_rate": 7.829665071770335e-05,
+      "loss": 0.0459,
+      "step": 1409
+    },
+    {
+      "epoch": 128.18,
+      "learning_rate": 7.82775119617225e-05,
+      "loss": 0.0517,
+      "step": 1410
+    },
+    {
+      "epoch": 128.27,
+      "learning_rate": 7.825837320574163e-05,
+      "loss": 0.0549,
+      "step": 1411
+    },
+    {
+      "epoch": 128.36,
+      "learning_rate": 7.823923444976077e-05,
+      "loss": 0.0567,
+      "step": 1412
+    },
+    {
+      "epoch": 128.45,
+      "learning_rate": 7.82200956937799e-05,
+      "loss": 0.0566,
+      "step": 1413
+    },
+    {
+      "epoch": 128.55,
+      "learning_rate": 7.820095693779904e-05,
+      "loss": 0.0526,
+      "step": 1414
+    },
+    {
+      "epoch": 128.64,
+      "learning_rate": 7.818181818181818e-05,
+      "loss": 0.0505,
+      "step": 1415
+    },
+    {
+      "epoch": 128.73,
+      "learning_rate": 7.816267942583733e-05,
+      "loss": 0.0484,
+      "step": 1416
+    },
+    {
+      "epoch": 128.82,
+      "learning_rate": 7.814354066985646e-05,
+      "loss": 0.0533,
+      "step": 1417
+    },
+    {
+      "epoch": 128.91,
+      "learning_rate": 7.81244019138756e-05,
+      "loss": 0.0451,
+      "step": 1418
+    },
+    {
+      "epoch": 129.0,
+      "learning_rate": 7.810526315789474e-05,
+      "loss": 0.0487,
+      "step": 1419
+    },
+    {
+      "epoch": 129.09,
+      "learning_rate": 7.808612440191387e-05,
+      "loss": 0.0577,
+      "step": 1420
+    },
+    {
+      "epoch": 129.09,
+      "eval_accuracy_NAT": 0.9914389447376277,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9512800978582772,
+      "eval_accuracy_concrete tiles": 0.982021647404146,
+      "eval_accuracy_dark bricks": 0.820484199992302,
+      "eval_accuracy_grass": 0.8767602064889538,
+      "eval_accuracy_red bricks": 0.8863636363636364,
+      "eval_accuracy_road": 0.9749246360583661,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6378573594684752,
+      "eval_accuracy_speedway bricks": 0.9887802352566354,
+      "eval_accuracy_steel": 0.8645867686828046,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787178228566193,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8720549070333038,
+      "eval_iou_concrete tiles": 0.9455369759773905,
+      "eval_iou_dark bricks": 0.7901623545110831,
+      "eval_iou_grass": 0.8048098146912376,
+      "eval_iou_red bricks": 0.8144884973078805,
+      "eval_iou_road": 0.9683230746134214,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4246957543273555,
+      "eval_iou_speedway bricks": 0.9728362665970085,
+      "eval_iou_steel": 0.8223079577177629,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09080400317907333,
+      "eval_mean_accuracy": 0.6903459794085557,
+      "eval_mean_iou": 0.6456871865871587,
+      "eval_overall_accuracy": 0.9759491814507378,
+      "eval_runtime": 3.639,
+      "eval_samples_per_second": 9.893,
+      "eval_steps_per_second": 0.55,
+      "step": 1420
+    },
+    {
+      "epoch": 129.18,
+      "learning_rate": 7.806698564593302e-05,
+      "loss": 0.0453,
+      "step": 1421
+    },
+    {
+      "epoch": 129.27,
+      "learning_rate": 7.804784688995216e-05,
+      "loss": 0.0536,
+      "step": 1422
+    },
+    {
+      "epoch": 129.36,
+      "learning_rate": 7.80287081339713e-05,
+      "loss": 0.0519,
+      "step": 1423
+    },
+    {
+      "epoch": 129.45,
+      "learning_rate": 7.800956937799043e-05,
+      "loss": 0.0517,
+      "step": 1424
+    },
+    {
+      "epoch": 129.55,
+      "learning_rate": 7.799043062200957e-05,
+      "loss": 0.0591,
+      "step": 1425
+    },
+    {
+      "epoch": 129.64,
+      "learning_rate": 7.797129186602872e-05,
+      "loss": 0.0492,
+      "step": 1426
+    },
+    {
+      "epoch": 129.73,
+      "learning_rate": 7.795215311004785e-05,
+      "loss": 0.0518,
+      "step": 1427
+    },
+    {
+      "epoch": 129.82,
+      "learning_rate": 7.7933014354067e-05,
+      "loss": 0.05,
+      "step": 1428
+    },
+    {
+      "epoch": 129.91,
+      "learning_rate": 7.791387559808613e-05,
+      "loss": 0.0487,
+      "step": 1429
+    },
+    {
+      "epoch": 130.0,
+      "learning_rate": 7.789473684210526e-05,
+      "loss": 0.0483,
+      "step": 1430
+    },
+    {
+      "epoch": 130.09,
+      "learning_rate": 7.78755980861244e-05,
+      "loss": 0.0541,
+      "step": 1431
+    },
+    {
+      "epoch": 130.18,
+      "learning_rate": 7.785645933014355e-05,
+      "loss": 0.0477,
+      "step": 1432
+    },
+    {
+      "epoch": 130.27,
+      "learning_rate": 7.783732057416268e-05,
+      "loss": 0.0492,
+      "step": 1433
+    },
+    {
+      "epoch": 130.36,
+      "learning_rate": 7.781818181818183e-05,
+      "loss": 0.0497,
+      "step": 1434
+    },
+    {
+      "epoch": 130.45,
+      "learning_rate": 7.779904306220096e-05,
+      "loss": 0.0562,
+      "step": 1435
+    },
+    {
+      "epoch": 130.55,
+      "learning_rate": 7.777990430622009e-05,
+      "loss": 0.0468,
+      "step": 1436
+    },
+    {
+      "epoch": 130.64,
+      "learning_rate": 7.776076555023924e-05,
+      "loss": 0.054,
+      "step": 1437
+    },
+    {
+      "epoch": 130.73,
+      "learning_rate": 7.774162679425838e-05,
+      "loss": 0.0442,
+      "step": 1438
+    },
+    {
+      "epoch": 130.82,
+      "learning_rate": 7.772248803827751e-05,
+      "loss": 0.0511,
+      "step": 1439
+    },
+    {
+      "epoch": 130.91,
+      "learning_rate": 7.770334928229666e-05,
+      "loss": 0.0526,
+      "step": 1440
+    },
+    {
+      "epoch": 130.91,
+      "eval_accuracy_NAT": 0.992498886593456,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9479135370616765,
+      "eval_accuracy_concrete tiles": 0.9801565461994741,
+      "eval_accuracy_dark bricks": 0.810746314614526,
+      "eval_accuracy_grass": 0.8873887255912791,
+      "eval_accuracy_red bricks": 0.8957741477272727,
+      "eval_accuracy_road": 0.9721398280746856,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6013012433490476,
+      "eval_accuracy_speedway bricks": 0.9881279514184307,
+      "eval_accuracy_steel": 0.8765088054877647,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9780873812946939,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8797007642709552,
+      "eval_iou_concrete tiles": 0.9403085767922094,
+      "eval_iou_dark bricks": 0.7898901263734204,
+      "eval_iou_grass": 0.8133162488988548,
+      "eval_iou_red bricks": 0.8282712198325398,
+      "eval_iou_road": 0.9675744894925796,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.407499439503774,
+      "eval_iou_speedway bricks": 0.9733669918793904,
+      "eval_iou_steel": 0.8237445469831159,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09054722636938095,
+      "eval_mean_accuracy": 0.6886581527782779,
+      "eval_mean_iou": 0.6462892142555027,
+      "eval_overall_accuracy": 0.976268450419108,
+      "eval_runtime": 3.9147,
+      "eval_samples_per_second": 9.196,
+      "eval_steps_per_second": 0.511,
+      "step": 1440
+    },
+    {
+      "epoch": 131.0,
+      "learning_rate": 7.768421052631579e-05,
+      "loss": 0.0536,
+      "step": 1441
+    },
+    {
+      "epoch": 131.09,
+      "learning_rate": 7.766507177033492e-05,
+      "loss": 0.0477,
+      "step": 1442
+    },
+    {
+      "epoch": 131.18,
+      "learning_rate": 7.764593301435407e-05,
+      "loss": 0.0539,
+      "step": 1443
+    },
+    {
+      "epoch": 131.27,
+      "learning_rate": 7.762679425837321e-05,
+      "loss": 0.0497,
+      "step": 1444
+    },
+    {
+      "epoch": 131.36,
+      "learning_rate": 7.760765550239235e-05,
+      "loss": 0.0523,
+      "step": 1445
+    },
+    {
+      "epoch": 131.45,
+      "learning_rate": 7.758851674641149e-05,
+      "loss": 0.0589,
+      "step": 1446
+    },
+    {
+      "epoch": 131.55,
+      "learning_rate": 7.756937799043062e-05,
+      "loss": 0.0515,
+      "step": 1447
+    },
+    {
+      "epoch": 131.64,
+      "learning_rate": 7.755023923444976e-05,
+      "loss": 0.0505,
+      "step": 1448
+    },
+    {
+      "epoch": 131.73,
+      "learning_rate": 7.75311004784689e-05,
+      "loss": 0.0483,
+      "step": 1449
+    },
+    {
+      "epoch": 131.82,
+      "learning_rate": 7.751196172248805e-05,
+      "loss": 0.0668,
+      "step": 1450
+    },
+    {
+      "epoch": 131.91,
+      "learning_rate": 7.749282296650718e-05,
+      "loss": 0.051,
+      "step": 1451
+    },
+    {
+      "epoch": 132.0,
+      "learning_rate": 7.747368421052633e-05,
+      "loss": 0.047,
+      "step": 1452
+    },
+    {
+      "epoch": 132.09,
+      "learning_rate": 7.745454545454546e-05,
+      "loss": 0.0492,
+      "step": 1453
+    },
+    {
+      "epoch": 132.18,
+      "learning_rate": 7.743540669856459e-05,
+      "loss": 0.054,
+      "step": 1454
+    },
+    {
+      "epoch": 132.27,
+      "learning_rate": 7.741626794258373e-05,
+      "loss": 0.0482,
+      "step": 1455
+    },
+    {
+      "epoch": 132.36,
+      "learning_rate": 7.739712918660288e-05,
+      "loss": 0.0523,
+      "step": 1456
+    },
+    {
+      "epoch": 132.45,
+      "learning_rate": 7.737799043062201e-05,
+      "loss": 0.0482,
+      "step": 1457
+    },
+    {
+      "epoch": 132.55,
+      "learning_rate": 7.735885167464116e-05,
+      "loss": 0.0591,
+      "step": 1458
+    },
+    {
+      "epoch": 132.64,
+      "learning_rate": 7.733971291866029e-05,
+      "loss": 0.0523,
+      "step": 1459
+    },
+    {
+      "epoch": 132.73,
+      "learning_rate": 7.732057416267942e-05,
+      "loss": 0.0521,
+      "step": 1460
+    },
+    {
+      "epoch": 132.73,
+      "eval_accuracy_NAT": 0.9909484106342147,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9439890124039658,
+      "eval_accuracy_concrete tiles": 0.9848957377851159,
+      "eval_accuracy_dark bricks": 0.8513144220776722,
+      "eval_accuracy_grass": 0.8908803042112143,
+      "eval_accuracy_red bricks": 0.9138849431818182,
+      "eval_accuracy_road": 0.9803709127229328,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6766741102197227,
+      "eval_accuracy_speedway bricks": 0.9898193034984378,
+      "eval_accuracy_steel": 0.8783309148473056,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9788612115895479,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8818742936189514,
+      "eval_iou_concrete tiles": 0.9417887319825746,
+      "eval_iou_dark bricks": 0.807167360046712,
+      "eval_iou_grass": 0.8062144962740856,
+      "eval_iou_red bricks": 0.8244433765817716,
+      "eval_iou_road": 0.9726157386024873,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.43470175687163504,
+      "eval_iou_speedway bricks": 0.9722821854184522,
+      "eval_iou_steel": 0.8236699475014884,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08658526837825775,
+      "eval_mean_accuracy": 0.7000852362755692,
+      "eval_mean_iou": 0.6495091614221311,
+      "eval_overall_accuracy": 0.9766035079956055,
+      "eval_runtime": 3.4493,
+      "eval_samples_per_second": 10.437,
+      "eval_steps_per_second": 0.58,
+      "step": 1460
+    },
+    {
+      "epoch": 132.82,
+      "learning_rate": 7.730143540669857e-05,
+      "loss": 0.0468,
+      "step": 1461
+    },
+    {
+      "epoch": 132.91,
+      "learning_rate": 7.728229665071771e-05,
+      "loss": 0.0505,
+      "step": 1462
+    },
+    {
+      "epoch": 133.0,
+      "learning_rate": 7.726315789473684e-05,
+      "loss": 0.052,
+      "step": 1463
+    },
+    {
+      "epoch": 133.09,
+      "learning_rate": 7.724401913875599e-05,
+      "loss": 0.0485,
+      "step": 1464
+    },
+    {
+      "epoch": 133.18,
+      "learning_rate": 7.722488038277512e-05,
+      "loss": 0.053,
+      "step": 1465
+    },
+    {
+      "epoch": 133.27,
+      "learning_rate": 7.720574162679425e-05,
+      "loss": 0.0441,
+      "step": 1466
+    },
+    {
+      "epoch": 133.36,
+      "learning_rate": 7.71866028708134e-05,
+      "loss": 0.0484,
+      "step": 1467
+    },
+    {
+      "epoch": 133.45,
+      "learning_rate": 7.716746411483255e-05,
+      "loss": 0.049,
+      "step": 1468
+    },
+    {
+      "epoch": 133.55,
+      "learning_rate": 7.714832535885168e-05,
+      "loss": 0.0535,
+      "step": 1469
+    },
+    {
+      "epoch": 133.64,
+      "learning_rate": 7.712918660287082e-05,
+      "loss": 0.0534,
+      "step": 1470
+    },
+    {
+      "epoch": 133.73,
+      "learning_rate": 7.711004784688996e-05,
+      "loss": 0.0494,
+      "step": 1471
+    },
+    {
+      "epoch": 133.82,
+      "learning_rate": 7.709090909090909e-05,
+      "loss": 0.0512,
+      "step": 1472
+    },
+    {
+      "epoch": 133.91,
+      "learning_rate": 7.707177033492823e-05,
+      "loss": 0.0519,
+      "step": 1473
+    },
+    {
+      "epoch": 134.0,
+      "learning_rate": 7.705263157894738e-05,
+      "loss": 0.0458,
+      "step": 1474
+    },
+    {
+      "epoch": 134.09,
+      "learning_rate": 7.703349282296651e-05,
+      "loss": 0.0502,
+      "step": 1475
+    },
+    {
+      "epoch": 134.18,
+      "learning_rate": 7.701435406698566e-05,
+      "loss": 0.0517,
+      "step": 1476
+    },
+    {
+      "epoch": 134.27,
+      "learning_rate": 7.699521531100479e-05,
+      "loss": 0.0482,
+      "step": 1477
+    },
+    {
+      "epoch": 134.36,
+      "learning_rate": 7.697607655502392e-05,
+      "loss": 0.0561,
+      "step": 1478
+    },
+    {
+      "epoch": 134.45,
+      "learning_rate": 7.695693779904307e-05,
+      "loss": 0.0512,
+      "step": 1479
+    },
+    {
+      "epoch": 134.55,
+      "learning_rate": 7.693779904306221e-05,
+      "loss": 0.0534,
+      "step": 1480
+    },
+    {
+      "epoch": 134.55,
+      "eval_accuracy_NAT": 0.9930360037915816,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.944281406927336,
+      "eval_accuracy_concrete tiles": 0.9792392833119306,
+      "eval_accuracy_dark bricks": 0.8256418151726261,
+      "eval_accuracy_grass": 0.8636082441798285,
+      "eval_accuracy_red bricks": 0.9071377840909091,
+      "eval_accuracy_road": 0.9828788118797515,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6690927135886197,
+      "eval_accuracy_speedway bricks": 0.9870840559141809,
+      "eval_accuracy_steel": 0.898522524899413,
+      "eval_accuracy_tiles": 0.039720586221065605,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9783642312265745,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8882241471975292,
+      "eval_iou_concrete tiles": 0.948526580778913,
+      "eval_iou_dark bricks": 0.7979689011234283,
+      "eval_iou_grass": 0.808648298386912,
+      "eval_iou_red bricks": 0.8272344559585493,
+      "eval_iou_road": 0.9742903136488241,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4315817551347026,
+      "eval_iou_speedway bricks": 0.9727457390874137,
+      "eval_iou_steel": 0.8208674168813366,
+      "eval_iou_tiles": 0.039720586221065605,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08533377200365067,
+      "eval_mean_accuracy": 0.6992494792290186,
+      "eval_mean_iou": 0.65293634043425,
+      "eval_overall_accuracy": 0.9769903818766276,
+      "eval_runtime": 3.4755,
+      "eval_samples_per_second": 10.358,
+      "eval_steps_per_second": 0.575,
+      "step": 1480
+    },
+    {
+      "epoch": 134.64,
+      "learning_rate": 7.691866028708134e-05,
+      "loss": 0.0466,
+      "step": 1481
+    },
+    {
+      "epoch": 134.73,
+      "learning_rate": 7.689952153110049e-05,
+      "loss": 0.0505,
+      "step": 1482
+    },
+    {
+      "epoch": 134.82,
+      "learning_rate": 7.688038277511962e-05,
+      "loss": 0.0602,
+      "step": 1483
+    },
+    {
+      "epoch": 134.91,
+      "learning_rate": 7.686124401913875e-05,
+      "loss": 0.0542,
+      "step": 1484
+    },
+    {
+      "epoch": 135.0,
+      "learning_rate": 7.68421052631579e-05,
+      "loss": 0.0481,
+      "step": 1485
+    },
+    {
+      "epoch": 135.09,
+      "learning_rate": 7.682296650717704e-05,
+      "loss": 0.0467,
+      "step": 1486
+    },
+    {
+      "epoch": 135.18,
+      "learning_rate": 7.680382775119618e-05,
+      "loss": 0.0515,
+      "step": 1487
+    },
+    {
+      "epoch": 135.27,
+      "learning_rate": 7.678468899521532e-05,
+      "loss": 0.0456,
+      "step": 1488
+    },
+    {
+      "epoch": 135.36,
+      "learning_rate": 7.676555023923445e-05,
+      "loss": 0.0551,
+      "step": 1489
+    },
+    {
+      "epoch": 135.45,
+      "learning_rate": 7.674641148325359e-05,
+      "loss": 0.0492,
+      "step": 1490
+    },
+    {
+      "epoch": 135.55,
+      "learning_rate": 7.672727272727273e-05,
+      "loss": 0.0525,
+      "step": 1491
+    },
+    {
+      "epoch": 135.64,
+      "learning_rate": 7.670813397129188e-05,
+      "loss": 0.045,
+      "step": 1492
+    },
+    {
+      "epoch": 135.73,
+      "learning_rate": 7.668899521531101e-05,
+      "loss": 0.0531,
+      "step": 1493
+    },
+    {
+      "epoch": 135.82,
+      "learning_rate": 7.666985645933015e-05,
+      "loss": 0.0517,
+      "step": 1494
+    },
+    {
+      "epoch": 135.91,
+      "learning_rate": 7.665071770334929e-05,
+      "loss": 0.0488,
+      "step": 1495
+    },
+    {
+      "epoch": 136.0,
+      "learning_rate": 7.663157894736842e-05,
+      "loss": 0.0471,
+      "step": 1496
+    },
+    {
+      "epoch": 136.09,
+      "learning_rate": 7.661244019138756e-05,
+      "loss": 0.0472,
+      "step": 1497
+    },
+    {
+      "epoch": 136.18,
+      "learning_rate": 7.659330143540671e-05,
+      "loss": 0.0483,
+      "step": 1498
+    },
+    {
+      "epoch": 136.27,
+      "learning_rate": 7.657416267942584e-05,
+      "loss": 0.0479,
+      "step": 1499
+    },
+    {
+      "epoch": 136.36,
+      "learning_rate": 7.655502392344497e-05,
+      "loss": 0.0454,
+      "step": 1500
+    },
+    {
+      "epoch": 136.36,
+      "eval_accuracy_NAT": 0.9927540349076772,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9514115412678655,
+      "eval_accuracy_concrete tiles": 0.9881673087506879,
+      "eval_accuracy_dark bricks": 0.8172895577537431,
+      "eval_accuracy_grass": 0.8571283775518407,
+      "eval_accuracy_red bricks": 0.9174360795454546,
+      "eval_accuracy_road": 0.9805704321833354,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6427370220274033,
+      "eval_accuracy_speedway bricks": 0.9840905497889882,
+      "eval_accuracy_steel": 0.9012433216806279,
+      "eval_accuracy_tiles": 0.013765237638679633,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.97815852830808,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8802079714101355,
+      "eval_iou_concrete tiles": 0.9136371346186465,
+      "eval_iou_dark bricks": 0.7975810389512827,
+      "eval_iou_grass": 0.8042046788611754,
+      "eval_iou_red bricks": 0.7780454750790543,
+      "eval_iou_road": 0.9712023512390954,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4287632183908046,
+      "eval_iou_speedway bricks": 0.9735098784312461,
+      "eval_iou_steel": 0.8268720167628615,
+      "eval_iou_tiles": 0.013765237638679633,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08769172430038452,
+      "eval_mean_accuracy": 0.6958918048535618,
+      "eval_mean_iou": 0.643534425360851,
+      "eval_overall_accuracy": 0.9763675265842013,
+      "eval_runtime": 3.523,
+      "eval_samples_per_second": 10.218,
+      "eval_steps_per_second": 0.568,
+      "step": 1500
+    },
+    {
+      "epoch": 136.45,
+      "learning_rate": 7.653588516746412e-05,
+      "loss": 0.0556,
+      "step": 1501
+    },
+    {
+      "epoch": 136.55,
+      "learning_rate": 7.651674641148325e-05,
+      "loss": 0.0463,
+      "step": 1502
+    },
+    {
+      "epoch": 136.64,
+      "learning_rate": 7.64976076555024e-05,
+      "loss": 0.062,
+      "step": 1503
+    },
+    {
+      "epoch": 136.73,
+      "learning_rate": 7.647846889952154e-05,
+      "loss": 0.0561,
+      "step": 1504
+    },
+    {
+      "epoch": 136.82,
+      "learning_rate": 7.645933014354067e-05,
+      "loss": 0.0592,
+      "step": 1505
+    },
+    {
+      "epoch": 136.91,
+      "learning_rate": 7.64401913875598e-05,
+      "loss": 0.0539,
+      "step": 1506
+    },
+    {
+      "epoch": 137.0,
+      "learning_rate": 7.642105263157895e-05,
+      "loss": 0.0448,
+      "step": 1507
+    },
+    {
+      "epoch": 137.09,
+      "learning_rate": 7.640191387559808e-05,
+      "loss": 0.0477,
+      "step": 1508
+    },
+    {
+      "epoch": 137.18,
+      "learning_rate": 7.638277511961723e-05,
+      "loss": 0.0453,
+      "step": 1509
+    },
+    {
+      "epoch": 137.27,
+      "learning_rate": 7.636363636363637e-05,
+      "loss": 0.0496,
+      "step": 1510
+    },
+    {
+      "epoch": 137.36,
+      "learning_rate": 7.63444976076555e-05,
+      "loss": 0.0523,
+      "step": 1511
+    },
+    {
+      "epoch": 137.45,
+      "learning_rate": 7.632535885167464e-05,
+      "loss": 0.0465,
+      "step": 1512
+    },
+    {
+      "epoch": 137.55,
+      "learning_rate": 7.630622009569378e-05,
+      "loss": 0.045,
+      "step": 1513
+    },
+    {
+      "epoch": 137.64,
+      "learning_rate": 7.628708133971292e-05,
+      "loss": 0.0531,
+      "step": 1514
+    },
+    {
+      "epoch": 137.73,
+      "learning_rate": 7.626794258373206e-05,
+      "loss": 0.0459,
+      "step": 1515
+    },
+    {
+      "epoch": 137.82,
+      "learning_rate": 7.624880382775121e-05,
+      "loss": 0.0571,
+      "step": 1516
+    },
+    {
+      "epoch": 137.91,
+      "learning_rate": 7.622966507177034e-05,
+      "loss": 0.0455,
+      "step": 1517
+    },
+    {
+      "epoch": 138.0,
+      "learning_rate": 7.621052631578947e-05,
+      "loss": 0.0553,
+      "step": 1518
+    },
+    {
+      "epoch": 138.09,
+      "learning_rate": 7.619138755980862e-05,
+      "loss": 0.0552,
+      "step": 1519
+    },
+    {
+      "epoch": 138.18,
+      "learning_rate": 7.617224880382775e-05,
+      "loss": 0.0454,
+      "step": 1520
+    },
+    {
+      "epoch": 138.18,
+      "eval_accuracy_NAT": 0.9916466277015548,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9438294025494657,
+      "eval_accuracy_concrete tiles": 0.9838561731792331,
+      "eval_accuracy_dark bricks": 0.8507755667603248,
+      "eval_accuracy_grass": 0.8860011813349085,
+      "eval_accuracy_red bricks": 0.9117542613636364,
+      "eval_accuracy_road": 0.9769694082252804,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6518898354147713,
+      "eval_accuracy_speedway bricks": 0.9886233492270949,
+      "eval_accuracy_steel": 0.8817525229206516,
+      "eval_accuracy_tiles": 0.08012601013559786,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787244009962874,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8800677851635536,
+      "eval_iou_concrete tiles": 0.9382161705105403,
+      "eval_iou_dark bricks": 0.7961388848869039,
+      "eval_iou_grass": 0.8045958770412455,
+      "eval_iou_red bricks": 0.8148206918438591,
+      "eval_iou_road": 0.9679893225856308,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4281678919349582,
+      "eval_iou_speedway bricks": 0.9734169942779669,
+      "eval_iou_steel": 0.8260738589403922,
+      "eval_iou_tiles": 0.08012601013559786,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08710777759552002,
+      "eval_mean_accuracy": 0.703632641447117,
+      "eval_mean_iou": 0.6529490683320719,
+      "eval_overall_accuracy": 0.9764205084906684,
+      "eval_runtime": 3.7378,
+      "eval_samples_per_second": 9.631,
+      "eval_steps_per_second": 0.535,
+      "step": 1520
+    },
+    {
+      "epoch": 138.27,
+      "learning_rate": 7.61531100478469e-05,
+      "loss": 0.0511,
+      "step": 1521
+    },
+    {
+      "epoch": 138.36,
+      "learning_rate": 7.613397129186604e-05,
+      "loss": 0.0566,
+      "step": 1522
+    },
+    {
+      "epoch": 138.45,
+      "learning_rate": 7.611483253588517e-05,
+      "loss": 0.0479,
+      "step": 1523
+    },
+    {
+      "epoch": 138.55,
+      "learning_rate": 7.60956937799043e-05,
+      "loss": 0.0435,
+      "step": 1524
+    },
+    {
+      "epoch": 138.64,
+      "learning_rate": 7.607655502392345e-05,
+      "loss": 0.0428,
+      "step": 1525
+    },
+    {
+      "epoch": 138.73,
+      "learning_rate": 7.605741626794258e-05,
+      "loss": 0.0459,
+      "step": 1526
+    },
+    {
+      "epoch": 138.82,
+      "learning_rate": 7.603827751196173e-05,
+      "loss": 0.0519,
+      "step": 1527
+    },
+    {
+      "epoch": 138.91,
+      "learning_rate": 7.601913875598087e-05,
+      "loss": 0.051,
+      "step": 1528
+    },
+    {
+      "epoch": 139.0,
+      "learning_rate": 7.6e-05,
+      "loss": 0.0468,
+      "step": 1529
+    },
+    {
+      "epoch": 139.09,
+      "learning_rate": 7.598086124401914e-05,
+      "loss": 0.0549,
+      "step": 1530
+    },
+    {
+      "epoch": 139.18,
+      "learning_rate": 7.596172248803828e-05,
+      "loss": 0.0497,
+      "step": 1531
+    },
+    {
+      "epoch": 139.27,
+      "learning_rate": 7.594258373205741e-05,
+      "loss": 0.0423,
+      "step": 1532
+    },
+    {
+      "epoch": 139.36,
+      "learning_rate": 7.592344497607656e-05,
+      "loss": 0.0709,
+      "step": 1533
+    },
+    {
+      "epoch": 139.45,
+      "learning_rate": 7.59043062200957e-05,
+      "loss": 0.0586,
+      "step": 1534
+    },
+    {
+      "epoch": 139.55,
+      "learning_rate": 7.588516746411484e-05,
+      "loss": 0.0446,
+      "step": 1535
+    },
+    {
+      "epoch": 139.64,
+      "learning_rate": 7.586602870813397e-05,
+      "loss": 0.0505,
+      "step": 1536
+    },
+    {
+      "epoch": 139.73,
+      "learning_rate": 7.584688995215311e-05,
+      "loss": 0.051,
+      "step": 1537
+    },
+    {
+      "epoch": 139.82,
+      "learning_rate": 7.582775119617225e-05,
+      "loss": 0.0446,
+      "step": 1538
+    },
+    {
+      "epoch": 139.91,
+      "learning_rate": 7.580861244019139e-05,
+      "loss": 0.0516,
+      "step": 1539
+    },
+    {
+      "epoch": 140.0,
+      "learning_rate": 7.578947368421054e-05,
+      "loss": 0.0472,
+      "step": 1540
+    },
+    {
+      "epoch": 140.0,
+      "eval_accuracy_NAT": 0.9926986292571479,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9461658762178634,
+      "eval_accuracy_concrete tiles": 0.9852932183697181,
+      "eval_accuracy_dark bricks": 0.8223317039374928,
+      "eval_accuracy_grass": 0.8785671595885599,
+      "eval_accuracy_red bricks": 0.8872514204545454,
+      "eval_accuracy_road": 0.9755340773192324,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6278499159154192,
+      "eval_accuracy_speedway bricks": 0.9887567023522044,
+      "eval_accuracy_steel": 0.8633912670668161,
+      "eval_accuracy_tiles": 0.08533077660594439,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9788607353330866,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8797192105571643,
+      "eval_iou_concrete tiles": 0.9311161836516513,
+      "eval_iou_dark bricks": 0.7982141522827467,
+      "eval_iou_grass": 0.8114093699765653,
+      "eval_iou_red bricks": 0.8036346091990993,
+      "eval_iou_road": 0.967234675685753,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4150537634408602,
+      "eval_iou_speedway bricks": 0.9733979326400155,
+      "eval_iou_steel": 0.8192372384118912,
+      "eval_iou_tiles": 0.08532493323289735,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0869722068309784,
+      "eval_mean_accuracy": 0.696397749775765,
+      "eval_mean_iou": 0.6510156003393639,
+      "eval_overall_accuracy": 0.9765008290608724,
+      "eval_runtime": 3.4021,
+      "eval_samples_per_second": 10.582,
+      "eval_steps_per_second": 0.588,
+      "step": 1540
+    },
+    {
+      "epoch": 140.09,
+      "learning_rate": 7.577033492822967e-05,
+      "loss": 0.0467,
+      "step": 1541
+    },
+    {
+      "epoch": 140.18,
+      "learning_rate": 7.57511961722488e-05,
+      "loss": 0.0449,
+      "step": 1542
+    },
+    {
+      "epoch": 140.27,
+      "learning_rate": 7.573205741626795e-05,
+      "loss": 0.0512,
+      "step": 1543
+    },
+    {
+      "epoch": 140.36,
+      "learning_rate": 7.571291866028708e-05,
+      "loss": 0.0462,
+      "step": 1544
+    },
+    {
+      "epoch": 140.45,
+      "learning_rate": 7.569377990430622e-05,
+      "loss": 0.0555,
+      "step": 1545
+    },
+    {
+      "epoch": 140.55,
+      "learning_rate": 7.567464114832537e-05,
+      "loss": 0.042,
+      "step": 1546
+    },
+    {
+      "epoch": 140.64,
+      "learning_rate": 7.56555023923445e-05,
+      "loss": 0.0506,
+      "step": 1547
+    },
+    {
+      "epoch": 140.73,
+      "learning_rate": 7.563636363636363e-05,
+      "loss": 0.0488,
+      "step": 1548
+    },
+    {
+      "epoch": 140.82,
+      "learning_rate": 7.561722488038278e-05,
+      "loss": 0.0514,
+      "step": 1549
+    },
+    {
+      "epoch": 140.91,
+      "learning_rate": 7.559808612440191e-05,
+      "loss": 0.0499,
+      "step": 1550
+    },
+    {
+      "epoch": 141.0,
+      "learning_rate": 7.557894736842106e-05,
+      "loss": 0.0458,
+      "step": 1551
+    },
+    {
+      "epoch": 141.09,
+      "learning_rate": 7.55598086124402e-05,
+      "loss": 0.0527,
+      "step": 1552
+    },
+    {
+      "epoch": 141.18,
+      "learning_rate": 7.554066985645933e-05,
+      "loss": 0.0486,
+      "step": 1553
+    },
+    {
+      "epoch": 141.27,
+      "learning_rate": 7.552153110047847e-05,
+      "loss": 0.0452,
+      "step": 1554
+    },
+    {
+      "epoch": 141.36,
+      "learning_rate": 7.550239234449761e-05,
+      "loss": 0.0494,
+      "step": 1555
+    },
+    {
+      "epoch": 141.45,
+      "learning_rate": 7.548325358851674e-05,
+      "loss": 0.0432,
+      "step": 1556
+    },
+    {
+      "epoch": 141.55,
+      "learning_rate": 7.546411483253589e-05,
+      "loss": 0.0496,
+      "step": 1557
+    },
+    {
+      "epoch": 141.64,
+      "learning_rate": 7.544497607655504e-05,
+      "loss": 0.0397,
+      "step": 1558
+    },
+    {
+      "epoch": 141.73,
+      "learning_rate": 7.542583732057417e-05,
+      "loss": 0.0542,
+      "step": 1559
+    },
+    {
+      "epoch": 141.82,
+      "learning_rate": 7.54066985645933e-05,
+      "loss": 0.046,
+      "step": 1560
+    },
+    {
+      "epoch": 141.82,
+      "eval_accuracy_NAT": 0.9913417201725269,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9441311858878063,
+      "eval_accuracy_concrete tiles": 0.9889928453494772,
+      "eval_accuracy_dark bricks": 0.8468496208767946,
+      "eval_accuracy_grass": 0.8955671980343706,
+      "eval_accuracy_red bricks": 0.9378551136363636,
+      "eval_accuracy_road": 0.9694819870203522,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6312960052931933,
+      "eval_accuracy_speedway bricks": 0.9895513904326072,
+      "eval_accuracy_steel": 0.8826512103423257,
+      "eval_accuracy_tiles": 0.0028078345432132586,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787357284985452,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8752427727696612,
+      "eval_iou_concrete tiles": 0.9153060358244433,
+      "eval_iou_dark bricks": 0.7999854561320583,
+      "eval_iou_grass": 0.8130171844857346,
+      "eval_iou_red bricks": 0.7933313307299489,
+      "eval_iou_road": 0.9628606153188403,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4166029909398537,
+      "eval_iou_speedway bricks": 0.9727276016553602,
+      "eval_iou_steel": 0.8245465398390265,
+      "eval_iou_tiles": 0.0028078345432132586,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0897364690899849,
+      "eval_mean_accuracy": 0.6985020085837715,
+      "eval_mean_iou": 0.6427049300566682,
+      "eval_overall_accuracy": 0.9758934444851346,
+      "eval_runtime": 3.4999,
+      "eval_samples_per_second": 10.286,
+      "eval_steps_per_second": 0.571,
+      "step": 1560
+    },
+    {
+      "epoch": 141.91,
+      "learning_rate": 7.538755980861245e-05,
+      "loss": 0.0584,
+      "step": 1561
+    },
+    {
+      "epoch": 142.0,
+      "learning_rate": 7.536842105263158e-05,
+      "loss": 0.0472,
+      "step": 1562
+    },
+    {
+      "epoch": 142.09,
+      "learning_rate": 7.534928229665072e-05,
+      "loss": 0.0481,
+      "step": 1563
+    },
+    {
+      "epoch": 142.18,
+      "learning_rate": 7.533014354066987e-05,
+      "loss": 0.0489,
+      "step": 1564
+    },
+    {
+      "epoch": 142.27,
+      "learning_rate": 7.5311004784689e-05,
+      "loss": 0.0475,
+      "step": 1565
+    },
+    {
+      "epoch": 142.36,
+      "learning_rate": 7.529186602870813e-05,
+      "loss": 0.0454,
+      "step": 1566
+    },
+    {
+      "epoch": 142.45,
+      "learning_rate": 7.527272727272728e-05,
+      "loss": 0.0467,
+      "step": 1567
+    },
+    {
+      "epoch": 142.55,
+      "learning_rate": 7.525358851674642e-05,
+      "loss": 0.0486,
+      "step": 1568
+    },
+    {
+      "epoch": 142.64,
+      "learning_rate": 7.523444976076556e-05,
+      "loss": 0.0495,
+      "step": 1569
+    },
+    {
+      "epoch": 142.73,
+      "learning_rate": 7.52153110047847e-05,
+      "loss": 0.0448,
+      "step": 1570
+    },
+    {
+      "epoch": 142.82,
+      "learning_rate": 7.519617224880383e-05,
+      "loss": 0.049,
+      "step": 1571
+    },
+    {
+      "epoch": 142.91,
+      "learning_rate": 7.517703349282296e-05,
+      "loss": 0.0523,
+      "step": 1572
+    },
+    {
+      "epoch": 143.0,
+      "learning_rate": 7.515789473684211e-05,
+      "loss": 0.0454,
+      "step": 1573
+    },
+    {
+      "epoch": 143.09,
+      "learning_rate": 7.513875598086126e-05,
+      "loss": 0.0446,
+      "step": 1574
+    },
+    {
+      "epoch": 143.18,
+      "learning_rate": 7.511961722488039e-05,
+      "loss": 0.0509,
+      "step": 1575
+    },
+    {
+      "epoch": 143.27,
+      "learning_rate": 7.510047846889952e-05,
+      "loss": 0.0444,
+      "step": 1576
+    },
+    {
+      "epoch": 143.36,
+      "learning_rate": 7.508133971291867e-05,
+      "loss": 0.0446,
+      "step": 1577
+    },
+    {
+      "epoch": 143.45,
+      "learning_rate": 7.50622009569378e-05,
+      "loss": 0.0459,
+      "step": 1578
+    },
+    {
+      "epoch": 143.55,
+      "learning_rate": 7.504306220095694e-05,
+      "loss": 0.0557,
+      "step": 1579
+    },
+    {
+      "epoch": 143.64,
+      "learning_rate": 7.502392344497609e-05,
+      "loss": 0.054,
+      "step": 1580
+    },
+    {
+      "epoch": 143.64,
+      "eval_accuracy_NAT": 0.9919349488256473,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9473971522382935,
+      "eval_accuracy_concrete tiles": 0.9763651929309607,
+      "eval_accuracy_dark bricks": 0.8168276817674455,
+      "eval_accuracy_grass": 0.9026377321165817,
+      "eval_accuracy_red bricks": 0.8948863636363636,
+      "eval_accuracy_road": 0.9786635098254265,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6302759628373722,
+      "eval_accuracy_speedway bricks": 0.9873290394833864,
+      "eval_accuracy_steel": 0.8902282171360728,
+      "eval_accuracy_tiles": 0.022942062731132723,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9792204258651726,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8839384151782083,
+      "eval_iou_concrete tiles": 0.9529678593810618,
+      "eval_iou_dark bricks": 0.7928419322299847,
+      "eval_iou_grass": 0.8160337712729318,
+      "eval_iou_red bricks": 0.8236639973851937,
+      "eval_iou_road": 0.970983578236124,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4205279131794353,
+      "eval_iou_speedway bricks": 0.9740534967297501,
+      "eval_iou_steel": 0.8271144373884467,
+      "eval_iou_tiles": 0.022942062731132723,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08511282503604889,
+      "eval_mean_accuracy": 0.6953452202714371,
+      "eval_mean_iou": 0.651099068429034,
+      "eval_overall_accuracy": 0.9771089553833008,
+      "eval_runtime": 4.1356,
+      "eval_samples_per_second": 8.705,
+      "eval_steps_per_second": 0.484,
+      "step": 1580
+    },
+    {
+      "epoch": 143.73,
+      "learning_rate": 7.500478468899522e-05,
+      "loss": 0.0523,
+      "step": 1581
+    },
+    {
+      "epoch": 143.82,
+      "learning_rate": 7.498564593301435e-05,
+      "loss": 0.0421,
+      "step": 1582
+    },
+    {
+      "epoch": 143.91,
+      "learning_rate": 7.49665071770335e-05,
+      "loss": 0.054,
+      "step": 1583
+    },
+    {
+      "epoch": 144.0,
+      "learning_rate": 7.494736842105263e-05,
+      "loss": 0.0444,
+      "step": 1584
+    },
+    {
+      "epoch": 144.09,
+      "learning_rate": 7.492822966507178e-05,
+      "loss": 0.0431,
+      "step": 1585
+    },
+    {
+      "epoch": 144.18,
+      "learning_rate": 7.490909090909092e-05,
+      "loss": 0.0507,
+      "step": 1586
+    },
+    {
+      "epoch": 144.27,
+      "learning_rate": 7.488995215311005e-05,
+      "loss": 0.045,
+      "step": 1587
+    },
+    {
+      "epoch": 144.36,
+      "learning_rate": 7.487081339712919e-05,
+      "loss": 0.0542,
+      "step": 1588
+    },
+    {
+      "epoch": 144.45,
+      "learning_rate": 7.485167464114833e-05,
+      "loss": 0.0466,
+      "step": 1589
+    },
+    {
+      "epoch": 144.55,
+      "learning_rate": 7.483253588516746e-05,
+      "loss": 0.0485,
+      "step": 1590
+    },
+    {
+      "epoch": 144.64,
+      "learning_rate": 7.481339712918661e-05,
+      "loss": 0.0542,
+      "step": 1591
+    },
+    {
+      "epoch": 144.73,
+      "learning_rate": 7.479425837320575e-05,
+      "loss": 0.0447,
+      "step": 1592
+    },
+    {
+      "epoch": 144.82,
+      "learning_rate": 7.477511961722489e-05,
+      "loss": 0.0532,
+      "step": 1593
+    },
+    {
+      "epoch": 144.91,
+      "learning_rate": 7.475598086124402e-05,
+      "loss": 0.0462,
+      "step": 1594
+    },
+    {
+      "epoch": 145.0,
+      "learning_rate": 7.473684210526316e-05,
+      "loss": 0.0408,
+      "step": 1595
+    },
+    {
+      "epoch": 145.09,
+      "learning_rate": 7.47177033492823e-05,
+      "loss": 0.0468,
+      "step": 1596
+    },
+    {
+      "epoch": 145.18,
+      "learning_rate": 7.469856459330144e-05,
+      "loss": 0.0533,
+      "step": 1597
+    },
+    {
+      "epoch": 145.27,
+      "learning_rate": 7.467942583732059e-05,
+      "loss": 0.0455,
+      "step": 1598
+    },
+    {
+      "epoch": 145.36,
+      "learning_rate": 7.466028708133972e-05,
+      "loss": 0.0455,
+      "step": 1599
+    },
+    {
+      "epoch": 145.45,
+      "learning_rate": 7.464114832535885e-05,
+      "loss": 0.0538,
+      "step": 1600
+    },
+    {
+      "epoch": 145.45,
+      "eval_accuracy_NAT": 0.9931623427908779,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9493352719000816,
+      "eval_accuracy_concrete tiles": 0.9832446645875375,
+      "eval_accuracy_dark bricks": 0.8404218467341519,
+      "eval_accuracy_grass": 0.8807935215314014,
+      "eval_accuracy_red bricks": 0.9026988636363636,
+      "eval_accuracy_road": 0.974312776379798,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6037548589860227,
+      "eval_accuracy_speedway bricks": 0.9874201540620811,
+      "eval_accuracy_steel": 0.8810764461447134,
+      "eval_accuracy_tiles": 0.07485275989590467,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.978936453338282,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8816061859933287,
+      "eval_iou_concrete tiles": 0.941035320282094,
+      "eval_iou_dark bricks": 0.8105650011136685,
+      "eval_iou_grass": 0.8225997036147252,
+      "eval_iou_red bricks": 0.8138306387065791,
+      "eval_iou_road": 0.9670788659738133,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.41114406938760184,
+      "eval_iou_speedway bricks": 0.9736437669237218,
+      "eval_iou_steel": 0.8252046332046332,
+      "eval_iou_tiles": 0.07485275989590467,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08630597591400146,
+      "eval_mean_accuracy": 0.6977748851268409,
+      "eval_mean_iou": 0.6538844152641808,
+      "eval_overall_accuracy": 0.9769258499145508,
+      "eval_runtime": 4.0474,
+      "eval_samples_per_second": 8.895,
+      "eval_steps_per_second": 0.494,
+      "step": 1600
+    },
+    {
+      "epoch": 145.55,
+      "learning_rate": 7.4622009569378e-05,
+      "loss": 0.0426,
+      "step": 1601
+    },
+    {
+      "epoch": 145.64,
+      "learning_rate": 7.460287081339713e-05,
+      "loss": 0.0456,
+      "step": 1602
+    },
+    {
+      "epoch": 145.73,
+      "learning_rate": 7.458373205741626e-05,
+      "loss": 0.0502,
+      "step": 1603
+    },
+    {
+      "epoch": 145.82,
+      "learning_rate": 7.456459330143542e-05,
+      "loss": 0.0448,
+      "step": 1604
+    },
+    {
+      "epoch": 145.91,
+      "learning_rate": 7.454545454545455e-05,
+      "loss": 0.0479,
+      "step": 1605
+    },
+    {
+      "epoch": 146.0,
+      "learning_rate": 7.452631578947368e-05,
+      "loss": 0.0456,
+      "step": 1606
+    },
+    {
+      "epoch": 146.09,
+      "learning_rate": 7.450717703349283e-05,
+      "loss": 0.0463,
+      "step": 1607
+    },
+    {
+      "epoch": 146.18,
+      "learning_rate": 7.448803827751196e-05,
+      "loss": 0.0502,
+      "step": 1608
+    },
+    {
+      "epoch": 146.27,
+      "learning_rate": 7.446889952153109e-05,
+      "loss": 0.0524,
+      "step": 1609
+    },
+    {
+      "epoch": 146.36,
+      "learning_rate": 7.444976076555025e-05,
+      "loss": 0.0544,
+      "step": 1610
+    },
+    {
+      "epoch": 146.45,
+      "learning_rate": 7.443062200956938e-05,
+      "loss": 0.0464,
+      "step": 1611
+    },
+    {
+      "epoch": 146.55,
+      "learning_rate": 7.441148325358852e-05,
+      "loss": 0.0565,
+      "step": 1612
+    },
+    {
+      "epoch": 146.64,
+      "learning_rate": 7.439234449760766e-05,
+      "loss": 0.0464,
+      "step": 1613
+    },
+    {
+      "epoch": 146.73,
+      "learning_rate": 7.43732057416268e-05,
+      "loss": 0.0445,
+      "step": 1614
+    },
+    {
+      "epoch": 146.82,
+      "learning_rate": 7.435406698564593e-05,
+      "loss": 0.0423,
+      "step": 1615
+    },
+    {
+      "epoch": 146.91,
+      "learning_rate": 7.433492822966508e-05,
+      "loss": 0.0479,
+      "step": 1616
+    },
+    {
+      "epoch": 147.0,
+      "learning_rate": 7.431578947368422e-05,
+      "loss": 0.0493,
+      "step": 1617
+    },
+    {
+      "epoch": 147.09,
+      "learning_rate": 7.429665071770335e-05,
+      "loss": 0.0525,
+      "step": 1618
+    },
+    {
+      "epoch": 147.18,
+      "learning_rate": 7.42775119617225e-05,
+      "loss": 0.0522,
+      "step": 1619
+    },
+    {
+      "epoch": 147.27,
+      "learning_rate": 7.425837320574163e-05,
+      "loss": 0.0553,
+      "step": 1620
+    },
+    {
+      "epoch": 147.27,
+      "eval_accuracy_NAT": 0.9921498662853441,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.95333758959612,
+      "eval_accuracy_concrete tiles": 0.9801871216290589,
+      "eval_accuracy_dark bricks": 0.8421538816827682,
+      "eval_accuracy_grass": 0.8760087656448237,
+      "eval_accuracy_red bricks": 0.8609730113636364,
+      "eval_accuracy_road": 0.9726356037035648,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6472307225760208,
+      "eval_accuracy_speedway bricks": 0.9868402791605874,
+      "eval_accuracy_steel": 0.88838961809907,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9789861436309871,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8731390862220825,
+      "eval_iou_concrete tiles": 0.9440763318314339,
+      "eval_iou_dark bricks": 0.8132317413120238,
+      "eval_iou_grass": 0.8220055491056495,
+      "eval_iou_red bricks": 0.807897367544152,
+      "eval_iou_road": 0.9638362349422495,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4260411940840214,
+      "eval_iou_speedway bricks": 0.9737318732718729,
+      "eval_iou_steel": 0.830681345113095,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09003419429063797,
+      "eval_mean_accuracy": 0.6923004969031534,
+      "eval_mean_iou": 0.6487405282351975,
+      "eval_overall_accuracy": 0.9763547049628364,
+      "eval_runtime": 3.7779,
+      "eval_samples_per_second": 9.529,
+      "eval_steps_per_second": 0.529,
+      "step": 1620
+    },
+    {
+      "epoch": 147.36,
+      "learning_rate": 7.423923444976076e-05,
+      "loss": 0.0462,
+      "step": 1621
+    },
+    {
+      "epoch": 147.45,
+      "learning_rate": 7.422009569377992e-05,
+      "loss": 0.0419,
+      "step": 1622
+    },
+    {
+      "epoch": 147.55,
+      "learning_rate": 7.420095693779905e-05,
+      "loss": 0.0506,
+      "step": 1623
+    },
+    {
+      "epoch": 147.64,
+      "learning_rate": 7.418181818181818e-05,
+      "loss": 0.0418,
+      "step": 1624
+    },
+    {
+      "epoch": 147.73,
+      "learning_rate": 7.416267942583733e-05,
+      "loss": 0.0492,
+      "step": 1625
+    },
+    {
+      "epoch": 147.82,
+      "learning_rate": 7.414354066985646e-05,
+      "loss": 0.0482,
+      "step": 1626
+    },
+    {
+      "epoch": 147.91,
+      "learning_rate": 7.412440191387559e-05,
+      "loss": 0.0525,
+      "step": 1627
+    },
+    {
+      "epoch": 148.0,
+      "learning_rate": 7.410526315789475e-05,
+      "loss": 0.0425,
+      "step": 1628
+    },
+    {
+      "epoch": 148.09,
+      "learning_rate": 7.408612440191388e-05,
+      "loss": 0.045,
+      "step": 1629
+    },
+    {
+      "epoch": 148.18,
+      "learning_rate": 7.406698564593301e-05,
+      "loss": 0.0446,
+      "step": 1630
+    },
+    {
+      "epoch": 148.27,
+      "learning_rate": 7.404784688995216e-05,
+      "loss": 0.045,
+      "step": 1631
+    },
+    {
+      "epoch": 148.36,
+      "learning_rate": 7.402870813397129e-05,
+      "loss": 0.0425,
+      "step": 1632
+    },
+    {
+      "epoch": 148.45,
+      "learning_rate": 7.400956937799042e-05,
+      "loss": 0.045,
+      "step": 1633
+    },
+    {
+      "epoch": 148.55,
+      "learning_rate": 7.399043062200958e-05,
+      "loss": 0.0461,
+      "step": 1634
+    },
+    {
+      "epoch": 148.64,
+      "learning_rate": 7.397129186602871e-05,
+      "loss": 0.0486,
+      "step": 1635
+    },
+    {
+      "epoch": 148.73,
+      "learning_rate": 7.395215311004785e-05,
+      "loss": 0.0461,
+      "step": 1636
+    },
+    {
+      "epoch": 148.82,
+      "learning_rate": 7.393301435406699e-05,
+      "loss": 0.0475,
+      "step": 1637
+    },
+    {
+      "epoch": 148.91,
+      "learning_rate": 7.391387559808612e-05,
+      "loss": 0.0525,
+      "step": 1638
+    },
+    {
+      "epoch": 149.0,
+      "learning_rate": 7.389473684210527e-05,
+      "loss": 0.0733,
+      "step": 1639
+    },
+    {
+      "epoch": 149.09,
+      "learning_rate": 7.387559808612442e-05,
+      "loss": 0.0475,
+      "step": 1640
+    },
+    {
+      "epoch": 149.09,
+      "eval_accuracy_NAT": 0.9921392792183639,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9503559702133139,
+      "eval_accuracy_concrete tiles": 0.9866996881306183,
+      "eval_accuracy_dark bricks": 0.8487741041530349,
+      "eval_accuracy_grass": 0.9025433651268537,
+      "eval_accuracy_red bricks": 0.9156605113636364,
+      "eval_accuracy_road": 0.976411962945004,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6331706779147024,
+      "eval_accuracy_speedway bricks": 0.9847259382086271,
+      "eval_accuracy_steel": 0.9052997823362575,
+      "eval_accuracy_tiles": 0.06238871387481167,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9791981841077735,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8843694068132971,
+      "eval_iou_concrete tiles": 0.9312613626525842,
+      "eval_iou_dark bricks": 0.8053759906504511,
+      "eval_iou_grass": 0.8241987271555054,
+      "eval_iou_red bricks": 0.8091950415816727,
+      "eval_iou_road": 0.9697386024369564,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.42141284403669726,
+      "eval_iou_speedway bricks": 0.9739739034788001,
+      "eval_iou_steel": 0.8297401252899881,
+      "eval_iou_tiles": 0.06238871387481167,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08501545339822769,
+      "eval_mean_accuracy": 0.7044746148834788,
+      "eval_mean_iou": 0.6531425309291183,
+      "eval_overall_accuracy": 0.9772101508246528,
+      "eval_runtime": 3.6488,
+      "eval_samples_per_second": 9.866,
+      "eval_steps_per_second": 0.548,
+      "step": 1640
+    },
+    {
+      "epoch": 149.18,
+      "learning_rate": 7.385645933014355e-05,
+      "loss": 0.0503,
+      "step": 1641
+    },
+    {
+      "epoch": 149.27,
+      "learning_rate": 7.383732057416268e-05,
+      "loss": 0.0453,
+      "step": 1642
+    },
+    {
+      "epoch": 149.36,
+      "learning_rate": 7.381818181818182e-05,
+      "loss": 0.0408,
+      "step": 1643
+    },
+    {
+      "epoch": 149.45,
+      "learning_rate": 7.379904306220096e-05,
+      "loss": 0.051,
+      "step": 1644
+    },
+    {
+      "epoch": 149.55,
+      "learning_rate": 7.37799043062201e-05,
+      "loss": 0.0489,
+      "step": 1645
+    },
+    {
+      "epoch": 149.64,
+      "learning_rate": 7.376076555023925e-05,
+      "loss": 0.0447,
+      "step": 1646
+    },
+    {
+      "epoch": 149.73,
+      "learning_rate": 7.374162679425838e-05,
+      "loss": 0.049,
+      "step": 1647
+    },
+    {
+      "epoch": 149.82,
+      "learning_rate": 7.372248803827751e-05,
+      "loss": 0.0458,
+      "step": 1648
+    },
+    {
+      "epoch": 149.91,
+      "learning_rate": 7.370334928229666e-05,
+      "loss": 0.0594,
+      "step": 1649
+    },
+    {
+      "epoch": 150.0,
+      "learning_rate": 7.368421052631579e-05,
+      "loss": 0.0517,
+      "step": 1650
+    },
+    {
+      "epoch": 150.09,
+      "learning_rate": 7.366507177033493e-05,
+      "loss": 0.0427,
+      "step": 1651
+    },
+    {
+      "epoch": 150.18,
+      "learning_rate": 7.364593301435408e-05,
+      "loss": 0.0498,
+      "step": 1652
+    },
+    {
+      "epoch": 150.27,
+      "learning_rate": 7.362679425837321e-05,
+      "loss": 0.0437,
+      "step": 1653
+    },
+    {
+      "epoch": 150.36,
+      "learning_rate": 7.360765550239234e-05,
+      "loss": 0.0473,
+      "step": 1654
+    },
+    {
+      "epoch": 150.45,
+      "learning_rate": 7.358851674641149e-05,
+      "loss": 0.0543,
+      "step": 1655
+    },
+    {
+      "epoch": 150.55,
+      "learning_rate": 7.356937799043062e-05,
+      "loss": 0.051,
+      "step": 1656
+    },
+    {
+      "epoch": 150.64,
+      "learning_rate": 7.355023923444977e-05,
+      "loss": 0.0454,
+      "step": 1657
+    },
+    {
+      "epoch": 150.73,
+      "learning_rate": 7.35311004784689e-05,
+      "loss": 0.0456,
+      "step": 1658
+    },
+    {
+      "epoch": 150.82,
+      "learning_rate": 7.351196172248805e-05,
+      "loss": 0.0481,
+      "step": 1659
+    },
+    {
+      "epoch": 150.91,
+      "learning_rate": 7.349282296650718e-05,
+      "loss": 0.046,
+      "step": 1660
+    },
+    {
+      "epoch": 150.91,
+      "eval_accuracy_NAT": 0.9925085914048544,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9531149405553886,
+      "eval_accuracy_concrete tiles": 0.9787500764385739,
+      "eval_accuracy_dark bricks": 0.8538932296678342,
+      "eval_accuracy_grass": 0.8979508382934254,
+      "eval_accuracy_red bricks": 0.9085582386363636,
+      "eval_accuracy_road": 0.9813987402462191,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6511179113941499,
+      "eval_accuracy_speedway bricks": 0.9844151831885758,
+      "eval_accuracy_steel": 0.8964118461842886,
+      "eval_accuracy_tiles": 0.0611560060265717,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9791510469388235,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8915179260326115,
+      "eval_iou_concrete tiles": 0.9490927419354839,
+      "eval_iou_dark bricks": 0.8049125607720775,
+      "eval_iou_grass": 0.8260025270142972,
+      "eval_iou_red bricks": 0.8328450520833334,
+      "eval_iou_road": 0.9727532160266176,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4306946040082426,
+      "eval_iou_speedway bricks": 0.9743620371570392,
+      "eval_iou_steel": 0.8321712042005036,
+      "eval_iou_tiles": 0.0611560060265717,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08345242589712143,
+      "eval_mean_accuracy": 0.7045596616950959,
+      "eval_mean_iou": 0.6580506863227386,
+      "eval_overall_accuracy": 0.977828237745497,
+      "eval_runtime": 3.3211,
+      "eval_samples_per_second": 10.84,
+      "eval_steps_per_second": 0.602,
+      "step": 1660
+    },
+    {
+      "epoch": 151.0,
+      "learning_rate": 7.347368421052632e-05,
+      "loss": 0.0515,
+      "step": 1661
+    },
+    {
+      "epoch": 151.09,
+      "learning_rate": 7.345454545454545e-05,
+      "loss": 0.0444,
+      "step": 1662
+    },
+    {
+      "epoch": 151.18,
+      "learning_rate": 7.34354066985646e-05,
+      "loss": 0.0528,
+      "step": 1663
+    },
+    {
+      "epoch": 151.27,
+      "learning_rate": 7.341626794258373e-05,
+      "loss": 0.0464,
+      "step": 1664
+    },
+    {
+      "epoch": 151.36,
+      "learning_rate": 7.339712918660288e-05,
+      "loss": 0.0475,
+      "step": 1665
+    },
+    {
+      "epoch": 151.45,
+      "learning_rate": 7.337799043062201e-05,
+      "loss": 0.0424,
+      "step": 1666
+    },
+    {
+      "epoch": 151.55,
+      "learning_rate": 7.335885167464116e-05,
+      "loss": 0.0443,
+      "step": 1667
+    },
+    {
+      "epoch": 151.64,
+      "learning_rate": 7.333971291866029e-05,
+      "loss": 0.0474,
+      "step": 1668
+    },
+    {
+      "epoch": 151.73,
+      "learning_rate": 7.332057416267943e-05,
+      "loss": 0.0501,
+      "step": 1669
+    },
+    {
+      "epoch": 151.82,
+      "learning_rate": 7.330143540669856e-05,
+      "loss": 0.0501,
+      "step": 1670
+    },
+    {
+      "epoch": 151.91,
+      "learning_rate": 7.328229665071771e-05,
+      "loss": 0.0446,
+      "step": 1671
+    },
+    {
+      "epoch": 152.0,
+      "learning_rate": 7.326315789473684e-05,
+      "loss": 0.0476,
+      "step": 1672
+    },
+    {
+      "epoch": 152.09,
+      "learning_rate": 7.324401913875599e-05,
+      "loss": 0.0436,
+      "step": 1673
+    },
+    {
+      "epoch": 152.18,
+      "learning_rate": 7.322488038277512e-05,
+      "loss": 0.0466,
+      "step": 1674
+    },
+    {
+      "epoch": 152.27,
+      "learning_rate": 7.320574162679427e-05,
+      "loss": 0.0636,
+      "step": 1675
+    },
+    {
+      "epoch": 152.36,
+      "learning_rate": 7.31866028708134e-05,
+      "loss": 0.0458,
+      "step": 1676
+    },
+    {
+      "epoch": 152.45,
+      "learning_rate": 7.316746411483254e-05,
+      "loss": 0.0584,
+      "step": 1677
+    },
+    {
+      "epoch": 152.55,
+      "learning_rate": 7.314832535885168e-05,
+      "loss": 0.0469,
+      "step": 1678
+    },
+    {
+      "epoch": 152.64,
+      "learning_rate": 7.312918660287081e-05,
+      "loss": 0.0439,
+      "step": 1679
+    },
+    {
+      "epoch": 152.73,
+      "learning_rate": 7.311004784688995e-05,
+      "loss": 0.0472,
+      "step": 1680
+    },
+    {
+      "epoch": 152.73,
+      "eval_accuracy_NAT": 0.9927157450154324,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9408799733894159,
+      "eval_accuracy_concrete tiles": 0.9845288326300985,
+      "eval_accuracy_dark bricks": 0.8359570455332743,
+      "eval_accuracy_grass": 0.8947598360111423,
+      "eval_accuracy_red bricks": 0.931640625,
+      "eval_accuracy_road": 0.9858969971716605,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6512006175392165,
+      "eval_accuracy_speedway bricks": 0.985736646283551,
+      "eval_accuracy_steel": 0.8899891168128752,
+      "eval_accuracy_tiles": 0.07122312012053143,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9786831498929034,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8884388583239612,
+      "eval_iou_concrete tiles": 0.9377093101138647,
+      "eval_iou_dark bricks": 0.8009662192063727,
+      "eval_iou_grass": 0.8210030754824081,
+      "eval_iou_red bricks": 0.8282557221783741,
+      "eval_iou_road": 0.9747980339620972,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.43012965256027386,
+      "eval_iou_speedway bricks": 0.9741375454908118,
+      "eval_iou_steel": 0.8296377708264482,
+      "eval_iou_tiles": 0.07122312012053143,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08421332389116287,
+      "eval_mean_accuracy": 0.7049637350390152,
+      "eval_mean_iou": 0.6565371121660035,
+      "eval_overall_accuracy": 0.9774333106146919,
+      "eval_runtime": 4.476,
+      "eval_samples_per_second": 8.043,
+      "eval_steps_per_second": 0.447,
+      "step": 1680
+    },
+    {
+      "epoch": 152.82,
+      "learning_rate": 7.30909090909091e-05,
+      "loss": 0.042,
+      "step": 1681
+    },
+    {
+      "epoch": 152.91,
+      "learning_rate": 7.307177033492823e-05,
+      "loss": 0.0641,
+      "step": 1682
+    },
+    {
+      "epoch": 153.0,
+      "learning_rate": 7.305263157894738e-05,
+      "loss": 0.0462,
+      "step": 1683
+    },
+    {
+      "epoch": 153.09,
+      "learning_rate": 7.303349282296651e-05,
+      "loss": 0.0505,
+      "step": 1684
+    },
+    {
+      "epoch": 153.18,
+      "learning_rate": 7.301435406698564e-05,
+      "loss": 0.0454,
+      "step": 1685
+    },
+    {
+      "epoch": 153.27,
+      "learning_rate": 7.299521531100479e-05,
+      "loss": 0.0499,
+      "step": 1686
+    },
+    {
+      "epoch": 153.36,
+      "learning_rate": 7.297607655502393e-05,
+      "loss": 0.0469,
+      "step": 1687
+    },
+    {
+      "epoch": 153.45,
+      "learning_rate": 7.295693779904306e-05,
+      "loss": 0.0445,
+      "step": 1688
+    },
+    {
+      "epoch": 153.55,
+      "learning_rate": 7.293779904306221e-05,
+      "loss": 0.0476,
+      "step": 1689
+    },
+    {
+      "epoch": 153.64,
+      "learning_rate": 7.291866028708134e-05,
+      "loss": 0.0455,
+      "step": 1690
+    },
+    {
+      "epoch": 153.73,
+      "learning_rate": 7.289952153110047e-05,
+      "loss": 0.0516,
+      "step": 1691
+    },
+    {
+      "epoch": 153.82,
+      "learning_rate": 7.288038277511962e-05,
+      "loss": 0.0496,
+      "step": 1692
+    },
+    {
+      "epoch": 153.91,
+      "learning_rate": 7.286124401913876e-05,
+      "loss": 0.0448,
+      "step": 1693
+    },
+    {
+      "epoch": 154.0,
+      "learning_rate": 7.28421052631579e-05,
+      "loss": 0.0568,
+      "step": 1694
+    },
+    {
+      "epoch": 154.09,
+      "learning_rate": 7.282296650717704e-05,
+      "loss": 0.0458,
+      "step": 1695
+    },
+    {
+      "epoch": 154.18,
+      "learning_rate": 7.280382775119617e-05,
+      "loss": 0.0464,
+      "step": 1696
+    },
+    {
+      "epoch": 154.27,
+      "learning_rate": 7.27846889952153e-05,
+      "loss": 0.0549,
+      "step": 1697
+    },
+    {
+      "epoch": 154.36,
+      "learning_rate": 7.276555023923445e-05,
+      "loss": 0.0511,
+      "step": 1698
+    },
+    {
+      "epoch": 154.45,
+      "learning_rate": 7.27464114832536e-05,
+      "loss": 0.0523,
+      "step": 1699
+    },
+    {
+      "epoch": 154.55,
+      "learning_rate": 7.272727272727273e-05,
+      "loss": 0.047,
+      "step": 1700
+    },
+    {
+      "epoch": 154.55,
+      "eval_accuracy_NAT": 0.9929484840378792,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9456830228765183,
+      "eval_accuracy_concrete tiles": 0.9848040114963615,
+      "eval_accuracy_dark bricks": 0.8400369500789038,
+      "eval_accuracy_grass": 0.88574254588158,
+      "eval_accuracy_red bricks": 0.939453125,
+      "eval_accuracy_road": 0.9690817388906959,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6245968075428004,
+      "eval_accuracy_speedway bricks": 0.9876397945034376,
+      "eval_accuracy_steel": 0.8951338961809907,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9786417448473447,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8799385477858358,
+      "eval_iou_concrete tiles": 0.9331614323791865,
+      "eval_iou_dark bricks": 0.8028619776339023,
+      "eval_iou_grass": 0.8180072238056351,
+      "eval_iou_red bricks": 0.8250428816466552,
+      "eval_iou_road": 0.962262021401264,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.41792256184169263,
+      "eval_iou_speedway bricks": 0.9741113035405025,
+      "eval_iou_steel": 0.8284421451027073,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08950748294591904,
+      "eval_mean_accuracy": 0.6973169520376282,
+      "eval_mean_iou": 0.6477224492295943,
+      "eval_overall_accuracy": 0.9764103359646268,
+      "eval_runtime": 3.6028,
+      "eval_samples_per_second": 9.992,
+      "eval_steps_per_second": 0.555,
+      "step": 1700
+    },
+    {
+      "epoch": 154.64,
+      "learning_rate": 7.270813397129187e-05,
+      "loss": 0.0503,
+      "step": 1701
+    },
+    {
+      "epoch": 154.73,
+      "learning_rate": 7.2688995215311e-05,
+      "loss": 0.0518,
+      "step": 1702
+    },
+    {
+      "epoch": 154.82,
+      "learning_rate": 7.266985645933014e-05,
+      "loss": 0.0452,
+      "step": 1703
+    },
+    {
+      "epoch": 154.91,
+      "learning_rate": 7.265071770334928e-05,
+      "loss": 0.0481,
+      "step": 1704
+    },
+    {
+      "epoch": 155.0,
+      "learning_rate": 7.263157894736843e-05,
+      "loss": 0.0474,
+      "step": 1705
+    },
+    {
+      "epoch": 155.09,
+      "learning_rate": 7.261244019138756e-05,
+      "loss": 0.047,
+      "step": 1706
+    },
+    {
+      "epoch": 155.18,
+      "learning_rate": 7.25933014354067e-05,
+      "loss": 0.0455,
+      "step": 1707
+    },
+    {
+      "epoch": 155.27,
+      "learning_rate": 7.257416267942584e-05,
+      "loss": 0.0446,
+      "step": 1708
+    },
+    {
+      "epoch": 155.36,
+      "learning_rate": 7.255502392344497e-05,
+      "loss": 0.0505,
+      "step": 1709
+    },
+    {
+      "epoch": 155.45,
+      "learning_rate": 7.253588516746413e-05,
+      "loss": 0.0515,
+      "step": 1710
+    },
+    {
+      "epoch": 155.55,
+      "learning_rate": 7.251674641148326e-05,
+      "loss": 0.0466,
+      "step": 1711
+    },
+    {
+      "epoch": 155.64,
+      "learning_rate": 7.24976076555024e-05,
+      "loss": 0.0414,
+      "step": 1712
+    },
+    {
+      "epoch": 155.73,
+      "learning_rate": 7.247846889952154e-05,
+      "loss": 0.0406,
+      "step": 1713
+    },
+    {
+      "epoch": 155.82,
+      "learning_rate": 7.245933014354067e-05,
+      "loss": 0.0499,
+      "step": 1714
+    },
+    {
+      "epoch": 155.91,
+      "learning_rate": 7.24401913875598e-05,
+      "loss": 0.0509,
+      "step": 1715
+    },
+    {
+      "epoch": 156.0,
+      "learning_rate": 7.242105263157896e-05,
+      "loss": 0.0427,
+      "step": 1716
+    },
+    {
+      "epoch": 156.09,
+      "learning_rate": 7.24019138755981e-05,
+      "loss": 0.0516,
+      "step": 1717
+    },
+    {
+      "epoch": 156.18,
+      "learning_rate": 7.238277511961723e-05,
+      "loss": 0.0494,
+      "step": 1718
+    },
+    {
+      "epoch": 156.27,
+      "learning_rate": 7.236363636363637e-05,
+      "loss": 0.045,
+      "step": 1719
+    },
+    {
+      "epoch": 156.36,
+      "learning_rate": 7.23444976076555e-05,
+      "loss": 0.0449,
+      "step": 1720
+    },
+    {
+      "epoch": 156.36,
+      "eval_accuracy_NAT": 0.9926827486566776,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9527997446242328,
+      "eval_accuracy_concrete tiles": 0.9853543692288876,
+      "eval_accuracy_dark bricks": 0.8538932296678342,
+      "eval_accuracy_grass": 0.8600956951177315,
+      "eval_accuracy_red bricks": 0.9263139204545454,
+      "eval_accuracy_road": 0.9723151633580697,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6287321147961293,
+      "eval_accuracy_speedway bricks": 0.9871588784821156,
+      "eval_accuracy_steel": 0.888579249389882,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9791064024191267,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8735758125784725,
+      "eval_iou_concrete tiles": 0.9357704927553064,
+      "eval_iou_dark bricks": 0.8076082999635966,
+      "eval_iou_grass": 0.8005100613504915,
+      "eval_iou_red bricks": 0.8087118276236243,
+      "eval_iou_road": 0.9662476867985291,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.41233795584805366,
+      "eval_iou_speedway bricks": 0.9748158485268478,
+      "eval_iou_steel": 0.8344159614744389,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09230832010507584,
+      "eval_mean_accuracy": 0.695994239521239,
+      "eval_mean_iou": 0.6456231037952682,
+      "eval_overall_accuracy": 0.9761981964111328,
+      "eval_runtime": 3.5636,
+      "eval_samples_per_second": 10.102,
+      "eval_steps_per_second": 0.561,
+      "step": 1720
+    },
+    {
+      "epoch": 156.45,
+      "learning_rate": 7.232535885167464e-05,
+      "loss": 0.0434,
+      "step": 1721
+    },
+    {
+      "epoch": 156.55,
+      "learning_rate": 7.23062200956938e-05,
+      "loss": 0.0398,
+      "step": 1722
+    },
+    {
+      "epoch": 156.64,
+      "learning_rate": 7.228708133971293e-05,
+      "loss": 0.051,
+      "step": 1723
+    },
+    {
+      "epoch": 156.73,
+      "learning_rate": 7.226794258373206e-05,
+      "loss": 0.059,
+      "step": 1724
+    },
+    {
+      "epoch": 156.82,
+      "learning_rate": 7.22488038277512e-05,
+      "loss": 0.0505,
+      "step": 1725
+    },
+    {
+      "epoch": 156.91,
+      "learning_rate": 7.222966507177034e-05,
+      "loss": 0.0462,
+      "step": 1726
+    },
+    {
+      "epoch": 157.0,
+      "learning_rate": 7.221052631578947e-05,
+      "loss": 0.0434,
+      "step": 1727
+    },
+    {
+      "epoch": 157.09,
+      "learning_rate": 7.219138755980863e-05,
+      "loss": 0.0469,
+      "step": 1728
+    },
+    {
+      "epoch": 157.18,
+      "learning_rate": 7.217224880382776e-05,
+      "loss": 0.0407,
+      "step": 1729
+    },
+    {
+      "epoch": 157.27,
+      "learning_rate": 7.215311004784689e-05,
+      "loss": 0.049,
+      "step": 1730
+    },
+    {
+      "epoch": 157.36,
+      "learning_rate": 7.213397129186604e-05,
+      "loss": 0.0466,
+      "step": 1731
+    },
+    {
+      "epoch": 157.45,
+      "learning_rate": 7.211483253588517e-05,
+      "loss": 0.0443,
+      "step": 1732
+    },
+    {
+      "epoch": 157.55,
+      "learning_rate": 7.20956937799043e-05,
+      "loss": 0.0548,
+      "step": 1733
+    },
+    {
+      "epoch": 157.64,
+      "learning_rate": 7.207655502392345e-05,
+      "loss": 0.0516,
+      "step": 1734
+    },
+    {
+      "epoch": 157.73,
+      "learning_rate": 7.205741626794259e-05,
+      "loss": 0.0489,
+      "step": 1735
+    },
+    {
+      "epoch": 157.82,
+      "learning_rate": 7.203827751196172e-05,
+      "loss": 0.043,
+      "step": 1736
+    },
+    {
+      "epoch": 157.91,
+      "learning_rate": 7.201913875598087e-05,
+      "loss": 0.0436,
+      "step": 1737
+    },
+    {
+      "epoch": 158.0,
+      "learning_rate": 7.2e-05,
+      "loss": 0.0638,
+      "step": 1738
+    },
+    {
+      "epoch": 158.09,
+      "learning_rate": 7.198086124401913e-05,
+      "loss": 0.0518,
+      "step": 1739
+    },
+    {
+      "epoch": 158.18,
+      "learning_rate": 7.196172248803828e-05,
+      "loss": 0.0485,
+      "step": 1740
+    },
+    {
+      "epoch": 158.18,
+      "eval_accuracy_NAT": 0.992552527732822,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9478974419502983,
+      "eval_accuracy_concrete tiles": 0.9776493609735217,
+      "eval_accuracy_dark bricks": 0.8441553442900581,
+      "eval_accuracy_grass": 0.900638549963826,
+      "eval_accuracy_red bricks": 0.9018110795454546,
+      "eval_accuracy_road": 0.9752632145366251,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6388774019242963,
+      "eval_accuracy_speedway bricks": 0.9891784844085457,
+      "eval_accuracy_steel": 0.8833025526020711,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.97978366703535,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8849400833948986,
+      "eval_iou_concrete tiles": 0.9470426206202055,
+      "eval_iou_dark bricks": 0.8031346125677458,
+      "eval_iou_grass": 0.828413628151301,
+      "eval_iou_red bricks": 0.8396429161844933,
+      "eval_iou_road": 0.967961835038555,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.42297583412426076,
+      "eval_iou_speedway bricks": 0.974302802223045,
+      "eval_iou_steel": 0.8296536075768018,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08880563080310822,
+      "eval_mean_accuracy": 0.6962558429175013,
+      "eval_mean_iou": 0.6521424313012812,
+      "eval_overall_accuracy": 0.977477921379937,
+      "eval_runtime": 3.7095,
+      "eval_samples_per_second": 9.705,
+      "eval_steps_per_second": 0.539,
+      "step": 1740
+    },
+    {
+      "epoch": 158.27,
+      "learning_rate": 7.194258373205742e-05,
+      "loss": 0.0494,
+      "step": 1741
+    },
+    {
+      "epoch": 158.36,
+      "learning_rate": 7.192344497607656e-05,
+      "loss": 0.0414,
+      "step": 1742
+    },
+    {
+      "epoch": 158.45,
+      "learning_rate": 7.19043062200957e-05,
+      "loss": 0.0471,
+      "step": 1743
+    },
+    {
+      "epoch": 158.55,
+      "learning_rate": 7.188516746411483e-05,
+      "loss": 0.0441,
+      "step": 1744
+    },
+    {
+      "epoch": 158.64,
+      "learning_rate": 7.186602870813397e-05,
+      "loss": 0.0422,
+      "step": 1745
+    },
+    {
+      "epoch": 158.73,
+      "learning_rate": 7.184688995215311e-05,
+      "loss": 0.0412,
+      "step": 1746
+    },
+    {
+      "epoch": 158.82,
+      "learning_rate": 7.182775119617226e-05,
+      "loss": 0.0475,
+      "step": 1747
+    },
+    {
+      "epoch": 158.91,
+      "learning_rate": 7.180861244019139e-05,
+      "loss": 0.0469,
+      "step": 1748
+    },
+    {
+      "epoch": 159.0,
+      "learning_rate": 7.178947368421054e-05,
+      "loss": 0.0445,
+      "step": 1749
+    },
+    {
+      "epoch": 159.09,
+      "learning_rate": 7.177033492822967e-05,
+      "loss": 0.0504,
+      "step": 1750
+    },
+    {
+      "epoch": 159.18,
+      "learning_rate": 7.17511961722488e-05,
+      "loss": 0.049,
+      "step": 1751
+    },
+    {
+      "epoch": 159.27,
+      "learning_rate": 7.173205741626794e-05,
+      "loss": 0.0525,
+      "step": 1752
+    },
+    {
+      "epoch": 159.36,
+      "learning_rate": 7.171291866028709e-05,
+      "loss": 0.0481,
+      "step": 1753
+    },
+    {
+      "epoch": 159.45,
+      "learning_rate": 7.169377990430622e-05,
+      "loss": 0.0419,
+      "step": 1754
+    },
+    {
+      "epoch": 159.55,
+      "learning_rate": 7.167464114832535e-05,
+      "loss": 0.0455,
+      "step": 1755
+    },
+    {
+      "epoch": 159.64,
+      "learning_rate": 7.16555023923445e-05,
+      "loss": 0.051,
+      "step": 1756
+    },
+    {
+      "epoch": 159.73,
+      "learning_rate": 7.163636363636363e-05,
+      "loss": 0.0478,
+      "step": 1757
+    },
+    {
+      "epoch": 159.82,
+      "learning_rate": 7.161722488038278e-05,
+      "loss": 0.0474,
+      "step": 1758
+    },
+    {
+      "epoch": 159.91,
+      "learning_rate": 7.159808612440192e-05,
+      "loss": 0.049,
+      "step": 1759
+    },
+    {
+      "epoch": 160.0,
+      "learning_rate": 7.157894736842105e-05,
+      "loss": 0.047,
+      "step": 1760
+    },
+    {
+      "epoch": 160.0,
+      "eval_accuracy_NAT": 0.9918191968933311,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9529754495901112,
+      "eval_accuracy_concrete tiles": 0.9863633584051856,
+      "eval_accuracy_dark bricks": 0.8652861706631769,
+      "eval_accuracy_grass": 0.8938616020718797,
+      "eval_accuracy_red bricks": 0.9133522727272727,
+      "eval_accuracy_road": 0.9783128392586582,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6441981639235795,
+      "eval_accuracy_speedway bricks": 0.9845117284375238,
+      "eval_accuracy_steel": 0.9083586175054416,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9794217766492862,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.883233056536224,
+      "eval_iou_concrete tiles": 0.9294687103837732,
+      "eval_iou_dark bricks": 0.80392647689887,
+      "eval_iou_grass": 0.8221378561003992,
+      "eval_iou_red bricks": 0.8477257745550428,
+      "eval_iou_road": 0.967997128499641,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4282258507889384,
+      "eval_iou_speedway bricks": 0.9743173569235289,
+      "eval_iou_steel": 0.8297034325907852,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0899038016796112,
+      "eval_mean_accuracy": 0.701464569190474,
+      "eval_mean_iou": 0.6512428784558838,
+      "eval_overall_accuracy": 0.9771187040540907,
+      "eval_runtime": 3.9964,
+      "eval_samples_per_second": 9.008,
+      "eval_steps_per_second": 0.5,
+      "step": 1760
+    },
+    {
+      "epoch": 160.09,
+      "learning_rate": 7.155980861244019e-05,
+      "loss": 0.042,
+      "step": 1761
+    },
+    {
+      "epoch": 160.18,
+      "learning_rate": 7.154066985645933e-05,
+      "loss": 0.0449,
+      "step": 1762
+    },
+    {
+      "epoch": 160.27,
+      "learning_rate": 7.152153110047846e-05,
+      "loss": 0.0459,
+      "step": 1763
+    },
+    {
+      "epoch": 160.36,
+      "learning_rate": 7.150239234449761e-05,
+      "loss": 0.0441,
+      "step": 1764
+    },
+    {
+      "epoch": 160.45,
+      "learning_rate": 7.148325358851676e-05,
+      "loss": 0.0481,
+      "step": 1765
+    },
+    {
+      "epoch": 160.55,
+      "learning_rate": 7.146411483253589e-05,
+      "loss": 0.0447,
+      "step": 1766
+    },
+    {
+      "epoch": 160.64,
+      "learning_rate": 7.144497607655502e-05,
+      "loss": 0.0495,
+      "step": 1767
+    },
+    {
+      "epoch": 160.73,
+      "learning_rate": 7.142583732057417e-05,
+      "loss": 0.0477,
+      "step": 1768
+    },
+    {
+      "epoch": 160.82,
+      "learning_rate": 7.14066985645933e-05,
+      "loss": 0.041,
+      "step": 1769
+    },
+    {
+      "epoch": 160.91,
+      "learning_rate": 7.138755980861244e-05,
+      "loss": 0.0508,
+      "step": 1770
+    },
+    {
+      "epoch": 161.0,
+      "learning_rate": 7.136842105263159e-05,
+      "loss": 0.0498,
+      "step": 1771
+    },
+    {
+      "epoch": 161.09,
+      "learning_rate": 7.134928229665072e-05,
+      "loss": 0.043,
+      "step": 1772
+    },
+    {
+      "epoch": 161.18,
+      "learning_rate": 7.133014354066985e-05,
+      "loss": 0.0434,
+      "step": 1773
+    },
+    {
+      "epoch": 161.27,
+      "learning_rate": 7.1311004784689e-05,
+      "loss": 0.051,
+      "step": 1774
+    },
+    {
+      "epoch": 161.36,
+      "learning_rate": 7.129186602870813e-05,
+      "loss": 0.0391,
+      "step": 1775
+    },
+    {
+      "epoch": 161.45,
+      "learning_rate": 7.127272727272728e-05,
+      "loss": 0.0472,
+      "step": 1776
+    },
+    {
+      "epoch": 161.55,
+      "learning_rate": 7.125358851674642e-05,
+      "loss": 0.0422,
+      "step": 1777
+    },
+    {
+      "epoch": 161.64,
+      "learning_rate": 7.123444976076555e-05,
+      "loss": 0.0408,
+      "step": 1778
+    },
+    {
+      "epoch": 161.73,
+      "learning_rate": 7.121531100478468e-05,
+      "loss": 0.044,
+      "step": 1779
+    },
+    {
+      "epoch": 161.82,
+      "learning_rate": 7.119617224880383e-05,
+      "loss": 0.0573,
+      "step": 1780
+    },
+    {
+      "epoch": 161.82,
+      "eval_accuracy_NAT": 0.9930889391264823,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9508683312588523,
+      "eval_accuracy_concrete tiles": 0.9769461260930716,
+      "eval_accuracy_dark bricks": 0.8354566798814518,
+      "eval_accuracy_grass": 0.8661421726077094,
+      "eval_accuracy_red bricks": 0.8973721590909091,
+      "eval_accuracy_road": 0.9766949178161205,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6550051002122791,
+      "eval_accuracy_speedway bricks": 0.9849739388168621,
+      "eval_accuracy_steel": 0.9008145900666183,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9786863948397175,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8782483520540325,
+      "eval_iou_concrete tiles": 0.9518305579552563,
+      "eval_iou_dark bricks": 0.8019655656543264,
+      "eval_iou_grass": 0.8099050930767622,
+      "eval_iou_red bricks": 0.8302940693280763,
+      "eval_iou_road": 0.9679214555599095,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4319111418131579,
+      "eval_iou_speedway bricks": 0.9746962045799649,
+      "eval_iou_steel": 0.8354974382503633,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09000352770090103,
+      "eval_mean_accuracy": 0.6951817657669506,
+      "eval_mean_iou": 0.6508427902393513,
+      "eval_overall_accuracy": 0.9766340255737305,
+      "eval_runtime": 3.3905,
+      "eval_samples_per_second": 10.618,
+      "eval_steps_per_second": 0.59,
+      "step": 1780
+    },
+    {
+      "epoch": 161.91,
+      "learning_rate": 7.117703349282298e-05,
+      "loss": 0.0451,
+      "step": 1781
+    },
+    {
+      "epoch": 162.0,
+      "learning_rate": 7.115789473684211e-05,
+      "loss": 0.048,
+      "step": 1782
+    },
+    {
+      "epoch": 162.09,
+      "learning_rate": 7.113875598086125e-05,
+      "loss": 0.0546,
+      "step": 1783
+    },
+    {
+      "epoch": 162.18,
+      "learning_rate": 7.111961722488039e-05,
+      "loss": 0.0458,
+      "step": 1784
+    },
+    {
+      "epoch": 162.27,
+      "learning_rate": 7.110047846889952e-05,
+      "loss": 0.0451,
+      "step": 1785
+    },
+    {
+      "epoch": 162.36,
+      "learning_rate": 7.108133971291866e-05,
+      "loss": 0.0398,
+      "step": 1786
+    },
+    {
+      "epoch": 162.45,
+      "learning_rate": 7.106220095693781e-05,
+      "loss": 0.0422,
+      "step": 1787
+    },
+    {
+      "epoch": 162.55,
+      "learning_rate": 7.104306220095694e-05,
+      "loss": 0.048,
+      "step": 1788
+    },
+    {
+      "epoch": 162.64,
+      "learning_rate": 7.102392344497609e-05,
+      "loss": 0.0406,
+      "step": 1789
+    },
+    {
+      "epoch": 162.73,
+      "learning_rate": 7.100478468899522e-05,
+      "loss": 0.0486,
+      "step": 1790
+    },
+    {
+      "epoch": 162.82,
+      "learning_rate": 7.098564593301435e-05,
+      "loss": 0.0451,
+      "step": 1791
+    },
+    {
+      "epoch": 162.91,
+      "learning_rate": 7.09665071770335e-05,
+      "loss": 0.0473,
+      "step": 1792
+    },
+    {
+      "epoch": 163.0,
+      "learning_rate": 7.094736842105264e-05,
+      "loss": 0.0637,
+      "step": 1793
+    },
+    {
+      "epoch": 163.09,
+      "learning_rate": 7.092822966507177e-05,
+      "loss": 0.0436,
+      "step": 1794
+    },
+    {
+      "epoch": 163.18,
+      "learning_rate": 7.090909090909092e-05,
+      "loss": 0.0469,
+      "step": 1795
+    },
+    {
+      "epoch": 163.27,
+      "learning_rate": 7.088995215311005e-05,
+      "loss": 0.0549,
+      "step": 1796
+    },
+    {
+      "epoch": 163.36,
+      "learning_rate": 7.087081339712918e-05,
+      "loss": 0.0461,
+      "step": 1797
+    },
+    {
+      "epoch": 163.45,
+      "learning_rate": 7.085167464114833e-05,
+      "loss": 0.0463,
+      "step": 1798
+    },
+    {
+      "epoch": 163.55,
+      "learning_rate": 7.083253588516747e-05,
+      "loss": 0.0431,
+      "step": 1799
+    },
+    {
+      "epoch": 163.64,
+      "learning_rate": 7.08133971291866e-05,
+      "loss": 0.0415,
+      "step": 1800
+    },
+    {
+      "epoch": 163.64,
+      "eval_accuracy_NAT": 0.9937215163785456,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9400765590797888,
+      "eval_accuracy_concrete tiles": 0.9874946492998227,
+      "eval_accuracy_dark bricks": 0.8513914014087217,
+      "eval_accuracy_grass": 0.8667188597671582,
+      "eval_accuracy_red bricks": 0.9185014204545454,
+      "eval_accuracy_road": 0.9773503090133219,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6619799851128939,
+      "eval_accuracy_speedway bricks": 0.9875142856798053,
+      "eval_accuracy_steel": 0.8986626871578391,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9783037125862509,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8812303939750175,
+      "eval_iou_concrete tiles": 0.9224024675843948,
+      "eval_iou_dark bricks": 0.802204975701748,
+      "eval_iou_grass": 0.8114255236998305,
+      "eval_iou_red bricks": 0.8243824701195219,
+      "eval_iou_road": 0.970285999654265,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.43761618370694366,
+      "eval_iou_speedway bricks": 0.9745814490551117,
+      "eval_iou_steel": 0.8338331369819,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08884698152542114,
+      "eval_mean_accuracy": 0.6987239748732648,
+      "eval_mean_iou": 0.6489435625434603,
+      "eval_overall_accuracy": 0.976774533589681,
+      "eval_runtime": 3.6478,
+      "eval_samples_per_second": 9.869,
+      "eval_steps_per_second": 0.548,
+      "step": 1800
+    },
+    {
+      "epoch": 163.73,
+      "learning_rate": 7.079425837320575e-05,
+      "loss": 0.0427,
+      "step": 1801
+    },
+    {
+      "epoch": 163.82,
+      "learning_rate": 7.077511961722488e-05,
+      "loss": 0.076,
+      "step": 1802
+    },
+    {
+      "epoch": 163.91,
+      "learning_rate": 7.075598086124402e-05,
+      "loss": 0.0494,
+      "step": 1803
+    },
+    {
+      "epoch": 164.0,
+      "learning_rate": 7.073684210526316e-05,
+      "loss": 0.0459,
+      "step": 1804
+    },
+    {
+      "epoch": 164.09,
+      "learning_rate": 7.07177033492823e-05,
+      "loss": 0.0475,
+      "step": 1805
+    },
+    {
+      "epoch": 164.18,
+      "learning_rate": 7.069856459330144e-05,
+      "loss": 0.0476,
+      "step": 1806
+    },
+    {
+      "epoch": 164.27,
+      "learning_rate": 7.067942583732058e-05,
+      "loss": 0.0626,
+      "step": 1807
+    },
+    {
+      "epoch": 164.36,
+      "learning_rate": 7.066028708133972e-05,
+      "loss": 0.0395,
+      "step": 1808
+    },
+    {
+      "epoch": 164.45,
+      "learning_rate": 7.064114832535885e-05,
+      "loss": 0.0446,
+      "step": 1809
+    },
+    {
+      "epoch": 164.55,
+      "learning_rate": 7.0622009569378e-05,
+      "loss": 0.0479,
+      "step": 1810
+    },
+    {
+      "epoch": 164.64,
+      "learning_rate": 7.060287081339714e-05,
+      "loss": 0.046,
+      "step": 1811
+    },
+    {
+      "epoch": 164.73,
+      "learning_rate": 7.058373205741627e-05,
+      "loss": 0.0454,
+      "step": 1812
+    },
+    {
+      "epoch": 164.82,
+      "learning_rate": 7.056459330143542e-05,
+      "loss": 0.0437,
+      "step": 1813
+    },
+    {
+      "epoch": 164.91,
+      "learning_rate": 7.054545454545455e-05,
+      "loss": 0.0443,
+      "step": 1814
+    },
+    {
+      "epoch": 165.0,
+      "learning_rate": 7.052631578947368e-05,
+      "loss": 0.0426,
+      "step": 1815
+    },
+    {
+      "epoch": 165.09,
+      "learning_rate": 7.050717703349283e-05,
+      "loss": 0.0443,
+      "step": 1816
+    },
+    {
+      "epoch": 165.18,
+      "learning_rate": 7.048803827751197e-05,
+      "loss": 0.0424,
+      "step": 1817
+    },
+    {
+      "epoch": 165.27,
+      "learning_rate": 7.04688995215311e-05,
+      "loss": 0.0486,
+      "step": 1818
+    },
+    {
+      "epoch": 165.36,
+      "learning_rate": 7.044976076555025e-05,
+      "loss": 0.0478,
+      "step": 1819
+    },
+    {
+      "epoch": 165.45,
+      "learning_rate": 7.043062200956938e-05,
+      "loss": 0.0455,
+      "step": 1820
+    },
+    {
+      "epoch": 165.45,
+      "eval_accuracy_NAT": 0.9929539540224855,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9468861324520366,
+      "eval_accuracy_concrete tiles": 0.9873723475814835,
+      "eval_accuracy_dark bricks": 0.8540471883299334,
+      "eval_accuracy_grass": 0.8556884071900656,
+      "eval_accuracy_red bricks": 0.9225852272727273,
+      "eval_accuracy_road": 0.977800134705866,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6581203650097869,
+      "eval_accuracy_speedway bricks": 0.9880736447158975,
+      "eval_accuracy_steel": 0.887466196161203,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9788367703201306,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.87555298141784,
+      "eval_iou_concrete tiles": 0.934512096307443,
+      "eval_iou_dark bricks": 0.810409057706355,
+      "eval_iou_grass": 0.7997981124494056,
+      "eval_iou_red bricks": 0.8069576021121292,
+      "eval_iou_road": 0.971042651865764,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4316427086158575,
+      "eval_iou_speedway bricks": 0.9748815250523909,
+      "eval_iou_steel": 0.8326745004602805,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09079986810684204,
+      "eval_mean_accuracy": 0.6977687382647297,
+      "eval_mean_iou": 0.6474083081775075,
+      "eval_overall_accuracy": 0.9765053855048286,
+      "eval_runtime": 3.6425,
+      "eval_samples_per_second": 9.883,
+      "eval_steps_per_second": 0.549,
+      "step": 1820
+    },
+    {
+      "epoch": 165.55,
+      "learning_rate": 7.041148325358851e-05,
+      "loss": 0.0457,
+      "step": 1821
+    },
+    {
+      "epoch": 165.64,
+      "learning_rate": 7.039234449760766e-05,
+      "loss": 0.042,
+      "step": 1822
+    },
+    {
+      "epoch": 165.73,
+      "learning_rate": 7.03732057416268e-05,
+      "loss": 0.0404,
+      "step": 1823
+    },
+    {
+      "epoch": 165.82,
+      "learning_rate": 7.035406698564594e-05,
+      "loss": 0.0429,
+      "step": 1824
+    },
+    {
+      "epoch": 165.91,
+      "learning_rate": 7.033492822966508e-05,
+      "loss": 0.0445,
+      "step": 1825
+    },
+    {
+      "epoch": 166.0,
+      "learning_rate": 7.031578947368421e-05,
+      "loss": 0.0472,
+      "step": 1826
+    },
+    {
+      "epoch": 166.09,
+      "learning_rate": 7.029665071770335e-05,
+      "loss": 0.046,
+      "step": 1827
+    },
+    {
+      "epoch": 166.18,
+      "learning_rate": 7.027751196172249e-05,
+      "loss": 0.041,
+      "step": 1828
+    },
+    {
+      "epoch": 166.27,
+      "learning_rate": 7.025837320574164e-05,
+      "loss": 0.0417,
+      "step": 1829
+    },
+    {
+      "epoch": 166.36,
+      "learning_rate": 7.023923444976077e-05,
+      "loss": 0.0456,
+      "step": 1830
+    },
+    {
+      "epoch": 166.45,
+      "learning_rate": 7.022009569377991e-05,
+      "loss": 0.0478,
+      "step": 1831
+    },
+    {
+      "epoch": 166.55,
+      "learning_rate": 7.020095693779905e-05,
+      "loss": 0.0438,
+      "step": 1832
+    },
+    {
+      "epoch": 166.64,
+      "learning_rate": 7.018181818181818e-05,
+      "loss": 0.0463,
+      "step": 1833
+    },
+    {
+      "epoch": 166.73,
+      "learning_rate": 7.016267942583732e-05,
+      "loss": 0.0444,
+      "step": 1834
+    },
+    {
+      "epoch": 166.82,
+      "learning_rate": 7.014354066985647e-05,
+      "loss": 0.0451,
+      "step": 1835
+    },
+    {
+      "epoch": 166.91,
+      "learning_rate": 7.01244019138756e-05,
+      "loss": 0.0551,
+      "step": 1836
+    },
+    {
+      "epoch": 167.0,
+      "learning_rate": 7.010526315789473e-05,
+      "loss": 0.0467,
+      "step": 1837
+    },
+    {
+      "epoch": 167.09,
+      "learning_rate": 7.008612440191388e-05,
+      "loss": 0.0456,
+      "step": 1838
+    },
+    {
+      "epoch": 167.18,
+      "learning_rate": 7.006698564593301e-05,
+      "loss": 0.047,
+      "step": 1839
+    },
+    {
+      "epoch": 167.27,
+      "learning_rate": 7.004784688995216e-05,
+      "loss": 0.0476,
+      "step": 1840
+    },
+    {
+      "epoch": 167.27,
+      "eval_accuracy_NAT": 0.9928449072325901,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9470846388257007,
+      "eval_accuracy_concrete tiles": 0.9843148046230049,
+      "eval_accuracy_dark bricks": 0.8445017512797813,
+      "eval_accuracy_grass": 0.8735971647962197,
+      "eval_accuracy_red bricks": 0.9083806818181818,
+      "eval_accuracy_road": 0.982256069321525,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6508697929589502,
+      "eval_accuracy_speedway bricks": 0.9875607480808615,
+      "eval_accuracy_steel": 0.8919596332695733,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787755820563614,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8840074765231913,
+      "eval_iou_concrete tiles": 0.9495619856649854,
+      "eval_iou_dark bricks": 0.8069214078187635,
+      "eval_iou_grass": 0.8086856928397874,
+      "eval_iou_red bricks": 0.8311941510966694,
+      "eval_iou_road": 0.9743036473132415,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.43159299477167196,
+      "eval_iou_speedway bricks": 0.9753940281037423,
+      "eval_iou_steel": 0.836961735443841,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08697095513343811,
+      "eval_mean_accuracy": 0.6971823224774145,
+      "eval_mean_iou": 0.6521075924332504,
+      "eval_overall_accuracy": 0.9772834777832031,
+      "eval_runtime": 3.4976,
+      "eval_samples_per_second": 10.293,
+      "eval_steps_per_second": 0.572,
+      "step": 1840
+    },
+    {
+      "epoch": 167.36,
+      "learning_rate": 7.00287081339713e-05,
+      "loss": 0.0456,
+      "step": 1841
+    },
+    {
+      "epoch": 167.45,
+      "learning_rate": 7.000956937799043e-05,
+      "loss": 0.04,
+      "step": 1842
+    },
+    {
+      "epoch": 167.55,
+      "learning_rate": 6.999043062200957e-05,
+      "loss": 0.0439,
+      "step": 1843
+    },
+    {
+      "epoch": 167.64,
+      "learning_rate": 6.997129186602871e-05,
+      "loss": 0.0403,
+      "step": 1844
+    },
+    {
+      "epoch": 167.73,
+      "learning_rate": 6.995215311004784e-05,
+      "loss": 0.0422,
+      "step": 1845
+    },
+    {
+      "epoch": 167.82,
+      "learning_rate": 6.993301435406699e-05,
+      "loss": 0.0417,
+      "step": 1846
+    },
+    {
+      "epoch": 167.91,
+      "learning_rate": 6.991387559808614e-05,
+      "loss": 0.0462,
+      "step": 1847
+    },
+    {
+      "epoch": 168.0,
+      "learning_rate": 6.989473684210527e-05,
+      "loss": 0.0461,
+      "step": 1848
+    },
+    {
+      "epoch": 168.09,
+      "learning_rate": 6.98755980861244e-05,
+      "loss": 0.045,
+      "step": 1849
+    },
+    {
+      "epoch": 168.18,
+      "learning_rate": 6.985645933014354e-05,
+      "loss": 0.0456,
+      "step": 1850
+    },
+    {
+      "epoch": 168.27,
+      "learning_rate": 6.983732057416268e-05,
+      "loss": 0.042,
+      "step": 1851
+    },
+    {
+      "epoch": 168.36,
+      "learning_rate": 6.981818181818182e-05,
+      "loss": 0.046,
+      "step": 1852
+    },
+    {
+      "epoch": 168.45,
+      "learning_rate": 6.979904306220097e-05,
+      "loss": 0.046,
+      "step": 1853
+    },
+    {
+      "epoch": 168.55,
+      "learning_rate": 6.97799043062201e-05,
+      "loss": 0.0486,
+      "step": 1854
+    },
+    {
+      "epoch": 168.64,
+      "learning_rate": 6.976076555023923e-05,
+      "loss": 0.0411,
+      "step": 1855
+    },
+    {
+      "epoch": 168.73,
+      "learning_rate": 6.974162679425838e-05,
+      "loss": 0.0429,
+      "step": 1856
+    },
+    {
+      "epoch": 168.82,
+      "learning_rate": 6.972248803827751e-05,
+      "loss": 0.0487,
+      "step": 1857
+    },
+    {
+      "epoch": 168.91,
+      "learning_rate": 6.970334928229666e-05,
+      "loss": 0.0551,
+      "step": 1858
+    },
+    {
+      "epoch": 169.0,
+      "learning_rate": 6.96842105263158e-05,
+      "loss": 0.0396,
+      "step": 1859
+    },
+    {
+      "epoch": 169.09,
+      "learning_rate": 6.966507177033493e-05,
+      "loss": 0.0466,
+      "step": 1860
+    },
+    {
+      "epoch": 169.09,
+      "eval_accuracy_NAT": 0.9934056688803048,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9468257757843684,
+      "eval_accuracy_concrete tiles": 0.9806763285024155,
+      "eval_accuracy_dark bricks": 0.8360340248643239,
+      "eval_accuracy_grass": 0.877917075881545,
+      "eval_accuracy_red bricks": 0.9153053977272727,
+      "eval_accuracy_road": 0.9792935076367585,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6479475091665977,
+      "eval_accuracy_speedway bricks": 0.984950405912431,
+      "eval_accuracy_steel": 0.8996520678055537,
+      "eval_accuracy_tiles": 0.046637446925078756,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9783656473735897,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8865817665749847,
+      "eval_iou_concrete tiles": 0.9391543687046147,
+      "eval_iou_dark bricks": 0.7956118823486319,
+      "eval_iou_grass": 0.8096642555989634,
+      "eval_iou_red bricks": 0.8157936382338977,
+      "eval_iou_road": 0.9725280399644547,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.430064043915828,
+      "eval_iou_speedway bricks": 0.9750088402964605,
+      "eval_iou_steel": 0.835300692020332,
+      "eval_iou_tiles": 0.046637446925078756,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08733034133911133,
+      "eval_mean_accuracy": 0.7006650160835886,
+      "eval_mean_iou": 0.6526700478428336,
+      "eval_overall_accuracy": 0.977140744527181,
+      "eval_runtime": 4.0027,
+      "eval_samples_per_second": 8.994,
+      "eval_steps_per_second": 0.5,
+      "step": 1860
+    },
+    {
+      "epoch": 169.18,
+      "learning_rate": 6.964593301435406e-05,
+      "loss": 0.0405,
+      "step": 1861
+    },
+    {
+      "epoch": 169.27,
+      "learning_rate": 6.962679425837321e-05,
+      "loss": 0.0414,
+      "step": 1862
+    },
+    {
+      "epoch": 169.36,
+      "learning_rate": 6.960765550239234e-05,
+      "loss": 0.0493,
+      "step": 1863
+    },
+    {
+      "epoch": 169.45,
+      "learning_rate": 6.958851674641149e-05,
+      "loss": 0.0427,
+      "step": 1864
+    },
+    {
+      "epoch": 169.55,
+      "learning_rate": 6.956937799043063e-05,
+      "loss": 0.0397,
+      "step": 1865
+    },
+    {
+      "epoch": 169.64,
+      "learning_rate": 6.955023923444977e-05,
+      "loss": 0.0479,
+      "step": 1866
+    },
+    {
+      "epoch": 169.73,
+      "learning_rate": 6.95311004784689e-05,
+      "loss": 0.0503,
+      "step": 1867
+    },
+    {
+      "epoch": 169.82,
+      "learning_rate": 6.951196172248804e-05,
+      "loss": 0.0421,
+      "step": 1868
+    },
+    {
+      "epoch": 169.91,
+      "learning_rate": 6.949282296650717e-05,
+      "loss": 0.0415,
+      "step": 1869
+    },
+    {
+      "epoch": 170.0,
+      "learning_rate": 6.947368421052632e-05,
+      "loss": 0.0437,
+      "step": 1870
+    },
+    {
+      "epoch": 170.09,
+      "learning_rate": 6.945454545454547e-05,
+      "loss": 0.0469,
+      "step": 1871
+    },
+    {
+      "epoch": 170.18,
+      "learning_rate": 6.94354066985646e-05,
+      "loss": 0.0427,
+      "step": 1872
+    },
+    {
+      "epoch": 170.27,
+      "learning_rate": 6.941626794258373e-05,
+      "loss": 0.0502,
+      "step": 1873
+    },
+    {
+      "epoch": 170.36,
+      "learning_rate": 6.939712918660288e-05,
+      "loss": 0.0404,
+      "step": 1874
+    },
+    {
+      "epoch": 170.45,
+      "learning_rate": 6.937799043062201e-05,
+      "loss": 0.0417,
+      "step": 1875
+    },
+    {
+      "epoch": 170.55,
+      "learning_rate": 6.935885167464115e-05,
+      "loss": 0.0376,
+      "step": 1876
+    },
+    {
+      "epoch": 170.64,
+      "learning_rate": 6.93397129186603e-05,
+      "loss": 0.043,
+      "step": 1877
+    },
+    {
+      "epoch": 170.73,
+      "learning_rate": 6.932057416267943e-05,
+      "loss": 0.0457,
+      "step": 1878
+    },
+    {
+      "epoch": 170.82,
+      "learning_rate": 6.930143540669856e-05,
+      "loss": 0.0384,
+      "step": 1879
+    },
+    {
+      "epoch": 170.91,
+      "learning_rate": 6.928229665071771e-05,
+      "loss": 0.0458,
+      "step": 1880
+    },
+    {
+      "epoch": 170.91,
+      "eval_accuracy_NAT": 0.9925133555849954,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.945095551311215,
+      "eval_accuracy_concrete tiles": 0.9849874640738703,
+      "eval_accuracy_dark bricks": 0.8558946922751242,
+      "eval_accuracy_grass": 0.8768790389945372,
+      "eval_accuracy_red bricks": 0.8971946022727273,
+      "eval_accuracy_road": 0.9753091644729602,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6460176991150443,
+      "eval_accuracy_speedway bricks": 0.9885714561557855,
+      "eval_accuracy_steel": 0.8943753710177429,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9790650771252278,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8766185457223615,
+      "eval_iou_concrete tiles": 0.9401721873631986,
+      "eval_iou_dark bricks": 0.8086475871849885,
+      "eval_iou_grass": 0.8070030975486585,
+      "eval_iou_red bricks": 0.8241722394389169,
+      "eval_iou_road": 0.9697415643203665,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4268772543447371,
+      "eval_iou_speedway bricks": 0.9745117073646592,
+      "eval_iou_steel": 0.8339573323082837,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08924874663352966,
+      "eval_mean_accuracy": 0.6966798765595387,
+      "eval_mean_iou": 0.6492897379016459,
+      "eval_overall_accuracy": 0.9766348732842339,
+      "eval_runtime": 3.8643,
+      "eval_samples_per_second": 9.316,
+      "eval_steps_per_second": 0.518,
+      "step": 1880
+    },
+    {
+      "epoch": 171.0,
+      "learning_rate": 6.926315789473684e-05,
+      "loss": 0.05,
+      "step": 1881
+    },
+    {
+      "epoch": 171.09,
+      "learning_rate": 6.924401913875599e-05,
+      "loss": 0.0424,
+      "step": 1882
+    },
+    {
+      "epoch": 171.18,
+      "learning_rate": 6.922488038277513e-05,
+      "loss": 0.0483,
+      "step": 1883
+    },
+    {
+      "epoch": 171.27,
+      "learning_rate": 6.920574162679426e-05,
+      "loss": 0.0454,
+      "step": 1884
+    },
+    {
+      "epoch": 171.36,
+      "learning_rate": 6.91866028708134e-05,
+      "loss": 0.0375,
+      "step": 1885
+    },
+    {
+      "epoch": 171.45,
+      "learning_rate": 6.916746411483254e-05,
+      "loss": 0.0451,
+      "step": 1886
+    },
+    {
+      "epoch": 171.55,
+      "learning_rate": 6.914832535885167e-05,
+      "loss": 0.044,
+      "step": 1887
+    },
+    {
+      "epoch": 171.64,
+      "learning_rate": 6.912918660287082e-05,
+      "loss": 0.0427,
+      "step": 1888
+    },
+    {
+      "epoch": 171.73,
+      "learning_rate": 6.911004784688996e-05,
+      "loss": 0.0441,
+      "step": 1889
+    },
+    {
+      "epoch": 171.82,
+      "learning_rate": 6.90909090909091e-05,
+      "loss": 0.0411,
+      "step": 1890
+    },
+    {
+      "epoch": 171.91,
+      "learning_rate": 6.907177033492823e-05,
+      "loss": 0.0465,
+      "step": 1891
+    },
+    {
+      "epoch": 172.0,
+      "learning_rate": 6.905263157894737e-05,
+      "loss": 0.0465,
+      "step": 1892
+    },
+    {
+      "epoch": 172.09,
+      "learning_rate": 6.90334928229665e-05,
+      "loss": 0.0374,
+      "step": 1893
+    },
+    {
+      "epoch": 172.18,
+      "learning_rate": 6.901435406698565e-05,
+      "loss": 0.0462,
+      "step": 1894
+    },
+    {
+      "epoch": 172.27,
+      "learning_rate": 6.89952153110048e-05,
+      "loss": 0.0442,
+      "step": 1895
+    },
+    {
+      "epoch": 172.36,
+      "learning_rate": 6.897607655502393e-05,
+      "loss": 0.0434,
+      "step": 1896
+    },
+    {
+      "epoch": 172.45,
+      "learning_rate": 6.895693779904306e-05,
+      "loss": 0.0565,
+      "step": 1897
+    },
+    {
+      "epoch": 172.55,
+      "learning_rate": 6.89377990430622e-05,
+      "loss": 0.0445,
+      "step": 1898
+    },
+    {
+      "epoch": 172.64,
+      "learning_rate": 6.891866028708134e-05,
+      "loss": 0.04,
+      "step": 1899
+    },
+    {
+      "epoch": 172.73,
+      "learning_rate": 6.889952153110048e-05,
+      "loss": 0.0438,
+      "step": 1900
+    },
+    {
+      "epoch": 172.73,
+      "eval_accuracy_NAT": 0.9921009893261191,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.951689181939139,
+      "eval_accuracy_concrete tiles": 0.9901547116736991,
+      "eval_accuracy_dark bricks": 0.8628228320695893,
+      "eval_accuracy_grass": 0.8791822925586386,
+      "eval_accuracy_red bricks": 0.9245383522727273,
+      "eval_accuracy_road": 0.9778400385979465,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6517244231246382,
+      "eval_accuracy_speedway bricks": 0.9870502650770492,
+      "eval_accuracy_steel": 0.8968240881208364,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9792697093990299,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8781677450655208,
+      "eval_iou_concrete tiles": 0.9245439232592001,
+      "eval_iou_dark bricks": 0.8122690050003624,
+      "eval_iou_grass": 0.8131139592390865,
+      "eval_iou_red bricks": 0.8185819839647854,
+      "eval_iou_road": 0.9723889310420707,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4300058207217695,
+      "eval_iou_speedway bricks": 0.9752013544852419,
+      "eval_iou_steel": 0.83734142135731,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08844412863254547,
+      "eval_mean_accuracy": 0.7010713211354141,
+      "eval_mean_iou": 0.6492987579641828,
+      "eval_overall_accuracy": 0.9770393371582031,
+      "eval_runtime": 3.5585,
+      "eval_samples_per_second": 10.116,
+      "eval_steps_per_second": 0.562,
+      "step": 1900
+    },
+    {
+      "epoch": 172.82,
+      "learning_rate": 6.888038277511963e-05,
+      "loss": 0.052,
+      "step": 1901
+    },
+    {
+      "epoch": 172.91,
+      "learning_rate": 6.886124401913876e-05,
+      "loss": 0.0446,
+      "step": 1902
+    },
+    {
+      "epoch": 173.0,
+      "learning_rate": 6.88421052631579e-05,
+      "loss": 0.0401,
+      "step": 1903
+    },
+    {
+      "epoch": 173.09,
+      "learning_rate": 6.882296650717704e-05,
+      "loss": 0.0445,
+      "step": 1904
+    },
+    {
+      "epoch": 173.18,
+      "learning_rate": 6.880382775119617e-05,
+      "loss": 0.0441,
+      "step": 1905
+    },
+    {
+      "epoch": 173.27,
+      "learning_rate": 6.878468899521532e-05,
+      "loss": 0.0437,
+      "step": 1906
+    },
+    {
+      "epoch": 173.36,
+      "learning_rate": 6.876555023923446e-05,
+      "loss": 0.0413,
+      "step": 1907
+    },
+    {
+      "epoch": 173.45,
+      "learning_rate": 6.87464114832536e-05,
+      "loss": 0.0449,
+      "step": 1908
+    },
+    {
+      "epoch": 173.55,
+      "learning_rate": 6.872727272727273e-05,
+      "loss": 0.0406,
+      "step": 1909
+    },
+    {
+      "epoch": 173.64,
+      "learning_rate": 6.870813397129187e-05,
+      "loss": 0.0433,
+      "step": 1910
+    },
+    {
+      "epoch": 173.73,
+      "learning_rate": 6.8688995215311e-05,
+      "loss": 0.047,
+      "step": 1911
+    },
+    {
+      "epoch": 173.82,
+      "learning_rate": 6.866985645933015e-05,
+      "loss": 0.0388,
+      "step": 1912
+    },
+    {
+      "epoch": 173.91,
+      "learning_rate": 6.865071770334928e-05,
+      "loss": 0.0455,
+      "step": 1913
+    },
+    {
+      "epoch": 174.0,
+      "learning_rate": 6.863157894736843e-05,
+      "loss": 0.0522,
+      "step": 1914
+    },
+    {
+      "epoch": 174.09,
+      "learning_rate": 6.861244019138756e-05,
+      "loss": 0.0459,
+      "step": 1915
+    },
+    {
+      "epoch": 174.18,
+      "learning_rate": 6.85933014354067e-05,
+      "loss": 0.046,
+      "step": 1916
+    },
+    {
+      "epoch": 174.27,
+      "learning_rate": 6.857416267942584e-05,
+      "loss": 0.0477,
+      "step": 1917
+    },
+    {
+      "epoch": 174.36,
+      "learning_rate": 6.855502392344498e-05,
+      "loss": 0.0381,
+      "step": 1918
+    },
+    {
+      "epoch": 174.45,
+      "learning_rate": 6.853588516746411e-05,
+      "loss": 0.0549,
+      "step": 1919
+    },
+    {
+      "epoch": 174.55,
+      "learning_rate": 6.851674641148326e-05,
+      "loss": 0.0412,
+      "step": 1920
+    },
+    {
+      "epoch": 174.55,
+      "eval_accuracy_NAT": 0.9930074187107352,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9542711060560539,
+      "eval_accuracy_concrete tiles": 0.9813795633828656,
+      "eval_accuracy_dark bricks": 0.8560486509372234,
+      "eval_accuracy_grass": 0.8827822184630763,
+      "eval_accuracy_red bricks": 0.9122869318181818,
+      "eval_accuracy_road": 0.9818945158750984,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6328674220494582,
+      "eval_accuracy_speedway bricks": 0.983234314112381,
+      "eval_accuracy_steel": 0.9009629971637755,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9790841326853269,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8850434016763571,
+      "eval_iou_concrete tiles": 0.9468979555712895,
+      "eval_iou_dark bricks": 0.8093228048469852,
+      "eval_iou_grass": 0.8212322108459785,
+      "eval_iou_red bricks": 0.8261778420968001,
+      "eval_iou_road": 0.9744347295990247,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.42128057844414674,
+      "eval_iou_speedway bricks": 0.974516171475049,
+      "eval_iou_steel": 0.8369983991666475,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08633160591125488,
+      "eval_mean_accuracy": 0.6983642414283732,
+      "eval_mean_iou": 0.6519221712621235,
+      "eval_overall_accuracy": 0.9775063196818033,
+      "eval_runtime": 3.6753,
+      "eval_samples_per_second": 9.795,
+      "eval_steps_per_second": 0.544,
+      "step": 1920
+    },
+    {
+      "epoch": 174.64,
+      "learning_rate": 6.849760765550239e-05,
+      "loss": 0.0463,
+      "step": 1921
+    },
+    {
+      "epoch": 174.73,
+      "learning_rate": 6.847846889952154e-05,
+      "loss": 0.0456,
+      "step": 1922
+    },
+    {
+      "epoch": 174.82,
+      "learning_rate": 6.845933014354068e-05,
+      "loss": 0.0411,
+      "step": 1923
+    },
+    {
+      "epoch": 174.91,
+      "learning_rate": 6.844019138755981e-05,
+      "loss": 0.0392,
+      "step": 1924
+    },
+    {
+      "epoch": 175.0,
+      "learning_rate": 6.842105263157895e-05,
+      "loss": 0.0446,
+      "step": 1925
+    },
+    {
+      "epoch": 175.09,
+      "learning_rate": 6.840191387559809e-05,
+      "loss": 0.0569,
+      "step": 1926
+    },
+    {
+      "epoch": 175.18,
+      "learning_rate": 6.838277511961722e-05,
+      "loss": 0.0391,
+      "step": 1927
+    },
+    {
+      "epoch": 175.27,
+      "learning_rate": 6.836363636363637e-05,
+      "loss": 0.0427,
+      "step": 1928
+    },
+    {
+      "epoch": 175.36,
+      "learning_rate": 6.834449760765552e-05,
+      "loss": 0.0452,
+      "step": 1929
+    },
+    {
+      "epoch": 175.45,
+      "learning_rate": 6.832535885167465e-05,
+      "loss": 0.0502,
+      "step": 1930
+    },
+    {
+      "epoch": 175.55,
+      "learning_rate": 6.830622009569378e-05,
+      "loss": 0.0382,
+      "step": 1931
+    },
+    {
+      "epoch": 175.64,
+      "learning_rate": 6.828708133971292e-05,
+      "loss": 0.0389,
+      "step": 1932
+    },
+    {
+      "epoch": 175.73,
+      "learning_rate": 6.826794258373206e-05,
+      "loss": 0.0422,
+      "step": 1933
+    },
+    {
+      "epoch": 175.82,
+      "learning_rate": 6.82488038277512e-05,
+      "loss": 0.053,
+      "step": 1934
+    },
+    {
+      "epoch": 175.91,
+      "learning_rate": 6.822966507177035e-05,
+      "loss": 0.0459,
+      "step": 1935
+    },
+    {
+      "epoch": 176.0,
+      "learning_rate": 6.821052631578948e-05,
+      "loss": 0.0403,
+      "step": 1936
+    },
+    {
+      "epoch": 176.09,
+      "learning_rate": 6.819138755980861e-05,
+      "loss": 0.0391,
+      "step": 1937
+    },
+    {
+      "epoch": 176.18,
+      "learning_rate": 6.817224880382776e-05,
+      "loss": 0.0503,
+      "step": 1938
+    },
+    {
+      "epoch": 176.27,
+      "learning_rate": 6.815311004784689e-05,
+      "loss": 0.0452,
+      "step": 1939
+    },
+    {
+      "epoch": 176.36,
+      "learning_rate": 6.813397129186602e-05,
+      "loss": 0.0413,
+      "step": 1940
+    },
+    {
+      "epoch": 176.36,
+      "eval_accuracy_NAT": 0.9923328460929841,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9491072578222242,
+      "eval_accuracy_concrete tiles": 0.9807069039320002,
+      "eval_accuracy_dark bricks": 0.8582040722066125,
+      "eval_accuracy_grass": 0.8947982818217722,
+      "eval_accuracy_red bricks": 0.9328835227272727,
+      "eval_accuracy_road": 0.9834918807671704,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6290629393763957,
+      "eval_accuracy_speedway bricks": 0.9870327662506774,
+      "eval_accuracy_steel": 0.8848278477672977,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9792082952306463,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8865779951688401,
+      "eval_iou_concrete tiles": 0.953138000713182,
+      "eval_iou_dark bricks": 0.8170392085012825,
+      "eval_iou_grass": 0.8243588298745834,
+      "eval_iou_red bricks": 0.8304093567251462,
+      "eval_iou_road": 0.9722130911412045,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4238900241501022,
+      "eval_iou_speedway bricks": 0.975141046246,
+      "eval_iou_steel": 0.8330538866377905,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08655188232660294,
+      "eval_mean_accuracy": 0.699419101443416,
+      "eval_mean_iou": 0.6534638257222137,
+      "eval_overall_accuracy": 0.9776584837171767,
+      "eval_runtime": 3.5,
+      "eval_samples_per_second": 10.286,
+      "eval_steps_per_second": 0.571,
+      "step": 1940
+    },
+    {
+      "epoch": 176.45,
+      "learning_rate": 6.811483253588518e-05,
+      "loss": 0.0429,
+      "step": 1941
+    },
+    {
+      "epoch": 176.55,
+      "learning_rate": 6.809569377990431e-05,
+      "loss": 0.0467,
+      "step": 1942
+    },
+    {
+      "epoch": 176.64,
+      "learning_rate": 6.807655502392344e-05,
+      "loss": 0.0428,
+      "step": 1943
+    },
+    {
+      "epoch": 176.73,
+      "learning_rate": 6.805741626794259e-05,
+      "loss": 0.0485,
+      "step": 1944
+    },
+    {
+      "epoch": 176.82,
+      "learning_rate": 6.803827751196172e-05,
+      "loss": 0.0417,
+      "step": 1945
+    },
+    {
+      "epoch": 176.91,
+      "learning_rate": 6.801913875598085e-05,
+      "loss": 0.0392,
+      "step": 1946
+    },
+    {
+      "epoch": 177.0,
+      "learning_rate": 6.800000000000001e-05,
+      "loss": 0.0391,
+      "step": 1947
+    },
+    {
+      "epoch": 177.09,
+      "learning_rate": 6.798086124401915e-05,
+      "loss": 0.0409,
+      "step": 1948
+    },
+    {
+      "epoch": 177.18,
+      "learning_rate": 6.796172248803828e-05,
+      "loss": 0.049,
+      "step": 1949
+    },
+    {
+      "epoch": 177.27,
+      "learning_rate": 6.794258373205742e-05,
+      "loss": 0.0481,
+      "step": 1950
+    },
+    {
+      "epoch": 177.36,
+      "learning_rate": 6.792344497607655e-05,
+      "loss": 0.0407,
+      "step": 1951
+    },
+    {
+      "epoch": 177.45,
+      "learning_rate": 6.790430622009569e-05,
+      "loss": 0.0444,
+      "step": 1952
+    },
+    {
+      "epoch": 177.55,
+      "learning_rate": 6.788516746411485e-05,
+      "loss": 0.0549,
+      "step": 1953
+    },
+    {
+      "epoch": 177.64,
+      "learning_rate": 6.786602870813398e-05,
+      "loss": 0.0385,
+      "step": 1954
+    },
+    {
+      "epoch": 177.73,
+      "learning_rate": 6.784688995215311e-05,
+      "loss": 0.0376,
+      "step": 1955
+    },
+    {
+      "epoch": 177.82,
+      "learning_rate": 6.782775119617226e-05,
+      "loss": 0.0426,
+      "step": 1956
+    },
+    {
+      "epoch": 177.91,
+      "learning_rate": 6.780861244019139e-05,
+      "loss": 0.0456,
+      "step": 1957
+    },
+    {
+      "epoch": 178.0,
+      "learning_rate": 6.778947368421052e-05,
+      "loss": 0.0494,
+      "step": 1958
+    },
+    {
+      "epoch": 178.09,
+      "learning_rate": 6.777033492822968e-05,
+      "loss": 0.0416,
+      "step": 1959
+    },
+    {
+      "epoch": 178.18,
+      "learning_rate": 6.775119617224881e-05,
+      "loss": 0.0411,
+      "step": 1960
+    },
+    {
+      "epoch": 178.18,
+      "eval_accuracy_NAT": 0.9917256778016732,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9461712412549895,
+      "eval_accuracy_concrete tiles": 0.9907662202653947,
+      "eval_accuracy_dark bricks": 0.870636234171125,
+      "eval_accuracy_grass": 0.8964549467525523,
+      "eval_accuracy_red bricks": 0.9240056818181818,
+      "eval_accuracy_road": 0.9804229087035226,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6980122956468999,
+      "eval_accuracy_speedway bricks": 0.9901976401927526,
+      "eval_accuracy_steel": 0.8767066816173076,
+      "eval_accuracy_tiles": 0.025544445966305986,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9793471737907663,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.884370584791436,
+      "eval_iou_concrete tiles": 0.9282685917268247,
+      "eval_iou_dark bricks": 0.8100848762668768,
+      "eval_iou_grass": 0.8265930602418957,
+      "eval_iou_red bricks": 0.814142678347935,
+      "eval_iou_road": 0.9734859763711459,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.45387566327262296,
+      "eval_iou_speedway bricks": 0.9747510567170137,
+      "eval_iou_steel": 0.8333254964655726,
+      "eval_iou_tiles": 0.025544445966305986,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08562561869621277,
+      "eval_mean_accuracy": 0.706972613399285,
+      "eval_mean_iou": 0.6541376618429534,
+      "eval_overall_accuracy": 0.9776629341973199,
+      "eval_runtime": 3.4082,
+      "eval_samples_per_second": 10.563,
+      "eval_steps_per_second": 0.587,
+      "step": 1960
+    },
+    {
+      "epoch": 178.27,
+      "learning_rate": 6.773205741626794e-05,
+      "loss": 0.0455,
+      "step": 1961
+    },
+    {
+      "epoch": 178.36,
+      "learning_rate": 6.771291866028709e-05,
+      "loss": 0.0412,
+      "step": 1962
+    },
+    {
+      "epoch": 178.45,
+      "learning_rate": 6.769377990430622e-05,
+      "loss": 0.0375,
+      "step": 1963
+    },
+    {
+      "epoch": 178.55,
+      "learning_rate": 6.767464114832535e-05,
+      "loss": 0.0471,
+      "step": 1964
+    },
+    {
+      "epoch": 178.64,
+      "learning_rate": 6.765550239234451e-05,
+      "loss": 0.0462,
+      "step": 1965
+    },
+    {
+      "epoch": 178.73,
+      "learning_rate": 6.763636363636364e-05,
+      "loss": 0.039,
+      "step": 1966
+    },
+    {
+      "epoch": 178.82,
+      "learning_rate": 6.761722488038278e-05,
+      "loss": 0.0509,
+      "step": 1967
+    },
+    {
+      "epoch": 178.91,
+      "learning_rate": 6.759808612440192e-05,
+      "loss": 0.0468,
+      "step": 1968
+    },
+    {
+      "epoch": 179.0,
+      "learning_rate": 6.757894736842105e-05,
+      "loss": 0.0432,
+      "step": 1969
+    },
+    {
+      "epoch": 179.09,
+      "learning_rate": 6.755980861244018e-05,
+      "loss": 0.0423,
+      "step": 1970
+    },
+    {
+      "epoch": 179.18,
+      "learning_rate": 6.754066985645934e-05,
+      "loss": 0.0418,
+      "step": 1971
+    },
+    {
+      "epoch": 179.27,
+      "learning_rate": 6.752153110047848e-05,
+      "loss": 0.0431,
+      "step": 1972
+    },
+    {
+      "epoch": 179.36,
+      "learning_rate": 6.750239234449761e-05,
+      "loss": 0.0408,
+      "step": 1973
+    },
+    {
+      "epoch": 179.45,
+      "learning_rate": 6.748325358851675e-05,
+      "loss": 0.0475,
+      "step": 1974
+    },
+    {
+      "epoch": 179.55,
+      "learning_rate": 6.746411483253589e-05,
+      "loss": 0.051,
+      "step": 1975
+    },
+    {
+      "epoch": 179.64,
+      "learning_rate": 6.744497607655502e-05,
+      "loss": 0.0532,
+      "step": 1976
+    },
+    {
+      "epoch": 179.73,
+      "learning_rate": 6.742583732057418e-05,
+      "loss": 0.0372,
+      "step": 1977
+    },
+    {
+      "epoch": 179.82,
+      "learning_rate": 6.740669856459331e-05,
+      "loss": 0.0409,
+      "step": 1978
+    },
+    {
+      "epoch": 179.91,
+      "learning_rate": 6.738755980861244e-05,
+      "loss": 0.0517,
+      "step": 1979
+    },
+    {
+      "epoch": 180.0,
+      "learning_rate": 6.736842105263159e-05,
+      "loss": 0.044,
+      "step": 1980
+    },
+    {
+      "epoch": 180.0,
+      "eval_accuracy_NAT": 0.9917359119664206,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9501936778402507,
+      "eval_accuracy_concrete tiles": 0.9870360178560509,
+      "eval_accuracy_dark bricks": 0.8804126092144259,
+      "eval_accuracy_grass": 0.894885658664113,
+      "eval_accuracy_red bricks": 0.9151278409090909,
+      "eval_accuracy_road": 0.9800734473456052,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6918644721969509,
+      "eval_accuracy_speedway bricks": 0.9877550453943692,
+      "eval_accuracy_steel": 0.8918359606886089,
+      "eval_accuracy_tiles": 0.0839611012190111,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9794926257347558,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8861860926497869,
+      "eval_iou_concrete tiles": 0.9393586684513764,
+      "eval_iou_dark bricks": 0.8058481592390347,
+      "eval_iou_grass": 0.8243570425890868,
+      "eval_iou_red bricks": 0.7932892104048023,
+      "eval_iou_road": 0.9729649267912641,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4498619725379127,
+      "eval_iou_speedway bricks": 0.974855048642674,
+      "eval_iou_steel": 0.8340838641025246,
+      "eval_iou_tiles": 0.0839611012190111,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08479425311088562,
+      "eval_mean_accuracy": 0.7119139802534536,
+      "eval_mean_iou": 0.65725067018171,
+      "eval_overall_accuracy": 0.9777496125962999,
+      "eval_runtime": 3.4671,
+      "eval_samples_per_second": 10.383,
+      "eval_steps_per_second": 0.577,
+      "step": 1980
+    },
+    {
+      "epoch": 180.09,
+      "learning_rate": 6.734928229665072e-05,
+      "loss": 0.0377,
+      "step": 1981
+    },
+    {
+      "epoch": 180.18,
+      "learning_rate": 6.733014354066985e-05,
+      "loss": 0.041,
+      "step": 1982
+    },
+    {
+      "epoch": 180.27,
+      "learning_rate": 6.731100478468901e-05,
+      "loss": 0.0441,
+      "step": 1983
+    },
+    {
+      "epoch": 180.36,
+      "learning_rate": 6.729186602870814e-05,
+      "loss": 0.0519,
+      "step": 1984
+    },
+    {
+      "epoch": 180.45,
+      "learning_rate": 6.727272727272727e-05,
+      "loss": 0.0417,
+      "step": 1985
+    },
+    {
+      "epoch": 180.55,
+      "learning_rate": 6.725358851674642e-05,
+      "loss": 0.0391,
+      "step": 1986
+    },
+    {
+      "epoch": 180.64,
+      "learning_rate": 6.723444976076555e-05,
+      "loss": 0.0444,
+      "step": 1987
+    },
+    {
+      "epoch": 180.73,
+      "learning_rate": 6.721531100478468e-05,
+      "loss": 0.0429,
+      "step": 1988
+    },
+    {
+      "epoch": 180.82,
+      "learning_rate": 6.719617224880383e-05,
+      "loss": 0.0429,
+      "step": 1989
+    },
+    {
+      "epoch": 180.91,
+      "learning_rate": 6.717703349282297e-05,
+      "loss": 0.0427,
+      "step": 1990
+    },
+    {
+      "epoch": 181.0,
+      "learning_rate": 6.71578947368421e-05,
+      "loss": 0.043,
+      "step": 1991
+    },
+    {
+      "epoch": 181.09,
+      "learning_rate": 6.713875598086125e-05,
+      "loss": 0.0393,
+      "step": 1992
+    },
+    {
+      "epoch": 181.18,
+      "learning_rate": 6.711961722488038e-05,
+      "loss": 0.0503,
+      "step": 1993
+    },
+    {
+      "epoch": 181.27,
+      "learning_rate": 6.710047846889953e-05,
+      "loss": 0.0417,
+      "step": 1994
+    },
+    {
+      "epoch": 181.36,
+      "learning_rate": 6.708133971291866e-05,
+      "loss": 0.0357,
+      "step": 1995
+    },
+    {
+      "epoch": 181.45,
+      "learning_rate": 6.70622009569378e-05,
+      "loss": 0.05,
+      "step": 1996
+    },
+    {
+      "epoch": 181.55,
+      "learning_rate": 6.704306220095694e-05,
+      "loss": 0.0434,
+      "step": 1997
+    },
+    {
+      "epoch": 181.64,
+      "learning_rate": 6.702392344497608e-05,
+      "loss": 0.0442,
+      "step": 1998
+    },
+    {
+      "epoch": 181.73,
+      "learning_rate": 6.700478468899522e-05,
+      "loss": 0.0418,
+      "step": 1999
+    },
+    {
+      "epoch": 181.82,
+      "learning_rate": 6.698564593301436e-05,
+      "loss": 0.0437,
+      "step": 2000
+    },
+    {
+      "epoch": 181.82,
+      "eval_accuracy_NAT": 0.9933733783260154,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9471517017897764,
+      "eval_accuracy_concrete tiles": 0.9842536537638353,
+      "eval_accuracy_dark bricks": 0.8430391439898387,
+      "eval_accuracy_grass": 0.866795751388418,
+      "eval_accuracy_red bricks": 0.9181463068181818,
+      "eval_accuracy_road": 0.9801447906678098,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6367270421525653,
+      "eval_accuracy_speedway bricks": 0.9871944795426651,
+      "eval_accuracy_steel": 0.9018616845854495,
+      "eval_accuracy_tiles": 6.848376934666484e-05,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.979067157579612,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.882548641244576,
+      "eval_iou_concrete tiles": 0.9438238484768523,
+      "eval_iou_dark bricks": 0.7931558935361217,
+      "eval_iou_grass": 0.8120103070843196,
+      "eval_iou_red bricks": 0.7565471836137527,
+      "eval_iou_road": 0.9726376039454264,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4277591539644028,
+      "eval_iou_speedway bricks": 0.9751601293671239,
+      "eval_iou_steel": 0.836698945951321,
+      "eval_iou_tiles": 6.848376934666484e-05,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09004754573106766,
+      "eval_mean_accuracy": 0.6968274166764541,
+      "eval_mean_iou": 0.6445751806563735,
+      "eval_overall_accuracy": 0.9772251976860894,
+      "eval_runtime": 3.5076,
+      "eval_samples_per_second": 10.263,
+      "eval_steps_per_second": 0.57,
+      "step": 2000
+    },
+    {
+      "epoch": 181.91,
+      "learning_rate": 6.69665071770335e-05,
+      "loss": 0.0501,
+      "step": 2001
+    },
+    {
+      "epoch": 182.0,
+      "learning_rate": 6.694736842105264e-05,
+      "loss": 0.0349,
+      "step": 2002
+    },
+    {
+      "epoch": 182.09,
+      "learning_rate": 6.692822966507177e-05,
+      "loss": 0.0433,
+      "step": 2003
+    },
+    {
+      "epoch": 182.18,
+      "learning_rate": 6.690909090909092e-05,
+      "loss": 0.0432,
+      "step": 2004
+    },
+    {
+      "epoch": 182.27,
+      "learning_rate": 6.688995215311005e-05,
+      "loss": 0.0438,
+      "step": 2005
+    },
+    {
+      "epoch": 182.36,
+      "learning_rate": 6.68708133971292e-05,
+      "loss": 0.0415,
+      "step": 2006
+    },
+    {
+      "epoch": 182.45,
+      "learning_rate": 6.685167464114833e-05,
+      "loss": 0.0393,
+      "step": 2007
+    },
+    {
+      "epoch": 182.55,
+      "learning_rate": 6.683253588516747e-05,
+      "loss": 0.0401,
+      "step": 2008
+    },
+    {
+      "epoch": 182.64,
+      "learning_rate": 6.68133971291866e-05,
+      "loss": 0.0415,
+      "step": 2009
+    },
+    {
+      "epoch": 182.73,
+      "learning_rate": 6.679425837320575e-05,
+      "loss": 0.0411,
+      "step": 2010
+    },
+    {
+      "epoch": 182.82,
+      "learning_rate": 6.677511961722488e-05,
+      "loss": 0.0412,
+      "step": 2011
+    },
+    {
+      "epoch": 182.91,
+      "learning_rate": 6.675598086124403e-05,
+      "loss": 0.0437,
+      "step": 2012
+    },
+    {
+      "epoch": 183.0,
+      "learning_rate": 6.673684210526316e-05,
+      "loss": 0.0433,
+      "step": 2013
+    },
+    {
+      "epoch": 183.09,
+      "learning_rate": 6.67177033492823e-05,
+      "loss": 0.0391,
+      "step": 2014
+    },
+    {
+      "epoch": 183.18,
+      "learning_rate": 6.669856459330144e-05,
+      "loss": 0.0478,
+      "step": 2015
+    },
+    {
+      "epoch": 183.27,
+      "learning_rate": 6.667942583732057e-05,
+      "loss": 0.0397,
+      "step": 2016
+    },
+    {
+      "epoch": 183.36,
+      "learning_rate": 6.666028708133971e-05,
+      "loss": 0.0399,
+      "step": 2017
+    },
+    {
+      "epoch": 183.45,
+      "learning_rate": 6.664114832535886e-05,
+      "loss": 0.0464,
+      "step": 2018
+    },
+    {
+      "epoch": 183.55,
+      "learning_rate": 6.662200956937799e-05,
+      "loss": 0.0464,
+      "step": 2019
+    },
+    {
+      "epoch": 183.64,
+      "learning_rate": 6.660287081339714e-05,
+      "loss": 0.0384,
+      "step": 2020
+    },
+    {
+      "epoch": 183.64,
+      "eval_accuracy_NAT": 0.9923030258543234,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9458627516202412,
+      "eval_accuracy_concrete tiles": 0.9899101082370207,
+      "eval_accuracy_dark bricks": 0.877217966975867,
+      "eval_accuracy_grass": 0.8865568980521954,
+      "eval_accuracy_red bricks": 0.919921875,
+      "eval_accuracy_road": 0.9815051506251005,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6548672566371682,
+      "eval_accuracy_speedway bricks": 0.988976342793561,
+      "eval_accuracy_steel": 0.8866417122881076,
+      "eval_accuracy_tiles": 0.00013696753869332968,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9795218490218404,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8839437248211944,
+      "eval_iou_concrete tiles": 0.9194070540126086,
+      "eval_iou_dark bricks": 0.8166182951736,
+      "eval_iou_grass": 0.8160541250048257,
+      "eval_iou_red bricks": 0.7821557971014492,
+      "eval_iou_road": 0.9732938112436508,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.43548564514354854,
+      "eval_iou_speedway bricks": 0.9751927495927238,
+      "eval_iou_steel": 0.8360920844963109,
+      "eval_iou_tiles": 0.00013696753869332968,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08922382444143295,
+      "eval_mean_accuracy": 0.7018384658170983,
+      "eval_mean_iou": 0.6475309310115727,
+      "eval_overall_accuracy": 0.9775008095635308,
+      "eval_runtime": 3.3735,
+      "eval_samples_per_second": 10.671,
+      "eval_steps_per_second": 0.593,
+      "step": 2020
+    },
+    {
+      "epoch": 183.73,
+      "learning_rate": 6.658373205741627e-05,
+      "loss": 0.0413,
+      "step": 2021
+    },
+    {
+      "epoch": 183.82,
+      "learning_rate": 6.65645933014354e-05,
+      "loss": 0.0413,
+      "step": 2022
+    },
+    {
+      "epoch": 183.91,
+      "learning_rate": 6.654545454545455e-05,
+      "loss": 0.0453,
+      "step": 2023
+    },
+    {
+      "epoch": 184.0,
+      "learning_rate": 6.652631578947369e-05,
+      "loss": 0.0423,
+      "step": 2024
+    },
+    {
+      "epoch": 184.09,
+      "learning_rate": 6.650717703349282e-05,
+      "loss": 0.0403,
+      "step": 2025
+    },
+    {
+      "epoch": 184.18,
+      "learning_rate": 6.648803827751197e-05,
+      "loss": 0.0403,
+      "step": 2026
+    },
+    {
+      "epoch": 184.27,
+      "learning_rate": 6.64688995215311e-05,
+      "loss": 0.0413,
+      "step": 2027
+    },
+    {
+      "epoch": 184.36,
+      "learning_rate": 6.644976076555023e-05,
+      "loss": 0.0447,
+      "step": 2028
+    },
+    {
+      "epoch": 184.45,
+      "learning_rate": 6.643062200956938e-05,
+      "loss": 0.0428,
+      "step": 2029
+    },
+    {
+      "epoch": 184.55,
+      "learning_rate": 6.641148325358852e-05,
+      "loss": 0.0452,
+      "step": 2030
+    },
+    {
+      "epoch": 184.64,
+      "learning_rate": 6.639234449760766e-05,
+      "loss": 0.0422,
+      "step": 2031
+    },
+    {
+      "epoch": 184.73,
+      "learning_rate": 6.63732057416268e-05,
+      "loss": 0.0431,
+      "step": 2032
+    },
+    {
+      "epoch": 184.82,
+      "learning_rate": 6.635406698564593e-05,
+      "loss": 0.0419,
+      "step": 2033
+    },
+    {
+      "epoch": 184.91,
+      "learning_rate": 6.633492822966507e-05,
+      "loss": 0.0495,
+      "step": 2034
+    },
+    {
+      "epoch": 185.0,
+      "learning_rate": 6.631578947368421e-05,
+      "loss": 0.047,
+      "step": 2035
+    },
+    {
+      "epoch": 185.09,
+      "learning_rate": 6.629665071770336e-05,
+      "loss": 0.0389,
+      "step": 2036
+    },
+    {
+      "epoch": 185.18,
+      "learning_rate": 6.627751196172249e-05,
+      "loss": 0.0541,
+      "step": 2037
+    },
+    {
+      "epoch": 185.27,
+      "learning_rate": 6.625837320574163e-05,
+      "loss": 0.045,
+      "step": 2038
+    },
+    {
+      "epoch": 185.36,
+      "learning_rate": 6.623923444976077e-05,
+      "loss": 0.0403,
+      "step": 2039
+    },
+    {
+      "epoch": 185.45,
+      "learning_rate": 6.62200956937799e-05,
+      "loss": 0.044,
+      "step": 2040
+    },
+    {
+      "epoch": 185.45,
+      "eval_accuracy_NAT": 0.993605587995113,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9497926413150779,
+      "eval_accuracy_concrete tiles": 0.9792392833119306,
+      "eval_accuracy_dark bricks": 0.8671721642738924,
+      "eval_accuracy_grass": 0.8754355735590685,
+      "eval_accuracy_red bricks": 0.9172585227272727,
+      "eval_accuracy_road": 0.9807892989853528,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6492983762026852,
+      "eval_accuracy_speedway bricks": 0.9869519096046834,
+      "eval_accuracy_steel": 0.8973517577996174,
+      "eval_accuracy_tiles": 0.00013696753869332968,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9793256577602958,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8874684024979729,
+      "eval_iou_concrete tiles": 0.9537806367074659,
+      "eval_iou_dark bricks": 0.8174891146589259,
+      "eval_iou_grass": 0.821271070570221,
+      "eval_iou_red bricks": 0.7977146386658431,
+      "eval_iou_road": 0.9735587165973096,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.43363467309852155,
+      "eval_iou_speedway bricks": 0.9757760695845153,
+      "eval_iou_steel": 0.8395596936059921,
+      "eval_iou_tiles": 0.00013696753869332968,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08602378517389297,
+      "eval_mean_accuracy": 0.6997716987164144,
+      "eval_mean_iou": 0.6522858185604428,
+      "eval_overall_accuracy": 0.9778881072998047,
+      "eval_runtime": 3.7372,
+      "eval_samples_per_second": 9.633,
+      "eval_steps_per_second": 0.535,
+      "step": 2040
+    },
+    {
+      "epoch": 185.55,
+      "learning_rate": 6.620095693779904e-05,
+      "loss": 0.0419,
+      "step": 2041
+    },
+    {
+      "epoch": 185.64,
+      "learning_rate": 6.618181818181819e-05,
+      "loss": 0.0427,
+      "step": 2042
+    },
+    {
+      "epoch": 185.73,
+      "learning_rate": 6.616267942583732e-05,
+      "loss": 0.0414,
+      "step": 2043
+    },
+    {
+      "epoch": 185.82,
+      "learning_rate": 6.614354066985647e-05,
+      "loss": 0.0393,
+      "step": 2044
+    },
+    {
+      "epoch": 185.91,
+      "learning_rate": 6.61244019138756e-05,
+      "loss": 0.0441,
+      "step": 2045
+    },
+    {
+      "epoch": 186.0,
+      "learning_rate": 6.610526315789473e-05,
+      "loss": 0.0444,
+      "step": 2046
+    },
+    {
+      "epoch": 186.09,
+      "learning_rate": 6.608612440191388e-05,
+      "loss": 0.0433,
+      "step": 2047
+    },
+    {
+      "epoch": 186.18,
+      "learning_rate": 6.606698564593302e-05,
+      "loss": 0.0461,
+      "step": 2048
+    },
+    {
+      "epoch": 186.27,
+      "learning_rate": 6.604784688995215e-05,
+      "loss": 0.0438,
+      "step": 2049
+    },
+    {
+      "epoch": 186.36,
+      "learning_rate": 6.60287081339713e-05,
+      "loss": 0.0356,
+      "step": 2050
+    },
+    {
+      "epoch": 186.45,
+      "learning_rate": 6.600956937799043e-05,
+      "loss": 0.0453,
+      "step": 2051
+    },
+    {
+      "epoch": 186.55,
+      "learning_rate": 6.599043062200956e-05,
+      "loss": 0.0478,
+      "step": 2052
+    },
+    {
+      "epoch": 186.64,
+      "learning_rate": 6.597129186602871e-05,
+      "loss": 0.046,
+      "step": 2053
+    },
+    {
+      "epoch": 186.73,
+      "learning_rate": 6.595215311004786e-05,
+      "loss": 0.0469,
+      "step": 2054
+    },
+    {
+      "epoch": 186.82,
+      "learning_rate": 6.593301435406699e-05,
+      "loss": 0.0354,
+      "step": 2055
+    },
+    {
+      "epoch": 186.91,
+      "learning_rate": 6.591387559808613e-05,
+      "loss": 0.0427,
+      "step": 2056
+    },
+    {
+      "epoch": 187.0,
+      "learning_rate": 6.589473684210526e-05,
+      "loss": 0.0424,
+      "step": 2057
+    },
+    {
+      "epoch": 187.09,
+      "learning_rate": 6.58755980861244e-05,
+      "loss": 0.0397,
+      "step": 2058
+    },
+    {
+      "epoch": 187.18,
+      "learning_rate": 6.585645933014354e-05,
+      "loss": 0.0457,
+      "step": 2059
+    },
+    {
+      "epoch": 187.27,
+      "learning_rate": 6.583732057416269e-05,
+      "loss": 0.0397,
+      "step": 2060
+    },
+    {
+      "epoch": 187.27,
+      "eval_accuracy_NAT": 0.994215932406518,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9488229108545431,
+      "eval_accuracy_concrete tiles": 0.9801259707698893,
+      "eval_accuracy_dark bricks": 0.8497363457911551,
+      "eval_accuracy_grass": 0.8691339556894557,
+      "eval_accuracy_red bricks": 0.9287997159090909,
+      "eval_accuracy_road": 0.9778654319838159,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6550602376423235,
+      "eval_accuracy_speedway bricks": 0.9878763303633601,
+      "eval_accuracy_steel": 0.8919266539146494,
+      "eval_accuracy_tiles": 0.00041090261607998904,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9793014916044781,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8882387581442478,
+      "eval_iou_concrete tiles": 0.95481488100557,
+      "eval_iou_dark bricks": 0.8130892751915144,
+      "eval_iou_grass": 0.8197215226592476,
+      "eval_iou_red bricks": 0.7311993290466872,
+      "eval_iou_road": 0.9727019150156908,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.43753107334229474,
+      "eval_iou_speedway bricks": 0.9752703769341892,
+      "eval_iou_steel": 0.8392357043668495,
+      "eval_iou_tiles": 0.00041090261607998904,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08747704327106476,
+      "eval_mean_accuracy": 0.6987672606108369,
+      "eval_mean_iou": 0.6470396330712961,
+      "eval_overall_accuracy": 0.9778078926934136,
+      "eval_runtime": 4.1389,
+      "eval_samples_per_second": 8.698,
+      "eval_steps_per_second": 0.483,
+      "step": 2060
+    },
+    {
+      "epoch": 187.36,
+      "learning_rate": 6.581818181818182e-05,
+      "loss": 0.0493,
+      "step": 2061
+    },
+    {
+      "epoch": 187.45,
+      "learning_rate": 6.579904306220097e-05,
+      "loss": 0.0397,
+      "step": 2062
+    },
+    {
+      "epoch": 187.55,
+      "learning_rate": 6.57799043062201e-05,
+      "loss": 0.0435,
+      "step": 2063
+    },
+    {
+      "epoch": 187.64,
+      "learning_rate": 6.576076555023923e-05,
+      "loss": 0.0406,
+      "step": 2064
+    },
+    {
+      "epoch": 187.73,
+      "learning_rate": 6.574162679425838e-05,
+      "loss": 0.0399,
+      "step": 2065
+    },
+    {
+      "epoch": 187.82,
+      "learning_rate": 6.572248803827752e-05,
+      "loss": 0.041,
+      "step": 2066
+    },
+    {
+      "epoch": 187.91,
+      "learning_rate": 6.570334928229665e-05,
+      "loss": 0.0369,
+      "step": 2067
+    },
+    {
+      "epoch": 188.0,
+      "learning_rate": 6.56842105263158e-05,
+      "loss": 0.0435,
+      "step": 2068
+    },
+    {
+      "epoch": 188.09,
+      "learning_rate": 6.566507177033493e-05,
+      "loss": 0.0414,
+      "step": 2069
+    },
+    {
+      "epoch": 188.18,
+      "learning_rate": 6.564593301435406e-05,
+      "loss": 0.0447,
+      "step": 2070
+    },
+    {
+      "epoch": 188.27,
+      "learning_rate": 6.562679425837321e-05,
+      "loss": 0.0486,
+      "step": 2071
+    },
+    {
+      "epoch": 188.36,
+      "learning_rate": 6.560765550239235e-05,
+      "loss": 0.0422,
+      "step": 2072
+    },
+    {
+      "epoch": 188.45,
+      "learning_rate": 6.558851674641149e-05,
+      "loss": 0.0402,
+      "step": 2073
+    },
+    {
+      "epoch": 188.55,
+      "learning_rate": 6.556937799043063e-05,
+      "loss": 0.0428,
+      "step": 2074
+    },
+    {
+      "epoch": 188.64,
+      "learning_rate": 6.555023923444976e-05,
+      "loss": 0.0451,
+      "step": 2075
+    },
+    {
+      "epoch": 188.73,
+      "learning_rate": 6.55311004784689e-05,
+      "loss": 0.0414,
+      "step": 2076
+    },
+    {
+      "epoch": 188.82,
+      "learning_rate": 6.551196172248804e-05,
+      "loss": 0.0411,
+      "step": 2077
+    },
+    {
+      "epoch": 188.91,
+      "learning_rate": 6.549282296650719e-05,
+      "loss": 0.04,
+      "step": 2078
+    },
+    {
+      "epoch": 189.0,
+      "learning_rate": 6.547368421052632e-05,
+      "loss": 0.041,
+      "step": 2079
+    },
+    {
+      "epoch": 189.09,
+      "learning_rate": 6.545454545454546e-05,
+      "loss": 0.0397,
+      "step": 2080
+    },
+    {
+      "epoch": 189.09,
+      "eval_accuracy_NAT": 0.9912300266158864,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9541007661273015,
+      "eval_accuracy_concrete tiles": 0.9886259401944597,
+      "eval_accuracy_dark bricks": 0.8845310034255802,
+      "eval_accuracy_grass": 0.9009216509330099,
+      "eval_accuracy_red bricks": 0.9074928977272727,
+      "eval_accuracy_road": 0.9806441939232419,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6497946130730846,
+      "eval_accuracy_speedway bricks": 0.9855616580198328,
+      "eval_accuracy_steel": 0.912085284611833,
+      "eval_accuracy_tiles": 0.00198602931105328,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9796630954660506,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8840573226005699,
+      "eval_iou_concrete tiles": 0.939887215859543,
+      "eval_iou_dark bricks": 0.813170093061109,
+      "eval_iou_grass": 0.8201863937049965,
+      "eval_iou_red bricks": 0.8260869565217391,
+      "eval_iou_road": 0.9733469197191027,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4312742443094489,
+      "eval_iou_speedway bricks": 0.9753436011599077,
+      "eval_iou_steel": 0.836869936227674,
+      "eval_iou_tiles": 0.00198602931105328,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08754990249872208,
+      "eval_mean_accuracy": 0.704382620304812,
+      "eval_mean_iou": 0.6524516775339381,
+      "eval_overall_accuracy": 0.9775863223605685,
+      "eval_runtime": 4.0243,
+      "eval_samples_per_second": 8.946,
+      "eval_steps_per_second": 0.497,
+      "step": 2080
+    },
+    {
+      "epoch": 189.18,
+      "learning_rate": 6.54354066985646e-05,
+      "loss": 0.0432,
+      "step": 2081
+    },
+    {
+      "epoch": 189.27,
+      "learning_rate": 6.541626794258373e-05,
+      "loss": 0.0397,
+      "step": 2082
+    },
+    {
+      "epoch": 189.36,
+      "learning_rate": 6.539712918660287e-05,
+      "loss": 0.0406,
+      "step": 2083
+    },
+    {
+      "epoch": 189.45,
+      "learning_rate": 6.537799043062202e-05,
+      "loss": 0.0488,
+      "step": 2084
+    },
+    {
+      "epoch": 189.55,
+      "learning_rate": 6.535885167464115e-05,
+      "loss": 0.0414,
+      "step": 2085
+    },
+    {
+      "epoch": 189.64,
+      "learning_rate": 6.53397129186603e-05,
+      "loss": 0.0421,
+      "step": 2086
+    },
+    {
+      "epoch": 189.73,
+      "learning_rate": 6.532057416267943e-05,
+      "loss": 0.0472,
+      "step": 2087
+    },
+    {
+      "epoch": 189.82,
+      "learning_rate": 6.530143540669856e-05,
+      "loss": 0.0404,
+      "step": 2088
+    },
+    {
+      "epoch": 189.91,
+      "learning_rate": 6.52822966507177e-05,
+      "loss": 0.0358,
+      "step": 2089
+    },
+    {
+      "epoch": 190.0,
+      "learning_rate": 6.526315789473685e-05,
+      "loss": 0.0383,
+      "step": 2090
+    },
+    {
+      "epoch": 190.09,
+      "learning_rate": 6.524401913875598e-05,
+      "loss": 0.0542,
+      "step": 2091
+    },
+    {
+      "epoch": 190.18,
+      "learning_rate": 6.522488038277512e-05,
+      "loss": 0.0405,
+      "step": 2092
+    },
+    {
+      "epoch": 190.27,
+      "learning_rate": 6.520574162679426e-05,
+      "loss": 0.0412,
+      "step": 2093
+    },
+    {
+      "epoch": 190.36,
+      "learning_rate": 6.518660287081339e-05,
+      "loss": 0.0411,
+      "step": 2094
+    },
+    {
+      "epoch": 190.45,
+      "learning_rate": 6.516746411483254e-05,
+      "loss": 0.0498,
+      "step": 2095
+    },
+    {
+      "epoch": 190.55,
+      "learning_rate": 6.514832535885168e-05,
+      "loss": 0.0364,
+      "step": 2096
+    },
+    {
+      "epoch": 190.64,
+      "learning_rate": 6.512918660287082e-05,
+      "loss": 0.0423,
+      "step": 2097
+    },
+    {
+      "epoch": 190.73,
+      "learning_rate": 6.511004784688995e-05,
+      "loss": 0.0414,
+      "step": 2098
+    },
+    {
+      "epoch": 190.82,
+      "learning_rate": 6.50909090909091e-05,
+      "loss": 0.0391,
+      "step": 2099
+    },
+    {
+      "epoch": 190.91,
+      "learning_rate": 6.507177033492823e-05,
+      "loss": 0.0417,
+      "step": 2100
+    },
+    {
+      "epoch": 190.91,
+      "eval_accuracy_NAT": 0.991197736061597,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9502540345079188,
+      "eval_accuracy_concrete tiles": 0.9822968262704092,
+      "eval_accuracy_dark bricks": 0.8675955505946653,
+      "eval_accuracy_grass": 0.8970351289856947,
+      "eval_accuracy_red bricks": 0.9106889204545454,
+      "eval_accuracy_road": 0.980269339179455,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6612080610922725,
+      "eval_accuracy_speedway bricks": 0.9887615296146517,
+      "eval_accuracy_steel": 0.8964448255392125,
+      "eval_accuracy_tiles": 0.006848376934666484,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9794945288636141,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8841520561980459,
+      "eval_iou_concrete tiles": 0.9528427796067266,
+      "eval_iou_dark bricks": 0.8146956773167559,
+      "eval_iou_grass": 0.8107125484076795,
+      "eval_iou_red bricks": 0.8251287001287001,
+      "eval_iou_road": 0.974251738082791,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4354709855472438,
+      "eval_iou_speedway bricks": 0.9753645350017291,
+      "eval_iou_steel": 0.8354373967497791,
+      "eval_iou_tiles": 0.006848376934666484,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08729197829961777,
+      "eval_mean_accuracy": 0.7025077176334682,
+      "eval_mean_iou": 0.6534153325259793,
+      "eval_overall_accuracy": 0.9774580001831055,
+      "eval_runtime": 3.7981,
+      "eval_samples_per_second": 9.478,
+      "eval_steps_per_second": 0.527,
+      "step": 2100
+    },
+    {
+      "epoch": 191.0,
+      "learning_rate": 6.505263157894737e-05,
+      "loss": 0.0391,
+      "step": 2101
+    },
+    {
+      "epoch": 191.09,
+      "learning_rate": 6.503349282296652e-05,
+      "loss": 0.0396,
+      "step": 2102
+    },
+    {
+      "epoch": 191.18,
+      "learning_rate": 6.501435406698565e-05,
+      "loss": 0.0424,
+      "step": 2103
+    },
+    {
+      "epoch": 191.27,
+      "learning_rate": 6.499521531100478e-05,
+      "loss": 0.0434,
+      "step": 2104
+    },
+    {
+      "epoch": 191.36,
+      "learning_rate": 6.497607655502393e-05,
+      "loss": 0.036,
+      "step": 2105
+    },
+    {
+      "epoch": 191.45,
+      "learning_rate": 6.495693779904306e-05,
+      "loss": 0.0472,
+      "step": 2106
+    },
+    {
+      "epoch": 191.55,
+      "learning_rate": 6.49377990430622e-05,
+      "loss": 0.0445,
+      "step": 2107
+    },
+    {
+      "epoch": 191.64,
+      "learning_rate": 6.491866028708135e-05,
+      "loss": 0.0388,
+      "step": 2108
+    },
+    {
+      "epoch": 191.73,
+      "learning_rate": 6.489952153110048e-05,
+      "loss": 0.0416,
+      "step": 2109
+    },
+    {
+      "epoch": 191.82,
+      "learning_rate": 6.488038277511961e-05,
+      "loss": 0.0392,
+      "step": 2110
+    },
+    {
+      "epoch": 191.91,
+      "learning_rate": 6.486124401913876e-05,
+      "loss": 0.0415,
+      "step": 2111
+    },
+    {
+      "epoch": 192.0,
+      "learning_rate": 6.484210526315789e-05,
+      "loss": 0.0429,
+      "step": 2112
+    },
+    {
+      "epoch": 192.09,
+      "learning_rate": 6.482296650717704e-05,
+      "loss": 0.0423,
+      "step": 2113
+    },
+    {
+      "epoch": 192.18,
+      "learning_rate": 6.480382775119618e-05,
+      "loss": 0.044,
+      "step": 2114
+    },
+    {
+      "epoch": 192.27,
+      "learning_rate": 6.478468899521531e-05,
+      "loss": 0.0389,
+      "step": 2115
+    },
+    {
+      "epoch": 192.36,
+      "learning_rate": 6.476555023923445e-05,
+      "loss": 0.0441,
+      "step": 2116
+    },
+    {
+      "epoch": 192.45,
+      "learning_rate": 6.474641148325359e-05,
+      "loss": 0.0395,
+      "step": 2117
+    },
+    {
+      "epoch": 192.55,
+      "learning_rate": 6.472727272727272e-05,
+      "loss": 0.0414,
+      "step": 2118
+    },
+    {
+      "epoch": 192.64,
+      "learning_rate": 6.470813397129187e-05,
+      "loss": 0.0409,
+      "step": 2119
+    },
+    {
+      "epoch": 192.73,
+      "learning_rate": 6.468899521531101e-05,
+      "loss": 0.0399,
+      "step": 2120
+    },
+    {
+      "epoch": 192.73,
+      "eval_accuracy_NAT": 0.9934376065323615,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9436885703249066,
+      "eval_accuracy_concrete tiles": 0.9882896104690271,
+      "eval_accuracy_dark bricks": 0.8584350101997613,
+      "eval_accuracy_grass": 0.8779799872080303,
+      "eval_accuracy_red bricks": 0.9041193181818182,
+      "eval_accuracy_road": 0.9786514177369172,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6418548231466932,
+      "eval_accuracy_speedway bricks": 0.9876560865141976,
+      "eval_accuracy_steel": 0.8982586900600225,
+      "eval_accuracy_tiles": 0.0012327078482399672,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9792014729152166,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8824992505609818,
+      "eval_iou_concrete tiles": 0.9265586928479289,
+      "eval_iou_dark bricks": 0.8126730797259875,
+      "eval_iou_grass": 0.8113882796779059,
+      "eval_iou_red bricks": 0.8227500403942478,
+      "eval_iou_road": 0.9729150112458421,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4276006464883926,
+      "eval_iou_speedway bricks": 0.9755509834797338,
+      "eval_iou_steel": 0.8374044980092543,
+      "eval_iou_tiles": 0.0012327078482399672,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08815740048885345,
+      "eval_mean_accuracy": 0.6979695252478442,
+      "eval_mean_iou": 0.6499826663995177,
+      "eval_overall_accuracy": 0.9773026572333442,
+      "eval_runtime": 3.3986,
+      "eval_samples_per_second": 10.593,
+      "eval_steps_per_second": 0.588,
+      "step": 2120
+    },
+    {
+      "epoch": 192.82,
+      "learning_rate": 6.466985645933015e-05,
+      "loss": 0.0432,
+      "step": 2121
+    },
+    {
+      "epoch": 192.91,
+      "learning_rate": 6.465071770334928e-05,
+      "loss": 0.0392,
+      "step": 2122
+    },
+    {
+      "epoch": 193.0,
+      "learning_rate": 6.463157894736842e-05,
+      "loss": 0.0365,
+      "step": 2123
+    },
+    {
+      "epoch": 193.09,
+      "learning_rate": 6.461244019138756e-05,
+      "loss": 0.036,
+      "step": 2124
+    },
+    {
+      "epoch": 193.18,
+      "learning_rate": 6.45933014354067e-05,
+      "loss": 0.04,
+      "step": 2125
+    },
+    {
+      "epoch": 193.27,
+      "learning_rate": 6.457416267942585e-05,
+      "loss": 0.0413,
+      "step": 2126
+    },
+    {
+      "epoch": 193.36,
+      "learning_rate": 6.455502392344498e-05,
+      "loss": 0.0349,
+      "step": 2127
+    },
+    {
+      "epoch": 193.45,
+      "learning_rate": 6.453588516746411e-05,
+      "loss": 0.045,
+      "step": 2128
+    },
+    {
+      "epoch": 193.55,
+      "learning_rate": 6.451674641148326e-05,
+      "loss": 0.0417,
+      "step": 2129
+    },
+    {
+      "epoch": 193.64,
+      "learning_rate": 6.449760765550239e-05,
+      "loss": 0.0392,
+      "step": 2130
+    },
+    {
+      "epoch": 193.73,
+      "learning_rate": 6.447846889952153e-05,
+      "loss": 0.0399,
+      "step": 2131
+    },
+    {
+      "epoch": 193.82,
+      "learning_rate": 6.445933014354068e-05,
+      "loss": 0.0466,
+      "step": 2132
+    },
+    {
+      "epoch": 193.91,
+      "learning_rate": 6.444019138755981e-05,
+      "loss": 0.0461,
+      "step": 2133
+    },
+    {
+      "epoch": 194.0,
+      "learning_rate": 6.442105263157894e-05,
+      "loss": 0.05,
+      "step": 2134
+    },
+    {
+      "epoch": 194.09,
+      "learning_rate": 6.440191387559809e-05,
+      "loss": 0.0482,
+      "step": 2135
+    },
+    {
+      "epoch": 194.18,
+      "learning_rate": 6.438277511961724e-05,
+      "loss": 0.0407,
+      "step": 2136
+    },
+    {
+      "epoch": 194.27,
+      "learning_rate": 6.436363636363637e-05,
+      "loss": 0.0405,
+      "step": 2137
+    },
+    {
+      "epoch": 194.36,
+      "learning_rate": 6.434449760765551e-05,
+      "loss": 0.0404,
+      "step": 2138
+    },
+    {
+      "epoch": 194.45,
+      "learning_rate": 6.432535885167464e-05,
+      "loss": 0.0481,
+      "step": 2139
+    },
+    {
+      "epoch": 194.55,
+      "learning_rate": 6.430622009569378e-05,
+      "loss": 0.0464,
+      "step": 2140
+    },
+    {
+      "epoch": 194.55,
+      "eval_accuracy_NAT": 0.9926255784949849,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9523356689128288,
+      "eval_accuracy_concrete tiles": 0.9833975417354613,
+      "eval_accuracy_dark bricks": 0.875832339016974,
+      "eval_accuracy_grass": 0.8898807131348364,
+      "eval_accuracy_red bricks": 0.9272017045454546,
+      "eval_accuracy_road": 0.9843093059503958,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6454387560995782,
+      "eval_accuracy_speedway bricks": 0.9840362430864551,
+      "eval_accuracy_steel": 0.9077897236330057,
+      "eval_accuracy_tiles": 0.05588275578687851,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.97952273875686,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8884027216468016,
+      "eval_iou_concrete tiles": 0.9517369947327928,
+      "eval_iou_dark bricks": 0.8236209642391776,
+      "eval_iou_grass": 0.823876598099269,
+      "eval_iou_red bricks": 0.8140296180826189,
+      "eval_iou_road": 0.974557683696852,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.43093007417769513,
+      "eval_iou_speedway bricks": 0.9750584898293765,
+      "eval_iou_steel": 0.8369299999239873,
+      "eval_iou_tiles": 0.05588275578687851,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0844007208943367,
+      "eval_mean_accuracy": 0.707594640799758,
+      "eval_mean_iou": 0.6580422029978699,
+      "eval_overall_accuracy": 0.9779846403333876,
+      "eval_runtime": 4.0364,
+      "eval_samples_per_second": 8.919,
+      "eval_steps_per_second": 0.495,
+      "step": 2140
+    },
+    {
+      "epoch": 194.64,
+      "learning_rate": 6.428708133971292e-05,
+      "loss": 0.0393,
+      "step": 2141
+    },
+    {
+      "epoch": 194.73,
+      "learning_rate": 6.426794258373207e-05,
+      "loss": 0.0439,
+      "step": 2142
+    },
+    {
+      "epoch": 194.82,
+      "learning_rate": 6.42488038277512e-05,
+      "loss": 0.0383,
+      "step": 2143
+    },
+    {
+      "epoch": 194.91,
+      "learning_rate": 6.422966507177035e-05,
+      "loss": 0.0337,
+      "step": 2144
+    },
+    {
+      "epoch": 195.0,
+      "learning_rate": 6.421052631578948e-05,
+      "loss": 0.0433,
+      "step": 2145
+    },
+    {
+      "epoch": 195.09,
+      "learning_rate": 6.419138755980861e-05,
+      "loss": 0.0393,
+      "step": 2146
+    },
+    {
+      "epoch": 195.18,
+      "learning_rate": 6.417224880382775e-05,
+      "loss": 0.0402,
+      "step": 2147
+    },
+    {
+      "epoch": 195.27,
+      "learning_rate": 6.41531100478469e-05,
+      "loss": 0.0412,
+      "step": 2148
+    },
+    {
+      "epoch": 195.36,
+      "learning_rate": 6.413397129186603e-05,
+      "loss": 0.0441,
+      "step": 2149
+    },
+    {
+      "epoch": 195.45,
+      "learning_rate": 6.411483253588518e-05,
+      "loss": 0.0448,
+      "step": 2150
+    },
+    {
+      "epoch": 195.55,
+      "learning_rate": 6.409569377990431e-05,
+      "loss": 0.0352,
+      "step": 2151
+    },
+    {
+      "epoch": 195.64,
+      "learning_rate": 6.407655502392344e-05,
+      "loss": 0.0438,
+      "step": 2152
+    },
+    {
+      "epoch": 195.73,
+      "learning_rate": 6.405741626794259e-05,
+      "loss": 0.041,
+      "step": 2153
+    },
+    {
+      "epoch": 195.82,
+      "learning_rate": 6.403827751196173e-05,
+      "loss": 0.0435,
+      "step": 2154
+    },
+    {
+      "epoch": 195.91,
+      "learning_rate": 6.401913875598087e-05,
+      "loss": 0.0432,
+      "step": 2155
+    },
+    {
+      "epoch": 196.0,
+      "learning_rate": 6.400000000000001e-05,
+      "loss": 0.0396,
+      "step": 2156
+    },
+    {
+      "epoch": 196.09,
+      "learning_rate": 6.398086124401914e-05,
+      "loss": 0.0371,
+      "step": 2157
+    },
+    {
+      "epoch": 196.18,
+      "learning_rate": 6.396172248803827e-05,
+      "loss": 0.036,
+      "step": 2158
+    },
+    {
+      "epoch": 196.27,
+      "learning_rate": 6.394258373205742e-05,
+      "loss": 0.0428,
+      "step": 2159
+    },
+    {
+      "epoch": 196.36,
+      "learning_rate": 6.392344497607657e-05,
+      "loss": 0.0411,
+      "step": 2160
+    },
+    {
+      "epoch": 196.36,
+      "eval_accuracy_NAT": 0.9914747643142439,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.956733658096914,
+      "eval_accuracy_concrete tiles": 0.9882284596098575,
+      "eval_accuracy_dark bricks": 0.8709441514953236,
+      "eval_accuracy_grass": 0.8873747452965046,
+      "eval_accuracy_red bricks": 0.8975497159090909,
+      "eval_accuracy_road": 0.9823963375482323,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6750199873183911,
+      "eval_accuracy_speedway bricks": 0.9869283767002524,
+      "eval_accuracy_steel": 0.9026284545874282,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.980022220206784,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.883892392907151,
+      "eval_iou_concrete tiles": 0.9450584795321637,
+      "eval_iou_dark bricks": 0.8253875615538938,
+      "eval_iou_grass": 0.821062333252486,
+      "eval_iou_red bricks": 0.822620016273393,
+      "eval_iou_road": 0.9740362193301642,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4416725291772643,
+      "eval_iou_speedway bricks": 0.9754944148522168,
+      "eval_iou_steel": 0.8395616530801616,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0853801816701889,
+      "eval_mean_accuracy": 0.7030214346827877,
+      "eval_mean_iou": 0.6545236784742829,
+      "eval_overall_accuracy": 0.9778517617119683,
+      "eval_runtime": 3.7657,
+      "eval_samples_per_second": 9.56,
+      "eval_steps_per_second": 0.531,
+      "step": 2160
+    },
+    {
+      "epoch": 196.45,
+      "learning_rate": 6.39043062200957e-05,
+      "loss": 0.0441,
+      "step": 2161
+    },
+    {
+      "epoch": 196.55,
+      "learning_rate": 6.388516746411484e-05,
+      "loss": 0.0421,
+      "step": 2162
+    },
+    {
+      "epoch": 196.64,
+      "learning_rate": 6.386602870813398e-05,
+      "loss": 0.0415,
+      "step": 2163
+    },
+    {
+      "epoch": 196.73,
+      "learning_rate": 6.384688995215311e-05,
+      "loss": 0.0405,
+      "step": 2164
+    },
+    {
+      "epoch": 196.82,
+      "learning_rate": 6.382775119617225e-05,
+      "loss": 0.038,
+      "step": 2165
+    },
+    {
+      "epoch": 196.91,
+      "learning_rate": 6.38086124401914e-05,
+      "loss": 0.0394,
+      "step": 2166
+    },
+    {
+      "epoch": 197.0,
+      "learning_rate": 6.378947368421053e-05,
+      "loss": 0.0377,
+      "step": 2167
+    },
+    {
+      "epoch": 197.09,
+      "learning_rate": 6.377033492822966e-05,
+      "loss": 0.0378,
+      "step": 2168
+    },
+    {
+      "epoch": 197.18,
+      "learning_rate": 6.375119617224881e-05,
+      "loss": 0.0449,
+      "step": 2169
+    },
+    {
+      "epoch": 197.27,
+      "learning_rate": 6.373205741626794e-05,
+      "loss": 0.0414,
+      "step": 2170
+    },
+    {
+      "epoch": 197.36,
+      "learning_rate": 6.371291866028709e-05,
+      "loss": 0.035,
+      "step": 2171
+    },
+    {
+      "epoch": 197.45,
+      "learning_rate": 6.369377990430623e-05,
+      "loss": 0.0429,
+      "step": 2172
+    },
+    {
+      "epoch": 197.55,
+      "learning_rate": 6.367464114832536e-05,
+      "loss": 0.0391,
+      "step": 2173
+    },
+    {
+      "epoch": 197.64,
+      "learning_rate": 6.36555023923445e-05,
+      "loss": 0.0455,
+      "step": 2174
+    },
+    {
+      "epoch": 197.73,
+      "learning_rate": 6.363636363636364e-05,
+      "loss": 0.0379,
+      "step": 2175
+    },
+    {
+      "epoch": 197.82,
+      "learning_rate": 6.361722488038277e-05,
+      "loss": 0.0451,
+      "step": 2176
+    },
+    {
+      "epoch": 197.91,
+      "learning_rate": 6.359808612440192e-05,
+      "loss": 0.0424,
+      "step": 2177
+    },
+    {
+      "epoch": 198.0,
+      "learning_rate": 6.357894736842106e-05,
+      "loss": 0.0419,
+      "step": 2178
+    },
+    {
+      "epoch": 198.09,
+      "learning_rate": 6.35598086124402e-05,
+      "loss": 0.0377,
+      "step": 2179
+    },
+    {
+      "epoch": 198.18,
+      "learning_rate": 6.354066985645933e-05,
+      "loss": 0.0403,
+      "step": 2180
+    },
+    {
+      "epoch": 198.18,
+      "eval_accuracy_NAT": 0.9916487451149508,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9577852053736212,
+      "eval_accuracy_concrete tiles": 0.9836115697425549,
+      "eval_accuracy_dark bricks": 0.8691736268811824,
+      "eval_accuracy_grass": 0.9089952711652926,
+      "eval_accuracy_red bricks": 0.9103338068181818,
+      "eval_accuracy_road": 0.9803467285459143,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6450252253742453,
+      "eval_accuracy_speedway bricks": 0.9854005481356509,
+      "eval_accuracy_steel": 0.9004600620011872,
+      "eval_accuracy_tiles": 0.07998904259690454,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.979796190590094,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8885999176224495,
+      "eval_iou_concrete tiles": 0.9507344031681295,
+      "eval_iou_dark bricks": 0.8181291210781827,
+      "eval_iou_grass": 0.8302389723486711,
+      "eval_iou_red bricks": 0.8499668435013262,
+      "eval_iou_road": 0.9745863260706236,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4303529714716648,
+      "eval_iou_speedway bricks": 0.9753955943337118,
+      "eval_iou_steel": 0.8400895357065936,
+      "eval_iou_tiles": 0.07998904259690454,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08364246040582657,
+      "eval_mean_accuracy": 0.7086746024422835,
+      "eval_mean_iou": 0.6629137629606424,
+      "eval_overall_accuracy": 0.9782142639160156,
+      "eval_runtime": 3.619,
+      "eval_samples_per_second": 9.948,
+      "eval_steps_per_second": 0.553,
+      "step": 2180
+    },
+    {
+      "epoch": 198.27,
+      "learning_rate": 6.352153110047847e-05,
+      "loss": 0.0432,
+      "step": 2181
+    },
+    {
+      "epoch": 198.36,
+      "learning_rate": 6.35023923444976e-05,
+      "loss": 0.0386,
+      "step": 2182
+    },
+    {
+      "epoch": 198.45,
+      "learning_rate": 6.348325358851675e-05,
+      "loss": 0.0446,
+      "step": 2183
+    },
+    {
+      "epoch": 198.55,
+      "learning_rate": 6.34641148325359e-05,
+      "loss": 0.0415,
+      "step": 2184
+    },
+    {
+      "epoch": 198.64,
+      "learning_rate": 6.344497607655503e-05,
+      "loss": 0.0385,
+      "step": 2185
+    },
+    {
+      "epoch": 198.73,
+      "learning_rate": 6.342583732057416e-05,
+      "loss": 0.0366,
+      "step": 2186
+    },
+    {
+      "epoch": 198.82,
+      "learning_rate": 6.34066985645933e-05,
+      "loss": 0.0465,
+      "step": 2187
+    },
+    {
+      "epoch": 198.91,
+      "learning_rate": 6.338755980861244e-05,
+      "loss": 0.0408,
+      "step": 2188
+    },
+    {
+      "epoch": 199.0,
+      "learning_rate": 6.336842105263158e-05,
+      "loss": 0.0404,
+      "step": 2189
+    },
+    {
+      "epoch": 199.09,
+      "learning_rate": 6.334928229665073e-05,
+      "loss": 0.0416,
+      "step": 2190
+    },
+    {
+      "epoch": 199.18,
+      "learning_rate": 6.333014354066986e-05,
+      "loss": 0.043,
+      "step": 2191
+    },
+    {
+      "epoch": 199.27,
+      "learning_rate": 6.331100478468899e-05,
+      "loss": 0.0414,
+      "step": 2192
+    },
+    {
+      "epoch": 199.36,
+      "learning_rate": 6.329186602870814e-05,
+      "loss": 0.0467,
+      "step": 2193
+    },
+    {
+      "epoch": 199.45,
+      "learning_rate": 6.327272727272727e-05,
+      "loss": 0.0365,
+      "step": 2194
+    },
+    {
+      "epoch": 199.55,
+      "learning_rate": 6.32535885167464e-05,
+      "loss": 0.0374,
+      "step": 2195
+    },
+    {
+      "epoch": 199.64,
+      "learning_rate": 6.323444976076556e-05,
+      "loss": 0.0346,
+      "step": 2196
+    },
+    {
+      "epoch": 199.73,
+      "learning_rate": 6.32153110047847e-05,
+      "loss": 0.0412,
+      "step": 2197
+    },
+    {
+      "epoch": 199.82,
+      "learning_rate": 6.319617224880383e-05,
+      "loss": 0.0371,
+      "step": 2198
+    },
+    {
+      "epoch": 199.91,
+      "learning_rate": 6.317703349282297e-05,
+      "loss": 0.0548,
+      "step": 2199
+    },
+    {
+      "epoch": 200.0,
+      "learning_rate": 6.31578947368421e-05,
+      "loss": 0.0406,
+      "step": 2200
+    },
+    {
+      "epoch": 200.0,
+      "eval_accuracy_NAT": 0.9925652322131981,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9546882376926048,
+      "eval_accuracy_concrete tiles": 0.9801259707698893,
+      "eval_accuracy_dark bricks": 0.8675570609291405,
+      "eval_accuracy_grass": 0.8980976313885578,
+      "eval_accuracy_red bricks": 0.9163707386363636,
+      "eval_accuracy_road": 0.9817651305280494,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6801201995974968,
+      "eval_accuracy_speedway bricks": 0.9864124630261867,
+      "eval_accuracy_steel": 0.896527273926522,
+      "eval_accuracy_tiles": 0.05026708670045199,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9799573395101151,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8925483557478291,
+      "eval_iou_concrete tiles": 0.9535935268919562,
+      "eval_iou_dark bricks": 0.8272773985172135,
+      "eval_iou_grass": 0.8295165476544039,
+      "eval_iou_red bricks": 0.8491280026324449,
+      "eval_iou_road": 0.9746700808036427,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4473335871910642,
+      "eval_iou_speedway bricks": 0.9755971798002657,
+      "eval_iou_steel": 0.8386265829618547,
+      "eval_iou_tiles": 0.05026708670045199,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0839119628071785,
+      "eval_mean_accuracy": 0.7080382327237278,
+      "eval_mean_iou": 0.6629627452624032,
+      "eval_overall_accuracy": 0.9785170025295682,
+      "eval_runtime": 3.232,
+      "eval_samples_per_second": 11.139,
+      "eval_steps_per_second": 0.619,
+      "step": 2200
+    },
+    {
+      "epoch": 200.09,
+      "learning_rate": 6.313875598086124e-05,
+      "loss": 0.0429,
+      "step": 2201
+    },
+    {
+      "epoch": 200.18,
+      "learning_rate": 6.31196172248804e-05,
+      "loss": 0.0438,
+      "step": 2202
+    },
+    {
+      "epoch": 200.27,
+      "learning_rate": 6.310047846889953e-05,
+      "loss": 0.0408,
+      "step": 2203
+    },
+    {
+      "epoch": 200.36,
+      "learning_rate": 6.308133971291866e-05,
+      "loss": 0.0368,
+      "step": 2204
+    },
+    {
+      "epoch": 200.45,
+      "learning_rate": 6.30622009569378e-05,
+      "loss": 0.0365,
+      "step": 2205
+    },
+    {
+      "epoch": 200.55,
+      "learning_rate": 6.304306220095694e-05,
+      "loss": 0.0456,
+      "step": 2206
+    },
+    {
+      "epoch": 200.64,
+      "learning_rate": 6.302392344497607e-05,
+      "loss": 0.0426,
+      "step": 2207
+    },
+    {
+      "epoch": 200.73,
+      "learning_rate": 6.300478468899523e-05,
+      "loss": 0.0375,
+      "step": 2208
+    },
+    {
+      "epoch": 200.82,
+      "learning_rate": 6.298564593301436e-05,
+      "loss": 0.041,
+      "step": 2209
+    },
+    {
+      "epoch": 200.91,
+      "learning_rate": 6.296650717703349e-05,
+      "loss": 0.0381,
+      "step": 2210
+    },
+    {
+      "epoch": 201.0,
+      "learning_rate": 6.294736842105264e-05,
+      "loss": 0.0452,
+      "step": 2211
+    },
+    {
+      "epoch": 201.09,
+      "learning_rate": 6.292822966507177e-05,
+      "loss": 0.0413,
+      "step": 2212
+    },
+    {
+      "epoch": 201.18,
+      "learning_rate": 6.290909090909091e-05,
+      "loss": 0.037,
+      "step": 2213
+    },
+    {
+      "epoch": 201.27,
+      "learning_rate": 6.288995215311006e-05,
+      "loss": 0.0406,
+      "step": 2214
+    },
+    {
+      "epoch": 201.36,
+      "learning_rate": 6.287081339712919e-05,
+      "loss": 0.0389,
+      "step": 2215
+    },
+    {
+      "epoch": 201.45,
+      "learning_rate": 6.285167464114832e-05,
+      "loss": 0.0362,
+      "step": 2216
+    },
+    {
+      "epoch": 201.55,
+      "learning_rate": 6.283253588516747e-05,
+      "loss": 0.0405,
+      "step": 2217
+    },
+    {
+      "epoch": 201.64,
+      "learning_rate": 6.28133971291866e-05,
+      "loss": 0.0443,
+      "step": 2218
+    },
+    {
+      "epoch": 201.73,
+      "learning_rate": 6.279425837320575e-05,
+      "loss": 0.0389,
+      "step": 2219
+    },
+    {
+      "epoch": 201.82,
+      "learning_rate": 6.277511961722489e-05,
+      "loss": 0.0461,
+      "step": 2220
+    },
+    {
+      "epoch": 201.82,
+      "eval_accuracy_NAT": 0.9921849800574948,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.949952251169578,
+      "eval_accuracy_concrete tiles": 0.9859047269614138,
+      "eval_accuracy_dark bricks": 0.8532389053539124,
+      "eval_accuracy_grass": 0.9119066675520854,
+      "eval_accuracy_red bricks": 0.9332386363636364,
+      "eval_accuracy_road": 0.9810928104069351,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6701678934744851,
+      "eval_accuracy_speedway bricks": 0.9873230054053271,
+      "eval_accuracy_steel": 0.8954059758591122,
+      "eval_accuracy_tiles": 0.008834406245719765,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9803125143830257,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.887904632121845,
+      "eval_iou_concrete tiles": 0.9410751809479337,
+      "eval_iou_dark bricks": 0.8167114909921527,
+      "eval_iou_grass": 0.8325526184793292,
+      "eval_iou_red bricks": 0.8421727287293703,
+      "eval_iou_road": 0.9734416973510085,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.44193360724284625,
+      "eval_iou_speedway bricks": 0.9755289782923645,
+      "eval_iou_steel": 0.8393189739784996,
+      "eval_iou_tiles": 0.008834406245719765,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08475645631551743,
+      "eval_mean_accuracy": 0.7053269429884383,
+      "eval_mean_iou": 0.6569066791356997,
+      "eval_overall_accuracy": 0.9783080418904623,
+      "eval_runtime": 3.3751,
+      "eval_samples_per_second": 10.666,
+      "eval_steps_per_second": 0.593,
+      "step": 2220
+    },
+    {
+      "epoch": 201.91,
+      "learning_rate": 6.275598086124402e-05,
+      "loss": 0.0386,
+      "step": 2221
+    },
+    {
+      "epoch": 202.0,
+      "learning_rate": 6.273684210526316e-05,
+      "loss": 0.0482,
+      "step": 2222
+    },
+    {
+      "epoch": 202.09,
+      "learning_rate": 6.27177033492823e-05,
+      "loss": 0.039,
+      "step": 2223
+    },
+    {
+      "epoch": 202.18,
+      "learning_rate": 6.269856459330143e-05,
+      "loss": 0.0382,
+      "step": 2224
+    },
+    {
+      "epoch": 202.27,
+      "learning_rate": 6.267942583732058e-05,
+      "loss": 0.0404,
+      "step": 2225
+    },
+    {
+      "epoch": 202.36,
+      "learning_rate": 6.266028708133973e-05,
+      "loss": 0.0404,
+      "step": 2226
+    },
+    {
+      "epoch": 202.45,
+      "learning_rate": 6.264114832535886e-05,
+      "loss": 0.0454,
+      "step": 2227
+    },
+    {
+      "epoch": 202.55,
+      "learning_rate": 6.262200956937799e-05,
+      "loss": 0.0404,
+      "step": 2228
+    },
+    {
+      "epoch": 202.64,
+      "learning_rate": 6.260287081339713e-05,
+      "loss": 0.0428,
+      "step": 2229
+    },
+    {
+      "epoch": 202.73,
+      "learning_rate": 6.258373205741627e-05,
+      "loss": 0.0448,
+      "step": 2230
+    },
+    {
+      "epoch": 202.82,
+      "learning_rate": 6.256459330143541e-05,
+      "loss": 0.0415,
+      "step": 2231
+    },
+    {
+      "epoch": 202.91,
+      "learning_rate": 6.254545454545456e-05,
+      "loss": 0.0383,
+      "step": 2232
+    },
+    {
+      "epoch": 203.0,
+      "learning_rate": 6.252631578947369e-05,
+      "loss": 0.0359,
+      "step": 2233
+    },
+    {
+      "epoch": 203.09,
+      "learning_rate": 6.250717703349282e-05,
+      "loss": 0.037,
+      "step": 2234
+    },
+    {
+      "epoch": 203.18,
+      "learning_rate": 6.248803827751197e-05,
+      "loss": 0.0475,
+      "step": 2235
+    },
+    {
+      "epoch": 203.27,
+      "learning_rate": 6.24688995215311e-05,
+      "loss": 0.0367,
+      "step": 2236
+    },
+    {
+      "epoch": 203.36,
+      "learning_rate": 6.244976076555024e-05,
+      "loss": 0.0416,
+      "step": 2237
+    },
+    {
+      "epoch": 203.45,
+      "learning_rate": 6.243062200956939e-05,
+      "loss": 0.0423,
+      "step": 2238
+    },
+    {
+      "epoch": 203.55,
+      "learning_rate": 6.241148325358852e-05,
+      "loss": 0.0348,
+      "step": 2239
+    },
+    {
+      "epoch": 203.64,
+      "learning_rate": 6.239234449760765e-05,
+      "loss": 0.0378,
+      "step": 2240
+    },
+    {
+      "epoch": 203.64,
+      "eval_accuracy_NAT": 0.9930635301657299,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9490549487102451,
+      "eval_accuracy_concrete tiles": 0.9763346175013758,
+      "eval_accuracy_dark bricks": 0.867056695277318,
+      "eval_accuracy_grass": 0.914678260991133,
+      "eval_accuracy_red bricks": 0.8971946022727273,
+      "eval_accuracy_road": 0.9813902757842626,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6515314421194828,
+      "eval_accuracy_speedway bricks": 0.9850885862999879,
+      "eval_accuracy_steel": 0.9041455049139239,
+      "eval_accuracy_tiles": 0.050335570469798654,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9801660261953216,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8912817295968237,
+      "eval_iou_concrete tiles": 0.9496505576208178,
+      "eval_iou_dark bricks": 0.8096829846883761,
+      "eval_iou_grass": 0.8343482197510712,
+      "eval_iou_red bricks": 0.8518206338503035,
+      "eval_iou_road": 0.9746114929167482,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4363230189793959,
+      "eval_iou_speedway bricks": 0.9755978333620179,
+      "eval_iou_steel": 0.8419541333005751,
+      "eval_iou_tiles": 0.050335570469798654,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08429981768131256,
+      "eval_mean_accuracy": 0.7053749257312295,
+      "eval_mean_iou": 0.6612132462100961,
+      "eval_overall_accuracy": 0.9785707261827257,
+      "eval_runtime": 3.2737,
+      "eval_samples_per_second": 10.997,
+      "eval_steps_per_second": 0.611,
+      "step": 2240
+    },
+    {
+      "epoch": 203.73,
+      "learning_rate": 6.23732057416268e-05,
+      "loss": 0.0403,
+      "step": 2241
+    },
+    {
+      "epoch": 203.82,
+      "learning_rate": 6.235406698564593e-05,
+      "loss": 0.0419,
+      "step": 2242
+    },
+    {
+      "epoch": 203.91,
+      "learning_rate": 6.233492822966508e-05,
+      "loss": 0.0377,
+      "step": 2243
+    },
+    {
+      "epoch": 204.0,
+      "learning_rate": 6.231578947368422e-05,
+      "loss": 0.0376,
+      "step": 2244
+    },
+    {
+      "epoch": 204.09,
+      "learning_rate": 6.229665071770336e-05,
+      "loss": 0.0473,
+      "step": 2245
+    },
+    {
+      "epoch": 204.18,
+      "learning_rate": 6.227751196172249e-05,
+      "loss": 0.0446,
+      "step": 2246
+    },
+    {
+      "epoch": 204.27,
+      "learning_rate": 6.225837320574163e-05,
+      "loss": 0.0422,
+      "step": 2247
+    },
+    {
+      "epoch": 204.36,
+      "learning_rate": 6.223923444976076e-05,
+      "loss": 0.0351,
+      "step": 2248
+    },
+    {
+      "epoch": 204.45,
+      "learning_rate": 6.222009569377991e-05,
+      "loss": 0.041,
+      "step": 2249
+    },
+    {
+      "epoch": 204.55,
+      "learning_rate": 6.220095693779904e-05,
+      "loss": 0.0403,
+      "step": 2250
+    },
+    {
+      "epoch": 204.64,
+      "learning_rate": 6.218181818181819e-05,
+      "loss": 0.0416,
+      "step": 2251
+    },
+    {
+      "epoch": 204.73,
+      "learning_rate": 6.216267942583732e-05,
+      "loss": 0.0376,
+      "step": 2252
+    },
+    {
+      "epoch": 204.82,
+      "learning_rate": 6.214354066985647e-05,
+      "loss": 0.0406,
+      "step": 2253
+    },
+    {
+      "epoch": 204.91,
+      "learning_rate": 6.21244019138756e-05,
+      "loss": 0.0375,
+      "step": 2254
+    },
+    {
+      "epoch": 205.0,
+      "learning_rate": 6.210526315789474e-05,
+      "loss": 0.0363,
+      "step": 2255
+    },
+    {
+      "epoch": 205.09,
+      "learning_rate": 6.208612440191387e-05,
+      "loss": 0.035,
+      "step": 2256
+    },
+    {
+      "epoch": 205.18,
+      "learning_rate": 6.206698564593302e-05,
+      "loss": 0.042,
+      "step": 2257
+    },
+    {
+      "epoch": 205.27,
+      "learning_rate": 6.204784688995215e-05,
+      "loss": 0.0397,
+      "step": 2258
+    },
+    {
+      "epoch": 205.36,
+      "learning_rate": 6.20287081339713e-05,
+      "loss": 0.0381,
+      "step": 2259
+    },
+    {
+      "epoch": 205.45,
+      "learning_rate": 6.200956937799043e-05,
+      "loss": 0.036,
+      "step": 2260
+    },
+    {
+      "epoch": 205.45,
+      "eval_accuracy_NAT": 0.9922142709428066,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9492521138246277,
+      "eval_accuracy_concrete tiles": 0.9851403412217942,
+      "eval_accuracy_dark bricks": 0.877217966975867,
+      "eval_accuracy_grass": 0.9124064630902743,
+      "eval_accuracy_red bricks": 0.9032315340909091,
+      "eval_accuracy_road": 0.9826901752990072,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6611253549472059,
+      "eval_accuracy_speedway bricks": 0.9867443373194453,
+      "eval_accuracy_steel": 0.902686168458545,
+      "eval_accuracy_tiles": 0.08690590330091769,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9801083856286206,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8922322447255788,
+      "eval_iou_concrete tiles": 0.9461721434234869,
+      "eval_iou_dark bricks": 0.8201741759032676,
+      "eval_iou_grass": 0.8339616011244929,
+      "eval_iou_red bricks": 0.8512382864792504,
+      "eval_iou_road": 0.9725738371376359,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4404466729112715,
+      "eval_iou_speedway bricks": 0.9758235017164926,
+      "eval_iou_steel": 0.8399631746518854,
+      "eval_iou_tiles": 0.08690590330091769,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08280381560325623,
+      "eval_mean_accuracy": 0.7107395868824153,
+      "eval_mean_iou": 0.6645846097694539,
+      "eval_overall_accuracy": 0.9785488976372613,
+      "eval_runtime": 4.1025,
+      "eval_samples_per_second": 8.775,
+      "eval_steps_per_second": 0.488,
+      "step": 2260
+    },
+    {
+      "epoch": 205.55,
+      "learning_rate": 6.199043062200958e-05,
+      "loss": 0.0429,
+      "step": 2261
+    },
+    {
+      "epoch": 205.64,
+      "learning_rate": 6.197129186602871e-05,
+      "loss": 0.0418,
+      "step": 2262
+    },
+    {
+      "epoch": 205.73,
+      "learning_rate": 6.195215311004785e-05,
+      "loss": 0.0498,
+      "step": 2263
+    },
+    {
+      "epoch": 205.82,
+      "learning_rate": 6.193301435406699e-05,
+      "loss": 0.0353,
+      "step": 2264
+    },
+    {
+      "epoch": 205.91,
+      "learning_rate": 6.191387559808613e-05,
+      "loss": 0.0389,
+      "step": 2265
+    },
+    {
+      "epoch": 206.0,
+      "learning_rate": 6.189473684210526e-05,
+      "loss": 0.0408,
+      "step": 2266
+    },
+    {
+      "epoch": 206.09,
+      "learning_rate": 6.187559808612441e-05,
+      "loss": 0.0472,
+      "step": 2267
+    },
+    {
+      "epoch": 206.18,
+      "learning_rate": 6.185645933014354e-05,
+      "loss": 0.0368,
+      "step": 2268
+    },
+    {
+      "epoch": 206.27,
+      "learning_rate": 6.183732057416269e-05,
+      "loss": 0.0449,
+      "step": 2269
+    },
+    {
+      "epoch": 206.36,
+      "learning_rate": 6.181818181818182e-05,
+      "loss": 0.0421,
+      "step": 2270
+    },
+    {
+      "epoch": 206.45,
+      "learning_rate": 6.179904306220095e-05,
+      "loss": 0.0412,
+      "step": 2271
+    },
+    {
+      "epoch": 206.55,
+      "learning_rate": 6.17799043062201e-05,
+      "loss": 0.038,
+      "step": 2272
+    },
+    {
+      "epoch": 206.64,
+      "learning_rate": 6.176076555023924e-05,
+      "loss": 0.0428,
+      "step": 2273
+    },
+    {
+      "epoch": 206.73,
+      "learning_rate": 6.174162679425837e-05,
+      "loss": 0.043,
+      "step": 2274
+    },
+    {
+      "epoch": 206.82,
+      "learning_rate": 6.172248803827752e-05,
+      "loss": 0.0434,
+      "step": 2275
+    },
+    {
+      "epoch": 206.91,
+      "learning_rate": 6.170334928229665e-05,
+      "loss": 0.0404,
+      "step": 2276
+    },
+    {
+      "epoch": 207.0,
+      "learning_rate": 6.168421052631578e-05,
+      "loss": 0.0418,
+      "step": 2277
+    },
+    {
+      "epoch": 207.09,
+      "learning_rate": 6.166507177033493e-05,
+      "loss": 0.0425,
+      "step": 2278
+    },
+    {
+      "epoch": 207.18,
+      "learning_rate": 6.164593301435407e-05,
+      "loss": 0.0374,
+      "step": 2279
+    },
+    {
+      "epoch": 207.27,
+      "learning_rate": 6.16267942583732e-05,
+      "loss": 0.0397,
+      "step": 2280
+    },
+    {
+      "epoch": 207.27,
+      "eval_accuracy_NAT": 0.9930107712819456,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9487115863341774,
+      "eval_accuracy_concrete tiles": 0.98217452455207,
+      "eval_accuracy_dark bricks": 0.8824140718217158,
+      "eval_accuracy_grass": 0.9043992492581706,
+      "eval_accuracy_red bricks": 0.9174360795454546,
+      "eval_accuracy_road": 0.9811363419255683,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6630551649987594,
+      "eval_accuracy_speedway bricks": 0.9860715376158392,
+      "eval_accuracy_steel": 0.899066684255656,
+      "eval_accuracy_tiles": 0.15559512395562253,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9799808276499766,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8932399172085281,
+      "eval_iou_concrete tiles": 0.9528654485049833,
+      "eval_iou_dark bricks": 0.8116260133819521,
+      "eval_iou_grass": 0.829807975987378,
+      "eval_iou_red bricks": 0.8423540919465276,
+      "eval_iou_road": 0.9739371023886688,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4363863991000472,
+      "eval_iou_speedway bricks": 0.9756046742931785,
+      "eval_iou_steel": 0.841014962208854,
+      "eval_iou_tiles": 0.15545672254533013,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08099492639303207,
+      "eval_mean_accuracy": 0.7163900873496137,
+      "eval_mean_iou": 0.668636471939648,
+      "eval_overall_accuracy": 0.9785671234130859,
+      "eval_runtime": 4.0424,
+      "eval_samples_per_second": 8.906,
+      "eval_steps_per_second": 0.495,
+      "step": 2280
+    },
+    {
+      "epoch": 207.36,
+      "learning_rate": 6.160765550239235e-05,
+      "loss": 0.0371,
+      "step": 2281
+    },
+    {
+      "epoch": 207.45,
+      "learning_rate": 6.158851674641148e-05,
+      "loss": 0.042,
+      "step": 2282
+    },
+    {
+      "epoch": 207.55,
+      "learning_rate": 6.156937799043062e-05,
+      "loss": 0.0392,
+      "step": 2283
+    },
+    {
+      "epoch": 207.64,
+      "learning_rate": 6.155023923444977e-05,
+      "loss": 0.0438,
+      "step": 2284
+    },
+    {
+      "epoch": 207.73,
+      "learning_rate": 6.15311004784689e-05,
+      "loss": 0.0421,
+      "step": 2285
+    },
+    {
+      "epoch": 207.82,
+      "learning_rate": 6.151196172248804e-05,
+      "loss": 0.0375,
+      "step": 2286
+    },
+    {
+      "epoch": 207.91,
+      "learning_rate": 6.149282296650718e-05,
+      "loss": 0.0441,
+      "step": 2287
+    },
+    {
+      "epoch": 208.0,
+      "learning_rate": 6.147368421052632e-05,
+      "loss": 0.039,
+      "step": 2288
+    },
+    {
+      "epoch": 208.09,
+      "learning_rate": 6.145454545454545e-05,
+      "loss": 0.0426,
+      "step": 2289
+    },
+    {
+      "epoch": 208.18,
+      "learning_rate": 6.143540669856461e-05,
+      "loss": 0.0376,
+      "step": 2290
+    },
+    {
+      "epoch": 208.27,
+      "learning_rate": 6.141626794258374e-05,
+      "loss": 0.0433,
+      "step": 2291
+    },
+    {
+      "epoch": 208.36,
+      "learning_rate": 6.139712918660287e-05,
+      "loss": 0.0335,
+      "step": 2292
+    },
+    {
+      "epoch": 208.45,
+      "learning_rate": 6.137799043062202e-05,
+      "loss": 0.0393,
+      "step": 2293
+    },
+    {
+      "epoch": 208.55,
+      "learning_rate": 6.135885167464115e-05,
+      "loss": 0.0425,
+      "step": 2294
+    },
+    {
+      "epoch": 208.64,
+      "learning_rate": 6.133971291866028e-05,
+      "loss": 0.0451,
+      "step": 2295
+    },
+    {
+      "epoch": 208.73,
+      "learning_rate": 6.132057416267944e-05,
+      "loss": 0.0433,
+      "step": 2296
+    },
+    {
+      "epoch": 208.82,
+      "learning_rate": 6.130143540669857e-05,
+      "loss": 0.0354,
+      "step": 2297
+    },
+    {
+      "epoch": 208.91,
+      "learning_rate": 6.12822966507177e-05,
+      "loss": 0.0429,
+      "step": 2298
+    },
+    {
+      "epoch": 209.0,
+      "learning_rate": 6.126315789473685e-05,
+      "loss": 0.0369,
+      "step": 2299
+    },
+    {
+      "epoch": 209.09,
+      "learning_rate": 6.124401913875598e-05,
+      "loss": 0.0402,
+      "step": 2300
+    },
+    {
+      "epoch": 209.09,
+      "eval_accuracy_NAT": 0.991681917924822,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9522847010601313,
+      "eval_accuracy_concrete tiles": 0.9703112578731731,
+      "eval_accuracy_dark bricks": 0.8850698587429275,
+      "eval_accuracy_grass": 0.9142798225900594,
+      "eval_accuracy_red bricks": 0.9035866477272727,
+      "eval_accuracy_road": 0.9810601617679601,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6982604140820996,
+      "eval_accuracy_speedway bricks": 0.9870412139599603,
+      "eval_accuracy_steel": 0.908284413956863,
+      "eval_accuracy_tiles": 0.022736611423092728,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9802348454008861,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8917101536645253,
+      "eval_iou_concrete tiles": 0.9557871276691865,
+      "eval_iou_dark bricks": 0.820780982295831,
+      "eval_iou_grass": 0.8308010391722193,
+      "eval_iou_red bricks": 0.8459109042553191,
+      "eval_iou_road": 0.9740731091498693,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4528355860687978,
+      "eval_iou_speedway bricks": 0.9758533014686318,
+      "eval_iou_steel": 0.8415439968832835,
+      "eval_iou_tiles": 0.022736611423092728,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0833815336227417,
+      "eval_mean_accuracy": 0.708815155469874,
+      "eval_mean_iou": 0.6609436659578187,
+      "eval_overall_accuracy": 0.9785207112630209,
+      "eval_runtime": 3.3309,
+      "eval_samples_per_second": 10.808,
+      "eval_steps_per_second": 0.6,
+      "step": 2300
+    },
+    {
+      "epoch": 209.18,
+      "learning_rate": 6.122488038277511e-05,
+      "loss": 0.0401,
+      "step": 2301
+    },
+    {
+      "epoch": 209.27,
+      "learning_rate": 6.120574162679427e-05,
+      "loss": 0.0403,
+      "step": 2302
+    },
+    {
+      "epoch": 209.36,
+      "learning_rate": 6.11866028708134e-05,
+      "loss": 0.0375,
+      "step": 2303
+    },
+    {
+      "epoch": 209.45,
+      "learning_rate": 6.116746411483254e-05,
+      "loss": 0.0382,
+      "step": 2304
+    },
+    {
+      "epoch": 209.55,
+      "learning_rate": 6.114832535885168e-05,
+      "loss": 0.0461,
+      "step": 2305
+    },
+    {
+      "epoch": 209.64,
+      "learning_rate": 6.112918660287081e-05,
+      "loss": 0.0349,
+      "step": 2306
+    },
+    {
+      "epoch": 209.73,
+      "learning_rate": 6.111004784688995e-05,
+      "loss": 0.0382,
+      "step": 2307
+    },
+    {
+      "epoch": 209.82,
+      "learning_rate": 6.10909090909091e-05,
+      "loss": 0.0393,
+      "step": 2308
+    },
+    {
+      "epoch": 209.91,
+      "learning_rate": 6.107177033492824e-05,
+      "loss": 0.0376,
+      "step": 2309
+    },
+    {
+      "epoch": 210.0,
+      "learning_rate": 6.105263157894737e-05,
+      "loss": 0.0396,
+      "step": 2310
+    },
+    {
+      "epoch": 210.09,
+      "learning_rate": 6.103349282296651e-05,
+      "loss": 0.0382,
+      "step": 2311
+    },
+    {
+      "epoch": 210.18,
+      "learning_rate": 6.1014354066985646e-05,
+      "loss": 0.0419,
+      "step": 2312
+    },
+    {
+      "epoch": 210.27,
+      "learning_rate": 6.0995215311004785e-05,
+      "loss": 0.0384,
+      "step": 2313
+    },
+    {
+      "epoch": 210.36,
+      "learning_rate": 6.097607655502393e-05,
+      "loss": 0.0395,
+      "step": 2314
+    },
+    {
+      "epoch": 210.45,
+      "learning_rate": 6.095693779904307e-05,
+      "loss": 0.0335,
+      "step": 2315
+    },
+    {
+      "epoch": 210.55,
+      "learning_rate": 6.09377990430622e-05,
+      "loss": 0.0398,
+      "step": 2316
+    },
+    {
+      "epoch": 210.64,
+      "learning_rate": 6.091866028708134e-05,
+      "loss": 0.0391,
+      "step": 2317
+    },
+    {
+      "epoch": 210.73,
+      "learning_rate": 6.089952153110048e-05,
+      "loss": 0.039,
+      "step": 2318
+    },
+    {
+      "epoch": 210.82,
+      "learning_rate": 6.088038277511962e-05,
+      "loss": 0.0329,
+      "step": 2319
+    },
+    {
+      "epoch": 210.91,
+      "learning_rate": 6.086124401913876e-05,
+      "loss": 0.038,
+      "step": 2320
+    },
+    {
+      "epoch": 210.91,
+      "eval_accuracy_NAT": 0.9931325225522172,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9488483947808919,
+      "eval_accuracy_concrete tiles": 0.9864245092643552,
+      "eval_accuracy_dark bricks": 0.8747931180478041,
+      "eval_accuracy_grass": 0.8902267254305057,
+      "eval_accuracy_red bricks": 0.90625,
+      "eval_accuracy_road": 0.9797300320319424,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.673421001847104,
+      "eval_accuracy_speedway bricks": 0.9883198351007147,
+      "eval_accuracy_steel": 0.8965602532814458,
+      "eval_accuracy_tiles": 0.030817696205999178,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9795849744242185,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8913313820826808,
+      "eval_iou_concrete tiles": 0.9433057512938218,
+      "eval_iou_dark bricks": 0.8225841476655809,
+      "eval_iou_grass": 0.8260385925085131,
+      "eval_iou_red bricks": 0.8498168498168498,
+      "eval_iou_road": 0.9734934547660957,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4438367613925431,
+      "eval_iou_speedway bricks": 0.9761292195064099,
+      "eval_iou_steel": 0.8421256427730623,
+      "eval_iou_tiles": 0.030817696205999178,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0861285850405693,
+      "eval_mean_accuracy": 0.7052710837340753,
+      "eval_mean_iou": 0.6599280363412134,
+      "eval_overall_accuracy": 0.9782945844862196,
+      "eval_runtime": 3.2187,
+      "eval_samples_per_second": 11.185,
+      "eval_steps_per_second": 0.621,
+      "step": 2320
+    },
+    {
+      "epoch": 211.0,
+      "learning_rate": 6.08421052631579e-05,
+      "loss": 0.0451,
+      "step": 2321
+    },
+    {
+      "epoch": 211.09,
+      "learning_rate": 6.0822966507177034e-05,
+      "loss": 0.0371,
+      "step": 2322
+    },
+    {
+      "epoch": 211.18,
+      "learning_rate": 6.080382775119617e-05,
+      "loss": 0.042,
+      "step": 2323
+    },
+    {
+      "epoch": 211.27,
+      "learning_rate": 6.078468899521531e-05,
+      "loss": 0.041,
+      "step": 2324
+    },
+    {
+      "epoch": 211.36,
+      "learning_rate": 6.076555023923445e-05,
+      "loss": 0.0405,
+      "step": 2325
+    },
+    {
+      "epoch": 211.45,
+      "learning_rate": 6.0746411483253596e-05,
+      "loss": 0.04,
+      "step": 2326
+    },
+    {
+      "epoch": 211.55,
+      "learning_rate": 6.0727272727272735e-05,
+      "loss": 0.0448,
+      "step": 2327
+    },
+    {
+      "epoch": 211.64,
+      "learning_rate": 6.070813397129187e-05,
+      "loss": 0.0387,
+      "step": 2328
+    },
+    {
+      "epoch": 211.73,
+      "learning_rate": 6.0688995215311006e-05,
+      "loss": 0.0461,
+      "step": 2329
+    },
+    {
+      "epoch": 211.82,
+      "learning_rate": 6.0669856459330144e-05,
+      "loss": 0.0354,
+      "step": 2330
+    },
+    {
+      "epoch": 211.91,
+      "learning_rate": 6.065071770334928e-05,
+      "loss": 0.0387,
+      "step": 2331
+    },
+    {
+      "epoch": 212.0,
+      "learning_rate": 6.063157894736843e-05,
+      "loss": 0.0462,
+      "step": 2332
+    },
+    {
+      "epoch": 212.09,
+      "learning_rate": 6.061244019138757e-05,
+      "loss": 0.0396,
+      "step": 2333
+    },
+    {
+      "epoch": 212.18,
+      "learning_rate": 6.05933014354067e-05,
+      "loss": 0.0385,
+      "step": 2334
+    },
+    {
+      "epoch": 212.27,
+      "learning_rate": 6.057416267942584e-05,
+      "loss": 0.0436,
+      "step": 2335
+    },
+    {
+      "epoch": 212.36,
+      "learning_rate": 6.055502392344498e-05,
+      "loss": 0.05,
+      "step": 2336
+    },
+    {
+      "epoch": 212.45,
+      "learning_rate": 6.0535885167464116e-05,
+      "loss": 0.0411,
+      "step": 2337
+    },
+    {
+      "epoch": 212.55,
+      "learning_rate": 6.051674641148326e-05,
+      "loss": 0.0376,
+      "step": 2338
+    },
+    {
+      "epoch": 212.64,
+      "learning_rate": 6.04976076555024e-05,
+      "loss": 0.0377,
+      "step": 2339
+    },
+    {
+      "epoch": 212.73,
+      "learning_rate": 6.047846889952153e-05,
+      "loss": 0.0377,
+      "step": 2340
+    },
+    {
+      "epoch": 212.73,
+      "eval_accuracy_NAT": 0.9925361177790027,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9463482874801493,
+      "eval_accuracy_concrete tiles": 0.9850791903626246,
+      "eval_accuracy_dark bricks": 0.8929217505099881,
+      "eval_accuracy_grass": 0.897370656060283,
+      "eval_accuracy_red bricks": 0.9202769886363636,
+      "eval_accuracy_road": 0.9845221267081586,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.7124858710335511,
+      "eval_accuracy_speedway bricks": 0.9882963021962837,
+      "eval_accuracy_steel": 0.8964530703779434,
+      "eval_accuracy_tiles": 0.12505136282701,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9800173737727116,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.894729900923304,
+      "eval_iou_concrete tiles": 0.9416613082363945,
+      "eval_iou_dark bricks": 0.8274423083782145,
+      "eval_iou_grass": 0.8285455202591938,
+      "eval_iou_red bricks": 0.8473107732548635,
+      "eval_iou_road": 0.9744102521754636,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4608416547788873,
+      "eval_iou_speedway bricks": 0.9759297460336754,
+      "eval_iou_steel": 0.8396515641772142,
+      "eval_iou_tiles": 0.12495722986381988,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08102604001760483,
+      "eval_mean_accuracy": 0.7185647479977969,
+      "eval_mean_iou": 0.6688844332195186,
+      "eval_overall_accuracy": 0.978719499376085,
+      "eval_runtime": 3.7611,
+      "eval_samples_per_second": 9.572,
+      "eval_steps_per_second": 0.532,
+      "step": 2340
+    },
+    {
+      "epoch": 212.82,
+      "learning_rate": 6.045933014354067e-05,
+      "loss": 0.0395,
+      "step": 2341
+    },
+    {
+      "epoch": 212.91,
+      "learning_rate": 6.044019138755981e-05,
+      "loss": 0.0405,
+      "step": 2342
+    },
+    {
+      "epoch": 213.0,
+      "learning_rate": 6.042105263157894e-05,
+      "loss": 0.0388,
+      "step": 2343
+    },
+    {
+      "epoch": 213.09,
+      "learning_rate": 6.0401913875598094e-05,
+      "loss": 0.0409,
+      "step": 2344
+    },
+    {
+      "epoch": 213.18,
+      "learning_rate": 6.038277511961723e-05,
+      "loss": 0.034,
+      "step": 2345
+    },
+    {
+      "epoch": 213.27,
+      "learning_rate": 6.0363636363636365e-05,
+      "loss": 0.0386,
+      "step": 2346
+    },
+    {
+      "epoch": 213.36,
+      "learning_rate": 6.0344497607655504e-05,
+      "loss": 0.0414,
+      "step": 2347
+    },
+    {
+      "epoch": 213.45,
+      "learning_rate": 6.032535885167464e-05,
+      "loss": 0.043,
+      "step": 2348
+    },
+    {
+      "epoch": 213.55,
+      "learning_rate": 6.0306220095693774e-05,
+      "loss": 0.0401,
+      "step": 2349
+    },
+    {
+      "epoch": 213.64,
+      "learning_rate": 6.028708133971293e-05,
+      "loss": 0.0387,
+      "step": 2350
+    },
+    {
+      "epoch": 213.73,
+      "learning_rate": 6.0267942583732065e-05,
+      "loss": 0.038,
+      "step": 2351
+    },
+    {
+      "epoch": 213.82,
+      "learning_rate": 6.02488038277512e-05,
+      "loss": 0.0365,
+      "step": 2352
+    },
+    {
+      "epoch": 213.91,
+      "learning_rate": 6.0229665071770336e-05,
+      "loss": 0.04,
+      "step": 2353
+    },
+    {
+      "epoch": 214.0,
+      "learning_rate": 6.0210526315789475e-05,
+      "loss": 0.0384,
+      "step": 2354
+    },
+    {
+      "epoch": 214.09,
+      "learning_rate": 6.019138755980862e-05,
+      "loss": 0.0427,
+      "step": 2355
+    },
+    {
+      "epoch": 214.18,
+      "learning_rate": 6.017224880382776e-05,
+      "loss": 0.0386,
+      "step": 2356
+    },
+    {
+      "epoch": 214.27,
+      "learning_rate": 6.015311004784689e-05,
+      "loss": 0.04,
+      "step": 2357
+    },
+    {
+      "epoch": 214.36,
+      "learning_rate": 6.013397129186603e-05,
+      "loss": 0.036,
+      "step": 2358
+    },
+    {
+      "epoch": 214.45,
+      "learning_rate": 6.011483253588517e-05,
+      "loss": 0.0407,
+      "step": 2359
+    },
+    {
+      "epoch": 214.55,
+      "learning_rate": 6.009569377990431e-05,
+      "loss": 0.0402,
+      "step": 2360
+    },
+    {
+      "epoch": 214.55,
+      "eval_accuracy_NAT": 0.9926700441763014,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.94890875144856,
+      "eval_accuracy_concrete tiles": 0.98217452455207,
+      "eval_accuracy_dark bricks": 0.8757938493514491,
+      "eval_accuracy_grass": 0.8974964787132537,
+      "eval_accuracy_red bricks": 0.8986150568181818,
+      "eval_accuracy_road": 0.981755456857242,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6762605794943898,
+      "eval_accuracy_speedway bricks": 0.989087973237657,
+      "eval_accuracy_steel": 0.8951503858584526,
+      "eval_accuracy_tiles": 0.039720586221065605,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9800790892282519,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8913016090587031,
+      "eval_iou_concrete tiles": 0.9421892415087699,
+      "eval_iou_dark bricks": 0.8261564156560889,
+      "eval_iou_grass": 0.8290978009240575,
+      "eval_iou_red bricks": 0.8478807170380298,
+      "eval_iou_road": 0.9752282822556041,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4472115367085377,
+      "eval_iou_speedway bricks": 0.9758172440397382,
+      "eval_iou_steel": 0.8401247369072675,
+      "eval_iou_tiles": 0.039720586221065605,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08310440182685852,
+      "eval_mean_accuracy": 0.7059718220560479,
+      "eval_mean_iou": 0.6611390199650857,
+      "eval_overall_accuracy": 0.9785444471571181,
+      "eval_runtime": 3.498,
+      "eval_samples_per_second": 10.292,
+      "eval_steps_per_second": 0.572,
+      "step": 2360
+    },
+    {
+      "epoch": 214.64,
+      "learning_rate": 6.007655502392345e-05,
+      "loss": 0.0394,
+      "step": 2361
+    },
+    {
+      "epoch": 214.73,
+      "learning_rate": 6.005741626794259e-05,
+      "loss": 0.0414,
+      "step": 2362
+    },
+    {
+      "epoch": 214.82,
+      "learning_rate": 6.0038277511961724e-05,
+      "loss": 0.0348,
+      "step": 2363
+    },
+    {
+      "epoch": 214.91,
+      "learning_rate": 6.001913875598086e-05,
+      "loss": 0.0392,
+      "step": 2364
+    },
+    {
+      "epoch": 215.0,
+      "learning_rate": 6e-05,
+      "loss": 0.0433,
+      "step": 2365
+    },
+    {
+      "epoch": 215.09,
+      "learning_rate": 5.998086124401914e-05,
+      "loss": 0.035,
+      "step": 2366
+    },
+    {
+      "epoch": 215.18,
+      "learning_rate": 5.9961722488038286e-05,
+      "loss": 0.039,
+      "step": 2367
+    },
+    {
+      "epoch": 215.27,
+      "learning_rate": 5.9942583732057425e-05,
+      "loss": 0.0329,
+      "step": 2368
+    },
+    {
+      "epoch": 215.36,
+      "learning_rate": 5.992344497607656e-05,
+      "loss": 0.0421,
+      "step": 2369
+    },
+    {
+      "epoch": 215.45,
+      "learning_rate": 5.9904306220095695e-05,
+      "loss": 0.0372,
+      "step": 2370
+    },
+    {
+      "epoch": 215.55,
+      "learning_rate": 5.9885167464114834e-05,
+      "loss": 0.0393,
+      "step": 2371
+    },
+    {
+      "epoch": 215.64,
+      "learning_rate": 5.986602870813397e-05,
+      "loss": 0.0374,
+      "step": 2372
+    },
+    {
+      "epoch": 215.73,
+      "learning_rate": 5.984688995215312e-05,
+      "loss": 0.0344,
+      "step": 2373
+    },
+    {
+      "epoch": 215.82,
+      "learning_rate": 5.982775119617226e-05,
+      "loss": 0.0411,
+      "step": 2374
+    },
+    {
+      "epoch": 215.91,
+      "learning_rate": 5.980861244019139e-05,
+      "loss": 0.0418,
+      "step": 2375
+    },
+    {
+      "epoch": 216.0,
+      "learning_rate": 5.978947368421053e-05,
+      "loss": 0.0433,
+      "step": 2376
+    },
+    {
+      "epoch": 216.09,
+      "learning_rate": 5.977033492822967e-05,
+      "loss": 0.035,
+      "step": 2377
+    },
+    {
+      "epoch": 216.18,
+      "learning_rate": 5.9751196172248806e-05,
+      "loss": 0.0385,
+      "step": 2378
+    },
+    {
+      "epoch": 216.27,
+      "learning_rate": 5.973205741626795e-05,
+      "loss": 0.0398,
+      "step": 2379
+    },
+    {
+      "epoch": 216.36,
+      "learning_rate": 5.971291866028709e-05,
+      "loss": 0.0353,
+      "step": 2380
+    },
+    {
+      "epoch": 216.36,
+      "eval_accuracy_NAT": 0.9933837889418791,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9492655264174428,
+      "eval_accuracy_concrete tiles": 0.9858130006726594,
+      "eval_accuracy_dark bricks": 0.8682498749085871,
+      "eval_accuracy_grass": 0.8897444052607849,
+      "eval_accuracy_red bricks": 0.9112215909090909,
+      "eval_accuracy_road": 0.9806308926258817,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6590852700355636,
+      "eval_accuracy_speedway bricks": 0.9871172433435068,
+      "eval_accuracy_steel": 0.9061902249192005,
+      "eval_accuracy_tiles": 0.05677304478838515,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9796474776991615,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8927175373204926,
+      "eval_iou_concrete tiles": 0.9374854617352872,
+      "eval_iou_dark bricks": 0.8177039910102585,
+      "eval_iou_grass": 0.8266526818940491,
+      "eval_iou_red bricks": 0.8377407770159974,
+      "eval_iou_road": 0.9748149163201795,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4388135313228465,
+      "eval_iou_speedway bricks": 0.9760676366261638,
+      "eval_iou_steel": 0.8409720415627343,
+      "eval_iou_tiles": 0.05677304478838515,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0833871066570282,
+      "eval_mean_accuracy": 0.7067288356017678,
+      "eval_mean_iou": 0.6599530074842735,
+      "eval_overall_accuracy": 0.9784232245551215,
+      "eval_runtime": 5.1709,
+      "eval_samples_per_second": 6.962,
+      "eval_steps_per_second": 0.387,
+      "step": 2380
+    },
+    {
+      "epoch": 216.45,
+      "learning_rate": 5.969377990430622e-05,
+      "loss": 0.0369,
+      "step": 2381
+    },
+    {
+      "epoch": 216.55,
+      "learning_rate": 5.967464114832536e-05,
+      "loss": 0.0364,
+      "step": 2382
+    },
+    {
+      "epoch": 216.64,
+      "learning_rate": 5.96555023923445e-05,
+      "loss": 0.0377,
+      "step": 2383
+    },
+    {
+      "epoch": 216.73,
+      "learning_rate": 5.963636363636363e-05,
+      "loss": 0.0393,
+      "step": 2384
+    },
+    {
+      "epoch": 216.82,
+      "learning_rate": 5.9617224880382784e-05,
+      "loss": 0.0395,
+      "step": 2385
+    },
+    {
+      "epoch": 216.91,
+      "learning_rate": 5.959808612440192e-05,
+      "loss": 0.0364,
+      "step": 2386
+    },
+    {
+      "epoch": 217.0,
+      "learning_rate": 5.9578947368421055e-05,
+      "loss": 0.0417,
+      "step": 2387
+    },
+    {
+      "epoch": 217.09,
+      "learning_rate": 5.9559808612440193e-05,
+      "loss": 0.0424,
+      "step": 2388
+    },
+    {
+      "epoch": 217.18,
+      "learning_rate": 5.954066985645933e-05,
+      "loss": 0.0365,
+      "step": 2389
+    },
+    {
+      "epoch": 217.27,
+      "learning_rate": 5.9521531100478464e-05,
+      "loss": 0.0372,
+      "step": 2390
+    },
+    {
+      "epoch": 217.36,
+      "learning_rate": 5.9502392344497617e-05,
+      "loss": 0.0394,
+      "step": 2391
+    },
+    {
+      "epoch": 217.45,
+      "learning_rate": 5.9483253588516755e-05,
+      "loss": 0.0427,
+      "step": 2392
+    },
+    {
+      "epoch": 217.55,
+      "learning_rate": 5.946411483253589e-05,
+      "loss": 0.0419,
+      "step": 2393
+    },
+    {
+      "epoch": 217.64,
+      "learning_rate": 5.9444976076555026e-05,
+      "loss": 0.04,
+      "step": 2394
+    },
+    {
+      "epoch": 217.73,
+      "learning_rate": 5.9425837320574165e-05,
+      "loss": 0.0379,
+      "step": 2395
+    },
+    {
+      "epoch": 217.82,
+      "learning_rate": 5.94066985645933e-05,
+      "loss": 0.0378,
+      "step": 2396
+    },
+    {
+      "epoch": 217.91,
+      "learning_rate": 5.938755980861245e-05,
+      "loss": 0.0339,
+      "step": 2397
+    },
+    {
+      "epoch": 218.0,
+      "learning_rate": 5.936842105263158e-05,
+      "loss": 0.05,
+      "step": 2398
+    },
+    {
+      "epoch": 218.09,
+      "learning_rate": 5.934928229665072e-05,
+      "loss": 0.0428,
+      "step": 2399
+    },
+    {
+      "epoch": 218.18,
+      "learning_rate": 5.933014354066986e-05,
+      "loss": 0.0385,
+      "step": 2400
+    },
+    {
+      "epoch": 218.18,
+      "eval_accuracy_NAT": 0.9906740291483128,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9540900360530495,
+      "eval_accuracy_concrete tiles": 0.993151103773008,
+      "eval_accuracy_dark bricks": 0.8853007967360763,
+      "eval_accuracy_grass": 0.9213223960827214,
+      "eval_accuracy_red bricks": 0.9176136363636364,
+      "eval_accuracy_road": 0.9827905396336339,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6841176632757148,
+      "eval_accuracy_speedway bricks": 0.9842263165453213,
+      "eval_accuracy_steel": 0.9103043994459469,
+      "eval_accuracy_tiles": 0.06341597041501164,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9800500180930248,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8876602892069022,
+      "eval_iou_concrete tiles": 0.8902104801578601,
+      "eval_iou_dark bricks": 0.8176098393288781,
+      "eval_iou_grass": 0.8297037244539849,
+      "eval_iou_red bricks": 0.82688,
+      "eval_iou_road": 0.9736169692816902,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4481506898793614,
+      "eval_iou_speedway bricks": 0.9754194581205669,
+      "eval_iou_steel": 0.8416025733864883,
+      "eval_iou_tiles": 0.06341597041501164,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08436784148216248,
+      "eval_mean_accuracy": 0.7143851451901871,
+      "eval_mean_iou": 0.6564861547941361,
+      "eval_overall_accuracy": 0.9780516094631619,
+      "eval_runtime": 3.6563,
+      "eval_samples_per_second": 9.846,
+      "eval_steps_per_second": 0.547,
+      "step": 2400
+    },
+    {
+      "epoch": 218.27,
+      "learning_rate": 5.9311004784689e-05,
+      "loss": 0.0398,
+      "step": 2401
+    },
+    {
+      "epoch": 218.36,
+      "learning_rate": 5.929186602870813e-05,
+      "loss": 0.035,
+      "step": 2402
+    },
+    {
+      "epoch": 218.45,
+      "learning_rate": 5.927272727272728e-05,
+      "loss": 0.0396,
+      "step": 2403
+    },
+    {
+      "epoch": 218.55,
+      "learning_rate": 5.9253588516746414e-05,
+      "loss": 0.0371,
+      "step": 2404
+    },
+    {
+      "epoch": 218.64,
+      "learning_rate": 5.923444976076555e-05,
+      "loss": 0.0362,
+      "step": 2405
+    },
+    {
+      "epoch": 218.73,
+      "learning_rate": 5.921531100478469e-05,
+      "loss": 0.0426,
+      "step": 2406
+    },
+    {
+      "epoch": 218.82,
+      "learning_rate": 5.919617224880383e-05,
+      "loss": 0.049,
+      "step": 2407
+    },
+    {
+      "epoch": 218.91,
+      "learning_rate": 5.917703349282296e-05,
+      "loss": 0.0362,
+      "step": 2408
+    },
+    {
+      "epoch": 219.0,
+      "learning_rate": 5.9157894736842114e-05,
+      "loss": 0.0381,
+      "step": 2409
+    },
+    {
+      "epoch": 219.09,
+      "learning_rate": 5.9138755980861246e-05,
+      "loss": 0.0358,
+      "step": 2410
+    },
+    {
+      "epoch": 219.18,
+      "learning_rate": 5.9119617224880385e-05,
+      "loss": 0.0435,
+      "step": 2411
+    },
+    {
+      "epoch": 219.27,
+      "learning_rate": 5.9100478468899524e-05,
+      "loss": 0.0374,
+      "step": 2412
+    },
+    {
+      "epoch": 219.36,
+      "learning_rate": 5.908133971291866e-05,
+      "loss": 0.0357,
+      "step": 2413
+    },
+    {
+      "epoch": 219.45,
+      "learning_rate": 5.9062200956937795e-05,
+      "loss": 0.0393,
+      "step": 2414
+    },
+    {
+      "epoch": 219.55,
+      "learning_rate": 5.904306220095695e-05,
+      "loss": 0.0422,
+      "step": 2415
+    },
+    {
+      "epoch": 219.64,
+      "learning_rate": 5.902392344497608e-05,
+      "loss": 0.0406,
+      "step": 2416
+    },
+    {
+      "epoch": 219.73,
+      "learning_rate": 5.900478468899522e-05,
+      "loss": 0.0392,
+      "step": 2417
+    },
+    {
+      "epoch": 219.82,
+      "learning_rate": 5.898564593301436e-05,
+      "loss": 0.0383,
+      "step": 2418
+    },
+    {
+      "epoch": 219.91,
+      "learning_rate": 5.896650717703349e-05,
+      "loss": 0.038,
+      "step": 2419
+    },
+    {
+      "epoch": 220.0,
+      "learning_rate": 5.894736842105263e-05,
+      "loss": 0.0391,
+      "step": 2420
+    },
+    {
+      "epoch": 220.0,
+      "eval_accuracy_NAT": 0.9920738158542034,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9436456500278982,
+      "eval_accuracy_concrete tiles": 0.9856295480951507,
+      "eval_accuracy_dark bricks": 0.8955005581001501,
+      "eval_accuracy_grass": 0.9143252585480764,
+      "eval_accuracy_red bricks": 0.9245383522727273,
+      "eval_accuracy_road": 0.9836720528859583,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6555013370826785,
+      "eval_accuracy_speedway bricks": 0.98925632401551,
+      "eval_accuracy_steel": 0.8992563155464679,
+      "eval_accuracy_tiles": 0.04985618408437201,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9800723099779995,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8911835715136207,
+      "eval_iou_concrete tiles": 0.941142123087703,
+      "eval_iou_dark bricks": 0.8187352641024739,
+      "eval_iou_grass": 0.832879119759564,
+      "eval_iou_red bricks": 0.841331394409436,
+      "eval_iou_road": 0.9755162783112143,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4360352099761599,
+      "eval_iou_speedway bricks": 0.9755309356165256,
+      "eval_iou_steel": 0.8426676349926217,
+      "eval_iou_tiles": 0.04985618408437201,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08479706943035126,
+      "eval_mean_accuracy": 0.7102504151163995,
+      "eval_mean_iou": 0.6603807712178222,
+      "eval_overall_accuracy": 0.9785486857096354,
+      "eval_runtime": 3.2998,
+      "eval_samples_per_second": 10.91,
+      "eval_steps_per_second": 0.606,
+      "step": 2420
+    },
+    {
+      "epoch": 220.09,
+      "learning_rate": 5.892822966507178e-05,
+      "loss": 0.0338,
+      "step": 2421
+    },
+    {
+      "epoch": 220.18,
+      "learning_rate": 5.890909090909091e-05,
+      "loss": 0.0373,
+      "step": 2422
+    },
+    {
+      "epoch": 220.27,
+      "learning_rate": 5.888995215311005e-05,
+      "loss": 0.0351,
+      "step": 2423
+    },
+    {
+      "epoch": 220.36,
+      "learning_rate": 5.887081339712919e-05,
+      "loss": 0.0363,
+      "step": 2424
+    },
+    {
+      "epoch": 220.45,
+      "learning_rate": 5.885167464114832e-05,
+      "loss": 0.0402,
+      "step": 2425
+    },
+    {
+      "epoch": 220.55,
+      "learning_rate": 5.8832535885167474e-05,
+      "loss": 0.0371,
+      "step": 2426
+    },
+    {
+      "epoch": 220.64,
+      "learning_rate": 5.881339712918661e-05,
+      "loss": 0.0481,
+      "step": 2427
+    },
+    {
+      "epoch": 220.73,
+      "learning_rate": 5.8794258373205744e-05,
+      "loss": 0.0364,
+      "step": 2428
+    },
+    {
+      "epoch": 220.82,
+      "learning_rate": 5.877511961722488e-05,
+      "loss": 0.0379,
+      "step": 2429
+    },
+    {
+      "epoch": 220.91,
+      "learning_rate": 5.875598086124402e-05,
+      "loss": 0.0461,
+      "step": 2430
+    },
+    {
+      "epoch": 221.0,
+      "learning_rate": 5.8736842105263154e-05,
+      "loss": 0.0444,
+      "step": 2431
+    },
+    {
+      "epoch": 221.09,
+      "learning_rate": 5.8717703349282306e-05,
+      "loss": 0.0372,
+      "step": 2432
+    },
+    {
+      "epoch": 221.18,
+      "learning_rate": 5.869856459330144e-05,
+      "loss": 0.0344,
+      "step": 2433
+    },
+    {
+      "epoch": 221.27,
+      "learning_rate": 5.867942583732058e-05,
+      "loss": 0.0385,
+      "step": 2434
+    },
+    {
+      "epoch": 221.36,
+      "learning_rate": 5.8660287081339716e-05,
+      "loss": 0.0425,
+      "step": 2435
+    },
+    {
+      "epoch": 221.45,
+      "learning_rate": 5.8641148325358855e-05,
+      "loss": 0.0371,
+      "step": 2436
+    },
+    {
+      "epoch": 221.55,
+      "learning_rate": 5.862200956937799e-05,
+      "loss": 0.0429,
+      "step": 2437
+    },
+    {
+      "epoch": 221.64,
+      "learning_rate": 5.860287081339714e-05,
+      "loss": 0.0415,
+      "step": 2438
+    },
+    {
+      "epoch": 221.73,
+      "learning_rate": 5.858373205741627e-05,
+      "loss": 0.0375,
+      "step": 2439
+    },
+    {
+      "epoch": 221.82,
+      "learning_rate": 5.856459330143541e-05,
+      "loss": 0.0377,
+      "step": 2440
+    },
+    {
+      "epoch": 221.82,
+      "eval_accuracy_NAT": 0.9936154692576278,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9506564122923731,
+      "eval_accuracy_concrete tiles": 0.9817770439674678,
+      "eval_accuracy_dark bricks": 0.8679419575843886,
+      "eval_accuracy_grass": 0.8850714917324032,
+      "eval_accuracy_red bricks": 0.9037642045454546,
+      "eval_accuracy_road": 0.9830722852958994,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6504286935185951,
+      "eval_accuracy_speedway bricks": 0.9874865289207327,
+      "eval_accuracy_steel": 0.8976403271552008,
+      "eval_accuracy_tiles": 0.04896589508286536,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9797462139309524,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8928824277849863,
+      "eval_iou_concrete tiles": 0.9432465777568886,
+      "eval_iou_dark bricks": 0.8273711245643002,
+      "eval_iou_grass": 0.8294459638721934,
+      "eval_iou_red bricks": 0.8521680897371505,
+      "eval_iou_road": 0.975685510196244,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4344295499742211,
+      "eval_iou_speedway bricks": 0.976282502909729,
+      "eval_iou_steel": 0.8440225438589691,
+      "eval_iou_tiles": 0.04896589508286536,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08422482013702393,
+      "eval_mean_accuracy": 0.7038784853348468,
+      "eval_mean_iou": 0.6618651076668077,
+      "eval_overall_accuracy": 0.9786348342895508,
+      "eval_runtime": 3.4338,
+      "eval_samples_per_second": 10.484,
+      "eval_steps_per_second": 0.582,
+      "step": 2440
+    },
+    {
+      "epoch": 221.91,
+      "learning_rate": 5.854545454545455e-05,
+      "loss": 0.0355,
+      "step": 2441
+    },
+    {
+      "epoch": 222.0,
+      "learning_rate": 5.852631578947369e-05,
+      "loss": 0.0357,
+      "step": 2442
+    },
+    {
+      "epoch": 222.09,
+      "learning_rate": 5.850717703349282e-05,
+      "loss": 0.0377,
+      "step": 2443
+    },
+    {
+      "epoch": 222.18,
+      "learning_rate": 5.848803827751197e-05,
+      "loss": 0.0362,
+      "step": 2444
+    },
+    {
+      "epoch": 222.27,
+      "learning_rate": 5.8468899521531104e-05,
+      "loss": 0.0361,
+      "step": 2445
+    },
+    {
+      "epoch": 222.36,
+      "learning_rate": 5.844976076555024e-05,
+      "loss": 0.0477,
+      "step": 2446
+    },
+    {
+      "epoch": 222.45,
+      "learning_rate": 5.843062200956938e-05,
+      "loss": 0.0328,
+      "step": 2447
+    },
+    {
+      "epoch": 222.55,
+      "learning_rate": 5.841148325358852e-05,
+      "loss": 0.0382,
+      "step": 2448
+    },
+    {
+      "epoch": 222.64,
+      "learning_rate": 5.839234449760765e-05,
+      "loss": 0.0384,
+      "step": 2449
+    },
+    {
+      "epoch": 222.73,
+      "learning_rate": 5.8373205741626804e-05,
+      "loss": 0.0324,
+      "step": 2450
+    },
+    {
+      "epoch": 222.82,
+      "learning_rate": 5.8354066985645936e-05,
+      "loss": 0.0362,
+      "step": 2451
+    },
+    {
+      "epoch": 222.91,
+      "learning_rate": 5.8334928229665075e-05,
+      "loss": 0.0388,
+      "step": 2452
+    },
+    {
+      "epoch": 223.0,
+      "learning_rate": 5.8315789473684214e-05,
+      "loss": 0.0449,
+      "step": 2453
+    },
+    {
+      "epoch": 223.09,
+      "learning_rate": 5.829665071770335e-05,
+      "loss": 0.0391,
+      "step": 2454
+    },
+    {
+      "epoch": 223.18,
+      "learning_rate": 5.8277511961722485e-05,
+      "loss": 0.0334,
+      "step": 2455
+    },
+    {
+      "epoch": 223.27,
+      "learning_rate": 5.825837320574164e-05,
+      "loss": 0.0335,
+      "step": 2456
+    },
+    {
+      "epoch": 223.36,
+      "learning_rate": 5.823923444976077e-05,
+      "loss": 0.0413,
+      "step": 2457
+    },
+    {
+      "epoch": 223.45,
+      "learning_rate": 5.822009569377991e-05,
+      "loss": 0.0402,
+      "step": 2458
+    },
+    {
+      "epoch": 223.55,
+      "learning_rate": 5.8200956937799047e-05,
+      "loss": 0.0348,
+      "step": 2459
+    },
+    {
+      "epoch": 223.64,
+      "learning_rate": 5.818181818181818e-05,
+      "loss": 0.0341,
+      "step": 2460
+    },
+    {
+      "epoch": 223.64,
+      "eval_accuracy_NAT": 0.99306882369922,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9523088437271986,
+      "eval_accuracy_concrete tiles": 0.9839478994679876,
+      "eval_accuracy_dark bricks": 0.8779107809553135,
+      "eval_accuracy_grass": 0.8955951586239196,
+      "eval_accuracy_red bricks": 0.9279119318181818,
+      "eval_accuracy_road": 0.9813273969240145,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6720425660959942,
+      "eval_accuracy_speedway bricks": 0.9865132321297762,
+      "eval_accuracy_steel": 0.9035271420091023,
+      "eval_accuracy_tiles": 0.006848376934666484,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9801028026218221,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.891032213452872,
+      "eval_iou_concrete tiles": 0.9490680665329716,
+      "eval_iou_dark bricks": 0.8228354978354978,
+      "eval_iou_grass": 0.8312248194787754,
+      "eval_iou_red bricks": 0.8179683831585538,
+      "eval_iou_road": 0.9746789682787511,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.44206078630494705,
+      "eval_iou_speedway bricks": 0.9764060035511037,
+      "eval_iou_steel": 0.8448941829536255,
+      "eval_iou_tiles": 0.006846970215679562,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08620060235261917,
+      "eval_mean_accuracy": 0.706230934798875,
+      "eval_mean_iou": 0.6567014380295847,
+      "eval_overall_accuracy": 0.9785752826266818,
+      "eval_runtime": 3.3388,
+      "eval_samples_per_second": 10.782,
+      "eval_steps_per_second": 0.599,
+      "step": 2460
+    },
+    {
+      "epoch": 223.73,
+      "learning_rate": 5.816267942583732e-05,
+      "loss": 0.0397,
+      "step": 2461
+    },
+    {
+      "epoch": 223.82,
+      "learning_rate": 5.814354066985647e-05,
+      "loss": 0.0409,
+      "step": 2462
+    },
+    {
+      "epoch": 223.91,
+      "learning_rate": 5.81244019138756e-05,
+      "loss": 0.0379,
+      "step": 2463
+    },
+    {
+      "epoch": 224.0,
+      "learning_rate": 5.810526315789474e-05,
+      "loss": 0.0387,
+      "step": 2464
+    },
+    {
+      "epoch": 224.09,
+      "learning_rate": 5.808612440191388e-05,
+      "loss": 0.039,
+      "step": 2465
+    },
+    {
+      "epoch": 224.18,
+      "learning_rate": 5.806698564593301e-05,
+      "loss": 0.0476,
+      "step": 2466
+    },
+    {
+      "epoch": 224.27,
+      "learning_rate": 5.804784688995215e-05,
+      "loss": 0.0377,
+      "step": 2467
+    },
+    {
+      "epoch": 224.36,
+      "learning_rate": 5.80287081339713e-05,
+      "loss": 0.0395,
+      "step": 2468
+    },
+    {
+      "epoch": 224.45,
+      "learning_rate": 5.8009569377990434e-05,
+      "loss": 0.0407,
+      "step": 2469
+    },
+    {
+      "epoch": 224.55,
+      "learning_rate": 5.799043062200957e-05,
+      "loss": 0.0362,
+      "step": 2470
+    },
+    {
+      "epoch": 224.64,
+      "learning_rate": 5.797129186602871e-05,
+      "loss": 0.0357,
+      "step": 2471
+    },
+    {
+      "epoch": 224.73,
+      "learning_rate": 5.7952153110047844e-05,
+      "loss": 0.0391,
+      "step": 2472
+    },
+    {
+      "epoch": 224.82,
+      "learning_rate": 5.793301435406698e-05,
+      "loss": 0.038,
+      "step": 2473
+    },
+    {
+      "epoch": 224.91,
+      "learning_rate": 5.791387559808613e-05,
+      "loss": 0.0362,
+      "step": 2474
+    },
+    {
+      "epoch": 225.0,
+      "learning_rate": 5.789473684210527e-05,
+      "loss": 0.0423,
+      "step": 2475
+    },
+    {
+      "epoch": 225.09,
+      "learning_rate": 5.7875598086124406e-05,
+      "loss": 0.0373,
+      "step": 2476
+    },
+    {
+      "epoch": 225.18,
+      "learning_rate": 5.7856459330143545e-05,
+      "loss": 0.0391,
+      "step": 2477
+    },
+    {
+      "epoch": 225.27,
+      "learning_rate": 5.7837320574162677e-05,
+      "loss": 0.0392,
+      "step": 2478
+    },
+    {
+      "epoch": 225.36,
+      "learning_rate": 5.7818181818181815e-05,
+      "loss": 0.0371,
+      "step": 2479
+    },
+    {
+      "epoch": 225.45,
+      "learning_rate": 5.779904306220096e-05,
+      "loss": 0.0376,
+      "step": 2480
+    },
+    {
+      "epoch": 225.45,
+      "eval_accuracy_NAT": 0.9925797012047376,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9508696725181338,
+      "eval_accuracy_concrete tiles": 0.9901852871032838,
+      "eval_accuracy_dark bricks": 0.8852623070705515,
+      "eval_accuracy_grass": 0.898209473746754,
+      "eval_accuracy_red bricks": 0.896484375,
+      "eval_accuracy_road": 0.982303228466711,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6883632453891324,
+      "eval_accuracy_speedway bricks": 0.9879783062825614,
+      "eval_accuracy_steel": 0.9003611239364158,
+      "eval_accuracy_tiles": 0.011573757019586357,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9801210280431639,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8917425468834709,
+      "eval_iou_concrete tiles": 0.9339851185326181,
+      "eval_iou_dark bricks": 0.8258527827648114,
+      "eval_iou_grass": 0.8317921563423452,
+      "eval_iou_red bricks": 0.8455870038519511,
+      "eval_iou_road": 0.9744147048874751,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4519603229193064,
+      "eval_iou_speedway bricks": 0.9764716126058053,
+      "eval_iou_steel": 0.8455451370875952,
+      "eval_iou_tiles": 0.011573757019586357,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0835827887058258,
+      "eval_mean_accuracy": 0.7064746521336822,
+      "eval_mean_iou": 0.6591573977644714,
+      "eval_overall_accuracy": 0.9786423577202691,
+      "eval_runtime": 4.1886,
+      "eval_samples_per_second": 8.595,
+      "eval_steps_per_second": 0.477,
+      "step": 2480
+    },
+    {
+      "epoch": 225.55,
+      "learning_rate": 5.77799043062201e-05,
+      "loss": 0.0415,
+      "step": 2481
+    },
+    {
+      "epoch": 225.64,
+      "learning_rate": 5.776076555023924e-05,
+      "loss": 0.0365,
+      "step": 2482
+    },
+    {
+      "epoch": 225.73,
+      "learning_rate": 5.774162679425838e-05,
+      "loss": 0.0361,
+      "step": 2483
+    },
+    {
+      "epoch": 225.82,
+      "learning_rate": 5.772248803827751e-05,
+      "loss": 0.0342,
+      "step": 2484
+    },
+    {
+      "epoch": 225.91,
+      "learning_rate": 5.770334928229665e-05,
+      "loss": 0.0376,
+      "step": 2485
+    },
+    {
+      "epoch": 226.0,
+      "learning_rate": 5.7684210526315794e-05,
+      "loss": 0.0484,
+      "step": 2486
+    },
+    {
+      "epoch": 226.09,
+      "learning_rate": 5.766507177033493e-05,
+      "loss": 0.0373,
+      "step": 2487
+    },
+    {
+      "epoch": 226.18,
+      "learning_rate": 5.764593301435407e-05,
+      "loss": 0.0369,
+      "step": 2488
+    },
+    {
+      "epoch": 226.27,
+      "learning_rate": 5.762679425837321e-05,
+      "loss": 0.0406,
+      "step": 2489
+    },
+    {
+      "epoch": 226.36,
+      "learning_rate": 5.760765550239234e-05,
+      "loss": 0.0364,
+      "step": 2490
+    },
+    {
+      "epoch": 226.45,
+      "learning_rate": 5.758851674641148e-05,
+      "loss": 0.0358,
+      "step": 2491
+    },
+    {
+      "epoch": 226.55,
+      "learning_rate": 5.7569377990430626e-05,
+      "loss": 0.0394,
+      "step": 2492
+    },
+    {
+      "epoch": 226.64,
+      "learning_rate": 5.7550239234449765e-05,
+      "loss": 0.0349,
+      "step": 2493
+    },
+    {
+      "epoch": 226.73,
+      "learning_rate": 5.7531100478468904e-05,
+      "loss": 0.0354,
+      "step": 2494
+    },
+    {
+      "epoch": 226.82,
+      "learning_rate": 5.7511961722488036e-05,
+      "loss": 0.0344,
+      "step": 2495
+    },
+    {
+      "epoch": 226.91,
+      "learning_rate": 5.7492822966507175e-05,
+      "loss": 0.0477,
+      "step": 2496
+    },
+    {
+      "epoch": 227.0,
+      "learning_rate": 5.747368421052633e-05,
+      "loss": 0.0382,
+      "step": 2497
+    },
+    {
+      "epoch": 227.09,
+      "learning_rate": 5.745454545454546e-05,
+      "loss": 0.0399,
+      "step": 2498
+    },
+    {
+      "epoch": 227.18,
+      "learning_rate": 5.74354066985646e-05,
+      "loss": 0.0344,
+      "step": 2499
+    },
+    {
+      "epoch": 227.27,
+      "learning_rate": 5.7416267942583736e-05,
+      "loss": 0.0368,
+      "step": 2500
+    },
+    {
+      "epoch": 227.27,
+      "eval_accuracy_NAT": 0.9931381689879399,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.950744935404953,
+      "eval_accuracy_concrete tiles": 0.9790252553048371,
+      "eval_accuracy_dark bricks": 0.8900350255956275,
+      "eval_accuracy_grass": 0.8858613783871633,
+      "eval_accuracy_red bricks": 0.9002130681818182,
+      "eval_accuracy_road": 0.9779803068246539,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6298624321120393,
+      "eval_accuracy_speedway bricks": 0.9880205448289762,
+      "eval_accuracy_steel": 0.9005919794208825,
+      "eval_accuracy_tiles": 0.0032187371592932476,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9798642598926022,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8876362425116707,
+      "eval_iou_concrete tiles": 0.9480384900074019,
+      "eval_iou_dark bricks": 0.8185486725663716,
+      "eval_iou_grass": 0.8167882106131551,
+      "eval_iou_red bricks": 0.829380009815148,
+      "eval_iou_road": 0.9732633770359629,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.4178310168251646,
+      "eval_iou_speedway bricks": 0.9764050498250116,
+      "eval_iou_steel": 0.8446566656356326,
+      "eval_iou_tiles": 0.0032187371592932476,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08745203167200089,
+      "eval_mean_accuracy": 0.6998993717083218,
+      "eval_mean_iou": 0.6535100562990319,
+      "eval_overall_accuracy": 0.9779639773898654,
+      "eval_runtime": 3.5217,
+      "eval_samples_per_second": 10.222,
+      "eval_steps_per_second": 0.568,
+      "step": 2500
+    },
+    {
+      "epoch": 227.36,
+      "learning_rate": 5.739712918660287e-05,
+      "loss": 0.0403,
+      "step": 2501
+    },
+    {
+      "epoch": 227.45,
+      "learning_rate": 5.737799043062201e-05,
+      "loss": 0.042,
+      "step": 2502
+    },
+    {
+      "epoch": 227.55,
+      "learning_rate": 5.735885167464116e-05,
+      "loss": 0.0354,
+      "step": 2503
+    },
+    {
+      "epoch": 227.64,
+      "learning_rate": 5.733971291866029e-05,
+      "loss": 0.0441,
+      "step": 2504
+    },
+    {
+      "epoch": 227.73,
+      "learning_rate": 5.732057416267943e-05,
+      "loss": 0.0348,
+      "step": 2505
+    },
+    {
+      "epoch": 227.82,
+      "learning_rate": 5.730143540669857e-05,
+      "loss": 0.0379,
+      "step": 2506
+    },
+    {
+      "epoch": 227.91,
+      "learning_rate": 5.72822966507177e-05,
+      "loss": 0.043,
+      "step": 2507
+    },
+    {
+      "epoch": 228.0,
+      "learning_rate": 5.726315789473684e-05,
+      "loss": 0.046,
+      "step": 2508
+    },
+    {
+      "epoch": 228.09,
+      "learning_rate": 5.7244019138755985e-05,
+      "loss": 0.0376,
+      "step": 2509
+    },
+    {
+      "epoch": 228.18,
+      "learning_rate": 5.7224880382775124e-05,
+      "loss": 0.0401,
+      "step": 2510
+    },
+    {
+      "epoch": 228.27,
+      "learning_rate": 5.720574162679426e-05,
+      "loss": 0.0382,
+      "step": 2511
+    },
+    {
+      "epoch": 228.36,
+      "learning_rate": 5.71866028708134e-05,
+      "loss": 0.051,
+      "step": 2512
+    },
+    {
+      "epoch": 228.45,
+      "learning_rate": 5.7167464114832534e-05,
+      "loss": 0.0355,
+      "step": 2513
+    },
+    {
+      "epoch": 228.55,
+      "learning_rate": 5.714832535885167e-05,
+      "loss": 0.0425,
+      "step": 2514
+    },
+    {
+      "epoch": 228.64,
+      "learning_rate": 5.712918660287082e-05,
+      "loss": 0.0441,
+      "step": 2515
+    },
+    {
+      "epoch": 228.73,
+      "learning_rate": 5.711004784688996e-05,
+      "loss": 0.0379,
+      "step": 2516
+    },
+    {
+      "epoch": 228.82,
+      "learning_rate": 5.7090909090909096e-05,
+      "loss": 0.0368,
+      "step": 2517
+    },
+    {
+      "epoch": 228.91,
+      "learning_rate": 5.7071770334928234e-05,
+      "loss": 0.0371,
+      "step": 2518
+    },
+    {
+      "epoch": 229.0,
+      "learning_rate": 5.7052631578947366e-05,
+      "loss": 0.0403,
+      "step": 2519
+    },
+    {
+      "epoch": 229.09,
+      "learning_rate": 5.7033492822966505e-05,
+      "loss": 0.0421,
+      "step": 2520
+    },
+    {
+      "epoch": 229.09,
+      "eval_accuracy_NAT": 0.9925622325442204,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9516046826044037,
+      "eval_accuracy_concrete tiles": 0.9846817097780224,
+      "eval_accuracy_dark bricks": 0.8881875216504369,
+      "eval_accuracy_grass": 0.89374626463999,
+      "eval_accuracy_red bricks": 0.9161931818181818,
+      "eval_accuracy_road": 0.9768110018658093,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6814159292035398,
+      "eval_accuracy_speedway bricks": 0.987893225781926,
+      "eval_accuracy_steel": 0.9120440604181782,
+      "eval_accuracy_tiles": 0.02020271195726613,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9800537456809018,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8887394056915519,
+      "eval_iou_concrete tiles": 0.9480423903444215,
+      "eval_iou_dark bricks": 0.8174861839308488,
+      "eval_iou_grass": 0.8293527755795701,
+      "eval_iou_red bricks": 0.8267905784329435,
+      "eval_iou_road": 0.9698938028659418,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.448486717956162,
+      "eval_iou_speedway bricks": 0.9762780632461742,
+      "eval_iou_steel": 0.8431530968459885,
+      "eval_iou_tiles": 0.02020271195726613,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08583278954029083,
+      "eval_mean_accuracy": 0.7081032709432288,
+      "eval_mean_iou": 0.6575753440409053,
+      "eval_overall_accuracy": 0.978195932176378,
+      "eval_runtime": 3.7313,
+      "eval_samples_per_second": 9.648,
+      "eval_steps_per_second": 0.536,
+      "step": 2520
+    },
+    {
+      "epoch": 229.18,
+      "learning_rate": 5.701435406698565e-05,
+      "loss": 0.0363,
+      "step": 2521
+    },
+    {
+      "epoch": 229.27,
+      "learning_rate": 5.699521531100479e-05,
+      "loss": 0.0334,
+      "step": 2522
+    },
+    {
+      "epoch": 229.36,
+      "learning_rate": 5.697607655502393e-05,
+      "loss": 0.038,
+      "step": 2523
+    },
+    {
+      "epoch": 229.45,
+      "learning_rate": 5.695693779904307e-05,
+      "loss": 0.0381,
+      "step": 2524
+    },
+    {
+      "epoch": 229.55,
+      "learning_rate": 5.69377990430622e-05,
+      "loss": 0.0404,
+      "step": 2525
+    },
+    {
+      "epoch": 229.64,
+      "learning_rate": 5.691866028708134e-05,
+      "loss": 0.041,
+      "step": 2526
+    },
+    {
+      "epoch": 229.73,
+      "learning_rate": 5.6899521531100483e-05,
+      "loss": 0.0395,
+      "step": 2527
+    },
+    {
+      "epoch": 229.82,
+      "learning_rate": 5.688038277511962e-05,
+      "loss": 0.036,
+      "step": 2528
+    },
+    {
+      "epoch": 229.91,
+      "learning_rate": 5.686124401913876e-05,
+      "loss": 0.0395,
+      "step": 2529
+    },
+    {
+      "epoch": 230.0,
+      "learning_rate": 5.68421052631579e-05,
+      "loss": 0.0435,
+      "step": 2530
+    },
+    {
+      "epoch": 230.09,
+      "learning_rate": 5.682296650717703e-05,
+      "loss": 0.0343,
+      "step": 2531
+    },
+    {
+      "epoch": 230.18,
+      "learning_rate": 5.680382775119617e-05,
+      "loss": 0.0386,
+      "step": 2532
+    },
+    {
+      "epoch": 230.27,
+      "learning_rate": 5.6784688995215316e-05,
+      "loss": 0.0349,
+      "step": 2533
+    },
+    {
+      "epoch": 230.36,
+      "learning_rate": 5.6765550239234455e-05,
+      "loss": 0.0361,
+      "step": 2534
+    },
+    {
+      "epoch": 230.45,
+      "learning_rate": 5.6746411483253594e-05,
+      "loss": 0.0359,
+      "step": 2535
+    },
+    {
+      "epoch": 230.55,
+      "learning_rate": 5.6727272727272726e-05,
+      "loss": 0.0432,
+      "step": 2536
+    },
+    {
+      "epoch": 230.64,
+      "learning_rate": 5.6708133971291864e-05,
+      "loss": 0.038,
+      "step": 2537
+    },
+    {
+      "epoch": 230.73,
+      "learning_rate": 5.6688995215311e-05,
+      "loss": 0.0427,
+      "step": 2538
+    },
+    {
+      "epoch": 230.82,
+      "learning_rate": 5.666985645933015e-05,
+      "loss": 0.0364,
+      "step": 2539
+    },
+    {
+      "epoch": 230.91,
+      "learning_rate": 5.665071770334929e-05,
+      "loss": 0.0375,
+      "step": 2540
+    },
+    {
+      "epoch": 230.91,
+      "eval_accuracy_NAT": 0.9922843220359918,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9548317524357268,
+      "eval_accuracy_concrete tiles": 0.9865773864122791,
+      "eval_accuracy_dark bricks": 0.8768715599861437,
+      "eval_accuracy_grass": 0.8876613413393821,
+      "eval_accuracy_red bricks": 0.9129971590909091,
+      "eval_accuracy_road": 0.9832500389969855,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.7583050754004356,
+      "eval_accuracy_speedway bricks": 0.9856841498044355,
+      "eval_accuracy_steel": 0.9147236330057384,
+      "eval_accuracy_tiles": 0.009998630324613067,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9802486579804061,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8928639066464488,
+      "eval_iou_concrete tiles": 0.9508191890617633,
+      "eval_iou_dark bricks": 0.8176434698345476,
+      "eval_iou_grass": 0.8263837596353142,
+      "eval_iou_red bricks": 0.8356899073622623,
+      "eval_iou_road": 0.9735368840977063,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.47721988965612966,
+      "eval_iou_speedway bricks": 0.9761853255088504,
+      "eval_iou_steel": 0.8426117203876416,
+      "eval_iou_tiles": 0.009998630324613067,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0855443924665451,
+      "eval_mean_accuracy": 0.7125526960640491,
+      "eval_mean_iou": 0.6602462569612064,
+      "eval_overall_accuracy": 0.9785635206434462,
+      "eval_runtime": 3.3735,
+      "eval_samples_per_second": 10.671,
+      "eval_steps_per_second": 0.593,
+      "step": 2540
+    },
+    {
+      "epoch": 231.0,
+      "learning_rate": 5.6631578947368426e-05,
+      "loss": 0.0389,
+      "step": 2541
+    },
+    {
+      "epoch": 231.09,
+      "learning_rate": 5.661244019138756e-05,
+      "loss": 0.0384,
+      "step": 2542
+    },
+    {
+      "epoch": 231.18,
+      "learning_rate": 5.65933014354067e-05,
+      "loss": 0.0354,
+      "step": 2543
+    },
+    {
+      "epoch": 231.27,
+      "learning_rate": 5.6574162679425836e-05,
+      "loss": 0.0418,
+      "step": 2544
+    },
+    {
+      "epoch": 231.36,
+      "learning_rate": 5.655502392344498e-05,
+      "loss": 0.0619,
+      "step": 2545
+    },
+    {
+      "epoch": 231.45,
+      "learning_rate": 5.653588516746412e-05,
+      "loss": 0.0351,
+      "step": 2546
+    },
+    {
+      "epoch": 231.55,
+      "learning_rate": 5.651674641148326e-05,
+      "loss": 0.0386,
+      "step": 2547
+    },
+    {
+      "epoch": 231.64,
+      "learning_rate": 5.649760765550239e-05,
+      "loss": 0.0337,
+      "step": 2548
+    },
+    {
+      "epoch": 231.73,
+      "learning_rate": 5.647846889952153e-05,
+      "loss": 0.0433,
+      "step": 2549
+    },
+    {
+      "epoch": 231.82,
+      "learning_rate": 5.645933014354067e-05,
+      "loss": 0.039,
+      "step": 2550
+    },
+    {
+      "epoch": 231.91,
+      "learning_rate": 5.6440191387559814e-05,
+      "loss": 0.0361,
+      "step": 2551
+    },
+    {
+      "epoch": 232.0,
+      "learning_rate": 5.642105263157895e-05,
+      "loss": 0.0406,
+      "step": 2552
+    },
+    {
+      "epoch": 232.09,
+      "learning_rate": 5.640191387559809e-05,
+      "loss": 0.0351,
+      "step": 2553
+    },
+    {
+      "epoch": 232.18,
+      "learning_rate": 5.6382775119617224e-05,
+      "loss": 0.0375,
+      "step": 2554
+    },
+    {
+      "epoch": 232.27,
+      "learning_rate": 5.636363636363636e-05,
+      "loss": 0.0412,
+      "step": 2555
+    },
+    {
+      "epoch": 232.36,
+      "learning_rate": 5.63444976076555e-05,
+      "loss": 0.0345,
+      "step": 2556
+    },
+    {
+      "epoch": 232.45,
+      "learning_rate": 5.632535885167465e-05,
+      "loss": 0.046,
+      "step": 2557
+    },
+    {
+      "epoch": 232.55,
+      "learning_rate": 5.6306220095693786e-05,
+      "loss": 0.051,
+      "step": 2558
+    },
+    {
+      "epoch": 232.64,
+      "learning_rate": 5.6287081339712924e-05,
+      "loss": 0.0374,
+      "step": 2559
+    },
+    {
+      "epoch": 232.73,
+      "learning_rate": 5.6267942583732056e-05,
+      "loss": 0.0375,
+      "step": 2560
+    },
+    {
+      "epoch": 232.73,
+      "eval_accuracy_NAT": 0.9924867114664288,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.947487016610155,
+      "eval_accuracy_concrete tiles": 0.9861187549685073,
+      "eval_accuracy_dark bricks": 0.8918055502097687,
+      "eval_accuracy_grass": 0.9059126161675118,
+      "eval_accuracy_red bricks": 0.9279119318181818,
+      "eval_accuracy_road": 0.9896068499262988,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.7319218151241971,
+      "eval_accuracy_speedway bricks": 0.9873694678063833,
+      "eval_accuracy_steel": 0.9052668029813337,
+      "eval_accuracy_tiles": 0.038145459526092315,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9805761027033906,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8994312473978328,
+      "eval_iou_concrete tiles": 0.940154496429092,
+      "eval_iou_dark bricks": 0.8167077899189285,
+      "eval_iou_grass": 0.8330665689601975,
+      "eval_iou_red bricks": 0.847963654064579,
+      "eval_iou_road": 0.9779750317570143,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.475984724886602,
+      "eval_iou_speedway bricks": 0.9764941568086621,
+      "eval_iou_steel": 0.8454259160872545,
+      "eval_iou_tiles": 0.038145459526092315,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08097036182880402,
+      "eval_mean_accuracy": 0.7156948443542197,
+      "eval_mean_iou": 0.6639942421953572,
+      "eval_overall_accuracy": 0.9793801837497287,
+      "eval_runtime": 3.2107,
+      "eval_samples_per_second": 11.213,
+      "eval_steps_per_second": 0.623,
+      "step": 2560
+    }
+  ],
+  "max_steps": 5500,
+  "num_train_epochs": 500,
+  "total_flos": 1.4330123656699576e+18,
+  "trial_name": null,
+  "trial_params": null
+}