Training in progress, step 2080

{checkpoint-1760 → checkpoint-2080}/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e84dcf39c02b88e16f530ff284c0bc4216e54b335b8b5ea41963d3ee5c107a66
+oid sha256:b5b7ad2f316d805e27c5b10d0a3c94bd63ddf2cbdb13ca5af5e62f70d50d0600
 size 29861573

{checkpoint-1760 → checkpoint-2080}/pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1fb9acddb8f35176a9002507f0a2e87e4968d076f827099d52a7f3fe2b46c764
+oid sha256:1ee07344711c4a458a60014f92aa7b147657559d8f5abcee058f54728be5f80c
 size 14944077

{checkpoint-1760 → checkpoint-2080}/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:48c744284fbbac2949716ed104a6b5dbf1c7a9350e6cc80a22007dd6db447a5e
+oid sha256:619ccb22e51b388897af62c7c75338d89654df4cb133c535868532a317ea2dfe
 size 14575

{checkpoint-1760 → checkpoint-2080}/scheduler.pt

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

{checkpoint-1760 → checkpoint-2080}/trainer_state.json

@@ -1,8 +1,8 @@
 {
   "best_metric": 0.09530654549598694,
   "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0/checkpoint-1440",
-  "epoch": 160.0,
-  "global_step": 1760,
+  "epoch": 189.0909090909091,
+  "global_step": 2080,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -13998,11 +13998,2555 @@
       "eval_samples_per_second": 10.809,
       "eval_steps_per_second": 0.601,
       "step": 1760
+    },
+    {
+      "epoch": 160.09,
+      "learning_rate": 7.155980861244019e-05,
+      "loss": 0.0413,
+      "step": 1761
+    },
+    {
+      "epoch": 160.18,
+      "learning_rate": 7.154066985645933e-05,
+      "loss": 0.0436,
+      "step": 1762
+    },
+    {
+      "epoch": 160.27,
+      "learning_rate": 7.152153110047846e-05,
+      "loss": 0.0454,
+      "step": 1763
+    },
+    {
+      "epoch": 160.36,
+      "learning_rate": 7.150239234449761e-05,
+      "loss": 0.0448,
+      "step": 1764
+    },
+    {
+      "epoch": 160.45,
+      "learning_rate": 7.148325358851676e-05,
+      "loss": 0.0479,
+      "step": 1765
+    },
+    {
+      "epoch": 160.55,
+      "learning_rate": 7.146411483253589e-05,
+      "loss": 0.0449,
+      "step": 1766
+    },
+    {
+      "epoch": 160.64,
+      "learning_rate": 7.144497607655502e-05,
+      "loss": 0.0491,
+      "step": 1767
+    },
+    {
+      "epoch": 160.73,
+      "learning_rate": 7.142583732057417e-05,
+      "loss": 0.048,
+      "step": 1768
+    },
+    {
+      "epoch": 160.82,
+      "learning_rate": 7.14066985645933e-05,
+      "loss": 0.0407,
+      "step": 1769
+    },
+    {
+      "epoch": 160.91,
+      "learning_rate": 7.138755980861244e-05,
+      "loss": 0.0496,
+      "step": 1770
+    },
+    {
+      "epoch": 161.0,
+      "learning_rate": 7.136842105263159e-05,
+      "loss": 0.0498,
+      "step": 1771
+    },
+    {
+      "epoch": 161.09,
+      "learning_rate": 7.134928229665072e-05,
+      "loss": 0.0423,
+      "step": 1772
+    },
+    {
+      "epoch": 161.18,
+      "learning_rate": 7.133014354066985e-05,
+      "loss": 0.0429,
+      "step": 1773
+    },
+    {
+      "epoch": 161.27,
+      "learning_rate": 7.1311004784689e-05,
+      "loss": 0.0469,
+      "step": 1774
+    },
+    {
+      "epoch": 161.36,
+      "learning_rate": 7.129186602870813e-05,
+      "loss": 0.039,
+      "step": 1775
+    },
+    {
+      "epoch": 161.45,
+      "learning_rate": 7.127272727272728e-05,
+      "loss": 0.047,
+      "step": 1776
+    },
+    {
+      "epoch": 161.55,
+      "learning_rate": 7.125358851674642e-05,
+      "loss": 0.041,
+      "step": 1777
+    },
+    {
+      "epoch": 161.64,
+      "learning_rate": 7.123444976076555e-05,
+      "loss": 0.0396,
+      "step": 1778
+    },
+    {
+      "epoch": 161.73,
+      "learning_rate": 7.121531100478468e-05,
+      "loss": 0.043,
+      "step": 1779
+    },
+    {
+      "epoch": 161.82,
+      "learning_rate": 7.119617224880383e-05,
+      "loss": 0.0569,
+      "step": 1780
+    },
+    {
+      "epoch": 161.82,
+      "eval_accuracy_NAT": 0.9930390034605593,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9492856453066655,
+      "eval_accuracy_concrete tiles": 0.9792698587415153,
+      "eval_accuracy_dark bricks": 0.8537777606712598,
+      "eval_accuracy_grass": 0.8787489034206286,
+      "eval_accuracy_red bricks": 0.8966619318181818,
+      "eval_accuracy_road": 0.8813221973259555,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.44752295095525596,
+      "eval_accuracy_speedway bricks": 0.9856455317048564,
+      "eval_accuracy_steel": 0.9042114636237715,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787156413862954,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8746314288327076,
+      "eval_iou_concrete tiles": 0.9403957954078337,
+      "eval_iou_dark bricks": 0.813839154681538,
+      "eval_iou_grass": 0.8278320135653491,
+      "eval_iou_red bricks": 0.8456128600133959,
+      "eval_iou_road": 0.8728029995485349,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3010180429098596,
+      "eval_iou_speedway bricks": 0.9744170522760448,
+      "eval_iou_steel": 0.8304181242711978,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12074758112430573,
+      "eval_mean_accuracy": 0.674575788232973,
+      "eval_mean_iou": 0.635360239453289,
+      "eval_overall_accuracy": 0.9679258134629991,
+      "eval_runtime": 3.3537,
+      "eval_samples_per_second": 10.734,
+      "eval_steps_per_second": 0.596,
+      "step": 1780
+    },
+    {
+      "epoch": 161.91,
+      "learning_rate": 7.117703349282298e-05,
+      "loss": 0.0444,
+      "step": 1781
+    },
+    {
+      "epoch": 162.0,
+      "learning_rate": 7.115789473684211e-05,
+      "loss": 0.0471,
+      "step": 1782
+    },
+    {
+      "epoch": 162.09,
+      "learning_rate": 7.113875598086125e-05,
+      "loss": 0.0541,
+      "step": 1783
+    },
+    {
+      "epoch": 162.18,
+      "learning_rate": 7.111961722488039e-05,
+      "loss": 0.0454,
+      "step": 1784
+    },
+    {
+      "epoch": 162.27,
+      "learning_rate": 7.110047846889952e-05,
+      "loss": 0.0471,
+      "step": 1785
+    },
+    {
+      "epoch": 162.36,
+      "learning_rate": 7.108133971291866e-05,
+      "loss": 0.0406,
+      "step": 1786
+    },
+    {
+      "epoch": 162.45,
+      "learning_rate": 7.106220095693781e-05,
+      "loss": 0.0432,
+      "step": 1787
+    },
+    {
+      "epoch": 162.55,
+      "learning_rate": 7.104306220095694e-05,
+      "loss": 0.0484,
+      "step": 1788
+    },
+    {
+      "epoch": 162.64,
+      "learning_rate": 7.102392344497609e-05,
+      "loss": 0.0418,
+      "step": 1789
+    },
+    {
+      "epoch": 162.73,
+      "learning_rate": 7.100478468899522e-05,
+      "loss": 0.0493,
+      "step": 1790
+    },
+    {
+      "epoch": 162.82,
+      "learning_rate": 7.098564593301435e-05,
+      "loss": 0.0459,
+      "step": 1791
+    },
+    {
+      "epoch": 162.91,
+      "learning_rate": 7.09665071770335e-05,
+      "loss": 0.0475,
+      "step": 1792
+    },
+    {
+      "epoch": 163.0,
+      "learning_rate": 7.094736842105264e-05,
+      "loss": 0.0628,
+      "step": 1793
+    },
+    {
+      "epoch": 163.09,
+      "learning_rate": 7.092822966507177e-05,
+      "loss": 0.0439,
+      "step": 1794
+    },
+    {
+      "epoch": 163.18,
+      "learning_rate": 7.090909090909092e-05,
+      "loss": 0.0467,
+      "step": 1795
+    },
+    {
+      "epoch": 163.27,
+      "learning_rate": 7.088995215311005e-05,
+      "loss": 0.0547,
+      "step": 1796
+    },
+    {
+      "epoch": 163.36,
+      "learning_rate": 7.087081339712918e-05,
+      "loss": 0.0465,
+      "step": 1797
+    },
+    {
+      "epoch": 163.45,
+      "learning_rate": 7.085167464114833e-05,
+      "loss": 0.0454,
+      "step": 1798
+    },
+    {
+      "epoch": 163.55,
+      "learning_rate": 7.083253588516747e-05,
+      "loss": 0.0468,
+      "step": 1799
+    },
+    {
+      "epoch": 163.64,
+      "learning_rate": 7.08133971291866e-05,
+      "loss": 0.0414,
+      "step": 1800
+    },
+    {
+      "epoch": 163.64,
+      "eval_accuracy_NAT": 0.9933093265707855,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9372478432550754,
+      "eval_accuracy_concrete tiles": 0.9910719745612426,
+      "eval_accuracy_dark bricks": 0.8719833724644933,
+      "eval_accuracy_grass": 0.8813142875117522,
+      "eval_accuracy_red bricks": 0.9128196022727273,
+      "eval_accuracy_road": 0.8683050640457468,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48851763019325667,
+      "eval_accuracy_speedway bricks": 0.9872536135076457,
+      "eval_accuracy_steel": 0.9058109623375766,
+      "eval_accuracy_tiles": 0.1494315847144227,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9779009757340675,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8811723374366657,
+      "eval_iou_concrete tiles": 0.9083622912229571,
+      "eval_iou_dark bricks": 0.8177814677110782,
+      "eval_iou_grass": 0.8256598658166423,
+      "eval_iou_red bricks": 0.8482098663586867,
+      "eval_iou_road": 0.8607135964980624,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3222697099208875,
+      "eval_iou_speedway bricks": 0.9744303817069401,
+      "eval_iou_steel": 0.8261010143543548,
+      "eval_iou_tiles": 0.14934976043805612,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14603829383850098,
+      "eval_mean_accuracy": 0.6913127124180558,
+      "eval_mean_iou": 0.6455347128614153,
+      "eval_overall_accuracy": 0.9668665991889106,
+      "eval_runtime": 3.9858,
+      "eval_samples_per_second": 9.032,
+      "eval_steps_per_second": 0.502,
+      "step": 1800
+    },
+    {
+      "epoch": 163.73,
+      "learning_rate": 7.079425837320575e-05,
+      "loss": 0.0432,
+      "step": 1801
+    },
+    {
+      "epoch": 163.82,
+      "learning_rate": 7.077511961722488e-05,
+      "loss": 0.0726,
+      "step": 1802
+    },
+    {
+      "epoch": 163.91,
+      "learning_rate": 7.075598086124402e-05,
+      "loss": 0.0472,
+      "step": 1803
+    },
+    {
+      "epoch": 164.0,
+      "learning_rate": 7.073684210526316e-05,
+      "loss": 0.0452,
+      "step": 1804
+    },
+    {
+      "epoch": 164.09,
+      "learning_rate": 7.07177033492823e-05,
+      "loss": 0.0485,
+      "step": 1805
+    },
+    {
+      "epoch": 164.18,
+      "learning_rate": 7.069856459330144e-05,
+      "loss": 0.0502,
+      "step": 1806
+    },
+    {
+      "epoch": 164.27,
+      "learning_rate": 7.067942583732058e-05,
+      "loss": 0.0617,
+      "step": 1807
+    },
+    {
+      "epoch": 164.36,
+      "learning_rate": 7.066028708133972e-05,
+      "loss": 0.0436,
+      "step": 1808
+    },
+    {
+      "epoch": 164.45,
+      "learning_rate": 7.064114832535885e-05,
+      "loss": 0.0502,
+      "step": 1809
+    },
+    {
+      "epoch": 164.55,
+      "learning_rate": 7.0622009569378e-05,
+      "loss": 0.0512,
+      "step": 1810
+    },
+    {
+      "epoch": 164.64,
+      "learning_rate": 7.060287081339714e-05,
+      "loss": 0.0466,
+      "step": 1811
+    },
+    {
+      "epoch": 164.73,
+      "learning_rate": 7.058373205741627e-05,
+      "loss": 0.0485,
+      "step": 1812
+    },
+    {
+      "epoch": 164.82,
+      "learning_rate": 7.056459330143542e-05,
+      "loss": 0.0444,
+      "step": 1813
+    },
+    {
+      "epoch": 164.91,
+      "learning_rate": 7.054545454545455e-05,
+      "loss": 0.0472,
+      "step": 1814
+    },
+    {
+      "epoch": 165.0,
+      "learning_rate": 7.052631578947368e-05,
+      "loss": 0.0425,
+      "step": 1815
+    },
+    {
+      "epoch": 165.09,
+      "learning_rate": 7.050717703349283e-05,
+      "loss": 0.0481,
+      "step": 1816
+    },
+    {
+      "epoch": 165.18,
+      "learning_rate": 7.048803827751197e-05,
+      "loss": 0.0431,
+      "step": 1817
+    },
+    {
+      "epoch": 165.27,
+      "learning_rate": 7.04688995215311e-05,
+      "loss": 0.05,
+      "step": 1818
+    },
+    {
+      "epoch": 165.36,
+      "learning_rate": 7.044976076555025e-05,
+      "loss": 0.0479,
+      "step": 1819
+    },
+    {
+      "epoch": 165.45,
+      "learning_rate": 7.043062200956938e-05,
+      "loss": 0.0447,
+      "step": 1820
+    },
+    {
+      "epoch": 165.45,
+      "eval_accuracy_NAT": 0.9922640301576132,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9439326795141423,
+      "eval_accuracy_concrete tiles": 0.985598972665566,
+      "eval_accuracy_dark bricks": 0.8512374427466225,
+      "eval_accuracy_grass": 0.8828311494947871,
+      "eval_accuracy_red bricks": 0.9451349431818182,
+      "eval_accuracy_road": 0.915026475627791,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5713340501199239,
+      "eval_accuracy_speedway bricks": 0.9875836775774867,
+      "eval_accuracy_steel": 0.8870539542246554,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9784177243160237,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8773594667271714,
+      "eval_iou_concrete tiles": 0.9301419667590027,
+      "eval_iou_dark bricks": 0.8094872076424728,
+      "eval_iou_grass": 0.8214994243490591,
+      "eval_iou_red bricks": 0.8043215472952554,
+      "eval_iou_road": 0.9049644694576061,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3752172653534183,
+      "eval_iou_speedway bricks": 0.9742733920870055,
+      "eval_iou_steel": 0.8290106333795654,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10441258549690247,
+      "eval_mean_accuracy": 0.6893844134854159,
+      "eval_mean_iou": 0.6388225459512754,
+      "eval_overall_accuracy": 0.97075441148546,
+      "eval_runtime": 3.4346,
+      "eval_samples_per_second": 10.482,
+      "eval_steps_per_second": 0.582,
+      "step": 1820
+    },
+    {
+      "epoch": 165.55,
+      "learning_rate": 7.041148325358851e-05,
+      "loss": 0.0471,
+      "step": 1821
+    },
+    {
+      "epoch": 165.64,
+      "learning_rate": 7.039234449760766e-05,
+      "loss": 0.042,
+      "step": 1822
+    },
+    {
+      "epoch": 165.73,
+      "learning_rate": 7.03732057416268e-05,
+      "loss": 0.0415,
+      "step": 1823
+    },
+    {
+      "epoch": 165.82,
+      "learning_rate": 7.035406698564594e-05,
+      "loss": 0.0439,
+      "step": 1824
+    },
+    {
+      "epoch": 165.91,
+      "learning_rate": 7.033492822966508e-05,
+      "loss": 0.0468,
+      "step": 1825
+    },
+    {
+      "epoch": 166.0,
+      "learning_rate": 7.031578947368421e-05,
+      "loss": 0.0486,
+      "step": 1826
+    },
+    {
+      "epoch": 166.09,
+      "learning_rate": 7.029665071770335e-05,
+      "loss": 0.0461,
+      "step": 1827
+    },
+    {
+      "epoch": 166.18,
+      "learning_rate": 7.027751196172249e-05,
+      "loss": 0.0411,
+      "step": 1828
+    },
+    {
+      "epoch": 166.27,
+      "learning_rate": 7.025837320574164e-05,
+      "loss": 0.0433,
+      "step": 1829
+    },
+    {
+      "epoch": 166.36,
+      "learning_rate": 7.023923444976077e-05,
+      "loss": 0.0479,
+      "step": 1830
+    },
+    {
+      "epoch": 166.45,
+      "learning_rate": 7.022009569377991e-05,
+      "loss": 0.0495,
+      "step": 1831
+    },
+    {
+      "epoch": 166.55,
+      "learning_rate": 7.020095693779905e-05,
+      "loss": 0.0442,
+      "step": 1832
+    },
+    {
+      "epoch": 166.64,
+      "learning_rate": 7.018181818181818e-05,
+      "loss": 0.0466,
+      "step": 1833
+    },
+    {
+      "epoch": 166.73,
+      "learning_rate": 7.016267942583732e-05,
+      "loss": 0.0452,
+      "step": 1834
+    },
+    {
+      "epoch": 166.82,
+      "learning_rate": 7.014354066985647e-05,
+      "loss": 0.0464,
+      "step": 1835
+    },
+    {
+      "epoch": 166.91,
+      "learning_rate": 7.01244019138756e-05,
+      "loss": 0.0567,
+      "step": 1836
+    },
+    {
+      "epoch": 167.0,
+      "learning_rate": 7.010526315789473e-05,
+      "loss": 0.0468,
+      "step": 1837
+    },
+    {
+      "epoch": 167.09,
+      "learning_rate": 7.008612440191388e-05,
+      "loss": 0.0462,
+      "step": 1838
+    },
+    {
+      "epoch": 167.18,
+      "learning_rate": 7.006698564593301e-05,
+      "loss": 0.0456,
+      "step": 1839
+    },
+    {
+      "epoch": 167.27,
+      "learning_rate": 7.004784688995216e-05,
+      "loss": 0.0485,
+      "step": 1840
+    },
+    {
+      "epoch": 167.27,
+      "eval_accuracy_NAT": 0.992405896855147,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.944899727456114,
+      "eval_accuracy_concrete tiles": 0.982449703418333,
+      "eval_accuracy_dark bricks": 0.8715599861437204,
+      "eval_accuracy_grass": 0.888444237846755,
+      "eval_accuracy_red bricks": 0.9263139204545454,
+      "eval_accuracy_road": 0.8881457628717259,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5142668100239848,
+      "eval_accuracy_speedway bricks": 0.9870575059707203,
+      "eval_accuracy_steel": 0.8977475100587032,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787805300386256,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8791844711567262,
+      "eval_iou_concrete tiles": 0.9453924914675768,
+      "eval_iou_dark bricks": 0.805463664496852,
+      "eval_iou_grass": 0.8205844185190685,
+      "eval_iou_red bricks": 0.845954272742014,
+      "eval_iou_road": 0.8793363120037545,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3444620895963364,
+      "eval_iou_speedway bricks": 0.9747215515412685,
+      "eval_iou_steel": 0.8303858091773624,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.12035813927650452,
+      "eval_mean_accuracy": 0.6840993123922884,
+      "eval_mean_iou": 0.6387896623645835,
+      "eval_overall_accuracy": 0.9685901006062826,
+      "eval_runtime": 3.5728,
+      "eval_samples_per_second": 10.076,
+      "eval_steps_per_second": 0.56,
+      "step": 1840
+    },
+    {
+      "epoch": 167.36,
+      "learning_rate": 7.00287081339713e-05,
+      "loss": 0.0456,
+      "step": 1841
+    },
+    {
+      "epoch": 167.45,
+      "learning_rate": 7.000956937799043e-05,
+      "loss": 0.0413,
+      "step": 1842
+    },
+    {
+      "epoch": 167.55,
+      "learning_rate": 6.999043062200957e-05,
+      "loss": 0.0451,
+      "step": 1843
+    },
+    {
+      "epoch": 167.64,
+      "learning_rate": 6.997129186602871e-05,
+      "loss": 0.0407,
+      "step": 1844
+    },
+    {
+      "epoch": 167.73,
+      "learning_rate": 6.995215311004784e-05,
+      "loss": 0.0418,
+      "step": 1845
+    },
+    {
+      "epoch": 167.82,
+      "learning_rate": 6.993301435406699e-05,
+      "loss": 0.0421,
+      "step": 1846
+    },
+    {
+      "epoch": 167.91,
+      "learning_rate": 6.991387559808614e-05,
+      "loss": 0.0467,
+      "step": 1847
+    },
+    {
+      "epoch": 168.0,
+      "learning_rate": 6.989473684210527e-05,
+      "loss": 0.047,
+      "step": 1848
+    },
+    {
+      "epoch": 168.09,
+      "learning_rate": 6.98755980861244e-05,
+      "loss": 0.0562,
+      "step": 1849
+    },
+    {
+      "epoch": 168.18,
+      "learning_rate": 6.985645933014354e-05,
+      "loss": 0.0459,
+      "step": 1850
+    },
+    {
+      "epoch": 168.27,
+      "learning_rate": 6.983732057416268e-05,
+      "loss": 0.042,
+      "step": 1851
+    },
+    {
+      "epoch": 168.36,
+      "learning_rate": 6.981818181818182e-05,
+      "loss": 0.0459,
+      "step": 1852
+    },
+    {
+      "epoch": 168.45,
+      "learning_rate": 6.979904306220097e-05,
+      "loss": 0.0459,
+      "step": 1853
+    },
+    {
+      "epoch": 168.55,
+      "learning_rate": 6.97799043062201e-05,
+      "loss": 0.0546,
+      "step": 1854
+    },
+    {
+      "epoch": 168.64,
+      "learning_rate": 6.976076555023923e-05,
+      "loss": 0.0454,
+      "step": 1855
+    },
+    {
+      "epoch": 168.73,
+      "learning_rate": 6.974162679425838e-05,
+      "loss": 0.043,
+      "step": 1856
+    },
+    {
+      "epoch": 168.82,
+      "learning_rate": 6.972248803827751e-05,
+      "loss": 0.0502,
+      "step": 1857
+    },
+    {
+      "epoch": 168.91,
+      "learning_rate": 6.970334928229666e-05,
+      "loss": 0.0527,
+      "step": 1858
+    },
+    {
+      "epoch": 169.0,
+      "learning_rate": 6.96842105263158e-05,
+      "loss": 0.0396,
+      "step": 1859
+    },
+    {
+      "epoch": 169.09,
+      "learning_rate": 6.966507177033493e-05,
+      "loss": 0.0461,
+      "step": 1860
+    },
+    {
+      "epoch": 169.09,
+      "eval_accuracy_NAT": 0.9925281774787676,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9475098180179407,
+      "eval_accuracy_concrete tiles": 0.9845594080596832,
+      "eval_accuracy_dark bricks": 0.844771178938455,
+      "eval_accuracy_grass": 0.893718304050441,
+      "eval_accuracy_red bricks": 0.9366122159090909,
+      "eval_accuracy_road": 0.8708419842149877,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.6013288120640697,
+      "eval_accuracy_speedway bricks": 0.9849799728949213,
+      "eval_accuracy_steel": 0.9035106523316404,
+      "eval_accuracy_tiles": 0.011505273250239693,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9782664537706919,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.885822733382822,
+      "eval_iou_concrete tiles": 0.9331188965197484,
+      "eval_iou_dark bricks": 0.8058452048758995,
+      "eval_iou_grass": 0.8221804946432935,
+      "eval_iou_red bricks": 0.8450817045818648,
+      "eval_iou_road": 0.8638630725900478,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.39869125737995575,
+      "eval_iou_speedway bricks": 0.9745276051937011,
+      "eval_iou_steel": 0.8296363030706801,
+      "eval_iou_tiles": 0.011505273250239693,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.13130919635295868,
+      "eval_mean_accuracy": 0.690143522862326,
+      "eval_mean_iou": 0.6421953076353033,
+      "eval_overall_accuracy": 0.9675146738688151,
+      "eval_runtime": 3.8455,
+      "eval_samples_per_second": 9.362,
+      "eval_steps_per_second": 0.52,
+      "step": 1860
+    },
+    {
+      "epoch": 169.18,
+      "learning_rate": 6.964593301435406e-05,
+      "loss": 0.0413,
+      "step": 1861
+    },
+    {
+      "epoch": 169.27,
+      "learning_rate": 6.962679425837321e-05,
+      "loss": 0.0415,
+      "step": 1862
+    },
+    {
+      "epoch": 169.36,
+      "learning_rate": 6.960765550239234e-05,
+      "loss": 0.0516,
+      "step": 1863
+    },
+    {
+      "epoch": 169.45,
+      "learning_rate": 6.958851674641149e-05,
+      "loss": 0.0424,
+      "step": 1864
+    },
+    {
+      "epoch": 169.55,
+      "learning_rate": 6.956937799043063e-05,
+      "loss": 0.039,
+      "step": 1865
+    },
+    {
+      "epoch": 169.64,
+      "learning_rate": 6.955023923444977e-05,
+      "loss": 0.0479,
+      "step": 1866
+    },
+    {
+      "epoch": 169.73,
+      "learning_rate": 6.95311004784689e-05,
+      "loss": 0.0488,
+      "step": 1867
+    },
+    {
+      "epoch": 169.82,
+      "learning_rate": 6.951196172248804e-05,
+      "loss": 0.0445,
+      "step": 1868
+    },
+    {
+      "epoch": 169.91,
+      "learning_rate": 6.949282296650717e-05,
+      "loss": 0.0417,
+      "step": 1869
+    },
+    {
+      "epoch": 170.0,
+      "learning_rate": 6.947368421052632e-05,
+      "loss": 0.0435,
+      "step": 1870
+    },
+    {
+      "epoch": 170.09,
+      "learning_rate": 6.945454545454547e-05,
+      "loss": 0.0461,
+      "step": 1871
+    },
+    {
+      "epoch": 170.18,
+      "learning_rate": 6.94354066985646e-05,
+      "loss": 0.0425,
+      "step": 1872
+    },
+    {
+      "epoch": 170.27,
+      "learning_rate": 6.941626794258373e-05,
+      "loss": 0.05,
+      "step": 1873
+    },
+    {
+      "epoch": 170.36,
+      "learning_rate": 6.939712918660288e-05,
+      "loss": 0.0407,
+      "step": 1874
+    },
+    {
+      "epoch": 170.45,
+      "learning_rate": 6.937799043062201e-05,
+      "loss": 0.044,
+      "step": 1875
+    },
+    {
+      "epoch": 170.55,
+      "learning_rate": 6.935885167464115e-05,
+      "loss": 0.038,
+      "step": 1876
+    },
+    {
+      "epoch": 170.64,
+      "learning_rate": 6.93397129186603e-05,
+      "loss": 0.0429,
+      "step": 1877
+    },
+    {
+      "epoch": 170.73,
+      "learning_rate": 6.932057416267943e-05,
+      "loss": 0.0451,
+      "step": 1878
+    },
+    {
+      "epoch": 170.82,
+      "learning_rate": 6.930143540669856e-05,
+      "loss": 0.0394,
+      "step": 1879
+    },
+    {
+      "epoch": 170.91,
+      "learning_rate": 6.928229665071771e-05,
+      "loss": 0.0443,
+      "step": 1880
+    },
+    {
+      "epoch": 170.91,
+      "eval_accuracy_NAT": 0.9920214098726517,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.943066226018284,
+      "eval_accuracy_concrete tiles": 0.9879838561731792,
+      "eval_accuracy_dark bricks": 0.8724837381163157,
+      "eval_accuracy_grass": 0.8903630333045572,
+      "eval_accuracy_red bricks": 0.9209872159090909,
+      "eval_accuracy_road": 0.9294063872829924,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.46177597662172964,
+      "eval_accuracy_speedway bricks": 0.9887367898946088,
+      "eval_accuracy_steel": 0.8923966097223138,
+      "eval_accuracy_tiles": 0.0,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9785171190347933,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8746502928894235,
+      "eval_iou_concrete tiles": 0.9393587022878572,
+      "eval_iou_dark bricks": 0.8114551637730446,
+      "eval_iou_grass": 0.8182545265328541,
+      "eval_iou_red bricks": 0.8130094043887147,
+      "eval_iou_road": 0.9230565069727,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3031564468254543,
+      "eval_iou_speedway bricks": 0.9741363884160983,
+      "eval_iou_steel": 0.8253231156353654,
+      "eval_iou_tiles": 0.0,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1050824448466301,
+      "eval_mean_accuracy": 0.6830170186858248,
+      "eval_mean_iou": 0.6354552051351003,
+      "eval_overall_accuracy": 0.9719311396280924,
+      "eval_runtime": 3.224,
+      "eval_samples_per_second": 11.166,
+      "eval_steps_per_second": 0.62,
+      "step": 1880
+    },
+    {
+      "epoch": 171.0,
+      "learning_rate": 6.926315789473684e-05,
+      "loss": 0.049,
+      "step": 1881
+    },
+    {
+      "epoch": 171.09,
+      "learning_rate": 6.924401913875599e-05,
+      "loss": 0.0426,
+      "step": 1882
+    },
+    {
+      "epoch": 171.18,
+      "learning_rate": 6.922488038277513e-05,
+      "loss": 0.0481,
+      "step": 1883
+    },
+    {
+      "epoch": 171.27,
+      "learning_rate": 6.920574162679426e-05,
+      "loss": 0.045,
+      "step": 1884
+    },
+    {
+      "epoch": 171.36,
+      "learning_rate": 6.91866028708134e-05,
+      "loss": 0.0374,
+      "step": 1885
+    },
+    {
+      "epoch": 171.45,
+      "learning_rate": 6.916746411483254e-05,
+      "loss": 0.0448,
+      "step": 1886
+    },
+    {
+      "epoch": 171.55,
+      "learning_rate": 6.914832535885167e-05,
+      "loss": 0.0444,
+      "step": 1887
+    },
+    {
+      "epoch": 171.64,
+      "learning_rate": 6.912918660287082e-05,
+      "loss": 0.0417,
+      "step": 1888
+    },
+    {
+      "epoch": 171.73,
+      "learning_rate": 6.911004784688996e-05,
+      "loss": 0.0443,
+      "step": 1889
+    },
+    {
+      "epoch": 171.82,
+      "learning_rate": 6.90909090909091e-05,
+      "loss": 0.0411,
+      "step": 1890
+    },
+    {
+      "epoch": 171.91,
+      "learning_rate": 6.907177033492823e-05,
+      "loss": 0.0447,
+      "step": 1891
+    },
+    {
+      "epoch": 172.0,
+      "learning_rate": 6.905263157894737e-05,
+      "loss": 0.0451,
+      "step": 1892
+    },
+    {
+      "epoch": 172.09,
+      "learning_rate": 6.90334928229665e-05,
+      "loss": 0.0388,
+      "step": 1893
+    },
+    {
+      "epoch": 172.18,
+      "learning_rate": 6.901435406698565e-05,
+      "loss": 0.0456,
+      "step": 1894
+    },
+    {
+      "epoch": 172.27,
+      "learning_rate": 6.89952153110048e-05,
+      "loss": 0.0444,
+      "step": 1895
+    },
+    {
+      "epoch": 172.36,
+      "learning_rate": 6.897607655502393e-05,
+      "loss": 0.0425,
+      "step": 1896
+    },
+    {
+      "epoch": 172.45,
+      "learning_rate": 6.895693779904306e-05,
+      "loss": 0.0542,
+      "step": 1897
+    },
+    {
+      "epoch": 172.55,
+      "learning_rate": 6.89377990430622e-05,
+      "loss": 0.0445,
+      "step": 1898
+    },
+    {
+      "epoch": 172.64,
+      "learning_rate": 6.891866028708134e-05,
+      "loss": 0.0402,
+      "step": 1899
+    },
+    {
+      "epoch": 172.73,
+      "learning_rate": 6.889952153110048e-05,
+      "loss": 0.0435,
+      "step": 1900
+    },
+    {
+      "epoch": 172.73,
+      "eval_accuracy_NAT": 0.992198743244569,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9460116314004893,
+      "eval_accuracy_concrete tiles": 0.9903381642512077,
+      "eval_accuracy_dark bricks": 0.8757553596859243,
+      "eval_accuracy_grass": 0.9039029487936753,
+      "eval_accuracy_red bricks": 0.9556107954545454,
+      "eval_accuracy_road": 0.9051303103918199,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.47864803021531166,
+      "eval_accuracy_speedway bricks": 0.9865753831337863,
+      "eval_accuracy_steel": 0.8964778048941363,
+      "eval_accuracy_tiles": 0.0007533214628133132,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787434839117086,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8756522548806605,
+      "eval_iou_concrete tiles": 0.9229760920981392,
+      "eval_iou_dark bricks": 0.7989115168539326,
+      "eval_iou_grass": 0.8330659760473641,
+      "eval_iou_red bricks": 0.8367537313432836,
+      "eval_iou_road": 0.898967881093753,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.31242352263729933,
+      "eval_iou_speedway bricks": 0.9745921480945452,
+      "eval_iou_steel": 0.8308461132888614,
+      "eval_iou_tiles": 0.0007533214628133132,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10976611822843552,
+      "eval_mean_accuracy": 0.6870309609944829,
+      "eval_mean_iou": 0.6356681570547971,
+      "eval_overall_accuracy": 0.9703302383422852,
+      "eval_runtime": 3.3141,
+      "eval_samples_per_second": 10.863,
+      "eval_steps_per_second": 0.603,
+      "step": 1900
+    },
+    {
+      "epoch": 172.82,
+      "learning_rate": 6.888038277511963e-05,
+      "loss": 0.0536,
+      "step": 1901
+    },
+    {
+      "epoch": 172.91,
+      "learning_rate": 6.886124401913876e-05,
+      "loss": 0.0433,
+      "step": 1902
+    },
+    {
+      "epoch": 173.0,
+      "learning_rate": 6.88421052631579e-05,
+      "loss": 0.0414,
+      "step": 1903
+    },
+    {
+      "epoch": 173.09,
+      "learning_rate": 6.882296650717704e-05,
+      "loss": 0.0437,
+      "step": 1904
+    },
+    {
+      "epoch": 173.18,
+      "learning_rate": 6.880382775119617e-05,
+      "loss": 0.0441,
+      "step": 1905
+    },
+    {
+      "epoch": 173.27,
+      "learning_rate": 6.878468899521532e-05,
+      "loss": 0.044,
+      "step": 1906
+    },
+    {
+      "epoch": 173.36,
+      "learning_rate": 6.876555023923446e-05,
+      "loss": 0.0419,
+      "step": 1907
+    },
+    {
+      "epoch": 173.45,
+      "learning_rate": 6.87464114832536e-05,
+      "loss": 0.0438,
+      "step": 1908
+    },
+    {
+      "epoch": 173.55,
+      "learning_rate": 6.872727272727273e-05,
+      "loss": 0.0413,
+      "step": 1909
+    },
+    {
+      "epoch": 173.64,
+      "learning_rate": 6.870813397129187e-05,
+      "loss": 0.0425,
+      "step": 1910
+    },
+    {
+      "epoch": 173.73,
+      "learning_rate": 6.8688995215311e-05,
+      "loss": 0.0454,
+      "step": 1911
+    },
+    {
+      "epoch": 173.82,
+      "learning_rate": 6.866985645933015e-05,
+      "loss": 0.0394,
+      "step": 1912
+    },
+    {
+      "epoch": 173.91,
+      "learning_rate": 6.865071770334928e-05,
+      "loss": 0.047,
+      "step": 1913
+    },
+    {
+      "epoch": 174.0,
+      "learning_rate": 6.863157894736843e-05,
+      "loss": 0.0558,
+      "step": 1914
+    },
+    {
+      "epoch": 174.09,
+      "learning_rate": 6.861244019138756e-05,
+      "loss": 0.045,
+      "step": 1915
+    },
+    {
+      "epoch": 174.18,
+      "learning_rate": 6.85933014354067e-05,
+      "loss": 0.047,
+      "step": 1916
+    },
+    {
+      "epoch": 174.27,
+      "learning_rate": 6.857416267942584e-05,
+      "loss": 0.0484,
+      "step": 1917
+    },
+    {
+      "epoch": 174.36,
+      "learning_rate": 6.855502392344498e-05,
+      "loss": 0.0379,
+      "step": 1918
+    },
+    {
+      "epoch": 174.45,
+      "learning_rate": 6.853588516746411e-05,
+      "loss": 0.052,
+      "step": 1919
+    },
+    {
+      "epoch": 174.55,
+      "learning_rate": 6.851674641148326e-05,
+      "loss": 0.0406,
+      "step": 1920
+    },
+    {
+      "epoch": 174.55,
+      "eval_accuracy_NAT": 0.9925837595804133,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9485492939611142,
+      "eval_accuracy_concrete tiles": 0.9919586620192014,
+      "eval_accuracy_dark bricks": 0.8798352642315538,
+      "eval_accuracy_grass": 0.8833868662120741,
+      "eval_accuracy_red bricks": 0.9554332386363636,
+      "eval_accuracy_road": 0.8674646638943538,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4360543655060238,
+      "eval_accuracy_speedway bricks": 0.983531794160702,
+      "eval_accuracy_steel": 0.9023398852318448,
+      "eval_accuracy_tiles": 0.015819750719079578,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.978735546489165,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8694489044723493,
+      "eval_iou_concrete tiles": 0.9024478442280945,
+      "eval_iou_dark bricks": 0.8014796115143228,
+      "eval_iou_grass": 0.8222491875168759,
+      "eval_iou_red bricks": 0.820649687357023,
+      "eval_iou_road": 0.8602888887556438,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.2921823622861787,
+      "eval_iou_speedway bricks": 0.9738587513405806,
+      "eval_iou_steel": 0.8313431475320177,
+      "eval_iou_tiles": 0.015819750719079578,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.14024387300014496,
+      "eval_mean_accuracy": 0.6813044264732865,
+      "eval_mean_iou": 0.6283464370931793,
+      "eval_overall_accuracy": 0.9662564595540365,
+      "eval_runtime": 3.5891,
+      "eval_samples_per_second": 10.03,
+      "eval_steps_per_second": 0.557,
+      "step": 1920
+    },
+    {
+      "epoch": 174.64,
+      "learning_rate": 6.849760765550239e-05,
+      "loss": 0.0548,
+      "step": 1921
+    },
+    {
+      "epoch": 174.73,
+      "learning_rate": 6.847846889952154e-05,
+      "loss": 0.0469,
+      "step": 1922
+    },
+    {
+      "epoch": 174.82,
+      "learning_rate": 6.845933014354068e-05,
+      "loss": 0.042,
+      "step": 1923
+    },
+    {
+      "epoch": 174.91,
+      "learning_rate": 6.844019138755981e-05,
+      "loss": 0.0409,
+      "step": 1924
+    },
+    {
+      "epoch": 175.0,
+      "learning_rate": 6.842105263157895e-05,
+      "loss": 0.0435,
+      "step": 1925
+    },
+    {
+      "epoch": 175.09,
+      "learning_rate": 6.840191387559809e-05,
+      "loss": 0.0472,
+      "step": 1926
+    },
+    {
+      "epoch": 175.18,
+      "learning_rate": 6.838277511961722e-05,
+      "loss": 0.0385,
+      "step": 1927
+    },
+    {
+      "epoch": 175.27,
+      "learning_rate": 6.836363636363637e-05,
+      "loss": 0.0423,
+      "step": 1928
+    },
+    {
+      "epoch": 175.36,
+      "learning_rate": 6.834449760765552e-05,
+      "loss": 0.0451,
+      "step": 1929
+    },
+    {
+      "epoch": 175.45,
+      "learning_rate": 6.832535885167465e-05,
+      "loss": 0.0493,
+      "step": 1930
+    },
+    {
+      "epoch": 175.55,
+      "learning_rate": 6.830622009569378e-05,
+      "loss": 0.0376,
+      "step": 1931
+    },
+    {
+      "epoch": 175.64,
+      "learning_rate": 6.828708133971292e-05,
+      "loss": 0.039,
+      "step": 1932
+    },
+    {
+      "epoch": 175.73,
+      "learning_rate": 6.826794258373206e-05,
+      "loss": 0.0433,
+      "step": 1933
+    },
+    {
+      "epoch": 175.82,
+      "learning_rate": 6.82488038277512e-05,
+      "loss": 0.0486,
+      "step": 1934
+    },
+    {
+      "epoch": 175.91,
+      "learning_rate": 6.822966507177035e-05,
+      "loss": 0.0433,
+      "step": 1935
+    },
+    {
+      "epoch": 176.0,
+      "learning_rate": 6.821052631578948e-05,
+      "loss": 0.0398,
+      "step": 1936
+    },
+    {
+      "epoch": 176.09,
+      "learning_rate": 6.819138755980861e-05,
+      "loss": 0.0389,
+      "step": 1937
+    },
+    {
+      "epoch": 176.18,
+      "learning_rate": 6.817224880382776e-05,
+      "loss": 0.0493,
+      "step": 1938
+    },
+    {
+      "epoch": 176.27,
+      "learning_rate": 6.815311004784689e-05,
+      "loss": 0.0439,
+      "step": 1939
+    },
+    {
+      "epoch": 176.36,
+      "learning_rate": 6.813397129186602e-05,
+      "loss": 0.0405,
+      "step": 1940
+    },
+    {
+      "epoch": 176.36,
+      "eval_accuracy_NAT": 0.9919912367317583,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9462087965148719,
+      "eval_accuracy_concrete tiles": 0.9792087078823457,
+      "eval_accuracy_dark bricks": 0.8778722912897887,
+      "eval_accuracy_grass": 0.9040637221835822,
+      "eval_accuracy_red bricks": 0.9209872159090909,
+      "eval_accuracy_road": 0.9058135133925926,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5158106580652276,
+      "eval_accuracy_speedway bricks": 0.9873181781428797,
+      "eval_accuracy_steel": 0.8948453268254073,
+      "eval_accuracy_tiles": 0.001712094233666621,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9790039935755902,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8802783597389099,
+      "eval_iou_concrete tiles": 0.949565629910754,
+      "eval_iou_dark bricks": 0.8014618033593366,
+      "eval_iou_grass": 0.8317299035369775,
+      "eval_iou_red bricks": 0.8453389830508474,
+      "eval_iou_road": 0.8977649835031358,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.33883264818269077,
+      "eval_iou_speedway bricks": 0.9749040281656406,
+      "eval_iou_steel": 0.8308695752026763,
+      "eval_iou_tiles": 0.001712094233666621,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11012466251850128,
+      "eval_mean_accuracy": 0.6866024416285548,
+      "eval_mean_iou": 0.6408816924969405,
+      "eval_overall_accuracy": 0.9704863230387369,
+      "eval_runtime": 3.4713,
+      "eval_samples_per_second": 10.371,
+      "eval_steps_per_second": 0.576,
+      "step": 1940
+    },
+    {
+      "epoch": 176.45,
+      "learning_rate": 6.811483253588518e-05,
+      "loss": 0.0423,
+      "step": 1941
+    },
+    {
+      "epoch": 176.55,
+      "learning_rate": 6.809569377990431e-05,
+      "loss": 0.047,
+      "step": 1942
+    },
+    {
+      "epoch": 176.64,
+      "learning_rate": 6.807655502392344e-05,
+      "loss": 0.0449,
+      "step": 1943
+    },
+    {
+      "epoch": 176.73,
+      "learning_rate": 6.805741626794259e-05,
+      "loss": 0.0478,
+      "step": 1944
+    },
+    {
+      "epoch": 176.82,
+      "learning_rate": 6.803827751196172e-05,
+      "loss": 0.0397,
+      "step": 1945
+    },
+    {
+      "epoch": 176.91,
+      "learning_rate": 6.801913875598085e-05,
+      "loss": 0.0389,
+      "step": 1946
+    },
+    {
+      "epoch": 177.0,
+      "learning_rate": 6.800000000000001e-05,
+      "loss": 0.0403,
+      "step": 1947
+    },
+    {
+      "epoch": 177.09,
+      "learning_rate": 6.798086124401915e-05,
+      "loss": 0.0403,
+      "step": 1948
+    },
+    {
+      "epoch": 177.18,
+      "learning_rate": 6.796172248803828e-05,
+      "loss": 0.047,
+      "step": 1949
+    },
+    {
+      "epoch": 177.27,
+      "learning_rate": 6.794258373205742e-05,
+      "loss": 0.0476,
+      "step": 1950
+    },
+    {
+      "epoch": 177.36,
+      "learning_rate": 6.792344497607655e-05,
+      "loss": 0.0412,
+      "step": 1951
+    },
+    {
+      "epoch": 177.45,
+      "learning_rate": 6.790430622009569e-05,
+      "loss": 0.0442,
+      "step": 1952
+    },
+    {
+      "epoch": 177.55,
+      "learning_rate": 6.788516746411485e-05,
+      "loss": 0.0509,
+      "step": 1953
+    },
+    {
+      "epoch": 177.64,
+      "learning_rate": 6.786602870813398e-05,
+      "loss": 0.0379,
+      "step": 1954
+    },
+    {
+      "epoch": 177.73,
+      "learning_rate": 6.784688995215311e-05,
+      "loss": 0.0379,
+      "step": 1955
+    },
+    {
+      "epoch": 177.82,
+      "learning_rate": 6.782775119617226e-05,
+      "loss": 0.0422,
+      "step": 1956
+    },
+    {
+      "epoch": 177.91,
+      "learning_rate": 6.780861244019139e-05,
+      "loss": 0.0448,
+      "step": 1957
+    },
+    {
+      "epoch": 178.0,
+      "learning_rate": 6.778947368421052e-05,
+      "loss": 0.0479,
+      "step": 1958
+    },
+    {
+      "epoch": 178.09,
+      "learning_rate": 6.777033492822968e-05,
+      "loss": 0.0409,
+      "step": 1959
+    },
+    {
+      "epoch": 178.18,
+      "learning_rate": 6.775119617224881e-05,
+      "loss": 0.0406,
+      "step": 1960
+    },
+    {
+      "epoch": 178.18,
+      "eval_accuracy_NAT": 0.9917911411658337,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9413748980642946,
+      "eval_accuracy_concrete tiles": 0.9896960802299273,
+      "eval_accuracy_dark bricks": 0.883183865132212,
+      "eval_accuracy_grass": 0.9071044362970393,
+      "eval_accuracy_red bricks": 0.9396306818181818,
+      "eval_accuracy_road": 0.9390425726160145,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5455848702891958,
+      "eval_accuracy_speedway bricks": 0.9902477230406443,
+      "eval_accuracy_steel": 0.8801612690455775,
+      "eval_accuracy_tiles": 0.019449390494452816,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787979952365885,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8794255771293391,
+      "eval_iou_concrete tiles": 0.9280369276642105,
+      "eval_iou_dark bricks": 0.8102115038310794,
+      "eval_iou_grass": 0.8313782521510165,
+      "eval_iou_red bricks": 0.8349637109498265,
+      "eval_iou_road": 0.9326055758711173,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.35438005873504763,
+      "eval_iou_speedway bricks": 0.9742153138536812,
+      "eval_iou_steel": 0.8268760069401413,
+      "eval_iou_tiles": 0.019449390494452816,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09781524538993835,
+      "eval_mean_accuracy": 0.6944051483225674,
+      "eval_mean_iou": 0.6438723317581924,
+      "eval_overall_accuracy": 0.9735180536905924,
+      "eval_runtime": 3.4973,
+      "eval_samples_per_second": 10.294,
+      "eval_steps_per_second": 0.572,
+      "step": 1960
+    },
+    {
+      "epoch": 178.27,
+      "learning_rate": 6.773205741626794e-05,
+      "loss": 0.0449,
+      "step": 1961
+    },
+    {
+      "epoch": 178.36,
+      "learning_rate": 6.771291866028709e-05,
+      "loss": 0.0405,
+      "step": 1962
+    },
+    {
+      "epoch": 178.45,
+      "learning_rate": 6.769377990430622e-05,
+      "loss": 0.0373,
+      "step": 1963
+    },
+    {
+      "epoch": 178.55,
+      "learning_rate": 6.767464114832535e-05,
+      "loss": 0.0467,
+      "step": 1964
+    },
+    {
+      "epoch": 178.64,
+      "learning_rate": 6.765550239234451e-05,
+      "loss": 0.0463,
+      "step": 1965
+    },
+    {
+      "epoch": 178.73,
+      "learning_rate": 6.763636363636364e-05,
+      "loss": 0.0394,
+      "step": 1966
+    },
+    {
+      "epoch": 178.82,
+      "learning_rate": 6.761722488038278e-05,
+      "loss": 0.0481,
+      "step": 1967
+    },
+    {
+      "epoch": 178.91,
+      "learning_rate": 6.759808612440192e-05,
+      "loss": 0.0466,
+      "step": 1968
+    },
+    {
+      "epoch": 179.0,
+      "learning_rate": 6.757894736842105e-05,
+      "loss": 0.0438,
+      "step": 1969
+    },
+    {
+      "epoch": 179.09,
+      "learning_rate": 6.755980861244018e-05,
+      "loss": 0.0421,
+      "step": 1970
+    },
+    {
+      "epoch": 179.18,
+      "learning_rate": 6.754066985645934e-05,
+      "loss": 0.0429,
+      "step": 1971
+    },
+    {
+      "epoch": 179.27,
+      "learning_rate": 6.752153110047848e-05,
+      "loss": 0.0429,
+      "step": 1972
+    },
+    {
+      "epoch": 179.36,
+      "learning_rate": 6.750239234449761e-05,
+      "loss": 0.0399,
+      "step": 1973
+    },
+    {
+      "epoch": 179.45,
+      "learning_rate": 6.748325358851675e-05,
+      "loss": 0.0518,
+      "step": 1974
+    },
+    {
+      "epoch": 179.55,
+      "learning_rate": 6.746411483253589e-05,
+      "loss": 0.0502,
+      "step": 1975
+    },
+    {
+      "epoch": 179.64,
+      "learning_rate": 6.744497607655502e-05,
+      "loss": 0.044,
+      "step": 1976
+    },
+    {
+      "epoch": 179.73,
+      "learning_rate": 6.742583732057418e-05,
+      "loss": 0.0373,
+      "step": 1977
+    },
+    {
+      "epoch": 179.82,
+      "learning_rate": 6.740669856459331e-05,
+      "loss": 0.0407,
+      "step": 1978
+    },
+    {
+      "epoch": 179.91,
+      "learning_rate": 6.738755980861244e-05,
+      "loss": 0.0534,
+      "step": 1979
+    },
+    {
+      "epoch": 180.0,
+      "learning_rate": 6.736842105263159e-05,
+      "loss": 0.0456,
+      "step": 1980
+    },
+    {
+      "epoch": 180.0,
+      "eval_accuracy_NAT": 0.9922395034524425,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9444128503369243,
+      "eval_accuracy_concrete tiles": 0.9868219898489574,
+      "eval_accuracy_dark bricks": 0.8744082213925561,
+      "eval_accuracy_grass": 0.8798813072973644,
+      "eval_accuracy_red bricks": 0.9273792613636364,
+      "eval_accuracy_road": 0.9261717536067677,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.46905411738758856,
+      "eval_accuracy_speedway bricks": 0.9875474731091312,
+      "eval_accuracy_steel": 0.8952823032781478,
+      "eval_accuracy_tiles": 0.13347486645664977,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.979057033751834,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8721908076873053,
+      "eval_iou_concrete tiles": 0.929285076732602,
+      "eval_iou_dark bricks": 0.8041769911504425,
+      "eval_iou_grass": 0.812185246721404,
+      "eval_iou_red bricks": 0.8410628019323672,
+      "eval_iou_road": 0.9185334778812851,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.31319490464619687,
+      "eval_iou_speedway bricks": 0.9746629568796816,
+      "eval_iou_steel": 0.8296937559216357,
+      "eval_iou_tiles": 0.1334657262206396,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.1043887734413147,
+      "eval_mean_accuracy": 0.6935902805792435,
+      "eval_mean_iou": 0.6467314445788765,
+      "eval_overall_accuracy": 0.9716350767347548,
+      "eval_runtime": 3.5826,
+      "eval_samples_per_second": 10.049,
+      "eval_steps_per_second": 0.558,
+      "step": 1980
+    },
+    {
+      "epoch": 180.09,
+      "learning_rate": 6.734928229665072e-05,
+      "loss": 0.0367,
+      "step": 1981
+    },
+    {
+      "epoch": 180.18,
+      "learning_rate": 6.733014354066985e-05,
+      "loss": 0.0422,
+      "step": 1982
+    },
+    {
+      "epoch": 180.27,
+      "learning_rate": 6.731100478468901e-05,
+      "loss": 0.0443,
+      "step": 1983
+    },
+    {
+      "epoch": 180.36,
+      "learning_rate": 6.729186602870814e-05,
+      "loss": 0.0525,
+      "step": 1984
+    },
+    {
+      "epoch": 180.45,
+      "learning_rate": 6.727272727272727e-05,
+      "loss": 0.0429,
+      "step": 1985
+    },
+    {
+      "epoch": 180.55,
+      "learning_rate": 6.725358851674642e-05,
+      "loss": 0.0385,
+      "step": 1986
+    },
+    {
+      "epoch": 180.64,
+      "learning_rate": 6.723444976076555e-05,
+      "loss": 0.0416,
+      "step": 1987
+    },
+    {
+      "epoch": 180.73,
+      "learning_rate": 6.721531100478468e-05,
+      "loss": 0.0439,
+      "step": 1988
+    },
+    {
+      "epoch": 180.82,
+      "learning_rate": 6.719617224880383e-05,
+      "loss": 0.0423,
+      "step": 1989
+    },
+    {
+      "epoch": 180.91,
+      "learning_rate": 6.717703349282297e-05,
+      "loss": 0.0428,
+      "step": 1990
+    },
+    {
+      "epoch": 181.0,
+      "learning_rate": 6.71578947368421e-05,
+      "loss": 0.0434,
+      "step": 1991
+    },
+    {
+      "epoch": 181.09,
+      "learning_rate": 6.713875598086125e-05,
+      "loss": 0.0376,
+      "step": 1992
+    },
+    {
+      "epoch": 181.18,
+      "learning_rate": 6.711961722488038e-05,
+      "loss": 0.048,
+      "step": 1993
+    },
+    {
+      "epoch": 181.27,
+      "learning_rate": 6.710047846889953e-05,
+      "loss": 0.0425,
+      "step": 1994
+    },
+    {
+      "epoch": 181.36,
+      "learning_rate": 6.708133971291866e-05,
+      "loss": 0.0361,
+      "step": 1995
+    },
+    {
+      "epoch": 181.45,
+      "learning_rate": 6.70622009569378e-05,
+      "loss": 0.0485,
+      "step": 1996
+    },
+    {
+      "epoch": 181.55,
+      "learning_rate": 6.704306220095694e-05,
+      "loss": 0.0415,
+      "step": 1997
+    },
+    {
+      "epoch": 181.64,
+      "learning_rate": 6.702392344497608e-05,
+      "loss": 0.0435,
+      "step": 1998
+    },
+    {
+      "epoch": 181.73,
+      "learning_rate": 6.700478468899522e-05,
+      "loss": 0.0434,
+      "step": 1999
+    },
+    {
+      "epoch": 181.82,
+      "learning_rate": 6.698564593301436e-05,
+      "loss": 0.042,
+      "step": 2000
+    },
+    {
+      "epoch": 181.82,
+      "eval_accuracy_NAT": 0.9927947951155508,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9445469762650758,
+      "eval_accuracy_concrete tiles": 0.9852014920809637,
+      "eval_accuracy_dark bricks": 0.8683653439051615,
+      "eval_accuracy_grass": 0.8680225222548817,
+      "eval_accuracy_red bricks": 0.9385653409090909,
+      "eval_accuracy_road": 0.9425178388535733,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.49596118324924876,
+      "eval_accuracy_speedway bricks": 0.9886082640319468,
+      "eval_accuracy_steel": 0.9001714926456038,
+      "eval_accuracy_tiles": 0.08033146144363786,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9787256693140206,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8760236202351823,
+      "eval_iou_concrete tiles": 0.9470373853750294,
+      "eval_iou_dark bricks": 0.802996867881549,
+      "eval_iou_grass": 0.8114671075417078,
+      "eval_iou_red bricks": 0.8076394194041253,
+      "eval_iou_road": 0.9342271486205632,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3328030190913127,
+      "eval_iou_speedway bricks": 0.9749794694183597,
+      "eval_iou_steel": 0.8353289518985793,
+      "eval_iou_tiles": 0.08033146144363786,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09672261029481888,
+      "eval_mean_accuracy": 0.6926989777503644,
+      "eval_mean_iou": 0.6447353938633898,
+      "eval_overall_accuracy": 0.973306761847602,
+      "eval_runtime": 3.773,
+      "eval_samples_per_second": 9.541,
+      "eval_steps_per_second": 0.53,
+      "step": 2000
+    },
+    {
+      "epoch": 181.91,
+      "learning_rate": 6.69665071770335e-05,
+      "loss": 0.0504,
+      "step": 2001
+    },
+    {
+      "epoch": 182.0,
+      "learning_rate": 6.694736842105264e-05,
+      "loss": 0.0356,
+      "step": 2002
+    },
+    {
+      "epoch": 182.09,
+      "learning_rate": 6.692822966507177e-05,
+      "loss": 0.0462,
+      "step": 2003
+    },
+    {
+      "epoch": 182.18,
+      "learning_rate": 6.690909090909092e-05,
+      "loss": 0.0444,
+      "step": 2004
+    },
+    {
+      "epoch": 182.27,
+      "learning_rate": 6.688995215311005e-05,
+      "loss": 0.0435,
+      "step": 2005
+    },
+    {
+      "epoch": 182.36,
+      "learning_rate": 6.68708133971292e-05,
+      "loss": 0.0409,
+      "step": 2006
+    },
+    {
+      "epoch": 182.45,
+      "learning_rate": 6.685167464114833e-05,
+      "loss": 0.0397,
+      "step": 2007
+    },
+    {
+      "epoch": 182.55,
+      "learning_rate": 6.683253588516747e-05,
+      "loss": 0.0404,
+      "step": 2008
+    },
+    {
+      "epoch": 182.64,
+      "learning_rate": 6.68133971291866e-05,
+      "loss": 0.0423,
+      "step": 2009
+    },
+    {
+      "epoch": 182.73,
+      "learning_rate": 6.679425837320575e-05,
+      "loss": 0.0409,
+      "step": 2010
+    },
+    {
+      "epoch": 182.82,
+      "learning_rate": 6.677511961722488e-05,
+      "loss": 0.0417,
+      "step": 2011
+    },
+    {
+      "epoch": 182.91,
+      "learning_rate": 6.675598086124403e-05,
+      "loss": 0.0423,
+      "step": 2012
+    },
+    {
+      "epoch": 183.0,
+      "learning_rate": 6.673684210526316e-05,
+      "loss": 0.0433,
+      "step": 2013
+    },
+    {
+      "epoch": 183.09,
+      "learning_rate": 6.67177033492823e-05,
+      "loss": 0.0385,
+      "step": 2014
+    },
+    {
+      "epoch": 183.18,
+      "learning_rate": 6.669856459330144e-05,
+      "loss": 0.0465,
+      "step": 2015
+    },
+    {
+      "epoch": 183.27,
+      "learning_rate": 6.667942583732057e-05,
+      "loss": 0.0397,
+      "step": 2016
+    },
+    {
+      "epoch": 183.36,
+      "learning_rate": 6.666028708133971e-05,
+      "loss": 0.0424,
+      "step": 2017
+    },
+    {
+      "epoch": 183.45,
+      "learning_rate": 6.664114832535886e-05,
+      "loss": 0.048,
+      "step": 2018
+    },
+    {
+      "epoch": 183.55,
+      "learning_rate": 6.662200956937799e-05,
+      "loss": 0.0457,
+      "step": 2019
+    },
+    {
+      "epoch": 183.64,
+      "learning_rate": 6.660287081339714e-05,
+      "loss": 0.0374,
+      "step": 2020
+    },
+    {
+      "epoch": 183.64,
+      "eval_accuracy_NAT": 0.9924062497573797,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9454939053178248,
+      "eval_accuracy_concrete tiles": 0.9917752094416926,
+      "eval_accuracy_dark bricks": 0.8842230861013818,
+      "eval_accuracy_grass": 0.8587605769667653,
+      "eval_accuracy_red bricks": 0.9190340909090909,
+      "eval_accuracy_road": 0.919061605563328,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.45979102914013176,
+      "eval_accuracy_speedway bricks": 0.9891531412806969,
+      "eval_accuracy_steel": 0.8893872435855155,
+      "eval_accuracy_tiles": 0.03670730036981235,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9790052617396954,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.865493816322747,
+      "eval_iou_concrete tiles": 0.9283097704767901,
+      "eval_iou_dark bricks": 0.8206401371722511,
+      "eval_iou_grass": 0.7978089201755981,
+      "eval_iou_red bricks": 0.8330919040721069,
+      "eval_iou_road": 0.9113890657582179,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.30605204242668915,
+      "eval_iou_speedway bricks": 0.9745489091243091,
+      "eval_iou_steel": 0.8341801028496307,
+      "eval_iou_tiles": 0.03670730036981235,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11285177618265152,
+      "eval_mean_accuracy": 0.6835225721872017,
+      "eval_mean_iou": 0.6374790177298344,
+      "eval_overall_accuracy": 0.9706172943115234,
+      "eval_runtime": 3.2685,
+      "eval_samples_per_second": 11.014,
+      "eval_steps_per_second": 0.612,
+      "step": 2020
+    },
+    {
+      "epoch": 183.73,
+      "learning_rate": 6.658373205741627e-05,
+      "loss": 0.0418,
+      "step": 2021
+    },
+    {
+      "epoch": 183.82,
+      "learning_rate": 6.65645933014354e-05,
+      "loss": 0.0404,
+      "step": 2022
+    },
+    {
+      "epoch": 183.91,
+      "learning_rate": 6.654545454545455e-05,
+      "loss": 0.0463,
+      "step": 2023
+    },
+    {
+      "epoch": 184.0,
+      "learning_rate": 6.652631578947369e-05,
+      "loss": 0.0434,
+      "step": 2024
+    },
+    {
+      "epoch": 184.09,
+      "learning_rate": 6.650717703349282e-05,
+      "loss": 0.0404,
+      "step": 2025
+    },
+    {
+      "epoch": 184.18,
+      "learning_rate": 6.648803827751197e-05,
+      "loss": 0.0402,
+      "step": 2026
+    },
+    {
+      "epoch": 184.27,
+      "learning_rate": 6.64688995215311e-05,
+      "loss": 0.0408,
+      "step": 2027
+    },
+    {
+      "epoch": 184.36,
+      "learning_rate": 6.644976076555023e-05,
+      "loss": 0.0458,
+      "step": 2028
+    },
+    {
+      "epoch": 184.45,
+      "learning_rate": 6.643062200956938e-05,
+      "loss": 0.0424,
+      "step": 2029
+    },
+    {
+      "epoch": 184.55,
+      "learning_rate": 6.641148325358852e-05,
+      "loss": 0.0433,
+      "step": 2030
+    },
+    {
+      "epoch": 184.64,
+      "learning_rate": 6.639234449760766e-05,
+      "loss": 0.0425,
+      "step": 2031
+    },
+    {
+      "epoch": 184.73,
+      "learning_rate": 6.63732057416268e-05,
+      "loss": 0.0435,
+      "step": 2032
+    },
+    {
+      "epoch": 184.82,
+      "learning_rate": 6.635406698564593e-05,
+      "loss": 0.0427,
+      "step": 2033
+    },
+    {
+      "epoch": 184.91,
+      "learning_rate": 6.633492822966507e-05,
+      "loss": 0.0498,
+      "step": 2034
+    },
+    {
+      "epoch": 185.0,
+      "learning_rate": 6.631578947368421e-05,
+      "loss": 0.0466,
+      "step": 2035
+    },
+    {
+      "epoch": 185.09,
+      "learning_rate": 6.629665071770336e-05,
+      "loss": 0.0417,
+      "step": 2036
+    },
+    {
+      "epoch": 185.18,
+      "learning_rate": 6.627751196172249e-05,
+      "loss": 0.0522,
+      "step": 2037
+    },
+    {
+      "epoch": 185.27,
+      "learning_rate": 6.625837320574163e-05,
+      "loss": 0.0455,
+      "step": 2038
+    },
+    {
+      "epoch": 185.36,
+      "learning_rate": 6.623923444976077e-05,
+      "loss": 0.0406,
+      "step": 2039
+    },
+    {
+      "epoch": 185.45,
+      "learning_rate": 6.62200956937799e-05,
+      "loss": 0.0442,
+      "step": 2040
+    },
+    {
+      "epoch": 185.45,
+      "eval_accuracy_NAT": 0.9925906411739505,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9447776728614962,
+      "eval_accuracy_concrete tiles": 0.9874640738702378,
+      "eval_accuracy_dark bricks": 0.8626303837419653,
+      "eval_accuracy_grass": 0.8853266321120381,
+      "eval_accuracy_red bricks": 0.9625355113636364,
+      "eval_accuracy_road": 0.9060456814919703,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5086152234444352,
+      "eval_accuracy_speedway bricks": 0.9888061817922902,
+      "eval_accuracy_steel": 0.8950349581162192,
+      "eval_accuracy_tiles": 0.08943980276674428,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9791917652725288,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8785884630352135,
+      "eval_iou_concrete tiles": 0.9385643708224354,
+      "eval_iou_dark bricks": 0.8144487244712552,
+      "eval_iou_grass": 0.8235724434358246,
+      "eval_iou_red bricks": 0.8273809523809523,
+      "eval_iou_road": 0.8990046360730637,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3366482975073902,
+      "eval_iou_speedway bricks": 0.9749617292375536,
+      "eval_iou_steel": 0.8356966897613549,
+      "eval_iou_tiles": 0.08942143101677508,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.11241282522678375,
+      "eval_mean_accuracy": 0.6940974432873064,
+      "eval_mean_iou": 0.6459599617703344,
+      "eval_overall_accuracy": 0.9705687628851997,
+      "eval_runtime": 3.8428,
+      "eval_samples_per_second": 9.368,
+      "eval_steps_per_second": 0.52,
+      "step": 2040
+    },
+    {
+      "epoch": 185.55,
+      "learning_rate": 6.620095693779904e-05,
+      "loss": 0.044,
+      "step": 2041
+    },
+    {
+      "epoch": 185.64,
+      "learning_rate": 6.618181818181819e-05,
+      "loss": 0.0445,
+      "step": 2042
+    },
+    {
+      "epoch": 185.73,
+      "learning_rate": 6.616267942583732e-05,
+      "loss": 0.0404,
+      "step": 2043
+    },
+    {
+      "epoch": 185.82,
+      "learning_rate": 6.614354066985647e-05,
+      "loss": 0.0393,
+      "step": 2044
+    },
+    {
+      "epoch": 185.91,
+      "learning_rate": 6.61244019138756e-05,
+      "loss": 0.0464,
+      "step": 2045
+    },
+    {
+      "epoch": 186.0,
+      "learning_rate": 6.610526315789473e-05,
+      "loss": 0.0439,
+      "step": 2046
+    },
+    {
+      "epoch": 186.09,
+      "learning_rate": 6.608612440191388e-05,
+      "loss": 0.0423,
+      "step": 2047
+    },
+    {
+      "epoch": 186.18,
+      "learning_rate": 6.606698564593302e-05,
+      "loss": 0.0464,
+      "step": 2048
+    },
+    {
+      "epoch": 186.27,
+      "learning_rate": 6.604784688995215e-05,
+      "loss": 0.0439,
+      "step": 2049
+    },
+    {
+      "epoch": 186.36,
+      "learning_rate": 6.60287081339713e-05,
+      "loss": 0.0355,
+      "step": 2050
+    },
+    {
+      "epoch": 186.45,
+      "learning_rate": 6.600956937799043e-05,
+      "loss": 0.0448,
+      "step": 2051
+    },
+    {
+      "epoch": 186.55,
+      "learning_rate": 6.599043062200956e-05,
+      "loss": 0.0427,
+      "step": 2052
+    },
+    {
+      "epoch": 186.64,
+      "learning_rate": 6.597129186602871e-05,
+      "loss": 0.0464,
+      "step": 2053
+    },
+    {
+      "epoch": 186.73,
+      "learning_rate": 6.595215311004786e-05,
+      "loss": 0.0476,
+      "step": 2054
+    },
+    {
+      "epoch": 186.82,
+      "learning_rate": 6.593301435406699e-05,
+      "loss": 0.0356,
+      "step": 2055
+    },
+    {
+      "epoch": 186.91,
+      "learning_rate": 6.591387559808613e-05,
+      "loss": 0.0429,
+      "step": 2056
+    },
+    {
+      "epoch": 187.0,
+      "learning_rate": 6.589473684210526e-05,
+      "loss": 0.0424,
+      "step": 2057
+    },
+    {
+      "epoch": 187.09,
+      "learning_rate": 6.58755980861244e-05,
+      "loss": 0.0397,
+      "step": 2058
+    },
+    {
+      "epoch": 187.18,
+      "learning_rate": 6.585645933014354e-05,
+      "loss": 0.0448,
+      "step": 2059
+    },
+    {
+      "epoch": 187.27,
+      "learning_rate": 6.583732057416269e-05,
+      "loss": 0.039,
+      "step": 2060
+    },
+    {
+      "epoch": 187.27,
+      "eval_accuracy_NAT": 0.9939258467712622,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9453208828705094,
+      "eval_accuracy_concrete tiles": 0.9836727206017245,
+      "eval_accuracy_dark bricks": 0.8662869019668219,
+      "eval_accuracy_grass": 0.8743346253455754,
+      "eval_accuracy_red bricks": 0.9382102272727273,
+      "eval_accuracy_road": 0.9282963335578431,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5814793372480909,
+      "eval_accuracy_speedway bricks": 0.9877212545572375,
+      "eval_accuracy_steel": 0.8960820526350505,
+      "eval_accuracy_tiles": 0.05307492124366525,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9789039772273064,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8853638805212686,
+      "eval_iou_concrete tiles": 0.9353141262319389,
+      "eval_iou_dark bricks": 0.8152937767152069,
+      "eval_iou_grass": 0.8229500430946569,
+      "eval_iou_red bricks": 0.83147128245476,
+      "eval_iou_road": 0.9221622413776535,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3904118463674225,
+      "eval_iou_speedway bricks": 0.9749494331027944,
+      "eval_iou_steel": 0.8369063019774533,
+      "eval_iou_tiles": 0.05307492124366525,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10347142815589905,
+      "eval_mean_accuracy": 0.6960311618515775,
+      "eval_mean_iou": 0.6497539869472404,
+      "eval_overall_accuracy": 0.9730592303805881,
+      "eval_runtime": 3.4146,
+      "eval_samples_per_second": 10.543,
+      "eval_steps_per_second": 0.586,
+      "step": 2060
+    },
+    {
+      "epoch": 187.36,
+      "learning_rate": 6.581818181818182e-05,
+      "loss": 0.0492,
+      "step": 2061
+    },
+    {
+      "epoch": 187.45,
+      "learning_rate": 6.579904306220097e-05,
+      "loss": 0.0399,
+      "step": 2062
+    },
+    {
+      "epoch": 187.55,
+      "learning_rate": 6.57799043062201e-05,
+      "loss": 0.0429,
+      "step": 2063
+    },
+    {
+      "epoch": 187.64,
+      "learning_rate": 6.576076555023923e-05,
+      "loss": 0.043,
+      "step": 2064
+    },
+    {
+      "epoch": 187.73,
+      "learning_rate": 6.574162679425838e-05,
+      "loss": 0.0406,
+      "step": 2065
+    },
+    {
+      "epoch": 187.82,
+      "learning_rate": 6.572248803827752e-05,
+      "loss": 0.0413,
+      "step": 2066
+    },
+    {
+      "epoch": 187.91,
+      "learning_rate": 6.570334928229665e-05,
+      "loss": 0.0372,
+      "step": 2067
+    },
+    {
+      "epoch": 188.0,
+      "learning_rate": 6.56842105263158e-05,
+      "loss": 0.0436,
+      "step": 2068
+    },
+    {
+      "epoch": 188.09,
+      "learning_rate": 6.566507177033493e-05,
+      "loss": 0.0425,
+      "step": 2069
+    },
+    {
+      "epoch": 188.18,
+      "learning_rate": 6.564593301435406e-05,
+      "loss": 0.0429,
+      "step": 2070
+    },
+    {
+      "epoch": 188.27,
+      "learning_rate": 6.562679425837321e-05,
+      "loss": 0.0461,
+      "step": 2071
+    },
+    {
+      "epoch": 188.36,
+      "learning_rate": 6.560765550239235e-05,
+      "loss": 0.0422,
+      "step": 2072
+    },
+    {
+      "epoch": 188.45,
+      "learning_rate": 6.558851674641149e-05,
+      "loss": 0.0402,
+      "step": 2073
+    },
+    {
+      "epoch": 188.55,
+      "learning_rate": 6.556937799043063e-05,
+      "loss": 0.0432,
+      "step": 2074
+    },
+    {
+      "epoch": 188.64,
+      "learning_rate": 6.555023923444976e-05,
+      "loss": 0.0455,
+      "step": 2075
+    },
+    {
+      "epoch": 188.73,
+      "learning_rate": 6.55311004784689e-05,
+      "loss": 0.0444,
+      "step": 2076
+    },
+    {
+      "epoch": 188.82,
+      "learning_rate": 6.551196172248804e-05,
+      "loss": 0.0409,
+      "step": 2077
+    },
+    {
+      "epoch": 188.91,
+      "learning_rate": 6.549282296650719e-05,
+      "loss": 0.0388,
+      "step": 2078
+    },
+    {
+      "epoch": 189.0,
+      "learning_rate": 6.547368421052632e-05,
+      "loss": 0.0407,
+      "step": 2079
+    },
+    {
+      "epoch": 189.09,
+      "learning_rate": 6.545454545454546e-05,
+      "loss": 0.0399,
+      "step": 2080
+    },
+    {
+      "epoch": 189.09,
+      "eval_accuracy_NAT": 0.9914188293103655,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9486512296665093,
+      "eval_accuracy_concrete tiles": 0.9881978841802728,
+      "eval_accuracy_dark bricks": 0.8916131018821446,
+      "eval_accuracy_grass": 0.9078523820674759,
+      "eval_accuracy_red bricks": 0.9398082386363636,
+      "eval_accuracy_road": 0.9178499782947012,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.539878146279602,
+      "eval_accuracy_speedway bricks": 0.9859828366683683,
+      "eval_accuracy_steel": 0.9133879691313238,
+      "eval_accuracy_tiles": 0.14361046431995617,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9794614431605171,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8828215981017595,
+      "eval_iou_concrete tiles": 0.9358892685469392,
+      "eval_iou_dark bricks": 0.8145504412954042,
+      "eval_iou_grass": 0.8275756600641665,
+      "eval_iou_red bricks": 0.824840268038024,
+      "eval_iou_road": 0.9102355783932222,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.35696968592209116,
+      "eval_iou_speedway bricks": 0.9751024319914832,
+      "eval_iou_steel": 0.8304697221847404,
+      "eval_iou_tiles": 0.14361046431995617,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.10652013123035431,
+      "eval_mean_accuracy": 0.7052500815720835,
+      "eval_mean_iou": 0.6524251201552542,
+      "eval_overall_accuracy": 0.9719012578328451,
+      "eval_runtime": 3.3447,
+      "eval_samples_per_second": 10.763,
+      "eval_steps_per_second": 0.598,
+      "step": 2080
     }
   ],
   "max_steps": 5500,
   "num_train_epochs": 500,
-  "total_flos": 9.851872301442662e+17,
+  "total_flos": 1.164313775853011e+18,
   "trial_name": null,
   "trial_params": null
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7a89a602af7a559cba27b8f9e51253d3ee9a3753bb186dff447f69ecd6684727
+oid sha256:1ee07344711c4a458a60014f92aa7b147657559d8f5abcee058f54728be5f80c
 size 14944077

runs/May20_21-20-16_robolidar/events.out.tfevents.1716258023.robolidar.3008813.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d73c44d74aa0698e3509494fff6a546e2969f8bf5614cf0ca42e0173ff88e8dc
-size 529002
+oid sha256:6d499e5d7d9c5067262d51e4433e25fc8569f72c273cd769f95e5cee2ec36381
+size 549978