fayetitchenal commited on
Commit
f69dbc3
1 Parent(s): de1bcbf

Training in progress, step 10000, checkpoint

Browse files
checkpoint-10000/config.json ADDED
@@ -0,0 +1,92 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b5",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 768,
9
+ "depths": [
10
+ 3,
11
+ 6,
12
+ 40,
13
+ 3
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
+ "hidden_act": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_sizes": [
25
+ 64,
26
+ 128,
27
+ 320,
28
+ 512
29
+ ],
30
+ "id2label": {
31
+ "0": "background",
32
+ "1": "double_plant",
33
+ "2": "drydown",
34
+ "3": "endrow",
35
+ "4": "nutrient_deficiency",
36
+ "5": "planter_skip",
37
+ "6": "water",
38
+ "7": "waterway",
39
+ "8": "weed_cluster"
40
+ },
41
+ "image_size": 224,
42
+ "initializer_range": 0.02,
43
+ "label2id": {
44
+ "background": 0,
45
+ "double_plant": 1,
46
+ "drydown": 2,
47
+ "endrow": 3,
48
+ "nutrient_deficiency": 4,
49
+ "planter_skip": 5,
50
+ "water": 6,
51
+ "waterway": 7,
52
+ "weed_cluster": 8
53
+ },
54
+ "layer_norm_eps": 1e-06,
55
+ "mlp_ratios": [
56
+ 4,
57
+ 4,
58
+ 4,
59
+ 4
60
+ ],
61
+ "model_type": "segformer",
62
+ "num_attention_heads": [
63
+ 1,
64
+ 2,
65
+ 5,
66
+ 8
67
+ ],
68
+ "num_channels": 4,
69
+ "num_encoder_blocks": 4,
70
+ "patch_sizes": [
71
+ 7,
72
+ 3,
73
+ 3,
74
+ 3
75
+ ],
76
+ "reshape_last_stage": true,
77
+ "semantic_loss_ignore_index": 255,
78
+ "sr_ratios": [
79
+ 8,
80
+ 4,
81
+ 2,
82
+ 1
83
+ ],
84
+ "strides": [
85
+ 4,
86
+ 2,
87
+ 2,
88
+ 2
89
+ ],
90
+ "torch_dtype": "float32",
91
+ "transformers_version": "4.40.2"
92
+ }
checkpoint-10000/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fa8b100497283f63057cc90fd08a8b8fb23e243da8e0f4c5ce5793a3f15c4f6a
3
+ size 338562516
checkpoint-10000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ea43b4154a3007e56f049756204193a32057849e7848712686acda81d282cd66
3
+ size 677518143
checkpoint-10000/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d5be3cb7c203eba1178fc70d1be9e93d7ca0270f34a0c65b811a4cb85d825f7
3
+ size 14575
checkpoint-10000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1fe95b9050b33c9621c656793c07e622f0c03b299402070da8b08725730fe0f7
3
+ size 563
checkpoint-10000/trainer_state.json ADDED
@@ -0,0 +1,741 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.6666666666666665,
5
+ "eval_steps": 500,
6
+ "global_step": 10000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.08333333333333333,
13
+ "grad_norm": 10.64107608795166,
14
+ "learning_rate": 2.4999999999999998e-06,
15
+ "loss": 4.1627,
16
+ "step": 500
17
+ },
18
+ {
19
+ "epoch": 0.08333333333333333,
20
+ "eval_accuracy_background": 0.014990974836900402,
21
+ "eval_accuracy_double_plant": 0.09625850817128659,
22
+ "eval_accuracy_drydown": 0.11748866381009407,
23
+ "eval_accuracy_endrow": 0.019624906998737878,
24
+ "eval_accuracy_nutrient_deficiency": 0.39043216558572647,
25
+ "eval_accuracy_planter_skip": 0.03244089779472675,
26
+ "eval_accuracy_water": 0.4902746667077927,
27
+ "eval_accuracy_waterway": 0.37818186421979577,
28
+ "eval_accuracy_weed_cluster": 0.5157577782392276,
29
+ "eval_iou_background": 0.01485215844086368,
30
+ "eval_iou_double_plant": 0.025830652956995166,
31
+ "eval_iou_drydown": 0.09098524126821965,
32
+ "eval_iou_endrow": 0.011190399377301194,
33
+ "eval_iou_nutrient_deficiency": 0.16172838363740288,
34
+ "eval_iou_planter_skip": 0.007145844854739052,
35
+ "eval_iou_water": 0.052629358314533047,
36
+ "eval_iou_waterway": 0.008009602169192399,
37
+ "eval_iou_weed_cluster": 0.055193229922437775,
38
+ "eval_loss": 3.9555599689483643,
39
+ "eval_mean_accuracy": 0.22838338070714315,
40
+ "eval_mean_iou": 0.047507207882409426,
41
+ "eval_overall_accuracy": 0.09656532361323403,
42
+ "eval_runtime": 348.036,
43
+ "eval_samples_per_second": 8.166,
44
+ "eval_steps_per_second": 2.724,
45
+ "step": 500
46
+ },
47
+ {
48
+ "epoch": 0.16666666666666666,
49
+ "grad_norm": 12.677851676940918,
50
+ "learning_rate": 4.9999999999999996e-06,
51
+ "loss": 3.7698,
52
+ "step": 1000
53
+ },
54
+ {
55
+ "epoch": 0.16666666666666666,
56
+ "eval_accuracy_background": 0.29766305371935275,
57
+ "eval_accuracy_double_plant": 0.04300475016987658,
58
+ "eval_accuracy_drydown": 0.7567223885695242,
59
+ "eval_accuracy_endrow": 0.006538411852764754,
60
+ "eval_accuracy_nutrient_deficiency": 0.5373960102281351,
61
+ "eval_accuracy_planter_skip": 0.1493345947032268,
62
+ "eval_accuracy_water": 0.2900425128140623,
63
+ "eval_accuracy_waterway": 0.37062750835043584,
64
+ "eval_accuracy_weed_cluster": 0.242435460451953,
65
+ "eval_iou_background": 0.2681855463534359,
66
+ "eval_iou_double_plant": 0.02937987848319999,
67
+ "eval_iou_drydown": 0.30192234804386486,
68
+ "eval_iou_endrow": 0.006310524335748097,
69
+ "eval_iou_nutrient_deficiency": 0.19394253102070813,
70
+ "eval_iou_planter_skip": 0.08231306246610917,
71
+ "eval_iou_water": 0.10160883330705332,
72
+ "eval_iou_waterway": 0.050415164283578985,
73
+ "eval_iou_weed_cluster": 0.07122746911823011,
74
+ "eval_loss": 3.33693528175354,
75
+ "eval_mean_accuracy": 0.29930718787325905,
76
+ "eval_mean_iou": 0.12281170637910319,
77
+ "eval_overall_accuracy": 0.37769782397579943,
78
+ "eval_runtime": 260.2627,
79
+ "eval_samples_per_second": 10.92,
80
+ "eval_steps_per_second": 3.642,
81
+ "step": 1000
82
+ },
83
+ {
84
+ "epoch": 0.25,
85
+ "grad_norm": 12.100262641906738,
86
+ "learning_rate": 7.5e-06,
87
+ "loss": 3.2674,
88
+ "step": 1500
89
+ },
90
+ {
91
+ "epoch": 0.25,
92
+ "eval_accuracy_background": 0.48419612667845113,
93
+ "eval_accuracy_double_plant": 0.013344873041957285,
94
+ "eval_accuracy_drydown": 0.8872887901430043,
95
+ "eval_accuracy_endrow": 0.015151865793565866,
96
+ "eval_accuracy_nutrient_deficiency": 0.37784363150955347,
97
+ "eval_accuracy_planter_skip": 0.1236399017700026,
98
+ "eval_accuracy_water": 0.6308252089425159,
99
+ "eval_accuracy_waterway": 0.4153131997350911,
100
+ "eval_accuracy_weed_cluster": 0.2533755890828509,
101
+ "eval_iou_background": 0.4220656310747218,
102
+ "eval_iou_double_plant": 0.012970922185309422,
103
+ "eval_iou_drydown": 0.3650381802310761,
104
+ "eval_iou_endrow": 0.014768985148023165,
105
+ "eval_iou_nutrient_deficiency": 0.20002194862446826,
106
+ "eval_iou_planter_skip": 0.09927397940127604,
107
+ "eval_iou_water": 0.27391064172674834,
108
+ "eval_iou_waterway": 0.10636423788784055,
109
+ "eval_iou_weed_cluster": 0.10586226367786514,
110
+ "eval_loss": 2.763476610183716,
111
+ "eval_mean_accuracy": 0.3556643540774437,
112
+ "eval_mean_iou": 0.17780853221748097,
113
+ "eval_overall_accuracy": 0.5152837603493446,
114
+ "eval_runtime": 258.1136,
115
+ "eval_samples_per_second": 11.011,
116
+ "eval_steps_per_second": 3.673,
117
+ "step": 1500
118
+ },
119
+ {
120
+ "epoch": 0.3333333333333333,
121
+ "grad_norm": 13.388419151306152,
122
+ "learning_rate": 9.999999999999999e-06,
123
+ "loss": 2.8338,
124
+ "step": 2000
125
+ },
126
+ {
127
+ "epoch": 0.3333333333333333,
128
+ "eval_accuracy_background": 0.6007479203520831,
129
+ "eval_accuracy_double_plant": 0.09356295249957611,
130
+ "eval_accuracy_drydown": 0.8340823867683501,
131
+ "eval_accuracy_endrow": 0.0035413015410058713,
132
+ "eval_accuracy_nutrient_deficiency": 0.44282241132222916,
133
+ "eval_accuracy_planter_skip": 0.11649665949385082,
134
+ "eval_accuracy_water": 0.921699186070695,
135
+ "eval_accuracy_waterway": 0.20547725725890914,
136
+ "eval_accuracy_weed_cluster": 0.3415216522793772,
137
+ "eval_iou_background": 0.5156964205169746,
138
+ "eval_iou_double_plant": 0.07643992513295463,
139
+ "eval_iou_drydown": 0.4276993634847662,
140
+ "eval_iou_endrow": 0.0035320673127108605,
141
+ "eval_iou_nutrient_deficiency": 0.23942297475711152,
142
+ "eval_iou_planter_skip": 0.09915967157890872,
143
+ "eval_iou_water": 0.37311176318003697,
144
+ "eval_iou_waterway": 0.1337041070891874,
145
+ "eval_iou_weed_cluster": 0.17456818424609255,
146
+ "eval_loss": 2.2476296424865723,
147
+ "eval_mean_accuracy": 0.39555019195400853,
148
+ "eval_mean_iou": 0.22703716414430483,
149
+ "eval_overall_accuracy": 0.6012494768590696,
150
+ "eval_runtime": 261.9728,
151
+ "eval_samples_per_second": 10.848,
152
+ "eval_steps_per_second": 3.619,
153
+ "step": 2000
154
+ },
155
+ {
156
+ "epoch": 0.4166666666666667,
157
+ "grad_norm": 31.341535568237305,
158
+ "learning_rate": 1.25e-05,
159
+ "loss": 2.5604,
160
+ "step": 2500
161
+ },
162
+ {
163
+ "epoch": 0.4166666666666667,
164
+ "eval_accuracy_background": 0.7311239150003357,
165
+ "eval_accuracy_double_plant": 0.18906004388106473,
166
+ "eval_accuracy_drydown": 0.7049642054247133,
167
+ "eval_accuracy_endrow": 0.03270138529568011,
168
+ "eval_accuracy_nutrient_deficiency": 0.2058331492617626,
169
+ "eval_accuracy_planter_skip": 0.07169662420163618,
170
+ "eval_accuracy_water": 0.9476640149608306,
171
+ "eval_accuracy_waterway": 0.4427819971981128,
172
+ "eval_accuracy_weed_cluster": 0.38249564148248877,
173
+ "eval_iou_background": 0.5934145085795158,
174
+ "eval_iou_double_plant": 0.10950501729968068,
175
+ "eval_iou_drydown": 0.42479480844402817,
176
+ "eval_iou_endrow": 0.030470447854384766,
177
+ "eval_iou_nutrient_deficiency": 0.16681197423835636,
178
+ "eval_iou_planter_skip": 0.06706541524720425,
179
+ "eval_iou_water": 0.36387793500767635,
180
+ "eval_iou_waterway": 0.19571015933545038,
181
+ "eval_iou_weed_cluster": 0.22406855402046205,
182
+ "eval_loss": 2.0893733501434326,
183
+ "eval_mean_accuracy": 0.41203566407851394,
184
+ "eval_mean_iou": 0.24174653555852876,
185
+ "eval_overall_accuracy": 0.6543493727987272,
186
+ "eval_runtime": 259.8765,
187
+ "eval_samples_per_second": 10.936,
188
+ "eval_steps_per_second": 3.648,
189
+ "step": 2500
190
+ },
191
+ {
192
+ "epoch": 0.5,
193
+ "grad_norm": 22.26491928100586,
194
+ "learning_rate": 1.5e-05,
195
+ "loss": 2.4032,
196
+ "step": 3000
197
+ },
198
+ {
199
+ "epoch": 0.5,
200
+ "eval_accuracy_background": 0.6732741082932401,
201
+ "eval_accuracy_double_plant": 0.10359602903136565,
202
+ "eval_accuracy_drydown": 0.8666517772278536,
203
+ "eval_accuracy_endrow": 0.035669653990401784,
204
+ "eval_accuracy_nutrient_deficiency": 0.2536647750774501,
205
+ "eval_accuracy_planter_skip": 0.24370499629921916,
206
+ "eval_accuracy_water": 0.9767091931497752,
207
+ "eval_accuracy_waterway": 0.6842904410593037,
208
+ "eval_accuracy_weed_cluster": 0.3698701654570609,
209
+ "eval_iou_background": 0.5681057490790173,
210
+ "eval_iou_double_plant": 0.0852643551765096,
211
+ "eval_iou_drydown": 0.4543271536581631,
212
+ "eval_iou_endrow": 0.03370061965697607,
213
+ "eval_iou_nutrient_deficiency": 0.19633806402515588,
214
+ "eval_iou_planter_skip": 0.18150485646008124,
215
+ "eval_iou_water": 0.3528168740365702,
216
+ "eval_iou_waterway": 0.16230336403850296,
217
+ "eval_iou_weed_cluster": 0.22603294672854196,
218
+ "eval_loss": 1.9570292234420776,
219
+ "eval_mean_accuracy": 0.4674923488428522,
220
+ "eval_mean_iou": 0.2511548869843909,
221
+ "eval_overall_accuracy": 0.6432186597875551,
222
+ "eval_runtime": 259.2993,
223
+ "eval_samples_per_second": 10.96,
224
+ "eval_steps_per_second": 3.656,
225
+ "step": 3000
226
+ },
227
+ {
228
+ "epoch": 0.5833333333333334,
229
+ "grad_norm": 31.124582290649414,
230
+ "learning_rate": 1.7500000000000002e-05,
231
+ "loss": 2.3362,
232
+ "step": 3500
233
+ },
234
+ {
235
+ "epoch": 0.5833333333333334,
236
+ "eval_accuracy_background": 0.5814538721345998,
237
+ "eval_accuracy_double_plant": 0.06263123204822073,
238
+ "eval_accuracy_drydown": 0.9572617680178966,
239
+ "eval_accuracy_endrow": 0.1022870538789039,
240
+ "eval_accuracy_nutrient_deficiency": 0.155304235691705,
241
+ "eval_accuracy_planter_skip": 0.28837366612584614,
242
+ "eval_accuracy_water": 0.957647648655113,
243
+ "eval_accuracy_waterway": 0.4168520985122232,
244
+ "eval_accuracy_weed_cluster": 0.20255586893627922,
245
+ "eval_iou_background": 0.5035989534908836,
246
+ "eval_iou_double_plant": 0.05887727499276805,
247
+ "eval_iou_drydown": 0.35033214248192274,
248
+ "eval_iou_endrow": 0.08288492557854046,
249
+ "eval_iou_nutrient_deficiency": 0.1308384862727777,
250
+ "eval_iou_planter_skip": 0.23854848545793084,
251
+ "eval_iou_water": 0.33185434658316154,
252
+ "eval_iou_waterway": 0.19380165775707148,
253
+ "eval_iou_weed_cluster": 0.16104566462906844,
254
+ "eval_loss": 2.403474807739258,
255
+ "eval_mean_accuracy": 0.4138186048889764,
256
+ "eval_mean_iou": 0.22797577080490272,
257
+ "eval_overall_accuracy": 0.5765593656585961,
258
+ "eval_runtime": 258.5568,
259
+ "eval_samples_per_second": 10.992,
260
+ "eval_steps_per_second": 3.667,
261
+ "step": 3500
262
+ },
263
+ {
264
+ "epoch": 0.6666666666666666,
265
+ "grad_norm": 27.981061935424805,
266
+ "learning_rate": 1.9999999999999998e-05,
267
+ "loss": 2.2454,
268
+ "step": 4000
269
+ },
270
+ {
271
+ "epoch": 0.6666666666666666,
272
+ "eval_accuracy_background": 0.6165356472563173,
273
+ "eval_accuracy_double_plant": 0.2464305201601748,
274
+ "eval_accuracy_drydown": 0.9388557439528662,
275
+ "eval_accuracy_endrow": 0.08952317035344279,
276
+ "eval_accuracy_nutrient_deficiency": 0.30307985133610366,
277
+ "eval_accuracy_planter_skip": 0.34189676046879824,
278
+ "eval_accuracy_water": 0.9684396578613939,
279
+ "eval_accuracy_waterway": 0.4364168499156771,
280
+ "eval_accuracy_weed_cluster": 0.48568842831047665,
281
+ "eval_iou_background": 0.5406762574665314,
282
+ "eval_iou_double_plant": 0.12820291315879165,
283
+ "eval_iou_drydown": 0.445376041256294,
284
+ "eval_iou_endrow": 0.07582305802889684,
285
+ "eval_iou_nutrient_deficiency": 0.2473209944226292,
286
+ "eval_iou_planter_skip": 0.2571484714289717,
287
+ "eval_iou_water": 0.3523385474174709,
288
+ "eval_iou_waterway": 0.2182523975088832,
289
+ "eval_iou_weed_cluster": 0.22609664246227254,
290
+ "eval_loss": 1.9709433317184448,
291
+ "eval_mean_accuracy": 0.49187406995725,
292
+ "eval_mean_iou": 0.27680392479452676,
293
+ "eval_overall_accuracy": 0.6261566315021039,
294
+ "eval_runtime": 262.6128,
295
+ "eval_samples_per_second": 10.822,
296
+ "eval_steps_per_second": 3.61,
297
+ "step": 4000
298
+ },
299
+ {
300
+ "epoch": 0.75,
301
+ "grad_norm": 19.56971549987793,
302
+ "learning_rate": 2.25e-05,
303
+ "loss": 2.2631,
304
+ "step": 4500
305
+ },
306
+ {
307
+ "epoch": 0.75,
308
+ "eval_accuracy_background": 0.7136164920788886,
309
+ "eval_accuracy_double_plant": 0.2274296812949631,
310
+ "eval_accuracy_drydown": 0.8423346225534782,
311
+ "eval_accuracy_endrow": 0.3372638527840968,
312
+ "eval_accuracy_nutrient_deficiency": 0.39363536828312395,
313
+ "eval_accuracy_planter_skip": 0.24472503933611423,
314
+ "eval_accuracy_water": 0.8132625823496894,
315
+ "eval_accuracy_waterway": 0.5279086387447126,
316
+ "eval_accuracy_weed_cluster": 0.2613040701496915,
317
+ "eval_iou_background": 0.6023960172505115,
318
+ "eval_iou_double_plant": 0.14551908714826825,
319
+ "eval_iou_drydown": 0.48571101575483516,
320
+ "eval_iou_endrow": 0.15976738972856136,
321
+ "eval_iou_nutrient_deficiency": 0.26876578893765557,
322
+ "eval_iou_planter_skip": 0.21772117232934346,
323
+ "eval_iou_water": 0.3885318463525075,
324
+ "eval_iou_waterway": 0.25285421749201065,
325
+ "eval_iou_weed_cluster": 0.19549177279724894,
326
+ "eval_loss": 1.8094414472579956,
327
+ "eval_mean_accuracy": 0.48460892750830653,
328
+ "eval_mean_iou": 0.3018620341989936,
329
+ "eval_overall_accuracy": 0.6746520910480136,
330
+ "eval_runtime": 263.1854,
331
+ "eval_samples_per_second": 10.798,
332
+ "eval_steps_per_second": 3.602,
333
+ "step": 4500
334
+ },
335
+ {
336
+ "epoch": 0.8333333333333334,
337
+ "grad_norm": 15.023950576782227,
338
+ "learning_rate": 2.5e-05,
339
+ "loss": 2.2139,
340
+ "step": 5000
341
+ },
342
+ {
343
+ "epoch": 0.8333333333333334,
344
+ "eval_accuracy_background": 0.6023184738684322,
345
+ "eval_accuracy_double_plant": 0.19349261148429422,
346
+ "eval_accuracy_drydown": 0.4844650786080634,
347
+ "eval_accuracy_endrow": 0.16074521211886753,
348
+ "eval_accuracy_nutrient_deficiency": 0.6620136299007383,
349
+ "eval_accuracy_planter_skip": 0.20313071353995618,
350
+ "eval_accuracy_water": 0.9327820785252834,
351
+ "eval_accuracy_waterway": 0.42715639009708817,
352
+ "eval_accuracy_weed_cluster": 0.5896869029400611,
353
+ "eval_iou_background": 0.5071775649990886,
354
+ "eval_iou_double_plant": 0.14848297634630775,
355
+ "eval_iou_drydown": 0.34958365519874773,
356
+ "eval_iou_endrow": 0.1266887334828955,
357
+ "eval_iou_nutrient_deficiency": 0.3268729234134353,
358
+ "eval_iou_planter_skip": 0.1708553631916936,
359
+ "eval_iou_water": 0.3656991869753895,
360
+ "eval_iou_waterway": 0.15825121509439088,
361
+ "eval_iou_weed_cluster": 0.18557336284286047,
362
+ "eval_loss": 2.1465232372283936,
363
+ "eval_mean_accuracy": 0.47286567678697605,
364
+ "eval_mean_iou": 0.2599094423938677,
365
+ "eval_overall_accuracy": 0.585979408046041,
366
+ "eval_runtime": 258.2944,
367
+ "eval_samples_per_second": 11.003,
368
+ "eval_steps_per_second": 3.67,
369
+ "step": 5000
370
+ },
371
+ {
372
+ "epoch": 0.9166666666666666,
373
+ "grad_norm": 15.944084167480469,
374
+ "learning_rate": 2.75e-05,
375
+ "loss": 2.1107,
376
+ "step": 5500
377
+ },
378
+ {
379
+ "epoch": 0.9166666666666666,
380
+ "eval_accuracy_background": 0.6320754005058506,
381
+ "eval_accuracy_double_plant": 0.18459879167713122,
382
+ "eval_accuracy_drydown": 0.7435807472335404,
383
+ "eval_accuracy_endrow": 0.4730924983472199,
384
+ "eval_accuracy_nutrient_deficiency": 0.7501714321465206,
385
+ "eval_accuracy_planter_skip": 0.394123356093544,
386
+ "eval_accuracy_water": 0.8982667262993669,
387
+ "eval_accuracy_waterway": 0.2070484619961853,
388
+ "eval_accuracy_weed_cluster": 0.06027005934327404,
389
+ "eval_iou_background": 0.5431209987641049,
390
+ "eval_iou_double_plant": 0.1123139870852644,
391
+ "eval_iou_drydown": 0.4467521390448892,
392
+ "eval_iou_endrow": 0.1507248222065432,
393
+ "eval_iou_nutrient_deficiency": 0.3257302284667843,
394
+ "eval_iou_planter_skip": 0.2504630445874714,
395
+ "eval_iou_water": 0.38357052112337536,
396
+ "eval_iou_waterway": 0.1656467966651648,
397
+ "eval_iou_weed_cluster": 0.0586611852117363,
398
+ "eval_loss": 2.075209856033325,
399
+ "eval_mean_accuracy": 0.48258083040473704,
400
+ "eval_mean_iou": 0.27077596923948155,
401
+ "eval_overall_accuracy": 0.6281377263734175,
402
+ "eval_runtime": 259.3294,
403
+ "eval_samples_per_second": 10.959,
404
+ "eval_steps_per_second": 3.656,
405
+ "step": 5500
406
+ },
407
+ {
408
+ "epoch": 1.0,
409
+ "grad_norm": 4.946102142333984,
410
+ "learning_rate": 3e-05,
411
+ "loss": 2.1338,
412
+ "step": 6000
413
+ },
414
+ {
415
+ "epoch": 1.0,
416
+ "eval_accuracy_background": 0.6964471310778703,
417
+ "eval_accuracy_double_plant": 0.2429333892576808,
418
+ "eval_accuracy_drydown": 0.8654738033211103,
419
+ "eval_accuracy_endrow": 0.05382795612758288,
420
+ "eval_accuracy_nutrient_deficiency": 0.30272693922024607,
421
+ "eval_accuracy_planter_skip": 0.39966472396096286,
422
+ "eval_accuracy_water": 0.9846543263110985,
423
+ "eval_accuracy_waterway": 0.7568692730591431,
424
+ "eval_accuracy_weed_cluster": 0.10615040627525124,
425
+ "eval_iou_background": 0.5855221723102959,
426
+ "eval_iou_double_plant": 0.1361405520488392,
427
+ "eval_iou_drydown": 0.4768776656490699,
428
+ "eval_iou_endrow": 0.05007269671892798,
429
+ "eval_iou_nutrient_deficiency": 0.22420499455655446,
430
+ "eval_iou_planter_skip": 0.2752533913799669,
431
+ "eval_iou_water": 0.29633141374004973,
432
+ "eval_iou_waterway": 0.1110136523015455,
433
+ "eval_iou_weed_cluster": 0.10019203575135141,
434
+ "eval_loss": 2.053126335144043,
435
+ "eval_mean_accuracy": 0.489860883178994,
436
+ "eval_mean_iou": 0.25062317493962233,
437
+ "eval_overall_accuracy": 0.6527557080115521,
438
+ "eval_runtime": 261.8254,
439
+ "eval_samples_per_second": 10.855,
440
+ "eval_steps_per_second": 3.621,
441
+ "step": 6000
442
+ },
443
+ {
444
+ "epoch": 1.0833333333333333,
445
+ "grad_norm": 16.1883602142334,
446
+ "learning_rate": 3.25e-05,
447
+ "loss": 2.0743,
448
+ "step": 6500
449
+ },
450
+ {
451
+ "epoch": 1.0833333333333333,
452
+ "eval_accuracy_background": 0.6163441350441362,
453
+ "eval_accuracy_double_plant": 0.3179624244479489,
454
+ "eval_accuracy_drydown": 0.637096555419927,
455
+ "eval_accuracy_endrow": 0.45733081369282536,
456
+ "eval_accuracy_nutrient_deficiency": 0.8510315773068933,
457
+ "eval_accuracy_planter_skip": 0.5177706104082623,
458
+ "eval_accuracy_water": 0.9512477714625793,
459
+ "eval_accuracy_waterway": 0.5107166291437305,
460
+ "eval_accuracy_weed_cluster": 0.24186705045667198,
461
+ "eval_iou_background": 0.5350583256945234,
462
+ "eval_iou_double_plant": 0.17959294330409034,
463
+ "eval_iou_drydown": 0.4634150029792384,
464
+ "eval_iou_endrow": 0.16707740028365953,
465
+ "eval_iou_nutrient_deficiency": 0.31532389062386124,
466
+ "eval_iou_planter_skip": 0.2948590617515534,
467
+ "eval_iou_water": 0.34520224120688753,
468
+ "eval_iou_waterway": 0.11975132688004231,
469
+ "eval_iou_weed_cluster": 0.2113183362317597,
470
+ "eval_loss": 1.8855762481689453,
471
+ "eval_mean_accuracy": 0.5668186185981083,
472
+ "eval_mean_iou": 0.29239983655062396,
473
+ "eval_overall_accuracy": 0.6228305054183748,
474
+ "eval_runtime": 259.3394,
475
+ "eval_samples_per_second": 10.959,
476
+ "eval_steps_per_second": 3.655,
477
+ "step": 6500
478
+ },
479
+ {
480
+ "epoch": 1.1666666666666667,
481
+ "grad_norm": 9.279212951660156,
482
+ "learning_rate": 3.5000000000000004e-05,
483
+ "loss": 2.0725,
484
+ "step": 7000
485
+ },
486
+ {
487
+ "epoch": 1.1666666666666667,
488
+ "eval_accuracy_background": 0.7373301969750995,
489
+ "eval_accuracy_double_plant": 0.33336127425920425,
490
+ "eval_accuracy_drydown": 0.6301703388238222,
491
+ "eval_accuracy_endrow": 0.37253283281326316,
492
+ "eval_accuracy_nutrient_deficiency": 0.34279961797329944,
493
+ "eval_accuracy_planter_skip": 0.5799829420962596,
494
+ "eval_accuracy_water": 0.9469093944591689,
495
+ "eval_accuracy_waterway": 0.3649137190752288,
496
+ "eval_accuracy_weed_cluster": 0.4194977852900649,
497
+ "eval_iou_background": 0.6045108089230737,
498
+ "eval_iou_double_plant": 0.15103332950916856,
499
+ "eval_iou_drydown": 0.4284455196537357,
500
+ "eval_iou_endrow": 0.18315189002409357,
501
+ "eval_iou_nutrient_deficiency": 0.2540752037931447,
502
+ "eval_iou_planter_skip": 0.2268832410220617,
503
+ "eval_iou_water": 0.3615440261043886,
504
+ "eval_iou_waterway": 0.2757438894003987,
505
+ "eval_iou_weed_cluster": 0.24843146863053017,
506
+ "eval_loss": 1.8260598182678223,
507
+ "eval_mean_accuracy": 0.5252775668628233,
508
+ "eval_mean_iou": 0.3037577085622884,
509
+ "eval_overall_accuracy": 0.6668996089543237,
510
+ "eval_runtime": 257.8864,
511
+ "eval_samples_per_second": 11.02,
512
+ "eval_steps_per_second": 3.676,
513
+ "step": 7000
514
+ },
515
+ {
516
+ "epoch": 1.25,
517
+ "grad_norm": 12.69166088104248,
518
+ "learning_rate": 3.7500000000000003e-05,
519
+ "loss": 2.1051,
520
+ "step": 7500
521
+ },
522
+ {
523
+ "epoch": 1.25,
524
+ "eval_accuracy_background": 0.6136021677052333,
525
+ "eval_accuracy_double_plant": 0.21675932855086671,
526
+ "eval_accuracy_drydown": 0.8978391275698134,
527
+ "eval_accuracy_endrow": 0.31490227354840317,
528
+ "eval_accuracy_nutrient_deficiency": 0.6391839292944432,
529
+ "eval_accuracy_planter_skip": 0.4095695820323414,
530
+ "eval_accuracy_water": 0.9815472332260521,
531
+ "eval_accuracy_waterway": 0.3863849586942106,
532
+ "eval_accuracy_weed_cluster": 0.41607248209352754,
533
+ "eval_iou_background": 0.5478072558942303,
534
+ "eval_iou_double_plant": 0.15633489787457702,
535
+ "eval_iou_drydown": 0.4606893739380495,
536
+ "eval_iou_endrow": 0.18368040175633407,
537
+ "eval_iou_nutrient_deficiency": 0.36167812896859036,
538
+ "eval_iou_planter_skip": 0.3133978757416113,
539
+ "eval_iou_water": 0.35102829123972634,
540
+ "eval_iou_waterway": 0.2651990401867848,
541
+ "eval_iou_weed_cluster": 0.27107161264823754,
542
+ "eval_loss": 1.7672898769378662,
543
+ "eval_mean_accuracy": 0.5417623425238768,
544
+ "eval_mean_iou": 0.32343187536090456,
545
+ "eval_overall_accuracy": 0.6462444076512336,
546
+ "eval_runtime": 263.0897,
547
+ "eval_samples_per_second": 10.802,
548
+ "eval_steps_per_second": 3.603,
549
+ "step": 7500
550
+ },
551
+ {
552
+ "epoch": 1.3333333333333333,
553
+ "grad_norm": 26.389280319213867,
554
+ "learning_rate": 3.9999999999999996e-05,
555
+ "loss": 2.0567,
556
+ "step": 8000
557
+ },
558
+ {
559
+ "epoch": 1.3333333333333333,
560
+ "eval_accuracy_background": 0.6516741544849038,
561
+ "eval_accuracy_double_plant": 0.22472217370629263,
562
+ "eval_accuracy_drydown": 0.9205657832973893,
563
+ "eval_accuracy_endrow": 0.1334717500678728,
564
+ "eval_accuracy_nutrient_deficiency": 0.579114420966078,
565
+ "eval_accuracy_planter_skip": 0.6448299356407252,
566
+ "eval_accuracy_water": 0.8737892085078403,
567
+ "eval_accuracy_waterway": 0.4514893265909921,
568
+ "eval_accuracy_weed_cluster": 0.36867061298945264,
569
+ "eval_iou_background": 0.5743802241628614,
570
+ "eval_iou_double_plant": 0.17379933416714446,
571
+ "eval_iou_drydown": 0.4870945365813324,
572
+ "eval_iou_endrow": 0.1124492323800547,
573
+ "eval_iou_nutrient_deficiency": 0.35858088912475333,
574
+ "eval_iou_planter_skip": 0.22081427365766507,
575
+ "eval_iou_water": 0.4188886664719046,
576
+ "eval_iou_waterway": 0.23882510300705312,
577
+ "eval_iou_weed_cluster": 0.22928729873239137,
578
+ "eval_loss": 1.7339626550674438,
579
+ "eval_mean_accuracy": 0.5387030406946163,
580
+ "eval_mean_iou": 0.3126799509205734,
581
+ "eval_overall_accuracy": 0.6655563976415058,
582
+ "eval_runtime": 262.6835,
583
+ "eval_samples_per_second": 10.819,
584
+ "eval_steps_per_second": 3.609,
585
+ "step": 8000
586
+ },
587
+ {
588
+ "epoch": 1.4166666666666667,
589
+ "grad_norm": 16.169328689575195,
590
+ "learning_rate": 4.25e-05,
591
+ "loss": 2.0731,
592
+ "step": 8500
593
+ },
594
+ {
595
+ "epoch": 1.4166666666666667,
596
+ "eval_accuracy_background": 0.7486936916842247,
597
+ "eval_accuracy_double_plant": 0.2930715215459597,
598
+ "eval_accuracy_drydown": 0.7255160536916373,
599
+ "eval_accuracy_endrow": 0.4577216435093236,
600
+ "eval_accuracy_nutrient_deficiency": 0.4830792419842683,
601
+ "eval_accuracy_planter_skip": 0.7417698065300401,
602
+ "eval_accuracy_water": 0.9756065468432125,
603
+ "eval_accuracy_waterway": 0.273890366668282,
604
+ "eval_accuracy_weed_cluster": 0.35391598936308855,
605
+ "eval_iou_background": 0.6323410133400419,
606
+ "eval_iou_double_plant": 0.17111536697625435,
607
+ "eval_iou_drydown": 0.5110135373059105,
608
+ "eval_iou_endrow": 0.16559443160045,
609
+ "eval_iou_nutrient_deficiency": 0.3404264291064937,
610
+ "eval_iou_planter_skip": 0.21812035844976185,
611
+ "eval_iou_water": 0.35579748623298973,
612
+ "eval_iou_waterway": 0.20254597755885112,
613
+ "eval_iou_weed_cluster": 0.24793962096921052,
614
+ "eval_loss": 1.6961616277694702,
615
+ "eval_mean_accuracy": 0.5614738735355596,
616
+ "eval_mean_iou": 0.31609935794888483,
617
+ "eval_overall_accuracy": 0.6986505820712626,
618
+ "eval_runtime": 262.6198,
619
+ "eval_samples_per_second": 10.822,
620
+ "eval_steps_per_second": 3.61,
621
+ "step": 8500
622
+ },
623
+ {
624
+ "epoch": 1.5,
625
+ "grad_norm": 9.605523109436035,
626
+ "learning_rate": 4.5e-05,
627
+ "loss": 2.0062,
628
+ "step": 9000
629
+ },
630
+ {
631
+ "epoch": 1.5,
632
+ "eval_accuracy_background": 0.6385280729636846,
633
+ "eval_accuracy_double_plant": 0.38205242780086,
634
+ "eval_accuracy_drydown": 0.9496890843481527,
635
+ "eval_accuracy_endrow": 0.34276586617075205,
636
+ "eval_accuracy_nutrient_deficiency": 0.4966985380725304,
637
+ "eval_accuracy_planter_skip": 0.7026444652494228,
638
+ "eval_accuracy_water": 0.7442517574456261,
639
+ "eval_accuracy_waterway": 0.46809502667206265,
640
+ "eval_accuracy_weed_cluster": 0.17348492439927163,
641
+ "eval_iou_background": 0.5609579182369832,
642
+ "eval_iou_double_plant": 0.16239615673750246,
643
+ "eval_iou_drydown": 0.45202834914092793,
644
+ "eval_iou_endrow": 0.18367824595658963,
645
+ "eval_iou_nutrient_deficiency": 0.34617965104488957,
646
+ "eval_iou_planter_skip": 0.2580270293767756,
647
+ "eval_iou_water": 0.3425501321917906,
648
+ "eval_iou_waterway": 0.19866797232183966,
649
+ "eval_iou_weed_cluster": 0.13472834814631707,
650
+ "eval_loss": 1.9120891094207764,
651
+ "eval_mean_accuracy": 0.5442455736802626,
652
+ "eval_mean_iou": 0.2932459781281795,
653
+ "eval_overall_accuracy": 0.6456323905823719,
654
+ "eval_runtime": 258.4421,
655
+ "eval_samples_per_second": 10.997,
656
+ "eval_steps_per_second": 3.668,
657
+ "step": 9000
658
+ },
659
+ {
660
+ "epoch": 1.5833333333333335,
661
+ "grad_norm": 9.02011489868164,
662
+ "learning_rate": 4.7499999999999996e-05,
663
+ "loss": 2.0069,
664
+ "step": 9500
665
+ },
666
+ {
667
+ "epoch": 1.5833333333333335,
668
+ "eval_accuracy_background": 0.5921818727290818,
669
+ "eval_accuracy_double_plant": 0.3459265635975844,
670
+ "eval_accuracy_drydown": 0.9022492563261855,
671
+ "eval_accuracy_endrow": 0.39012380134585,
672
+ "eval_accuracy_nutrient_deficiency": 0.6176093671439052,
673
+ "eval_accuracy_planter_skip": 0.7456583223453638,
674
+ "eval_accuracy_water": 0.930193012579653,
675
+ "eval_accuracy_waterway": 0.6027143991593465,
676
+ "eval_accuracy_weed_cluster": 0.38137543613044333,
677
+ "eval_iou_background": 0.5323074634115271,
678
+ "eval_iou_double_plant": 0.132029668948867,
679
+ "eval_iou_drydown": 0.4864590469093194,
680
+ "eval_iou_endrow": 0.2008338441059683,
681
+ "eval_iou_nutrient_deficiency": 0.36281261663843545,
682
+ "eval_iou_planter_skip": 0.18127130680379444,
683
+ "eval_iou_water": 0.3502148412760265,
684
+ "eval_iou_waterway": 0.14129351940530874,
685
+ "eval_iou_weed_cluster": 0.21689374484941243,
686
+ "eval_loss": 1.9075040817260742,
687
+ "eval_mean_accuracy": 0.6120035590397126,
688
+ "eval_mean_iou": 0.28934622803873994,
689
+ "eval_overall_accuracy": 0.6314946434667654,
690
+ "eval_runtime": 258.4278,
691
+ "eval_samples_per_second": 10.997,
692
+ "eval_steps_per_second": 3.668,
693
+ "step": 9500
694
+ },
695
+ {
696
+ "epoch": 1.6666666666666665,
697
+ "grad_norm": 27.558940887451172,
698
+ "learning_rate": 5e-05,
699
+ "loss": 1.9695,
700
+ "step": 10000
701
+ },
702
+ {
703
+ "epoch": 1.6666666666666665,
704
+ "eval_accuracy_background": 0.6535160468633152,
705
+ "eval_accuracy_double_plant": 0.27480931114887563,
706
+ "eval_accuracy_drydown": 0.8507298791874478,
707
+ "eval_accuracy_endrow": 0.3002457236689853,
708
+ "eval_accuracy_nutrient_deficiency": 0.5805433393846879,
709
+ "eval_accuracy_planter_skip": 0.5759424736901442,
710
+ "eval_accuracy_water": 0.9136638372037981,
711
+ "eval_accuracy_waterway": 0.300740548651248,
712
+ "eval_accuracy_weed_cluster": 0.2748341582674426,
713
+ "eval_iou_background": 0.5715406090850156,
714
+ "eval_iou_double_plant": 0.1830877307497648,
715
+ "eval_iou_drydown": 0.44761615381963227,
716
+ "eval_iou_endrow": 0.1607454389243595,
717
+ "eval_iou_nutrient_deficiency": 0.3589882634051457,
718
+ "eval_iou_planter_skip": 0.3142600611891689,
719
+ "eval_iou_water": 0.381127146789507,
720
+ "eval_iou_waterway": 0.11509687922710479,
721
+ "eval_iou_weed_cluster": 0.191443588646255,
722
+ "eval_loss": 1.9981021881103516,
723
+ "eval_mean_accuracy": 0.5250028131184383,
724
+ "eval_mean_iou": 0.3026562079817726,
725
+ "eval_overall_accuracy": 0.6541074851018032,
726
+ "eval_runtime": 262.1608,
727
+ "eval_samples_per_second": 10.841,
728
+ "eval_steps_per_second": 3.616,
729
+ "step": 10000
730
+ }
731
+ ],
732
+ "logging_steps": 500,
733
+ "max_steps": 120000,
734
+ "num_input_tokens_seen": 0,
735
+ "num_train_epochs": 20,
736
+ "save_steps": 500,
737
+ "total_flos": 1.596835724525568e+19,
738
+ "train_batch_size": 3,
739
+ "trial_name": null,
740
+ "trial_params": null
741
+ }
checkpoint-10000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a680a40140b6ca5fa47ac87d8120967d6766117ead261eaa84c073e1e9509b8
3
+ size 4539