diff --git "a/checkpoint-1360/trainer_state.json" "b/checkpoint-1360/trainer_state.json"
deleted file mode 100644--- "a/checkpoint-1360/trainer_state.json"
+++ /dev/null
@@ -1,10828 +0,0 @@
-{
-  "best_metric": 0.07582816481590271,
-  "best_model_checkpoint": "/robodata/smodak/Projects/lifelong_concept_learner/scripts/terrainseg/training/models/safety-utcustom-terrain-final/checkpoint-1360",
-  "epoch": 80.0,
-  "global_step": 1360,
-  "is_hyper_param_search": false,
-  "is_local_process_zero": true,
-  "is_world_process_zero": true,
-  "log_history": [
-    {
-      "epoch": 0.06,
-      "learning_rate": 2.9411764705882356e-07,
-      "loss": 2.7081,
-      "step": 1
-    },
-    {
-      "epoch": 0.12,
-      "learning_rate": 5.882352941176471e-07,
-      "loss": 2.7218,
-      "step": 2
-    },
-    {
-      "epoch": 0.18,
-      "learning_rate": 8.823529411764706e-07,
-      "loss": 2.709,
-      "step": 3
-    },
-    {
-      "epoch": 0.24,
-      "learning_rate": 1.1764705882352942e-06,
-      "loss": 2.6939,
-      "step": 4
-    },
-    {
-      "epoch": 0.29,
-      "learning_rate": 1.4705882352941177e-06,
-      "loss": 2.6979,
-      "step": 5
-    },
-    {
-      "epoch": 0.35,
-      "learning_rate": 1.7647058823529412e-06,
-      "loss": 2.6788,
-      "step": 6
-    },
-    {
-      "epoch": 0.41,
-      "learning_rate": 2.058823529411765e-06,
-      "loss": 2.6769,
-      "step": 7
-    },
-    {
-      "epoch": 0.47,
-      "learning_rate": 2.3529411764705885e-06,
-      "loss": 2.6464,
-      "step": 8
-    },
-    {
-      "epoch": 0.53,
-      "learning_rate": 2.647058823529412e-06,
-      "loss": 2.6424,
-      "step": 9
-    },
-    {
-      "epoch": 0.59,
-      "learning_rate": 2.9411764705882355e-06,
-      "loss": 2.6198,
-      "step": 10
-    },
-    {
-      "epoch": 0.65,
-      "learning_rate": 3.2352941176470594e-06,
-      "loss": 2.6228,
-      "step": 11
-    },
-    {
-      "epoch": 0.71,
-      "learning_rate": 3.5294117647058825e-06,
-      "loss": 2.6158,
-      "step": 12
-    },
-    {
-      "epoch": 0.76,
-      "learning_rate": 3.823529411764706e-06,
-      "loss": 2.632,
-      "step": 13
-    },
-    {
-      "epoch": 0.82,
-      "learning_rate": 4.11764705882353e-06,
-      "loss": 2.587,
-      "step": 14
-    },
-    {
-      "epoch": 0.88,
-      "learning_rate": 4.411764705882353e-06,
-      "loss": 2.5767,
-      "step": 15
-    },
-    {
-      "epoch": 0.94,
-      "learning_rate": 4.705882352941177e-06,
-      "loss": 2.5638,
-      "step": 16
-    },
-    {
-      "epoch": 1.0,
-      "learning_rate": 5e-06,
-      "loss": 2.517,
-      "step": 17
-    },
-    {
-      "epoch": 1.06,
-      "learning_rate": 5.294117647058824e-06,
-      "loss": 2.5202,
-      "step": 18
-    },
-    {
-      "epoch": 1.12,
-      "learning_rate": 5.588235294117647e-06,
-      "loss": 2.5257,
-      "step": 19
-    },
-    {
-      "epoch": 1.18,
-      "learning_rate": 5.882352941176471e-06,
-      "loss": 2.51,
-      "step": 20
-    },
-    {
-      "epoch": 1.18,
-      "eval_accuracy_NAT": 0.0901846148886272,
-      "eval_accuracy_REST": 0.1772357723577236,
-      "eval_accuracy_concrete": 0.21836230658804573,
-      "eval_accuracy_concrete tiles": 0.10897949563987744,
-      "eval_accuracy_dark bricks": 0.01268591007407691,
-      "eval_accuracy_grass": 0.027127403891645564,
-      "eval_accuracy_red bricks": 0.0003844463971880492,
-      "eval_accuracy_road": 0.0070585682671182835,
-      "eval_accuracy_rough concrete": 0.002699336785465293,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.6741853129093485,
-      "eval_accuracy_steel": 0.0007721641086127534,
-      "eval_accuracy_tiles": 0.2798933624163107,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.09010334674005777,
-      "eval_iou_REST": 2.227093222444523e-05,
-      "eval_iou_concrete": 0.07965033823417267,
-      "eval_iou_concrete tiles": 0.0011714161291131616,
-      "eval_iou_dark bricks": 0.0030587532148061655,
-      "eval_iou_grass": 0.019163923774789638,
-      "eval_iou_red bricks": 5.560869275085081e-06,
-      "eval_iou_road": 0.006794427417222495,
-      "eval_iou_rough concrete": 0.00013145523557474666,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.4632836614619225,
-      "eval_iou_steel": 0.0006867765509148131,
-      "eval_iou_tiles": 0.0032256814378599824,
-      "eval_iou_unlabeled": 0.0,
-      "eval_loss": 2.507078170776367,
-      "eval_mean_accuracy": 0.12304374571723385,
-      "eval_mean_iou": 0.04766411514270953,
-      "eval_overall_accuracy": 0.20164948802883342,
-      "eval_runtime": 18.8176,
-      "eval_samples_per_second": 6.271,
-      "eval_steps_per_second": 0.425,
-      "step": 20
-    },
-    {
-      "epoch": 1.24,
-      "learning_rate": 6.1764705882352944e-06,
-      "loss": 2.4771,
-      "step": 21
-    },
-    {
-      "epoch": 1.29,
-      "learning_rate": 6.470588235294119e-06,
-      "loss": 2.4758,
-      "step": 22
-    },
-    {
-      "epoch": 1.35,
-      "learning_rate": 6.7647058823529414e-06,
-      "loss": 2.4283,
-      "step": 23
-    },
-    {
-      "epoch": 1.41,
-      "learning_rate": 7.058823529411765e-06,
-      "loss": 2.3783,
-      "step": 24
-    },
-    {
-      "epoch": 1.47,
-      "learning_rate": 7.3529411764705884e-06,
-      "loss": 2.3059,
-      "step": 25
-    },
-    {
-      "epoch": 1.53,
-      "learning_rate": 7.647058823529413e-06,
-      "loss": 2.3405,
-      "step": 26
-    },
-    {
-      "epoch": 1.59,
-      "learning_rate": 7.941176470588235e-06,
-      "loss": 2.2917,
-      "step": 27
-    },
-    {
-      "epoch": 1.65,
-      "learning_rate": 8.23529411764706e-06,
-      "loss": 2.2857,
-      "step": 28
-    },
-    {
-      "epoch": 1.71,
-      "learning_rate": 8.529411764705883e-06,
-      "loss": 2.1291,
-      "step": 29
-    },
-    {
-      "epoch": 1.76,
-      "learning_rate": 8.823529411764707e-06,
-      "loss": 2.1234,
-      "step": 30
-    },
-    {
-      "epoch": 1.82,
-      "learning_rate": 9.117647058823529e-06,
-      "loss": 2.0916,
-      "step": 31
-    },
-    {
-      "epoch": 1.88,
-      "learning_rate": 9.411764705882354e-06,
-      "loss": 2.0543,
-      "step": 32
-    },
-    {
-      "epoch": 1.94,
-      "learning_rate": 9.705882352941177e-06,
-      "loss": 1.9605,
-      "step": 33
-    },
-    {
-      "epoch": 2.0,
-      "learning_rate": 1e-05,
-      "loss": 1.9835,
-      "step": 34
-    },
-    {
-      "epoch": 2.06,
-      "learning_rate": 1.0294117647058824e-05,
-      "loss": 2.0005,
-      "step": 35
-    },
-    {
-      "epoch": 2.12,
-      "learning_rate": 1.0588235294117648e-05,
-      "loss": 1.8628,
-      "step": 36
-    },
-    {
-      "epoch": 2.18,
-      "learning_rate": 1.0882352941176471e-05,
-      "loss": 1.8321,
-      "step": 37
-    },
-    {
-      "epoch": 2.24,
-      "learning_rate": 1.1176470588235295e-05,
-      "loss": 1.7578,
-      "step": 38
-    },
-    {
-      "epoch": 2.29,
-      "learning_rate": 1.1470588235294118e-05,
-      "loss": 1.7233,
-      "step": 39
-    },
-    {
-      "epoch": 2.35,
-      "learning_rate": 1.1764705882352942e-05,
-      "loss": 1.5504,
-      "step": 40
-    },
-    {
-      "epoch": 2.35,
-      "eval_accuracy_NAT": 0.760154401121494,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.3386920222776326,
-      "eval_accuracy_concrete tiles": 0.007895357058684892,
-      "eval_accuracy_dark bricks": 0.0014308136975288556,
-      "eval_accuracy_grass": 0.07454476982063306,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.3332036483520477,
-      "eval_accuracy_rough concrete": 0.0009832021404012656,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9879532008612127,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 0.00875518797903602,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.7577068331202973,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.2037716446695068,
-      "eval_iou_concrete tiles": 0.00033778197232406094,
-      "eval_iou_dark bricks": 0.0009957339958239116,
-      "eval_iou_grass": 0.06824458242953867,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.31270597640653264,
-      "eval_iou_rough concrete": 0.00024414351495637967,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.571054819328365,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 0.0007685393497465681,
-      "eval_iou_unlabeled": 0.0,
-      "eval_loss": 1.840782642364502,
-      "eval_mean_accuracy": 0.1933548156391286,
-      "eval_mean_iou": 0.13684500391336366,
-      "eval_overall_accuracy": 0.6935860908637612,
-      "eval_runtime": 7.985,
-      "eval_samples_per_second": 14.778,
-      "eval_steps_per_second": 1.002,
-      "step": 40
-    },
-    {
-      "epoch": 2.41,
-      "learning_rate": 1.2058823529411765e-05,
-      "loss": 1.7505,
-      "step": 41
-    },
-    {
-      "epoch": 2.47,
-      "learning_rate": 1.2352941176470589e-05,
-      "loss": 1.4808,
-      "step": 42
-    },
-    {
-      "epoch": 2.53,
-      "learning_rate": 1.2647058823529412e-05,
-      "loss": 1.4041,
-      "step": 43
-    },
-    {
-      "epoch": 2.59,
-      "learning_rate": 1.2941176470588238e-05,
-      "loss": 1.3261,
-      "step": 44
-    },
-    {
-      "epoch": 2.65,
-      "learning_rate": 1.323529411764706e-05,
-      "loss": 1.3337,
-      "step": 45
-    },
-    {
-      "epoch": 2.71,
-      "learning_rate": 1.3529411764705883e-05,
-      "loss": 1.3177,
-      "step": 46
-    },
-    {
-      "epoch": 2.76,
-      "learning_rate": 1.3823529411764708e-05,
-      "loss": 1.2551,
-      "step": 47
-    },
-    {
-      "epoch": 2.82,
-      "learning_rate": 1.411764705882353e-05,
-      "loss": 1.119,
-      "step": 48
-    },
-    {
-      "epoch": 2.88,
-      "learning_rate": 1.4411764705882352e-05,
-      "loss": 1.1259,
-      "step": 49
-    },
-    {
-      "epoch": 2.94,
-      "learning_rate": 1.4705882352941177e-05,
-      "loss": 1.0199,
-      "step": 50
-    },
-    {
-      "epoch": 3.0,
-      "learning_rate": 1.5e-05,
-      "loss": 1.1612,
-      "step": 51
-    },
-    {
-      "epoch": 3.06,
-      "learning_rate": 1.5294117647058826e-05,
-      "loss": 0.8821,
-      "step": 52
-    },
-    {
-      "epoch": 3.12,
-      "learning_rate": 1.558823529411765e-05,
-      "loss": 0.8667,
-      "step": 53
-    },
-    {
-      "epoch": 3.18,
-      "learning_rate": 1.588235294117647e-05,
-      "loss": 0.8607,
-      "step": 54
-    },
-    {
-      "epoch": 3.24,
-      "learning_rate": 1.6176470588235296e-05,
-      "loss": 0.8021,
-      "step": 55
-    },
-    {
-      "epoch": 3.29,
-      "learning_rate": 1.647058823529412e-05,
-      "loss": 0.7418,
-      "step": 56
-    },
-    {
-      "epoch": 3.35,
-      "learning_rate": 1.676470588235294e-05,
-      "loss": 0.7114,
-      "step": 57
-    },
-    {
-      "epoch": 3.41,
-      "learning_rate": 1.7058823529411767e-05,
-      "loss": 0.7799,
-      "step": 58
-    },
-    {
-      "epoch": 3.47,
-      "learning_rate": 1.735294117647059e-05,
-      "loss": 0.6974,
-      "step": 59
-    },
-    {
-      "epoch": 3.53,
-      "learning_rate": 1.7647058823529414e-05,
-      "loss": 0.6271,
-      "step": 60
-    },
-    {
-      "epoch": 3.53,
-      "eval_accuracy_NAT": 0.9551305368117847,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8126940616930818,
-      "eval_accuracy_concrete tiles": 0.0,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.25898348643298624,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.8978859032247204,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9793115987478005,
-      "eval_accuracy_steel": 0.0,
-      "eval_accuracy_tiles": 3.0294768093550245e-05,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9404775296004152,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.5259192497701041,
-      "eval_iou_concrete tiles": 0.0,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.2391512912549761,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.6993177996437074,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8262999758534085,
-      "eval_iou_steel": 0.0,
-      "eval_iou_tiles": 2.891343318105592e-05,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.8082596063613892,
-      "eval_mean_accuracy": 0.30031045243680515,
-      "eval_mean_iou": 0.24855344304275326,
-      "eval_overall_accuracy": 0.8893904281874835,
-      "eval_runtime": 6.7648,
-      "eval_samples_per_second": 17.443,
-      "eval_steps_per_second": 1.183,
-      "step": 60
-    },
-    {
-      "epoch": 3.59,
-      "learning_rate": 1.7941176470588237e-05,
-      "loss": 0.6901,
-      "step": 61
-    },
-    {
-      "epoch": 3.65,
-      "learning_rate": 1.8235294117647057e-05,
-      "loss": 0.5778,
-      "step": 62
-    },
-    {
-      "epoch": 3.71,
-      "learning_rate": 1.8529411764705884e-05,
-      "loss": 0.6319,
-      "step": 63
-    },
-    {
-      "epoch": 3.76,
-      "learning_rate": 1.8823529411764708e-05,
-      "loss": 0.5626,
-      "step": 64
-    },
-    {
-      "epoch": 3.82,
-      "learning_rate": 1.9117647058823528e-05,
-      "loss": 0.7374,
-      "step": 65
-    },
-    {
-      "epoch": 3.88,
-      "learning_rate": 1.9411764705882355e-05,
-      "loss": 0.6702,
-      "step": 66
-    },
-    {
-      "epoch": 3.94,
-      "learning_rate": 1.9705882352941178e-05,
-      "loss": 0.5477,
-      "step": 67
-    },
-    {
-      "epoch": 4.0,
-      "learning_rate": 2e-05,
-      "loss": 0.586,
-      "step": 68
-    },
-    {
-      "epoch": 4.06,
-      "learning_rate": 2.0294117647058825e-05,
-      "loss": 0.505,
-      "step": 69
-    },
-    {
-      "epoch": 4.12,
-      "learning_rate": 2.058823529411765e-05,
-      "loss": 0.5758,
-      "step": 70
-    },
-    {
-      "epoch": 4.18,
-      "learning_rate": 2.0882352941176472e-05,
-      "loss": 0.5694,
-      "step": 71
-    },
-    {
-      "epoch": 4.24,
-      "learning_rate": 2.1176470588235296e-05,
-      "loss": 0.4535,
-      "step": 72
-    },
-    {
-      "epoch": 4.29,
-      "learning_rate": 2.1470588235294116e-05,
-      "loss": 0.5105,
-      "step": 73
-    },
-    {
-      "epoch": 4.35,
-      "learning_rate": 2.1764705882352943e-05,
-      "loss": 0.5525,
-      "step": 74
-    },
-    {
-      "epoch": 4.41,
-      "learning_rate": 2.2058823529411766e-05,
-      "loss": 0.465,
-      "step": 75
-    },
-    {
-      "epoch": 4.47,
-      "learning_rate": 2.235294117647059e-05,
-      "loss": 0.5477,
-      "step": 76
-    },
-    {
-      "epoch": 4.53,
-      "learning_rate": 2.2647058823529413e-05,
-      "loss": 0.3823,
-      "step": 77
-    },
-    {
-      "epoch": 4.59,
-      "learning_rate": 2.2941176470588237e-05,
-      "loss": 0.4698,
-      "step": 78
-    },
-    {
-      "epoch": 4.65,
-      "learning_rate": 2.323529411764706e-05,
-      "loss": 0.4636,
-      "step": 79
-    },
-    {
-      "epoch": 4.71,
-      "learning_rate": 2.3529411764705884e-05,
-      "loss": 0.3963,
-      "step": 80
-    },
-    {
-      "epoch": 4.71,
-      "eval_accuracy_NAT": 0.969774210544504,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.7972872258293342,
-      "eval_accuracy_concrete tiles": 0.04426113598868725,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7826429655535243,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9650914038630805,
-      "eval_accuracy_rough concrete": 0.0,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9776932262095467,
-      "eval_accuracy_steel": 0.0003548456421456666,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9531585041831687,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6900984540650801,
-      "eval_iou_concrete tiles": 0.04426113598868725,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.5632131076991799,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.727967248165449,
-      "eval_iou_rough concrete": 0.0,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.8641130344685216,
-      "eval_iou_steel": 0.00035449307490288885,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.4204064607620239,
-      "eval_mean_accuracy": 0.34900807797160177,
-      "eval_mean_iou": 0.29562815212653765,
-      "eval_overall_accuracy": 0.9185845003289691,
-      "eval_runtime": 5.4734,
-      "eval_samples_per_second": 21.559,
-      "eval_steps_per_second": 1.462,
-      "step": 80
-    },
-    {
-      "epoch": 4.76,
-      "learning_rate": 2.3823529411764707e-05,
-      "loss": 0.3853,
-      "step": 81
-    },
-    {
-      "epoch": 4.82,
-      "learning_rate": 2.411764705882353e-05,
-      "loss": 0.378,
-      "step": 82
-    },
-    {
-      "epoch": 4.88,
-      "learning_rate": 2.4411764705882354e-05,
-      "loss": 0.4374,
-      "step": 83
-    },
-    {
-      "epoch": 4.94,
-      "learning_rate": 2.4705882352941178e-05,
-      "loss": 0.3066,
-      "step": 84
-    },
-    {
-      "epoch": 5.0,
-      "learning_rate": 2.5e-05,
-      "loss": 0.3689,
-      "step": 85
-    },
-    {
-      "epoch": 5.06,
-      "learning_rate": 2.5294117647058825e-05,
-      "loss": 0.3284,
-      "step": 86
-    },
-    {
-      "epoch": 5.12,
-      "learning_rate": 2.5588235294117645e-05,
-      "loss": 0.3295,
-      "step": 87
-    },
-    {
-      "epoch": 5.18,
-      "learning_rate": 2.5882352941176475e-05,
-      "loss": 0.3386,
-      "step": 88
-    },
-    {
-      "epoch": 5.24,
-      "learning_rate": 2.6176470588235295e-05,
-      "loss": 0.3517,
-      "step": 89
-    },
-    {
-      "epoch": 5.29,
-      "learning_rate": 2.647058823529412e-05,
-      "loss": 0.3205,
-      "step": 90
-    },
-    {
-      "epoch": 5.35,
-      "learning_rate": 2.6764705882352942e-05,
-      "loss": 0.4159,
-      "step": 91
-    },
-    {
-      "epoch": 5.41,
-      "learning_rate": 2.7058823529411766e-05,
-      "loss": 0.3534,
-      "step": 92
-    },
-    {
-      "epoch": 5.47,
-      "learning_rate": 2.7352941176470593e-05,
-      "loss": 0.325,
-      "step": 93
-    },
-    {
-      "epoch": 5.53,
-      "learning_rate": 2.7647058823529416e-05,
-      "loss": 0.2642,
-      "step": 94
-    },
-    {
-      "epoch": 5.59,
-      "learning_rate": 2.7941176470588236e-05,
-      "loss": 0.3485,
-      "step": 95
-    },
-    {
-      "epoch": 5.65,
-      "learning_rate": 2.823529411764706e-05,
-      "loss": 0.3246,
-      "step": 96
-    },
-    {
-      "epoch": 5.71,
-      "learning_rate": 2.8529411764705883e-05,
-      "loss": 0.4612,
-      "step": 97
-    },
-    {
-      "epoch": 5.76,
-      "learning_rate": 2.8823529411764703e-05,
-      "loss": 0.2731,
-      "step": 98
-    },
-    {
-      "epoch": 5.82,
-      "learning_rate": 2.9117647058823534e-05,
-      "loss": 0.4205,
-      "step": 99
-    },
-    {
-      "epoch": 5.88,
-      "learning_rate": 2.9411764705882354e-05,
-      "loss": 0.3558,
-      "step": 100
-    },
-    {
-      "epoch": 5.88,
-      "eval_accuracy_NAT": 0.9806404490547624,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9025246556766533,
-      "eval_accuracy_concrete tiles": 0.30502003299552205,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.7816418565745797,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.913452910724182,
-      "eval_accuracy_rough concrete": 0.0030807000399239656,
-      "eval_accuracy_rough red sidewalk": 0.0,
-      "eval_accuracy_speedway bricks": 0.9632305210940226,
-      "eval_accuracy_steel": 0.28737998970447853,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9617550029709626,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.6868428500250288,
-      "eval_iou_concrete tiles": 0.3027439238344757,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6345173453939056,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8305275811769873,
-      "eval_iou_rough concrete": 0.0030807000399239656,
-      "eval_iou_rough red sidewalk": 0.0,
-      "eval_iou_speedway bricks": 0.891543623735271,
-      "eval_iou_steel": 0.2804400149240265,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.30909788608551025,
-      "eval_mean_accuracy": 0.3951516242972403,
-      "eval_mean_iou": 0.35318854170004477,
-      "eval_overall_accuracy": 0.9322030665510792,
-      "eval_runtime": 5.2886,
-      "eval_samples_per_second": 22.312,
-      "eval_steps_per_second": 1.513,
-      "step": 100
-    },
-    {
-      "epoch": 5.94,
-      "learning_rate": 2.9705882352941177e-05,
-      "loss": 0.4249,
-      "step": 101
-    },
-    {
-      "epoch": 6.0,
-      "learning_rate": 3e-05,
-      "loss": 0.4043,
-      "step": 102
-    },
-    {
-      "epoch": 6.06,
-      "learning_rate": 3.0294117647058824e-05,
-      "loss": 0.3018,
-      "step": 103
-    },
-    {
-      "epoch": 6.12,
-      "learning_rate": 3.058823529411765e-05,
-      "loss": 0.3319,
-      "step": 104
-    },
-    {
-      "epoch": 6.18,
-      "learning_rate": 3.0882352941176475e-05,
-      "loss": 0.3451,
-      "step": 105
-    },
-    {
-      "epoch": 6.24,
-      "learning_rate": 3.11764705882353e-05,
-      "loss": 0.3506,
-      "step": 106
-    },
-    {
-      "epoch": 6.29,
-      "learning_rate": 3.147058823529412e-05,
-      "loss": 0.3709,
-      "step": 107
-    },
-    {
-      "epoch": 6.35,
-      "learning_rate": 3.176470588235294e-05,
-      "loss": 0.2675,
-      "step": 108
-    },
-    {
-      "epoch": 6.41,
-      "learning_rate": 3.205882352941177e-05,
-      "loss": 0.3261,
-      "step": 109
-    },
-    {
-      "epoch": 6.47,
-      "learning_rate": 3.235294117647059e-05,
-      "loss": 0.2528,
-      "step": 110
-    },
-    {
-      "epoch": 6.53,
-      "learning_rate": 3.2647058823529416e-05,
-      "loss": 0.2736,
-      "step": 111
-    },
-    {
-      "epoch": 6.59,
-      "learning_rate": 3.294117647058824e-05,
-      "loss": 0.3192,
-      "step": 112
-    },
-    {
-      "epoch": 6.65,
-      "learning_rate": 3.3235294117647056e-05,
-      "loss": 0.2764,
-      "step": 113
-    },
-    {
-      "epoch": 6.71,
-      "learning_rate": 3.352941176470588e-05,
-      "loss": 0.2485,
-      "step": 114
-    },
-    {
-      "epoch": 6.76,
-      "learning_rate": 3.382352941176471e-05,
-      "loss": 0.295,
-      "step": 115
-    },
-    {
-      "epoch": 6.82,
-      "learning_rate": 3.411764705882353e-05,
-      "loss": 0.2182,
-      "step": 116
-    },
-    {
-      "epoch": 6.88,
-      "learning_rate": 3.441176470588236e-05,
-      "loss": 0.2945,
-      "step": 117
-    },
-    {
-      "epoch": 6.94,
-      "learning_rate": 3.470588235294118e-05,
-      "loss": 0.2284,
-      "step": 118
-    },
-    {
-      "epoch": 7.0,
-      "learning_rate": 3.5e-05,
-      "loss": 0.3391,
-      "step": 119
-    },
-    {
-      "epoch": 7.06,
-      "learning_rate": 3.529411764705883e-05,
-      "loss": 0.2654,
-      "step": 120
-    },
-    {
-      "epoch": 7.06,
-      "eval_accuracy_NAT": 0.9793312019858504,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8161395883560691,
-      "eval_accuracy_concrete tiles": 0.38750883808625974,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.8487237279860109,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9769372718377128,
-      "eval_accuracy_rough concrete": 0.00018472282637841962,
-      "eval_accuracy_rough red sidewalk": 0.0009475180291624993,
-      "eval_accuracy_speedway bricks": 0.9861728882500396,
-      "eval_accuracy_steel": 0.4149295056650357,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9635845098252436,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7301235451822234,
-      "eval_iou_concrete tiles": 0.30457172495554236,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.649146550195586,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.7846078239913129,
-      "eval_iou_rough concrete": 0.0001839500605255038,
-      "eval_iou_rough red sidewalk": 0.0009475180291624993,
-      "eval_iou_speedway bricks": 0.9013550388480807,
-      "eval_iou_steel": 0.40151082824726514,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.25543102622032166,
-      "eval_mean_accuracy": 0.41622117407865533,
-      "eval_mean_iou": 0.3643101145642263,
-      "eval_overall_accuracy": 0.9363230042538401,
-      "eval_runtime": 5.3581,
-      "eval_samples_per_second": 22.023,
-      "eval_steps_per_second": 1.493,
-      "step": 120
-    },
-    {
-      "epoch": 7.12,
-      "learning_rate": 3.558823529411765e-05,
-      "loss": 0.3335,
-      "step": 121
-    },
-    {
-      "epoch": 7.18,
-      "learning_rate": 3.5882352941176474e-05,
-      "loss": 0.2502,
-      "step": 122
-    },
-    {
-      "epoch": 7.24,
-      "learning_rate": 3.61764705882353e-05,
-      "loss": 0.2615,
-      "step": 123
-    },
-    {
-      "epoch": 7.29,
-      "learning_rate": 3.6470588235294114e-05,
-      "loss": 0.2804,
-      "step": 124
-    },
-    {
-      "epoch": 7.35,
-      "learning_rate": 3.6764705882352945e-05,
-      "loss": 0.2703,
-      "step": 125
-    },
-    {
-      "epoch": 7.41,
-      "learning_rate": 3.705882352941177e-05,
-      "loss": 0.2235,
-      "step": 126
-    },
-    {
-      "epoch": 7.47,
-      "learning_rate": 3.735294117647059e-05,
-      "loss": 0.2855,
-      "step": 127
-    },
-    {
-      "epoch": 7.53,
-      "learning_rate": 3.7647058823529415e-05,
-      "loss": 0.2263,
-      "step": 128
-    },
-    {
-      "epoch": 7.59,
-      "learning_rate": 3.794117647058824e-05,
-      "loss": 0.1922,
-      "step": 129
-    },
-    {
-      "epoch": 7.65,
-      "learning_rate": 3.8235294117647055e-05,
-      "loss": 0.2025,
-      "step": 130
-    },
-    {
-      "epoch": 7.71,
-      "learning_rate": 3.8529411764705886e-05,
-      "loss": 0.2237,
-      "step": 131
-    },
-    {
-      "epoch": 7.76,
-      "learning_rate": 3.882352941176471e-05,
-      "loss": 0.2142,
-      "step": 132
-    },
-    {
-      "epoch": 7.82,
-      "learning_rate": 3.911764705882353e-05,
-      "loss": 0.2622,
-      "step": 133
-    },
-    {
-      "epoch": 7.88,
-      "learning_rate": 3.9411764705882356e-05,
-      "loss": 0.2454,
-      "step": 134
-    },
-    {
-      "epoch": 7.94,
-      "learning_rate": 3.970588235294117e-05,
-      "loss": 0.2398,
-      "step": 135
-    },
-    {
-      "epoch": 8.0,
-      "learning_rate": 4e-05,
-      "loss": 0.3103,
-      "step": 136
-    },
-    {
-      "epoch": 8.06,
-      "learning_rate": 4.029411764705883e-05,
-      "loss": 0.2222,
-      "step": 137
-    },
-    {
-      "epoch": 8.12,
-      "learning_rate": 4.058823529411765e-05,
-      "loss": 0.2128,
-      "step": 138
-    },
-    {
-      "epoch": 8.18,
-      "learning_rate": 4.0882352941176474e-05,
-      "loss": 0.2087,
-      "step": 139
-    },
-    {
-      "epoch": 8.24,
-      "learning_rate": 4.11764705882353e-05,
-      "loss": 0.1938,
-      "step": 140
-    },
-    {
-      "epoch": 8.24,
-      "eval_accuracy_NAT": 0.9830271765403628,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9010176438483495,
-      "eval_accuracy_concrete tiles": 0.6649304737214236,
-      "eval_accuracy_dark bricks": 0.0,
-      "eval_accuracy_grass": 0.8719394262831794,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9517197778325257,
-      "eval_accuracy_rough concrete": 0.1663399257533414,
-      "eval_accuracy_rough red sidewalk": 0.03742696215191872,
-      "eval_accuracy_speedway bricks": 0.9778834415332932,
-      "eval_accuracy_steel": 0.644497143742472,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9680687096130417,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7740222344798603,
-      "eval_iou_concrete tiles": 0.18868164278023367,
-      "eval_iou_dark bricks": 0.0,
-      "eval_iou_grass": 0.6641972532280409,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.8746486517016726,
-      "eval_iou_rough concrete": 0.16600853979090596,
-      "eval_iou_rough red sidewalk": 0.03742696215191872,
-      "eval_iou_speedway bricks": 0.9354015193799212,
-      "eval_iou_steel": 0.5996131393140197,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.2058570384979248,
-      "eval_mean_accuracy": 0.4768293824159129,
-      "eval_mean_iou": 0.4006206655722781,
-      "eval_overall_accuracy": 0.9478311700336004,
-      "eval_runtime": 5.5104,
-      "eval_samples_per_second": 21.414,
-      "eval_steps_per_second": 1.452,
-      "step": 140
-    },
-    {
-      "epoch": 8.29,
-      "learning_rate": 4.147058823529412e-05,
-      "loss": 0.2178,
-      "step": 141
-    },
-    {
-      "epoch": 8.35,
-      "learning_rate": 4.1764705882352944e-05,
-      "loss": 0.2016,
-      "step": 142
-    },
-    {
-      "epoch": 8.41,
-      "learning_rate": 4.205882352941177e-05,
-      "loss": 0.291,
-      "step": 143
-    },
-    {
-      "epoch": 8.47,
-      "learning_rate": 4.235294117647059e-05,
-      "loss": 0.2516,
-      "step": 144
-    },
-    {
-      "epoch": 8.53,
-      "learning_rate": 4.2647058823529415e-05,
-      "loss": 0.2561,
-      "step": 145
-    },
-    {
-      "epoch": 8.59,
-      "learning_rate": 4.294117647058823e-05,
-      "loss": 0.2003,
-      "step": 146
-    },
-    {
-      "epoch": 8.65,
-      "learning_rate": 4.323529411764706e-05,
-      "loss": 0.2749,
-      "step": 147
-    },
-    {
-      "epoch": 8.71,
-      "learning_rate": 4.3529411764705885e-05,
-      "loss": 0.2009,
-      "step": 148
-    },
-    {
-      "epoch": 8.76,
-      "learning_rate": 4.382352941176471e-05,
-      "loss": 0.2451,
-      "step": 149
-    },
-    {
-      "epoch": 8.82,
-      "learning_rate": 4.411764705882353e-05,
-      "loss": 0.1868,
-      "step": 150
-    },
-    {
-      "epoch": 8.88,
-      "learning_rate": 4.4411764705882356e-05,
-      "loss": 0.1741,
-      "step": 151
-    },
-    {
-      "epoch": 8.94,
-      "learning_rate": 4.470588235294118e-05,
-      "loss": 0.1841,
-      "step": 152
-    },
-    {
-      "epoch": 9.0,
-      "learning_rate": 4.5e-05,
-      "loss": 0.1486,
-      "step": 153
-    },
-    {
-      "epoch": 9.06,
-      "learning_rate": 4.5294117647058826e-05,
-      "loss": 0.2007,
-      "step": 154
-    },
-    {
-      "epoch": 9.12,
-      "learning_rate": 4.558823529411765e-05,
-      "loss": 0.1838,
-      "step": 155
-    },
-    {
-      "epoch": 9.18,
-      "learning_rate": 4.588235294117647e-05,
-      "loss": 0.1689,
-      "step": 156
-    },
-    {
-      "epoch": 9.24,
-      "learning_rate": 4.61764705882353e-05,
-      "loss": 0.1427,
-      "step": 157
-    },
-    {
-      "epoch": 9.29,
-      "learning_rate": 4.647058823529412e-05,
-      "loss": 0.1998,
-      "step": 158
-    },
-    {
-      "epoch": 9.35,
-      "learning_rate": 4.6764705882352944e-05,
-      "loss": 0.1375,
-      "step": 159
-    },
-    {
-      "epoch": 9.41,
-      "learning_rate": 4.705882352941177e-05,
-      "loss": 0.202,
-      "step": 160
-    },
-    {
-      "epoch": 9.41,
-      "eval_accuracy_NAT": 0.9873472336178203,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.8667412464894995,
-      "eval_accuracy_concrete tiles": 0.7498939429648833,
-      "eval_accuracy_dark bricks": 0.028214306222843252,
-      "eval_accuracy_grass": 0.855339752542514,
-      "eval_accuracy_red bricks": 0.0,
-      "eval_accuracy_road": 0.9538643897689238,
-      "eval_accuracy_rough concrete": 0.5565400818739237,
-      "eval_accuracy_rough red sidewalk": 0.16307838079696793,
-      "eval_accuracy_speedway bricks": 0.9877744670029537,
-      "eval_accuracy_steel": 0.679419452538146,
-      "eval_accuracy_tiles": 0.01720742827713654,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9704972906134676,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7774759671425241,
-      "eval_iou_concrete tiles": 0.20069383120978931,
-      "eval_iou_dark bricks": 0.028214306222843252,
-      "eval_iou_grass": 0.7106618898582445,
-      "eval_iou_red bricks": 0.0,
-      "eval_iou_road": 0.9073164433288142,
-      "eval_iou_rough concrete": 0.49950262592121164,
-      "eval_iou_rough red sidewalk": 0.1493012048192771,
-      "eval_iou_speedway bricks": 0.9308492599585683,
-      "eval_iou_steel": 0.6268308483977987,
-      "eval_iou_tiles": 0.01656459609215515,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.17850728332996368,
-      "eval_mean_accuracy": 0.5265708216996624,
-      "eval_mean_iou": 0.4475314048895918,
-      "eval_overall_accuracy": 0.9523853689937268,
-      "eval_runtime": 5.1814,
-      "eval_samples_per_second": 22.774,
-      "eval_steps_per_second": 1.544,
-      "step": 160
-    },
-    {
-      "epoch": 9.47,
-      "learning_rate": 4.735294117647059e-05,
-      "loss": 0.1802,
-      "step": 161
-    },
-    {
-      "epoch": 9.53,
-      "learning_rate": 4.7647058823529414e-05,
-      "loss": 0.1469,
-      "step": 162
-    },
-    {
-      "epoch": 9.59,
-      "learning_rate": 4.794117647058824e-05,
-      "loss": 0.1746,
-      "step": 163
-    },
-    {
-      "epoch": 9.65,
-      "learning_rate": 4.823529411764706e-05,
-      "loss": 0.1482,
-      "step": 164
-    },
-    {
-      "epoch": 9.71,
-      "learning_rate": 4.8529411764705885e-05,
-      "loss": 0.1973,
-      "step": 165
-    },
-    {
-      "epoch": 9.76,
-      "learning_rate": 4.882352941176471e-05,
-      "loss": 0.2324,
-      "step": 166
-    },
-    {
-      "epoch": 9.82,
-      "learning_rate": 4.911764705882353e-05,
-      "loss": 0.1866,
-      "step": 167
-    },
-    {
-      "epoch": 9.88,
-      "learning_rate": 4.9411764705882355e-05,
-      "loss": 0.1695,
-      "step": 168
-    },
-    {
-      "epoch": 9.94,
-      "learning_rate": 4.970588235294118e-05,
-      "loss": 0.1591,
-      "step": 169
-    },
-    {
-      "epoch": 10.0,
-      "learning_rate": 5e-05,
-      "loss": 0.1731,
-      "step": 170
-    },
-    {
-      "epoch": 10.06,
-      "learning_rate": 4.998452012383901e-05,
-      "loss": 0.2182,
-      "step": 171
-    },
-    {
-      "epoch": 10.12,
-      "learning_rate": 4.996904024767802e-05,
-      "loss": 0.1816,
-      "step": 172
-    },
-    {
-      "epoch": 10.18,
-      "learning_rate": 4.9953560371517026e-05,
-      "loss": 0.1562,
-      "step": 173
-    },
-    {
-      "epoch": 10.24,
-      "learning_rate": 4.9938080495356036e-05,
-      "loss": 0.1389,
-      "step": 174
-    },
-    {
-      "epoch": 10.29,
-      "learning_rate": 4.9922600619195046e-05,
-      "loss": 0.14,
-      "step": 175
-    },
-    {
-      "epoch": 10.35,
-      "learning_rate": 4.9907120743034056e-05,
-      "loss": 0.1198,
-      "step": 176
-    },
-    {
-      "epoch": 10.41,
-      "learning_rate": 4.9891640866873066e-05,
-      "loss": 0.1172,
-      "step": 177
-    },
-    {
-      "epoch": 10.47,
-      "learning_rate": 4.9876160990712076e-05,
-      "loss": 0.1214,
-      "step": 178
-    },
-    {
-      "epoch": 10.53,
-      "learning_rate": 4.9860681114551086e-05,
-      "loss": 0.1756,
-      "step": 179
-    },
-    {
-      "epoch": 10.59,
-      "learning_rate": 4.9845201238390096e-05,
-      "loss": 0.174,
-      "step": 180
-    },
-    {
-      "epoch": 10.59,
-      "eval_accuracy_NAT": 0.9880436657214318,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9067070625843705,
-      "eval_accuracy_concrete tiles": 0.7554560452510016,
-      "eval_accuracy_dark bricks": 0.0684062936661371,
-      "eval_accuracy_grass": 0.8559283062144455,
-      "eval_accuracy_red bricks": 0.08809314586994728,
-      "eval_accuracy_road": 0.9470382925190827,
-      "eval_accuracy_rough concrete": 0.5194048349710104,
-      "eval_accuracy_rough red sidewalk": 0.47680460824641485,
-      "eval_accuracy_speedway bricks": 0.9830751203154764,
-      "eval_accuracy_steel": 0.7461504245653141,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9711355215780687,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.7933555654989429,
-      "eval_iou_concrete tiles": 0.21096068920582062,
-      "eval_iou_dark bricks": 0.06803708604228342,
-      "eval_iou_grass": 0.740562538990161,
-      "eval_iou_red bricks": 0.08809314586994728,
-      "eval_iou_road": 0.9141089895430123,
-      "eval_iou_rough concrete": 0.5090669111763916,
-      "eval_iou_rough red sidewalk": 0.38462705037367756,
-      "eval_iou_speedway bricks": 0.9481883072256478,
-      "eval_iou_steel": 0.6882840269421739,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.15712417662143707,
-      "eval_mean_accuracy": 0.564239061532664,
-      "eval_mean_iou": 0.4858784486497021,
-      "eval_overall_accuracy": 0.9572272543179787,
-      "eval_runtime": 5.3736,
-      "eval_samples_per_second": 21.959,
-      "eval_steps_per_second": 1.489,
-      "step": 180
-    },
-    {
-      "epoch": 10.65,
-      "learning_rate": 4.9829721362229106e-05,
-      "loss": 0.1558,
-      "step": 181
-    },
-    {
-      "epoch": 10.71,
-      "learning_rate": 4.9814241486068116e-05,
-      "loss": 0.3122,
-      "step": 182
-    },
-    {
-      "epoch": 10.76,
-      "learning_rate": 4.9798761609907126e-05,
-      "loss": 0.1763,
-      "step": 183
-    },
-    {
-      "epoch": 10.82,
-      "learning_rate": 4.9783281733746136e-05,
-      "loss": 0.1451,
-      "step": 184
-    },
-    {
-      "epoch": 10.88,
-      "learning_rate": 4.976780185758514e-05,
-      "loss": 0.1583,
-      "step": 185
-    },
-    {
-      "epoch": 10.94,
-      "learning_rate": 4.975232198142415e-05,
-      "loss": 0.1434,
-      "step": 186
-    },
-    {
-      "epoch": 11.0,
-      "learning_rate": 4.973684210526316e-05,
-      "loss": 0.1714,
-      "step": 187
-    },
-    {
-      "epoch": 11.06,
-      "learning_rate": 4.972136222910217e-05,
-      "loss": 0.1208,
-      "step": 188
-    },
-    {
-      "epoch": 11.12,
-      "learning_rate": 4.970588235294118e-05,
-      "loss": 0.1512,
-      "step": 189
-    },
-    {
-      "epoch": 11.18,
-      "learning_rate": 4.969040247678019e-05,
-      "loss": 0.1312,
-      "step": 190
-    },
-    {
-      "epoch": 11.24,
-      "learning_rate": 4.967492260061919e-05,
-      "loss": 0.1599,
-      "step": 191
-    },
-    {
-      "epoch": 11.29,
-      "learning_rate": 4.96594427244582e-05,
-      "loss": 0.126,
-      "step": 192
-    },
-    {
-      "epoch": 11.35,
-      "learning_rate": 4.964396284829721e-05,
-      "loss": 0.1264,
-      "step": 193
-    },
-    {
-      "epoch": 11.41,
-      "learning_rate": 4.962848297213622e-05,
-      "loss": 0.1599,
-      "step": 194
-    },
-    {
-      "epoch": 11.47,
-      "learning_rate": 4.961300309597524e-05,
-      "loss": 0.1406,
-      "step": 195
-    },
-    {
-      "epoch": 11.53,
-      "learning_rate": 4.959752321981425e-05,
-      "loss": 0.1432,
-      "step": 196
-    },
-    {
-      "epoch": 11.59,
-      "learning_rate": 4.958204334365325e-05,
-      "loss": 0.1213,
-      "step": 197
-    },
-    {
-      "epoch": 11.65,
-      "learning_rate": 4.956656346749226e-05,
-      "loss": 0.1459,
-      "step": 198
-    },
-    {
-      "epoch": 11.71,
-      "learning_rate": 4.955108359133127e-05,
-      "loss": 0.1344,
-      "step": 199
-    },
-    {
-      "epoch": 11.76,
-      "learning_rate": 4.953560371517028e-05,
-      "loss": 0.1317,
-      "step": 200
-    },
-    {
-      "epoch": 11.76,
-      "eval_accuracy_NAT": 0.9864662076511677,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9249773355816844,
-      "eval_accuracy_concrete tiles": 0.7600047136460052,
-      "eval_accuracy_dark bricks": 0.7836265145569742,
-      "eval_accuracy_grass": 0.894897750627349,
-      "eval_accuracy_red bricks": 0.10456942003514938,
-      "eval_accuracy_road": 0.9491855765361391,
-      "eval_accuracy_rough concrete": 0.5790703078912399,
-      "eval_accuracy_rough red sidewalk": 0.5756548026380105,
-      "eval_accuracy_speedway bricks": 0.9715143127606035,
-      "eval_accuracy_steel": 0.7851359658548531,
-      "eval_accuracy_tiles": 0.0,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9719829052737282,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8087278598858759,
-      "eval_iou_concrete tiles": 0.6147203477067368,
-      "eval_iou_dark bricks": 0.7371681176904863,
-      "eval_iou_grass": 0.7440049088827965,
-      "eval_iou_red bricks": 0.10435735818032338,
-      "eval_iou_road": 0.9256206487360326,
-      "eval_iou_rough concrete": 0.5756129055186671,
-      "eval_iou_rough red sidewalk": 0.4396544792492275,
-      "eval_iou_speedway bricks": 0.9522459713674365,
-      "eval_iou_steel": 0.7135061360560284,
-      "eval_iou_tiles": 0.0,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.13907073438167572,
-      "eval_mean_accuracy": 0.6396233005983981,
-      "eval_mean_iou": 0.5836616645036414,
-      "eval_overall_accuracy": 0.9632077621201337,
-      "eval_runtime": 5.8073,
-      "eval_samples_per_second": 20.319,
-      "eval_steps_per_second": 1.378,
-      "step": 200
-    },
-    {
-      "epoch": 11.82,
-      "learning_rate": 4.952012383900929e-05,
-      "loss": 0.1202,
-      "step": 201
-    },
-    {
-      "epoch": 11.88,
-      "learning_rate": 4.95046439628483e-05,
-      "loss": 0.1488,
-      "step": 202
-    },
-    {
-      "epoch": 11.94,
-      "learning_rate": 4.9489164086687305e-05,
-      "loss": 0.1498,
-      "step": 203
-    },
-    {
-      "epoch": 12.0,
-      "learning_rate": 4.9473684210526315e-05,
-      "loss": 0.1198,
-      "step": 204
-    },
-    {
-      "epoch": 12.06,
-      "learning_rate": 4.9458204334365325e-05,
-      "loss": 0.1387,
-      "step": 205
-    },
-    {
-      "epoch": 12.12,
-      "learning_rate": 4.9442724458204335e-05,
-      "loss": 0.1239,
-      "step": 206
-    },
-    {
-      "epoch": 12.18,
-      "learning_rate": 4.9427244582043345e-05,
-      "loss": 0.1445,
-      "step": 207
-    },
-    {
-      "epoch": 12.24,
-      "learning_rate": 4.9411764705882355e-05,
-      "loss": 0.1116,
-      "step": 208
-    },
-    {
-      "epoch": 12.29,
-      "learning_rate": 4.9396284829721365e-05,
-      "loss": 0.1086,
-      "step": 209
-    },
-    {
-      "epoch": 12.35,
-      "learning_rate": 4.9380804953560375e-05,
-      "loss": 0.1052,
-      "step": 210
-    },
-    {
-      "epoch": 12.41,
-      "learning_rate": 4.9365325077399385e-05,
-      "loss": 0.185,
-      "step": 211
-    },
-    {
-      "epoch": 12.47,
-      "learning_rate": 4.9349845201238395e-05,
-      "loss": 0.0983,
-      "step": 212
-    },
-    {
-      "epoch": 12.53,
-      "learning_rate": 4.9334365325077405e-05,
-      "loss": 0.1275,
-      "step": 213
-    },
-    {
-      "epoch": 12.59,
-      "learning_rate": 4.9318885448916415e-05,
-      "loss": 0.1293,
-      "step": 214
-    },
-    {
-      "epoch": 12.65,
-      "learning_rate": 4.930340557275542e-05,
-      "loss": 0.1142,
-      "step": 215
-    },
-    {
-      "epoch": 12.71,
-      "learning_rate": 4.928792569659443e-05,
-      "loss": 0.1127,
-      "step": 216
-    },
-    {
-      "epoch": 12.76,
-      "learning_rate": 4.927244582043344e-05,
-      "loss": 0.1655,
-      "step": 217
-    },
-    {
-      "epoch": 12.82,
-      "learning_rate": 4.925696594427245e-05,
-      "loss": 0.1393,
-      "step": 218
-    },
-    {
-      "epoch": 12.88,
-      "learning_rate": 4.924148606811146e-05,
-      "loss": 0.1265,
-      "step": 219
-    },
-    {
-      "epoch": 12.94,
-      "learning_rate": 4.922600619195047e-05,
-      "loss": 0.0915,
-      "step": 220
-    },
-    {
-      "epoch": 12.94,
-      "eval_accuracy_NAT": 0.9860258024913594,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9262420330556528,
-      "eval_accuracy_concrete tiles": 0.7638934716002829,
-      "eval_accuracy_dark bricks": 0.7600061252225179,
-      "eval_accuracy_grass": 0.9003149046013164,
-      "eval_accuracy_red bricks": 0.5878185413005272,
-      "eval_accuracy_road": 0.9337452256598837,
-      "eval_accuracy_rough concrete": 0.6387834512182768,
-      "eval_accuracy_rough red sidewalk": 0.5331292910910745,
-      "eval_accuracy_speedway bricks": 0.9804688276498366,
-      "eval_accuracy_steel": 0.7936647558312134,
-      "eval_accuracy_tiles": 0.011542306643642643,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9732919784830467,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.811574716554503,
-      "eval_iou_concrete tiles": 0.6126800499035954,
-      "eval_iou_dark bricks": 0.7370453216509963,
-      "eval_iou_grass": 0.7637841543433552,
-      "eval_iou_red bricks": 0.5786969451203028,
-      "eval_iou_road": 0.90648387138614,
-      "eval_iou_rough concrete": 0.56731583403895,
-      "eval_iou_rough red sidewalk": 0.4643523825118716,
-      "eval_iou_speedway bricks": 0.956141941385667,
-      "eval_iou_steel": 0.7260183788232067,
-      "eval_iou_tiles": 0.011447629349197765,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.12515389919281006,
-      "eval_mean_accuracy": 0.6781257489511988,
-      "eval_mean_iou": 0.623756400273141,
-      "eval_overall_accuracy": 0.9643849841618942,
-      "eval_runtime": 5.8731,
-      "eval_samples_per_second": 20.092,
-      "eval_steps_per_second": 1.362,
-      "step": 220
-    },
-    {
-      "epoch": 13.0,
-      "learning_rate": 4.921052631578947e-05,
-      "loss": 0.1054,
-      "step": 221
-    },
-    {
-      "epoch": 13.06,
-      "learning_rate": 4.919504643962848e-05,
-      "loss": 0.1411,
-      "step": 222
-    },
-    {
-      "epoch": 13.12,
-      "learning_rate": 4.917956656346749e-05,
-      "loss": 0.1017,
-      "step": 223
-    },
-    {
-      "epoch": 13.18,
-      "learning_rate": 4.91640866873065e-05,
-      "loss": 0.0983,
-      "step": 224
-    },
-    {
-      "epoch": 13.24,
-      "learning_rate": 4.914860681114551e-05,
-      "loss": 0.1682,
-      "step": 225
-    },
-    {
-      "epoch": 13.29,
-      "learning_rate": 4.913312693498452e-05,
-      "loss": 0.1545,
-      "step": 226
-    },
-    {
-      "epoch": 13.35,
-      "learning_rate": 4.911764705882353e-05,
-      "loss": 0.1017,
-      "step": 227
-    },
-    {
-      "epoch": 13.41,
-      "learning_rate": 4.910216718266254e-05,
-      "loss": 0.1027,
-      "step": 228
-    },
-    {
-      "epoch": 13.47,
-      "learning_rate": 4.908668730650155e-05,
-      "loss": 0.1169,
-      "step": 229
-    },
-    {
-      "epoch": 13.53,
-      "learning_rate": 4.907120743034056e-05,
-      "loss": 0.0948,
-      "step": 230
-    },
-    {
-      "epoch": 13.59,
-      "learning_rate": 4.905572755417957e-05,
-      "loss": 0.1222,
-      "step": 231
-    },
-    {
-      "epoch": 13.65,
-      "learning_rate": 4.904024767801858e-05,
-      "loss": 0.1146,
-      "step": 232
-    },
-    {
-      "epoch": 13.71,
-      "learning_rate": 4.9024767801857585e-05,
-      "loss": 0.1355,
-      "step": 233
-    },
-    {
-      "epoch": 13.76,
-      "learning_rate": 4.9009287925696595e-05,
-      "loss": 0.0935,
-      "step": 234
-    },
-    {
-      "epoch": 13.82,
-      "learning_rate": 4.8993808049535605e-05,
-      "loss": 0.1122,
-      "step": 235
-    },
-    {
-      "epoch": 13.88,
-      "learning_rate": 4.8978328173374615e-05,
-      "loss": 0.0929,
-      "step": 236
-    },
-    {
-      "epoch": 13.94,
-      "learning_rate": 4.8962848297213625e-05,
-      "loss": 0.083,
-      "step": 237
-    },
-    {
-      "epoch": 14.0,
-      "learning_rate": 4.8947368421052635e-05,
-      "loss": 0.1423,
-      "step": 238
-    },
-    {
-      "epoch": 14.06,
-      "learning_rate": 4.893188854489164e-05,
-      "loss": 0.1276,
-      "step": 239
-    },
-    {
-      "epoch": 14.12,
-      "learning_rate": 4.891640866873065e-05,
-      "loss": 0.1027,
-      "step": 240
-    },
-    {
-      "epoch": 14.12,
-      "eval_accuracy_NAT": 0.9917472418339723,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9190345187117132,
-      "eval_accuracy_concrete tiles": 0.7522743341975018,
-      "eval_accuracy_dark bricks": 0.7271979021112877,
-      "eval_accuracy_grass": 0.8350431669107934,
-      "eval_accuracy_red bricks": 0.6341168717047452,
-      "eval_accuracy_road": 0.9736051872567405,
-      "eval_accuracy_rough concrete": 0.6122906226351009,
-      "eval_accuracy_rough red sidewalk": 0.5560201234781431,
-      "eval_accuracy_speedway bricks": 0.9814349844023434,
-      "eval_accuracy_steel": 0.7841863789251675,
-      "eval_accuracy_tiles": 0.058953618710048776,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9740554919478627,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8296911062490576,
-      "eval_iou_concrete tiles": 0.6982172153560101,
-      "eval_iou_dark bricks": 0.7163416439221454,
-      "eval_iou_grass": 0.7689727795781048,
-      "eval_iou_red bricks": 0.6251218191662155,
-      "eval_iou_road": 0.9231815060770597,
-      "eval_iou_rough concrete": 0.594851191682249,
-      "eval_iou_rough red sidewalk": 0.47678587549491225,
-      "eval_iou_speedway bricks": 0.9582911258333994,
-      "eval_iou_steel": 0.7348050774234488,
-      "eval_iou_tiles": 0.058793317018641045,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11748708039522171,
-      "eval_mean_accuracy": 0.6789157654521198,
-      "eval_mean_iou": 0.6430083192114698,
-      "eval_overall_accuracy": 0.9672419661182469,
-      "eval_runtime": 5.5775,
-      "eval_samples_per_second": 21.156,
-      "eval_steps_per_second": 1.434,
-      "step": 240
-    },
-    {
-      "epoch": 14.18,
-      "learning_rate": 4.890092879256966e-05,
-      "loss": 0.1013,
-      "step": 241
-    },
-    {
-      "epoch": 14.24,
-      "learning_rate": 4.888544891640867e-05,
-      "loss": 0.0807,
-      "step": 242
-    },
-    {
-      "epoch": 14.29,
-      "learning_rate": 4.886996904024768e-05,
-      "loss": 0.0938,
-      "step": 243
-    },
-    {
-      "epoch": 14.35,
-      "learning_rate": 4.8854489164086695e-05,
-      "loss": 0.1197,
-      "step": 244
-    },
-    {
-      "epoch": 14.41,
-      "learning_rate": 4.88390092879257e-05,
-      "loss": 0.1158,
-      "step": 245
-    },
-    {
-      "epoch": 14.47,
-      "learning_rate": 4.882352941176471e-05,
-      "loss": 0.121,
-      "step": 246
-    },
-    {
-      "epoch": 14.53,
-      "learning_rate": 4.880804953560372e-05,
-      "loss": 0.1065,
-      "step": 247
-    },
-    {
-      "epoch": 14.59,
-      "learning_rate": 4.879256965944273e-05,
-      "loss": 0.1111,
-      "step": 248
-    },
-    {
-      "epoch": 14.65,
-      "learning_rate": 4.877708978328174e-05,
-      "loss": 0.0954,
-      "step": 249
-    },
-    {
-      "epoch": 14.71,
-      "learning_rate": 4.876160990712075e-05,
-      "loss": 0.0999,
-      "step": 250
-    },
-    {
-      "epoch": 14.76,
-      "learning_rate": 4.874613003095975e-05,
-      "loss": 0.1146,
-      "step": 251
-    },
-    {
-      "epoch": 14.82,
-      "learning_rate": 4.873065015479876e-05,
-      "loss": 0.1087,
-      "step": 252
-    },
-    {
-      "epoch": 14.88,
-      "learning_rate": 4.871517027863777e-05,
-      "loss": 0.1084,
-      "step": 253
-    },
-    {
-      "epoch": 14.94,
-      "learning_rate": 4.869969040247678e-05,
-      "loss": 0.0894,
-      "step": 254
-    },
-    {
-      "epoch": 15.0,
-      "learning_rate": 4.868421052631579e-05,
-      "loss": 0.1068,
-      "step": 255
-    },
-    {
-      "epoch": 15.06,
-      "learning_rate": 4.86687306501548e-05,
-      "loss": 0.0922,
-      "step": 256
-    },
-    {
-      "epoch": 15.12,
-      "learning_rate": 4.8653250773993805e-05,
-      "loss": 0.0908,
-      "step": 257
-    },
-    {
-      "epoch": 15.18,
-      "learning_rate": 4.863777089783282e-05,
-      "loss": 0.1101,
-      "step": 258
-    },
-    {
-      "epoch": 15.24,
-      "learning_rate": 4.862229102167183e-05,
-      "loss": 0.0947,
-      "step": 259
-    },
-    {
-      "epoch": 15.29,
-      "learning_rate": 4.860681114551084e-05,
-      "loss": 0.1055,
-      "step": 260
-    },
-    {
-      "epoch": 15.29,
-      "eval_accuracy_NAT": 0.9880657156308897,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9169668205146237,
-      "eval_accuracy_concrete tiles": 0.6840678765024747,
-      "eval_accuracy_dark bricks": 0.7566324675076087,
-      "eval_accuracy_grass": 0.8498658249024439,
-      "eval_accuracy_red bricks": 0.6606985940246046,
-      "eval_accuracy_road": 0.9807342984532393,
-      "eval_accuracy_rough concrete": 0.6125766450759449,
-      "eval_accuracy_rough red sidewalk": 0.6318290857955016,
-      "eval_accuracy_speedway bricks": 0.9880714428192173,
-      "eval_accuracy_steel": 0.7434690909454388,
-      "eval_accuracy_tiles": 0.030385652397830896,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9746038269106396,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8274392746410625,
-      "eval_iou_concrete tiles": 0.666950067786484,
-      "eval_iou_dark bricks": 0.738972547040184,
-      "eval_iou_grass": 0.7776974436725635,
-      "eval_iou_red bricks": 0.6507979442791453,
-      "eval_iou_road": 0.9132698891078013,
-      "eval_iou_rough concrete": 0.5830686510277235,
-      "eval_iou_rough red sidewalk": 0.5123983070486785,
-      "eval_iou_speedway bricks": 0.9579088057628538,
-      "eval_iou_steel": 0.7130391226405146,
-      "eval_iou_tiles": 0.030056937368894216,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.11388663947582245,
-      "eval_mean_accuracy": 0.6802587318899861,
-      "eval_mean_iou": 0.6420156013297343,
-      "eval_overall_accuracy": 0.9669402494268903,
-      "eval_runtime": 6.1549,
-      "eval_samples_per_second": 19.172,
-      "eval_steps_per_second": 1.3,
-      "step": 260
-    },
-    {
-      "epoch": 15.35,
-      "learning_rate": 4.859133126934985e-05,
-      "loss": 0.0871,
-      "step": 261
-    },
-    {
-      "epoch": 15.41,
-      "learning_rate": 4.857585139318886e-05,
-      "loss": 0.1223,
-      "step": 262
-    },
-    {
-      "epoch": 15.47,
-      "learning_rate": 4.8560371517027865e-05,
-      "loss": 0.0979,
-      "step": 263
-    },
-    {
-      "epoch": 15.53,
-      "learning_rate": 4.8544891640866875e-05,
-      "loss": 0.0874,
-      "step": 264
-    },
-    {
-      "epoch": 15.59,
-      "learning_rate": 4.8529411764705885e-05,
-      "loss": 0.0922,
-      "step": 265
-    },
-    {
-      "epoch": 15.65,
-      "learning_rate": 4.8513931888544895e-05,
-      "loss": 0.0995,
-      "step": 266
-    },
-    {
-      "epoch": 15.71,
-      "learning_rate": 4.8498452012383905e-05,
-      "loss": 0.0835,
-      "step": 267
-    },
-    {
-      "epoch": 15.76,
-      "learning_rate": 4.8482972136222915e-05,
-      "loss": 0.0942,
-      "step": 268
-    },
-    {
-      "epoch": 15.82,
-      "learning_rate": 4.846749226006192e-05,
-      "loss": 0.0956,
-      "step": 269
-    },
-    {
-      "epoch": 15.88,
-      "learning_rate": 4.845201238390093e-05,
-      "loss": 0.0978,
-      "step": 270
-    },
-    {
-      "epoch": 15.94,
-      "learning_rate": 4.843653250773994e-05,
-      "loss": 0.0927,
-      "step": 271
-    },
-    {
-      "epoch": 16.0,
-      "learning_rate": 4.842105263157895e-05,
-      "loss": 0.0825,
-      "step": 272
-    },
-    {
-      "epoch": 16.06,
-      "learning_rate": 4.840557275541796e-05,
-      "loss": 0.0847,
-      "step": 273
-    },
-    {
-      "epoch": 16.12,
-      "learning_rate": 4.839009287925697e-05,
-      "loss": 0.0886,
-      "step": 274
-    },
-    {
-      "epoch": 16.18,
-      "learning_rate": 4.837461300309598e-05,
-      "loss": 0.0814,
-      "step": 275
-    },
-    {
-      "epoch": 16.24,
-      "learning_rate": 4.835913312693499e-05,
-      "loss": 0.0678,
-      "step": 276
-    },
-    {
-      "epoch": 16.29,
-      "learning_rate": 4.8343653250774e-05,
-      "loss": 0.0912,
-      "step": 277
-    },
-    {
-      "epoch": 16.35,
-      "learning_rate": 4.832817337461301e-05,
-      "loss": 0.1018,
-      "step": 278
-    },
-    {
-      "epoch": 16.41,
-      "learning_rate": 4.831269349845202e-05,
-      "loss": 0.0735,
-      "step": 279
-    },
-    {
-      "epoch": 16.47,
-      "learning_rate": 4.829721362229103e-05,
-      "loss": 0.089,
-      "step": 280
-    },
-    {
-      "epoch": 16.47,
-      "eval_accuracy_NAT": 0.9882539215817883,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9266035935906707,
-      "eval_accuracy_concrete tiles": 0.7337732736271506,
-      "eval_accuracy_dark bricks": 0.7679306318549853,
-      "eval_accuracy_grass": 0.8684279750347266,
-      "eval_accuracy_red bricks": 0.7258897188049209,
-      "eval_accuracy_road": 0.9741289150657786,
-      "eval_accuracy_rough concrete": 0.6240711719173634,
-      "eval_accuracy_rough red sidewalk": 0.7053068529617459,
-      "eval_accuracy_speedway bricks": 0.9871671488882171,
-      "eval_accuracy_steel": 0.7572206090350697,
-      "eval_accuracy_tiles": 0.06749674331242994,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9757167219517451,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8338523416250841,
-      "eval_iou_concrete tiles": 0.6868753722946588,
-      "eval_iou_dark bricks": 0.7545810625902224,
-      "eval_iou_grass": 0.7875470668392022,
-      "eval_iou_red bricks": 0.7145483051305617,
-      "eval_iou_road": 0.936732714837276,
-      "eval_iou_rough concrete": 0.5637154375709819,
-      "eval_iou_rough red sidewalk": 0.5270045513288757,
-      "eval_iou_speedway bricks": 0.9597518102996859,
-      "eval_iou_steel": 0.724264609221694,
-      "eval_iou_tiles": 0.06645588498478792,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10488714277744293,
-      "eval_mean_accuracy": 0.7020208119749882,
-      "eval_mean_iou": 0.656234298359598,
-      "eval_overall_accuracy": 0.968710850861113,
-      "eval_runtime": 5.5741,
-      "eval_samples_per_second": 21.169,
-      "eval_steps_per_second": 1.435,
-      "step": 280
-    },
-    {
-      "epoch": 16.53,
-      "learning_rate": 4.828173374613003e-05,
-      "loss": 0.1157,
-      "step": 281
-    },
-    {
-      "epoch": 16.59,
-      "learning_rate": 4.826625386996904e-05,
-      "loss": 0.0849,
-      "step": 282
-    },
-    {
-      "epoch": 16.65,
-      "learning_rate": 4.825077399380805e-05,
-      "loss": 0.0923,
-      "step": 283
-    },
-    {
-      "epoch": 16.71,
-      "learning_rate": 4.823529411764706e-05,
-      "loss": 0.0892,
-      "step": 284
-    },
-    {
-      "epoch": 16.76,
-      "learning_rate": 4.821981424148607e-05,
-      "loss": 0.0897,
-      "step": 285
-    },
-    {
-      "epoch": 16.82,
-      "learning_rate": 4.820433436532508e-05,
-      "loss": 0.0713,
-      "step": 286
-    },
-    {
-      "epoch": 16.88,
-      "learning_rate": 4.8188854489164085e-05,
-      "loss": 0.0838,
-      "step": 287
-    },
-    {
-      "epoch": 16.94,
-      "learning_rate": 4.8173374613003095e-05,
-      "loss": 0.108,
-      "step": 288
-    },
-    {
-      "epoch": 17.0,
-      "learning_rate": 4.8157894736842105e-05,
-      "loss": 0.0924,
-      "step": 289
-    },
-    {
-      "epoch": 17.06,
-      "learning_rate": 4.8142414860681115e-05,
-      "loss": 0.0913,
-      "step": 290
-    },
-    {
-      "epoch": 17.12,
-      "learning_rate": 4.8126934984520125e-05,
-      "loss": 0.0698,
-      "step": 291
-    },
-    {
-      "epoch": 17.18,
-      "learning_rate": 4.811145510835914e-05,
-      "loss": 0.1019,
-      "step": 292
-    },
-    {
-      "epoch": 17.24,
-      "learning_rate": 4.8095975232198145e-05,
-      "loss": 0.074,
-      "step": 293
-    },
-    {
-      "epoch": 17.29,
-      "learning_rate": 4.8080495356037155e-05,
-      "loss": 0.0724,
-      "step": 294
-    },
-    {
-      "epoch": 17.35,
-      "learning_rate": 4.8065015479876165e-05,
-      "loss": 0.0794,
-      "step": 295
-    },
-    {
-      "epoch": 17.41,
-      "learning_rate": 4.8049535603715175e-05,
-      "loss": 0.067,
-      "step": 296
-    },
-    {
-      "epoch": 17.47,
-      "learning_rate": 4.8034055727554185e-05,
-      "loss": 0.0773,
-      "step": 297
-    },
-    {
-      "epoch": 17.53,
-      "learning_rate": 4.8018575851393195e-05,
-      "loss": 0.103,
-      "step": 298
-    },
-    {
-      "epoch": 17.59,
-      "learning_rate": 4.80030959752322e-05,
-      "loss": 0.0926,
-      "step": 299
-    },
-    {
-      "epoch": 17.65,
-      "learning_rate": 4.798761609907121e-05,
-      "loss": 0.0982,
-      "step": 300
-    },
-    {
-      "epoch": 17.65,
-      "eval_accuracy_NAT": 0.9931665767148967,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9314965089916206,
-      "eval_accuracy_concrete tiles": 0.7459816167805797,
-      "eval_accuracy_dark bricks": 0.8016145703730644,
-      "eval_accuracy_grass": 0.8160921171657072,
-      "eval_accuracy_red bricks": 0.8456722319859402,
-      "eval_accuracy_road": 0.9701795798527256,
-      "eval_accuracy_rough concrete": 0.6210738950893522,
-      "eval_accuracy_rough red sidewalk": 0.6152550402695163,
-      "eval_accuracy_speedway bricks": 0.9799415370631988,
-      "eval_accuracy_steel": 0.8352466676995507,
-      "eval_accuracy_tiles": 0.09173255778727014,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9752539993478923,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8405579023170859,
-      "eval_iou_concrete tiles": 0.7205427062465853,
-      "eval_iou_dark bricks": 0.7828242705198423,
-      "eval_iou_grass": 0.7712687805109247,
-      "eval_iou_red bricks": 0.8008946218662228,
-      "eval_iou_road": 0.9368960548615971,
-      "eval_iou_rough concrete": 0.5511559532119212,
-      "eval_iou_rough red sidewalk": 0.5013173858163503,
-      "eval_iou_speedway bricks": 0.9640669157177488,
-      "eval_iou_steel": 0.7744820666819596,
-      "eval_iou_tiles": 0.08814881662833687,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10236875712871552,
-      "eval_mean_accuracy": 0.7113425307518016,
-      "eval_mean_iou": 0.6698007287481899,
-      "eval_overall_accuracy": 0.9696271864034361,
-      "eval_runtime": 5.9798,
-      "eval_samples_per_second": 19.733,
-      "eval_steps_per_second": 1.338,
-      "step": 300
-    },
-    {
-      "epoch": 17.71,
-      "learning_rate": 4.797213622291022e-05,
-      "loss": 0.0788,
-      "step": 301
-    },
-    {
-      "epoch": 17.76,
-      "learning_rate": 4.795665634674923e-05,
-      "loss": 0.0778,
-      "step": 302
-    },
-    {
-      "epoch": 17.82,
-      "learning_rate": 4.794117647058824e-05,
-      "loss": 0.078,
-      "step": 303
-    },
-    {
-      "epoch": 17.88,
-      "learning_rate": 4.792569659442725e-05,
-      "loss": 0.0882,
-      "step": 304
-    },
-    {
-      "epoch": 17.94,
-      "learning_rate": 4.791021671826625e-05,
-      "loss": 0.0745,
-      "step": 305
-    },
-    {
-      "epoch": 18.0,
-      "learning_rate": 4.789473684210526e-05,
-      "loss": 0.0797,
-      "step": 306
-    },
-    {
-      "epoch": 18.06,
-      "learning_rate": 4.787925696594428e-05,
-      "loss": 0.0779,
-      "step": 307
-    },
-    {
-      "epoch": 18.12,
-      "learning_rate": 4.786377708978329e-05,
-      "loss": 0.0661,
-      "step": 308
-    },
-    {
-      "epoch": 18.18,
-      "learning_rate": 4.78482972136223e-05,
-      "loss": 0.0659,
-      "step": 309
-    },
-    {
-      "epoch": 18.24,
-      "learning_rate": 4.783281733746131e-05,
-      "loss": 0.0739,
-      "step": 310
-    },
-    {
-      "epoch": 18.29,
-      "learning_rate": 4.781733746130031e-05,
-      "loss": 0.0877,
-      "step": 311
-    },
-    {
-      "epoch": 18.35,
-      "learning_rate": 4.780185758513932e-05,
-      "loss": 0.0735,
-      "step": 312
-    },
-    {
-      "epoch": 18.41,
-      "learning_rate": 4.778637770897833e-05,
-      "loss": 0.0671,
-      "step": 313
-    },
-    {
-      "epoch": 18.47,
-      "learning_rate": 4.777089783281734e-05,
-      "loss": 0.0732,
-      "step": 314
-    },
-    {
-      "epoch": 18.53,
-      "learning_rate": 4.775541795665635e-05,
-      "loss": 0.0977,
-      "step": 315
-    },
-    {
-      "epoch": 18.59,
-      "learning_rate": 4.773993808049536e-05,
-      "loss": 0.0752,
-      "step": 316
-    },
-    {
-      "epoch": 18.65,
-      "learning_rate": 4.7724458204334364e-05,
-      "loss": 0.0708,
-      "step": 317
-    },
-    {
-      "epoch": 18.71,
-      "learning_rate": 4.7708978328173374e-05,
-      "loss": 0.0807,
-      "step": 318
-    },
-    {
-      "epoch": 18.76,
-      "learning_rate": 4.7693498452012384e-05,
-      "loss": 0.0666,
-      "step": 319
-    },
-    {
-      "epoch": 18.82,
-      "learning_rate": 4.7678018575851394e-05,
-      "loss": 0.0787,
-      "step": 320
-    },
-    {
-      "epoch": 18.82,
-      "eval_accuracy_NAT": 0.9902132366415152,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9379877818528716,
-      "eval_accuracy_concrete tiles": 0.7392411029931653,
-      "eval_accuracy_dark bricks": 0.8253450222996382,
-      "eval_accuracy_grass": 0.8625017505213715,
-      "eval_accuracy_red bricks": 0.8736269771528998,
-      "eval_accuracy_road": 0.9765295123292473,
-      "eval_accuracy_rough concrete": 0.6109022220368373,
-      "eval_accuracy_rough red sidewalk": 0.7007873423623279,
-      "eval_accuracy_speedway bricks": 0.9772675551607303,
-      "eval_accuracy_steel": 0.8889108237916507,
-      "eval_accuracy_tiles": 0.11981580780999121,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9772417072398732,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8491310902138964,
-      "eval_iou_concrete tiles": 0.6966970969103308,
-      "eval_iou_dark bricks": 0.802353926311872,
-      "eval_iou_grass": 0.7964497227547327,
-      "eval_iou_red bricks": 0.8242395979066273,
-      "eval_iou_road": 0.9342291068271499,
-      "eval_iou_rough concrete": 0.5883997084430976,
-      "eval_iou_rough red sidewalk": 0.5513809670319267,
-      "eval_iou_speedway bricks": 0.9625142068997178,
-      "eval_iou_steel": 0.773288725413471,
-      "eval_iou_tiles": 0.11640912435614423,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.10046244412660599,
-      "eval_mean_accuracy": 0.7310099334578652,
-      "eval_mean_iou": 0.682487306177603,
-      "eval_overall_accuracy": 0.971074799359855,
-      "eval_runtime": 6.0333,
-      "eval_samples_per_second": 19.558,
-      "eval_steps_per_second": 1.326,
-      "step": 320
-    },
-    {
-      "epoch": 18.88,
-      "learning_rate": 4.7662538699690404e-05,
-      "loss": 0.083,
-      "step": 321
-    },
-    {
-      "epoch": 18.94,
-      "learning_rate": 4.7647058823529414e-05,
-      "loss": 0.0779,
-      "step": 322
-    },
-    {
-      "epoch": 19.0,
-      "learning_rate": 4.7631578947368424e-05,
-      "loss": 0.087,
-      "step": 323
-    },
-    {
-      "epoch": 19.06,
-      "learning_rate": 4.7616099071207434e-05,
-      "loss": 0.0676,
-      "step": 324
-    },
-    {
-      "epoch": 19.12,
-      "learning_rate": 4.7600619195046444e-05,
-      "loss": 0.0646,
-      "step": 325
-    },
-    {
-      "epoch": 19.18,
-      "learning_rate": 4.7585139318885454e-05,
-      "loss": 0.0675,
-      "step": 326
-    },
-    {
-      "epoch": 19.24,
-      "learning_rate": 4.7569659442724464e-05,
-      "loss": 0.0754,
-      "step": 327
-    },
-    {
-      "epoch": 19.29,
-      "learning_rate": 4.755417956656347e-05,
-      "loss": 0.0707,
-      "step": 328
-    },
-    {
-      "epoch": 19.35,
-      "learning_rate": 4.753869969040248e-05,
-      "loss": 0.0819,
-      "step": 329
-    },
-    {
-      "epoch": 19.41,
-      "learning_rate": 4.752321981424149e-05,
-      "loss": 0.077,
-      "step": 330
-    },
-    {
-      "epoch": 19.47,
-      "learning_rate": 4.75077399380805e-05,
-      "loss": 0.0674,
-      "step": 331
-    },
-    {
-      "epoch": 19.53,
-      "learning_rate": 4.749226006191951e-05,
-      "loss": 0.0637,
-      "step": 332
-    },
-    {
-      "epoch": 19.59,
-      "learning_rate": 4.747678018575852e-05,
-      "loss": 0.0711,
-      "step": 333
-    },
-    {
-      "epoch": 19.65,
-      "learning_rate": 4.746130030959752e-05,
-      "loss": 0.0842,
-      "step": 334
-    },
-    {
-      "epoch": 19.71,
-      "learning_rate": 4.744582043343653e-05,
-      "loss": 0.0651,
-      "step": 335
-    },
-    {
-      "epoch": 19.76,
-      "learning_rate": 4.743034055727554e-05,
-      "loss": 0.071,
-      "step": 336
-    },
-    {
-      "epoch": 19.82,
-      "learning_rate": 4.741486068111455e-05,
-      "loss": 0.0744,
-      "step": 337
-    },
-    {
-      "epoch": 19.88,
-      "learning_rate": 4.739938080495356e-05,
-      "loss": 0.0751,
-      "step": 338
-    },
-    {
-      "epoch": 19.94,
-      "learning_rate": 4.738390092879257e-05,
-      "loss": 0.0786,
-      "step": 339
-    },
-    {
-      "epoch": 20.0,
-      "learning_rate": 4.736842105263158e-05,
-      "loss": 0.0659,
-      "step": 340
-    },
-    {
-      "epoch": 20.0,
-      "eval_accuracy_NAT": 0.9924402235850037,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9213479710568117,
-      "eval_accuracy_concrete tiles": 0.8307801084138581,
-      "eval_accuracy_dark bricks": 0.8274697088605172,
-      "eval_accuracy_grass": 0.867474177444201,
-      "eval_accuracy_red bricks": 0.867805360281195,
-      "eval_accuracy_road": 0.9700555953101778,
-      "eval_accuracy_rough concrete": 0.6168431464852013,
-      "eval_accuracy_rough red sidewalk": 0.7044646147136014,
-      "eval_accuracy_speedway bricks": 0.9838130186794876,
-      "eval_accuracy_steel": 0.8907150389580533,
-      "eval_accuracy_tiles": 0.1436274955315217,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.977106375940702,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8576227324911442,
-      "eval_iou_concrete tiles": 0.7540268241031893,
-      "eval_iou_dark bricks": 0.8016820124806439,
-      "eval_iou_grass": 0.80373527797601,
-      "eval_iou_red bricks": 0.8045315682281059,
-      "eval_iou_road": 0.94770066933286,
-      "eval_iou_rough concrete": 0.5887793058731188,
-      "eval_iou_rough red sidewalk": 0.5287908465372523,
-      "eval_iou_speedway bricks": 0.9634395180447106,
-      "eval_iou_steel": 0.7817665804716818,
-      "eval_iou_tiles": 0.1388002459232368,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09484652429819107,
-      "eval_mean_accuracy": 0.7397566507168947,
-      "eval_mean_iou": 0.688306304415589,
-      "eval_overall_accuracy": 0.9722490795588089,
-      "eval_runtime": 5.4896,
-      "eval_samples_per_second": 21.495,
-      "eval_steps_per_second": 1.457,
-      "step": 340
-    },
-    {
-      "epoch": 20.06,
-      "learning_rate": 4.735294117647059e-05,
-      "loss": 0.0644,
-      "step": 341
-    },
-    {
-      "epoch": 20.12,
-      "learning_rate": 4.73374613003096e-05,
-      "loss": 0.0652,
-      "step": 342
-    },
-    {
-      "epoch": 20.18,
-      "learning_rate": 4.732198142414861e-05,
-      "loss": 0.065,
-      "step": 343
-    },
-    {
-      "epoch": 20.24,
-      "learning_rate": 4.730650154798762e-05,
-      "loss": 0.103,
-      "step": 344
-    },
-    {
-      "epoch": 20.29,
-      "learning_rate": 4.729102167182663e-05,
-      "loss": 0.0546,
-      "step": 345
-    },
-    {
-      "epoch": 20.35,
-      "learning_rate": 4.7275541795665634e-05,
-      "loss": 0.0758,
-      "step": 346
-    },
-    {
-      "epoch": 20.41,
-      "learning_rate": 4.7260061919504644e-05,
-      "loss": 0.0645,
-      "step": 347
-    },
-    {
-      "epoch": 20.47,
-      "learning_rate": 4.7244582043343654e-05,
-      "loss": 0.0779,
-      "step": 348
-    },
-    {
-      "epoch": 20.53,
-      "learning_rate": 4.7229102167182664e-05,
-      "loss": 0.066,
-      "step": 349
-    },
-    {
-      "epoch": 20.59,
-      "learning_rate": 4.7213622291021674e-05,
-      "loss": 0.0758,
-      "step": 350
-    },
-    {
-      "epoch": 20.65,
-      "learning_rate": 4.7198142414860684e-05,
-      "loss": 0.0775,
-      "step": 351
-    },
-    {
-      "epoch": 20.71,
-      "learning_rate": 4.718266253869969e-05,
-      "loss": 0.066,
-      "step": 352
-    },
-    {
-      "epoch": 20.76,
-      "learning_rate": 4.71671826625387e-05,
-      "loss": 0.073,
-      "step": 353
-    },
-    {
-      "epoch": 20.82,
-      "learning_rate": 4.715170278637771e-05,
-      "loss": 0.0638,
-      "step": 354
-    },
-    {
-      "epoch": 20.88,
-      "learning_rate": 4.7136222910216724e-05,
-      "loss": 0.0726,
-      "step": 355
-    },
-    {
-      "epoch": 20.94,
-      "learning_rate": 4.7120743034055734e-05,
-      "loss": 0.0797,
-      "step": 356
-    },
-    {
-      "epoch": 21.0,
-      "learning_rate": 4.7105263157894744e-05,
-      "loss": 0.0823,
-      "step": 357
-    },
-    {
-      "epoch": 21.06,
-      "learning_rate": 4.708978328173375e-05,
-      "loss": 0.0607,
-      "step": 358
-    },
-    {
-      "epoch": 21.12,
-      "learning_rate": 4.707430340557276e-05,
-      "loss": 0.0682,
-      "step": 359
-    },
-    {
-      "epoch": 21.18,
-      "learning_rate": 4.705882352941177e-05,
-      "loss": 0.0615,
-      "step": 360
-    },
-    {
-      "epoch": 21.18,
-      "eval_accuracy_NAT": 0.9913629588153516,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9522984701326538,
-      "eval_accuracy_concrete tiles": 0.7668159321234975,
-      "eval_accuracy_dark bricks": 0.8173152384051452,
-      "eval_accuracy_grass": 0.8567212450843468,
-      "eval_accuracy_red bricks": 0.7849297012302284,
-      "eval_accuracy_road": 0.9600828558770543,
-      "eval_accuracy_rough concrete": 0.608846435743271,
-      "eval_accuracy_rough red sidewalk": 0.6382661924063198,
-      "eval_accuracy_speedway bricks": 0.9756119963835097,
-      "eval_accuracy_steel": 0.8969873105199239,
-      "eval_accuracy_tiles": 0.039383198521615315,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9765665415339276,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8440068262299035,
-      "eval_iou_concrete tiles": 0.7239714292071827,
-      "eval_iou_dark bricks": 0.800682565056982,
-      "eval_iou_grass": 0.7904247081326513,
-      "eval_iou_red bricks": 0.7613466865544428,
-      "eval_iou_road": 0.9455310239636968,
-      "eval_iou_rough concrete": 0.6013111898400443,
-      "eval_iou_rough red sidewalk": 0.5301668405239455,
-      "eval_iou_speedway bricks": 0.9648085398083491,
-      "eval_iou_steel": 0.7869885794342117,
-      "eval_iou_tiles": 0.03864791747183161,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09980905055999756,
-      "eval_mean_accuracy": 0.7145093488648399,
-      "eval_mean_iou": 0.6741886805967052,
-      "eval_overall_accuracy": 0.9711400694766287,
-      "eval_runtime": 6.4095,
-      "eval_samples_per_second": 18.41,
-      "eval_steps_per_second": 1.248,
-      "step": 360
-    },
-    {
-      "epoch": 21.24,
-      "learning_rate": 4.704334365325078e-05,
-      "loss": 0.0638,
-      "step": 361
-    },
-    {
-      "epoch": 21.29,
-      "learning_rate": 4.702786377708979e-05,
-      "loss": 0.0664,
-      "step": 362
-    },
-    {
-      "epoch": 21.35,
-      "learning_rate": 4.70123839009288e-05,
-      "loss": 0.0702,
-      "step": 363
-    },
-    {
-      "epoch": 21.41,
-      "learning_rate": 4.69969040247678e-05,
-      "loss": 0.0618,
-      "step": 364
-    },
-    {
-      "epoch": 21.47,
-      "learning_rate": 4.698142414860681e-05,
-      "loss": 0.0729,
-      "step": 365
-    },
-    {
-      "epoch": 21.53,
-      "learning_rate": 4.696594427244582e-05,
-      "loss": 0.092,
-      "step": 366
-    },
-    {
-      "epoch": 21.59,
-      "learning_rate": 4.695046439628483e-05,
-      "loss": 0.0798,
-      "step": 367
-    },
-    {
-      "epoch": 21.65,
-      "learning_rate": 4.693498452012384e-05,
-      "loss": 0.0659,
-      "step": 368
-    },
-    {
-      "epoch": 21.71,
-      "learning_rate": 4.691950464396285e-05,
-      "loss": 0.0672,
-      "step": 369
-    },
-    {
-      "epoch": 21.76,
-      "learning_rate": 4.690402476780186e-05,
-      "loss": 0.0756,
-      "step": 370
-    },
-    {
-      "epoch": 21.82,
-      "learning_rate": 4.688854489164087e-05,
-      "loss": 0.0689,
-      "step": 371
-    },
-    {
-      "epoch": 21.88,
-      "learning_rate": 4.687306501547988e-05,
-      "loss": 0.0713,
-      "step": 372
-    },
-    {
-      "epoch": 21.94,
-      "learning_rate": 4.685758513931889e-05,
-      "loss": 0.0562,
-      "step": 373
-    },
-    {
-      "epoch": 22.0,
-      "learning_rate": 4.68421052631579e-05,
-      "loss": 0.0657,
-      "step": 374
-    },
-    {
-      "epoch": 22.06,
-      "learning_rate": 4.682662538699691e-05,
-      "loss": 0.0626,
-      "step": 375
-    },
-    {
-      "epoch": 22.12,
-      "learning_rate": 4.6811145510835914e-05,
-      "loss": 0.065,
-      "step": 376
-    },
-    {
-      "epoch": 22.18,
-      "learning_rate": 4.6795665634674924e-05,
-      "loss": 0.0681,
-      "step": 377
-    },
-    {
-      "epoch": 22.24,
-      "learning_rate": 4.6780185758513934e-05,
-      "loss": 0.0616,
-      "step": 378
-    },
-    {
-      "epoch": 22.29,
-      "learning_rate": 4.6764705882352944e-05,
-      "loss": 0.0726,
-      "step": 379
-    },
-    {
-      "epoch": 22.35,
-      "learning_rate": 4.6749226006191954e-05,
-      "loss": 0.0671,
-      "step": 380
-    },
-    {
-      "epoch": 22.35,
-      "eval_accuracy_NAT": 0.9918347945306951,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9463292815111854,
-      "eval_accuracy_concrete tiles": 0.7548432712703276,
-      "eval_accuracy_dark bricks": 0.8054093371360757,
-      "eval_accuracy_grass": 0.8488552536458156,
-      "eval_accuracy_red bricks": 0.8876318101933216,
-      "eval_accuracy_road": 0.967290526244909,
-      "eval_accuracy_rough concrete": 0.6129401319278508,
-      "eval_accuracy_rough red sidewalk": 0.7476669248527963,
-      "eval_accuracy_speedway bricks": 0.9791248107271846,
-      "eval_accuracy_steel": 0.8704263645314289,
-      "eval_accuracy_tiles": 0.1060013935593323,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9774210083281781,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8496639620054871,
-      "eval_iou_concrete tiles": 0.7044694703501672,
-      "eval_iou_dark bricks": 0.7850040111191955,
-      "eval_iou_grass": 0.7914839253842089,
-      "eval_iou_red bricks": 0.8270815209047644,
-      "eval_iou_road": 0.9404892116236474,
-      "eval_iou_rough concrete": 0.5939658159140778,
-      "eval_iou_rough red sidewalk": 0.5748613785248016,
-      "eval_iou_speedway bricks": 0.9670993189974229,
-      "eval_iou_steel": 0.7935183012536479,
-      "eval_iou_tiles": 0.10403472779710403,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09570205211639404,
-      "eval_mean_accuracy": 0.7321810692408403,
-      "eval_mean_iou": 0.685314819400208,
-      "eval_overall_accuracy": 0.9719316514871889,
-      "eval_runtime": 5.6048,
-      "eval_samples_per_second": 21.053,
-      "eval_steps_per_second": 1.427,
-      "step": 380
-    },
-    {
-      "epoch": 22.41,
-      "learning_rate": 4.6733746130030964e-05,
-      "loss": 0.0598,
-      "step": 381
-    },
-    {
-      "epoch": 22.47,
-      "learning_rate": 4.671826625386997e-05,
-      "loss": 0.0697,
-      "step": 382
-    },
-    {
-      "epoch": 22.53,
-      "learning_rate": 4.670278637770898e-05,
-      "loss": 0.0661,
-      "step": 383
-    },
-    {
-      "epoch": 22.59,
-      "learning_rate": 4.668730650154799e-05,
-      "loss": 0.0627,
-      "step": 384
-    },
-    {
-      "epoch": 22.65,
-      "learning_rate": 4.6671826625387e-05,
-      "loss": 0.0702,
-      "step": 385
-    },
-    {
-      "epoch": 22.71,
-      "learning_rate": 4.665634674922601e-05,
-      "loss": 0.0587,
-      "step": 386
-    },
-    {
-      "epoch": 22.76,
-      "learning_rate": 4.664086687306502e-05,
-      "loss": 0.0846,
-      "step": 387
-    },
-    {
-      "epoch": 22.82,
-      "learning_rate": 4.662538699690403e-05,
-      "loss": 0.0624,
-      "step": 388
-    },
-    {
-      "epoch": 22.88,
-      "learning_rate": 4.660990712074304e-05,
-      "loss": 0.0665,
-      "step": 389
-    },
-    {
-      "epoch": 22.94,
-      "learning_rate": 4.659442724458205e-05,
-      "loss": 0.0736,
-      "step": 390
-    },
-    {
-      "epoch": 23.0,
-      "learning_rate": 4.657894736842106e-05,
-      "loss": 0.0619,
-      "step": 391
-    },
-    {
-      "epoch": 23.06,
-      "learning_rate": 4.656346749226007e-05,
-      "loss": 0.08,
-      "step": 392
-    },
-    {
-      "epoch": 23.12,
-      "learning_rate": 4.654798761609908e-05,
-      "loss": 0.063,
-      "step": 393
-    },
-    {
-      "epoch": 23.18,
-      "learning_rate": 4.653250773993808e-05,
-      "loss": 0.0666,
-      "step": 394
-    },
-    {
-      "epoch": 23.24,
-      "learning_rate": 4.651702786377709e-05,
-      "loss": 0.0528,
-      "step": 395
-    },
-    {
-      "epoch": 23.29,
-      "learning_rate": 4.65015479876161e-05,
-      "loss": 0.0542,
-      "step": 396
-    },
-    {
-      "epoch": 23.35,
-      "learning_rate": 4.648606811145511e-05,
-      "loss": 0.0632,
-      "step": 397
-    },
-    {
-      "epoch": 23.41,
-      "learning_rate": 4.647058823529412e-05,
-      "loss": 0.0504,
-      "step": 398
-    },
-    {
-      "epoch": 23.47,
-      "learning_rate": 4.645510835913313e-05,
-      "loss": 0.0689,
-      "step": 399
-    },
-    {
-      "epoch": 23.53,
-      "learning_rate": 4.6439628482972134e-05,
-      "loss": 0.0917,
-      "step": 400
-    },
-    {
-      "epoch": 23.53,
-      "eval_accuracy_NAT": 0.9909434175063019,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9335657359858984,
-      "eval_accuracy_concrete tiles": 0.7849870374734857,
-      "eval_accuracy_dark bricks": 0.8096204276170971,
-      "eval_accuracy_grass": 0.8686938650376409,
-      "eval_accuracy_red bricks": 0.8738466608084359,
-      "eval_accuracy_road": 0.9780349625721663,
-      "eval_accuracy_rough concrete": 0.6157526859294835,
-      "eval_accuracy_rough red sidewalk": 0.6848976154129599,
-      "eval_accuracy_speedway bricks": 0.9868086529807598,
-      "eval_accuracy_steel": 0.8361887578903177,
-      "eval_accuracy_tiles": 0.10460783422702899,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9777327551888214,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8544803622196483,
-      "eval_iou_concrete tiles": 0.732021978021978,
-      "eval_iou_dark bricks": 0.7838293614024684,
-      "eval_iou_grass": 0.8003822054755468,
-      "eval_iou_red bricks": 0.8379502843901412,
-      "eval_iou_road": 0.9416817001622897,
-      "eval_iou_rough concrete": 0.599613545631789,
-      "eval_iou_rough red sidewalk": 0.5411039883077764,
-      "eval_iou_speedway bricks": 0.9665345059616663,
-      "eval_iou_steel": 0.7894853567442969,
-      "eval_iou_tiles": 0.1043202416918429,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.09354794770479202,
-      "eval_mean_accuracy": 0.7283036656493521,
-      "eval_mean_iou": 0.6868566373229436,
-      "eval_overall_accuracy": 0.9724575301348153,
-      "eval_runtime": 6.1543,
-      "eval_samples_per_second": 19.174,
-      "eval_steps_per_second": 1.3,
-      "step": 400
-    },
-    {
-      "epoch": 23.59,
-      "learning_rate": 4.6424148606811144e-05,
-      "loss": 0.0554,
-      "step": 401
-    },
-    {
-      "epoch": 23.65,
-      "learning_rate": 4.6408668730650154e-05,
-      "loss": 0.0656,
-      "step": 402
-    },
-    {
-      "epoch": 23.71,
-      "learning_rate": 4.6393188854489164e-05,
-      "loss": 0.0757,
-      "step": 403
-    },
-    {
-      "epoch": 23.76,
-      "learning_rate": 4.637770897832818e-05,
-      "loss": 0.0679,
-      "step": 404
-    },
-    {
-      "epoch": 23.82,
-      "learning_rate": 4.636222910216719e-05,
-      "loss": 0.0602,
-      "step": 405
-    },
-    {
-      "epoch": 23.88,
-      "learning_rate": 4.6346749226006194e-05,
-      "loss": 0.0704,
-      "step": 406
-    },
-    {
-      "epoch": 23.94,
-      "learning_rate": 4.6331269349845204e-05,
-      "loss": 0.0603,
-      "step": 407
-    },
-    {
-      "epoch": 24.0,
-      "learning_rate": 4.6315789473684214e-05,
-      "loss": 0.0736,
-      "step": 408
-    },
-    {
-      "epoch": 24.06,
-      "learning_rate": 4.6300309597523224e-05,
-      "loss": 0.0631,
-      "step": 409
-    },
-    {
-      "epoch": 24.12,
-      "learning_rate": 4.6284829721362234e-05,
-      "loss": 0.058,
-      "step": 410
-    },
-    {
-      "epoch": 24.18,
-      "learning_rate": 4.6269349845201244e-05,
-      "loss": 0.0653,
-      "step": 411
-    },
-    {
-      "epoch": 24.24,
-      "learning_rate": 4.625386996904025e-05,
-      "loss": 0.0701,
-      "step": 412
-    },
-    {
-      "epoch": 24.29,
-      "learning_rate": 4.623839009287926e-05,
-      "loss": 0.0489,
-      "step": 413
-    },
-    {
-      "epoch": 24.35,
-      "learning_rate": 4.622291021671827e-05,
-      "loss": 0.0624,
-      "step": 414
-    },
-    {
-      "epoch": 24.41,
-      "learning_rate": 4.620743034055728e-05,
-      "loss": 0.065,
-      "step": 415
-    },
-    {
-      "epoch": 24.47,
-      "learning_rate": 4.619195046439629e-05,
-      "loss": 0.0596,
-      "step": 416
-    },
-    {
-      "epoch": 24.53,
-      "learning_rate": 4.61764705882353e-05,
-      "loss": 0.0619,
-      "step": 417
-    },
-    {
-      "epoch": 24.59,
-      "learning_rate": 4.616099071207431e-05,
-      "loss": 0.0546,
-      "step": 418
-    },
-    {
-      "epoch": 24.65,
-      "learning_rate": 4.614551083591332e-05,
-      "loss": 0.0572,
-      "step": 419
-    },
-    {
-      "epoch": 24.71,
-      "learning_rate": 4.613003095975233e-05,
-      "loss": 0.0649,
-      "step": 420
-    },
-    {
-      "epoch": 24.71,
-      "eval_accuracy_NAT": 0.9925458906327725,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9282929144836717,
-      "eval_accuracy_concrete tiles": 0.8514023096865425,
-      "eval_accuracy_dark bricks": 0.8465153226269548,
-      "eval_accuracy_grass": 0.8541730537041032,
-      "eval_accuracy_red bricks": 0.9041080843585237,
-      "eval_accuracy_road": 0.9788007808888516,
-      "eval_accuracy_rough concrete": 0.623808984679923,
-      "eval_accuracy_rough red sidewalk": 0.6470946540431196,
-      "eval_accuracy_speedway bricks": 0.9883149527066442,
-      "eval_accuracy_steel": 0.8562525301493851,
-      "eval_accuracy_tiles": 0.13562967675482443,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9782598114699761,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8609499442060025,
-      "eval_iou_concrete tiles": 0.7683064293158085,
-      "eval_iou_dark bricks": 0.816763934547335,
-      "eval_iou_grass": 0.8033950859006216,
-      "eval_iou_red bricks": 0.8423476436575756,
-      "eval_iou_road": 0.9410264736316956,
-      "eval_iou_rough concrete": 0.6038078648963536,
-      "eval_iou_rough red sidewalk": 0.544589232258922,
-      "eval_iou_speedway bricks": 0.9664483963533285,
-      "eval_iou_steel": 0.7954582387326526,
-      "eval_iou_tiles": 0.13351027346196284,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08796583861112595,
-      "eval_mean_accuracy": 0.7389953195934861,
-      "eval_mean_iou": 0.696527948340941,
-      "eval_overall_accuracy": 0.9733388545149464,
-      "eval_runtime": 8.4846,
-      "eval_samples_per_second": 13.908,
-      "eval_steps_per_second": 0.943,
-      "step": 420
-    },
-    {
-      "epoch": 24.76,
-      "learning_rate": 4.611455108359134e-05,
-      "loss": 0.0556,
-      "step": 421
-    },
-    {
-      "epoch": 24.82,
-      "learning_rate": 4.609907120743035e-05,
-      "loss": 0.0623,
-      "step": 422
-    },
-    {
-      "epoch": 24.88,
-      "learning_rate": 4.608359133126936e-05,
-      "loss": 0.0669,
-      "step": 423
-    },
-    {
-      "epoch": 24.94,
-      "learning_rate": 4.606811145510836e-05,
-      "loss": 0.0577,
-      "step": 424
-    },
-    {
-      "epoch": 25.0,
-      "learning_rate": 4.605263157894737e-05,
-      "loss": 0.0598,
-      "step": 425
-    },
-    {
-      "epoch": 25.06,
-      "learning_rate": 4.603715170278638e-05,
-      "loss": 0.0633,
-      "step": 426
-    },
-    {
-      "epoch": 25.12,
-      "learning_rate": 4.602167182662539e-05,
-      "loss": 0.0547,
-      "step": 427
-    },
-    {
-      "epoch": 25.18,
-      "learning_rate": 4.60061919504644e-05,
-      "loss": 0.0553,
-      "step": 428
-    },
-    {
-      "epoch": 25.24,
-      "learning_rate": 4.599071207430341e-05,
-      "loss": 0.0568,
-      "step": 429
-    },
-    {
-      "epoch": 25.29,
-      "learning_rate": 4.597523219814241e-05,
-      "loss": 0.052,
-      "step": 430
-    },
-    {
-      "epoch": 25.35,
-      "learning_rate": 4.595975232198142e-05,
-      "loss": 0.0671,
-      "step": 431
-    },
-    {
-      "epoch": 25.41,
-      "learning_rate": 4.594427244582043e-05,
-      "loss": 0.0603,
-      "step": 432
-    },
-    {
-      "epoch": 25.47,
-      "learning_rate": 4.592879256965944e-05,
-      "loss": 0.0586,
-      "step": 433
-    },
-    {
-      "epoch": 25.53,
-      "learning_rate": 4.591331269349845e-05,
-      "loss": 0.0734,
-      "step": 434
-    },
-    {
-      "epoch": 25.59,
-      "learning_rate": 4.589783281733746e-05,
-      "loss": 0.0559,
-      "step": 435
-    },
-    {
-      "epoch": 25.65,
-      "learning_rate": 4.588235294117647e-05,
-      "loss": 0.0498,
-      "step": 436
-    },
-    {
-      "epoch": 25.71,
-      "learning_rate": 4.586687306501548e-05,
-      "loss": 0.0595,
-      "step": 437
-    },
-    {
-      "epoch": 25.76,
-      "learning_rate": 4.585139318885449e-05,
-      "loss": 0.0491,
-      "step": 438
-    },
-    {
-      "epoch": 25.82,
-      "learning_rate": 4.58359133126935e-05,
-      "loss": 0.0623,
-      "step": 439
-    },
-    {
-      "epoch": 25.88,
-      "learning_rate": 4.582043343653251e-05,
-      "loss": 0.0648,
-      "step": 440
-    },
-    {
-      "epoch": 25.88,
-      "eval_accuracy_NAT": 0.9919881734852373,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9411478056409559,
-      "eval_accuracy_concrete tiles": 0.8343860476078246,
-      "eval_accuracy_dark bricks": 0.8282257910150642,
-      "eval_accuracy_grass": 0.8704670958755824,
-      "eval_accuracy_red bricks": 0.8488027240773286,
-      "eval_accuracy_road": 0.9759058487035867,
-      "eval_accuracy_rough concrete": 0.6163366484128734,
-      "eval_accuracy_rough red sidewalk": 0.7025846186239932,
-      "eval_accuracy_speedway bricks": 0.9869561983805307,
-      "eval_accuracy_steel": 0.834177132947168,
-      "eval_accuracy_tiles": 0.10306280105425793,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9788527237848805,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8594148468537138,
-      "eval_iou_concrete tiles": 0.79045726533893,
-      "eval_iou_dark bricks": 0.8074231652018138,
-      "eval_iou_grass": 0.8148623846709846,
-      "eval_iou_red bricks": 0.8211572180011689,
-      "eval_iou_road": 0.9523440987621357,
-      "eval_iou_rough concrete": 0.5988443790853457,
-      "eval_iou_rough red sidewalk": 0.5235994978591764,
-      "eval_iou_speedway bricks": 0.9685424964705743,
-      "eval_iou_steel": 0.792603356412229,
-      "eval_iou_tiles": 0.1028105167724388,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08859998732805252,
-      "eval_mean_accuracy": 0.733387760448031,
-      "eval_mean_iou": 0.6931470730164146,
-      "eval_overall_accuracy": 0.973915197081485,
-      "eval_runtime": 10.9462,
-      "eval_samples_per_second": 10.78,
-      "eval_steps_per_second": 0.731,
-      "step": 440
-    },
-    {
-      "epoch": 25.94,
-      "learning_rate": 4.580495356037152e-05,
-      "loss": 0.05,
-      "step": 441
-    },
-    {
-      "epoch": 26.0,
-      "learning_rate": 4.5789473684210527e-05,
-      "loss": 0.0547,
-      "step": 442
-    },
-    {
-      "epoch": 26.06,
-      "learning_rate": 4.5773993808049537e-05,
-      "loss": 0.0636,
-      "step": 443
-    },
-    {
-      "epoch": 26.12,
-      "learning_rate": 4.5758513931888547e-05,
-      "loss": 0.0503,
-      "step": 444
-    },
-    {
-      "epoch": 26.18,
-      "learning_rate": 4.5743034055727557e-05,
-      "loss": 0.0486,
-      "step": 445
-    },
-    {
-      "epoch": 26.24,
-      "learning_rate": 4.5727554179566567e-05,
-      "loss": 0.0596,
-      "step": 446
-    },
-    {
-      "epoch": 26.29,
-      "learning_rate": 4.5712074303405577e-05,
-      "loss": 0.0578,
-      "step": 447
-    },
-    {
-      "epoch": 26.35,
-      "learning_rate": 4.569659442724458e-05,
-      "loss": 0.0709,
-      "step": 448
-    },
-    {
-      "epoch": 26.41,
-      "learning_rate": 4.568111455108359e-05,
-      "loss": 0.0521,
-      "step": 449
-    },
-    {
-      "epoch": 26.47,
-      "learning_rate": 4.56656346749226e-05,
-      "loss": 0.0509,
-      "step": 450
-    },
-    {
-      "epoch": 26.53,
-      "learning_rate": 4.565015479876161e-05,
-      "loss": 0.0582,
-      "step": 451
-    },
-    {
-      "epoch": 26.59,
-      "learning_rate": 4.563467492260062e-05,
-      "loss": 0.0571,
-      "step": 452
-    },
-    {
-      "epoch": 26.65,
-      "learning_rate": 4.561919504643964e-05,
-      "loss": 0.0596,
-      "step": 453
-    },
-    {
-      "epoch": 26.71,
-      "learning_rate": 4.560371517027864e-05,
-      "loss": 0.0588,
-      "step": 454
-    },
-    {
-      "epoch": 26.76,
-      "learning_rate": 4.558823529411765e-05,
-      "loss": 0.0409,
-      "step": 455
-    },
-    {
-      "epoch": 26.82,
-      "learning_rate": 4.557275541795666e-05,
-      "loss": 0.0521,
-      "step": 456
-    },
-    {
-      "epoch": 26.88,
-      "learning_rate": 4.555727554179567e-05,
-      "loss": 0.0638,
-      "step": 457
-    },
-    {
-      "epoch": 26.94,
-      "learning_rate": 4.554179566563468e-05,
-      "loss": 0.0627,
-      "step": 458
-    },
-    {
-      "epoch": 27.0,
-      "learning_rate": 4.552631578947369e-05,
-      "loss": 0.0633,
-      "step": 459
-    },
-    {
-      "epoch": 27.06,
-      "learning_rate": 4.551083591331269e-05,
-      "loss": 0.0633,
-      "step": 460
-    },
-    {
-      "epoch": 27.06,
-      "eval_accuracy_NAT": 0.9912690984428182,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9297857849379844,
-      "eval_accuracy_concrete tiles": 0.8931180768324298,
-      "eval_accuracy_dark bricks": 0.8600290948069598,
-      "eval_accuracy_grass": 0.8679898715779673,
-      "eval_accuracy_red bricks": 0.9230008787346221,
-      "eval_accuracy_road": 0.9769875069540899,
-      "eval_accuracy_rough concrete": 0.6186665395455818,
-      "eval_accuracy_rough red sidewalk": 0.7836726099609713,
-      "eval_accuracy_speedway bricks": 0.9904856351443888,
-      "eval_accuracy_steel": 0.8313108797673012,
-      "eval_accuracy_tiles": 0.13214577842406616,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9784511593154107,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8657692444570981,
-      "eval_iou_concrete tiles": 0.7636887608069164,
-      "eval_iou_dark bricks": 0.8113273984723451,
-      "eval_iou_grass": 0.8026909321920439,
-      "eval_iou_red bricks": 0.8375778719162721,
-      "eval_iou_road": 0.945710151690017,
-      "eval_iou_rough concrete": 0.6000277405335429,
-      "eval_iou_rough red sidewalk": 0.6096657754792814,
-      "eval_iou_speedway bricks": 0.9673607821937777,
-      "eval_iou_steel": 0.7921859893603341,
-      "eval_iou_tiles": 0.13162341581170792,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08854084461927414,
-      "eval_mean_accuracy": 0.7537278273176292,
-      "eval_mean_iou": 0.7004676324791345,
-      "eval_overall_accuracy": 0.9738638603081138,
-      "eval_runtime": 5.7428,
-      "eval_samples_per_second": 20.548,
-      "eval_steps_per_second": 1.393,
-      "step": 460
-    },
-    {
-      "epoch": 27.12,
-      "learning_rate": 4.54953560371517e-05,
-      "loss": 0.0449,
-      "step": 461
-    },
-    {
-      "epoch": 27.18,
-      "learning_rate": 4.547987616099071e-05,
-      "loss": 0.0513,
-      "step": 462
-    },
-    {
-      "epoch": 27.24,
-      "learning_rate": 4.546439628482972e-05,
-      "loss": 0.0553,
-      "step": 463
-    },
-    {
-      "epoch": 27.29,
-      "learning_rate": 4.544891640866873e-05,
-      "loss": 0.0509,
-      "step": 464
-    },
-    {
-      "epoch": 27.35,
-      "learning_rate": 4.5433436532507736e-05,
-      "loss": 0.0608,
-      "step": 465
-    },
-    {
-      "epoch": 27.41,
-      "learning_rate": 4.5417956656346746e-05,
-      "loss": 0.0623,
-      "step": 466
-    },
-    {
-      "epoch": 27.47,
-      "learning_rate": 4.540247678018576e-05,
-      "loss": 0.0538,
-      "step": 467
-    },
-    {
-      "epoch": 27.53,
-      "learning_rate": 4.538699690402477e-05,
-      "loss": 0.0662,
-      "step": 468
-    },
-    {
-      "epoch": 27.59,
-      "learning_rate": 4.537151702786378e-05,
-      "loss": 0.0469,
-      "step": 469
-    },
-    {
-      "epoch": 27.65,
-      "learning_rate": 4.535603715170279e-05,
-      "loss": 0.0562,
-      "step": 470
-    },
-    {
-      "epoch": 27.71,
-      "learning_rate": 4.5340557275541796e-05,
-      "loss": 0.0491,
-      "step": 471
-    },
-    {
-      "epoch": 27.76,
-      "learning_rate": 4.5325077399380806e-05,
-      "loss": 0.0563,
-      "step": 472
-    },
-    {
-      "epoch": 27.82,
-      "learning_rate": 4.5309597523219816e-05,
-      "loss": 0.0535,
-      "step": 473
-    },
-    {
-      "epoch": 27.88,
-      "learning_rate": 4.5294117647058826e-05,
-      "loss": 0.0462,
-      "step": 474
-    },
-    {
-      "epoch": 27.94,
-      "learning_rate": 4.5278637770897836e-05,
-      "loss": 0.0533,
-      "step": 475
-    },
-    {
-      "epoch": 28.0,
-      "learning_rate": 4.5263157894736846e-05,
-      "loss": 0.0563,
-      "step": 476
-    },
-    {
-      "epoch": 28.06,
-      "learning_rate": 4.524767801857585e-05,
-      "loss": 0.0529,
-      "step": 477
-    },
-    {
-      "epoch": 28.12,
-      "learning_rate": 4.523219814241486e-05,
-      "loss": 0.0604,
-      "step": 478
-    },
-    {
-      "epoch": 28.18,
-      "learning_rate": 4.521671826625387e-05,
-      "loss": 0.0485,
-      "step": 479
-    },
-    {
-      "epoch": 28.24,
-      "learning_rate": 4.520123839009288e-05,
-      "loss": 0.0504,
-      "step": 480
-    },
-    {
-      "epoch": 28.24,
-      "eval_accuracy_NAT": 0.9911260166342576,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9331674843183628,
-      "eval_accuracy_concrete tiles": 0.8847749234032524,
-      "eval_accuracy_dark bricks": 0.8469077196945045,
-      "eval_accuracy_grass": 0.8925321812064025,
-      "eval_accuracy_red bricks": 0.9044925307557118,
-      "eval_accuracy_road": 0.9779024273715117,
-      "eval_accuracy_rough concrete": 0.6204899326059624,
-      "eval_accuracy_rough red sidewalk": 0.7195572233209755,
-      "eval_accuracy_speedway bricks": 0.9882148754552316,
-      "eval_accuracy_steel": 0.8626022680134142,
-      "eval_accuracy_tiles": 0.17510375958072041,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9790429664294764,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8675571401566566,
-      "eval_iou_concrete tiles": 0.7879481151876416,
-      "eval_iou_dark bricks": 0.8161326619076605,
-      "eval_iou_grass": 0.8155797060723207,
-      "eval_iou_red bricks": 0.8463870901428718,
-      "eval_iou_road": 0.9503152653229957,
-      "eval_iou_rough concrete": 0.5960606077952111,
-      "eval_iou_rough red sidewalk": 0.536717522997532,
-      "eval_iou_speedway bricks": 0.9697299086152923,
-      "eval_iou_steel": 0.8085007213925687,
-      "eval_iou_tiles": 0.17393397731034274,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08616523444652557,
-      "eval_mean_accuracy": 0.7536054878738698,
-      "eval_mean_iou": 0.7036850525638899,
-      "eval_overall_accuracy": 0.974603491314387,
-      "eval_runtime": 5.9007,
-      "eval_samples_per_second": 19.997,
-      "eval_steps_per_second": 1.356,
-      "step": 480
-    },
-    {
-      "epoch": 28.29,
-      "learning_rate": 4.518575851393189e-05,
-      "loss": 0.0567,
-      "step": 481
-    },
-    {
-      "epoch": 28.35,
-      "learning_rate": 4.51702786377709e-05,
-      "loss": 0.0511,
-      "step": 482
-    },
-    {
-      "epoch": 28.41,
-      "learning_rate": 4.515479876160991e-05,
-      "loss": 0.0489,
-      "step": 483
-    },
-    {
-      "epoch": 28.47,
-      "learning_rate": 4.513931888544892e-05,
-      "loss": 0.0492,
-      "step": 484
-    },
-    {
-      "epoch": 28.53,
-      "learning_rate": 4.512383900928793e-05,
-      "loss": 0.0502,
-      "step": 485
-    },
-    {
-      "epoch": 28.59,
-      "learning_rate": 4.510835913312694e-05,
-      "loss": 0.0536,
-      "step": 486
-    },
-    {
-      "epoch": 28.65,
-      "learning_rate": 4.509287925696595e-05,
-      "loss": 0.0479,
-      "step": 487
-    },
-    {
-      "epoch": 28.71,
-      "learning_rate": 4.507739938080496e-05,
-      "loss": 0.048,
-      "step": 488
-    },
-    {
-      "epoch": 28.76,
-      "learning_rate": 4.506191950464396e-05,
-      "loss": 0.0573,
-      "step": 489
-    },
-    {
-      "epoch": 28.82,
-      "learning_rate": 4.504643962848297e-05,
-      "loss": 0.0667,
-      "step": 490
-    },
-    {
-      "epoch": 28.88,
-      "learning_rate": 4.503095975232198e-05,
-      "loss": 0.0514,
-      "step": 491
-    },
-    {
-      "epoch": 28.94,
-      "learning_rate": 4.501547987616099e-05,
-      "loss": 0.052,
-      "step": 492
-    },
-    {
-      "epoch": 29.0,
-      "learning_rate": 4.5e-05,
-      "loss": 0.0556,
-      "step": 493
-    },
-    {
-      "epoch": 29.06,
-      "learning_rate": 4.498452012383901e-05,
-      "loss": 0.0601,
-      "step": 494
-    },
-    {
-      "epoch": 29.12,
-      "learning_rate": 4.4969040247678016e-05,
-      "loss": 0.0497,
-      "step": 495
-    },
-    {
-      "epoch": 29.18,
-      "learning_rate": 4.4953560371517026e-05,
-      "loss": 0.0451,
-      "step": 496
-    },
-    {
-      "epoch": 29.24,
-      "learning_rate": 4.4938080495356036e-05,
-      "loss": 0.0537,
-      "step": 497
-    },
-    {
-      "epoch": 29.29,
-      "learning_rate": 4.4922600619195046e-05,
-      "loss": 0.0474,
-      "step": 498
-    },
-    {
-      "epoch": 29.35,
-      "learning_rate": 4.4907120743034056e-05,
-      "loss": 0.0533,
-      "step": 499
-    },
-    {
-      "epoch": 29.41,
-      "learning_rate": 4.4891640866873066e-05,
-      "loss": 0.0603,
-      "step": 500
-    },
-    {
-      "epoch": 29.41,
-      "eval_accuracy_NAT": 0.9932370932957547,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.942857383096701,
-      "eval_accuracy_concrete tiles": 0.8190666980909733,
-      "eval_accuracy_dark bricks": 0.8274314262197806,
-      "eval_accuracy_grass": 0.8522957378116401,
-      "eval_accuracy_red bricks": 0.8900483304042179,
-      "eval_accuracy_road": 0.9719020030451031,
-      "eval_accuracy_rough concrete": 0.6163962364213825,
-      "eval_accuracy_rough red sidewalk": 0.7169628287173163,
-      "eval_accuracy_speedway bricks": 0.9857528722513885,
-      "eval_accuracy_steel": 0.8697291678120018,
-      "eval_accuracy_tiles": 0.138083552970402,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9782592362006232,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8635047513437755,
-      "eval_iou_concrete tiles": 0.7952267630772047,
-      "eval_iou_dark bricks": 0.8104902971782132,
-      "eval_iou_grass": 0.8026895039834602,
-      "eval_iou_red bricks": 0.8525883838383839,
-      "eval_iou_road": 0.9459653181836624,
-      "eval_iou_rough concrete": 0.6043007863159986,
-      "eval_iou_rough red sidewalk": 0.5906685996084554,
-      "eval_iou_speedway bricks": 0.9711059681367812,
-      "eval_iou_steel": 0.8125372075052353,
-      "eval_iou_tiles": 0.137033251157477,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08850965648889542,
-      "eval_mean_accuracy": 0.7402894869335893,
-      "eval_mean_iou": 0.7049515435791746,
-      "eval_overall_accuracy": 0.9742760092525159,
-      "eval_runtime": 8.0307,
-      "eval_samples_per_second": 14.694,
-      "eval_steps_per_second": 0.996,
-      "step": 500
-    },
-    {
-      "epoch": 29.47,
-      "learning_rate": 4.4876160990712076e-05,
-      "loss": 0.0463,
-      "step": 501
-    },
-    {
-      "epoch": 29.53,
-      "learning_rate": 4.4860681114551086e-05,
-      "loss": 0.0528,
-      "step": 502
-    },
-    {
-      "epoch": 29.59,
-      "learning_rate": 4.4845201238390096e-05,
-      "loss": 0.0575,
-      "step": 503
-    },
-    {
-      "epoch": 29.65,
-      "learning_rate": 4.4829721362229106e-05,
-      "loss": 0.055,
-      "step": 504
-    },
-    {
-      "epoch": 29.71,
-      "learning_rate": 4.4814241486068116e-05,
-      "loss": 0.0654,
-      "step": 505
-    },
-    {
-      "epoch": 29.76,
-      "learning_rate": 4.4798761609907126e-05,
-      "loss": 0.057,
-      "step": 506
-    },
-    {
-      "epoch": 29.82,
-      "learning_rate": 4.478328173374613e-05,
-      "loss": 0.0434,
-      "step": 507
-    },
-    {
-      "epoch": 29.88,
-      "learning_rate": 4.476780185758514e-05,
-      "loss": 0.0492,
-      "step": 508
-    },
-    {
-      "epoch": 29.94,
-      "learning_rate": 4.475232198142415e-05,
-      "loss": 0.06,
-      "step": 509
-    },
-    {
-      "epoch": 30.0,
-      "learning_rate": 4.473684210526316e-05,
-      "loss": 0.0457,
-      "step": 510
-    },
-    {
-      "epoch": 30.06,
-      "learning_rate": 4.472136222910217e-05,
-      "loss": 0.0561,
-      "step": 511
-    },
-    {
-      "epoch": 30.12,
-      "learning_rate": 4.470588235294118e-05,
-      "loss": 0.046,
-      "step": 512
-    },
-    {
-      "epoch": 30.18,
-      "learning_rate": 4.469040247678018e-05,
-      "loss": 0.0678,
-      "step": 513
-    },
-    {
-      "epoch": 30.24,
-      "learning_rate": 4.467492260061919e-05,
-      "loss": 0.0558,
-      "step": 514
-    },
-    {
-      "epoch": 30.29,
-      "learning_rate": 4.46594427244582e-05,
-      "loss": 0.0437,
-      "step": 515
-    },
-    {
-      "epoch": 30.35,
-      "learning_rate": 4.464396284829722e-05,
-      "loss": 0.0544,
-      "step": 516
-    },
-    {
-      "epoch": 30.41,
-      "learning_rate": 4.462848297213623e-05,
-      "loss": 0.0559,
-      "step": 517
-    },
-    {
-      "epoch": 30.47,
-      "learning_rate": 4.461300309597524e-05,
-      "loss": 0.0521,
-      "step": 518
-    },
-    {
-      "epoch": 30.53,
-      "learning_rate": 4.459752321981424e-05,
-      "loss": 0.0522,
-      "step": 519
-    },
-    {
-      "epoch": 30.59,
-      "learning_rate": 4.458204334365325e-05,
-      "loss": 0.0447,
-      "step": 520
-    },
-    {
-      "epoch": 30.59,
-      "eval_accuracy_NAT": 0.9901760375277598,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9483530267891132,
-      "eval_accuracy_concrete tiles": 0.8590855526749941,
-      "eval_accuracy_dark bricks": 0.875892464062171,
-      "eval_accuracy_grass": 0.8866419133482459,
-      "eval_accuracy_red bricks": 0.9196507029876977,
-      "eval_accuracy_road": 0.9787553555176595,
-      "eval_accuracy_rough concrete": 0.6182732586894213,
-      "eval_accuracy_rough red sidewalk": 0.736898307251521,
-      "eval_accuracy_speedway bricks": 0.985416996544593,
-      "eval_accuracy_steel": 0.8896280118148605,
-      "eval_accuracy_tiles": 0.2665636644551486,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9794976770895875,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8698962107547464,
-      "eval_iou_concrete tiles": 0.8193077095976624,
-      "eval_iou_dark bricks": 0.8351637822077632,
-      "eval_iou_grass": 0.8186954037349569,
-      "eval_iou_red bricks": 0.8540317233641047,
-      "eval_iou_road": 0.9470304594120126,
-      "eval_iou_rough concrete": 0.6008338641496322,
-      "eval_iou_rough red sidewalk": 0.5913070763512168,
-      "eval_iou_speedway bricks": 0.9713626391149616,
-      "eval_iou_steel": 0.8156706945456378,
-      "eval_iou_tiles": 0.24534351996430961,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08243369311094284,
-      "eval_mean_accuracy": 0.7657950224356297,
-      "eval_mean_iou": 0.7190877507912763,
-      "eval_overall_accuracy": 0.9753200724973516,
-      "eval_runtime": 5.6665,
-      "eval_samples_per_second": 20.824,
-      "eval_steps_per_second": 1.412,
-      "step": 520
-    },
-    {
-      "epoch": 30.65,
-      "learning_rate": 4.456656346749226e-05,
-      "loss": 0.0519,
-      "step": 521
-    },
-    {
-      "epoch": 30.71,
-      "learning_rate": 4.455108359133127e-05,
-      "loss": 0.0496,
-      "step": 522
-    },
-    {
-      "epoch": 30.76,
-      "learning_rate": 4.453560371517028e-05,
-      "loss": 0.0504,
-      "step": 523
-    },
-    {
-      "epoch": 30.82,
-      "learning_rate": 4.452012383900929e-05,
-      "loss": 0.0531,
-      "step": 524
-    },
-    {
-      "epoch": 30.88,
-      "learning_rate": 4.4504643962848296e-05,
-      "loss": 0.0479,
-      "step": 525
-    },
-    {
-      "epoch": 30.94,
-      "learning_rate": 4.4489164086687306e-05,
-      "loss": 0.0583,
-      "step": 526
-    },
-    {
-      "epoch": 31.0,
-      "learning_rate": 4.4473684210526316e-05,
-      "loss": 0.0534,
-      "step": 527
-    },
-    {
-      "epoch": 31.06,
-      "learning_rate": 4.4458204334365326e-05,
-      "loss": 0.0472,
-      "step": 528
-    },
-    {
-      "epoch": 31.12,
-      "learning_rate": 4.4442724458204336e-05,
-      "loss": 0.0586,
-      "step": 529
-    },
-    {
-      "epoch": 31.18,
-      "learning_rate": 4.4427244582043346e-05,
-      "loss": 0.0465,
-      "step": 530
-    },
-    {
-      "epoch": 31.24,
-      "learning_rate": 4.4411764705882356e-05,
-      "loss": 0.0502,
-      "step": 531
-    },
-    {
-      "epoch": 31.29,
-      "learning_rate": 4.4396284829721366e-05,
-      "loss": 0.0479,
-      "step": 532
-    },
-    {
-      "epoch": 31.35,
-      "learning_rate": 4.4380804953560376e-05,
-      "loss": 0.0519,
-      "step": 533
-    },
-    {
-      "epoch": 31.41,
-      "learning_rate": 4.4365325077399386e-05,
-      "loss": 0.0551,
-      "step": 534
-    },
-    {
-      "epoch": 31.47,
-      "learning_rate": 4.4349845201238396e-05,
-      "loss": 0.0585,
-      "step": 535
-    },
-    {
-      "epoch": 31.53,
-      "learning_rate": 4.4334365325077406e-05,
-      "loss": 0.0463,
-      "step": 536
-    },
-    {
-      "epoch": 31.59,
-      "learning_rate": 4.431888544891641e-05,
-      "loss": 0.0478,
-      "step": 537
-    },
-    {
-      "epoch": 31.65,
-      "learning_rate": 4.430340557275542e-05,
-      "loss": 0.0489,
-      "step": 538
-    },
-    {
-      "epoch": 31.71,
-      "learning_rate": 4.428792569659443e-05,
-      "loss": 0.0542,
-      "step": 539
-    },
-    {
-      "epoch": 31.76,
-      "learning_rate": 4.427244582043344e-05,
-      "loss": 0.0517,
-      "step": 540
-    },
-    {
-      "epoch": 31.76,
-      "eval_accuracy_NAT": 0.9909922615600154,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9372061842903858,
-      "eval_accuracy_concrete tiles": 0.893683714353052,
-      "eval_accuracy_dark bricks": 0.8919950998219858,
-      "eval_accuracy_grass": 0.8814584776330681,
-      "eval_accuracy_red bricks": 0.9331063268892794,
-      "eval_accuracy_road": 0.9784058473675463,
-      "eval_accuracy_rough concrete": 0.6244704115743748,
-      "eval_accuracy_rough red sidewalk": 0.8062024830988351,
-      "eval_accuracy_speedway bricks": 0.9904345683277433,
-      "eval_accuracy_steel": 0.8430882566083754,
-      "eval_accuracy_tiles": 0.3040079978187767,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9798501078345754,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8732045489507899,
-      "eval_iou_concrete tiles": 0.8217002188657985,
-      "eval_iou_dark bricks": 0.8340544724796298,
-      "eval_iou_grass": 0.8191193203301274,
-      "eval_iou_red bricks": 0.8353820434654342,
-      "eval_iou_road": 0.9494913348518607,
-      "eval_iou_rough concrete": 0.6023739042965943,
-      "eval_iou_rough red sidewalk": 0.5985060795194445,
-      "eval_iou_speedway bricks": 0.9701875387971075,
-      "eval_iou_steel": 0.8044703660606985,
-      "eval_iou_tiles": 0.2819136981683335,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08074591308832169,
-      "eval_mean_accuracy": 0.7750039714879567,
-      "eval_mean_iou": 0.7207887410477227,
-      "eval_overall_accuracy": 0.9755485987259169,
-      "eval_runtime": 5.8591,
-      "eval_samples_per_second": 20.14,
-      "eval_steps_per_second": 1.365,
-      "step": 540
-    },
-    {
-      "epoch": 31.82,
-      "learning_rate": 4.425696594427245e-05,
-      "loss": 0.0505,
-      "step": 541
-    },
-    {
-      "epoch": 31.88,
-      "learning_rate": 4.424148606811146e-05,
-      "loss": 0.054,
-      "step": 542
-    },
-    {
-      "epoch": 31.94,
-      "learning_rate": 4.422600619195046e-05,
-      "loss": 0.0489,
-      "step": 543
-    },
-    {
-      "epoch": 32.0,
-      "learning_rate": 4.421052631578947e-05,
-      "loss": 0.0477,
-      "step": 544
-    },
-    {
-      "epoch": 32.06,
-      "learning_rate": 4.419504643962848e-05,
-      "loss": 0.042,
-      "step": 545
-    },
-    {
-      "epoch": 32.12,
-      "learning_rate": 4.417956656346749e-05,
-      "loss": 0.0537,
-      "step": 546
-    },
-    {
-      "epoch": 32.18,
-      "learning_rate": 4.41640866873065e-05,
-      "loss": 0.0536,
-      "step": 547
-    },
-    {
-      "epoch": 32.24,
-      "learning_rate": 4.414860681114551e-05,
-      "loss": 0.0545,
-      "step": 548
-    },
-    {
-      "epoch": 32.29,
-      "learning_rate": 4.413312693498452e-05,
-      "loss": 0.0469,
-      "step": 549
-    },
-    {
-      "epoch": 32.35,
-      "learning_rate": 4.411764705882353e-05,
-      "loss": 0.0414,
-      "step": 550
-    },
-    {
-      "epoch": 32.41,
-      "learning_rate": 4.410216718266254e-05,
-      "loss": 0.0474,
-      "step": 551
-    },
-    {
-      "epoch": 32.47,
-      "learning_rate": 4.408668730650155e-05,
-      "loss": 0.0579,
-      "step": 552
-    },
-    {
-      "epoch": 32.53,
-      "learning_rate": 4.407120743034056e-05,
-      "loss": 0.0456,
-      "step": 553
-    },
-    {
-      "epoch": 32.59,
-      "learning_rate": 4.405572755417957e-05,
-      "loss": 0.0478,
-      "step": 554
-    },
-    {
-      "epoch": 32.65,
-      "learning_rate": 4.4040247678018576e-05,
-      "loss": 0.0454,
-      "step": 555
-    },
-    {
-      "epoch": 32.71,
-      "learning_rate": 4.4024767801857586e-05,
-      "loss": 0.0446,
-      "step": 556
-    },
-    {
-      "epoch": 32.76,
-      "learning_rate": 4.4009287925696596e-05,
-      "loss": 0.0398,
-      "step": 557
-    },
-    {
-      "epoch": 32.82,
-      "learning_rate": 4.3993808049535606e-05,
-      "loss": 0.0474,
-      "step": 558
-    },
-    {
-      "epoch": 32.88,
-      "learning_rate": 4.3978328173374616e-05,
-      "loss": 0.0519,
-      "step": 559
-    },
-    {
-      "epoch": 32.94,
-      "learning_rate": 4.3962848297213626e-05,
-      "loss": 0.0512,
-      "step": 560
-    },
-    {
-      "epoch": 32.94,
-      "eval_accuracy_NAT": 0.9933581251948573,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9411397794557176,
-      "eval_accuracy_concrete tiles": 0.8627386283290125,
-      "eval_accuracy_dark bricks": 0.7835116666347645,
-      "eval_accuracy_grass": 0.8619198204438187,
-      "eval_accuracy_red bricks": 0.835951230228471,
-      "eval_accuracy_road": 0.9788659796569156,
-      "eval_accuracy_rough concrete": 0.6165869180486119,
-      "eval_accuracy_rough red sidewalk": 0.7804014167650531,
-      "eval_accuracy_speedway bricks": 0.9852581410444912,
-      "eval_accuracy_steel": 0.8938186888703414,
-      "eval_accuracy_tiles": 0.12823775333999818,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9793455250994089,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.873954996019624,
-      "eval_iou_concrete tiles": 0.8109616960942866,
-      "eval_iou_dark bricks": 0.7668982056122042,
-      "eval_iou_grass": 0.8081869093206233,
-      "eval_iou_red bricks": 0.822312263641275,
-      "eval_iou_road": 0.9476417280956264,
-      "eval_iou_rough concrete": 0.6077005749590364,
-      "eval_iou_rough red sidewalk": 0.6209246528214057,
-      "eval_iou_speedway bricks": 0.9704413251282072,
-      "eval_iou_steel": 0.8201518854988787,
-      "eval_iou_tiles": 0.1261774174317396,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08563321828842163,
-      "eval_mean_accuracy": 0.743214472924004,
-      "eval_mean_iou": 0.704207475363255,
-      "eval_overall_accuracy": 0.9751662884728384,
-      "eval_runtime": 5.9098,
-      "eval_samples_per_second": 19.967,
-      "eval_steps_per_second": 1.354,
-      "step": 560
-    },
-    {
-      "epoch": 33.0,
-      "learning_rate": 4.394736842105263e-05,
-      "loss": 0.0474,
-      "step": 561
-    },
-    {
-      "epoch": 33.06,
-      "learning_rate": 4.393188854489164e-05,
-      "loss": 0.0477,
-      "step": 562
-    },
-    {
-      "epoch": 33.12,
-      "learning_rate": 4.391640866873065e-05,
-      "loss": 0.0486,
-      "step": 563
-    },
-    {
-      "epoch": 33.18,
-      "learning_rate": 4.390092879256966e-05,
-      "loss": 0.0484,
-      "step": 564
-    },
-    {
-      "epoch": 33.24,
-      "learning_rate": 4.3885448916408676e-05,
-      "loss": 0.0493,
-      "step": 565
-    },
-    {
-      "epoch": 33.29,
-      "learning_rate": 4.3869969040247686e-05,
-      "loss": 0.0467,
-      "step": 566
-    },
-    {
-      "epoch": 33.35,
-      "learning_rate": 4.385448916408669e-05,
-      "loss": 0.0537,
-      "step": 567
-    },
-    {
-      "epoch": 33.41,
-      "learning_rate": 4.38390092879257e-05,
-      "loss": 0.0462,
-      "step": 568
-    },
-    {
-      "epoch": 33.47,
-      "learning_rate": 4.382352941176471e-05,
-      "loss": 0.0494,
-      "step": 569
-    },
-    {
-      "epoch": 33.53,
-      "learning_rate": 4.380804953560372e-05,
-      "loss": 0.0471,
-      "step": 570
-    },
-    {
-      "epoch": 33.59,
-      "learning_rate": 4.379256965944273e-05,
-      "loss": 0.0422,
-      "step": 571
-    },
-    {
-      "epoch": 33.65,
-      "learning_rate": 4.377708978328174e-05,
-      "loss": 0.0595,
-      "step": 572
-    },
-    {
-      "epoch": 33.71,
-      "learning_rate": 4.376160990712074e-05,
-      "loss": 0.0455,
-      "step": 573
-    },
-    {
-      "epoch": 33.76,
-      "learning_rate": 4.374613003095975e-05,
-      "loss": 0.0476,
-      "step": 574
-    },
-    {
-      "epoch": 33.82,
-      "learning_rate": 4.373065015479876e-05,
-      "loss": 0.0349,
-      "step": 575
-    },
-    {
-      "epoch": 33.88,
-      "learning_rate": 4.371517027863777e-05,
-      "loss": 0.049,
-      "step": 576
-    },
-    {
-      "epoch": 33.94,
-      "learning_rate": 4.369969040247678e-05,
-      "loss": 0.0478,
-      "step": 577
-    },
-    {
-      "epoch": 34.0,
-      "learning_rate": 4.368421052631579e-05,
-      "loss": 0.05,
-      "step": 578
-    },
-    {
-      "epoch": 34.06,
-      "learning_rate": 4.36687306501548e-05,
-      "loss": 0.0458,
-      "step": 579
-    },
-    {
-      "epoch": 34.12,
-      "learning_rate": 4.365325077399381e-05,
-      "loss": 0.0474,
-      "step": 580
-    },
-    {
-      "epoch": 34.12,
-      "eval_accuracy_NAT": 0.9925247572232188,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9510559402179147,
-      "eval_accuracy_concrete tiles": 0.8676879566344567,
-      "eval_accuracy_dark bricks": 0.8660681813831518,
-      "eval_accuracy_grass": 0.8697148447997214,
-      "eval_accuracy_red bricks": 0.8806019332161688,
-      "eval_accuracy_road": 0.9743426815184473,
-      "eval_accuracy_rough concrete": 0.6254416961130742,
-      "eval_accuracy_rough red sidewalk": 0.7290173636438836,
-      "eval_accuracy_speedway bricks": 0.9838051358822872,
-      "eval_accuracy_steel": 0.8986790746025479,
-      "eval_accuracy_tiles": 0.21272986155290982,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.979491261992737,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8723648091991095,
-      "eval_iou_concrete tiles": 0.8248605292048484,
-      "eval_iou_dark bricks": 0.8305294749304771,
-      "eval_iou_grass": 0.8163165799109914,
-      "eval_iou_red bricks": 0.8460767241834204,
-      "eval_iou_road": 0.9531798785620255,
-      "eval_iou_rough concrete": 0.5899867344185629,
-      "eval_iou_rough red sidewalk": 0.6061121392487371,
-      "eval_iou_speedway bricks": 0.9723500352882056,
-      "eval_iou_steel": 0.8223677591079971,
-      "eval_iou_tiles": 0.21011999162153266,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07961232215166092,
-      "eval_mean_accuracy": 0.7578207251375216,
-      "eval_mean_iou": 0.7172119936668189,
-      "eval_overall_accuracy": 0.9757948729951503,
-      "eval_runtime": 5.7515,
-      "eval_samples_per_second": 20.516,
-      "eval_steps_per_second": 1.391,
-      "step": 580
-    },
-    {
-      "epoch": 34.18,
-      "learning_rate": 4.363777089783282e-05,
-      "loss": 0.0518,
-      "step": 581
-    },
-    {
-      "epoch": 34.24,
-      "learning_rate": 4.362229102167183e-05,
-      "loss": 0.0598,
-      "step": 582
-    },
-    {
-      "epoch": 34.29,
-      "learning_rate": 4.360681114551084e-05,
-      "loss": 0.0466,
-      "step": 583
-    },
-    {
-      "epoch": 34.35,
-      "learning_rate": 4.359133126934985e-05,
-      "loss": 0.0496,
-      "step": 584
-    },
-    {
-      "epoch": 34.41,
-      "learning_rate": 4.3575851393188855e-05,
-      "loss": 0.0381,
-      "step": 585
-    },
-    {
-      "epoch": 34.47,
-      "learning_rate": 4.3560371517027865e-05,
-      "loss": 0.0472,
-      "step": 586
-    },
-    {
-      "epoch": 34.53,
-      "learning_rate": 4.3544891640866875e-05,
-      "loss": 0.0462,
-      "step": 587
-    },
-    {
-      "epoch": 34.59,
-      "learning_rate": 4.3529411764705885e-05,
-      "loss": 0.0465,
-      "step": 588
-    },
-    {
-      "epoch": 34.65,
-      "learning_rate": 4.3513931888544895e-05,
-      "loss": 0.0469,
-      "step": 589
-    },
-    {
-      "epoch": 34.71,
-      "learning_rate": 4.3498452012383905e-05,
-      "loss": 0.0488,
-      "step": 590
-    },
-    {
-      "epoch": 34.76,
-      "learning_rate": 4.348297213622291e-05,
-      "loss": 0.0476,
-      "step": 591
-    },
-    {
-      "epoch": 34.82,
-      "learning_rate": 4.346749226006192e-05,
-      "loss": 0.0514,
-      "step": 592
-    },
-    {
-      "epoch": 34.88,
-      "learning_rate": 4.345201238390093e-05,
-      "loss": 0.0475,
-      "step": 593
-    },
-    {
-      "epoch": 34.94,
-      "learning_rate": 4.343653250773994e-05,
-      "loss": 0.0436,
-      "step": 594
-    },
-    {
-      "epoch": 35.0,
-      "learning_rate": 4.342105263157895e-05,
-      "loss": 0.0492,
-      "step": 595
-    },
-    {
-      "epoch": 35.06,
-      "learning_rate": 4.340557275541796e-05,
-      "loss": 0.0441,
-      "step": 596
-    },
-    {
-      "epoch": 35.12,
-      "learning_rate": 4.339009287925697e-05,
-      "loss": 0.051,
-      "step": 597
-    },
-    {
-      "epoch": 35.18,
-      "learning_rate": 4.337461300309598e-05,
-      "loss": 0.0404,
-      "step": 598
-    },
-    {
-      "epoch": 35.24,
-      "learning_rate": 4.335913312693499e-05,
-      "loss": 0.0456,
-      "step": 599
-    },
-    {
-      "epoch": 35.29,
-      "learning_rate": 4.3343653250774e-05,
-      "loss": 0.0471,
-      "step": 600
-    },
-    {
-      "epoch": 35.29,
-      "eval_accuracy_NAT": 0.9921806923768611,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9352401511063141,
-      "eval_accuracy_concrete tiles": 0.9168041480084845,
-      "eval_accuracy_dark bricks": 0.8880184905154758,
-      "eval_accuracy_grass": 0.8596167020555852,
-      "eval_accuracy_red bricks": 0.9219573813708261,
-      "eval_accuracy_road": 0.9796841707545048,
-      "eval_accuracy_rough concrete": 0.6205256854110679,
-      "eval_accuracy_rough red sidewalk": 0.8089923972958136,
-      "eval_accuracy_speedway bricks": 0.9875815098367027,
-      "eval_accuracy_steel": 0.8911223617726289,
-      "eval_accuracy_tiles": 0.456420976097428,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9789283325219831,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8784533738659873,
-      "eval_iou_concrete tiles": 0.8173814376667857,
-      "eval_iou_dark bricks": 0.8203265905152598,
-      "eval_iou_grass": 0.8033537812057023,
-      "eval_iou_red bricks": 0.8567418597529856,
-      "eval_iou_road": 0.9529836330101246,
-      "eval_iou_rough concrete": 0.6047773086550244,
-      "eval_iou_rough red sidewalk": 0.5827776182711528,
-      "eval_iou_speedway bricks": 0.9728446971237983,
-      "eval_iou_steel": 0.8207736729324953,
-      "eval_iou_tiles": 0.4101377470463331,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08150813728570938,
-      "eval_mean_accuracy": 0.7890880512770533,
-      "eval_mean_iou": 0.7307292348128948,
-      "eval_overall_accuracy": 0.9756607443599378,
-      "eval_runtime": 8.5574,
-      "eval_samples_per_second": 13.789,
-      "eval_steps_per_second": 0.935,
-      "step": 600
-    },
-    {
-      "epoch": 35.35,
-      "learning_rate": 4.332817337461301e-05,
-      "loss": 0.0514,
-      "step": 601
-    },
-    {
-      "epoch": 35.41,
-      "learning_rate": 4.331269349845202e-05,
-      "loss": 0.052,
-      "step": 602
-    },
-    {
-      "epoch": 35.47,
-      "learning_rate": 4.329721362229102e-05,
-      "loss": 0.0427,
-      "step": 603
-    },
-    {
-      "epoch": 35.53,
-      "learning_rate": 4.328173374613003e-05,
-      "loss": 0.0551,
-      "step": 604
-    },
-    {
-      "epoch": 35.59,
-      "learning_rate": 4.326625386996904e-05,
-      "loss": 0.0451,
-      "step": 605
-    },
-    {
-      "epoch": 35.65,
-      "learning_rate": 4.325077399380805e-05,
-      "loss": 0.0483,
-      "step": 606
-    },
-    {
-      "epoch": 35.71,
-      "learning_rate": 4.323529411764706e-05,
-      "loss": 0.0421,
-      "step": 607
-    },
-    {
-      "epoch": 35.76,
-      "learning_rate": 4.321981424148607e-05,
-      "loss": 0.0522,
-      "step": 608
-    },
-    {
-      "epoch": 35.82,
-      "learning_rate": 4.3204334365325075e-05,
-      "loss": 0.0513,
-      "step": 609
-    },
-    {
-      "epoch": 35.88,
-      "learning_rate": 4.3188854489164085e-05,
-      "loss": 0.0469,
-      "step": 610
-    },
-    {
-      "epoch": 35.94,
-      "learning_rate": 4.3173374613003095e-05,
-      "loss": 0.0493,
-      "step": 611
-    },
-    {
-      "epoch": 36.0,
-      "learning_rate": 4.3157894736842105e-05,
-      "loss": 0.0591,
-      "step": 612
-    },
-    {
-      "epoch": 36.06,
-      "learning_rate": 4.314241486068112e-05,
-      "loss": 0.0483,
-      "step": 613
-    },
-    {
-      "epoch": 36.12,
-      "learning_rate": 4.3126934984520125e-05,
-      "loss": 0.045,
-      "step": 614
-    },
-    {
-      "epoch": 36.18,
-      "learning_rate": 4.3111455108359135e-05,
-      "loss": 0.0452,
-      "step": 615
-    },
-    {
-      "epoch": 36.24,
-      "learning_rate": 4.3095975232198145e-05,
-      "loss": 0.0451,
-      "step": 616
-    },
-    {
-      "epoch": 36.29,
-      "learning_rate": 4.3080495356037155e-05,
-      "loss": 0.0451,
-      "step": 617
-    },
-    {
-      "epoch": 36.35,
-      "learning_rate": 4.3065015479876165e-05,
-      "loss": 0.0414,
-      "step": 618
-    },
-    {
-      "epoch": 36.41,
-      "learning_rate": 4.3049535603715175e-05,
-      "loss": 0.05,
-      "step": 619
-    },
-    {
-      "epoch": 36.47,
-      "learning_rate": 4.303405572755418e-05,
-      "loss": 0.0522,
-      "step": 620
-    },
-    {
-      "epoch": 36.47,
-      "eval_accuracy_NAT": 0.9943427156506753,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9352936590079024,
-      "eval_accuracy_concrete tiles": 0.8649776101814753,
-      "eval_accuracy_dark bricks": 0.8185402829087151,
-      "eval_accuracy_grass": 0.8543376973358011,
-      "eval_accuracy_red bricks": 0.8782952548330404,
-      "eval_accuracy_road": 0.9782813284088668,
-      "eval_accuracy_rough concrete": 0.6238447374850286,
-      "eval_accuracy_rough red sidewalk": 0.7478699644304739,
-      "eval_accuracy_speedway bricks": 0.9850266267180213,
-      "eval_accuracy_steel": 0.9007806604127204,
-      "eval_accuracy_tiles": 0.10754642673210336,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9785511215548038,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8728486742140519,
-      "eval_iou_concrete tiles": 0.8187070580886946,
-      "eval_iou_dark bricks": 0.7977836751255778,
-      "eval_iou_grass": 0.8036221777391295,
-      "eval_iou_red bricks": 0.8518617162946786,
-      "eval_iou_road": 0.9475414435277907,
-      "eval_iou_rough concrete": 0.6092469739292365,
-      "eval_iou_rough red sidewalk": 0.603051305840049,
-      "eval_iou_speedway bricks": 0.9727465494249605,
-      "eval_iou_steel": 0.8254196148473816,
-      "eval_iou_tiles": 0.10750734381151389,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08459129929542542,
-      "eval_mean_accuracy": 0.7453182280080634,
-      "eval_mean_iou": 0.7068375118767591,
-      "eval_overall_accuracy": 0.9751562668105304,
-      "eval_runtime": 5.4145,
-      "eval_samples_per_second": 21.793,
-      "eval_steps_per_second": 1.478,
-      "step": 620
-    },
-    {
-      "epoch": 36.53,
-      "learning_rate": 4.301857585139319e-05,
-      "loss": 0.0406,
-      "step": 621
-    },
-    {
-      "epoch": 36.59,
-      "learning_rate": 4.30030959752322e-05,
-      "loss": 0.0507,
-      "step": 622
-    },
-    {
-      "epoch": 36.65,
-      "learning_rate": 4.298761609907121e-05,
-      "loss": 0.0402,
-      "step": 623
-    },
-    {
-      "epoch": 36.71,
-      "learning_rate": 4.297213622291022e-05,
-      "loss": 0.0433,
-      "step": 624
-    },
-    {
-      "epoch": 36.76,
-      "learning_rate": 4.295665634674923e-05,
-      "loss": 0.0478,
-      "step": 625
-    },
-    {
-      "epoch": 36.82,
-      "learning_rate": 4.294117647058823e-05,
-      "loss": 0.0557,
-      "step": 626
-    },
-    {
-      "epoch": 36.88,
-      "learning_rate": 4.292569659442724e-05,
-      "loss": 0.0417,
-      "step": 627
-    },
-    {
-      "epoch": 36.94,
-      "learning_rate": 4.291021671826626e-05,
-      "loss": 0.041,
-      "step": 628
-    },
-    {
-      "epoch": 37.0,
-      "learning_rate": 4.289473684210527e-05,
-      "loss": 0.0399,
-      "step": 629
-    },
-    {
-      "epoch": 37.06,
-      "learning_rate": 4.287925696594428e-05,
-      "loss": 0.0414,
-      "step": 630
-    },
-    {
-      "epoch": 37.12,
-      "learning_rate": 4.286377708978329e-05,
-      "loss": 0.0394,
-      "step": 631
-    },
-    {
-      "epoch": 37.18,
-      "learning_rate": 4.284829721362229e-05,
-      "loss": 0.0432,
-      "step": 632
-    },
-    {
-      "epoch": 37.24,
-      "learning_rate": 4.28328173374613e-05,
-      "loss": 0.0461,
-      "step": 633
-    },
-    {
-      "epoch": 37.29,
-      "learning_rate": 4.281733746130031e-05,
-      "loss": 0.0482,
-      "step": 634
-    },
-    {
-      "epoch": 37.35,
-      "learning_rate": 4.280185758513932e-05,
-      "loss": 0.0425,
-      "step": 635
-    },
-    {
-      "epoch": 37.41,
-      "learning_rate": 4.278637770897833e-05,
-      "loss": 0.0429,
-      "step": 636
-    },
-    {
-      "epoch": 37.47,
-      "learning_rate": 4.277089783281734e-05,
-      "loss": 0.0533,
-      "step": 637
-    },
-    {
-      "epoch": 37.53,
-      "learning_rate": 4.2755417956656345e-05,
-      "loss": 0.0495,
-      "step": 638
-    },
-    {
-      "epoch": 37.59,
-      "learning_rate": 4.2739938080495355e-05,
-      "loss": 0.0452,
-      "step": 639
-    },
-    {
-      "epoch": 37.65,
-      "learning_rate": 4.2724458204334365e-05,
-      "loss": 0.043,
-      "step": 640
-    },
-    {
-      "epoch": 37.65,
-      "eval_accuracy_NAT": 0.993806239736212,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9391362907405341,
-      "eval_accuracy_concrete tiles": 0.91006363422107,
-      "eval_accuracy_dark bricks": 0.8166548628524396,
-      "eval_accuracy_grass": 0.8672234270855806,
-      "eval_accuracy_red bricks": 0.9003734622144113,
-      "eval_accuracy_road": 0.980038488649803,
-      "eval_accuracy_rough concrete": 0.6178918954349627,
-      "eval_accuracy_rough red sidewalk": 0.7609171372923544,
-      "eval_accuracy_speedway bricks": 0.9853395394938421,
-      "eval_accuracy_steel": 0.9130003448499903,
-      "eval_accuracy_tiles": 0.17068072343906207,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9796457416816873,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8762803509415095,
-      "eval_iou_concrete tiles": 0.8324853397723353,
-      "eval_iou_dark bricks": 0.7964735586907923,
-      "eval_iou_grass": 0.8144555358175916,
-      "eval_iou_red bricks": 0.8581898131183584,
-      "eval_iou_road": 0.9519905976698708,
-      "eval_iou_rough concrete": 0.6100795443848254,
-      "eval_iou_rough red sidewalk": 0.6210778296096243,
-      "eval_iou_speedway bricks": 0.9727566141361427,
-      "eval_iou_steel": 0.8250808462205521,
-      "eval_iou_tiles": 0.16930612735523035,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08132755756378174,
-      "eval_mean_accuracy": 0.7580866189238662,
-      "eval_mean_iou": 0.7159862999537322,
-      "eval_overall_accuracy": 0.9760765140339479,
-      "eval_runtime": 8.4358,
-      "eval_samples_per_second": 13.988,
-      "eval_steps_per_second": 0.948,
-      "step": 640
-    },
-    {
-      "epoch": 37.71,
-      "learning_rate": 4.2708978328173375e-05,
-      "loss": 0.0445,
-      "step": 641
-    },
-    {
-      "epoch": 37.76,
-      "learning_rate": 4.2693498452012385e-05,
-      "loss": 0.0405,
-      "step": 642
-    },
-    {
-      "epoch": 37.82,
-      "learning_rate": 4.2678018575851395e-05,
-      "loss": 0.0485,
-      "step": 643
-    },
-    {
-      "epoch": 37.88,
-      "learning_rate": 4.2662538699690405e-05,
-      "loss": 0.0433,
-      "step": 644
-    },
-    {
-      "epoch": 37.94,
-      "learning_rate": 4.2647058823529415e-05,
-      "loss": 0.0396,
-      "step": 645
-    },
-    {
-      "epoch": 38.0,
-      "learning_rate": 4.2631578947368425e-05,
-      "loss": 0.0477,
-      "step": 646
-    },
-    {
-      "epoch": 38.06,
-      "learning_rate": 4.2616099071207435e-05,
-      "loss": 0.0409,
-      "step": 647
-    },
-    {
-      "epoch": 38.12,
-      "learning_rate": 4.2600619195046445e-05,
-      "loss": 0.0468,
-      "step": 648
-    },
-    {
-      "epoch": 38.18,
-      "learning_rate": 4.2585139318885455e-05,
-      "loss": 0.0399,
-      "step": 649
-    },
-    {
-      "epoch": 38.24,
-      "learning_rate": 4.256965944272446e-05,
-      "loss": 0.0434,
-      "step": 650
-    },
-    {
-      "epoch": 38.29,
-      "learning_rate": 4.255417956656347e-05,
-      "loss": 0.0477,
-      "step": 651
-    },
-    {
-      "epoch": 38.35,
-      "learning_rate": 4.253869969040248e-05,
-      "loss": 0.0409,
-      "step": 652
-    },
-    {
-      "epoch": 38.41,
-      "learning_rate": 4.252321981424149e-05,
-      "loss": 0.0421,
-      "step": 653
-    },
-    {
-      "epoch": 38.47,
-      "learning_rate": 4.25077399380805e-05,
-      "loss": 0.0427,
-      "step": 654
-    },
-    {
-      "epoch": 38.53,
-      "learning_rate": 4.249226006191951e-05,
-      "loss": 0.0521,
-      "step": 655
-    },
-    {
-      "epoch": 38.59,
-      "learning_rate": 4.247678018575851e-05,
-      "loss": 0.0439,
-      "step": 656
-    },
-    {
-      "epoch": 38.65,
-      "learning_rate": 4.246130030959752e-05,
-      "loss": 0.0389,
-      "step": 657
-    },
-    {
-      "epoch": 38.71,
-      "learning_rate": 4.244582043343653e-05,
-      "loss": 0.0415,
-      "step": 658
-    },
-    {
-      "epoch": 38.76,
-      "learning_rate": 4.243034055727554e-05,
-      "loss": 0.0498,
-      "step": 659
-    },
-    {
-      "epoch": 38.82,
-      "learning_rate": 4.241486068111455e-05,
-      "loss": 0.0429,
-      "step": 660
-    },
-    {
-      "epoch": 38.82,
-      "eval_accuracy_NAT": 0.993064360019708,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9416037694023474,
-      "eval_accuracy_concrete tiles": 0.8665095451331605,
-      "eval_accuracy_dark bricks": 0.8162672511149819,
-      "eval_accuracy_grass": 0.8894030059763746,
-      "eval_accuracy_red bricks": 0.9094903339191565,
-      "eval_accuracy_road": 0.9792309858748472,
-      "eval_accuracy_rough concrete": 0.6195305656689648,
-      "eval_accuracy_rough red sidewalk": 0.7703095977560367,
-      "eval_accuracy_speedway bricks": 0.9875611173830758,
-      "eval_accuracy_steel": 0.8956428953405269,
-      "eval_accuracy_tiles": 0.19309885182828926,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804051300764475,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8790174711985862,
-      "eval_iou_concrete tiles": 0.8424067454862065,
-      "eval_iou_dark bricks": 0.7969845067000579,
-      "eval_iou_grass": 0.8264285946653607,
-      "eval_iou_red bricks": 0.8588766142834915,
-      "eval_iou_road": 0.954441135075388,
-      "eval_iou_rough concrete": 0.6086180251481021,
-      "eval_iou_rough red sidewalk": 0.6179531264138992,
-      "eval_iou_speedway bricks": 0.9729326781278663,
-      "eval_iou_steel": 0.8281644253431305,
-      "eval_iou_tiles": 0.19039369137941334,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07921610027551651,
-      "eval_mean_accuracy": 0.7585932522628822,
-      "eval_mean_iou": 0.719740164915227,
-      "eval_overall_accuracy": 0.9767600237312963,
-      "eval_runtime": 5.7289,
-      "eval_samples_per_second": 20.597,
-      "eval_steps_per_second": 1.396,
-      "step": 660
-    },
-    {
-      "epoch": 38.88,
-      "learning_rate": 4.239938080495356e-05,
-      "loss": 0.0436,
-      "step": 661
-    },
-    {
-      "epoch": 38.94,
-      "learning_rate": 4.238390092879257e-05,
-      "loss": 0.0458,
-      "step": 662
-    },
-    {
-      "epoch": 39.0,
-      "learning_rate": 4.236842105263158e-05,
-      "loss": 0.0416,
-      "step": 663
-    },
-    {
-      "epoch": 39.06,
-      "learning_rate": 4.235294117647059e-05,
-      "loss": 0.0426,
-      "step": 664
-    },
-    {
-      "epoch": 39.12,
-      "learning_rate": 4.23374613003096e-05,
-      "loss": 0.0467,
-      "step": 665
-    },
-    {
-      "epoch": 39.18,
-      "learning_rate": 4.232198142414861e-05,
-      "loss": 0.0417,
-      "step": 666
-    },
-    {
-      "epoch": 39.24,
-      "learning_rate": 4.230650154798762e-05,
-      "loss": 0.0422,
-      "step": 667
-    },
-    {
-      "epoch": 39.29,
-      "learning_rate": 4.2291021671826625e-05,
-      "loss": 0.0403,
-      "step": 668
-    },
-    {
-      "epoch": 39.35,
-      "learning_rate": 4.2275541795665635e-05,
-      "loss": 0.0423,
-      "step": 669
-    },
-    {
-      "epoch": 39.41,
-      "learning_rate": 4.2260061919504645e-05,
-      "loss": 0.0476,
-      "step": 670
-    },
-    {
-      "epoch": 39.47,
-      "learning_rate": 4.2244582043343655e-05,
-      "loss": 0.0569,
-      "step": 671
-    },
-    {
-      "epoch": 39.53,
-      "learning_rate": 4.2229102167182665e-05,
-      "loss": 0.0468,
-      "step": 672
-    },
-    {
-      "epoch": 39.59,
-      "learning_rate": 4.2213622291021675e-05,
-      "loss": 0.0447,
-      "step": 673
-    },
-    {
-      "epoch": 39.65,
-      "learning_rate": 4.219814241486068e-05,
-      "loss": 0.0396,
-      "step": 674
-    },
-    {
-      "epoch": 39.71,
-      "learning_rate": 4.218266253869969e-05,
-      "loss": 0.0423,
-      "step": 675
-    },
-    {
-      "epoch": 39.76,
-      "learning_rate": 4.2167182662538705e-05,
-      "loss": 0.0372,
-      "step": 676
-    },
-    {
-      "epoch": 39.82,
-      "learning_rate": 4.2151702786377715e-05,
-      "loss": 0.042,
-      "step": 677
-    },
-    {
-      "epoch": 39.88,
-      "learning_rate": 4.2136222910216725e-05,
-      "loss": 0.039,
-      "step": 678
-    },
-    {
-      "epoch": 39.94,
-      "learning_rate": 4.2120743034055735e-05,
-      "loss": 0.0381,
-      "step": 679
-    },
-    {
-      "epoch": 40.0,
-      "learning_rate": 4.210526315789474e-05,
-      "loss": 0.0506,
-      "step": 680
-    },
-    {
-      "epoch": 40.0,
-      "eval_accuracy_NAT": 0.9915816788218814,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9473501358336301,
-      "eval_accuracy_concrete tiles": 0.8905962762196559,
-      "eval_accuracy_dark bricks": 0.8488218517313324,
-      "eval_accuracy_grass": 0.8902943525341872,
-      "eval_accuracy_red bricks": 0.8935632688927944,
-      "eval_accuracy_road": 0.9795794251926971,
-      "eval_accuracy_rough concrete": 0.6164141128239353,
-      "eval_accuracy_rough red sidewalk": 0.7506824385805277,
-      "eval_accuracy_speedway bricks": 0.9894197438707825,
-      "eval_accuracy_steel": 0.8787777316867162,
-      "eval_accuracy_tiles": 0.21100305977157746,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807303099744141,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8769597034864961,
-      "eval_iou_concrete tiles": 0.8471126255380201,
-      "eval_iou_dark bricks": 0.819791839979295,
-      "eval_iou_grass": 0.8269456409679492,
-      "eval_iou_red bricks": 0.8523233275708523,
-      "eval_iou_road": 0.9515856732881748,
-      "eval_iou_rough concrete": 0.6061064140197923,
-      "eval_iou_rough red sidewalk": 0.614039404321804,
-      "eval_iou_speedway bricks": 0.9726276228723906,
-      "eval_iou_steel": 0.8268979803941394,
-      "eval_iou_tiles": 0.2069160155669766,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0794433206319809,
-      "eval_mean_accuracy": 0.7606218519969014,
-      "eval_mean_iou": 0.7216951198446389,
-      "eval_overall_accuracy": 0.9767023830090539,
-      "eval_runtime": 5.9542,
-      "eval_samples_per_second": 19.818,
-      "eval_steps_per_second": 1.344,
-      "step": 680
-    },
-    {
-      "epoch": 40.06,
-      "learning_rate": 4.208978328173375e-05,
-      "loss": 0.0406,
-      "step": 681
-    },
-    {
-      "epoch": 40.12,
-      "learning_rate": 4.207430340557276e-05,
-      "loss": 0.0418,
-      "step": 682
-    },
-    {
-      "epoch": 40.18,
-      "learning_rate": 4.205882352941177e-05,
-      "loss": 0.0452,
-      "step": 683
-    },
-    {
-      "epoch": 40.24,
-      "learning_rate": 4.204334365325078e-05,
-      "loss": 0.0494,
-      "step": 684
-    },
-    {
-      "epoch": 40.29,
-      "learning_rate": 4.202786377708979e-05,
-      "loss": 0.0416,
-      "step": 685
-    },
-    {
-      "epoch": 40.35,
-      "learning_rate": 4.201238390092879e-05,
-      "loss": 0.0399,
-      "step": 686
-    },
-    {
-      "epoch": 40.41,
-      "learning_rate": 4.19969040247678e-05,
-      "loss": 0.0435,
-      "step": 687
-    },
-    {
-      "epoch": 40.47,
-      "learning_rate": 4.198142414860681e-05,
-      "loss": 0.0427,
-      "step": 688
-    },
-    {
-      "epoch": 40.53,
-      "learning_rate": 4.196594427244582e-05,
-      "loss": 0.0414,
-      "step": 689
-    },
-    {
-      "epoch": 40.59,
-      "learning_rate": 4.195046439628483e-05,
-      "loss": 0.0405,
-      "step": 690
-    },
-    {
-      "epoch": 40.65,
-      "learning_rate": 4.193498452012384e-05,
-      "loss": 0.0402,
-      "step": 691
-    },
-    {
-      "epoch": 40.71,
-      "learning_rate": 4.191950464396285e-05,
-      "loss": 0.0497,
-      "step": 692
-    },
-    {
-      "epoch": 40.76,
-      "learning_rate": 4.190402476780186e-05,
-      "loss": 0.039,
-      "step": 693
-    },
-    {
-      "epoch": 40.82,
-      "learning_rate": 4.188854489164087e-05,
-      "loss": 0.0387,
-      "step": 694
-    },
-    {
-      "epoch": 40.88,
-      "learning_rate": 4.187306501547988e-05,
-      "loss": 0.037,
-      "step": 695
-    },
-    {
-      "epoch": 40.94,
-      "learning_rate": 4.185758513931889e-05,
-      "loss": 0.034,
-      "step": 696
-    },
-    {
-      "epoch": 41.0,
-      "learning_rate": 4.18421052631579e-05,
-      "loss": 0.0512,
-      "step": 697
-    },
-    {
-      "epoch": 41.06,
-      "learning_rate": 4.1826625386996904e-05,
-      "loss": 0.0444,
-      "step": 698
-    },
-    {
-      "epoch": 41.12,
-      "learning_rate": 4.1811145510835914e-05,
-      "loss": 0.0391,
-      "step": 699
-    },
-    {
-      "epoch": 41.18,
-      "learning_rate": 4.1795665634674924e-05,
-      "loss": 0.0426,
-      "step": 700
-    },
-    {
-      "epoch": 41.18,
-      "eval_accuracy_NAT": 0.992102682061493,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.951264621034109,
-      "eval_accuracy_concrete tiles": 0.9061984444968183,
-      "eval_accuracy_dark bricks": 0.8437015485328178,
-      "eval_accuracy_grass": 0.8701397010677234,
-      "eval_accuracy_red bricks": 0.9255821616871704,
-      "eval_accuracy_road": 0.9787628373435029,
-      "eval_accuracy_rough concrete": 0.6143285325261144,
-      "eval_accuracy_rough red sidewalk": 0.7620000150399687,
-      "eval_accuracy_speedway bricks": 0.9887498774739131,
-      "eval_accuracy_steel": 0.8858521543128739,
-      "eval_accuracy_tiles": 0.214062831349026,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9804335180753472,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8763075924244403,
-      "eval_iou_concrete tiles": 0.8416881922858018,
-      "eval_iou_dark bricks": 0.8206457738907016,
-      "eval_iou_grass": 0.8174202747051524,
-      "eval_iou_red bricks": 0.85792099368764,
-      "eval_iou_road": 0.9523834781568671,
-      "eval_iou_rough concrete": 0.6066004930658931,
-      "eval_iou_rough red sidewalk": 0.6145234456492734,
-      "eval_iou_speedway bricks": 0.9741434647027937,
-      "eval_iou_steel": 0.831835460859771,
-      "eval_iou_tiles": 0.21037275217339527,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08128701895475388,
-      "eval_mean_accuracy": 0.7640573389942715,
-      "eval_mean_iou": 0.7218673415136212,
-      "eval_overall_accuracy": 0.9766193325236693,
-      "eval_runtime": 5.6917,
-      "eval_samples_per_second": 20.732,
-      "eval_steps_per_second": 1.406,
-      "step": 700
-    },
-    {
-      "epoch": 41.24,
-      "learning_rate": 4.1780185758513934e-05,
-      "loss": 0.0396,
-      "step": 701
-    },
-    {
-      "epoch": 41.29,
-      "learning_rate": 4.1764705882352944e-05,
-      "loss": 0.0365,
-      "step": 702
-    },
-    {
-      "epoch": 41.35,
-      "learning_rate": 4.1749226006191954e-05,
-      "loss": 0.0408,
-      "step": 703
-    },
-    {
-      "epoch": 41.41,
-      "learning_rate": 4.173374613003096e-05,
-      "loss": 0.0419,
-      "step": 704
-    },
-    {
-      "epoch": 41.47,
-      "learning_rate": 4.171826625386997e-05,
-      "loss": 0.0479,
-      "step": 705
-    },
-    {
-      "epoch": 41.53,
-      "learning_rate": 4.170278637770898e-05,
-      "loss": 0.036,
-      "step": 706
-    },
-    {
-      "epoch": 41.59,
-      "learning_rate": 4.168730650154799e-05,
-      "loss": 0.0452,
-      "step": 707
-    },
-    {
-      "epoch": 41.65,
-      "learning_rate": 4.1671826625387e-05,
-      "loss": 0.0507,
-      "step": 708
-    },
-    {
-      "epoch": 41.71,
-      "learning_rate": 4.165634674922601e-05,
-      "loss": 0.0429,
-      "step": 709
-    },
-    {
-      "epoch": 41.76,
-      "learning_rate": 4.164086687306502e-05,
-      "loss": 0.0424,
-      "step": 710
-    },
-    {
-      "epoch": 41.82,
-      "learning_rate": 4.162538699690403e-05,
-      "loss": 0.0419,
-      "step": 711
-    },
-    {
-      "epoch": 41.88,
-      "learning_rate": 4.160990712074304e-05,
-      "loss": 0.0385,
-      "step": 712
-    },
-    {
-      "epoch": 41.94,
-      "learning_rate": 4.159442724458205e-05,
-      "loss": 0.0481,
-      "step": 713
-    },
-    {
-      "epoch": 42.0,
-      "learning_rate": 4.157894736842106e-05,
-      "loss": 0.0431,
-      "step": 714
-    },
-    {
-      "epoch": 42.06,
-      "learning_rate": 4.156346749226007e-05,
-      "loss": 0.0409,
-      "step": 715
-    },
-    {
-      "epoch": 42.12,
-      "learning_rate": 4.154798761609907e-05,
-      "loss": 0.0368,
-      "step": 716
-    },
-    {
-      "epoch": 42.18,
-      "learning_rate": 4.153250773993808e-05,
-      "loss": 0.0384,
-      "step": 717
-    },
-    {
-      "epoch": 42.24,
-      "learning_rate": 4.151702786377709e-05,
-      "loss": 0.0476,
-      "step": 718
-    },
-    {
-      "epoch": 42.29,
-      "learning_rate": 4.15015479876161e-05,
-      "loss": 0.0493,
-      "step": 719
-    },
-    {
-      "epoch": 42.35,
-      "learning_rate": 4.148606811145511e-05,
-      "loss": 0.043,
-      "step": 720
-    },
-    {
-      "epoch": 42.35,
-      "eval_accuracy_NAT": 0.9923592481228869,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9485104928995015,
-      "eval_accuracy_concrete tiles": 0.8466650954513316,
-      "eval_accuracy_dark bricks": 0.8488074957410562,
-      "eval_accuracy_grass": 0.8917695216250894,
-      "eval_accuracy_red bricks": 0.914213532513181,
-      "eval_accuracy_road": 0.9799428281622338,
-      "eval_accuracy_rough concrete": 0.6664680399716361,
-      "eval_accuracy_rough red sidewalk": 0.7487197226629768,
-      "eval_accuracy_speedway bricks": 0.9883259200766619,
-      "eval_accuracy_steel": 0.8803970272931275,
-      "eval_accuracy_tiles": 0.1910691023660214,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9809562210409813,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8793828877096302,
-      "eval_iou_concrete tiles": 0.8212138530117727,
-      "eval_iou_dark bricks": 0.8281021676307324,
-      "eval_iou_grass": 0.8290248008909109,
-      "eval_iou_red bricks": 0.8632921896068873,
-      "eval_iou_road": 0.9535515005278239,
-      "eval_iou_rough concrete": 0.6520834183569358,
-      "eval_iou_rough red sidewalk": 0.6152192047455742,
-      "eval_iou_speedway bricks": 0.9744320413766746,
-      "eval_iou_steel": 0.8306992207302265,
-      "eval_iou_tiles": 0.1892118921189212,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07583378255367279,
-      "eval_mean_accuracy": 0.7613267712989003,
-      "eval_mean_iou": 0.724397645980544,
-      "eval_overall_accuracy": 0.9773474224543167,
-      "eval_runtime": 8.5023,
-      "eval_samples_per_second": 13.879,
-      "eval_steps_per_second": 0.941,
-      "step": 720
-    },
-    {
-      "epoch": 42.41,
-      "learning_rate": 4.147058823529412e-05,
-      "loss": 0.0457,
-      "step": 721
-    },
-    {
-      "epoch": 42.47,
-      "learning_rate": 4.1455108359133124e-05,
-      "loss": 0.0415,
-      "step": 722
-    },
-    {
-      "epoch": 42.53,
-      "learning_rate": 4.1439628482972134e-05,
-      "loss": 0.0349,
-      "step": 723
-    },
-    {
-      "epoch": 42.59,
-      "learning_rate": 4.1424148606811144e-05,
-      "loss": 0.0385,
-      "step": 724
-    },
-    {
-      "epoch": 42.65,
-      "learning_rate": 4.140866873065016e-05,
-      "loss": 0.0425,
-      "step": 725
-    },
-    {
-      "epoch": 42.71,
-      "learning_rate": 4.139318885448917e-05,
-      "loss": 0.0403,
-      "step": 726
-    },
-    {
-      "epoch": 42.76,
-      "learning_rate": 4.137770897832818e-05,
-      "loss": 0.0443,
-      "step": 727
-    },
-    {
-      "epoch": 42.82,
-      "learning_rate": 4.1362229102167184e-05,
-      "loss": 0.0391,
-      "step": 728
-    },
-    {
-      "epoch": 42.88,
-      "learning_rate": 4.1346749226006194e-05,
-      "loss": 0.0367,
-      "step": 729
-    },
-    {
-      "epoch": 42.94,
-      "learning_rate": 4.1331269349845204e-05,
-      "loss": 0.039,
-      "step": 730
-    },
-    {
-      "epoch": 43.0,
-      "learning_rate": 4.1315789473684214e-05,
-      "loss": 0.043,
-      "step": 731
-    },
-    {
-      "epoch": 43.06,
-      "learning_rate": 4.1300309597523224e-05,
-      "loss": 0.0381,
-      "step": 732
-    },
-    {
-      "epoch": 43.12,
-      "learning_rate": 4.1284829721362234e-05,
-      "loss": 0.0398,
-      "step": 733
-    },
-    {
-      "epoch": 43.18,
-      "learning_rate": 4.126934984520124e-05,
-      "loss": 0.0358,
-      "step": 734
-    },
-    {
-      "epoch": 43.24,
-      "learning_rate": 4.125386996904025e-05,
-      "loss": 0.0442,
-      "step": 735
-    },
-    {
-      "epoch": 43.29,
-      "learning_rate": 4.123839009287926e-05,
-      "loss": 0.0393,
-      "step": 736
-    },
-    {
-      "epoch": 43.35,
-      "learning_rate": 4.122291021671827e-05,
-      "loss": 0.0439,
-      "step": 737
-    },
-    {
-      "epoch": 43.41,
-      "learning_rate": 4.120743034055728e-05,
-      "loss": 0.0387,
-      "step": 738
-    },
-    {
-      "epoch": 43.47,
-      "learning_rate": 4.119195046439629e-05,
-      "loss": 0.0426,
-      "step": 739
-    },
-    {
-      "epoch": 43.53,
-      "learning_rate": 4.11764705882353e-05,
-      "loss": 0.0369,
-      "step": 740
-    },
-    {
-      "epoch": 43.53,
-      "eval_accuracy_NAT": 0.9927773877262009,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9525602766511392,
-      "eval_accuracy_concrete tiles": 0.8837379212821117,
-      "eval_accuracy_dark bricks": 0.8390884903240625,
-      "eval_accuracy_grass": 0.876139731347012,
-      "eval_accuracy_red bricks": 0.9155316344463972,
-      "eval_accuracy_road": 0.9744639939803367,
-      "eval_accuracy_rough concrete": 0.617081498519238,
-      "eval_accuracy_rough red sidewalk": 0.7683694417915611,
-      "eval_accuracy_speedway bricks": 0.9881799169632998,
-      "eval_accuracy_steel": 0.8889408107473249,
-      "eval_accuracy_tiles": 0.224302462964646,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9808259955460622,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8761423222027079,
-      "eval_iou_concrete tiles": 0.8354388075663391,
-      "eval_iou_dark bricks": 0.8188763835053472,
-      "eval_iou_grass": 0.8242899301614433,
-      "eval_iou_red bricks": 0.8647611142812678,
-      "eval_iou_road": 0.9536246163887512,
-      "eval_iou_rough concrete": 0.608209459261043,
-      "eval_iou_rough red sidewalk": 0.6258544652701212,
-      "eval_iou_speedway bricks": 0.9747296153357259,
-      "eval_iou_steel": 0.8328401189333459,
-      "eval_iou_tiles": 0.21711973255916248,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08001357316970825,
-      "eval_mean_accuracy": 0.7631671974417947,
-      "eval_mean_iou": 0.724054812385486,
-      "eval_overall_accuracy": 0.9769962440102787,
-      "eval_runtime": 5.4467,
-      "eval_samples_per_second": 21.664,
-      "eval_steps_per_second": 1.469,
-      "step": 740
-    },
-    {
-      "epoch": 43.59,
-      "learning_rate": 4.116099071207431e-05,
-      "loss": 0.0394,
-      "step": 741
-    },
-    {
-      "epoch": 43.65,
-      "learning_rate": 4.114551083591332e-05,
-      "loss": 0.0407,
-      "step": 742
-    },
-    {
-      "epoch": 43.71,
-      "learning_rate": 4.113003095975233e-05,
-      "loss": 0.0399,
-      "step": 743
-    },
-    {
-      "epoch": 43.76,
-      "learning_rate": 4.111455108359134e-05,
-      "loss": 0.0389,
-      "step": 744
-    },
-    {
-      "epoch": 43.82,
-      "learning_rate": 4.109907120743035e-05,
-      "loss": 0.0447,
-      "step": 745
-    },
-    {
-      "epoch": 43.88,
-      "learning_rate": 4.108359133126935e-05,
-      "loss": 0.0367,
-      "step": 746
-    },
-    {
-      "epoch": 43.94,
-      "learning_rate": 4.106811145510836e-05,
-      "loss": 0.0384,
-      "step": 747
-    },
-    {
-      "epoch": 44.0,
-      "learning_rate": 4.105263157894737e-05,
-      "loss": 0.0375,
-      "step": 748
-    },
-    {
-      "epoch": 44.06,
-      "learning_rate": 4.103715170278638e-05,
-      "loss": 0.0339,
-      "step": 749
-    },
-    {
-      "epoch": 44.12,
-      "learning_rate": 4.102167182662539e-05,
-      "loss": 0.0487,
-      "step": 750
-    },
-    {
-      "epoch": 44.18,
-      "learning_rate": 4.10061919504644e-05,
-      "loss": 0.0603,
-      "step": 751
-    },
-    {
-      "epoch": 44.24,
-      "learning_rate": 4.0990712074303404e-05,
-      "loss": 0.0491,
-      "step": 752
-    },
-    {
-      "epoch": 44.29,
-      "learning_rate": 4.0975232198142414e-05,
-      "loss": 0.0369,
-      "step": 753
-    },
-    {
-      "epoch": 44.35,
-      "learning_rate": 4.0959752321981424e-05,
-      "loss": 0.0374,
-      "step": 754
-    },
-    {
-      "epoch": 44.41,
-      "learning_rate": 4.0944272445820434e-05,
-      "loss": 0.0528,
-      "step": 755
-    },
-    {
-      "epoch": 44.47,
-      "learning_rate": 4.0928792569659444e-05,
-      "loss": 0.0449,
-      "step": 756
-    },
-    {
-      "epoch": 44.53,
-      "learning_rate": 4.0913312693498454e-05,
-      "loss": 0.0336,
-      "step": 757
-    },
-    {
-      "epoch": 44.59,
-      "learning_rate": 4.0897832817337464e-05,
-      "loss": 0.0357,
-      "step": 758
-    },
-    {
-      "epoch": 44.65,
-      "learning_rate": 4.0882352941176474e-05,
-      "loss": 0.0454,
-      "step": 759
-    },
-    {
-      "epoch": 44.71,
-      "learning_rate": 4.0866873065015484e-05,
-      "loss": 0.0373,
-      "step": 760
-    },
-    {
-      "epoch": 44.71,
-      "eval_accuracy_NAT": 0.9927799215788771,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9443819760926696,
-      "eval_accuracy_concrete tiles": 0.8854819703040302,
-      "eval_accuracy_dark bricks": 0.8560955534712784,
-      "eval_accuracy_grass": 0.8600633215622598,
-      "eval_accuracy_red bricks": 0.8865883128295254,
-      "eval_accuracy_road": 0.9836190767320026,
-      "eval_accuracy_rough concrete": 0.614864824602697,
-      "eval_accuracy_rough red sidewalk": 0.7556982681476022,
-      "eval_accuracy_speedway bricks": 0.9885584625940709,
-      "eval_accuracy_steel": 0.8776082404154193,
-      "eval_accuracy_tiles": 0.20333848344390923,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9802811033779495,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8743327277784108,
-      "eval_iou_concrete tiles": 0.8404769361549819,
-      "eval_iou_dark bricks": 0.8347097412353144,
-      "eval_iou_grass": 0.8130648339724684,
-      "eval_iou_red bricks": 0.8579400510204082,
-      "eval_iou_road": 0.9458303635067712,
-      "eval_iou_rough concrete": 0.6082179507583125,
-      "eval_iou_rough red sidewalk": 0.6063963311609945,
-      "eval_iou_speedway bricks": 0.9741491155827552,
-      "eval_iou_steel": 0.8308712891900332,
-      "eval_iou_tiles": 0.19649287157118184,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08317837119102478,
-      "eval_mean_accuracy": 0.757621416290334,
-      "eval_mean_iou": 0.7202125627161217,
-      "eval_overall_accuracy": 0.9762467529814122,
-      "eval_runtime": 5.9253,
-      "eval_samples_per_second": 19.915,
-      "eval_steps_per_second": 1.35,
-      "step": 760
-    },
-    {
-      "epoch": 44.76,
-      "learning_rate": 4.0851393188854494e-05,
-      "loss": 0.0368,
-      "step": 761
-    },
-    {
-      "epoch": 44.82,
-      "learning_rate": 4.0835913312693504e-05,
-      "loss": 0.042,
-      "step": 762
-    },
-    {
-      "epoch": 44.88,
-      "learning_rate": 4.082043343653251e-05,
-      "loss": 0.0483,
-      "step": 763
-    },
-    {
-      "epoch": 44.94,
-      "learning_rate": 4.080495356037152e-05,
-      "loss": 0.0441,
-      "step": 764
-    },
-    {
-      "epoch": 45.0,
-      "learning_rate": 4.078947368421053e-05,
-      "loss": 0.0399,
-      "step": 765
-    },
-    {
-      "epoch": 45.06,
-      "learning_rate": 4.077399380804954e-05,
-      "loss": 0.0419,
-      "step": 766
-    },
-    {
-      "epoch": 45.12,
-      "learning_rate": 4.075851393188855e-05,
-      "loss": 0.0378,
-      "step": 767
-    },
-    {
-      "epoch": 45.18,
-      "learning_rate": 4.074303405572756e-05,
-      "loss": 0.0421,
-      "step": 768
-    },
-    {
-      "epoch": 45.24,
-      "learning_rate": 4.072755417956656e-05,
-      "loss": 0.039,
-      "step": 769
-    },
-    {
-      "epoch": 45.29,
-      "learning_rate": 4.071207430340557e-05,
-      "loss": 0.0456,
-      "step": 770
-    },
-    {
-      "epoch": 45.35,
-      "learning_rate": 4.069659442724458e-05,
-      "loss": 0.0489,
-      "step": 771
-    },
-    {
-      "epoch": 45.41,
-      "learning_rate": 4.068111455108359e-05,
-      "loss": 0.0467,
-      "step": 772
-    },
-    {
-      "epoch": 45.47,
-      "learning_rate": 4.06656346749226e-05,
-      "loss": 0.0413,
-      "step": 773
-    },
-    {
-      "epoch": 45.53,
-      "learning_rate": 4.065015479876162e-05,
-      "loss": 0.0379,
-      "step": 774
-    },
-    {
-      "epoch": 45.59,
-      "learning_rate": 4.063467492260062e-05,
-      "loss": 0.0513,
-      "step": 775
-    },
-    {
-      "epoch": 45.65,
-      "learning_rate": 4.061919504643963e-05,
-      "loss": 0.0432,
-      "step": 776
-    },
-    {
-      "epoch": 45.71,
-      "learning_rate": 4.060371517027864e-05,
-      "loss": 0.0403,
-      "step": 777
-    },
-    {
-      "epoch": 45.76,
-      "learning_rate": 4.058823529411765e-05,
-      "loss": 0.0389,
-      "step": 778
-    },
-    {
-      "epoch": 45.82,
-      "learning_rate": 4.057275541795666e-05,
-      "loss": 0.0458,
-      "step": 779
-    },
-    {
-      "epoch": 45.88,
-      "learning_rate": 4.055727554179567e-05,
-      "loss": 0.0332,
-      "step": 780
-    },
-    {
-      "epoch": 45.88,
-      "eval_accuracy_NAT": 0.9922975191587566,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9401074591543611,
-      "eval_accuracy_concrete tiles": 0.8655903841621494,
-      "eval_accuracy_dark bricks": 0.8542053480849109,
-      "eval_accuracy_grass": 0.8879079661023365,
-      "eval_accuracy_red bricks": 0.891146748681898,
-      "eval_accuracy_road": 0.9791310500582247,
-      "eval_accuracy_rough concrete": 0.6247504752143679,
-      "eval_accuracy_rough red sidewalk": 0.8070447213469796,
-      "eval_accuracy_speedway bricks": 0.9859261224246387,
-      "eval_accuracy_steel": 0.9102140568852549,
-      "eval_accuracy_tiles": 0.22954345784483018,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9806645176463016,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8770076222436308,
-      "eval_iou_concrete tiles": 0.8293327311730835,
-      "eval_iou_dark bricks": 0.8286216948900772,
-      "eval_iou_grass": 0.8239650451950314,
-      "eval_iou_red bricks": 0.8598378464310318,
-      "eval_iou_road": 0.945054655970685,
-      "eval_iou_rough concrete": 0.5886596896265187,
-      "eval_iou_rough red sidewalk": 0.5982596286241478,
-      "eval_iou_speedway bricks": 0.9745539449151847,
-      "eval_iou_steel": 0.8352553504231953,
-      "eval_iou_tiles": 0.22587568937248473,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08175460249185562,
-      "eval_mean_accuracy": 0.7667588699322084,
-      "eval_mean_iou": 0.7205452628085671,
-      "eval_overall_accuracy": 0.976465936434471,
-      "eval_runtime": 7.852,
-      "eval_samples_per_second": 15.028,
-      "eval_steps_per_second": 1.019,
-      "step": 780
-    },
-    {
-      "epoch": 45.94,
-      "learning_rate": 4.0541795665634674e-05,
-      "loss": 0.0346,
-      "step": 781
-    },
-    {
-      "epoch": 46.0,
-      "learning_rate": 4.0526315789473684e-05,
-      "loss": 0.0372,
-      "step": 782
-    },
-    {
-      "epoch": 46.06,
-      "learning_rate": 4.0510835913312694e-05,
-      "loss": 0.0364,
-      "step": 783
-    },
-    {
-      "epoch": 46.12,
-      "learning_rate": 4.0495356037151704e-05,
-      "loss": 0.0379,
-      "step": 784
-    },
-    {
-      "epoch": 46.18,
-      "learning_rate": 4.0479876160990714e-05,
-      "loss": 0.0333,
-      "step": 785
-    },
-    {
-      "epoch": 46.24,
-      "learning_rate": 4.0464396284829724e-05,
-      "loss": 0.0403,
-      "step": 786
-    },
-    {
-      "epoch": 46.29,
-      "learning_rate": 4.044891640866873e-05,
-      "loss": 0.0401,
-      "step": 787
-    },
-    {
-      "epoch": 46.35,
-      "learning_rate": 4.0433436532507744e-05,
-      "loss": 0.0395,
-      "step": 788
-    },
-    {
-      "epoch": 46.41,
-      "learning_rate": 4.0417956656346754e-05,
-      "loss": 0.0389,
-      "step": 789
-    },
-    {
-      "epoch": 46.47,
-      "learning_rate": 4.0402476780185764e-05,
-      "loss": 0.0436,
-      "step": 790
-    },
-    {
-      "epoch": 46.53,
-      "learning_rate": 4.0386996904024774e-05,
-      "loss": 0.0323,
-      "step": 791
-    },
-    {
-      "epoch": 46.59,
-      "learning_rate": 4.0371517027863784e-05,
-      "loss": 0.0415,
-      "step": 792
-    },
-    {
-      "epoch": 46.65,
-      "learning_rate": 4.035603715170279e-05,
-      "loss": 0.0397,
-      "step": 793
-    },
-    {
-      "epoch": 46.71,
-      "learning_rate": 4.03405572755418e-05,
-      "loss": 0.0414,
-      "step": 794
-    },
-    {
-      "epoch": 46.76,
-      "learning_rate": 4.032507739938081e-05,
-      "loss": 0.0384,
-      "step": 795
-    },
-    {
-      "epoch": 46.82,
-      "learning_rate": 4.030959752321982e-05,
-      "loss": 0.0424,
-      "step": 796
-    },
-    {
-      "epoch": 46.88,
-      "learning_rate": 4.029411764705883e-05,
-      "loss": 0.0348,
-      "step": 797
-    },
-    {
-      "epoch": 46.94,
-      "learning_rate": 4.027863777089784e-05,
-      "loss": 0.0426,
-      "step": 798
-    },
-    {
-      "epoch": 47.0,
-      "learning_rate": 4.026315789473684e-05,
-      "loss": 0.0394,
-      "step": 799
-    },
-    {
-      "epoch": 47.06,
-      "learning_rate": 4.024767801857585e-05,
-      "loss": 0.0386,
-      "step": 800
-    },
-    {
-      "epoch": 47.06,
-      "eval_accuracy_NAT": 0.9925614711311426,
-      "eval_accuracy_REST": 0.032520325203252036,
-      "eval_accuracy_concrete": 0.9593274974048668,
-      "eval_accuracy_concrete tiles": 0.844025453688428,
-      "eval_accuracy_dark bricks": 0.8593495779338859,
-      "eval_accuracy_grass": 0.8724560666447142,
-      "eval_accuracy_red bricks": 0.898011862917399,
-      "eval_accuracy_road": 0.9703276131211986,
-      "eval_accuracy_rough concrete": 0.6201145281523546,
-      "eval_accuracy_rough red sidewalk": 0.7499454801134013,
-      "eval_accuracy_speedway bricks": 0.9851378427046082,
-      "eval_accuracy_steel": 0.9042666440098557,
-      "eval_accuracy_tiles": 0.19943045835984125,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9810036198475386,
-      "eval_iou_REST": 0.032467532467532464,
-      "eval_iou_concrete": 0.8732396179205504,
-      "eval_iou_concrete tiles": 0.8141311266709103,
-      "eval_iou_dark bricks": 0.8340144900613041,
-      "eval_iou_grass": 0.8221526718918581,
-      "eval_iou_red bricks": 0.8615765623353356,
-      "eval_iou_road": 0.9528672167616816,
-      "eval_iou_rough concrete": 0.57871907375586,
-      "eval_iou_rough red sidewalk": 0.5950582366700081,
-      "eval_iou_speedway bricks": 0.975265346173186,
-      "eval_iou_steel": 0.8392392058091613,
-      "eval_iou_tiles": 0.1944871188844245,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08345451205968857,
-      "eval_mean_accuracy": 0.7605749862603807,
-      "eval_mean_iou": 0.7195555245576423,
-      "eval_overall_accuracy": 0.976670927920584,
-      "eval_runtime": 7.5362,
-      "eval_samples_per_second": 15.658,
-      "eval_steps_per_second": 1.062,
-      "step": 800
-    },
-    {
-      "epoch": 47.12,
-      "learning_rate": 4.023219814241486e-05,
-      "loss": 0.0389,
-      "step": 801
-    },
-    {
-      "epoch": 47.18,
-      "learning_rate": 4.021671826625387e-05,
-      "loss": 0.0433,
-      "step": 802
-    },
-    {
-      "epoch": 47.24,
-      "learning_rate": 4.020123839009288e-05,
-      "loss": 0.0384,
-      "step": 803
-    },
-    {
-      "epoch": 47.29,
-      "learning_rate": 4.018575851393189e-05,
-      "loss": 0.0359,
-      "step": 804
-    },
-    {
-      "epoch": 47.35,
-      "learning_rate": 4.01702786377709e-05,
-      "loss": 0.0408,
-      "step": 805
-    },
-    {
-      "epoch": 47.41,
-      "learning_rate": 4.015479876160991e-05,
-      "loss": 0.043,
-      "step": 806
-    },
-    {
-      "epoch": 47.47,
-      "learning_rate": 4.013931888544892e-05,
-      "loss": 0.0336,
-      "step": 807
-    },
-    {
-      "epoch": 47.53,
-      "learning_rate": 4.012383900928793e-05,
-      "loss": 0.0477,
-      "step": 808
-    },
-    {
-      "epoch": 47.59,
-      "learning_rate": 4.010835913312694e-05,
-      "loss": 0.0382,
-      "step": 809
-    },
-    {
-      "epoch": 47.65,
-      "learning_rate": 4.009287925696595e-05,
-      "loss": 0.0373,
-      "step": 810
-    },
-    {
-      "epoch": 47.71,
-      "learning_rate": 4.007739938080495e-05,
-      "loss": 0.0354,
-      "step": 811
-    },
-    {
-      "epoch": 47.76,
-      "learning_rate": 4.006191950464396e-05,
-      "loss": 0.0483,
-      "step": 812
-    },
-    {
-      "epoch": 47.82,
-      "learning_rate": 4.004643962848297e-05,
-      "loss": 0.0442,
-      "step": 813
-    },
-    {
-      "epoch": 47.88,
-      "learning_rate": 4.003095975232198e-05,
-      "loss": 0.0437,
-      "step": 814
-    },
-    {
-      "epoch": 47.94,
-      "learning_rate": 4.001547987616099e-05,
-      "loss": 0.0392,
-      "step": 815
-    },
-    {
-      "epoch": 48.0,
-      "learning_rate": 4e-05,
-      "loss": 0.0403,
-      "step": 816
-    },
-    {
-      "epoch": 48.06,
-      "learning_rate": 3.9984520123839006e-05,
-      "loss": 0.0385,
-      "step": 817
-    },
-    {
-      "epoch": 48.12,
-      "learning_rate": 3.9969040247678017e-05,
-      "loss": 0.0353,
-      "step": 818
-    },
-    {
-      "epoch": 48.18,
-      "learning_rate": 3.9953560371517027e-05,
-      "loss": 0.0362,
-      "step": 819
-    },
-    {
-      "epoch": 48.24,
-      "learning_rate": 3.9938080495356037e-05,
-      "loss": 0.0457,
-      "step": 820
-    },
-    {
-      "epoch": 48.24,
-      "eval_accuracy_NAT": 0.993284913026046,
-      "eval_accuracy_REST": 0.011382113821138212,
-      "eval_accuracy_concrete": 0.9413209419225236,
-      "eval_accuracy_concrete tiles": 0.9300730615130803,
-      "eval_accuracy_dark bricks": 0.884673544781119,
-      "eval_accuracy_grass": 0.8748254209767341,
-      "eval_accuracy_red bricks": 0.9043277680140598,
-      "eval_accuracy_road": 0.9777367583706935,
-      "eval_accuracy_rough concrete": 0.6237732318748175,
-      "eval_accuracy_rough red sidewalk": 0.7846050880214169,
-      "eval_accuracy_speedway bricks": 0.9886994961178938,
-      "eval_accuracy_steel": 0.8896030226851319,
-      "eval_accuracy_tiles": 0.24520585294919567,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9809656395022599,
-      "eval_iou_REST": 0.011363636363636364,
-      "eval_iou_concrete": 0.8789259208108553,
-      "eval_iou_concrete tiles": 0.8180046846174575,
-      "eval_iou_dark bricks": 0.8469760345988739,
-      "eval_iou_grass": 0.8197962701548546,
-      "eval_iou_red bricks": 0.8569346864428832,
-      "eval_iou_road": 0.9515196400171004,
-      "eval_iou_rough concrete": 0.6110368498164221,
-      "eval_iou_rough red sidewalk": 0.6043103797790945,
-      "eval_iou_speedway bricks": 0.97510861397648,
-      "eval_iou_steel": 0.8364921448745483,
-      "eval_iou_tiles": 0.23762550642945218,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0798560082912445,
-      "eval_mean_accuracy": 0.773039324159527,
-      "eval_mean_iou": 0.7253123082603015,
-      "eval_overall_accuracy": 0.9771031201896021,
-      "eval_runtime": 5.5393,
-      "eval_samples_per_second": 21.302,
-      "eval_steps_per_second": 1.444,
-      "step": 820
-    },
-    {
-      "epoch": 48.29,
-      "learning_rate": 3.9922600619195047e-05,
-      "loss": 0.0346,
-      "step": 821
-    },
-    {
-      "epoch": 48.35,
-      "learning_rate": 3.9907120743034057e-05,
-      "loss": 0.0423,
-      "step": 822
-    },
-    {
-      "epoch": 48.41,
-      "learning_rate": 3.9891640866873067e-05,
-      "loss": 0.0369,
-      "step": 823
-    },
-    {
-      "epoch": 48.47,
-      "learning_rate": 3.9876160990712077e-05,
-      "loss": 0.0436,
-      "step": 824
-    },
-    {
-      "epoch": 48.53,
-      "learning_rate": 3.9860681114551087e-05,
-      "loss": 0.0397,
-      "step": 825
-    },
-    {
-      "epoch": 48.59,
-      "learning_rate": 3.9845201238390097e-05,
-      "loss": 0.0513,
-      "step": 826
-    },
-    {
-      "epoch": 48.65,
-      "learning_rate": 3.9829721362229107e-05,
-      "loss": 0.0333,
-      "step": 827
-    },
-    {
-      "epoch": 48.71,
-      "learning_rate": 3.9814241486068117e-05,
-      "loss": 0.0362,
-      "step": 828
-    },
-    {
-      "epoch": 48.76,
-      "learning_rate": 3.979876160990712e-05,
-      "loss": 0.0405,
-      "step": 829
-    },
-    {
-      "epoch": 48.82,
-      "learning_rate": 3.978328173374613e-05,
-      "loss": 0.0367,
-      "step": 830
-    },
-    {
-      "epoch": 48.88,
-      "learning_rate": 3.976780185758514e-05,
-      "loss": 0.0366,
-      "step": 831
-    },
-    {
-      "epoch": 48.94,
-      "learning_rate": 3.975232198142415e-05,
-      "loss": 0.0449,
-      "step": 832
-    },
-    {
-      "epoch": 49.0,
-      "learning_rate": 3.973684210526316e-05,
-      "loss": 0.0387,
-      "step": 833
-    },
-    {
-      "epoch": 49.06,
-      "learning_rate": 3.972136222910217e-05,
-      "loss": 0.0348,
-      "step": 834
-    },
-    {
-      "epoch": 49.12,
-      "learning_rate": 3.970588235294117e-05,
-      "loss": 0.0416,
-      "step": 835
-    },
-    {
-      "epoch": 49.18,
-      "learning_rate": 3.969040247678018e-05,
-      "loss": 0.0352,
-      "step": 836
-    },
-    {
-      "epoch": 49.24,
-      "learning_rate": 3.96749226006192e-05,
-      "loss": 0.0413,
-      "step": 837
-    },
-    {
-      "epoch": 49.29,
-      "learning_rate": 3.965944272445821e-05,
-      "loss": 0.0322,
-      "step": 838
-    },
-    {
-      "epoch": 49.35,
-      "learning_rate": 3.964396284829722e-05,
-      "loss": 0.0416,
-      "step": 839
-    },
-    {
-      "epoch": 49.41,
-      "learning_rate": 3.962848297213623e-05,
-      "loss": 0.045,
-      "step": 840
-    },
-    {
-      "epoch": 49.41,
-      "eval_accuracy_NAT": 0.9926908054411413,
-      "eval_accuracy_REST": 0.0,
-      "eval_accuracy_concrete": 0.9410973553337441,
-      "eval_accuracy_concrete tiles": 0.8221305679943436,
-      "eval_accuracy_dark bricks": 0.8123193537890243,
-      "eval_accuracy_grass": 0.88069865673506,
-      "eval_accuracy_red bricks": 0.8406195079086116,
-      "eval_accuracy_road": 0.9731642939481114,
-      "eval_accuracy_rough concrete": 0.6200370637412927,
-      "eval_accuracy_rough red sidewalk": 0.7597740996698726,
-      "eval_accuracy_speedway bricks": 0.9920198673907873,
-      "eval_accuracy_steel": 0.8703638917071074,
-      "eval_accuracy_tiles": 0.17676997182586568,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9807335603411328,
-      "eval_iou_REST": 0.0,
-      "eval_iou_concrete": 0.8722015791095635,
-      "eval_iou_concrete tiles": 0.7926333250017042,
-      "eval_iou_dark bricks": 0.7954042808411741,
-      "eval_iou_grass": 0.8199669985895554,
-      "eval_iou_red bricks": 0.8261456253036109,
-      "eval_iou_road": 0.9523591859959761,
-      "eval_iou_rough concrete": 0.6057257951846505,
-      "eval_iou_rough red sidewalk": 0.6041221949163184,
-      "eval_iou_speedway bricks": 0.9719564469144301,
-      "eval_iou_steel": 0.8282038783007764,
-      "eval_iou_tiles": 0.17556792537987062,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0856236144900322,
-      "eval_mean_accuracy": 0.7447450334988432,
-      "eval_mean_iou": 0.7096169842983664,
-      "eval_overall_accuracy": 0.9761548769676079,
-      "eval_runtime": 5.6297,
-      "eval_samples_per_second": 20.96,
-      "eval_steps_per_second": 1.421,
-      "step": 840
-    },
-    {
-      "epoch": 49.47,
-      "learning_rate": 3.961300309597523e-05,
-      "loss": 0.0331,
-      "step": 841
-    },
-    {
-      "epoch": 49.53,
-      "learning_rate": 3.959752321981424e-05,
-      "loss": 0.0397,
-      "step": 842
-    },
-    {
-      "epoch": 49.59,
-      "learning_rate": 3.958204334365325e-05,
-      "loss": 0.0355,
-      "step": 843
-    },
-    {
-      "epoch": 49.65,
-      "learning_rate": 3.956656346749226e-05,
-      "loss": 0.0337,
-      "step": 844
-    },
-    {
-      "epoch": 49.71,
-      "learning_rate": 3.955108359133127e-05,
-      "loss": 0.0476,
-      "step": 845
-    },
-    {
-      "epoch": 49.76,
-      "learning_rate": 3.953560371517028e-05,
-      "loss": 0.0418,
-      "step": 846
-    },
-    {
-      "epoch": 49.82,
-      "learning_rate": 3.9520123839009286e-05,
-      "loss": 0.04,
-      "step": 847
-    },
-    {
-      "epoch": 49.88,
-      "learning_rate": 3.9504643962848296e-05,
-      "loss": 0.0385,
-      "step": 848
-    },
-    {
-      "epoch": 49.94,
-      "learning_rate": 3.9489164086687306e-05,
-      "loss": 0.0404,
-      "step": 849
-    },
-    {
-      "epoch": 50.0,
-      "learning_rate": 3.9473684210526316e-05,
-      "loss": 0.0332,
-      "step": 850
-    },
-    {
-      "epoch": 50.06,
-      "learning_rate": 3.9458204334365326e-05,
-      "loss": 0.0366,
-      "step": 851
-    },
-    {
-      "epoch": 50.12,
-      "learning_rate": 3.9442724458204336e-05,
-      "loss": 0.0356,
-      "step": 852
-    },
-    {
-      "epoch": 50.18,
-      "learning_rate": 3.9427244582043346e-05,
-      "loss": 0.0462,
-      "step": 853
-    },
-    {
-      "epoch": 50.24,
-      "learning_rate": 3.9411764705882356e-05,
-      "loss": 0.0339,
-      "step": 854
-    },
-    {
-      "epoch": 50.29,
-      "learning_rate": 3.9396284829721366e-05,
-      "loss": 0.0345,
-      "step": 855
-    },
-    {
-      "epoch": 50.35,
-      "learning_rate": 3.9380804953560376e-05,
-      "loss": 0.0393,
-      "step": 856
-    },
-    {
-      "epoch": 50.41,
-      "learning_rate": 3.9365325077399386e-05,
-      "loss": 0.0454,
-      "step": 857
-    },
-    {
-      "epoch": 50.47,
-      "learning_rate": 3.9349845201238396e-05,
-      "loss": 0.0374,
-      "step": 858
-    },
-    {
-      "epoch": 50.53,
-      "learning_rate": 3.93343653250774e-05,
-      "loss": 0.0394,
-      "step": 859
-    },
-    {
-      "epoch": 50.59,
-      "learning_rate": 3.931888544891641e-05,
-      "loss": 0.0414,
-      "step": 860
-    },
-    {
-      "epoch": 50.59,
-      "eval_accuracy_NAT": 0.9925857314227221,
-      "eval_accuracy_REST": 0.12195121951219512,
-      "eval_accuracy_concrete": 0.9468819416947328,
-      "eval_accuracy_concrete tiles": 0.8721659203393826,
-      "eval_accuracy_dark bricks": 0.8332982409126581,
-      "eval_accuracy_grass": 0.8964788972283096,
-      "eval_accuracy_red bricks": 0.8888949912126538,
-      "eval_accuracy_road": 0.9774647405596726,
-      "eval_accuracy_rough concrete": 0.6200728165463982,
-      "eval_accuracy_rough red sidewalk": 0.7201889020070839,
-      "eval_accuracy_speedway bricks": 0.990332606059609,
-      "eval_accuracy_steel": 0.8716133481935359,
-      "eval_accuracy_tiles": 0.21218455572722592,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9814399439643317,
-      "eval_iou_REST": 0.12175324675324675,
-      "eval_iou_concrete": 0.877990003317107,
-      "eval_iou_concrete tiles": 0.8280043854742354,
-      "eval_iou_dark bricks": 0.8171370920438281,
-      "eval_iou_grass": 0.8289540342020185,
-      "eval_iou_red bricks": 0.8561227188574452,
-      "eval_iou_road": 0.9513924722233781,
-      "eval_iou_rough concrete": 0.6119880495894987,
-      "eval_iou_rough red sidewalk": 0.5965045592705167,
-      "eval_iou_speedway bricks": 0.9745007669094427,
-      "eval_iou_steel": 0.8298664065381093,
-      "eval_iou_tiles": 0.2088377362991234,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08020060509443283,
-      "eval_mean_accuracy": 0.7649318393397061,
-      "eval_mean_iou": 0.7295762627263294,
-      "eval_overall_accuracy": 0.9771873021529893,
-      "eval_runtime": 5.851,
-      "eval_samples_per_second": 20.167,
-      "eval_steps_per_second": 1.367,
-      "step": 860
-    },
-    {
-      "epoch": 50.65,
-      "learning_rate": 3.930340557275542e-05,
-      "loss": 0.0366,
-      "step": 861
-    },
-    {
-      "epoch": 50.71,
-      "learning_rate": 3.928792569659443e-05,
-      "loss": 0.0387,
-      "step": 862
-    },
-    {
-      "epoch": 50.76,
-      "learning_rate": 3.927244582043344e-05,
-      "loss": 0.0357,
-      "step": 863
-    },
-    {
-      "epoch": 50.82,
-      "learning_rate": 3.925696594427245e-05,
-      "loss": 0.0407,
-      "step": 864
-    },
-    {
-      "epoch": 50.88,
-      "learning_rate": 3.924148606811145e-05,
-      "loss": 0.0329,
-      "step": 865
-    },
-    {
-      "epoch": 50.94,
-      "learning_rate": 3.922600619195046e-05,
-      "loss": 0.0373,
-      "step": 866
-    },
-    {
-      "epoch": 51.0,
-      "learning_rate": 3.921052631578947e-05,
-      "loss": 0.0337,
-      "step": 867
-    },
-    {
-      "epoch": 51.06,
-      "learning_rate": 3.919504643962848e-05,
-      "loss": 0.0376,
-      "step": 868
-    },
-    {
-      "epoch": 51.12,
-      "learning_rate": 3.917956656346749e-05,
-      "loss": 0.0329,
-      "step": 869
-    },
-    {
-      "epoch": 51.18,
-      "learning_rate": 3.91640866873065e-05,
-      "loss": 0.0431,
-      "step": 870
-    },
-    {
-      "epoch": 51.24,
-      "learning_rate": 3.914860681114551e-05,
-      "loss": 0.0336,
-      "step": 871
-    },
-    {
-      "epoch": 51.29,
-      "learning_rate": 3.913312693498452e-05,
-      "loss": 0.0293,
-      "step": 872
-    },
-    {
-      "epoch": 51.35,
-      "learning_rate": 3.911764705882353e-05,
-      "loss": 0.0379,
-      "step": 873
-    },
-    {
-      "epoch": 51.41,
-      "learning_rate": 3.910216718266254e-05,
-      "loss": 0.0294,
-      "step": 874
-    },
-    {
-      "epoch": 51.47,
-      "learning_rate": 3.908668730650155e-05,
-      "loss": 0.0377,
-      "step": 875
-    },
-    {
-      "epoch": 51.53,
-      "learning_rate": 3.907120743034056e-05,
-      "loss": 0.0342,
-      "step": 876
-    },
-    {
-      "epoch": 51.59,
-      "learning_rate": 3.9055727554179566e-05,
-      "loss": 0.0444,
-      "step": 877
-    },
-    {
-      "epoch": 51.65,
-      "learning_rate": 3.9040247678018576e-05,
-      "loss": 0.0369,
-      "step": 878
-    },
-    {
-      "epoch": 51.71,
-      "learning_rate": 3.9024767801857586e-05,
-      "loss": 0.0404,
-      "step": 879
-    },
-    {
-      "epoch": 51.76,
-      "learning_rate": 3.9009287925696596e-05,
-      "loss": 0.038,
-      "step": 880
-    },
-    {
-      "epoch": 51.76,
-      "eval_accuracy_NAT": 0.9934121986892003,
-      "eval_accuracy_REST": 0.03414634146341464,
-      "eval_accuracy_concrete": 0.9429991790359099,
-      "eval_accuracy_concrete tiles": 0.8840207400424228,
-      "eval_accuracy_dark bricks": 0.8399642057309114,
-      "eval_accuracy_grass": 0.8809948260265625,
-      "eval_accuracy_red bricks": 0.8958699472759226,
-      "eval_accuracy_road": 0.9795115543439749,
-      "eval_accuracy_rough concrete": 0.6205018502076642,
-      "eval_accuracy_rough red sidewalk": 0.7668428849667993,
-      "eval_accuracy_speedway bricks": 0.9909476356063893,
-      "eval_accuracy_steel": 0.8687745830563705,
-      "eval_accuracy_tiles": 0.1804659335332788,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9812413400145908,
-      "eval_iou_REST": 0.03414634146341464,
-      "eval_iou_concrete": 0.8798285343221981,
-      "eval_iou_concrete tiles": 0.8269909163065526,
-      "eval_iou_dark bricks": 0.8211959877987163,
-      "eval_iou_grass": 0.82433791603805,
-      "eval_iou_red bricks": 0.8576235541535226,
-      "eval_iou_road": 0.9539101529545061,
-      "eval_iou_rough concrete": 0.6107806909496158,
-      "eval_iou_rough red sidewalk": 0.6117803748410165,
-      "eval_iou_speedway bricks": 0.97457153797357,
-      "eval_iou_steel": 0.8285751056390896,
-      "eval_iou_tiles": 0.17920640173279986,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08046015352010727,
-      "eval_mean_accuracy": 0.7598809138445246,
-      "eval_mean_iou": 0.7218606810913571,
-      "eval_overall_accuracy": 0.9772612684864109,
-      "eval_runtime": 5.5554,
-      "eval_samples_per_second": 21.241,
-      "eval_steps_per_second": 1.44,
-      "step": 880
-    },
-    {
-      "epoch": 51.82,
-      "learning_rate": 3.8993808049535606e-05,
-      "loss": 0.0367,
-      "step": 881
-    },
-    {
-      "epoch": 51.88,
-      "learning_rate": 3.8978328173374616e-05,
-      "loss": 0.0367,
-      "step": 882
-    },
-    {
-      "epoch": 51.94,
-      "learning_rate": 3.896284829721362e-05,
-      "loss": 0.0403,
-      "step": 883
-    },
-    {
-      "epoch": 52.0,
-      "learning_rate": 3.894736842105263e-05,
-      "loss": 0.0419,
-      "step": 884
-    },
-    {
-      "epoch": 52.06,
-      "learning_rate": 3.893188854489164e-05,
-      "loss": 0.0356,
-      "step": 885
-    },
-    {
-      "epoch": 52.12,
-      "learning_rate": 3.8916408668730656e-05,
-      "loss": 0.0359,
-      "step": 886
-    },
-    {
-      "epoch": 52.18,
-      "learning_rate": 3.8900928792569666e-05,
-      "loss": 0.044,
-      "step": 887
-    },
-    {
-      "epoch": 52.24,
-      "learning_rate": 3.8885448916408676e-05,
-      "loss": 0.0315,
-      "step": 888
-    },
-    {
-      "epoch": 52.29,
-      "learning_rate": 3.886996904024768e-05,
-      "loss": 0.0376,
-      "step": 889
-    },
-    {
-      "epoch": 52.35,
-      "learning_rate": 3.885448916408669e-05,
-      "loss": 0.0388,
-      "step": 890
-    },
-    {
-      "epoch": 52.41,
-      "learning_rate": 3.88390092879257e-05,
-      "loss": 0.0376,
-      "step": 891
-    },
-    {
-      "epoch": 52.47,
-      "learning_rate": 3.882352941176471e-05,
-      "loss": 0.0388,
-      "step": 892
-    },
-    {
-      "epoch": 52.53,
-      "learning_rate": 3.880804953560372e-05,
-      "loss": 0.0333,
-      "step": 893
-    },
-    {
-      "epoch": 52.59,
-      "learning_rate": 3.879256965944273e-05,
-      "loss": 0.0375,
-      "step": 894
-    },
-    {
-      "epoch": 52.65,
-      "learning_rate": 3.877708978328173e-05,
-      "loss": 0.0365,
-      "step": 895
-    },
-    {
-      "epoch": 52.71,
-      "learning_rate": 3.876160990712074e-05,
-      "loss": 0.0364,
-      "step": 896
-    },
-    {
-      "epoch": 52.76,
-      "learning_rate": 3.874613003095975e-05,
-      "loss": 0.0352,
-      "step": 897
-    },
-    {
-      "epoch": 52.82,
-      "learning_rate": 3.873065015479876e-05,
-      "loss": 0.0356,
-      "step": 898
-    },
-    {
-      "epoch": 52.88,
-      "learning_rate": 3.871517027863777e-05,
-      "loss": 0.0347,
-      "step": 899
-    },
-    {
-      "epoch": 52.94,
-      "learning_rate": 3.869969040247678e-05,
-      "loss": 0.0348,
-      "step": 900
-    },
-    {
-      "epoch": 52.94,
-      "eval_accuracy_NAT": 0.9888540133719492,
-      "eval_accuracy_REST": 0.3008130081300813,
-      "eval_accuracy_concrete": 0.9529829890736865,
-      "eval_accuracy_concrete tiles": 0.901249116191374,
-      "eval_accuracy_dark bricks": 0.8564975211990123,
-      "eval_accuracy_grass": 0.9121957404610779,
-      "eval_accuracy_red bricks": 0.9022407732864675,
-      "eval_accuracy_road": 0.9801090315791836,
-      "eval_accuracy_rough concrete": 0.6241903479343817,
-      "eval_accuracy_rough red sidewalk": 0.8273035592085969,
-      "eval_accuracy_speedway bricks": 0.9887646148773745,
-      "eval_accuracy_steel": 0.8903377030991518,
-      "eval_accuracy_tiles": 0.20206610318398013,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9810375330982319,
-      "eval_iou_REST": 0.29983792544570503,
-      "eval_iou_concrete": 0.877285158428386,
-      "eval_iou_concrete tiles": 0.8302575014112641,
-      "eval_iou_dark bricks": 0.8291632115111113,
-      "eval_iou_grass": 0.8277350139310445,
-      "eval_iou_red bricks": 0.8709574806489238,
-      "eval_iou_road": 0.9462856099116447,
-      "eval_iou_rough concrete": 0.585037698966769,
-      "eval_iou_rough red sidewalk": 0.6260363851568591,
-      "eval_iou_speedway bricks": 0.9756787477607302,
-      "eval_iou_steel": 0.8367543447627994,
-      "eval_iou_tiles": 0.19808154901553174,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08102715015411377,
-      "eval_mean_accuracy": 0.7944311170458707,
-      "eval_mean_iou": 0.7449344738499231,
-      "eval_overall_accuracy": 0.9767890542240466,
-      "eval_runtime": 5.6704,
-      "eval_samples_per_second": 20.81,
-      "eval_steps_per_second": 1.411,
-      "step": 900
-    },
-    {
-      "epoch": 53.0,
-      "learning_rate": 3.868421052631579e-05,
-      "loss": 0.0401,
-      "step": 901
-    },
-    {
-      "epoch": 53.06,
-      "learning_rate": 3.86687306501548e-05,
-      "loss": 0.0358,
-      "step": 902
-    },
-    {
-      "epoch": 53.12,
-      "learning_rate": 3.865325077399381e-05,
-      "loss": 0.0372,
-      "step": 903
-    },
-    {
-      "epoch": 53.18,
-      "learning_rate": 3.863777089783282e-05,
-      "loss": 0.0398,
-      "step": 904
-    },
-    {
-      "epoch": 53.24,
-      "learning_rate": 3.862229102167183e-05,
-      "loss": 0.0396,
-      "step": 905
-    },
-    {
-      "epoch": 53.29,
-      "learning_rate": 3.8606811145510836e-05,
-      "loss": 0.0405,
-      "step": 906
-    },
-    {
-      "epoch": 53.35,
-      "learning_rate": 3.8591331269349846e-05,
-      "loss": 0.0309,
-      "step": 907
-    },
-    {
-      "epoch": 53.41,
-      "learning_rate": 3.8575851393188856e-05,
-      "loss": 0.0356,
-      "step": 908
-    },
-    {
-      "epoch": 53.47,
-      "learning_rate": 3.8560371517027866e-05,
-      "loss": 0.0368,
-      "step": 909
-    },
-    {
-      "epoch": 53.53,
-      "learning_rate": 3.8544891640866876e-05,
-      "loss": 0.0343,
-      "step": 910
-    },
-    {
-      "epoch": 53.59,
-      "learning_rate": 3.8529411764705886e-05,
-      "loss": 0.0337,
-      "step": 911
-    },
-    {
-      "epoch": 53.65,
-      "learning_rate": 3.851393188854489e-05,
-      "loss": 0.0344,
-      "step": 912
-    },
-    {
-      "epoch": 53.71,
-      "learning_rate": 3.84984520123839e-05,
-      "loss": 0.0422,
-      "step": 913
-    },
-    {
-      "epoch": 53.76,
-      "learning_rate": 3.848297213622291e-05,
-      "loss": 0.0366,
-      "step": 914
-    },
-    {
-      "epoch": 53.82,
-      "learning_rate": 3.846749226006192e-05,
-      "loss": 0.0342,
-      "step": 915
-    },
-    {
-      "epoch": 53.88,
-      "learning_rate": 3.845201238390093e-05,
-      "loss": 0.0374,
-      "step": 916
-    },
-    {
-      "epoch": 53.94,
-      "learning_rate": 3.843653250773994e-05,
-      "loss": 0.0311,
-      "step": 917
-    },
-    {
-      "epoch": 54.0,
-      "learning_rate": 3.842105263157895e-05,
-      "loss": 0.0397,
-      "step": 918
-    },
-    {
-      "epoch": 54.06,
-      "learning_rate": 3.840557275541796e-05,
-      "loss": 0.0368,
-      "step": 919
-    },
-    {
-      "epoch": 54.12,
-      "learning_rate": 3.839009287925697e-05,
-      "loss": 0.0386,
-      "step": 920
-    },
-    {
-      "epoch": 54.12,
-      "eval_accuracy_NAT": 0.9937696336518063,
-      "eval_accuracy_REST": 0.05365853658536585,
-      "eval_accuracy_concrete": 0.9511832125838354,
-      "eval_accuracy_concrete tiles": 0.8834786707518265,
-      "eval_accuracy_dark bricks": 0.8349826771050667,
-      "eval_accuracy_grass": 0.8486016646038901,
-      "eval_accuracy_red bricks": 0.9054261862917399,
-      "eval_accuracy_road": 0.9782642270926534,
-      "eval_accuracy_rough concrete": 0.616437948027339,
-      "eval_accuracy_rough red sidewalk": 0.7137668353649824,
-      "eval_accuracy_speedway bricks": 0.9895650615235185,
-      "eval_accuracy_steel": 0.8807443761963546,
-      "eval_accuracy_tiles": 0.19912751067890574,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9802639671263945,
-      "eval_iou_REST": 0.05365853658536585,
-      "eval_iou_concrete": 0.8779370132905309,
-      "eval_iou_concrete tiles": 0.8351379049146728,
-      "eval_iou_dark bricks": 0.819423311731004,
-      "eval_iou_grass": 0.8058242656770866,
-      "eval_iou_red bricks": 0.8772415261001437,
-      "eval_iou_road": 0.9516963894859102,
-      "eval_iou_rough concrete": 0.6105622248191036,
-      "eval_iou_rough red sidewalk": 0.6058455194776181,
-      "eval_iou_speedway bricks": 0.9750876922217045,
-      "eval_iou_steel": 0.8357127783429009,
-      "eval_iou_tiles": 0.1962734032070232,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08540533483028412,
-      "eval_mean_accuracy": 0.7576158877274832,
-      "eval_mean_iou": 0.7249741948445737,
-      "eval_overall_accuracy": 0.9766210135767015,
-      "eval_runtime": 5.7762,
-      "eval_samples_per_second": 20.429,
-      "eval_steps_per_second": 1.385,
-      "step": 920
-    },
-    {
-      "epoch": 54.18,
-      "learning_rate": 3.837461300309598e-05,
-      "loss": 0.0342,
-      "step": 921
-    },
-    {
-      "epoch": 54.24,
-      "learning_rate": 3.835913312693499e-05,
-      "loss": 0.037,
-      "step": 922
-    },
-    {
-      "epoch": 54.29,
-      "learning_rate": 3.8343653250774e-05,
-      "loss": 0.0303,
-      "step": 923
-    },
-    {
-      "epoch": 54.35,
-      "learning_rate": 3.8328173374613e-05,
-      "loss": 0.0331,
-      "step": 924
-    },
-    {
-      "epoch": 54.41,
-      "learning_rate": 3.831269349845201e-05,
-      "loss": 0.0398,
-      "step": 925
-    },
-    {
-      "epoch": 54.47,
-      "learning_rate": 3.829721362229102e-05,
-      "loss": 0.0359,
-      "step": 926
-    },
-    {
-      "epoch": 54.53,
-      "learning_rate": 3.828173374613003e-05,
-      "loss": 0.0387,
-      "step": 927
-    },
-    {
-      "epoch": 54.59,
-      "learning_rate": 3.826625386996904e-05,
-      "loss": 0.0303,
-      "step": 928
-    },
-    {
-      "epoch": 54.65,
-      "learning_rate": 3.825077399380805e-05,
-      "loss": 0.0363,
-      "step": 929
-    },
-    {
-      "epoch": 54.71,
-      "learning_rate": 3.8235294117647055e-05,
-      "loss": 0.0398,
-      "step": 930
-    },
-    {
-      "epoch": 54.76,
-      "learning_rate": 3.8219814241486065e-05,
-      "loss": 0.0346,
-      "step": 931
-    },
-    {
-      "epoch": 54.82,
-      "learning_rate": 3.8204334365325075e-05,
-      "loss": 0.0339,
-      "step": 932
-    },
-    {
-      "epoch": 54.88,
-      "learning_rate": 3.8188854489164085e-05,
-      "loss": 0.0331,
-      "step": 933
-    },
-    {
-      "epoch": 54.94,
-      "learning_rate": 3.81733746130031e-05,
-      "loss": 0.0343,
-      "step": 934
-    },
-    {
-      "epoch": 55.0,
-      "learning_rate": 3.815789473684211e-05,
-      "loss": 0.0422,
-      "step": 935
-    },
-    {
-      "epoch": 55.06,
-      "learning_rate": 3.8142414860681115e-05,
-      "loss": 0.0307,
-      "step": 936
-    },
-    {
-      "epoch": 55.12,
-      "learning_rate": 3.8126934984520125e-05,
-      "loss": 0.0324,
-      "step": 937
-    },
-    {
-      "epoch": 55.18,
-      "learning_rate": 3.8111455108359135e-05,
-      "loss": 0.034,
-      "step": 938
-    },
-    {
-      "epoch": 55.24,
-      "learning_rate": 3.8095975232198145e-05,
-      "loss": 0.03,
-      "step": 939
-    },
-    {
-      "epoch": 55.29,
-      "learning_rate": 3.8080495356037155e-05,
-      "loss": 0.0387,
-      "step": 940
-    },
-    {
-      "epoch": 55.29,
-      "eval_accuracy_NAT": 0.9936041245514744,
-      "eval_accuracy_REST": 0.08292682926829269,
-      "eval_accuracy_concrete": 0.9422481574171889,
-      "eval_accuracy_concrete tiles": 0.8882394532170634,
-      "eval_accuracy_dark bricks": 0.8624552571636391,
-      "eval_accuracy_grass": 0.8795215493911971,
-      "eval_accuracy_red bricks": 0.8723088752196837,
-      "eval_accuracy_road": 0.9796440895446293,
-      "eval_accuracy_rough concrete": 0.623093928577813,
-      "eval_accuracy_rough red sidewalk": 0.8036532084013265,
-      "eval_accuracy_speedway bricks": 0.9909808804467558,
-      "eval_accuracy_steel": 0.8757065676430753,
-      "eval_accuracy_tiles": 0.2240601048198976,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9810823956322647,
-      "eval_iou_REST": 0.08279220779220779,
-      "eval_iou_concrete": 0.8819949505881023,
-      "eval_iou_concrete tiles": 0.827980139724944,
-      "eval_iou_dark bricks": 0.8400208806215713,
-      "eval_iou_grass": 0.826028110636755,
-      "eval_iou_red bricks": 0.8475001333973641,
-      "eval_iou_road": 0.9556556740770245,
-      "eval_iou_rough concrete": 0.612638634193212,
-      "eval_iou_rough red sidewalk": 0.6468049822667135,
-      "eval_iou_speedway bricks": 0.9754247112745985,
-      "eval_iou_steel": 0.8335117212771626,
-      "eval_iou_tiles": 0.22174916799088537,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07935668528079987,
-      "eval_mean_accuracy": 0.7706494635124644,
-      "eval_mean_iou": 0.7333218238056005,
-      "eval_overall_accuracy": 0.9777301206427106,
-      "eval_runtime": 5.3545,
-      "eval_samples_per_second": 22.038,
-      "eval_steps_per_second": 1.494,
-      "step": 940
-    },
-    {
-      "epoch": 55.35,
-      "learning_rate": 3.8065015479876166e-05,
-      "loss": 0.0383,
-      "step": 941
-    },
-    {
-      "epoch": 55.41,
-      "learning_rate": 3.804953560371517e-05,
-      "loss": 0.0361,
-      "step": 942
-    },
-    {
-      "epoch": 55.47,
-      "learning_rate": 3.803405572755418e-05,
-      "loss": 0.045,
-      "step": 943
-    },
-    {
-      "epoch": 55.53,
-      "learning_rate": 3.801857585139319e-05,
-      "loss": 0.0375,
-      "step": 944
-    },
-    {
-      "epoch": 55.59,
-      "learning_rate": 3.80030959752322e-05,
-      "loss": 0.0346,
-      "step": 945
-    },
-    {
-      "epoch": 55.65,
-      "learning_rate": 3.798761609907121e-05,
-      "loss": 0.0341,
-      "step": 946
-    },
-    {
-      "epoch": 55.71,
-      "learning_rate": 3.797213622291022e-05,
-      "loss": 0.0339,
-      "step": 947
-    },
-    {
-      "epoch": 55.76,
-      "learning_rate": 3.795665634674922e-05,
-      "loss": 0.0317,
-      "step": 948
-    },
-    {
-      "epoch": 55.82,
-      "learning_rate": 3.794117647058824e-05,
-      "loss": 0.034,
-      "step": 949
-    },
-    {
-      "epoch": 55.88,
-      "learning_rate": 3.792569659442725e-05,
-      "loss": 0.0385,
-      "step": 950
-    },
-    {
-      "epoch": 55.94,
-      "learning_rate": 3.791021671826626e-05,
-      "loss": 0.0381,
-      "step": 951
-    },
-    {
-      "epoch": 56.0,
-      "learning_rate": 3.789473684210527e-05,
-      "loss": 0.0362,
-      "step": 952
-    },
-    {
-      "epoch": 56.06,
-      "learning_rate": 3.787925696594428e-05,
-      "loss": 0.0383,
-      "step": 953
-    },
-    {
-      "epoch": 56.12,
-      "learning_rate": 3.786377708978328e-05,
-      "loss": 0.0335,
-      "step": 954
-    },
-    {
-      "epoch": 56.18,
-      "learning_rate": 3.784829721362229e-05,
-      "loss": 0.0355,
-      "step": 955
-    },
-    {
-      "epoch": 56.24,
-      "learning_rate": 3.78328173374613e-05,
-      "loss": 0.0321,
-      "step": 956
-    },
-    {
-      "epoch": 56.29,
-      "learning_rate": 3.781733746130031e-05,
-      "loss": 0.0383,
-      "step": 957
-    },
-    {
-      "epoch": 56.35,
-      "learning_rate": 3.780185758513932e-05,
-      "loss": 0.0369,
-      "step": 958
-    },
-    {
-      "epoch": 56.41,
-      "learning_rate": 3.778637770897833e-05,
-      "loss": 0.0379,
-      "step": 959
-    },
-    {
-      "epoch": 56.47,
-      "learning_rate": 3.7770897832817335e-05,
-      "loss": 0.0283,
-      "step": 960
-    },
-    {
-      "epoch": 56.47,
-      "eval_accuracy_NAT": 0.9940290570364307,
-      "eval_accuracy_REST": 0.1073170731707317,
-      "eval_accuracy_concrete": 0.947219423674036,
-      "eval_accuracy_concrete tiles": 0.8996464765496112,
-      "eval_accuracy_dark bricks": 0.8595505617977528,
-      "eval_accuracy_grass": 0.8739274508245429,
-      "eval_accuracy_red bricks": 0.9006480667838312,
-      "eval_accuracy_road": 0.9807049055659974,
-      "eval_accuracy_rough concrete": 0.6202932921778821,
-      "eval_accuracy_rough red sidewalk": 0.786327164439498,
-      "eval_accuracy_speedway bricks": 0.9863600190009686,
-      "eval_accuracy_steel": 0.8993787702349478,
-      "eval_accuracy_tiles": 0.22711987639734618,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9810132306957249,
-      "eval_iou_REST": 0.10696920583468396,
-      "eval_iou_concrete": 0.8835801707236987,
-      "eval_iou_concrete tiles": 0.8253762324857291,
-      "eval_iou_dark bricks": 0.8396454848872975,
-      "eval_iou_grass": 0.8187282816821502,
-      "eval_iou_red bricks": 0.858586387434555,
-      "eval_iou_road": 0.954342682108701,
-      "eval_iou_rough concrete": 0.6133852653660089,
-      "eval_iou_rough red sidewalk": 0.6249439693041435,
-      "eval_iou_speedway bricks": 0.9763386266792573,
-      "eval_iou_steel": 0.8405589277331571,
-      "eval_iou_tiles": 0.22450813044650075,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0812796875834465,
-      "eval_mean_accuracy": 0.775578625973352,
-      "eval_mean_iou": 0.7344597381062775,
-      "eval_overall_accuracy": 0.977639764042224,
-      "eval_runtime": 5.7,
-      "eval_samples_per_second": 20.702,
-      "eval_steps_per_second": 1.404,
-      "step": 960
-    },
-    {
-      "epoch": 56.53,
-      "learning_rate": 3.7755417956656345e-05,
-      "loss": 0.0333,
-      "step": 961
-    },
-    {
-      "epoch": 56.59,
-      "learning_rate": 3.7739938080495355e-05,
-      "loss": 0.0327,
-      "step": 962
-    },
-    {
-      "epoch": 56.65,
-      "learning_rate": 3.7724458204334365e-05,
-      "loss": 0.0384,
-      "step": 963
-    },
-    {
-      "epoch": 56.71,
-      "learning_rate": 3.7708978328173375e-05,
-      "loss": 0.032,
-      "step": 964
-    },
-    {
-      "epoch": 56.76,
-      "learning_rate": 3.7693498452012385e-05,
-      "loss": 0.0304,
-      "step": 965
-    },
-    {
-      "epoch": 56.82,
-      "learning_rate": 3.7678018575851395e-05,
-      "loss": 0.0387,
-      "step": 966
-    },
-    {
-      "epoch": 56.88,
-      "learning_rate": 3.7662538699690405e-05,
-      "loss": 0.0335,
-      "step": 967
-    },
-    {
-      "epoch": 56.94,
-      "learning_rate": 3.7647058823529415e-05,
-      "loss": 0.0353,
-      "step": 968
-    },
-    {
-      "epoch": 57.0,
-      "learning_rate": 3.7631578947368425e-05,
-      "loss": 0.0365,
-      "step": 969
-    },
-    {
-      "epoch": 57.06,
-      "learning_rate": 3.7616099071207435e-05,
-      "loss": 0.0321,
-      "step": 970
-    },
-    {
-      "epoch": 57.12,
-      "learning_rate": 3.7600619195046445e-05,
-      "loss": 0.0316,
-      "step": 971
-    },
-    {
-      "epoch": 57.18,
-      "learning_rate": 3.758513931888545e-05,
-      "loss": 0.0374,
-      "step": 972
-    },
-    {
-      "epoch": 57.24,
-      "learning_rate": 3.756965944272446e-05,
-      "loss": 0.0364,
-      "step": 973
-    },
-    {
-      "epoch": 57.29,
-      "learning_rate": 3.755417956656347e-05,
-      "loss": 0.0328,
-      "step": 974
-    },
-    {
-      "epoch": 57.35,
-      "learning_rate": 3.753869969040248e-05,
-      "loss": 0.0309,
-      "step": 975
-    },
-    {
-      "epoch": 57.41,
-      "learning_rate": 3.752321981424149e-05,
-      "loss": 0.0406,
-      "step": 976
-    },
-    {
-      "epoch": 57.47,
-      "learning_rate": 3.75077399380805e-05,
-      "loss": 0.0333,
-      "step": 977
-    },
-    {
-      "epoch": 57.53,
-      "learning_rate": 3.74922600619195e-05,
-      "loss": 0.0378,
-      "step": 978
-    },
-    {
-      "epoch": 57.59,
-      "learning_rate": 3.747678018575851e-05,
-      "loss": 0.0367,
-      "step": 979
-    },
-    {
-      "epoch": 57.65,
-      "learning_rate": 3.746130030959752e-05,
-      "loss": 0.0384,
-      "step": 980
-    },
-    {
-      "epoch": 57.65,
-      "eval_accuracy_NAT": 0.9925220616352655,
-      "eval_accuracy_REST": 0.15772357723577235,
-      "eval_accuracy_concrete": 0.9498726511942199,
-      "eval_accuracy_concrete tiles": 0.9201037002121141,
-      "eval_accuracy_dark bricks": 0.849343452711368,
-      "eval_accuracy_grass": 0.8902697506122094,
-      "eval_accuracy_red bricks": 0.8871375219683656,
-      "eval_accuracy_road": 0.9787938334791398,
-      "eval_accuracy_rough concrete": 0.6250186212526592,
-      "eval_accuracy_rough red sidewalk": 0.7860263650651607,
-      "eval_accuracy_speedway bricks": 0.9885589766895405,
-      "eval_accuracy_steel": 0.8913947432866703,
-      "eval_accuracy_tiles": 0.24911387803326365,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9816768379657402,
-      "eval_iou_REST": 0.15746753246753248,
-      "eval_iou_concrete": 0.8824866621097391,
-      "eval_iou_concrete tiles": 0.8382899229134011,
-      "eval_iou_dark bricks": 0.8279833739964453,
-      "eval_iou_grass": 0.8279874402019143,
-      "eval_iou_red bricks": 0.8572414159104177,
-      "eval_iou_road": 0.9501891537987437,
-      "eval_iou_rough concrete": 0.5993440299872006,
-      "eval_iou_rough red sidewalk": 0.6357272318116022,
-      "eval_iou_speedway bricks": 0.9760409479396596,
-      "eval_iou_steel": 0.8398661725441814,
-      "eval_iou_tiles": 0.24450655645089353,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07997491955757141,
-      "eval_mean_accuracy": 0.7819907025673654,
-      "eval_mean_iou": 0.7399082521613438,
-      "eval_overall_accuracy": 0.9777147648698192,
-      "eval_runtime": 7.695,
-      "eval_samples_per_second": 15.335,
-      "eval_steps_per_second": 1.04,
-      "step": 980
-    },
-    {
-      "epoch": 57.71,
-      "learning_rate": 3.744582043343653e-05,
-      "loss": 0.0299,
-      "step": 981
-    },
-    {
-      "epoch": 57.76,
-      "learning_rate": 3.743034055727554e-05,
-      "loss": 0.0298,
-      "step": 982
-    },
-    {
-      "epoch": 57.82,
-      "learning_rate": 3.741486068111456e-05,
-      "loss": 0.0384,
-      "step": 983
-    },
-    {
-      "epoch": 57.88,
-      "learning_rate": 3.739938080495356e-05,
-      "loss": 0.0324,
-      "step": 984
-    },
-    {
-      "epoch": 57.94,
-      "learning_rate": 3.738390092879257e-05,
-      "loss": 0.0359,
-      "step": 985
-    },
-    {
-      "epoch": 58.0,
-      "learning_rate": 3.736842105263158e-05,
-      "loss": 0.0425,
-      "step": 986
-    },
-    {
-      "epoch": 58.06,
-      "learning_rate": 3.735294117647059e-05,
-      "loss": 0.0325,
-      "step": 987
-    },
-    {
-      "epoch": 58.12,
-      "learning_rate": 3.73374613003096e-05,
-      "loss": 0.0305,
-      "step": 988
-    },
-    {
-      "epoch": 58.18,
-      "learning_rate": 3.732198142414861e-05,
-      "loss": 0.0395,
-      "step": 989
-    },
-    {
-      "epoch": 58.24,
-      "learning_rate": 3.7306501547987615e-05,
-      "loss": 0.0323,
-      "step": 990
-    },
-    {
-      "epoch": 58.29,
-      "learning_rate": 3.7291021671826625e-05,
-      "loss": 0.0319,
-      "step": 991
-    },
-    {
-      "epoch": 58.35,
-      "learning_rate": 3.7275541795665635e-05,
-      "loss": 0.0294,
-      "step": 992
-    },
-    {
-      "epoch": 58.41,
-      "learning_rate": 3.7260061919504645e-05,
-      "loss": 0.0308,
-      "step": 993
-    },
-    {
-      "epoch": 58.47,
-      "learning_rate": 3.7244582043343655e-05,
-      "loss": 0.0353,
-      "step": 994
-    },
-    {
-      "epoch": 58.53,
-      "learning_rate": 3.7229102167182665e-05,
-      "loss": 0.0312,
-      "step": 995
-    },
-    {
-      "epoch": 58.59,
-      "learning_rate": 3.721362229102167e-05,
-      "loss": 0.0331,
-      "step": 996
-    },
-    {
-      "epoch": 58.65,
-      "learning_rate": 3.7198142414860685e-05,
-      "loss": 0.034,
-      "step": 997
-    },
-    {
-      "epoch": 58.71,
-      "learning_rate": 3.7182662538699695e-05,
-      "loss": 0.0392,
-      "step": 998
-    },
-    {
-      "epoch": 58.76,
-      "learning_rate": 3.7167182662538705e-05,
-      "loss": 0.0345,
-      "step": 999
-    },
-    {
-      "epoch": 58.82,
-      "learning_rate": 3.7151702786377715e-05,
-      "loss": 0.0284,
-      "step": 1000
-    },
-    {
-      "epoch": 58.82,
-      "eval_accuracy_NAT": 0.9921813393179699,
-      "eval_accuracy_REST": 0.43414634146341463,
-      "eval_accuracy_concrete": 0.9423184820878477,
-      "eval_accuracy_concrete tiles": 0.8988215884987037,
-      "eval_accuracy_dark bricks": 0.8605794077675478,
-      "eval_accuracy_grass": 0.8817253138637507,
-      "eval_accuracy_red bricks": 0.890322934973638,
-      "eval_accuracy_road": 0.9849903885258718,
-      "eval_accuracy_rough concrete": 0.6222239436535791,
-      "eval_accuracy_rough red sidewalk": 0.7714300754254431,
-      "eval_accuracy_speedway bricks": 0.991181720410207,
-      "eval_accuracy_steel": 0.8787052632105035,
-      "eval_accuracy_tiles": 0.21878881517161985,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.981312315122162,
-      "eval_iou_REST": 0.4070121951219512,
-      "eval_iou_concrete": 0.8805584107693506,
-      "eval_iou_concrete tiles": 0.8373109096098535,
-      "eval_iou_dark bricks": 0.8309975001270731,
-      "eval_iou_grass": 0.8225141163396466,
-      "eval_iou_red bricks": 0.8639415902792582,
-      "eval_iou_road": 0.9464345316648095,
-      "eval_iou_rough concrete": 0.6075898545917922,
-      "eval_iou_rough red sidewalk": 0.6204277143410103,
-      "eval_iou_speedway bricks": 0.9753300572952552,
-      "eval_iou_steel": 0.8363440466365238,
-      "eval_iou_tiles": 0.21588497294711984,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08268687129020691,
-      "eval_mean_accuracy": 0.7974935087976996,
-      "eval_mean_iou": 0.7558198626804467,
-      "eval_overall_accuracy": 0.9772288435596531,
-      "eval_runtime": 5.8338,
-      "eval_samples_per_second": 20.227,
-      "eval_steps_per_second": 1.371,
-      "step": 1000
-    },
-    {
-      "epoch": 58.88,
-      "learning_rate": 3.7136222910216725e-05,
-      "loss": 0.0356,
-      "step": 1001
-    },
-    {
-      "epoch": 58.94,
-      "learning_rate": 3.712074303405573e-05,
-      "loss": 0.0352,
-      "step": 1002
-    },
-    {
-      "epoch": 59.0,
-      "learning_rate": 3.710526315789474e-05,
-      "loss": 0.0406,
-      "step": 1003
-    },
-    {
-      "epoch": 59.06,
-      "learning_rate": 3.708978328173375e-05,
-      "loss": 0.0397,
-      "step": 1004
-    },
-    {
-      "epoch": 59.12,
-      "learning_rate": 3.707430340557276e-05,
-      "loss": 0.0319,
-      "step": 1005
-    },
-    {
-      "epoch": 59.18,
-      "learning_rate": 3.705882352941177e-05,
-      "loss": 0.0337,
-      "step": 1006
-    },
-    {
-      "epoch": 59.24,
-      "learning_rate": 3.704334365325078e-05,
-      "loss": 0.0354,
-      "step": 1007
-    },
-    {
-      "epoch": 59.29,
-      "learning_rate": 3.702786377708978e-05,
-      "loss": 0.0315,
-      "step": 1008
-    },
-    {
-      "epoch": 59.35,
-      "learning_rate": 3.701238390092879e-05,
-      "loss": 0.0377,
-      "step": 1009
-    },
-    {
-      "epoch": 59.41,
-      "learning_rate": 3.69969040247678e-05,
-      "loss": 0.0327,
-      "step": 1010
-    },
-    {
-      "epoch": 59.47,
-      "learning_rate": 3.698142414860681e-05,
-      "loss": 0.0303,
-      "step": 1011
-    },
-    {
-      "epoch": 59.53,
-      "learning_rate": 3.696594427244582e-05,
-      "loss": 0.0369,
-      "step": 1012
-    },
-    {
-      "epoch": 59.59,
-      "learning_rate": 3.695046439628483e-05,
-      "loss": 0.0313,
-      "step": 1013
-    },
-    {
-      "epoch": 59.65,
-      "learning_rate": 3.693498452012384e-05,
-      "loss": 0.0363,
-      "step": 1014
-    },
-    {
-      "epoch": 59.71,
-      "learning_rate": 3.691950464396285e-05,
-      "loss": 0.0306,
-      "step": 1015
-    },
-    {
-      "epoch": 59.76,
-      "learning_rate": 3.690402476780186e-05,
-      "loss": 0.0338,
-      "step": 1016
-    },
-    {
-      "epoch": 59.82,
-      "learning_rate": 3.688854489164087e-05,
-      "loss": 0.0349,
-      "step": 1017
-    },
-    {
-      "epoch": 59.88,
-      "learning_rate": 3.687306501547988e-05,
-      "loss": 0.0354,
-      "step": 1018
-    },
-    {
-      "epoch": 59.94,
-      "learning_rate": 3.685758513931889e-05,
-      "loss": 0.0277,
-      "step": 1019
-    },
-    {
-      "epoch": 60.0,
-      "learning_rate": 3.6842105263157895e-05,
-      "loss": 0.0282,
-      "step": 1020
-    },
-    {
-      "epoch": 60.0,
-      "eval_accuracy_NAT": 0.9918096716509706,
-      "eval_accuracy_REST": 0.4016260162601626,
-      "eval_accuracy_concrete": 0.9463885224022296,
-      "eval_accuracy_concrete tiles": 0.9250765967475842,
-      "eval_accuracy_dark bricks": 0.8827402714239229,
-      "eval_accuracy_grass": 0.8949971045430287,
-      "eval_accuracy_red bricks": 0.8991102811950791,
-      "eval_accuracy_road": 0.983558153292992,
-      "eval_accuracy_rough concrete": 0.623308445408446,
-      "eval_accuracy_rough red sidewalk": 0.7761375856338218,
-      "eval_accuracy_speedway bricks": 0.9871097415607801,
-      "eval_accuracy_steel": 0.9167012349627912,
-      "eval_accuracy_tiles": 0.24320639825502136,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9814752227349574,
-      "eval_iou_REST": 0.39268680445151033,
-      "eval_iou_concrete": 0.8841362393150186,
-      "eval_iou_concrete tiles": 0.8398271177011789,
-      "eval_iou_dark bricks": 0.8436370953726825,
-      "eval_iou_grass": 0.8236017348736243,
-      "eval_iou_red bricks": 0.8689029244732233,
-      "eval_iou_road": 0.9497141984320003,
-      "eval_iou_rough concrete": 0.6083691985576364,
-      "eval_iou_rough red sidewalk": 0.6327748042695899,
-      "eval_iou_speedway bricks": 0.9759902446231568,
-      "eval_iou_steel": 0.8398063253322344,
-      "eval_iou_tiles": 0.23962033250753664,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08094064146280289,
-      "eval_mean_accuracy": 0.80552077102591,
-      "eval_mean_iou": 0.7600417109726423,
-      "eval_overall_accuracy": 0.9776828248622054,
-      "eval_runtime": 5.4926,
-      "eval_samples_per_second": 21.484,
-      "eval_steps_per_second": 1.457,
-      "step": 1020
-    },
-    {
-      "epoch": 60.06,
-      "learning_rate": 3.6826625386996905e-05,
-      "loss": 0.0361,
-      "step": 1021
-    },
-    {
-      "epoch": 60.12,
-      "learning_rate": 3.6811145510835915e-05,
-      "loss": 0.0357,
-      "step": 1022
-    },
-    {
-      "epoch": 60.18,
-      "learning_rate": 3.6795665634674925e-05,
-      "loss": 0.0345,
-      "step": 1023
-    },
-    {
-      "epoch": 60.24,
-      "learning_rate": 3.6780185758513935e-05,
-      "loss": 0.0378,
-      "step": 1024
-    },
-    {
-      "epoch": 60.29,
-      "learning_rate": 3.6764705882352945e-05,
-      "loss": 0.0332,
-      "step": 1025
-    },
-    {
-      "epoch": 60.35,
-      "learning_rate": 3.674922600619195e-05,
-      "loss": 0.0356,
-      "step": 1026
-    },
-    {
-      "epoch": 60.41,
-      "learning_rate": 3.673374613003096e-05,
-      "loss": 0.0365,
-      "step": 1027
-    },
-    {
-      "epoch": 60.47,
-      "learning_rate": 3.671826625386997e-05,
-      "loss": 0.0371,
-      "step": 1028
-    },
-    {
-      "epoch": 60.53,
-      "learning_rate": 3.670278637770898e-05,
-      "loss": 0.0294,
-      "step": 1029
-    },
-    {
-      "epoch": 60.59,
-      "learning_rate": 3.668730650154799e-05,
-      "loss": 0.033,
-      "step": 1030
-    },
-    {
-      "epoch": 60.65,
-      "learning_rate": 3.6671826625387e-05,
-      "loss": 0.0325,
-      "step": 1031
-    },
-    {
-      "epoch": 60.71,
-      "learning_rate": 3.665634674922601e-05,
-      "loss": 0.0361,
-      "step": 1032
-    },
-    {
-      "epoch": 60.76,
-      "learning_rate": 3.664086687306502e-05,
-      "loss": 0.0338,
-      "step": 1033
-    },
-    {
-      "epoch": 60.82,
-      "learning_rate": 3.662538699690403e-05,
-      "loss": 0.0291,
-      "step": 1034
-    },
-    {
-      "epoch": 60.88,
-      "learning_rate": 3.660990712074304e-05,
-      "loss": 0.0327,
-      "step": 1035
-    },
-    {
-      "epoch": 60.94,
-      "learning_rate": 3.659442724458205e-05,
-      "loss": 0.0334,
-      "step": 1036
-    },
-    {
-      "epoch": 61.0,
-      "learning_rate": 3.657894736842106e-05,
-      "loss": 0.033,
-      "step": 1037
-    },
-    {
-      "epoch": 61.06,
-      "learning_rate": 3.656346749226006e-05,
-      "loss": 0.033,
-      "step": 1038
-    },
-    {
-      "epoch": 61.12,
-      "learning_rate": 3.654798761609907e-05,
-      "loss": 0.0321,
-      "step": 1039
-    },
-    {
-      "epoch": 61.18,
-      "learning_rate": 3.653250773993808e-05,
-      "loss": 0.0327,
-      "step": 1040
-    },
-    {
-      "epoch": 61.18,
-      "eval_accuracy_NAT": 0.9918416413240966,
-      "eval_accuracy_REST": 0.4113821138211382,
-      "eval_accuracy_concrete": 0.9499039915365788,
-      "eval_accuracy_concrete tiles": 0.9102050436012256,
-      "eval_accuracy_dark bricks": 0.8241056218057922,
-      "eval_accuracy_grass": 0.9009176516897736,
-      "eval_accuracy_red bricks": 0.8756041300527241,
-      "eval_accuracy_road": 0.9789685875541965,
-      "eval_accuracy_rough concrete": 0.6215505991574256,
-      "eval_accuracy_rough red sidewalk": 0.7633310522714113,
-      "eval_accuracy_speedway bricks": 0.9883005580334957,
-      "eval_accuracy_steel": 0.906825530894061,
-      "eval_accuracy_tiles": 0.22954345784483018,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9817055662529158,
-      "eval_iou_REST": 0.4080645161290323,
-      "eval_iou_concrete": 0.8795901885339427,
-      "eval_iou_concrete tiles": 0.8338011140377392,
-      "eval_iou_dark bricks": 0.8083731147818005,
-      "eval_iou_grass": 0.8279318464824104,
-      "eval_iou_red bricks": 0.8530230069555912,
-      "eval_iou_road": 0.9510194461302683,
-      "eval_iou_rough concrete": 0.6061564030892429,
-      "eval_iou_rough red sidewalk": 0.6305957631856868,
-      "eval_iou_speedway bricks": 0.9756481291007124,
-      "eval_iou_steel": 0.8437646774336057,
-      "eval_iou_tiles": 0.22592957032531233,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08245375007390976,
-      "eval_mean_accuracy": 0.7963446138143653,
-      "eval_mean_iou": 0.75581564172602,
-      "eval_overall_accuracy": 0.9775115514205674,
-      "eval_runtime": 5.7704,
-      "eval_samples_per_second": 20.449,
-      "eval_steps_per_second": 1.386,
-      "step": 1040
-    },
-    {
-      "epoch": 61.24,
-      "learning_rate": 3.651702786377709e-05,
-      "loss": 0.0337,
-      "step": 1041
-    },
-    {
-      "epoch": 61.29,
-      "learning_rate": 3.65015479876161e-05,
-      "loss": 0.0332,
-      "step": 1042
-    },
-    {
-      "epoch": 61.35,
-      "learning_rate": 3.648606811145511e-05,
-      "loss": 0.0339,
-      "step": 1043
-    },
-    {
-      "epoch": 61.41,
-      "learning_rate": 3.6470588235294114e-05,
-      "loss": 0.0358,
-      "step": 1044
-    },
-    {
-      "epoch": 61.47,
-      "learning_rate": 3.6455108359133124e-05,
-      "loss": 0.0273,
-      "step": 1045
-    },
-    {
-      "epoch": 61.53,
-      "learning_rate": 3.643962848297214e-05,
-      "loss": 0.0359,
-      "step": 1046
-    },
-    {
-      "epoch": 61.59,
-      "learning_rate": 3.642414860681115e-05,
-      "loss": 0.0344,
-      "step": 1047
-    },
-    {
-      "epoch": 61.65,
-      "learning_rate": 3.640866873065016e-05,
-      "loss": 0.0306,
-      "step": 1048
-    },
-    {
-      "epoch": 61.71,
-      "learning_rate": 3.639318885448917e-05,
-      "loss": 0.0332,
-      "step": 1049
-    },
-    {
-      "epoch": 61.76,
-      "learning_rate": 3.6377708978328174e-05,
-      "loss": 0.0335,
-      "step": 1050
-    },
-    {
-      "epoch": 61.82,
-      "learning_rate": 3.6362229102167184e-05,
-      "loss": 0.0317,
-      "step": 1051
-    },
-    {
-      "epoch": 61.88,
-      "learning_rate": 3.6346749226006194e-05,
-      "loss": 0.0338,
-      "step": 1052
-    },
-    {
-      "epoch": 61.94,
-      "learning_rate": 3.6331269349845204e-05,
-      "loss": 0.0323,
-      "step": 1053
-    },
-    {
-      "epoch": 62.0,
-      "learning_rate": 3.6315789473684214e-05,
-      "loss": 0.034,
-      "step": 1054
-    },
-    {
-      "epoch": 62.06,
-      "learning_rate": 3.630030959752322e-05,
-      "loss": 0.032,
-      "step": 1055
-    },
-    {
-      "epoch": 62.12,
-      "learning_rate": 3.628482972136223e-05,
-      "loss": 0.0381,
-      "step": 1056
-    },
-    {
-      "epoch": 62.18,
-      "learning_rate": 3.626934984520124e-05,
-      "loss": 0.0295,
-      "step": 1057
-    },
-    {
-      "epoch": 62.24,
-      "learning_rate": 3.625386996904025e-05,
-      "loss": 0.0362,
-      "step": 1058
-    },
-    {
-      "epoch": 62.29,
-      "learning_rate": 3.623839009287926e-05,
-      "loss": 0.0334,
-      "step": 1059
-    },
-    {
-      "epoch": 62.35,
-      "learning_rate": 3.622291021671827e-05,
-      "loss": 0.033,
-      "step": 1060
-    },
-    {
-      "epoch": 62.35,
-      "eval_accuracy_NAT": 0.9923818910616945,
-      "eval_accuracy_REST": 0.3024390243902439,
-      "eval_accuracy_concrete": 0.9527479365059952,
-      "eval_accuracy_concrete tiles": 0.8912326184303558,
-      "eval_accuracy_dark bricks": 0.8402082575656069,
-      "eval_accuracy_grass": 0.884502492363942,
-      "eval_accuracy_red bricks": 0.9115224077328646,
-      "eval_accuracy_road": 0.9812949009753629,
-      "eval_accuracy_rough concrete": 0.6220332620263498,
-      "eval_accuracy_rough red sidewalk": 0.7280021657554953,
-      "eval_accuracy_speedway bricks": 0.9887152617122944,
-      "eval_accuracy_steel": 0.9029147320915402,
-      "eval_accuracy_tiles": 0.22975552122148504,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.981627653224521,
-      "eval_iou_REST": 0.2995169082125604,
-      "eval_iou_concrete": 0.8805582323302258,
-      "eval_iou_concrete tiles": 0.8412680756395996,
-      "eval_iou_dark bricks": 0.8209007560066764,
-      "eval_iou_grass": 0.8232177077315448,
-      "eval_iou_red bricks": 0.870502465121158,
-      "eval_iou_road": 0.9529414451163437,
-      "eval_iou_rough concrete": 0.6107477182307512,
-      "eval_iou_rough red sidewalk": 0.6131847807498148,
-      "eval_iou_speedway bricks": 0.9762220118128542,
-      "eval_iou_steel": 0.8434578402971173,
-      "eval_iou_tiles": 0.2261652700325053,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08240743726491928,
-      "eval_mean_accuracy": 0.7867500362948638,
-      "eval_mean_iou": 0.7492546818850516,
-      "eval_overall_accuracy": 0.9776363049523306,
-      "eval_runtime": 8.754,
-      "eval_samples_per_second": 13.48,
-      "eval_steps_per_second": 0.914,
-      "step": 1060
-    },
-    {
-      "epoch": 62.41,
-      "learning_rate": 3.620743034055728e-05,
-      "loss": 0.0358,
-      "step": 1061
-    },
-    {
-      "epoch": 62.47,
-      "learning_rate": 3.619195046439629e-05,
-      "loss": 0.0284,
-      "step": 1062
-    },
-    {
-      "epoch": 62.53,
-      "learning_rate": 3.61764705882353e-05,
-      "loss": 0.0284,
-      "step": 1063
-    },
-    {
-      "epoch": 62.59,
-      "learning_rate": 3.616099071207431e-05,
-      "loss": 0.032,
-      "step": 1064
-    },
-    {
-      "epoch": 62.65,
-      "learning_rate": 3.614551083591332e-05,
-      "loss": 0.0395,
-      "step": 1065
-    },
-    {
-      "epoch": 62.71,
-      "learning_rate": 3.613003095975233e-05,
-      "loss": 0.0343,
-      "step": 1066
-    },
-    {
-      "epoch": 62.76,
-      "learning_rate": 3.611455108359133e-05,
-      "loss": 0.0366,
-      "step": 1067
-    },
-    {
-      "epoch": 62.82,
-      "learning_rate": 3.609907120743034e-05,
-      "loss": 0.0325,
-      "step": 1068
-    },
-    {
-      "epoch": 62.88,
-      "learning_rate": 3.608359133126935e-05,
-      "loss": 0.0324,
-      "step": 1069
-    },
-    {
-      "epoch": 62.94,
-      "learning_rate": 3.606811145510836e-05,
-      "loss": 0.0373,
-      "step": 1070
-    },
-    {
-      "epoch": 63.0,
-      "learning_rate": 3.605263157894737e-05,
-      "loss": 0.0282,
-      "step": 1071
-    },
-    {
-      "epoch": 63.06,
-      "learning_rate": 3.603715170278638e-05,
-      "loss": 0.0337,
-      "step": 1072
-    },
-    {
-      "epoch": 63.12,
-      "learning_rate": 3.6021671826625384e-05,
-      "loss": 0.0357,
-      "step": 1073
-    },
-    {
-      "epoch": 63.18,
-      "learning_rate": 3.6006191950464394e-05,
-      "loss": 0.0292,
-      "step": 1074
-    },
-    {
-      "epoch": 63.24,
-      "learning_rate": 3.5990712074303404e-05,
-      "loss": 0.0352,
-      "step": 1075
-    },
-    {
-      "epoch": 63.29,
-      "learning_rate": 3.5975232198142414e-05,
-      "loss": 0.0284,
-      "step": 1076
-    },
-    {
-      "epoch": 63.35,
-      "learning_rate": 3.5959752321981424e-05,
-      "loss": 0.0357,
-      "step": 1077
-    },
-    {
-      "epoch": 63.41,
-      "learning_rate": 3.5944272445820434e-05,
-      "loss": 0.0294,
-      "step": 1078
-    },
-    {
-      "epoch": 63.47,
-      "learning_rate": 3.5928792569659444e-05,
-      "loss": 0.0306,
-      "step": 1079
-    },
-    {
-      "epoch": 63.53,
-      "learning_rate": 3.5913312693498454e-05,
-      "loss": 0.0368,
-      "step": 1080
-    },
-    {
-      "epoch": 63.53,
-      "eval_accuracy_NAT": 0.9934459474503754,
-      "eval_accuracy_REST": 0.11056910569105691,
-      "eval_accuracy_concrete": 0.9446781805478903,
-      "eval_accuracy_concrete tiles": 0.8895592740985152,
-      "eval_accuracy_dark bricks": 0.8378969431311372,
-      "eval_accuracy_grass": 0.8800883398244558,
-      "eval_accuracy_red bricks": 0.9107535149384886,
-      "eval_accuracy_road": 0.9836361780482161,
-      "eval_accuracy_rough concrete": 0.6245478759854367,
-      "eval_accuracy_rough red sidewalk": 0.7779198219267703,
-      "eval_accuracy_speedway bricks": 0.99053790151713,
-      "eval_accuracy_steel": 0.8842428543583541,
-      "eval_accuracy_tiles": 0.21472931624708413,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9817367145617313,
-      "eval_iou_REST": 0.11056910569105691,
-      "eval_iou_concrete": 0.8837308416053454,
-      "eval_iou_concrete tiles": 0.837917637917638,
-      "eval_iou_dark bricks": 0.8200726884419757,
-      "eval_iou_grass": 0.8245979860738014,
-      "eval_iou_red bricks": 0.8740775880244571,
-      "eval_iou_road": 0.9521563154077513,
-      "eval_iou_rough concrete": 0.6120636292498336,
-      "eval_iou_rough red sidewalk": 0.6341772927905837,
-      "eval_iou_speedway bricks": 0.9760166340414632,
-      "eval_iou_steel": 0.8398827466710973,
-      "eval_iou_tiles": 0.21284646106723523,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08029883354902267,
-      "eval_mean_accuracy": 0.7725080964434545,
-      "eval_mean_iou": 0.7353727416572284,
-      "eval_overall_accuracy": 0.9778749174990896,
-      "eval_runtime": 6.9943,
-      "eval_samples_per_second": 16.871,
-      "eval_steps_per_second": 1.144,
-      "step": 1080
-    },
-    {
-      "epoch": 63.59,
-      "learning_rate": 3.5897832817337464e-05,
-      "loss": 0.0323,
-      "step": 1081
-    },
-    {
-      "epoch": 63.65,
-      "learning_rate": 3.5882352941176474e-05,
-      "loss": 0.0312,
-      "step": 1082
-    },
-    {
-      "epoch": 63.71,
-      "learning_rate": 3.5866873065015484e-05,
-      "loss": 0.041,
-      "step": 1083
-    },
-    {
-      "epoch": 63.76,
-      "learning_rate": 3.5851393188854494e-05,
-      "loss": 0.0355,
-      "step": 1084
-    },
-    {
-      "epoch": 63.82,
-      "learning_rate": 3.58359133126935e-05,
-      "loss": 0.0352,
-      "step": 1085
-    },
-    {
-      "epoch": 63.88,
-      "learning_rate": 3.582043343653251e-05,
-      "loss": 0.0306,
-      "step": 1086
-    },
-    {
-      "epoch": 63.94,
-      "learning_rate": 3.580495356037152e-05,
-      "loss": 0.032,
-      "step": 1087
-    },
-    {
-      "epoch": 64.0,
-      "learning_rate": 3.578947368421053e-05,
-      "loss": 0.0325,
-      "step": 1088
-    },
-    {
-      "epoch": 64.06,
-      "learning_rate": 3.577399380804954e-05,
-      "loss": 0.0285,
-      "step": 1089
-    },
-    {
-      "epoch": 64.12,
-      "learning_rate": 3.575851393188855e-05,
-      "loss": 0.0371,
-      "step": 1090
-    },
-    {
-      "epoch": 64.18,
-      "learning_rate": 3.574303405572755e-05,
-      "loss": 0.0324,
-      "step": 1091
-    },
-    {
-      "epoch": 64.24,
-      "learning_rate": 3.572755417956656e-05,
-      "loss": 0.0266,
-      "step": 1092
-    },
-    {
-      "epoch": 64.29,
-      "learning_rate": 3.571207430340557e-05,
-      "loss": 0.0286,
-      "step": 1093
-    },
-    {
-      "epoch": 64.35,
-      "learning_rate": 3.569659442724458e-05,
-      "loss": 0.0333,
-      "step": 1094
-    },
-    {
-      "epoch": 64.41,
-      "learning_rate": 3.56811145510836e-05,
-      "loss": 0.0316,
-      "step": 1095
-    },
-    {
-      "epoch": 64.47,
-      "learning_rate": 3.566563467492261e-05,
-      "loss": 0.0336,
-      "step": 1096
-    },
-    {
-      "epoch": 64.53,
-      "learning_rate": 3.565015479876161e-05,
-      "loss": 0.0296,
-      "step": 1097
-    },
-    {
-      "epoch": 64.59,
-      "learning_rate": 3.563467492260062e-05,
-      "loss": 0.0284,
-      "step": 1098
-    },
-    {
-      "epoch": 64.65,
-      "learning_rate": 3.561919504643963e-05,
-      "loss": 0.032,
-      "step": 1099
-    },
-    {
-      "epoch": 64.71,
-      "learning_rate": 3.560371517027864e-05,
-      "loss": 0.0395,
-      "step": 1100
-    },
-    {
-      "epoch": 64.71,
-      "eval_accuracy_NAT": 0.9933450785491634,
-      "eval_accuracy_REST": 0.055284552845528454,
-      "eval_accuracy_concrete": 0.9391015106045016,
-      "eval_accuracy_concrete tiles": 0.9200565637520622,
-      "eval_accuracy_dark bricks": 0.8680636640315449,
-      "eval_accuracy_grass": 0.8875058193007755,
-      "eval_accuracy_red bricks": 0.9199802284710018,
-      "eval_accuracy_road": 0.9820190348337778,
-      "eval_accuracy_rough concrete": 0.6581912655897127,
-      "eval_accuracy_rough red sidewalk": 0.7844246083968145,
-      "eval_accuracy_speedway bricks": 0.9909495206231111,
-      "eval_accuracy_steel": 0.8916371378450374,
-      "eval_accuracy_tiles": 0.23357266200127239,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9816389172822669,
-      "eval_iou_REST": 0.055284552845528454,
-      "eval_iou_concrete": 0.8853906696483588,
-      "eval_iou_concrete tiles": 0.8478596094954717,
-      "eval_iou_dark bricks": 0.8406523129396717,
-      "eval_iou_grass": 0.8273295245555029,
-      "eval_iou_red bricks": 0.8751828631138976,
-      "eval_iou_road": 0.9525531288424973,
-      "eval_iou_rough concrete": 0.644120477009651,
-      "eval_iou_rough red sidewalk": 0.6287756861184952,
-      "eval_iou_speedway bricks": 0.9760494212665044,
-      "eval_iou_steel": 0.8417931011704087,
-      "eval_iou_tiles": 0.23059668012561688,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.0765690729022026,
-      "eval_mean_accuracy": 0.7787793574495617,
-      "eval_mean_iou": 0.7374789957241441,
-      "eval_overall_accuracy": 0.9781521942655919,
-      "eval_runtime": 5.6286,
-      "eval_samples_per_second": 20.964,
-      "eval_steps_per_second": 1.421,
-      "step": 1100
-    },
-    {
-      "epoch": 64.76,
-      "learning_rate": 3.558823529411765e-05,
-      "loss": 0.0287,
-      "step": 1101
-    },
-    {
-      "epoch": 64.82,
-      "learning_rate": 3.557275541795666e-05,
-      "loss": 0.036,
-      "step": 1102
-    },
-    {
-      "epoch": 64.88,
-      "learning_rate": 3.5557275541795664e-05,
-      "loss": 0.0326,
-      "step": 1103
-    },
-    {
-      "epoch": 64.94,
-      "learning_rate": 3.5541795665634674e-05,
-      "loss": 0.0356,
-      "step": 1104
-    },
-    {
-      "epoch": 65.0,
-      "learning_rate": 3.5526315789473684e-05,
-      "loss": 0.0336,
-      "step": 1105
-    },
-    {
-      "epoch": 65.06,
-      "learning_rate": 3.5510835913312694e-05,
-      "loss": 0.0334,
-      "step": 1106
-    },
-    {
-      "epoch": 65.12,
-      "learning_rate": 3.5495356037151704e-05,
-      "loss": 0.031,
-      "step": 1107
-    },
-    {
-      "epoch": 65.18,
-      "learning_rate": 3.5479876160990714e-05,
-      "loss": 0.0319,
-      "step": 1108
-    },
-    {
-      "epoch": 65.24,
-      "learning_rate": 3.5464396284829724e-05,
-      "loss": 0.029,
-      "step": 1109
-    },
-    {
-      "epoch": 65.29,
-      "learning_rate": 3.5448916408668734e-05,
-      "loss": 0.0344,
-      "step": 1110
-    },
-    {
-      "epoch": 65.35,
-      "learning_rate": 3.5433436532507744e-05,
-      "loss": 0.0335,
-      "step": 1111
-    },
-    {
-      "epoch": 65.41,
-      "learning_rate": 3.5417956656346754e-05,
-      "loss": 0.0339,
-      "step": 1112
-    },
-    {
-      "epoch": 65.47,
-      "learning_rate": 3.5402476780185764e-05,
-      "loss": 0.0353,
-      "step": 1113
-    },
-    {
-      "epoch": 65.53,
-      "learning_rate": 3.5386996904024774e-05,
-      "loss": 0.0325,
-      "step": 1114
-    },
-    {
-      "epoch": 65.59,
-      "learning_rate": 3.537151702786378e-05,
-      "loss": 0.0299,
-      "step": 1115
-    },
-    {
-      "epoch": 65.65,
-      "learning_rate": 3.535603715170279e-05,
-      "loss": 0.0298,
-      "step": 1116
-    },
-    {
-      "epoch": 65.71,
-      "learning_rate": 3.53405572755418e-05,
-      "loss": 0.0343,
-      "step": 1117
-    },
-    {
-      "epoch": 65.76,
-      "learning_rate": 3.532507739938081e-05,
-      "loss": 0.0285,
-      "step": 1118
-    },
-    {
-      "epoch": 65.82,
-      "learning_rate": 3.530959752321982e-05,
-      "loss": 0.0274,
-      "step": 1119
-    },
-    {
-      "epoch": 65.88,
-      "learning_rate": 3.529411764705883e-05,
-      "loss": 0.0355,
-      "step": 1120
-    },
-    {
-      "epoch": 65.88,
-      "eval_accuracy_NAT": 0.9935288098240596,
-      "eval_accuracy_REST": 0.15772357723577235,
-      "eval_accuracy_concrete": 0.947173941957686,
-      "eval_accuracy_concrete tiles": 0.9117841150129625,
-      "eval_accuracy_dark bricks": 0.8493243113909997,
-      "eval_accuracy_grass": 0.8630335305272002,
-      "eval_accuracy_red bricks": 0.9045474516695958,
-      "eval_accuracy_road": 0.9832118516396688,
-      "eval_accuracy_rough concrete": 0.6616652464857972,
-      "eval_accuracy_rough red sidewalk": 0.7929823505967107,
-      "eval_accuracy_speedway bricks": 0.9890675884741167,
-      "eval_accuracy_steel": 0.8980943289668994,
-      "eval_accuracy_tiles": 0.23039171135144962,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9811323848611445,
-      "eval_iou_REST": 0.15746753246753248,
-      "eval_iou_concrete": 0.8863057884077346,
-      "eval_iou_concrete tiles": 0.8400356103704347,
-      "eval_iou_dark bricks": 0.8323758248251864,
-      "eval_iou_grass": 0.815390612009176,
-      "eval_iou_red bricks": 0.8783531544984268,
-      "eval_iou_road": 0.9513252874989658,
-      "eval_iou_rough concrete": 0.6450675915114125,
-      "eval_iou_rough red sidewalk": 0.62166910342876,
-      "eval_iou_speedway bricks": 0.9768907139886711,
-      "eval_iou_steel": 0.8456050859501099,
-      "eval_iou_tiles": 0.22557394554191137,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07918473333120346,
-      "eval_mean_accuracy": 0.7832714473179168,
-      "eval_mean_iou": 0.7428609719507281,
-      "eval_overall_accuracy": 0.9778169858253608,
-      "eval_runtime": 5.5719,
-      "eval_samples_per_second": 21.178,
-      "eval_steps_per_second": 1.436,
-      "step": 1120
-    },
-    {
-      "epoch": 65.94,
-      "learning_rate": 3.527863777089783e-05,
-      "loss": 0.0304,
-      "step": 1121
-    },
-    {
-      "epoch": 66.0,
-      "learning_rate": 3.526315789473684e-05,
-      "loss": 0.0325,
-      "step": 1122
-    },
-    {
-      "epoch": 66.06,
-      "learning_rate": 3.524767801857585e-05,
-      "loss": 0.0322,
-      "step": 1123
-    },
-    {
-      "epoch": 66.12,
-      "learning_rate": 3.523219814241486e-05,
-      "loss": 0.0287,
-      "step": 1124
-    },
-    {
-      "epoch": 66.18,
-      "learning_rate": 3.521671826625387e-05,
-      "loss": 0.0385,
-      "step": 1125
-    },
-    {
-      "epoch": 66.24,
-      "learning_rate": 3.520123839009288e-05,
-      "loss": 0.0318,
-      "step": 1126
-    },
-    {
-      "epoch": 66.29,
-      "learning_rate": 3.518575851393189e-05,
-      "loss": 0.026,
-      "step": 1127
-    },
-    {
-      "epoch": 66.35,
-      "learning_rate": 3.51702786377709e-05,
-      "loss": 0.0298,
-      "step": 1128
-    },
-    {
-      "epoch": 66.41,
-      "learning_rate": 3.515479876160991e-05,
-      "loss": 0.0311,
-      "step": 1129
-    },
-    {
-      "epoch": 66.47,
-      "learning_rate": 3.513931888544892e-05,
-      "loss": 0.0336,
-      "step": 1130
-    },
-    {
-      "epoch": 66.53,
-      "learning_rate": 3.512383900928793e-05,
-      "loss": 0.0332,
-      "step": 1131
-    },
-    {
-      "epoch": 66.59,
-      "learning_rate": 3.510835913312694e-05,
-      "loss": 0.0305,
-      "step": 1132
-    },
-    {
-      "epoch": 66.65,
-      "learning_rate": 3.5092879256965944e-05,
-      "loss": 0.0365,
-      "step": 1133
-    },
-    {
-      "epoch": 66.71,
-      "learning_rate": 3.5077399380804954e-05,
-      "loss": 0.0289,
-      "step": 1134
-    },
-    {
-      "epoch": 66.76,
-      "learning_rate": 3.5061919504643964e-05,
-      "loss": 0.0302,
-      "step": 1135
-    },
-    {
-      "epoch": 66.82,
-      "learning_rate": 3.5046439628482974e-05,
-      "loss": 0.0345,
-      "step": 1136
-    },
-    {
-      "epoch": 66.88,
-      "learning_rate": 3.5030959752321984e-05,
-      "loss": 0.0273,
-      "step": 1137
-    },
-    {
-      "epoch": 66.94,
-      "learning_rate": 3.5015479876160994e-05,
-      "loss": 0.0344,
-      "step": 1138
-    },
-    {
-      "epoch": 67.0,
-      "learning_rate": 3.5e-05,
-      "loss": 0.0459,
-      "step": 1139
-    },
-    {
-      "epoch": 67.06,
-      "learning_rate": 3.498452012383901e-05,
-      "loss": 0.0326,
-      "step": 1140
-    },
-    {
-      "epoch": 67.06,
-      "eval_accuracy_NAT": 0.9919153386987394,
-      "eval_accuracy_REST": 0.191869918699187,
-      "eval_accuracy_concrete": 0.9546646659807463,
-      "eval_accuracy_concrete tiles": 0.8761253829837379,
-      "eval_accuracy_dark bricks": 0.8499320483126926,
-      "eval_accuracy_grass": 0.8893254152993676,
-      "eval_accuracy_red bricks": 0.9099297012302284,
-      "eval_accuracy_road": 0.9822333357025782,
-      "eval_accuracy_rough concrete": 0.6300418903699819,
-      "eval_accuracy_rough red sidewalk": 0.7587739417502012,
-      "eval_accuracy_speedway bricks": 0.9882316692405713,
-      "eval_accuracy_steel": 0.902150064721846,
-      "eval_accuracy_tiles": 0.1801023963161562,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9819518422256411,
-      "eval_iou_REST": 0.191869918699187,
-      "eval_iou_concrete": 0.8805101971033416,
-      "eval_iou_concrete tiles": 0.8212707670554966,
-      "eval_iou_dark bricks": 0.8274994525640966,
-      "eval_iou_grass": 0.8292444489932045,
-      "eval_iou_red bricks": 0.8830615073019934,
-      "eval_iou_road": 0.9491662320090478,
-      "eval_iou_rough concrete": 0.6198658646687069,
-      "eval_iou_rough red sidewalk": 0.6252889376390464,
-      "eval_iou_speedway bricks": 0.9767151019960202,
-      "eval_iou_steel": 0.8457860285538912,
-      "eval_iou_tiles": 0.17682926829268292,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07983653992414474,
-      "eval_mean_accuracy": 0.7773304437927717,
-      "eval_mean_iou": 0.7391584282386428,
-      "eval_overall_accuracy": 0.9778036667128741,
-      "eval_runtime": 5.6682,
-      "eval_samples_per_second": 20.818,
-      "eval_steps_per_second": 1.411,
-      "step": 1140
-    },
-    {
-      "epoch": 67.12,
-      "learning_rate": 3.496904024767802e-05,
-      "loss": 0.0302,
-      "step": 1141
-    },
-    {
-      "epoch": 67.18,
-      "learning_rate": 3.495356037151703e-05,
-      "loss": 0.0326,
-      "step": 1142
-    },
-    {
-      "epoch": 67.24,
-      "learning_rate": 3.493808049535604e-05,
-      "loss": 0.0379,
-      "step": 1143
-    },
-    {
-      "epoch": 67.29,
-      "learning_rate": 3.4922600619195054e-05,
-      "loss": 0.0328,
-      "step": 1144
-    },
-    {
-      "epoch": 67.35,
-      "learning_rate": 3.490712074303406e-05,
-      "loss": 0.0247,
-      "step": 1145
-    },
-    {
-      "epoch": 67.41,
-      "learning_rate": 3.489164086687307e-05,
-      "loss": 0.0304,
-      "step": 1146
-    },
-    {
-      "epoch": 67.47,
-      "learning_rate": 3.487616099071208e-05,
-      "loss": 0.033,
-      "step": 1147
-    },
-    {
-      "epoch": 67.53,
-      "learning_rate": 3.486068111455109e-05,
-      "loss": 0.0538,
-      "step": 1148
-    },
-    {
-      "epoch": 67.59,
-      "learning_rate": 3.48452012383901e-05,
-      "loss": 0.0312,
-      "step": 1149
-    },
-    {
-      "epoch": 67.65,
-      "learning_rate": 3.482972136222911e-05,
-      "loss": 0.0284,
-      "step": 1150
-    },
-    {
-      "epoch": 67.71,
-      "learning_rate": 3.481424148606811e-05,
-      "loss": 0.034,
-      "step": 1151
-    },
-    {
-      "epoch": 67.76,
-      "learning_rate": 3.479876160990712e-05,
-      "loss": 0.031,
-      "step": 1152
-    },
-    {
-      "epoch": 67.82,
-      "learning_rate": 3.478328173374613e-05,
-      "loss": 0.03,
-      "step": 1153
-    },
-    {
-      "epoch": 67.88,
-      "learning_rate": 3.476780185758514e-05,
-      "loss": 0.031,
-      "step": 1154
-    },
-    {
-      "epoch": 67.94,
-      "learning_rate": 3.475232198142415e-05,
-      "loss": 0.0372,
-      "step": 1155
-    },
-    {
-      "epoch": 68.0,
-      "learning_rate": 3.473684210526316e-05,
-      "loss": 0.0288,
-      "step": 1156
-    },
-    {
-      "epoch": 68.06,
-      "learning_rate": 3.4721362229102163e-05,
-      "loss": 0.0307,
-      "step": 1157
-    },
-    {
-      "epoch": 68.12,
-      "learning_rate": 3.470588235294118e-05,
-      "loss": 0.0331,
-      "step": 1158
-    },
-    {
-      "epoch": 68.18,
-      "learning_rate": 3.469040247678019e-05,
-      "loss": 0.032,
-      "step": 1159
-    },
-    {
-      "epoch": 68.24,
-      "learning_rate": 3.46749226006192e-05,
-      "loss": 0.03,
-      "step": 1160
-    },
-    {
-      "epoch": 68.24,
-      "eval_accuracy_NAT": 0.9934777014764652,
-      "eval_accuracy_REST": 0.3056910569105691,
-      "eval_accuracy_concrete": 0.9438457504788957,
-      "eval_accuracy_concrete tiles": 0.9096629743106293,
-      "eval_accuracy_dark bricks": 0.8682311505847673,
-      "eval_accuracy_grass": 0.8853427426222621,
-      "eval_accuracy_red bricks": 0.9111928822495606,
-      "eval_accuracy_road": 0.9814338491695975,
-      "eval_accuracy_rough concrete": 0.7519768321822916,
-      "eval_accuracy_rough red sidewalk": 0.7964265034328729,
-      "eval_accuracy_speedway bricks": 0.9891467591764328,
-      "eval_accuracy_steel": 0.9062058004767926,
-      "eval_accuracy_tiles": 0.22775606652731073,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9820215251429533,
-      "eval_iou_REST": 0.29984051036682613,
-      "eval_iou_concrete": 0.8899700377609263,
-      "eval_iou_concrete tiles": 0.8444809101848813,
-      "eval_iou_dark bricks": 0.8442967761149579,
-      "eval_iou_grass": 0.828061638998384,
-      "eval_iou_red bricks": 0.8802992518703242,
-      "eval_iou_road": 0.9556096261359331,
-      "eval_iou_rough concrete": 0.738022831477496,
-      "eval_iou_rough red sidewalk": 0.6150754702735979,
-      "eval_iou_speedway bricks": 0.976805894926519,
-      "eval_iou_steel": 0.8482410179640718,
-      "eval_iou_tiles": 0.22415026833631485,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07349683344364166,
-      "eval_mean_accuracy": 0.8054146207383421,
-      "eval_mean_iou": 0.7636058276579374,
-      "eval_overall_accuracy": 0.9789129677465407,
-      "eval_runtime": 5.4514,
-      "eval_samples_per_second": 21.646,
-      "eval_steps_per_second": 1.468,
-      "step": 1160
-    },
-    {
-      "epoch": 68.29,
-      "learning_rate": 3.465944272445821e-05,
-      "loss": 0.032,
-      "step": 1161
-    },
-    {
-      "epoch": 68.35,
-      "learning_rate": 3.464396284829722e-05,
-      "loss": 0.0315,
-      "step": 1162
-    },
-    {
-      "epoch": 68.41,
-      "learning_rate": 3.4628482972136223e-05,
-      "loss": 0.0397,
-      "step": 1163
-    },
-    {
-      "epoch": 68.47,
-      "learning_rate": 3.4613003095975233e-05,
-      "loss": 0.0333,
-      "step": 1164
-    },
-    {
-      "epoch": 68.53,
-      "learning_rate": 3.4597523219814243e-05,
-      "loss": 0.033,
-      "step": 1165
-    },
-    {
-      "epoch": 68.59,
-      "learning_rate": 3.4582043343653253e-05,
-      "loss": 0.0376,
-      "step": 1166
-    },
-    {
-      "epoch": 68.65,
-      "learning_rate": 3.4566563467492263e-05,
-      "loss": 0.0414,
-      "step": 1167
-    },
-    {
-      "epoch": 68.71,
-      "learning_rate": 3.4551083591331273e-05,
-      "loss": 0.033,
-      "step": 1168
-    },
-    {
-      "epoch": 68.76,
-      "learning_rate": 3.453560371517028e-05,
-      "loss": 0.0398,
-      "step": 1169
-    },
-    {
-      "epoch": 68.82,
-      "learning_rate": 3.452012383900929e-05,
-      "loss": 0.0334,
-      "step": 1170
-    },
-    {
-      "epoch": 68.88,
-      "learning_rate": 3.45046439628483e-05,
-      "loss": 0.0396,
-      "step": 1171
-    },
-    {
-      "epoch": 68.94,
-      "learning_rate": 3.448916408668731e-05,
-      "loss": 0.0335,
-      "step": 1172
-    },
-    {
-      "epoch": 69.0,
-      "learning_rate": 3.447368421052632e-05,
-      "loss": 0.0326,
-      "step": 1173
-    },
-    {
-      "epoch": 69.06,
-      "learning_rate": 3.445820433436533e-05,
-      "loss": 0.0343,
-      "step": 1174
-    },
-    {
-      "epoch": 69.12,
-      "learning_rate": 3.444272445820434e-05,
-      "loss": 0.0459,
-      "step": 1175
-    },
-    {
-      "epoch": 69.18,
-      "learning_rate": 3.442724458204335e-05,
-      "loss": 0.0314,
-      "step": 1176
-    },
-    {
-      "epoch": 69.24,
-      "learning_rate": 3.441176470588236e-05,
-      "loss": 0.0305,
-      "step": 1177
-    },
-    {
-      "epoch": 69.29,
-      "learning_rate": 3.439628482972137e-05,
-      "loss": 0.0354,
-      "step": 1178
-    },
-    {
-      "epoch": 69.35,
-      "learning_rate": 3.438080495356038e-05,
-      "loss": 0.0398,
-      "step": 1179
-    },
-    {
-      "epoch": 69.41,
-      "learning_rate": 3.436532507739939e-05,
-      "loss": 0.0304,
-      "step": 1180
-    },
-    {
-      "epoch": 69.41,
-      "eval_accuracy_NAT": 0.9931456589523792,
-      "eval_accuracy_REST": 0.2634146341463415,
-      "eval_accuracy_concrete": 0.945193767399623,
-      "eval_accuracy_concrete tiles": 0.9119019561630921,
-      "eval_accuracy_dark bricks": 0.888994697854258,
-      "eval_accuracy_grass": 0.8893727266877864,
-      "eval_accuracy_red bricks": 0.9082271528998243,
-      "eval_accuracy_road": 0.9828548616637123,
-      "eval_accuracy_rough concrete": 0.6426328365679691,
-      "eval_accuracy_rough red sidewalk": 0.7215800991133938,
-      "eval_accuracy_speedway bricks": 0.9898546686380514,
-      "eval_accuracy_steel": 0.9067880471994683,
-      "eval_accuracy_tiles": 0.2100033324244903,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.98200447224178,
-      "eval_iou_REST": 0.26129032258064516,
-      "eval_iou_concrete": 0.8846762397166505,
-      "eval_iou_concrete tiles": 0.8445084686572376,
-      "eval_iou_dark bricks": 0.8599699108899433,
-      "eval_iou_grass": 0.8291459103928835,
-      "eval_iou_red bricks": 0.8779464854533872,
-      "eval_iou_road": 0.9485877067067877,
-      "eval_iou_rough concrete": 0.6299121536376805,
-      "eval_iou_rough red sidewalk": 0.622748778255875,
-      "eval_iou_speedway bricks": 0.977983933489584,
-      "eval_iou_steel": 0.8494635738180013,
-      "eval_iou_tiles": 0.20392433736357485,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07490196824073792,
-      "eval_mean_accuracy": 0.7887664953623378,
-      "eval_mean_iou": 0.7517047917849256,
-      "eval_overall_accuracy": 0.978399438373113,
-      "eval_runtime": 7.2052,
-      "eval_samples_per_second": 16.377,
-      "eval_steps_per_second": 1.11,
-      "step": 1180
-    },
-    {
-      "epoch": 69.47,
-      "learning_rate": 3.434984520123839e-05,
-      "loss": 0.0381,
-      "step": 1181
-    },
-    {
-      "epoch": 69.53,
-      "learning_rate": 3.43343653250774e-05,
-      "loss": 0.0305,
-      "step": 1182
-    },
-    {
-      "epoch": 69.59,
-      "learning_rate": 3.431888544891641e-05,
-      "loss": 0.0355,
-      "step": 1183
-    },
-    {
-      "epoch": 69.65,
-      "learning_rate": 3.430340557275542e-05,
-      "loss": 0.0428,
-      "step": 1184
-    },
-    {
-      "epoch": 69.71,
-      "learning_rate": 3.428792569659443e-05,
-      "loss": 0.038,
-      "step": 1185
-    },
-    {
-      "epoch": 69.76,
-      "learning_rate": 3.427244582043344e-05,
-      "loss": 0.0352,
-      "step": 1186
-    },
-    {
-      "epoch": 69.82,
-      "learning_rate": 3.425696594427244e-05,
-      "loss": 0.034,
-      "step": 1187
-    },
-    {
-      "epoch": 69.88,
-      "learning_rate": 3.424148606811145e-05,
-      "loss": 0.035,
-      "step": 1188
-    },
-    {
-      "epoch": 69.94,
-      "learning_rate": 3.422600619195046e-05,
-      "loss": 0.0308,
-      "step": 1189
-    },
-    {
-      "epoch": 70.0,
-      "learning_rate": 3.421052631578947e-05,
-      "loss": 0.0307,
-      "step": 1190
-    },
-    {
-      "epoch": 70.06,
-      "learning_rate": 3.419504643962848e-05,
-      "loss": 0.031,
-      "step": 1191
-    },
-    {
-      "epoch": 70.12,
-      "learning_rate": 3.41795665634675e-05,
-      "loss": 0.0365,
-      "step": 1192
-    },
-    {
-      "epoch": 70.18,
-      "learning_rate": 3.41640866873065e-05,
-      "loss": 0.0376,
-      "step": 1193
-    },
-    {
-      "epoch": 70.24,
-      "learning_rate": 3.414860681114551e-05,
-      "loss": 0.0354,
-      "step": 1194
-    },
-    {
-      "epoch": 70.29,
-      "learning_rate": 3.413312693498452e-05,
-      "loss": 0.0276,
-      "step": 1195
-    },
-    {
-      "epoch": 70.35,
-      "learning_rate": 3.411764705882353e-05,
-      "loss": 0.0324,
-      "step": 1196
-    },
-    {
-      "epoch": 70.41,
-      "learning_rate": 3.410216718266254e-05,
-      "loss": 0.0336,
-      "step": 1197
-    },
-    {
-      "epoch": 70.47,
-      "learning_rate": 3.4086687306501546e-05,
-      "loss": 0.0322,
-      "step": 1198
-    },
-    {
-      "epoch": 70.53,
-      "learning_rate": 3.4071207430340556e-05,
-      "loss": 0.0346,
-      "step": 1199
-    },
-    {
-      "epoch": 70.59,
-      "learning_rate": 3.4055727554179566e-05,
-      "loss": 0.0366,
-      "step": 1200
-    },
-    {
-      "epoch": 70.59,
-      "eval_accuracy_NAT": 0.992674955383976,
-      "eval_accuracy_REST": 0.04715447154471545,
-      "eval_accuracy_concrete": 0.958235171813872,
-      "eval_accuracy_concrete tiles": 0.8674994107942493,
-      "eval_accuracy_dark bricks": 0.8251440384357713,
-      "eval_accuracy_grass": 0.8618933260663041,
-      "eval_accuracy_red bricks": 0.9024055360281195,
-      "eval_accuracy_road": 0.9774027482883987,
-      "eval_accuracy_rough concrete": 0.6224027076791067,
-      "eval_accuracy_rough red sidewalk": 0.8008482542356312,
-      "eval_accuracy_speedway bricks": 0.9870333127009685,
-      "eval_accuracy_steel": 0.9060183820038283,
-      "eval_accuracy_tiles": 0.14244599957587326,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9819493271821238,
-      "eval_iou_REST": 0.04707792207792208,
-      "eval_iou_concrete": 0.8767137131319575,
-      "eval_iou_concrete tiles": 0.8272390156197326,
-      "eval_iou_dark bricks": 0.8106283554443995,
-      "eval_iou_grass": 0.8130420850341017,
-      "eval_iou_red bricks": 0.8766940561306157,
-      "eval_iou_road": 0.9598719622793611,
-      "eval_iou_rough concrete": 0.6054954928842642,
-      "eval_iou_rough red sidewalk": 0.5239165834436141,
-      "eval_iou_speedway bricks": 0.9776720750888546,
-      "eval_iou_steel": 0.8490466622641244,
-      "eval_iou_tiles": 0.13535219781801433,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08301250636577606,
-      "eval_mean_accuracy": 0.7608583318885243,
-      "eval_mean_iou": 0.7142076498768526,
-      "eval_overall_accuracy": 0.9770683676509534,
-      "eval_runtime": 5.8395,
-      "eval_samples_per_second": 20.207,
-      "eval_steps_per_second": 1.37,
-      "step": 1200
-    },
-    {
-      "epoch": 70.65,
-      "learning_rate": 3.4040247678018576e-05,
-      "loss": 0.0324,
-      "step": 1201
-    },
-    {
-      "epoch": 70.71,
-      "learning_rate": 3.4024767801857586e-05,
-      "loss": 0.0378,
-      "step": 1202
-    },
-    {
-      "epoch": 70.76,
-      "learning_rate": 3.4009287925696596e-05,
-      "loss": 0.0355,
-      "step": 1203
-    },
-    {
-      "epoch": 70.82,
-      "learning_rate": 3.39938080495356e-05,
-      "loss": 0.038,
-      "step": 1204
-    },
-    {
-      "epoch": 70.88,
-      "learning_rate": 3.397832817337461e-05,
-      "loss": 0.039,
-      "step": 1205
-    },
-    {
-      "epoch": 70.94,
-      "learning_rate": 3.396284829721362e-05,
-      "loss": 0.0297,
-      "step": 1206
-    },
-    {
-      "epoch": 71.0,
-      "learning_rate": 3.3947368421052636e-05,
-      "loss": 0.0321,
-      "step": 1207
-    },
-    {
-      "epoch": 71.06,
-      "learning_rate": 3.3931888544891646e-05,
-      "loss": 0.0328,
-      "step": 1208
-    },
-    {
-      "epoch": 71.12,
-      "learning_rate": 3.3916408668730656e-05,
-      "loss": 0.0335,
-      "step": 1209
-    },
-    {
-      "epoch": 71.18,
-      "learning_rate": 3.390092879256966e-05,
-      "loss": 0.0323,
-      "step": 1210
-    },
-    {
-      "epoch": 71.24,
-      "learning_rate": 3.388544891640867e-05,
-      "loss": 0.0333,
-      "step": 1211
-    },
-    {
-      "epoch": 71.29,
-      "learning_rate": 3.386996904024768e-05,
-      "loss": 0.0339,
-      "step": 1212
-    },
-    {
-      "epoch": 71.35,
-      "learning_rate": 3.385448916408669e-05,
-      "loss": 0.0317,
-      "step": 1213
-    },
-    {
-      "epoch": 71.41,
-      "learning_rate": 3.38390092879257e-05,
-      "loss": 0.0318,
-      "step": 1214
-    },
-    {
-      "epoch": 71.47,
-      "learning_rate": 3.382352941176471e-05,
-      "loss": 0.0317,
-      "step": 1215
-    },
-    {
-      "epoch": 71.53,
-      "learning_rate": 3.380804953560371e-05,
-      "loss": 0.0288,
-      "step": 1216
-    },
-    {
-      "epoch": 71.59,
-      "learning_rate": 3.379256965944272e-05,
-      "loss": 0.0303,
-      "step": 1217
-    },
-    {
-      "epoch": 71.65,
-      "learning_rate": 3.377708978328173e-05,
-      "loss": 0.0386,
-      "step": 1218
-    },
-    {
-      "epoch": 71.71,
-      "learning_rate": 3.376160990712074e-05,
-      "loss": 0.0318,
-      "step": 1219
-    },
-    {
-      "epoch": 71.76,
-      "learning_rate": 3.374613003095975e-05,
-      "loss": 0.0296,
-      "step": 1220
-    },
-    {
-      "epoch": 71.76,
-      "eval_accuracy_NAT": 0.9916076642897511,
-      "eval_accuracy_REST": 0.14308943089430895,
-      "eval_accuracy_concrete": 0.9524127477224744,
-      "eval_accuracy_concrete tiles": 0.9023332547725665,
-      "eval_accuracy_dark bricks": 0.8711693432612981,
-      "eval_accuracy_grass": 0.9014191524070142,
-      "eval_accuracy_red bricks": 0.890322934973638,
-      "eval_accuracy_road": 0.9790663857062923,
-      "eval_accuracy_rough concrete": 0.6242618535445927,
-      "eval_accuracy_rough red sidewalk": 0.7556456282570932,
-      "eval_accuracy_speedway bricks": 0.991555981912065,
-      "eval_accuracy_steel": 0.885989594526381,
-      "eval_accuracy_tiles": 0.17537641249356237,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9818573901591024,
-      "eval_iou_REST": 0.14308943089430895,
-      "eval_iou_concrete": 0.8808498965890104,
-      "eval_iou_concrete tiles": 0.839660503980525,
-      "eval_iou_dark bricks": 0.8471303198667299,
-      "eval_iou_grass": 0.8321373634388728,
-      "eval_iou_red bricks": 0.8664813726014218,
-      "eval_iou_road": 0.9586484952029115,
-      "eval_iou_rough concrete": 0.6103291581706962,
-      "eval_iou_rough red sidewalk": 0.6133155109589292,
-      "eval_iou_speedway bricks": 0.977225058515616,
-      "eval_iou_steel": 0.8448727388841627,
-      "eval_iou_tiles": 0.1710192023633678,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07912295311689377,
-      "eval_mean_accuracy": 0.7741731065200798,
-      "eval_mean_iou": 0.7358935724327427,
-      "eval_overall_accuracy": 0.978185427390923,
-      "eval_runtime": 13.2108,
-      "eval_samples_per_second": 8.932,
-      "eval_steps_per_second": 0.606,
-      "step": 1220
-    },
-    {
-      "epoch": 71.82,
-      "learning_rate": 3.373065015479876e-05,
-      "loss": 0.0351,
-      "step": 1221
-    },
-    {
-      "epoch": 71.88,
-      "learning_rate": 3.371517027863777e-05,
-      "loss": 0.027,
-      "step": 1222
-    },
-    {
-      "epoch": 71.94,
-      "learning_rate": 3.369969040247678e-05,
-      "loss": 0.0298,
-      "step": 1223
-    },
-    {
-      "epoch": 72.0,
-      "learning_rate": 3.368421052631579e-05,
-      "loss": 0.0299,
-      "step": 1224
-    },
-    {
-      "epoch": 72.06,
-      "learning_rate": 3.36687306501548e-05,
-      "loss": 0.0306,
-      "step": 1225
-    },
-    {
-      "epoch": 72.12,
-      "learning_rate": 3.365325077399381e-05,
-      "loss": 0.0309,
-      "step": 1226
-    },
-    {
-      "epoch": 72.18,
-      "learning_rate": 3.363777089783282e-05,
-      "loss": 0.0277,
-      "step": 1227
-    },
-    {
-      "epoch": 72.24,
-      "learning_rate": 3.3622291021671826e-05,
-      "loss": 0.0363,
-      "step": 1228
-    },
-    {
-      "epoch": 72.29,
-      "learning_rate": 3.3606811145510836e-05,
-      "loss": 0.0299,
-      "step": 1229
-    },
-    {
-      "epoch": 72.35,
-      "learning_rate": 3.3591331269349846e-05,
-      "loss": 0.032,
-      "step": 1230
-    },
-    {
-      "epoch": 72.41,
-      "learning_rate": 3.3575851393188856e-05,
-      "loss": 0.0406,
-      "step": 1231
-    },
-    {
-      "epoch": 72.47,
-      "learning_rate": 3.3560371517027866e-05,
-      "loss": 0.0306,
-      "step": 1232
-    },
-    {
-      "epoch": 72.53,
-      "learning_rate": 3.3544891640866876e-05,
-      "loss": 0.0264,
-      "step": 1233
-    },
-    {
-      "epoch": 72.59,
-      "learning_rate": 3.352941176470588e-05,
-      "loss": 0.0306,
-      "step": 1234
-    },
-    {
-      "epoch": 72.65,
-      "learning_rate": 3.351393188854489e-05,
-      "loss": 0.0321,
-      "step": 1235
-    },
-    {
-      "epoch": 72.71,
-      "learning_rate": 3.34984520123839e-05,
-      "loss": 0.0303,
-      "step": 1236
-    },
-    {
-      "epoch": 72.76,
-      "learning_rate": 3.348297213622291e-05,
-      "loss": 0.0339,
-      "step": 1237
-    },
-    {
-      "epoch": 72.82,
-      "learning_rate": 3.346749226006192e-05,
-      "loss": 0.0351,
-      "step": 1238
-    },
-    {
-      "epoch": 72.88,
-      "learning_rate": 3.345201238390093e-05,
-      "loss": 0.0348,
-      "step": 1239
-    },
-    {
-      "epoch": 72.94,
-      "learning_rate": 3.343653250773994e-05,
-      "loss": 0.0313,
-      "step": 1240
-    },
-    {
-      "epoch": 72.94,
-      "eval_accuracy_NAT": 0.9927213194967726,
-      "eval_accuracy_REST": 0.08943089430894309,
-      "eval_accuracy_concrete": 0.9468108526254798,
-      "eval_accuracy_concrete tiles": 0.9154843271270328,
-      "eval_accuracy_dark bricks": 0.856550159830025,
-      "eval_accuracy_grass": 0.8992655380061846,
-      "eval_accuracy_red bricks": 0.8841168717047452,
-      "eval_accuracy_road": 0.9796259193961525,
-      "eval_accuracy_rough concrete": 0.6253403964986086,
-      "eval_accuracy_rough red sidewalk": 0.784311808631438,
-      "eval_accuracy_speedway bricks": 0.9926374674149154,
-      "eval_accuracy_steel": 0.8652661092424795,
-      "eval_accuracy_tiles": 0.2090338998454967,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9821372358073259,
-      "eval_iou_REST": 0.08943089430894309,
-      "eval_iou_concrete": 0.8860208515889054,
-      "eval_iou_concrete tiles": 0.8653152149699265,
-      "eval_iou_dark bricks": 0.839343324455115,
-      "eval_iou_grass": 0.8308539785950206,
-      "eval_iou_red bricks": 0.8594308899685014,
-      "eval_iou_road": 0.9600674794861629,
-      "eval_iou_rough concrete": 0.6149735127279546,
-      "eval_iou_rough red sidewalk": 0.6126972377897618,
-      "eval_iou_speedway bricks": 0.9755335515396683,
-      "eval_iou_steel": 0.8341532160925078,
-      "eval_iou_tiles": 0.2064508407635689,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07735487818717957,
-      "eval_mean_accuracy": 0.7723535049329442,
-      "eval_mean_iou": 0.7351083252379511,
-      "eval_overall_accuracy": 0.9783551490913003,
-      "eval_runtime": 5.5504,
-      "eval_samples_per_second": 21.26,
-      "eval_steps_per_second": 1.441,
-      "step": 1240
-    },
-    {
-      "epoch": 73.0,
-      "learning_rate": 3.342105263157895e-05,
-      "loss": 0.0332,
-      "step": 1241
-    },
-    {
-      "epoch": 73.06,
-      "learning_rate": 3.340557275541796e-05,
-      "loss": 0.031,
-      "step": 1242
-    },
-    {
-      "epoch": 73.12,
-      "learning_rate": 3.339009287925697e-05,
-      "loss": 0.0293,
-      "step": 1243
-    },
-    {
-      "epoch": 73.18,
-      "learning_rate": 3.337461300309598e-05,
-      "loss": 0.0289,
-      "step": 1244
-    },
-    {
-      "epoch": 73.24,
-      "learning_rate": 3.335913312693499e-05,
-      "loss": 0.0255,
-      "step": 1245
-    },
-    {
-      "epoch": 73.29,
-      "learning_rate": 3.334365325077399e-05,
-      "loss": 0.0296,
-      "step": 1246
-    },
-    {
-      "epoch": 73.35,
-      "learning_rate": 3.3328173374613e-05,
-      "loss": 0.0336,
-      "step": 1247
-    },
-    {
-      "epoch": 73.41,
-      "learning_rate": 3.331269349845201e-05,
-      "loss": 0.0289,
-      "step": 1248
-    },
-    {
-      "epoch": 73.47,
-      "learning_rate": 3.329721362229102e-05,
-      "loss": 0.0294,
-      "step": 1249
-    },
-    {
-      "epoch": 73.53,
-      "learning_rate": 3.328173374613003e-05,
-      "loss": 0.034,
-      "step": 1250
-    },
-    {
-      "epoch": 73.59,
-      "learning_rate": 3.326625386996904e-05,
-      "loss": 0.0299,
-      "step": 1251
-    },
-    {
-      "epoch": 73.65,
-      "learning_rate": 3.3250773993808046e-05,
-      "loss": 0.0307,
-      "step": 1252
-    },
-    {
-      "epoch": 73.71,
-      "learning_rate": 3.3235294117647056e-05,
-      "loss": 0.0339,
-      "step": 1253
-    },
-    {
-      "epoch": 73.76,
-      "learning_rate": 3.3219814241486066e-05,
-      "loss": 0.032,
-      "step": 1254
-    },
-    {
-      "epoch": 73.82,
-      "learning_rate": 3.320433436532508e-05,
-      "loss": 0.0294,
-      "step": 1255
-    },
-    {
-      "epoch": 73.88,
-      "learning_rate": 3.318885448916409e-05,
-      "loss": 0.0282,
-      "step": 1256
-    },
-    {
-      "epoch": 73.94,
-      "learning_rate": 3.31733746130031e-05,
-      "loss": 0.0349,
-      "step": 1257
-    },
-    {
-      "epoch": 74.0,
-      "learning_rate": 3.3157894736842106e-05,
-      "loss": 0.0268,
-      "step": 1258
-    },
-    {
-      "epoch": 74.06,
-      "learning_rate": 3.3142414860681116e-05,
-      "loss": 0.0285,
-      "step": 1259
-    },
-    {
-      "epoch": 74.12,
-      "learning_rate": 3.3126934984520126e-05,
-      "loss": 0.0272,
-      "step": 1260
-    },
-    {
-      "epoch": 74.12,
-      "eval_accuracy_NAT": 0.9942002268714645,
-      "eval_accuracy_REST": 0.22276422764227644,
-      "eval_accuracy_concrete": 0.9498260228799787,
-      "eval_accuracy_concrete tiles": 0.9140702333254772,
-      "eval_accuracy_dark bricks": 0.8848888846352622,
-      "eval_accuracy_grass": 0.8712098846737596,
-      "eval_accuracy_red bricks": 0.9234951669595782,
-      "eval_accuracy_road": 0.97808092235949,
-      "eval_accuracy_rough concrete": 0.6260733290032714,
-      "eval_accuracy_rough red sidewalk": 0.772994232171997,
-      "eval_accuracy_speedway bricks": 0.9907025834325538,
-      "eval_accuracy_steel": 0.8942984801611299,
-      "eval_accuracy_tiles": 0.264382441152413,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9816555179486792,
-      "eval_iou_REST": 0.21642969984202212,
-      "eval_iou_concrete": 0.8902363972700686,
-      "eval_iou_concrete tiles": 0.855441351625568,
-      "eval_iou_dark bricks": 0.8587450193652651,
-      "eval_iou_grass": 0.8241404527621606,
-      "eval_iou_red bricks": 0.8853261728005054,
-      "eval_iou_road": 0.9616836897156543,
-      "eval_iou_rough concrete": 0.617017653069614,
-      "eval_iou_rough red sidewalk": 0.6000770586930378,
-      "eval_iou_speedway bricks": 0.977204896135035,
-      "eval_iou_steel": 0.8476513278193066,
-      "eval_iou_tiles": 0.25355918414782963,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07605685293674469,
-      "eval_mean_accuracy": 0.7913066642514347,
-      "eval_mean_iou": 0.7514744939380574,
-      "eval_overall_accuracy": 0.9786853143724344,
-      "eval_runtime": 5.5689,
-      "eval_samples_per_second": 21.189,
-      "eval_steps_per_second": 1.437,
-      "step": 1260
-    },
-    {
-      "epoch": 74.18,
-      "learning_rate": 3.3111455108359136e-05,
-      "loss": 0.0309,
-      "step": 1261
-    },
-    {
-      "epoch": 74.24,
-      "learning_rate": 3.3095975232198146e-05,
-      "loss": 0.0314,
-      "step": 1262
-    },
-    {
-      "epoch": 74.29,
-      "learning_rate": 3.3080495356037156e-05,
-      "loss": 0.0328,
-      "step": 1263
-    },
-    {
-      "epoch": 74.35,
-      "learning_rate": 3.306501547987616e-05,
-      "loss": 0.0331,
-      "step": 1264
-    },
-    {
-      "epoch": 74.41,
-      "learning_rate": 3.304953560371517e-05,
-      "loss": 0.0306,
-      "step": 1265
-    },
-    {
-      "epoch": 74.47,
-      "learning_rate": 3.303405572755418e-05,
-      "loss": 0.0313,
-      "step": 1266
-    },
-    {
-      "epoch": 74.53,
-      "learning_rate": 3.301857585139319e-05,
-      "loss": 0.0261,
-      "step": 1267
-    },
-    {
-      "epoch": 74.59,
-      "learning_rate": 3.30030959752322e-05,
-      "loss": 0.0322,
-      "step": 1268
-    },
-    {
-      "epoch": 74.65,
-      "learning_rate": 3.298761609907121e-05,
-      "loss": 0.0277,
-      "step": 1269
-    },
-    {
-      "epoch": 74.71,
-      "learning_rate": 3.297213622291022e-05,
-      "loss": 0.0288,
-      "step": 1270
-    },
-    {
-      "epoch": 74.76,
-      "learning_rate": 3.295665634674923e-05,
-      "loss": 0.0329,
-      "step": 1271
-    },
-    {
-      "epoch": 74.82,
-      "learning_rate": 3.294117647058824e-05,
-      "loss": 0.0311,
-      "step": 1272
-    },
-    {
-      "epoch": 74.88,
-      "learning_rate": 3.292569659442725e-05,
-      "loss": 0.0312,
-      "step": 1273
-    },
-    {
-      "epoch": 74.94,
-      "learning_rate": 3.291021671826626e-05,
-      "loss": 0.0284,
-      "step": 1274
-    },
-    {
-      "epoch": 75.0,
-      "learning_rate": 3.289473684210527e-05,
-      "loss": 0.0292,
-      "step": 1275
-    },
-    {
-      "epoch": 75.06,
-      "learning_rate": 3.287925696594427e-05,
-      "loss": 0.0302,
-      "step": 1276
-    },
-    {
-      "epoch": 75.12,
-      "learning_rate": 3.286377708978328e-05,
-      "loss": 0.0342,
-      "step": 1277
-    },
-    {
-      "epoch": 75.18,
-      "learning_rate": 3.284829721362229e-05,
-      "loss": 0.0294,
-      "step": 1278
-    },
-    {
-      "epoch": 75.24,
-      "learning_rate": 3.28328173374613e-05,
-      "loss": 0.0272,
-      "step": 1279
-    },
-    {
-      "epoch": 75.29,
-      "learning_rate": 3.281733746130031e-05,
-      "loss": 0.0329,
-      "step": 1280
-    },
-    {
-      "epoch": 75.29,
-      "eval_accuracy_NAT": 0.9928449391603104,
-      "eval_accuracy_REST": 0.1902439024390244,
-      "eval_accuracy_concrete": 0.9563448140906179,
-      "eval_accuracy_concrete tiles": 0.8934716002828188,
-      "eval_accuracy_dark bricks": 0.8582058840418813,
-      "eval_accuracy_grass": 0.8855253645815592,
-      "eval_accuracy_red bricks": 0.8926296133567663,
-      "eval_accuracy_road": 0.9793902418820853,
-      "eval_accuracy_rough concrete": 0.6249173216381936,
-      "eval_accuracy_rough red sidewalk": 0.7686627211815399,
-      "eval_accuracy_speedway bricks": 0.9893386881517445,
-      "eval_accuracy_steel": 0.8970098007366796,
-      "eval_accuracy_tiles": 0.21088188069920324,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9820734284240179,
-      "eval_iou_REST": 0.18962722852512157,
-      "eval_iou_concrete": 0.8856006348028841,
-      "eval_iou_concrete tiles": 0.8492003046458492,
-      "eval_iou_dark bricks": 0.8390065215153868,
-      "eval_iou_grass": 0.8260845002789371,
-      "eval_iou_red bricks": 0.8720356261401437,
-      "eval_iou_road": 0.9609563440959449,
-      "eval_iou_rough concrete": 0.6129052995768755,
-      "eval_iou_rough red sidewalk": 0.5995917313874094,
-      "eval_iou_speedway bricks": 0.9775231301854868,
-      "eval_iou_steel": 0.8492074757511238,
-      "eval_iou_tiles": 0.20707401237505949,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07827547192573547,
-      "eval_mean_accuracy": 0.7799589824801865,
-      "eval_mean_iou": 0.7423758644387879,
-      "eval_overall_accuracy": 0.9784607644808494,
-      "eval_runtime": 6.2956,
-      "eval_samples_per_second": 18.743,
-      "eval_steps_per_second": 1.271,
-      "step": 1280
-    },
-    {
-      "epoch": 75.35,
-      "learning_rate": 3.280185758513932e-05,
-      "loss": 0.0345,
-      "step": 1281
-    },
-    {
-      "epoch": 75.41,
-      "learning_rate": 3.2786377708978326e-05,
-      "loss": 0.0353,
-      "step": 1282
-    },
-    {
-      "epoch": 75.47,
-      "learning_rate": 3.2770897832817336e-05,
-      "loss": 0.0303,
-      "step": 1283
-    },
-    {
-      "epoch": 75.53,
-      "learning_rate": 3.2755417956656346e-05,
-      "loss": 0.0284,
-      "step": 1284
-    },
-    {
-      "epoch": 75.59,
-      "learning_rate": 3.2739938080495356e-05,
-      "loss": 0.0282,
-      "step": 1285
-    },
-    {
-      "epoch": 75.65,
-      "learning_rate": 3.2724458204334366e-05,
-      "loss": 0.0288,
-      "step": 1286
-    },
-    {
-      "epoch": 75.71,
-      "learning_rate": 3.2708978328173376e-05,
-      "loss": 0.0285,
-      "step": 1287
-    },
-    {
-      "epoch": 75.76,
-      "learning_rate": 3.2693498452012386e-05,
-      "loss": 0.0289,
-      "step": 1288
-    },
-    {
-      "epoch": 75.82,
-      "learning_rate": 3.2678018575851396e-05,
-      "loss": 0.0266,
-      "step": 1289
-    },
-    {
-      "epoch": 75.88,
-      "learning_rate": 3.2662538699690406e-05,
-      "loss": 0.0244,
-      "step": 1290
-    },
-    {
-      "epoch": 75.94,
-      "learning_rate": 3.2647058823529416e-05,
-      "loss": 0.033,
-      "step": 1291
-    },
-    {
-      "epoch": 76.0,
-      "learning_rate": 3.2631578947368426e-05,
-      "loss": 0.0308,
-      "step": 1292
-    },
-    {
-      "epoch": 76.06,
-      "learning_rate": 3.2616099071207436e-05,
-      "loss": 0.0316,
-      "step": 1293
-    },
-    {
-      "epoch": 76.12,
-      "learning_rate": 3.260061919504644e-05,
-      "loss": 0.0321,
-      "step": 1294
-    },
-    {
-      "epoch": 76.18,
-      "learning_rate": 3.258513931888545e-05,
-      "loss": 0.0283,
-      "step": 1295
-    },
-    {
-      "epoch": 76.24,
-      "learning_rate": 3.256965944272446e-05,
-      "loss": 0.0281,
-      "step": 1296
-    },
-    {
-      "epoch": 76.29,
-      "learning_rate": 3.255417956656347e-05,
-      "loss": 0.0296,
-      "step": 1297
-    },
-    {
-      "epoch": 76.35,
-      "learning_rate": 3.253869969040248e-05,
-      "loss": 0.0286,
-      "step": 1298
-    },
-    {
-      "epoch": 76.41,
-      "learning_rate": 3.252321981424149e-05,
-      "loss": 0.0279,
-      "step": 1299
-    },
-    {
-      "epoch": 76.47,
-      "learning_rate": 3.250773993808049e-05,
-      "loss": 0.0297,
-      "step": 1300
-    },
-    {
-      "epoch": 76.47,
-      "eval_accuracy_NAT": 0.992488851991681,
-      "eval_accuracy_REST": 0.2601626016260163,
-      "eval_accuracy_concrete": 0.9521742553611096,
-      "eval_accuracy_concrete tiles": 0.9211407023332547,
-      "eval_accuracy_dark bricks": 0.8558323603162146,
-      "eval_accuracy_grass": 0.9004804944607826,
-      "eval_accuracy_red bricks": 0.8834578207381371,
-      "eval_accuracy_road": 0.9812932977269678,
-      "eval_accuracy_rough concrete": 0.6232965278067442,
-      "eval_accuracy_rough red sidewalk": 0.7873047624060941,
-      "eval_accuracy_speedway bricks": 0.9882248146343102,
-      "eval_accuracy_steel": 0.9161164893271427,
-      "eval_accuracy_tiles": 0.2258777909055106,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9823089468549129,
-      "eval_iou_REST": 0.25806451612903225,
-      "eval_iou_concrete": 0.8877026974541958,
-      "eval_iou_concrete tiles": 0.8522459659834278,
-      "eval_iou_dark bricks": 0.8368740231907387,
-      "eval_iou_grass": 0.8291762110430432,
-      "eval_iou_red bricks": 0.8674036128336479,
-      "eval_iou_road": 0.9597156076083275,
-      "eval_iou_rough concrete": 0.614172812250458,
-      "eval_iou_rough red sidewalk": 0.6164766705136963,
-      "eval_iou_speedway bricks": 0.9772265422709538,
-      "eval_iou_steel": 0.8510678800260005,
-      "eval_iou_tiles": 0.22112818079364138,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07827833294868469,
-      "eval_mean_accuracy": 0.7913731361256896,
-      "eval_mean_iou": 0.7502741282270828,
-      "eval_overall_accuracy": 0.9786628464520988,
-      "eval_runtime": 5.5012,
-      "eval_samples_per_second": 21.45,
-      "eval_steps_per_second": 1.454,
-      "step": 1300
-    },
-    {
-      "epoch": 76.53,
-      "learning_rate": 3.24922600619195e-05,
-      "loss": 0.0318,
-      "step": 1301
-    },
-    {
-      "epoch": 76.59,
-      "learning_rate": 3.247678018575851e-05,
-      "loss": 0.0295,
-      "step": 1302
-    },
-    {
-      "epoch": 76.65,
-      "learning_rate": 3.246130030959752e-05,
-      "loss": 0.0265,
-      "step": 1303
-    },
-    {
-      "epoch": 76.71,
-      "learning_rate": 3.244582043343654e-05,
-      "loss": 0.0291,
-      "step": 1304
-    },
-    {
-      "epoch": 76.76,
-      "learning_rate": 3.243034055727555e-05,
-      "loss": 0.0284,
-      "step": 1305
-    },
-    {
-      "epoch": 76.82,
-      "learning_rate": 3.241486068111455e-05,
-      "loss": 0.0298,
-      "step": 1306
-    },
-    {
-      "epoch": 76.88,
-      "learning_rate": 3.239938080495356e-05,
-      "loss": 0.0343,
-      "step": 1307
-    },
-    {
-      "epoch": 76.94,
-      "learning_rate": 3.238390092879257e-05,
-      "loss": 0.0295,
-      "step": 1308
-    },
-    {
-      "epoch": 77.0,
-      "learning_rate": 3.236842105263158e-05,
-      "loss": 0.0333,
-      "step": 1309
-    },
-    {
-      "epoch": 77.06,
-      "learning_rate": 3.235294117647059e-05,
-      "loss": 0.0275,
-      "step": 1310
-    },
-    {
-      "epoch": 77.12,
-      "learning_rate": 3.23374613003096e-05,
-      "loss": 0.0253,
-      "step": 1311
-    },
-    {
-      "epoch": 77.18,
-      "learning_rate": 3.2321981424148605e-05,
-      "loss": 0.0312,
-      "step": 1312
-    },
-    {
-      "epoch": 77.24,
-      "learning_rate": 3.2306501547987615e-05,
-      "loss": 0.0301,
-      "step": 1313
-    },
-    {
-      "epoch": 77.29,
-      "learning_rate": 3.2291021671826625e-05,
-      "loss": 0.0316,
-      "step": 1314
-    },
-    {
-      "epoch": 77.35,
-      "learning_rate": 3.2275541795665635e-05,
-      "loss": 0.0372,
-      "step": 1315
-    },
-    {
-      "epoch": 77.41,
-      "learning_rate": 3.2260061919504645e-05,
-      "loss": 0.037,
-      "step": 1316
-    },
-    {
-      "epoch": 77.47,
-      "learning_rate": 3.2244582043343655e-05,
-      "loss": 0.0305,
-      "step": 1317
-    },
-    {
-      "epoch": 77.53,
-      "learning_rate": 3.2229102167182665e-05,
-      "loss": 0.0297,
-      "step": 1318
-    },
-    {
-      "epoch": 77.59,
-      "learning_rate": 3.2213622291021675e-05,
-      "loss": 0.0282,
-      "step": 1319
-    },
-    {
-      "epoch": 77.65,
-      "learning_rate": 3.2198142414860685e-05,
-      "loss": 0.0316,
-      "step": 1320
-    },
-    {
-      "epoch": 77.65,
-      "eval_accuracy_NAT": 0.9949590887920742,
-      "eval_accuracy_REST": 0.06829268292682927,
-      "eval_accuracy_concrete": 0.9489343519199399,
-      "eval_accuracy_concrete tiles": 0.9051143059156257,
-      "eval_accuracy_dark bricks": 0.8368776678215263,
-      "eval_accuracy_grass": 0.8689285295241989,
-      "eval_accuracy_red bricks": 0.8564367311072056,
-      "eval_accuracy_road": 0.9787991776404565,
-      "eval_accuracy_rough concrete": 0.622158396844219,
-      "eval_accuracy_rough red sidewalk": 0.7372968664225178,
-      "eval_accuracy_speedway bricks": 0.9889505460722078,
-      "eval_accuracy_steel": 0.9112061253354791,
-      "eval_accuracy_tiles": 0.22015207973582962,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.981318691427329,
-      "eval_iou_REST": 0.06807131280388978,
-      "eval_iou_concrete": 0.8872889012142022,
-      "eval_iou_concrete tiles": 0.8525885800550573,
-      "eval_iou_dark bricks": 0.8249517672753346,
-      "eval_iou_grass": 0.8239352772760756,
-      "eval_iou_red bricks": 0.8418268192614986,
-      "eval_iou_road": 0.9598942168303413,
-      "eval_iou_rough concrete": 0.6145488148701861,
-      "eval_iou_rough red sidewalk": 0.6000416164311462,
-      "eval_iou_speedway bricks": 0.9772518101809055,
-      "eval_iou_steel": 0.8517559944406162,
-      "eval_iou_tiles": 0.21822822822822824,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.08024384081363678,
-      "eval_mean_accuracy": 0.7644697346198547,
-      "eval_mean_iou": 0.7309001561765238,
-      "eval_overall_accuracy": 0.9783175840216168,
-      "eval_runtime": 7.305,
-      "eval_samples_per_second": 16.153,
-      "eval_steps_per_second": 1.095,
-      "step": 1320
-    },
-    {
-      "epoch": 77.71,
-      "learning_rate": 3.2182662538699695e-05,
-      "loss": 0.0311,
-      "step": 1321
-    },
-    {
-      "epoch": 77.76,
-      "learning_rate": 3.2167182662538705e-05,
-      "loss": 0.0303,
-      "step": 1322
-    },
-    {
-      "epoch": 77.82,
-      "learning_rate": 3.2151702786377715e-05,
-      "loss": 0.0348,
-      "step": 1323
-    },
-    {
-      "epoch": 77.88,
-      "learning_rate": 3.213622291021672e-05,
-      "loss": 0.0324,
-      "step": 1324
-    },
-    {
-      "epoch": 77.94,
-      "learning_rate": 3.212074303405573e-05,
-      "loss": 0.0321,
-      "step": 1325
-    },
-    {
-      "epoch": 78.0,
-      "learning_rate": 3.210526315789474e-05,
-      "loss": 0.0286,
-      "step": 1326
-    },
-    {
-      "epoch": 78.06,
-      "learning_rate": 3.208978328173375e-05,
-      "loss": 0.0312,
-      "step": 1327
-    },
-    {
-      "epoch": 78.12,
-      "learning_rate": 3.207430340557276e-05,
-      "loss": 0.026,
-      "step": 1328
-    },
-    {
-      "epoch": 78.18,
-      "learning_rate": 3.205882352941177e-05,
-      "loss": 0.0293,
-      "step": 1329
-    },
-    {
-      "epoch": 78.24,
-      "learning_rate": 3.204334365325077e-05,
-      "loss": 0.029,
-      "step": 1330
-    },
-    {
-      "epoch": 78.29,
-      "learning_rate": 3.202786377708978e-05,
-      "loss": 0.03,
-      "step": 1331
-    },
-    {
-      "epoch": 78.35,
-      "learning_rate": 3.201238390092879e-05,
-      "loss": 0.0307,
-      "step": 1332
-    },
-    {
-      "epoch": 78.41,
-      "learning_rate": 3.19969040247678e-05,
-      "loss": 0.0253,
-      "step": 1333
-    },
-    {
-      "epoch": 78.47,
-      "learning_rate": 3.198142414860681e-05,
-      "loss": 0.0307,
-      "step": 1334
-    },
-    {
-      "epoch": 78.53,
-      "learning_rate": 3.196594427244582e-05,
-      "loss": 0.0331,
-      "step": 1335
-    },
-    {
-      "epoch": 78.59,
-      "learning_rate": 3.195046439628483e-05,
-      "loss": 0.0307,
-      "step": 1336
-    },
-    {
-      "epoch": 78.65,
-      "learning_rate": 3.193498452012384e-05,
-      "loss": 0.0324,
-      "step": 1337
-    },
-    {
-      "epoch": 78.71,
-      "learning_rate": 3.191950464396285e-05,
-      "loss": 0.0329,
-      "step": 1338
-    },
-    {
-      "epoch": 78.76,
-      "learning_rate": 3.190402476780186e-05,
-      "loss": 0.0285,
-      "step": 1339
-    },
-    {
-      "epoch": 78.82,
-      "learning_rate": 3.188854489164087e-05,
-      "loss": 0.0295,
-      "step": 1340
-    },
-    {
-      "epoch": 78.82,
-      "eval_accuracy_NAT": 0.9931076511622378,
-      "eval_accuracy_REST": 0.23089430894308943,
-      "eval_accuracy_concrete": 0.9574501344577127,
-      "eval_accuracy_concrete tiles": 0.891821824181004,
-      "eval_accuracy_dark bricks": 0.8444624160174569,
-      "eval_accuracy_grass": 0.8917903386359938,
-      "eval_accuracy_red bricks": 0.8780206502636204,
-      "eval_accuracy_road": 0.9784314993418666,
-      "eval_accuracy_rough concrete": 0.6227423593276089,
-      "eval_accuracy_rough red sidewalk": 0.7669857646696094,
-      "eval_accuracy_speedway bricks": 0.9876396026247658,
-      "eval_accuracy_steel": 0.9197898913972422,
-      "eval_accuracy_tiles": 0.21251779817625496,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.982500950046869,
-      "eval_iou_REST": 0.22756410256410256,
-      "eval_iou_concrete": 0.8858469625941018,
-      "eval_iou_concrete tiles": 0.8542532057070615,
-      "eval_iou_dark bricks": 0.8293690518152979,
-      "eval_iou_grass": 0.8328935275195306,
-      "eval_iou_red bricks": 0.8618793465955038,
-      "eval_iou_road": 0.960303673072041,
-      "eval_iou_rough concrete": 0.6135943307049629,
-      "eval_iou_rough red sidewalk": 0.6129829074212083,
-      "eval_iou_speedway bricks": 0.9771381370643655,
-      "eval_iou_steel": 0.8528193994888774,
-      "eval_iou_tiles": 0.2103574427252009,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07932768762111664,
-      "eval_mean_accuracy": 0.7827426337844972,
-      "eval_mean_iou": 0.7462694644091633,
-      "eval_overall_accuracy": 0.9787218449479442,
-      "eval_runtime": 6.1176,
-      "eval_samples_per_second": 19.288,
-      "eval_steps_per_second": 1.308,
-      "step": 1340
-    },
-    {
-      "epoch": 78.88,
-      "learning_rate": 3.187306501547988e-05,
-      "loss": 0.0265,
-      "step": 1341
-    },
-    {
-      "epoch": 78.94,
-      "learning_rate": 3.1857585139318885e-05,
-      "loss": 0.0276,
-      "step": 1342
-    },
-    {
-      "epoch": 79.0,
-      "learning_rate": 3.1842105263157895e-05,
-      "loss": 0.0265,
-      "step": 1343
-    },
-    {
-      "epoch": 79.06,
-      "learning_rate": 3.1826625386996905e-05,
-      "loss": 0.0255,
-      "step": 1344
-    },
-    {
-      "epoch": 79.12,
-      "learning_rate": 3.1811145510835915e-05,
-      "loss": 0.0286,
-      "step": 1345
-    },
-    {
-      "epoch": 79.18,
-      "learning_rate": 3.1795665634674925e-05,
-      "loss": 0.0276,
-      "step": 1346
-    },
-    {
-      "epoch": 79.24,
-      "learning_rate": 3.178018575851393e-05,
-      "loss": 0.0315,
-      "step": 1347
-    },
-    {
-      "epoch": 79.29,
-      "learning_rate": 3.176470588235294e-05,
-      "loss": 0.0312,
-      "step": 1348
-    },
-    {
-      "epoch": 79.35,
-      "learning_rate": 3.174922600619195e-05,
-      "loss": 0.025,
-      "step": 1349
-    },
-    {
-      "epoch": 79.41,
-      "learning_rate": 3.173374613003096e-05,
-      "loss": 0.032,
-      "step": 1350
-    },
-    {
-      "epoch": 79.47,
-      "learning_rate": 3.171826625386997e-05,
-      "loss": 0.0298,
-      "step": 1351
-    },
-    {
-      "epoch": 79.53,
-      "learning_rate": 3.170278637770898e-05,
-      "loss": 0.0309,
-      "step": 1352
-    },
-    {
-      "epoch": 79.59,
-      "learning_rate": 3.168730650154799e-05,
-      "loss": 0.0284,
-      "step": 1353
-    },
-    {
-      "epoch": 79.65,
-      "learning_rate": 3.1671826625387e-05,
-      "loss": 0.0279,
-      "step": 1354
-    },
-    {
-      "epoch": 79.71,
-      "learning_rate": 3.165634674922601e-05,
-      "loss": 0.0271,
-      "step": 1355
-    },
-    {
-      "epoch": 79.76,
-      "learning_rate": 3.164086687306502e-05,
-      "loss": 0.0286,
-      "step": 1356
-    },
-    {
-      "epoch": 79.82,
-      "learning_rate": 3.162538699690403e-05,
-      "loss": 0.0291,
-      "step": 1357
-    },
-    {
-      "epoch": 79.88,
-      "learning_rate": 3.160990712074304e-05,
-      "loss": 0.0302,
-      "step": 1358
-    },
-    {
-      "epoch": 79.94,
-      "learning_rate": 3.159442724458204e-05,
-      "loss": 0.0272,
-      "step": 1359
-    },
-    {
-      "epoch": 80.0,
-      "learning_rate": 3.157894736842105e-05,
-      "loss": 0.0363,
-      "step": 1360
-    },
-    {
-      "epoch": 80.0,
-      "eval_accuracy_NAT": 0.9934704233889913,
-      "eval_accuracy_REST": 0.22926829268292684,
-      "eval_accuracy_concrete": 0.9454731550857731,
-      "eval_accuracy_concrete tiles": 0.9084138581192552,
-      "eval_accuracy_dark bricks": 0.8572009647225466,
-      "eval_accuracy_grass": 0.897609639411522,
-      "eval_accuracy_red bricks": 0.8853251318101933,
-      "eval_accuracy_road": 0.9826731601789439,
-      "eval_accuracy_rough concrete": 0.6364833540898229,
-      "eval_accuracy_rough red sidewalk": 0.7868009234540793,
-      "eval_accuracy_speedway bricks": 0.9907786695620524,
-      "eval_accuracy_steel": 0.8975195789831424,
-      "eval_accuracy_tiles": 0.22757429791874942,
-      "eval_accuracy_unlabeled": NaN,
-      "eval_iou_NAT": 0.9823985430335701,
-      "eval_iou_REST": 0.2266881028938907,
-      "eval_iou_concrete": 0.8881201981762045,
-      "eval_iou_concrete tiles": 0.8518952370427672,
-      "eval_iou_dark bricks": 0.8386950272259497,
-      "eval_iou_grass": 0.835193096016228,
-      "eval_iou_red bricks": 0.8649460750120728,
-      "eval_iou_road": 0.9614746893655648,
-      "eval_iou_rough concrete": 0.6243365831988964,
-      "eval_iou_rough red sidewalk": 0.6142808496647605,
-      "eval_iou_speedway bricks": 0.9772434400055711,
-      "eval_iou_steel": 0.8497443650292307,
-      "eval_iou_tiles": 0.2253284540164377,
-      "eval_iou_unlabeled": NaN,
-      "eval_loss": 0.07582816481590271,
-      "eval_mean_accuracy": 0.7875839576467689,
-      "eval_mean_iou": 0.7492572815908572,
-      "eval_overall_accuracy": 0.9789746817895921,
-      "eval_runtime": 5.7191,
-      "eval_samples_per_second": 20.633,
-      "eval_steps_per_second": 1.399,
-      "step": 1360
-    }
-  ],
-  "max_steps": 3400,
-  "num_train_epochs": 200,
-  "total_flos": 8.591048192714342e+18,
-  "trial_name": null,
-  "trial_params": null
-}