{
  "best_metric": 0.17850728332996368,
  "best_model_checkpoint": "/robodata/smodak/Projects/lifelong_concept_learner/scripts/terrainseg/training/models/safety-utcustom-terrain-final/checkpoint-160",
  "epoch": 9.411764705882353,
  "global_step": 160,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.06,
      "learning_rate": 2.9411764705882356e-07,
      "loss": 2.7081,
      "step": 1
    },
    {
      "epoch": 0.12,
      "learning_rate": 5.882352941176471e-07,
      "loss": 2.7218,
      "step": 2
    },
    {
      "epoch": 0.18,
      "learning_rate": 8.823529411764706e-07,
      "loss": 2.709,
      "step": 3
    },
    {
      "epoch": 0.24,
      "learning_rate": 1.1764705882352942e-06,
      "loss": 2.6939,
      "step": 4
    },
    {
      "epoch": 0.29,
      "learning_rate": 1.4705882352941177e-06,
      "loss": 2.6979,
      "step": 5
    },
    {
      "epoch": 0.35,
      "learning_rate": 1.7647058823529412e-06,
      "loss": 2.6788,
      "step": 6
    },
    {
      "epoch": 0.41,
      "learning_rate": 2.058823529411765e-06,
      "loss": 2.6769,
      "step": 7
    },
    {
      "epoch": 0.47,
      "learning_rate": 2.3529411764705885e-06,
      "loss": 2.6464,
      "step": 8
    },
    {
      "epoch": 0.53,
      "learning_rate": 2.647058823529412e-06,
      "loss": 2.6424,
      "step": 9
    },
    {
      "epoch": 0.59,
      "learning_rate": 2.9411764705882355e-06,
      "loss": 2.6198,
      "step": 10
    },
    {
      "epoch": 0.65,
      "learning_rate": 3.2352941176470594e-06,
      "loss": 2.6228,
      "step": 11
    },
    {
      "epoch": 0.71,
      "learning_rate": 3.5294117647058825e-06,
      "loss": 2.6158,
      "step": 12
    },
    {
      "epoch": 0.76,
      "learning_rate": 3.823529411764706e-06,
      "loss": 2.632,
      "step": 13
    },
    {
      "epoch": 0.82,
      "learning_rate": 4.11764705882353e-06,
      "loss": 2.587,
      "step": 14
    },
    {
      "epoch": 0.88,
      "learning_rate": 4.411764705882353e-06,
      "loss": 2.5767,
      "step": 15
    },
    {
      "epoch": 0.94,
      "learning_rate": 4.705882352941177e-06,
      "loss": 2.5638,
      "step": 16
    },
    {
      "epoch": 1.0,
      "learning_rate": 5e-06,
      "loss": 2.517,
      "step": 17
    },
    {
      "epoch": 1.06,
      "learning_rate": 5.294117647058824e-06,
      "loss": 2.5202,
      "step": 18
    },
    {
      "epoch": 1.12,
      "learning_rate": 5.588235294117647e-06,
      "loss": 2.5257,
      "step": 19
    },
    {
      "epoch": 1.18,
      "learning_rate": 5.882352941176471e-06,
      "loss": 2.51,
      "step": 20
    },
    {
      "epoch": 1.18,
      "eval_accuracy_NAT": 0.0901846148886272,
      "eval_accuracy_REST": 0.1772357723577236,
      "eval_accuracy_concrete": 0.21836230658804573,
      "eval_accuracy_concrete tiles": 0.10897949563987744,
      "eval_accuracy_dark bricks": 0.01268591007407691,
      "eval_accuracy_grass": 0.027127403891645564,
      "eval_accuracy_red bricks": 0.0003844463971880492,
      "eval_accuracy_road": 0.0070585682671182835,
      "eval_accuracy_rough concrete": 0.002699336785465293,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.6741853129093485,
      "eval_accuracy_steel": 0.0007721641086127534,
      "eval_accuracy_tiles": 0.2798933624163107,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.09010334674005777,
      "eval_iou_REST": 2.227093222444523e-05,
      "eval_iou_concrete": 0.07965033823417267,
      "eval_iou_concrete tiles": 0.0011714161291131616,
      "eval_iou_dark bricks": 0.0030587532148061655,
      "eval_iou_grass": 0.019163923774789638,
      "eval_iou_red bricks": 5.560869275085081e-06,
      "eval_iou_road": 0.006794427417222495,
      "eval_iou_rough concrete": 0.00013145523557474666,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.4632836614619225,
      "eval_iou_steel": 0.0006867765509148131,
      "eval_iou_tiles": 0.0032256814378599824,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 2.507078170776367,
      "eval_mean_accuracy": 0.12304374571723385,
      "eval_mean_iou": 0.04766411514270953,
      "eval_overall_accuracy": 0.20164948802883342,
      "eval_runtime": 18.8176,
      "eval_samples_per_second": 6.271,
      "eval_steps_per_second": 0.425,
      "step": 20
    },
    {
      "epoch": 1.24,
      "learning_rate": 6.1764705882352944e-06,
      "loss": 2.4771,
      "step": 21
    },
    {
      "epoch": 1.29,
      "learning_rate": 6.470588235294119e-06,
      "loss": 2.4758,
      "step": 22
    },
    {
      "epoch": 1.35,
      "learning_rate": 6.7647058823529414e-06,
      "loss": 2.4283,
      "step": 23
    },
    {
      "epoch": 1.41,
      "learning_rate": 7.058823529411765e-06,
      "loss": 2.3783,
      "step": 24
    },
    {
      "epoch": 1.47,
      "learning_rate": 7.3529411764705884e-06,
      "loss": 2.3059,
      "step": 25
    },
    {
      "epoch": 1.53,
      "learning_rate": 7.647058823529413e-06,
      "loss": 2.3405,
      "step": 26
    },
    {
      "epoch": 1.59,
      "learning_rate": 7.941176470588235e-06,
      "loss": 2.2917,
      "step": 27
    },
    {
      "epoch": 1.65,
      "learning_rate": 8.23529411764706e-06,
      "loss": 2.2857,
      "step": 28
    },
    {
      "epoch": 1.71,
      "learning_rate": 8.529411764705883e-06,
      "loss": 2.1291,
      "step": 29
    },
    {
      "epoch": 1.76,
      "learning_rate": 8.823529411764707e-06,
      "loss": 2.1234,
      "step": 30
    },
    {
      "epoch": 1.82,
      "learning_rate": 9.117647058823529e-06,
      "loss": 2.0916,
      "step": 31
    },
    {
      "epoch": 1.88,
      "learning_rate": 9.411764705882354e-06,
      "loss": 2.0543,
      "step": 32
    },
    {
      "epoch": 1.94,
      "learning_rate": 9.705882352941177e-06,
      "loss": 1.9605,
      "step": 33
    },
    {
      "epoch": 2.0,
      "learning_rate": 1e-05,
      "loss": 1.9835,
      "step": 34
    },
    {
      "epoch": 2.06,
      "learning_rate": 1.0294117647058824e-05,
      "loss": 2.0005,
      "step": 35
    },
    {
      "epoch": 2.12,
      "learning_rate": 1.0588235294117648e-05,
      "loss": 1.8628,
      "step": 36
    },
    {
      "epoch": 2.18,
      "learning_rate": 1.0882352941176471e-05,
      "loss": 1.8321,
      "step": 37
    },
    {
      "epoch": 2.24,
      "learning_rate": 1.1176470588235295e-05,
      "loss": 1.7578,
      "step": 38
    },
    {
      "epoch": 2.29,
      "learning_rate": 1.1470588235294118e-05,
      "loss": 1.7233,
      "step": 39
    },
    {
      "epoch": 2.35,
      "learning_rate": 1.1764705882352942e-05,
      "loss": 1.5504,
      "step": 40
    },
    {
      "epoch": 2.35,
      "eval_accuracy_NAT": 0.760154401121494,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.3386920222776326,
      "eval_accuracy_concrete tiles": 0.007895357058684892,
      "eval_accuracy_dark bricks": 0.0014308136975288556,
      "eval_accuracy_grass": 0.07454476982063306,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.3332036483520477,
      "eval_accuracy_rough concrete": 0.0009832021404012656,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9879532008612127,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 0.00875518797903602,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.7577068331202973,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.2037716446695068,
      "eval_iou_concrete tiles": 0.00033778197232406094,
      "eval_iou_dark bricks": 0.0009957339958239116,
      "eval_iou_grass": 0.06824458242953867,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.31270597640653264,
      "eval_iou_rough concrete": 0.00024414351495637967,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.571054819328365,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 0.0007685393497465681,
      "eval_iou_unlabeled": 0.0,
      "eval_loss": 1.840782642364502,
      "eval_mean_accuracy": 0.1933548156391286,
      "eval_mean_iou": 0.13684500391336366,
      "eval_overall_accuracy": 0.6935860908637612,
      "eval_runtime": 7.985,
      "eval_samples_per_second": 14.778,
      "eval_steps_per_second": 1.002,
      "step": 40
    },
    {
      "epoch": 2.41,
      "learning_rate": 1.2058823529411765e-05,
      "loss": 1.7505,
      "step": 41
    },
    {
      "epoch": 2.47,
      "learning_rate": 1.2352941176470589e-05,
      "loss": 1.4808,
      "step": 42
    },
    {
      "epoch": 2.53,
      "learning_rate": 1.2647058823529412e-05,
      "loss": 1.4041,
      "step": 43
    },
    {
      "epoch": 2.59,
      "learning_rate": 1.2941176470588238e-05,
      "loss": 1.3261,
      "step": 44
    },
    {
      "epoch": 2.65,
      "learning_rate": 1.323529411764706e-05,
      "loss": 1.3337,
      "step": 45
    },
    {
      "epoch": 2.71,
      "learning_rate": 1.3529411764705883e-05,
      "loss": 1.3177,
      "step": 46
    },
    {
      "epoch": 2.76,
      "learning_rate": 1.3823529411764708e-05,
      "loss": 1.2551,
      "step": 47
    },
    {
      "epoch": 2.82,
      "learning_rate": 1.411764705882353e-05,
      "loss": 1.119,
      "step": 48
    },
    {
      "epoch": 2.88,
      "learning_rate": 1.4411764705882352e-05,
      "loss": 1.1259,
      "step": 49
    },
    {
      "epoch": 2.94,
      "learning_rate": 1.4705882352941177e-05,
      "loss": 1.0199,
      "step": 50
    },
    {
      "epoch": 3.0,
      "learning_rate": 1.5e-05,
      "loss": 1.1612,
      "step": 51
    },
    {
      "epoch": 3.06,
      "learning_rate": 1.5294117647058826e-05,
      "loss": 0.8821,
      "step": 52
    },
    {
      "epoch": 3.12,
      "learning_rate": 1.558823529411765e-05,
      "loss": 0.8667,
      "step": 53
    },
    {
      "epoch": 3.18,
      "learning_rate": 1.588235294117647e-05,
      "loss": 0.8607,
      "step": 54
    },
    {
      "epoch": 3.24,
      "learning_rate": 1.6176470588235296e-05,
      "loss": 0.8021,
      "step": 55
    },
    {
      "epoch": 3.29,
      "learning_rate": 1.647058823529412e-05,
      "loss": 0.7418,
      "step": 56
    },
    {
      "epoch": 3.35,
      "learning_rate": 1.676470588235294e-05,
      "loss": 0.7114,
      "step": 57
    },
    {
      "epoch": 3.41,
      "learning_rate": 1.7058823529411767e-05,
      "loss": 0.7799,
      "step": 58
    },
    {
      "epoch": 3.47,
      "learning_rate": 1.735294117647059e-05,
      "loss": 0.6974,
      "step": 59
    },
    {
      "epoch": 3.53,
      "learning_rate": 1.7647058823529414e-05,
      "loss": 0.6271,
      "step": 60
    },
    {
      "epoch": 3.53,
      "eval_accuracy_NAT": 0.9551305368117847,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8126940616930818,
      "eval_accuracy_concrete tiles": 0.0,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.25898348643298624,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.8978859032247204,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9793115987478005,
      "eval_accuracy_steel": 0.0,
      "eval_accuracy_tiles": 3.0294768093550245e-05,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9404775296004152,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.5259192497701041,
      "eval_iou_concrete tiles": 0.0,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.2391512912549761,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.6993177996437074,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8262999758534085,
      "eval_iou_steel": 0.0,
      "eval_iou_tiles": 2.891343318105592e-05,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.8082596063613892,
      "eval_mean_accuracy": 0.30031045243680515,
      "eval_mean_iou": 0.24855344304275326,
      "eval_overall_accuracy": 0.8893904281874835,
      "eval_runtime": 6.7648,
      "eval_samples_per_second": 17.443,
      "eval_steps_per_second": 1.183,
      "step": 60
    },
    {
      "epoch": 3.59,
      "learning_rate": 1.7941176470588237e-05,
      "loss": 0.6901,
      "step": 61
    },
    {
      "epoch": 3.65,
      "learning_rate": 1.8235294117647057e-05,
      "loss": 0.5778,
      "step": 62
    },
    {
      "epoch": 3.71,
      "learning_rate": 1.8529411764705884e-05,
      "loss": 0.6319,
      "step": 63
    },
    {
      "epoch": 3.76,
      "learning_rate": 1.8823529411764708e-05,
      "loss": 0.5626,
      "step": 64
    },
    {
      "epoch": 3.82,
      "learning_rate": 1.9117647058823528e-05,
      "loss": 0.7374,
      "step": 65
    },
    {
      "epoch": 3.88,
      "learning_rate": 1.9411764705882355e-05,
      "loss": 0.6702,
      "step": 66
    },
    {
      "epoch": 3.94,
      "learning_rate": 1.9705882352941178e-05,
      "loss": 0.5477,
      "step": 67
    },
    {
      "epoch": 4.0,
      "learning_rate": 2e-05,
      "loss": 0.586,
      "step": 68
    },
    {
      "epoch": 4.06,
      "learning_rate": 2.0294117647058825e-05,
      "loss": 0.505,
      "step": 69
    },
    {
      "epoch": 4.12,
      "learning_rate": 2.058823529411765e-05,
      "loss": 0.5758,
      "step": 70
    },
    {
      "epoch": 4.18,
      "learning_rate": 2.0882352941176472e-05,
      "loss": 0.5694,
      "step": 71
    },
    {
      "epoch": 4.24,
      "learning_rate": 2.1176470588235296e-05,
      "loss": 0.4535,
      "step": 72
    },
    {
      "epoch": 4.29,
      "learning_rate": 2.1470588235294116e-05,
      "loss": 0.5105,
      "step": 73
    },
    {
      "epoch": 4.35,
      "learning_rate": 2.1764705882352943e-05,
      "loss": 0.5525,
      "step": 74
    },
    {
      "epoch": 4.41,
      "learning_rate": 2.2058823529411766e-05,
      "loss": 0.465,
      "step": 75
    },
    {
      "epoch": 4.47,
      "learning_rate": 2.235294117647059e-05,
      "loss": 0.5477,
      "step": 76
    },
    {
      "epoch": 4.53,
      "learning_rate": 2.2647058823529413e-05,
      "loss": 0.3823,
      "step": 77
    },
    {
      "epoch": 4.59,
      "learning_rate": 2.2941176470588237e-05,
      "loss": 0.4698,
      "step": 78
    },
    {
      "epoch": 4.65,
      "learning_rate": 2.323529411764706e-05,
      "loss": 0.4636,
      "step": 79
    },
    {
      "epoch": 4.71,
      "learning_rate": 2.3529411764705884e-05,
      "loss": 0.3963,
      "step": 80
    },
    {
      "epoch": 4.71,
      "eval_accuracy_NAT": 0.969774210544504,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.7972872258293342,
      "eval_accuracy_concrete tiles": 0.04426113598868725,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7826429655535243,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9650914038630805,
      "eval_accuracy_rough concrete": 0.0,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9776932262095467,
      "eval_accuracy_steel": 0.0003548456421456666,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9531585041831687,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.6900984540650801,
      "eval_iou_concrete tiles": 0.04426113598868725,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.5632131076991799,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.727967248165449,
      "eval_iou_rough concrete": 0.0,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.8641130344685216,
      "eval_iou_steel": 0.00035449307490288885,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.4204064607620239,
      "eval_mean_accuracy": 0.34900807797160177,
      "eval_mean_iou": 0.29562815212653765,
      "eval_overall_accuracy": 0.9185845003289691,
      "eval_runtime": 5.4734,
      "eval_samples_per_second": 21.559,
      "eval_steps_per_second": 1.462,
      "step": 80
    },
    {
      "epoch": 4.76,
      "learning_rate": 2.3823529411764707e-05,
      "loss": 0.3853,
      "step": 81
    },
    {
      "epoch": 4.82,
      "learning_rate": 2.411764705882353e-05,
      "loss": 0.378,
      "step": 82
    },
    {
      "epoch": 4.88,
      "learning_rate": 2.4411764705882354e-05,
      "loss": 0.4374,
      "step": 83
    },
    {
      "epoch": 4.94,
      "learning_rate": 2.4705882352941178e-05,
      "loss": 0.3066,
      "step": 84
    },
    {
      "epoch": 5.0,
      "learning_rate": 2.5e-05,
      "loss": 0.3689,
      "step": 85
    },
    {
      "epoch": 5.06,
      "learning_rate": 2.5294117647058825e-05,
      "loss": 0.3284,
      "step": 86
    },
    {
      "epoch": 5.12,
      "learning_rate": 2.5588235294117645e-05,
      "loss": 0.3295,
      "step": 87
    },
    {
      "epoch": 5.18,
      "learning_rate": 2.5882352941176475e-05,
      "loss": 0.3386,
      "step": 88
    },
    {
      "epoch": 5.24,
      "learning_rate": 2.6176470588235295e-05,
      "loss": 0.3517,
      "step": 89
    },
    {
      "epoch": 5.29,
      "learning_rate": 2.647058823529412e-05,
      "loss": 0.3205,
      "step": 90
    },
    {
      "epoch": 5.35,
      "learning_rate": 2.6764705882352942e-05,
      "loss": 0.4159,
      "step": 91
    },
    {
      "epoch": 5.41,
      "learning_rate": 2.7058823529411766e-05,
      "loss": 0.3534,
      "step": 92
    },
    {
      "epoch": 5.47,
      "learning_rate": 2.7352941176470593e-05,
      "loss": 0.325,
      "step": 93
    },
    {
      "epoch": 5.53,
      "learning_rate": 2.7647058823529416e-05,
      "loss": 0.2642,
      "step": 94
    },
    {
      "epoch": 5.59,
      "learning_rate": 2.7941176470588236e-05,
      "loss": 0.3485,
      "step": 95
    },
    {
      "epoch": 5.65,
      "learning_rate": 2.823529411764706e-05,
      "loss": 0.3246,
      "step": 96
    },
    {
      "epoch": 5.71,
      "learning_rate": 2.8529411764705883e-05,
      "loss": 0.4612,
      "step": 97
    },
    {
      "epoch": 5.76,
      "learning_rate": 2.8823529411764703e-05,
      "loss": 0.2731,
      "step": 98
    },
    {
      "epoch": 5.82,
      "learning_rate": 2.9117647058823534e-05,
      "loss": 0.4205,
      "step": 99
    },
    {
      "epoch": 5.88,
      "learning_rate": 2.9411764705882354e-05,
      "loss": 0.3558,
      "step": 100
    },
    {
      "epoch": 5.88,
      "eval_accuracy_NAT": 0.9806404490547624,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.9025246556766533,
      "eval_accuracy_concrete tiles": 0.30502003299552205,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.7816418565745797,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.913452910724182,
      "eval_accuracy_rough concrete": 0.0030807000399239656,
      "eval_accuracy_rough red sidewalk": 0.0,
      "eval_accuracy_speedway bricks": 0.9632305210940226,
      "eval_accuracy_steel": 0.28737998970447853,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9617550029709626,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.6868428500250288,
      "eval_iou_concrete tiles": 0.3027439238344757,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.6345173453939056,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8305275811769873,
      "eval_iou_rough concrete": 0.0030807000399239656,
      "eval_iou_rough red sidewalk": 0.0,
      "eval_iou_speedway bricks": 0.891543623735271,
      "eval_iou_steel": 0.2804400149240265,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.30909788608551025,
      "eval_mean_accuracy": 0.3951516242972403,
      "eval_mean_iou": 0.35318854170004477,
      "eval_overall_accuracy": 0.9322030665510792,
      "eval_runtime": 5.2886,
      "eval_samples_per_second": 22.312,
      "eval_steps_per_second": 1.513,
      "step": 100
    },
    {
      "epoch": 5.94,
      "learning_rate": 2.9705882352941177e-05,
      "loss": 0.4249,
      "step": 101
    },
    {
      "epoch": 6.0,
      "learning_rate": 3e-05,
      "loss": 0.4043,
      "step": 102
    },
    {
      "epoch": 6.06,
      "learning_rate": 3.0294117647058824e-05,
      "loss": 0.3018,
      "step": 103
    },
    {
      "epoch": 6.12,
      "learning_rate": 3.058823529411765e-05,
      "loss": 0.3319,
      "step": 104
    },
    {
      "epoch": 6.18,
      "learning_rate": 3.0882352941176475e-05,
      "loss": 0.3451,
      "step": 105
    },
    {
      "epoch": 6.24,
      "learning_rate": 3.11764705882353e-05,
      "loss": 0.3506,
      "step": 106
    },
    {
      "epoch": 6.29,
      "learning_rate": 3.147058823529412e-05,
      "loss": 0.3709,
      "step": 107
    },
    {
      "epoch": 6.35,
      "learning_rate": 3.176470588235294e-05,
      "loss": 0.2675,
      "step": 108
    },
    {
      "epoch": 6.41,
      "learning_rate": 3.205882352941177e-05,
      "loss": 0.3261,
      "step": 109
    },
    {
      "epoch": 6.47,
      "learning_rate": 3.235294117647059e-05,
      "loss": 0.2528,
      "step": 110
    },
    {
      "epoch": 6.53,
      "learning_rate": 3.2647058823529416e-05,
      "loss": 0.2736,
      "step": 111
    },
    {
      "epoch": 6.59,
      "learning_rate": 3.294117647058824e-05,
      "loss": 0.3192,
      "step": 112
    },
    {
      "epoch": 6.65,
      "learning_rate": 3.3235294117647056e-05,
      "loss": 0.2764,
      "step": 113
    },
    {
      "epoch": 6.71,
      "learning_rate": 3.352941176470588e-05,
      "loss": 0.2485,
      "step": 114
    },
    {
      "epoch": 6.76,
      "learning_rate": 3.382352941176471e-05,
      "loss": 0.295,
      "step": 115
    },
    {
      "epoch": 6.82,
      "learning_rate": 3.411764705882353e-05,
      "loss": 0.2182,
      "step": 116
    },
    {
      "epoch": 6.88,
      "learning_rate": 3.441176470588236e-05,
      "loss": 0.2945,
      "step": 117
    },
    {
      "epoch": 6.94,
      "learning_rate": 3.470588235294118e-05,
      "loss": 0.2284,
      "step": 118
    },
    {
      "epoch": 7.0,
      "learning_rate": 3.5e-05,
      "loss": 0.3391,
      "step": 119
    },
    {
      "epoch": 7.06,
      "learning_rate": 3.529411764705883e-05,
      "loss": 0.2654,
      "step": 120
    },
    {
      "epoch": 7.06,
      "eval_accuracy_NAT": 0.9793312019858504,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8161395883560691,
      "eval_accuracy_concrete tiles": 0.38750883808625974,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.8487237279860109,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9769372718377128,
      "eval_accuracy_rough concrete": 0.00018472282637841962,
      "eval_accuracy_rough red sidewalk": 0.0009475180291624993,
      "eval_accuracy_speedway bricks": 0.9861728882500396,
      "eval_accuracy_steel": 0.4149295056650357,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9635845098252436,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7301235451822234,
      "eval_iou_concrete tiles": 0.30457172495554236,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.649146550195586,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.7846078239913129,
      "eval_iou_rough concrete": 0.0001839500605255038,
      "eval_iou_rough red sidewalk": 0.0009475180291624993,
      "eval_iou_speedway bricks": 0.9013550388480807,
      "eval_iou_steel": 0.40151082824726514,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.25543102622032166,
      "eval_mean_accuracy": 0.41622117407865533,
      "eval_mean_iou": 0.3643101145642263,
      "eval_overall_accuracy": 0.9363230042538401,
      "eval_runtime": 5.3581,
      "eval_samples_per_second": 22.023,
      "eval_steps_per_second": 1.493,
      "step": 120
    },
    {
      "epoch": 7.12,
      "learning_rate": 3.558823529411765e-05,
      "loss": 0.3335,
      "step": 121
    },
    {
      "epoch": 7.18,
      "learning_rate": 3.5882352941176474e-05,
      "loss": 0.2502,
      "step": 122
    },
    {
      "epoch": 7.24,
      "learning_rate": 3.61764705882353e-05,
      "loss": 0.2615,
      "step": 123
    },
    {
      "epoch": 7.29,
      "learning_rate": 3.6470588235294114e-05,
      "loss": 0.2804,
      "step": 124
    },
    {
      "epoch": 7.35,
      "learning_rate": 3.6764705882352945e-05,
      "loss": 0.2703,
      "step": 125
    },
    {
      "epoch": 7.41,
      "learning_rate": 3.705882352941177e-05,
      "loss": 0.2235,
      "step": 126
    },
    {
      "epoch": 7.47,
      "learning_rate": 3.735294117647059e-05,
      "loss": 0.2855,
      "step": 127
    },
    {
      "epoch": 7.53,
      "learning_rate": 3.7647058823529415e-05,
      "loss": 0.2263,
      "step": 128
    },
    {
      "epoch": 7.59,
      "learning_rate": 3.794117647058824e-05,
      "loss": 0.1922,
      "step": 129
    },
    {
      "epoch": 7.65,
      "learning_rate": 3.8235294117647055e-05,
      "loss": 0.2025,
      "step": 130
    },
    {
      "epoch": 7.71,
      "learning_rate": 3.8529411764705886e-05,
      "loss": 0.2237,
      "step": 131
    },
    {
      "epoch": 7.76,
      "learning_rate": 3.882352941176471e-05,
      "loss": 0.2142,
      "step": 132
    },
    {
      "epoch": 7.82,
      "learning_rate": 3.911764705882353e-05,
      "loss": 0.2622,
      "step": 133
    },
    {
      "epoch": 7.88,
      "learning_rate": 3.9411764705882356e-05,
      "loss": 0.2454,
      "step": 134
    },
    {
      "epoch": 7.94,
      "learning_rate": 3.970588235294117e-05,
      "loss": 0.2398,
      "step": 135
    },
    {
      "epoch": 8.0,
      "learning_rate": 4e-05,
      "loss": 0.3103,
      "step": 136
    },
    {
      "epoch": 8.06,
      "learning_rate": 4.029411764705883e-05,
      "loss": 0.2222,
      "step": 137
    },
    {
      "epoch": 8.12,
      "learning_rate": 4.058823529411765e-05,
      "loss": 0.2128,
      "step": 138
    },
    {
      "epoch": 8.18,
      "learning_rate": 4.0882352941176474e-05,
      "loss": 0.2087,
      "step": 139
    },
    {
      "epoch": 8.24,
      "learning_rate": 4.11764705882353e-05,
      "loss": 0.1938,
      "step": 140
    },
    {
      "epoch": 8.24,
      "eval_accuracy_NAT": 0.9830271765403628,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.9010176438483495,
      "eval_accuracy_concrete tiles": 0.6649304737214236,
      "eval_accuracy_dark bricks": 0.0,
      "eval_accuracy_grass": 0.8719394262831794,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9517197778325257,
      "eval_accuracy_rough concrete": 0.1663399257533414,
      "eval_accuracy_rough red sidewalk": 0.03742696215191872,
      "eval_accuracy_speedway bricks": 0.9778834415332932,
      "eval_accuracy_steel": 0.644497143742472,
      "eval_accuracy_tiles": 0.0,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9680687096130417,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7740222344798603,
      "eval_iou_concrete tiles": 0.18868164278023367,
      "eval_iou_dark bricks": 0.0,
      "eval_iou_grass": 0.6641972532280409,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.8746486517016726,
      "eval_iou_rough concrete": 0.16600853979090596,
      "eval_iou_rough red sidewalk": 0.03742696215191872,
      "eval_iou_speedway bricks": 0.9354015193799212,
      "eval_iou_steel": 0.5996131393140197,
      "eval_iou_tiles": 0.0,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.2058570384979248,
      "eval_mean_accuracy": 0.4768293824159129,
      "eval_mean_iou": 0.4006206655722781,
      "eval_overall_accuracy": 0.9478311700336004,
      "eval_runtime": 5.5104,
      "eval_samples_per_second": 21.414,
      "eval_steps_per_second": 1.452,
      "step": 140
    },
    {
      "epoch": 8.29,
      "learning_rate": 4.147058823529412e-05,
      "loss": 0.2178,
      "step": 141
    },
    {
      "epoch": 8.35,
      "learning_rate": 4.1764705882352944e-05,
      "loss": 0.2016,
      "step": 142
    },
    {
      "epoch": 8.41,
      "learning_rate": 4.205882352941177e-05,
      "loss": 0.291,
      "step": 143
    },
    {
      "epoch": 8.47,
      "learning_rate": 4.235294117647059e-05,
      "loss": 0.2516,
      "step": 144
    },
    {
      "epoch": 8.53,
      "learning_rate": 4.2647058823529415e-05,
      "loss": 0.2561,
      "step": 145
    },
    {
      "epoch": 8.59,
      "learning_rate": 4.294117647058823e-05,
      "loss": 0.2003,
      "step": 146
    },
    {
      "epoch": 8.65,
      "learning_rate": 4.323529411764706e-05,
      "loss": 0.2749,
      "step": 147
    },
    {
      "epoch": 8.71,
      "learning_rate": 4.3529411764705885e-05,
      "loss": 0.2009,
      "step": 148
    },
    {
      "epoch": 8.76,
      "learning_rate": 4.382352941176471e-05,
      "loss": 0.2451,
      "step": 149
    },
    {
      "epoch": 8.82,
      "learning_rate": 4.411764705882353e-05,
      "loss": 0.1868,
      "step": 150
    },
    {
      "epoch": 8.88,
      "learning_rate": 4.4411764705882356e-05,
      "loss": 0.1741,
      "step": 151
    },
    {
      "epoch": 8.94,
      "learning_rate": 4.470588235294118e-05,
      "loss": 0.1841,
      "step": 152
    },
    {
      "epoch": 9.0,
      "learning_rate": 4.5e-05,
      "loss": 0.1486,
      "step": 153
    },
    {
      "epoch": 9.06,
      "learning_rate": 4.5294117647058826e-05,
      "loss": 0.2007,
      "step": 154
    },
    {
      "epoch": 9.12,
      "learning_rate": 4.558823529411765e-05,
      "loss": 0.1838,
      "step": 155
    },
    {
      "epoch": 9.18,
      "learning_rate": 4.588235294117647e-05,
      "loss": 0.1689,
      "step": 156
    },
    {
      "epoch": 9.24,
      "learning_rate": 4.61764705882353e-05,
      "loss": 0.1427,
      "step": 157
    },
    {
      "epoch": 9.29,
      "learning_rate": 4.647058823529412e-05,
      "loss": 0.1998,
      "step": 158
    },
    {
      "epoch": 9.35,
      "learning_rate": 4.6764705882352944e-05,
      "loss": 0.1375,
      "step": 159
    },
    {
      "epoch": 9.41,
      "learning_rate": 4.705882352941177e-05,
      "loss": 0.202,
      "step": 160
    },
    {
      "epoch": 9.41,
      "eval_accuracy_NAT": 0.9873472336178203,
      "eval_accuracy_REST": 0.0,
      "eval_accuracy_concrete": 0.8667412464894995,
      "eval_accuracy_concrete tiles": 0.7498939429648833,
      "eval_accuracy_dark bricks": 0.028214306222843252,
      "eval_accuracy_grass": 0.855339752542514,
      "eval_accuracy_red bricks": 0.0,
      "eval_accuracy_road": 0.9538643897689238,
      "eval_accuracy_rough concrete": 0.5565400818739237,
      "eval_accuracy_rough red sidewalk": 0.16307838079696793,
      "eval_accuracy_speedway bricks": 0.9877744670029537,
      "eval_accuracy_steel": 0.679419452538146,
      "eval_accuracy_tiles": 0.01720742827713654,
      "eval_accuracy_unlabeled": NaN,
      "eval_iou_NAT": 0.9704972906134676,
      "eval_iou_REST": 0.0,
      "eval_iou_concrete": 0.7774759671425241,
      "eval_iou_concrete tiles": 0.20069383120978931,
      "eval_iou_dark bricks": 0.028214306222843252,
      "eval_iou_grass": 0.7106618898582445,
      "eval_iou_red bricks": 0.0,
      "eval_iou_road": 0.9073164433288142,
      "eval_iou_rough concrete": 0.49950262592121164,
      "eval_iou_rough red sidewalk": 0.1493012048192771,
      "eval_iou_speedway bricks": 0.9308492599585683,
      "eval_iou_steel": 0.6268308483977987,
      "eval_iou_tiles": 0.01656459609215515,
      "eval_iou_unlabeled": NaN,
      "eval_loss": 0.17850728332996368,
      "eval_mean_accuracy": 0.5265708216996624,
      "eval_mean_iou": 0.4475314048895918,
      "eval_overall_accuracy": 0.9523853689937268,
      "eval_runtime": 5.1814,
      "eval_samples_per_second": 22.774,
      "eval_steps_per_second": 1.544,
      "step": 160
    }
  ],
  "max_steps": 3400,
  "num_train_epochs": 200,
  "total_flos": 1.011204696661033e+18,
  "trial_name": null,
  "trial_params": null
}