{
  "best_metric": 0.8469406962394714,
  "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0-optim/checkpoint-160",
  "epoch": 14.545454545454545,
  "global_step": 160,
  "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
    }
  ],
  "max_steps": 5500,
  "num_train_epochs": 500,
  "total_flos": 8.957204410820198e+16,
  "trial_name": null,
  "trial_params": null
}