heroza commited on
Commit
9da20ef
1 Parent(s): da2ec7a

End of training

Browse files
README.md CHANGED
@@ -2,6 +2,8 @@
2
  license: other
3
  base_model: nvidia/mit-b0
4
  tags:
 
 
5
  - generated_from_trainer
6
  model-index:
7
  - name: segformer-finetuned-biofilm_MRCNNv1_halfjoin
@@ -13,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
13
 
14
  # segformer-finetuned-biofilm_MRCNNv1_halfjoin
15
 
16
- This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
  - Loss: 0.0208
19
  - Mean Iou: 0.4961
 
2
  license: other
3
  base_model: nvidia/mit-b0
4
  tags:
5
+ - image-segmentation
6
+ - vision
7
  - generated_from_trainer
8
  model-index:
9
  - name: segformer-finetuned-biofilm_MRCNNv1_halfjoin
 
15
 
16
  # segformer-finetuned-biofilm_MRCNNv1_halfjoin
17
 
18
+ This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the heroza/biofilm_MRCNNv1_halfjoin dataset.
19
  It achieves the following results on the evaluation set:
20
  - Loss: 0.0208
21
  - Mean Iou: 0.4961
all_results.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.92,
3
+ "eval_accuracy_background": 0.9922576669620311,
4
+ "eval_accuracy_biofilm": NaN,
5
+ "eval_iou_background": 0.9922576669620311,
6
+ "eval_iou_biofilm": 0.0,
7
+ "eval_loss": 0.020812883973121643,
8
+ "eval_mean_accuracy": 0.9922576669620311,
9
+ "eval_mean_iou": 0.49612883348101555,
10
+ "eval_overall_accuracy": 0.9922576669620311,
11
+ "eval_runtime": 21.6884,
12
+ "eval_samples_per_second": 11.757,
13
+ "eval_steps_per_second": 1.475,
14
+ "train_loss": 0.03354803665876389,
15
+ "train_runtime": 13691.4846,
16
+ "train_samples_per_second": 5.843,
17
+ "train_steps_per_second": 0.73
18
+ }
eval_results.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.92,
3
+ "eval_accuracy_background": 0.9922576669620311,
4
+ "eval_accuracy_biofilm": NaN,
5
+ "eval_iou_background": 0.9922576669620311,
6
+ "eval_iou_biofilm": 0.0,
7
+ "eval_loss": 0.020812883973121643,
8
+ "eval_mean_accuracy": 0.9922576669620311,
9
+ "eval_mean_iou": 0.49612883348101555,
10
+ "eval_overall_accuracy": 0.9922576669620311,
11
+ "eval_runtime": 21.6884,
12
+ "eval_samples_per_second": 11.757,
13
+ "eval_steps_per_second": 1.475
14
+ }
runs/Jul10_10-42-15_gpu-1-130.local/events.out.tfevents.1720618387.gpu-1-130.local.221507.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3dae1377fcc27daceae1650cb2d78cd1a004564a010518df1f4f5a9f95ea58b4
3
+ size 764
train_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 20.92,
3
+ "train_loss": 0.03354803665876389,
4
+ "train_runtime": 13691.4846,
5
+ "train_samples_per_second": 5.843,
6
+ "train_steps_per_second": 0.73
7
+ }
trainer_state.json ADDED
@@ -0,0 +1,945 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 20.92050209205021,
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.21,
13
+ "learning_rate": 5.9401e-05,
14
+ "loss": 0.3126,
15
+ "step": 100
16
+ },
17
+ {
18
+ "epoch": 0.42,
19
+ "learning_rate": 5.8802000000000004e-05,
20
+ "loss": 0.1588,
21
+ "step": 200
22
+ },
23
+ {
24
+ "epoch": 0.63,
25
+ "learning_rate": 5.8203e-05,
26
+ "loss": 0.1026,
27
+ "step": 300
28
+ },
29
+ {
30
+ "epoch": 0.84,
31
+ "learning_rate": 5.7604e-05,
32
+ "loss": 0.0713,
33
+ "step": 400
34
+ },
35
+ {
36
+ "epoch": 1.0,
37
+ "eval_accuracy_background": 0.99061704638999,
38
+ "eval_accuracy_biofilm": NaN,
39
+ "eval_iou_background": 0.99061704638999,
40
+ "eval_iou_biofilm": 0.0,
41
+ "eval_loss": 0.038060422986745834,
42
+ "eval_mean_accuracy": 0.99061704638999,
43
+ "eval_mean_iou": 0.495308523194995,
44
+ "eval_overall_accuracy": 0.99061704638999,
45
+ "eval_runtime": 20.5844,
46
+ "eval_samples_per_second": 12.388,
47
+ "eval_steps_per_second": 1.555,
48
+ "step": 478
49
+ },
50
+ {
51
+ "epoch": 1.05,
52
+ "learning_rate": 5.7005e-05,
53
+ "loss": 0.0599,
54
+ "step": 500
55
+ },
56
+ {
57
+ "epoch": 1.26,
58
+ "learning_rate": 5.6406e-05,
59
+ "loss": 0.05,
60
+ "step": 600
61
+ },
62
+ {
63
+ "epoch": 1.46,
64
+ "learning_rate": 5.5806999999999996e-05,
65
+ "loss": 0.0471,
66
+ "step": 700
67
+ },
68
+ {
69
+ "epoch": 1.67,
70
+ "learning_rate": 5.5208000000000004e-05,
71
+ "loss": 0.0435,
72
+ "step": 800
73
+ },
74
+ {
75
+ "epoch": 1.88,
76
+ "learning_rate": 5.4609000000000005e-05,
77
+ "loss": 0.044,
78
+ "step": 900
79
+ },
80
+ {
81
+ "epoch": 2.0,
82
+ "eval_accuracy_background": 0.99493468531224,
83
+ "eval_accuracy_biofilm": NaN,
84
+ "eval_iou_background": 0.99493468531224,
85
+ "eval_iou_biofilm": 0.0,
86
+ "eval_loss": 0.020206088200211525,
87
+ "eval_mean_accuracy": 0.99493468531224,
88
+ "eval_mean_iou": 0.49746734265612,
89
+ "eval_overall_accuracy": 0.99493468531224,
90
+ "eval_runtime": 20.9376,
91
+ "eval_samples_per_second": 12.179,
92
+ "eval_steps_per_second": 1.528,
93
+ "step": 956
94
+ },
95
+ {
96
+ "epoch": 2.09,
97
+ "learning_rate": 5.401e-05,
98
+ "loss": 0.0486,
99
+ "step": 1000
100
+ },
101
+ {
102
+ "epoch": 2.3,
103
+ "learning_rate": 5.3411e-05,
104
+ "loss": 0.042,
105
+ "step": 1100
106
+ },
107
+ {
108
+ "epoch": 2.51,
109
+ "learning_rate": 5.2812e-05,
110
+ "loss": 0.0378,
111
+ "step": 1200
112
+ },
113
+ {
114
+ "epoch": 2.72,
115
+ "learning_rate": 5.2213e-05,
116
+ "loss": 0.0429,
117
+ "step": 1300
118
+ },
119
+ {
120
+ "epoch": 2.93,
121
+ "learning_rate": 5.1614000000000004e-05,
122
+ "loss": 0.041,
123
+ "step": 1400
124
+ },
125
+ {
126
+ "epoch": 3.0,
127
+ "eval_accuracy_background": 0.9944646370264222,
128
+ "eval_accuracy_biofilm": NaN,
129
+ "eval_iou_background": 0.9944646370264222,
130
+ "eval_iou_biofilm": 0.0,
131
+ "eval_loss": 0.01811068318784237,
132
+ "eval_mean_accuracy": 0.9944646370264222,
133
+ "eval_mean_iou": 0.4972323185132111,
134
+ "eval_overall_accuracy": 0.9944646370264222,
135
+ "eval_runtime": 20.8434,
136
+ "eval_samples_per_second": 12.234,
137
+ "eval_steps_per_second": 1.535,
138
+ "step": 1434
139
+ },
140
+ {
141
+ "epoch": 3.14,
142
+ "learning_rate": 5.1015e-05,
143
+ "loss": 0.0348,
144
+ "step": 1500
145
+ },
146
+ {
147
+ "epoch": 3.35,
148
+ "learning_rate": 5.0416e-05,
149
+ "loss": 0.0338,
150
+ "step": 1600
151
+ },
152
+ {
153
+ "epoch": 3.56,
154
+ "learning_rate": 4.9817e-05,
155
+ "loss": 0.0378,
156
+ "step": 1700
157
+ },
158
+ {
159
+ "epoch": 3.77,
160
+ "learning_rate": 4.9218e-05,
161
+ "loss": 0.0316,
162
+ "step": 1800
163
+ },
164
+ {
165
+ "epoch": 3.97,
166
+ "learning_rate": 4.8619e-05,
167
+ "loss": 0.0361,
168
+ "step": 1900
169
+ },
170
+ {
171
+ "epoch": 4.0,
172
+ "eval_accuracy_background": 0.9925826278137663,
173
+ "eval_accuracy_biofilm": NaN,
174
+ "eval_iou_background": 0.9925826278137663,
175
+ "eval_iou_biofilm": 0.0,
176
+ "eval_loss": 0.020319059491157532,
177
+ "eval_mean_accuracy": 0.9925826278137663,
178
+ "eval_mean_iou": 0.49629131390688314,
179
+ "eval_overall_accuracy": 0.9925826278137663,
180
+ "eval_runtime": 21.9308,
181
+ "eval_samples_per_second": 11.627,
182
+ "eval_steps_per_second": 1.459,
183
+ "step": 1912
184
+ },
185
+ {
186
+ "epoch": 4.18,
187
+ "learning_rate": 4.8020000000000004e-05,
188
+ "loss": 0.0302,
189
+ "step": 2000
190
+ },
191
+ {
192
+ "epoch": 4.39,
193
+ "learning_rate": 4.7421000000000006e-05,
194
+ "loss": 0.0303,
195
+ "step": 2100
196
+ },
197
+ {
198
+ "epoch": 4.6,
199
+ "learning_rate": 4.6822e-05,
200
+ "loss": 0.0344,
201
+ "step": 2200
202
+ },
203
+ {
204
+ "epoch": 4.81,
205
+ "learning_rate": 4.6223e-05,
206
+ "loss": 0.0357,
207
+ "step": 2300
208
+ },
209
+ {
210
+ "epoch": 5.0,
211
+ "eval_accuracy_background": 0.9941524381895419,
212
+ "eval_accuracy_biofilm": NaN,
213
+ "eval_iou_background": 0.9941524381895419,
214
+ "eval_iou_biofilm": 0.0,
215
+ "eval_loss": 0.01632993295788765,
216
+ "eval_mean_accuracy": 0.9941524381895419,
217
+ "eval_mean_iou": 0.49707621909477095,
218
+ "eval_overall_accuracy": 0.9941524381895419,
219
+ "eval_runtime": 21.4286,
220
+ "eval_samples_per_second": 11.9,
221
+ "eval_steps_per_second": 1.493,
222
+ "step": 2390
223
+ },
224
+ {
225
+ "epoch": 5.02,
226
+ "learning_rate": 4.5624e-05,
227
+ "loss": 0.0258,
228
+ "step": 2400
229
+ },
230
+ {
231
+ "epoch": 5.23,
232
+ "learning_rate": 4.5025000000000003e-05,
233
+ "loss": 0.0278,
234
+ "step": 2500
235
+ },
236
+ {
237
+ "epoch": 5.44,
238
+ "learning_rate": 4.4426000000000005e-05,
239
+ "loss": 0.0317,
240
+ "step": 2600
241
+ },
242
+ {
243
+ "epoch": 5.65,
244
+ "learning_rate": 4.3827e-05,
245
+ "loss": 0.0286,
246
+ "step": 2700
247
+ },
248
+ {
249
+ "epoch": 5.86,
250
+ "learning_rate": 4.3228e-05,
251
+ "loss": 0.0336,
252
+ "step": 2800
253
+ },
254
+ {
255
+ "epoch": 6.0,
256
+ "eval_accuracy_background": 0.9915140030588883,
257
+ "eval_accuracy_biofilm": NaN,
258
+ "eval_iou_background": 0.9915140030588883,
259
+ "eval_iou_biofilm": 0.0,
260
+ "eval_loss": 0.03403358906507492,
261
+ "eval_mean_accuracy": 0.9915140030588883,
262
+ "eval_mean_iou": 0.49575700152944413,
263
+ "eval_overall_accuracy": 0.9915140030588883,
264
+ "eval_runtime": 21.1945,
265
+ "eval_samples_per_second": 12.031,
266
+ "eval_steps_per_second": 1.51,
267
+ "step": 2868
268
+ },
269
+ {
270
+ "epoch": 6.07,
271
+ "learning_rate": 4.2629e-05,
272
+ "loss": 0.035,
273
+ "step": 2900
274
+ },
275
+ {
276
+ "epoch": 6.28,
277
+ "learning_rate": 4.203e-05,
278
+ "loss": 0.0262,
279
+ "step": 3000
280
+ },
281
+ {
282
+ "epoch": 6.49,
283
+ "learning_rate": 4.1431e-05,
284
+ "loss": 0.0293,
285
+ "step": 3100
286
+ },
287
+ {
288
+ "epoch": 6.69,
289
+ "learning_rate": 4.0832e-05,
290
+ "loss": 0.0291,
291
+ "step": 3200
292
+ },
293
+ {
294
+ "epoch": 6.9,
295
+ "learning_rate": 4.0233e-05,
296
+ "loss": 0.0295,
297
+ "step": 3300
298
+ },
299
+ {
300
+ "epoch": 7.0,
301
+ "eval_accuracy_background": 0.9955253040954527,
302
+ "eval_accuracy_biofilm": NaN,
303
+ "eval_iou_background": 0.9955253040954527,
304
+ "eval_iou_biofilm": 0.0,
305
+ "eval_loss": 0.012596790678799152,
306
+ "eval_mean_accuracy": 0.9955253040954527,
307
+ "eval_mean_iou": 0.49776265204772635,
308
+ "eval_overall_accuracy": 0.9955253040954527,
309
+ "eval_runtime": 21.3112,
310
+ "eval_samples_per_second": 11.966,
311
+ "eval_steps_per_second": 1.502,
312
+ "step": 3346
313
+ },
314
+ {
315
+ "epoch": 7.11,
316
+ "learning_rate": 3.9634e-05,
317
+ "loss": 0.0372,
318
+ "step": 3400
319
+ },
320
+ {
321
+ "epoch": 7.32,
322
+ "learning_rate": 3.9035e-05,
323
+ "loss": 0.0282,
324
+ "step": 3500
325
+ },
326
+ {
327
+ "epoch": 7.53,
328
+ "learning_rate": 3.8436e-05,
329
+ "loss": 0.0302,
330
+ "step": 3600
331
+ },
332
+ {
333
+ "epoch": 7.74,
334
+ "learning_rate": 3.7837000000000004e-05,
335
+ "loss": 0.0318,
336
+ "step": 3700
337
+ },
338
+ {
339
+ "epoch": 7.95,
340
+ "learning_rate": 3.7238000000000005e-05,
341
+ "loss": 0.0251,
342
+ "step": 3800
343
+ },
344
+ {
345
+ "epoch": 8.0,
346
+ "eval_accuracy_background": 0.9914670494104436,
347
+ "eval_accuracy_biofilm": NaN,
348
+ "eval_iou_background": 0.9914670494104436,
349
+ "eval_iou_biofilm": 0.0,
350
+ "eval_loss": 0.022036196663975716,
351
+ "eval_mean_accuracy": 0.9914670494104436,
352
+ "eval_mean_iou": 0.4957335247052218,
353
+ "eval_overall_accuracy": 0.9914670494104436,
354
+ "eval_runtime": 21.5063,
355
+ "eval_samples_per_second": 11.857,
356
+ "eval_steps_per_second": 1.488,
357
+ "step": 3824
358
+ },
359
+ {
360
+ "epoch": 8.16,
361
+ "learning_rate": 3.6639e-05,
362
+ "loss": 0.0249,
363
+ "step": 3900
364
+ },
365
+ {
366
+ "epoch": 8.37,
367
+ "learning_rate": 3.604e-05,
368
+ "loss": 0.0325,
369
+ "step": 4000
370
+ },
371
+ {
372
+ "epoch": 8.58,
373
+ "learning_rate": 3.544100000000001e-05,
374
+ "loss": 0.0296,
375
+ "step": 4100
376
+ },
377
+ {
378
+ "epoch": 8.79,
379
+ "learning_rate": 3.4842e-05,
380
+ "loss": 0.0265,
381
+ "step": 4200
382
+ },
383
+ {
384
+ "epoch": 9.0,
385
+ "learning_rate": 3.4243000000000004e-05,
386
+ "loss": 0.0265,
387
+ "step": 4300
388
+ },
389
+ {
390
+ "epoch": 9.0,
391
+ "eval_accuracy_background": 0.9932536159220945,
392
+ "eval_accuracy_biofilm": NaN,
393
+ "eval_iou_background": 0.9932536159220945,
394
+ "eval_iou_biofilm": 0.0,
395
+ "eval_loss": 0.01821214146912098,
396
+ "eval_mean_accuracy": 0.9932536159220945,
397
+ "eval_mean_iou": 0.49662680796104725,
398
+ "eval_overall_accuracy": 0.9932536159220945,
399
+ "eval_runtime": 21.4325,
400
+ "eval_samples_per_second": 11.898,
401
+ "eval_steps_per_second": 1.493,
402
+ "step": 4302
403
+ },
404
+ {
405
+ "epoch": 9.21,
406
+ "learning_rate": 3.3644000000000005e-05,
407
+ "loss": 0.0294,
408
+ "step": 4400
409
+ },
410
+ {
411
+ "epoch": 9.41,
412
+ "learning_rate": 3.3045000000000006e-05,
413
+ "loss": 0.0246,
414
+ "step": 4500
415
+ },
416
+ {
417
+ "epoch": 9.62,
418
+ "learning_rate": 3.2446e-05,
419
+ "loss": 0.0253,
420
+ "step": 4600
421
+ },
422
+ {
423
+ "epoch": 9.83,
424
+ "learning_rate": 3.1847e-05,
425
+ "loss": 0.0238,
426
+ "step": 4700
427
+ },
428
+ {
429
+ "epoch": 10.0,
430
+ "eval_accuracy_background": 0.9940080441637696,
431
+ "eval_accuracy_biofilm": NaN,
432
+ "eval_iou_background": 0.9940080441637696,
433
+ "eval_iou_biofilm": 0.0,
434
+ "eval_loss": 0.015465262345969677,
435
+ "eval_mean_accuracy": 0.9940080441637696,
436
+ "eval_mean_iou": 0.4970040220818848,
437
+ "eval_overall_accuracy": 0.9940080441637696,
438
+ "eval_runtime": 22.1026,
439
+ "eval_samples_per_second": 11.537,
440
+ "eval_steps_per_second": 1.448,
441
+ "step": 4780
442
+ },
443
+ {
444
+ "epoch": 10.04,
445
+ "learning_rate": 3.1248e-05,
446
+ "loss": 0.0262,
447
+ "step": 4800
448
+ },
449
+ {
450
+ "epoch": 10.25,
451
+ "learning_rate": 3.0649000000000004e-05,
452
+ "loss": 0.0253,
453
+ "step": 4900
454
+ },
455
+ {
456
+ "epoch": 10.46,
457
+ "learning_rate": 3.0050000000000002e-05,
458
+ "loss": 0.0266,
459
+ "step": 5000
460
+ },
461
+ {
462
+ "epoch": 10.67,
463
+ "learning_rate": 2.9451e-05,
464
+ "loss": 0.028,
465
+ "step": 5100
466
+ },
467
+ {
468
+ "epoch": 10.88,
469
+ "learning_rate": 2.8851999999999998e-05,
470
+ "loss": 0.0258,
471
+ "step": 5200
472
+ },
473
+ {
474
+ "epoch": 11.0,
475
+ "eval_accuracy_background": 0.9931246419763669,
476
+ "eval_accuracy_biofilm": NaN,
477
+ "eval_iou_background": 0.9931246419763669,
478
+ "eval_iou_biofilm": 0.0,
479
+ "eval_loss": 0.01807301491498947,
480
+ "eval_mean_accuracy": 0.9931246419763669,
481
+ "eval_mean_iou": 0.4965623209881834,
482
+ "eval_overall_accuracy": 0.9931246419763669,
483
+ "eval_runtime": 21.5251,
484
+ "eval_samples_per_second": 11.847,
485
+ "eval_steps_per_second": 1.487,
486
+ "step": 5258
487
+ },
488
+ {
489
+ "epoch": 11.09,
490
+ "learning_rate": 2.8253e-05,
491
+ "loss": 0.0228,
492
+ "step": 5300
493
+ },
494
+ {
495
+ "epoch": 11.3,
496
+ "learning_rate": 2.7653999999999996e-05,
497
+ "loss": 0.0246,
498
+ "step": 5400
499
+ },
500
+ {
501
+ "epoch": 11.51,
502
+ "learning_rate": 2.7054999999999998e-05,
503
+ "loss": 0.0262,
504
+ "step": 5500
505
+ },
506
+ {
507
+ "epoch": 11.72,
508
+ "learning_rate": 2.6455999999999995e-05,
509
+ "loss": 0.0306,
510
+ "step": 5600
511
+ },
512
+ {
513
+ "epoch": 11.92,
514
+ "learning_rate": 2.5857e-05,
515
+ "loss": 0.0264,
516
+ "step": 5700
517
+ },
518
+ {
519
+ "epoch": 12.0,
520
+ "eval_accuracy_background": 0.9937508887389549,
521
+ "eval_accuracy_biofilm": NaN,
522
+ "eval_iou_background": 0.9937508887389549,
523
+ "eval_iou_biofilm": 0.0,
524
+ "eval_loss": 0.01792028360068798,
525
+ "eval_mean_accuracy": 0.9937508887389549,
526
+ "eval_mean_iou": 0.49687544436947745,
527
+ "eval_overall_accuracy": 0.9937508887389549,
528
+ "eval_runtime": 21.5267,
529
+ "eval_samples_per_second": 11.846,
530
+ "eval_steps_per_second": 1.487,
531
+ "step": 5736
532
+ },
533
+ {
534
+ "epoch": 12.13,
535
+ "learning_rate": 2.5258e-05,
536
+ "loss": 0.028,
537
+ "step": 5800
538
+ },
539
+ {
540
+ "epoch": 12.34,
541
+ "learning_rate": 2.4659e-05,
542
+ "loss": 0.0268,
543
+ "step": 5900
544
+ },
545
+ {
546
+ "epoch": 12.55,
547
+ "learning_rate": 2.406e-05,
548
+ "loss": 0.0266,
549
+ "step": 6000
550
+ },
551
+ {
552
+ "epoch": 12.76,
553
+ "learning_rate": 2.3460999999999998e-05,
554
+ "loss": 0.0242,
555
+ "step": 6100
556
+ },
557
+ {
558
+ "epoch": 12.97,
559
+ "learning_rate": 2.2862e-05,
560
+ "loss": 0.0265,
561
+ "step": 6200
562
+ },
563
+ {
564
+ "epoch": 13.0,
565
+ "eval_accuracy_background": 0.991736174383473,
566
+ "eval_accuracy_biofilm": NaN,
567
+ "eval_iou_background": 0.991736174383473,
568
+ "eval_iou_biofilm": 0.0,
569
+ "eval_loss": 0.022188672795891762,
570
+ "eval_mean_accuracy": 0.991736174383473,
571
+ "eval_mean_iou": 0.4958680871917365,
572
+ "eval_overall_accuracy": 0.991736174383473,
573
+ "eval_runtime": 22.1635,
574
+ "eval_samples_per_second": 11.505,
575
+ "eval_steps_per_second": 1.444,
576
+ "step": 6214
577
+ },
578
+ {
579
+ "epoch": 13.18,
580
+ "learning_rate": 2.2263e-05,
581
+ "loss": 0.0233,
582
+ "step": 6300
583
+ },
584
+ {
585
+ "epoch": 13.39,
586
+ "learning_rate": 2.1663999999999998e-05,
587
+ "loss": 0.0232,
588
+ "step": 6400
589
+ },
590
+ {
591
+ "epoch": 13.6,
592
+ "learning_rate": 2.1065e-05,
593
+ "loss": 0.0311,
594
+ "step": 6500
595
+ },
596
+ {
597
+ "epoch": 13.81,
598
+ "learning_rate": 2.0465999999999997e-05,
599
+ "loss": 0.0219,
600
+ "step": 6600
601
+ },
602
+ {
603
+ "epoch": 14.0,
604
+ "eval_accuracy_background": 0.9924663828634505,
605
+ "eval_accuracy_biofilm": NaN,
606
+ "eval_iou_background": 0.9924663828634505,
607
+ "eval_iou_biofilm": 0.0,
608
+ "eval_loss": 0.020027143880724907,
609
+ "eval_mean_accuracy": 0.9924663828634505,
610
+ "eval_mean_iou": 0.49623319143172523,
611
+ "eval_overall_accuracy": 0.9924663828634505,
612
+ "eval_runtime": 22.2537,
613
+ "eval_samples_per_second": 11.459,
614
+ "eval_steps_per_second": 1.438,
615
+ "step": 6692
616
+ },
617
+ {
618
+ "epoch": 14.02,
619
+ "learning_rate": 1.9866999999999998e-05,
620
+ "loss": 0.0234,
621
+ "step": 6700
622
+ },
623
+ {
624
+ "epoch": 14.23,
625
+ "learning_rate": 1.9267999999999996e-05,
626
+ "loss": 0.0281,
627
+ "step": 6800
628
+ },
629
+ {
630
+ "epoch": 14.44,
631
+ "learning_rate": 1.8669e-05,
632
+ "loss": 0.0243,
633
+ "step": 6900
634
+ },
635
+ {
636
+ "epoch": 14.64,
637
+ "learning_rate": 1.807e-05,
638
+ "loss": 0.0238,
639
+ "step": 7000
640
+ },
641
+ {
642
+ "epoch": 14.85,
643
+ "learning_rate": 1.7471e-05,
644
+ "loss": 0.0213,
645
+ "step": 7100
646
+ },
647
+ {
648
+ "epoch": 15.0,
649
+ "eval_accuracy_background": 0.9916155213374698,
650
+ "eval_accuracy_biofilm": NaN,
651
+ "eval_iou_background": 0.9916155213374698,
652
+ "eval_iou_biofilm": 0.0,
653
+ "eval_loss": 0.023391906172037125,
654
+ "eval_mean_accuracy": 0.9916155213374698,
655
+ "eval_mean_iou": 0.4958077606687349,
656
+ "eval_overall_accuracy": 0.9916155213374698,
657
+ "eval_runtime": 21.7106,
658
+ "eval_samples_per_second": 11.745,
659
+ "eval_steps_per_second": 1.474,
660
+ "step": 7170
661
+ },
662
+ {
663
+ "epoch": 15.06,
664
+ "learning_rate": 1.6872e-05,
665
+ "loss": 0.0231,
666
+ "step": 7200
667
+ },
668
+ {
669
+ "epoch": 15.27,
670
+ "learning_rate": 1.6272999999999998e-05,
671
+ "loss": 0.0252,
672
+ "step": 7300
673
+ },
674
+ {
675
+ "epoch": 15.48,
676
+ "learning_rate": 1.5674e-05,
677
+ "loss": 0.024,
678
+ "step": 7400
679
+ },
680
+ {
681
+ "epoch": 15.69,
682
+ "learning_rate": 1.5075000000000002e-05,
683
+ "loss": 0.0205,
684
+ "step": 7500
685
+ },
686
+ {
687
+ "epoch": 15.9,
688
+ "learning_rate": 1.4476e-05,
689
+ "loss": 0.0192,
690
+ "step": 7600
691
+ },
692
+ {
693
+ "epoch": 16.0,
694
+ "eval_accuracy_background": 0.9922337608850452,
695
+ "eval_accuracy_biofilm": NaN,
696
+ "eval_iou_background": 0.9922337608850452,
697
+ "eval_iou_biofilm": 0.0,
698
+ "eval_loss": 0.019924867898225784,
699
+ "eval_mean_accuracy": 0.9922337608850452,
700
+ "eval_mean_iou": 0.4961168804425226,
701
+ "eval_overall_accuracy": 0.9922337608850452,
702
+ "eval_runtime": 21.6803,
703
+ "eval_samples_per_second": 11.762,
704
+ "eval_steps_per_second": 1.476,
705
+ "step": 7648
706
+ },
707
+ {
708
+ "epoch": 16.11,
709
+ "learning_rate": 1.3877e-05,
710
+ "loss": 0.0234,
711
+ "step": 7700
712
+ },
713
+ {
714
+ "epoch": 16.32,
715
+ "learning_rate": 1.3277999999999999e-05,
716
+ "loss": 0.0244,
717
+ "step": 7800
718
+ },
719
+ {
720
+ "epoch": 16.53,
721
+ "learning_rate": 1.2678999999999998e-05,
722
+ "loss": 0.0187,
723
+ "step": 7900
724
+ },
725
+ {
726
+ "epoch": 16.74,
727
+ "learning_rate": 1.2079999999999998e-05,
728
+ "loss": 0.0232,
729
+ "step": 8000
730
+ },
731
+ {
732
+ "epoch": 16.95,
733
+ "learning_rate": 1.1480999999999997e-05,
734
+ "loss": 0.0232,
735
+ "step": 8100
736
+ },
737
+ {
738
+ "epoch": 17.0,
739
+ "eval_accuracy_background": 0.9922697355685758,
740
+ "eval_accuracy_biofilm": NaN,
741
+ "eval_iou_background": 0.9922697355685758,
742
+ "eval_iou_biofilm": 0.0,
743
+ "eval_loss": 0.020755073055624962,
744
+ "eval_mean_accuracy": 0.9922697355685758,
745
+ "eval_mean_iou": 0.4961348677842879,
746
+ "eval_overall_accuracy": 0.9922697355685758,
747
+ "eval_runtime": 21.7178,
748
+ "eval_samples_per_second": 11.742,
749
+ "eval_steps_per_second": 1.473,
750
+ "step": 8126
751
+ },
752
+ {
753
+ "epoch": 17.15,
754
+ "learning_rate": 1.0882000000000004e-05,
755
+ "loss": 0.0256,
756
+ "step": 8200
757
+ },
758
+ {
759
+ "epoch": 17.36,
760
+ "learning_rate": 1.0283000000000003e-05,
761
+ "loss": 0.0195,
762
+ "step": 8300
763
+ },
764
+ {
765
+ "epoch": 17.57,
766
+ "learning_rate": 9.684000000000002e-06,
767
+ "loss": 0.0212,
768
+ "step": 8400
769
+ },
770
+ {
771
+ "epoch": 17.78,
772
+ "learning_rate": 9.085000000000002e-06,
773
+ "loss": 0.023,
774
+ "step": 8500
775
+ },
776
+ {
777
+ "epoch": 17.99,
778
+ "learning_rate": 8.486000000000001e-06,
779
+ "loss": 0.0219,
780
+ "step": 8600
781
+ },
782
+ {
783
+ "epoch": 18.0,
784
+ "eval_accuracy_background": 0.9909196860980203,
785
+ "eval_accuracy_biofilm": NaN,
786
+ "eval_iou_background": 0.9909196860980203,
787
+ "eval_iou_biofilm": 0.0,
788
+ "eval_loss": 0.02454926259815693,
789
+ "eval_mean_accuracy": 0.9909196860980203,
790
+ "eval_mean_iou": 0.49545984304901014,
791
+ "eval_overall_accuracy": 0.9909196860980203,
792
+ "eval_runtime": 22.2518,
793
+ "eval_samples_per_second": 11.46,
794
+ "eval_steps_per_second": 1.438,
795
+ "step": 8604
796
+ },
797
+ {
798
+ "epoch": 18.2,
799
+ "learning_rate": 7.887000000000001e-06,
800
+ "loss": 0.0223,
801
+ "step": 8700
802
+ },
803
+ {
804
+ "epoch": 18.41,
805
+ "learning_rate": 7.2879999999999995e-06,
806
+ "loss": 0.0249,
807
+ "step": 8800
808
+ },
809
+ {
810
+ "epoch": 18.62,
811
+ "learning_rate": 6.688999999999999e-06,
812
+ "loss": 0.0259,
813
+ "step": 8900
814
+ },
815
+ {
816
+ "epoch": 18.83,
817
+ "learning_rate": 6.0899999999999984e-06,
818
+ "loss": 0.0201,
819
+ "step": 9000
820
+ },
821
+ {
822
+ "epoch": 19.0,
823
+ "eval_accuracy_background": 0.9921929323433422,
824
+ "eval_accuracy_biofilm": NaN,
825
+ "eval_iou_background": 0.9921929323433422,
826
+ "eval_iou_biofilm": 0.0,
827
+ "eval_loss": 0.021141713485121727,
828
+ "eval_mean_accuracy": 0.9921929323433422,
829
+ "eval_mean_iou": 0.4960964661716711,
830
+ "eval_overall_accuracy": 0.9921929323433422,
831
+ "eval_runtime": 22.2034,
832
+ "eval_samples_per_second": 11.485,
833
+ "eval_steps_per_second": 1.441,
834
+ "step": 9082
835
+ },
836
+ {
837
+ "epoch": 19.04,
838
+ "learning_rate": 5.490999999999998e-06,
839
+ "loss": 0.0263,
840
+ "step": 9100
841
+ },
842
+ {
843
+ "epoch": 19.25,
844
+ "learning_rate": 4.891999999999997e-06,
845
+ "loss": 0.0246,
846
+ "step": 9200
847
+ },
848
+ {
849
+ "epoch": 19.46,
850
+ "learning_rate": 4.292999999999997e-06,
851
+ "loss": 0.0176,
852
+ "step": 9300
853
+ },
854
+ {
855
+ "epoch": 19.67,
856
+ "learning_rate": 3.694000000000003e-06,
857
+ "loss": 0.0185,
858
+ "step": 9400
859
+ },
860
+ {
861
+ "epoch": 19.87,
862
+ "learning_rate": 3.0950000000000026e-06,
863
+ "loss": 0.0192,
864
+ "step": 9500
865
+ },
866
+ {
867
+ "epoch": 20.0,
868
+ "eval_accuracy_background": 0.9923248780389713,
869
+ "eval_accuracy_biofilm": NaN,
870
+ "eval_iou_background": 0.9923248780389713,
871
+ "eval_iou_biofilm": 0.0,
872
+ "eval_loss": 0.020733820274472237,
873
+ "eval_mean_accuracy": 0.9923248780389713,
874
+ "eval_mean_iou": 0.49616243901948565,
875
+ "eval_overall_accuracy": 0.9923248780389713,
876
+ "eval_runtime": 21.616,
877
+ "eval_samples_per_second": 11.797,
878
+ "eval_steps_per_second": 1.48,
879
+ "step": 9560
880
+ },
881
+ {
882
+ "epoch": 20.08,
883
+ "learning_rate": 2.496000000000002e-06,
884
+ "loss": 0.0242,
885
+ "step": 9600
886
+ },
887
+ {
888
+ "epoch": 20.29,
889
+ "learning_rate": 1.8970000000000013e-06,
890
+ "loss": 0.0194,
891
+ "step": 9700
892
+ },
893
+ {
894
+ "epoch": 20.5,
895
+ "learning_rate": 1.298000000000001e-06,
896
+ "loss": 0.021,
897
+ "step": 9800
898
+ },
899
+ {
900
+ "epoch": 20.71,
901
+ "learning_rate": 6.990000000000005e-07,
902
+ "loss": 0.0233,
903
+ "step": 9900
904
+ },
905
+ {
906
+ "epoch": 20.92,
907
+ "learning_rate": 1e-07,
908
+ "loss": 0.0175,
909
+ "step": 10000
910
+ },
911
+ {
912
+ "epoch": 20.92,
913
+ "eval_accuracy_background": 0.9922576669620311,
914
+ "eval_accuracy_biofilm": NaN,
915
+ "eval_iou_background": 0.9922576669620311,
916
+ "eval_iou_biofilm": 0.0,
917
+ "eval_loss": 0.020812883973121643,
918
+ "eval_mean_accuracy": 0.9922576669620311,
919
+ "eval_mean_iou": 0.49612883348101555,
920
+ "eval_overall_accuracy": 0.9922576669620311,
921
+ "eval_runtime": 21.7462,
922
+ "eval_samples_per_second": 11.726,
923
+ "eval_steps_per_second": 1.472,
924
+ "step": 10000
925
+ },
926
+ {
927
+ "epoch": 20.92,
928
+ "step": 10000,
929
+ "total_flos": 1.4004836461707264e+18,
930
+ "train_loss": 0.03354803665876389,
931
+ "train_runtime": 13691.4846,
932
+ "train_samples_per_second": 5.843,
933
+ "train_steps_per_second": 0.73
934
+ }
935
+ ],
936
+ "logging_steps": 100,
937
+ "max_steps": 10000,
938
+ "num_input_tokens_seen": 0,
939
+ "num_train_epochs": 21,
940
+ "save_steps": 500,
941
+ "total_flos": 1.4004836461707264e+18,
942
+ "train_batch_size": 8,
943
+ "trial_name": null,
944
+ "trial_params": null
945
+ }