{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.6666666666666666,
  "eval_steps": 500,
  "global_step": 4000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.08333333333333333,
      "grad_norm": 10.64107608795166,
      "learning_rate": 2.4999999999999998e-06,
      "loss": 4.1627,
      "step": 500
    },
    {
      "epoch": 0.08333333333333333,
      "eval_accuracy_background": 0.014990974836900402,
      "eval_accuracy_double_plant": 0.09625850817128659,
      "eval_accuracy_drydown": 0.11748866381009407,
      "eval_accuracy_endrow": 0.019624906998737878,
      "eval_accuracy_nutrient_deficiency": 0.39043216558572647,
      "eval_accuracy_planter_skip": 0.03244089779472675,
      "eval_accuracy_water": 0.4902746667077927,
      "eval_accuracy_waterway": 0.37818186421979577,
      "eval_accuracy_weed_cluster": 0.5157577782392276,
      "eval_iou_background": 0.01485215844086368,
      "eval_iou_double_plant": 0.025830652956995166,
      "eval_iou_drydown": 0.09098524126821965,
      "eval_iou_endrow": 0.011190399377301194,
      "eval_iou_nutrient_deficiency": 0.16172838363740288,
      "eval_iou_planter_skip": 0.007145844854739052,
      "eval_iou_water": 0.052629358314533047,
      "eval_iou_waterway": 0.008009602169192399,
      "eval_iou_weed_cluster": 0.055193229922437775,
      "eval_loss": 3.9555599689483643,
      "eval_mean_accuracy": 0.22838338070714315,
      "eval_mean_iou": 0.047507207882409426,
      "eval_overall_accuracy": 0.09656532361323403,
      "eval_runtime": 348.036,
      "eval_samples_per_second": 8.166,
      "eval_steps_per_second": 2.724,
      "step": 500
    },
    {
      "epoch": 0.16666666666666666,
      "grad_norm": 12.677851676940918,
      "learning_rate": 4.9999999999999996e-06,
      "loss": 3.7698,
      "step": 1000
    },
    {
      "epoch": 0.16666666666666666,
      "eval_accuracy_background": 0.29766305371935275,
      "eval_accuracy_double_plant": 0.04300475016987658,
      "eval_accuracy_drydown": 0.7567223885695242,
      "eval_accuracy_endrow": 0.006538411852764754,
      "eval_accuracy_nutrient_deficiency": 0.5373960102281351,
      "eval_accuracy_planter_skip": 0.1493345947032268,
      "eval_accuracy_water": 0.2900425128140623,
      "eval_accuracy_waterway": 0.37062750835043584,
      "eval_accuracy_weed_cluster": 0.242435460451953,
      "eval_iou_background": 0.2681855463534359,
      "eval_iou_double_plant": 0.02937987848319999,
      "eval_iou_drydown": 0.30192234804386486,
      "eval_iou_endrow": 0.006310524335748097,
      "eval_iou_nutrient_deficiency": 0.19394253102070813,
      "eval_iou_planter_skip": 0.08231306246610917,
      "eval_iou_water": 0.10160883330705332,
      "eval_iou_waterway": 0.050415164283578985,
      "eval_iou_weed_cluster": 0.07122746911823011,
      "eval_loss": 3.33693528175354,
      "eval_mean_accuracy": 0.29930718787325905,
      "eval_mean_iou": 0.12281170637910319,
      "eval_overall_accuracy": 0.37769782397579943,
      "eval_runtime": 260.2627,
      "eval_samples_per_second": 10.92,
      "eval_steps_per_second": 3.642,
      "step": 1000
    },
    {
      "epoch": 0.25,
      "grad_norm": 12.100262641906738,
      "learning_rate": 7.5e-06,
      "loss": 3.2674,
      "step": 1500
    },
    {
      "epoch": 0.25,
      "eval_accuracy_background": 0.48419612667845113,
      "eval_accuracy_double_plant": 0.013344873041957285,
      "eval_accuracy_drydown": 0.8872887901430043,
      "eval_accuracy_endrow": 0.015151865793565866,
      "eval_accuracy_nutrient_deficiency": 0.37784363150955347,
      "eval_accuracy_planter_skip": 0.1236399017700026,
      "eval_accuracy_water": 0.6308252089425159,
      "eval_accuracy_waterway": 0.4153131997350911,
      "eval_accuracy_weed_cluster": 0.2533755890828509,
      "eval_iou_background": 0.4220656310747218,
      "eval_iou_double_plant": 0.012970922185309422,
      "eval_iou_drydown": 0.3650381802310761,
      "eval_iou_endrow": 0.014768985148023165,
      "eval_iou_nutrient_deficiency": 0.20002194862446826,
      "eval_iou_planter_skip": 0.09927397940127604,
      "eval_iou_water": 0.27391064172674834,
      "eval_iou_waterway": 0.10636423788784055,
      "eval_iou_weed_cluster": 0.10586226367786514,
      "eval_loss": 2.763476610183716,
      "eval_mean_accuracy": 0.3556643540774437,
      "eval_mean_iou": 0.17780853221748097,
      "eval_overall_accuracy": 0.5152837603493446,
      "eval_runtime": 258.1136,
      "eval_samples_per_second": 11.011,
      "eval_steps_per_second": 3.673,
      "step": 1500
    },
    {
      "epoch": 0.3333333333333333,
      "grad_norm": 13.388419151306152,
      "learning_rate": 9.999999999999999e-06,
      "loss": 2.8338,
      "step": 2000
    },
    {
      "epoch": 0.3333333333333333,
      "eval_accuracy_background": 0.6007479203520831,
      "eval_accuracy_double_plant": 0.09356295249957611,
      "eval_accuracy_drydown": 0.8340823867683501,
      "eval_accuracy_endrow": 0.0035413015410058713,
      "eval_accuracy_nutrient_deficiency": 0.44282241132222916,
      "eval_accuracy_planter_skip": 0.11649665949385082,
      "eval_accuracy_water": 0.921699186070695,
      "eval_accuracy_waterway": 0.20547725725890914,
      "eval_accuracy_weed_cluster": 0.3415216522793772,
      "eval_iou_background": 0.5156964205169746,
      "eval_iou_double_plant": 0.07643992513295463,
      "eval_iou_drydown": 0.4276993634847662,
      "eval_iou_endrow": 0.0035320673127108605,
      "eval_iou_nutrient_deficiency": 0.23942297475711152,
      "eval_iou_planter_skip": 0.09915967157890872,
      "eval_iou_water": 0.37311176318003697,
      "eval_iou_waterway": 0.1337041070891874,
      "eval_iou_weed_cluster": 0.17456818424609255,
      "eval_loss": 2.2476296424865723,
      "eval_mean_accuracy": 0.39555019195400853,
      "eval_mean_iou": 0.22703716414430483,
      "eval_overall_accuracy": 0.6012494768590696,
      "eval_runtime": 261.9728,
      "eval_samples_per_second": 10.848,
      "eval_steps_per_second": 3.619,
      "step": 2000
    },
    {
      "epoch": 0.4166666666666667,
      "grad_norm": 31.341535568237305,
      "learning_rate": 1.25e-05,
      "loss": 2.5604,
      "step": 2500
    },
    {
      "epoch": 0.4166666666666667,
      "eval_accuracy_background": 0.7311239150003357,
      "eval_accuracy_double_plant": 0.18906004388106473,
      "eval_accuracy_drydown": 0.7049642054247133,
      "eval_accuracy_endrow": 0.03270138529568011,
      "eval_accuracy_nutrient_deficiency": 0.2058331492617626,
      "eval_accuracy_planter_skip": 0.07169662420163618,
      "eval_accuracy_water": 0.9476640149608306,
      "eval_accuracy_waterway": 0.4427819971981128,
      "eval_accuracy_weed_cluster": 0.38249564148248877,
      "eval_iou_background": 0.5934145085795158,
      "eval_iou_double_plant": 0.10950501729968068,
      "eval_iou_drydown": 0.42479480844402817,
      "eval_iou_endrow": 0.030470447854384766,
      "eval_iou_nutrient_deficiency": 0.16681197423835636,
      "eval_iou_planter_skip": 0.06706541524720425,
      "eval_iou_water": 0.36387793500767635,
      "eval_iou_waterway": 0.19571015933545038,
      "eval_iou_weed_cluster": 0.22406855402046205,
      "eval_loss": 2.0893733501434326,
      "eval_mean_accuracy": 0.41203566407851394,
      "eval_mean_iou": 0.24174653555852876,
      "eval_overall_accuracy": 0.6543493727987272,
      "eval_runtime": 259.8765,
      "eval_samples_per_second": 10.936,
      "eval_steps_per_second": 3.648,
      "step": 2500
    },
    {
      "epoch": 0.5,
      "grad_norm": 22.26491928100586,
      "learning_rate": 1.5e-05,
      "loss": 2.4032,
      "step": 3000
    },
    {
      "epoch": 0.5,
      "eval_accuracy_background": 0.6732741082932401,
      "eval_accuracy_double_plant": 0.10359602903136565,
      "eval_accuracy_drydown": 0.8666517772278536,
      "eval_accuracy_endrow": 0.035669653990401784,
      "eval_accuracy_nutrient_deficiency": 0.2536647750774501,
      "eval_accuracy_planter_skip": 0.24370499629921916,
      "eval_accuracy_water": 0.9767091931497752,
      "eval_accuracy_waterway": 0.6842904410593037,
      "eval_accuracy_weed_cluster": 0.3698701654570609,
      "eval_iou_background": 0.5681057490790173,
      "eval_iou_double_plant": 0.0852643551765096,
      "eval_iou_drydown": 0.4543271536581631,
      "eval_iou_endrow": 0.03370061965697607,
      "eval_iou_nutrient_deficiency": 0.19633806402515588,
      "eval_iou_planter_skip": 0.18150485646008124,
      "eval_iou_water": 0.3528168740365702,
      "eval_iou_waterway": 0.16230336403850296,
      "eval_iou_weed_cluster": 0.22603294672854196,
      "eval_loss": 1.9570292234420776,
      "eval_mean_accuracy": 0.4674923488428522,
      "eval_mean_iou": 0.2511548869843909,
      "eval_overall_accuracy": 0.6432186597875551,
      "eval_runtime": 259.2993,
      "eval_samples_per_second": 10.96,
      "eval_steps_per_second": 3.656,
      "step": 3000
    },
    {
      "epoch": 0.5833333333333334,
      "grad_norm": 31.124582290649414,
      "learning_rate": 1.7500000000000002e-05,
      "loss": 2.3362,
      "step": 3500
    },
    {
      "epoch": 0.5833333333333334,
      "eval_accuracy_background": 0.5814538721345998,
      "eval_accuracy_double_plant": 0.06263123204822073,
      "eval_accuracy_drydown": 0.9572617680178966,
      "eval_accuracy_endrow": 0.1022870538789039,
      "eval_accuracy_nutrient_deficiency": 0.155304235691705,
      "eval_accuracy_planter_skip": 0.28837366612584614,
      "eval_accuracy_water": 0.957647648655113,
      "eval_accuracy_waterway": 0.4168520985122232,
      "eval_accuracy_weed_cluster": 0.20255586893627922,
      "eval_iou_background": 0.5035989534908836,
      "eval_iou_double_plant": 0.05887727499276805,
      "eval_iou_drydown": 0.35033214248192274,
      "eval_iou_endrow": 0.08288492557854046,
      "eval_iou_nutrient_deficiency": 0.1308384862727777,
      "eval_iou_planter_skip": 0.23854848545793084,
      "eval_iou_water": 0.33185434658316154,
      "eval_iou_waterway": 0.19380165775707148,
      "eval_iou_weed_cluster": 0.16104566462906844,
      "eval_loss": 2.403474807739258,
      "eval_mean_accuracy": 0.4138186048889764,
      "eval_mean_iou": 0.22797577080490272,
      "eval_overall_accuracy": 0.5765593656585961,
      "eval_runtime": 258.5568,
      "eval_samples_per_second": 10.992,
      "eval_steps_per_second": 3.667,
      "step": 3500
    },
    {
      "epoch": 0.6666666666666666,
      "grad_norm": 27.981061935424805,
      "learning_rate": 1.9999999999999998e-05,
      "loss": 2.2454,
      "step": 4000
    },
    {
      "epoch": 0.6666666666666666,
      "eval_accuracy_background": 0.6165356472563173,
      "eval_accuracy_double_plant": 0.2464305201601748,
      "eval_accuracy_drydown": 0.9388557439528662,
      "eval_accuracy_endrow": 0.08952317035344279,
      "eval_accuracy_nutrient_deficiency": 0.30307985133610366,
      "eval_accuracy_planter_skip": 0.34189676046879824,
      "eval_accuracy_water": 0.9684396578613939,
      "eval_accuracy_waterway": 0.4364168499156771,
      "eval_accuracy_weed_cluster": 0.48568842831047665,
      "eval_iou_background": 0.5406762574665314,
      "eval_iou_double_plant": 0.12820291315879165,
      "eval_iou_drydown": 0.445376041256294,
      "eval_iou_endrow": 0.07582305802889684,
      "eval_iou_nutrient_deficiency": 0.2473209944226292,
      "eval_iou_planter_skip": 0.2571484714289717,
      "eval_iou_water": 0.3523385474174709,
      "eval_iou_waterway": 0.2182523975088832,
      "eval_iou_weed_cluster": 0.22609664246227254,
      "eval_loss": 1.9709433317184448,
      "eval_mean_accuracy": 0.49187406995725,
      "eval_mean_iou": 0.27680392479452676,
      "eval_overall_accuracy": 0.6261566315021039,
      "eval_runtime": 262.6128,
      "eval_samples_per_second": 10.822,
      "eval_steps_per_second": 3.61,
      "step": 4000
    }
  ],
  "logging_steps": 500,
  "max_steps": 120000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 20,
  "save_steps": 500,
  "total_flos": 6.387342898102272e+18,
  "train_batch_size": 3,
  "trial_name": null,
  "trial_params": null
}