Training in progress, step 4880

{checkpoint-4560 → checkpoint-4880}/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0bcbd8c473cee434a9a670db122182d2befde0477cfdf6c1bdd1c320b333b85c
+oid sha256:83e26c1d71b81aaa587a3aeb051a5a2bc7a9aaab46a4762bd8850149ad7a1d6f
 size 29861573

{checkpoint-4560 → checkpoint-4880}/pytorch_model.bin

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

{checkpoint-4560 → checkpoint-4880}/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5d81c007175793253c88c9a2fc62b88be3230f1ecff15003433a149cbe6825de
+oid sha256:0e2b096535d9d2059b8423d8a799da5b87dee74ac5a7aef4e102821790578a5f
 size 14575

{checkpoint-4560 → checkpoint-4880}/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6d070d4b02d70a49061f261f22e103fdce10fe20ee0f11fc1e104cd9adf2f29d
+oid sha256:8ed0c500b98e02f6210b971f28adc720c9ea05689190a25377cacdd433428861
 size 627

{checkpoint-4560 → checkpoint-4880}/trainer_state.json

@@ -1,8 +1,8 @@
 {
   "best_metric": 0.08047032356262207,
   "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0-512sq/checkpoint-2560",
-  "epoch": 414.54545454545456,
-  "global_step": 4560,
+  "epoch": 443.6363636363636,
+  "global_step": 4880,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -36258,11 +36258,2555 @@
       "eval_samples_per_second": 10.545,
       "eval_steps_per_second": 0.586,
       "step": 4560
+    },
+    {
+      "epoch": 414.64,
+      "learning_rate": 1.797129186602871e-05,
+      "loss": 0.0282,
+      "step": 4561
+    },
+    {
+      "epoch": 414.73,
+      "learning_rate": 1.7952153110047847e-05,
+      "loss": 0.0309,
+      "step": 4562
+    },
+    {
+      "epoch": 414.82,
+      "learning_rate": 1.7933014354066986e-05,
+      "loss": 0.0305,
+      "step": 4563
+    },
+    {
+      "epoch": 414.91,
+      "learning_rate": 1.7913875598086125e-05,
+      "loss": 0.0273,
+      "step": 4564
+    },
+    {
+      "epoch": 415.0,
+      "learning_rate": 1.7894736842105264e-05,
+      "loss": 0.0291,
+      "step": 4565
+    },
+    {
+      "epoch": 415.09,
+      "learning_rate": 1.7875598086124403e-05,
+      "loss": 0.0297,
+      "step": 4566
+    },
+    {
+      "epoch": 415.18,
+      "learning_rate": 1.785645933014354e-05,
+      "loss": 0.0385,
+      "step": 4567
+    },
+    {
+      "epoch": 415.27,
+      "learning_rate": 1.783732057416268e-05,
+      "loss": 0.0307,
+      "step": 4568
+    },
+    {
+      "epoch": 415.36,
+      "learning_rate": 1.781818181818182e-05,
+      "loss": 0.03,
+      "step": 4569
+    },
+    {
+      "epoch": 415.45,
+      "learning_rate": 1.7799043062200958e-05,
+      "loss": 0.0402,
+      "step": 4570
+    },
+    {
+      "epoch": 415.55,
+      "learning_rate": 1.7779904306220096e-05,
+      "loss": 0.028,
+      "step": 4571
+    },
+    {
+      "epoch": 415.64,
+      "learning_rate": 1.7760765550239235e-05,
+      "loss": 0.0283,
+      "step": 4572
+    },
+    {
+      "epoch": 415.73,
+      "learning_rate": 1.7741626794258374e-05,
+      "loss": 0.028,
+      "step": 4573
+    },
+    {
+      "epoch": 415.82,
+      "learning_rate": 1.7722488038277513e-05,
+      "loss": 0.0264,
+      "step": 4574
+    },
+    {
+      "epoch": 415.91,
+      "learning_rate": 1.770334928229665e-05,
+      "loss": 0.0306,
+      "step": 4575
+    },
+    {
+      "epoch": 416.0,
+      "learning_rate": 1.768421052631579e-05,
+      "loss": 0.0252,
+      "step": 4576
+    },
+    {
+      "epoch": 416.09,
+      "learning_rate": 1.766507177033493e-05,
+      "loss": 0.0295,
+      "step": 4577
+    },
+    {
+      "epoch": 416.18,
+      "learning_rate": 1.7645933014354068e-05,
+      "loss": 0.0299,
+      "step": 4578
+    },
+    {
+      "epoch": 416.27,
+      "learning_rate": 1.7626794258373207e-05,
+      "loss": 0.0282,
+      "step": 4579
+    },
+    {
+      "epoch": 416.36,
+      "learning_rate": 1.7607655502392345e-05,
+      "loss": 0.0323,
+      "step": 4580
+    },
+    {
+      "epoch": 416.36,
+      "eval_accuracy_NAT": 0.9937931555317778,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9526575711403923,
+      "eval_accuracy_concrete tiles": 0.9675288937809576,
+      "eval_accuracy_dark bricks": 0.8724067587852662,
+      "eval_accuracy_grass": 0.8955776832554514,
+      "eval_accuracy_red bricks": 0.8831676136363636,
+      "eval_accuracy_road": 0.9818787961600364,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4764976704435806,
+      "eval_accuracy_speedway bricks": 0.9879752892435317,
+      "eval_accuracy_steel": 0.900872303937735,
+      "eval_accuracy_tiles": 0.06601835365018491,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809128257806169,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8872298884890244,
+      "eval_iou_concrete tiles": 0.950555722439171,
+      "eval_iou_dark bricks": 0.8352433946272617,
+      "eval_iou_grass": 0.8366965766754176,
+      "eval_iou_red bricks": 0.8368102288021534,
+      "eval_iou_road": 0.9743478057402455,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3156030311330229,
+      "eval_iou_speedway bricks": 0.9772729307515864,
+      "eval_iou_steel": 0.8501525007002583,
+      "eval_iou_tiles": 0.0660093125171186,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08955886960029602,
+      "eval_mean_accuracy": 0.6906441607357908,
+      "eval_mean_iou": 0.6546795552042982,
+      "eval_overall_accuracy": 0.978549427456326,
+      "eval_runtime": 4.2035,
+      "eval_samples_per_second": 8.564,
+      "eval_steps_per_second": 0.476,
+      "step": 4580
+    },
+    {
+      "epoch": 416.45,
+      "learning_rate": 1.7588516746411484e-05,
+      "loss": 0.0285,
+      "step": 4581
+    },
+    {
+      "epoch": 416.55,
+      "learning_rate": 1.7569377990430623e-05,
+      "loss": 0.0274,
+      "step": 4582
+    },
+    {
+      "epoch": 416.64,
+      "learning_rate": 1.7550239234449762e-05,
+      "loss": 0.0272,
+      "step": 4583
+    },
+    {
+      "epoch": 416.73,
+      "learning_rate": 1.75311004784689e-05,
+      "loss": 0.0367,
+      "step": 4584
+    },
+    {
+      "epoch": 416.82,
+      "learning_rate": 1.751196172248804e-05,
+      "loss": 0.0334,
+      "step": 4585
+    },
+    {
+      "epoch": 416.91,
+      "learning_rate": 1.7492822966507178e-05,
+      "loss": 0.0254,
+      "step": 4586
+    },
+    {
+      "epoch": 417.0,
+      "learning_rate": 1.7473684210526317e-05,
+      "loss": 0.0325,
+      "step": 4587
+    },
+    {
+      "epoch": 417.09,
+      "learning_rate": 1.7454545454545456e-05,
+      "loss": 0.0318,
+      "step": 4588
+    },
+    {
+      "epoch": 417.18,
+      "learning_rate": 1.7435406698564594e-05,
+      "loss": 0.033,
+      "step": 4589
+    },
+    {
+      "epoch": 417.27,
+      "learning_rate": 1.7416267942583733e-05,
+      "loss": 0.026,
+      "step": 4590
+    },
+    {
+      "epoch": 417.36,
+      "learning_rate": 1.7397129186602872e-05,
+      "loss": 0.031,
+      "step": 4591
+    },
+    {
+      "epoch": 417.45,
+      "learning_rate": 1.737799043062201e-05,
+      "loss": 0.03,
+      "step": 4592
+    },
+    {
+      "epoch": 417.55,
+      "learning_rate": 1.735885167464115e-05,
+      "loss": 0.0311,
+      "step": 4593
+    },
+    {
+      "epoch": 417.64,
+      "learning_rate": 1.7339712918660288e-05,
+      "loss": 0.0307,
+      "step": 4594
+    },
+    {
+      "epoch": 417.73,
+      "learning_rate": 1.7320574162679427e-05,
+      "loss": 0.031,
+      "step": 4595
+    },
+    {
+      "epoch": 417.82,
+      "learning_rate": 1.7301435406698566e-05,
+      "loss": 0.0309,
+      "step": 4596
+    },
+    {
+      "epoch": 417.91,
+      "learning_rate": 1.7282296650717705e-05,
+      "loss": 0.0335,
+      "step": 4597
+    },
+    {
+      "epoch": 418.0,
+      "learning_rate": 1.7263157894736843e-05,
+      "loss": 0.0287,
+      "step": 4598
+    },
+    {
+      "epoch": 418.09,
+      "learning_rate": 1.7244019138755982e-05,
+      "loss": 0.0298,
+      "step": 4599
+    },
+    {
+      "epoch": 418.18,
+      "learning_rate": 1.722488038277512e-05,
+      "loss": 0.0294,
+      "step": 4600
+    },
+    {
+      "epoch": 418.18,
+      "eval_accuracy_NAT": 0.9931079958470466,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9536554680458389,
+      "eval_accuracy_concrete tiles": 0.9765486455084694,
+      "eval_accuracy_dark bricks": 0.8884954389746353,
+      "eval_accuracy_grass": 0.9024455030634321,
+      "eval_accuracy_red bricks": 0.8980823863636364,
+      "eval_accuracy_road": 0.9795329309892417,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.47795881233975684,
+      "eval_accuracy_speedway bricks": 0.9884115530872154,
+      "eval_accuracy_steel": 0.8972363300573841,
+      "eval_accuracy_tiles": 0.0734146007396247,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.980815294229793,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8869626722586134,
+      "eval_iou_concrete tiles": 0.9545997967601172,
+      "eval_iou_dark bricks": 0.8389910590971869,
+      "eval_iou_grass": 0.8361587958510228,
+      "eval_iou_red bricks": 0.8472361809045226,
+      "eval_iou_road": 0.9725509170104355,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.31342878837184074,
+      "eval_iou_speedway bricks": 0.977246002688241,
+      "eval_iou_steel": 0.8490995911488406,
+      "eval_iou_tiles": 0.07339449541284404,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09073410928249359,
+      "eval_mean_accuracy": 0.6945299742320217,
+      "eval_mean_iou": 0.6561910456718044,
+      "eval_overall_accuracy": 0.9783508512708876,
+      "eval_runtime": 3.7048,
+      "eval_samples_per_second": 9.717,
+      "eval_steps_per_second": 0.54,
+      "step": 4600
+    },
+    {
+      "epoch": 418.27,
+      "learning_rate": 1.720574162679426e-05,
+      "loss": 0.0278,
+      "step": 4601
+    },
+    {
+      "epoch": 418.36,
+      "learning_rate": 1.71866028708134e-05,
+      "loss": 0.0361,
+      "step": 4602
+    },
+    {
+      "epoch": 418.45,
+      "learning_rate": 1.7167464114832537e-05,
+      "loss": 0.0292,
+      "step": 4603
+    },
+    {
+      "epoch": 418.55,
+      "learning_rate": 1.7148325358851676e-05,
+      "loss": 0.0275,
+      "step": 4604
+    },
+    {
+      "epoch": 418.64,
+      "learning_rate": 1.7129186602870815e-05,
+      "loss": 0.029,
+      "step": 4605
+    },
+    {
+      "epoch": 418.73,
+      "learning_rate": 1.7110047846889954e-05,
+      "loss": 0.0278,
+      "step": 4606
+    },
+    {
+      "epoch": 418.82,
+      "learning_rate": 1.7090909090909092e-05,
+      "loss": 0.0281,
+      "step": 4607
+    },
+    {
+      "epoch": 418.91,
+      "learning_rate": 1.707177033492823e-05,
+      "loss": 0.0357,
+      "step": 4608
+    },
+    {
+      "epoch": 419.0,
+      "learning_rate": 1.705263157894737e-05,
+      "loss": 0.0294,
+      "step": 4609
+    },
+    {
+      "epoch": 419.09,
+      "learning_rate": 1.7033492822966505e-05,
+      "loss": 0.032,
+      "step": 4610
+    },
+    {
+      "epoch": 419.18,
+      "learning_rate": 1.7014354066985647e-05,
+      "loss": 0.0345,
+      "step": 4611
+    },
+    {
+      "epoch": 419.27,
+      "learning_rate": 1.6995215311004786e-05,
+      "loss": 0.0307,
+      "step": 4612
+    },
+    {
+      "epoch": 419.36,
+      "learning_rate": 1.697607655502392e-05,
+      "loss": 0.0391,
+      "step": 4613
+    },
+    {
+      "epoch": 419.45,
+      "learning_rate": 1.6956937799043064e-05,
+      "loss": 0.0282,
+      "step": 4614
+    },
+    {
+      "epoch": 419.55,
+      "learning_rate": 1.6937799043062203e-05,
+      "loss": 0.0306,
+      "step": 4615
+    },
+    {
+      "epoch": 419.64,
+      "learning_rate": 1.6918660287081338e-05,
+      "loss": 0.0275,
+      "step": 4616
+    },
+    {
+      "epoch": 419.73,
+      "learning_rate": 1.689952153110048e-05,
+      "loss": 0.0304,
+      "step": 4617
+    },
+    {
+      "epoch": 419.82,
+      "learning_rate": 1.688038277511962e-05,
+      "loss": 0.0295,
+      "step": 4618
+    },
+    {
+      "epoch": 419.91,
+      "learning_rate": 1.6861244019138754e-05,
+      "loss": 0.0321,
+      "step": 4619
+    },
+    {
+      "epoch": 420.0,
+      "learning_rate": 1.6842105263157896e-05,
+      "loss": 0.0308,
+      "step": 4620
+    },
+    {
+      "epoch": 420.0,
+      "eval_accuracy_NAT": 0.9935867077256652,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9538392205674063,
+      "eval_accuracy_concrete tiles": 0.9761205894942824,
+      "eval_accuracy_dark bricks": 0.884184596435857,
+      "eval_accuracy_grass": 0.8984471387579207,
+      "eval_accuracy_red bricks": 0.8868963068181818,
+      "eval_accuracy_road": 0.9723453935793428,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4149367297990241,
+      "eval_accuracy_speedway bricks": 0.9879481358922652,
+      "eval_accuracy_steel": 0.9009877316799684,
+      "eval_accuracy_tiles": 0.07594850020545131,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809304661528708,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8798750869170372,
+      "eval_iou_concrete tiles": 0.9531273324376772,
+      "eval_iou_dark bricks": 0.83897593221577,
+      "eval_iou_grass": 0.8374031676949839,
+      "eval_iou_red bricks": 0.8387909319899244,
+      "eval_iou_road": 0.9660838568529712,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.2758412139872443,
+      "eval_iou_speedway bricks": 0.9771772481514704,
+      "eval_iou_steel": 0.8484657908630703,
+      "eval_iou_tiles": 0.07591730558598028,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09380853921175003,
+      "eval_mean_accuracy": 0.6880954654581051,
+      "eval_mean_iou": 0.6517375640653077,
+      "eval_overall_accuracy": 0.9776102701822916,
+      "eval_runtime": 3.7937,
+      "eval_samples_per_second": 9.489,
+      "eval_steps_per_second": 0.527,
+      "step": 4620
+    },
+    {
+      "epoch": 420.09,
+      "learning_rate": 1.6822966507177035e-05,
+      "loss": 0.0243,
+      "step": 4621
+    },
+    {
+      "epoch": 420.18,
+      "learning_rate": 1.680382775119617e-05,
+      "loss": 0.0288,
+      "step": 4622
+    },
+    {
+      "epoch": 420.27,
+      "learning_rate": 1.6784688995215313e-05,
+      "loss": 0.0275,
+      "step": 4623
+    },
+    {
+      "epoch": 420.36,
+      "learning_rate": 1.676555023923445e-05,
+      "loss": 0.0263,
+      "step": 4624
+    },
+    {
+      "epoch": 420.45,
+      "learning_rate": 1.674641148325359e-05,
+      "loss": 0.0355,
+      "step": 4625
+    },
+    {
+      "epoch": 420.55,
+      "learning_rate": 1.672727272727273e-05,
+      "loss": 0.0314,
+      "step": 4626
+    },
+    {
+      "epoch": 420.64,
+      "learning_rate": 1.6708133971291868e-05,
+      "loss": 0.0271,
+      "step": 4627
+    },
+    {
+      "epoch": 420.73,
+      "learning_rate": 1.6688995215311007e-05,
+      "loss": 0.0311,
+      "step": 4628
+    },
+    {
+      "epoch": 420.82,
+      "learning_rate": 1.6669856459330142e-05,
+      "loss": 0.0302,
+      "step": 4629
+    },
+    {
+      "epoch": 420.91,
+      "learning_rate": 1.6650717703349284e-05,
+      "loss": 0.0315,
+      "step": 4630
+    },
+    {
+      "epoch": 421.0,
+      "learning_rate": 1.6631578947368423e-05,
+      "loss": 0.0336,
+      "step": 4631
+    },
+    {
+      "epoch": 421.09,
+      "learning_rate": 1.661244019138756e-05,
+      "loss": 0.0256,
+      "step": 4632
+    },
+    {
+      "epoch": 421.18,
+      "learning_rate": 1.65933014354067e-05,
+      "loss": 0.028,
+      "step": 4633
+    },
+    {
+      "epoch": 421.27,
+      "learning_rate": 1.657416267942584e-05,
+      "loss": 0.0299,
+      "step": 4634
+    },
+    {
+      "epoch": 421.36,
+      "learning_rate": 1.6555023923444975e-05,
+      "loss": 0.0315,
+      "step": 4635
+    },
+    {
+      "epoch": 421.45,
+      "learning_rate": 1.6535885167464117e-05,
+      "loss": 0.0284,
+      "step": 4636
+    },
+    {
+      "epoch": 421.55,
+      "learning_rate": 1.6516746411483256e-05,
+      "loss": 0.0274,
+      "step": 4637
+    },
+    {
+      "epoch": 421.64,
+      "learning_rate": 1.649760765550239e-05,
+      "loss": 0.0294,
+      "step": 4638
+    },
+    {
+      "epoch": 421.73,
+      "learning_rate": 1.6478468899521533e-05,
+      "loss": 0.0306,
+      "step": 4639
+    },
+    {
+      "epoch": 421.82,
+      "learning_rate": 1.6459330143540672e-05,
+      "loss": 0.0263,
+      "step": 4640
+    },
+    {
+      "epoch": 421.82,
+      "eval_accuracy_NAT": 0.9937619236841864,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9481133846946221,
+      "eval_accuracy_concrete tiles": 0.9737968568458387,
+      "eval_accuracy_dark bricks": 0.8850698587429275,
+      "eval_accuracy_grass": 0.8998766238986149,
+      "eval_accuracy_red bricks": 0.8792613636363636,
+      "eval_accuracy_road": 0.9793213194403297,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.49455517878311694,
+      "eval_accuracy_speedway bricks": 0.988671018443763,
+      "eval_accuracy_steel": 0.8988110942549964,
+      "eval_accuracy_tiles": 0.07451034104917134,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807345907025632,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8876124296353406,
+      "eval_iou_concrete tiles": 0.9527641498145267,
+      "eval_iou_dark bricks": 0.837796480489671,
+      "eval_iou_grass": 0.8362891071614177,
+      "eval_iou_red bricks": 0.8346536322265296,
+      "eval_iou_road": 0.971343725375669,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3242241862314519,
+      "eval_iou_speedway bricks": 0.97713003361108,
+      "eval_iou_steel": 0.8468697320686414,
+      "eval_iou_tiles": 0.07448993564288649,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09013354778289795,
+      "eval_mean_accuracy": 0.6935191510364563,
+      "eval_mean_iou": 0.6556852309969059,
+      "eval_overall_accuracy": 0.9783103730943468,
+      "eval_runtime": 3.5907,
+      "eval_samples_per_second": 10.026,
+      "eval_steps_per_second": 0.557,
+      "step": 4640
+    },
+    {
+      "epoch": 421.91,
+      "learning_rate": 1.6440191387559807e-05,
+      "loss": 0.0328,
+      "step": 4641
+    },
+    {
+      "epoch": 422.0,
+      "learning_rate": 1.642105263157895e-05,
+      "loss": 0.0355,
+      "step": 4642
+    },
+    {
+      "epoch": 422.09,
+      "learning_rate": 1.640191387559809e-05,
+      "loss": 0.0319,
+      "step": 4643
+    },
+    {
+      "epoch": 422.18,
+      "learning_rate": 1.6382775119617224e-05,
+      "loss": 0.0293,
+      "step": 4644
+    },
+    {
+      "epoch": 422.27,
+      "learning_rate": 1.6363636363636366e-05,
+      "loss": 0.0347,
+      "step": 4645
+    },
+    {
+      "epoch": 422.36,
+      "learning_rate": 1.6344497607655505e-05,
+      "loss": 0.0293,
+      "step": 4646
+    },
+    {
+      "epoch": 422.45,
+      "learning_rate": 1.632535885167464e-05,
+      "loss": 0.0324,
+      "step": 4647
+    },
+    {
+      "epoch": 422.55,
+      "learning_rate": 1.630622009569378e-05,
+      "loss": 0.0298,
+      "step": 4648
+    },
+    {
+      "epoch": 422.64,
+      "learning_rate": 1.628708133971292e-05,
+      "loss": 0.0295,
+      "step": 4649
+    },
+    {
+      "epoch": 422.73,
+      "learning_rate": 1.6267942583732056e-05,
+      "loss": 0.0266,
+      "step": 4650
+    },
+    {
+      "epoch": 422.82,
+      "learning_rate": 1.6248803827751195e-05,
+      "loss": 0.0313,
+      "step": 4651
+    },
+    {
+      "epoch": 422.91,
+      "learning_rate": 1.6229665071770337e-05,
+      "loss": 0.0282,
+      "step": 4652
+    },
+    {
+      "epoch": 423.0,
+      "learning_rate": 1.6210526315789473e-05,
+      "loss": 0.0288,
+      "step": 4653
+    },
+    {
+      "epoch": 423.09,
+      "learning_rate": 1.619138755980861e-05,
+      "loss": 0.0311,
+      "step": 4654
+    },
+    {
+      "epoch": 423.18,
+      "learning_rate": 1.6172248803827754e-05,
+      "loss": 0.0275,
+      "step": 4655
+    },
+    {
+      "epoch": 423.27,
+      "learning_rate": 1.615311004784689e-05,
+      "loss": 0.0297,
+      "step": 4656
+    },
+    {
+      "epoch": 423.36,
+      "learning_rate": 1.6133971291866028e-05,
+      "loss": 0.0333,
+      "step": 4657
+    },
+    {
+      "epoch": 423.45,
+      "learning_rate": 1.611483253588517e-05,
+      "loss": 0.0284,
+      "step": 4658
+    },
+    {
+      "epoch": 423.55,
+      "learning_rate": 1.609569377990431e-05,
+      "loss": 0.0315,
+      "step": 4659
+    },
+    {
+      "epoch": 423.64,
+      "learning_rate": 1.6076555023923444e-05,
+      "loss": 0.0317,
+      "step": 4660
+    },
+    {
+      "epoch": 423.64,
+      "eval_accuracy_NAT": 0.9936654049235508,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9518300141636981,
+      "eval_accuracy_concrete tiles": 0.9775882101143522,
+      "eval_accuracy_dark bricks": 0.8837612101150841,
+      "eval_accuracy_grass": 0.8998416731616786,
+      "eval_accuracy_red bricks": 0.8852982954545454,
+      "eval_accuracy_road": 0.9788340082734069,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4781517933449122,
+      "eval_accuracy_speedway bricks": 0.9878896053350904,
+      "eval_accuracy_steel": 0.9041619945913858,
+      "eval_accuracy_tiles": 0.0701273798109848,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807365130251586,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8878143738419946,
+      "eval_iou_concrete tiles": 0.9541330945986273,
+      "eval_iou_dark bricks": 0.8388499196258951,
+      "eval_iou_grass": 0.8377396063476362,
+      "eval_iou_red bricks": 0.8339187155042649,
+      "eval_iou_road": 0.9716201731548731,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3153741249204473,
+      "eval_iou_speedway bricks": 0.9772354210116025,
+      "eval_iou_steel": 0.8499899238865894,
+      "eval_iou_tiles": 0.07011297500855872,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09068559855222702,
+      "eval_mean_accuracy": 0.6931653530222068,
+      "eval_mean_iou": 0.6551942185327422,
+      "eval_overall_accuracy": 0.9783771302964952,
+      "eval_runtime": 3.6443,
+      "eval_samples_per_second": 9.878,
+      "eval_steps_per_second": 0.549,
+      "step": 4660
+    },
+    {
+      "epoch": 423.73,
+      "learning_rate": 1.6057416267942586e-05,
+      "loss": 0.0281,
+      "step": 4661
+    },
+    {
+      "epoch": 423.82,
+      "learning_rate": 1.6038277511961725e-05,
+      "loss": 0.0329,
+      "step": 4662
+    },
+    {
+      "epoch": 423.91,
+      "learning_rate": 1.601913875598086e-05,
+      "loss": 0.0312,
+      "step": 4663
+    },
+    {
+      "epoch": 424.0,
+      "learning_rate": 1.6000000000000003e-05,
+      "loss": 0.0301,
+      "step": 4664
+    },
+    {
+      "epoch": 424.09,
+      "learning_rate": 1.598086124401914e-05,
+      "loss": 0.0328,
+      "step": 4665
+    },
+    {
+      "epoch": 424.18,
+      "learning_rate": 1.5961722488038277e-05,
+      "loss": 0.0291,
+      "step": 4666
+    },
+    {
+      "epoch": 424.27,
+      "learning_rate": 1.5942583732057416e-05,
+      "loss": 0.0344,
+      "step": 4667
+    },
+    {
+      "epoch": 424.36,
+      "learning_rate": 1.5923444976076558e-05,
+      "loss": 0.0343,
+      "step": 4668
+    },
+    {
+      "epoch": 424.45,
+      "learning_rate": 1.5904306220095693e-05,
+      "loss": 0.0281,
+      "step": 4669
+    },
+    {
+      "epoch": 424.55,
+      "learning_rate": 1.5885167464114832e-05,
+      "loss": 0.0307,
+      "step": 4670
+    },
+    {
+      "epoch": 424.64,
+      "learning_rate": 1.5866028708133974e-05,
+      "loss": 0.0293,
+      "step": 4671
+    },
+    {
+      "epoch": 424.73,
+      "learning_rate": 1.584688995215311e-05,
+      "loss": 0.0317,
+      "step": 4672
+    },
+    {
+      "epoch": 424.82,
+      "learning_rate": 1.5827751196172248e-05,
+      "loss": 0.0264,
+      "step": 4673
+    },
+    {
+      "epoch": 424.91,
+      "learning_rate": 1.580861244019139e-05,
+      "loss": 0.0264,
+      "step": 4674
+    },
+    {
+      "epoch": 425.0,
+      "learning_rate": 1.5789473684210526e-05,
+      "loss": 0.0324,
+      "step": 4675
+    },
+    {
+      "epoch": 425.09,
+      "learning_rate": 1.5770334928229665e-05,
+      "loss": 0.032,
+      "step": 4676
+    },
+    {
+      "epoch": 425.18,
+      "learning_rate": 1.5751196172248807e-05,
+      "loss": 0.0329,
+      "step": 4677
+    },
+    {
+      "epoch": 425.27,
+      "learning_rate": 1.5732057416267942e-05,
+      "loss": 0.0288,
+      "step": 4678
+    },
+    {
+      "epoch": 425.36,
+      "learning_rate": 1.571291866028708e-05,
+      "loss": 0.034,
+      "step": 4679
+    },
+    {
+      "epoch": 425.45,
+      "learning_rate": 1.5693779904306223e-05,
+      "loss": 0.0289,
+      "step": 4680
+    },
+    {
+      "epoch": 425.45,
+      "eval_accuracy_NAT": 0.9934796018980494,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9506604360702176,
+      "eval_accuracy_concrete tiles": 0.9773741821072586,
+      "eval_accuracy_dark bricks": 0.8811824025249221,
+      "eval_accuracy_grass": 0.9021938577574908,
+      "eval_accuracy_red bricks": 0.8920454545454546,
+      "eval_accuracy_road": 0.9816490464783606,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4694125106828771,
+      "eval_accuracy_speedway bricks": 0.9882341511922735,
+      "eval_accuracy_steel": 0.9056460655629576,
+      "eval_accuracy_tiles": 0.07088070127379811,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808047340576567,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8879665450609365,
+      "eval_iou_concrete tiles": 0.9542374399235798,
+      "eval_iou_dark bricks": 0.8377794854905405,
+      "eval_iou_grass": 0.8374225864323092,
+      "eval_iou_red bricks": 0.8398528920093614,
+      "eval_iou_road": 0.9740321601432113,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3088742154337336,
+      "eval_iou_speedway bricks": 0.9773084907872807,
+      "eval_iou_steel": 0.850633460335161,
+      "eval_iou_tiles": 0.07087099424815119,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09045922756195068,
+      "eval_mean_accuracy": 0.6932891084687431,
+      "eval_mean_iou": 0.6553679233786093,
+      "eval_overall_accuracy": 0.9785344865587022,
+      "eval_runtime": 3.8905,
+      "eval_samples_per_second": 9.253,
+      "eval_steps_per_second": 0.514,
+      "step": 4680
+    },
+    {
+      "epoch": 425.55,
+      "learning_rate": 1.567464114832536e-05,
+      "loss": 0.0341,
+      "step": 4681
+    },
+    {
+      "epoch": 425.64,
+      "learning_rate": 1.5655502392344497e-05,
+      "loss": 0.0284,
+      "step": 4682
+    },
+    {
+      "epoch": 425.73,
+      "learning_rate": 1.563636363636364e-05,
+      "loss": 0.0259,
+      "step": 4683
+    },
+    {
+      "epoch": 425.82,
+      "learning_rate": 1.5617224880382775e-05,
+      "loss": 0.0309,
+      "step": 4684
+    },
+    {
+      "epoch": 425.91,
+      "learning_rate": 1.5598086124401914e-05,
+      "loss": 0.0329,
+      "step": 4685
+    },
+    {
+      "epoch": 426.0,
+      "learning_rate": 1.5578947368421056e-05,
+      "loss": 0.0301,
+      "step": 4686
+    },
+    {
+      "epoch": 426.09,
+      "learning_rate": 1.555980861244019e-05,
+      "loss": 0.0319,
+      "step": 4687
+    },
+    {
+      "epoch": 426.18,
+      "learning_rate": 1.554066985645933e-05,
+      "loss": 0.0307,
+      "step": 4688
+    },
+    {
+      "epoch": 426.27,
+      "learning_rate": 1.552153110047847e-05,
+      "loss": 0.0258,
+      "step": 4689
+    },
+    {
+      "epoch": 426.36,
+      "learning_rate": 1.5502392344497607e-05,
+      "loss": 0.0319,
+      "step": 4690
+    },
+    {
+      "epoch": 426.45,
+      "learning_rate": 1.5483253588516746e-05,
+      "loss": 0.0256,
+      "step": 4691
+    },
+    {
+      "epoch": 426.55,
+      "learning_rate": 1.5464114832535885e-05,
+      "loss": 0.0331,
+      "step": 4692
+    },
+    {
+      "epoch": 426.64,
+      "learning_rate": 1.5444976076555024e-05,
+      "loss": 0.0296,
+      "step": 4693
+    },
+    {
+      "epoch": 426.73,
+      "learning_rate": 1.5425837320574163e-05,
+      "loss": 0.0279,
+      "step": 4694
+    },
+    {
+      "epoch": 426.82,
+      "learning_rate": 1.54066985645933e-05,
+      "loss": 0.0302,
+      "step": 4695
+    },
+    {
+      "epoch": 426.91,
+      "learning_rate": 1.5387559808612444e-05,
+      "loss": 0.0303,
+      "step": 4696
+    },
+    {
+      "epoch": 427.0,
+      "learning_rate": 1.536842105263158e-05,
+      "loss": 0.0288,
+      "step": 4697
+    },
+    {
+      "epoch": 427.09,
+      "learning_rate": 1.5349282296650718e-05,
+      "loss": 0.0288,
+      "step": 4698
+    },
+    {
+      "epoch": 427.18,
+      "learning_rate": 1.533014354066986e-05,
+      "loss": 0.0311,
+      "step": 4699
+    },
+    {
+      "epoch": 427.27,
+      "learning_rate": 1.5311004784688995e-05,
+      "loss": 0.0301,
+      "step": 4700
+    },
+    {
+      "epoch": 427.27,
+      "eval_accuracy_NAT": 0.9938520902046339,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9513149705995966,
+      "eval_accuracy_concrete tiles": 0.9804011496361524,
+      "eval_accuracy_dark bricks": 0.883183865132212,
+      "eval_accuracy_grass": 0.8971679417860525,
+      "eval_accuracy_red bricks": 0.8971946022727273,
+      "eval_accuracy_road": 0.9820674327407807,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4819562760179748,
+      "eval_accuracy_speedway bricks": 0.9878672792462713,
+      "eval_accuracy_steel": 0.9031726139436712,
+      "eval_accuracy_tiles": 0.06910012327078482,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809410987752452,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8883063728314288,
+      "eval_iou_concrete tiles": 0.9557092188012279,
+      "eval_iou_dark bricks": 0.8386389386352838,
+      "eval_iou_grass": 0.8382890341037121,
+      "eval_iou_red bricks": 0.8447007689735875,
+      "eval_iou_road": 0.9740581227579705,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.3170475154153065,
+      "eval_iou_speedway bricks": 0.9772938375724697,
+      "eval_iou_steel": 0.8507875361148218,
+      "eval_iou_tiles": 0.06909066009312517,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09060151875019073,
+      "eval_mean_accuracy": 0.6944060265269891,
+      "eval_mean_iou": 0.6565279310826291,
+      "eval_overall_accuracy": 0.9786624908447266,
+      "eval_runtime": 4.2015,
+      "eval_samples_per_second": 8.568,
+      "eval_steps_per_second": 0.476,
+      "step": 4700
+    },
+    {
+      "epoch": 427.36,
+      "learning_rate": 1.5291866028708134e-05,
+      "loss": 0.0283,
+      "step": 4701
+    },
+    {
+      "epoch": 427.45,
+      "learning_rate": 1.5272727272727276e-05,
+      "loss": 0.0294,
+      "step": 4702
+    },
+    {
+      "epoch": 427.55,
+      "learning_rate": 1.5253588516746412e-05,
+      "loss": 0.0281,
+      "step": 4703
+    },
+    {
+      "epoch": 427.64,
+      "learning_rate": 1.523444976076555e-05,
+      "loss": 0.0297,
+      "step": 4704
+    },
+    {
+      "epoch": 427.73,
+      "learning_rate": 1.521531100478469e-05,
+      "loss": 0.0313,
+      "step": 4705
+    },
+    {
+      "epoch": 427.82,
+      "learning_rate": 1.5196172248803828e-05,
+      "loss": 0.0308,
+      "step": 4706
+    },
+    {
+      "epoch": 427.91,
+      "learning_rate": 1.5177033492822967e-05,
+      "loss": 0.0315,
+      "step": 4707
+    },
+    {
+      "epoch": 428.0,
+      "learning_rate": 1.5157894736842107e-05,
+      "loss": 0.0318,
+      "step": 4708
+    },
+    {
+      "epoch": 428.09,
+      "learning_rate": 1.5138755980861244e-05,
+      "loss": 0.0308,
+      "step": 4709
+    },
+    {
+      "epoch": 428.18,
+      "learning_rate": 1.5119617224880383e-05,
+      "loss": 0.0317,
+      "step": 4710
+    },
+    {
+      "epoch": 428.27,
+      "learning_rate": 1.5100478468899523e-05,
+      "loss": 0.0288,
+      "step": 4711
+    },
+    {
+      "epoch": 428.36,
+      "learning_rate": 1.508133971291866e-05,
+      "loss": 0.0291,
+      "step": 4712
+    },
+    {
+      "epoch": 428.45,
+      "learning_rate": 1.50622009569378e-05,
+      "loss": 0.0306,
+      "step": 4713
+    },
+    {
+      "epoch": 428.55,
+      "learning_rate": 1.504306220095694e-05,
+      "loss": 0.0311,
+      "step": 4714
+    },
+    {
+      "epoch": 428.64,
+      "learning_rate": 1.5023923444976077e-05,
+      "loss": 0.0296,
+      "step": 4715
+    },
+    {
+      "epoch": 428.73,
+      "learning_rate": 1.5004784688995216e-05,
+      "loss": 0.0314,
+      "step": 4716
+    },
+    {
+      "epoch": 428.82,
+      "learning_rate": 1.4985645933014356e-05,
+      "loss": 0.0329,
+      "step": 4717
+    },
+    {
+      "epoch": 428.91,
+      "learning_rate": 1.4966507177033493e-05,
+      "loss": 0.0287,
+      "step": 4718
+    },
+    {
+      "epoch": 429.0,
+      "learning_rate": 1.4947368421052632e-05,
+      "loss": 0.0311,
+      "step": 4719
+    },
+    {
+      "epoch": 429.09,
+      "learning_rate": 1.4928229665071772e-05,
+      "loss": 0.03,
+      "step": 4720
+    },
+    {
+      "epoch": 429.09,
+      "eval_accuracy_NAT": 0.9936444072407068,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9467748079316709,
+      "eval_accuracy_concrete tiles": 0.984161927475081,
+      "eval_accuracy_dark bricks": 0.8815288095146453,
+      "eval_accuracy_grass": 0.9020156089991157,
+      "eval_accuracy_red bricks": 0.8998579545454546,
+      "eval_accuracy_road": 0.9819622315707502,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4853747966807267,
+      "eval_accuracy_speedway bricks": 0.9896750890328218,
+      "eval_accuracy_steel": 0.9001632478068728,
+      "eval_accuracy_tiles": 0.07779756197781126,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809980260788931,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8883291531434483,
+      "eval_iou_concrete tiles": 0.9551902190040952,
+      "eval_iou_dark bricks": 0.8387841054751877,
+      "eval_iou_grass": 0.8381680480137182,
+      "eval_iou_red bricks": 0.8449483161053685,
+      "eval_iou_road": 0.9738801942795466,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.319800919114308,
+      "eval_iou_speedway bricks": 0.9769885642736902,
+      "eval_iou_steel": 0.8490407571291926,
+      "eval_iou_tiles": 0.07778158165011982,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08905282616615295,
+      "eval_mean_accuracy": 0.695612034059666,
+      "eval_mean_iou": 0.6572238372513515,
+      "eval_overall_accuracy": 0.9786322911580404,
+      "eval_runtime": 4.0999,
+      "eval_samples_per_second": 8.781,
+      "eval_steps_per_second": 0.488,
+      "step": 4720
+    },
+    {
+      "epoch": 429.18,
+      "learning_rate": 1.4909090909090908e-05,
+      "loss": 0.0332,
+      "step": 4721
+    },
+    {
+      "epoch": 429.27,
+      "learning_rate": 1.4889952153110048e-05,
+      "loss": 0.0287,
+      "step": 4722
+    },
+    {
+      "epoch": 429.36,
+      "learning_rate": 1.4870813397129189e-05,
+      "loss": 0.0316,
+      "step": 4723
+    },
+    {
+      "epoch": 429.45,
+      "learning_rate": 1.4851674641148324e-05,
+      "loss": 0.0304,
+      "step": 4724
+    },
+    {
+      "epoch": 429.55,
+      "learning_rate": 1.4832535885167465e-05,
+      "loss": 0.0251,
+      "step": 4725
+    },
+    {
+      "epoch": 429.64,
+      "learning_rate": 1.4813397129186603e-05,
+      "loss": 0.0289,
+      "step": 4726
+    },
+    {
+      "epoch": 429.73,
+      "learning_rate": 1.479425837320574e-05,
+      "loss": 0.0282,
+      "step": 4727
+    },
+    {
+      "epoch": 429.82,
+      "learning_rate": 1.4775119617224881e-05,
+      "loss": 0.0276,
+      "step": 4728
+    },
+    {
+      "epoch": 429.91,
+      "learning_rate": 1.475598086124402e-05,
+      "loss": 0.0313,
+      "step": 4729
+    },
+    {
+      "epoch": 430.0,
+      "learning_rate": 1.4736842105263157e-05,
+      "loss": 0.0315,
+      "step": 4730
+    },
+    {
+      "epoch": 430.09,
+      "learning_rate": 1.4717703349282297e-05,
+      "loss": 0.0312,
+      "step": 4731
+    },
+    {
+      "epoch": 430.18,
+      "learning_rate": 1.4698564593301436e-05,
+      "loss": 0.0286,
+      "step": 4732
+    },
+    {
+      "epoch": 430.27,
+      "learning_rate": 1.4679425837320577e-05,
+      "loss": 0.0294,
+      "step": 4733
+    },
+    {
+      "epoch": 430.36,
+      "learning_rate": 1.4660287081339714e-05,
+      "loss": 0.0294,
+      "step": 4734
+    },
+    {
+      "epoch": 430.45,
+      "learning_rate": 1.4641148325358852e-05,
+      "loss": 0.0311,
+      "step": 4735
+    },
+    {
+      "epoch": 430.55,
+      "learning_rate": 1.4622009569377993e-05,
+      "loss": 0.0349,
+      "step": 4736
+    },
+    {
+      "epoch": 430.64,
+      "learning_rate": 1.460287081339713e-05,
+      "loss": 0.0303,
+      "step": 4737
+    },
+    {
+      "epoch": 430.73,
+      "learning_rate": 1.4583732057416269e-05,
+      "loss": 0.028,
+      "step": 4738
+    },
+    {
+      "epoch": 430.82,
+      "learning_rate": 1.456459330143541e-05,
+      "loss": 0.0315,
+      "step": 4739
+    },
+    {
+      "epoch": 430.91,
+      "learning_rate": 1.4545454545454545e-05,
+      "loss": 0.0247,
+      "step": 4740
+    },
+    {
+      "epoch": 430.91,
+      "eval_accuracy_NAT": 0.993786450389357,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9486324520365681,
+      "eval_accuracy_concrete tiles": 0.9801259707698893,
+      "eval_accuracy_dark bricks": 0.8734459797544359,
+      "eval_accuracy_grass": 0.900638549963826,
+      "eval_accuracy_red bricks": 0.9007457386363636,
+      "eval_accuracy_road": 0.9803443101282124,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5065200011027486,
+      "eval_accuracy_speedway bricks": 0.9883325066646392,
+      "eval_accuracy_steel": 0.9055306378207242,
+      "eval_accuracy_tiles": 0.07903026982605123,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9807589666718967,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8887121357765472,
+      "eval_iou_concrete tiles": 0.9548717643203956,
+      "eval_iou_dark bricks": 0.8328929017103428,
+      "eval_iou_grass": 0.8366873818939822,
+      "eval_iou_red bricks": 0.8433915211970074,
+      "eval_iou_road": 0.9733752107075452,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.33251289476065515,
+      "eval_iou_speedway bricks": 0.9772897051277462,
+      "eval_iou_steel": 0.849393676916414,
+      "eval_iou_tiles": 0.07900862659181158,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08833417296409607,
+      "eval_mean_accuracy": 0.6967025282379089,
+      "eval_mean_iou": 0.6576072912057188,
+      "eval_overall_accuracy": 0.9785615073310004,
+      "eval_runtime": 3.8926,
+      "eval_samples_per_second": 9.248,
+      "eval_steps_per_second": 0.514,
+      "step": 4740
+    },
+    {
+      "epoch": 431.0,
+      "learning_rate": 1.4526315789473685e-05,
+      "loss": 0.0298,
+      "step": 4741
+    },
+    {
+      "epoch": 431.09,
+      "learning_rate": 1.4507177033492826e-05,
+      "loss": 0.0331,
+      "step": 4742
+    },
+    {
+      "epoch": 431.18,
+      "learning_rate": 1.4488038277511961e-05,
+      "loss": 0.0395,
+      "step": 4743
+    },
+    {
+      "epoch": 431.27,
+      "learning_rate": 1.4468899521531101e-05,
+      "loss": 0.0269,
+      "step": 4744
+    },
+    {
+      "epoch": 431.36,
+      "learning_rate": 1.444976076555024e-05,
+      "loss": 0.0277,
+      "step": 4745
+    },
+    {
+      "epoch": 431.45,
+      "learning_rate": 1.4430622009569377e-05,
+      "loss": 0.0287,
+      "step": 4746
+    },
+    {
+      "epoch": 431.55,
+      "learning_rate": 1.4411483253588518e-05,
+      "loss": 0.0301,
+      "step": 4747
+    },
+    {
+      "epoch": 431.64,
+      "learning_rate": 1.4392344497607657e-05,
+      "loss": 0.0283,
+      "step": 4748
+    },
+    {
+      "epoch": 431.73,
+      "learning_rate": 1.4373205741626794e-05,
+      "loss": 0.0293,
+      "step": 4749
+    },
+    {
+      "epoch": 431.82,
+      "learning_rate": 1.4354066985645934e-05,
+      "loss": 0.0332,
+      "step": 4750
+    },
+    {
+      "epoch": 431.91,
+      "learning_rate": 1.4334928229665073e-05,
+      "loss": 0.0327,
+      "step": 4751
+    },
+    {
+      "epoch": 432.0,
+      "learning_rate": 1.431578947368421e-05,
+      "loss": 0.0285,
+      "step": 4752
+    },
+    {
+      "epoch": 432.09,
+      "learning_rate": 1.429665071770335e-05,
+      "loss": 0.0311,
+      "step": 4753
+    },
+    {
+      "epoch": 432.18,
+      "learning_rate": 1.427751196172249e-05,
+      "loss": 0.0284,
+      "step": 4754
+    },
+    {
+      "epoch": 432.27,
+      "learning_rate": 1.4258373205741626e-05,
+      "loss": 0.0326,
+      "step": 4755
+    },
+    {
+      "epoch": 432.36,
+      "learning_rate": 1.4239234449760767e-05,
+      "loss": 0.0249,
+      "step": 4756
+    },
+    {
+      "epoch": 432.45,
+      "learning_rate": 1.4220095693779906e-05,
+      "loss": 0.0321,
+      "step": 4757
+    },
+    {
+      "epoch": 432.55,
+      "learning_rate": 1.4200956937799043e-05,
+      "loss": 0.0296,
+      "step": 4758
+    },
+    {
+      "epoch": 432.64,
+      "learning_rate": 1.4181818181818181e-05,
+      "loss": 0.0301,
+      "step": 4759
+    },
+    {
+      "epoch": 432.73,
+      "learning_rate": 1.4162679425837322e-05,
+      "loss": 0.0281,
+      "step": 4760
+    },
+    {
+      "epoch": 432.73,
+      "eval_accuracy_NAT": 0.993656053014385,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9499991952444311,
+      "eval_accuracy_concrete tiles": 0.968904788112273,
+      "eval_accuracy_dark bricks": 0.8808359955351988,
+      "eval_accuracy_grass": 0.9008412642380564,
+      "eval_accuracy_red bricks": 0.8838778409090909,
+      "eval_accuracy_road": 0.981658720149168,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.48713919444214704,
+      "eval_accuracy_speedway bricks": 0.987837108855975,
+      "eval_accuracy_steel": 0.9069405052437174,
+      "eval_accuracy_tiles": 0.07601698397479797,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808024400324441,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8887016573545815,
+      "eval_iou_concrete tiles": 0.950679506795068,
+      "eval_iou_dark bricks": 0.8338799008890833,
+      "eval_iou_grass": 0.8367507271971744,
+      "eval_iou_red bricks": 0.8345347862531434,
+      "eval_iou_road": 0.974068639155491,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.32144221498608355,
+      "eval_iou_speedway bricks": 0.9771572437561442,
+      "eval_iou_steel": 0.8487927961295401,
+      "eval_iou_tiles": 0.07600136939404313,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.0886818915605545,
+      "eval_mean_accuracy": 0.6936698192091723,
+      "eval_mean_iou": 0.6556008678417536,
+      "eval_overall_accuracy": 0.9785358640882704,
+      "eval_runtime": 3.8635,
+      "eval_samples_per_second": 9.318,
+      "eval_steps_per_second": 0.518,
+      "step": 4760
+    },
+    {
+      "epoch": 432.82,
+      "learning_rate": 1.4143540669856459e-05,
+      "loss": 0.0295,
+      "step": 4761
+    },
+    {
+      "epoch": 432.91,
+      "learning_rate": 1.4124401913875598e-05,
+      "loss": 0.0335,
+      "step": 4762
+    },
+    {
+      "epoch": 433.0,
+      "learning_rate": 1.4105263157894738e-05,
+      "loss": 0.0277,
+      "step": 4763
+    },
+    {
+      "epoch": 433.09,
+      "learning_rate": 1.4086124401913875e-05,
+      "loss": 0.0288,
+      "step": 4764
+    },
+    {
+      "epoch": 433.18,
+      "learning_rate": 1.4066985645933014e-05,
+      "loss": 0.0336,
+      "step": 4765
+    },
+    {
+      "epoch": 433.27,
+      "learning_rate": 1.4047846889952155e-05,
+      "loss": 0.0277,
+      "step": 4766
+    },
+    {
+      "epoch": 433.36,
+      "learning_rate": 1.4028708133971293e-05,
+      "loss": 0.0307,
+      "step": 4767
+    },
+    {
+      "epoch": 433.45,
+      "learning_rate": 1.400956937799043e-05,
+      "loss": 0.0271,
+      "step": 4768
+    },
+    {
+      "epoch": 433.55,
+      "learning_rate": 1.3990430622009571e-05,
+      "loss": 0.0266,
+      "step": 4769
+    },
+    {
+      "epoch": 433.64,
+      "learning_rate": 1.397129186602871e-05,
+      "loss": 0.0301,
+      "step": 4770
+    },
+    {
+      "epoch": 433.73,
+      "learning_rate": 1.3952153110047847e-05,
+      "loss": 0.0268,
+      "step": 4771
+    },
+    {
+      "epoch": 433.82,
+      "learning_rate": 1.3933014354066987e-05,
+      "loss": 0.0323,
+      "step": 4772
+    },
+    {
+      "epoch": 433.91,
+      "learning_rate": 1.3913875598086126e-05,
+      "loss": 0.028,
+      "step": 4773
+    },
+    {
+      "epoch": 434.0,
+      "learning_rate": 1.3894736842105263e-05,
+      "loss": 0.0324,
+      "step": 4774
+    },
+    {
+      "epoch": 434.09,
+      "learning_rate": 1.3875598086124404e-05,
+      "loss": 0.0272,
+      "step": 4775
+    },
+    {
+      "epoch": 434.18,
+      "learning_rate": 1.3856459330143542e-05,
+      "loss": 0.0272,
+      "step": 4776
+    },
+    {
+      "epoch": 434.27,
+      "learning_rate": 1.383732057416268e-05,
+      "loss": 0.0303,
+      "step": 4777
+    },
+    {
+      "epoch": 434.36,
+      "learning_rate": 1.3818181818181818e-05,
+      "loss": 0.0331,
+      "step": 4778
+    },
+    {
+      "epoch": 434.45,
+      "learning_rate": 1.3799043062200959e-05,
+      "loss": 0.027,
+      "step": 4779
+    },
+    {
+      "epoch": 434.55,
+      "learning_rate": 1.3779904306220096e-05,
+      "loss": 0.0321,
+      "step": 4780
+    },
+    {
+      "epoch": 434.55,
+      "eval_accuracy_NAT": 0.9935923541613878,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9499844413923344,
+      "eval_accuracy_concrete tiles": 0.9787806518681588,
+      "eval_accuracy_dark bricks": 0.8829914168045879,
+      "eval_accuracy_grass": 0.8971155156806482,
+      "eval_accuracy_red bricks": 0.8856534090909091,
+      "eval_accuracy_road": 0.9819876249566196,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.4973396190003584,
+      "eval_accuracy_speedway bricks": 0.9888804009524189,
+      "eval_accuracy_steel": 0.9018287052305257,
+      "eval_accuracy_tiles": 0.08334474729489111,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809577432204597,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8881852518797558,
+      "eval_iou_concrete tiles": 0.9547840610832736,
+      "eval_iou_dark bricks": 0.8353420966391144,
+      "eval_iou_grass": 0.837378640776699,
+      "eval_iou_red bricks": 0.8320266889074228,
+      "eval_iou_road": 0.9741892762416823,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.325332275342194,
+      "eval_iou_speedway bricks": 0.9772513007051388,
+      "eval_iou_steel": 0.8494028297637722,
+      "eval_iou_tiles": 0.0832820091699172,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08831140398979187,
+      "eval_mean_accuracy": 0.6954999143409876,
+      "eval_mean_iou": 0.6567793979791868,
+      "eval_overall_accuracy": 0.978621588812934,
+      "eval_runtime": 3.8396,
+      "eval_samples_per_second": 9.376,
+      "eval_steps_per_second": 0.521,
+      "step": 4780
+    },
+    {
+      "epoch": 434.64,
+      "learning_rate": 1.3760765550239235e-05,
+      "loss": 0.0316,
+      "step": 4781
+    },
+    {
+      "epoch": 434.73,
+      "learning_rate": 1.3741626794258375e-05,
+      "loss": 0.031,
+      "step": 4782
+    },
+    {
+      "epoch": 434.82,
+      "learning_rate": 1.3722488038277512e-05,
+      "loss": 0.0291,
+      "step": 4783
+    },
+    {
+      "epoch": 434.91,
+      "learning_rate": 1.370334928229665e-05,
+      "loss": 0.0284,
+      "step": 4784
+    },
+    {
+      "epoch": 435.0,
+      "learning_rate": 1.3684210526315791e-05,
+      "loss": 0.0302,
+      "step": 4785
+    },
+    {
+      "epoch": 435.09,
+      "learning_rate": 1.3665071770334928e-05,
+      "loss": 0.0269,
+      "step": 4786
+    },
+    {
+      "epoch": 435.18,
+      "learning_rate": 1.3645933014354067e-05,
+      "loss": 0.0309,
+      "step": 4787
+    },
+    {
+      "epoch": 435.27,
+      "learning_rate": 1.3626794258373208e-05,
+      "loss": 0.0287,
+      "step": 4788
+    },
+    {
+      "epoch": 435.36,
+      "learning_rate": 1.3607655502392345e-05,
+      "loss": 0.0331,
+      "step": 4789
+    },
+    {
+      "epoch": 435.45,
+      "learning_rate": 1.3588516746411483e-05,
+      "loss": 0.032,
+      "step": 4790
+    },
+    {
+      "epoch": 435.55,
+      "learning_rate": 1.3569377990430624e-05,
+      "loss": 0.0283,
+      "step": 4791
+    },
+    {
+      "epoch": 435.64,
+      "learning_rate": 1.3550239234449761e-05,
+      "loss": 0.0345,
+      "step": 4792
+    },
+    {
+      "epoch": 435.73,
+      "learning_rate": 1.35311004784689e-05,
+      "loss": 0.0339,
+      "step": 4793
+    },
+    {
+      "epoch": 435.82,
+      "learning_rate": 1.351196172248804e-05,
+      "loss": 0.0245,
+      "step": 4794
+    },
+    {
+      "epoch": 435.91,
+      "learning_rate": 1.3492822966507176e-05,
+      "loss": 0.032,
+      "step": 4795
+    },
+    {
+      "epoch": 436.0,
+      "learning_rate": 1.3473684210526316e-05,
+      "loss": 0.0299,
+      "step": 4796
+    },
+    {
+      "epoch": 436.09,
+      "learning_rate": 1.3454545454545457e-05,
+      "loss": 0.0285,
+      "step": 4797
+    },
+    {
+      "epoch": 436.18,
+      "learning_rate": 1.3435406698564592e-05,
+      "loss": 0.0288,
+      "step": 4798
+    },
+    {
+      "epoch": 436.27,
+      "learning_rate": 1.3416267942583732e-05,
+      "loss": 0.0283,
+      "step": 4799
+    },
+    {
+      "epoch": 436.36,
+      "learning_rate": 1.3397129186602871e-05,
+      "loss": 0.034,
+      "step": 4800
+    },
+    {
+      "epoch": 436.36,
+      "eval_accuracy_NAT": 0.9939838991885366,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.950982338297781,
+      "eval_accuracy_concrete tiles": 0.9787806518681588,
+      "eval_accuracy_dark bricks": 0.8828374581424887,
+      "eval_accuracy_grass": 0.8953819591286082,
+      "eval_accuracy_red bricks": 0.8776633522727273,
+      "eval_accuracy_road": 0.981304421955847,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.3913930471700714,
+      "eval_accuracy_speedway bricks": 0.9877025489152538,
+      "eval_accuracy_steel": 0.9105434997691445,
+      "eval_accuracy_tiles": 0.07197644158334475,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9809212013313454,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.884173167860492,
+      "eval_iou_concrete tiles": 0.9546132283652412,
+      "eval_iou_dark bricks": 0.835591985428051,
+      "eval_iou_grass": 0.8375036777926705,
+      "eval_iou_red bricks": 0.8258980785296575,
+      "eval_iou_road": 0.9741133032126104,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.2621888158380734,
+      "eval_iou_speedway bricks": 0.9772247903461624,
+      "eval_iou_steel": 0.8523555198814522,
+      "eval_iou_tiles": 0.0719616569667922,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09137748926877975,
+      "eval_mean_accuracy": 0.6863499706378432,
+      "eval_mean_iou": 0.650503494273273,
+      "eval_overall_accuracy": 0.9782982932196723,
+      "eval_runtime": 3.9924,
+      "eval_samples_per_second": 9.017,
+      "eval_steps_per_second": 0.501,
+      "step": 4800
+    },
+    {
+      "epoch": 436.45,
+      "learning_rate": 1.3377990430622008e-05,
+      "loss": 0.0267,
+      "step": 4801
+    },
+    {
+      "epoch": 436.55,
+      "learning_rate": 1.3358851674641149e-05,
+      "loss": 0.0299,
+      "step": 4802
+    },
+    {
+      "epoch": 436.64,
+      "learning_rate": 1.3339712918660288e-05,
+      "loss": 0.0284,
+      "step": 4803
+    },
+    {
+      "epoch": 436.73,
+      "learning_rate": 1.3320574162679428e-05,
+      "loss": 0.0326,
+      "step": 4804
+    },
+    {
+      "epoch": 436.82,
+      "learning_rate": 1.3301435406698565e-05,
+      "loss": 0.0305,
+      "step": 4805
+    },
+    {
+      "epoch": 436.91,
+      "learning_rate": 1.3282296650717704e-05,
+      "loss": 0.0317,
+      "step": 4806
+    },
+    {
+      "epoch": 437.0,
+      "learning_rate": 1.3263157894736844e-05,
+      "loss": 0.0297,
+      "step": 4807
+    },
+    {
+      "epoch": 437.09,
+      "learning_rate": 1.3244019138755981e-05,
+      "loss": 0.0285,
+      "step": 4808
+    },
+    {
+      "epoch": 437.18,
+      "learning_rate": 1.322488038277512e-05,
+      "loss": 0.0286,
+      "step": 4809
+    },
+    {
+      "epoch": 437.27,
+      "learning_rate": 1.320574162679426e-05,
+      "loss": 0.0294,
+      "step": 4810
+    },
+    {
+      "epoch": 437.36,
+      "learning_rate": 1.3186602870813398e-05,
+      "loss": 0.0291,
+      "step": 4811
+    },
+    {
+      "epoch": 437.45,
+      "learning_rate": 1.3167464114832537e-05,
+      "loss": 0.027,
+      "step": 4812
+    },
+    {
+      "epoch": 437.55,
+      "learning_rate": 1.3148325358851677e-05,
+      "loss": 0.0282,
+      "step": 4813
+    },
+    {
+      "epoch": 437.64,
+      "learning_rate": 1.3129186602870812e-05,
+      "loss": 0.0302,
+      "step": 4814
+    },
+    {
+      "epoch": 437.73,
+      "learning_rate": 1.3110047846889953e-05,
+      "loss": 0.034,
+      "step": 4815
+    },
+    {
+      "epoch": 437.82,
+      "learning_rate": 1.3090909090909093e-05,
+      "loss": 0.0305,
+      "step": 4816
+    },
+    {
+      "epoch": 437.91,
+      "learning_rate": 1.3071770334928229e-05,
+      "loss": 0.0318,
+      "step": 4817
+    },
+    {
+      "epoch": 438.0,
+      "learning_rate": 1.305263157894737e-05,
+      "loss": 0.0273,
+      "step": 4818
+    },
+    {
+      "epoch": 438.09,
+      "learning_rate": 1.3033492822966508e-05,
+      "loss": 0.0283,
+      "step": 4819
+    },
+    {
+      "epoch": 438.18,
+      "learning_rate": 1.3014354066985645e-05,
+      "loss": 0.0327,
+      "step": 4820
+    },
+    {
+      "epoch": 438.18,
+      "eval_accuracy_NAT": 0.9935360662552768,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9500434568007211,
+      "eval_accuracy_concrete tiles": 0.9839478994679876,
+      "eval_accuracy_dark bricks": 0.8878796043262384,
+      "eval_accuracy_grass": 0.8986428628847639,
+      "eval_accuracy_red bricks": 0.8824573863636364,
+      "eval_accuracy_road": 0.981964649988452,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.438232293992777,
+      "eval_accuracy_speedway bricks": 0.9883916406296198,
+      "eval_accuracy_steel": 0.9060747971769673,
+      "eval_accuracy_tiles": 0.07745514313107793,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.980845442076097,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8862788081715797,
+      "eval_iou_concrete tiles": 0.9559186098321699,
+      "eval_iou_dark bricks": 0.8362212716595374,
+      "eval_iou_grass": 0.8372283095358265,
+      "eval_iou_red bricks": 0.8304093567251462,
+      "eval_iou_road": 0.97401926036273,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.28902869195243464,
+      "eval_iou_speedway bricks": 0.9772805376303028,
+      "eval_iou_steel": 0.8507989594946117,
+      "eval_iou_tiles": 0.07743393126112556,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08980442583560944,
+      "eval_mean_accuracy": 0.6914327539244245,
+      "eval_mean_iou": 0.6534971675924278,
+      "eval_overall_accuracy": 0.9783986409505209,
+      "eval_runtime": 4.1851,
+      "eval_samples_per_second": 8.602,
+      "eval_steps_per_second": 0.478,
+      "step": 4820
+    },
+    {
+      "epoch": 438.27,
+      "learning_rate": 1.2995215311004786e-05,
+      "loss": 0.0283,
+      "step": 4821
+    },
+    {
+      "epoch": 438.36,
+      "learning_rate": 1.2976076555023924e-05,
+      "loss": 0.0294,
+      "step": 4822
+    },
+    {
+      "epoch": 438.45,
+      "learning_rate": 1.2956937799043061e-05,
+      "loss": 0.0249,
+      "step": 4823
+    },
+    {
+      "epoch": 438.55,
+      "learning_rate": 1.2937799043062202e-05,
+      "loss": 0.0299,
+      "step": 4824
+    },
+    {
+      "epoch": 438.64,
+      "learning_rate": 1.291866028708134e-05,
+      "loss": 0.0312,
+      "step": 4825
+    },
+    {
+      "epoch": 438.73,
+      "learning_rate": 1.2899521531100478e-05,
+      "loss": 0.0289,
+      "step": 4826
+    },
+    {
+      "epoch": 438.82,
+      "learning_rate": 1.2880382775119618e-05,
+      "loss": 0.0283,
+      "step": 4827
+    },
+    {
+      "epoch": 438.91,
+      "learning_rate": 1.2861244019138757e-05,
+      "loss": 0.0317,
+      "step": 4828
+    },
+    {
+      "epoch": 439.0,
+      "learning_rate": 1.2842105263157894e-05,
+      "loss": 0.0354,
+      "step": 4829
+    },
+    {
+      "epoch": 439.09,
+      "learning_rate": 1.2822966507177035e-05,
+      "loss": 0.0277,
+      "step": 4830
+    },
+    {
+      "epoch": 439.18,
+      "learning_rate": 1.2803827751196173e-05,
+      "loss": 0.0319,
+      "step": 4831
+    },
+    {
+      "epoch": 439.27,
+      "learning_rate": 1.278468899521531e-05,
+      "loss": 0.03,
+      "step": 4832
+    },
+    {
+      "epoch": 439.36,
+      "learning_rate": 1.276555023923445e-05,
+      "loss": 0.0284,
+      "step": 4833
+    },
+    {
+      "epoch": 439.45,
+      "learning_rate": 1.274641148325359e-05,
+      "loss": 0.0267,
+      "step": 4834
+    },
+    {
+      "epoch": 439.55,
+      "learning_rate": 1.2727272727272727e-05,
+      "loss": 0.0285,
+      "step": 4835
+    },
+    {
+      "epoch": 439.64,
+      "learning_rate": 1.2708133971291866e-05,
+      "loss": 0.0302,
+      "step": 4836
+    },
+    {
+      "epoch": 439.73,
+      "learning_rate": 1.2688995215311006e-05,
+      "loss": 0.0294,
+      "step": 4837
+    },
+    {
+      "epoch": 439.82,
+      "learning_rate": 1.2669856459330145e-05,
+      "loss": 0.0349,
+      "step": 4838
+    },
+    {
+      "epoch": 439.91,
+      "learning_rate": 1.2650717703349282e-05,
+      "loss": 0.0311,
+      "step": 4839
+    },
+    {
+      "epoch": 440.0,
+      "learning_rate": 1.2631578947368422e-05,
+      "loss": 0.0291,
+      "step": 4840
+    },
+    {
+      "epoch": 440.0,
+      "eval_accuracy_NAT": 0.9935316549773684,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9529472831451994,
+      "eval_accuracy_concrete tiles": 0.98287775943252,
+      "eval_accuracy_dark bricks": 0.8857241830568492,
+      "eval_accuracy_grass": 0.8975069639343346,
+      "eval_accuracy_red bricks": 0.875,
+      "eval_accuracy_road": 0.9803684943052309,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.42808700686461004,
+      "eval_accuracy_speedway bricks": 0.9879698585732785,
+      "eval_accuracy_steel": 0.9070476881472198,
+      "eval_accuracy_tiles": 0.0777290782084646,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9808598820870905,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8852350557004325,
+      "eval_iou_concrete tiles": 0.9562708234174203,
+      "eval_iou_dark bricks": 0.8361614766905272,
+      "eval_iou_grass": 0.8373643072127096,
+      "eval_iou_red bricks": 0.8232542599398597,
+      "eval_iou_road": 0.9735476673939487,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.2839327835579367,
+      "eval_iou_speedway bricks": 0.9772873916134296,
+      "eval_iou_steel": 0.8519037626123789,
+      "eval_iou_tiles": 0.07770779131863617,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.09085646271705627,
+      "eval_mean_accuracy": 0.6899069208188521,
+      "eval_mean_iou": 0.6525788616572593,
+      "eval_overall_accuracy": 0.9783368640475802,
+      "eval_runtime": 3.7187,
+      "eval_samples_per_second": 9.681,
+      "eval_steps_per_second": 0.538,
+      "step": 4840
+    },
+    {
+      "epoch": 440.09,
+      "learning_rate": 1.2612440191387561e-05,
+      "loss": 0.0351,
+      "step": 4841
+    },
+    {
+      "epoch": 440.18,
+      "learning_rate": 1.2593301435406698e-05,
+      "loss": 0.0292,
+      "step": 4842
+    },
+    {
+      "epoch": 440.27,
+      "learning_rate": 1.2574162679425839e-05,
+      "loss": 0.0284,
+      "step": 4843
+    },
+    {
+      "epoch": 440.36,
+      "learning_rate": 1.2555023923444977e-05,
+      "loss": 0.0315,
+      "step": 4844
+    },
+    {
+      "epoch": 440.45,
+      "learning_rate": 1.2535885167464115e-05,
+      "loss": 0.0324,
+      "step": 4845
+    },
+    {
+      "epoch": 440.55,
+      "learning_rate": 1.2516746411483255e-05,
+      "loss": 0.0277,
+      "step": 4846
+    },
+    {
+      "epoch": 440.64,
+      "learning_rate": 1.2497607655502392e-05,
+      "loss": 0.0433,
+      "step": 4847
+    },
+    {
+      "epoch": 440.73,
+      "learning_rate": 1.2478468899521533e-05,
+      "loss": 0.0279,
+      "step": 4848
+    },
+    {
+      "epoch": 440.82,
+      "learning_rate": 1.2459330143540671e-05,
+      "loss": 0.0301,
+      "step": 4849
+    },
+    {
+      "epoch": 440.91,
+      "learning_rate": 1.2440191387559808e-05,
+      "loss": 0.0325,
+      "step": 4850
+    },
+    {
+      "epoch": 441.0,
+      "learning_rate": 1.2421052631578949e-05,
+      "loss": 0.0263,
+      "step": 4851
+    },
+    {
+      "epoch": 441.09,
+      "learning_rate": 1.2401913875598086e-05,
+      "loss": 0.0283,
+      "step": 4852
+    },
+    {
+      "epoch": 441.18,
+      "learning_rate": 1.2382775119617225e-05,
+      "loss": 0.0307,
+      "step": 4853
+    },
+    {
+      "epoch": 441.27,
+      "learning_rate": 1.2363636363636365e-05,
+      "loss": 0.0285,
+      "step": 4854
+    },
+    {
+      "epoch": 441.36,
+      "learning_rate": 1.2344497607655502e-05,
+      "loss": 0.0258,
+      "step": 4855
+    },
+    {
+      "epoch": 441.45,
+      "learning_rate": 1.2325358851674641e-05,
+      "loss": 0.0313,
+      "step": 4856
+    },
+    {
+      "epoch": 441.55,
+      "learning_rate": 1.2306220095693782e-05,
+      "loss": 0.031,
+      "step": 4857
+    },
+    {
+      "epoch": 441.64,
+      "learning_rate": 1.2287081339712919e-05,
+      "loss": 0.0262,
+      "step": 4858
+    },
+    {
+      "epoch": 441.73,
+      "learning_rate": 1.2267942583732057e-05,
+      "loss": 0.0297,
+      "step": 4859
+    },
+    {
+      "epoch": 441.82,
+      "learning_rate": 1.2248803827751198e-05,
+      "loss": 0.03,
+      "step": 4860
+    },
+    {
+      "epoch": 441.82,
+      "eval_accuracy_NAT": 0.9931341106122642,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9522645821709086,
+      "eval_accuracy_concrete tiles": 0.983305815446707,
+      "eval_accuracy_dark bricks": 0.8856856933913244,
+      "eval_accuracy_grass": 0.9036303330455723,
+      "eval_accuracy_red bricks": 0.892578125,
+      "eval_accuracy_road": 0.98098156319265,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.5448405149835966,
+      "eval_accuracy_speedway bricks": 0.987243355574945,
+      "eval_accuracy_steel": 0.9102054613811754,
+      "eval_accuracy_tiles": 0.07759211066977126,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.981022816125752,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8889325717490979,
+      "eval_iou_concrete tiles": 0.9564312267657993,
+      "eval_iou_dark bricks": 0.8351843786295006,
+      "eval_iou_grass": 0.8376010781671159,
+      "eval_iou_red bricks": 0.836996336996337,
+      "eval_iou_road": 0.9727804364483146,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.35402335912869015,
+      "eval_iou_speedway bricks": 0.9772272792431193,
+      "eval_iou_steel": 0.8517695530402981,
+      "eval_iou_tiles": 0.07757086128988087,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08871608227491379,
+      "eval_mean_accuracy": 0.7008816665745319,
+      "eval_mean_iou": 0.6591953767372234,
+      "eval_overall_accuracy": 0.978656980726454,
+      "eval_runtime": 3.5134,
+      "eval_samples_per_second": 10.246,
+      "eval_steps_per_second": 0.569,
+      "step": 4860
+    },
+    {
+      "epoch": 441.91,
+      "learning_rate": 1.2229665071770335e-05,
+      "loss": 0.0298,
+      "step": 4861
+    },
+    {
+      "epoch": 442.0,
+      "learning_rate": 1.2210526315789474e-05,
+      "loss": 0.0349,
+      "step": 4862
+    },
+    {
+      "epoch": 442.09,
+      "learning_rate": 1.2191387559808613e-05,
+      "loss": 0.0276,
+      "step": 4863
+    },
+    {
+      "epoch": 442.18,
+      "learning_rate": 1.2172248803827751e-05,
+      "loss": 0.0289,
+      "step": 4864
+    },
+    {
+      "epoch": 442.27,
+      "learning_rate": 1.2153110047846892e-05,
+      "loss": 0.0302,
+      "step": 4865
+    },
+    {
+      "epoch": 442.36,
+      "learning_rate": 1.2133971291866029e-05,
+      "loss": 0.0311,
+      "step": 4866
+    },
+    {
+      "epoch": 442.45,
+      "learning_rate": 1.2114832535885168e-05,
+      "loss": 0.0296,
+      "step": 4867
+    },
+    {
+      "epoch": 442.55,
+      "learning_rate": 1.2095693779904308e-05,
+      "loss": 0.0313,
+      "step": 4868
+    },
+    {
+      "epoch": 442.64,
+      "learning_rate": 1.2076555023923445e-05,
+      "loss": 0.0277,
+      "step": 4869
+    },
+    {
+      "epoch": 442.73,
+      "learning_rate": 1.2057416267942584e-05,
+      "loss": 0.0321,
+      "step": 4870
+    },
+    {
+      "epoch": 442.82,
+      "learning_rate": 1.2038277511961724e-05,
+      "loss": 0.0291,
+      "step": 4871
+    },
+    {
+      "epoch": 442.91,
+      "learning_rate": 1.2019138755980862e-05,
+      "loss": 0.0302,
+      "step": 4872
+    },
+    {
+      "epoch": 443.0,
+      "learning_rate": 1.2e-05,
+      "loss": 0.0252,
+      "step": 4873
+    },
+    {
+      "epoch": 443.09,
+      "learning_rate": 1.1980861244019139e-05,
+      "loss": 0.0303,
+      "step": 4874
+    },
+    {
+      "epoch": 443.18,
+      "learning_rate": 1.1961722488038278e-05,
+      "loss": 0.0272,
+      "step": 4875
+    },
+    {
+      "epoch": 443.27,
+      "learning_rate": 1.1942583732057417e-05,
+      "loss": 0.0317,
+      "step": 4876
+    },
+    {
+      "epoch": 443.36,
+      "learning_rate": 1.1923444976076555e-05,
+      "loss": 0.0285,
+      "step": 4877
+    },
+    {
+      "epoch": 443.45,
+      "learning_rate": 1.1904306220095694e-05,
+      "loss": 0.0313,
+      "step": 4878
+    },
+    {
+      "epoch": 443.55,
+      "learning_rate": 1.1885167464114833e-05,
+      "loss": 0.0293,
+      "step": 4879
+    },
+    {
+      "epoch": 443.64,
+      "learning_rate": 1.1866028708133972e-05,
+      "loss": 0.035,
+      "step": 4880
+    },
+    {
+      "epoch": 443.64,
+      "eval_accuracy_NAT": 0.9934324894499877,
+      "eval_accuracy_REST": 0.0,
+      "eval_accuracy_concrete": 0.9508549186660372,
+      "eval_accuracy_concrete tiles": 0.9852626429401333,
+      "eval_accuracy_dark bricks": 0.8869943420191678,
+      "eval_accuracy_grass": 0.9012362075654365,
+      "eval_accuracy_red bricks": 0.8831676136363636,
+      "eval_accuracy_road": 0.9800649828836487,
+      "eval_accuracy_rough concrete": 0.0,
+      "eval_accuracy_rough red sidewalk": 0.46753783806136795,
+      "eval_accuracy_speedway bricks": 0.9883162146538792,
+      "eval_accuracy_steel": 0.9056543104016885,
+      "eval_accuracy_tiles": 0.07944117244213121,
+      "eval_accuracy_unlabeled": NaN,
+      "eval_iou_NAT": 0.9810547336156806,
+      "eval_iou_REST": 0.0,
+      "eval_iou_concrete": 0.8853818274464158,
+      "eval_iou_concrete tiles": 0.9550964758884377,
+      "eval_iou_dark bricks": 0.8379085917899866,
+      "eval_iou_grass": 0.8385631266239785,
+      "eval_iou_red bricks": 0.8330262937531402,
+      "eval_iou_road": 0.9723973855079975,
+      "eval_iou_rough concrete": 0.0,
+      "eval_iou_rough red sidewalk": 0.30747892303508295,
+      "eval_iou_speedway bricks": 0.9772491053220065,
+      "eval_iou_steel": 0.8513796310649512,
+      "eval_iou_tiles": 0.0794194166780775,
+      "eval_iou_unlabeled": NaN,
+      "eval_loss": 0.08974248915910721,
+      "eval_mean_accuracy": 0.6939971332861418,
+      "eval_mean_iou": 0.6553042700558274,
+      "eval_overall_accuracy": 0.9784123102823893,
+      "eval_runtime": 3.9568,
+      "eval_samples_per_second": 9.098,
+      "eval_steps_per_second": 0.505,
+      "step": 4880
     }
   ],
   "max_steps": 5500,
   "num_train_epochs": 500,
-  "total_flos": 2.5525401194688676e+18,
+  "total_flos": 2.7316666651776123e+18,
   "trial_name": null,
   "trial_params": null
 }

pytorch_model.bin

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

runs/May20_22-12-23_robolidar/events.out.tfevents.1716261149.robolidar.3147080.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:898351be8bc90432f2603653888f2c9c6625ad9df5f7bdef60a12c9c5522e15c
-size 1263186
+oid sha256:1e87d8c5b85371d64cf20b4d131ca1a10af20787ee1d5f4fb8c7b508bd0f6bf0
+size 1284162