File size: 32,648 Bytes
6d795fe
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 5.0,
  "eval_steps": 500,
  "global_step": 11250,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.2222222222222222,
      "grad_norm": 11.793246269226074,
      "learning_rate": 2.2222222222222223e-05,
      "loss": 3.818,
      "step": 500
    },
    {
      "epoch": 0.2222222222222222,
      "eval_accuracy_background": 0.19581472704788175,
      "eval_accuracy_double_plant": 0.09380215019766877,
      "eval_accuracy_drydown": 0.7107036127331178,
      "eval_accuracy_endrow": 0.02059395079032939,
      "eval_accuracy_nutrient_deficiency": 0.8348166019686053,
      "eval_accuracy_planter_skip": 0.09523015915808461,
      "eval_accuracy_water": 0.9933604093368263,
      "eval_accuracy_waterway": 0.5081417568068439,
      "eval_accuracy_weed_cluster": 0.8598379016004754,
      "eval_iou_background": 0.18808867925529232,
      "eval_iou_double_plant": 0.06203210246822897,
      "eval_iou_drydown": 0.40717237628074626,
      "eval_iou_endrow": 0.0177635377630849,
      "eval_iou_nutrient_deficiency": 0.24616684931318317,
      "eval_iou_planter_skip": 0.08719568132734196,
      "eval_iou_water": 0.2711830537857341,
      "eval_iou_waterway": 0.11493665271237699,
      "eval_iou_weed_cluster": 0.15174042308492286,
      "eval_loss": 3.0543487071990967,
      "eval_mean_accuracy": 0.47914458551553707,
      "eval_mean_iou": 0.1718088173323235,
      "eval_overall_accuracy": 0.3698257009825982,
      "eval_runtime": 233.3045,
      "eval_samples_per_second": 12.182,
      "eval_steps_per_second": 1.526,
      "step": 500
    },
    {
      "epoch": 0.4444444444444444,
      "grad_norm": 18.055360794067383,
      "learning_rate": 4.4444444444444447e-05,
      "loss": 3.0561,
      "step": 1000
    },
    {
      "epoch": 0.4444444444444444,
      "eval_accuracy_background": 0.33281217700519594,
      "eval_accuracy_double_plant": 0.6079763269177568,
      "eval_accuracy_drydown": 0.7946684041913001,
      "eval_accuracy_endrow": 0.349301342050525,
      "eval_accuracy_nutrient_deficiency": 0.7314999843672831,
      "eval_accuracy_planter_skip": 0.6712138070986369,
      "eval_accuracy_water": 0.995121665952459,
      "eval_accuracy_waterway": 0.5701844801294333,
      "eval_accuracy_weed_cluster": 0.5467936481137453,
      "eval_iou_background": 0.3071003729227365,
      "eval_iou_double_plant": 0.05692166554215038,
      "eval_iou_drydown": 0.45070670178291583,
      "eval_iou_endrow": 0.12562451540796912,
      "eval_iou_nutrient_deficiency": 0.2946294028718886,
      "eval_iou_planter_skip": 0.13924388642718094,
      "eval_iou_water": 0.2832997795518204,
      "eval_iou_waterway": 0.10604657522478783,
      "eval_iou_weed_cluster": 0.21995946645651349,
      "eval_loss": 2.7527103424072266,
      "eval_mean_accuracy": 0.6221746484251484,
      "eval_mean_iou": 0.2203924851319959,
      "eval_overall_accuracy": 0.4604543823237725,
      "eval_runtime": 234.4163,
      "eval_samples_per_second": 12.124,
      "eval_steps_per_second": 1.519,
      "step": 1000
    },
    {
      "epoch": 0.6666666666666666,
      "grad_norm": 17.42981719970703,
      "learning_rate": 6.666666666666667e-05,
      "loss": 2.8425,
      "step": 1500
    },
    {
      "epoch": 0.6666666666666666,
      "eval_accuracy_background": 0.3979410567243586,
      "eval_accuracy_double_plant": 0.35090828194540236,
      "eval_accuracy_drydown": 0.990671482434014,
      "eval_accuracy_endrow": 0.2249281722842768,
      "eval_accuracy_nutrient_deficiency": 0.64935174422475,
      "eval_accuracy_planter_skip": 0.5494840964457799,
      "eval_accuracy_water": 0.9732364233609112,
      "eval_accuracy_waterway": 0.32217860917602775,
      "eval_accuracy_weed_cluster": 0.4993965740432129,
      "eval_iou_background": 0.3749426436452205,
      "eval_iou_double_plant": 0.1311586100117858,
      "eval_iou_drydown": 0.377378210407481,
      "eval_iou_endrow": 0.10949437953809864,
      "eval_iou_nutrient_deficiency": 0.3194503546667628,
      "eval_iou_planter_skip": 0.2767050562366308,
      "eval_iou_water": 0.33601793640817545,
      "eval_iou_waterway": 0.17211292189101207,
      "eval_iou_weed_cluster": 0.23999589853669023,
      "eval_loss": 2.7259955406188965,
      "eval_mean_accuracy": 0.5508996045154149,
      "eval_mean_iou": 0.2596951123713175,
      "eval_overall_accuracy": 0.5187459341357686,
      "eval_runtime": 234.4997,
      "eval_samples_per_second": 12.119,
      "eval_steps_per_second": 1.518,
      "step": 1500
    },
    {
      "epoch": 0.8888888888888888,
      "grad_norm": 18.323888778686523,
      "learning_rate": 8.888888888888889e-05,
      "loss": 2.812,
      "step": 2000
    },
    {
      "epoch": 0.8888888888888888,
      "eval_accuracy_background": 0.3444616006736896,
      "eval_accuracy_double_plant": 0.2994677094262697,
      "eval_accuracy_drydown": 0.9219474623777996,
      "eval_accuracy_endrow": 0.08748836836582362,
      "eval_accuracy_nutrient_deficiency": 0.8971747201033099,
      "eval_accuracy_planter_skip": 0.4553112332178167,
      "eval_accuracy_water": 0.9835769083095529,
      "eval_accuracy_waterway": 0.3921009308045625,
      "eval_accuracy_weed_cluster": 0.7258725441601974,
      "eval_iou_background": 0.32680002116655066,
      "eval_iou_double_plant": 0.1689556986493995,
      "eval_iou_drydown": 0.4498001115160996,
      "eval_iou_endrow": 0.07367449850616072,
      "eval_iou_nutrient_deficiency": 0.3152010347192682,
      "eval_iou_planter_skip": 0.28500014113702443,
      "eval_iou_water": 0.3018250386605094,
      "eval_iou_waterway": 0.11357719882212529,
      "eval_iou_weed_cluster": 0.2522793183619568,
      "eval_loss": 2.639113187789917,
      "eval_mean_accuracy": 0.5674890530487802,
      "eval_mean_iou": 0.25412367350434384,
      "eval_overall_accuracy": 0.5034953329987246,
      "eval_runtime": 234.7079,
      "eval_samples_per_second": 12.109,
      "eval_steps_per_second": 1.517,
      "step": 2000
    },
    {
      "epoch": 1.1111111111111112,
      "grad_norm": 10.998316764831543,
      "learning_rate": 9.996239762521151e-05,
      "loss": 2.759,
      "step": 2500
    },
    {
      "epoch": 1.1111111111111112,
      "eval_accuracy_background": 0.3645173692584377,
      "eval_accuracy_double_plant": 0.4073825238114058,
      "eval_accuracy_drydown": 0.8372221461982765,
      "eval_accuracy_endrow": 0.8605400739588867,
      "eval_accuracy_nutrient_deficiency": 0.8390058376603935,
      "eval_accuracy_planter_skip": 0.2988588850601193,
      "eval_accuracy_water": 0.9820329895300751,
      "eval_accuracy_waterway": 0.7229222574531815,
      "eval_accuracy_weed_cluster": 0.7084585704978812,
      "eval_iou_background": 0.3475988528346604,
      "eval_iou_double_plant": 0.17493931420950803,
      "eval_iou_drydown": 0.5046039726642503,
      "eval_iou_endrow": 0.0706611160751562,
      "eval_iou_nutrient_deficiency": 0.32536830412947676,
      "eval_iou_planter_skip": 0.22907108809940935,
      "eval_iou_water": 0.3462153514773095,
      "eval_iou_waterway": 0.16650439461800443,
      "eval_iou_weed_cluster": 0.27121508109484677,
      "eval_loss": 2.593132972717285,
      "eval_mean_accuracy": 0.6689934059365175,
      "eval_mean_iou": 0.2706863861336246,
      "eval_overall_accuracy": 0.5069850298500573,
      "eval_runtime": 234.81,
      "eval_samples_per_second": 12.103,
      "eval_steps_per_second": 1.516,
      "step": 2500
    },
    {
      "epoch": 1.3333333333333333,
      "grad_norm": 10.760461807250977,
      "learning_rate": 9.966191788709716e-05,
      "loss": 2.6787,
      "step": 3000
    },
    {
      "epoch": 1.3333333333333333,
      "eval_accuracy_background": 0.5289161626135687,
      "eval_accuracy_double_plant": 0.5163129324203556,
      "eval_accuracy_drydown": 0.8819552587741175,
      "eval_accuracy_endrow": 0.178768114089838,
      "eval_accuracy_nutrient_deficiency": 0.7478542163797729,
      "eval_accuracy_planter_skip": 0.6799633353103509,
      "eval_accuracy_water": 0.7100977137717751,
      "eval_accuracy_waterway": 0.6130317877550806,
      "eval_accuracy_weed_cluster": 0.42893667138117214,
      "eval_iou_background": 0.4823402935296036,
      "eval_iou_double_plant": 0.1500710257143286,
      "eval_iou_drydown": 0.4647006334399156,
      "eval_iou_endrow": 0.11586272712955523,
      "eval_iou_nutrient_deficiency": 0.35666112710148684,
      "eval_iou_planter_skip": 0.2656678841196218,
      "eval_iou_water": 0.29190270392369394,
      "eval_iou_waterway": 0.13711629169569617,
      "eval_iou_weed_cluster": 0.2189985744693017,
      "eval_loss": 2.7097718715667725,
      "eval_mean_accuracy": 0.5873151324995591,
      "eval_mean_iou": 0.2759245845692448,
      "eval_overall_accuracy": 0.5953067288821898,
      "eval_runtime": 235.7529,
      "eval_samples_per_second": 12.055,
      "eval_steps_per_second": 1.51,
      "step": 3000
    },
    {
      "epoch": 1.5555555555555556,
      "grad_norm": 7.579084873199463,
      "learning_rate": 9.906276553136923e-05,
      "loss": 2.5987,
      "step": 3500
    },
    {
      "epoch": 1.5555555555555556,
      "eval_accuracy_background": 0.44845745777345614,
      "eval_accuracy_double_plant": 0.43811743603015324,
      "eval_accuracy_drydown": 0.8981147894499133,
      "eval_accuracy_endrow": 0.36041987177051604,
      "eval_accuracy_nutrient_deficiency": 0.8981291552297538,
      "eval_accuracy_planter_skip": 0.7046899431893379,
      "eval_accuracy_water": 0.9661965673637197,
      "eval_accuracy_waterway": 0.4783779264725078,
      "eval_accuracy_weed_cluster": 0.5656546529493687,
      "eval_iou_background": 0.4213139993195897,
      "eval_iou_double_plant": 0.18850742709320553,
      "eval_iou_drydown": 0.48988628203968987,
      "eval_iou_endrow": 0.16427164736852504,
      "eval_iou_nutrient_deficiency": 0.3396249313110222,
      "eval_iou_planter_skip": 0.2878780809609733,
      "eval_iou_water": 0.34577830408088805,
      "eval_iou_waterway": 0.0966323667968418,
      "eval_iou_weed_cluster": 0.2644303620138278,
      "eval_loss": 2.5603725910186768,
      "eval_mean_accuracy": 0.6397953111365252,
      "eval_mean_iou": 0.28870260010939597,
      "eval_overall_accuracy": 0.5673802237308736,
      "eval_runtime": 236.5689,
      "eval_samples_per_second": 12.013,
      "eval_steps_per_second": 1.505,
      "step": 3500
    },
    {
      "epoch": 1.7777777777777777,
      "grad_norm": 30.224449157714844,
      "learning_rate": 9.816854393079403e-05,
      "loss": 2.5903,
      "step": 4000
    },
    {
      "epoch": 1.7777777777777777,
      "eval_accuracy_background": 0.48452708688190493,
      "eval_accuracy_double_plant": 0.45566285012742336,
      "eval_accuracy_drydown": 0.7706441910353777,
      "eval_accuracy_endrow": 0.4346390238476997,
      "eval_accuracy_nutrient_deficiency": 0.9163122738336061,
      "eval_accuracy_planter_skip": 0.6643195710035243,
      "eval_accuracy_water": 0.9529314710981239,
      "eval_accuracy_waterway": 0.5652691719502191,
      "eval_accuracy_weed_cluster": 0.7199459132433054,
      "eval_iou_background": 0.4513627214329878,
      "eval_iou_double_plant": 0.1934941937869372,
      "eval_iou_drydown": 0.47740329126353237,
      "eval_iou_endrow": 0.1927445666887825,
      "eval_iou_nutrient_deficiency": 0.3408237922045895,
      "eval_iou_planter_skip": 0.1993885722146088,
      "eval_iou_water": 0.36254479085706903,
      "eval_iou_waterway": 0.10718524988313392,
      "eval_iou_weed_cluster": 0.32265929353301015,
      "eval_loss": 2.4760828018188477,
      "eval_mean_accuracy": 0.6626946170023538,
      "eval_mean_iou": 0.2941784968738501,
      "eval_overall_accuracy": 0.5830334441140828,
      "eval_runtime": 237.1396,
      "eval_samples_per_second": 11.984,
      "eval_steps_per_second": 1.501,
      "step": 4000
    },
    {
      "epoch": 2.0,
      "grad_norm": 4.368917942047119,
      "learning_rate": 9.698463103929542e-05,
      "loss": 2.5103,
      "step": 4500
    },
    {
      "epoch": 2.0,
      "eval_accuracy_background": 0.5553855635459563,
      "eval_accuracy_double_plant": 0.34347881100417776,
      "eval_accuracy_drydown": 0.9424376735830182,
      "eval_accuracy_endrow": 0.6260475378930491,
      "eval_accuracy_nutrient_deficiency": 0.5435000667444707,
      "eval_accuracy_planter_skip": 0.6585385101783727,
      "eval_accuracy_water": 0.9437360659302082,
      "eval_accuracy_waterway": 0.6383330299192482,
      "eval_accuracy_weed_cluster": 0.5861360126229717,
      "eval_iou_background": 0.5073018855198684,
      "eval_iou_double_plant": 0.24133669629976653,
      "eval_iou_drydown": 0.46988570286035836,
      "eval_iou_endrow": 0.16053099249771158,
      "eval_iou_nutrient_deficiency": 0.3529766621270875,
      "eval_iou_planter_skip": 0.2706109321909118,
      "eval_iou_water": 0.384981488447517,
      "eval_iou_waterway": 0.08851952603929993,
      "eval_iou_weed_cluster": 0.29452505578421073,
      "eval_loss": 2.4755921363830566,
      "eval_mean_accuracy": 0.6486214746023858,
      "eval_mean_iou": 0.30785210464074797,
      "eval_overall_accuracy": 0.6176850367895912,
      "eval_runtime": 237.7605,
      "eval_samples_per_second": 11.953,
      "eval_steps_per_second": 1.497,
      "step": 4500
    },
    {
      "epoch": 2.2222222222222223,
      "grad_norm": 6.629311561584473,
      "learning_rate": 9.551814704830734e-05,
      "loss": 2.4353,
      "step": 5000
    },
    {
      "epoch": 2.2222222222222223,
      "eval_accuracy_background": 0.42591159650814375,
      "eval_accuracy_double_plant": 0.4570086359771186,
      "eval_accuracy_drydown": 0.9760591262490192,
      "eval_accuracy_endrow": 0.3551154320957804,
      "eval_accuracy_nutrient_deficiency": 0.6534028202751002,
      "eval_accuracy_planter_skip": 0.5061506487395288,
      "eval_accuracy_water": 0.9413386637859449,
      "eval_accuracy_waterway": 0.7338525861575946,
      "eval_accuracy_weed_cluster": 0.4176907906433075,
      "eval_iou_background": 0.39859839745224973,
      "eval_iou_double_plant": 0.12951991556215345,
      "eval_iou_drydown": 0.3929487238858669,
      "eval_iou_endrow": 0.19821401528351465,
      "eval_iou_nutrient_deficiency": 0.3492317794369294,
      "eval_iou_planter_skip": 0.342223631855141,
      "eval_iou_water": 0.3651455303939148,
      "eval_iou_waterway": 0.06171327109097799,
      "eval_iou_weed_cluster": 0.2469832540646728,
      "eval_loss": 2.772688388824463,
      "eval_mean_accuracy": 0.6073922556035042,
      "eval_mean_iou": 0.27606427989171345,
      "eval_overall_accuracy": 0.5348577429690352,
      "eval_runtime": 235.7496,
      "eval_samples_per_second": 12.055,
      "eval_steps_per_second": 1.51,
      "step": 5000
    },
    {
      "epoch": 2.4444444444444446,
      "grad_norm": 1.828034520149231,
      "learning_rate": 9.377791156510455e-05,
      "loss": 2.4343,
      "step": 5500
    },
    {
      "epoch": 2.4444444444444446,
      "eval_accuracy_background": 0.5359506409790418,
      "eval_accuracy_double_plant": 0.3328800104029485,
      "eval_accuracy_drydown": 0.8475921756634195,
      "eval_accuracy_endrow": 0.5246471478576723,
      "eval_accuracy_nutrient_deficiency": 0.8727763125095295,
      "eval_accuracy_planter_skip": 0.6149790942645249,
      "eval_accuracy_water": 0.9438895752632102,
      "eval_accuracy_waterway": 0.5362387742245587,
      "eval_accuracy_weed_cluster": 0.7168151228394442,
      "eval_iou_background": 0.49774515001598935,
      "eval_iou_double_plant": 0.2583518771192604,
      "eval_iou_drydown": 0.49474623169120435,
      "eval_iou_endrow": 0.2086936981590746,
      "eval_iou_nutrient_deficiency": 0.389153487897357,
      "eval_iou_planter_skip": 0.33342953856833324,
      "eval_iou_water": 0.38659165736650947,
      "eval_iou_waterway": 0.1222013652080862,
      "eval_iou_weed_cluster": 0.32389362495378293,
      "eval_loss": 2.37630033493042,
      "eval_mean_accuracy": 0.6584187615560388,
      "eval_mean_iou": 0.33497851455328864,
      "eval_overall_accuracy": 0.6244067353340839,
      "eval_runtime": 235.2095,
      "eval_samples_per_second": 12.083,
      "eval_steps_per_second": 1.514,
      "step": 5500
    },
    {
      "epoch": 2.6666666666666665,
      "grad_norm": 4.322945594787598,
      "learning_rate": 9.177439057064683e-05,
      "loss": 2.429,
      "step": 6000
    },
    {
      "epoch": 2.6666666666666665,
      "eval_accuracy_background": 0.47508722733281505,
      "eval_accuracy_double_plant": 0.3903129553680362,
      "eval_accuracy_drydown": 0.9349512677593288,
      "eval_accuracy_endrow": 0.8204605885347837,
      "eval_accuracy_nutrient_deficiency": 0.7945893080826532,
      "eval_accuracy_planter_skip": 0.6755439657665518,
      "eval_accuracy_water": 0.9602463423637888,
      "eval_accuracy_waterway": 0.5908533095491616,
      "eval_accuracy_weed_cluster": 0.5223400460036128,
      "eval_iou_background": 0.4473297070942847,
      "eval_iou_double_plant": 0.21522522959513646,
      "eval_iou_drydown": 0.5045700487256383,
      "eval_iou_endrow": 0.09909322405768327,
      "eval_iou_nutrient_deficiency": 0.3792928669727522,
      "eval_iou_planter_skip": 0.3511935674413129,
      "eval_iou_water": 0.3847106205841943,
      "eval_iou_waterway": 0.11231181305994274,
      "eval_iou_weed_cluster": 0.2714616419201456,
      "eval_loss": 2.444581985473633,
      "eval_mean_accuracy": 0.6849316678623035,
      "eval_mean_iou": 0.30724319105012116,
      "eval_overall_accuracy": 0.5833008089471211,
      "eval_runtime": 235.7629,
      "eval_samples_per_second": 12.054,
      "eval_steps_per_second": 1.51,
      "step": 6000
    },
    {
      "epoch": 2.888888888888889,
      "grad_norm": 7.450633525848389,
      "learning_rate": 8.951963347593797e-05,
      "loss": 2.3447,
      "step": 6500
    },
    {
      "epoch": 2.888888888888889,
      "eval_accuracy_background": 0.47111716561960343,
      "eval_accuracy_double_plant": 0.5341091774148928,
      "eval_accuracy_drydown": 0.941938268968352,
      "eval_accuracy_endrow": 0.7311690146391068,
      "eval_accuracy_nutrient_deficiency": 0.7887680533155264,
      "eval_accuracy_planter_skip": 0.6696045801451882,
      "eval_accuracy_water": 0.937607014146304,
      "eval_accuracy_waterway": 0.6133181757260874,
      "eval_accuracy_weed_cluster": 0.791186490124323,
      "eval_iou_background": 0.44812222855512157,
      "eval_iou_double_plant": 0.19977490534329798,
      "eval_iou_drydown": 0.49045901876145187,
      "eval_iou_endrow": 0.1522359916466749,
      "eval_iou_nutrient_deficiency": 0.3821088495393014,
      "eval_iou_planter_skip": 0.32711815378396797,
      "eval_iou_water": 0.40061432551146625,
      "eval_iou_waterway": 0.11404079137084261,
      "eval_iou_weed_cluster": 0.361816623385642,
      "eval_loss": 2.3683271408081055,
      "eval_mean_accuracy": 0.7198686600110427,
      "eval_mean_iou": 0.31958787643308517,
      "eval_overall_accuracy": 0.5941963743061932,
      "eval_runtime": 237.4801,
      "eval_samples_per_second": 11.967,
      "eval_steps_per_second": 1.499,
      "step": 6500
    },
    {
      "epoch": 3.111111111111111,
      "grad_norm": 5.715631484985352,
      "learning_rate": 8.702720065545024e-05,
      "loss": 2.3524,
      "step": 7000
    },
    {
      "epoch": 3.111111111111111,
      "eval_accuracy_background": 0.5733005382373251,
      "eval_accuracy_double_plant": 0.4085941294733635,
      "eval_accuracy_drydown": 0.9132687213925081,
      "eval_accuracy_endrow": 0.5473116504244036,
      "eval_accuracy_nutrient_deficiency": 0.6654354745523554,
      "eval_accuracy_planter_skip": 0.6567846831788482,
      "eval_accuracy_water": 0.9470583573485105,
      "eval_accuracy_waterway": 0.6380754553721537,
      "eval_accuracy_weed_cluster": 0.6830693870978963,
      "eval_iou_background": 0.5288164660085214,
      "eval_iou_double_plant": 0.22413327990472365,
      "eval_iou_drydown": 0.48867003992878844,
      "eval_iou_endrow": 0.21578814947417663,
      "eval_iou_nutrient_deficiency": 0.36825716293633476,
      "eval_iou_planter_skip": 0.36583626453645124,
      "eval_iou_water": 0.39093973802796883,
      "eval_iou_waterway": 0.11535891831010817,
      "eval_iou_weed_cluster": 0.35715285838197053,
      "eval_loss": 2.431631088256836,
      "eval_mean_accuracy": 0.6703220441197071,
      "eval_mean_iou": 0.33943920861211596,
      "eval_overall_accuracy": 0.6409103665091717,
      "eval_runtime": 236.9321,
      "eval_samples_per_second": 11.995,
      "eval_steps_per_second": 1.503,
      "step": 7000
    },
    {
      "epoch": 3.3333333333333335,
      "grad_norm": 4.925364971160889,
      "learning_rate": 8.43120818934367e-05,
      "loss": 2.2918,
      "step": 7500
    },
    {
      "epoch": 3.3333333333333335,
      "eval_accuracy_background": 0.5207783038430598,
      "eval_accuracy_double_plant": 0.511623956358611,
      "eval_accuracy_drydown": 0.7907720914484996,
      "eval_accuracy_endrow": 0.8393525385113304,
      "eval_accuracy_nutrient_deficiency": 0.7608358372413602,
      "eval_accuracy_planter_skip": 0.7122150276210597,
      "eval_accuracy_water": 0.9862896196935537,
      "eval_accuracy_waterway": 0.7186587438482248,
      "eval_accuracy_weed_cluster": 0.6876786054552028,
      "eval_iou_background": 0.4833249353556612,
      "eval_iou_double_plant": 0.2770853527046839,
      "eval_iou_drydown": 0.49633234539651844,
      "eval_iou_endrow": 0.10903224364863091,
      "eval_iou_nutrient_deficiency": 0.34932685343766695,
      "eval_iou_planter_skip": 0.4145313361231283,
      "eval_iou_water": 0.3976396288185673,
      "eval_iou_waterway": 0.1271568225173445,
      "eval_iou_weed_cluster": 0.33225514281148677,
      "eval_loss": 2.4041595458984375,
      "eval_mean_accuracy": 0.7253560804467668,
      "eval_mean_iou": 0.3318538512015209,
      "eval_overall_accuracy": 0.6005182097024693,
      "eval_runtime": 235.653,
      "eval_samples_per_second": 12.06,
      "eval_steps_per_second": 1.511,
      "step": 7500
    },
    {
      "epoch": 3.5555555555555554,
      "grad_norm": 6.189846992492676,
      "learning_rate": 8.139060623360493e-05,
      "loss": 2.2655,
      "step": 8000
    },
    {
      "epoch": 3.5555555555555554,
      "eval_accuracy_background": 0.660148747631652,
      "eval_accuracy_double_plant": 0.5109490714476035,
      "eval_accuracy_drydown": 0.7900188075274437,
      "eval_accuracy_endrow": 0.6243997662274682,
      "eval_accuracy_nutrient_deficiency": 0.6157337403081103,
      "eval_accuracy_planter_skip": 0.6394037576405194,
      "eval_accuracy_water": 0.960773194952663,
      "eval_accuracy_waterway": 0.7705940673160164,
      "eval_accuracy_weed_cluster": 0.4267326357404514,
      "eval_iou_background": 0.5818582932469415,
      "eval_iou_double_plant": 0.23837002693064407,
      "eval_iou_drydown": 0.49145768489122593,
      "eval_iou_endrow": 0.19309707416375207,
      "eval_iou_nutrient_deficiency": 0.3764808008201772,
      "eval_iou_planter_skip": 0.298692826471094,
      "eval_iou_water": 0.4087861680876253,
      "eval_iou_waterway": 0.13748257441187617,
      "eval_iou_weed_cluster": 0.2731917403167738,
      "eval_loss": 2.3547887802124023,
      "eval_mean_accuracy": 0.6665281987546587,
      "eval_mean_iou": 0.3332685765933456,
      "eval_overall_accuracy": 0.6675489762542709,
      "eval_runtime": 235.8283,
      "eval_samples_per_second": 12.051,
      "eval_steps_per_second": 1.51,
      "step": 8000
    },
    {
      "epoch": 3.7777777777777777,
      "grad_norm": 14.043067932128906,
      "learning_rate": 7.828034377432693e-05,
      "loss": 2.086,
      "step": 8500
    },
    {
      "epoch": 3.7777777777777777,
      "eval_accuracy_background": 0.5113278819399572,
      "eval_accuracy_double_plant": 0.5218351961325509,
      "eval_accuracy_drydown": 0.9519253082367062,
      "eval_accuracy_endrow": 0.6842155529197043,
      "eval_accuracy_nutrient_deficiency": 0.8463735263255534,
      "eval_accuracy_planter_skip": 0.628710706775615,
      "eval_accuracy_water": 0.9762014177950836,
      "eval_accuracy_waterway": 0.648506351395421,
      "eval_accuracy_weed_cluster": 0.6639366229727125,
      "eval_iou_background": 0.4848819009356754,
      "eval_iou_double_plant": 0.2879336679760272,
      "eval_iou_drydown": 0.4894900709205923,
      "eval_iou_endrow": 0.19852289972941828,
      "eval_iou_nutrient_deficiency": 0.3829074318401349,
      "eval_iou_planter_skip": 0.3657439037882685,
      "eval_iou_water": 0.4192142240551455,
      "eval_iou_waterway": 0.15043160267945166,
      "eval_iou_weed_cluster": 0.3861382244896622,
      "eval_loss": 2.1546285152435303,
      "eval_mean_accuracy": 0.7147813960548115,
      "eval_mean_iou": 0.3516959918238196,
      "eval_overall_accuracy": 0.6220420455837136,
      "eval_runtime": 235.6843,
      "eval_samples_per_second": 12.059,
      "eval_steps_per_second": 1.51,
      "step": 8500
    },
    {
      "epoch": 4.0,
      "grad_norm": 18.264894485473633,
      "learning_rate": 7.500000000000001e-05,
      "loss": 1.9832,
      "step": 9000
    },
    {
      "epoch": 4.0,
      "eval_accuracy_background": 0.5073105699646216,
      "eval_accuracy_double_plant": 0.6275967531581745,
      "eval_accuracy_drydown": 0.9145034082156106,
      "eval_accuracy_endrow": 0.7164097057049754,
      "eval_accuracy_nutrient_deficiency": 0.789913970524425,
      "eval_accuracy_planter_skip": 0.6452701079843733,
      "eval_accuracy_water": 0.9795285230949086,
      "eval_accuracy_waterway": 0.6648322120109195,
      "eval_accuracy_weed_cluster": 0.597354833490017,
      "eval_iou_background": 0.4765089227639876,
      "eval_iou_double_plant": 0.23250998862027894,
      "eval_iou_drydown": 0.49137847672958673,
      "eval_iou_endrow": 0.14681924325595433,
      "eval_iou_nutrient_deficiency": 0.37568387674202897,
      "eval_iou_planter_skip": 0.379643445665233,
      "eval_iou_water": 0.39053093405963807,
      "eval_iou_waterway": 0.11916817100899851,
      "eval_iou_weed_cluster": 0.3378135364676862,
      "eval_loss": 2.1923012733459473,
      "eval_mean_accuracy": 0.7158577871275584,
      "eval_mean_iou": 0.3277840661459325,
      "eval_overall_accuracy": 0.6071390321455458,
      "eval_runtime": 236.5123,
      "eval_samples_per_second": 12.016,
      "eval_steps_per_second": 1.505,
      "step": 9000
    },
    {
      "epoch": 4.222222222222222,
      "grad_norm": 89.83843231201172,
      "learning_rate": 7.156930328406268e-05,
      "loss": 1.8817,
      "step": 9500
    },
    {
      "epoch": 4.222222222222222,
      "eval_accuracy_background": 0.6665794060557415,
      "eval_accuracy_double_plant": 0.6738756911395181,
      "eval_accuracy_drydown": 0.8593353095770565,
      "eval_accuracy_endrow": 0.528491718138303,
      "eval_accuracy_nutrient_deficiency": 0.5577531153361905,
      "eval_accuracy_planter_skip": 0.4867982608780899,
      "eval_accuracy_water": 0.9665819453233988,
      "eval_accuracy_waterway": 0.5507764562893811,
      "eval_accuracy_weed_cluster": 0.37931683965997287,
      "eval_iou_background": 0.5948571171387201,
      "eval_iou_double_plant": 0.14222147527739493,
      "eval_iou_drydown": 0.5067474903020236,
      "eval_iou_endrow": 0.20952001702399436,
      "eval_iou_nutrient_deficiency": 0.36320208791310965,
      "eval_iou_planter_skip": 0.36895812412485374,
      "eval_iou_water": 0.43796631436628747,
      "eval_iou_waterway": 0.14631637863228455,
      "eval_iou_weed_cluster": 0.27229885013621663,
      "eval_loss": 2.3856027126312256,
      "eval_mean_accuracy": 0.6299454158219614,
      "eval_mean_iou": 0.3380097616572095,
      "eval_overall_accuracy": 0.6742306501318911,
      "eval_runtime": 237.4369,
      "eval_samples_per_second": 11.969,
      "eval_steps_per_second": 1.499,
      "step": 9500
    },
    {
      "epoch": 4.444444444444445,
      "grad_norm": 1.9701658487319946,
      "learning_rate": 6.800888624023553e-05,
      "loss": 1.8543,
      "step": 10000
    },
    {
      "epoch": 4.444444444444445,
      "eval_accuracy_background": 0.5527180589769325,
      "eval_accuracy_double_plant": 0.4246191003744297,
      "eval_accuracy_drydown": 0.8353909056852167,
      "eval_accuracy_endrow": 0.6753884637676755,
      "eval_accuracy_nutrient_deficiency": 0.8477198680961048,
      "eval_accuracy_planter_skip": 0.4963550984995907,
      "eval_accuracy_water": 0.9550066106182281,
      "eval_accuracy_waterway": 0.5634499971404859,
      "eval_accuracy_weed_cluster": 0.6110140820972658,
      "eval_iou_background": 0.5084311951913111,
      "eval_iou_double_plant": 0.27885082999204747,
      "eval_iou_drydown": 0.49256853689719754,
      "eval_iou_endrow": 0.18385884370260622,
      "eval_iou_nutrient_deficiency": 0.3838676199858262,
      "eval_iou_planter_skip": 0.33103959389402615,
      "eval_iou_water": 0.42705295972807256,
      "eval_iou_waterway": 0.11857219976282583,
      "eval_iou_weed_cluster": 0.32056690645254854,
      "eval_loss": 2.1975200176239014,
      "eval_mean_accuracy": 0.6624069094728811,
      "eval_mean_iou": 0.33831207617849574,
      "eval_overall_accuracy": 0.628797372897935,
      "eval_runtime": 236.3076,
      "eval_samples_per_second": 12.027,
      "eval_steps_per_second": 1.507,
      "step": 10000
    },
    {
      "epoch": 4.666666666666667,
      "grad_norm": 2.4208943843841553,
      "learning_rate": 6.434016163555452e-05,
      "loss": 1.8277,
      "step": 10500
    },
    {
      "epoch": 4.666666666666667,
      "eval_accuracy_background": 0.5099154200330375,
      "eval_accuracy_double_plant": 0.4769406088529602,
      "eval_accuracy_drydown": 0.9297325261412196,
      "eval_accuracy_endrow": 0.7408691239954655,
      "eval_accuracy_nutrient_deficiency": 0.8186861646399846,
      "eval_accuracy_planter_skip": 0.5929283224924147,
      "eval_accuracy_water": 0.9649930078370883,
      "eval_accuracy_waterway": 0.6425751517397851,
      "eval_accuracy_weed_cluster": 0.5683679700188848,
      "eval_iou_background": 0.4796036750402124,
      "eval_iou_double_plant": 0.2854657885552567,
      "eval_iou_drydown": 0.4818667626801526,
      "eval_iou_endrow": 0.1440330128654955,
      "eval_iou_nutrient_deficiency": 0.38900088125772797,
      "eval_iou_planter_skip": 0.3263530494113775,
      "eval_iou_water": 0.42266776750368934,
      "eval_iou_waterway": 0.13362089548065528,
      "eval_iou_weed_cluster": 0.3196479507378374,
      "eval_loss": 2.244593620300293,
      "eval_mean_accuracy": 0.6938898106389823,
      "eval_mean_iou": 0.33136219817026724,
      "eval_overall_accuracy": 0.6106220341578938,
      "eval_runtime": 234.8197,
      "eval_samples_per_second": 12.103,
      "eval_steps_per_second": 1.516,
      "step": 10500
    },
    {
      "epoch": 4.888888888888889,
      "grad_norm": 5.2590179443359375,
      "learning_rate": 6.058519361147055e-05,
      "loss": 1.8497,
      "step": 11000
    },
    {
      "epoch": 4.888888888888889,
      "eval_accuracy_background": 0.594185126170412,
      "eval_accuracy_double_plant": 0.5687051962472893,
      "eval_accuracy_drydown": 0.9177861283215228,
      "eval_accuracy_endrow": 0.6953884844921907,
      "eval_accuracy_nutrient_deficiency": 0.5195121987300781,
      "eval_accuracy_planter_skip": 0.5532373254383343,
      "eval_accuracy_water": 0.970162492570799,
      "eval_accuracy_waterway": 0.775521162805013,
      "eval_accuracy_weed_cluster": 0.6501758234958537,
      "eval_iou_background": 0.540882435487803,
      "eval_iou_double_plant": 0.28772831895138,
      "eval_iou_drydown": 0.5010427869734542,
      "eval_iou_endrow": 0.15782366525775998,
      "eval_iou_nutrient_deficiency": 0.36487790306357637,
      "eval_iou_planter_skip": 0.3377777205835922,
      "eval_iou_water": 0.44182527301028707,
      "eval_iou_waterway": 0.09707009096755505,
      "eval_iou_weed_cluster": 0.32469730465714464,
      "eval_loss": 2.1957969665527344,
      "eval_mean_accuracy": 0.6938526598079436,
      "eval_mean_iou": 0.33930283321695026,
      "eval_overall_accuracy": 0.6446164491310151,
      "eval_runtime": 235.4392,
      "eval_samples_per_second": 12.071,
      "eval_steps_per_second": 1.512,
      "step": 11000
    }
  ],
  "logging_steps": 500,
  "max_steps": 22500,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 10,
  "save_steps": 2250,
  "total_flos": 4.790507173576704e+19,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}