diff --git "a/checkpoint-5200/trainer_state.json" "b/checkpoint-5200/trainer_state.json"
deleted file mode 100644--- "a/checkpoint-5200/trainer_state.json"
+++ /dev/null
@@ -1,41356 +0,0 @@
-{
-  "best_metric": 0.08471038937568665,
-  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0/checkpoint-2560",
-  "epoch": 472.72727272727275,
-  "global_step": 5200,
-  "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.6936,
-      "step": 1
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 7.272727272727273e-07,
-      "loss": 2.6918,
-      "step": 2
-    },
-    {
-      "epoch": 0.27,
-      "learning_rate": 1.090909090909091e-06,
-      "loss": 2.6894,
-      "step": 3
-    },
-    {
-      "epoch": 0.36,
-      "learning_rate": 1.4545454545454546e-06,
-      "loss": 2.6788,
-      "step": 4
-    },
-    {
-      "epoch": 0.45,
-      "learning_rate": 1.818181818181818e-06,
-      "loss": 2.6793,
-      "step": 5
-    },
-    {
-      "epoch": 0.55,
-      "learning_rate": 2.181818181818182e-06,
-      "loss": 2.6842,
-      "step": 6
-    },
-    {
-      "epoch": 0.64,
-      "learning_rate": 2.5454545454545456e-06,
-      "loss": 2.6901,
-      "step": 7
-    },
-    {
-      "epoch": 0.73,
-      "learning_rate": 2.9090909090909093e-06,
-      "loss": 2.6793,
-      "step": 8
-    },
-    {
-      "epoch": 0.82,
-      "learning_rate": 3.2727272727272733e-06,
-      "loss": 2.6703,
-      "step": 9
-    },
-    {
-      "epoch": 0.91,
-      "learning_rate": 3.636363636363636e-06,
-      "loss": 2.6668,
-      "step": 10
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 4.000000000000001e-06,
-      "loss": 2.6638,
-      "step": 11
-    },
-    {
-      "epoch": 1.09,
-      "learning_rate": 4.363636363636364e-06,
-      "loss": 2.6632,
-      "step": 12
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 4.727272727272727e-06,
-      "loss": 2.6546,
-      "step": 13
-    },
-    {
-      "epoch": 1.27,
-      "learning_rate": 5.090909090909091e-06,
-      "loss": 2.6434,
-      "step": 14
-    },
-    {
-      "epoch": 1.36,
-      "learning_rate": 5.4545454545454545e-06,
-      "loss": 2.6392,
-      "step": 15
-    },
-    {
-      "epoch": 1.45,
-      "learning_rate": 5.8181818181818185e-06,
-      "loss": 2.6424,
-      "step": 16
-    },
-    {
-      "epoch": 1.55,
-      "learning_rate": 6.181818181818183e-06,
-      "loss": 2.6164,
-      "step": 17
-    },
-    {
-      "epoch": 1.64,
-      "learning_rate": 6.545454545454547e-06,
-      "loss": 2.6205,
-      "step": 18
-    },
-    {
-      "epoch": 1.73,
-      "learning_rate": 6.909090909090909e-06,
-      "loss": 2.6095,
-      "step": 19
-    },
-    {
-      "epoch": 1.82,
-      "learning_rate": 7.272727272727272e-06,
-      "loss": 2.6108,
-      "step": 20
-    },
-    {
-      "epoch": 1.82,
-      "eval_accuracy_NAT": 0.09511473910290841,
-      "eval_accuracy_REST": 0.25,
-      "eval_accuracy_concrete": 0.01301558006781407,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.6411223586467033,
-      "eval_accuracy_grass": 0.16874215792839992,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.005832014288011783,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.1063325338405977,
-      "eval_accuracy_speedway bricks": 0.06347065688180568,
-      "eval_accuracy_steel": 0.013958511971505837,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.09360158321654327,
-      "eval_iou_REST": 7.769355796094811e-07,
-      "eval_iou_concrete": 0.011760286008604496,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.005873411187090554,
-      "eval_iou_grass": 0.020184891600275597,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.005779806508903394,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.017025014456033796,
-      "eval_iou_speedway bricks": 0.06224845291701409,
-      "eval_iou_steel": 0.006497743252786392,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": 0.0,
-      "eval_loss": 2.605102300643921,
-      "eval_mean_accuracy": 0.10442988867136511,
-      "eval_mean_iou": 0.015926569005916515,
-      "eval_overall_accuracy": 0.07727368672688802,
-      "eval_runtime": 3.9824,
-      "eval_samples_per_second": 9.04,
-      "eval_steps_per_second": 0.502,
-      "step": 20
-    },
-    {
-      "epoch": 1.91,
-      "learning_rate": 7.636363636363638e-06,
-      "loss": 2.5966,
-      "step": 21
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 8.000000000000001e-06,
-      "loss": 2.5833,
-      "step": 22
-    },
-    {
-      "epoch": 2.09,
-      "learning_rate": 8.363636363636365e-06,
-      "loss": 2.5726,
-      "step": 23
-    },
-    {
-      "epoch": 2.18,
-      "learning_rate": 8.727272727272728e-06,
-      "loss": 2.564,
-      "step": 24
-    },
-    {
-      "epoch": 2.27,
-      "learning_rate": 9.090909090909091e-06,
-      "loss": 2.5487,
-      "step": 25
-    },
-    {
-      "epoch": 2.36,
-      "learning_rate": 9.454545454545454e-06,
-      "loss": 2.5331,
-      "step": 26
-    },
-    {
-      "epoch": 2.45,
-      "learning_rate": 9.818181818181818e-06,
-      "loss": 2.5263,
-      "step": 27
-    },
-    {
-      "epoch": 2.55,
-      "learning_rate": 1.0181818181818182e-05,
-      "loss": 2.5167,
-      "step": 28
-    },
-    {
-      "epoch": 2.64,
-      "learning_rate": 1.0545454545454546e-05,
-      "loss": 2.4959,
-      "step": 29
-    },
-    {
-      "epoch": 2.73,
-      "learning_rate": 1.0909090909090909e-05,
-      "loss": 2.4863,
-      "step": 30
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 1.1272727272727274e-05,
-      "loss": 2.4883,
-      "step": 31
-    },
-    {
-      "epoch": 2.91,
-      "learning_rate": 1.1636363636363637e-05,
-      "loss": 2.4681,
-      "step": 32
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 1.2e-05,
-      "loss": 2.4457,
-      "step": 33
-    },
-    {
-      "epoch": 3.09,
-      "learning_rate": 1.2363636363636365e-05,
-      "loss": 2.4198,
-      "step": 34
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 1.2727272727272727e-05,
-      "loss": 2.4107,
-      "step": 35
-    },
-    {
-      "epoch": 3.27,
-      "learning_rate": 1.3090909090909093e-05,
-      "loss": 2.3918,
-      "step": 36
-    },
-    {
-      "epoch": 3.36,
-      "learning_rate": 1.3454545454545457e-05,
-      "loss": 2.3891,
-      "step": 37
-    },
-    {
-      "epoch": 3.45,
-      "learning_rate": 1.3818181818181818e-05,
-      "loss": 2.3568,
-      "step": 38
-    },
-    {
-      "epoch": 3.55,
-      "learning_rate": 1.4181818181818181e-05,
-      "loss": 2.3432,
-      "step": 39
-    },
-    {
-      "epoch": 3.64,
-      "learning_rate": 1.4545454545454545e-05,
-      "loss": 2.3421,
-      "step": 40
-    },
-    {
-      "epoch": 3.64,
-      "eval_accuracy_NAT": 0.6849137118750895,
-      "eval_accuracy_REST": 0.125,
-      "eval_accuracy_concrete": 0.05468314090733508,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.5916631384473269,
-      "eval_accuracy_grass": 0.18795108294858398,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.1476613296218683,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.02789953960245913,
-      "eval_accuracy_speedway bricks": 0.7465795828521156,
-      "eval_accuracy_steel": 0.02932689136600488,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.6762243132368754,
-      "eval_iou_REST": 8.163178676471188e-07,
-      "eval_iou_concrete": 0.04730494108637764,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.009574402487401365,
-      "eval_iou_grass": 0.0756298502760737,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.14108906887268777,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.005906833131964022,
-      "eval_iou_speedway bricks": 0.6970837094053498,
-      "eval_iou_steel": 0.012883666683569613,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": 0.0,
-      "eval_loss": 2.2775866985321045,
-      "eval_mean_accuracy": 0.1996675705862141,
-      "eval_mean_iou": 0.11897840010701195,
-      "eval_overall_accuracy": 0.5674870808919271,
-      "eval_runtime": 3.5096,
-      "eval_samples_per_second": 10.258,
-      "eval_steps_per_second": 0.57,
-      "step": 40
-    },
-    {
-      "epoch": 3.73,
-      "learning_rate": 1.4909090909090908e-05,
-      "loss": 2.3287,
-      "step": 41
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 1.5272727272727276e-05,
-      "loss": 2.308,
-      "step": 42
-    },
-    {
-      "epoch": 3.91,
-      "learning_rate": 1.563636363636364e-05,
-      "loss": 2.2771,
-      "step": 43
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 1.6000000000000003e-05,
-      "loss": 2.2752,
-      "step": 44
-    },
-    {
-      "epoch": 4.09,
-      "learning_rate": 1.6363636363636366e-05,
-      "loss": 2.2581,
-      "step": 45
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 1.672727272727273e-05,
-      "loss": 2.2201,
-      "step": 46
-    },
-    {
-      "epoch": 4.27,
-      "learning_rate": 1.7090909090909092e-05,
-      "loss": 2.1681,
-      "step": 47
-    },
-    {
-      "epoch": 4.36,
-      "learning_rate": 1.7454545454545456e-05,
-      "loss": 2.1538,
-      "step": 48
-    },
-    {
-      "epoch": 4.45,
-      "learning_rate": 1.781818181818182e-05,
-      "loss": 2.1502,
-      "step": 49
-    },
-    {
-      "epoch": 4.55,
-      "learning_rate": 1.8181818181818182e-05,
-      "loss": 2.1401,
-      "step": 50
-    },
-    {
-      "epoch": 4.64,
-      "learning_rate": 1.8545454545454545e-05,
-      "loss": 2.0827,
-      "step": 51
-    },
-    {
-      "epoch": 4.73,
-      "learning_rate": 1.890909090909091e-05,
-      "loss": 2.0856,
-      "step": 52
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 1.9272727272727272e-05,
-      "loss": 2.0984,
-      "step": 53
-    },
-    {
-      "epoch": 4.91,
-      "learning_rate": 1.9636363636363635e-05,
-      "loss": 2.049,
-      "step": 54
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 2e-05,
-      "loss": 2.0259,
-      "step": 55
-    },
-    {
-      "epoch": 5.09,
-      "learning_rate": 2.0363636363636365e-05,
-      "loss": 1.961,
-      "step": 56
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 2.0727272727272728e-05,
-      "loss": 1.9549,
-      "step": 57
-    },
-    {
-      "epoch": 5.27,
-      "learning_rate": 2.109090909090909e-05,
-      "loss": 1.9661,
-      "step": 58
-    },
-    {
-      "epoch": 5.36,
-      "learning_rate": 2.1454545454545455e-05,
-      "loss": 1.9233,
-      "step": 59
-    },
-    {
-      "epoch": 5.45,
-      "learning_rate": 2.1818181818181818e-05,
-      "loss": 1.9173,
-      "step": 60
-    },
-    {
-      "epoch": 5.45,
-      "eval_accuracy_NAT": 0.9258208329480817,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.5797794433237478,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.18390362187752587,
-      "eval_accuracy_grass": 0.419499715151494,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.6828861880537421,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0004686681553772779,
-      "eval_accuracy_speedway bricks": 0.9326892558412893,
-      "eval_accuracy_steel": 0.00015665193588813403,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.915879998687335,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.3659033117650445,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.02468574498974441,
-      "eval_iou_grass": 0.19461708465144334,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.5963247243226463,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.00032108185698636346,
-      "eval_iou_speedway bricks": 0.7993201887295917,
-      "eval_iou_steel": 0.00015567773071030014,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": 0.0,
-      "eval_loss": 1.6415036916732788,
-      "eval_mean_accuracy": 0.286554182868242,
-      "eval_mean_iou": 0.20694341519525014,
-      "eval_overall_accuracy": 0.8386445575290256,
-      "eval_runtime": 3.435,
-      "eval_samples_per_second": 10.48,
-      "eval_steps_per_second": 0.582,
-      "step": 60
-    },
-    {
-      "epoch": 5.55,
-      "learning_rate": 2.2181818181818184e-05,
-      "loss": 1.8719,
-      "step": 61
-    },
-    {
-      "epoch": 5.64,
-      "learning_rate": 2.2545454545454548e-05,
-      "loss": 1.851,
-      "step": 62
-    },
-    {
-      "epoch": 5.73,
-      "learning_rate": 2.290909090909091e-05,
-      "loss": 1.8528,
-      "step": 63
-    },
-    {
-      "epoch": 5.82,
-      "learning_rate": 2.3272727272727274e-05,
-      "loss": 1.8134,
-      "step": 64
-    },
-    {
-      "epoch": 5.91,
-      "learning_rate": 2.3636363636363637e-05,
-      "loss": 1.7904,
-      "step": 65
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 2.4e-05,
-      "loss": 1.7922,
-      "step": 66
-    },
-    {
-      "epoch": 6.09,
-      "learning_rate": 2.4363636363636364e-05,
-      "loss": 1.7283,
-      "step": 67
-    },
-    {
-      "epoch": 6.18,
-      "learning_rate": 2.472727272727273e-05,
-      "loss": 1.7191,
-      "step": 68
-    },
-    {
-      "epoch": 6.27,
-      "learning_rate": 2.5090909090909094e-05,
-      "loss": 1.7944,
-      "step": 69
-    },
-    {
-      "epoch": 6.36,
-      "learning_rate": 2.5454545454545454e-05,
-      "loss": 1.678,
-      "step": 70
-    },
-    {
-      "epoch": 6.45,
-      "learning_rate": 2.581818181818182e-05,
-      "loss": 1.7206,
-      "step": 71
-    },
-    {
-      "epoch": 6.55,
-      "learning_rate": 2.6181818181818187e-05,
-      "loss": 1.6827,
-      "step": 72
-    },
-    {
-      "epoch": 6.64,
-      "learning_rate": 2.6545454545454547e-05,
-      "loss": 1.6447,
-      "step": 73
-    },
-    {
-      "epoch": 6.73,
-      "learning_rate": 2.6909090909090913e-05,
-      "loss": 1.6101,
-      "step": 74
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 2.7272727272727273e-05,
-      "loss": 1.599,
-      "step": 75
-    },
-    {
-      "epoch": 6.91,
-      "learning_rate": 2.7636363636363636e-05,
-      "loss": 1.6276,
-      "step": 76
-    },
-    {
-      "epoch": 7.0,
-      "learning_rate": 2.8000000000000003e-05,
-      "loss": 1.6611,
-      "step": 77
-    },
-    {
-      "epoch": 7.09,
-      "learning_rate": 2.8363636363636363e-05,
-      "loss": 1.5806,
-      "step": 78
-    },
-    {
-      "epoch": 7.18,
-      "learning_rate": 2.872727272727273e-05,
-      "loss": 1.5683,
-      "step": 79
-    },
-    {
-      "epoch": 7.27,
-      "learning_rate": 2.909090909090909e-05,
-      "loss": 1.5459,
-      "step": 80
-    },
-    {
-      "epoch": 7.27,
-      "eval_accuracy_NAT": 0.9541145930013841,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.7506236855659041,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0003079173241984527,
-      "eval_accuracy_grass": 0.33006776947891947,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.8611586397367794,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9632361726084233,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9413840553822712,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.4971771539137395,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0002478775484910454,
-      "eval_iou_grass": 0.2048493314707643,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.715494766157398,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8327570610740783,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": 0.0,
-      "eval_loss": 1.2371957302093506,
-      "eval_mean_accuracy": 0.2968852905935084,
-      "eval_mean_iou": 0.22799358896762448,
-      "eval_overall_accuracy": 0.8868987825181749,
-      "eval_runtime": 3.6626,
-      "eval_samples_per_second": 9.829,
-      "eval_steps_per_second": 0.546,
-      "step": 80
-    },
-    {
-      "epoch": 7.36,
-      "learning_rate": 2.9454545454545456e-05,
-      "loss": 1.5754,
-      "step": 81
-    },
-    {
-      "epoch": 7.45,
-      "learning_rate": 2.9818181818181816e-05,
-      "loss": 1.5397,
-      "step": 82
-    },
-    {
-      "epoch": 7.55,
-      "learning_rate": 3.0181818181818182e-05,
-      "loss": 1.5087,
-      "step": 83
-    },
-    {
-      "epoch": 7.64,
-      "learning_rate": 3.054545454545455e-05,
-      "loss": 1.5129,
-      "step": 84
-    },
-    {
-      "epoch": 7.73,
-      "learning_rate": 3.090909090909091e-05,
-      "loss": 1.4476,
-      "step": 85
-    },
-    {
-      "epoch": 7.82,
-      "learning_rate": 3.127272727272728e-05,
-      "loss": 1.4488,
-      "step": 86
-    },
-    {
-      "epoch": 7.91,
-      "learning_rate": 3.1636363636363635e-05,
-      "loss": 1.4656,
-      "step": 87
-    },
-    {
-      "epoch": 8.0,
-      "learning_rate": 3.2000000000000005e-05,
-      "loss": 1.5033,
-      "step": 88
-    },
-    {
-      "epoch": 8.09,
-      "learning_rate": 3.236363636363636e-05,
-      "loss": 1.4288,
-      "step": 89
-    },
-    {
-      "epoch": 8.18,
-      "learning_rate": 3.272727272727273e-05,
-      "loss": 1.4925,
-      "step": 90
-    },
-    {
-      "epoch": 8.27,
-      "learning_rate": 3.3090909090909095e-05,
-      "loss": 1.4681,
-      "step": 91
-    },
-    {
-      "epoch": 8.36,
-      "learning_rate": 3.345454545454546e-05,
-      "loss": 1.3993,
-      "step": 92
-    },
-    {
-      "epoch": 8.45,
-      "learning_rate": 3.381818181818182e-05,
-      "loss": 1.3844,
-      "step": 93
-    },
-    {
-      "epoch": 8.55,
-      "learning_rate": 3.4181818181818185e-05,
-      "loss": 1.4013,
-      "step": 94
-    },
-    {
-      "epoch": 8.64,
-      "learning_rate": 3.454545454545455e-05,
-      "loss": 1.3718,
-      "step": 95
-    },
-    {
-      "epoch": 8.73,
-      "learning_rate": 3.490909090909091e-05,
-      "loss": 1.3602,
-      "step": 96
-    },
-    {
-      "epoch": 8.82,
-      "learning_rate": 3.5272727272727274e-05,
-      "loss": 1.381,
-      "step": 97
-    },
-    {
-      "epoch": 8.91,
-      "learning_rate": 3.563636363636364e-05,
-      "loss": 1.3243,
-      "step": 98
-    },
-    {
-      "epoch": 9.0,
-      "learning_rate": 3.6e-05,
-      "loss": 1.3197,
-      "step": 99
-    },
-    {
-      "epoch": 9.09,
-      "learning_rate": 3.6363636363636364e-05,
-      "loss": 1.3972,
-      "step": 100
-    },
-    {
-      "epoch": 9.09,
-      "eval_accuracy_NAT": 0.9441565742509826,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.7732024443109147,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.3750982989476333,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9302879005353167,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9784070516649832,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9364044962303619,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.5272474841475918,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.2266649488999605,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.7831659271067786,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8404302629096495,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 1.1419544219970703,
-      "eval_mean_accuracy": 0.30778094382383314,
-      "eval_mean_iou": 0.25491639379187253,
-      "eval_overall_accuracy": 0.8927888870239258,
-      "eval_runtime": 3.338,
-      "eval_samples_per_second": 10.785,
-      "eval_steps_per_second": 0.599,
-      "step": 100
-    },
-    {
-      "epoch": 9.18,
-      "learning_rate": 3.672727272727273e-05,
-      "loss": 1.3272,
-      "step": 101
-    },
-    {
-      "epoch": 9.27,
-      "learning_rate": 3.709090909090909e-05,
-      "loss": 1.3119,
-      "step": 102
-    },
-    {
-      "epoch": 9.36,
-      "learning_rate": 3.745454545454546e-05,
-      "loss": 1.3227,
-      "step": 103
-    },
-    {
-      "epoch": 9.45,
-      "learning_rate": 3.781818181818182e-05,
-      "loss": 1.2673,
-      "step": 104
-    },
-    {
-      "epoch": 9.55,
-      "learning_rate": 3.818181818181819e-05,
-      "loss": 1.2853,
-      "step": 105
-    },
-    {
-      "epoch": 9.64,
-      "learning_rate": 3.8545454545454544e-05,
-      "loss": 1.2571,
-      "step": 106
-    },
-    {
-      "epoch": 9.73,
-      "learning_rate": 3.8909090909090914e-05,
-      "loss": 1.2697,
-      "step": 107
-    },
-    {
-      "epoch": 9.82,
-      "learning_rate": 3.927272727272727e-05,
-      "loss": 1.2571,
-      "step": 108
-    },
-    {
-      "epoch": 9.91,
-      "learning_rate": 3.963636363636364e-05,
-      "loss": 1.2733,
-      "step": 109
-    },
-    {
-      "epoch": 10.0,
-      "learning_rate": 4e-05,
-      "loss": 1.2868,
-      "step": 110
-    },
-    {
-      "epoch": 10.09,
-      "learning_rate": 4.0363636363636367e-05,
-      "loss": 1.2433,
-      "step": 111
-    },
-    {
-      "epoch": 10.18,
-      "learning_rate": 4.072727272727273e-05,
-      "loss": 1.2005,
-      "step": 112
-    },
-    {
-      "epoch": 10.27,
-      "learning_rate": 4.109090909090909e-05,
-      "loss": 1.2367,
-      "step": 113
-    },
-    {
-      "epoch": 10.36,
-      "learning_rate": 4.1454545454545456e-05,
-      "loss": 1.2342,
-      "step": 114
-    },
-    {
-      "epoch": 10.45,
-      "learning_rate": 4.181818181818182e-05,
-      "loss": 1.2305,
-      "step": 115
-    },
-    {
-      "epoch": 10.55,
-      "learning_rate": 4.218181818181818e-05,
-      "loss": 1.2059,
-      "step": 116
-    },
-    {
-      "epoch": 10.64,
-      "learning_rate": 4.254545454545455e-05,
-      "loss": 1.1661,
-      "step": 117
-    },
-    {
-      "epoch": 10.73,
-      "learning_rate": 4.290909090909091e-05,
-      "loss": 1.2296,
-      "step": 118
-    },
-    {
-      "epoch": 10.82,
-      "learning_rate": 4.327272727272728e-05,
-      "loss": 1.204,
-      "step": 119
-    },
-    {
-      "epoch": 10.91,
-      "learning_rate": 4.3636363636363636e-05,
-      "loss": 1.1514,
-      "step": 120
-    },
-    {
-      "epoch": 10.91,
-      "eval_accuracy_NAT": 0.9494411087341185,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8586969934331946,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.534651908135483,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9496376605678203,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9744118885819555,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9412789614506244,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6159207732573185,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.32522535929926016,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8377841239429528,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8578146032380104,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.9918751120567322,
-      "eval_mean_accuracy": 0.3282184276501978,
-      "eval_mean_iou": 0.27523260162985896,
-      "eval_overall_accuracy": 0.9085481431749132,
-      "eval_runtime": 3.5459,
-      "eval_samples_per_second": 10.153,
-      "eval_steps_per_second": 0.564,
-      "step": 120
-    },
-    {
-      "epoch": 11.0,
-      "learning_rate": 4.4000000000000006e-05,
-      "loss": 1.1483,
-      "step": 121
-    },
-    {
-      "epoch": 11.09,
-      "learning_rate": 4.436363636363637e-05,
-      "loss": 1.125,
-      "step": 122
-    },
-    {
-      "epoch": 11.18,
-      "learning_rate": 4.472727272727273e-05,
-      "loss": 1.0994,
-      "step": 123
-    },
-    {
-      "epoch": 11.27,
-      "learning_rate": 4.5090909090909095e-05,
-      "loss": 1.1775,
-      "step": 124
-    },
-    {
-      "epoch": 11.36,
-      "learning_rate": 4.545454545454546e-05,
-      "loss": 1.1118,
-      "step": 125
-    },
-    {
-      "epoch": 11.45,
-      "learning_rate": 4.581818181818182e-05,
-      "loss": 1.1555,
-      "step": 126
-    },
-    {
-      "epoch": 11.55,
-      "learning_rate": 4.618181818181818e-05,
-      "loss": 1.0831,
-      "step": 127
-    },
-    {
-      "epoch": 11.64,
-      "learning_rate": 4.654545454545455e-05,
-      "loss": 1.0728,
-      "step": 128
-    },
-    {
-      "epoch": 11.73,
-      "learning_rate": 4.690909090909091e-05,
-      "loss": 1.1166,
-      "step": 129
-    },
-    {
-      "epoch": 11.82,
-      "learning_rate": 4.7272727272727275e-05,
-      "loss": 1.1782,
-      "step": 130
-    },
-    {
-      "epoch": 11.91,
-      "learning_rate": 4.763636363636364e-05,
-      "loss": 1.0924,
-      "step": 131
-    },
-    {
-      "epoch": 12.0,
-      "learning_rate": 4.8e-05,
-      "loss": 1.0445,
-      "step": 132
-    },
-    {
-      "epoch": 12.09,
-      "learning_rate": 4.8363636363636364e-05,
-      "loss": 1.0956,
-      "step": 133
-    },
-    {
-      "epoch": 12.18,
-      "learning_rate": 4.872727272727273e-05,
-      "loss": 1.0544,
-      "step": 134
-    },
-    {
-      "epoch": 12.27,
-      "learning_rate": 4.909090909090909e-05,
-      "loss": 1.0478,
-      "step": 135
-    },
-    {
-      "epoch": 12.36,
-      "learning_rate": 4.945454545454546e-05,
-      "loss": 1.0406,
-      "step": 136
-    },
-    {
-      "epoch": 12.45,
-      "learning_rate": 4.981818181818182e-05,
-      "loss": 1.031,
-      "step": 137
-    },
-    {
-      "epoch": 12.55,
-      "learning_rate": 5.018181818181819e-05,
-      "loss": 1.0459,
-      "step": 138
-    },
-    {
-      "epoch": 12.64,
-      "learning_rate": 5.054545454545455e-05,
-      "loss": 1.084,
-      "step": 139
-    },
-    {
-      "epoch": 12.73,
-      "learning_rate": 5.090909090909091e-05,
-      "loss": 1.0387,
-      "step": 140
-    },
-    {
-      "epoch": 12.73,
-      "eval_accuracy_NAT": 0.9525459425771603,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8759268101635264,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.726360195304718,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9417076689234535,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9726644195759974,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9436753187840712,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6566357551437371,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.4501492382136467,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8345871752335147,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8664112146619203,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.8869848847389221,
-      "eval_mean_accuracy": 0.34378500281114277,
-      "eval_mean_iou": 0.28857374631053,
-      "eval_overall_accuracy": 0.916584332784017,
-      "eval_runtime": 3.5134,
-      "eval_samples_per_second": 10.246,
-      "eval_steps_per_second": 0.569,
-      "step": 140
-    },
-    {
-      "epoch": 12.82,
-      "learning_rate": 5.127272727272727e-05,
-      "loss": 0.9896,
-      "step": 141
-    },
-    {
-      "epoch": 12.91,
-      "learning_rate": 5.163636363636364e-05,
-      "loss": 1.0116,
-      "step": 142
-    },
-    {
-      "epoch": 13.0,
-      "learning_rate": 5.2000000000000004e-05,
-      "loss": 0.9974,
-      "step": 143
-    },
-    {
-      "epoch": 13.09,
-      "learning_rate": 5.2363636363636374e-05,
-      "loss": 0.9904,
-      "step": 144
-    },
-    {
-      "epoch": 13.18,
-      "learning_rate": 5.272727272727272e-05,
-      "loss": 1.0003,
-      "step": 145
-    },
-    {
-      "epoch": 13.27,
-      "learning_rate": 5.309090909090909e-05,
-      "loss": 1.0049,
-      "step": 146
-    },
-    {
-      "epoch": 13.36,
-      "learning_rate": 5.3454545454545457e-05,
-      "loss": 0.9292,
-      "step": 147
-    },
-    {
-      "epoch": 13.45,
-      "learning_rate": 5.3818181818181827e-05,
-      "loss": 0.9974,
-      "step": 148
-    },
-    {
-      "epoch": 13.55,
-      "learning_rate": 5.418181818181819e-05,
-      "loss": 0.9906,
-      "step": 149
-    },
-    {
-      "epoch": 13.64,
-      "learning_rate": 5.4545454545454546e-05,
-      "loss": 0.9568,
-      "step": 150
-    },
-    {
-      "epoch": 13.73,
-      "learning_rate": 5.490909090909091e-05,
-      "loss": 0.9848,
-      "step": 151
-    },
-    {
-      "epoch": 13.82,
-      "learning_rate": 5.527272727272727e-05,
-      "loss": 0.9129,
-      "step": 152
-    },
-    {
-      "epoch": 13.91,
-      "learning_rate": 5.563636363636364e-05,
-      "loss": 0.9442,
-      "step": 153
-    },
-    {
-      "epoch": 14.0,
-      "learning_rate": 5.6000000000000006e-05,
-      "loss": 0.9283,
-      "step": 154
-    },
-    {
-      "epoch": 14.09,
-      "learning_rate": 5.636363636363636e-05,
-      "loss": 0.8675,
-      "step": 155
-    },
-    {
-      "epoch": 14.18,
-      "learning_rate": 5.6727272727272726e-05,
-      "loss": 0.9106,
-      "step": 156
-    },
-    {
-      "epoch": 14.27,
-      "learning_rate": 5.7090909090909096e-05,
-      "loss": 0.8956,
-      "step": 157
-    },
-    {
-      "epoch": 14.36,
-      "learning_rate": 5.745454545454546e-05,
-      "loss": 0.9227,
-      "step": 158
-    },
-    {
-      "epoch": 14.45,
-      "learning_rate": 5.7818181818181815e-05,
-      "loss": 0.9498,
-      "step": 159
-    },
-    {
-      "epoch": 14.55,
-      "learning_rate": 5.818181818181818e-05,
-      "loss": 0.9516,
-      "step": 160
-    },
-    {
-      "epoch": 14.55,
-      "eval_accuracy_NAT": 0.9519405387970128,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8994203077385295,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7479562556576506,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.8920224864477918,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9753701001777639,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9435492601648952,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6126403745025453,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.49066488745818515,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8458624244224112,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8661472256739096,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.7788939476013184,
-      "eval_mean_accuracy": 0.3435930529860576,
-      "eval_mean_iou": 0.2891433978632266,
-      "eval_overall_accuracy": 0.9148527781168619,
-      "eval_runtime": 3.4609,
-      "eval_samples_per_second": 10.402,
-      "eval_steps_per_second": 0.578,
-      "step": 160
-    },
-    {
-      "epoch": 14.64,
-      "learning_rate": 5.854545454545455e-05,
-      "loss": 0.9021,
-      "step": 161
-    },
-    {
-      "epoch": 14.73,
-      "learning_rate": 5.890909090909091e-05,
-      "loss": 0.8649,
-      "step": 162
-    },
-    {
-      "epoch": 14.82,
-      "learning_rate": 5.927272727272728e-05,
-      "loss": 0.8786,
-      "step": 163
-    },
-    {
-      "epoch": 14.91,
-      "learning_rate": 5.963636363636363e-05,
-      "loss": 0.8389,
-      "step": 164
-    },
-    {
-      "epoch": 15.0,
-      "learning_rate": 6e-05,
-      "loss": 0.8666,
-      "step": 165
-    },
-    {
-      "epoch": 15.09,
-      "learning_rate": 6.0363636363636365e-05,
-      "loss": 0.8747,
-      "step": 166
-    },
-    {
-      "epoch": 15.18,
-      "learning_rate": 6.0727272727272735e-05,
-      "loss": 0.8764,
-      "step": 167
-    },
-    {
-      "epoch": 15.27,
-      "learning_rate": 6.10909090909091e-05,
-      "loss": 0.8476,
-      "step": 168
-    },
-    {
-      "epoch": 15.36,
-      "learning_rate": 6.145454545454545e-05,
-      "loss": 0.8698,
-      "step": 169
-    },
-    {
-      "epoch": 15.45,
-      "learning_rate": 6.181818181818182e-05,
-      "loss": 0.8277,
-      "step": 170
-    },
-    {
-      "epoch": 15.55,
-      "learning_rate": 6.218181818181819e-05,
-      "loss": 0.827,
-      "step": 171
-    },
-    {
-      "epoch": 15.64,
-      "learning_rate": 6.254545454545456e-05,
-      "loss": 0.7985,
-      "step": 172
-    },
-    {
-      "epoch": 15.73,
-      "learning_rate": 6.290909090909091e-05,
-      "loss": 0.8279,
-      "step": 173
-    },
-    {
-      "epoch": 15.82,
-      "learning_rate": 6.327272727272727e-05,
-      "loss": 0.7842,
-      "step": 174
-    },
-    {
-      "epoch": 15.91,
-      "learning_rate": 6.363636363636364e-05,
-      "loss": 0.8432,
-      "step": 175
-    },
-    {
-      "epoch": 16.0,
-      "learning_rate": 6.400000000000001e-05,
-      "loss": 0.7764,
-      "step": 176
-    },
-    {
-      "epoch": 16.09,
-      "learning_rate": 6.436363636363637e-05,
-      "loss": 0.8342,
-      "step": 177
-    },
-    {
-      "epoch": 16.18,
-      "learning_rate": 6.472727272727272e-05,
-      "loss": 0.7997,
-      "step": 178
-    },
-    {
-      "epoch": 16.27,
-      "learning_rate": 6.50909090909091e-05,
-      "loss": 0.7965,
-      "step": 179
-    },
-    {
-      "epoch": 16.36,
-      "learning_rate": 6.545454545454546e-05,
-      "loss": 0.7909,
-      "step": 180
-    },
-    {
-      "epoch": 16.36,
-      "eval_accuracy_NAT": 0.9619333184173323,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8907691853727627,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7823897216872818,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9575265391112556,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9698676243955362,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9486718471978234,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6816816200342622,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.5504504814643598,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8747294990969495,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8693016400978921,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.6643147468566895,
-      "eval_mean_accuracy": 0.3509604914603207,
-      "eval_mean_iou": 0.30191039137625286,
-      "eval_overall_accuracy": 0.9259880913628472,
-      "eval_runtime": 3.6989,
-      "eval_samples_per_second": 9.733,
-      "eval_steps_per_second": 0.541,
-      "step": 180
-    },
-    {
-      "epoch": 16.45,
-      "learning_rate": 6.581818181818182e-05,
-      "loss": 0.7994,
-      "step": 181
-    },
-    {
-      "epoch": 16.55,
-      "learning_rate": 6.618181818181819e-05,
-      "loss": 0.8043,
-      "step": 182
-    },
-    {
-      "epoch": 16.64,
-      "learning_rate": 6.654545454545455e-05,
-      "loss": 0.7247,
-      "step": 183
-    },
-    {
-      "epoch": 16.73,
-      "learning_rate": 6.690909090909092e-05,
-      "loss": 0.74,
-      "step": 184
-    },
-    {
-      "epoch": 16.82,
-      "learning_rate": 6.727272727272727e-05,
-      "loss": 0.7753,
-      "step": 185
-    },
-    {
-      "epoch": 16.91,
-      "learning_rate": 6.763636363636364e-05,
-      "loss": 0.7471,
-      "step": 186
-    },
-    {
-      "epoch": 17.0,
-      "learning_rate": 6.800000000000001e-05,
-      "loss": 0.7165,
-      "step": 187
-    },
-    {
-      "epoch": 17.09,
-      "learning_rate": 6.836363636363637e-05,
-      "loss": 0.7639,
-      "step": 188
-    },
-    {
-      "epoch": 17.18,
-      "learning_rate": 6.872727272727273e-05,
-      "loss": 0.7767,
-      "step": 189
-    },
-    {
-      "epoch": 17.27,
-      "learning_rate": 6.90909090909091e-05,
-      "loss": 0.769,
-      "step": 190
-    },
-    {
-      "epoch": 17.36,
-      "learning_rate": 6.945454545454547e-05,
-      "loss": 0.7041,
-      "step": 191
-    },
-    {
-      "epoch": 17.45,
-      "learning_rate": 6.981818181818182e-05,
-      "loss": 0.6912,
-      "step": 192
-    },
-    {
-      "epoch": 17.55,
-      "learning_rate": 7.018181818181818e-05,
-      "loss": 0.67,
-      "step": 193
-    },
-    {
-      "epoch": 17.64,
-      "learning_rate": 7.054545454545455e-05,
-      "loss": 0.6731,
-      "step": 194
-    },
-    {
-      "epoch": 17.73,
-      "learning_rate": 7.090909090909092e-05,
-      "loss": 0.6915,
-      "step": 195
-    },
-    {
-      "epoch": 17.82,
-      "learning_rate": 7.127272727272728e-05,
-      "loss": 0.7215,
-      "step": 196
-    },
-    {
-      "epoch": 17.91,
-      "learning_rate": 7.163636363636363e-05,
-      "loss": 0.6632,
-      "step": 197
-    },
-    {
-      "epoch": 18.0,
-      "learning_rate": 7.2e-05,
-      "loss": 0.6253,
-      "step": 198
-    },
-    {
-      "epoch": 18.09,
-      "learning_rate": 7.236363636363637e-05,
-      "loss": 0.6285,
-      "step": 199
-    },
-    {
-      "epoch": 18.18,
-      "learning_rate": 7.272727272727273e-05,
-      "loss": 0.7189,
-      "step": 200
-    },
-    {
-      "epoch": 18.18,
-      "eval_accuracy_NAT": 0.9610858237055723,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8973292845186489,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7523705337327038,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9528190890546042,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9795505094572106,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9488019245157138,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6700592619152306,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.5677699443217994,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8822359066226277,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8696800959165809,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.5797037482261658,
-      "eval_mean_accuracy": 0.3494734800360569,
-      "eval_mean_iou": 0.302965164099381,
-      "eval_overall_accuracy": 0.9263751771714952,
-      "eval_runtime": 3.3252,
-      "eval_samples_per_second": 10.826,
-      "eval_steps_per_second": 0.601,
-      "step": 200
-    },
-    {
-      "epoch": 18.27,
-      "learning_rate": 7.30909090909091e-05,
-      "loss": 0.6902,
-      "step": 201
-    },
-    {
-      "epoch": 18.36,
-      "learning_rate": 7.345454545454545e-05,
-      "loss": 0.6411,
-      "step": 202
-    },
-    {
-      "epoch": 18.45,
-      "learning_rate": 7.381818181818182e-05,
-      "loss": 0.7037,
-      "step": 203
-    },
-    {
-      "epoch": 18.55,
-      "learning_rate": 7.418181818181818e-05,
-      "loss": 0.6229,
-      "step": 204
-    },
-    {
-      "epoch": 18.64,
-      "learning_rate": 7.454545454545455e-05,
-      "loss": 0.6343,
-      "step": 205
-    },
-    {
-      "epoch": 18.73,
-      "learning_rate": 7.490909090909092e-05,
-      "loss": 0.6315,
-      "step": 206
-    },
-    {
-      "epoch": 18.82,
-      "learning_rate": 7.527272727272728e-05,
-      "loss": 0.6426,
-      "step": 207
-    },
-    {
-      "epoch": 18.91,
-      "learning_rate": 7.563636363636363e-05,
-      "loss": 0.6563,
-      "step": 208
-    },
-    {
-      "epoch": 19.0,
-      "learning_rate": 7.6e-05,
-      "loss": 0.5825,
-      "step": 209
-    },
-    {
-      "epoch": 19.09,
-      "learning_rate": 7.636363636363637e-05,
-      "loss": 0.613,
-      "step": 210
-    },
-    {
-      "epoch": 19.18,
-      "learning_rate": 7.672727272727273e-05,
-      "loss": 0.6389,
-      "step": 211
-    },
-    {
-      "epoch": 19.27,
-      "learning_rate": 7.709090909090909e-05,
-      "loss": 0.6512,
-      "step": 212
-    },
-    {
-      "epoch": 19.36,
-      "learning_rate": 7.745454545454546e-05,
-      "loss": 0.6031,
-      "step": 213
-    },
-    {
-      "epoch": 19.45,
-      "learning_rate": 7.781818181818183e-05,
-      "loss": 0.5767,
-      "step": 214
-    },
-    {
-      "epoch": 19.55,
-      "learning_rate": 7.818181818181818e-05,
-      "loss": 0.6033,
-      "step": 215
-    },
-    {
-      "epoch": 19.64,
-      "learning_rate": 7.854545454545454e-05,
-      "loss": 0.5905,
-      "step": 216
-    },
-    {
-      "epoch": 19.73,
-      "learning_rate": 7.890909090909091e-05,
-      "loss": 0.5525,
-      "step": 217
-    },
-    {
-      "epoch": 19.82,
-      "learning_rate": 7.927272727272728e-05,
-      "loss": 0.5875,
-      "step": 218
-    },
-    {
-      "epoch": 19.91,
-      "learning_rate": 7.963636363636364e-05,
-      "loss": 0.6272,
-      "step": 219
-    },
-    {
-      "epoch": 20.0,
-      "learning_rate": 8e-05,
-      "loss": 0.5829,
-      "step": 220
-    },
-    {
-      "epoch": 20.0,
-      "eval_accuracy_NAT": 0.9617046377705614,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9070292716425598,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7869612780785483,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9461418377797958,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9750901189558149,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9498686804080231,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.681961380679045,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.5801510392645403,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8684081139783777,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8727223630879569,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.4973033368587494,
-      "eval_mean_accuracy": 0.3520713187867139,
-      "eval_mean_iou": 0.30408550595522643,
-      "eval_overall_accuracy": 0.9271934297349718,
-      "eval_runtime": 3.6242,
-      "eval_samples_per_second": 9.933,
-      "eval_steps_per_second": 0.552,
-      "step": 220
-    },
-    {
-      "epoch": 20.09,
-      "learning_rate": 8.036363636363636e-05,
-      "loss": 0.5671,
-      "step": 221
-    },
-    {
-      "epoch": 20.18,
-      "learning_rate": 8.072727272727273e-05,
-      "loss": 0.5755,
-      "step": 222
-    },
-    {
-      "epoch": 20.27,
-      "learning_rate": 8.109090909090909e-05,
-      "loss": 0.5232,
-      "step": 223
-    },
-    {
-      "epoch": 20.36,
-      "learning_rate": 8.145454545454546e-05,
-      "loss": 0.5439,
-      "step": 224
-    },
-    {
-      "epoch": 20.45,
-      "learning_rate": 8.181818181818183e-05,
-      "loss": 0.6096,
-      "step": 225
-    },
-    {
-      "epoch": 20.55,
-      "learning_rate": 8.218181818181819e-05,
-      "loss": 0.5649,
-      "step": 226
-    },
-    {
-      "epoch": 20.64,
-      "learning_rate": 8.254545454545454e-05,
-      "loss": 0.4974,
-      "step": 227
-    },
-    {
-      "epoch": 20.73,
-      "learning_rate": 8.290909090909091e-05,
-      "loss": 0.5754,
-      "step": 228
-    },
-    {
-      "epoch": 20.82,
-      "learning_rate": 8.327272727272728e-05,
-      "loss": 0.573,
-      "step": 229
-    },
-    {
-      "epoch": 20.91,
-      "learning_rate": 8.363636363636364e-05,
-      "loss": 0.5127,
-      "step": 230
-    },
-    {
-      "epoch": 21.0,
-      "learning_rate": 8.4e-05,
-      "loss": 0.5175,
-      "step": 231
-    },
-    {
-      "epoch": 21.09,
-      "learning_rate": 8.436363636363637e-05,
-      "loss": 0.4559,
-      "step": 232
-    },
-    {
-      "epoch": 21.18,
-      "learning_rate": 8.472727272727274e-05,
-      "loss": 0.5219,
-      "step": 233
-    },
-    {
-      "epoch": 21.27,
-      "learning_rate": 8.50909090909091e-05,
-      "loss": 0.5869,
-      "step": 234
-    },
-    {
-      "epoch": 21.36,
-      "learning_rate": 8.545454545454545e-05,
-      "loss": 0.5397,
-      "step": 235
-    },
-    {
-      "epoch": 21.45,
-      "learning_rate": 8.581818181818182e-05,
-      "loss": 0.5005,
-      "step": 236
-    },
-    {
-      "epoch": 21.55,
-      "learning_rate": 8.618181818181819e-05,
-      "loss": 0.5077,
-      "step": 237
-    },
-    {
-      "epoch": 21.64,
-      "learning_rate": 8.654545454545456e-05,
-      "loss": 0.5302,
-      "step": 238
-    },
-    {
-      "epoch": 21.73,
-      "learning_rate": 8.690909090909091e-05,
-      "loss": 0.4989,
-      "step": 239
-    },
-    {
-      "epoch": 21.82,
-      "learning_rate": 8.727272727272727e-05,
-      "loss": 0.5048,
-      "step": 240
-    },
-    {
-      "epoch": 21.82,
-      "eval_accuracy_NAT": 0.9648992852318179,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9155663869693978,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7918578763233223,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9534394131951288,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9688599333596419,
-      "eval_accuracy_steel": 0.02671327748829233,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9519438077397182,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6809167508730699,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6172144657086427,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8674866078353928,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8809945905135638,
-      "eval_iou_steel": 0.02670733215183613,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.43356192111968994,
-      "eval_mean_accuracy": 0.3554873978898155,
-      "eval_mean_iou": 0.30963565806324794,
-      "eval_overall_accuracy": 0.9298235575358073,
-      "eval_runtime": 3.386,
-      "eval_samples_per_second": 10.632,
-      "eval_steps_per_second": 0.591,
-      "step": 240
-    },
-    {
-      "epoch": 21.91,
-      "learning_rate": 8.763636363636364e-05,
-      "loss": 0.4869,
-      "step": 241
-    },
-    {
-      "epoch": 22.0,
-      "learning_rate": 8.800000000000001e-05,
-      "loss": 0.4789,
-      "step": 242
-    },
-    {
-      "epoch": 22.09,
-      "learning_rate": 8.836363636363637e-05,
-      "loss": 0.4677,
-      "step": 243
-    },
-    {
-      "epoch": 22.18,
-      "learning_rate": 8.872727272727274e-05,
-      "loss": 0.4919,
-      "step": 244
-    },
-    {
-      "epoch": 22.27,
-      "learning_rate": 8.90909090909091e-05,
-      "loss": 0.4705,
-      "step": 245
-    },
-    {
-      "epoch": 22.36,
-      "learning_rate": 8.945454545454546e-05,
-      "loss": 0.4609,
-      "step": 246
-    },
-    {
-      "epoch": 22.45,
-      "learning_rate": 8.981818181818182e-05,
-      "loss": 0.466,
-      "step": 247
-    },
-    {
-      "epoch": 22.55,
-      "learning_rate": 9.018181818181819e-05,
-      "loss": 0.4155,
-      "step": 248
-    },
-    {
-      "epoch": 22.64,
-      "learning_rate": 9.054545454545455e-05,
-      "loss": 0.4654,
-      "step": 249
-    },
-    {
-      "epoch": 22.73,
-      "learning_rate": 9.090909090909092e-05,
-      "loss": 0.4972,
-      "step": 250
-    },
-    {
-      "epoch": 22.82,
-      "learning_rate": 9.127272727272727e-05,
-      "loss": 0.4541,
-      "step": 251
-    },
-    {
-      "epoch": 22.91,
-      "learning_rate": 9.163636363636364e-05,
-      "loss": 0.5064,
-      "step": 252
-    },
-    {
-      "epoch": 23.0,
-      "learning_rate": 9.200000000000001e-05,
-      "loss": 0.4212,
-      "step": 253
-    },
-    {
-      "epoch": 23.09,
-      "learning_rate": 9.236363636363636e-05,
-      "loss": 0.4343,
-      "step": 254
-    },
-    {
-      "epoch": 23.18,
-      "learning_rate": 9.272727272727273e-05,
-      "loss": 0.4641,
-      "step": 255
-    },
-    {
-      "epoch": 23.27,
-      "learning_rate": 9.30909090909091e-05,
-      "loss": 0.4081,
-      "step": 256
-    },
-    {
-      "epoch": 23.36,
-      "learning_rate": 9.345454545454547e-05,
-      "loss": 0.4325,
-      "step": 257
-    },
-    {
-      "epoch": 23.45,
-      "learning_rate": 9.381818181818182e-05,
-      "loss": 0.4329,
-      "step": 258
-    },
-    {
-      "epoch": 23.55,
-      "learning_rate": 9.418181818181818e-05,
-      "loss": 0.3952,
-      "step": 259
-    },
-    {
-      "epoch": 23.64,
-      "learning_rate": 9.454545454545455e-05,
-      "loss": 0.4344,
-      "step": 260
-    },
-    {
-      "epoch": 23.64,
-      "eval_accuracy_NAT": 0.9641530734608346,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9097144727241513,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.8163653330630476,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9233688074903234,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9851181532824781,
-      "eval_accuracy_steel": 0.11447958577930216,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9528103888603495,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6748117850842851,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6165737047586114,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.891319598049778,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.878271142291361,
-      "eval_iou_steel": 0.11441826736875067,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.3822040557861328,
-      "eval_mean_accuracy": 0.3625538019846259,
-      "eval_mean_iou": 0.3175542220317797,
-      "eval_overall_accuracy": 0.9310041003757052,
-      "eval_runtime": 3.3973,
-      "eval_samples_per_second": 10.597,
-      "eval_steps_per_second": 0.589,
-      "step": 260
-    },
-    {
-      "epoch": 23.73,
-      "learning_rate": 9.490909090909092e-05,
-      "loss": 0.4273,
-      "step": 261
-    },
-    {
-      "epoch": 23.82,
-      "learning_rate": 9.527272727272728e-05,
-      "loss": 0.435,
-      "step": 262
-    },
-    {
-      "epoch": 23.91,
-      "learning_rate": 9.563636363636365e-05,
-      "loss": 0.4056,
-      "step": 263
-    },
-    {
-      "epoch": 24.0,
-      "learning_rate": 9.6e-05,
-      "loss": 0.5192,
-      "step": 264
-    },
-    {
-      "epoch": 24.09,
-      "learning_rate": 9.636363636363637e-05,
-      "loss": 0.3617,
-      "step": 265
-    },
-    {
-      "epoch": 24.18,
-      "learning_rate": 9.672727272727273e-05,
-      "loss": 0.4952,
-      "step": 266
-    },
-    {
-      "epoch": 24.27,
-      "learning_rate": 9.70909090909091e-05,
-      "loss": 0.3757,
-      "step": 267
-    },
-    {
-      "epoch": 24.36,
-      "learning_rate": 9.745454545454546e-05,
-      "loss": 0.3992,
-      "step": 268
-    },
-    {
-      "epoch": 24.45,
-      "learning_rate": 9.781818181818183e-05,
-      "loss": 0.4048,
-      "step": 269
-    },
-    {
-      "epoch": 24.55,
-      "learning_rate": 9.818181818181818e-05,
-      "loss": 0.4214,
-      "step": 270
-    },
-    {
-      "epoch": 24.64,
-      "learning_rate": 9.854545454545455e-05,
-      "loss": 0.4157,
-      "step": 271
-    },
-    {
-      "epoch": 24.73,
-      "learning_rate": 9.890909090909092e-05,
-      "loss": 0.3977,
-      "step": 272
-    },
-    {
-      "epoch": 24.82,
-      "learning_rate": 9.927272727272728e-05,
-      "loss": 0.4002,
-      "step": 273
-    },
-    {
-      "epoch": 24.91,
-      "learning_rate": 9.963636363636363e-05,
-      "loss": 0.3822,
-      "step": 274
-    },
-    {
-      "epoch": 25.0,
-      "learning_rate": 0.0001,
-      "loss": 0.3723,
-      "step": 275
-    },
-    {
-      "epoch": 25.09,
-      "learning_rate": 9.998086124401914e-05,
-      "loss": 0.3567,
-      "step": 276
-    },
-    {
-      "epoch": 25.18,
-      "learning_rate": 9.996172248803828e-05,
-      "loss": 0.4088,
-      "step": 277
-    },
-    {
-      "epoch": 25.27,
-      "learning_rate": 9.994258373205743e-05,
-      "loss": 0.3251,
-      "step": 278
-    },
-    {
-      "epoch": 25.36,
-      "learning_rate": 9.992344497607656e-05,
-      "loss": 0.3713,
-      "step": 279
-    },
-    {
-      "epoch": 25.45,
-      "learning_rate": 9.99043062200957e-05,
-      "loss": 0.3584,
-      "step": 280
-    },
-    {
-      "epoch": 25.45,
-      "eval_accuracy_NAT": 0.9740316892088849,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9090894458989657,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7792092046260796,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9691252704093293,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9794708596268284,
-      "eval_accuracy_steel": 0.3456236396016094,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.95805661815918,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7268800229928941,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6364220262053609,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9027517768841731,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.9025692579805912,
-      "eval_iou_steel": 0.33965596869201664,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.3192930519580841,
-      "eval_mean_accuracy": 0.3812730853362844,
-      "eval_mean_iou": 0.3435642823780166,
-      "eval_overall_accuracy": 0.9417492548624674,
-      "eval_runtime": 3.3199,
-      "eval_samples_per_second": 10.844,
-      "eval_steps_per_second": 0.602,
-      "step": 280
-    },
-    {
-      "epoch": 25.55,
-      "learning_rate": 9.988516746411484e-05,
-      "loss": 0.389,
-      "step": 281
-    },
-    {
-      "epoch": 25.64,
-      "learning_rate": 9.986602870813397e-05,
-      "loss": 0.334,
-      "step": 282
-    },
-    {
-      "epoch": 25.73,
-      "learning_rate": 9.984688995215312e-05,
-      "loss": 0.3992,
-      "step": 283
-    },
-    {
-      "epoch": 25.82,
-      "learning_rate": 9.982775119617226e-05,
-      "loss": 0.3609,
-      "step": 284
-    },
-    {
-      "epoch": 25.91,
-      "learning_rate": 9.980861244019139e-05,
-      "loss": 0.3966,
-      "step": 285
-    },
-    {
-      "epoch": 26.0,
-      "learning_rate": 9.978947368421054e-05,
-      "loss": 0.3505,
-      "step": 286
-    },
-    {
-      "epoch": 26.09,
-      "learning_rate": 9.977033492822967e-05,
-      "loss": 0.3183,
-      "step": 287
-    },
-    {
-      "epoch": 26.18,
-      "learning_rate": 9.97511961722488e-05,
-      "loss": 0.3844,
-      "step": 288
-    },
-    {
-      "epoch": 26.27,
-      "learning_rate": 9.973205741626795e-05,
-      "loss": 0.3578,
-      "step": 289
-    },
-    {
-      "epoch": 26.36,
-      "learning_rate": 9.97129186602871e-05,
-      "loss": 0.3075,
-      "step": 290
-    },
-    {
-      "epoch": 26.45,
-      "learning_rate": 9.969377990430623e-05,
-      "loss": 0.3744,
-      "step": 291
-    },
-    {
-      "epoch": 26.55,
-      "learning_rate": 9.967464114832537e-05,
-      "loss": 0.3404,
-      "step": 292
-    },
-    {
-      "epoch": 26.64,
-      "learning_rate": 9.96555023923445e-05,
-      "loss": 0.3195,
-      "step": 293
-    },
-    {
-      "epoch": 26.73,
-      "learning_rate": 9.963636363636363e-05,
-      "loss": 0.3336,
-      "step": 294
-    },
-    {
-      "epoch": 26.82,
-      "learning_rate": 9.961722488038278e-05,
-      "loss": 0.3474,
-      "step": 295
-    },
-    {
-      "epoch": 26.91,
-      "learning_rate": 9.959808612440193e-05,
-      "loss": 0.3756,
-      "step": 296
-    },
-    {
-      "epoch": 27.0,
-      "learning_rate": 9.957894736842106e-05,
-      "loss": 0.3434,
-      "step": 297
-    },
-    {
-      "epoch": 27.09,
-      "learning_rate": 9.95598086124402e-05,
-      "loss": 0.3108,
-      "step": 298
-    },
-    {
-      "epoch": 27.18,
-      "learning_rate": 9.954066985645934e-05,
-      "loss": 0.3695,
-      "step": 299
-    },
-    {
-      "epoch": 27.27,
-      "learning_rate": 9.952153110047847e-05,
-      "loss": 0.3006,
-      "step": 300
-    },
-    {
-      "epoch": 27.27,
-      "eval_accuracy_NAT": 0.9688187938789814,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.927841591913816,
-      "eval_accuracy_concrete tiles": 0.07698893169449031,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.8511413163146545,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9460825865461004,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9802510659198892,
-      "eval_accuracy_steel": 0.44872534793219443,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9582652306653726,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7206670271205051,
-      "eval_iou_concrete tiles": 0.07698893169449031,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6336097328715978,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9164337703896493,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.9045607087593204,
-      "eval_iou_steel": 0.42411514424200863,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.29558247327804565,
-      "eval_mean_accuracy": 0.3999884334000097,
-      "eval_mean_iou": 0.3565108112109957,
-      "eval_overall_accuracy": 0.9419907463921441,
-      "eval_runtime": 3.3499,
-      "eval_samples_per_second": 10.746,
-      "eval_steps_per_second": 0.597,
-      "step": 300
-    },
-    {
-      "epoch": 27.36,
-      "learning_rate": 9.950239234449761e-05,
-      "loss": 0.3462,
-      "step": 301
-    },
-    {
-      "epoch": 27.45,
-      "learning_rate": 9.948325358851676e-05,
-      "loss": 0.3003,
-      "step": 302
-    },
-    {
-      "epoch": 27.55,
-      "learning_rate": 9.946411483253589e-05,
-      "loss": 0.3295,
-      "step": 303
-    },
-    {
-      "epoch": 27.64,
-      "learning_rate": 9.944497607655504e-05,
-      "loss": 0.2829,
-      "step": 304
-    },
-    {
-      "epoch": 27.73,
-      "learning_rate": 9.942583732057417e-05,
-      "loss": 0.2852,
-      "step": 305
-    },
-    {
-      "epoch": 27.82,
-      "learning_rate": 9.94066985645933e-05,
-      "loss": 0.3215,
-      "step": 306
-    },
-    {
-      "epoch": 27.91,
-      "learning_rate": 9.938755980861245e-05,
-      "loss": 0.3226,
-      "step": 307
-    },
-    {
-      "epoch": 28.0,
-      "learning_rate": 9.936842105263159e-05,
-      "loss": 0.3365,
-      "step": 308
-    },
-    {
-      "epoch": 28.09,
-      "learning_rate": 9.934928229665072e-05,
-      "loss": 0.329,
-      "step": 309
-    },
-    {
-      "epoch": 28.18,
-      "learning_rate": 9.933014354066987e-05,
-      "loss": 0.2809,
-      "step": 310
-    },
-    {
-      "epoch": 28.27,
-      "learning_rate": 9.9311004784689e-05,
-      "loss": 0.2855,
-      "step": 311
-    },
-    {
-      "epoch": 28.36,
-      "learning_rate": 9.929186602870813e-05,
-      "loss": 0.289,
-      "step": 312
-    },
-    {
-      "epoch": 28.45,
-      "learning_rate": 9.927272727272728e-05,
-      "loss": 0.2965,
-      "step": 313
-    },
-    {
-      "epoch": 28.55,
-      "learning_rate": 9.925358851674642e-05,
-      "loss": 0.289,
-      "step": 314
-    },
-    {
-      "epoch": 28.64,
-      "learning_rate": 9.923444976076556e-05,
-      "loss": 0.3306,
-      "step": 315
-    },
-    {
-      "epoch": 28.73,
-      "learning_rate": 9.921531100478469e-05,
-      "loss": 0.3364,
-      "step": 316
-    },
-    {
-      "epoch": 28.82,
-      "learning_rate": 9.919617224880383e-05,
-      "loss": 0.2645,
-      "step": 317
-    },
-    {
-      "epoch": 28.91,
-      "learning_rate": 9.917703349282297e-05,
-      "loss": 0.3016,
-      "step": 318
-    },
-    {
-      "epoch": 29.0,
-      "learning_rate": 9.915789473684211e-05,
-      "loss": 0.3284,
-      "step": 319
-    },
-    {
-      "epoch": 29.09,
-      "learning_rate": 9.913875598086126e-05,
-      "loss": 0.2623,
-      "step": 320
-    },
-    {
-      "epoch": 29.09,
-      "eval_accuracy_NAT": 0.9863562703315799,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8830395081333963,
-      "eval_accuracy_concrete tiles": 0.4547483642145172,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.8140201386146226,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9644021006376158,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.976765179025062,
-      "eval_accuracy_steel": 0.5599317327353077,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9602198795475638,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7984248970082988,
-      "eval_iou_concrete tiles": 0.4547483642145172,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6255153206335053,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9279770271150786,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.9274175621449311,
-      "eval_iou_steel": 0.5065979396823739,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.28042370080947876,
-      "eval_mean_accuracy": 0.4337894841301617,
-      "eval_mean_iou": 0.40006930694971304,
-      "eval_overall_accuracy": 0.9515891604953342,
-      "eval_runtime": 3.569,
-      "eval_samples_per_second": 10.087,
-      "eval_steps_per_second": 0.56,
-      "step": 320
-    },
-    {
-      "epoch": 29.18,
-      "learning_rate": 9.911961722488039e-05,
-      "loss": 0.3107,
-      "step": 321
-    },
-    {
-      "epoch": 29.27,
-      "learning_rate": 9.910047846889952e-05,
-      "loss": 0.2664,
-      "step": 322
-    },
-    {
-      "epoch": 29.36,
-      "learning_rate": 9.908133971291867e-05,
-      "loss": 0.2748,
-      "step": 323
-    },
-    {
-      "epoch": 29.45,
-      "learning_rate": 9.90622009569378e-05,
-      "loss": 0.3124,
-      "step": 324
-    },
-    {
-      "epoch": 29.55,
-      "learning_rate": 9.904306220095694e-05,
-      "loss": 0.2631,
-      "step": 325
-    },
-    {
-      "epoch": 29.64,
-      "learning_rate": 9.902392344497609e-05,
-      "loss": 0.2503,
-      "step": 326
-    },
-    {
-      "epoch": 29.73,
-      "learning_rate": 9.900478468899522e-05,
-      "loss": 0.2743,
-      "step": 327
-    },
-    {
-      "epoch": 29.82,
-      "learning_rate": 9.898564593301435e-05,
-      "loss": 0.281,
-      "step": 328
-    },
-    {
-      "epoch": 29.91,
-      "learning_rate": 9.89665071770335e-05,
-      "loss": 0.3247,
-      "step": 329
-    },
-    {
-      "epoch": 30.0,
-      "learning_rate": 9.894736842105263e-05,
-      "loss": 0.2906,
-      "step": 330
-    },
-    {
-      "epoch": 30.09,
-      "learning_rate": 9.892822966507178e-05,
-      "loss": 0.2804,
-      "step": 331
-    },
-    {
-      "epoch": 30.18,
-      "learning_rate": 9.890909090909092e-05,
-      "loss": 0.287,
-      "step": 332
-    },
-    {
-      "epoch": 30.27,
-      "learning_rate": 9.888995215311005e-05,
-      "loss": 0.2577,
-      "step": 333
-    },
-    {
-      "epoch": 30.36,
-      "learning_rate": 9.887081339712919e-05,
-      "loss": 0.2597,
-      "step": 334
-    },
-    {
-      "epoch": 30.45,
-      "learning_rate": 9.885167464114833e-05,
-      "loss": 0.2207,
-      "step": 335
-    },
-    {
-      "epoch": 30.55,
-      "learning_rate": 9.883253588516746e-05,
-      "loss": 0.2619,
-      "step": 336
-    },
-    {
-      "epoch": 30.64,
-      "learning_rate": 9.881339712918661e-05,
-      "loss": 0.2584,
-      "step": 337
-    },
-    {
-      "epoch": 30.73,
-      "learning_rate": 9.879425837320575e-05,
-      "loss": 0.2461,
-      "step": 338
-    },
-    {
-      "epoch": 30.82,
-      "learning_rate": 9.877511961722489e-05,
-      "loss": 0.2442,
-      "step": 339
-    },
-    {
-      "epoch": 30.91,
-      "learning_rate": 9.875598086124402e-05,
-      "loss": 0.2954,
-      "step": 340
-    },
-    {
-      "epoch": 30.91,
-      "eval_accuracy_NAT": 0.9780385411586345,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9337431327524787,
-      "eval_accuracy_concrete tiles": 0.5960374243258117,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.8581069981860567,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9683731425040538,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.000303255865244121,
-      "eval_accuracy_speedway bricks": 0.9773728106856282,
-      "eval_accuracy_steel": 0.5902315150715652,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9631424672668348,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7737282983554485,
-      "eval_iou_concrete tiles": 0.5960374243258117,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6824275306929647,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9282016926718113,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.000303255865244121,
-      "eval_iou_speedway bricks": 0.9314139852837332,
-      "eval_iou_steel": 0.554039516759409,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.22423362731933594,
-      "eval_mean_accuracy": 0.4540159092730364,
-      "eval_mean_iou": 0.417638013170866,
-      "eval_overall_accuracy": 0.9532714419894748,
-      "eval_runtime": 3.7621,
-      "eval_samples_per_second": 9.569,
-      "eval_steps_per_second": 0.532,
-      "step": 340
-    },
-    {
-      "epoch": 31.0,
-      "learning_rate": 9.873684210526316e-05,
-      "loss": 0.2519,
-      "step": 341
-    },
-    {
-      "epoch": 31.09,
-      "learning_rate": 9.87177033492823e-05,
-      "loss": 0.2534,
-      "step": 342
-    },
-    {
-      "epoch": 31.18,
-      "learning_rate": 9.869856459330143e-05,
-      "loss": 0.2352,
-      "step": 343
-    },
-    {
-      "epoch": 31.27,
-      "learning_rate": 9.867942583732059e-05,
-      "loss": 0.2653,
-      "step": 344
-    },
-    {
-      "epoch": 31.36,
-      "learning_rate": 9.866028708133972e-05,
-      "loss": 0.2425,
-      "step": 345
-    },
-    {
-      "epoch": 31.45,
-      "learning_rate": 9.864114832535885e-05,
-      "loss": 0.2384,
-      "step": 346
-    },
-    {
-      "epoch": 31.55,
-      "learning_rate": 9.8622009569378e-05,
-      "loss": 0.2317,
-      "step": 347
-    },
-    {
-      "epoch": 31.64,
-      "learning_rate": 9.860287081339713e-05,
-      "loss": 0.2528,
-      "step": 348
-    },
-    {
-      "epoch": 31.73,
-      "learning_rate": 9.858373205741627e-05,
-      "loss": 0.2559,
-      "step": 349
-    },
-    {
-      "epoch": 31.82,
-      "learning_rate": 9.856459330143542e-05,
-      "loss": 0.2501,
-      "step": 350
-    },
-    {
-      "epoch": 31.91,
-      "learning_rate": 9.854545454545455e-05,
-      "loss": 0.2653,
-      "step": 351
-    },
-    {
-      "epoch": 32.0,
-      "learning_rate": 9.852631578947368e-05,
-      "loss": 0.2571,
-      "step": 352
-    },
-    {
-      "epoch": 32.09,
-      "learning_rate": 9.850717703349283e-05,
-      "loss": 0.2514,
-      "step": 353
-    },
-    {
-      "epoch": 32.18,
-      "learning_rate": 9.848803827751196e-05,
-      "loss": 0.1948,
-      "step": 354
-    },
-    {
-      "epoch": 32.27,
-      "learning_rate": 9.846889952153111e-05,
-      "loss": 0.2095,
-      "step": 355
-    },
-    {
-      "epoch": 32.36,
-      "learning_rate": 9.844976076555025e-05,
-      "loss": 0.2438,
-      "step": 356
-    },
-    {
-      "epoch": 32.45,
-      "learning_rate": 9.843062200956938e-05,
-      "loss": 0.2716,
-      "step": 357
-    },
-    {
-      "epoch": 32.55,
-      "learning_rate": 9.841148325358852e-05,
-      "loss": 0.2256,
-      "step": 358
-    },
-    {
-      "epoch": 32.64,
-      "learning_rate": 9.839234449760766e-05,
-      "loss": 0.2219,
-      "step": 359
-    },
-    {
-      "epoch": 32.73,
-      "learning_rate": 9.83732057416268e-05,
-      "loss": 0.23,
-      "step": 360
-    },
-    {
-      "epoch": 32.73,
-      "eval_accuracy_NAT": 0.9899087606567651,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9128355830722349,
-      "eval_accuracy_concrete tiles": 0.7983550418883385,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7866851672567516,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9574201287323743,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.05136051608634522,
-      "eval_accuracy_speedway bricks": 0.9771881678970152,
-      "eval_accuracy_steel": 0.7017347140689928,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9622421996765484,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8176590941583679,
-      "eval_iou_concrete tiles": 0.7976721451701595,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6783643355716025,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9405345947063197,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.05084468218662155,
-      "eval_iou_speedway bricks": 0.935758543002045,
-      "eval_iou_steel": 0.6334576253525949,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.22010985016822815,
-      "eval_mean_accuracy": 0.4750375445891398,
-      "eval_mean_iou": 0.44742563229417376,
-      "eval_overall_accuracy": 0.9579209221733941,
-      "eval_runtime": 3.2647,
-      "eval_samples_per_second": 11.027,
-      "eval_steps_per_second": 0.613,
-      "step": 360
-    },
-    {
-      "epoch": 32.82,
-      "learning_rate": 9.835406698564594e-05,
-      "loss": 0.2189,
-      "step": 361
-    },
-    {
-      "epoch": 32.91,
-      "learning_rate": 9.833492822966509e-05,
-      "loss": 0.2745,
-      "step": 362
-    },
-    {
-      "epoch": 33.0,
-      "learning_rate": 9.831578947368422e-05,
-      "loss": 0.2492,
-      "step": 363
-    },
-    {
-      "epoch": 33.09,
-      "learning_rate": 9.829665071770335e-05,
-      "loss": 0.2473,
-      "step": 364
-    },
-    {
-      "epoch": 33.18,
-      "learning_rate": 9.82775119617225e-05,
-      "loss": 0.1956,
-      "step": 365
-    },
-    {
-      "epoch": 33.27,
-      "learning_rate": 9.825837320574163e-05,
-      "loss": 0.2128,
-      "step": 366
-    },
-    {
-      "epoch": 33.36,
-      "learning_rate": 9.823923444976077e-05,
-      "loss": 0.2026,
-      "step": 367
-    },
-    {
-      "epoch": 33.45,
-      "learning_rate": 9.822009569377992e-05,
-      "loss": 0.2038,
-      "step": 368
-    },
-    {
-      "epoch": 33.55,
-      "learning_rate": 9.820095693779905e-05,
-      "loss": 0.1899,
-      "step": 369
-    },
-    {
-      "epoch": 33.64,
-      "learning_rate": 9.818181818181818e-05,
-      "loss": 0.2627,
-      "step": 370
-    },
-    {
-      "epoch": 33.73,
-      "learning_rate": 9.816267942583733e-05,
-      "loss": 0.2367,
-      "step": 371
-    },
-    {
-      "epoch": 33.82,
-      "learning_rate": 9.814354066985646e-05,
-      "loss": 0.1731,
-      "step": 372
-    },
-    {
-      "epoch": 33.91,
-      "learning_rate": 9.81244019138756e-05,
-      "loss": 0.2467,
-      "step": 373
-    },
-    {
-      "epoch": 34.0,
-      "learning_rate": 9.810526315789475e-05,
-      "loss": 0.2316,
-      "step": 374
-    },
-    {
-      "epoch": 34.09,
-      "learning_rate": 9.808612440191388e-05,
-      "loss": 0.2053,
-      "step": 375
-    },
-    {
-      "epoch": 34.18,
-      "learning_rate": 9.806698564593301e-05,
-      "loss": 0.2205,
-      "step": 376
-    },
-    {
-      "epoch": 34.27,
-      "learning_rate": 9.804784688995216e-05,
-      "loss": 0.2145,
-      "step": 377
-    },
-    {
-      "epoch": 34.36,
-      "learning_rate": 9.802870813397129e-05,
-      "loss": 0.1797,
-      "step": 378
-    },
-    {
-      "epoch": 34.45,
-      "learning_rate": 9.800956937799044e-05,
-      "loss": 0.1891,
-      "step": 379
-    },
-    {
-      "epoch": 34.55,
-      "learning_rate": 9.799043062200958e-05,
-      "loss": 0.2269,
-      "step": 380
-    },
-    {
-      "epoch": 34.55,
-      "eval_accuracy_NAT": 0.9863255678373375,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9324032147302459,
-      "eval_accuracy_concrete tiles": 0.6486577386412279,
-      "eval_accuracy_dark bricks": 0.007967360763634964,
-      "eval_accuracy_grass": 0.8395411667255004,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9571262909815995,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.06426267471673146,
-      "eval_accuracy_speedway bricks": 0.9594606499667522,
-      "eval_accuracy_steel": 0.8058587824022162,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9662984571502657,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7832327773177176,
-      "eval_iou_concrete tiles": 0.6462667925792792,
-      "eval_iou_dark bricks": 0.007967360763634964,
-      "eval_iou_grass": 0.7368109469370477,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9331737038058792,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.048589831780376465,
-      "eval_iou_speedway bricks": 0.9318452595485027,
-      "eval_iou_steel": 0.6573165564873534,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.20067629218101501,
-      "eval_mean_accuracy": 0.47704641898194206,
-      "eval_mean_iou": 0.4393462835669275,
-      "eval_overall_accuracy": 0.9566695955064561,
-      "eval_runtime": 3.3404,
-      "eval_samples_per_second": 10.777,
-      "eval_steps_per_second": 0.599,
-      "step": 380
-    },
-    {
-      "epoch": 34.64,
-      "learning_rate": 9.797129186602872e-05,
-      "loss": 0.1986,
-      "step": 381
-    },
-    {
-      "epoch": 34.73,
-      "learning_rate": 9.795215311004785e-05,
-      "loss": 0.2067,
-      "step": 382
-    },
-    {
-      "epoch": 34.82,
-      "learning_rate": 9.793301435406699e-05,
-      "loss": 0.2059,
-      "step": 383
-    },
-    {
-      "epoch": 34.91,
-      "learning_rate": 9.791387559808612e-05,
-      "loss": 0.1862,
-      "step": 384
-    },
-    {
-      "epoch": 35.0,
-      "learning_rate": 9.789473684210527e-05,
-      "loss": 0.2052,
-      "step": 385
-    },
-    {
-      "epoch": 35.09,
-      "learning_rate": 9.787559808612442e-05,
-      "loss": 0.2041,
-      "step": 386
-    },
-    {
-      "epoch": 35.18,
-      "learning_rate": 9.785645933014355e-05,
-      "loss": 0.182,
-      "step": 387
-    },
-    {
-      "epoch": 35.27,
-      "learning_rate": 9.783732057416268e-05,
-      "loss": 0.1796,
-      "step": 388
-    },
-    {
-      "epoch": 35.36,
-      "learning_rate": 9.781818181818183e-05,
-      "loss": 0.2155,
-      "step": 389
-    },
-    {
-      "epoch": 35.45,
-      "learning_rate": 9.779904306220096e-05,
-      "loss": 0.2244,
-      "step": 390
-    },
-    {
-      "epoch": 35.55,
-      "learning_rate": 9.77799043062201e-05,
-      "loss": 0.1773,
-      "step": 391
-    },
-    {
-      "epoch": 35.64,
-      "learning_rate": 9.776076555023923e-05,
-      "loss": 0.2018,
-      "step": 392
-    },
-    {
-      "epoch": 35.73,
-      "learning_rate": 9.774162679425838e-05,
-      "loss": 0.1777,
-      "step": 393
-    },
-    {
-      "epoch": 35.82,
-      "learning_rate": 9.772248803827751e-05,
-      "loss": 0.2042,
-      "step": 394
-    },
-    {
-      "epoch": 35.91,
-      "learning_rate": 9.770334928229666e-05,
-      "loss": 0.2192,
-      "step": 395
-    },
-    {
-      "epoch": 36.0,
-      "learning_rate": 9.768421052631579e-05,
-      "loss": 0.168,
-      "step": 396
-    },
-    {
-      "epoch": 36.09,
-      "learning_rate": 9.766507177033494e-05,
-      "loss": 0.1912,
-      "step": 397
-    },
-    {
-      "epoch": 36.18,
-      "learning_rate": 9.764593301435407e-05,
-      "loss": 0.2357,
-      "step": 398
-    },
-    {
-      "epoch": 36.27,
-      "learning_rate": 9.762679425837321e-05,
-      "loss": 0.1589,
-      "step": 399
-    },
-    {
-      "epoch": 36.36,
-      "learning_rate": 9.760765550239235e-05,
-      "loss": 0.2013,
-      "step": 400
-    },
-    {
-      "epoch": 36.36,
-      "eval_accuracy_NAT": 0.9866195353971526,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9297716640199151,
-      "eval_accuracy_concrete tiles": 0.9206567602274812,
-      "eval_accuracy_dark bricks": 0.06497055540587353,
-      "eval_accuracy_grass": 0.8123949293470852,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9425021191385112,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.11143274611970336,
-      "eval_accuracy_speedway bricks": 0.9790852820388426,
-      "eval_accuracy_steel": 0.7739017874810369,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9663282002856404,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7973902220749851,
-      "eval_iou_concrete tiles": 0.9056757000631636,
-      "eval_iou_dark bricks": 0.06494555807779616,
-      "eval_iou_grass": 0.7393694175127873,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9248623570174189,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0789437706294799,
-      "eval_iou_speedway bricks": 0.9390875312528938,
-      "eval_iou_steel": 0.6864085763594349,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.18810129165649414,
-      "eval_mean_accuracy": 0.5016411830135079,
-      "eval_mean_iou": 0.4694624102518154,
-      "eval_overall_accuracy": 0.9588501188490126,
-      "eval_runtime": 3.3523,
-      "eval_samples_per_second": 10.739,
-      "eval_steps_per_second": 0.597,
-      "step": 400
-    },
-    {
-      "epoch": 36.45,
-      "learning_rate": 9.758851674641149e-05,
-      "loss": 0.1684,
-      "step": 401
-    },
-    {
-      "epoch": 36.55,
-      "learning_rate": 9.756937799043062e-05,
-      "loss": 0.1775,
-      "step": 402
-    },
-    {
-      "epoch": 36.64,
-      "learning_rate": 9.755023923444977e-05,
-      "loss": 0.2015,
-      "step": 403
-    },
-    {
-      "epoch": 36.73,
-      "learning_rate": 9.75311004784689e-05,
-      "loss": 0.1897,
-      "step": 404
-    },
-    {
-      "epoch": 36.82,
-      "learning_rate": 9.751196172248805e-05,
-      "loss": 0.203,
-      "step": 405
-    },
-    {
-      "epoch": 36.91,
-      "learning_rate": 9.749282296650718e-05,
-      "loss": 0.1627,
-      "step": 406
-    },
-    {
-      "epoch": 37.0,
-      "learning_rate": 9.747368421052632e-05,
-      "loss": 0.1958,
-      "step": 407
-    },
-    {
-      "epoch": 37.09,
-      "learning_rate": 9.745454545454546e-05,
-      "loss": 0.1773,
-      "step": 408
-    },
-    {
-      "epoch": 37.18,
-      "learning_rate": 9.74354066985646e-05,
-      "loss": 0.1828,
-      "step": 409
-    },
-    {
-      "epoch": 37.27,
-      "learning_rate": 9.741626794258373e-05,
-      "loss": 0.1771,
-      "step": 410
-    },
-    {
-      "epoch": 37.36,
-      "learning_rate": 9.739712918660288e-05,
-      "loss": 0.1984,
-      "step": 411
-    },
-    {
-      "epoch": 37.45,
-      "learning_rate": 9.737799043062201e-05,
-      "loss": 0.1759,
-      "step": 412
-    },
-    {
-      "epoch": 37.55,
-      "learning_rate": 9.735885167464116e-05,
-      "loss": 0.1688,
-      "step": 413
-    },
-    {
-      "epoch": 37.64,
-      "learning_rate": 9.733971291866029e-05,
-      "loss": 0.1501,
-      "step": 414
-    },
-    {
-      "epoch": 37.73,
-      "learning_rate": 9.732057416267943e-05,
-      "loss": 0.2254,
-      "step": 415
-    },
-    {
-      "epoch": 37.82,
-      "learning_rate": 9.730143540669857e-05,
-      "loss": 0.1663,
-      "step": 416
-    },
-    {
-      "epoch": 37.91,
-      "learning_rate": 9.728229665071771e-05,
-      "loss": 0.1675,
-      "step": 417
-    },
-    {
-      "epoch": 38.0,
-      "learning_rate": 9.726315789473684e-05,
-      "loss": 0.1798,
-      "step": 418
-    },
-    {
-      "epoch": 38.09,
-      "learning_rate": 9.724401913875598e-05,
-      "loss": 0.1869,
-      "step": 419
-    },
-    {
-      "epoch": 38.18,
-      "learning_rate": 9.722488038277513e-05,
-      "loss": 0.1744,
-      "step": 420
-    },
-    {
-      "epoch": 38.18,
-      "eval_accuracy_NAT": 0.9872814035345276,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9265861732263188,
-      "eval_accuracy_concrete tiles": 0.9576224545954871,
-      "eval_accuracy_dark bricks": 0.44698048573957894,
-      "eval_accuracy_grass": 0.8165645522635845,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9749331005203226,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.15289609351308137,
-      "eval_accuracy_speedway bricks": 0.9732497251477444,
-      "eval_accuracy_steel": 0.7794752984631621,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9679813587175312,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.815121023407062,
-      "eval_iou_concrete tiles": 0.9066697545159796,
-      "eval_iou_dark bricks": 0.44388808195092117,
-      "eval_iou_grass": 0.739656247724517,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9384433092046377,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.10387907621420142,
-      "eval_iou_speedway bricks": 0.9480872164858768,
-      "eval_iou_steel": 0.692004772396226,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1691574603319168,
-      "eval_mean_accuracy": 0.5396607143849083,
-      "eval_mean_iou": 0.5042869877397655,
-      "eval_overall_accuracy": 0.9623503155178494,
-      "eval_runtime": 3.5497,
-      "eval_samples_per_second": 10.142,
-      "eval_steps_per_second": 0.563,
-      "step": 420
-    },
-    {
-      "epoch": 38.27,
-      "learning_rate": 9.720574162679427e-05,
-      "loss": 0.1551,
-      "step": 421
-    },
-    {
-      "epoch": 38.36,
-      "learning_rate": 9.71866028708134e-05,
-      "loss": 0.1663,
-      "step": 422
-    },
-    {
-      "epoch": 38.45,
-      "learning_rate": 9.716746411483254e-05,
-      "loss": 0.209,
-      "step": 423
-    },
-    {
-      "epoch": 38.55,
-      "learning_rate": 9.714832535885168e-05,
-      "loss": 0.1747,
-      "step": 424
-    },
-    {
-      "epoch": 38.64,
-      "learning_rate": 9.712918660287081e-05,
-      "loss": 0.1779,
-      "step": 425
-    },
-    {
-      "epoch": 38.73,
-      "learning_rate": 9.711004784688997e-05,
-      "loss": 0.1582,
-      "step": 426
-    },
-    {
-      "epoch": 38.82,
-      "learning_rate": 9.70909090909091e-05,
-      "loss": 0.1623,
-      "step": 427
-    },
-    {
-      "epoch": 38.91,
-      "learning_rate": 9.707177033492823e-05,
-      "loss": 0.1641,
-      "step": 428
-    },
-    {
-      "epoch": 39.0,
-      "learning_rate": 9.705263157894738e-05,
-      "loss": 0.1564,
-      "step": 429
-    },
-    {
-      "epoch": 39.09,
-      "learning_rate": 9.703349282296651e-05,
-      "loss": 0.1544,
-      "step": 430
-    },
-    {
-      "epoch": 39.18,
-      "learning_rate": 9.701435406698564e-05,
-      "loss": 0.1757,
-      "step": 431
-    },
-    {
-      "epoch": 39.27,
-      "learning_rate": 9.69952153110048e-05,
-      "loss": 0.1663,
-      "step": 432
-    },
-    {
-      "epoch": 39.36,
-      "learning_rate": 9.697607655502393e-05,
-      "loss": 0.1729,
-      "step": 433
-    },
-    {
-      "epoch": 39.45,
-      "learning_rate": 9.695693779904306e-05,
-      "loss": 0.1415,
-      "step": 434
-    },
-    {
-      "epoch": 39.55,
-      "learning_rate": 9.693779904306221e-05,
-      "loss": 0.162,
-      "step": 435
-    },
-    {
-      "epoch": 39.64,
-      "learning_rate": 9.691866028708134e-05,
-      "loss": 0.1516,
-      "step": 436
-    },
-    {
-      "epoch": 39.73,
-      "learning_rate": 9.689952153110047e-05,
-      "loss": 0.1952,
-      "step": 437
-    },
-    {
-      "epoch": 39.82,
-      "learning_rate": 9.688038277511963e-05,
-      "loss": 0.1811,
-      "step": 438
-    },
-    {
-      "epoch": 39.91,
-      "learning_rate": 9.686124401913876e-05,
-      "loss": 0.1563,
-      "step": 439
-    },
-    {
-      "epoch": 40.0,
-      "learning_rate": 9.68421052631579e-05,
-      "loss": 0.1487,
-      "step": 440
-    },
-    {
-      "epoch": 40.0,
-      "eval_accuracy_NAT": 0.9854990708084214,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9130233593716468,
-      "eval_accuracy_concrete tiles": 0.9764874946492998,
-      "eval_accuracy_dark bricks": 0.4822370193603018,
-      "eval_accuracy_grass": 0.888377831446576,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9663670650203692,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5104898960659444,
-      "eval_accuracy_speedway bricks": 0.9811471265116875,
-      "eval_accuracy_steel": 0.7723682474770793,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9682572236447702,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8369987015669725,
-      "eval_iou_concrete tiles": 0.8658768029497885,
-      "eval_iou_dark bricks": 0.48013029315960915,
-      "eval_iou_grass": 0.7608266184552388,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9352708011745113,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3195120267798599,
-      "eval_iou_speedway bricks": 0.9518359901843249,
-      "eval_iou_steel": 0.6969334007856207,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.15717998147010803,
-      "eval_mean_accuracy": 0.5750767008239482,
-      "eval_mean_iou": 0.5242801429769766,
-      "eval_overall_accuracy": 0.9644674725002713,
-      "eval_runtime": 3.2977,
-      "eval_samples_per_second": 10.917,
-      "eval_steps_per_second": 0.606,
-      "step": 440
-    },
-    {
-      "epoch": 40.09,
-      "learning_rate": 9.682296650717704e-05,
-      "loss": 0.1583,
-      "step": 441
-    },
-    {
-      "epoch": 40.18,
-      "learning_rate": 9.680382775119617e-05,
-      "loss": 0.1562,
-      "step": 442
-    },
-    {
-      "epoch": 40.27,
-      "learning_rate": 9.67846889952153e-05,
-      "loss": 0.1468,
-      "step": 443
-    },
-    {
-      "epoch": 40.36,
-      "learning_rate": 9.676555023923446e-05,
-      "loss": 0.1619,
-      "step": 444
-    },
-    {
-      "epoch": 40.45,
-      "learning_rate": 9.67464114832536e-05,
-      "loss": 0.1481,
-      "step": 445
-    },
-    {
-      "epoch": 40.55,
-      "learning_rate": 9.672727272727273e-05,
-      "loss": 0.1648,
-      "step": 446
-    },
-    {
-      "epoch": 40.64,
-      "learning_rate": 9.670813397129187e-05,
-      "loss": 0.1371,
-      "step": 447
-    },
-    {
-      "epoch": 40.73,
-      "learning_rate": 9.6688995215311e-05,
-      "loss": 0.1399,
-      "step": 448
-    },
-    {
-      "epoch": 40.82,
-      "learning_rate": 9.666985645933014e-05,
-      "loss": 0.1585,
-      "step": 449
-    },
-    {
-      "epoch": 40.91,
-      "learning_rate": 9.66507177033493e-05,
-      "loss": 0.2047,
-      "step": 450
-    },
-    {
-      "epoch": 41.0,
-      "learning_rate": 9.663157894736843e-05,
-      "loss": 0.1351,
-      "step": 451
-    },
-    {
-      "epoch": 41.09,
-      "learning_rate": 9.661244019138756e-05,
-      "loss": 0.1642,
-      "step": 452
-    },
-    {
-      "epoch": 41.18,
-      "learning_rate": 9.659330143540671e-05,
-      "loss": 0.1371,
-      "step": 453
-    },
-    {
-      "epoch": 41.27,
-      "learning_rate": 9.657416267942584e-05,
-      "loss": 0.1615,
-      "step": 454
-    },
-    {
-      "epoch": 41.36,
-      "learning_rate": 9.655502392344497e-05,
-      "loss": 0.1449,
-      "step": 455
-    },
-    {
-      "epoch": 41.45,
-      "learning_rate": 9.653588516746413e-05,
-      "loss": 0.1626,
-      "step": 456
-    },
-    {
-      "epoch": 41.55,
-      "learning_rate": 9.651674641148326e-05,
-      "loss": 0.1469,
-      "step": 457
-    },
-    {
-      "epoch": 41.64,
-      "learning_rate": 9.64976076555024e-05,
-      "loss": 0.1539,
-      "step": 458
-    },
-    {
-      "epoch": 41.73,
-      "learning_rate": 9.647846889952154e-05,
-      "loss": 0.1418,
-      "step": 459
-    },
-    {
-      "epoch": 41.82,
-      "learning_rate": 9.645933014354067e-05,
-      "loss": 0.1535,
-      "step": 460
-    },
-    {
-      "epoch": 41.82,
-      "eval_accuracy_NAT": 0.9876124258287733,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9293263659384523,
-      "eval_accuracy_concrete tiles": 0.9816241668195438,
-      "eval_accuracy_dark bricks": 0.4997498171740888,
-      "eval_accuracy_grass": 0.7922108787663787,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.961718866197413,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4520166515038734,
-      "eval_accuracy_speedway bricks": 0.9859514594624602,
-      "eval_accuracy_steel": 0.7584014906668426,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9691685609917259,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8188484670138827,
-      "eval_iou_concrete tiles": 0.8976904149424002,
-      "eval_iou_dark bricks": 0.49659603763481985,
-      "eval_iou_grass": 0.7413021722493671,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9431274450396009,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.30079436423342937,
-      "eval_iou_speedway bricks": 0.9503931635663969,
-      "eval_iou_steel": 0.7010303778560215,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14945927262306213,
-      "eval_mean_accuracy": 0.5652778555659866,
-      "eval_mean_iou": 0.5245346925790495,
-      "eval_overall_accuracy": 0.9642071194118924,
-      "eval_runtime": 3.2482,
-      "eval_samples_per_second": 11.083,
-      "eval_steps_per_second": 0.616,
-      "step": 460
-    },
-    {
-      "epoch": 41.91,
-      "learning_rate": 9.64401913875598e-05,
-      "loss": 0.139,
-      "step": 461
-    },
-    {
-      "epoch": 42.0,
-      "learning_rate": 9.642105263157896e-05,
-      "loss": 0.1787,
-      "step": 462
-    },
-    {
-      "epoch": 42.09,
-      "learning_rate": 9.64019138755981e-05,
-      "loss": 0.1197,
-      "step": 463
-    },
-    {
-      "epoch": 42.18,
-      "learning_rate": 9.638277511961723e-05,
-      "loss": 0.1348,
-      "step": 464
-    },
-    {
-      "epoch": 42.27,
-      "learning_rate": 9.636363636363637e-05,
-      "loss": 0.1931,
-      "step": 465
-    },
-    {
-      "epoch": 42.36,
-      "learning_rate": 9.63444976076555e-05,
-      "loss": 0.1553,
-      "step": 466
-    },
-    {
-      "epoch": 42.45,
-      "learning_rate": 9.632535885167464e-05,
-      "loss": 0.173,
-      "step": 467
-    },
-    {
-      "epoch": 42.55,
-      "learning_rate": 9.63062200956938e-05,
-      "loss": 0.1385,
-      "step": 468
-    },
-    {
-      "epoch": 42.64,
-      "learning_rate": 9.628708133971293e-05,
-      "loss": 0.1554,
-      "step": 469
-    },
-    {
-      "epoch": 42.73,
-      "learning_rate": 9.626794258373206e-05,
-      "loss": 0.1524,
-      "step": 470
-    },
-    {
-      "epoch": 42.82,
-      "learning_rate": 9.62488038277512e-05,
-      "loss": 0.1848,
-      "step": 471
-    },
-    {
-      "epoch": 42.91,
-      "learning_rate": 9.622966507177034e-05,
-      "loss": 0.136,
-      "step": 472
-    },
-    {
-      "epoch": 43.0,
-      "learning_rate": 9.621052631578947e-05,
-      "loss": 0.1304,
-      "step": 473
-    },
-    {
-      "epoch": 43.09,
-      "learning_rate": 9.619138755980861e-05,
-      "loss": 0.1465,
-      "step": 474
-    },
-    {
-      "epoch": 43.18,
-      "learning_rate": 9.617224880382776e-05,
-      "loss": 0.1184,
-      "step": 475
-    },
-    {
-      "epoch": 43.27,
-      "learning_rate": 9.615311004784689e-05,
-      "loss": 0.1246,
-      "step": 476
-    },
-    {
-      "epoch": 43.36,
-      "learning_rate": 9.613397129186604e-05,
-      "loss": 0.1453,
-      "step": 477
-    },
-    {
-      "epoch": 43.45,
-      "learning_rate": 9.611483253588517e-05,
-      "loss": 0.1274,
-      "step": 478
-    },
-    {
-      "epoch": 43.55,
-      "learning_rate": 9.60956937799043e-05,
-      "loss": 0.1481,
-      "step": 479
-    },
-    {
-      "epoch": 43.64,
-      "learning_rate": 9.607655502392345e-05,
-      "loss": 0.1375,
-      "step": 480
-    },
-    {
-      "epoch": 43.64,
-      "eval_accuracy_NAT": 0.9902858366923745,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9111308425254303,
-      "eval_accuracy_concrete tiles": 0.9678957989359751,
-      "eval_accuracy_dark bricks": 0.5515953966360032,
-      "eval_accuracy_grass": 0.8097386733399273,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.964414192726125,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.35128056681278086,
-      "eval_accuracy_speedway bricks": 0.9868125224015148,
-      "eval_accuracy_steel": 0.7366186267396609,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9694640467147071,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8278836568202813,
-      "eval_iou_concrete tiles": 0.8613174434740021,
-      "eval_iou_dark bricks": 0.5358184401405818,
-      "eval_iou_grass": 0.7511590960671789,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9439751542503015,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2399532974275922,
-      "eval_iou_speedway bricks": 0.94943779129308,
-      "eval_iou_steel": 0.6912578241661315,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14554911851882935,
-      "eval_mean_accuracy": 0.5592132659084454,
-      "eval_mean_iou": 0.5207897500272197,
-      "eval_overall_accuracy": 0.9647218916151259,
-      "eval_runtime": 3.2155,
-      "eval_samples_per_second": 11.196,
-      "eval_steps_per_second": 0.622,
-      "step": 480
-    },
-    {
-      "epoch": 43.73,
-      "learning_rate": 9.605741626794259e-05,
-      "loss": 0.157,
-      "step": 481
-    },
-    {
-      "epoch": 43.82,
-      "learning_rate": 9.603827751196172e-05,
-      "loss": 0.1361,
-      "step": 482
-    },
-    {
-      "epoch": 43.91,
-      "learning_rate": 9.601913875598087e-05,
-      "loss": 0.1401,
-      "step": 483
-    },
-    {
-      "epoch": 44.0,
-      "learning_rate": 9.6e-05,
-      "loss": 0.1382,
-      "step": 484
-    },
-    {
-      "epoch": 44.09,
-      "learning_rate": 9.598086124401913e-05,
-      "loss": 0.1597,
-      "step": 485
-    },
-    {
-      "epoch": 44.18,
-      "learning_rate": 9.596172248803828e-05,
-      "loss": 0.1453,
-      "step": 486
-    },
-    {
-      "epoch": 44.27,
-      "learning_rate": 9.594258373205743e-05,
-      "loss": 0.1354,
-      "step": 487
-    },
-    {
-      "epoch": 44.36,
-      "learning_rate": 9.592344497607656e-05,
-      "loss": 0.1368,
-      "step": 488
-    },
-    {
-      "epoch": 44.45,
-      "learning_rate": 9.59043062200957e-05,
-      "loss": 0.1376,
-      "step": 489
-    },
-    {
-      "epoch": 44.55,
-      "learning_rate": 9.588516746411484e-05,
-      "loss": 0.1416,
-      "step": 490
-    },
-    {
-      "epoch": 44.64,
-      "learning_rate": 9.586602870813398e-05,
-      "loss": 0.1415,
-      "step": 491
-    },
-    {
-      "epoch": 44.73,
-      "learning_rate": 9.584688995215311e-05,
-      "loss": 0.14,
-      "step": 492
-    },
-    {
-      "epoch": 44.82,
-      "learning_rate": 9.582775119617226e-05,
-      "loss": 0.1213,
-      "step": 493
-    },
-    {
-      "epoch": 44.91,
-      "learning_rate": 9.580861244019139e-05,
-      "loss": 0.1313,
-      "step": 494
-    },
-    {
-      "epoch": 45.0,
-      "learning_rate": 9.578947368421052e-05,
-      "loss": 0.1434,
-      "step": 495
-    },
-    {
-      "epoch": 45.09,
-      "learning_rate": 9.577033492822967e-05,
-      "loss": 0.1125,
-      "step": 496
-    },
-    {
-      "epoch": 45.18,
-      "learning_rate": 9.575119617224881e-05,
-      "loss": 0.1427,
-      "step": 497
-    },
-    {
-      "epoch": 45.27,
-      "learning_rate": 9.573205741626795e-05,
-      "loss": 0.1253,
-      "step": 498
-    },
-    {
-      "epoch": 45.36,
-      "learning_rate": 9.571291866028709e-05,
-      "loss": 0.1608,
-      "step": 499
-    },
-    {
-      "epoch": 45.45,
-      "learning_rate": 9.569377990430622e-05,
-      "loss": 0.1309,
-      "step": 500
-    },
-    {
-      "epoch": 45.45,
-      "eval_accuracy_NAT": 0.9897674233125803,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9317258787930812,
-      "eval_accuracy_concrete tiles": 0.9816547422491286,
-      "eval_accuracy_dark bricks": 0.669835649128209,
-      "eval_accuracy_grass": 0.8455736639207038,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9641118905133939,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.3113334987456235,
-      "eval_accuracy_speedway bricks": 0.9772750586210683,
-      "eval_accuracy_steel": 0.8290927379460458,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9705295962768544,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8297568894075545,
-      "eval_iou_concrete tiles": 0.8985726280436608,
-      "eval_iou_dark bricks": 0.6453682414892828,
-      "eval_iou_grass": 0.7728427085096936,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9514574229163733,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.21027837259100643,
-      "eval_iou_speedway bricks": 0.9568115517357497,
-      "eval_iou_steel": 0.7419083524542389,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1379835158586502,
-      "eval_mean_accuracy": 0.5769515802484488,
-      "eval_mean_iou": 0.5367327510326473,
-      "eval_overall_accuracy": 0.9668308893839518,
-      "eval_runtime": 3.407,
-      "eval_samples_per_second": 10.567,
-      "eval_steps_per_second": 0.587,
-      "step": 500
-    },
-    {
-      "epoch": 45.55,
-      "learning_rate": 9.567464114832535e-05,
-      "loss": 0.1226,
-      "step": 501
-    },
-    {
-      "epoch": 45.64,
-      "learning_rate": 9.56555023923445e-05,
-      "loss": 0.1349,
-      "step": 502
-    },
-    {
-      "epoch": 45.73,
-      "learning_rate": 9.563636363636365e-05,
-      "loss": 0.121,
-      "step": 503
-    },
-    {
-      "epoch": 45.82,
-      "learning_rate": 9.561722488038278e-05,
-      "loss": 0.1623,
-      "step": 504
-    },
-    {
-      "epoch": 45.91,
-      "learning_rate": 9.559808612440192e-05,
-      "loss": 0.1316,
-      "step": 505
-    },
-    {
-      "epoch": 46.0,
-      "learning_rate": 9.557894736842106e-05,
-      "loss": 0.1296,
-      "step": 506
-    },
-    {
-      "epoch": 46.09,
-      "learning_rate": 9.555980861244019e-05,
-      "loss": 0.1205,
-      "step": 507
-    },
-    {
-      "epoch": 46.18,
-      "learning_rate": 9.554066985645933e-05,
-      "loss": 0.1351,
-      "step": 508
-    },
-    {
-      "epoch": 46.27,
-      "learning_rate": 9.552153110047848e-05,
-      "loss": 0.1459,
-      "step": 509
-    },
-    {
-      "epoch": 46.36,
-      "learning_rate": 9.550239234449761e-05,
-      "loss": 0.1367,
-      "step": 510
-    },
-    {
-      "epoch": 46.45,
-      "learning_rate": 9.548325358851676e-05,
-      "loss": 0.1382,
-      "step": 511
-    },
-    {
-      "epoch": 46.55,
-      "learning_rate": 9.546411483253589e-05,
-      "loss": 0.1301,
-      "step": 512
-    },
-    {
-      "epoch": 46.64,
-      "learning_rate": 9.544497607655502e-05,
-      "loss": 0.1182,
-      "step": 513
-    },
-    {
-      "epoch": 46.73,
-      "learning_rate": 9.542583732057417e-05,
-      "loss": 0.1268,
-      "step": 514
-    },
-    {
-      "epoch": 46.82,
-      "learning_rate": 9.540669856459331e-05,
-      "loss": 0.1367,
-      "step": 515
-    },
-    {
-      "epoch": 46.91,
-      "learning_rate": 9.538755980861244e-05,
-      "loss": 0.1179,
-      "step": 516
-    },
-    {
-      "epoch": 47.0,
-      "learning_rate": 9.536842105263159e-05,
-      "loss": 0.1309,
-      "step": 517
-    },
-    {
-      "epoch": 47.09,
-      "learning_rate": 9.534928229665072e-05,
-      "loss": 0.1273,
-      "step": 518
-    },
-    {
-      "epoch": 47.18,
-      "learning_rate": 9.533014354066985e-05,
-      "loss": 0.146,
-      "step": 519
-    },
-    {
-      "epoch": 47.27,
-      "learning_rate": 9.5311004784689e-05,
-      "loss": 0.1186,
-      "step": 520
-    },
-    {
-      "epoch": 47.27,
-      "eval_accuracy_NAT": 0.9896784919499472,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9158735353448646,
-      "eval_accuracy_concrete tiles": 0.9876781018773314,
-      "eval_accuracy_dark bricks": 0.7515876987028983,
-      "eval_accuracy_grass": 0.8389120534606472,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9626995345755133,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.28307556584787585,
-      "eval_accuracy_speedway bricks": 0.9887579091678161,
-      "eval_accuracy_steel": 0.7942665391464943,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9706582024192398,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.838931773198992,
-      "eval_iou_concrete tiles": 0.8375596349305123,
-      "eval_iou_dark bricks": 0.7111847616272717,
-      "eval_iou_grass": 0.7714933144767292,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9498607076172353,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.19217667976792063,
-      "eval_iou_speedway bricks": 0.9532614102090502,
-      "eval_iou_steel": 0.7309846116490121,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13483025133609772,
-      "eval_mean_accuracy": 0.5778868792364145,
-      "eval_mean_iou": 0.5350854689150741,
-      "eval_overall_accuracy": 0.9669055938720703,
-      "eval_runtime": 3.61,
-      "eval_samples_per_second": 9.972,
-      "eval_steps_per_second": 0.554,
-      "step": 520
-    },
-    {
-      "epoch": 47.36,
-      "learning_rate": 9.529186602870814e-05,
-      "loss": 0.1128,
-      "step": 521
-    },
-    {
-      "epoch": 47.45,
-      "learning_rate": 9.527272727272728e-05,
-      "loss": 0.1088,
-      "step": 522
-    },
-    {
-      "epoch": 47.55,
-      "learning_rate": 9.525358851674642e-05,
-      "loss": 0.1156,
-      "step": 523
-    },
-    {
-      "epoch": 47.64,
-      "learning_rate": 9.523444976076555e-05,
-      "loss": 0.1075,
-      "step": 524
-    },
-    {
-      "epoch": 47.73,
-      "learning_rate": 9.521531100478469e-05,
-      "loss": 0.1408,
-      "step": 525
-    },
-    {
-      "epoch": 47.82,
-      "learning_rate": 9.519617224880383e-05,
-      "loss": 0.1204,
-      "step": 526
-    },
-    {
-      "epoch": 47.91,
-      "learning_rate": 9.517703349282298e-05,
-      "loss": 0.1224,
-      "step": 527
-    },
-    {
-      "epoch": 48.0,
-      "learning_rate": 9.515789473684211e-05,
-      "loss": 0.1187,
-      "step": 528
-    },
-    {
-      "epoch": 48.09,
-      "learning_rate": 9.513875598086125e-05,
-      "loss": 0.1416,
-      "step": 529
-    },
-    {
-      "epoch": 48.18,
-      "learning_rate": 9.511961722488039e-05,
-      "loss": 0.1307,
-      "step": 530
-    },
-    {
-      "epoch": 48.27,
-      "learning_rate": 9.510047846889952e-05,
-      "loss": 0.1041,
-      "step": 531
-    },
-    {
-      "epoch": 48.36,
-      "learning_rate": 9.508133971291866e-05,
-      "loss": 0.1448,
-      "step": 532
-    },
-    {
-      "epoch": 48.45,
-      "learning_rate": 9.506220095693781e-05,
-      "loss": 0.1072,
-      "step": 533
-    },
-    {
-      "epoch": 48.55,
-      "learning_rate": 9.504306220095694e-05,
-      "loss": 0.1211,
-      "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.1189,
-      "step": 536
-    },
-    {
-      "epoch": 48.82,
-      "learning_rate": 9.498564593301435e-05,
-      "loss": 0.1024,
-      "step": 537
-    },
-    {
-      "epoch": 48.91,
-      "learning_rate": 9.49665071770335e-05,
-      "loss": 0.1227,
-      "step": 538
-    },
-    {
-      "epoch": 49.0,
-      "learning_rate": 9.494736842105264e-05,
-      "loss": 0.1197,
-      "step": 539
-    },
-    {
-      "epoch": 49.09,
-      "learning_rate": 9.492822966507177e-05,
-      "loss": 0.1157,
-      "step": 540
-    },
-    {
-      "epoch": 49.09,
-      "eval_accuracy_NAT": 0.9880929022185552,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9352560732220268,
-      "eval_accuracy_concrete tiles": 0.9901547116736991,
-      "eval_accuracy_dark bricks": 0.7314576036334244,
-      "eval_accuracy_grass": 0.8557792791061,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9594999679559655,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.34127312325972486,
-      "eval_accuracy_speedway bricks": 0.9764791637250536,
-      "eval_accuracy_steel": 0.8611156915770728,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9709480795732678,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8278919948329249,
-      "eval_iou_concrete tiles": 0.8843255051884217,
-      "eval_iou_dark bricks": 0.6998600574500994,
-      "eval_iou_grass": 0.767205913225484,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9454584882796815,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.21930306305029496,
-      "eval_iou_speedway bricks": 0.9589203602749466,
-      "eval_iou_steel": 0.7521893814997264,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13406065106391907,
-      "eval_mean_accuracy": 0.5876237320285863,
-      "eval_mean_iou": 0.5404694494903728,
-      "eval_overall_accuracy": 0.9665954377916124,
-      "eval_runtime": 3.8919,
-      "eval_samples_per_second": 9.25,
-      "eval_steps_per_second": 0.514,
-      "step": 540
-    },
-    {
-      "epoch": 49.18,
-      "learning_rate": 9.490909090909092e-05,
-      "loss": 0.1128,
-      "step": 541
-    },
-    {
-      "epoch": 49.27,
-      "learning_rate": 9.488995215311005e-05,
-      "loss": 0.1172,
-      "step": 542
-    },
-    {
-      "epoch": 49.36,
-      "learning_rate": 9.487081339712918e-05,
-      "loss": 0.1164,
-      "step": 543
-    },
-    {
-      "epoch": 49.45,
-      "learning_rate": 9.485167464114833e-05,
-      "loss": 0.11,
-      "step": 544
-    },
-    {
-      "epoch": 49.55,
-      "learning_rate": 9.483253588516747e-05,
-      "loss": 0.1063,
-      "step": 545
-    },
-    {
-      "epoch": 49.64,
-      "learning_rate": 9.48133971291866e-05,
-      "loss": 0.1057,
-      "step": 546
-    },
-    {
-      "epoch": 49.73,
-      "learning_rate": 9.479425837320575e-05,
-      "loss": 0.1367,
-      "step": 547
-    },
-    {
-      "epoch": 49.82,
-      "learning_rate": 9.477511961722488e-05,
-      "loss": 0.115,
-      "step": 548
-    },
-    {
-      "epoch": 49.91,
-      "learning_rate": 9.475598086124402e-05,
-      "loss": 0.1122,
-      "step": 549
-    },
-    {
-      "epoch": 50.0,
-      "learning_rate": 9.473684210526316e-05,
-      "loss": 0.1385,
-      "step": 550
-    },
-    {
-      "epoch": 50.09,
-      "learning_rate": 9.471770334928231e-05,
-      "loss": 0.1091,
-      "step": 551
-    },
-    {
-      "epoch": 50.18,
-      "learning_rate": 9.469856459330144e-05,
-      "loss": 0.1148,
-      "step": 552
-    },
-    {
-      "epoch": 50.27,
-      "learning_rate": 9.467942583732058e-05,
-      "loss": 0.112,
-      "step": 553
-    },
-    {
-      "epoch": 50.36,
-      "learning_rate": 9.466028708133972e-05,
-      "loss": 0.1073,
-      "step": 554
-    },
-    {
-      "epoch": 50.45,
-      "learning_rate": 9.464114832535885e-05,
-      "loss": 0.104,
-      "step": 555
-    },
-    {
-      "epoch": 50.55,
-      "learning_rate": 9.4622009569378e-05,
-      "loss": 0.0947,
-      "step": 556
-    },
-    {
-      "epoch": 50.64,
-      "learning_rate": 9.460287081339714e-05,
-      "loss": 0.1166,
-      "step": 557
-    },
-    {
-      "epoch": 50.73,
-      "learning_rate": 9.458373205741627e-05,
-      "loss": 0.1145,
-      "step": 558
-    },
-    {
-      "epoch": 50.82,
-      "learning_rate": 9.456459330143542e-05,
-      "loss": 0.122,
-      "step": 559
-    },
-    {
-      "epoch": 50.91,
-      "learning_rate": 9.454545454545455e-05,
-      "loss": 0.127,
-      "step": 560
-    },
-    {
-      "epoch": 50.91,
-      "eval_accuracy_NAT": 0.9905360443753384,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9207302352032276,
-      "eval_accuracy_concrete tiles": 0.9757231089096802,
-      "eval_accuracy_dark bricks": 0.7773372849389939,
-      "eval_accuracy_grass": 0.8448536787398162,
-      "eval_accuracy_red bricks": 0.012428977272727272,
-      "eval_accuracy_road": 0.9744554630242072,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.3134562898023323,
-      "eval_accuracy_speedway bricks": 0.9880845060564042,
-      "eval_accuracy_steel": 0.7858815381571137,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9721527207738906,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8493891899048366,
-      "eval_iou_concrete tiles": 0.9197867127828218,
-      "eval_iou_dark bricks": 0.7353894330553836,
-      "eval_iou_grass": 0.782120970925304,
-      "eval_iou_red bricks": 0.012428977272727272,
-      "eval_iou_road": 0.9568637043141045,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.21134221825684493,
-      "eval_iou_speedway bricks": 0.9561169032571479,
-      "eval_iou_steel": 0.7330067596145713,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12320339679718018,
-      "eval_mean_accuracy": 0.5833451635753724,
-      "eval_mean_iou": 0.5483536607813563,
-      "eval_overall_accuracy": 0.9689422183566623,
-      "eval_runtime": 3.2366,
-      "eval_samples_per_second": 11.123,
-      "eval_steps_per_second": 0.618,
-      "step": 560
-    },
-    {
-      "epoch": 51.0,
-      "learning_rate": 9.452631578947368e-05,
-      "loss": 0.1259,
-      "step": 561
-    },
-    {
-      "epoch": 51.09,
-      "learning_rate": 9.450717703349283e-05,
-      "loss": 0.1086,
-      "step": 562
-    },
-    {
-      "epoch": 51.18,
-      "learning_rate": 9.448803827751197e-05,
-      "loss": 0.1027,
-      "step": 563
-    },
-    {
-      "epoch": 51.27,
-      "learning_rate": 9.44688995215311e-05,
-      "loss": 0.1263,
-      "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.1138,
-      "step": 566
-    },
-    {
-      "epoch": 51.55,
-      "learning_rate": 9.441148325358851e-05,
-      "loss": 0.0995,
-      "step": 567
-    },
-    {
-      "epoch": 51.64,
-      "learning_rate": 9.439234449760766e-05,
-      "loss": 0.1104,
-      "step": 568
-    },
-    {
-      "epoch": 51.73,
-      "learning_rate": 9.43732057416268e-05,
-      "loss": 0.1062,
-      "step": 569
-    },
-    {
-      "epoch": 51.82,
-      "learning_rate": 9.435406698564594e-05,
-      "loss": 0.0955,
-      "step": 570
-    },
-    {
-      "epoch": 51.91,
-      "learning_rate": 9.433492822966507e-05,
-      "loss": 0.1132,
-      "step": 571
-    },
-    {
-      "epoch": 52.0,
-      "learning_rate": 9.431578947368421e-05,
-      "loss": 0.1047,
-      "step": 572
-    },
-    {
-      "epoch": 52.09,
-      "learning_rate": 9.429665071770335e-05,
-      "loss": 0.1096,
-      "step": 573
-    },
-    {
-      "epoch": 52.18,
-      "learning_rate": 9.427751196172249e-05,
-      "loss": 0.1146,
-      "step": 574
-    },
-    {
-      "epoch": 52.27,
-      "learning_rate": 9.425837320574164e-05,
-      "loss": 0.0906,
-      "step": 575
-    },
-    {
-      "epoch": 52.36,
-      "learning_rate": 9.423923444976077e-05,
-      "loss": 0.1122,
-      "step": 576
-    },
-    {
-      "epoch": 52.45,
-      "learning_rate": 9.42200956937799e-05,
-      "loss": 0.1222,
-      "step": 577
-    },
-    {
-      "epoch": 52.55,
-      "learning_rate": 9.420095693779905e-05,
-      "loss": 0.1114,
-      "step": 578
-    },
-    {
-      "epoch": 52.64,
-      "learning_rate": 9.418181818181818e-05,
-      "loss": 0.0979,
-      "step": 579
-    },
-    {
-      "epoch": 52.73,
-      "learning_rate": 9.416267942583733e-05,
-      "loss": 0.114,
-      "step": 580
-    },
-    {
-      "epoch": 52.73,
-      "eval_accuracy_NAT": 0.990019571957824,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9227877269410705,
-      "eval_accuracy_concrete tiles": 0.9834281171650462,
-      "eval_accuracy_dark bricks": 0.7965436280358724,
-      "eval_accuracy_grass": 0.85826427650227,
-      "eval_accuracy_red bricks": 0.07066761363636363,
-      "eval_accuracy_road": 0.9742402238487425,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5022468502743087,
-      "eval_accuracy_speedway bricks": 0.9841448564915215,
-      "eval_accuracy_steel": 0.8427461908845063,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9720439459156511,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8596154975454765,
-      "eval_iou_concrete tiles": 0.9143994314143568,
-      "eval_iou_dark bricks": 0.7516434823666146,
-      "eval_iou_grass": 0.7847726669883769,
-      "eval_iou_red bricks": 0.07066761363636363,
-      "eval_iou_road": 0.955756957460438,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3258276250603616,
-      "eval_iou_speedway bricks": 0.9593209492405911,
-      "eval_iou_steel": 0.7577150312455986,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11953094601631165,
-      "eval_mean_accuracy": 0.6096222350567327,
-      "eval_mean_iou": 0.5655202462210638,
-      "eval_overall_accuracy": 0.9700613021850586,
-      "eval_runtime": 3.1995,
-      "eval_samples_per_second": 11.252,
-      "eval_steps_per_second": 0.625,
-      "step": 580
-    },
-    {
-      "epoch": 52.82,
-      "learning_rate": 9.414354066985647e-05,
-      "loss": 0.1206,
-      "step": 581
-    },
-    {
-      "epoch": 52.91,
-      "learning_rate": 9.41244019138756e-05,
-      "loss": 0.093,
-      "step": 582
-    },
-    {
-      "epoch": 53.0,
-      "learning_rate": 9.410526315789473e-05,
-      "loss": 0.1116,
-      "step": 583
-    },
-    {
-      "epoch": 53.09,
-      "learning_rate": 9.408612440191388e-05,
-      "loss": 0.0964,
-      "step": 584
-    },
-    {
-      "epoch": 53.18,
-      "learning_rate": 9.406698564593301e-05,
-      "loss": 0.108,
-      "step": 585
-    },
-    {
-      "epoch": 53.27,
-      "learning_rate": 9.404784688995216e-05,
-      "loss": 0.1034,
-      "step": 586
-    },
-    {
-      "epoch": 53.36,
-      "learning_rate": 9.40287081339713e-05,
-      "loss": 0.0919,
-      "step": 587
-    },
-    {
-      "epoch": 53.45,
-      "learning_rate": 9.400956937799044e-05,
-      "loss": 0.1046,
-      "step": 588
-    },
-    {
-      "epoch": 53.55,
-      "learning_rate": 9.399043062200957e-05,
-      "loss": 0.1011,
-      "step": 589
-    },
-    {
-      "epoch": 53.64,
-      "learning_rate": 9.397129186602871e-05,
-      "loss": 0.1041,
-      "step": 590
-    },
-    {
-      "epoch": 53.73,
-      "learning_rate": 9.395215311004784e-05,
-      "loss": 0.1085,
-      "step": 591
-    },
-    {
-      "epoch": 53.82,
-      "learning_rate": 9.393301435406699e-05,
-      "loss": 0.1386,
-      "step": 592
-    },
-    {
-      "epoch": 53.91,
-      "learning_rate": 9.391387559808614e-05,
-      "loss": 0.0914,
-      "step": 593
-    },
-    {
-      "epoch": 54.0,
-      "learning_rate": 9.389473684210527e-05,
-      "loss": 0.0961,
-      "step": 594
-    },
-    {
-      "epoch": 54.09,
-      "learning_rate": 9.38755980861244e-05,
-      "loss": 0.128,
-      "step": 595
-    },
-    {
-      "epoch": 54.18,
-      "learning_rate": 9.385645933014355e-05,
-      "loss": 0.0961,
-      "step": 596
-    },
-    {
-      "epoch": 54.27,
-      "learning_rate": 9.383732057416268e-05,
-      "loss": 0.1044,
-      "step": 597
-    },
-    {
-      "epoch": 54.36,
-      "learning_rate": 9.381818181818182e-05,
-      "loss": 0.0977,
-      "step": 598
-    },
-    {
-      "epoch": 54.45,
-      "learning_rate": 9.379904306220097e-05,
-      "loss": 0.1233,
-      "step": 599
-    },
-    {
-      "epoch": 54.55,
-      "learning_rate": 9.37799043062201e-05,
-      "loss": 0.0963,
-      "step": 600
-    },
-    {
-      "epoch": 54.55,
-      "eval_accuracy_NAT": 0.9901933763074146,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9337109425297223,
-      "eval_accuracy_concrete tiles": 0.990582767687886,
-      "eval_accuracy_dark bricks": 0.785266156037104,
-      "eval_accuracy_grass": 0.8520395502539171,
-      "eval_accuracy_red bricks": 0.32954545454545453,
-      "eval_accuracy_road": 0.9295635844336126,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.39770628291015353,
-      "eval_accuracy_speedway bricks": 0.9823473046376717,
-      "eval_accuracy_steel": 0.8614125057713871,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9731258312827636,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8200186586064,
-      "eval_iou_concrete tiles": 0.8685790884718498,
-      "eval_iou_dark bricks": 0.7394440216012468,
-      "eval_iou_grass": 0.7886507890292902,
-      "eval_iou_red bricks": 0.327683615819209,
-      "eval_iou_road": 0.9177020709677728,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2600731940363086,
-      "eval_iou_speedway bricks": 0.9617787245908302,
-      "eval_iou_steel": 0.7622699050801456,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12740296125411987,
-      "eval_mean_accuracy": 0.6194129173164864,
-      "eval_mean_iou": 0.5707173768835244,
-      "eval_overall_accuracy": 0.9665955437554253,
-      "eval_runtime": 3.7868,
-      "eval_samples_per_second": 9.507,
-      "eval_steps_per_second": 0.528,
-      "step": 600
-    },
-    {
-      "epoch": 54.64,
-      "learning_rate": 9.376076555023923e-05,
-      "loss": 0.1073,
-      "step": 601
-    },
-    {
-      "epoch": 54.73,
-      "learning_rate": 9.374162679425838e-05,
-      "loss": 0.0898,
-      "step": 602
-    },
-    {
-      "epoch": 54.82,
-      "learning_rate": 9.372248803827751e-05,
-      "loss": 0.0938,
-      "step": 603
-    },
-    {
-      "epoch": 54.91,
-      "learning_rate": 9.370334928229666e-05,
-      "loss": 0.096,
-      "step": 604
-    },
-    {
-      "epoch": 55.0,
-      "learning_rate": 9.36842105263158e-05,
-      "loss": 0.0991,
-      "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.1,
-      "step": 607
-    },
-    {
-      "epoch": 55.27,
-      "learning_rate": 9.362679425837321e-05,
-      "loss": 0.0915,
-      "step": 608
-    },
-    {
-      "epoch": 55.36,
-      "learning_rate": 9.360765550239234e-05,
-      "loss": 0.1202,
-      "step": 609
-    },
-    {
-      "epoch": 55.45,
-      "learning_rate": 9.358851674641149e-05,
-      "loss": 0.1249,
-      "step": 610
-    },
-    {
-      "epoch": 55.55,
-      "learning_rate": 9.356937799043063e-05,
-      "loss": 0.0817,
-      "step": 611
-    },
-    {
-      "epoch": 55.64,
-      "learning_rate": 9.355023923444977e-05,
-      "loss": 0.1032,
-      "step": 612
-    },
-    {
-      "epoch": 55.73,
-      "learning_rate": 9.35311004784689e-05,
-      "loss": 0.0918,
-      "step": 613
-    },
-    {
-      "epoch": 55.82,
-      "learning_rate": 9.351196172248804e-05,
-      "loss": 0.0941,
-      "step": 614
-    },
-    {
-      "epoch": 55.91,
-      "learning_rate": 9.349282296650718e-05,
-      "loss": 0.0974,
-      "step": 615
-    },
-    {
-      "epoch": 56.0,
-      "learning_rate": 9.347368421052632e-05,
-      "loss": 0.1097,
-      "step": 616
-    },
-    {
-      "epoch": 56.09,
-      "learning_rate": 9.345454545454547e-05,
-      "loss": 0.0873,
-      "step": 617
-    },
-    {
-      "epoch": 56.18,
-      "learning_rate": 9.34354066985646e-05,
-      "loss": 0.0933,
-      "step": 618
-    },
-    {
-      "epoch": 56.27,
-      "learning_rate": 9.341626794258373e-05,
-      "loss": 0.1118,
-      "step": 619
-    },
-    {
-      "epoch": 56.36,
-      "learning_rate": 9.339712918660288e-05,
-      "loss": 0.0854,
-      "step": 620
-    },
-    {
-      "epoch": 56.36,
-      "eval_accuracy_NAT": 0.9911206269237582,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.928115208807245,
-      "eval_accuracy_concrete tiles": 0.9813795633828656,
-      "eval_accuracy_dark bricks": 0.7771833262768947,
-      "eval_accuracy_grass": 0.8529063285299371,
-      "eval_accuracy_red bricks": 0.4177911931818182,
-      "eval_accuracy_road": 0.9738508585987447,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.38196454663248147,
-      "eval_accuracy_speedway bricks": 0.9828608046805136,
-      "eval_accuracy_steel": 0.8564408680166216,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9731393396754336,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8537964578403042,
-      "eval_iou_concrete tiles": 0.9142360715506437,
-      "eval_iou_dark bricks": 0.7443506469569064,
-      "eval_iou_grass": 0.7871612249769365,
-      "eval_iou_red bricks": 0.414479478597851,
-      "eval_iou_road": 0.9590828382284116,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2555895808736718,
-      "eval_iou_speedway bricks": 0.9621888178519298,
-      "eval_iou_steel": 0.7664258888979067,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11508533358573914,
-      "eval_mean_accuracy": 0.6264317942331447,
-      "eval_mean_iou": 0.5869577188807689,
-      "eval_overall_accuracy": 0.9705817964341905,
-      "eval_runtime": 3.4783,
-      "eval_samples_per_second": 10.35,
-      "eval_steps_per_second": 0.575,
-      "step": 620
-    },
-    {
-      "epoch": 56.45,
-      "learning_rate": 9.337799043062201e-05,
-      "loss": 0.0836,
-      "step": 621
-    },
-    {
-      "epoch": 56.55,
-      "learning_rate": 9.335885167464115e-05,
-      "loss": 0.1111,
-      "step": 622
-    },
-    {
-      "epoch": 56.64,
-      "learning_rate": 9.33397129186603e-05,
-      "loss": 0.1059,
-      "step": 623
-    },
-    {
-      "epoch": 56.73,
-      "learning_rate": 9.332057416267943e-05,
-      "loss": 0.098,
-      "step": 624
-    },
-    {
-      "epoch": 56.82,
-      "learning_rate": 9.330143540669856e-05,
-      "loss": 0.0988,
-      "step": 625
-    },
-    {
-      "epoch": 56.91,
-      "learning_rate": 9.328229665071771e-05,
-      "loss": 0.094,
-      "step": 626
-    },
-    {
-      "epoch": 57.0,
-      "learning_rate": 9.326315789473684e-05,
-      "loss": 0.1058,
-      "step": 627
-    },
-    {
-      "epoch": 57.09,
-      "learning_rate": 9.324401913875599e-05,
-      "loss": 0.0867,
-      "step": 628
-    },
-    {
-      "epoch": 57.18,
-      "learning_rate": 9.322488038277513e-05,
-      "loss": 0.1019,
-      "step": 629
-    },
-    {
-      "epoch": 57.27,
-      "learning_rate": 9.320574162679426e-05,
-      "loss": 0.0864,
-      "step": 630
-    },
-    {
-      "epoch": 57.36,
-      "learning_rate": 9.31866028708134e-05,
-      "loss": 0.0943,
-      "step": 631
-    },
-    {
-      "epoch": 57.45,
-      "learning_rate": 9.316746411483254e-05,
-      "loss": 0.0971,
-      "step": 632
-    },
-    {
-      "epoch": 57.55,
-      "learning_rate": 9.314832535885169e-05,
-      "loss": 0.0836,
-      "step": 633
-    },
-    {
-      "epoch": 57.64,
-      "learning_rate": 9.312918660287082e-05,
-      "loss": 0.105,
-      "step": 634
-    },
-    {
-      "epoch": 57.73,
-      "learning_rate": 9.311004784688996e-05,
-      "loss": 0.0929,
-      "step": 635
-    },
-    {
-      "epoch": 57.82,
-      "learning_rate": 9.30909090909091e-05,
-      "loss": 0.1089,
-      "step": 636
-    },
-    {
-      "epoch": 57.91,
-      "learning_rate": 9.307177033492823e-05,
-      "loss": 0.1117,
-      "step": 637
-    },
-    {
-      "epoch": 58.0,
-      "learning_rate": 9.305263157894737e-05,
-      "loss": 0.0931,
-      "step": 638
-    },
-    {
-      "epoch": 58.09,
-      "learning_rate": 9.303349282296652e-05,
-      "loss": 0.0907,
-      "step": 639
-    },
-    {
-      "epoch": 58.18,
-      "learning_rate": 9.301435406698565e-05,
-      "loss": 0.0803,
-      "step": 640
-    },
-    {
-      "epoch": 58.18,
-      "eval_accuracy_NAT": 0.9885320890471145,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.930003701875617,
-      "eval_accuracy_concrete tiles": 0.9932122546321775,
-      "eval_accuracy_dark bricks": 0.788460798275663,
-      "eval_accuracy_grass": 0.8814121495751738,
-      "eval_accuracy_red bricks": 0.6344105113636364,
-      "eval_accuracy_road": 0.9720152795630402,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.483555261489262,
-      "eval_accuracy_speedway bricks": 0.9849377343485066,
-      "eval_accuracy_steel": 0.855294835433019,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9731420574095275,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8548217880738367,
-      "eval_iou_concrete tiles": 0.8437183449780525,
-      "eval_iou_dark bricks": 0.7499816943691879,
-      "eval_iou_grass": 0.7902452017234626,
-      "eval_iou_red bricks": 0.6293817156949093,
-      "eval_iou_road": 0.9597908109657083,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3173511850913696,
-      "eval_iou_speedway bricks": 0.9632592940548853,
-      "eval_iou_steel": 0.7716172893685705,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10878638923168182,
-      "eval_mean_accuracy": 0.6547565088925545,
-      "eval_mean_iou": 0.6041007216715008,
-      "eval_overall_accuracy": 0.9708216985066732,
-      "eval_runtime": 3.7605,
-      "eval_samples_per_second": 9.573,
-      "eval_steps_per_second": 0.532,
-      "step": 640
-    },
-    {
-      "epoch": 58.27,
-      "learning_rate": 9.29952153110048e-05,
-      "loss": 0.1023,
-      "step": 641
-    },
-    {
-      "epoch": 58.36,
-      "learning_rate": 9.297607655502393e-05,
-      "loss": 0.0855,
-      "step": 642
-    },
-    {
-      "epoch": 58.45,
-      "learning_rate": 9.295693779904306e-05,
-      "loss": 0.111,
-      "step": 643
-    },
-    {
-      "epoch": 58.55,
-      "learning_rate": 9.29377990430622e-05,
-      "loss": 0.076,
-      "step": 644
-    },
-    {
-      "epoch": 58.64,
-      "learning_rate": 9.291866028708135e-05,
-      "loss": 0.0935,
-      "step": 645
-    },
-    {
-      "epoch": 58.73,
-      "learning_rate": 9.289952153110048e-05,
-      "loss": 0.0968,
-      "step": 646
-    },
-    {
-      "epoch": 58.82,
-      "learning_rate": 9.288038277511963e-05,
-      "loss": 0.0967,
-      "step": 647
-    },
-    {
-      "epoch": 58.91,
-      "learning_rate": 9.286124401913876e-05,
-      "loss": 0.0924,
-      "step": 648
-    },
-    {
-      "epoch": 59.0,
-      "learning_rate": 9.28421052631579e-05,
-      "loss": 0.0919,
-      "step": 649
-    },
-    {
-      "epoch": 59.09,
-      "learning_rate": 9.282296650717704e-05,
-      "loss": 0.1093,
-      "step": 650
-    },
-    {
-      "epoch": 59.18,
-      "learning_rate": 9.280382775119618e-05,
-      "loss": 0.09,
-      "step": 651
-    },
-    {
-      "epoch": 59.27,
-      "learning_rate": 9.278468899521532e-05,
-      "loss": 0.0935,
-      "step": 652
-    },
-    {
-      "epoch": 59.36,
-      "learning_rate": 9.276555023923445e-05,
-      "loss": 0.0917,
-      "step": 653
-    },
-    {
-      "epoch": 59.45,
-      "learning_rate": 9.27464114832536e-05,
-      "loss": 0.0834,
-      "step": 654
-    },
-    {
-      "epoch": 59.55,
-      "learning_rate": 9.272727272727273e-05,
-      "loss": 0.0881,
-      "step": 655
-    },
-    {
-      "epoch": 59.64,
-      "learning_rate": 9.270813397129187e-05,
-      "loss": 0.0991,
-      "step": 656
-    },
-    {
-      "epoch": 59.73,
-      "learning_rate": 9.268899521531102e-05,
-      "loss": 0.115,
-      "step": 657
-    },
-    {
-      "epoch": 59.82,
-      "learning_rate": 9.266985645933015e-05,
-      "loss": 0.1007,
-      "step": 658
-    },
-    {
-      "epoch": 59.91,
-      "learning_rate": 9.265071770334928e-05,
-      "loss": 0.0982,
-      "step": 659
-    },
-    {
-      "epoch": 60.0,
-      "learning_rate": 9.263157894736843e-05,
-      "loss": 0.0862,
-      "step": 660
-    },
-    {
-      "epoch": 60.0,
-      "eval_accuracy_NAT": 0.9898018312802658,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.940473571827117,
-      "eval_accuracy_concrete tiles": 0.9419678346480768,
-      "eval_accuracy_dark bricks": 0.7965051383703475,
-      "eval_accuracy_grass": 0.7932663910218547,
-      "eval_accuracy_red bricks": 0.5671164772727273,
-      "eval_accuracy_road": 0.9696899709427114,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.3429272461610564,
-      "eval_accuracy_speedway bricks": 0.9828197729497108,
-      "eval_accuracy_steel": 0.8620391135149397,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9732964870209403,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8298237952716209,
-      "eval_iou_concrete tiles": 0.9168229027170193,
-      "eval_iou_dark bricks": 0.7531663997670694,
-      "eval_iou_grass": 0.745598848917082,
-      "eval_iou_red bricks": 0.5623239436619718,
-      "eval_iou_road": 0.9525812505939371,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.22686071747733946,
-      "eval_iou_speedway bricks": 0.9633653334879025,
-      "eval_iou_steel": 0.7727813624840166,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11864052712917328,
-      "eval_mean_accuracy": 0.6297390267683698,
-      "eval_mean_iou": 0.5920477724152999,
-      "eval_overall_accuracy": 0.9685138066609701,
-      "eval_runtime": 3.1688,
-      "eval_samples_per_second": 11.361,
-      "eval_steps_per_second": 0.631,
-      "step": 660
-    },
-    {
-      "epoch": 60.09,
-      "learning_rate": 9.261244019138756e-05,
-      "loss": 0.0844,
-      "step": 661
-    },
-    {
-      "epoch": 60.18,
-      "learning_rate": 9.25933014354067e-05,
-      "loss": 0.0822,
-      "step": 662
-    },
-    {
-      "epoch": 60.27,
-      "learning_rate": 9.257416267942585e-05,
-      "loss": 0.0895,
-      "step": 663
-    },
-    {
-      "epoch": 60.36,
-      "learning_rate": 9.255502392344498e-05,
-      "loss": 0.0966,
-      "step": 664
-    },
-    {
-      "epoch": 60.45,
-      "learning_rate": 9.253588516746411e-05,
-      "loss": 0.0833,
-      "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.1081,
-      "step": 667
-    },
-    {
-      "epoch": 60.73,
-      "learning_rate": 9.247846889952154e-05,
-      "loss": 0.0777,
-      "step": 668
-    },
-    {
-      "epoch": 60.82,
-      "learning_rate": 9.245933014354068e-05,
-      "loss": 0.0902,
-      "step": 669
-    },
-    {
-      "epoch": 60.91,
-      "learning_rate": 9.244019138755981e-05,
-      "loss": 0.0918,
-      "step": 670
-    },
-    {
-      "epoch": 61.0,
-      "learning_rate": 9.242105263157895e-05,
-      "loss": 0.1051,
-      "step": 671
-    },
-    {
-      "epoch": 61.09,
-      "learning_rate": 9.240191387559809e-05,
-      "loss": 0.096,
-      "step": 672
-    },
-    {
-      "epoch": 61.18,
-      "learning_rate": 9.238277511961722e-05,
-      "loss": 0.1032,
-      "step": 673
-    },
-    {
-      "epoch": 61.27,
-      "learning_rate": 9.236363636363636e-05,
-      "loss": 0.0816,
-      "step": 674
-    },
-    {
-      "epoch": 61.36,
-      "learning_rate": 9.234449760765552e-05,
-      "loss": 0.0988,
-      "step": 675
-    },
-    {
-      "epoch": 61.45,
-      "learning_rate": 9.232535885167465e-05,
-      "loss": 0.0964,
-      "step": 676
-    },
-    {
-      "epoch": 61.55,
-      "learning_rate": 9.230622009569378e-05,
-      "loss": 0.1047,
-      "step": 677
-    },
-    {
-      "epoch": 61.64,
-      "learning_rate": 9.228708133971293e-05,
-      "loss": 0.092,
-      "step": 678
-    },
-    {
-      "epoch": 61.73,
-      "learning_rate": 9.226794258373206e-05,
-      "loss": 0.1072,
-      "step": 679
-    },
-    {
-      "epoch": 61.82,
-      "learning_rate": 9.224880382775119e-05,
-      "loss": 0.0869,
-      "step": 680
-    },
-    {
-      "epoch": 61.82,
-      "eval_accuracy_NAT": 0.9892585382930683,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9361855659041161,
-      "eval_accuracy_concrete tiles": 0.9920198128783709,
-      "eval_accuracy_dark bricks": 0.8274893191178169,
-      "eval_accuracy_grass": 0.8739466721655826,
-      "eval_accuracy_red bricks": 0.7446732954545454,
-      "eval_accuracy_road": 0.9579304148674647,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.31599261158437403,
-      "eval_accuracy_speedway bricks": 0.9839306467204182,
-      "eval_accuracy_steel": 0.8543879031726139,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9740629424416096,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.833847028690483,
-      "eval_iou_concrete tiles": 0.8834581347855684,
-      "eval_iou_dark bricks": 0.7753813971940707,
-      "eval_iou_grass": 0.7941126965425034,
-      "eval_iou_red bricks": 0.7348869809006483,
-      "eval_iou_road": 0.947283332655731,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2109816481675778,
-      "eval_iou_speedway bricks": 0.9648691060441634,
-      "eval_iou_steel": 0.7770470905818836,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11289845407009125,
-      "eval_mean_accuracy": 0.6519857523198748,
-      "eval_mean_iou": 0.6073792583080184,
-      "eval_overall_accuracy": 0.9696222941080729,
-      "eval_runtime": 3.221,
-      "eval_samples_per_second": 11.177,
-      "eval_steps_per_second": 0.621,
-      "step": 680
-    },
-    {
-      "epoch": 61.91,
-      "learning_rate": 9.222966507177035e-05,
-      "loss": 0.0898,
-      "step": 681
-    },
-    {
-      "epoch": 62.0,
-      "learning_rate": 9.221052631578948e-05,
-      "loss": 0.0902,
-      "step": 682
-    },
-    {
-      "epoch": 62.09,
-      "learning_rate": 9.219138755980861e-05,
-      "loss": 0.0773,
-      "step": 683
-    },
-    {
-      "epoch": 62.18,
-      "learning_rate": 9.217224880382776e-05,
-      "loss": 0.0949,
-      "step": 684
-    },
-    {
-      "epoch": 62.27,
-      "learning_rate": 9.215311004784689e-05,
-      "loss": 0.0988,
-      "step": 685
-    },
-    {
-      "epoch": 62.36,
-      "learning_rate": 9.213397129186602e-05,
-      "loss": 0.0834,
-      "step": 686
-    },
-    {
-      "epoch": 62.45,
-      "learning_rate": 9.211483253588518e-05,
-      "loss": 0.1017,
-      "step": 687
-    },
-    {
-      "epoch": 62.55,
-      "learning_rate": 9.209569377990431e-05,
-      "loss": 0.0866,
-      "step": 688
-    },
-    {
-      "epoch": 62.64,
-      "learning_rate": 9.207655502392344e-05,
-      "loss": 0.1042,
-      "step": 689
-    },
-    {
-      "epoch": 62.73,
-      "learning_rate": 9.205741626794259e-05,
-      "loss": 0.0948,
-      "step": 690
-    },
-    {
-      "epoch": 62.82,
-      "learning_rate": 9.203827751196172e-05,
-      "loss": 0.0876,
-      "step": 691
-    },
-    {
-      "epoch": 62.91,
-      "learning_rate": 9.201913875598085e-05,
-      "loss": 0.0784,
-      "step": 692
-    },
-    {
-      "epoch": 63.0,
-      "learning_rate": 9.200000000000001e-05,
-      "loss": 0.0937,
-      "step": 693
-    },
-    {
-      "epoch": 63.09,
-      "learning_rate": 9.198086124401915e-05,
-      "loss": 0.0923,
-      "step": 694
-    },
-    {
-      "epoch": 63.18,
-      "learning_rate": 9.196172248803828e-05,
-      "loss": 0.0901,
-      "step": 695
-    },
-    {
-      "epoch": 63.27,
-      "learning_rate": 9.194258373205742e-05,
-      "loss": 0.095,
-      "step": 696
-    },
-    {
-      "epoch": 63.36,
-      "learning_rate": 9.192344497607656e-05,
-      "loss": 0.0894,
-      "step": 697
-    },
-    {
-      "epoch": 63.45,
-      "learning_rate": 9.190430622009569e-05,
-      "loss": 0.1109,
-      "step": 698
-    },
-    {
-      "epoch": 63.55,
-      "learning_rate": 9.188516746411485e-05,
-      "loss": 0.0963,
-      "step": 699
-    },
-    {
-      "epoch": 63.64,
-      "learning_rate": 9.186602870813398e-05,
-      "loss": 0.1056,
-      "step": 700
-    },
-    {
-      "epoch": 63.64,
-      "eval_accuracy_NAT": 0.9891419041051706,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9336720460105584,
-      "eval_accuracy_concrete tiles": 0.9671314131963554,
-      "eval_accuracy_dark bricks": 0.8278357261075401,
-      "eval_accuracy_grass": 0.8652439386684468,
-      "eval_accuracy_red bricks": 0.8052201704545454,
-      "eval_accuracy_road": 0.970147051888361,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.36398974443801174,
-      "eval_accuracy_speedway bricks": 0.987283783897942,
-      "eval_accuracy_steel": 0.8504880944528725,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9740856906039906,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8467215078862128,
-      "eval_iou_concrete tiles": 0.9119241192411924,
-      "eval_iou_dark bricks": 0.7726684868515591,
-      "eval_iou_grass": 0.7933325215028264,
-      "eval_iou_red bricks": 0.7949167397020158,
-      "eval_iou_road": 0.9599185928674239,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.23865731535374715,
-      "eval_iou_speedway bricks": 0.9625075150860573,
-      "eval_iou_steel": 0.7814460167873701,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10999203473329544,
-      "eval_mean_accuracy": 0.6584733748630618,
-      "eval_mean_iou": 0.6181675773755688,
-      "eval_overall_accuracy": 0.9708344141642252,
-      "eval_runtime": 3.6074,
-      "eval_samples_per_second": 9.98,
-      "eval_steps_per_second": 0.554,
-      "step": 700
-    },
-    {
-      "epoch": 63.73,
-      "learning_rate": 9.184688995215311e-05,
-      "loss": 0.093,
-      "step": 701
-    },
-    {
-      "epoch": 63.82,
-      "learning_rate": 9.182775119617226e-05,
-      "loss": 0.0992,
-      "step": 702
-    },
-    {
-      "epoch": 63.91,
-      "learning_rate": 9.180861244019139e-05,
-      "loss": 0.0861,
-      "step": 703
-    },
-    {
-      "epoch": 64.0,
-      "learning_rate": 9.178947368421052e-05,
-      "loss": 0.0872,
-      "step": 704
-    },
-    {
-      "epoch": 64.09,
-      "learning_rate": 9.177033492822968e-05,
-      "loss": 0.0877,
-      "step": 705
-    },
-    {
-      "epoch": 64.18,
-      "learning_rate": 9.175119617224881e-05,
-      "loss": 0.0917,
-      "step": 706
-    },
-    {
-      "epoch": 64.27,
-      "learning_rate": 9.173205741626794e-05,
-      "loss": 0.0883,
-      "step": 707
-    },
-    {
-      "epoch": 64.36,
-      "learning_rate": 9.171291866028709e-05,
-      "loss": 0.0951,
-      "step": 708
-    },
-    {
-      "epoch": 64.45,
-      "learning_rate": 9.169377990430622e-05,
-      "loss": 0.0967,
-      "step": 709
-    },
-    {
-      "epoch": 64.55,
-      "learning_rate": 9.167464114832537e-05,
-      "loss": 0.0891,
-      "step": 710
-    },
-    {
-      "epoch": 64.64,
-      "learning_rate": 9.165550239234451e-05,
-      "loss": 0.0918,
-      "step": 711
-    },
-    {
-      "epoch": 64.73,
-      "learning_rate": 9.163636363636364e-05,
-      "loss": 0.0784,
-      "step": 712
-    },
-    {
-      "epoch": 64.82,
-      "learning_rate": 9.161722488038278e-05,
-      "loss": 0.0871,
-      "step": 713
-    },
-    {
-      "epoch": 64.91,
-      "learning_rate": 9.159808612440192e-05,
-      "loss": 0.0862,
-      "step": 714
-    },
-    {
-      "epoch": 65.0,
-      "learning_rate": 9.157894736842105e-05,
-      "loss": 0.0841,
-      "step": 715
-    },
-    {
-      "epoch": 65.09,
-      "learning_rate": 9.15598086124402e-05,
-      "loss": 0.0929,
-      "step": 716
-    },
-    {
-      "epoch": 65.18,
-      "learning_rate": 9.154066985645934e-05,
-      "loss": 0.0797,
-      "step": 717
-    },
-    {
-      "epoch": 65.27,
-      "learning_rate": 9.152153110047848e-05,
-      "loss": 0.0755,
-      "step": 718
-    },
-    {
-      "epoch": 65.36,
-      "learning_rate": 9.150239234449761e-05,
-      "loss": 0.0811,
-      "step": 719
-    },
-    {
-      "epoch": 65.45,
-      "learning_rate": 9.148325358851675e-05,
-      "loss": 0.1043,
-      "step": 720
-    },
-    {
-      "epoch": 65.45,
-      "eval_accuracy_NAT": 0.9892400109258531,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9349650199579381,
-      "eval_accuracy_concrete tiles": 0.971748303063658,
-      "eval_accuracy_dark bricks": 0.8461568068973481,
-      "eval_accuracy_grass": 0.8515851906737454,
-      "eval_accuracy_red bricks": 0.8018465909090909,
-      "eval_accuracy_road": 0.9770649357245035,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.43004438563118574,
-      "eval_accuracy_speedway bricks": 0.9864878890019273,
-      "eval_accuracy_steel": 0.840437636039839,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9739380956591636,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8530040161232296,
-      "eval_iou_concrete tiles": 0.9140112734383987,
-      "eval_iou_dark bricks": 0.7931308175193016,
-      "eval_iou_grass": 0.7845854129769764,
-      "eval_iou_red bricks": 0.785527917898765,
-      "eval_iou_road": 0.9615096279478227,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2843989863078634,
-      "eval_iou_speedway bricks": 0.9637027487610799,
-      "eval_iou_steel": 0.7689725407362704,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10348805785179138,
-      "eval_mean_accuracy": 0.6638135976019299,
-      "eval_mean_iou": 0.6217524182591441,
-      "eval_overall_accuracy": 0.9712369706895616,
-      "eval_runtime": 3.3896,
-      "eval_samples_per_second": 10.621,
-      "eval_steps_per_second": 0.59,
-      "step": 720
-    },
-    {
-      "epoch": 65.55,
-      "learning_rate": 9.146411483253589e-05,
-      "loss": 0.0789,
-      "step": 721
-    },
-    {
-      "epoch": 65.64,
-      "learning_rate": 9.144497607655503e-05,
-      "loss": 0.1081,
-      "step": 722
-    },
-    {
-      "epoch": 65.73,
-      "learning_rate": 9.142583732057418e-05,
-      "loss": 0.0881,
-      "step": 723
-    },
-    {
-      "epoch": 65.82,
-      "learning_rate": 9.140669856459331e-05,
-      "loss": 0.0787,
-      "step": 724
-    },
-    {
-      "epoch": 65.91,
-      "learning_rate": 9.138755980861244e-05,
-      "loss": 0.0827,
-      "step": 725
-    },
-    {
-      "epoch": 66.0,
-      "learning_rate": 9.136842105263159e-05,
-      "loss": 0.0883,
-      "step": 726
-    },
-    {
-      "epoch": 66.09,
-      "learning_rate": 9.134928229665072e-05,
-      "loss": 0.0777,
-      "step": 727
-    },
-    {
-      "epoch": 66.18,
-      "learning_rate": 9.133014354066986e-05,
-      "loss": 0.0948,
-      "step": 728
-    },
-    {
-      "epoch": 66.27,
-      "learning_rate": 9.1311004784689e-05,
-      "loss": 0.09,
-      "step": 729
-    },
-    {
-      "epoch": 66.36,
-      "learning_rate": 9.129186602870814e-05,
-      "loss": 0.0835,
-      "step": 730
-    },
-    {
-      "epoch": 66.45,
-      "learning_rate": 9.127272727272727e-05,
-      "loss": 0.0737,
-      "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.0845,
-      "step": 733
-    },
-    {
-      "epoch": 66.73,
-      "learning_rate": 9.12153110047847e-05,
-      "loss": 0.0798,
-      "step": 734
-    },
-    {
-      "epoch": 66.82,
-      "learning_rate": 9.119617224880383e-05,
-      "loss": 0.0842,
-      "step": 735
-    },
-    {
-      "epoch": 66.91,
-      "learning_rate": 9.117703349282297e-05,
-      "loss": 0.0742,
-      "step": 736
-    },
-    {
-      "epoch": 67.0,
-      "learning_rate": 9.11578947368421e-05,
-      "loss": 0.0832,
-      "step": 737
-    },
-    {
-      "epoch": 67.09,
-      "learning_rate": 9.113875598086125e-05,
-      "loss": 0.0729,
-      "step": 738
-    },
-    {
-      "epoch": 67.18,
-      "learning_rate": 9.111961722488038e-05,
-      "loss": 0.0849,
-      "step": 739
-    },
-    {
-      "epoch": 67.27,
-      "learning_rate": 9.110047846889953e-05,
-      "loss": 0.0689,
-      "step": 740
-    },
-    {
-      "epoch": 67.27,
-      "eval_accuracy_NAT": 0.9897222518267984,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9395548092192798,
-      "eval_accuracy_concrete tiles": 0.9696385984223078,
-      "eval_accuracy_dark bricks": 0.8106693352834764,
-      "eval_accuracy_grass": 0.8453499792043115,
-      "eval_accuracy_red bricks": 0.8110795454545454,
-      "eval_accuracy_road": 0.9525288789303822,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.424365230336614,
-      "eval_accuracy_speedway bricks": 0.9849703183700266,
-      "eval_accuracy_steel": 0.8638529780357497,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9741131803107411,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8333595058626466,
-      "eval_iou_concrete tiles": 0.9161105815061964,
-      "eval_iou_dark bricks": 0.7605532083920125,
-      "eval_iou_grass": 0.7742260748205031,
-      "eval_iou_red bricks": 0.8005608131791097,
-      "eval_iou_road": 0.9433620391294611,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2848498306778438,
-      "eval_iou_speedway bricks": 0.9655526913216869,
-      "eval_iou_steel": 0.7824051256776737,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11377041786909103,
-      "eval_mean_accuracy": 0.6609024557756531,
-      "eval_mean_iou": 0.6180840808367597,
-      "eval_overall_accuracy": 0.969463136461046,
-      "eval_runtime": 3.4235,
-      "eval_samples_per_second": 10.515,
-      "eval_steps_per_second": 0.584,
-      "step": 740
-    },
-    {
-      "epoch": 67.36,
-      "learning_rate": 9.108133971291866e-05,
-      "loss": 0.0933,
-      "step": 741
-    },
-    {
-      "epoch": 67.45,
-      "learning_rate": 9.106220095693781e-05,
-      "loss": 0.074,
-      "step": 742
-    },
-    {
-      "epoch": 67.55,
-      "learning_rate": 9.104306220095694e-05,
-      "loss": 0.0708,
-      "step": 743
-    },
-    {
-      "epoch": 67.64,
-      "learning_rate": 9.102392344497608e-05,
-      "loss": 0.0767,
-      "step": 744
-    },
-    {
-      "epoch": 67.73,
-      "learning_rate": 9.100478468899522e-05,
-      "loss": 0.0843,
-      "step": 745
-    },
-    {
-      "epoch": 67.82,
-      "learning_rate": 9.098564593301436e-05,
-      "loss": 0.0881,
-      "step": 746
-    },
-    {
-      "epoch": 67.91,
-      "learning_rate": 9.09665071770335e-05,
-      "loss": 0.0939,
-      "step": 747
-    },
-    {
-      "epoch": 68.0,
-      "learning_rate": 9.094736842105264e-05,
-      "loss": 0.0897,
-      "step": 748
-    },
-    {
-      "epoch": 68.09,
-      "learning_rate": 9.092822966507177e-05,
-      "loss": 0.0775,
-      "step": 749
-    },
-    {
-      "epoch": 68.18,
-      "learning_rate": 9.090909090909092e-05,
-      "loss": 0.0768,
-      "step": 750
-    },
-    {
-      "epoch": 68.27,
-      "learning_rate": 9.088995215311005e-05,
-      "loss": 0.0793,
-      "step": 751
-    },
-    {
-      "epoch": 68.36,
-      "learning_rate": 9.08708133971292e-05,
-      "loss": 0.0805,
-      "step": 752
-    },
-    {
-      "epoch": 68.45,
-      "learning_rate": 9.085167464114833e-05,
-      "loss": 0.1074,
-      "step": 753
-    },
-    {
-      "epoch": 68.55,
-      "learning_rate": 9.083253588516747e-05,
-      "loss": 0.0737,
-      "step": 754
-    },
-    {
-      "epoch": 68.64,
-      "learning_rate": 9.08133971291866e-05,
-      "loss": 0.0826,
-      "step": 755
-    },
-    {
-      "epoch": 68.73,
-      "learning_rate": 9.079425837320574e-05,
-      "loss": 0.0765,
-      "step": 756
-    },
-    {
-      "epoch": 68.82,
-      "learning_rate": 9.077511961722488e-05,
-      "loss": 0.072,
-      "step": 757
-    },
-    {
-      "epoch": 68.91,
-      "learning_rate": 9.075598086124403e-05,
-      "loss": 0.0824,
-      "step": 758
-    },
-    {
-      "epoch": 69.0,
-      "learning_rate": 9.073684210526316e-05,
-      "loss": 0.0771,
-      "step": 759
-    },
-    {
-      "epoch": 69.09,
-      "learning_rate": 9.07177033492823e-05,
-      "loss": 0.0783,
-      "step": 760
-    },
-    {
-      "epoch": 69.09,
-      "eval_accuracy_NAT": 0.9892846530582861,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9250571376453925,
-      "eval_accuracy_concrete tiles": 0.987586375588577,
-      "eval_accuracy_dark bricks": 0.83183865132212,
-      "eval_accuracy_grass": 0.8574743898475099,
-      "eval_accuracy_red bricks": 0.8545809659090909,
-      "eval_accuracy_road": 0.9739633150218806,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4850715408154826,
-      "eval_accuracy_speedway bricks": 0.9870460412224077,
-      "eval_accuracy_steel": 0.8592193786689533,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9736016609644527,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8536089696066577,
-      "eval_iou_concrete tiles": 0.9021086440441279,
-      "eval_iou_dark bricks": 0.7815144282924712,
-      "eval_iou_grass": 0.7814979629411372,
-      "eval_iou_red bricks": 0.8101329742467598,
-      "eval_iou_road": 0.9612082185209097,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.29370034052213395,
-      "eval_iou_speedway bricks": 0.9639871860149355,
-      "eval_iou_steel": 0.7762086712995032,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10631099343299866,
-      "eval_mean_accuracy": 0.6731632653153616,
-      "eval_mean_iou": 0.6228899274194684,
-      "eval_overall_accuracy": 0.9709856245252821,
-      "eval_runtime": 3.6249,
-      "eval_samples_per_second": 9.931,
-      "eval_steps_per_second": 0.552,
-      "step": 760
-    },
-    {
-      "epoch": 69.18,
-      "learning_rate": 9.069856459330144e-05,
-      "loss": 0.1039,
-      "step": 761
-    },
-    {
-      "epoch": 69.27,
-      "learning_rate": 9.067942583732057e-05,
-      "loss": 0.0817,
-      "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.0754,
-      "step": 764
-    },
-    {
-      "epoch": 69.55,
-      "learning_rate": 9.062200956937799e-05,
-      "loss": 0.0768,
-      "step": 765
-    },
-    {
-      "epoch": 69.64,
-      "learning_rate": 9.060287081339714e-05,
-      "loss": 0.0831,
-      "step": 766
-    },
-    {
-      "epoch": 69.73,
-      "learning_rate": 9.058373205741627e-05,
-      "loss": 0.0865,
-      "step": 767
-    },
-    {
-      "epoch": 69.82,
-      "learning_rate": 9.05645933014354e-05,
-      "loss": 0.0903,
-      "step": 768
-    },
-    {
-      "epoch": 69.91,
-      "learning_rate": 9.054545454545455e-05,
-      "loss": 0.1031,
-      "step": 769
-    },
-    {
-      "epoch": 70.0,
-      "learning_rate": 9.052631578947369e-05,
-      "loss": 0.0876,
-      "step": 770
-    },
-    {
-      "epoch": 70.09,
-      "learning_rate": 9.050717703349282e-05,
-      "loss": 0.0777,
-      "step": 771
-    },
-    {
-      "epoch": 70.18,
-      "learning_rate": 9.048803827751197e-05,
-      "loss": 0.0778,
-      "step": 772
-    },
-    {
-      "epoch": 70.27,
-      "learning_rate": 9.04688995215311e-05,
-      "loss": 0.0786,
-      "step": 773
-    },
-    {
-      "epoch": 70.36,
-      "learning_rate": 9.044976076555023e-05,
-      "loss": 0.085,
-      "step": 774
-    },
-    {
-      "epoch": 70.45,
-      "learning_rate": 9.043062200956938e-05,
-      "loss": 0.0808,
-      "step": 775
-    },
-    {
-      "epoch": 70.55,
-      "learning_rate": 9.041148325358853e-05,
-      "loss": 0.0794,
-      "step": 776
-    },
-    {
-      "epoch": 70.64,
-      "learning_rate": 9.039234449760766e-05,
-      "loss": 0.0813,
-      "step": 777
-    },
-    {
-      "epoch": 70.73,
-      "learning_rate": 9.03732057416268e-05,
-      "loss": 0.0711,
-      "step": 778
-    },
-    {
-      "epoch": 70.82,
-      "learning_rate": 9.035406698564593e-05,
-      "loss": 0.0799,
-      "step": 779
-    },
-    {
-      "epoch": 70.91,
-      "learning_rate": 9.033492822966507e-05,
-      "loss": 0.0747,
-      "step": 780
-    },
-    {
-      "epoch": 70.91,
-      "eval_accuracy_NAT": 0.9907446095948471,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9231351130949826,
-      "eval_accuracy_concrete tiles": 0.9902464379624534,
-      "eval_accuracy_dark bricks": 0.8434625303106116,
-      "eval_accuracy_grass": 0.8665021651981532,
-      "eval_accuracy_red bricks": 0.8634588068181818,
-      "eval_accuracy_road": 0.9773164511654959,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5318005127780994,
-      "eval_accuracy_speedway bricks": 0.9884912029175974,
-      "eval_accuracy_steel": 0.8526482422003826,
-      "eval_accuracy_tiles": 0.08991918915217094,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9741119929549645,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8680594093371944,
-      "eval_iou_concrete tiles": 0.897221375737596,
-      "eval_iou_dark bricks": 0.792377784205959,
-      "eval_iou_grass": 0.7965768411447373,
-      "eval_iou_red bricks": 0.8365731980044727,
-      "eval_iou_road": 0.9650598813120157,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.35168641750227897,
-      "eval_iou_speedway bricks": 0.9643313354218984,
-      "eval_iou_steel": 0.7785825064369928,
-      "eval_iou_tiles": 0.08991918915217094,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09868098050355911,
-      "eval_mean_accuracy": 0.6859788662456135,
-      "eval_mean_iou": 0.6395769177854062,
-      "eval_overall_accuracy": 0.9728126525878906,
-      "eval_runtime": 3.3936,
-      "eval_samples_per_second": 10.608,
-      "eval_steps_per_second": 0.589,
-      "step": 780
-    },
-    {
-      "epoch": 71.0,
-      "learning_rate": 9.031578947368423e-05,
-      "loss": 0.0748,
-      "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.0817,
-      "step": 783
-    },
-    {
-      "epoch": 71.27,
-      "learning_rate": 9.025837320574164e-05,
-      "loss": 0.0933,
-      "step": 784
-    },
-    {
-      "epoch": 71.36,
-      "learning_rate": 9.023923444976077e-05,
-      "loss": 0.0773,
-      "step": 785
-    },
-    {
-      "epoch": 71.45,
-      "learning_rate": 9.02200956937799e-05,
-      "loss": 0.0864,
-      "step": 786
-    },
-    {
-      "epoch": 71.55,
-      "learning_rate": 9.020095693779906e-05,
-      "loss": 0.0724,
-      "step": 787
-    },
-    {
-      "epoch": 71.64,
-      "learning_rate": 9.018181818181819e-05,
-      "loss": 0.0811,
-      "step": 788
-    },
-    {
-      "epoch": 71.73,
-      "learning_rate": 9.016267942583732e-05,
-      "loss": 0.0723,
-      "step": 789
-    },
-    {
-      "epoch": 71.82,
-      "learning_rate": 9.014354066985647e-05,
-      "loss": 0.0837,
-      "step": 790
-    },
-    {
-      "epoch": 71.91,
-      "learning_rate": 9.01244019138756e-05,
-      "loss": 0.0779,
-      "step": 791
-    },
-    {
-      "epoch": 72.0,
-      "learning_rate": 9.010526315789473e-05,
-      "loss": 0.0641,
-      "step": 792
-    },
-    {
-      "epoch": 72.09,
-      "learning_rate": 9.008612440191389e-05,
-      "loss": 0.0757,
-      "step": 793
-    },
-    {
-      "epoch": 72.18,
-      "learning_rate": 9.006698564593302e-05,
-      "loss": 0.086,
-      "step": 794
-    },
-    {
-      "epoch": 72.27,
-      "learning_rate": 9.004784688995216e-05,
-      "loss": 0.0811,
-      "step": 795
-    },
-    {
-      "epoch": 72.36,
-      "learning_rate": 9.00287081339713e-05,
-      "loss": 0.0787,
-      "step": 796
-    },
-    {
-      "epoch": 72.45,
-      "learning_rate": 9.000956937799043e-05,
-      "loss": 0.0787,
-      "step": 797
-    },
-    {
-      "epoch": 72.55,
-      "learning_rate": 8.999043062200956e-05,
-      "loss": 0.0707,
-      "step": 798
-    },
-    {
-      "epoch": 72.64,
-      "learning_rate": 8.997129186602872e-05,
-      "loss": 0.0812,
-      "step": 799
-    },
-    {
-      "epoch": 72.73,
-      "learning_rate": 8.995215311004786e-05,
-      "loss": 0.0832,
-      "step": 800
-    },
-    {
-      "epoch": 72.73,
-      "eval_accuracy_NAT": 0.9908023091098889,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9322972552470064,
-      "eval_accuracy_concrete tiles": 0.9887788173423837,
-      "eval_accuracy_dark bricks": 0.8481967591701628,
-      "eval_accuracy_grass": 0.8780324133134347,
-      "eval_accuracy_red bricks": 0.8787286931818182,
-      "eval_accuracy_road": 0.9773720747726385,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4578612190885783,
-      "eval_accuracy_speedway bricks": 0.9861910123614123,
-      "eval_accuracy_steel": 0.8581475496339291,
-      "eval_accuracy_tiles": 0.10354745925215723,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9749560892064526,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.867676118313059,
-      "eval_iou_concrete tiles": 0.9074048093380847,
-      "eval_iou_dark bricks": 0.7940402839332684,
-      "eval_iou_grass": 0.8050013778783237,
-      "eval_iou_red bricks": 0.8399524779361847,
-      "eval_iou_road": 0.9661600030600633,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.29879639458107693,
-      "eval_iou_speedway bricks": 0.9670450544801531,
-      "eval_iou_steel": 0.7852653815685239,
-      "eval_iou_tiles": 0.10354745925215723,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09568993747234344,
-      "eval_mean_accuracy": 0.6846119663441085,
-      "eval_mean_iou": 0.6392188807344112,
-      "eval_overall_accuracy": 0.9733462863498263,
-      "eval_runtime": 3.871,
-      "eval_samples_per_second": 9.3,
-      "eval_steps_per_second": 0.517,
-      "step": 800
-    },
-    {
-      "epoch": 72.82,
-      "learning_rate": 8.993301435406699e-05,
-      "loss": 0.0737,
-      "step": 801
-    },
-    {
-      "epoch": 72.91,
-      "learning_rate": 8.991387559808613e-05,
-      "loss": 0.0725,
-      "step": 802
-    },
-    {
-      "epoch": 73.0,
-      "learning_rate": 8.989473684210527e-05,
-      "loss": 0.072,
-      "step": 803
-    },
-    {
-      "epoch": 73.09,
-      "learning_rate": 8.98755980861244e-05,
-      "loss": 0.0662,
-      "step": 804
-    },
-    {
-      "epoch": 73.18,
-      "learning_rate": 8.985645933014354e-05,
-      "loss": 0.0783,
-      "step": 805
-    },
-    {
-      "epoch": 73.27,
-      "learning_rate": 8.983732057416269e-05,
-      "loss": 0.0814,
-      "step": 806
-    },
-    {
-      "epoch": 73.36,
-      "learning_rate": 8.981818181818182e-05,
-      "loss": 0.0711,
-      "step": 807
-    },
-    {
-      "epoch": 73.45,
-      "learning_rate": 8.979904306220097e-05,
-      "loss": 0.0824,
-      "step": 808
-    },
-    {
-      "epoch": 73.55,
-      "learning_rate": 8.97799043062201e-05,
-      "loss": 0.0794,
-      "step": 809
-    },
-    {
-      "epoch": 73.64,
-      "learning_rate": 8.976076555023923e-05,
-      "loss": 0.074,
-      "step": 810
-    },
-    {
-      "epoch": 73.73,
-      "learning_rate": 8.974162679425838e-05,
-      "loss": 0.0843,
-      "step": 811
-    },
-    {
-      "epoch": 73.82,
-      "learning_rate": 8.972248803827752e-05,
-      "loss": 0.0763,
-      "step": 812
-    },
-    {
-      "epoch": 73.91,
-      "learning_rate": 8.970334928229665e-05,
-      "loss": 0.084,
-      "step": 813
-    },
-    {
-      "epoch": 74.0,
-      "learning_rate": 8.96842105263158e-05,
-      "loss": 0.0828,
-      "step": 814
-    },
-    {
-      "epoch": 74.09,
-      "learning_rate": 8.966507177033493e-05,
-      "loss": 0.0803,
-      "step": 815
-    },
-    {
-      "epoch": 74.18,
-      "learning_rate": 8.964593301435406e-05,
-      "loss": 0.0707,
-      "step": 816
-    },
-    {
-      "epoch": 74.27,
-      "learning_rate": 8.962679425837321e-05,
-      "loss": 0.0668,
-      "step": 817
-    },
-    {
-      "epoch": 74.36,
-      "learning_rate": 8.960765550239235e-05,
-      "loss": 0.0782,
-      "step": 818
-    },
-    {
-      "epoch": 74.45,
-      "learning_rate": 8.958851674641149e-05,
-      "loss": 0.0718,
-      "step": 819
-    },
-    {
-      "epoch": 74.55,
-      "learning_rate": 8.956937799043063e-05,
-      "loss": 0.0768,
-      "step": 820
-    },
-    {
-      "epoch": 74.55,
-      "eval_accuracy_NAT": 0.9908855940367992,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9388265054294176,
-      "eval_accuracy_concrete tiles": 0.982449703418333,
-      "eval_accuracy_dark bricks": 0.8645163773526808,
-      "eval_accuracy_grass": 0.8632133008524485,
-      "eval_accuracy_red bricks": 0.8893821022727273,
-      "eval_accuracy_road": 0.9661614995157118,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4956579273840046,
-      "eval_accuracy_speedway bricks": 0.9870955206624935,
-      "eval_accuracy_steel": 0.8382857331310599,
-      "eval_accuracy_tiles": 0.03848787837282564,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9749679590570397,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8589876850766691,
-      "eval_iou_concrete tiles": 0.916303076966949,
-      "eval_iou_dark bricks": 0.8054868208714363,
-      "eval_iou_grass": 0.800438170056651,
-      "eval_iou_red bricks": 0.8261586673264061,
-      "eval_iou_road": 0.9569998466889607,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32982333840876155,
-      "eval_iou_speedway bricks": 0.9666439561855137,
-      "eval_iou_steel": 0.7759537819294671,
-      "eval_iou_tiles": 0.03848787837282564,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0993707925081253,
-      "eval_mean_accuracy": 0.6811509340329618,
-      "eval_mean_iou": 0.6346347062262061,
-      "eval_overall_accuracy": 0.9724579917060004,
-      "eval_runtime": 3.6557,
-      "eval_samples_per_second": 9.848,
-      "eval_steps_per_second": 0.547,
-      "step": 820
-    },
-    {
-      "epoch": 74.64,
-      "learning_rate": 8.955023923444976e-05,
-      "loss": 0.0826,
-      "step": 821
-    },
-    {
-      "epoch": 74.73,
-      "learning_rate": 8.95311004784689e-05,
-      "loss": 0.0816,
-      "step": 822
-    },
-    {
-      "epoch": 74.82,
-      "learning_rate": 8.951196172248804e-05,
-      "loss": 0.0869,
-      "step": 823
-    },
-    {
-      "epoch": 74.91,
-      "learning_rate": 8.949282296650719e-05,
-      "loss": 0.0852,
-      "step": 824
-    },
-    {
-      "epoch": 75.0,
-      "learning_rate": 8.947368421052632e-05,
-      "loss": 0.0678,
-      "step": 825
-    },
-    {
-      "epoch": 75.09,
-      "learning_rate": 8.945454545454546e-05,
-      "loss": 0.0701,
-      "step": 826
-    },
-    {
-      "epoch": 75.18,
-      "learning_rate": 8.94354066985646e-05,
-      "loss": 0.0722,
-      "step": 827
-    },
-    {
-      "epoch": 75.27,
-      "learning_rate": 8.941626794258373e-05,
-      "loss": 0.0753,
-      "step": 828
-    },
-    {
-      "epoch": 75.36,
-      "learning_rate": 8.939712918660287e-05,
-      "loss": 0.0853,
-      "step": 829
-    },
-    {
-      "epoch": 75.45,
-      "learning_rate": 8.937799043062202e-05,
-      "loss": 0.0661,
-      "step": 830
-    },
-    {
-      "epoch": 75.55,
-      "learning_rate": 8.935885167464115e-05,
-      "loss": 0.074,
-      "step": 831
-    },
-    {
-      "epoch": 75.64,
-      "learning_rate": 8.933971291866028e-05,
-      "loss": 0.0804,
-      "step": 832
-    },
-    {
-      "epoch": 75.73,
-      "learning_rate": 8.932057416267943e-05,
-      "loss": 0.0771,
-      "step": 833
-    },
-    {
-      "epoch": 75.82,
-      "learning_rate": 8.930143540669856e-05,
-      "loss": 0.0722,
-      "step": 834
-    },
-    {
-      "epoch": 75.91,
-      "learning_rate": 8.92822966507177e-05,
-      "loss": 0.0722,
-      "step": 835
-    },
-    {
-      "epoch": 76.0,
-      "learning_rate": 8.926315789473685e-05,
-      "loss": 0.0672,
-      "step": 836
-    },
-    {
-      "epoch": 76.09,
-      "learning_rate": 8.924401913875598e-05,
-      "loss": 0.0729,
-      "step": 837
-    },
-    {
-      "epoch": 76.18,
-      "learning_rate": 8.922488038277512e-05,
-      "loss": 0.0893,
-      "step": 838
-    },
-    {
-      "epoch": 76.27,
-      "learning_rate": 8.920574162679426e-05,
-      "loss": 0.0844,
-      "step": 839
-    },
-    {
-      "epoch": 76.36,
-      "learning_rate": 8.91866028708134e-05,
-      "loss": 0.0738,
-      "step": 840
-    },
-    {
-      "epoch": 76.36,
-      "eval_accuracy_NAT": 0.9916820943759382,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.941506341473883,
-      "eval_accuracy_concrete tiles": 0.9726961413807864,
-      "eval_accuracy_dark bricks": 0.8335321966052115,
-      "eval_accuracy_grass": 0.8527350699189492,
-      "eval_accuracy_red bricks": 0.7977627840909091,
-      "eval_accuracy_road": 0.9683429122827808,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.44286383811650537,
-      "eval_accuracy_speedway bricks": 0.9845859475976525,
-      "eval_accuracy_steel": 0.8555586702724095,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9751643463183423,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8557247469544939,
-      "eval_iou_concrete tiles": 0.9230791550603529,
-      "eval_iou_dark bricks": 0.786632764257174,
-      "eval_iou_grass": 0.7920747721805415,
-      "eval_iou_red bricks": 0.7734549836460665,
-      "eval_iou_road": 0.9582332190994026,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2943849875384841,
-      "eval_iou_speedway bricks": 0.9685526340782388,
-      "eval_iou_steel": 0.7892018922166618,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10128644108772278,
-      "eval_mean_accuracy": 0.6647127689319251,
-      "eval_mean_iou": 0.6243464231807506,
-      "eval_overall_accuracy": 0.9723665449354384,
-      "eval_runtime": 3.7787,
-      "eval_samples_per_second": 9.527,
-      "eval_steps_per_second": 0.529,
-      "step": 840
-    },
-    {
-      "epoch": 76.45,
-      "learning_rate": 8.916746411483254e-05,
-      "loss": 0.0721,
-      "step": 841
-    },
-    {
-      "epoch": 76.55,
-      "learning_rate": 8.914832535885168e-05,
-      "loss": 0.0673,
-      "step": 842
-    },
-    {
-      "epoch": 76.64,
-      "learning_rate": 8.912918660287082e-05,
-      "loss": 0.0647,
-      "step": 843
-    },
-    {
-      "epoch": 76.73,
-      "learning_rate": 8.911004784688995e-05,
-      "loss": 0.0674,
-      "step": 844
-    },
-    {
-      "epoch": 76.82,
-      "learning_rate": 8.90909090909091e-05,
-      "loss": 0.0702,
-      "step": 845
-    },
-    {
-      "epoch": 76.91,
-      "learning_rate": 8.907177033492823e-05,
-      "loss": 0.0736,
-      "step": 846
-    },
-    {
-      "epoch": 77.0,
-      "learning_rate": 8.905263157894737e-05,
-      "loss": 0.0794,
-      "step": 847
-    },
-    {
-      "epoch": 77.09,
-      "learning_rate": 8.903349282296652e-05,
-      "loss": 0.0624,
-      "step": 848
-    },
-    {
-      "epoch": 77.18,
-      "learning_rate": 8.901435406698565e-05,
-      "loss": 0.0688,
-      "step": 849
-    },
-    {
-      "epoch": 77.27,
-      "learning_rate": 8.899521531100478e-05,
-      "loss": 0.0696,
-      "step": 850
-    },
-    {
-      "epoch": 77.36,
-      "learning_rate": 8.897607655502393e-05,
-      "loss": 0.0777,
-      "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.0833,
-      "step": 853
-    },
-    {
-      "epoch": 77.64,
-      "learning_rate": 8.891866028708135e-05,
-      "loss": 0.0577,
-      "step": 854
-    },
-    {
-      "epoch": 77.73,
-      "learning_rate": 8.889952153110048e-05,
-      "loss": 0.077,
-      "step": 855
-    },
-    {
-      "epoch": 77.82,
-      "learning_rate": 8.888038277511961e-05,
-      "loss": 0.069,
-      "step": 856
-    },
-    {
-      "epoch": 77.91,
-      "learning_rate": 8.886124401913876e-05,
-      "loss": 0.0797,
-      "step": 857
-    },
-    {
-      "epoch": 78.0,
-      "learning_rate": 8.88421052631579e-05,
-      "loss": 0.0723,
-      "step": 858
-    },
-    {
-      "epoch": 78.09,
-      "learning_rate": 8.882296650717704e-05,
-      "loss": 0.0698,
-      "step": 859
-    },
-    {
-      "epoch": 78.18,
-      "learning_rate": 8.880382775119618e-05,
-      "loss": 0.073,
-      "step": 860
-    },
-    {
-      "epoch": 78.18,
-      "eval_accuracy_NAT": 0.9887180685237323,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.923360444654277,
-      "eval_accuracy_concrete tiles": 0.9877698281660857,
-      "eval_accuracy_dark bricks": 0.8652861706631769,
-      "eval_accuracy_grass": 0.8928620109955019,
-      "eval_accuracy_red bricks": 0.8924005681818182,
-      "eval_accuracy_road": 0.9728157758223527,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5640283406390428,
-      "eval_accuracy_speedway bricks": 0.9908245809031084,
-      "eval_accuracy_steel": 0.8369500692566454,
-      "eval_accuracy_tiles": 0.1644295302013423,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9744004329312975,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8639812050315697,
-      "eval_iou_concrete tiles": 0.8955232155232156,
-      "eval_iou_dark bricks": 0.8030936305504948,
-      "eval_iou_grass": 0.7960183967643631,
-      "eval_iou_red bricks": 0.8266447368421053,
-      "eval_iou_road": 0.9628633151214492,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3589865066413995,
-      "eval_iou_speedway bricks": 0.9648917760364791,
-      "eval_iou_steel": 0.7718135093214926,
-      "eval_iou_tiles": 0.16441827021844826,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09554906189441681,
-      "eval_mean_accuracy": 0.698418876000545,
-      "eval_mean_iou": 0.6448180765371011,
-      "eval_overall_accuracy": 0.9725341796875,
-      "eval_runtime": 3.4495,
-      "eval_samples_per_second": 10.436,
-      "eval_steps_per_second": 0.58,
-      "step": 860
-    },
-    {
-      "epoch": 78.27,
-      "learning_rate": 8.878468899521531e-05,
-      "loss": 0.0798,
-      "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.0712,
-      "step": 863
-    },
-    {
-      "epoch": 78.55,
-      "learning_rate": 8.872727272727274e-05,
-      "loss": 0.0703,
-      "step": 864
-    },
-    {
-      "epoch": 78.64,
-      "learning_rate": 8.870813397129187e-05,
-      "loss": 0.0712,
-      "step": 865
-    },
-    {
-      "epoch": 78.73,
-      "learning_rate": 8.868899521531102e-05,
-      "loss": 0.0699,
-      "step": 866
-    },
-    {
-      "epoch": 78.82,
-      "learning_rate": 8.866985645933015e-05,
-      "loss": 0.0831,
-      "step": 867
-    },
-    {
-      "epoch": 78.91,
-      "learning_rate": 8.865071770334928e-05,
-      "loss": 0.0774,
-      "step": 868
-    },
-    {
-      "epoch": 79.0,
-      "learning_rate": 8.863157894736842e-05,
-      "loss": 0.0683,
-      "step": 869
-    },
-    {
-      "epoch": 79.09,
-      "learning_rate": 8.861244019138757e-05,
-      "loss": 0.0769,
-      "step": 870
-    },
-    {
-      "epoch": 79.18,
-      "learning_rate": 8.85933014354067e-05,
-      "loss": 0.0692,
-      "step": 871
-    },
-    {
-      "epoch": 79.27,
-      "learning_rate": 8.857416267942585e-05,
-      "loss": 0.0747,
-      "step": 872
-    },
-    {
-      "epoch": 79.36,
-      "learning_rate": 8.855502392344498e-05,
-      "loss": 0.0703,
-      "step": 873
-    },
-    {
-      "epoch": 79.45,
-      "learning_rate": 8.853588516746411e-05,
-      "loss": 0.0671,
-      "step": 874
-    },
-    {
-      "epoch": 79.55,
-      "learning_rate": 8.851674641148326e-05,
-      "loss": 0.0769,
-      "step": 875
-    },
-    {
-      "epoch": 79.64,
-      "learning_rate": 8.84976076555024e-05,
-      "loss": 0.0715,
-      "step": 876
-    },
-    {
-      "epoch": 79.73,
-      "learning_rate": 8.847846889952154e-05,
-      "loss": 0.0742,
-      "step": 877
-    },
-    {
-      "epoch": 79.82,
-      "learning_rate": 8.845933014354068e-05,
-      "loss": 0.0691,
-      "step": 878
-    },
-    {
-      "epoch": 79.91,
-      "learning_rate": 8.844019138755981e-05,
-      "loss": 0.0689,
-      "step": 879
-    },
-    {
-      "epoch": 80.0,
-      "learning_rate": 8.842105263157894e-05,
-      "loss": 0.0829,
-      "step": 880
-    },
-    {
-      "epoch": 80.0,
-      "eval_accuracy_NAT": 0.9917106794567847,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9318385445727284,
-      "eval_accuracy_concrete tiles": 0.972482113373693,
-      "eval_accuracy_dark bricks": 0.85474000230938,
-      "eval_accuracy_grass": 0.8552550180520556,
-      "eval_accuracy_red bricks": 0.890625,
-      "eval_accuracy_road": 0.9764180089892586,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.42510958564221324,
-      "eval_accuracy_speedway bricks": 0.9885406823576832,
-      "eval_accuracy_steel": 0.853843743816371,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9750679343229547,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8629907781896234,
-      "eval_iou_concrete tiles": 0.9300271937775958,
-      "eval_iou_dark bricks": 0.8017546393241389,
-      "eval_iou_grass": 0.7926809911144369,
-      "eval_iou_red bricks": 0.8399196249162759,
-      "eval_iou_road": 0.9666924855173008,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2869477836912427,
-      "eval_iou_speedway bricks": 0.9675137647783492,
-      "eval_iou_steel": 0.7917568177126737,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09802630543708801,
-      "eval_mean_accuracy": 0.6723510291207823,
-      "eval_mean_iou": 0.631950154872661,
-      "eval_overall_accuracy": 0.9731212192111545,
-      "eval_runtime": 3.7841,
-      "eval_samples_per_second": 9.514,
-      "eval_steps_per_second": 0.529,
-      "step": 880
-    },
-    {
-      "epoch": 80.09,
-      "learning_rate": 8.840191387559809e-05,
-      "loss": 0.0725,
-      "step": 881
-    },
-    {
-      "epoch": 80.18,
-      "learning_rate": 8.838277511961724e-05,
-      "loss": 0.0833,
-      "step": 882
-    },
-    {
-      "epoch": 80.27,
-      "learning_rate": 8.836363636363637e-05,
-      "loss": 0.0612,
-      "step": 883
-    },
-    {
-      "epoch": 80.36,
-      "learning_rate": 8.834449760765551e-05,
-      "loss": 0.0729,
-      "step": 884
-    },
-    {
-      "epoch": 80.45,
-      "learning_rate": 8.832535885167465e-05,
-      "loss": 0.0683,
-      "step": 885
-    },
-    {
-      "epoch": 80.55,
-      "learning_rate": 8.830622009569378e-05,
-      "loss": 0.075,
-      "step": 886
-    },
-    {
-      "epoch": 80.64,
-      "learning_rate": 8.828708133971292e-05,
-      "loss": 0.0633,
-      "step": 887
-    },
-    {
-      "epoch": 80.73,
-      "learning_rate": 8.826794258373207e-05,
-      "loss": 0.0796,
-      "step": 888
-    },
-    {
-      "epoch": 80.82,
-      "learning_rate": 8.82488038277512e-05,
-      "loss": 0.0599,
-      "step": 889
-    },
-    {
-      "epoch": 80.91,
-      "learning_rate": 8.822966507177035e-05,
-      "loss": 0.0697,
-      "step": 890
-    },
-    {
-      "epoch": 81.0,
-      "learning_rate": 8.821052631578948e-05,
-      "loss": 0.0762,
-      "step": 891
-    },
-    {
-      "epoch": 81.09,
-      "learning_rate": 8.819138755980861e-05,
-      "loss": 0.0748,
-      "step": 892
-    },
-    {
-      "epoch": 81.18,
-      "learning_rate": 8.817224880382776e-05,
-      "loss": 0.072,
-      "step": 893
-    },
-    {
-      "epoch": 81.27,
-      "learning_rate": 8.81531100478469e-05,
-      "loss": 0.0687,
-      "step": 894
-    },
-    {
-      "epoch": 81.36,
-      "learning_rate": 8.813397129186603e-05,
-      "loss": 0.068,
-      "step": 895
-    },
-    {
-      "epoch": 81.45,
-      "learning_rate": 8.811483253588518e-05,
-      "loss": 0.0739,
-      "step": 896
-    },
-    {
-      "epoch": 81.55,
-      "learning_rate": 8.809569377990431e-05,
-      "loss": 0.0775,
-      "step": 897
-    },
-    {
-      "epoch": 81.64,
-      "learning_rate": 8.807655502392344e-05,
-      "loss": 0.0746,
-      "step": 898
-    },
-    {
-      "epoch": 81.73,
-      "learning_rate": 8.805741626794259e-05,
-      "loss": 0.0713,
-      "step": 899
-    },
-    {
-      "epoch": 81.82,
-      "learning_rate": 8.803827751196173e-05,
-      "loss": 0.0874,
-      "step": 900
-    },
-    {
-      "epoch": 81.82,
-      "eval_accuracy_NAT": 0.991639922559134,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.943866957809348,
-      "eval_accuracy_concrete tiles": 0.9780774169877087,
-      "eval_accuracy_dark bricks": 0.8520457257226435,
-      "eval_accuracy_grass": 0.8556220007898867,
-      "eval_accuracy_red bricks": 0.9273792613636364,
-      "eval_accuracy_road": 0.9486170882976395,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.44923221128663193,
-      "eval_accuracy_speedway bricks": 0.9838328946558584,
-      "eval_accuracy_steel": 0.8673735241738671,
-      "eval_accuracy_tiles": 0.09786330639638406,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9763561788580174,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8471363342526372,
-      "eval_iou_concrete tiles": 0.9211299239806496,
-      "eval_iou_dark bricks": 0.7994582881906825,
-      "eval_iou_grass": 0.8000026143021938,
-      "eval_iou_red bricks": 0.7541149292520936,
-      "eval_iou_road": 0.9386448234149704,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2936512227208016,
-      "eval_iou_speedway bricks": 0.9681307830201059,
-      "eval_iou_steel": 0.792034632034632,
-      "eval_iou_tiles": 0.09786330639638406,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10007079690694809,
-      "eval_mean_accuracy": 0.6842731007725184,
-      "eval_mean_iou": 0.6298863874171666,
-      "eval_overall_accuracy": 0.9712291293674045,
-      "eval_runtime": 3.6304,
-      "eval_samples_per_second": 9.916,
-      "eval_steps_per_second": 0.551,
-      "step": 900
-    },
-    {
-      "epoch": 81.91,
-      "learning_rate": 8.801913875598087e-05,
-      "loss": 0.0714,
-      "step": 901
-    },
-    {
-      "epoch": 82.0,
-      "learning_rate": 8.800000000000001e-05,
-      "loss": 0.0708,
-      "step": 902
-    },
-    {
-      "epoch": 82.09,
-      "learning_rate": 8.798086124401914e-05,
-      "loss": 0.0728,
-      "step": 903
-    },
-    {
-      "epoch": 82.18,
-      "learning_rate": 8.796172248803828e-05,
-      "loss": 0.0668,
-      "step": 904
-    },
-    {
-      "epoch": 82.27,
-      "learning_rate": 8.794258373205742e-05,
-      "loss": 0.0713,
-      "step": 905
-    },
-    {
-      "epoch": 82.36,
-      "learning_rate": 8.792344497607657e-05,
-      "loss": 0.0718,
-      "step": 906
-    },
-    {
-      "epoch": 82.45,
-      "learning_rate": 8.79043062200957e-05,
-      "loss": 0.067,
-      "step": 907
-    },
-    {
-      "epoch": 82.55,
-      "learning_rate": 8.788516746411483e-05,
-      "loss": 0.07,
-      "step": 908
-    },
-    {
-      "epoch": 82.64,
-      "learning_rate": 8.786602870813398e-05,
-      "loss": 0.0634,
-      "step": 909
-    },
-    {
-      "epoch": 82.73,
-      "learning_rate": 8.784688995215311e-05,
-      "loss": 0.0704,
-      "step": 910
-    },
-    {
-      "epoch": 82.82,
-      "learning_rate": 8.782775119617225e-05,
-      "loss": 0.0782,
-      "step": 911
-    },
-    {
-      "epoch": 82.91,
-      "learning_rate": 8.78086124401914e-05,
-      "loss": 0.0731,
-      "step": 912
-    },
-    {
-      "epoch": 83.0,
-      "learning_rate": 8.778947368421053e-05,
-      "loss": 0.082,
-      "step": 913
-    },
-    {
-      "epoch": 83.09,
-      "learning_rate": 8.777033492822966e-05,
-      "loss": 0.0639,
-      "step": 914
-    },
-    {
-      "epoch": 83.18,
-      "learning_rate": 8.775119617224881e-05,
-      "loss": 0.0765,
-      "step": 915
-    },
-    {
-      "epoch": 83.27,
-      "learning_rate": 8.773205741626794e-05,
-      "loss": 0.072,
-      "step": 916
-    },
-    {
-      "epoch": 83.36,
-      "learning_rate": 8.771291866028709e-05,
-      "loss": 0.0852,
-      "step": 917
-    },
-    {
-      "epoch": 83.45,
-      "learning_rate": 8.769377990430623e-05,
-      "loss": 0.0822,
-      "step": 918
-    },
-    {
-      "epoch": 83.55,
-      "learning_rate": 8.767464114832536e-05,
-      "loss": 0.0666,
-      "step": 919
-    },
-    {
-      "epoch": 83.64,
-      "learning_rate": 8.76555023923445e-05,
-      "loss": 0.0605,
-      "step": 920
-    },
-    {
-      "epoch": 83.64,
-      "eval_accuracy_NAT": 0.9913203695874502,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9396889351474312,
-      "eval_accuracy_concrete tiles": 0.9915611814345991,
-      "eval_accuracy_dark bricks": 0.8553558369577768,
-      "eval_accuracy_grass": 0.8582398109864147,
-      "eval_accuracy_red bricks": 0.8995028409090909,
-      "eval_accuracy_road": 0.972344184370492,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4101949108152069,
-      "eval_accuracy_speedway bricks": 0.9840356396786492,
-      "eval_accuracy_steel": 0.8832036145372997,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9758365849049263,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.856032716338253,
-      "eval_iou_concrete tiles": 0.8948428575370437,
-      "eval_iou_dark bricks": 0.8008865503820095,
-      "eval_iou_grass": 0.797162047662796,
-      "eval_iou_red bricks": 0.8104303311470165,
-      "eval_iou_road": 0.9616426888625791,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2729340548472897,
-      "eval_iou_speedway bricks": 0.9691186459412288,
-      "eval_iou_steel": 0.7995253093699154,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09636644273996353,
-      "eval_mean_accuracy": 0.6758036403403392,
-      "eval_mean_iou": 0.6260316759225428,
-      "eval_overall_accuracy": 0.9728425343831381,
-      "eval_runtime": 3.3293,
-      "eval_samples_per_second": 10.813,
-      "eval_steps_per_second": 0.601,
-      "step": 920
-    },
-    {
-      "epoch": 83.73,
-      "learning_rate": 8.763636363636364e-05,
-      "loss": 0.0814,
-      "step": 921
-    },
-    {
-      "epoch": 83.82,
-      "learning_rate": 8.761722488038277e-05,
-      "loss": 0.0695,
-      "step": 922
-    },
-    {
-      "epoch": 83.91,
-      "learning_rate": 8.759808612440192e-05,
-      "loss": 0.0704,
-      "step": 923
-    },
-    {
-      "epoch": 84.0,
-      "learning_rate": 8.757894736842106e-05,
-      "loss": 0.0717,
-      "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.0643,
-      "step": 926
-    },
-    {
-      "epoch": 84.27,
-      "learning_rate": 8.752153110047847e-05,
-      "loss": 0.072,
-      "step": 927
-    },
-    {
-      "epoch": 84.36,
-      "learning_rate": 8.75023923444976e-05,
-      "loss": 0.0687,
-      "step": 928
-    },
-    {
-      "epoch": 84.45,
-      "learning_rate": 8.748325358851675e-05,
-      "loss": 0.0699,
-      "step": 929
-    },
-    {
-      "epoch": 84.55,
-      "learning_rate": 8.74641148325359e-05,
-      "loss": 0.061,
-      "step": 930
-    },
-    {
-      "epoch": 84.64,
-      "learning_rate": 8.744497607655503e-05,
-      "loss": 0.0715,
-      "step": 931
-    },
-    {
-      "epoch": 84.73,
-      "learning_rate": 8.742583732057416e-05,
-      "loss": 0.0957,
-      "step": 932
-    },
-    {
-      "epoch": 84.82,
-      "learning_rate": 8.74066985645933e-05,
-      "loss": 0.0636,
-      "step": 933
-    },
-    {
-      "epoch": 84.91,
-      "learning_rate": 8.738755980861244e-05,
-      "loss": 0.0711,
-      "step": 934
-    },
-    {
-      "epoch": 85.0,
-      "learning_rate": 8.736842105263158e-05,
-      "loss": 0.0659,
-      "step": 935
-    },
-    {
-      "epoch": 85.09,
-      "learning_rate": 8.734928229665073e-05,
-      "loss": 0.0705,
-      "step": 936
-    },
-    {
-      "epoch": 85.18,
-      "learning_rate": 8.733014354066986e-05,
-      "loss": 0.0712,
-      "step": 937
-    },
-    {
-      "epoch": 85.27,
-      "learning_rate": 8.7311004784689e-05,
-      "loss": 0.0588,
-      "step": 938
-    },
-    {
-      "epoch": 85.36,
-      "learning_rate": 8.729186602870814e-05,
-      "loss": 0.0685,
-      "step": 939
-    },
-    {
-      "epoch": 85.45,
-      "learning_rate": 8.727272727272727e-05,
-      "loss": 0.0756,
-      "step": 940
-    },
-    {
-      "epoch": 85.45,
-      "eval_accuracy_NAT": 0.9910576338752265,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9381880660114168,
-      "eval_accuracy_concrete tiles": 0.9542285819115759,
-      "eval_accuracy_dark bricks": 0.8326084446326162,
-      "eval_accuracy_grass": 0.8402541617590007,
-      "eval_accuracy_red bricks": 0.8474786931818182,
-      "eval_accuracy_road": 0.970324805589447,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.42607449066798997,
-      "eval_accuracy_speedway bricks": 0.9844332854227535,
-      "eval_accuracy_steel": 0.881529912274916,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.975599982352183,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.848327075328909,
-      "eval_iou_concrete tiles": 0.9277072619719985,
-      "eval_iou_dark bricks": 0.7830588235294118,
-      "eval_iou_grass": 0.7718270338991213,
-      "eval_iou_red bricks": 0.8109072375127421,
-      "eval_iou_road": 0.9550700074744465,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.28460150265173834,
-      "eval_iou_speedway bricks": 0.970255788090183,
-      "eval_iou_steel": 0.8011134172017713,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10112927109003067,
-      "eval_mean_accuracy": 0.6666290827174431,
-      "eval_mean_iou": 0.6252667792317312,
-      "eval_overall_accuracy": 0.9717302322387695,
-      "eval_runtime": 3.3889,
-      "eval_samples_per_second": 10.623,
-      "eval_steps_per_second": 0.59,
-      "step": 940
-    },
-    {
-      "epoch": 85.55,
-      "learning_rate": 8.725358851674642e-05,
-      "loss": 0.0832,
-      "step": 941
-    },
-    {
-      "epoch": 85.64,
-      "learning_rate": 8.723444976076556e-05,
-      "loss": 0.0772,
-      "step": 942
-    },
-    {
-      "epoch": 85.73,
-      "learning_rate": 8.72153110047847e-05,
-      "loss": 0.0599,
-      "step": 943
-    },
-    {
-      "epoch": 85.82,
-      "learning_rate": 8.719617224880383e-05,
-      "loss": 0.0712,
-      "step": 944
-    },
-    {
-      "epoch": 85.91,
-      "learning_rate": 8.717703349282297e-05,
-      "loss": 0.0664,
-      "step": 945
-    },
-    {
-      "epoch": 86.0,
-      "learning_rate": 8.71578947368421e-05,
-      "loss": 0.0655,
-      "step": 946
-    },
-    {
-      "epoch": 86.09,
-      "learning_rate": 8.713875598086125e-05,
-      "loss": 0.0684,
-      "step": 947
-    },
-    {
-      "epoch": 86.18,
-      "learning_rate": 8.71196172248804e-05,
-      "loss": 0.0624,
-      "step": 948
-    },
-    {
-      "epoch": 86.27,
-      "learning_rate": 8.710047846889953e-05,
-      "loss": 0.0764,
-      "step": 949
-    },
-    {
-      "epoch": 86.36,
-      "learning_rate": 8.708133971291866e-05,
-      "loss": 0.0688,
-      "step": 950
-    },
-    {
-      "epoch": 86.45,
-      "learning_rate": 8.70622009569378e-05,
-      "loss": 0.0604,
-      "step": 951
-    },
-    {
-      "epoch": 86.55,
-      "learning_rate": 8.704306220095694e-05,
-      "loss": 0.0709,
-      "step": 952
-    },
-    {
-      "epoch": 86.64,
-      "learning_rate": 8.702392344497608e-05,
-      "loss": 0.0633,
-      "step": 953
-    },
-    {
-      "epoch": 86.73,
-      "learning_rate": 8.700478468899523e-05,
-      "loss": 0.0697,
-      "step": 954
-    },
-    {
-      "epoch": 86.82,
-      "learning_rate": 8.698564593301436e-05,
-      "loss": 0.0616,
-      "step": 955
-    },
-    {
-      "epoch": 86.91,
-      "learning_rate": 8.696650717703349e-05,
-      "loss": 0.0618,
-      "step": 956
-    },
-    {
-      "epoch": 87.0,
-      "learning_rate": 8.694736842105264e-05,
-      "loss": 0.0712,
-      "step": 957
-    },
-    {
-      "epoch": 87.09,
-      "learning_rate": 8.692822966507177e-05,
-      "loss": 0.0664,
-      "step": 958
-    },
-    {
-      "epoch": 87.18,
-      "learning_rate": 8.690909090909091e-05,
-      "loss": 0.0788,
-      "step": 959
-    },
-    {
-      "epoch": 87.27,
-      "learning_rate": 8.688995215311006e-05,
-      "loss": 0.0713,
-      "step": 960
-    },
-    {
-      "epoch": 87.27,
-      "eval_accuracy_NAT": 0.9925719373556189,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9410033692433152,
-      "eval_accuracy_concrete tiles": 0.9769461260930716,
-      "eval_accuracy_dark bricks": 0.8435010199761364,
-      "eval_accuracy_grass": 0.8020704816561057,
-      "eval_accuracy_red bricks": 0.8290127840909091,
-      "eval_accuracy_road": 0.9683888622191159,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4094229867945855,
-      "eval_accuracy_speedway bricks": 0.986498146934628,
-      "eval_accuracy_steel": 0.8696326099861487,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9753546396575332,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8490141574564742,
-      "eval_iou_concrete tiles": 0.9267090112822297,
-      "eval_iou_dark bricks": 0.7975761546020308,
-      "eval_iou_grass": 0.7614784533246618,
-      "eval_iou_red bricks": 0.7922959443407432,
-      "eval_iou_road": 0.9560467634358925,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.27487599022728954,
-      "eval_iou_speedway bricks": 0.9699984099145389,
-      "eval_iou_steel": 0.8027215026103898,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10321125388145447,
-      "eval_mean_accuracy": 0.6630037172576642,
-      "eval_mean_iou": 0.6235439251424449,
-      "eval_overall_accuracy": 0.9717781278822157,
-      "eval_runtime": 3.8609,
-      "eval_samples_per_second": 9.324,
-      "eval_steps_per_second": 0.518,
-      "step": 960
-    },
-    {
-      "epoch": 87.36,
-      "learning_rate": 8.687081339712919e-05,
-      "loss": 0.0726,
-      "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.0635,
-      "step": 963
-    },
-    {
-      "epoch": 87.64,
-      "learning_rate": 8.68133971291866e-05,
-      "loss": 0.0591,
-      "step": 964
-    },
-    {
-      "epoch": 87.73,
-      "learning_rate": 8.679425837320575e-05,
-      "loss": 0.0583,
-      "step": 965
-    },
-    {
-      "epoch": 87.82,
-      "learning_rate": 8.677511961722489e-05,
-      "loss": 0.0625,
-      "step": 966
-    },
-    {
-      "epoch": 87.91,
-      "learning_rate": 8.675598086124403e-05,
-      "loss": 0.0617,
-      "step": 967
-    },
-    {
-      "epoch": 88.0,
-      "learning_rate": 8.673684210526316e-05,
-      "loss": 0.0622,
-      "step": 968
-    },
-    {
-      "epoch": 88.09,
-      "learning_rate": 8.67177033492823e-05,
-      "loss": 0.0699,
-      "step": 969
-    },
-    {
-      "epoch": 88.18,
-      "learning_rate": 8.669856459330143e-05,
-      "loss": 0.0687,
-      "step": 970
-    },
-    {
-      "epoch": 88.27,
-      "learning_rate": 8.667942583732058e-05,
-      "loss": 0.0628,
-      "step": 971
-    },
-    {
-      "epoch": 88.36,
-      "learning_rate": 8.666028708133973e-05,
-      "loss": 0.0758,
-      "step": 972
-    },
-    {
-      "epoch": 88.45,
-      "learning_rate": 8.664114832535886e-05,
-      "loss": 0.0659,
-      "step": 973
-    },
-    {
-      "epoch": 88.55,
-      "learning_rate": 8.662200956937799e-05,
-      "loss": 0.0641,
-      "step": 974
-    },
-    {
-      "epoch": 88.64,
-      "learning_rate": 8.660287081339714e-05,
-      "loss": 0.0611,
-      "step": 975
-    },
-    {
-      "epoch": 88.73,
-      "learning_rate": 8.658373205741627e-05,
-      "loss": 0.075,
-      "step": 976
-    },
-    {
-      "epoch": 88.82,
-      "learning_rate": 8.656459330143541e-05,
-      "loss": 0.0602,
-      "step": 977
-    },
-    {
-      "epoch": 88.91,
-      "learning_rate": 8.654545454545456e-05,
-      "loss": 0.0683,
-      "step": 978
-    },
-    {
-      "epoch": 89.0,
-      "learning_rate": 8.652631578947369e-05,
-      "loss": 0.0585,
-      "step": 979
-    },
-    {
-      "epoch": 89.09,
-      "learning_rate": 8.650717703349282e-05,
-      "loss": 0.0644,
-      "step": 980
-    },
-    {
-      "epoch": 89.09,
-      "eval_accuracy_NAT": 0.992146160811901,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9329263058500364,
-      "eval_accuracy_concrete tiles": 0.9787195010089892,
-      "eval_accuracy_dark bricks": 0.8584350101997613,
-      "eval_accuracy_grass": 0.8607178182351974,
-      "eval_accuracy_red bricks": 0.8837002840909091,
-      "eval_accuracy_road": 0.9775075061639421,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4111046784109393,
-      "eval_accuracy_speedway bricks": 0.9845165556999712,
-      "eval_accuracy_steel": 0.8909620077831277,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9756775651918609,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8626565794369342,
-      "eval_iou_concrete tiles": 0.9305232558139535,
-      "eval_iou_dark bricks": 0.8043203865988676,
-      "eval_iou_grass": 0.7930710223430224,
-      "eval_iou_red bricks": 0.8277066356228172,
-      "eval_iou_road": 0.9671216029325205,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2751037727146942,
-      "eval_iou_speedway bricks": 0.9699998097566311,
-      "eval_iou_steel": 0.7982198256758753,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09656993299722672,
-      "eval_mean_accuracy": 0.6746719867888289,
-      "eval_mean_iou": 0.6311077273913214,
-      "eval_overall_accuracy": 0.9734739727444119,
-      "eval_runtime": 3.8957,
-      "eval_samples_per_second": 9.241,
-      "eval_steps_per_second": 0.513,
-      "step": 980
-    },
-    {
-      "epoch": 89.18,
-      "learning_rate": 8.648803827751197e-05,
-      "loss": 0.0605,
-      "step": 981
-    },
-    {
-      "epoch": 89.27,
-      "learning_rate": 8.64688995215311e-05,
-      "loss": 0.0645,
-      "step": 982
-    },
-    {
-      "epoch": 89.36,
-      "learning_rate": 8.644976076555025e-05,
-      "loss": 0.0539,
-      "step": 983
-    },
-    {
-      "epoch": 89.45,
-      "learning_rate": 8.643062200956938e-05,
-      "loss": 0.0676,
-      "step": 984
-    },
-    {
-      "epoch": 89.55,
-      "learning_rate": 8.641148325358852e-05,
-      "loss": 0.0654,
-      "step": 985
-    },
-    {
-      "epoch": 89.64,
-      "learning_rate": 8.639234449760765e-05,
-      "loss": 0.0681,
-      "step": 986
-    },
-    {
-      "epoch": 89.73,
-      "learning_rate": 8.63732057416268e-05,
-      "loss": 0.065,
-      "step": 987
-    },
-    {
-      "epoch": 89.82,
-      "learning_rate": 8.635406698564593e-05,
-      "loss": 0.0629,
-      "step": 988
-    },
-    {
-      "epoch": 89.91,
-      "learning_rate": 8.633492822966508e-05,
-      "loss": 0.0795,
-      "step": 989
-    },
-    {
-      "epoch": 90.0,
-      "learning_rate": 8.631578947368421e-05,
-      "loss": 0.0646,
-      "step": 990
-    },
-    {
-      "epoch": 90.09,
-      "learning_rate": 8.629665071770336e-05,
-      "loss": 0.0717,
-      "step": 991
-    },
-    {
-      "epoch": 90.18,
-      "learning_rate": 8.627751196172249e-05,
-      "loss": 0.0585,
-      "step": 992
-    },
-    {
-      "epoch": 90.27,
-      "learning_rate": 8.625837320574163e-05,
-      "loss": 0.0663,
-      "step": 993
-    },
-    {
-      "epoch": 90.36,
-      "learning_rate": 8.623923444976078e-05,
-      "loss": 0.0726,
-      "step": 994
-    },
-    {
-      "epoch": 90.45,
-      "learning_rate": 8.622009569377991e-05,
-      "loss": 0.0661,
-      "step": 995
-    },
-    {
-      "epoch": 90.55,
-      "learning_rate": 8.620095693779904e-05,
-      "loss": 0.0708,
-      "step": 996
-    },
-    {
-      "epoch": 90.64,
-      "learning_rate": 8.618181818181819e-05,
-      "loss": 0.0631,
-      "step": 997
-    },
-    {
-      "epoch": 90.73,
-      "learning_rate": 8.616267942583732e-05,
-      "loss": 0.0721,
-      "step": 998
-    },
-    {
-      "epoch": 90.82,
-      "learning_rate": 8.614354066985647e-05,
-      "loss": 0.0656,
-      "step": 999
-    },
-    {
-      "epoch": 90.91,
-      "learning_rate": 8.612440191387561e-05,
-      "loss": 0.0585,
-      "step": 1000
-    },
-    {
-      "epoch": 90.91,
-      "eval_accuracy_NAT": 0.9915735769393919,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9356651573028886,
-      "eval_accuracy_concrete tiles": 0.9636763896532746,
-      "eval_accuracy_dark bricks": 0.8332242792810131,
-      "eval_accuracy_grass": 0.8753237312008724,
-      "eval_accuracy_red bricks": 0.8565340909090909,
-      "eval_accuracy_road": 0.9651336719924255,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.45215449507898436,
-      "eval_accuracy_speedway bricks": 0.9859484424234306,
-      "eval_accuracy_steel": 0.8990996636105798,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9764090900403279,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8608809936631147,
-      "eval_iou_concrete tiles": 0.9221991397723616,
-      "eval_iou_dark bricks": 0.7890075445566206,
-      "eval_iou_grass": 0.8054758657952966,
-      "eval_iou_red bricks": 0.8141772151898734,
-      "eval_iou_road": 0.9567346361449307,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.28836416063014275,
-      "eval_iou_speedway bricks": 0.9684689997718066,
-      "eval_iou_steel": 0.7975164000965357,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09861599653959274,
-      "eval_mean_accuracy": 0.6737179614147656,
-      "eval_mean_iou": 0.6291718496662315,
-      "eval_overall_accuracy": 0.9730810589260526,
-      "eval_runtime": 3.1995,
-      "eval_samples_per_second": 11.252,
-      "eval_steps_per_second": 0.625,
-      "step": 1000
-    },
-    {
-      "epoch": 91.0,
-      "learning_rate": 8.610526315789474e-05,
-      "loss": 0.0695,
-      "step": 1001
-    },
-    {
-      "epoch": 91.09,
-      "learning_rate": 8.608612440191388e-05,
-      "loss": 0.0631,
-      "step": 1002
-    },
-    {
-      "epoch": 91.18,
-      "learning_rate": 8.606698564593302e-05,
-      "loss": 0.0729,
-      "step": 1003
-    },
-    {
-      "epoch": 91.27,
-      "learning_rate": 8.604784688995215e-05,
-      "loss": 0.0683,
-      "step": 1004
-    },
-    {
-      "epoch": 91.36,
-      "learning_rate": 8.60287081339713e-05,
-      "loss": 0.063,
-      "step": 1005
-    },
-    {
-      "epoch": 91.45,
-      "learning_rate": 8.600956937799044e-05,
-      "loss": 0.0651,
-      "step": 1006
-    },
-    {
-      "epoch": 91.55,
-      "learning_rate": 8.599043062200958e-05,
-      "loss": 0.0618,
-      "step": 1007
-    },
-    {
-      "epoch": 91.64,
-      "learning_rate": 8.597129186602871e-05,
-      "loss": 0.0702,
-      "step": 1008
-    },
-    {
-      "epoch": 91.73,
-      "learning_rate": 8.595215311004785e-05,
-      "loss": 0.0563,
-      "step": 1009
-    },
-    {
-      "epoch": 91.82,
-      "learning_rate": 8.593301435406699e-05,
-      "loss": 0.0572,
-      "step": 1010
-    },
-    {
-      "epoch": 91.91,
-      "learning_rate": 8.591387559808612e-05,
-      "loss": 0.0575,
-      "step": 1011
-    },
-    {
-      "epoch": 92.0,
-      "learning_rate": 8.589473684210528e-05,
-      "loss": 0.073,
-      "step": 1012
-    },
-    {
-      "epoch": 92.09,
-      "learning_rate": 8.587559808612441e-05,
-      "loss": 0.0614,
-      "step": 1013
-    },
-    {
-      "epoch": 92.18,
-      "learning_rate": 8.585645933014354e-05,
-      "loss": 0.0667,
-      "step": 1014
-    },
-    {
-      "epoch": 92.27,
-      "learning_rate": 8.583732057416269e-05,
-      "loss": 0.0686,
-      "step": 1015
-    },
-    {
-      "epoch": 92.36,
-      "learning_rate": 8.581818181818182e-05,
-      "loss": 0.0609,
-      "step": 1016
-    },
-    {
-      "epoch": 92.45,
-      "learning_rate": 8.579904306220095e-05,
-      "loss": 0.0787,
-      "step": 1017
-    },
-    {
-      "epoch": 92.55,
-      "learning_rate": 8.577990430622011e-05,
-      "loss": 0.0663,
-      "step": 1018
-    },
-    {
-      "epoch": 92.64,
-      "learning_rate": 8.576076555023924e-05,
-      "loss": 0.0616,
-      "step": 1019
-    },
-    {
-      "epoch": 92.73,
-      "learning_rate": 8.574162679425837e-05,
-      "loss": 0.0686,
-      "step": 1020
-    },
-    {
-      "epoch": 92.73,
-      "eval_accuracy_NAT": 0.9922105654693635,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9391188999527876,
-      "eval_accuracy_concrete tiles": 0.9905521922583013,
-      "eval_accuracy_dark bricks": 0.8324929756360417,
-      "eval_accuracy_grass": 0.8400095066004467,
-      "eval_accuracy_red bricks": 0.8636363636363636,
-      "eval_accuracy_road": 0.979171377542815,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4167562649904888,
-      "eval_accuracy_speedway bricks": 0.9828040843467567,
-      "eval_accuracy_steel": 0.8998499439350967,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9762357140501005,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8621722726533909,
-      "eval_iou_concrete tiles": 0.8986186619327637,
-      "eval_iou_dark bricks": 0.7922710622710623,
-      "eval_iou_grass": 0.7878534573753188,
-      "eval_iou_red bricks": 0.8132419327871593,
-      "eval_iou_road": 0.9667541776652352,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2703906417686199,
-      "eval_iou_speedway bricks": 0.9695859206225329,
-      "eval_iou_steel": 0.7963125100322491,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09539979696273804,
-      "eval_mean_accuracy": 0.6720463211052662,
-      "eval_mean_iou": 0.6256489500891101,
-      "eval_overall_accuracy": 0.9733126958211263,
-      "eval_runtime": 3.526,
-      "eval_samples_per_second": 10.21,
-      "eval_steps_per_second": 0.567,
-      "step": 1020
-    },
-    {
-      "epoch": 92.82,
-      "learning_rate": 8.572248803827752e-05,
-      "loss": 0.0694,
-      "step": 1021
-    },
-    {
-      "epoch": 92.91,
-      "learning_rate": 8.570334928229665e-05,
-      "loss": 0.062,
-      "step": 1022
-    },
-    {
-      "epoch": 93.0,
-      "learning_rate": 8.568421052631578e-05,
-      "loss": 0.0611,
-      "step": 1023
-    },
-    {
-      "epoch": 93.09,
-      "learning_rate": 8.566507177033494e-05,
-      "loss": 0.07,
-      "step": 1024
-    },
-    {
-      "epoch": 93.18,
-      "learning_rate": 8.564593301435407e-05,
-      "loss": 0.054,
-      "step": 1025
-    },
-    {
-      "epoch": 93.27,
-      "learning_rate": 8.56267942583732e-05,
-      "loss": 0.0699,
-      "step": 1026
-    },
-    {
-      "epoch": 93.36,
-      "learning_rate": 8.560765550239235e-05,
-      "loss": 0.0682,
-      "step": 1027
-    },
-    {
-      "epoch": 93.45,
-      "learning_rate": 8.558851674641148e-05,
-      "loss": 0.0618,
-      "step": 1028
-    },
-    {
-      "epoch": 93.55,
-      "learning_rate": 8.556937799043062e-05,
-      "loss": 0.0626,
-      "step": 1029
-    },
-    {
-      "epoch": 93.64,
-      "learning_rate": 8.555023923444977e-05,
-      "loss": 0.0635,
-      "step": 1030
-    },
-    {
-      "epoch": 93.73,
-      "learning_rate": 8.55311004784689e-05,
-      "loss": 0.0639,
-      "step": 1031
-    },
-    {
-      "epoch": 93.82,
-      "learning_rate": 8.551196172248804e-05,
-      "loss": 0.0607,
-      "step": 1032
-    },
-    {
-      "epoch": 93.91,
-      "learning_rate": 8.549282296650718e-05,
-      "loss": 0.061,
-      "step": 1033
-    },
-    {
-      "epoch": 94.0,
-      "learning_rate": 8.547368421052632e-05,
-      "loss": 0.0622,
-      "step": 1034
-    },
-    {
-      "epoch": 94.09,
-      "learning_rate": 8.545454545454545e-05,
-      "loss": 0.0648,
-      "step": 1035
-    },
-    {
-      "epoch": 94.18,
-      "learning_rate": 8.543540669856461e-05,
-      "loss": 0.0664,
-      "step": 1036
-    },
-    {
-      "epoch": 94.27,
-      "learning_rate": 8.541626794258374e-05,
-      "loss": 0.0569,
-      "step": 1037
-    },
-    {
-      "epoch": 94.36,
-      "learning_rate": 8.539712918660287e-05,
-      "loss": 0.0702,
-      "step": 1038
-    },
-    {
-      "epoch": 94.45,
-      "learning_rate": 8.537799043062202e-05,
-      "loss": 0.0676,
-      "step": 1039
-    },
-    {
-      "epoch": 94.55,
-      "learning_rate": 8.535885167464115e-05,
-      "loss": 0.0553,
-      "step": 1040
-    },
-    {
-      "epoch": 94.55,
-      "eval_accuracy_NAT": 0.9927983241378775,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9316856410146358,
-      "eval_accuracy_concrete tiles": 0.9789641044456675,
-      "eval_accuracy_dark bricks": 0.8705592548400755,
-      "eval_accuracy_grass": 0.892540464215688,
-      "eval_accuracy_red bricks": 0.900390625,
-      "eval_accuracy_road": 0.9688592444621258,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.3991674248063298,
-      "eval_accuracy_speedway bricks": 0.9838697025320199,
-      "eval_accuracy_steel": 0.8924708132708924,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9764078532201541,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.862087795482272,
-      "eval_iou_concrete tiles": 0.9305394094396652,
-      "eval_iou_dark bricks": 0.8125157164924381,
-      "eval_iou_grass": 0.8160486489250198,
-      "eval_iou_red bricks": 0.8325398128386143,
-      "eval_iou_road": 0.9615446465645319,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2666138803469166,
-      "eval_iou_speedway bricks": 0.9705447655872862,
-      "eval_iou_steel": 0.8045337991006727,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09706210345029831,
-      "eval_mean_accuracy": 0.6777927383634855,
-      "eval_mean_iou": 0.6333366406151979,
-      "eval_overall_accuracy": 0.9738786485460069,
-      "eval_runtime": 3.445,
-      "eval_samples_per_second": 10.45,
-      "eval_steps_per_second": 0.581,
-      "step": 1040
-    },
-    {
-      "epoch": 94.64,
-      "learning_rate": 8.533971291866028e-05,
-      "loss": 0.0616,
-      "step": 1041
-    },
-    {
-      "epoch": 94.73,
-      "learning_rate": 8.532057416267944e-05,
-      "loss": 0.0608,
-      "step": 1042
-    },
-    {
-      "epoch": 94.82,
-      "learning_rate": 8.530143540669857e-05,
-      "loss": 0.0675,
-      "step": 1043
-    },
-    {
-      "epoch": 94.91,
-      "learning_rate": 8.52822966507177e-05,
-      "loss": 0.0613,
-      "step": 1044
-    },
-    {
-      "epoch": 95.0,
-      "learning_rate": 8.526315789473685e-05,
-      "loss": 0.0622,
-      "step": 1045
-    },
-    {
-      "epoch": 95.09,
-      "learning_rate": 8.524401913875598e-05,
-      "loss": 0.0659,
-      "step": 1046
-    },
-    {
-      "epoch": 95.18,
-      "learning_rate": 8.522488038277511e-05,
-      "loss": 0.0583,
-      "step": 1047
-    },
-    {
-      "epoch": 95.27,
-      "learning_rate": 8.520574162679427e-05,
-      "loss": 0.0561,
-      "step": 1048
-    },
-    {
-      "epoch": 95.36,
-      "learning_rate": 8.51866028708134e-05,
-      "loss": 0.0567,
-      "step": 1049
-    },
-    {
-      "epoch": 95.45,
-      "learning_rate": 8.516746411483254e-05,
-      "loss": 0.0547,
-      "step": 1050
-    },
-    {
-      "epoch": 95.55,
-      "learning_rate": 8.514832535885168e-05,
-      "loss": 0.0612,
-      "step": 1051
-    },
-    {
-      "epoch": 95.64,
-      "learning_rate": 8.512918660287081e-05,
-      "loss": 0.0642,
-      "step": 1052
-    },
-    {
-      "epoch": 95.73,
-      "learning_rate": 8.511004784688995e-05,
-      "loss": 0.0635,
-      "step": 1053
-    },
-    {
-      "epoch": 95.82,
-      "learning_rate": 8.50909090909091e-05,
-      "loss": 0.0677,
-      "step": 1054
-    },
-    {
-      "epoch": 95.91,
-      "learning_rate": 8.507177033492824e-05,
-      "loss": 0.0592,
-      "step": 1055
-    },
-    {
-      "epoch": 96.0,
-      "learning_rate": 8.505263157894737e-05,
-      "loss": 0.0656,
-      "step": 1056
-    },
-    {
-      "epoch": 96.09,
-      "learning_rate": 8.503349282296651e-05,
-      "loss": 0.0675,
-      "step": 1057
-    },
-    {
-      "epoch": 96.18,
-      "learning_rate": 8.501435406698565e-05,
-      "loss": 0.0604,
-      "step": 1058
-    },
-    {
-      "epoch": 96.27,
-      "learning_rate": 8.499521531100478e-05,
-      "loss": 0.0575,
-      "step": 1059
-    },
-    {
-      "epoch": 96.36,
-      "learning_rate": 8.497607655502394e-05,
-      "loss": 0.072,
-      "step": 1060
-    },
-    {
-      "epoch": 96.36,
-      "eval_accuracy_NAT": 0.9920134695724167,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9422333040044637,
-      "eval_accuracy_concrete tiles": 0.9773436066776738,
-      "eval_accuracy_dark bricks": 0.8566259959200955,
-      "eval_accuracy_grass": 0.8643596850239588,
-      "eval_accuracy_red bricks": 0.8987926136363636,
-      "eval_accuracy_road": 0.9613561035421355,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.42147051525928375,
-      "eval_accuracy_speedway bricks": 0.9820830120186766,
-      "eval_accuracy_steel": 0.8991656223204274,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9768918185833649,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.863907079129082,
-      "eval_iou_concrete tiles": 0.9230702590314477,
-      "eval_iou_dark bricks": 0.8076937035020867,
-      "eval_iou_grass": 0.8003572850221847,
-      "eval_iou_red bricks": 0.8032370675975881,
-      "eval_iou_road": 0.9518501136192211,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2746528214433286,
-      "eval_iou_speedway bricks": 0.9699589205760059,
-      "eval_iou_steel": 0.7998621155002714,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.097042515873909,
-      "eval_mean_accuracy": 0.6765726098442688,
-      "eval_mean_iou": 0.62857547569266,
-      "eval_overall_accuracy": 0.9725418090820312,
-      "eval_runtime": 3.5177,
-      "eval_samples_per_second": 10.234,
-      "eval_steps_per_second": 0.569,
-      "step": 1060
-    },
-    {
-      "epoch": 96.45,
-      "learning_rate": 8.495693779904307e-05,
-      "loss": 0.058,
-      "step": 1061
-    },
-    {
-      "epoch": 96.55,
-      "learning_rate": 8.49377990430622e-05,
-      "loss": 0.0657,
-      "step": 1062
-    },
-    {
-      "epoch": 96.64,
-      "learning_rate": 8.491866028708135e-05,
-      "loss": 0.0584,
-      "step": 1063
-    },
-    {
-      "epoch": 96.73,
-      "learning_rate": 8.489952153110048e-05,
-      "loss": 0.0654,
-      "step": 1064
-    },
-    {
-      "epoch": 96.82,
-      "learning_rate": 8.488038277511963e-05,
-      "loss": 0.0551,
-      "step": 1065
-    },
-    {
-      "epoch": 96.91,
-      "learning_rate": 8.486124401913876e-05,
-      "loss": 0.0622,
-      "step": 1066
-    },
-    {
-      "epoch": 97.0,
-      "learning_rate": 8.48421052631579e-05,
-      "loss": 0.06,
-      "step": 1067
-    },
-    {
-      "epoch": 97.09,
-      "learning_rate": 8.482296650717703e-05,
-      "loss": 0.0632,
-      "step": 1068
-    },
-    {
-      "epoch": 97.18,
-      "learning_rate": 8.480382775119618e-05,
-      "loss": 0.0695,
-      "step": 1069
-    },
-    {
-      "epoch": 97.27,
-      "learning_rate": 8.478468899521531e-05,
-      "loss": 0.0584,
-      "step": 1070
-    },
-    {
-      "epoch": 97.36,
-      "learning_rate": 8.476555023923446e-05,
-      "loss": 0.0567,
-      "step": 1071
-    },
-    {
-      "epoch": 97.45,
-      "learning_rate": 8.474641148325359e-05,
-      "loss": 0.0638,
-      "step": 1072
-    },
-    {
-      "epoch": 97.55,
-      "learning_rate": 8.472727272727274e-05,
-      "loss": 0.0597,
-      "step": 1073
-    },
-    {
-      "epoch": 97.64,
-      "learning_rate": 8.470813397129187e-05,
-      "loss": 0.0569,
-      "step": 1074
-    },
-    {
-      "epoch": 97.73,
-      "learning_rate": 8.468899521531101e-05,
-      "loss": 0.059,
-      "step": 1075
-    },
-    {
-      "epoch": 97.82,
-      "learning_rate": 8.466985645933014e-05,
-      "loss": 0.0567,
-      "step": 1076
-    },
-    {
-      "epoch": 97.91,
-      "learning_rate": 8.465071770334929e-05,
-      "loss": 0.0477,
-      "step": 1077
-    },
-    {
-      "epoch": 98.0,
-      "learning_rate": 8.463157894736842e-05,
-      "loss": 0.0636,
-      "step": 1078
-    },
-    {
-      "epoch": 98.09,
-      "learning_rate": 8.461244019138757e-05,
-      "loss": 0.063,
-      "step": 1079
-    },
-    {
-      "epoch": 98.18,
-      "learning_rate": 8.45933014354067e-05,
-      "loss": 0.0561,
-      "step": 1080
-    },
-    {
-      "epoch": 98.18,
-      "eval_accuracy_NAT": 0.9914687649762886,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9443511524099747,
-      "eval_accuracy_concrete tiles": 0.9741331865712713,
-      "eval_accuracy_dark bricks": 0.8606289211346754,
-      "eval_accuracy_grass": 0.8585333971766794,
-      "eval_accuracy_red bricks": 0.9298650568181818,
-      "eval_accuracy_road": 0.9737093811631864,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4085683566288975,
-      "eval_accuracy_speedway bricks": 0.9860160240976942,
-      "eval_accuracy_steel": 0.8814392190488755,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.976708252946885,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8619977032265015,
-      "eval_iou_concrete tiles": 0.9324787075248047,
-      "eval_iou_dark bricks": 0.8081246159962413,
-      "eval_iou_grass": 0.7956756931847629,
-      "eval_iou_red bricks": 0.7989321128909229,
-      "eval_iou_road": 0.964323693828769,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.26915602695192603,
-      "eval_iou_speedway bricks": 0.9706940602217043,
-      "eval_iou_steel": 0.805782507763273,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09496220201253891,
-      "eval_mean_accuracy": 0.677593343078902,
-      "eval_mean_iou": 0.6295287211181376,
-      "eval_overall_accuracy": 0.973719596862793,
-      "eval_runtime": 3.5692,
-      "eval_samples_per_second": 10.086,
-      "eval_steps_per_second": 0.56,
-      "step": 1080
-    },
-    {
-      "epoch": 98.27,
-      "learning_rate": 8.457416267942585e-05,
-      "loss": 0.0619,
-      "step": 1081
-    },
-    {
-      "epoch": 98.36,
-      "learning_rate": 8.455502392344498e-05,
-      "loss": 0.0603,
-      "step": 1082
-    },
-    {
-      "epoch": 98.45,
-      "learning_rate": 8.453588516746412e-05,
-      "loss": 0.0642,
-      "step": 1083
-    },
-    {
-      "epoch": 98.55,
-      "learning_rate": 8.451674641148326e-05,
-      "loss": 0.0627,
-      "step": 1084
-    },
-    {
-      "epoch": 98.64,
-      "learning_rate": 8.44976076555024e-05,
-      "loss": 0.0538,
-      "step": 1085
-    },
-    {
-      "epoch": 98.73,
-      "learning_rate": 8.447846889952153e-05,
-      "loss": 0.0597,
-      "step": 1086
-    },
-    {
-      "epoch": 98.82,
-      "learning_rate": 8.445933014354066e-05,
-      "loss": 0.0652,
-      "step": 1087
-    },
-    {
-      "epoch": 98.91,
-      "learning_rate": 8.444019138755981e-05,
-      "loss": 0.0527,
-      "step": 1088
-    },
-    {
-      "epoch": 99.0,
-      "learning_rate": 8.442105263157896e-05,
-      "loss": 0.06,
-      "step": 1089
-    },
-    {
-      "epoch": 99.09,
-      "learning_rate": 8.440191387559809e-05,
-      "loss": 0.0629,
-      "step": 1090
-    },
-    {
-      "epoch": 99.18,
-      "learning_rate": 8.438277511961723e-05,
-      "loss": 0.0614,
-      "step": 1091
-    },
-    {
-      "epoch": 99.27,
-      "learning_rate": 8.436363636363637e-05,
-      "loss": 0.0547,
-      "step": 1092
-    },
-    {
-      "epoch": 99.36,
-      "learning_rate": 8.43444976076555e-05,
-      "loss": 0.058,
-      "step": 1093
-    },
-    {
-      "epoch": 99.45,
-      "learning_rate": 8.432535885167464e-05,
-      "loss": 0.06,
-      "step": 1094
-    },
-    {
-      "epoch": 99.55,
-      "learning_rate": 8.430622009569379e-05,
-      "loss": 0.0618,
-      "step": 1095
-    },
-    {
-      "epoch": 99.64,
-      "learning_rate": 8.428708133971292e-05,
-      "loss": 0.0584,
-      "step": 1096
-    },
-    {
-      "epoch": 99.73,
-      "learning_rate": 8.426794258373207e-05,
-      "loss": 0.0541,
-      "step": 1097
-    },
-    {
-      "epoch": 99.82,
-      "learning_rate": 8.42488038277512e-05,
-      "loss": 0.0611,
-      "step": 1098
-    },
-    {
-      "epoch": 99.91,
-      "learning_rate": 8.422966507177033e-05,
-      "loss": 0.0627,
-      "step": 1099
-    },
-    {
-      "epoch": 100.0,
-      "learning_rate": 8.421052631578948e-05,
-      "loss": 0.0611,
-      "step": 1100
-    },
-    {
-      "epoch": 100.0,
-      "eval_accuracy_NAT": 0.9910858660538402,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9463415811837418,
-      "eval_accuracy_concrete tiles": 0.9766403717972237,
-      "eval_accuracy_dark bricks": 0.8566644855856203,
-      "eval_accuracy_grass": 0.8698714162388114,
-      "eval_accuracy_red bricks": 0.875,
-      "eval_accuracy_road": 0.9717468351981349,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.38778154550216415,
-      "eval_accuracy_speedway bricks": 0.9856129476833364,
-      "eval_accuracy_steel": 0.8766489677461908,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.976963633637572,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8580159986769109,
-      "eval_iou_concrete tiles": 0.9323953529102691,
-      "eval_iou_dark bricks": 0.8081699346405229,
-      "eval_iou_grass": 0.7921480632738153,
-      "eval_iou_red bricks": 0.8141417478936065,
-      "eval_iou_road": 0.9652340292132645,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2592811059907834,
-      "eval_iou_speedway bricks": 0.9710863859171426,
-      "eval_iou_steel": 0.8052086725382245,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09601221978664398,
-      "eval_mean_accuracy": 0.6721072320760817,
-      "eval_mean_iou": 0.6294342249763163,
-      "eval_overall_accuracy": 0.9735714594523112,
-      "eval_runtime": 3.2239,
-      "eval_samples_per_second": 11.167,
-      "eval_steps_per_second": 0.62,
-      "step": 1100
-    },
-    {
-      "epoch": 100.09,
-      "learning_rate": 8.419138755980862e-05,
-      "loss": 0.0626,
-      "step": 1101
-    },
-    {
-      "epoch": 100.18,
-      "learning_rate": 8.417224880382775e-05,
-      "loss": 0.0591,
-      "step": 1102
-    },
-    {
-      "epoch": 100.27,
-      "learning_rate": 8.41531100478469e-05,
-      "loss": 0.0612,
-      "step": 1103
-    },
-    {
-      "epoch": 100.36,
-      "learning_rate": 8.413397129186603e-05,
-      "loss": 0.0556,
-      "step": 1104
-    },
-    {
-      "epoch": 100.45,
-      "learning_rate": 8.411483253588516e-05,
-      "loss": 0.0558,
-      "step": 1105
-    },
-    {
-      "epoch": 100.55,
-      "learning_rate": 8.409569377990431e-05,
-      "loss": 0.0663,
-      "step": 1106
-    },
-    {
-      "epoch": 100.64,
-      "learning_rate": 8.407655502392345e-05,
-      "loss": 0.0497,
-      "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.0543,
-      "step": 1109
-    },
-    {
-      "epoch": 100.91,
-      "learning_rate": 8.401913875598086e-05,
-      "loss": 0.0569,
-      "step": 1110
-    },
-    {
-      "epoch": 101.0,
-      "learning_rate": 8.4e-05,
-      "loss": 0.057,
-      "step": 1111
-    },
-    {
-      "epoch": 101.09,
-      "learning_rate": 8.398086124401914e-05,
-      "loss": 0.0554,
-      "step": 1112
-    },
-    {
-      "epoch": 101.18,
-      "learning_rate": 8.396172248803829e-05,
-      "loss": 0.0514,
-      "step": 1113
-    },
-    {
-      "epoch": 101.27,
-      "learning_rate": 8.394258373205742e-05,
-      "loss": 0.0544,
-      "step": 1114
-    },
-    {
-      "epoch": 101.36,
-      "learning_rate": 8.392344497607656e-05,
-      "loss": 0.0653,
-      "step": 1115
-    },
-    {
-      "epoch": 101.45,
-      "learning_rate": 8.39043062200957e-05,
-      "loss": 0.0686,
-      "step": 1116
-    },
-    {
-      "epoch": 101.55,
-      "learning_rate": 8.388516746411483e-05,
-      "loss": 0.0624,
-      "step": 1117
-    },
-    {
-      "epoch": 101.64,
-      "learning_rate": 8.386602870813397e-05,
-      "loss": 0.0625,
-      "step": 1118
-    },
-    {
-      "epoch": 101.73,
-      "learning_rate": 8.384688995215312e-05,
-      "loss": 0.0552,
-      "step": 1119
-    },
-    {
-      "epoch": 101.82,
-      "learning_rate": 8.382775119617225e-05,
-      "loss": 0.067,
-      "step": 1120
-    },
-    {
-      "epoch": 101.82,
-      "eval_accuracy_NAT": 0.9918479584252937,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9381049079359629,
-      "eval_accuracy_concrete tiles": 0.9845899834892681,
-      "eval_accuracy_dark bricks": 0.8498903044532543,
-      "eval_accuracy_grass": 0.8775640734384884,
-      "eval_accuracy_red bricks": 0.9094460227272727,
-      "eval_accuracy_road": 0.9424948638854057,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5554820389821631,
-      "eval_accuracy_speedway bricks": 0.9870707809424506,
-      "eval_accuracy_steel": 0.878067080007915,
-      "eval_accuracy_tiles": 0.11162854403506368,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9770867265436101,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8687984756112709,
-      "eval_iou_concrete tiles": 0.9298068316345682,
-      "eval_iou_dark bricks": 0.8044959376252414,
-      "eval_iou_grass": 0.816391161284449,
-      "eval_iou_red bricks": 0.82893672115229,
-      "eval_iou_road": 0.9332301241503501,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3561844826671852,
-      "eval_iou_speedway bricks": 0.9707221548038323,
-      "eval_iou_steel": 0.8053646105099178,
-      "eval_iou_tiles": 0.11162854403506368,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09573851525783539,
-      "eval_mean_accuracy": 0.6943220429478877,
-      "eval_mean_iou": 0.6463573669244446,
-      "eval_overall_accuracy": 0.9721692403157552,
-      "eval_runtime": 3.2876,
-      "eval_samples_per_second": 10.95,
-      "eval_steps_per_second": 0.608,
-      "step": 1120
-    },
-    {
-      "epoch": 101.91,
-      "learning_rate": 8.38086124401914e-05,
-      "loss": 0.0595,
-      "step": 1121
-    },
-    {
-      "epoch": 102.0,
-      "learning_rate": 8.378947368421053e-05,
-      "loss": 0.0646,
-      "step": 1122
-    },
-    {
-      "epoch": 102.09,
-      "learning_rate": 8.377033492822966e-05,
-      "loss": 0.0552,
-      "step": 1123
-    },
-    {
-      "epoch": 102.18,
-      "learning_rate": 8.37511961722488e-05,
-      "loss": 0.0578,
-      "step": 1124
-    },
-    {
-      "epoch": 102.27,
-      "learning_rate": 8.373205741626795e-05,
-      "loss": 0.0716,
-      "step": 1125
-    },
-    {
-      "epoch": 102.36,
-      "learning_rate": 8.371291866028708e-05,
-      "loss": 0.0558,
-      "step": 1126
-    },
-    {
-      "epoch": 102.45,
-      "learning_rate": 8.369377990430623e-05,
-      "loss": 0.0521,
-      "step": 1127
-    },
-    {
-      "epoch": 102.55,
-      "learning_rate": 8.367464114832536e-05,
-      "loss": 0.0548,
-      "step": 1128
-    },
-    {
-      "epoch": 102.64,
-      "learning_rate": 8.36555023923445e-05,
-      "loss": 0.0567,
-      "step": 1129
-    },
-    {
-      "epoch": 102.73,
-      "learning_rate": 8.363636363636364e-05,
-      "loss": 0.0644,
-      "step": 1130
-    },
-    {
-      "epoch": 102.82,
-      "learning_rate": 8.361722488038278e-05,
-      "loss": 0.0683,
-      "step": 1131
-    },
-    {
-      "epoch": 102.91,
-      "learning_rate": 8.359808612440192e-05,
-      "loss": 0.0505,
-      "step": 1132
-    },
-    {
-      "epoch": 103.0,
-      "learning_rate": 8.357894736842106e-05,
-      "loss": 0.0656,
-      "step": 1133
-    },
-    {
-      "epoch": 103.09,
-      "learning_rate": 8.35598086124402e-05,
-      "loss": 0.0605,
-      "step": 1134
-    },
-    {
-      "epoch": 103.18,
-      "learning_rate": 8.354066985645933e-05,
-      "loss": 0.0691,
-      "step": 1135
-    },
-    {
-      "epoch": 103.27,
-      "learning_rate": 8.352153110047849e-05,
-      "loss": 0.0603,
-      "step": 1136
-    },
-    {
-      "epoch": 103.36,
-      "learning_rate": 8.350239234449762e-05,
-      "loss": 0.0528,
-      "step": 1137
-    },
-    {
-      "epoch": 103.45,
-      "learning_rate": 8.348325358851675e-05,
-      "loss": 0.067,
-      "step": 1138
-    },
-    {
-      "epoch": 103.55,
-      "learning_rate": 8.34641148325359e-05,
-      "loss": 0.0554,
-      "step": 1139
-    },
-    {
-      "epoch": 103.64,
-      "learning_rate": 8.344497607655503e-05,
-      "loss": 0.0582,
-      "step": 1140
-    },
-    {
-      "epoch": 103.64,
-      "eval_accuracy_NAT": 0.9905542188403209,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.945705824284304,
-      "eval_accuracy_concrete tiles": 0.9711673699015471,
-      "eval_accuracy_dark bricks": 0.8490435318117086,
-      "eval_accuracy_grass": 0.8863996197359821,
-      "eval_accuracy_red bricks": 0.8884943181818182,
-      "eval_accuracy_road": 0.973118078035084,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4749538224023378,
-      "eval_accuracy_speedway bricks": 0.9845153488843593,
-      "eval_accuracy_steel": 0.8883648835828771,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9772139612140531,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8706800211900773,
-      "eval_iou_concrete tiles": 0.9371275151944297,
-      "eval_iou_dark bricks": 0.8044271023265991,
-      "eval_iou_grass": 0.80607574659598,
-      "eval_iou_red bricks": 0.8093158660844251,
-      "eval_iou_road": 0.96071111790246,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31404717634620294,
-      "eval_iou_speedway bricks": 0.9711573888618725,
-      "eval_iou_steel": 0.8071011235955056,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09244033694267273,
-      "eval_mean_accuracy": 0.6809474627431029,
-      "eval_mean_iou": 0.6352197707162774,
-      "eval_overall_accuracy": 0.974084218343099,
-      "eval_runtime": 3.3208,
-      "eval_samples_per_second": 10.841,
-      "eval_steps_per_second": 0.602,
-      "step": 1140
-    },
-    {
-      "epoch": 103.73,
-      "learning_rate": 8.342583732057416e-05,
-      "loss": 0.0538,
-      "step": 1141
-    },
-    {
-      "epoch": 103.82,
-      "learning_rate": 8.34066985645933e-05,
-      "loss": 0.0633,
-      "step": 1142
-    },
-    {
-      "epoch": 103.91,
-      "learning_rate": 8.338755980861245e-05,
-      "loss": 0.0585,
-      "step": 1143
-    },
-    {
-      "epoch": 104.0,
-      "learning_rate": 8.336842105263158e-05,
-      "loss": 0.0593,
-      "step": 1144
-    },
-    {
-      "epoch": 104.09,
-      "learning_rate": 8.334928229665073e-05,
-      "loss": 0.0571,
-      "step": 1145
-    },
-    {
-      "epoch": 104.18,
-      "learning_rate": 8.333014354066986e-05,
-      "loss": 0.0583,
-      "step": 1146
-    },
-    {
-      "epoch": 104.27,
-      "learning_rate": 8.331100478468899e-05,
-      "loss": 0.0592,
-      "step": 1147
-    },
-    {
-      "epoch": 104.36,
-      "learning_rate": 8.329186602870814e-05,
-      "loss": 0.057,
-      "step": 1148
-    },
-    {
-      "epoch": 104.45,
-      "learning_rate": 8.327272727272728e-05,
-      "loss": 0.0519,
-      "step": 1149
-    },
-    {
-      "epoch": 104.55,
-      "learning_rate": 8.325358851674641e-05,
-      "loss": 0.0584,
-      "step": 1150
-    },
-    {
-      "epoch": 104.64,
-      "learning_rate": 8.323444976076556e-05,
-      "loss": 0.0524,
-      "step": 1151
-    },
-    {
-      "epoch": 104.73,
-      "learning_rate": 8.321531100478469e-05,
-      "loss": 0.0586,
-      "step": 1152
-    },
-    {
-      "epoch": 104.82,
-      "learning_rate": 8.319617224880382e-05,
-      "loss": 0.0558,
-      "step": 1153
-    },
-    {
-      "epoch": 104.91,
-      "learning_rate": 8.317703349282297e-05,
-      "loss": 0.0542,
-      "step": 1154
-    },
-    {
-      "epoch": 105.0,
-      "learning_rate": 8.315789473684212e-05,
-      "loss": 0.0715,
-      "step": 1155
-    },
-    {
-      "epoch": 105.09,
-      "learning_rate": 8.313875598086125e-05,
-      "loss": 0.0552,
-      "step": 1156
-    },
-    {
-      "epoch": 105.18,
-      "learning_rate": 8.311961722488039e-05,
-      "loss": 0.0584,
-      "step": 1157
-    },
-    {
-      "epoch": 105.27,
-      "learning_rate": 8.310047846889952e-05,
-      "loss": 0.0579,
-      "step": 1158
-    },
-    {
-      "epoch": 105.36,
-      "learning_rate": 8.308133971291866e-05,
-      "loss": 0.0689,
-      "step": 1159
-    },
-    {
-      "epoch": 105.45,
-      "learning_rate": 8.30622009569378e-05,
-      "loss": 0.0593,
-      "step": 1160
-    },
-    {
-      "epoch": 105.45,
-      "eval_accuracy_NAT": 0.9921701581637227,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9405231984205331,
-      "eval_accuracy_concrete tiles": 0.9787195010089892,
-      "eval_accuracy_dark bricks": 0.8588583965205342,
-      "eval_accuracy_grass": 0.8798428614867344,
-      "eval_accuracy_red bricks": 0.9325284090909091,
-      "eval_accuracy_road": 0.9819428842291354,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5258456703333058,
-      "eval_accuracy_speedway bricks": 0.9832331072967693,
-      "eval_accuracy_steel": 0.8870704439021173,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9771828050160812,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8716996940702333,
-      "eval_iou_concrete tiles": 0.9406958974961797,
-      "eval_iou_dark bricks": 0.8108874191438331,
-      "eval_iou_grass": 0.8177028519456896,
-      "eval_iou_red bricks": 0.8159080316917818,
-      "eval_iou_road": 0.9683148387958399,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.34184020932649917,
-      "eval_iou_speedway bricks": 0.9714549225652176,
-      "eval_iou_steel": 0.812743616860553,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08860820531845093,
-      "eval_mean_accuracy": 0.6892872792655962,
-      "eval_mean_iou": 0.6406484836086085,
-      "eval_overall_accuracy": 0.9752529991997613,
-      "eval_runtime": 3.8142,
-      "eval_samples_per_second": 9.439,
-      "eval_steps_per_second": 0.524,
-      "step": 1160
-    },
-    {
-      "epoch": 105.55,
-      "learning_rate": 8.304306220095695e-05,
-      "loss": 0.0565,
-      "step": 1161
-    },
-    {
-      "epoch": 105.64,
-      "learning_rate": 8.302392344497608e-05,
-      "loss": 0.0448,
-      "step": 1162
-    },
-    {
-      "epoch": 105.73,
-      "learning_rate": 8.300478468899521e-05,
-      "loss": 0.057,
-      "step": 1163
-    },
-    {
-      "epoch": 105.82,
-      "learning_rate": 8.298564593301436e-05,
-      "loss": 0.0719,
-      "step": 1164
-    },
-    {
-      "epoch": 105.91,
-      "learning_rate": 8.296650717703349e-05,
-      "loss": 0.0577,
-      "step": 1165
-    },
-    {
-      "epoch": 106.0,
-      "learning_rate": 8.294736842105263e-05,
-      "loss": 0.0588,
-      "step": 1166
-    },
-    {
-      "epoch": 106.09,
-      "learning_rate": 8.292822966507178e-05,
-      "loss": 0.0607,
-      "step": 1167
-    },
-    {
-      "epoch": 106.18,
-      "learning_rate": 8.290909090909091e-05,
-      "loss": 0.0532,
-      "step": 1168
-    },
-    {
-      "epoch": 106.27,
-      "learning_rate": 8.288995215311004e-05,
-      "loss": 0.0618,
-      "step": 1169
-    },
-    {
-      "epoch": 106.36,
-      "learning_rate": 8.287081339712919e-05,
-      "loss": 0.0573,
-      "step": 1170
-    },
-    {
-      "epoch": 106.45,
-      "learning_rate": 8.285167464114832e-05,
-      "loss": 0.0533,
-      "step": 1171
-    },
-    {
-      "epoch": 106.55,
-      "learning_rate": 8.283253588516747e-05,
-      "loss": 0.0567,
-      "step": 1172
-    },
-    {
-      "epoch": 106.64,
-      "learning_rate": 8.281339712918661e-05,
-      "loss": 0.0614,
-      "step": 1173
-    },
-    {
-      "epoch": 106.73,
-      "learning_rate": 8.279425837320575e-05,
-      "loss": 0.0574,
-      "step": 1174
-    },
-    {
-      "epoch": 106.82,
-      "learning_rate": 8.277511961722488e-05,
-      "loss": 0.0578,
-      "step": 1175
-    },
-    {
-      "epoch": 106.91,
-      "learning_rate": 8.275598086124402e-05,
-      "loss": 0.054,
-      "step": 1176
-    },
-    {
-      "epoch": 107.0,
-      "learning_rate": 8.273684210526315e-05,
-      "loss": 0.0631,
-      "step": 1177
-    },
-    {
-      "epoch": 107.09,
-      "learning_rate": 8.27177033492823e-05,
-      "loss": 0.0583,
-      "step": 1178
-    },
-    {
-      "epoch": 107.18,
-      "learning_rate": 8.269856459330145e-05,
-      "loss": 0.0523,
-      "step": 1179
-    },
-    {
-      "epoch": 107.27,
-      "learning_rate": 8.267942583732058e-05,
-      "loss": 0.0548,
-      "step": 1180
-    },
-    {
-      "epoch": 107.27,
-      "eval_accuracy_NAT": 0.9917689083251754,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9454992703549508,
-      "eval_accuracy_concrete tiles": 0.9789946798752522,
-      "eval_accuracy_dark bricks": 0.8647088256803048,
-      "eval_accuracy_grass": 0.8888985974269268,
-      "eval_accuracy_red bricks": 0.8595525568181818,
-      "eval_accuracy_road": 0.9207327321953066,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4293000303255865,
-      "eval_accuracy_speedway bricks": 0.9853927038341739,
-      "eval_accuracy_steel": 0.8736066222544687,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9777837157125583,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8702781447142628,
-      "eval_iou_concrete tiles": 0.9342612044817927,
-      "eval_iou_dark bricks": 0.8105202395555234,
-      "eval_iou_grass": 0.8139648015566942,
-      "eval_iou_red bricks": 0.7992405481261351,
-      "eval_iou_road": 0.9118740726281136,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2855832890127827,
-      "eval_iou_speedway bricks": 0.9715937317608198,
-      "eval_iou_steel": 0.8125987392057917,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11004399508237839,
-      "eval_mean_accuracy": 0.6721888405454097,
-      "eval_mean_iou": 0.629822960519575,
-      "eval_overall_accuracy": 0.9701244566175673,
-      "eval_runtime": 3.3874,
-      "eval_samples_per_second": 10.628,
-      "eval_steps_per_second": 0.59,
-      "step": 1180
-    },
-    {
-      "epoch": 107.36,
-      "learning_rate": 8.266028708133971e-05,
-      "loss": 0.0551,
-      "step": 1181
-    },
-    {
-      "epoch": 107.45,
-      "learning_rate": 8.264114832535886e-05,
-      "loss": 0.0636,
-      "step": 1182
-    },
-    {
-      "epoch": 107.55,
-      "learning_rate": 8.262200956937799e-05,
-      "loss": 0.0506,
-      "step": 1183
-    },
-    {
-      "epoch": 107.64,
-      "learning_rate": 8.260287081339713e-05,
-      "loss": 0.056,
-      "step": 1184
-    },
-    {
-      "epoch": 107.73,
-      "learning_rate": 8.258373205741628e-05,
-      "loss": 0.0615,
-      "step": 1185
-    },
-    {
-      "epoch": 107.82,
-      "learning_rate": 8.256459330143541e-05,
-      "loss": 0.0671,
-      "step": 1186
-    },
-    {
-      "epoch": 107.91,
-      "learning_rate": 8.254545454545454e-05,
-      "loss": 0.0529,
-      "step": 1187
-    },
-    {
-      "epoch": 108.0,
-      "learning_rate": 8.252631578947369e-05,
-      "loss": 0.0571,
-      "step": 1188
-    },
-    {
-      "epoch": 108.09,
-      "learning_rate": 8.250717703349282e-05,
-      "loss": 0.0609,
-      "step": 1189
-    },
-    {
-      "epoch": 108.18,
-      "learning_rate": 8.248803827751197e-05,
-      "loss": 0.0677,
-      "step": 1190
-    },
-    {
-      "epoch": 108.27,
-      "learning_rate": 8.246889952153111e-05,
-      "loss": 0.05,
-      "step": 1191
-    },
-    {
-      "epoch": 108.36,
-      "learning_rate": 8.244976076555024e-05,
-      "loss": 0.0584,
-      "step": 1192
-    },
-    {
-      "epoch": 108.45,
-      "learning_rate": 8.243062200956938e-05,
-      "loss": 0.0537,
-      "step": 1193
-    },
-    {
-      "epoch": 108.55,
-      "learning_rate": 8.241148325358852e-05,
-      "loss": 0.0528,
-      "step": 1194
-    },
-    {
-      "epoch": 108.64,
-      "learning_rate": 8.239234449760765e-05,
-      "loss": 0.0561,
-      "step": 1195
-    },
-    {
-      "epoch": 108.73,
-      "learning_rate": 8.23732057416268e-05,
-      "loss": 0.057,
-      "step": 1196
-    },
-    {
-      "epoch": 108.82,
-      "learning_rate": 8.235406698564594e-05,
-      "loss": 0.0642,
-      "step": 1197
-    },
-    {
-      "epoch": 108.91,
-      "learning_rate": 8.233492822966508e-05,
-      "loss": 0.0633,
-      "step": 1198
-    },
-    {
-      "epoch": 109.0,
-      "learning_rate": 8.231578947368421e-05,
-      "loss": 0.0531,
-      "step": 1199
-    },
-    {
-      "epoch": 109.09,
-      "learning_rate": 8.229665071770335e-05,
-      "loss": 0.0593,
-      "step": 1200
-    },
-    {
-      "epoch": 109.09,
-      "eval_accuracy_NAT": 0.9918334894337543,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9376274196317439,
-      "eval_accuracy_concrete tiles": 0.9839173240384027,
-      "eval_accuracy_dark bricks": 0.8464647242215465,
-      "eval_accuracy_grass": 0.887982888119196,
-      "eval_accuracy_red bricks": 0.9138849431818182,
-      "eval_accuracy_road": 0.9463365204047948,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5118407631020319,
-      "eval_accuracy_speedway bricks": 0.9891012482093874,
-      "eval_accuracy_steel": 0.86359738803509,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9774768586547675,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8706945728014827,
-      "eval_iou_concrete tiles": 0.9318892621336731,
-      "eval_iou_dark bricks": 0.8062174646235061,
-      "eval_iou_grass": 0.8239324942680447,
-      "eval_iou_red bricks": 0.8297597936482347,
-      "eval_iou_road": 0.9371731573167855,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3339689163908476,
-      "eval_iou_speedway bricks": 0.9713874279618957,
-      "eval_iou_steel": 0.8117235874425561,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0992363840341568,
-      "eval_mean_accuracy": 0.6825066698752127,
-      "eval_mean_iou": 0.6380171950185994,
-      "eval_overall_accuracy": 0.9725963804456923,
-      "eval_runtime": 3.7015,
-      "eval_samples_per_second": 9.726,
-      "eval_steps_per_second": 0.54,
-      "step": 1200
-    },
-    {
-      "epoch": 109.18,
-      "learning_rate": 8.227751196172249e-05,
-      "loss": 0.0933,
-      "step": 1201
-    },
-    {
-      "epoch": 109.27,
-      "learning_rate": 8.225837320574163e-05,
-      "loss": 0.072,
-      "step": 1202
-    },
-    {
-      "epoch": 109.36,
-      "learning_rate": 8.223923444976078e-05,
-      "loss": 0.0499,
-      "step": 1203
-    },
-    {
-      "epoch": 109.45,
-      "learning_rate": 8.222009569377991e-05,
-      "loss": 0.065,
-      "step": 1204
-    },
-    {
-      "epoch": 109.55,
-      "learning_rate": 8.220095693779904e-05,
-      "loss": 0.068,
-      "step": 1205
-    },
-    {
-      "epoch": 109.64,
-      "learning_rate": 8.218181818181819e-05,
-      "loss": 0.0557,
-      "step": 1206
-    },
-    {
-      "epoch": 109.73,
-      "learning_rate": 8.216267942583733e-05,
-      "loss": 0.0697,
-      "step": 1207
-    },
-    {
-      "epoch": 109.82,
-      "learning_rate": 8.214354066985646e-05,
-      "loss": 0.047,
-      "step": 1208
-    },
-    {
-      "epoch": 109.91,
-      "learning_rate": 8.212440191387561e-05,
-      "loss": 0.056,
-      "step": 1209
-    },
-    {
-      "epoch": 110.0,
-      "learning_rate": 8.210526315789474e-05,
-      "loss": 0.06,
-      "step": 1210
-    },
-    {
-      "epoch": 110.09,
-      "learning_rate": 8.208612440191387e-05,
-      "loss": 0.0562,
-      "step": 1211
-    },
-    {
-      "epoch": 110.18,
-      "learning_rate": 8.206698564593302e-05,
-      "loss": 0.0512,
-      "step": 1212
-    },
-    {
-      "epoch": 110.27,
-      "learning_rate": 8.204784688995216e-05,
-      "loss": 0.0643,
-      "step": 1213
-    },
-    {
-      "epoch": 110.36,
-      "learning_rate": 8.20287081339713e-05,
-      "loss": 0.0596,
-      "step": 1214
-    },
-    {
-      "epoch": 110.45,
-      "learning_rate": 8.200956937799044e-05,
-      "loss": 0.0564,
-      "step": 1215
-    },
-    {
-      "epoch": 110.55,
-      "learning_rate": 8.199043062200957e-05,
-      "loss": 0.0595,
-      "step": 1216
-    },
-    {
-      "epoch": 110.64,
-      "learning_rate": 8.19712918660287e-05,
-      "loss": 0.0585,
-      "step": 1217
-    },
-    {
-      "epoch": 110.73,
-      "learning_rate": 8.195215311004785e-05,
-      "loss": 0.0551,
-      "step": 1218
-    },
-    {
-      "epoch": 110.82,
-      "learning_rate": 8.1933014354067e-05,
-      "loss": 0.0581,
-      "step": 1219
-    },
-    {
-      "epoch": 110.91,
-      "learning_rate": 8.191387559808613e-05,
-      "loss": 0.0622,
-      "step": 1220
-    },
-    {
-      "epoch": 110.91,
-      "eval_accuracy_NAT": 0.9912388491717031,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9373698978496932,
-      "eval_accuracy_concrete tiles": 0.9893291750749098,
-      "eval_accuracy_dark bricks": 0.8838381894461337,
-      "eval_accuracy_grass": 0.8492295110042395,
-      "eval_accuracy_red bricks": 0.921875,
-      "eval_accuracy_road": 0.9360461530834221,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4517133956386293,
-      "eval_accuracy_speedway bricks": 0.9884429302931235,
-      "eval_accuracy_steel": 0.876632478068729,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9769963315941174,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8579097324774773,
-      "eval_iou_concrete tiles": 0.9169925749589072,
-      "eval_iou_dark bricks": 0.8236961044551259,
-      "eval_iou_grass": 0.7775576818458191,
-      "eval_iou_red bricks": 0.7852389594676346,
-      "eval_iou_road": 0.9283654082551698,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2904317923993193,
-      "eval_iou_speedway bricks": 0.9714858606519703,
-      "eval_iou_steel": 0.8147384714410507,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1068648174405098,
-      "eval_mean_accuracy": 0.6789011984331217,
-      "eval_mean_iou": 0.6264163782728147,
-      "eval_overall_accuracy": 0.970089488559299,
-      "eval_runtime": 3.6924,
-      "eval_samples_per_second": 9.75,
-      "eval_steps_per_second": 0.542,
-      "step": 1220
-    },
-    {
-      "epoch": 111.0,
-      "learning_rate": 8.189473684210527e-05,
-      "loss": 0.0575,
-      "step": 1221
-    },
-    {
-      "epoch": 111.09,
-      "learning_rate": 8.18755980861244e-05,
-      "loss": 0.0618,
-      "step": 1222
-    },
-    {
-      "epoch": 111.18,
-      "learning_rate": 8.185645933014354e-05,
-      "loss": 0.0563,
-      "step": 1223
-    },
-    {
-      "epoch": 111.27,
-      "learning_rate": 8.183732057416268e-05,
-      "loss": 0.0538,
-      "step": 1224
-    },
-    {
-      "epoch": 111.36,
-      "learning_rate": 8.181818181818183e-05,
-      "loss": 0.0637,
-      "step": 1225
-    },
-    {
-      "epoch": 111.45,
-      "learning_rate": 8.179904306220096e-05,
-      "loss": 0.0924,
-      "step": 1226
-    },
-    {
-      "epoch": 111.55,
-      "learning_rate": 8.177990430622011e-05,
-      "loss": 0.0593,
-      "step": 1227
-    },
-    {
-      "epoch": 111.64,
-      "learning_rate": 8.176076555023924e-05,
-      "loss": 0.0699,
-      "step": 1228
-    },
-    {
-      "epoch": 111.73,
-      "learning_rate": 8.174162679425837e-05,
-      "loss": 0.0517,
-      "step": 1229
-    },
-    {
-      "epoch": 111.82,
-      "learning_rate": 8.172248803827752e-05,
-      "loss": 0.0507,
-      "step": 1230
-    },
-    {
-      "epoch": 111.91,
-      "learning_rate": 8.170334928229666e-05,
-      "loss": 0.0571,
-      "step": 1231
-    },
-    {
-      "epoch": 112.0,
-      "learning_rate": 8.16842105263158e-05,
-      "loss": 0.0562,
-      "step": 1232
-    },
-    {
-      "epoch": 112.09,
-      "learning_rate": 8.166507177033494e-05,
-      "loss": 0.0509,
-      "step": 1233
-    },
-    {
-      "epoch": 112.18,
-      "learning_rate": 8.164593301435407e-05,
-      "loss": 0.0582,
-      "step": 1234
-    },
-    {
-      "epoch": 112.27,
-      "learning_rate": 8.16267942583732e-05,
-      "loss": 0.0742,
-      "step": 1235
-    },
-    {
-      "epoch": 112.36,
-      "learning_rate": 8.160765550239235e-05,
-      "loss": 0.0626,
-      "step": 1236
-    },
-    {
-      "epoch": 112.45,
-      "learning_rate": 8.15885167464115e-05,
-      "loss": 0.0501,
-      "step": 1237
-    },
-    {
-      "epoch": 112.55,
-      "learning_rate": 8.156937799043063e-05,
-      "loss": 0.0523,
-      "step": 1238
-    },
-    {
-      "epoch": 112.64,
-      "learning_rate": 8.155023923444977e-05,
-      "loss": 0.0479,
-      "step": 1239
-    },
-    {
-      "epoch": 112.73,
-      "learning_rate": 8.15311004784689e-05,
-      "loss": 0.0491,
-      "step": 1240
-    },
-    {
-      "epoch": 112.73,
-      "eval_accuracy_NAT": 0.9914094774011998,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9446368406369372,
-      "eval_accuracy_concrete tiles": 0.9848651623555311,
-      "eval_accuracy_dark bricks": 0.8584734998652862,
-      "eval_accuracy_grass": 0.8614447935634723,
-      "eval_accuracy_red bricks": 0.9050071022727273,
-      "eval_accuracy_road": 0.9001253949578409,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4678686626416343,
-      "eval_accuracy_speedway bricks": 0.9853058131101207,
-      "eval_accuracy_steel": 0.8875568893872435,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9774155804574186,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8608262228398705,
-      "eval_iou_concrete tiles": 0.924753100597152,
-      "eval_iou_dark bricks": 0.8160995243322356,
-      "eval_iou_grass": 0.7881871382431006,
-      "eval_iou_red bricks": 0.810462712672921,
-      "eval_iou_road": 0.892805783870074,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3068175654909334,
-      "eval_iou_speedway bricks": 0.9722646790708808,
-      "eval_iou_steel": 0.8140748358994525,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11632745712995529,
-      "eval_mean_accuracy": 0.6758995104763071,
-      "eval_mean_iou": 0.6279774725749262,
-      "eval_overall_accuracy": 0.9675448735555013,
-      "eval_runtime": 3.3552,
-      "eval_samples_per_second": 10.73,
-      "eval_steps_per_second": 0.596,
-      "step": 1240
-    },
-    {
-      "epoch": 112.82,
-      "learning_rate": 8.151196172248804e-05,
-      "loss": 0.0559,
-      "step": 1241
-    },
-    {
-      "epoch": 112.91,
-      "learning_rate": 8.149282296650718e-05,
-      "loss": 0.0601,
-      "step": 1242
-    },
-    {
-      "epoch": 113.0,
-      "learning_rate": 8.147368421052633e-05,
-      "loss": 0.0589,
-      "step": 1243
-    },
-    {
-      "epoch": 113.09,
-      "learning_rate": 8.145454545454546e-05,
-      "loss": 0.0564,
-      "step": 1244
-    },
-    {
-      "epoch": 113.18,
-      "learning_rate": 8.143540669856459e-05,
-      "loss": 0.0586,
-      "step": 1245
-    },
-    {
-      "epoch": 113.27,
-      "learning_rate": 8.141626794258374e-05,
-      "loss": 0.0506,
-      "step": 1246
-    },
-    {
-      "epoch": 113.36,
-      "learning_rate": 8.139712918660287e-05,
-      "loss": 0.0528,
-      "step": 1247
-    },
-    {
-      "epoch": 113.45,
-      "learning_rate": 8.137799043062201e-05,
-      "loss": 0.0552,
-      "step": 1248
-    },
-    {
-      "epoch": 113.55,
-      "learning_rate": 8.135885167464116e-05,
-      "loss": 0.0545,
-      "step": 1249
-    },
-    {
-      "epoch": 113.64,
-      "learning_rate": 8.133971291866029e-05,
-      "loss": 0.0589,
-      "step": 1250
-    },
-    {
-      "epoch": 113.73,
-      "learning_rate": 8.132057416267942e-05,
-      "loss": 0.0508,
-      "step": 1251
-    },
-    {
-      "epoch": 113.82,
-      "learning_rate": 8.130143540669857e-05,
-      "loss": 0.0533,
-      "step": 1252
-    },
-    {
-      "epoch": 113.91,
-      "learning_rate": 8.12822966507177e-05,
-      "loss": 0.0573,
-      "step": 1253
-    },
-    {
-      "epoch": 114.0,
-      "learning_rate": 8.126315789473685e-05,
-      "loss": 0.0601,
-      "step": 1254
-    },
-    {
-      "epoch": 114.09,
-      "learning_rate": 8.124401913875599e-05,
-      "loss": 0.0632,
-      "step": 1255
-    },
-    {
-      "epoch": 114.18,
-      "learning_rate": 8.122488038277512e-05,
-      "loss": 0.0586,
-      "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.0499,
-      "step": 1258
-    },
-    {
-      "epoch": 114.45,
-      "learning_rate": 8.116746411483253e-05,
-      "loss": 0.0513,
-      "step": 1259
-    },
-    {
-      "epoch": 114.55,
-      "learning_rate": 8.114832535885168e-05,
-      "loss": 0.0468,
-      "step": 1260
-    },
-    {
-      "epoch": 114.55,
-      "eval_accuracy_NAT": 0.991638863852436,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9413722155457316,
-      "eval_accuracy_concrete tiles": 0.9876475264477466,
-      "eval_accuracy_dark bricks": 0.8520457257226435,
-      "eval_accuracy_grass": 0.8875180433179434,
-      "eval_accuracy_red bricks": 0.8908025568181818,
-      "eval_accuracy_road": 0.9044446889733454,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5028533620047969,
-      "eval_accuracy_speedway bricks": 0.9839837466073396,
-      "eval_accuracy_steel": 0.9030159620077831,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9772855600886666,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.865725933281116,
-      "eval_iou_concrete tiles": 0.917592250660455,
-      "eval_iou_dark bricks": 0.811443862028518,
-      "eval_iou_grass": 0.8086812521894208,
-      "eval_iou_red bricks": 0.8307666832256996,
-      "eval_iou_road": 0.8968619113910542,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3364817001180638,
-      "eval_iou_speedway bricks": 0.9718616743894989,
-      "eval_iou_steel": 0.810156076632887,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1142939031124115,
-      "eval_mean_accuracy": 0.6804094377921497,
-      "eval_mean_iou": 0.6328351464619524,
-      "eval_overall_accuracy": 0.9686781565348307,
-      "eval_runtime": 3.4965,
-      "eval_samples_per_second": 10.296,
-      "eval_steps_per_second": 0.572,
-      "step": 1260
-    },
-    {
-      "epoch": 114.64,
-      "learning_rate": 8.112918660287083e-05,
-      "loss": 0.0473,
-      "step": 1261
-    },
-    {
-      "epoch": 114.73,
-      "learning_rate": 8.111004784688996e-05,
-      "loss": 0.0676,
-      "step": 1262
-    },
-    {
-      "epoch": 114.82,
-      "learning_rate": 8.109090909090909e-05,
-      "loss": 0.0537,
-      "step": 1263
-    },
-    {
-      "epoch": 114.91,
-      "learning_rate": 8.107177033492824e-05,
-      "loss": 0.0573,
-      "step": 1264
-    },
-    {
-      "epoch": 115.0,
-      "learning_rate": 8.105263157894737e-05,
-      "loss": 0.0703,
-      "step": 1265
-    },
-    {
-      "epoch": 115.09,
-      "learning_rate": 8.10334928229665e-05,
-      "loss": 0.0568,
-      "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.0709,
-      "step": 1268
-    },
-    {
-      "epoch": 115.36,
-      "learning_rate": 8.097607655502392e-05,
-      "loss": 0.0644,
-      "step": 1269
-    },
-    {
-      "epoch": 115.45,
-      "learning_rate": 8.095693779904307e-05,
-      "loss": 0.0615,
-      "step": 1270
-    },
-    {
-      "epoch": 115.55,
-      "learning_rate": 8.09377990430622e-05,
-      "loss": 0.0497,
-      "step": 1271
-    },
-    {
-      "epoch": 115.64,
-      "learning_rate": 8.091866028708133e-05,
-      "loss": 0.0713,
-      "step": 1272
-    },
-    {
-      "epoch": 115.73,
-      "learning_rate": 8.089952153110049e-05,
-      "loss": 0.0532,
-      "step": 1273
-    },
-    {
-      "epoch": 115.82,
-      "learning_rate": 8.088038277511962e-05,
-      "loss": 0.0641,
-      "step": 1274
-    },
-    {
-      "epoch": 115.91,
-      "learning_rate": 8.086124401913875e-05,
-      "loss": 0.053,
-      "step": 1275
-    },
-    {
-      "epoch": 116.0,
-      "learning_rate": 8.08421052631579e-05,
-      "loss": 0.0453,
-      "step": 1276
-    },
-    {
-      "epoch": 116.09,
-      "learning_rate": 8.082296650717703e-05,
-      "loss": 0.0583,
-      "step": 1277
-    },
-    {
-      "epoch": 116.18,
-      "learning_rate": 8.080382775119618e-05,
-      "loss": 0.055,
-      "step": 1278
-    },
-    {
-      "epoch": 116.27,
-      "learning_rate": 8.078468899521532e-05,
-      "loss": 0.0499,
-      "step": 1279
-    },
-    {
-      "epoch": 116.36,
-      "learning_rate": 8.076555023923446e-05,
-      "loss": 0.0488,
-      "step": 1280
-    },
-    {
-      "epoch": 116.36,
-      "eval_accuracy_NAT": 0.9918013753305812,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9368803382119404,
-      "eval_accuracy_concrete tiles": 0.9873111967223139,
-      "eval_accuracy_dark bricks": 0.8737538970786344,
-      "eval_accuracy_grass": 0.8592638675786479,
-      "eval_accuracy_red bricks": 0.8888494318181818,
-      "eval_accuracy_road": 0.9018025676340741,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.40611474099192235,
-      "eval_accuracy_speedway bricks": 0.9868813108913902,
-      "eval_accuracy_steel": 0.8932375832728712,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9776638458899988,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8509476048870757,
-      "eval_iou_concrete tiles": 0.924634195229505,
-      "eval_iou_dark bricks": 0.818407960199005,
-      "eval_iou_grass": 0.7931259194260201,
-      "eval_iou_red bricks": 0.8181075339107697,
-      "eval_iou_road": 0.892009721720534,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.26999633431085046,
-      "eval_iou_speedway bricks": 0.9714206803157464,
-      "eval_iou_steel": 0.8077525275118547,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1215834766626358,
-      "eval_mean_accuracy": 0.671222793040812,
-      "eval_mean_iou": 0.6249281787231814,
-      "eval_overall_accuracy": 0.9674015045166016,
-      "eval_runtime": 3.2691,
-      "eval_samples_per_second": 11.012,
-      "eval_steps_per_second": 0.612,
-      "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.0638,
-      "step": 1282
-    },
-    {
-      "epoch": 116.64,
-      "learning_rate": 8.070813397129187e-05,
-      "loss": 0.0555,
-      "step": 1283
-    },
-    {
-      "epoch": 116.73,
-      "learning_rate": 8.068899521531101e-05,
-      "loss": 0.0508,
-      "step": 1284
-    },
-    {
-      "epoch": 116.82,
-      "learning_rate": 8.066985645933016e-05,
-      "loss": 0.0589,
-      "step": 1285
-    },
-    {
-      "epoch": 116.91,
-      "learning_rate": 8.065071770334929e-05,
-      "loss": 0.0672,
-      "step": 1286
-    },
-    {
-      "epoch": 117.0,
-      "learning_rate": 8.063157894736842e-05,
-      "loss": 0.0642,
-      "step": 1287
-    },
-    {
-      "epoch": 117.09,
-      "learning_rate": 8.061244019138757e-05,
-      "loss": 0.0579,
-      "step": 1288
-    },
-    {
-      "epoch": 117.18,
-      "learning_rate": 8.05933014354067e-05,
-      "loss": 0.0612,
-      "step": 1289
-    },
-    {
-      "epoch": 117.27,
-      "learning_rate": 8.057416267942584e-05,
-      "loss": 0.0554,
-      "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.0513,
-      "step": 1292
-    },
-    {
-      "epoch": 117.55,
-      "learning_rate": 8.051674641148325e-05,
-      "loss": 0.0577,
-      "step": 1293
-    },
-    {
-      "epoch": 117.64,
-      "learning_rate": 8.04976076555024e-05,
-      "loss": 0.0606,
-      "step": 1294
-    },
-    {
-      "epoch": 117.73,
-      "learning_rate": 8.047846889952153e-05,
-      "loss": 0.0622,
-      "step": 1295
-    },
-    {
-      "epoch": 117.82,
-      "learning_rate": 8.045933014354068e-05,
-      "loss": 0.0539,
-      "step": 1296
-    },
-    {
-      "epoch": 117.91,
-      "learning_rate": 8.044019138755982e-05,
-      "loss": 0.0477,
-      "step": 1297
-    },
-    {
-      "epoch": 118.0,
-      "learning_rate": 8.042105263157895e-05,
-      "loss": 0.0524,
-      "step": 1298
-    },
-    {
-      "epoch": 118.09,
-      "learning_rate": 8.040191387559809e-05,
-      "loss": 0.059,
-      "step": 1299
-    },
-    {
-      "epoch": 118.18,
-      "learning_rate": 8.038277511961723e-05,
-      "loss": 0.0502,
-      "step": 1300
-    },
-    {
-      "epoch": 118.18,
-      "eval_accuracy_NAT": 0.9919864725516172,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9447159749345465,
-      "eval_accuracy_concrete tiles": 0.9933345563505167,
-      "eval_accuracy_dark bricks": 0.8621685077556676,
-      "eval_accuracy_grass": 0.8526651684450767,
-      "eval_accuracy_red bricks": 0.9241832386363636,
-      "eval_accuracy_road": 0.8100949591710631,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4109668348358283,
-      "eval_accuracy_speedway bricks": 0.9859550799092958,
-      "eval_accuracy_steel": 0.8831046764725282,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9774632586959534,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8604061688807452,
-      "eval_iou_concrete tiles": 0.8911075758406934,
-      "eval_iou_dark bricks": 0.8152569515213277,
-      "eval_iou_grass": 0.7948819872538414,
-      "eval_iou_red bricks": 0.8271094867312887,
-      "eval_iou_road": 0.8038333199348715,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2725128880114073,
-      "eval_iou_speedway bricks": 0.9726198469501945,
-      "eval_iou_steel": 0.8083041535860903,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.16381952166557312,
-      "eval_mean_accuracy": 0.6660904206971157,
-      "eval_mean_iou": 0.6171919721081858,
-      "eval_overall_accuracy": 0.9596310721503364,
-      "eval_runtime": 4.2502,
-      "eval_samples_per_second": 8.47,
-      "eval_steps_per_second": 0.471,
-      "step": 1300
-    },
-    {
-      "epoch": 118.27,
-      "learning_rate": 8.036363636363636e-05,
-      "loss": 0.0524,
-      "step": 1301
-    },
-    {
-      "epoch": 118.36,
-      "learning_rate": 8.034449760765551e-05,
-      "loss": 0.0544,
-      "step": 1302
-    },
-    {
-      "epoch": 118.45,
-      "learning_rate": 8.032535885167465e-05,
-      "loss": 0.0489,
-      "step": 1303
-    },
-    {
-      "epoch": 118.55,
-      "learning_rate": 8.030622009569379e-05,
-      "loss": 0.0591,
-      "step": 1304
-    },
-    {
-      "epoch": 118.64,
-      "learning_rate": 8.028708133971292e-05,
-      "loss": 0.0514,
-      "step": 1305
-    },
-    {
-      "epoch": 118.73,
-      "learning_rate": 8.026794258373206e-05,
-      "loss": 0.0673,
-      "step": 1306
-    },
-    {
-      "epoch": 118.82,
-      "learning_rate": 8.02488038277512e-05,
-      "loss": 0.0554,
-      "step": 1307
-    },
-    {
-      "epoch": 118.91,
-      "learning_rate": 8.022966507177034e-05,
-      "loss": 0.0548,
-      "step": 1308
-    },
-    {
-      "epoch": 119.0,
-      "learning_rate": 8.021052631578949e-05,
-      "loss": 0.0576,
-      "step": 1309
-    },
-    {
-      "epoch": 119.09,
-      "learning_rate": 8.019138755980862e-05,
-      "loss": 0.0531,
-      "step": 1310
-    },
-    {
-      "epoch": 119.18,
-      "learning_rate": 8.017224880382775e-05,
-      "loss": 0.0588,
-      "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.0514,
-      "step": 1313
-    },
-    {
-      "epoch": 119.45,
-      "learning_rate": 8.011483253588517e-05,
-      "loss": 0.0612,
-      "step": 1314
-    },
-    {
-      "epoch": 119.55,
-      "learning_rate": 8.009569377990432e-05,
-      "loss": 0.0544,
-      "step": 1315
-    },
-    {
-      "epoch": 119.64,
-      "learning_rate": 8.007655502392345e-05,
-      "loss": 0.0538,
-      "step": 1316
-    },
-    {
-      "epoch": 119.73,
-      "learning_rate": 8.005741626794258e-05,
-      "loss": 0.0587,
-      "step": 1317
-    },
-    {
-      "epoch": 119.82,
-      "learning_rate": 8.003827751196173e-05,
-      "loss": 0.0534,
-      "step": 1318
-    },
-    {
-      "epoch": 119.91,
-      "learning_rate": 8.001913875598086e-05,
-      "loss": 0.0463,
-      "step": 1319
-    },
-    {
-      "epoch": 120.0,
-      "learning_rate": 8e-05,
-      "loss": 0.0471,
-      "step": 1320
-    },
-    {
-      "epoch": 120.0,
-      "eval_accuracy_NAT": 0.9916586263774656,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9445080797459119,
-      "eval_accuracy_concrete tiles": 0.9839173240384027,
-      "eval_accuracy_dark bricks": 0.8459643585697241,
-      "eval_accuracy_grass": 0.8727688323308297,
-      "eval_accuracy_red bricks": 0.9241832386363636,
-      "eval_accuracy_road": 0.8645407968928169,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4731342872108731,
-      "eval_accuracy_speedway bricks": 0.9885829209040979,
-      "eval_accuracy_steel": 0.8708198667634061,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9774136470136462,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8690879201845313,
-      "eval_iou_concrete tiles": 0.9319162491674148,
-      "eval_iou_dark bricks": 0.8093905358129257,
-      "eval_iou_grass": 0.8054173130823788,
-      "eval_iou_red bricks": 0.8309386973180076,
-      "eval_iou_road": 0.8590364369940141,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3193999851112931,
-      "eval_iou_speedway bricks": 0.9720258228650185,
-      "eval_iou_steel": 0.8064565390018936,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14239585399627686,
-      "eval_mean_accuracy": 0.6738521793438377,
-      "eval_mean_iou": 0.6293140881962402,
-      "eval_overall_accuracy": 0.9652636845906576,
-      "eval_runtime": 3.2751,
-      "eval_samples_per_second": 10.992,
-      "eval_steps_per_second": 0.611,
-      "step": 1320
-    },
-    {
-      "epoch": 120.09,
-      "learning_rate": 7.998086124401914e-05,
-      "loss": 0.0477,
-      "step": 1321
-    },
-    {
-      "epoch": 120.18,
-      "learning_rate": 7.996172248803828e-05,
-      "loss": 0.0508,
-      "step": 1322
-    },
-    {
-      "epoch": 120.27,
-      "learning_rate": 7.994258373205742e-05,
-      "loss": 0.0553,
-      "step": 1323
-    },
-    {
-      "epoch": 120.36,
-      "learning_rate": 7.992344497607656e-05,
-      "loss": 0.0478,
-      "step": 1324
-    },
-    {
-      "epoch": 120.45,
-      "learning_rate": 7.99043062200957e-05,
-      "loss": 0.0652,
-      "step": 1325
-    },
-    {
-      "epoch": 120.55,
-      "learning_rate": 7.988516746411484e-05,
-      "loss": 0.0489,
-      "step": 1326
-    },
-    {
-      "epoch": 120.64,
-      "learning_rate": 7.986602870813397e-05,
-      "loss": 0.0573,
-      "step": 1327
-    },
-    {
-      "epoch": 120.73,
-      "learning_rate": 7.984688995215312e-05,
-      "loss": 0.0501,
-      "step": 1328
-    },
-    {
-      "epoch": 120.82,
-      "learning_rate": 7.982775119617225e-05,
-      "loss": 0.0513,
-      "step": 1329
-    },
-    {
-      "epoch": 120.91,
-      "learning_rate": 7.98086124401914e-05,
-      "loss": 0.0609,
-      "step": 1330
-    },
-    {
-      "epoch": 121.0,
-      "learning_rate": 7.978947368421053e-05,
-      "loss": 0.0576,
-      "step": 1331
-    },
-    {
-      "epoch": 121.09,
-      "learning_rate": 7.977033492822967e-05,
-      "loss": 0.0597,
-      "step": 1332
-    },
-    {
-      "epoch": 121.18,
-      "learning_rate": 7.97511961722488e-05,
-      "loss": 0.0524,
-      "step": 1333
-    },
-    {
-      "epoch": 121.27,
-      "learning_rate": 7.973205741626795e-05,
-      "loss": 0.0454,
-      "step": 1334
-    },
-    {
-      "epoch": 121.36,
-      "learning_rate": 7.971291866028708e-05,
-      "loss": 0.0551,
-      "step": 1335
-    },
-    {
-      "epoch": 121.45,
-      "learning_rate": 7.969377990430623e-05,
-      "loss": 0.0573,
-      "step": 1336
-    },
-    {
-      "epoch": 121.55,
-      "learning_rate": 7.967464114832536e-05,
-      "loss": 0.0553,
-      "step": 1337
-    },
-    {
-      "epoch": 121.64,
-      "learning_rate": 7.96555023923445e-05,
-      "loss": 0.0545,
-      "step": 1338
-    },
-    {
-      "epoch": 121.73,
-      "learning_rate": 7.963636363636364e-05,
-      "loss": 0.0522,
-      "step": 1339
-    },
-    {
-      "epoch": 121.82,
-      "learning_rate": 7.961722488038278e-05,
-      "loss": 0.0518,
-      "step": 1340
-    },
-    {
-      "epoch": 121.82,
-      "eval_accuracy_NAT": 0.9917105030056683,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9401382570067385,
-      "eval_accuracy_concrete tiles": 0.9917446340121079,
-      "eval_accuracy_dark bricks": 0.8784881259381856,
-      "eval_accuracy_grass": 0.8560309244120412,
-      "eval_accuracy_red bricks": 0.9200994318181818,
-      "eval_accuracy_road": 0.8729907483430815,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4371846828219337,
-      "eval_accuracy_speedway bricks": 0.9890692675956734,
-      "eval_accuracy_steel": 0.8778362245234483,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.977888084679696,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8642510671562474,
-      "eval_iou_concrete tiles": 0.9045680182943834,
-      "eval_iou_dark bricks": 0.8157255182273052,
-      "eval_iou_grass": 0.7942259924379503,
-      "eval_iou_red bricks": 0.8186413902053713,
-      "eval_iou_road": 0.8660652591170825,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.28958036594718967,
-      "eval_iou_speedway bricks": 0.9721758285035099,
-      "eval_iou_steel": 0.8123278578458675,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13630938529968262,
-      "eval_mean_accuracy": 0.6734840614982354,
-      "eval_mean_iou": 0.6242653371088156,
-      "eval_overall_accuracy": 0.9653342564900717,
-      "eval_runtime": 3.324,
-      "eval_samples_per_second": 10.83,
-      "eval_steps_per_second": 0.602,
-      "step": 1340
-    },
-    {
-      "epoch": 121.91,
-      "learning_rate": 7.959808612440191e-05,
-      "loss": 0.0538,
-      "step": 1341
-    },
-    {
-      "epoch": 122.0,
-      "learning_rate": 7.957894736842106e-05,
-      "loss": 0.0668,
-      "step": 1342
-    },
-    {
-      "epoch": 122.09,
-      "learning_rate": 7.955980861244019e-05,
-      "loss": 0.0483,
-      "step": 1343
-    },
-    {
-      "epoch": 122.18,
-      "learning_rate": 7.954066985645934e-05,
-      "loss": 0.0479,
-      "step": 1344
-    },
-    {
-      "epoch": 122.27,
-      "learning_rate": 7.952153110047847e-05,
-      "loss": 0.0565,
-      "step": 1345
-    },
-    {
-      "epoch": 122.36,
-      "learning_rate": 7.950239234449761e-05,
-      "loss": 0.0467,
-      "step": 1346
-    },
-    {
-      "epoch": 122.45,
-      "learning_rate": 7.948325358851675e-05,
-      "loss": 0.0639,
-      "step": 1347
-    },
-    {
-      "epoch": 122.55,
-      "learning_rate": 7.946411483253588e-05,
-      "loss": 0.0561,
-      "step": 1348
-    },
-    {
-      "epoch": 122.64,
-      "learning_rate": 7.944497607655504e-05,
-      "loss": 0.0533,
-      "step": 1349
-    },
-    {
-      "epoch": 122.73,
-      "learning_rate": 7.942583732057417e-05,
-      "loss": 0.0511,
-      "step": 1350
-    },
-    {
-      "epoch": 122.82,
-      "learning_rate": 7.94066985645933e-05,
-      "loss": 0.0505,
-      "step": 1351
-    },
-    {
-      "epoch": 122.91,
-      "learning_rate": 7.938755980861245e-05,
-      "loss": 0.0518,
-      "step": 1352
-    },
-    {
-      "epoch": 123.0,
-      "learning_rate": 7.936842105263158e-05,
-      "loss": 0.0539,
-      "step": 1353
-    },
-    {
-      "epoch": 123.09,
-      "learning_rate": 7.934928229665071e-05,
-      "loss": 0.0552,
-      "step": 1354
-    },
-    {
-      "epoch": 123.18,
-      "learning_rate": 7.933014354066987e-05,
-      "loss": 0.0519,
-      "step": 1355
-    },
-    {
-      "epoch": 123.27,
-      "learning_rate": 7.9311004784689e-05,
-      "loss": 0.0578,
-      "step": 1356
-    },
-    {
-      "epoch": 123.36,
-      "learning_rate": 7.929186602870813e-05,
-      "loss": 0.0541,
-      "step": 1357
-    },
-    {
-      "epoch": 123.45,
-      "learning_rate": 7.927272727272728e-05,
-      "loss": 0.0572,
-      "step": 1358
-    },
-    {
-      "epoch": 123.55,
-      "learning_rate": 7.925358851674641e-05,
-      "loss": 0.0599,
-      "step": 1359
-    },
-    {
-      "epoch": 123.64,
-      "learning_rate": 7.923444976076554e-05,
-      "loss": 0.054,
-      "step": 1360
-    },
-    {
-      "epoch": 123.64,
-      "eval_accuracy_NAT": 0.9910841015426768,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9396486973689858,
-      "eval_accuracy_concrete tiles": 0.9936403106463646,
-      "eval_accuracy_dark bricks": 0.8761017666756475,
-      "eval_accuracy_grass": 0.8899541096824026,
-      "eval_accuracy_red bricks": 0.9284446022727273,
-      "eval_accuracy_road": 0.8862908364944068,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5216827943649547,
-      "eval_accuracy_speedway bricks": 0.9875782469072333,
-      "eval_accuracy_steel": 0.8888760635841962,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9780780764569964,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8690537033637171,
-      "eval_iou_concrete tiles": 0.8889679131219739,
-      "eval_iou_dark bricks": 0.8135096497498213,
-      "eval_iou_grass": 0.8122900536888344,
-      "eval_iou_red bricks": 0.8096934035305048,
-      "eval_iou_road": 0.8788023030236177,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3431809938338774,
-      "eval_iou_speedway bricks": 0.9728104649752854,
-      "eval_iou_steel": 0.8191689018228236,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11979474127292633,
-      "eval_mean_accuracy": 0.6848693484261227,
-      "eval_mean_iou": 0.6296581125821118,
-      "eval_overall_accuracy": 0.9673231972588433,
-      "eval_runtime": 3.3371,
-      "eval_samples_per_second": 10.788,
-      "eval_steps_per_second": 0.599,
-      "step": 1360
-    },
-    {
-      "epoch": 123.73,
-      "learning_rate": 7.92153110047847e-05,
-      "loss": 0.0598,
-      "step": 1361
-    },
-    {
-      "epoch": 123.82,
-      "learning_rate": 7.919617224880384e-05,
-      "loss": 0.0463,
-      "step": 1362
-    },
-    {
-      "epoch": 123.91,
-      "learning_rate": 7.917703349282297e-05,
-      "loss": 0.0468,
-      "step": 1363
-    },
-    {
-      "epoch": 124.0,
-      "learning_rate": 7.915789473684211e-05,
-      "loss": 0.0542,
-      "step": 1364
-    },
-    {
-      "epoch": 124.09,
-      "learning_rate": 7.913875598086124e-05,
-      "loss": 0.0532,
-      "step": 1365
-    },
-    {
-      "epoch": 124.18,
-      "learning_rate": 7.911961722488038e-05,
-      "loss": 0.0489,
-      "step": 1366
-    },
-    {
-      "epoch": 124.27,
-      "learning_rate": 7.910047846889954e-05,
-      "loss": 0.0492,
-      "step": 1367
-    },
-    {
-      "epoch": 124.36,
-      "learning_rate": 7.908133971291867e-05,
-      "loss": 0.05,
-      "step": 1368
-    },
-    {
-      "epoch": 124.45,
-      "learning_rate": 7.90622009569378e-05,
-      "loss": 0.0531,
-      "step": 1369
-    },
-    {
-      "epoch": 124.55,
-      "learning_rate": 7.904306220095695e-05,
-      "loss": 0.0495,
-      "step": 1370
-    },
-    {
-      "epoch": 124.64,
-      "learning_rate": 7.902392344497608e-05,
-      "loss": 0.0546,
-      "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.0488,
-      "step": 1373
-    },
-    {
-      "epoch": 124.91,
-      "learning_rate": 7.89665071770335e-05,
-      "loss": 0.0527,
-      "step": 1374
-    },
-    {
-      "epoch": 125.0,
-      "learning_rate": 7.894736842105263e-05,
-      "loss": 0.0479,
-      "step": 1375
-    },
-    {
-      "epoch": 125.09,
-      "learning_rate": 7.892822966507178e-05,
-      "loss": 0.0552,
-      "step": 1376
-    },
-    {
-      "epoch": 125.18,
-      "learning_rate": 7.890909090909091e-05,
-      "loss": 0.0482,
-      "step": 1377
-    },
-    {
-      "epoch": 125.27,
-      "learning_rate": 7.888995215311004e-05,
-      "loss": 0.07,
-      "step": 1378
-    },
-    {
-      "epoch": 125.36,
-      "learning_rate": 7.88708133971292e-05,
-      "loss": 0.0457,
-      "step": 1379
-    },
-    {
-      "epoch": 125.45,
-      "learning_rate": 7.885167464114833e-05,
-      "loss": 0.0519,
-      "step": 1380
-    },
-    {
-      "epoch": 125.45,
-      "eval_accuracy_NAT": 0.9897083121886079,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9451196939782823,
-      "eval_accuracy_concrete tiles": 0.9916223322937687,
-      "eval_accuracy_dark bricks": 0.867864978253339,
-      "eval_accuracy_grass": 0.8966786314689446,
-      "eval_accuracy_red bricks": 0.9231178977272727,
-      "eval_accuracy_road": 0.8728347604013122,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5174923496815813,
-      "eval_accuracy_speedway bricks": 0.9857088895244784,
-      "eval_accuracy_steel": 0.8935508871446475,
-      "eval_accuracy_tiles": 0.0028763183125599234,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9779495365216413,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8706672119784067,
-      "eval_iou_concrete tiles": 0.9128317712291367,
-      "eval_iou_dark bricks": 0.8078824793980652,
-      "eval_iou_grass": 0.8042123543167384,
-      "eval_iou_red bricks": 0.8062965260545906,
-      "eval_iou_road": 0.863738727958492,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3420496373774555,
-      "eval_iou_speedway bricks": 0.9726074049552923,
-      "eval_iou_steel": 0.8206959221536481,
-      "eval_iou_tiles": 0.0028763183125599234,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13092216849327087,
-      "eval_mean_accuracy": 0.6835826962288304,
-      "eval_mean_iou": 0.6293698377120021,
-      "eval_overall_accuracy": 0.9656412336561415,
-      "eval_runtime": 3.6426,
-      "eval_samples_per_second": 9.883,
-      "eval_steps_per_second": 0.549,
-      "step": 1380
-    },
-    {
-      "epoch": 125.55,
-      "learning_rate": 7.883253588516747e-05,
-      "loss": 0.0546,
-      "step": 1381
-    },
-    {
-      "epoch": 125.64,
-      "learning_rate": 7.881339712918661e-05,
-      "loss": 0.0509,
-      "step": 1382
-    },
-    {
-      "epoch": 125.73,
-      "learning_rate": 7.879425837320574e-05,
-      "loss": 0.0498,
-      "step": 1383
-    },
-    {
-      "epoch": 125.82,
-      "learning_rate": 7.877511961722487e-05,
-      "loss": 0.0479,
-      "step": 1384
-    },
-    {
-      "epoch": 125.91,
-      "learning_rate": 7.875598086124403e-05,
-      "loss": 0.0531,
-      "step": 1385
-    },
-    {
-      "epoch": 126.0,
-      "learning_rate": 7.873684210526317e-05,
-      "loss": 0.0466,
-      "step": 1386
-    },
-    {
-      "epoch": 126.09,
-      "learning_rate": 7.87177033492823e-05,
-      "loss": 0.0477,
-      "step": 1387
-    },
-    {
-      "epoch": 126.18,
-      "learning_rate": 7.869856459330144e-05,
-      "loss": 0.0437,
-      "step": 1388
-    },
-    {
-      "epoch": 126.27,
-      "learning_rate": 7.867942583732058e-05,
-      "loss": 0.0563,
-      "step": 1389
-    },
-    {
-      "epoch": 126.36,
-      "learning_rate": 7.866028708133971e-05,
-      "loss": 0.0544,
-      "step": 1390
-    },
-    {
-      "epoch": 126.45,
-      "learning_rate": 7.864114832535887e-05,
-      "loss": 0.0503,
-      "step": 1391
-    },
-    {
-      "epoch": 126.55,
-      "learning_rate": 7.8622009569378e-05,
-      "loss": 0.055,
-      "step": 1392
-    },
-    {
-      "epoch": 126.64,
-      "learning_rate": 7.860287081339713e-05,
-      "loss": 0.0541,
-      "step": 1393
-    },
-    {
-      "epoch": 126.73,
-      "learning_rate": 7.858373205741628e-05,
-      "loss": 0.0526,
-      "step": 1394
-    },
-    {
-      "epoch": 126.82,
-      "learning_rate": 7.856459330143541e-05,
-      "loss": 0.0511,
-      "step": 1395
-    },
-    {
-      "epoch": 126.91,
-      "learning_rate": 7.854545454545454e-05,
-      "loss": 0.0496,
-      "step": 1396
-    },
-    {
-      "epoch": 127.0,
-      "learning_rate": 7.852631578947369e-05,
-      "loss": 0.0507,
-      "step": 1397
-    },
-    {
-      "epoch": 127.09,
-      "learning_rate": 7.850717703349283e-05,
-      "loss": 0.0508,
-      "step": 1398
-    },
-    {
-      "epoch": 127.18,
-      "learning_rate": 7.848803827751196e-05,
-      "loss": 0.0599,
-      "step": 1399
-    },
-    {
-      "epoch": 127.27,
-      "learning_rate": 7.846889952153111e-05,
-      "loss": 0.0485,
-      "step": 1400
-    },
-    {
-      "epoch": 127.27,
-      "eval_accuracy_NAT": 0.9913581301263461,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9484969848491351,
-      "eval_accuracy_concrete tiles": 0.9869748669968813,
-      "eval_accuracy_dark bricks": 0.853931719333359,
-      "eval_accuracy_grass": 0.8946549838003335,
-      "eval_accuracy_red bricks": 0.9094460227272727,
-      "eval_accuracy_road": 0.8884613663818174,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5687701596228599,
-      "eval_accuracy_speedway bricks": 0.9837665197972068,
-      "eval_accuracy_steel": 0.8992645603851989,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9781912871005799,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8774342360763868,
-      "eval_iou_concrete tiles": 0.9327053656563322,
-      "eval_iou_dark bricks": 0.8122574503917405,
-      "eval_iou_grass": 0.8080611658637911,
-      "eval_iou_red bricks": 0.8236050812027658,
-      "eval_iou_road": 0.8812408322269039,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.37549140943506115,
-      "eval_iou_speedway bricks": 0.9730232186169095,
-      "eval_iou_steel": 0.8219849123150779,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1189020499587059,
-      "eval_mean_accuracy": 0.6865481010784932,
-      "eval_mean_iou": 0.6372303814527347,
-      "eval_overall_accuracy": 0.9680691825018989,
-      "eval_runtime": 3.8603,
-      "eval_samples_per_second": 9.326,
-      "eval_steps_per_second": 0.518,
-      "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.045,
-      "step": 1403
-    },
-    {
-      "epoch": 127.64,
-      "learning_rate": 7.839234449760766e-05,
-      "loss": 0.0554,
-      "step": 1404
-    },
-    {
-      "epoch": 127.73,
-      "learning_rate": 7.83732057416268e-05,
-      "loss": 0.0585,
-      "step": 1405
-    },
-    {
-      "epoch": 127.82,
-      "learning_rate": 7.835406698564594e-05,
-      "loss": 0.0476,
-      "step": 1406
-    },
-    {
-      "epoch": 127.91,
-      "learning_rate": 7.833492822966507e-05,
-      "loss": 0.057,
-      "step": 1407
-    },
-    {
-      "epoch": 128.0,
-      "learning_rate": 7.83157894736842e-05,
-      "loss": 0.0529,
-      "step": 1408
-    },
-    {
-      "epoch": 128.09,
-      "learning_rate": 7.829665071770335e-05,
-      "loss": 0.0452,
-      "step": 1409
-    },
-    {
-      "epoch": 128.18,
-      "learning_rate": 7.82775119617225e-05,
-      "loss": 0.0498,
-      "step": 1410
-    },
-    {
-      "epoch": 128.27,
-      "learning_rate": 7.825837320574163e-05,
-      "loss": 0.0531,
-      "step": 1411
-    },
-    {
-      "epoch": 128.36,
-      "learning_rate": 7.823923444976077e-05,
-      "loss": 0.0545,
-      "step": 1412
-    },
-    {
-      "epoch": 128.45,
-      "learning_rate": 7.82200956937799e-05,
-      "loss": 0.0553,
-      "step": 1413
-    },
-    {
-      "epoch": 128.55,
-      "learning_rate": 7.820095693779904e-05,
-      "loss": 0.0522,
-      "step": 1414
-    },
-    {
-      "epoch": 128.64,
-      "learning_rate": 7.818181818181818e-05,
-      "loss": 0.0511,
-      "step": 1415
-    },
-    {
-      "epoch": 128.73,
-      "learning_rate": 7.816267942583733e-05,
-      "loss": 0.0481,
-      "step": 1416
-    },
-    {
-      "epoch": 128.82,
-      "learning_rate": 7.814354066985646e-05,
-      "loss": 0.0528,
-      "step": 1417
-    },
-    {
-      "epoch": 128.91,
-      "learning_rate": 7.81244019138756e-05,
-      "loss": 0.0441,
-      "step": 1418
-    },
-    {
-      "epoch": 129.0,
-      "learning_rate": 7.810526315789474e-05,
-      "loss": 0.0492,
-      "step": 1419
-    },
-    {
-      "epoch": 129.09,
-      "learning_rate": 7.808612440191387e-05,
-      "loss": 0.0573,
-      "step": 1420
-    },
-    {
-      "epoch": 129.09,
-      "eval_accuracy_NAT": 0.9920323498418645,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9442934782608695,
-      "eval_accuracy_concrete tiles": 0.9802482724882284,
-      "eval_accuracy_dark bricks": 0.8537777606712598,
-      "eval_accuracy_grass": 0.8625702072928207,
-      "eval_accuracy_red bricks": 0.9037642045454546,
-      "eval_accuracy_road": 0.8754962290821984,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5081189865740358,
-      "eval_accuracy_speedway bricks": 0.9892502899374508,
-      "eval_accuracy_steel": 0.8761872567772574,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9784310122006782,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.868754288027088,
-      "eval_iou_concrete tiles": 0.9450257921886515,
-      "eval_iou_dark bricks": 0.8114277353038007,
-      "eval_iou_grass": 0.7994920486829872,
-      "eval_iou_red bricks": 0.8035996210925166,
-      "eval_iou_road": 0.8695491345030223,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33200634074287566,
-      "eval_iou_speedway bricks": 0.9725250703690631,
-      "eval_iou_steel": 0.8194106035838756,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12597930431365967,
-      "eval_mean_accuracy": 0.6758260796516493,
-      "eval_mean_iou": 0.630786280514966,
-      "eval_overall_accuracy": 0.9664392471313477,
-      "eval_runtime": 3.4421,
-      "eval_samples_per_second": 10.459,
-      "eval_steps_per_second": 0.581,
-      "step": 1420
-    },
-    {
-      "epoch": 129.18,
-      "learning_rate": 7.806698564593302e-05,
-      "loss": 0.0444,
-      "step": 1421
-    },
-    {
-      "epoch": 129.27,
-      "learning_rate": 7.804784688995216e-05,
-      "loss": 0.051,
-      "step": 1422
-    },
-    {
-      "epoch": 129.36,
-      "learning_rate": 7.80287081339713e-05,
-      "loss": 0.0489,
-      "step": 1423
-    },
-    {
-      "epoch": 129.45,
-      "learning_rate": 7.800956937799043e-05,
-      "loss": 0.0503,
-      "step": 1424
-    },
-    {
-      "epoch": 129.55,
-      "learning_rate": 7.799043062200957e-05,
-      "loss": 0.0616,
-      "step": 1425
-    },
-    {
-      "epoch": 129.64,
-      "learning_rate": 7.797129186602872e-05,
-      "loss": 0.0483,
-      "step": 1426
-    },
-    {
-      "epoch": 129.73,
-      "learning_rate": 7.795215311004785e-05,
-      "loss": 0.0523,
-      "step": 1427
-    },
-    {
-      "epoch": 129.82,
-      "learning_rate": 7.7933014354067e-05,
-      "loss": 0.0505,
-      "step": 1428
-    },
-    {
-      "epoch": 129.91,
-      "learning_rate": 7.791387559808613e-05,
-      "loss": 0.0483,
-      "step": 1429
-    },
-    {
-      "epoch": 130.0,
-      "learning_rate": 7.789473684210526e-05,
-      "loss": 0.0481,
-      "step": 1430
-    },
-    {
-      "epoch": 130.09,
-      "learning_rate": 7.78755980861244e-05,
-      "loss": 0.0571,
-      "step": 1431
-    },
-    {
-      "epoch": 130.18,
-      "learning_rate": 7.785645933014355e-05,
-      "loss": 0.047,
-      "step": 1432
-    },
-    {
-      "epoch": 130.27,
-      "learning_rate": 7.783732057416268e-05,
-      "loss": 0.0662,
-      "step": 1433
-    },
-    {
-      "epoch": 130.36,
-      "learning_rate": 7.781818181818183e-05,
-      "loss": 0.0502,
-      "step": 1434
-    },
-    {
-      "epoch": 130.45,
-      "learning_rate": 7.779904306220096e-05,
-      "loss": 0.0561,
-      "step": 1435
-    },
-    {
-      "epoch": 130.55,
-      "learning_rate": 7.777990430622009e-05,
-      "loss": 0.0447,
-      "step": 1436
-    },
-    {
-      "epoch": 130.64,
-      "learning_rate": 7.776076555023924e-05,
-      "loss": 0.0533,
-      "step": 1437
-    },
-    {
-      "epoch": 130.73,
-      "learning_rate": 7.774162679425838e-05,
-      "loss": 0.0449,
-      "step": 1438
-    },
-    {
-      "epoch": 130.82,
-      "learning_rate": 7.772248803827751e-05,
-      "loss": 0.0492,
-      "step": 1439
-    },
-    {
-      "epoch": 130.91,
-      "learning_rate": 7.770334928229666e-05,
-      "loss": 0.055,
-      "step": 1440
-    },
-    {
-      "epoch": 130.91,
-      "eval_accuracy_NAT": 0.993018358679948,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9416874114768874,
-      "eval_accuracy_concrete tiles": 0.9845594080596832,
-      "eval_accuracy_dark bricks": 0.8546245333128055,
-      "eval_accuracy_grass": 0.8574114785210246,
-      "eval_accuracy_red bricks": 0.9254261363636364,
-      "eval_accuracy_road": 0.9675871567509525,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.392909326496292,
-      "eval_accuracy_speedway bricks": 0.9885388721342655,
-      "eval_accuracy_steel": 0.8703911351493965,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9778120681249312,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8698354095557854,
-      "eval_iou_concrete tiles": 0.9225854511073547,
-      "eval_iou_dark bricks": 0.8140788267644363,
-      "eval_iou_grass": 0.7989760391866937,
-      "eval_iou_red bricks": 0.8193680238956139,
-      "eval_iou_road": 0.9595118154990665,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.26591040543313993,
-      "eval_iou_speedway bricks": 0.9725012228538321,
-      "eval_iou_steel": 0.8170328692273758,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09530654549598694,
-      "eval_mean_accuracy": 0.675088755149607,
-      "eval_mean_iou": 0.6321240101267869,
-      "eval_overall_accuracy": 0.9741270277235243,
-      "eval_runtime": 3.4607,
-      "eval_samples_per_second": 10.402,
-      "eval_steps_per_second": 0.578,
-      "step": 1440
-    },
-    {
-      "epoch": 131.0,
-      "learning_rate": 7.768421052631579e-05,
-      "loss": 0.0505,
-      "step": 1441
-    },
-    {
-      "epoch": 131.09,
-      "learning_rate": 7.766507177033492e-05,
-      "loss": 0.0476,
-      "step": 1442
-    },
-    {
-      "epoch": 131.18,
-      "learning_rate": 7.764593301435407e-05,
-      "loss": 0.0568,
-      "step": 1443
-    },
-    {
-      "epoch": 131.27,
-      "learning_rate": 7.762679425837321e-05,
-      "loss": 0.0489,
-      "step": 1444
-    },
-    {
-      "epoch": 131.36,
-      "learning_rate": 7.760765550239235e-05,
-      "loss": 0.0521,
-      "step": 1445
-    },
-    {
-      "epoch": 131.45,
-      "learning_rate": 7.758851674641149e-05,
-      "loss": 0.0569,
-      "step": 1446
-    },
-    {
-      "epoch": 131.55,
-      "learning_rate": 7.756937799043062e-05,
-      "loss": 0.0512,
-      "step": 1447
-    },
-    {
-      "epoch": 131.64,
-      "learning_rate": 7.755023923444976e-05,
-      "loss": 0.051,
-      "step": 1448
-    },
-    {
-      "epoch": 131.73,
-      "learning_rate": 7.75311004784689e-05,
-      "loss": 0.0484,
-      "step": 1449
-    },
-    {
-      "epoch": 131.82,
-      "learning_rate": 7.751196172248805e-05,
-      "loss": 0.066,
-      "step": 1450
-    },
-    {
-      "epoch": 131.91,
-      "learning_rate": 7.749282296650718e-05,
-      "loss": 0.0551,
-      "step": 1451
-    },
-    {
-      "epoch": 132.0,
-      "learning_rate": 7.747368421052633e-05,
-      "loss": 0.0465,
-      "step": 1452
-    },
-    {
-      "epoch": 132.09,
-      "learning_rate": 7.745454545454546e-05,
-      "loss": 0.0476,
-      "step": 1453
-    },
-    {
-      "epoch": 132.18,
-      "learning_rate": 7.743540669856459e-05,
-      "loss": 0.0539,
-      "step": 1454
-    },
-    {
-      "epoch": 132.27,
-      "learning_rate": 7.741626794258373e-05,
-      "loss": 0.0468,
-      "step": 1455
-    },
-    {
-      "epoch": 132.36,
-      "learning_rate": 7.739712918660288e-05,
-      "loss": 0.0528,
-      "step": 1456
-    },
-    {
-      "epoch": 132.45,
-      "learning_rate": 7.737799043062201e-05,
-      "loss": 0.0484,
-      "step": 1457
-    },
-    {
-      "epoch": 132.55,
-      "learning_rate": 7.735885167464116e-05,
-      "loss": 0.0574,
-      "step": 1458
-    },
-    {
-      "epoch": 132.64,
-      "learning_rate": 7.733971291866029e-05,
-      "loss": 0.0531,
-      "step": 1459
-    },
-    {
-      "epoch": 132.73,
-      "learning_rate": 7.732057416267942e-05,
-      "loss": 0.0516,
-      "step": 1460
-    },
-    {
-      "epoch": 132.73,
-      "eval_accuracy_NAT": 0.9909898766465536,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9402509227863857,
-      "eval_accuracy_concrete tiles": 0.9883507613281967,
-      "eval_accuracy_dark bricks": 0.8693275855432816,
-      "eval_accuracy_grass": 0.8819503909239926,
-      "eval_accuracy_red bricks": 0.9074928977272727,
-      "eval_accuracy_road": 0.8681031261676423,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4978909933008023,
-      "eval_accuracy_speedway bricks": 0.9890837493830155,
-      "eval_accuracy_steel": 0.8781577732339555,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9784708601446531,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8639945672268701,
-      "eval_iou_concrete tiles": 0.9211763700094041,
-      "eval_iou_dark bricks": 0.8158797818155546,
-      "eval_iou_grass": 0.8049847993288098,
-      "eval_iou_red bricks": 0.8165841188688289,
-      "eval_iou_road": 0.8579948131416347,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3304121919537496,
-      "eval_iou_speedway bricks": 0.9728334056406857,
-      "eval_iou_steel": 0.8217603308335648,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14384014904499054,
-      "eval_mean_accuracy": 0.6778152366954692,
-      "eval_mean_iou": 0.6295454799202889,
-      "eval_overall_accuracy": 0.9654634263780382,
-      "eval_runtime": 3.4168,
-      "eval_samples_per_second": 10.536,
-      "eval_steps_per_second": 0.585,
-      "step": 1460
-    },
-    {
-      "epoch": 132.82,
-      "learning_rate": 7.730143540669857e-05,
-      "loss": 0.0472,
-      "step": 1461
-    },
-    {
-      "epoch": 132.91,
-      "learning_rate": 7.728229665071771e-05,
-      "loss": 0.0497,
-      "step": 1462
-    },
-    {
-      "epoch": 133.0,
-      "learning_rate": 7.726315789473684e-05,
-      "loss": 0.0518,
-      "step": 1463
-    },
-    {
-      "epoch": 133.09,
-      "learning_rate": 7.724401913875599e-05,
-      "loss": 0.0486,
-      "step": 1464
-    },
-    {
-      "epoch": 133.18,
-      "learning_rate": 7.722488038277512e-05,
-      "loss": 0.0554,
-      "step": 1465
-    },
-    {
-      "epoch": 133.27,
-      "learning_rate": 7.720574162679425e-05,
-      "loss": 0.0446,
-      "step": 1466
-    },
-    {
-      "epoch": 133.36,
-      "learning_rate": 7.71866028708134e-05,
-      "loss": 0.0482,
-      "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.0531,
-      "step": 1469
-    },
-    {
-      "epoch": 133.64,
-      "learning_rate": 7.712918660287082e-05,
-      "loss": 0.0533,
-      "step": 1470
-    },
-    {
-      "epoch": 133.73,
-      "learning_rate": 7.711004784688996e-05,
-      "loss": 0.0499,
-      "step": 1471
-    },
-    {
-      "epoch": 133.82,
-      "learning_rate": 7.709090909090909e-05,
-      "loss": 0.0506,
-      "step": 1472
-    },
-    {
-      "epoch": 133.91,
-      "learning_rate": 7.707177033492823e-05,
-      "loss": 0.0506,
-      "step": 1473
-    },
-    {
-      "epoch": 134.0,
-      "learning_rate": 7.705263157894738e-05,
-      "loss": 0.0456,
-      "step": 1474
-    },
-    {
-      "epoch": 134.09,
-      "learning_rate": 7.703349282296651e-05,
-      "loss": 0.0498,
-      "step": 1475
-    },
-    {
-      "epoch": 134.18,
-      "learning_rate": 7.701435406698566e-05,
-      "loss": 0.0533,
-      "step": 1476
-    },
-    {
-      "epoch": 134.27,
-      "learning_rate": 7.699521531100479e-05,
-      "loss": 0.0492,
-      "step": 1477
-    },
-    {
-      "epoch": 134.36,
-      "learning_rate": 7.697607655502392e-05,
-      "loss": 0.0531,
-      "step": 1478
-    },
-    {
-      "epoch": 134.45,
-      "learning_rate": 7.695693779904307e-05,
-      "loss": 0.0502,
-      "step": 1479
-    },
-    {
-      "epoch": 134.55,
-      "learning_rate": 7.693779904306221e-05,
-      "loss": 0.054,
-      "step": 1480
-    },
-    {
-      "epoch": 134.55,
-      "eval_accuracy_NAT": 0.9926668680562074,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9424049851924975,
-      "eval_accuracy_concrete tiles": 0.97719072952975,
-      "eval_accuracy_dark bricks": 0.8533158846849621,
-      "eval_accuracy_grass": 0.8533711733311897,
-      "eval_accuracy_red bricks": 0.9083806818181818,
-      "eval_accuracy_road": 0.8704635018446482,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.45375348055027154,
-      "eval_accuracy_speedway bricks": 0.987821420253021,
-      "eval_accuracy_steel": 0.8903271552008443,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9782088677642624,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8684024160088394,
-      "eval_iou_concrete tiles": 0.9410240555899066,
-      "eval_iou_dark bricks": 0.8102181778313782,
-      "eval_iou_grass": 0.797720834953296,
-      "eval_iou_red bricks": 0.8129667884951534,
-      "eval_iou_road": 0.8611149988336833,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.303464424654756,
-      "eval_iou_speedway bricks": 0.9726187321689348,
-      "eval_iou_steel": 0.8226000578941756,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13006871938705444,
-      "eval_mean_accuracy": 0.6715150677278133,
-      "eval_mean_iou": 0.6283337964764912,
-      "eval_overall_accuracy": 0.9656639099121094,
-      "eval_runtime": 3.4782,
-      "eval_samples_per_second": 10.35,
-      "eval_steps_per_second": 0.575,
-      "step": 1480
-    },
-    {
-      "epoch": 134.64,
-      "learning_rate": 7.691866028708134e-05,
-      "loss": 0.0463,
-      "step": 1481
-    },
-    {
-      "epoch": 134.73,
-      "learning_rate": 7.689952153110049e-05,
-      "loss": 0.0485,
-      "step": 1482
-    },
-    {
-      "epoch": 134.82,
-      "learning_rate": 7.688038277511962e-05,
-      "loss": 0.0566,
-      "step": 1483
-    },
-    {
-      "epoch": 134.91,
-      "learning_rate": 7.686124401913875e-05,
-      "loss": 0.0547,
-      "step": 1484
-    },
-    {
-      "epoch": 135.0,
-      "learning_rate": 7.68421052631579e-05,
-      "loss": 0.0476,
-      "step": 1485
-    },
-    {
-      "epoch": 135.09,
-      "learning_rate": 7.682296650717704e-05,
-      "loss": 0.0471,
-      "step": 1486
-    },
-    {
-      "epoch": 135.18,
-      "learning_rate": 7.680382775119618e-05,
-      "loss": 0.0503,
-      "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.054,
-      "step": 1489
-    },
-    {
-      "epoch": 135.45,
-      "learning_rate": 7.674641148325359e-05,
-      "loss": 0.049,
-      "step": 1490
-    },
-    {
-      "epoch": 135.55,
-      "learning_rate": 7.672727272727273e-05,
-      "loss": 0.0536,
-      "step": 1491
-    },
-    {
-      "epoch": 135.64,
-      "learning_rate": 7.670813397129188e-05,
-      "loss": 0.0454,
-      "step": 1492
-    },
-    {
-      "epoch": 135.73,
-      "learning_rate": 7.668899521531101e-05,
-      "loss": 0.0545,
-      "step": 1493
-    },
-    {
-      "epoch": 135.82,
-      "learning_rate": 7.666985645933015e-05,
-      "loss": 0.0498,
-      "step": 1494
-    },
-    {
-      "epoch": 135.91,
-      "learning_rate": 7.665071770334929e-05,
-      "loss": 0.0481,
-      "step": 1495
-    },
-    {
-      "epoch": 136.0,
-      "learning_rate": 7.663157894736842e-05,
-      "loss": 0.0467,
-      "step": 1496
-    },
-    {
-      "epoch": 136.09,
-      "learning_rate": 7.661244019138756e-05,
-      "loss": 0.0462,
-      "step": 1497
-    },
-    {
-      "epoch": 136.18,
-      "learning_rate": 7.659330143540671e-05,
-      "loss": 0.0481,
-      "step": 1498
-    },
-    {
-      "epoch": 136.27,
-      "learning_rate": 7.657416267942584e-05,
-      "loss": 0.0472,
-      "step": 1499
-    },
-    {
-      "epoch": 136.36,
-      "learning_rate": 7.655502392344497e-05,
-      "loss": 0.0456,
-      "step": 1500
-    },
-    {
-      "epoch": 136.36,
-      "eval_accuracy_NAT": 0.993093526855507,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9476426026868106,
-      "eval_accuracy_concrete tiles": 0.984742860637192,
-      "eval_accuracy_dark bricks": 0.8567799545821947,
-      "eval_accuracy_grass": 0.8516935379582479,
-      "eval_accuracy_red bricks": 0.9197443181818182,
-      "eval_accuracy_road": 0.8739617430503744,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4267361398285226,
-      "eval_accuracy_speedway bricks": 0.9851543577508336,
-      "eval_accuracy_steel": 0.8957605039245432,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9778048765233055,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8704423963247189,
-      "eval_iou_concrete tiles": 0.9285837850305616,
-      "eval_iou_dark bricks": 0.816161912444086,
-      "eval_iou_grass": 0.8045111044348409,
-      "eval_iou_red bricks": 0.7811793093047805,
-      "eval_iou_road": 0.8658496409039995,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.288825033119997,
-      "eval_iou_speedway bricks": 0.9733682461496637,
-      "eval_iou_steel": 0.8264553967396678,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12295994162559509,
-      "eval_mean_accuracy": 0.6719468881120034,
-      "eval_mean_iou": 0.6256293616135093,
-      "eval_overall_accuracy": 0.9661297268337674,
-      "eval_runtime": 3.5747,
-      "eval_samples_per_second": 10.071,
-      "eval_steps_per_second": 0.559,
-      "step": 1500
-    },
-    {
-      "epoch": 136.45,
-      "learning_rate": 7.653588516746412e-05,
-      "loss": 0.0529,
-      "step": 1501
-    },
-    {
-      "epoch": 136.55,
-      "learning_rate": 7.651674641148325e-05,
-      "loss": 0.0462,
-      "step": 1502
-    },
-    {
-      "epoch": 136.64,
-      "learning_rate": 7.64976076555024e-05,
-      "loss": 0.0595,
-      "step": 1503
-    },
-    {
-      "epoch": 136.73,
-      "learning_rate": 7.647846889952154e-05,
-      "loss": 0.0558,
-      "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.0522,
-      "step": 1506
-    },
-    {
-      "epoch": 137.0,
-      "learning_rate": 7.642105263157895e-05,
-      "loss": 0.0441,
-      "step": 1507
-    },
-    {
-      "epoch": 137.09,
-      "learning_rate": 7.640191387559808e-05,
-      "loss": 0.0483,
-      "step": 1508
-    },
-    {
-      "epoch": 137.18,
-      "learning_rate": 7.638277511961723e-05,
-      "loss": 0.0459,
-      "step": 1509
-    },
-    {
-      "epoch": 137.27,
-      "learning_rate": 7.636363636363637e-05,
-      "loss": 0.048,
-      "step": 1510
-    },
-    {
-      "epoch": 137.36,
-      "learning_rate": 7.63444976076555e-05,
-      "loss": 0.048,
-      "step": 1511
-    },
-    {
-      "epoch": 137.45,
-      "learning_rate": 7.632535885167464e-05,
-      "loss": 0.0462,
-      "step": 1512
-    },
-    {
-      "epoch": 137.55,
-      "learning_rate": 7.630622009569378e-05,
-      "loss": 0.0447,
-      "step": 1513
-    },
-    {
-      "epoch": 137.64,
-      "learning_rate": 7.628708133971292e-05,
-      "loss": 0.0556,
-      "step": 1514
-    },
-    {
-      "epoch": 137.73,
-      "learning_rate": 7.626794258373206e-05,
-      "loss": 0.0463,
-      "step": 1515
-    },
-    {
-      "epoch": 137.82,
-      "learning_rate": 7.624880382775121e-05,
-      "loss": 0.0542,
-      "step": 1516
-    },
-    {
-      "epoch": 137.91,
-      "learning_rate": 7.622966507177034e-05,
-      "loss": 0.0458,
-      "step": 1517
-    },
-    {
-      "epoch": 138.0,
-      "learning_rate": 7.621052631578947e-05,
-      "loss": 0.0568,
-      "step": 1518
-    },
-    {
-      "epoch": 138.09,
-      "learning_rate": 7.619138755980862e-05,
-      "loss": 0.055,
-      "step": 1519
-    },
-    {
-      "epoch": 138.18,
-      "learning_rate": 7.617224880382775e-05,
-      "loss": 0.0466,
-      "step": 1520
-    },
-    {
-      "epoch": 138.18,
-      "eval_accuracy_NAT": 0.9923010848920437,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9380888128245848,
-      "eval_accuracy_concrete tiles": 0.9833363908762918,
-      "eval_accuracy_dark bricks": 0.8707901928332242,
-      "eval_accuracy_grass": 0.8776095093965056,
-      "eval_accuracy_red bricks": 0.9224076704545454,
-      "eval_accuracy_road": 0.9090602391573266,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.3660849667796984,
-      "eval_accuracy_speedway bricks": 0.9886661911813156,
-      "eval_accuracy_steel": 0.8817112987269969,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9784387703705122,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.867462512557828,
-      "eval_iou_concrete tiles": 0.9318247667613142,
-      "eval_iou_dark bricks": 0.8111286390362829,
-      "eval_iou_grass": 0.8071068110957539,
-      "eval_iou_red bricks": 0.7824973640608526,
-      "eval_iou_road": 0.9004745637654634,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.24667484024372122,
-      "eval_iou_speedway bricks": 0.9722801783069386,
-      "eval_iou_steel": 0.8200179429964805,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10966861248016357,
-      "eval_mean_accuracy": 0.6715427967017332,
-      "eval_mean_iou": 0.624454337630396,
-      "eval_overall_accuracy": 0.9689988030327691,
-      "eval_runtime": 3.7551,
-      "eval_samples_per_second": 9.587,
-      "eval_steps_per_second": 0.533,
-      "step": 1520
-    },
-    {
-      "epoch": 138.27,
-      "learning_rate": 7.61531100478469e-05,
-      "loss": 0.0509,
-      "step": 1521
-    },
-    {
-      "epoch": 138.36,
-      "learning_rate": 7.613397129186604e-05,
-      "loss": 0.0567,
-      "step": 1522
-    },
-    {
-      "epoch": 138.45,
-      "learning_rate": 7.611483253588517e-05,
-      "loss": 0.0486,
-      "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.0439,
-      "step": 1525
-    },
-    {
-      "epoch": 138.73,
-      "learning_rate": 7.605741626794258e-05,
-      "loss": 0.0462,
-      "step": 1526
-    },
-    {
-      "epoch": 138.82,
-      "learning_rate": 7.603827751196173e-05,
-      "loss": 0.0517,
-      "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.0467,
-      "step": 1529
-    },
-    {
-      "epoch": 139.09,
-      "learning_rate": 7.598086124401914e-05,
-      "loss": 0.055,
-      "step": 1530
-    },
-    {
-      "epoch": 139.18,
-      "learning_rate": 7.596172248803828e-05,
-      "loss": 0.0508,
-      "step": 1531
-    },
-    {
-      "epoch": 139.27,
-      "learning_rate": 7.594258373205741e-05,
-      "loss": 0.0421,
-      "step": 1532
-    },
-    {
-      "epoch": 139.36,
-      "learning_rate": 7.592344497607656e-05,
-      "loss": 0.0627,
-      "step": 1533
-    },
-    {
-      "epoch": 139.45,
-      "learning_rate": 7.59043062200957e-05,
-      "loss": 0.0577,
-      "step": 1534
-    },
-    {
-      "epoch": 139.55,
-      "learning_rate": 7.588516746411484e-05,
-      "loss": 0.0447,
-      "step": 1535
-    },
-    {
-      "epoch": 139.64,
-      "learning_rate": 7.586602870813397e-05,
-      "loss": 0.0496,
-      "step": 1536
-    },
-    {
-      "epoch": 139.73,
-      "learning_rate": 7.584688995215311e-05,
-      "loss": 0.0503,
-      "step": 1537
-    },
-    {
-      "epoch": 139.82,
-      "learning_rate": 7.582775119617225e-05,
-      "loss": 0.043,
-      "step": 1538
-    },
-    {
-      "epoch": 139.91,
-      "learning_rate": 7.580861244019139e-05,
-      "loss": 0.0497,
-      "step": 1539
-    },
-    {
-      "epoch": 140.0,
-      "learning_rate": 7.578947368421054e-05,
-      "loss": 0.0485,
-      "step": 1540
-    },
-    {
-      "epoch": 140.0,
-      "eval_accuracy_NAT": 0.9925677025288269,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.938391937422207,
-      "eval_accuracy_concrete tiles": 0.9894209013636641,
-      "eval_accuracy_dark bricks": 0.8532773950194373,
-      "eval_accuracy_grass": 0.8471569323039176,
-      "eval_accuracy_red bricks": 0.8936434659090909,
-      "eval_accuracy_road": 0.869190204924624,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.42907948060540896,
-      "eval_accuracy_speedway bricks": 0.9893745919454713,
-      "eval_accuracy_steel": 0.8739858848360926,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9780978808871226,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8597929275861009,
-      "eval_iou_concrete tiles": 0.9145893392120287,
-      "eval_iou_dark bricks": 0.8104481977041749,
-      "eval_iou_grass": 0.7874481826569465,
-      "eval_iou_red bricks": 0.825894322284214,
-      "eval_iou_road": 0.8610708146408955,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2891378253357855,
-      "eval_iou_speedway bricks": 0.9730299591951063,
-      "eval_iou_steel": 0.8191266584757092,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12985126674175262,
-      "eval_mean_accuracy": 0.6673914228352877,
-      "eval_mean_iou": 0.6245104698444681,
-      "eval_overall_accuracy": 0.9649887084960938,
-      "eval_runtime": 3.4436,
-      "eval_samples_per_second": 10.454,
-      "eval_steps_per_second": 0.581,
-      "step": 1540
-    },
-    {
-      "epoch": 140.09,
-      "learning_rate": 7.577033492822967e-05,
-      "loss": 0.0461,
-      "step": 1541
-    },
-    {
-      "epoch": 140.18,
-      "learning_rate": 7.57511961722488e-05,
-      "loss": 0.0441,
-      "step": 1542
-    },
-    {
-      "epoch": 140.27,
-      "learning_rate": 7.573205741626795e-05,
-      "loss": 0.0498,
-      "step": 1543
-    },
-    {
-      "epoch": 140.36,
-      "learning_rate": 7.571291866028708e-05,
-      "loss": 0.0461,
-      "step": 1544
-    },
-    {
-      "epoch": 140.45,
-      "learning_rate": 7.569377990430622e-05,
-      "loss": 0.0556,
-      "step": 1545
-    },
-    {
-      "epoch": 140.55,
-      "learning_rate": 7.567464114832537e-05,
-      "loss": 0.0416,
-      "step": 1546
-    },
-    {
-      "epoch": 140.64,
-      "learning_rate": 7.56555023923445e-05,
-      "loss": 0.0496,
-      "step": 1547
-    },
-    {
-      "epoch": 140.73,
-      "learning_rate": 7.563636363636363e-05,
-      "loss": 0.0498,
-      "step": 1548
-    },
-    {
-      "epoch": 140.82,
-      "learning_rate": 7.561722488038278e-05,
-      "loss": 0.0529,
-      "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.0465,
-      "step": 1551
-    },
-    {
-      "epoch": 141.09,
-      "learning_rate": 7.55598086124402e-05,
-      "loss": 0.0526,
-      "step": 1552
-    },
-    {
-      "epoch": 141.18,
-      "learning_rate": 7.554066985645933e-05,
-      "loss": 0.0483,
-      "step": 1553
-    },
-    {
-      "epoch": 141.27,
-      "learning_rate": 7.552153110047847e-05,
-      "loss": 0.0455,
-      "step": 1554
-    },
-    {
-      "epoch": 141.36,
-      "learning_rate": 7.550239234449761e-05,
-      "loss": 0.0484,
-      "step": 1555
-    },
-    {
-      "epoch": 141.45,
-      "learning_rate": 7.548325358851674e-05,
-      "loss": 0.0439,
-      "step": 1556
-    },
-    {
-      "epoch": 141.55,
-      "learning_rate": 7.546411483253589e-05,
-      "loss": 0.0494,
-      "step": 1557
-    },
-    {
-      "epoch": 141.64,
-      "learning_rate": 7.544497607655504e-05,
-      "loss": 0.0404,
-      "step": 1558
-    },
-    {
-      "epoch": 141.73,
-      "learning_rate": 7.542583732057417e-05,
-      "loss": 0.0526,
-      "step": 1559
-    },
-    {
-      "epoch": 141.82,
-      "learning_rate": 7.54066985645933e-05,
-      "loss": 0.0461,
-      "step": 1560
-    },
-    {
-      "epoch": 141.82,
-      "eval_accuracy_NAT": 0.991451649218004,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9410570196145758,
-      "eval_accuracy_concrete tiles": 0.9919280865896166,
-      "eval_accuracy_dark bricks": 0.8720603517955429,
-      "eval_accuracy_grass": 0.8722340860557045,
-      "eval_accuracy_red bricks": 0.9137073863636364,
-      "eval_accuracy_road": 0.8644634075263578,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4308163096518071,
-      "eval_accuracy_speedway bricks": 0.9901047153906402,
-      "eval_accuracy_steel": 0.867761031594222,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9786343922694996,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.859771878400502,
-      "eval_iou_concrete tiles": 0.9039538577279946,
-      "eval_iou_dark bricks": 0.8170278749414013,
-      "eval_iou_grass": 0.7986258672333018,
-      "eval_iou_red bricks": 0.8112880340532871,
-      "eval_iou_road": 0.8566190567484663,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.28941032669086597,
-      "eval_iou_speedway bricks": 0.972295715417493,
-      "eval_iou_steel": 0.8163330204996548,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.142292320728302,
-      "eval_mean_accuracy": 0.671968003369239,
-      "eval_mean_iou": 0.6233815403063436,
-      "eval_overall_accuracy": 0.9650023778279623,
-      "eval_runtime": 3.2163,
-      "eval_samples_per_second": 11.193,
-      "eval_steps_per_second": 0.622,
-      "step": 1560
-    },
-    {
-      "epoch": 141.91,
-      "learning_rate": 7.538755980861245e-05,
-      "loss": 0.0582,
-      "step": 1561
-    },
-    {
-      "epoch": 142.0,
-      "learning_rate": 7.536842105263158e-05,
-      "loss": 0.047,
-      "step": 1562
-    },
-    {
-      "epoch": 142.09,
-      "learning_rate": 7.534928229665072e-05,
-      "loss": 0.0476,
-      "step": 1563
-    },
-    {
-      "epoch": 142.18,
-      "learning_rate": 7.533014354066987e-05,
-      "loss": 0.049,
-      "step": 1564
-    },
-    {
-      "epoch": 142.27,
-      "learning_rate": 7.5311004784689e-05,
-      "loss": 0.0479,
-      "step": 1565
-    },
-    {
-      "epoch": 142.36,
-      "learning_rate": 7.529186602870813e-05,
-      "loss": 0.0457,
-      "step": 1566
-    },
-    {
-      "epoch": 142.45,
-      "learning_rate": 7.527272727272728e-05,
-      "loss": 0.0469,
-      "step": 1567
-    },
-    {
-      "epoch": 142.55,
-      "learning_rate": 7.525358851674642e-05,
-      "loss": 0.0482,
-      "step": 1568
-    },
-    {
-      "epoch": 142.64,
-      "learning_rate": 7.523444976076556e-05,
-      "loss": 0.0488,
-      "step": 1569
-    },
-    {
-      "epoch": 142.73,
-      "learning_rate": 7.52153110047847e-05,
-      "loss": 0.0446,
-      "step": 1570
-    },
-    {
-      "epoch": 142.82,
-      "learning_rate": 7.519617224880383e-05,
-      "loss": 0.0489,
-      "step": 1571
-    },
-    {
-      "epoch": 142.91,
-      "learning_rate": 7.517703349282296e-05,
-      "loss": 0.0525,
-      "step": 1572
-    },
-    {
-      "epoch": 143.0,
-      "learning_rate": 7.515789473684211e-05,
-      "loss": 0.0437,
-      "step": 1573
-    },
-    {
-      "epoch": 143.09,
-      "learning_rate": 7.513875598086126e-05,
-      "loss": 0.0448,
-      "step": 1574
-    },
-    {
-      "epoch": 143.18,
-      "learning_rate": 7.511961722488039e-05,
-      "loss": 0.0512,
-      "step": 1575
-    },
-    {
-      "epoch": 143.27,
-      "learning_rate": 7.510047846889952e-05,
-      "loss": 0.0441,
-      "step": 1576
-    },
-    {
-      "epoch": 143.36,
-      "learning_rate": 7.508133971291867e-05,
-      "loss": 0.0445,
-      "step": 1577
-    },
-    {
-      "epoch": 143.45,
-      "learning_rate": 7.50622009569378e-05,
-      "loss": 0.0464,
-      "step": 1578
-    },
-    {
-      "epoch": 143.55,
-      "learning_rate": 7.504306220095694e-05,
-      "loss": 0.0548,
-      "step": 1579
-    },
-    {
-      "epoch": 143.64,
-      "learning_rate": 7.502392344497609e-05,
-      "loss": 0.0531,
-      "step": 1580
-    },
-    {
-      "epoch": 143.64,
-      "eval_accuracy_NAT": 0.9913203695874502,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.944383342632731,
-      "eval_accuracy_concrete tiles": 0.9780162661285391,
-      "eval_accuracy_dark bricks": 0.8504676494361264,
-      "eval_accuracy_grass": 0.9010684440281423,
-      "eval_accuracy_red bricks": 0.9137073863636364,
-      "eval_accuracy_road": 0.8707246909564479,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5600860143908692,
-      "eval_accuracy_speedway bricks": 0.9884562052648538,
-      "eval_accuracy_steel": 0.8899314029417584,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9788353777925833,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8786886473399808,
-      "eval_iou_concrete tiles": 0.9436528306339794,
-      "eval_iou_dark bricks": 0.8183703703703704,
-      "eval_iou_grass": 0.8176818535022756,
-      "eval_iou_red bricks": 0.8045653533458411,
-      "eval_iou_road": 0.8641779437673117,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3641382277028965,
-      "eval_iou_speedway bricks": 0.973649830961404,
-      "eval_iou_steel": 0.8223158440054548,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13118228316307068,
-      "eval_mean_accuracy": 0.6837047516715812,
-      "eval_mean_iou": 0.6358520214940074,
-      "eval_overall_accuracy": 0.9669938617282443,
-      "eval_runtime": 3.6546,
-      "eval_samples_per_second": 9.851,
-      "eval_steps_per_second": 0.547,
-      "step": 1580
-    },
-    {
-      "epoch": 143.73,
-      "learning_rate": 7.500478468899522e-05,
-      "loss": 0.0522,
-      "step": 1581
-    },
-    {
-      "epoch": 143.82,
-      "learning_rate": 7.498564593301435e-05,
-      "loss": 0.0424,
-      "step": 1582
-    },
-    {
-      "epoch": 143.91,
-      "learning_rate": 7.49665071770335e-05,
-      "loss": 0.058,
-      "step": 1583
-    },
-    {
-      "epoch": 144.0,
-      "learning_rate": 7.494736842105263e-05,
-      "loss": 0.044,
-      "step": 1584
-    },
-    {
-      "epoch": 144.09,
-      "learning_rate": 7.492822966507178e-05,
-      "loss": 0.0435,
-      "step": 1585
-    },
-    {
-      "epoch": 144.18,
-      "learning_rate": 7.490909090909092e-05,
-      "loss": 0.0521,
-      "step": 1586
-    },
-    {
-      "epoch": 144.27,
-      "learning_rate": 7.488995215311005e-05,
-      "loss": 0.044,
-      "step": 1587
-    },
-    {
-      "epoch": 144.36,
-      "learning_rate": 7.487081339712919e-05,
-      "loss": 0.053,
-      "step": 1588
-    },
-    {
-      "epoch": 144.45,
-      "learning_rate": 7.485167464114833e-05,
-      "loss": 0.0458,
-      "step": 1589
-    },
-    {
-      "epoch": 144.55,
-      "learning_rate": 7.483253588516746e-05,
-      "loss": 0.0496,
-      "step": 1590
-    },
-    {
-      "epoch": 144.64,
-      "learning_rate": 7.481339712918661e-05,
-      "loss": 0.0546,
-      "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.0533,
-      "step": 1593
-    },
-    {
-      "epoch": 144.91,
-      "learning_rate": 7.475598086124402e-05,
-      "loss": 0.0452,
-      "step": 1594
-    },
-    {
-      "epoch": 145.0,
-      "learning_rate": 7.473684210526316e-05,
-      "loss": 0.0404,
-      "step": 1595
-    },
-    {
-      "epoch": 145.09,
-      "learning_rate": 7.47177033492823e-05,
-      "loss": 0.0451,
-      "step": 1596
-    },
-    {
-      "epoch": 145.18,
-      "learning_rate": 7.469856459330144e-05,
-      "loss": 0.0529,
-      "step": 1597
-    },
-    {
-      "epoch": 145.27,
-      "learning_rate": 7.467942583732059e-05,
-      "loss": 0.0474,
-      "step": 1598
-    },
-    {
-      "epoch": 145.36,
-      "learning_rate": 7.466028708133972e-05,
-      "loss": 0.0456,
-      "step": 1599
-    },
-    {
-      "epoch": 145.45,
-      "learning_rate": 7.464114832535885e-05,
-      "loss": 0.0518,
-      "step": 1600
-    },
-    {
-      "epoch": 145.45,
-      "eval_accuracy_NAT": 0.9926781609276529,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9449788617537234,
-      "eval_accuracy_concrete tiles": 0.9849263132147007,
-      "eval_accuracy_dark bricks": 0.874831607713329,
-      "eval_accuracy_grass": 0.872125738771202,
-      "eval_accuracy_red bricks": 0.9142400568181818,
-      "eval_accuracy_road": 0.8686182491381363,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5539381909409202,
-      "eval_accuracy_speedway bricks": 0.9884899961019856,
-      "eval_accuracy_steel": 0.8684041290152364,
-      "eval_accuracy_tiles": 0.10299958909738392,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9777273119888188,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8809445810128513,
-      "eval_iou_concrete tiles": 0.9252355238970589,
-      "eval_iou_dark bricks": 0.8196538045438153,
-      "eval_iou_grass": 0.816802838662372,
-      "eval_iou_red bricks": 0.7990378646803228,
-      "eval_iou_road": 0.8620666481053005,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.37566137566137564,
-      "eval_iou_speedway bricks": 0.9731399938814123,
-      "eval_iou_steel": 0.8153885456825677,
-      "eval_iou_tiles": 0.10299958909738392,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13914626836776733,
-      "eval_mean_accuracy": 0.6897100687301887,
-      "eval_mean_iou": 0.6422044674779446,
-      "eval_overall_accuracy": 0.9667505688137479,
-      "eval_runtime": 3.4608,
-      "eval_samples_per_second": 10.402,
-      "eval_steps_per_second": 0.578,
-      "step": 1600
-    },
-    {
-      "epoch": 145.55,
-      "learning_rate": 7.4622009569378e-05,
-      "loss": 0.0425,
-      "step": 1601
-    },
-    {
-      "epoch": 145.64,
-      "learning_rate": 7.460287081339713e-05,
-      "loss": 0.0455,
-      "step": 1602
-    },
-    {
-      "epoch": 145.73,
-      "learning_rate": 7.458373205741626e-05,
-      "loss": 0.0501,
-      "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.0458,
-      "step": 1606
-    },
-    {
-      "epoch": 146.09,
-      "learning_rate": 7.450717703349283e-05,
-      "loss": 0.0459,
-      "step": 1607
-    },
-    {
-      "epoch": 146.18,
-      "learning_rate": 7.448803827751196e-05,
-      "loss": 0.0504,
-      "step": 1608
-    },
-    {
-      "epoch": 146.27,
-      "learning_rate": 7.446889952153109e-05,
-      "loss": 0.0517,
-      "step": 1609
-    },
-    {
-      "epoch": 146.36,
-      "learning_rate": 7.444976076555025e-05,
-      "loss": 0.0576,
-      "step": 1610
-    },
-    {
-      "epoch": 146.45,
-      "learning_rate": 7.443062200956938e-05,
-      "loss": 0.0463,
-      "step": 1611
-    },
-    {
-      "epoch": 146.55,
-      "learning_rate": 7.441148325358852e-05,
-      "loss": 0.0561,
-      "step": 1612
-    },
-    {
-      "epoch": 146.64,
-      "learning_rate": 7.439234449760766e-05,
-      "loss": 0.0455,
-      "step": 1613
-    },
-    {
-      "epoch": 146.73,
-      "learning_rate": 7.43732057416268e-05,
-      "loss": 0.0434,
-      "step": 1614
-    },
-    {
-      "epoch": 146.82,
-      "learning_rate": 7.435406698564593e-05,
-      "loss": 0.0422,
-      "step": 1615
-    },
-    {
-      "epoch": 146.91,
-      "learning_rate": 7.433492822966508e-05,
-      "loss": 0.0542,
-      "step": 1616
-    },
-    {
-      "epoch": 147.0,
-      "learning_rate": 7.431578947368422e-05,
-      "loss": 0.0513,
-      "step": 1617
-    },
-    {
-      "epoch": 147.09,
-      "learning_rate": 7.429665071770335e-05,
-      "loss": 0.0518,
-      "step": 1618
-    },
-    {
-      "epoch": 147.18,
-      "learning_rate": 7.42775119617225e-05,
-      "loss": 0.0516,
-      "step": 1619
-    },
-    {
-      "epoch": 147.27,
-      "learning_rate": 7.425837320574163e-05,
-      "loss": 0.0528,
-      "step": 1620
-    },
-    {
-      "epoch": 147.27,
-      "eval_accuracy_NAT": 0.9917383822820494,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9486404995922572,
-      "eval_accuracy_concrete tiles": 0.986730263560203,
-      "eval_accuracy_dark bricks": 0.8604364728070513,
-      "eval_accuracy_grass": 0.8674528252428202,
-      "eval_accuracy_red bricks": 0.8934659090909091,
-      "eval_accuracy_road": 0.8694513940364238,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4855126402558377,
-      "eval_accuracy_speedway bricks": 0.9869428584875946,
-      "eval_accuracy_steel": 0.8872435855154673,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.978853656399956,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8640311174147332,
-      "eval_iou_concrete tiles": 0.9303773748090063,
-      "eval_iou_dark bricks": 0.8232672902703101,
-      "eval_iou_grass": 0.8084515208568134,
-      "eval_iou_red bricks": 0.8082235785415998,
-      "eval_iou_road": 0.8611510998157991,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3206839412203871,
-      "eval_iou_speedway bricks": 0.9735606541296188,
-      "eval_iou_steel": 0.8265701930240954,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14763912558555603,
-      "eval_mean_accuracy": 0.675201140836201,
-      "eval_mean_iou": 0.6303977251140244,
-      "eval_overall_accuracy": 0.9659091101752387,
-      "eval_runtime": 3.5858,
-      "eval_samples_per_second": 10.04,
-      "eval_steps_per_second": 0.558,
-      "step": 1620
-    },
-    {
-      "epoch": 147.36,
-      "learning_rate": 7.423923444976076e-05,
-      "loss": 0.0456,
-      "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.0415,
-      "step": 1624
-    },
-    {
-      "epoch": 147.73,
-      "learning_rate": 7.416267942583733e-05,
-      "loss": 0.0477,
-      "step": 1625
-    },
-    {
-      "epoch": 147.82,
-      "learning_rate": 7.414354066985646e-05,
-      "loss": 0.0488,
-      "step": 1626
-    },
-    {
-      "epoch": 147.91,
-      "learning_rate": 7.412440191387559e-05,
-      "loss": 0.0503,
-      "step": 1627
-    },
-    {
-      "epoch": 148.0,
-      "learning_rate": 7.410526315789475e-05,
-      "loss": 0.0441,
-      "step": 1628
-    },
-    {
-      "epoch": 148.09,
-      "learning_rate": 7.408612440191388e-05,
-      "loss": 0.0451,
-      "step": 1629
-    },
-    {
-      "epoch": 148.18,
-      "learning_rate": 7.406698564593301e-05,
-      "loss": 0.0442,
-      "step": 1630
-    },
-    {
-      "epoch": 148.27,
-      "learning_rate": 7.404784688995216e-05,
-      "loss": 0.0447,
-      "step": 1631
-    },
-    {
-      "epoch": 148.36,
-      "learning_rate": 7.402870813397129e-05,
-      "loss": 0.0432,
-      "step": 1632
-    },
-    {
-      "epoch": 148.45,
-      "learning_rate": 7.400956937799042e-05,
-      "loss": 0.0453,
-      "step": 1633
-    },
-    {
-      "epoch": 148.55,
-      "learning_rate": 7.399043062200958e-05,
-      "loss": 0.0468,
-      "step": 1634
-    },
-    {
-      "epoch": 148.64,
-      "learning_rate": 7.397129186602871e-05,
-      "loss": 0.0483,
-      "step": 1635
-    },
-    {
-      "epoch": 148.73,
-      "learning_rate": 7.395215311004785e-05,
-      "loss": 0.0453,
-      "step": 1636
-    },
-    {
-      "epoch": 148.82,
-      "learning_rate": 7.393301435406699e-05,
-      "loss": 0.049,
-      "step": 1637
-    },
-    {
-      "epoch": 148.91,
-      "learning_rate": 7.391387559808612e-05,
-      "loss": 0.0516,
-      "step": 1638
-    },
-    {
-      "epoch": 149.0,
-      "learning_rate": 7.389473684210527e-05,
-      "loss": 0.0739,
-      "step": 1639
-    },
-    {
-      "epoch": 149.09,
-      "learning_rate": 7.387559808612442e-05,
-      "loss": 0.0474,
-      "step": 1640
-    },
-    {
-      "epoch": 149.09,
-      "eval_accuracy_NAT": 0.9926737496497445,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.944897044937551,
-      "eval_accuracy_concrete tiles": 0.9866385372714487,
-      "eval_accuracy_dark bricks": 0.8567414649166699,
-      "eval_accuracy_grass": 0.8743835563772863,
-      "eval_accuracy_red bricks": 0.94140625,
-      "eval_accuracy_road": 0.8710149010806699,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.43875609957819867,
-      "eval_accuracy_speedway bricks": 0.9850415204911257,
-      "eval_accuracy_steel": 0.9085977178286393,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9788229186077324,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8704061776061776,
-      "eval_iou_concrete tiles": 0.9320105132426422,
-      "eval_iou_dark bricks": 0.8185863489261548,
-      "eval_iou_grass": 0.8091204579634211,
-      "eval_iou_red bricks": 0.7523768979707677,
-      "eval_iou_road": 0.8633459302221056,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.29556513018608627,
-      "eval_iou_speedway bricks": 0.9739605240470638,
-      "eval_iou_steel": 0.8236569105204938,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12765635550022125,
-      "eval_mean_accuracy": 0.6769346801639488,
-      "eval_mean_iou": 0.6244501391763573,
-      "eval_overall_accuracy": 0.9663011762830946,
-      "eval_runtime": 3.2909,
-      "eval_samples_per_second": 10.939,
-      "eval_steps_per_second": 0.608,
-      "step": 1640
-    },
-    {
-      "epoch": 149.18,
-      "learning_rate": 7.385645933014355e-05,
-      "loss": 0.0496,
-      "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.0405,
-      "step": 1643
-    },
-    {
-      "epoch": 149.45,
-      "learning_rate": 7.379904306220096e-05,
-      "loss": 0.0503,
-      "step": 1644
-    },
-    {
-      "epoch": 149.55,
-      "learning_rate": 7.37799043062201e-05,
-      "loss": 0.048,
-      "step": 1645
-    },
-    {
-      "epoch": 149.64,
-      "learning_rate": 7.376076555023925e-05,
-      "loss": 0.0438,
-      "step": 1646
-    },
-    {
-      "epoch": 149.73,
-      "learning_rate": 7.374162679425838e-05,
-      "loss": 0.0461,
-      "step": 1647
-    },
-    {
-      "epoch": 149.82,
-      "learning_rate": 7.372248803827751e-05,
-      "loss": 0.0451,
-      "step": 1648
-    },
-    {
-      "epoch": 149.91,
-      "learning_rate": 7.370334928229666e-05,
-      "loss": 0.0495,
-      "step": 1649
-    },
-    {
-      "epoch": 150.0,
-      "learning_rate": 7.368421052631579e-05,
-      "loss": 0.0502,
-      "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.0485,
-      "step": 1652
-    },
-    {
-      "epoch": 150.27,
-      "learning_rate": 7.362679425837321e-05,
-      "loss": 0.043,
-      "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.0541,
-      "step": 1655
-    },
-    {
-      "epoch": 150.55,
-      "learning_rate": 7.356937799043062e-05,
-      "loss": 0.0518,
-      "step": 1656
-    },
-    {
-      "epoch": 150.64,
-      "learning_rate": 7.355023923444977e-05,
-      "loss": 0.0464,
-      "step": 1657
-    },
-    {
-      "epoch": 150.73,
-      "learning_rate": 7.35311004784689e-05,
-      "loss": 0.0461,
-      "step": 1658
-    },
-    {
-      "epoch": 150.82,
-      "learning_rate": 7.351196172248805e-05,
-      "loss": 0.0484,
-      "step": 1659
-    },
-    {
-      "epoch": 150.91,
-      "learning_rate": 7.349282296650718e-05,
-      "loss": 0.0451,
-      "step": 1660
-    },
-    {
-      "epoch": 150.91,
-      "eval_accuracy_NAT": 0.9926852189723063,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9469102751191039,
-      "eval_accuracy_concrete tiles": 0.979178132452761,
-      "eval_accuracy_dark bricks": 0.8765636426619453,
-      "eval_accuracy_grass": 0.8896046023130397,
-      "eval_accuracy_red bricks": 0.9186789772727273,
-      "eval_accuracy_road": 0.948738009182732,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6197998511289389,
-      "eval_accuracy_speedway bricks": 0.986338243866058,
-      "eval_accuracy_steel": 0.8970961677989578,
-      "eval_accuracy_tiles": 0.0726612792768114,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9782830860456725,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8878484339728232,
-      "eval_iou_concrete tiles": 0.9373079287031346,
-      "eval_iou_dark bricks": 0.8248161964434465,
-      "eval_iou_grass": 0.824290451701491,
-      "eval_iou_red bricks": 0.8227063126093178,
-      "eval_iou_road": 0.9420275238390257,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.4160328651529451,
-      "eval_iou_speedway bricks": 0.9745461027706913,
-      "eval_iou_steel": 0.8275869937250427,
-      "eval_iou_tiles": 0.0726612792768114,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08876092731952667,
-      "eval_mean_accuracy": 0.7021734153881062,
-      "eval_mean_iou": 0.6544697826338771,
-      "eval_overall_accuracy": 0.9746428595648872,
-      "eval_runtime": 3.67,
-      "eval_samples_per_second": 9.809,
-      "eval_steps_per_second": 0.545,
-      "step": 1660
-    },
-    {
-      "epoch": 151.0,
-      "learning_rate": 7.347368421052632e-05,
-      "loss": 0.0499,
-      "step": 1661
-    },
-    {
-      "epoch": 151.09,
-      "learning_rate": 7.345454545454545e-05,
-      "loss": 0.0439,
-      "step": 1662
-    },
-    {
-      "epoch": 151.18,
-      "learning_rate": 7.34354066985646e-05,
-      "loss": 0.0504,
-      "step": 1663
-    },
-    {
-      "epoch": 151.27,
-      "learning_rate": 7.341626794258373e-05,
-      "loss": 0.0599,
-      "step": 1664
-    },
-    {
-      "epoch": 151.36,
-      "learning_rate": 7.339712918660288e-05,
-      "loss": 0.0468,
-      "step": 1665
-    },
-    {
-      "epoch": 151.45,
-      "learning_rate": 7.337799043062201e-05,
-      "loss": 0.0426,
-      "step": 1666
-    },
-    {
-      "epoch": 151.55,
-      "learning_rate": 7.335885167464116e-05,
-      "loss": 0.0439,
-      "step": 1667
-    },
-    {
-      "epoch": 151.64,
-      "learning_rate": 7.333971291866029e-05,
-      "loss": 0.0485,
-      "step": 1668
-    },
-    {
-      "epoch": 151.73,
-      "learning_rate": 7.332057416267943e-05,
-      "loss": 0.0497,
-      "step": 1669
-    },
-    {
-      "epoch": 151.82,
-      "learning_rate": 7.330143540669856e-05,
-      "loss": 0.0513,
-      "step": 1670
-    },
-    {
-      "epoch": 151.91,
-      "learning_rate": 7.328229665071771e-05,
-      "loss": 0.0456,
-      "step": 1671
-    },
-    {
-      "epoch": 152.0,
-      "learning_rate": 7.326315789473684e-05,
-      "loss": 0.0454,
-      "step": 1672
-    },
-    {
-      "epoch": 152.09,
-      "learning_rate": 7.324401913875599e-05,
-      "loss": 0.0437,
-      "step": 1673
-    },
-    {
-      "epoch": 152.18,
-      "learning_rate": 7.322488038277512e-05,
-      "loss": 0.0461,
-      "step": 1674
-    },
-    {
-      "epoch": 152.27,
-      "learning_rate": 7.320574162679427e-05,
-      "loss": 0.0585,
-      "step": 1675
-    },
-    {
-      "epoch": 152.36,
-      "learning_rate": 7.31866028708134e-05,
-      "loss": 0.0417,
-      "step": 1676
-    },
-    {
-      "epoch": 152.45,
-      "learning_rate": 7.316746411483254e-05,
-      "loss": 0.0573,
-      "step": 1677
-    },
-    {
-      "epoch": 152.55,
-      "learning_rate": 7.314832535885168e-05,
-      "loss": 0.0476,
-      "step": 1678
-    },
-    {
-      "epoch": 152.64,
-      "learning_rate": 7.312918660287081e-05,
-      "loss": 0.0431,
-      "step": 1679
-    },
-    {
-      "epoch": 152.73,
-      "learning_rate": 7.311004784688995e-05,
-      "loss": 0.0453,
-      "step": 1680
-    },
-    {
-      "epoch": 152.73,
-      "eval_accuracy_NAT": 0.99301200643976,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9433519142452466,
-      "eval_accuracy_concrete tiles": 0.9864856601235248,
-      "eval_accuracy_dark bricks": 0.8579731342134637,
-      "eval_accuracy_grass": 0.8838831666765694,
-      "eval_accuracy_red bricks": 0.9005681818181818,
-      "eval_accuracy_road": 0.9243289193179578,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.39125520359496047,
-      "eval_accuracy_speedway bricks": 0.9866888238013002,
-      "eval_accuracy_steel": 0.8978134687685508,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9784378831470583,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8747640001691488,
-      "eval_iou_concrete tiles": 0.9313550025980024,
-      "eval_iou_dark bricks": 0.8226068344527271,
-      "eval_iou_grass": 0.8193472929146971,
-      "eval_iou_red bricks": 0.8158275695673154,
-      "eval_iou_road": 0.917705136885982,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2646773591943305,
-      "eval_iou_speedway bricks": 0.9738665478710289,
-      "eval_iou_steel": 0.8246546710287168,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10811636596918106,
-      "eval_mean_accuracy": 0.6742584983845781,
-      "eval_mean_iou": 0.6325570998330006,
-      "eval_overall_accuracy": 0.9712887869940864,
-      "eval_runtime": 3.4095,
-      "eval_samples_per_second": 10.559,
-      "eval_steps_per_second": 0.587,
-      "step": 1680
-    },
-    {
-      "epoch": 152.82,
-      "learning_rate": 7.30909090909091e-05,
-      "loss": 0.0403,
-      "step": 1681
-    },
-    {
-      "epoch": 152.91,
-      "learning_rate": 7.307177033492823e-05,
-      "loss": 0.0588,
-      "step": 1682
-    },
-    {
-      "epoch": 153.0,
-      "learning_rate": 7.305263157894738e-05,
-      "loss": 0.045,
-      "step": 1683
-    },
-    {
-      "epoch": 153.09,
-      "learning_rate": 7.303349282296651e-05,
-      "loss": 0.051,
-      "step": 1684
-    },
-    {
-      "epoch": 153.18,
-      "learning_rate": 7.301435406698564e-05,
-      "loss": 0.0449,
-      "step": 1685
-    },
-    {
-      "epoch": 153.27,
-      "learning_rate": 7.299521531100479e-05,
-      "loss": 0.0492,
-      "step": 1686
-    },
-    {
-      "epoch": 153.36,
-      "learning_rate": 7.297607655502393e-05,
-      "loss": 0.046,
-      "step": 1687
-    },
-    {
-      "epoch": 153.45,
-      "learning_rate": 7.295693779904306e-05,
-      "loss": 0.0436,
-      "step": 1688
-    },
-    {
-      "epoch": 153.55,
-      "learning_rate": 7.293779904306221e-05,
-      "loss": 0.0458,
-      "step": 1689
-    },
-    {
-      "epoch": 153.64,
-      "learning_rate": 7.291866028708134e-05,
-      "loss": 0.0456,
-      "step": 1690
-    },
-    {
-      "epoch": 153.73,
-      "learning_rate": 7.289952153110047e-05,
-      "loss": 0.0443,
-      "step": 1691
-    },
-    {
-      "epoch": 153.82,
-      "learning_rate": 7.288038277511962e-05,
-      "loss": 0.0478,
-      "step": 1692
-    },
-    {
-      "epoch": 153.91,
-      "learning_rate": 7.286124401913876e-05,
-      "loss": 0.044,
-      "step": 1693
-    },
-    {
-      "epoch": 154.0,
-      "learning_rate": 7.28421052631579e-05,
-      "loss": 0.0533,
-      "step": 1694
-    },
-    {
-      "epoch": 154.09,
-      "learning_rate": 7.282296650717704e-05,
-      "loss": 0.0448,
-      "step": 1695
-    },
-    {
-      "epoch": 154.18,
-      "learning_rate": 7.280382775119617e-05,
-      "loss": 0.0454,
-      "step": 1696
-    },
-    {
-      "epoch": 154.27,
-      "learning_rate": 7.27846889952153e-05,
-      "loss": 0.0466,
-      "step": 1697
-    },
-    {
-      "epoch": 154.36,
-      "learning_rate": 7.276555023923445e-05,
-      "loss": 0.0487,
-      "step": 1698
-    },
-    {
-      "epoch": 154.45,
-      "learning_rate": 7.27464114832536e-05,
-      "loss": 0.0538,
-      "step": 1699
-    },
-    {
-      "epoch": 154.55,
-      "learning_rate": 7.272727272727273e-05,
-      "loss": 0.0443,
-      "step": 1700
-    },
-    {
-      "epoch": 154.55,
-      "eval_accuracy_NAT": 0.9924235419667806,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9455247542812997,
-      "eval_accuracy_concrete tiles": 0.9844371063413441,
-      "eval_accuracy_dark bricks": 0.8664408606289211,
-      "eval_accuracy_grass": 0.887276883233083,
-      "eval_accuracy_red bricks": 0.9135298295454546,
-      "eval_accuracy_road": 0.8743208780790992,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.3527968461390015,
-      "eval_accuracy_speedway bricks": 0.987067160495615,
-      "eval_accuracy_steel": 0.898877052964844,
-      "eval_accuracy_tiles": 0.005341734009039857,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9788732284057954,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.87227442469357,
-      "eval_iou_concrete tiles": 0.9309796437659033,
-      "eval_iou_dark bricks": 0.822139439757496,
-      "eval_iou_grass": 0.8204968261560937,
-      "eval_iou_red bricks": 0.8196590728054803,
-      "eval_iou_road": 0.8678564527028243,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.240477309029409,
-      "eval_iou_speedway bricks": 0.9737516637736648,
-      "eval_iou_steel": 0.8236854034451496,
-      "eval_iou_tiles": 0.005341734009039857,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12572631239891052,
-      "eval_mean_accuracy": 0.6698489728988063,
-      "eval_mean_iou": 0.6273488614264943,
-      "eval_overall_accuracy": 0.9667922125922309,
-      "eval_runtime": 3.275,
-      "eval_samples_per_second": 10.992,
-      "eval_steps_per_second": 0.611,
-      "step": 1700
-    },
-    {
-      "epoch": 154.64,
-      "learning_rate": 7.270813397129187e-05,
-      "loss": 0.0484,
-      "step": 1701
-    },
-    {
-      "epoch": 154.73,
-      "learning_rate": 7.2688995215311e-05,
-      "loss": 0.0539,
-      "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.0453,
-      "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.0457,
-      "step": 1706
-    },
-    {
-      "epoch": 155.18,
-      "learning_rate": 7.25933014354067e-05,
-      "loss": 0.0448,
-      "step": 1707
-    },
-    {
-      "epoch": 155.27,
-      "learning_rate": 7.257416267942584e-05,
-      "loss": 0.0492,
-      "step": 1708
-    },
-    {
-      "epoch": 155.36,
-      "learning_rate": 7.255502392344497e-05,
-      "loss": 0.0461,
-      "step": 1709
-    },
-    {
-      "epoch": 155.45,
-      "learning_rate": 7.253588516746413e-05,
-      "loss": 0.0505,
-      "step": 1710
-    },
-    {
-      "epoch": 155.55,
-      "learning_rate": 7.251674641148326e-05,
-      "loss": 0.0449,
-      "step": 1711
-    },
-    {
-      "epoch": 155.64,
-      "learning_rate": 7.24976076555024e-05,
-      "loss": 0.0411,
-      "step": 1712
-    },
-    {
-      "epoch": 155.73,
-      "learning_rate": 7.247846889952154e-05,
-      "loss": 0.0402,
-      "step": 1713
-    },
-    {
-      "epoch": 155.82,
-      "learning_rate": 7.245933014354067e-05,
-      "loss": 0.0483,
-      "step": 1714
-    },
-    {
-      "epoch": 155.91,
-      "learning_rate": 7.24401913875598e-05,
-      "loss": 0.0475,
-      "step": 1715
-    },
-    {
-      "epoch": 156.0,
-      "learning_rate": 7.242105263157896e-05,
-      "loss": 0.0419,
-      "step": 1716
-    },
-    {
-      "epoch": 156.09,
-      "learning_rate": 7.24019138755981e-05,
-      "loss": 0.0486,
-      "step": 1717
-    },
-    {
-      "epoch": 156.18,
-      "learning_rate": 7.238277511961723e-05,
-      "loss": 0.0486,
-      "step": 1718
-    },
-    {
-      "epoch": 156.27,
-      "learning_rate": 7.236363636363637e-05,
-      "loss": 0.0438,
-      "step": 1719
-    },
-    {
-      "epoch": 156.36,
-      "learning_rate": 7.23444976076555e-05,
-      "loss": 0.0417,
-      "step": 1720
-    },
-    {
-      "epoch": 156.36,
-      "eval_accuracy_NAT": 0.9929417788954583,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9489838619683248,
-      "eval_accuracy_concrete tiles": 0.9862410566868465,
-      "eval_accuracy_dark bricks": 0.8794888572418306,
-      "eval_accuracy_grass": 0.8583831090078534,
-      "eval_accuracy_red bricks": 0.900390625,
-      "eval_accuracy_road": 0.8707186449121933,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.40162104044330493,
-      "eval_accuracy_speedway bricks": 0.9863720347031897,
-      "eval_accuracy_steel": 0.8962551942484005,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9789846677739332,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.868326004543318,
-      "eval_iou_concrete tiles": 0.93206576704135,
-      "eval_iou_dark bricks": 0.8122134148508868,
-      "eval_iou_grass": 0.8055510000590392,
-      "eval_iou_red bricks": 0.8281887963416625,
-      "eval_iou_road": 0.8637479742965066,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.26952322806238554,
-      "eval_iou_speedway bricks": 0.9743610682113367,
-      "eval_iou_steel": 0.8308112074104645,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13718970119953156,
-      "eval_mean_accuracy": 0.6708766310082617,
-      "eval_mean_iou": 0.6279825483531448,
-      "eval_overall_accuracy": 0.9662430021497939,
-      "eval_runtime": 3.5774,
-      "eval_samples_per_second": 10.063,
-      "eval_steps_per_second": 0.559,
-      "step": 1720
-    },
-    {
-      "epoch": 156.45,
-      "learning_rate": 7.232535885167464e-05,
-      "loss": 0.0428,
-      "step": 1721
-    },
-    {
-      "epoch": 156.55,
-      "learning_rate": 7.23062200956938e-05,
-      "loss": 0.0391,
-      "step": 1722
-    },
-    {
-      "epoch": 156.64,
-      "learning_rate": 7.228708133971293e-05,
-      "loss": 0.0554,
-      "step": 1723
-    },
-    {
-      "epoch": 156.73,
-      "learning_rate": 7.226794258373206e-05,
-      "loss": 0.0633,
-      "step": 1724
-    },
-    {
-      "epoch": 156.82,
-      "learning_rate": 7.22488038277512e-05,
-      "loss": 0.0501,
-      "step": 1725
-    },
-    {
-      "epoch": 156.91,
-      "learning_rate": 7.222966507177034e-05,
-      "loss": 0.0458,
-      "step": 1726
-    },
-    {
-      "epoch": 157.0,
-      "learning_rate": 7.221052631578947e-05,
-      "loss": 0.044,
-      "step": 1727
-    },
-    {
-      "epoch": 157.09,
-      "learning_rate": 7.219138755980863e-05,
-      "loss": 0.0454,
-      "step": 1728
-    },
-    {
-      "epoch": 157.18,
-      "learning_rate": 7.217224880382776e-05,
-      "loss": 0.0393,
-      "step": 1729
-    },
-    {
-      "epoch": 157.27,
-      "learning_rate": 7.215311004784689e-05,
-      "loss": 0.0477,
-      "step": 1730
-    },
-    {
-      "epoch": 157.36,
-      "learning_rate": 7.213397129186604e-05,
-      "loss": 0.0432,
-      "step": 1731
-    },
-    {
-      "epoch": 157.45,
-      "learning_rate": 7.211483253588517e-05,
-      "loss": 0.0444,
-      "step": 1732
-    },
-    {
-      "epoch": 157.55,
-      "learning_rate": 7.20956937799043e-05,
-      "loss": 0.0561,
-      "step": 1733
-    },
-    {
-      "epoch": 157.64,
-      "learning_rate": 7.207655502392345e-05,
-      "loss": 0.0509,
-      "step": 1734
-    },
-    {
-      "epoch": 157.73,
-      "learning_rate": 7.205741626794259e-05,
-      "loss": 0.0474,
-      "step": 1735
-    },
-    {
-      "epoch": 157.82,
-      "learning_rate": 7.203827751196172e-05,
-      "loss": 0.0418,
-      "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.0617,
-      "step": 1738
-    },
-    {
-      "epoch": 158.09,
-      "learning_rate": 7.198086124401913e-05,
-      "loss": 0.0506,
-      "step": 1739
-    },
-    {
-      "epoch": 158.18,
-      "learning_rate": 7.196172248803828e-05,
-      "loss": 0.0475,
-      "step": 1740
-    },
-    {
-      "epoch": 158.18,
-      "eval_accuracy_NAT": 0.9930121828908762,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9459190845100648,
-      "eval_accuracy_concrete tiles": 0.9752950528954932,
-      "eval_accuracy_dark bricks": 0.8883799699780609,
-      "eval_accuracy_grass": 0.8680469877707372,
-      "eval_accuracy_red bricks": 0.9161931818181818,
-      "eval_accuracy_road": 0.8860647144392838,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.44691643922476776,
-      "eval_accuracy_speedway bricks": 0.9888828145836426,
-      "eval_accuracy_steel": 0.8897829958446013,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9791981999021095,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8702337455932854,
-      "eval_iou_concrete tiles": 0.9425844389941196,
-      "eval_iou_dark bricks": 0.818040049618997,
-      "eval_iou_grass": 0.8095168250739886,
-      "eval_iou_red bricks": 0.8168434383409846,
-      "eval_iou_road": 0.8800320900828194,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.29832535885167466,
-      "eval_iou_speedway bricks": 0.9739895815689456,
-      "eval_iou_steel": 0.8291397444663834,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11910075694322586,
-      "eval_mean_accuracy": 0.6768071864581314,
-      "eval_mean_iou": 0.6321464209610237,
-      "eval_overall_accuracy": 0.968208736843533,
-      "eval_runtime": 3.7624,
-      "eval_samples_per_second": 9.568,
-      "eval_steps_per_second": 0.532,
-      "step": 1740
-    },
-    {
-      "epoch": 158.27,
-      "learning_rate": 7.194258373205742e-05,
-      "loss": 0.0479,
-      "step": 1741
-    },
-    {
-      "epoch": 158.36,
-      "learning_rate": 7.192344497607656e-05,
-      "loss": 0.0411,
-      "step": 1742
-    },
-    {
-      "epoch": 158.45,
-      "learning_rate": 7.19043062200957e-05,
-      "loss": 0.0461,
-      "step": 1743
-    },
-    {
-      "epoch": 158.55,
-      "learning_rate": 7.188516746411483e-05,
-      "loss": 0.0429,
-      "step": 1744
-    },
-    {
-      "epoch": 158.64,
-      "learning_rate": 7.186602870813397e-05,
-      "loss": 0.0409,
-      "step": 1745
-    },
-    {
-      "epoch": 158.73,
-      "learning_rate": 7.184688995215311e-05,
-      "loss": 0.0403,
-      "step": 1746
-    },
-    {
-      "epoch": 158.82,
-      "learning_rate": 7.182775119617226e-05,
-      "loss": 0.0452,
-      "step": 1747
-    },
-    {
-      "epoch": 158.91,
-      "learning_rate": 7.180861244019139e-05,
-      "loss": 0.0459,
-      "step": 1748
-    },
-    {
-      "epoch": 159.0,
-      "learning_rate": 7.178947368421054e-05,
-      "loss": 0.0433,
-      "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.0481,
-      "step": 1751
-    },
-    {
-      "epoch": 159.27,
-      "learning_rate": 7.173205741626794e-05,
-      "loss": 0.0506,
-      "step": 1752
-    },
-    {
-      "epoch": 159.36,
-      "learning_rate": 7.171291866028709e-05,
-      "loss": 0.0462,
-      "step": 1753
-    },
-    {
-      "epoch": 159.45,
-      "learning_rate": 7.169377990430622e-05,
-      "loss": 0.0411,
-      "step": 1754
-    },
-    {
-      "epoch": 159.55,
-      "learning_rate": 7.167464114832535e-05,
-      "loss": 0.045,
-      "step": 1755
-    },
-    {
-      "epoch": 159.64,
-      "learning_rate": 7.16555023923445e-05,
-      "loss": 0.0514,
-      "step": 1756
-    },
-    {
-      "epoch": 159.73,
-      "learning_rate": 7.163636363636363e-05,
-      "loss": 0.0479,
-      "step": 1757
-    },
-    {
-      "epoch": 159.82,
-      "learning_rate": 7.161722488038278e-05,
-      "loss": 0.0461,
-      "step": 1758
-    },
-    {
-      "epoch": 159.91,
-      "learning_rate": 7.159808612440192e-05,
-      "loss": 0.0491,
-      "step": 1759
-    },
-    {
-      "epoch": 160.0,
-      "learning_rate": 7.157894736842105e-05,
-      "loss": 0.0468,
-      "step": 1760
-    },
-    {
-      "epoch": 160.0,
-      "eval_accuracy_NAT": 0.9919785322513821,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9455019528735139,
-      "eval_accuracy_concrete tiles": 0.9856906989543203,
-      "eval_accuracy_dark bricks": 0.8582810515376621,
-      "eval_accuracy_grass": 0.8940992670830464,
-      "eval_accuracy_red bricks": 0.9371448863636364,
-      "eval_accuracy_road": 0.8714707728174688,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4859261709811706,
-      "eval_accuracy_speedway bricks": 0.9858832743803907,
-      "eval_accuracy_steel": 0.9074187058901128,
-      "eval_accuracy_tiles": 0.2383920010957403,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9788417946247757,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8815325302530665,
-      "eval_iou_concrete tiles": 0.9147348409613256,
-      "eval_iou_dark bricks": 0.8043211657769441,
-      "eval_iou_grass": 0.8221080302855013,
-      "eval_iou_red bricks": 0.8352587434720684,
-      "eval_iou_road": 0.8638309195362345,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3264194970184081,
-      "eval_iou_speedway bricks": 0.9743969131495298,
-      "eval_iou_steel": 0.8223668479885229,
-      "eval_iou_tiles": 0.23835935360175295,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13528358936309814,
-      "eval_mean_accuracy": 0.7001374857098804,
-      "eval_mean_iou": 0.6509362028206254,
-      "eval_overall_accuracy": 0.9672504001193576,
-      "eval_runtime": 3.3305,
-      "eval_samples_per_second": 10.809,
-      "eval_steps_per_second": 0.601,
-      "step": 1760
-    },
-    {
-      "epoch": 160.09,
-      "learning_rate": 7.155980861244019e-05,
-      "loss": 0.0413,
-      "step": 1761
-    },
-    {
-      "epoch": 160.18,
-      "learning_rate": 7.154066985645933e-05,
-      "loss": 0.0436,
-      "step": 1762
-    },
-    {
-      "epoch": 160.27,
-      "learning_rate": 7.152153110047846e-05,
-      "loss": 0.0454,
-      "step": 1763
-    },
-    {
-      "epoch": 160.36,
-      "learning_rate": 7.150239234449761e-05,
-      "loss": 0.0448,
-      "step": 1764
-    },
-    {
-      "epoch": 160.45,
-      "learning_rate": 7.148325358851676e-05,
-      "loss": 0.0479,
-      "step": 1765
-    },
-    {
-      "epoch": 160.55,
-      "learning_rate": 7.146411483253589e-05,
-      "loss": 0.0449,
-      "step": 1766
-    },
-    {
-      "epoch": 160.64,
-      "learning_rate": 7.144497607655502e-05,
-      "loss": 0.0491,
-      "step": 1767
-    },
-    {
-      "epoch": 160.73,
-      "learning_rate": 7.142583732057417e-05,
-      "loss": 0.048,
-      "step": 1768
-    },
-    {
-      "epoch": 160.82,
-      "learning_rate": 7.14066985645933e-05,
-      "loss": 0.0407,
-      "step": 1769
-    },
-    {
-      "epoch": 160.91,
-      "learning_rate": 7.138755980861244e-05,
-      "loss": 0.0496,
-      "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.0423,
-      "step": 1772
-    },
-    {
-      "epoch": 161.18,
-      "learning_rate": 7.133014354066985e-05,
-      "loss": 0.0429,
-      "step": 1773
-    },
-    {
-      "epoch": 161.27,
-      "learning_rate": 7.1311004784689e-05,
-      "loss": 0.0469,
-      "step": 1774
-    },
-    {
-      "epoch": 161.36,
-      "learning_rate": 7.129186602870813e-05,
-      "loss": 0.039,
-      "step": 1775
-    },
-    {
-      "epoch": 161.45,
-      "learning_rate": 7.127272727272728e-05,
-      "loss": 0.047,
-      "step": 1776
-    },
-    {
-      "epoch": 161.55,
-      "learning_rate": 7.125358851674642e-05,
-      "loss": 0.041,
-      "step": 1777
-    },
-    {
-      "epoch": 161.64,
-      "learning_rate": 7.123444976076555e-05,
-      "loss": 0.0396,
-      "step": 1778
-    },
-    {
-      "epoch": 161.73,
-      "learning_rate": 7.121531100478468e-05,
-      "loss": 0.043,
-      "step": 1779
-    },
-    {
-      "epoch": 161.82,
-      "learning_rate": 7.119617224880383e-05,
-      "loss": 0.0569,
-      "step": 1780
-    },
-    {
-      "epoch": 161.82,
-      "eval_accuracy_NAT": 0.9930390034605593,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9492856453066655,
-      "eval_accuracy_concrete tiles": 0.9792698587415153,
-      "eval_accuracy_dark bricks": 0.8537777606712598,
-      "eval_accuracy_grass": 0.8787489034206286,
-      "eval_accuracy_red bricks": 0.8966619318181818,
-      "eval_accuracy_road": 0.8813221973259555,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.44752295095525596,
-      "eval_accuracy_speedway bricks": 0.9856455317048564,
-      "eval_accuracy_steel": 0.9042114636237715,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9787156413862954,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8746314288327076,
-      "eval_iou_concrete tiles": 0.9403957954078337,
-      "eval_iou_dark bricks": 0.813839154681538,
-      "eval_iou_grass": 0.8278320135653491,
-      "eval_iou_red bricks": 0.8456128600133959,
-      "eval_iou_road": 0.8728029995485349,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3010180429098596,
-      "eval_iou_speedway bricks": 0.9744170522760448,
-      "eval_iou_steel": 0.8304181242711978,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12074758112430573,
-      "eval_mean_accuracy": 0.674575788232973,
-      "eval_mean_iou": 0.635360239453289,
-      "eval_overall_accuracy": 0.9679258134629991,
-      "eval_runtime": 3.3537,
-      "eval_samples_per_second": 10.734,
-      "eval_steps_per_second": 0.596,
-      "step": 1780
-    },
-    {
-      "epoch": 161.91,
-      "learning_rate": 7.117703349282298e-05,
-      "loss": 0.0444,
-      "step": 1781
-    },
-    {
-      "epoch": 162.0,
-      "learning_rate": 7.115789473684211e-05,
-      "loss": 0.0471,
-      "step": 1782
-    },
-    {
-      "epoch": 162.09,
-      "learning_rate": 7.113875598086125e-05,
-      "loss": 0.0541,
-      "step": 1783
-    },
-    {
-      "epoch": 162.18,
-      "learning_rate": 7.111961722488039e-05,
-      "loss": 0.0454,
-      "step": 1784
-    },
-    {
-      "epoch": 162.27,
-      "learning_rate": 7.110047846889952e-05,
-      "loss": 0.0471,
-      "step": 1785
-    },
-    {
-      "epoch": 162.36,
-      "learning_rate": 7.108133971291866e-05,
-      "loss": 0.0406,
-      "step": 1786
-    },
-    {
-      "epoch": 162.45,
-      "learning_rate": 7.106220095693781e-05,
-      "loss": 0.0432,
-      "step": 1787
-    },
-    {
-      "epoch": 162.55,
-      "learning_rate": 7.104306220095694e-05,
-      "loss": 0.0484,
-      "step": 1788
-    },
-    {
-      "epoch": 162.64,
-      "learning_rate": 7.102392344497609e-05,
-      "loss": 0.0418,
-      "step": 1789
-    },
-    {
-      "epoch": 162.73,
-      "learning_rate": 7.100478468899522e-05,
-      "loss": 0.0493,
-      "step": 1790
-    },
-    {
-      "epoch": 162.82,
-      "learning_rate": 7.098564593301435e-05,
-      "loss": 0.0459,
-      "step": 1791
-    },
-    {
-      "epoch": 162.91,
-      "learning_rate": 7.09665071770335e-05,
-      "loss": 0.0475,
-      "step": 1792
-    },
-    {
-      "epoch": 163.0,
-      "learning_rate": 7.094736842105264e-05,
-      "loss": 0.0628,
-      "step": 1793
-    },
-    {
-      "epoch": 163.09,
-      "learning_rate": 7.092822966507177e-05,
-      "loss": 0.0439,
-      "step": 1794
-    },
-    {
-      "epoch": 163.18,
-      "learning_rate": 7.090909090909092e-05,
-      "loss": 0.0467,
-      "step": 1795
-    },
-    {
-      "epoch": 163.27,
-      "learning_rate": 7.088995215311005e-05,
-      "loss": 0.0547,
-      "step": 1796
-    },
-    {
-      "epoch": 163.36,
-      "learning_rate": 7.087081339712918e-05,
-      "loss": 0.0465,
-      "step": 1797
-    },
-    {
-      "epoch": 163.45,
-      "learning_rate": 7.085167464114833e-05,
-      "loss": 0.0454,
-      "step": 1798
-    },
-    {
-      "epoch": 163.55,
-      "learning_rate": 7.083253588516747e-05,
-      "loss": 0.0468,
-      "step": 1799
-    },
-    {
-      "epoch": 163.64,
-      "learning_rate": 7.08133971291866e-05,
-      "loss": 0.0414,
-      "step": 1800
-    },
-    {
-      "epoch": 163.64,
-      "eval_accuracy_NAT": 0.9933093265707855,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9372478432550754,
-      "eval_accuracy_concrete tiles": 0.9910719745612426,
-      "eval_accuracy_dark bricks": 0.8719833724644933,
-      "eval_accuracy_grass": 0.8813142875117522,
-      "eval_accuracy_red bricks": 0.9128196022727273,
-      "eval_accuracy_road": 0.8683050640457468,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48851763019325667,
-      "eval_accuracy_speedway bricks": 0.9872536135076457,
-      "eval_accuracy_steel": 0.9058109623375766,
-      "eval_accuracy_tiles": 0.1494315847144227,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9779009757340675,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8811723374366657,
-      "eval_iou_concrete tiles": 0.9083622912229571,
-      "eval_iou_dark bricks": 0.8177814677110782,
-      "eval_iou_grass": 0.8256598658166423,
-      "eval_iou_red bricks": 0.8482098663586867,
-      "eval_iou_road": 0.8607135964980624,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3222697099208875,
-      "eval_iou_speedway bricks": 0.9744303817069401,
-      "eval_iou_steel": 0.8261010143543548,
-      "eval_iou_tiles": 0.14934976043805612,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14603829383850098,
-      "eval_mean_accuracy": 0.6913127124180558,
-      "eval_mean_iou": 0.6455347128614153,
-      "eval_overall_accuracy": 0.9668665991889106,
-      "eval_runtime": 3.9858,
-      "eval_samples_per_second": 9.032,
-      "eval_steps_per_second": 0.502,
-      "step": 1800
-    },
-    {
-      "epoch": 163.73,
-      "learning_rate": 7.079425837320575e-05,
-      "loss": 0.0432,
-      "step": 1801
-    },
-    {
-      "epoch": 163.82,
-      "learning_rate": 7.077511961722488e-05,
-      "loss": 0.0726,
-      "step": 1802
-    },
-    {
-      "epoch": 163.91,
-      "learning_rate": 7.075598086124402e-05,
-      "loss": 0.0472,
-      "step": 1803
-    },
-    {
-      "epoch": 164.0,
-      "learning_rate": 7.073684210526316e-05,
-      "loss": 0.0452,
-      "step": 1804
-    },
-    {
-      "epoch": 164.09,
-      "learning_rate": 7.07177033492823e-05,
-      "loss": 0.0485,
-      "step": 1805
-    },
-    {
-      "epoch": 164.18,
-      "learning_rate": 7.069856459330144e-05,
-      "loss": 0.0502,
-      "step": 1806
-    },
-    {
-      "epoch": 164.27,
-      "learning_rate": 7.067942583732058e-05,
-      "loss": 0.0617,
-      "step": 1807
-    },
-    {
-      "epoch": 164.36,
-      "learning_rate": 7.066028708133972e-05,
-      "loss": 0.0436,
-      "step": 1808
-    },
-    {
-      "epoch": 164.45,
-      "learning_rate": 7.064114832535885e-05,
-      "loss": 0.0502,
-      "step": 1809
-    },
-    {
-      "epoch": 164.55,
-      "learning_rate": 7.0622009569378e-05,
-      "loss": 0.0512,
-      "step": 1810
-    },
-    {
-      "epoch": 164.64,
-      "learning_rate": 7.060287081339714e-05,
-      "loss": 0.0466,
-      "step": 1811
-    },
-    {
-      "epoch": 164.73,
-      "learning_rate": 7.058373205741627e-05,
-      "loss": 0.0485,
-      "step": 1812
-    },
-    {
-      "epoch": 164.82,
-      "learning_rate": 7.056459330143542e-05,
-      "loss": 0.0444,
-      "step": 1813
-    },
-    {
-      "epoch": 164.91,
-      "learning_rate": 7.054545454545455e-05,
-      "loss": 0.0472,
-      "step": 1814
-    },
-    {
-      "epoch": 165.0,
-      "learning_rate": 7.052631578947368e-05,
-      "loss": 0.0425,
-      "step": 1815
-    },
-    {
-      "epoch": 165.09,
-      "learning_rate": 7.050717703349283e-05,
-      "loss": 0.0481,
-      "step": 1816
-    },
-    {
-      "epoch": 165.18,
-      "learning_rate": 7.048803827751197e-05,
-      "loss": 0.0431,
-      "step": 1817
-    },
-    {
-      "epoch": 165.27,
-      "learning_rate": 7.04688995215311e-05,
-      "loss": 0.05,
-      "step": 1818
-    },
-    {
-      "epoch": 165.36,
-      "learning_rate": 7.044976076555025e-05,
-      "loss": 0.0479,
-      "step": 1819
-    },
-    {
-      "epoch": 165.45,
-      "learning_rate": 7.043062200956938e-05,
-      "loss": 0.0447,
-      "step": 1820
-    },
-    {
-      "epoch": 165.45,
-      "eval_accuracy_NAT": 0.9922640301576132,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9439326795141423,
-      "eval_accuracy_concrete tiles": 0.985598972665566,
-      "eval_accuracy_dark bricks": 0.8512374427466225,
-      "eval_accuracy_grass": 0.8828311494947871,
-      "eval_accuracy_red bricks": 0.9451349431818182,
-      "eval_accuracy_road": 0.915026475627791,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5713340501199239,
-      "eval_accuracy_speedway bricks": 0.9875836775774867,
-      "eval_accuracy_steel": 0.8870539542246554,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9784177243160237,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8773594667271714,
-      "eval_iou_concrete tiles": 0.9301419667590027,
-      "eval_iou_dark bricks": 0.8094872076424728,
-      "eval_iou_grass": 0.8214994243490591,
-      "eval_iou_red bricks": 0.8043215472952554,
-      "eval_iou_road": 0.9049644694576061,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3752172653534183,
-      "eval_iou_speedway bricks": 0.9742733920870055,
-      "eval_iou_steel": 0.8290106333795654,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10441258549690247,
-      "eval_mean_accuracy": 0.6893844134854159,
-      "eval_mean_iou": 0.6388225459512754,
-      "eval_overall_accuracy": 0.97075441148546,
-      "eval_runtime": 3.4346,
-      "eval_samples_per_second": 10.482,
-      "eval_steps_per_second": 0.582,
-      "step": 1820
-    },
-    {
-      "epoch": 165.55,
-      "learning_rate": 7.041148325358851e-05,
-      "loss": 0.0471,
-      "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.0415,
-      "step": 1823
-    },
-    {
-      "epoch": 165.82,
-      "learning_rate": 7.035406698564594e-05,
-      "loss": 0.0439,
-      "step": 1824
-    },
-    {
-      "epoch": 165.91,
-      "learning_rate": 7.033492822966508e-05,
-      "loss": 0.0468,
-      "step": 1825
-    },
-    {
-      "epoch": 166.0,
-      "learning_rate": 7.031578947368421e-05,
-      "loss": 0.0486,
-      "step": 1826
-    },
-    {
-      "epoch": 166.09,
-      "learning_rate": 7.029665071770335e-05,
-      "loss": 0.0461,
-      "step": 1827
-    },
-    {
-      "epoch": 166.18,
-      "learning_rate": 7.027751196172249e-05,
-      "loss": 0.0411,
-      "step": 1828
-    },
-    {
-      "epoch": 166.27,
-      "learning_rate": 7.025837320574164e-05,
-      "loss": 0.0433,
-      "step": 1829
-    },
-    {
-      "epoch": 166.36,
-      "learning_rate": 7.023923444976077e-05,
-      "loss": 0.0479,
-      "step": 1830
-    },
-    {
-      "epoch": 166.45,
-      "learning_rate": 7.022009569377991e-05,
-      "loss": 0.0495,
-      "step": 1831
-    },
-    {
-      "epoch": 166.55,
-      "learning_rate": 7.020095693779905e-05,
-      "loss": 0.0442,
-      "step": 1832
-    },
-    {
-      "epoch": 166.64,
-      "learning_rate": 7.018181818181818e-05,
-      "loss": 0.0466,
-      "step": 1833
-    },
-    {
-      "epoch": 166.73,
-      "learning_rate": 7.016267942583732e-05,
-      "loss": 0.0452,
-      "step": 1834
-    },
-    {
-      "epoch": 166.82,
-      "learning_rate": 7.014354066985647e-05,
-      "loss": 0.0464,
-      "step": 1835
-    },
-    {
-      "epoch": 166.91,
-      "learning_rate": 7.01244019138756e-05,
-      "loss": 0.0567,
-      "step": 1836
-    },
-    {
-      "epoch": 167.0,
-      "learning_rate": 7.010526315789473e-05,
-      "loss": 0.0468,
-      "step": 1837
-    },
-    {
-      "epoch": 167.09,
-      "learning_rate": 7.008612440191388e-05,
-      "loss": 0.0462,
-      "step": 1838
-    },
-    {
-      "epoch": 167.18,
-      "learning_rate": 7.006698564593301e-05,
-      "loss": 0.0456,
-      "step": 1839
-    },
-    {
-      "epoch": 167.27,
-      "learning_rate": 7.004784688995216e-05,
-      "loss": 0.0485,
-      "step": 1840
-    },
-    {
-      "epoch": 167.27,
-      "eval_accuracy_NAT": 0.992405896855147,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.944899727456114,
-      "eval_accuracy_concrete tiles": 0.982449703418333,
-      "eval_accuracy_dark bricks": 0.8715599861437204,
-      "eval_accuracy_grass": 0.888444237846755,
-      "eval_accuracy_red bricks": 0.9263139204545454,
-      "eval_accuracy_road": 0.8881457628717259,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5142668100239848,
-      "eval_accuracy_speedway bricks": 0.9870575059707203,
-      "eval_accuracy_steel": 0.8977475100587032,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9787805300386256,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8791844711567262,
-      "eval_iou_concrete tiles": 0.9453924914675768,
-      "eval_iou_dark bricks": 0.805463664496852,
-      "eval_iou_grass": 0.8205844185190685,
-      "eval_iou_red bricks": 0.845954272742014,
-      "eval_iou_road": 0.8793363120037545,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3444620895963364,
-      "eval_iou_speedway bricks": 0.9747215515412685,
-      "eval_iou_steel": 0.8303858091773624,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12035813927650452,
-      "eval_mean_accuracy": 0.6840993123922884,
-      "eval_mean_iou": 0.6387896623645835,
-      "eval_overall_accuracy": 0.9685901006062826,
-      "eval_runtime": 3.5728,
-      "eval_samples_per_second": 10.076,
-      "eval_steps_per_second": 0.56,
-      "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.0413,
-      "step": 1842
-    },
-    {
-      "epoch": 167.55,
-      "learning_rate": 6.999043062200957e-05,
-      "loss": 0.0451,
-      "step": 1843
-    },
-    {
-      "epoch": 167.64,
-      "learning_rate": 6.997129186602871e-05,
-      "loss": 0.0407,
-      "step": 1844
-    },
-    {
-      "epoch": 167.73,
-      "learning_rate": 6.995215311004784e-05,
-      "loss": 0.0418,
-      "step": 1845
-    },
-    {
-      "epoch": 167.82,
-      "learning_rate": 6.993301435406699e-05,
-      "loss": 0.0421,
-      "step": 1846
-    },
-    {
-      "epoch": 167.91,
-      "learning_rate": 6.991387559808614e-05,
-      "loss": 0.0467,
-      "step": 1847
-    },
-    {
-      "epoch": 168.0,
-      "learning_rate": 6.989473684210527e-05,
-      "loss": 0.047,
-      "step": 1848
-    },
-    {
-      "epoch": 168.09,
-      "learning_rate": 6.98755980861244e-05,
-      "loss": 0.0562,
-      "step": 1849
-    },
-    {
-      "epoch": 168.18,
-      "learning_rate": 6.985645933014354e-05,
-      "loss": 0.0459,
-      "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.0459,
-      "step": 1852
-    },
-    {
-      "epoch": 168.45,
-      "learning_rate": 6.979904306220097e-05,
-      "loss": 0.0459,
-      "step": 1853
-    },
-    {
-      "epoch": 168.55,
-      "learning_rate": 6.97799043062201e-05,
-      "loss": 0.0546,
-      "step": 1854
-    },
-    {
-      "epoch": 168.64,
-      "learning_rate": 6.976076555023923e-05,
-      "loss": 0.0454,
-      "step": 1855
-    },
-    {
-      "epoch": 168.73,
-      "learning_rate": 6.974162679425838e-05,
-      "loss": 0.043,
-      "step": 1856
-    },
-    {
-      "epoch": 168.82,
-      "learning_rate": 6.972248803827751e-05,
-      "loss": 0.0502,
-      "step": 1857
-    },
-    {
-      "epoch": 168.91,
-      "learning_rate": 6.970334928229666e-05,
-      "loss": 0.0527,
-      "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.0461,
-      "step": 1860
-    },
-    {
-      "epoch": 169.09,
-      "eval_accuracy_NAT": 0.9925281774787676,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9475098180179407,
-      "eval_accuracy_concrete tiles": 0.9845594080596832,
-      "eval_accuracy_dark bricks": 0.844771178938455,
-      "eval_accuracy_grass": 0.893718304050441,
-      "eval_accuracy_red bricks": 0.9366122159090909,
-      "eval_accuracy_road": 0.8708419842149877,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6013288120640697,
-      "eval_accuracy_speedway bricks": 0.9849799728949213,
-      "eval_accuracy_steel": 0.9035106523316404,
-      "eval_accuracy_tiles": 0.011505273250239693,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9782664537706919,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.885822733382822,
-      "eval_iou_concrete tiles": 0.9331188965197484,
-      "eval_iou_dark bricks": 0.8058452048758995,
-      "eval_iou_grass": 0.8221804946432935,
-      "eval_iou_red bricks": 0.8450817045818648,
-      "eval_iou_road": 0.8638630725900478,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.39869125737995575,
-      "eval_iou_speedway bricks": 0.9745276051937011,
-      "eval_iou_steel": 0.8296363030706801,
-      "eval_iou_tiles": 0.011505273250239693,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13130919635295868,
-      "eval_mean_accuracy": 0.690143522862326,
-      "eval_mean_iou": 0.6421953076353033,
-      "eval_overall_accuracy": 0.9675146738688151,
-      "eval_runtime": 3.8455,
-      "eval_samples_per_second": 9.362,
-      "eval_steps_per_second": 0.52,
-      "step": 1860
-    },
-    {
-      "epoch": 169.18,
-      "learning_rate": 6.964593301435406e-05,
-      "loss": 0.0413,
-      "step": 1861
-    },
-    {
-      "epoch": 169.27,
-      "learning_rate": 6.962679425837321e-05,
-      "loss": 0.0415,
-      "step": 1862
-    },
-    {
-      "epoch": 169.36,
-      "learning_rate": 6.960765550239234e-05,
-      "loss": 0.0516,
-      "step": 1863
-    },
-    {
-      "epoch": 169.45,
-      "learning_rate": 6.958851674641149e-05,
-      "loss": 0.0424,
-      "step": 1864
-    },
-    {
-      "epoch": 169.55,
-      "learning_rate": 6.956937799043063e-05,
-      "loss": 0.039,
-      "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.0488,
-      "step": 1867
-    },
-    {
-      "epoch": 169.82,
-      "learning_rate": 6.951196172248804e-05,
-      "loss": 0.0445,
-      "step": 1868
-    },
-    {
-      "epoch": 169.91,
-      "learning_rate": 6.949282296650717e-05,
-      "loss": 0.0417,
-      "step": 1869
-    },
-    {
-      "epoch": 170.0,
-      "learning_rate": 6.947368421052632e-05,
-      "loss": 0.0435,
-      "step": 1870
-    },
-    {
-      "epoch": 170.09,
-      "learning_rate": 6.945454545454547e-05,
-      "loss": 0.0461,
-      "step": 1871
-    },
-    {
-      "epoch": 170.18,
-      "learning_rate": 6.94354066985646e-05,
-      "loss": 0.0425,
-      "step": 1872
-    },
-    {
-      "epoch": 170.27,
-      "learning_rate": 6.941626794258373e-05,
-      "loss": 0.05,
-      "step": 1873
-    },
-    {
-      "epoch": 170.36,
-      "learning_rate": 6.939712918660288e-05,
-      "loss": 0.0407,
-      "step": 1874
-    },
-    {
-      "epoch": 170.45,
-      "learning_rate": 6.937799043062201e-05,
-      "loss": 0.044,
-      "step": 1875
-    },
-    {
-      "epoch": 170.55,
-      "learning_rate": 6.935885167464115e-05,
-      "loss": 0.038,
-      "step": 1876
-    },
-    {
-      "epoch": 170.64,
-      "learning_rate": 6.93397129186603e-05,
-      "loss": 0.0429,
-      "step": 1877
-    },
-    {
-      "epoch": 170.73,
-      "learning_rate": 6.932057416267943e-05,
-      "loss": 0.0451,
-      "step": 1878
-    },
-    {
-      "epoch": 170.82,
-      "learning_rate": 6.930143540669856e-05,
-      "loss": 0.0394,
-      "step": 1879
-    },
-    {
-      "epoch": 170.91,
-      "learning_rate": 6.928229665071771e-05,
-      "loss": 0.0443,
-      "step": 1880
-    },
-    {
-      "epoch": 170.91,
-      "eval_accuracy_NAT": 0.9920214098726517,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.943066226018284,
-      "eval_accuracy_concrete tiles": 0.9879838561731792,
-      "eval_accuracy_dark bricks": 0.8724837381163157,
-      "eval_accuracy_grass": 0.8903630333045572,
-      "eval_accuracy_red bricks": 0.9209872159090909,
-      "eval_accuracy_road": 0.9294063872829924,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46177597662172964,
-      "eval_accuracy_speedway bricks": 0.9887367898946088,
-      "eval_accuracy_steel": 0.8923966097223138,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9785171190347933,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8746502928894235,
-      "eval_iou_concrete tiles": 0.9393587022878572,
-      "eval_iou_dark bricks": 0.8114551637730446,
-      "eval_iou_grass": 0.8182545265328541,
-      "eval_iou_red bricks": 0.8130094043887147,
-      "eval_iou_road": 0.9230565069727,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3031564468254543,
-      "eval_iou_speedway bricks": 0.9741363884160983,
-      "eval_iou_steel": 0.8253231156353654,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1050824448466301,
-      "eval_mean_accuracy": 0.6830170186858248,
-      "eval_mean_iou": 0.6354552051351003,
-      "eval_overall_accuracy": 0.9719311396280924,
-      "eval_runtime": 3.224,
-      "eval_samples_per_second": 11.166,
-      "eval_steps_per_second": 0.62,
-      "step": 1880
-    },
-    {
-      "epoch": 171.0,
-      "learning_rate": 6.926315789473684e-05,
-      "loss": 0.049,
-      "step": 1881
-    },
-    {
-      "epoch": 171.09,
-      "learning_rate": 6.924401913875599e-05,
-      "loss": 0.0426,
-      "step": 1882
-    },
-    {
-      "epoch": 171.18,
-      "learning_rate": 6.922488038277513e-05,
-      "loss": 0.0481,
-      "step": 1883
-    },
-    {
-      "epoch": 171.27,
-      "learning_rate": 6.920574162679426e-05,
-      "loss": 0.045,
-      "step": 1884
-    },
-    {
-      "epoch": 171.36,
-      "learning_rate": 6.91866028708134e-05,
-      "loss": 0.0374,
-      "step": 1885
-    },
-    {
-      "epoch": 171.45,
-      "learning_rate": 6.916746411483254e-05,
-      "loss": 0.0448,
-      "step": 1886
-    },
-    {
-      "epoch": 171.55,
-      "learning_rate": 6.914832535885167e-05,
-      "loss": 0.0444,
-      "step": 1887
-    },
-    {
-      "epoch": 171.64,
-      "learning_rate": 6.912918660287082e-05,
-      "loss": 0.0417,
-      "step": 1888
-    },
-    {
-      "epoch": 171.73,
-      "learning_rate": 6.911004784688996e-05,
-      "loss": 0.0443,
-      "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.0447,
-      "step": 1891
-    },
-    {
-      "epoch": 172.0,
-      "learning_rate": 6.905263157894737e-05,
-      "loss": 0.0451,
-      "step": 1892
-    },
-    {
-      "epoch": 172.09,
-      "learning_rate": 6.90334928229665e-05,
-      "loss": 0.0388,
-      "step": 1893
-    },
-    {
-      "epoch": 172.18,
-      "learning_rate": 6.901435406698565e-05,
-      "loss": 0.0456,
-      "step": 1894
-    },
-    {
-      "epoch": 172.27,
-      "learning_rate": 6.89952153110048e-05,
-      "loss": 0.0444,
-      "step": 1895
-    },
-    {
-      "epoch": 172.36,
-      "learning_rate": 6.897607655502393e-05,
-      "loss": 0.0425,
-      "step": 1896
-    },
-    {
-      "epoch": 172.45,
-      "learning_rate": 6.895693779904306e-05,
-      "loss": 0.0542,
-      "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.0402,
-      "step": 1899
-    },
-    {
-      "epoch": 172.73,
-      "learning_rate": 6.889952153110048e-05,
-      "loss": 0.0435,
-      "step": 1900
-    },
-    {
-      "epoch": 172.73,
-      "eval_accuracy_NAT": 0.992198743244569,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9460116314004893,
-      "eval_accuracy_concrete tiles": 0.9903381642512077,
-      "eval_accuracy_dark bricks": 0.8757553596859243,
-      "eval_accuracy_grass": 0.9039029487936753,
-      "eval_accuracy_red bricks": 0.9556107954545454,
-      "eval_accuracy_road": 0.9051303103918199,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47864803021531166,
-      "eval_accuracy_speedway bricks": 0.9865753831337863,
-      "eval_accuracy_steel": 0.8964778048941363,
-      "eval_accuracy_tiles": 0.0007533214628133132,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9787434839117086,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8756522548806605,
-      "eval_iou_concrete tiles": 0.9229760920981392,
-      "eval_iou_dark bricks": 0.7989115168539326,
-      "eval_iou_grass": 0.8330659760473641,
-      "eval_iou_red bricks": 0.8367537313432836,
-      "eval_iou_road": 0.898967881093753,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31242352263729933,
-      "eval_iou_speedway bricks": 0.9745921480945452,
-      "eval_iou_steel": 0.8308461132888614,
-      "eval_iou_tiles": 0.0007533214628133132,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10976611822843552,
-      "eval_mean_accuracy": 0.6870309609944829,
-      "eval_mean_iou": 0.6356681570547971,
-      "eval_overall_accuracy": 0.9703302383422852,
-      "eval_runtime": 3.3141,
-      "eval_samples_per_second": 10.863,
-      "eval_steps_per_second": 0.603,
-      "step": 1900
-    },
-    {
-      "epoch": 172.82,
-      "learning_rate": 6.888038277511963e-05,
-      "loss": 0.0536,
-      "step": 1901
-    },
-    {
-      "epoch": 172.91,
-      "learning_rate": 6.886124401913876e-05,
-      "loss": 0.0433,
-      "step": 1902
-    },
-    {
-      "epoch": 173.0,
-      "learning_rate": 6.88421052631579e-05,
-      "loss": 0.0414,
-      "step": 1903
-    },
-    {
-      "epoch": 173.09,
-      "learning_rate": 6.882296650717704e-05,
-      "loss": 0.0437,
-      "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.044,
-      "step": 1906
-    },
-    {
-      "epoch": 173.36,
-      "learning_rate": 6.876555023923446e-05,
-      "loss": 0.0419,
-      "step": 1907
-    },
-    {
-      "epoch": 173.45,
-      "learning_rate": 6.87464114832536e-05,
-      "loss": 0.0438,
-      "step": 1908
-    },
-    {
-      "epoch": 173.55,
-      "learning_rate": 6.872727272727273e-05,
-      "loss": 0.0413,
-      "step": 1909
-    },
-    {
-      "epoch": 173.64,
-      "learning_rate": 6.870813397129187e-05,
-      "loss": 0.0425,
-      "step": 1910
-    },
-    {
-      "epoch": 173.73,
-      "learning_rate": 6.8688995215311e-05,
-      "loss": 0.0454,
-      "step": 1911
-    },
-    {
-      "epoch": 173.82,
-      "learning_rate": 6.866985645933015e-05,
-      "loss": 0.0394,
-      "step": 1912
-    },
-    {
-      "epoch": 173.91,
-      "learning_rate": 6.865071770334928e-05,
-      "loss": 0.047,
-      "step": 1913
-    },
-    {
-      "epoch": 174.0,
-      "learning_rate": 6.863157894736843e-05,
-      "loss": 0.0558,
-      "step": 1914
-    },
-    {
-      "epoch": 174.09,
-      "learning_rate": 6.861244019138756e-05,
-      "loss": 0.045,
-      "step": 1915
-    },
-    {
-      "epoch": 174.18,
-      "learning_rate": 6.85933014354067e-05,
-      "loss": 0.047,
-      "step": 1916
-    },
-    {
-      "epoch": 174.27,
-      "learning_rate": 6.857416267942584e-05,
-      "loss": 0.0484,
-      "step": 1917
-    },
-    {
-      "epoch": 174.36,
-      "learning_rate": 6.855502392344498e-05,
-      "loss": 0.0379,
-      "step": 1918
-    },
-    {
-      "epoch": 174.45,
-      "learning_rate": 6.853588516746411e-05,
-      "loss": 0.052,
-      "step": 1919
-    },
-    {
-      "epoch": 174.55,
-      "learning_rate": 6.851674641148326e-05,
-      "loss": 0.0406,
-      "step": 1920
-    },
-    {
-      "epoch": 174.55,
-      "eval_accuracy_NAT": 0.9925837595804133,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9485492939611142,
-      "eval_accuracy_concrete tiles": 0.9919586620192014,
-      "eval_accuracy_dark bricks": 0.8798352642315538,
-      "eval_accuracy_grass": 0.8833868662120741,
-      "eval_accuracy_red bricks": 0.9554332386363636,
-      "eval_accuracy_road": 0.8674646638943538,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4360543655060238,
-      "eval_accuracy_speedway bricks": 0.983531794160702,
-      "eval_accuracy_steel": 0.9023398852318448,
-      "eval_accuracy_tiles": 0.015819750719079578,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.978735546489165,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8694489044723493,
-      "eval_iou_concrete tiles": 0.9024478442280945,
-      "eval_iou_dark bricks": 0.8014796115143228,
-      "eval_iou_grass": 0.8222491875168759,
-      "eval_iou_red bricks": 0.820649687357023,
-      "eval_iou_road": 0.8602888887556438,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2921823622861787,
-      "eval_iou_speedway bricks": 0.9738587513405806,
-      "eval_iou_steel": 0.8313431475320177,
-      "eval_iou_tiles": 0.015819750719079578,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14024387300014496,
-      "eval_mean_accuracy": 0.6813044264732865,
-      "eval_mean_iou": 0.6283464370931793,
-      "eval_overall_accuracy": 0.9662564595540365,
-      "eval_runtime": 3.5891,
-      "eval_samples_per_second": 10.03,
-      "eval_steps_per_second": 0.557,
-      "step": 1920
-    },
-    {
-      "epoch": 174.64,
-      "learning_rate": 6.849760765550239e-05,
-      "loss": 0.0548,
-      "step": 1921
-    },
-    {
-      "epoch": 174.73,
-      "learning_rate": 6.847846889952154e-05,
-      "loss": 0.0469,
-      "step": 1922
-    },
-    {
-      "epoch": 174.82,
-      "learning_rate": 6.845933014354068e-05,
-      "loss": 0.042,
-      "step": 1923
-    },
-    {
-      "epoch": 174.91,
-      "learning_rate": 6.844019138755981e-05,
-      "loss": 0.0409,
-      "step": 1924
-    },
-    {
-      "epoch": 175.0,
-      "learning_rate": 6.842105263157895e-05,
-      "loss": 0.0435,
-      "step": 1925
-    },
-    {
-      "epoch": 175.09,
-      "learning_rate": 6.840191387559809e-05,
-      "loss": 0.0472,
-      "step": 1926
-    },
-    {
-      "epoch": 175.18,
-      "learning_rate": 6.838277511961722e-05,
-      "loss": 0.0385,
-      "step": 1927
-    },
-    {
-      "epoch": 175.27,
-      "learning_rate": 6.836363636363637e-05,
-      "loss": 0.0423,
-      "step": 1928
-    },
-    {
-      "epoch": 175.36,
-      "learning_rate": 6.834449760765552e-05,
-      "loss": 0.0451,
-      "step": 1929
-    },
-    {
-      "epoch": 175.45,
-      "learning_rate": 6.832535885167465e-05,
-      "loss": 0.0493,
-      "step": 1930
-    },
-    {
-      "epoch": 175.55,
-      "learning_rate": 6.830622009569378e-05,
-      "loss": 0.0376,
-      "step": 1931
-    },
-    {
-      "epoch": 175.64,
-      "learning_rate": 6.828708133971292e-05,
-      "loss": 0.039,
-      "step": 1932
-    },
-    {
-      "epoch": 175.73,
-      "learning_rate": 6.826794258373206e-05,
-      "loss": 0.0433,
-      "step": 1933
-    },
-    {
-      "epoch": 175.82,
-      "learning_rate": 6.82488038277512e-05,
-      "loss": 0.0486,
-      "step": 1934
-    },
-    {
-      "epoch": 175.91,
-      "learning_rate": 6.822966507177035e-05,
-      "loss": 0.0433,
-      "step": 1935
-    },
-    {
-      "epoch": 176.0,
-      "learning_rate": 6.821052631578948e-05,
-      "loss": 0.0398,
-      "step": 1936
-    },
-    {
-      "epoch": 176.09,
-      "learning_rate": 6.819138755980861e-05,
-      "loss": 0.0389,
-      "step": 1937
-    },
-    {
-      "epoch": 176.18,
-      "learning_rate": 6.817224880382776e-05,
-      "loss": 0.0493,
-      "step": 1938
-    },
-    {
-      "epoch": 176.27,
-      "learning_rate": 6.815311004784689e-05,
-      "loss": 0.0439,
-      "step": 1939
-    },
-    {
-      "epoch": 176.36,
-      "learning_rate": 6.813397129186602e-05,
-      "loss": 0.0405,
-      "step": 1940
-    },
-    {
-      "epoch": 176.36,
-      "eval_accuracy_NAT": 0.9919912367317583,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9462087965148719,
-      "eval_accuracy_concrete tiles": 0.9792087078823457,
-      "eval_accuracy_dark bricks": 0.8778722912897887,
-      "eval_accuracy_grass": 0.9040637221835822,
-      "eval_accuracy_red bricks": 0.9209872159090909,
-      "eval_accuracy_road": 0.9058135133925926,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5158106580652276,
-      "eval_accuracy_speedway bricks": 0.9873181781428797,
-      "eval_accuracy_steel": 0.8948453268254073,
-      "eval_accuracy_tiles": 0.001712094233666621,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9790039935755902,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8802783597389099,
-      "eval_iou_concrete tiles": 0.949565629910754,
-      "eval_iou_dark bricks": 0.8014618033593366,
-      "eval_iou_grass": 0.8317299035369775,
-      "eval_iou_red bricks": 0.8453389830508474,
-      "eval_iou_road": 0.8977649835031358,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33883264818269077,
-      "eval_iou_speedway bricks": 0.9749040281656406,
-      "eval_iou_steel": 0.8308695752026763,
-      "eval_iou_tiles": 0.001712094233666621,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11012466251850128,
-      "eval_mean_accuracy": 0.6866024416285548,
-      "eval_mean_iou": 0.6408816924969405,
-      "eval_overall_accuracy": 0.9704863230387369,
-      "eval_runtime": 3.4713,
-      "eval_samples_per_second": 10.371,
-      "eval_steps_per_second": 0.576,
-      "step": 1940
-    },
-    {
-      "epoch": 176.45,
-      "learning_rate": 6.811483253588518e-05,
-      "loss": 0.0423,
-      "step": 1941
-    },
-    {
-      "epoch": 176.55,
-      "learning_rate": 6.809569377990431e-05,
-      "loss": 0.047,
-      "step": 1942
-    },
-    {
-      "epoch": 176.64,
-      "learning_rate": 6.807655502392344e-05,
-      "loss": 0.0449,
-      "step": 1943
-    },
-    {
-      "epoch": 176.73,
-      "learning_rate": 6.805741626794259e-05,
-      "loss": 0.0478,
-      "step": 1944
-    },
-    {
-      "epoch": 176.82,
-      "learning_rate": 6.803827751196172e-05,
-      "loss": 0.0397,
-      "step": 1945
-    },
-    {
-      "epoch": 176.91,
-      "learning_rate": 6.801913875598085e-05,
-      "loss": 0.0389,
-      "step": 1946
-    },
-    {
-      "epoch": 177.0,
-      "learning_rate": 6.800000000000001e-05,
-      "loss": 0.0403,
-      "step": 1947
-    },
-    {
-      "epoch": 177.09,
-      "learning_rate": 6.798086124401915e-05,
-      "loss": 0.0403,
-      "step": 1948
-    },
-    {
-      "epoch": 177.18,
-      "learning_rate": 6.796172248803828e-05,
-      "loss": 0.047,
-      "step": 1949
-    },
-    {
-      "epoch": 177.27,
-      "learning_rate": 6.794258373205742e-05,
-      "loss": 0.0476,
-      "step": 1950
-    },
-    {
-      "epoch": 177.36,
-      "learning_rate": 6.792344497607655e-05,
-      "loss": 0.0412,
-      "step": 1951
-    },
-    {
-      "epoch": 177.45,
-      "learning_rate": 6.790430622009569e-05,
-      "loss": 0.0442,
-      "step": 1952
-    },
-    {
-      "epoch": 177.55,
-      "learning_rate": 6.788516746411485e-05,
-      "loss": 0.0509,
-      "step": 1953
-    },
-    {
-      "epoch": 177.64,
-      "learning_rate": 6.786602870813398e-05,
-      "loss": 0.0379,
-      "step": 1954
-    },
-    {
-      "epoch": 177.73,
-      "learning_rate": 6.784688995215311e-05,
-      "loss": 0.0379,
-      "step": 1955
-    },
-    {
-      "epoch": 177.82,
-      "learning_rate": 6.782775119617226e-05,
-      "loss": 0.0422,
-      "step": 1956
-    },
-    {
-      "epoch": 177.91,
-      "learning_rate": 6.780861244019139e-05,
-      "loss": 0.0448,
-      "step": 1957
-    },
-    {
-      "epoch": 178.0,
-      "learning_rate": 6.778947368421052e-05,
-      "loss": 0.0479,
-      "step": 1958
-    },
-    {
-      "epoch": 178.09,
-      "learning_rate": 6.777033492822968e-05,
-      "loss": 0.0409,
-      "step": 1959
-    },
-    {
-      "epoch": 178.18,
-      "learning_rate": 6.775119617224881e-05,
-      "loss": 0.0406,
-      "step": 1960
-    },
-    {
-      "epoch": 178.18,
-      "eval_accuracy_NAT": 0.9917911411658337,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9413748980642946,
-      "eval_accuracy_concrete tiles": 0.9896960802299273,
-      "eval_accuracy_dark bricks": 0.883183865132212,
-      "eval_accuracy_grass": 0.9071044362970393,
-      "eval_accuracy_red bricks": 0.9396306818181818,
-      "eval_accuracy_road": 0.9390425726160145,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5455848702891958,
-      "eval_accuracy_speedway bricks": 0.9902477230406443,
-      "eval_accuracy_steel": 0.8801612690455775,
-      "eval_accuracy_tiles": 0.019449390494452816,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9787979952365885,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8794255771293391,
-      "eval_iou_concrete tiles": 0.9280369276642105,
-      "eval_iou_dark bricks": 0.8102115038310794,
-      "eval_iou_grass": 0.8313782521510165,
-      "eval_iou_red bricks": 0.8349637109498265,
-      "eval_iou_road": 0.9326055758711173,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.35438005873504763,
-      "eval_iou_speedway bricks": 0.9742153138536812,
-      "eval_iou_steel": 0.8268760069401413,
-      "eval_iou_tiles": 0.019449390494452816,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09781524538993835,
-      "eval_mean_accuracy": 0.6944051483225674,
-      "eval_mean_iou": 0.6438723317581924,
-      "eval_overall_accuracy": 0.9735180536905924,
-      "eval_runtime": 3.4973,
-      "eval_samples_per_second": 10.294,
-      "eval_steps_per_second": 0.572,
-      "step": 1960
-    },
-    {
-      "epoch": 178.27,
-      "learning_rate": 6.773205741626794e-05,
-      "loss": 0.0449,
-      "step": 1961
-    },
-    {
-      "epoch": 178.36,
-      "learning_rate": 6.771291866028709e-05,
-      "loss": 0.0405,
-      "step": 1962
-    },
-    {
-      "epoch": 178.45,
-      "learning_rate": 6.769377990430622e-05,
-      "loss": 0.0373,
-      "step": 1963
-    },
-    {
-      "epoch": 178.55,
-      "learning_rate": 6.767464114832535e-05,
-      "loss": 0.0467,
-      "step": 1964
-    },
-    {
-      "epoch": 178.64,
-      "learning_rate": 6.765550239234451e-05,
-      "loss": 0.0463,
-      "step": 1965
-    },
-    {
-      "epoch": 178.73,
-      "learning_rate": 6.763636363636364e-05,
-      "loss": 0.0394,
-      "step": 1966
-    },
-    {
-      "epoch": 178.82,
-      "learning_rate": 6.761722488038278e-05,
-      "loss": 0.0481,
-      "step": 1967
-    },
-    {
-      "epoch": 178.91,
-      "learning_rate": 6.759808612440192e-05,
-      "loss": 0.0466,
-      "step": 1968
-    },
-    {
-      "epoch": 179.0,
-      "learning_rate": 6.757894736842105e-05,
-      "loss": 0.0438,
-      "step": 1969
-    },
-    {
-      "epoch": 179.09,
-      "learning_rate": 6.755980861244018e-05,
-      "loss": 0.0421,
-      "step": 1970
-    },
-    {
-      "epoch": 179.18,
-      "learning_rate": 6.754066985645934e-05,
-      "loss": 0.0429,
-      "step": 1971
-    },
-    {
-      "epoch": 179.27,
-      "learning_rate": 6.752153110047848e-05,
-      "loss": 0.0429,
-      "step": 1972
-    },
-    {
-      "epoch": 179.36,
-      "learning_rate": 6.750239234449761e-05,
-      "loss": 0.0399,
-      "step": 1973
-    },
-    {
-      "epoch": 179.45,
-      "learning_rate": 6.748325358851675e-05,
-      "loss": 0.0518,
-      "step": 1974
-    },
-    {
-      "epoch": 179.55,
-      "learning_rate": 6.746411483253589e-05,
-      "loss": 0.0502,
-      "step": 1975
-    },
-    {
-      "epoch": 179.64,
-      "learning_rate": 6.744497607655502e-05,
-      "loss": 0.044,
-      "step": 1976
-    },
-    {
-      "epoch": 179.73,
-      "learning_rate": 6.742583732057418e-05,
-      "loss": 0.0373,
-      "step": 1977
-    },
-    {
-      "epoch": 179.82,
-      "learning_rate": 6.740669856459331e-05,
-      "loss": 0.0407,
-      "step": 1978
-    },
-    {
-      "epoch": 179.91,
-      "learning_rate": 6.738755980861244e-05,
-      "loss": 0.0534,
-      "step": 1979
-    },
-    {
-      "epoch": 180.0,
-      "learning_rate": 6.736842105263159e-05,
-      "loss": 0.0456,
-      "step": 1980
-    },
-    {
-      "epoch": 180.0,
-      "eval_accuracy_NAT": 0.9922395034524425,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9444128503369243,
-      "eval_accuracy_concrete tiles": 0.9868219898489574,
-      "eval_accuracy_dark bricks": 0.8744082213925561,
-      "eval_accuracy_grass": 0.8798813072973644,
-      "eval_accuracy_red bricks": 0.9273792613636364,
-      "eval_accuracy_road": 0.9261717536067677,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46905411738758856,
-      "eval_accuracy_speedway bricks": 0.9875474731091312,
-      "eval_accuracy_steel": 0.8952823032781478,
-      "eval_accuracy_tiles": 0.13347486645664977,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.979057033751834,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8721908076873053,
-      "eval_iou_concrete tiles": 0.929285076732602,
-      "eval_iou_dark bricks": 0.8041769911504425,
-      "eval_iou_grass": 0.812185246721404,
-      "eval_iou_red bricks": 0.8410628019323672,
-      "eval_iou_road": 0.9185334778812851,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31319490464619687,
-      "eval_iou_speedway bricks": 0.9746629568796816,
-      "eval_iou_steel": 0.8296937559216357,
-      "eval_iou_tiles": 0.1334657262206396,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1043887734413147,
-      "eval_mean_accuracy": 0.6935902805792435,
-      "eval_mean_iou": 0.6467314445788765,
-      "eval_overall_accuracy": 0.9716350767347548,
-      "eval_runtime": 3.5826,
-      "eval_samples_per_second": 10.049,
-      "eval_steps_per_second": 0.558,
-      "step": 1980
-    },
-    {
-      "epoch": 180.09,
-      "learning_rate": 6.734928229665072e-05,
-      "loss": 0.0367,
-      "step": 1981
-    },
-    {
-      "epoch": 180.18,
-      "learning_rate": 6.733014354066985e-05,
-      "loss": 0.0422,
-      "step": 1982
-    },
-    {
-      "epoch": 180.27,
-      "learning_rate": 6.731100478468901e-05,
-      "loss": 0.0443,
-      "step": 1983
-    },
-    {
-      "epoch": 180.36,
-      "learning_rate": 6.729186602870814e-05,
-      "loss": 0.0525,
-      "step": 1984
-    },
-    {
-      "epoch": 180.45,
-      "learning_rate": 6.727272727272727e-05,
-      "loss": 0.0429,
-      "step": 1985
-    },
-    {
-      "epoch": 180.55,
-      "learning_rate": 6.725358851674642e-05,
-      "loss": 0.0385,
-      "step": 1986
-    },
-    {
-      "epoch": 180.64,
-      "learning_rate": 6.723444976076555e-05,
-      "loss": 0.0416,
-      "step": 1987
-    },
-    {
-      "epoch": 180.73,
-      "learning_rate": 6.721531100478468e-05,
-      "loss": 0.0439,
-      "step": 1988
-    },
-    {
-      "epoch": 180.82,
-      "learning_rate": 6.719617224880383e-05,
-      "loss": 0.0423,
-      "step": 1989
-    },
-    {
-      "epoch": 180.91,
-      "learning_rate": 6.717703349282297e-05,
-      "loss": 0.0428,
-      "step": 1990
-    },
-    {
-      "epoch": 181.0,
-      "learning_rate": 6.71578947368421e-05,
-      "loss": 0.0434,
-      "step": 1991
-    },
-    {
-      "epoch": 181.09,
-      "learning_rate": 6.713875598086125e-05,
-      "loss": 0.0376,
-      "step": 1992
-    },
-    {
-      "epoch": 181.18,
-      "learning_rate": 6.711961722488038e-05,
-      "loss": 0.048,
-      "step": 1993
-    },
-    {
-      "epoch": 181.27,
-      "learning_rate": 6.710047846889953e-05,
-      "loss": 0.0425,
-      "step": 1994
-    },
-    {
-      "epoch": 181.36,
-      "learning_rate": 6.708133971291866e-05,
-      "loss": 0.0361,
-      "step": 1995
-    },
-    {
-      "epoch": 181.45,
-      "learning_rate": 6.70622009569378e-05,
-      "loss": 0.0485,
-      "step": 1996
-    },
-    {
-      "epoch": 181.55,
-      "learning_rate": 6.704306220095694e-05,
-      "loss": 0.0415,
-      "step": 1997
-    },
-    {
-      "epoch": 181.64,
-      "learning_rate": 6.702392344497608e-05,
-      "loss": 0.0435,
-      "step": 1998
-    },
-    {
-      "epoch": 181.73,
-      "learning_rate": 6.700478468899522e-05,
-      "loss": 0.0434,
-      "step": 1999
-    },
-    {
-      "epoch": 181.82,
-      "learning_rate": 6.698564593301436e-05,
-      "loss": 0.042,
-      "step": 2000
-    },
-    {
-      "epoch": 181.82,
-      "eval_accuracy_NAT": 0.9927947951155508,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9445469762650758,
-      "eval_accuracy_concrete tiles": 0.9852014920809637,
-      "eval_accuracy_dark bricks": 0.8683653439051615,
-      "eval_accuracy_grass": 0.8680225222548817,
-      "eval_accuracy_red bricks": 0.9385653409090909,
-      "eval_accuracy_road": 0.9425178388535733,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.49596118324924876,
-      "eval_accuracy_speedway bricks": 0.9886082640319468,
-      "eval_accuracy_steel": 0.9001714926456038,
-      "eval_accuracy_tiles": 0.08033146144363786,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9787256693140206,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8760236202351823,
-      "eval_iou_concrete tiles": 0.9470373853750294,
-      "eval_iou_dark bricks": 0.802996867881549,
-      "eval_iou_grass": 0.8114671075417078,
-      "eval_iou_red bricks": 0.8076394194041253,
-      "eval_iou_road": 0.9342271486205632,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3328030190913127,
-      "eval_iou_speedway bricks": 0.9749794694183597,
-      "eval_iou_steel": 0.8353289518985793,
-      "eval_iou_tiles": 0.08033146144363786,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09672261029481888,
-      "eval_mean_accuracy": 0.6926989777503644,
-      "eval_mean_iou": 0.6447353938633898,
-      "eval_overall_accuracy": 0.973306761847602,
-      "eval_runtime": 3.773,
-      "eval_samples_per_second": 9.541,
-      "eval_steps_per_second": 0.53,
-      "step": 2000
-    },
-    {
-      "epoch": 181.91,
-      "learning_rate": 6.69665071770335e-05,
-      "loss": 0.0504,
-      "step": 2001
-    },
-    {
-      "epoch": 182.0,
-      "learning_rate": 6.694736842105264e-05,
-      "loss": 0.0356,
-      "step": 2002
-    },
-    {
-      "epoch": 182.09,
-      "learning_rate": 6.692822966507177e-05,
-      "loss": 0.0462,
-      "step": 2003
-    },
-    {
-      "epoch": 182.18,
-      "learning_rate": 6.690909090909092e-05,
-      "loss": 0.0444,
-      "step": 2004
-    },
-    {
-      "epoch": 182.27,
-      "learning_rate": 6.688995215311005e-05,
-      "loss": 0.0435,
-      "step": 2005
-    },
-    {
-      "epoch": 182.36,
-      "learning_rate": 6.68708133971292e-05,
-      "loss": 0.0409,
-      "step": 2006
-    },
-    {
-      "epoch": 182.45,
-      "learning_rate": 6.685167464114833e-05,
-      "loss": 0.0397,
-      "step": 2007
-    },
-    {
-      "epoch": 182.55,
-      "learning_rate": 6.683253588516747e-05,
-      "loss": 0.0404,
-      "step": 2008
-    },
-    {
-      "epoch": 182.64,
-      "learning_rate": 6.68133971291866e-05,
-      "loss": 0.0423,
-      "step": 2009
-    },
-    {
-      "epoch": 182.73,
-      "learning_rate": 6.679425837320575e-05,
-      "loss": 0.0409,
-      "step": 2010
-    },
-    {
-      "epoch": 182.82,
-      "learning_rate": 6.677511961722488e-05,
-      "loss": 0.0417,
-      "step": 2011
-    },
-    {
-      "epoch": 182.91,
-      "learning_rate": 6.675598086124403e-05,
-      "loss": 0.0423,
-      "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.0385,
-      "step": 2014
-    },
-    {
-      "epoch": 183.18,
-      "learning_rate": 6.669856459330144e-05,
-      "loss": 0.0465,
-      "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.0424,
-      "step": 2017
-    },
-    {
-      "epoch": 183.45,
-      "learning_rate": 6.664114832535886e-05,
-      "loss": 0.048,
-      "step": 2018
-    },
-    {
-      "epoch": 183.55,
-      "learning_rate": 6.662200956937799e-05,
-      "loss": 0.0457,
-      "step": 2019
-    },
-    {
-      "epoch": 183.64,
-      "learning_rate": 6.660287081339714e-05,
-      "loss": 0.0374,
-      "step": 2020
-    },
-    {
-      "epoch": 183.64,
-      "eval_accuracy_NAT": 0.9924062497573797,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9454939053178248,
-      "eval_accuracy_concrete tiles": 0.9917752094416926,
-      "eval_accuracy_dark bricks": 0.8842230861013818,
-      "eval_accuracy_grass": 0.8587605769667653,
-      "eval_accuracy_red bricks": 0.9190340909090909,
-      "eval_accuracy_road": 0.919061605563328,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.45979102914013176,
-      "eval_accuracy_speedway bricks": 0.9891531412806969,
-      "eval_accuracy_steel": 0.8893872435855155,
-      "eval_accuracy_tiles": 0.03670730036981235,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9790052617396954,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.865493816322747,
-      "eval_iou_concrete tiles": 0.9283097704767901,
-      "eval_iou_dark bricks": 0.8206401371722511,
-      "eval_iou_grass": 0.7978089201755981,
-      "eval_iou_red bricks": 0.8330919040721069,
-      "eval_iou_road": 0.9113890657582179,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.30605204242668915,
-      "eval_iou_speedway bricks": 0.9745489091243091,
-      "eval_iou_steel": 0.8341801028496307,
-      "eval_iou_tiles": 0.03670730036981235,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11285177618265152,
-      "eval_mean_accuracy": 0.6835225721872017,
-      "eval_mean_iou": 0.6374790177298344,
-      "eval_overall_accuracy": 0.9706172943115234,
-      "eval_runtime": 3.2685,
-      "eval_samples_per_second": 11.014,
-      "eval_steps_per_second": 0.612,
-      "step": 2020
-    },
-    {
-      "epoch": 183.73,
-      "learning_rate": 6.658373205741627e-05,
-      "loss": 0.0418,
-      "step": 2021
-    },
-    {
-      "epoch": 183.82,
-      "learning_rate": 6.65645933014354e-05,
-      "loss": 0.0404,
-      "step": 2022
-    },
-    {
-      "epoch": 183.91,
-      "learning_rate": 6.654545454545455e-05,
-      "loss": 0.0463,
-      "step": 2023
-    },
-    {
-      "epoch": 184.0,
-      "learning_rate": 6.652631578947369e-05,
-      "loss": 0.0434,
-      "step": 2024
-    },
-    {
-      "epoch": 184.09,
-      "learning_rate": 6.650717703349282e-05,
-      "loss": 0.0404,
-      "step": 2025
-    },
-    {
-      "epoch": 184.18,
-      "learning_rate": 6.648803827751197e-05,
-      "loss": 0.0402,
-      "step": 2026
-    },
-    {
-      "epoch": 184.27,
-      "learning_rate": 6.64688995215311e-05,
-      "loss": 0.0408,
-      "step": 2027
-    },
-    {
-      "epoch": 184.36,
-      "learning_rate": 6.644976076555023e-05,
-      "loss": 0.0458,
-      "step": 2028
-    },
-    {
-      "epoch": 184.45,
-      "learning_rate": 6.643062200956938e-05,
-      "loss": 0.0424,
-      "step": 2029
-    },
-    {
-      "epoch": 184.55,
-      "learning_rate": 6.641148325358852e-05,
-      "loss": 0.0433,
-      "step": 2030
-    },
-    {
-      "epoch": 184.64,
-      "learning_rate": 6.639234449760766e-05,
-      "loss": 0.0425,
-      "step": 2031
-    },
-    {
-      "epoch": 184.73,
-      "learning_rate": 6.63732057416268e-05,
-      "loss": 0.0435,
-      "step": 2032
-    },
-    {
-      "epoch": 184.82,
-      "learning_rate": 6.635406698564593e-05,
-      "loss": 0.0427,
-      "step": 2033
-    },
-    {
-      "epoch": 184.91,
-      "learning_rate": 6.633492822966507e-05,
-      "loss": 0.0498,
-      "step": 2034
-    },
-    {
-      "epoch": 185.0,
-      "learning_rate": 6.631578947368421e-05,
-      "loss": 0.0466,
-      "step": 2035
-    },
-    {
-      "epoch": 185.09,
-      "learning_rate": 6.629665071770336e-05,
-      "loss": 0.0417,
-      "step": 2036
-    },
-    {
-      "epoch": 185.18,
-      "learning_rate": 6.627751196172249e-05,
-      "loss": 0.0522,
-      "step": 2037
-    },
-    {
-      "epoch": 185.27,
-      "learning_rate": 6.625837320574163e-05,
-      "loss": 0.0455,
-      "step": 2038
-    },
-    {
-      "epoch": 185.36,
-      "learning_rate": 6.623923444976077e-05,
-      "loss": 0.0406,
-      "step": 2039
-    },
-    {
-      "epoch": 185.45,
-      "learning_rate": 6.62200956937799e-05,
-      "loss": 0.0442,
-      "step": 2040
-    },
-    {
-      "epoch": 185.45,
-      "eval_accuracy_NAT": 0.9925906411739505,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9447776728614962,
-      "eval_accuracy_concrete tiles": 0.9874640738702378,
-      "eval_accuracy_dark bricks": 0.8626303837419653,
-      "eval_accuracy_grass": 0.8853266321120381,
-      "eval_accuracy_red bricks": 0.9625355113636364,
-      "eval_accuracy_road": 0.9060456814919703,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5086152234444352,
-      "eval_accuracy_speedway bricks": 0.9888061817922902,
-      "eval_accuracy_steel": 0.8950349581162192,
-      "eval_accuracy_tiles": 0.08943980276674428,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9791917652725288,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8785884630352135,
-      "eval_iou_concrete tiles": 0.9385643708224354,
-      "eval_iou_dark bricks": 0.8144487244712552,
-      "eval_iou_grass": 0.8235724434358246,
-      "eval_iou_red bricks": 0.8273809523809523,
-      "eval_iou_road": 0.8990046360730637,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3366482975073902,
-      "eval_iou_speedway bricks": 0.9749617292375536,
-      "eval_iou_steel": 0.8356966897613549,
-      "eval_iou_tiles": 0.08942143101677508,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11241282522678375,
-      "eval_mean_accuracy": 0.6940974432873064,
-      "eval_mean_iou": 0.6459599617703344,
-      "eval_overall_accuracy": 0.9705687628851997,
-      "eval_runtime": 3.8428,
-      "eval_samples_per_second": 9.368,
-      "eval_steps_per_second": 0.52,
-      "step": 2040
-    },
-    {
-      "epoch": 185.55,
-      "learning_rate": 6.620095693779904e-05,
-      "loss": 0.044,
-      "step": 2041
-    },
-    {
-      "epoch": 185.64,
-      "learning_rate": 6.618181818181819e-05,
-      "loss": 0.0445,
-      "step": 2042
-    },
-    {
-      "epoch": 185.73,
-      "learning_rate": 6.616267942583732e-05,
-      "loss": 0.0404,
-      "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.0464,
-      "step": 2045
-    },
-    {
-      "epoch": 186.0,
-      "learning_rate": 6.610526315789473e-05,
-      "loss": 0.0439,
-      "step": 2046
-    },
-    {
-      "epoch": 186.09,
-      "learning_rate": 6.608612440191388e-05,
-      "loss": 0.0423,
-      "step": 2047
-    },
-    {
-      "epoch": 186.18,
-      "learning_rate": 6.606698564593302e-05,
-      "loss": 0.0464,
-      "step": 2048
-    },
-    {
-      "epoch": 186.27,
-      "learning_rate": 6.604784688995215e-05,
-      "loss": 0.0439,
-      "step": 2049
-    },
-    {
-      "epoch": 186.36,
-      "learning_rate": 6.60287081339713e-05,
-      "loss": 0.0355,
-      "step": 2050
-    },
-    {
-      "epoch": 186.45,
-      "learning_rate": 6.600956937799043e-05,
-      "loss": 0.0448,
-      "step": 2051
-    },
-    {
-      "epoch": 186.55,
-      "learning_rate": 6.599043062200956e-05,
-      "loss": 0.0427,
-      "step": 2052
-    },
-    {
-      "epoch": 186.64,
-      "learning_rate": 6.597129186602871e-05,
-      "loss": 0.0464,
-      "step": 2053
-    },
-    {
-      "epoch": 186.73,
-      "learning_rate": 6.595215311004786e-05,
-      "loss": 0.0476,
-      "step": 2054
-    },
-    {
-      "epoch": 186.82,
-      "learning_rate": 6.593301435406699e-05,
-      "loss": 0.0356,
-      "step": 2055
-    },
-    {
-      "epoch": 186.91,
-      "learning_rate": 6.591387559808613e-05,
-      "loss": 0.0429,
-      "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.0448,
-      "step": 2059
-    },
-    {
-      "epoch": 187.27,
-      "learning_rate": 6.583732057416269e-05,
-      "loss": 0.039,
-      "step": 2060
-    },
-    {
-      "epoch": 187.27,
-      "eval_accuracy_NAT": 0.9939258467712622,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9453208828705094,
-      "eval_accuracy_concrete tiles": 0.9836727206017245,
-      "eval_accuracy_dark bricks": 0.8662869019668219,
-      "eval_accuracy_grass": 0.8743346253455754,
-      "eval_accuracy_red bricks": 0.9382102272727273,
-      "eval_accuracy_road": 0.9282963335578431,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5814793372480909,
-      "eval_accuracy_speedway bricks": 0.9877212545572375,
-      "eval_accuracy_steel": 0.8960820526350505,
-      "eval_accuracy_tiles": 0.05307492124366525,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9789039772273064,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8853638805212686,
-      "eval_iou_concrete tiles": 0.9353141262319389,
-      "eval_iou_dark bricks": 0.8152937767152069,
-      "eval_iou_grass": 0.8229500430946569,
-      "eval_iou_red bricks": 0.83147128245476,
-      "eval_iou_road": 0.9221622413776535,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3904118463674225,
-      "eval_iou_speedway bricks": 0.9749494331027944,
-      "eval_iou_steel": 0.8369063019774533,
-      "eval_iou_tiles": 0.05307492124366525,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10347142815589905,
-      "eval_mean_accuracy": 0.6960311618515775,
-      "eval_mean_iou": 0.6497539869472404,
-      "eval_overall_accuracy": 0.9730592303805881,
-      "eval_runtime": 3.4146,
-      "eval_samples_per_second": 10.543,
-      "eval_steps_per_second": 0.586,
-      "step": 2060
-    },
-    {
-      "epoch": 187.36,
-      "learning_rate": 6.581818181818182e-05,
-      "loss": 0.0492,
-      "step": 2061
-    },
-    {
-      "epoch": 187.45,
-      "learning_rate": 6.579904306220097e-05,
-      "loss": 0.0399,
-      "step": 2062
-    },
-    {
-      "epoch": 187.55,
-      "learning_rate": 6.57799043062201e-05,
-      "loss": 0.0429,
-      "step": 2063
-    },
-    {
-      "epoch": 187.64,
-      "learning_rate": 6.576076555023923e-05,
-      "loss": 0.043,
-      "step": 2064
-    },
-    {
-      "epoch": 187.73,
-      "learning_rate": 6.574162679425838e-05,
-      "loss": 0.0406,
-      "step": 2065
-    },
-    {
-      "epoch": 187.82,
-      "learning_rate": 6.572248803827752e-05,
-      "loss": 0.0413,
-      "step": 2066
-    },
-    {
-      "epoch": 187.91,
-      "learning_rate": 6.570334928229665e-05,
-      "loss": 0.0372,
-      "step": 2067
-    },
-    {
-      "epoch": 188.0,
-      "learning_rate": 6.56842105263158e-05,
-      "loss": 0.0436,
-      "step": 2068
-    },
-    {
-      "epoch": 188.09,
-      "learning_rate": 6.566507177033493e-05,
-      "loss": 0.0425,
-      "step": 2069
-    },
-    {
-      "epoch": 188.18,
-      "learning_rate": 6.564593301435406e-05,
-      "loss": 0.0429,
-      "step": 2070
-    },
-    {
-      "epoch": 188.27,
-      "learning_rate": 6.562679425837321e-05,
-      "loss": 0.0461,
-      "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.0432,
-      "step": 2074
-    },
-    {
-      "epoch": 188.64,
-      "learning_rate": 6.555023923444976e-05,
-      "loss": 0.0455,
-      "step": 2075
-    },
-    {
-      "epoch": 188.73,
-      "learning_rate": 6.55311004784689e-05,
-      "loss": 0.0444,
-      "step": 2076
-    },
-    {
-      "epoch": 188.82,
-      "learning_rate": 6.551196172248804e-05,
-      "loss": 0.0409,
-      "step": 2077
-    },
-    {
-      "epoch": 188.91,
-      "learning_rate": 6.549282296650719e-05,
-      "loss": 0.0388,
-      "step": 2078
-    },
-    {
-      "epoch": 189.0,
-      "learning_rate": 6.547368421052632e-05,
-      "loss": 0.0407,
-      "step": 2079
-    },
-    {
-      "epoch": 189.09,
-      "learning_rate": 6.545454545454546e-05,
-      "loss": 0.0399,
-      "step": 2080
-    },
-    {
-      "epoch": 189.09,
-      "eval_accuracy_NAT": 0.9914188293103655,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9486512296665093,
-      "eval_accuracy_concrete tiles": 0.9881978841802728,
-      "eval_accuracy_dark bricks": 0.8916131018821446,
-      "eval_accuracy_grass": 0.9078523820674759,
-      "eval_accuracy_red bricks": 0.9398082386363636,
-      "eval_accuracy_road": 0.9178499782947012,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.539878146279602,
-      "eval_accuracy_speedway bricks": 0.9859828366683683,
-      "eval_accuracy_steel": 0.9133879691313238,
-      "eval_accuracy_tiles": 0.14361046431995617,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9794614431605171,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8828215981017595,
-      "eval_iou_concrete tiles": 0.9358892685469392,
-      "eval_iou_dark bricks": 0.8145504412954042,
-      "eval_iou_grass": 0.8275756600641665,
-      "eval_iou_red bricks": 0.824840268038024,
-      "eval_iou_road": 0.9102355783932222,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.35696968592209116,
-      "eval_iou_speedway bricks": 0.9751024319914832,
-      "eval_iou_steel": 0.8304697221847404,
-      "eval_iou_tiles": 0.14361046431995617,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10652013123035431,
-      "eval_mean_accuracy": 0.7052500815720835,
-      "eval_mean_iou": 0.6524251201552542,
-      "eval_overall_accuracy": 0.9719012578328451,
-      "eval_runtime": 3.3447,
-      "eval_samples_per_second": 10.763,
-      "eval_steps_per_second": 0.598,
-      "step": 2080
-    },
-    {
-      "epoch": 189.18,
-      "learning_rate": 6.54354066985646e-05,
-      "loss": 0.0437,
-      "step": 2081
-    },
-    {
-      "epoch": 189.27,
-      "learning_rate": 6.541626794258373e-05,
-      "loss": 0.0403,
-      "step": 2082
-    },
-    {
-      "epoch": 189.36,
-      "learning_rate": 6.539712918660287e-05,
-      "loss": 0.0399,
-      "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.0405,
-      "step": 2085
-    },
-    {
-      "epoch": 189.64,
-      "learning_rate": 6.53397129186603e-05,
-      "loss": 0.0412,
-      "step": 2086
-    },
-    {
-      "epoch": 189.73,
-      "learning_rate": 6.532057416267943e-05,
-      "loss": 0.0477,
-      "step": 2087
-    },
-    {
-      "epoch": 189.82,
-      "learning_rate": 6.530143540669856e-05,
-      "loss": 0.0408,
-      "step": 2088
-    },
-    {
-      "epoch": 189.91,
-      "learning_rate": 6.52822966507177e-05,
-      "loss": 0.0367,
-      "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.0543,
-      "step": 2091
-    },
-    {
-      "epoch": 190.18,
-      "learning_rate": 6.522488038277512e-05,
-      "loss": 0.0416,
-      "step": 2092
-    },
-    {
-      "epoch": 190.27,
-      "learning_rate": 6.520574162679426e-05,
-      "loss": 0.0413,
-      "step": 2093
-    },
-    {
-      "epoch": 190.36,
-      "learning_rate": 6.518660287081339e-05,
-      "loss": 0.0397,
-      "step": 2094
-    },
-    {
-      "epoch": 190.45,
-      "learning_rate": 6.516746411483254e-05,
-      "loss": 0.0483,
-      "step": 2095
-    },
-    {
-      "epoch": 190.55,
-      "learning_rate": 6.514832535885168e-05,
-      "loss": 0.0366,
-      "step": 2096
-    },
-    {
-      "epoch": 190.64,
-      "learning_rate": 6.512918660287082e-05,
-      "loss": 0.0432,
-      "step": 2097
-    },
-    {
-      "epoch": 190.73,
-      "learning_rate": 6.511004784688995e-05,
-      "loss": 0.0416,
-      "step": 2098
-    },
-    {
-      "epoch": 190.82,
-      "learning_rate": 6.50909090909091e-05,
-      "loss": 0.0394,
-      "step": 2099
-    },
-    {
-      "epoch": 190.91,
-      "learning_rate": 6.507177033492823e-05,
-      "loss": 0.0453,
-      "step": 2100
-    },
-    {
-      "epoch": 190.91,
-      "eval_accuracy_NAT": 0.9908030149143542,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9477874586892141,
-      "eval_accuracy_concrete tiles": 0.9827554577141809,
-      "eval_accuracy_dark bricks": 0.8860705900465725,
-      "eval_accuracy_grass": 0.8996424539611417,
-      "eval_accuracy_red bricks": 0.9151278409090909,
-      "eval_accuracy_road": 0.9166468154880306,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5587075786397596,
-      "eval_accuracy_speedway bricks": 0.9893957112186786,
-      "eval_accuracy_steel": 0.8935921113383022,
-      "eval_accuracy_tiles": 0.12977674291192987,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9793174076552572,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8821565046308723,
-      "eval_iou_concrete tiles": 0.9472753529221066,
-      "eval_iou_dark bricks": 0.8283616998308805,
-      "eval_iou_grass": 0.8144994399195002,
-      "eval_iou_red bricks": 0.8376401755241346,
-      "eval_iou_road": 0.9090402167148135,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.36637440115700987,
-      "eval_iou_speedway bricks": 0.9747450157921675,
-      "eval_iou_steel": 0.8313606971089309,
-      "eval_iou_tiles": 0.12977674291192987,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10846742987632751,
-      "eval_mean_accuracy": 0.7007927519870196,
-      "eval_mean_iou": 0.6538882810898156,
-      "eval_overall_accuracy": 0.9714558919270834,
-      "eval_runtime": 3.3367,
-      "eval_samples_per_second": 10.789,
-      "eval_steps_per_second": 0.599,
-      "step": 2100
-    },
-    {
-      "epoch": 191.0,
-      "learning_rate": 6.505263157894737e-05,
-      "loss": 0.0394,
-      "step": 2101
-    },
-    {
-      "epoch": 191.09,
-      "learning_rate": 6.503349282296652e-05,
-      "loss": 0.0398,
-      "step": 2102
-    },
-    {
-      "epoch": 191.18,
-      "learning_rate": 6.501435406698565e-05,
-      "loss": 0.0439,
-      "step": 2103
-    },
-    {
-      "epoch": 191.27,
-      "learning_rate": 6.499521531100478e-05,
-      "loss": 0.0495,
-      "step": 2104
-    },
-    {
-      "epoch": 191.36,
-      "learning_rate": 6.497607655502393e-05,
-      "loss": 0.0368,
-      "step": 2105
-    },
-    {
-      "epoch": 191.45,
-      "learning_rate": 6.495693779904306e-05,
-      "loss": 0.0466,
-      "step": 2106
-    },
-    {
-      "epoch": 191.55,
-      "learning_rate": 6.49377990430622e-05,
-      "loss": 0.0439,
-      "step": 2107
-    },
-    {
-      "epoch": 191.64,
-      "learning_rate": 6.491866028708135e-05,
-      "loss": 0.0404,
-      "step": 2108
-    },
-    {
-      "epoch": 191.73,
-      "learning_rate": 6.489952153110048e-05,
-      "loss": 0.0415,
-      "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.0409,
-      "step": 2111
-    },
-    {
-      "epoch": 192.0,
-      "learning_rate": 6.484210526315789e-05,
-      "loss": 0.0432,
-      "step": 2112
-    },
-    {
-      "epoch": 192.09,
-      "learning_rate": 6.482296650717704e-05,
-      "loss": 0.0434,
-      "step": 2113
-    },
-    {
-      "epoch": 192.18,
-      "learning_rate": 6.480382775119618e-05,
-      "loss": 0.0467,
-      "step": 2114
-    },
-    {
-      "epoch": 192.27,
-      "learning_rate": 6.478468899521531e-05,
-      "loss": 0.0399,
-      "step": 2115
-    },
-    {
-      "epoch": 192.36,
-      "learning_rate": 6.476555023923445e-05,
-      "loss": 0.0452,
-      "step": 2116
-    },
-    {
-      "epoch": 192.45,
-      "learning_rate": 6.474641148325359e-05,
-      "loss": 0.042,
-      "step": 2117
-    },
-    {
-      "epoch": 192.55,
-      "learning_rate": 6.472727272727272e-05,
-      "loss": 0.0413,
-      "step": 2118
-    },
-    {
-      "epoch": 192.64,
-      "learning_rate": 6.470813397129187e-05,
-      "loss": 0.041,
-      "step": 2119
-    },
-    {
-      "epoch": 192.73,
-      "learning_rate": 6.468899521531101e-05,
-      "loss": 0.0424,
-      "step": 2120
-    },
-    {
-      "epoch": 192.73,
-      "eval_accuracy_NAT": 0.9935327136840664,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9427912678655737,
-      "eval_accuracy_concrete tiles": 0.9848651623555311,
-      "eval_accuracy_dark bricks": 0.8553173472922521,
-      "eval_accuracy_grass": 0.8620948772704873,
-      "eval_accuracy_red bricks": 0.8790838068181818,
-      "eval_accuracy_road": 0.8894638005192342,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.45764066936840075,
-      "eval_accuracy_speedway bricks": 0.9885358550952359,
-      "eval_accuracy_steel": 0.8939548842424642,
-      "eval_accuracy_tiles": 0.0784823996712779,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9791588086365555,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8688431987184558,
-      "eval_iou_concrete tiles": 0.9333275382475661,
-      "eval_iou_dark bricks": 0.8213942485399571,
-      "eval_iou_grass": 0.8021202562518293,
-      "eval_iou_red bricks": 0.8395794471765304,
-      "eval_iou_road": 0.8821388396529848,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3031133022916096,
-      "eval_iou_speedway bricks": 0.9752040287872564,
-      "eval_iou_steel": 0.8347396298463339,
-      "eval_iou_tiles": 0.0784823996712779,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12418189644813538,
-      "eval_mean_accuracy": 0.6789048295525159,
-      "eval_mean_iou": 0.639853976755412,
-      "eval_overall_accuracy": 0.9684669706556532,
-      "eval_runtime": 3.2279,
-      "eval_samples_per_second": 11.153,
-      "eval_steps_per_second": 0.62,
-      "step": 2120
-    },
-    {
-      "epoch": 192.82,
-      "learning_rate": 6.466985645933015e-05,
-      "loss": 0.0422,
-      "step": 2121
-    },
-    {
-      "epoch": 192.91,
-      "learning_rate": 6.465071770334928e-05,
-      "loss": 0.039,
-      "step": 2122
-    },
-    {
-      "epoch": 193.0,
-      "learning_rate": 6.463157894736842e-05,
-      "loss": 0.0382,
-      "step": 2123
-    },
-    {
-      "epoch": 193.09,
-      "learning_rate": 6.461244019138756e-05,
-      "loss": 0.0354,
-      "step": 2124
-    },
-    {
-      "epoch": 193.18,
-      "learning_rate": 6.45933014354067e-05,
-      "loss": 0.0416,
-      "step": 2125
-    },
-    {
-      "epoch": 193.27,
-      "learning_rate": 6.457416267942585e-05,
-      "loss": 0.0411,
-      "step": 2126
-    },
-    {
-      "epoch": 193.36,
-      "learning_rate": 6.455502392344498e-05,
-      "loss": 0.0354,
-      "step": 2127
-    },
-    {
-      "epoch": 193.45,
-      "learning_rate": 6.453588516746411e-05,
-      "loss": 0.0452,
-      "step": 2128
-    },
-    {
-      "epoch": 193.55,
-      "learning_rate": 6.451674641148326e-05,
-      "loss": 0.0432,
-      "step": 2129
-    },
-    {
-      "epoch": 193.64,
-      "learning_rate": 6.449760765550239e-05,
-      "loss": 0.0388,
-      "step": 2130
-    },
-    {
-      "epoch": 193.73,
-      "learning_rate": 6.447846889952153e-05,
-      "loss": 0.04,
-      "step": 2131
-    },
-    {
-      "epoch": 193.82,
-      "learning_rate": 6.445933014354068e-05,
-      "loss": 0.0474,
-      "step": 2132
-    },
-    {
-      "epoch": 193.91,
-      "learning_rate": 6.444019138755981e-05,
-      "loss": 0.045,
-      "step": 2133
-    },
-    {
-      "epoch": 194.0,
-      "learning_rate": 6.442105263157894e-05,
-      "loss": 0.0518,
-      "step": 2134
-    },
-    {
-      "epoch": 194.09,
-      "learning_rate": 6.440191387559809e-05,
-      "loss": 0.0478,
-      "step": 2135
-    },
-    {
-      "epoch": 194.18,
-      "learning_rate": 6.438277511961724e-05,
-      "loss": 0.0416,
-      "step": 2136
-    },
-    {
-      "epoch": 194.27,
-      "learning_rate": 6.436363636363637e-05,
-      "loss": 0.0408,
-      "step": 2137
-    },
-    {
-      "epoch": 194.36,
-      "learning_rate": 6.434449760765551e-05,
-      "loss": 0.0398,
-      "step": 2138
-    },
-    {
-      "epoch": 194.45,
-      "learning_rate": 6.432535885167464e-05,
-      "loss": 0.0493,
-      "step": 2139
-    },
-    {
-      "epoch": 194.55,
-      "learning_rate": 6.430622009569378e-05,
-      "loss": 0.0447,
-      "step": 2140
-    },
-    {
-      "epoch": 194.55,
-      "eval_accuracy_NAT": 0.9928463188415207,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9496357139791407,
-      "eval_accuracy_concrete tiles": 0.983886748608818,
-      "eval_accuracy_dark bricks": 0.8604749624725762,
-      "eval_accuracy_grass": 0.8868644645372348,
-      "eval_accuracy_red bricks": 0.9291548295454546,
-      "eval_accuracy_road": 0.9397354492875946,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5942712210183884,
-      "eval_accuracy_speedway bricks": 0.9861083454920007,
-      "eval_accuracy_steel": 0.9056543104016885,
-      "eval_accuracy_tiles": 0.11847692096973017,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9795589160273369,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8869449531860307,
-      "eval_iou_concrete tiles": 0.9445797986321073,
-      "eval_iou_dark bricks": 0.8228798586572438,
-      "eval_iou_grass": 0.821935157862004,
-      "eval_iou_red bricks": 0.8297130172823847,
-      "eval_iou_road": 0.9313966713447293,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3982853553083774,
-      "eval_iou_speedway bricks": 0.9754710426508035,
-      "eval_iou_steel": 0.8337001730471477,
-      "eval_iou_tiles": 0.11847692096973017,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09687285125255585,
-      "eval_mean_accuracy": 0.7036237911657037,
-      "eval_mean_iou": 0.6571493742282997,
-      "eval_overall_accuracy": 0.9741036097208658,
-      "eval_runtime": 3.554,
-      "eval_samples_per_second": 10.13,
-      "eval_steps_per_second": 0.563,
-      "step": 2140
-    },
-    {
-      "epoch": 194.64,
-      "learning_rate": 6.428708133971292e-05,
-      "loss": 0.039,
-      "step": 2141
-    },
-    {
-      "epoch": 194.73,
-      "learning_rate": 6.426794258373207e-05,
-      "loss": 0.044,
-      "step": 2142
-    },
-    {
-      "epoch": 194.82,
-      "learning_rate": 6.42488038277512e-05,
-      "loss": 0.0382,
-      "step": 2143
-    },
-    {
-      "epoch": 194.91,
-      "learning_rate": 6.422966507177035e-05,
-      "loss": 0.0349,
-      "step": 2144
-    },
-    {
-      "epoch": 195.0,
-      "learning_rate": 6.421052631578948e-05,
-      "loss": 0.0426,
-      "step": 2145
-    },
-    {
-      "epoch": 195.09,
-      "learning_rate": 6.419138755980861e-05,
-      "loss": 0.0392,
-      "step": 2146
-    },
-    {
-      "epoch": 195.18,
-      "learning_rate": 6.417224880382775e-05,
-      "loss": 0.0404,
-      "step": 2147
-    },
-    {
-      "epoch": 195.27,
-      "learning_rate": 6.41531100478469e-05,
-      "loss": 0.0407,
-      "step": 2148
-    },
-    {
-      "epoch": 195.36,
-      "learning_rate": 6.413397129186603e-05,
-      "loss": 0.045,
-      "step": 2149
-    },
-    {
-      "epoch": 195.45,
-      "learning_rate": 6.411483253588518e-05,
-      "loss": 0.0457,
-      "step": 2150
-    },
-    {
-      "epoch": 195.55,
-      "learning_rate": 6.409569377990431e-05,
-      "loss": 0.0353,
-      "step": 2151
-    },
-    {
-      "epoch": 195.64,
-      "learning_rate": 6.407655502392344e-05,
-      "loss": 0.0434,
-      "step": 2152
-    },
-    {
-      "epoch": 195.73,
-      "learning_rate": 6.405741626794259e-05,
-      "loss": 0.0405,
-      "step": 2153
-    },
-    {
-      "epoch": 195.82,
-      "learning_rate": 6.403827751196173e-05,
-      "loss": 0.0424,
-      "step": 2154
-    },
-    {
-      "epoch": 195.91,
-      "learning_rate": 6.401913875598087e-05,
-      "loss": 0.0454,
-      "step": 2155
-    },
-    {
-      "epoch": 196.0,
-      "learning_rate": 6.400000000000001e-05,
-      "loss": 0.0399,
-      "step": 2156
-    },
-    {
-      "epoch": 196.09,
-      "learning_rate": 6.398086124401914e-05,
-      "loss": 0.0383,
-      "step": 2157
-    },
-    {
-      "epoch": 196.18,
-      "learning_rate": 6.396172248803827e-05,
-      "loss": 0.0358,
-      "step": 2158
-    },
-    {
-      "epoch": 196.27,
-      "learning_rate": 6.394258373205742e-05,
-      "loss": 0.0445,
-      "step": 2159
-    },
-    {
-      "epoch": 196.36,
-      "learning_rate": 6.392344497607657e-05,
-      "loss": 0.041,
-      "step": 2160
-    },
-    {
-      "epoch": 196.36,
-      "eval_accuracy_NAT": 0.9911310375396221,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9519936477960428,
-      "eval_accuracy_concrete tiles": 0.9870971687152205,
-      "eval_accuracy_dark bricks": 0.8729841037681383,
-      "eval_accuracy_grass": 0.8945431414421373,
-      "eval_accuracy_red bricks": 0.9222301136363636,
-      "eval_accuracy_road": 0.9329820178551779,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5270035563642379,
-      "eval_accuracy_speedway bricks": 0.9877224613728494,
-      "eval_accuracy_steel": 0.9020183365213377,
-      "eval_accuracy_tiles": 0.03328311190247911,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9801395814858749,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8793835929572597,
-      "eval_iou_concrete tiles": 0.9361208571345725,
-      "eval_iou_dark bricks": 0.815804618372779,
-      "eval_iou_grass": 0.8241609268686947,
-      "eval_iou_red bricks": 0.832505209168136,
-      "eval_iou_road": 0.9229113014063212,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3409492214671732,
-      "eval_iou_speedway bricks": 0.9750754877590396,
-      "eval_iou_steel": 0.8312931682965192,
-      "eval_iou_tiles": 0.03328311190247911,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10372213274240494,
-      "eval_mean_accuracy": 0.6925375920702774,
-      "eval_mean_iou": 0.64397131360145,
-      "eval_overall_accuracy": 0.9727885988023546,
-      "eval_runtime": 3.6334,
-      "eval_samples_per_second": 9.908,
-      "eval_steps_per_second": 0.55,
-      "step": 2160
-    },
-    {
-      "epoch": 196.45,
-      "learning_rate": 6.39043062200957e-05,
-      "loss": 0.0422,
-      "step": 2161
-    },
-    {
-      "epoch": 196.55,
-      "learning_rate": 6.388516746411484e-05,
-      "loss": 0.0416,
-      "step": 2162
-    },
-    {
-      "epoch": 196.64,
-      "learning_rate": 6.386602870813398e-05,
-      "loss": 0.0421,
-      "step": 2163
-    },
-    {
-      "epoch": 196.73,
-      "learning_rate": 6.384688995215311e-05,
-      "loss": 0.0404,
-      "step": 2164
-    },
-    {
-      "epoch": 196.82,
-      "learning_rate": 6.382775119617225e-05,
-      "loss": 0.0385,
-      "step": 2165
-    },
-    {
-      "epoch": 196.91,
-      "learning_rate": 6.38086124401914e-05,
-      "loss": 0.0401,
-      "step": 2166
-    },
-    {
-      "epoch": 197.0,
-      "learning_rate": 6.378947368421053e-05,
-      "loss": 0.0361,
-      "step": 2167
-    },
-    {
-      "epoch": 197.09,
-      "learning_rate": 6.377033492822966e-05,
-      "loss": 0.0384,
-      "step": 2168
-    },
-    {
-      "epoch": 197.18,
-      "learning_rate": 6.375119617224881e-05,
-      "loss": 0.0445,
-      "step": 2169
-    },
-    {
-      "epoch": 197.27,
-      "learning_rate": 6.373205741626794e-05,
-      "loss": 0.04,
-      "step": 2170
-    },
-    {
-      "epoch": 197.36,
-      "learning_rate": 6.371291866028709e-05,
-      "loss": 0.0351,
-      "step": 2171
-    },
-    {
-      "epoch": 197.45,
-      "learning_rate": 6.369377990430623e-05,
-      "loss": 0.0426,
-      "step": 2172
-    },
-    {
-      "epoch": 197.55,
-      "learning_rate": 6.367464114832536e-05,
-      "loss": 0.0389,
-      "step": 2173
-    },
-    {
-      "epoch": 197.64,
-      "learning_rate": 6.36555023923445e-05,
-      "loss": 0.0397,
-      "step": 2174
-    },
-    {
-      "epoch": 197.73,
-      "learning_rate": 6.363636363636364e-05,
-      "loss": 0.0374,
-      "step": 2175
-    },
-    {
-      "epoch": 197.82,
-      "learning_rate": 6.361722488038277e-05,
-      "loss": 0.0443,
-      "step": 2176
-    },
-    {
-      "epoch": 197.91,
-      "learning_rate": 6.359808612440192e-05,
-      "loss": 0.0404,
-      "step": 2177
-    },
-    {
-      "epoch": 198.0,
-      "learning_rate": 6.357894736842106e-05,
-      "loss": 0.0414,
-      "step": 2178
-    },
-    {
-      "epoch": 198.09,
-      "learning_rate": 6.35598086124402e-05,
-      "loss": 0.0383,
-      "step": 2179
-    },
-    {
-      "epoch": 198.18,
-      "learning_rate": 6.354066985645933e-05,
-      "loss": 0.0407,
-      "step": 2180
-    },
-    {
-      "epoch": 198.18,
-      "eval_accuracy_NAT": 0.9922264460698337,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9549095454740547,
-      "eval_accuracy_concrete tiles": 0.9853849446584725,
-      "eval_accuracy_dark bricks": 0.8594357415034063,
-      "eval_accuracy_grass": 0.8957594270875201,
-      "eval_accuracy_red bricks": 0.9149502840909091,
-      "eval_accuracy_road": 0.937661656108258,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5668127808562843,
-      "eval_accuracy_speedway bricks": 0.9863756551500252,
-      "eval_accuracy_steel": 0.9018534397467185,
-      "eval_accuracy_tiles": 0.08129023421449116,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9795040918657033,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8872049088806476,
-      "eval_iou_concrete tiles": 0.9427250921429825,
-      "eval_iou_dark bricks": 0.8159394869546153,
-      "eval_iou_grass": 0.8284132304599243,
-      "eval_iou_red bricks": 0.8389775317486161,
-      "eval_iou_road": 0.9311401944573531,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3776149282788767,
-      "eval_iou_speedway bricks": 0.9755072154145268,
-      "eval_iou_steel": 0.8361348713127097,
-      "eval_iou_tiles": 0.08129023421449116,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09696460515260696,
-      "eval_mean_accuracy": 0.6982046273046133,
-      "eval_mean_iou": 0.6534193681331112,
-      "eval_overall_accuracy": 0.974064826965332,
-      "eval_runtime": 3.3143,
-      "eval_samples_per_second": 10.862,
-      "eval_steps_per_second": 0.603,
-      "step": 2180
-    },
-    {
-      "epoch": 198.27,
-      "learning_rate": 6.352153110047847e-05,
-      "loss": 0.042,
-      "step": 2181
-    },
-    {
-      "epoch": 198.36,
-      "learning_rate": 6.35023923444976e-05,
-      "loss": 0.0377,
-      "step": 2182
-    },
-    {
-      "epoch": 198.45,
-      "learning_rate": 6.348325358851675e-05,
-      "loss": 0.0442,
-      "step": 2183
-    },
-    {
-      "epoch": 198.55,
-      "learning_rate": 6.34641148325359e-05,
-      "loss": 0.0423,
-      "step": 2184
-    },
-    {
-      "epoch": 198.64,
-      "learning_rate": 6.344497607655503e-05,
-      "loss": 0.0395,
-      "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.0434,
-      "step": 2187
-    },
-    {
-      "epoch": 198.91,
-      "learning_rate": 6.338755980861244e-05,
-      "loss": 0.0401,
-      "step": 2188
-    },
-    {
-      "epoch": 199.0,
-      "learning_rate": 6.336842105263158e-05,
-      "loss": 0.0399,
-      "step": 2189
-    },
-    {
-      "epoch": 199.09,
-      "learning_rate": 6.334928229665073e-05,
-      "loss": 0.0417,
-      "step": 2190
-    },
-    {
-      "epoch": 199.18,
-      "learning_rate": 6.333014354066986e-05,
-      "loss": 0.0448,
-      "step": 2191
-    },
-    {
-      "epoch": 199.27,
-      "learning_rate": 6.331100478468899e-05,
-      "loss": 0.0405,
-      "step": 2192
-    },
-    {
-      "epoch": 199.36,
-      "learning_rate": 6.329186602870814e-05,
-      "loss": 0.0455,
-      "step": 2193
-    },
-    {
-      "epoch": 199.45,
-      "learning_rate": 6.327272727272727e-05,
-      "loss": 0.0366,
-      "step": 2194
-    },
-    {
-      "epoch": 199.55,
-      "learning_rate": 6.32535885167464e-05,
-      "loss": 0.0377,
-      "step": 2195
-    },
-    {
-      "epoch": 199.64,
-      "learning_rate": 6.323444976076556e-05,
-      "loss": 0.0341,
-      "step": 2196
-    },
-    {
-      "epoch": 199.73,
-      "learning_rate": 6.32153110047847e-05,
-      "loss": 0.0421,
-      "step": 2197
-    },
-    {
-      "epoch": 199.82,
-      "learning_rate": 6.319617224880383e-05,
-      "loss": 0.0362,
-      "step": 2198
-    },
-    {
-      "epoch": 199.91,
-      "learning_rate": 6.317703349282297e-05,
-      "loss": 0.0579,
-      "step": 2199
-    },
-    {
-      "epoch": 200.0,
-      "learning_rate": 6.31578947368421e-05,
-      "loss": 0.0423,
-      "step": 2200
-    },
-    {
-      "epoch": 200.0,
-      "eval_accuracy_NAT": 0.9931145245383509,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9535575561182883,
-      "eval_accuracy_concrete tiles": 0.9799730936219654,
-      "eval_accuracy_dark bricks": 0.8527770293676148,
-      "eval_accuracy_grass": 0.8819329155555244,
-      "eval_accuracy_red bricks": 0.9151278409090909,
-      "eval_accuracy_road": 0.9100554180416379,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.49948997877208945,
-      "eval_accuracy_speedway bricks": 0.9866870135778825,
-      "eval_accuracy_steel": 0.8952740584394169,
-      "eval_accuracy_tiles": 0.045815641692918775,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.979164865325187,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8851457683841473,
-      "eval_iou_concrete tiles": 0.9488720468944283,
-      "eval_iou_dark bricks": 0.818742840249806,
-      "eval_iou_grass": 0.825585237775851,
-      "eval_iou_red bricks": 0.8417442430181283,
-      "eval_iou_road": 0.902977795214939,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3363531726877808,
-      "eval_iou_speedway bricks": 0.9757475979124424,
-      "eval_iou_steel": 0.8350392580573222,
-      "eval_iou_tiles": 0.045815641692918775,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11419142782688141,
-      "eval_mean_accuracy": 0.6856773131257523,
-      "eval_mean_iou": 0.64578372824715,
-      "eval_overall_accuracy": 0.9712724685668945,
-      "eval_runtime": 3.3604,
-      "eval_samples_per_second": 10.713,
-      "eval_steps_per_second": 0.595,
-      "step": 2200
-    },
-    {
-      "epoch": 200.09,
-      "learning_rate": 6.313875598086124e-05,
-      "loss": 0.043,
-      "step": 2201
-    },
-    {
-      "epoch": 200.18,
-      "learning_rate": 6.31196172248804e-05,
-      "loss": 0.0445,
-      "step": 2202
-    },
-    {
-      "epoch": 200.27,
-      "learning_rate": 6.310047846889953e-05,
-      "loss": 0.041,
-      "step": 2203
-    },
-    {
-      "epoch": 200.36,
-      "learning_rate": 6.308133971291866e-05,
-      "loss": 0.0369,
-      "step": 2204
-    },
-    {
-      "epoch": 200.45,
-      "learning_rate": 6.30622009569378e-05,
-      "loss": 0.0367,
-      "step": 2205
-    },
-    {
-      "epoch": 200.55,
-      "learning_rate": 6.304306220095694e-05,
-      "loss": 0.0445,
-      "step": 2206
-    },
-    {
-      "epoch": 200.64,
-      "learning_rate": 6.302392344497607e-05,
-      "loss": 0.0416,
-      "step": 2207
-    },
-    {
-      "epoch": 200.73,
-      "learning_rate": 6.300478468899523e-05,
-      "loss": 0.0361,
-      "step": 2208
-    },
-    {
-      "epoch": 200.82,
-      "learning_rate": 6.298564593301436e-05,
-      "loss": 0.0404,
-      "step": 2209
-    },
-    {
-      "epoch": 200.91,
-      "learning_rate": 6.296650717703349e-05,
-      "loss": 0.0376,
-      "step": 2210
-    },
-    {
-      "epoch": 201.0,
-      "learning_rate": 6.294736842105264e-05,
-      "loss": 0.0444,
-      "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.0366,
-      "step": 2213
-    },
-    {
-      "epoch": 201.27,
-      "learning_rate": 6.288995215311006e-05,
-      "loss": 0.0412,
-      "step": 2214
-    },
-    {
-      "epoch": 201.36,
-      "learning_rate": 6.287081339712919e-05,
-      "loss": 0.0392,
-      "step": 2215
-    },
-    {
-      "epoch": 201.45,
-      "learning_rate": 6.285167464114832e-05,
-      "loss": 0.0363,
-      "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.0427,
-      "step": 2218
-    },
-    {
-      "epoch": 201.73,
-      "learning_rate": 6.279425837320575e-05,
-      "loss": 0.0383,
-      "step": 2219
-    },
-    {
-      "epoch": 201.82,
-      "learning_rate": 6.277511961722489e-05,
-      "loss": 0.0473,
-      "step": 2220
-    },
-    {
-      "epoch": 201.82,
-      "eval_accuracy_NAT": 0.9925163552539732,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9470497660843813,
-      "eval_accuracy_concrete tiles": 0.9885036384761205,
-      "eval_accuracy_dark bricks": 0.8598976174897041,
-      "eval_accuracy_grass": 0.9050283625230238,
-      "eval_accuracy_red bricks": 0.9083806818181818,
-      "eval_accuracy_road": 0.9181788831021528,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4843271855098834,
-      "eval_accuracy_speedway bricks": 0.9876198820458421,
-      "eval_accuracy_steel": 0.9042526878174263,
-      "eval_accuracy_tiles": 0.03314614436378578,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799070348896157,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8820303599624248,
-      "eval_iou_concrete tiles": 0.9340960966166826,
-      "eval_iou_dark bricks": 0.8230244980659421,
-      "eval_iou_grass": 0.8343854018644008,
-      "eval_iou_red bricks": 0.8494105927278764,
-      "eval_iou_road": 0.9101699222308526,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32553830189378496,
-      "eval_iou_speedway bricks": 0.9753367597765363,
-      "eval_iou_steel": 0.836702776930119,
-      "eval_iou_tiles": 0.03314614436378578,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10966138541698456,
-      "eval_mean_accuracy": 0.6868385541911135,
-      "eval_mean_iou": 0.6449036837940016,
-      "eval_overall_accuracy": 0.9720576604207357,
-      "eval_runtime": 3.2404,
-      "eval_samples_per_second": 11.11,
-      "eval_steps_per_second": 0.617,
-      "step": 2220
-    },
-    {
-      "epoch": 201.91,
-      "learning_rate": 6.275598086124402e-05,
-      "loss": 0.039,
-      "step": 2221
-    },
-    {
-      "epoch": 202.0,
-      "learning_rate": 6.273684210526316e-05,
-      "loss": 0.0474,
-      "step": 2222
-    },
-    {
-      "epoch": 202.09,
-      "learning_rate": 6.27177033492823e-05,
-      "loss": 0.0392,
-      "step": 2223
-    },
-    {
-      "epoch": 202.18,
-      "learning_rate": 6.269856459330143e-05,
-      "loss": 0.0374,
-      "step": 2224
-    },
-    {
-      "epoch": 202.27,
-      "learning_rate": 6.267942583732058e-05,
-      "loss": 0.0397,
-      "step": 2225
-    },
-    {
-      "epoch": 202.36,
-      "learning_rate": 6.266028708133973e-05,
-      "loss": 0.0421,
-      "step": 2226
-    },
-    {
-      "epoch": 202.45,
-      "learning_rate": 6.264114832535886e-05,
-      "loss": 0.0426,
-      "step": 2227
-    },
-    {
-      "epoch": 202.55,
-      "learning_rate": 6.262200956937799e-05,
-      "loss": 0.0401,
-      "step": 2228
-    },
-    {
-      "epoch": 202.64,
-      "learning_rate": 6.260287081339713e-05,
-      "loss": 0.0422,
-      "step": 2229
-    },
-    {
-      "epoch": 202.73,
-      "learning_rate": 6.258373205741627e-05,
-      "loss": 0.05,
-      "step": 2230
-    },
-    {
-      "epoch": 202.82,
-      "learning_rate": 6.256459330143541e-05,
-      "loss": 0.0402,
-      "step": 2231
-    },
-    {
-      "epoch": 202.91,
-      "learning_rate": 6.254545454545456e-05,
-      "loss": 0.0392,
-      "step": 2232
-    },
-    {
-      "epoch": 203.0,
-      "learning_rate": 6.252631578947369e-05,
-      "loss": 0.0361,
-      "step": 2233
-    },
-    {
-      "epoch": 203.09,
-      "learning_rate": 6.250717703349282e-05,
-      "loss": 0.0369,
-      "step": 2234
-    },
-    {
-      "epoch": 203.18,
-      "learning_rate": 6.248803827751197e-05,
-      "loss": 0.0499,
-      "step": 2235
-    },
-    {
-      "epoch": 203.27,
-      "learning_rate": 6.24688995215311e-05,
-      "loss": 0.0364,
-      "step": 2236
-    },
-    {
-      "epoch": 203.36,
-      "learning_rate": 6.244976076555024e-05,
-      "loss": 0.0417,
-      "step": 2237
-    },
-    {
-      "epoch": 203.45,
-      "learning_rate": 6.243062200956939e-05,
-      "loss": 0.0416,
-      "step": 2238
-    },
-    {
-      "epoch": 203.55,
-      "learning_rate": 6.241148325358852e-05,
-      "loss": 0.0359,
-      "step": 2239
-    },
-    {
-      "epoch": 203.64,
-      "learning_rate": 6.239234449760765e-05,
-      "loss": 0.0381,
-      "step": 2240
-    },
-    {
-      "epoch": 203.64,
-      "eval_accuracy_NAT": 0.9926790431832346,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9463563350358385,
-      "eval_accuracy_concrete tiles": 0.9801259707698893,
-      "eval_accuracy_dark bricks": 0.8731380624302375,
-      "eval_accuracy_grass": 0.9132347955556643,
-      "eval_accuracy_red bricks": 0.9332386363636364,
-      "eval_accuracy_road": 0.9424537507844742,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4723899319052739,
-      "eval_accuracy_speedway bricks": 0.9868414859761991,
-      "eval_accuracy_steel": 0.9039064045907262,
-      "eval_accuracy_tiles": 0.09485002054513081,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9797065095830372,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8858651640532895,
-      "eval_iou_concrete tiles": 0.94449027695934,
-      "eval_iou_dark bricks": 0.8194263834705967,
-      "eval_iou_grass": 0.8374797274341502,
-      "eval_iou_red bricks": 0.8610747051114024,
-      "eval_iou_road": 0.9350523614541956,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3155791296020038,
-      "eval_iou_speedway bricks": 0.9758083570804634,
-      "eval_iou_steel": 0.837596455038582,
-      "eval_iou_tiles": 0.09485002054513081,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09588578343391418,
-      "eval_mean_accuracy": 0.695324187472331,
-      "eval_mean_iou": 0.6528406992563224,
-      "eval_overall_accuracy": 0.9744073020087348,
-      "eval_runtime": 3.4227,
-      "eval_samples_per_second": 10.518,
-      "eval_steps_per_second": 0.584,
-      "step": 2240
-    },
-    {
-      "epoch": 203.73,
-      "learning_rate": 6.23732057416268e-05,
-      "loss": 0.0392,
-      "step": 2241
-    },
-    {
-      "epoch": 203.82,
-      "learning_rate": 6.235406698564593e-05,
-      "loss": 0.0418,
-      "step": 2242
-    },
-    {
-      "epoch": 203.91,
-      "learning_rate": 6.233492822966508e-05,
-      "loss": 0.0368,
-      "step": 2243
-    },
-    {
-      "epoch": 204.0,
-      "learning_rate": 6.231578947368422e-05,
-      "loss": 0.0388,
-      "step": 2244
-    },
-    {
-      "epoch": 204.09,
-      "learning_rate": 6.229665071770336e-05,
-      "loss": 0.0458,
-      "step": 2245
-    },
-    {
-      "epoch": 204.18,
-      "learning_rate": 6.227751196172249e-05,
-      "loss": 0.0445,
-      "step": 2246
-    },
-    {
-      "epoch": 204.27,
-      "learning_rate": 6.225837320574163e-05,
-      "loss": 0.0408,
-      "step": 2247
-    },
-    {
-      "epoch": 204.36,
-      "learning_rate": 6.223923444976076e-05,
-      "loss": 0.0353,
-      "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.0373,
-      "step": 2250
-    },
-    {
-      "epoch": 204.64,
-      "learning_rate": 6.218181818181819e-05,
-      "loss": 0.0409,
-      "step": 2251
-    },
-    {
-      "epoch": 204.73,
-      "learning_rate": 6.216267942583732e-05,
-      "loss": 0.0379,
-      "step": 2252
-    },
-    {
-      "epoch": 204.82,
-      "learning_rate": 6.214354066985647e-05,
-      "loss": 0.0404,
-      "step": 2253
-    },
-    {
-      "epoch": 204.91,
-      "learning_rate": 6.21244019138756e-05,
-      "loss": 0.0362,
-      "step": 2254
-    },
-    {
-      "epoch": 205.0,
-      "learning_rate": 6.210526315789474e-05,
-      "loss": 0.0356,
-      "step": 2255
-    },
-    {
-      "epoch": 205.09,
-      "learning_rate": 6.208612440191387e-05,
-      "loss": 0.0347,
-      "step": 2256
-    },
-    {
-      "epoch": 205.18,
-      "learning_rate": 6.206698564593302e-05,
-      "loss": 0.0402,
-      "step": 2257
-    },
-    {
-      "epoch": 205.27,
-      "learning_rate": 6.204784688995215e-05,
-      "loss": 0.0387,
-      "step": 2258
-    },
-    {
-      "epoch": 205.36,
-      "learning_rate": 6.20287081339713e-05,
-      "loss": 0.0372,
-      "step": 2259
-    },
-    {
-      "epoch": 205.45,
-      "learning_rate": 6.200956937799043e-05,
-      "loss": 0.0362,
-      "step": 2260
-    },
-    {
-      "epoch": 205.45,
-      "eval_accuracy_NAT": 0.992046289480055,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9459861474741406,
-      "eval_accuracy_concrete tiles": 0.9868525652785421,
-      "eval_accuracy_dark bricks": 0.8806435472075748,
-      "eval_accuracy_grass": 0.9109245518441756,
-      "eval_accuracy_red bricks": 0.9389204545454546,
-      "eval_accuracy_road": 0.9502035703100532,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5285198356904585,
-      "eval_accuracy_speedway bricks": 0.9874829084738972,
-      "eval_accuracy_steel": 0.9014411978101708,
-      "eval_accuracy_tiles": 0.15792357211340913,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9798090752766652,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8868492261883805,
-      "eval_iou_concrete tiles": 0.9368668543728774,
-      "eval_iou_dark bricks": 0.8099975218607286,
-      "eval_iou_grass": 0.8348233018042979,
-      "eval_iou_red bricks": 0.8578844905905256,
-      "eval_iou_road": 0.9408900910350311,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3510079279345259,
-      "eval_iou_speedway bricks": 0.975195097417611,
-      "eval_iou_steel": 0.8325325332947527,
-      "eval_iou_tiles": 0.15790194467269242,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08997855335474014,
-      "eval_mean_accuracy": 0.7062265107867641,
-      "eval_mean_iou": 0.6587506203421606,
-      "eval_overall_accuracy": 0.9750488069322374,
-      "eval_runtime": 3.4704,
-      "eval_samples_per_second": 10.373,
-      "eval_steps_per_second": 0.576,
-      "step": 2260
-    },
-    {
-      "epoch": 205.55,
-      "learning_rate": 6.199043062200958e-05,
-      "loss": 0.0426,
-      "step": 2261
-    },
-    {
-      "epoch": 205.64,
-      "learning_rate": 6.197129186602871e-05,
-      "loss": 0.042,
-      "step": 2262
-    },
-    {
-      "epoch": 205.73,
-      "learning_rate": 6.195215311004785e-05,
-      "loss": 0.0476,
-      "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.039,
-      "step": 2265
-    },
-    {
-      "epoch": 206.0,
-      "learning_rate": 6.189473684210526e-05,
-      "loss": 0.0418,
-      "step": 2266
-    },
-    {
-      "epoch": 206.09,
-      "learning_rate": 6.187559808612441e-05,
-      "loss": 0.0433,
-      "step": 2267
-    },
-    {
-      "epoch": 206.18,
-      "learning_rate": 6.185645933014354e-05,
-      "loss": 0.0366,
-      "step": 2268
-    },
-    {
-      "epoch": 206.27,
-      "learning_rate": 6.183732057416269e-05,
-      "loss": 0.0478,
-      "step": 2269
-    },
-    {
-      "epoch": 206.36,
-      "learning_rate": 6.181818181818182e-05,
-      "loss": 0.0386,
-      "step": 2270
-    },
-    {
-      "epoch": 206.45,
-      "learning_rate": 6.179904306220095e-05,
-      "loss": 0.04,
-      "step": 2271
-    },
-    {
-      "epoch": 206.55,
-      "learning_rate": 6.17799043062201e-05,
-      "loss": 0.0379,
-      "step": 2272
-    },
-    {
-      "epoch": 206.64,
-      "learning_rate": 6.176076555023924e-05,
-      "loss": 0.0421,
-      "step": 2273
-    },
-    {
-      "epoch": 206.73,
-      "learning_rate": 6.174162679425837e-05,
-      "loss": 0.0401,
-      "step": 2274
-    },
-    {
-      "epoch": 206.82,
-      "learning_rate": 6.172248803827752e-05,
-      "loss": 0.0413,
-      "step": 2275
-    },
-    {
-      "epoch": 206.91,
-      "learning_rate": 6.170334928229665e-05,
-      "loss": 0.0399,
-      "step": 2276
-    },
-    {
-      "epoch": 207.0,
-      "learning_rate": 6.168421052631578e-05,
-      "loss": 0.041,
-      "step": 2277
-    },
-    {
-      "epoch": 207.09,
-      "learning_rate": 6.166507177033493e-05,
-      "loss": 0.0409,
-      "step": 2278
-    },
-    {
-      "epoch": 207.18,
-      "learning_rate": 6.164593301435407e-05,
-      "loss": 0.0375,
-      "step": 2279
-    },
-    {
-      "epoch": 207.27,
-      "learning_rate": 6.16267942583732e-05,
-      "loss": 0.039,
-      "step": 2280
-    },
-    {
-      "epoch": 207.27,
-      "eval_accuracy_NAT": 0.9929387792264807,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.945460373835787,
-      "eval_accuracy_concrete tiles": 0.980890356509509,
-      "eval_accuracy_dark bricks": 0.8770640083137677,
-      "eval_accuracy_grass": 0.902945298601621,
-      "eval_accuracy_red bricks": 0.9458451704545454,
-      "eval_accuracy_road": 0.9511153137836508,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.501860888263998,
-      "eval_accuracy_speedway bricks": 0.9875112686407757,
-      "eval_accuracy_steel": 0.8980360794142866,
-      "eval_accuracy_tiles": 0.25332146281331325,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9796330846041801,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.888484570425988,
-      "eval_iou_concrete tiles": 0.950801695266886,
-      "eval_iou_dark bricks": 0.8113294880011394,
-      "eval_iou_grass": 0.8337840890753591,
-      "eval_iou_red bricks": 0.8490596110934013,
-      "eval_iou_road": 0.9429773031347857,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3351868900754925,
-      "eval_iou_speedway bricks": 0.9755381814670295,
-      "eval_iou_steel": 0.8354887702503682,
-      "eval_iou_tiles": 0.2531654233112039,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08814884722232819,
-      "eval_mean_accuracy": 0.710537615373672,
-      "eval_mean_iou": 0.6658037774389102,
-      "eval_overall_accuracy": 0.975361188252767,
-      "eval_runtime": 3.897,
-      "eval_samples_per_second": 9.238,
-      "eval_steps_per_second": 0.513,
-      "step": 2280
-    },
-    {
-      "epoch": 207.36,
-      "learning_rate": 6.160765550239235e-05,
-      "loss": 0.0372,
-      "step": 2281
-    },
-    {
-      "epoch": 207.45,
-      "learning_rate": 6.158851674641148e-05,
-      "loss": 0.0412,
-      "step": 2282
-    },
-    {
-      "epoch": 207.55,
-      "learning_rate": 6.156937799043062e-05,
-      "loss": 0.0391,
-      "step": 2283
-    },
-    {
-      "epoch": 207.64,
-      "learning_rate": 6.155023923444977e-05,
-      "loss": 0.0437,
-      "step": 2284
-    },
-    {
-      "epoch": 207.73,
-      "learning_rate": 6.15311004784689e-05,
-      "loss": 0.0405,
-      "step": 2285
-    },
-    {
-      "epoch": 207.82,
-      "learning_rate": 6.151196172248804e-05,
-      "loss": 0.0371,
-      "step": 2286
-    },
-    {
-      "epoch": 207.91,
-      "learning_rate": 6.149282296650718e-05,
-      "loss": 0.0434,
-      "step": 2287
-    },
-    {
-      "epoch": 208.0,
-      "learning_rate": 6.147368421052632e-05,
-      "loss": 0.0378,
-      "step": 2288
-    },
-    {
-      "epoch": 208.09,
-      "learning_rate": 6.145454545454545e-05,
-      "loss": 0.0418,
-      "step": 2289
-    },
-    {
-      "epoch": 208.18,
-      "learning_rate": 6.143540669856461e-05,
-      "loss": 0.0377,
-      "step": 2290
-    },
-    {
-      "epoch": 208.27,
-      "learning_rate": 6.141626794258374e-05,
-      "loss": 0.0436,
-      "step": 2291
-    },
-    {
-      "epoch": 208.36,
-      "learning_rate": 6.139712918660287e-05,
-      "loss": 0.0323,
-      "step": 2292
-    },
-    {
-      "epoch": 208.45,
-      "learning_rate": 6.137799043062202e-05,
-      "loss": 0.0403,
-      "step": 2293
-    },
-    {
-      "epoch": 208.55,
-      "learning_rate": 6.135885167464115e-05,
-      "loss": 0.0419,
-      "step": 2294
-    },
-    {
-      "epoch": 208.64,
-      "learning_rate": 6.133971291866028e-05,
-      "loss": 0.0447,
-      "step": 2295
-    },
-    {
-      "epoch": 208.73,
-      "learning_rate": 6.132057416267944e-05,
-      "loss": 0.0437,
-      "step": 2296
-    },
-    {
-      "epoch": 208.82,
-      "learning_rate": 6.130143540669857e-05,
-      "loss": 0.0353,
-      "step": 2297
-    },
-    {
-      "epoch": 208.91,
-      "learning_rate": 6.12822966507177e-05,
-      "loss": 0.0431,
-      "step": 2298
-    },
-    {
-      "epoch": 209.0,
-      "learning_rate": 6.126315789473685e-05,
-      "loss": 0.0361,
-      "step": 2299
-    },
-    {
-      "epoch": 209.09,
-      "learning_rate": 6.124401913875598e-05,
-      "loss": 0.0398,
-      "step": 2300
-    },
-    {
-      "epoch": 209.09,
-      "eval_accuracy_NAT": 0.9918717793259991,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9448688784926391,
-      "eval_accuracy_concrete tiles": 0.9800648199107197,
-      "eval_accuracy_dark bricks": 0.8733305107578615,
-      "eval_accuracy_grass": 0.9177644110626073,
-      "eval_accuracy_red bricks": 0.8957741477272727,
-      "eval_accuracy_road": 0.9274051466347113,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4698536101232322,
-      "eval_accuracy_speedway bricks": 0.9874744607646142,
-      "eval_accuracy_steel": 0.9060500626607744,
-      "eval_accuracy_tiles": 0.13292699630187646,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9793520299957785,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.884435031192209,
-      "eval_iou_concrete tiles": 0.9503676470588235,
-      "eval_iou_dark bricks": 0.8197550489540807,
-      "eval_iou_grass": 0.8301760331832667,
-      "eval_iou_red bricks": 0.8480416876786014,
-      "eval_iou_road": 0.9201561603931364,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3127844663045074,
-      "eval_iou_speedway bricks": 0.975472746303745,
-      "eval_iou_steel": 0.8326425773406778,
-      "eval_iou_tiles": 0.13292699630187646,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09984607994556427,
-      "eval_mean_accuracy": 0.6944142172124852,
-      "eval_mean_iou": 0.6527777249774387,
-      "eval_overall_accuracy": 0.972789446512858,
-      "eval_runtime": 3.5145,
-      "eval_samples_per_second": 10.243,
-      "eval_steps_per_second": 0.569,
-      "step": 2300
-    },
-    {
-      "epoch": 209.18,
-      "learning_rate": 6.122488038277511e-05,
-      "loss": 0.0398,
-      "step": 2301
-    },
-    {
-      "epoch": 209.27,
-      "learning_rate": 6.120574162679427e-05,
-      "loss": 0.0393,
-      "step": 2302
-    },
-    {
-      "epoch": 209.36,
-      "learning_rate": 6.11866028708134e-05,
-      "loss": 0.0379,
-      "step": 2303
-    },
-    {
-      "epoch": 209.45,
-      "learning_rate": 6.116746411483254e-05,
-      "loss": 0.0386,
-      "step": 2304
-    },
-    {
-      "epoch": 209.55,
-      "learning_rate": 6.114832535885168e-05,
-      "loss": 0.0455,
-      "step": 2305
-    },
-    {
-      "epoch": 209.64,
-      "learning_rate": 6.112918660287081e-05,
-      "loss": 0.0345,
-      "step": 2306
-    },
-    {
-      "epoch": 209.73,
-      "learning_rate": 6.111004784688995e-05,
-      "loss": 0.038,
-      "step": 2307
-    },
-    {
-      "epoch": 209.82,
-      "learning_rate": 6.10909090909091e-05,
-      "loss": 0.0391,
-      "step": 2308
-    },
-    {
-      "epoch": 209.91,
-      "learning_rate": 6.107177033492824e-05,
-      "loss": 0.0372,
-      "step": 2309
-    },
-    {
-      "epoch": 210.0,
-      "learning_rate": 6.105263157894737e-05,
-      "loss": 0.0383,
-      "step": 2310
-    },
-    {
-      "epoch": 210.09,
-      "learning_rate": 6.103349282296651e-05,
-      "loss": 0.0395,
-      "step": 2311
-    },
-    {
-      "epoch": 210.18,
-      "learning_rate": 6.1014354066985646e-05,
-      "loss": 0.04,
-      "step": 2312
-    },
-    {
-      "epoch": 210.27,
-      "learning_rate": 6.0995215311004785e-05,
-      "loss": 0.0382,
-      "step": 2313
-    },
-    {
-      "epoch": 210.36,
-      "learning_rate": 6.097607655502393e-05,
-      "loss": 0.0394,
-      "step": 2314
-    },
-    {
-      "epoch": 210.45,
-      "learning_rate": 6.095693779904307e-05,
-      "loss": 0.0334,
-      "step": 2315
-    },
-    {
-      "epoch": 210.55,
-      "learning_rate": 6.09377990430622e-05,
-      "loss": 0.0388,
-      "step": 2316
-    },
-    {
-      "epoch": 210.64,
-      "learning_rate": 6.091866028708134e-05,
-      "loss": 0.0394,
-      "step": 2317
-    },
-    {
-      "epoch": 210.73,
-      "learning_rate": 6.089952153110048e-05,
-      "loss": 0.0392,
-      "step": 2318
-    },
-    {
-      "epoch": 210.82,
-      "learning_rate": 6.088038277511962e-05,
-      "loss": 0.0332,
-      "step": 2319
-    },
-    {
-      "epoch": 210.91,
-      "learning_rate": 6.086124401913876e-05,
-      "loss": 0.0365,
-      "step": 2320
-    },
-    {
-      "epoch": 210.91,
-      "eval_accuracy_NAT": 0.9932682134606793,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.944474548263874,
-      "eval_accuracy_concrete tiles": 0.986149330398092,
-      "eval_accuracy_dark bricks": 0.867672529925715,
-      "eval_accuracy_grass": 0.8949485699905982,
-      "eval_accuracy_red bricks": 0.9243607954545454,
-      "eval_accuracy_road": 0.9260871089872029,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.49359027375734016,
-      "eval_accuracy_speedway bricks": 0.9888055783844842,
-      "eval_accuracy_steel": 0.8963788668293649,
-      "eval_accuracy_tiles": 0.3112587316805917,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9794711197237296,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8889991453023824,
-      "eval_iou_concrete tiles": 0.936797467251445,
-      "eval_iou_dark bricks": 0.8176048164804874,
-      "eval_iou_grass": 0.8323689651137087,
-      "eval_iou_red bricks": 0.8537225319776977,
-      "eval_iou_road": 0.919241045387003,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33062490766730684,
-      "eval_iou_speedway bricks": 0.9752524125371589,
-      "eval_iou_steel": 0.8317204342204916,
-      "eval_iou_tiles": 0.311088295687885,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09843525290489197,
-      "eval_mean_accuracy": 0.7097688113178837,
-      "eval_mean_iou": 0.6674531647191766,
-      "eval_overall_accuracy": 0.9732888539632162,
-      "eval_runtime": 3.294,
-      "eval_samples_per_second": 10.929,
-      "eval_steps_per_second": 0.607,
-      "step": 2320
-    },
-    {
-      "epoch": 211.0,
-      "learning_rate": 6.08421052631579e-05,
-      "loss": 0.0441,
-      "step": 2321
-    },
-    {
-      "epoch": 211.09,
-      "learning_rate": 6.0822966507177034e-05,
-      "loss": 0.0367,
-      "step": 2322
-    },
-    {
-      "epoch": 211.18,
-      "learning_rate": 6.080382775119617e-05,
-      "loss": 0.0415,
-      "step": 2323
-    },
-    {
-      "epoch": 211.27,
-      "learning_rate": 6.078468899521531e-05,
-      "loss": 0.0419,
-      "step": 2324
-    },
-    {
-      "epoch": 211.36,
-      "learning_rate": 6.076555023923445e-05,
-      "loss": 0.0399,
-      "step": 2325
-    },
-    {
-      "epoch": 211.45,
-      "learning_rate": 6.0746411483253596e-05,
-      "loss": 0.0398,
-      "step": 2326
-    },
-    {
-      "epoch": 211.55,
-      "learning_rate": 6.0727272727272735e-05,
-      "loss": 0.0442,
-      "step": 2327
-    },
-    {
-      "epoch": 211.64,
-      "learning_rate": 6.070813397129187e-05,
-      "loss": 0.0384,
-      "step": 2328
-    },
-    {
-      "epoch": 211.73,
-      "learning_rate": 6.0688995215311006e-05,
-      "loss": 0.0477,
-      "step": 2329
-    },
-    {
-      "epoch": 211.82,
-      "learning_rate": 6.0669856459330144e-05,
-      "loss": 0.0357,
-      "step": 2330
-    },
-    {
-      "epoch": 211.91,
-      "learning_rate": 6.065071770334928e-05,
-      "loss": 0.0378,
-      "step": 2331
-    },
-    {
-      "epoch": 212.0,
-      "learning_rate": 6.063157894736843e-05,
-      "loss": 0.0447,
-      "step": 2332
-    },
-    {
-      "epoch": 212.09,
-      "learning_rate": 6.061244019138757e-05,
-      "loss": 0.0391,
-      "step": 2333
-    },
-    {
-      "epoch": 212.18,
-      "learning_rate": 6.05933014354067e-05,
-      "loss": 0.0391,
-      "step": 2334
-    },
-    {
-      "epoch": 212.27,
-      "learning_rate": 6.057416267942584e-05,
-      "loss": 0.0441,
-      "step": 2335
-    },
-    {
-      "epoch": 212.36,
-      "learning_rate": 6.055502392344498e-05,
-      "loss": 0.0495,
-      "step": 2336
-    },
-    {
-      "epoch": 212.45,
-      "learning_rate": 6.0535885167464116e-05,
-      "loss": 0.0404,
-      "step": 2337
-    },
-    {
-      "epoch": 212.55,
-      "learning_rate": 6.051674641148326e-05,
-      "loss": 0.0375,
-      "step": 2338
-    },
-    {
-      "epoch": 212.64,
-      "learning_rate": 6.04976076555024e-05,
-      "loss": 0.0395,
-      "step": 2339
-    },
-    {
-      "epoch": 212.73,
-      "learning_rate": 6.047846889952153e-05,
-      "loss": 0.0367,
-      "step": 2340
-    },
-    {
-      "epoch": 212.73,
-      "eval_accuracy_NAT": 0.9933696728525723,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9422011137817073,
-      "eval_accuracy_concrete tiles": 0.9843148046230049,
-      "eval_accuracy_dark bricks": 0.8790654709210577,
-      "eval_accuracy_grass": 0.8750441253053821,
-      "eval_accuracy_red bricks": 0.9140625,
-      "eval_accuracy_road": 0.9204401036533827,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4681719185068784,
-      "eval_accuracy_speedway bricks": 0.9884121564950212,
-      "eval_accuracy_steel": 0.898167996833982,
-      "eval_accuracy_tiles": 0.25058211203944664,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9795715106746491,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8815538488567628,
-      "eval_iou_concrete tiles": 0.9418390333811182,
-      "eval_iou_dark bricks": 0.8253767482201583,
-      "eval_iou_grass": 0.8146985129022811,
-      "eval_iou_red bricks": 0.8511904761904762,
-      "eval_iou_road": 0.9096343140804793,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3140511151385139,
-      "eval_iou_speedway bricks": 0.9756017665119132,
-      "eval_iou_steel": 0.8384541970044487,
-      "eval_iou_tiles": 0.250376351443821,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10671775043010712,
-      "eval_mean_accuracy": 0.7010639980778796,
-      "eval_mean_iou": 0.6601806057234325,
-      "eval_overall_accuracy": 0.97185304429796,
-      "eval_runtime": 3.3814,
-      "eval_samples_per_second": 10.646,
-      "eval_steps_per_second": 0.591,
-      "step": 2340
-    },
-    {
-      "epoch": 212.82,
-      "learning_rate": 6.045933014354067e-05,
-      "loss": 0.0379,
-      "step": 2341
-    },
-    {
-      "epoch": 212.91,
-      "learning_rate": 6.044019138755981e-05,
-      "loss": 0.0416,
-      "step": 2342
-    },
-    {
-      "epoch": 213.0,
-      "learning_rate": 6.042105263157894e-05,
-      "loss": 0.0386,
-      "step": 2343
-    },
-    {
-      "epoch": 213.09,
-      "learning_rate": 6.0401913875598094e-05,
-      "loss": 0.0423,
-      "step": 2344
-    },
-    {
-      "epoch": 213.18,
-      "learning_rate": 6.038277511961723e-05,
-      "loss": 0.0336,
-      "step": 2345
-    },
-    {
-      "epoch": 213.27,
-      "learning_rate": 6.0363636363636365e-05,
-      "loss": 0.0382,
-      "step": 2346
-    },
-    {
-      "epoch": 213.36,
-      "learning_rate": 6.0344497607655504e-05,
-      "loss": 0.0405,
-      "step": 2347
-    },
-    {
-      "epoch": 213.45,
-      "learning_rate": 6.032535885167464e-05,
-      "loss": 0.041,
-      "step": 2348
-    },
-    {
-      "epoch": 213.55,
-      "learning_rate": 6.0306220095693774e-05,
-      "loss": 0.0392,
-      "step": 2349
-    },
-    {
-      "epoch": 213.64,
-      "learning_rate": 6.028708133971293e-05,
-      "loss": 0.0388,
-      "step": 2350
-    },
-    {
-      "epoch": 213.73,
-      "learning_rate": 6.0267942583732065e-05,
-      "loss": 0.0382,
-      "step": 2351
-    },
-    {
-      "epoch": 213.82,
-      "learning_rate": 6.02488038277512e-05,
-      "loss": 0.0369,
-      "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.0391,
-      "step": 2354
-    },
-    {
-      "epoch": 214.09,
-      "learning_rate": 6.019138755980862e-05,
-      "loss": 0.0429,
-      "step": 2355
-    },
-    {
-      "epoch": 214.18,
-      "learning_rate": 6.017224880382776e-05,
-      "loss": 0.0391,
-      "step": 2356
-    },
-    {
-      "epoch": 214.27,
-      "learning_rate": 6.015311004784689e-05,
-      "loss": 0.0394,
-      "step": 2357
-    },
-    {
-      "epoch": 214.36,
-      "learning_rate": 6.013397129186603e-05,
-      "loss": 0.0361,
-      "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.0391,
-      "step": 2360
-    },
-    {
-      "epoch": 214.55,
-      "eval_accuracy_NAT": 0.9929993019593838,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9450244645692948,
-      "eval_accuracy_concrete tiles": 0.9831835137283679,
-      "eval_accuracy_dark bricks": 0.8814133405180709,
-      "eval_accuracy_grass": 0.8900554668195179,
-      "eval_accuracy_red bricks": 0.9225852272727273,
-      "eval_accuracy_road": 0.9553535908061432,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4654977531497257,
-      "eval_accuracy_speedway bricks": 0.9888206635796324,
-      "eval_accuracy_steel": 0.9011114042609326,
-      "eval_accuracy_tiles": 0.28386522394192576,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9796788112978512,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8893124230854186,
-      "eval_iou_concrete tiles": 0.9416932673441298,
-      "eval_iou_dark bricks": 0.8309746716017128,
-      "eval_iou_grass": 0.8246147469586138,
-      "eval_iou_red bricks": 0.8428223844282239,
-      "eval_iou_road": 0.947853169228259,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3119688123568103,
-      "eval_iou_speedway bricks": 0.9753177899271396,
-      "eval_iou_steel": 0.8331605427656655,
-      "eval_iou_tiles": 0.2834575668467483,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08702244609594345,
-      "eval_mean_accuracy": 0.7084546115850556,
-      "eval_mean_iou": 0.6662195527569672,
-      "eval_overall_accuracy": 0.9755267037285699,
-      "eval_runtime": 3.9243,
-      "eval_samples_per_second": 9.174,
-      "eval_steps_per_second": 0.51,
-      "step": 2360
-    },
-    {
-      "epoch": 214.64,
-      "learning_rate": 6.007655502392345e-05,
-      "loss": 0.0388,
-      "step": 2361
-    },
-    {
-      "epoch": 214.73,
-      "learning_rate": 6.005741626794259e-05,
-      "loss": 0.0416,
-      "step": 2362
-    },
-    {
-      "epoch": 214.82,
-      "learning_rate": 6.0038277511961724e-05,
-      "loss": 0.0343,
-      "step": 2363
-    },
-    {
-      "epoch": 214.91,
-      "learning_rate": 6.001913875598086e-05,
-      "loss": 0.0378,
-      "step": 2364
-    },
-    {
-      "epoch": 215.0,
-      "learning_rate": 6e-05,
-      "loss": 0.0433,
-      "step": 2365
-    },
-    {
-      "epoch": 215.09,
-      "learning_rate": 5.998086124401914e-05,
-      "loss": 0.0352,
-      "step": 2366
-    },
-    {
-      "epoch": 215.18,
-      "learning_rate": 5.9961722488038286e-05,
-      "loss": 0.0429,
-      "step": 2367
-    },
-    {
-      "epoch": 215.27,
-      "learning_rate": 5.9942583732057425e-05,
-      "loss": 0.0327,
-      "step": 2368
-    },
-    {
-      "epoch": 215.36,
-      "learning_rate": 5.992344497607656e-05,
-      "loss": 0.0423,
-      "step": 2369
-    },
-    {
-      "epoch": 215.45,
-      "learning_rate": 5.9904306220095695e-05,
-      "loss": 0.0358,
-      "step": 2370
-    },
-    {
-      "epoch": 215.55,
-      "learning_rate": 5.9885167464114834e-05,
-      "loss": 0.0395,
-      "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.0352,
-      "step": 2373
-    },
-    {
-      "epoch": 215.82,
-      "learning_rate": 5.982775119617226e-05,
-      "loss": 0.042,
-      "step": 2374
-    },
-    {
-      "epoch": 215.91,
-      "learning_rate": 5.980861244019139e-05,
-      "loss": 0.0406,
-      "step": 2375
-    },
-    {
-      "epoch": 216.0,
-      "learning_rate": 5.978947368421053e-05,
-      "loss": 0.0438,
-      "step": 2376
-    },
-    {
-      "epoch": 216.09,
-      "learning_rate": 5.977033492822967e-05,
-      "loss": 0.0355,
-      "step": 2377
-    },
-    {
-      "epoch": 216.18,
-      "learning_rate": 5.9751196172248806e-05,
-      "loss": 0.0391,
-      "step": 2378
-    },
-    {
-      "epoch": 216.27,
-      "learning_rate": 5.973205741626795e-05,
-      "loss": 0.0395,
-      "step": 2379
-    },
-    {
-      "epoch": 216.36,
-      "learning_rate": 5.971291866028709e-05,
-      "loss": 0.0346,
-      "step": 2380
-    },
-    {
-      "epoch": 216.36,
-      "eval_accuracy_NAT": 0.9926599864626704,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9459995600669556,
-      "eval_accuracy_concrete tiles": 0.9841925029046658,
-      "eval_accuracy_dark bricks": 0.867056695277318,
-      "eval_accuracy_grass": 0.8844913094992608,
-      "eval_accuracy_red bricks": 0.9169034090909091,
-      "eval_accuracy_road": 0.9382577960717641,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4853472279657045,
-      "eval_accuracy_speedway bricks": 0.9879083109770741,
-      "eval_accuracy_steel": 0.9087626146032584,
-      "eval_accuracy_tiles": 0.23729626078619367,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9795635150726215,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8821961500456541,
-      "eval_iou_concrete tiles": 0.9414465795092276,
-      "eval_iou_dark bricks": 0.8190742828055121,
-      "eval_iou_grass": 0.8151414518503248,
-      "eval_iou_red bricks": 0.8275641025641025,
-      "eval_iou_road": 0.9313351753739495,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32318763424081653,
-      "eval_iou_speedway bricks": 0.9756622995074655,
-      "eval_iou_steel": 0.8361299915038233,
-      "eval_iou_tiles": 0.2371500923961399,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09500674903392792,
-      "eval_mean_accuracy": 0.7037596672081364,
-      "eval_mean_iou": 0.6591116365284336,
-      "eval_overall_accuracy": 0.973636097378201,
-      "eval_runtime": 3.5802,
-      "eval_samples_per_second": 10.055,
-      "eval_steps_per_second": 0.559,
-      "step": 2380
-    },
-    {
-      "epoch": 216.45,
-      "learning_rate": 5.969377990430622e-05,
-      "loss": 0.0367,
-      "step": 2381
-    },
-    {
-      "epoch": 216.55,
-      "learning_rate": 5.967464114832536e-05,
-      "loss": 0.0358,
-      "step": 2382
-    },
-    {
-      "epoch": 216.64,
-      "learning_rate": 5.96555023923445e-05,
-      "loss": 0.0368,
-      "step": 2383
-    },
-    {
-      "epoch": 216.73,
-      "learning_rate": 5.963636363636363e-05,
-      "loss": 0.0401,
-      "step": 2384
-    },
-    {
-      "epoch": 216.82,
-      "learning_rate": 5.9617224880382784e-05,
-      "loss": 0.0405,
-      "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.0413,
-      "step": 2387
-    },
-    {
-      "epoch": 217.09,
-      "learning_rate": 5.9559808612440193e-05,
-      "loss": 0.039,
-      "step": 2388
-    },
-    {
-      "epoch": 217.18,
-      "learning_rate": 5.954066985645933e-05,
-      "loss": 0.0364,
-      "step": 2389
-    },
-    {
-      "epoch": 217.27,
-      "learning_rate": 5.9521531100478464e-05,
-      "loss": 0.038,
-      "step": 2390
-    },
-    {
-      "epoch": 217.36,
-      "learning_rate": 5.9502392344497617e-05,
-      "loss": 0.0409,
-      "step": 2391
-    },
-    {
-      "epoch": 217.45,
-      "learning_rate": 5.9483253588516755e-05,
-      "loss": 0.0421,
-      "step": 2392
-    },
-    {
-      "epoch": 217.55,
-      "learning_rate": 5.946411483253589e-05,
-      "loss": 0.0399,
-      "step": 2393
-    },
-    {
-      "epoch": 217.64,
-      "learning_rate": 5.9444976076555026e-05,
-      "loss": 0.0392,
-      "step": 2394
-    },
-    {
-      "epoch": 217.73,
-      "learning_rate": 5.9425837320574165e-05,
-      "loss": 0.0375,
-      "step": 2395
-    },
-    {
-      "epoch": 217.82,
-      "learning_rate": 5.94066985645933e-05,
-      "loss": 0.0367,
-      "step": 2396
-    },
-    {
-      "epoch": 217.91,
-      "learning_rate": 5.938755980861245e-05,
-      "loss": 0.0335,
-      "step": 2397
-    },
-    {
-      "epoch": 218.0,
-      "learning_rate": 5.936842105263158e-05,
-      "loss": 0.0483,
-      "step": 2398
-    },
-    {
-      "epoch": 218.09,
-      "learning_rate": 5.934928229665072e-05,
-      "loss": 0.0425,
-      "step": 2399
-    },
-    {
-      "epoch": 218.18,
-      "learning_rate": 5.933014354066986e-05,
-      "loss": 0.0383,
-      "step": 2400
-    },
-    {
-      "epoch": 218.18,
-      "eval_accuracy_NAT": 0.9920965780482107,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9487692604832826,
-      "eval_accuracy_concrete tiles": 0.9919280865896166,
-      "eval_accuracy_dark bricks": 0.8750625457064778,
-      "eval_accuracy_grass": 0.8886609324157599,
-      "eval_accuracy_red bricks": 0.9273792613636364,
-      "eval_accuracy_road": 0.9158330179313581,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.43382129958922616,
-      "eval_accuracy_speedway bricks": 0.984389236652921,
-      "eval_accuracy_steel": 0.9121100191280258,
-      "eval_accuracy_tiles": 0.35248596082728395,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9797040553729156,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.877888714451484,
-      "eval_iou_concrete tiles": 0.9046092072609654,
-      "eval_iou_dark bricks": 0.819988458486619,
-      "eval_iou_grass": 0.8133879300692589,
-      "eval_iou_red bricks": 0.839845634346358,
-      "eval_iou_road": 0.9074192330963092,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2902035998819711,
-      "eval_iou_speedway bricks": 0.9750112658783142,
-      "eval_iou_steel": 0.8367660295441309,
-      "eval_iou_tiles": 0.3521724255901471,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10760101675987244,
-      "eval_mean_accuracy": 0.7094258614412153,
-      "eval_mean_iou": 0.6613074272291133,
-      "eval_overall_accuracy": 0.9711382124159071,
-      "eval_runtime": 3.4252,
-      "eval_samples_per_second": 10.51,
-      "eval_steps_per_second": 0.584,
-      "step": 2400
-    },
-    {
-      "epoch": 218.27,
-      "learning_rate": 5.9311004784689e-05,
-      "loss": 0.0396,
-      "step": 2401
-    },
-    {
-      "epoch": 218.36,
-      "learning_rate": 5.929186602870813e-05,
-      "loss": 0.0348,
-      "step": 2402
-    },
-    {
-      "epoch": 218.45,
-      "learning_rate": 5.927272727272728e-05,
-      "loss": 0.039,
-      "step": 2403
-    },
-    {
-      "epoch": 218.55,
-      "learning_rate": 5.9253588516746414e-05,
-      "loss": 0.0363,
-      "step": 2404
-    },
-    {
-      "epoch": 218.64,
-      "learning_rate": 5.923444976076555e-05,
-      "loss": 0.0358,
-      "step": 2405
-    },
-    {
-      "epoch": 218.73,
-      "learning_rate": 5.921531100478469e-05,
-      "loss": 0.0416,
-      "step": 2406
-    },
-    {
-      "epoch": 218.82,
-      "learning_rate": 5.919617224880383e-05,
-      "loss": 0.0472,
-      "step": 2407
-    },
-    {
-      "epoch": 218.91,
-      "learning_rate": 5.917703349282296e-05,
-      "loss": 0.0358,
-      "step": 2408
-    },
-    {
-      "epoch": 219.0,
-      "learning_rate": 5.9157894736842114e-05,
-      "loss": 0.038,
-      "step": 2409
-    },
-    {
-      "epoch": 219.09,
-      "learning_rate": 5.9138755980861246e-05,
-      "loss": 0.0363,
-      "step": 2410
-    },
-    {
-      "epoch": 219.18,
-      "learning_rate": 5.9119617224880385e-05,
-      "loss": 0.0424,
-      "step": 2411
-    },
-    {
-      "epoch": 219.27,
-      "learning_rate": 5.9100478468899524e-05,
-      "loss": 0.0371,
-      "step": 2412
-    },
-    {
-      "epoch": 219.36,
-      "learning_rate": 5.908133971291866e-05,
-      "loss": 0.0358,
-      "step": 2413
-    },
-    {
-      "epoch": 219.45,
-      "learning_rate": 5.9062200956937795e-05,
-      "loss": 0.0415,
-      "step": 2414
-    },
-    {
-      "epoch": 219.55,
-      "learning_rate": 5.904306220095695e-05,
-      "loss": 0.0419,
-      "step": 2415
-    },
-    {
-      "epoch": 219.64,
-      "learning_rate": 5.902392344497608e-05,
-      "loss": 0.0403,
-      "step": 2416
-    },
-    {
-      "epoch": 219.73,
-      "learning_rate": 5.900478468899522e-05,
-      "loss": 0.039,
-      "step": 2417
-    },
-    {
-      "epoch": 219.82,
-      "learning_rate": 5.898564593301436e-05,
-      "loss": 0.0379,
-      "step": 2418
-    },
-    {
-      "epoch": 219.91,
-      "learning_rate": 5.896650717703349e-05,
-      "loss": 0.0379,
-      "step": 2419
-    },
-    {
-      "epoch": 220.0,
-      "learning_rate": 5.894736842105263e-05,
-      "loss": 0.0377,
-      "step": 2420
-    },
-    {
-      "epoch": 220.0,
-      "eval_accuracy_NAT": 0.9917378529287003,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9429052749045024,
-      "eval_accuracy_concrete tiles": 0.9887176664832141,
-      "eval_accuracy_dark bricks": 0.8882645009814865,
-      "eval_accuracy_grass": 0.9015717346400249,
-      "eval_accuracy_red bricks": 0.9211647727272727,
-      "eval_accuracy_road": 0.9110542245525021,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.39660353430926587,
-      "eval_accuracy_speedway bricks": 0.9897716342817697,
-      "eval_accuracy_steel": 0.898687421674032,
-      "eval_accuracy_tiles": 0.21312149020682097,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9798341812319151,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8756840113552424,
-      "eval_iou_concrete tiles": 0.9306414942297176,
-      "eval_iou_dark bricks": 0.8250393250393251,
-      "eval_iou_grass": 0.8202379741039404,
-      "eval_iou_red bricks": 0.8446760013025073,
-      "eval_iou_road": 0.9030175694285472,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2656154797732686,
-      "eval_iou_speedway bricks": 0.9752724907425685,
-      "eval_iou_steel": 0.8400058569215711,
-      "eval_iou_tiles": 0.21300479123887747,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11215254664421082,
-      "eval_mean_accuracy": 0.6956615467453532,
-      "eval_mean_iou": 0.6517714750282677,
-      "eval_overall_accuracy": 0.9708676868014865,
-      "eval_runtime": 3.4086,
-      "eval_samples_per_second": 10.562,
-      "eval_steps_per_second": 0.587,
-      "step": 2420
-    },
-    {
-      "epoch": 220.09,
-      "learning_rate": 5.892822966507178e-05,
-      "loss": 0.0341,
-      "step": 2421
-    },
-    {
-      "epoch": 220.18,
-      "learning_rate": 5.890909090909091e-05,
-      "loss": 0.0379,
-      "step": 2422
-    },
-    {
-      "epoch": 220.27,
-      "learning_rate": 5.888995215311005e-05,
-      "loss": 0.0352,
-      "step": 2423
-    },
-    {
-      "epoch": 220.36,
-      "learning_rate": 5.887081339712919e-05,
-      "loss": 0.0358,
-      "step": 2424
-    },
-    {
-      "epoch": 220.45,
-      "learning_rate": 5.885167464114832e-05,
-      "loss": 0.0414,
-      "step": 2425
-    },
-    {
-      "epoch": 220.55,
-      "learning_rate": 5.8832535885167474e-05,
-      "loss": 0.0381,
-      "step": 2426
-    },
-    {
-      "epoch": 220.64,
-      "learning_rate": 5.881339712918661e-05,
-      "loss": 0.0512,
-      "step": 2427
-    },
-    {
-      "epoch": 220.73,
-      "learning_rate": 5.8794258373205744e-05,
-      "loss": 0.0367,
-      "step": 2428
-    },
-    {
-      "epoch": 220.82,
-      "learning_rate": 5.877511961722488e-05,
-      "loss": 0.0365,
-      "step": 2429
-    },
-    {
-      "epoch": 220.91,
-      "learning_rate": 5.875598086124402e-05,
-      "loss": 0.0437,
-      "step": 2430
-    },
-    {
-      "epoch": 221.0,
-      "learning_rate": 5.8736842105263154e-05,
-      "loss": 0.0425,
-      "step": 2431
-    },
-    {
-      "epoch": 221.09,
-      "learning_rate": 5.8717703349282306e-05,
-      "loss": 0.038,
-      "step": 2432
-    },
-    {
-      "epoch": 221.18,
-      "learning_rate": 5.869856459330144e-05,
-      "loss": 0.0338,
-      "step": 2433
-    },
-    {
-      "epoch": 221.27,
-      "learning_rate": 5.867942583732058e-05,
-      "loss": 0.0373,
-      "step": 2434
-    },
-    {
-      "epoch": 221.36,
-      "learning_rate": 5.8660287081339716e-05,
-      "loss": 0.0416,
-      "step": 2435
-    },
-    {
-      "epoch": 221.45,
-      "learning_rate": 5.8641148325358855e-05,
-      "loss": 0.0374,
-      "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.0401,
-      "step": 2438
-    },
-    {
-      "epoch": 221.73,
-      "learning_rate": 5.858373205741627e-05,
-      "loss": 0.0376,
-      "step": 2439
-    },
-    {
-      "epoch": 221.82,
-      "learning_rate": 5.856459330143541e-05,
-      "loss": 0.0379,
-      "step": 2440
-    },
-    {
-      "epoch": 221.82,
-      "eval_accuracy_NAT": 0.9941405877798426,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9487960856689128,
-      "eval_accuracy_concrete tiles": 0.9752339020363237,
-      "eval_accuracy_dark bricks": 0.8636696047111351,
-      "eval_accuracy_grass": 0.8726220392356973,
-      "eval_accuracy_red bricks": 0.9094460227272727,
-      "eval_accuracy_road": 0.9364621209281403,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.43186392082265046,
-      "eval_accuracy_speedway bricks": 0.9868161428483504,
-      "eval_accuracy_steel": 0.9008558142602731,
-      "eval_accuracy_tiles": 0.28475551294343243,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9790513285199397,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8878011925335502,
-      "eval_iou_concrete tiles": 0.9456550742684336,
-      "eval_iou_dark bricks": 0.8238123210221014,
-      "eval_iou_grass": 0.8249093059676343,
-      "eval_iou_red bricks": 0.8525299600532623,
-      "eval_iou_road": 0.9274965538137994,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.29184365451971084,
-      "eval_iou_speedway bricks": 0.9758508386061976,
-      "eval_iou_steel": 0.8389229280877136,
-      "eval_iou_tiles": 0.28448275862068967,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09870745241641998,
-      "eval_mean_accuracy": 0.7003585964586179,
-      "eval_mean_iou": 0.6640273781548487,
-      "eval_overall_accuracy": 0.9737585915459527,
-      "eval_runtime": 3.299,
-      "eval_samples_per_second": 10.912,
-      "eval_steps_per_second": 0.606,
-      "step": 2440
-    },
-    {
-      "epoch": 221.91,
-      "learning_rate": 5.854545454545455e-05,
-      "loss": 0.0344,
-      "step": 2441
-    },
-    {
-      "epoch": 222.0,
-      "learning_rate": 5.852631578947369e-05,
-      "loss": 0.0346,
-      "step": 2442
-    },
-    {
-      "epoch": 222.09,
-      "learning_rate": 5.850717703349282e-05,
-      "loss": 0.0379,
-      "step": 2443
-    },
-    {
-      "epoch": 222.18,
-      "learning_rate": 5.848803827751197e-05,
-      "loss": 0.0363,
-      "step": 2444
-    },
-    {
-      "epoch": 222.27,
-      "learning_rate": 5.8468899521531104e-05,
-      "loss": 0.0358,
-      "step": 2445
-    },
-    {
-      "epoch": 222.36,
-      "learning_rate": 5.844976076555024e-05,
-      "loss": 0.0478,
-      "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.0396,
-      "step": 2449
-    },
-    {
-      "epoch": 222.73,
-      "learning_rate": 5.8373205741626804e-05,
-      "loss": 0.0341,
-      "step": 2450
-    },
-    {
-      "epoch": 222.82,
-      "learning_rate": 5.8354066985645936e-05,
-      "loss": 0.0372,
-      "step": 2451
-    },
-    {
-      "epoch": 222.91,
-      "learning_rate": 5.8334928229665075e-05,
-      "loss": 0.0394,
-      "step": 2452
-    },
-    {
-      "epoch": 223.0,
-      "learning_rate": 5.8315789473684214e-05,
-      "loss": 0.0482,
-      "step": 2453
-    },
-    {
-      "epoch": 223.09,
-      "learning_rate": 5.829665071770335e-05,
-      "loss": 0.0389,
-      "step": 2454
-    },
-    {
-      "epoch": 223.18,
-      "learning_rate": 5.8277511961722485e-05,
-      "loss": 0.034,
-      "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.04,
-      "step": 2457
-    },
-    {
-      "epoch": 223.45,
-      "learning_rate": 5.822009569377991e-05,
-      "loss": 0.0392,
-      "step": 2458
-    },
-    {
-      "epoch": 223.55,
-      "learning_rate": 5.8200956937799047e-05,
-      "loss": 0.0354,
-      "step": 2459
-    },
-    {
-      "epoch": 223.64,
-      "learning_rate": 5.818181818181818e-05,
-      "loss": 0.0343,
-      "step": 2460
-    },
-    {
-      "epoch": 223.64,
-      "eval_accuracy_NAT": 0.992632460088522,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9500984484312631,
-      "eval_accuracy_concrete tiles": 0.9846205589188528,
-      "eval_accuracy_dark bricks": 0.8693275855432816,
-      "eval_accuracy_grass": 0.8866023340102126,
-      "eval_accuracy_red bricks": 0.9149502840909091,
-      "eval_accuracy_road": 0.9285708239670031,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.41463347393378,
-      "eval_accuracy_speedway bricks": 0.9874955800378216,
-      "eval_accuracy_steel": 0.899058439416925,
-      "eval_accuracy_tiles": 0.1344336392275031,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.979568418182255,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8692306465209938,
-      "eval_iou_concrete tiles": 0.9488214496169711,
-      "eval_iou_dark bricks": 0.8168535262206148,
-      "eval_iou_grass": 0.8190259715101187,
-      "eval_iou_red bricks": 0.8453083989501312,
-      "eval_iou_road": 0.9197424918405844,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2795019513101654,
-      "eval_iou_speedway bricks": 0.9759648241865325,
-      "eval_iou_steel": 0.8444917715392062,
-      "eval_iou_tiles": 0.1344336392275031,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10607396066188812,
-      "eval_mean_accuracy": 0.6894172021281595,
-      "eval_mean_iou": 0.6486879299311598,
-      "eval_overall_accuracy": 0.9725369347466363,
-      "eval_runtime": 3.4958,
-      "eval_samples_per_second": 10.298,
-      "eval_steps_per_second": 0.572,
-      "step": 2460
-    },
-    {
-      "epoch": 223.73,
-      "learning_rate": 5.816267942583732e-05,
-      "loss": 0.0395,
-      "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.0389,
-      "step": 2463
-    },
-    {
-      "epoch": 224.0,
-      "learning_rate": 5.810526315789474e-05,
-      "loss": 0.0378,
-      "step": 2464
-    },
-    {
-      "epoch": 224.09,
-      "learning_rate": 5.808612440191388e-05,
-      "loss": 0.0388,
-      "step": 2465
-    },
-    {
-      "epoch": 224.18,
-      "learning_rate": 5.806698564593301e-05,
-      "loss": 0.0474,
-      "step": 2466
-    },
-    {
-      "epoch": 224.27,
-      "learning_rate": 5.804784688995215e-05,
-      "loss": 0.0376,
-      "step": 2467
-    },
-    {
-      "epoch": 224.36,
-      "learning_rate": 5.80287081339713e-05,
-      "loss": 0.0396,
-      "step": 2468
-    },
-    {
-      "epoch": 224.45,
-      "learning_rate": 5.8009569377990434e-05,
-      "loss": 0.0404,
-      "step": 2469
-    },
-    {
-      "epoch": 224.55,
-      "learning_rate": 5.799043062200957e-05,
-      "loss": 0.0359,
-      "step": 2470
-    },
-    {
-      "epoch": 224.64,
-      "learning_rate": 5.797129186602871e-05,
-      "loss": 0.035,
-      "step": 2471
-    },
-    {
-      "epoch": 224.73,
-      "learning_rate": 5.7952153110047844e-05,
-      "loss": 0.0389,
-      "step": 2472
-    },
-    {
-      "epoch": 224.82,
-      "learning_rate": 5.793301435406698e-05,
-      "loss": 0.0379,
-      "step": 2473
-    },
-    {
-      "epoch": 224.91,
-      "learning_rate": 5.791387559808613e-05,
-      "loss": 0.037,
-      "step": 2474
-    },
-    {
-      "epoch": 225.0,
-      "learning_rate": 5.789473684210527e-05,
-      "loss": 0.042,
-      "step": 2475
-    },
-    {
-      "epoch": 225.09,
-      "learning_rate": 5.7875598086124406e-05,
-      "loss": 0.037,
-      "step": 2476
-    },
-    {
-      "epoch": 225.18,
-      "learning_rate": 5.7856459330143545e-05,
-      "loss": 0.0385,
-      "step": 2477
-    },
-    {
-      "epoch": 225.27,
-      "learning_rate": 5.7837320574162677e-05,
-      "loss": 0.0389,
-      "step": 2478
-    },
-    {
-      "epoch": 225.36,
-      "learning_rate": 5.7818181818181815e-05,
-      "loss": 0.0368,
-      "step": 2479
-    },
-    {
-      "epoch": 225.45,
-      "learning_rate": 5.779904306220096e-05,
-      "loss": 0.0383,
-      "step": 2480
-    },
-    {
-      "epoch": 225.45,
-      "eval_accuracy_NAT": 0.9928187924673725,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9464944847418344,
-      "eval_accuracy_concrete tiles": 0.9900018345257751,
-      "eval_accuracy_dark bricks": 0.8780647396174127,
-      "eval_accuracy_grass": 0.8943579025363749,
-      "eval_accuracy_red bricks": 0.8929332386363636,
-      "eval_accuracy_road": 0.9289045656098585,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47076337771896454,
-      "eval_accuracy_speedway bricks": 0.9886577434720326,
-      "eval_accuracy_steel": 0.905398720401029,
-      "eval_accuracy_tiles": 0.24667853718668675,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9794551977016392,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8830055995245096,
-      "eval_iou_concrete tiles": 0.9397474967348716,
-      "eval_iou_dark bricks": 0.8206410302528868,
-      "eval_iou_grass": 0.8283943023632243,
-      "eval_iou_red bricks": 0.8506427604871448,
-      "eval_iou_road": 0.9212548000009594,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31426100079135766,
-      "eval_iou_speedway bricks": 0.9760918247408985,
-      "eval_iou_steel": 0.8437106241740681,
-      "eval_iou_tiles": 0.2463916820575963,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10051467269659042,
-      "eval_mean_accuracy": 0.7026979951472081,
-      "eval_mean_iou": 0.6618151014483965,
-      "eval_overall_accuracy": 0.9733330408732096,
-      "eval_runtime": 3.7118,
-      "eval_samples_per_second": 9.699,
-      "eval_steps_per_second": 0.539,
-      "step": 2480
-    },
-    {
-      "epoch": 225.55,
-      "learning_rate": 5.77799043062201e-05,
-      "loss": 0.042,
-      "step": 2481
-    },
-    {
-      "epoch": 225.64,
-      "learning_rate": 5.776076555023924e-05,
-      "loss": 0.0359,
-      "step": 2482
-    },
-    {
-      "epoch": 225.73,
-      "learning_rate": 5.774162679425838e-05,
-      "loss": 0.0367,
-      "step": 2483
-    },
-    {
-      "epoch": 225.82,
-      "learning_rate": 5.772248803827751e-05,
-      "loss": 0.0339,
-      "step": 2484
-    },
-    {
-      "epoch": 225.91,
-      "learning_rate": 5.770334928229665e-05,
-      "loss": 0.0368,
-      "step": 2485
-    },
-    {
-      "epoch": 226.0,
-      "learning_rate": 5.7684210526315794e-05,
-      "loss": 0.0442,
-      "step": 2486
-    },
-    {
-      "epoch": 226.09,
-      "learning_rate": 5.766507177033493e-05,
-      "loss": 0.0377,
-      "step": 2487
-    },
-    {
-      "epoch": 226.18,
-      "learning_rate": 5.764593301435407e-05,
-      "loss": 0.0374,
-      "step": 2488
-    },
-    {
-      "epoch": 226.27,
-      "learning_rate": 5.762679425837321e-05,
-      "loss": 0.04,
-      "step": 2489
-    },
-    {
-      "epoch": 226.36,
-      "learning_rate": 5.760765550239234e-05,
-      "loss": 0.0367,
-      "step": 2490
-    },
-    {
-      "epoch": 226.45,
-      "learning_rate": 5.758851674641148e-05,
-      "loss": 0.0351,
-      "step": 2491
-    },
-    {
-      "epoch": 226.55,
-      "learning_rate": 5.7569377990430626e-05,
-      "loss": 0.0401,
-      "step": 2492
-    },
-    {
-      "epoch": 226.64,
-      "learning_rate": 5.7550239234449765e-05,
-      "loss": 0.0351,
-      "step": 2493
-    },
-    {
-      "epoch": 226.73,
-      "learning_rate": 5.7531100478468904e-05,
-      "loss": 0.0348,
-      "step": 2494
-    },
-    {
-      "epoch": 226.82,
-      "learning_rate": 5.7511961722488036e-05,
-      "loss": 0.0347,
-      "step": 2495
-    },
-    {
-      "epoch": 226.91,
-      "learning_rate": 5.7492822966507175e-05,
-      "loss": 0.048,
-      "step": 2496
-    },
-    {
-      "epoch": 227.0,
-      "learning_rate": 5.747368421052633e-05,
-      "loss": 0.037,
-      "step": 2497
-    },
-    {
-      "epoch": 227.09,
-      "learning_rate": 5.745454545454546e-05,
-      "loss": 0.0404,
-      "step": 2498
-    },
-    {
-      "epoch": 227.18,
-      "learning_rate": 5.74354066985646e-05,
-      "loss": 0.0342,
-      "step": 2499
-    },
-    {
-      "epoch": 227.27,
-      "learning_rate": 5.7416267942583736e-05,
-      "loss": 0.0363,
-      "step": 2500
-    },
-    {
-      "epoch": 227.27,
-      "eval_accuracy_NAT": 0.992913723167961,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9474105648311086,
-      "eval_accuracy_concrete tiles": 0.9831223628691983,
-      "eval_accuracy_dark bricks": 0.882568030483815,
-      "eval_accuracy_grass": 0.8824886322728115,
-      "eval_accuracy_red bricks": 0.9078480113636364,
-      "eval_accuracy_road": 0.918691587654945,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.39933283709646294,
-      "eval_accuracy_speedway bricks": 0.9880899367266575,
-      "eval_accuracy_steel": 0.905382230723567,
-      "eval_accuracy_tiles": 0.22880427338720724,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9791764148436669,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.877117018225076,
-      "eval_iou_concrete tiles": 0.9481599433828733,
-      "eval_iou_dark bricks": 0.8188700807085209,
-      "eval_iou_grass": 0.8129108903240762,
-      "eval_iou_red bricks": 0.8479270315091211,
-      "eval_iou_road": 0.9124149128832824,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.27243830875714714,
-      "eval_iou_speedway bricks": 0.9762174878041074,
-      "eval_iou_steel": 0.8418647797054561,
-      "eval_iou_tiles": 0.2286633358428581,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10736905783414841,
-      "eval_mean_accuracy": 0.6951270915828746,
-      "eval_mean_iou": 0.6550584772297066,
-      "eval_overall_accuracy": 0.9718029234144423,
-      "eval_runtime": 3.9755,
-      "eval_samples_per_second": 9.055,
-      "eval_steps_per_second": 0.503,
-      "step": 2500
-    },
-    {
-      "epoch": 227.36,
-      "learning_rate": 5.739712918660287e-05,
-      "loss": 0.0394,
-      "step": 2501
-    },
-    {
-      "epoch": 227.45,
-      "learning_rate": 5.737799043062201e-05,
-      "loss": 0.0432,
-      "step": 2502
-    },
-    {
-      "epoch": 227.55,
-      "learning_rate": 5.735885167464116e-05,
-      "loss": 0.0346,
-      "step": 2503
-    },
-    {
-      "epoch": 227.64,
-      "learning_rate": 5.733971291866029e-05,
-      "loss": 0.0457,
-      "step": 2504
-    },
-    {
-      "epoch": 227.73,
-      "learning_rate": 5.732057416267943e-05,
-      "loss": 0.0345,
-      "step": 2505
-    },
-    {
-      "epoch": 227.82,
-      "learning_rate": 5.730143540669857e-05,
-      "loss": 0.0372,
-      "step": 2506
-    },
-    {
-      "epoch": 227.91,
-      "learning_rate": 5.72822966507177e-05,
-      "loss": 0.0425,
-      "step": 2507
-    },
-    {
-      "epoch": 228.0,
-      "learning_rate": 5.726315789473684e-05,
-      "loss": 0.0441,
-      "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.0414,
-      "step": 2510
-    },
-    {
-      "epoch": 228.27,
-      "learning_rate": 5.720574162679426e-05,
-      "loss": 0.0377,
-      "step": 2511
-    },
-    {
-      "epoch": 228.36,
-      "learning_rate": 5.71866028708134e-05,
-      "loss": 0.0472,
-      "step": 2512
-    },
-    {
-      "epoch": 228.45,
-      "learning_rate": 5.7167464114832534e-05,
-      "loss": 0.0343,
-      "step": 2513
-    },
-    {
-      "epoch": 228.55,
-      "learning_rate": 5.714832535885167e-05,
-      "loss": 0.0421,
-      "step": 2514
-    },
-    {
-      "epoch": 228.64,
-      "learning_rate": 5.712918660287082e-05,
-      "loss": 0.043,
-      "step": 2515
-    },
-    {
-      "epoch": 228.73,
-      "learning_rate": 5.711004784688996e-05,
-      "loss": 0.0394,
-      "step": 2516
-    },
-    {
-      "epoch": 228.82,
-      "learning_rate": 5.7090909090909096e-05,
-      "loss": 0.0365,
-      "step": 2517
-    },
-    {
-      "epoch": 228.91,
-      "learning_rate": 5.7071770334928234e-05,
-      "loss": 0.0365,
-      "step": 2518
-    },
-    {
-      "epoch": 229.0,
-      "learning_rate": 5.7052631578947366e-05,
-      "loss": 0.0394,
-      "step": 2519
-    },
-    {
-      "epoch": 229.09,
-      "learning_rate": 5.7033492822966505e-05,
-      "loss": 0.0385,
-      "step": 2520
-    },
-    {
-      "epoch": 229.09,
-      "eval_accuracy_NAT": 0.993321678148929,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.946031750289712,
-      "eval_accuracy_concrete tiles": 0.9879227053140096,
-      "eval_accuracy_dark bricks": 0.8674415919325661,
-      "eval_accuracy_grass": 0.8836874425497262,
-      "eval_accuracy_red bricks": 0.9275568181818182,
-      "eval_accuracy_road": 0.9127301880198843,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5260110826234389,
-      "eval_accuracy_speedway bricks": 0.9870430241833781,
-      "eval_accuracy_steel": 0.9115741046105138,
-      "eval_accuracy_tiles": 0.4080947815367758,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9793118636997256,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8820838424299573,
-      "eval_iou_concrete tiles": 0.9277571998736612,
-      "eval_iou_dark bricks": 0.8191995928901167,
-      "eval_iou_grass": 0.8219567302222005,
-      "eval_iou_red bricks": 0.8637566137566137,
-      "eval_iou_road": 0.9056480621605991,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3288634561687752,
-      "eval_iou_speedway bricks": 0.9759086532208701,
-      "eval_iou_steel": 0.8386149878640776,
-      "eval_iou_tiles": 0.4074808533916849,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10428911447525024,
-      "eval_mean_accuracy": 0.719339628260827,
-      "eval_mean_iou": 0.6731216812060216,
-      "eval_overall_accuracy": 0.9720997280544705,
-      "eval_runtime": 3.7206,
-      "eval_samples_per_second": 9.676,
-      "eval_steps_per_second": 0.538,
-      "step": 2520
-    },
-    {
-      "epoch": 229.18,
-      "learning_rate": 5.701435406698565e-05,
-      "loss": 0.0364,
-      "step": 2521
-    },
-    {
-      "epoch": 229.27,
-      "learning_rate": 5.699521531100479e-05,
-      "loss": 0.0344,
-      "step": 2522
-    },
-    {
-      "epoch": 229.36,
-      "learning_rate": 5.697607655502393e-05,
-      "loss": 0.0382,
-      "step": 2523
-    },
-    {
-      "epoch": 229.45,
-      "learning_rate": 5.695693779904307e-05,
-      "loss": 0.0395,
-      "step": 2524
-    },
-    {
-      "epoch": 229.55,
-      "learning_rate": 5.69377990430622e-05,
-      "loss": 0.0407,
-      "step": 2525
-    },
-    {
-      "epoch": 229.64,
-      "learning_rate": 5.691866028708134e-05,
-      "loss": 0.0406,
-      "step": 2526
-    },
-    {
-      "epoch": 229.73,
-      "learning_rate": 5.6899521531100483e-05,
-      "loss": 0.0375,
-      "step": 2527
-    },
-    {
-      "epoch": 229.82,
-      "learning_rate": 5.688038277511962e-05,
-      "loss": 0.0381,
-      "step": 2528
-    },
-    {
-      "epoch": 229.91,
-      "learning_rate": 5.686124401913876e-05,
-      "loss": 0.039,
-      "step": 2529
-    },
-    {
-      "epoch": 230.0,
-      "learning_rate": 5.68421052631579e-05,
-      "loss": 0.043,
-      "step": 2530
-    },
-    {
-      "epoch": 230.09,
-      "learning_rate": 5.682296650717703e-05,
-      "loss": 0.0335,
-      "step": 2531
-    },
-    {
-      "epoch": 230.18,
-      "learning_rate": 5.680382775119617e-05,
-      "loss": 0.0372,
-      "step": 2532
-    },
-    {
-      "epoch": 230.27,
-      "learning_rate": 5.6784688995215316e-05,
-      "loss": 0.0333,
-      "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.0376,
-      "step": 2535
-    },
-    {
-      "epoch": 230.55,
-      "learning_rate": 5.6727272727272726e-05,
-      "loss": 0.0414,
-      "step": 2536
-    },
-    {
-      "epoch": 230.64,
-      "learning_rate": 5.6708133971291864e-05,
-      "loss": 0.0374,
-      "step": 2537
-    },
-    {
-      "epoch": 230.73,
-      "learning_rate": 5.6688995215311e-05,
-      "loss": 0.0425,
-      "step": 2538
-    },
-    {
-      "epoch": 230.82,
-      "learning_rate": 5.666985645933015e-05,
-      "loss": 0.0357,
-      "step": 2539
-    },
-    {
-      "epoch": 230.91,
-      "learning_rate": 5.665071770334929e-05,
-      "loss": 0.0373,
-      "step": 2540
-    },
-    {
-      "epoch": 230.91,
-      "eval_accuracy_NAT": 0.9927026876328235,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.949701435683935,
-      "eval_accuracy_concrete tiles": 0.9797284901852871,
-      "eval_accuracy_dark bricks": 0.8734844694199607,
-      "eval_accuracy_grass": 0.8679945616653327,
-      "eval_accuracy_red bricks": 0.9151278409090909,
-      "eval_accuracy_road": 0.9632980929567212,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4731067184958509,
-      "eval_accuracy_speedway bricks": 0.9869959583745159,
-      "eval_accuracy_steel": 0.9074516852450366,
-      "eval_accuracy_tiles": 0.07471579235721135,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9793662282969383,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8766308371527546,
-      "eval_iou_concrete tiles": 0.9507744347516468,
-      "eval_iou_dark bricks": 0.8183326121448147,
-      "eval_iou_grass": 0.8046318135869988,
-      "eval_iou_red bricks": 0.8548681373362084,
-      "eval_iou_road": 0.9544711161166258,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3179376018971395,
-      "eval_iou_speedway bricks": 0.9757328869757764,
-      "eval_iou_steel": 0.8372547677187218,
-      "eval_iou_tiles": 0.07471579235721135,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09160269051790237,
-      "eval_mean_accuracy": 0.6911005948404436,
-      "eval_mean_iou": 0.6495935560257566,
-      "eval_overall_accuracy": 0.9751738442314996,
-      "eval_runtime": 4.129,
-      "eval_samples_per_second": 8.719,
-      "eval_steps_per_second": 0.484,
-      "step": 2540
-    },
-    {
-      "epoch": 231.0,
-      "learning_rate": 5.6631578947368426e-05,
-      "loss": 0.0391,
-      "step": 2541
-    },
-    {
-      "epoch": 231.09,
-      "learning_rate": 5.661244019138756e-05,
-      "loss": 0.0372,
-      "step": 2542
-    },
-    {
-      "epoch": 231.18,
-      "learning_rate": 5.65933014354067e-05,
-      "loss": 0.0355,
-      "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.0502,
-      "step": 2545
-    },
-    {
-      "epoch": 231.45,
-      "learning_rate": 5.653588516746412e-05,
-      "loss": 0.0348,
-      "step": 2546
-    },
-    {
-      "epoch": 231.55,
-      "learning_rate": 5.651674641148326e-05,
-      "loss": 0.0385,
-      "step": 2547
-    },
-    {
-      "epoch": 231.64,
-      "learning_rate": 5.649760765550239e-05,
-      "loss": 0.0332,
-      "step": 2548
-    },
-    {
-      "epoch": 231.73,
-      "learning_rate": 5.647846889952153e-05,
-      "loss": 0.0423,
-      "step": 2549
-    },
-    {
-      "epoch": 231.82,
-      "learning_rate": 5.645933014354067e-05,
-      "loss": 0.0391,
-      "step": 2550
-    },
-    {
-      "epoch": 231.91,
-      "learning_rate": 5.6440191387559814e-05,
-      "loss": 0.0358,
-      "step": 2551
-    },
-    {
-      "epoch": 232.0,
-      "learning_rate": 5.642105263157895e-05,
-      "loss": 0.0398,
-      "step": 2552
-    },
-    {
-      "epoch": 232.09,
-      "learning_rate": 5.640191387559809e-05,
-      "loss": 0.0366,
-      "step": 2553
-    },
-    {
-      "epoch": 232.18,
-      "learning_rate": 5.6382775119617224e-05,
-      "loss": 0.0371,
-      "step": 2554
-    },
-    {
-      "epoch": 232.27,
-      "learning_rate": 5.636363636363636e-05,
-      "loss": 0.0411,
-      "step": 2555
-    },
-    {
-      "epoch": 232.36,
-      "learning_rate": 5.63444976076555e-05,
-      "loss": 0.0335,
-      "step": 2556
-    },
-    {
-      "epoch": 232.45,
-      "learning_rate": 5.632535885167465e-05,
-      "loss": 0.0435,
-      "step": 2557
-    },
-    {
-      "epoch": 232.55,
-      "learning_rate": 5.6306220095693786e-05,
-      "loss": 0.0426,
-      "step": 2558
-    },
-    {
-      "epoch": 232.64,
-      "learning_rate": 5.6287081339712924e-05,
-      "loss": 0.0373,
-      "step": 2559
-    },
-    {
-      "epoch": 232.73,
-      "learning_rate": 5.6267942583732056e-05,
-      "loss": 0.0366,
-      "step": 2560
-    },
-    {
-      "epoch": 232.73,
-      "eval_accuracy_NAT": 0.9922774404424547,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9458775054723378,
-      "eval_accuracy_concrete tiles": 0.9879532807435945,
-      "eval_accuracy_dark bricks": 0.8738693660752088,
-      "eval_accuracy_grass": 0.9042140103524082,
-      "eval_accuracy_red bricks": 0.939453125,
-      "eval_accuracy_road": 0.9676947763386848,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6271882667548866,
-      "eval_accuracy_speedway bricks": 0.9886565366564208,
-      "eval_accuracy_steel": 0.9045577468504716,
-      "eval_accuracy_tiles": 0.18593343377619503,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9797952466516631,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8933094514859863,
-      "eval_iou_concrete tiles": 0.9317454367196286,
-      "eval_iou_dark bricks": 0.8193431974016601,
-      "eval_iou_grass": 0.8330521190888659,
-      "eval_iou_red bricks": 0.8634138381201044,
-      "eval_iou_road": 0.9571096942711267,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.41775313085313454,
-      "eval_iou_speedway bricks": 0.9759179598853049,
-      "eval_iou_steel": 0.840717865407899,
-      "eval_iou_tiles": 0.1858952413557001,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08471038937568665,
-      "eval_mean_accuracy": 0.7167442683432819,
-      "eval_mean_iou": 0.669081013941621,
-      "eval_overall_accuracy": 0.9771625730726454,
-      "eval_runtime": 3.8341,
-      "eval_samples_per_second": 9.39,
-      "eval_steps_per_second": 0.522,
-      "step": 2560
-    },
-    {
-      "epoch": 232.82,
-      "learning_rate": 5.6248803827751195e-05,
-      "loss": 0.0344,
-      "step": 2561
-    },
-    {
-      "epoch": 232.91,
-      "learning_rate": 5.6229665071770334e-05,
-      "loss": 0.0373,
-      "step": 2562
-    },
-    {
-      "epoch": 233.0,
-      "learning_rate": 5.621052631578948e-05,
-      "loss": 0.0489,
-      "step": 2563
-    },
-    {
-      "epoch": 233.09,
-      "learning_rate": 5.619138755980862e-05,
-      "loss": 0.0373,
-      "step": 2564
-    },
-    {
-      "epoch": 233.18,
-      "learning_rate": 5.617224880382776e-05,
-      "loss": 0.0385,
-      "step": 2565
-    },
-    {
-      "epoch": 233.27,
-      "learning_rate": 5.615311004784689e-05,
-      "loss": 0.0408,
-      "step": 2566
-    },
-    {
-      "epoch": 233.36,
-      "learning_rate": 5.613397129186603e-05,
-      "loss": 0.0404,
-      "step": 2567
-    },
-    {
-      "epoch": 233.45,
-      "learning_rate": 5.611483253588517e-05,
-      "loss": 0.0338,
-      "step": 2568
-    },
-    {
-      "epoch": 233.55,
-      "learning_rate": 5.609569377990431e-05,
-      "loss": 0.058,
-      "step": 2569
-    },
-    {
-      "epoch": 233.64,
-      "learning_rate": 5.607655502392345e-05,
-      "loss": 0.0377,
-      "step": 2570
-    },
-    {
-      "epoch": 233.73,
-      "learning_rate": 5.605741626794259e-05,
-      "loss": 0.0396,
-      "step": 2571
-    },
-    {
-      "epoch": 233.82,
-      "learning_rate": 5.603827751196172e-05,
-      "loss": 0.035,
-      "step": 2572
-    },
-    {
-      "epoch": 233.91,
-      "learning_rate": 5.601913875598086e-05,
-      "loss": 0.036,
-      "step": 2573
-    },
-    {
-      "epoch": 234.0,
-      "learning_rate": 5.6000000000000006e-05,
-      "loss": 0.0395,
-      "step": 2574
-    },
-    {
-      "epoch": 234.09,
-      "learning_rate": 5.5980861244019145e-05,
-      "loss": 0.0399,
-      "step": 2575
-    },
-    {
-      "epoch": 234.18,
-      "learning_rate": 5.5961722488038283e-05,
-      "loss": 0.0402,
-      "step": 2576
-    },
-    {
-      "epoch": 234.27,
-      "learning_rate": 5.5942583732057415e-05,
-      "loss": 0.0444,
-      "step": 2577
-    },
-    {
-      "epoch": 234.36,
-      "learning_rate": 5.5923444976076554e-05,
-      "loss": 0.0451,
-      "step": 2578
-    },
-    {
-      "epoch": 234.45,
-      "learning_rate": 5.590430622009569e-05,
-      "loss": 0.0371,
-      "step": 2579
-    },
-    {
-      "epoch": 234.55,
-      "learning_rate": 5.588516746411484e-05,
-      "loss": 0.0414,
-      "step": 2580
-    },
-    {
-      "epoch": 234.55,
-      "eval_accuracy_NAT": 0.992455303167721,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9569496008412378,
-      "eval_accuracy_concrete tiles": 0.9794838867486089,
-      "eval_accuracy_dark bricks": 0.8493899388014318,
-      "eval_accuracy_grass": 0.8855852675653666,
-      "eval_accuracy_red bricks": 0.9268465909090909,
-      "eval_accuracy_road": 0.9563257947222871,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4653323408595925,
-      "eval_accuracy_speedway bricks": 0.9870454378146017,
-      "eval_accuracy_steel": 0.8921245300441923,
-      "eval_accuracy_tiles": 0.07088070127379811,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.979768926087773,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8794039777545507,
-      "eval_iou_concrete tiles": 0.9493539592223803,
-      "eval_iou_dark bricks": 0.8146780862374483,
-      "eval_iou_grass": 0.827517848161623,
-      "eval_iou_red bricks": 0.8441138421733506,
-      "eval_iou_road": 0.9475831429266358,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3098826855642659,
-      "eval_iou_speedway bricks": 0.9756421236961895,
-      "eval_iou_steel": 0.8384332261438922,
-      "eval_iou_tiles": 0.07088070127379811,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0939382016658783,
-      "eval_mean_accuracy": 0.6894168763652253,
-      "eval_mean_iou": 0.6490198860955314,
-      "eval_overall_accuracy": 0.9752359390258789,
-      "eval_runtime": 4.5839,
-      "eval_samples_per_second": 7.853,
-      "eval_steps_per_second": 0.436,
-      "step": 2580
-    },
-    {
-      "epoch": 234.64,
-      "learning_rate": 5.586602870813398e-05,
-      "loss": 0.0359,
-      "step": 2581
-    },
-    {
-      "epoch": 234.73,
-      "learning_rate": 5.5846889952153116e-05,
-      "loss": 0.0363,
-      "step": 2582
-    },
-    {
-      "epoch": 234.82,
-      "learning_rate": 5.582775119617225e-05,
-      "loss": 0.0364,
-      "step": 2583
-    },
-    {
-      "epoch": 234.91,
-      "learning_rate": 5.580861244019139e-05,
-      "loss": 0.0376,
-      "step": 2584
-    },
-    {
-      "epoch": 235.0,
-      "learning_rate": 5.5789473684210526e-05,
-      "loss": 0.0395,
-      "step": 2585
-    },
-    {
-      "epoch": 235.09,
-      "learning_rate": 5.577033492822967e-05,
-      "loss": 0.0356,
-      "step": 2586
-    },
-    {
-      "epoch": 235.18,
-      "learning_rate": 5.575119617224881e-05,
-      "loss": 0.0353,
-      "step": 2587
-    },
-    {
-      "epoch": 235.27,
-      "learning_rate": 5.573205741626795e-05,
-      "loss": 0.041,
-      "step": 2588
-    },
-    {
-      "epoch": 235.36,
-      "learning_rate": 5.571291866028708e-05,
-      "loss": 0.0402,
-      "step": 2589
-    },
-    {
-      "epoch": 235.45,
-      "learning_rate": 5.569377990430622e-05,
-      "loss": 0.04,
-      "step": 2590
-    },
-    {
-      "epoch": 235.55,
-      "learning_rate": 5.567464114832536e-05,
-      "loss": 0.0321,
-      "step": 2591
-    },
-    {
-      "epoch": 235.64,
-      "learning_rate": 5.5655502392344504e-05,
-      "loss": 0.0347,
-      "step": 2592
-    },
-    {
-      "epoch": 235.73,
-      "learning_rate": 5.563636363636364e-05,
-      "loss": 0.0367,
-      "step": 2593
-    },
-    {
-      "epoch": 235.82,
-      "learning_rate": 5.561722488038278e-05,
-      "loss": 0.0428,
-      "step": 2594
-    },
-    {
-      "epoch": 235.91,
-      "learning_rate": 5.5598086124401913e-05,
-      "loss": 0.0348,
-      "step": 2595
-    },
-    {
-      "epoch": 236.0,
-      "learning_rate": 5.557894736842105e-05,
-      "loss": 0.0427,
-      "step": 2596
-    },
-    {
-      "epoch": 236.09,
-      "learning_rate": 5.555980861244019e-05,
-      "loss": 0.037,
-      "step": 2597
-    },
-    {
-      "epoch": 236.18,
-      "learning_rate": 5.5540669856459337e-05,
-      "loss": 0.034,
-      "step": 2598
-    },
-    {
-      "epoch": 236.27,
-      "learning_rate": 5.5521531100478475e-05,
-      "loss": 0.0323,
-      "step": 2599
-    },
-    {
-      "epoch": 236.36,
-      "learning_rate": 5.5502392344497614e-05,
-      "loss": 0.0346,
-      "step": 2600
-    },
-    {
-      "epoch": 236.36,
-      "eval_accuracy_NAT": 0.9931625192419943,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.951689181939139,
-      "eval_accuracy_concrete tiles": 0.9789641044456675,
-      "eval_accuracy_dark bricks": 0.876833070320619,
-      "eval_accuracy_grass": 0.8859033192714868,
-      "eval_accuracy_red bricks": 0.9321732954545454,
-      "eval_accuracy_road": 0.940245735422685,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4570065889228903,
-      "eval_accuracy_speedway bricks": 0.9872053408831718,
-      "eval_accuracy_steel": 0.9076660510520415,
-      "eval_accuracy_tiles": 0.14354198055060952,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9796065486692087,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8828640396172653,
-      "eval_iou_concrete tiles": 0.9537682454572535,
-      "eval_iou_dark bricks": 0.8076364023114829,
-      "eval_iou_grass": 0.8273340905826904,
-      "eval_iou_red bricks": 0.8563040287065732,
-      "eval_iou_road": 0.9333017260004322,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.30589385887215825,
-      "eval_iou_speedway bricks": 0.976241537211209,
-      "eval_iou_steel": 0.8420001988573351,
-      "eval_iou_tiles": 0.14353215092789154,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09956098347902298,
-      "eval_mean_accuracy": 0.6964916298080654,
-      "eval_mean_iou": 0.6544986790164231,
-      "eval_overall_accuracy": 0.9742307662963867,
-      "eval_runtime": 4.2058,
-      "eval_samples_per_second": 8.56,
-      "eval_steps_per_second": 0.476,
-      "step": 2600
-    },
-    {
-      "epoch": 236.45,
-      "learning_rate": 5.5483253588516746e-05,
-      "loss": 0.0364,
-      "step": 2601
-    },
-    {
-      "epoch": 236.55,
-      "learning_rate": 5.5464114832535885e-05,
-      "loss": 0.0359,
-      "step": 2602
-    },
-    {
-      "epoch": 236.64,
-      "learning_rate": 5.5444976076555024e-05,
-      "loss": 0.0378,
-      "step": 2603
-    },
-    {
-      "epoch": 236.73,
-      "learning_rate": 5.542583732057417e-05,
-      "loss": 0.0353,
-      "step": 2604
-    },
-    {
-      "epoch": 236.82,
-      "learning_rate": 5.540669856459331e-05,
-      "loss": 0.0417,
-      "step": 2605
-    },
-    {
-      "epoch": 236.91,
-      "learning_rate": 5.538755980861245e-05,
-      "loss": 0.0428,
-      "step": 2606
-    },
-    {
-      "epoch": 237.0,
-      "learning_rate": 5.536842105263158e-05,
-      "loss": 0.0449,
-      "step": 2607
-    },
-    {
-      "epoch": 237.09,
-      "learning_rate": 5.534928229665072e-05,
-      "loss": 0.0454,
-      "step": 2608
-    },
-    {
-      "epoch": 237.18,
-      "learning_rate": 5.5330143540669856e-05,
-      "loss": 0.0348,
-      "step": 2609
-    },
-    {
-      "epoch": 237.27,
-      "learning_rate": 5.5311004784689e-05,
-      "loss": 0.0361,
-      "step": 2610
-    },
-    {
-      "epoch": 237.36,
-      "learning_rate": 5.529186602870814e-05,
-      "loss": 0.0362,
-      "step": 2611
-    },
-    {
-      "epoch": 237.45,
-      "learning_rate": 5.527272727272727e-05,
-      "loss": 0.0372,
-      "step": 2612
-    },
-    {
-      "epoch": 237.55,
-      "learning_rate": 5.525358851674641e-05,
-      "loss": 0.0438,
-      "step": 2613
-    },
-    {
-      "epoch": 237.64,
-      "learning_rate": 5.523444976076555e-05,
-      "loss": 0.0361,
-      "step": 2614
-    },
-    {
-      "epoch": 237.73,
-      "learning_rate": 5.521531100478469e-05,
-      "loss": 0.0371,
-      "step": 2615
-    },
-    {
-      "epoch": 237.82,
-      "learning_rate": 5.5196172248803835e-05,
-      "loss": 0.0409,
-      "step": 2616
-    },
-    {
-      "epoch": 237.91,
-      "learning_rate": 5.517703349282297e-05,
-      "loss": 0.0391,
-      "step": 2617
-    },
-    {
-      "epoch": 238.0,
-      "learning_rate": 5.5157894736842105e-05,
-      "loss": 0.0386,
-      "step": 2618
-    },
-    {
-      "epoch": 238.09,
-      "learning_rate": 5.5138755980861244e-05,
-      "loss": 0.0381,
-      "step": 2619
-    },
-    {
-      "epoch": 238.18,
-      "learning_rate": 5.511961722488038e-05,
-      "loss": 0.029,
-      "step": 2620
-    },
-    {
-      "epoch": 238.18,
-      "eval_accuracy_NAT": 0.9933709080103866,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9480879007682733,
-      "eval_accuracy_concrete tiles": 0.9776493609735217,
-      "eval_accuracy_dark bricks": 0.859320272506832,
-      "eval_accuracy_grass": 0.8935819961763893,
-      "eval_accuracy_red bricks": 0.9163707386363636,
-      "eval_accuracy_road": 0.9551540713457406,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.45946020455986547,
-      "eval_accuracy_speedway bricks": 0.9881092457764471,
-      "eval_accuracy_steel": 0.9020595607149924,
-      "eval_accuracy_tiles": 0.20038350910834132,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9791815413632373,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8856058233208465,
-      "eval_iou_concrete tiles": 0.9505901239706276,
-      "eval_iou_dark bricks": 0.8095583436072231,
-      "eval_iou_grass": 0.8318821623098999,
-      "eval_iou_red bricks": 0.8605969651492413,
-      "eval_iou_road": 0.948422118082696,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31006511627906974,
-      "eval_iou_speedway bricks": 0.9764100289128149,
-      "eval_iou_steel": 0.841950949233149,
-      "eval_iou_tiles": 0.20026007802340703,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08897086977958679,
-      "eval_mean_accuracy": 0.6995036745059349,
-      "eval_mean_iou": 0.661117173096324,
-      "eval_overall_accuracy": 0.9757324854532877,
-      "eval_runtime": 4.4731,
-      "eval_samples_per_second": 8.048,
-      "eval_steps_per_second": 0.447,
-      "step": 2620
-    },
-    {
-      "epoch": 238.27,
-      "learning_rate": 5.510047846889952e-05,
-      "loss": 0.0336,
-      "step": 2621
-    },
-    {
-      "epoch": 238.36,
-      "learning_rate": 5.508133971291867e-05,
-      "loss": 0.0386,
-      "step": 2622
-    },
-    {
-      "epoch": 238.45,
-      "learning_rate": 5.5062200956937806e-05,
-      "loss": 0.0419,
-      "step": 2623
-    },
-    {
-      "epoch": 238.55,
-      "learning_rate": 5.504306220095694e-05,
-      "loss": 0.0377,
-      "step": 2624
-    },
-    {
-      "epoch": 238.64,
-      "learning_rate": 5.502392344497608e-05,
-      "loss": 0.0348,
-      "step": 2625
-    },
-    {
-      "epoch": 238.73,
-      "learning_rate": 5.5004784688995216e-05,
-      "loss": 0.0443,
-      "step": 2626
-    },
-    {
-      "epoch": 238.82,
-      "learning_rate": 5.4985645933014354e-05,
-      "loss": 0.0311,
-      "step": 2627
-    },
-    {
-      "epoch": 238.91,
-      "learning_rate": 5.49665071770335e-05,
-      "loss": 0.0403,
-      "step": 2628
-    },
-    {
-      "epoch": 239.0,
-      "learning_rate": 5.494736842105264e-05,
-      "loss": 0.0435,
-      "step": 2629
-    },
-    {
-      "epoch": 239.09,
-      "learning_rate": 5.492822966507177e-05,
-      "loss": 0.0343,
-      "step": 2630
-    },
-    {
-      "epoch": 239.18,
-      "learning_rate": 5.490909090909091e-05,
-      "loss": 0.0369,
-      "step": 2631
-    },
-    {
-      "epoch": 239.27,
-      "learning_rate": 5.488995215311005e-05,
-      "loss": 0.0359,
-      "step": 2632
-    },
-    {
-      "epoch": 239.36,
-      "learning_rate": 5.487081339712919e-05,
-      "loss": 0.0352,
-      "step": 2633
-    },
-    {
-      "epoch": 239.45,
-      "learning_rate": 5.485167464114833e-05,
-      "loss": 0.0357,
-      "step": 2634
-    },
-    {
-      "epoch": 239.55,
-      "learning_rate": 5.483253588516747e-05,
-      "loss": 0.0401,
-      "step": 2635
-    },
-    {
-      "epoch": 239.64,
-      "learning_rate": 5.48133971291866e-05,
-      "loss": 0.0434,
-      "step": 2636
-    },
-    {
-      "epoch": 239.73,
-      "learning_rate": 5.479425837320574e-05,
-      "loss": 0.0358,
-      "step": 2637
-    },
-    {
-      "epoch": 239.82,
-      "learning_rate": 5.477511961722488e-05,
-      "loss": 0.0371,
-      "step": 2638
-    },
-    {
-      "epoch": 239.91,
-      "learning_rate": 5.4755980861244026e-05,
-      "loss": 0.0502,
-      "step": 2639
-    },
-    {
-      "epoch": 240.0,
-      "learning_rate": 5.4736842105263165e-05,
-      "loss": 0.0364,
-      "step": 2640
-    },
-    {
-      "epoch": 240.0,
-      "eval_accuracy_NAT": 0.9918371949071974,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9496223013863256,
-      "eval_accuracy_concrete tiles": 0.9856601235247355,
-      "eval_accuracy_dark bricks": 0.8706747238366499,
-      "eval_accuracy_grass": 0.9182886721166515,
-      "eval_accuracy_red bricks": 0.9154829545454546,
-      "eval_accuracy_road": 0.9547417311275752,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46232735092217353,
-      "eval_accuracy_speedway bricks": 0.9874473074133476,
-      "eval_accuracy_steel": 0.9125799749356902,
-      "eval_accuracy_tiles": 0.16353924119983565,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.97960751085213,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8839823704943004,
-      "eval_iou_concrete tiles": 0.9466435661008986,
-      "eval_iou_dark bricks": 0.8114865834409528,
-      "eval_iou_grass": 0.8399122807017544,
-      "eval_iou_red bricks": 0.8460781096160157,
-      "eval_iou_road": 0.9471654338641222,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3082493934269539,
-      "eval_iou_speedway bricks": 0.9760550686897328,
-      "eval_iou_steel": 0.8407328355601468,
-      "eval_iou_tiles": 0.16352804218311306,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09092303365468979,
-      "eval_mean_accuracy": 0.7009385827627413,
-      "eval_mean_iou": 0.6571877842253941,
-      "eval_overall_accuracy": 0.9756770663791232,
-      "eval_runtime": 4.1362,
-      "eval_samples_per_second": 8.704,
-      "eval_steps_per_second": 0.484,
-      "step": 2640
-    },
-    {
-      "epoch": 240.09,
-      "learning_rate": 5.4717703349282304e-05,
-      "loss": 0.0377,
-      "step": 2641
-    },
-    {
-      "epoch": 240.18,
-      "learning_rate": 5.4698564593301436e-05,
-      "loss": 0.0359,
-      "step": 2642
-    },
-    {
-      "epoch": 240.27,
-      "learning_rate": 5.4679425837320575e-05,
-      "loss": 0.0439,
-      "step": 2643
-    },
-    {
-      "epoch": 240.36,
-      "learning_rate": 5.4660287081339714e-05,
-      "loss": 0.0398,
-      "step": 2644
-    },
-    {
-      "epoch": 240.45,
-      "learning_rate": 5.464114832535886e-05,
-      "loss": 0.0395,
-      "step": 2645
-    },
-    {
-      "epoch": 240.55,
-      "learning_rate": 5.4622009569378e-05,
-      "loss": 0.0397,
-      "step": 2646
-    },
-    {
-      "epoch": 240.64,
-      "learning_rate": 5.460287081339714e-05,
-      "loss": 0.0342,
-      "step": 2647
-    },
-    {
-      "epoch": 240.73,
-      "learning_rate": 5.458373205741627e-05,
-      "loss": 0.0446,
-      "step": 2648
-    },
-    {
-      "epoch": 240.82,
-      "learning_rate": 5.456459330143541e-05,
-      "loss": 0.0417,
-      "step": 2649
-    },
-    {
-      "epoch": 240.91,
-      "learning_rate": 5.4545454545454546e-05,
-      "loss": 0.0359,
-      "step": 2650
-    },
-    {
-      "epoch": 241.0,
-      "learning_rate": 5.452631578947369e-05,
-      "loss": 0.0344,
-      "step": 2651
-    },
-    {
-      "epoch": 241.09,
-      "learning_rate": 5.450717703349283e-05,
-      "loss": 0.0413,
-      "step": 2652
-    },
-    {
-      "epoch": 241.18,
-      "learning_rate": 5.448803827751196e-05,
-      "loss": 0.0371,
-      "step": 2653
-    },
-    {
-      "epoch": 241.27,
-      "learning_rate": 5.44688995215311e-05,
-      "loss": 0.0355,
-      "step": 2654
-    },
-    {
-      "epoch": 241.36,
-      "learning_rate": 5.444976076555024e-05,
-      "loss": 0.0352,
-      "step": 2655
-    },
-    {
-      "epoch": 241.45,
-      "learning_rate": 5.443062200956938e-05,
-      "loss": 0.0371,
-      "step": 2656
-    },
-    {
-      "epoch": 241.55,
-      "learning_rate": 5.4411483253588524e-05,
-      "loss": 0.0409,
-      "step": 2657
-    },
-    {
-      "epoch": 241.64,
-      "learning_rate": 5.439234449760766e-05,
-      "loss": 0.036,
-      "step": 2658
-    },
-    {
-      "epoch": 241.73,
-      "learning_rate": 5.4373205741626795e-05,
-      "loss": 0.0339,
-      "step": 2659
-    },
-    {
-      "epoch": 241.82,
-      "learning_rate": 5.4354066985645934e-05,
-      "loss": 0.0378,
-      "step": 2660
-    },
-    {
-      "epoch": 241.82,
-      "eval_accuracy_NAT": 0.9926626332294154,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9481120434353406,
-      "eval_accuracy_concrete tiles": 0.9868219898489574,
-      "eval_accuracy_dark bricks": 0.8814518301835957,
-      "eval_accuracy_grass": 0.9002680721523013,
-      "eval_accuracy_red bricks": 0.9213423295454546,
-      "eval_accuracy_road": 0.9080541773933568,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.45411187384556007,
-      "eval_accuracy_speedway bricks": 0.98792520639564,
-      "eval_accuracy_steel": 0.8953977310203812,
-      "eval_accuracy_tiles": 0.06519654841802493,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9798626588570859,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.883286724995814,
-      "eval_iou_concrete tiles": 0.9400029124799767,
-      "eval_iou_dark bricks": 0.8246075183638196,
-      "eval_iou_grass": 0.8391577863712034,
-      "eval_iou_red bricks": 0.8540157998683344,
-      "eval_iou_road": 0.8976769984232804,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.302671713645217,
-      "eval_iou_speedway bricks": 0.9756534030471384,
-      "eval_iou_steel": 0.8417638122403422,
-      "eval_iou_tiles": 0.06516084873374402,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11888762563467026,
-      "eval_mean_accuracy": 0.6877957258052328,
-      "eval_mean_iou": 0.6464507828481505,
-      "eval_overall_accuracy": 0.9711323844061958,
-      "eval_runtime": 4.6308,
-      "eval_samples_per_second": 7.774,
-      "eval_steps_per_second": 0.432,
-      "step": 2660
-    },
-    {
-      "epoch": 241.91,
-      "learning_rate": 5.433492822966507e-05,
-      "loss": 0.042,
-      "step": 2661
-    },
-    {
-      "epoch": 242.0,
-      "learning_rate": 5.431578947368421e-05,
-      "loss": 0.0358,
-      "step": 2662
-    },
-    {
-      "epoch": 242.09,
-      "learning_rate": 5.429665071770336e-05,
-      "loss": 0.0364,
-      "step": 2663
-    },
-    {
-      "epoch": 242.18,
-      "learning_rate": 5.4277511961722496e-05,
-      "loss": 0.0507,
-      "step": 2664
-    },
-    {
-      "epoch": 242.27,
-      "learning_rate": 5.425837320574163e-05,
-      "loss": 0.0346,
-      "step": 2665
-    },
-    {
-      "epoch": 242.36,
-      "learning_rate": 5.423923444976077e-05,
-      "loss": 0.0335,
-      "step": 2666
-    },
-    {
-      "epoch": 242.45,
-      "learning_rate": 5.4220095693779905e-05,
-      "loss": 0.0431,
-      "step": 2667
-    },
-    {
-      "epoch": 242.55,
-      "learning_rate": 5.4200956937799044e-05,
-      "loss": 0.0387,
-      "step": 2668
-    },
-    {
-      "epoch": 242.64,
-      "learning_rate": 5.418181818181819e-05,
-      "loss": 0.0369,
-      "step": 2669
-    },
-    {
-      "epoch": 242.73,
-      "learning_rate": 5.416267942583733e-05,
-      "loss": 0.0469,
-      "step": 2670
-    },
-    {
-      "epoch": 242.82,
-      "learning_rate": 5.414354066985646e-05,
-      "loss": 0.0352,
-      "step": 2671
-    },
-    {
-      "epoch": 242.91,
-      "learning_rate": 5.41244019138756e-05,
-      "loss": 0.0455,
-      "step": 2672
-    },
-    {
-      "epoch": 243.0,
-      "learning_rate": 5.410526315789474e-05,
-      "loss": 0.039,
-      "step": 2673
-    },
-    {
-      "epoch": 243.09,
-      "learning_rate": 5.408612440191388e-05,
-      "loss": 0.0529,
-      "step": 2674
-    },
-    {
-      "epoch": 243.18,
-      "learning_rate": 5.406698564593302e-05,
-      "loss": 0.0331,
-      "step": 2675
-    },
-    {
-      "epoch": 243.27,
-      "learning_rate": 5.404784688995216e-05,
-      "loss": 0.0397,
-      "step": 2676
-    },
-    {
-      "epoch": 243.36,
-      "learning_rate": 5.402870813397129e-05,
-      "loss": 0.0368,
-      "step": 2677
-    },
-    {
-      "epoch": 243.45,
-      "learning_rate": 5.400956937799043e-05,
-      "loss": 0.0376,
-      "step": 2678
-    },
-    {
-      "epoch": 243.55,
-      "learning_rate": 5.399043062200957e-05,
-      "loss": 0.0383,
-      "step": 2679
-    },
-    {
-      "epoch": 243.64,
-      "learning_rate": 5.39712918660287e-05,
-      "loss": 0.0364,
-      "step": 2680
-    },
-    {
-      "epoch": 243.64,
-      "eval_accuracy_NAT": 0.99224973761719,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9446583007854414,
-      "eval_accuracy_concrete tiles": 0.9890539962086468,
-      "eval_accuracy_dark bricks": 0.8686347715638351,
-      "eval_accuracy_grass": 0.9113963867928155,
-      "eval_accuracy_red bricks": 0.96484375,
-      "eval_accuracy_road": 0.9082512784360577,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.49201885700107517,
-      "eval_accuracy_speedway bricks": 0.9883361271114748,
-      "eval_accuracy_steel": 0.8976568168326627,
-      "eval_accuracy_tiles": 0.09197370223257088,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9795448298302779,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.883473699263173,
-      "eval_iou_concrete tiles": 0.9163479788107985,
-      "eval_iou_dark bricks": 0.8138184703039919,
-      "eval_iou_grass": 0.8338134994356317,
-      "eval_iou_red bricks": 0.8565573770491803,
-      "eval_iou_road": 0.8990551154712573,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32707180295422056,
-      "eval_iou_speedway bricks": 0.9762023252334978,
-      "eval_iou_steel": 0.8433646278738304,
-      "eval_iou_tiles": 0.0918165037259862,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11633595079183578,
-      "eval_mean_accuracy": 0.6960825941985977,
-      "eval_mean_iou": 0.6477743253809111,
-      "eval_overall_accuracy": 0.9712529712253146,
-      "eval_runtime": 3.5851,
-      "eval_samples_per_second": 10.041,
-      "eval_steps_per_second": 0.558,
-      "step": 2680
-    },
-    {
-      "epoch": 243.73,
-      "learning_rate": 5.3952153110047855e-05,
-      "loss": 0.0361,
-      "step": 2681
-    },
-    {
-      "epoch": 243.82,
-      "learning_rate": 5.3933014354066994e-05,
-      "loss": 0.0388,
-      "step": 2682
-    },
-    {
-      "epoch": 243.91,
-      "learning_rate": 5.3913875598086126e-05,
-      "loss": 0.0418,
-      "step": 2683
-    },
-    {
-      "epoch": 244.0,
-      "learning_rate": 5.3894736842105265e-05,
-      "loss": 0.0341,
-      "step": 2684
-    },
-    {
-      "epoch": 244.09,
-      "learning_rate": 5.3875598086124403e-05,
-      "loss": 0.0324,
-      "step": 2685
-    },
-    {
-      "epoch": 244.18,
-      "learning_rate": 5.3856459330143535e-05,
-      "loss": 0.0362,
-      "step": 2686
-    },
-    {
-      "epoch": 244.27,
-      "learning_rate": 5.383732057416269e-05,
-      "loss": 0.0385,
-      "step": 2687
-    },
-    {
-      "epoch": 244.36,
-      "learning_rate": 5.3818181818181827e-05,
-      "loss": 0.0451,
-      "step": 2688
-    },
-    {
-      "epoch": 244.45,
-      "learning_rate": 5.379904306220096e-05,
-      "loss": 0.0359,
-      "step": 2689
-    },
-    {
-      "epoch": 244.55,
-      "learning_rate": 5.37799043062201e-05,
-      "loss": 0.0368,
-      "step": 2690
-    },
-    {
-      "epoch": 244.64,
-      "learning_rate": 5.3760765550239236e-05,
-      "loss": 0.0393,
-      "step": 2691
-    },
-    {
-      "epoch": 244.73,
-      "learning_rate": 5.374162679425837e-05,
-      "loss": 0.0364,
-      "step": 2692
-    },
-    {
-      "epoch": 244.82,
-      "learning_rate": 5.372248803827752e-05,
-      "loss": 0.0385,
-      "step": 2693
-    },
-    {
-      "epoch": 244.91,
-      "learning_rate": 5.370334928229665e-05,
-      "loss": 0.0336,
-      "step": 2694
-    },
-    {
-      "epoch": 245.0,
-      "learning_rate": 5.368421052631579e-05,
-      "loss": 0.0414,
-      "step": 2695
-    },
-    {
-      "epoch": 245.09,
-      "learning_rate": 5.366507177033493e-05,
-      "loss": 0.0392,
-      "step": 2696
-    },
-    {
-      "epoch": 245.18,
-      "learning_rate": 5.364593301435407e-05,
-      "loss": 0.0423,
-      "step": 2697
-    },
-    {
-      "epoch": 245.27,
-      "learning_rate": 5.36267942583732e-05,
-      "loss": 0.0367,
-      "step": 2698
-    },
-    {
-      "epoch": 245.36,
-      "learning_rate": 5.360765550239235e-05,
-      "loss": 0.0378,
-      "step": 2699
-    },
-    {
-      "epoch": 245.45,
-      "learning_rate": 5.3588516746411485e-05,
-      "loss": 0.0339,
-      "step": 2700
-    },
-    {
-      "epoch": 245.45,
-      "eval_accuracy_NAT": 0.9929299566706639,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9489369178934718,
-      "eval_accuracy_concrete tiles": 0.9853237937993029,
-      "eval_accuracy_dark bricks": 0.8640160117008583,
-      "eval_accuracy_grass": 0.8986218924426022,
-      "eval_accuracy_red bricks": 0.9231178977272727,
-      "eval_accuracy_road": 0.9173795960516913,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5220136189452209,
-      "eval_accuracy_speedway bricks": 0.9885774902338447,
-      "eval_accuracy_steel": 0.8957275245696195,
-      "eval_accuracy_tiles": 0.09320641008081085,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799872521394463,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8850239551669357,
-      "eval_iou_concrete tiles": 0.9477120338783672,
-      "eval_iou_dark bricks": 0.8061770515352846,
-      "eval_iou_grass": 0.8335067705344783,
-      "eval_iou_red bricks": 0.8478473581213307,
-      "eval_iou_road": 0.9102668549605315,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.34518275453468233,
-      "eval_iou_speedway bricks": 0.9760963272552868,
-      "eval_iou_steel": 0.8393219972342184,
-      "eval_iou_tiles": 0.09309165526675786,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10759963095188141,
-      "eval_mean_accuracy": 0.6946039315473354,
-      "eval_mean_iou": 0.6510933854328708,
-      "eval_overall_accuracy": 0.9724963506062826,
-      "eval_runtime": 4.4408,
-      "eval_samples_per_second": 8.107,
-      "eval_steps_per_second": 0.45,
-      "step": 2700
-    },
-    {
-      "epoch": 245.55,
-      "learning_rate": 5.3569377990430624e-05,
-      "loss": 0.0324,
-      "step": 2701
-    },
-    {
-      "epoch": 245.64,
-      "learning_rate": 5.355023923444976e-05,
-      "loss": 0.04,
-      "step": 2702
-    },
-    {
-      "epoch": 245.73,
-      "learning_rate": 5.35311004784689e-05,
-      "loss": 0.0402,
-      "step": 2703
-    },
-    {
-      "epoch": 245.82,
-      "learning_rate": 5.3511961722488033e-05,
-      "loss": 0.0375,
-      "step": 2704
-    },
-    {
-      "epoch": 245.91,
-      "learning_rate": 5.3492822966507186e-05,
-      "loss": 0.0392,
-      "step": 2705
-    },
-    {
-      "epoch": 246.0,
-      "learning_rate": 5.347368421052632e-05,
-      "loss": 0.0378,
-      "step": 2706
-    },
-    {
-      "epoch": 246.09,
-      "learning_rate": 5.3454545454545457e-05,
-      "loss": 0.0368,
-      "step": 2707
-    },
-    {
-      "epoch": 246.18,
-      "learning_rate": 5.3435406698564595e-05,
-      "loss": 0.042,
-      "step": 2708
-    },
-    {
-      "epoch": 246.27,
-      "learning_rate": 5.3416267942583734e-05,
-      "loss": 0.0335,
-      "step": 2709
-    },
-    {
-      "epoch": 246.36,
-      "learning_rate": 5.339712918660288e-05,
-      "loss": 0.0349,
-      "step": 2710
-    },
-    {
-      "epoch": 246.45,
-      "learning_rate": 5.337799043062202e-05,
-      "loss": 0.0345,
-      "step": 2711
-    },
-    {
-      "epoch": 246.55,
-      "learning_rate": 5.335885167464115e-05,
-      "loss": 0.0364,
-      "step": 2712
-    },
-    {
-      "epoch": 246.64,
-      "learning_rate": 5.333971291866029e-05,
-      "loss": 0.0385,
-      "step": 2713
-    },
-    {
-      "epoch": 246.73,
-      "learning_rate": 5.332057416267943e-05,
-      "loss": 0.0345,
-      "step": 2714
-    },
-    {
-      "epoch": 246.82,
-      "learning_rate": 5.330143540669856e-05,
-      "loss": 0.0384,
-      "step": 2715
-    },
-    {
-      "epoch": 246.91,
-      "learning_rate": 5.328229665071771e-05,
-      "loss": 0.0409,
-      "step": 2716
-    },
-    {
-      "epoch": 247.0,
-      "learning_rate": 5.326315789473685e-05,
-      "loss": 0.0338,
-      "step": 2717
-    },
-    {
-      "epoch": 247.09,
-      "learning_rate": 5.324401913875598e-05,
-      "loss": 0.0383,
-      "step": 2718
-    },
-    {
-      "epoch": 247.18,
-      "learning_rate": 5.322488038277512e-05,
-      "loss": 0.0344,
-      "step": 2719
-    },
-    {
-      "epoch": 247.27,
-      "learning_rate": 5.320574162679426e-05,
-      "loss": 0.0353,
-      "step": 2720
-    },
-    {
-      "epoch": 247.27,
-      "eval_accuracy_NAT": 0.992500474653503,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9479725524700631,
-      "eval_accuracy_concrete tiles": 0.9854460955176421,
-      "eval_accuracy_dark bricks": 0.8580886032100381,
-      "eval_accuracy_grass": 0.9092154608079911,
-      "eval_accuracy_red bricks": 0.9392755681818182,
-      "eval_accuracy_road": 0.9461333733178393,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47870316764535603,
-      "eval_accuracy_speedway bricks": 0.9880265789070354,
-      "eval_accuracy_steel": 0.9056048413693029,
-      "eval_accuracy_tiles": 0.20367073003698125,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9801403090563118,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8871386002655975,
-      "eval_iou_concrete tiles": 0.9402532236419862,
-      "eval_iou_dark bricks": 0.8099251616653346,
-      "eval_iou_grass": 0.8380863402061856,
-      "eval_iou_red bricks": 0.8565414507772021,
-      "eval_iou_road": 0.9372581808250421,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.318909785483397,
-      "eval_iou_speedway bricks": 0.9762782785802742,
-      "eval_iou_steel": 0.8380434285016709,
-      "eval_iou_tiles": 0.2032114793303724,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09617145359516144,
-      "eval_mean_accuracy": 0.7042028804705822,
-      "eval_mean_iou": 0.6604450952564135,
-      "eval_overall_accuracy": 0.9750315348307291,
-      "eval_runtime": 3.5328,
-      "eval_samples_per_second": 10.19,
-      "eval_steps_per_second": 0.566,
-      "step": 2720
-    },
-    {
-      "epoch": 247.36,
-      "learning_rate": 5.318660287081339e-05,
-      "loss": 0.042,
-      "step": 2721
-    },
-    {
-      "epoch": 247.45,
-      "learning_rate": 5.3167464114832545e-05,
-      "loss": 0.0401,
-      "step": 2722
-    },
-    {
-      "epoch": 247.55,
-      "learning_rate": 5.3148325358851684e-05,
-      "loss": 0.041,
-      "step": 2723
-    },
-    {
-      "epoch": 247.64,
-      "learning_rate": 5.3129186602870816e-05,
-      "loss": 0.0372,
-      "step": 2724
-    },
-    {
-      "epoch": 247.73,
-      "learning_rate": 5.3110047846889955e-05,
-      "loss": 0.0325,
-      "step": 2725
-    },
-    {
-      "epoch": 247.82,
-      "learning_rate": 5.309090909090909e-05,
-      "loss": 0.043,
-      "step": 2726
-    },
-    {
-      "epoch": 247.91,
-      "learning_rate": 5.3071770334928225e-05,
-      "loss": 0.04,
-      "step": 2727
-    },
-    {
-      "epoch": 248.0,
-      "learning_rate": 5.305263157894738e-05,
-      "loss": 0.0359,
-      "step": 2728
-    },
-    {
-      "epoch": 248.09,
-      "learning_rate": 5.303349282296651e-05,
-      "loss": 0.0411,
-      "step": 2729
-    },
-    {
-      "epoch": 248.18,
-      "learning_rate": 5.301435406698565e-05,
-      "loss": 0.0328,
-      "step": 2730
-    },
-    {
-      "epoch": 248.27,
-      "learning_rate": 5.299521531100479e-05,
-      "loss": 0.0363,
-      "step": 2731
-    },
-    {
-      "epoch": 248.36,
-      "learning_rate": 5.2976076555023926e-05,
-      "loss": 0.039,
-      "step": 2732
-    },
-    {
-      "epoch": 248.45,
-      "learning_rate": 5.295693779904306e-05,
-      "loss": 0.0406,
-      "step": 2733
-    },
-    {
-      "epoch": 248.55,
-      "learning_rate": 5.293779904306221e-05,
-      "loss": 0.0446,
-      "step": 2734
-    },
-    {
-      "epoch": 248.64,
-      "learning_rate": 5.291866028708134e-05,
-      "loss": 0.0409,
-      "step": 2735
-    },
-    {
-      "epoch": 248.73,
-      "learning_rate": 5.289952153110048e-05,
-      "loss": 0.0365,
-      "step": 2736
-    },
-    {
-      "epoch": 248.82,
-      "learning_rate": 5.288038277511962e-05,
-      "loss": 0.0386,
-      "step": 2737
-    },
-    {
-      "epoch": 248.91,
-      "learning_rate": 5.286124401913876e-05,
-      "loss": 0.0339,
-      "step": 2738
-    },
-    {
-      "epoch": 249.0,
-      "learning_rate": 5.284210526315789e-05,
-      "loss": 0.0393,
-      "step": 2739
-    },
-    {
-      "epoch": 249.09,
-      "learning_rate": 5.282296650717704e-05,
-      "loss": 0.0328,
-      "step": 2740
-    },
-    {
-      "epoch": 249.09,
-      "eval_accuracy_NAT": 0.9935515939535143,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9462584231082879,
-      "eval_accuracy_concrete tiles": 0.9652663119916834,
-      "eval_accuracy_dark bricks": 0.870636234171125,
-      "eval_accuracy_grass": 0.8975174491554154,
-      "eval_accuracy_red bricks": 0.9248934659090909,
-      "eval_accuracy_road": 0.9470451167914369,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4561519587572023,
-      "eval_accuracy_speedway bricks": 0.9892689955794344,
-      "eval_accuracy_steel": 0.8997097816766704,
-      "eval_accuracy_tiles": 0.08991918915217094,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9795770493451709,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8851411709206973,
-      "eval_iou_concrete tiles": 0.9499593777269582,
-      "eval_iou_dark bricks": 0.8176691729323309,
-      "eval_iou_grass": 0.8336958639049412,
-      "eval_iou_red bricks": 0.8588623248145095,
-      "eval_iou_road": 0.9398512210839333,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.30979217375023405,
-      "eval_iou_speedway bricks": 0.9763788812440706,
-      "eval_iou_steel": 0.8431381638928809,
-      "eval_iou_tiles": 0.08988225629791895,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09706311672925949,
-      "eval_mean_accuracy": 0.6907860400189256,
-      "eval_mean_iou": 0.6526113581472036,
-      "eval_overall_accuracy": 0.9750883314344618,
-      "eval_runtime": 4.0496,
-      "eval_samples_per_second": 8.89,
-      "eval_steps_per_second": 0.494,
-      "step": 2740
-    },
-    {
-      "epoch": 249.18,
-      "learning_rate": 5.2803827751196175e-05,
-      "loss": 0.0335,
-      "step": 2741
-    },
-    {
-      "epoch": 249.27,
-      "learning_rate": 5.2784688995215314e-05,
-      "loss": 0.0366,
-      "step": 2742
-    },
-    {
-      "epoch": 249.36,
-      "learning_rate": 5.276555023923445e-05,
-      "loss": 0.0398,
-      "step": 2743
-    },
-    {
-      "epoch": 249.45,
-      "learning_rate": 5.274641148325359e-05,
-      "loss": 0.0363,
-      "step": 2744
-    },
-    {
-      "epoch": 249.55,
-      "learning_rate": 5.272727272727272e-05,
-      "loss": 0.04,
-      "step": 2745
-    },
-    {
-      "epoch": 249.64,
-      "learning_rate": 5.2708133971291876e-05,
-      "loss": 0.033,
-      "step": 2746
-    },
-    {
-      "epoch": 249.73,
-      "learning_rate": 5.268899521531101e-05,
-      "loss": 0.0389,
-      "step": 2747
-    },
-    {
-      "epoch": 249.82,
-      "learning_rate": 5.2669856459330146e-05,
-      "loss": 0.0358,
-      "step": 2748
-    },
-    {
-      "epoch": 249.91,
-      "learning_rate": 5.2650717703349285e-05,
-      "loss": 0.037,
-      "step": 2749
-    },
-    {
-      "epoch": 250.0,
-      "learning_rate": 5.2631578947368424e-05,
-      "loss": 0.0371,
-      "step": 2750
-    },
-    {
-      "epoch": 250.09,
-      "learning_rate": 5.2612440191387556e-05,
-      "loss": 0.0372,
-      "step": 2751
-    },
-    {
-      "epoch": 250.18,
-      "learning_rate": 5.259330143540671e-05,
-      "loss": 0.0388,
-      "step": 2752
-    },
-    {
-      "epoch": 250.27,
-      "learning_rate": 5.257416267942584e-05,
-      "loss": 0.0347,
-      "step": 2753
-    },
-    {
-      "epoch": 250.36,
-      "learning_rate": 5.255502392344498e-05,
-      "loss": 0.0387,
-      "step": 2754
-    },
-    {
-      "epoch": 250.45,
-      "learning_rate": 5.253588516746412e-05,
-      "loss": 0.0341,
-      "step": 2755
-    },
-    {
-      "epoch": 250.55,
-      "learning_rate": 5.251674641148325e-05,
-      "loss": 0.0395,
-      "step": 2756
-    },
-    {
-      "epoch": 250.64,
-      "learning_rate": 5.249760765550239e-05,
-      "loss": 0.036,
-      "step": 2757
-    },
-    {
-      "epoch": 250.73,
-      "learning_rate": 5.247846889952154e-05,
-      "loss": 0.0344,
-      "step": 2758
-    },
-    {
-      "epoch": 250.82,
-      "learning_rate": 5.245933014354067e-05,
-      "loss": 0.051,
-      "step": 2759
-    },
-    {
-      "epoch": 250.91,
-      "learning_rate": 5.244019138755981e-05,
-      "loss": 0.0356,
-      "step": 2760
-    },
-    {
-      "epoch": 250.91,
-      "eval_accuracy_NAT": 0.9925807599114357,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9497604510923215,
-      "eval_accuracy_concrete tiles": 0.9795756130373632,
-      "eval_accuracy_dark bricks": 0.8613602247796467,
-      "eval_accuracy_grass": 0.9064333821478626,
-      "eval_accuracy_red bricks": 0.939453125,
-      "eval_accuracy_road": 0.944006374949062,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4340694180244259,
-      "eval_accuracy_speedway bricks": 0.9880308027616769,
-      "eval_accuracy_steel": 0.9107248862212255,
-      "eval_accuracy_tiles": 0.08087933159841118,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9801673303635121,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8818208988680091,
-      "eval_iou_concrete tiles": 0.9521516880646695,
-      "eval_iou_dark bricks": 0.8090744757772957,
-      "eval_iou_grass": 0.8357319171699074,
-      "eval_iou_red bricks": 0.8499598393574297,
-      "eval_iou_road": 0.9369569939835023,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2907986111111111,
-      "eval_iou_speedway bricks": 0.976182342704557,
-      "eval_iou_steel": 0.8418758145525772,
-      "eval_iou_tiles": 0.08080186097427476,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10148781538009644,
-      "eval_mean_accuracy": 0.6912980284248794,
-      "eval_mean_iou": 0.648886290225142,
-      "eval_overall_accuracy": 0.9746440251668295,
-      "eval_runtime": 3.9075,
-      "eval_samples_per_second": 9.213,
-      "eval_steps_per_second": 0.512,
-      "step": 2760
-    },
-    {
-      "epoch": 251.0,
-      "learning_rate": 5.242105263157895e-05,
-      "loss": 0.0359,
-      "step": 2761
-    },
-    {
-      "epoch": 251.09,
-      "learning_rate": 5.240191387559808e-05,
-      "loss": 0.0409,
-      "step": 2762
-    },
-    {
-      "epoch": 251.18,
-      "learning_rate": 5.238277511961722e-05,
-      "loss": 0.0323,
-      "step": 2763
-    },
-    {
-      "epoch": 251.27,
-      "learning_rate": 5.2363636363636374e-05,
-      "loss": 0.0371,
-      "step": 2764
-    },
-    {
-      "epoch": 251.36,
-      "learning_rate": 5.2344497607655506e-05,
-      "loss": 0.0373,
-      "step": 2765
-    },
-    {
-      "epoch": 251.45,
-      "learning_rate": 5.2325358851674644e-05,
-      "loss": 0.0337,
-      "step": 2766
-    },
-    {
-      "epoch": 251.55,
-      "learning_rate": 5.230622009569378e-05,
-      "loss": 0.0361,
-      "step": 2767
-    },
-    {
-      "epoch": 251.64,
-      "learning_rate": 5.2287081339712915e-05,
-      "loss": 0.038,
-      "step": 2768
-    },
-    {
-      "epoch": 251.73,
-      "learning_rate": 5.2267942583732054e-05,
-      "loss": 0.0373,
-      "step": 2769
-    },
-    {
-      "epoch": 251.82,
-      "learning_rate": 5.22488038277512e-05,
-      "loss": 0.0376,
-      "step": 2770
-    },
-    {
-      "epoch": 251.91,
-      "learning_rate": 5.222966507177034e-05,
-      "loss": 0.0367,
-      "step": 2771
-    },
-    {
-      "epoch": 252.0,
-      "learning_rate": 5.221052631578948e-05,
-      "loss": 0.0342,
-      "step": 2772
-    },
-    {
-      "epoch": 252.09,
-      "learning_rate": 5.2191387559808616e-05,
-      "loss": 0.0374,
-      "step": 2773
-    },
-    {
-      "epoch": 252.18,
-      "learning_rate": 5.217224880382775e-05,
-      "loss": 0.0398,
-      "step": 2774
-    },
-    {
-      "epoch": 252.27,
-      "learning_rate": 5.2153110047846887e-05,
-      "loss": 0.0348,
-      "step": 2775
-    },
-    {
-      "epoch": 252.36,
-      "learning_rate": 5.213397129186603e-05,
-      "loss": 0.0342,
-      "step": 2776
-    },
-    {
-      "epoch": 252.45,
-      "learning_rate": 5.211483253588517e-05,
-      "loss": 0.0435,
-      "step": 2777
-    },
-    {
-      "epoch": 252.55,
-      "learning_rate": 5.209569377990431e-05,
-      "loss": 0.0435,
-      "step": 2778
-    },
-    {
-      "epoch": 252.64,
-      "learning_rate": 5.207655502392345e-05,
-      "loss": 0.0317,
-      "step": 2779
-    },
-    {
-      "epoch": 252.73,
-      "learning_rate": 5.205741626794258e-05,
-      "loss": 0.0383,
-      "step": 2780
-    },
-    {
-      "epoch": 252.73,
-      "eval_accuracy_NAT": 0.9927822670862909,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9431426777973304,
-      "eval_accuracy_concrete tiles": 0.9885647893352901,
-      "eval_accuracy_dark bricks": 0.8666333089565452,
-      "eval_accuracy_grass": 0.9144161304641109,
-      "eval_accuracy_red bricks": 0.9359019886363636,
-      "eval_accuracy_road": 0.9496050119288453,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46403661125354945,
-      "eval_accuracy_speedway bricks": 0.98888764184609,
-      "eval_accuracy_steel": 0.9036425697513356,
-      "eval_accuracy_tiles": 0.08108478290645117,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9800480061592808,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8831041775509384,
-      "eval_iou_concrete tiles": 0.932806324110672,
-      "eval_iou_dark bricks": 0.8104236403556131,
-      "eval_iou_grass": 0.8372695852534562,
-      "eval_iou_red bricks": 0.8523609314359638,
-      "eval_iou_road": 0.9420186241898549,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3137839752432795,
-      "eval_iou_speedway bricks": 0.9761885332076891,
-      "eval_iou_steel": 0.8420935360690879,
-      "eval_iou_tiles": 0.0810292909937038,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09606492519378662,
-      "eval_mean_accuracy": 0.6945152138432463,
-      "eval_mean_iou": 0.6500866634284261,
-      "eval_overall_accuracy": 0.9751933415730795,
-      "eval_runtime": 4.1153,
-      "eval_samples_per_second": 8.748,
-      "eval_steps_per_second": 0.486,
-      "step": 2780
-    },
-    {
-      "epoch": 252.82,
-      "learning_rate": 5.203827751196173e-05,
-      "loss": 0.0479,
-      "step": 2781
-    },
-    {
-      "epoch": 252.91,
-      "learning_rate": 5.2019138755980865e-05,
-      "loss": 0.0389,
-      "step": 2782
-    },
-    {
-      "epoch": 253.0,
-      "learning_rate": 5.2000000000000004e-05,
-      "loss": 0.0348,
-      "step": 2783
-    },
-    {
-      "epoch": 253.09,
-      "learning_rate": 5.198086124401914e-05,
-      "loss": 0.0337,
-      "step": 2784
-    },
-    {
-      "epoch": 253.18,
-      "learning_rate": 5.196172248803828e-05,
-      "loss": 0.0372,
-      "step": 2785
-    },
-    {
-      "epoch": 253.27,
-      "learning_rate": 5.194258373205741e-05,
-      "loss": 0.0401,
-      "step": 2786
-    },
-    {
-      "epoch": 253.36,
-      "learning_rate": 5.1923444976076565e-05,
-      "loss": 0.0366,
-      "step": 2787
-    },
-    {
-      "epoch": 253.45,
-      "learning_rate": 5.19043062200957e-05,
-      "loss": 0.0353,
-      "step": 2788
-    },
-    {
-      "epoch": 253.55,
-      "learning_rate": 5.1885167464114836e-05,
-      "loss": 0.0377,
-      "step": 2789
-    },
-    {
-      "epoch": 253.64,
-      "learning_rate": 5.1866028708133975e-05,
-      "loss": 0.0325,
-      "step": 2790
-    },
-    {
-      "epoch": 253.73,
-      "learning_rate": 5.184688995215311e-05,
-      "loss": 0.0393,
-      "step": 2791
-    },
-    {
-      "epoch": 253.82,
-      "learning_rate": 5.1827751196172246e-05,
-      "loss": 0.0334,
-      "step": 2792
-    },
-    {
-      "epoch": 253.91,
-      "learning_rate": 5.18086124401914e-05,
-      "loss": 0.0429,
-      "step": 2793
-    },
-    {
-      "epoch": 254.0,
-      "learning_rate": 5.178947368421053e-05,
-      "loss": 0.0334,
-      "step": 2794
-    },
-    {
-      "epoch": 254.09,
-      "learning_rate": 5.177033492822967e-05,
-      "loss": 0.0409,
-      "step": 2795
-    },
-    {
-      "epoch": 254.18,
-      "learning_rate": 5.175119617224881e-05,
-      "loss": 0.033,
-      "step": 2796
-    },
-    {
-      "epoch": 254.27,
-      "learning_rate": 5.173205741626794e-05,
-      "loss": 0.04,
-      "step": 2797
-    },
-    {
-      "epoch": 254.36,
-      "learning_rate": 5.171291866028708e-05,
-      "loss": 0.0377,
-      "step": 2798
-    },
-    {
-      "epoch": 254.45,
-      "learning_rate": 5.169377990430623e-05,
-      "loss": 0.0444,
-      "step": 2799
-    },
-    {
-      "epoch": 254.55,
-      "learning_rate": 5.167464114832536e-05,
-      "loss": 0.0368,
-      "step": 2800
-    },
-    {
-      "epoch": 254.55,
-      "eval_accuracy_NAT": 0.992989950050218,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9435316429889695,
-      "eval_accuracy_concrete tiles": 0.9887176664832141,
-      "eval_accuracy_dark bricks": 0.8617066317693699,
-      "eval_accuracy_grass": 0.909225946029072,
-      "eval_accuracy_red bricks": 0.8796164772727273,
-      "eval_accuracy_road": 0.9319445166610841,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4621343699170182,
-      "eval_accuracy_speedway bricks": 0.9893661442361883,
-      "eval_accuracy_steel": 0.9046237055603192,
-      "eval_accuracy_tiles": 0.06834680180797151,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9802694173006918,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8825871870181456,
-      "eval_iou_concrete tiles": 0.9358666396550227,
-      "eval_iou_dark bricks": 0.819983152034575,
-      "eval_iou_grass": 0.8368881253860407,
-      "eval_iou_red bricks": 0.8386659895039783,
-      "eval_iou_road": 0.9239040787164462,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31298194514460687,
-      "eval_iou_speedway bricks": 0.9759080963842801,
-      "eval_iou_steel": 0.8427423690799883,
-      "eval_iou_tiles": 0.06831872946330778,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10566778481006622,
-      "eval_mean_accuracy": 0.687092604059704,
-      "eval_mean_iou": 0.6475473638220832,
-      "eval_overall_accuracy": 0.9736668268839518,
-      "eval_runtime": 3.8523,
-      "eval_samples_per_second": 9.345,
-      "eval_steps_per_second": 0.519,
-      "step": 2800
-    },
-    {
-      "epoch": 254.64,
-      "learning_rate": 5.16555023923445e-05,
-      "loss": 0.0315,
-      "step": 2801
-    },
-    {
-      "epoch": 254.73,
-      "learning_rate": 5.163636363636364e-05,
-      "loss": 0.0383,
-      "step": 2802
-    },
-    {
-      "epoch": 254.82,
-      "learning_rate": 5.161722488038277e-05,
-      "loss": 0.043,
-      "step": 2803
-    },
-    {
-      "epoch": 254.91,
-      "learning_rate": 5.159808612440191e-05,
-      "loss": 0.0314,
-      "step": 2804
-    },
-    {
-      "epoch": 255.0,
-      "learning_rate": 5.157894736842106e-05,
-      "loss": 0.0337,
-      "step": 2805
-    },
-    {
-      "epoch": 255.09,
-      "learning_rate": 5.1559808612440195e-05,
-      "loss": 0.0385,
-      "step": 2806
-    },
-    {
-      "epoch": 255.18,
-      "learning_rate": 5.1540669856459334e-05,
-      "loss": 0.0338,
-      "step": 2807
-    },
-    {
-      "epoch": 255.27,
-      "learning_rate": 5.152153110047847e-05,
-      "loss": 0.0394,
-      "step": 2808
-    },
-    {
-      "epoch": 255.36,
-      "learning_rate": 5.1502392344497605e-05,
-      "loss": 0.0373,
-      "step": 2809
-    },
-    {
-      "epoch": 255.45,
-      "learning_rate": 5.1483253588516744e-05,
-      "loss": 0.0392,
-      "step": 2810
-    },
-    {
-      "epoch": 255.55,
-      "learning_rate": 5.146411483253589e-05,
-      "loss": 0.0392,
-      "step": 2811
-    },
-    {
-      "epoch": 255.64,
-      "learning_rate": 5.144497607655503e-05,
-      "loss": 0.0401,
-      "step": 2812
-    },
-    {
-      "epoch": 255.73,
-      "learning_rate": 5.142583732057417e-05,
-      "loss": 0.036,
-      "step": 2813
-    },
-    {
-      "epoch": 255.82,
-      "learning_rate": 5.1406698564593306e-05,
-      "loss": 0.0385,
-      "step": 2814
-    },
-    {
-      "epoch": 255.91,
-      "learning_rate": 5.138755980861244e-05,
-      "loss": 0.0365,
-      "step": 2815
-    },
-    {
-      "epoch": 256.0,
-      "learning_rate": 5.1368421052631576e-05,
-      "loss": 0.0372,
-      "step": 2816
-    },
-    {
-      "epoch": 256.09,
-      "learning_rate": 5.134928229665072e-05,
-      "loss": 0.0345,
-      "step": 2817
-    },
-    {
-      "epoch": 256.18,
-      "learning_rate": 5.133014354066986e-05,
-      "loss": 0.0311,
-      "step": 2818
-    },
-    {
-      "epoch": 256.27,
-      "learning_rate": 5.1311004784689e-05,
-      "loss": 0.0353,
-      "step": 2819
-    },
-    {
-      "epoch": 256.36,
-      "learning_rate": 5.129186602870814e-05,
-      "loss": 0.0384,
-      "step": 2820
-    },
-    {
-      "epoch": 256.36,
-      "eval_accuracy_NAT": 0.9929414259932257,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9478344027640672,
-      "eval_accuracy_concrete tiles": 0.9822356754112396,
-      "eval_accuracy_dark bricks": 0.8569724029098187,
-      "eval_accuracy_grass": 0.9111202759710189,
-      "eval_accuracy_red bricks": 0.9019886363636364,
-      "eval_accuracy_road": 0.9511237782456072,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.44256058225126127,
-      "eval_accuracy_speedway bricks": 0.9888098022391257,
-      "eval_accuracy_steel": 0.9009300178088516,
-      "eval_accuracy_tiles": 0.048007122312012054,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803511278162742,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8846187067110473,
-      "eval_iou_concrete tiles": 0.9493203309692672,
-      "eval_iou_dark bricks": 0.8201642907135227,
-      "eval_iou_grass": 0.8403754960461375,
-      "eval_iou_red bricks": 0.8516345347862532,
-      "eval_iou_road": 0.9428150190284978,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2983108170888075,
-      "eval_iou_speedway bricks": 0.9761954262573503,
-      "eval_iou_steel": 0.8415572413261966,
-      "eval_iou_tiles": 0.04800054779512462,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09818898886442184,
-      "eval_mean_accuracy": 0.6865018555592204,
-      "eval_mean_iou": 0.6487187337337291,
-      "eval_overall_accuracy": 0.9754417207505968,
-      "eval_runtime": 4.3946,
-      "eval_samples_per_second": 8.192,
-      "eval_steps_per_second": 0.455,
-      "step": 2820
-    },
-    {
-      "epoch": 256.45,
-      "learning_rate": 5.127272727272727e-05,
-      "loss": 0.0413,
-      "step": 2821
-    },
-    {
-      "epoch": 256.55,
-      "learning_rate": 5.125358851674641e-05,
-      "loss": 0.0375,
-      "step": 2822
-    },
-    {
-      "epoch": 256.64,
-      "learning_rate": 5.1234449760765555e-05,
-      "loss": 0.0358,
-      "step": 2823
-    },
-    {
-      "epoch": 256.73,
-      "learning_rate": 5.1215311004784693e-05,
-      "loss": 0.0356,
-      "step": 2824
-    },
-    {
-      "epoch": 256.82,
-      "learning_rate": 5.119617224880383e-05,
-      "loss": 0.0384,
-      "step": 2825
-    },
-    {
-      "epoch": 256.91,
-      "learning_rate": 5.117703349282297e-05,
-      "loss": 0.0345,
-      "step": 2826
-    },
-    {
-      "epoch": 257.0,
-      "learning_rate": 5.11578947368421e-05,
-      "loss": 0.0389,
-      "step": 2827
-    },
-    {
-      "epoch": 257.09,
-      "learning_rate": 5.113875598086124e-05,
-      "loss": 0.0389,
-      "step": 2828
-    },
-    {
-      "epoch": 257.18,
-      "learning_rate": 5.111961722488039e-05,
-      "loss": 0.0365,
-      "step": 2829
-    },
-    {
-      "epoch": 257.27,
-      "learning_rate": 5.1100478468899526e-05,
-      "loss": 0.0354,
-      "step": 2830
-    },
-    {
-      "epoch": 257.36,
-      "learning_rate": 5.1081339712918665e-05,
-      "loss": 0.0352,
-      "step": 2831
-    },
-    {
-      "epoch": 257.45,
-      "learning_rate": 5.10622009569378e-05,
-      "loss": 0.0352,
-      "step": 2832
-    },
-    {
-      "epoch": 257.55,
-      "learning_rate": 5.1043062200956936e-05,
-      "loss": 0.033,
-      "step": 2833
-    },
-    {
-      "epoch": 257.64,
-      "learning_rate": 5.1023923444976074e-05,
-      "loss": 0.0395,
-      "step": 2834
-    },
-    {
-      "epoch": 257.73,
-      "learning_rate": 5.100478468899522e-05,
-      "loss": 0.0358,
-      "step": 2835
-    },
-    {
-      "epoch": 257.82,
-      "learning_rate": 5.098564593301436e-05,
-      "loss": 0.0336,
-      "step": 2836
-    },
-    {
-      "epoch": 257.91,
-      "learning_rate": 5.09665071770335e-05,
-      "loss": 0.0381,
-      "step": 2837
-    },
-    {
-      "epoch": 258.0,
-      "learning_rate": 5.094736842105263e-05,
-      "loss": 0.041,
-      "step": 2838
-    },
-    {
-      "epoch": 258.09,
-      "learning_rate": 5.092822966507177e-05,
-      "loss": 0.0373,
-      "step": 2839
-    },
-    {
-      "epoch": 258.18,
-      "learning_rate": 5.090909090909091e-05,
-      "loss": 0.0397,
-      "step": 2840
-    },
-    {
-      "epoch": 258.18,
-      "eval_accuracy_NAT": 0.9930589424367052,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9459459096956951,
-      "eval_accuracy_concrete tiles": 0.9887176664832141,
-      "eval_accuracy_dark bricks": 0.8675570609291405,
-      "eval_accuracy_grass": 0.8996389588874482,
-      "eval_accuracy_red bricks": 0.9296875,
-      "eval_accuracy_road": 0.8718432091435536,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4512722961982742,
-      "eval_accuracy_speedway bricks": 0.9887470478273095,
-      "eval_accuracy_steel": 0.9032385726535189,
-      "eval_accuracy_tiles": 0.08478290645117108,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.97975283152225,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8837843730889916,
-      "eval_iou_concrete tiles": 0.934622387930287,
-      "eval_iou_dark bricks": 0.8212789214793222,
-      "eval_iou_grass": 0.8376053913513197,
-      "eval_iou_red bricks": 0.8609010194015126,
-      "eval_iou_road": 0.8646187718627121,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3021058265507632,
-      "eval_iou_speedway bricks": 0.9763428546233143,
-      "eval_iou_steel": 0.8403804848112918,
-      "eval_iou_tiles": 0.08469010808592146,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14663034677505493,
-      "eval_mean_accuracy": 0.6864992362081561,
-      "eval_mean_iou": 0.6450833054390527,
-      "eval_overall_accuracy": 0.9682447645399306,
-      "eval_runtime": 4.064,
-      "eval_samples_per_second": 8.858,
-      "eval_steps_per_second": 0.492,
-      "step": 2840
-    },
-    {
-      "epoch": 258.27,
-      "learning_rate": 5.088995215311005e-05,
-      "loss": 0.0321,
-      "step": 2841
-    },
-    {
-      "epoch": 258.36,
-      "learning_rate": 5.087081339712919e-05,
-      "loss": 0.0328,
-      "step": 2842
-    },
-    {
-      "epoch": 258.45,
-      "learning_rate": 5.085167464114833e-05,
-      "loss": 0.0353,
-      "step": 2843
-    },
-    {
-      "epoch": 258.55,
-      "learning_rate": 5.083253588516746e-05,
-      "loss": 0.036,
-      "step": 2844
-    },
-    {
-      "epoch": 258.64,
-      "learning_rate": 5.08133971291866e-05,
-      "loss": 0.0353,
-      "step": 2845
-    },
-    {
-      "epoch": 258.73,
-      "learning_rate": 5.079425837320574e-05,
-      "loss": 0.0361,
-      "step": 2846
-    },
-    {
-      "epoch": 258.82,
-      "learning_rate": 5.0775119617224885e-05,
-      "loss": 0.0421,
-      "step": 2847
-    },
-    {
-      "epoch": 258.91,
-      "learning_rate": 5.0755980861244024e-05,
-      "loss": 0.04,
-      "step": 2848
-    },
-    {
-      "epoch": 259.0,
-      "learning_rate": 5.073684210526316e-05,
-      "loss": 0.0343,
-      "step": 2849
-    },
-    {
-      "epoch": 259.09,
-      "learning_rate": 5.0717703349282295e-05,
-      "loss": 0.0402,
-      "step": 2850
-    },
-    {
-      "epoch": 259.18,
-      "learning_rate": 5.0698564593301434e-05,
-      "loss": 0.044,
-      "step": 2851
-    },
-    {
-      "epoch": 259.27,
-      "learning_rate": 5.067942583732058e-05,
-      "loss": 0.0322,
-      "step": 2852
-    },
-    {
-      "epoch": 259.36,
-      "learning_rate": 5.066028708133972e-05,
-      "loss": 0.0397,
-      "step": 2853
-    },
-    {
-      "epoch": 259.45,
-      "learning_rate": 5.064114832535886e-05,
-      "loss": 0.0403,
-      "step": 2854
-    },
-    {
-      "epoch": 259.55,
-      "learning_rate": 5.0622009569377996e-05,
-      "loss": 0.0305,
-      "step": 2855
-    },
-    {
-      "epoch": 259.64,
-      "learning_rate": 5.060287081339713e-05,
-      "loss": 0.0355,
-      "step": 2856
-    },
-    {
-      "epoch": 259.73,
-      "learning_rate": 5.0583732057416266e-05,
-      "loss": 0.0369,
-      "step": 2857
-    },
-    {
-      "epoch": 259.82,
-      "learning_rate": 5.056459330143541e-05,
-      "loss": 0.0377,
-      "step": 2858
-    },
-    {
-      "epoch": 259.91,
-      "learning_rate": 5.054545454545455e-05,
-      "loss": 0.037,
-      "step": 2859
-    },
-    {
-      "epoch": 260.0,
-      "learning_rate": 5.052631578947369e-05,
-      "loss": 0.038,
-      "step": 2860
-    },
-    {
-      "epoch": 260.0,
-      "eval_accuracy_NAT": 0.9928856674404636,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.948472842182068,
-      "eval_accuracy_concrete tiles": 0.9878309790252553,
-      "eval_accuracy_dark bricks": 0.8650552326700282,
-      "eval_accuracy_grass": 0.8948087670428531,
-      "eval_accuracy_red bricks": 0.9424715909090909,
-      "eval_accuracy_road": 0.8743801293127945,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4671794447660795,
-      "eval_accuracy_speedway bricks": 0.9866779624607935,
-      "eval_accuracy_steel": 0.9071878504056461,
-      "eval_accuracy_tiles": 0.13785782769483632,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799679449501628,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.883954219198836,
-      "eval_iou_concrete tiles": 0.9326789838337183,
-      "eval_iou_dark bricks": 0.8136335662310393,
-      "eval_iou_grass": 0.829978020268036,
-      "eval_iou_red bricks": 0.8543376790600354,
-      "eval_iou_road": 0.8654150419960889,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3113413805140642,
-      "eval_iou_speedway bricks": 0.9762588652376412,
-      "eval_iou_steel": 0.8424006247320389,
-      "eval_iou_tiles": 0.13775405460890988,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14678435027599335,
-      "eval_mean_accuracy": 0.6926775610699931,
-      "eval_mean_iou": 0.6482861831254284,
-      "eval_overall_accuracy": 0.9682445526123047,
-      "eval_runtime": 3.5491,
-      "eval_samples_per_second": 10.144,
-      "eval_steps_per_second": 0.564,
-      "step": 2860
-    },
-    {
-      "epoch": 260.09,
-      "learning_rate": 5.050717703349283e-05,
-      "loss": 0.0343,
-      "step": 2861
-    },
-    {
-      "epoch": 260.18,
-      "learning_rate": 5.048803827751196e-05,
-      "loss": 0.0381,
-      "step": 2862
-    },
-    {
-      "epoch": 260.27,
-      "learning_rate": 5.04688995215311e-05,
-      "loss": 0.0404,
-      "step": 2863
-    },
-    {
-      "epoch": 260.36,
-      "learning_rate": 5.0449760765550245e-05,
-      "loss": 0.0354,
-      "step": 2864
-    },
-    {
-      "epoch": 260.45,
-      "learning_rate": 5.043062200956938e-05,
-      "loss": 0.0365,
-      "step": 2865
-    },
-    {
-      "epoch": 260.55,
-      "learning_rate": 5.041148325358852e-05,
-      "loss": 0.0325,
-      "step": 2866
-    },
-    {
-      "epoch": 260.64,
-      "learning_rate": 5.039234449760766e-05,
-      "loss": 0.0355,
-      "step": 2867
-    },
-    {
-      "epoch": 260.73,
-      "learning_rate": 5.037320574162679e-05,
-      "loss": 0.034,
-      "step": 2868
-    },
-    {
-      "epoch": 260.82,
-      "learning_rate": 5.035406698564593e-05,
-      "loss": 0.0406,
-      "step": 2869
-    },
-    {
-      "epoch": 260.91,
-      "learning_rate": 5.033492822966508e-05,
-      "loss": 0.0359,
-      "step": 2870
-    },
-    {
-      "epoch": 261.0,
-      "learning_rate": 5.0315789473684216e-05,
-      "loss": 0.0366,
-      "step": 2871
-    },
-    {
-      "epoch": 261.09,
-      "learning_rate": 5.0296650717703355e-05,
-      "loss": 0.0377,
-      "step": 2872
-    },
-    {
-      "epoch": 261.18,
-      "learning_rate": 5.027751196172249e-05,
-      "loss": 0.0415,
-      "step": 2873
-    },
-    {
-      "epoch": 261.27,
-      "learning_rate": 5.0258373205741626e-05,
-      "loss": 0.0343,
-      "step": 2874
-    },
-    {
-      "epoch": 261.36,
-      "learning_rate": 5.0239234449760764e-05,
-      "loss": 0.0376,
-      "step": 2875
-    },
-    {
-      "epoch": 261.45,
-      "learning_rate": 5.022009569377991e-05,
-      "loss": 0.0355,
-      "step": 2876
-    },
-    {
-      "epoch": 261.55,
-      "learning_rate": 5.020095693779905e-05,
-      "loss": 0.0328,
-      "step": 2877
-    },
-    {
-      "epoch": 261.64,
-      "learning_rate": 5.018181818181819e-05,
-      "loss": 0.0372,
-      "step": 2878
-    },
-    {
-      "epoch": 261.73,
-      "learning_rate": 5.016267942583732e-05,
-      "loss": 0.035,
-      "step": 2879
-    },
-    {
-      "epoch": 261.82,
-      "learning_rate": 5.014354066985646e-05,
-      "loss": 0.0387,
-      "step": 2880
-    },
-    {
-      "epoch": 261.82,
-      "eval_accuracy_NAT": 0.99207205134304,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9505652066612301,
-      "eval_accuracy_concrete tiles": 0.9880755824619336,
-      "eval_accuracy_dark bricks": 0.8918055502097687,
-      "eval_accuracy_grass": 0.9010789292492232,
-      "eval_accuracy_red bricks": 0.9176136363636364,
-      "eval_accuracy_road": 0.8766268393578134,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4036611253549472,
-      "eval_accuracy_speedway bricks": 0.9873338667458338,
-      "eval_accuracy_steel": 0.9034282039443309,
-      "eval_accuracy_tiles": 0.14114504862347624,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803229661526144,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8770980349447661,
-      "eval_iou_concrete tiles": 0.9370759148640028,
-      "eval_iou_dark bricks": 0.8182363950983508,
-      "eval_iou_grass": 0.8240344170704579,
-      "eval_iou_red bricks": 0.8543560919160191,
-      "eval_iou_road": 0.8691939059467017,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2717772621809745,
-      "eval_iou_speedway bricks": 0.976319699275038,
-      "eval_iou_steel": 0.8453034838152251,
-      "eval_iou_tiles": 0.1411353831404506,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13669130206108093,
-      "eval_mean_accuracy": 0.6887235415627104,
-      "eval_mean_iou": 0.6457579657234309,
-      "eval_overall_accuracy": 0.9681957032945421,
-      "eval_runtime": 3.759,
-      "eval_samples_per_second": 9.577,
-      "eval_steps_per_second": 0.532,
-      "step": 2880
-    },
-    {
-      "epoch": 261.91,
-      "learning_rate": 5.01244019138756e-05,
-      "loss": 0.0393,
-      "step": 2881
-    },
-    {
-      "epoch": 262.0,
-      "learning_rate": 5.010526315789474e-05,
-      "loss": 0.0381,
-      "step": 2882
-    },
-    {
-      "epoch": 262.09,
-      "learning_rate": 5.008612440191388e-05,
-      "loss": 0.0371,
-      "step": 2883
-    },
-    {
-      "epoch": 262.18,
-      "learning_rate": 5.006698564593302e-05,
-      "loss": 0.0345,
-      "step": 2884
-    },
-    {
-      "epoch": 262.27,
-      "learning_rate": 5.004784688995215e-05,
-      "loss": 0.0353,
-      "step": 2885
-    },
-    {
-      "epoch": 262.36,
-      "learning_rate": 5.002870813397129e-05,
-      "loss": 0.0375,
-      "step": 2886
-    },
-    {
-      "epoch": 262.45,
-      "learning_rate": 5.000956937799043e-05,
-      "loss": 0.0352,
-      "step": 2887
-    },
-    {
-      "epoch": 262.55,
-      "learning_rate": 4.999043062200957e-05,
-      "loss": 0.0327,
-      "step": 2888
-    },
-    {
-      "epoch": 262.64,
-      "learning_rate": 4.9971291866028714e-05,
-      "loss": 0.0385,
-      "step": 2889
-    },
-    {
-      "epoch": 262.73,
-      "learning_rate": 4.995215311004785e-05,
-      "loss": 0.037,
-      "step": 2890
-    },
-    {
-      "epoch": 262.82,
-      "learning_rate": 4.9933014354066985e-05,
-      "loss": 0.038,
-      "step": 2891
-    },
-    {
-      "epoch": 262.91,
-      "learning_rate": 4.991387559808613e-05,
-      "loss": 0.0418,
-      "step": 2892
-    },
-    {
-      "epoch": 263.0,
-      "learning_rate": 4.989473684210527e-05,
-      "loss": 0.0344,
-      "step": 2893
-    },
-    {
-      "epoch": 263.09,
-      "learning_rate": 4.98755980861244e-05,
-      "loss": 0.0357,
-      "step": 2894
-    },
-    {
-      "epoch": 263.18,
-      "learning_rate": 4.985645933014355e-05,
-      "loss": 0.0369,
-      "step": 2895
-    },
-    {
-      "epoch": 263.27,
-      "learning_rate": 4.9837320574162685e-05,
-      "loss": 0.032,
-      "step": 2896
-    },
-    {
-      "epoch": 263.36,
-      "learning_rate": 4.981818181818182e-05,
-      "loss": 0.0355,
-      "step": 2897
-    },
-    {
-      "epoch": 263.45,
-      "learning_rate": 4.979904306220096e-05,
-      "loss": 0.0346,
-      "step": 2898
-    },
-    {
-      "epoch": 263.55,
-      "learning_rate": 4.97799043062201e-05,
-      "loss": 0.0348,
-      "step": 2899
-    },
-    {
-      "epoch": 263.64,
-      "learning_rate": 4.9760765550239234e-05,
-      "loss": 0.0406,
-      "step": 2900
-    },
-    {
-      "epoch": 263.64,
-      "eval_accuracy_NAT": 0.9931443447770116,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9495981587192583,
-      "eval_accuracy_concrete tiles": 0.9789335290160827,
-      "eval_accuracy_dark bricks": 0.8421538816827682,
-      "eval_accuracy_grass": 0.9035918872349423,
-      "eval_accuracy_red bricks": 0.9176136363636364,
-      "eval_accuracy_road": 0.8733631846691665,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.42684641468861134,
-      "eval_accuracy_speedway bricks": 0.9882341511922735,
-      "eval_accuracy_steel": 0.9003693687751467,
-      "eval_accuracy_tiles": 0.11409395973154363,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980086536214577,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8846524378891495,
-      "eval_iou_concrete tiles": 0.9478654745692463,
-      "eval_iou_dark bricks": 0.8113620350799124,
-      "eval_iou_grass": 0.8350495151840105,
-      "eval_iou_red bricks": 0.8518213284984342,
-      "eval_iou_road": 0.8671046249741883,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2895155107612334,
-      "eval_iou_speedway bricks": 0.9761103276955224,
-      "eval_iou_steel": 0.8426364604392043,
-      "eval_iou_tiles": 0.11404709748083242,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13875362277030945,
-      "eval_mean_accuracy": 0.6836878859115725,
-      "eval_mean_iou": 0.64617318067587,
-      "eval_overall_accuracy": 0.9685511059231229,
-      "eval_runtime": 3.5529,
-      "eval_samples_per_second": 10.133,
-      "eval_steps_per_second": 0.563,
-      "step": 2900
-    },
-    {
-      "epoch": 263.73,
-      "learning_rate": 4.974162679425838e-05,
-      "loss": 0.033,
-      "step": 2901
-    },
-    {
-      "epoch": 263.82,
-      "learning_rate": 4.972248803827752e-05,
-      "loss": 0.0348,
-      "step": 2902
-    },
-    {
-      "epoch": 263.91,
-      "learning_rate": 4.970334928229665e-05,
-      "loss": 0.0367,
-      "step": 2903
-    },
-    {
-      "epoch": 264.0,
-      "learning_rate": 4.9684210526315796e-05,
-      "loss": 0.0368,
-      "step": 2904
-    },
-    {
-      "epoch": 264.09,
-      "learning_rate": 4.9665071770334934e-05,
-      "loss": 0.0338,
-      "step": 2905
-    },
-    {
-      "epoch": 264.18,
-      "learning_rate": 4.9645933014354066e-05,
-      "loss": 0.0328,
-      "step": 2906
-    },
-    {
-      "epoch": 264.27,
-      "learning_rate": 4.962679425837321e-05,
-      "loss": 0.0381,
-      "step": 2907
-    },
-    {
-      "epoch": 264.36,
-      "learning_rate": 4.9607655502392344e-05,
-      "loss": 0.0368,
-      "step": 2908
-    },
-    {
-      "epoch": 264.45,
-      "learning_rate": 4.958851674641148e-05,
-      "loss": 0.0344,
-      "step": 2909
-    },
-    {
-      "epoch": 264.55,
-      "learning_rate": 4.956937799043063e-05,
-      "loss": 0.0358,
-      "step": 2910
-    },
-    {
-      "epoch": 264.64,
-      "learning_rate": 4.955023923444976e-05,
-      "loss": 0.0377,
-      "step": 2911
-    },
-    {
-      "epoch": 264.73,
-      "learning_rate": 4.95311004784689e-05,
-      "loss": 0.035,
-      "step": 2912
-    },
-    {
-      "epoch": 264.82,
-      "learning_rate": 4.9511961722488045e-05,
-      "loss": 0.0346,
-      "step": 2913
-    },
-    {
-      "epoch": 264.91,
-      "learning_rate": 4.9492822966507177e-05,
-      "loss": 0.0314,
-      "step": 2914
-    },
-    {
-      "epoch": 265.0,
-      "learning_rate": 4.9473684210526315e-05,
-      "loss": 0.039,
-      "step": 2915
-    },
-    {
-      "epoch": 265.09,
-      "learning_rate": 4.945454545454546e-05,
-      "loss": 0.0339,
-      "step": 2916
-    },
-    {
-      "epoch": 265.18,
-      "learning_rate": 4.943540669856459e-05,
-      "loss": 0.0374,
-      "step": 2917
-    },
-    {
-      "epoch": 265.27,
-      "learning_rate": 4.941626794258373e-05,
-      "loss": 0.0383,
-      "step": 2918
-    },
-    {
-      "epoch": 265.36,
-      "learning_rate": 4.939712918660288e-05,
-      "loss": 0.0307,
-      "step": 2919
-    },
-    {
-      "epoch": 265.45,
-      "learning_rate": 4.937799043062201e-05,
-      "loss": 0.0333,
-      "step": 2920
-    },
-    {
-      "epoch": 265.45,
-      "eval_accuracy_NAT": 0.9935699448696133,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9488349821880767,
-      "eval_accuracy_concrete tiles": 0.9863327829756008,
-      "eval_accuracy_dark bricks": 0.8563950579269466,
-      "eval_accuracy_grass": 0.9013200893340836,
-      "eval_accuracy_red bricks": 0.9224076704545454,
-      "eval_accuracy_road": 0.8689290158128241,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4075758828880986,
-      "eval_accuracy_speedway bricks": 0.9876428115424672,
-      "eval_accuracy_steel": 0.9088945320229537,
-      "eval_accuracy_tiles": 0.07526366251198466,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799512778512969,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8830421985275577,
-      "eval_iou_concrete tiles": 0.9399749409947842,
-      "eval_iou_dark bricks": 0.8099155503785672,
-      "eval_iou_grass": 0.8373776240806585,
-      "eval_iou_red bricks": 0.8575437438098382,
-      "eval_iou_road": 0.8637713736214202,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2764087798676289,
-      "eval_iou_speedway bricks": 0.976194750487419,
-      "eval_iou_steel": 0.843184947223497,
-      "eval_iou_tiles": 0.07522759942501198,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14231668412685394,
-      "eval_mean_accuracy": 0.6813204948097842,
-      "eval_mean_iou": 0.6417379066359754,
-      "eval_overall_accuracy": 0.9682282341851128,
-      "eval_runtime": 3.8314,
-      "eval_samples_per_second": 9.396,
-      "eval_steps_per_second": 0.522,
-      "step": 2920
-    },
-    {
-      "epoch": 265.55,
-      "learning_rate": 4.935885167464115e-05,
-      "loss": 0.0375,
-      "step": 2921
-    },
-    {
-      "epoch": 265.64,
-      "learning_rate": 4.9339712918660294e-05,
-      "loss": 0.0396,
-      "step": 2922
-    },
-    {
-      "epoch": 265.73,
-      "learning_rate": 4.9320574162679426e-05,
-      "loss": 0.0344,
-      "step": 2923
-    },
-    {
-      "epoch": 265.82,
-      "learning_rate": 4.9301435406698564e-05,
-      "loss": 0.0389,
-      "step": 2924
-    },
-    {
-      "epoch": 265.91,
-      "learning_rate": 4.928229665071771e-05,
-      "loss": 0.0386,
-      "step": 2925
-    },
-    {
-      "epoch": 266.0,
-      "learning_rate": 4.926315789473684e-05,
-      "loss": 0.0399,
-      "step": 2926
-    },
-    {
-      "epoch": 266.09,
-      "learning_rate": 4.924401913875598e-05,
-      "loss": 0.0297,
-      "step": 2927
-    },
-    {
-      "epoch": 266.18,
-      "learning_rate": 4.9224880382775126e-05,
-      "loss": 0.0353,
-      "step": 2928
-    },
-    {
-      "epoch": 266.27,
-      "learning_rate": 4.920574162679426e-05,
-      "loss": 0.0345,
-      "step": 2929
-    },
-    {
-      "epoch": 266.36,
-      "learning_rate": 4.91866028708134e-05,
-      "loss": 0.0321,
-      "step": 2930
-    },
-    {
-      "epoch": 266.45,
-      "learning_rate": 4.916746411483254e-05,
-      "loss": 0.0358,
-      "step": 2931
-    },
-    {
-      "epoch": 266.55,
-      "learning_rate": 4.9148325358851675e-05,
-      "loss": 0.0345,
-      "step": 2932
-    },
-    {
-      "epoch": 266.64,
-      "learning_rate": 4.912918660287081e-05,
-      "loss": 0.0389,
-      "step": 2933
-    },
-    {
-      "epoch": 266.73,
-      "learning_rate": 4.911004784688996e-05,
-      "loss": 0.0392,
-      "step": 2934
-    },
-    {
-      "epoch": 266.82,
-      "learning_rate": 4.909090909090909e-05,
-      "loss": 0.0361,
-      "step": 2935
-    },
-    {
-      "epoch": 266.91,
-      "learning_rate": 4.907177033492823e-05,
-      "loss": 0.0363,
-      "step": 2936
-    },
-    {
-      "epoch": 267.0,
-      "learning_rate": 4.9052631578947375e-05,
-      "loss": 0.0341,
-      "step": 2937
-    },
-    {
-      "epoch": 267.09,
-      "learning_rate": 4.903349282296651e-05,
-      "loss": 0.0366,
-      "step": 2938
-    },
-    {
-      "epoch": 267.18,
-      "learning_rate": 4.9014354066985646e-05,
-      "loss": 0.0322,
-      "step": 2939
-    },
-    {
-      "epoch": 267.27,
-      "learning_rate": 4.899521531100479e-05,
-      "loss": 0.0371,
-      "step": 2940
-    },
-    {
-      "epoch": 267.27,
-      "eval_accuracy_NAT": 0.9929350737530376,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9499643225031117,
-      "eval_accuracy_concrete tiles": 0.9827248822845961,
-      "eval_accuracy_dark bricks": 0.8675955505946653,
-      "eval_accuracy_grass": 0.907352586529287,
-      "eval_accuracy_red bricks": 0.9064275568181818,
-      "eval_accuracy_road": 0.8840610553733009,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.45369834312022717,
-      "eval_accuracy_speedway bricks": 0.988847213523093,
-      "eval_accuracy_steel": 0.8951421410197217,
-      "eval_accuracy_tiles": 0.06321051910697165,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980411810199274,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8832553081640848,
-      "eval_iou_concrete tiles": 0.939410767522067,
-      "eval_iou_dark bricks": 0.8169693015838498,
-      "eval_iou_grass": 0.8421727042992788,
-      "eval_iou_red bricks": 0.8548225050234427,
-      "eval_iou_road": 0.8775204014638402,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3021351594484936,
-      "eval_iou_speedway bricks": 0.9764161651333182,
-      "eval_iou_steel": 0.8445872358963189,
-      "eval_iou_tiles": 0.06317590691307323,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1303626447916031,
-      "eval_mean_accuracy": 0.6839968649712456,
-      "eval_mean_iou": 0.644682866588234,
-      "eval_overall_accuracy": 0.9696473015679253,
-      "eval_runtime": 4.0474,
-      "eval_samples_per_second": 8.895,
-      "eval_steps_per_second": 0.494,
-      "step": 2940
-    },
-    {
-      "epoch": 267.36,
-      "learning_rate": 4.8976076555023924e-05,
-      "loss": 0.0372,
-      "step": 2941
-    },
-    {
-      "epoch": 267.45,
-      "learning_rate": 4.895693779904306e-05,
-      "loss": 0.035,
-      "step": 2942
-    },
-    {
-      "epoch": 267.55,
-      "learning_rate": 4.893779904306221e-05,
-      "loss": 0.0407,
-      "step": 2943
-    },
-    {
-      "epoch": 267.64,
-      "learning_rate": 4.891866028708134e-05,
-      "loss": 0.0368,
-      "step": 2944
-    },
-    {
-      "epoch": 267.73,
-      "learning_rate": 4.889952153110048e-05,
-      "loss": 0.0355,
-      "step": 2945
-    },
-    {
-      "epoch": 267.82,
-      "learning_rate": 4.888038277511962e-05,
-      "loss": 0.0364,
-      "step": 2946
-    },
-    {
-      "epoch": 267.91,
-      "learning_rate": 4.8861244019138756e-05,
-      "loss": 0.0366,
-      "step": 2947
-    },
-    {
-      "epoch": 268.0,
-      "learning_rate": 4.8842105263157895e-05,
-      "loss": 0.0362,
-      "step": 2948
-    },
-    {
-      "epoch": 268.09,
-      "learning_rate": 4.8822966507177034e-05,
-      "loss": 0.0336,
-      "step": 2949
-    },
-    {
-      "epoch": 268.18,
-      "learning_rate": 4.880382775119617e-05,
-      "loss": 0.0328,
-      "step": 2950
-    },
-    {
-      "epoch": 268.27,
-      "learning_rate": 4.878468899521531e-05,
-      "loss": 0.0353,
-      "step": 2951
-    },
-    {
-      "epoch": 268.36,
-      "learning_rate": 4.876555023923445e-05,
-      "loss": 0.0332,
-      "step": 2952
-    },
-    {
-      "epoch": 268.45,
-      "learning_rate": 4.874641148325359e-05,
-      "loss": 0.0364,
-      "step": 2953
-    },
-    {
-      "epoch": 268.55,
-      "learning_rate": 4.872727272727273e-05,
-      "loss": 0.0351,
-      "step": 2954
-    },
-    {
-      "epoch": 268.64,
-      "learning_rate": 4.8708133971291866e-05,
-      "loss": 0.0398,
-      "step": 2955
-    },
-    {
-      "epoch": 268.73,
-      "learning_rate": 4.8688995215311005e-05,
-      "loss": 0.0424,
-      "step": 2956
-    },
-    {
-      "epoch": 268.82,
-      "learning_rate": 4.8669856459330144e-05,
-      "loss": 0.035,
-      "step": 2957
-    },
-    {
-      "epoch": 268.91,
-      "learning_rate": 4.865071770334928e-05,
-      "loss": 0.0314,
-      "step": 2958
-    },
-    {
-      "epoch": 269.0,
-      "learning_rate": 4.863157894736842e-05,
-      "loss": 0.041,
-      "step": 2959
-    },
-    {
-      "epoch": 269.09,
-      "learning_rate": 4.861244019138757e-05,
-      "loss": 0.0352,
-      "step": 2960
-    },
-    {
-      "epoch": 269.09,
-      "eval_accuracy_NAT": 0.9930307102580915,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9512358363019872,
-      "eval_accuracy_concrete tiles": 0.9828166085733504,
-      "eval_accuracy_dark bricks": 0.8505061391016512,
-      "eval_accuracy_grass": 0.9078314116253141,
-      "eval_accuracy_red bricks": 0.9236505681818182,
-      "eval_accuracy_road": 0.8856439097591619,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4673724257712348,
-      "eval_accuracy_speedway bricks": 0.9871775841240993,
-      "eval_accuracy_steel": 0.9098839126706681,
-      "eval_accuracy_tiles": 0.07231886043007807,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803620817391853,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8852323761355438,
-      "eval_iou_concrete tiles": 0.9449392950583532,
-      "eval_iou_dark bricks": 0.8169550428867199,
-      "eval_iou_grass": 0.8421608933041099,
-      "eval_iou_red bricks": 0.860545905707196,
-      "eval_iou_road": 0.8788331117499965,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3128783404695113,
-      "eval_iou_speedway bricks": 0.9766319294582153,
-      "eval_iou_steel": 0.8422795997649267,
-      "eval_iou_tiles": 0.0722495894909688,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12793147563934326,
-      "eval_mean_accuracy": 0.687035997445958,
-      "eval_mean_iou": 0.6471590896742098,
-      "eval_overall_accuracy": 0.9698848724365234,
-      "eval_runtime": 4.314,
-      "eval_samples_per_second": 8.345,
-      "eval_steps_per_second": 0.464,
-      "step": 2960
-    },
-    {
-      "epoch": 269.18,
-      "learning_rate": 4.85933014354067e-05,
-      "loss": 0.0382,
-      "step": 2961
-    },
-    {
-      "epoch": 269.27,
-      "learning_rate": 4.857416267942584e-05,
-      "loss": 0.0372,
-      "step": 2962
-    },
-    {
-      "epoch": 269.36,
-      "learning_rate": 4.8555023923444983e-05,
-      "loss": 0.0339,
-      "step": 2963
-    },
-    {
-      "epoch": 269.45,
-      "learning_rate": 4.8535885167464115e-05,
-      "loss": 0.0396,
-      "step": 2964
-    },
-    {
-      "epoch": 269.55,
-      "learning_rate": 4.8516746411483254e-05,
-      "loss": 0.0314,
-      "step": 2965
-    },
-    {
-      "epoch": 269.64,
-      "learning_rate": 4.84976076555024e-05,
-      "loss": 0.0354,
-      "step": 2966
-    },
-    {
-      "epoch": 269.73,
-      "learning_rate": 4.847846889952153e-05,
-      "loss": 0.0359,
-      "step": 2967
-    },
-    {
-      "epoch": 269.82,
-      "learning_rate": 4.845933014354067e-05,
-      "loss": 0.0333,
-      "step": 2968
-    },
-    {
-      "epoch": 269.91,
-      "learning_rate": 4.8440191387559816e-05,
-      "loss": 0.0344,
-      "step": 2969
-    },
-    {
-      "epoch": 270.0,
-      "learning_rate": 4.842105263157895e-05,
-      "loss": 0.0373,
-      "step": 2970
-    },
-    {
-      "epoch": 270.09,
-      "learning_rate": 4.840191387559809e-05,
-      "loss": 0.0345,
-      "step": 2971
-    },
-    {
-      "epoch": 270.18,
-      "learning_rate": 4.838277511961723e-05,
-      "loss": 0.0343,
-      "step": 2972
-    },
-    {
-      "epoch": 270.27,
-      "learning_rate": 4.8363636363636364e-05,
-      "loss": 0.0341,
-      "step": 2973
-    },
-    {
-      "epoch": 270.36,
-      "learning_rate": 4.83444976076555e-05,
-      "loss": 0.037,
-      "step": 2974
-    },
-    {
-      "epoch": 270.45,
-      "learning_rate": 4.832535885167465e-05,
-      "loss": 0.0341,
-      "step": 2975
-    },
-    {
-      "epoch": 270.55,
-      "learning_rate": 4.830622009569378e-05,
-      "loss": 0.0327,
-      "step": 2976
-    },
-    {
-      "epoch": 270.64,
-      "learning_rate": 4.828708133971292e-05,
-      "loss": 0.0387,
-      "step": 2977
-    },
-    {
-      "epoch": 270.73,
-      "learning_rate": 4.8267942583732065e-05,
-      "loss": 0.035,
-      "step": 2978
-    },
-    {
-      "epoch": 270.82,
-      "learning_rate": 4.82488038277512e-05,
-      "loss": 0.0373,
-      "step": 2979
-    },
-    {
-      "epoch": 270.91,
-      "learning_rate": 4.8229665071770336e-05,
-      "loss": 0.0334,
-      "step": 2980
-    },
-    {
-      "epoch": 270.91,
-      "eval_accuracy_NAT": 0.9927540349076772,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9533711210781579,
-      "eval_accuracy_concrete tiles": 0.9826637314254265,
-      "eval_accuracy_dark bricks": 0.8471190485354683,
-      "eval_accuracy_grass": 0.9089113893966454,
-      "eval_accuracy_red bricks": 0.8959517045454546,
-      "eval_accuracy_road": 0.9138982837698779,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48066054641193173,
-      "eval_accuracy_speedway bricks": 0.9866574465953921,
-      "eval_accuracy_steel": 0.9112030868676209,
-      "eval_accuracy_tiles": 0.07320914943158471,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9802328839009256,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8832089125691479,
-      "eval_iou_concrete tiles": 0.9480810643381811,
-      "eval_iou_dark bricks": 0.8157524091919941,
-      "eval_iou_grass": 0.8407258455589968,
-      "eval_iou_red bricks": 0.842685370741483,
-      "eval_iou_road": 0.9075399174572907,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31931649603487117,
-      "eval_iou_speedway bricks": 0.976523079477298,
-      "eval_iou_steel": 0.8438807611251947,
-      "eval_iou_tiles": 0.07313902572523262,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11521327495574951,
-      "eval_mean_accuracy": 0.6881845802280953,
-      "eval_mean_iou": 0.648545058932355,
-      "eval_overall_accuracy": 0.972347789340549,
-      "eval_runtime": 3.406,
-      "eval_samples_per_second": 10.57,
-      "eval_steps_per_second": 0.587,
-      "step": 2980
-    },
-    {
-      "epoch": 271.0,
-      "learning_rate": 4.821052631578948e-05,
-      "loss": 0.0327,
-      "step": 2981
-    },
-    {
-      "epoch": 271.09,
-      "learning_rate": 4.8191387559808613e-05,
-      "loss": 0.0366,
-      "step": 2982
-    },
-    {
-      "epoch": 271.18,
-      "learning_rate": 4.817224880382775e-05,
-      "loss": 0.0344,
-      "step": 2983
-    },
-    {
-      "epoch": 271.27,
-      "learning_rate": 4.81531100478469e-05,
-      "loss": 0.0331,
-      "step": 2984
-    },
-    {
-      "epoch": 271.36,
-      "learning_rate": 4.813397129186603e-05,
-      "loss": 0.0316,
-      "step": 2985
-    },
-    {
-      "epoch": 271.45,
-      "learning_rate": 4.811483253588517e-05,
-      "loss": 0.0408,
-      "step": 2986
-    },
-    {
-      "epoch": 271.55,
-      "learning_rate": 4.809569377990431e-05,
-      "loss": 0.0334,
-      "step": 2987
-    },
-    {
-      "epoch": 271.64,
-      "learning_rate": 4.8076555023923446e-05,
-      "loss": 0.0408,
-      "step": 2988
-    },
-    {
-      "epoch": 271.73,
-      "learning_rate": 4.8057416267942585e-05,
-      "loss": 0.0346,
-      "step": 2989
-    },
-    {
-      "epoch": 271.82,
-      "learning_rate": 4.8038277511961724e-05,
-      "loss": 0.0354,
-      "step": 2990
-    },
-    {
-      "epoch": 271.91,
-      "learning_rate": 4.801913875598086e-05,
-      "loss": 0.0353,
-      "step": 2991
-    },
-    {
-      "epoch": 272.0,
-      "learning_rate": 4.8e-05,
-      "loss": 0.0334,
-      "step": 2992
-    },
-    {
-      "epoch": 272.09,
-      "learning_rate": 4.798086124401914e-05,
-      "loss": 0.0343,
-      "step": 2993
-    },
-    {
-      "epoch": 272.18,
-      "learning_rate": 4.796172248803828e-05,
-      "loss": 0.0345,
-      "step": 2994
-    },
-    {
-      "epoch": 272.27,
-      "learning_rate": 4.794258373205742e-05,
-      "loss": 0.0327,
-      "step": 2995
-    },
-    {
-      "epoch": 272.36,
-      "learning_rate": 4.7923444976076556e-05,
-      "loss": 0.0373,
-      "step": 2996
-    },
-    {
-      "epoch": 272.45,
-      "learning_rate": 4.7904306220095695e-05,
-      "loss": 0.0335,
-      "step": 2997
-    },
-    {
-      "epoch": 272.55,
-      "learning_rate": 4.7885167464114834e-05,
-      "loss": 0.0378,
-      "step": 2998
-    },
-    {
-      "epoch": 272.64,
-      "learning_rate": 4.786602870813397e-05,
-      "loss": 0.0368,
-      "step": 2999
-    },
-    {
-      "epoch": 272.73,
-      "learning_rate": 4.784688995215311e-05,
-      "loss": 0.0331,
-      "step": 3000
-    },
-    {
-      "epoch": 272.73,
-      "eval_accuracy_NAT": 0.9938067422677356,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9505813017726082,
-      "eval_accuracy_concrete tiles": 0.9864245092643552,
-      "eval_accuracy_dark bricks": 0.8774873946345406,
-      "eval_accuracy_grass": 0.881174484564007,
-      "eval_accuracy_red bricks": 0.9305752840909091,
-      "eval_accuracy_road": 0.9348333166059443,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.3710473354836931,
-      "eval_accuracy_speedway bricks": 0.9862507497341989,
-      "eval_accuracy_steel": 0.9114916562232043,
-      "eval_accuracy_tiles": 0.07067524996575812,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9796927710822416,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.882351183547243,
-      "eval_iou_concrete tiles": 0.9417637271214643,
-      "eval_iou_dark bricks": 0.8227948606900534,
-      "eval_iou_grass": 0.8242145869430187,
-      "eval_iou_red bricks": 0.8556734693877551,
-      "eval_iou_road": 0.9269301878331683,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2542504155961916,
-      "eval_iou_speedway bricks": 0.9764144546817137,
-      "eval_iou_steel": 0.8466239345693478,
-      "eval_iou_tiles": 0.070641385447327,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10943185538053513,
-      "eval_mean_accuracy": 0.6841806172774579,
-      "eval_mean_iou": 0.644719305915348,
-      "eval_overall_accuracy": 0.973377545674642,
-      "eval_runtime": 3.8975,
-      "eval_samples_per_second": 9.237,
-      "eval_steps_per_second": 0.513,
-      "step": 3000
-    },
-    {
-      "epoch": 272.82,
-      "learning_rate": 4.782775119617225e-05,
-      "loss": 0.0418,
-      "step": 3001
-    },
-    {
-      "epoch": 272.91,
-      "learning_rate": 4.780861244019139e-05,
-      "loss": 0.0358,
-      "step": 3002
-    },
-    {
-      "epoch": 273.0,
-      "learning_rate": 4.778947368421053e-05,
-      "loss": 0.0347,
-      "step": 3003
-    },
-    {
-      "epoch": 273.09,
-      "learning_rate": 4.7770334928229667e-05,
-      "loss": 0.0336,
-      "step": 3004
-    },
-    {
-      "epoch": 273.18,
-      "learning_rate": 4.7751196172248805e-05,
-      "loss": 0.0354,
-      "step": 3005
-    },
-    {
-      "epoch": 273.27,
-      "learning_rate": 4.7732057416267944e-05,
-      "loss": 0.0346,
-      "step": 3006
-    },
-    {
-      "epoch": 273.36,
-      "learning_rate": 4.771291866028708e-05,
-      "loss": 0.038,
-      "step": 3007
-    },
-    {
-      "epoch": 273.45,
-      "learning_rate": 4.769377990430622e-05,
-      "loss": 0.0359,
-      "step": 3008
-    },
-    {
-      "epoch": 273.55,
-      "learning_rate": 4.767464114832536e-05,
-      "loss": 0.0349,
-      "step": 3009
-    },
-    {
-      "epoch": 273.64,
-      "learning_rate": 4.76555023923445e-05,
-      "loss": 0.0398,
-      "step": 3010
-    },
-    {
-      "epoch": 273.73,
-      "learning_rate": 4.763636363636364e-05,
-      "loss": 0.0318,
-      "step": 3011
-    },
-    {
-      "epoch": 273.82,
-      "learning_rate": 4.761722488038278e-05,
-      "loss": 0.0338,
-      "step": 3012
-    },
-    {
-      "epoch": 273.91,
-      "learning_rate": 4.7598086124401916e-05,
-      "loss": 0.0355,
-      "step": 3013
-    },
-    {
-      "epoch": 274.0,
-      "learning_rate": 4.7578947368421054e-05,
-      "loss": 0.0376,
-      "step": 3014
-    },
-    {
-      "epoch": 274.09,
-      "learning_rate": 4.755980861244019e-05,
-      "loss": 0.0351,
-      "step": 3015
-    },
-    {
-      "epoch": 274.18,
-      "learning_rate": 4.754066985645933e-05,
-      "loss": 0.0449,
-      "step": 3016
-    },
-    {
-      "epoch": 274.27,
-      "learning_rate": 4.752153110047847e-05,
-      "loss": 0.0355,
-      "step": 3017
-    },
-    {
-      "epoch": 274.36,
-      "learning_rate": 4.750239234449761e-05,
-      "loss": 0.0355,
-      "step": 3018
-    },
-    {
-      "epoch": 274.45,
-      "learning_rate": 4.748325358851675e-05,
-      "loss": 0.0341,
-      "step": 3019
-    },
-    {
-      "epoch": 274.55,
-      "learning_rate": 4.746411483253589e-05,
-      "loss": 0.0397,
-      "step": 3020
-    },
-    {
-      "epoch": 274.55,
-      "eval_accuracy_NAT": 0.9913193108807522,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9530693377398172,
-      "eval_accuracy_concrete tiles": 0.9741943374304409,
-      "eval_accuracy_dark bricks": 0.8560871406027482,
-      "eval_accuracy_grass": 0.9212664749036233,
-      "eval_accuracy_red bricks": 0.9380326704545454,
-      "eval_accuracy_road": 0.8730149325201001,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48837978661814574,
-      "eval_accuracy_speedway bricks": 0.9886698116281512,
-      "eval_accuracy_steel": 0.9014329529714399,
-      "eval_accuracy_tiles": 0.05793726886727846,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980070780904891,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8829995576166972,
-      "eval_iou_concrete tiles": 0.9505086363771964,
-      "eval_iou_dark bricks": 0.8194076038903625,
-      "eval_iou_grass": 0.8370966000393794,
-      "eval_iou_red bricks": 0.8477214377406932,
-      "eval_iou_road": 0.8663854604801305,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3220382119289571,
-      "eval_iou_speedway bricks": 0.9766384330649055,
-      "eval_iou_steel": 0.8445441764896723,
-      "eval_iou_tiles": 0.0579333013764295,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.14592590928077698,
-      "eval_mean_accuracy": 0.6879541557397725,
-      "eval_mean_iou": 0.6450264769161012,
-      "eval_overall_accuracy": 0.9685086144341363,
-      "eval_runtime": 3.9233,
-      "eval_samples_per_second": 9.176,
-      "eval_steps_per_second": 0.51,
-      "step": 3020
-    },
-    {
-      "epoch": 274.64,
-      "learning_rate": 4.7444976076555026e-05,
-      "loss": 0.0333,
-      "step": 3021
-    },
-    {
-      "epoch": 274.73,
-      "learning_rate": 4.7425837320574165e-05,
-      "loss": 0.035,
-      "step": 3022
-    },
-    {
-      "epoch": 274.82,
-      "learning_rate": 4.74066985645933e-05,
-      "loss": 0.033,
-      "step": 3023
-    },
-    {
-      "epoch": 274.91,
-      "learning_rate": 4.738755980861244e-05,
-      "loss": 0.0357,
-      "step": 3024
-    },
-    {
-      "epoch": 275.0,
-      "learning_rate": 4.736842105263158e-05,
-      "loss": 0.0368,
-      "step": 3025
-    },
-    {
-      "epoch": 275.09,
-      "learning_rate": 4.734928229665072e-05,
-      "loss": 0.0363,
-      "step": 3026
-    },
-    {
-      "epoch": 275.18,
-      "learning_rate": 4.733014354066986e-05,
-      "loss": 0.0323,
-      "step": 3027
-    },
-    {
-      "epoch": 275.27,
-      "learning_rate": 4.7311004784689e-05,
-      "loss": 0.033,
-      "step": 3028
-    },
-    {
-      "epoch": 275.36,
-      "learning_rate": 4.7291866028708136e-05,
-      "loss": 0.0327,
-      "step": 3029
-    },
-    {
-      "epoch": 275.45,
-      "learning_rate": 4.7272727272727275e-05,
-      "loss": 0.0379,
-      "step": 3030
-    },
-    {
-      "epoch": 275.55,
-      "learning_rate": 4.7253588516746414e-05,
-      "loss": 0.0341,
-      "step": 3031
-    },
-    {
-      "epoch": 275.64,
-      "learning_rate": 4.723444976076555e-05,
-      "loss": 0.0366,
-      "step": 3032
-    },
-    {
-      "epoch": 275.73,
-      "learning_rate": 4.721531100478469e-05,
-      "loss": 0.0313,
-      "step": 3033
-    },
-    {
-      "epoch": 275.82,
-      "learning_rate": 4.719617224880383e-05,
-      "loss": 0.0371,
-      "step": 3034
-    },
-    {
-      "epoch": 275.91,
-      "learning_rate": 4.717703349282297e-05,
-      "loss": 0.0377,
-      "step": 3035
-    },
-    {
-      "epoch": 276.0,
-      "learning_rate": 4.715789473684211e-05,
-      "loss": 0.0328,
-      "step": 3036
-    },
-    {
-      "epoch": 276.09,
-      "learning_rate": 4.7138755980861246e-05,
-      "loss": 0.0374,
-      "step": 3037
-    },
-    {
-      "epoch": 276.18,
-      "learning_rate": 4.7119617224880385e-05,
-      "loss": 0.0383,
-      "step": 3038
-    },
-    {
-      "epoch": 276.27,
-      "learning_rate": 4.7100478468899524e-05,
-      "loss": 0.0336,
-      "step": 3039
-    },
-    {
-      "epoch": 276.36,
-      "learning_rate": 4.708133971291866e-05,
-      "loss": 0.0376,
-      "step": 3040
-    },
-    {
-      "epoch": 276.36,
-      "eval_accuracy_NAT": 0.9931949862474,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9506698248851883,
-      "eval_accuracy_concrete tiles": 0.9711673699015471,
-      "eval_accuracy_dark bricks": 0.8650552326700282,
-      "eval_accuracy_grass": 0.8959446659932825,
-      "eval_accuracy_red bricks": 0.9225852272727273,
-      "eval_accuracy_road": 0.9020818948786378,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4304027789264742,
-      "eval_accuracy_speedway bricks": 0.9868982063099561,
-      "eval_accuracy_steel": 0.9154244442978695,
-      "eval_accuracy_tiles": 0.08567319545267772,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9795535009474061,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8851940013687749,
-      "eval_iou_concrete tiles": 0.9473009245451834,
-      "eval_iou_dark bricks": 0.8257099819978692,
-      "eval_iou_grass": 0.8328324653426424,
-      "eval_iou_red bricks": 0.8581337737407102,
-      "eval_iou_road": 0.8954734782556506,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2909212880143113,
-      "eval_iou_speedway bricks": 0.97643655858771,
-      "eval_iou_steel": 0.8430779978131454,
-      "eval_iou_tiles": 0.08550922761449077,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12555526196956635,
-      "eval_mean_accuracy": 0.6860844482181375,
-      "eval_mean_iou": 0.6477033229406073,
-      "eval_overall_accuracy": 0.9709186553955078,
-      "eval_runtime": 3.7938,
-      "eval_samples_per_second": 9.489,
-      "eval_steps_per_second": 0.527,
-      "step": 3040
-    },
-    {
-      "epoch": 276.45,
-      "learning_rate": 4.70622009569378e-05,
-      "loss": 0.0331,
-      "step": 3041
-    },
-    {
-      "epoch": 276.55,
-      "learning_rate": 4.704306220095694e-05,
-      "loss": 0.0377,
-      "step": 3042
-    },
-    {
-      "epoch": 276.64,
-      "learning_rate": 4.702392344497608e-05,
-      "loss": 0.0325,
-      "step": 3043
-    },
-    {
-      "epoch": 276.73,
-      "learning_rate": 4.700478468899522e-05,
-      "loss": 0.0337,
-      "step": 3044
-    },
-    {
-      "epoch": 276.82,
-      "learning_rate": 4.6985645933014356e-05,
-      "loss": 0.0352,
-      "step": 3045
-    },
-    {
-      "epoch": 276.91,
-      "learning_rate": 4.6966507177033495e-05,
-      "loss": 0.0377,
-      "step": 3046
-    },
-    {
-      "epoch": 277.0,
-      "learning_rate": 4.6947368421052634e-05,
-      "loss": 0.0367,
-      "step": 3047
-    },
-    {
-      "epoch": 277.09,
-      "learning_rate": 4.692822966507177e-05,
-      "loss": 0.0381,
-      "step": 3048
-    },
-    {
-      "epoch": 277.18,
-      "learning_rate": 4.690909090909091e-05,
-      "loss": 0.0346,
-      "step": 3049
-    },
-    {
-      "epoch": 277.27,
-      "learning_rate": 4.688995215311005e-05,
-      "loss": 0.0313,
-      "step": 3050
-    },
-    {
-      "epoch": 277.36,
-      "learning_rate": 4.687081339712919e-05,
-      "loss": 0.0318,
-      "step": 3051
-    },
-    {
-      "epoch": 277.45,
-      "learning_rate": 4.685167464114833e-05,
-      "loss": 0.0373,
-      "step": 3052
-    },
-    {
-      "epoch": 277.55,
-      "learning_rate": 4.683253588516747e-05,
-      "loss": 0.0565,
-      "step": 3053
-    },
-    {
-      "epoch": 277.64,
-      "learning_rate": 4.6813397129186605e-05,
-      "loss": 0.0321,
-      "step": 3054
-    },
-    {
-      "epoch": 277.73,
-      "learning_rate": 4.6794258373205744e-05,
-      "loss": 0.043,
-      "step": 3055
-    },
-    {
-      "epoch": 277.82,
-      "learning_rate": 4.677511961722488e-05,
-      "loss": 0.0324,
-      "step": 3056
-    },
-    {
-      "epoch": 277.91,
-      "learning_rate": 4.675598086124402e-05,
-      "loss": 0.0367,
-      "step": 3057
-    },
-    {
-      "epoch": 278.0,
-      "learning_rate": 4.673684210526316e-05,
-      "loss": 0.0341,
-      "step": 3058
-    },
-    {
-      "epoch": 278.09,
-      "learning_rate": 4.67177033492823e-05,
-      "loss": 0.0366,
-      "step": 3059
-    },
-    {
-      "epoch": 278.18,
-      "learning_rate": 4.669856459330144e-05,
-      "loss": 0.0355,
-      "step": 3060
-    },
-    {
-      "epoch": 278.18,
-      "eval_accuracy_NAT": 0.9924637728213052,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9472898514957724,
-      "eval_accuracy_concrete tiles": 0.9912554271387513,
-      "eval_accuracy_dark bricks": 0.8890727839575074,
-      "eval_accuracy_grass": 0.9166389973332588,
-      "eval_accuracy_red bricks": 0.9005681818181818,
-      "eval_accuracy_road": 0.9433642850492209,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.40809968847352024,
-      "eval_accuracy_speedway bricks": 0.9887657534692932,
-      "eval_accuracy_steel": 0.9083751071829035,
-      "eval_accuracy_tiles": 0.06916860704013149,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9801551321861384,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8834226217651967,
-      "eval_iou_concrete tiles": 0.9335675411063438,
-      "eval_iou_dark bricks": 0.8201604885669649,
-      "eval_iou_grass": 0.8377446001111601,
-      "eval_iou_red bricks": 0.8488702928870293,
-      "eval_iou_road": 0.9369100403034044,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2742515191937157,
-      "eval_iou_speedway bricks": 0.9762909385556741,
-      "eval_iou_steel": 0.8458148765152503,
-      "eval_iou_tiles": 0.06914493051276785,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10423687845468521,
-      "eval_mean_accuracy": 0.6888509581369112,
-      "eval_mean_iou": 0.6466409985925881,
-      "eval_overall_accuracy": 0.9747062259250217,
-      "eval_runtime": 3.8207,
-      "eval_samples_per_second": 9.422,
-      "eval_steps_per_second": 0.523,
-      "step": 3060
-    },
-    {
-      "epoch": 278.27,
-      "learning_rate": 4.667942583732058e-05,
-      "loss": 0.0384,
-      "step": 3061
-    },
-    {
-      "epoch": 278.36,
-      "learning_rate": 4.6660287081339716e-05,
-      "loss": 0.0319,
-      "step": 3062
-    },
-    {
-      "epoch": 278.45,
-      "learning_rate": 4.6641148325358854e-05,
-      "loss": 0.0315,
-      "step": 3063
-    },
-    {
-      "epoch": 278.55,
-      "learning_rate": 4.662200956937799e-05,
-      "loss": 0.0329,
-      "step": 3064
-    },
-    {
-      "epoch": 278.64,
-      "learning_rate": 4.660287081339713e-05,
-      "loss": 0.0381,
-      "step": 3065
-    },
-    {
-      "epoch": 278.73,
-      "learning_rate": 4.658373205741627e-05,
-      "loss": 0.0313,
-      "step": 3066
-    },
-    {
-      "epoch": 278.82,
-      "learning_rate": 4.656459330143541e-05,
-      "loss": 0.041,
-      "step": 3067
-    },
-    {
-      "epoch": 278.91,
-      "learning_rate": 4.654545454545455e-05,
-      "loss": 0.0325,
-      "step": 3068
-    },
-    {
-      "epoch": 279.0,
-      "learning_rate": 4.652631578947369e-05,
-      "loss": 0.033,
-      "step": 3069
-    },
-    {
-      "epoch": 279.09,
-      "learning_rate": 4.6507177033492826e-05,
-      "loss": 0.0386,
-      "step": 3070
-    },
-    {
-      "epoch": 279.18,
-      "learning_rate": 4.6488038277511965e-05,
-      "loss": 0.0302,
-      "step": 3071
-    },
-    {
-      "epoch": 279.27,
-      "learning_rate": 4.64688995215311e-05,
-      "loss": 0.0392,
-      "step": 3072
-    },
-    {
-      "epoch": 279.36,
-      "learning_rate": 4.644976076555024e-05,
-      "loss": 0.0398,
-      "step": 3073
-    },
-    {
-      "epoch": 279.45,
-      "learning_rate": 4.643062200956938e-05,
-      "loss": 0.0346,
-      "step": 3074
-    },
-    {
-      "epoch": 279.55,
-      "learning_rate": 4.641148325358852e-05,
-      "loss": 0.0345,
-      "step": 3075
-    },
-    {
-      "epoch": 279.64,
-      "learning_rate": 4.639234449760766e-05,
-      "loss": 0.0352,
-      "step": 3076
-    },
-    {
-      "epoch": 279.73,
-      "learning_rate": 4.63732057416268e-05,
-      "loss": 0.0293,
-      "step": 3077
-    },
-    {
-      "epoch": 279.82,
-      "learning_rate": 4.6354066985645936e-05,
-      "loss": 0.0386,
-      "step": 3078
-    },
-    {
-      "epoch": 279.91,
-      "learning_rate": 4.6334928229665075e-05,
-      "loss": 0.0326,
-      "step": 3079
-    },
-    {
-      "epoch": 280.0,
-      "learning_rate": 4.6315789473684214e-05,
-      "loss": 0.0385,
-      "step": 3080
-    },
-    {
-      "epoch": 280.0,
-      "eval_accuracy_NAT": 0.9919801203114291,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9463496287394308,
-      "eval_accuracy_concrete tiles": 0.9900935608145295,
-      "eval_accuracy_dark bricks": 0.8563565682614218,
-      "eval_accuracy_grass": 0.9175197559040532,
-      "eval_accuracy_red bricks": 0.9225852272727273,
-      "eval_accuracy_road": 0.9407983438675578,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5342541284150746,
-      "eval_accuracy_speedway bricks": 0.9892653751325988,
-      "eval_accuracy_steel": 0.9007403865180397,
-      "eval_accuracy_tiles": 0.18216682646212848,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9802134945146072,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8885597777492324,
-      "eval_iou_concrete tiles": 0.9319097502014504,
-      "eval_iou_dark bricks": 0.8131944444444444,
-      "eval_iou_grass": 0.8336524812560138,
-      "eval_iou_red bricks": 0.8547458463563086,
-      "eval_iou_road": 0.9323417472749758,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.35466041983126223,
-      "eval_iou_speedway bricks": 0.9766254066383392,
-      "eval_iou_steel": 0.8449984144049377,
-      "eval_iou_tiles": 0.18193010054031872,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10033272951841354,
-      "eval_mean_accuracy": 0.7055469170537685,
-      "eval_mean_iou": 0.6609870679393761,
-      "eval_overall_accuracy": 0.9747117360432943,
-      "eval_runtime": 4.1555,
-      "eval_samples_per_second": 8.663,
-      "eval_steps_per_second": 0.481,
-      "step": 3080
-    },
-    {
-      "epoch": 280.09,
-      "learning_rate": 4.629665071770335e-05,
-      "loss": 0.036,
-      "step": 3081
-    },
-    {
-      "epoch": 280.18,
-      "learning_rate": 4.627751196172249e-05,
-      "loss": 0.0474,
-      "step": 3082
-    },
-    {
-      "epoch": 280.27,
-      "learning_rate": 4.625837320574163e-05,
-      "loss": 0.0329,
-      "step": 3083
-    },
-    {
-      "epoch": 280.36,
-      "learning_rate": 4.623923444976077e-05,
-      "loss": 0.0351,
-      "step": 3084
-    },
-    {
-      "epoch": 280.45,
-      "learning_rate": 4.622009569377991e-05,
-      "loss": 0.0336,
-      "step": 3085
-    },
-    {
-      "epoch": 280.55,
-      "learning_rate": 4.6200956937799046e-05,
-      "loss": 0.0308,
-      "step": 3086
-    },
-    {
-      "epoch": 280.64,
-      "learning_rate": 4.618181818181818e-05,
-      "loss": 0.0326,
-      "step": 3087
-    },
-    {
-      "epoch": 280.73,
-      "learning_rate": 4.6162679425837324e-05,
-      "loss": 0.0357,
-      "step": 3088
-    },
-    {
-      "epoch": 280.82,
-      "learning_rate": 4.614354066985646e-05,
-      "loss": 0.0355,
-      "step": 3089
-    },
-    {
-      "epoch": 280.91,
-      "learning_rate": 4.6124401913875595e-05,
-      "loss": 0.0368,
-      "step": 3090
-    },
-    {
-      "epoch": 281.0,
-      "learning_rate": 4.610526315789474e-05,
-      "loss": 0.0365,
-      "step": 3091
-    },
-    {
-      "epoch": 281.09,
-      "learning_rate": 4.608612440191388e-05,
-      "loss": 0.0361,
-      "step": 3092
-    },
-    {
-      "epoch": 281.18,
-      "learning_rate": 4.606698564593301e-05,
-      "loss": 0.0339,
-      "step": 3093
-    },
-    {
-      "epoch": 281.27,
-      "learning_rate": 4.6047846889952156e-05,
-      "loss": 0.0336,
-      "step": 3094
-    },
-    {
-      "epoch": 281.36,
-      "learning_rate": 4.6028708133971295e-05,
-      "loss": 0.031,
-      "step": 3095
-    },
-    {
-      "epoch": 281.45,
-      "learning_rate": 4.600956937799043e-05,
-      "loss": 0.0398,
-      "step": 3096
-    },
-    {
-      "epoch": 281.55,
-      "learning_rate": 4.599043062200957e-05,
-      "loss": 0.0314,
-      "step": 3097
-    },
-    {
-      "epoch": 281.64,
-      "learning_rate": 4.597129186602871e-05,
-      "loss": 0.0346,
-      "step": 3098
-    },
-    {
-      "epoch": 281.73,
-      "learning_rate": 4.5952153110047844e-05,
-      "loss": 0.0356,
-      "step": 3099
-    },
-    {
-      "epoch": 281.82,
-      "learning_rate": 4.593301435406699e-05,
-      "loss": 0.0375,
-      "step": 3100
-    },
-    {
-      "epoch": 281.82,
-      "eval_accuracy_NAT": 0.9925885237605544,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9454791514657281,
-      "eval_accuracy_concrete tiles": 0.9864245092643552,
-      "eval_accuracy_dark bricks": 0.8594357415034063,
-      "eval_accuracy_grass": 0.9030117050018,
-      "eval_accuracy_red bricks": 0.9343039772727273,
-      "eval_accuracy_road": 0.9259347486719863,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5050864279215945,
-      "eval_accuracy_speedway bricks": 0.9886643809578978,
-      "eval_accuracy_steel": 0.9064210804036673,
-      "eval_accuracy_tiles": 0.09402821531297083,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799251950473554,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8844322045594268,
-      "eval_iou_concrete tiles": 0.9324007976648073,
-      "eval_iou_dark bricks": 0.8194128440366972,
-      "eval_iou_grass": 0.8304202771849528,
-      "eval_iou_red bricks": 0.8554706551780198,
-      "eval_iou_road": 0.9167647400387186,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3365603644646925,
-      "eval_iou_speedway bricks": 0.9766249186378364,
-      "eval_iou_steel": 0.8460478825331107,
-      "eval_iou_tiles": 0.09392529757832808,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10817250609397888,
-      "eval_mean_accuracy": 0.6954906508874376,
-      "eval_mean_iou": 0.6516911674556881,
-      "eval_overall_accuracy": 0.9729877048068576,
-      "eval_runtime": 3.9017,
-      "eval_samples_per_second": 9.227,
-      "eval_steps_per_second": 0.513,
-      "step": 3100
-    },
-    {
-      "epoch": 281.91,
-      "learning_rate": 4.591387559808613e-05,
-      "loss": 0.0291,
-      "step": 3101
-    },
-    {
-      "epoch": 282.0,
-      "learning_rate": 4.589473684210526e-05,
-      "loss": 0.0372,
-      "step": 3102
-    },
-    {
-      "epoch": 282.09,
-      "learning_rate": 4.5875598086124405e-05,
-      "loss": 0.0333,
-      "step": 3103
-    },
-    {
-      "epoch": 282.18,
-      "learning_rate": 4.5856459330143544e-05,
-      "loss": 0.0301,
-      "step": 3104
-    },
-    {
-      "epoch": 282.27,
-      "learning_rate": 4.583732057416268e-05,
-      "loss": 0.0357,
-      "step": 3105
-    },
-    {
-      "epoch": 282.36,
-      "learning_rate": 4.581818181818182e-05,
-      "loss": 0.0294,
-      "step": 3106
-    },
-    {
-      "epoch": 282.45,
-      "learning_rate": 4.579904306220096e-05,
-      "loss": 0.037,
-      "step": 3107
-    },
-    {
-      "epoch": 282.55,
-      "learning_rate": 4.57799043062201e-05,
-      "loss": 0.0312,
-      "step": 3108
-    },
-    {
-      "epoch": 282.64,
-      "learning_rate": 4.576076555023924e-05,
-      "loss": 0.0323,
-      "step": 3109
-    },
-    {
-      "epoch": 282.73,
-      "learning_rate": 4.574162679425838e-05,
-      "loss": 0.033,
-      "step": 3110
-    },
-    {
-      "epoch": 282.82,
-      "learning_rate": 4.5722488038277516e-05,
-      "loss": 0.0373,
-      "step": 3111
-    },
-    {
-      "epoch": 282.91,
-      "learning_rate": 4.5703349282296654e-05,
-      "loss": 0.0356,
-      "step": 3112
-    },
-    {
-      "epoch": 283.0,
-      "learning_rate": 4.568421052631579e-05,
-      "loss": 0.0343,
-      "step": 3113
-    },
-    {
-      "epoch": 283.09,
-      "learning_rate": 4.566507177033493e-05,
-      "loss": 0.0329,
-      "step": 3114
-    },
-    {
-      "epoch": 283.18,
-      "learning_rate": 4.564593301435407e-05,
-      "loss": 0.0356,
-      "step": 3115
-    },
-    {
-      "epoch": 283.27,
-      "learning_rate": 4.562679425837321e-05,
-      "loss": 0.0305,
-      "step": 3116
-    },
-    {
-      "epoch": 283.36,
-      "learning_rate": 4.560765550239235e-05,
-      "loss": 0.0443,
-      "step": 3117
-    },
-    {
-      "epoch": 283.45,
-      "learning_rate": 4.558851674641149e-05,
-      "loss": 0.033,
-      "step": 3118
-    },
-    {
-      "epoch": 283.55,
-      "learning_rate": 4.5569377990430626e-05,
-      "loss": 0.035,
-      "step": 3119
-    },
-    {
-      "epoch": 283.64,
-      "learning_rate": 4.5550239234449765e-05,
-      "loss": 0.0357,
-      "step": 3120
-    },
-    {
-      "epoch": 283.64,
-      "eval_accuracy_NAT": 0.9928228508430481,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9492802802695395,
-      "eval_accuracy_concrete tiles": 0.9841925029046658,
-      "eval_accuracy_dark bricks": 0.8815672991801701,
-      "eval_accuracy_grass": 0.9009076706382354,
-      "eval_accuracy_red bricks": 0.9266690340909091,
-      "eval_accuracy_road": 0.9296071159522459,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46684862018581313,
-      "eval_accuracy_speedway bricks": 0.988013907343111,
-      "eval_accuracy_steel": 0.9061902249192005,
-      "eval_accuracy_tiles": 0.08676893576222436,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799327393309462,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8842723259855644,
-      "eval_iou_concrete tiles": 0.9434056271981243,
-      "eval_iou_dark bricks": 0.8139592736060273,
-      "eval_iou_grass": 0.8306880694031318,
-      "eval_iou_red bricks": 0.8554335354859859,
-      "eval_iou_road": 0.9224593380089874,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31427351854944974,
-      "eval_iou_speedway bricks": 0.976852266151212,
-      "eval_iou_steel": 0.8448051898140675,
-      "eval_iou_tiles": 0.08670362006432629,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10754900425672531,
-      "eval_mean_accuracy": 0.6932975724683972,
-      "eval_mean_iou": 0.6502142695075249,
-      "eval_overall_accuracy": 0.9734599855211046,
-      "eval_runtime": 3.8401,
-      "eval_samples_per_second": 9.375,
-      "eval_steps_per_second": 0.521,
-      "step": 3120
-    },
-    {
-      "epoch": 283.73,
-      "learning_rate": 4.5531100478468903e-05,
-      "loss": 0.0354,
-      "step": 3121
-    },
-    {
-      "epoch": 283.82,
-      "learning_rate": 4.551196172248804e-05,
-      "loss": 0.0365,
-      "step": 3122
-    },
-    {
-      "epoch": 283.91,
-      "learning_rate": 4.549282296650718e-05,
-      "loss": 0.0354,
-      "step": 3123
-    },
-    {
-      "epoch": 284.0,
-      "learning_rate": 4.547368421052632e-05,
-      "loss": 0.0329,
-      "step": 3124
-    },
-    {
-      "epoch": 284.09,
-      "learning_rate": 4.545454545454546e-05,
-      "loss": 0.0343,
-      "step": 3125
-    },
-    {
-      "epoch": 284.18,
-      "learning_rate": 4.54354066985646e-05,
-      "loss": 0.0338,
-      "step": 3126
-    },
-    {
-      "epoch": 284.27,
-      "learning_rate": 4.5416267942583736e-05,
-      "loss": 0.0335,
-      "step": 3127
-    },
-    {
-      "epoch": 284.36,
-      "learning_rate": 4.539712918660287e-05,
-      "loss": 0.0344,
-      "step": 3128
-    },
-    {
-      "epoch": 284.45,
-      "learning_rate": 4.5377990430622014e-05,
-      "loss": 0.0325,
-      "step": 3129
-    },
-    {
-      "epoch": 284.55,
-      "learning_rate": 4.535885167464115e-05,
-      "loss": 0.0311,
-      "step": 3130
-    },
-    {
-      "epoch": 284.64,
-      "learning_rate": 4.5339712918660284e-05,
-      "loss": 0.0339,
-      "step": 3131
-    },
-    {
-      "epoch": 284.73,
-      "learning_rate": 4.532057416267943e-05,
-      "loss": 0.0437,
-      "step": 3132
-    },
-    {
-      "epoch": 284.82,
-      "learning_rate": 4.530143540669857e-05,
-      "loss": 0.0351,
-      "step": 3133
-    },
-    {
-      "epoch": 284.91,
-      "learning_rate": 4.52822966507177e-05,
-      "loss": 0.0336,
-      "step": 3134
-    },
-    {
-      "epoch": 285.0,
-      "learning_rate": 4.5263157894736846e-05,
-      "loss": 0.0336,
-      "step": 3135
-    },
-    {
-      "epoch": 285.09,
-      "learning_rate": 4.5244019138755985e-05,
-      "loss": 0.0372,
-      "step": 3136
-    },
-    {
-      "epoch": 285.18,
-      "learning_rate": 4.522488038277512e-05,
-      "loss": 0.0355,
-      "step": 3137
-    },
-    {
-      "epoch": 285.27,
-      "learning_rate": 4.520574162679426e-05,
-      "loss": 0.0342,
-      "step": 3138
-    },
-    {
-      "epoch": 285.36,
-      "learning_rate": 4.51866028708134e-05,
-      "loss": 0.0376,
-      "step": 3139
-    },
-    {
-      "epoch": 285.45,
-      "learning_rate": 4.5167464114832533e-05,
-      "loss": 0.0341,
-      "step": 3140
-    },
-    {
-      "epoch": 285.45,
-      "eval_accuracy_NAT": 0.9921101647841685,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9491649319713292,
-      "eval_accuracy_concrete tiles": 0.9902770133920381,
-      "eval_accuracy_dark bricks": 0.8668257572841692,
-      "eval_accuracy_grass": 0.915017283139415,
-      "eval_accuracy_red bricks": 0.9300426136363636,
-      "eval_accuracy_road": 0.9249153856106566,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47641496429851404,
-      "eval_accuracy_speedway bricks": 0.9883083703524023,
-      "eval_accuracy_steel": 0.9075423784710771,
-      "eval_accuracy_tiles": 0.08711135460895768,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9800954056693266,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8869007782832651,
-      "eval_iou_concrete tiles": 0.9256358959702772,
-      "eval_iou_dark bricks": 0.821784345922277,
-      "eval_iou_grass": 0.8347293208390592,
-      "eval_iou_red bricks": 0.8571428571428571,
-      "eval_iou_road": 0.9172588710741225,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32104705816783397,
-      "eval_iou_speedway bricks": 0.97673200961777,
-      "eval_iou_steel": 0.8439057301010473,
-      "eval_iou_tiles": 0.08707557502738225,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10906367003917694,
-      "eval_mean_accuracy": 0.6944407859653148,
-      "eval_mean_iou": 0.6501775267550168,
-      "eval_overall_accuracy": 0.973128424750434,
-      "eval_runtime": 3.7884,
-      "eval_samples_per_second": 9.503,
-      "eval_steps_per_second": 0.528,
-      "step": 3140
-    },
-    {
-      "epoch": 285.55,
-      "learning_rate": 4.514832535885168e-05,
-      "loss": 0.0322,
-      "step": 3141
-    },
-    {
-      "epoch": 285.64,
-      "learning_rate": 4.512918660287082e-05,
-      "loss": 0.0371,
-      "step": 3142
-    },
-    {
-      "epoch": 285.73,
-      "learning_rate": 4.511004784688995e-05,
-      "loss": 0.0399,
-      "step": 3143
-    },
-    {
-      "epoch": 285.82,
-      "learning_rate": 4.5090909090909095e-05,
-      "loss": 0.0361,
-      "step": 3144
-    },
-    {
-      "epoch": 285.91,
-      "learning_rate": 4.5071770334928234e-05,
-      "loss": 0.0334,
-      "step": 3145
-    },
-    {
-      "epoch": 286.0,
-      "learning_rate": 4.5052631578947366e-05,
-      "loss": 0.0305,
-      "step": 3146
-    },
-    {
-      "epoch": 286.09,
-      "learning_rate": 4.503349282296651e-05,
-      "loss": 0.0458,
-      "step": 3147
-    },
-    {
-      "epoch": 286.18,
-      "learning_rate": 4.501435406698565e-05,
-      "loss": 0.0401,
-      "step": 3148
-    },
-    {
-      "epoch": 286.27,
-      "learning_rate": 4.499521531100478e-05,
-      "loss": 0.0353,
-      "step": 3149
-    },
-    {
-      "epoch": 286.36,
-      "learning_rate": 4.497607655502393e-05,
-      "loss": 0.0362,
-      "step": 3150
-    },
-    {
-      "epoch": 286.45,
-      "learning_rate": 4.495693779904307e-05,
-      "loss": 0.0328,
-      "step": 3151
-    },
-    {
-      "epoch": 286.55,
-      "learning_rate": 4.49377990430622e-05,
-      "loss": 0.0341,
-      "step": 3152
-    },
-    {
-      "epoch": 286.64,
-      "learning_rate": 4.4918660287081344e-05,
-      "loss": 0.0333,
-      "step": 3153
-    },
-    {
-      "epoch": 286.73,
-      "learning_rate": 4.489952153110048e-05,
-      "loss": 0.0319,
-      "step": 3154
-    },
-    {
-      "epoch": 286.82,
-      "learning_rate": 4.4880382775119615e-05,
-      "loss": 0.0385,
-      "step": 3155
-    },
-    {
-      "epoch": 286.91,
-      "learning_rate": 4.486124401913876e-05,
-      "loss": 0.0342,
-      "step": 3156
-    },
-    {
-      "epoch": 287.0,
-      "learning_rate": 4.48421052631579e-05,
-      "loss": 0.0363,
-      "step": 3157
-    },
-    {
-      "epoch": 287.09,
-      "learning_rate": 4.482296650717703e-05,
-      "loss": 0.0345,
-      "step": 3158
-    },
-    {
-      "epoch": 287.18,
-      "learning_rate": 4.480382775119618e-05,
-      "loss": 0.0371,
-      "step": 3159
-    },
-    {
-      "epoch": 287.27,
-      "learning_rate": 4.4784688995215316e-05,
-      "loss": 0.0433,
-      "step": 3160
-    },
-    {
-      "epoch": 287.27,
-      "eval_accuracy_NAT": 0.9924717131215403,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9521465513541354,
-      "eval_accuracy_concrete tiles": 0.9842230783342506,
-      "eval_accuracy_dark bricks": 0.8641314806974327,
-      "eval_accuracy_grass": 0.9170898618397368,
-      "eval_accuracy_red bricks": 0.91015625,
-      "eval_accuracy_road": 0.9447971975375671,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46180354533675183,
-      "eval_accuracy_speedway bricks": 0.9875872980243222,
-      "eval_accuracy_steel": 0.9109474968669613,
-      "eval_accuracy_tiles": 0.13737844130940968,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9801556294114285,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8906422902337859,
-      "eval_iou_concrete tiles": 0.9473219540906416,
-      "eval_iou_dark bricks": 0.8262853777924993,
-      "eval_iou_grass": 0.8397580529019251,
-      "eval_iou_red bricks": 0.8629629629629629,
-      "eval_iou_road": 0.9380684076122648,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31175091193329857,
-      "eval_iou_speedway bricks": 0.9766338214663562,
-      "eval_iou_steel": 0.8442371171832019,
-      "eval_iou_tiles": 0.13729381972486482,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09478067606687546,
-      "eval_mean_accuracy": 0.697133301109393,
-      "eval_mean_iou": 0.6580854111779407,
-      "eval_overall_accuracy": 0.9752846823798286,
-      "eval_runtime": 4.0609,
-      "eval_samples_per_second": 8.865,
-      "eval_steps_per_second": 0.492,
-      "step": 3160
-    },
-    {
-      "epoch": 287.36,
-      "learning_rate": 4.476555023923445e-05,
-      "loss": 0.0309,
-      "step": 3161
-    },
-    {
-      "epoch": 287.45,
-      "learning_rate": 4.474641148325359e-05,
-      "loss": 0.0319,
-      "step": 3162
-    },
-    {
-      "epoch": 287.55,
-      "learning_rate": 4.472727272727273e-05,
-      "loss": 0.0318,
-      "step": 3163
-    },
-    {
-      "epoch": 287.64,
-      "learning_rate": 4.4708133971291864e-05,
-      "loss": 0.0312,
-      "step": 3164
-    },
-    {
-      "epoch": 287.73,
-      "learning_rate": 4.468899521531101e-05,
-      "loss": 0.0339,
-      "step": 3165
-    },
-    {
-      "epoch": 287.82,
-      "learning_rate": 4.466985645933014e-05,
-      "loss": 0.0421,
-      "step": 3166
-    },
-    {
-      "epoch": 287.91,
-      "learning_rate": 4.465071770334928e-05,
-      "loss": 0.0379,
-      "step": 3167
-    },
-    {
-      "epoch": 288.0,
-      "learning_rate": 4.4631578947368426e-05,
-      "loss": 0.0374,
-      "step": 3168
-    },
-    {
-      "epoch": 288.09,
-      "learning_rate": 4.461244019138756e-05,
-      "loss": 0.0314,
-      "step": 3169
-    },
-    {
-      "epoch": 288.18,
-      "learning_rate": 4.45933014354067e-05,
-      "loss": 0.033,
-      "step": 3170
-    },
-    {
-      "epoch": 288.27,
-      "learning_rate": 4.457416267942584e-05,
-      "loss": 0.0336,
-      "step": 3171
-    },
-    {
-      "epoch": 288.36,
-      "learning_rate": 4.4555023923444974e-05,
-      "loss": 0.0363,
-      "step": 3172
-    },
-    {
-      "epoch": 288.45,
-      "learning_rate": 4.453588516746411e-05,
-      "loss": 0.0335,
-      "step": 3173
-    },
-    {
-      "epoch": 288.55,
-      "learning_rate": 4.451674641148326e-05,
-      "loss": 0.0319,
-      "step": 3174
-    },
-    {
-      "epoch": 288.64,
-      "learning_rate": 4.449760765550239e-05,
-      "loss": 0.0342,
-      "step": 3175
-    },
-    {
-      "epoch": 288.73,
-      "learning_rate": 4.4478468899521536e-05,
-      "loss": 0.0346,
-      "step": 3176
-    },
-    {
-      "epoch": 288.82,
-      "learning_rate": 4.4459330143540675e-05,
-      "loss": 0.0397,
-      "step": 3177
-    },
-    {
-      "epoch": 288.91,
-      "learning_rate": 4.444019138755981e-05,
-      "loss": 0.0326,
-      "step": 3178
-    },
-    {
-      "epoch": 289.0,
-      "learning_rate": 4.442105263157895e-05,
-      "loss": 0.037,
-      "step": 3179
-    },
-    {
-      "epoch": 289.09,
-      "learning_rate": 4.440191387559809e-05,
-      "loss": 0.032,
-      "step": 3180
-    },
-    {
-      "epoch": 289.09,
-      "eval_accuracy_NAT": 0.9933331474714908,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9507154277007597,
-      "eval_accuracy_concrete tiles": 0.9872194704335596,
-      "eval_accuracy_dark bricks": 0.8715214964781957,
-      "eval_accuracy_grass": 0.9006769957744559,
-      "eval_accuracy_red bricks": 0.9362571022727273,
-      "eval_accuracy_road": 0.9380643226556161,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4890138670636562,
-      "eval_accuracy_speedway bricks": 0.9875227333890882,
-      "eval_accuracy_steel": 0.9115246355781281,
-      "eval_accuracy_tiles": 0.10916312833858376,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803276530319581,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8896127774772615,
-      "eval_iou_concrete tiles": 0.9414508980639141,
-      "eval_iou_dark bricks": 0.8247013403263403,
-      "eval_iou_grass": 0.8362370807846445,
-      "eval_iou_red bricks": 0.8707067371202114,
-      "eval_iou_road": 0.9315411790417134,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3281715416921056,
-      "eval_iou_speedway bricks": 0.9765204430267207,
-      "eval_iou_steel": 0.8439336804018259,
-      "eval_iou_tiles": 0.10907349117284795,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10123680531978607,
-      "eval_mean_accuracy": 0.6980778713197124,
-      "eval_mean_iou": 0.6563289863184264,
-      "eval_overall_accuracy": 0.9747046364678277,
-      "eval_runtime": 4.4223,
-      "eval_samples_per_second": 8.141,
-      "eval_steps_per_second": 0.452,
-      "step": 3180
-    },
-    {
-      "epoch": 289.18,
-      "learning_rate": 4.438277511961722e-05,
-      "loss": 0.0336,
-      "step": 3181
-    },
-    {
-      "epoch": 289.27,
-      "learning_rate": 4.436363636363637e-05,
-      "loss": 0.0303,
-      "step": 3182
-    },
-    {
-      "epoch": 289.36,
-      "learning_rate": 4.434449760765551e-05,
-      "loss": 0.0324,
-      "step": 3183
-    },
-    {
-      "epoch": 289.45,
-      "learning_rate": 4.432535885167464e-05,
-      "loss": 0.0328,
-      "step": 3184
-    },
-    {
-      "epoch": 289.55,
-      "learning_rate": 4.4306220095693785e-05,
-      "loss": 0.0335,
-      "step": 3185
-    },
-    {
-      "epoch": 289.64,
-      "learning_rate": 4.4287081339712924e-05,
-      "loss": 0.0369,
-      "step": 3186
-    },
-    {
-      "epoch": 289.73,
-      "learning_rate": 4.4267942583732056e-05,
-      "loss": 0.0377,
-      "step": 3187
-    },
-    {
-      "epoch": 289.82,
-      "learning_rate": 4.42488038277512e-05,
-      "loss": 0.0334,
-      "step": 3188
-    },
-    {
-      "epoch": 289.91,
-      "learning_rate": 4.422966507177034e-05,
-      "loss": 0.0286,
-      "step": 3189
-    },
-    {
-      "epoch": 290.0,
-      "learning_rate": 4.421052631578947e-05,
-      "loss": 0.0422,
-      "step": 3190
-    },
-    {
-      "epoch": 290.09,
-      "learning_rate": 4.419138755980862e-05,
-      "loss": 0.0356,
-      "step": 3191
-    },
-    {
-      "epoch": 290.18,
-      "learning_rate": 4.417224880382776e-05,
-      "loss": 0.0321,
-      "step": 3192
-    },
-    {
-      "epoch": 290.27,
-      "learning_rate": 4.415311004784689e-05,
-      "loss": 0.0375,
-      "step": 3193
-    },
-    {
-      "epoch": 290.36,
-      "learning_rate": 4.4133971291866034e-05,
-      "loss": 0.034,
-      "step": 3194
-    },
-    {
-      "epoch": 290.45,
-      "learning_rate": 4.411483253588517e-05,
-      "loss": 0.0391,
-      "step": 3195
-    },
-    {
-      "epoch": 290.55,
-      "learning_rate": 4.4095693779904305e-05,
-      "loss": 0.0314,
-      "step": 3196
-    },
-    {
-      "epoch": 290.64,
-      "learning_rate": 4.407655502392345e-05,
-      "loss": 0.0385,
-      "step": 3197
-    },
-    {
-      "epoch": 290.73,
-      "learning_rate": 4.405741626794259e-05,
-      "loss": 0.0309,
-      "step": 3198
-    },
-    {
-      "epoch": 290.82,
-      "learning_rate": 4.403827751196172e-05,
-      "loss": 0.0346,
-      "step": 3199
-    },
-    {
-      "epoch": 290.91,
-      "learning_rate": 4.401913875598087e-05,
-      "loss": 0.0324,
-      "step": 3200
-    },
-    {
-      "epoch": 290.91,
-      "eval_accuracy_NAT": 0.9928801974558572,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9476721103910039,
-      "eval_accuracy_concrete tiles": 0.9883507613281967,
-      "eval_accuracy_dark bricks": 0.8773334359724414,
-      "eval_accuracy_grass": 0.9009845622594952,
-      "eval_accuracy_red bricks": 0.9296875,
-      "eval_accuracy_road": 0.9350376729017505,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47454029167700496,
-      "eval_accuracy_speedway bricks": 0.9887313592243555,
-      "eval_accuracy_steel": 0.901037200712354,
-      "eval_accuracy_tiles": 0.10971099849335707,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9802268080423674,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8864591017900965,
-      "eval_iou_concrete tiles": 0.932362272858379,
-      "eval_iou_dark bricks": 0.823661198236612,
-      "eval_iou_grass": 0.8309598102041079,
-      "eval_iou_red bricks": 0.8634564643799473,
-      "eval_iou_road": 0.9273574180054591,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3160784457747255,
-      "eval_iou_speedway bricks": 0.976684677489055,
-      "eval_iou_steel": 0.8447280344430446,
-      "eval_iou_tiles": 0.10956093557652852,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10652707517147064,
-      "eval_mean_accuracy": 0.6958435454166013,
-      "eval_mean_iou": 0.6531950128307942,
-      "eval_overall_accuracy": 0.973974969651964,
-      "eval_runtime": 3.7046,
-      "eval_samples_per_second": 9.718,
-      "eval_steps_per_second": 0.54,
-      "step": 3200
-    },
-    {
-      "epoch": 291.0,
-      "learning_rate": 4.4000000000000006e-05,
-      "loss": 0.0374,
-      "step": 3201
-    },
-    {
-      "epoch": 291.09,
-      "learning_rate": 4.398086124401914e-05,
-      "loss": 0.0314,
-      "step": 3202
-    },
-    {
-      "epoch": 291.18,
-      "learning_rate": 4.396172248803828e-05,
-      "loss": 0.0343,
-      "step": 3203
-    },
-    {
-      "epoch": 291.27,
-      "learning_rate": 4.3942583732057415e-05,
-      "loss": 0.0325,
-      "step": 3204
-    },
-    {
-      "epoch": 291.36,
-      "learning_rate": 4.3923444976076554e-05,
-      "loss": 0.0332,
-      "step": 3205
-    },
-    {
-      "epoch": 291.45,
-      "learning_rate": 4.39043062200957e-05,
-      "loss": 0.035,
-      "step": 3206
-    },
-    {
-      "epoch": 291.55,
-      "learning_rate": 4.388516746411483e-05,
-      "loss": 0.0344,
-      "step": 3207
-    },
-    {
-      "epoch": 291.64,
-      "learning_rate": 4.386602870813397e-05,
-      "loss": 0.032,
-      "step": 3208
-    },
-    {
-      "epoch": 291.73,
-      "learning_rate": 4.3846889952153116e-05,
-      "loss": 0.0327,
-      "step": 3209
-    },
-    {
-      "epoch": 291.82,
-      "learning_rate": 4.382775119617225e-05,
-      "loss": 0.0336,
-      "step": 3210
-    },
-    {
-      "epoch": 291.91,
-      "learning_rate": 4.380861244019139e-05,
-      "loss": 0.0353,
-      "step": 3211
-    },
-    {
-      "epoch": 292.0,
-      "learning_rate": 4.378947368421053e-05,
-      "loss": 0.0354,
-      "step": 3212
-    },
-    {
-      "epoch": 292.09,
-      "learning_rate": 4.3770334928229664e-05,
-      "loss": 0.0306,
-      "step": 3213
-    },
-    {
-      "epoch": 292.18,
-      "learning_rate": 4.37511961722488e-05,
-      "loss": 0.0294,
-      "step": 3214
-    },
-    {
-      "epoch": 292.27,
-      "learning_rate": 4.373205741626795e-05,
-      "loss": 0.0361,
-      "step": 3215
-    },
-    {
-      "epoch": 292.36,
-      "learning_rate": 4.371291866028708e-05,
-      "loss": 0.0324,
-      "step": 3216
-    },
-    {
-      "epoch": 292.45,
-      "learning_rate": 4.369377990430622e-05,
-      "loss": 0.0394,
-      "step": 3217
-    },
-    {
-      "epoch": 292.55,
-      "learning_rate": 4.3674641148325365e-05,
-      "loss": 0.0433,
-      "step": 3218
-    },
-    {
-      "epoch": 292.64,
-      "learning_rate": 4.36555023923445e-05,
-      "loss": 0.0342,
-      "step": 3219
-    },
-    {
-      "epoch": 292.73,
-      "learning_rate": 4.3636363636363636e-05,
-      "loss": 0.0399,
-      "step": 3220
-    },
-    {
-      "epoch": 292.73,
-      "eval_accuracy_NAT": 0.9926534577713659,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9483896841066141,
-      "eval_accuracy_concrete tiles": 0.9868219898489574,
-      "eval_accuracy_dark bricks": 0.8857626727223741,
-      "eval_accuracy_grass": 0.911787835046502,
-      "eval_accuracy_red bricks": 0.9387428977272727,
-      "eval_accuracy_road": 0.9391526106214487,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.42260083257519365,
-      "eval_accuracy_speedway bricks": 0.9881110559998648,
-      "eval_accuracy_steel": 0.9022162126508806,
-      "eval_accuracy_tiles": 0.23476236132036707,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980299244743683,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8879123500973191,
-      "eval_iou_concrete tiles": 0.9325878409616274,
-      "eval_iou_dark bricks": 0.8257265877287406,
-      "eval_iou_grass": 0.837922650230135,
-      "eval_iou_red bricks": 0.8685723673402332,
-      "eval_iou_road": 0.9310990960749994,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.28797129492213186,
-      "eval_iou_speedway bricks": 0.9763210031557289,
-      "eval_iou_steel": 0.8409129332206255,
-      "eval_iou_tiles": 0.23442522054298023,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10238160192966461,
-      "eval_mean_accuracy": 0.7039232007992955,
-      "eval_mean_iou": 0.6618269683860157,
-      "eval_overall_accuracy": 0.9745070139567057,
-      "eval_runtime": 3.7615,
-      "eval_samples_per_second": 9.571,
-      "eval_steps_per_second": 0.532,
-      "step": 3220
-    },
-    {
-      "epoch": 292.82,
-      "learning_rate": 4.361722488038278e-05,
-      "loss": 0.0313,
-      "step": 3221
-    },
-    {
-      "epoch": 292.91,
-      "learning_rate": 4.359808612440191e-05,
-      "loss": 0.0335,
-      "step": 3222
-    },
-    {
-      "epoch": 293.0,
-      "learning_rate": 4.357894736842105e-05,
-      "loss": 0.0345,
-      "step": 3223
-    },
-    {
-      "epoch": 293.09,
-      "learning_rate": 4.35598086124402e-05,
-      "loss": 0.0313,
-      "step": 3224
-    },
-    {
-      "epoch": 293.18,
-      "learning_rate": 4.354066985645933e-05,
-      "loss": 0.0369,
-      "step": 3225
-    },
-    {
-      "epoch": 293.27,
-      "learning_rate": 4.352153110047847e-05,
-      "loss": 0.0383,
-      "step": 3226
-    },
-    {
-      "epoch": 293.36,
-      "learning_rate": 4.3502392344497614e-05,
-      "loss": 0.0312,
-      "step": 3227
-    },
-    {
-      "epoch": 293.45,
-      "learning_rate": 4.3483253588516746e-05,
-      "loss": 0.0316,
-      "step": 3228
-    },
-    {
-      "epoch": 293.55,
-      "learning_rate": 4.3464114832535885e-05,
-      "loss": 0.0359,
-      "step": 3229
-    },
-    {
-      "epoch": 293.64,
-      "learning_rate": 4.344497607655503e-05,
-      "loss": 0.0302,
-      "step": 3230
-    },
-    {
-      "epoch": 293.73,
-      "learning_rate": 4.342583732057416e-05,
-      "loss": 0.0387,
-      "step": 3231
-    },
-    {
-      "epoch": 293.82,
-      "learning_rate": 4.34066985645933e-05,
-      "loss": 0.0347,
-      "step": 3232
-    },
-    {
-      "epoch": 293.91,
-      "learning_rate": 4.3387559808612446e-05,
-      "loss": 0.0338,
-      "step": 3233
-    },
-    {
-      "epoch": 294.0,
-      "learning_rate": 4.336842105263158e-05,
-      "loss": 0.0351,
-      "step": 3234
-    },
-    {
-      "epoch": 294.09,
-      "learning_rate": 4.334928229665072e-05,
-      "loss": 0.0298,
-      "step": 3235
-    },
-    {
-      "epoch": 294.18,
-      "learning_rate": 4.333014354066986e-05,
-      "loss": 0.035,
-      "step": 3236
-    },
-    {
-      "epoch": 294.27,
-      "learning_rate": 4.3311004784688995e-05,
-      "loss": 0.0339,
-      "step": 3237
-    },
-    {
-      "epoch": 294.36,
-      "learning_rate": 4.3291866028708134e-05,
-      "loss": 0.033,
-      "step": 3238
-    },
-    {
-      "epoch": 294.45,
-      "learning_rate": 4.327272727272728e-05,
-      "loss": 0.0339,
-      "step": 3239
-    },
-    {
-      "epoch": 294.55,
-      "learning_rate": 4.325358851674641e-05,
-      "loss": 0.0358,
-      "step": 3240
-    },
-    {
-      "epoch": 294.55,
-      "eval_accuracy_NAT": 0.9930843513974575,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9515912700115885,
-      "eval_accuracy_concrete tiles": 0.9855683972359812,
-      "eval_accuracy_dark bricks": 0.8733305107578615,
-      "eval_accuracy_grass": 0.9006630154796814,
-      "eval_accuracy_red bricks": 0.9321732954545454,
-      "eval_accuracy_road": 0.9472083599863118,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4539188928404047,
-      "eval_accuracy_speedway bricks": 0.9888417828528397,
-      "eval_accuracy_steel": 0.900319899742761,
-      "eval_accuracy_tiles": 0.15751266949732914,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805308467173962,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8855123039825064,
-      "eval_iou_concrete tiles": 0.943038530177584,
-      "eval_iou_dark bricks": 0.8259919912631962,
-      "eval_iou_grass": 0.8391869140313342,
-      "eval_iou_red bricks": 0.8612204724409449,
-      "eval_iou_road": 0.9396016706609954,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3038775999852352,
-      "eval_iou_speedway bricks": 0.9766470573861987,
-      "eval_iou_steel": 0.8444994393101581,
-      "eval_iou_tiles": 0.15727571115973743,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09677817672491074,
-      "eval_mean_accuracy": 0.6987855727120587,
-      "eval_mean_iou": 0.6582601951627143,
-      "eval_overall_accuracy": 0.9754496680365669,
-      "eval_runtime": 3.9921,
-      "eval_samples_per_second": 9.018,
-      "eval_steps_per_second": 0.501,
-      "step": 3240
-    },
-    {
-      "epoch": 294.64,
-      "learning_rate": 4.323444976076555e-05,
-      "loss": 0.0331,
-      "step": 3241
-    },
-    {
-      "epoch": 294.73,
-      "learning_rate": 4.321531100478469e-05,
-      "loss": 0.0318,
-      "step": 3242
-    },
-    {
-      "epoch": 294.82,
-      "learning_rate": 4.319617224880383e-05,
-      "loss": 0.04,
-      "step": 3243
-    },
-    {
-      "epoch": 294.91,
-      "learning_rate": 4.3177033492822966e-05,
-      "loss": 0.0327,
-      "step": 3244
-    },
-    {
-      "epoch": 295.0,
-      "learning_rate": 4.3157894736842105e-05,
-      "loss": 0.0332,
-      "step": 3245
-    },
-    {
-      "epoch": 295.09,
-      "learning_rate": 4.3138755980861244e-05,
-      "loss": 0.032,
-      "step": 3246
-    },
-    {
-      "epoch": 295.18,
-      "learning_rate": 4.311961722488039e-05,
-      "loss": 0.0337,
-      "step": 3247
-    },
-    {
-      "epoch": 295.27,
-      "learning_rate": 4.310047846889952e-05,
-      "loss": 0.0306,
-      "step": 3248
-    },
-    {
-      "epoch": 295.36,
-      "learning_rate": 4.308133971291866e-05,
-      "loss": 0.0334,
-      "step": 3249
-    },
-    {
-      "epoch": 295.45,
-      "learning_rate": 4.3062200956937806e-05,
-      "loss": 0.0384,
-      "step": 3250
-    },
-    {
-      "epoch": 295.55,
-      "learning_rate": 4.304306220095694e-05,
-      "loss": 0.031,
-      "step": 3251
-    },
-    {
-      "epoch": 295.64,
-      "learning_rate": 4.3023923444976076e-05,
-      "loss": 0.0317,
-      "step": 3252
-    },
-    {
-      "epoch": 295.73,
-      "learning_rate": 4.300478468899522e-05,
-      "loss": 0.0364,
-      "step": 3253
-    },
-    {
-      "epoch": 295.82,
-      "learning_rate": 4.2985645933014354e-05,
-      "loss": 0.0329,
-      "step": 3254
-    },
-    {
-      "epoch": 295.91,
-      "learning_rate": 4.296650717703349e-05,
-      "loss": 0.0383,
-      "step": 3255
-    },
-    {
-      "epoch": 296.0,
-      "learning_rate": 4.294736842105264e-05,
-      "loss": 0.032,
-      "step": 3256
-    },
-    {
-      "epoch": 296.09,
-      "learning_rate": 4.292822966507177e-05,
-      "loss": 0.0317,
-      "step": 3257
-    },
-    {
-      "epoch": 296.18,
-      "learning_rate": 4.290909090909091e-05,
-      "loss": 0.0317,
-      "step": 3258
-    },
-    {
-      "epoch": 296.27,
-      "learning_rate": 4.2889952153110055e-05,
-      "loss": 0.033,
-      "step": 3259
-    },
-    {
-      "epoch": 296.36,
-      "learning_rate": 4.287081339712919e-05,
-      "loss": 0.0318,
-      "step": 3260
-    },
-    {
-      "epoch": 296.36,
-      "eval_accuracy_NAT": 0.9930144767553887,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9477029593544787,
-      "eval_accuracy_concrete tiles": 0.990582767687886,
-      "eval_accuracy_dark bricks": 0.8898810669335283,
-      "eval_accuracy_grass": 0.884659073036555,
-      "eval_accuracy_red bricks": 0.9339488636363636,
-      "eval_accuracy_road": 0.9442929574467314,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46715187605105724,
-      "eval_accuracy_speedway bricks": 0.9889443621798469,
-      "eval_accuracy_steel": 0.9081772310533606,
-      "eval_accuracy_tiles": 0.17353787152444872,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803066075943281,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8838498882330057,
-      "eval_iou_concrete tiles": 0.9357633874415112,
-      "eval_iou_dark bricks": 0.8275170908049679,
-      "eval_iou_grass": 0.8240579766764987,
-      "eval_iou_red bricks": 0.8511326860841424,
-      "eval_iou_road": 0.9362458712736559,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3131410197179975,
-      "eval_iou_speedway bricks": 0.9764795112514292,
-      "eval_iou_steel": 0.8420042806910258,
-      "eval_iou_tiles": 0.1730165232828076,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09844598919153214,
-      "eval_mean_accuracy": 0.7016841158199727,
-      "eval_mean_iou": 0.6571934494654901,
-      "eval_overall_accuracy": 0.9746184878879123,
-      "eval_runtime": 3.9111,
-      "eval_samples_per_second": 9.205,
-      "eval_steps_per_second": 0.511,
-      "step": 3260
-    },
-    {
-      "epoch": 296.45,
-      "learning_rate": 4.2851674641148325e-05,
-      "loss": 0.0322,
-      "step": 3261
-    },
-    {
-      "epoch": 296.55,
-      "learning_rate": 4.283253588516747e-05,
-      "loss": 0.0348,
-      "step": 3262
-    },
-    {
-      "epoch": 296.64,
-      "learning_rate": 4.28133971291866e-05,
-      "loss": 0.0348,
-      "step": 3263
-    },
-    {
-      "epoch": 296.73,
-      "learning_rate": 4.279425837320574e-05,
-      "loss": 0.0321,
-      "step": 3264
-    },
-    {
-      "epoch": 296.82,
-      "learning_rate": 4.277511961722489e-05,
-      "loss": 0.0303,
-      "step": 3265
-    },
-    {
-      "epoch": 296.91,
-      "learning_rate": 4.275598086124402e-05,
-      "loss": 0.0345,
-      "step": 3266
-    },
-    {
-      "epoch": 297.0,
-      "learning_rate": 4.273684210526316e-05,
-      "loss": 0.0322,
-      "step": 3267
-    },
-    {
-      "epoch": 297.09,
-      "learning_rate": 4.2717703349282304e-05,
-      "loss": 0.0312,
-      "step": 3268
-    },
-    {
-      "epoch": 297.18,
-      "learning_rate": 4.2698564593301436e-05,
-      "loss": 0.0333,
-      "step": 3269
-    },
-    {
-      "epoch": 297.27,
-      "learning_rate": 4.2679425837320574e-05,
-      "loss": 0.0341,
-      "step": 3270
-    },
-    {
-      "epoch": 297.36,
-      "learning_rate": 4.266028708133972e-05,
-      "loss": 0.0315,
-      "step": 3271
-    },
-    {
-      "epoch": 297.45,
-      "learning_rate": 4.264114832535885e-05,
-      "loss": 0.0378,
-      "step": 3272
-    },
-    {
-      "epoch": 297.55,
-      "learning_rate": 4.262200956937799e-05,
-      "loss": 0.0337,
-      "step": 3273
-    },
-    {
-      "epoch": 297.64,
-      "learning_rate": 4.2602870813397136e-05,
-      "loss": 0.0371,
-      "step": 3274
-    },
-    {
-      "epoch": 297.73,
-      "learning_rate": 4.258373205741627e-05,
-      "loss": 0.0431,
-      "step": 3275
-    },
-    {
-      "epoch": 297.82,
-      "learning_rate": 4.256459330143541e-05,
-      "loss": 0.036,
-      "step": 3276
-    },
-    {
-      "epoch": 297.91,
-      "learning_rate": 4.254545454545455e-05,
-      "loss": 0.031,
-      "step": 3277
-    },
-    {
-      "epoch": 298.0,
-      "learning_rate": 4.2526315789473685e-05,
-      "loss": 0.0347,
-      "step": 3278
-    },
-    {
-      "epoch": 298.09,
-      "learning_rate": 4.2507177033492823e-05,
-      "loss": 0.033,
-      "step": 3279
-    },
-    {
-      "epoch": 298.18,
-      "learning_rate": 4.248803827751197e-05,
-      "loss": 0.0367,
-      "step": 3280
-    },
-    {
-      "epoch": 298.18,
-      "eval_accuracy_NAT": 0.9927363897960437,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9519949890553243,
-      "eval_accuracy_concrete tiles": 0.980890356509509,
-      "eval_accuracy_dark bricks": 0.8761787460066972,
-      "eval_accuracy_grass": 0.90380159165656,
-      "eval_accuracy_red bricks": 0.9501065340909091,
-      "eval_accuracy_road": 0.9459314354397348,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46877843023736665,
-      "eval_accuracy_speedway bricks": 0.9875866946165163,
-      "eval_accuracy_steel": 0.9102466855748301,
-      "eval_accuracy_tiles": 0.23400903985755375,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9800911502519332,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8910874556984115,
-      "eval_iou_concrete tiles": 0.957384583246292,
-      "eval_iou_dark bricks": 0.8173201206376561,
-      "eval_iou_grass": 0.836131831747356,
-      "eval_iou_red bricks": 0.85616,
-      "eval_iou_road": 0.93759610901564,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3176655208489015,
-      "eval_iou_speedway bricks": 0.9769048720435723,
-      "eval_iou_steel": 0.8468034515819751,
-      "eval_iou_tiles": 0.23325824288347327,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09470751136541367,
-      "eval_mean_accuracy": 0.7078662225262342,
-      "eval_mean_iou": 0.66541564138117,
-      "eval_overall_accuracy": 0.9753408432006836,
-      "eval_runtime": 3.978,
-      "eval_samples_per_second": 9.05,
-      "eval_steps_per_second": 0.503,
-      "step": 3280
-    },
-    {
-      "epoch": 298.27,
-      "learning_rate": 4.24688995215311e-05,
-      "loss": 0.0337,
-      "step": 3281
-    },
-    {
-      "epoch": 298.36,
-      "learning_rate": 4.244976076555024e-05,
-      "loss": 0.0381,
-      "step": 3282
-    },
-    {
-      "epoch": 298.45,
-      "learning_rate": 4.243062200956938e-05,
-      "loss": 0.031,
-      "step": 3283
-    },
-    {
-      "epoch": 298.55,
-      "learning_rate": 4.241148325358852e-05,
-      "loss": 0.0419,
-      "step": 3284
-    },
-    {
-      "epoch": 298.64,
-      "learning_rate": 4.2392344497607656e-05,
-      "loss": 0.0331,
-      "step": 3285
-    },
-    {
-      "epoch": 298.73,
-      "learning_rate": 4.2373205741626795e-05,
-      "loss": 0.0346,
-      "step": 3286
-    },
-    {
-      "epoch": 298.82,
-      "learning_rate": 4.2354066985645934e-05,
-      "loss": 0.0375,
-      "step": 3287
-    },
-    {
-      "epoch": 298.91,
-      "learning_rate": 4.233492822966507e-05,
-      "loss": 0.0338,
-      "step": 3288
-    },
-    {
-      "epoch": 299.0,
-      "learning_rate": 4.231578947368421e-05,
-      "loss": 0.0282,
-      "step": 3289
-    },
-    {
-      "epoch": 299.09,
-      "learning_rate": 4.229665071770335e-05,
-      "loss": 0.0343,
-      "step": 3290
-    },
-    {
-      "epoch": 299.18,
-      "learning_rate": 4.227751196172249e-05,
-      "loss": 0.0338,
-      "step": 3291
-    },
-    {
-      "epoch": 299.27,
-      "learning_rate": 4.225837320574163e-05,
-      "loss": 0.0322,
-      "step": 3292
-    },
-    {
-      "epoch": 299.36,
-      "learning_rate": 4.2239234449760766e-05,
-      "loss": 0.0336,
-      "step": 3293
-    },
-    {
-      "epoch": 299.45,
-      "learning_rate": 4.2220095693779905e-05,
-      "loss": 0.0361,
-      "step": 3294
-    },
-    {
-      "epoch": 299.55,
-      "learning_rate": 4.2200956937799044e-05,
-      "loss": 0.0375,
-      "step": 3295
-    },
-    {
-      "epoch": 299.64,
-      "learning_rate": 4.218181818181818e-05,
-      "loss": 0.032,
-      "step": 3296
-    },
-    {
-      "epoch": 299.73,
-      "learning_rate": 4.216267942583732e-05,
-      "loss": 0.0368,
-      "step": 3297
-    },
-    {
-      "epoch": 299.82,
-      "learning_rate": 4.214354066985646e-05,
-      "loss": 0.0363,
-      "step": 3298
-    },
-    {
-      "epoch": 299.91,
-      "learning_rate": 4.21244019138756e-05,
-      "loss": 0.035,
-      "step": 3299
-    },
-    {
-      "epoch": 300.0,
-      "learning_rate": 4.210526315789474e-05,
-      "loss": 0.034,
-      "step": 3300
-    },
-    {
-      "epoch": 300.0,
-      "eval_accuracy_NAT": 0.9935969418904126,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9505866668097344,
-      "eval_accuracy_concrete tiles": 0.9853237937993029,
-      "eval_accuracy_dark bricks": 0.8730995727647127,
-      "eval_accuracy_grass": 0.8847010139208785,
-      "eval_accuracy_red bricks": 0.9224076704545454,
-      "eval_accuracy_road": 0.9404609745981497,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4984423676012461,
-      "eval_accuracy_speedway bricks": 0.9870979342937172,
-      "eval_accuracy_steel": 0.9078309478266605,
-      "eval_accuracy_tiles": 0.11964114504862347,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799122482331225,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8865514998373821,
-      "eval_iou_concrete tiles": 0.9432184042615466,
-      "eval_iou_dark bricks": 0.8185328185328186,
-      "eval_iou_grass": 0.8279543514311788,
-      "eval_iou_red bricks": 0.8477480417754569,
-      "eval_iou_road": 0.9332599763849824,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3341279961560496,
-      "eval_iou_speedway bricks": 0.976564206957072,
-      "eval_iou_steel": 0.8477161268467691,
-      "eval_iou_tiles": 0.11899734350521082,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10274531692266464,
-      "eval_mean_accuracy": 0.697168386846768,
-      "eval_mean_iou": 0.6549679241478146,
-      "eval_overall_accuracy": 0.974498430887858,
-      "eval_runtime": 4.0766,
-      "eval_samples_per_second": 8.831,
-      "eval_steps_per_second": 0.491,
-      "step": 3300
-    },
-    {
-      "epoch": 300.09,
-      "learning_rate": 4.2086124401913877e-05,
-      "loss": 0.0314,
-      "step": 3301
-    },
-    {
-      "epoch": 300.18,
-      "learning_rate": 4.2066985645933015e-05,
-      "loss": 0.0354,
-      "step": 3302
-    },
-    {
-      "epoch": 300.27,
-      "learning_rate": 4.2047846889952154e-05,
-      "loss": 0.0327,
-      "step": 3303
-    },
-    {
-      "epoch": 300.36,
-      "learning_rate": 4.202870813397129e-05,
-      "loss": 0.0327,
-      "step": 3304
-    },
-    {
-      "epoch": 300.45,
-      "learning_rate": 4.200956937799043e-05,
-      "loss": 0.033,
-      "step": 3305
-    },
-    {
-      "epoch": 300.55,
-      "learning_rate": 4.199043062200957e-05,
-      "loss": 0.0328,
-      "step": 3306
-    },
-    {
-      "epoch": 300.64,
-      "learning_rate": 4.197129186602871e-05,
-      "loss": 0.0356,
-      "step": 3307
-    },
-    {
-      "epoch": 300.73,
-      "learning_rate": 4.195215311004785e-05,
-      "loss": 0.0325,
-      "step": 3308
-    },
-    {
-      "epoch": 300.82,
-      "learning_rate": 4.193301435406699e-05,
-      "loss": 0.0361,
-      "step": 3309
-    },
-    {
-      "epoch": 300.91,
-      "learning_rate": 4.1913875598086126e-05,
-      "loss": 0.0339,
-      "step": 3310
-    },
-    {
-      "epoch": 301.0,
-      "learning_rate": 4.1894736842105264e-05,
-      "loss": 0.0371,
-      "step": 3311
-    },
-    {
-      "epoch": 301.09,
-      "learning_rate": 4.18755980861244e-05,
-      "loss": 0.0342,
-      "step": 3312
-    },
-    {
-      "epoch": 301.18,
-      "learning_rate": 4.185645933014354e-05,
-      "loss": 0.0338,
-      "step": 3313
-    },
-    {
-      "epoch": 301.27,
-      "learning_rate": 4.183732057416268e-05,
-      "loss": 0.0385,
-      "step": 3314
-    },
-    {
-      "epoch": 301.36,
-      "learning_rate": 4.181818181818182e-05,
-      "loss": 0.0333,
-      "step": 3315
-    },
-    {
-      "epoch": 301.45,
-      "learning_rate": 4.179904306220096e-05,
-      "loss": 0.0372,
-      "step": 3316
-    },
-    {
-      "epoch": 301.55,
-      "learning_rate": 4.17799043062201e-05,
-      "loss": 0.0409,
-      "step": 3317
-    },
-    {
-      "epoch": 301.64,
-      "learning_rate": 4.176076555023924e-05,
-      "loss": 0.0358,
-      "step": 3318
-    },
-    {
-      "epoch": 301.73,
-      "learning_rate": 4.1741626794258375e-05,
-      "loss": 0.0273,
-      "step": 3319
-    },
-    {
-      "epoch": 301.82,
-      "learning_rate": 4.172248803827751e-05,
-      "loss": 0.0394,
-      "step": 3320
-    },
-    {
-      "epoch": 301.82,
-      "eval_accuracy_NAT": 0.9931478737993383,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.949328565603674,
-      "eval_accuracy_concrete tiles": 0.9920809637375405,
-      "eval_accuracy_dark bricks": 0.8672876332704669,
-      "eval_accuracy_grass": 0.891645725350119,
-      "eval_accuracy_red bricks": 0.9263139204545454,
-      "eval_accuracy_road": 0.9365225813706866,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47765555647451274,
-      "eval_accuracy_speedway bricks": 0.9873815359625018,
-      "eval_accuracy_steel": 0.9111618626739661,
-      "eval_accuracy_tiles": 0.21134091220380768,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804644284008392,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8835423633814807,
-      "eval_iou_concrete tiles": 0.9032123371562187,
-      "eval_iou_dark bricks": 0.8245087635844707,
-      "eval_iou_grass": 0.8296179949204738,
-      "eval_iou_red bricks": 0.845954272742014,
-      "eval_iou_road": 0.9295302013422819,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3217276660538874,
-      "eval_iou_speedway bricks": 0.9765579587759053,
-      "eval_iou_steel": 0.8464278055206642,
-      "eval_iou_tiles": 0.21026095251073107,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10323752462863922,
-      "eval_mean_accuracy": 0.7033743946847046,
-      "eval_mean_iou": 0.6578311341837667,
-      "eval_overall_accuracy": 0.9741591347588433,
-      "eval_runtime": 3.8214,
-      "eval_samples_per_second": 9.421,
-      "eval_steps_per_second": 0.523,
-      "step": 3320
-    },
-    {
-      "epoch": 301.91,
-      "learning_rate": 4.170334928229665e-05,
-      "loss": 0.0351,
-      "step": 3321
-    },
-    {
-      "epoch": 302.0,
-      "learning_rate": 4.168421052631579e-05,
-      "loss": 0.0359,
-      "step": 3322
-    },
-    {
-      "epoch": 302.09,
-      "learning_rate": 4.166507177033493e-05,
-      "loss": 0.0396,
-      "step": 3323
-    },
-    {
-      "epoch": 302.18,
-      "learning_rate": 4.164593301435407e-05,
-      "loss": 0.0333,
-      "step": 3324
-    },
-    {
-      "epoch": 302.27,
-      "learning_rate": 4.162679425837321e-05,
-      "loss": 0.0363,
-      "step": 3325
-    },
-    {
-      "epoch": 302.36,
-      "learning_rate": 4.1607655502392346e-05,
-      "loss": 0.0407,
-      "step": 3326
-    },
-    {
-      "epoch": 302.45,
-      "learning_rate": 4.1588516746411485e-05,
-      "loss": 0.0282,
-      "step": 3327
-    },
-    {
-      "epoch": 302.55,
-      "learning_rate": 4.1569377990430624e-05,
-      "loss": 0.0411,
-      "step": 3328
-    },
-    {
-      "epoch": 302.64,
-      "learning_rate": 4.155023923444976e-05,
-      "loss": 0.0324,
-      "step": 3329
-    },
-    {
-      "epoch": 302.73,
-      "learning_rate": 4.15311004784689e-05,
-      "loss": 0.0345,
-      "step": 3330
-    },
-    {
-      "epoch": 302.82,
-      "learning_rate": 4.151196172248804e-05,
-      "loss": 0.031,
-      "step": 3331
-    },
-    {
-      "epoch": 302.91,
-      "learning_rate": 4.149282296650718e-05,
-      "loss": 0.0317,
-      "step": 3332
-    },
-    {
-      "epoch": 303.0,
-      "learning_rate": 4.147368421052632e-05,
-      "loss": 0.0322,
-      "step": 3333
-    },
-    {
-      "epoch": 303.09,
-      "learning_rate": 4.1454545454545456e-05,
-      "loss": 0.0347,
-      "step": 3334
-    },
-    {
-      "epoch": 303.18,
-      "learning_rate": 4.1435406698564595e-05,
-      "loss": 0.0334,
-      "step": 3335
-    },
-    {
-      "epoch": 303.27,
-      "learning_rate": 4.1416267942583734e-05,
-      "loss": 0.0383,
-      "step": 3336
-    },
-    {
-      "epoch": 303.36,
-      "learning_rate": 4.139712918660287e-05,
-      "loss": 0.0345,
-      "step": 3337
-    },
-    {
-      "epoch": 303.45,
-      "learning_rate": 4.137799043062201e-05,
-      "loss": 0.0324,
-      "step": 3338
-    },
-    {
-      "epoch": 303.55,
-      "learning_rate": 4.135885167464115e-05,
-      "loss": 0.033,
-      "step": 3339
-    },
-    {
-      "epoch": 303.64,
-      "learning_rate": 4.133971291866029e-05,
-      "loss": 0.0353,
-      "step": 3340
-    },
-    {
-      "epoch": 303.64,
-      "eval_accuracy_NAT": 0.9924214245533846,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9497497210180694,
-      "eval_accuracy_concrete tiles": 0.9866079618418638,
-      "eval_accuracy_dark bricks": 0.8725607174473654,
-      "eval_accuracy_grass": 0.9075657860245984,
-      "eval_accuracy_red bricks": 0.9360795454545454,
-      "eval_accuracy_road": 0.9413860193691074,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4908334022551209,
-      "eval_accuracy_speedway bricks": 0.9881864819756054,
-      "eval_accuracy_steel": 0.9108568036409208,
-      "eval_accuracy_tiles": 0.25873168059169976,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9801066480787662,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8885998431372549,
-      "eval_iou_concrete tiles": 0.9276945634361613,
-      "eval_iou_dark bricks": 0.8240639767357325,
-      "eval_iou_grass": 0.8344607513906608,
-      "eval_iou_red bricks": 0.8294524858401511,
-      "eval_iou_road": 0.933524871365344,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3313666734910384,
-      "eval_iou_speedway bricks": 0.9767699853992309,
-      "eval_iou_steel": 0.8496781289176364,
-      "eval_iou_tiles": 0.25809536821970214,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09785914421081543,
-      "eval_mean_accuracy": 0.7103830418594063,
-      "eval_mean_iou": 0.6641394843085906,
-      "eval_overall_accuracy": 0.9749277962578667,
-      "eval_runtime": 3.6597,
-      "eval_samples_per_second": 9.837,
-      "eval_steps_per_second": 0.546,
-      "step": 3340
-    },
-    {
-      "epoch": 303.73,
-      "learning_rate": 4.132057416267943e-05,
-      "loss": 0.031,
-      "step": 3341
-    },
-    {
-      "epoch": 303.82,
-      "learning_rate": 4.1301435406698566e-05,
-      "loss": 0.0377,
-      "step": 3342
-    },
-    {
-      "epoch": 303.91,
-      "learning_rate": 4.1282296650717705e-05,
-      "loss": 0.0376,
-      "step": 3343
-    },
-    {
-      "epoch": 304.0,
-      "learning_rate": 4.1263157894736844e-05,
-      "loss": 0.032,
-      "step": 3344
-    },
-    {
-      "epoch": 304.09,
-      "learning_rate": 4.124401913875598e-05,
-      "loss": 0.0315,
-      "step": 3345
-    },
-    {
-      "epoch": 304.18,
-      "learning_rate": 4.122488038277512e-05,
-      "loss": 0.0275,
-      "step": 3346
-    },
-    {
-      "epoch": 304.27,
-      "learning_rate": 4.120574162679426e-05,
-      "loss": 0.0294,
-      "step": 3347
-    },
-    {
-      "epoch": 304.36,
-      "learning_rate": 4.11866028708134e-05,
-      "loss": 0.0375,
-      "step": 3348
-    },
-    {
-      "epoch": 304.45,
-      "learning_rate": 4.116746411483254e-05,
-      "loss": 0.0313,
-      "step": 3349
-    },
-    {
-      "epoch": 304.55,
-      "learning_rate": 4.114832535885168e-05,
-      "loss": 0.0353,
-      "step": 3350
-    },
-    {
-      "epoch": 304.64,
-      "learning_rate": 4.1129186602870815e-05,
-      "loss": 0.0391,
-      "step": 3351
-    },
-    {
-      "epoch": 304.73,
-      "learning_rate": 4.1110047846889954e-05,
-      "loss": 0.0369,
-      "step": 3352
-    },
-    {
-      "epoch": 304.82,
-      "learning_rate": 4.109090909090909e-05,
-      "loss": 0.0353,
-      "step": 3353
-    },
-    {
-      "epoch": 304.91,
-      "learning_rate": 4.107177033492823e-05,
-      "loss": 0.0362,
-      "step": 3354
-    },
-    {
-      "epoch": 305.0,
-      "learning_rate": 4.105263157894737e-05,
-      "loss": 0.0337,
-      "step": 3355
-    },
-    {
-      "epoch": 305.09,
-      "learning_rate": 4.103349282296651e-05,
-      "loss": 0.0332,
-      "step": 3356
-    },
-    {
-      "epoch": 305.18,
-      "learning_rate": 4.101435406698565e-05,
-      "loss": 0.0352,
-      "step": 3357
-    },
-    {
-      "epoch": 305.27,
-      "learning_rate": 4.099521531100479e-05,
-      "loss": 0.0353,
-      "step": 3358
-    },
-    {
-      "epoch": 305.36,
-      "learning_rate": 4.0976076555023926e-05,
-      "loss": 0.0351,
-      "step": 3359
-    },
-    {
-      "epoch": 305.45,
-      "learning_rate": 4.0956937799043064e-05,
-      "loss": 0.0325,
-      "step": 3360
-    },
-    {
-      "epoch": 305.45,
-      "eval_accuracy_NAT": 0.9934497816593887,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9499750525773638,
-      "eval_accuracy_concrete tiles": 0.9793310096006849,
-      "eval_accuracy_dark bricks": 0.8844925137600554,
-      "eval_accuracy_grass": 0.8941447030410636,
-      "eval_accuracy_red bricks": 0.9275568181818182,
-      "eval_accuracy_road": 0.9438189475771687,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47205910732500755,
-      "eval_accuracy_speedway bricks": 0.987082849098569,
-      "eval_accuracy_steel": 0.9120523052569092,
-      "eval_accuracy_tiles": 0.23784413094096699,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9801469653790512,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.890473442487745,
-      "eval_iou_concrete tiles": 0.9392411002287255,
-      "eval_iou_dark bricks": 0.8263808975834293,
-      "eval_iou_grass": 0.8308623651793224,
-      "eval_iou_red bricks": 0.8488787780305492,
-      "eval_iou_road": 0.9351704637356449,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31944703555837467,
-      "eval_iou_speedway bricks": 0.9767510317748438,
-      "eval_iou_steel": 0.8476185367946792,
-      "eval_iou_tiles": 0.23758380079354222,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09690146893262863,
-      "eval_mean_accuracy": 0.7062928630014612,
-      "eval_mean_iou": 0.6640426475035314,
-      "eval_overall_accuracy": 0.9750890731811523,
-      "eval_runtime": 3.9827,
-      "eval_samples_per_second": 9.039,
-      "eval_steps_per_second": 0.502,
-      "step": 3360
-    },
-    {
-      "epoch": 305.55,
-      "learning_rate": 4.09377990430622e-05,
-      "loss": 0.0315,
-      "step": 3361
-    },
-    {
-      "epoch": 305.64,
-      "learning_rate": 4.091866028708134e-05,
-      "loss": 0.0297,
-      "step": 3362
-    },
-    {
-      "epoch": 305.73,
-      "learning_rate": 4.089952153110048e-05,
-      "loss": 0.0407,
-      "step": 3363
-    },
-    {
-      "epoch": 305.82,
-      "learning_rate": 4.088038277511962e-05,
-      "loss": 0.0328,
-      "step": 3364
-    },
-    {
-      "epoch": 305.91,
-      "learning_rate": 4.086124401913876e-05,
-      "loss": 0.0296,
-      "step": 3365
-    },
-    {
-      "epoch": 306.0,
-      "learning_rate": 4.08421052631579e-05,
-      "loss": 0.0307,
-      "step": 3366
-    },
-    {
-      "epoch": 306.09,
-      "learning_rate": 4.0822966507177036e-05,
-      "loss": 0.0351,
-      "step": 3367
-    },
-    {
-      "epoch": 306.18,
-      "learning_rate": 4.0803827751196175e-05,
-      "loss": 0.0377,
-      "step": 3368
-    },
-    {
-      "epoch": 306.27,
-      "learning_rate": 4.0784688995215313e-05,
-      "loss": 0.0459,
-      "step": 3369
-    },
-    {
-      "epoch": 306.36,
-      "learning_rate": 4.076555023923445e-05,
-      "loss": 0.0345,
-      "step": 3370
-    },
-    {
-      "epoch": 306.45,
-      "learning_rate": 4.074641148325359e-05,
-      "loss": 0.033,
-      "step": 3371
-    },
-    {
-      "epoch": 306.55,
-      "learning_rate": 4.072727272727273e-05,
-      "loss": 0.0335,
-      "step": 3372
-    },
-    {
-      "epoch": 306.64,
-      "learning_rate": 4.070813397129187e-05,
-      "loss": 0.0314,
-      "step": 3373
-    },
-    {
-      "epoch": 306.73,
-      "learning_rate": 4.068899521531101e-05,
-      "loss": 0.0346,
-      "step": 3374
-    },
-    {
-      "epoch": 306.82,
-      "learning_rate": 4.0669856459330146e-05,
-      "loss": 0.0298,
-      "step": 3375
-    },
-    {
-      "epoch": 306.91,
-      "learning_rate": 4.0650717703349285e-05,
-      "loss": 0.0354,
-      "step": 3376
-    },
-    {
-      "epoch": 307.0,
-      "learning_rate": 4.0631578947368424e-05,
-      "loss": 0.0353,
-      "step": 3377
-    },
-    {
-      "epoch": 307.09,
-      "learning_rate": 4.061244019138756e-05,
-      "loss": 0.0334,
-      "step": 3378
-    },
-    {
-      "epoch": 307.18,
-      "learning_rate": 4.05933014354067e-05,
-      "loss": 0.0424,
-      "step": 3379
-    },
-    {
-      "epoch": 307.27,
-      "learning_rate": 4.057416267942584e-05,
-      "loss": 0.0356,
-      "step": 3380
-    },
-    {
-      "epoch": 307.27,
-      "eval_accuracy_NAT": 0.9919252440142488,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9490562899695266,
-      "eval_accuracy_concrete tiles": 0.9857212743839051,
-      "eval_accuracy_dark bricks": 0.8814903198491205,
-      "eval_accuracy_grass": 0.9202494084587773,
-      "eval_accuracy_red bricks": 0.9190340909090909,
-      "eval_accuracy_road": 0.9480245759606862,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4478537755355223,
-      "eval_accuracy_speedway bricks": 0.989936968020593,
-      "eval_accuracy_steel": 0.8986214629641844,
-      "eval_accuracy_tiles": 0.19312422955759484,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804480020873324,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8876425073260809,
-      "eval_iou_concrete tiles": 0.9404060439880987,
-      "eval_iou_dark bricks": 0.8299931141956294,
-      "eval_iou_grass": 0.8392374480454905,
-      "eval_iou_red bricks": 0.8552544613350959,
-      "eval_iou_road": 0.9393330529726025,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.30404836324842316,
-      "eval_iou_speedway bricks": 0.9762730079877461,
-      "eval_iou_steel": 0.8482065729160992,
-      "eval_iou_tiles": 0.19293924466338258,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09550207108259201,
-      "eval_mean_accuracy": 0.7019259722787117,
-      "eval_mean_iou": 0.6610601399050755,
-      "eval_overall_accuracy": 0.9754361046685113,
-      "eval_runtime": 3.9327,
-      "eval_samples_per_second": 9.154,
-      "eval_steps_per_second": 0.509,
-      "step": 3380
-    },
-    {
-      "epoch": 307.36,
-      "learning_rate": 4.055502392344498e-05,
-      "loss": 0.0301,
-      "step": 3381
-    },
-    {
-      "epoch": 307.45,
-      "learning_rate": 4.053588516746412e-05,
-      "loss": 0.0364,
-      "step": 3382
-    },
-    {
-      "epoch": 307.55,
-      "learning_rate": 4.051674641148325e-05,
-      "loss": 0.0318,
-      "step": 3383
-    },
-    {
-      "epoch": 307.64,
-      "learning_rate": 4.0497607655502395e-05,
-      "loss": 0.0331,
-      "step": 3384
-    },
-    {
-      "epoch": 307.73,
-      "learning_rate": 4.0478468899521534e-05,
-      "loss": 0.0276,
-      "step": 3385
-    },
-    {
-      "epoch": 307.82,
-      "learning_rate": 4.0459330143540666e-05,
-      "loss": 0.0325,
-      "step": 3386
-    },
-    {
-      "epoch": 307.91,
-      "learning_rate": 4.044019138755981e-05,
-      "loss": 0.0353,
-      "step": 3387
-    },
-    {
-      "epoch": 308.0,
-      "learning_rate": 4.042105263157895e-05,
-      "loss": 0.0378,
-      "step": 3388
-    },
-    {
-      "epoch": 308.09,
-      "learning_rate": 4.040191387559809e-05,
-      "loss": 0.0369,
-      "step": 3389
-    },
-    {
-      "epoch": 308.18,
-      "learning_rate": 4.038277511961723e-05,
-      "loss": 0.0317,
-      "step": 3390
-    },
-    {
-      "epoch": 308.27,
-      "learning_rate": 4.0363636363636367e-05,
-      "loss": 0.0312,
-      "step": 3391
-    },
-    {
-      "epoch": 308.36,
-      "learning_rate": 4.0344497607655505e-05,
-      "loss": 0.0345,
-      "step": 3392
-    },
-    {
-      "epoch": 308.45,
-      "learning_rate": 4.0325358851674644e-05,
-      "loss": 0.0337,
-      "step": 3393
-    },
-    {
-      "epoch": 308.55,
-      "learning_rate": 4.030622009569378e-05,
-      "loss": 0.0325,
-      "step": 3394
-    },
-    {
-      "epoch": 308.64,
-      "learning_rate": 4.028708133971292e-05,
-      "loss": 0.0315,
-      "step": 3395
-    },
-    {
-      "epoch": 308.73,
-      "learning_rate": 4.026794258373206e-05,
-      "loss": 0.0303,
-      "step": 3396
-    },
-    {
-      "epoch": 308.82,
-      "learning_rate": 4.02488038277512e-05,
-      "loss": 0.033,
-      "step": 3397
-    },
-    {
-      "epoch": 308.91,
-      "learning_rate": 4.022966507177034e-05,
-      "loss": 0.0337,
-      "step": 3398
-    },
-    {
-      "epoch": 309.0,
-      "learning_rate": 4.021052631578948e-05,
-      "loss": 0.0349,
-      "step": 3399
-    },
-    {
-      "epoch": 309.09,
-      "learning_rate": 4.0191387559808616e-05,
-      "loss": 0.033,
-      "step": 3400
-    },
-    {
-      "epoch": 309.09,
-      "eval_accuracy_NAT": 0.9933112675330652,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9497845937593888,
-      "eval_accuracy_concrete tiles": 0.9833669663058766,
-      "eval_accuracy_dark bricks": 0.8824140718217158,
-      "eval_accuracy_grass": 0.89370781882936,
-      "eval_accuracy_red bricks": 0.9463778409090909,
-      "eval_accuracy_road": 0.9699705073961259,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48975822236925537,
-      "eval_accuracy_speedway bricks": 0.9877821987456359,
-      "eval_accuracy_steel": 0.9079793549238176,
-      "eval_accuracy_tiles": 0.18339953431036846,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9800280496216469,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8895545374719396,
-      "eval_iou_concrete tiles": 0.9402718900745505,
-      "eval_iou_dark bricks": 0.8299004524886878,
-      "eval_iou_grass": 0.8338251838325208,
-      "eval_iou_red bricks": 0.8562248995983935,
-      "eval_iou_road": 0.9614832785360433,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3288718575289718,
-      "eval_iou_speedway bricks": 0.9766324975808001,
-      "eval_iou_steel": 0.846173941773533,
-      "eval_iou_tiles": 0.18292349726775955,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08639038354158401,
-      "eval_mean_accuracy": 0.7067578751464385,
-      "eval_mean_iou": 0.6635300065980652,
-      "eval_overall_accuracy": 0.9773430294460721,
-      "eval_runtime": 4.1266,
-      "eval_samples_per_second": 8.724,
-      "eval_steps_per_second": 0.485,
-      "step": 3400
-    },
-    {
-      "epoch": 309.18,
-      "learning_rate": 4.0172248803827754e-05,
-      "loss": 0.0329,
-      "step": 3401
-    },
-    {
-      "epoch": 309.27,
-      "learning_rate": 4.015311004784689e-05,
-      "loss": 0.0329,
-      "step": 3402
-    },
-    {
-      "epoch": 309.36,
-      "learning_rate": 4.013397129186603e-05,
-      "loss": 0.037,
-      "step": 3403
-    },
-    {
-      "epoch": 309.45,
-      "learning_rate": 4.011483253588517e-05,
-      "loss": 0.0355,
-      "step": 3404
-    },
-    {
-      "epoch": 309.55,
-      "learning_rate": 4.009569377990431e-05,
-      "loss": 0.0339,
-      "step": 3405
-    },
-    {
-      "epoch": 309.64,
-      "learning_rate": 4.007655502392345e-05,
-      "loss": 0.0348,
-      "step": 3406
-    },
-    {
-      "epoch": 309.73,
-      "learning_rate": 4.005741626794259e-05,
-      "loss": 0.0379,
-      "step": 3407
-    },
-    {
-      "epoch": 309.82,
-      "learning_rate": 4.0038277511961726e-05,
-      "loss": 0.0337,
-      "step": 3408
-    },
-    {
-      "epoch": 309.91,
-      "learning_rate": 4.0019138755980864e-05,
-      "loss": 0.0333,
-      "step": 3409
-    },
-    {
-      "epoch": 310.0,
-      "learning_rate": 4e-05,
-      "loss": 0.0342,
-      "step": 3410
-    },
-    {
-      "epoch": 310.09,
-      "learning_rate": 3.998086124401914e-05,
-      "loss": 0.037,
-      "step": 3411
-    },
-    {
-      "epoch": 310.18,
-      "learning_rate": 3.996172248803828e-05,
-      "loss": 0.0326,
-      "step": 3412
-    },
-    {
-      "epoch": 310.27,
-      "learning_rate": 3.994258373205742e-05,
-      "loss": 0.0331,
-      "step": 3413
-    },
-    {
-      "epoch": 310.36,
-      "learning_rate": 3.992344497607656e-05,
-      "loss": 0.0341,
-      "step": 3414
-    },
-    {
-      "epoch": 310.45,
-      "learning_rate": 3.99043062200957e-05,
-      "loss": 0.0347,
-      "step": 3415
-    },
-    {
-      "epoch": 310.55,
-      "learning_rate": 3.9885167464114836e-05,
-      "loss": 0.0307,
-      "step": 3416
-    },
-    {
-      "epoch": 310.64,
-      "learning_rate": 3.9866028708133975e-05,
-      "loss": 0.029,
-      "step": 3417
-    },
-    {
-      "epoch": 310.73,
-      "learning_rate": 3.9846889952153113e-05,
-      "loss": 0.0337,
-      "step": 3418
-    },
-    {
-      "epoch": 310.82,
-      "learning_rate": 3.982775119617225e-05,
-      "loss": 0.0339,
-      "step": 3419
-    },
-    {
-      "epoch": 310.91,
-      "learning_rate": 3.980861244019139e-05,
-      "loss": 0.0304,
-      "step": 3420
-    },
-    {
-      "epoch": 310.91,
-      "eval_accuracy_NAT": 0.9926220494726582,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9542778123524615,
-      "eval_accuracy_concrete tiles": 0.9782302941356326,
-      "eval_accuracy_dark bricks": 0.8792579192486817,
-      "eval_accuracy_grass": 0.9044796359531241,
-      "eval_accuracy_red bricks": 0.9311079545454546,
-      "eval_accuracy_road": 0.9565289418092424,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48782841231770185,
-      "eval_accuracy_speedway bricks": 0.9869917345198744,
-      "eval_accuracy_steel": 0.9089357562166084,
-      "eval_accuracy_tiles": 0.16744281605259553,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799671699755961,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8887388529621547,
-      "eval_iou_concrete tiles": 0.9522590630394666,
-      "eval_iou_dark bricks": 0.8260948179221062,
-      "eval_iou_grass": 0.8346110059373881,
-      "eval_iou_red bricks": 0.8533767290480065,
-      "eval_iou_road": 0.9489043126943304,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3264940863885455,
-      "eval_iou_speedway bricks": 0.9766072234601249,
-      "eval_iou_steel": 0.8462069865442627,
-      "eval_iou_tiles": 0.1671794871794872,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09159811586141586,
-      "eval_mean_accuracy": 0.7036694866633874,
-      "eval_mean_iou": 0.6615722873193437,
-      "eval_overall_accuracy": 0.9762385686238607,
-      "eval_runtime": 3.8251,
-      "eval_samples_per_second": 9.411,
-      "eval_steps_per_second": 0.523,
-      "step": 3420
-    },
-    {
-      "epoch": 311.0,
-      "learning_rate": 3.978947368421053e-05,
-      "loss": 0.0304,
-      "step": 3421
-    },
-    {
-      "epoch": 311.09,
-      "learning_rate": 3.977033492822967e-05,
-      "loss": 0.0369,
-      "step": 3422
-    },
-    {
-      "epoch": 311.18,
-      "learning_rate": 3.975119617224881e-05,
-      "loss": 0.0344,
-      "step": 3423
-    },
-    {
-      "epoch": 311.27,
-      "learning_rate": 3.973205741626794e-05,
-      "loss": 0.0327,
-      "step": 3424
-    },
-    {
-      "epoch": 311.36,
-      "learning_rate": 3.9712918660287085e-05,
-      "loss": 0.0346,
-      "step": 3425
-    },
-    {
-      "epoch": 311.45,
-      "learning_rate": 3.9693779904306224e-05,
-      "loss": 0.0387,
-      "step": 3426
-    },
-    {
-      "epoch": 311.55,
-      "learning_rate": 3.9674641148325356e-05,
-      "loss": 0.0299,
-      "step": 3427
-    },
-    {
-      "epoch": 311.64,
-      "learning_rate": 3.96555023923445e-05,
-      "loss": 0.0344,
-      "step": 3428
-    },
-    {
-      "epoch": 311.73,
-      "learning_rate": 3.963636363636364e-05,
-      "loss": 0.0328,
-      "step": 3429
-    },
-    {
-      "epoch": 311.82,
-      "learning_rate": 3.961722488038277e-05,
-      "loss": 0.0327,
-      "step": 3430
-    },
-    {
-      "epoch": 311.91,
-      "learning_rate": 3.959808612440192e-05,
-      "loss": 0.0316,
-      "step": 3431
-    },
-    {
-      "epoch": 312.0,
-      "learning_rate": 3.9578947368421056e-05,
-      "loss": 0.0321,
-      "step": 3432
-    },
-    {
-      "epoch": 312.09,
-      "learning_rate": 3.955980861244019e-05,
-      "loss": 0.0308,
-      "step": 3433
-    },
-    {
-      "epoch": 312.18,
-      "learning_rate": 3.9540669856459334e-05,
-      "loss": 0.0322,
-      "step": 3434
-    },
-    {
-      "epoch": 312.27,
-      "learning_rate": 3.952153110047847e-05,
-      "loss": 0.0328,
-      "step": 3435
-    },
-    {
-      "epoch": 312.36,
-      "learning_rate": 3.9502392344497605e-05,
-      "loss": 0.0317,
-      "step": 3436
-    },
-    {
-      "epoch": 312.45,
-      "learning_rate": 3.948325358851675e-05,
-      "loss": 0.0328,
-      "step": 3437
-    },
-    {
-      "epoch": 312.55,
-      "learning_rate": 3.946411483253589e-05,
-      "loss": 0.0316,
-      "step": 3438
-    },
-    {
-      "epoch": 312.64,
-      "learning_rate": 3.944497607655502e-05,
-      "loss": 0.0352,
-      "step": 3439
-    },
-    {
-      "epoch": 312.73,
-      "learning_rate": 3.9425837320574167e-05,
-      "loss": 0.0318,
-      "step": 3440
-    },
-    {
-      "epoch": 312.73,
-      "eval_accuracy_NAT": 0.9934326659011041,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9474172711275162,
-      "eval_accuracy_concrete tiles": 0.9825720051366722,
-      "eval_accuracy_dark bricks": 0.8819521958354182,
-      "eval_accuracy_grass": 0.9018618257565961,
-      "eval_accuracy_red bricks": 0.9378551136363636,
-      "eval_accuracy_road": 0.956197618584089,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4831141620489069,
-      "eval_accuracy_speedway bricks": 0.987724875004073,
-      "eval_accuracy_steel": 0.9065777323395554,
-      "eval_accuracy_tiles": 0.1431310779345295,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9799850654301261,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8897549522223593,
-      "eval_iou_concrete tiles": 0.951106901858648,
-      "eval_iou_dark bricks": 0.8228534492045821,
-      "eval_iou_grass": 0.8358345291349091,
-      "eval_iou_red bricks": 0.8622265752530199,
-      "eval_iou_road": 0.9475367445293609,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3261796184271754,
-      "eval_iou_speedway bricks": 0.9767786154409825,
-      "eval_iou_steel": 0.8443876179725236,
-      "eval_iou_tiles": 0.14270107879284447,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09347935020923615,
-      "eval_mean_accuracy": 0.7016797341003712,
-      "eval_mean_iou": 0.6599496267897332,
-      "eval_overall_accuracy": 0.9761441548665365,
-      "eval_runtime": 4.1115,
-      "eval_samples_per_second": 8.756,
-      "eval_steps_per_second": 0.486,
-      "step": 3440
-    },
-    {
-      "epoch": 312.82,
-      "learning_rate": 3.9406698564593305e-05,
-      "loss": 0.035,
-      "step": 3441
-    },
-    {
-      "epoch": 312.91,
-      "learning_rate": 3.938755980861244e-05,
-      "loss": 0.0329,
-      "step": 3442
-    },
-    {
-      "epoch": 313.0,
-      "learning_rate": 3.936842105263158e-05,
-      "loss": 0.037,
-      "step": 3443
-    },
-    {
-      "epoch": 313.09,
-      "learning_rate": 3.934928229665072e-05,
-      "loss": 0.0362,
-      "step": 3444
-    },
-    {
-      "epoch": 313.18,
-      "learning_rate": 3.9330143540669854e-05,
-      "loss": 0.0352,
-      "step": 3445
-    },
-    {
-      "epoch": 313.27,
-      "learning_rate": 3.9311004784689e-05,
-      "loss": 0.0318,
-      "step": 3446
-    },
-    {
-      "epoch": 313.36,
-      "learning_rate": 3.929186602870814e-05,
-      "loss": 0.0291,
-      "step": 3447
-    },
-    {
-      "epoch": 313.45,
-      "learning_rate": 3.927272727272727e-05,
-      "loss": 0.0315,
-      "step": 3448
-    },
-    {
-      "epoch": 313.55,
-      "learning_rate": 3.9253588516746416e-05,
-      "loss": 0.0314,
-      "step": 3449
-    },
-    {
-      "epoch": 313.64,
-      "learning_rate": 3.9234449760765554e-05,
-      "loss": 0.0343,
-      "step": 3450
-    },
-    {
-      "epoch": 313.73,
-      "learning_rate": 3.9215311004784686e-05,
-      "loss": 0.033,
-      "step": 3451
-    },
-    {
-      "epoch": 313.82,
-      "learning_rate": 3.919617224880383e-05,
-      "loss": 0.0344,
-      "step": 3452
-    },
-    {
-      "epoch": 313.91,
-      "learning_rate": 3.917703349282297e-05,
-      "loss": 0.0325,
-      "step": 3453
-    },
-    {
-      "epoch": 314.0,
-      "learning_rate": 3.91578947368421e-05,
-      "loss": 0.0356,
-      "step": 3454
-    },
-    {
-      "epoch": 314.09,
-      "learning_rate": 3.913875598086125e-05,
-      "loss": 0.0337,
-      "step": 3455
-    },
-    {
-      "epoch": 314.18,
-      "learning_rate": 3.911961722488039e-05,
-      "loss": 0.0318,
-      "step": 3456
-    },
-    {
-      "epoch": 314.27,
-      "learning_rate": 3.910047846889952e-05,
-      "loss": 0.0341,
-      "step": 3457
-    },
-    {
-      "epoch": 314.36,
-      "learning_rate": 3.9081339712918665e-05,
-      "loss": 0.0386,
-      "step": 3458
-    },
-    {
-      "epoch": 314.45,
-      "learning_rate": 3.90622009569378e-05,
-      "loss": 0.0306,
-      "step": 3459
-    },
-    {
-      "epoch": 314.55,
-      "learning_rate": 3.9043062200956935e-05,
-      "loss": 0.0352,
-      "step": 3460
-    },
-    {
-      "epoch": 314.55,
-      "eval_accuracy_NAT": 0.992452127047627,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9502634233228894,
-      "eval_accuracy_concrete tiles": 0.9835809943129701,
-      "eval_accuracy_dark bricks": 0.8791809399176321,
-      "eval_accuracy_grass": 0.922098302442707,
-      "eval_accuracy_red bricks": 0.9412286931818182,
-      "eval_accuracy_road": 0.948189028364412,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5006202960879993,
-      "eval_accuracy_speedway bricks": 0.9886197287802594,
-      "eval_accuracy_steel": 0.9002209616779896,
-      "eval_accuracy_tiles": 0.16627859197370223,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804187000847677,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8900499618721945,
-      "eval_iou_concrete tiles": 0.9476521534201379,
-      "eval_iou_dark bricks": 0.8220095005038146,
-      "eval_iou_grass": 0.8406395574843393,
-      "eval_iou_red bricks": 0.84585926280517,
-      "eval_iou_road": 0.9416836795964934,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.335817583311758,
-      "eval_iou_speedway bricks": 0.976884178818636,
-      "eval_iou_steel": 0.8442695204367258,
-      "eval_iou_tiles": 0.16598304621274268,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09379665553569794,
-      "eval_mean_accuracy": 0.7055948528546158,
-      "eval_mean_iou": 0.6608667034266754,
-      "eval_overall_accuracy": 0.9758683310614692,
-      "eval_runtime": 4.3735,
-      "eval_samples_per_second": 8.231,
-      "eval_steps_per_second": 0.457,
-      "step": 3460
-    },
-    {
-      "epoch": 314.64,
-      "learning_rate": 3.902392344497608e-05,
-      "loss": 0.0337,
-      "step": 3461
-    },
-    {
-      "epoch": 314.73,
-      "learning_rate": 3.900478468899521e-05,
-      "loss": 0.0301,
-      "step": 3462
-    },
-    {
-      "epoch": 314.82,
-      "learning_rate": 3.898564593301436e-05,
-      "loss": 0.032,
-      "step": 3463
-    },
-    {
-      "epoch": 314.91,
-      "learning_rate": 3.89665071770335e-05,
-      "loss": 0.0399,
-      "step": 3464
-    },
-    {
-      "epoch": 315.0,
-      "learning_rate": 3.894736842105263e-05,
-      "loss": 0.0333,
-      "step": 3465
-    },
-    {
-      "epoch": 315.09,
-      "learning_rate": 3.8928229665071775e-05,
-      "loss": 0.0325,
-      "step": 3466
-    },
-    {
-      "epoch": 315.18,
-      "learning_rate": 3.8909090909090914e-05,
-      "loss": 0.0326,
-      "step": 3467
-    },
-    {
-      "epoch": 315.27,
-      "learning_rate": 3.8889952153110046e-05,
-      "loss": 0.0335,
-      "step": 3468
-    },
-    {
-      "epoch": 315.36,
-      "learning_rate": 3.887081339712919e-05,
-      "loss": 0.0328,
-      "step": 3469
-    },
-    {
-      "epoch": 315.45,
-      "learning_rate": 3.885167464114833e-05,
-      "loss": 0.0345,
-      "step": 3470
-    },
-    {
-      "epoch": 315.55,
-      "learning_rate": 3.883253588516746e-05,
-      "loss": 0.0322,
-      "step": 3471
-    },
-    {
-      "epoch": 315.64,
-      "learning_rate": 3.881339712918661e-05,
-      "loss": 0.0335,
-      "step": 3472
-    },
-    {
-      "epoch": 315.73,
-      "learning_rate": 3.8794258373205746e-05,
-      "loss": 0.0311,
-      "step": 3473
-    },
-    {
-      "epoch": 315.82,
-      "learning_rate": 3.877511961722488e-05,
-      "loss": 0.0316,
-      "step": 3474
-    },
-    {
-      "epoch": 315.91,
-      "learning_rate": 3.8755980861244024e-05,
-      "loss": 0.0316,
-      "step": 3475
-    },
-    {
-      "epoch": 316.0,
-      "learning_rate": 3.873684210526316e-05,
-      "loss": 0.0327,
-      "step": 3476
-    },
-    {
-      "epoch": 316.09,
-      "learning_rate": 3.8717703349282295e-05,
-      "loss": 0.0361,
-      "step": 3477
-    },
-    {
-      "epoch": 316.18,
-      "learning_rate": 3.869856459330144e-05,
-      "loss": 0.0327,
-      "step": 3478
-    },
-    {
-      "epoch": 316.27,
-      "learning_rate": 3.867942583732058e-05,
-      "loss": 0.0401,
-      "step": 3479
-    },
-    {
-      "epoch": 316.36,
-      "learning_rate": 3.866028708133971e-05,
-      "loss": 0.0336,
-      "step": 3480
-    },
-    {
-      "epoch": 316.36,
-      "eval_accuracy_NAT": 0.9937419847080404,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9459204257693463,
-      "eval_accuracy_concrete tiles": 0.9837644468904788,
-      "eval_accuracy_dark bricks": 0.8775258843000654,
-      "eval_accuracy_grass": 0.8762534208033776,
-      "eval_accuracy_red bricks": 0.9376775568181818,
-      "eval_accuracy_road": 0.9447959883287161,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.45684117663275714,
-      "eval_accuracy_speedway bricks": 0.9881001946593582,
-      "eval_accuracy_steel": 0.9053245168524504,
-      "eval_accuracy_tiles": 0.1986714148746747,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9798897172429728,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8836730832603463,
-      "eval_iou_concrete tiles": 0.9341791998141804,
-      "eval_iou_dark bricks": 0.8226824955796919,
-      "eval_iou_grass": 0.8149943274722631,
-      "eval_iou_red bricks": 0.855361192095886,
-      "eval_iou_road": 0.9370333326137691,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3086652013560332,
-      "eval_iou_speedway bricks": 0.9767254737992335,
-      "eval_iou_steel": 0.845720755414523,
-      "eval_iou_tiles": 0.19771007973829482,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09824936836957932,
-      "eval_mean_accuracy": 0.7006628469721115,
-      "eval_mean_iou": 0.6582026814143995,
-      "eval_overall_accuracy": 0.9744627210828993,
-      "eval_runtime": 3.8459,
-      "eval_samples_per_second": 9.361,
-      "eval_steps_per_second": 0.52,
-      "step": 3480
-    },
-    {
-      "epoch": 316.45,
-      "learning_rate": 3.8641148325358856e-05,
-      "loss": 0.0361,
-      "step": 3481
-    },
-    {
-      "epoch": 316.55,
-      "learning_rate": 3.8622009569377995e-05,
-      "loss": 0.0288,
-      "step": 3482
-    },
-    {
-      "epoch": 316.64,
-      "learning_rate": 3.860287081339713e-05,
-      "loss": 0.0309,
-      "step": 3483
-    },
-    {
-      "epoch": 316.73,
-      "learning_rate": 3.858373205741627e-05,
-      "loss": 0.0345,
-      "step": 3484
-    },
-    {
-      "epoch": 316.82,
-      "learning_rate": 3.856459330143541e-05,
-      "loss": 0.0336,
-      "step": 3485
-    },
-    {
-      "epoch": 316.91,
-      "learning_rate": 3.8545454545454544e-05,
-      "loss": 0.0312,
-      "step": 3486
-    },
-    {
-      "epoch": 317.0,
-      "learning_rate": 3.852631578947369e-05,
-      "loss": 0.0307,
-      "step": 3487
-    },
-    {
-      "epoch": 317.09,
-      "learning_rate": 3.850717703349283e-05,
-      "loss": 0.0334,
-      "step": 3488
-    },
-    {
-      "epoch": 317.18,
-      "learning_rate": 3.848803827751196e-05,
-      "loss": 0.0372,
-      "step": 3489
-    },
-    {
-      "epoch": 317.27,
-      "learning_rate": 3.8468899521531105e-05,
-      "loss": 0.0314,
-      "step": 3490
-    },
-    {
-      "epoch": 317.36,
-      "learning_rate": 3.8449760765550244e-05,
-      "loss": 0.034,
-      "step": 3491
-    },
-    {
-      "epoch": 317.45,
-      "learning_rate": 3.8430622009569376e-05,
-      "loss": 0.0379,
-      "step": 3492
-    },
-    {
-      "epoch": 317.55,
-      "learning_rate": 3.841148325358852e-05,
-      "loss": 0.0309,
-      "step": 3493
-    },
-    {
-      "epoch": 317.64,
-      "learning_rate": 3.839234449760766e-05,
-      "loss": 0.0334,
-      "step": 3494
-    },
-    {
-      "epoch": 317.73,
-      "learning_rate": 3.837320574162679e-05,
-      "loss": 0.0356,
-      "step": 3495
-    },
-    {
-      "epoch": 317.82,
-      "learning_rate": 3.835406698564594e-05,
-      "loss": 0.0325,
-      "step": 3496
-    },
-    {
-      "epoch": 317.91,
-      "learning_rate": 3.833492822966508e-05,
-      "loss": 0.0335,
-      "step": 3497
-    },
-    {
-      "epoch": 318.0,
-      "learning_rate": 3.831578947368421e-05,
-      "loss": 0.0349,
-      "step": 3498
-    },
-    {
-      "epoch": 318.09,
-      "learning_rate": 3.8296650717703354e-05,
-      "loss": 0.0335,
-      "step": 3499
-    },
-    {
-      "epoch": 318.18,
-      "learning_rate": 3.8277511961722486e-05,
-      "loss": 0.0346,
-      "step": 3500
-    },
-    {
-      "epoch": 318.18,
-      "eval_accuracy_NAT": 0.9925664673710125,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9499173784282587,
-      "eval_accuracy_concrete tiles": 0.9859964532501682,
-      "eval_accuracy_dark bricks": 0.8826450098148647,
-      "eval_accuracy_grass": 0.9102534976949989,
-      "eval_accuracy_red bricks": 0.912109375,
-      "eval_accuracy_road": 0.9394089628978448,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.482700631323574,
-      "eval_accuracy_speedway bricks": 0.9877978873485899,
-      "eval_accuracy_steel": 0.9062067145966625,
-      "eval_accuracy_tiles": 0.22777701684700727,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9801730281671734,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8883743217037247,
-      "eval_iou_concrete tiles": 0.9415199556217336,
-      "eval_iou_dark bricks": 0.8270937026617615,
-      "eval_iou_grass": 0.8342510642796053,
-      "eval_iou_red bricks": 0.8564521507169056,
-      "eval_iou_road": 0.9321343682396235,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32534329301150194,
-      "eval_iou_speedway bricks": 0.9766188767133748,
-      "eval_iou_steel": 0.8448660199548019,
-      "eval_iou_tiles": 0.2273566204115114,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09969106316566467,
-      "eval_mean_accuracy": 0.7059522611209986,
-      "eval_mean_iou": 0.6641679539601321,
-      "eval_overall_accuracy": 0.9747405582004123,
-      "eval_runtime": 3.5896,
-      "eval_samples_per_second": 10.029,
-      "eval_steps_per_second": 0.557,
-      "step": 3500
-    },
-    {
-      "epoch": 318.27,
-      "learning_rate": 3.8258373205741625e-05,
-      "loss": 0.0329,
-      "step": 3501
-    },
-    {
-      "epoch": 318.36,
-      "learning_rate": 3.823923444976077e-05,
-      "loss": 0.035,
-      "step": 3502
-    },
-    {
-      "epoch": 318.45,
-      "learning_rate": 3.82200956937799e-05,
-      "loss": 0.0366,
-      "step": 3503
-    },
-    {
-      "epoch": 318.55,
-      "learning_rate": 3.820095693779904e-05,
-      "loss": 0.035,
-      "step": 3504
-    },
-    {
-      "epoch": 318.64,
-      "learning_rate": 3.818181818181819e-05,
-      "loss": 0.0314,
-      "step": 3505
-    },
-    {
-      "epoch": 318.73,
-      "learning_rate": 3.816267942583732e-05,
-      "loss": 0.0357,
-      "step": 3506
-    },
-    {
-      "epoch": 318.82,
-      "learning_rate": 3.814354066985646e-05,
-      "loss": 0.0354,
-      "step": 3507
-    },
-    {
-      "epoch": 318.91,
-      "learning_rate": 3.8124401913875603e-05,
-      "loss": 0.0365,
-      "step": 3508
-    },
-    {
-      "epoch": 319.0,
-      "learning_rate": 3.8105263157894735e-05,
-      "loss": 0.0372,
-      "step": 3509
-    },
-    {
-      "epoch": 319.09,
-      "learning_rate": 3.8086124401913874e-05,
-      "loss": 0.0341,
-      "step": 3510
-    },
-    {
-      "epoch": 319.18,
-      "learning_rate": 3.806698564593302e-05,
-      "loss": 0.0331,
-      "step": 3511
-    },
-    {
-      "epoch": 319.27,
-      "learning_rate": 3.804784688995215e-05,
-      "loss": 0.0382,
-      "step": 3512
-    },
-    {
-      "epoch": 319.36,
-      "learning_rate": 3.802870813397129e-05,
-      "loss": 0.0324,
-      "step": 3513
-    },
-    {
-      "epoch": 319.45,
-      "learning_rate": 3.8009569377990436e-05,
-      "loss": 0.0332,
-      "step": 3514
-    },
-    {
-      "epoch": 319.55,
-      "learning_rate": 3.799043062200957e-05,
-      "loss": 0.0311,
-      "step": 3515
-    },
-    {
-      "epoch": 319.64,
-      "learning_rate": 3.797129186602871e-05,
-      "loss": 0.0344,
-      "step": 3516
-    },
-    {
-      "epoch": 319.73,
-      "learning_rate": 3.795215311004785e-05,
-      "loss": 0.0279,
-      "step": 3517
-    },
-    {
-      "epoch": 319.82,
-      "learning_rate": 3.7933014354066984e-05,
-      "loss": 0.0298,
-      "step": 3518
-    },
-    {
-      "epoch": 319.91,
-      "learning_rate": 3.791387559808612e-05,
-      "loss": 0.0309,
-      "step": 3519
-    },
-    {
-      "epoch": 320.0,
-      "learning_rate": 3.789473684210527e-05,
-      "loss": 0.0339,
-      "step": 3520
-    },
-    {
-      "epoch": 320.0,
-      "eval_accuracy_NAT": 0.99340108115128,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9494707390875144,
-      "eval_accuracy_concrete tiles": 0.9855378218063964,
-      "eval_accuracy_dark bricks": 0.8826065201493399,
-      "eval_accuracy_grass": 0.903756155698543,
-      "eval_accuracy_red bricks": 0.9174360795454546,
-      "eval_accuracy_road": 0.9326216736176022,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4431946626967717,
-      "eval_accuracy_speedway bricks": 0.9875504901481608,
-      "eval_accuracy_steel": 0.9070064639535651,
-      "eval_accuracy_tiles": 0.20620462950280782,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9800356196561595,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8908508132107099,
-      "eval_iou_concrete tiles": 0.9410820122040232,
-      "eval_iou_dark bricks": 0.8237597442253116,
-      "eval_iou_grass": 0.8363196501804727,
-      "eval_iou_red bricks": 0.8643358982937437,
-      "eval_iou_road": 0.9257050815803813,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3016361453017112,
-      "eval_iou_speedway bricks": 0.9767034939686416,
-      "eval_iou_steel": 0.846874518860662,
-      "eval_iou_tiles": 0.20592258241006703,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10429579764604568,
-      "eval_mean_accuracy": 0.7006758705659565,
-      "eval_mean_iou": 0.6610173507609142,
-      "eval_overall_accuracy": 0.9741978115505643,
-      "eval_runtime": 3.8417,
-      "eval_samples_per_second": 9.371,
-      "eval_steps_per_second": 0.521,
-      "step": 3520
-    },
-    {
-      "epoch": 320.09,
-      "learning_rate": 3.78755980861244e-05,
-      "loss": 0.0303,
-      "step": 3521
-    },
-    {
-      "epoch": 320.18,
-      "learning_rate": 3.785645933014354e-05,
-      "loss": 0.0336,
-      "step": 3522
-    },
-    {
-      "epoch": 320.27,
-      "learning_rate": 3.7837320574162685e-05,
-      "loss": 0.033,
-      "step": 3523
-    },
-    {
-      "epoch": 320.36,
-      "learning_rate": 3.781818181818182e-05,
-      "loss": 0.0344,
-      "step": 3524
-    },
-    {
-      "epoch": 320.45,
-      "learning_rate": 3.7799043062200956e-05,
-      "loss": 0.0337,
-      "step": 3525
-    },
-    {
-      "epoch": 320.55,
-      "learning_rate": 3.77799043062201e-05,
-      "loss": 0.0296,
-      "step": 3526
-    },
-    {
-      "epoch": 320.64,
-      "learning_rate": 3.7760765550239233e-05,
-      "loss": 0.0398,
-      "step": 3527
-    },
-    {
-      "epoch": 320.73,
-      "learning_rate": 3.774162679425837e-05,
-      "loss": 0.0295,
-      "step": 3528
-    },
-    {
-      "epoch": 320.82,
-      "learning_rate": 3.772248803827752e-05,
-      "loss": 0.0302,
-      "step": 3529
-    },
-    {
-      "epoch": 320.91,
-      "learning_rate": 3.770334928229665e-05,
-      "loss": 0.0343,
-      "step": 3530
-    },
-    {
-      "epoch": 321.0,
-      "learning_rate": 3.768421052631579e-05,
-      "loss": 0.0313,
-      "step": 3531
-    },
-    {
-      "epoch": 321.09,
-      "learning_rate": 3.7665071770334934e-05,
-      "loss": 0.0292,
-      "step": 3532
-    },
-    {
-      "epoch": 321.18,
-      "learning_rate": 3.7645933014354066e-05,
-      "loss": 0.0328,
-      "step": 3533
-    },
-    {
-      "epoch": 321.27,
-      "learning_rate": 3.762679425837321e-05,
-      "loss": 0.0354,
-      "step": 3534
-    },
-    {
-      "epoch": 321.36,
-      "learning_rate": 3.760765550239235e-05,
-      "loss": 0.0299,
-      "step": 3535
-    },
-    {
-      "epoch": 321.45,
-      "learning_rate": 3.758851674641148e-05,
-      "loss": 0.03,
-      "step": 3536
-    },
-    {
-      "epoch": 321.55,
-      "learning_rate": 3.756937799043063e-05,
-      "loss": 0.0315,
-      "step": 3537
-    },
-    {
-      "epoch": 321.64,
-      "learning_rate": 3.755023923444976e-05,
-      "loss": 0.0373,
-      "step": 3538
-    },
-    {
-      "epoch": 321.73,
-      "learning_rate": 3.75311004784689e-05,
-      "loss": 0.0317,
-      "step": 3539
-    },
-    {
-      "epoch": 321.82,
-      "learning_rate": 3.7511961722488044e-05,
-      "loss": 0.0372,
-      "step": 3540
-    },
-    {
-      "epoch": 321.82,
-      "eval_accuracy_NAT": 0.9935463004200242,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.94852783381261,
-      "eval_accuracy_concrete tiles": 0.985874151531829,
-      "eval_accuracy_dark bricks": 0.8798737538970787,
-      "eval_accuracy_grass": 0.9042978921210554,
-      "eval_accuracy_red bricks": 0.9279119318181818,
-      "eval_accuracy_road": 0.9408455030127438,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4754224905577151,
-      "eval_accuracy_speedway bricks": 0.9864571152038252,
-      "eval_accuracy_steel": 0.9148225710705099,
-      "eval_accuracy_tiles": 0.1707985207505821,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803005570462274,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8890643346902446,
-      "eval_iou_concrete tiles": 0.9394557426723384,
-      "eval_iou_dark bricks": 0.8260461082604611,
-      "eval_iou_grass": 0.8380406687871269,
-      "eval_iou_red bricks": 0.8638016528925619,
-      "eval_iou_road": 0.9336068317816556,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3210163812360387,
-      "eval_iou_speedway bricks": 0.9766041766479049,
-      "eval_iou_steel": 0.847744202926233,
-      "eval_iou_tiles": 0.17054157549234136,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09993595629930496,
-      "eval_mean_accuracy": 0.7021829280150889,
-      "eval_mean_iou": 0.6604786332640872,
-      "eval_overall_accuracy": 0.9749250411987305,
-      "eval_runtime": 3.3444,
-      "eval_samples_per_second": 10.764,
-      "eval_steps_per_second": 0.598,
-      "step": 3540
-    },
-    {
-      "epoch": 321.91,
-      "learning_rate": 3.7492822966507176e-05,
-      "loss": 0.0378,
-      "step": 3541
-    },
-    {
-      "epoch": 322.0,
-      "learning_rate": 3.7473684210526315e-05,
-      "loss": 0.0342,
-      "step": 3542
-    },
-    {
-      "epoch": 322.09,
-      "learning_rate": 3.745454545454546e-05,
-      "loss": 0.0359,
-      "step": 3543
-    },
-    {
-      "epoch": 322.18,
-      "learning_rate": 3.743540669856459e-05,
-      "loss": 0.028,
-      "step": 3544
-    },
-    {
-      "epoch": 322.27,
-      "learning_rate": 3.741626794258373e-05,
-      "loss": 0.0323,
-      "step": 3545
-    },
-    {
-      "epoch": 322.36,
-      "learning_rate": 3.739712918660288e-05,
-      "loss": 0.0278,
-      "step": 3546
-    },
-    {
-      "epoch": 322.45,
-      "learning_rate": 3.737799043062201e-05,
-      "loss": 0.0349,
-      "step": 3547
-    },
-    {
-      "epoch": 322.55,
-      "learning_rate": 3.735885167464115e-05,
-      "loss": 0.0295,
-      "step": 3548
-    },
-    {
-      "epoch": 322.64,
-      "learning_rate": 3.733971291866029e-05,
-      "loss": 0.0336,
-      "step": 3549
-    },
-    {
-      "epoch": 322.73,
-      "learning_rate": 3.7320574162679425e-05,
-      "loss": 0.0389,
-      "step": 3550
-    },
-    {
-      "epoch": 322.82,
-      "learning_rate": 3.7301435406698564e-05,
-      "loss": 0.0311,
-      "step": 3551
-    },
-    {
-      "epoch": 322.91,
-      "learning_rate": 3.728229665071771e-05,
-      "loss": 0.0364,
-      "step": 3552
-    },
-    {
-      "epoch": 323.0,
-      "learning_rate": 3.726315789473684e-05,
-      "loss": 0.0297,
-      "step": 3553
-    },
-    {
-      "epoch": 323.09,
-      "learning_rate": 3.724401913875598e-05,
-      "loss": 0.0318,
-      "step": 3554
-    },
-    {
-      "epoch": 323.18,
-      "learning_rate": 3.7224880382775126e-05,
-      "loss": 0.0378,
-      "step": 3555
-    },
-    {
-      "epoch": 323.27,
-      "learning_rate": 3.720574162679426e-05,
-      "loss": 0.0338,
-      "step": 3556
-    },
-    {
-      "epoch": 323.36,
-      "learning_rate": 3.71866028708134e-05,
-      "loss": 0.0342,
-      "step": 3557
-    },
-    {
-      "epoch": 323.45,
-      "learning_rate": 3.716746411483254e-05,
-      "loss": 0.033,
-      "step": 3558
-    },
-    {
-      "epoch": 323.55,
-      "learning_rate": 3.7148325358851674e-05,
-      "loss": 0.0378,
-      "step": 3559
-    },
-    {
-      "epoch": 323.64,
-      "learning_rate": 3.712918660287081e-05,
-      "loss": 0.0344,
-      "step": 3560
-    },
-    {
-      "epoch": 323.64,
-      "eval_accuracy_NAT": 0.9929959493881734,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9506148332546461,
-      "eval_accuracy_concrete tiles": 0.9843759554821745,
-      "eval_accuracy_dark bricks": 0.8820676648319926,
-      "eval_accuracy_grass": 0.8959796167302188,
-      "eval_accuracy_red bricks": 0.9341264204545454,
-      "eval_accuracy_road": 0.9342698252814131,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.461252171036308,
-      "eval_accuracy_speedway bricks": 0.987332056522416,
-      "eval_accuracy_steel": 0.9120193259019853,
-      "eval_accuracy_tiles": 0.10190384878783729,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980479605159768,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8835130697974178,
-      "eval_iou_concrete tiles": 0.9402196133403423,
-      "eval_iou_dark bricks": 0.8309884690695482,
-      "eval_iou_grass": 0.8306358849736736,
-      "eval_iou_red bricks": 0.8522598412441277,
-      "eval_iou_road": 0.9263070437935201,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.30979317495880165,
-      "eval_iou_speedway bricks": 0.9767678419845532,
-      "eval_iou_steel": 0.8507233113122659,
-      "eval_iou_tiles": 0.10157689944706123,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10518258064985275,
-      "eval_mean_accuracy": 0.6951490513593624,
-      "eval_mean_iou": 0.6525588273139292,
-      "eval_overall_accuracy": 0.9738921059502496,
-      "eval_runtime": 3.2802,
-      "eval_samples_per_second": 10.975,
-      "eval_steps_per_second": 0.61,
-      "step": 3560
-    },
-    {
-      "epoch": 323.73,
-      "learning_rate": 3.711004784688996e-05,
-      "loss": 0.0309,
-      "step": 3561
-    },
-    {
-      "epoch": 323.82,
-      "learning_rate": 3.709090909090909e-05,
-      "loss": 0.0357,
-      "step": 3562
-    },
-    {
-      "epoch": 323.91,
-      "learning_rate": 3.707177033492823e-05,
-      "loss": 0.0353,
-      "step": 3563
-    },
-    {
-      "epoch": 324.0,
-      "learning_rate": 3.7052631578947375e-05,
-      "loss": 0.0259,
-      "step": 3564
-    },
-    {
-      "epoch": 324.09,
-      "learning_rate": 3.703349282296651e-05,
-      "loss": 0.0394,
-      "step": 3565
-    },
-    {
-      "epoch": 324.18,
-      "learning_rate": 3.7014354066985646e-05,
-      "loss": 0.0348,
-      "step": 3566
-    },
-    {
-      "epoch": 324.27,
-      "learning_rate": 3.699521531100479e-05,
-      "loss": 0.0307,
-      "step": 3567
-    },
-    {
-      "epoch": 324.36,
-      "learning_rate": 3.697607655502392e-05,
-      "loss": 0.0305,
-      "step": 3568
-    },
-    {
-      "epoch": 324.45,
-      "learning_rate": 3.695693779904306e-05,
-      "loss": 0.0335,
-      "step": 3569
-    },
-    {
-      "epoch": 324.55,
-      "learning_rate": 3.693779904306221e-05,
-      "loss": 0.031,
-      "step": 3570
-    },
-    {
-      "epoch": 324.64,
-      "learning_rate": 3.691866028708134e-05,
-      "loss": 0.0329,
-      "step": 3571
-    },
-    {
-      "epoch": 324.73,
-      "learning_rate": 3.689952153110048e-05,
-      "loss": 0.0343,
-      "step": 3572
-    },
-    {
-      "epoch": 324.82,
-      "learning_rate": 3.6880382775119624e-05,
-      "loss": 0.0332,
-      "step": 3573
-    },
-    {
-      "epoch": 324.91,
-      "learning_rate": 3.6861244019138756e-05,
-      "loss": 0.03,
-      "step": 3574
-    },
-    {
-      "epoch": 325.0,
-      "learning_rate": 3.6842105263157895e-05,
-      "loss": 0.0326,
-      "step": 3575
-    },
-    {
-      "epoch": 325.09,
-      "learning_rate": 3.682296650717704e-05,
-      "loss": 0.0346,
-      "step": 3576
-    },
-    {
-      "epoch": 325.18,
-      "learning_rate": 3.680382775119617e-05,
-      "loss": 0.0303,
-      "step": 3577
-    },
-    {
-      "epoch": 325.27,
-      "learning_rate": 3.678468899521531e-05,
-      "loss": 0.0345,
-      "step": 3578
-    },
-    {
-      "epoch": 325.36,
-      "learning_rate": 3.676555023923445e-05,
-      "loss": 0.0352,
-      "step": 3579
-    },
-    {
-      "epoch": 325.45,
-      "learning_rate": 3.674641148325359e-05,
-      "loss": 0.0337,
-      "step": 3580
-    },
-    {
-      "epoch": 325.45,
-      "eval_accuracy_NAT": 0.9930543547076804,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9502607408043263,
-      "eval_accuracy_concrete tiles": 0.9814712896716199,
-      "eval_accuracy_dark bricks": 0.8615141834417459,
-      "eval_accuracy_grass": 0.9066990077485784,
-      "eval_accuracy_red bricks": 0.9183238636363636,
-      "eval_accuracy_road": 0.9054833993762901,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5176853306867367,
-      "eval_accuracy_speedway bricks": 0.9883681077251888,
-      "eval_accuracy_steel": 0.9115658597717828,
-      "eval_accuracy_tiles": 0.17059306944254213,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808335935909698,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8914742435509787,
-      "eval_iou_concrete tiles": 0.9548740220721659,
-      "eval_iou_dark bricks": 0.8254840494191407,
-      "eval_iou_grass": 0.8395044948837932,
-      "eval_iou_red bricks": 0.8612822647793505,
-      "eval_iou_road": 0.8978337787593537,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3435796099096132,
-      "eval_iou_speedway bricks": 0.9767898822707165,
-      "eval_iou_steel": 0.8512627040344933,
-      "eval_iou_tiles": 0.17032478632478631,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11398448050022125,
-      "eval_mean_accuracy": 0.7003860928471427,
-      "eval_mean_iou": 0.6610187253534895,
-      "eval_overall_accuracy": 0.9721248414781358,
-      "eval_runtime": 3.9223,
-      "eval_samples_per_second": 9.178,
-      "eval_steps_per_second": 0.51,
-      "step": 3580
-    },
-    {
-      "epoch": 325.55,
-      "learning_rate": 3.672727272727273e-05,
-      "loss": 0.0349,
-      "step": 3581
-    },
-    {
-      "epoch": 325.64,
-      "learning_rate": 3.6708133971291866e-05,
-      "loss": 0.0297,
-      "step": 3582
-    },
-    {
-      "epoch": 325.73,
-      "learning_rate": 3.6688995215311005e-05,
-      "loss": 0.0325,
-      "step": 3583
-    },
-    {
-      "epoch": 325.82,
-      "learning_rate": 3.6669856459330144e-05,
-      "loss": 0.0295,
-      "step": 3584
-    },
-    {
-      "epoch": 325.91,
-      "learning_rate": 3.665071770334928e-05,
-      "loss": 0.0313,
-      "step": 3585
-    },
-    {
-      "epoch": 326.0,
-      "learning_rate": 3.663157894736842e-05,
-      "loss": 0.0339,
-      "step": 3586
-    },
-    {
-      "epoch": 326.09,
-      "learning_rate": 3.661244019138756e-05,
-      "loss": 0.0323,
-      "step": 3587
-    },
-    {
-      "epoch": 326.18,
-      "learning_rate": 3.65933014354067e-05,
-      "loss": 0.0298,
-      "step": 3588
-    },
-    {
-      "epoch": 326.27,
-      "learning_rate": 3.657416267942584e-05,
-      "loss": 0.034,
-      "step": 3589
-    },
-    {
-      "epoch": 326.36,
-      "learning_rate": 3.6555023923444976e-05,
-      "loss": 0.0376,
-      "step": 3590
-    },
-    {
-      "epoch": 326.45,
-      "learning_rate": 3.6535885167464115e-05,
-      "loss": 0.0322,
-      "step": 3591
-    },
-    {
-      "epoch": 326.55,
-      "learning_rate": 3.6516746411483254e-05,
-      "loss": 0.0255,
-      "step": 3592
-    },
-    {
-      "epoch": 326.64,
-      "learning_rate": 3.649760765550239e-05,
-      "loss": 0.0349,
-      "step": 3593
-    },
-    {
-      "epoch": 326.73,
-      "learning_rate": 3.647846889952153e-05,
-      "loss": 0.0343,
-      "step": 3594
-    },
-    {
-      "epoch": 326.82,
-      "learning_rate": 3.645933014354067e-05,
-      "loss": 0.0345,
-      "step": 3595
-    },
-    {
-      "epoch": 326.91,
-      "learning_rate": 3.644019138755981e-05,
-      "loss": 0.0319,
-      "step": 3596
-    },
-    {
-      "epoch": 327.0,
-      "learning_rate": 3.642105263157895e-05,
-      "loss": 0.0313,
-      "step": 3597
-    },
-    {
-      "epoch": 327.09,
-      "learning_rate": 3.6401913875598087e-05,
-      "loss": 0.0321,
-      "step": 3598
-    },
-    {
-      "epoch": 327.18,
-      "learning_rate": 3.6382775119617225e-05,
-      "loss": 0.0363,
-      "step": 3599
-    },
-    {
-      "epoch": 327.27,
-      "learning_rate": 3.6363636363636364e-05,
-      "loss": 0.0299,
-      "step": 3600
-    },
-    {
-      "epoch": 327.27,
-      "eval_accuracy_NAT": 0.9936129989419991,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9506228808103352,
-      "eval_accuracy_concrete tiles": 0.9800953953403045,
-      "eval_accuracy_dark bricks": 0.880797505869674,
-      "eval_accuracy_grass": 0.8959236955511207,
-      "eval_accuracy_red bricks": 0.9241832386363636,
-      "eval_accuracy_road": 0.9349264256874654,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46260303807239544,
-      "eval_accuracy_speedway bricks": 0.9876078138897236,
-      "eval_accuracy_steel": 0.9117060220302091,
-      "eval_accuracy_tiles": 0.15888234488426242,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804411301260745,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8908331741259514,
-      "eval_iou_concrete tiles": 0.9512433972342572,
-      "eval_iou_dark bricks": 0.8308161487075225,
-      "eval_iou_grass": 0.8346677954512154,
-      "eval_iou_red bricks": 0.8613271553863975,
-      "eval_iou_road": 0.927378322166488,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31424397917525,
-      "eval_iou_speedway bricks": 0.9767427010620706,
-      "eval_iou_steel": 0.8499277501076062,
-      "eval_iou_tiles": 0.15868673050615595,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10279962420463562,
-      "eval_mean_accuracy": 0.698535489208758,
-      "eval_mean_iou": 0.6597160218499223,
-      "eval_overall_accuracy": 0.974432627360026,
-      "eval_runtime": 3.6536,
-      "eval_samples_per_second": 9.853,
-      "eval_steps_per_second": 0.547,
-      "step": 3600
-    },
-    {
-      "epoch": 327.36,
-      "learning_rate": 3.63444976076555e-05,
-      "loss": 0.0292,
-      "step": 3601
-    },
-    {
-      "epoch": 327.45,
-      "learning_rate": 3.632535885167464e-05,
-      "loss": 0.0324,
-      "step": 3602
-    },
-    {
-      "epoch": 327.55,
-      "learning_rate": 3.630622009569378e-05,
-      "loss": 0.0321,
-      "step": 3603
-    },
-    {
-      "epoch": 327.64,
-      "learning_rate": 3.628708133971292e-05,
-      "loss": 0.0337,
-      "step": 3604
-    },
-    {
-      "epoch": 327.73,
-      "learning_rate": 3.6267942583732065e-05,
-      "loss": 0.0331,
-      "step": 3605
-    },
-    {
-      "epoch": 327.82,
-      "learning_rate": 3.62488038277512e-05,
-      "loss": 0.0269,
-      "step": 3606
-    },
-    {
-      "epoch": 327.91,
-      "learning_rate": 3.6229665071770336e-05,
-      "loss": 0.0319,
-      "step": 3607
-    },
-    {
-      "epoch": 328.0,
-      "learning_rate": 3.621052631578948e-05,
-      "loss": 0.0344,
-      "step": 3608
-    },
-    {
-      "epoch": 328.09,
-      "learning_rate": 3.619138755980861e-05,
-      "loss": 0.0315,
-      "step": 3609
-    },
-    {
-      "epoch": 328.18,
-      "learning_rate": 3.617224880382775e-05,
-      "loss": 0.0338,
-      "step": 3610
-    },
-    {
-      "epoch": 328.27,
-      "learning_rate": 3.61531100478469e-05,
-      "loss": 0.0301,
-      "step": 3611
-    },
-    {
-      "epoch": 328.36,
-      "learning_rate": 3.613397129186603e-05,
-      "loss": 0.0407,
-      "step": 3612
-    },
-    {
-      "epoch": 328.45,
-      "learning_rate": 3.611483253588517e-05,
-      "loss": 0.0291,
-      "step": 3613
-    },
-    {
-      "epoch": 328.55,
-      "learning_rate": 3.6095693779904314e-05,
-      "loss": 0.0327,
-      "step": 3614
-    },
-    {
-      "epoch": 328.64,
-      "learning_rate": 3.6076555023923446e-05,
-      "loss": 0.0329,
-      "step": 3615
-    },
-    {
-      "epoch": 328.73,
-      "learning_rate": 3.6057416267942585e-05,
-      "loss": 0.0307,
-      "step": 3616
-    },
-    {
-      "epoch": 328.82,
-      "learning_rate": 3.603827751196172e-05,
-      "loss": 0.0393,
-      "step": 3617
-    },
-    {
-      "epoch": 328.91,
-      "learning_rate": 3.601913875598086e-05,
-      "loss": 0.0321,
-      "step": 3618
-    },
-    {
-      "epoch": 329.0,
-      "learning_rate": 3.6e-05,
-      "loss": 0.0313,
-      "step": 3619
-    },
-    {
-      "epoch": 329.09,
-      "learning_rate": 3.598086124401914e-05,
-      "loss": 0.0327,
-      "step": 3620
-    },
-    {
-      "epoch": 329.09,
-      "eval_accuracy_NAT": 0.9917262071550221,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9535159770805613,
-      "eval_accuracy_concrete tiles": 0.9845288326300985,
-      "eval_accuracy_dark bricks": 0.88557022439475,
-      "eval_accuracy_grass": 0.9199558222685126,
-      "eval_accuracy_red bricks": 0.9302201704545454,
-      "eval_accuracy_road": 0.8882715205922221,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.507733024563725,
-      "eval_accuracy_speedway bricks": 0.9880718344924797,
-      "eval_accuracy_steel": 0.9097602400897038,
-      "eval_accuracy_tiles": 0.10190384878783729,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808498403615272,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8871151297646293,
-      "eval_iou_concrete tiles": 0.9519304676875776,
-      "eval_iou_dark bricks": 0.8328386302758272,
-      "eval_iou_grass": 0.8374615416432019,
-      "eval_iou_red bricks": 0.8455455132343448,
-      "eval_iou_road": 0.8816891333127691,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3361073090610457,
-      "eval_iou_speedway bricks": 0.9768209589689932,
-      "eval_iou_steel": 0.8499168130141418,
-      "eval_iou_tiles": 0.10172967799275313,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12743033468723297,
-      "eval_mean_accuracy": 0.6970198232699585,
-      "eval_mean_iou": 0.6524619242551394,
-      "eval_overall_accuracy": 0.9703422122531467,
-      "eval_runtime": 3.7034,
-      "eval_samples_per_second": 9.721,
-      "eval_steps_per_second": 0.54,
-      "step": 3620
-    },
-    {
-      "epoch": 329.18,
-      "learning_rate": 3.596172248803828e-05,
-      "loss": 0.0344,
-      "step": 3621
-    },
-    {
-      "epoch": 329.27,
-      "learning_rate": 3.594258373205742e-05,
-      "loss": 0.0298,
-      "step": 3622
-    },
-    {
-      "epoch": 329.36,
-      "learning_rate": 3.5923444976076556e-05,
-      "loss": 0.0329,
-      "step": 3623
-    },
-    {
-      "epoch": 329.45,
-      "learning_rate": 3.5904306220095695e-05,
-      "loss": 0.0327,
-      "step": 3624
-    },
-    {
-      "epoch": 329.55,
-      "learning_rate": 3.5885167464114834e-05,
-      "loss": 0.0276,
-      "step": 3625
-    },
-    {
-      "epoch": 329.64,
-      "learning_rate": 3.586602870813397e-05,
-      "loss": 0.0341,
-      "step": 3626
-    },
-    {
-      "epoch": 329.73,
-      "learning_rate": 3.584688995215311e-05,
-      "loss": 0.0309,
-      "step": 3627
-    },
-    {
-      "epoch": 329.82,
-      "learning_rate": 3.582775119617225e-05,
-      "loss": 0.0339,
-      "step": 3628
-    },
-    {
-      "epoch": 329.91,
-      "learning_rate": 3.580861244019139e-05,
-      "loss": 0.0388,
-      "step": 3629
-    },
-    {
-      "epoch": 330.0,
-      "learning_rate": 3.578947368421053e-05,
-      "loss": 0.0319,
-      "step": 3630
-    },
-    {
-      "epoch": 330.09,
-      "learning_rate": 3.5770334928229666e-05,
-      "loss": 0.031,
-      "step": 3631
-    },
-    {
-      "epoch": 330.18,
-      "learning_rate": 3.5751196172248805e-05,
-      "loss": 0.0335,
-      "step": 3632
-    },
-    {
-      "epoch": 330.27,
-      "learning_rate": 3.5732057416267944e-05,
-      "loss": 0.0309,
-      "step": 3633
-    },
-    {
-      "epoch": 330.36,
-      "learning_rate": 3.571291866028708e-05,
-      "loss": 0.0355,
-      "step": 3634
-    },
-    {
-      "epoch": 330.45,
-      "learning_rate": 3.569377990430622e-05,
-      "loss": 0.0331,
-      "step": 3635
-    },
-    {
-      "epoch": 330.55,
-      "learning_rate": 3.567464114832536e-05,
-      "loss": 0.0337,
-      "step": 3636
-    },
-    {
-      "epoch": 330.64,
-      "learning_rate": 3.56555023923445e-05,
-      "loss": 0.0327,
-      "step": 3637
-    },
-    {
-      "epoch": 330.73,
-      "learning_rate": 3.563636363636364e-05,
-      "loss": 0.0341,
-      "step": 3638
-    },
-    {
-      "epoch": 330.82,
-      "learning_rate": 3.5617224880382776e-05,
-      "loss": 0.0297,
-      "step": 3639
-    },
-    {
-      "epoch": 330.91,
-      "learning_rate": 3.5598086124401915e-05,
-      "loss": 0.036,
-      "step": 3640
-    },
-    {
-      "epoch": 330.91,
-      "eval_accuracy_NAT": 0.9931993975253084,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9486633010000429,
-      "eval_accuracy_concrete tiles": 0.986730263560203,
-      "eval_accuracy_dark bricks": 0.8781032292829375,
-      "eval_accuracy_grass": 0.9053219487132886,
-      "eval_accuracy_red bricks": 0.9147727272727273,
-      "eval_accuracy_road": 0.8852920299835426,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47106663358420864,
-      "eval_accuracy_speedway bricks": 0.9882118251034543,
-      "eval_accuracy_steel": 0.910271420091023,
-      "eval_accuracy_tiles": 0.20189015203396796,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805989949678708,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8895279154556534,
-      "eval_iou_concrete tiles": 0.9502944640753828,
-      "eval_iou_dark bricks": 0.8298715943399658,
-      "eval_iou_grass": 0.8365024414188648,
-      "eval_iou_red bricks": 0.8486246087959151,
-      "eval_iou_road": 0.8788994504214875,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31780898353947734,
-      "eval_iou_speedway bricks": 0.9765282794758993,
-      "eval_iou_steel": 0.8481600983329493,
-      "eval_iou_tiles": 0.20157264957264956,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12760543823242188,
-      "eval_mean_accuracy": 0.6987325329346695,
-      "eval_mean_iou": 0.6583376523381628,
-      "eval_overall_accuracy": 0.9701615439520942,
-      "eval_runtime": 3.5794,
-      "eval_samples_per_second": 10.058,
-      "eval_steps_per_second": 0.559,
-      "step": 3640
-    },
-    {
-      "epoch": 331.0,
-      "learning_rate": 3.5578947368421054e-05,
-      "loss": 0.0326,
-      "step": 3641
-    },
-    {
-      "epoch": 331.09,
-      "learning_rate": 3.555980861244019e-05,
-      "loss": 0.0355,
-      "step": 3642
-    },
-    {
-      "epoch": 331.18,
-      "learning_rate": 3.554066985645933e-05,
-      "loss": 0.0332,
-      "step": 3643
-    },
-    {
-      "epoch": 331.27,
-      "learning_rate": 3.552153110047847e-05,
-      "loss": 0.0313,
-      "step": 3644
-    },
-    {
-      "epoch": 331.36,
-      "learning_rate": 3.550239234449761e-05,
-      "loss": 0.0332,
-      "step": 3645
-    },
-    {
-      "epoch": 331.45,
-      "learning_rate": 3.548325358851675e-05,
-      "loss": 0.0276,
-      "step": 3646
-    },
-    {
-      "epoch": 331.55,
-      "learning_rate": 3.546411483253589e-05,
-      "loss": 0.0322,
-      "step": 3647
-    },
-    {
-      "epoch": 331.64,
-      "learning_rate": 3.5444976076555025e-05,
-      "loss": 0.0322,
-      "step": 3648
-    },
-    {
-      "epoch": 331.73,
-      "learning_rate": 3.5425837320574164e-05,
-      "loss": 0.0349,
-      "step": 3649
-    },
-    {
-      "epoch": 331.82,
-      "learning_rate": 3.54066985645933e-05,
-      "loss": 0.0332,
-      "step": 3650
-    },
-    {
-      "epoch": 331.91,
-      "learning_rate": 3.538755980861244e-05,
-      "loss": 0.0351,
-      "step": 3651
-    },
-    {
-      "epoch": 332.0,
-      "learning_rate": 3.536842105263158e-05,
-      "loss": 0.0348,
-      "step": 3652
-    },
-    {
-      "epoch": 332.09,
-      "learning_rate": 3.534928229665072e-05,
-      "loss": 0.031,
-      "step": 3653
-    },
-    {
-      "epoch": 332.18,
-      "learning_rate": 3.533014354066986e-05,
-      "loss": 0.0334,
-      "step": 3654
-    },
-    {
-      "epoch": 332.27,
-      "learning_rate": 3.5311004784689e-05,
-      "loss": 0.0336,
-      "step": 3655
-    },
-    {
-      "epoch": 332.36,
-      "learning_rate": 3.5291866028708136e-05,
-      "loss": 0.0327,
-      "step": 3656
-    },
-    {
-      "epoch": 332.45,
-      "learning_rate": 3.5272727272727274e-05,
-      "loss": 0.0304,
-      "step": 3657
-    },
-    {
-      "epoch": 332.55,
-      "learning_rate": 3.525358851674641e-05,
-      "loss": 0.0321,
-      "step": 3658
-    },
-    {
-      "epoch": 332.64,
-      "learning_rate": 3.523444976076555e-05,
-      "loss": 0.0355,
-      "step": 3659
-    },
-    {
-      "epoch": 332.73,
-      "learning_rate": 3.521531100478469e-05,
-      "loss": 0.0338,
-      "step": 3660
-    },
-    {
-      "epoch": 332.73,
-      "eval_accuracy_NAT": 0.9931409922058013,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9479913301000042,
-      "eval_accuracy_concrete tiles": 0.98715831957439,
-      "eval_accuracy_dark bricks": 0.8854162657326508,
-      "eval_accuracy_grass": 0.90257831586379,
-      "eval_accuracy_red bricks": 0.9343039772727273,
-      "eval_accuracy_road": 0.922016911994989,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4772971631792242,
-      "eval_accuracy_speedway bricks": 0.9887488580507273,
-      "eval_accuracy_steel": 0.9093562429918871,
-      "eval_accuracy_tiles": 0.1739487741405287,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804854748118699,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8889604403095549,
-      "eval_iou_concrete tiles": 0.946886822887644,
-      "eval_iou_dark bricks": 0.8333574844225474,
-      "eval_iou_grass": 0.8354945161603416,
-      "eval_iou_red bricks": 0.8491205421978376,
-      "eval_iou_road": 0.9144885733303749,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3204449544680536,
-      "eval_iou_speedway bricks": 0.9769640094321342,
-      "eval_iou_steel": 0.8509551584729809,
-      "eval_iou_tiles": 0.173746494288255,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10963268578052521,
-      "eval_mean_accuracy": 0.7016890116235939,
-      "eval_mean_iou": 0.6593003439062765,
-      "eval_overall_accuracy": 0.9733049604627821,
-      "eval_runtime": 3.8497,
-      "eval_samples_per_second": 9.351,
-      "eval_steps_per_second": 0.52,
-      "step": 3660
-    },
-    {
-      "epoch": 332.82,
-      "learning_rate": 3.519617224880383e-05,
-      "loss": 0.0345,
-      "step": 3661
-    },
-    {
-      "epoch": 332.91,
-      "learning_rate": 3.517703349282297e-05,
-      "loss": 0.0306,
-      "step": 3662
-    },
-    {
-      "epoch": 333.0,
-      "learning_rate": 3.515789473684211e-05,
-      "loss": 0.0296,
-      "step": 3663
-    },
-    {
-      "epoch": 333.09,
-      "learning_rate": 3.5138755980861246e-05,
-      "loss": 0.0325,
-      "step": 3664
-    },
-    {
-      "epoch": 333.18,
-      "learning_rate": 3.5119617224880385e-05,
-      "loss": 0.0285,
-      "step": 3665
-    },
-    {
-      "epoch": 333.27,
-      "learning_rate": 3.5100478468899523e-05,
-      "loss": 0.0345,
-      "step": 3666
-    },
-    {
-      "epoch": 333.36,
-      "learning_rate": 3.508133971291866e-05,
-      "loss": 0.0334,
-      "step": 3667
-    },
-    {
-      "epoch": 333.45,
-      "learning_rate": 3.50622009569378e-05,
-      "loss": 0.0374,
-      "step": 3668
-    },
-    {
-      "epoch": 333.55,
-      "learning_rate": 3.504306220095694e-05,
-      "loss": 0.036,
-      "step": 3669
-    },
-    {
-      "epoch": 333.64,
-      "learning_rate": 3.502392344497608e-05,
-      "loss": 0.033,
-      "step": 3670
-    },
-    {
-      "epoch": 333.73,
-      "learning_rate": 3.500478468899522e-05,
-      "loss": 0.0324,
-      "step": 3671
-    },
-    {
-      "epoch": 333.82,
-      "learning_rate": 3.4985645933014356e-05,
-      "loss": 0.0311,
-      "step": 3672
-    },
-    {
-      "epoch": 333.91,
-      "learning_rate": 3.4966507177033495e-05,
-      "loss": 0.0305,
-      "step": 3673
-    },
-    {
-      "epoch": 334.0,
-      "learning_rate": 3.4947368421052634e-05,
-      "loss": 0.0274,
-      "step": 3674
-    },
-    {
-      "epoch": 334.09,
-      "learning_rate": 3.492822966507177e-05,
-      "loss": 0.037,
-      "step": 3675
-    },
-    {
-      "epoch": 334.18,
-      "learning_rate": 3.490909090909091e-05,
-      "loss": 0.0313,
-      "step": 3676
-    },
-    {
-      "epoch": 334.27,
-      "learning_rate": 3.488995215311005e-05,
-      "loss": 0.0285,
-      "step": 3677
-    },
-    {
-      "epoch": 334.36,
-      "learning_rate": 3.487081339712919e-05,
-      "loss": 0.0364,
-      "step": 3678
-    },
-    {
-      "epoch": 334.45,
-      "learning_rate": 3.485167464114833e-05,
-      "loss": 0.0317,
-      "step": 3679
-    },
-    {
-      "epoch": 334.55,
-      "learning_rate": 3.4832535885167466e-05,
-      "loss": 0.034,
-      "step": 3680
-    },
-    {
-      "epoch": 334.55,
-      "eval_accuracy_NAT": 0.9929982432526857,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9467037211897507,
-      "eval_accuracy_concrete tiles": 0.984742860637192,
-      "eval_accuracy_dark bricks": 0.8764481736653709,
-      "eval_accuracy_grass": 0.9051541851759944,
-      "eval_accuracy_red bricks": 0.9343039772727273,
-      "eval_accuracy_road": 0.9283870242216625,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5120613128222093,
-      "eval_accuracy_speedway bricks": 0.9898132694203785,
-      "eval_accuracy_steel": 0.8996273332893608,
-      "eval_accuracy_tiles": 0.162648952198329,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806515301085883,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.888618071441251,
-      "eval_iou_concrete tiles": 0.9497788263049248,
-      "eval_iou_dark bricks": 0.8348977047737772,
-      "eval_iou_grass": 0.83821249518557,
-      "eval_iou_red bricks": 0.853251175612129,
-      "eval_iou_road": 0.9205229410156681,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33959849343620874,
-      "eval_iou_speedway bricks": 0.9766397873789598,
-      "eval_iou_steel": 0.8495460844921285,
-      "eval_iou_tiles": 0.16204967248908297,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10704771429300308,
-      "eval_mean_accuracy": 0.7025299271650508,
-      "eval_mean_iou": 0.6610589832490992,
-      "eval_overall_accuracy": 0.9738987816704644,
-      "eval_runtime": 3.7142,
-      "eval_samples_per_second": 9.692,
-      "eval_steps_per_second": 0.538,
-      "step": 3680
-    },
-    {
-      "epoch": 334.64,
-      "learning_rate": 3.4813397129186605e-05,
-      "loss": 0.0316,
-      "step": 3681
-    },
-    {
-      "epoch": 334.73,
-      "learning_rate": 3.4794258373205744e-05,
-      "loss": 0.0271,
-      "step": 3682
-    },
-    {
-      "epoch": 334.82,
-      "learning_rate": 3.477511961722488e-05,
-      "loss": 0.0306,
-      "step": 3683
-    },
-    {
-      "epoch": 334.91,
-      "learning_rate": 3.475598086124402e-05,
-      "loss": 0.0312,
-      "step": 3684
-    },
-    {
-      "epoch": 335.0,
-      "learning_rate": 3.473684210526316e-05,
-      "loss": 0.0319,
-      "step": 3685
-    },
-    {
-      "epoch": 335.09,
-      "learning_rate": 3.47177033492823e-05,
-      "loss": 0.0347,
-      "step": 3686
-    },
-    {
-      "epoch": 335.18,
-      "learning_rate": 3.469856459330144e-05,
-      "loss": 0.0302,
-      "step": 3687
-    },
-    {
-      "epoch": 335.27,
-      "learning_rate": 3.4679425837320577e-05,
-      "loss": 0.0372,
-      "step": 3688
-    },
-    {
-      "epoch": 335.36,
-      "learning_rate": 3.4660287081339715e-05,
-      "loss": 0.0327,
-      "step": 3689
-    },
-    {
-      "epoch": 335.45,
-      "learning_rate": 3.4641148325358854e-05,
-      "loss": 0.0304,
-      "step": 3690
-    },
-    {
-      "epoch": 335.55,
-      "learning_rate": 3.462200956937799e-05,
-      "loss": 0.0315,
-      "step": 3691
-    },
-    {
-      "epoch": 335.64,
-      "learning_rate": 3.460287081339713e-05,
-      "loss": 0.0322,
-      "step": 3692
-    },
-    {
-      "epoch": 335.73,
-      "learning_rate": 3.458373205741627e-05,
-      "loss": 0.0289,
-      "step": 3693
-    },
-    {
-      "epoch": 335.82,
-      "learning_rate": 3.456459330143541e-05,
-      "loss": 0.032,
-      "step": 3694
-    },
-    {
-      "epoch": 335.91,
-      "learning_rate": 3.454545454545455e-05,
-      "loss": 0.0289,
-      "step": 3695
-    },
-    {
-      "epoch": 336.0,
-      "learning_rate": 3.452631578947369e-05,
-      "loss": 0.032,
-      "step": 3696
-    },
-    {
-      "epoch": 336.09,
-      "learning_rate": 3.4507177033492826e-05,
-      "loss": 0.0329,
-      "step": 3697
-    },
-    {
-      "epoch": 336.18,
-      "learning_rate": 3.4488038277511964e-05,
-      "loss": 0.0321,
-      "step": 3698
-    },
-    {
-      "epoch": 336.27,
-      "learning_rate": 3.44688995215311e-05,
-      "loss": 0.0297,
-      "step": 3699
-    },
-    {
-      "epoch": 336.36,
-      "learning_rate": 3.444976076555024e-05,
-      "loss": 0.0314,
-      "step": 3700
-    },
-    {
-      "epoch": 336.36,
-      "eval_accuracy_NAT": 0.992787560619781,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9507543242199236,
-      "eval_accuracy_concrete tiles": 0.9848651623555311,
-      "eval_accuracy_dark bricks": 0.8785651052692353,
-      "eval_accuracy_grass": 0.9107393129384133,
-      "eval_accuracy_red bricks": 0.9307528409090909,
-      "eval_accuracy_road": 0.9141183597807463,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48060540898188736,
-      "eval_accuracy_speedway bricks": 0.9869857004418152,
-      "eval_accuracy_steel": 0.910279664929754,
-      "eval_accuracy_tiles": 0.17038761813450212,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980445858951377,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8884193111746127,
-      "eval_iou_concrete tiles": 0.9470202569605739,
-      "eval_iou_dark bricks": 0.8312454479242535,
-      "eval_iou_grass": 0.8383539078762374,
-      "eval_iou_red bricks": 0.8596261069203017,
-      "eval_iou_road": 0.906042729859365,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3238528701467583,
-      "eval_iou_speedway bricks": 0.976842711270844,
-      "eval_iou_steel": 0.8519967588841301,
-      "eval_iou_tiles": 0.16995696427351595,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1142367273569107,
-      "eval_mean_accuracy": 0.7008339275831292,
-      "eval_mean_iou": 0.6595233018647669,
-      "eval_overall_accuracy": 0.9725467893812392,
-      "eval_runtime": 3.7494,
-      "eval_samples_per_second": 9.601,
-      "eval_steps_per_second": 0.533,
-      "step": 3700
-    },
-    {
-      "epoch": 336.45,
-      "learning_rate": 3.443062200956938e-05,
-      "loss": 0.0342,
-      "step": 3701
-    },
-    {
-      "epoch": 336.55,
-      "learning_rate": 3.441148325358852e-05,
-      "loss": 0.0323,
-      "step": 3702
-    },
-    {
-      "epoch": 336.64,
-      "learning_rate": 3.439234449760766e-05,
-      "loss": 0.0323,
-      "step": 3703
-    },
-    {
-      "epoch": 336.73,
-      "learning_rate": 3.43732057416268e-05,
-      "loss": 0.0359,
-      "step": 3704
-    },
-    {
-      "epoch": 336.82,
-      "learning_rate": 3.4354066985645936e-05,
-      "loss": 0.0317,
-      "step": 3705
-    },
-    {
-      "epoch": 336.91,
-      "learning_rate": 3.4334928229665075e-05,
-      "loss": 0.0345,
-      "step": 3706
-    },
-    {
-      "epoch": 337.0,
-      "learning_rate": 3.431578947368421e-05,
-      "loss": 0.0301,
-      "step": 3707
-    },
-    {
-      "epoch": 337.09,
-      "learning_rate": 3.429665071770335e-05,
-      "loss": 0.0286,
-      "step": 3708
-    },
-    {
-      "epoch": 337.18,
-      "learning_rate": 3.427751196172249e-05,
-      "loss": 0.0321,
-      "step": 3709
-    },
-    {
-      "epoch": 337.27,
-      "learning_rate": 3.425837320574163e-05,
-      "loss": 0.0277,
-      "step": 3710
-    },
-    {
-      "epoch": 337.36,
-      "learning_rate": 3.423923444976077e-05,
-      "loss": 0.032,
-      "step": 3711
-    },
-    {
-      "epoch": 337.45,
-      "learning_rate": 3.422009569377991e-05,
-      "loss": 0.0329,
-      "step": 3712
-    },
-    {
-      "epoch": 337.55,
-      "learning_rate": 3.4200956937799046e-05,
-      "loss": 0.0354,
-      "step": 3713
-    },
-    {
-      "epoch": 337.64,
-      "learning_rate": 3.4181818181818185e-05,
-      "loss": 0.0443,
-      "step": 3714
-    },
-    {
-      "epoch": 337.73,
-      "learning_rate": 3.4162679425837324e-05,
-      "loss": 0.0286,
-      "step": 3715
-    },
-    {
-      "epoch": 337.82,
-      "learning_rate": 3.414354066985646e-05,
-      "loss": 0.0302,
-      "step": 3716
-    },
-    {
-      "epoch": 337.91,
-      "learning_rate": 3.41244019138756e-05,
-      "loss": 0.0329,
-      "step": 3717
-    },
-    {
-      "epoch": 338.0,
-      "learning_rate": 3.410526315789474e-05,
-      "loss": 0.035,
-      "step": 3718
-    },
-    {
-      "epoch": 338.09,
-      "learning_rate": 3.408612440191388e-05,
-      "loss": 0.0359,
-      "step": 3719
-    },
-    {
-      "epoch": 338.18,
-      "learning_rate": 3.406698564593301e-05,
-      "loss": 0.0296,
-      "step": 3720
-    },
-    {
-      "epoch": 338.18,
-      "eval_accuracy_NAT": 0.993375319288295,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9511540194858148,
-      "eval_accuracy_concrete tiles": 0.9853849446584725,
-      "eval_accuracy_dark bricks": 0.8774873946345406,
-      "eval_accuracy_grass": 0.8980417102094598,
-      "eval_accuracy_red bricks": 0.9621803977272727,
-      "eval_accuracy_road": 0.926845282936733,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5043145039009732,
-      "eval_accuracy_speedway bricks": 0.9886776559296282,
-      "eval_accuracy_steel": 0.9035601213640261,
-      "eval_accuracy_tiles": 0.1762772222983153,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807711471041393,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.889642436878856,
-      "eval_iou_concrete tiles": 0.9500619067271977,
-      "eval_iou_dark bricks": 0.8337173157798501,
-      "eval_iou_grass": 0.8395386450147849,
-      "eval_iou_red bricks": 0.8382057231245166,
-      "eval_iou_road": 0.9195201391596437,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3362745638706594,
-      "eval_iou_speedway bricks": 0.9769437483267835,
-      "eval_iou_steel": 0.8501885153062017,
-      "eval_iou_tiles": 0.1757836508912108,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10652291029691696,
-      "eval_mean_accuracy": 0.7051768132641177,
-      "eval_mean_iou": 0.6608190609372188,
-      "eval_overall_accuracy": 0.9739902284410265,
-      "eval_runtime": 3.6209,
-      "eval_samples_per_second": 9.942,
-      "eval_steps_per_second": 0.552,
-      "step": 3720
-    },
-    {
-      "epoch": 338.27,
-      "learning_rate": 3.4047846889952156e-05,
-      "loss": 0.0359,
-      "step": 3721
-    },
-    {
-      "epoch": 338.36,
-      "learning_rate": 3.4028708133971295e-05,
-      "loss": 0.0312,
-      "step": 3722
-    },
-    {
-      "epoch": 338.45,
-      "learning_rate": 3.400956937799043e-05,
-      "loss": 0.0318,
-      "step": 3723
-    },
-    {
-      "epoch": 338.55,
-      "learning_rate": 3.399043062200957e-05,
-      "loss": 0.0336,
-      "step": 3724
-    },
-    {
-      "epoch": 338.64,
-      "learning_rate": 3.397129186602871e-05,
-      "loss": 0.0361,
-      "step": 3725
-    },
-    {
-      "epoch": 338.73,
-      "learning_rate": 3.395215311004784e-05,
-      "loss": 0.0324,
-      "step": 3726
-    },
-    {
-      "epoch": 338.82,
-      "learning_rate": 3.393301435406699e-05,
-      "loss": 0.0322,
-      "step": 3727
-    },
-    {
-      "epoch": 338.91,
-      "learning_rate": 3.391387559808613e-05,
-      "loss": 0.0308,
-      "step": 3728
-    },
-    {
-      "epoch": 339.0,
-      "learning_rate": 3.389473684210526e-05,
-      "loss": 0.034,
-      "step": 3729
-    },
-    {
-      "epoch": 339.09,
-      "learning_rate": 3.3875598086124405e-05,
-      "loss": 0.0312,
-      "step": 3730
-    },
-    {
-      "epoch": 339.18,
-      "learning_rate": 3.3856459330143544e-05,
-      "loss": 0.0295,
-      "step": 3731
-    },
-    {
-      "epoch": 339.27,
-      "learning_rate": 3.3837320574162676e-05,
-      "loss": 0.0362,
-      "step": 3732
-    },
-    {
-      "epoch": 339.36,
-      "learning_rate": 3.381818181818182e-05,
-      "loss": 0.0302,
-      "step": 3733
-    },
-    {
-      "epoch": 339.45,
-      "learning_rate": 3.379904306220096e-05,
-      "loss": 0.0306,
-      "step": 3734
-    },
-    {
-      "epoch": 339.55,
-      "learning_rate": 3.377990430622009e-05,
-      "loss": 0.0376,
-      "step": 3735
-    },
-    {
-      "epoch": 339.64,
-      "learning_rate": 3.376076555023924e-05,
-      "loss": 0.0318,
-      "step": 3736
-    },
-    {
-      "epoch": 339.73,
-      "learning_rate": 3.3741626794258377e-05,
-      "loss": 0.0315,
-      "step": 3737
-    },
-    {
-      "epoch": 339.82,
-      "learning_rate": 3.372248803827751e-05,
-      "loss": 0.0316,
-      "step": 3738
-    },
-    {
-      "epoch": 339.91,
-      "learning_rate": 3.3703349282296654e-05,
-      "loss": 0.03,
-      "step": 3739
-    },
-    {
-      "epoch": 340.0,
-      "learning_rate": 3.368421052631579e-05,
-      "loss": 0.0309,
-      "step": 3740
-    },
-    {
-      "epoch": 340.0,
-      "eval_accuracy_NAT": 0.9931582844152022,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9509461242971802,
-      "eval_accuracy_concrete tiles": 0.9798813673332111,
-      "eval_accuracy_dark bricks": 0.8736769177475848,
-      "eval_accuracy_grass": 0.8999814761094238,
-      "eval_accuracy_red bricks": 0.9488636363636364,
-      "eval_accuracy_road": 0.9294874042760043,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5073746312684366,
-      "eval_accuracy_speedway bricks": 0.9878497804198995,
-      "eval_accuracy_steel": 0.9064210804036673,
-      "eval_accuracy_tiles": 0.1805916997671552,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804314964766591,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8878641019861246,
-      "eval_iou_concrete tiles": 0.9562856205054755,
-      "eval_iou_dark bricks": 0.8322273143904675,
-      "eval_iou_grass": 0.8380770119543955,
-      "eval_iou_red bricks": 0.8588878174220508,
-      "eval_iou_road": 0.9208852873805576,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3386200551977921,
-      "eval_iou_speedway bricks": 0.9770169888441893,
-      "eval_iou_steel": 0.8498148677792636,
-      "eval_iou_tiles": 0.18025839086745504,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10638141632080078,
-      "eval_mean_accuracy": 0.7044794155693386,
-      "eval_mean_iou": 0.6631053040618794,
-      "eval_overall_accuracy": 0.9740061230129666,
-      "eval_runtime": 3.5247,
-      "eval_samples_per_second": 10.214,
-      "eval_steps_per_second": 0.567,
-      "step": 3740
-    },
-    {
-      "epoch": 340.09,
-      "learning_rate": 3.3665071770334925e-05,
-      "loss": 0.0334,
-      "step": 3741
-    },
-    {
-      "epoch": 340.18,
-      "learning_rate": 3.364593301435407e-05,
-      "loss": 0.0285,
-      "step": 3742
-    },
-    {
-      "epoch": 340.27,
-      "learning_rate": 3.362679425837321e-05,
-      "loss": 0.0318,
-      "step": 3743
-    },
-    {
-      "epoch": 340.36,
-      "learning_rate": 3.360765550239234e-05,
-      "loss": 0.0285,
-      "step": 3744
-    },
-    {
-      "epoch": 340.45,
-      "learning_rate": 3.358851674641149e-05,
-      "loss": 0.0304,
-      "step": 3745
-    },
-    {
-      "epoch": 340.55,
-      "learning_rate": 3.3569377990430626e-05,
-      "loss": 0.0324,
-      "step": 3746
-    },
-    {
-      "epoch": 340.64,
-      "learning_rate": 3.3550239234449764e-05,
-      "loss": 0.0349,
-      "step": 3747
-    },
-    {
-      "epoch": 340.73,
-      "learning_rate": 3.35311004784689e-05,
-      "loss": 0.0333,
-      "step": 3748
-    },
-    {
-      "epoch": 340.82,
-      "learning_rate": 3.351196172248804e-05,
-      "loss": 0.0265,
-      "step": 3749
-    },
-    {
-      "epoch": 340.91,
-      "learning_rate": 3.349282296650718e-05,
-      "loss": 0.0358,
-      "step": 3750
-    },
-    {
-      "epoch": 341.0,
-      "learning_rate": 3.347368421052632e-05,
-      "loss": 0.0393,
-      "step": 3751
-    },
-    {
-      "epoch": 341.09,
-      "learning_rate": 3.345454545454546e-05,
-      "loss": 0.0313,
-      "step": 3752
-    },
-    {
-      "epoch": 341.18,
-      "learning_rate": 3.34354066985646e-05,
-      "loss": 0.0323,
-      "step": 3753
-    },
-    {
-      "epoch": 341.27,
-      "learning_rate": 3.3416267942583736e-05,
-      "loss": 0.0331,
-      "step": 3754
-    },
-    {
-      "epoch": 341.36,
-      "learning_rate": 3.3397129186602875e-05,
-      "loss": 0.0339,
-      "step": 3755
-    },
-    {
-      "epoch": 341.45,
-      "learning_rate": 3.337799043062201e-05,
-      "loss": 0.0327,
-      "step": 3756
-    },
-    {
-      "epoch": 341.55,
-      "learning_rate": 3.335885167464115e-05,
-      "loss": 0.0317,
-      "step": 3757
-    },
-    {
-      "epoch": 341.64,
-      "learning_rate": 3.3339712918660284e-05,
-      "loss": 0.0319,
-      "step": 3758
-    },
-    {
-      "epoch": 341.73,
-      "learning_rate": 3.332057416267943e-05,
-      "loss": 0.0351,
-      "step": 3759
-    },
-    {
-      "epoch": 341.82,
-      "learning_rate": 3.330143540669857e-05,
-      "loss": 0.029,
-      "step": 3760
-    },
-    {
-      "epoch": 341.82,
-      "eval_accuracy_NAT": 0.9933350884337705,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9478035538005923,
-      "eval_accuracy_concrete tiles": 0.9790252553048371,
-      "eval_accuracy_dark bricks": 0.880797505869674,
-      "eval_accuracy_grass": 0.9059475669044482,
-      "eval_accuracy_red bricks": 0.9518821022727273,
-      "eval_accuracy_road": 0.9473558834661246,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.582940479144267,
-      "eval_accuracy_speedway bricks": 0.9880127005274991,
-      "eval_accuracy_steel": 0.9048628058835169,
-      "eval_accuracy_tiles": 0.4498698808382413,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980218616095378,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8992909028898379,
-      "eval_iou_concrete tiles": 0.94443133553563,
-      "eval_iou_dark bricks": 0.8312688437647572,
-      "eval_iou_grass": 0.8408331549206226,
-      "eval_iou_red bricks": 0.864816905952573,
-      "eval_iou_road": 0.9387064810144176,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3931612807258934,
-      "eval_iou_speedway bricks": 0.9769857138509952,
-      "eval_iou_steel": 0.8475088033607215,
-      "eval_iou_tiles": 0.4479067230328651,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09146282076835632,
-      "eval_mean_accuracy": 0.7332179094188999,
-      "eval_mean_iou": 0.6896252893187454,
-      "eval_overall_accuracy": 0.9763448503282335,
-      "eval_runtime": 3.6264,
-      "eval_samples_per_second": 9.927,
-      "eval_steps_per_second": 0.552,
-      "step": 3760
-    },
-    {
-      "epoch": 341.91,
-      "learning_rate": 3.32822966507177e-05,
-      "loss": 0.0336,
-      "step": 3761
-    },
-    {
-      "epoch": 342.0,
-      "learning_rate": 3.3263157894736846e-05,
-      "loss": 0.0364,
-      "step": 3762
-    },
-    {
-      "epoch": 342.09,
-      "learning_rate": 3.3244019138755985e-05,
-      "loss": 0.031,
-      "step": 3763
-    },
-    {
-      "epoch": 342.18,
-      "learning_rate": 3.322488038277512e-05,
-      "loss": 0.0321,
-      "step": 3764
-    },
-    {
-      "epoch": 342.27,
-      "learning_rate": 3.320574162679426e-05,
-      "loss": 0.0331,
-      "step": 3765
-    },
-    {
-      "epoch": 342.36,
-      "learning_rate": 3.31866028708134e-05,
-      "loss": 0.0291,
-      "step": 3766
-    },
-    {
-      "epoch": 342.45,
-      "learning_rate": 3.316746411483253e-05,
-      "loss": 0.0312,
-      "step": 3767
-    },
-    {
-      "epoch": 342.55,
-      "learning_rate": 3.314832535885168e-05,
-      "loss": 0.0326,
-      "step": 3768
-    },
-    {
-      "epoch": 342.64,
-      "learning_rate": 3.312918660287082e-05,
-      "loss": 0.0334,
-      "step": 3769
-    },
-    {
-      "epoch": 342.73,
-      "learning_rate": 3.311004784688995e-05,
-      "loss": 0.0285,
-      "step": 3770
-    },
-    {
-      "epoch": 342.82,
-      "learning_rate": 3.3090909090909095e-05,
-      "loss": 0.0323,
-      "step": 3771
-    },
-    {
-      "epoch": 342.91,
-      "learning_rate": 3.3071770334928234e-05,
-      "loss": 0.0353,
-      "step": 3772
-    },
-    {
-      "epoch": 343.0,
-      "learning_rate": 3.3052631578947366e-05,
-      "loss": 0.0336,
-      "step": 3773
-    },
-    {
-      "epoch": 343.09,
-      "learning_rate": 3.303349282296651e-05,
-      "loss": 0.0352,
-      "step": 3774
-    },
-    {
-      "epoch": 343.18,
-      "learning_rate": 3.301435406698565e-05,
-      "loss": 0.028,
-      "step": 3775
-    },
-    {
-      "epoch": 343.27,
-      "learning_rate": 3.299521531100478e-05,
-      "loss": 0.0311,
-      "step": 3776
-    },
-    {
-      "epoch": 343.36,
-      "learning_rate": 3.297607655502393e-05,
-      "loss": 0.0331,
-      "step": 3777
-    },
-    {
-      "epoch": 343.45,
-      "learning_rate": 3.2956937799043066e-05,
-      "loss": 0.0299,
-      "step": 3778
-    },
-    {
-      "epoch": 343.55,
-      "learning_rate": 3.29377990430622e-05,
-      "loss": 0.03,
-      "step": 3779
-    },
-    {
-      "epoch": 343.64,
-      "learning_rate": 3.2918660287081344e-05,
-      "loss": 0.038,
-      "step": 3780
-    },
-    {
-      "epoch": 343.64,
-      "eval_accuracy_NAT": 0.9931614605352962,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9470511073436628,
-      "eval_accuracy_concrete tiles": 0.9900324099553599,
-      "eval_accuracy_dark bricks": 0.8836457411185097,
-      "eval_accuracy_grass": 0.904940985680683,
-      "eval_accuracy_red bricks": 0.9446022727272727,
-      "eval_accuracy_road": 0.9071436431286103,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4949687095084498,
-      "eval_accuracy_speedway bricks": 0.9887253251462962,
-      "eval_accuracy_steel": 0.9056708000791505,
-      "eval_accuracy_tiles": 0.10190384878783729,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9800008670823671,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8811785304684995,
-      "eval_iou_concrete tiles": 0.9401585319822304,
-      "eval_iou_dark bricks": 0.8325657298277426,
-      "eval_iou_grass": 0.8385062794297669,
-      "eval_iou_red bricks": 0.8609807412202621,
-      "eval_iou_road": 0.9003144276696349,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3331663233684054,
-      "eval_iou_speedway bricks": 0.9768062255436143,
-      "eval_iou_steel": 0.8490983156706784,
-      "eval_iou_tiles": 0.10163934426229508,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11874689161777496,
-      "eval_mean_accuracy": 0.6970651003085483,
-      "eval_mean_iou": 0.6534165628096535,
-      "eval_overall_accuracy": 0.9719274308946397,
-      "eval_runtime": 3.4934,
-      "eval_samples_per_second": 10.305,
-      "eval_steps_per_second": 0.573,
-      "step": 3780
-    },
-    {
-      "epoch": 343.73,
-      "learning_rate": 3.289952153110048e-05,
-      "loss": 0.0305,
-      "step": 3781
-    },
-    {
-      "epoch": 343.82,
-      "learning_rate": 3.2880382775119615e-05,
-      "loss": 0.0285,
-      "step": 3782
-    },
-    {
-      "epoch": 343.91,
-      "learning_rate": 3.286124401913876e-05,
-      "loss": 0.0335,
-      "step": 3783
-    },
-    {
-      "epoch": 344.0,
-      "learning_rate": 3.28421052631579e-05,
-      "loss": 0.0402,
-      "step": 3784
-    },
-    {
-      "epoch": 344.09,
-      "learning_rate": 3.282296650717703e-05,
-      "loss": 0.0348,
-      "step": 3785
-    },
-    {
-      "epoch": 344.18,
-      "learning_rate": 3.280382775119618e-05,
-      "loss": 0.0319,
-      "step": 3786
-    },
-    {
-      "epoch": 344.27,
-      "learning_rate": 3.2784688995215315e-05,
-      "loss": 0.028,
-      "step": 3787
-    },
-    {
-      "epoch": 344.36,
-      "learning_rate": 3.276555023923445e-05,
-      "loss": 0.032,
-      "step": 3788
-    },
-    {
-      "epoch": 344.45,
-      "learning_rate": 3.274641148325359e-05,
-      "loss": 0.0305,
-      "step": 3789
-    },
-    {
-      "epoch": 344.55,
-      "learning_rate": 3.272727272727273e-05,
-      "loss": 0.03,
-      "step": 3790
-    },
-    {
-      "epoch": 344.64,
-      "learning_rate": 3.2708133971291864e-05,
-      "loss": 0.0333,
-      "step": 3791
-    },
-    {
-      "epoch": 344.73,
-      "learning_rate": 3.268899521531101e-05,
-      "loss": 0.0365,
-      "step": 3792
-    },
-    {
-      "epoch": 344.82,
-      "learning_rate": 3.266985645933015e-05,
-      "loss": 0.0306,
-      "step": 3793
-    },
-    {
-      "epoch": 344.91,
-      "learning_rate": 3.265071770334928e-05,
-      "loss": 0.0338,
-      "step": 3794
-    },
-    {
-      "epoch": 345.0,
-      "learning_rate": 3.2631578947368426e-05,
-      "loss": 0.0303,
-      "step": 3795
-    },
-    {
-      "epoch": 345.09,
-      "learning_rate": 3.261244019138756e-05,
-      "loss": 0.0317,
-      "step": 3796
-    },
-    {
-      "epoch": 345.18,
-      "learning_rate": 3.2593301435406696e-05,
-      "loss": 0.034,
-      "step": 3797
-    },
-    {
-      "epoch": 345.27,
-      "learning_rate": 3.257416267942584e-05,
-      "loss": 0.0382,
-      "step": 3798
-    },
-    {
-      "epoch": 345.36,
-      "learning_rate": 3.2555023923444974e-05,
-      "loss": 0.0321,
-      "step": 3799
-    },
-    {
-      "epoch": 345.45,
-      "learning_rate": 3.253588516746411e-05,
-      "loss": 0.0306,
-      "step": 3800
-    },
-    {
-      "epoch": 345.45,
-      "eval_accuracy_NAT": 0.9933728489726663,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9485251512940469,
-      "eval_accuracy_concrete tiles": 0.9878615544548401,
-      "eval_accuracy_dark bricks": 0.8732920210923367,
-      "eval_accuracy_grass": 0.9036443133403468,
-      "eval_accuracy_red bricks": 0.9275568181818182,
-      "eval_accuracy_road": 0.9016647178250686,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5013095139635542,
-      "eval_accuracy_speedway bricks": 0.9885811106806802,
-      "eval_accuracy_steel": 0.9084905349251369,
-      "eval_accuracy_tiles": 0.10998493357074374,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807536725124438,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8864951732212496,
-      "eval_iou_concrete tiles": 0.9406917836138123,
-      "eval_iou_dark bricks": 0.8312206916764361,
-      "eval_iou_grass": 0.8393080321637142,
-      "eval_iou_red bricks": 0.8573773182340391,
-      "eval_iou_road": 0.8941031030682953,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33424627318346417,
-      "eval_iou_speedway bricks": 0.977006798907738,
-      "eval_iou_steel": 0.8513799604401039,
-      "eval_iou_tiles": 0.10972193755551,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11992505937814713,
-      "eval_mean_accuracy": 0.6957141167924031,
-      "eval_mean_iou": 0.6540234418905236,
-      "eval_overall_accuracy": 0.9716529846191406,
-      "eval_runtime": 3.3415,
-      "eval_samples_per_second": 10.774,
-      "eval_steps_per_second": 0.599,
-      "step": 3800
-    },
-    {
-      "epoch": 345.55,
-      "learning_rate": 3.251674641148326e-05,
-      "loss": 0.0295,
-      "step": 3801
-    },
-    {
-      "epoch": 345.64,
-      "learning_rate": 3.249760765550239e-05,
-      "loss": 0.034,
-      "step": 3802
-    },
-    {
-      "epoch": 345.73,
-      "learning_rate": 3.247846889952153e-05,
-      "loss": 0.0338,
-      "step": 3803
-    },
-    {
-      "epoch": 345.82,
-      "learning_rate": 3.2459330143540675e-05,
-      "loss": 0.0307,
-      "step": 3804
-    },
-    {
-      "epoch": 345.91,
-      "learning_rate": 3.244019138755981e-05,
-      "loss": 0.0317,
-      "step": 3805
-    },
-    {
-      "epoch": 346.0,
-      "learning_rate": 3.2421052631578945e-05,
-      "loss": 0.0291,
-      "step": 3806
-    },
-    {
-      "epoch": 346.09,
-      "learning_rate": 3.240191387559809e-05,
-      "loss": 0.0287,
-      "step": 3807
-    },
-    {
-      "epoch": 346.18,
-      "learning_rate": 3.238277511961722e-05,
-      "loss": 0.0318,
-      "step": 3808
-    },
-    {
-      "epoch": 346.27,
-      "learning_rate": 3.236363636363636e-05,
-      "loss": 0.0332,
-      "step": 3809
-    },
-    {
-      "epoch": 346.36,
-      "learning_rate": 3.234449760765551e-05,
-      "loss": 0.0263,
-      "step": 3810
-    },
-    {
-      "epoch": 346.45,
-      "learning_rate": 3.232535885167464e-05,
-      "loss": 0.0344,
-      "step": 3811
-    },
-    {
-      "epoch": 346.55,
-      "learning_rate": 3.230622009569378e-05,
-      "loss": 0.0314,
-      "step": 3812
-    },
-    {
-      "epoch": 346.64,
-      "learning_rate": 3.2287081339712924e-05,
-      "loss": 0.0296,
-      "step": 3813
-    },
-    {
-      "epoch": 346.73,
-      "learning_rate": 3.2267942583732056e-05,
-      "loss": 0.0339,
-      "step": 3814
-    },
-    {
-      "epoch": 346.82,
-      "learning_rate": 3.2248803827751194e-05,
-      "loss": 0.0319,
-      "step": 3815
-    },
-    {
-      "epoch": 346.91,
-      "learning_rate": 3.222966507177034e-05,
-      "loss": 0.0325,
-      "step": 3816
-    },
-    {
-      "epoch": 347.0,
-      "learning_rate": 3.221052631578947e-05,
-      "loss": 0.0326,
-      "step": 3817
-    },
-    {
-      "epoch": 347.09,
-      "learning_rate": 3.219138755980862e-05,
-      "loss": 0.0342,
-      "step": 3818
-    },
-    {
-      "epoch": 347.18,
-      "learning_rate": 3.2172248803827756e-05,
-      "loss": 0.0354,
-      "step": 3819
-    },
-    {
-      "epoch": 347.27,
-      "learning_rate": 3.215311004784689e-05,
-      "loss": 0.0303,
-      "step": 3820
-    },
-    {
-      "epoch": 347.27,
-      "eval_accuracy_NAT": 0.9930024780794778,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9499562749474226,
-      "eval_accuracy_concrete tiles": 0.9843148046230049,
-      "eval_accuracy_dark bricks": 0.8742927523959817,
-      "eval_accuracy_grass": 0.9129027635547695,
-      "eval_accuracy_red bricks": 0.9364346590909091,
-      "eval_accuracy_road": 0.9303507793955649,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5892537148843492,
-      "eval_accuracy_speedway bricks": 0.9879107246082979,
-      "eval_accuracy_steel": 0.9045247674955478,
-      "eval_accuracy_tiles": 0.19778112587316807,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805856539099207,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8941690527772307,
-      "eval_iou_concrete tiles": 0.9479681978798586,
-      "eval_iou_dark bricks": 0.8327223403475328,
-      "eval_iou_grass": 0.8406429125457419,
-      "eval_iou_red bricks": 0.8705843512710465,
-      "eval_iou_road": 0.9227621409758836,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3937657743961976,
-      "eval_iou_speedway bricks": 0.9771699881227357,
-      "eval_iou_steel": 0.8501266960611861,
-      "eval_iou_tiles": 0.1975376196990424,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10520409792661667,
-      "eval_mean_accuracy": 0.7123634496114224,
-      "eval_mean_iou": 0.6698488252297212,
-      "eval_overall_accuracy": 0.9746390448676215,
-      "eval_runtime": 3.6576,
-      "eval_samples_per_second": 9.842,
-      "eval_steps_per_second": 0.547,
-      "step": 3820
-    },
-    {
-      "epoch": 347.36,
-      "learning_rate": 3.2133971291866034e-05,
-      "loss": 0.0319,
-      "step": 3821
-    },
-    {
-      "epoch": 347.45,
-      "learning_rate": 3.211483253588517e-05,
-      "loss": 0.0336,
-      "step": 3822
-    },
-    {
-      "epoch": 347.55,
-      "learning_rate": 3.2095693779904305e-05,
-      "loss": 0.0282,
-      "step": 3823
-    },
-    {
-      "epoch": 347.64,
-      "learning_rate": 3.207655502392345e-05,
-      "loss": 0.0339,
-      "step": 3824
-    },
-    {
-      "epoch": 347.73,
-      "learning_rate": 3.205741626794259e-05,
-      "loss": 0.0333,
-      "step": 3825
-    },
-    {
-      "epoch": 347.82,
-      "learning_rate": 3.203827751196172e-05,
-      "loss": 0.0318,
-      "step": 3826
-    },
-    {
-      "epoch": 347.91,
-      "learning_rate": 3.2019138755980867e-05,
-      "loss": 0.0302,
-      "step": 3827
-    },
-    {
-      "epoch": 348.0,
-      "learning_rate": 3.2000000000000005e-05,
-      "loss": 0.0302,
-      "step": 3828
-    },
-    {
-      "epoch": 348.09,
-      "learning_rate": 3.198086124401914e-05,
-      "loss": 0.0296,
-      "step": 3829
-    },
-    {
-      "epoch": 348.18,
-      "learning_rate": 3.196172248803828e-05,
-      "loss": 0.031,
-      "step": 3830
-    },
-    {
-      "epoch": 348.27,
-      "learning_rate": 3.194258373205742e-05,
-      "loss": 0.0301,
-      "step": 3831
-    },
-    {
-      "epoch": 348.36,
-      "learning_rate": 3.1923444976076554e-05,
-      "loss": 0.0299,
-      "step": 3832
-    },
-    {
-      "epoch": 348.45,
-      "learning_rate": 3.19043062200957e-05,
-      "loss": 0.031,
-      "step": 3833
-    },
-    {
-      "epoch": 348.55,
-      "learning_rate": 3.188516746411483e-05,
-      "loss": 0.0314,
-      "step": 3834
-    },
-    {
-      "epoch": 348.64,
-      "learning_rate": 3.186602870813397e-05,
-      "loss": 0.0337,
-      "step": 3835
-    },
-    {
-      "epoch": 348.73,
-      "learning_rate": 3.1846889952153116e-05,
-      "loss": 0.0317,
-      "step": 3836
-    },
-    {
-      "epoch": 348.82,
-      "learning_rate": 3.182775119617225e-05,
-      "loss": 0.032,
-      "step": 3837
-    },
-    {
-      "epoch": 348.91,
-      "learning_rate": 3.1808612440191386e-05,
-      "loss": 0.0367,
-      "step": 3838
-    },
-    {
-      "epoch": 349.0,
-      "learning_rate": 3.178947368421053e-05,
-      "loss": 0.03,
-      "step": 3839
-    },
-    {
-      "epoch": 349.09,
-      "learning_rate": 3.1770334928229664e-05,
-      "loss": 0.0342,
-      "step": 3840
-    },
-    {
-      "epoch": 349.09,
-      "eval_accuracy_NAT": 0.9932669783028649,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9508723550366969,
-      "eval_accuracy_concrete tiles": 0.9845899834892681,
-      "eval_accuracy_dark bricks": 0.8755244216927754,
-      "eval_accuracy_grass": 0.9046019635324011,
-      "eval_accuracy_red bricks": 0.9254261363636364,
-      "eval_accuracy_road": 0.9138825640548158,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4923772502963637,
-      "eval_accuracy_speedway bricks": 0.9877260818196848,
-      "eval_accuracy_steel": 0.9108485588021898,
-      "eval_accuracy_tiles": 0.162238049582249,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805364063159743,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8890644594933534,
-      "eval_iou_concrete tiles": 0.9509494138144878,
-      "eval_iou_dark bricks": 0.8322783652262998,
-      "eval_iou_grass": 0.840462149945446,
-      "eval_iou_red bricks": 0.8624855204368691,
-      "eval_iou_road": 0.9068480757185607,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32826054991912956,
-      "eval_iou_speedway bricks": 0.9770170924407566,
-      "eval_iou_steel": 0.8506059532792313,
-      "eval_iou_tiles": 0.16179483677093293,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11583779752254486,
-      "eval_mean_accuracy": 0.7001041802286883,
-      "eval_mean_iou": 0.6600232941046955,
-      "eval_overall_accuracy": 0.9727947447035048,
-      "eval_runtime": 3.4162,
-      "eval_samples_per_second": 10.538,
-      "eval_steps_per_second": 0.585,
-      "step": 3840
-    },
-    {
-      "epoch": 349.18,
-      "learning_rate": 3.17511961722488e-05,
-      "loss": 0.0322,
-      "step": 3841
-    },
-    {
-      "epoch": 349.27,
-      "learning_rate": 3.173205741626795e-05,
-      "loss": 0.0331,
-      "step": 3842
-    },
-    {
-      "epoch": 349.36,
-      "learning_rate": 3.171291866028708e-05,
-      "loss": 0.0306,
-      "step": 3843
-    },
-    {
-      "epoch": 349.45,
-      "learning_rate": 3.169377990430622e-05,
-      "loss": 0.0317,
-      "step": 3844
-    },
-    {
-      "epoch": 349.55,
-      "learning_rate": 3.1674641148325365e-05,
-      "loss": 0.0292,
-      "step": 3845
-    },
-    {
-      "epoch": 349.64,
-      "learning_rate": 3.1655502392344497e-05,
-      "loss": 0.0394,
-      "step": 3846
-    },
-    {
-      "epoch": 349.73,
-      "learning_rate": 3.1636363636363635e-05,
-      "loss": 0.0317,
-      "step": 3847
-    },
-    {
-      "epoch": 349.82,
-      "learning_rate": 3.161722488038278e-05,
-      "loss": 0.0321,
-      "step": 3848
-    },
-    {
-      "epoch": 349.91,
-      "learning_rate": 3.159808612440191e-05,
-      "loss": 0.0336,
-      "step": 3849
-    },
-    {
-      "epoch": 350.0,
-      "learning_rate": 3.157894736842105e-05,
-      "loss": 0.0302,
-      "step": 3850
-    },
-    {
-      "epoch": 350.09,
-      "learning_rate": 3.15598086124402e-05,
-      "loss": 0.0357,
-      "step": 3851
-    },
-    {
-      "epoch": 350.18,
-      "learning_rate": 3.154066985645933e-05,
-      "loss": 0.0318,
-      "step": 3852
-    },
-    {
-      "epoch": 350.27,
-      "learning_rate": 3.152153110047847e-05,
-      "loss": 0.0303,
-      "step": 3853
-    },
-    {
-      "epoch": 350.36,
-      "learning_rate": 3.1502392344497614e-05,
-      "loss": 0.0323,
-      "step": 3854
-    },
-    {
-      "epoch": 350.45,
-      "learning_rate": 3.1483253588516746e-05,
-      "loss": 0.0364,
-      "step": 3855
-    },
-    {
-      "epoch": 350.55,
-      "learning_rate": 3.1464114832535884e-05,
-      "loss": 0.0272,
-      "step": 3856
-    },
-    {
-      "epoch": 350.64,
-      "learning_rate": 3.144497607655503e-05,
-      "loss": 0.037,
-      "step": 3857
-    },
-    {
-      "epoch": 350.73,
-      "learning_rate": 3.142583732057416e-05,
-      "loss": 0.0268,
-      "step": 3858
-    },
-    {
-      "epoch": 350.82,
-      "learning_rate": 3.14066985645933e-05,
-      "loss": 0.0361,
-      "step": 3859
-    },
-    {
-      "epoch": 350.91,
-      "learning_rate": 3.1387559808612446e-05,
-      "loss": 0.0324,
-      "step": 3860
-    },
-    {
-      "epoch": 350.91,
-      "eval_accuracy_NAT": 0.992860082028595,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9505611828833855,
-      "eval_accuracy_concrete tiles": 0.9847122852076071,
-      "eval_accuracy_dark bricks": 0.8813748508525461,
-      "eval_accuracy_grass": 0.9148530146758145,
-      "eval_accuracy_red bricks": 0.9282670454545454,
-      "eval_accuracy_road": 0.9118353734701996,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5101039340556337,
-      "eval_accuracy_speedway bricks": 0.9880863162798219,
-      "eval_accuracy_steel": 0.9069817294373722,
-      "eval_accuracy_tiles": 0.14819887686618272,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806870038355497,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8909231705293434,
-      "eval_iou_concrete tiles": 0.9528684280600018,
-      "eval_iou_dark bricks": 0.8356079404466501,
-      "eval_iou_grass": 0.8439544353915648,
-      "eval_iou_red bricks": 0.8652763985435287,
-      "eval_iou_road": 0.9045908758292247,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.339685337151879,
-      "eval_iou_speedway bricks": 0.9769401764268609,
-      "eval_iou_steel": 0.8486087432789996,
-      "eval_iou_tiles": 0.14793546622914958,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11561170965433121,
-      "eval_mean_accuracy": 0.7013718993239771,
-      "eval_mean_iou": 0.660544459670981,
-      "eval_overall_accuracy": 0.9727354049682617,
-      "eval_runtime": 3.3367,
-      "eval_samples_per_second": 10.789,
-      "eval_steps_per_second": 0.599,
-      "step": 3860
-    },
-    {
-      "epoch": 351.0,
-      "learning_rate": 3.136842105263158e-05,
-      "loss": 0.0349,
-      "step": 3861
-    },
-    {
-      "epoch": 351.09,
-      "learning_rate": 3.134928229665072e-05,
-      "loss": 0.0338,
-      "step": 3862
-    },
-    {
-      "epoch": 351.18,
-      "learning_rate": 3.133014354066986e-05,
-      "loss": 0.0328,
-      "step": 3863
-    },
-    {
-      "epoch": 351.27,
-      "learning_rate": 3.1311004784688995e-05,
-      "loss": 0.0291,
-      "step": 3864
-    },
-    {
-      "epoch": 351.36,
-      "learning_rate": 3.129186602870813e-05,
-      "loss": 0.0293,
-      "step": 3865
-    },
-    {
-      "epoch": 351.45,
-      "learning_rate": 3.127272727272728e-05,
-      "loss": 0.0324,
-      "step": 3866
-    },
-    {
-      "epoch": 351.55,
-      "learning_rate": 3.125358851674641e-05,
-      "loss": 0.0327,
-      "step": 3867
-    },
-    {
-      "epoch": 351.64,
-      "learning_rate": 3.123444976076555e-05,
-      "loss": 0.0349,
-      "step": 3868
-    },
-    {
-      "epoch": 351.73,
-      "learning_rate": 3.1215311004784695e-05,
-      "loss": 0.029,
-      "step": 3869
-    },
-    {
-      "epoch": 351.82,
-      "learning_rate": 3.119617224880383e-05,
-      "loss": 0.0332,
-      "step": 3870
-    },
-    {
-      "epoch": 351.91,
-      "learning_rate": 3.1177033492822966e-05,
-      "loss": 0.0364,
-      "step": 3871
-    },
-    {
-      "epoch": 352.0,
-      "learning_rate": 3.115789473684211e-05,
-      "loss": 0.0329,
-      "step": 3872
-    },
-    {
-      "epoch": 352.09,
-      "learning_rate": 3.1138755980861244e-05,
-      "loss": 0.027,
-      "step": 3873
-    },
-    {
-      "epoch": 352.18,
-      "learning_rate": 3.111961722488038e-05,
-      "loss": 0.0356,
-      "step": 3874
-    },
-    {
-      "epoch": 352.27,
-      "learning_rate": 3.110047846889952e-05,
-      "loss": 0.0296,
-      "step": 3875
-    },
-    {
-      "epoch": 352.36,
-      "learning_rate": 3.108133971291866e-05,
-      "loss": 0.0284,
-      "step": 3876
-    },
-    {
-      "epoch": 352.45,
-      "learning_rate": 3.10622009569378e-05,
-      "loss": 0.0291,
-      "step": 3877
-    },
-    {
-      "epoch": 352.55,
-      "learning_rate": 3.104306220095694e-05,
-      "loss": 0.0298,
-      "step": 3878
-    },
-    {
-      "epoch": 352.64,
-      "learning_rate": 3.1023923444976076e-05,
-      "loss": 0.0338,
-      "step": 3879
-    },
-    {
-      "epoch": 352.73,
-      "learning_rate": 3.1004784688995215e-05,
-      "loss": 0.0304,
-      "step": 3880
-    },
-    {
-      "epoch": 352.73,
-      "eval_accuracy_NAT": 0.9924847705041491,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.950846871110348,
-      "eval_accuracy_concrete tiles": 0.9863327829756008,
-      "eval_accuracy_dark bricks": 0.8874177283399407,
-      "eval_accuracy_grass": 0.9135178965248483,
-      "eval_accuracy_red bricks": 0.9287997159090909,
-      "eval_accuracy_road": 0.9170797122566618,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47553276541780387,
-      "eval_accuracy_speedway bricks": 0.9886680014047333,
-      "eval_accuracy_steel": 0.9107248862212255,
-      "eval_accuracy_tiles": 0.1809341186138885,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803868629194666,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.889556025688322,
-      "eval_iou_concrete tiles": 0.9523795465281059,
-      "eval_iou_dark bricks": 0.8364533449426789,
-      "eval_iou_grass": 0.8424701688337641,
-      "eval_iou_red bricks": 0.8606449489963804,
-      "eval_iou_road": 0.909698607527675,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32000074207372503,
-      "eval_iou_speedway bricks": 0.976852083705956,
-      "eval_iou_steel": 0.8511917146357815,
-      "eval_iou_tiles": 0.18016912165848337,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11291428655385971,
-      "eval_mean_accuracy": 0.7024876345598686,
-      "eval_mean_iou": 0.6615233205777185,
-      "eval_overall_accuracy": 0.9730422761705186,
-      "eval_runtime": 3.7385,
-      "eval_samples_per_second": 9.629,
-      "eval_steps_per_second": 0.535,
-      "step": 3880
-    },
-    {
-      "epoch": 352.82,
-      "learning_rate": 3.0985645933014354e-05,
-      "loss": 0.0296,
-      "step": 3881
-    },
-    {
-      "epoch": 352.91,
-      "learning_rate": 3.096650717703349e-05,
-      "loss": 0.0341,
-      "step": 3882
-    },
-    {
-      "epoch": 353.0,
-      "learning_rate": 3.094736842105263e-05,
-      "loss": 0.0307,
-      "step": 3883
-    },
-    {
-      "epoch": 353.09,
-      "learning_rate": 3.092822966507177e-05,
-      "loss": 0.04,
-      "step": 3884
-    },
-    {
-      "epoch": 353.18,
-      "learning_rate": 3.090909090909091e-05,
-      "loss": 0.0273,
-      "step": 3885
-    },
-    {
-      "epoch": 353.27,
-      "learning_rate": 3.088995215311005e-05,
-      "loss": 0.0309,
-      "step": 3886
-    },
-    {
-      "epoch": 353.36,
-      "learning_rate": 3.0870813397129186e-05,
-      "loss": 0.037,
-      "step": 3887
-    },
-    {
-      "epoch": 353.45,
-      "learning_rate": 3.0851674641148325e-05,
-      "loss": 0.0343,
-      "step": 3888
-    },
-    {
-      "epoch": 353.55,
-      "learning_rate": 3.0832535885167464e-05,
-      "loss": 0.0328,
-      "step": 3889
-    },
-    {
-      "epoch": 353.64,
-      "learning_rate": 3.08133971291866e-05,
-      "loss": 0.0267,
-      "step": 3890
-    },
-    {
-      "epoch": 353.73,
-      "learning_rate": 3.079425837320574e-05,
-      "loss": 0.035,
-      "step": 3891
-    },
-    {
-      "epoch": 353.82,
-      "learning_rate": 3.077511961722489e-05,
-      "loss": 0.0334,
-      "step": 3892
-    },
-    {
-      "epoch": 353.91,
-      "learning_rate": 3.075598086124402e-05,
-      "loss": 0.0306,
-      "step": 3893
-    },
-    {
-      "epoch": 354.0,
-      "learning_rate": 3.073684210526316e-05,
-      "loss": 0.0281,
-      "step": 3894
-    },
-    {
-      "epoch": 354.09,
-      "learning_rate": 3.0717703349282303e-05,
-      "loss": 0.0309,
-      "step": 3895
-    },
-    {
-      "epoch": 354.18,
-      "learning_rate": 3.0698564593301435e-05,
-      "loss": 0.0303,
-      "step": 3896
-    },
-    {
-      "epoch": 354.27,
-      "learning_rate": 3.0679425837320574e-05,
-      "loss": 0.0327,
-      "step": 3897
-    },
-    {
-      "epoch": 354.36,
-      "learning_rate": 3.066028708133972e-05,
-      "loss": 0.0305,
-      "step": 3898
-    },
-    {
-      "epoch": 354.45,
-      "learning_rate": 3.064114832535885e-05,
-      "loss": 0.0302,
-      "step": 3899
-    },
-    {
-      "epoch": 354.55,
-      "learning_rate": 3.062200956937799e-05,
-      "loss": 0.0344,
-      "step": 3900
-    },
-    {
-      "epoch": 354.55,
-      "eval_accuracy_NAT": 0.9933116204352979,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9540806472380788,
-      "eval_accuracy_concrete tiles": 0.9841313520454963,
-      "eval_accuracy_dark bricks": 0.8655940879873754,
-      "eval_accuracy_grass": 0.9029068527909911,
-      "eval_accuracy_red bricks": 0.9247159090909091,
-      "eval_accuracy_road": 0.9238198423917183,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47638739558349186,
-      "eval_accuracy_speedway bricks": 0.9873127474726264,
-      "eval_accuracy_steel": 0.9118626739660972,
-      "eval_accuracy_tiles": 0.15381454595260924,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805637374868706,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.888548569429958,
-      "eval_iou_concrete tiles": 0.9552455853984271,
-      "eval_iou_dark bricks": 0.8314477965099083,
-      "eval_iou_grass": 0.8410584814931777,
-      "eval_iou_red bricks": 0.8562972706346597,
-      "eval_iou_road": 0.9170246604610408,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3189663128749423,
-      "eval_iou_speedway bricks": 0.9770071085873288,
-      "eval_iou_steel": 0.8520253301080073,
-      "eval_iou_tiles": 0.15353065828149565,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10896729677915573,
-      "eval_mean_accuracy": 0.6983028980734378,
-      "eval_mean_iou": 0.659362731635832,
-      "eval_overall_accuracy": 0.9737310409545898,
-      "eval_runtime": 3.7751,
-      "eval_samples_per_second": 9.536,
-      "eval_steps_per_second": 0.53,
-      "step": 3900
-    },
-    {
-      "epoch": 354.64,
-      "learning_rate": 3.0602870813397136e-05,
-      "loss": 0.0348,
-      "step": 3901
-    },
-    {
-      "epoch": 354.73,
-      "learning_rate": 3.058373205741627e-05,
-      "loss": 0.0305,
-      "step": 3902
-    },
-    {
-      "epoch": 354.82,
-      "learning_rate": 3.056459330143541e-05,
-      "loss": 0.0266,
-      "step": 3903
-    },
-    {
-      "epoch": 354.91,
-      "learning_rate": 3.054545454545455e-05,
-      "loss": 0.0302,
-      "step": 3904
-    },
-    {
-      "epoch": 355.0,
-      "learning_rate": 3.0526315789473684e-05,
-      "loss": 0.0313,
-      "step": 3905
-    },
-    {
-      "epoch": 355.09,
-      "learning_rate": 3.0507177033492823e-05,
-      "loss": 0.0342,
-      "step": 3906
-    },
-    {
-      "epoch": 355.18,
-      "learning_rate": 3.0488038277511965e-05,
-      "loss": 0.0323,
-      "step": 3907
-    },
-    {
-      "epoch": 355.27,
-      "learning_rate": 3.04688995215311e-05,
-      "loss": 0.0337,
-      "step": 3908
-    },
-    {
-      "epoch": 355.36,
-      "learning_rate": 3.044976076555024e-05,
-      "loss": 0.0331,
-      "step": 3909
-    },
-    {
-      "epoch": 355.45,
-      "learning_rate": 3.043062200956938e-05,
-      "loss": 0.0311,
-      "step": 3910
-    },
-    {
-      "epoch": 355.55,
-      "learning_rate": 3.0411483253588517e-05,
-      "loss": 0.0296,
-      "step": 3911
-    },
-    {
-      "epoch": 355.64,
-      "learning_rate": 3.0392344497607656e-05,
-      "loss": 0.0306,
-      "step": 3912
-    },
-    {
-      "epoch": 355.73,
-      "learning_rate": 3.0373205741626798e-05,
-      "loss": 0.0319,
-      "step": 3913
-    },
-    {
-      "epoch": 355.82,
-      "learning_rate": 3.0354066985645933e-05,
-      "loss": 0.0304,
-      "step": 3914
-    },
-    {
-      "epoch": 355.91,
-      "learning_rate": 3.0334928229665072e-05,
-      "loss": 0.0271,
-      "step": 3915
-    },
-    {
-      "epoch": 356.0,
-      "learning_rate": 3.0315789473684214e-05,
-      "loss": 0.0304,
-      "step": 3916
-    },
-    {
-      "epoch": 356.09,
-      "learning_rate": 3.029665071770335e-05,
-      "loss": 0.0343,
-      "step": 3917
-    },
-    {
-      "epoch": 356.18,
-      "learning_rate": 3.027751196172249e-05,
-      "loss": 0.0314,
-      "step": 3918
-    },
-    {
-      "epoch": 356.27,
-      "learning_rate": 3.025837320574163e-05,
-      "loss": 0.0355,
-      "step": 3919
-    },
-    {
-      "epoch": 356.36,
-      "learning_rate": 3.0239234449760766e-05,
-      "loss": 0.0376,
-      "step": 3920
-    },
-    {
-      "epoch": 356.36,
-      "eval_accuracy_NAT": 0.992303378756556,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.951313629340315,
-      "eval_accuracy_concrete tiles": 0.9895737785115881,
-      "eval_accuracy_dark bricks": 0.8870713213502175,
-      "eval_accuracy_grass": 0.9136437191778188,
-      "eval_accuracy_red bricks": 0.9454900568181818,
-      "eval_accuracy_road": 0.9261644983536621,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48060540898188736,
-      "eval_accuracy_speedway bricks": 0.9891742605539042,
-      "eval_accuracy_steel": 0.9051101510454456,
-      "eval_accuracy_tiles": 0.2173674839063142,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805370216966892,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8898006548657024,
-      "eval_iou_concrete tiles": 0.9365685678733686,
-      "eval_iou_dark bricks": 0.8321718721790937,
-      "eval_iou_grass": 0.8433990327378552,
-      "eval_iou_red bricks": 0.8623481781376519,
-      "eval_iou_road": 0.9184800205298919,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3221889554224884,
-      "eval_iou_speedway bricks": 0.9768904345587521,
-      "eval_iou_steel": 0.8515016598926499,
-      "eval_iou_tiles": 0.2166848716548334,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10837341845035553,
-      "eval_mean_accuracy": 0.7075244374458378,
-      "eval_mean_iou": 0.6638900976576136,
-      "eval_overall_accuracy": 0.9738829930623373,
-      "eval_runtime": 3.5459,
-      "eval_samples_per_second": 10.152,
-      "eval_steps_per_second": 0.564,
-      "step": 3920
-    },
-    {
-      "epoch": 356.45,
-      "learning_rate": 3.0220095693779905e-05,
-      "loss": 0.0272,
-      "step": 3921
-    },
-    {
-      "epoch": 356.55,
-      "learning_rate": 3.0200956937799047e-05,
-      "loss": 0.0324,
-      "step": 3922
-    },
-    {
-      "epoch": 356.64,
-      "learning_rate": 3.0181818181818182e-05,
-      "loss": 0.0318,
-      "step": 3923
-    },
-    {
-      "epoch": 356.73,
-      "learning_rate": 3.016267942583732e-05,
-      "loss": 0.0283,
-      "step": 3924
-    },
-    {
-      "epoch": 356.82,
-      "learning_rate": 3.0143540669856463e-05,
-      "loss": 0.0306,
-      "step": 3925
-    },
-    {
-      "epoch": 356.91,
-      "learning_rate": 3.01244019138756e-05,
-      "loss": 0.0328,
-      "step": 3926
-    },
-    {
-      "epoch": 357.0,
-      "learning_rate": 3.0105263157894737e-05,
-      "loss": 0.0252,
-      "step": 3927
-    },
-    {
-      "epoch": 357.09,
-      "learning_rate": 3.008612440191388e-05,
-      "loss": 0.0371,
-      "step": 3928
-    },
-    {
-      "epoch": 357.18,
-      "learning_rate": 3.0066985645933015e-05,
-      "loss": 0.0327,
-      "step": 3929
-    },
-    {
-      "epoch": 357.27,
-      "learning_rate": 3.0047846889952154e-05,
-      "loss": 0.0377,
-      "step": 3930
-    },
-    {
-      "epoch": 357.36,
-      "learning_rate": 3.0028708133971296e-05,
-      "loss": 0.0334,
-      "step": 3931
-    },
-    {
-      "epoch": 357.45,
-      "learning_rate": 3.000956937799043e-05,
-      "loss": 0.0282,
-      "step": 3932
-    },
-    {
-      "epoch": 357.55,
-      "learning_rate": 2.999043062200957e-05,
-      "loss": 0.0309,
-      "step": 3933
-    },
-    {
-      "epoch": 357.64,
-      "learning_rate": 2.9971291866028712e-05,
-      "loss": 0.0325,
-      "step": 3934
-    },
-    {
-      "epoch": 357.73,
-      "learning_rate": 2.9952153110047848e-05,
-      "loss": 0.0344,
-      "step": 3935
-    },
-    {
-      "epoch": 357.82,
-      "learning_rate": 2.9933014354066986e-05,
-      "loss": 0.0293,
-      "step": 3936
-    },
-    {
-      "epoch": 357.91,
-      "learning_rate": 2.991387559808613e-05,
-      "loss": 0.0348,
-      "step": 3937
-    },
-    {
-      "epoch": 358.0,
-      "learning_rate": 2.9894736842105264e-05,
-      "loss": 0.0282,
-      "step": 3938
-    },
-    {
-      "epoch": 358.09,
-      "learning_rate": 2.9875598086124403e-05,
-      "loss": 0.0352,
-      "step": 3939
-    },
-    {
-      "epoch": 358.18,
-      "learning_rate": 2.9856459330143545e-05,
-      "loss": 0.0327,
-      "step": 3940
-    },
-    {
-      "epoch": 358.18,
-      "eval_accuracy_NAT": 0.9930338863781856,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9498851882055024,
-      "eval_accuracy_concrete tiles": 0.9888399682015532,
-      "eval_accuracy_dark bricks": 0.8805665678765252,
-      "eval_accuracy_grass": 0.9122946207320781,
-      "eval_accuracy_red bricks": 0.9337713068181818,
-      "eval_accuracy_road": 0.9326071631113911,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4839412234995727,
-      "eval_accuracy_speedway bricks": 0.9894313122792282,
-      "eval_accuracy_steel": 0.9014824220038256,
-      "eval_accuracy_tiles": 0.20052047664703465,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805602865115339,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8908652237465439,
-      "eval_iou_concrete tiles": 0.9423368298368299,
-      "eval_iou_dark bricks": 0.8331997960521523,
-      "eval_iou_grass": 0.8446553559698281,
-      "eval_iou_red bricks": 0.8635467980295567,
-      "eval_iou_road": 0.9264189902595882,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3270850413654319,
-      "eval_iou_speedway bricks": 0.9768095895078717,
-      "eval_iou_steel": 0.847582576879249,
-      "eval_iou_tiles": 0.20024620434961019,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10282377898693085,
-      "eval_mean_accuracy": 0.7051057027502368,
-      "eval_mean_iou": 0.6641005148083228,
-      "eval_overall_accuracy": 0.9746903313530816,
-      "eval_runtime": 3.9956,
-      "eval_samples_per_second": 9.01,
-      "eval_steps_per_second": 0.501,
-      "step": 3940
-    },
-    {
-      "epoch": 358.27,
-      "learning_rate": 2.983732057416268e-05,
-      "loss": 0.0341,
-      "step": 3941
-    },
-    {
-      "epoch": 358.36,
-      "learning_rate": 2.9818181818181816e-05,
-      "loss": 0.03,
-      "step": 3942
-    },
-    {
-      "epoch": 358.45,
-      "learning_rate": 2.979904306220096e-05,
-      "loss": 0.0337,
-      "step": 3943
-    },
-    {
-      "epoch": 358.55,
-      "learning_rate": 2.9779904306220097e-05,
-      "loss": 0.0316,
-      "step": 3944
-    },
-    {
-      "epoch": 358.64,
-      "learning_rate": 2.9760765550239232e-05,
-      "loss": 0.0311,
-      "step": 3945
-    },
-    {
-      "epoch": 358.73,
-      "learning_rate": 2.9741626794258378e-05,
-      "loss": 0.032,
-      "step": 3946
-    },
-    {
-      "epoch": 358.82,
-      "learning_rate": 2.9722488038277513e-05,
-      "loss": 0.0324,
-      "step": 3947
-    },
-    {
-      "epoch": 358.91,
-      "learning_rate": 2.970334928229665e-05,
-      "loss": 0.0289,
-      "step": 3948
-    },
-    {
-      "epoch": 359.0,
-      "learning_rate": 2.968421052631579e-05,
-      "loss": 0.0286,
-      "step": 3949
-    },
-    {
-      "epoch": 359.09,
-      "learning_rate": 2.966507177033493e-05,
-      "loss": 0.0273,
-      "step": 3950
-    },
-    {
-      "epoch": 359.18,
-      "learning_rate": 2.9645933014354065e-05,
-      "loss": 0.0306,
-      "step": 3951
-    },
-    {
-      "epoch": 359.27,
-      "learning_rate": 2.9626794258373207e-05,
-      "loss": 0.0324,
-      "step": 3952
-    },
-    {
-      "epoch": 359.36,
-      "learning_rate": 2.9607655502392346e-05,
-      "loss": 0.0311,
-      "step": 3953
-    },
-    {
-      "epoch": 359.45,
-      "learning_rate": 2.958851674641148e-05,
-      "loss": 0.0316,
-      "step": 3954
-    },
-    {
-      "epoch": 359.55,
-      "learning_rate": 2.9569377990430623e-05,
-      "loss": 0.0323,
-      "step": 3955
-    },
-    {
-      "epoch": 359.64,
-      "learning_rate": 2.9550239234449762e-05,
-      "loss": 0.0301,
-      "step": 3956
-    },
-    {
-      "epoch": 359.73,
-      "learning_rate": 2.9531100478468897e-05,
-      "loss": 0.0321,
-      "step": 3957
-    },
-    {
-      "epoch": 359.82,
-      "learning_rate": 2.951196172248804e-05,
-      "loss": 0.032,
-      "step": 3958
-    },
-    {
-      "epoch": 359.91,
-      "learning_rate": 2.949282296650718e-05,
-      "loss": 0.0301,
-      "step": 3959
-    },
-    {
-      "epoch": 360.0,
-      "learning_rate": 2.9473684210526314e-05,
-      "loss": 0.0363,
-      "step": 3960
-    },
-    {
-      "epoch": 360.0,
-      "eval_accuracy_NAT": 0.9933024449772484,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9498261727971158,
-      "eval_accuracy_concrete tiles": 0.9869748669968813,
-      "eval_accuracy_dark bricks": 0.8880335629883376,
-      "eval_accuracy_grass": 0.9041301285837612,
-      "eval_accuracy_red bricks": 0.9247159090909091,
-      "eval_accuracy_road": 0.9264547084778841,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4705152592837648,
-      "eval_accuracy_speedway bricks": 0.9884718938678079,
-      "eval_accuracy_steel": 0.9086224523448322,
-      "eval_accuracy_tiles": 0.1675112998219422,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9802825009873638,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8912684238811004,
-      "eval_iou_concrete tiles": 0.9444119368051492,
-      "eval_iou_dark bricks": 0.8332550832460544,
-      "eval_iou_grass": 0.8395630288101104,
-      "eval_iou_red bricks": 0.8628230616302187,
-      "eval_iou_road": 0.9194249438984291,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31825386465772837,
-      "eval_iou_speedway bricks": 0.9771820703996239,
-      "eval_iou_steel": 0.8498226403454657,
-      "eval_iou_tiles": 0.1670650911822963,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10982490330934525,
-      "eval_mean_accuracy": 0.700658361479268,
-      "eval_mean_iou": 0.6602578958341184,
-      "eval_overall_accuracy": 0.9738895628187392,
-      "eval_runtime": 3.8163,
-      "eval_samples_per_second": 9.433,
-      "eval_steps_per_second": 0.524,
-      "step": 3960
-    },
-    {
-      "epoch": 360.09,
-      "learning_rate": 2.9454545454545456e-05,
-      "loss": 0.0316,
-      "step": 3961
-    },
-    {
-      "epoch": 360.18,
-      "learning_rate": 2.9435406698564595e-05,
-      "loss": 0.029,
-      "step": 3962
-    },
-    {
-      "epoch": 360.27,
-      "learning_rate": 2.9416267942583737e-05,
-      "loss": 0.0281,
-      "step": 3963
-    },
-    {
-      "epoch": 360.36,
-      "learning_rate": 2.9397129186602872e-05,
-      "loss": 0.0315,
-      "step": 3964
-    },
-    {
-      "epoch": 360.45,
-      "learning_rate": 2.937799043062201e-05,
-      "loss": 0.0288,
-      "step": 3965
-    },
-    {
-      "epoch": 360.55,
-      "learning_rate": 2.9358851674641153e-05,
-      "loss": 0.0305,
-      "step": 3966
-    },
-    {
-      "epoch": 360.64,
-      "learning_rate": 2.933971291866029e-05,
-      "loss": 0.0304,
-      "step": 3967
-    },
-    {
-      "epoch": 360.73,
-      "learning_rate": 2.9320574162679427e-05,
-      "loss": 0.0462,
-      "step": 3968
-    },
-    {
-      "epoch": 360.82,
-      "learning_rate": 2.930143540669857e-05,
-      "loss": 0.0305,
-      "step": 3969
-    },
-    {
-      "epoch": 360.91,
-      "learning_rate": 2.9282296650717705e-05,
-      "loss": 0.0347,
-      "step": 3970
-    },
-    {
-      "epoch": 361.0,
-      "learning_rate": 2.9263157894736844e-05,
-      "loss": 0.0367,
-      "step": 3971
-    },
-    {
-      "epoch": 361.09,
-      "learning_rate": 2.9244019138755986e-05,
-      "loss": 0.0298,
-      "step": 3972
-    },
-    {
-      "epoch": 361.18,
-      "learning_rate": 2.922488038277512e-05,
-      "loss": 0.0342,
-      "step": 3973
-    },
-    {
-      "epoch": 361.27,
-      "learning_rate": 2.920574162679426e-05,
-      "loss": 0.035,
-      "step": 3974
-    },
-    {
-      "epoch": 361.36,
-      "learning_rate": 2.9186602870813402e-05,
-      "loss": 0.029,
-      "step": 3975
-    },
-    {
-      "epoch": 361.45,
-      "learning_rate": 2.9167464114832538e-05,
-      "loss": 0.0302,
-      "step": 3976
-    },
-    {
-      "epoch": 361.55,
-      "learning_rate": 2.9148325358851676e-05,
-      "loss": 0.0344,
-      "step": 3977
-    },
-    {
-      "epoch": 361.64,
-      "learning_rate": 2.912918660287082e-05,
-      "loss": 0.0326,
-      "step": 3978
-    },
-    {
-      "epoch": 361.73,
-      "learning_rate": 2.9110047846889954e-05,
-      "loss": 0.0303,
-      "step": 3979
-    },
-    {
-      "epoch": 361.82,
-      "learning_rate": 2.909090909090909e-05,
-      "loss": 0.0317,
-      "step": 3980
-    },
-    {
-      "epoch": 361.82,
-      "eval_accuracy_NAT": 0.9933098559241345,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9528587600326194,
-      "eval_accuracy_concrete tiles": 0.9868219898489574,
-      "eval_accuracy_dark bricks": 0.8820676648319926,
-      "eval_accuracy_grass": 0.9043433280790726,
-      "eval_accuracy_red bricks": 0.9110440340909091,
-      "eval_accuracy_road": 0.9528952692122126,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4346207923248697,
-      "eval_accuracy_speedway bricks": 0.9886317969363779,
-      "eval_accuracy_steel": 0.9063138975001649,
-      "eval_accuracy_tiles": 0.0913573483084509,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807914347047291,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8865025737014506,
-      "eval_iou_concrete tiles": 0.9502149208031561,
-      "eval_iou_dark bricks": 0.8330122496455963,
-      "eval_iou_grass": 0.8424733581003422,
-      "eval_iou_red bricks": 0.853459747172322,
-      "eval_iou_road": 0.9462644485591667,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2928990784780024,
-      "eval_iou_speedway bricks": 0.9770866083504887,
-      "eval_iou_steel": 0.8508456209605635,
-      "eval_iou_tiles": 0.09113889458222313,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0957815945148468,
-      "eval_mean_accuracy": 0.6926357490069047,
-      "eval_mean_iou": 0.6542068411583107,
-      "eval_overall_accuracy": 0.9761745664808485,
-      "eval_runtime": 3.3465,
-      "eval_samples_per_second": 10.757,
-      "eval_steps_per_second": 0.598,
-      "step": 3980
-    },
-    {
-      "epoch": 361.91,
-      "learning_rate": 2.9071770334928235e-05,
-      "loss": 0.0341,
-      "step": 3981
-    },
-    {
-      "epoch": 362.0,
-      "learning_rate": 2.905263157894737e-05,
-      "loss": 0.0362,
-      "step": 3982
-    },
-    {
-      "epoch": 362.09,
-      "learning_rate": 2.9033492822966506e-05,
-      "loss": 0.0312,
-      "step": 3983
-    },
-    {
-      "epoch": 362.18,
-      "learning_rate": 2.901435406698565e-05,
-      "loss": 0.0326,
-      "step": 3984
-    },
-    {
-      "epoch": 362.27,
-      "learning_rate": 2.8995215311004787e-05,
-      "loss": 0.0334,
-      "step": 3985
-    },
-    {
-      "epoch": 362.36,
-      "learning_rate": 2.8976076555023922e-05,
-      "loss": 0.0271,
-      "step": 3986
-    },
-    {
-      "epoch": 362.45,
-      "learning_rate": 2.8956937799043064e-05,
-      "loss": 0.0363,
-      "step": 3987
-    },
-    {
-      "epoch": 362.55,
-      "learning_rate": 2.8937799043062203e-05,
-      "loss": 0.0352,
-      "step": 3988
-    },
-    {
-      "epoch": 362.64,
-      "learning_rate": 2.8918660287081338e-05,
-      "loss": 0.0307,
-      "step": 3989
-    },
-    {
-      "epoch": 362.73,
-      "learning_rate": 2.889952153110048e-05,
-      "loss": 0.0335,
-      "step": 3990
-    },
-    {
-      "epoch": 362.82,
-      "learning_rate": 2.888038277511962e-05,
-      "loss": 0.028,
-      "step": 3991
-    },
-    {
-      "epoch": 362.91,
-      "learning_rate": 2.8861244019138755e-05,
-      "loss": 0.0309,
-      "step": 3992
-    },
-    {
-      "epoch": 363.0,
-      "learning_rate": 2.8842105263157897e-05,
-      "loss": 0.0293,
-      "step": 3993
-    },
-    {
-      "epoch": 363.09,
-      "learning_rate": 2.8822966507177036e-05,
-      "loss": 0.0303,
-      "step": 3994
-    },
-    {
-      "epoch": 363.18,
-      "learning_rate": 2.880382775119617e-05,
-      "loss": 0.0314,
-      "step": 3995
-    },
-    {
-      "epoch": 363.27,
-      "learning_rate": 2.8784688995215313e-05,
-      "loss": 0.0296,
-      "step": 3996
-    },
-    {
-      "epoch": 363.36,
-      "learning_rate": 2.8765550239234452e-05,
-      "loss": 0.0331,
-      "step": 3997
-    },
-    {
-      "epoch": 363.45,
-      "learning_rate": 2.8746411483253587e-05,
-      "loss": 0.0361,
-      "step": 3998
-    },
-    {
-      "epoch": 363.55,
-      "learning_rate": 2.872727272727273e-05,
-      "loss": 0.0341,
-      "step": 3999
-    },
-    {
-      "epoch": 363.64,
-      "learning_rate": 2.8708133971291868e-05,
-      "loss": 0.0414,
-      "step": 4000
-    },
-    {
-      "epoch": 363.64,
-      "eval_accuracy_NAT": 0.9928976661163744,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9532691853727627,
-      "eval_accuracy_concrete tiles": 0.9842842291934202,
-      "eval_accuracy_dark bricks": 0.8827989684769639,
-      "eval_accuracy_grass": 0.8955916635502259,
-      "eval_accuracy_red bricks": 0.9287997159090909,
-      "eval_accuracy_road": 0.9345455248994241,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.43539271634549115,
-      "eval_accuracy_speedway bricks": 0.9883240589553562,
-      "eval_accuracy_steel": 0.9051513752391003,
-      "eval_accuracy_tiles": 0.09101492946171758,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803222687979825,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8853520719715183,
-      "eval_iou_concrete tiles": 0.9485532441510991,
-      "eval_iou_dark bricks": 0.8341880341880342,
-      "eval_iou_grass": 0.8283892825738375,
-      "eval_iou_red bricks": 0.8606449489963804,
-      "eval_iou_road": 0.9276335925506722,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.2939654530563621,
-      "eval_iou_speedway bricks": 0.9770225193626434,
-      "eval_iou_steel": 0.8529960218796618,
-      "eval_iou_tiles": 0.09075389237913138,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10924650728702545,
-      "eval_mean_accuracy": 0.6916976948861483,
-      "eval_mean_iou": 0.6522939484544096,
-      "eval_overall_accuracy": 0.9740233951144748,
-      "eval_runtime": 3.3507,
-      "eval_samples_per_second": 10.744,
-      "eval_steps_per_second": 0.597,
-      "step": 4000
-    },
-    {
-      "epoch": 363.73,
-      "learning_rate": 2.8688995215311004e-05,
-      "loss": 0.0324,
-      "step": 4001
-    },
-    {
-      "epoch": 363.82,
-      "learning_rate": 2.8669856459330146e-05,
-      "loss": 0.0281,
-      "step": 4002
-    },
-    {
-      "epoch": 363.91,
-      "learning_rate": 2.8650717703349285e-05,
-      "loss": 0.0341,
-      "step": 4003
-    },
-    {
-      "epoch": 364.0,
-      "learning_rate": 2.863157894736842e-05,
-      "loss": 0.0294,
-      "step": 4004
-    },
-    {
-      "epoch": 364.09,
-      "learning_rate": 2.8612440191387562e-05,
-      "loss": 0.0337,
-      "step": 4005
-    },
-    {
-      "epoch": 364.18,
-      "learning_rate": 2.85933014354067e-05,
-      "loss": 0.0299,
-      "step": 4006
-    },
-    {
-      "epoch": 364.27,
-      "learning_rate": 2.8574162679425836e-05,
-      "loss": 0.0305,
-      "step": 4007
-    },
-    {
-      "epoch": 364.36,
-      "learning_rate": 2.855502392344498e-05,
-      "loss": 0.0258,
-      "step": 4008
-    },
-    {
-      "epoch": 364.45,
-      "learning_rate": 2.8535885167464117e-05,
-      "loss": 0.0312,
-      "step": 4009
-    },
-    {
-      "epoch": 364.55,
-      "learning_rate": 2.8516746411483253e-05,
-      "loss": 0.0274,
-      "step": 4010
-    },
-    {
-      "epoch": 364.64,
-      "learning_rate": 2.8497607655502395e-05,
-      "loss": 0.0322,
-      "step": 4011
-    },
-    {
-      "epoch": 364.73,
-      "learning_rate": 2.8478468899521534e-05,
-      "loss": 0.0313,
-      "step": 4012
-    },
-    {
-      "epoch": 364.82,
-      "learning_rate": 2.845933014354067e-05,
-      "loss": 0.0317,
-      "step": 4013
-    },
-    {
-      "epoch": 364.91,
-      "learning_rate": 2.844019138755981e-05,
-      "loss": 0.0328,
-      "step": 4014
-    },
-    {
-      "epoch": 365.0,
-      "learning_rate": 2.842105263157895e-05,
-      "loss": 0.0314,
-      "step": 4015
-    },
-    {
-      "epoch": 365.09,
-      "learning_rate": 2.8401913875598085e-05,
-      "loss": 0.0329,
-      "step": 4016
-    },
-    {
-      "epoch": 365.18,
-      "learning_rate": 2.8382775119617227e-05,
-      "loss": 0.0252,
-      "step": 4017
-    },
-    {
-      "epoch": 365.27,
-      "learning_rate": 2.8363636363636363e-05,
-      "loss": 0.0324,
-      "step": 4018
-    },
-    {
-      "epoch": 365.36,
-      "learning_rate": 2.83444976076555e-05,
-      "loss": 0.031,
-      "step": 4019
-    },
-    {
-      "epoch": 365.45,
-      "learning_rate": 2.8325358851674644e-05,
-      "loss": 0.0338,
-      "step": 4020
-    },
-    {
-      "epoch": 365.45,
-      "eval_accuracy_NAT": 0.9933084443152038,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9502607408043263,
-      "eval_accuracy_concrete tiles": 0.9873723475814835,
-      "eval_accuracy_dark bricks": 0.8856087140602749,
-      "eval_accuracy_grass": 0.8979578284408126,
-      "eval_accuracy_red bricks": 0.9428267045454546,
-      "eval_accuracy_road": 0.9240822407123691,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4642295922587048,
-      "eval_accuracy_speedway bricks": 0.9884085360481857,
-      "eval_accuracy_steel": 0.9051266407229075,
-      "eval_accuracy_tiles": 0.1655937542802356,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804716231376136,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8884346644061335,
-      "eval_iou_concrete tiles": 0.9483994126284875,
-      "eval_iou_dark bricks": 0.8362651740931889,
-      "eval_iou_grass": 0.8322864464046804,
-      "eval_iou_red bricks": 0.8608949416342413,
-      "eval_iou_road": 0.917478951039632,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31220914063224253,
-      "eval_iou_speedway bricks": 0.977060570163352,
-      "eval_iou_steel": 0.8508044516089032,
-      "eval_iou_tiles": 0.16485988954796482,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11302583664655685,
-      "eval_mean_accuracy": 0.7003673495207661,
-      "eval_mean_iou": 0.659166558868957,
-      "eval_overall_accuracy": 0.9734547932942709,
-      "eval_runtime": 3.8504,
-      "eval_samples_per_second": 9.35,
-      "eval_steps_per_second": 0.519,
-      "step": 4020
-    },
-    {
-      "epoch": 365.55,
-      "learning_rate": 2.830622009569378e-05,
-      "loss": 0.03,
-      "step": 4021
-    },
-    {
-      "epoch": 365.64,
-      "learning_rate": 2.8287081339712918e-05,
-      "loss": 0.0358,
-      "step": 4022
-    },
-    {
-      "epoch": 365.73,
-      "learning_rate": 2.826794258373206e-05,
-      "loss": 0.0296,
-      "step": 4023
-    },
-    {
-      "epoch": 365.82,
-      "learning_rate": 2.8248803827751195e-05,
-      "loss": 0.038,
-      "step": 4024
-    },
-    {
-      "epoch": 365.91,
-      "learning_rate": 2.8229665071770334e-05,
-      "loss": 0.0301,
-      "step": 4025
-    },
-    {
-      "epoch": 366.0,
-      "learning_rate": 2.8210526315789476e-05,
-      "loss": 0.0308,
-      "step": 4026
-    },
-    {
-      "epoch": 366.09,
-      "learning_rate": 2.8191387559808612e-05,
-      "loss": 0.0305,
-      "step": 4027
-    },
-    {
-      "epoch": 366.18,
-      "learning_rate": 2.817224880382775e-05,
-      "loss": 0.0276,
-      "step": 4028
-    },
-    {
-      "epoch": 366.27,
-      "learning_rate": 2.8153110047846893e-05,
-      "loss": 0.0286,
-      "step": 4029
-    },
-    {
-      "epoch": 366.36,
-      "learning_rate": 2.8133971291866028e-05,
-      "loss": 0.0293,
-      "step": 4030
-    },
-    {
-      "epoch": 366.45,
-      "learning_rate": 2.8114832535885167e-05,
-      "loss": 0.0343,
-      "step": 4031
-    },
-    {
-      "epoch": 366.55,
-      "learning_rate": 2.809569377990431e-05,
-      "loss": 0.031,
-      "step": 4032
-    },
-    {
-      "epoch": 366.64,
-      "learning_rate": 2.8076555023923444e-05,
-      "loss": 0.03,
-      "step": 4033
-    },
-    {
-      "epoch": 366.73,
-      "learning_rate": 2.8057416267942587e-05,
-      "loss": 0.0301,
-      "step": 4034
-    },
-    {
-      "epoch": 366.82,
-      "learning_rate": 2.8038277511961725e-05,
-      "loss": 0.0344,
-      "step": 4035
-    },
-    {
-      "epoch": 366.91,
-      "learning_rate": 2.801913875598086e-05,
-      "loss": 0.0296,
-      "step": 4036
-    },
-    {
-      "epoch": 367.0,
-      "learning_rate": 2.8000000000000003e-05,
-      "loss": 0.0296,
-      "step": 4037
-    },
-    {
-      "epoch": 367.09,
-      "learning_rate": 2.7980861244019142e-05,
-      "loss": 0.0303,
-      "step": 4038
-    },
-    {
-      "epoch": 367.18,
-      "learning_rate": 2.7961722488038277e-05,
-      "loss": 0.0335,
-      "step": 4039
-    },
-    {
-      "epoch": 367.27,
-      "learning_rate": 2.794258373205742e-05,
-      "loss": 0.0321,
-      "step": 4040
-    },
-    {
-      "epoch": 367.27,
-      "eval_accuracy_NAT": 0.9930097125752476,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9508871088887935,
-      "eval_accuracy_concrete tiles": 0.9856601235247355,
-      "eval_accuracy_dark bricks": 0.8816442785112197,
-      "eval_accuracy_grass": 0.9042140103524082,
-      "eval_accuracy_red bricks": 0.9312855113636364,
-      "eval_accuracy_road": 0.9234135482178075,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.446668320789568,
-      "eval_accuracy_speedway bricks": 0.9887669602849051,
-      "eval_accuracy_steel": 0.89780522392982,
-      "eval_accuracy_tiles": 0.09437063415970415,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980392074882017,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8862973214218742,
-      "eval_iou_concrete tiles": 0.9470885480933074,
-      "eval_iou_dark bricks": 0.8320377769705776,
-      "eval_iou_grass": 0.8335513512816878,
-      "eval_iou_red bricks": 0.8589911562397642,
-      "eval_iou_road": 0.9167798158639376,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.30153914872233906,
-      "eval_iou_speedway bricks": 0.976940540721009,
-      "eval_iou_steel": 0.8482613030879008,
-      "eval_iou_tiles": 0.09424155382300643,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11465374380350113,
-      "eval_mean_accuracy": 0.6921327255844497,
-      "eval_mean_iou": 0.6520092762390324,
-      "eval_overall_accuracy": 0.9732233683268229,
-      "eval_runtime": 3.4755,
-      "eval_samples_per_second": 10.358,
-      "eval_steps_per_second": 0.575,
-      "step": 4040
-    },
-    {
-      "epoch": 367.36,
-      "learning_rate": 2.7923444976076558e-05,
-      "loss": 0.0325,
-      "step": 4041
-    },
-    {
-      "epoch": 367.45,
-      "learning_rate": 2.7904306220095693e-05,
-      "loss": 0.0276,
-      "step": 4042
-    },
-    {
-      "epoch": 367.55,
-      "learning_rate": 2.7885167464114836e-05,
-      "loss": 0.035,
-      "step": 4043
-    },
-    {
-      "epoch": 367.64,
-      "learning_rate": 2.7866028708133974e-05,
-      "loss": 0.0308,
-      "step": 4044
-    },
-    {
-      "epoch": 367.73,
-      "learning_rate": 2.784688995215311e-05,
-      "loss": 0.0305,
-      "step": 4045
-    },
-    {
-      "epoch": 367.82,
-      "learning_rate": 2.7827751196172252e-05,
-      "loss": 0.038,
-      "step": 4046
-    },
-    {
-      "epoch": 367.91,
-      "learning_rate": 2.780861244019139e-05,
-      "loss": 0.0337,
-      "step": 4047
-    },
-    {
-      "epoch": 368.0,
-      "learning_rate": 2.7789473684210526e-05,
-      "loss": 0.0289,
-      "step": 4048
-    },
-    {
-      "epoch": 368.09,
-      "learning_rate": 2.7770334928229668e-05,
-      "loss": 0.0346,
-      "step": 4049
-    },
-    {
-      "epoch": 368.18,
-      "learning_rate": 2.7751196172248807e-05,
-      "loss": 0.0313,
-      "step": 4050
-    },
-    {
-      "epoch": 368.27,
-      "learning_rate": 2.7732057416267942e-05,
-      "loss": 0.0316,
-      "step": 4051
-    },
-    {
-      "epoch": 368.36,
-      "learning_rate": 2.7712918660287085e-05,
-      "loss": 0.0341,
-      "step": 4052
-    },
-    {
-      "epoch": 368.45,
-      "learning_rate": 2.7693779904306223e-05,
-      "loss": 0.0297,
-      "step": 4053
-    },
-    {
-      "epoch": 368.55,
-      "learning_rate": 2.767464114832536e-05,
-      "loss": 0.0321,
-      "step": 4054
-    },
-    {
-      "epoch": 368.64,
-      "learning_rate": 2.76555023923445e-05,
-      "loss": 0.0303,
-      "step": 4055
-    },
-    {
-      "epoch": 368.73,
-      "learning_rate": 2.7636363636363636e-05,
-      "loss": 0.0321,
-      "step": 4056
-    },
-    {
-      "epoch": 368.82,
-      "learning_rate": 2.7617224880382775e-05,
-      "loss": 0.0335,
-      "step": 4057
-    },
-    {
-      "epoch": 368.91,
-      "learning_rate": 2.7598086124401917e-05,
-      "loss": 0.0361,
-      "step": 4058
-    },
-    {
-      "epoch": 369.0,
-      "learning_rate": 2.7578947368421053e-05,
-      "loss": 0.0282,
-      "step": 4059
-    },
-    {
-      "epoch": 369.09,
-      "learning_rate": 2.755980861244019e-05,
-      "loss": 0.0291,
-      "step": 4060
-    },
-    {
-      "epoch": 369.09,
-      "eval_accuracy_NAT": 0.9929592475559755,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9520969247607194,
-      "eval_accuracy_concrete tiles": 0.9839478994679876,
-      "eval_accuracy_dark bricks": 0.8798737538970787,
-      "eval_accuracy_grass": 0.9074993796244194,
-      "eval_accuracy_red bricks": 0.9261363636363636,
-      "eval_accuracy_road": 0.9230846434103559,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4733824056460728,
-      "eval_accuracy_speedway bricks": 0.9884827552083145,
-      "eval_accuracy_steel": 0.9033210210408285,
-      "eval_accuracy_tiles": 0.15559512395562253,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804961520763141,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8896147922193969,
-      "eval_iou_concrete tiles": 0.9518752957879791,
-      "eval_iou_dark bricks": 0.8304272013949433,
-      "eval_iou_grass": 0.838335668761018,
-      "eval_iou_red bricks": 0.8603001814283358,
-      "eval_iou_road": 0.9162772721216711,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3185477886613238,
-      "eval_iou_speedway bricks": 0.9771759035569633,
-      "eval_iou_steel": 0.851231450547743,
-      "eval_iou_tiles": 0.1553398058252427,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11287505179643631,
-      "eval_mean_accuracy": 0.6989522706310568,
-      "eval_mean_iou": 0.6592016547985331,
-      "eval_overall_accuracy": 0.9735639360215929,
-      "eval_runtime": 3.5902,
-      "eval_samples_per_second": 10.027,
-      "eval_steps_per_second": 0.557,
-      "step": 4060
-    },
-    {
-      "epoch": 369.18,
-      "learning_rate": 2.7540669856459334e-05,
-      "loss": 0.0379,
-      "step": 4061
-    },
-    {
-      "epoch": 369.27,
-      "learning_rate": 2.752153110047847e-05,
-      "loss": 0.0345,
-      "step": 4062
-    },
-    {
-      "epoch": 369.36,
-      "learning_rate": 2.7502392344497608e-05,
-      "loss": 0.0291,
-      "step": 4063
-    },
-    {
-      "epoch": 369.45,
-      "learning_rate": 2.748325358851675e-05,
-      "loss": 0.0345,
-      "step": 4064
-    },
-    {
-      "epoch": 369.55,
-      "learning_rate": 2.7464114832535885e-05,
-      "loss": 0.0337,
-      "step": 4065
-    },
-    {
-      "epoch": 369.64,
-      "learning_rate": 2.7444976076555024e-05,
-      "loss": 0.0265,
-      "step": 4066
-    },
-    {
-      "epoch": 369.73,
-      "learning_rate": 2.7425837320574166e-05,
-      "loss": 0.0261,
-      "step": 4067
-    },
-    {
-      "epoch": 369.82,
-      "learning_rate": 2.74066985645933e-05,
-      "loss": 0.032,
-      "step": 4068
-    },
-    {
-      "epoch": 369.91,
-      "learning_rate": 2.738755980861244e-05,
-      "loss": 0.0285,
-      "step": 4069
-    },
-    {
-      "epoch": 370.0,
-      "learning_rate": 2.7368421052631583e-05,
-      "loss": 0.0304,
-      "step": 4070
-    },
-    {
-      "epoch": 370.09,
-      "learning_rate": 2.7349282296650718e-05,
-      "loss": 0.0296,
-      "step": 4071
-    },
-    {
-      "epoch": 370.18,
-      "learning_rate": 2.7330143540669857e-05,
-      "loss": 0.0315,
-      "step": 4072
-    },
-    {
-      "epoch": 370.27,
-      "learning_rate": 2.7311004784689e-05,
-      "loss": 0.0347,
-      "step": 4073
-    },
-    {
-      "epoch": 370.36,
-      "learning_rate": 2.7291866028708134e-05,
-      "loss": 0.0334,
-      "step": 4074
-    },
-    {
-      "epoch": 370.45,
-      "learning_rate": 2.7272727272727273e-05,
-      "loss": 0.0294,
-      "step": 4075
-    },
-    {
-      "epoch": 370.55,
-      "learning_rate": 2.7253588516746415e-05,
-      "loss": 0.0342,
-      "step": 4076
-    },
-    {
-      "epoch": 370.64,
-      "learning_rate": 2.723444976076555e-05,
-      "loss": 0.0328,
-      "step": 4077
-    },
-    {
-      "epoch": 370.73,
-      "learning_rate": 2.721531100478469e-05,
-      "loss": 0.0287,
-      "step": 4078
-    },
-    {
-      "epoch": 370.82,
-      "learning_rate": 2.719617224880383e-05,
-      "loss": 0.0253,
-      "step": 4079
-    },
-    {
-      "epoch": 370.91,
-      "learning_rate": 2.7177033492822967e-05,
-      "loss": 0.0344,
-      "step": 4080
-    },
-    {
-      "epoch": 370.91,
-      "eval_accuracy_NAT": 0.993318502028835,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9505826430318898,
-      "eval_accuracy_concrete tiles": 0.98217452455207,
-      "eval_accuracy_dark bricks": 0.8806820368730995,
-      "eval_accuracy_grass": 0.9042454660156509,
-      "eval_accuracy_red bricks": 0.931640625,
-      "eval_accuracy_road": 0.9327026906106142,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6770600722300334,
-      "eval_accuracy_speedway bricks": 0.9885630084465025,
-      "eval_accuracy_steel": 0.9077732339555439,
-      "eval_accuracy_tiles": 0.0985481440898507,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806887401620619,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8968771710918448,
-      "eval_iou_concrete tiles": 0.9521029076143336,
-      "eval_iou_dark bricks": 0.8337949129072225,
-      "eval_iou_grass": 0.8409939018840448,
-      "eval_iou_red bricks": 0.8678465100893152,
-      "eval_iou_road": 0.9255699817605837,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.4496256018747368,
-      "eval_iou_speedway bricks": 0.9770553318785549,
-      "eval_iou_steel": 0.8514774915511147,
-      "eval_iou_tiles": 0.09843354538614132,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10659582167863846,
-      "eval_mean_accuracy": 0.7113300728333916,
-      "eval_mean_iou": 0.6672666227846118,
-      "eval_overall_accuracy": 0.9751694997151693,
-      "eval_runtime": 3.1337,
-      "eval_samples_per_second": 11.488,
-      "eval_steps_per_second": 0.638,
-      "step": 4080
-    },
-    {
-      "epoch": 371.0,
-      "learning_rate": 2.7157894736842106e-05,
-      "loss": 0.0309,
-      "step": 4081
-    },
-    {
-      "epoch": 371.09,
-      "learning_rate": 2.7138755980861248e-05,
-      "loss": 0.0312,
-      "step": 4082
-    },
-    {
-      "epoch": 371.18,
-      "learning_rate": 2.7119617224880383e-05,
-      "loss": 0.0305,
-      "step": 4083
-    },
-    {
-      "epoch": 371.27,
-      "learning_rate": 2.7100478468899522e-05,
-      "loss": 0.0315,
-      "step": 4084
-    },
-    {
-      "epoch": 371.36,
-      "learning_rate": 2.7081339712918664e-05,
-      "loss": 0.0293,
-      "step": 4085
-    },
-    {
-      "epoch": 371.45,
-      "learning_rate": 2.70622009569378e-05,
-      "loss": 0.0277,
-      "step": 4086
-    },
-    {
-      "epoch": 371.55,
-      "learning_rate": 2.704306220095694e-05,
-      "loss": 0.0282,
-      "step": 4087
-    },
-    {
-      "epoch": 371.64,
-      "learning_rate": 2.702392344497608e-05,
-      "loss": 0.0322,
-      "step": 4088
-    },
-    {
-      "epoch": 371.73,
-      "learning_rate": 2.7004784688995216e-05,
-      "loss": 0.0306,
-      "step": 4089
-    },
-    {
-      "epoch": 371.82,
-      "learning_rate": 2.698564593301435e-05,
-      "loss": 0.0304,
-      "step": 4090
-    },
-    {
-      "epoch": 371.91,
-      "learning_rate": 2.6966507177033497e-05,
-      "loss": 0.0324,
-      "step": 4091
-    },
-    {
-      "epoch": 372.0,
-      "learning_rate": 2.6947368421052632e-05,
-      "loss": 0.0318,
-      "step": 4092
-    },
-    {
-      "epoch": 372.09,
-      "learning_rate": 2.6928229665071768e-05,
-      "loss": 0.033,
-      "step": 4093
-    },
-    {
-      "epoch": 372.18,
-      "learning_rate": 2.6909090909090913e-05,
-      "loss": 0.0279,
-      "step": 4094
-    },
-    {
-      "epoch": 372.27,
-      "learning_rate": 2.688995215311005e-05,
-      "loss": 0.0304,
-      "step": 4095
-    },
-    {
-      "epoch": 372.36,
-      "learning_rate": 2.6870813397129184e-05,
-      "loss": 0.033,
-      "step": 4096
-    },
-    {
-      "epoch": 372.45,
-      "learning_rate": 2.6851674641148326e-05,
-      "loss": 0.0355,
-      "step": 4097
-    },
-    {
-      "epoch": 372.55,
-      "learning_rate": 2.6832535885167465e-05,
-      "loss": 0.0336,
-      "step": 4098
-    },
-    {
-      "epoch": 372.64,
-      "learning_rate": 2.68133971291866e-05,
-      "loss": 0.0298,
-      "step": 4099
-    },
-    {
-      "epoch": 372.73,
-      "learning_rate": 2.6794258373205743e-05,
-      "loss": 0.0309,
-      "step": 4100
-    },
-    {
-      "epoch": 372.73,
-      "eval_accuracy_NAT": 0.9929324269862926,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9500930833941371,
-      "eval_accuracy_concrete tiles": 0.9871888950039748,
-      "eval_accuracy_dark bricks": 0.8884569493091105,
-      "eval_accuracy_grass": 0.9113509508347983,
-      "eval_accuracy_red bricks": 0.9424715909090909,
-      "eval_accuracy_road": 0.9324838238085967,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6373886913130979,
-      "eval_accuracy_speedway bricks": 0.9889039338568499,
-      "eval_accuracy_steel": 0.9056295758854956,
-      "eval_accuracy_tiles": 0.09615121216271744,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808058128456946,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8964061952289296,
-      "eval_iou_concrete tiles": 0.9527561378659113,
-      "eval_iou_dark bricks": 0.8328702868482771,
-      "eval_iou_grass": 0.8428979192055678,
-      "eval_iou_red bricks": 0.859177727419877,
-      "eval_iou_road": 0.9247890553420925,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.42798963346908553,
-      "eval_iou_speedway bricks": 0.9769267727406037,
-      "eval_iou_steel": 0.8497559239689625,
-      "eval_iou_tiles": 0.0959016393442623,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10778361558914185,
-      "eval_mean_accuracy": 0.7102347025741663,
-      "eval_mean_iou": 0.6646367003291741,
-      "eval_overall_accuracy": 0.9750165939331055,
-      "eval_runtime": 3.5145,
-      "eval_samples_per_second": 10.243,
-      "eval_steps_per_second": 0.569,
-      "step": 4100
-    },
-    {
-      "epoch": 372.82,
-      "learning_rate": 2.677511961722488e-05,
-      "loss": 0.031,
-      "step": 4101
-    },
-    {
-      "epoch": 372.91,
-      "learning_rate": 2.6755980861244017e-05,
-      "loss": 0.0317,
-      "step": 4102
-    },
-    {
-      "epoch": 373.0,
-      "learning_rate": 2.673684210526316e-05,
-      "loss": 0.031,
-      "step": 4103
-    },
-    {
-      "epoch": 373.09,
-      "learning_rate": 2.6717703349282298e-05,
-      "loss": 0.0346,
-      "step": 4104
-    },
-    {
-      "epoch": 373.18,
-      "learning_rate": 2.669856459330144e-05,
-      "loss": 0.0299,
-      "step": 4105
-    },
-    {
-      "epoch": 373.27,
-      "learning_rate": 2.6679425837320575e-05,
-      "loss": 0.0289,
-      "step": 4106
-    },
-    {
-      "epoch": 373.36,
-      "learning_rate": 2.6660287081339714e-05,
-      "loss": 0.031,
-      "step": 4107
-    },
-    {
-      "epoch": 373.45,
-      "learning_rate": 2.6641148325358856e-05,
-      "loss": 0.0319,
-      "step": 4108
-    },
-    {
-      "epoch": 373.55,
-      "learning_rate": 2.662200956937799e-05,
-      "loss": 0.0273,
-      "step": 4109
-    },
-    {
-      "epoch": 373.64,
-      "learning_rate": 2.660287081339713e-05,
-      "loss": 0.033,
-      "step": 4110
-    },
-    {
-      "epoch": 373.73,
-      "learning_rate": 2.6583732057416272e-05,
-      "loss": 0.031,
-      "step": 4111
-    },
-    {
-      "epoch": 373.82,
-      "learning_rate": 2.6564593301435408e-05,
-      "loss": 0.0322,
-      "step": 4112
-    },
-    {
-      "epoch": 373.91,
-      "learning_rate": 2.6545454545454547e-05,
-      "loss": 0.0277,
-      "step": 4113
-    },
-    {
-      "epoch": 374.0,
-      "learning_rate": 2.652631578947369e-05,
-      "loss": 0.0302,
-      "step": 4114
-    },
-    {
-      "epoch": 374.09,
-      "learning_rate": 2.6507177033492824e-05,
-      "loss": 0.032,
-      "step": 4115
-    },
-    {
-      "epoch": 374.18,
-      "learning_rate": 2.6488038277511963e-05,
-      "loss": 0.0318,
-      "step": 4116
-    },
-    {
-      "epoch": 374.27,
-      "learning_rate": 2.6468899521531105e-05,
-      "loss": 0.029,
-      "step": 4117
-    },
-    {
-      "epoch": 374.36,
-      "learning_rate": 2.644976076555024e-05,
-      "loss": 0.0296,
-      "step": 4118
-    },
-    {
-      "epoch": 374.45,
-      "learning_rate": 2.643062200956938e-05,
-      "loss": 0.0342,
-      "step": 4119
-    },
-    {
-      "epoch": 374.55,
-      "learning_rate": 2.641148325358852e-05,
-      "loss": 0.032,
-      "step": 4120
-    },
-    {
-      "epoch": 374.55,
-      "eval_accuracy_NAT": 0.9933059739995751,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9509488068157431,
-      "eval_accuracy_concrete tiles": 0.9861799058276769,
-      "eval_accuracy_dark bricks": 0.874831607713329,
-      "eval_accuracy_grass": 0.9038854734252072,
-      "eval_accuracy_red bricks": 0.9236505681818182,
-      "eval_accuracy_road": 0.9329662981401159,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6269952857497312,
-      "eval_accuracy_speedway bricks": 0.9880320095772887,
-      "eval_accuracy_steel": 0.9096035881538157,
-      "eval_accuracy_tiles": 0.08676893576222436,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803401003126476,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8952129264748032,
-      "eval_iou_concrete tiles": 0.9546557745811874,
-      "eval_iou_dark bricks": 0.8326556031798367,
-      "eval_iou_grass": 0.8383237220941804,
-      "eval_iou_red bricks": 0.860545905707196,
-      "eval_iou_road": 0.9261716527379977,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.4209717723276261,
-      "eval_iou_speedway bricks": 0.977232795087994,
-      "eval_iou_steel": 0.8512785691137209,
-      "eval_iou_tiles": 0.08663247863247864,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10789303481578827,
-      "eval_mean_accuracy": 0.7059360348728096,
-      "eval_mean_iou": 0.6633862538653592,
-      "eval_overall_accuracy": 0.9749157163831923,
-      "eval_runtime": 3.568,
-      "eval_samples_per_second": 10.09,
-      "eval_steps_per_second": 0.561,
-      "step": 4120
-    },
-    {
-      "epoch": 374.64,
-      "learning_rate": 2.6392344497607657e-05,
-      "loss": 0.0333,
-      "step": 4121
-    },
-    {
-      "epoch": 374.73,
-      "learning_rate": 2.6373205741626796e-05,
-      "loss": 0.033,
-      "step": 4122
-    },
-    {
-      "epoch": 374.82,
-      "learning_rate": 2.6354066985645938e-05,
-      "loss": 0.0334,
-      "step": 4123
-    },
-    {
-      "epoch": 374.91,
-      "learning_rate": 2.6334928229665073e-05,
-      "loss": 0.0254,
-      "step": 4124
-    },
-    {
-      "epoch": 375.0,
-      "learning_rate": 2.6315789473684212e-05,
-      "loss": 0.0302,
-      "step": 4125
-    },
-    {
-      "epoch": 375.09,
-      "learning_rate": 2.6296650717703354e-05,
-      "loss": 0.0389,
-      "step": 4126
-    },
-    {
-      "epoch": 375.18,
-      "learning_rate": 2.627751196172249e-05,
-      "loss": 0.0279,
-      "step": 4127
-    },
-    {
-      "epoch": 375.27,
-      "learning_rate": 2.6258373205741625e-05,
-      "loss": 0.0324,
-      "step": 4128
-    },
-    {
-      "epoch": 375.36,
-      "learning_rate": 2.623923444976077e-05,
-      "loss": 0.0311,
-      "step": 4129
-    },
-    {
-      "epoch": 375.45,
-      "learning_rate": 2.6220095693779906e-05,
-      "loss": 0.0288,
-      "step": 4130
-    },
-    {
-      "epoch": 375.55,
-      "learning_rate": 2.620095693779904e-05,
-      "loss": 0.0333,
-      "step": 4131
-    },
-    {
-      "epoch": 375.64,
-      "learning_rate": 2.6181818181818187e-05,
-      "loss": 0.0336,
-      "step": 4132
-    },
-    {
-      "epoch": 375.73,
-      "learning_rate": 2.6162679425837322e-05,
-      "loss": 0.0319,
-      "step": 4133
-    },
-    {
-      "epoch": 375.82,
-      "learning_rate": 2.6143540669856458e-05,
-      "loss": 0.0368,
-      "step": 4134
-    },
-    {
-      "epoch": 375.91,
-      "learning_rate": 2.61244019138756e-05,
-      "loss": 0.0309,
-      "step": 4135
-    },
-    {
-      "epoch": 376.0,
-      "learning_rate": 2.610526315789474e-05,
-      "loss": 0.0285,
-      "step": 4136
-    },
-    {
-      "epoch": 376.09,
-      "learning_rate": 2.6086124401913874e-05,
-      "loss": 0.03,
-      "step": 4137
-    },
-    {
-      "epoch": 376.18,
-      "learning_rate": 2.6066985645933016e-05,
-      "loss": 0.034,
-      "step": 4138
-    },
-    {
-      "epoch": 376.27,
-      "learning_rate": 2.6047846889952155e-05,
-      "loss": 0.0301,
-      "step": 4139
-    },
-    {
-      "epoch": 376.36,
-      "learning_rate": 2.602870813397129e-05,
-      "loss": 0.034,
-      "step": 4140
-    },
-    {
-      "epoch": 376.36,
-      "eval_accuracy_NAT": 0.9928166750539764,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9525998969912872,
-      "eval_accuracy_concrete tiles": 0.9856906989543203,
-      "eval_accuracy_dark bricks": 0.8747931180478041,
-      "eval_accuracy_grass": 0.9109245518441756,
-      "eval_accuracy_red bricks": 0.9248934659090909,
-      "eval_accuracy_road": 0.9340787702829669,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6111708433269926,
-      "eval_accuracy_speedway bricks": 0.9888490237465108,
-      "eval_accuracy_steel": 0.9032962865246356,
-      "eval_accuracy_tiles": 0.09704150116422408,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807343800738734,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8939926615435934,
-      "eval_iou_concrete tiles": 0.9555963955418544,
-      "eval_iou_dark bricks": 0.8301859224896811,
-      "eval_iou_grass": 0.8421469284357188,
-      "eval_iou_red bricks": 0.8639907115607895,
-      "eval_iou_road": 0.9272203710461119,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.409105168945727,
-      "eval_iou_speedway bricks": 0.9772470987143761,
-      "eval_iou_steel": 0.8521018860587206,
-      "eval_iou_tiles": 0.0968293016263496,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10703253000974655,
-      "eval_mean_accuracy": 0.7058580639881527,
-      "eval_mean_iou": 0.6637808327720612,
-      "eval_overall_accuracy": 0.9750796424018012,
-      "eval_runtime": 3.4098,
-      "eval_samples_per_second": 10.558,
-      "eval_steps_per_second": 0.587,
-      "step": 4140
-    },
-    {
-      "epoch": 376.45,
-      "learning_rate": 2.6009569377990432e-05,
-      "loss": 0.0338,
-      "step": 4141
-    },
-    {
-      "epoch": 376.55,
-      "learning_rate": 2.599043062200957e-05,
-      "loss": 0.0322,
-      "step": 4142
-    },
-    {
-      "epoch": 376.64,
-      "learning_rate": 2.5971291866028707e-05,
-      "loss": 0.0306,
-      "step": 4143
-    },
-    {
-      "epoch": 376.73,
-      "learning_rate": 2.595215311004785e-05,
-      "loss": 0.0326,
-      "step": 4144
-    },
-    {
-      "epoch": 376.82,
-      "learning_rate": 2.5933014354066988e-05,
-      "loss": 0.03,
-      "step": 4145
-    },
-    {
-      "epoch": 376.91,
-      "learning_rate": 2.5913875598086123e-05,
-      "loss": 0.0282,
-      "step": 4146
-    },
-    {
-      "epoch": 377.0,
-      "learning_rate": 2.5894736842105265e-05,
-      "loss": 0.0292,
-      "step": 4147
-    },
-    {
-      "epoch": 377.09,
-      "learning_rate": 2.5875598086124404e-05,
-      "loss": 0.0281,
-      "step": 4148
-    },
-    {
-      "epoch": 377.18,
-      "learning_rate": 2.585645933014354e-05,
-      "loss": 0.0327,
-      "step": 4149
-    },
-    {
-      "epoch": 377.27,
-      "learning_rate": 2.583732057416268e-05,
-      "loss": 0.0342,
-      "step": 4150
-    },
-    {
-      "epoch": 377.36,
-      "learning_rate": 2.581818181818182e-05,
-      "loss": 0.0307,
-      "step": 4151
-    },
-    {
-      "epoch": 377.45,
-      "learning_rate": 2.5799043062200956e-05,
-      "loss": 0.0342,
-      "step": 4152
-    },
-    {
-      "epoch": 377.55,
-      "learning_rate": 2.5779904306220098e-05,
-      "loss": 0.0348,
-      "step": 4153
-    },
-    {
-      "epoch": 377.64,
-      "learning_rate": 2.5760765550239237e-05,
-      "loss": 0.0293,
-      "step": 4154
-    },
-    {
-      "epoch": 377.73,
-      "learning_rate": 2.5741626794258372e-05,
-      "loss": 0.0311,
-      "step": 4155
-    },
-    {
-      "epoch": 377.82,
-      "learning_rate": 2.5722488038277514e-05,
-      "loss": 0.0264,
-      "step": 4156
-    },
-    {
-      "epoch": 377.91,
-      "learning_rate": 2.5703349282296653e-05,
-      "loss": 0.0326,
-      "step": 4157
-    },
-    {
-      "epoch": 378.0,
-      "learning_rate": 2.5684210526315788e-05,
-      "loss": 0.0365,
-      "step": 4158
-    },
-    {
-      "epoch": 378.09,
-      "learning_rate": 2.566507177033493e-05,
-      "loss": 0.0291,
-      "step": 4159
-    },
-    {
-      "epoch": 378.18,
-      "learning_rate": 2.564593301435407e-05,
-      "loss": 0.0339,
-      "step": 4160
-    },
-    {
-      "epoch": 378.18,
-      "eval_accuracy_NAT": 0.9922333276633708,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9493795334563715,
-      "eval_accuracy_concrete tiles": 0.9890845716382315,
-      "eval_accuracy_dark bricks": 0.8846464724221547,
-      "eval_accuracy_grass": 0.923087408298004,
-      "eval_accuracy_red bricks": 0.9229403409090909,
-      "eval_accuracy_road": 0.9353339290702273,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5771510489896066,
-      "eval_accuracy_speedway bricks": 0.9895405290921006,
-      "eval_accuracy_steel": 0.9008558142602731,
-      "eval_accuracy_tiles": 0.1073140665662238,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806945376268775,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8927920950929841,
-      "eval_iou_concrete tiles": 0.9525898878058836,
-      "eval_iou_dark bricks": 0.8315484804630969,
-      "eval_iou_grass": 0.8430132718788104,
-      "eval_iou_red bricks": 0.8586058804096465,
-      "eval_iou_road": 0.9275010761765098,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3883107970248363,
-      "eval_iou_speedway bricks": 0.9770536106871047,
-      "eval_iou_steel": 0.8506134587238813,
-      "eval_iou_tiles": 0.10722594772136308,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1065874919295311,
-      "eval_mean_accuracy": 0.7055051571050505,
-      "eval_mean_iou": 0.6623037725854609,
-      "eval_overall_accuracy": 0.9749665790134006,
-      "eval_runtime": 3.2902,
-      "eval_samples_per_second": 10.942,
-      "eval_steps_per_second": 0.608,
-      "step": 4160
-    },
-    {
-      "epoch": 378.27,
-      "learning_rate": 2.5626794258373205e-05,
-      "loss": 0.0334,
-      "step": 4161
-    },
-    {
-      "epoch": 378.36,
-      "learning_rate": 2.5607655502392347e-05,
-      "loss": 0.0334,
-      "step": 4162
-    },
-    {
-      "epoch": 378.45,
-      "learning_rate": 2.5588516746411485e-05,
-      "loss": 0.0346,
-      "step": 4163
-    },
-    {
-      "epoch": 378.55,
-      "learning_rate": 2.556937799043062e-05,
-      "loss": 0.0316,
-      "step": 4164
-    },
-    {
-      "epoch": 378.64,
-      "learning_rate": 2.5550239234449763e-05,
-      "loss": 0.0265,
-      "step": 4165
-    },
-    {
-      "epoch": 378.73,
-      "learning_rate": 2.55311004784689e-05,
-      "loss": 0.026,
-      "step": 4166
-    },
-    {
-      "epoch": 378.82,
-      "learning_rate": 2.5511961722488037e-05,
-      "loss": 0.0302,
-      "step": 4167
-    },
-    {
-      "epoch": 378.91,
-      "learning_rate": 2.549282296650718e-05,
-      "loss": 0.0312,
-      "step": 4168
-    },
-    {
-      "epoch": 379.0,
-      "learning_rate": 2.5473684210526315e-05,
-      "loss": 0.032,
-      "step": 4169
-    },
-    {
-      "epoch": 379.09,
-      "learning_rate": 2.5454545454545454e-05,
-      "loss": 0.0293,
-      "step": 4170
-    },
-    {
-      "epoch": 379.18,
-      "learning_rate": 2.5435406698564596e-05,
-      "loss": 0.0316,
-      "step": 4171
-    },
-    {
-      "epoch": 379.27,
-      "learning_rate": 2.541626794258373e-05,
-      "loss": 0.0295,
-      "step": 4172
-    },
-    {
-      "epoch": 379.36,
-      "learning_rate": 2.539712918660287e-05,
-      "loss": 0.0276,
-      "step": 4173
-    },
-    {
-      "epoch": 379.45,
-      "learning_rate": 2.5377990430622012e-05,
-      "loss": 0.0278,
-      "step": 4174
-    },
-    {
-      "epoch": 379.55,
-      "learning_rate": 2.5358851674641147e-05,
-      "loss": 0.0342,
-      "step": 4175
-    },
-    {
-      "epoch": 379.64,
-      "learning_rate": 2.533971291866029e-05,
-      "loss": 0.0373,
-      "step": 4176
-    },
-    {
-      "epoch": 379.73,
-      "learning_rate": 2.532057416267943e-05,
-      "loss": 0.0295,
-      "step": 4177
-    },
-    {
-      "epoch": 379.82,
-      "learning_rate": 2.5301435406698564e-05,
-      "loss": 0.0344,
-      "step": 4178
-    },
-    {
-      "epoch": 379.91,
-      "learning_rate": 2.5282296650717706e-05,
-      "loss": 0.033,
-      "step": 4179
-    },
-    {
-      "epoch": 380.0,
-      "learning_rate": 2.5263157894736845e-05,
-      "loss": 0.0322,
-      "step": 4180
-    },
-    {
-      "epoch": 380.0,
-      "eval_accuracy_NAT": 0.993248627386766,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.950504849993562,
-      "eval_accuracy_concrete tiles": 0.9844982572005136,
-      "eval_accuracy_dark bricks": 0.8757938493514491,
-      "eval_accuracy_grass": 0.9031305375073834,
-      "eval_accuracy_red bricks": 0.9229403409090909,
-      "eval_accuracy_road": 0.9287775986805114,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6225842913461803,
-      "eval_accuracy_speedway bricks": 0.9889522064813239,
-      "eval_accuracy_steel": 0.9081937207308225,
-      "eval_accuracy_tiles": 0.12121627174359677,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804441707165582,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8947418923935083,
-      "eval_iou_concrete tiles": 0.9528586647727273,
-      "eval_iou_dark bricks": 0.832229984272704,
-      "eval_iou_grass": 0.8392858302856289,
-      "eval_iou_red bricks": 0.861023687261885,
-      "eval_iou_road": 0.9224951628535981,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.4172378752886836,
-      "eval_iou_speedway bricks": 0.9771340572754513,
-      "eval_iou_steel": 0.8496771854583041,
-      "eval_iou_tiles": 0.12090989821709133,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10924029350280762,
-      "eval_mean_accuracy": 0.7076800424100924,
-      "eval_mean_iou": 0.6652337237535492,
-      "eval_overall_accuracy": 0.9746324751112196,
-      "eval_runtime": 3.6447,
-      "eval_samples_per_second": 9.877,
-      "eval_steps_per_second": 0.549,
-      "step": 4180
-    },
-    {
-      "epoch": 380.09,
-      "learning_rate": 2.524401913875598e-05,
-      "loss": 0.0278,
-      "step": 4181
-    },
-    {
-      "epoch": 380.18,
-      "learning_rate": 2.5224880382775122e-05,
-      "loss": 0.0295,
-      "step": 4182
-    },
-    {
-      "epoch": 380.27,
-      "learning_rate": 2.520574162679426e-05,
-      "loss": 0.0298,
-      "step": 4183
-    },
-    {
-      "epoch": 380.36,
-      "learning_rate": 2.5186602870813396e-05,
-      "loss": 0.0392,
-      "step": 4184
-    },
-    {
-      "epoch": 380.45,
-      "learning_rate": 2.516746411483254e-05,
-      "loss": 0.0329,
-      "step": 4185
-    },
-    {
-      "epoch": 380.55,
-      "learning_rate": 2.5148325358851677e-05,
-      "loss": 0.0293,
-      "step": 4186
-    },
-    {
-      "epoch": 380.64,
-      "learning_rate": 2.5129186602870813e-05,
-      "loss": 0.0321,
-      "step": 4187
-    },
-    {
-      "epoch": 380.73,
-      "learning_rate": 2.5110047846889955e-05,
-      "loss": 0.0283,
-      "step": 4188
-    },
-    {
-      "epoch": 380.82,
-      "learning_rate": 2.5090909090909094e-05,
-      "loss": 0.0311,
-      "step": 4189
-    },
-    {
-      "epoch": 380.91,
-      "learning_rate": 2.507177033492823e-05,
-      "loss": 0.0307,
-      "step": 4190
-    },
-    {
-      "epoch": 381.0,
-      "learning_rate": 2.505263157894737e-05,
-      "loss": 0.0335,
-      "step": 4191
-    },
-    {
-      "epoch": 381.09,
-      "learning_rate": 2.503349282296651e-05,
-      "loss": 0.0348,
-      "step": 4192
-    },
-    {
-      "epoch": 381.18,
-      "learning_rate": 2.5014354066985645e-05,
-      "loss": 0.0339,
-      "step": 4193
-    },
-    {
-      "epoch": 381.27,
-      "learning_rate": 2.4995215311004784e-05,
-      "loss": 0.0265,
-      "step": 4194
-    },
-    {
-      "epoch": 381.36,
-      "learning_rate": 2.4976076555023926e-05,
-      "loss": 0.0343,
-      "step": 4195
-    },
-    {
-      "epoch": 381.45,
-      "learning_rate": 2.4956937799043065e-05,
-      "loss": 0.034,
-      "step": 4196
-    },
-    {
-      "epoch": 381.55,
-      "learning_rate": 2.49377990430622e-05,
-      "loss": 0.0309,
-      "step": 4197
-    },
-    {
-      "epoch": 381.64,
-      "learning_rate": 2.4918660287081343e-05,
-      "loss": 0.032,
-      "step": 4198
-    },
-    {
-      "epoch": 381.73,
-      "learning_rate": 2.489952153110048e-05,
-      "loss": 0.033,
-      "step": 4199
-    },
-    {
-      "epoch": 381.82,
-      "learning_rate": 2.4880382775119617e-05,
-      "loss": 0.033,
-      "step": 4200
-    },
-    {
-      "epoch": 381.82,
-      "eval_accuracy_NAT": 0.9930432382873513,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9522927486158205,
-      "eval_accuracy_concrete tiles": 0.9872806212927292,
-      "eval_accuracy_dark bricks": 0.8829529271390631,
-      "eval_accuracy_grass": 0.905073798481041,
-      "eval_accuracy_red bricks": 0.9376775568181818,
-      "eval_accuracy_road": 0.9377644388605867,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5947398891737656,
-      "eval_accuracy_speedway bricks": 0.9885074949283574,
-      "eval_accuracy_steel": 0.907921641052701,
-      "eval_accuracy_tiles": 0.15634844541843584,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980663838797888,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8956762743851363,
-      "eval_iou_concrete tiles": 0.9451469382976232,
-      "eval_iou_dark bricks": 0.8355795148247979,
-      "eval_iou_grass": 0.8418682826286257,
-      "eval_iou_red bricks": 0.8694435297991439,
-      "eval_iou_road": 0.929620297471207,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.4017617699642431,
-      "eval_iou_speedway bricks": 0.977370015255298,
-      "eval_iou_steel": 0.8527509970186239,
-      "eval_iou_tiles": 0.15572987721691678,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1055811271071434,
-      "eval_mean_accuracy": 0.7110463692360026,
-      "eval_mean_iou": 0.6681239488968849,
-      "eval_overall_accuracy": 0.9754007127549913,
-      "eval_runtime": 3.2696,
-      "eval_samples_per_second": 11.011,
-      "eval_steps_per_second": 0.612,
-      "step": 4200
-    },
-    {
-      "epoch": 381.91,
-      "learning_rate": 2.486124401913876e-05,
-      "loss": 0.0402,
-      "step": 4201
-    },
-    {
-      "epoch": 382.0,
-      "learning_rate": 2.4842105263157898e-05,
-      "loss": 0.0284,
-      "step": 4202
-    },
-    {
-      "epoch": 382.09,
-      "learning_rate": 2.4822966507177033e-05,
-      "loss": 0.0375,
-      "step": 4203
-    },
-    {
-      "epoch": 382.18,
-      "learning_rate": 2.4803827751196172e-05,
-      "loss": 0.0328,
-      "step": 4204
-    },
-    {
-      "epoch": 382.27,
-      "learning_rate": 2.4784688995215314e-05,
-      "loss": 0.0351,
-      "step": 4205
-    },
-    {
-      "epoch": 382.36,
-      "learning_rate": 2.476555023923445e-05,
-      "loss": 0.0264,
-      "step": 4206
-    },
-    {
-      "epoch": 382.45,
-      "learning_rate": 2.4746411483253588e-05,
-      "loss": 0.0326,
-      "step": 4207
-    },
-    {
-      "epoch": 382.55,
-      "learning_rate": 2.472727272727273e-05,
-      "loss": 0.0309,
-      "step": 4208
-    },
-    {
-      "epoch": 382.64,
-      "learning_rate": 2.4708133971291866e-05,
-      "loss": 0.0306,
-      "step": 4209
-    },
-    {
-      "epoch": 382.73,
-      "learning_rate": 2.4688995215311005e-05,
-      "loss": 0.0321,
-      "step": 4210
-    },
-    {
-      "epoch": 382.82,
-      "learning_rate": 2.4669856459330147e-05,
-      "loss": 0.0277,
-      "step": 4211
-    },
-    {
-      "epoch": 382.91,
-      "learning_rate": 2.4650717703349282e-05,
-      "loss": 0.0321,
-      "step": 4212
-    },
-    {
-      "epoch": 383.0,
-      "learning_rate": 2.463157894736842e-05,
-      "loss": 0.0313,
-      "step": 4213
-    },
-    {
-      "epoch": 383.09,
-      "learning_rate": 2.4612440191387563e-05,
-      "loss": 0.0322,
-      "step": 4214
-    },
-    {
-      "epoch": 383.18,
-      "learning_rate": 2.45933014354067e-05,
-      "loss": 0.0287,
-      "step": 4215
-    },
-    {
-      "epoch": 383.27,
-      "learning_rate": 2.4574162679425837e-05,
-      "loss": 0.031,
-      "step": 4216
-    },
-    {
-      "epoch": 383.36,
-      "learning_rate": 2.455502392344498e-05,
-      "loss": 0.029,
-      "step": 4217
-    },
-    {
-      "epoch": 383.45,
-      "learning_rate": 2.4535885167464115e-05,
-      "loss": 0.0302,
-      "step": 4218
-    },
-    {
-      "epoch": 383.55,
-      "learning_rate": 2.4516746411483254e-05,
-      "loss": 0.0285,
-      "step": 4219
-    },
-    {
-      "epoch": 383.64,
-      "learning_rate": 2.4497607655502396e-05,
-      "loss": 0.0313,
-      "step": 4220
-    },
-    {
-      "epoch": 383.64,
-      "eval_accuracy_NAT": 0.9930308867092078,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9510882977810207,
-      "eval_accuracy_concrete tiles": 0.9852932183697181,
-      "eval_accuracy_dark bricks": 0.8870713213502175,
-      "eval_accuracy_grass": 0.9022602641576698,
-      "eval_accuracy_red bricks": 0.92578125,
-      "eval_accuracy_road": 0.9402433170049832,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5171339563862928,
-      "eval_accuracy_speedway bricks": 0.988976342793561,
-      "eval_accuracy_steel": 0.9061490007255458,
-      "eval_accuracy_tiles": 0.09984933570743734,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980622934200569,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8893444370586834,
-      "eval_iou_concrete tiles": 0.950589970501475,
-      "eval_iou_dark bricks": 0.8370682453782733,
-      "eval_iou_grass": 0.83498398939095,
-      "eval_iou_red bricks": 0.865250580816462,
-      "eval_iou_road": 0.9327518596401997,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3484803448019618,
-      "eval_iou_speedway bricks": 0.9774673523512808,
-      "eval_iou_steel": 0.8540224258106628,
-      "eval_iou_tiles": 0.0995833617922273,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10420788079500198,
-      "eval_mean_accuracy": 0.6997597839219734,
-      "eval_mean_iou": 0.6592435001340573,
-      "eval_overall_accuracy": 0.9751012590196397,
-      "eval_runtime": 3.507,
-      "eval_samples_per_second": 10.265,
-      "eval_steps_per_second": 0.57,
-      "step": 4220
-    },
-    {
-      "epoch": 383.73,
-      "learning_rate": 2.447846889952153e-05,
-      "loss": 0.0276,
-      "step": 4221
-    },
-    {
-      "epoch": 383.82,
-      "learning_rate": 2.445933014354067e-05,
-      "loss": 0.0378,
-      "step": 4222
-    },
-    {
-      "epoch": 383.91,
-      "learning_rate": 2.444019138755981e-05,
-      "loss": 0.0316,
-      "step": 4223
-    },
-    {
-      "epoch": 384.0,
-      "learning_rate": 2.4421052631578948e-05,
-      "loss": 0.0316,
-      "step": 4224
-    },
-    {
-      "epoch": 384.09,
-      "learning_rate": 2.4401913875598086e-05,
-      "loss": 0.0294,
-      "step": 4225
-    },
-    {
-      "epoch": 384.18,
-      "learning_rate": 2.4382775119617225e-05,
-      "loss": 0.0309,
-      "step": 4226
-    },
-    {
-      "epoch": 384.27,
-      "learning_rate": 2.4363636363636364e-05,
-      "loss": 0.0333,
-      "step": 4227
-    },
-    {
-      "epoch": 384.36,
-      "learning_rate": 2.4344497607655503e-05,
-      "loss": 0.0336,
-      "step": 4228
-    },
-    {
-      "epoch": 384.45,
-      "learning_rate": 2.432535885167464e-05,
-      "loss": 0.0309,
-      "step": 4229
-    },
-    {
-      "epoch": 384.55,
-      "learning_rate": 2.4306220095693784e-05,
-      "loss": 0.0304,
-      "step": 4230
-    },
-    {
-      "epoch": 384.64,
-      "learning_rate": 2.428708133971292e-05,
-      "loss": 0.0322,
-      "step": 4231
-    },
-    {
-      "epoch": 384.73,
-      "learning_rate": 2.4267942583732058e-05,
-      "loss": 0.0294,
-      "step": 4232
-    },
-    {
-      "epoch": 384.82,
-      "learning_rate": 2.42488038277512e-05,
-      "loss": 0.0309,
-      "step": 4233
-    },
-    {
-      "epoch": 384.91,
-      "learning_rate": 2.4229665071770335e-05,
-      "loss": 0.032,
-      "step": 4234
-    },
-    {
-      "epoch": 385.0,
-      "learning_rate": 2.4210526315789474e-05,
-      "loss": 0.033,
-      "step": 4235
-    },
-    {
-      "epoch": 385.09,
-      "learning_rate": 2.4191387559808616e-05,
-      "loss": 0.0272,
-      "step": 4236
-    },
-    {
-      "epoch": 385.18,
-      "learning_rate": 2.417224880382775e-05,
-      "loss": 0.0277,
-      "step": 4237
-    },
-    {
-      "epoch": 385.27,
-      "learning_rate": 2.415311004784689e-05,
-      "loss": 0.0334,
-      "step": 4238
-    },
-    {
-      "epoch": 385.36,
-      "learning_rate": 2.4133971291866033e-05,
-      "loss": 0.0281,
-      "step": 4239
-    },
-    {
-      "epoch": 385.45,
-      "learning_rate": 2.4114832535885168e-05,
-      "loss": 0.0322,
-      "step": 4240
-    },
-    {
-      "epoch": 385.45,
-      "eval_accuracy_NAT": 0.9928899022672557,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.953186027297309,
-      "eval_accuracy_concrete tiles": 0.985874151531829,
-      "eval_accuracy_dark bricks": 0.8831453754666871,
-      "eval_accuracy_grass": 0.905685436377426,
-      "eval_accuracy_red bricks": 0.935546875,
-      "eval_accuracy_road": 0.9385480061959861,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6182008656576516,
-      "eval_accuracy_speedway bricks": 0.987724875004073,
-      "eval_accuracy_steel": 0.9115741046105138,
-      "eval_accuracy_tiles": 0.15456786741542256,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808115029885397,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8926617754380941,
-      "eval_iou_concrete tiles": 0.9520210221736691,
-      "eval_iou_dark bricks": 0.8351836348414807,
-      "eval_iou_grass": 0.8369869606364321,
-      "eval_iou_red bricks": 0.8617926071311743,
-      "eval_iou_road": 0.9314421286073616,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.40717606042998256,
-      "eval_iou_speedway bricks": 0.9774832230395231,
-      "eval_iou_steel": 0.853590371119535,
-      "eval_iou_tiles": 0.15384091064003816,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10452328622341156,
-      "eval_mean_accuracy": 0.7128418066787809,
-      "eval_mean_iou": 0.6679223228496792,
-      "eval_overall_accuracy": 0.9754577212863498,
-      "eval_runtime": 3.3934,
-      "eval_samples_per_second": 10.609,
-      "eval_steps_per_second": 0.589,
-      "step": 4240
-    },
-    {
-      "epoch": 385.55,
-      "learning_rate": 2.4095693779904307e-05,
-      "loss": 0.0321,
-      "step": 4241
-    },
-    {
-      "epoch": 385.64,
-      "learning_rate": 2.407655502392345e-05,
-      "loss": 0.029,
-      "step": 4242
-    },
-    {
-      "epoch": 385.73,
-      "learning_rate": 2.4057416267942584e-05,
-      "loss": 0.0318,
-      "step": 4243
-    },
-    {
-      "epoch": 385.82,
-      "learning_rate": 2.4038277511961723e-05,
-      "loss": 0.0284,
-      "step": 4244
-    },
-    {
-      "epoch": 385.91,
-      "learning_rate": 2.4019138755980862e-05,
-      "loss": 0.0352,
-      "step": 4245
-    },
-    {
-      "epoch": 386.0,
-      "learning_rate": 2.4e-05,
-      "loss": 0.032,
-      "step": 4246
-    },
-    {
-      "epoch": 386.09,
-      "learning_rate": 2.398086124401914e-05,
-      "loss": 0.03,
-      "step": 4247
-    },
-    {
-      "epoch": 386.18,
-      "learning_rate": 2.3961722488038278e-05,
-      "loss": 0.0308,
-      "step": 4248
-    },
-    {
-      "epoch": 386.27,
-      "learning_rate": 2.3942583732057417e-05,
-      "loss": 0.0316,
-      "step": 4249
-    },
-    {
-      "epoch": 386.36,
-      "learning_rate": 2.3923444976076556e-05,
-      "loss": 0.0304,
-      "step": 4250
-    },
-    {
-      "epoch": 386.45,
-      "learning_rate": 2.3904306220095694e-05,
-      "loss": 0.0415,
-      "step": 4251
-    },
-    {
-      "epoch": 386.55,
-      "learning_rate": 2.3885167464114833e-05,
-      "loss": 0.0354,
-      "step": 4252
-    },
-    {
-      "epoch": 386.64,
-      "learning_rate": 2.3866028708133972e-05,
-      "loss": 0.0259,
-      "step": 4253
-    },
-    {
-      "epoch": 386.73,
-      "learning_rate": 2.384688995215311e-05,
-      "loss": 0.0282,
-      "step": 4254
-    },
-    {
-      "epoch": 386.82,
-      "learning_rate": 2.382775119617225e-05,
-      "loss": 0.028,
-      "step": 4255
-    },
-    {
-      "epoch": 386.91,
-      "learning_rate": 2.380861244019139e-05,
-      "loss": 0.0329,
-      "step": 4256
-    },
-    {
-      "epoch": 387.0,
-      "learning_rate": 2.3789473684210527e-05,
-      "loss": 0.0343,
-      "step": 4257
-    },
-    {
-      "epoch": 387.09,
-      "learning_rate": 2.3770334928229666e-05,
-      "loss": 0.0304,
-      "step": 4258
-    },
-    {
-      "epoch": 387.18,
-      "learning_rate": 2.3751196172248805e-05,
-      "loss": 0.0325,
-      "step": 4259
-    },
-    {
-      "epoch": 387.27,
-      "learning_rate": 2.3732057416267943e-05,
-      "loss": 0.0296,
-      "step": 4260
-    },
-    {
-      "epoch": 387.27,
-      "eval_accuracy_NAT": 0.9931090545537445,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9536474204901498,
-      "eval_accuracy_concrete tiles": 0.9852932183697181,
-      "eval_accuracy_dark bricks": 0.8889188252954082,
-      "eval_accuracy_grass": 0.9027181188115352,
-      "eval_accuracy_red bricks": 0.9279119318181818,
-      "eval_accuracy_road": 0.936903482158728,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5808176880875583,
-      "eval_accuracy_speedway bricks": 0.9875426458466837,
-      "eval_accuracy_steel": 0.9089769804102632,
-      "eval_accuracy_tiles": 0.1780578003013286,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806519385783904,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8940315660820537,
-      "eval_iou_concrete tiles": 0.9544189077123564,
-      "eval_iou_dark bricks": 0.8327323862407153,
-      "eval_iou_grass": 0.8369236252875798,
-      "eval_iou_red bricks": 0.864516129032258,
-      "eval_iou_road": 0.9292023299406001,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.39239351101674397,
-      "eval_iou_speedway bricks": 0.9773203916426011,
-      "eval_iou_steel": 0.8535505249140627,
-      "eval_iou_tiles": 0.17736544102599086,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10419435054063797,
-      "eval_mean_accuracy": 0.7110690127802537,
-      "eval_mean_iou": 0.6687005193441039,
-      "eval_overall_accuracy": 0.9752283096313477,
-      "eval_runtime": 3.6101,
-      "eval_samples_per_second": 9.972,
-      "eval_steps_per_second": 0.554,
-      "step": 4260
-    },
-    {
-      "epoch": 387.36,
-      "learning_rate": 2.3712918660287082e-05,
-      "loss": 0.0337,
-      "step": 4261
-    },
-    {
-      "epoch": 387.45,
-      "learning_rate": 2.369377990430622e-05,
-      "loss": 0.0322,
-      "step": 4262
-    },
-    {
-      "epoch": 387.55,
-      "learning_rate": 2.367464114832536e-05,
-      "loss": 0.0261,
-      "step": 4263
-    },
-    {
-      "epoch": 387.64,
-      "learning_rate": 2.36555023923445e-05,
-      "loss": 0.0337,
-      "step": 4264
-    },
-    {
-      "epoch": 387.73,
-      "learning_rate": 2.3636363636363637e-05,
-      "loss": 0.0346,
-      "step": 4265
-    },
-    {
-      "epoch": 387.82,
-      "learning_rate": 2.3617224880382776e-05,
-      "loss": 0.0304,
-      "step": 4266
-    },
-    {
-      "epoch": 387.91,
-      "learning_rate": 2.3598086124401915e-05,
-      "loss": 0.0293,
-      "step": 4267
-    },
-    {
-      "epoch": 388.0,
-      "learning_rate": 2.3578947368421054e-05,
-      "loss": 0.032,
-      "step": 4268
-    },
-    {
-      "epoch": 388.09,
-      "learning_rate": 2.3559808612440192e-05,
-      "loss": 0.0318,
-      "step": 4269
-    },
-    {
-      "epoch": 388.18,
-      "learning_rate": 2.354066985645933e-05,
-      "loss": 0.0344,
-      "step": 4270
-    },
-    {
-      "epoch": 388.27,
-      "learning_rate": 2.352153110047847e-05,
-      "loss": 0.0303,
-      "step": 4271
-    },
-    {
-      "epoch": 388.36,
-      "learning_rate": 2.350239234449761e-05,
-      "loss": 0.0313,
-      "step": 4272
-    },
-    {
-      "epoch": 388.45,
-      "learning_rate": 2.3483253588516748e-05,
-      "loss": 0.0287,
-      "step": 4273
-    },
-    {
-      "epoch": 388.55,
-      "learning_rate": 2.3464114832535886e-05,
-      "loss": 0.0324,
-      "step": 4274
-    },
-    {
-      "epoch": 388.64,
-      "learning_rate": 2.3444976076555025e-05,
-      "loss": 0.0289,
-      "step": 4275
-    },
-    {
-      "epoch": 388.73,
-      "learning_rate": 2.3425837320574164e-05,
-      "loss": 0.0346,
-      "step": 4276
-    },
-    {
-      "epoch": 388.82,
-      "learning_rate": 2.3406698564593303e-05,
-      "loss": 0.0321,
-      "step": 4277
-    },
-    {
-      "epoch": 388.91,
-      "learning_rate": 2.338755980861244e-05,
-      "loss": 0.0283,
-      "step": 4278
-    },
-    {
-      "epoch": 389.0,
-      "learning_rate": 2.336842105263158e-05,
-      "loss": 0.0337,
-      "step": 4279
-    },
-    {
-      "epoch": 389.09,
-      "learning_rate": 2.334928229665072e-05,
-      "loss": 0.0273,
-      "step": 4280
-    },
-    {
-      "epoch": 389.09,
-      "eval_accuracy_NAT": 0.9925975227674876,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9519279260912485,
-      "eval_accuracy_concrete tiles": 0.9848957377851159,
-      "eval_accuracy_dark bricks": 0.8643239290250567,
-      "eval_accuracy_grass": 0.9032493700129667,
-      "eval_accuracy_red bricks": 0.9200994318181818,
-      "eval_accuracy_road": 0.9263797375291268,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6260579494389766,
-      "eval_accuracy_speedway bricks": 0.988782648887859,
-      "eval_accuracy_steel": 0.9076907855682342,
-      "eval_accuracy_tiles": 0.20962881797014107,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980552419430045,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8925677103245668,
-      "eval_iou_concrete tiles": 0.9541751829141858,
-      "eval_iou_dark bricks": 0.8290629845676734,
-      "eval_iou_grass": 0.8315743070616324,
-      "eval_iou_red bricks": 0.8716568544995795,
-      "eval_iou_road": 0.9193615744629785,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.4166177441843399,
-      "eval_iou_speedway bricks": 0.9772316191796131,
-      "eval_iou_steel": 0.8535057524730982,
-      "eval_iou_tiles": 0.20928483522494187,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1096707433462143,
-      "eval_mean_accuracy": 0.7135102966841842,
-      "eval_mean_iou": 0.6719685372555888,
-      "eval_overall_accuracy": 0.9742293887668185,
-      "eval_runtime": 3.4101,
-      "eval_samples_per_second": 10.557,
-      "eval_steps_per_second": 0.586,
-      "step": 4280
-    },
-    {
-      "epoch": 389.18,
-      "learning_rate": 2.3330143540669858e-05,
-      "loss": 0.0314,
-      "step": 4281
-    },
-    {
-      "epoch": 389.27,
-      "learning_rate": 2.3311004784688997e-05,
-      "loss": 0.0299,
-      "step": 4282
-    },
-    {
-      "epoch": 389.36,
-      "learning_rate": 2.3291866028708135e-05,
-      "loss": 0.0273,
-      "step": 4283
-    },
-    {
-      "epoch": 389.45,
-      "learning_rate": 2.3272727272727274e-05,
-      "loss": 0.0316,
-      "step": 4284
-    },
-    {
-      "epoch": 389.55,
-      "learning_rate": 2.3253588516746413e-05,
-      "loss": 0.0296,
-      "step": 4285
-    },
-    {
-      "epoch": 389.64,
-      "learning_rate": 2.323444976076555e-05,
-      "loss": 0.0281,
-      "step": 4286
-    },
-    {
-      "epoch": 389.73,
-      "learning_rate": 2.321531100478469e-05,
-      "loss": 0.0294,
-      "step": 4287
-    },
-    {
-      "epoch": 389.82,
-      "learning_rate": 2.319617224880383e-05,
-      "loss": 0.0306,
-      "step": 4288
-    },
-    {
-      "epoch": 389.91,
-      "learning_rate": 2.3177033492822968e-05,
-      "loss": 0.034,
-      "step": 4289
-    },
-    {
-      "epoch": 390.0,
-      "learning_rate": 2.3157894736842107e-05,
-      "loss": 0.0332,
-      "step": 4290
-    },
-    {
-      "epoch": 390.09,
-      "learning_rate": 2.3138755980861246e-05,
-      "loss": 0.029,
-      "step": 4291
-    },
-    {
-      "epoch": 390.18,
-      "learning_rate": 2.3119617224880384e-05,
-      "loss": 0.0362,
-      "step": 4292
-    },
-    {
-      "epoch": 390.27,
-      "learning_rate": 2.3100478468899523e-05,
-      "loss": 0.0431,
-      "step": 4293
-    },
-    {
-      "epoch": 390.36,
-      "learning_rate": 2.3081339712918662e-05,
-      "loss": 0.0294,
-      "step": 4294
-    },
-    {
-      "epoch": 390.45,
-      "learning_rate": 2.3062200956937797e-05,
-      "loss": 0.0246,
-      "step": 4295
-    },
-    {
-      "epoch": 390.55,
-      "learning_rate": 2.304306220095694e-05,
-      "loss": 0.0301,
-      "step": 4296
-    },
-    {
-      "epoch": 390.64,
-      "learning_rate": 2.3023923444976078e-05,
-      "loss": 0.0278,
-      "step": 4297
-    },
-    {
-      "epoch": 390.73,
-      "learning_rate": 2.3004784688995214e-05,
-      "loss": 0.0329,
-      "step": 4298
-    },
-    {
-      "epoch": 390.82,
-      "learning_rate": 2.2985645933014356e-05,
-      "loss": 0.0322,
-      "step": 4299
-    },
-    {
-      "epoch": 390.91,
-      "learning_rate": 2.2966507177033495e-05,
-      "loss": 0.034,
-      "step": 4300
-    },
-    {
-      "epoch": 390.91,
-      "eval_accuracy_NAT": 0.9929446021133197,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9509903858534701,
-      "eval_accuracy_concrete tiles": 0.9866079618418638,
-      "eval_accuracy_dark bricks": 0.8779877602863632,
-      "eval_accuracy_grass": 0.9133081921032304,
-      "eval_accuracy_red bricks": 0.9437144886363636,
-      "eval_accuracy_road": 0.9271379114786569,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5054448212168831,
-      "eval_accuracy_speedway bricks": 0.988702999057477,
-      "eval_accuracy_steel": 0.9114009629971638,
-      "eval_accuracy_tiles": 0.2157923572113409,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807106636140293,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8937872265327089,
-      "eval_iou_concrete tiles": 0.9505125486037469,
-      "eval_iou_dark bricks": 0.8313349611866322,
-      "eval_iou_grass": 0.8431409875842131,
-      "eval_iou_red bricks": 0.8786576293602248,
-      "eval_iou_road": 0.9201151087424801,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.34252512797518964,
-      "eval_iou_speedway bricks": 0.9769810236247702,
-      "eval_iou_steel": 0.852881722089345,
-      "eval_iou_tiles": 0.215659434672507,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10719604045152664,
-      "eval_mean_accuracy": 0.7087717263689334,
-      "eval_mean_iou": 0.6681774179989113,
-      "eval_overall_accuracy": 0.9743724399142795,
-      "eval_runtime": 3.2454,
-      "eval_samples_per_second": 11.092,
-      "eval_steps_per_second": 0.616,
-      "step": 4300
-    },
-    {
-      "epoch": 391.0,
-      "learning_rate": 2.294736842105263e-05,
-      "loss": 0.0327,
-      "step": 4301
-    },
-    {
-      "epoch": 391.09,
-      "learning_rate": 2.2928229665071772e-05,
-      "loss": 0.0299,
-      "step": 4302
-    },
-    {
-      "epoch": 391.18,
-      "learning_rate": 2.290909090909091e-05,
-      "loss": 0.0335,
-      "step": 4303
-    },
-    {
-      "epoch": 391.27,
-      "learning_rate": 2.288995215311005e-05,
-      "loss": 0.029,
-      "step": 4304
-    },
-    {
-      "epoch": 391.36,
-      "learning_rate": 2.287081339712919e-05,
-      "loss": 0.0309,
-      "step": 4305
-    },
-    {
-      "epoch": 391.45,
-      "learning_rate": 2.2851674641148327e-05,
-      "loss": 0.0312,
-      "step": 4306
-    },
-    {
-      "epoch": 391.55,
-      "learning_rate": 2.2832535885167466e-05,
-      "loss": 0.0285,
-      "step": 4307
-    },
-    {
-      "epoch": 391.64,
-      "learning_rate": 2.2813397129186605e-05,
-      "loss": 0.0324,
-      "step": 4308
-    },
-    {
-      "epoch": 391.73,
-      "learning_rate": 2.2794258373205744e-05,
-      "loss": 0.0314,
-      "step": 4309
-    },
-    {
-      "epoch": 391.82,
-      "learning_rate": 2.2775119617224882e-05,
-      "loss": 0.0325,
-      "step": 4310
-    },
-    {
-      "epoch": 391.91,
-      "learning_rate": 2.275598086124402e-05,
-      "loss": 0.0327,
-      "step": 4311
-    },
-    {
-      "epoch": 392.0,
-      "learning_rate": 2.273684210526316e-05,
-      "loss": 0.0301,
-      "step": 4312
-    },
-    {
-      "epoch": 392.09,
-      "learning_rate": 2.27177033492823e-05,
-      "loss": 0.0302,
-      "step": 4313
-    },
-    {
-      "epoch": 392.18,
-      "learning_rate": 2.2698564593301434e-05,
-      "loss": 0.0309,
-      "step": 4314
-    },
-    {
-      "epoch": 392.27,
-      "learning_rate": 2.2679425837320576e-05,
-      "loss": 0.0291,
-      "step": 4315
-    },
-    {
-      "epoch": 392.36,
-      "learning_rate": 2.2660287081339715e-05,
-      "loss": 0.0339,
-      "step": 4316
-    },
-    {
-      "epoch": 392.45,
-      "learning_rate": 2.264114832535885e-05,
-      "loss": 0.0272,
-      "step": 4317
-    },
-    {
-      "epoch": 392.55,
-      "learning_rate": 2.2622009569377993e-05,
-      "loss": 0.0331,
-      "step": 4318
-    },
-    {
-      "epoch": 392.64,
-      "learning_rate": 2.260287081339713e-05,
-      "loss": 0.0331,
-      "step": 4319
-    },
-    {
-      "epoch": 392.73,
-      "learning_rate": 2.2583732057416267e-05,
-      "loss": 0.0312,
-      "step": 4320
-    },
-    {
-      "epoch": 392.73,
-      "eval_accuracy_NAT": 0.9932267474483404,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9523571290613331,
-      "eval_accuracy_concrete tiles": 0.982021647404146,
-      "eval_accuracy_dark bricks": 0.8803741195489011,
-      "eval_accuracy_grass": 0.9076601530143263,
-      "eval_accuracy_red bricks": 0.9302201704545454,
-      "eval_accuracy_road": 0.9248065568140733,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.502577674854575,
-      "eval_accuracy_speedway bricks": 0.9881092457764471,
-      "eval_accuracy_steel": 0.908094782666051,
-      "eval_accuracy_tiles": 0.22818791946308725,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805621089074468,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8942589854484727,
-      "eval_iou_concrete tiles": 0.9563767382306524,
-      "eval_iou_dark bricks": 0.8289721658451725,
-      "eval_iou_grass": 0.8420020231626182,
-      "eval_iou_red bricks": 0.8662367724867724,
-      "eval_iou_road": 0.9177992106093716,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33874054666740994,
-      "eval_iou_speedway bricks": 0.9773506010455405,
-      "eval_iou_steel": 0.8530127013630731,
-      "eval_iou_tiles": 0.22789138909787293,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10883994400501251,
-      "eval_mean_accuracy": 0.7075104728081404,
-      "eval_mean_iou": 0.6679387109895694,
-      "eval_overall_accuracy": 0.9741183386908637,
-      "eval_runtime": 3.4105,
-      "eval_samples_per_second": 10.556,
-      "eval_steps_per_second": 0.586,
-      "step": 4320
-    },
-    {
-      "epoch": 392.82,
-      "learning_rate": 2.256459330143541e-05,
-      "loss": 0.0302,
-      "step": 4321
-    },
-    {
-      "epoch": 392.91,
-      "learning_rate": 2.2545454545454548e-05,
-      "loss": 0.0301,
-      "step": 4322
-    },
-    {
-      "epoch": 393.0,
-      "learning_rate": 2.2526315789473683e-05,
-      "loss": 0.03,
-      "step": 4323
-    },
-    {
-      "epoch": 393.09,
-      "learning_rate": 2.2507177033492825e-05,
-      "loss": 0.0301,
-      "step": 4324
-    },
-    {
-      "epoch": 393.18,
-      "learning_rate": 2.2488038277511964e-05,
-      "loss": 0.0296,
-      "step": 4325
-    },
-    {
-      "epoch": 393.27,
-      "learning_rate": 2.24688995215311e-05,
-      "loss": 0.0318,
-      "step": 4326
-    },
-    {
-      "epoch": 393.36,
-      "learning_rate": 2.244976076555024e-05,
-      "loss": 0.0282,
-      "step": 4327
-    },
-    {
-      "epoch": 393.45,
-      "learning_rate": 2.243062200956938e-05,
-      "loss": 0.0308,
-      "step": 4328
-    },
-    {
-      "epoch": 393.55,
-      "learning_rate": 2.2411483253588516e-05,
-      "loss": 0.028,
-      "step": 4329
-    },
-    {
-      "epoch": 393.64,
-      "learning_rate": 2.2392344497607658e-05,
-      "loss": 0.0291,
-      "step": 4330
-    },
-    {
-      "epoch": 393.73,
-      "learning_rate": 2.2373205741626797e-05,
-      "loss": 0.031,
-      "step": 4331
-    },
-    {
-      "epoch": 393.82,
-      "learning_rate": 2.2354066985645932e-05,
-      "loss": 0.0267,
-      "step": 4332
-    },
-    {
-      "epoch": 393.91,
-      "learning_rate": 2.233492822966507e-05,
-      "loss": 0.0296,
-      "step": 4333
-    },
-    {
-      "epoch": 394.0,
-      "learning_rate": 2.2315789473684213e-05,
-      "loss": 0.034,
-      "step": 4334
-    },
-    {
-      "epoch": 394.09,
-      "learning_rate": 2.229665071770335e-05,
-      "loss": 0.0274,
-      "step": 4335
-    },
-    {
-      "epoch": 394.18,
-      "learning_rate": 2.2277511961722487e-05,
-      "loss": 0.031,
-      "step": 4336
-    },
-    {
-      "epoch": 394.27,
-      "learning_rate": 2.225837320574163e-05,
-      "loss": 0.0293,
-      "step": 4337
-    },
-    {
-      "epoch": 394.36,
-      "learning_rate": 2.2239234449760768e-05,
-      "loss": 0.0308,
-      "step": 4338
-    },
-    {
-      "epoch": 394.45,
-      "learning_rate": 2.2220095693779903e-05,
-      "loss": 0.0316,
-      "step": 4339
-    },
-    {
-      "epoch": 394.55,
-      "learning_rate": 2.2200956937799046e-05,
-      "loss": 0.03,
-      "step": 4340
-    },
-    {
-      "epoch": 394.55,
-      "eval_accuracy_NAT": 0.9928500243149638,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9513216768960041,
-      "eval_accuracy_concrete tiles": 0.9866079618418638,
-      "eval_accuracy_dark bricks": 0.8864169970362957,
-      "eval_accuracy_grass": 0.9113928917191219,
-      "eval_accuracy_red bricks": 0.9247159090909091,
-      "eval_accuracy_road": 0.9199455372333544,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48443746036997215,
-      "eval_accuracy_speedway bricks": 0.9890463380990482,
-      "eval_accuracy_steel": 0.9055388826594551,
-      "eval_accuracy_tiles": 0.21099849335707438,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805538516231118,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8915944805427441,
-      "eval_iou_concrete tiles": 0.9525890063175296,
-      "eval_iou_dark bricks": 0.8311378974340466,
-      "eval_iou_grass": 0.8418155698674802,
-      "eval_iou_red bricks": 0.8611111111111112,
-      "eval_iou_road": 0.9134354127420363,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32620479691096754,
-      "eval_iou_speedway bricks": 0.9772480387679444,
-      "eval_iou_steel": 0.8538454960312832,
-      "eval_iou_tiles": 0.21072430066342931,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11214879900217056,
-      "eval_mean_accuracy": 0.7048670902013895,
-      "eval_mean_iou": 0.6646353816932064,
-      "eval_overall_accuracy": 0.973562134636773,
-      "eval_runtime": 4.1411,
-      "eval_samples_per_second": 8.693,
-      "eval_steps_per_second": 0.483,
-      "step": 4340
-    },
-    {
-      "epoch": 394.64,
-      "learning_rate": 2.2181818181818184e-05,
-      "loss": 0.0315,
-      "step": 4341
-    },
-    {
-      "epoch": 394.73,
-      "learning_rate": 2.216267942583732e-05,
-      "loss": 0.032,
-      "step": 4342
-    },
-    {
-      "epoch": 394.82,
-      "learning_rate": 2.2143540669856462e-05,
-      "loss": 0.0317,
-      "step": 4343
-    },
-    {
-      "epoch": 394.91,
-      "learning_rate": 2.21244019138756e-05,
-      "loss": 0.032,
-      "step": 4344
-    },
-    {
-      "epoch": 395.0,
-      "learning_rate": 2.2105263157894736e-05,
-      "loss": 0.0306,
-      "step": 4345
-    },
-    {
-      "epoch": 395.09,
-      "learning_rate": 2.208612440191388e-05,
-      "loss": 0.0303,
-      "step": 4346
-    },
-    {
-      "epoch": 395.18,
-      "learning_rate": 2.2066985645933017e-05,
-      "loss": 0.0286,
-      "step": 4347
-    },
-    {
-      "epoch": 395.27,
-      "learning_rate": 2.2047846889952152e-05,
-      "loss": 0.0286,
-      "step": 4348
-    },
-    {
-      "epoch": 395.36,
-      "learning_rate": 2.2028708133971295e-05,
-      "loss": 0.0311,
-      "step": 4349
-    },
-    {
-      "epoch": 395.45,
-      "learning_rate": 2.2009569377990433e-05,
-      "loss": 0.0298,
-      "step": 4350
-    },
-    {
-      "epoch": 395.55,
-      "learning_rate": 2.199043062200957e-05,
-      "loss": 0.0394,
-      "step": 4351
-    },
-    {
-      "epoch": 395.64,
-      "learning_rate": 2.1971291866028708e-05,
-      "loss": 0.0376,
-      "step": 4352
-    },
-    {
-      "epoch": 395.73,
-      "learning_rate": 2.195215311004785e-05,
-      "loss": 0.0272,
-      "step": 4353
-    },
-    {
-      "epoch": 395.82,
-      "learning_rate": 2.1933014354066985e-05,
-      "loss": 0.0595,
-      "step": 4354
-    },
-    {
-      "epoch": 395.91,
-      "learning_rate": 2.1913875598086124e-05,
-      "loss": 0.0276,
-      "step": 4355
-    },
-    {
-      "epoch": 396.0,
-      "learning_rate": 2.1894736842105266e-05,
-      "loss": 0.03,
-      "step": 4356
-    },
-    {
-      "epoch": 396.09,
-      "learning_rate": 2.18755980861244e-05,
-      "loss": 0.0328,
-      "step": 4357
-    },
-    {
-      "epoch": 396.18,
-      "learning_rate": 2.185645933014354e-05,
-      "loss": 0.0309,
-      "step": 4358
-    },
-    {
-      "epoch": 396.27,
-      "learning_rate": 2.1837320574162682e-05,
-      "loss": 0.0283,
-      "step": 4359
-    },
-    {
-      "epoch": 396.36,
-      "learning_rate": 2.1818181818181818e-05,
-      "loss": 0.0329,
-      "step": 4360
-    },
-    {
-      "epoch": 396.36,
-      "eval_accuracy_NAT": 0.9930735878793611,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9494385488647581,
-      "eval_accuracy_concrete tiles": 0.9844065309117593,
-      "eval_accuracy_dark bricks": 0.8832223547977368,
-      "eval_accuracy_grass": 0.9077824805936033,
-      "eval_accuracy_red bricks": 0.9398082386363636,
-      "eval_accuracy_road": 0.9293725294351665,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5040388167507512,
-      "eval_accuracy_speedway bricks": 0.9884646529741368,
-      "eval_accuracy_steel": 0.9066107116944793,
-      "eval_accuracy_tiles": 0.15100671140939598,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805005745680635,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8906367955389586,
-      "eval_iou_concrete tiles": 0.9507722292767917,
-      "eval_iou_dark bricks": 0.8313528005217014,
-      "eval_iou_grass": 0.8398608272757845,
-      "eval_iou_red bricks": 0.8647279856232641,
-      "eval_iou_road": 0.9215409794787863,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33762996066554635,
-      "eval_iou_speedway bricks": 0.9769756009120062,
-      "eval_iou_steel": 0.8549890755845145,
-      "eval_iou_tiles": 0.150831110199056,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11021122336387634,
-      "eval_mean_accuracy": 0.7028634741498087,
-      "eval_mean_iou": 0.6615244568957286,
-      "eval_overall_accuracy": 0.9741509755452474,
-      "eval_runtime": 3.4044,
-      "eval_samples_per_second": 10.575,
-      "eval_steps_per_second": 0.587,
-      "step": 4360
-    },
-    {
-      "epoch": 396.45,
-      "learning_rate": 2.1799043062200957e-05,
-      "loss": 0.0284,
-      "step": 4361
-    },
-    {
-      "epoch": 396.55,
-      "learning_rate": 2.17799043062201e-05,
-      "loss": 0.031,
-      "step": 4362
-    },
-    {
-      "epoch": 396.64,
-      "learning_rate": 2.1760765550239234e-05,
-      "loss": 0.0356,
-      "step": 4363
-    },
-    {
-      "epoch": 396.73,
-      "learning_rate": 2.1741626794258373e-05,
-      "loss": 0.0318,
-      "step": 4364
-    },
-    {
-      "epoch": 396.82,
-      "learning_rate": 2.1722488038277515e-05,
-      "loss": 0.0314,
-      "step": 4365
-    },
-    {
-      "epoch": 396.91,
-      "learning_rate": 2.170334928229665e-05,
-      "loss": 0.0391,
-      "step": 4366
-    },
-    {
-      "epoch": 397.0,
-      "learning_rate": 2.168421052631579e-05,
-      "loss": 0.0302,
-      "step": 4367
-    },
-    {
-      "epoch": 397.09,
-      "learning_rate": 2.166507177033493e-05,
-      "loss": 0.0371,
-      "step": 4368
-    },
-    {
-      "epoch": 397.18,
-      "learning_rate": 2.1645933014354067e-05,
-      "loss": 0.0283,
-      "step": 4369
-    },
-    {
-      "epoch": 397.27,
-      "learning_rate": 2.1626794258373206e-05,
-      "loss": 0.029,
-      "step": 4370
-    },
-    {
-      "epoch": 397.36,
-      "learning_rate": 2.1607655502392344e-05,
-      "loss": 0.0297,
-      "step": 4371
-    },
-    {
-      "epoch": 397.45,
-      "learning_rate": 2.1588516746411483e-05,
-      "loss": 0.0313,
-      "step": 4372
-    },
-    {
-      "epoch": 397.55,
-      "learning_rate": 2.1569377990430622e-05,
-      "loss": 0.0317,
-      "step": 4373
-    },
-    {
-      "epoch": 397.64,
-      "learning_rate": 2.155023923444976e-05,
-      "loss": 0.0283,
-      "step": 4374
-    },
-    {
-      "epoch": 397.73,
-      "learning_rate": 2.1531100478468903e-05,
-      "loss": 0.0284,
-      "step": 4375
-    },
-    {
-      "epoch": 397.82,
-      "learning_rate": 2.1511961722488038e-05,
-      "loss": 0.0293,
-      "step": 4376
-    },
-    {
-      "epoch": 397.91,
-      "learning_rate": 2.1492822966507177e-05,
-      "loss": 0.0324,
-      "step": 4377
-    },
-    {
-      "epoch": 398.0,
-      "learning_rate": 2.147368421052632e-05,
-      "loss": 0.0395,
-      "step": 4378
-    },
-    {
-      "epoch": 398.09,
-      "learning_rate": 2.1454545454545455e-05,
-      "loss": 0.0326,
-      "step": 4379
-    },
-    {
-      "epoch": 398.18,
-      "learning_rate": 2.1435406698564593e-05,
-      "loss": 0.0279,
-      "step": 4380
-    },
-    {
-      "epoch": 398.18,
-      "eval_accuracy_NAT": 0.9927062166551502,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9529875209236448,
-      "eval_accuracy_concrete tiles": 0.981593591389959,
-      "eval_accuracy_dark bricks": 0.8717524344713444,
-      "eval_accuracy_grass": 0.9106938769803962,
-      "eval_accuracy_red bricks": 0.9536576704545454,
-      "eval_accuracy_road": 0.9275055109693381,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.604388939431533,
-      "eval_accuracy_speedway bricks": 0.9882504432030335,
-      "eval_accuracy_steel": 0.9078639271815844,
-      "eval_accuracy_tiles": 0.4518559101492946,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804878601373496,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8991769026532993,
-      "eval_iou_concrete tiles": 0.9543400713436385,
-      "eval_iou_dark bricks": 0.8284198975859547,
-      "eval_iou_grass": 0.8404617677227071,
-      "eval_iou_red bricks": 0.8582614253755193,
-      "eval_iou_road": 0.9208390658051752,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.40458043442154024,
-      "eval_iou_speedway bricks": 0.9772598711611488,
-      "eval_iou_steel": 0.8528883243226496,
-      "eval_iou_tiles": 0.4495162828723259,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10495162755250931,
-      "eval_mean_accuracy": 0.7340966186007557,
-      "eval_mean_iou": 0.6897101464154852,
-      "eval_overall_accuracy": 0.9749320348103842,
-      "eval_runtime": 3.695,
-      "eval_samples_per_second": 9.743,
-      "eval_steps_per_second": 0.541,
-      "step": 4380
-    },
-    {
-      "epoch": 398.27,
-      "learning_rate": 2.1416267942583736e-05,
-      "loss": 0.0338,
-      "step": 4381
-    },
-    {
-      "epoch": 398.36,
-      "learning_rate": 2.139712918660287e-05,
-      "loss": 0.0335,
-      "step": 4382
-    },
-    {
-      "epoch": 398.45,
-      "learning_rate": 2.137799043062201e-05,
-      "loss": 0.0276,
-      "step": 4383
-    },
-    {
-      "epoch": 398.55,
-      "learning_rate": 2.1358851674641152e-05,
-      "loss": 0.0292,
-      "step": 4384
-    },
-    {
-      "epoch": 398.64,
-      "learning_rate": 2.1339712918660287e-05,
-      "loss": 0.0303,
-      "step": 4385
-    },
-    {
-      "epoch": 398.73,
-      "learning_rate": 2.1320574162679426e-05,
-      "loss": 0.033,
-      "step": 4386
-    },
-    {
-      "epoch": 398.82,
-      "learning_rate": 2.1301435406698568e-05,
-      "loss": 0.0305,
-      "step": 4387
-    },
-    {
-      "epoch": 398.91,
-      "learning_rate": 2.1282296650717704e-05,
-      "loss": 0.0291,
-      "step": 4388
-    },
-    {
-      "epoch": 399.0,
-      "learning_rate": 2.1263157894736842e-05,
-      "loss": 0.033,
-      "step": 4389
-    },
-    {
-      "epoch": 399.09,
-      "learning_rate": 2.1244019138755985e-05,
-      "loss": 0.0267,
-      "step": 4390
-    },
-    {
-      "epoch": 399.18,
-      "learning_rate": 2.122488038277512e-05,
-      "loss": 0.0291,
-      "step": 4391
-    },
-    {
-      "epoch": 399.27,
-      "learning_rate": 2.120574162679426e-05,
-      "loss": 0.0315,
-      "step": 4392
-    },
-    {
-      "epoch": 399.36,
-      "learning_rate": 2.1186602870813397e-05,
-      "loss": 0.032,
-      "step": 4393
-    },
-    {
-      "epoch": 399.45,
-      "learning_rate": 2.1167464114832536e-05,
-      "loss": 0.0295,
-      "step": 4394
-    },
-    {
-      "epoch": 399.55,
-      "learning_rate": 2.1148325358851675e-05,
-      "loss": 0.0329,
-      "step": 4395
-    },
-    {
-      "epoch": 399.64,
-      "learning_rate": 2.1129186602870814e-05,
-      "loss": 0.0299,
-      "step": 4396
-    },
-    {
-      "epoch": 399.73,
-      "learning_rate": 2.1110047846889953e-05,
-      "loss": 0.0326,
-      "step": 4397
-    },
-    {
-      "epoch": 399.82,
-      "learning_rate": 2.109090909090909e-05,
-      "loss": 0.0336,
-      "step": 4398
-    },
-    {
-      "epoch": 399.91,
-      "learning_rate": 2.107177033492823e-05,
-      "loss": 0.0334,
-      "step": 4399
-    },
-    {
-      "epoch": 400.0,
-      "learning_rate": 2.105263157894737e-05,
-      "loss": 0.0459,
-      "step": 4400
-    },
-    {
-      "epoch": 400.0,
-      "eval_accuracy_NAT": 0.9933305007047457,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.95159261127087,
-      "eval_accuracy_concrete tiles": 0.9836727206017245,
-      "eval_accuracy_dark bricks": 0.8784111466071359,
-      "eval_accuracy_grass": 0.9024664735055938,
-      "eval_accuracy_red bricks": 0.9353693181818182,
-      "eval_accuracy_road": 0.9367910257355919,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6358724119868773,
-      "eval_accuracy_speedway bricks": 0.9884918063254033,
-      "eval_accuracy_steel": 0.9037909768484929,
-      "eval_accuracy_tiles": 0.3663881660046569,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808779559662952,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.900980379706648,
-      "eval_iou_concrete tiles": 0.9567027477102414,
-      "eval_iou_dark bricks": 0.8329805095262428,
-      "eval_iou_grass": 0.8415030389936287,
-      "eval_iou_red bricks": 0.8583998696431481,
-      "eval_iou_road": 0.9278323105445452,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.42564774488816715,
-      "eval_iou_speedway bricks": 0.9772932818849762,
-      "eval_iou_steel": 0.8513766455671624,
-      "eval_iou_tiles": 0.36573694284933006,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1011911928653717,
-      "eval_mean_accuracy": 0.7289367044440701,
-      "eval_mean_iou": 0.6861024174831065,
-      "eval_overall_accuracy": 0.9757544199625651,
-      "eval_runtime": 3.4989,
-      "eval_samples_per_second": 10.289,
-      "eval_steps_per_second": 0.572,
-      "step": 4400
-    },
-    {
-      "epoch": 400.09,
-      "learning_rate": 2.1033492822966508e-05,
-      "loss": 0.0339,
-      "step": 4401
-    },
-    {
-      "epoch": 400.18,
-      "learning_rate": 2.1014354066985646e-05,
-      "loss": 0.0276,
-      "step": 4402
-    },
-    {
-      "epoch": 400.27,
-      "learning_rate": 2.0995215311004785e-05,
-      "loss": 0.0306,
-      "step": 4403
-    },
-    {
-      "epoch": 400.36,
-      "learning_rate": 2.0976076555023924e-05,
-      "loss": 0.0269,
-      "step": 4404
-    },
-    {
-      "epoch": 400.45,
-      "learning_rate": 2.0956937799043063e-05,
-      "loss": 0.0322,
-      "step": 4405
-    },
-    {
-      "epoch": 400.55,
-      "learning_rate": 2.09377990430622e-05,
-      "loss": 0.0279,
-      "step": 4406
-    },
-    {
-      "epoch": 400.64,
-      "learning_rate": 2.091866028708134e-05,
-      "loss": 0.0352,
-      "step": 4407
-    },
-    {
-      "epoch": 400.73,
-      "learning_rate": 2.089952153110048e-05,
-      "loss": 0.041,
-      "step": 4408
-    },
-    {
-      "epoch": 400.82,
-      "learning_rate": 2.088038277511962e-05,
-      "loss": 0.034,
-      "step": 4409
-    },
-    {
-      "epoch": 400.91,
-      "learning_rate": 2.0861244019138757e-05,
-      "loss": 0.0315,
-      "step": 4410
-    },
-    {
-      "epoch": 401.0,
-      "learning_rate": 2.0842105263157895e-05,
-      "loss": 0.0291,
-      "step": 4411
-    },
-    {
-      "epoch": 401.09,
-      "learning_rate": 2.0822966507177034e-05,
-      "loss": 0.0322,
-      "step": 4412
-    },
-    {
-      "epoch": 401.18,
-      "learning_rate": 2.0803827751196173e-05,
-      "loss": 0.0311,
-      "step": 4413
-    },
-    {
-      "epoch": 401.27,
-      "learning_rate": 2.0784688995215312e-05,
-      "loss": 0.0315,
-      "step": 4414
-    },
-    {
-      "epoch": 401.36,
-      "learning_rate": 2.076555023923445e-05,
-      "loss": 0.0321,
-      "step": 4415
-    },
-    {
-      "epoch": 401.45,
-      "learning_rate": 2.074641148325359e-05,
-      "loss": 0.0309,
-      "step": 4416
-    },
-    {
-      "epoch": 401.55,
-      "learning_rate": 2.0727272727272728e-05,
-      "loss": 0.0307,
-      "step": 4417
-    },
-    {
-      "epoch": 401.64,
-      "learning_rate": 2.0708133971291867e-05,
-      "loss": 0.0325,
-      "step": 4418
-    },
-    {
-      "epoch": 401.73,
-      "learning_rate": 2.0688995215311006e-05,
-      "loss": 0.033,
-      "step": 4419
-    },
-    {
-      "epoch": 401.82,
-      "learning_rate": 2.0669856459330144e-05,
-      "loss": 0.0265,
-      "step": 4420
-    },
-    {
-      "epoch": 401.82,
-      "eval_accuracy_NAT": 0.9929994784105001,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9491407893042619,
-      "eval_accuracy_concrete tiles": 0.985874151531829,
-      "eval_accuracy_dark bricks": 0.8944998267965052,
-      "eval_accuracy_grass": 0.9083242170161158,
-      "eval_accuracy_red bricks": 0.9311079545454546,
-      "eval_accuracy_road": 0.9291500350065962,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.609516720425661,
-      "eval_accuracy_speedway bricks": 0.9896575902064499,
-      "eval_accuracy_steel": 0.9052585581426027,
-      "eval_accuracy_tiles": 0.3307766059443912,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806295099952411,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8995500019067716,
-      "eval_iou_concrete tiles": 0.9522459466643041,
-      "eval_iou_dark bricks": 0.8343804976124655,
-      "eval_iou_grass": 0.8407562331977834,
-      "eval_iou_red bricks": 0.8617912900575185,
-      "eval_iou_road": 0.9222625104571375,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.4087295718405679,
-      "eval_iou_speedway bricks": 0.9770970082654278,
-      "eval_iou_steel": 0.8505856651482755,
-      "eval_iou_tiles": 0.3296928327645051,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10490593314170837,
-      "eval_mean_accuracy": 0.7251004559484898,
-      "eval_mean_iou": 0.6813631590699999,
-      "eval_overall_accuracy": 0.9749865002102323,
-      "eval_runtime": 4.0981,
-      "eval_samples_per_second": 8.785,
-      "eval_steps_per_second": 0.488,
-      "step": 4420
-    },
-    {
-      "epoch": 401.91,
-      "learning_rate": 2.0650717703349283e-05,
-      "loss": 0.0275,
-      "step": 4421
-    },
-    {
-      "epoch": 402.0,
-      "learning_rate": 2.0631578947368422e-05,
-      "loss": 0.028,
-      "step": 4422
-    },
-    {
-      "epoch": 402.09,
-      "learning_rate": 2.061244019138756e-05,
-      "loss": 0.0284,
-      "step": 4423
-    },
-    {
-      "epoch": 402.18,
-      "learning_rate": 2.05933014354067e-05,
-      "loss": 0.0317,
-      "step": 4424
-    },
-    {
-      "epoch": 402.27,
-      "learning_rate": 2.057416267942584e-05,
-      "loss": 0.0296,
-      "step": 4425
-    },
-    {
-      "epoch": 402.36,
-      "learning_rate": 2.0555023923444977e-05,
-      "loss": 0.0284,
-      "step": 4426
-    },
-    {
-      "epoch": 402.45,
-      "learning_rate": 2.0535885167464116e-05,
-      "loss": 0.0289,
-      "step": 4427
-    },
-    {
-      "epoch": 402.55,
-      "learning_rate": 2.0516746411483255e-05,
-      "loss": 0.033,
-      "step": 4428
-    },
-    {
-      "epoch": 402.64,
-      "learning_rate": 2.0497607655502393e-05,
-      "loss": 0.0291,
-      "step": 4429
-    },
-    {
-      "epoch": 402.73,
-      "learning_rate": 2.0478468899521532e-05,
-      "loss": 0.0279,
-      "step": 4430
-    },
-    {
-      "epoch": 402.82,
-      "learning_rate": 2.045933014354067e-05,
-      "loss": 0.0352,
-      "step": 4431
-    },
-    {
-      "epoch": 402.91,
-      "learning_rate": 2.044019138755981e-05,
-      "loss": 0.0323,
-      "step": 4432
-    },
-    {
-      "epoch": 403.0,
-      "learning_rate": 2.042105263157895e-05,
-      "loss": 0.0358,
-      "step": 4433
-    },
-    {
-      "epoch": 403.09,
-      "learning_rate": 2.0401913875598087e-05,
-      "loss": 0.0316,
-      "step": 4434
-    },
-    {
-      "epoch": 403.18,
-      "learning_rate": 2.0382775119617226e-05,
-      "loss": 0.0253,
-      "step": 4435
-    },
-    {
-      "epoch": 403.27,
-      "learning_rate": 2.0363636363636365e-05,
-      "loss": 0.0303,
-      "step": 4436
-    },
-    {
-      "epoch": 403.36,
-      "learning_rate": 2.0344497607655504e-05,
-      "loss": 0.0301,
-      "step": 4437
-    },
-    {
-      "epoch": 403.45,
-      "learning_rate": 2.0325358851674642e-05,
-      "loss": 0.0322,
-      "step": 4438
-    },
-    {
-      "epoch": 403.55,
-      "learning_rate": 2.030622009569378e-05,
-      "loss": 0.0289,
-      "step": 4439
-    },
-    {
-      "epoch": 403.64,
-      "learning_rate": 2.028708133971292e-05,
-      "loss": 0.0257,
-      "step": 4440
-    },
-    {
-      "epoch": 403.64,
-      "eval_accuracy_NAT": 0.9927215679022715,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9549967273273531,
-      "eval_accuracy_concrete tiles": 0.9817464685378829,
-      "eval_accuracy_dark bricks": 0.8828374581424887,
-      "eval_accuracy_grass": 0.9028089907275695,
-      "eval_accuracy_red bricks": 0.9163707386363636,
-      "eval_accuracy_road": 0.9303084570857825,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5079535742839026,
-      "eval_accuracy_speedway bricks": 0.9882257034829905,
-      "eval_accuracy_steel": 0.9065859771782864,
-      "eval_accuracy_tiles": 0.22627037392138064,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806990626080748,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8918704961175641,
-      "eval_iou_concrete tiles": 0.9588783372155528,
-      "eval_iou_dark bricks": 0.837208453480308,
-      "eval_iou_grass": 0.8368587303387167,
-      "eval_iou_red bricks": 0.8560291922375186,
-      "eval_iou_road": 0.9223086697995588,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3427205595134019,
-      "eval_iou_speedway bricks": 0.9773071348139272,
-      "eval_iou_steel": 0.8546467794713156,
-      "eval_iou_tiles": 0.22586819797648347,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10712853819131851,
-      "eval_mean_accuracy": 0.7069866182481749,
-      "eval_mean_iou": 0.6680304318132634,
-      "eval_overall_accuracy": 0.974374877081977,
-      "eval_runtime": 3.6459,
-      "eval_samples_per_second": 9.874,
-      "eval_steps_per_second": 0.549,
-      "step": 4440
-    },
-    {
-      "epoch": 403.73,
-      "learning_rate": 2.026794258373206e-05,
-      "loss": 0.034,
-      "step": 4441
-    },
-    {
-      "epoch": 403.82,
-      "learning_rate": 2.0248803827751198e-05,
-      "loss": 0.0394,
-      "step": 4442
-    },
-    {
-      "epoch": 403.91,
-      "learning_rate": 2.0229665071770333e-05,
-      "loss": 0.0319,
-      "step": 4443
-    },
-    {
-      "epoch": 404.0,
-      "learning_rate": 2.0210526315789475e-05,
-      "loss": 0.0315,
-      "step": 4444
-    },
-    {
-      "epoch": 404.09,
-      "learning_rate": 2.0191387559808614e-05,
-      "loss": 0.0308,
-      "step": 4445
-    },
-    {
-      "epoch": 404.18,
-      "learning_rate": 2.0172248803827753e-05,
-      "loss": 0.0286,
-      "step": 4446
-    },
-    {
-      "epoch": 404.27,
-      "learning_rate": 2.015311004784689e-05,
-      "loss": 0.0268,
-      "step": 4447
-    },
-    {
-      "epoch": 404.36,
-      "learning_rate": 2.013397129186603e-05,
-      "loss": 0.0307,
-      "step": 4448
-    },
-    {
-      "epoch": 404.45,
-      "learning_rate": 2.011483253588517e-05,
-      "loss": 0.0297,
-      "step": 4449
-    },
-    {
-      "epoch": 404.55,
-      "learning_rate": 2.0095693779904308e-05,
-      "loss": 0.0274,
-      "step": 4450
-    },
-    {
-      "epoch": 404.64,
-      "learning_rate": 2.0076555023923447e-05,
-      "loss": 0.0318,
-      "step": 4451
-    },
-    {
-      "epoch": 404.73,
-      "learning_rate": 2.0057416267942585e-05,
-      "loss": 0.0329,
-      "step": 4452
-    },
-    {
-      "epoch": 404.82,
-      "learning_rate": 2.0038277511961724e-05,
-      "loss": 0.0299,
-      "step": 4453
-    },
-    {
-      "epoch": 404.91,
-      "learning_rate": 2.0019138755980863e-05,
-      "loss": 0.0332,
-      "step": 4454
-    },
-    {
-      "epoch": 405.0,
-      "learning_rate": 2e-05,
-      "loss": 0.0278,
-      "step": 4455
-    },
-    {
-      "epoch": 405.09,
-      "learning_rate": 1.998086124401914e-05,
-      "loss": 0.0292,
-      "step": 4456
-    },
-    {
-      "epoch": 405.18,
-      "learning_rate": 1.996172248803828e-05,
-      "loss": 0.0313,
-      "step": 4457
-    },
-    {
-      "epoch": 405.27,
-      "learning_rate": 1.9942583732057418e-05,
-      "loss": 0.0288,
-      "step": 4458
-    },
-    {
-      "epoch": 405.36,
-      "learning_rate": 1.9923444976076557e-05,
-      "loss": 0.032,
-      "step": 4459
-    },
-    {
-      "epoch": 405.45,
-      "learning_rate": 1.9904306220095696e-05,
-      "loss": 0.0316,
-      "step": 4460
-    },
-    {
-      "epoch": 405.45,
-      "eval_accuracy_NAT": 0.9927766206505682,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9525140563972703,
-      "eval_accuracy_concrete tiles": 0.9866691127010334,
-      "eval_accuracy_dark bricks": 0.8779492706208383,
-      "eval_accuracy_grass": 0.903784116288092,
-      "eval_accuracy_red bricks": 0.9300426136363636,
-      "eval_accuracy_road": 0.9284341833668486,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.517299368676426,
-      "eval_accuracy_speedway bricks": 0.9886191253724534,
-      "eval_accuracy_steel": 0.9098014642833586,
-      "eval_accuracy_tiles": 0.24215860840980688,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807117363533491,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8913495052897534,
-      "eval_iou_concrete tiles": 0.9532108465764755,
-      "eval_iou_dark bricks": 0.8342476775656499,
-      "eval_iou_grass": 0.8372418303616237,
-      "eval_iou_red bricks": 0.8596750369276218,
-      "eval_iou_road": 0.9208611781412062,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.34810032650638173,
-      "eval_iou_speedway bricks": 0.9771556680519307,
-      "eval_iou_steel": 0.8531556119096033,
-      "eval_iou_tiles": 0.24154655372634742,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10701620578765869,
-      "eval_mean_accuracy": 0.7100037338771584,
-      "eval_mean_iou": 0.6690196901084572,
-      "eval_overall_accuracy": 0.9742598003811307,
-      "eval_runtime": 3.426,
-      "eval_samples_per_second": 10.508,
-      "eval_steps_per_second": 0.584,
-      "step": 4460
-    },
-    {
-      "epoch": 405.55,
-      "learning_rate": 1.9885167464114834e-05,
-      "loss": 0.0296,
-      "step": 4461
-    },
-    {
-      "epoch": 405.64,
-      "learning_rate": 1.986602870813397e-05,
-      "loss": 0.0281,
-      "step": 4462
-    },
-    {
-      "epoch": 405.73,
-      "learning_rate": 1.9846889952153112e-05,
-      "loss": 0.028,
-      "step": 4463
-    },
-    {
-      "epoch": 405.82,
-      "learning_rate": 1.982775119617225e-05,
-      "loss": 0.0281,
-      "step": 4464
-    },
-    {
-      "epoch": 405.91,
-      "learning_rate": 1.9808612440191386e-05,
-      "loss": 0.0331,
-      "step": 4465
-    },
-    {
-      "epoch": 406.0,
-      "learning_rate": 1.9789473684210528e-05,
-      "loss": 0.029,
-      "step": 4466
-    },
-    {
-      "epoch": 406.09,
-      "learning_rate": 1.9770334928229667e-05,
-      "loss": 0.0314,
-      "step": 4467
-    },
-    {
-      "epoch": 406.18,
-      "learning_rate": 1.9751196172248802e-05,
-      "loss": 0.0319,
-      "step": 4468
-    },
-    {
-      "epoch": 406.27,
-      "learning_rate": 1.9732057416267945e-05,
-      "loss": 0.0296,
-      "step": 4469
-    },
-    {
-      "epoch": 406.36,
-      "learning_rate": 1.9712918660287083e-05,
-      "loss": 0.0323,
-      "step": 4470
-    },
-    {
-      "epoch": 406.45,
-      "learning_rate": 1.969377990430622e-05,
-      "loss": 0.0255,
-      "step": 4471
-    },
-    {
-      "epoch": 406.55,
-      "learning_rate": 1.967464114832536e-05,
-      "loss": 0.0269,
-      "step": 4472
-    },
-    {
-      "epoch": 406.64,
-      "learning_rate": 1.96555023923445e-05,
-      "loss": 0.0322,
-      "step": 4473
-    },
-    {
-      "epoch": 406.73,
-      "learning_rate": 1.9636363636363635e-05,
-      "loss": 0.0287,
-      "step": 4474
-    },
-    {
-      "epoch": 406.82,
-      "learning_rate": 1.9617224880382777e-05,
-      "loss": 0.0281,
-      "step": 4475
-    },
-    {
-      "epoch": 406.91,
-      "learning_rate": 1.9598086124401916e-05,
-      "loss": 0.0315,
-      "step": 4476
-    },
-    {
-      "epoch": 407.0,
-      "learning_rate": 1.957894736842105e-05,
-      "loss": 0.0311,
-      "step": 4477
-    },
-    {
-      "epoch": 407.09,
-      "learning_rate": 1.9559808612440194e-05,
-      "loss": 0.0324,
-      "step": 4478
-    },
-    {
-      "epoch": 407.18,
-      "learning_rate": 1.9540669856459332e-05,
-      "loss": 0.031,
-      "step": 4479
-    },
-    {
-      "epoch": 407.27,
-      "learning_rate": 1.9521531100478468e-05,
-      "loss": 0.0288,
-      "step": 4480
-    },
-    {
-      "epoch": 407.27,
-      "eval_accuracy_NAT": 0.9927970889800631,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9532584552985107,
-      "eval_accuracy_concrete tiles": 0.9851097657922094,
-      "eval_accuracy_dark bricks": 0.875832339016974,
-      "eval_accuracy_grass": 0.9072092885078482,
-      "eval_accuracy_red bricks": 0.9403409090909091,
-      "eval_accuracy_road": 0.9352130081851348,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5335097731094753,
-      "eval_accuracy_speedway bricks": 0.9883614702393236,
-      "eval_accuracy_steel": 0.9092160807334608,
-      "eval_accuracy_tiles": 0.24681550472538008,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806690510302613,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8943281456053636,
-      "eval_iou_concrete tiles": 0.9546087523332643,
-      "eval_iou_dark bricks": 0.8327234135987704,
-      "eval_iou_grass": 0.8388992130310424,
-      "eval_iou_red bricks": 0.8625407166123779,
-      "eval_iou_road": 0.9272466346638021,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.36041270905501543,
-      "eval_iou_speedway bricks": 0.9773113475481984,
-      "eval_iou_steel": 0.8549665074737952,
-      "eval_iou_tiles": 0.24614123753585576,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10339589416980743,
-      "eval_mean_accuracy": 0.7128972064368685,
-      "eval_mean_iou": 0.6715267483452112,
-      "eval_overall_accuracy": 0.9750404357910156,
-      "eval_runtime": 3.4888,
-      "eval_samples_per_second": 10.319,
-      "eval_steps_per_second": 0.573,
-      "step": 4480
-    },
-    {
-      "epoch": 407.36,
-      "learning_rate": 1.9502392344497606e-05,
-      "loss": 0.0333,
-      "step": 4481
-    },
-    {
-      "epoch": 407.45,
-      "learning_rate": 1.948325358851675e-05,
-      "loss": 0.03,
-      "step": 4482
-    },
-    {
-      "epoch": 407.55,
-      "learning_rate": 1.9464114832535887e-05,
-      "loss": 0.029,
-      "step": 4483
-    },
-    {
-      "epoch": 407.64,
-      "learning_rate": 1.9444976076555023e-05,
-      "loss": 0.0314,
-      "step": 4484
-    },
-    {
-      "epoch": 407.73,
-      "learning_rate": 1.9425837320574165e-05,
-      "loss": 0.0289,
-      "step": 4485
-    },
-    {
-      "epoch": 407.82,
-      "learning_rate": 1.9406698564593304e-05,
-      "loss": 0.0328,
-      "step": 4486
-    },
-    {
-      "epoch": 407.91,
-      "learning_rate": 1.938755980861244e-05,
-      "loss": 0.0281,
-      "step": 4487
-    },
-    {
-      "epoch": 408.0,
-      "learning_rate": 1.936842105263158e-05,
-      "loss": 0.0348,
-      "step": 4488
-    },
-    {
-      "epoch": 408.09,
-      "learning_rate": 1.934928229665072e-05,
-      "loss": 0.0287,
-      "step": 4489
-    },
-    {
-      "epoch": 408.18,
-      "learning_rate": 1.9330143540669855e-05,
-      "loss": 0.031,
-      "step": 4490
-    },
-    {
-      "epoch": 408.27,
-      "learning_rate": 1.9311004784688998e-05,
-      "loss": 0.028,
-      "step": 4491
-    },
-    {
-      "epoch": 408.36,
-      "learning_rate": 1.9291866028708136e-05,
-      "loss": 0.0323,
-      "step": 4492
-    },
-    {
-      "epoch": 408.45,
-      "learning_rate": 1.9272727272727272e-05,
-      "loss": 0.0356,
-      "step": 4493
-    },
-    {
-      "epoch": 408.55,
-      "learning_rate": 1.9253588516746414e-05,
-      "loss": 0.0276,
-      "step": 4494
-    },
-    {
-      "epoch": 408.64,
-      "learning_rate": 1.9234449760765553e-05,
-      "loss": 0.0339,
-      "step": 4495
-    },
-    {
-      "epoch": 408.73,
-      "learning_rate": 1.9215311004784688e-05,
-      "loss": 0.0302,
-      "step": 4496
-    },
-    {
-      "epoch": 408.82,
-      "learning_rate": 1.919617224880383e-05,
-      "loss": 0.0329,
-      "step": 4497
-    },
-    {
-      "epoch": 408.91,
-      "learning_rate": 1.917703349282297e-05,
-      "loss": 0.0287,
-      "step": 4498
-    },
-    {
-      "epoch": 409.0,
-      "learning_rate": 1.9157894736842104e-05,
-      "loss": 0.0334,
-      "step": 4499
-    },
-    {
-      "epoch": 409.09,
-      "learning_rate": 1.9138755980861243e-05,
-      "loss": 0.028,
-      "step": 4500
-    },
-    {
-      "epoch": 409.09,
-      "eval_accuracy_NAT": 0.993071999819314,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9511540194858148,
-      "eval_accuracy_concrete tiles": 0.9863327829756008,
-      "eval_accuracy_dark bricks": 0.8798737538970787,
-      "eval_accuracy_grass": 0.909414680008528,
-      "eval_accuracy_red bricks": 0.9383877840909091,
-      "eval_accuracy_road": 0.9237243148924953,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5015852011137761,
-      "eval_accuracy_speedway bricks": 0.9889570337437713,
-      "eval_accuracy_steel": 0.9066931600817888,
-      "eval_accuracy_tiles": 0.3071497055197918,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806886531701277,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8942431246745025,
-      "eval_iou_concrete tiles": 0.9521546635182999,
-      "eval_iou_dark bricks": 0.831454135447734,
-      "eval_iou_grass": 0.8412620919766178,
-      "eval_iou_red bricks": 0.8665354976225611,
-      "eval_iou_road": 0.9165799770347467,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3395796782261376,
-      "eval_iou_speedway bricks": 0.9772734184144104,
-      "eval_iou_steel": 0.8530306086039188,
-      "eval_iou_tiles": 0.3057259713701431,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10897794365882874,
-      "eval_mean_accuracy": 0.7143341873560669,
-      "eval_mean_iou": 0.673732909235323,
-      "eval_overall_accuracy": 0.974156379699707,
-      "eval_runtime": 3.9335,
-      "eval_samples_per_second": 9.152,
-      "eval_steps_per_second": 0.508,
-      "step": 4500
-    },
-    {
-      "epoch": 409.18,
-      "learning_rate": 1.9119617224880385e-05,
-      "loss": 0.0272,
-      "step": 4501
-    },
-    {
-      "epoch": 409.27,
-      "learning_rate": 1.910047846889952e-05,
-      "loss": 0.0313,
-      "step": 4502
-    },
-    {
-      "epoch": 409.36,
-      "learning_rate": 1.908133971291866e-05,
-      "loss": 0.0347,
-      "step": 4503
-    },
-    {
-      "epoch": 409.45,
-      "learning_rate": 1.9062200956937802e-05,
-      "loss": 0.0307,
-      "step": 4504
-    },
-    {
-      "epoch": 409.55,
-      "learning_rate": 1.9043062200956937e-05,
-      "loss": 0.0291,
-      "step": 4505
-    },
-    {
-      "epoch": 409.64,
-      "learning_rate": 1.9023923444976076e-05,
-      "loss": 0.0345,
-      "step": 4506
-    },
-    {
-      "epoch": 409.73,
-      "learning_rate": 1.9004784688995218e-05,
-      "loss": 0.0311,
-      "step": 4507
-    },
-    {
-      "epoch": 409.82,
-      "learning_rate": 1.8985645933014353e-05,
-      "loss": 0.0305,
-      "step": 4508
-    },
-    {
-      "epoch": 409.91,
-      "learning_rate": 1.8966507177033492e-05,
-      "loss": 0.0289,
-      "step": 4509
-    },
-    {
-      "epoch": 410.0,
-      "learning_rate": 1.8947368421052634e-05,
-      "loss": 0.0284,
-      "step": 4510
-    },
-    {
-      "epoch": 410.09,
-      "learning_rate": 1.892822966507177e-05,
-      "loss": 0.0283,
-      "step": 4511
-    },
-    {
-      "epoch": 410.18,
-      "learning_rate": 1.890909090909091e-05,
-      "loss": 0.031,
-      "step": 4512
-    },
-    {
-      "epoch": 410.27,
-      "learning_rate": 1.888995215311005e-05,
-      "loss": 0.0265,
-      "step": 4513
-    },
-    {
-      "epoch": 410.36,
-      "learning_rate": 1.8870813397129186e-05,
-      "loss": 0.0302,
-      "step": 4514
-    },
-    {
-      "epoch": 410.45,
-      "learning_rate": 1.8851674641148325e-05,
-      "loss": 0.0271,
-      "step": 4515
-    },
-    {
-      "epoch": 410.55,
-      "learning_rate": 1.8832535885167467e-05,
-      "loss": 0.0304,
-      "step": 4516
-    },
-    {
-      "epoch": 410.64,
-      "learning_rate": 1.8813397129186606e-05,
-      "loss": 0.0335,
-      "step": 4517
-    },
-    {
-      "epoch": 410.73,
-      "learning_rate": 1.879425837320574e-05,
-      "loss": 0.0321,
-      "step": 4518
-    },
-    {
-      "epoch": 410.82,
-      "learning_rate": 1.877511961722488e-05,
-      "loss": 0.0303,
-      "step": 4519
-    },
-    {
-      "epoch": 410.91,
-      "learning_rate": 1.8755980861244022e-05,
-      "loss": 0.0331,
-      "step": 4520
-    },
-    {
-      "epoch": 410.91,
-      "eval_accuracy_NAT": 0.9931371102812419,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9518045302373492,
-      "eval_accuracy_concrete tiles": 0.9852932183697181,
-      "eval_accuracy_dark bricks": 0.8679804472499134,
-      "eval_accuracy_grass": 0.9056959215985069,
-      "eval_accuracy_red bricks": 0.9401633522727273,
-      "eval_accuracy_road": 0.9285490582076864,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5479833484961266,
-      "eval_accuracy_speedway bricks": 0.988558181184055,
-      "eval_accuracy_steel": 0.9070476881472198,
-      "eval_accuracy_tiles": 0.30091768250924533,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807080078695033,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8955955681816747,
-      "eval_iou_concrete tiles": 0.9524442868120825,
-      "eval_iou_dark bricks": 0.8322937811404318,
-      "eval_iou_grass": 0.838763032086202,
-      "eval_iou_red bricks": 0.866045142296369,
-      "eval_iou_road": 0.9208867590553901,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3684883764042861,
-      "eval_iou_speedway bricks": 0.9773810617549394,
-      "eval_iou_steel": 0.8544843066742266,
-      "eval_iou_tiles": 0.2994820065430752,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10701356828212738,
-      "eval_mean_accuracy": 0.7167023491195222,
-      "eval_mean_iou": 0.67589017913986,
-      "eval_overall_accuracy": 0.9746248457166884,
-      "eval_runtime": 3.2714,
-      "eval_samples_per_second": 11.004,
-      "eval_steps_per_second": 0.611,
-      "step": 4520
-    },
-    {
-      "epoch": 411.0,
-      "learning_rate": 1.8736842105263158e-05,
-      "loss": 0.0331,
-      "step": 4521
-    },
-    {
-      "epoch": 411.09,
-      "learning_rate": 1.8717703349282296e-05,
-      "loss": 0.0271,
-      "step": 4522
-    },
-    {
-      "epoch": 411.18,
-      "learning_rate": 1.869856459330144e-05,
-      "loss": 0.0531,
-      "step": 4523
-    },
-    {
-      "epoch": 411.27,
-      "learning_rate": 1.8679425837320574e-05,
-      "loss": 0.0311,
-      "step": 4524
-    },
-    {
-      "epoch": 411.36,
-      "learning_rate": 1.8660287081339713e-05,
-      "loss": 0.0304,
-      "step": 4525
-    },
-    {
-      "epoch": 411.45,
-      "learning_rate": 1.8641148325358855e-05,
-      "loss": 0.0324,
-      "step": 4526
-    },
-    {
-      "epoch": 411.55,
-      "learning_rate": 1.862200956937799e-05,
-      "loss": 0.0313,
-      "step": 4527
-    },
-    {
-      "epoch": 411.64,
-      "learning_rate": 1.860287081339713e-05,
-      "loss": 0.0279,
-      "step": 4528
-    },
-    {
-      "epoch": 411.73,
-      "learning_rate": 1.858373205741627e-05,
-      "loss": 0.03,
-      "step": 4529
-    },
-    {
-      "epoch": 411.82,
-      "learning_rate": 1.8564593301435407e-05,
-      "loss": 0.0393,
-      "step": 4530
-    },
-    {
-      "epoch": 411.91,
-      "learning_rate": 1.8545454545454545e-05,
-      "loss": 0.0295,
-      "step": 4531
-    },
-    {
-      "epoch": 412.0,
-      "learning_rate": 1.8526315789473687e-05,
-      "loss": 0.0299,
-      "step": 4532
-    },
-    {
-      "epoch": 412.09,
-      "learning_rate": 1.8507177033492823e-05,
-      "loss": 0.0349,
-      "step": 4533
-    },
-    {
-      "epoch": 412.18,
-      "learning_rate": 1.848803827751196e-05,
-      "loss": 0.0319,
-      "step": 4534
-    },
-    {
-      "epoch": 412.27,
-      "learning_rate": 1.8468899521531104e-05,
-      "loss": 0.0303,
-      "step": 4535
-    },
-    {
-      "epoch": 412.36,
-      "learning_rate": 1.844976076555024e-05,
-      "loss": 0.0347,
-      "step": 4536
-    },
-    {
-      "epoch": 412.45,
-      "learning_rate": 1.8430622009569378e-05,
-      "loss": 0.0321,
-      "step": 4537
-    },
-    {
-      "epoch": 412.55,
-      "learning_rate": 1.841148325358852e-05,
-      "loss": 0.0297,
-      "step": 4538
-    },
-    {
-      "epoch": 412.64,
-      "learning_rate": 1.8392344497607656e-05,
-      "loss": 0.0275,
-      "step": 4539
-    },
-    {
-      "epoch": 412.73,
-      "learning_rate": 1.8373205741626794e-05,
-      "loss": 0.0338,
-      "step": 4540
-    },
-    {
-      "epoch": 412.73,
-      "eval_accuracy_NAT": 0.9933530864476368,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9498932357611914,
-      "eval_accuracy_concrete tiles": 0.9842230783342506,
-      "eval_accuracy_dark bricks": 0.8837996997806089,
-      "eval_accuracy_grass": 0.8950114813170835,
-      "eval_accuracy_red bricks": 0.9291548295454546,
-      "eval_accuracy_road": 0.9255405465865848,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4973671877153806,
-      "eval_accuracy_speedway bricks": 0.9895471665779657,
-      "eval_accuracy_steel": 0.9047556229800146,
-      "eval_accuracy_tiles": 0.1936036159430215,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9803234309374976,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8908055374568251,
-      "eval_iou_concrete tiles": 0.9559871703492516,
-      "eval_iou_dark bricks": 0.8332849470169836,
-      "eval_iou_grass": 0.8323381904108743,
-      "eval_iou_red bricks": 0.8638164410696599,
-      "eval_iou_road": 0.9183535122903466,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33702596674761814,
-      "eval_iou_speedway bricks": 0.9773792736812871,
-      "eval_iou_steel": 0.8556346539208272,
-      "eval_iou_tiles": 0.19333880454110244,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11009025573730469,
-      "eval_mean_accuracy": 0.7035576577683995,
-      "eval_mean_iou": 0.6644836868017134,
-      "eval_overall_accuracy": 0.9738328721788194,
-      "eval_runtime": 3.3603,
-      "eval_samples_per_second": 10.713,
-      "eval_steps_per_second": 0.595,
-      "step": 4540
-    },
-    {
-      "epoch": 412.82,
-      "learning_rate": 1.8354066985645933e-05,
-      "loss": 0.0331,
-      "step": 4541
-    },
-    {
-      "epoch": 412.91,
-      "learning_rate": 1.8334928229665072e-05,
-      "loss": 0.0302,
-      "step": 4542
-    },
-    {
-      "epoch": 413.0,
-      "learning_rate": 1.831578947368421e-05,
-      "loss": 0.03,
-      "step": 4543
-    },
-    {
-      "epoch": 413.09,
-      "learning_rate": 1.829665071770335e-05,
-      "loss": 0.0298,
-      "step": 4544
-    },
-    {
-      "epoch": 413.18,
-      "learning_rate": 1.8277511961722488e-05,
-      "loss": 0.0335,
-      "step": 4545
-    },
-    {
-      "epoch": 413.27,
-      "learning_rate": 1.8258373205741627e-05,
-      "loss": 0.0301,
-      "step": 4546
-    },
-    {
-      "epoch": 413.36,
-      "learning_rate": 1.8239234449760766e-05,
-      "loss": 0.0325,
-      "step": 4547
-    },
-    {
-      "epoch": 413.45,
-      "learning_rate": 1.8220095693779905e-05,
-      "loss": 0.0285,
-      "step": 4548
-    },
-    {
-      "epoch": 413.55,
-      "learning_rate": 1.8200956937799043e-05,
-      "loss": 0.0278,
-      "step": 4549
-    },
-    {
-      "epoch": 413.64,
-      "learning_rate": 1.8181818181818182e-05,
-      "loss": 0.033,
-      "step": 4550
-    },
-    {
-      "epoch": 413.73,
-      "learning_rate": 1.816267942583732e-05,
-      "loss": 0.0313,
-      "step": 4551
-    },
-    {
-      "epoch": 413.82,
-      "learning_rate": 1.814354066985646e-05,
-      "loss": 0.0255,
-      "step": 4552
-    },
-    {
-      "epoch": 413.91,
-      "learning_rate": 1.81244019138756e-05,
-      "loss": 0.029,
-      "step": 4553
-    },
-    {
-      "epoch": 414.0,
-      "learning_rate": 1.810526315789474e-05,
-      "loss": 0.0325,
-      "step": 4554
-    },
-    {
-      "epoch": 414.09,
-      "learning_rate": 1.8086124401913876e-05,
-      "loss": 0.0305,
-      "step": 4555
-    },
-    {
-      "epoch": 414.18,
-      "learning_rate": 1.8066985645933015e-05,
-      "loss": 0.0316,
-      "step": 4556
-    },
-    {
-      "epoch": 414.27,
-      "learning_rate": 1.8047846889952157e-05,
-      "loss": 0.0321,
-      "step": 4557
-    },
-    {
-      "epoch": 414.36,
-      "learning_rate": 1.8028708133971292e-05,
-      "loss": 0.032,
-      "step": 4558
-    },
-    {
-      "epoch": 414.45,
-      "learning_rate": 1.800956937799043e-05,
-      "loss": 0.0322,
-      "step": 4559
-    },
-    {
-      "epoch": 414.55,
-      "learning_rate": 1.799043062200957e-05,
-      "loss": 0.0287,
-      "step": 4560
-    },
-    {
-      "epoch": 414.55,
-      "eval_accuracy_NAT": 0.993071470465965,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9505853255504528,
-      "eval_accuracy_concrete tiles": 0.9843759554821745,
-      "eval_accuracy_dark bricks": 0.8802971402178515,
-      "eval_accuracy_grass": 0.901438921839667,
-      "eval_accuracy_red bricks": 0.9334161931818182,
-      "eval_accuracy_road": 0.928877963015138,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5477903674909712,
-      "eval_accuracy_speedway bricks": 0.9885050812971337,
-      "eval_accuracy_steel": 0.9123161400962997,
-      "eval_accuracy_tiles": 0.21818928913847419,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806592515834142,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8943490369727263,
-      "eval_iou_concrete tiles": 0.957557551603117,
-      "eval_iou_dark bricks": 0.8309777277186353,
-      "eval_iou_grass": 0.8360654674883059,
-      "eval_iou_red bricks": 0.8633601576613565,
-      "eval_iou_road": 0.9199971256452327,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3688988730668548,
-      "eval_iou_speedway bricks": 0.9772871667391487,
-      "eval_iou_steel": 0.8542390414871771,
-      "eval_iou_tiles": 0.21769730099077553,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10849345475435257,
-      "eval_mean_accuracy": 0.7106818344443036,
-      "eval_mean_iou": 0.6693145154582111,
-      "eval_overall_accuracy": 0.9743452072143555,
-      "eval_runtime": 3.5969,
-      "eval_samples_per_second": 10.009,
-      "eval_steps_per_second": 0.556,
-      "step": 4560
-    },
-    {
-      "epoch": 414.64,
-      "learning_rate": 1.797129186602871e-05,
-      "loss": 0.0279,
-      "step": 4561
-    },
-    {
-      "epoch": 414.73,
-      "learning_rate": 1.7952153110047847e-05,
-      "loss": 0.031,
-      "step": 4562
-    },
-    {
-      "epoch": 414.82,
-      "learning_rate": 1.7933014354066986e-05,
-      "loss": 0.0309,
-      "step": 4563
-    },
-    {
-      "epoch": 414.91,
-      "learning_rate": 1.7913875598086125e-05,
-      "loss": 0.0278,
-      "step": 4564
-    },
-    {
-      "epoch": 415.0,
-      "learning_rate": 1.7894736842105264e-05,
-      "loss": 0.0304,
-      "step": 4565
-    },
-    {
-      "epoch": 415.09,
-      "learning_rate": 1.7875598086124403e-05,
-      "loss": 0.0304,
-      "step": 4566
-    },
-    {
-      "epoch": 415.18,
-      "learning_rate": 1.785645933014354e-05,
-      "loss": 0.0366,
-      "step": 4567
-    },
-    {
-      "epoch": 415.27,
-      "learning_rate": 1.783732057416268e-05,
-      "loss": 0.0312,
-      "step": 4568
-    },
-    {
-      "epoch": 415.36,
-      "learning_rate": 1.781818181818182e-05,
-      "loss": 0.0309,
-      "step": 4569
-    },
-    {
-      "epoch": 415.45,
-      "learning_rate": 1.7799043062200958e-05,
-      "loss": 0.0427,
-      "step": 4570
-    },
-    {
-      "epoch": 415.55,
-      "learning_rate": 1.7779904306220096e-05,
-      "loss": 0.0283,
-      "step": 4571
-    },
-    {
-      "epoch": 415.64,
-      "learning_rate": 1.7760765550239235e-05,
-      "loss": 0.0288,
-      "step": 4572
-    },
-    {
-      "epoch": 415.73,
-      "learning_rate": 1.7741626794258374e-05,
-      "loss": 0.0286,
-      "step": 4573
-    },
-    {
-      "epoch": 415.82,
-      "learning_rate": 1.7722488038277513e-05,
-      "loss": 0.0284,
-      "step": 4574
-    },
-    {
-      "epoch": 415.91,
-      "learning_rate": 1.770334928229665e-05,
-      "loss": 0.0307,
-      "step": 4575
-    },
-    {
-      "epoch": 416.0,
-      "learning_rate": 1.768421052631579e-05,
-      "loss": 0.0251,
-      "step": 4576
-    },
-    {
-      "epoch": 416.09,
-      "learning_rate": 1.766507177033493e-05,
-      "loss": 0.0301,
-      "step": 4577
-    },
-    {
-      "epoch": 416.18,
-      "learning_rate": 1.7645933014354068e-05,
-      "loss": 0.0298,
-      "step": 4578
-    },
-    {
-      "epoch": 416.27,
-      "learning_rate": 1.7626794258373207e-05,
-      "loss": 0.029,
-      "step": 4579
-    },
-    {
-      "epoch": 416.36,
-      "learning_rate": 1.7607655502392345e-05,
-      "loss": 0.0326,
-      "step": 4580
-    },
-    {
-      "epoch": 416.36,
-      "eval_accuracy_NAT": 0.9932378638686695,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9538030065668054,
-      "eval_accuracy_concrete tiles": 0.9819604965449764,
-      "eval_accuracy_dark bricks": 0.8720218621300181,
-      "eval_accuracy_grass": 0.89579787289815,
-      "eval_accuracy_red bricks": 0.9192116477272727,
-      "eval_accuracy_road": 0.9393376195756402,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4789237173655336,
-      "eval_accuracy_speedway bricks": 0.9886136947022002,
-      "eval_accuracy_steel": 0.9065447529846317,
-      "eval_accuracy_tiles": 0.1779893165319819,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980669946386031,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8888151464474892,
-      "eval_iou_concrete tiles": 0.9614129621314175,
-      "eval_iou_dark bricks": 0.8289488127035235,
-      "eval_iou_grass": 0.8343386730817434,
-      "eval_iou_red bricks": 0.8619713619713619,
-      "eval_iou_road": 0.9320579857603252,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3229476502082094,
-      "eval_iou_speedway bricks": 0.9774283670570493,
-      "eval_iou_steel": 0.8551595943255351,
-      "eval_iou_tiles": 0.17758797403484797,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10411757975816727,
-      "eval_mean_accuracy": 0.700572450068914,
-      "eval_mean_iou": 0.6631798826236565,
-      "eval_overall_accuracy": 0.9750232696533203,
-      "eval_runtime": 3.826,
-      "eval_samples_per_second": 9.409,
-      "eval_steps_per_second": 0.523,
-      "step": 4580
-    },
-    {
-      "epoch": 416.45,
-      "learning_rate": 1.7588516746411484e-05,
-      "loss": 0.0287,
-      "step": 4581
-    },
-    {
-      "epoch": 416.55,
-      "learning_rate": 1.7569377990430623e-05,
-      "loss": 0.0281,
-      "step": 4582
-    },
-    {
-      "epoch": 416.64,
-      "learning_rate": 1.7550239234449762e-05,
-      "loss": 0.0279,
-      "step": 4583
-    },
-    {
-      "epoch": 416.73,
-      "learning_rate": 1.75311004784689e-05,
-      "loss": 0.0381,
-      "step": 4584
-    },
-    {
-      "epoch": 416.82,
-      "learning_rate": 1.751196172248804e-05,
-      "loss": 0.0331,
-      "step": 4585
-    },
-    {
-      "epoch": 416.91,
-      "learning_rate": 1.7492822966507178e-05,
-      "loss": 0.0256,
-      "step": 4586
-    },
-    {
-      "epoch": 417.0,
-      "learning_rate": 1.7473684210526317e-05,
-      "loss": 0.0328,
-      "step": 4587
-    },
-    {
-      "epoch": 417.09,
-      "learning_rate": 1.7454545454545456e-05,
-      "loss": 0.0319,
-      "step": 4588
-    },
-    {
-      "epoch": 417.18,
-      "learning_rate": 1.7435406698564594e-05,
-      "loss": 0.0281,
-      "step": 4589
-    },
-    {
-      "epoch": 417.27,
-      "learning_rate": 1.7416267942583733e-05,
-      "loss": 0.0261,
-      "step": 4590
-    },
-    {
-      "epoch": 417.36,
-      "learning_rate": 1.7397129186602872e-05,
-      "loss": 0.0316,
-      "step": 4591
-    },
-    {
-      "epoch": 417.45,
-      "learning_rate": 1.737799043062201e-05,
-      "loss": 0.0306,
-      "step": 4592
-    },
-    {
-      "epoch": 417.55,
-      "learning_rate": 1.735885167464115e-05,
-      "loss": 0.0312,
-      "step": 4593
-    },
-    {
-      "epoch": 417.64,
-      "learning_rate": 1.7339712918660288e-05,
-      "loss": 0.0306,
-      "step": 4594
-    },
-    {
-      "epoch": 417.73,
-      "learning_rate": 1.7320574162679427e-05,
-      "loss": 0.0352,
-      "step": 4595
-    },
-    {
-      "epoch": 417.82,
-      "learning_rate": 1.7301435406698566e-05,
-      "loss": 0.032,
-      "step": 4596
-    },
-    {
-      "epoch": 417.91,
-      "learning_rate": 1.7282296650717705e-05,
-      "loss": 0.0313,
-      "step": 4597
-    },
-    {
-      "epoch": 418.0,
-      "learning_rate": 1.7263157894736843e-05,
-      "loss": 0.0293,
-      "step": 4598
-    },
-    {
-      "epoch": 418.09,
-      "learning_rate": 1.7244019138755982e-05,
-      "loss": 0.033,
-      "step": 4599
-    },
-    {
-      "epoch": 418.18,
-      "learning_rate": 1.722488038277512e-05,
-      "loss": 0.0295,
-      "step": 4600
-    },
-    {
-      "epoch": 418.18,
-      "eval_accuracy_NAT": 0.9926527519669006,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9531793210009013,
-      "eval_accuracy_concrete tiles": 0.9851709166513789,
-      "eval_accuracy_dark bricks": 0.8812208921904469,
-      "eval_accuracy_grass": 0.9061992122103895,
-      "eval_accuracy_red bricks": 0.9353693181818182,
-      "eval_accuracy_road": 0.9111291955012594,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5387202602486698,
-      "eval_accuracy_speedway bricks": 0.9890499585458837,
-      "eval_accuracy_steel": 0.9036095903964119,
-      "eval_accuracy_tiles": 0.2321599780851938,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804916145889977,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8926835273221504,
-      "eval_iou_concrete tiles": 0.9554323330565769,
-      "eval_iou_dark bricks": 0.8348526837806302,
-      "eval_iou_grass": 0.8365862708719851,
-      "eval_iou_red bricks": 0.8643150123051682,
-      "eval_iou_road": 0.9044382160712036,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3599771571733844,
-      "eval_iou_speedway bricks": 0.9774235271177072,
-      "eval_iou_steel": 0.8528949416342413,
-      "eval_iou_tiles": 0.2311783960720131,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11710582673549652,
-      "eval_mean_accuracy": 0.7098816457676348,
-      "eval_mean_iou": 0.6684825907687736,
-      "eval_overall_accuracy": 0.9728646808200412,
-      "eval_runtime": 3.5134,
-      "eval_samples_per_second": 10.246,
-      "eval_steps_per_second": 0.569,
-      "step": 4600
-    },
-    {
-      "epoch": 418.27,
-      "learning_rate": 1.720574162679426e-05,
-      "loss": 0.0277,
-      "step": 4601
-    },
-    {
-      "epoch": 418.36,
-      "learning_rate": 1.71866028708134e-05,
-      "loss": 0.0366,
-      "step": 4602
-    },
-    {
-      "epoch": 418.45,
-      "learning_rate": 1.7167464114832537e-05,
-      "loss": 0.0294,
-      "step": 4603
-    },
-    {
-      "epoch": 418.55,
-      "learning_rate": 1.7148325358851676e-05,
-      "loss": 0.0273,
-      "step": 4604
-    },
-    {
-      "epoch": 418.64,
-      "learning_rate": 1.7129186602870815e-05,
-      "loss": 0.0288,
-      "step": 4605
-    },
-    {
-      "epoch": 418.73,
-      "learning_rate": 1.7110047846889954e-05,
-      "loss": 0.0284,
-      "step": 4606
-    },
-    {
-      "epoch": 418.82,
-      "learning_rate": 1.7090909090909092e-05,
-      "loss": 0.0291,
-      "step": 4607
-    },
-    {
-      "epoch": 418.91,
-      "learning_rate": 1.707177033492823e-05,
-      "loss": 0.0367,
-      "step": 4608
-    },
-    {
-      "epoch": 419.0,
-      "learning_rate": 1.705263157894737e-05,
-      "loss": 0.0301,
-      "step": 4609
-    },
-    {
-      "epoch": 419.09,
-      "learning_rate": 1.7033492822966505e-05,
-      "loss": 0.0323,
-      "step": 4610
-    },
-    {
-      "epoch": 419.18,
-      "learning_rate": 1.7014354066985647e-05,
-      "loss": 0.0351,
-      "step": 4611
-    },
-    {
-      "epoch": 419.27,
-      "learning_rate": 1.6995215311004786e-05,
-      "loss": 0.0317,
-      "step": 4612
-    },
-    {
-      "epoch": 419.36,
-      "learning_rate": 1.697607655502392e-05,
-      "loss": 0.0332,
-      "step": 4613
-    },
-    {
-      "epoch": 419.45,
-      "learning_rate": 1.6956937799043064e-05,
-      "loss": 0.0294,
-      "step": 4614
-    },
-    {
-      "epoch": 419.55,
-      "learning_rate": 1.6937799043062203e-05,
-      "loss": 0.0309,
-      "step": 4615
-    },
-    {
-      "epoch": 419.64,
-      "learning_rate": 1.6918660287081338e-05,
-      "loss": 0.0279,
-      "step": 4616
-    },
-    {
-      "epoch": 419.73,
-      "learning_rate": 1.689952153110048e-05,
-      "loss": 0.0303,
-      "step": 4617
-    },
-    {
-      "epoch": 419.82,
-      "learning_rate": 1.688038277511962e-05,
-      "loss": 0.0299,
-      "step": 4618
-    },
-    {
-      "epoch": 419.91,
-      "learning_rate": 1.6861244019138754e-05,
-      "loss": 0.0342,
-      "step": 4619
-    },
-    {
-      "epoch": 420.0,
-      "learning_rate": 1.6842105263157896e-05,
-      "loss": 0.0323,
-      "step": 4620
-    },
-    {
-      "epoch": 420.0,
-      "eval_accuracy_NAT": 0.9930587659855888,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9519721876475385,
-      "eval_accuracy_concrete tiles": 0.986883140708127,
-      "eval_accuracy_dark bricks": 0.8866864246949694,
-      "eval_accuracy_grass": 0.8852846912277146,
-      "eval_accuracy_red bricks": 0.9275568181818182,
-      "eval_accuracy_road": 0.8719520379401369,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4612246023212858,
-      "eval_accuracy_speedway bricks": 0.9887554955365925,
-      "eval_accuracy_steel": 0.9058274520150386,
-      "eval_accuracy_tiles": 0.20182166826462128,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806087734536512,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8833969760182415,
-      "eval_iou_concrete tiles": 0.9529672276350752,
-      "eval_iou_dark bricks": 0.8359460047898977,
-      "eval_iou_grass": 0.8226026974626444,
-      "eval_iou_red bricks": 0.866334991708126,
-      "eval_iou_road": 0.8648573769056491,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.311632672068548,
-      "eval_iou_speedway bricks": 0.977247909521228,
-      "eval_iou_steel": 0.8532817633216058,
-      "eval_iou_tiles": 0.2008450896203912,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.15534144639968872,
-      "eval_mean_accuracy": 0.697001791117187,
-      "eval_mean_iou": 0.6576708832696199,
-      "eval_overall_accuracy": 0.9685942331949869,
-      "eval_runtime": 3.3771,
-      "eval_samples_per_second": 10.66,
-      "eval_steps_per_second": 0.592,
-      "step": 4620
-    },
-    {
-      "epoch": 420.09,
-      "learning_rate": 1.6822966507177035e-05,
-      "loss": 0.0252,
-      "step": 4621
-    },
-    {
-      "epoch": 420.18,
-      "learning_rate": 1.680382775119617e-05,
-      "loss": 0.0287,
-      "step": 4622
-    },
-    {
-      "epoch": 420.27,
-      "learning_rate": 1.6784688995215313e-05,
-      "loss": 0.028,
-      "step": 4623
-    },
-    {
-      "epoch": 420.36,
-      "learning_rate": 1.676555023923445e-05,
-      "loss": 0.0267,
-      "step": 4624
-    },
-    {
-      "epoch": 420.45,
-      "learning_rate": 1.674641148325359e-05,
-      "loss": 0.0357,
-      "step": 4625
-    },
-    {
-      "epoch": 420.55,
-      "learning_rate": 1.672727272727273e-05,
-      "loss": 0.0328,
-      "step": 4626
-    },
-    {
-      "epoch": 420.64,
-      "learning_rate": 1.6708133971291868e-05,
-      "loss": 0.0273,
-      "step": 4627
-    },
-    {
-      "epoch": 420.73,
-      "learning_rate": 1.6688995215311007e-05,
-      "loss": 0.0309,
-      "step": 4628
-    },
-    {
-      "epoch": 420.82,
-      "learning_rate": 1.6669856459330142e-05,
-      "loss": 0.0314,
-      "step": 4629
-    },
-    {
-      "epoch": 420.91,
-      "learning_rate": 1.6650717703349284e-05,
-      "loss": 0.0318,
-      "step": 4630
-    },
-    {
-      "epoch": 421.0,
-      "learning_rate": 1.6631578947368423e-05,
-      "loss": 0.0333,
-      "step": 4631
-    },
-    {
-      "epoch": 421.09,
-      "learning_rate": 1.661244019138756e-05,
-      "loss": 0.0261,
-      "step": 4632
-    },
-    {
-      "epoch": 421.18,
-      "learning_rate": 1.65933014354067e-05,
-      "loss": 0.0283,
-      "step": 4633
-    },
-    {
-      "epoch": 421.27,
-      "learning_rate": 1.657416267942584e-05,
-      "loss": 0.0304,
-      "step": 4634
-    },
-    {
-      "epoch": 421.36,
-      "learning_rate": 1.6555023923444975e-05,
-      "loss": 0.0323,
-      "step": 4635
-    },
-    {
-      "epoch": 421.45,
-      "learning_rate": 1.6535885167464117e-05,
-      "loss": 0.0285,
-      "step": 4636
-    },
-    {
-      "epoch": 421.55,
-      "learning_rate": 1.6516746411483256e-05,
-      "loss": 0.0274,
-      "step": 4637
-    },
-    {
-      "epoch": 421.64,
-      "learning_rate": 1.649760765550239e-05,
-      "loss": 0.0299,
-      "step": 4638
-    },
-    {
-      "epoch": 421.73,
-      "learning_rate": 1.6478468899521533e-05,
-      "loss": 0.0306,
-      "step": 4639
-    },
-    {
-      "epoch": 421.82,
-      "learning_rate": 1.6459330143540672e-05,
-      "loss": 0.0268,
-      "step": 4640
-    },
-    {
-      "epoch": 421.82,
-      "eval_accuracy_NAT": 0.9931545789417592,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9489972745611399,
-      "eval_accuracy_concrete tiles": 0.9814101388124503,
-      "eval_accuracy_dark bricks": 0.8870328316846927,
-      "eval_accuracy_grass": 0.900365934215723,
-      "eval_accuracy_red bricks": 0.9254261363636364,
-      "eval_accuracy_road": 0.8731721296707203,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4929837620268519,
-      "eval_accuracy_speedway bricks": 0.9889322940237284,
-      "eval_accuracy_steel": 0.9067508739529054,
-      "eval_accuracy_tiles": 0.22154499383646076,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804138346681902,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8900851034387325,
-      "eval_iou_concrete tiles": 0.9558665872543181,
-      "eval_iou_dark bricks": 0.8347580411474935,
-      "eval_iou_grass": 0.8337324910027704,
-      "eval_iou_red bricks": 0.868232550391471,
-      "eval_iou_road": 0.8657228184049213,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3318918316969506,
-      "eval_iou_speedway bricks": 0.9772017734868956,
-      "eval_iou_steel": 0.8530253554336951,
-      "eval_iou_tiles": 0.22048800436205016,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1444719433784485,
-      "eval_mean_accuracy": 0.7015208421607745,
-      "eval_mean_iou": 0.6624167993298068,
-      "eval_overall_accuracy": 0.9691571129692925,
-      "eval_runtime": 3.455,
-      "eval_samples_per_second": 10.42,
-      "eval_steps_per_second": 0.579,
-      "step": 4640
-    },
-    {
-      "epoch": 421.91,
-      "learning_rate": 1.6440191387559807e-05,
-      "loss": 0.0334,
-      "step": 4641
-    },
-    {
-      "epoch": 422.0,
-      "learning_rate": 1.642105263157895e-05,
-      "loss": 0.0352,
-      "step": 4642
-    },
-    {
-      "epoch": 422.09,
-      "learning_rate": 1.640191387559809e-05,
-      "loss": 0.0319,
-      "step": 4643
-    },
-    {
-      "epoch": 422.18,
-      "learning_rate": 1.6382775119617224e-05,
-      "loss": 0.0293,
-      "step": 4644
-    },
-    {
-      "epoch": 422.27,
-      "learning_rate": 1.6363636363636366e-05,
-      "loss": 0.0367,
-      "step": 4645
-    },
-    {
-      "epoch": 422.36,
-      "learning_rate": 1.6344497607655505e-05,
-      "loss": 0.0292,
-      "step": 4646
-    },
-    {
-      "epoch": 422.45,
-      "learning_rate": 1.632535885167464e-05,
-      "loss": 0.0344,
-      "step": 4647
-    },
-    {
-      "epoch": 422.55,
-      "learning_rate": 1.630622009569378e-05,
-      "loss": 0.0297,
-      "step": 4648
-    },
-    {
-      "epoch": 422.64,
-      "learning_rate": 1.628708133971292e-05,
-      "loss": 0.0298,
-      "step": 4649
-    },
-    {
-      "epoch": 422.73,
-      "learning_rate": 1.6267942583732056e-05,
-      "loss": 0.0269,
-      "step": 4650
-    },
-    {
-      "epoch": 422.82,
-      "learning_rate": 1.6248803827751195e-05,
-      "loss": 0.0312,
-      "step": 4651
-    },
-    {
-      "epoch": 422.91,
-      "learning_rate": 1.6229665071770337e-05,
-      "loss": 0.0282,
-      "step": 4652
-    },
-    {
-      "epoch": 423.0,
-      "learning_rate": 1.6210526315789473e-05,
-      "loss": 0.0293,
-      "step": 4653
-    },
-    {
-      "epoch": 423.09,
-      "learning_rate": 1.619138755980861e-05,
-      "loss": 0.0321,
-      "step": 4654
-    },
-    {
-      "epoch": 423.18,
-      "learning_rate": 1.6172248803827754e-05,
-      "loss": 0.0291,
-      "step": 4655
-    },
-    {
-      "epoch": 423.27,
-      "learning_rate": 1.615311004784689e-05,
-      "loss": 0.0293,
-      "step": 4656
-    },
-    {
-      "epoch": 423.36,
-      "learning_rate": 1.6133971291866028e-05,
-      "loss": 0.0337,
-      "step": 4657
-    },
-    {
-      "epoch": 423.45,
-      "learning_rate": 1.611483253588517e-05,
-      "loss": 0.0281,
-      "step": 4658
-    },
-    {
-      "epoch": 423.55,
-      "learning_rate": 1.609569377990431e-05,
-      "loss": 0.033,
-      "step": 4659
-    },
-    {
-      "epoch": 423.64,
-      "learning_rate": 1.6076555023923444e-05,
-      "loss": 0.0328,
-      "step": 4660
-    },
-    {
-      "epoch": 423.64,
-      "eval_accuracy_NAT": 0.9933867886108568,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9512546139319284,
-      "eval_accuracy_concrete tiles": 0.9860576041093377,
-      "eval_accuracy_dark bricks": 0.8820676648319926,
-      "eval_accuracy_grass": 0.8970910501647927,
-      "eval_accuracy_red bricks": 0.9247159090909091,
-      "eval_accuracy_road": 0.8761878965449276,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5015852011137761,
-      "eval_accuracy_speedway bricks": 0.9883958644842613,
-      "eval_accuracy_steel": 0.9091666117010752,
-      "eval_accuracy_tiles": 0.18963155732091494,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805479302697537,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8896316161823956,
-      "eval_iou_concrete tiles": 0.9552442166997424,
-      "eval_iou_dark bricks": 0.836112225911197,
-      "eval_iou_grass": 0.8358233742551043,
-      "eval_iou_red bricks": 0.8586974443528442,
-      "eval_iou_road": 0.8693069140633904,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.336601790868053,
-      "eval_iou_speedway bricks": 0.9773511887359172,
-      "eval_iou_steel": 0.853840971915724,
-      "eval_iou_tiles": 0.1889715416638231,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13898244500160217,
-      "eval_mean_accuracy": 0.6999646739926747,
-      "eval_mean_iou": 0.6601637857629188,
-      "eval_overall_accuracy": 0.9695624245537652,
-      "eval_runtime": 3.9734,
-      "eval_samples_per_second": 9.06,
-      "eval_steps_per_second": 0.503,
-      "step": 4660
-    },
-    {
-      "epoch": 423.73,
-      "learning_rate": 1.6057416267942586e-05,
-      "loss": 0.028,
-      "step": 4661
-    },
-    {
-      "epoch": 423.82,
-      "learning_rate": 1.6038277511961725e-05,
-      "loss": 0.0321,
-      "step": 4662
-    },
-    {
-      "epoch": 423.91,
-      "learning_rate": 1.601913875598086e-05,
-      "loss": 0.0318,
-      "step": 4663
-    },
-    {
-      "epoch": 424.0,
-      "learning_rate": 1.6000000000000003e-05,
-      "loss": 0.0312,
-      "step": 4664
-    },
-    {
-      "epoch": 424.09,
-      "learning_rate": 1.598086124401914e-05,
-      "loss": 0.0317,
-      "step": 4665
-    },
-    {
-      "epoch": 424.18,
-      "learning_rate": 1.5961722488038277e-05,
-      "loss": 0.0294,
-      "step": 4666
-    },
-    {
-      "epoch": 424.27,
-      "learning_rate": 1.5942583732057416e-05,
-      "loss": 0.0349,
-      "step": 4667
-    },
-    {
-      "epoch": 424.36,
-      "learning_rate": 1.5923444976076558e-05,
-      "loss": 0.0348,
-      "step": 4668
-    },
-    {
-      "epoch": 424.45,
-      "learning_rate": 1.5904306220095693e-05,
-      "loss": 0.0283,
-      "step": 4669
-    },
-    {
-      "epoch": 424.55,
-      "learning_rate": 1.5885167464114832e-05,
-      "loss": 0.0312,
-      "step": 4670
-    },
-    {
-      "epoch": 424.64,
-      "learning_rate": 1.5866028708133974e-05,
-      "loss": 0.0297,
-      "step": 4671
-    },
-    {
-      "epoch": 424.73,
-      "learning_rate": 1.584688995215311e-05,
-      "loss": 0.0325,
-      "step": 4672
-    },
-    {
-      "epoch": 424.82,
-      "learning_rate": 1.5827751196172248e-05,
-      "loss": 0.0258,
-      "step": 4673
-    },
-    {
-      "epoch": 424.91,
-      "learning_rate": 1.580861244019139e-05,
-      "loss": 0.0276,
-      "step": 4674
-    },
-    {
-      "epoch": 425.0,
-      "learning_rate": 1.5789473684210526e-05,
-      "loss": 0.0327,
-      "step": 4675
-    },
-    {
-      "epoch": 425.09,
-      "learning_rate": 1.5770334928229665e-05,
-      "loss": 0.0319,
-      "step": 4676
-    },
-    {
-      "epoch": 425.18,
-      "learning_rate": 1.5751196172248807e-05,
-      "loss": 0.0303,
-      "step": 4677
-    },
-    {
-      "epoch": 425.27,
-      "learning_rate": 1.5732057416267942e-05,
-      "loss": 0.0282,
-      "step": 4678
-    },
-    {
-      "epoch": 425.36,
-      "learning_rate": 1.571291866028708e-05,
-      "loss": 0.0345,
-      "step": 4679
-    },
-    {
-      "epoch": 425.45,
-      "learning_rate": 1.5693779904306223e-05,
-      "loss": 0.0288,
-      "step": 4680
-    },
-    {
-      "epoch": 425.45,
-      "eval_accuracy_NAT": 0.9931438154236627,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9511982810421048,
-      "eval_accuracy_concrete tiles": 0.982449703418333,
-      "eval_accuracy_dark bricks": 0.8748700973788538,
-      "eval_accuracy_grass": 0.906919197391277,
-      "eval_accuracy_red bricks": 0.9289772727272727,
-      "eval_accuracy_road": 0.8841807670495425,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.528023598820059,
-      "eval_accuracy_speedway bricks": 0.9885805072728743,
-      "eval_accuracy_steel": 0.9061819800804696,
-      "eval_accuracy_tiles": 0.21099849335707438,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805284025471556,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8917456225833831,
-      "eval_iou_concrete tiles": 0.9537265144994213,
-      "eval_iou_dark bricks": 0.8356310429763611,
-      "eval_iou_grass": 0.8392032444599682,
-      "eval_iou_red bricks": 0.8616600790513834,
-      "eval_iou_road": 0.8774317572010732,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3548034530028528,
-      "eval_iou_speedway bricks": 0.9775740148802473,
-      "eval_iou_steel": 0.8546644997239481,
-      "eval_iou_tiles": 0.21042207348722852,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1314036101102829,
-      "eval_mean_accuracy": 0.7042710549201173,
-      "eval_mean_iou": 0.6644146695702324,
-      "eval_overall_accuracy": 0.9705094231499566,
-      "eval_runtime": 3.386,
-      "eval_samples_per_second": 10.632,
-      "eval_steps_per_second": 0.591,
-      "step": 4680
-    },
-    {
-      "epoch": 425.55,
-      "learning_rate": 1.567464114832536e-05,
-      "loss": 0.0353,
-      "step": 4681
-    },
-    {
-      "epoch": 425.64,
-      "learning_rate": 1.5655502392344497e-05,
-      "loss": 0.0288,
-      "step": 4682
-    },
-    {
-      "epoch": 425.73,
-      "learning_rate": 1.563636363636364e-05,
-      "loss": 0.0263,
-      "step": 4683
-    },
-    {
-      "epoch": 425.82,
-      "learning_rate": 1.5617224880382775e-05,
-      "loss": 0.0314,
-      "step": 4684
-    },
-    {
-      "epoch": 425.91,
-      "learning_rate": 1.5598086124401914e-05,
-      "loss": 0.0325,
-      "step": 4685
-    },
-    {
-      "epoch": 426.0,
-      "learning_rate": 1.5578947368421056e-05,
-      "loss": 0.0299,
-      "step": 4686
-    },
-    {
-      "epoch": 426.09,
-      "learning_rate": 1.555980861244019e-05,
-      "loss": 0.0328,
-      "step": 4687
-    },
-    {
-      "epoch": 426.18,
-      "learning_rate": 1.554066985645933e-05,
-      "loss": 0.0312,
-      "step": 4688
-    },
-    {
-      "epoch": 426.27,
-      "learning_rate": 1.552153110047847e-05,
-      "loss": 0.0259,
-      "step": 4689
-    },
-    {
-      "epoch": 426.36,
-      "learning_rate": 1.5502392344497607e-05,
-      "loss": 0.0331,
-      "step": 4690
-    },
-    {
-      "epoch": 426.45,
-      "learning_rate": 1.5483253588516746e-05,
-      "loss": 0.0257,
-      "step": 4691
-    },
-    {
-      "epoch": 426.55,
-      "learning_rate": 1.5464114832535885e-05,
-      "loss": 0.0338,
-      "step": 4692
-    },
-    {
-      "epoch": 426.64,
-      "learning_rate": 1.5444976076555024e-05,
-      "loss": 0.0305,
-      "step": 4693
-    },
-    {
-      "epoch": 426.73,
-      "learning_rate": 1.5425837320574163e-05,
-      "loss": 0.0279,
-      "step": 4694
-    },
-    {
-      "epoch": 426.82,
-      "learning_rate": 1.54066985645933e-05,
-      "loss": 0.0295,
-      "step": 4695
-    },
-    {
-      "epoch": 426.91,
-      "learning_rate": 1.5387559808612444e-05,
-      "loss": 0.03,
-      "step": 4696
-    },
-    {
-      "epoch": 427.0,
-      "learning_rate": 1.536842105263158e-05,
-      "loss": 0.0309,
-      "step": 4697
-    },
-    {
-      "epoch": 427.09,
-      "learning_rate": 1.5349282296650718e-05,
-      "loss": 0.0286,
-      "step": 4698
-    },
-    {
-      "epoch": 427.18,
-      "learning_rate": 1.533014354066986e-05,
-      "loss": 0.0311,
-      "step": 4699
-    },
-    {
-      "epoch": 427.27,
-      "learning_rate": 1.5311004784688995e-05,
-      "loss": 0.0438,
-      "step": 4700
-    },
-    {
-      "epoch": 427.27,
-      "eval_accuracy_NAT": 0.9931757530757194,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9510011159277222,
-      "eval_accuracy_concrete tiles": 0.985874151531829,
-      "eval_accuracy_dark bricks": 0.8794118779107809,
-      "eval_accuracy_grass": 0.9073840421925297,
-      "eval_accuracy_red bricks": 0.9346590909090909,
-      "eval_accuracy_road": 0.8866681096558955,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5763791249689852,
-      "eval_accuracy_speedway bricks": 0.9880670072300324,
-      "eval_accuracy_steel": 0.9107413758986874,
-      "eval_accuracy_tiles": 0.21394329543898097,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806908024666514,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8941963424308041,
-      "eval_iou_concrete tiles": 0.9567952522255193,
-      "eval_iou_dark bricks": 0.8347826086956521,
-      "eval_iou_grass": 0.8405005098985059,
-      "eval_iou_red bricks": 0.8619616833142296,
-      "eval_iou_road": 0.8792206639871846,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3874321294220114,
-      "eval_iou_speedway bricks": 0.9774214112781919,
-      "eval_iou_steel": 0.8545854028377354,
-      "eval_iou_tiles": 0.21315502183406113,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1298498809337616,
-      "eval_mean_accuracy": 0.7097926880569426,
-      "eval_mean_iou": 0.6677493714146574,
-      "eval_overall_accuracy": 0.9709316889444987,
-      "eval_runtime": 3.6977,
-      "eval_samples_per_second": 9.736,
-      "eval_steps_per_second": 0.541,
-      "step": 4700
-    },
-    {
-      "epoch": 427.36,
-      "learning_rate": 1.5291866028708134e-05,
-      "loss": 0.0286,
-      "step": 4701
-    },
-    {
-      "epoch": 427.45,
-      "learning_rate": 1.5272727272727276e-05,
-      "loss": 0.0316,
-      "step": 4702
-    },
-    {
-      "epoch": 427.55,
-      "learning_rate": 1.5253588516746412e-05,
-      "loss": 0.0302,
-      "step": 4703
-    },
-    {
-      "epoch": 427.64,
-      "learning_rate": 1.523444976076555e-05,
-      "loss": 0.0299,
-      "step": 4704
-    },
-    {
-      "epoch": 427.73,
-      "learning_rate": 1.521531100478469e-05,
-      "loss": 0.0315,
-      "step": 4705
-    },
-    {
-      "epoch": 427.82,
-      "learning_rate": 1.5196172248803828e-05,
-      "loss": 0.0327,
-      "step": 4706
-    },
-    {
-      "epoch": 427.91,
-      "learning_rate": 1.5177033492822967e-05,
-      "loss": 0.0319,
-      "step": 4707
-    },
-    {
-      "epoch": 428.0,
-      "learning_rate": 1.5157894736842107e-05,
-      "loss": 0.0321,
-      "step": 4708
-    },
-    {
-      "epoch": 428.09,
-      "learning_rate": 1.5138755980861244e-05,
-      "loss": 0.0306,
-      "step": 4709
-    },
-    {
-      "epoch": 428.18,
-      "learning_rate": 1.5119617224880383e-05,
-      "loss": 0.0354,
-      "step": 4710
-    },
-    {
-      "epoch": 428.27,
-      "learning_rate": 1.5100478468899523e-05,
-      "loss": 0.0289,
-      "step": 4711
-    },
-    {
-      "epoch": 428.36,
-      "learning_rate": 1.508133971291866e-05,
-      "loss": 0.0301,
-      "step": 4712
-    },
-    {
-      "epoch": 428.45,
-      "learning_rate": 1.50622009569378e-05,
-      "loss": 0.0309,
-      "step": 4713
-    },
-    {
-      "epoch": 428.55,
-      "learning_rate": 1.504306220095694e-05,
-      "loss": 0.0316,
-      "step": 4714
-    },
-    {
-      "epoch": 428.64,
-      "learning_rate": 1.5023923444976077e-05,
-      "loss": 0.0295,
-      "step": 4715
-    },
-    {
-      "epoch": 428.73,
-      "learning_rate": 1.5004784688995216e-05,
-      "loss": 0.0329,
-      "step": 4716
-    },
-    {
-      "epoch": 428.82,
-      "learning_rate": 1.4985645933014356e-05,
-      "loss": 0.0333,
-      "step": 4717
-    },
-    {
-      "epoch": 428.91,
-      "learning_rate": 1.4966507177033493e-05,
-      "loss": 0.028,
-      "step": 4718
-    },
-    {
-      "epoch": 429.0,
-      "learning_rate": 1.4947368421052632e-05,
-      "loss": 0.0313,
-      "step": 4719
-    },
-    {
-      "epoch": 429.09,
-      "learning_rate": 1.4928229665071772e-05,
-      "loss": 0.0309,
-      "step": 4720
-    },
-    {
-      "epoch": 429.09,
-      "eval_accuracy_NAT": 0.9932242771327117,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9471221940855831,
-      "eval_accuracy_concrete tiles": 0.9890845716382315,
-      "eval_accuracy_dark bricks": 0.875832339016974,
-      "eval_accuracy_grass": 0.9055281580612127,
-      "eval_accuracy_red bricks": 0.9431818181818182,
-      "eval_accuracy_road": 0.8721334192677757,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5932787472775894,
-      "eval_accuracy_speedway bricks": 0.9899242964566687,
-      "eval_accuracy_steel": 0.9044835433018931,
-      "eval_accuracy_tiles": 0.36666210108204356,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807154778130862,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8964119099454773,
-      "eval_iou_concrete tiles": 0.9516930952310906,
-      "eval_iou_dark bricks": 0.8327234135987704,
-      "eval_iou_grass": 0.8384991051461379,
-      "eval_iou_red bricks": 0.8679738562091504,
-      "eval_iou_road": 0.865172178431723,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3990209893940518,
-      "eval_iou_speedway bricks": 0.9770780140830223,
-      "eval_iou_steel": 0.8530891558769781,
-      "eval_iou_tiles": 0.36406908744730043,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1460125744342804,
-      "eval_mean_accuracy": 0.7215734973463463,
-      "eval_mean_iou": 0.6789574063982146,
-      "eval_overall_accuracy": 0.9698777728610568,
-      "eval_runtime": 3.5626,
-      "eval_samples_per_second": 10.105,
-      "eval_steps_per_second": 0.561,
-      "step": 4720
-    },
-    {
-      "epoch": 429.18,
-      "learning_rate": 1.4909090909090908e-05,
-      "loss": 0.0337,
-      "step": 4721
-    },
-    {
-      "epoch": 429.27,
-      "learning_rate": 1.4889952153110048e-05,
-      "loss": 0.0287,
-      "step": 4722
-    },
-    {
-      "epoch": 429.36,
-      "learning_rate": 1.4870813397129189e-05,
-      "loss": 0.0353,
-      "step": 4723
-    },
-    {
-      "epoch": 429.45,
-      "learning_rate": 1.4851674641148324e-05,
-      "loss": 0.0302,
-      "step": 4724
-    },
-    {
-      "epoch": 429.55,
-      "learning_rate": 1.4832535885167465e-05,
-      "loss": 0.0251,
-      "step": 4725
-    },
-    {
-      "epoch": 429.64,
-      "learning_rate": 1.4813397129186603e-05,
-      "loss": 0.0295,
-      "step": 4726
-    },
-    {
-      "epoch": 429.73,
-      "learning_rate": 1.479425837320574e-05,
-      "loss": 0.0296,
-      "step": 4727
-    },
-    {
-      "epoch": 429.82,
-      "learning_rate": 1.4775119617224881e-05,
-      "loss": 0.028,
-      "step": 4728
-    },
-    {
-      "epoch": 429.91,
-      "learning_rate": 1.475598086124402e-05,
-      "loss": 0.0324,
-      "step": 4729
-    },
-    {
-      "epoch": 430.0,
-      "learning_rate": 1.4736842105263157e-05,
-      "loss": 0.0317,
-      "step": 4730
-    },
-    {
-      "epoch": 430.09,
-      "learning_rate": 1.4717703349282297e-05,
-      "loss": 0.0309,
-      "step": 4731
-    },
-    {
-      "epoch": 430.18,
-      "learning_rate": 1.4698564593301436e-05,
-      "loss": 0.0293,
-      "step": 4732
-    },
-    {
-      "epoch": 430.27,
-      "learning_rate": 1.4679425837320577e-05,
-      "loss": 0.03,
-      "step": 4733
-    },
-    {
-      "epoch": 430.36,
-      "learning_rate": 1.4660287081339714e-05,
-      "loss": 0.03,
-      "step": 4734
-    },
-    {
-      "epoch": 430.45,
-      "learning_rate": 1.4641148325358852e-05,
-      "loss": 0.0315,
-      "step": 4735
-    },
-    {
-      "epoch": 430.55,
-      "learning_rate": 1.4622009569377993e-05,
-      "loss": 0.0358,
-      "step": 4736
-    },
-    {
-      "epoch": 430.64,
-      "learning_rate": 1.460287081339713e-05,
-      "loss": 0.0303,
-      "step": 4737
-    },
-    {
-      "epoch": 430.73,
-      "learning_rate": 1.4583732057416269e-05,
-      "loss": 0.0282,
-      "step": 4738
-    },
-    {
-      "epoch": 430.82,
-      "learning_rate": 1.456459330143541e-05,
-      "loss": 0.0319,
-      "step": 4739
-    },
-    {
-      "epoch": 430.91,
-      "learning_rate": 1.4545454545454545e-05,
-      "loss": 0.0254,
-      "step": 4740
-    },
-    {
-      "epoch": 430.91,
-      "eval_accuracy_NAT": 0.993475190620141,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9494117236791279,
-      "eval_accuracy_concrete tiles": 0.9853237937993029,
-      "eval_accuracy_dark bricks": 0.8601670451483776,
-      "eval_accuracy_grass": 0.9066046407588504,
-      "eval_accuracy_red bricks": 0.9392755681818182,
-      "eval_accuracy_road": 0.8992015593957342,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.6280704656355968,
-      "eval_accuracy_speedway bricks": 0.9885195630844759,
-      "eval_accuracy_steel": 0.9066766704043269,
-      "eval_accuracy_tiles": 0.43329680865634845,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805212808272282,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.9008018569586956,
-      "eval_iou_concrete tiles": 0.9566869526495473,
-      "eval_iou_dark bricks": 0.8289317507418398,
-      "eval_iou_grass": 0.8409029020462149,
-      "eval_iou_red bricks": 0.8683519369665135,
-      "eval_iou_road": 0.8919301115830204,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.4233942908117752,
-      "eval_iou_speedway bricks": 0.9773860292626118,
-      "eval_iou_steel": 0.8540087599403579,
-      "eval_iou_tiles": 0.4307891332470893,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11915155500173569,
-      "eval_mean_accuracy": 0.7300017714895461,
-      "eval_mean_iou": 0.6887465388488381,
-      "eval_overall_accuracy": 0.9725738101535373,
-      "eval_runtime": 3.8035,
-      "eval_samples_per_second": 9.465,
-      "eval_steps_per_second": 0.526,
-      "step": 4740
-    },
-    {
-      "epoch": 431.0,
-      "learning_rate": 1.4526315789473685e-05,
-      "loss": 0.0299,
-      "step": 4741
-    },
-    {
-      "epoch": 431.09,
-      "learning_rate": 1.4507177033492826e-05,
-      "loss": 0.0408,
-      "step": 4742
-    },
-    {
-      "epoch": 431.18,
-      "learning_rate": 1.4488038277511961e-05,
-      "loss": 0.0469,
-      "step": 4743
-    },
-    {
-      "epoch": 431.27,
-      "learning_rate": 1.4468899521531101e-05,
-      "loss": 0.0275,
-      "step": 4744
-    },
-    {
-      "epoch": 431.36,
-      "learning_rate": 1.444976076555024e-05,
-      "loss": 0.0281,
-      "step": 4745
-    },
-    {
-      "epoch": 431.45,
-      "learning_rate": 1.4430622009569377e-05,
-      "loss": 0.0281,
-      "step": 4746
-    },
-    {
-      "epoch": 431.55,
-      "learning_rate": 1.4411483253588518e-05,
-      "loss": 0.0307,
-      "step": 4747
-    },
-    {
-      "epoch": 431.64,
-      "learning_rate": 1.4392344497607657e-05,
-      "loss": 0.0287,
-      "step": 4748
-    },
-    {
-      "epoch": 431.73,
-      "learning_rate": 1.4373205741626794e-05,
-      "loss": 0.0301,
-      "step": 4749
-    },
-    {
-      "epoch": 431.82,
-      "learning_rate": 1.4354066985645934e-05,
-      "loss": 0.0332,
-      "step": 4750
-    },
-    {
-      "epoch": 431.91,
-      "learning_rate": 1.4334928229665073e-05,
-      "loss": 0.033,
-      "step": 4751
-    },
-    {
-      "epoch": 432.0,
-      "learning_rate": 1.431578947368421e-05,
-      "loss": 0.0289,
-      "step": 4752
-    },
-    {
-      "epoch": 432.09,
-      "learning_rate": 1.429665071770335e-05,
-      "loss": 0.0322,
-      "step": 4753
-    },
-    {
-      "epoch": 432.18,
-      "learning_rate": 1.427751196172249e-05,
-      "loss": 0.0283,
-      "step": 4754
-    },
-    {
-      "epoch": 432.27,
-      "learning_rate": 1.4258373205741626e-05,
-      "loss": 0.0332,
-      "step": 4755
-    },
-    {
-      "epoch": 432.36,
-      "learning_rate": 1.4239234449760767e-05,
-      "loss": 0.025,
-      "step": 4756
-    },
-    {
-      "epoch": 432.45,
-      "learning_rate": 1.4220095693779906e-05,
-      "loss": 0.032,
-      "step": 4757
-    },
-    {
-      "epoch": 432.55,
-      "learning_rate": 1.4200956937799043e-05,
-      "loss": 0.032,
-      "step": 4758
-    },
-    {
-      "epoch": 432.64,
-      "learning_rate": 1.4181818181818181e-05,
-      "loss": 0.0308,
-      "step": 4759
-    },
-    {
-      "epoch": 432.73,
-      "learning_rate": 1.4162679425837322e-05,
-      "loss": 0.0291,
-      "step": 4760
-    },
-    {
-      "epoch": 432.73,
-      "eval_accuracy_NAT": 0.9932131607123825,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.949844950427057,
-      "eval_accuracy_concrete tiles": 0.9828471840029352,
-      "eval_accuracy_dark bricks": 0.8714830068126708,
-      "eval_accuracy_grass": 0.9028404463908122,
-      "eval_accuracy_red bricks": 0.9334161931818182,
-      "eval_accuracy_road": 0.9075378452140118,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5480109172111488,
-      "eval_accuracy_speedway bricks": 0.9880796787939568,
-      "eval_accuracy_steel": 0.9112772904161994,
-      "eval_accuracy_tiles": 0.31064237775647174,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805208537288099,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8940484637001987,
-      "eval_iou_concrete tiles": 0.958465024748047,
-      "eval_iou_dark bricks": 0.8298028292897457,
-      "eval_iou_grass": 0.8362972387603073,
-      "eval_iou_red bricks": 0.8629349967170059,
-      "eval_iou_road": 0.8994179420530575,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.36966507354992284,
-      "eval_iou_speedway bricks": 0.9772292029087133,
-      "eval_iou_steel": 0.8530885065721938,
-      "eval_iou_tiles": 0.308844556410431,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11700248718261719,
-      "eval_mean_accuracy": 0.7153225423784204,
-      "eval_mean_iou": 0.6746395914183411,
-      "eval_overall_accuracy": 0.9725705252753364,
-      "eval_runtime": 3.3921,
-      "eval_samples_per_second": 10.613,
-      "eval_steps_per_second": 0.59,
-      "step": 4760
-    },
-    {
-      "epoch": 432.82,
-      "learning_rate": 1.4143540669856459e-05,
-      "loss": 0.0295,
-      "step": 4761
-    },
-    {
-      "epoch": 432.91,
-      "learning_rate": 1.4124401913875598e-05,
-      "loss": 0.0326,
-      "step": 4762
-    },
-    {
-      "epoch": 433.0,
-      "learning_rate": 1.4105263157894738e-05,
-      "loss": 0.0279,
-      "step": 4763
-    },
-    {
-      "epoch": 433.09,
-      "learning_rate": 1.4086124401913875e-05,
-      "loss": 0.0296,
-      "step": 4764
-    },
-    {
-      "epoch": 433.18,
-      "learning_rate": 1.4066985645933014e-05,
-      "loss": 0.0331,
-      "step": 4765
-    },
-    {
-      "epoch": 433.27,
-      "learning_rate": 1.4047846889952155e-05,
-      "loss": 0.0281,
-      "step": 4766
-    },
-    {
-      "epoch": 433.36,
-      "learning_rate": 1.4028708133971293e-05,
-      "loss": 0.0316,
-      "step": 4767
-    },
-    {
-      "epoch": 433.45,
-      "learning_rate": 1.400956937799043e-05,
-      "loss": 0.0278,
-      "step": 4768
-    },
-    {
-      "epoch": 433.55,
-      "learning_rate": 1.3990430622009571e-05,
-      "loss": 0.0272,
-      "step": 4769
-    },
-    {
-      "epoch": 433.64,
-      "learning_rate": 1.397129186602871e-05,
-      "loss": 0.0301,
-      "step": 4770
-    },
-    {
-      "epoch": 433.73,
-      "learning_rate": 1.3952153110047847e-05,
-      "loss": 0.0268,
-      "step": 4771
-    },
-    {
-      "epoch": 433.82,
-      "learning_rate": 1.3933014354066987e-05,
-      "loss": 0.0318,
-      "step": 4772
-    },
-    {
-      "epoch": 433.91,
-      "learning_rate": 1.3913875598086126e-05,
-      "loss": 0.0282,
-      "step": 4773
-    },
-    {
-      "epoch": 434.0,
-      "learning_rate": 1.3894736842105263e-05,
-      "loss": 0.0343,
-      "step": 4774
-    },
-    {
-      "epoch": 434.09,
-      "learning_rate": 1.3875598086124404e-05,
-      "loss": 0.0275,
-      "step": 4775
-    },
-    {
-      "epoch": 434.18,
-      "learning_rate": 1.3856459330143542e-05,
-      "loss": 0.0274,
-      "step": 4776
-    },
-    {
-      "epoch": 434.27,
-      "learning_rate": 1.383732057416268e-05,
-      "loss": 0.03,
-      "step": 4777
-    },
-    {
-      "epoch": 434.36,
-      "learning_rate": 1.3818181818181818e-05,
-      "loss": 0.0337,
-      "step": 4778
-    },
-    {
-      "epoch": 434.45,
-      "learning_rate": 1.3799043062200959e-05,
-      "loss": 0.0275,
-      "step": 4779
-    },
-    {
-      "epoch": 434.55,
-      "learning_rate": 1.3779904306220096e-05,
-      "loss": 0.0323,
-      "step": 4780
-    },
-    {
-      "epoch": 434.55,
-      "eval_accuracy_NAT": 0.9931868694960485,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.950231233100133,
-      "eval_accuracy_concrete tiles": 0.9884730630465358,
-      "eval_accuracy_dark bricks": 0.8741387937338825,
-      "eval_accuracy_grass": 0.899743811098257,
-      "eval_accuracy_red bricks": 0.94140625,
-      "eval_accuracy_road": 0.916721786436788,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5754693573732529,
-      "eval_accuracy_speedway bricks": 0.9892599444623456,
-      "eval_accuracy_steel": 0.9055223929819932,
-      "eval_accuracy_tiles": 0.32961238186549785,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807945706649605,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8948797752638679,
-      "eval_iou_concrete tiles": 0.9514685973276826,
-      "eval_iou_dark bricks": 0.831843820965497,
-      "eval_iou_grass": 0.8364400804494251,
-      "eval_iou_red bricks": 0.8628152969894223,
-      "eval_iou_road": 0.9091418210250372,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.38552748226950356,
-      "eval_iou_speedway bricks": 0.9771578974346442,
-      "eval_iou_steel": 0.8525176784730146,
-      "eval_iou_tiles": 0.32668159913120204,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11128445714712143,
-      "eval_mean_accuracy": 0.7202896833534412,
-      "eval_mean_iou": 0.6776360476918659,
-      "eval_overall_accuracy": 0.973595937093099,
-      "eval_runtime": 3.6858,
-      "eval_samples_per_second": 9.767,
-      "eval_steps_per_second": 0.543,
-      "step": 4780
-    },
-    {
-      "epoch": 434.64,
-      "learning_rate": 1.3760765550239235e-05,
-      "loss": 0.0322,
-      "step": 4781
-    },
-    {
-      "epoch": 434.73,
-      "learning_rate": 1.3741626794258375e-05,
-      "loss": 0.0322,
-      "step": 4782
-    },
-    {
-      "epoch": 434.82,
-      "learning_rate": 1.3722488038277512e-05,
-      "loss": 0.0299,
-      "step": 4783
-    },
-    {
-      "epoch": 434.91,
-      "learning_rate": 1.370334928229665e-05,
-      "loss": 0.0288,
-      "step": 4784
-    },
-    {
-      "epoch": 435.0,
-      "learning_rate": 1.3684210526315791e-05,
-      "loss": 0.0304,
-      "step": 4785
-    },
-    {
-      "epoch": 435.09,
-      "learning_rate": 1.3665071770334928e-05,
-      "loss": 0.0275,
-      "step": 4786
-    },
-    {
-      "epoch": 435.18,
-      "learning_rate": 1.3645933014354067e-05,
-      "loss": 0.0317,
-      "step": 4787
-    },
-    {
-      "epoch": 435.27,
-      "learning_rate": 1.3626794258373208e-05,
-      "loss": 0.029,
-      "step": 4788
-    },
-    {
-      "epoch": 435.36,
-      "learning_rate": 1.3607655502392345e-05,
-      "loss": 0.0338,
-      "step": 4789
-    },
-    {
-      "epoch": 435.45,
-      "learning_rate": 1.3588516746411483e-05,
-      "loss": 0.0327,
-      "step": 4790
-    },
-    {
-      "epoch": 435.55,
-      "learning_rate": 1.3569377990430624e-05,
-      "loss": 0.028,
-      "step": 4791
-    },
-    {
-      "epoch": 435.64,
-      "learning_rate": 1.3550239234449761e-05,
-      "loss": 0.0346,
-      "step": 4792
-    },
-    {
-      "epoch": 435.73,
-      "learning_rate": 1.35311004784689e-05,
-      "loss": 0.0339,
-      "step": 4793
-    },
-    {
-      "epoch": 435.82,
-      "learning_rate": 1.351196172248804e-05,
-      "loss": 0.0249,
-      "step": 4794
-    },
-    {
-      "epoch": 435.91,
-      "learning_rate": 1.3492822966507176e-05,
-      "loss": 0.0316,
-      "step": 4795
-    },
-    {
-      "epoch": 436.0,
-      "learning_rate": 1.3473684210526316e-05,
-      "loss": 0.0297,
-      "step": 4796
-    },
-    {
-      "epoch": 436.09,
-      "learning_rate": 1.3454545454545457e-05,
-      "loss": 0.0294,
-      "step": 4797
-    },
-    {
-      "epoch": 436.18,
-      "learning_rate": 1.3435406698564592e-05,
-      "loss": 0.0294,
-      "step": 4798
-    },
-    {
-      "epoch": 436.27,
-      "learning_rate": 1.3416267942583732e-05,
-      "loss": 0.0285,
-      "step": 4799
-    },
-    {
-      "epoch": 436.36,
-      "learning_rate": 1.3397129186602871e-05,
-      "loss": 0.0339,
-      "step": 4800
-    },
-    {
-      "epoch": 436.36,
-      "eval_accuracy_NAT": 0.9934878951005172,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9509662431864029,
-      "eval_accuracy_concrete tiles": 0.9863939338347704,
-      "eval_accuracy_dark bricks": 0.8759478080135483,
-      "eval_accuracy_grass": 0.8929144371009062,
-      "eval_accuracy_red bricks": 0.923828125,
-      "eval_accuracy_road": 0.9216372204157985,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.43382129958922616,
-      "eval_accuracy_speedway bricks": 0.9880012357791865,
-      "eval_accuracy_steel": 0.9104033375107183,
-      "eval_accuracy_tiles": 0.2570195863580331,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805864166569137,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8868533660551342,
-      "eval_iou_concrete tiles": 0.954410981598722,
-      "eval_iou_dark bricks": 0.8307052124397722,
-      "eval_iou_grass": 0.829643725827036,
-      "eval_iou_red bricks": 0.8651479880279348,
-      "eval_iou_road": 0.9142534300307318,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.29582283716208596,
-      "eval_iou_speedway bricks": 0.9771726150157046,
-      "eval_iou_steel": 0.8534033032174296,
-      "eval_iou_tiles": 0.25631744297227155,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11220533400774002,
-      "eval_mean_accuracy": 0.7026477786068543,
-      "eval_mean_iou": 0.6649474860772103,
-      "eval_overall_accuracy": 0.9732306798299154,
-      "eval_runtime": 3.546,
-      "eval_samples_per_second": 10.152,
-      "eval_steps_per_second": 0.564,
-      "step": 4800
-    },
-    {
-      "epoch": 436.45,
-      "learning_rate": 1.3377990430622008e-05,
-      "loss": 0.0274,
-      "step": 4801
-    },
-    {
-      "epoch": 436.55,
-      "learning_rate": 1.3358851674641149e-05,
-      "loss": 0.0301,
-      "step": 4802
-    },
-    {
-      "epoch": 436.64,
-      "learning_rate": 1.3339712918660288e-05,
-      "loss": 0.0281,
-      "step": 4803
-    },
-    {
-      "epoch": 436.73,
-      "learning_rate": 1.3320574162679428e-05,
-      "loss": 0.0327,
-      "step": 4804
-    },
-    {
-      "epoch": 436.82,
-      "learning_rate": 1.3301435406698565e-05,
-      "loss": 0.0306,
-      "step": 4805
-    },
-    {
-      "epoch": 436.91,
-      "learning_rate": 1.3282296650717704e-05,
-      "loss": 0.0324,
-      "step": 4806
-    },
-    {
-      "epoch": 437.0,
-      "learning_rate": 1.3263157894736844e-05,
-      "loss": 0.0304,
-      "step": 4807
-    },
-    {
-      "epoch": 437.09,
-      "learning_rate": 1.3244019138755981e-05,
-      "loss": 0.0284,
-      "step": 4808
-    },
-    {
-      "epoch": 437.18,
-      "learning_rate": 1.322488038277512e-05,
-      "loss": 0.0291,
-      "step": 4809
-    },
-    {
-      "epoch": 437.27,
-      "learning_rate": 1.320574162679426e-05,
-      "loss": 0.0297,
-      "step": 4810
-    },
-    {
-      "epoch": 437.36,
-      "learning_rate": 1.3186602870813398e-05,
-      "loss": 0.0299,
-      "step": 4811
-    },
-    {
-      "epoch": 437.45,
-      "learning_rate": 1.3167464114832537e-05,
-      "loss": 0.0273,
-      "step": 4812
-    },
-    {
-      "epoch": 437.55,
-      "learning_rate": 1.3148325358851677e-05,
-      "loss": 0.0289,
-      "step": 4813
-    },
-    {
-      "epoch": 437.64,
-      "learning_rate": 1.3129186602870812e-05,
-      "loss": 0.0299,
-      "step": 4814
-    },
-    {
-      "epoch": 437.73,
-      "learning_rate": 1.3110047846889953e-05,
-      "loss": 0.034,
-      "step": 4815
-    },
-    {
-      "epoch": 437.82,
-      "learning_rate": 1.3090909090909093e-05,
-      "loss": 0.0309,
-      "step": 4816
-    },
-    {
-      "epoch": 437.91,
-      "learning_rate": 1.3071770334928229e-05,
-      "loss": 0.0321,
-      "step": 4817
-    },
-    {
-      "epoch": 438.0,
-      "learning_rate": 1.305263157894737e-05,
-      "loss": 0.0274,
-      "step": 4818
-    },
-    {
-      "epoch": 438.09,
-      "learning_rate": 1.3033492822966508e-05,
-      "loss": 0.0291,
-      "step": 4819
-    },
-    {
-      "epoch": 438.18,
-      "learning_rate": 1.3014354066985645e-05,
-      "loss": 0.03,
-      "step": 4820
-    },
-    {
-      "epoch": 438.18,
-      "eval_accuracy_NAT": 0.9933574977255452,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9493486844928967,
-      "eval_accuracy_concrete tiles": 0.9876781018773314,
-      "eval_accuracy_dark bricks": 0.8782571879450367,
-      "eval_accuracy_grass": 0.8991251830544847,
-      "eval_accuracy_red bricks": 0.9284446022727273,
-      "eval_accuracy_road": 0.922324051043124,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.46607669616519176,
-      "eval_accuracy_speedway bricks": 0.9886257628583186,
-      "eval_accuracy_steel": 0.9095211397665062,
-      "eval_accuracy_tiles": 0.2963292699630188,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805195302825362,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8899825349144538,
-      "eval_iou_concrete tiles": 0.9499206022466623,
-      "eval_iou_dark bricks": 0.8316810030616708,
-      "eval_iou_grass": 0.8352326593161128,
-      "eval_iou_red bricks": 0.8638691557905172,
-      "eval_iou_road": 0.9148338006951657,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.31525752433521054,
-      "eval_iou_speedway bricks": 0.97729628874072,
-      "eval_iou_steel": 0.8528202115158637,
-      "eval_iou_tiles": 0.29511662801800576,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1112009584903717,
-      "eval_mean_accuracy": 0.7091606290126293,
-      "eval_mean_iou": 0.6697330722243784,
-      "eval_overall_accuracy": 0.9735697640313042,
-      "eval_runtime": 4.031,
-      "eval_samples_per_second": 8.931,
-      "eval_steps_per_second": 0.496,
-      "step": 4820
-    },
-    {
-      "epoch": 438.27,
-      "learning_rate": 1.2995215311004786e-05,
-      "loss": 0.029,
-      "step": 4821
-    },
-    {
-      "epoch": 438.36,
-      "learning_rate": 1.2976076555023924e-05,
-      "loss": 0.0293,
-      "step": 4822
-    },
-    {
-      "epoch": 438.45,
-      "learning_rate": 1.2956937799043061e-05,
-      "loss": 0.0261,
-      "step": 4823
-    },
-    {
-      "epoch": 438.55,
-      "learning_rate": 1.2937799043062202e-05,
-      "loss": 0.0297,
-      "step": 4824
-    },
-    {
-      "epoch": 438.64,
-      "learning_rate": 1.291866028708134e-05,
-      "loss": 0.0299,
-      "step": 4825
-    },
-    {
-      "epoch": 438.73,
-      "learning_rate": 1.2899521531100478e-05,
-      "loss": 0.0286,
-      "step": 4826
-    },
-    {
-      "epoch": 438.82,
-      "learning_rate": 1.2880382775119618e-05,
-      "loss": 0.0288,
-      "step": 4827
-    },
-    {
-      "epoch": 438.91,
-      "learning_rate": 1.2861244019138757e-05,
-      "loss": 0.0316,
-      "step": 4828
-    },
-    {
-      "epoch": 439.0,
-      "learning_rate": 1.2842105263157894e-05,
-      "loss": 0.0362,
-      "step": 4829
-    },
-    {
-      "epoch": 439.09,
-      "learning_rate": 1.2822966507177035e-05,
-      "loss": 0.0294,
-      "step": 4830
-    },
-    {
-      "epoch": 439.18,
-      "learning_rate": 1.2803827751196173e-05,
-      "loss": 0.0314,
-      "step": 4831
-    },
-    {
-      "epoch": 439.27,
-      "learning_rate": 1.278468899521531e-05,
-      "loss": 0.0303,
-      "step": 4832
-    },
-    {
-      "epoch": 439.36,
-      "learning_rate": 1.276555023923445e-05,
-      "loss": 0.0292,
-      "step": 4833
-    },
-    {
-      "epoch": 439.45,
-      "learning_rate": 1.274641148325359e-05,
-      "loss": 0.0262,
-      "step": 4834
-    },
-    {
-      "epoch": 439.55,
-      "learning_rate": 1.2727272727272727e-05,
-      "loss": 0.0281,
-      "step": 4835
-    },
-    {
-      "epoch": 439.64,
-      "learning_rate": 1.2708133971291866e-05,
-      "loss": 0.0307,
-      "step": 4836
-    },
-    {
-      "epoch": 439.73,
-      "learning_rate": 1.2688995215311006e-05,
-      "loss": 0.0299,
-      "step": 4837
-    },
-    {
-      "epoch": 439.82,
-      "learning_rate": 1.2669856459330145e-05,
-      "loss": 0.0331,
-      "step": 4838
-    },
-    {
-      "epoch": 439.91,
-      "learning_rate": 1.2650717703349282e-05,
-      "loss": 0.0316,
-      "step": 4839
-    },
-    {
-      "epoch": 440.0,
-      "learning_rate": 1.2631578947368422e-05,
-      "loss": 0.0296,
-      "step": 4840
-    },
-    {
-      "epoch": 440.0,
-      "eval_accuracy_NAT": 0.9930215348000421,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9513149705995966,
-      "eval_accuracy_concrete tiles": 0.987739252736501,
-      "eval_accuracy_dark bricks": 0.8774873946345406,
-      "eval_accuracy_grass": 0.8968673654484005,
-      "eval_accuracy_red bricks": 0.9185014204545454,
-      "eval_accuracy_road": 0.9200700857449996,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48454773523006095,
-      "eval_accuracy_speedway bricks": 0.9887373933024147,
-      "eval_accuracy_steel": 0.9088862871842227,
-      "eval_accuracy_tiles": 0.2524996575811533,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9805000848481428,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8887237697615643,
-      "eval_iou_concrete tiles": 0.9526126444916254,
-      "eval_iou_dark bricks": 0.8304374749571996,
-      "eval_iou_grass": 0.8308262643268794,
-      "eval_iou_red bricks": 0.8601596275357499,
-      "eval_iou_road": 0.912913482188921,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3264669279491799,
-      "eval_iou_speedway bricks": 0.9773943967355547,
-      "eval_iou_steel": 0.854378187342086,
-      "eval_iou_tiles": 0.2515178388703186,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11227044463157654,
-      "eval_mean_accuracy": 0.7061286998243445,
-      "eval_mean_iou": 0.6666100537697862,
-      "eval_overall_accuracy": 0.9732641643948026,
-      "eval_runtime": 3.3843,
-      "eval_samples_per_second": 10.637,
-      "eval_steps_per_second": 0.591,
-      "step": 4840
-    },
-    {
-      "epoch": 440.09,
-      "learning_rate": 1.2612440191387561e-05,
-      "loss": 0.0323,
-      "step": 4841
-    },
-    {
-      "epoch": 440.18,
-      "learning_rate": 1.2593301435406698e-05,
-      "loss": 0.0294,
-      "step": 4842
-    },
-    {
-      "epoch": 440.27,
-      "learning_rate": 1.2574162679425839e-05,
-      "loss": 0.0289,
-      "step": 4843
-    },
-    {
-      "epoch": 440.36,
-      "learning_rate": 1.2555023923444977e-05,
-      "loss": 0.0323,
-      "step": 4844
-    },
-    {
-      "epoch": 440.45,
-      "learning_rate": 1.2535885167464115e-05,
-      "loss": 0.0318,
-      "step": 4845
-    },
-    {
-      "epoch": 440.55,
-      "learning_rate": 1.2516746411483255e-05,
-      "loss": 0.0286,
-      "step": 4846
-    },
-    {
-      "epoch": 440.64,
-      "learning_rate": 1.2497607655502392e-05,
-      "loss": 0.0433,
-      "step": 4847
-    },
-    {
-      "epoch": 440.73,
-      "learning_rate": 1.2478468899521533e-05,
-      "loss": 0.0284,
-      "step": 4848
-    },
-    {
-      "epoch": 440.82,
-      "learning_rate": 1.2459330143540671e-05,
-      "loss": 0.0304,
-      "step": 4849
-    },
-    {
-      "epoch": 440.91,
-      "learning_rate": 1.2440191387559808e-05,
-      "loss": 0.0322,
-      "step": 4850
-    },
-    {
-      "epoch": 441.0,
-      "learning_rate": 1.2421052631578949e-05,
-      "loss": 0.0264,
-      "step": 4851
-    },
-    {
-      "epoch": 441.09,
-      "learning_rate": 1.2401913875598086e-05,
-      "loss": 0.028,
-      "step": 4852
-    },
-    {
-      "epoch": 441.18,
-      "learning_rate": 1.2382775119617225e-05,
-      "loss": 0.0312,
-      "step": 4853
-    },
-    {
-      "epoch": 441.27,
-      "learning_rate": 1.2363636363636365e-05,
-      "loss": 0.0289,
-      "step": 4854
-    },
-    {
-      "epoch": 441.36,
-      "learning_rate": 1.2344497607655502e-05,
-      "loss": 0.027,
-      "step": 4855
-    },
-    {
-      "epoch": 441.45,
-      "learning_rate": 1.2325358851674641e-05,
-      "loss": 0.031,
-      "step": 4856
-    },
-    {
-      "epoch": 441.55,
-      "learning_rate": 1.2306220095693782e-05,
-      "loss": 0.0306,
-      "step": 4857
-    },
-    {
-      "epoch": 441.64,
-      "learning_rate": 1.2287081339712919e-05,
-      "loss": 0.026,
-      "step": 4858
-    },
-    {
-      "epoch": 441.73,
-      "learning_rate": 1.2267942583732057e-05,
-      "loss": 0.0297,
-      "step": 4859
-    },
-    {
-      "epoch": 441.82,
-      "learning_rate": 1.2248803827751198e-05,
-      "loss": 0.0304,
-      "step": 4860
-    },
-    {
-      "epoch": 441.82,
-      "eval_accuracy_NAT": 0.9928404959546817,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9523920018026525,
-      "eval_accuracy_concrete tiles": 0.9884730630465358,
-      "eval_accuracy_dark bricks": 0.876024787344598,
-      "eval_accuracy_grass": 0.9014703775029097,
-      "eval_accuracy_red bricks": 0.9307528409090909,
-      "eval_accuracy_road": 0.9244462125764976,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5139635541587406,
-      "eval_accuracy_speedway bricks": 0.9875854878009044,
-      "eval_accuracy_steel": 0.910090033638942,
-      "eval_accuracy_tiles": 0.2457882481851801,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808025310056913,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8888044680989551,
-      "eval_iou_concrete tiles": 0.9510487453299209,
-      "eval_iou_dark bricks": 0.8286910613508102,
-      "eval_iou_grass": 0.8325081176690832,
-      "eval_iou_red bricks": 0.8606140206862584,
-      "eval_iou_road": 0.9161289162906908,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3432385160637025,
-      "eval_iou_speedway bricks": 0.9775052199548002,
-      "eval_iou_steel": 0.8535120004948658,
-      "eval_iou_tiles": 0.24506657562307954,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1113537847995758,
-      "eval_mean_accuracy": 0.7095251617631333,
-      "eval_mean_iou": 0.6675323209667583,
-      "eval_overall_accuracy": 0.9736852645874023,
-      "eval_runtime": 3.4315,
-      "eval_samples_per_second": 10.491,
-      "eval_steps_per_second": 0.583,
-      "step": 4860
-    },
-    {
-      "epoch": 441.91,
-      "learning_rate": 1.2229665071770335e-05,
-      "loss": 0.0297,
-      "step": 4861
-    },
-    {
-      "epoch": 442.0,
-      "learning_rate": 1.2210526315789474e-05,
-      "loss": 0.0346,
-      "step": 4862
-    },
-    {
-      "epoch": 442.09,
-      "learning_rate": 1.2191387559808613e-05,
-      "loss": 0.0281,
-      "step": 4863
-    },
-    {
-      "epoch": 442.18,
-      "learning_rate": 1.2172248803827751e-05,
-      "loss": 0.0283,
-      "step": 4864
-    },
-    {
-      "epoch": 442.27,
-      "learning_rate": 1.2153110047846892e-05,
-      "loss": 0.0304,
-      "step": 4865
-    },
-    {
-      "epoch": 442.36,
-      "learning_rate": 1.2133971291866029e-05,
-      "loss": 0.0319,
-      "step": 4866
-    },
-    {
-      "epoch": 442.45,
-      "learning_rate": 1.2114832535885168e-05,
-      "loss": 0.0297,
-      "step": 4867
-    },
-    {
-      "epoch": 442.55,
-      "learning_rate": 1.2095693779904308e-05,
-      "loss": 0.0311,
-      "step": 4868
-    },
-    {
-      "epoch": 442.64,
-      "learning_rate": 1.2076555023923445e-05,
-      "loss": 0.0282,
-      "step": 4869
-    },
-    {
-      "epoch": 442.73,
-      "learning_rate": 1.2057416267942584e-05,
-      "loss": 0.0327,
-      "step": 4870
-    },
-    {
-      "epoch": 442.82,
-      "learning_rate": 1.2038277511961724e-05,
-      "loss": 0.0289,
-      "step": 4871
-    },
-    {
-      "epoch": 442.91,
-      "learning_rate": 1.2019138755980862e-05,
-      "loss": 0.0302,
-      "step": 4872
-    },
-    {
-      "epoch": 443.0,
-      "learning_rate": 1.2e-05,
-      "loss": 0.0253,
-      "step": 4873
-    },
-    {
-      "epoch": 443.09,
-      "learning_rate": 1.1980861244019139e-05,
-      "loss": 0.0307,
-      "step": 4874
-    },
-    {
-      "epoch": 443.18,
-      "learning_rate": 1.1961722488038278e-05,
-      "loss": 0.0272,
-      "step": 4875
-    },
-    {
-      "epoch": 443.27,
-      "learning_rate": 1.1942583732057417e-05,
-      "loss": 0.0322,
-      "step": 4876
-    },
-    {
-      "epoch": 443.36,
-      "learning_rate": 1.1923444976076555e-05,
-      "loss": 0.0285,
-      "step": 4877
-    },
-    {
-      "epoch": 443.45,
-      "learning_rate": 1.1904306220095694e-05,
-      "loss": 0.0311,
-      "step": 4878
-    },
-    {
-      "epoch": 443.55,
-      "learning_rate": 1.1885167464114833e-05,
-      "loss": 0.0303,
-      "step": 4879
-    },
-    {
-      "epoch": 443.64,
-      "learning_rate": 1.1866028708133972e-05,
-      "loss": 0.0346,
-      "step": 4880
-    },
-    {
-      "epoch": 443.64,
-      "eval_accuracy_NAT": 0.992863258148689,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9517522211253702,
-      "eval_accuracy_concrete tiles": 0.9892680242157402,
-      "eval_accuracy_dark bricks": 0.8779492706208383,
-      "eval_accuracy_grass": 0.9035359660558443,
-      "eval_accuracy_red bricks": 0.9327059659090909,
-      "eval_accuracy_road": 0.9250508170019601,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.49642985140462603,
-      "eval_accuracy_speedway bricks": 0.9887500648663391,
-      "eval_accuracy_steel": 0.9084328210540202,
-      "eval_accuracy_tiles": 0.2571565538967265,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9809175203079192,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8892481324556094,
-      "eval_iou_concrete tiles": 0.9486600598135225,
-      "eval_iou_dark bricks": 0.8321173208813658,
-      "eval_iou_grass": 0.8343860645711021,
-      "eval_iou_red bricks": 0.8594568062827225,
-      "eval_iou_road": 0.917690314531801,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33380294744647326,
-      "eval_iou_speedway bricks": 0.9774145021038203,
-      "eval_iou_steel": 0.8545877608004343,
-      "eval_iou_tiles": 0.25584247462015397,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1097680851817131,
-      "eval_mean_accuracy": 0.7095303703307111,
-      "eval_mean_iou": 0.6680095310626866,
-      "eval_overall_accuracy": 0.9739066229926215,
-      "eval_runtime": 3.423,
-      "eval_samples_per_second": 10.517,
-      "eval_steps_per_second": 0.584,
-      "step": 4880
-    },
-    {
-      "epoch": 443.73,
-      "learning_rate": 1.184688995215311e-05,
-      "loss": 0.0288,
-      "step": 4881
-    },
-    {
-      "epoch": 443.82,
-      "learning_rate": 1.182775119617225e-05,
-      "loss": 0.0308,
-      "step": 4882
-    },
-    {
-      "epoch": 443.91,
-      "learning_rate": 1.1808612440191388e-05,
-      "loss": 0.0285,
-      "step": 4883
-    },
-    {
-      "epoch": 444.0,
-      "learning_rate": 1.1789473684210527e-05,
-      "loss": 0.0299,
-      "step": 4884
-    },
-    {
-      "epoch": 444.09,
-      "learning_rate": 1.1770334928229666e-05,
-      "loss": 0.0323,
-      "step": 4885
-    },
-    {
-      "epoch": 444.18,
-      "learning_rate": 1.1751196172248804e-05,
-      "loss": 0.0257,
-      "step": 4886
-    },
-    {
-      "epoch": 444.27,
-      "learning_rate": 1.1732057416267943e-05,
-      "loss": 0.0306,
-      "step": 4887
-    },
-    {
-      "epoch": 444.36,
-      "learning_rate": 1.1712918660287082e-05,
-      "loss": 0.0306,
-      "step": 4888
-    },
-    {
-      "epoch": 444.45,
-      "learning_rate": 1.169377990430622e-05,
-      "loss": 0.028,
-      "step": 4889
-    },
-    {
-      "epoch": 444.55,
-      "learning_rate": 1.167464114832536e-05,
-      "loss": 0.0292,
-      "step": 4890
-    },
-    {
-      "epoch": 444.64,
-      "learning_rate": 1.1655502392344498e-05,
-      "loss": 0.025,
-      "step": 4891
-    },
-    {
-      "epoch": 444.73,
-      "learning_rate": 1.1636363636363637e-05,
-      "loss": 0.0314,
-      "step": 4892
-    },
-    {
-      "epoch": 444.82,
-      "learning_rate": 1.1617224880382776e-05,
-      "loss": 0.033,
-      "step": 4893
-    },
-    {
-      "epoch": 444.91,
-      "learning_rate": 1.1598086124401915e-05,
-      "loss": 0.0294,
-      "step": 4894
-    },
-    {
-      "epoch": 445.0,
-      "learning_rate": 1.1578947368421053e-05,
-      "loss": 0.0277,
-      "step": 4895
-    },
-    {
-      "epoch": 445.09,
-      "learning_rate": 1.1559808612440192e-05,
-      "loss": 0.0287,
-      "step": 4896
-    },
-    {
-      "epoch": 445.18,
-      "learning_rate": 1.1540669856459331e-05,
-      "loss": 0.0265,
-      "step": 4897
-    },
-    {
-      "epoch": 445.27,
-      "learning_rate": 1.152153110047847e-05,
-      "loss": 0.0295,
-      "step": 4898
-    },
-    {
-      "epoch": 445.36,
-      "learning_rate": 1.1502392344497607e-05,
-      "loss": 0.0369,
-      "step": 4899
-    },
-    {
-      "epoch": 445.45,
-      "learning_rate": 1.1483253588516747e-05,
-      "loss": 0.0283,
-      "step": 4900
-    },
-    {
-      "epoch": 445.45,
-      "eval_accuracy_NAT": 0.9930799401195491,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.949193098416241,
-      "eval_accuracy_concrete tiles": 0.9894209013636641,
-      "eval_accuracy_dark bricks": 0.8812978715214965,
-      "eval_accuracy_grass": 0.9031759734654006,
-      "eval_accuracy_red bricks": 0.9348366477272727,
-      "eval_accuracy_road": 0.9236578084056944,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4617208391916853,
-      "eval_accuracy_speedway bricks": 0.9886993786106415,
-      "eval_accuracy_steel": 0.90791339621397,
-      "eval_accuracy_tiles": 0.26736063552937955,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806139629019868,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8884592496993223,
-      "eval_iou_concrete tiles": 0.9468910021945867,
-      "eval_iou_dark bricks": 0.831982849460412,
-      "eval_iou_grass": 0.8321922440277983,
-      "eval_iou_red bricks": 0.8614201570680629,
-      "eval_iou_road": 0.9161907914701095,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3134627262348163,
-      "eval_iou_speedway bricks": 0.9774938374421029,
-      "eval_iou_steel": 0.8543774439823723,
-      "eval_iou_tiles": 0.2659400544959128,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11146039515733719,
-      "eval_mean_accuracy": 0.7077197300434612,
-      "eval_mean_iou": 0.6668480245367294,
-      "eval_overall_accuracy": 0.9735794067382812,
-      "eval_runtime": 4.399,
-      "eval_samples_per_second": 8.184,
-      "eval_steps_per_second": 0.455,
-      "step": 4900
-    },
-    {
-      "epoch": 445.55,
-      "learning_rate": 1.1464114832535886e-05,
-      "loss": 0.0332,
-      "step": 4901
-    },
-    {
-      "epoch": 445.64,
-      "learning_rate": 1.1444976076555025e-05,
-      "loss": 0.0296,
-      "step": 4902
-    },
-    {
-      "epoch": 445.73,
-      "learning_rate": 1.1425837320574164e-05,
-      "loss": 0.0298,
-      "step": 4903
-    },
-    {
-      "epoch": 445.82,
-      "learning_rate": 1.1406698564593302e-05,
-      "loss": 0.0293,
-      "step": 4904
-    },
-    {
-      "epoch": 445.91,
-      "learning_rate": 1.1387559808612441e-05,
-      "loss": 0.0275,
-      "step": 4905
-    },
-    {
-      "epoch": 446.0,
-      "learning_rate": 1.136842105263158e-05,
-      "loss": 0.0331,
-      "step": 4906
-    },
-    {
-      "epoch": 446.09,
-      "learning_rate": 1.1349282296650717e-05,
-      "loss": 0.0298,
-      "step": 4907
-    },
-    {
-      "epoch": 446.18,
-      "learning_rate": 1.1330143540669858e-05,
-      "loss": 0.0322,
-      "step": 4908
-    },
-    {
-      "epoch": 446.27,
-      "learning_rate": 1.1311004784688996e-05,
-      "loss": 0.0329,
-      "step": 4909
-    },
-    {
-      "epoch": 446.36,
-      "learning_rate": 1.1291866028708133e-05,
-      "loss": 0.0294,
-      "step": 4910
-    },
-    {
-      "epoch": 446.45,
-      "learning_rate": 1.1272727272727274e-05,
-      "loss": 0.0285,
-      "step": 4911
-    },
-    {
-      "epoch": 446.55,
-      "learning_rate": 1.1253588516746413e-05,
-      "loss": 0.03,
-      "step": 4912
-    },
-    {
-      "epoch": 446.64,
-      "learning_rate": 1.123444976076555e-05,
-      "loss": 0.0294,
-      "step": 4913
-    },
-    {
-      "epoch": 446.73,
-      "learning_rate": 1.121531100478469e-05,
-      "loss": 0.033,
-      "step": 4914
-    },
-    {
-      "epoch": 446.82,
-      "learning_rate": 1.1196172248803829e-05,
-      "loss": 0.0262,
-      "step": 4915
-    },
-    {
-      "epoch": 446.91,
-      "learning_rate": 1.1177033492822966e-05,
-      "loss": 0.0304,
-      "step": 4916
-    },
-    {
-      "epoch": 447.0,
-      "learning_rate": 1.1157894736842106e-05,
-      "loss": 0.0264,
-      "step": 4917
-    },
-    {
-      "epoch": 447.09,
-      "learning_rate": 1.1138755980861244e-05,
-      "loss": 0.0281,
-      "step": 4918
-    },
-    {
-      "epoch": 447.18,
-      "learning_rate": 1.1119617224880384e-05,
-      "loss": 0.0292,
-      "step": 4919
-    },
-    {
-      "epoch": 447.27,
-      "learning_rate": 1.1100478468899523e-05,
-      "loss": 0.027,
-      "step": 4920
-    },
-    {
-      "epoch": 447.27,
-      "eval_accuracy_NAT": 0.9928408488569144,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9508629662217263,
-      "eval_accuracy_concrete tiles": 0.9897572310890969,
-      "eval_accuracy_dark bricks": 0.8775643739655903,
-      "eval_accuracy_grass": 0.9054128206293229,
-      "eval_accuracy_red bricks": 0.9343039772727273,
-      "eval_accuracy_road": 0.9188512032232671,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48446502908499434,
-      "eval_accuracy_speedway bricks": 0.9890378903897652,
-      "eval_accuracy_steel": 0.9059841039509268,
-      "eval_accuracy_tiles": 0.2420901246404602,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806529658870141,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.88954094596375,
-      "eval_iou_concrete tiles": 0.9479340537058186,
-      "eval_iou_dark bricks": 0.8313885647607935,
-      "eval_iou_grass": 0.8342909039380628,
-      "eval_iou_red bricks": 0.8596634536840385,
-      "eval_iou_road": 0.9118561531972289,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32741466686540466,
-      "eval_iou_speedway bricks": 0.9773964470931006,
-      "eval_iou_steel": 0.8545044519615849,
-      "eval_iou_tiles": 0.2413793103448276,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11351735144853592,
-      "eval_mean_accuracy": 0.7070131207172915,
-      "eval_mean_iou": 0.6658478398001251,
-      "eval_overall_accuracy": 0.9732879002888998,
-      "eval_runtime": 3.5854,
-      "eval_samples_per_second": 10.041,
-      "eval_steps_per_second": 0.558,
-      "step": 4920
-    },
-    {
-      "epoch": 447.36,
-      "learning_rate": 1.108133971291866e-05,
-      "loss": 0.0303,
-      "step": 4921
-    },
-    {
-      "epoch": 447.45,
-      "learning_rate": 1.10622009569378e-05,
-      "loss": 0.0316,
-      "step": 4922
-    },
-    {
-      "epoch": 447.55,
-      "learning_rate": 1.104306220095694e-05,
-      "loss": 0.0292,
-      "step": 4923
-    },
-    {
-      "epoch": 447.64,
-      "learning_rate": 1.1023923444976076e-05,
-      "loss": 0.026,
-      "step": 4924
-    },
-    {
-      "epoch": 447.73,
-      "learning_rate": 1.1004784688995217e-05,
-      "loss": 0.0318,
-      "step": 4925
-    },
-    {
-      "epoch": 447.82,
-      "learning_rate": 1.0985645933014354e-05,
-      "loss": 0.0281,
-      "step": 4926
-    },
-    {
-      "epoch": 447.91,
-      "learning_rate": 1.0966507177033493e-05,
-      "loss": 0.0291,
-      "step": 4927
-    },
-    {
-      "epoch": 448.0,
-      "learning_rate": 1.0947368421052633e-05,
-      "loss": 0.0305,
-      "step": 4928
-    },
-    {
-      "epoch": 448.09,
-      "learning_rate": 1.092822966507177e-05,
-      "loss": 0.0312,
-      "step": 4929
-    },
-    {
-      "epoch": 448.18,
-      "learning_rate": 1.0909090909090909e-05,
-      "loss": 0.0321,
-      "step": 4930
-    },
-    {
-      "epoch": 448.27,
-      "learning_rate": 1.088995215311005e-05,
-      "loss": 0.0257,
-      "step": 4931
-    },
-    {
-      "epoch": 448.36,
-      "learning_rate": 1.0870813397129186e-05,
-      "loss": 0.027,
-      "step": 4932
-    },
-    {
-      "epoch": 448.45,
-      "learning_rate": 1.0851674641148325e-05,
-      "loss": 0.0283,
-      "step": 4933
-    },
-    {
-      "epoch": 448.55,
-      "learning_rate": 1.0832535885167466e-05,
-      "loss": 0.0275,
-      "step": 4934
-    },
-    {
-      "epoch": 448.64,
-      "learning_rate": 1.0813397129186603e-05,
-      "loss": 0.0295,
-      "step": 4935
-    },
-    {
-      "epoch": 448.73,
-      "learning_rate": 1.0794258373205742e-05,
-      "loss": 0.0298,
-      "step": 4936
-    },
-    {
-      "epoch": 448.82,
-      "learning_rate": 1.077511961722488e-05,
-      "loss": 0.0299,
-      "step": 4937
-    },
-    {
-      "epoch": 448.91,
-      "learning_rate": 1.0755980861244019e-05,
-      "loss": 0.0321,
-      "step": 4938
-    },
-    {
-      "epoch": 449.0,
-      "learning_rate": 1.073684210526316e-05,
-      "loss": 0.0348,
-      "step": 4939
-    },
-    {
-      "epoch": 449.09,
-      "learning_rate": 1.0717703349282297e-05,
-      "loss": 0.0307,
-      "step": 4940
-    },
-    {
-      "epoch": 449.09,
-      "eval_accuracy_NAT": 0.9928232037452808,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9506228808103352,
-      "eval_accuracy_concrete tiles": 0.9882284596098575,
-      "eval_accuracy_dark bricks": 0.8797197952349795,
-      "eval_accuracy_grass": 0.9082787810580986,
-      "eval_accuracy_red bricks": 0.9330610795454546,
-      "eval_accuracy_road": 0.9208645359600575,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4991867229068453,
-      "eval_accuracy_speedway bricks": 0.9891525378728909,
-      "eval_accuracy_steel": 0.9067756084690983,
-      "eval_accuracy_tiles": 0.2577044240514998,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807382651174125,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8908454908703778,
-      "eval_iou_concrete tiles": 0.9498075171177525,
-      "eval_iou_dark bricks": 0.8318835304822566,
-      "eval_iou_grass": 0.8360136271951977,
-      "eval_iou_red bricks": 0.8623235969806367,
-      "eval_iou_road": 0.9140340411946224,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33620513582264144,
-      "eval_iou_speedway bricks": 0.97737802874878,
-      "eval_iou_steel": 0.8557633950108156,
-      "eval_iou_tiles": 0.2563176895306859,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11107033491134644,
-      "eval_mean_accuracy": 0.7097244637895691,
-      "eval_mean_iou": 0.6685623321593214,
-      "eval_overall_accuracy": 0.9736326005723741,
-      "eval_runtime": 3.4981,
-      "eval_samples_per_second": 10.291,
-      "eval_steps_per_second": 0.572,
-      "step": 4940
-    },
-    {
-      "epoch": 449.18,
-      "learning_rate": 1.0698564593301435e-05,
-      "loss": 0.0288,
-      "step": 4941
-    },
-    {
-      "epoch": 449.27,
-      "learning_rate": 1.0679425837320576e-05,
-      "loss": 0.0321,
-      "step": 4942
-    },
-    {
-      "epoch": 449.36,
-      "learning_rate": 1.0660287081339713e-05,
-      "loss": 0.0283,
-      "step": 4943
-    },
-    {
-      "epoch": 449.45,
-      "learning_rate": 1.0641148325358852e-05,
-      "loss": 0.0271,
-      "step": 4944
-    },
-    {
-      "epoch": 449.55,
-      "learning_rate": 1.0622009569377992e-05,
-      "loss": 0.0289,
-      "step": 4945
-    },
-    {
-      "epoch": 449.64,
-      "learning_rate": 1.060287081339713e-05,
-      "loss": 0.0361,
-      "step": 4946
-    },
-    {
-      "epoch": 449.73,
-      "learning_rate": 1.0583732057416268e-05,
-      "loss": 0.0325,
-      "step": 4947
-    },
-    {
-      "epoch": 449.82,
-      "learning_rate": 1.0564593301435407e-05,
-      "loss": 0.0277,
-      "step": 4948
-    },
-    {
-      "epoch": 449.91,
-      "learning_rate": 1.0545454545454546e-05,
-      "loss": 0.0299,
-      "step": 4949
-    },
-    {
-      "epoch": 450.0,
-      "learning_rate": 1.0526315789473684e-05,
-      "loss": 0.0279,
-      "step": 4950
-    },
-    {
-      "epoch": 450.09,
-      "learning_rate": 1.0507177033492823e-05,
-      "loss": 0.0268,
-      "step": 4951
-    },
-    {
-      "epoch": 450.18,
-      "learning_rate": 1.0488038277511962e-05,
-      "loss": 0.0339,
-      "step": 4952
-    },
-    {
-      "epoch": 450.27,
-      "learning_rate": 1.04688995215311e-05,
-      "loss": 0.0346,
-      "step": 4953
-    },
-    {
-      "epoch": 450.36,
-      "learning_rate": 1.044976076555024e-05,
-      "loss": 0.0337,
-      "step": 4954
-    },
-    {
-      "epoch": 450.45,
-      "learning_rate": 1.0430622009569378e-05,
-      "loss": 0.0302,
-      "step": 4955
-    },
-    {
-      "epoch": 450.55,
-      "learning_rate": 1.0411483253588517e-05,
-      "loss": 0.0266,
-      "step": 4956
-    },
-    {
-      "epoch": 450.64,
-      "learning_rate": 1.0392344497607656e-05,
-      "loss": 0.0266,
-      "step": 4957
-    },
-    {
-      "epoch": 450.73,
-      "learning_rate": 1.0373205741626795e-05,
-      "loss": 0.03,
-      "step": 4958
-    },
-    {
-      "epoch": 450.82,
-      "learning_rate": 1.0354066985645933e-05,
-      "loss": 0.0283,
-      "step": 4959
-    },
-    {
-      "epoch": 450.91,
-      "learning_rate": 1.0334928229665072e-05,
-      "loss": 0.0292,
-      "step": 4960
-    },
-    {
-      "epoch": 450.91,
-      "eval_accuracy_NAT": 0.9932918579102683,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9518434267565131,
-      "eval_accuracy_concrete tiles": 0.9860576041093377,
-      "eval_accuracy_dark bricks": 0.8725992071128902,
-      "eval_accuracy_grass": 0.8997123554350144,
-      "eval_accuracy_red bricks": 0.9295099431818182,
-      "eval_accuracy_road": 0.92195282392589,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4959336145342266,
-      "eval_accuracy_speedway bricks": 0.9878763303633601,
-      "eval_accuracy_steel": 0.9132972759052833,
-      "eval_accuracy_tiles": 0.2733187234625394,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806421054153571,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8910127073370088,
-      "eval_iou_concrete tiles": 0.9521419503409997,
-      "eval_iou_dark bricks": 0.8330944769044207,
-      "eval_iou_grass": 0.8350135912756337,
-      "eval_iou_red bricks": 0.865146256817055,
-      "eval_iou_road": 0.9146663467594398,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33444262660816537,
-      "eval_iou_speedway bricks": 0.9775210322364924,
-      "eval_iou_steel": 0.8556200951615893,
-      "eval_iou_tiles": 0.2712935898307389,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11137256026268005,
-      "eval_mean_accuracy": 0.7096456278997801,
-      "eval_mean_iou": 0.6700457522066846,
-      "eval_overall_accuracy": 0.9736882315741645,
-      "eval_runtime": 3.4834,
-      "eval_samples_per_second": 10.335,
-      "eval_steps_per_second": 0.574,
-      "step": 4960
-    },
-    {
-      "epoch": 451.0,
-      "learning_rate": 1.0315789473684211e-05,
-      "loss": 0.0325,
-      "step": 4961
-    },
-    {
-      "epoch": 451.09,
-      "learning_rate": 1.029665071770335e-05,
-      "loss": 0.0272,
-      "step": 4962
-    },
-    {
-      "epoch": 451.18,
-      "learning_rate": 1.0277511961722489e-05,
-      "loss": 0.0297,
-      "step": 4963
-    },
-    {
-      "epoch": 451.27,
-      "learning_rate": 1.0258373205741627e-05,
-      "loss": 0.0252,
-      "step": 4964
-    },
-    {
-      "epoch": 451.36,
-      "learning_rate": 1.0239234449760766e-05,
-      "loss": 0.0295,
-      "step": 4965
-    },
-    {
-      "epoch": 451.45,
-      "learning_rate": 1.0220095693779905e-05,
-      "loss": 0.0295,
-      "step": 4966
-    },
-    {
-      "epoch": 451.55,
-      "learning_rate": 1.0200956937799044e-05,
-      "loss": 0.0303,
-      "step": 4967
-    },
-    {
-      "epoch": 451.64,
-      "learning_rate": 1.0181818181818182e-05,
-      "loss": 0.0286,
-      "step": 4968
-    },
-    {
-      "epoch": 451.73,
-      "learning_rate": 1.0162679425837321e-05,
-      "loss": 0.0292,
-      "step": 4969
-    },
-    {
-      "epoch": 451.82,
-      "learning_rate": 1.014354066985646e-05,
-      "loss": 0.0254,
-      "step": 4970
-    },
-    {
-      "epoch": 451.91,
-      "learning_rate": 1.0124401913875599e-05,
-      "loss": 0.029,
-      "step": 4971
-    },
-    {
-      "epoch": 452.0,
-      "learning_rate": 1.0105263157894738e-05,
-      "loss": 0.0303,
-      "step": 4972
-    },
-    {
-      "epoch": 452.09,
-      "learning_rate": 1.0086124401913876e-05,
-      "loss": 0.031,
-      "step": 4973
-    },
-    {
-      "epoch": 452.18,
-      "learning_rate": 1.0066985645933015e-05,
-      "loss": 0.0405,
-      "step": 4974
-    },
-    {
-      "epoch": 452.27,
-      "learning_rate": 1.0047846889952154e-05,
-      "loss": 0.0261,
-      "step": 4975
-    },
-    {
-      "epoch": 452.36,
-      "learning_rate": 1.0028708133971293e-05,
-      "loss": 0.0312,
-      "step": 4976
-    },
-    {
-      "epoch": 452.45,
-      "learning_rate": 1.0009569377990431e-05,
-      "loss": 0.0307,
-      "step": 4977
-    },
-    {
-      "epoch": 452.55,
-      "learning_rate": 9.99043062200957e-06,
-      "loss": 0.033,
-      "step": 4978
-    },
-    {
-      "epoch": 452.64,
-      "learning_rate": 9.971291866028709e-06,
-      "loss": 0.0284,
-      "step": 4979
-    },
-    {
-      "epoch": 452.73,
-      "learning_rate": 9.952153110047848e-06,
-      "loss": 0.0329,
-      "step": 4980
-    },
-    {
-      "epoch": 452.73,
-      "eval_accuracy_NAT": 0.9932819766477534,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9499012833168805,
-      "eval_accuracy_concrete tiles": 0.9850791903626246,
-      "eval_accuracy_dark bricks": 0.874023324737308,
-      "eval_accuracy_grass": 0.905224086649867,
-      "eval_accuracy_red bricks": 0.9351917613636364,
-      "eval_accuracy_road": 0.9335709025655784,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5081465552890579,
-      "eval_accuracy_speedway bricks": 0.9885117187829988,
-      "eval_accuracy_steel": 0.9105764791240684,
-      "eval_accuracy_tiles": 0.27708533077660596,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806673714049983,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.893616376960814,
-      "eval_iou_concrete tiles": 0.9537313874663272,
-      "eval_iou_dark bricks": 0.8323131620422974,
-      "eval_iou_grass": 0.8363634015125584,
-      "eval_iou_red bricks": 0.8648604269293925,
-      "eval_iou_road": 0.9257461536709046,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.34283801127169244,
-      "eval_iou_speedway bricks": 0.9776080747706384,
-      "eval_iou_steel": 0.8557746697144628,
-      "eval_iou_tiles": 0.2753879662401307,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10519985854625702,
-      "eval_mean_accuracy": 0.7123532776627984,
-      "eval_mean_iou": 0.6722236155372474,
-      "eval_overall_accuracy": 0.9748473697238498,
-      "eval_runtime": 4.0208,
-      "eval_samples_per_second": 8.953,
-      "eval_steps_per_second": 0.497,
-      "step": 4980
-    },
-    {
-      "epoch": 452.82,
-      "learning_rate": 9.933014354066985e-06,
-      "loss": 0.0273,
-      "step": 4981
-    },
-    {
-      "epoch": 452.91,
-      "learning_rate": 9.913875598086125e-06,
-      "loss": 0.0287,
-      "step": 4982
-    },
-    {
-      "epoch": 453.0,
-      "learning_rate": 9.894736842105264e-06,
-      "loss": 0.0261,
-      "step": 4983
-    },
-    {
-      "epoch": 453.09,
-      "learning_rate": 9.875598086124401e-06,
-      "loss": 0.0304,
-      "step": 4984
-    },
-    {
-      "epoch": 453.18,
-      "learning_rate": 9.856459330143542e-06,
-      "loss": 0.0309,
-      "step": 4985
-    },
-    {
-      "epoch": 453.27,
-      "learning_rate": 9.83732057416268e-06,
-      "loss": 0.028,
-      "step": 4986
-    },
-    {
-      "epoch": 453.36,
-      "learning_rate": 9.818181818181818e-06,
-      "loss": 0.0283,
-      "step": 4987
-    },
-    {
-      "epoch": 453.45,
-      "learning_rate": 9.799043062200958e-06,
-      "loss": 0.0293,
-      "step": 4988
-    },
-    {
-      "epoch": 453.55,
-      "learning_rate": 9.779904306220097e-06,
-      "loss": 0.0343,
-      "step": 4989
-    },
-    {
-      "epoch": 453.64,
-      "learning_rate": 9.760765550239234e-06,
-      "loss": 0.0315,
-      "step": 4990
-    },
-    {
-      "epoch": 453.73,
-      "learning_rate": 9.741626794258374e-06,
-      "loss": 0.0298,
-      "step": 4991
-    },
-    {
-      "epoch": 453.82,
-      "learning_rate": 9.722488038277511e-06,
-      "loss": 0.0292,
-      "step": 4992
-    },
-    {
-      "epoch": 453.91,
-      "learning_rate": 9.703349282296652e-06,
-      "loss": 0.0286,
-      "step": 4993
-    },
-    {
-      "epoch": 454.0,
-      "learning_rate": 9.68421052631579e-06,
-      "loss": 0.0283,
-      "step": 4994
-    },
-    {
-      "epoch": 454.09,
-      "learning_rate": 9.665071770334928e-06,
-      "loss": 0.028,
-      "step": 4995
-    },
-    {
-      "epoch": 454.18,
-      "learning_rate": 9.645933014354068e-06,
-      "loss": 0.0305,
-      "step": 4996
-    },
-    {
-      "epoch": 454.27,
-      "learning_rate": 9.626794258373207e-06,
-      "loss": 0.026,
-      "step": 4997
-    },
-    {
-      "epoch": 454.36,
-      "learning_rate": 9.607655502392344e-06,
-      "loss": 0.0299,
-      "step": 4998
-    },
-    {
-      "epoch": 454.45,
-      "learning_rate": 9.588516746411485e-06,
-      "loss": 0.031,
-      "step": 4999
-    },
-    {
-      "epoch": 454.55,
-      "learning_rate": 9.569377990430622e-06,
-      "loss": 0.0314,
-      "step": 5000
-    },
-    {
-      "epoch": 454.55,
-      "eval_accuracy_NAT": 0.993021005446693,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.951597976307996,
-      "eval_accuracy_concrete tiles": 0.9868525652785421,
-      "eval_accuracy_dark bricks": 0.8730225934336631,
-      "eval_accuracy_grass": 0.9056819413037324,
-      "eval_accuracy_red bricks": 0.939453125,
-      "eval_accuracy_road": 0.939891437229364,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.49601632067929313,
-      "eval_accuracy_speedway bricks": 0.9890240120102289,
-      "eval_accuracy_steel": 0.9098426884770134,
-      "eval_accuracy_tiles": 0.22086015614299412,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808122783572064,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8909787088688349,
-      "eval_iou_concrete tiles": 0.9546570439823716,
-      "eval_iou_dark bricks": 0.8333455801307958,
-      "eval_iou_grass": 0.8362328521777855,
-      "eval_iou_red bricks": 0.8597660058498537,
-      "eval_iou_road": 0.9332289579308697,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.33413189221312234,
-      "eval_iou_speedway bricks": 0.9774579214533796,
-      "eval_iou_steel": 0.8565784366995265,
-      "eval_iou_tiles": 0.2198364008179959,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10168782621622086,
-      "eval_mean_accuracy": 0.7080972170238092,
-      "eval_mean_iou": 0.6674635444985955,
-      "eval_overall_accuracy": 0.9753452936808268,
-      "eval_runtime": 3.8102,
-      "eval_samples_per_second": 9.448,
-      "eval_steps_per_second": 0.525,
-      "step": 5000
-    },
-    {
-      "epoch": 454.64,
-      "learning_rate": 9.55023923444976e-06,
-      "loss": 0.0305,
-      "step": 5001
-    },
-    {
-      "epoch": 454.73,
-      "learning_rate": 9.531100478468901e-06,
-      "loss": 0.0322,
-      "step": 5002
-    },
-    {
-      "epoch": 454.82,
-      "learning_rate": 9.511961722488038e-06,
-      "loss": 0.0267,
-      "step": 5003
-    },
-    {
-      "epoch": 454.91,
-      "learning_rate": 9.492822966507177e-06,
-      "loss": 0.0286,
-      "step": 5004
-    },
-    {
-      "epoch": 455.0,
-      "learning_rate": 9.473684210526317e-06,
-      "loss": 0.0297,
-      "step": 5005
-    },
-    {
-      "epoch": 455.09,
-      "learning_rate": 9.454545454545454e-06,
-      "loss": 0.029,
-      "step": 5006
-    },
-    {
-      "epoch": 455.18,
-      "learning_rate": 9.435406698564593e-06,
-      "loss": 0.0282,
-      "step": 5007
-    },
-    {
-      "epoch": 455.27,
-      "learning_rate": 9.416267942583734e-06,
-      "loss": 0.0305,
-      "step": 5008
-    },
-    {
-      "epoch": 455.36,
-      "learning_rate": 9.39712918660287e-06,
-      "loss": 0.0277,
-      "step": 5009
-    },
-    {
-      "epoch": 455.45,
-      "learning_rate": 9.377990430622011e-06,
-      "loss": 0.0327,
-      "step": 5010
-    },
-    {
-      "epoch": 455.55,
-      "learning_rate": 9.358851674641148e-06,
-      "loss": 0.0285,
-      "step": 5011
-    },
-    {
-      "epoch": 455.64,
-      "learning_rate": 9.339712918660287e-06,
-      "loss": 0.0284,
-      "step": 5012
-    },
-    {
-      "epoch": 455.73,
-      "learning_rate": 9.320574162679427e-06,
-      "loss": 0.0284,
-      "step": 5013
-    },
-    {
-      "epoch": 455.82,
-      "learning_rate": 9.301435406698564e-06,
-      "loss": 0.0288,
-      "step": 5014
-    },
-    {
-      "epoch": 455.91,
-      "learning_rate": 9.282296650717703e-06,
-      "loss": 0.0316,
-      "step": 5015
-    },
-    {
-      "epoch": 456.0,
-      "learning_rate": 9.263157894736844e-06,
-      "loss": 0.0317,
-      "step": 5016
-    },
-    {
-      "epoch": 456.09,
-      "learning_rate": 9.24401913875598e-06,
-      "loss": 0.0304,
-      "step": 5017
-    },
-    {
-      "epoch": 456.18,
-      "learning_rate": 9.22488038277512e-06,
-      "loss": 0.0267,
-      "step": 5018
-    },
-    {
-      "epoch": 456.27,
-      "learning_rate": 9.20574162679426e-06,
-      "loss": 0.0324,
-      "step": 5019
-    },
-    {
-      "epoch": 456.36,
-      "learning_rate": 9.186602870813397e-06,
-      "loss": 0.03,
-      "step": 5020
-    },
-    {
-      "epoch": 456.36,
-      "eval_accuracy_NAT": 0.9929534246691365,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9522578758745011,
-      "eval_accuracy_concrete tiles": 0.9881061578915183,
-      "eval_accuracy_dark bricks": 0.8754089526962011,
-      "eval_accuracy_grass": 0.9043118724158299,
-      "eval_accuracy_red bricks": 0.9403409090909091,
-      "eval_accuracy_road": 0.9388672373326303,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48951010393405564,
-      "eval_accuracy_speedway bricks": 0.9886722252593748,
-      "eval_accuracy_steel": 0.9099910955741706,
-      "eval_accuracy_tiles": 0.22277770168470074,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807709970934281,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8907654224333375,
-      "eval_iou_concrete tiles": 0.9533882054459096,
-      "eval_iou_dark bricks": 0.8322599531615925,
-      "eval_iou_grass": 0.8353263641822655,
-      "eval_iou_red bricks": 0.8601591684261816,
-      "eval_iou_road": 0.9313509927896351,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3303749185970788,
-      "eval_iou_speedway bricks": 0.9775521357241129,
-      "eval_iou_steel": 0.8571373100250841,
-      "eval_iou_tiles": 0.2216241994822183,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10293947160243988,
-      "eval_mean_accuracy": 0.7079382735710021,
-      "eval_mean_iou": 0.666977666720065,
-      "eval_overall_accuracy": 0.9751550886366103,
-      "eval_runtime": 3.3068,
-      "eval_samples_per_second": 10.887,
-      "eval_steps_per_second": 0.605,
-      "step": 5020
-    },
-    {
-      "epoch": 456.45,
-      "learning_rate": 9.167464114832536e-06,
-      "loss": 0.0315,
-      "step": 5021
-    },
-    {
-      "epoch": 456.55,
-      "learning_rate": 9.148325358851675e-06,
-      "loss": 0.0337,
-      "step": 5022
-    },
-    {
-      "epoch": 456.64,
-      "learning_rate": 9.129186602870813e-06,
-      "loss": 0.0304,
-      "step": 5023
-    },
-    {
-      "epoch": 456.73,
-      "learning_rate": 9.110047846889952e-06,
-      "loss": 0.0253,
-      "step": 5024
-    },
-    {
-      "epoch": 456.82,
-      "learning_rate": 9.090909090909091e-06,
-      "loss": 0.0257,
-      "step": 5025
-    },
-    {
-      "epoch": 456.91,
-      "learning_rate": 9.07177033492823e-06,
-      "loss": 0.0323,
-      "step": 5026
-    },
-    {
-      "epoch": 457.0,
-      "learning_rate": 9.05263157894737e-06,
-      "loss": 0.0255,
-      "step": 5027
-    },
-    {
-      "epoch": 457.09,
-      "learning_rate": 9.033492822966507e-06,
-      "loss": 0.0339,
-      "step": 5028
-    },
-    {
-      "epoch": 457.18,
-      "learning_rate": 9.014354066985646e-06,
-      "loss": 0.0348,
-      "step": 5029
-    },
-    {
-      "epoch": 457.27,
-      "learning_rate": 8.995215311004785e-06,
-      "loss": 0.0282,
-      "step": 5030
-    },
-    {
-      "epoch": 457.36,
-      "learning_rate": 8.976076555023924e-06,
-      "loss": 0.0313,
-      "step": 5031
-    },
-    {
-      "epoch": 457.45,
-      "learning_rate": 8.956937799043062e-06,
-      "loss": 0.0318,
-      "step": 5032
-    },
-    {
-      "epoch": 457.55,
-      "learning_rate": 8.937799043062201e-06,
-      "loss": 0.0273,
-      "step": 5033
-    },
-    {
-      "epoch": 457.64,
-      "learning_rate": 8.91866028708134e-06,
-      "loss": 0.0261,
-      "step": 5034
-    },
-    {
-      "epoch": 457.73,
-      "learning_rate": 8.899521531100479e-06,
-      "loss": 0.0261,
-      "step": 5035
-    },
-    {
-      "epoch": 457.82,
-      "learning_rate": 8.880382775119618e-06,
-      "loss": 0.0292,
-      "step": 5036
-    },
-    {
-      "epoch": 457.91,
-      "learning_rate": 8.861244019138756e-06,
-      "loss": 0.0315,
-      "step": 5037
-    },
-    {
-      "epoch": 458.0,
-      "learning_rate": 8.842105263157895e-06,
-      "loss": 0.0323,
-      "step": 5038
-    },
-    {
-      "epoch": 458.09,
-      "learning_rate": 8.822966507177034e-06,
-      "loss": 0.025,
-      "step": 5039
-    },
-    {
-      "epoch": 458.18,
-      "learning_rate": 8.803827751196173e-06,
-      "loss": 0.0311,
-      "step": 5040
-    },
-    {
-      "epoch": 458.18,
-      "eval_accuracy_NAT": 0.9930829397885269,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9527943795871068,
-      "eval_accuracy_concrete tiles": 0.9856906989543203,
-      "eval_accuracy_dark bricks": 0.8703668065124514,
-      "eval_accuracy_grass": 0.9032179143497241,
-      "eval_accuracy_red bricks": 0.9412286931818182,
-      "eval_accuracy_road": 0.9387100401820101,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.48598130841121495,
-      "eval_accuracy_speedway bricks": 0.9883994849310969,
-      "eval_accuracy_steel": 0.9077897236330057,
-      "eval_accuracy_tiles": 0.23941925763594027,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807174002557003,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8906951171652149,
-      "eval_iou_concrete tiles": 0.9545212293480192,
-      "eval_iou_dark bricks": 0.8315437228800471,
-      "eval_iou_grass": 0.834547568300717,
-      "eval_iou_red bricks": 0.8613909652258693,
-      "eval_iou_road": 0.9311637792989952,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.32795668917787574,
-      "eval_iou_speedway bricks": 0.9776546015799764,
-      "eval_iou_steel": 0.8571005986252637,
-      "eval_iou_tiles": 0.23813091751243104,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10318261384963989,
-      "eval_mean_accuracy": 0.7082062497820935,
-      "eval_mean_iou": 0.6681094299515469,
-      "eval_overall_accuracy": 0.975142584906684,
-      "eval_runtime": 3.598,
-      "eval_samples_per_second": 10.005,
-      "eval_steps_per_second": 0.556,
-      "step": 5040
-    },
-    {
-      "epoch": 458.27,
-      "learning_rate": 8.784688995215311e-06,
-      "loss": 0.0293,
-      "step": 5041
-    },
-    {
-      "epoch": 458.36,
-      "learning_rate": 8.76555023923445e-06,
-      "loss": 0.0278,
-      "step": 5042
-    },
-    {
-      "epoch": 458.45,
-      "learning_rate": 8.746411483253589e-06,
-      "loss": 0.0297,
-      "step": 5043
-    },
-    {
-      "epoch": 458.55,
-      "learning_rate": 8.727272727272728e-06,
-      "loss": 0.0324,
-      "step": 5044
-    },
-    {
-      "epoch": 458.64,
-      "learning_rate": 8.708133971291867e-06,
-      "loss": 0.0297,
-      "step": 5045
-    },
-    {
-      "epoch": 458.73,
-      "learning_rate": 8.688995215311005e-06,
-      "loss": 0.0262,
-      "step": 5046
-    },
-    {
-      "epoch": 458.82,
-      "learning_rate": 8.669856459330144e-06,
-      "loss": 0.0284,
-      "step": 5047
-    },
-    {
-      "epoch": 458.91,
-      "learning_rate": 8.650717703349283e-06,
-      "loss": 0.0286,
-      "step": 5048
-    },
-    {
-      "epoch": 459.0,
-      "learning_rate": 8.631578947368422e-06,
-      "loss": 0.0353,
-      "step": 5049
-    },
-    {
-      "epoch": 459.09,
-      "learning_rate": 8.61244019138756e-06,
-      "loss": 0.0366,
-      "step": 5050
-    },
-    {
-      "epoch": 459.18,
-      "learning_rate": 8.5933014354067e-06,
-      "loss": 0.029,
-      "step": 5051
-    },
-    {
-      "epoch": 459.27,
-      "learning_rate": 8.574162679425838e-06,
-      "loss": 0.0298,
-      "step": 5052
-    },
-    {
-      "epoch": 459.36,
-      "learning_rate": 8.555023923444977e-06,
-      "loss": 0.0316,
-      "step": 5053
-    },
-    {
-      "epoch": 459.45,
-      "learning_rate": 8.535885167464116e-06,
-      "loss": 0.0287,
-      "step": 5054
-    },
-    {
-      "epoch": 459.55,
-      "learning_rate": 8.516746411483253e-06,
-      "loss": 0.0293,
-      "step": 5055
-    },
-    {
-      "epoch": 459.64,
-      "learning_rate": 8.497607655502393e-06,
-      "loss": 0.0279,
-      "step": 5056
-    },
-    {
-      "epoch": 459.73,
-      "learning_rate": 8.478468899521532e-06,
-      "loss": 0.0347,
-      "step": 5057
-    },
-    {
-      "epoch": 459.82,
-      "learning_rate": 8.459330143540669e-06,
-      "loss": 0.03,
-      "step": 5058
-    },
-    {
-      "epoch": 459.91,
-      "learning_rate": 8.44019138755981e-06,
-      "loss": 0.0272,
-      "step": 5059
-    },
-    {
-      "epoch": 460.0,
-      "learning_rate": 8.421052631578948e-06,
-      "loss": 0.0269,
-      "step": 5060
-    },
-    {
-      "epoch": 460.0,
-      "eval_accuracy_NAT": 0.993197633014145,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9534194064122924,
-      "eval_accuracy_concrete tiles": 0.9823885525591635,
-      "eval_accuracy_dark bricks": 0.8792964089142066,
-      "eval_accuracy_grass": 0.9008098085748139,
-      "eval_accuracy_red bricks": 0.9385653409090909,
-      "eval_accuracy_road": 0.9368962269056225,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4996002536321782,
-      "eval_accuracy_speedway bricks": 0.9877797851144121,
-      "eval_accuracy_steel": 0.9113927181584328,
-      "eval_accuracy_tiles": 0.24626763457060677,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807723872600012,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8912436479805813,
-      "eval_iou_concrete tiles": 0.9550561797752809,
-      "eval_iou_dark bricks": 0.8334853515268708,
-      "eval_iou_grass": 0.8345459549597681,
-      "eval_iou_red bricks": 0.8662733529990168,
-      "eval_iou_road": 0.9295529801324504,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3354435065896639,
-      "eval_iou_speedway bricks": 0.9775311427187753,
-      "eval_iou_steel": 0.8566878240450118,
-      "eval_iou_tiles": 0.24532678400873242,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10305383801460266,
-      "eval_mean_accuracy": 0.7099702899049973,
-      "eval_mean_iou": 0.6696860855381657,
-      "eval_overall_accuracy": 0.9750408596462674,
-      "eval_runtime": 3.7535,
-      "eval_samples_per_second": 9.591,
-      "eval_steps_per_second": 0.533,
-      "step": 5060
-    },
-    {
-      "epoch": 460.09,
-      "learning_rate": 8.401913875598085e-06,
-      "loss": 0.0277,
-      "step": 5061
-    },
-    {
-      "epoch": 460.18,
-      "learning_rate": 8.382775119617226e-06,
-      "loss": 0.0296,
-      "step": 5062
-    },
-    {
-      "epoch": 460.27,
-      "learning_rate": 8.363636363636365e-06,
-      "loss": 0.0308,
-      "step": 5063
-    },
-    {
-      "epoch": 460.36,
-      "learning_rate": 8.344497607655503e-06,
-      "loss": 0.0279,
-      "step": 5064
-    },
-    {
-      "epoch": 460.45,
-      "learning_rate": 8.325358851674642e-06,
-      "loss": 0.0286,
-      "step": 5065
-    },
-    {
-      "epoch": 460.55,
-      "learning_rate": 8.30622009569378e-06,
-      "loss": 0.0299,
-      "step": 5066
-    },
-    {
-      "epoch": 460.64,
-      "learning_rate": 8.28708133971292e-06,
-      "loss": 0.0246,
-      "step": 5067
-    },
-    {
-      "epoch": 460.73,
-      "learning_rate": 8.267942583732058e-06,
-      "loss": 0.0318,
-      "step": 5068
-    },
-    {
-      "epoch": 460.82,
-      "learning_rate": 8.248803827751196e-06,
-      "loss": 0.0287,
-      "step": 5069
-    },
-    {
-      "epoch": 460.91,
-      "learning_rate": 8.229665071770336e-06,
-      "loss": 0.028,
-      "step": 5070
-    },
-    {
-      "epoch": 461.0,
-      "learning_rate": 8.210526315789475e-06,
-      "loss": 0.0269,
-      "step": 5071
-    },
-    {
-      "epoch": 461.09,
-      "learning_rate": 8.191387559808612e-06,
-      "loss": 0.0291,
-      "step": 5072
-    },
-    {
-      "epoch": 461.18,
-      "learning_rate": 8.172248803827752e-06,
-      "loss": 0.0298,
-      "step": 5073
-    },
-    {
-      "epoch": 461.27,
-      "learning_rate": 8.15311004784689e-06,
-      "loss": 0.0292,
-      "step": 5074
-    },
-    {
-      "epoch": 461.36,
-      "learning_rate": 8.133971291866028e-06,
-      "loss": 0.0304,
-      "step": 5075
-    },
-    {
-      "epoch": 461.45,
-      "learning_rate": 8.114832535885169e-06,
-      "loss": 0.0285,
-      "step": 5076
-    },
-    {
-      "epoch": 461.55,
-      "learning_rate": 8.095693779904306e-06,
-      "loss": 0.036,
-      "step": 5077
-    },
-    {
-      "epoch": 461.64,
-      "learning_rate": 8.076555023923445e-06,
-      "loss": 0.0305,
-      "step": 5078
-    },
-    {
-      "epoch": 461.73,
-      "learning_rate": 8.057416267942585e-06,
-      "loss": 0.0293,
-      "step": 5079
-    },
-    {
-      "epoch": 461.82,
-      "learning_rate": 8.038277511961722e-06,
-      "loss": 0.0261,
-      "step": 5080
-    },
-    {
-      "epoch": 461.82,
-      "eval_accuracy_NAT": 0.9932726247385877,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9515751749002103,
-      "eval_accuracy_concrete tiles": 0.9827554577141809,
-      "eval_accuracy_dark bricks": 0.8798737538970787,
-      "eval_accuracy_grass": 0.9002366164890586,
-      "eval_accuracy_red bricks": 0.9389204545454546,
-      "eval_accuracy_road": 0.9369724070632307,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5139911228737628,
-      "eval_accuracy_speedway bricks": 0.9883578497924881,
-      "eval_accuracy_steel": 0.9087296352483345,
-      "eval_accuracy_tiles": 0.26085467744144636,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807680011094948,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8920069855423414,
-      "eval_iou_concrete tiles": 0.953825152828061,
-      "eval_iou_dark bricks": 0.8338196673475343,
-      "eval_iou_grass": 0.8339123199502708,
-      "eval_iou_red bricks": 0.867311792684927,
-      "eval_iou_road": 0.9292573151382798,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.344093165753096,
-      "eval_iou_speedway bricks": 0.9775813283374803,
-      "eval_iou_steel": 0.8565810743596121,
-      "eval_iou_tiles": 0.25961014176663033,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10353000462055206,
-      "eval_mean_accuracy": 0.7119645980541409,
-      "eval_mean_iou": 0.6714436111398252,
-      "eval_overall_accuracy": 0.9750777350531684,
-      "eval_runtime": 3.2724,
-      "eval_samples_per_second": 11.001,
-      "eval_steps_per_second": 0.611,
-      "step": 5080
-    },
-    {
-      "epoch": 461.91,
-      "learning_rate": 8.019138755980863e-06,
-      "loss": 0.0346,
-      "step": 5081
-    },
-    {
-      "epoch": 462.0,
-      "learning_rate": 8.000000000000001e-06,
-      "loss": 0.0245,
-      "step": 5082
-    },
-    {
-      "epoch": 462.09,
-      "learning_rate": 7.980861244019138e-06,
-      "loss": 0.0275,
-      "step": 5083
-    },
-    {
-      "epoch": 462.18,
-      "learning_rate": 7.961722488038279e-06,
-      "loss": 0.0264,
-      "step": 5084
-    },
-    {
-      "epoch": 462.27,
-      "learning_rate": 7.942583732057416e-06,
-      "loss": 0.0278,
-      "step": 5085
-    },
-    {
-      "epoch": 462.36,
-      "learning_rate": 7.923444976076555e-06,
-      "loss": 0.0306,
-      "step": 5086
-    },
-    {
-      "epoch": 462.45,
-      "learning_rate": 7.904306220095695e-06,
-      "loss": 0.0293,
-      "step": 5087
-    },
-    {
-      "epoch": 462.55,
-      "learning_rate": 7.885167464114832e-06,
-      "loss": 0.0284,
-      "step": 5088
-    },
-    {
-      "epoch": 462.64,
-      "learning_rate": 7.866028708133971e-06,
-      "loss": 0.0284,
-      "step": 5089
-    },
-    {
-      "epoch": 462.73,
-      "learning_rate": 7.846889952153112e-06,
-      "loss": 0.033,
-      "step": 5090
-    },
-    {
-      "epoch": 462.82,
-      "learning_rate": 7.827751196172249e-06,
-      "loss": 0.0283,
-      "step": 5091
-    },
-    {
-      "epoch": 462.91,
-      "learning_rate": 7.808612440191387e-06,
-      "loss": 0.0245,
-      "step": 5092
-    },
-    {
-      "epoch": 463.0,
-      "learning_rate": 7.789473684210528e-06,
-      "loss": 0.0273,
-      "step": 5093
-    },
-    {
-      "epoch": 463.09,
-      "learning_rate": 7.770334928229665e-06,
-      "loss": 0.0397,
-      "step": 5094
-    },
-    {
-      "epoch": 463.18,
-      "learning_rate": 7.751196172248804e-06,
-      "loss": 0.0277,
-      "step": 5095
-    },
-    {
-      "epoch": 463.27,
-      "learning_rate": 7.732057416267943e-06,
-      "loss": 0.0277,
-      "step": 5096
-    },
-    {
-      "epoch": 463.36,
-      "learning_rate": 7.712918660287081e-06,
-      "loss": 0.0265,
-      "step": 5097
-    },
-    {
-      "epoch": 463.45,
-      "learning_rate": 7.693779904306222e-06,
-      "loss": 0.0288,
-      "step": 5098
-    },
-    {
-      "epoch": 463.55,
-      "learning_rate": 7.674641148325359e-06,
-      "loss": 0.0292,
-      "step": 5099
-    },
-    {
-      "epoch": 463.64,
-      "learning_rate": 7.655502392344498e-06,
-      "loss": 0.0336,
-      "step": 5100
-    },
-    {
-      "epoch": 463.64,
-      "eval_accuracy_NAT": 0.9930603540456359,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.952349081505644,
-      "eval_accuracy_concrete tiles": 0.9831835137283679,
-      "eval_accuracy_dark bricks": 0.8741387937338825,
-      "eval_accuracy_grass": 0.9039274143095307,
-      "eval_accuracy_red bricks": 0.927734375,
-      "eval_accuracy_road": 0.9356035826439835,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5061340390924379,
-      "eval_accuracy_speedway bricks": 0.9884562052648538,
-      "eval_accuracy_steel": 0.9097932194446277,
-      "eval_accuracy_tiles": 0.2520202711957266,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.980851314128064,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8913360858545599,
-      "eval_iou_concrete tiles": 0.9549490689870223,
-      "eval_iou_dark bricks": 0.832697807435653,
-      "eval_iou_grass": 0.8354891230609199,
-      "eval_iou_red bricks": 0.8667883211678832,
-      "eval_iou_road": 0.9279687595572511,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3398619004424369,
-      "eval_iou_speedway bricks": 0.9775846758696747,
-      "eval_iou_steel": 0.8562925829931867,
-      "eval_iou_tiles": 0.25076660988074956,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1046426072716713,
-      "eval_mean_accuracy": 0.7097231423049762,
-      "eval_mean_iou": 0.6703527884136462,
-      "eval_overall_accuracy": 0.9749694400363498,
-      "eval_runtime": 3.3903,
-      "eval_samples_per_second": 10.618,
-      "eval_steps_per_second": 0.59,
-      "step": 5100
-    },
-    {
-      "epoch": 463.73,
-      "learning_rate": 7.636363636363638e-06,
-      "loss": 0.0264,
-      "step": 5101
-    },
-    {
-      "epoch": 463.82,
-      "learning_rate": 7.617224880382775e-06,
-      "loss": 0.0297,
-      "step": 5102
-    },
-    {
-      "epoch": 463.91,
-      "learning_rate": 7.598086124401914e-06,
-      "loss": 0.0318,
-      "step": 5103
-    },
-    {
-      "epoch": 464.0,
-      "learning_rate": 7.578947368421054e-06,
-      "loss": 0.0286,
-      "step": 5104
-    },
-    {
-      "epoch": 464.09,
-      "learning_rate": 7.5598086124401915e-06,
-      "loss": 0.0295,
-      "step": 5105
-    },
-    {
-      "epoch": 464.18,
-      "learning_rate": 7.54066985645933e-06,
-      "loss": 0.0337,
-      "step": 5106
-    },
-    {
-      "epoch": 464.27,
-      "learning_rate": 7.52153110047847e-06,
-      "loss": 0.0297,
-      "step": 5107
-    },
-    {
-      "epoch": 464.36,
-      "learning_rate": 7.502392344497608e-06,
-      "loss": 0.0289,
-      "step": 5108
-    },
-    {
-      "epoch": 464.45,
-      "learning_rate": 7.483253588516747e-06,
-      "loss": 0.0289,
-      "step": 5109
-    },
-    {
-      "epoch": 464.55,
-      "learning_rate": 7.464114832535886e-06,
-      "loss": 0.0237,
-      "step": 5110
-    },
-    {
-      "epoch": 464.64,
-      "learning_rate": 7.444976076555024e-06,
-      "loss": 0.0294,
-      "step": 5111
-    },
-    {
-      "epoch": 464.73,
-      "learning_rate": 7.425837320574162e-06,
-      "loss": 0.028,
-      "step": 5112
-    },
-    {
-      "epoch": 464.82,
-      "learning_rate": 7.406698564593302e-06,
-      "loss": 0.0328,
-      "step": 5113
-    },
-    {
-      "epoch": 464.91,
-      "learning_rate": 7.3875598086124405e-06,
-      "loss": 0.0306,
-      "step": 5114
-    },
-    {
-      "epoch": 465.0,
-      "learning_rate": 7.3684210526315784e-06,
-      "loss": 0.0311,
-      "step": 5115
-    },
-    {
-      "epoch": 465.09,
-      "learning_rate": 7.349282296650718e-06,
-      "loss": 0.0343,
-      "step": 5116
-    },
-    {
-      "epoch": 465.18,
-      "learning_rate": 7.330143540669857e-06,
-      "loss": 0.0281,
-      "step": 5117
-    },
-    {
-      "epoch": 465.27,
-      "learning_rate": 7.3110047846889965e-06,
-      "loss": 0.0294,
-      "step": 5118
-    },
-    {
-      "epoch": 465.36,
-      "learning_rate": 7.291866028708134e-06,
-      "loss": 0.0297,
-      "step": 5119
-    },
-    {
-      "epoch": 465.45,
-      "learning_rate": 7.272727272727272e-06,
-      "loss": 0.029,
-      "step": 5120
-    },
-    {
-      "epoch": 465.45,
-      "eval_accuracy_NAT": 0.993097761682299,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9517790463110005,
-      "eval_accuracy_concrete tiles": 0.9864245092643552,
-      "eval_accuracy_dark bricks": 0.8792579192486817,
-      "eval_accuracy_grass": 0.902606276453339,
-      "eval_accuracy_red bricks": 0.9263139204545454,
-      "eval_accuracy_road": 0.9360280149506582,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47415432966669424,
-      "eval_accuracy_speedway bricks": 0.9886692082203452,
-      "eval_accuracy_steel": 0.9101477475100587,
-      "eval_accuracy_tiles": 0.24318586495000685,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806891444502527,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8907455432345787,
-      "eval_iou_concrete tiles": 0.9512044107674618,
-      "eval_iou_dark bricks": 0.8325072886297377,
-      "eval_iou_grass": 0.8349126295200685,
-      "eval_iou_red bricks": 0.8673316708229426,
-      "eval_iou_road": 0.9284888365387153,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3212004631531767,
-      "eval_iou_speedway bricks": 0.9775065789512933,
-      "eval_iou_steel": 0.856201039323664,
-      "eval_iou_tiles": 0.2421741799086135,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.1045546680688858,
-      "eval_mean_accuracy": 0.7070511229778448,
-      "eval_mean_iou": 0.667920137330808,
-      "eval_overall_accuracy": 0.9748738606770834,
-      "eval_runtime": 3.5549,
-      "eval_samples_per_second": 10.127,
-      "eval_steps_per_second": 0.563,
-      "step": 5120
-    },
-    {
-      "epoch": 465.55,
-      "learning_rate": 7.253588516746413e-06,
-      "loss": 0.0278,
-      "step": 5121
-    },
-    {
-      "epoch": 465.64,
-      "learning_rate": 7.234449760765551e-06,
-      "loss": 0.0328,
-      "step": 5122
-    },
-    {
-      "epoch": 465.73,
-      "learning_rate": 7.215311004784689e-06,
-      "loss": 0.0308,
-      "step": 5123
-    },
-    {
-      "epoch": 465.82,
-      "learning_rate": 7.196172248803828e-06,
-      "loss": 0.0321,
-      "step": 5124
-    },
-    {
-      "epoch": 465.91,
-      "learning_rate": 7.177033492822967e-06,
-      "loss": 0.0288,
-      "step": 5125
-    },
-    {
-      "epoch": 466.0,
-      "learning_rate": 7.157894736842105e-06,
-      "loss": 0.0298,
-      "step": 5126
-    },
-    {
-      "epoch": 466.09,
-      "learning_rate": 7.138755980861245e-06,
-      "loss": 0.0295,
-      "step": 5127
-    },
-    {
-      "epoch": 466.18,
-      "learning_rate": 7.119617224880383e-06,
-      "loss": 0.0271,
-      "step": 5128
-    },
-    {
-      "epoch": 466.27,
-      "learning_rate": 7.100478468899521e-06,
-      "loss": 0.0328,
-      "step": 5129
-    },
-    {
-      "epoch": 466.36,
-      "learning_rate": 7.081339712918661e-06,
-      "loss": 0.0331,
-      "step": 5130
-    },
-    {
-      "epoch": 466.45,
-      "learning_rate": 7.062200956937799e-06,
-      "loss": 0.0312,
-      "step": 5131
-    },
-    {
-      "epoch": 466.55,
-      "learning_rate": 7.043062200956938e-06,
-      "loss": 0.0282,
-      "step": 5132
-    },
-    {
-      "epoch": 466.64,
-      "learning_rate": 7.023923444976077e-06,
-      "loss": 0.0274,
-      "step": 5133
-    },
-    {
-      "epoch": 466.73,
-      "learning_rate": 7.004784688995215e-06,
-      "loss": 0.027,
-      "step": 5134
-    },
-    {
-      "epoch": 466.82,
-      "learning_rate": 6.985645933014355e-06,
-      "loss": 0.0274,
-      "step": 5135
-    },
-    {
-      "epoch": 466.91,
-      "learning_rate": 6.966507177033494e-06,
-      "loss": 0.0317,
-      "step": 5136
-    },
-    {
-      "epoch": 467.0,
-      "learning_rate": 6.9473684210526315e-06,
-      "loss": 0.0311,
-      "step": 5137
-    },
-    {
-      "epoch": 467.09,
-      "learning_rate": 6.928229665071771e-06,
-      "loss": 0.0325,
-      "step": 5138
-    },
-    {
-      "epoch": 467.18,
-      "learning_rate": 6.909090909090909e-06,
-      "loss": 0.0298,
-      "step": 5139
-    },
-    {
-      "epoch": 467.27,
-      "learning_rate": 6.889952153110048e-06,
-      "loss": 0.0287,
-      "step": 5140
-    },
-    {
-      "epoch": 467.27,
-      "eval_accuracy_NAT": 0.9932161603813603,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9527139040302159,
-      "eval_accuracy_concrete tiles": 0.9869137161377117,
-      "eval_accuracy_dark bricks": 0.8827989684769639,
-      "eval_accuracy_grass": 0.8981885033045922,
-      "eval_accuracy_red bricks": 0.9323508522727273,
-      "eval_accuracy_road": 0.9372517343077944,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.47823449948997876,
-      "eval_accuracy_speedway bricks": 0.98790891438488,
-      "eval_accuracy_steel": 0.9113267594485852,
-      "eval_accuracy_tiles": 0.2333242021640871,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806763663770552,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8907427401970558,
-      "eval_iou_concrete tiles": 0.9501074382597945,
-      "eval_iou_dark bricks": 0.8326738064984571,
-      "eval_iou_grass": 0.834243364670441,
-      "eval_iou_red bricks": 0.8672171758876961,
-      "eval_iou_road": 0.9290251824861263,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3237226141156272,
-      "eval_iou_speedway bricks": 0.9775602789112479,
-      "eval_iou_steel": 0.8556775252368861,
-      "eval_iou_tiles": 0.232464519650655,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10398384928703308,
-      "eval_mean_accuracy": 0.7072483241845305,
-      "eval_mean_iou": 0.6672393086377724,
-      "eval_overall_accuracy": 0.9748892254299588,
-      "eval_runtime": 3.9381,
-      "eval_samples_per_second": 9.141,
-      "eval_steps_per_second": 0.508,
-      "step": 5140
-    },
-    {
-      "epoch": 467.36,
-      "learning_rate": 6.8708133971291875e-06,
-      "loss": 0.0288,
-      "step": 5141
-    },
-    {
-      "epoch": 467.45,
-      "learning_rate": 6.851674641148325e-06,
-      "loss": 0.028,
-      "step": 5142
-    },
-    {
-      "epoch": 467.55,
-      "learning_rate": 6.832535885167464e-06,
-      "loss": 0.0272,
-      "step": 5143
-    },
-    {
-      "epoch": 467.64,
-      "learning_rate": 6.813397129186604e-06,
-      "loss": 0.0283,
-      "step": 5144
-    },
-    {
-      "epoch": 467.73,
-      "learning_rate": 6.794258373205742e-06,
-      "loss": 0.0331,
-      "step": 5145
-    },
-    {
-      "epoch": 467.82,
-      "learning_rate": 6.7751196172248805e-06,
-      "loss": 0.0301,
-      "step": 5146
-    },
-    {
-      "epoch": 467.91,
-      "learning_rate": 6.75598086124402e-06,
-      "loss": 0.0293,
-      "step": 5147
-    },
-    {
-      "epoch": 468.0,
-      "learning_rate": 6.736842105263158e-06,
-      "loss": 0.0279,
-      "step": 5148
-    },
-    {
-      "epoch": 468.09,
-      "learning_rate": 6.717703349282296e-06,
-      "loss": 0.0275,
-      "step": 5149
-    },
-    {
-      "epoch": 468.18,
-      "learning_rate": 6.698564593301436e-06,
-      "loss": 0.0264,
-      "step": 5150
-    },
-    {
-      "epoch": 468.27,
-      "learning_rate": 6.679425837320574e-06,
-      "loss": 0.0305,
-      "step": 5151
-    },
-    {
-      "epoch": 468.36,
-      "learning_rate": 6.660287081339714e-06,
-      "loss": 0.0336,
-      "step": 5152
-    },
-    {
-      "epoch": 468.45,
-      "learning_rate": 6.641148325358852e-06,
-      "loss": 0.0275,
-      "step": 5153
-    },
-    {
-      "epoch": 468.55,
-      "learning_rate": 6.622009569377991e-06,
-      "loss": 0.0255,
-      "step": 5154
-    },
-    {
-      "epoch": 468.64,
-      "learning_rate": 6.60287081339713e-06,
-      "loss": 0.0314,
-      "step": 5155
-    },
-    {
-      "epoch": 468.73,
-      "learning_rate": 6.583732057416268e-06,
-      "loss": 0.0303,
-      "step": 5156
-    },
-    {
-      "epoch": 468.82,
-      "learning_rate": 6.564593301435406e-06,
-      "loss": 0.0292,
-      "step": 5157
-    },
-    {
-      "epoch": 468.91,
-      "learning_rate": 6.545454545454547e-06,
-      "loss": 0.0297,
-      "step": 5158
-    },
-    {
-      "epoch": 469.0,
-      "learning_rate": 6.526315789473685e-06,
-      "loss": 0.0357,
-      "step": 5159
-    },
-    {
-      "epoch": 469.09,
-      "learning_rate": 6.5071770334928226e-06,
-      "loss": 0.0264,
-      "step": 5160
-    },
-    {
-      "epoch": 469.09,
-      "eval_accuracy_NAT": 0.9929959493881734,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9515550560109876,
-      "eval_accuracy_concrete tiles": 0.9874334984406531,
-      "eval_accuracy_dark bricks": 0.8816442785112197,
-      "eval_accuracy_grass": 0.8997053652876271,
-      "eval_accuracy_red bricks": 0.9300426136363636,
-      "eval_accuracy_road": 0.9333943580733434,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.4850715408154826,
-      "eval_accuracy_speedway bricks": 0.9889341042471462,
-      "eval_accuracy_steel": 0.909908647186861,
-      "eval_accuracy_tiles": 0.23969319271332695,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807453116310796,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8896525779207344,
-      "eval_iou_concrete tiles": 0.9521773741781407,
-      "eval_iou_dark bricks": 0.8318866896676956,
-      "eval_iou_grass": 0.8332988042134158,
-      "eval_iou_red bricks": 0.8670749875848369,
-      "eval_iou_road": 0.9257131686912139,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3274463095991365,
-      "eval_iou_speedway bricks": 0.9775282105176516,
-      "eval_iou_steel": 0.8556575540014576,
-      "eval_iou_tiles": 0.23841961852861035,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10607215017080307,
-      "eval_mean_accuracy": 0.7077214311008604,
-      "eval_mean_iou": 0.667661585117998,
-      "eval_overall_accuracy": 0.9745685789320204,
-      "eval_runtime": 3.587,
-      "eval_samples_per_second": 10.036,
-      "eval_steps_per_second": 0.558,
-      "step": 5160
-    },
-    {
-      "epoch": 469.18,
-      "learning_rate": 6.488038277511962e-06,
-      "loss": 0.0323,
-      "step": 5161
-    },
-    {
-      "epoch": 469.27,
-      "learning_rate": 6.468899521531101e-06,
-      "loss": 0.0302,
-      "step": 5162
-    },
-    {
-      "epoch": 469.36,
-      "learning_rate": 6.449760765550239e-06,
-      "loss": 0.0273,
-      "step": 5163
-    },
-    {
-      "epoch": 469.45,
-      "learning_rate": 6.4306220095693785e-06,
-      "loss": 0.0343,
-      "step": 5164
-    },
-    {
-      "epoch": 469.55,
-      "learning_rate": 6.411483253588517e-06,
-      "loss": 0.0266,
-      "step": 5165
-    },
-    {
-      "epoch": 469.64,
-      "learning_rate": 6.392344497607655e-06,
-      "loss": 0.0294,
-      "step": 5166
-    },
-    {
-      "epoch": 469.73,
-      "learning_rate": 6.373205741626795e-06,
-      "loss": 0.0284,
-      "step": 5167
-    },
-    {
-      "epoch": 469.82,
-      "learning_rate": 6.354066985645933e-06,
-      "loss": 0.0292,
-      "step": 5168
-    },
-    {
-      "epoch": 469.91,
-      "learning_rate": 6.334928229665072e-06,
-      "loss": 0.0308,
-      "step": 5169
-    },
-    {
-      "epoch": 470.0,
-      "learning_rate": 6.315789473684211e-06,
-      "loss": 0.0307,
-      "step": 5170
-    },
-    {
-      "epoch": 470.09,
-      "learning_rate": 6.296650717703349e-06,
-      "loss": 0.0298,
-      "step": 5171
-    },
-    {
-      "epoch": 470.18,
-      "learning_rate": 6.277511961722489e-06,
-      "loss": 0.027,
-      "step": 5172
-    },
-    {
-      "epoch": 470.27,
-      "learning_rate": 6.2583732057416275e-06,
-      "loss": 0.0383,
-      "step": 5173
-    },
-    {
-      "epoch": 470.36,
-      "learning_rate": 6.239234449760766e-06,
-      "loss": 0.0309,
-      "step": 5174
-    },
-    {
-      "epoch": 470.45,
-      "learning_rate": 6.220095693779904e-06,
-      "loss": 0.029,
-      "step": 5175
-    },
-    {
-      "epoch": 470.55,
-      "learning_rate": 6.200956937799043e-06,
-      "loss": 0.0317,
-      "step": 5176
-    },
-    {
-      "epoch": 470.64,
-      "learning_rate": 6.181818181818183e-06,
-      "loss": 0.0302,
-      "step": 5177
-    },
-    {
-      "epoch": 470.73,
-      "learning_rate": 6.1626794258373205e-06,
-      "loss": 0.0271,
-      "step": 5178
-    },
-    {
-      "epoch": 470.82,
-      "learning_rate": 6.143540669856459e-06,
-      "loss": 0.0289,
-      "step": 5179
-    },
-    {
-      "epoch": 470.91,
-      "learning_rate": 6.124401913875599e-06,
-      "loss": 0.0283,
-      "step": 5180
-    },
-    {
-      "epoch": 470.91,
-      "eval_accuracy_NAT": 0.9931120542227222,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9509756320013735,
-      "eval_accuracy_concrete tiles": 0.9864245092643552,
-      "eval_accuracy_dark bricks": 0.8847234517532042,
-      "eval_accuracy_grass": 0.8997228406560952,
-      "eval_accuracy_red bricks": 0.9293323863636364,
-      "eval_accuracy_road": 0.9353448119498855,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5075951809886141,
-      "eval_accuracy_speedway bricks": 0.9888810043602249,
-      "eval_accuracy_steel": 0.9093150187982323,
-      "eval_accuracy_tiles": 0.24606218326256676,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806931763552178,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8914169543690956,
-      "eval_iou_concrete tiles": 0.9523275378575434,
-      "eval_iou_dark bricks": 0.8319820471984943,
-      "eval_iou_grass": 0.8333635480738103,
-      "eval_iou_red bricks": 0.8681373362083264,
-      "eval_iou_road": 0.9276253224427338,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.341601885007143,
-      "eval_iou_speedway bricks": 0.9774628969237386,
-      "eval_iou_steel": 0.8551921466455755,
-      "eval_iou_tiles": 0.24500511421752472,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10369506478309631,
-      "eval_mean_accuracy": 0.7101145441246854,
-      "eval_mean_iou": 0.6696006127153235,
-      "eval_overall_accuracy": 0.9748480055067275,
-      "eval_runtime": 3.633,
-      "eval_samples_per_second": 9.909,
-      "eval_steps_per_second": 0.551,
-      "step": 5180
-    },
-    {
-      "epoch": 471.0,
-      "learning_rate": 6.105263157894737e-06,
-      "loss": 0.0286,
-      "step": 5181
-    },
-    {
-      "epoch": 471.09,
-      "learning_rate": 6.086124401913876e-06,
-      "loss": 0.0305,
-      "step": 5182
-    },
-    {
-      "epoch": 471.18,
-      "learning_rate": 6.0669856459330144e-06,
-      "loss": 0.0254,
-      "step": 5183
-    },
-    {
-      "epoch": 471.27,
-      "learning_rate": 6.047846889952154e-06,
-      "loss": 0.0288,
-      "step": 5184
-    },
-    {
-      "epoch": 471.36,
-      "learning_rate": 6.028708133971292e-06,
-      "loss": 0.0268,
-      "step": 5185
-    },
-    {
-      "epoch": 471.45,
-      "learning_rate": 6.009569377990431e-06,
-      "loss": 0.0277,
-      "step": 5186
-    },
-    {
-      "epoch": 471.55,
-      "learning_rate": 5.9904306220095695e-06,
-      "loss": 0.0251,
-      "step": 5187
-    },
-    {
-      "epoch": 471.64,
-      "learning_rate": 5.971291866028708e-06,
-      "loss": 0.0314,
-      "step": 5188
-    },
-    {
-      "epoch": 471.73,
-      "learning_rate": 5.952153110047847e-06,
-      "loss": 0.0345,
-      "step": 5189
-    },
-    {
-      "epoch": 471.82,
-      "learning_rate": 5.933014354066986e-06,
-      "loss": 0.0297,
-      "step": 5190
-    },
-    {
-      "epoch": 471.91,
-      "learning_rate": 5.913875598086125e-06,
-      "loss": 0.0286,
-      "step": 5191
-    },
-    {
-      "epoch": 472.0,
-      "learning_rate": 5.8947368421052634e-06,
-      "loss": 0.0347,
-      "step": 5192
-    },
-    {
-      "epoch": 472.09,
-      "learning_rate": 5.875598086124402e-06,
-      "loss": 0.0284,
-      "step": 5193
-    },
-    {
-      "epoch": 472.18,
-      "learning_rate": 5.856459330143541e-06,
-      "loss": 0.0278,
-      "step": 5194
-    },
-    {
-      "epoch": 472.27,
-      "learning_rate": 5.83732057416268e-06,
-      "loss": 0.0293,
-      "step": 5195
-    },
-    {
-      "epoch": 472.36,
-      "learning_rate": 5.8181818181818185e-06,
-      "loss": 0.027,
-      "step": 5196
-    },
-    {
-      "epoch": 472.45,
-      "learning_rate": 5.799043062200957e-06,
-      "loss": 0.0255,
-      "step": 5197
-    },
-    {
-      "epoch": 472.55,
-      "learning_rate": 5.779904306220096e-06,
-      "loss": 0.0302,
-      "step": 5198
-    },
-    {
-      "epoch": 472.64,
-      "learning_rate": 5.760765550239235e-06,
-      "loss": 0.0303,
-      "step": 5199
-    },
-    {
-      "epoch": 472.73,
-      "learning_rate": 5.741626794258374e-06,
-      "loss": 0.0283,
-      "step": 5200
-    },
-    {
-      "epoch": 472.73,
-      "eval_accuracy_NAT": 0.9930781756083858,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9521291149834756,
-      "eval_accuracy_concrete tiles": 0.986730263560203,
-      "eval_accuracy_dark bricks": 0.8833378237943112,
-      "eval_accuracy_grass": 0.9004358356895955,
-      "eval_accuracy_red bricks": 0.9311079545454546,
-      "eval_accuracy_road": 0.935967554508112,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.5189534915777576,
-      "eval_accuracy_speedway bricks": 0.9884531882258242,
-      "eval_accuracy_steel": 0.9105022755754898,
-      "eval_accuracy_tiles": 0.25962196959320644,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808106269309271,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8918982758252431,
-      "eval_iou_concrete tiles": 0.9519202406937644,
-      "eval_iou_dark bricks": 0.832849470169836,
-      "eval_iou_grass": 0.834285936341292,
-      "eval_iou_red bricks": 0.8666336142786316,
-      "eval_iou_road": 0.9282529441393039,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.3487152887126952,
-      "eval_iou_speedway bricks": 0.9774994405740456,
-      "eval_iou_steel": 0.8545329330196856,
-      "eval_iou_tiles": 0.2584890222282831,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10386094450950623,
-      "eval_mean_accuracy": 0.7123321267432167,
-      "eval_mean_iou": 0.6712221379164391,
-      "eval_overall_accuracy": 0.974998050265842,
-      "eval_runtime": 3.7046,
-      "eval_samples_per_second": 9.718,
-      "eval_steps_per_second": 0.54,
-      "step": 5200
-    }
-  ],
-  "max_steps": 5500,
-  "num_train_epochs": 500,
-  "total_flos": 2.910793210886357e+18,
-  "trial_name": null,
-  "trial_params": null
-}