sam1120 commited on
Commit
dc074c5
1 Parent(s): 5c3abf6

Training in progress, step 80

Browse files
checkpoint-80/config.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b5",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 768,
9
+ "depths": [
10
+ 3,
11
+ 6,
12
+ 40,
13
+ 3
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
+ "hidden_act": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_sizes": [
25
+ 64,
26
+ 128,
27
+ 320,
28
+ 512
29
+ ],
30
+ "id2label": {
31
+ "0": "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:87a4663a66ad188bf8b1bc9a95d5e59bc352ec66c778886a3388d05802187814
3
+ size 677522687
checkpoint-80/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f55a4f3ecf4cc23deb2be02b27e900a87f34d795d3bd90b902f2b3477a56cadf
3
+ size 338830281
checkpoint-80/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:03882aa57658d3f758c785af19723b50a9ba1ea14d770d9e4c07ba418ec025e5
3
+ size 14575
checkpoint-80/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7207c7808d37314625904a148be0869b3f50ac0397ac3480d2f4ad9419998b03
3
+ size 627
checkpoint-80/trainer_state.json ADDED
@@ -0,0 +1,652 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.4204064607620239,
3
+ "best_model_checkpoint": "/robodata/smodak/Projects/lifelong_concept_learner/scripts/terrainseg/training/models/safety-utcustom-terrain-final/checkpoint-80",
4
+ "epoch": 4.705882352941177,
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.06,
12
+ "learning_rate": 2.9411764705882356e-07,
13
+ "loss": 2.7081,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 0.12,
18
+ "learning_rate": 5.882352941176471e-07,
19
+ "loss": 2.7218,
20
+ "step": 2
21
+ },
22
+ {
23
+ "epoch": 0.18,
24
+ "learning_rate": 8.823529411764706e-07,
25
+ "loss": 2.709,
26
+ "step": 3
27
+ },
28
+ {
29
+ "epoch": 0.24,
30
+ "learning_rate": 1.1764705882352942e-06,
31
+ "loss": 2.6939,
32
+ "step": 4
33
+ },
34
+ {
35
+ "epoch": 0.29,
36
+ "learning_rate": 1.4705882352941177e-06,
37
+ "loss": 2.6979,
38
+ "step": 5
39
+ },
40
+ {
41
+ "epoch": 0.35,
42
+ "learning_rate": 1.7647058823529412e-06,
43
+ "loss": 2.6788,
44
+ "step": 6
45
+ },
46
+ {
47
+ "epoch": 0.41,
48
+ "learning_rate": 2.058823529411765e-06,
49
+ "loss": 2.6769,
50
+ "step": 7
51
+ },
52
+ {
53
+ "epoch": 0.47,
54
+ "learning_rate": 2.3529411764705885e-06,
55
+ "loss": 2.6464,
56
+ "step": 8
57
+ },
58
+ {
59
+ "epoch": 0.53,
60
+ "learning_rate": 2.647058823529412e-06,
61
+ "loss": 2.6424,
62
+ "step": 9
63
+ },
64
+ {
65
+ "epoch": 0.59,
66
+ "learning_rate": 2.9411764705882355e-06,
67
+ "loss": 2.6198,
68
+ "step": 10
69
+ },
70
+ {
71
+ "epoch": 0.65,
72
+ "learning_rate": 3.2352941176470594e-06,
73
+ "loss": 2.6228,
74
+ "step": 11
75
+ },
76
+ {
77
+ "epoch": 0.71,
78
+ "learning_rate": 3.5294117647058825e-06,
79
+ "loss": 2.6158,
80
+ "step": 12
81
+ },
82
+ {
83
+ "epoch": 0.76,
84
+ "learning_rate": 3.823529411764706e-06,
85
+ "loss": 2.632,
86
+ "step": 13
87
+ },
88
+ {
89
+ "epoch": 0.82,
90
+ "learning_rate": 4.11764705882353e-06,
91
+ "loss": 2.587,
92
+ "step": 14
93
+ },
94
+ {
95
+ "epoch": 0.88,
96
+ "learning_rate": 4.411764705882353e-06,
97
+ "loss": 2.5767,
98
+ "step": 15
99
+ },
100
+ {
101
+ "epoch": 0.94,
102
+ "learning_rate": 4.705882352941177e-06,
103
+ "loss": 2.5638,
104
+ "step": 16
105
+ },
106
+ {
107
+ "epoch": 1.0,
108
+ "learning_rate": 5e-06,
109
+ "loss": 2.517,
110
+ "step": 17
111
+ },
112
+ {
113
+ "epoch": 1.06,
114
+ "learning_rate": 5.294117647058824e-06,
115
+ "loss": 2.5202,
116
+ "step": 18
117
+ },
118
+ {
119
+ "epoch": 1.12,
120
+ "learning_rate": 5.588235294117647e-06,
121
+ "loss": 2.5257,
122
+ "step": 19
123
+ },
124
+ {
125
+ "epoch": 1.18,
126
+ "learning_rate": 5.882352941176471e-06,
127
+ "loss": 2.51,
128
+ "step": 20
129
+ },
130
+ {
131
+ "epoch": 1.18,
132
+ "eval_accuracy_NAT": 0.0901846148886272,
133
+ "eval_accuracy_REST": 0.1772357723577236,
134
+ "eval_accuracy_concrete": 0.21836230658804573,
135
+ "eval_accuracy_concrete tiles": 0.10897949563987744,
136
+ "eval_accuracy_dark bricks": 0.01268591007407691,
137
+ "eval_accuracy_grass": 0.027127403891645564,
138
+ "eval_accuracy_red bricks": 0.0003844463971880492,
139
+ "eval_accuracy_road": 0.0070585682671182835,
140
+ "eval_accuracy_rough concrete": 0.002699336785465293,
141
+ "eval_accuracy_rough red sidewalk": 0.0,
142
+ "eval_accuracy_speedway bricks": 0.6741853129093485,
143
+ "eval_accuracy_steel": 0.0007721641086127534,
144
+ "eval_accuracy_tiles": 0.2798933624163107,
145
+ "eval_accuracy_unlabeled": NaN,
146
+ "eval_iou_NAT": 0.09010334674005777,
147
+ "eval_iou_REST": 2.227093222444523e-05,
148
+ "eval_iou_concrete": 0.07965033823417267,
149
+ "eval_iou_concrete tiles": 0.0011714161291131616,
150
+ "eval_iou_dark bricks": 0.0030587532148061655,
151
+ "eval_iou_grass": 0.019163923774789638,
152
+ "eval_iou_red bricks": 5.560869275085081e-06,
153
+ "eval_iou_road": 0.006794427417222495,
154
+ "eval_iou_rough concrete": 0.00013145523557474666,
155
+ "eval_iou_rough red sidewalk": 0.0,
156
+ "eval_iou_speedway bricks": 0.4632836614619225,
157
+ "eval_iou_steel": 0.0006867765509148131,
158
+ "eval_iou_tiles": 0.0032256814378599824,
159
+ "eval_iou_unlabeled": 0.0,
160
+ "eval_loss": 2.507078170776367,
161
+ "eval_mean_accuracy": 0.12304374571723385,
162
+ "eval_mean_iou": 0.04766411514270953,
163
+ "eval_overall_accuracy": 0.20164948802883342,
164
+ "eval_runtime": 18.8176,
165
+ "eval_samples_per_second": 6.271,
166
+ "eval_steps_per_second": 0.425,
167
+ "step": 20
168
+ },
169
+ {
170
+ "epoch": 1.24,
171
+ "learning_rate": 6.1764705882352944e-06,
172
+ "loss": 2.4771,
173
+ "step": 21
174
+ },
175
+ {
176
+ "epoch": 1.29,
177
+ "learning_rate": 6.470588235294119e-06,
178
+ "loss": 2.4758,
179
+ "step": 22
180
+ },
181
+ {
182
+ "epoch": 1.35,
183
+ "learning_rate": 6.7647058823529414e-06,
184
+ "loss": 2.4283,
185
+ "step": 23
186
+ },
187
+ {
188
+ "epoch": 1.41,
189
+ "learning_rate": 7.058823529411765e-06,
190
+ "loss": 2.3783,
191
+ "step": 24
192
+ },
193
+ {
194
+ "epoch": 1.47,
195
+ "learning_rate": 7.3529411764705884e-06,
196
+ "loss": 2.3059,
197
+ "step": 25
198
+ },
199
+ {
200
+ "epoch": 1.53,
201
+ "learning_rate": 7.647058823529413e-06,
202
+ "loss": 2.3405,
203
+ "step": 26
204
+ },
205
+ {
206
+ "epoch": 1.59,
207
+ "learning_rate": 7.941176470588235e-06,
208
+ "loss": 2.2917,
209
+ "step": 27
210
+ },
211
+ {
212
+ "epoch": 1.65,
213
+ "learning_rate": 8.23529411764706e-06,
214
+ "loss": 2.2857,
215
+ "step": 28
216
+ },
217
+ {
218
+ "epoch": 1.71,
219
+ "learning_rate": 8.529411764705883e-06,
220
+ "loss": 2.1291,
221
+ "step": 29
222
+ },
223
+ {
224
+ "epoch": 1.76,
225
+ "learning_rate": 8.823529411764707e-06,
226
+ "loss": 2.1234,
227
+ "step": 30
228
+ },
229
+ {
230
+ "epoch": 1.82,
231
+ "learning_rate": 9.117647058823529e-06,
232
+ "loss": 2.0916,
233
+ "step": 31
234
+ },
235
+ {
236
+ "epoch": 1.88,
237
+ "learning_rate": 9.411764705882354e-06,
238
+ "loss": 2.0543,
239
+ "step": 32
240
+ },
241
+ {
242
+ "epoch": 1.94,
243
+ "learning_rate": 9.705882352941177e-06,
244
+ "loss": 1.9605,
245
+ "step": 33
246
+ },
247
+ {
248
+ "epoch": 2.0,
249
+ "learning_rate": 1e-05,
250
+ "loss": 1.9835,
251
+ "step": 34
252
+ },
253
+ {
254
+ "epoch": 2.06,
255
+ "learning_rate": 1.0294117647058824e-05,
256
+ "loss": 2.0005,
257
+ "step": 35
258
+ },
259
+ {
260
+ "epoch": 2.12,
261
+ "learning_rate": 1.0588235294117648e-05,
262
+ "loss": 1.8628,
263
+ "step": 36
264
+ },
265
+ {
266
+ "epoch": 2.18,
267
+ "learning_rate": 1.0882352941176471e-05,
268
+ "loss": 1.8321,
269
+ "step": 37
270
+ },
271
+ {
272
+ "epoch": 2.24,
273
+ "learning_rate": 1.1176470588235295e-05,
274
+ "loss": 1.7578,
275
+ "step": 38
276
+ },
277
+ {
278
+ "epoch": 2.29,
279
+ "learning_rate": 1.1470588235294118e-05,
280
+ "loss": 1.7233,
281
+ "step": 39
282
+ },
283
+ {
284
+ "epoch": 2.35,
285
+ "learning_rate": 1.1764705882352942e-05,
286
+ "loss": 1.5504,
287
+ "step": 40
288
+ },
289
+ {
290
+ "epoch": 2.35,
291
+ "eval_accuracy_NAT": 0.760154401121494,
292
+ "eval_accuracy_REST": 0.0,
293
+ "eval_accuracy_concrete": 0.3386920222776326,
294
+ "eval_accuracy_concrete tiles": 0.007895357058684892,
295
+ "eval_accuracy_dark bricks": 0.0014308136975288556,
296
+ "eval_accuracy_grass": 0.07454476982063306,
297
+ "eval_accuracy_red bricks": 0.0,
298
+ "eval_accuracy_road": 0.3332036483520477,
299
+ "eval_accuracy_rough concrete": 0.0009832021404012656,
300
+ "eval_accuracy_rough red sidewalk": 0.0,
301
+ "eval_accuracy_speedway bricks": 0.9879532008612127,
302
+ "eval_accuracy_steel": 0.0,
303
+ "eval_accuracy_tiles": 0.00875518797903602,
304
+ "eval_accuracy_unlabeled": NaN,
305
+ "eval_iou_NAT": 0.7577068331202973,
306
+ "eval_iou_REST": 0.0,
307
+ "eval_iou_concrete": 0.2037716446695068,
308
+ "eval_iou_concrete tiles": 0.00033778197232406094,
309
+ "eval_iou_dark bricks": 0.0009957339958239116,
310
+ "eval_iou_grass": 0.06824458242953867,
311
+ "eval_iou_red bricks": 0.0,
312
+ "eval_iou_road": 0.31270597640653264,
313
+ "eval_iou_rough concrete": 0.00024414351495637967,
314
+ "eval_iou_rough red sidewalk": 0.0,
315
+ "eval_iou_speedway bricks": 0.571054819328365,
316
+ "eval_iou_steel": 0.0,
317
+ "eval_iou_tiles": 0.0007685393497465681,
318
+ "eval_iou_unlabeled": 0.0,
319
+ "eval_loss": 1.840782642364502,
320
+ "eval_mean_accuracy": 0.1933548156391286,
321
+ "eval_mean_iou": 0.13684500391336366,
322
+ "eval_overall_accuracy": 0.6935860908637612,
323
+ "eval_runtime": 7.985,
324
+ "eval_samples_per_second": 14.778,
325
+ "eval_steps_per_second": 1.002,
326
+ "step": 40
327
+ },
328
+ {
329
+ "epoch": 2.41,
330
+ "learning_rate": 1.2058823529411765e-05,
331
+ "loss": 1.7505,
332
+ "step": 41
333
+ },
334
+ {
335
+ "epoch": 2.47,
336
+ "learning_rate": 1.2352941176470589e-05,
337
+ "loss": 1.4808,
338
+ "step": 42
339
+ },
340
+ {
341
+ "epoch": 2.53,
342
+ "learning_rate": 1.2647058823529412e-05,
343
+ "loss": 1.4041,
344
+ "step": 43
345
+ },
346
+ {
347
+ "epoch": 2.59,
348
+ "learning_rate": 1.2941176470588238e-05,
349
+ "loss": 1.3261,
350
+ "step": 44
351
+ },
352
+ {
353
+ "epoch": 2.65,
354
+ "learning_rate": 1.323529411764706e-05,
355
+ "loss": 1.3337,
356
+ "step": 45
357
+ },
358
+ {
359
+ "epoch": 2.71,
360
+ "learning_rate": 1.3529411764705883e-05,
361
+ "loss": 1.3177,
362
+ "step": 46
363
+ },
364
+ {
365
+ "epoch": 2.76,
366
+ "learning_rate": 1.3823529411764708e-05,
367
+ "loss": 1.2551,
368
+ "step": 47
369
+ },
370
+ {
371
+ "epoch": 2.82,
372
+ "learning_rate": 1.411764705882353e-05,
373
+ "loss": 1.119,
374
+ "step": 48
375
+ },
376
+ {
377
+ "epoch": 2.88,
378
+ "learning_rate": 1.4411764705882352e-05,
379
+ "loss": 1.1259,
380
+ "step": 49
381
+ },
382
+ {
383
+ "epoch": 2.94,
384
+ "learning_rate": 1.4705882352941177e-05,
385
+ "loss": 1.0199,
386
+ "step": 50
387
+ },
388
+ {
389
+ "epoch": 3.0,
390
+ "learning_rate": 1.5e-05,
391
+ "loss": 1.1612,
392
+ "step": 51
393
+ },
394
+ {
395
+ "epoch": 3.06,
396
+ "learning_rate": 1.5294117647058826e-05,
397
+ "loss": 0.8821,
398
+ "step": 52
399
+ },
400
+ {
401
+ "epoch": 3.12,
402
+ "learning_rate": 1.558823529411765e-05,
403
+ "loss": 0.8667,
404
+ "step": 53
405
+ },
406
+ {
407
+ "epoch": 3.18,
408
+ "learning_rate": 1.588235294117647e-05,
409
+ "loss": 0.8607,
410
+ "step": 54
411
+ },
412
+ {
413
+ "epoch": 3.24,
414
+ "learning_rate": 1.6176470588235296e-05,
415
+ "loss": 0.8021,
416
+ "step": 55
417
+ },
418
+ {
419
+ "epoch": 3.29,
420
+ "learning_rate": 1.647058823529412e-05,
421
+ "loss": 0.7418,
422
+ "step": 56
423
+ },
424
+ {
425
+ "epoch": 3.35,
426
+ "learning_rate": 1.676470588235294e-05,
427
+ "loss": 0.7114,
428
+ "step": 57
429
+ },
430
+ {
431
+ "epoch": 3.41,
432
+ "learning_rate": 1.7058823529411767e-05,
433
+ "loss": 0.7799,
434
+ "step": 58
435
+ },
436
+ {
437
+ "epoch": 3.47,
438
+ "learning_rate": 1.735294117647059e-05,
439
+ "loss": 0.6974,
440
+ "step": 59
441
+ },
442
+ {
443
+ "epoch": 3.53,
444
+ "learning_rate": 1.7647058823529414e-05,
445
+ "loss": 0.6271,
446
+ "step": 60
447
+ },
448
+ {
449
+ "epoch": 3.53,
450
+ "eval_accuracy_NAT": 0.9551305368117847,
451
+ "eval_accuracy_REST": 0.0,
452
+ "eval_accuracy_concrete": 0.8126940616930818,
453
+ "eval_accuracy_concrete tiles": 0.0,
454
+ "eval_accuracy_dark bricks": 0.0,
455
+ "eval_accuracy_grass": 0.25898348643298624,
456
+ "eval_accuracy_red bricks": 0.0,
457
+ "eval_accuracy_road": 0.8978859032247204,
458
+ "eval_accuracy_rough concrete": 0.0,
459
+ "eval_accuracy_rough red sidewalk": 0.0,
460
+ "eval_accuracy_speedway bricks": 0.9793115987478005,
461
+ "eval_accuracy_steel": 0.0,
462
+ "eval_accuracy_tiles": 3.0294768093550245e-05,
463
+ "eval_accuracy_unlabeled": NaN,
464
+ "eval_iou_NAT": 0.9404775296004152,
465
+ "eval_iou_REST": 0.0,
466
+ "eval_iou_concrete": 0.5259192497701041,
467
+ "eval_iou_concrete tiles": 0.0,
468
+ "eval_iou_dark bricks": 0.0,
469
+ "eval_iou_grass": 0.2391512912549761,
470
+ "eval_iou_red bricks": 0.0,
471
+ "eval_iou_road": 0.6993177996437074,
472
+ "eval_iou_rough concrete": 0.0,
473
+ "eval_iou_rough red sidewalk": 0.0,
474
+ "eval_iou_speedway bricks": 0.8262999758534085,
475
+ "eval_iou_steel": 0.0,
476
+ "eval_iou_tiles": 2.891343318105592e-05,
477
+ "eval_iou_unlabeled": NaN,
478
+ "eval_loss": 0.8082596063613892,
479
+ "eval_mean_accuracy": 0.30031045243680515,
480
+ "eval_mean_iou": 0.24855344304275326,
481
+ "eval_overall_accuracy": 0.8893904281874835,
482
+ "eval_runtime": 6.7648,
483
+ "eval_samples_per_second": 17.443,
484
+ "eval_steps_per_second": 1.183,
485
+ "step": 60
486
+ },
487
+ {
488
+ "epoch": 3.59,
489
+ "learning_rate": 1.7941176470588237e-05,
490
+ "loss": 0.6901,
491
+ "step": 61
492
+ },
493
+ {
494
+ "epoch": 3.65,
495
+ "learning_rate": 1.8235294117647057e-05,
496
+ "loss": 0.5778,
497
+ "step": 62
498
+ },
499
+ {
500
+ "epoch": 3.71,
501
+ "learning_rate": 1.8529411764705884e-05,
502
+ "loss": 0.6319,
503
+ "step": 63
504
+ },
505
+ {
506
+ "epoch": 3.76,
507
+ "learning_rate": 1.8823529411764708e-05,
508
+ "loss": 0.5626,
509
+ "step": 64
510
+ },
511
+ {
512
+ "epoch": 3.82,
513
+ "learning_rate": 1.9117647058823528e-05,
514
+ "loss": 0.7374,
515
+ "step": 65
516
+ },
517
+ {
518
+ "epoch": 3.88,
519
+ "learning_rate": 1.9411764705882355e-05,
520
+ "loss": 0.6702,
521
+ "step": 66
522
+ },
523
+ {
524
+ "epoch": 3.94,
525
+ "learning_rate": 1.9705882352941178e-05,
526
+ "loss": 0.5477,
527
+ "step": 67
528
+ },
529
+ {
530
+ "epoch": 4.0,
531
+ "learning_rate": 2e-05,
532
+ "loss": 0.586,
533
+ "step": 68
534
+ },
535
+ {
536
+ "epoch": 4.06,
537
+ "learning_rate": 2.0294117647058825e-05,
538
+ "loss": 0.505,
539
+ "step": 69
540
+ },
541
+ {
542
+ "epoch": 4.12,
543
+ "learning_rate": 2.058823529411765e-05,
544
+ "loss": 0.5758,
545
+ "step": 70
546
+ },
547
+ {
548
+ "epoch": 4.18,
549
+ "learning_rate": 2.0882352941176472e-05,
550
+ "loss": 0.5694,
551
+ "step": 71
552
+ },
553
+ {
554
+ "epoch": 4.24,
555
+ "learning_rate": 2.1176470588235296e-05,
556
+ "loss": 0.4535,
557
+ "step": 72
558
+ },
559
+ {
560
+ "epoch": 4.29,
561
+ "learning_rate": 2.1470588235294116e-05,
562
+ "loss": 0.5105,
563
+ "step": 73
564
+ },
565
+ {
566
+ "epoch": 4.35,
567
+ "learning_rate": 2.1764705882352943e-05,
568
+ "loss": 0.5525,
569
+ "step": 74
570
+ },
571
+ {
572
+ "epoch": 4.41,
573
+ "learning_rate": 2.2058823529411766e-05,
574
+ "loss": 0.465,
575
+ "step": 75
576
+ },
577
+ {
578
+ "epoch": 4.47,
579
+ "learning_rate": 2.235294117647059e-05,
580
+ "loss": 0.5477,
581
+ "step": 76
582
+ },
583
+ {
584
+ "epoch": 4.53,
585
+ "learning_rate": 2.2647058823529413e-05,
586
+ "loss": 0.3823,
587
+ "step": 77
588
+ },
589
+ {
590
+ "epoch": 4.59,
591
+ "learning_rate": 2.2941176470588237e-05,
592
+ "loss": 0.4698,
593
+ "step": 78
594
+ },
595
+ {
596
+ "epoch": 4.65,
597
+ "learning_rate": 2.323529411764706e-05,
598
+ "loss": 0.4636,
599
+ "step": 79
600
+ },
601
+ {
602
+ "epoch": 4.71,
603
+ "learning_rate": 2.3529411764705884e-05,
604
+ "loss": 0.3963,
605
+ "step": 80
606
+ },
607
+ {
608
+ "epoch": 4.71,
609
+ "eval_accuracy_NAT": 0.969774210544504,
610
+ "eval_accuracy_REST": 0.0,
611
+ "eval_accuracy_concrete": 0.7972872258293342,
612
+ "eval_accuracy_concrete tiles": 0.04426113598868725,
613
+ "eval_accuracy_dark bricks": 0.0,
614
+ "eval_accuracy_grass": 0.7826429655535243,
615
+ "eval_accuracy_red bricks": 0.0,
616
+ "eval_accuracy_road": 0.9650914038630805,
617
+ "eval_accuracy_rough concrete": 0.0,
618
+ "eval_accuracy_rough red sidewalk": 0.0,
619
+ "eval_accuracy_speedway bricks": 0.9776932262095467,
620
+ "eval_accuracy_steel": 0.0003548456421456666,
621
+ "eval_accuracy_tiles": 0.0,
622
+ "eval_accuracy_unlabeled": NaN,
623
+ "eval_iou_NAT": 0.9531585041831687,
624
+ "eval_iou_REST": 0.0,
625
+ "eval_iou_concrete": 0.6900984540650801,
626
+ "eval_iou_concrete tiles": 0.04426113598868725,
627
+ "eval_iou_dark bricks": 0.0,
628
+ "eval_iou_grass": 0.5632131076991799,
629
+ "eval_iou_red bricks": 0.0,
630
+ "eval_iou_road": 0.727967248165449,
631
+ "eval_iou_rough concrete": 0.0,
632
+ "eval_iou_rough red sidewalk": 0.0,
633
+ "eval_iou_speedway bricks": 0.8641130344685216,
634
+ "eval_iou_steel": 0.00035449307490288885,
635
+ "eval_iou_tiles": 0.0,
636
+ "eval_iou_unlabeled": NaN,
637
+ "eval_loss": 0.4204064607620239,
638
+ "eval_mean_accuracy": 0.34900807797160177,
639
+ "eval_mean_iou": 0.29562815212653765,
640
+ "eval_overall_accuracy": 0.9185845003289691,
641
+ "eval_runtime": 5.4734,
642
+ "eval_samples_per_second": 21.559,
643
+ "eval_steps_per_second": 1.462,
644
+ "step": 80
645
+ }
646
+ ],
647
+ "max_steps": 3400,
648
+ "num_train_epochs": 200,
649
+ "total_flos": 5.0620116674543616e+17,
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:94f6bdec54864366f6b4fd33ad685e0ecfd6f5af579d8074dce48dc71114c3e0
3
+ size 4219
config.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b5",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 768,
9
+ "depths": [
10
+ 3,
11
+ 6,
12
+ 40,
13
+ 3
14
+ ],
15
+ "downsampling_rates": [
16
+ 1,
17
+ 4,
18
+ 8,
19
+ 16
20
+ ],
21
+ "drop_path_rate": 0.1,
22
+ "hidden_act": "gelu",
23
+ "hidden_dropout_prob": 0.0,
24
+ "hidden_sizes": [
25
+ 64,
26
+ 128,
27
+ 320,
28
+ 512
29
+ ],
30
+ "id2label": {
31
+ "0": "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:f55a4f3ecf4cc23deb2be02b27e900a87f34d795d3bd90b902f2b3477a56cadf
3
+ size 338830281
runs/Dec09_04-08-30_robolidar/events.out.tfevents.1702116518.robolidar.2934636.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0ece70d24b0a61928750150282228b210f880411eba0bfefcecaa465ddf22a3f
3
+ size 25862
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:94f6bdec54864366f6b4fd33ad685e0ecfd6f5af579d8074dce48dc71114c3e0
3
+ size 4219