Ariana03 commited on
Commit
764d3fb
1 Parent(s): 92081c2

🍻 cheers

Browse files
README.md CHANGED
@@ -3,6 +3,7 @@ library_name: transformers
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224-in21k
5
  tags:
 
6
  - generated_from_trainer
7
  metrics:
8
  - accuracy
@@ -16,10 +17,10 @@ should probably proofread and complete it, then remove this comment. -->
16
 
17
  # finetuned-indian-food
18
 
19
- This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on an unknown dataset.
20
  It achieves the following results on the evaluation set:
21
- - Loss: 0.2259
22
- - Accuracy: 0.9490
23
 
24
  ## Model description
25
 
 
3
  license: apache-2.0
4
  base_model: google/vit-base-patch16-224-in21k
5
  tags:
6
+ - image-classification
7
  - generated_from_trainer
8
  metrics:
9
  - accuracy
 
17
 
18
  # finetuned-indian-food
19
 
20
+ This model is a fine-tuned version of [google/vit-base-patch16-224-in21k](https://huggingface.co/google/vit-base-patch16-224-in21k) on the indian_food_images dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.2146
23
+ - Accuracy: 0.9426
24
 
25
  ## Model description
26
 
all_results.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "eval_accuracy": 0.9426142401700319,
4
+ "eval_loss": 0.21462436020374298,
5
+ "eval_runtime": 26.9124,
6
+ "eval_samples_per_second": 34.965,
7
+ "eval_steps_per_second": 4.385,
8
+ "total_flos": 1.651775717862015e+18,
9
+ "train_loss": 0.4770685946261203,
10
+ "train_runtime": 1116.421,
11
+ "train_samples_per_second": 19.09,
12
+ "train_steps_per_second": 1.193
13
+ }
eval_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "eval_accuracy": 0.9426142401700319,
4
+ "eval_loss": 0.21462436020374298,
5
+ "eval_runtime": 26.9124,
6
+ "eval_samples_per_second": 34.965,
7
+ "eval_steps_per_second": 4.385
8
+ }
runs/Oct17_14-21-43_609077bf8eb4/events.out.tfevents.1729176099.609077bf8eb4.269.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f0db8c8d1f8ab4222621c68e643e39fd8b2b05dbac1b7eef8c6773b29922968a
3
+ size 411
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 4.0,
3
+ "total_flos": 1.651775717862015e+18,
4
+ "train_loss": 0.4770685946261203,
5
+ "train_runtime": 1116.421,
6
+ "train_samples_per_second": 19.09,
7
+ "train_steps_per_second": 1.193
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,1090 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.21462436020374298,
3
+ "best_model_checkpoint": "finetuned-indian-food/checkpoint-1200",
4
+ "epoch": 4.0,
5
+ "eval_steps": 100,
6
+ "global_step": 1332,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.03003003003003003,
13
+ "grad_norm": 2.086472749710083,
14
+ "learning_rate": 0.0001984984984984985,
15
+ "loss": 2.8715,
16
+ "step": 10
17
+ },
18
+ {
19
+ "epoch": 0.06006006006006006,
20
+ "grad_norm": 2.6774790287017822,
21
+ "learning_rate": 0.00019699699699699701,
22
+ "loss": 2.5719,
23
+ "step": 20
24
+ },
25
+ {
26
+ "epoch": 0.09009009009009009,
27
+ "grad_norm": 2.241564989089966,
28
+ "learning_rate": 0.0001954954954954955,
29
+ "loss": 2.2151,
30
+ "step": 30
31
+ },
32
+ {
33
+ "epoch": 0.12012012012012012,
34
+ "grad_norm": 2.272333860397339,
35
+ "learning_rate": 0.00019399399399399402,
36
+ "loss": 2.0446,
37
+ "step": 40
38
+ },
39
+ {
40
+ "epoch": 0.15015015015015015,
41
+ "grad_norm": 2.3084256649017334,
42
+ "learning_rate": 0.0001924924924924925,
43
+ "loss": 1.6935,
44
+ "step": 50
45
+ },
46
+ {
47
+ "epoch": 0.18018018018018017,
48
+ "grad_norm": 2.6124863624572754,
49
+ "learning_rate": 0.000190990990990991,
50
+ "loss": 1.4559,
51
+ "step": 60
52
+ },
53
+ {
54
+ "epoch": 0.21021021021021022,
55
+ "grad_norm": 1.7842832803726196,
56
+ "learning_rate": 0.0001894894894894895,
57
+ "loss": 1.2871,
58
+ "step": 70
59
+ },
60
+ {
61
+ "epoch": 0.24024024024024024,
62
+ "grad_norm": 1.8386505842208862,
63
+ "learning_rate": 0.000187987987987988,
64
+ "loss": 1.244,
65
+ "step": 80
66
+ },
67
+ {
68
+ "epoch": 0.2702702702702703,
69
+ "grad_norm": 3.2887415885925293,
70
+ "learning_rate": 0.0001864864864864865,
71
+ "loss": 1.2332,
72
+ "step": 90
73
+ },
74
+ {
75
+ "epoch": 0.3003003003003003,
76
+ "grad_norm": 1.8092305660247803,
77
+ "learning_rate": 0.000184984984984985,
78
+ "loss": 1.0574,
79
+ "step": 100
80
+ },
81
+ {
82
+ "epoch": 0.3003003003003003,
83
+ "eval_accuracy": 0.8575982996811902,
84
+ "eval_loss": 0.9444788694381714,
85
+ "eval_runtime": 26.2853,
86
+ "eval_samples_per_second": 35.8,
87
+ "eval_steps_per_second": 4.489,
88
+ "step": 100
89
+ },
90
+ {
91
+ "epoch": 0.3303303303303303,
92
+ "grad_norm": 2.07977557182312,
93
+ "learning_rate": 0.0001834834834834835,
94
+ "loss": 1.0456,
95
+ "step": 110
96
+ },
97
+ {
98
+ "epoch": 0.36036036036036034,
99
+ "grad_norm": 1.9540412425994873,
100
+ "learning_rate": 0.000181981981981982,
101
+ "loss": 1.0319,
102
+ "step": 120
103
+ },
104
+ {
105
+ "epoch": 0.39039039039039036,
106
+ "grad_norm": 1.9397859573364258,
107
+ "learning_rate": 0.0001804804804804805,
108
+ "loss": 0.7043,
109
+ "step": 130
110
+ },
111
+ {
112
+ "epoch": 0.42042042042042044,
113
+ "grad_norm": 3.5629427433013916,
114
+ "learning_rate": 0.00017897897897897898,
115
+ "loss": 0.8621,
116
+ "step": 140
117
+ },
118
+ {
119
+ "epoch": 0.45045045045045046,
120
+ "grad_norm": 2.1689531803131104,
121
+ "learning_rate": 0.0001774774774774775,
122
+ "loss": 0.7846,
123
+ "step": 150
124
+ },
125
+ {
126
+ "epoch": 0.4804804804804805,
127
+ "grad_norm": 5.137659549713135,
128
+ "learning_rate": 0.000175975975975976,
129
+ "loss": 0.8796,
130
+ "step": 160
131
+ },
132
+ {
133
+ "epoch": 0.5105105105105106,
134
+ "grad_norm": 3.2067863941192627,
135
+ "learning_rate": 0.0001744744744744745,
136
+ "loss": 0.7393,
137
+ "step": 170
138
+ },
139
+ {
140
+ "epoch": 0.5405405405405406,
141
+ "grad_norm": 3.543780565261841,
142
+ "learning_rate": 0.000172972972972973,
143
+ "loss": 0.7776,
144
+ "step": 180
145
+ },
146
+ {
147
+ "epoch": 0.5705705705705706,
148
+ "grad_norm": 5.8286333084106445,
149
+ "learning_rate": 0.00017147147147147148,
150
+ "loss": 0.7427,
151
+ "step": 190
152
+ },
153
+ {
154
+ "epoch": 0.6006006006006006,
155
+ "grad_norm": 2.206890106201172,
156
+ "learning_rate": 0.00016996996996997,
157
+ "loss": 0.8399,
158
+ "step": 200
159
+ },
160
+ {
161
+ "epoch": 0.6006006006006006,
162
+ "eval_accuracy": 0.8862911795961743,
163
+ "eval_loss": 0.5542067885398865,
164
+ "eval_runtime": 26.2959,
165
+ "eval_samples_per_second": 35.785,
166
+ "eval_steps_per_second": 4.487,
167
+ "step": 200
168
+ },
169
+ {
170
+ "epoch": 0.6306306306306306,
171
+ "grad_norm": 5.071735382080078,
172
+ "learning_rate": 0.00016846846846846846,
173
+ "loss": 0.4854,
174
+ "step": 210
175
+ },
176
+ {
177
+ "epoch": 0.6606606606606606,
178
+ "grad_norm": 1.8520969152450562,
179
+ "learning_rate": 0.00016696696696696697,
180
+ "loss": 0.5468,
181
+ "step": 220
182
+ },
183
+ {
184
+ "epoch": 0.6906906906906907,
185
+ "grad_norm": 2.1558125019073486,
186
+ "learning_rate": 0.00016546546546546546,
187
+ "loss": 0.5233,
188
+ "step": 230
189
+ },
190
+ {
191
+ "epoch": 0.7207207207207207,
192
+ "grad_norm": 4.998787879943848,
193
+ "learning_rate": 0.00016396396396396395,
194
+ "loss": 0.6996,
195
+ "step": 240
196
+ },
197
+ {
198
+ "epoch": 0.7507507507507507,
199
+ "grad_norm": 2.3446874618530273,
200
+ "learning_rate": 0.00016246246246246247,
201
+ "loss": 0.6156,
202
+ "step": 250
203
+ },
204
+ {
205
+ "epoch": 0.7807807807807807,
206
+ "grad_norm": 9.410783767700195,
207
+ "learning_rate": 0.00016096096096096096,
208
+ "loss": 0.7073,
209
+ "step": 260
210
+ },
211
+ {
212
+ "epoch": 0.8108108108108109,
213
+ "grad_norm": 1.848683476448059,
214
+ "learning_rate": 0.00015945945945945947,
215
+ "loss": 0.5893,
216
+ "step": 270
217
+ },
218
+ {
219
+ "epoch": 0.8408408408408409,
220
+ "grad_norm": 4.705769062042236,
221
+ "learning_rate": 0.00015795795795795796,
222
+ "loss": 0.617,
223
+ "step": 280
224
+ },
225
+ {
226
+ "epoch": 0.8708708708708709,
227
+ "grad_norm": 3.885349988937378,
228
+ "learning_rate": 0.00015645645645645645,
229
+ "loss": 0.5976,
230
+ "step": 290
231
+ },
232
+ {
233
+ "epoch": 0.9009009009009009,
234
+ "grad_norm": 2.5080394744873047,
235
+ "learning_rate": 0.00015495495495495496,
236
+ "loss": 0.6418,
237
+ "step": 300
238
+ },
239
+ {
240
+ "epoch": 0.9009009009009009,
241
+ "eval_accuracy": 0.8671625929861849,
242
+ "eval_loss": 0.5741079449653625,
243
+ "eval_runtime": 26.4803,
244
+ "eval_samples_per_second": 35.536,
245
+ "eval_steps_per_second": 4.456,
246
+ "step": 300
247
+ },
248
+ {
249
+ "epoch": 0.9309309309309309,
250
+ "grad_norm": 5.160592555999756,
251
+ "learning_rate": 0.00015345345345345345,
252
+ "loss": 0.6029,
253
+ "step": 310
254
+ },
255
+ {
256
+ "epoch": 0.960960960960961,
257
+ "grad_norm": 3.9884884357452393,
258
+ "learning_rate": 0.00015195195195195194,
259
+ "loss": 0.6916,
260
+ "step": 320
261
+ },
262
+ {
263
+ "epoch": 0.990990990990991,
264
+ "grad_norm": 4.166062355041504,
265
+ "learning_rate": 0.00015045045045045046,
266
+ "loss": 0.7854,
267
+ "step": 330
268
+ },
269
+ {
270
+ "epoch": 1.021021021021021,
271
+ "grad_norm": 4.180619716644287,
272
+ "learning_rate": 0.00014894894894894895,
273
+ "loss": 0.5521,
274
+ "step": 340
275
+ },
276
+ {
277
+ "epoch": 1.0510510510510511,
278
+ "grad_norm": 4.4930315017700195,
279
+ "learning_rate": 0.00014744744744744746,
280
+ "loss": 0.4041,
281
+ "step": 350
282
+ },
283
+ {
284
+ "epoch": 1.0810810810810811,
285
+ "grad_norm": 4.935527324676514,
286
+ "learning_rate": 0.00014594594594594595,
287
+ "loss": 0.4967,
288
+ "step": 360
289
+ },
290
+ {
291
+ "epoch": 1.1111111111111112,
292
+ "grad_norm": 12.9185791015625,
293
+ "learning_rate": 0.00014444444444444444,
294
+ "loss": 0.4248,
295
+ "step": 370
296
+ },
297
+ {
298
+ "epoch": 1.1411411411411412,
299
+ "grad_norm": 3.98164439201355,
300
+ "learning_rate": 0.00014294294294294295,
301
+ "loss": 0.5054,
302
+ "step": 380
303
+ },
304
+ {
305
+ "epoch": 1.1711711711711712,
306
+ "grad_norm": 4.026088237762451,
307
+ "learning_rate": 0.00014144144144144144,
308
+ "loss": 0.4668,
309
+ "step": 390
310
+ },
311
+ {
312
+ "epoch": 1.2012012012012012,
313
+ "grad_norm": 5.09647798538208,
314
+ "learning_rate": 0.00013993993993993996,
315
+ "loss": 0.3785,
316
+ "step": 400
317
+ },
318
+ {
319
+ "epoch": 1.2012012012012012,
320
+ "eval_accuracy": 0.8841657810839533,
321
+ "eval_loss": 0.4702335000038147,
322
+ "eval_runtime": 26.6715,
323
+ "eval_samples_per_second": 35.281,
324
+ "eval_steps_per_second": 4.424,
325
+ "step": 400
326
+ },
327
+ {
328
+ "epoch": 1.2312312312312312,
329
+ "grad_norm": 0.3809890151023865,
330
+ "learning_rate": 0.00013843843843843845,
331
+ "loss": 0.37,
332
+ "step": 410
333
+ },
334
+ {
335
+ "epoch": 1.2612612612612613,
336
+ "grad_norm": 5.375821113586426,
337
+ "learning_rate": 0.00013693693693693693,
338
+ "loss": 0.4845,
339
+ "step": 420
340
+ },
341
+ {
342
+ "epoch": 1.2912912912912913,
343
+ "grad_norm": 1.7599005699157715,
344
+ "learning_rate": 0.00013543543543543545,
345
+ "loss": 0.6018,
346
+ "step": 430
347
+ },
348
+ {
349
+ "epoch": 1.3213213213213213,
350
+ "grad_norm": 1.0588518381118774,
351
+ "learning_rate": 0.00013393393393393394,
352
+ "loss": 0.4875,
353
+ "step": 440
354
+ },
355
+ {
356
+ "epoch": 1.3513513513513513,
357
+ "grad_norm": 2.4473206996917725,
358
+ "learning_rate": 0.00013243243243243243,
359
+ "loss": 0.5371,
360
+ "step": 450
361
+ },
362
+ {
363
+ "epoch": 1.3813813813813813,
364
+ "grad_norm": 9.87109661102295,
365
+ "learning_rate": 0.00013093093093093094,
366
+ "loss": 0.4605,
367
+ "step": 460
368
+ },
369
+ {
370
+ "epoch": 1.4114114114114114,
371
+ "grad_norm": 2.05519700050354,
372
+ "learning_rate": 0.00012942942942942943,
373
+ "loss": 0.4135,
374
+ "step": 470
375
+ },
376
+ {
377
+ "epoch": 1.4414414414414414,
378
+ "grad_norm": 3.4596476554870605,
379
+ "learning_rate": 0.00012792792792792795,
380
+ "loss": 0.543,
381
+ "step": 480
382
+ },
383
+ {
384
+ "epoch": 1.4714714714714714,
385
+ "grad_norm": 2.090067148208618,
386
+ "learning_rate": 0.00012642642642642644,
387
+ "loss": 0.365,
388
+ "step": 490
389
+ },
390
+ {
391
+ "epoch": 1.5015015015015014,
392
+ "grad_norm": 1.422653317451477,
393
+ "learning_rate": 0.00012492492492492492,
394
+ "loss": 0.4451,
395
+ "step": 500
396
+ },
397
+ {
398
+ "epoch": 1.5015015015015014,
399
+ "eval_accuracy": 0.9117959617428267,
400
+ "eval_loss": 0.3684743642807007,
401
+ "eval_runtime": 26.6937,
402
+ "eval_samples_per_second": 35.252,
403
+ "eval_steps_per_second": 4.421,
404
+ "step": 500
405
+ },
406
+ {
407
+ "epoch": 1.5315315315315314,
408
+ "grad_norm": 3.5416722297668457,
409
+ "learning_rate": 0.00012342342342342344,
410
+ "loss": 0.3413,
411
+ "step": 510
412
+ },
413
+ {
414
+ "epoch": 1.5615615615615615,
415
+ "grad_norm": 0.6044331789016724,
416
+ "learning_rate": 0.00012192192192192193,
417
+ "loss": 0.3424,
418
+ "step": 520
419
+ },
420
+ {
421
+ "epoch": 1.5915915915915915,
422
+ "grad_norm": 8.660246849060059,
423
+ "learning_rate": 0.00012042042042042043,
424
+ "loss": 0.4565,
425
+ "step": 530
426
+ },
427
+ {
428
+ "epoch": 1.6216216216216215,
429
+ "grad_norm": 4.744662761688232,
430
+ "learning_rate": 0.00011891891891891893,
431
+ "loss": 0.4159,
432
+ "step": 540
433
+ },
434
+ {
435
+ "epoch": 1.6516516516516515,
436
+ "grad_norm": 2.420365333557129,
437
+ "learning_rate": 0.00011741741741741743,
438
+ "loss": 0.356,
439
+ "step": 550
440
+ },
441
+ {
442
+ "epoch": 1.6816816816816815,
443
+ "grad_norm": 4.598628997802734,
444
+ "learning_rate": 0.00011591591591591592,
445
+ "loss": 0.4904,
446
+ "step": 560
447
+ },
448
+ {
449
+ "epoch": 1.7117117117117115,
450
+ "grad_norm": 3.6854140758514404,
451
+ "learning_rate": 0.00011441441441441443,
452
+ "loss": 0.353,
453
+ "step": 570
454
+ },
455
+ {
456
+ "epoch": 1.7417417417417418,
457
+ "grad_norm": 4.791896343231201,
458
+ "learning_rate": 0.00011291291291291293,
459
+ "loss": 0.4928,
460
+ "step": 580
461
+ },
462
+ {
463
+ "epoch": 1.7717717717717718,
464
+ "grad_norm": 4.51821231842041,
465
+ "learning_rate": 0.00011141141141141143,
466
+ "loss": 0.365,
467
+ "step": 590
468
+ },
469
+ {
470
+ "epoch": 1.8018018018018018,
471
+ "grad_norm": 4.391872882843018,
472
+ "learning_rate": 0.00010990990990990993,
473
+ "loss": 0.4535,
474
+ "step": 600
475
+ },
476
+ {
477
+ "epoch": 1.8018018018018018,
478
+ "eval_accuracy": 0.9096705632306057,
479
+ "eval_loss": 0.37811535596847534,
480
+ "eval_runtime": 26.4312,
481
+ "eval_samples_per_second": 35.602,
482
+ "eval_steps_per_second": 4.464,
483
+ "step": 600
484
+ },
485
+ {
486
+ "epoch": 1.8318318318318318,
487
+ "grad_norm": 2.613373279571533,
488
+ "learning_rate": 0.00010840840840840842,
489
+ "loss": 0.3374,
490
+ "step": 610
491
+ },
492
+ {
493
+ "epoch": 1.8618618618618619,
494
+ "grad_norm": 2.3293354511260986,
495
+ "learning_rate": 0.00010690690690690692,
496
+ "loss": 0.2425,
497
+ "step": 620
498
+ },
499
+ {
500
+ "epoch": 1.8918918918918919,
501
+ "grad_norm": 4.844559669494629,
502
+ "learning_rate": 0.0001054054054054054,
503
+ "loss": 0.4471,
504
+ "step": 630
505
+ },
506
+ {
507
+ "epoch": 1.921921921921922,
508
+ "grad_norm": 4.249619960784912,
509
+ "learning_rate": 0.0001039039039039039,
510
+ "loss": 0.4115,
511
+ "step": 640
512
+ },
513
+ {
514
+ "epoch": 1.951951951951952,
515
+ "grad_norm": 5.946290969848633,
516
+ "learning_rate": 0.0001024024024024024,
517
+ "loss": 0.4435,
518
+ "step": 650
519
+ },
520
+ {
521
+ "epoch": 1.981981981981982,
522
+ "grad_norm": 1.4038317203521729,
523
+ "learning_rate": 0.00010090090090090089,
524
+ "loss": 0.2987,
525
+ "step": 660
526
+ },
527
+ {
528
+ "epoch": 2.012012012012012,
529
+ "grad_norm": 2.397927761077881,
530
+ "learning_rate": 9.93993993993994e-05,
531
+ "loss": 0.224,
532
+ "step": 670
533
+ },
534
+ {
535
+ "epoch": 2.042042042042042,
536
+ "grad_norm": 3.2759602069854736,
537
+ "learning_rate": 9.789789789789791e-05,
538
+ "loss": 0.3491,
539
+ "step": 680
540
+ },
541
+ {
542
+ "epoch": 2.0720720720720722,
543
+ "grad_norm": 0.820797324180603,
544
+ "learning_rate": 9.639639639639641e-05,
545
+ "loss": 0.3682,
546
+ "step": 690
547
+ },
548
+ {
549
+ "epoch": 2.1021021021021022,
550
+ "grad_norm": 1.0604861974716187,
551
+ "learning_rate": 9.48948948948949e-05,
552
+ "loss": 0.4618,
553
+ "step": 700
554
+ },
555
+ {
556
+ "epoch": 2.1021021021021022,
557
+ "eval_accuracy": 0.9287991498405951,
558
+ "eval_loss": 0.3000124990940094,
559
+ "eval_runtime": 26.5936,
560
+ "eval_samples_per_second": 35.384,
561
+ "eval_steps_per_second": 4.437,
562
+ "step": 700
563
+ },
564
+ {
565
+ "epoch": 2.1321321321321323,
566
+ "grad_norm": 1.994698405265808,
567
+ "learning_rate": 9.33933933933934e-05,
568
+ "loss": 0.2813,
569
+ "step": 710
570
+ },
571
+ {
572
+ "epoch": 2.1621621621621623,
573
+ "grad_norm": 1.6023216247558594,
574
+ "learning_rate": 9.18918918918919e-05,
575
+ "loss": 0.3322,
576
+ "step": 720
577
+ },
578
+ {
579
+ "epoch": 2.1921921921921923,
580
+ "grad_norm": 6.890353679656982,
581
+ "learning_rate": 9.039039039039039e-05,
582
+ "loss": 0.2649,
583
+ "step": 730
584
+ },
585
+ {
586
+ "epoch": 2.2222222222222223,
587
+ "grad_norm": 4.888972282409668,
588
+ "learning_rate": 8.888888888888889e-05,
589
+ "loss": 0.2021,
590
+ "step": 740
591
+ },
592
+ {
593
+ "epoch": 2.2522522522522523,
594
+ "grad_norm": 4.375626564025879,
595
+ "learning_rate": 8.738738738738738e-05,
596
+ "loss": 0.3112,
597
+ "step": 750
598
+ },
599
+ {
600
+ "epoch": 2.2822822822822824,
601
+ "grad_norm": 8.04282283782959,
602
+ "learning_rate": 8.588588588588588e-05,
603
+ "loss": 0.2771,
604
+ "step": 760
605
+ },
606
+ {
607
+ "epoch": 2.3123123123123124,
608
+ "grad_norm": 3.198418140411377,
609
+ "learning_rate": 8.438438438438439e-05,
610
+ "loss": 0.291,
611
+ "step": 770
612
+ },
613
+ {
614
+ "epoch": 2.3423423423423424,
615
+ "grad_norm": 4.040826320648193,
616
+ "learning_rate": 8.288288288288289e-05,
617
+ "loss": 0.2622,
618
+ "step": 780
619
+ },
620
+ {
621
+ "epoch": 2.3723723723723724,
622
+ "grad_norm": 4.993142127990723,
623
+ "learning_rate": 8.138138138138138e-05,
624
+ "loss": 0.3394,
625
+ "step": 790
626
+ },
627
+ {
628
+ "epoch": 2.4024024024024024,
629
+ "grad_norm": 3.0009608268737793,
630
+ "learning_rate": 7.987987987987988e-05,
631
+ "loss": 0.2321,
632
+ "step": 800
633
+ },
634
+ {
635
+ "epoch": 2.4024024024024024,
636
+ "eval_accuracy": 0.9181721572794899,
637
+ "eval_loss": 0.31457969546318054,
638
+ "eval_runtime": 26.9039,
639
+ "eval_samples_per_second": 34.976,
640
+ "eval_steps_per_second": 4.386,
641
+ "step": 800
642
+ },
643
+ {
644
+ "epoch": 2.4324324324324325,
645
+ "grad_norm": 1.5650168657302856,
646
+ "learning_rate": 7.837837837837838e-05,
647
+ "loss": 0.1994,
648
+ "step": 810
649
+ },
650
+ {
651
+ "epoch": 2.4624624624624625,
652
+ "grad_norm": 2.945512533187866,
653
+ "learning_rate": 7.687687687687688e-05,
654
+ "loss": 0.2863,
655
+ "step": 820
656
+ },
657
+ {
658
+ "epoch": 2.4924924924924925,
659
+ "grad_norm": 6.051520824432373,
660
+ "learning_rate": 7.537537537537538e-05,
661
+ "loss": 0.3662,
662
+ "step": 830
663
+ },
664
+ {
665
+ "epoch": 2.5225225225225225,
666
+ "grad_norm": 2.5240094661712646,
667
+ "learning_rate": 7.387387387387387e-05,
668
+ "loss": 0.2587,
669
+ "step": 840
670
+ },
671
+ {
672
+ "epoch": 2.5525525525525525,
673
+ "grad_norm": 2.640320062637329,
674
+ "learning_rate": 7.237237237237238e-05,
675
+ "loss": 0.1814,
676
+ "step": 850
677
+ },
678
+ {
679
+ "epoch": 2.5825825825825826,
680
+ "grad_norm": 2.48514723777771,
681
+ "learning_rate": 7.087087087087088e-05,
682
+ "loss": 0.2494,
683
+ "step": 860
684
+ },
685
+ {
686
+ "epoch": 2.6126126126126126,
687
+ "grad_norm": 4.28499698638916,
688
+ "learning_rate": 6.936936936936938e-05,
689
+ "loss": 0.1975,
690
+ "step": 870
691
+ },
692
+ {
693
+ "epoch": 2.6426426426426426,
694
+ "grad_norm": 2.4091761112213135,
695
+ "learning_rate": 6.786786786786787e-05,
696
+ "loss": 0.1894,
697
+ "step": 880
698
+ },
699
+ {
700
+ "epoch": 2.6726726726726726,
701
+ "grad_norm": 1.1737655401229858,
702
+ "learning_rate": 6.636636636636637e-05,
703
+ "loss": 0.3093,
704
+ "step": 890
705
+ },
706
+ {
707
+ "epoch": 2.7027027027027026,
708
+ "grad_norm": 1.0858086347579956,
709
+ "learning_rate": 6.486486486486487e-05,
710
+ "loss": 0.1816,
711
+ "step": 900
712
+ },
713
+ {
714
+ "epoch": 2.7027027027027026,
715
+ "eval_accuracy": 0.9213602550478215,
716
+ "eval_loss": 0.30446675419807434,
717
+ "eval_runtime": 26.7771,
718
+ "eval_samples_per_second": 35.142,
719
+ "eval_steps_per_second": 4.407,
720
+ "step": 900
721
+ },
722
+ {
723
+ "epoch": 2.7327327327327327,
724
+ "grad_norm": 6.119840145111084,
725
+ "learning_rate": 6.336336336336337e-05,
726
+ "loss": 0.3395,
727
+ "step": 910
728
+ },
729
+ {
730
+ "epoch": 2.7627627627627627,
731
+ "grad_norm": 1.7285120487213135,
732
+ "learning_rate": 6.186186186186186e-05,
733
+ "loss": 0.2643,
734
+ "step": 920
735
+ },
736
+ {
737
+ "epoch": 2.7927927927927927,
738
+ "grad_norm": 4.141608715057373,
739
+ "learning_rate": 6.0360360360360365e-05,
740
+ "loss": 0.1987,
741
+ "step": 930
742
+ },
743
+ {
744
+ "epoch": 2.8228228228228227,
745
+ "grad_norm": 3.5640854835510254,
746
+ "learning_rate": 5.8858858858858854e-05,
747
+ "loss": 0.2566,
748
+ "step": 940
749
+ },
750
+ {
751
+ "epoch": 2.8528528528528527,
752
+ "grad_norm": 0.9691357612609863,
753
+ "learning_rate": 5.7357357357357356e-05,
754
+ "loss": 0.2895,
755
+ "step": 950
756
+ },
757
+ {
758
+ "epoch": 2.8828828828828827,
759
+ "grad_norm": 2.255638360977173,
760
+ "learning_rate": 5.585585585585585e-05,
761
+ "loss": 0.2138,
762
+ "step": 960
763
+ },
764
+ {
765
+ "epoch": 2.9129129129129128,
766
+ "grad_norm": 3.9691970348358154,
767
+ "learning_rate": 5.435435435435435e-05,
768
+ "loss": 0.1973,
769
+ "step": 970
770
+ },
771
+ {
772
+ "epoch": 2.942942942942943,
773
+ "grad_norm": 2.5032782554626465,
774
+ "learning_rate": 5.2852852852852855e-05,
775
+ "loss": 0.2642,
776
+ "step": 980
777
+ },
778
+ {
779
+ "epoch": 2.972972972972973,
780
+ "grad_norm": 4.827852725982666,
781
+ "learning_rate": 5.135135135135135e-05,
782
+ "loss": 0.2679,
783
+ "step": 990
784
+ },
785
+ {
786
+ "epoch": 3.003003003003003,
787
+ "grad_norm": 0.1294238269329071,
788
+ "learning_rate": 4.984984984984985e-05,
789
+ "loss": 0.2332,
790
+ "step": 1000
791
+ },
792
+ {
793
+ "epoch": 3.003003003003003,
794
+ "eval_accuracy": 0.9043570669500531,
795
+ "eval_loss": 0.3445766568183899,
796
+ "eval_runtime": 26.6419,
797
+ "eval_samples_per_second": 35.32,
798
+ "eval_steps_per_second": 4.429,
799
+ "step": 1000
800
+ },
801
+ {
802
+ "epoch": 3.033033033033033,
803
+ "grad_norm": 3.9579365253448486,
804
+ "learning_rate": 4.834834834834835e-05,
805
+ "loss": 0.196,
806
+ "step": 1010
807
+ },
808
+ {
809
+ "epoch": 3.063063063063063,
810
+ "grad_norm": 3.364076852798462,
811
+ "learning_rate": 4.684684684684685e-05,
812
+ "loss": 0.2462,
813
+ "step": 1020
814
+ },
815
+ {
816
+ "epoch": 3.093093093093093,
817
+ "grad_norm": 2.6246888637542725,
818
+ "learning_rate": 4.5345345345345345e-05,
819
+ "loss": 0.1611,
820
+ "step": 1030
821
+ },
822
+ {
823
+ "epoch": 3.123123123123123,
824
+ "grad_norm": 0.28202980756759644,
825
+ "learning_rate": 4.384384384384385e-05,
826
+ "loss": 0.1105,
827
+ "step": 1040
828
+ },
829
+ {
830
+ "epoch": 3.153153153153153,
831
+ "grad_norm": 0.38636475801467896,
832
+ "learning_rate": 4.234234234234234e-05,
833
+ "loss": 0.1293,
834
+ "step": 1050
835
+ },
836
+ {
837
+ "epoch": 3.1831831831831834,
838
+ "grad_norm": 2.700636625289917,
839
+ "learning_rate": 4.0840840840840845e-05,
840
+ "loss": 0.2852,
841
+ "step": 1060
842
+ },
843
+ {
844
+ "epoch": 3.2132132132132134,
845
+ "grad_norm": 5.883915424346924,
846
+ "learning_rate": 3.933933933933934e-05,
847
+ "loss": 0.1586,
848
+ "step": 1070
849
+ },
850
+ {
851
+ "epoch": 3.2432432432432434,
852
+ "grad_norm": 0.6246801614761353,
853
+ "learning_rate": 3.783783783783784e-05,
854
+ "loss": 0.1551,
855
+ "step": 1080
856
+ },
857
+ {
858
+ "epoch": 3.2732732732732734,
859
+ "grad_norm": 1.7150659561157227,
860
+ "learning_rate": 3.633633633633634e-05,
861
+ "loss": 0.1931,
862
+ "step": 1090
863
+ },
864
+ {
865
+ "epoch": 3.3033033033033035,
866
+ "grad_norm": 0.42575761675834656,
867
+ "learning_rate": 3.483483483483483e-05,
868
+ "loss": 0.1173,
869
+ "step": 1100
870
+ },
871
+ {
872
+ "epoch": 3.3033033033033035,
873
+ "eval_accuracy": 0.9415515409139213,
874
+ "eval_loss": 0.23813442885875702,
875
+ "eval_runtime": 26.6996,
876
+ "eval_samples_per_second": 35.244,
877
+ "eval_steps_per_second": 4.42,
878
+ "step": 1100
879
+ },
880
+ {
881
+ "epoch": 3.3333333333333335,
882
+ "grad_norm": 1.8270289897918701,
883
+ "learning_rate": 3.3333333333333335e-05,
884
+ "loss": 0.3149,
885
+ "step": 1110
886
+ },
887
+ {
888
+ "epoch": 3.3633633633633635,
889
+ "grad_norm": 0.11395557224750519,
890
+ "learning_rate": 3.183183183183183e-05,
891
+ "loss": 0.2494,
892
+ "step": 1120
893
+ },
894
+ {
895
+ "epoch": 3.3933933933933935,
896
+ "grad_norm": 3.1319284439086914,
897
+ "learning_rate": 3.0330330330330332e-05,
898
+ "loss": 0.1403,
899
+ "step": 1130
900
+ },
901
+ {
902
+ "epoch": 3.4234234234234235,
903
+ "grad_norm": 1.6305768489837646,
904
+ "learning_rate": 2.882882882882883e-05,
905
+ "loss": 0.2106,
906
+ "step": 1140
907
+ },
908
+ {
909
+ "epoch": 3.4534534534534536,
910
+ "grad_norm": 2.2145802974700928,
911
+ "learning_rate": 2.732732732732733e-05,
912
+ "loss": 0.1435,
913
+ "step": 1150
914
+ },
915
+ {
916
+ "epoch": 3.4834834834834836,
917
+ "grad_norm": 1.2664133310317993,
918
+ "learning_rate": 2.582582582582583e-05,
919
+ "loss": 0.2026,
920
+ "step": 1160
921
+ },
922
+ {
923
+ "epoch": 3.5135135135135136,
924
+ "grad_norm": 0.11266277730464935,
925
+ "learning_rate": 2.4324324324324327e-05,
926
+ "loss": 0.2054,
927
+ "step": 1170
928
+ },
929
+ {
930
+ "epoch": 3.5435435435435436,
931
+ "grad_norm": 4.44317626953125,
932
+ "learning_rate": 2.2822822822822822e-05,
933
+ "loss": 0.1943,
934
+ "step": 1180
935
+ },
936
+ {
937
+ "epoch": 3.5735735735735736,
938
+ "grad_norm": 2.41487979888916,
939
+ "learning_rate": 2.132132132132132e-05,
940
+ "loss": 0.2326,
941
+ "step": 1190
942
+ },
943
+ {
944
+ "epoch": 3.6036036036036037,
945
+ "grad_norm": 6.012082099914551,
946
+ "learning_rate": 1.981981981981982e-05,
947
+ "loss": 0.2694,
948
+ "step": 1200
949
+ },
950
+ {
951
+ "epoch": 3.6036036036036037,
952
+ "eval_accuracy": 0.9426142401700319,
953
+ "eval_loss": 0.21462436020374298,
954
+ "eval_runtime": 26.5228,
955
+ "eval_samples_per_second": 35.479,
956
+ "eval_steps_per_second": 4.449,
957
+ "step": 1200
958
+ },
959
+ {
960
+ "epoch": 3.6336336336336337,
961
+ "grad_norm": 4.844743251800537,
962
+ "learning_rate": 1.831831831831832e-05,
963
+ "loss": 0.1067,
964
+ "step": 1210
965
+ },
966
+ {
967
+ "epoch": 3.6636636636636637,
968
+ "grad_norm": 0.3227928876876831,
969
+ "learning_rate": 1.6816816816816817e-05,
970
+ "loss": 0.0959,
971
+ "step": 1220
972
+ },
973
+ {
974
+ "epoch": 3.6936936936936937,
975
+ "grad_norm": 1.930338978767395,
976
+ "learning_rate": 1.5315315315315316e-05,
977
+ "loss": 0.2313,
978
+ "step": 1230
979
+ },
980
+ {
981
+ "epoch": 3.7237237237237237,
982
+ "grad_norm": 4.675666332244873,
983
+ "learning_rate": 1.3813813813813815e-05,
984
+ "loss": 0.1652,
985
+ "step": 1240
986
+ },
987
+ {
988
+ "epoch": 3.7537537537537538,
989
+ "grad_norm": 0.07279150187969208,
990
+ "learning_rate": 1.2312312312312313e-05,
991
+ "loss": 0.1007,
992
+ "step": 1250
993
+ },
994
+ {
995
+ "epoch": 3.7837837837837838,
996
+ "grad_norm": 0.06191149726510048,
997
+ "learning_rate": 1.0810810810810812e-05,
998
+ "loss": 0.1737,
999
+ "step": 1260
1000
+ },
1001
+ {
1002
+ "epoch": 3.813813813813814,
1003
+ "grad_norm": 0.23188060522079468,
1004
+ "learning_rate": 9.309309309309309e-06,
1005
+ "loss": 0.1923,
1006
+ "step": 1270
1007
+ },
1008
+ {
1009
+ "epoch": 3.843843843843844,
1010
+ "grad_norm": 0.0755883976817131,
1011
+ "learning_rate": 7.807807807807808e-06,
1012
+ "loss": 0.0793,
1013
+ "step": 1280
1014
+ },
1015
+ {
1016
+ "epoch": 3.873873873873874,
1017
+ "grad_norm": 3.1278254985809326,
1018
+ "learning_rate": 6.306306306306306e-06,
1019
+ "loss": 0.1839,
1020
+ "step": 1290
1021
+ },
1022
+ {
1023
+ "epoch": 3.903903903903904,
1024
+ "grad_norm": 0.0690230205655098,
1025
+ "learning_rate": 4.804804804804805e-06,
1026
+ "loss": 0.1227,
1027
+ "step": 1300
1028
+ },
1029
+ {
1030
+ "epoch": 3.903903903903904,
1031
+ "eval_accuracy": 0.9489904357066951,
1032
+ "eval_loss": 0.22585047781467438,
1033
+ "eval_runtime": 26.5743,
1034
+ "eval_samples_per_second": 35.41,
1035
+ "eval_steps_per_second": 4.44,
1036
+ "step": 1300
1037
+ },
1038
+ {
1039
+ "epoch": 3.933933933933934,
1040
+ "grad_norm": 4.087301254272461,
1041
+ "learning_rate": 3.3033033033033035e-06,
1042
+ "loss": 0.1755,
1043
+ "step": 1310
1044
+ },
1045
+ {
1046
+ "epoch": 3.963963963963964,
1047
+ "grad_norm": 1.5509567260742188,
1048
+ "learning_rate": 1.801801801801802e-06,
1049
+ "loss": 0.2882,
1050
+ "step": 1320
1051
+ },
1052
+ {
1053
+ "epoch": 3.993993993993994,
1054
+ "grad_norm": 0.1417875587940216,
1055
+ "learning_rate": 3.003003003003003e-07,
1056
+ "loss": 0.1315,
1057
+ "step": 1330
1058
+ },
1059
+ {
1060
+ "epoch": 4.0,
1061
+ "step": 1332,
1062
+ "total_flos": 1.651775717862015e+18,
1063
+ "train_loss": 0.4770685946261203,
1064
+ "train_runtime": 1116.421,
1065
+ "train_samples_per_second": 19.09,
1066
+ "train_steps_per_second": 1.193
1067
+ }
1068
+ ],
1069
+ "logging_steps": 10,
1070
+ "max_steps": 1332,
1071
+ "num_input_tokens_seen": 0,
1072
+ "num_train_epochs": 4,
1073
+ "save_steps": 100,
1074
+ "stateful_callbacks": {
1075
+ "TrainerControl": {
1076
+ "args": {
1077
+ "should_epoch_stop": false,
1078
+ "should_evaluate": false,
1079
+ "should_log": false,
1080
+ "should_save": true,
1081
+ "should_training_stop": true
1082
+ },
1083
+ "attributes": {}
1084
+ }
1085
+ },
1086
+ "total_flos": 1.651775717862015e+18,
1087
+ "train_batch_size": 16,
1088
+ "trial_name": null,
1089
+ "trial_params": null
1090
+ }