Training in progress, step 2960

{checkpoint-2640 → checkpoint-2960}/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:052b768c40df274c8ba852e722e9455f3201e4b95a53864a313ce0ba1e60a976
+oid sha256:790f2c3156cc6430ca0d06e42821e7e6799403d65966f576b991999558b734f8
 size 677523903

{checkpoint-2640 → checkpoint-2960}/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8ae4a55c76dae309ad126417181d3a225d0005c38dd46f51824d3b661757421c
+oid sha256:b5d6df664b36ccc3d54e1cf8e501a508daed6e77a51faf1ec4337b95da2566c3
 size 338830281

{checkpoint-2640 → checkpoint-2960}/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:58bd8d14a2a607463a2f502a360d3dd8bd8fc0a077b71dec16d9be9e895eb926
+oid sha256:d41d98419757fe02252c7edeabecb3ba3b0841f9d2231daf03262ff889cd44b5
 size 14575

{checkpoint-2640 → checkpoint-2960}/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e2baa48e7e9ebfa39f62c8f313327b90eebacc825ed9876fc29feee648f59133
+oid sha256:d81d0fa64deabbd69059db40b8a28460fa23953407913e4394f6a4e27528435d
 size 627

{checkpoint-2640 → checkpoint-2960}/trainer_state.json

@@ -1,8 +1,8 @@
 {
   "best_metric": 0.07582816481590271,
   "best_model_checkpoint": "/robodata/smodak/Projects/lifelong_concept_learner/scripts/terrainseg/training/models/safety-utcustom-terrain-final/checkpoint-1360",
-  "epoch": 155.2941176470588,
-  "global_step": 2640,
+  "epoch": 174.11764705882354,
+  "global_step": 2960,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -20994,11 +20994,2555 @@
       "eval_samples_per_second": 21.686,
       "eval_steps_per_second": 1.47,
       "step": 2640
+    },
+    {
+      "epoch": 155.35,
+      "learning_rate": 1.174922600619195e-05,
+      "loss": 0.0206,
+      "step": 2641
+    },
+    {
+      "epoch": 155.41,
+      "learning_rate": 1.173374613003096e-05,
+      "loss": 0.0219,
+      "step": 2642
+    },
+    {
+      "epoch": 155.47,
+      "learning_rate": 1.171826625386997e-05,
+      "loss": 0.0212,
+      "step": 2643
+    },
+    {
+      "epoch": 155.53,
+      "learning_rate": 1.1702786377708978e-05,
+      "loss": 0.0237,
+      "step": 2644
+    },
+    {
+      "epoch": 155.59,
+      "learning_rate": 1.1687306501547988e-05,
+      "loss": 0.0248,
+      "step": 2645
+    },
+    {
+      "epoch": 155.65,
+      "learning_rate": 1.1671826625386997e-05,
+      "loss": 0.0241,
+      "step": 2646
+    },
+    {
+      "epoch": 155.71,
+      "learning_rate": 1.1656346749226007e-05,
+      "loss": 0.0239,
+      "step": 2647
+    },
+    {
+      "epoch": 155.76,
+      "learning_rate": 1.1640866873065017e-05,
+      "loss": 0.0246,
+      "step": 2648
+    },
+    {
+      "epoch": 155.82,
+      "learning_rate": 1.1625386996904025e-05,
+      "loss": 0.0204,
+      "step": 2649
+    },
+    {
+      "epoch": 155.88,
+      "learning_rate": 1.1609907120743033e-05,
+      "loss": 0.0191,
+      "step": 2650
+    },
+    {
+      "epoch": 155.94,
+      "learning_rate": 1.1594427244582045e-05,
+      "loss": 0.0248,
+      "step": 2651
+    },
+    {
+      "epoch": 156.0,
+      "learning_rate": 1.1578947368421053e-05,
+      "loss": 0.0219,
+      "step": 2652
+    },
+    {
+      "epoch": 156.06,
+      "learning_rate": 1.1563467492260062e-05,
+      "loss": 0.0237,
+      "step": 2653
+    },
+    {
+      "epoch": 156.12,
+      "learning_rate": 1.1547987616099072e-05,
+      "loss": 0.0216,
+      "step": 2654
+    },
+    {
+      "epoch": 156.18,
+      "learning_rate": 1.1532507739938082e-05,
+      "loss": 0.0244,
+      "step": 2655
+    },
+    {
+      "epoch": 156.24,
+      "learning_rate": 1.151702786377709e-05,
+      "loss": 0.0252,
+      "step": 2656
+    },
+    {
+      "epoch": 156.29,
+      "learning_rate": 1.15015479876161e-05,
+      "loss": 0.0222,
+      "step": 2657
+    },
+    {
+      "epoch": 156.35,
+      "learning_rate": 1.1486068111455108e-05,
+      "loss": 0.0187,
+      "step": 2658
+    },
+    {
+      "epoch": 156.41,
+      "learning_rate": 1.1470588235294118e-05,
+      "loss": 0.0231,
+      "step": 2659
+    },
+    {
+      "epoch": 156.47,
+      "learning_rate": 1.1455108359133128e-05,
+      "loss": 0.0203,
+      "step": 2660
+    },
+    {
+      "epoch": 156.47,
+      "eval_accuracy_NAT": 0.9940302430951301,
+      "eval_accuracy_REST": 0.34959349593495936,
+      "eval_accuracy_concrete": 0.9558009444909029,
+      "eval_accuracy_concrete tiles": 0.9386518972425171,
+      "eval_accuracy_dark bricks": 0.8629002928622016,
+      "eval_accuracy_grass": 0.8938739321819634,
+      "eval_accuracy_red bricks": 0.9250878734622144,
+      "eval_accuracy_road": 0.9807946874761183,
+      "eval_accuracy_rough concrete": 0.6244823291760766,
+      "eval_accuracy_rough red sidewalk": 0.77346047120222,
+      "eval_accuracy_speedway bricks": 0.9892427236640886,
+      "eval_accuracy_steel": 0.9230934543473589,
+      "eval_accuracy_tiles": 0.24987124723560242,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834143702762304,
+      "eval_iou_REST": 0.34290271132376393,
+      "eval_iou_concrete": 0.8925630217609332,
+      "eval_iou_concrete tiles": 0.8694712483080819,
+      "eval_iou_dark bricks": 0.8462802004918433,
+      "eval_iou_grass": 0.8399726135802853,
+      "eval_iou_red bricks": 0.9049588996937624,
+      "eval_iou_road": 0.9607700969688531,
+      "eval_iou_rough concrete": 0.617746052142954,
+      "eval_iou_rough red sidewalk": 0.6219779156537619,
+      "eval_iou_speedway bricks": 0.9795577537998109,
+      "eval_iou_steel": 0.8663931907787495,
+      "eval_iou_tiles": 0.24600334049152947,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08094289898872375,
+      "eval_mean_accuracy": 0.8046833532593349,
+      "eval_mean_iou": 0.7670778011746585,
+      "eval_overall_accuracy": 0.9799908783476231,
+      "eval_runtime": 5.3629,
+      "eval_samples_per_second": 22.003,
+      "eval_steps_per_second": 1.492,
+      "step": 2660
+    },
+    {
+      "epoch": 156.53,
+      "learning_rate": 1.1439628482972137e-05,
+      "loss": 0.021,
+      "step": 2661
+    },
+    {
+      "epoch": 156.59,
+      "learning_rate": 1.1424148606811145e-05,
+      "loss": 0.0237,
+      "step": 2662
+    },
+    {
+      "epoch": 156.65,
+      "learning_rate": 1.1408668730650155e-05,
+      "loss": 0.0207,
+      "step": 2663
+    },
+    {
+      "epoch": 156.71,
+      "learning_rate": 1.1393188854489165e-05,
+      "loss": 0.0208,
+      "step": 2664
+    },
+    {
+      "epoch": 156.76,
+      "learning_rate": 1.1377708978328173e-05,
+      "loss": 0.0235,
+      "step": 2665
+    },
+    {
+      "epoch": 156.82,
+      "learning_rate": 1.1362229102167183e-05,
+      "loss": 0.0214,
+      "step": 2666
+    },
+    {
+      "epoch": 156.88,
+      "learning_rate": 1.1346749226006193e-05,
+      "loss": 0.0224,
+      "step": 2667
+    },
+    {
+      "epoch": 156.94,
+      "learning_rate": 1.1331269349845202e-05,
+      "loss": 0.0215,
+      "step": 2668
+    },
+    {
+      "epoch": 157.0,
+      "learning_rate": 1.1315789473684212e-05,
+      "loss": 0.0237,
+      "step": 2669
+    },
+    {
+      "epoch": 157.06,
+      "learning_rate": 1.130030959752322e-05,
+      "loss": 0.0198,
+      "step": 2670
+    },
+    {
+      "epoch": 157.12,
+      "learning_rate": 1.128482972136223e-05,
+      "loss": 0.0216,
+      "step": 2671
+    },
+    {
+      "epoch": 157.18,
+      "learning_rate": 1.126934984520124e-05,
+      "loss": 0.0188,
+      "step": 2672
+    },
+    {
+      "epoch": 157.24,
+      "learning_rate": 1.1253869969040248e-05,
+      "loss": 0.0204,
+      "step": 2673
+    },
+    {
+      "epoch": 157.29,
+      "learning_rate": 1.1238390092879257e-05,
+      "loss": 0.022,
+      "step": 2674
+    },
+    {
+      "epoch": 157.35,
+      "learning_rate": 1.1222910216718267e-05,
+      "loss": 0.0253,
+      "step": 2675
+    },
+    {
+      "epoch": 157.41,
+      "learning_rate": 1.1207430340557277e-05,
+      "loss": 0.0212,
+      "step": 2676
+    },
+    {
+      "epoch": 157.47,
+      "learning_rate": 1.1191950464396285e-05,
+      "loss": 0.0283,
+      "step": 2677
+    },
+    {
+      "epoch": 157.53,
+      "learning_rate": 1.1176470588235295e-05,
+      "loss": 0.0193,
+      "step": 2678
+    },
+    {
+      "epoch": 157.59,
+      "learning_rate": 1.1160990712074305e-05,
+      "loss": 0.0237,
+      "step": 2679
+    },
+    {
+      "epoch": 157.65,
+      "learning_rate": 1.1145510835913313e-05,
+      "loss": 0.0225,
+      "step": 2680
+    },
+    {
+      "epoch": 157.65,
+      "eval_accuracy_NAT": 0.9938389102622057,
+      "eval_accuracy_REST": 0.3203252032520325,
+      "eval_accuracy_concrete": 0.9571459038172537,
+      "eval_accuracy_concrete tiles": 0.9291774687720952,
+      "eval_accuracy_dark bricks": 0.8643215358995463,
+      "eval_accuracy_grass": 0.8980392268183659,
+      "eval_accuracy_red bricks": 0.9264608963093146,
+      "eval_accuracy_road": 0.98097425129636,
+      "eval_accuracy_rough concrete": 0.6219736740178406,
+      "eval_accuracy_rough red sidewalk": 0.7661660863745403,
+      "eval_accuracy_speedway bricks": 0.9885450961118617,
+      "eval_accuracy_steel": 0.9237181825905731,
+      "eval_accuracy_tiles": 0.242691387197431,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835807089243077,
+      "eval_iou_REST": 0.3152,
+      "eval_iou_concrete": 0.8912357634404546,
+      "eval_iou_concrete tiles": 0.865285428966486,
+      "eval_iou_dark bricks": 0.8466687604591968,
+      "eval_iou_grass": 0.8418850387512319,
+      "eval_iou_red bricks": 0.9053292545483819,
+      "eval_iou_road": 0.9601985261193249,
+      "eval_iou_rough concrete": 0.6168570601202049,
+      "eval_iou_rough red sidewalk": 0.622196166083457,
+      "eval_iou_speedway bricks": 0.9794348540885546,
+      "eval_iou_steel": 0.8654449930815534,
+      "eval_iou_tiles": 0.23876371006199332,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08075359463691711,
+      "eval_mean_accuracy": 0.8010290632861093,
+      "eval_mean_iou": 0.7640061742034729,
+      "eval_overall_accuracy": 0.9799637552035056,
+      "eval_runtime": 5.3885,
+      "eval_samples_per_second": 21.899,
+      "eval_steps_per_second": 1.485,
+      "step": 2680
+    },
+    {
+      "epoch": 157.71,
+      "learning_rate": 1.1130030959752323e-05,
+      "loss": 0.0209,
+      "step": 2681
+    },
+    {
+      "epoch": 157.76,
+      "learning_rate": 1.1114551083591331e-05,
+      "loss": 0.0219,
+      "step": 2682
+    },
+    {
+      "epoch": 157.82,
+      "learning_rate": 1.1099071207430341e-05,
+      "loss": 0.0226,
+      "step": 2683
+    },
+    {
+      "epoch": 157.88,
+      "learning_rate": 1.1083591331269351e-05,
+      "loss": 0.019,
+      "step": 2684
+    },
+    {
+      "epoch": 157.94,
+      "learning_rate": 1.106811145510836e-05,
+      "loss": 0.0239,
+      "step": 2685
+    },
+    {
+      "epoch": 158.0,
+      "learning_rate": 1.1052631578947368e-05,
+      "loss": 0.0245,
+      "step": 2686
+    },
+    {
+      "epoch": 158.06,
+      "learning_rate": 1.1037151702786378e-05,
+      "loss": 0.0236,
+      "step": 2687
+    },
+    {
+      "epoch": 158.12,
+      "learning_rate": 1.1021671826625388e-05,
+      "loss": 0.0214,
+      "step": 2688
+    },
+    {
+      "epoch": 158.18,
+      "learning_rate": 1.1006191950464396e-05,
+      "loss": 0.0217,
+      "step": 2689
+    },
+    {
+      "epoch": 158.24,
+      "learning_rate": 1.0990712074303406e-05,
+      "loss": 0.0202,
+      "step": 2690
+    },
+    {
+      "epoch": 158.29,
+      "learning_rate": 1.0975232198142415e-05,
+      "loss": 0.0218,
+      "step": 2691
+    },
+    {
+      "epoch": 158.35,
+      "learning_rate": 1.0959752321981425e-05,
+      "loss": 0.0237,
+      "step": 2692
+    },
+    {
+      "epoch": 158.41,
+      "learning_rate": 1.0944272445820435e-05,
+      "loss": 0.0223,
+      "step": 2693
+    },
+    {
+      "epoch": 158.47,
+      "learning_rate": 1.0928792569659443e-05,
+      "loss": 0.0202,
+      "step": 2694
+    },
+    {
+      "epoch": 158.53,
+      "learning_rate": 1.0913312693498453e-05,
+      "loss": 0.0201,
+      "step": 2695
+    },
+    {
+      "epoch": 158.59,
+      "learning_rate": 1.0897832817337463e-05,
+      "loss": 0.0232,
+      "step": 2696
+    },
+    {
+      "epoch": 158.65,
+      "learning_rate": 1.0882352941176471e-05,
+      "loss": 0.0235,
+      "step": 2697
+    },
+    {
+      "epoch": 158.71,
+      "learning_rate": 1.086687306501548e-05,
+      "loss": 0.0201,
+      "step": 2698
+    },
+    {
+      "epoch": 158.76,
+      "learning_rate": 1.085139318885449e-05,
+      "loss": 0.0239,
+      "step": 2699
+    },
+    {
+      "epoch": 158.82,
+      "learning_rate": 1.08359133126935e-05,
+      "loss": 0.0269,
+      "step": 2700
+    },
+    {
+      "epoch": 158.82,
+      "eval_accuracy_NAT": 0.9942922003324307,
+      "eval_accuracy_REST": 0.2552845528455285,
+      "eval_accuracy_concrete": 0.9557722795436234,
+      "eval_accuracy_concrete tiles": 0.9247466415272213,
+      "eval_accuracy_dark bricks": 0.8602875026319315,
+      "eval_accuracy_grass": 0.8893642106378711,
+      "eval_accuracy_red bricks": 0.9212434094903339,
+      "eval_accuracy_road": 0.980666427604517,
+      "eval_accuracy_rough concrete": 0.6250424564560628,
+      "eval_accuracy_rough red sidewalk": 0.7760774257589544,
+      "eval_accuracy_speedway bricks": 0.9899199587626887,
+      "eval_accuracy_steel": 0.9176583186313954,
+      "eval_accuracy_tiles": 0.24444848374685693,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9834257991120192,
+      "eval_iou_REST": 0.2516025641025641,
+      "eval_iou_concrete": 0.8928482296861182,
+      "eval_iou_concrete tiles": 0.8697107392219883,
+      "eval_iou_dark bricks": 0.843891997446393,
+      "eval_iou_grass": 0.83900462570363,
+      "eval_iou_red bricks": 0.9014402407566638,
+      "eval_iou_road": 0.9608989917238093,
+      "eval_iou_rough concrete": 0.6184423088261305,
+      "eval_iou_rough red sidewalk": 0.6271657156053065,
+      "eval_iou_speedway bricks": 0.9795101626805487,
+      "eval_iou_steel": 0.8653123836543491,
+      "eval_iou_tiles": 0.24177503445796128,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0806136280298233,
+      "eval_mean_accuracy": 0.7949849129207244,
+      "eval_mean_iou": 0.7596175994598062,
+      "eval_overall_accuracy": 0.9800087880280058,
+      "eval_runtime": 5.1938,
+      "eval_samples_per_second": 22.719,
+      "eval_steps_per_second": 1.54,
+      "step": 2700
+    },
+    {
+      "epoch": 158.88,
+      "learning_rate": 1.0820433436532508e-05,
+      "loss": 0.0219,
+      "step": 2701
+    },
+    {
+      "epoch": 158.94,
+      "learning_rate": 1.0804953560371518e-05,
+      "loss": 0.0205,
+      "step": 2702
+    },
+    {
+      "epoch": 159.0,
+      "learning_rate": 1.0789473684210526e-05,
+      "loss": 0.019,
+      "step": 2703
+    },
+    {
+      "epoch": 159.06,
+      "learning_rate": 1.0773993808049536e-05,
+      "loss": 0.0212,
+      "step": 2704
+    },
+    {
+      "epoch": 159.12,
+      "learning_rate": 1.0758513931888545e-05,
+      "loss": 0.0181,
+      "step": 2705
+    },
+    {
+      "epoch": 159.18,
+      "learning_rate": 1.0743034055727555e-05,
+      "loss": 0.024,
+      "step": 2706
+    },
+    {
+      "epoch": 159.24,
+      "learning_rate": 1.0727554179566565e-05,
+      "loss": 0.0213,
+      "step": 2707
+    },
+    {
+      "epoch": 159.29,
+      "learning_rate": 1.0712074303405573e-05,
+      "loss": 0.0236,
+      "step": 2708
+    },
+    {
+      "epoch": 159.35,
+      "learning_rate": 1.0696594427244583e-05,
+      "loss": 0.0215,
+      "step": 2709
+    },
+    {
+      "epoch": 159.41,
+      "learning_rate": 1.0681114551083591e-05,
+      "loss": 0.0214,
+      "step": 2710
+    },
+    {
+      "epoch": 159.47,
+      "learning_rate": 1.0665634674922601e-05,
+      "loss": 0.0248,
+      "step": 2711
+    },
+    {
+      "epoch": 159.53,
+      "learning_rate": 1.0650154798761611e-05,
+      "loss": 0.0214,
+      "step": 2712
+    },
+    {
+      "epoch": 159.59,
+      "learning_rate": 1.063467492260062e-05,
+      "loss": 0.0223,
+      "step": 2713
+    },
+    {
+      "epoch": 159.65,
+      "learning_rate": 1.0619195046439628e-05,
+      "loss": 0.0233,
+      "step": 2714
+    },
+    {
+      "epoch": 159.71,
+      "learning_rate": 1.0603715170278638e-05,
+      "loss": 0.0216,
+      "step": 2715
+    },
+    {
+      "epoch": 159.76,
+      "learning_rate": 1.0588235294117648e-05,
+      "loss": 0.0201,
+      "step": 2716
+    },
+    {
+      "epoch": 159.82,
+      "learning_rate": 1.0572755417956656e-05,
+      "loss": 0.0237,
+      "step": 2717
+    },
+    {
+      "epoch": 159.88,
+      "learning_rate": 1.0557275541795666e-05,
+      "loss": 0.0226,
+      "step": 2718
+    },
+    {
+      "epoch": 159.94,
+      "learning_rate": 1.0541795665634676e-05,
+      "loss": 0.0192,
+      "step": 2719
+    },
+    {
+      "epoch": 160.0,
+      "learning_rate": 1.0526315789473684e-05,
+      "loss": 0.0204,
+      "step": 2720
+    },
+    {
+      "epoch": 160.0,
+      "eval_accuracy_NAT": 0.9939970334515457,
+      "eval_accuracy_REST": 0.34796747967479674,
+      "eval_accuracy_concrete": 0.954261827921646,
+      "eval_accuracy_concrete tiles": 0.9310157907141173,
+      "eval_accuracy_dark bricks": 0.866838619527975,
+      "eval_accuracy_grass": 0.89485043923893,
+      "eval_accuracy_red bricks": 0.9281085237258347,
+      "eval_accuracy_road": 0.9815920363445723,
+      "eval_accuracy_rough concrete": 0.6257038833505145,
+      "eval_accuracy_rough red sidewalk": 0.7981485798509539,
+      "eval_accuracy_speedway bricks": 0.9892302140076621,
+      "eval_accuracy_steel": 0.9247552314743087,
+      "eval_accuracy_tiles": 0.2526886606683026,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983518218056789,
+      "eval_iou_REST": 0.33914421553090335,
+      "eval_iou_concrete": 0.8931911820102879,
+      "eval_iou_concrete tiles": 0.8665028844677444,
+      "eval_iou_dark bricks": 0.8496761150694441,
+      "eval_iou_grass": 0.8404022756638686,
+      "eval_iou_red bricks": 0.9049480561208096,
+      "eval_iou_road": 0.9605439586989672,
+      "eval_iou_rough concrete": 0.6186517568873282,
+      "eval_iou_rough red sidewalk": 0.6307841342667982,
+      "eval_iou_speedway bricks": 0.9796196710495166,
+      "eval_iou_steel": 0.8656120472684063,
+      "eval_iou_tiles": 0.2475074183976261,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08046878129243851,
+      "eval_mean_accuracy": 0.8068583323039353,
+      "eval_mean_iou": 0.7677001487298837,
+      "eval_overall_accuracy": 0.9800751572948391,
+      "eval_runtime": 5.2952,
+      "eval_samples_per_second": 22.284,
+      "eval_steps_per_second": 1.511,
+      "step": 2720
+    },
+    {
+      "epoch": 160.06,
+      "learning_rate": 1.0510835913312694e-05,
+      "loss": 0.0212,
+      "step": 2721
+    },
+    {
+      "epoch": 160.12,
+      "learning_rate": 1.0495356037151703e-05,
+      "loss": 0.0231,
+      "step": 2722
+    },
+    {
+      "epoch": 160.18,
+      "learning_rate": 1.0479876160990713e-05,
+      "loss": 0.0237,
+      "step": 2723
+    },
+    {
+      "epoch": 160.24,
+      "learning_rate": 1.0464396284829723e-05,
+      "loss": 0.0232,
+      "step": 2724
+    },
+    {
+      "epoch": 160.29,
+      "learning_rate": 1.0448916408668731e-05,
+      "loss": 0.0216,
+      "step": 2725
+    },
+    {
+      "epoch": 160.35,
+      "learning_rate": 1.043343653250774e-05,
+      "loss": 0.0204,
+      "step": 2726
+    },
+    {
+      "epoch": 160.41,
+      "learning_rate": 1.041795665634675e-05,
+      "loss": 0.0224,
+      "step": 2727
+    },
+    {
+      "epoch": 160.47,
+      "learning_rate": 1.040247678018576e-05,
+      "loss": 0.0204,
+      "step": 2728
+    },
+    {
+      "epoch": 160.53,
+      "learning_rate": 1.0386996904024768e-05,
+      "loss": 0.0199,
+      "step": 2729
+    },
+    {
+      "epoch": 160.59,
+      "learning_rate": 1.0371517027863778e-05,
+      "loss": 0.0202,
+      "step": 2730
+    },
+    {
+      "epoch": 160.65,
+      "learning_rate": 1.0356037151702786e-05,
+      "loss": 0.0224,
+      "step": 2731
+    },
+    {
+      "epoch": 160.71,
+      "learning_rate": 1.0340557275541796e-05,
+      "loss": 0.0267,
+      "step": 2732
+    },
+    {
+      "epoch": 160.76,
+      "learning_rate": 1.0325077399380806e-05,
+      "loss": 0.0227,
+      "step": 2733
+    },
+    {
+      "epoch": 160.82,
+      "learning_rate": 1.0309597523219814e-05,
+      "loss": 0.0203,
+      "step": 2734
+    },
+    {
+      "epoch": 160.88,
+      "learning_rate": 1.0294117647058824e-05,
+      "loss": 0.0226,
+      "step": 2735
+    },
+    {
+      "epoch": 160.94,
+      "learning_rate": 1.0278637770897834e-05,
+      "loss": 0.0194,
+      "step": 2736
+    },
+    {
+      "epoch": 161.0,
+      "learning_rate": 1.0263157894736843e-05,
+      "loss": 0.0234,
+      "step": 2737
+    },
+    {
+      "epoch": 161.06,
+      "learning_rate": 1.0247678018575851e-05,
+      "loss": 0.023,
+      "step": 2738
+    },
+    {
+      "epoch": 161.12,
+      "learning_rate": 1.0232198142414861e-05,
+      "loss": 0.0218,
+      "step": 2739
+    },
+    {
+      "epoch": 161.18,
+      "learning_rate": 1.0216718266253871e-05,
+      "loss": 0.0188,
+      "step": 2740
+    },
+    {
+      "epoch": 161.18,
+      "eval_accuracy_NAT": 0.9934050284252445,
+      "eval_accuracy_REST": 0.36585365853658536,
+      "eval_accuracy_concrete": 0.9578820196633895,
+      "eval_accuracy_concrete tiles": 0.9386283290124912,
+      "eval_accuracy_dark bricks": 0.867656910973719,
+      "eval_accuracy_grass": 0.8992210653010707,
+      "eval_accuracy_red bricks": 0.9273945518453427,
+      "eval_accuracy_road": 0.9822787610737703,
+      "eval_accuracy_rough concrete": 0.6254774489181797,
+      "eval_accuracy_rough red sidewalk": 0.7694147196173832,
+      "eval_accuracy_speedway bricks": 0.9895726015904057,
+      "eval_accuracy_steel": 0.9167612088741397,
+      "eval_accuracy_tiles": 0.2531430821897058,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837008827549625,
+      "eval_iou_REST": 0.35714285714285715,
+      "eval_iou_concrete": 0.892026902069264,
+      "eval_iou_concrete tiles": 0.8708753362052,
+      "eval_iou_dark bricks": 0.849418389479947,
+      "eval_iou_grass": 0.8420361509835194,
+      "eval_iou_red bricks": 0.9019335541074671,
+      "eval_iou_road": 0.9608230274555067,
+      "eval_iou_rough concrete": 0.6191464889255905,
+      "eval_iou_rough red sidewalk": 0.6212641933329285,
+      "eval_iou_speedway bricks": 0.9793934745804969,
+      "eval_iou_steel": 0.8653554933883089,
+      "eval_iou_tiles": 0.248439079502884,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08005556464195251,
+      "eval_mean_accuracy": 0.8066684143093406,
+      "eval_mean_iou": 0.768581217686841,
+      "eval_overall_accuracy": 0.9800700494798563,
+      "eval_runtime": 5.4771,
+      "eval_samples_per_second": 21.544,
+      "eval_steps_per_second": 1.461,
+      "step": 2740
+    },
+    {
+      "epoch": 161.24,
+      "learning_rate": 1.020123839009288e-05,
+      "loss": 0.0216,
+      "step": 2741
+    },
+    {
+      "epoch": 161.29,
+      "learning_rate": 1.018575851393189e-05,
+      "loss": 0.024,
+      "step": 2742
+    },
+    {
+      "epoch": 161.35,
+      "learning_rate": 1.0170278637770898e-05,
+      "loss": 0.0208,
+      "step": 2743
+    },
+    {
+      "epoch": 161.41,
+      "learning_rate": 1.0154798761609908e-05,
+      "loss": 0.0194,
+      "step": 2744
+    },
+    {
+      "epoch": 161.47,
+      "learning_rate": 1.0139318885448918e-05,
+      "loss": 0.0217,
+      "step": 2745
+    },
+    {
+      "epoch": 161.53,
+      "learning_rate": 1.0123839009287926e-05,
+      "loss": 0.0198,
+      "step": 2746
+    },
+    {
+      "epoch": 161.59,
+      "learning_rate": 1.0108359133126936e-05,
+      "loss": 0.0206,
+      "step": 2747
+    },
+    {
+      "epoch": 161.65,
+      "learning_rate": 1.0092879256965946e-05,
+      "loss": 0.026,
+      "step": 2748
+    },
+    {
+      "epoch": 161.71,
+      "learning_rate": 1.0077399380804954e-05,
+      "loss": 0.0222,
+      "step": 2749
+    },
+    {
+      "epoch": 161.76,
+      "learning_rate": 1.0061919504643963e-05,
+      "loss": 0.0183,
+      "step": 2750
+    },
+    {
+      "epoch": 161.82,
+      "learning_rate": 1.0046439628482973e-05,
+      "loss": 0.0225,
+      "step": 2751
+    },
+    {
+      "epoch": 161.88,
+      "learning_rate": 1.0030959752321983e-05,
+      "loss": 0.0251,
+      "step": 2752
+    },
+    {
+      "epoch": 161.94,
+      "learning_rate": 1.001547987616099e-05,
+      "loss": 0.0222,
+      "step": 2753
+    },
+    {
+      "epoch": 162.0,
+      "learning_rate": 1e-05,
+      "loss": 0.0237,
+      "step": 2754
+    },
+    {
+      "epoch": 162.06,
+      "learning_rate": 9.984520123839009e-06,
+      "loss": 0.0187,
+      "step": 2755
+    },
+    {
+      "epoch": 162.12,
+      "learning_rate": 9.969040247678019e-06,
+      "loss": 0.0206,
+      "step": 2756
+    },
+    {
+      "epoch": 162.18,
+      "learning_rate": 9.953560371517029e-06,
+      "loss": 0.0226,
+      "step": 2757
+    },
+    {
+      "epoch": 162.24,
+      "learning_rate": 9.938080495356037e-06,
+      "loss": 0.0226,
+      "step": 2758
+    },
+    {
+      "epoch": 162.29,
+      "learning_rate": 9.922600619195046e-06,
+      "loss": 0.0271,
+      "step": 2759
+    },
+    {
+      "epoch": 162.35,
+      "learning_rate": 9.907120743034057e-06,
+      "loss": 0.0218,
+      "step": 2760
+    },
+    {
+      "epoch": 162.35,
+      "eval_accuracy_NAT": 0.9935705375255764,
+      "eval_accuracy_REST": 0.35772357723577236,
+      "eval_accuracy_concrete": 0.9578369201463365,
+      "eval_accuracy_concrete tiles": 0.932076361065284,
+      "eval_accuracy_dark bricks": 0.8665467143923588,
+      "eval_accuracy_grass": 0.8999950796156044,
+      "eval_accuracy_red bricks": 0.9289872583479789,
+      "eval_accuracy_road": 0.9800422295627247,
+      "eval_accuracy_rough concrete": 0.6247147224092624,
+      "eval_accuracy_rough red sidewalk": 0.7756487866505237,
+      "eval_accuracy_speedway bricks": 0.9893912972547987,
+      "eval_accuracy_steel": 0.9216165967804005,
+      "eval_accuracy_tiles": 0.2514768699445606,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837143175727022,
+      "eval_iou_REST": 0.3514376996805112,
+      "eval_iou_concrete": 0.8918607866436488,
+      "eval_iou_concrete tiles": 0.8731977655604867,
+      "eval_iou_dark bricks": 0.8478191658707418,
+      "eval_iou_grass": 0.8421416112552349,
+      "eval_iou_red bricks": 0.9040619989310529,
+      "eval_iou_road": 0.9609052710893089,
+      "eval_iou_rough concrete": 0.618263843840302,
+      "eval_iou_rough red sidewalk": 0.6240282172437367,
+      "eval_iou_speedway bricks": 0.9795047563853914,
+      "eval_iou_steel": 0.8673546420701349,
+      "eval_iou_tiles": 0.24692863729660589,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08054724335670471,
+      "eval_mean_accuracy": 0.8061251500716294,
+      "eval_mean_iou": 0.7685552856492198,
+      "eval_overall_accuracy": 0.9800904160839016,
+      "eval_runtime": 5.5636,
+      "eval_samples_per_second": 21.209,
+      "eval_steps_per_second": 1.438,
+      "step": 2760
+    },
+    {
+      "epoch": 162.41,
+      "learning_rate": 9.891640866873066e-06,
+      "loss": 0.02,
+      "step": 2761
+    },
+    {
+      "epoch": 162.47,
+      "learning_rate": 9.876160990712074e-06,
+      "loss": 0.0239,
+      "step": 2762
+    },
+    {
+      "epoch": 162.53,
+      "learning_rate": 9.860681114551084e-06,
+      "loss": 0.0216,
+      "step": 2763
+    },
+    {
+      "epoch": 162.59,
+      "learning_rate": 9.845201238390094e-06,
+      "loss": 0.025,
+      "step": 2764
+    },
+    {
+      "epoch": 162.65,
+      "learning_rate": 9.829721362229102e-06,
+      "loss": 0.0201,
+      "step": 2765
+    },
+    {
+      "epoch": 162.71,
+      "learning_rate": 9.814241486068112e-06,
+      "loss": 0.0207,
+      "step": 2766
+    },
+    {
+      "epoch": 162.76,
+      "learning_rate": 9.79876160990712e-06,
+      "loss": 0.0209,
+      "step": 2767
+    },
+    {
+      "epoch": 162.82,
+      "learning_rate": 9.78328173374613e-06,
+      "loss": 0.0259,
+      "step": 2768
+    },
+    {
+      "epoch": 162.88,
+      "learning_rate": 9.76780185758514e-06,
+      "loss": 0.0178,
+      "step": 2769
+    },
+    {
+      "epoch": 162.94,
+      "learning_rate": 9.752321981424149e-06,
+      "loss": 0.0223,
+      "step": 2770
+    },
+    {
+      "epoch": 163.0,
+      "learning_rate": 9.736842105263157e-06,
+      "loss": 0.0252,
+      "step": 2771
+    },
+    {
+      "epoch": 163.06,
+      "learning_rate": 9.721362229102169e-06,
+      "loss": 0.0217,
+      "step": 2772
+    },
+    {
+      "epoch": 163.12,
+      "learning_rate": 9.705882352941177e-06,
+      "loss": 0.0182,
+      "step": 2773
+    },
+    {
+      "epoch": 163.18,
+      "learning_rate": 9.690402476780186e-06,
+      "loss": 0.0217,
+      "step": 2774
+    },
+    {
+      "epoch": 163.24,
+      "learning_rate": 9.674922600619196e-06,
+      "loss": 0.0222,
+      "step": 2775
+    },
+    {
+      "epoch": 163.29,
+      "learning_rate": 9.659442724458206e-06,
+      "loss": 0.0223,
+      "step": 2776
+    },
+    {
+      "epoch": 163.35,
+      "learning_rate": 9.643962848297214e-06,
+      "loss": 0.023,
+      "step": 2777
+    },
+    {
+      "epoch": 163.41,
+      "learning_rate": 9.628482972136222e-06,
+      "loss": 0.0213,
+      "step": 2778
+    },
+    {
+      "epoch": 163.47,
+      "learning_rate": 9.613003095975232e-06,
+      "loss": 0.0225,
+      "step": 2779
+    },
+    {
+      "epoch": 163.53,
+      "learning_rate": 9.597523219814242e-06,
+      "loss": 0.0203,
+      "step": 2780
+    },
+    {
+      "epoch": 163.53,
+      "eval_accuracy_NAT": 0.9931384886884234,
+      "eval_accuracy_REST": 0.32682926829268294,
+      "eval_accuracy_concrete": 0.9579603705192865,
+      "eval_accuracy_concrete tiles": 0.9177468772095215,
+      "eval_accuracy_dark bricks": 0.8569138449170224,
+      "eval_accuracy_grass": 0.9023313159757311,
+      "eval_accuracy_red bricks": 0.9230008787346221,
+      "eval_accuracy_road": 0.9821793596732794,
+      "eval_accuracy_rough concrete": 0.6227483181284599,
+      "eval_accuracy_rough red sidewalk": 0.7698132787883801,
+      "eval_accuracy_speedway bricks": 0.9898034304562494,
+      "eval_accuracy_steel": 0.9155167502136571,
+      "eval_accuracy_tiles": 0.24114635402465995,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835999365252531,
+      "eval_iou_REST": 0.3216,
+      "eval_iou_concrete": 0.8909636457803461,
+      "eval_iou_concrete tiles": 0.871316372423978,
+      "eval_iou_dark bricks": 0.8405195073410687,
+      "eval_iou_grass": 0.8406492755462934,
+      "eval_iou_red bricks": 0.9017545742340506,
+      "eval_iou_road": 0.959926438136008,
+      "eval_iou_rough concrete": 0.6166923353809296,
+      "eval_iou_rough red sidewalk": 0.6241555495939322,
+      "eval_iou_speedway bricks": 0.9795240786802043,
+      "eval_iou_steel": 0.8663756408558618,
+      "eval_iou_tiles": 0.2384161499985024,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08148735761642456,
+      "eval_mean_accuracy": 0.7999329642786136,
+      "eval_mean_iou": 0.7642687311151098,
+      "eval_overall_accuracy": 0.9799140671552238,
+      "eval_runtime": 5.3192,
+      "eval_samples_per_second": 22.184,
+      "eval_steps_per_second": 1.504,
+      "step": 2780
+    },
+    {
+      "epoch": 163.59,
+      "learning_rate": 9.58204334365325e-06,
+      "loss": 0.0227,
+      "step": 2781
+    },
+    {
+      "epoch": 163.65,
+      "learning_rate": 9.56656346749226e-06,
+      "loss": 0.0205,
+      "step": 2782
+    },
+    {
+      "epoch": 163.71,
+      "learning_rate": 9.551083591331269e-06,
+      "loss": 0.0232,
+      "step": 2783
+    },
+    {
+      "epoch": 163.76,
+      "learning_rate": 9.535603715170279e-06,
+      "loss": 0.0217,
+      "step": 2784
+    },
+    {
+      "epoch": 163.82,
+      "learning_rate": 9.520123839009289e-06,
+      "loss": 0.0247,
+      "step": 2785
+    },
+    {
+      "epoch": 163.88,
+      "learning_rate": 9.504643962848297e-06,
+      "loss": 0.021,
+      "step": 2786
+    },
+    {
+      "epoch": 163.94,
+      "learning_rate": 9.489164086687305e-06,
+      "loss": 0.0231,
+      "step": 2787
+    },
+    {
+      "epoch": 164.0,
+      "learning_rate": 9.473684210526317e-06,
+      "loss": 0.0213,
+      "step": 2788
+    },
+    {
+      "epoch": 164.06,
+      "learning_rate": 9.458204334365325e-06,
+      "loss": 0.0208,
+      "step": 2789
+    },
+    {
+      "epoch": 164.12,
+      "learning_rate": 9.442724458204334e-06,
+      "loss": 0.019,
+      "step": 2790
+    },
+    {
+      "epoch": 164.18,
+      "learning_rate": 9.427244582043344e-06,
+      "loss": 0.0186,
+      "step": 2791
+    },
+    {
+      "epoch": 164.24,
+      "learning_rate": 9.411764705882354e-06,
+      "loss": 0.0214,
+      "step": 2792
+    },
+    {
+      "epoch": 164.29,
+      "learning_rate": 9.396284829721362e-06,
+      "loss": 0.0215,
+      "step": 2793
+    },
+    {
+      "epoch": 164.35,
+      "learning_rate": 9.380804953560372e-06,
+      "loss": 0.0224,
+      "step": 2794
+    },
+    {
+      "epoch": 164.41,
+      "learning_rate": 9.36532507739938e-06,
+      "loss": 0.0219,
+      "step": 2795
+    },
+    {
+      "epoch": 164.47,
+      "learning_rate": 9.34984520123839e-06,
+      "loss": 0.022,
+      "step": 2796
+    },
+    {
+      "epoch": 164.53,
+      "learning_rate": 9.3343653250774e-06,
+      "loss": 0.0231,
+      "step": 2797
+    },
+    {
+      "epoch": 164.59,
+      "learning_rate": 9.318885448916409e-06,
+      "loss": 0.0229,
+      "step": 2798
+    },
+    {
+      "epoch": 164.65,
+      "learning_rate": 9.303405572755417e-06,
+      "loss": 0.0228,
+      "step": 2799
+    },
+    {
+      "epoch": 164.71,
+      "learning_rate": 9.287925696594429e-06,
+      "loss": 0.0229,
+      "step": 2800
+    },
+    {
+      "epoch": 164.71,
+      "eval_accuracy_NAT": 0.9933407656084381,
+      "eval_accuracy_REST": 0.3300813008130081,
+      "eval_accuracy_concrete": 0.9548225142904317,
+      "eval_accuracy_concrete tiles": 0.9398538769738393,
+      "eval_accuracy_dark bricks": 0.8573923779262294,
+      "eval_accuracy_grass": 0.9024126915638117,
+      "eval_accuracy_red bricks": 0.930305360281195,
+      "eval_accuracy_road": 0.9801042218339986,
+      "eval_accuracy_rough concrete": 0.6250603328586155,
+      "eval_accuracy_rough red sidewalk": 0.789056918761609,
+      "eval_accuracy_speedway bricks": 0.9911529310639103,
+      "eval_accuracy_steel": 0.9162364371498398,
+      "eval_accuracy_tiles": 0.25123451179981215,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836287408897749,
+      "eval_iou_REST": 0.3248,
+      "eval_iou_concrete": 0.8924182216449549,
+      "eval_iou_concrete tiles": 0.8792997001234786,
+      "eval_iou_dark bricks": 0.8400977146768945,
+      "eval_iou_grass": 0.8409675101076236,
+      "eval_iou_red bricks": 0.9061196105702365,
+      "eval_iou_road": 0.9611498756355792,
+      "eval_iou_rough concrete": 0.6189855192191944,
+      "eval_iou_rough red sidewalk": 0.6315599400509206,
+      "eval_iou_speedway bricks": 0.9792372809616524,
+      "eval_iou_steel": 0.866876300359372,
+      "eval_iou_tiles": 0.24814482345900657,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08053811639547348,
+      "eval_mean_accuracy": 0.8046964800711337,
+      "eval_mean_iou": 0.7671757875152837,
+      "eval_overall_accuracy": 0.9800550493143373,
+      "eval_runtime": 5.3606,
+      "eval_samples_per_second": 22.012,
+      "eval_steps_per_second": 1.492,
+      "step": 2800
+    },
+    {
+      "epoch": 164.76,
+      "learning_rate": 9.272445820433437e-06,
+      "loss": 0.0194,
+      "step": 2801
+    },
+    {
+      "epoch": 164.82,
+      "learning_rate": 9.256965944272445e-06,
+      "loss": 0.0188,
+      "step": 2802
+    },
+    {
+      "epoch": 164.88,
+      "learning_rate": 9.241486068111455e-06,
+      "loss": 0.0224,
+      "step": 2803
+    },
+    {
+      "epoch": 164.94,
+      "learning_rate": 9.226006191950465e-06,
+      "loss": 0.0234,
+      "step": 2804
+    },
+    {
+      "epoch": 165.0,
+      "learning_rate": 9.210526315789474e-06,
+      "loss": 0.0228,
+      "step": 2805
+    },
+    {
+      "epoch": 165.06,
+      "learning_rate": 9.195046439628484e-06,
+      "loss": 0.0195,
+      "step": 2806
+    },
+    {
+      "epoch": 165.12,
+      "learning_rate": 9.179566563467492e-06,
+      "loss": 0.0222,
+      "step": 2807
+    },
+    {
+      "epoch": 165.18,
+      "learning_rate": 9.164086687306502e-06,
+      "loss": 0.0204,
+      "step": 2808
+    },
+    {
+      "epoch": 165.24,
+      "learning_rate": 9.148606811145512e-06,
+      "loss": 0.0203,
+      "step": 2809
+    },
+    {
+      "epoch": 165.29,
+      "learning_rate": 9.13312693498452e-06,
+      "loss": 0.0199,
+      "step": 2810
+    },
+    {
+      "epoch": 165.35,
+      "learning_rate": 9.117647058823529e-06,
+      "loss": 0.0228,
+      "step": 2811
+    },
+    {
+      "epoch": 165.41,
+      "learning_rate": 9.10216718266254e-06,
+      "loss": 0.0199,
+      "step": 2812
+    },
+    {
+      "epoch": 165.47,
+      "learning_rate": 9.086687306501549e-06,
+      "loss": 0.0229,
+      "step": 2813
+    },
+    {
+      "epoch": 165.53,
+      "learning_rate": 9.071207430340557e-06,
+      "loss": 0.0255,
+      "step": 2814
+    },
+    {
+      "epoch": 165.59,
+      "learning_rate": 9.055727554179567e-06,
+      "loss": 0.0239,
+      "step": 2815
+    },
+    {
+      "epoch": 165.65,
+      "learning_rate": 9.040247678018577e-06,
+      "loss": 0.0251,
+      "step": 2816
+    },
+    {
+      "epoch": 165.71,
+      "learning_rate": 9.024767801857585e-06,
+      "loss": 0.0207,
+      "step": 2817
+    },
+    {
+      "epoch": 165.76,
+      "learning_rate": 9.009287925696595e-06,
+      "loss": 0.0227,
+      "step": 2818
+    },
+    {
+      "epoch": 165.82,
+      "learning_rate": 8.993808049535604e-06,
+      "loss": 0.0207,
+      "step": 2819
+    },
+    {
+      "epoch": 165.88,
+      "learning_rate": 8.978328173374614e-06,
+      "loss": 0.0189,
+      "step": 2820
+    },
+    {
+      "epoch": 165.88,
+      "eval_accuracy_NAT": 0.9940474948580312,
+      "eval_accuracy_REST": 0.33495934959349594,
+      "eval_accuracy_concrete": 0.955562452129538,
+      "eval_accuracy_concrete tiles": 0.9350930945086025,
+      "eval_accuracy_dark bricks": 0.8622303466493119,
+      "eval_accuracy_grass": 0.892068529599897,
+      "eval_accuracy_red bricks": 0.9278339191564148,
+      "eval_accuracy_road": 0.9820655290372333,
+      "eval_accuracy_rough concrete": 0.624816022023728,
+      "eval_accuracy_rough red sidewalk": 0.781582054309327,
+      "eval_accuracy_speedway bricks": 0.9889923591704007,
+      "eval_accuracy_steel": 0.9258422586175014,
+      "eval_accuracy_tiles": 0.25238571298736706,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835063978476642,
+      "eval_iou_REST": 0.32854864433811803,
+      "eval_iou_concrete": 0.8929697642896813,
+      "eval_iou_concrete tiles": 0.8755213275371273,
+      "eval_iou_dark bricks": 0.8449886510720517,
+      "eval_iou_grass": 0.8390086707491481,
+      "eval_iou_red bricks": 0.9041477120685042,
+      "eval_iou_road": 0.9605225675306259,
+      "eval_iou_rough concrete": 0.6186669184067215,
+      "eval_iou_rough red sidewalk": 0.6254641303235823,
+      "eval_iou_speedway bricks": 0.979499484644301,
+      "eval_iou_steel": 0.8664957832649643,
+      "eval_iou_tiles": 0.24715201139195445,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08159501105546951,
+      "eval_mean_accuracy": 0.8044214709723729,
+      "eval_mean_iou": 0.7666532356511111,
+      "eval_overall_accuracy": 0.9800159001754503,
+      "eval_runtime": 5.2209,
+      "eval_samples_per_second": 22.601,
+      "eval_steps_per_second": 1.532,
+      "step": 2820
+    },
+    {
+      "epoch": 165.94,
+      "learning_rate": 8.962848297213624e-06,
+      "loss": 0.0214,
+      "step": 2821
+    },
+    {
+      "epoch": 166.0,
+      "learning_rate": 8.947368421052632e-06,
+      "loss": 0.0193,
+      "step": 2822
+    },
+    {
+      "epoch": 166.06,
+      "learning_rate": 8.93188854489164e-06,
+      "loss": 0.0234,
+      "step": 2823
+    },
+    {
+      "epoch": 166.12,
+      "learning_rate": 8.916408668730652e-06,
+      "loss": 0.0227,
+      "step": 2824
+    },
+    {
+      "epoch": 166.18,
+      "learning_rate": 8.90092879256966e-06,
+      "loss": 0.0187,
+      "step": 2825
+    },
+    {
+      "epoch": 166.24,
+      "learning_rate": 8.885448916408668e-06,
+      "loss": 0.0223,
+      "step": 2826
+    },
+    {
+      "epoch": 166.29,
+      "learning_rate": 8.869969040247678e-06,
+      "loss": 0.027,
+      "step": 2827
+    },
+    {
+      "epoch": 166.35,
+      "learning_rate": 8.854489164086688e-06,
+      "loss": 0.0221,
+      "step": 2828
+    },
+    {
+      "epoch": 166.41,
+      "learning_rate": 8.839009287925697e-06,
+      "loss": 0.0182,
+      "step": 2829
+    },
+    {
+      "epoch": 166.47,
+      "learning_rate": 8.823529411764707e-06,
+      "loss": 0.0195,
+      "step": 2830
+    },
+    {
+      "epoch": 166.53,
+      "learning_rate": 8.808049535603715e-06,
+      "loss": 0.0218,
+      "step": 2831
+    },
+    {
+      "epoch": 166.59,
+      "learning_rate": 8.792569659442725e-06,
+      "loss": 0.0192,
+      "step": 2832
+    },
+    {
+      "epoch": 166.65,
+      "learning_rate": 8.777089783281735e-06,
+      "loss": 0.0189,
+      "step": 2833
+    },
+    {
+      "epoch": 166.71,
+      "learning_rate": 8.761609907120743e-06,
+      "loss": 0.0213,
+      "step": 2834
+    },
+    {
+      "epoch": 166.76,
+      "learning_rate": 8.746130030959752e-06,
+      "loss": 0.0233,
+      "step": 2835
+    },
+    {
+      "epoch": 166.82,
+      "learning_rate": 8.730650154798763e-06,
+      "loss": 0.0246,
+      "step": 2836
+    },
+    {
+      "epoch": 166.88,
+      "learning_rate": 8.715170278637772e-06,
+      "loss": 0.0241,
+      "step": 2837
+    },
+    {
+      "epoch": 166.94,
+      "learning_rate": 8.69969040247678e-06,
+      "loss": 0.0218,
+      "step": 2838
+    },
+    {
+      "epoch": 167.0,
+      "learning_rate": 8.68421052631579e-06,
+      "loss": 0.0179,
+      "step": 2839
+    },
+    {
+      "epoch": 167.06,
+      "learning_rate": 8.6687306501548e-06,
+      "loss": 0.0235,
+      "step": 2840
+    },
+    {
+      "epoch": 167.06,
+      "eval_accuracy_NAT": 0.9932580110582722,
+      "eval_accuracy_REST": 0.3772357723577236,
+      "eval_accuracy_concrete": 0.9555295829899909,
+      "eval_accuracy_concrete tiles": 0.9379212821117134,
+      "eval_accuracy_dark bricks": 0.8621824933483911,
+      "eval_accuracy_grass": 0.9031819747395035,
+      "eval_accuracy_red bricks": 0.9265158172231986,
+      "eval_accuracy_road": 0.9821911168281762,
+      "eval_accuracy_rough concrete": 0.6231356401837694,
+      "eval_accuracy_rough red sidewalk": 0.7864249242361576,
+      "eval_accuracy_speedway bricks": 0.9898682064854172,
+      "eval_accuracy_steel": 0.9224262445836061,
+      "eval_accuracy_tiles": 0.25387015662395107,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837118504619963,
+      "eval_iou_REST": 0.3700159489633174,
+      "eval_iou_concrete": 0.8925125295724327,
+      "eval_iou_concrete tiles": 0.877240163121349,
+      "eval_iou_dark bricks": 0.8443359310929804,
+      "eval_iou_grass": 0.8411711259965966,
+      "eval_iou_red bricks": 0.9049458212638128,
+      "eval_iou_road": 0.9603020712667065,
+      "eval_iou_rough concrete": 0.6175897381382658,
+      "eval_iou_rough red sidewalk": 0.6271732285825662,
+      "eval_iou_speedway bricks": 0.979474234228307,
+      "eval_iou_steel": 0.8668567282018285,
+      "eval_iou_tiles": 0.24858354839666577,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08091432601213455,
+      "eval_mean_accuracy": 0.8087493248284514,
+      "eval_mean_iou": 0.7703009937912942,
+      "eval_overall_accuracy": 0.9800649416648736,
+      "eval_runtime": 5.3919,
+      "eval_samples_per_second": 21.885,
+      "eval_steps_per_second": 1.484,
+      "step": 2840
+    },
+    {
+      "epoch": 167.12,
+      "learning_rate": 8.653250773993808e-06,
+      "loss": 0.0205,
+      "step": 2841
+    },
+    {
+      "epoch": 167.18,
+      "learning_rate": 8.637770897832818e-06,
+      "loss": 0.0224,
+      "step": 2842
+    },
+    {
+      "epoch": 167.24,
+      "learning_rate": 8.622291021671827e-06,
+      "loss": 0.0242,
+      "step": 2843
+    },
+    {
+      "epoch": 167.29,
+      "learning_rate": 8.606811145510837e-06,
+      "loss": 0.022,
+      "step": 2844
+    },
+    {
+      "epoch": 167.35,
+      "learning_rate": 8.591331269349847e-06,
+      "loss": 0.0235,
+      "step": 2845
+    },
+    {
+      "epoch": 167.41,
+      "learning_rate": 8.575851393188855e-06,
+      "loss": 0.0224,
+      "step": 2846
+    },
+    {
+      "epoch": 167.47,
+      "learning_rate": 8.560371517027863e-06,
+      "loss": 0.0191,
+      "step": 2847
+    },
+    {
+      "epoch": 167.53,
+      "learning_rate": 8.544891640866875e-06,
+      "loss": 0.0206,
+      "step": 2848
+    },
+    {
+      "epoch": 167.59,
+      "learning_rate": 8.529411764705883e-06,
+      "loss": 0.0219,
+      "step": 2849
+    },
+    {
+      "epoch": 167.65,
+      "learning_rate": 8.513931888544892e-06,
+      "loss": 0.0209,
+      "step": 2850
+    },
+    {
+      "epoch": 167.71,
+      "learning_rate": 8.4984520123839e-06,
+      "loss": 0.0214,
+      "step": 2851
+    },
+    {
+      "epoch": 167.76,
+      "learning_rate": 8.482972136222912e-06,
+      "loss": 0.0219,
+      "step": 2852
+    },
+    {
+      "epoch": 167.82,
+      "learning_rate": 8.46749226006192e-06,
+      "loss": 0.0172,
+      "step": 2853
+    },
+    {
+      "epoch": 167.88,
+      "learning_rate": 8.452012383900928e-06,
+      "loss": 0.0218,
+      "step": 2854
+    },
+    {
+      "epoch": 167.94,
+      "learning_rate": 8.436532507739938e-06,
+      "loss": 0.0213,
+      "step": 2855
+    },
+    {
+      "epoch": 168.0,
+      "learning_rate": 8.421052631578948e-06,
+      "loss": 0.0244,
+      "step": 2856
+    },
+    {
+      "epoch": 168.06,
+      "learning_rate": 8.405572755417957e-06,
+      "loss": 0.0188,
+      "step": 2857
+    },
+    {
+      "epoch": 168.12,
+      "learning_rate": 8.390092879256967e-06,
+      "loss": 0.0214,
+      "step": 2858
+    },
+    {
+      "epoch": 168.18,
+      "learning_rate": 8.374613003095975e-06,
+      "loss": 0.0215,
+      "step": 2859
+    },
+    {
+      "epoch": 168.24,
+      "learning_rate": 8.359133126934985e-06,
+      "loss": 0.0218,
+      "step": 2860
+    },
+    {
+      "epoch": 168.24,
+      "eval_accuracy_NAT": 0.9934739276533305,
+      "eval_accuracy_REST": 0.33495934959349594,
+      "eval_accuracy_concrete": 0.9582152974504249,
+      "eval_accuracy_concrete tiles": 0.9313693141645062,
+      "eval_accuracy_dark bricks": 0.8572009647225466,
+      "eval_accuracy_grass": 0.8960862127044326,
+      "eval_accuracy_red bricks": 0.9207491212653779,
+      "eval_accuracy_road": 0.9808614894925772,
+      "eval_accuracy_rough concrete": 0.6219617564161388,
+      "eval_accuracy_rough red sidewalk": 0.7770099038194,
+      "eval_accuracy_speedway bricks": 0.990580057345636,
+      "eval_accuracy_steel": 0.9125030611683917,
+      "eval_accuracy_tiles": 0.2425702081250568,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9836868716362848,
+      "eval_iou_REST": 0.329073482428115,
+      "eval_iou_concrete": 0.8914491801833099,
+      "eval_iou_concrete tiles": 0.8748920719962806,
+      "eval_iou_dark bricks": 0.8394771866681663,
+      "eval_iou_grass": 0.8411808752639213,
+      "eval_iou_red bricks": 0.900327587132807,
+      "eval_iou_road": 0.9612753603756716,
+      "eval_iou_rough concrete": 0.6170494516863232,
+      "eval_iou_rough red sidewalk": 0.6243225116464554,
+      "eval_iou_speedway bricks": 0.9792170629935452,
+      "eval_iou_steel": 0.8658209839952579,
+      "eval_iou_tiles": 0.24000359690666026,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08251631259918213,
+      "eval_mean_accuracy": 0.8013492818401011,
+      "eval_mean_iou": 0.7652135556086767,
+      "eval_overall_accuracy": 0.9799988633495266,
+      "eval_runtime": 5.2123,
+      "eval_samples_per_second": 22.639,
+      "eval_steps_per_second": 1.535,
+      "step": 2860
+    },
+    {
+      "epoch": 168.29,
+      "learning_rate": 8.343653250773995e-06,
+      "loss": 0.0212,
+      "step": 2861
+    },
+    {
+      "epoch": 168.35,
+      "learning_rate": 8.328173374613003e-06,
+      "loss": 0.0229,
+      "step": 2862
+    },
+    {
+      "epoch": 168.41,
+      "learning_rate": 8.312693498452011e-06,
+      "loss": 0.0225,
+      "step": 2863
+    },
+    {
+      "epoch": 168.47,
+      "learning_rate": 8.297213622291023e-06,
+      "loss": 0.0222,
+      "step": 2864
+    },
+    {
+      "epoch": 168.53,
+      "learning_rate": 8.281733746130031e-06,
+      "loss": 0.0194,
+      "step": 2865
+    },
+    {
+      "epoch": 168.59,
+      "learning_rate": 8.26625386996904e-06,
+      "loss": 0.0234,
+      "step": 2866
+    },
+    {
+      "epoch": 168.65,
+      "learning_rate": 8.25077399380805e-06,
+      "loss": 0.0221,
+      "step": 2867
+    },
+    {
+      "epoch": 168.71,
+      "learning_rate": 8.23529411764706e-06,
+      "loss": 0.0248,
+      "step": 2868
+    },
+    {
+      "epoch": 168.76,
+      "learning_rate": 8.219814241486068e-06,
+      "loss": 0.0207,
+      "step": 2869
+    },
+    {
+      "epoch": 168.82,
+      "learning_rate": 8.204334365325078e-06,
+      "loss": 0.0188,
+      "step": 2870
+    },
+    {
+      "epoch": 168.88,
+      "learning_rate": 8.188854489164086e-06,
+      "loss": 0.0217,
+      "step": 2871
+    },
+    {
+      "epoch": 168.94,
+      "learning_rate": 8.173374613003096e-06,
+      "loss": 0.0209,
+      "step": 2872
+    },
+    {
+      "epoch": 169.0,
+      "learning_rate": 8.157894736842106e-06,
+      "loss": 0.0196,
+      "step": 2873
+    },
+    {
+      "epoch": 169.06,
+      "learning_rate": 8.142414860681115e-06,
+      "loss": 0.0237,
+      "step": 2874
+    },
+    {
+      "epoch": 169.12,
+      "learning_rate": 8.126934984520123e-06,
+      "loss": 0.0206,
+      "step": 2875
+    },
+    {
+      "epoch": 169.18,
+      "learning_rate": 8.111455108359135e-06,
+      "loss": 0.0211,
+      "step": 2876
+    },
+    {
+      "epoch": 169.24,
+      "learning_rate": 8.095975232198143e-06,
+      "loss": 0.0223,
+      "step": 2877
+    },
+    {
+      "epoch": 169.29,
+      "learning_rate": 8.080495356037151e-06,
+      "loss": 0.0218,
+      "step": 2878
+    },
+    {
+      "epoch": 169.35,
+      "learning_rate": 8.065015479876161e-06,
+      "loss": 0.0237,
+      "step": 2879
+    },
+    {
+      "epoch": 169.41,
+      "learning_rate": 8.049535603715171e-06,
+      "loss": 0.0225,
+      "step": 2880
+    },
+    {
+      "epoch": 169.41,
+      "eval_accuracy_NAT": 0.9941456142595309,
+      "eval_accuracy_REST": 0.3089430894308943,
+      "eval_accuracy_concrete": 0.9553170801808262,
+      "eval_accuracy_concrete tiles": 0.9302616073532878,
+      "eval_accuracy_dark bricks": 0.8597467603315276,
+      "eval_accuracy_grass": 0.8912774831855326,
+      "eval_accuracy_red bricks": 0.9232205623901582,
+      "eval_accuracy_road": 0.9819297873397888,
+      "eval_accuracy_rough concrete": 0.6236302206543956,
+      "eval_accuracy_rough red sidewalk": 0.7836801299453298,
+      "eval_accuracy_speedway bricks": 0.9895869962635542,
+      "eval_accuracy_steel": 0.9214266793944634,
+      "eval_accuracy_tiles": 0.24496349480444726,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835355386718644,
+      "eval_iou_REST": 0.304,
+      "eval_iou_concrete": 0.8927851069491498,
+      "eval_iou_concrete tiles": 0.8759265012649239,
+      "eval_iou_dark bricks": 0.8426100496196453,
+      "eval_iou_grass": 0.8391563597747466,
+      "eval_iou_red bricks": 0.9028896766569986,
+      "eval_iou_road": 0.9611046655779586,
+      "eval_iou_rough concrete": 0.6184743939770001,
+      "eval_iou_rough red sidewalk": 0.6285881451724783,
+      "eval_iou_speedway bricks": 0.9794404459336035,
+      "eval_iou_steel": 0.8664952777980134,
+      "eval_iou_tiles": 0.24138754552510597,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08182083070278168,
+      "eval_mean_accuracy": 0.800625346579518,
+      "eval_mean_iou": 0.7643379774554991,
+      "eval_overall_accuracy": 0.9800418271856793,
+      "eval_runtime": 5.3416,
+      "eval_samples_per_second": 22.091,
+      "eval_steps_per_second": 1.498,
+      "step": 2880
+    },
+    {
+      "epoch": 169.47,
+      "learning_rate": 8.03405572755418e-06,
+      "loss": 0.0162,
+      "step": 2881
+    },
+    {
+      "epoch": 169.53,
+      "learning_rate": 8.01857585139319e-06,
+      "loss": 0.0217,
+      "step": 2882
+    },
+    {
+      "epoch": 169.59,
+      "learning_rate": 8.003095975232198e-06,
+      "loss": 0.0188,
+      "step": 2883
+    },
+    {
+      "epoch": 169.65,
+      "learning_rate": 7.987616099071208e-06,
+      "loss": 0.0227,
+      "step": 2884
+    },
+    {
+      "epoch": 169.71,
+      "learning_rate": 7.972136222910218e-06,
+      "loss": 0.0198,
+      "step": 2885
+    },
+    {
+      "epoch": 169.76,
+      "learning_rate": 7.956656346749226e-06,
+      "loss": 0.0199,
+      "step": 2886
+    },
+    {
+      "epoch": 169.82,
+      "learning_rate": 7.941176470588235e-06,
+      "loss": 0.0219,
+      "step": 2887
+    },
+    {
+      "epoch": 169.88,
+      "learning_rate": 7.925696594427245e-06,
+      "loss": 0.026,
+      "step": 2888
+    },
+    {
+      "epoch": 169.94,
+      "learning_rate": 7.910216718266255e-06,
+      "loss": 0.0215,
+      "step": 2889
+    },
+    {
+      "epoch": 170.0,
+      "learning_rate": 7.894736842105263e-06,
+      "loss": 0.0232,
+      "step": 2890
+    },
+    {
+      "epoch": 170.06,
+      "learning_rate": 7.879256965944273e-06,
+      "loss": 0.0207,
+      "step": 2891
+    },
+    {
+      "epoch": 170.12,
+      "learning_rate": 7.863777089783283e-06,
+      "loss": 0.0217,
+      "step": 2892
+    },
+    {
+      "epoch": 170.18,
+      "learning_rate": 7.848297213622291e-06,
+      "loss": 0.0202,
+      "step": 2893
+    },
+    {
+      "epoch": 170.24,
+      "learning_rate": 7.832817337461301e-06,
+      "loss": 0.0191,
+      "step": 2894
+    },
+    {
+      "epoch": 170.29,
+      "learning_rate": 7.81733746130031e-06,
+      "loss": 0.0226,
+      "step": 2895
+    },
+    {
+      "epoch": 170.35,
+      "learning_rate": 7.80185758513932e-06,
+      "loss": 0.0245,
+      "step": 2896
+    },
+    {
+      "epoch": 170.41,
+      "learning_rate": 7.78637770897833e-06,
+      "loss": 0.0233,
+      "step": 2897
+    },
+    {
+      "epoch": 170.47,
+      "learning_rate": 7.770897832817338e-06,
+      "loss": 0.0201,
+      "step": 2898
+    },
+    {
+      "epoch": 170.53,
+      "learning_rate": 7.755417956656346e-06,
+      "loss": 0.018,
+      "step": 2899
+    },
+    {
+      "epoch": 170.59,
+      "learning_rate": 7.739938080495356e-06,
+      "loss": 0.0204,
+      "step": 2900
+    },
+    {
+      "epoch": 170.59,
+      "eval_accuracy_NAT": 0.9937209513333699,
+      "eval_accuracy_REST": 0.38699186991869916,
+      "eval_accuracy_concrete": 0.9556748187228734,
+      "eval_accuracy_concrete tiles": 0.9337968418571765,
+      "eval_accuracy_dark bricks": 0.8589045422353234,
+      "eval_accuracy_grass": 0.8985615445465108,
+      "eval_accuracy_red bricks": 0.9321726713532513,
+      "eval_accuracy_road": 0.9815300440732984,
+      "eval_accuracy_rough concrete": 0.6253344376977577,
+      "eval_accuracy_rough red sidewalk": 0.7735507110145211,
+      "eval_accuracy_speedway bricks": 0.9901903729796905,
+      "eval_accuracy_steel": 0.9196149674891422,
+      "eval_accuracy_tiles": 0.2531127874216123,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837273476144998,
+      "eval_iou_REST": 0.37658227848101267,
+      "eval_iou_concrete": 0.8927871485255489,
+      "eval_iou_concrete tiles": 0.8769976537252645,
+      "eval_iou_dark bricks": 0.8423693177018313,
+      "eval_iou_grass": 0.8411554776859621,
+      "eval_iou_red bricks": 0.906144893492072,
+      "eval_iou_road": 0.9611352096151199,
+      "eval_iou_rough concrete": 0.6191984989556413,
+      "eval_iou_rough red sidewalk": 0.6237516296273838,
+      "eval_iou_speedway bricks": 0.9793305381747947,
+      "eval_iou_steel": 0.8666632126304684,
+      "eval_iou_tiles": 0.24970113568439928,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08075252920389175,
+      "eval_mean_accuracy": 0.807935120049479,
+      "eval_mean_iou": 0.7707341801472308,
+      "eval_overall_accuracy": 0.9801128840042372,
+      "eval_runtime": 5.3903,
+      "eval_samples_per_second": 21.891,
+      "eval_steps_per_second": 1.484,
+      "step": 2900
+    },
+    {
+      "epoch": 170.65,
+      "learning_rate": 7.724458204334366e-06,
+      "loss": 0.0234,
+      "step": 2901
+    },
+    {
+      "epoch": 170.71,
+      "learning_rate": 7.708978328173374e-06,
+      "loss": 0.019,
+      "step": 2902
+    },
+    {
+      "epoch": 170.76,
+      "learning_rate": 7.693498452012384e-06,
+      "loss": 0.0224,
+      "step": 2903
+    },
+    {
+      "epoch": 170.82,
+      "learning_rate": 7.678018575851394e-06,
+      "loss": 0.0221,
+      "step": 2904
+    },
+    {
+      "epoch": 170.88,
+      "learning_rate": 7.662538699690403e-06,
+      "loss": 0.0196,
+      "step": 2905
+    },
+    {
+      "epoch": 170.94,
+      "learning_rate": 7.647058823529413e-06,
+      "loss": 0.0215,
+      "step": 2906
+    },
+    {
+      "epoch": 171.0,
+      "learning_rate": 7.631578947368421e-06,
+      "loss": 0.0215,
+      "step": 2907
+    },
+    {
+      "epoch": 171.06,
+      "learning_rate": 7.61609907120743e-06,
+      "loss": 0.0223,
+      "step": 2908
+    },
+    {
+      "epoch": 171.12,
+      "learning_rate": 7.60061919504644e-06,
+      "loss": 0.0214,
+      "step": 2909
+    },
+    {
+      "epoch": 171.18,
+      "learning_rate": 7.585139318885449e-06,
+      "loss": 0.0197,
+      "step": 2910
+    },
+    {
+      "epoch": 171.24,
+      "learning_rate": 7.5696594427244586e-06,
+      "loss": 0.0213,
+      "step": 2911
+    },
+    {
+      "epoch": 171.29,
+      "learning_rate": 7.5541795665634686e-06,
+      "loss": 0.0212,
+      "step": 2912
+    },
+    {
+      "epoch": 171.35,
+      "learning_rate": 7.538699690402478e-06,
+      "loss": 0.0223,
+      "step": 2913
+    },
+    {
+      "epoch": 171.41,
+      "learning_rate": 7.523219814241486e-06,
+      "loss": 0.0213,
+      "step": 2914
+    },
+    {
+      "epoch": 171.47,
+      "learning_rate": 7.507739938080496e-06,
+      "loss": 0.0201,
+      "step": 2915
+    },
+    {
+      "epoch": 171.53,
+      "learning_rate": 7.492260061919505e-06,
+      "loss": 0.0231,
+      "step": 2916
+    },
+    {
+      "epoch": 171.59,
+      "learning_rate": 7.476780185758514e-06,
+      "loss": 0.0198,
+      "step": 2917
+    },
+    {
+      "epoch": 171.65,
+      "learning_rate": 7.461300309597524e-06,
+      "loss": 0.0257,
+      "step": 2918
+    },
+    {
+      "epoch": 171.71,
+      "learning_rate": 7.445820433436533e-06,
+      "loss": 0.0203,
+      "step": 2919
+    },
+    {
+      "epoch": 171.76,
+      "learning_rate": 7.430340557275542e-06,
+      "loss": 0.0184,
+      "step": 2920
+    },
+    {
+      "epoch": 171.76,
+      "eval_accuracy_NAT": 0.9939313689290036,
+      "eval_accuracy_REST": 0.35772357723577236,
+      "eval_accuracy_concrete": 0.957523516722748,
+      "eval_accuracy_concrete tiles": 0.9345510252180061,
+      "eval_accuracy_dark bricks": 0.8620676454261815,
+      "eval_accuracy_grass": 0.891021055460302,
+      "eval_accuracy_red bricks": 0.9336555360281195,
+      "eval_accuracy_road": 0.9812890223979145,
+      "eval_accuracy_rough concrete": 0.6253225200960558,
+      "eval_accuracy_rough red sidewalk": 0.7792809390956467,
+      "eval_accuracy_speedway bricks": 0.9891556701645723,
+      "eval_accuracy_steel": 0.9239006032375916,
+      "eval_accuracy_tiles": 0.25447605198582207,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9835952775509318,
+      "eval_iou_REST": 0.3486529318541997,
+      "eval_iou_concrete": 0.8926954867781031,
+      "eval_iou_concrete tiles": 0.8714753521900618,
+      "eval_iou_dark bricks": 0.844979784050507,
+      "eval_iou_grass": 0.8385279615881573,
+      "eval_iou_red bricks": 0.9021917953616728,
+      "eval_iou_road": 0.9612035165120488,
+      "eval_iou_rough concrete": 0.6189274150265699,
+      "eval_iou_rough red sidewalk": 0.6237540328405644,
+      "eval_iou_speedway bricks": 0.9795029661467786,
+      "eval_iou_steel": 0.8667543452473052,
+      "eval_iou_tiles": 0.24884464984002844,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08169034123420715,
+      "eval_mean_accuracy": 0.8064537332305951,
+      "eval_mean_iou": 0.7677773473066869,
+      "eval_overall_accuracy": 0.9800321934586864,
+      "eval_runtime": 5.4316,
+      "eval_samples_per_second": 21.725,
+      "eval_steps_per_second": 1.473,
+      "step": 2920
+    },
+    {
+      "epoch": 171.82,
+      "learning_rate": 7.414860681114552e-06,
+      "loss": 0.0216,
+      "step": 2921
+    },
+    {
+      "epoch": 171.88,
+      "learning_rate": 7.399380804953561e-06,
+      "loss": 0.0216,
+      "step": 2922
+    },
+    {
+      "epoch": 171.94,
+      "learning_rate": 7.38390092879257e-06,
+      "loss": 0.0218,
+      "step": 2923
+    },
+    {
+      "epoch": 172.0,
+      "learning_rate": 7.3684210526315784e-06,
+      "loss": 0.0226,
+      "step": 2924
+    },
+    {
+      "epoch": 172.06,
+      "learning_rate": 7.3529411764705884e-06,
+      "loss": 0.0215,
+      "step": 2925
+    },
+    {
+      "epoch": 172.12,
+      "learning_rate": 7.337461300309598e-06,
+      "loss": 0.0241,
+      "step": 2926
+    },
+    {
+      "epoch": 172.18,
+      "learning_rate": 7.321981424148607e-06,
+      "loss": 0.0216,
+      "step": 2927
+    },
+    {
+      "epoch": 172.24,
+      "learning_rate": 7.306501547987617e-06,
+      "loss": 0.0217,
+      "step": 2928
+    },
+    {
+      "epoch": 172.29,
+      "learning_rate": 7.291021671826626e-06,
+      "loss": 0.0201,
+      "step": 2929
+    },
+    {
+      "epoch": 172.35,
+      "learning_rate": 7.275541795665634e-06,
+      "loss": 0.0209,
+      "step": 2930
+    },
+    {
+      "epoch": 172.41,
+      "learning_rate": 7.260061919504644e-06,
+      "loss": 0.0198,
+      "step": 2931
+    },
+    {
+      "epoch": 172.47,
+      "learning_rate": 7.244582043343653e-06,
+      "loss": 0.021,
+      "step": 2932
+    },
+    {
+      "epoch": 172.53,
+      "learning_rate": 7.2291021671826625e-06,
+      "loss": 0.0216,
+      "step": 2933
+    },
+    {
+      "epoch": 172.59,
+      "learning_rate": 7.2136222910216725e-06,
+      "loss": 0.0201,
+      "step": 2934
+    },
+    {
+      "epoch": 172.65,
+      "learning_rate": 7.198142414860682e-06,
+      "loss": 0.0212,
+      "step": 2935
+    },
+    {
+      "epoch": 172.71,
+      "learning_rate": 7.18266253869969e-06,
+      "loss": 0.0242,
+      "step": 2936
+    },
+    {
+      "epoch": 172.76,
+      "learning_rate": 7.1671826625387e-06,
+      "loss": 0.022,
+      "step": 2937
+    },
+    {
+      "epoch": 172.82,
+      "learning_rate": 7.151702786377709e-06,
+      "loss": 0.0204,
+      "step": 2938
+    },
+    {
+      "epoch": 172.88,
+      "learning_rate": 7.136222910216718e-06,
+      "loss": 0.0236,
+      "step": 2939
+    },
+    {
+      "epoch": 172.94,
+      "learning_rate": 7.120743034055728e-06,
+      "loss": 0.0234,
+      "step": 2940
+    },
+    {
+      "epoch": 172.94,
+      "eval_accuracy_NAT": 0.9938183698820017,
+      "eval_accuracy_REST": 0.3300813008130081,
+      "eval_accuracy_concrete": 0.9575693806383951,
+      "eval_accuracy_concrete tiles": 0.934291774687721,
+      "eval_accuracy_dark bricks": 0.8639817774630094,
+      "eval_accuracy_grass": 0.8945135821533873,
+      "eval_accuracy_red bricks": 0.9314037785588752,
+      "eval_accuracy_road": 0.9801320114728456,
+      "eval_accuracy_rough concrete": 0.6258945649777439,
+      "eval_accuracy_rough red sidewalk": 0.7919069928334549,
+      "eval_accuracy_speedway bricks": 0.9890096670512101,
+      "eval_accuracy_steel": 0.9267643575044856,
+      "eval_accuracy_tiles": 0.25296131358114454,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9837181551881644,
+      "eval_iou_REST": 0.3232484076433121,
+      "eval_iou_concrete": 0.8927789645852301,
+      "eval_iou_concrete tiles": 0.8738454755869062,
+      "eval_iou_dark bricks": 0.8452700867985655,
+      "eval_iou_grass": 0.8399657384231678,
+      "eval_iou_red bricks": 0.9007807935411909,
+      "eval_iou_road": 0.9615253346027678,
+      "eval_iou_rough concrete": 0.619369409211791,
+      "eval_iou_rough red sidewalk": 0.6259034282725501,
+      "eval_iou_speedway bricks": 0.9794250345942486,
+      "eval_iou_steel": 0.8671255585145559,
+      "eval_iou_tiles": 0.24743673324245838,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08140265941619873,
+      "eval_mean_accuracy": 0.8055637593551755,
+      "eval_mean_iou": 0.7661840861696084,
+      "eval_overall_accuracy": 0.9800936165502516,
+      "eval_runtime": 5.3597,
+      "eval_samples_per_second": 22.016,
+      "eval_steps_per_second": 1.493,
+      "step": 2940
+    },
+    {
+      "epoch": 173.0,
+      "learning_rate": 7.1052631578947375e-06,
+      "loss": 0.0192,
+      "step": 2941
+    },
+    {
+      "epoch": 173.06,
+      "learning_rate": 7.089783281733746e-06,
+      "loss": 0.0183,
+      "step": 2942
+    },
+    {
+      "epoch": 173.12,
+      "learning_rate": 7.074303405572756e-06,
+      "loss": 0.0209,
+      "step": 2943
+    },
+    {
+      "epoch": 173.18,
+      "learning_rate": 7.058823529411765e-06,
+      "loss": 0.022,
+      "step": 2944
+    },
+    {
+      "epoch": 173.24,
+      "learning_rate": 7.043343653250774e-06,
+      "loss": 0.0209,
+      "step": 2945
+    },
+    {
+      "epoch": 173.29,
+      "learning_rate": 7.027863777089784e-06,
+      "loss": 0.0192,
+      "step": 2946
+    },
+    {
+      "epoch": 173.35,
+      "learning_rate": 7.012383900928793e-06,
+      "loss": 0.0217,
+      "step": 2947
+    },
+    {
+      "epoch": 173.41,
+      "learning_rate": 6.9969040247678016e-06,
+      "loss": 0.0221,
+      "step": 2948
+    },
+    {
+      "epoch": 173.47,
+      "learning_rate": 6.9814241486068116e-06,
+      "loss": 0.0249,
+      "step": 2949
+    },
+    {
+      "epoch": 173.53,
+      "learning_rate": 6.965944272445821e-06,
+      "loss": 0.0212,
+      "step": 2950
+    },
+    {
+      "epoch": 173.59,
+      "learning_rate": 6.95046439628483e-06,
+      "loss": 0.0233,
+      "step": 2951
+    },
+    {
+      "epoch": 173.65,
+      "learning_rate": 6.93498452012384e-06,
+      "loss": 0.0178,
+      "step": 2952
+    },
+    {
+      "epoch": 173.71,
+      "learning_rate": 6.919504643962848e-06,
+      "loss": 0.0181,
+      "step": 2953
+    },
+    {
+      "epoch": 173.76,
+      "learning_rate": 6.904024767801857e-06,
+      "loss": 0.0214,
+      "step": 2954
+    },
+    {
+      "epoch": 173.82,
+      "learning_rate": 6.888544891640867e-06,
+      "loss": 0.0229,
+      "step": 2955
+    },
+    {
+      "epoch": 173.88,
+      "learning_rate": 6.8730650154798765e-06,
+      "loss": 0.0242,
+      "step": 2956
+    },
+    {
+      "epoch": 173.94,
+      "learning_rate": 6.857585139318886e-06,
+      "loss": 0.0186,
+      "step": 2957
+    },
+    {
+      "epoch": 174.0,
+      "learning_rate": 6.842105263157896e-06,
+      "loss": 0.0215,
+      "step": 2958
+    },
+    {
+      "epoch": 174.06,
+      "learning_rate": 6.826625386996904e-06,
+      "loss": 0.0231,
+      "step": 2959
+    },
+    {
+      "epoch": 174.12,
+      "learning_rate": 6.811145510835913e-06,
+      "loss": 0.0213,
+      "step": 2960
+    },
+    {
+      "epoch": 174.12,
+      "eval_accuracy_NAT": 0.9934057831898715,
+      "eval_accuracy_REST": 0.35284552845528455,
+      "eval_accuracy_concrete": 0.9570136628604713,
+      "eval_accuracy_concrete tiles": 0.9311807683242989,
+      "eval_accuracy_dark bricks": 0.8599381735352104,
+      "eval_accuracy_grass": 0.9020313617731551,
+      "eval_accuracy_red bricks": 0.9336555360281195,
+      "eval_accuracy_road": 0.9818009930520559,
+      "eval_accuracy_rough concrete": 0.6259481941854022,
+      "eval_accuracy_rough red sidewalk": 0.7750847878236413,
+      "eval_accuracy_speedway bricks": 0.9894387654031571,
+      "eval_accuracy_steel": 0.9217715293847176,
+      "eval_accuracy_tiles": 0.24747796055621193,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.983709087337114,
+      "eval_iou_REST": 0.34609250398724084,
+      "eval_iou_concrete": 0.8921336865339017,
+      "eval_iou_concrete tiles": 0.8754514635173163,
+      "eval_iou_dark bricks": 0.8432010135135135,
+      "eval_iou_grass": 0.8416207139489673,
+      "eval_iou_red bricks": 0.9034863945578231,
+      "eval_iou_road": 0.9609152704783583,
+      "eval_iou_rough concrete": 0.6192435567921903,
+      "eval_iou_rough red sidewalk": 0.6199289069595395,
+      "eval_iou_speedway bricks": 0.9795760136241345,
+      "eval_iou_steel": 0.8671924957682904,
+      "eval_iou_tiles": 0.24399641577060932,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08147281408309937,
+      "eval_mean_accuracy": 0.8055071572747383,
+      "eval_mean_iou": 0.7674267325222307,
+      "eval_overall_accuracy": 0.9800656852075609,
+      "eval_runtime": 5.4672,
+      "eval_samples_per_second": 21.583,
+      "eval_steps_per_second": 1.463,
+      "step": 2960
     }
   ],
   "max_steps": 3400,
   "num_train_epochs": 200,
-  "total_flos": 1.6677092855513088e+19,
+  "total_flos": 1.8698304612005315e+19,
   "trial_name": null,
   "trial_params": null
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5882d92c0409ce4cb6f64ab74a4554784137589744c85456376d8d79ddbf107b
+oid sha256:b5d6df664b36ccc3d54e1cf8e501a508daed6e77a51faf1ec4337b95da2566c3
 size 338830281

runs/Dec09_04-08-30_robolidar/events.out.tfevents.1702116518.robolidar.2934636.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2281832a571232c7bb3364088cff8ba87d60b9e088469014866cc4fe59c529c0
-size 759809
+oid sha256:4183b0ea15fa8d4c417a54df163174009f920799f92ac59d6d9a48b12145f459
+size 780785