{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 1.0,
  "eval_steps": 500,
  "global_step": 6000,
  "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
    }
  ],
  "logging_steps": 500,
  "max_steps": 120000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 20,
  "save_steps": 500,
  "total_flos": 9.581014347153408e+18,
  "train_batch_size": 3,
  "trial_name": null,
  "trial_params": null
}