{
  "best_metric": 0.18835988640785217,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0-optim/checkpoint-400",
  "epoch": 36.36363636363637,
  "global_step": 400,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.09,
      "learning_rate": 3.6363636363636366e-07,
      "loss": 2.757,
      "step": 1
    },
    {
      "epoch": 0.18,
      "learning_rate": 7.272727272727273e-07,
      "loss": 2.7552,
      "step": 2
    },
    {
      "epoch": 0.27,
      "learning_rate": 1.090909090909091e-06,
      "loss": 2.7635,
      "step": 3
    },
    {
      "epoch": 0.36,
      "learning_rate": 1.4545454545454546e-06,
      "loss": 2.7567,
      "step": 4
    },
    {
      "epoch": 0.45,
      "learning_rate": 1.818181818181818e-06,
      "loss": 2.7588,
      "step": 5
    },
    {
      "epoch": 0.55,
      "learning_rate": 2.181818181818182e-06,
      "loss": 2.7621,
      "step": 6
    },
    {
      "epoch": 0.64,
      "learning_rate": 2.5454545454545456e-06,
      "loss": 2.7553,
      "step": 7
    },
    {
      "epoch": 0.73,
      "learning_rate": 2.9090909090909093e-06,
      "loss": 2.7371,
      "step": 8
    },
    {
      "epoch": 0.82,
      "learning_rate": 3.2727272727272733e-06,
      "loss": 2.7444,
      "step": 9
    },
    {
      "epoch": 0.91,
      "learning_rate": 3.636363636363636e-06,
      "loss": 2.7363,
      "step": 10
    },
    {
      "epoch": 1.0,
      "learning_rate": 4.000000000000001e-06,
      "loss": 2.7364,
      "step": 11
    },
    {
      "epoch": 1.09,
      "learning_rate": 4.363636363636364e-06,
      "loss": 2.7248,
      "step": 12
    },
    {
      "epoch": 1.18,
      "learning_rate": 4.727272727272727e-06,
      "loss": 2.7344,
      "step": 13
    },
    {
      "epoch": 1.27,
      "learning_rate": 5.090909090909091e-06,
      "loss": 2.7259,
      "step": 14
    },
    {
      "epoch": 1.36,
      "learning_rate": 5.4545454545454545e-06,
      "loss": 2.7086,
      "step": 15
    },
    {
      "epoch": 1.45,
      "learning_rate": 5.8181818181818185e-06,
      "loss": 2.7087,
      "step": 16
    },
    {
      "epoch": 1.55,
      "learning_rate": 6.181818181818183e-06,
      "loss": 2.7055,
      "step": 17
    },
    {
      "epoch": 1.64,
      "learning_rate": 6.545454545454547e-06,
      "loss": 2.6959,
      "step": 18
    },
    {
      "epoch": 1.73,
      "learning_rate": 6.909090909090909e-06,
      "loss": 2.6983,
      "step": 19
    },
    {
      "epoch": 1.82,
      "learning_rate": 7.272727272727272e-06,
      "loss": 2.6746,
      "step": 20
    },
    {
      "epoch": 1.82,
      "eval_accuracy_NAT": 0.019056367662015652,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.0010381346838920125,
      "eval_accuracy_concrete tiles": 0.09181801504311135,
      "eval_accuracy_dark bricks": 0.09326045956660636,
      "eval_accuracy_grass": 0.007958282800392846,
      "eval_accuracy_red bricks": 0.00870028409090909,
      "eval_accuracy_road": 0.05730198902763889,
      "eval_accuracy_rough concrete": 0.03398947127489128,
      "eval_accuracy_rough red sidewalk": 0.06001709260331376,
      "eval_accuracy_speedway bricks": 0.03133919121631325,
      "eval_accuracy_steel": 0.00842622518303542,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.018937014149192417,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.0009231352727125269,
      "eval_iou_concrete tiles": 0.002698064724802789,
      "eval_iou_dark bricks": 0.0018963436490879863,
      "eval_iou_grass": 0.006090841001497967,
      "eval_iou_red bricks": 1.3048864535742704e-05,
      "eval_iou_road": 0.0465812133841859,
      "eval_iou_rough concrete": 0.004291938525567382,
      "eval_iou_rough red sidewalk": 0.0013826848729701939,
      "eval_iou_speedway bricks": 0.030977499755458056,
      "eval_iou_steel": 0.0016645601782480097,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 2.647437810897827,
      "eval_mean_accuracy": 0.03176196255016307,
      "eval_mean_iou": 0.008246881741304212,
      "eval_overall_accuracy": 0.02327410380045573,
      "eval_runtime": 3.4336,
      "eval_samples_per_second": 10.485,
      "eval_steps_per_second": 0.582,
      "step": 20
    },
    {
      "epoch": 1.91,
      "learning_rate": 7.636363636363638e-06,
      "loss": 2.6783,
      "step": 21
    },
    {
      "epoch": 2.0,
      "learning_rate": 8.000000000000001e-06,
      "loss": 2.6636,
      "step": 22
    },
    {
      "epoch": 2.09,
      "learning_rate": 8.363636363636365e-06,
      "loss": 2.6626,
      "step": 23
    },
    {
      "epoch": 2.18,
      "learning_rate": 8.727272727272728e-06,
      "loss": 2.6414,
      "step": 24
    },
    {
      "epoch": 2.27,
      "learning_rate": 9.090909090909091e-06,
      "loss": 2.6365,
      "step": 25
    },
    {
      "epoch": 2.36,
      "learning_rate": 9.454545454545454e-06,
      "loss": 2.6263,
      "step": 26
    },
    {
      "epoch": 2.45,
      "learning_rate": 9.818181818181818e-06,
      "loss": 2.6182,
      "step": 27
    },
    {
      "epoch": 2.55,
      "learning_rate": 1.0181818181818182e-05,
      "loss": 2.6144,
      "step": 28
    },
    {
      "epoch": 2.64,
      "learning_rate": 1.0545454545454546e-05,
      "loss": 2.5878,
      "step": 29
    },
    {
      "epoch": 2.73,
      "learning_rate": 1.0909090909090909e-05,
      "loss": 2.5814,
      "step": 30
    },
    {
      "epoch": 2.82,
      "learning_rate": 1.1272727272727274e-05,
      "loss": 2.5739,
      "step": 31
    },
    {
      "epoch": 2.91,
      "learning_rate": 1.1636363636363637e-05,
      "loss": 2.5536,
      "step": 32
    },
    {
      "epoch": 3.0,
      "learning_rate": 1.2e-05,
      "loss": 2.539,
      "step": 33
    },
    {
      "epoch": 3.09,
      "learning_rate": 1.2363636363636365e-05,
      "loss": 2.5256,
      "step": 34
    },
    {
      "epoch": 3.18,
      "learning_rate": 1.2727272727272727e-05,
      "loss": 2.4979,
      "step": 35
    },
    {
      "epoch": 3.27,
      "learning_rate": 1.3090909090909093e-05,
      "loss": 2.4936,
      "step": 36
    },
    {
      "epoch": 3.36,
      "learning_rate": 1.3454545454545457e-05,
      "loss": 2.4862,
      "step": 37
    },
    {
      "epoch": 3.45,
      "learning_rate": 1.3818181818181818e-05,
      "loss": 2.4491,
      "step": 38
    },
    {
      "epoch": 3.55,
      "learning_rate": 1.4181818181818181e-05,
      "loss": 2.4468,
      "step": 39
    },
    {
      "epoch": 3.64,
      "learning_rate": 1.4545454545454545e-05,
      "loss": 2.4385,
      "step": 40
    },
    {
      "epoch": 3.64,
      "eval_accuracy_NAT": 0.447892573737157,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.04360568050130907,
      "eval_accuracy_concrete tiles": 0.10279459426404941,
      "eval_accuracy_dark bricks": 0.07266848851083484,
      "eval_accuracy_grass": 0.017545269942016727,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.37485716220448445,
      "eval_accuracy_rough concrete": 0.2949759670405127,
      "eval_accuracy_rough red sidewalk": 0.05480660546411932,
      "eval_accuracy_speedway bricks": 0.7969013802350152,
      "eval_accuracy_steel": 0.0050211067871512435,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.4472536503231145,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.039962583094048695,
      "eval_iou_concrete tiles": 0.0041023463419931645,
      "eval_iou_dark bricks": 0.007329448622040366,
      "eval_iou_grass": 0.010573184550513284,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.2852347789122688,
      "eval_iou_rough concrete": 0.048997709321445884,
      "eval_iou_rough red sidewalk": 0.003679975417468036,
      "eval_iou_speedway bricks": 0.7166237181229831,
      "eval_iou_steel": 0.0015635711039736272,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 2.4373269081115723,
      "eval_mean_accuracy": 0.17008221759128078,
      "eval_mean_iou": 0.11180864041498925,
      "eval_overall_accuracy": 0.4471186531914605,
      "eval_runtime": 3.2402,
      "eval_samples_per_second": 11.111,
      "eval_steps_per_second": 0.617,
      "step": 40
    },
    {
      "epoch": 3.73,
      "learning_rate": 1.4909090909090908e-05,
      "loss": 2.4028,
      "step": 41
    },
    {
      "epoch": 3.82,
      "learning_rate": 1.5272727272727276e-05,
      "loss": 2.4083,
      "step": 42
    },
    {
      "epoch": 3.91,
      "learning_rate": 1.563636363636364e-05,
      "loss": 2.3613,
      "step": 43
    },
    {
      "epoch": 4.0,
      "learning_rate": 1.6000000000000003e-05,
      "loss": 2.3597,
      "step": 44
    },
    {
      "epoch": 4.09,
      "learning_rate": 1.6363636363636366e-05,
      "loss": 2.3331,
      "step": 45
    },
    {
      "epoch": 4.18,
      "learning_rate": 1.672727272727273e-05,
      "loss": 2.3068,
      "step": 46
    },
    {
      "epoch": 4.27,
      "learning_rate": 1.7090909090909092e-05,
      "loss": 2.2492,
      "step": 47
    },
    {
      "epoch": 4.36,
      "learning_rate": 1.7454545454545456e-05,
      "loss": 2.2469,
      "step": 48
    },
    {
      "epoch": 4.45,
      "learning_rate": 1.781818181818182e-05,
      "loss": 2.2194,
      "step": 49
    },
    {
      "epoch": 4.55,
      "learning_rate": 1.8181818181818182e-05,
      "loss": 2.2206,
      "step": 50
    },
    {
      "epoch": 4.64,
      "learning_rate": 1.8545454545454545e-05,
      "loss": 2.1524,
      "step": 51
    },
    {
      "epoch": 4.73,
      "learning_rate": 1.890909090909091e-05,
      "loss": 2.1454,
      "step": 52
    },
    {
      "epoch": 4.82,
      "learning_rate": 1.9272727272727272e-05,
      "loss": 2.1673,
      "step": 53
    },
    {
      "epoch": 4.91,
      "learning_rate": 1.9636363636363635e-05,
      "loss": 2.1274,
      "step": 54
    },
    {
      "epoch": 5.0,
      "learning_rate": 2e-05,
      "loss": 2.0958,
      "step": 55
    },
    {
      "epoch": 5.09,
      "learning_rate": 2.0363636363636365e-05,
      "loss": 2.0027,
      "step": 56
    },
    {
      "epoch": 5.18,
      "learning_rate": 2.0727272727272728e-05,
      "loss": 2.0081,
      "step": 57
    },
    {
      "epoch": 5.27,
      "learning_rate": 2.109090909090909e-05,
      "loss": 2.016,
      "step": 58
    },
    {
      "epoch": 5.36,
      "learning_rate": 2.1454545454545455e-05,
      "loss": 1.9957,
      "step": 59
    },
    {
      "epoch": 5.45,
      "learning_rate": 2.1818181818181818e-05,
      "loss": 1.9773,
      "step": 60
    },
    {
      "epoch": 5.45,
      "eval_accuracy_NAT": 0.898956856290447,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.24511379243744366,
      "eval_accuracy_concrete tiles": 0.007154650522839846,
      "eval_accuracy_dark bricks": 0.009314499057003194,
      "eval_accuracy_grass": 0.07155464372966304,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.7720496210944066,
      "eval_accuracy_rough concrete": 0.0036621652552071412,
      "eval_accuracy_rough red sidewalk": 0.010007443553055993,
      "eval_accuracy_speedway bricks": 0.9787968531076106,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.0009587727708533077,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.89400043378223,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.19197997304386832,
      "eval_iou_concrete tiles": 0.0013976657786909726,
      "eval_iou_dark bricks": 0.005338039042682255,
      "eval_iou_grass": 0.07039652848458174,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.5405843619796408,
      "eval_iou_rough concrete": 0.0013066024253807521,
      "eval_iou_rough red sidewalk": 0.003800927719548077,
      "eval_iou_speedway bricks": 0.6998719868699171,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.00034965034965034965,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.916090488433838,
      "eval_mean_accuracy": 0.23058225367834848,
      "eval_mean_iou": 0.1720732978197279,
      "eval_overall_accuracy": 0.8010207282172309,
      "eval_runtime": 3.5175,
      "eval_samples_per_second": 10.235,
      "eval_steps_per_second": 0.569,
      "step": 60
    },
    {
      "epoch": 5.55,
      "learning_rate": 2.2181818181818184e-05,
      "loss": 1.9281,
      "step": 61
    },
    {
      "epoch": 5.64,
      "learning_rate": 2.2545454545454548e-05,
      "loss": 1.8834,
      "step": 62
    },
    {
      "epoch": 5.73,
      "learning_rate": 2.290909090909091e-05,
      "loss": 1.8846,
      "step": 63
    },
    {
      "epoch": 5.82,
      "learning_rate": 2.3272727272727274e-05,
      "loss": 1.8532,
      "step": 64
    },
    {
      "epoch": 5.91,
      "learning_rate": 2.3636363636363637e-05,
      "loss": 1.8357,
      "step": 65
    },
    {
      "epoch": 6.0,
      "learning_rate": 2.4e-05,
      "loss": 1.8409,
      "step": 66
    },
    {
      "epoch": 6.09,
      "learning_rate": 2.4363636363636364e-05,
      "loss": 1.7691,
      "step": 67
    },
    {
      "epoch": 6.18,
      "learning_rate": 2.472727272727273e-05,
      "loss": 1.7549,
      "step": 68
    },
    {
      "epoch": 6.27,
      "learning_rate": 2.5090909090909094e-05,
      "loss": 1.8554,
      "step": 69
    },
    {
      "epoch": 6.36,
      "learning_rate": 2.5454545454545454e-05,
      "loss": 1.7217,
      "step": 70
    },
    {
      "epoch": 6.45,
      "learning_rate": 2.581818181818182e-05,
      "loss": 1.7575,
      "step": 71
    },
    {
      "epoch": 6.55,
      "learning_rate": 2.6181818181818187e-05,
      "loss": 1.7587,
      "step": 72
    },
    {
      "epoch": 6.64,
      "learning_rate": 2.6545454545454547e-05,
      "loss": 1.6981,
      "step": 73
    },
    {
      "epoch": 6.73,
      "learning_rate": 2.6909090909090913e-05,
      "loss": 1.6573,
      "step": 74
    },
    {
      "epoch": 6.82,
      "learning_rate": 2.7272727272727273e-05,
      "loss": 1.6428,
      "step": 75
    },
    {
      "epoch": 6.91,
      "learning_rate": 2.7636363636363636e-05,
      "loss": 1.6877,
      "step": 76
    },
    {
      "epoch": 7.0,
      "learning_rate": 2.8000000000000003e-05,
      "loss": 1.7226,
      "step": 77
    },
    {
      "epoch": 7.09,
      "learning_rate": 2.8363636363636363e-05,
      "loss": 1.6332,
      "step": 78
    },
    {
      "epoch": 7.18,
      "learning_rate": 2.872727272727273e-05,
      "loss": 1.6322,
      "step": 79
    },
    {
      "epoch": 7.27,
      "learning_rate": 2.909090909090909e-05,
      "loss": 1.6038,
      "step": 80
    },
    {
      "epoch": 7.27,
      "eval_accuracy_NAT": 0.9470286690715778,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.7309205867204601,
      "eval_accuracy_concrete tiles": 0.0014064697609001407,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.029533372711163616,
      "eval_accuracy_red bricks": 0.015447443181818182,
      "eval_accuracy_road": 0.8218001008480181,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0020400849116422684,
      "eval_accuracy_speedway bricks": 0.9799789289994171,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.937594640128183,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.4290493301504721,
      "eval_iou_concrete tiles": 0.0011021395883748233,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.029093488223162548,
      "eval_iou_red bricks": 0.002611828279795857,
      "eval_iou_road": 0.7049759862244951,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0009831927190593237,
      "eval_iou_speedway bricks": 0.7986861621141007,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.4904648065567017,
      "eval_mean_accuracy": 0.27139658893884594,
      "eval_mean_iou": 0.2074354833876888,
      "eval_overall_accuracy": 0.8714875115288628,
      "eval_runtime": 3.8608,
      "eval_samples_per_second": 9.325,
      "eval_steps_per_second": 0.518,
      "step": 80
    },
    {
      "epoch": 7.36,
      "learning_rate": 2.9454545454545456e-05,
      "loss": 1.6326,
      "step": 81
    },
    {
      "epoch": 7.45,
      "learning_rate": 2.9818181818181816e-05,
      "loss": 1.5668,
      "step": 82
    },
    {
      "epoch": 7.55,
      "learning_rate": 3.0181818181818182e-05,
      "loss": 1.5495,
      "step": 83
    },
    {
      "epoch": 7.64,
      "learning_rate": 3.054545454545455e-05,
      "loss": 1.5698,
      "step": 84
    },
    {
      "epoch": 7.73,
      "learning_rate": 3.090909090909091e-05,
      "loss": 1.4842,
      "step": 85
    },
    {
      "epoch": 7.82,
      "learning_rate": 3.127272727272728e-05,
      "loss": 1.4979,
      "step": 86
    },
    {
      "epoch": 7.91,
      "learning_rate": 3.1636363636363635e-05,
      "loss": 1.5206,
      "step": 87
    },
    {
      "epoch": 8.0,
      "learning_rate": 3.2000000000000005e-05,
      "loss": 1.555,
      "step": 88
    },
    {
      "epoch": 8.09,
      "learning_rate": 3.236363636363636e-05,
      "loss": 1.47,
      "step": 89
    },
    {
      "epoch": 8.18,
      "learning_rate": 3.272727272727273e-05,
      "loss": 1.5321,
      "step": 90
    },
    {
      "epoch": 8.27,
      "learning_rate": 3.3090909090909095e-05,
      "loss": 1.5188,
      "step": 91
    },
    {
      "epoch": 8.36,
      "learning_rate": 3.345454545454546e-05,
      "loss": 1.4527,
      "step": 92
    },
    {
      "epoch": 8.45,
      "learning_rate": 3.381818181818182e-05,
      "loss": 1.4205,
      "step": 93
    },
    {
      "epoch": 8.55,
      "learning_rate": 3.4181818181818185e-05,
      "loss": 1.4396,
      "step": 94
    },
    {
      "epoch": 8.64,
      "learning_rate": 3.454545454545455e-05,
      "loss": 1.4194,
      "step": 95
    },
    {
      "epoch": 8.73,
      "learning_rate": 3.490909090909091e-05,
      "loss": 1.4087,
      "step": 96
    },
    {
      "epoch": 8.82,
      "learning_rate": 3.5272727272727274e-05,
      "loss": 1.4411,
      "step": 97
    },
    {
      "epoch": 8.91,
      "learning_rate": 3.563636363636364e-05,
      "loss": 1.3689,
      "step": 98
    },
    {
      "epoch": 9.0,
      "learning_rate": 3.6e-05,
      "loss": 1.3591,
      "step": 99
    },
    {
      "epoch": 9.09,
      "learning_rate": 3.6363636363636364e-05,
      "loss": 1.4392,
      "step": 100
    },
    {
      "epoch": 9.09,
      "eval_accuracy_NAT": 0.9504804058093353,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.76187819219709,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.07590950555192456,
      "eval_accuracy_red bricks": 0.0005326704545454545,
      "eval_accuracy_road": 0.8848434134998494,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0008821988807101701,
      "eval_accuracy_speedway bricks": 0.9805213926169435,
      "eval_accuracy_steel": 0.0003380383879691313,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9410302391336439,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.44589299171062546,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.06707307657862148,
      "eval_iou_red bricks": 0.0003791229622140781,
      "eval_iou_road": 0.7530169393154503,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0006904289289721239,
      "eval_iou_speedway bricks": 0.8219766311954673,
      "eval_iou_steel": 0.00033548260399960723,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.2885687351226807,
      "eval_mean_accuracy": 0.281183524415259,
      "eval_mean_iou": 0.21645677945921385,
      "eval_overall_accuracy": 0.883018069797092,
      "eval_runtime": 3.8233,
      "eval_samples_per_second": 9.416,
      "eval_steps_per_second": 0.523,
      "step": 100
    },
    {
      "epoch": 9.18,
      "learning_rate": 3.672727272727273e-05,
      "loss": 1.3807,
      "step": 101
    },
    {
      "epoch": 9.27,
      "learning_rate": 3.709090909090909e-05,
      "loss": 1.3464,
      "step": 102
    },
    {
      "epoch": 9.36,
      "learning_rate": 3.745454545454546e-05,
      "loss": 1.377,
      "step": 103
    },
    {
      "epoch": 9.45,
      "learning_rate": 3.781818181818182e-05,
      "loss": 1.314,
      "step": 104
    },
    {
      "epoch": 9.55,
      "learning_rate": 3.818181818181819e-05,
      "loss": 1.3202,
      "step": 105
    },
    {
      "epoch": 9.64,
      "learning_rate": 3.8545454545454544e-05,
      "loss": 1.2905,
      "step": 106
    },
    {
      "epoch": 9.73,
      "learning_rate": 3.8909090909090914e-05,
      "loss": 1.3153,
      "step": 107
    },
    {
      "epoch": 9.82,
      "learning_rate": 3.927272727272727e-05,
      "loss": 1.3037,
      "step": 108
    },
    {
      "epoch": 9.91,
      "learning_rate": 3.963636363636364e-05,
      "loss": 1.3246,
      "step": 109
    },
    {
      "epoch": 10.0,
      "learning_rate": 4e-05,
      "loss": 1.3242,
      "step": 110
    },
    {
      "epoch": 10.09,
      "learning_rate": 4.0363636363636367e-05,
      "loss": 1.2856,
      "step": 111
    },
    {
      "epoch": 10.18,
      "learning_rate": 4.072727272727273e-05,
      "loss": 1.2349,
      "step": 112
    },
    {
      "epoch": 10.27,
      "learning_rate": 4.109090909090909e-05,
      "loss": 1.2786,
      "step": 113
    },
    {
      "epoch": 10.36,
      "learning_rate": 4.1454545454545456e-05,
      "loss": 1.2887,
      "step": 114
    },
    {
      "epoch": 10.45,
      "learning_rate": 4.181818181818182e-05,
      "loss": 1.2777,
      "step": 115
    },
    {
      "epoch": 10.55,
      "learning_rate": 4.218181818181818e-05,
      "loss": 1.2586,
      "step": 116
    },
    {
      "epoch": 10.64,
      "learning_rate": 4.254545454545455e-05,
      "loss": 1.2073,
      "step": 117
    },
    {
      "epoch": 10.73,
      "learning_rate": 4.290909090909091e-05,
      "loss": 1.2722,
      "step": 118
    },
    {
      "epoch": 10.82,
      "learning_rate": 4.327272727272728e-05,
      "loss": 1.2427,
      "step": 119
    },
    {
      "epoch": 10.91,
      "learning_rate": 4.3636363636363636e-05,
      "loss": 1.1951,
      "step": 120
    },
    {
      "epoch": 10.91,
      "eval_accuracy_NAT": 0.9539229670890436,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8772452680372548,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.14066972602117317,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.8862182839633513,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.981366766953044,
      "eval_accuracy_steel": 0.00023085548446672384,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9437268461219142,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.5137735030918505,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.1162642232152916,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8261045014174364,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.844953008348876,
      "eval_iou_steel": 0.0002300380384327837,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 1.1216386556625366,
      "eval_mean_accuracy": 0.2953579898114103,
      "eval_mean_iou": 0.24961939386413856,
      "eval_overall_accuracy": 0.8964270485772027,
      "eval_runtime": 3.543,
      "eval_samples_per_second": 10.161,
      "eval_steps_per_second": 0.564,
      "step": 120
    },
    {
      "epoch": 11.0,
      "learning_rate": 4.4000000000000006e-05,
      "loss": 1.1897,
      "step": 121
    },
    {
      "epoch": 11.09,
      "learning_rate": 4.436363636363637e-05,
      "loss": 1.163,
      "step": 122
    },
    {
      "epoch": 11.18,
      "learning_rate": 4.472727272727273e-05,
      "loss": 1.1332,
      "step": 123
    },
    {
      "epoch": 11.27,
      "learning_rate": 4.5090909090909095e-05,
      "loss": 1.2278,
      "step": 124
    },
    {
      "epoch": 11.36,
      "learning_rate": 4.545454545454546e-05,
      "loss": 1.1608,
      "step": 125
    },
    {
      "epoch": 11.45,
      "learning_rate": 4.581818181818182e-05,
      "loss": 1.2089,
      "step": 126
    },
    {
      "epoch": 11.55,
      "learning_rate": 4.618181818181818e-05,
      "loss": 1.1246,
      "step": 127
    },
    {
      "epoch": 11.64,
      "learning_rate": 4.654545454545455e-05,
      "loss": 1.1045,
      "step": 128
    },
    {
      "epoch": 11.73,
      "learning_rate": 4.690909090909091e-05,
      "loss": 1.1625,
      "step": 129
    },
    {
      "epoch": 11.82,
      "learning_rate": 4.7272727272727275e-05,
      "loss": 1.2094,
      "step": 130
    },
    {
      "epoch": 11.91,
      "learning_rate": 4.763636363636364e-05,
      "loss": 1.1249,
      "step": 131
    },
    {
      "epoch": 12.0,
      "learning_rate": 4.8e-05,
      "loss": 1.0953,
      "step": 132
    },
    {
      "epoch": 12.09,
      "learning_rate": 4.8363636363636364e-05,
      "loss": 1.1209,
      "step": 133
    },
    {
      "epoch": 12.18,
      "learning_rate": 4.872727272727273e-05,
      "loss": 1.1054,
      "step": 134
    },
    {
      "epoch": 12.27,
      "learning_rate": 4.909090909090909e-05,
      "loss": 1.0896,
      "step": 135
    },
    {
      "epoch": 12.36,
      "learning_rate": 4.945454545454546e-05,
      "loss": 1.0754,
      "step": 136
    },
    {
      "epoch": 12.45,
      "learning_rate": 4.981818181818182e-05,
      "loss": 1.0641,
      "step": 137
    },
    {
      "epoch": 12.55,
      "learning_rate": 5.018181818181819e-05,
      "loss": 1.0827,
      "step": 138
    },
    {
      "epoch": 12.64,
      "learning_rate": 5.054545454545455e-05,
      "loss": 1.1269,
      "step": 139
    },
    {
      "epoch": 12.73,
      "learning_rate": 5.090909090909091e-05,
      "loss": 1.0666,
      "step": 140
    },
    {
      "epoch": 12.73,
      "eval_accuracy_NAT": 0.9561173131717935,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8869868341988927,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.3516009185053667,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9451321483892734,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.975032795214252,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9454440385038502,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.60397101197788,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.27263931573897915,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.805971688307403,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8606970792130627,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.9632436037063599,
      "eval_mean_accuracy": 0.31652846226765985,
      "eval_mean_iou": 0.2683633179800904,
      "eval_overall_accuracy": 0.9089568456013998,
      "eval_runtime": 3.8233,
      "eval_samples_per_second": 9.416,
      "eval_steps_per_second": 0.523,
      "step": 140
    },
    {
      "epoch": 12.82,
      "learning_rate": 5.127272727272727e-05,
      "loss": 1.025,
      "step": 141
    },
    {
      "epoch": 12.91,
      "learning_rate": 5.163636363636364e-05,
      "loss": 1.057,
      "step": 142
    },
    {
      "epoch": 13.0,
      "learning_rate": 5.2000000000000004e-05,
      "loss": 1.0304,
      "step": 143
    },
    {
      "epoch": 13.09,
      "learning_rate": 5.2363636363636374e-05,
      "loss": 1.0345,
      "step": 144
    },
    {
      "epoch": 13.18,
      "learning_rate": 5.272727272727272e-05,
      "loss": 1.0249,
      "step": 145
    },
    {
      "epoch": 13.27,
      "learning_rate": 5.309090909090909e-05,
      "loss": 1.0423,
      "step": 146
    },
    {
      "epoch": 13.36,
      "learning_rate": 5.3454545454545457e-05,
      "loss": 0.961,
      "step": 147
    },
    {
      "epoch": 13.45,
      "learning_rate": 5.3818181818181827e-05,
      "loss": 1.0416,
      "step": 148
    },
    {
      "epoch": 13.55,
      "learning_rate": 5.418181818181819e-05,
      "loss": 1.0241,
      "step": 149
    },
    {
      "epoch": 13.64,
      "learning_rate": 5.4545454545454546e-05,
      "loss": 0.9978,
      "step": 150
    },
    {
      "epoch": 13.73,
      "learning_rate": 5.490909090909091e-05,
      "loss": 1.0115,
      "step": 151
    },
    {
      "epoch": 13.82,
      "learning_rate": 5.527272727272727e-05,
      "loss": 0.9345,
      "step": 152
    },
    {
      "epoch": 13.91,
      "learning_rate": 5.563636363636364e-05,
      "loss": 0.9835,
      "step": 153
    },
    {
      "epoch": 14.0,
      "learning_rate": 5.6000000000000006e-05,
      "loss": 0.9658,
      "step": 154
    },
    {
      "epoch": 14.09,
      "learning_rate": 5.636363636363636e-05,
      "loss": 0.8986,
      "step": 155
    },
    {
      "epoch": 14.18,
      "learning_rate": 5.6727272727272726e-05,
      "loss": 0.9494,
      "step": 156
    },
    {
      "epoch": 14.27,
      "learning_rate": 5.7090909090909096e-05,
      "loss": 0.9374,
      "step": 157
    },
    {
      "epoch": 14.36,
      "learning_rate": 5.745454545454546e-05,
      "loss": 0.9526,
      "step": 158
    },
    {
      "epoch": 14.45,
      "learning_rate": 5.7818181818181815e-05,
      "loss": 0.9791,
      "step": 159
    },
    {
      "epoch": 14.55,
      "learning_rate": 5.818181818181818e-05,
      "loss": 0.9858,
      "step": 160
    },
    {
      "epoch": 14.55,
      "eval_accuracy_NAT": 0.9542294626781186,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.9107794325936736,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.4873705512080722,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.8653860338796135,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9734096282163145,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9443239861289295,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.5657941221562315,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.35750086526259117,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8179847069984342,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8611517704146181,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.8469406962394714,
      "eval_mean_accuracy": 0.3223980852750609,
      "eval_mean_iou": 0.27282734238160034,
      "eval_overall_accuracy": 0.9065458509657118,
      "eval_runtime": 3.5282,
      "eval_samples_per_second": 10.203,
      "eval_steps_per_second": 0.567,
      "step": 160
    },
    {
      "epoch": 14.64,
      "learning_rate": 5.854545454545455e-05,
      "loss": 0.9323,
      "step": 161
    },
    {
      "epoch": 14.73,
      "learning_rate": 5.890909090909091e-05,
      "loss": 0.8957,
      "step": 162
    },
    {
      "epoch": 14.82,
      "learning_rate": 5.927272727272728e-05,
      "loss": 0.9079,
      "step": 163
    },
    {
      "epoch": 14.91,
      "learning_rate": 5.963636363636363e-05,
      "loss": 0.8696,
      "step": 164
    },
    {
      "epoch": 15.0,
      "learning_rate": 6e-05,
      "loss": 0.8968,
      "step": 165
    },
    {
      "epoch": 15.09,
      "learning_rate": 6.0363636363636365e-05,
      "loss": 0.8978,
      "step": 166
    },
    {
      "epoch": 15.18,
      "learning_rate": 6.0727272727272735e-05,
      "loss": 0.9059,
      "step": 167
    },
    {
      "epoch": 15.27,
      "learning_rate": 6.10909090909091e-05,
      "loss": 0.8761,
      "step": 168
    },
    {
      "epoch": 15.36,
      "learning_rate": 6.145454545454545e-05,
      "loss": 0.905,
      "step": 169
    },
    {
      "epoch": 15.45,
      "learning_rate": 6.181818181818182e-05,
      "loss": 0.8571,
      "step": 170
    },
    {
      "epoch": 15.55,
      "learning_rate": 6.218181818181819e-05,
      "loss": 0.8471,
      "step": 171
    },
    {
      "epoch": 15.64,
      "learning_rate": 6.254545454545456e-05,
      "loss": 0.8319,
      "step": 172
    },
    {
      "epoch": 15.73,
      "learning_rate": 6.290909090909091e-05,
      "loss": 0.861,
      "step": 173
    },
    {
      "epoch": 15.82,
      "learning_rate": 6.327272727272727e-05,
      "loss": 0.8161,
      "step": 174
    },
    {
      "epoch": 15.91,
      "learning_rate": 6.363636363636364e-05,
      "loss": 0.8789,
      "step": 175
    },
    {
      "epoch": 16.0,
      "learning_rate": 6.400000000000001e-05,
      "loss": 0.7966,
      "step": 176
    },
    {
      "epoch": 16.09,
      "learning_rate": 6.436363636363637e-05,
      "loss": 0.8516,
      "step": 177
    },
    {
      "epoch": 16.18,
      "learning_rate": 6.472727272727272e-05,
      "loss": 0.8139,
      "step": 178
    },
    {
      "epoch": 16.27,
      "learning_rate": 6.50909090909091e-05,
      "loss": 0.8312,
      "step": 179
    },
    {
      "epoch": 16.36,
      "learning_rate": 6.545454545454546e-05,
      "loss": 0.8171,
      "step": 180
    },
    {
      "epoch": 16.36,
      "eval_accuracy_NAT": 0.9580538641735771,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8946829799562213,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7175526095967734,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9461273272735847,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9772442848229662,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9459735019378309,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.6848531827515401,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.48368507602636773,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8659028381838987,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8638598572414595,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.7246155738830566,
      "eval_mean_accuracy": 0.34566623583254785,
      "eval_mean_iou": 0.2957134197031613,
      "eval_overall_accuracy": 0.9222983254326714,
      "eval_runtime": 3.4993,
      "eval_samples_per_second": 10.288,
      "eval_steps_per_second": 0.572,
      "step": 180
    },
    {
      "epoch": 16.45,
      "learning_rate": 6.581818181818182e-05,
      "loss": 0.8185,
      "step": 181
    },
    {
      "epoch": 16.55,
      "learning_rate": 6.618181818181819e-05,
      "loss": 0.8349,
      "step": 182
    },
    {
      "epoch": 16.64,
      "learning_rate": 6.654545454545455e-05,
      "loss": 0.7596,
      "step": 183
    },
    {
      "epoch": 16.73,
      "learning_rate": 6.690909090909092e-05,
      "loss": 0.7586,
      "step": 184
    },
    {
      "epoch": 16.82,
      "learning_rate": 6.727272727272727e-05,
      "loss": 0.7928,
      "step": 185
    },
    {
      "epoch": 16.91,
      "learning_rate": 6.763636363636364e-05,
      "loss": 0.7646,
      "step": 186
    },
    {
      "epoch": 17.0,
      "learning_rate": 6.800000000000001e-05,
      "loss": 0.7323,
      "step": 187
    },
    {
      "epoch": 17.09,
      "learning_rate": 6.836363636363637e-05,
      "loss": 0.7875,
      "step": 188
    },
    {
      "epoch": 17.18,
      "learning_rate": 6.872727272727273e-05,
      "loss": 0.8039,
      "step": 189
    },
    {
      "epoch": 17.27,
      "learning_rate": 6.90909090909091e-05,
      "loss": 0.8027,
      "step": 190
    },
    {
      "epoch": 17.36,
      "learning_rate": 6.945454545454547e-05,
      "loss": 0.7244,
      "step": 191
    },
    {
      "epoch": 17.45,
      "learning_rate": 6.981818181818182e-05,
      "loss": 0.7035,
      "step": 192
    },
    {
      "epoch": 17.55,
      "learning_rate": 7.018181818181818e-05,
      "loss": 0.6842,
      "step": 193
    },
    {
      "epoch": 17.64,
      "learning_rate": 7.054545454545455e-05,
      "loss": 0.6885,
      "step": 194
    },
    {
      "epoch": 17.73,
      "learning_rate": 7.090909090909092e-05,
      "loss": 0.7147,
      "step": 195
    },
    {
      "epoch": 17.82,
      "learning_rate": 7.127272727272728e-05,
      "loss": 0.7371,
      "step": 196
    },
    {
      "epoch": 17.91,
      "learning_rate": 7.163636363636363e-05,
      "loss": 0.6814,
      "step": 197
    },
    {
      "epoch": 18.0,
      "learning_rate": 7.2e-05,
      "loss": 0.6467,
      "step": 198
    },
    {
      "epoch": 18.09,
      "learning_rate": 7.236363636363637e-05,
      "loss": 0.639,
      "step": 199
    },
    {
      "epoch": 18.18,
      "learning_rate": 7.272727272727273e-05,
      "loss": 0.7362,
      "step": 200
    },
    {
      "epoch": 18.18,
      "eval_accuracy_NAT": 0.9589253562371588,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8930412785956479,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7201005183194288,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9467536974583639,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9858386222027523,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9477355622967787,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.6862279892606659,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.5313936273928227,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.861003861003861,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8618114030437445,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.6129858493804932,
      "eval_mean_accuracy": 0.3465122671394886,
      "eval_mean_iou": 0.29909018792291325,
      "eval_overall_accuracy": 0.9243333604600694,
      "eval_runtime": 3.4286,
      "eval_samples_per_second": 10.5,
      "eval_steps_per_second": 0.583,
      "step": 200
    },
    {
      "epoch": 18.27,
      "learning_rate": 7.30909090909091e-05,
      "loss": 0.7156,
      "step": 201
    },
    {
      "epoch": 18.36,
      "learning_rate": 7.345454545454545e-05,
      "loss": 0.6647,
      "step": 202
    },
    {
      "epoch": 18.45,
      "learning_rate": 7.381818181818182e-05,
      "loss": 0.7283,
      "step": 203
    },
    {
      "epoch": 18.55,
      "learning_rate": 7.418181818181818e-05,
      "loss": 0.6373,
      "step": 204
    },
    {
      "epoch": 18.64,
      "learning_rate": 7.454545454545455e-05,
      "loss": 0.661,
      "step": 205
    },
    {
      "epoch": 18.73,
      "learning_rate": 7.490909090909092e-05,
      "loss": 0.6538,
      "step": 206
    },
    {
      "epoch": 18.82,
      "learning_rate": 7.527272727272728e-05,
      "loss": 0.6686,
      "step": 207
    },
    {
      "epoch": 18.91,
      "learning_rate": 7.563636363636363e-05,
      "loss": 0.6689,
      "step": 208
    },
    {
      "epoch": 19.0,
      "learning_rate": 7.6e-05,
      "loss": 0.5993,
      "step": 209
    },
    {
      "epoch": 19.09,
      "learning_rate": 7.636363636363637e-05,
      "loss": 0.6439,
      "step": 210
    },
    {
      "epoch": 19.18,
      "learning_rate": 7.672727272727273e-05,
      "loss": 0.6698,
      "step": 211
    },
    {
      "epoch": 19.27,
      "learning_rate": 7.709090909090909e-05,
      "loss": 0.6703,
      "step": 212
    },
    {
      "epoch": 19.36,
      "learning_rate": 7.745454545454546e-05,
      "loss": 0.6178,
      "step": 213
    },
    {
      "epoch": 19.45,
      "learning_rate": 7.781818181818183e-05,
      "loss": 0.6013,
      "step": 214
    },
    {
      "epoch": 19.55,
      "learning_rate": 7.818181818181818e-05,
      "loss": 0.6247,
      "step": 215
    },
    {
      "epoch": 19.64,
      "learning_rate": 7.854545454545454e-05,
      "loss": 0.624,
      "step": 216
    },
    {
      "epoch": 19.73,
      "learning_rate": 7.890909090909091e-05,
      "loss": 0.5774,
      "step": 217
    },
    {
      "epoch": 19.82,
      "learning_rate": 7.927272727272728e-05,
      "loss": 0.6187,
      "step": 218
    },
    {
      "epoch": 19.91,
      "learning_rate": 7.963636363636364e-05,
      "loss": 0.6401,
      "step": 219
    },
    {
      "epoch": 20.0,
      "learning_rate": 8e-05,
      "loss": 0.608,
      "step": 220
    },
    {
      "epoch": 20.0,
      "eval_accuracy_NAT": 0.9587116739352763,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8941330636508005,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7906241153094713,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9608881397168275,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9820896495045418,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9479116908257735,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7102665429327287,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.5579340180148183,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8665623415357604,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8661181583018343,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.5255590081214905,
      "eval_mean_accuracy": 0.3528035878551475,
      "eval_mean_iou": 0.3037532885854551,
      "eval_overall_accuracy": 0.9270096884833442,
      "eval_runtime": 3.8134,
      "eval_samples_per_second": 9.44,
      "eval_steps_per_second": 0.524,
      "step": 220
    },
    {
      "epoch": 20.09,
      "learning_rate": 8.036363636363636e-05,
      "loss": 0.5913,
      "step": 221
    },
    {
      "epoch": 20.18,
      "learning_rate": 8.072727272727273e-05,
      "loss": 0.6022,
      "step": 222
    },
    {
      "epoch": 20.27,
      "learning_rate": 8.109090909090909e-05,
      "loss": 0.5652,
      "step": 223
    },
    {
      "epoch": 20.36,
      "learning_rate": 8.145454545454546e-05,
      "loss": 0.5683,
      "step": 224
    },
    {
      "epoch": 20.45,
      "learning_rate": 8.181818181818183e-05,
      "loss": 0.6305,
      "step": 225
    },
    {
      "epoch": 20.55,
      "learning_rate": 8.218181818181819e-05,
      "loss": 0.6003,
      "step": 226
    },
    {
      "epoch": 20.64,
      "learning_rate": 8.254545454545454e-05,
      "loss": 0.5246,
      "step": 227
    },
    {
      "epoch": 20.73,
      "learning_rate": 8.290909090909091e-05,
      "loss": 0.5921,
      "step": 228
    },
    {
      "epoch": 20.82,
      "learning_rate": 8.327272727272728e-05,
      "loss": 0.603,
      "step": 229
    },
    {
      "epoch": 20.91,
      "learning_rate": 8.363636363636364e-05,
      "loss": 0.5426,
      "step": 230
    },
    {
      "epoch": 21.0,
      "learning_rate": 8.4e-05,
      "loss": 0.5409,
      "step": 231
    },
    {
      "epoch": 21.09,
      "learning_rate": 8.436363636363637e-05,
      "loss": 0.4839,
      "step": 232
    },
    {
      "epoch": 21.18,
      "learning_rate": 8.472727272727274e-05,
      "loss": 0.5619,
      "step": 233
    },
    {
      "epoch": 21.27,
      "learning_rate": 8.50909090909091e-05,
      "loss": 0.5996,
      "step": 234
    },
    {
      "epoch": 21.36,
      "learning_rate": 8.545454545454545e-05,
      "loss": 0.5703,
      "step": 235
    },
    {
      "epoch": 21.45,
      "learning_rate": 8.581818181818182e-05,
      "loss": 0.5253,
      "step": 236
    },
    {
      "epoch": 21.55,
      "learning_rate": 8.618181818181819e-05,
      "loss": 0.5508,
      "step": 237
    },
    {
      "epoch": 21.64,
      "learning_rate": 8.654545454545456e-05,
      "loss": 0.5668,
      "step": 238
    },
    {
      "epoch": 21.73,
      "learning_rate": 8.690909090909091e-05,
      "loss": 0.5009,
      "step": 239
    },
    {
      "epoch": 21.82,
      "learning_rate": 8.727272727272727e-05,
      "loss": 0.5261,
      "step": 240
    },
    {
      "epoch": 21.82,
      "eval_accuracy_NAT": 0.9583323040351547,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.9321604468002919,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7454677631877868,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9518480943473114,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9563380145710917,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.948486096354183,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.6478410704181764,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.5856762745792591,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8490727427463814,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.871423285203022,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.47437721490859985,
      "eval_mean_accuracy": 0.3495497402262797,
      "eval_mean_iou": 0.3001922668693094,
      "eval_overall_accuracy": 0.9231026967366537,
      "eval_runtime": 3.2633,
      "eval_samples_per_second": 11.032,
      "eval_steps_per_second": 0.613,
      "step": 240
    },
    {
      "epoch": 21.91,
      "learning_rate": 8.763636363636364e-05,
      "loss": 0.5156,
      "step": 241
    },
    {
      "epoch": 22.0,
      "learning_rate": 8.800000000000001e-05,
      "loss": 0.5294,
      "step": 242
    },
    {
      "epoch": 22.09,
      "learning_rate": 8.836363636363637e-05,
      "loss": 0.505,
      "step": 243
    },
    {
      "epoch": 22.18,
      "learning_rate": 8.872727272727274e-05,
      "loss": 0.5347,
      "step": 244
    },
    {
      "epoch": 22.27,
      "learning_rate": 8.90909090909091e-05,
      "loss": 0.5135,
      "step": 245
    },
    {
      "epoch": 22.36,
      "learning_rate": 8.945454545454546e-05,
      "loss": 0.4949,
      "step": 246
    },
    {
      "epoch": 22.45,
      "learning_rate": 8.981818181818182e-05,
      "loss": 0.5015,
      "step": 247
    },
    {
      "epoch": 22.55,
      "learning_rate": 9.018181818181819e-05,
      "loss": 0.4432,
      "step": 248
    },
    {
      "epoch": 22.64,
      "learning_rate": 9.054545454545455e-05,
      "loss": 0.5089,
      "step": 249
    },
    {
      "epoch": 22.73,
      "learning_rate": 9.090909090909092e-05,
      "loss": 0.5342,
      "step": 250
    },
    {
      "epoch": 22.82,
      "learning_rate": 9.127272727272727e-05,
      "loss": 0.4953,
      "step": 251
    },
    {
      "epoch": 22.91,
      "learning_rate": 9.163636363636364e-05,
      "loss": 0.5519,
      "step": 252
    },
    {
      "epoch": 23.0,
      "learning_rate": 9.200000000000001e-05,
      "loss": 0.4735,
      "step": 253
    },
    {
      "epoch": 23.09,
      "learning_rate": 9.236363636363636e-05,
      "loss": 0.4638,
      "step": 254
    },
    {
      "epoch": 23.18,
      "learning_rate": 9.272727272727273e-05,
      "loss": 0.488,
      "step": 255
    },
    {
      "epoch": 23.27,
      "learning_rate": 9.30909090909091e-05,
      "loss": 0.4371,
      "step": 256
    },
    {
      "epoch": 23.36,
      "learning_rate": 9.345454545454547e-05,
      "loss": 0.4664,
      "step": 257
    },
    {
      "epoch": 23.45,
      "learning_rate": 9.381818181818182e-05,
      "loss": 0.4603,
      "step": 258
    },
    {
      "epoch": 23.55,
      "learning_rate": 9.418181818181818e-05,
      "loss": 0.4394,
      "step": 259
    },
    {
      "epoch": 23.64,
      "learning_rate": 9.454545454545455e-05,
      "loss": 0.4777,
      "step": 260
    },
    {
      "epoch": 23.64,
      "eval_accuracy_NAT": 0.9674894111685087,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8888887398600799,
      "eval_accuracy_concrete tiles": 0.10248883996820156,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7898062680651622,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9544164539466763,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9888592816792116,
      "eval_accuracy_steel": 0.0009646461315216675,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9545549900462824,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.725248497197943,
      "eval_iou_concrete tiles": 0.10248883996820156,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.6226222227120437,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8777720566238544,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8638473573125913,
      "eval_iou_steel": 0.0009646461315216675,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.4128577411174774,
      "eval_mean_accuracy": 0.36099335698610474,
      "eval_mean_iou": 0.31903835461480295,
      "eval_overall_accuracy": 0.9328311284383138,
      "eval_runtime": 4.1127,
      "eval_samples_per_second": 8.753,
      "eval_steps_per_second": 0.486,
      "step": 260
    },
    {
      "epoch": 23.73,
      "learning_rate": 9.490909090909092e-05,
      "loss": 0.4488,
      "step": 261
    },
    {
      "epoch": 23.82,
      "learning_rate": 9.527272727272728e-05,
      "loss": 0.4711,
      "step": 262
    },
    {
      "epoch": 23.91,
      "learning_rate": 9.563636363636365e-05,
      "loss": 0.4346,
      "step": 263
    },
    {
      "epoch": 24.0,
      "learning_rate": 9.6e-05,
      "loss": 0.5418,
      "step": 264
    },
    {
      "epoch": 24.09,
      "learning_rate": 9.636363636363637e-05,
      "loss": 0.4004,
      "step": 265
    },
    {
      "epoch": 24.18,
      "learning_rate": 9.672727272727273e-05,
      "loss": 0.5282,
      "step": 266
    },
    {
      "epoch": 24.27,
      "learning_rate": 9.70909090909091e-05,
      "loss": 0.4103,
      "step": 267
    },
    {
      "epoch": 24.36,
      "learning_rate": 9.745454545454546e-05,
      "loss": 0.4233,
      "step": 268
    },
    {
      "epoch": 24.45,
      "learning_rate": 9.781818181818183e-05,
      "loss": 0.4435,
      "step": 269
    },
    {
      "epoch": 24.55,
      "learning_rate": 9.818181818181818e-05,
      "loss": 0.4471,
      "step": 270
    },
    {
      "epoch": 24.64,
      "learning_rate": 9.854545454545455e-05,
      "loss": 0.4428,
      "step": 271
    },
    {
      "epoch": 24.73,
      "learning_rate": 9.890909090909092e-05,
      "loss": 0.4204,
      "step": 272
    },
    {
      "epoch": 24.82,
      "learning_rate": 9.927272727272728e-05,
      "loss": 0.4329,
      "step": 273
    },
    {
      "epoch": 24.91,
      "learning_rate": 9.963636363636363e-05,
      "loss": 0.4094,
      "step": 274
    },
    {
      "epoch": 25.0,
      "learning_rate": 0.0001,
      "loss": 0.3999,
      "step": 275
    },
    {
      "epoch": 25.09,
      "learning_rate": 9.998086124401914e-05,
      "loss": 0.3871,
      "step": 276
    },
    {
      "epoch": 25.18,
      "learning_rate": 9.996172248803828e-05,
      "loss": 0.4378,
      "step": 277
    },
    {
      "epoch": 25.27,
      "learning_rate": 9.994258373205743e-05,
      "loss": 0.3492,
      "step": 278
    },
    {
      "epoch": 25.36,
      "learning_rate": 9.992344497607656e-05,
      "loss": 0.408,
      "step": 279
    },
    {
      "epoch": 25.45,
      "learning_rate": 9.99043062200957e-05,
      "loss": 0.3775,
      "step": 280
    },
    {
      "epoch": 25.45,
      "eval_accuracy_NAT": 0.9729683947818464,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8722732198806816,
      "eval_accuracy_concrete tiles": 0.7616339509570109,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.8074948360286177,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9740443320148926,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.986779334972189,
      "eval_accuracy_steel": 0.014296550359474969,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9569670815303015,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7644293101036613,
      "eval_iou_concrete tiles": 0.7351335399144164,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.6186385763477944,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8826613018104198,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8804635809724196,
      "eval_iou_steel": 0.014296550359474969,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.3502545952796936,
      "eval_mean_accuracy": 0.41457620146113183,
      "eval_mean_iou": 0.37327614931065295,
      "eval_overall_accuracy": 0.9391554726494683,
      "eval_runtime": 3.892,
      "eval_samples_per_second": 9.25,
      "eval_steps_per_second": 0.514,
      "step": 280
    },
    {
      "epoch": 25.55,
      "learning_rate": 9.988516746411484e-05,
      "loss": 0.4176,
      "step": 281
    },
    {
      "epoch": 25.64,
      "learning_rate": 9.986602870813397e-05,
      "loss": 0.3613,
      "step": 282
    },
    {
      "epoch": 25.73,
      "learning_rate": 9.984688995215312e-05,
      "loss": 0.4248,
      "step": 283
    },
    {
      "epoch": 25.82,
      "learning_rate": 9.982775119617226e-05,
      "loss": 0.4028,
      "step": 284
    },
    {
      "epoch": 25.91,
      "learning_rate": 9.980861244019139e-05,
      "loss": 0.414,
      "step": 285
    },
    {
      "epoch": 26.0,
      "learning_rate": 9.978947368421054e-05,
      "loss": 0.3797,
      "step": 286
    },
    {
      "epoch": 26.09,
      "learning_rate": 9.977033492822967e-05,
      "loss": 0.3478,
      "step": 287
    },
    {
      "epoch": 26.18,
      "learning_rate": 9.97511961722488e-05,
      "loss": 0.4247,
      "step": 288
    },
    {
      "epoch": 26.27,
      "learning_rate": 9.973205741626795e-05,
      "loss": 0.3849,
      "step": 289
    },
    {
      "epoch": 26.36,
      "learning_rate": 9.97129186602871e-05,
      "loss": 0.3291,
      "step": 290
    },
    {
      "epoch": 26.45,
      "learning_rate": 9.969377990430623e-05,
      "loss": 0.3979,
      "step": 291
    },
    {
      "epoch": 26.55,
      "learning_rate": 9.967464114832537e-05,
      "loss": 0.3622,
      "step": 292
    },
    {
      "epoch": 26.64,
      "learning_rate": 9.96555023923445e-05,
      "loss": 0.3409,
      "step": 293
    },
    {
      "epoch": 26.73,
      "learning_rate": 9.963636363636363e-05,
      "loss": 0.3457,
      "step": 294
    },
    {
      "epoch": 26.82,
      "learning_rate": 9.961722488038278e-05,
      "loss": 0.3679,
      "step": 295
    },
    {
      "epoch": 26.91,
      "learning_rate": 9.959808612440193e-05,
      "loss": 0.3988,
      "step": 296
    },
    {
      "epoch": 27.0,
      "learning_rate": 9.957894736842106e-05,
      "loss": 0.3584,
      "step": 297
    },
    {
      "epoch": 27.09,
      "learning_rate": 9.95598086124402e-05,
      "loss": 0.3431,
      "step": 298
    },
    {
      "epoch": 27.18,
      "learning_rate": 9.954066985645934e-05,
      "loss": 0.3898,
      "step": 299
    },
    {
      "epoch": 27.27,
      "learning_rate": 9.952153110047847e-05,
      "loss": 0.3272,
      "step": 300
    },
    {
      "epoch": 27.27,
      "eval_accuracy_NAT": 0.9755313472466215,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8942215867633804,
      "eval_accuracy_concrete tiles": 0.7856356631810677,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7729774882303393,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9490753784521401,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9889141917895506,
      "eval_accuracy_steel": 0.12353241870589012,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9595498328439508,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7432702405611682,
      "eval_iou_concrete tiles": 0.6156998059089929,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.6411552053527684,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.9023240039777659,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8827876336194475,
      "eval_iou_steel": 0.12300000820930443,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.31202512979507446,
      "eval_mean_accuracy": 0.4222990826437684,
      "eval_mean_iou": 0.3744451331133383,
      "eval_overall_accuracy": 0.941056039598253,
      "eval_runtime": 3.7112,
      "eval_samples_per_second": 9.7,
      "eval_steps_per_second": 0.539,
      "step": 300
    },
    {
      "epoch": 27.36,
      "learning_rate": 9.950239234449761e-05,
      "loss": 0.3692,
      "step": 301
    },
    {
      "epoch": 27.45,
      "learning_rate": 9.948325358851676e-05,
      "loss": 0.312,
      "step": 302
    },
    {
      "epoch": 27.55,
      "learning_rate": 9.946411483253589e-05,
      "loss": 0.3441,
      "step": 303
    },
    {
      "epoch": 27.64,
      "learning_rate": 9.944497607655504e-05,
      "loss": 0.3033,
      "step": 304
    },
    {
      "epoch": 27.73,
      "learning_rate": 9.942583732057417e-05,
      "loss": 0.3007,
      "step": 305
    },
    {
      "epoch": 27.82,
      "learning_rate": 9.94066985645933e-05,
      "loss": 0.3405,
      "step": 306
    },
    {
      "epoch": 27.91,
      "learning_rate": 9.938755980861245e-05,
      "loss": 0.3206,
      "step": 307
    },
    {
      "epoch": 28.0,
      "learning_rate": 9.936842105263159e-05,
      "loss": 0.3518,
      "step": 308
    },
    {
      "epoch": 28.09,
      "learning_rate": 9.934928229665072e-05,
      "loss": 0.3508,
      "step": 309
    },
    {
      "epoch": 28.18,
      "learning_rate": 9.933014354066987e-05,
      "loss": 0.2985,
      "step": 310
    },
    {
      "epoch": 28.27,
      "learning_rate": 9.9311004784689e-05,
      "loss": 0.3159,
      "step": 311
    },
    {
      "epoch": 28.36,
      "learning_rate": 9.929186602870813e-05,
      "loss": 0.3215,
      "step": 312
    },
    {
      "epoch": 28.45,
      "learning_rate": 9.927272727272728e-05,
      "loss": 0.3105,
      "step": 313
    },
    {
      "epoch": 28.55,
      "learning_rate": 9.925358851674642e-05,
      "loss": 0.3098,
      "step": 314
    },
    {
      "epoch": 28.64,
      "learning_rate": 9.923444976076556e-05,
      "loss": 0.3491,
      "step": 315
    },
    {
      "epoch": 28.73,
      "learning_rate": 9.921531100478469e-05,
      "loss": 0.3527,
      "step": 316
    },
    {
      "epoch": 28.82,
      "learning_rate": 9.919617224880383e-05,
      "loss": 0.2871,
      "step": 317
    },
    {
      "epoch": 28.91,
      "learning_rate": 9.917703349282297e-05,
      "loss": 0.3123,
      "step": 318
    },
    {
      "epoch": 29.0,
      "learning_rate": 9.915789473684211e-05,
      "loss": 0.3469,
      "step": 319
    },
    {
      "epoch": 29.09,
      "learning_rate": 9.913875598086126e-05,
      "loss": 0.2718,
      "step": 320
    },
    {
      "epoch": 29.09,
      "eval_accuracy_NAT": 0.9825863922310691,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8046442443881712,
      "eval_accuracy_concrete tiles": 0.8758637558857701,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7965098194095422,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9697637326826177,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9828523569712307,
      "eval_accuracy_steel": 0.39059923487896575,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9586311515822584,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7106968754627573,
      "eval_iou_concrete tiles": 0.30282784502352134,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.644390092178929,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.9171158004759503,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.9192782290288645,
      "eval_iou_steel": 0.3835287069719245,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.28215399384498596,
      "eval_mean_accuracy": 0.4463707335728744,
      "eval_mean_iou": 0.3720360539018619,
      "eval_overall_accuracy": 0.9434228473239474,
      "eval_runtime": 4.2101,
      "eval_samples_per_second": 8.551,
      "eval_steps_per_second": 0.475,
      "step": 320
    },
    {
      "epoch": 29.18,
      "learning_rate": 9.911961722488039e-05,
      "loss": 0.3449,
      "step": 321
    },
    {
      "epoch": 29.27,
      "learning_rate": 9.910047846889952e-05,
      "loss": 0.2981,
      "step": 322
    },
    {
      "epoch": 29.36,
      "learning_rate": 9.908133971291867e-05,
      "loss": 0.3002,
      "step": 323
    },
    {
      "epoch": 29.45,
      "learning_rate": 9.90622009569378e-05,
      "loss": 0.3164,
      "step": 324
    },
    {
      "epoch": 29.55,
      "learning_rate": 9.904306220095694e-05,
      "loss": 0.2772,
      "step": 325
    },
    {
      "epoch": 29.64,
      "learning_rate": 9.902392344497609e-05,
      "loss": 0.2679,
      "step": 326
    },
    {
      "epoch": 29.73,
      "learning_rate": 9.900478468899522e-05,
      "loss": 0.2789,
      "step": 327
    },
    {
      "epoch": 29.82,
      "learning_rate": 9.898564593301435e-05,
      "loss": 0.3047,
      "step": 328
    },
    {
      "epoch": 29.91,
      "learning_rate": 9.89665071770335e-05,
      "loss": 0.3267,
      "step": 329
    },
    {
      "epoch": 30.0,
      "learning_rate": 9.894736842105263e-05,
      "loss": 0.3023,
      "step": 330
    },
    {
      "epoch": 30.09,
      "learning_rate": 9.892822966507178e-05,
      "loss": 0.2884,
      "step": 331
    },
    {
      "epoch": 30.18,
      "learning_rate": 9.890909090909092e-05,
      "loss": 0.291,
      "step": 332
    },
    {
      "epoch": 30.27,
      "learning_rate": 9.888995215311005e-05,
      "loss": 0.2564,
      "step": 333
    },
    {
      "epoch": 30.36,
      "learning_rate": 9.887081339712919e-05,
      "loss": 0.2616,
      "step": 334
    },
    {
      "epoch": 30.45,
      "learning_rate": 9.885167464114833e-05,
      "loss": 0.234,
      "step": 335
    },
    {
      "epoch": 30.55,
      "learning_rate": 9.883253588516746e-05,
      "loss": 0.2786,
      "step": 336
    },
    {
      "epoch": 30.64,
      "learning_rate": 9.881339712918661e-05,
      "loss": 0.2706,
      "step": 337
    },
    {
      "epoch": 30.73,
      "learning_rate": 9.879425837320575e-05,
      "loss": 0.2773,
      "step": 338
    },
    {
      "epoch": 30.82,
      "learning_rate": 9.877511961722489e-05,
      "loss": 0.2576,
      "step": 339
    },
    {
      "epoch": 30.91,
      "learning_rate": 9.875598086124402e-05,
      "loss": 0.286,
      "step": 340
    },
    {
      "epoch": 30.91,
      "eval_accuracy_NAT": 0.977698166955223,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.9108478368170307,
      "eval_accuracy_concrete tiles": 0.8354736134042683,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.8607387886773593,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9504163910678161,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0005789430154660492,
      "eval_accuracy_speedway bricks": 0.9874183438386632,
      "eval_accuracy_steel": 0.47383088186795064,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9621091350951422,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7653984394459729,
      "eval_iou_concrete tiles": 0.8104701171585348,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.6783810659754455,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.9239618416180283,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0005789430154660492,
      "eval_iou_speedway bricks": 0.9187664611554629,
      "eval_iou_steel": 0.45589763523429505,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.23237627744674683,
      "eval_mean_accuracy": 0.4613079204341367,
      "eval_mean_iou": 0.424274126053719,
      "eval_overall_accuracy": 0.950863414340549,
      "eval_runtime": 3.4041,
      "eval_samples_per_second": 10.576,
      "eval_steps_per_second": 0.588,
      "step": 340
    },
    {
      "epoch": 31.0,
      "learning_rate": 9.873684210526316e-05,
      "loss": 0.2626,
      "step": 341
    },
    {
      "epoch": 31.09,
      "learning_rate": 9.87177033492823e-05,
      "loss": 0.2658,
      "step": 342
    },
    {
      "epoch": 31.18,
      "learning_rate": 9.869856459330143e-05,
      "loss": 0.2426,
      "step": 343
    },
    {
      "epoch": 31.27,
      "learning_rate": 9.867942583732059e-05,
      "loss": 0.2669,
      "step": 344
    },
    {
      "epoch": 31.36,
      "learning_rate": 9.866028708133972e-05,
      "loss": 0.2462,
      "step": 345
    },
    {
      "epoch": 31.45,
      "learning_rate": 9.864114832535885e-05,
      "loss": 0.2497,
      "step": 346
    },
    {
      "epoch": 31.55,
      "learning_rate": 9.8622009569378e-05,
      "loss": 0.2362,
      "step": 347
    },
    {
      "epoch": 31.64,
      "learning_rate": 9.860287081339713e-05,
      "loss": 0.2559,
      "step": 348
    },
    {
      "epoch": 31.73,
      "learning_rate": 9.858373205741627e-05,
      "loss": 0.2661,
      "step": 349
    },
    {
      "epoch": 31.82,
      "learning_rate": 9.856459330143542e-05,
      "loss": 0.2389,
      "step": 350
    },
    {
      "epoch": 31.91,
      "learning_rate": 9.854545454545455e-05,
      "loss": 0.2693,
      "step": 351
    },
    {
      "epoch": 32.0,
      "learning_rate": 9.852631578947368e-05,
      "loss": 0.2688,
      "step": 352
    },
    {
      "epoch": 32.09,
      "learning_rate": 9.850717703349283e-05,
      "loss": 0.2313,
      "step": 353
    },
    {
      "epoch": 32.18,
      "learning_rate": 9.848803827751196e-05,
      "loss": 0.1979,
      "step": 354
    },
    {
      "epoch": 32.27,
      "learning_rate": 9.846889952153111e-05,
      "loss": 0.2052,
      "step": 355
    },
    {
      "epoch": 32.36,
      "learning_rate": 9.844976076555025e-05,
      "loss": 0.2669,
      "step": 356
    },
    {
      "epoch": 32.45,
      "learning_rate": 9.843062200956938e-05,
      "loss": 0.2743,
      "step": 357
    },
    {
      "epoch": 32.55,
      "learning_rate": 9.841148325358852e-05,
      "loss": 0.2396,
      "step": 358
    },
    {
      "epoch": 32.64,
      "learning_rate": 9.839234449760766e-05,
      "loss": 0.2322,
      "step": 359
    },
    {
      "epoch": 32.73,
      "learning_rate": 9.83732057416268e-05,
      "loss": 0.2553,
      "step": 360
    },
    {
      "epoch": 32.73,
      "eval_accuracy_NAT": 0.986369504165305,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.9002438409373793,
      "eval_accuracy_concrete tiles": 0.9165902280927047,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.8603193798341238,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9630828537812566,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.026521103851349488,
      "eval_accuracy_speedway bricks": 0.9764043411571189,
      "eval_accuracy_steel": 0.6370292197084625,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.964050345451529,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.802261703556906,
      "eval_iou_concrete tiles": 0.8505603631720812,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.6887008099827931,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.9318697956093869,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.026312910284463895,
      "eval_iou_speedway bricks": 0.9343460443813649,
      "eval_iou_steel": 0.5962418489794343,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.21057012677192688,
      "eval_mean_accuracy": 0.4820431131944385,
      "eval_mean_iou": 0.4457187554936892,
      "eval_overall_accuracy": 0.9568744235568576,
      "eval_runtime": 3.4643,
      "eval_samples_per_second": 10.392,
      "eval_steps_per_second": 0.577,
      "step": 360
    },
    {
      "epoch": 32.82,
      "learning_rate": 9.835406698564594e-05,
      "loss": 0.2174,
      "step": 361
    },
    {
      "epoch": 32.91,
      "learning_rate": 9.833492822966509e-05,
      "loss": 0.3019,
      "step": 362
    },
    {
      "epoch": 33.0,
      "learning_rate": 9.831578947368422e-05,
      "loss": 0.2601,
      "step": 363
    },
    {
      "epoch": 33.09,
      "learning_rate": 9.829665071770335e-05,
      "loss": 0.2548,
      "step": 364
    },
    {
      "epoch": 33.18,
      "learning_rate": 9.82775119617225e-05,
      "loss": 0.2142,
      "step": 365
    },
    {
      "epoch": 33.27,
      "learning_rate": 9.825837320574163e-05,
      "loss": 0.2233,
      "step": 366
    },
    {
      "epoch": 33.36,
      "learning_rate": 9.823923444976077e-05,
      "loss": 0.2116,
      "step": 367
    },
    {
      "epoch": 33.45,
      "learning_rate": 9.822009569377992e-05,
      "loss": 0.2108,
      "step": 368
    },
    {
      "epoch": 33.55,
      "learning_rate": 9.820095693779905e-05,
      "loss": 0.1959,
      "step": 369
    },
    {
      "epoch": 33.64,
      "learning_rate": 9.818181818181818e-05,
      "loss": 0.263,
      "step": 370
    },
    {
      "epoch": 33.73,
      "learning_rate": 9.816267942583733e-05,
      "loss": 0.2373,
      "step": 371
    },
    {
      "epoch": 33.82,
      "learning_rate": 9.814354066985646e-05,
      "loss": 0.1877,
      "step": 372
    },
    {
      "epoch": 33.91,
      "learning_rate": 9.81244019138756e-05,
      "loss": 0.2542,
      "step": 373
    },
    {
      "epoch": 34.0,
      "learning_rate": 9.810526315789475e-05,
      "loss": 0.2463,
      "step": 374
    },
    {
      "epoch": 34.09,
      "learning_rate": 9.808612440191388e-05,
      "loss": 0.2061,
      "step": 375
    },
    {
      "epoch": 34.18,
      "learning_rate": 9.806698564593301e-05,
      "loss": 0.2288,
      "step": 376
    },
    {
      "epoch": 34.27,
      "learning_rate": 9.804784688995216e-05,
      "loss": 0.2206,
      "step": 377
    },
    {
      "epoch": 34.36,
      "learning_rate": 9.802870813397129e-05,
      "loss": 0.1926,
      "step": 378
    },
    {
      "epoch": 34.45,
      "learning_rate": 9.800956937799044e-05,
      "loss": 0.199,
      "step": 379
    },
    {
      "epoch": 34.55,
      "learning_rate": 9.799043062200958e-05,
      "loss": 0.2305,
      "step": 380
    },
    {
      "epoch": 34.55,
      "eval_accuracy_NAT": 0.985468897667528,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8977423923773552,
      "eval_accuracy_concrete tiles": 0.9639515685195377,
      "eval_accuracy_dark bricks": 0.04491743966744929,
      "eval_accuracy_grass": 0.8741843371767494,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9721228991507727,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.14093127119344967,
      "eval_accuracy_speedway bricks": 0.9746079961188809,
      "eval_accuracy_steel": 0.7391003231976783,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.965356884261195,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.8089298835905171,
      "eval_iou_concrete tiles": 0.8073495518565941,
      "eval_iou_dark bricks": 0.04491225369458128,
      "eval_iou_grass": 0.7313335516543666,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.9402549192061903,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.11247029833670685,
      "eval_iou_speedway bricks": 0.9375066387515026,
      "eval_iou_steel": 0.6287497808171137,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.2024267166852951,
      "eval_mean_accuracy": 0.5071559326976464,
      "eval_mean_iou": 0.459758750936059,
      "eval_overall_accuracy": 0.9590724309285482,
      "eval_runtime": 4.213,
      "eval_samples_per_second": 8.545,
      "eval_steps_per_second": 0.475,
      "step": 380
    },
    {
      "epoch": 34.64,
      "learning_rate": 9.797129186602872e-05,
      "loss": 0.2066,
      "step": 381
    },
    {
      "epoch": 34.73,
      "learning_rate": 9.795215311004785e-05,
      "loss": 0.2228,
      "step": 382
    },
    {
      "epoch": 34.82,
      "learning_rate": 9.793301435406699e-05,
      "loss": 0.204,
      "step": 383
    },
    {
      "epoch": 34.91,
      "learning_rate": 9.791387559808612e-05,
      "loss": 0.1985,
      "step": 384
    },
    {
      "epoch": 35.0,
      "learning_rate": 9.789473684210527e-05,
      "loss": 0.2294,
      "step": 385
    },
    {
      "epoch": 35.09,
      "learning_rate": 9.787559808612442e-05,
      "loss": 0.2086,
      "step": 386
    },
    {
      "epoch": 35.18,
      "learning_rate": 9.785645933014355e-05,
      "loss": 0.187,
      "step": 387
    },
    {
      "epoch": 35.27,
      "learning_rate": 9.783732057416268e-05,
      "loss": 0.1915,
      "step": 388
    },
    {
      "epoch": 35.36,
      "learning_rate": 9.781818181818183e-05,
      "loss": 0.2294,
      "step": 389
    },
    {
      "epoch": 35.45,
      "learning_rate": 9.779904306220096e-05,
      "loss": 0.2292,
      "step": 390
    },
    {
      "epoch": 35.55,
      "learning_rate": 9.77799043062201e-05,
      "loss": 0.1793,
      "step": 391
    },
    {
      "epoch": 35.64,
      "learning_rate": 9.776076555023923e-05,
      "loss": 0.2091,
      "step": 392
    },
    {
      "epoch": 35.73,
      "learning_rate": 9.774162679425838e-05,
      "loss": 0.1883,
      "step": 393
    },
    {
      "epoch": 35.82,
      "learning_rate": 9.772248803827751e-05,
      "loss": 0.2175,
      "step": 394
    },
    {
      "epoch": 35.91,
      "learning_rate": 9.770334928229666e-05,
      "loss": 0.2277,
      "step": 395
    },
    {
      "epoch": 36.0,
      "learning_rate": 9.768421052631579e-05,
      "loss": 0.1767,
      "step": 396
    },
    {
      "epoch": 36.09,
      "learning_rate": 9.766507177033494e-05,
      "loss": 0.2035,
      "step": 397
    },
    {
      "epoch": 36.18,
      "learning_rate": 9.764593301435407e-05,
      "loss": 0.229,
      "step": 398
    },
    {
      "epoch": 36.27,
      "learning_rate": 9.762679425837321e-05,
      "loss": 0.1671,
      "step": 399
    },
    {
      "epoch": 36.36,
      "learning_rate": 9.760765550239235e-05,
      "loss": 0.2019,
      "step": 400
    },
    {
      "epoch": 36.36,
      "eval_accuracy_NAT": 0.9836902704148648,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.9404172389372935,
      "eval_accuracy_concrete tiles": 0.8294196783464808,
      "eval_accuracy_dark bricks": 0.19722104614910896,
      "eval_accuracy_grass": 0.8577819563325493,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9300545232270883,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.12643012709177626,
      "eval_accuracy_speedway bricks": 0.9749531453838699,
      "eval_accuracy_steel": 0.7743717432887013,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9671312829982205,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7636547208281054,
      "eval_iou_concrete tiles": 0.8252060961883613,
      "eval_iou_dark bricks": 0.19474004256612953,
      "eval_iou_grass": 0.7586279376717473,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.9198591645308932,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.09111499642374632,
      "eval_iou_speedway bricks": 0.9437740799671028,
      "eval_iou_steel": 0.6910348379501895,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.18835988640785217,
      "eval_mean_accuracy": 0.5087953637824411,
      "eval_mean_iou": 0.47347255070188426,
      "eval_overall_accuracy": 0.9576032426622179,
      "eval_runtime": 3.4211,
      "eval_samples_per_second": 10.523,
      "eval_steps_per_second": 0.585,
      "step": 400
    }
  ],
  "max_steps": 5500,
  "num_train_epochs": 500,
  "total_flos": 2.239125677628457e+17,
  "trial_name": null,
  "trial_params": null
}