File size: 15,082 Bytes
475c70b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
{
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.8333333333333334,
  "eval_steps": 500,
  "global_step": 5000,
  "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
    }
  ],
  "logging_steps": 500,
  "max_steps": 120000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 20,
  "save_steps": 500,
  "total_flos": 7.98417862262784e+18,
  "train_batch_size": 3,
  "trial_name": null,
  "trial_params": null
}