sam1120 commited on
Commit
800fb2b
·
1 Parent(s): 3683d0e

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:e7a3e46b2fceab7e4c4bbaaeff9b8aee0d52f0a6ac4e4d8134e791955b3cea3e
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:7bc52a7477531b0c11554f19c984fa7d75f249562a682b88850f09f22686232e
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.4965529441833496,
3
+ "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0-512sq/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.6331,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.18,
18
+ "learning_rate": 7.272727272727273e-07,
19
+ "loss": 2.6235,
20
+ "step": 2
21
+ },
22
+ {
23
+ "epoch": 0.27,
24
+ "learning_rate": 1.090909090909091e-06,
25
+ "loss": 2.6281,
26
+ "step": 3
27
+ },
28
+ {
29
+ "epoch": 0.36,
30
+ "learning_rate": 1.4545454545454546e-06,
31
+ "loss": 2.6298,
32
+ "step": 4
33
+ },
34
+ {
35
+ "epoch": 0.45,
36
+ "learning_rate": 1.818181818181818e-06,
37
+ "loss": 2.6104,
38
+ "step": 5
39
+ },
40
+ {
41
+ "epoch": 0.55,
42
+ "learning_rate": 2.181818181818182e-06,
43
+ "loss": 2.6114,
44
+ "step": 6
45
+ },
46
+ {
47
+ "epoch": 0.64,
48
+ "learning_rate": 2.5454545454545456e-06,
49
+ "loss": 2.6423,
50
+ "step": 7
51
+ },
52
+ {
53
+ "epoch": 0.73,
54
+ "learning_rate": 2.9090909090909093e-06,
55
+ "loss": 2.6415,
56
+ "step": 8
57
+ },
58
+ {
59
+ "epoch": 0.82,
60
+ "learning_rate": 3.2727272727272733e-06,
61
+ "loss": 2.6322,
62
+ "step": 9
63
+ },
64
+ {
65
+ "epoch": 0.91,
66
+ "learning_rate": 3.636363636363636e-06,
67
+ "loss": 2.6275,
68
+ "step": 10
69
+ },
70
+ {
71
+ "epoch": 1.0,
72
+ "learning_rate": 4.000000000000001e-06,
73
+ "loss": 2.6138,
74
+ "step": 11
75
+ },
76
+ {
77
+ "epoch": 1.09,
78
+ "learning_rate": 4.363636363636364e-06,
79
+ "loss": 2.6339,
80
+ "step": 12
81
+ },
82
+ {
83
+ "epoch": 1.18,
84
+ "learning_rate": 4.727272727272727e-06,
85
+ "loss": 2.6064,
86
+ "step": 13
87
+ },
88
+ {
89
+ "epoch": 1.27,
90
+ "learning_rate": 5.090909090909091e-06,
91
+ "loss": 2.5849,
92
+ "step": 14
93
+ },
94
+ {
95
+ "epoch": 1.36,
96
+ "learning_rate": 5.4545454545454545e-06,
97
+ "loss": 2.594,
98
+ "step": 15
99
+ },
100
+ {
101
+ "epoch": 1.45,
102
+ "learning_rate": 5.8181818181818185e-06,
103
+ "loss": 2.5859,
104
+ "step": 16
105
+ },
106
+ {
107
+ "epoch": 1.55,
108
+ "learning_rate": 6.181818181818183e-06,
109
+ "loss": 2.5824,
110
+ "step": 17
111
+ },
112
+ {
113
+ "epoch": 1.64,
114
+ "learning_rate": 6.545454545454547e-06,
115
+ "loss": 2.5595,
116
+ "step": 18
117
+ },
118
+ {
119
+ "epoch": 1.73,
120
+ "learning_rate": 6.909090909090909e-06,
121
+ "loss": 2.542,
122
+ "step": 19
123
+ },
124
+ {
125
+ "epoch": 1.82,
126
+ "learning_rate": 7.272727272727272e-06,
127
+ "loss": 2.5605,
128
+ "step": 20
129
+ },
130
+ {
131
+ "epoch": 1.82,
132
+ "eval_accuracy_NAT": 0.06958914416267946,
133
+ "eval_accuracy_REST": 0.0,
134
+ "eval_accuracy_concrete": 0.012212165758187047,
135
+ "eval_accuracy_concrete tiles": 0.055341527548462055,
136
+ "eval_accuracy_dark bricks": 0.013009506947384628,
137
+ "eval_accuracy_grass": 0.21269969977316971,
138
+ "eval_accuracy_red bricks": 0.0,
139
+ "eval_accuracy_road": 0.0030363234246729392,
140
+ "eval_accuracy_rough concrete": 0.014248111696040284,
141
+ "eval_accuracy_rough red sidewalk": 0.06566867918286329,
142
+ "eval_accuracy_speedway bricks": 0.8828568221889945,
143
+ "eval_accuracy_steel": 0.01111404260932656,
144
+ "eval_accuracy_tiles": 0.014518559101492946,
145
+ "eval_accuracy_unlabeled": NaN,
146
+ "eval_iou_NAT": 0.0694410794425303,
147
+ "eval_iou_REST": 0.0,
148
+ "eval_iou_concrete": 0.011557707296143877,
149
+ "eval_iou_concrete tiles": 0.002648498842562108,
150
+ "eval_iou_dark bricks": 0.0007563670215027535,
151
+ "eval_iou_grass": 0.04560076309737809,
152
+ "eval_iou_red bricks": 0.0,
153
+ "eval_iou_road": 0.002986320724423608,
154
+ "eval_iou_rough concrete": 0.000721262470019813,
155
+ "eval_iou_rough red sidewalk": 0.0012369258187233804,
156
+ "eval_iou_speedway bricks": 0.44652589366643575,
157
+ "eval_iou_steel": 0.0028144253916289634,
158
+ "eval_iou_tiles": 0.0003408135476600417,
159
+ "eval_iou_unlabeled": 0.0,
160
+ "eval_loss": 2.592318058013916,
161
+ "eval_mean_accuracy": 0.10417650633794412,
162
+ "eval_mean_iou": 0.04175928980850062,
163
+ "eval_overall_accuracy": 0.2051790025499132,
164
+ "eval_runtime": 3.505,
165
+ "eval_samples_per_second": 10.271,
166
+ "eval_steps_per_second": 0.571,
167
+ "step": 20
168
+ },
169
+ {
170
+ "epoch": 1.91,
171
+ "learning_rate": 7.636363636363638e-06,
172
+ "loss": 2.5567,
173
+ "step": 21
174
+ },
175
+ {
176
+ "epoch": 2.0,
177
+ "learning_rate": 8.000000000000001e-06,
178
+ "loss": 2.527,
179
+ "step": 22
180
+ },
181
+ {
182
+ "epoch": 2.09,
183
+ "learning_rate": 8.363636363636365e-06,
184
+ "loss": 2.5298,
185
+ "step": 23
186
+ },
187
+ {
188
+ "epoch": 2.18,
189
+ "learning_rate": 8.727272727272728e-06,
190
+ "loss": 2.5135,
191
+ "step": 24
192
+ },
193
+ {
194
+ "epoch": 2.27,
195
+ "learning_rate": 9.090909090909091e-06,
196
+ "loss": 2.4713,
197
+ "step": 25
198
+ },
199
+ {
200
+ "epoch": 2.36,
201
+ "learning_rate": 9.454545454545454e-06,
202
+ "loss": 2.4836,
203
+ "step": 26
204
+ },
205
+ {
206
+ "epoch": 2.45,
207
+ "learning_rate": 9.818181818181818e-06,
208
+ "loss": 2.4907,
209
+ "step": 27
210
+ },
211
+ {
212
+ "epoch": 2.55,
213
+ "learning_rate": 1.0181818181818182e-05,
214
+ "loss": 2.4879,
215
+ "step": 28
216
+ },
217
+ {
218
+ "epoch": 2.64,
219
+ "learning_rate": 1.0545454545454546e-05,
220
+ "loss": 2.402,
221
+ "step": 29
222
+ },
223
+ {
224
+ "epoch": 2.73,
225
+ "learning_rate": 1.0909090909090909e-05,
226
+ "loss": 2.4356,
227
+ "step": 30
228
+ },
229
+ {
230
+ "epoch": 2.82,
231
+ "learning_rate": 1.1272727272727274e-05,
232
+ "loss": 2.4624,
233
+ "step": 31
234
+ },
235
+ {
236
+ "epoch": 2.91,
237
+ "learning_rate": 1.1636363636363637e-05,
238
+ "loss": 2.4085,
239
+ "step": 32
240
+ },
241
+ {
242
+ "epoch": 3.0,
243
+ "learning_rate": 1.2e-05,
244
+ "loss": 2.377,
245
+ "step": 33
246
+ },
247
+ {
248
+ "epoch": 3.09,
249
+ "learning_rate": 1.2363636363636365e-05,
250
+ "loss": 2.3522,
251
+ "step": 34
252
+ },
253
+ {
254
+ "epoch": 3.18,
255
+ "learning_rate": 1.2727272727272727e-05,
256
+ "loss": 2.3316,
257
+ "step": 35
258
+ },
259
+ {
260
+ "epoch": 3.27,
261
+ "learning_rate": 1.3090909090909093e-05,
262
+ "loss": 2.3399,
263
+ "step": 36
264
+ },
265
+ {
266
+ "epoch": 3.36,
267
+ "learning_rate": 1.3454545454545457e-05,
268
+ "loss": 2.339,
269
+ "step": 37
270
+ },
271
+ {
272
+ "epoch": 3.45,
273
+ "learning_rate": 1.3818181818181818e-05,
274
+ "loss": 2.2832,
275
+ "step": 38
276
+ },
277
+ {
278
+ "epoch": 3.55,
279
+ "learning_rate": 1.4181818181818181e-05,
280
+ "loss": 2.2694,
281
+ "step": 39
282
+ },
283
+ {
284
+ "epoch": 3.64,
285
+ "learning_rate": 1.4545454545454545e-05,
286
+ "loss": 2.2612,
287
+ "step": 40
288
+ },
289
+ {
290
+ "epoch": 3.64,
291
+ "eval_accuracy_NAT": 0.6487876396698812,
292
+ "eval_accuracy_REST": 0.0,
293
+ "eval_accuracy_concrete": 0.22674658783638782,
294
+ "eval_accuracy_concrete tiles": 0.11138628997737418,
295
+ "eval_accuracy_dark bricks": 0.012932527616335013,
296
+ "eval_accuracy_grass": 0.33075979407025796,
297
+ "eval_accuracy_red bricks": 0.0,
298
+ "eval_accuracy_road": 0.10575498768420785,
299
+ "eval_accuracy_rough concrete": 0.04217212176699474,
300
+ "eval_accuracy_rough red sidewalk": 0.10583629697019822,
301
+ "eval_accuracy_speedway bricks": 0.9638444076767955,
302
+ "eval_accuracy_steel": 0.014024470681353473,
303
+ "eval_accuracy_tiles": 0.0,
304
+ "eval_accuracy_unlabeled": NaN,
305
+ "eval_iou_NAT": 0.6444485535759669,
306
+ "eval_iou_REST": 0.0,
307
+ "eval_iou_concrete": 0.18658421370958841,
308
+ "eval_iou_concrete tiles": 0.010319440942941394,
309
+ "eval_iou_dark bricks": 0.0009821746984782138,
310
+ "eval_iou_grass": 0.12849579967711758,
311
+ "eval_iou_red bricks": 0.0,
312
+ "eval_iou_road": 0.10435270254146284,
313
+ "eval_iou_rough concrete": 0.0032174975988823887,
314
+ "eval_iou_rough red sidewalk": 0.003593977022513144,
315
+ "eval_iou_speedway bricks": 0.6287866411845635,
316
+ "eval_iou_steel": 0.008100501459614167,
317
+ "eval_iou_tiles": 0.0,
318
+ "eval_iou_unlabeled": 0.0,
319
+ "eval_loss": 2.284590482711792,
320
+ "eval_mean_accuracy": 0.19709577876536816,
321
+ "eval_mean_iou": 0.12277725017222348,
322
+ "eval_overall_accuracy": 0.5971706178453233,
323
+ "eval_runtime": 3.4424,
324
+ "eval_samples_per_second": 10.458,
325
+ "eval_steps_per_second": 0.581,
326
+ "step": 40
327
+ },
328
+ {
329
+ "epoch": 3.73,
330
+ "learning_rate": 1.4909090909090908e-05,
331
+ "loss": 2.2556,
332
+ "step": 41
333
+ },
334
+ {
335
+ "epoch": 3.82,
336
+ "learning_rate": 1.5272727272727276e-05,
337
+ "loss": 2.256,
338
+ "step": 42
339
+ },
340
+ {
341
+ "epoch": 3.91,
342
+ "learning_rate": 1.563636363636364e-05,
343
+ "loss": 2.1922,
344
+ "step": 43
345
+ },
346
+ {
347
+ "epoch": 4.0,
348
+ "learning_rate": 1.6000000000000003e-05,
349
+ "loss": 2.2101,
350
+ "step": 44
351
+ },
352
+ {
353
+ "epoch": 4.09,
354
+ "learning_rate": 1.6363636363636366e-05,
355
+ "loss": 2.1992,
356
+ "step": 45
357
+ },
358
+ {
359
+ "epoch": 4.18,
360
+ "learning_rate": 1.672727272727273e-05,
361
+ "loss": 2.1483,
362
+ "step": 46
363
+ },
364
+ {
365
+ "epoch": 4.27,
366
+ "learning_rate": 1.7090909090909092e-05,
367
+ "loss": 2.0835,
368
+ "step": 47
369
+ },
370
+ {
371
+ "epoch": 4.36,
372
+ "learning_rate": 1.7454545454545456e-05,
373
+ "loss": 2.0778,
374
+ "step": 48
375
+ },
376
+ {
377
+ "epoch": 4.45,
378
+ "learning_rate": 1.781818181818182e-05,
379
+ "loss": 2.0845,
380
+ "step": 49
381
+ },
382
+ {
383
+ "epoch": 4.55,
384
+ "learning_rate": 1.8181818181818182e-05,
385
+ "loss": 2.0901,
386
+ "step": 50
387
+ },
388
+ {
389
+ "epoch": 4.64,
390
+ "learning_rate": 1.8545454545454545e-05,
391
+ "loss": 2.0222,
392
+ "step": 51
393
+ },
394
+ {
395
+ "epoch": 4.73,
396
+ "learning_rate": 1.890909090909091e-05,
397
+ "loss": 2.015,
398
+ "step": 52
399
+ },
400
+ {
401
+ "epoch": 4.82,
402
+ "learning_rate": 1.9272727272727272e-05,
403
+ "loss": 2.0583,
404
+ "step": 53
405
+ },
406
+ {
407
+ "epoch": 4.91,
408
+ "learning_rate": 1.9636363636363635e-05,
409
+ "loss": 1.9883,
410
+ "step": 54
411
+ },
412
+ {
413
+ "epoch": 5.0,
414
+ "learning_rate": 2e-05,
415
+ "loss": 1.9745,
416
+ "step": 55
417
+ },
418
+ {
419
+ "epoch": 5.09,
420
+ "learning_rate": 2.0363636363636365e-05,
421
+ "loss": 1.8931,
422
+ "step": 56
423
+ },
424
+ {
425
+ "epoch": 5.18,
426
+ "learning_rate": 2.0727272727272728e-05,
427
+ "loss": 1.9003,
428
+ "step": 57
429
+ },
430
+ {
431
+ "epoch": 5.27,
432
+ "learning_rate": 2.109090909090909e-05,
433
+ "loss": 1.9138,
434
+ "step": 58
435
+ },
436
+ {
437
+ "epoch": 5.36,
438
+ "learning_rate": 2.1454545454545455e-05,
439
+ "loss": 1.9265,
440
+ "step": 59
441
+ },
442
+ {
443
+ "epoch": 5.45,
444
+ "learning_rate": 2.1818181818181818e-05,
445
+ "loss": 1.9031,
446
+ "step": 60
447
+ },
448
+ {
449
+ "epoch": 5.45,
450
+ "eval_accuracy_NAT": 0.926172676474055,
451
+ "eval_accuracy_REST": 0.0,
452
+ "eval_accuracy_concrete": 0.5041069359199966,
453
+ "eval_accuracy_concrete tiles": 0.03005564728184431,
454
+ "eval_accuracy_dark bricks": 0.015626804203071474,
455
+ "eval_accuracy_grass": 0.179513975052164,
456
+ "eval_accuracy_red bricks": 0.0,
457
+ "eval_accuracy_road": 0.5167807958287132,
458
+ "eval_accuracy_rough concrete": 0.0,
459
+ "eval_accuracy_rough red sidewalk": 0.045460811071595955,
460
+ "eval_accuracy_speedway bricks": 0.9734023873226434,
461
+ "eval_accuracy_steel": 0.0017231712947694744,
462
+ "eval_accuracy_tiles": 0.0,
463
+ "eval_accuracy_unlabeled": NaN,
464
+ "eval_iou_NAT": 0.91595721698184,
465
+ "eval_iou_REST": 0.0,
466
+ "eval_iou_concrete": 0.2939118888370162,
467
+ "eval_iou_concrete tiles": 0.009215596201261871,
468
+ "eval_iou_dark bricks": 0.005187769131495892,
469
+ "eval_iou_grass": 0.13645988591590083,
470
+ "eval_iou_red bricks": 0.0,
471
+ "eval_iou_road": 0.47866524647780845,
472
+ "eval_iou_rough concrete": 0.0,
473
+ "eval_iou_rough red sidewalk": 0.01148048873881714,
474
+ "eval_iou_speedway bricks": 0.6862571181348763,
475
+ "eval_iou_steel": 0.0016102439249888285,
476
+ "eval_iou_tiles": 0.0,
477
+ "eval_iou_unlabeled": 0.0,
478
+ "eval_loss": 1.8314470052719116,
479
+ "eval_mean_accuracy": 0.2456033234191426,
480
+ "eval_mean_iou": 0.18133896102457184,
481
+ "eval_overall_accuracy": 0.8180291917588975,
482
+ "eval_runtime": 3.3468,
483
+ "eval_samples_per_second": 10.757,
484
+ "eval_steps_per_second": 0.598,
485
+ "step": 60
486
+ },
487
+ {
488
+ "epoch": 5.55,
489
+ "learning_rate": 2.2181818181818184e-05,
490
+ "loss": 1.8533,
491
+ "step": 61
492
+ },
493
+ {
494
+ "epoch": 5.64,
495
+ "learning_rate": 2.2545454545454548e-05,
496
+ "loss": 1.8064,
497
+ "step": 62
498
+ },
499
+ {
500
+ "epoch": 5.73,
501
+ "learning_rate": 2.290909090909091e-05,
502
+ "loss": 1.8108,
503
+ "step": 63
504
+ },
505
+ {
506
+ "epoch": 5.82,
507
+ "learning_rate": 2.3272727272727274e-05,
508
+ "loss": 1.7959,
509
+ "step": 64
510
+ },
511
+ {
512
+ "epoch": 5.91,
513
+ "learning_rate": 2.3636363636363637e-05,
514
+ "loss": 1.7832,
515
+ "step": 65
516
+ },
517
+ {
518
+ "epoch": 6.0,
519
+ "learning_rate": 2.4e-05,
520
+ "loss": 1.7869,
521
+ "step": 66
522
+ },
523
+ {
524
+ "epoch": 6.09,
525
+ "learning_rate": 2.4363636363636364e-05,
526
+ "loss": 1.7181,
527
+ "step": 67
528
+ },
529
+ {
530
+ "epoch": 6.18,
531
+ "learning_rate": 2.472727272727273e-05,
532
+ "loss": 1.6959,
533
+ "step": 68
534
+ },
535
+ {
536
+ "epoch": 6.27,
537
+ "learning_rate": 2.5090909090909094e-05,
538
+ "loss": 1.8088,
539
+ "step": 69
540
+ },
541
+ {
542
+ "epoch": 6.36,
543
+ "learning_rate": 2.5454545454545454e-05,
544
+ "loss": 1.6729,
545
+ "step": 70
546
+ },
547
+ {
548
+ "epoch": 6.45,
549
+ "learning_rate": 2.581818181818182e-05,
550
+ "loss": 1.7154,
551
+ "step": 71
552
+ },
553
+ {
554
+ "epoch": 6.55,
555
+ "learning_rate": 2.6181818181818187e-05,
556
+ "loss": 1.7189,
557
+ "step": 72
558
+ },
559
+ {
560
+ "epoch": 6.64,
561
+ "learning_rate": 2.6545454545454547e-05,
562
+ "loss": 1.659,
563
+ "step": 73
564
+ },
565
+ {
566
+ "epoch": 6.73,
567
+ "learning_rate": 2.6909090909090913e-05,
568
+ "loss": 1.6129,
569
+ "step": 74
570
+ },
571
+ {
572
+ "epoch": 6.82,
573
+ "learning_rate": 2.7272727272727273e-05,
574
+ "loss": 1.6158,
575
+ "step": 75
576
+ },
577
+ {
578
+ "epoch": 6.91,
579
+ "learning_rate": 2.7636363636363636e-05,
580
+ "loss": 1.6459,
581
+ "step": 76
582
+ },
583
+ {
584
+ "epoch": 7.0,
585
+ "learning_rate": 2.8000000000000003e-05,
586
+ "loss": 1.7042,
587
+ "step": 77
588
+ },
589
+ {
590
+ "epoch": 7.09,
591
+ "learning_rate": 2.8363636363636363e-05,
592
+ "loss": 1.6068,
593
+ "step": 78
594
+ },
595
+ {
596
+ "epoch": 7.18,
597
+ "learning_rate": 2.872727272727273e-05,
598
+ "loss": 1.5862,
599
+ "step": 79
600
+ },
601
+ {
602
+ "epoch": 7.27,
603
+ "learning_rate": 2.909090909090909e-05,
604
+ "loss": 1.5736,
605
+ "step": 80
606
+ },
607
+ {
608
+ "epoch": 7.27,
609
+ "eval_accuracy_NAT": 0.9479160417356296,
610
+ "eval_accuracy_REST": 0.0,
611
+ "eval_accuracy_concrete": 0.6931882806129018,
612
+ "eval_accuracy_concrete tiles": 0.001651073197578426,
613
+ "eval_accuracy_dark bricks": 0.0016550556175666834,
614
+ "eval_accuracy_grass": 0.17391486699497058,
615
+ "eval_accuracy_red bricks": 0.0,
616
+ "eval_accuracy_road": 0.7966799961789001,
617
+ "eval_accuracy_rough concrete": 0.0,
618
+ "eval_accuracy_rough red sidewalk": 0.012212940754831418,
619
+ "eval_accuracy_speedway bricks": 0.9742284526089543,
620
+ "eval_accuracy_steel": 0.0,
621
+ "eval_accuracy_tiles": 0.0,
622
+ "eval_accuracy_unlabeled": NaN,
623
+ "eval_iou_NAT": 0.9379327936554384,
624
+ "eval_iou_REST": 0.0,
625
+ "eval_iou_concrete": 0.3900379457951712,
626
+ "eval_iou_concrete tiles": 0.0012524352908433063,
627
+ "eval_iou_dark bricks": 0.0015475976246175993,
628
+ "eval_iou_grass": 0.1382072497701638,
629
+ "eval_iou_red bricks": 0.0,
630
+ "eval_iou_road": 0.6778795352287588,
631
+ "eval_iou_rough concrete": 0.0,
632
+ "eval_iou_rough red sidewalk": 0.009622276765351115,
633
+ "eval_iou_speedway bricks": 0.8110189073519661,
634
+ "eval_iou_steel": 0.0,
635
+ "eval_iou_tiles": 0.0,
636
+ "eval_iou_unlabeled": 0.0,
637
+ "eval_loss": 1.4965529441833496,
638
+ "eval_mean_accuracy": 0.27703436213087174,
639
+ "eval_mean_iou": 0.21196419582016507,
640
+ "eval_overall_accuracy": 0.8702409532335069,
641
+ "eval_runtime": 3.4799,
642
+ "eval_samples_per_second": 10.345,
643
+ "eval_steps_per_second": 0.575,
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:dbffb6c5228fd97c01803fa2eb23fd31d3f0cc83956e63f26012ac5061e17a08
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:7bc52a7477531b0c11554f19c984fa7d75f249562a682b88850f09f22686232e
3
+ size 14944077
runs/May20_22-12-23_robolidar/events.out.tfevents.1716261149.robolidar.3147080.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff5c2115bd79a1e35acf2ba8b1cff093fd6f14ff45c12472fe30a86a5d4c9d28
3
+ size 25815
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbffb6c5228fd97c01803fa2eb23fd31d3f0cc83956e63f26012ac5061e17a08
3
+ size 4155