sam1120 commited on
Commit
2825920
1 Parent(s): 899dfc8

Training in progress, step 120

Browse files
checkpoint-120/config.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "dropoff",
33
+ "2": "undropoff"
34
+ },
35
+ "image_size": 224,
36
+ "initializer_range": 0.02,
37
+ "label2id": {
38
+ "dropoff": 1,
39
+ "undropoff": 2,
40
+ "unlabeled": 0
41
+ },
42
+ "layer_norm_eps": 1e-06,
43
+ "mlp_ratios": [
44
+ 4,
45
+ 4,
46
+ 4,
47
+ 4
48
+ ],
49
+ "model_type": "segformer",
50
+ "num_attention_heads": [
51
+ 1,
52
+ 2,
53
+ 5,
54
+ 8
55
+ ],
56
+ "num_channels": 6,
57
+ "num_encoder_blocks": 4,
58
+ "patch_sizes": [
59
+ 7,
60
+ 3,
61
+ 3,
62
+ 3
63
+ ],
64
+ "reshape_last_stage": true,
65
+ "semantic_loss_ignore_index": 255,
66
+ "sr_ratios": [
67
+ 8,
68
+ 4,
69
+ 2,
70
+ 1
71
+ ],
72
+ "strides": [
73
+ 4,
74
+ 2,
75
+ 2,
76
+ 2
77
+ ],
78
+ "torch_dtype": "float32",
79
+ "transformers_version": "4.30.2"
80
+ }
checkpoint-120/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:55a8a81d08bbde84e89e334f31c8f6f730f0216fb65d396a90c947e6b25013f9
3
+ size 29876485
checkpoint-120/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f0fc8bbb23c0420a3465abe203545dbf9fb563bede6e2fd8babb249ba6641f4
3
+ size 14951629
checkpoint-120/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d04b46b261d14fecd73ba66387b3f4629081598b88c439e27184f26d8ce1edea
3
+ size 14575
checkpoint-120/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2fc57cd04be113dc8e978540d0c7305911a11b36dd67e7a6268e8c5e6c00df77
3
+ size 627
checkpoint-120/trainer_state.json ADDED
@@ -0,0 +1,940 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.5707733035087585,
3
+ "best_model_checkpoint": "/robodata/smodak/Projects/nspl/scripts/terrainseg/training/models/dropoff-utcustom-train-SF-RGBD-b0_1/checkpoint-120",
4
+ "epoch": 60.0,
5
+ "global_step": 120,
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.5,
12
+ "learning_rate": 1.6666666666666667e-06,
13
+ "loss": 1.0746,
14
+ "step": 1
15
+ },
16
+ {
17
+ "epoch": 1.0,
18
+ "learning_rate": 3.3333333333333333e-06,
19
+ "loss": 1.0748,
20
+ "step": 2
21
+ },
22
+ {
23
+ "epoch": 1.5,
24
+ "learning_rate": 5e-06,
25
+ "loss": 1.0693,
26
+ "step": 3
27
+ },
28
+ {
29
+ "epoch": 2.0,
30
+ "learning_rate": 6.666666666666667e-06,
31
+ "loss": 1.0764,
32
+ "step": 4
33
+ },
34
+ {
35
+ "epoch": 2.5,
36
+ "learning_rate": 8.333333333333334e-06,
37
+ "loss": 1.0684,
38
+ "step": 5
39
+ },
40
+ {
41
+ "epoch": 3.0,
42
+ "learning_rate": 1e-05,
43
+ "loss": 1.0649,
44
+ "step": 6
45
+ },
46
+ {
47
+ "epoch": 3.5,
48
+ "learning_rate": 1.1666666666666668e-05,
49
+ "loss": 1.0601,
50
+ "step": 7
51
+ },
52
+ {
53
+ "epoch": 4.0,
54
+ "learning_rate": 1.3333333333333333e-05,
55
+ "loss": 1.0556,
56
+ "step": 8
57
+ },
58
+ {
59
+ "epoch": 4.5,
60
+ "learning_rate": 1.5000000000000002e-05,
61
+ "loss": 1.0471,
62
+ "step": 9
63
+ },
64
+ {
65
+ "epoch": 5.0,
66
+ "learning_rate": 1.6666666666666667e-05,
67
+ "loss": 1.0495,
68
+ "step": 10
69
+ },
70
+ {
71
+ "epoch": 5.0,
72
+ "eval_accuracy_dropoff": 0.20262391339250901,
73
+ "eval_accuracy_undropoff": 0.5118628585529819,
74
+ "eval_accuracy_unlabeled": NaN,
75
+ "eval_iou_dropoff": 0.047444479293322045,
76
+ "eval_iou_undropoff": 0.5081086334840771,
77
+ "eval_iou_unlabeled": 0.0,
78
+ "eval_loss": 1.0889850854873657,
79
+ "eval_mean_accuracy": 0.3572433859727454,
80
+ "eval_mean_iou": 0.18518437092579973,
81
+ "eval_overall_accuracy": 0.4990049362182617,
82
+ "eval_runtime": 2.7639,
83
+ "eval_samples_per_second": 7.236,
84
+ "eval_steps_per_second": 0.724,
85
+ "step": 10
86
+ },
87
+ {
88
+ "epoch": 5.5,
89
+ "learning_rate": 1.8333333333333333e-05,
90
+ "loss": 1.0418,
91
+ "step": 11
92
+ },
93
+ {
94
+ "epoch": 6.0,
95
+ "learning_rate": 2e-05,
96
+ "loss": 1.0375,
97
+ "step": 12
98
+ },
99
+ {
100
+ "epoch": 6.5,
101
+ "learning_rate": 1.9912280701754387e-05,
102
+ "loss": 1.034,
103
+ "step": 13
104
+ },
105
+ {
106
+ "epoch": 7.0,
107
+ "learning_rate": 1.9824561403508773e-05,
108
+ "loss": 1.0282,
109
+ "step": 14
110
+ },
111
+ {
112
+ "epoch": 7.5,
113
+ "learning_rate": 1.9736842105263158e-05,
114
+ "loss": 1.0212,
115
+ "step": 15
116
+ },
117
+ {
118
+ "epoch": 8.0,
119
+ "learning_rate": 1.9649122807017544e-05,
120
+ "loss": 1.0081,
121
+ "step": 16
122
+ },
123
+ {
124
+ "epoch": 8.5,
125
+ "learning_rate": 1.9561403508771933e-05,
126
+ "loss": 1.0061,
127
+ "step": 17
128
+ },
129
+ {
130
+ "epoch": 9.0,
131
+ "learning_rate": 1.9473684210526318e-05,
132
+ "loss": 1.0027,
133
+ "step": 18
134
+ },
135
+ {
136
+ "epoch": 9.5,
137
+ "learning_rate": 1.9385964912280704e-05,
138
+ "loss": 0.9979,
139
+ "step": 19
140
+ },
141
+ {
142
+ "epoch": 10.0,
143
+ "learning_rate": 1.929824561403509e-05,
144
+ "loss": 0.9941,
145
+ "step": 20
146
+ },
147
+ {
148
+ "epoch": 10.0,
149
+ "eval_accuracy_dropoff": 0.8224913415445309,
150
+ "eval_accuracy_undropoff": 0.8489730610750296,
151
+ "eval_accuracy_unlabeled": NaN,
152
+ "eval_iou_dropoff": 0.19307967569325465,
153
+ "eval_iou_undropoff": 0.8424968366811092,
154
+ "eval_iou_unlabeled": 0.0,
155
+ "eval_loss": 1.0478662252426147,
156
+ "eval_mean_accuracy": 0.8357322013097802,
157
+ "eval_mean_iou": 0.34519217079145464,
158
+ "eval_overall_accuracy": 0.847871971130371,
159
+ "eval_runtime": 2.9295,
160
+ "eval_samples_per_second": 6.827,
161
+ "eval_steps_per_second": 0.683,
162
+ "step": 20
163
+ },
164
+ {
165
+ "epoch": 10.5,
166
+ "learning_rate": 1.9210526315789474e-05,
167
+ "loss": 0.9861,
168
+ "step": 21
169
+ },
170
+ {
171
+ "epoch": 11.0,
172
+ "learning_rate": 1.912280701754386e-05,
173
+ "loss": 0.9817,
174
+ "step": 22
175
+ },
176
+ {
177
+ "epoch": 11.5,
178
+ "learning_rate": 1.9035087719298245e-05,
179
+ "loss": 0.9781,
180
+ "step": 23
181
+ },
182
+ {
183
+ "epoch": 12.0,
184
+ "learning_rate": 1.894736842105263e-05,
185
+ "loss": 0.9729,
186
+ "step": 24
187
+ },
188
+ {
189
+ "epoch": 12.5,
190
+ "learning_rate": 1.885964912280702e-05,
191
+ "loss": 0.9669,
192
+ "step": 25
193
+ },
194
+ {
195
+ "epoch": 13.0,
196
+ "learning_rate": 1.8771929824561405e-05,
197
+ "loss": 0.9634,
198
+ "step": 26
199
+ },
200
+ {
201
+ "epoch": 13.5,
202
+ "learning_rate": 1.868421052631579e-05,
203
+ "loss": 0.9642,
204
+ "step": 27
205
+ },
206
+ {
207
+ "epoch": 14.0,
208
+ "learning_rate": 1.8596491228070176e-05,
209
+ "loss": 0.9586,
210
+ "step": 28
211
+ },
212
+ {
213
+ "epoch": 14.5,
214
+ "learning_rate": 1.8508771929824562e-05,
215
+ "loss": 0.9559,
216
+ "step": 29
217
+ },
218
+ {
219
+ "epoch": 15.0,
220
+ "learning_rate": 1.8421052631578947e-05,
221
+ "loss": 0.9448,
222
+ "step": 30
223
+ },
224
+ {
225
+ "epoch": 15.0,
226
+ "eval_accuracy_dropoff": 0.7351131906695108,
227
+ "eval_accuracy_undropoff": 0.9082392930385471,
228
+ "eval_accuracy_unlabeled": NaN,
229
+ "eval_iou_dropoff": 0.23901293122734799,
230
+ "eval_iou_undropoff": 0.8979626128799195,
231
+ "eval_iou_unlabeled": 0.0,
232
+ "eval_loss": 0.9838544130325317,
233
+ "eval_mean_accuracy": 0.8216762418540289,
234
+ "eval_mean_iou": 0.37899184803575586,
235
+ "eval_overall_accuracy": 0.9010408401489258,
236
+ "eval_runtime": 3.1655,
237
+ "eval_samples_per_second": 6.318,
238
+ "eval_steps_per_second": 0.632,
239
+ "step": 30
240
+ },
241
+ {
242
+ "epoch": 15.5,
243
+ "learning_rate": 1.8333333333333333e-05,
244
+ "loss": 0.9406,
245
+ "step": 31
246
+ },
247
+ {
248
+ "epoch": 16.0,
249
+ "learning_rate": 1.824561403508772e-05,
250
+ "loss": 0.9378,
251
+ "step": 32
252
+ },
253
+ {
254
+ "epoch": 16.5,
255
+ "learning_rate": 1.8157894736842107e-05,
256
+ "loss": 0.935,
257
+ "step": 33
258
+ },
259
+ {
260
+ "epoch": 17.0,
261
+ "learning_rate": 1.8070175438596493e-05,
262
+ "loss": 0.9249,
263
+ "step": 34
264
+ },
265
+ {
266
+ "epoch": 17.5,
267
+ "learning_rate": 1.7982456140350878e-05,
268
+ "loss": 0.9162,
269
+ "step": 35
270
+ },
271
+ {
272
+ "epoch": 18.0,
273
+ "learning_rate": 1.7894736842105264e-05,
274
+ "loss": 0.9153,
275
+ "step": 36
276
+ },
277
+ {
278
+ "epoch": 18.5,
279
+ "learning_rate": 1.780701754385965e-05,
280
+ "loss": 0.8983,
281
+ "step": 37
282
+ },
283
+ {
284
+ "epoch": 19.0,
285
+ "learning_rate": 1.7719298245614035e-05,
286
+ "loss": 0.918,
287
+ "step": 38
288
+ },
289
+ {
290
+ "epoch": 19.5,
291
+ "learning_rate": 1.763157894736842e-05,
292
+ "loss": 0.9107,
293
+ "step": 39
294
+ },
295
+ {
296
+ "epoch": 20.0,
297
+ "learning_rate": 1.754385964912281e-05,
298
+ "loss": 0.8912,
299
+ "step": 40
300
+ },
301
+ {
302
+ "epoch": 20.0,
303
+ "eval_accuracy_dropoff": 0.4863276680657813,
304
+ "eval_accuracy_undropoff": 0.9437063733796893,
305
+ "eval_accuracy_unlabeled": NaN,
306
+ "eval_iou_dropoff": 0.23034404840794775,
307
+ "eval_iou_undropoff": 0.9232710241069885,
308
+ "eval_iou_unlabeled": 0.0,
309
+ "eval_loss": 0.9041398763656616,
310
+ "eval_mean_accuracy": 0.7150170207227353,
311
+ "eval_mean_iou": 0.3845383575049787,
312
+ "eval_overall_accuracy": 0.9246889114379883,
313
+ "eval_runtime": 2.9597,
314
+ "eval_samples_per_second": 6.757,
315
+ "eval_steps_per_second": 0.676,
316
+ "step": 40
317
+ },
318
+ {
319
+ "epoch": 20.5,
320
+ "learning_rate": 1.7456140350877195e-05,
321
+ "loss": 0.8927,
322
+ "step": 41
323
+ },
324
+ {
325
+ "epoch": 21.0,
326
+ "learning_rate": 1.736842105263158e-05,
327
+ "loss": 0.8849,
328
+ "step": 42
329
+ },
330
+ {
331
+ "epoch": 21.5,
332
+ "learning_rate": 1.7280701754385966e-05,
333
+ "loss": 0.8909,
334
+ "step": 43
335
+ },
336
+ {
337
+ "epoch": 22.0,
338
+ "learning_rate": 1.719298245614035e-05,
339
+ "loss": 0.8779,
340
+ "step": 44
341
+ },
342
+ {
343
+ "epoch": 22.5,
344
+ "learning_rate": 1.7105263157894737e-05,
345
+ "loss": 0.8723,
346
+ "step": 45
347
+ },
348
+ {
349
+ "epoch": 23.0,
350
+ "learning_rate": 1.7017543859649125e-05,
351
+ "loss": 0.8638,
352
+ "step": 46
353
+ },
354
+ {
355
+ "epoch": 23.5,
356
+ "learning_rate": 1.692982456140351e-05,
357
+ "loss": 0.867,
358
+ "step": 47
359
+ },
360
+ {
361
+ "epoch": 24.0,
362
+ "learning_rate": 1.6842105263157896e-05,
363
+ "loss": 0.8582,
364
+ "step": 48
365
+ },
366
+ {
367
+ "epoch": 24.5,
368
+ "learning_rate": 1.6754385964912282e-05,
369
+ "loss": 0.8498,
370
+ "step": 49
371
+ },
372
+ {
373
+ "epoch": 25.0,
374
+ "learning_rate": 1.6666666666666667e-05,
375
+ "loss": 0.8458,
376
+ "step": 50
377
+ },
378
+ {
379
+ "epoch": 25.0,
380
+ "eval_accuracy_dropoff": 0.38082983554668687,
381
+ "eval_accuracy_undropoff": 0.9565176118458433,
382
+ "eval_accuracy_unlabeled": NaN,
383
+ "eval_iou_dropoff": 0.21875946972192284,
384
+ "eval_iou_undropoff": 0.9316126713826287,
385
+ "eval_iou_unlabeled": 0.0,
386
+ "eval_loss": 0.7996953129768372,
387
+ "eval_mean_accuracy": 0.6686737236962651,
388
+ "eval_mean_iou": 0.38345738036818383,
389
+ "eval_overall_accuracy": 0.9325809478759766,
390
+ "eval_runtime": 2.9373,
391
+ "eval_samples_per_second": 6.809,
392
+ "eval_steps_per_second": 0.681,
393
+ "step": 50
394
+ },
395
+ {
396
+ "epoch": 25.5,
397
+ "learning_rate": 1.6578947368421053e-05,
398
+ "loss": 0.8489,
399
+ "step": 51
400
+ },
401
+ {
402
+ "epoch": 26.0,
403
+ "learning_rate": 1.649122807017544e-05,
404
+ "loss": 0.8322,
405
+ "step": 52
406
+ },
407
+ {
408
+ "epoch": 26.5,
409
+ "learning_rate": 1.6403508771929827e-05,
410
+ "loss": 0.8306,
411
+ "step": 53
412
+ },
413
+ {
414
+ "epoch": 27.0,
415
+ "learning_rate": 1.6315789473684213e-05,
416
+ "loss": 0.8211,
417
+ "step": 54
418
+ },
419
+ {
420
+ "epoch": 27.5,
421
+ "learning_rate": 1.62280701754386e-05,
422
+ "loss": 0.8128,
423
+ "step": 55
424
+ },
425
+ {
426
+ "epoch": 28.0,
427
+ "learning_rate": 1.6140350877192984e-05,
428
+ "loss": 0.8159,
429
+ "step": 56
430
+ },
431
+ {
432
+ "epoch": 28.5,
433
+ "learning_rate": 1.605263157894737e-05,
434
+ "loss": 0.8077,
435
+ "step": 57
436
+ },
437
+ {
438
+ "epoch": 29.0,
439
+ "learning_rate": 1.5964912280701755e-05,
440
+ "loss": 0.8115,
441
+ "step": 58
442
+ },
443
+ {
444
+ "epoch": 29.5,
445
+ "learning_rate": 1.5877192982456144e-05,
446
+ "loss": 0.7824,
447
+ "step": 59
448
+ },
449
+ {
450
+ "epoch": 30.0,
451
+ "learning_rate": 1.578947368421053e-05,
452
+ "loss": 0.8299,
453
+ "step": 60
454
+ },
455
+ {
456
+ "epoch": 30.0,
457
+ "eval_accuracy_dropoff": 0.306841900043579,
458
+ "eval_accuracy_undropoff": 0.9597373870247777,
459
+ "eval_accuracy_unlabeled": NaN,
460
+ "eval_iou_dropoff": 0.1934366306340696,
461
+ "eval_iou_undropoff": 0.9317996330434056,
462
+ "eval_iou_unlabeled": 0.0,
463
+ "eval_loss": 0.7386532425880432,
464
+ "eval_mean_accuracy": 0.6332896435341784,
465
+ "eval_mean_iou": 0.37507875455915846,
466
+ "eval_overall_accuracy": 0.9325904846191406,
467
+ "eval_runtime": 3.129,
468
+ "eval_samples_per_second": 6.392,
469
+ "eval_steps_per_second": 0.639,
470
+ "step": 60
471
+ },
472
+ {
473
+ "epoch": 30.5,
474
+ "learning_rate": 1.5701754385964915e-05,
475
+ "loss": 0.7911,
476
+ "step": 61
477
+ },
478
+ {
479
+ "epoch": 31.0,
480
+ "learning_rate": 1.56140350877193e-05,
481
+ "loss": 0.7694,
482
+ "step": 62
483
+ },
484
+ {
485
+ "epoch": 31.5,
486
+ "learning_rate": 1.5526315789473686e-05,
487
+ "loss": 0.7954,
488
+ "step": 63
489
+ },
490
+ {
491
+ "epoch": 32.0,
492
+ "learning_rate": 1.543859649122807e-05,
493
+ "loss": 0.7716,
494
+ "step": 64
495
+ },
496
+ {
497
+ "epoch": 32.5,
498
+ "learning_rate": 1.535087719298246e-05,
499
+ "loss": 0.7495,
500
+ "step": 65
501
+ },
502
+ {
503
+ "epoch": 33.0,
504
+ "learning_rate": 1.5263157894736846e-05,
505
+ "loss": 0.7472,
506
+ "step": 66
507
+ },
508
+ {
509
+ "epoch": 33.5,
510
+ "learning_rate": 1.517543859649123e-05,
511
+ "loss": 0.7637,
512
+ "step": 67
513
+ },
514
+ {
515
+ "epoch": 34.0,
516
+ "learning_rate": 1.5087719298245615e-05,
517
+ "loss": 0.7548,
518
+ "step": 68
519
+ },
520
+ {
521
+ "epoch": 34.5,
522
+ "learning_rate": 1.5000000000000002e-05,
523
+ "loss": 0.7478,
524
+ "step": 69
525
+ },
526
+ {
527
+ "epoch": 35.0,
528
+ "learning_rate": 1.4912280701754388e-05,
529
+ "loss": 0.7518,
530
+ "step": 70
531
+ },
532
+ {
533
+ "epoch": 35.0,
534
+ "eval_accuracy_dropoff": 0.2960801853253515,
535
+ "eval_accuracy_undropoff": 0.9683274343591943,
536
+ "eval_accuracy_unlabeled": NaN,
537
+ "eval_iou_dropoff": 0.1975453738560891,
538
+ "eval_iou_undropoff": 0.9397214711056545,
539
+ "eval_iou_unlabeled": 0.0,
540
+ "eval_loss": 0.6810304522514343,
541
+ "eval_mean_accuracy": 0.6322038098422729,
542
+ "eval_mean_iou": 0.3790889483205812,
543
+ "eval_overall_accuracy": 0.9403759002685547,
544
+ "eval_runtime": 2.9719,
545
+ "eval_samples_per_second": 6.73,
546
+ "eval_steps_per_second": 0.673,
547
+ "step": 70
548
+ },
549
+ {
550
+ "epoch": 35.5,
551
+ "learning_rate": 1.4824561403508773e-05,
552
+ "loss": 0.7382,
553
+ "step": 71
554
+ },
555
+ {
556
+ "epoch": 36.0,
557
+ "learning_rate": 1.4736842105263159e-05,
558
+ "loss": 0.7568,
559
+ "step": 72
560
+ },
561
+ {
562
+ "epoch": 36.5,
563
+ "learning_rate": 1.4649122807017544e-05,
564
+ "loss": 0.7411,
565
+ "step": 73
566
+ },
567
+ {
568
+ "epoch": 37.0,
569
+ "learning_rate": 1.4561403508771931e-05,
570
+ "loss": 0.6921,
571
+ "step": 74
572
+ },
573
+ {
574
+ "epoch": 37.5,
575
+ "learning_rate": 1.4473684210526317e-05,
576
+ "loss": 0.7174,
577
+ "step": 75
578
+ },
579
+ {
580
+ "epoch": 38.0,
581
+ "learning_rate": 1.4385964912280704e-05,
582
+ "loss": 0.7151,
583
+ "step": 76
584
+ },
585
+ {
586
+ "epoch": 38.5,
587
+ "learning_rate": 1.429824561403509e-05,
588
+ "loss": 0.73,
589
+ "step": 77
590
+ },
591
+ {
592
+ "epoch": 39.0,
593
+ "learning_rate": 1.4210526315789475e-05,
594
+ "loss": 0.7098,
595
+ "step": 78
596
+ },
597
+ {
598
+ "epoch": 39.5,
599
+ "learning_rate": 1.412280701754386e-05,
600
+ "loss": 0.6912,
601
+ "step": 79
602
+ },
603
+ {
604
+ "epoch": 40.0,
605
+ "learning_rate": 1.4035087719298246e-05,
606
+ "loss": 0.6943,
607
+ "step": 80
608
+ },
609
+ {
610
+ "epoch": 40.0,
611
+ "eval_accuracy_dropoff": 0.2411156219179339,
612
+ "eval_accuracy_undropoff": 0.9726190748643999,
613
+ "eval_accuracy_unlabeled": NaN,
614
+ "eval_iou_dropoff": 0.1691189482591643,
615
+ "eval_iou_undropoff": 0.941739253422002,
616
+ "eval_iou_unlabeled": 0.0,
617
+ "eval_loss": 0.6321617364883423,
618
+ "eval_mean_accuracy": 0.6068673483911668,
619
+ "eval_mean_iou": 0.3702860672270554,
620
+ "eval_overall_accuracy": 0.9422037124633789,
621
+ "eval_runtime": 2.9083,
622
+ "eval_samples_per_second": 6.877,
623
+ "eval_steps_per_second": 0.688,
624
+ "step": 80
625
+ },
626
+ {
627
+ "epoch": 40.5,
628
+ "learning_rate": 1.3947368421052631e-05,
629
+ "loss": 0.7307,
630
+ "step": 81
631
+ },
632
+ {
633
+ "epoch": 41.0,
634
+ "learning_rate": 1.385964912280702e-05,
635
+ "loss": 0.6983,
636
+ "step": 82
637
+ },
638
+ {
639
+ "epoch": 41.5,
640
+ "learning_rate": 1.3771929824561406e-05,
641
+ "loss": 0.6995,
642
+ "step": 83
643
+ },
644
+ {
645
+ "epoch": 42.0,
646
+ "learning_rate": 1.3684210526315791e-05,
647
+ "loss": 0.6687,
648
+ "step": 84
649
+ },
650
+ {
651
+ "epoch": 42.5,
652
+ "learning_rate": 1.3596491228070177e-05,
653
+ "loss": 0.672,
654
+ "step": 85
655
+ },
656
+ {
657
+ "epoch": 43.0,
658
+ "learning_rate": 1.3508771929824562e-05,
659
+ "loss": 0.6712,
660
+ "step": 86
661
+ },
662
+ {
663
+ "epoch": 43.5,
664
+ "learning_rate": 1.3421052631578948e-05,
665
+ "loss": 0.7337,
666
+ "step": 87
667
+ },
668
+ {
669
+ "epoch": 44.0,
670
+ "learning_rate": 1.3333333333333333e-05,
671
+ "loss": 0.6606,
672
+ "step": 88
673
+ },
674
+ {
675
+ "epoch": 44.5,
676
+ "learning_rate": 1.3245614035087719e-05,
677
+ "loss": 0.7239,
678
+ "step": 89
679
+ },
680
+ {
681
+ "epoch": 45.0,
682
+ "learning_rate": 1.3157894736842108e-05,
683
+ "loss": 0.6617,
684
+ "step": 90
685
+ },
686
+ {
687
+ "epoch": 45.0,
688
+ "eval_accuracy_dropoff": 0.27343287690084633,
689
+ "eval_accuracy_undropoff": 0.974586483073742,
690
+ "eval_accuracy_unlabeled": NaN,
691
+ "eval_iou_dropoff": 0.1892465949138013,
692
+ "eval_iou_undropoff": 0.9448538377714254,
693
+ "eval_iou_unlabeled": 0.0,
694
+ "eval_loss": 0.6071122884750366,
695
+ "eval_mean_accuracy": 0.6240096799872942,
696
+ "eval_mean_iou": 0.3780334775617422,
697
+ "eval_overall_accuracy": 0.9454330444335938,
698
+ "eval_runtime": 3.1015,
699
+ "eval_samples_per_second": 6.448,
700
+ "eval_steps_per_second": 0.645,
701
+ "step": 90
702
+ },
703
+ {
704
+ "epoch": 45.5,
705
+ "learning_rate": 1.3070175438596493e-05,
706
+ "loss": 0.6397,
707
+ "step": 91
708
+ },
709
+ {
710
+ "epoch": 46.0,
711
+ "learning_rate": 1.2982456140350879e-05,
712
+ "loss": 0.68,
713
+ "step": 92
714
+ },
715
+ {
716
+ "epoch": 46.5,
717
+ "learning_rate": 1.2894736842105264e-05,
718
+ "loss": 0.6642,
719
+ "step": 93
720
+ },
721
+ {
722
+ "epoch": 47.0,
723
+ "learning_rate": 1.280701754385965e-05,
724
+ "loss": 0.6341,
725
+ "step": 94
726
+ },
727
+ {
728
+ "epoch": 47.5,
729
+ "learning_rate": 1.2719298245614035e-05,
730
+ "loss": 0.6783,
731
+ "step": 95
732
+ },
733
+ {
734
+ "epoch": 48.0,
735
+ "learning_rate": 1.263157894736842e-05,
736
+ "loss": 0.6371,
737
+ "step": 96
738
+ },
739
+ {
740
+ "epoch": 48.5,
741
+ "learning_rate": 1.2543859649122808e-05,
742
+ "loss": 0.6431,
743
+ "step": 97
744
+ },
745
+ {
746
+ "epoch": 49.0,
747
+ "learning_rate": 1.2456140350877195e-05,
748
+ "loss": 0.6393,
749
+ "step": 98
750
+ },
751
+ {
752
+ "epoch": 49.5,
753
+ "learning_rate": 1.236842105263158e-05,
754
+ "loss": 0.6309,
755
+ "step": 99
756
+ },
757
+ {
758
+ "epoch": 50.0,
759
+ "learning_rate": 1.2280701754385966e-05,
760
+ "loss": 0.634,
761
+ "step": 100
762
+ },
763
+ {
764
+ "epoch": 50.0,
765
+ "eval_accuracy_dropoff": 0.24070735567329526,
766
+ "eval_accuracy_undropoff": 0.9804538810340934,
767
+ "eval_accuracy_unlabeled": NaN,
768
+ "eval_iou_dropoff": 0.18019759818954179,
769
+ "eval_iou_undropoff": 0.9494230040635037,
770
+ "eval_iou_unlabeled": 0.0,
771
+ "eval_loss": 0.5931987166404724,
772
+ "eval_mean_accuracy": 0.6105806183536944,
773
+ "eval_mean_iou": 0.3765402007510152,
774
+ "eval_overall_accuracy": 0.9496957778930664,
775
+ "eval_runtime": 2.967,
776
+ "eval_samples_per_second": 6.741,
777
+ "eval_steps_per_second": 0.674,
778
+ "step": 100
779
+ },
780
+ {
781
+ "epoch": 50.5,
782
+ "learning_rate": 1.2192982456140352e-05,
783
+ "loss": 0.6392,
784
+ "step": 101
785
+ },
786
+ {
787
+ "epoch": 51.0,
788
+ "learning_rate": 1.2105263157894737e-05,
789
+ "loss": 0.6654,
790
+ "step": 102
791
+ },
792
+ {
793
+ "epoch": 51.5,
794
+ "learning_rate": 1.2017543859649124e-05,
795
+ "loss": 0.6101,
796
+ "step": 103
797
+ },
798
+ {
799
+ "epoch": 52.0,
800
+ "learning_rate": 1.192982456140351e-05,
801
+ "loss": 0.6192,
802
+ "step": 104
803
+ },
804
+ {
805
+ "epoch": 52.5,
806
+ "learning_rate": 1.1842105263157895e-05,
807
+ "loss": 0.6381,
808
+ "step": 105
809
+ },
810
+ {
811
+ "epoch": 53.0,
812
+ "learning_rate": 1.1754385964912282e-05,
813
+ "loss": 0.6018,
814
+ "step": 106
815
+ },
816
+ {
817
+ "epoch": 53.5,
818
+ "learning_rate": 1.1666666666666668e-05,
819
+ "loss": 0.6106,
820
+ "step": 107
821
+ },
822
+ {
823
+ "epoch": 54.0,
824
+ "learning_rate": 1.1578947368421053e-05,
825
+ "loss": 0.6486,
826
+ "step": 108
827
+ },
828
+ {
829
+ "epoch": 54.5,
830
+ "learning_rate": 1.1491228070175439e-05,
831
+ "loss": 0.6192,
832
+ "step": 109
833
+ },
834
+ {
835
+ "epoch": 55.0,
836
+ "learning_rate": 1.1403508771929826e-05,
837
+ "loss": 0.6157,
838
+ "step": 110
839
+ },
840
+ {
841
+ "epoch": 55.0,
842
+ "eval_accuracy_dropoff": 0.32805798298126104,
843
+ "eval_accuracy_undropoff": 0.9795322679026485,
844
+ "eval_accuracy_unlabeled": NaN,
845
+ "eval_iou_dropoff": 0.24251662501229285,
846
+ "eval_iou_undropoff": 0.9520137977134819,
847
+ "eval_iou_unlabeled": 0.0,
848
+ "eval_loss": 0.5828702449798584,
849
+ "eval_mean_accuracy": 0.6537951254419547,
850
+ "eval_mean_iou": 0.39817680757525825,
851
+ "eval_overall_accuracy": 0.9524444580078125,
852
+ "eval_runtime": 2.9135,
853
+ "eval_samples_per_second": 6.865,
854
+ "eval_steps_per_second": 0.686,
855
+ "step": 110
856
+ },
857
+ {
858
+ "epoch": 55.5,
859
+ "learning_rate": 1.1315789473684212e-05,
860
+ "loss": 0.6085,
861
+ "step": 111
862
+ },
863
+ {
864
+ "epoch": 56.0,
865
+ "learning_rate": 1.1228070175438597e-05,
866
+ "loss": 0.5901,
867
+ "step": 112
868
+ },
869
+ {
870
+ "epoch": 56.5,
871
+ "learning_rate": 1.1140350877192983e-05,
872
+ "loss": 0.6228,
873
+ "step": 113
874
+ },
875
+ {
876
+ "epoch": 57.0,
877
+ "learning_rate": 1.105263157894737e-05,
878
+ "loss": 0.5944,
879
+ "step": 114
880
+ },
881
+ {
882
+ "epoch": 57.5,
883
+ "learning_rate": 1.0964912280701755e-05,
884
+ "loss": 0.6076,
885
+ "step": 115
886
+ },
887
+ {
888
+ "epoch": 58.0,
889
+ "learning_rate": 1.0877192982456142e-05,
890
+ "loss": 0.5844,
891
+ "step": 116
892
+ },
893
+ {
894
+ "epoch": 58.5,
895
+ "learning_rate": 1.0789473684210528e-05,
896
+ "loss": 0.5846,
897
+ "step": 117
898
+ },
899
+ {
900
+ "epoch": 59.0,
901
+ "learning_rate": 1.0701754385964913e-05,
902
+ "loss": 0.6332,
903
+ "step": 118
904
+ },
905
+ {
906
+ "epoch": 59.5,
907
+ "learning_rate": 1.0614035087719299e-05,
908
+ "loss": 0.5968,
909
+ "step": 119
910
+ },
911
+ {
912
+ "epoch": 60.0,
913
+ "learning_rate": 1.0526315789473684e-05,
914
+ "loss": 0.5814,
915
+ "step": 120
916
+ },
917
+ {
918
+ "epoch": 60.0,
919
+ "eval_accuracy_dropoff": 0.3607972659923393,
920
+ "eval_accuracy_undropoff": 0.9790086738303464,
921
+ "eval_accuracy_unlabeled": NaN,
922
+ "eval_iou_dropoff": 0.25863605423162545,
923
+ "eval_iou_undropoff": 0.9527589478644293,
924
+ "eval_iou_unlabeled": 0.0,
925
+ "eval_loss": 0.5707733035087585,
926
+ "eval_mean_accuracy": 0.6699029699113428,
927
+ "eval_mean_iou": 0.4037983340320182,
928
+ "eval_overall_accuracy": 0.9533039093017578,
929
+ "eval_runtime": 2.8254,
930
+ "eval_samples_per_second": 7.079,
931
+ "eval_steps_per_second": 0.708,
932
+ "step": 120
933
+ }
934
+ ],
935
+ "max_steps": 240,
936
+ "num_train_epochs": 120,
937
+ "total_flos": 6.107874830843904e+16,
938
+ "trial_name": null,
939
+ "trial_params": null
940
+ }
checkpoint-120/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ff0ba26522d947dfbd4b01b70efdc53619411a3a479d5339da8c89c9a1f71dbd
3
+ size 4219
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:fb1d892e1d5ec4bb9d58d904de05ef40a0b98eb63fa433a74838ef5873be602d
3
  size 14951629
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2f0fc8bbb23c0420a3465abe203545dbf9fb563bede6e2fd8babb249ba6641f4
3
  size 14951629
runs/Feb12_06-52-20_robovision.csres.utexas.edu/events.out.tfevents.1707742347.robovision.csres.utexas.edu.2754155.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f806f3b98c30c109a3e4a995330432d8f9ab8671ac29f6a66dac4f5a69c25c7e
3
- size 23332
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ad97b22aad76d9370cb5eb12216abc2bd7cac4bee7ecf066bd7702d6284742a0
3
+ size 32608