sam1120 commited on
Commit
1c790ff
1 Parent(s): 7d77e37

Training in progress, step 80

Browse files
checkpoint-80/config.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b0",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 256,
9
+ "depths": [
10
+ 2,
11
+ 2,
12
+ 2,
13
+ 2
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
+ 32,
26
+ 64,
27
+ 160,
28
+ 256
29
+ ],
30
+ "id2label": {
31
+ "0": "unlabeled",
32
+ "1": "NAT",
33
+ "2": "concrete",
34
+ "3": "grass",
35
+ "4": "speedway bricks",
36
+ "5": "steel",
37
+ "6": "rough concrete",
38
+ "7": "dark bricks",
39
+ "8": "road",
40
+ "9": "rough red sidewalk",
41
+ "10": "tiles",
42
+ "11": "red bricks",
43
+ "12": "concrete tiles",
44
+ "13": "REST"
45
+ },
46
+ "image_size": 224,
47
+ "initializer_range": 0.02,
48
+ "label2id": {
49
+ "NAT": 1,
50
+ "REST": 13,
51
+ "concrete": 2,
52
+ "concrete tiles": 12,
53
+ "dark bricks": 7,
54
+ "grass": 3,
55
+ "red bricks": 11,
56
+ "road": 8,
57
+ "rough concrete": 6,
58
+ "rough red sidewalk": 9,
59
+ "speedway bricks": 4,
60
+ "steel": 5,
61
+ "tiles": 10,
62
+ "unlabeled": 0
63
+ },
64
+ "layer_norm_eps": 1e-06,
65
+ "mlp_ratios": [
66
+ 4,
67
+ 4,
68
+ 4,
69
+ 4
70
+ ],
71
+ "model_type": "segformer",
72
+ "num_attention_heads": [
73
+ 1,
74
+ 2,
75
+ 5,
76
+ 8
77
+ ],
78
+ "num_channels": 3,
79
+ "num_encoder_blocks": 4,
80
+ "patch_sizes": [
81
+ 7,
82
+ 3,
83
+ 3,
84
+ 3
85
+ ],
86
+ "reshape_last_stage": true,
87
+ "semantic_loss_ignore_index": 255,
88
+ "sr_ratios": [
89
+ 8,
90
+ 4,
91
+ 2,
92
+ 1
93
+ ],
94
+ "strides": [
95
+ 4,
96
+ 2,
97
+ 2,
98
+ 2
99
+ ],
100
+ "torch_dtype": "float32",
101
+ "transformers_version": "4.30.2"
102
+ }
checkpoint-80/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7b88de03d6a828f8d99e5a8364422ba207abc2ae39bbe46c4721f4ea00ec7f6d
3
+ size 29861381
checkpoint-80/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4847e5ae2433df8ac0af66bf9328e1c21165e6bc8e3912e8d0c5f6761bf9b3df
3
+ size 14944077
checkpoint-80/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d36fa1ebf661b7d16f466ead99a6ca8e0ccccf39b14db15504c4d9f429f40ab9
3
+ size 14575
checkpoint-80/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4374eecad6e55b8c9a105a8e6ddd5f8fc33a804e43294396fb159541ebf4ae55
3
+ size 627
checkpoint-80/trainer_state.json ADDED
@@ -0,0 +1,652 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 1.4904648065567017,
3
+ "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0-optim/checkpoint-80",
4
+ "epoch": 7.2727272727272725,
5
+ "global_step": 80,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.09,
12
+ "learning_rate": 3.6363636363636366e-07,
13
+ "loss": 2.757,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.18,
18
+ "learning_rate": 7.272727272727273e-07,
19
+ "loss": 2.7552,
20
+ "step": 2
21
+ },
22
+ {
23
+ "epoch": 0.27,
24
+ "learning_rate": 1.090909090909091e-06,
25
+ "loss": 2.7635,
26
+ "step": 3
27
+ },
28
+ {
29
+ "epoch": 0.36,
30
+ "learning_rate": 1.4545454545454546e-06,
31
+ "loss": 2.7567,
32
+ "step": 4
33
+ },
34
+ {
35
+ "epoch": 0.45,
36
+ "learning_rate": 1.818181818181818e-06,
37
+ "loss": 2.7588,
38
+ "step": 5
39
+ },
40
+ {
41
+ "epoch": 0.55,
42
+ "learning_rate": 2.181818181818182e-06,
43
+ "loss": 2.7621,
44
+ "step": 6
45
+ },
46
+ {
47
+ "epoch": 0.64,
48
+ "learning_rate": 2.5454545454545456e-06,
49
+ "loss": 2.7553,
50
+ "step": 7
51
+ },
52
+ {
53
+ "epoch": 0.73,
54
+ "learning_rate": 2.9090909090909093e-06,
55
+ "loss": 2.7371,
56
+ "step": 8
57
+ },
58
+ {
59
+ "epoch": 0.82,
60
+ "learning_rate": 3.2727272727272733e-06,
61
+ "loss": 2.7444,
62
+ "step": 9
63
+ },
64
+ {
65
+ "epoch": 0.91,
66
+ "learning_rate": 3.636363636363636e-06,
67
+ "loss": 2.7363,
68
+ "step": 10
69
+ },
70
+ {
71
+ "epoch": 1.0,
72
+ "learning_rate": 4.000000000000001e-06,
73
+ "loss": 2.7364,
74
+ "step": 11
75
+ },
76
+ {
77
+ "epoch": 1.09,
78
+ "learning_rate": 4.363636363636364e-06,
79
+ "loss": 2.7248,
80
+ "step": 12
81
+ },
82
+ {
83
+ "epoch": 1.18,
84
+ "learning_rate": 4.727272727272727e-06,
85
+ "loss": 2.7344,
86
+ "step": 13
87
+ },
88
+ {
89
+ "epoch": 1.27,
90
+ "learning_rate": 5.090909090909091e-06,
91
+ "loss": 2.7259,
92
+ "step": 14
93
+ },
94
+ {
95
+ "epoch": 1.36,
96
+ "learning_rate": 5.4545454545454545e-06,
97
+ "loss": 2.7086,
98
+ "step": 15
99
+ },
100
+ {
101
+ "epoch": 1.45,
102
+ "learning_rate": 5.8181818181818185e-06,
103
+ "loss": 2.7087,
104
+ "step": 16
105
+ },
106
+ {
107
+ "epoch": 1.55,
108
+ "learning_rate": 6.181818181818183e-06,
109
+ "loss": 2.7055,
110
+ "step": 17
111
+ },
112
+ {
113
+ "epoch": 1.64,
114
+ "learning_rate": 6.545454545454547e-06,
115
+ "loss": 2.6959,
116
+ "step": 18
117
+ },
118
+ {
119
+ "epoch": 1.73,
120
+ "learning_rate": 6.909090909090909e-06,
121
+ "loss": 2.6983,
122
+ "step": 19
123
+ },
124
+ {
125
+ "epoch": 1.82,
126
+ "learning_rate": 7.272727272727272e-06,
127
+ "loss": 2.6746,
128
+ "step": 20
129
+ },
130
+ {
131
+ "epoch": 1.82,
132
+ "eval_accuracy_NAT": 0.019056367662015652,
133
+ "eval_accuracy_REST": 0.0,
134
+ "eval_accuracy_concrete": 0.0010381346838920125,
135
+ "eval_accuracy_concrete tiles": 0.09181801504311135,
136
+ "eval_accuracy_dark bricks": 0.09326045956660636,
137
+ "eval_accuracy_grass": 0.007958282800392846,
138
+ "eval_accuracy_red bricks": 0.00870028409090909,
139
+ "eval_accuracy_road": 0.05730198902763889,
140
+ "eval_accuracy_rough concrete": 0.03398947127489128,
141
+ "eval_accuracy_rough red sidewalk": 0.06001709260331376,
142
+ "eval_accuracy_speedway bricks": 0.03133919121631325,
143
+ "eval_accuracy_steel": 0.00842622518303542,
144
+ "eval_accuracy_tiles": 0.0,
145
+ "eval_accuracy_unlabeled": NaN,
146
+ "eval_iou_NAT": 0.018937014149192417,
147
+ "eval_iou_REST": 0.0,
148
+ "eval_iou_concrete": 0.0009231352727125269,
149
+ "eval_iou_concrete tiles": 0.002698064724802789,
150
+ "eval_iou_dark bricks": 0.0018963436490879863,
151
+ "eval_iou_grass": 0.006090841001497967,
152
+ "eval_iou_red bricks": 1.3048864535742704e-05,
153
+ "eval_iou_road": 0.0465812133841859,
154
+ "eval_iou_rough concrete": 0.004291938525567382,
155
+ "eval_iou_rough red sidewalk": 0.0013826848729701939,
156
+ "eval_iou_speedway bricks": 0.030977499755458056,
157
+ "eval_iou_steel": 0.0016645601782480097,
158
+ "eval_iou_tiles": 0.0,
159
+ "eval_iou_unlabeled": 0.0,
160
+ "eval_loss": 2.647437810897827,
161
+ "eval_mean_accuracy": 0.03176196255016307,
162
+ "eval_mean_iou": 0.008246881741304212,
163
+ "eval_overall_accuracy": 0.02327410380045573,
164
+ "eval_runtime": 3.4336,
165
+ "eval_samples_per_second": 10.485,
166
+ "eval_steps_per_second": 0.582,
167
+ "step": 20
168
+ },
169
+ {
170
+ "epoch": 1.91,
171
+ "learning_rate": 7.636363636363638e-06,
172
+ "loss": 2.6783,
173
+ "step": 21
174
+ },
175
+ {
176
+ "epoch": 2.0,
177
+ "learning_rate": 8.000000000000001e-06,
178
+ "loss": 2.6636,
179
+ "step": 22
180
+ },
181
+ {
182
+ "epoch": 2.09,
183
+ "learning_rate": 8.363636363636365e-06,
184
+ "loss": 2.6626,
185
+ "step": 23
186
+ },
187
+ {
188
+ "epoch": 2.18,
189
+ "learning_rate": 8.727272727272728e-06,
190
+ "loss": 2.6414,
191
+ "step": 24
192
+ },
193
+ {
194
+ "epoch": 2.27,
195
+ "learning_rate": 9.090909090909091e-06,
196
+ "loss": 2.6365,
197
+ "step": 25
198
+ },
199
+ {
200
+ "epoch": 2.36,
201
+ "learning_rate": 9.454545454545454e-06,
202
+ "loss": 2.6263,
203
+ "step": 26
204
+ },
205
+ {
206
+ "epoch": 2.45,
207
+ "learning_rate": 9.818181818181818e-06,
208
+ "loss": 2.6182,
209
+ "step": 27
210
+ },
211
+ {
212
+ "epoch": 2.55,
213
+ "learning_rate": 1.0181818181818182e-05,
214
+ "loss": 2.6144,
215
+ "step": 28
216
+ },
217
+ {
218
+ "epoch": 2.64,
219
+ "learning_rate": 1.0545454545454546e-05,
220
+ "loss": 2.5878,
221
+ "step": 29
222
+ },
223
+ {
224
+ "epoch": 2.73,
225
+ "learning_rate": 1.0909090909090909e-05,
226
+ "loss": 2.5814,
227
+ "step": 30
228
+ },
229
+ {
230
+ "epoch": 2.82,
231
+ "learning_rate": 1.1272727272727274e-05,
232
+ "loss": 2.5739,
233
+ "step": 31
234
+ },
235
+ {
236
+ "epoch": 2.91,
237
+ "learning_rate": 1.1636363636363637e-05,
238
+ "loss": 2.5536,
239
+ "step": 32
240
+ },
241
+ {
242
+ "epoch": 3.0,
243
+ "learning_rate": 1.2e-05,
244
+ "loss": 2.539,
245
+ "step": 33
246
+ },
247
+ {
248
+ "epoch": 3.09,
249
+ "learning_rate": 1.2363636363636365e-05,
250
+ "loss": 2.5256,
251
+ "step": 34
252
+ },
253
+ {
254
+ "epoch": 3.18,
255
+ "learning_rate": 1.2727272727272727e-05,
256
+ "loss": 2.4979,
257
+ "step": 35
258
+ },
259
+ {
260
+ "epoch": 3.27,
261
+ "learning_rate": 1.3090909090909093e-05,
262
+ "loss": 2.4936,
263
+ "step": 36
264
+ },
265
+ {
266
+ "epoch": 3.36,
267
+ "learning_rate": 1.3454545454545457e-05,
268
+ "loss": 2.4862,
269
+ "step": 37
270
+ },
271
+ {
272
+ "epoch": 3.45,
273
+ "learning_rate": 1.3818181818181818e-05,
274
+ "loss": 2.4491,
275
+ "step": 38
276
+ },
277
+ {
278
+ "epoch": 3.55,
279
+ "learning_rate": 1.4181818181818181e-05,
280
+ "loss": 2.4468,
281
+ "step": 39
282
+ },
283
+ {
284
+ "epoch": 3.64,
285
+ "learning_rate": 1.4545454545454545e-05,
286
+ "loss": 2.4385,
287
+ "step": 40
288
+ },
289
+ {
290
+ "epoch": 3.64,
291
+ "eval_accuracy_NAT": 0.447892573737157,
292
+ "eval_accuracy_REST": 0.0,
293
+ "eval_accuracy_concrete": 0.04360568050130907,
294
+ "eval_accuracy_concrete tiles": 0.10279459426404941,
295
+ "eval_accuracy_dark bricks": 0.07266848851083484,
296
+ "eval_accuracy_grass": 0.017545269942016727,
297
+ "eval_accuracy_red bricks": 0.0,
298
+ "eval_accuracy_road": 0.37485716220448445,
299
+ "eval_accuracy_rough concrete": 0.2949759670405127,
300
+ "eval_accuracy_rough red sidewalk": 0.05480660546411932,
301
+ "eval_accuracy_speedway bricks": 0.7969013802350152,
302
+ "eval_accuracy_steel": 0.0050211067871512435,
303
+ "eval_accuracy_tiles": 0.0,
304
+ "eval_accuracy_unlabeled": NaN,
305
+ "eval_iou_NAT": 0.4472536503231145,
306
+ "eval_iou_REST": 0.0,
307
+ "eval_iou_concrete": 0.039962583094048695,
308
+ "eval_iou_concrete tiles": 0.0041023463419931645,
309
+ "eval_iou_dark bricks": 0.007329448622040366,
310
+ "eval_iou_grass": 0.010573184550513284,
311
+ "eval_iou_red bricks": 0.0,
312
+ "eval_iou_road": 0.2852347789122688,
313
+ "eval_iou_rough concrete": 0.048997709321445884,
314
+ "eval_iou_rough red sidewalk": 0.003679975417468036,
315
+ "eval_iou_speedway bricks": 0.7166237181229831,
316
+ "eval_iou_steel": 0.0015635711039736272,
317
+ "eval_iou_tiles": 0.0,
318
+ "eval_iou_unlabeled": 0.0,
319
+ "eval_loss": 2.4373269081115723,
320
+ "eval_mean_accuracy": 0.17008221759128078,
321
+ "eval_mean_iou": 0.11180864041498925,
322
+ "eval_overall_accuracy": 0.4471186531914605,
323
+ "eval_runtime": 3.2402,
324
+ "eval_samples_per_second": 11.111,
325
+ "eval_steps_per_second": 0.617,
326
+ "step": 40
327
+ },
328
+ {
329
+ "epoch": 3.73,
330
+ "learning_rate": 1.4909090909090908e-05,
331
+ "loss": 2.4028,
332
+ "step": 41
333
+ },
334
+ {
335
+ "epoch": 3.82,
336
+ "learning_rate": 1.5272727272727276e-05,
337
+ "loss": 2.4083,
338
+ "step": 42
339
+ },
340
+ {
341
+ "epoch": 3.91,
342
+ "learning_rate": 1.563636363636364e-05,
343
+ "loss": 2.3613,
344
+ "step": 43
345
+ },
346
+ {
347
+ "epoch": 4.0,
348
+ "learning_rate": 1.6000000000000003e-05,
349
+ "loss": 2.3597,
350
+ "step": 44
351
+ },
352
+ {
353
+ "epoch": 4.09,
354
+ "learning_rate": 1.6363636363636366e-05,
355
+ "loss": 2.3331,
356
+ "step": 45
357
+ },
358
+ {
359
+ "epoch": 4.18,
360
+ "learning_rate": 1.672727272727273e-05,
361
+ "loss": 2.3068,
362
+ "step": 46
363
+ },
364
+ {
365
+ "epoch": 4.27,
366
+ "learning_rate": 1.7090909090909092e-05,
367
+ "loss": 2.2492,
368
+ "step": 47
369
+ },
370
+ {
371
+ "epoch": 4.36,
372
+ "learning_rate": 1.7454545454545456e-05,
373
+ "loss": 2.2469,
374
+ "step": 48
375
+ },
376
+ {
377
+ "epoch": 4.45,
378
+ "learning_rate": 1.781818181818182e-05,
379
+ "loss": 2.2194,
380
+ "step": 49
381
+ },
382
+ {
383
+ "epoch": 4.55,
384
+ "learning_rate": 1.8181818181818182e-05,
385
+ "loss": 2.2206,
386
+ "step": 50
387
+ },
388
+ {
389
+ "epoch": 4.64,
390
+ "learning_rate": 1.8545454545454545e-05,
391
+ "loss": 2.1524,
392
+ "step": 51
393
+ },
394
+ {
395
+ "epoch": 4.73,
396
+ "learning_rate": 1.890909090909091e-05,
397
+ "loss": 2.1454,
398
+ "step": 52
399
+ },
400
+ {
401
+ "epoch": 4.82,
402
+ "learning_rate": 1.9272727272727272e-05,
403
+ "loss": 2.1673,
404
+ "step": 53
405
+ },
406
+ {
407
+ "epoch": 4.91,
408
+ "learning_rate": 1.9636363636363635e-05,
409
+ "loss": 2.1274,
410
+ "step": 54
411
+ },
412
+ {
413
+ "epoch": 5.0,
414
+ "learning_rate": 2e-05,
415
+ "loss": 2.0958,
416
+ "step": 55
417
+ },
418
+ {
419
+ "epoch": 5.09,
420
+ "learning_rate": 2.0363636363636365e-05,
421
+ "loss": 2.0027,
422
+ "step": 56
423
+ },
424
+ {
425
+ "epoch": 5.18,
426
+ "learning_rate": 2.0727272727272728e-05,
427
+ "loss": 2.0081,
428
+ "step": 57
429
+ },
430
+ {
431
+ "epoch": 5.27,
432
+ "learning_rate": 2.109090909090909e-05,
433
+ "loss": 2.016,
434
+ "step": 58
435
+ },
436
+ {
437
+ "epoch": 5.36,
438
+ "learning_rate": 2.1454545454545455e-05,
439
+ "loss": 1.9957,
440
+ "step": 59
441
+ },
442
+ {
443
+ "epoch": 5.45,
444
+ "learning_rate": 2.1818181818181818e-05,
445
+ "loss": 1.9773,
446
+ "step": 60
447
+ },
448
+ {
449
+ "epoch": 5.45,
450
+ "eval_accuracy_NAT": 0.898956856290447,
451
+ "eval_accuracy_REST": 0.0,
452
+ "eval_accuracy_concrete": 0.24511379243744366,
453
+ "eval_accuracy_concrete tiles": 0.007154650522839846,
454
+ "eval_accuracy_dark bricks": 0.009314499057003194,
455
+ "eval_accuracy_grass": 0.07155464372966304,
456
+ "eval_accuracy_red bricks": 0.0,
457
+ "eval_accuracy_road": 0.7720496210944066,
458
+ "eval_accuracy_rough concrete": 0.0036621652552071412,
459
+ "eval_accuracy_rough red sidewalk": 0.010007443553055993,
460
+ "eval_accuracy_speedway bricks": 0.9787968531076106,
461
+ "eval_accuracy_steel": 0.0,
462
+ "eval_accuracy_tiles": 0.0009587727708533077,
463
+ "eval_accuracy_unlabeled": NaN,
464
+ "eval_iou_NAT": 0.89400043378223,
465
+ "eval_iou_REST": 0.0,
466
+ "eval_iou_concrete": 0.19197997304386832,
467
+ "eval_iou_concrete tiles": 0.0013976657786909726,
468
+ "eval_iou_dark bricks": 0.005338039042682255,
469
+ "eval_iou_grass": 0.07039652848458174,
470
+ "eval_iou_red bricks": 0.0,
471
+ "eval_iou_road": 0.5405843619796408,
472
+ "eval_iou_rough concrete": 0.0013066024253807521,
473
+ "eval_iou_rough red sidewalk": 0.003800927719548077,
474
+ "eval_iou_speedway bricks": 0.6998719868699171,
475
+ "eval_iou_steel": 0.0,
476
+ "eval_iou_tiles": 0.00034965034965034965,
477
+ "eval_iou_unlabeled": 0.0,
478
+ "eval_loss": 1.916090488433838,
479
+ "eval_mean_accuracy": 0.23058225367834848,
480
+ "eval_mean_iou": 0.1720732978197279,
481
+ "eval_overall_accuracy": 0.8010207282172309,
482
+ "eval_runtime": 3.5175,
483
+ "eval_samples_per_second": 10.235,
484
+ "eval_steps_per_second": 0.569,
485
+ "step": 60
486
+ },
487
+ {
488
+ "epoch": 5.55,
489
+ "learning_rate": 2.2181818181818184e-05,
490
+ "loss": 1.9281,
491
+ "step": 61
492
+ },
493
+ {
494
+ "epoch": 5.64,
495
+ "learning_rate": 2.2545454545454548e-05,
496
+ "loss": 1.8834,
497
+ "step": 62
498
+ },
499
+ {
500
+ "epoch": 5.73,
501
+ "learning_rate": 2.290909090909091e-05,
502
+ "loss": 1.8846,
503
+ "step": 63
504
+ },
505
+ {
506
+ "epoch": 5.82,
507
+ "learning_rate": 2.3272727272727274e-05,
508
+ "loss": 1.8532,
509
+ "step": 64
510
+ },
511
+ {
512
+ "epoch": 5.91,
513
+ "learning_rate": 2.3636363636363637e-05,
514
+ "loss": 1.8357,
515
+ "step": 65
516
+ },
517
+ {
518
+ "epoch": 6.0,
519
+ "learning_rate": 2.4e-05,
520
+ "loss": 1.8409,
521
+ "step": 66
522
+ },
523
+ {
524
+ "epoch": 6.09,
525
+ "learning_rate": 2.4363636363636364e-05,
526
+ "loss": 1.7691,
527
+ "step": 67
528
+ },
529
+ {
530
+ "epoch": 6.18,
531
+ "learning_rate": 2.472727272727273e-05,
532
+ "loss": 1.7549,
533
+ "step": 68
534
+ },
535
+ {
536
+ "epoch": 6.27,
537
+ "learning_rate": 2.5090909090909094e-05,
538
+ "loss": 1.8554,
539
+ "step": 69
540
+ },
541
+ {
542
+ "epoch": 6.36,
543
+ "learning_rate": 2.5454545454545454e-05,
544
+ "loss": 1.7217,
545
+ "step": 70
546
+ },
547
+ {
548
+ "epoch": 6.45,
549
+ "learning_rate": 2.581818181818182e-05,
550
+ "loss": 1.7575,
551
+ "step": 71
552
+ },
553
+ {
554
+ "epoch": 6.55,
555
+ "learning_rate": 2.6181818181818187e-05,
556
+ "loss": 1.7587,
557
+ "step": 72
558
+ },
559
+ {
560
+ "epoch": 6.64,
561
+ "learning_rate": 2.6545454545454547e-05,
562
+ "loss": 1.6981,
563
+ "step": 73
564
+ },
565
+ {
566
+ "epoch": 6.73,
567
+ "learning_rate": 2.6909090909090913e-05,
568
+ "loss": 1.6573,
569
+ "step": 74
570
+ },
571
+ {
572
+ "epoch": 6.82,
573
+ "learning_rate": 2.7272727272727273e-05,
574
+ "loss": 1.6428,
575
+ "step": 75
576
+ },
577
+ {
578
+ "epoch": 6.91,
579
+ "learning_rate": 2.7636363636363636e-05,
580
+ "loss": 1.6877,
581
+ "step": 76
582
+ },
583
+ {
584
+ "epoch": 7.0,
585
+ "learning_rate": 2.8000000000000003e-05,
586
+ "loss": 1.7226,
587
+ "step": 77
588
+ },
589
+ {
590
+ "epoch": 7.09,
591
+ "learning_rate": 2.8363636363636363e-05,
592
+ "loss": 1.6332,
593
+ "step": 78
594
+ },
595
+ {
596
+ "epoch": 7.18,
597
+ "learning_rate": 2.872727272727273e-05,
598
+ "loss": 1.6322,
599
+ "step": 79
600
+ },
601
+ {
602
+ "epoch": 7.27,
603
+ "learning_rate": 2.909090909090909e-05,
604
+ "loss": 1.6038,
605
+ "step": 80
606
+ },
607
+ {
608
+ "epoch": 7.27,
609
+ "eval_accuracy_NAT": 0.9470286690715778,
610
+ "eval_accuracy_REST": 0.0,
611
+ "eval_accuracy_concrete": 0.7309205867204601,
612
+ "eval_accuracy_concrete tiles": 0.0014064697609001407,
613
+ "eval_accuracy_dark bricks": 0.0,
614
+ "eval_accuracy_grass": 0.029533372711163616,
615
+ "eval_accuracy_red bricks": 0.015447443181818182,
616
+ "eval_accuracy_road": 0.8218001008480181,
617
+ "eval_accuracy_rough concrete": 0.0,
618
+ "eval_accuracy_rough red sidewalk": 0.0020400849116422684,
619
+ "eval_accuracy_speedway bricks": 0.9799789289994171,
620
+ "eval_accuracy_steel": 0.0,
621
+ "eval_accuracy_tiles": 0.0,
622
+ "eval_accuracy_unlabeled": NaN,
623
+ "eval_iou_NAT": 0.937594640128183,
624
+ "eval_iou_REST": 0.0,
625
+ "eval_iou_concrete": 0.4290493301504721,
626
+ "eval_iou_concrete tiles": 0.0011021395883748233,
627
+ "eval_iou_dark bricks": 0.0,
628
+ "eval_iou_grass": 0.029093488223162548,
629
+ "eval_iou_red bricks": 0.002611828279795857,
630
+ "eval_iou_road": 0.7049759862244951,
631
+ "eval_iou_rough concrete": 0.0,
632
+ "eval_iou_rough red sidewalk": 0.0009831927190593237,
633
+ "eval_iou_speedway bricks": 0.7986861621141007,
634
+ "eval_iou_steel": 0.0,
635
+ "eval_iou_tiles": 0.0,
636
+ "eval_iou_unlabeled": 0.0,
637
+ "eval_loss": 1.4904648065567017,
638
+ "eval_mean_accuracy": 0.27139658893884594,
639
+ "eval_mean_iou": 0.2074354833876888,
640
+ "eval_overall_accuracy": 0.8714875115288628,
641
+ "eval_runtime": 3.8608,
642
+ "eval_samples_per_second": 9.325,
643
+ "eval_steps_per_second": 0.518,
644
+ "step": 80
645
+ }
646
+ ],
647
+ "max_steps": 5500,
648
+ "num_train_epochs": 500,
649
+ "total_flos": 4.478602205410099e+16,
650
+ "trial_name": null,
651
+ "trial_params": null
652
+ }
checkpoint-80/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7935fa87beea0fca0caeec5cfb1e8512ed5be3a31fff1b806e2d5979c9b2318c
3
+ size 4155
config.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b0",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 256,
9
+ "depths": [
10
+ 2,
11
+ 2,
12
+ 2,
13
+ 2
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
+ 32,
26
+ 64,
27
+ 160,
28
+ 256
29
+ ],
30
+ "id2label": {
31
+ "0": "unlabeled",
32
+ "1": "NAT",
33
+ "2": "concrete",
34
+ "3": "grass",
35
+ "4": "speedway bricks",
36
+ "5": "steel",
37
+ "6": "rough concrete",
38
+ "7": "dark bricks",
39
+ "8": "road",
40
+ "9": "rough red sidewalk",
41
+ "10": "tiles",
42
+ "11": "red bricks",
43
+ "12": "concrete tiles",
44
+ "13": "REST"
45
+ },
46
+ "image_size": 224,
47
+ "initializer_range": 0.02,
48
+ "label2id": {
49
+ "NAT": 1,
50
+ "REST": 13,
51
+ "concrete": 2,
52
+ "concrete tiles": 12,
53
+ "dark bricks": 7,
54
+ "grass": 3,
55
+ "red bricks": 11,
56
+ "road": 8,
57
+ "rough concrete": 6,
58
+ "rough red sidewalk": 9,
59
+ "speedway bricks": 4,
60
+ "steel": 5,
61
+ "tiles": 10,
62
+ "unlabeled": 0
63
+ },
64
+ "layer_norm_eps": 1e-06,
65
+ "mlp_ratios": [
66
+ 4,
67
+ 4,
68
+ 4,
69
+ 4
70
+ ],
71
+ "model_type": "segformer",
72
+ "num_attention_heads": [
73
+ 1,
74
+ 2,
75
+ 5,
76
+ 8
77
+ ],
78
+ "num_channels": 3,
79
+ "num_encoder_blocks": 4,
80
+ "patch_sizes": [
81
+ 7,
82
+ 3,
83
+ 3,
84
+ 3
85
+ ],
86
+ "reshape_last_stage": true,
87
+ "semantic_loss_ignore_index": 255,
88
+ "sr_ratios": [
89
+ 8,
90
+ 4,
91
+ 2,
92
+ 1
93
+ ],
94
+ "strides": [
95
+ 4,
96
+ 2,
97
+ 2,
98
+ 2
99
+ ],
100
+ "torch_dtype": "float32",
101
+ "transformers_version": "4.30.2"
102
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4847e5ae2433df8ac0af66bf9328e1c21165e6bc8e3912e8d0c5f6761bf9b3df
3
+ size 14944077
runs/May20_23-57-43_robolidar/events.out.tfevents.1716267470.robolidar.3441566.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38cc726237a950e03a424a5caaab405be8a77f6bb35b98e047e20ca7ff55df49
3
+ size 25815
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7935fa87beea0fca0caeec5cfb1e8512ed5be3a31fff1b806e2d5979c9b2318c
3
+ size 4155