{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 2.0833333333333335,
  "eval_steps": 500,
  "global_step": 12500,
  "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
    },
    {
      "epoch": 0.75,
      "grad_norm": 19.56971549987793,
      "learning_rate": 2.25e-05,
      "loss": 2.2631,
      "step": 4500
    },
    {
      "epoch": 0.75,
      "eval_accuracy_background": 0.7136164920788886,
      "eval_accuracy_double_plant": 0.2274296812949631,
      "eval_accuracy_drydown": 0.8423346225534782,
      "eval_accuracy_endrow": 0.3372638527840968,
      "eval_accuracy_nutrient_deficiency": 0.39363536828312395,
      "eval_accuracy_planter_skip": 0.24472503933611423,
      "eval_accuracy_water": 0.8132625823496894,
      "eval_accuracy_waterway": 0.5279086387447126,
      "eval_accuracy_weed_cluster": 0.2613040701496915,
      "eval_iou_background": 0.6023960172505115,
      "eval_iou_double_plant": 0.14551908714826825,
      "eval_iou_drydown": 0.48571101575483516,
      "eval_iou_endrow": 0.15976738972856136,
      "eval_iou_nutrient_deficiency": 0.26876578893765557,
      "eval_iou_planter_skip": 0.21772117232934346,
      "eval_iou_water": 0.3885318463525075,
      "eval_iou_waterway": 0.25285421749201065,
      "eval_iou_weed_cluster": 0.19549177279724894,
      "eval_loss": 1.8094414472579956,
      "eval_mean_accuracy": 0.48460892750830653,
      "eval_mean_iou": 0.3018620341989936,
      "eval_overall_accuracy": 0.6746520910480136,
      "eval_runtime": 263.1854,
      "eval_samples_per_second": 10.798,
      "eval_steps_per_second": 3.602,
      "step": 4500
    },
    {
      "epoch": 0.8333333333333334,
      "grad_norm": 15.023950576782227,
      "learning_rate": 2.5e-05,
      "loss": 2.2139,
      "step": 5000
    },
    {
      "epoch": 0.8333333333333334,
      "eval_accuracy_background": 0.6023184738684322,
      "eval_accuracy_double_plant": 0.19349261148429422,
      "eval_accuracy_drydown": 0.4844650786080634,
      "eval_accuracy_endrow": 0.16074521211886753,
      "eval_accuracy_nutrient_deficiency": 0.6620136299007383,
      "eval_accuracy_planter_skip": 0.20313071353995618,
      "eval_accuracy_water": 0.9327820785252834,
      "eval_accuracy_waterway": 0.42715639009708817,
      "eval_accuracy_weed_cluster": 0.5896869029400611,
      "eval_iou_background": 0.5071775649990886,
      "eval_iou_double_plant": 0.14848297634630775,
      "eval_iou_drydown": 0.34958365519874773,
      "eval_iou_endrow": 0.1266887334828955,
      "eval_iou_nutrient_deficiency": 0.3268729234134353,
      "eval_iou_planter_skip": 0.1708553631916936,
      "eval_iou_water": 0.3656991869753895,
      "eval_iou_waterway": 0.15825121509439088,
      "eval_iou_weed_cluster": 0.18557336284286047,
      "eval_loss": 2.1465232372283936,
      "eval_mean_accuracy": 0.47286567678697605,
      "eval_mean_iou": 0.2599094423938677,
      "eval_overall_accuracy": 0.585979408046041,
      "eval_runtime": 258.2944,
      "eval_samples_per_second": 11.003,
      "eval_steps_per_second": 3.67,
      "step": 5000
    },
    {
      "epoch": 0.9166666666666666,
      "grad_norm": 15.944084167480469,
      "learning_rate": 2.75e-05,
      "loss": 2.1107,
      "step": 5500
    },
    {
      "epoch": 0.9166666666666666,
      "eval_accuracy_background": 0.6320754005058506,
      "eval_accuracy_double_plant": 0.18459879167713122,
      "eval_accuracy_drydown": 0.7435807472335404,
      "eval_accuracy_endrow": 0.4730924983472199,
      "eval_accuracy_nutrient_deficiency": 0.7501714321465206,
      "eval_accuracy_planter_skip": 0.394123356093544,
      "eval_accuracy_water": 0.8982667262993669,
      "eval_accuracy_waterway": 0.2070484619961853,
      "eval_accuracy_weed_cluster": 0.06027005934327404,
      "eval_iou_background": 0.5431209987641049,
      "eval_iou_double_plant": 0.1123139870852644,
      "eval_iou_drydown": 0.4467521390448892,
      "eval_iou_endrow": 0.1507248222065432,
      "eval_iou_nutrient_deficiency": 0.3257302284667843,
      "eval_iou_planter_skip": 0.2504630445874714,
      "eval_iou_water": 0.38357052112337536,
      "eval_iou_waterway": 0.1656467966651648,
      "eval_iou_weed_cluster": 0.0586611852117363,
      "eval_loss": 2.075209856033325,
      "eval_mean_accuracy": 0.48258083040473704,
      "eval_mean_iou": 0.27077596923948155,
      "eval_overall_accuracy": 0.6281377263734175,
      "eval_runtime": 259.3294,
      "eval_samples_per_second": 10.959,
      "eval_steps_per_second": 3.656,
      "step": 5500
    },
    {
      "epoch": 1.0,
      "grad_norm": 4.946102142333984,
      "learning_rate": 3e-05,
      "loss": 2.1338,
      "step": 6000
    },
    {
      "epoch": 1.0,
      "eval_accuracy_background": 0.6964471310778703,
      "eval_accuracy_double_plant": 0.2429333892576808,
      "eval_accuracy_drydown": 0.8654738033211103,
      "eval_accuracy_endrow": 0.05382795612758288,
      "eval_accuracy_nutrient_deficiency": 0.30272693922024607,
      "eval_accuracy_planter_skip": 0.39966472396096286,
      "eval_accuracy_water": 0.9846543263110985,
      "eval_accuracy_waterway": 0.7568692730591431,
      "eval_accuracy_weed_cluster": 0.10615040627525124,
      "eval_iou_background": 0.5855221723102959,
      "eval_iou_double_plant": 0.1361405520488392,
      "eval_iou_drydown": 0.4768776656490699,
      "eval_iou_endrow": 0.05007269671892798,
      "eval_iou_nutrient_deficiency": 0.22420499455655446,
      "eval_iou_planter_skip": 0.2752533913799669,
      "eval_iou_water": 0.29633141374004973,
      "eval_iou_waterway": 0.1110136523015455,
      "eval_iou_weed_cluster": 0.10019203575135141,
      "eval_loss": 2.053126335144043,
      "eval_mean_accuracy": 0.489860883178994,
      "eval_mean_iou": 0.25062317493962233,
      "eval_overall_accuracy": 0.6527557080115521,
      "eval_runtime": 261.8254,
      "eval_samples_per_second": 10.855,
      "eval_steps_per_second": 3.621,
      "step": 6000
    },
    {
      "epoch": 1.0833333333333333,
      "grad_norm": 16.1883602142334,
      "learning_rate": 3.25e-05,
      "loss": 2.0743,
      "step": 6500
    },
    {
      "epoch": 1.0833333333333333,
      "eval_accuracy_background": 0.6163441350441362,
      "eval_accuracy_double_plant": 0.3179624244479489,
      "eval_accuracy_drydown": 0.637096555419927,
      "eval_accuracy_endrow": 0.45733081369282536,
      "eval_accuracy_nutrient_deficiency": 0.8510315773068933,
      "eval_accuracy_planter_skip": 0.5177706104082623,
      "eval_accuracy_water": 0.9512477714625793,
      "eval_accuracy_waterway": 0.5107166291437305,
      "eval_accuracy_weed_cluster": 0.24186705045667198,
      "eval_iou_background": 0.5350583256945234,
      "eval_iou_double_plant": 0.17959294330409034,
      "eval_iou_drydown": 0.4634150029792384,
      "eval_iou_endrow": 0.16707740028365953,
      "eval_iou_nutrient_deficiency": 0.31532389062386124,
      "eval_iou_planter_skip": 0.2948590617515534,
      "eval_iou_water": 0.34520224120688753,
      "eval_iou_waterway": 0.11975132688004231,
      "eval_iou_weed_cluster": 0.2113183362317597,
      "eval_loss": 1.8855762481689453,
      "eval_mean_accuracy": 0.5668186185981083,
      "eval_mean_iou": 0.29239983655062396,
      "eval_overall_accuracy": 0.6228305054183748,
      "eval_runtime": 259.3394,
      "eval_samples_per_second": 10.959,
      "eval_steps_per_second": 3.655,
      "step": 6500
    },
    {
      "epoch": 1.1666666666666667,
      "grad_norm": 9.279212951660156,
      "learning_rate": 3.5000000000000004e-05,
      "loss": 2.0725,
      "step": 7000
    },
    {
      "epoch": 1.1666666666666667,
      "eval_accuracy_background": 0.7373301969750995,
      "eval_accuracy_double_plant": 0.33336127425920425,
      "eval_accuracy_drydown": 0.6301703388238222,
      "eval_accuracy_endrow": 0.37253283281326316,
      "eval_accuracy_nutrient_deficiency": 0.34279961797329944,
      "eval_accuracy_planter_skip": 0.5799829420962596,
      "eval_accuracy_water": 0.9469093944591689,
      "eval_accuracy_waterway": 0.3649137190752288,
      "eval_accuracy_weed_cluster": 0.4194977852900649,
      "eval_iou_background": 0.6045108089230737,
      "eval_iou_double_plant": 0.15103332950916856,
      "eval_iou_drydown": 0.4284455196537357,
      "eval_iou_endrow": 0.18315189002409357,
      "eval_iou_nutrient_deficiency": 0.2540752037931447,
      "eval_iou_planter_skip": 0.2268832410220617,
      "eval_iou_water": 0.3615440261043886,
      "eval_iou_waterway": 0.2757438894003987,
      "eval_iou_weed_cluster": 0.24843146863053017,
      "eval_loss": 1.8260598182678223,
      "eval_mean_accuracy": 0.5252775668628233,
      "eval_mean_iou": 0.3037577085622884,
      "eval_overall_accuracy": 0.6668996089543237,
      "eval_runtime": 257.8864,
      "eval_samples_per_second": 11.02,
      "eval_steps_per_second": 3.676,
      "step": 7000
    },
    {
      "epoch": 1.25,
      "grad_norm": 12.69166088104248,
      "learning_rate": 3.7500000000000003e-05,
      "loss": 2.1051,
      "step": 7500
    },
    {
      "epoch": 1.25,
      "eval_accuracy_background": 0.6136021677052333,
      "eval_accuracy_double_plant": 0.21675932855086671,
      "eval_accuracy_drydown": 0.8978391275698134,
      "eval_accuracy_endrow": 0.31490227354840317,
      "eval_accuracy_nutrient_deficiency": 0.6391839292944432,
      "eval_accuracy_planter_skip": 0.4095695820323414,
      "eval_accuracy_water": 0.9815472332260521,
      "eval_accuracy_waterway": 0.3863849586942106,
      "eval_accuracy_weed_cluster": 0.41607248209352754,
      "eval_iou_background": 0.5478072558942303,
      "eval_iou_double_plant": 0.15633489787457702,
      "eval_iou_drydown": 0.4606893739380495,
      "eval_iou_endrow": 0.18368040175633407,
      "eval_iou_nutrient_deficiency": 0.36167812896859036,
      "eval_iou_planter_skip": 0.3133978757416113,
      "eval_iou_water": 0.35102829123972634,
      "eval_iou_waterway": 0.2651990401867848,
      "eval_iou_weed_cluster": 0.27107161264823754,
      "eval_loss": 1.7672898769378662,
      "eval_mean_accuracy": 0.5417623425238768,
      "eval_mean_iou": 0.32343187536090456,
      "eval_overall_accuracy": 0.6462444076512336,
      "eval_runtime": 263.0897,
      "eval_samples_per_second": 10.802,
      "eval_steps_per_second": 3.603,
      "step": 7500
    },
    {
      "epoch": 1.3333333333333333,
      "grad_norm": 26.389280319213867,
      "learning_rate": 3.9999999999999996e-05,
      "loss": 2.0567,
      "step": 8000
    },
    {
      "epoch": 1.3333333333333333,
      "eval_accuracy_background": 0.6516741544849038,
      "eval_accuracy_double_plant": 0.22472217370629263,
      "eval_accuracy_drydown": 0.9205657832973893,
      "eval_accuracy_endrow": 0.1334717500678728,
      "eval_accuracy_nutrient_deficiency": 0.579114420966078,
      "eval_accuracy_planter_skip": 0.6448299356407252,
      "eval_accuracy_water": 0.8737892085078403,
      "eval_accuracy_waterway": 0.4514893265909921,
      "eval_accuracy_weed_cluster": 0.36867061298945264,
      "eval_iou_background": 0.5743802241628614,
      "eval_iou_double_plant": 0.17379933416714446,
      "eval_iou_drydown": 0.4870945365813324,
      "eval_iou_endrow": 0.1124492323800547,
      "eval_iou_nutrient_deficiency": 0.35858088912475333,
      "eval_iou_planter_skip": 0.22081427365766507,
      "eval_iou_water": 0.4188886664719046,
      "eval_iou_waterway": 0.23882510300705312,
      "eval_iou_weed_cluster": 0.22928729873239137,
      "eval_loss": 1.7339626550674438,
      "eval_mean_accuracy": 0.5387030406946163,
      "eval_mean_iou": 0.3126799509205734,
      "eval_overall_accuracy": 0.6655563976415058,
      "eval_runtime": 262.6835,
      "eval_samples_per_second": 10.819,
      "eval_steps_per_second": 3.609,
      "step": 8000
    },
    {
      "epoch": 1.4166666666666667,
      "grad_norm": 16.169328689575195,
      "learning_rate": 4.25e-05,
      "loss": 2.0731,
      "step": 8500
    },
    {
      "epoch": 1.4166666666666667,
      "eval_accuracy_background": 0.7486936916842247,
      "eval_accuracy_double_plant": 0.2930715215459597,
      "eval_accuracy_drydown": 0.7255160536916373,
      "eval_accuracy_endrow": 0.4577216435093236,
      "eval_accuracy_nutrient_deficiency": 0.4830792419842683,
      "eval_accuracy_planter_skip": 0.7417698065300401,
      "eval_accuracy_water": 0.9756065468432125,
      "eval_accuracy_waterway": 0.273890366668282,
      "eval_accuracy_weed_cluster": 0.35391598936308855,
      "eval_iou_background": 0.6323410133400419,
      "eval_iou_double_plant": 0.17111536697625435,
      "eval_iou_drydown": 0.5110135373059105,
      "eval_iou_endrow": 0.16559443160045,
      "eval_iou_nutrient_deficiency": 0.3404264291064937,
      "eval_iou_planter_skip": 0.21812035844976185,
      "eval_iou_water": 0.35579748623298973,
      "eval_iou_waterway": 0.20254597755885112,
      "eval_iou_weed_cluster": 0.24793962096921052,
      "eval_loss": 1.6961616277694702,
      "eval_mean_accuracy": 0.5614738735355596,
      "eval_mean_iou": 0.31609935794888483,
      "eval_overall_accuracy": 0.6986505820712626,
      "eval_runtime": 262.6198,
      "eval_samples_per_second": 10.822,
      "eval_steps_per_second": 3.61,
      "step": 8500
    },
    {
      "epoch": 1.5,
      "grad_norm": 9.605523109436035,
      "learning_rate": 4.5e-05,
      "loss": 2.0062,
      "step": 9000
    },
    {
      "epoch": 1.5,
      "eval_accuracy_background": 0.6385280729636846,
      "eval_accuracy_double_plant": 0.38205242780086,
      "eval_accuracy_drydown": 0.9496890843481527,
      "eval_accuracy_endrow": 0.34276586617075205,
      "eval_accuracy_nutrient_deficiency": 0.4966985380725304,
      "eval_accuracy_planter_skip": 0.7026444652494228,
      "eval_accuracy_water": 0.7442517574456261,
      "eval_accuracy_waterway": 0.46809502667206265,
      "eval_accuracy_weed_cluster": 0.17348492439927163,
      "eval_iou_background": 0.5609579182369832,
      "eval_iou_double_plant": 0.16239615673750246,
      "eval_iou_drydown": 0.45202834914092793,
      "eval_iou_endrow": 0.18367824595658963,
      "eval_iou_nutrient_deficiency": 0.34617965104488957,
      "eval_iou_planter_skip": 0.2580270293767756,
      "eval_iou_water": 0.3425501321917906,
      "eval_iou_waterway": 0.19866797232183966,
      "eval_iou_weed_cluster": 0.13472834814631707,
      "eval_loss": 1.9120891094207764,
      "eval_mean_accuracy": 0.5442455736802626,
      "eval_mean_iou": 0.2932459781281795,
      "eval_overall_accuracy": 0.6456323905823719,
      "eval_runtime": 258.4421,
      "eval_samples_per_second": 10.997,
      "eval_steps_per_second": 3.668,
      "step": 9000
    },
    {
      "epoch": 1.5833333333333335,
      "grad_norm": 9.02011489868164,
      "learning_rate": 4.7499999999999996e-05,
      "loss": 2.0069,
      "step": 9500
    },
    {
      "epoch": 1.5833333333333335,
      "eval_accuracy_background": 0.5921818727290818,
      "eval_accuracy_double_plant": 0.3459265635975844,
      "eval_accuracy_drydown": 0.9022492563261855,
      "eval_accuracy_endrow": 0.39012380134585,
      "eval_accuracy_nutrient_deficiency": 0.6176093671439052,
      "eval_accuracy_planter_skip": 0.7456583223453638,
      "eval_accuracy_water": 0.930193012579653,
      "eval_accuracy_waterway": 0.6027143991593465,
      "eval_accuracy_weed_cluster": 0.38137543613044333,
      "eval_iou_background": 0.5323074634115271,
      "eval_iou_double_plant": 0.132029668948867,
      "eval_iou_drydown": 0.4864590469093194,
      "eval_iou_endrow": 0.2008338441059683,
      "eval_iou_nutrient_deficiency": 0.36281261663843545,
      "eval_iou_planter_skip": 0.18127130680379444,
      "eval_iou_water": 0.3502148412760265,
      "eval_iou_waterway": 0.14129351940530874,
      "eval_iou_weed_cluster": 0.21689374484941243,
      "eval_loss": 1.9075040817260742,
      "eval_mean_accuracy": 0.6120035590397126,
      "eval_mean_iou": 0.28934622803873994,
      "eval_overall_accuracy": 0.6314946434667654,
      "eval_runtime": 258.4278,
      "eval_samples_per_second": 10.997,
      "eval_steps_per_second": 3.668,
      "step": 9500
    },
    {
      "epoch": 1.6666666666666665,
      "grad_norm": 27.558940887451172,
      "learning_rate": 5e-05,
      "loss": 1.9695,
      "step": 10000
    },
    {
      "epoch": 1.6666666666666665,
      "eval_accuracy_background": 0.6535160468633152,
      "eval_accuracy_double_plant": 0.27480931114887563,
      "eval_accuracy_drydown": 0.8507298791874478,
      "eval_accuracy_endrow": 0.3002457236689853,
      "eval_accuracy_nutrient_deficiency": 0.5805433393846879,
      "eval_accuracy_planter_skip": 0.5759424736901442,
      "eval_accuracy_water": 0.9136638372037981,
      "eval_accuracy_waterway": 0.300740548651248,
      "eval_accuracy_weed_cluster": 0.2748341582674426,
      "eval_iou_background": 0.5715406090850156,
      "eval_iou_double_plant": 0.1830877307497648,
      "eval_iou_drydown": 0.44761615381963227,
      "eval_iou_endrow": 0.1607454389243595,
      "eval_iou_nutrient_deficiency": 0.3589882634051457,
      "eval_iou_planter_skip": 0.3142600611891689,
      "eval_iou_water": 0.381127146789507,
      "eval_iou_waterway": 0.11509687922710479,
      "eval_iou_weed_cluster": 0.191443588646255,
      "eval_loss": 1.9981021881103516,
      "eval_mean_accuracy": 0.5250028131184383,
      "eval_mean_iou": 0.3026562079817726,
      "eval_overall_accuracy": 0.6541074851018032,
      "eval_runtime": 262.1608,
      "eval_samples_per_second": 10.841,
      "eval_steps_per_second": 3.616,
      "step": 10000
    },
    {
      "epoch": 1.75,
      "grad_norm": 7.4598588943481445,
      "learning_rate": 5.25e-05,
      "loss": 1.9823,
      "step": 10500
    },
    {
      "epoch": 1.75,
      "eval_accuracy_background": 0.6656899643375446,
      "eval_accuracy_double_plant": 0.3934759106723288,
      "eval_accuracy_drydown": 0.6984513863750703,
      "eval_accuracy_endrow": 0.5897825722024841,
      "eval_accuracy_nutrient_deficiency": 0.672282770766344,
      "eval_accuracy_planter_skip": 0.5736107366759635,
      "eval_accuracy_water": 0.9725679178509335,
      "eval_accuracy_waterway": 0.3880805850347442,
      "eval_accuracy_weed_cluster": 0.2804900072264515,
      "eval_iou_background": 0.5767367514980629,
      "eval_iou_double_plant": 0.20286343375264382,
      "eval_iou_drydown": 0.46795251362795115,
      "eval_iou_endrow": 0.13363987546509035,
      "eval_iou_nutrient_deficiency": 0.3626840890315017,
      "eval_iou_planter_skip": 0.30331216870602024,
      "eval_iou_water": 0.30117376418991876,
      "eval_iou_waterway": 0.2072977234811509,
      "eval_iou_weed_cluster": 0.2100599157730429,
      "eval_loss": 1.87357759475708,
      "eval_mean_accuracy": 0.5816035390157627,
      "eval_mean_iou": 0.30730224839170917,
      "eval_overall_accuracy": 0.6533269610793297,
      "eval_runtime": 262.5666,
      "eval_samples_per_second": 10.824,
      "eval_steps_per_second": 3.611,
      "step": 10500
    },
    {
      "epoch": 1.8333333333333335,
      "grad_norm": 9.449281692504883,
      "learning_rate": 5.5e-05,
      "loss": 1.8992,
      "step": 11000
    },
    {
      "epoch": 1.8333333333333335,
      "eval_accuracy_background": 0.591561636780099,
      "eval_accuracy_double_plant": 0.13478224563452249,
      "eval_accuracy_drydown": 0.8104318370166057,
      "eval_accuracy_endrow": 0.18974519899334122,
      "eval_accuracy_nutrient_deficiency": 0.8002396916859228,
      "eval_accuracy_planter_skip": 0.555837185249815,
      "eval_accuracy_water": 0.9841123189100232,
      "eval_accuracy_waterway": 0.6447291734607411,
      "eval_accuracy_weed_cluster": 0.4545554666421275,
      "eval_iou_background": 0.5351854519544275,
      "eval_iou_double_plant": 0.12235773192803939,
      "eval_iou_drydown": 0.48958264330531226,
      "eval_iou_endrow": 0.13265128465179174,
      "eval_iou_nutrient_deficiency": 0.3848348426109684,
      "eval_iou_planter_skip": 0.3516838331732518,
      "eval_iou_water": 0.33706546823750627,
      "eval_iou_waterway": 0.0876245178196897,
      "eval_iou_weed_cluster": 0.23276521867619226,
      "eval_loss": 1.8864208459854126,
      "eval_mean_accuracy": 0.5739994171525775,
      "eval_mean_iou": 0.29708344359524214,
      "eval_overall_accuracy": 0.6340729206003646,
      "eval_runtime": 266.7514,
      "eval_samples_per_second": 10.654,
      "eval_steps_per_second": 3.554,
      "step": 11000
    },
    {
      "epoch": 1.9166666666666665,
      "grad_norm": 179.62493896484375,
      "learning_rate": 5.75e-05,
      "loss": 1.9102,
      "step": 11500
    },
    {
      "epoch": 1.9166666666666665,
      "eval_accuracy_background": 0.5262653554571423,
      "eval_accuracy_double_plant": 0.12008314072155157,
      "eval_accuracy_drydown": 0.8570712958580466,
      "eval_accuracy_endrow": 0.3906394963666471,
      "eval_accuracy_nutrient_deficiency": 0.848468676066121,
      "eval_accuracy_planter_skip": 0.3694477258579194,
      "eval_accuracy_water": 0.9433195760325397,
      "eval_accuracy_waterway": 0.2684022825470543,
      "eval_accuracy_weed_cluster": 0.5539906575032456,
      "eval_iou_background": 0.48726462380145824,
      "eval_iou_double_plant": 0.10638346834228546,
      "eval_iou_drydown": 0.5237704902459679,
      "eval_iou_endrow": 0.1661201420060771,
      "eval_iou_nutrient_deficiency": 0.37603227109002707,
      "eval_iou_planter_skip": 0.3088275424635105,
      "eval_iou_water": 0.3755045365931884,
      "eval_iou_waterway": 0.1621157246203886,
      "eval_iou_weed_cluster": 0.18121117948664975,
      "eval_loss": 2.0767364501953125,
      "eval_mean_accuracy": 0.5419653562678075,
      "eval_mean_iou": 0.2985811087388392,
      "eval_overall_accuracy": 0.604590198595902,
      "eval_runtime": 264.8767,
      "eval_samples_per_second": 10.73,
      "eval_steps_per_second": 3.579,
      "step": 11500
    },
    {
      "epoch": 2.0,
      "grad_norm": 5.2313337326049805,
      "learning_rate": 6e-05,
      "loss": 1.9478,
      "step": 12000
    },
    {
      "epoch": 2.0,
      "eval_accuracy_background": 0.5917584355396107,
      "eval_accuracy_double_plant": 0.23022929832367192,
      "eval_accuracy_drydown": 0.8591774260328934,
      "eval_accuracy_endrow": 0.44466450809326863,
      "eval_accuracy_nutrient_deficiency": 0.4915070798422274,
      "eval_accuracy_planter_skip": 0.616599595119871,
      "eval_accuracy_water": 0.9331229726538546,
      "eval_accuracy_waterway": 0.6528746410873313,
      "eval_accuracy_weed_cluster": 0.6726995315007896,
      "eval_iou_background": 0.5297567240991148,
      "eval_iou_double_plant": 0.1865127044756558,
      "eval_iou_drydown": 0.48548291796169635,
      "eval_iou_endrow": 0.1696600303497616,
      "eval_iou_nutrient_deficiency": 0.3447111530279667,
      "eval_iou_planter_skip": 0.2809826623294383,
      "eval_iou_water": 0.34996397672243307,
      "eval_iou_waterway": 0.06688915132136661,
      "eval_iou_weed_cluster": 0.27669622818924317,
      "eval_loss": 1.8301477432250977,
      "eval_mean_accuracy": 0.6102926097992799,
      "eval_mean_iou": 0.29896172760851963,
      "eval_overall_accuracy": 0.6270241082274692,
      "eval_runtime": 263.5983,
      "eval_samples_per_second": 10.782,
      "eval_steps_per_second": 3.596,
      "step": 12000
    },
    {
      "epoch": 2.0833333333333335,
      "grad_norm": 24.462024688720703,
      "learning_rate": 5.999682695781323e-05,
      "loss": 1.9239,
      "step": 12500
    },
    {
      "epoch": 2.0833333333333335,
      "eval_accuracy_background": 0.7518201291388158,
      "eval_accuracy_double_plant": 0.23938112017827798,
      "eval_accuracy_drydown": 0.7562345973159798,
      "eval_accuracy_endrow": 0.2983546116536712,
      "eval_accuracy_nutrient_deficiency": 0.48790565425147486,
      "eval_accuracy_planter_skip": 0.6943331487027661,
      "eval_accuracy_water": 0.9779127526764066,
      "eval_accuracy_waterway": 0.5641306051386559,
      "eval_accuracy_weed_cluster": 0.3079581583540586,
      "eval_iou_background": 0.6365747880470238,
      "eval_iou_double_plant": 0.1964527495585162,
      "eval_iou_drydown": 0.519540905379824,
      "eval_iou_endrow": 0.20298551726617572,
      "eval_iou_nutrient_deficiency": 0.35134456727900654,
      "eval_iou_planter_skip": 0.23709257653367372,
      "eval_iou_water": 0.34692142879031984,
      "eval_iou_waterway": 0.09302081664144977,
      "eval_iou_weed_cluster": 0.23107629770672056,
      "eval_loss": 1.7629526853561401,
      "eval_mean_accuracy": 0.5642256419344563,
      "eval_mean_iou": 0.31277884968919,
      "eval_overall_accuracy": 0.702497874611006,
      "eval_runtime": 263.8984,
      "eval_samples_per_second": 10.769,
      "eval_steps_per_second": 3.592,
      "step": 12500
    }
  ],
  "logging_steps": 500,
  "max_steps": 120000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 20,
  "save_steps": 500,
  "total_flos": 1.99604465565696e+19,
  "train_batch_size": 3,
  "trial_name": null,
  "trial_params": null
}