emvisee commited on
Commit
4814da5
·
verified ·
1 Parent(s): 821ba50

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-b16-lr1e-5
@@ -14,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  # suas-2025-rtdetr-finetuned-b16-lr1e-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: 3.6523
20
  - Map: 0.8465
 
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-b16-lr1e-5
 
16
 
17
  # suas-2025-rtdetr-finetuned-b16-lr1e-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: 3.6523
22
  - Map: 0.8465
all_results.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "test_loss": 3.641826868057251,
4
+ "test_map": 0.8516,
5
+ "test_map_50": 0.9297,
6
+ "test_map_75": 0.9242,
7
+ "test_map_baseball-bat": 0.8122,
8
+ "test_map_basketball": 0.8229,
9
+ "test_map_car": -1.0,
10
+ "test_map_football": 0.7657,
11
+ "test_map_human": 0.9395,
12
+ "test_map_large": 0.9254,
13
+ "test_map_luggage": 0.8425,
14
+ "test_map_mattress": 0.9607,
15
+ "test_map_medium": 0.8596,
16
+ "test_map_motorcycle": 0.9338,
17
+ "test_map_skis": 0.6855,
18
+ "test_map_small": 0.7653,
19
+ "test_map_snowboard": 0.9849,
20
+ "test_map_soccer-ball": 0.8242,
21
+ "test_map_stop-sign": 0.9766,
22
+ "test_map_tennis-racket": 0.7259,
23
+ "test_map_umbrella": 0.8418,
24
+ "test_map_volleyball": 0.8063,
25
+ "test_mar_1": 0.804,
26
+ "test_mar_10": 0.9269,
27
+ "test_mar_100": 0.9297,
28
+ "test_mar_100_baseball-bat": 0.887,
29
+ "test_mar_100_basketball": 0.901,
30
+ "test_mar_100_car": -1.0,
31
+ "test_mar_100_football": 0.8143,
32
+ "test_mar_100_human": 0.9639,
33
+ "test_mar_100_luggage": 0.9124,
34
+ "test_mar_100_mattress": 0.9924,
35
+ "test_mar_100_motorcycle": 0.9768,
36
+ "test_mar_100_skis": 0.996,
37
+ "test_mar_100_snowboard": 0.9962,
38
+ "test_mar_100_soccer-ball": 0.8698,
39
+ "test_mar_100_stop-sign": 0.994,
40
+ "test_mar_100_tennis-racket": 0.8665,
41
+ "test_mar_100_umbrella": 0.9828,
42
+ "test_mar_100_volleyball": 0.8632,
43
+ "test_mar_large": 0.9933,
44
+ "test_mar_medium": 0.9462,
45
+ "test_mar_small": 0.8267,
46
+ "test_runtime": 72.5857,
47
+ "test_samples_per_second": 20.665,
48
+ "test_steps_per_second": 0.648,
49
+ "total_flos": 2.2065014071296e+19,
50
+ "train_loss": 7.963780066085189,
51
+ "train_runtime": 3254.086,
52
+ "train_samples_per_second": 21.511,
53
+ "train_steps_per_second": 1.346
54
+ }
test_results.json ADDED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "test_loss": 3.641826868057251,
4
+ "test_map": 0.8516,
5
+ "test_map_50": 0.9297,
6
+ "test_map_75": 0.9242,
7
+ "test_map_baseball-bat": 0.8122,
8
+ "test_map_basketball": 0.8229,
9
+ "test_map_car": -1.0,
10
+ "test_map_football": 0.7657,
11
+ "test_map_human": 0.9395,
12
+ "test_map_large": 0.9254,
13
+ "test_map_luggage": 0.8425,
14
+ "test_map_mattress": 0.9607,
15
+ "test_map_medium": 0.8596,
16
+ "test_map_motorcycle": 0.9338,
17
+ "test_map_skis": 0.6855,
18
+ "test_map_small": 0.7653,
19
+ "test_map_snowboard": 0.9849,
20
+ "test_map_soccer-ball": 0.8242,
21
+ "test_map_stop-sign": 0.9766,
22
+ "test_map_tennis-racket": 0.7259,
23
+ "test_map_umbrella": 0.8418,
24
+ "test_map_volleyball": 0.8063,
25
+ "test_mar_1": 0.804,
26
+ "test_mar_10": 0.9269,
27
+ "test_mar_100": 0.9297,
28
+ "test_mar_100_baseball-bat": 0.887,
29
+ "test_mar_100_basketball": 0.901,
30
+ "test_mar_100_car": -1.0,
31
+ "test_mar_100_football": 0.8143,
32
+ "test_mar_100_human": 0.9639,
33
+ "test_mar_100_luggage": 0.9124,
34
+ "test_mar_100_mattress": 0.9924,
35
+ "test_mar_100_motorcycle": 0.9768,
36
+ "test_mar_100_skis": 0.996,
37
+ "test_mar_100_snowboard": 0.9962,
38
+ "test_mar_100_soccer-ball": 0.8698,
39
+ "test_mar_100_stop-sign": 0.994,
40
+ "test_mar_100_tennis-racket": 0.8665,
41
+ "test_mar_100_umbrella": 0.9828,
42
+ "test_mar_100_volleyball": 0.8632,
43
+ "test_mar_large": 0.9933,
44
+ "test_mar_medium": 0.9462,
45
+ "test_mar_small": 0.8267,
46
+ "test_runtime": 72.5857,
47
+ "test_samples_per_second": 20.665,
48
+ "test_steps_per_second": 0.648
49
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 10.0,
3
+ "total_flos": 2.2065014071296e+19,
4
+ "train_loss": 7.963780066085189,
5
+ "train_runtime": 3254.086,
6
+ "train_samples_per_second": 21.511,
7
+ "train_steps_per_second": 1.346
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,612 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.8465,
3
+ "best_model_checkpoint": "rtdetr-finetuned/checkpoint-4380",
4
+ "epoch": 10.0,
5
+ "eval_steps": 500,
6
+ "global_step": 4380,
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": 38.200565338134766,
14
+ "learning_rate": 9.885844748858449e-06,
15
+ "loss": 26.4817,
16
+ "step": 438
17
+ },
18
+ {
19
+ "epoch": 1.0,
20
+ "eval_loss": 7.179637908935547,
21
+ "eval_map": 0.5246,
22
+ "eval_map_50": 0.6136,
23
+ "eval_map_75": 0.5998,
24
+ "eval_map_baseball-bat": 0.3868,
25
+ "eval_map_basketball": 0.1579,
26
+ "eval_map_car": -1.0,
27
+ "eval_map_football": 0.5107,
28
+ "eval_map_human": 0.8625,
29
+ "eval_map_large": 0.685,
30
+ "eval_map_luggage": 0.6942,
31
+ "eval_map_mattress": 0.568,
32
+ "eval_map_medium": 0.4541,
33
+ "eval_map_motorcycle": 0.7128,
34
+ "eval_map_skis": 0.085,
35
+ "eval_map_small": 0.4998,
36
+ "eval_map_snowboard": 0.9532,
37
+ "eval_map_soccer-ball": 0.5658,
38
+ "eval_map_stop-sign": 0.3822,
39
+ "eval_map_tennis-racket": 0.2726,
40
+ "eval_map_umbrella": 0.6033,
41
+ "eval_map_volleyball": 0.5895,
42
+ "eval_mar_1": 0.6337,
43
+ "eval_mar_10": 0.8316,
44
+ "eval_mar_100": 0.8375,
45
+ "eval_mar_100_baseball-bat": 0.8129,
46
+ "eval_mar_100_basketball": 0.8832,
47
+ "eval_mar_100_car": -1.0,
48
+ "eval_mar_100_football": 0.6099,
49
+ "eval_mar_100_human": 0.9246,
50
+ "eval_mar_100_luggage": 0.792,
51
+ "eval_mar_100_mattress": 0.8388,
52
+ "eval_mar_100_motorcycle": 0.9384,
53
+ "eval_mar_100_skis": 0.9099,
54
+ "eval_mar_100_snowboard": 0.9784,
55
+ "eval_mar_100_soccer-ball": 0.7478,
56
+ "eval_mar_100_stop-sign": 0.9553,
57
+ "eval_mar_100_tennis-racket": 0.7264,
58
+ "eval_mar_100_umbrella": 0.9515,
59
+ "eval_mar_100_volleyball": 0.6554,
60
+ "eval_mar_large": 0.9453,
61
+ "eval_mar_medium": 0.8446,
62
+ "eval_mar_small": 0.7077,
63
+ "eval_runtime": 72.0026,
64
+ "eval_samples_per_second": 20.833,
65
+ "eval_steps_per_second": 0.653,
66
+ "step": 438
67
+ },
68
+ {
69
+ "epoch": 2.0,
70
+ "grad_norm": 51.442848205566406,
71
+ "learning_rate": 9.719978134831347e-06,
72
+ "loss": 8.2849,
73
+ "step": 876
74
+ },
75
+ {
76
+ "epoch": 2.0,
77
+ "eval_loss": 5.165024757385254,
78
+ "eval_map": 0.6623,
79
+ "eval_map_50": 0.7385,
80
+ "eval_map_75": 0.7311,
81
+ "eval_map_baseball-bat": 0.7515,
82
+ "eval_map_basketball": 0.2886,
83
+ "eval_map_car": -1.0,
84
+ "eval_map_football": 0.6431,
85
+ "eval_map_human": 0.9057,
86
+ "eval_map_large": 0.8009,
87
+ "eval_map_luggage": 0.8014,
88
+ "eval_map_mattress": 0.8949,
89
+ "eval_map_medium": 0.674,
90
+ "eval_map_motorcycle": 0.8986,
91
+ "eval_map_skis": 0.1247,
92
+ "eval_map_small": 0.6108,
93
+ "eval_map_snowboard": 0.9665,
94
+ "eval_map_soccer-ball": 0.7817,
95
+ "eval_map_stop-sign": 0.8546,
96
+ "eval_map_tennis-racket": 0.0862,
97
+ "eval_map_umbrella": 0.7593,
98
+ "eval_map_volleyball": 0.5151,
99
+ "eval_mar_1": 0.6914,
100
+ "eval_mar_10": 0.87,
101
+ "eval_mar_100": 0.8726,
102
+ "eval_mar_100_baseball-bat": 0.8723,
103
+ "eval_mar_100_basketball": 0.8671,
104
+ "eval_mar_100_car": -1.0,
105
+ "eval_mar_100_football": 0.7019,
106
+ "eval_mar_100_human": 0.9462,
107
+ "eval_mar_100_luggage": 0.8587,
108
+ "eval_mar_100_mattress": 0.9605,
109
+ "eval_mar_100_motorcycle": 0.9603,
110
+ "eval_mar_100_skis": 0.9936,
111
+ "eval_mar_100_snowboard": 0.9862,
112
+ "eval_mar_100_soccer-ball": 0.8419,
113
+ "eval_mar_100_stop-sign": 0.9869,
114
+ "eval_mar_100_tennis-racket": 0.7197,
115
+ "eval_mar_100_umbrella": 0.9635,
116
+ "eval_mar_100_volleyball": 0.5581,
117
+ "eval_mar_large": 0.9696,
118
+ "eval_mar_medium": 0.9081,
119
+ "eval_mar_small": 0.7227,
120
+ "eval_runtime": 71.9707,
121
+ "eval_samples_per_second": 20.842,
122
+ "eval_steps_per_second": 0.653,
123
+ "step": 876
124
+ },
125
+ {
126
+ "epoch": 3.0,
127
+ "grad_norm": 21.441896438598633,
128
+ "learning_rate": 8.900848647881377e-06,
129
+ "loss": 6.559,
130
+ "step": 1314
131
+ },
132
+ {
133
+ "epoch": 3.0,
134
+ "eval_loss": 4.400482654571533,
135
+ "eval_map": 0.7504,
136
+ "eval_map_50": 0.8276,
137
+ "eval_map_75": 0.8215,
138
+ "eval_map_baseball-bat": 0.7706,
139
+ "eval_map_basketball": 0.6714,
140
+ "eval_map_car": -1.0,
141
+ "eval_map_football": 0.6961,
142
+ "eval_map_human": 0.9191,
143
+ "eval_map_large": 0.8594,
144
+ "eval_map_luggage": 0.8097,
145
+ "eval_map_mattress": 0.9315,
146
+ "eval_map_medium": 0.7272,
147
+ "eval_map_motorcycle": 0.9122,
148
+ "eval_map_skis": 0.356,
149
+ "eval_map_small": 0.7114,
150
+ "eval_map_snowboard": 0.9799,
151
+ "eval_map_soccer-ball": 0.8045,
152
+ "eval_map_stop-sign": 0.9288,
153
+ "eval_map_tennis-racket": 0.2706,
154
+ "eval_map_umbrella": 0.7865,
155
+ "eval_map_volleyball": 0.6683,
156
+ "eval_mar_1": 0.7435,
157
+ "eval_mar_10": 0.8949,
158
+ "eval_mar_100": 0.8966,
159
+ "eval_mar_100_baseball-bat": 0.8428,
160
+ "eval_mar_100_basketball": 0.8801,
161
+ "eval_mar_100_car": -1.0,
162
+ "eval_mar_100_football": 0.7498,
163
+ "eval_mar_100_human": 0.9553,
164
+ "eval_mar_100_luggage": 0.8663,
165
+ "eval_mar_100_mattress": 0.981,
166
+ "eval_mar_100_motorcycle": 0.9665,
167
+ "eval_mar_100_skis": 0.9926,
168
+ "eval_mar_100_snowboard": 0.9927,
169
+ "eval_mar_100_soccer-ball": 0.8594,
170
+ "eval_mar_100_stop-sign": 0.9846,
171
+ "eval_mar_100_tennis-racket": 0.7788,
172
+ "eval_mar_100_umbrella": 0.9736,
173
+ "eval_mar_100_volleyball": 0.7284,
174
+ "eval_mar_large": 0.975,
175
+ "eval_mar_medium": 0.9093,
176
+ "eval_mar_small": 0.7711,
177
+ "eval_runtime": 71.8257,
178
+ "eval_samples_per_second": 20.884,
179
+ "eval_steps_per_second": 0.654,
180
+ "step": 1314
181
+ },
182
+ {
183
+ "epoch": 4.0,
184
+ "grad_norm": 23.689577102661133,
185
+ "learning_rate": 7.641372933209267e-06,
186
+ "loss": 6.1007,
187
+ "step": 1752
188
+ },
189
+ {
190
+ "epoch": 4.0,
191
+ "eval_loss": 4.3049516677856445,
192
+ "eval_map": 0.7733,
193
+ "eval_map_50": 0.8501,
194
+ "eval_map_75": 0.8445,
195
+ "eval_map_baseball-bat": 0.7927,
196
+ "eval_map_basketball": 0.5734,
197
+ "eval_map_car": -1.0,
198
+ "eval_map_football": 0.7111,
199
+ "eval_map_human": 0.9262,
200
+ "eval_map_large": 0.8698,
201
+ "eval_map_luggage": 0.8357,
202
+ "eval_map_mattress": 0.9221,
203
+ "eval_map_medium": 0.8036,
204
+ "eval_map_motorcycle": 0.9326,
205
+ "eval_map_skis": 0.4091,
206
+ "eval_map_small": 0.672,
207
+ "eval_map_snowboard": 0.9788,
208
+ "eval_map_soccer-ball": 0.7975,
209
+ "eval_map_stop-sign": 0.9535,
210
+ "eval_map_tennis-racket": 0.5879,
211
+ "eval_map_umbrella": 0.798,
212
+ "eval_map_volleyball": 0.6073,
213
+ "eval_mar_1": 0.75,
214
+ "eval_mar_10": 0.8933,
215
+ "eval_mar_100": 0.8959,
216
+ "eval_mar_100_baseball-bat": 0.8788,
217
+ "eval_mar_100_basketball": 0.7517,
218
+ "eval_mar_100_car": -1.0,
219
+ "eval_mar_100_football": 0.7712,
220
+ "eval_mar_100_human": 0.959,
221
+ "eval_mar_100_luggage": 0.8953,
222
+ "eval_mar_100_mattress": 0.966,
223
+ "eval_mar_100_motorcycle": 0.9731,
224
+ "eval_mar_100_skis": 0.9936,
225
+ "eval_mar_100_snowboard": 0.9937,
226
+ "eval_mar_100_soccer-ball": 0.8563,
227
+ "eval_mar_100_stop-sign": 0.9877,
228
+ "eval_mar_100_tennis-racket": 0.8409,
229
+ "eval_mar_100_umbrella": 0.9753,
230
+ "eval_mar_100_volleyball": 0.6998,
231
+ "eval_mar_large": 0.977,
232
+ "eval_mar_medium": 0.93,
233
+ "eval_mar_small": 0.7471,
234
+ "eval_runtime": 71.9818,
235
+ "eval_samples_per_second": 20.839,
236
+ "eval_steps_per_second": 0.653,
237
+ "step": 1752
238
+ },
239
+ {
240
+ "epoch": 5.0,
241
+ "grad_norm": 27.84611701965332,
242
+ "learning_rate": 6.097187453407394e-06,
243
+ "loss": 5.7812,
244
+ "step": 2190
245
+ },
246
+ {
247
+ "epoch": 5.0,
248
+ "eval_loss": 3.9205009937286377,
249
+ "eval_map": 0.8281,
250
+ "eval_map_50": 0.91,
251
+ "eval_map_75": 0.9046,
252
+ "eval_map_baseball-bat": 0.8105,
253
+ "eval_map_basketball": 0.7005,
254
+ "eval_map_car": -1.0,
255
+ "eval_map_football": 0.7656,
256
+ "eval_map_human": 0.9326,
257
+ "eval_map_large": 0.9087,
258
+ "eval_map_luggage": 0.8291,
259
+ "eval_map_mattress": 0.9399,
260
+ "eval_map_medium": 0.8373,
261
+ "eval_map_motorcycle": 0.9223,
262
+ "eval_map_skis": 0.6417,
263
+ "eval_map_small": 0.741,
264
+ "eval_map_snowboard": 0.977,
265
+ "eval_map_soccer-ball": 0.8194,
266
+ "eval_map_stop-sign": 0.9422,
267
+ "eval_map_tennis-racket": 0.7258,
268
+ "eval_map_umbrella": 0.8433,
269
+ "eval_map_volleyball": 0.744,
270
+ "eval_mar_1": 0.7819,
271
+ "eval_mar_10": 0.9162,
272
+ "eval_mar_100": 0.9196,
273
+ "eval_mar_100_baseball-bat": 0.8866,
274
+ "eval_mar_100_basketball": 0.8344,
275
+ "eval_mar_100_car": -1.0,
276
+ "eval_mar_100_football": 0.8203,
277
+ "eval_mar_100_human": 0.9609,
278
+ "eval_mar_100_luggage": 0.9,
279
+ "eval_mar_100_mattress": 0.9791,
280
+ "eval_mar_100_motorcycle": 0.9695,
281
+ "eval_mar_100_skis": 0.9906,
282
+ "eval_mar_100_snowboard": 0.9935,
283
+ "eval_mar_100_soccer-ball": 0.8738,
284
+ "eval_mar_100_stop-sign": 0.9858,
285
+ "eval_mar_100_tennis-racket": 0.8819,
286
+ "eval_mar_100_umbrella": 0.9767,
287
+ "eval_mar_100_volleyball": 0.8208,
288
+ "eval_mar_large": 0.9776,
289
+ "eval_mar_medium": 0.9294,
290
+ "eval_mar_small": 0.8073,
291
+ "eval_runtime": 72.8493,
292
+ "eval_samples_per_second": 20.59,
293
+ "eval_steps_per_second": 0.645,
294
+ "step": 2190
295
+ },
296
+ {
297
+ "epoch": 6.0,
298
+ "grad_norm": 22.790102005004883,
299
+ "learning_rate": 4.451280000979843e-06,
300
+ "loss": 5.4833,
301
+ "step": 2628
302
+ },
303
+ {
304
+ "epoch": 6.0,
305
+ "eval_loss": 3.7494161128997803,
306
+ "eval_map": 0.8402,
307
+ "eval_map_50": 0.9191,
308
+ "eval_map_75": 0.915,
309
+ "eval_map_baseball-bat": 0.8153,
310
+ "eval_map_basketball": 0.8096,
311
+ "eval_map_car": -1.0,
312
+ "eval_map_football": 0.7586,
313
+ "eval_map_human": 0.9378,
314
+ "eval_map_large": 0.9247,
315
+ "eval_map_luggage": 0.8322,
316
+ "eval_map_mattress": 0.9545,
317
+ "eval_map_medium": 0.861,
318
+ "eval_map_motorcycle": 0.9241,
319
+ "eval_map_skis": 0.7303,
320
+ "eval_map_small": 0.7567,
321
+ "eval_map_snowboard": 0.9832,
322
+ "eval_map_soccer-ball": 0.8225,
323
+ "eval_map_stop-sign": 0.9542,
324
+ "eval_map_tennis-racket": 0.6731,
325
+ "eval_map_umbrella": 0.8273,
326
+ "eval_map_volleyball": 0.7398,
327
+ "eval_mar_1": 0.7947,
328
+ "eval_mar_10": 0.9219,
329
+ "eval_mar_100": 0.9246,
330
+ "eval_mar_100_baseball-bat": 0.8907,
331
+ "eval_mar_100_basketball": 0.9043,
332
+ "eval_mar_100_car": -1.0,
333
+ "eval_mar_100_football": 0.8083,
334
+ "eval_mar_100_human": 0.9627,
335
+ "eval_mar_100_luggage": 0.9015,
336
+ "eval_mar_100_mattress": 0.989,
337
+ "eval_mar_100_motorcycle": 0.9725,
338
+ "eval_mar_100_skis": 0.9926,
339
+ "eval_mar_100_snowboard": 0.9947,
340
+ "eval_mar_100_soccer-ball": 0.8707,
341
+ "eval_mar_100_stop-sign": 0.988,
342
+ "eval_mar_100_tennis-racket": 0.871,
343
+ "eval_mar_100_umbrella": 0.9776,
344
+ "eval_mar_100_volleyball": 0.8213,
345
+ "eval_mar_large": 0.9902,
346
+ "eval_mar_medium": 0.9378,
347
+ "eval_mar_small": 0.8235,
348
+ "eval_runtime": 72.3187,
349
+ "eval_samples_per_second": 20.742,
350
+ "eval_steps_per_second": 0.65,
351
+ "step": 2628
352
+ },
353
+ {
354
+ "epoch": 7.0,
355
+ "grad_norm": 26.117334365844727,
356
+ "learning_rate": 2.897985402843834e-06,
357
+ "loss": 5.3134,
358
+ "step": 3066
359
+ },
360
+ {
361
+ "epoch": 7.0,
362
+ "eval_loss": 3.7250254154205322,
363
+ "eval_map": 0.8437,
364
+ "eval_map_50": 0.9213,
365
+ "eval_map_75": 0.9165,
366
+ "eval_map_baseball-bat": 0.8142,
367
+ "eval_map_basketball": 0.8066,
368
+ "eval_map_car": -1.0,
369
+ "eval_map_football": 0.7341,
370
+ "eval_map_human": 0.9359,
371
+ "eval_map_large": 0.919,
372
+ "eval_map_luggage": 0.8417,
373
+ "eval_map_mattress": 0.9412,
374
+ "eval_map_medium": 0.854,
375
+ "eval_map_motorcycle": 0.9292,
376
+ "eval_map_skis": 0.713,
377
+ "eval_map_small": 0.7597,
378
+ "eval_map_snowboard": 0.9828,
379
+ "eval_map_soccer-ball": 0.828,
380
+ "eval_map_stop-sign": 0.963,
381
+ "eval_map_tennis-racket": 0.7265,
382
+ "eval_map_umbrella": 0.8307,
383
+ "eval_map_volleyball": 0.7646,
384
+ "eval_mar_1": 0.7917,
385
+ "eval_mar_10": 0.9213,
386
+ "eval_mar_100": 0.9241,
387
+ "eval_mar_100_baseball-bat": 0.8876,
388
+ "eval_mar_100_basketball": 0.9014,
389
+ "eval_mar_100_car": -1.0,
390
+ "eval_mar_100_football": 0.7887,
391
+ "eval_mar_100_human": 0.9624,
392
+ "eval_mar_100_luggage": 0.9133,
393
+ "eval_mar_100_mattress": 0.9782,
394
+ "eval_mar_100_motorcycle": 0.9777,
395
+ "eval_mar_100_skis": 0.9931,
396
+ "eval_mar_100_snowboard": 0.9934,
397
+ "eval_mar_100_soccer-ball": 0.8761,
398
+ "eval_mar_100_stop-sign": 0.9922,
399
+ "eval_mar_100_tennis-racket": 0.8554,
400
+ "eval_mar_100_umbrella": 0.9791,
401
+ "eval_mar_100_volleyball": 0.8392,
402
+ "eval_mar_large": 0.9775,
403
+ "eval_mar_medium": 0.9419,
404
+ "eval_mar_small": 0.8256,
405
+ "eval_runtime": 73.5967,
406
+ "eval_samples_per_second": 20.381,
407
+ "eval_steps_per_second": 0.639,
408
+ "step": 3066
409
+ },
410
+ {
411
+ "epoch": 8.0,
412
+ "grad_norm": 59.24980163574219,
413
+ "learning_rate": 1.6283784771634588e-06,
414
+ "loss": 5.2982,
415
+ "step": 3504
416
+ },
417
+ {
418
+ "epoch": 8.0,
419
+ "eval_loss": 3.782477855682373,
420
+ "eval_map": 0.8419,
421
+ "eval_map_50": 0.9181,
422
+ "eval_map_75": 0.9133,
423
+ "eval_map_baseball-bat": 0.8151,
424
+ "eval_map_basketball": 0.8242,
425
+ "eval_map_car": -1.0,
426
+ "eval_map_football": 0.7225,
427
+ "eval_map_human": 0.9375,
428
+ "eval_map_large": 0.9154,
429
+ "eval_map_luggage": 0.8383,
430
+ "eval_map_mattress": 0.9338,
431
+ "eval_map_medium": 0.8549,
432
+ "eval_map_motorcycle": 0.9288,
433
+ "eval_map_skis": 0.7353,
434
+ "eval_map_small": 0.7599,
435
+ "eval_map_snowboard": 0.9818,
436
+ "eval_map_soccer-ball": 0.8221,
437
+ "eval_map_stop-sign": 0.9675,
438
+ "eval_map_tennis-racket": 0.6801,
439
+ "eval_map_umbrella": 0.8273,
440
+ "eval_map_volleyball": 0.7726,
441
+ "eval_mar_1": 0.7909,
442
+ "eval_mar_10": 0.9198,
443
+ "eval_mar_100": 0.922,
444
+ "eval_mar_100_baseball-bat": 0.8931,
445
+ "eval_mar_100_basketball": 0.9066,
446
+ "eval_mar_100_car": -1.0,
447
+ "eval_mar_100_football": 0.772,
448
+ "eval_mar_100_human": 0.9627,
449
+ "eval_mar_100_luggage": 0.9052,
450
+ "eval_mar_100_mattress": 0.9744,
451
+ "eval_mar_100_motorcycle": 0.9755,
452
+ "eval_mar_100_skis": 0.996,
453
+ "eval_mar_100_snowboard": 0.9925,
454
+ "eval_mar_100_soccer-ball": 0.873,
455
+ "eval_mar_100_stop-sign": 0.9891,
456
+ "eval_mar_100_tennis-racket": 0.8466,
457
+ "eval_mar_100_umbrella": 0.9818,
458
+ "eval_mar_100_volleyball": 0.8397,
459
+ "eval_mar_large": 0.9768,
460
+ "eval_mar_medium": 0.9355,
461
+ "eval_mar_small": 0.8277,
462
+ "eval_runtime": 72.6914,
463
+ "eval_samples_per_second": 20.635,
464
+ "eval_steps_per_second": 0.647,
465
+ "step": 3504
466
+ },
467
+ {
468
+ "epoch": 9.0,
469
+ "grad_norm": 50.26298141479492,
470
+ "learning_rate": 7.955925565783265e-07,
471
+ "loss": 5.1855,
472
+ "step": 3942
473
+ },
474
+ {
475
+ "epoch": 9.0,
476
+ "eval_loss": 3.794715166091919,
477
+ "eval_map": 0.8365,
478
+ "eval_map_50": 0.913,
479
+ "eval_map_75": 0.9085,
480
+ "eval_map_baseball-bat": 0.8041,
481
+ "eval_map_basketball": 0.8009,
482
+ "eval_map_car": -1.0,
483
+ "eval_map_football": 0.7239,
484
+ "eval_map_human": 0.9351,
485
+ "eval_map_large": 0.9104,
486
+ "eval_map_luggage": 0.8444,
487
+ "eval_map_mattress": 0.9314,
488
+ "eval_map_medium": 0.8439,
489
+ "eval_map_motorcycle": 0.9241,
490
+ "eval_map_skis": 0.7034,
491
+ "eval_map_small": 0.7528,
492
+ "eval_map_snowboard": 0.984,
493
+ "eval_map_soccer-ball": 0.8168,
494
+ "eval_map_stop-sign": 0.9639,
495
+ "eval_map_tennis-racket": 0.6888,
496
+ "eval_map_umbrella": 0.8203,
497
+ "eval_map_volleyball": 0.7704,
498
+ "eval_mar_1": 0.7884,
499
+ "eval_mar_10": 0.9184,
500
+ "eval_mar_100": 0.9219,
501
+ "eval_mar_100_baseball-bat": 0.8864,
502
+ "eval_mar_100_basketball": 0.8955,
503
+ "eval_mar_100_car": -1.0,
504
+ "eval_mar_100_football": 0.7755,
505
+ "eval_mar_100_human": 0.9622,
506
+ "eval_mar_100_luggage": 0.9119,
507
+ "eval_mar_100_mattress": 0.9771,
508
+ "eval_mar_100_motorcycle": 0.9735,
509
+ "eval_mar_100_skis": 0.995,
510
+ "eval_mar_100_snowboard": 0.995,
511
+ "eval_mar_100_soccer-ball": 0.8704,
512
+ "eval_mar_100_stop-sign": 0.9922,
513
+ "eval_mar_100_tennis-racket": 0.8513,
514
+ "eval_mar_100_umbrella": 0.9808,
515
+ "eval_mar_100_volleyball": 0.8402,
516
+ "eval_mar_large": 0.979,
517
+ "eval_mar_medium": 0.9344,
518
+ "eval_mar_small": 0.8189,
519
+ "eval_runtime": 73.3675,
520
+ "eval_samples_per_second": 20.445,
521
+ "eval_steps_per_second": 0.641,
522
+ "step": 3942
523
+ },
524
+ {
525
+ "epoch": 10.0,
526
+ "grad_norm": 48.91480255126953,
527
+ "learning_rate": 5.000739137221696e-07,
528
+ "loss": 5.1499,
529
+ "step": 4380
530
+ },
531
+ {
532
+ "epoch": 10.0,
533
+ "eval_loss": 3.652301549911499,
534
+ "eval_map": 0.8465,
535
+ "eval_map_50": 0.9229,
536
+ "eval_map_75": 0.9193,
537
+ "eval_map_baseball-bat": 0.8128,
538
+ "eval_map_basketball": 0.8105,
539
+ "eval_map_car": -1.0,
540
+ "eval_map_football": 0.7611,
541
+ "eval_map_human": 0.9382,
542
+ "eval_map_large": 0.9144,
543
+ "eval_map_luggage": 0.8579,
544
+ "eval_map_mattress": 0.9384,
545
+ "eval_map_medium": 0.8561,
546
+ "eval_map_motorcycle": 0.9309,
547
+ "eval_map_skis": 0.7044,
548
+ "eval_map_small": 0.7682,
549
+ "eval_map_snowboard": 0.9834,
550
+ "eval_map_soccer-ball": 0.8245,
551
+ "eval_map_stop-sign": 0.9671,
552
+ "eval_map_tennis-racket": 0.7064,
553
+ "eval_map_umbrella": 0.8285,
554
+ "eval_map_volleyball": 0.787,
555
+ "eval_mar_1": 0.7945,
556
+ "eval_mar_10": 0.9245,
557
+ "eval_mar_100": 0.9271,
558
+ "eval_mar_100_baseball-bat": 0.892,
559
+ "eval_mar_100_basketball": 0.8993,
560
+ "eval_mar_100_car": -1.0,
561
+ "eval_mar_100_football": 0.8113,
562
+ "eval_mar_100_human": 0.9641,
563
+ "eval_mar_100_luggage": 0.9191,
564
+ "eval_mar_100_mattress": 0.977,
565
+ "eval_mar_100_motorcycle": 0.9773,
566
+ "eval_mar_100_skis": 0.995,
567
+ "eval_mar_100_snowboard": 0.9932,
568
+ "eval_mar_100_soccer-ball": 0.8733,
569
+ "eval_mar_100_stop-sign": 0.9925,
570
+ "eval_mar_100_tennis-racket": 0.8539,
571
+ "eval_mar_100_umbrella": 0.9803,
572
+ "eval_mar_100_volleyball": 0.8517,
573
+ "eval_mar_large": 0.9779,
574
+ "eval_mar_medium": 0.9357,
575
+ "eval_mar_small": 0.8316,
576
+ "eval_runtime": 73.2259,
577
+ "eval_samples_per_second": 20.485,
578
+ "eval_steps_per_second": 0.642,
579
+ "step": 4380
580
+ },
581
+ {
582
+ "epoch": 10.0,
583
+ "step": 4380,
584
+ "total_flos": 2.2065014071296e+19,
585
+ "train_loss": 7.963780066085189,
586
+ "train_runtime": 3254.086,
587
+ "train_samples_per_second": 21.511,
588
+ "train_steps_per_second": 1.346
589
+ }
590
+ ],
591
+ "logging_steps": 500,
592
+ "max_steps": 4380,
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": 16,
610
+ "trial_name": null,
611
+ "trial_params": null
612
+ }