diff --git "a/tmp-checkpoint-1040/trainer_state.json" "b/tmp-checkpoint-1040/trainer_state.json"
new file mode 100644--- /dev/null
+++ "b/tmp-checkpoint-1040/trainer_state.json"
@@ -0,0 +1,8081 @@
+{
+  "best_metric": 0.2938700318336487,
+  "best_model_checkpoint": "segments-ECHO\\checkpoint-980",
+  "epoch": 6.5,
+  "eval_steps": 20,
+  "global_step": 1040,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 0.01,
+      "learning_rate": 9.9975e-05,
+      "loss": 2.5254,
+      "step": 1
+    },
+    {
+      "epoch": 0.01,
+      "learning_rate": 9.995e-05,
+      "loss": 2.5039,
+      "step": 2
+    },
+    {
+      "epoch": 0.02,
+      "learning_rate": 9.9925e-05,
+      "loss": 2.4627,
+      "step": 3
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 9.99e-05,
+      "loss": 2.4211,
+      "step": 4
+    },
+    {
+      "epoch": 0.03,
+      "learning_rate": 9.9875e-05,
+      "loss": 2.3994,
+      "step": 5
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.985000000000001e-05,
+      "loss": 2.4166,
+      "step": 6
+    },
+    {
+      "epoch": 0.04,
+      "learning_rate": 9.9825e-05,
+      "loss": 2.3561,
+      "step": 7
+    },
+    {
+      "epoch": 0.05,
+      "learning_rate": 9.98e-05,
+      "loss": 2.2661,
+      "step": 8
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 9.977500000000001e-05,
+      "loss": 2.3066,
+      "step": 9
+    },
+    {
+      "epoch": 0.06,
+      "learning_rate": 9.975000000000001e-05,
+      "loss": 2.2332,
+      "step": 10
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 9.9725e-05,
+      "loss": 2.2073,
+      "step": 11
+    },
+    {
+      "epoch": 0.07,
+      "learning_rate": 9.970000000000001e-05,
+      "loss": 2.2208,
+      "step": 12
+    },
+    {
+      "epoch": 0.08,
+      "learning_rate": 9.967500000000001e-05,
+      "loss": 2.1216,
+      "step": 13
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 9.965000000000001e-05,
+      "loss": 2.1051,
+      "step": 14
+    },
+    {
+      "epoch": 0.09,
+      "learning_rate": 9.9625e-05,
+      "loss": 2.1227,
+      "step": 15
+    },
+    {
+      "epoch": 0.1,
+      "learning_rate": 9.960000000000001e-05,
+      "loss": 2.0446,
+      "step": 16
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 9.957500000000001e-05,
+      "loss": 2.0619,
+      "step": 17
+    },
+    {
+      "epoch": 0.11,
+      "learning_rate": 9.955000000000001e-05,
+      "loss": 2.015,
+      "step": 18
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 9.952500000000001e-05,
+      "loss": 1.9739,
+      "step": 19
+    },
+    {
+      "epoch": 0.12,
+      "learning_rate": 9.95e-05,
+      "loss": 2.0322,
+      "step": 20
+    },
+    {
+      "epoch": 0.12,
+      "eval_accuracy_AK": 0.6263424445960992,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.017283506468198647,
+      "eval_accuracy_LV": 0.3903317252296394,
+      "eval_accuracy_MK": 0.0010812535450935904,
+      "eval_accuracy_RA": 0.08495028901734104,
+      "eval_accuracy_RV": 0.46802064873474486,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.10870013111069489,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3646572320303637,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.017185325401453597,
+      "eval_iou_LV": 0.29703004778944986,
+      "eval_iou_MK": 0.0010515790925389602,
+      "eval_iou_RA": 0.07824425855254191,
+      "eval_iou_RV": 0.20849867275659106,
+      "eval_iou_TK": 0.0,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.08233900373139623,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 2.2124316692352295,
+      "eval_mean_accuracy": 0.1885233331890902,
+      "eval_mean_iou": 0.09536419266857593,
+      "eval_overall_accuracy": 0.30325268684745055,
+      "eval_runtime": 4.3378,
+      "eval_samples_per_second": 18.673,
+      "eval_steps_per_second": 9.452,
+      "step": 20
+    },
+    {
+      "epoch": 0.13,
+      "learning_rate": 9.9475e-05,
+      "loss": 1.9305,
+      "step": 21
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 9.945e-05,
+      "loss": 1.8974,
+      "step": 22
+    },
+    {
+      "epoch": 0.14,
+      "learning_rate": 9.9425e-05,
+      "loss": 1.8201,
+      "step": 23
+    },
+    {
+      "epoch": 0.15,
+      "learning_rate": 9.94e-05,
+      "loss": 1.9195,
+      "step": 24
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 9.9375e-05,
+      "loss": 1.8381,
+      "step": 25
+    },
+    {
+      "epoch": 0.16,
+      "learning_rate": 9.935000000000002e-05,
+      "loss": 1.8615,
+      "step": 26
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 9.9325e-05,
+      "loss": 1.9896,
+      "step": 27
+    },
+    {
+      "epoch": 0.17,
+      "learning_rate": 9.93e-05,
+      "loss": 1.7221,
+      "step": 28
+    },
+    {
+      "epoch": 0.18,
+      "learning_rate": 9.9275e-05,
+      "loss": 1.7829,
+      "step": 29
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 9.925000000000001e-05,
+      "loss": 1.7313,
+      "step": 30
+    },
+    {
+      "epoch": 0.19,
+      "learning_rate": 9.9225e-05,
+      "loss": 1.7434,
+      "step": 31
+    },
+    {
+      "epoch": 0.2,
+      "learning_rate": 9.92e-05,
+      "loss": 1.7066,
+      "step": 32
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 9.917500000000001e-05,
+      "loss": 1.7412,
+      "step": 33
+    },
+    {
+      "epoch": 0.21,
+      "learning_rate": 9.915000000000001e-05,
+      "loss": 1.6489,
+      "step": 34
+    },
+    {
+      "epoch": 0.22,
+      "learning_rate": 9.9125e-05,
+      "loss": 1.7579,
+      "step": 35
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 9.910000000000001e-05,
+      "loss": 1.5982,
+      "step": 36
+    },
+    {
+      "epoch": 0.23,
+      "learning_rate": 9.907500000000001e-05,
+      "loss": 1.6152,
+      "step": 37
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 9.905000000000001e-05,
+      "loss": 1.5696,
+      "step": 38
+    },
+    {
+      "epoch": 0.24,
+      "learning_rate": 9.9025e-05,
+      "loss": 1.5817,
+      "step": 39
+    },
+    {
+      "epoch": 0.25,
+      "learning_rate": 9.900000000000001e-05,
+      "loss": 1.6027,
+      "step": 40
+    },
+    {
+      "epoch": 0.25,
+      "eval_accuracy_AK": 0.4180375465440657,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 8.349196807267141e-05,
+      "eval_accuracy_LA": 0.08391232330500217,
+      "eval_accuracy_LV": 0.5149157899270969,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.026379190751445087,
+      "eval_accuracy_RV": 0.006093098673401302,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0014422176437535118,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3342379533570258,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 8.348987685243165e-05,
+      "eval_iou_LA": 0.07866686340848168,
+      "eval_iou_LV": 0.3418158910572593,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.026221859593881795,
+      "eval_iou_RV": 0.006072132718195484,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0014407739014099,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 1.5649173259735107,
+      "eval_mean_accuracy": 0.11676262875698191,
+      "eval_mean_iou": 0.07885389639131064,
+      "eval_overall_accuracy": 0.2640260828145788,
+      "eval_runtime": 4.3349,
+      "eval_samples_per_second": 18.686,
+      "eval_steps_per_second": 9.458,
+      "step": 40
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 9.897500000000001e-05,
+      "loss": 1.6129,
+      "step": 41
+    },
+    {
+      "epoch": 0.26,
+      "learning_rate": 9.895e-05,
+      "loss": 1.5371,
+      "step": 42
+    },
+    {
+      "epoch": 0.27,
+      "learning_rate": 9.8925e-05,
+      "loss": 1.4685,
+      "step": 43
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 9.89e-05,
+      "loss": 1.4916,
+      "step": 44
+    },
+    {
+      "epoch": 0.28,
+      "learning_rate": 9.8875e-05,
+      "loss": 1.4807,
+      "step": 45
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 9.885e-05,
+      "loss": 1.6242,
+      "step": 46
+    },
+    {
+      "epoch": 0.29,
+      "learning_rate": 9.8825e-05,
+      "loss": 1.4578,
+      "step": 47
+    },
+    {
+      "epoch": 0.3,
+      "learning_rate": 9.88e-05,
+      "loss": 1.3899,
+      "step": 48
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 9.8775e-05,
+      "loss": 1.4259,
+      "step": 49
+    },
+    {
+      "epoch": 0.31,
+      "learning_rate": 9.875000000000002e-05,
+      "loss": 1.3936,
+      "step": 50
+    },
+    {
+      "epoch": 0.32,
+      "learning_rate": 9.8725e-05,
+      "loss": 1.4484,
+      "step": 51
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 9.87e-05,
+      "loss": 1.4465,
+      "step": 52
+    },
+    {
+      "epoch": 0.33,
+      "learning_rate": 9.8675e-05,
+      "loss": 1.3384,
+      "step": 53
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 9.865000000000001e-05,
+      "loss": 1.3987,
+      "step": 54
+    },
+    {
+      "epoch": 0.34,
+      "learning_rate": 9.8625e-05,
+      "loss": 1.325,
+      "step": 55
+    },
+    {
+      "epoch": 0.35,
+      "learning_rate": 9.86e-05,
+      "loss": 1.3986,
+      "step": 56
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 9.857500000000001e-05,
+      "loss": 1.3668,
+      "step": 57
+    },
+    {
+      "epoch": 0.36,
+      "learning_rate": 9.855000000000001e-05,
+      "loss": 1.3011,
+      "step": 58
+    },
+    {
+      "epoch": 0.37,
+      "learning_rate": 9.8525e-05,
+      "loss": 1.3793,
+      "step": 59
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 9.850000000000001e-05,
+      "loss": 1.2877,
+      "step": 60
+    },
+    {
+      "epoch": 0.38,
+      "eval_accuracy_AK": 0.39300851262526487,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.24209088456176642,
+      "eval_accuracy_LV": 0.46650240138698945,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.05467976878612717,
+      "eval_accuracy_RV": 0.005259041892968094,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.005047761753137292,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.33124112303256126,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.18770266500854244,
+      "eval_iou_LV": 0.3611907523728232,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.0528826157540955,
+      "eval_iou_RV": 0.005251120698764411,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.004983818770226537,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 1.261634111404419,
+      "eval_mean_accuracy": 0.12962093011180592,
+      "eval_mean_iou": 0.09432520956370134,
+      "eval_overall_accuracy": 0.26848080197537355,
+      "eval_runtime": 4.3522,
+      "eval_samples_per_second": 18.611,
+      "eval_steps_per_second": 9.421,
+      "step": 60
+    },
+    {
+      "epoch": 0.38,
+      "learning_rate": 9.847500000000001e-05,
+      "loss": 1.3054,
+      "step": 61
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 9.845000000000001e-05,
+      "loss": 1.2689,
+      "step": 62
+    },
+    {
+      "epoch": 0.39,
+      "learning_rate": 9.842500000000001e-05,
+      "loss": 1.2455,
+      "step": 63
+    },
+    {
+      "epoch": 0.4,
+      "learning_rate": 9.84e-05,
+      "loss": 1.3569,
+      "step": 64
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 9.8375e-05,
+      "loss": 1.3127,
+      "step": 65
+    },
+    {
+      "epoch": 0.41,
+      "learning_rate": 9.835e-05,
+      "loss": 1.2319,
+      "step": 66
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 9.8325e-05,
+      "loss": 1.1801,
+      "step": 67
+    },
+    {
+      "epoch": 0.42,
+      "learning_rate": 9.83e-05,
+      "loss": 1.2914,
+      "step": 68
+    },
+    {
+      "epoch": 0.43,
+      "learning_rate": 9.8275e-05,
+      "loss": 1.2394,
+      "step": 69
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 9.825e-05,
+      "loss": 1.1518,
+      "step": 70
+    },
+    {
+      "epoch": 0.44,
+      "learning_rate": 9.8225e-05,
+      "loss": 1.266,
+      "step": 71
+    },
+    {
+      "epoch": 0.45,
+      "learning_rate": 9.82e-05,
+      "loss": 1.234,
+      "step": 72
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 9.8175e-05,
+      "loss": 1.1306,
+      "step": 73
+    },
+    {
+      "epoch": 0.46,
+      "learning_rate": 9.815000000000001e-05,
+      "loss": 1.106,
+      "step": 74
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 9.8125e-05,
+      "loss": 1.0982,
+      "step": 75
+    },
+    {
+      "epoch": 0.47,
+      "learning_rate": 9.81e-05,
+      "loss": 1.2036,
+      "step": 76
+    },
+    {
+      "epoch": 0.48,
+      "learning_rate": 9.807500000000001e-05,
+      "loss": 1.1903,
+      "step": 77
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 9.805000000000001e-05,
+      "loss": 1.2332,
+      "step": 78
+    },
+    {
+      "epoch": 0.49,
+      "learning_rate": 9.8025e-05,
+      "loss": 1.1477,
+      "step": 79
+    },
+    {
+      "epoch": 0.5,
+      "learning_rate": 9.8e-05,
+      "loss": 1.0981,
+      "step": 80
+    },
+    {
+      "epoch": 0.5,
+      "eval_accuracy_AK": 0.4862672557259677,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.07940032378468319,
+      "eval_accuracy_LV": 0.8150770489861409,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.008226589595375722,
+      "eval_accuracy_RV": 0.007904705873912139,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.4019844608304053,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.07502230751798765,
+      "eval_iou_LV": 0.47365251006325026,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.008195399704246958,
+      "eval_iou_RV": 0.00790374185772906,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 1.2207798957824707,
+      "eval_mean_accuracy": 0.15520843599623105,
+      "eval_mean_iou": 0.09667584199736193,
+      "eval_overall_accuracy": 0.3897765659817464,
+      "eval_runtime": 4.3143,
+      "eval_samples_per_second": 18.775,
+      "eval_steps_per_second": 9.503,
+      "step": 80
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 9.797500000000001e-05,
+      "loss": 1.0832,
+      "step": 81
+    },
+    {
+      "epoch": 0.51,
+      "learning_rate": 9.795000000000001e-05,
+      "loss": 1.1685,
+      "step": 82
+    },
+    {
+      "epoch": 0.52,
+      "learning_rate": 9.7925e-05,
+      "loss": 1.1672,
+      "step": 83
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 9.790000000000001e-05,
+      "loss": 1.0383,
+      "step": 84
+    },
+    {
+      "epoch": 0.53,
+      "learning_rate": 9.787500000000001e-05,
+      "loss": 1.0307,
+      "step": 85
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 9.785e-05,
+      "loss": 1.0283,
+      "step": 86
+    },
+    {
+      "epoch": 0.54,
+      "learning_rate": 9.7825e-05,
+      "loss": 1.0414,
+      "step": 87
+    },
+    {
+      "epoch": 0.55,
+      "learning_rate": 9.78e-05,
+      "loss": 1.0416,
+      "step": 88
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 9.7775e-05,
+      "loss": 0.9611,
+      "step": 89
+    },
+    {
+      "epoch": 0.56,
+      "learning_rate": 9.775e-05,
+      "loss": 0.9732,
+      "step": 90
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 9.7725e-05,
+      "loss": 0.9857,
+      "step": 91
+    },
+    {
+      "epoch": 0.57,
+      "learning_rate": 9.77e-05,
+      "loss": 0.9002,
+      "step": 92
+    },
+    {
+      "epoch": 0.58,
+      "learning_rate": 9.7675e-05,
+      "loss": 1.0053,
+      "step": 93
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 9.765e-05,
+      "loss": 0.9788,
+      "step": 94
+    },
+    {
+      "epoch": 0.59,
+      "learning_rate": 9.7625e-05,
+      "loss": 1.0315,
+      "step": 95
+    },
+    {
+      "epoch": 0.6,
+      "learning_rate": 9.76e-05,
+      "loss": 0.9625,
+      "step": 96
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 9.7575e-05,
+      "loss": 1.0484,
+      "step": 97
+    },
+    {
+      "epoch": 0.61,
+      "learning_rate": 9.755000000000001e-05,
+      "loss": 0.9898,
+      "step": 98
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 9.7525e-05,
+      "loss": 0.972,
+      "step": 99
+    },
+    {
+      "epoch": 0.62,
+      "learning_rate": 9.75e-05,
+      "loss": 1.0235,
+      "step": 100
+    },
+    {
+      "epoch": 0.62,
+      "eval_accuracy_AK": 0.4947409695532967,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.3015038749231761,
+      "eval_accuracy_LV": 0.8507606635581448,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.0422728323699422,
+      "eval_accuracy_RV": 0.01015217500484291,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.4058522236190776,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.22833526085383246,
+      "eval_iou_LV": 0.5319480132542299,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.041750206661581954,
+      "eval_iou_RV": 0.010136375439884668,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.9343371987342834,
+      "eval_mean_accuracy": 0.18882561282326696,
+      "eval_mean_iou": 0.12180220798286065,
+      "eval_overall_accuracy": 0.4418879928168235,
+      "eval_runtime": 4.3249,
+      "eval_samples_per_second": 18.729,
+      "eval_steps_per_second": 9.48,
+      "step": 100
+    },
+    {
+      "epoch": 0.63,
+      "learning_rate": 9.747500000000001e-05,
+      "loss": 0.9706,
+      "step": 101
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 9.745000000000001e-05,
+      "loss": 0.8611,
+      "step": 102
+    },
+    {
+      "epoch": 0.64,
+      "learning_rate": 9.7425e-05,
+      "loss": 0.9307,
+      "step": 103
+    },
+    {
+      "epoch": 0.65,
+      "learning_rate": 9.74e-05,
+      "loss": 0.9535,
+      "step": 104
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 9.737500000000001e-05,
+      "loss": 1.0598,
+      "step": 105
+    },
+    {
+      "epoch": 0.66,
+      "learning_rate": 9.735000000000001e-05,
+      "loss": 0.9147,
+      "step": 106
+    },
+    {
+      "epoch": 0.67,
+      "learning_rate": 9.7325e-05,
+      "loss": 1.0839,
+      "step": 107
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 9.730000000000001e-05,
+      "loss": 0.9645,
+      "step": 108
+    },
+    {
+      "epoch": 0.68,
+      "learning_rate": 9.7275e-05,
+      "loss": 0.853,
+      "step": 109
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 9.725e-05,
+      "loss": 0.961,
+      "step": 110
+    },
+    {
+      "epoch": 0.69,
+      "learning_rate": 9.7225e-05,
+      "loss": 0.8446,
+      "step": 111
+    },
+    {
+      "epoch": 0.7,
+      "learning_rate": 9.72e-05,
+      "loss": 0.9368,
+      "step": 112
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 9.7175e-05,
+      "loss": 1.0214,
+      "step": 113
+    },
+    {
+      "epoch": 0.71,
+      "learning_rate": 9.715e-05,
+      "loss": 0.8252,
+      "step": 114
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 9.7125e-05,
+      "loss": 0.9631,
+      "step": 115
+    },
+    {
+      "epoch": 0.72,
+      "learning_rate": 9.71e-05,
+      "loss": 0.8311,
+      "step": 116
+    },
+    {
+      "epoch": 0.73,
+      "learning_rate": 9.7075e-05,
+      "loss": 1.0585,
+      "step": 117
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 9.705e-05,
+      "loss": 0.938,
+      "step": 118
+    },
+    {
+      "epoch": 0.74,
+      "learning_rate": 9.7025e-05,
+      "loss": 0.8328,
+      "step": 119
+    },
+    {
+      "epoch": 0.75,
+      "learning_rate": 9.7e-05,
+      "loss": 0.8977,
+      "step": 120
+    },
+    {
+      "epoch": 0.75,
+      "eval_accuracy_AK": 0.4762520503334411,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.41777967651511744,
+      "eval_accuracy_LV": 0.8123734695196149,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.11882312138728324,
+      "eval_accuracy_RV": 0.17874643956406633,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3906352807157097,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.2994674528396919,
+      "eval_iou_LV": 0.6151112701084495,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.11241728554397401,
+      "eval_iou_RV": 0.17412438954072496,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.7805922031402588,
+      "eval_mean_accuracy": 0.22266386192439144,
+      "eval_mean_iou": 0.159175567874855,
+      "eval_overall_accuracy": 0.47636504990446155,
+      "eval_runtime": 4.334,
+      "eval_samples_per_second": 18.69,
+      "eval_steps_per_second": 9.46,
+      "step": 120
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 9.6975e-05,
+      "loss": 0.7832,
+      "step": 121
+    },
+    {
+      "epoch": 0.76,
+      "learning_rate": 9.695000000000001e-05,
+      "loss": 0.7775,
+      "step": 122
+    },
+    {
+      "epoch": 0.77,
+      "learning_rate": 9.6925e-05,
+      "loss": 0.7412,
+      "step": 123
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 9.69e-05,
+      "loss": 0.7278,
+      "step": 124
+    },
+    {
+      "epoch": 0.78,
+      "learning_rate": 9.687500000000001e-05,
+      "loss": 0.873,
+      "step": 125
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 9.685000000000001e-05,
+      "loss": 0.8006,
+      "step": 126
+    },
+    {
+      "epoch": 0.79,
+      "learning_rate": 9.682500000000001e-05,
+      "loss": 0.8247,
+      "step": 127
+    },
+    {
+      "epoch": 0.8,
+      "learning_rate": 9.680000000000001e-05,
+      "loss": 0.7759,
+      "step": 128
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 9.677500000000001e-05,
+      "loss": 0.9544,
+      "step": 129
+    },
+    {
+      "epoch": 0.81,
+      "learning_rate": 9.675000000000001e-05,
+      "loss": 0.7949,
+      "step": 130
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 9.6725e-05,
+      "loss": 0.7626,
+      "step": 131
+    },
+    {
+      "epoch": 0.82,
+      "learning_rate": 9.67e-05,
+      "loss": 0.7742,
+      "step": 132
+    },
+    {
+      "epoch": 0.83,
+      "learning_rate": 9.6675e-05,
+      "loss": 0.7878,
+      "step": 133
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 9.665e-05,
+      "loss": 0.8039,
+      "step": 134
+    },
+    {
+      "epoch": 0.84,
+      "learning_rate": 9.6625e-05,
+      "loss": 0.7243,
+      "step": 135
+    },
+    {
+      "epoch": 0.85,
+      "learning_rate": 9.66e-05,
+      "loss": 0.8162,
+      "step": 136
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 9.6575e-05,
+      "loss": 0.7009,
+      "step": 137
+    },
+    {
+      "epoch": 0.86,
+      "learning_rate": 9.655e-05,
+      "loss": 0.8104,
+      "step": 138
+    },
+    {
+      "epoch": 0.87,
+      "learning_rate": 9.652500000000002e-05,
+      "loss": 0.6472,
+      "step": 139
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 9.65e-05,
+      "loss": 0.6932,
+      "step": 140
+    },
+    {
+      "epoch": 0.88,
+      "eval_accuracy_AK": 0.12756067191073117,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.2809544153138163,
+      "eval_accuracy_LV": 0.6794323358065106,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.14148901734104047,
+      "eval_accuracy_RV": 0.20193142438961392,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.12167004028355512,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.2414392770766174,
+      "eval_iou_LV": 0.5674467912887458,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.13723674089091525,
+      "eval_iou_RV": 0.19411884390830997,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.624600350856781,
+      "eval_mean_accuracy": 0.1590408738624125,
+      "eval_mean_iou": 0.12619116934481434,
+      "eval_overall_accuracy": 0.3766346927713812,
+      "eval_runtime": 4.2359,
+      "eval_samples_per_second": 19.122,
+      "eval_steps_per_second": 9.679,
+      "step": 140
+    },
+    {
+      "epoch": 0.88,
+      "learning_rate": 9.6475e-05,
+      "loss": 0.6345,
+      "step": 141
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 9.645000000000001e-05,
+      "loss": 0.7438,
+      "step": 142
+    },
+    {
+      "epoch": 0.89,
+      "learning_rate": 9.642500000000001e-05,
+      "loss": 0.7369,
+      "step": 143
+    },
+    {
+      "epoch": 0.9,
+      "learning_rate": 9.64e-05,
+      "loss": 0.6622,
+      "step": 144
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 9.6375e-05,
+      "loss": 0.7326,
+      "step": 145
+    },
+    {
+      "epoch": 0.91,
+      "learning_rate": 9.635000000000001e-05,
+      "loss": 0.732,
+      "step": 146
+    },
+    {
+      "epoch": 0.92,
+      "learning_rate": 9.632500000000001e-05,
+      "loss": 0.6986,
+      "step": 147
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 9.63e-05,
+      "loss": 0.6736,
+      "step": 148
+    },
+    {
+      "epoch": 0.93,
+      "learning_rate": 9.627500000000001e-05,
+      "loss": 0.6711,
+      "step": 149
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 9.625000000000001e-05,
+      "loss": 0.6661,
+      "step": 150
+    },
+    {
+      "epoch": 0.94,
+      "learning_rate": 9.622500000000001e-05,
+      "loss": 0.7122,
+      "step": 151
+    },
+    {
+      "epoch": 0.95,
+      "learning_rate": 9.620000000000001e-05,
+      "loss": 0.6453,
+      "step": 152
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 9.6175e-05,
+      "loss": 0.7258,
+      "step": 153
+    },
+    {
+      "epoch": 0.96,
+      "learning_rate": 9.615e-05,
+      "loss": 0.7111,
+      "step": 154
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 9.6125e-05,
+      "loss": 0.6233,
+      "step": 155
+    },
+    {
+      "epoch": 0.97,
+      "learning_rate": 9.61e-05,
+      "loss": 0.6888,
+      "step": 156
+    },
+    {
+      "epoch": 0.98,
+      "learning_rate": 9.6075e-05,
+      "loss": 0.7224,
+      "step": 157
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 9.605e-05,
+      "loss": 0.5711,
+      "step": 158
+    },
+    {
+      "epoch": 0.99,
+      "learning_rate": 9.6025e-05,
+      "loss": 0.5744,
+      "step": 159
+    },
+    {
+      "epoch": 1.0,
+      "learning_rate": 9.6e-05,
+      "loss": 0.6168,
+      "step": 160
+    },
+    {
+      "epoch": 1.0,
+      "eval_accuracy_AK": 0.3864384660512673,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.43749531561511595,
+      "eval_accuracy_LV": 0.7500424011716184,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.14906820809248555,
+      "eval_accuracy_RV": 0.32609467710343737,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.34388650940782883,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.3522248822955276,
+      "eval_iou_LV": 0.6106392095578969,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.14754837454657801,
+      "eval_iou_RV": 0.29734260564647236,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.6124103665351868,
+      "eval_mean_accuracy": 0.22768211867043606,
+      "eval_mean_iou": 0.17516415814543035,
+      "eval_overall_accuracy": 0.4717479975253081,
+      "eval_runtime": 4.2982,
+      "eval_samples_per_second": 18.845,
+      "eval_steps_per_second": 9.539,
+      "step": 160
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 9.5975e-05,
+      "loss": 0.645,
+      "step": 161
+    },
+    {
+      "epoch": 1.01,
+      "learning_rate": 9.595e-05,
+      "loss": 0.5848,
+      "step": 162
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 9.592500000000001e-05,
+      "loss": 0.5205,
+      "step": 163
+    },
+    {
+      "epoch": 1.02,
+      "learning_rate": 9.59e-05,
+      "loss": 0.585,
+      "step": 164
+    },
+    {
+      "epoch": 1.03,
+      "learning_rate": 9.5875e-05,
+      "loss": 0.5965,
+      "step": 165
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 9.585000000000001e-05,
+      "loss": 0.5869,
+      "step": 166
+    },
+    {
+      "epoch": 1.04,
+      "learning_rate": 9.582500000000001e-05,
+      "loss": 0.6747,
+      "step": 167
+    },
+    {
+      "epoch": 1.05,
+      "learning_rate": 9.58e-05,
+      "loss": 0.7185,
+      "step": 168
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 9.5775e-05,
+      "loss": 0.4734,
+      "step": 169
+    },
+    {
+      "epoch": 1.06,
+      "learning_rate": 9.575000000000001e-05,
+      "loss": 0.8814,
+      "step": 170
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 9.572500000000001e-05,
+      "loss": 0.6958,
+      "step": 171
+    },
+    {
+      "epoch": 1.07,
+      "learning_rate": 9.57e-05,
+      "loss": 0.4995,
+      "step": 172
+    },
+    {
+      "epoch": 1.08,
+      "learning_rate": 9.567500000000001e-05,
+      "loss": 0.7137,
+      "step": 173
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 9.565000000000001e-05,
+      "loss": 0.7256,
+      "step": 174
+    },
+    {
+      "epoch": 1.09,
+      "learning_rate": 9.562500000000001e-05,
+      "loss": 0.6362,
+      "step": 175
+    },
+    {
+      "epoch": 1.1,
+      "learning_rate": 9.56e-05,
+      "loss": 0.4946,
+      "step": 176
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 9.5575e-05,
+      "loss": 0.8409,
+      "step": 177
+    },
+    {
+      "epoch": 1.11,
+      "learning_rate": 9.555e-05,
+      "loss": 0.6697,
+      "step": 178
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 9.5525e-05,
+      "loss": 0.5507,
+      "step": 179
+    },
+    {
+      "epoch": 1.12,
+      "learning_rate": 9.55e-05,
+      "loss": 0.5758,
+      "step": 180
+    },
+    {
+      "epoch": 1.12,
+      "eval_accuracy_AK": 0.40478670545837675,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.44330395287134056,
+      "eval_accuracy_LV": 0.6646006751881804,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.35957687861271675,
+      "eval_accuracy_RV": 0.395497169587958,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.36230851251935636,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.3865133809179174,
+      "eval_iou_LV": 0.6047703331700147,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.3133212450891508,
+      "eval_iou_RV": 0.37052795730365023,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.5658473968505859,
+      "eval_mean_accuracy": 0.25197393130206364,
+      "eval_mean_iou": 0.20374414290000895,
+      "eval_overall_accuracy": 0.4750379142508308,
+      "eval_runtime": 4.316,
+      "eval_samples_per_second": 18.768,
+      "eval_steps_per_second": 9.5,
+      "step": 180
+    },
+    {
+      "epoch": 1.13,
+      "learning_rate": 9.5475e-05,
+      "loss": 0.5877,
+      "step": 181
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 9.545e-05,
+      "loss": 0.6592,
+      "step": 182
+    },
+    {
+      "epoch": 1.14,
+      "learning_rate": 9.542500000000002e-05,
+      "loss": 0.5944,
+      "step": 183
+    },
+    {
+      "epoch": 1.15,
+      "learning_rate": 9.54e-05,
+      "loss": 0.5967,
+      "step": 184
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 9.5375e-05,
+      "loss": 0.507,
+      "step": 185
+    },
+    {
+      "epoch": 1.16,
+      "learning_rate": 9.535e-05,
+      "loss": 0.5218,
+      "step": 186
+    },
+    {
+      "epoch": 1.17,
+      "learning_rate": 9.532500000000001e-05,
+      "loss": 0.5396,
+      "step": 187
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 9.53e-05,
+      "loss": 0.6345,
+      "step": 188
+    },
+    {
+      "epoch": 1.18,
+      "learning_rate": 9.5275e-05,
+      "loss": 0.6127,
+      "step": 189
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 9.525000000000001e-05,
+      "loss": 0.5602,
+      "step": 190
+    },
+    {
+      "epoch": 1.19,
+      "learning_rate": 9.522500000000001e-05,
+      "loss": 0.4613,
+      "step": 191
+    },
+    {
+      "epoch": 1.2,
+      "learning_rate": 9.52e-05,
+      "loss": 0.565,
+      "step": 192
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 9.517500000000001e-05,
+      "loss": 0.5524,
+      "step": 193
+    },
+    {
+      "epoch": 1.21,
+      "learning_rate": 9.515000000000001e-05,
+      "loss": 0.5632,
+      "step": 194
+    },
+    {
+      "epoch": 1.22,
+      "learning_rate": 9.512500000000001e-05,
+      "loss": 0.5295,
+      "step": 195
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 9.51e-05,
+      "loss": 0.5285,
+      "step": 196
+    },
+    {
+      "epoch": 1.23,
+      "learning_rate": 9.507500000000001e-05,
+      "loss": 0.5475,
+      "step": 197
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 9.505e-05,
+      "loss": 0.7287,
+      "step": 198
+    },
+    {
+      "epoch": 1.24,
+      "learning_rate": 9.5025e-05,
+      "loss": 0.5132,
+      "step": 199
+    },
+    {
+      "epoch": 1.25,
+      "learning_rate": 9.5e-05,
+      "loss": 0.5081,
+      "step": 200
+    },
+    {
+      "epoch": 1.25,
+      "eval_accuracy_AK": 0.43958550339427466,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.6477024029020702,
+      "eval_accuracy_LV": 0.6459616585723055,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.4741248554913295,
+      "eval_accuracy_RV": 0.4867411876968553,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3965160199902263,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.46107110320882794,
+      "eval_iou_LV": 0.6097595840464746,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.39607477285369924,
+      "eval_iou_RV": 0.4523309463802323,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.5116393566131592,
+      "eval_mean_accuracy": 0.2993461786729817,
+      "eval_mean_iou": 0.23157524264794604,
+      "eval_overall_accuracy": 0.5279889295862557,
+      "eval_runtime": 4.3327,
+      "eval_samples_per_second": 18.695,
+      "eval_steps_per_second": 9.463,
+      "step": 200
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 9.4975e-05,
+      "loss": 0.5719,
+      "step": 201
+    },
+    {
+      "epoch": 1.26,
+      "learning_rate": 9.495e-05,
+      "loss": 0.5619,
+      "step": 202
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 9.4925e-05,
+      "loss": 0.4706,
+      "step": 203
+    },
+    {
+      "epoch": 1.27,
+      "learning_rate": 9.49e-05,
+      "loss": 0.5288,
+      "step": 204
+    },
+    {
+      "epoch": 1.28,
+      "learning_rate": 9.4875e-05,
+      "loss": 0.5602,
+      "step": 205
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 9.485e-05,
+      "loss": 0.6679,
+      "step": 206
+    },
+    {
+      "epoch": 1.29,
+      "learning_rate": 9.482500000000001e-05,
+      "loss": 0.5069,
+      "step": 207
+    },
+    {
+      "epoch": 1.3,
+      "learning_rate": 9.48e-05,
+      "loss": 0.4546,
+      "step": 208
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 9.4775e-05,
+      "loss": 0.4362,
+      "step": 209
+    },
+    {
+      "epoch": 1.31,
+      "learning_rate": 9.475e-05,
+      "loss": 0.5631,
+      "step": 210
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 9.472500000000001e-05,
+      "loss": 0.7127,
+      "step": 211
+    },
+    {
+      "epoch": 1.32,
+      "learning_rate": 9.47e-05,
+      "loss": 0.5257,
+      "step": 212
+    },
+    {
+      "epoch": 1.33,
+      "learning_rate": 9.4675e-05,
+      "loss": 0.4348,
+      "step": 213
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 9.465000000000001e-05,
+      "loss": 0.46,
+      "step": 214
+    },
+    {
+      "epoch": 1.34,
+      "learning_rate": 9.462500000000001e-05,
+      "loss": 0.4745,
+      "step": 215
+    },
+    {
+      "epoch": 1.35,
+      "learning_rate": 9.46e-05,
+      "loss": 0.3421,
+      "step": 216
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 9.457500000000001e-05,
+      "loss": 0.4425,
+      "step": 217
+    },
+    {
+      "epoch": 1.36,
+      "learning_rate": 9.455000000000001e-05,
+      "loss": 0.5037,
+      "step": 218
+    },
+    {
+      "epoch": 1.37,
+      "learning_rate": 9.452500000000001e-05,
+      "loss": 0.3668,
+      "step": 219
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 9.449999999999999e-05,
+      "loss": 0.6351,
+      "step": 220
+    },
+    {
+      "epoch": 1.38,
+      "eval_accuracy_AK": 0.07825688732444955,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.3882717992534964,
+      "eval_accuracy_LV": 0.37490644820865143,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.26013179190751445,
+      "eval_accuracy_RV": 0.09017857783453749,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.07743706849826289,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.34655040062615083,
+      "eval_iou_LV": 0.3623206410740071,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.25101123344823356,
+      "eval_iou_RV": 0.08965001970388889,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.487935334444046,
+      "eval_mean_accuracy": 0.13241616716984994,
+      "eval_mean_iou": 0.11269693633505433,
+      "eval_overall_accuracy": 0.2609329916944522,
+      "eval_runtime": 4.2777,
+      "eval_samples_per_second": 18.935,
+      "eval_steps_per_second": 9.585,
+      "step": 220
+    },
+    {
+      "epoch": 1.38,
+      "learning_rate": 9.4475e-05,
+      "loss": 0.5288,
+      "step": 221
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 9.445e-05,
+      "loss": 0.4605,
+      "step": 222
+    },
+    {
+      "epoch": 1.39,
+      "learning_rate": 9.4425e-05,
+      "loss": 0.5842,
+      "step": 223
+    },
+    {
+      "epoch": 1.4,
+      "learning_rate": 9.44e-05,
+      "loss": 0.6004,
+      "step": 224
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 9.4375e-05,
+      "loss": 0.3985,
+      "step": 225
+    },
+    {
+      "epoch": 1.41,
+      "learning_rate": 9.435e-05,
+      "loss": 0.6153,
+      "step": 226
+    },
+    {
+      "epoch": 1.42,
+      "learning_rate": 9.4325e-05,
+      "loss": 0.4359,
+      "step": 227
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 9.43e-05,
+      "loss": 0.4356,
+      "step": 228
+    },
+    {
+      "epoch": 1.43,
+      "learning_rate": 9.4275e-05,
+      "loss": 0.4503,
+      "step": 229
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 9.425e-05,
+      "loss": 0.5033,
+      "step": 230
+    },
+    {
+      "epoch": 1.44,
+      "learning_rate": 9.422500000000001e-05,
+      "loss": 0.5512,
+      "step": 231
+    },
+    {
+      "epoch": 1.45,
+      "learning_rate": 9.42e-05,
+      "loss": 0.4235,
+      "step": 232
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 9.4175e-05,
+      "loss": 0.514,
+      "step": 233
+    },
+    {
+      "epoch": 1.46,
+      "learning_rate": 9.415e-05,
+      "loss": 0.7272,
+      "step": 234
+    },
+    {
+      "epoch": 1.47,
+      "learning_rate": 9.412500000000001e-05,
+      "loss": 0.6195,
+      "step": 235
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 9.41e-05,
+      "loss": 0.4825,
+      "step": 236
+    },
+    {
+      "epoch": 1.48,
+      "learning_rate": 9.4075e-05,
+      "loss": 0.4703,
+      "step": 237
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 9.405000000000001e-05,
+      "loss": 0.4063,
+      "step": 238
+    },
+    {
+      "epoch": 1.49,
+      "learning_rate": 9.402500000000001e-05,
+      "loss": 0.4666,
+      "step": 239
+    },
+    {
+      "epoch": 1.5,
+      "learning_rate": 9.4e-05,
+      "loss": 0.6241,
+      "step": 240
+    },
+    {
+      "epoch": 1.5,
+      "eval_accuracy_AK": 0.35967936975443887,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.6915594879405196,
+      "eval_accuracy_LV": 0.743931248182807,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5367190751445087,
+      "eval_accuracy_RV": 0.44921401358884766,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3409631956326203,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.5265270020614439,
+      "eval_iou_LV": 0.6994661008529603,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.4399761940413079,
+      "eval_iou_RV": 0.43220164946389833,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.45926591753959656,
+      "eval_mean_accuracy": 0.3090114660679025,
+      "eval_mean_iou": 0.2439134142052231,
+      "eval_overall_accuracy": 0.5686198665199371,
+      "eval_runtime": 4.2801,
+      "eval_samples_per_second": 18.925,
+      "eval_steps_per_second": 9.579,
+      "step": 240
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 9.397500000000001e-05,
+      "loss": 0.4423,
+      "step": 241
+    },
+    {
+      "epoch": 1.51,
+      "learning_rate": 9.395000000000001e-05,
+      "loss": 0.44,
+      "step": 242
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 9.3925e-05,
+      "loss": 0.6371,
+      "step": 243
+    },
+    {
+      "epoch": 1.52,
+      "learning_rate": 9.39e-05,
+      "loss": 0.3726,
+      "step": 244
+    },
+    {
+      "epoch": 1.53,
+      "learning_rate": 9.3875e-05,
+      "loss": 0.6107,
+      "step": 245
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 9.385e-05,
+      "loss": 0.3358,
+      "step": 246
+    },
+    {
+      "epoch": 1.54,
+      "learning_rate": 9.3825e-05,
+      "loss": 0.5283,
+      "step": 247
+    },
+    {
+      "epoch": 1.55,
+      "learning_rate": 9.38e-05,
+      "loss": 0.4027,
+      "step": 248
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 9.3775e-05,
+      "loss": 0.426,
+      "step": 249
+    },
+    {
+      "epoch": 1.56,
+      "learning_rate": 9.375e-05,
+      "loss": 0.3243,
+      "step": 250
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 9.3725e-05,
+      "loss": 0.385,
+      "step": 251
+    },
+    {
+      "epoch": 1.57,
+      "learning_rate": 9.370000000000001e-05,
+      "loss": 0.3897,
+      "step": 252
+    },
+    {
+      "epoch": 1.58,
+      "learning_rate": 9.3675e-05,
+      "loss": 0.3262,
+      "step": 253
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 9.365e-05,
+      "loss": 0.3444,
+      "step": 254
+    },
+    {
+      "epoch": 1.59,
+      "learning_rate": 9.362500000000001e-05,
+      "loss": 0.3338,
+      "step": 255
+    },
+    {
+      "epoch": 1.6,
+      "learning_rate": 9.360000000000001e-05,
+      "loss": 0.4978,
+      "step": 256
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 9.3575e-05,
+      "loss": 0.4499,
+      "step": 257
+    },
+    {
+      "epoch": 1.61,
+      "learning_rate": 9.355000000000001e-05,
+      "loss": 0.5182,
+      "step": 258
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 9.352500000000001e-05,
+      "loss": 0.4065,
+      "step": 259
+    },
+    {
+      "epoch": 1.62,
+      "learning_rate": 9.350000000000001e-05,
+      "loss": 0.4315,
+      "step": 260
+    },
+    {
+      "epoch": 1.62,
+      "eval_accuracy_AK": 0.30245860421680254,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.5870977050261577,
+      "eval_accuracy_LV": 0.681133766947008,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.42533179190751447,
+      "eval_accuracy_RV": 0.3535450103674155,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.28860423271356356,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.5051754660337873,
+      "eval_iou_LV": 0.63978076233765,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.3951717367440801,
+      "eval_iou_RV": 0.34590916828559043,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.40823444724082947,
+      "eval_mean_accuracy": 0.2610629864960998,
+      "eval_mean_iou": 0.21746413661146713,
+      "eval_overall_accuracy": 0.49479411329927897,
+      "eval_runtime": 4.4554,
+      "eval_samples_per_second": 18.18,
+      "eval_steps_per_second": 9.202,
+      "step": 260
+    },
+    {
+      "epoch": 1.63,
+      "learning_rate": 9.3475e-05,
+      "loss": 0.4592,
+      "step": 261
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 9.345000000000001e-05,
+      "loss": 0.407,
+      "step": 262
+    },
+    {
+      "epoch": 1.64,
+      "learning_rate": 9.342500000000001e-05,
+      "loss": 0.3973,
+      "step": 263
+    },
+    {
+      "epoch": 1.65,
+      "learning_rate": 9.340000000000001e-05,
+      "loss": 0.5056,
+      "step": 264
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 9.3375e-05,
+      "loss": 0.3946,
+      "step": 265
+    },
+    {
+      "epoch": 1.66,
+      "learning_rate": 9.335e-05,
+      "loss": 0.5485,
+      "step": 266
+    },
+    {
+      "epoch": 1.67,
+      "learning_rate": 9.3325e-05,
+      "loss": 0.3967,
+      "step": 267
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 9.33e-05,
+      "loss": 0.4156,
+      "step": 268
+    },
+    {
+      "epoch": 1.68,
+      "learning_rate": 9.3275e-05,
+      "loss": 0.3342,
+      "step": 269
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 9.325e-05,
+      "loss": 0.8152,
+      "step": 270
+    },
+    {
+      "epoch": 1.69,
+      "learning_rate": 9.3225e-05,
+      "loss": 0.302,
+      "step": 271
+    },
+    {
+      "epoch": 1.7,
+      "learning_rate": 9.320000000000002e-05,
+      "loss": 0.398,
+      "step": 272
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 9.3175e-05,
+      "loss": 0.3698,
+      "step": 273
+    },
+    {
+      "epoch": 1.71,
+      "learning_rate": 9.315e-05,
+      "loss": 0.3218,
+      "step": 274
+    },
+    {
+      "epoch": 1.72,
+      "learning_rate": 9.3125e-05,
+      "loss": 0.4393,
+      "step": 275
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 9.310000000000001e-05,
+      "loss": 0.511,
+      "step": 276
+    },
+    {
+      "epoch": 1.73,
+      "learning_rate": 9.3075e-05,
+      "loss": 0.4566,
+      "step": 277
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 9.305e-05,
+      "loss": 0.3512,
+      "step": 278
+    },
+    {
+      "epoch": 1.74,
+      "learning_rate": 9.302500000000001e-05,
+      "loss": 0.4009,
+      "step": 279
+    },
+    {
+      "epoch": 1.75,
+      "learning_rate": 9.300000000000001e-05,
+      "loss": 0.5236,
+      "step": 280
+    },
+    {
+      "epoch": 1.75,
+      "eval_accuracy_AK": 0.10146007686505513,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.44335829173599556,
+      "eval_accuracy_LV": 0.2161026189117302,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.3687052023121387,
+      "eval_accuracy_RV": 0.08852301996713996,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.10123312258654107,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.3312914788160511,
+      "eval_iou_LV": 0.21381353985276833,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.3281976451024016,
+      "eval_iou_RV": 0.08844211560034837,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.4651370048522949,
+      "eval_mean_accuracy": 0.1353499121991177,
+      "eval_mean_iou": 0.10629779019581105,
+      "eval_overall_accuracy": 0.2191134361534966,
+      "eval_runtime": 4.3782,
+      "eval_samples_per_second": 18.501,
+      "eval_steps_per_second": 9.365,
+      "step": 280
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 9.2975e-05,
+      "loss": 0.4876,
+      "step": 281
+    },
+    {
+      "epoch": 1.76,
+      "learning_rate": 9.295000000000001e-05,
+      "loss": 0.4215,
+      "step": 282
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 9.292500000000001e-05,
+      "loss": 0.2739,
+      "step": 283
+    },
+    {
+      "epoch": 1.77,
+      "learning_rate": 9.290000000000001e-05,
+      "loss": 0.3876,
+      "step": 284
+    },
+    {
+      "epoch": 1.78,
+      "learning_rate": 9.2875e-05,
+      "loss": 0.5523,
+      "step": 285
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 9.285000000000001e-05,
+      "loss": 0.6985,
+      "step": 286
+    },
+    {
+      "epoch": 1.79,
+      "learning_rate": 9.2825e-05,
+      "loss": 0.4534,
+      "step": 287
+    },
+    {
+      "epoch": 1.8,
+      "learning_rate": 9.28e-05,
+      "loss": 0.3697,
+      "step": 288
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 9.2775e-05,
+      "loss": 0.7617,
+      "step": 289
+    },
+    {
+      "epoch": 1.81,
+      "learning_rate": 9.275e-05,
+      "loss": 0.5439,
+      "step": 290
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 9.2725e-05,
+      "loss": 0.459,
+      "step": 291
+    },
+    {
+      "epoch": 1.82,
+      "learning_rate": 9.27e-05,
+      "loss": 0.5957,
+      "step": 292
+    },
+    {
+      "epoch": 1.83,
+      "learning_rate": 9.2675e-05,
+      "loss": 0.3129,
+      "step": 293
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 9.265e-05,
+      "loss": 0.4936,
+      "step": 294
+    },
+    {
+      "epoch": 1.84,
+      "learning_rate": 9.2625e-05,
+      "loss": 0.3988,
+      "step": 295
+    },
+    {
+      "epoch": 1.85,
+      "learning_rate": 9.260000000000001e-05,
+      "loss": 0.3524,
+      "step": 296
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 9.2575e-05,
+      "loss": 0.6072,
+      "step": 297
+    },
+    {
+      "epoch": 1.86,
+      "learning_rate": 9.255e-05,
+      "loss": 0.5386,
+      "step": 298
+    },
+    {
+      "epoch": 1.87,
+      "learning_rate": 9.252500000000001e-05,
+      "loss": 0.5085,
+      "step": 299
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 9.250000000000001e-05,
+      "loss": 0.3688,
+      "step": 300
+    },
+    {
+      "epoch": 1.88,
+      "eval_accuracy_AK": 0.49032901117057576,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.6449217520348968,
+      "eval_accuracy_LV": 0.8529386031034966,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5493109826589595,
+      "eval_accuracy_RV": 0.5704769011113583,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.005141412249484922,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.44997074633917217,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.5536704169253884,
+      "eval_iou_LV": 0.7546264152561246,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5044011991303757,
+      "eval_iou_RV": 0.5277130223413169,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0051345840893361515,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.4278528094291687,
+      "eval_mean_accuracy": 0.3459020735920857,
+      "eval_mean_iou": 0.2795516384081714,
+      "eval_overall_accuracy": 0.6382408335112703,
+      "eval_runtime": 4.2979,
+      "eval_samples_per_second": 18.846,
+      "eval_steps_per_second": 9.54,
+      "step": 300
+    },
+    {
+      "epoch": 1.88,
+      "learning_rate": 9.2475e-05,
+      "loss": 0.4249,
+      "step": 301
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 9.245e-05,
+      "loss": 0.4207,
+      "step": 302
+    },
+    {
+      "epoch": 1.89,
+      "learning_rate": 9.242500000000001e-05,
+      "loss": 0.3904,
+      "step": 303
+    },
+    {
+      "epoch": 1.9,
+      "learning_rate": 9.240000000000001e-05,
+      "loss": 0.4973,
+      "step": 304
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 9.2375e-05,
+      "loss": 0.6232,
+      "step": 305
+    },
+    {
+      "epoch": 1.91,
+      "learning_rate": 9.235000000000001e-05,
+      "loss": 0.3847,
+      "step": 306
+    },
+    {
+      "epoch": 1.92,
+      "learning_rate": 9.232500000000001e-05,
+      "loss": 0.3851,
+      "step": 307
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 9.230000000000001e-05,
+      "loss": 0.4016,
+      "step": 308
+    },
+    {
+      "epoch": 1.93,
+      "learning_rate": 9.2275e-05,
+      "loss": 0.5811,
+      "step": 309
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 9.225e-05,
+      "loss": 0.462,
+      "step": 310
+    },
+    {
+      "epoch": 1.94,
+      "learning_rate": 9.2225e-05,
+      "loss": 0.3461,
+      "step": 311
+    },
+    {
+      "epoch": 1.95,
+      "learning_rate": 9.22e-05,
+      "loss": 0.3931,
+      "step": 312
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 9.2175e-05,
+      "loss": 0.4777,
+      "step": 313
+    },
+    {
+      "epoch": 1.96,
+      "learning_rate": 9.215e-05,
+      "loss": 0.4695,
+      "step": 314
+    },
+    {
+      "epoch": 1.97,
+      "learning_rate": 9.2125e-05,
+      "loss": 0.4359,
+      "step": 315
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 9.21e-05,
+      "loss": 0.3924,
+      "step": 316
+    },
+    {
+      "epoch": 1.98,
+      "learning_rate": 9.2075e-05,
+      "loss": 0.2749,
+      "step": 317
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 9.205e-05,
+      "loss": 0.3762,
+      "step": 318
+    },
+    {
+      "epoch": 1.99,
+      "learning_rate": 9.2025e-05,
+      "loss": 0.4244,
+      "step": 319
+    },
+    {
+      "epoch": 2.0,
+      "learning_rate": 9.200000000000001e-05,
+      "loss": 0.3659,
+      "step": 320
+    },
+    {
+      "epoch": 2.0,
+      "eval_accuracy_AK": 0.3810207966667864,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.41347566368365035,
+      "eval_accuracy_LV": 0.7155944778867794,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.3143630057803468,
+      "eval_accuracy_RV": 0.14762805013667768,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.00058063307735531,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.35596220474202417,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.3918907159335302,
+      "eval_iou_LV": 0.685085336371181,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.30116334703714204,
+      "eval_iou_RV": 0.14611245537449916,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.00058063307735531,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.39072567224502563,
+      "eval_mean_accuracy": 0.2191847363590662,
+      "eval_mean_iou": 0.18807946925357316,
+      "eval_overall_accuracy": 0.4460818290820098,
+      "eval_runtime": 4.2858,
+      "eval_samples_per_second": 18.899,
+      "eval_steps_per_second": 9.566,
+      "step": 320
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 9.1975e-05,
+      "loss": 0.4524,
+      "step": 321
+    },
+    {
+      "epoch": 2.01,
+      "learning_rate": 9.195e-05,
+      "loss": 0.2783,
+      "step": 322
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 9.192500000000001e-05,
+      "loss": 0.2814,
+      "step": 323
+    },
+    {
+      "epoch": 2.02,
+      "learning_rate": 9.190000000000001e-05,
+      "loss": 0.3788,
+      "step": 324
+    },
+    {
+      "epoch": 2.03,
+      "learning_rate": 9.1875e-05,
+      "loss": 0.4222,
+      "step": 325
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 9.185e-05,
+      "loss": 0.4028,
+      "step": 326
+    },
+    {
+      "epoch": 2.04,
+      "learning_rate": 9.182500000000001e-05,
+      "loss": 0.2685,
+      "step": 327
+    },
+    {
+      "epoch": 2.05,
+      "learning_rate": 9.180000000000001e-05,
+      "loss": 0.436,
+      "step": 328
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 9.1775e-05,
+      "loss": 0.2994,
+      "step": 329
+    },
+    {
+      "epoch": 2.06,
+      "learning_rate": 9.175000000000001e-05,
+      "loss": 0.3534,
+      "step": 330
+    },
+    {
+      "epoch": 2.07,
+      "learning_rate": 9.172500000000001e-05,
+      "loss": 0.2921,
+      "step": 331
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 9.17e-05,
+      "loss": 0.3496,
+      "step": 332
+    },
+    {
+      "epoch": 2.08,
+      "learning_rate": 9.1675e-05,
+      "loss": 0.325,
+      "step": 333
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 9.165e-05,
+      "loss": 0.5465,
+      "step": 334
+    },
+    {
+      "epoch": 2.09,
+      "learning_rate": 9.1625e-05,
+      "loss": 0.4318,
+      "step": 335
+    },
+    {
+      "epoch": 2.1,
+      "learning_rate": 9.16e-05,
+      "loss": 0.4708,
+      "step": 336
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 9.1575e-05,
+      "loss": 0.4728,
+      "step": 337
+    },
+    {
+      "epoch": 2.11,
+      "learning_rate": 9.155e-05,
+      "loss": 0.4177,
+      "step": 338
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 9.1525e-05,
+      "loss": 0.3304,
+      "step": 339
+    },
+    {
+      "epoch": 2.12,
+      "learning_rate": 9.15e-05,
+      "loss": 0.3243,
+      "step": 340
+    },
+    {
+      "epoch": 2.12,
+      "eval_accuracy_AK": 0.39044933730828635,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.7038307025827825,
+      "eval_accuracy_LV": 0.7312775271098284,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6434682080924855,
+      "eval_accuracy_RV": 0.47469848398969716,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.001020790410189174,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.37523515841191124,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.6222517460275389,
+      "eval_iou_LV": 0.7044636732492112,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.573302597886835,
+      "eval_iou_RV": 0.461041302647418,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.001020790410189174,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3845641314983368,
+      "eval_mean_accuracy": 0.327193894388141,
+      "eval_mean_iou": 0.27373152686331037,
+      "eval_overall_accuracy": 0.5846380215604794,
+      "eval_runtime": 4.3924,
+      "eval_samples_per_second": 18.441,
+      "eval_steps_per_second": 9.334,
+      "step": 340
+    },
+    {
+      "epoch": 2.13,
+      "learning_rate": 9.1475e-05,
+      "loss": 0.3596,
+      "step": 341
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 9.145e-05,
+      "loss": 0.2519,
+      "step": 342
+    },
+    {
+      "epoch": 2.14,
+      "learning_rate": 9.1425e-05,
+      "loss": 0.5022,
+      "step": 343
+    },
+    {
+      "epoch": 2.15,
+      "learning_rate": 9.140000000000001e-05,
+      "loss": 0.3462,
+      "step": 344
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 9.1375e-05,
+      "loss": 0.5754,
+      "step": 345
+    },
+    {
+      "epoch": 2.16,
+      "learning_rate": 9.135e-05,
+      "loss": 0.3346,
+      "step": 346
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 9.132500000000001e-05,
+      "loss": 0.3923,
+      "step": 347
+    },
+    {
+      "epoch": 2.17,
+      "learning_rate": 9.130000000000001e-05,
+      "loss": 0.3218,
+      "step": 348
+    },
+    {
+      "epoch": 2.18,
+      "learning_rate": 9.1275e-05,
+      "loss": 0.3723,
+      "step": 349
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 9.125e-05,
+      "loss": 0.3393,
+      "step": 350
+    },
+    {
+      "epoch": 2.19,
+      "learning_rate": 9.122500000000001e-05,
+      "loss": 0.2961,
+      "step": 351
+    },
+    {
+      "epoch": 2.2,
+      "learning_rate": 9.120000000000001e-05,
+      "loss": 0.4071,
+      "step": 352
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 9.1175e-05,
+      "loss": 0.3372,
+      "step": 353
+    },
+    {
+      "epoch": 2.21,
+      "learning_rate": 9.115e-05,
+      "loss": 0.2853,
+      "step": 354
+    },
+    {
+      "epoch": 2.22,
+      "learning_rate": 9.1125e-05,
+      "loss": 0.381,
+      "step": 355
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 9.11e-05,
+      "loss": 0.3936,
+      "step": 356
+    },
+    {
+      "epoch": 2.23,
+      "learning_rate": 9.1075e-05,
+      "loss": 0.163,
+      "step": 357
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 9.105e-05,
+      "loss": 0.4305,
+      "step": 358
+    },
+    {
+      "epoch": 2.24,
+      "learning_rate": 9.1025e-05,
+      "loss": 0.4934,
+      "step": 359
+    },
+    {
+      "epoch": 2.25,
+      "learning_rate": 9.1e-05,
+      "loss": 0.4169,
+      "step": 360
+    },
+    {
+      "epoch": 2.25,
+      "eval_accuracy_AK": 0.27411311854219794,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.30883774789764806,
+      "eval_accuracy_LV": 0.3286259059043968,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.3183791907514451,
+      "eval_accuracy_RV": 0.09681336499759648,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0005244427795467316,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.2685097225154223,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.2960098129364099,
+      "eval_iou_LV": 0.32406528627121994,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.3061222674998944,
+      "eval_iou_RV": 0.0965283763918149,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.0005244427795467316,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.4099118411540985,
+      "eval_mean_accuracy": 0.1474770856525368,
+      "eval_mean_iou": 0.12917599083943082,
+      "eval_overall_accuracy": 0.2563405767346112,
+      "eval_runtime": 4.4061,
+      "eval_samples_per_second": 18.384,
+      "eval_steps_per_second": 9.305,
+      "step": 360
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 9.0975e-05,
+      "loss": 0.3114,
+      "step": 361
+    },
+    {
+      "epoch": 2.26,
+      "learning_rate": 9.095e-05,
+      "loss": 0.7929,
+      "step": 362
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 9.0925e-05,
+      "loss": 0.35,
+      "step": 363
+    },
+    {
+      "epoch": 2.27,
+      "learning_rate": 9.090000000000001e-05,
+      "loss": 0.2621,
+      "step": 364
+    },
+    {
+      "epoch": 2.28,
+      "learning_rate": 9.0875e-05,
+      "loss": 0.5385,
+      "step": 365
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 9.085e-05,
+      "loss": 0.4585,
+      "step": 366
+    },
+    {
+      "epoch": 2.29,
+      "learning_rate": 9.0825e-05,
+      "loss": 0.385,
+      "step": 367
+    },
+    {
+      "epoch": 2.3,
+      "learning_rate": 9.080000000000001e-05,
+      "loss": 0.3166,
+      "step": 368
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 9.0775e-05,
+      "loss": 0.3616,
+      "step": 369
+    },
+    {
+      "epoch": 2.31,
+      "learning_rate": 9.075e-05,
+      "loss": 0.3681,
+      "step": 370
+    },
+    {
+      "epoch": 2.32,
+      "learning_rate": 9.072500000000001e-05,
+      "loss": 0.5442,
+      "step": 371
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 9.070000000000001e-05,
+      "loss": 0.378,
+      "step": 372
+    },
+    {
+      "epoch": 2.33,
+      "learning_rate": 9.0675e-05,
+      "loss": 0.4478,
+      "step": 373
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 9.065000000000001e-05,
+      "loss": 0.3697,
+      "step": 374
+    },
+    {
+      "epoch": 2.34,
+      "learning_rate": 9.062500000000001e-05,
+      "loss": 0.2358,
+      "step": 375
+    },
+    {
+      "epoch": 2.35,
+      "learning_rate": 9.06e-05,
+      "loss": 0.3334,
+      "step": 376
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 9.0575e-05,
+      "loss": 0.3563,
+      "step": 377
+    },
+    {
+      "epoch": 2.36,
+      "learning_rate": 9.055e-05,
+      "loss": 0.6663,
+      "step": 378
+    },
+    {
+      "epoch": 2.37,
+      "learning_rate": 9.0525e-05,
+      "loss": 0.6533,
+      "step": 379
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 9.05e-05,
+      "loss": 0.2951,
+      "step": 380
+    },
+    {
+      "epoch": 2.38,
+      "eval_accuracy_AK": 0.2519186331908576,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.6784076089400549,
+      "eval_accuracy_LV": 0.6694801750966477,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5730473988439306,
+      "eval_accuracy_RV": 0.21363333070261661,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.044156209027907845,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.2488424762434141,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.5966962824525233,
+      "eval_iou_LV": 0.6409329723335133,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5347448320204368,
+      "eval_iou_RV": 0.2125445239947749,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.043528434268833086,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3582638204097748,
+      "eval_mean_accuracy": 0.2700714839780017,
+      "eval_mean_iou": 0.22772895213134955,
+      "eval_overall_accuracy": 0.4961984462000887,
+      "eval_runtime": 4.2959,
+      "eval_samples_per_second": 18.855,
+      "eval_steps_per_second": 9.544,
+      "step": 380
+    },
+    {
+      "epoch": 2.38,
+      "learning_rate": 9.0475e-05,
+      "loss": 0.2944,
+      "step": 381
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 9.045e-05,
+      "loss": 0.2736,
+      "step": 382
+    },
+    {
+      "epoch": 2.39,
+      "learning_rate": 9.0425e-05,
+      "loss": 0.551,
+      "step": 383
+    },
+    {
+      "epoch": 2.4,
+      "learning_rate": 9.04e-05,
+      "loss": 0.4521,
+      "step": 384
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 9.037500000000001e-05,
+      "loss": 0.3252,
+      "step": 385
+    },
+    {
+      "epoch": 2.41,
+      "learning_rate": 9.035e-05,
+      "loss": 0.526,
+      "step": 386
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 9.0325e-05,
+      "loss": 0.3281,
+      "step": 387
+    },
+    {
+      "epoch": 2.42,
+      "learning_rate": 9.030000000000001e-05,
+      "loss": 0.2743,
+      "step": 388
+    },
+    {
+      "epoch": 2.43,
+      "learning_rate": 9.027500000000001e-05,
+      "loss": 0.2724,
+      "step": 389
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 9.025e-05,
+      "loss": 0.3944,
+      "step": 390
+    },
+    {
+      "epoch": 2.44,
+      "learning_rate": 9.0225e-05,
+      "loss": 0.2651,
+      "step": 391
+    },
+    {
+      "epoch": 2.45,
+      "learning_rate": 9.020000000000001e-05,
+      "loss": 0.279,
+      "step": 392
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 9.017500000000001e-05,
+      "loss": 0.3564,
+      "step": 393
+    },
+    {
+      "epoch": 2.46,
+      "learning_rate": 9.015e-05,
+      "loss": 0.2109,
+      "step": 394
+    },
+    {
+      "epoch": 2.47,
+      "learning_rate": 9.012500000000001e-05,
+      "loss": 0.288,
+      "step": 395
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 9.010000000000001e-05,
+      "loss": 0.4921,
+      "step": 396
+    },
+    {
+      "epoch": 2.48,
+      "learning_rate": 9.007500000000001e-05,
+      "loss": 0.5898,
+      "step": 397
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 9.005000000000001e-05,
+      "loss": 0.3059,
+      "step": 398
+    },
+    {
+      "epoch": 2.49,
+      "learning_rate": 9.0025e-05,
+      "loss": 0.4104,
+      "step": 399
+    },
+    {
+      "epoch": 2.5,
+      "learning_rate": 9e-05,
+      "loss": 0.3847,
+      "step": 400
+    },
+    {
+      "epoch": 2.5,
+      "eval_accuracy_AK": 0.25939860876644755,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.6408763172490294,
+      "eval_accuracy_LV": 0.6544210288274124,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5888300578034682,
+      "eval_accuracy_RV": 0.30665846361360033,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.10889679715302492,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.25350954942621534,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.5852675240761401,
+      "eval_iou_LV": 0.6303974896753823,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5346583501462249,
+      "eval_iou_RV": 0.30234655603244753,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.10333519955210749,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.35651251673698425,
+      "eval_mean_accuracy": 0.28434236371255367,
+      "eval_mean_iou": 0.2409514668908518,
+      "eval_overall_accuracy": 0.5032327031628971,
+      "eval_runtime": 4.3217,
+      "eval_samples_per_second": 18.743,
+      "eval_steps_per_second": 9.487,
+      "step": 400
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 8.9975e-05,
+      "loss": 0.2817,
+      "step": 401
+    },
+    {
+      "epoch": 2.51,
+      "learning_rate": 8.995e-05,
+      "loss": 0.2905,
+      "step": 402
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 8.9925e-05,
+      "loss": 0.2285,
+      "step": 403
+    },
+    {
+      "epoch": 2.52,
+      "learning_rate": 8.99e-05,
+      "loss": 0.3652,
+      "step": 404
+    },
+    {
+      "epoch": 2.53,
+      "learning_rate": 8.9875e-05,
+      "loss": 0.3379,
+      "step": 405
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 8.985e-05,
+      "loss": 0.3874,
+      "step": 406
+    },
+    {
+      "epoch": 2.54,
+      "learning_rate": 8.9825e-05,
+      "loss": 0.3345,
+      "step": 407
+    },
+    {
+      "epoch": 2.55,
+      "learning_rate": 8.98e-05,
+      "loss": 0.3567,
+      "step": 408
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 8.977500000000001e-05,
+      "loss": 0.3165,
+      "step": 409
+    },
+    {
+      "epoch": 2.56,
+      "learning_rate": 8.975e-05,
+      "loss": 0.56,
+      "step": 410
+    },
+    {
+      "epoch": 2.57,
+      "learning_rate": 8.9725e-05,
+      "loss": 0.2853,
+      "step": 411
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 8.970000000000001e-05,
+      "loss": 0.2311,
+      "step": 412
+    },
+    {
+      "epoch": 2.58,
+      "learning_rate": 8.967500000000001e-05,
+      "loss": 0.4779,
+      "step": 413
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 8.965e-05,
+      "loss": 0.3167,
+      "step": 414
+    },
+    {
+      "epoch": 2.59,
+      "learning_rate": 8.962500000000001e-05,
+      "loss": 0.5405,
+      "step": 415
+    },
+    {
+      "epoch": 2.6,
+      "learning_rate": 8.960000000000001e-05,
+      "loss": 0.4581,
+      "step": 416
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 8.957500000000001e-05,
+      "loss": 0.2187,
+      "step": 417
+    },
+    {
+      "epoch": 2.61,
+      "learning_rate": 8.955e-05,
+      "loss": 0.357,
+      "step": 418
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 8.952500000000001e-05,
+      "loss": 0.3782,
+      "step": 419
+    },
+    {
+      "epoch": 2.62,
+      "learning_rate": 8.950000000000001e-05,
+      "loss": 0.339,
+      "step": 420
+    },
+    {
+      "epoch": 2.62,
+      "eval_accuracy_AK": 0.5459963123929935,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0014611094412717498,
+      "eval_accuracy_LA": 0.6900117671748287,
+      "eval_accuracy_LV": 0.7530427888394732,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6410658959537572,
+      "eval_accuracy_RV": 0.5620197447248151,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.13366735343697322,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.5259966262489367,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0014611094412717498,
+      "eval_iou_LA": 0.615988732413214,
+      "eval_iou_LV": 0.7083221491805604,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5721889046190167,
+      "eval_iou_RV": 0.5346794670525439,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.12613337103872463,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.37152913212776184,
+      "eval_mean_accuracy": 0.36969610799601244,
+      "eval_mean_iou": 0.3084770359994268,
+      "eval_overall_accuracy": 0.6226656045201452,
+      "eval_runtime": 4.2758,
+      "eval_samples_per_second": 18.944,
+      "eval_steps_per_second": 9.589,
+      "step": 420
+    },
+    {
+      "epoch": 2.63,
+      "learning_rate": 8.9475e-05,
+      "loss": 0.4128,
+      "step": 421
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 8.945e-05,
+      "loss": 0.3237,
+      "step": 422
+    },
+    {
+      "epoch": 2.64,
+      "learning_rate": 8.9425e-05,
+      "loss": 0.3517,
+      "step": 423
+    },
+    {
+      "epoch": 2.65,
+      "learning_rate": 8.94e-05,
+      "loss": 0.332,
+      "step": 424
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 8.9375e-05,
+      "loss": 0.3496,
+      "step": 425
+    },
+    {
+      "epoch": 2.66,
+      "learning_rate": 8.935e-05,
+      "loss": 0.1985,
+      "step": 426
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 8.9325e-05,
+      "loss": 0.2829,
+      "step": 427
+    },
+    {
+      "epoch": 2.67,
+      "learning_rate": 8.93e-05,
+      "loss": 0.2325,
+      "step": 428
+    },
+    {
+      "epoch": 2.68,
+      "learning_rate": 8.927500000000002e-05,
+      "loss": 0.2639,
+      "step": 429
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 8.925e-05,
+      "loss": 0.2979,
+      "step": 430
+    },
+    {
+      "epoch": 2.69,
+      "learning_rate": 8.9225e-05,
+      "loss": 0.2318,
+      "step": 431
+    },
+    {
+      "epoch": 2.7,
+      "learning_rate": 8.92e-05,
+      "loss": 0.1811,
+      "step": 432
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 8.917500000000001e-05,
+      "loss": 0.6538,
+      "step": 433
+    },
+    {
+      "epoch": 2.71,
+      "learning_rate": 8.915e-05,
+      "loss": 0.4878,
+      "step": 434
+    },
+    {
+      "epoch": 2.72,
+      "learning_rate": 8.9125e-05,
+      "loss": 0.4138,
+      "step": 435
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 8.910000000000001e-05,
+      "loss": 0.4613,
+      "step": 436
+    },
+    {
+      "epoch": 2.73,
+      "learning_rate": 8.907500000000001e-05,
+      "loss": 0.2582,
+      "step": 437
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 8.905e-05,
+      "loss": 0.3046,
+      "step": 438
+    },
+    {
+      "epoch": 2.74,
+      "learning_rate": 8.902500000000001e-05,
+      "loss": 0.285,
+      "step": 439
+    },
+    {
+      "epoch": 2.75,
+      "learning_rate": 8.900000000000001e-05,
+      "loss": 0.7318,
+      "step": 440
+    },
+    {
+      "epoch": 2.75,
+      "eval_accuracy_AK": 0.3349526477736671,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.009710115886851684,
+      "eval_accuracy_LA": 0.6926181589243153,
+      "eval_accuracy_LV": 0.5246895965023745,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5171421965317919,
+      "eval_accuracy_RV": 0.23384082250554244,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.3424236748454767,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3234753400819781,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.009707441258712073,
+      "eval_iou_LA": 0.61741390426495,
+      "eval_iou_LV": 0.5099719498027067,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.4803398355371244,
+      "eval_iou_RV": 0.23217297304998388,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.3047787345064142,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3573572337627411,
+      "eval_mean_accuracy": 0.2950419125522244,
+      "eval_mean_iou": 0.247786017850187,
+      "eval_overall_accuracy": 0.45247879813193614,
+      "eval_runtime": 4.3183,
+      "eval_samples_per_second": 18.757,
+      "eval_steps_per_second": 9.494,
+      "step": 440
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 8.897500000000001e-05,
+      "loss": 0.3598,
+      "step": 441
+    },
+    {
+      "epoch": 2.76,
+      "learning_rate": 8.895e-05,
+      "loss": 0.2076,
+      "step": 442
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 8.8925e-05,
+      "loss": 0.5106,
+      "step": 443
+    },
+    {
+      "epoch": 2.77,
+      "learning_rate": 8.89e-05,
+      "loss": 0.3081,
+      "step": 444
+    },
+    {
+      "epoch": 2.78,
+      "learning_rate": 8.8875e-05,
+      "loss": 0.4885,
+      "step": 445
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 8.885e-05,
+      "loss": 0.2949,
+      "step": 446
+    },
+    {
+      "epoch": 2.79,
+      "learning_rate": 8.8825e-05,
+      "loss": 0.4082,
+      "step": 447
+    },
+    {
+      "epoch": 2.8,
+      "learning_rate": 8.88e-05,
+      "loss": 0.3334,
+      "step": 448
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 8.8775e-05,
+      "loss": 0.2932,
+      "step": 449
+    },
+    {
+      "epoch": 2.81,
+      "learning_rate": 8.875e-05,
+      "loss": 0.4338,
+      "step": 450
+    },
+    {
+      "epoch": 2.82,
+      "learning_rate": 8.8725e-05,
+      "loss": 0.3143,
+      "step": 451
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 8.87e-05,
+      "loss": 0.389,
+      "step": 452
+    },
+    {
+      "epoch": 2.83,
+      "learning_rate": 8.867500000000001e-05,
+      "loss": 0.3854,
+      "step": 453
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 8.865e-05,
+      "loss": 0.2451,
+      "step": 454
+    },
+    {
+      "epoch": 2.84,
+      "learning_rate": 8.8625e-05,
+      "loss": 0.2134,
+      "step": 455
+    },
+    {
+      "epoch": 2.85,
+      "learning_rate": 8.86e-05,
+      "loss": 0.2417,
+      "step": 456
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 8.857500000000001e-05,
+      "loss": 0.235,
+      "step": 457
+    },
+    {
+      "epoch": 2.86,
+      "learning_rate": 8.855e-05,
+      "loss": 0.4012,
+      "step": 458
+    },
+    {
+      "epoch": 2.87,
+      "learning_rate": 8.8525e-05,
+      "loss": 0.5499,
+      "step": 459
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 8.850000000000001e-05,
+      "loss": 0.2905,
+      "step": 460
+    },
+    {
+      "epoch": 2.88,
+      "eval_accuracy_AK": 0.17994145325239755,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.0,
+      "eval_accuracy_LA": 0.5863107283656368,
+      "eval_accuracy_LV": 0.4733653338789399,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.592478612716763,
+      "eval_accuracy_RV": 0.18409695865230774,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.02005057126802772,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.17864901087340904,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.0,
+      "eval_iou_LA": 0.519180546015059,
+      "eval_iou_LV": 0.4670651127270312,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5347625835021985,
+      "eval_iou_RV": 0.1834313604592352,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.019883725249823545,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.36089879274368286,
+      "eval_mean_accuracy": 0.22624929534823032,
+      "eval_mean_iou": 0.19029723388267566,
+      "eval_overall_accuracy": 0.39348121259902874,
+      "eval_runtime": 4.2753,
+      "eval_samples_per_second": 18.946,
+      "eval_steps_per_second": 9.59,
+      "step": 460
+    },
+    {
+      "epoch": 2.88,
+      "learning_rate": 8.847500000000001e-05,
+      "loss": 0.6448,
+      "step": 461
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 8.845e-05,
+      "loss": 0.377,
+      "step": 462
+    },
+    {
+      "epoch": 2.89,
+      "learning_rate": 8.842500000000001e-05,
+      "loss": 0.218,
+      "step": 463
+    },
+    {
+      "epoch": 2.9,
+      "learning_rate": 8.840000000000001e-05,
+      "loss": 0.4422,
+      "step": 464
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 8.837500000000001e-05,
+      "loss": 0.5321,
+      "step": 465
+    },
+    {
+      "epoch": 2.91,
+      "learning_rate": 8.834999999999999e-05,
+      "loss": 0.3462,
+      "step": 466
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 8.8325e-05,
+      "loss": 0.2453,
+      "step": 467
+    },
+    {
+      "epoch": 2.92,
+      "learning_rate": 8.83e-05,
+      "loss": 0.3069,
+      "step": 468
+    },
+    {
+      "epoch": 2.93,
+      "learning_rate": 8.8275e-05,
+      "loss": 0.1978,
+      "step": 469
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 8.825e-05,
+      "loss": 0.694,
+      "step": 470
+    },
+    {
+      "epoch": 2.94,
+      "learning_rate": 8.8225e-05,
+      "loss": 0.5458,
+      "step": 471
+    },
+    {
+      "epoch": 2.95,
+      "learning_rate": 8.82e-05,
+      "loss": 0.3155,
+      "step": 472
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 8.8175e-05,
+      "loss": 0.2446,
+      "step": 473
+    },
+    {
+      "epoch": 2.96,
+      "learning_rate": 8.815e-05,
+      "loss": 0.2936,
+      "step": 474
+    },
+    {
+      "epoch": 2.97,
+      "learning_rate": 8.8125e-05,
+      "loss": 0.3922,
+      "step": 475
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 8.81e-05,
+      "loss": 0.808,
+      "step": 476
+    },
+    {
+      "epoch": 2.98,
+      "learning_rate": 8.807500000000001e-05,
+      "loss": 0.3406,
+      "step": 477
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 8.805e-05,
+      "loss": 0.4646,
+      "step": 478
+    },
+    {
+      "epoch": 2.99,
+      "learning_rate": 8.8025e-05,
+      "loss": 0.438,
+      "step": 479
+    },
+    {
+      "epoch": 3.0,
+      "learning_rate": 8.800000000000001e-05,
+      "loss": 0.3793,
+      "step": 480
+    },
+    {
+      "epoch": 3.0,
+      "eval_accuracy_AK": 0.3403553512206219,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.02625822395885516,
+      "eval_accuracy_LA": 0.6168435490398885,
+      "eval_accuracy_LV": 0.8326890150005922,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5373757225433526,
+      "eval_accuracy_RV": 0.46348624972198105,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.0746207154897921,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3269633327199117,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.026018183773588027,
+      "eval_iou_LA": 0.569995428983018,
+      "eval_iou_LV": 0.7723172902076403,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5138798297495992,
+      "eval_iou_RV": 0.45115765838907307,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.07337557094445263,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.345205157995224,
+      "eval_mean_accuracy": 0.3212920918861204,
+      "eval_mean_iou": 0.2733707294767283,
+      "eval_overall_accuracy": 0.597336147474117,
+      "eval_runtime": 4.3431,
+      "eval_samples_per_second": 18.65,
+      "eval_steps_per_second": 9.44,
+      "step": 480
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 8.797500000000001e-05,
+      "loss": 0.2496,
+      "step": 481
+    },
+    {
+      "epoch": 3.01,
+      "learning_rate": 8.795e-05,
+      "loss": 0.3643,
+      "step": 482
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 8.7925e-05,
+      "loss": 0.3767,
+      "step": 483
+    },
+    {
+      "epoch": 3.02,
+      "learning_rate": 8.790000000000001e-05,
+      "loss": 0.4158,
+      "step": 484
+    },
+    {
+      "epoch": 3.03,
+      "learning_rate": 8.787500000000001e-05,
+      "loss": 0.4141,
+      "step": 485
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 8.785e-05,
+      "loss": 0.3872,
+      "step": 486
+    },
+    {
+      "epoch": 3.04,
+      "learning_rate": 8.782500000000001e-05,
+      "loss": 0.3086,
+      "step": 487
+    },
+    {
+      "epoch": 3.05,
+      "learning_rate": 8.78e-05,
+      "loss": 0.3353,
+      "step": 488
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 8.7775e-05,
+      "loss": 0.2774,
+      "step": 489
+    },
+    {
+      "epoch": 3.06,
+      "learning_rate": 8.775e-05,
+      "loss": 0.2329,
+      "step": 490
+    },
+    {
+      "epoch": 3.07,
+      "learning_rate": 8.7725e-05,
+      "loss": 0.2561,
+      "step": 491
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 8.77e-05,
+      "loss": 0.3373,
+      "step": 492
+    },
+    {
+      "epoch": 3.08,
+      "learning_rate": 8.7675e-05,
+      "loss": 0.3361,
+      "step": 493
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 8.765e-05,
+      "loss": 0.3181,
+      "step": 494
+    },
+    {
+      "epoch": 3.09,
+      "learning_rate": 8.7625e-05,
+      "loss": 0.505,
+      "step": 495
+    },
+    {
+      "epoch": 3.1,
+      "learning_rate": 8.76e-05,
+      "loss": 0.1839,
+      "step": 496
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 8.7575e-05,
+      "loss": 0.3403,
+      "step": 497
+    },
+    {
+      "epoch": 3.11,
+      "learning_rate": 8.755e-05,
+      "loss": 0.3126,
+      "step": 498
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 8.7525e-05,
+      "loss": 0.2319,
+      "step": 499
+    },
+    {
+      "epoch": 3.12,
+      "learning_rate": 8.75e-05,
+      "loss": 0.3922,
+      "step": 500
+    },
+    {
+      "epoch": 3.12,
+      "eval_accuracy_AK": 0.2863402895010955,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.004458471095080653,
+      "eval_accuracy_LA": 0.7205858104360601,
+      "eval_accuracy_LV": 0.2750887059431636,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6317965317919075,
+      "eval_accuracy_RV": 0.28468779371354364,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.14085034650683648,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.28242462890140646,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.004451929169306699,
+      "eval_iou_LA": 0.6103062417972364,
+      "eval_iou_LV": 0.2726041284556641,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5652043511625646,
+      "eval_iou_RV": 0.2824178776755036,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.13394606533433082,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.36945000290870667,
+      "eval_mean_accuracy": 0.2604231054430764,
+      "eval_mean_iou": 0.21513552224960125,
+      "eval_overall_accuracy": 0.36586704553542587,
+      "eval_runtime": 4.3536,
+      "eval_samples_per_second": 18.605,
+      "eval_steps_per_second": 9.418,
+      "step": 500
+    },
+    {
+      "epoch": 3.13,
+      "learning_rate": 8.747500000000001e-05,
+      "loss": 0.3017,
+      "step": 501
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 8.745000000000001e-05,
+      "loss": 0.4561,
+      "step": 502
+    },
+    {
+      "epoch": 3.14,
+      "learning_rate": 8.7425e-05,
+      "loss": 0.2648,
+      "step": 503
+    },
+    {
+      "epoch": 3.15,
+      "learning_rate": 8.740000000000001e-05,
+      "loss": 0.3374,
+      "step": 504
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 8.737500000000001e-05,
+      "loss": 0.3366,
+      "step": 505
+    },
+    {
+      "epoch": 3.16,
+      "learning_rate": 8.735000000000001e-05,
+      "loss": 0.3114,
+      "step": 506
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 8.7325e-05,
+      "loss": 0.2809,
+      "step": 507
+    },
+    {
+      "epoch": 3.17,
+      "learning_rate": 8.730000000000001e-05,
+      "loss": 0.24,
+      "step": 508
+    },
+    {
+      "epoch": 3.18,
+      "learning_rate": 8.727500000000001e-05,
+      "loss": 0.5009,
+      "step": 509
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 8.725e-05,
+      "loss": 0.2111,
+      "step": 510
+    },
+    {
+      "epoch": 3.19,
+      "learning_rate": 8.7225e-05,
+      "loss": 0.3674,
+      "step": 511
+    },
+    {
+      "epoch": 3.2,
+      "learning_rate": 8.72e-05,
+      "loss": 0.3971,
+      "step": 512
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 8.7175e-05,
+      "loss": 0.7098,
+      "step": 513
+    },
+    {
+      "epoch": 3.21,
+      "learning_rate": 8.715e-05,
+      "loss": 0.5057,
+      "step": 514
+    },
+    {
+      "epoch": 3.22,
+      "learning_rate": 8.7125e-05,
+      "loss": 0.4631,
+      "step": 515
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 8.71e-05,
+      "loss": 0.2942,
+      "step": 516
+    },
+    {
+      "epoch": 3.23,
+      "learning_rate": 8.7075e-05,
+      "loss": 0.3161,
+      "step": 517
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 8.705000000000002e-05,
+      "loss": 0.3454,
+      "step": 518
+    },
+    {
+      "epoch": 3.24,
+      "learning_rate": 8.7025e-05,
+      "loss": 0.378,
+      "step": 519
+    },
+    {
+      "epoch": 3.25,
+      "learning_rate": 8.7e-05,
+      "loss": 0.3299,
+      "step": 520
+    },
+    {
+      "epoch": 3.25,
+      "eval_accuracy_AK": 0.5038013481316523,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.042188491467120864,
+      "eval_accuracy_LA": 0.6557689136724079,
+      "eval_accuracy_LV": 0.802910604869539,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5865710982658959,
+      "eval_accuracy_RV": 0.5000986518772556,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.1575294999063495,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.476214188305993,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.04162410623084781,
+      "eval_iou_LA": 0.5947926580557444,
+      "eval_iou_LV": 0.7639444055240991,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5460544866344806,
+      "eval_iou_RV": 0.4932089636726434,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.14723362539060106,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.33257803320884705,
+      "eval_mean_accuracy": 0.3609854009100245,
+      "eval_mean_iou": 0.306307243381441,
+      "eval_overall_accuracy": 0.620217466287798,
+      "eval_runtime": 4.3818,
+      "eval_samples_per_second": 18.486,
+      "eval_steps_per_second": 9.357,
+      "step": 520
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 8.6975e-05,
+      "loss": 0.3804,
+      "step": 521
+    },
+    {
+      "epoch": 3.26,
+      "learning_rate": 8.695000000000001e-05,
+      "loss": 0.3072,
+      "step": 522
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 8.6925e-05,
+      "loss": 0.3441,
+      "step": 523
+    },
+    {
+      "epoch": 3.27,
+      "learning_rate": 8.69e-05,
+      "loss": 0.2871,
+      "step": 524
+    },
+    {
+      "epoch": 3.28,
+      "learning_rate": 8.687500000000001e-05,
+      "loss": 0.3628,
+      "step": 525
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 8.685000000000001e-05,
+      "loss": 0.2031,
+      "step": 526
+    },
+    {
+      "epoch": 3.29,
+      "learning_rate": 8.6825e-05,
+      "loss": 0.394,
+      "step": 527
+    },
+    {
+      "epoch": 3.3,
+      "learning_rate": 8.680000000000001e-05,
+      "loss": 0.2824,
+      "step": 528
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 8.677500000000001e-05,
+      "loss": 0.3225,
+      "step": 529
+    },
+    {
+      "epoch": 3.31,
+      "learning_rate": 8.675000000000001e-05,
+      "loss": 0.254,
+      "step": 530
+    },
+    {
+      "epoch": 3.32,
+      "learning_rate": 8.672500000000001e-05,
+      "loss": 0.2834,
+      "step": 531
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 8.67e-05,
+      "loss": 0.3819,
+      "step": 532
+    },
+    {
+      "epoch": 3.33,
+      "learning_rate": 8.6675e-05,
+      "loss": 0.2267,
+      "step": 533
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 8.665e-05,
+      "loss": 0.34,
+      "step": 534
+    },
+    {
+      "epoch": 3.34,
+      "learning_rate": 8.6625e-05,
+      "loss": 0.3332,
+      "step": 535
+    },
+    {
+      "epoch": 3.35,
+      "learning_rate": 8.66e-05,
+      "loss": 0.293,
+      "step": 536
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 8.6575e-05,
+      "loss": 0.3987,
+      "step": 537
+    },
+    {
+      "epoch": 3.36,
+      "learning_rate": 8.655e-05,
+      "loss": 0.256,
+      "step": 538
+    },
+    {
+      "epoch": 3.37,
+      "learning_rate": 8.6525e-05,
+      "loss": 0.4905,
+      "step": 539
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 8.65e-05,
+      "loss": 0.2547,
+      "step": 540
+    },
+    {
+      "epoch": 3.38,
+      "eval_accuracy_AK": 0.4163733342911533,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.08126273252513108,
+      "eval_accuracy_LA": 0.5281531531531531,
+      "eval_accuracy_LV": 0.6042402517687346,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6044161849710983,
+      "eval_accuracy_RV": 0.43105130615085485,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.14375351189361305,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.4010234086715867,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.08005362680024017,
+      "eval_iou_LA": 0.5006127363919239,
+      "eval_iou_LV": 0.5882347159941189,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5254440676739638,
+      "eval_iou_RV": 0.4189089311736939,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.13585273033011772,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.33227574825286865,
+      "eval_mean_accuracy": 0.3121389416393042,
+      "eval_mean_iou": 0.2650130217035645,
+      "eval_overall_accuracy": 0.5052236803924467,
+      "eval_runtime": 4.3675,
+      "eval_samples_per_second": 18.546,
+      "eval_steps_per_second": 9.387,
+      "step": 540
+    },
+    {
+      "epoch": 3.38,
+      "learning_rate": 8.6475e-05,
+      "loss": 0.264,
+      "step": 541
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 8.645000000000001e-05,
+      "loss": 0.3142,
+      "step": 542
+    },
+    {
+      "epoch": 3.39,
+      "learning_rate": 8.6425e-05,
+      "loss": 0.4571,
+      "step": 543
+    },
+    {
+      "epoch": 3.4,
+      "learning_rate": 8.64e-05,
+      "loss": 0.4592,
+      "step": 544
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 8.637500000000001e-05,
+      "loss": 0.3148,
+      "step": 545
+    },
+    {
+      "epoch": 3.41,
+      "learning_rate": 8.635000000000001e-05,
+      "loss": 0.5289,
+      "step": 546
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 8.6325e-05,
+      "loss": 0.2309,
+      "step": 547
+    },
+    {
+      "epoch": 3.42,
+      "learning_rate": 8.63e-05,
+      "loss": 0.3315,
+      "step": 548
+    },
+    {
+      "epoch": 3.43,
+      "learning_rate": 8.627500000000001e-05,
+      "loss": 0.3765,
+      "step": 549
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 8.625000000000001e-05,
+      "loss": 0.2992,
+      "step": 550
+    },
+    {
+      "epoch": 3.44,
+      "learning_rate": 8.6225e-05,
+      "loss": 0.3186,
+      "step": 551
+    },
+    {
+      "epoch": 3.45,
+      "learning_rate": 8.620000000000001e-05,
+      "loss": 0.3253,
+      "step": 552
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 8.617500000000001e-05,
+      "loss": 0.6042,
+      "step": 553
+    },
+    {
+      "epoch": 3.46,
+      "learning_rate": 8.615000000000001e-05,
+      "loss": 0.4992,
+      "step": 554
+    },
+    {
+      "epoch": 3.47,
+      "learning_rate": 8.6125e-05,
+      "loss": 0.3198,
+      "step": 555
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 8.61e-05,
+      "loss": 0.2827,
+      "step": 556
+    },
+    {
+      "epoch": 3.48,
+      "learning_rate": 8.6075e-05,
+      "loss": 0.3034,
+      "step": 557
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 8.605e-05,
+      "loss": 0.2449,
+      "step": 558
+    },
+    {
+      "epoch": 3.49,
+      "learning_rate": 8.6025e-05,
+      "loss": 0.2145,
+      "step": 559
+    },
+    {
+      "epoch": 3.5,
+      "learning_rate": 8.6e-05,
+      "loss": 0.2154,
+      "step": 560
+    },
+    {
+      "epoch": 3.5,
+      "eval_accuracy_AK": 0.49547729367958526,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.03395618341515546,
+      "eval_accuracy_LA": 0.6269393353419975,
+      "eval_accuracy_LV": 0.7327090714278023,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6264531791907515,
+      "eval_accuracy_RV": 0.43414359408519215,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.10787600674283573,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.46340828415618734,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.033748796760381054,
+      "eval_iou_LA": 0.5874153129317746,
+      "eval_iou_LV": 0.7034337354068507,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5827828671847609,
+      "eval_iou_RV": 0.4303848786863092,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.10224841776365427,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.32110071182250977,
+      "eval_mean_accuracy": 0.3397282959870355,
+      "eval_mean_iou": 0.2903422292889918,
+      "eval_overall_accuracy": 0.5796251279777058,
+      "eval_runtime": 4.3066,
+      "eval_samples_per_second": 18.808,
+      "eval_steps_per_second": 9.52,
+      "step": 560
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 8.5975e-05,
+      "loss": 0.3226,
+      "step": 561
+    },
+    {
+      "epoch": 3.51,
+      "learning_rate": 8.595e-05,
+      "loss": 0.3108,
+      "step": 562
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 8.5925e-05,
+      "loss": 0.2209,
+      "step": 563
+    },
+    {
+      "epoch": 3.52,
+      "learning_rate": 8.59e-05,
+      "loss": 0.4449,
+      "step": 564
+    },
+    {
+      "epoch": 3.53,
+      "learning_rate": 8.5875e-05,
+      "loss": 0.5756,
+      "step": 565
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 8.585000000000001e-05,
+      "loss": 0.2925,
+      "step": 566
+    },
+    {
+      "epoch": 3.54,
+      "learning_rate": 8.5825e-05,
+      "loss": 0.2096,
+      "step": 567
+    },
+    {
+      "epoch": 3.55,
+      "learning_rate": 8.58e-05,
+      "loss": 0.2382,
+      "step": 568
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 8.577500000000001e-05,
+      "loss": 0.5982,
+      "step": 569
+    },
+    {
+      "epoch": 3.56,
+      "learning_rate": 8.575000000000001e-05,
+      "loss": 0.5009,
+      "step": 570
+    },
+    {
+      "epoch": 3.57,
+      "learning_rate": 8.5725e-05,
+      "loss": 0.3325,
+      "step": 571
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 8.57e-05,
+      "loss": 0.2765,
+      "step": 572
+    },
+    {
+      "epoch": 3.58,
+      "learning_rate": 8.567500000000001e-05,
+      "loss": 0.4498,
+      "step": 573
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 8.565000000000001e-05,
+      "loss": 0.5,
+      "step": 574
+    },
+    {
+      "epoch": 3.59,
+      "learning_rate": 8.5625e-05,
+      "loss": 0.2084,
+      "step": 575
+    },
+    {
+      "epoch": 3.6,
+      "learning_rate": 8.560000000000001e-05,
+      "loss": 0.4352,
+      "step": 576
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 8.5575e-05,
+      "loss": 0.253,
+      "step": 577
+    },
+    {
+      "epoch": 3.61,
+      "learning_rate": 8.555e-05,
+      "loss": 0.4637,
+      "step": 578
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 8.5525e-05,
+      "loss": 0.4002,
+      "step": 579
+    },
+    {
+      "epoch": 3.62,
+      "learning_rate": 8.55e-05,
+      "loss": 0.3146,
+      "step": 580
+    },
+    {
+      "epoch": 3.62,
+      "eval_accuracy_AK": 0.4115902206577829,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.12932070934776074,
+      "eval_accuracy_LA": 0.8100088441186611,
+      "eval_accuracy_LV": 0.6732256119229403,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.7254265895953758,
+      "eval_accuracy_RV": 0.45181304213690726,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.2673440719235812,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.39574010066969234,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.12115452305526223,
+      "eval_iou_LA": 0.6516722493736414,
+      "eval_iou_LV": 0.6556960531749622,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5842697256932159,
+      "eval_iou_RV": 0.4443732909940901,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.24335498610471756,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3641601502895355,
+      "eval_mean_accuracy": 0.3854143433003344,
+      "eval_mean_iou": 0.3096260929065582,
+      "eval_overall_accuracy": 0.596707619532546,
+      "eval_runtime": 4.34,
+      "eval_samples_per_second": 18.664,
+      "eval_steps_per_second": 9.447,
+      "step": 580
+    },
+    {
+      "epoch": 3.63,
+      "learning_rate": 8.5475e-05,
+      "loss": 0.3344,
+      "step": 581
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 8.545e-05,
+      "loss": 0.3221,
+      "step": 582
+    },
+    {
+      "epoch": 3.64,
+      "learning_rate": 8.5425e-05,
+      "loss": 0.3741,
+      "step": 583
+    },
+    {
+      "epoch": 3.65,
+      "learning_rate": 8.54e-05,
+      "loss": 0.2699,
+      "step": 584
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 8.5375e-05,
+      "loss": 0.2557,
+      "step": 585
+    },
+    {
+      "epoch": 3.66,
+      "learning_rate": 8.535e-05,
+      "loss": 0.3452,
+      "step": 586
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 8.5325e-05,
+      "loss": 0.2136,
+      "step": 587
+    },
+    {
+      "epoch": 3.67,
+      "learning_rate": 8.53e-05,
+      "loss": 0.3068,
+      "step": 588
+    },
+    {
+      "epoch": 3.68,
+      "learning_rate": 8.5275e-05,
+      "loss": 0.2309,
+      "step": 589
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 8.525000000000001e-05,
+      "loss": 0.2584,
+      "step": 590
+    },
+    {
+      "epoch": 3.69,
+      "learning_rate": 8.5225e-05,
+      "loss": 0.3901,
+      "step": 591
+    },
+    {
+      "epoch": 3.7,
+      "learning_rate": 8.52e-05,
+      "loss": 0.3071,
+      "step": 592
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 8.517500000000001e-05,
+      "loss": 0.5091,
+      "step": 593
+    },
+    {
+      "epoch": 3.71,
+      "learning_rate": 8.515000000000001e-05,
+      "loss": 0.2102,
+      "step": 594
+    },
+    {
+      "epoch": 3.72,
+      "learning_rate": 8.5125e-05,
+      "loss": 0.3538,
+      "step": 595
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 8.510000000000001e-05,
+      "loss": 0.5395,
+      "step": 596
+    },
+    {
+      "epoch": 3.73,
+      "learning_rate": 8.507500000000001e-05,
+      "loss": 0.1686,
+      "step": 597
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 8.505000000000001e-05,
+      "loss": 0.2501,
+      "step": 598
+    },
+    {
+      "epoch": 3.74,
+      "learning_rate": 8.502499999999999e-05,
+      "loss": 0.2605,
+      "step": 599
+    },
+    {
+      "epoch": 3.75,
+      "learning_rate": 8.5e-05,
+      "loss": 0.2216,
+      "step": 600
+    },
+    {
+      "epoch": 3.75,
+      "eval_accuracy_AK": 0.4242304514924033,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.1378452392879805,
+      "eval_accuracy_LA": 0.7168289337590502,
+      "eval_accuracy_LV": 0.7614469702680292,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5415421965317919,
+      "eval_accuracy_RV": 0.4294244470113862,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.42475182618467877,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.412502073686447,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.13382833334684316,
+      "eval_iou_LA": 0.6520234110292263,
+      "eval_iou_LV": 0.7253677123518395,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5274254857276682,
+      "eval_iou_RV": 0.42124460272160075,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.3687158558792924,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.31775519251823425,
+      "eval_mean_accuracy": 0.38178556272614667,
+      "eval_mean_iou": 0.32411074747429175,
+      "eval_overall_accuracy": 0.5998258955701921,
+      "eval_runtime": 4.3384,
+      "eval_samples_per_second": 18.67,
+      "eval_steps_per_second": 9.45,
+      "step": 600
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 8.4975e-05,
+      "loss": 0.2721,
+      "step": 601
+    },
+    {
+      "epoch": 3.76,
+      "learning_rate": 8.495e-05,
+      "loss": 0.3165,
+      "step": 602
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 8.4925e-05,
+      "loss": 0.1555,
+      "step": 603
+    },
+    {
+      "epoch": 3.77,
+      "learning_rate": 8.49e-05,
+      "loss": 0.8972,
+      "step": 604
+    },
+    {
+      "epoch": 3.78,
+      "learning_rate": 8.4875e-05,
+      "loss": 0.2969,
+      "step": 605
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 8.485e-05,
+      "loss": 0.2469,
+      "step": 606
+    },
+    {
+      "epoch": 3.79,
+      "learning_rate": 8.4825e-05,
+      "loss": 0.461,
+      "step": 607
+    },
+    {
+      "epoch": 3.8,
+      "learning_rate": 8.48e-05,
+      "loss": 0.1906,
+      "step": 608
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 8.4775e-05,
+      "loss": 0.3693,
+      "step": 609
+    },
+    {
+      "epoch": 3.81,
+      "learning_rate": 8.475000000000001e-05,
+      "loss": 0.4496,
+      "step": 610
+    },
+    {
+      "epoch": 3.82,
+      "learning_rate": 8.4725e-05,
+      "loss": 0.2221,
+      "step": 611
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 8.47e-05,
+      "loss": 0.328,
+      "step": 612
+    },
+    {
+      "epoch": 3.83,
+      "learning_rate": 8.4675e-05,
+      "loss": 0.2693,
+      "step": 613
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 8.465000000000001e-05,
+      "loss": 0.4204,
+      "step": 614
+    },
+    {
+      "epoch": 3.84,
+      "learning_rate": 8.4625e-05,
+      "loss": 0.2607,
+      "step": 615
+    },
+    {
+      "epoch": 3.85,
+      "learning_rate": 8.46e-05,
+      "loss": 0.4076,
+      "step": 616
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 8.457500000000001e-05,
+      "loss": 0.4614,
+      "step": 617
+    },
+    {
+      "epoch": 3.86,
+      "learning_rate": 8.455000000000001e-05,
+      "loss": 0.3574,
+      "step": 618
+    },
+    {
+      "epoch": 3.87,
+      "learning_rate": 8.4525e-05,
+      "loss": 0.3057,
+      "step": 619
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 8.450000000000001e-05,
+      "loss": 0.2973,
+      "step": 620
+    },
+    {
+      "epoch": 3.88,
+      "eval_accuracy_AK": 0.487566299103241,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.19648999766222489,
+      "eval_accuracy_LA": 0.7738903629086658,
+      "eval_accuracy_LV": 0.7369323627278894,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5558080924855492,
+      "eval_accuracy_RV": 0.5177537505650062,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.4455796965723918,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.4691846616030255,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.18591166549487703,
+      "eval_iou_LA": 0.7010394671627502,
+      "eval_iou_LV": 0.7072031457086241,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5406997537028915,
+      "eval_iou_RV": 0.49743146996431115,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.38451405388805376,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3198661506175995,
+      "eval_mean_accuracy": 0.4126689513361076,
+      "eval_mean_iou": 0.3485984217524533,
+      "eval_overall_accuracy": 0.6216853089806131,
+      "eval_runtime": 4.3245,
+      "eval_samples_per_second": 18.73,
+      "eval_steps_per_second": 9.481,
+      "step": 620
+    },
+    {
+      "epoch": 3.88,
+      "learning_rate": 8.447500000000001e-05,
+      "loss": 0.3165,
+      "step": 621
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 8.445e-05,
+      "loss": 0.1703,
+      "step": 622
+    },
+    {
+      "epoch": 3.89,
+      "learning_rate": 8.442499999999999e-05,
+      "loss": 0.2809,
+      "step": 623
+    },
+    {
+      "epoch": 3.9,
+      "learning_rate": 8.44e-05,
+      "loss": 0.2829,
+      "step": 624
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 8.4375e-05,
+      "loss": 0.2978,
+      "step": 625
+    },
+    {
+      "epoch": 3.91,
+      "learning_rate": 8.435e-05,
+      "loss": 0.2781,
+      "step": 626
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 8.4325e-05,
+      "loss": 0.2826,
+      "step": 627
+    },
+    {
+      "epoch": 3.92,
+      "learning_rate": 8.43e-05,
+      "loss": 0.247,
+      "step": 628
+    },
+    {
+      "epoch": 3.93,
+      "learning_rate": 8.4275e-05,
+      "loss": 0.257,
+      "step": 629
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 8.425e-05,
+      "loss": 0.302,
+      "step": 630
+    },
+    {
+      "epoch": 3.94,
+      "learning_rate": 8.422500000000001e-05,
+      "loss": 0.3157,
+      "step": 631
+    },
+    {
+      "epoch": 3.95,
+      "learning_rate": 8.42e-05,
+      "loss": 0.2609,
+      "step": 632
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 8.4175e-05,
+      "loss": 0.5228,
+      "step": 633
+    },
+    {
+      "epoch": 3.96,
+      "learning_rate": 8.415000000000001e-05,
+      "loss": 0.2545,
+      "step": 634
+    },
+    {
+      "epoch": 3.97,
+      "learning_rate": 8.412500000000001e-05,
+      "loss": 0.2994,
+      "step": 635
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 8.41e-05,
+      "loss": 0.2663,
+      "step": 636
+    },
+    {
+      "epoch": 3.98,
+      "learning_rate": 8.4075e-05,
+      "loss": 0.2617,
+      "step": 637
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 8.405000000000001e-05,
+      "loss": 0.4235,
+      "step": 638
+    },
+    {
+      "epoch": 3.99,
+      "learning_rate": 8.402500000000001e-05,
+      "loss": 0.4619,
+      "step": 639
+    },
+    {
+      "epoch": 4.0,
+      "learning_rate": 8.4e-05,
+      "loss": 0.2434,
+      "step": 640
+    },
+    {
+      "epoch": 4.0,
+      "eval_accuracy_AK": 0.5376303533158531,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.12952109007113516,
+      "eval_accuracy_LA": 0.7444724258368185,
+      "eval_accuracy_LV": 0.7160925233946782,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6826867052023121,
+      "eval_accuracy_RV": 0.45819133441910187,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.3827495785727664,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.5133967107030772,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.12537277245726755,
+      "eval_iou_LA": 0.6769358683188712,
+      "eval_iou_LV": 0.6868721820820444,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.6280446430660521,
+      "eval_iou_RV": 0.44831580369851015,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.33601907424155225,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3179140090942383,
+      "eval_mean_accuracy": 0.40570489009029614,
+      "eval_mean_iou": 0.34149570545673746,
+      "eval_overall_accuracy": 0.6153890796007643,
+      "eval_runtime": 4.2849,
+      "eval_samples_per_second": 18.904,
+      "eval_steps_per_second": 9.569,
+      "step": 640
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 8.397500000000001e-05,
+      "loss": 0.2999,
+      "step": 641
+    },
+    {
+      "epoch": 4.01,
+      "learning_rate": 8.395000000000001e-05,
+      "loss": 0.5324,
+      "step": 642
+    },
+    {
+      "epoch": 4.02,
+      "learning_rate": 8.392500000000001e-05,
+      "loss": 0.3431,
+      "step": 643
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 8.39e-05,
+      "loss": 0.2549,
+      "step": 644
+    },
+    {
+      "epoch": 4.03,
+      "learning_rate": 8.3875e-05,
+      "loss": 0.2728,
+      "step": 645
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 8.385e-05,
+      "loss": 0.4355,
+      "step": 646
+    },
+    {
+      "epoch": 4.04,
+      "learning_rate": 8.3825e-05,
+      "loss": 0.5573,
+      "step": 647
+    },
+    {
+      "epoch": 4.05,
+      "learning_rate": 8.38e-05,
+      "loss": 0.3938,
+      "step": 648
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 8.3775e-05,
+      "loss": 0.3112,
+      "step": 649
+    },
+    {
+      "epoch": 4.06,
+      "learning_rate": 8.375e-05,
+      "loss": 0.3231,
+      "step": 650
+    },
+    {
+      "epoch": 4.07,
+      "learning_rate": 8.3725e-05,
+      "loss": 0.3148,
+      "step": 651
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 8.37e-05,
+      "loss": 0.2475,
+      "step": 652
+    },
+    {
+      "epoch": 4.08,
+      "learning_rate": 8.3675e-05,
+      "loss": 0.2026,
+      "step": 653
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 8.365e-05,
+      "loss": 0.3166,
+      "step": 654
+    },
+    {
+      "epoch": 4.09,
+      "learning_rate": 8.362500000000001e-05,
+      "loss": 0.2806,
+      "step": 655
+    },
+    {
+      "epoch": 4.1,
+      "learning_rate": 8.36e-05,
+      "loss": 0.2985,
+      "step": 656
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 8.3575e-05,
+      "loss": 0.2739,
+      "step": 657
+    },
+    {
+      "epoch": 4.11,
+      "learning_rate": 8.355000000000001e-05,
+      "loss": 0.2218,
+      "step": 658
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 8.352500000000001e-05,
+      "loss": 0.2365,
+      "step": 659
+    },
+    {
+      "epoch": 4.12,
+      "learning_rate": 8.35e-05,
+      "loss": 0.2283,
+      "step": 660
+    },
+    {
+      "epoch": 4.12,
+      "eval_accuracy_AK": 0.3463566921686242,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.07180309254249741,
+      "eval_accuracy_LA": 0.7014435400458695,
+      "eval_accuracy_LV": 0.6236741220938372,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6672624277456647,
+      "eval_accuracy_RV": 0.2267307126611613,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.12884435287507023,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3401171591406762,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.07067950951707815,
+      "eval_iou_LA": 0.6270416222092317,
+      "eval_iou_LV": 0.6078184822313483,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5911533175059352,
+      "eval_iou_RV": 0.22595054706394418,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.12122013110594206,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3310217559337616,
+      "eval_mean_accuracy": 0.30734610445919164,
+      "eval_mean_iou": 0.25839807687741556,
+      "eval_overall_accuracy": 0.5075532852629908,
+      "eval_runtime": 4.3828,
+      "eval_samples_per_second": 18.481,
+      "eval_steps_per_second": 9.355,
+      "step": 660
+    },
+    {
+      "epoch": 4.13,
+      "learning_rate": 8.347500000000001e-05,
+      "loss": 0.2858,
+      "step": 661
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 8.345000000000001e-05,
+      "loss": 0.4033,
+      "step": 662
+    },
+    {
+      "epoch": 4.14,
+      "learning_rate": 8.342500000000001e-05,
+      "loss": 0.2277,
+      "step": 663
+    },
+    {
+      "epoch": 4.15,
+      "learning_rate": 8.34e-05,
+      "loss": 0.2091,
+      "step": 664
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 8.337500000000001e-05,
+      "loss": 0.2882,
+      "step": 665
+    },
+    {
+      "epoch": 4.16,
+      "learning_rate": 8.335e-05,
+      "loss": 0.4355,
+      "step": 666
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 8.3325e-05,
+      "loss": 0.2315,
+      "step": 667
+    },
+    {
+      "epoch": 4.17,
+      "learning_rate": 8.33e-05,
+      "loss": 0.3257,
+      "step": 668
+    },
+    {
+      "epoch": 4.18,
+      "learning_rate": 8.3275e-05,
+      "loss": 0.2647,
+      "step": 669
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 8.325e-05,
+      "loss": 0.3527,
+      "step": 670
+    },
+    {
+      "epoch": 4.19,
+      "learning_rate": 8.3225e-05,
+      "loss": 0.1399,
+      "step": 671
+    },
+    {
+      "epoch": 4.2,
+      "learning_rate": 8.32e-05,
+      "loss": 0.3516,
+      "step": 672
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 8.3175e-05,
+      "loss": 0.1789,
+      "step": 673
+    },
+    {
+      "epoch": 4.21,
+      "learning_rate": 8.315e-05,
+      "loss": 0.4795,
+      "step": 674
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 8.312500000000001e-05,
+      "loss": 0.3938,
+      "step": 675
+    },
+    {
+      "epoch": 4.22,
+      "learning_rate": 8.31e-05,
+      "loss": 0.264,
+      "step": 676
+    },
+    {
+      "epoch": 4.23,
+      "learning_rate": 8.3075e-05,
+      "loss": 0.4486,
+      "step": 677
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 8.305e-05,
+      "loss": 0.4039,
+      "step": 678
+    },
+    {
+      "epoch": 4.24,
+      "learning_rate": 8.302500000000001e-05,
+      "loss": 0.5576,
+      "step": 679
+    },
+    {
+      "epoch": 4.25,
+      "learning_rate": 8.3e-05,
+      "loss": 0.6263,
+      "step": 680
+    },
+    {
+      "epoch": 4.25,
+      "eval_accuracy_AK": 0.42477521161835663,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.15117890658918612,
+      "eval_accuracy_LA": 0.6236059270585061,
+      "eval_accuracy_LV": 0.646090208156101,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5694011560693641,
+      "eval_accuracy_RV": 0.2823990701612151,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.3949709683461322,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.41397637737852494,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.14493832496858217,
+      "eval_iou_LA": 0.5851519094169773,
+      "eval_iou_LV": 0.624437579724314,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5497536593914149,
+      "eval_iou_RV": 0.28109422535814776,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.347932616156283,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.31528791785240173,
+      "eval_mean_accuracy": 0.3436023831109846,
+      "eval_mean_iou": 0.2947284692394244,
+      "eval_overall_accuracy": 0.5197622215287245,
+      "eval_runtime": 4.321,
+      "eval_samples_per_second": 18.746,
+      "eval_steps_per_second": 9.489,
+      "step": 680
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 8.2975e-05,
+      "loss": 0.3236,
+      "step": 681
+    },
+    {
+      "epoch": 4.26,
+      "learning_rate": 8.295000000000001e-05,
+      "loss": 0.2213,
+      "step": 682
+    },
+    {
+      "epoch": 4.27,
+      "learning_rate": 8.292500000000001e-05,
+      "loss": 0.1885,
+      "step": 683
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 8.29e-05,
+      "loss": 0.3411,
+      "step": 684
+    },
+    {
+      "epoch": 4.28,
+      "learning_rate": 8.287500000000001e-05,
+      "loss": 0.3991,
+      "step": 685
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 8.285000000000001e-05,
+      "loss": 0.2323,
+      "step": 686
+    },
+    {
+      "epoch": 4.29,
+      "learning_rate": 8.282500000000001e-05,
+      "loss": 0.255,
+      "step": 687
+    },
+    {
+      "epoch": 4.3,
+      "learning_rate": 8.28e-05,
+      "loss": 0.3426,
+      "step": 688
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 8.2775e-05,
+      "loss": 0.2286,
+      "step": 689
+    },
+    {
+      "epoch": 4.31,
+      "learning_rate": 8.275e-05,
+      "loss": 0.2001,
+      "step": 690
+    },
+    {
+      "epoch": 4.32,
+      "learning_rate": 8.2725e-05,
+      "loss": 0.245,
+      "step": 691
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 8.27e-05,
+      "loss": 0.2131,
+      "step": 692
+    },
+    {
+      "epoch": 4.33,
+      "learning_rate": 8.2675e-05,
+      "loss": 0.3293,
+      "step": 693
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 8.265e-05,
+      "loss": 0.5444,
+      "step": 694
+    },
+    {
+      "epoch": 4.34,
+      "learning_rate": 8.2625e-05,
+      "loss": 0.2451,
+      "step": 695
+    },
+    {
+      "epoch": 4.35,
+      "learning_rate": 8.26e-05,
+      "loss": 0.1994,
+      "step": 696
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 8.2575e-05,
+      "loss": 0.3549,
+      "step": 697
+    },
+    {
+      "epoch": 4.36,
+      "learning_rate": 8.255e-05,
+      "loss": 0.1955,
+      "step": 698
+    },
+    {
+      "epoch": 4.37,
+      "learning_rate": 8.252500000000001e-05,
+      "loss": 0.2614,
+      "step": 699
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 8.25e-05,
+      "loss": 0.1708,
+      "step": 700
+    },
+    {
+      "epoch": 4.38,
+      "eval_accuracy_AK": 0.39079654705889394,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.13309454630464548,
+      "eval_accuracy_LA": 0.7152849605012667,
+      "eval_accuracy_LV": 0.753067018080398,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5852971098265896,
+      "eval_accuracy_RV": 0.39017355555714994,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.12388087656864581,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.37878532925619257,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.1298232755110351,
+      "eval_iou_LA": 0.6612489173739823,
+      "eval_iou_LV": 0.7187677859074587,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5655935312478355,
+      "eval_iou_RV": 0.38693932149362475,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.11594456959040748,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.31732410192489624,
+      "eval_mean_accuracy": 0.3435105126552877,
+      "eval_mean_iou": 0.29571027303805364,
+      "eval_overall_accuracy": 0.5833735744515437,
+      "eval_runtime": 4.2501,
+      "eval_samples_per_second": 19.058,
+      "eval_steps_per_second": 9.647,
+      "step": 700
+    },
+    {
+      "epoch": 4.38,
+      "learning_rate": 8.2475e-05,
+      "loss": 0.2081,
+      "step": 701
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 8.245e-05,
+      "loss": 0.4086,
+      "step": 702
+    },
+    {
+      "epoch": 4.39,
+      "learning_rate": 8.242500000000001e-05,
+      "loss": 0.2864,
+      "step": 703
+    },
+    {
+      "epoch": 4.4,
+      "learning_rate": 8.24e-05,
+      "loss": 0.2048,
+      "step": 704
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 8.2375e-05,
+      "loss": 0.2829,
+      "step": 705
+    },
+    {
+      "epoch": 4.41,
+      "learning_rate": 8.235000000000001e-05,
+      "loss": 0.4265,
+      "step": 706
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 8.232500000000001e-05,
+      "loss": 0.2017,
+      "step": 707
+    },
+    {
+      "epoch": 4.42,
+      "learning_rate": 8.23e-05,
+      "loss": 0.2414,
+      "step": 708
+    },
+    {
+      "epoch": 4.43,
+      "learning_rate": 8.227500000000001e-05,
+      "loss": 0.2947,
+      "step": 709
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 8.225000000000001e-05,
+      "loss": 0.3867,
+      "step": 710
+    },
+    {
+      "epoch": 4.44,
+      "learning_rate": 8.2225e-05,
+      "loss": 0.2438,
+      "step": 711
+    },
+    {
+      "epoch": 4.45,
+      "learning_rate": 8.22e-05,
+      "loss": 0.2151,
+      "step": 712
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 8.2175e-05,
+      "loss": 0.4297,
+      "step": 713
+    },
+    {
+      "epoch": 4.46,
+      "learning_rate": 8.215e-05,
+      "loss": 0.2723,
+      "step": 714
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 8.2125e-05,
+      "loss": 0.316,
+      "step": 715
+    },
+    {
+      "epoch": 4.47,
+      "learning_rate": 8.21e-05,
+      "loss": 0.344,
+      "step": 716
+    },
+    {
+      "epoch": 4.48,
+      "learning_rate": 8.2075e-05,
+      "loss": 0.3114,
+      "step": 717
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 8.205e-05,
+      "loss": 0.2425,
+      "step": 718
+    },
+    {
+      "epoch": 4.49,
+      "learning_rate": 8.2025e-05,
+      "loss": 0.2098,
+      "step": 719
+    },
+    {
+      "epoch": 4.5,
+      "learning_rate": 8.2e-05,
+      "loss": 0.246,
+      "step": 720
+    },
+    {
+      "epoch": 4.5,
+      "eval_accuracy_AK": 0.3459226799803647,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.12365995391243362,
+      "eval_accuracy_LA": 0.5988929861642008,
+      "eval_accuracy_LV": 0.6685587909070351,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.4552231213872832,
+      "eval_accuracy_RV": 0.3447111831768057,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.12296310170443904,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.33663845082812416,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.12125057305651975,
+      "eval_iou_LA": 0.5639927088314544,
+      "eval_iou_LV": 0.6468718518436515,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.4442498922566976,
+      "eval_iou_RV": 0.341554877832043,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.11586759501937009,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.31382229924201965,
+      "eval_mean_accuracy": 0.2955479796925069,
+      "eval_mean_iou": 0.257042594966786,
+      "eval_overall_accuracy": 0.5052173841630668,
+      "eval_runtime": 4.3083,
+      "eval_samples_per_second": 18.801,
+      "eval_steps_per_second": 9.517,
+      "step": 720
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 8.1975e-05,
+      "loss": 0.597,
+      "step": 721
+    },
+    {
+      "epoch": 4.51,
+      "learning_rate": 8.195e-05,
+      "loss": 0.1902,
+      "step": 722
+    },
+    {
+      "epoch": 4.52,
+      "learning_rate": 8.192500000000001e-05,
+      "loss": 0.3064,
+      "step": 723
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 8.19e-05,
+      "loss": 0.2641,
+      "step": 724
+    },
+    {
+      "epoch": 4.53,
+      "learning_rate": 8.1875e-05,
+      "loss": 0.3346,
+      "step": 725
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 8.185000000000001e-05,
+      "loss": 0.654,
+      "step": 726
+    },
+    {
+      "epoch": 4.54,
+      "learning_rate": 8.182500000000001e-05,
+      "loss": 0.4987,
+      "step": 727
+    },
+    {
+      "epoch": 4.55,
+      "learning_rate": 8.18e-05,
+      "loss": 0.2002,
+      "step": 728
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 8.1775e-05,
+      "loss": 0.2177,
+      "step": 729
+    },
+    {
+      "epoch": 4.56,
+      "learning_rate": 8.175000000000001e-05,
+      "loss": 0.3221,
+      "step": 730
+    },
+    {
+      "epoch": 4.57,
+      "learning_rate": 8.172500000000001e-05,
+      "loss": 0.3144,
+      "step": 731
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 8.17e-05,
+      "loss": 0.3337,
+      "step": 732
+    },
+    {
+      "epoch": 4.58,
+      "learning_rate": 8.1675e-05,
+      "loss": 0.3435,
+      "step": 733
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 8.165e-05,
+      "loss": 0.4919,
+      "step": 734
+    },
+    {
+      "epoch": 4.59,
+      "learning_rate": 8.1625e-05,
+      "loss": 0.1782,
+      "step": 735
+    },
+    {
+      "epoch": 4.6,
+      "learning_rate": 8.16e-05,
+      "loss": 0.5148,
+      "step": 736
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 8.1575e-05,
+      "loss": 0.3272,
+      "step": 737
+    },
+    {
+      "epoch": 4.61,
+      "learning_rate": 8.155e-05,
+      "loss": 0.2598,
+      "step": 738
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 8.1525e-05,
+      "loss": 0.2946,
+      "step": 739
+    },
+    {
+      "epoch": 4.62,
+      "learning_rate": 8.15e-05,
+      "loss": 0.3876,
+      "step": 740
+    },
+    {
+      "epoch": 4.62,
+      "eval_accuracy_AK": 0.5539372388443902,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.23511338209264268,
+      "eval_accuracy_LA": 0.7478245716598462,
+      "eval_accuracy_LV": 0.7367378557660209,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6535815028901734,
+      "eval_accuracy_RV": 0.4775611821006034,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.4972841356059187,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.5273737604012311,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.22083676430223895,
+      "eval_iou_LA": 0.6824244144492283,
+      "eval_iou_LV": 0.7107680300448803,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.6166266742288731,
+      "eval_iou_RV": 0.4699577784995128,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.41794898031468175,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3083554208278656,
+      "eval_mean_accuracy": 0.433559985439955,
+      "eval_mean_iou": 0.3645936402240647,
+      "eval_overall_accuracy": 0.6320877749125372,
+      "eval_runtime": 4.3946,
+      "eval_samples_per_second": 18.432,
+      "eval_steps_per_second": 9.33,
+      "step": 740
+    },
+    {
+      "epoch": 4.63,
+      "learning_rate": 8.1475e-05,
+      "loss": 0.2593,
+      "step": 741
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 8.145e-05,
+      "loss": 0.2021,
+      "step": 742
+    },
+    {
+      "epoch": 4.64,
+      "learning_rate": 8.1425e-05,
+      "loss": 0.3468,
+      "step": 743
+    },
+    {
+      "epoch": 4.65,
+      "learning_rate": 8.14e-05,
+      "loss": 0.576,
+      "step": 744
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 8.1375e-05,
+      "loss": 0.2244,
+      "step": 745
+    },
+    {
+      "epoch": 4.66,
+      "learning_rate": 8.135e-05,
+      "loss": 0.1771,
+      "step": 746
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 8.132500000000001e-05,
+      "loss": 0.3237,
+      "step": 747
+    },
+    {
+      "epoch": 4.67,
+      "learning_rate": 8.13e-05,
+      "loss": 0.4276,
+      "step": 748
+    },
+    {
+      "epoch": 4.68,
+      "learning_rate": 8.1275e-05,
+      "loss": 0.313,
+      "step": 749
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 8.125000000000001e-05,
+      "loss": 0.2453,
+      "step": 750
+    },
+    {
+      "epoch": 4.69,
+      "learning_rate": 8.122500000000001e-05,
+      "loss": 0.2059,
+      "step": 751
+    },
+    {
+      "epoch": 4.7,
+      "learning_rate": 8.120000000000001e-05,
+      "loss": 0.199,
+      "step": 752
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 8.1175e-05,
+      "loss": 0.1687,
+      "step": 753
+    },
+    {
+      "epoch": 4.71,
+      "learning_rate": 8.115000000000001e-05,
+      "loss": 0.1383,
+      "step": 754
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 8.112500000000001e-05,
+      "loss": 0.3482,
+      "step": 755
+    },
+    {
+      "epoch": 4.72,
+      "learning_rate": 8.11e-05,
+      "loss": 0.3062,
+      "step": 756
+    },
+    {
+      "epoch": 4.73,
+      "learning_rate": 8.1075e-05,
+      "loss": 0.2911,
+      "step": 757
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 8.105e-05,
+      "loss": 0.2848,
+      "step": 758
+    },
+    {
+      "epoch": 4.74,
+      "learning_rate": 8.1025e-05,
+      "loss": 0.2975,
+      "step": 759
+    },
+    {
+      "epoch": 4.75,
+      "learning_rate": 8.1e-05,
+      "loss": 0.4766,
+      "step": 760
+    },
+    {
+      "epoch": 4.75,
+      "eval_accuracy_AK": 0.39724985932018725,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.11191263400460875,
+      "eval_accuracy_LA": 0.4988757476278275,
+      "eval_accuracy_LV": 0.6095168420145807,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.43483699421965316,
+      "eval_accuracy_RV": 0.34980700105467827,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.20196666042330025,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.38649686796529864,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.10923315133240975,
+      "eval_iou_LA": 0.47542213079626294,
+      "eval_iou_LV": 0.5954075463982322,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.4156966848948087,
+      "eval_iou_RV": 0.34344467846802995,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.18360136555963255,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3251356780529022,
+      "eval_mean_accuracy": 0.28935174874053726,
+      "eval_mean_iou": 0.25093024254146745,
+      "eval_overall_accuracy": 0.4715730718481897,
+      "eval_runtime": 4.3734,
+      "eval_samples_per_second": 18.521,
+      "eval_steps_per_second": 9.375,
+      "step": 760
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 8.0975e-05,
+      "loss": 0.3569,
+      "step": 761
+    },
+    {
+      "epoch": 4.76,
+      "learning_rate": 8.095e-05,
+      "loss": 0.176,
+      "step": 762
+    },
+    {
+      "epoch": 4.77,
+      "learning_rate": 8.0925e-05,
+      "loss": 0.5118,
+      "step": 763
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 8.090000000000001e-05,
+      "loss": 0.6116,
+      "step": 764
+    },
+    {
+      "epoch": 4.78,
+      "learning_rate": 8.0875e-05,
+      "loss": 0.2398,
+      "step": 765
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 8.085e-05,
+      "loss": 0.2234,
+      "step": 766
+    },
+    {
+      "epoch": 4.79,
+      "learning_rate": 8.082500000000001e-05,
+      "loss": 0.3343,
+      "step": 767
+    },
+    {
+      "epoch": 4.8,
+      "learning_rate": 8.080000000000001e-05,
+      "loss": 0.1105,
+      "step": 768
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 8.0775e-05,
+      "loss": 0.2273,
+      "step": 769
+    },
+    {
+      "epoch": 4.81,
+      "learning_rate": 8.075e-05,
+      "loss": 0.2676,
+      "step": 770
+    },
+    {
+      "epoch": 4.82,
+      "learning_rate": 8.072500000000001e-05,
+      "loss": 0.3629,
+      "step": 771
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 8.070000000000001e-05,
+      "loss": 0.2139,
+      "step": 772
+    },
+    {
+      "epoch": 4.83,
+      "learning_rate": 8.0675e-05,
+      "loss": 0.4253,
+      "step": 773
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 8.065000000000001e-05,
+      "loss": 0.5303,
+      "step": 774
+    },
+    {
+      "epoch": 4.84,
+      "learning_rate": 8.062500000000001e-05,
+      "loss": 0.2045,
+      "step": 775
+    },
+    {
+      "epoch": 4.85,
+      "learning_rate": 8.060000000000001e-05,
+      "loss": 0.2132,
+      "step": 776
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 8.057500000000001e-05,
+      "loss": 0.3776,
+      "step": 777
+    },
+    {
+      "epoch": 4.86,
+      "learning_rate": 8.055e-05,
+      "loss": 0.2538,
+      "step": 778
+    },
+    {
+      "epoch": 4.87,
+      "learning_rate": 8.0525e-05,
+      "loss": 0.5849,
+      "step": 779
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 8.05e-05,
+      "loss": 0.4431,
+      "step": 780
+    },
+    {
+      "epoch": 4.88,
+      "eval_accuracy_AK": 0.38657016630149776,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.21626924489864074,
+      "eval_accuracy_LA": 0.7363478286939186,
+      "eval_accuracy_LV": 0.7023511786179641,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5149734104046243,
+      "eval_accuracy_RV": 0.48621205490066655,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.2456358868702004,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.37711002035196933,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.2020672439347843,
+      "eval_iou_LA": 0.6460585954004205,
+      "eval_iou_LV": 0.6800729367128039,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5039004171907947,
+      "eval_iou_RV": 0.4735981410463603,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.2209483535645391,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3051846921443939,
+      "eval_mean_accuracy": 0.3653733078541681,
+      "eval_mean_iou": 0.3103755708201672,
+      "eval_overall_accuracy": 0.5780535343746749,
+      "eval_runtime": 4.3349,
+      "eval_samples_per_second": 18.686,
+      "eval_steps_per_second": 9.458,
+      "step": 780
+    },
+    {
+      "epoch": 4.88,
+      "learning_rate": 8.0475e-05,
+      "loss": 0.2619,
+      "step": 781
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 8.045e-05,
+      "loss": 0.1361,
+      "step": 782
+    },
+    {
+      "epoch": 4.89,
+      "learning_rate": 8.0425e-05,
+      "loss": 0.2012,
+      "step": 783
+    },
+    {
+      "epoch": 4.9,
+      "learning_rate": 8.04e-05,
+      "loss": 0.2021,
+      "step": 784
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 8.0375e-05,
+      "loss": 0.2149,
+      "step": 785
+    },
+    {
+      "epoch": 4.91,
+      "learning_rate": 8.035e-05,
+      "loss": 0.2373,
+      "step": 786
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 8.0325e-05,
+      "loss": 0.3817,
+      "step": 787
+    },
+    {
+      "epoch": 4.92,
+      "learning_rate": 8.030000000000001e-05,
+      "loss": 0.3508,
+      "step": 788
+    },
+    {
+      "epoch": 4.93,
+      "learning_rate": 8.0275e-05,
+      "loss": 0.3538,
+      "step": 789
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 8.025e-05,
+      "loss": 0.2399,
+      "step": 790
+    },
+    {
+      "epoch": 4.94,
+      "learning_rate": 8.022500000000001e-05,
+      "loss": 0.8289,
+      "step": 791
+    },
+    {
+      "epoch": 4.95,
+      "learning_rate": 8.020000000000001e-05,
+      "loss": 0.4189,
+      "step": 792
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 8.0175e-05,
+      "loss": 0.2794,
+      "step": 793
+    },
+    {
+      "epoch": 4.96,
+      "learning_rate": 8.015e-05,
+      "loss": 0.2775,
+      "step": 794
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 8.012500000000001e-05,
+      "loss": 0.1834,
+      "step": 795
+    },
+    {
+      "epoch": 4.97,
+      "learning_rate": 8.010000000000001e-05,
+      "loss": 0.3179,
+      "step": 796
+    },
+    {
+      "epoch": 4.98,
+      "learning_rate": 8.0075e-05,
+      "loss": 0.2557,
+      "step": 797
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 8.005000000000001e-05,
+      "loss": 0.2173,
+      "step": 798
+    },
+    {
+      "epoch": 4.99,
+      "learning_rate": 8.002500000000001e-05,
+      "loss": 0.2328,
+      "step": 799
+    },
+    {
+      "epoch": 5.0,
+      "learning_rate": 8e-05,
+      "loss": 0.2319,
+      "step": 800
+    },
+    {
+      "epoch": 5.0,
+      "eval_accuracy_AK": 0.36753648695568886,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.2907858264035,
+      "eval_accuracy_LA": 0.6571648603678554,
+      "eval_accuracy_LV": 0.6789867869872823,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6711190751445086,
+      "eval_accuracy_RV": 0.5432956184217135,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.33488480988949243,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.35846467783982927,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.26564764389120255,
+      "eval_iou_LA": 0.6112667120978944,
+      "eval_iou_LV": 0.6628492734140571,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5922214809056576,
+      "eval_iou_RV": 0.5264570312228425,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.2986511880402556,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3104110062122345,
+      "eval_mean_accuracy": 0.3937526071300046,
+      "eval_mean_iou": 0.3315558007411739,
+      "eval_overall_accuracy": 0.5874924034623786,
+      "eval_runtime": 4.3888,
+      "eval_samples_per_second": 18.456,
+      "eval_steps_per_second": 9.342,
+      "step": 800
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 7.9975e-05,
+      "loss": 0.2245,
+      "step": 801
+    },
+    {
+      "epoch": 5.01,
+      "learning_rate": 7.995e-05,
+      "loss": 0.2429,
+      "step": 802
+    },
+    {
+      "epoch": 5.02,
+      "learning_rate": 7.9925e-05,
+      "loss": 0.2182,
+      "step": 803
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 7.99e-05,
+      "loss": 0.6771,
+      "step": 804
+    },
+    {
+      "epoch": 5.03,
+      "learning_rate": 7.9875e-05,
+      "loss": 0.2512,
+      "step": 805
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 7.985e-05,
+      "loss": 0.3991,
+      "step": 806
+    },
+    {
+      "epoch": 5.04,
+      "learning_rate": 7.9825e-05,
+      "loss": 0.2515,
+      "step": 807
+    },
+    {
+      "epoch": 5.05,
+      "learning_rate": 7.98e-05,
+      "loss": 0.2359,
+      "step": 808
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 7.9775e-05,
+      "loss": 0.2257,
+      "step": 809
+    },
+    {
+      "epoch": 5.06,
+      "learning_rate": 7.975e-05,
+      "loss": 0.2834,
+      "step": 810
+    },
+    {
+      "epoch": 5.07,
+      "learning_rate": 7.9725e-05,
+      "loss": 0.2314,
+      "step": 811
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 7.970000000000001e-05,
+      "loss": 0.2958,
+      "step": 812
+    },
+    {
+      "epoch": 5.08,
+      "learning_rate": 7.9675e-05,
+      "loss": 0.2195,
+      "step": 813
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 7.965e-05,
+      "loss": 0.2106,
+      "step": 814
+    },
+    {
+      "epoch": 5.09,
+      "learning_rate": 7.962500000000001e-05,
+      "loss": 0.1791,
+      "step": 815
+    },
+    {
+      "epoch": 5.1,
+      "learning_rate": 7.960000000000001e-05,
+      "loss": 0.1667,
+      "step": 816
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 7.9575e-05,
+      "loss": 0.2145,
+      "step": 817
+    },
+    {
+      "epoch": 5.11,
+      "learning_rate": 7.955e-05,
+      "loss": 0.3587,
+      "step": 818
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 7.952500000000001e-05,
+      "loss": 0.3018,
+      "step": 819
+    },
+    {
+      "epoch": 5.12,
+      "learning_rate": 7.950000000000001e-05,
+      "loss": 0.3361,
+      "step": 820
+    },
+    {
+      "epoch": 5.12,
+      "eval_accuracy_AK": 0.4818014199681525,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.15584610760444845,
+      "eval_accuracy_LA": 0.7698243168293085,
+      "eval_accuracy_LV": 0.7234036968437375,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6026196531791908,
+      "eval_accuracy_RV": 0.3480115368886274,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.34614159955047763,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.4584358623832308,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.1497905532283692,
+      "eval_iou_LA": 0.6877418879648601,
+      "eval_iou_LV": 0.7027590202006465,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5676967846205452,
+      "eval_iou_RV": 0.3453264299609683,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.30765453062311676,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3124898672103882,
+      "eval_mean_accuracy": 0.38084981454043804,
+      "eval_mean_iou": 0.32194050689817366,
+      "eval_overall_accuracy": 0.5904584749984944,
+      "eval_runtime": 4.3558,
+      "eval_samples_per_second": 18.596,
+      "eval_steps_per_second": 9.413,
+      "step": 820
+    },
+    {
+      "epoch": 5.13,
+      "learning_rate": 7.9475e-05,
+      "loss": 0.2071,
+      "step": 821
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 7.945e-05,
+      "loss": 0.1533,
+      "step": 822
+    },
+    {
+      "epoch": 5.14,
+      "learning_rate": 7.9425e-05,
+      "loss": 0.3024,
+      "step": 823
+    },
+    {
+      "epoch": 5.15,
+      "learning_rate": 7.94e-05,
+      "loss": 0.2813,
+      "step": 824
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 7.9375e-05,
+      "loss": 0.2127,
+      "step": 825
+    },
+    {
+      "epoch": 5.16,
+      "learning_rate": 7.935e-05,
+      "loss": 0.5278,
+      "step": 826
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 7.9325e-05,
+      "loss": 0.3353,
+      "step": 827
+    },
+    {
+      "epoch": 5.17,
+      "learning_rate": 7.93e-05,
+      "loss": 0.2452,
+      "step": 828
+    },
+    {
+      "epoch": 5.18,
+      "learning_rate": 7.9275e-05,
+      "loss": 0.2378,
+      "step": 829
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 7.925e-05,
+      "loss": 0.4677,
+      "step": 830
+    },
+    {
+      "epoch": 5.19,
+      "learning_rate": 7.9225e-05,
+      "loss": 0.5807,
+      "step": 831
+    },
+    {
+      "epoch": 5.2,
+      "learning_rate": 7.920000000000001e-05,
+      "loss": 0.3998,
+      "step": 832
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 7.9175e-05,
+      "loss": 0.097,
+      "step": 833
+    },
+    {
+      "epoch": 5.21,
+      "learning_rate": 7.915e-05,
+      "loss": 0.1601,
+      "step": 834
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 7.9125e-05,
+      "loss": 0.2322,
+      "step": 835
+    },
+    {
+      "epoch": 5.22,
+      "learning_rate": 7.910000000000001e-05,
+      "loss": 0.2499,
+      "step": 836
+    },
+    {
+      "epoch": 5.23,
+      "learning_rate": 7.9075e-05,
+      "loss": 0.3892,
+      "step": 837
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 7.905e-05,
+      "loss": 0.1528,
+      "step": 838
+    },
+    {
+      "epoch": 5.24,
+      "learning_rate": 7.902500000000001e-05,
+      "loss": 0.345,
+      "step": 839
+    },
+    {
+      "epoch": 5.25,
+      "learning_rate": 7.900000000000001e-05,
+      "loss": 0.412,
+      "step": 840
+    },
+    {
+      "epoch": 5.25,
+      "eval_accuracy_AK": 0.28142846880499983,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.12377684266773537,
+      "eval_accuracy_LA": 0.6917768433991396,
+      "eval_accuracy_LV": 0.3970223608972357,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5965479768786127,
+      "eval_accuracy_RV": 0.27681896124954264,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.16140663045514142,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.27669122926794815,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.12077983445219319,
+      "eval_iou_LA": 0.644599370054544,
+      "eval_iou_LV": 0.3899106885547568,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5702933836710424,
+      "eval_iou_RV": 0.27519645936273524,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.14940446262937984,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.347745805978775,
+      "eval_mean_accuracy": 0.28097534270582303,
+      "eval_mean_iou": 0.24268754279925994,
+      "eval_overall_accuracy": 0.4099428411871951,
+      "eval_runtime": 4.392,
+      "eval_samples_per_second": 18.443,
+      "eval_steps_per_second": 9.335,
+      "step": 840
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 7.8975e-05,
+      "loss": 0.1544,
+      "step": 841
+    },
+    {
+      "epoch": 5.26,
+      "learning_rate": 7.895000000000001e-05,
+      "loss": 0.2241,
+      "step": 842
+    },
+    {
+      "epoch": 5.27,
+      "learning_rate": 7.892500000000001e-05,
+      "loss": 0.5474,
+      "step": 843
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 7.890000000000001e-05,
+      "loss": 0.2954,
+      "step": 844
+    },
+    {
+      "epoch": 5.28,
+      "learning_rate": 7.887499999999999e-05,
+      "loss": 0.1243,
+      "step": 845
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 7.885e-05,
+      "loss": 0.2063,
+      "step": 846
+    },
+    {
+      "epoch": 5.29,
+      "learning_rate": 7.8825e-05,
+      "loss": 0.1432,
+      "step": 847
+    },
+    {
+      "epoch": 5.3,
+      "learning_rate": 7.88e-05,
+      "loss": 0.2151,
+      "step": 848
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 7.8775e-05,
+      "loss": 0.5666,
+      "step": 849
+    },
+    {
+      "epoch": 5.31,
+      "learning_rate": 7.875e-05,
+      "loss": 0.1142,
+      "step": 850
+    },
+    {
+      "epoch": 5.32,
+      "learning_rate": 7.8725e-05,
+      "loss": 0.6193,
+      "step": 851
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 7.87e-05,
+      "loss": 0.3058,
+      "step": 852
+    },
+    {
+      "epoch": 5.33,
+      "learning_rate": 7.8675e-05,
+      "loss": 0.2948,
+      "step": 853
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 7.865e-05,
+      "loss": 0.2776,
+      "step": 854
+    },
+    {
+      "epoch": 5.34,
+      "learning_rate": 7.8625e-05,
+      "loss": 0.2357,
+      "step": 855
+    },
+    {
+      "epoch": 5.35,
+      "learning_rate": 7.860000000000001e-05,
+      "loss": 0.226,
+      "step": 856
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 7.8575e-05,
+      "loss": 0.337,
+      "step": 857
+    },
+    {
+      "epoch": 5.36,
+      "learning_rate": 7.855e-05,
+      "loss": 0.2459,
+      "step": 858
+    },
+    {
+      "epoch": 5.37,
+      "learning_rate": 7.8525e-05,
+      "loss": 0.192,
+      "step": 859
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 7.850000000000001e-05,
+      "loss": 0.1799,
+      "step": 860
+    },
+    {
+      "epoch": 5.38,
+      "eval_accuracy_AK": 0.5226524430396418,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.216745149116655,
+      "eval_accuracy_LA": 0.5953422224220893,
+      "eval_accuracy_LV": 0.7650961631650927,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6609317919075145,
+      "eval_accuracy_RV": 0.4804256738820052,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.39008241243678593,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.4965096407844565,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.20327622388573935,
+      "eval_iou_LA": 0.5642581621345154,
+      "eval_iou_LV": 0.7389385363012577,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.6239765124477479,
+      "eval_iou_RV": 0.4759032291307608,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.3413732625229478,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3131515681743622,
+      "eval_mean_accuracy": 0.4034750951077538,
+      "eval_mean_iou": 0.3444235567207426,
+      "eval_overall_accuracy": 0.6160553849186144,
+      "eval_runtime": 4.2191,
+      "eval_samples_per_second": 19.198,
+      "eval_steps_per_second": 9.718,
+      "step": 860
+    },
+    {
+      "epoch": 5.38,
+      "learning_rate": 7.8475e-05,
+      "loss": 0.2128,
+      "step": 861
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 7.845e-05,
+      "loss": 0.3902,
+      "step": 862
+    },
+    {
+      "epoch": 5.39,
+      "learning_rate": 7.842500000000001e-05,
+      "loss": 0.4554,
+      "step": 863
+    },
+    {
+      "epoch": 5.4,
+      "learning_rate": 7.840000000000001e-05,
+      "loss": 0.5527,
+      "step": 864
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 7.8375e-05,
+      "loss": 0.4556,
+      "step": 865
+    },
+    {
+      "epoch": 5.41,
+      "learning_rate": 7.835000000000001e-05,
+      "loss": 0.5762,
+      "step": 866
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 7.8325e-05,
+      "loss": 0.273,
+      "step": 867
+    },
+    {
+      "epoch": 5.42,
+      "learning_rate": 7.83e-05,
+      "loss": 0.2857,
+      "step": 868
+    },
+    {
+      "epoch": 5.43,
+      "learning_rate": 7.827499999999999e-05,
+      "loss": 0.376,
+      "step": 869
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 7.825e-05,
+      "loss": 0.1957,
+      "step": 870
+    },
+    {
+      "epoch": 5.44,
+      "learning_rate": 7.8225e-05,
+      "loss": 0.1664,
+      "step": 871
+    },
+    {
+      "epoch": 5.45,
+      "learning_rate": 7.82e-05,
+      "loss": 0.7699,
+      "step": 872
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 7.8175e-05,
+      "loss": 0.2756,
+      "step": 873
+    },
+    {
+      "epoch": 5.46,
+      "learning_rate": 7.815e-05,
+      "loss": 0.1803,
+      "step": 874
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 7.8125e-05,
+      "loss": 0.2256,
+      "step": 875
+    },
+    {
+      "epoch": 5.47,
+      "learning_rate": 7.81e-05,
+      "loss": 0.1776,
+      "step": 876
+    },
+    {
+      "epoch": 5.48,
+      "learning_rate": 7.807500000000001e-05,
+      "loss": 0.4746,
+      "step": 877
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 7.805e-05,
+      "loss": 0.3886,
+      "step": 878
+    },
+    {
+      "epoch": 5.49,
+      "learning_rate": 7.8025e-05,
+      "loss": 0.157,
+      "step": 879
+    },
+    {
+      "epoch": 5.5,
+      "learning_rate": 7.800000000000001e-05,
+      "loss": 0.1716,
+      "step": 880
+    },
+    {
+      "epoch": 5.5,
+      "eval_accuracy_AK": 0.3577457706260551,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.1934341916307651,
+      "eval_accuracy_LA": 0.6887975746128824,
+      "eval_accuracy_LV": 0.5684920258875978,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5785479768786127,
+      "eval_accuracy_RV": 0.3184697838268319,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.25475744521445964,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.34409691832880374,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.17979481289481444,
+      "eval_iou_LA": 0.6271087553971685,
+      "eval_iou_LV": 0.5591856135835227,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5486867322459971,
+      "eval_iou_RV": 0.3164716702671486,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.229489526476965,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.31855207681655884,
+      "eval_mean_accuracy": 0.32891608540857825,
+      "eval_mean_iou": 0.280483402919442,
+      "eval_overall_accuracy": 0.49546890483933664,
+      "eval_runtime": 4.4352,
+      "eval_samples_per_second": 18.263,
+      "eval_steps_per_second": 9.244,
+      "step": 880
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 7.797500000000001e-05,
+      "loss": 0.2101,
+      "step": 881
+    },
+    {
+      "epoch": 5.51,
+      "learning_rate": 7.795e-05,
+      "loss": 0.1803,
+      "step": 882
+    },
+    {
+      "epoch": 5.52,
+      "learning_rate": 7.792500000000001e-05,
+      "loss": 0.2757,
+      "step": 883
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 7.790000000000001e-05,
+      "loss": 0.3196,
+      "step": 884
+    },
+    {
+      "epoch": 5.53,
+      "learning_rate": 7.787500000000001e-05,
+      "loss": 0.3445,
+      "step": 885
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 7.785e-05,
+      "loss": 0.4656,
+      "step": 886
+    },
+    {
+      "epoch": 5.54,
+      "learning_rate": 7.782500000000001e-05,
+      "loss": 0.4186,
+      "step": 887
+    },
+    {
+      "epoch": 5.55,
+      "learning_rate": 7.780000000000001e-05,
+      "loss": 0.1456,
+      "step": 888
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 7.7775e-05,
+      "loss": 0.2785,
+      "step": 889
+    },
+    {
+      "epoch": 5.56,
+      "learning_rate": 7.775e-05,
+      "loss": 0.283,
+      "step": 890
+    },
+    {
+      "epoch": 5.57,
+      "learning_rate": 7.7725e-05,
+      "loss": 0.4395,
+      "step": 891
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 7.77e-05,
+      "loss": 0.3472,
+      "step": 892
+    },
+    {
+      "epoch": 5.58,
+      "learning_rate": 7.7675e-05,
+      "loss": 0.3324,
+      "step": 893
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 7.765e-05,
+      "loss": 0.1657,
+      "step": 894
+    },
+    {
+      "epoch": 5.59,
+      "learning_rate": 7.7625e-05,
+      "loss": 0.1965,
+      "step": 895
+    },
+    {
+      "epoch": 5.6,
+      "learning_rate": 7.76e-05,
+      "loss": 0.2531,
+      "step": 896
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 7.757500000000002e-05,
+      "loss": 0.3145,
+      "step": 897
+    },
+    {
+      "epoch": 5.61,
+      "learning_rate": 7.755e-05,
+      "loss": 0.2764,
+      "step": 898
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 7.7525e-05,
+      "loss": 0.2274,
+      "step": 899
+    },
+    {
+      "epoch": 5.62,
+      "learning_rate": 7.75e-05,
+      "loss": 0.4388,
+      "step": 900
+    },
+    {
+      "epoch": 5.62,
+      "eval_accuracy_AK": 0.6031542209930199,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.260720368700531,
+      "eval_accuracy_LA": 0.7979343736415284,
+      "eval_accuracy_LV": 0.8787790885497991,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.7475260115606936,
+      "eval_accuracy_RV": 0.6109385201500943,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.5234126240869077,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.5648180330074446,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.2445015150684717,
+      "eval_iou_LA": 0.7285842114538048,
+      "eval_iou_LV": 0.8319651997275415,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.6842260090749018,
+      "eval_iou_RV": 0.5955591147604619,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.44814176322014193,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3170972168445587,
+      "eval_mean_accuracy": 0.4913850230758416,
+      "eval_mean_iou": 0.4097795846312768,
+      "eval_overall_accuracy": 0.7347655886426971,
+      "eval_runtime": 4.4109,
+      "eval_samples_per_second": 18.364,
+      "eval_steps_per_second": 9.295,
+      "step": 900
+    },
+    {
+      "epoch": 5.63,
+      "learning_rate": 7.747500000000001e-05,
+      "loss": 0.3296,
+      "step": 901
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 7.745e-05,
+      "loss": 0.3717,
+      "step": 902
+    },
+    {
+      "epoch": 5.64,
+      "learning_rate": 7.7425e-05,
+      "loss": 0.2753,
+      "step": 903
+    },
+    {
+      "epoch": 5.65,
+      "learning_rate": 7.740000000000001e-05,
+      "loss": 0.3555,
+      "step": 904
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 7.737500000000001e-05,
+      "loss": 0.4339,
+      "step": 905
+    },
+    {
+      "epoch": 5.66,
+      "learning_rate": 7.735e-05,
+      "loss": 0.3049,
+      "step": 906
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 7.732500000000001e-05,
+      "loss": 0.2729,
+      "step": 907
+    },
+    {
+      "epoch": 5.67,
+      "learning_rate": 7.730000000000001e-05,
+      "loss": 0.2276,
+      "step": 908
+    },
+    {
+      "epoch": 5.68,
+      "learning_rate": 7.727500000000001e-05,
+      "loss": 0.2524,
+      "step": 909
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 7.725e-05,
+      "loss": 0.5198,
+      "step": 910
+    },
+    {
+      "epoch": 5.69,
+      "learning_rate": 7.722500000000001e-05,
+      "loss": 0.3153,
+      "step": 911
+    },
+    {
+      "epoch": 5.7,
+      "learning_rate": 7.72e-05,
+      "loss": 0.249,
+      "step": 912
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 7.7175e-05,
+      "loss": 0.7954,
+      "step": 913
+    },
+    {
+      "epoch": 5.71,
+      "learning_rate": 7.715e-05,
+      "loss": 0.2201,
+      "step": 914
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 7.7125e-05,
+      "loss": 0.5551,
+      "step": 915
+    },
+    {
+      "epoch": 5.72,
+      "learning_rate": 7.71e-05,
+      "loss": 0.168,
+      "step": 916
+    },
+    {
+      "epoch": 5.73,
+      "learning_rate": 7.7075e-05,
+      "loss": 0.2804,
+      "step": 917
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 7.705e-05,
+      "loss": 0.3459,
+      "step": 918
+    },
+    {
+      "epoch": 5.74,
+      "learning_rate": 7.7025e-05,
+      "loss": 0.4026,
+      "step": 919
+    },
+    {
+      "epoch": 5.75,
+      "learning_rate": 7.7e-05,
+      "loss": 0.2632,
+      "step": 920
+    },
+    {
+      "epoch": 5.75,
+      "eval_accuracy_AK": 0.38458867617303016,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.11798250008349197,
+      "eval_accuracy_LA": 0.6042406799478347,
+      "eval_accuracy_LV": 0.5144628377286972,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.461278612716763,
+      "eval_accuracy_RV": 0.34446006930742795,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.3895766997565087,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.3677147755013265,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.11406086044071354,
+      "eval_iou_LA": 0.5718156364648702,
+      "eval_iou_LV": 0.5044506097150527,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.4568349030356738,
+      "eval_iou_RV": 0.3404426559356137,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.3415913943176219,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3162875175476074,
+      "eval_mean_accuracy": 0.3129544528570838,
+      "eval_mean_iou": 0.26969108354108723,
+      "eval_overall_accuracy": 0.45514401940333643,
+      "eval_runtime": 4.4508,
+      "eval_samples_per_second": 18.199,
+      "eval_steps_per_second": 9.212,
+      "step": 920
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 7.697500000000001e-05,
+      "loss": 0.2812,
+      "step": 921
+    },
+    {
+      "epoch": 5.76,
+      "learning_rate": 7.695e-05,
+      "loss": 0.3751,
+      "step": 922
+    },
+    {
+      "epoch": 5.77,
+      "learning_rate": 7.6925e-05,
+      "loss": 0.2477,
+      "step": 923
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 7.69e-05,
+      "loss": 0.2371,
+      "step": 924
+    },
+    {
+      "epoch": 5.78,
+      "learning_rate": 7.687500000000001e-05,
+      "loss": 0.3081,
+      "step": 925
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 7.685e-05,
+      "loss": 0.2325,
+      "step": 926
+    },
+    {
+      "epoch": 5.79,
+      "learning_rate": 7.6825e-05,
+      "loss": 0.2423,
+      "step": 927
+    },
+    {
+      "epoch": 5.8,
+      "learning_rate": 7.680000000000001e-05,
+      "loss": 0.2129,
+      "step": 928
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 7.677500000000001e-05,
+      "loss": 0.258,
+      "step": 929
+    },
+    {
+      "epoch": 5.81,
+      "learning_rate": 7.675e-05,
+      "loss": 0.3016,
+      "step": 930
+    },
+    {
+      "epoch": 5.82,
+      "learning_rate": 7.672500000000001e-05,
+      "loss": 0.1213,
+      "step": 931
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 7.670000000000001e-05,
+      "loss": 0.1804,
+      "step": 932
+    },
+    {
+      "epoch": 5.83,
+      "learning_rate": 7.667500000000001e-05,
+      "loss": 0.256,
+      "step": 933
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 7.664999999999999e-05,
+      "loss": 0.3543,
+      "step": 934
+    },
+    {
+      "epoch": 5.84,
+      "learning_rate": 7.6625e-05,
+      "loss": 0.2376,
+      "step": 935
+    },
+    {
+      "epoch": 5.85,
+      "learning_rate": 7.66e-05,
+      "loss": 0.429,
+      "step": 936
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 7.6575e-05,
+      "loss": 0.2387,
+      "step": 937
+    },
+    {
+      "epoch": 5.86,
+      "learning_rate": 7.655e-05,
+      "loss": 0.3588,
+      "step": 938
+    },
+    {
+      "epoch": 5.87,
+      "learning_rate": 7.6525e-05,
+      "loss": 0.1281,
+      "step": 939
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 7.65e-05,
+      "loss": 0.3024,
+      "step": 940
+    },
+    {
+      "epoch": 5.88,
+      "eval_accuracy_AK": 0.505393723884439,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.3253097552015496,
+      "eval_accuracy_LA": 0.7012936397295798,
+      "eval_accuracy_LV": 0.7778286292710768,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.7137063583815029,
+      "eval_accuracy_RV": 0.6624473557709555,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.49496160329649747,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.49189822232839436,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.29189484803308285,
+      "eval_iou_LA": 0.6506215526038108,
+      "eval_iou_LV": 0.7479239148442677,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.6433552733047934,
+      "eval_iou_RV": 0.6322575012368674,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.4251321197886083,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.306339830160141,
+      "eval_mean_accuracy": 0.46454900728173343,
+      "eval_mean_iou": 0.3883083432139825,
+      "eval_overall_accuracy": 0.6757885342925501,
+      "eval_runtime": 4.3014,
+      "eval_samples_per_second": 18.831,
+      "eval_steps_per_second": 9.532,
+      "step": 940
+    },
+    {
+      "epoch": 5.88,
+      "learning_rate": 7.6475e-05,
+      "loss": 0.2299,
+      "step": 941
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 7.645e-05,
+      "loss": 0.1693,
+      "step": 942
+    },
+    {
+      "epoch": 5.89,
+      "learning_rate": 7.6425e-05,
+      "loss": 0.2296,
+      "step": 943
+    },
+    {
+      "epoch": 5.9,
+      "learning_rate": 7.64e-05,
+      "loss": 0.3425,
+      "step": 944
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 7.637500000000001e-05,
+      "loss": 0.3853,
+      "step": 945
+    },
+    {
+      "epoch": 5.91,
+      "learning_rate": 7.635e-05,
+      "loss": 0.2038,
+      "step": 946
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 7.6325e-05,
+      "loss": 0.2548,
+      "step": 947
+    },
+    {
+      "epoch": 5.92,
+      "learning_rate": 7.630000000000001e-05,
+      "loss": 0.2247,
+      "step": 948
+    },
+    {
+      "epoch": 5.93,
+      "learning_rate": 7.627500000000001e-05,
+      "loss": 0.435,
+      "step": 949
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 7.625e-05,
+      "loss": 0.2827,
+      "step": 950
+    },
+    {
+      "epoch": 5.94,
+      "learning_rate": 7.6225e-05,
+      "loss": 0.2335,
+      "step": 951
+    },
+    {
+      "epoch": 5.95,
+      "learning_rate": 7.620000000000001e-05,
+      "loss": 0.1822,
+      "step": 952
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 7.617500000000001e-05,
+      "loss": 0.1951,
+      "step": 953
+    },
+    {
+      "epoch": 5.96,
+      "learning_rate": 7.615e-05,
+      "loss": 0.2004,
+      "step": 954
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 7.612500000000001e-05,
+      "loss": 0.1502,
+      "step": 955
+    },
+    {
+      "epoch": 5.97,
+      "learning_rate": 7.61e-05,
+      "loss": 0.2307,
+      "step": 956
+    },
+    {
+      "epoch": 5.98,
+      "learning_rate": 7.6075e-05,
+      "loss": 0.5534,
+      "step": 957
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 7.605e-05,
+      "loss": 0.1755,
+      "step": 958
+    },
+    {
+      "epoch": 5.99,
+      "learning_rate": 7.6025e-05,
+      "loss": 0.3305,
+      "step": 959
+    },
+    {
+      "epoch": 6.0,
+      "learning_rate": 7.6e-05,
+      "loss": 0.2551,
+      "step": 960
+    },
+    {
+      "epoch": 6.0,
+      "eval_accuracy_AK": 0.24888653424805143,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.16664161907624486,
+      "eval_accuracy_LA": 0.6708844868162672,
+      "eval_accuracy_LV": 0.5010977192207876,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.4074219653179191,
+      "eval_accuracy_RV": 0.10843455613829917,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.2592433039895111,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.2469843910118367,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.15623605664232204,
+      "eval_iou_LA": 0.6180913284260244,
+      "eval_iou_LV": 0.49030402688939273,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.40419587477520463,
+      "eval_iou_RV": 0.1081457232046225,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.23686755029221251,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3489318788051605,
+      "eval_mean_accuracy": 0.2625122427563423,
+      "eval_mean_iou": 0.22608249512416156,
+      "eval_overall_accuracy": 0.40242048957289667,
+      "eval_runtime": 4.2851,
+      "eval_samples_per_second": 18.903,
+      "eval_steps_per_second": 9.568,
+      "step": 960
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 7.5975e-05,
+      "loss": 0.4217,
+      "step": 961
+    },
+    {
+      "epoch": 6.01,
+      "learning_rate": 7.595e-05,
+      "loss": 0.2589,
+      "step": 962
+    },
+    {
+      "epoch": 6.02,
+      "learning_rate": 7.5925e-05,
+      "loss": 0.1692,
+      "step": 963
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 7.59e-05,
+      "loss": 0.2896,
+      "step": 964
+    },
+    {
+      "epoch": 6.03,
+      "learning_rate": 7.5875e-05,
+      "loss": 0.2676,
+      "step": 965
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 7.585e-05,
+      "loss": 0.2148,
+      "step": 966
+    },
+    {
+      "epoch": 6.04,
+      "learning_rate": 7.5825e-05,
+      "loss": 0.199,
+      "step": 967
+    },
+    {
+      "epoch": 6.05,
+      "learning_rate": 7.58e-05,
+      "loss": 0.2372,
+      "step": 968
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 7.577500000000001e-05,
+      "loss": 0.3216,
+      "step": 969
+    },
+    {
+      "epoch": 6.06,
+      "learning_rate": 7.575e-05,
+      "loss": 0.2611,
+      "step": 970
+    },
+    {
+      "epoch": 6.07,
+      "learning_rate": 7.5725e-05,
+      "loss": 0.3112,
+      "step": 971
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 7.570000000000001e-05,
+      "loss": 0.3522,
+      "step": 972
+    },
+    {
+      "epoch": 6.08,
+      "learning_rate": 7.567500000000001e-05,
+      "loss": 0.1917,
+      "step": 973
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 7.565e-05,
+      "loss": 0.2933,
+      "step": 974
+    },
+    {
+      "epoch": 6.09,
+      "learning_rate": 7.5625e-05,
+      "loss": 0.1557,
+      "step": 975
+    },
+    {
+      "epoch": 6.1,
+      "learning_rate": 7.560000000000001e-05,
+      "loss": 0.7174,
+      "step": 976
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 7.557500000000001e-05,
+      "loss": 0.1847,
+      "step": 977
+    },
+    {
+      "epoch": 6.11,
+      "learning_rate": 7.555e-05,
+      "loss": 0.242,
+      "step": 978
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 7.5525e-05,
+      "loss": 0.2347,
+      "step": 979
+    },
+    {
+      "epoch": 6.12,
+      "learning_rate": 7.55e-05,
+      "loss": 0.3281,
+      "step": 980
+    },
+    {
+      "epoch": 6.12,
+      "eval_accuracy_AK": 0.5579421237263987,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.2545085662759243,
+      "eval_accuracy_LA": 0.6806130173434666,
+      "eval_accuracy_LV": 0.6802817053078191,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.641764161849711,
+      "eval_accuracy_RV": 0.5665433817146055,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.4660048698258101,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.5329940239613417,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.2371627299038372,
+      "eval_iou_LA": 0.6359278627564606,
+      "eval_iou_LV": 0.6658651613298489,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.6049351504148587,
+      "eval_iou_RV": 0.5508445165284572,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.40712801296002354,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.2938700318336487,
+      "eval_mean_accuracy": 0.42751753622708166,
+      "eval_mean_iou": 0.36348574578548276,
+      "eval_overall_accuracy": 0.6115746595929898,
+      "eval_runtime": 4.3374,
+      "eval_samples_per_second": 18.675,
+      "eval_steps_per_second": 9.453,
+      "step": 980
+    },
+    {
+      "epoch": 6.13,
+      "learning_rate": 7.5475e-05,
+      "loss": 0.5447,
+      "step": 981
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 7.545e-05,
+      "loss": 0.1878,
+      "step": 982
+    },
+    {
+      "epoch": 6.14,
+      "learning_rate": 7.5425e-05,
+      "loss": 0.2774,
+      "step": 983
+    },
+    {
+      "epoch": 6.15,
+      "learning_rate": 7.54e-05,
+      "loss": 0.17,
+      "step": 984
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 7.5375e-05,
+      "loss": 0.3285,
+      "step": 985
+    },
+    {
+      "epoch": 6.16,
+      "learning_rate": 7.535e-05,
+      "loss": 0.1058,
+      "step": 986
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 7.5325e-05,
+      "loss": 0.2625,
+      "step": 987
+    },
+    {
+      "epoch": 6.17,
+      "learning_rate": 7.53e-05,
+      "loss": 0.2679,
+      "step": 988
+    },
+    {
+      "epoch": 6.18,
+      "learning_rate": 7.5275e-05,
+      "loss": 0.187,
+      "step": 989
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 7.525e-05,
+      "loss": 0.1415,
+      "step": 990
+    },
+    {
+      "epoch": 6.19,
+      "learning_rate": 7.5225e-05,
+      "loss": 0.1139,
+      "step": 991
+    },
+    {
+      "epoch": 6.2,
+      "learning_rate": 7.52e-05,
+      "loss": 0.2784,
+      "step": 992
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 7.517500000000001e-05,
+      "loss": 0.426,
+      "step": 993
+    },
+    {
+      "epoch": 6.21,
+      "learning_rate": 7.515e-05,
+      "loss": 0.2224,
+      "step": 994
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 7.5125e-05,
+      "loss": 0.4038,
+      "step": 995
+    },
+    {
+      "epoch": 6.22,
+      "learning_rate": 7.510000000000001e-05,
+      "loss": 0.1417,
+      "step": 996
+    },
+    {
+      "epoch": 6.23,
+      "learning_rate": 7.507500000000001e-05,
+      "loss": 0.2275,
+      "step": 997
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 7.505e-05,
+      "loss": 0.2616,
+      "step": 998
+    },
+    {
+      "epoch": 6.24,
+      "learning_rate": 7.502500000000001e-05,
+      "loss": 0.2887,
+      "step": 999
+    },
+    {
+      "epoch": 6.25,
+      "learning_rate": 7.500000000000001e-05,
+      "loss": 0.1372,
+      "step": 1000
+    },
+    {
+      "epoch": 6.25,
+      "eval_accuracy_AK": 0.5647935299258887,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.2979160404769061,
+      "eval_accuracy_LA": 0.7023448156975611,
+      "eval_accuracy_LV": 0.7530104831849068,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.6673317919075145,
+      "eval_accuracy_RV": 0.5417171883856248,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.444980333395767,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.5389149899324547,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.2737126331857966,
+      "eval_iou_LA": 0.6582160894027408,
+      "eval_iou_LV": 0.7286595174611941,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.6358537246288369,
+      "eval_iou_RV": 0.5299874529485571,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.389863467786931,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.29983189702033997,
+      "eval_mean_accuracy": 0.44134379810824104,
+      "eval_mean_iou": 0.3755207875346511,
+      "eval_overall_accuracy": 0.6450046263598487,
+      "eval_runtime": 4.3746,
+      "eval_samples_per_second": 18.516,
+      "eval_steps_per_second": 9.372,
+      "step": 1000
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 7.4975e-05,
+      "loss": 0.1882,
+      "step": 1001
+    },
+    {
+      "epoch": 6.26,
+      "learning_rate": 7.495e-05,
+      "loss": 0.2553,
+      "step": 1002
+    },
+    {
+      "epoch": 6.27,
+      "learning_rate": 7.4925e-05,
+      "loss": 0.2875,
+      "step": 1003
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 7.49e-05,
+      "loss": 0.3197,
+      "step": 1004
+    },
+    {
+      "epoch": 6.28,
+      "learning_rate": 7.4875e-05,
+      "loss": 0.3018,
+      "step": 1005
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 7.485e-05,
+      "loss": 0.1417,
+      "step": 1006
+    },
+    {
+      "epoch": 6.29,
+      "learning_rate": 7.4825e-05,
+      "loss": 0.2227,
+      "step": 1007
+    },
+    {
+      "epoch": 6.3,
+      "learning_rate": 7.48e-05,
+      "loss": 0.3268,
+      "step": 1008
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 7.4775e-05,
+      "loss": 0.2264,
+      "step": 1009
+    },
+    {
+      "epoch": 6.31,
+      "learning_rate": 7.475000000000001e-05,
+      "loss": 0.1951,
+      "step": 1010
+    },
+    {
+      "epoch": 6.32,
+      "learning_rate": 7.4725e-05,
+      "loss": 0.3946,
+      "step": 1011
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 7.47e-05,
+      "loss": 0.128,
+      "step": 1012
+    },
+    {
+      "epoch": 6.33,
+      "learning_rate": 7.467500000000001e-05,
+      "loss": 0.4181,
+      "step": 1013
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 7.465000000000001e-05,
+      "loss": 0.1869,
+      "step": 1014
+    },
+    {
+      "epoch": 6.34,
+      "learning_rate": 7.4625e-05,
+      "loss": 0.2964,
+      "step": 1015
+    },
+    {
+      "epoch": 6.35,
+      "learning_rate": 7.46e-05,
+      "loss": 0.2407,
+      "step": 1016
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 7.457500000000001e-05,
+      "loss": 0.1833,
+      "step": 1017
+    },
+    {
+      "epoch": 6.36,
+      "learning_rate": 7.455000000000001e-05,
+      "loss": 0.1317,
+      "step": 1018
+    },
+    {
+      "epoch": 6.37,
+      "learning_rate": 7.4525e-05,
+      "loss": 0.3148,
+      "step": 1019
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 7.450000000000001e-05,
+      "loss": 0.3485,
+      "step": 1020
+    },
+    {
+      "epoch": 6.38,
+      "eval_accuracy_AK": 0.28109323180441315,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.1936930167317904,
+      "eval_accuracy_LA": 0.6262872689661375,
+      "eval_accuracy_LV": 0.5305019221864467,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.5157687861271676,
+      "eval_accuracy_RV": 0.295665057146342,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.20428919273272148,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.2769149566689175,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.18332029490553067,
+      "eval_iou_LA": 0.5911863807207606,
+      "eval_iou_LV": 0.5205640306622823,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.5070176036730194,
+      "eval_iou_RV": 0.29312454209848704,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.18464064735109148,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.3397650420665741,
+      "eval_mean_accuracy": 0.2941442750772243,
+      "eval_mean_iou": 0.2556768456080089,
+      "eval_overall_accuracy": 0.451493848857645,
+      "eval_runtime": 4.2855,
+      "eval_samples_per_second": 18.901,
+      "eval_steps_per_second": 9.567,
+      "step": 1020
+    },
+    {
+      "epoch": 6.38,
+      "learning_rate": 7.447500000000001e-05,
+      "loss": 0.2099,
+      "step": 1021
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 7.445000000000001e-05,
+      "loss": 0.49,
+      "step": 1022
+    },
+    {
+      "epoch": 6.39,
+      "learning_rate": 7.4425e-05,
+      "loss": 0.3984,
+      "step": 1023
+    },
+    {
+      "epoch": 6.4,
+      "learning_rate": 7.44e-05,
+      "loss": 0.2814,
+      "step": 1024
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 7.4375e-05,
+      "loss": 0.2811,
+      "step": 1025
+    },
+    {
+      "epoch": 6.41,
+      "learning_rate": 7.435e-05,
+      "loss": 0.2892,
+      "step": 1026
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 7.4325e-05,
+      "loss": 0.2035,
+      "step": 1027
+    },
+    {
+      "epoch": 6.42,
+      "learning_rate": 7.43e-05,
+      "loss": 0.2016,
+      "step": 1028
+    },
+    {
+      "epoch": 6.43,
+      "learning_rate": 7.4275e-05,
+      "loss": 0.1919,
+      "step": 1029
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 7.425e-05,
+      "loss": 0.2524,
+      "step": 1030
+    },
+    {
+      "epoch": 6.44,
+      "learning_rate": 7.4225e-05,
+      "loss": 0.2805,
+      "step": 1031
+    },
+    {
+      "epoch": 6.45,
+      "learning_rate": 7.42e-05,
+      "loss": 0.2513,
+      "step": 1032
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 7.4175e-05,
+      "loss": 0.3296,
+      "step": 1033
+    },
+    {
+      "epoch": 6.46,
+      "learning_rate": 7.415000000000001e-05,
+      "loss": 0.274,
+      "step": 1034
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 7.4125e-05,
+      "loss": 0.1785,
+      "step": 1035
+    },
+    {
+      "epoch": 6.47,
+      "learning_rate": 7.41e-05,
+      "loss": 0.2358,
+      "step": 1036
+    },
+    {
+      "epoch": 6.48,
+      "learning_rate": 7.407500000000001e-05,
+      "loss": 0.2124,
+      "step": 1037
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 7.405000000000001e-05,
+      "loss": 0.1447,
+      "step": 1038
+    },
+    {
+      "epoch": 6.49,
+      "learning_rate": 7.4025e-05,
+      "loss": 0.2454,
+      "step": 1039
+    },
+    {
+      "epoch": 6.5,
+      "learning_rate": 7.4e-05,
+      "loss": 0.3755,
+      "step": 1040
+    },
+    {
+      "epoch": 6.5,
+      "eval_accuracy_AK": 0.4597715599296002,
+      "eval_accuracy_AS": NaN,
+      "eval_accuracy_ASD": 0.3010219416892095,
+      "eval_accuracy_LA": 0.6986122978219484,
+      "eval_accuracy_LV": 0.6346486221638328,
+      "eval_accuracy_MK": 0.0,
+      "eval_accuracy_RA": 0.7049988439306358,
+      "eval_accuracy_RV": 0.4563940765825555,
+      "eval_accuracy_TK": NaN,
+      "eval_accuracy_VS": 0.0,
+      "eval_accuracy_VSD": 0.48811575201348567,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_AK": 0.44423300499164203,
+      "eval_iou_AS": NaN,
+      "eval_iou_ASD": 0.27584887759942467,
+      "eval_iou_LA": 0.6553671621222561,
+      "eval_iou_LV": 0.6187952547257844,
+      "eval_iou_MK": 0.0,
+      "eval_iou_RA": 0.6638913081344714,
+      "eval_iou_RV": 0.4508897402546773,
+      "eval_iou_TK": NaN,
+      "eval_iou_VS": 0.0,
+      "eval_iou_VSD": 0.41664135317910755,
+      "eval_iou_unlabeled": 0.0,
+      "eval_loss": 0.30338749289512634,
+      "eval_mean_accuracy": 0.41595145490347424,
+      "eval_mean_iou": 0.35256667010073633,
+      "eval_overall_accuracy": 0.5795123433470756,
+      "eval_runtime": 4.4722,
+      "eval_samples_per_second": 18.112,
+      "eval_steps_per_second": 9.168,
+      "step": 1040
+    }
+  ],
+  "logging_steps": 1,
+  "max_steps": 4000,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 25,
+  "save_steps": 20,
+  "total_flos": 3.648337119019008e+16,
+  "train_batch_size": 2,
+  "trial_name": null,
+  "trial_params": null
+}