emvisee commited on
Commit
897e33e
·
verified ·
1 Parent(s): b83dd51

End of training

Browse files
Files changed (5) hide show
  1. README.md +3 -1
  2. all_results.json +54 -0
  3. test_results.json +49 -0
  4. train_results.json +8 -0
  5. trainer_state.json +612 -0
README.md CHANGED
@@ -3,6 +3,8 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: PekingU/rtdetr_r50vd
5
  tags:
 
 
6
  - generated_from_trainer
7
  model-index:
8
  - name: suas-2025-rtdetr-finetuned-b8-lr3e-5
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # suas-2025-rtdetr-finetuned-b8-lr3e-5
16
 
17
- This model is a fine-tuned version of [PekingU/rtdetr_r50vd](https://huggingface.co/PekingU/rtdetr_r50vd) on an unknown dataset.
18
  It achieves the following results on the evaluation set:
19
  - Loss: 6.4174
20
  - Map: 0.6051
 
3
  license: apache-2.0
4
  base_model: PekingU/rtdetr_r50vd
5
  tags:
6
+ - object-detection
7
+ - vision
8
  - generated_from_trainer
9
  model-index:
10
  - name: suas-2025-rtdetr-finetuned-b8-lr3e-5
 
16
 
17
  # suas-2025-rtdetr-finetuned-b8-lr3e-5
18
 
19
+ This model is a fine-tuned version of [PekingU/rtdetr_r50vd](https://huggingface.co/PekingU/rtdetr_r50vd) on the mfly-auton/suas-2025-synthetic-data dataset.
20
  It achieves the following results on the evaluation set:
21
  - Loss: 6.4174
22
  - Map: 0.6051
all_results.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "test_loss": 5.249497413635254,
4
+ "test_map": 0.6926,
5
+ "test_map_50": 0.782,
6
+ "test_map_75": 0.7759,
7
+ "test_map_baseball-bat": 0.7144,
8
+ "test_map_basketball": 0.8079,
9
+ "test_map_car": -1.0,
10
+ "test_map_football": 0.7354,
11
+ "test_map_human": 0.8815,
12
+ "test_map_large": 0.69,
13
+ "test_map_luggage": 0.6487,
14
+ "test_map_mattress": 0.9218,
15
+ "test_map_medium": 0.6594,
16
+ "test_map_motorcycle": 0.8764,
17
+ "test_map_skis": 0.8277,
18
+ "test_map_small": 0.716,
19
+ "test_map_snowboard": 0.3425,
20
+ "test_map_soccer-ball": 0.7573,
21
+ "test_map_stop-sign": 0.5775,
22
+ "test_map_tennis-racket": 0.3563,
23
+ "test_map_umbrella": 0.44,
24
+ "test_map_volleyball": 0.8094,
25
+ "test_mar_1": 0.7429,
26
+ "test_mar_10": 0.8961,
27
+ "test_mar_100": 0.9058,
28
+ "test_mar_100_baseball-bat": 0.8322,
29
+ "test_mar_100_basketball": 0.8568,
30
+ "test_mar_100_car": -1.0,
31
+ "test_mar_100_football": 0.8332,
32
+ "test_mar_100_human": 0.9337,
33
+ "test_mar_100_luggage": 0.9392,
34
+ "test_mar_100_mattress": 0.9712,
35
+ "test_mar_100_motorcycle": 0.9261,
36
+ "test_mar_100_skis": 0.9781,
37
+ "test_mar_100_snowboard": 0.9661,
38
+ "test_mar_100_soccer-ball": 0.8701,
39
+ "test_mar_100_stop-sign": 0.9512,
40
+ "test_mar_100_tennis-racket": 0.9379,
41
+ "test_mar_100_umbrella": 0.792,
42
+ "test_mar_100_volleyball": 0.8933,
43
+ "test_mar_large": 0.9473,
44
+ "test_mar_medium": 0.8937,
45
+ "test_mar_small": 0.8053,
46
+ "test_runtime": 69.262,
47
+ "test_samples_per_second": 21.657,
48
+ "test_steps_per_second": 0.679,
49
+ "total_flos": 2.2065014071296e+19,
50
+ "train_loss": 6.704501674107143,
51
+ "train_runtime": 3543.7026,
52
+ "train_samples_per_second": 19.753,
53
+ "train_steps_per_second": 2.469
54
+ }
test_results.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "test_loss": 5.249497413635254,
4
+ "test_map": 0.6926,
5
+ "test_map_50": 0.782,
6
+ "test_map_75": 0.7759,
7
+ "test_map_baseball-bat": 0.7144,
8
+ "test_map_basketball": 0.8079,
9
+ "test_map_car": -1.0,
10
+ "test_map_football": 0.7354,
11
+ "test_map_human": 0.8815,
12
+ "test_map_large": 0.69,
13
+ "test_map_luggage": 0.6487,
14
+ "test_map_mattress": 0.9218,
15
+ "test_map_medium": 0.6594,
16
+ "test_map_motorcycle": 0.8764,
17
+ "test_map_skis": 0.8277,
18
+ "test_map_small": 0.716,
19
+ "test_map_snowboard": 0.3425,
20
+ "test_map_soccer-ball": 0.7573,
21
+ "test_map_stop-sign": 0.5775,
22
+ "test_map_tennis-racket": 0.3563,
23
+ "test_map_umbrella": 0.44,
24
+ "test_map_volleyball": 0.8094,
25
+ "test_mar_1": 0.7429,
26
+ "test_mar_10": 0.8961,
27
+ "test_mar_100": 0.9058,
28
+ "test_mar_100_baseball-bat": 0.8322,
29
+ "test_mar_100_basketball": 0.8568,
30
+ "test_mar_100_car": -1.0,
31
+ "test_mar_100_football": 0.8332,
32
+ "test_mar_100_human": 0.9337,
33
+ "test_mar_100_luggage": 0.9392,
34
+ "test_mar_100_mattress": 0.9712,
35
+ "test_mar_100_motorcycle": 0.9261,
36
+ "test_mar_100_skis": 0.9781,
37
+ "test_mar_100_snowboard": 0.9661,
38
+ "test_mar_100_soccer-ball": 0.8701,
39
+ "test_mar_100_stop-sign": 0.9512,
40
+ "test_mar_100_tennis-racket": 0.9379,
41
+ "test_mar_100_umbrella": 0.792,
42
+ "test_mar_100_volleyball": 0.8933,
43
+ "test_mar_large": 0.9473,
44
+ "test_mar_medium": 0.8937,
45
+ "test_mar_small": 0.8053,
46
+ "test_runtime": 69.262,
47
+ "test_samples_per_second": 21.657,
48
+ "test_steps_per_second": 0.679
49
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 2.2065014071296e+19,
4
+ "train_loss": 6.704501674107143,
5
+ "train_runtime": 3543.7026,
6
+ "train_samples_per_second": 19.753,
7
+ "train_steps_per_second": 2.469
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,612 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.6942,
3
+ "best_model_checkpoint": "rtdetr-finetuned/checkpoint-875",
4
+ "epoch": 10.0,
5
+ "eval_steps": 500,
6
+ "global_step": 8750,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 1.0,
13
+ "grad_norm": 29.419410705566406,
14
+ "learning_rate": 2.9794285714285713e-05,
15
+ "loss": 17.7551,
16
+ "step": 875
17
+ },
18
+ {
19
+ "epoch": 1.0,
20
+ "eval_loss": 5.250951290130615,
21
+ "eval_map": 0.6942,
22
+ "eval_map_50": 0.7838,
23
+ "eval_map_75": 0.7782,
24
+ "eval_map_baseball-bat": 0.6964,
25
+ "eval_map_basketball": 0.7915,
26
+ "eval_map_car": -1.0,
27
+ "eval_map_football": 0.7303,
28
+ "eval_map_human": 0.8746,
29
+ "eval_map_large": 0.7363,
30
+ "eval_map_luggage": 0.6526,
31
+ "eval_map_mattress": 0.9142,
32
+ "eval_map_medium": 0.6405,
33
+ "eval_map_motorcycle": 0.8867,
34
+ "eval_map_skis": 0.8376,
35
+ "eval_map_small": 0.7254,
36
+ "eval_map_snowboard": 0.3429,
37
+ "eval_map_soccer-ball": 0.7454,
38
+ "eval_map_stop-sign": 0.581,
39
+ "eval_map_tennis-racket": 0.3918,
40
+ "eval_map_umbrella": 0.4713,
41
+ "eval_map_volleyball": 0.8027,
42
+ "eval_mar_1": 0.7413,
43
+ "eval_mar_10": 0.8942,
44
+ "eval_mar_100": 0.9051,
45
+ "eval_mar_100_baseball-bat": 0.8173,
46
+ "eval_mar_100_basketball": 0.8415,
47
+ "eval_mar_100_car": -1.0,
48
+ "eval_mar_100_football": 0.8192,
49
+ "eval_mar_100_human": 0.9281,
50
+ "eval_mar_100_luggage": 0.9372,
51
+ "eval_mar_100_mattress": 0.9711,
52
+ "eval_mar_100_motorcycle": 0.9311,
53
+ "eval_mar_100_skis": 0.9594,
54
+ "eval_mar_100_snowboard": 0.9742,
55
+ "eval_mar_100_soccer-ball": 0.8686,
56
+ "eval_mar_100_stop-sign": 0.9542,
57
+ "eval_mar_100_tennis-racket": 0.944,
58
+ "eval_mar_100_umbrella": 0.8414,
59
+ "eval_mar_100_volleyball": 0.8846,
60
+ "eval_mar_large": 0.9496,
61
+ "eval_mar_medium": 0.8996,
62
+ "eval_mar_small": 0.8089,
63
+ "eval_runtime": 84.6749,
64
+ "eval_samples_per_second": 17.715,
65
+ "eval_steps_per_second": 0.555,
66
+ "step": 875
67
+ },
68
+ {
69
+ "epoch": 2.0,
70
+ "grad_norm": 39.169166564941406,
71
+ "learning_rate": 2.915224734506273e-05,
72
+ "loss": 6.6319,
73
+ "step": 1750
74
+ },
75
+ {
76
+ "epoch": 2.0,
77
+ "eval_loss": 5.667089939117432,
78
+ "eval_map": 0.6915,
79
+ "eval_map_50": 0.7796,
80
+ "eval_map_75": 0.7715,
81
+ "eval_map_baseball-bat": 0.6895,
82
+ "eval_map_basketball": 0.5032,
83
+ "eval_map_car": -1.0,
84
+ "eval_map_football": 0.7195,
85
+ "eval_map_human": 0.9186,
86
+ "eval_map_large": 0.8104,
87
+ "eval_map_luggage": 0.6184,
88
+ "eval_map_mattress": 0.895,
89
+ "eval_map_medium": 0.6468,
90
+ "eval_map_motorcycle": 0.901,
91
+ "eval_map_skis": 0.9222,
92
+ "eval_map_small": 0.6499,
93
+ "eval_map_snowboard": 0.2897,
94
+ "eval_map_soccer-ball": 0.7717,
95
+ "eval_map_stop-sign": 0.6219,
96
+ "eval_map_tennis-racket": 0.2013,
97
+ "eval_map_umbrella": 0.8724,
98
+ "eval_map_volleyball": 0.7562,
99
+ "eval_mar_1": 0.7323,
100
+ "eval_mar_10": 0.879,
101
+ "eval_mar_100": 0.892,
102
+ "eval_mar_100_baseball-bat": 0.7787,
103
+ "eval_mar_100_basketball": 0.6076,
104
+ "eval_mar_100_car": -1.0,
105
+ "eval_mar_100_football": 0.7926,
106
+ "eval_mar_100_human": 0.9593,
107
+ "eval_mar_100_luggage": 0.9288,
108
+ "eval_mar_100_mattress": 0.9984,
109
+ "eval_mar_100_motorcycle": 0.9643,
110
+ "eval_mar_100_skis": 0.9634,
111
+ "eval_mar_100_snowboard": 0.9699,
112
+ "eval_mar_100_soccer-ball": 0.8578,
113
+ "eval_mar_100_stop-sign": 0.9584,
114
+ "eval_mar_100_tennis-racket": 0.8995,
115
+ "eval_mar_100_umbrella": 0.9458,
116
+ "eval_mar_100_volleyball": 0.864,
117
+ "eval_mar_large": 0.9841,
118
+ "eval_mar_medium": 0.9168,
119
+ "eval_mar_small": 0.726,
120
+ "eval_runtime": 68.6398,
121
+ "eval_samples_per_second": 21.853,
122
+ "eval_steps_per_second": 0.685,
123
+ "step": 1750
124
+ },
125
+ {
126
+ "epoch": 3.0,
127
+ "grad_norm": 26.32632827758789,
128
+ "learning_rate": 2.669166949131109e-05,
129
+ "loss": 6.3092,
130
+ "step": 2625
131
+ },
132
+ {
133
+ "epoch": 3.0,
134
+ "eval_loss": 5.378567695617676,
135
+ "eval_map": 0.6861,
136
+ "eval_map_50": 0.7797,
137
+ "eval_map_75": 0.7703,
138
+ "eval_map_baseball-bat": 0.7854,
139
+ "eval_map_basketball": 0.5392,
140
+ "eval_map_car": -1.0,
141
+ "eval_map_football": 0.6451,
142
+ "eval_map_human": 0.9165,
143
+ "eval_map_large": 0.8039,
144
+ "eval_map_luggage": 0.6485,
145
+ "eval_map_mattress": 0.9011,
146
+ "eval_map_medium": 0.6505,
147
+ "eval_map_motorcycle": 0.9194,
148
+ "eval_map_skis": 0.9354,
149
+ "eval_map_small": 0.6794,
150
+ "eval_map_snowboard": 0.147,
151
+ "eval_map_soccer-ball": 0.8089,
152
+ "eval_map_stop-sign": 0.7102,
153
+ "eval_map_tennis-racket": 0.0527,
154
+ "eval_map_umbrella": 0.8651,
155
+ "eval_map_volleyball": 0.7312,
156
+ "eval_mar_1": 0.7008,
157
+ "eval_mar_10": 0.8593,
158
+ "eval_mar_100": 0.8895,
159
+ "eval_mar_100_baseball-bat": 0.863,
160
+ "eval_mar_100_basketball": 0.5953,
161
+ "eval_mar_100_car": -1.0,
162
+ "eval_mar_100_football": 0.7011,
163
+ "eval_mar_100_human": 0.9458,
164
+ "eval_mar_100_luggage": 0.9354,
165
+ "eval_mar_100_mattress": 0.9986,
166
+ "eval_mar_100_motorcycle": 0.9592,
167
+ "eval_mar_100_skis": 0.9827,
168
+ "eval_mar_100_snowboard": 0.9815,
169
+ "eval_mar_100_soccer-ball": 0.8686,
170
+ "eval_mar_100_stop-sign": 0.9732,
171
+ "eval_mar_100_tennis-racket": 0.8694,
172
+ "eval_mar_100_umbrella": 0.9418,
173
+ "eval_mar_100_volleyball": 0.838,
174
+ "eval_mar_large": 0.9851,
175
+ "eval_mar_medium": 0.9294,
176
+ "eval_mar_small": 0.738,
177
+ "eval_runtime": 70.9777,
178
+ "eval_samples_per_second": 21.133,
179
+ "eval_steps_per_second": 0.662,
180
+ "step": 2625
181
+ },
182
+ {
183
+ "epoch": 4.0,
184
+ "grad_norm": 40.14275360107422,
185
+ "learning_rate": 2.291434900526519e-05,
186
+ "loss": 5.8652,
187
+ "step": 3500
188
+ },
189
+ {
190
+ "epoch": 4.0,
191
+ "eval_loss": 5.853817939758301,
192
+ "eval_map": 0.6465,
193
+ "eval_map_50": 0.749,
194
+ "eval_map_75": 0.7306,
195
+ "eval_map_baseball-bat": 0.6314,
196
+ "eval_map_basketball": 0.4033,
197
+ "eval_map_car": -1.0,
198
+ "eval_map_football": 0.6013,
199
+ "eval_map_human": 0.9291,
200
+ "eval_map_large": 0.6801,
201
+ "eval_map_luggage": 0.729,
202
+ "eval_map_mattress": 0.64,
203
+ "eval_map_medium": 0.6486,
204
+ "eval_map_motorcycle": 0.9361,
205
+ "eval_map_skis": 0.8382,
206
+ "eval_map_small": 0.6014,
207
+ "eval_map_snowboard": 0.1472,
208
+ "eval_map_soccer-ball": 0.7465,
209
+ "eval_map_stop-sign": 0.803,
210
+ "eval_map_tennis-racket": 0.0352,
211
+ "eval_map_umbrella": 0.8569,
212
+ "eval_map_volleyball": 0.7539,
213
+ "eval_mar_1": 0.6644,
214
+ "eval_mar_10": 0.8092,
215
+ "eval_mar_100": 0.8383,
216
+ "eval_mar_100_baseball-bat": 0.7432,
217
+ "eval_mar_100_basketball": 0.446,
218
+ "eval_mar_100_car": -1.0,
219
+ "eval_mar_100_football": 0.6536,
220
+ "eval_mar_100_human": 0.9609,
221
+ "eval_mar_100_luggage": 0.9151,
222
+ "eval_mar_100_mattress": 0.9942,
223
+ "eval_mar_100_motorcycle": 0.9592,
224
+ "eval_mar_100_skis": 0.9396,
225
+ "eval_mar_100_snowboard": 0.893,
226
+ "eval_mar_100_soccer-ball": 0.8059,
227
+ "eval_mar_100_stop-sign": 0.952,
228
+ "eval_mar_100_tennis-racket": 0.7725,
229
+ "eval_mar_100_umbrella": 0.892,
230
+ "eval_mar_100_volleyball": 0.8091,
231
+ "eval_mar_large": 0.9506,
232
+ "eval_mar_medium": 0.8784,
233
+ "eval_mar_small": 0.6448,
234
+ "eval_runtime": 64.5124,
235
+ "eval_samples_per_second": 23.251,
236
+ "eval_steps_per_second": 0.729,
237
+ "step": 3500
238
+ },
239
+ {
240
+ "epoch": 5.0,
241
+ "grad_norm": 113.89815521240234,
242
+ "learning_rate": 1.8274856272528978e-05,
243
+ "loss": 5.4748,
244
+ "step": 4375
245
+ },
246
+ {
247
+ "epoch": 5.0,
248
+ "eval_loss": 5.892192840576172,
249
+ "eval_map": 0.6416,
250
+ "eval_map_50": 0.7467,
251
+ "eval_map_75": 0.7203,
252
+ "eval_map_baseball-bat": 0.6726,
253
+ "eval_map_basketball": 0.3624,
254
+ "eval_map_car": -1.0,
255
+ "eval_map_football": 0.5819,
256
+ "eval_map_human": 0.9266,
257
+ "eval_map_large": 0.7155,
258
+ "eval_map_luggage": 0.6374,
259
+ "eval_map_mattress": 0.7673,
260
+ "eval_map_medium": 0.6359,
261
+ "eval_map_motorcycle": 0.9372,
262
+ "eval_map_skis": 0.9629,
263
+ "eval_map_small": 0.5884,
264
+ "eval_map_snowboard": 0.0824,
265
+ "eval_map_soccer-ball": 0.7745,
266
+ "eval_map_stop-sign": 0.7208,
267
+ "eval_map_tennis-racket": 0.015,
268
+ "eval_map_umbrella": 0.7927,
269
+ "eval_map_volleyball": 0.7489,
270
+ "eval_mar_1": 0.6543,
271
+ "eval_mar_10": 0.8104,
272
+ "eval_mar_100": 0.8498,
273
+ "eval_mar_100_baseball-bat": 0.7857,
274
+ "eval_mar_100_basketball": 0.4249,
275
+ "eval_mar_100_car": -1.0,
276
+ "eval_mar_100_football": 0.641,
277
+ "eval_mar_100_human": 0.9518,
278
+ "eval_mar_100_luggage": 0.9073,
279
+ "eval_mar_100_mattress": 0.9959,
280
+ "eval_mar_100_motorcycle": 0.9692,
281
+ "eval_mar_100_skis": 0.9861,
282
+ "eval_mar_100_snowboard": 0.9316,
283
+ "eval_mar_100_soccer-ball": 0.8198,
284
+ "eval_mar_100_stop-sign": 0.9508,
285
+ "eval_mar_100_tennis-racket": 0.8404,
286
+ "eval_mar_100_umbrella": 0.8863,
287
+ "eval_mar_100_volleyball": 0.8061,
288
+ "eval_mar_large": 0.965,
289
+ "eval_mar_medium": 0.895,
290
+ "eval_mar_small": 0.6337,
291
+ "eval_runtime": 67.4079,
292
+ "eval_samples_per_second": 22.253,
293
+ "eval_steps_per_second": 0.697,
294
+ "step": 4375
295
+ },
296
+ {
297
+ "epoch": 6.0,
298
+ "grad_norm": 107.60013580322266,
299
+ "learning_rate": 1.3325915107375438e-05,
300
+ "loss": 5.352,
301
+ "step": 5250
302
+ },
303
+ {
304
+ "epoch": 6.0,
305
+ "eval_loss": 6.314184665679932,
306
+ "eval_map": 0.6409,
307
+ "eval_map_50": 0.741,
308
+ "eval_map_75": 0.7178,
309
+ "eval_map_baseball-bat": 0.7148,
310
+ "eval_map_basketball": 0.4792,
311
+ "eval_map_car": -1.0,
312
+ "eval_map_football": 0.5879,
313
+ "eval_map_human": 0.9101,
314
+ "eval_map_large": 0.6861,
315
+ "eval_map_luggage": 0.5811,
316
+ "eval_map_mattress": 0.8213,
317
+ "eval_map_medium": 0.5958,
318
+ "eval_map_motorcycle": 0.9139,
319
+ "eval_map_skis": 0.9479,
320
+ "eval_map_small": 0.6313,
321
+ "eval_map_snowboard": 0.052,
322
+ "eval_map_soccer-ball": 0.7854,
323
+ "eval_map_stop-sign": 0.6801,
324
+ "eval_map_tennis-racket": 0.0209,
325
+ "eval_map_umbrella": 0.7664,
326
+ "eval_map_volleyball": 0.7112,
327
+ "eval_mar_1": 0.6395,
328
+ "eval_mar_10": 0.7987,
329
+ "eval_mar_100": 0.8411,
330
+ "eval_mar_100_baseball-bat": 0.8051,
331
+ "eval_mar_100_basketball": 0.5386,
332
+ "eval_mar_100_car": -1.0,
333
+ "eval_mar_100_football": 0.6392,
334
+ "eval_mar_100_human": 0.9478,
335
+ "eval_mar_100_luggage": 0.8811,
336
+ "eval_mar_100_mattress": 0.9987,
337
+ "eval_mar_100_motorcycle": 0.96,
338
+ "eval_mar_100_skis": 0.9767,
339
+ "eval_mar_100_snowboard": 0.9105,
340
+ "eval_mar_100_soccer-ball": 0.8321,
341
+ "eval_mar_100_stop-sign": 0.9332,
342
+ "eval_mar_100_tennis-racket": 0.6995,
343
+ "eval_mar_100_umbrella": 0.8798,
344
+ "eval_mar_100_volleyball": 0.7733,
345
+ "eval_mar_large": 0.9544,
346
+ "eval_mar_medium": 0.8595,
347
+ "eval_mar_small": 0.6741,
348
+ "eval_runtime": 65.1215,
349
+ "eval_samples_per_second": 23.034,
350
+ "eval_steps_per_second": 0.722,
351
+ "step": 5250
352
+ },
353
+ {
354
+ "epoch": 7.0,
355
+ "grad_norm": 18.771818161010742,
356
+ "learning_rate": 8.66935435595516e-06,
357
+ "loss": 5.1405,
358
+ "step": 6125
359
+ },
360
+ {
361
+ "epoch": 7.0,
362
+ "eval_loss": 6.370442867279053,
363
+ "eval_map": 0.6464,
364
+ "eval_map_50": 0.7579,
365
+ "eval_map_75": 0.7185,
366
+ "eval_map_baseball-bat": 0.6828,
367
+ "eval_map_basketball": 0.5464,
368
+ "eval_map_car": -1.0,
369
+ "eval_map_football": 0.5058,
370
+ "eval_map_human": 0.9172,
371
+ "eval_map_large": 0.6971,
372
+ "eval_map_luggage": 0.6302,
373
+ "eval_map_mattress": 0.8037,
374
+ "eval_map_medium": 0.6209,
375
+ "eval_map_motorcycle": 0.9339,
376
+ "eval_map_skis": 0.9615,
377
+ "eval_map_small": 0.5862,
378
+ "eval_map_snowboard": 0.0249,
379
+ "eval_map_soccer-ball": 0.7298,
380
+ "eval_map_stop-sign": 0.7449,
381
+ "eval_map_tennis-racket": 0.0025,
382
+ "eval_map_umbrella": 0.8942,
383
+ "eval_map_volleyball": 0.6718,
384
+ "eval_mar_1": 0.6346,
385
+ "eval_mar_10": 0.7737,
386
+ "eval_mar_100": 0.8247,
387
+ "eval_mar_100_baseball-bat": 0.7855,
388
+ "eval_mar_100_basketball": 0.5882,
389
+ "eval_mar_100_car": -1.0,
390
+ "eval_mar_100_football": 0.5697,
391
+ "eval_mar_100_human": 0.9526,
392
+ "eval_mar_100_luggage": 0.9021,
393
+ "eval_mar_100_mattress": 0.9908,
394
+ "eval_mar_100_motorcycle": 0.9643,
395
+ "eval_mar_100_skis": 0.9767,
396
+ "eval_mar_100_snowboard": 0.6942,
397
+ "eval_mar_100_soccer-ball": 0.7859,
398
+ "eval_mar_100_stop-sign": 0.9567,
399
+ "eval_mar_100_tennis-racket": 0.713,
400
+ "eval_mar_100_umbrella": 0.9397,
401
+ "eval_mar_100_volleyball": 0.7272,
402
+ "eval_mar_large": 0.9372,
403
+ "eval_mar_medium": 0.865,
404
+ "eval_mar_small": 0.6346,
405
+ "eval_runtime": 64.5285,
406
+ "eval_samples_per_second": 23.246,
407
+ "eval_steps_per_second": 0.728,
408
+ "step": 6125
409
+ },
410
+ {
411
+ "epoch": 8.0,
412
+ "grad_norm": 24.313987731933594,
413
+ "learning_rate": 4.870494556629035e-06,
414
+ "loss": 4.9059,
415
+ "step": 7000
416
+ },
417
+ {
418
+ "epoch": 8.0,
419
+ "eval_loss": 6.487894535064697,
420
+ "eval_map": 0.6238,
421
+ "eval_map_50": 0.7487,
422
+ "eval_map_75": 0.6799,
423
+ "eval_map_baseball-bat": 0.6155,
424
+ "eval_map_basketball": 0.4736,
425
+ "eval_map_car": -1.0,
426
+ "eval_map_football": 0.4614,
427
+ "eval_map_human": 0.9159,
428
+ "eval_map_large": 0.6895,
429
+ "eval_map_luggage": 0.6436,
430
+ "eval_map_mattress": 0.867,
431
+ "eval_map_medium": 0.6248,
432
+ "eval_map_motorcycle": 0.9482,
433
+ "eval_map_skis": 0.9574,
434
+ "eval_map_small": 0.5369,
435
+ "eval_map_snowboard": 0.0317,
436
+ "eval_map_soccer-ball": 0.6742,
437
+ "eval_map_stop-sign": 0.6387,
438
+ "eval_map_tennis-racket": 0.0037,
439
+ "eval_map_umbrella": 0.8625,
440
+ "eval_map_volleyball": 0.6398,
441
+ "eval_mar_1": 0.6205,
442
+ "eval_mar_10": 0.7655,
443
+ "eval_mar_100": 0.8153,
444
+ "eval_mar_100_baseball-bat": 0.7369,
445
+ "eval_mar_100_basketball": 0.5339,
446
+ "eval_mar_100_car": -1.0,
447
+ "eval_mar_100_football": 0.5228,
448
+ "eval_mar_100_human": 0.9516,
449
+ "eval_mar_100_luggage": 0.8783,
450
+ "eval_mar_100_mattress": 0.9951,
451
+ "eval_mar_100_motorcycle": 0.9714,
452
+ "eval_mar_100_skis": 0.9767,
453
+ "eval_mar_100_snowboard": 0.806,
454
+ "eval_mar_100_soccer-ball": 0.7409,
455
+ "eval_mar_100_stop-sign": 0.9321,
456
+ "eval_mar_100_tennis-racket": 0.7368,
457
+ "eval_mar_100_umbrella": 0.9115,
458
+ "eval_mar_100_volleyball": 0.7199,
459
+ "eval_mar_large": 0.9561,
460
+ "eval_mar_medium": 0.8634,
461
+ "eval_mar_small": 0.5983,
462
+ "eval_runtime": 65.2856,
463
+ "eval_samples_per_second": 22.976,
464
+ "eval_steps_per_second": 0.72,
465
+ "step": 7000
466
+ },
467
+ {
468
+ "epoch": 9.0,
469
+ "grad_norm": 29.279457092285156,
470
+ "learning_rate": 2.3789297716181247e-06,
471
+ "loss": 4.8446,
472
+ "step": 7875
473
+ },
474
+ {
475
+ "epoch": 9.0,
476
+ "eval_loss": 6.516501426696777,
477
+ "eval_map": 0.6064,
478
+ "eval_map_50": 0.734,
479
+ "eval_map_75": 0.6583,
480
+ "eval_map_baseball-bat": 0.598,
481
+ "eval_map_basketball": 0.4926,
482
+ "eval_map_car": -1.0,
483
+ "eval_map_football": 0.4304,
484
+ "eval_map_human": 0.9153,
485
+ "eval_map_large": 0.6766,
486
+ "eval_map_luggage": 0.5701,
487
+ "eval_map_mattress": 0.7636,
488
+ "eval_map_medium": 0.6055,
489
+ "eval_map_motorcycle": 0.9459,
490
+ "eval_map_skis": 0.9583,
491
+ "eval_map_small": 0.5305,
492
+ "eval_map_snowboard": 0.0271,
493
+ "eval_map_soccer-ball": 0.6457,
494
+ "eval_map_stop-sign": 0.6268,
495
+ "eval_map_tennis-racket": 0.0014,
496
+ "eval_map_umbrella": 0.8662,
497
+ "eval_map_volleyball": 0.6484,
498
+ "eval_mar_1": 0.6057,
499
+ "eval_mar_10": 0.7479,
500
+ "eval_mar_100": 0.8021,
501
+ "eval_mar_100_baseball-bat": 0.7249,
502
+ "eval_mar_100_basketball": 0.5547,
503
+ "eval_mar_100_car": -1.0,
504
+ "eval_mar_100_football": 0.4976,
505
+ "eval_mar_100_human": 0.9478,
506
+ "eval_mar_100_luggage": 0.8526,
507
+ "eval_mar_100_mattress": 0.9867,
508
+ "eval_mar_100_motorcycle": 0.9684,
509
+ "eval_mar_100_skis": 0.9762,
510
+ "eval_mar_100_snowboard": 0.7862,
511
+ "eval_mar_100_soccer-ball": 0.7131,
512
+ "eval_mar_100_stop-sign": 0.9338,
513
+ "eval_mar_100_tennis-racket": 0.6518,
514
+ "eval_mar_100_umbrella": 0.914,
515
+ "eval_mar_100_volleyball": 0.7213,
516
+ "eval_mar_large": 0.9489,
517
+ "eval_mar_medium": 0.8441,
518
+ "eval_mar_small": 0.5908,
519
+ "eval_runtime": 64.9236,
520
+ "eval_samples_per_second": 23.104,
521
+ "eval_steps_per_second": 0.724,
522
+ "step": 7875
523
+ },
524
+ {
525
+ "epoch": 10.0,
526
+ "grad_norm": 87.04912567138672,
527
+ "learning_rate": 1.5001632845718623e-06,
528
+ "loss": 4.7656,
529
+ "step": 8750
530
+ },
531
+ {
532
+ "epoch": 10.0,
533
+ "eval_loss": 6.417407512664795,
534
+ "eval_map": 0.6051,
535
+ "eval_map_50": 0.7291,
536
+ "eval_map_75": 0.6597,
537
+ "eval_map_baseball-bat": 0.5841,
538
+ "eval_map_basketball": 0.4357,
539
+ "eval_map_car": -1.0,
540
+ "eval_map_football": 0.4363,
541
+ "eval_map_human": 0.9189,
542
+ "eval_map_large": 0.6723,
543
+ "eval_map_luggage": 0.5995,
544
+ "eval_map_mattress": 0.7881,
545
+ "eval_map_medium": 0.6131,
546
+ "eval_map_motorcycle": 0.9485,
547
+ "eval_map_skis": 0.9582,
548
+ "eval_map_small": 0.5252,
549
+ "eval_map_snowboard": 0.0257,
550
+ "eval_map_soccer-ball": 0.6432,
551
+ "eval_map_stop-sign": 0.5994,
552
+ "eval_map_tennis-racket": 0.0019,
553
+ "eval_map_umbrella": 0.8665,
554
+ "eval_map_volleyball": 0.6647,
555
+ "eval_mar_1": 0.6006,
556
+ "eval_mar_10": 0.7415,
557
+ "eval_mar_100": 0.7959,
558
+ "eval_mar_100_baseball-bat": 0.7219,
559
+ "eval_mar_100_basketball": 0.4955,
560
+ "eval_mar_100_car": -1.0,
561
+ "eval_mar_100_football": 0.5014,
562
+ "eval_mar_100_human": 0.9512,
563
+ "eval_mar_100_luggage": 0.8773,
564
+ "eval_mar_100_mattress": 0.9906,
565
+ "eval_mar_100_motorcycle": 0.9752,
566
+ "eval_mar_100_skis": 0.9782,
567
+ "eval_mar_100_snowboard": 0.777,
568
+ "eval_mar_100_soccer-ball": 0.7154,
569
+ "eval_mar_100_stop-sign": 0.9274,
570
+ "eval_mar_100_tennis-racket": 0.5736,
571
+ "eval_mar_100_umbrella": 0.9243,
572
+ "eval_mar_100_volleyball": 0.7338,
573
+ "eval_mar_large": 0.948,
574
+ "eval_mar_medium": 0.8299,
575
+ "eval_mar_small": 0.5831,
576
+ "eval_runtime": 64.5479,
577
+ "eval_samples_per_second": 23.239,
578
+ "eval_steps_per_second": 0.728,
579
+ "step": 8750
580
+ },
581
+ {
582
+ "epoch": 10.0,
583
+ "step": 8750,
584
+ "total_flos": 2.2065014071296e+19,
585
+ "train_loss": 6.704501674107143,
586
+ "train_runtime": 3543.7026,
587
+ "train_samples_per_second": 19.753,
588
+ "train_steps_per_second": 2.469
589
+ }
590
+ ],
591
+ "logging_steps": 500,
592
+ "max_steps": 8750,
593
+ "num_input_tokens_seen": 0,
594
+ "num_train_epochs": 10,
595
+ "save_steps": 500,
596
+ "stateful_callbacks": {
597
+ "TrainerControl": {
598
+ "args": {
599
+ "should_epoch_stop": false,
600
+ "should_evaluate": false,
601
+ "should_log": false,
602
+ "should_save": true,
603
+ "should_training_stop": true
604
+ },
605
+ "attributes": {}
606
+ }
607
+ },
608
+ "total_flos": 2.2065014071296e+19,
609
+ "train_batch_size": 8,
610
+ "trial_name": null,
611
+ "trial_params": null
612
+ }