sam1120 commited on
Commit
3238333
1 Parent(s): 5546ec0

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:cc31879e5c37d87089a752e4132673249c0958d03a2b33331c9df5f0f49d9207
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:2ad22c770dbf928f9aeb0e0a6371a0cf1f7b944d5d3d3ae5fa29dc5244b66c39
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.2371957302093506,
3
+ "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/safety-utcustom-terrain-b0/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.6936,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.18,
18
+ "learning_rate": 7.272727272727273e-07,
19
+ "loss": 2.6918,
20
+ "step": 2
21
+ },
22
+ {
23
+ "epoch": 0.27,
24
+ "learning_rate": 1.090909090909091e-06,
25
+ "loss": 2.6894,
26
+ "step": 3
27
+ },
28
+ {
29
+ "epoch": 0.36,
30
+ "learning_rate": 1.4545454545454546e-06,
31
+ "loss": 2.6788,
32
+ "step": 4
33
+ },
34
+ {
35
+ "epoch": 0.45,
36
+ "learning_rate": 1.818181818181818e-06,
37
+ "loss": 2.6793,
38
+ "step": 5
39
+ },
40
+ {
41
+ "epoch": 0.55,
42
+ "learning_rate": 2.181818181818182e-06,
43
+ "loss": 2.6842,
44
+ "step": 6
45
+ },
46
+ {
47
+ "epoch": 0.64,
48
+ "learning_rate": 2.5454545454545456e-06,
49
+ "loss": 2.6901,
50
+ "step": 7
51
+ },
52
+ {
53
+ "epoch": 0.73,
54
+ "learning_rate": 2.9090909090909093e-06,
55
+ "loss": 2.6793,
56
+ "step": 8
57
+ },
58
+ {
59
+ "epoch": 0.82,
60
+ "learning_rate": 3.2727272727272733e-06,
61
+ "loss": 2.6703,
62
+ "step": 9
63
+ },
64
+ {
65
+ "epoch": 0.91,
66
+ "learning_rate": 3.636363636363636e-06,
67
+ "loss": 2.6668,
68
+ "step": 10
69
+ },
70
+ {
71
+ "epoch": 1.0,
72
+ "learning_rate": 4.000000000000001e-06,
73
+ "loss": 2.6638,
74
+ "step": 11
75
+ },
76
+ {
77
+ "epoch": 1.09,
78
+ "learning_rate": 4.363636363636364e-06,
79
+ "loss": 2.6632,
80
+ "step": 12
81
+ },
82
+ {
83
+ "epoch": 1.18,
84
+ "learning_rate": 4.727272727272727e-06,
85
+ "loss": 2.6546,
86
+ "step": 13
87
+ },
88
+ {
89
+ "epoch": 1.27,
90
+ "learning_rate": 5.090909090909091e-06,
91
+ "loss": 2.6434,
92
+ "step": 14
93
+ },
94
+ {
95
+ "epoch": 1.36,
96
+ "learning_rate": 5.4545454545454545e-06,
97
+ "loss": 2.6392,
98
+ "step": 15
99
+ },
100
+ {
101
+ "epoch": 1.45,
102
+ "learning_rate": 5.8181818181818185e-06,
103
+ "loss": 2.6424,
104
+ "step": 16
105
+ },
106
+ {
107
+ "epoch": 1.55,
108
+ "learning_rate": 6.181818181818183e-06,
109
+ "loss": 2.6164,
110
+ "step": 17
111
+ },
112
+ {
113
+ "epoch": 1.64,
114
+ "learning_rate": 6.545454545454547e-06,
115
+ "loss": 2.6205,
116
+ "step": 18
117
+ },
118
+ {
119
+ "epoch": 1.73,
120
+ "learning_rate": 6.909090909090909e-06,
121
+ "loss": 2.6095,
122
+ "step": 19
123
+ },
124
+ {
125
+ "epoch": 1.82,
126
+ "learning_rate": 7.272727272727272e-06,
127
+ "loss": 2.6108,
128
+ "step": 20
129
+ },
130
+ {
131
+ "epoch": 1.82,
132
+ "eval_accuracy_NAT": 0.09511473910290841,
133
+ "eval_accuracy_REST": 0.25,
134
+ "eval_accuracy_concrete": 0.01301558006781407,
135
+ "eval_accuracy_concrete tiles": 0.0,
136
+ "eval_accuracy_dark bricks": 0.6411223586467033,
137
+ "eval_accuracy_grass": 0.16874215792839992,
138
+ "eval_accuracy_red bricks": 0.0,
139
+ "eval_accuracy_road": 0.005832014288011783,
140
+ "eval_accuracy_rough concrete": 0.0,
141
+ "eval_accuracy_rough red sidewalk": 0.1063325338405977,
142
+ "eval_accuracy_speedway bricks": 0.06347065688180568,
143
+ "eval_accuracy_steel": 0.013958511971505837,
144
+ "eval_accuracy_tiles": 0.0,
145
+ "eval_accuracy_unlabeled": NaN,
146
+ "eval_iou_NAT": 0.09360158321654327,
147
+ "eval_iou_REST": 7.769355796094811e-07,
148
+ "eval_iou_concrete": 0.011760286008604496,
149
+ "eval_iou_concrete tiles": 0.0,
150
+ "eval_iou_dark bricks": 0.005873411187090554,
151
+ "eval_iou_grass": 0.020184891600275597,
152
+ "eval_iou_red bricks": 0.0,
153
+ "eval_iou_road": 0.005779806508903394,
154
+ "eval_iou_rough concrete": 0.0,
155
+ "eval_iou_rough red sidewalk": 0.017025014456033796,
156
+ "eval_iou_speedway bricks": 0.06224845291701409,
157
+ "eval_iou_steel": 0.006497743252786392,
158
+ "eval_iou_tiles": 0.0,
159
+ "eval_iou_unlabeled": 0.0,
160
+ "eval_loss": 2.605102300643921,
161
+ "eval_mean_accuracy": 0.10442988867136511,
162
+ "eval_mean_iou": 0.015926569005916515,
163
+ "eval_overall_accuracy": 0.07727368672688802,
164
+ "eval_runtime": 3.9824,
165
+ "eval_samples_per_second": 9.04,
166
+ "eval_steps_per_second": 0.502,
167
+ "step": 20
168
+ },
169
+ {
170
+ "epoch": 1.91,
171
+ "learning_rate": 7.636363636363638e-06,
172
+ "loss": 2.5966,
173
+ "step": 21
174
+ },
175
+ {
176
+ "epoch": 2.0,
177
+ "learning_rate": 8.000000000000001e-06,
178
+ "loss": 2.5833,
179
+ "step": 22
180
+ },
181
+ {
182
+ "epoch": 2.09,
183
+ "learning_rate": 8.363636363636365e-06,
184
+ "loss": 2.5726,
185
+ "step": 23
186
+ },
187
+ {
188
+ "epoch": 2.18,
189
+ "learning_rate": 8.727272727272728e-06,
190
+ "loss": 2.564,
191
+ "step": 24
192
+ },
193
+ {
194
+ "epoch": 2.27,
195
+ "learning_rate": 9.090909090909091e-06,
196
+ "loss": 2.5487,
197
+ "step": 25
198
+ },
199
+ {
200
+ "epoch": 2.36,
201
+ "learning_rate": 9.454545454545454e-06,
202
+ "loss": 2.5331,
203
+ "step": 26
204
+ },
205
+ {
206
+ "epoch": 2.45,
207
+ "learning_rate": 9.818181818181818e-06,
208
+ "loss": 2.5263,
209
+ "step": 27
210
+ },
211
+ {
212
+ "epoch": 2.55,
213
+ "learning_rate": 1.0181818181818182e-05,
214
+ "loss": 2.5167,
215
+ "step": 28
216
+ },
217
+ {
218
+ "epoch": 2.64,
219
+ "learning_rate": 1.0545454545454546e-05,
220
+ "loss": 2.4959,
221
+ "step": 29
222
+ },
223
+ {
224
+ "epoch": 2.73,
225
+ "learning_rate": 1.0909090909090909e-05,
226
+ "loss": 2.4863,
227
+ "step": 30
228
+ },
229
+ {
230
+ "epoch": 2.82,
231
+ "learning_rate": 1.1272727272727274e-05,
232
+ "loss": 2.4883,
233
+ "step": 31
234
+ },
235
+ {
236
+ "epoch": 2.91,
237
+ "learning_rate": 1.1636363636363637e-05,
238
+ "loss": 2.4681,
239
+ "step": 32
240
+ },
241
+ {
242
+ "epoch": 3.0,
243
+ "learning_rate": 1.2e-05,
244
+ "loss": 2.4457,
245
+ "step": 33
246
+ },
247
+ {
248
+ "epoch": 3.09,
249
+ "learning_rate": 1.2363636363636365e-05,
250
+ "loss": 2.4198,
251
+ "step": 34
252
+ },
253
+ {
254
+ "epoch": 3.18,
255
+ "learning_rate": 1.2727272727272727e-05,
256
+ "loss": 2.4107,
257
+ "step": 35
258
+ },
259
+ {
260
+ "epoch": 3.27,
261
+ "learning_rate": 1.3090909090909093e-05,
262
+ "loss": 2.3918,
263
+ "step": 36
264
+ },
265
+ {
266
+ "epoch": 3.36,
267
+ "learning_rate": 1.3454545454545457e-05,
268
+ "loss": 2.3891,
269
+ "step": 37
270
+ },
271
+ {
272
+ "epoch": 3.45,
273
+ "learning_rate": 1.3818181818181818e-05,
274
+ "loss": 2.3568,
275
+ "step": 38
276
+ },
277
+ {
278
+ "epoch": 3.55,
279
+ "learning_rate": 1.4181818181818181e-05,
280
+ "loss": 2.3432,
281
+ "step": 39
282
+ },
283
+ {
284
+ "epoch": 3.64,
285
+ "learning_rate": 1.4545454545454545e-05,
286
+ "loss": 2.3421,
287
+ "step": 40
288
+ },
289
+ {
290
+ "epoch": 3.64,
291
+ "eval_accuracy_NAT": 0.6849137118750895,
292
+ "eval_accuracy_REST": 0.125,
293
+ "eval_accuracy_concrete": 0.05468314090733508,
294
+ "eval_accuracy_concrete tiles": 0.0,
295
+ "eval_accuracy_dark bricks": 0.5916631384473269,
296
+ "eval_accuracy_grass": 0.18795108294858398,
297
+ "eval_accuracy_red bricks": 0.0,
298
+ "eval_accuracy_road": 0.1476613296218683,
299
+ "eval_accuracy_rough concrete": 0.0,
300
+ "eval_accuracy_rough red sidewalk": 0.02789953960245913,
301
+ "eval_accuracy_speedway bricks": 0.7465795828521156,
302
+ "eval_accuracy_steel": 0.02932689136600488,
303
+ "eval_accuracy_tiles": 0.0,
304
+ "eval_accuracy_unlabeled": NaN,
305
+ "eval_iou_NAT": 0.6762243132368754,
306
+ "eval_iou_REST": 8.163178676471188e-07,
307
+ "eval_iou_concrete": 0.04730494108637764,
308
+ "eval_iou_concrete tiles": 0.0,
309
+ "eval_iou_dark bricks": 0.009574402487401365,
310
+ "eval_iou_grass": 0.0756298502760737,
311
+ "eval_iou_red bricks": 0.0,
312
+ "eval_iou_road": 0.14108906887268777,
313
+ "eval_iou_rough concrete": 0.0,
314
+ "eval_iou_rough red sidewalk": 0.005906833131964022,
315
+ "eval_iou_speedway bricks": 0.6970837094053498,
316
+ "eval_iou_steel": 0.012883666683569613,
317
+ "eval_iou_tiles": 0.0,
318
+ "eval_iou_unlabeled": 0.0,
319
+ "eval_loss": 2.2775866985321045,
320
+ "eval_mean_accuracy": 0.1996675705862141,
321
+ "eval_mean_iou": 0.11897840010701195,
322
+ "eval_overall_accuracy": 0.5674870808919271,
323
+ "eval_runtime": 3.5096,
324
+ "eval_samples_per_second": 10.258,
325
+ "eval_steps_per_second": 0.57,
326
+ "step": 40
327
+ },
328
+ {
329
+ "epoch": 3.73,
330
+ "learning_rate": 1.4909090909090908e-05,
331
+ "loss": 2.3287,
332
+ "step": 41
333
+ },
334
+ {
335
+ "epoch": 3.82,
336
+ "learning_rate": 1.5272727272727276e-05,
337
+ "loss": 2.308,
338
+ "step": 42
339
+ },
340
+ {
341
+ "epoch": 3.91,
342
+ "learning_rate": 1.563636363636364e-05,
343
+ "loss": 2.2771,
344
+ "step": 43
345
+ },
346
+ {
347
+ "epoch": 4.0,
348
+ "learning_rate": 1.6000000000000003e-05,
349
+ "loss": 2.2752,
350
+ "step": 44
351
+ },
352
+ {
353
+ "epoch": 4.09,
354
+ "learning_rate": 1.6363636363636366e-05,
355
+ "loss": 2.2581,
356
+ "step": 45
357
+ },
358
+ {
359
+ "epoch": 4.18,
360
+ "learning_rate": 1.672727272727273e-05,
361
+ "loss": 2.2201,
362
+ "step": 46
363
+ },
364
+ {
365
+ "epoch": 4.27,
366
+ "learning_rate": 1.7090909090909092e-05,
367
+ "loss": 2.1681,
368
+ "step": 47
369
+ },
370
+ {
371
+ "epoch": 4.36,
372
+ "learning_rate": 1.7454545454545456e-05,
373
+ "loss": 2.1538,
374
+ "step": 48
375
+ },
376
+ {
377
+ "epoch": 4.45,
378
+ "learning_rate": 1.781818181818182e-05,
379
+ "loss": 2.1502,
380
+ "step": 49
381
+ },
382
+ {
383
+ "epoch": 4.55,
384
+ "learning_rate": 1.8181818181818182e-05,
385
+ "loss": 2.1401,
386
+ "step": 50
387
+ },
388
+ {
389
+ "epoch": 4.64,
390
+ "learning_rate": 1.8545454545454545e-05,
391
+ "loss": 2.0827,
392
+ "step": 51
393
+ },
394
+ {
395
+ "epoch": 4.73,
396
+ "learning_rate": 1.890909090909091e-05,
397
+ "loss": 2.0856,
398
+ "step": 52
399
+ },
400
+ {
401
+ "epoch": 4.82,
402
+ "learning_rate": 1.9272727272727272e-05,
403
+ "loss": 2.0984,
404
+ "step": 53
405
+ },
406
+ {
407
+ "epoch": 4.91,
408
+ "learning_rate": 1.9636363636363635e-05,
409
+ "loss": 2.049,
410
+ "step": 54
411
+ },
412
+ {
413
+ "epoch": 5.0,
414
+ "learning_rate": 2e-05,
415
+ "loss": 2.0259,
416
+ "step": 55
417
+ },
418
+ {
419
+ "epoch": 5.09,
420
+ "learning_rate": 2.0363636363636365e-05,
421
+ "loss": 1.961,
422
+ "step": 56
423
+ },
424
+ {
425
+ "epoch": 5.18,
426
+ "learning_rate": 2.0727272727272728e-05,
427
+ "loss": 1.9549,
428
+ "step": 57
429
+ },
430
+ {
431
+ "epoch": 5.27,
432
+ "learning_rate": 2.109090909090909e-05,
433
+ "loss": 1.9661,
434
+ "step": 58
435
+ },
436
+ {
437
+ "epoch": 5.36,
438
+ "learning_rate": 2.1454545454545455e-05,
439
+ "loss": 1.9233,
440
+ "step": 59
441
+ },
442
+ {
443
+ "epoch": 5.45,
444
+ "learning_rate": 2.1818181818181818e-05,
445
+ "loss": 1.9173,
446
+ "step": 60
447
+ },
448
+ {
449
+ "epoch": 5.45,
450
+ "eval_accuracy_NAT": 0.9258208329480817,
451
+ "eval_accuracy_REST": 0.0,
452
+ "eval_accuracy_concrete": 0.5797794433237478,
453
+ "eval_accuracy_concrete tiles": 0.0,
454
+ "eval_accuracy_dark bricks": 0.18390362187752587,
455
+ "eval_accuracy_grass": 0.419499715151494,
456
+ "eval_accuracy_red bricks": 0.0,
457
+ "eval_accuracy_road": 0.6828861880537421,
458
+ "eval_accuracy_rough concrete": 0.0,
459
+ "eval_accuracy_rough red sidewalk": 0.0004686681553772779,
460
+ "eval_accuracy_speedway bricks": 0.9326892558412893,
461
+ "eval_accuracy_steel": 0.00015665193588813403,
462
+ "eval_accuracy_tiles": 0.0,
463
+ "eval_accuracy_unlabeled": NaN,
464
+ "eval_iou_NAT": 0.915879998687335,
465
+ "eval_iou_REST": 0.0,
466
+ "eval_iou_concrete": 0.3659033117650445,
467
+ "eval_iou_concrete tiles": 0.0,
468
+ "eval_iou_dark bricks": 0.02468574498974441,
469
+ "eval_iou_grass": 0.19461708465144334,
470
+ "eval_iou_red bricks": 0.0,
471
+ "eval_iou_road": 0.5963247243226463,
472
+ "eval_iou_rough concrete": 0.0,
473
+ "eval_iou_rough red sidewalk": 0.00032108185698636346,
474
+ "eval_iou_speedway bricks": 0.7993201887295917,
475
+ "eval_iou_steel": 0.00015567773071030014,
476
+ "eval_iou_tiles": 0.0,
477
+ "eval_iou_unlabeled": 0.0,
478
+ "eval_loss": 1.6415036916732788,
479
+ "eval_mean_accuracy": 0.286554182868242,
480
+ "eval_mean_iou": 0.20694341519525014,
481
+ "eval_overall_accuracy": 0.8386445575290256,
482
+ "eval_runtime": 3.435,
483
+ "eval_samples_per_second": 10.48,
484
+ "eval_steps_per_second": 0.582,
485
+ "step": 60
486
+ },
487
+ {
488
+ "epoch": 5.55,
489
+ "learning_rate": 2.2181818181818184e-05,
490
+ "loss": 1.8719,
491
+ "step": 61
492
+ },
493
+ {
494
+ "epoch": 5.64,
495
+ "learning_rate": 2.2545454545454548e-05,
496
+ "loss": 1.851,
497
+ "step": 62
498
+ },
499
+ {
500
+ "epoch": 5.73,
501
+ "learning_rate": 2.290909090909091e-05,
502
+ "loss": 1.8528,
503
+ "step": 63
504
+ },
505
+ {
506
+ "epoch": 5.82,
507
+ "learning_rate": 2.3272727272727274e-05,
508
+ "loss": 1.8134,
509
+ "step": 64
510
+ },
511
+ {
512
+ "epoch": 5.91,
513
+ "learning_rate": 2.3636363636363637e-05,
514
+ "loss": 1.7904,
515
+ "step": 65
516
+ },
517
+ {
518
+ "epoch": 6.0,
519
+ "learning_rate": 2.4e-05,
520
+ "loss": 1.7922,
521
+ "step": 66
522
+ },
523
+ {
524
+ "epoch": 6.09,
525
+ "learning_rate": 2.4363636363636364e-05,
526
+ "loss": 1.7283,
527
+ "step": 67
528
+ },
529
+ {
530
+ "epoch": 6.18,
531
+ "learning_rate": 2.472727272727273e-05,
532
+ "loss": 1.7191,
533
+ "step": 68
534
+ },
535
+ {
536
+ "epoch": 6.27,
537
+ "learning_rate": 2.5090909090909094e-05,
538
+ "loss": 1.7944,
539
+ "step": 69
540
+ },
541
+ {
542
+ "epoch": 6.36,
543
+ "learning_rate": 2.5454545454545454e-05,
544
+ "loss": 1.678,
545
+ "step": 70
546
+ },
547
+ {
548
+ "epoch": 6.45,
549
+ "learning_rate": 2.581818181818182e-05,
550
+ "loss": 1.7206,
551
+ "step": 71
552
+ },
553
+ {
554
+ "epoch": 6.55,
555
+ "learning_rate": 2.6181818181818187e-05,
556
+ "loss": 1.6827,
557
+ "step": 72
558
+ },
559
+ {
560
+ "epoch": 6.64,
561
+ "learning_rate": 2.6545454545454547e-05,
562
+ "loss": 1.6447,
563
+ "step": 73
564
+ },
565
+ {
566
+ "epoch": 6.73,
567
+ "learning_rate": 2.6909090909090913e-05,
568
+ "loss": 1.6101,
569
+ "step": 74
570
+ },
571
+ {
572
+ "epoch": 6.82,
573
+ "learning_rate": 2.7272727272727273e-05,
574
+ "loss": 1.599,
575
+ "step": 75
576
+ },
577
+ {
578
+ "epoch": 6.91,
579
+ "learning_rate": 2.7636363636363636e-05,
580
+ "loss": 1.6276,
581
+ "step": 76
582
+ },
583
+ {
584
+ "epoch": 7.0,
585
+ "learning_rate": 2.8000000000000003e-05,
586
+ "loss": 1.6611,
587
+ "step": 77
588
+ },
589
+ {
590
+ "epoch": 7.09,
591
+ "learning_rate": 2.8363636363636363e-05,
592
+ "loss": 1.5806,
593
+ "step": 78
594
+ },
595
+ {
596
+ "epoch": 7.18,
597
+ "learning_rate": 2.872727272727273e-05,
598
+ "loss": 1.5683,
599
+ "step": 79
600
+ },
601
+ {
602
+ "epoch": 7.27,
603
+ "learning_rate": 2.909090909090909e-05,
604
+ "loss": 1.5459,
605
+ "step": 80
606
+ },
607
+ {
608
+ "epoch": 7.27,
609
+ "eval_accuracy_NAT": 0.9541145930013841,
610
+ "eval_accuracy_REST": 0.0,
611
+ "eval_accuracy_concrete": 0.7506236855659041,
612
+ "eval_accuracy_concrete tiles": 0.0,
613
+ "eval_accuracy_dark bricks": 0.0003079173241984527,
614
+ "eval_accuracy_grass": 0.33006776947891947,
615
+ "eval_accuracy_red bricks": 0.0,
616
+ "eval_accuracy_road": 0.8611586397367794,
617
+ "eval_accuracy_rough concrete": 0.0,
618
+ "eval_accuracy_rough red sidewalk": 0.0,
619
+ "eval_accuracy_speedway bricks": 0.9632361726084233,
620
+ "eval_accuracy_steel": 0.0,
621
+ "eval_accuracy_tiles": 0.0,
622
+ "eval_accuracy_unlabeled": NaN,
623
+ "eval_iou_NAT": 0.9413840553822712,
624
+ "eval_iou_REST": 0.0,
625
+ "eval_iou_concrete": 0.4971771539137395,
626
+ "eval_iou_concrete tiles": 0.0,
627
+ "eval_iou_dark bricks": 0.0002478775484910454,
628
+ "eval_iou_grass": 0.2048493314707643,
629
+ "eval_iou_red bricks": 0.0,
630
+ "eval_iou_road": 0.715494766157398,
631
+ "eval_iou_rough concrete": 0.0,
632
+ "eval_iou_rough red sidewalk": 0.0,
633
+ "eval_iou_speedway bricks": 0.8327570610740783,
634
+ "eval_iou_steel": 0.0,
635
+ "eval_iou_tiles": 0.0,
636
+ "eval_iou_unlabeled": 0.0,
637
+ "eval_loss": 1.2371957302093506,
638
+ "eval_mean_accuracy": 0.2968852905935084,
639
+ "eval_mean_iou": 0.22799358896762448,
640
+ "eval_overall_accuracy": 0.8868987825181749,
641
+ "eval_runtime": 3.6626,
642
+ "eval_samples_per_second": 9.829,
643
+ "eval_steps_per_second": 0.546,
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:6ae3e621e780039b6372f57185ebc8826ff51d8891b4d0fda6ee4b4df223c8c5
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:2ad22c770dbf928f9aeb0e0a6371a0cf1f7b944d5d3d3ae5fa29dc5244b66c39
3
+ size 14944077
runs/May20_21-20-16_robolidar/events.out.tfevents.1716258023.robolidar.3008813.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c20f7e002bc0cea49d760b2de936e03ddad8cc09249466094d9fec6f994b15ca
3
+ size 25791
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6ae3e621e780039b6372f57185ebc8826ff51d8891b4d0fda6ee4b4df223c8c5
3
+ size 4155