Sebsa commited on
Commit
b720d6a
·
verified ·
1 Parent(s): 883c6e2

Upload folder using huggingface_hub

Browse files
README.md ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ ---
3
+ tags:
4
+ - autotrain
5
+ - text-regression
6
+ base_model: distilbert/distilbert-base-cased
7
+ widget:
8
+ - text: "I love AutoTrain"
9
+ ---
10
+
11
+ # Model Trained Using AutoTrain
12
+
13
+ - Problem type: Text Regression
14
+
15
+ ## Validation Metrics
16
+ loss: 0.04696119576692581
17
+
18
+ mse: 0.046961188316345215
19
+
20
+ mae: 0.16446031630039215
21
+
22
+ r2: 0.5230978727340698
23
+
24
+ rmse: 0.216705302926221
25
+
26
+ explained_variance: 0.5387225151062012
checkpoint-228/config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilbert/distilbert-base-cased",
3
+ "_num_labels": 1,
4
+ "activation": "gelu",
5
+ "architectures": [
6
+ "DistilBertForSequenceClassification"
7
+ ],
8
+ "attention_dropout": 0.1,
9
+ "dim": 768,
10
+ "dropout": 0.1,
11
+ "hidden_dim": 3072,
12
+ "id2label": {
13
+ "0": "target"
14
+ },
15
+ "initializer_range": 0.02,
16
+ "label2id": {
17
+ "target": 0
18
+ },
19
+ "max_position_embeddings": 512,
20
+ "model_type": "distilbert",
21
+ "n_heads": 12,
22
+ "n_layers": 6,
23
+ "output_past": true,
24
+ "pad_token_id": 0,
25
+ "problem_type": "regression",
26
+ "qa_dropout": 0.1,
27
+ "seq_classif_dropout": 0.2,
28
+ "sinusoidal_pos_embds": false,
29
+ "tie_weights_": true,
30
+ "torch_dtype": "float32",
31
+ "transformers_version": "4.48.0",
32
+ "vocab_size": 28996
33
+ }
checkpoint-228/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:705d4b85898e039d6d32d8851592d90fd631c16105efa2c7695c4151dbf63f1f
3
+ size 263141604
checkpoint-228/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d10e2a2a2cce3afe07a6e0e6bbaf108b3cd6776d133d26e7f3cd233ba2731bbe
3
+ size 526342522
checkpoint-228/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da289b7c136fb4f0be3b3b8bcd07137af289665b63b3ed6bfe09a03c11be6ecf
3
+ size 13990
checkpoint-228/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c573af203cd37dc5e2d965d1183bd1cf3b240b8b0449f0532e927393eeb729c8
3
+ size 1064
checkpoint-228/trainer_state.json ADDED
@@ -0,0 +1,613 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.04696119576692581,
3
+ "best_model_checkpoint": "autotrain-bert-tone/checkpoint-228",
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 228,
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.039473684210526314,
13
+ "grad_norm": 1.268290400505066,
14
+ "learning_rate": 6.521739130434783e-06,
15
+ "loss": 0.1598,
16
+ "step": 3
17
+ },
18
+ {
19
+ "epoch": 0.07894736842105263,
20
+ "grad_norm": 2.4602935314178467,
21
+ "learning_rate": 1.3043478260869566e-05,
22
+ "loss": 0.1091,
23
+ "step": 6
24
+ },
25
+ {
26
+ "epoch": 0.11842105263157894,
27
+ "grad_norm": 3.914072275161743,
28
+ "learning_rate": 1.956521739130435e-05,
29
+ "loss": 0.0937,
30
+ "step": 9
31
+ },
32
+ {
33
+ "epoch": 0.15789473684210525,
34
+ "grad_norm": 0.8280628323554993,
35
+ "learning_rate": 2.608695652173913e-05,
36
+ "loss": 0.0976,
37
+ "step": 12
38
+ },
39
+ {
40
+ "epoch": 0.19736842105263158,
41
+ "grad_norm": 1.6599308252334595,
42
+ "learning_rate": 3.260869565217392e-05,
43
+ "loss": 0.106,
44
+ "step": 15
45
+ },
46
+ {
47
+ "epoch": 0.23684210526315788,
48
+ "grad_norm": 1.8458030223846436,
49
+ "learning_rate": 3.91304347826087e-05,
50
+ "loss": 0.0606,
51
+ "step": 18
52
+ },
53
+ {
54
+ "epoch": 0.27631578947368424,
55
+ "grad_norm": 2.4959943294525146,
56
+ "learning_rate": 4.565217391304348e-05,
57
+ "loss": 0.1703,
58
+ "step": 21
59
+ },
60
+ {
61
+ "epoch": 0.3157894736842105,
62
+ "grad_norm": 1.732654333114624,
63
+ "learning_rate": 4.975609756097561e-05,
64
+ "loss": 0.0757,
65
+ "step": 24
66
+ },
67
+ {
68
+ "epoch": 0.35526315789473684,
69
+ "grad_norm": 1.4400056600570679,
70
+ "learning_rate": 4.902439024390244e-05,
71
+ "loss": 0.0641,
72
+ "step": 27
73
+ },
74
+ {
75
+ "epoch": 0.39473684210526316,
76
+ "grad_norm": 4.92896032333374,
77
+ "learning_rate": 4.829268292682927e-05,
78
+ "loss": 0.1041,
79
+ "step": 30
80
+ },
81
+ {
82
+ "epoch": 0.4342105263157895,
83
+ "grad_norm": 1.7027392387390137,
84
+ "learning_rate": 4.75609756097561e-05,
85
+ "loss": 0.0797,
86
+ "step": 33
87
+ },
88
+ {
89
+ "epoch": 0.47368421052631576,
90
+ "grad_norm": 1.845708966255188,
91
+ "learning_rate": 4.682926829268293e-05,
92
+ "loss": 0.1277,
93
+ "step": 36
94
+ },
95
+ {
96
+ "epoch": 0.5131578947368421,
97
+ "grad_norm": 2.450996160507202,
98
+ "learning_rate": 4.609756097560976e-05,
99
+ "loss": 0.0925,
100
+ "step": 39
101
+ },
102
+ {
103
+ "epoch": 0.5526315789473685,
104
+ "grad_norm": 1.4300225973129272,
105
+ "learning_rate": 4.536585365853659e-05,
106
+ "loss": 0.0921,
107
+ "step": 42
108
+ },
109
+ {
110
+ "epoch": 0.5921052631578947,
111
+ "grad_norm": 1.470561146736145,
112
+ "learning_rate": 4.4634146341463416e-05,
113
+ "loss": 0.0468,
114
+ "step": 45
115
+ },
116
+ {
117
+ "epoch": 0.631578947368421,
118
+ "grad_norm": 1.5631119012832642,
119
+ "learning_rate": 4.390243902439025e-05,
120
+ "loss": 0.0562,
121
+ "step": 48
122
+ },
123
+ {
124
+ "epoch": 0.6710526315789473,
125
+ "grad_norm": 2.8870129585266113,
126
+ "learning_rate": 4.317073170731707e-05,
127
+ "loss": 0.0699,
128
+ "step": 51
129
+ },
130
+ {
131
+ "epoch": 0.7105263157894737,
132
+ "grad_norm": 3.935586929321289,
133
+ "learning_rate": 4.2439024390243905e-05,
134
+ "loss": 0.0971,
135
+ "step": 54
136
+ },
137
+ {
138
+ "epoch": 0.75,
139
+ "grad_norm": 1.9553509950637817,
140
+ "learning_rate": 4.170731707317073e-05,
141
+ "loss": 0.0584,
142
+ "step": 57
143
+ },
144
+ {
145
+ "epoch": 0.7894736842105263,
146
+ "grad_norm": 2.7788777351379395,
147
+ "learning_rate": 4.097560975609756e-05,
148
+ "loss": 0.0691,
149
+ "step": 60
150
+ },
151
+ {
152
+ "epoch": 0.8289473684210527,
153
+ "grad_norm": 1.335891604423523,
154
+ "learning_rate": 4.0243902439024395e-05,
155
+ "loss": 0.0866,
156
+ "step": 63
157
+ },
158
+ {
159
+ "epoch": 0.868421052631579,
160
+ "grad_norm": 2.5660781860351562,
161
+ "learning_rate": 3.951219512195122e-05,
162
+ "loss": 0.0626,
163
+ "step": 66
164
+ },
165
+ {
166
+ "epoch": 0.9078947368421053,
167
+ "grad_norm": 1.0248908996582031,
168
+ "learning_rate": 3.878048780487805e-05,
169
+ "loss": 0.1038,
170
+ "step": 69
171
+ },
172
+ {
173
+ "epoch": 0.9473684210526315,
174
+ "grad_norm": 1.2834587097167969,
175
+ "learning_rate": 3.804878048780488e-05,
176
+ "loss": 0.0724,
177
+ "step": 72
178
+ },
179
+ {
180
+ "epoch": 0.9868421052631579,
181
+ "grad_norm": 1.8682374954223633,
182
+ "learning_rate": 3.731707317073171e-05,
183
+ "loss": 0.06,
184
+ "step": 75
185
+ },
186
+ {
187
+ "epoch": 1.0,
188
+ "eval_explained_variance": 0.4236792325973511,
189
+ "eval_loss": 0.10101328790187836,
190
+ "eval_mae": 0.25109994411468506,
191
+ "eval_mse": 0.10101328045129776,
192
+ "eval_r2": -0.025814056396484375,
193
+ "eval_rmse": 0.317825864981593,
194
+ "eval_runtime": 12.7804,
195
+ "eval_samples_per_second": 11.815,
196
+ "eval_steps_per_second": 0.782,
197
+ "step": 76
198
+ },
199
+ {
200
+ "epoch": 1.0263157894736843,
201
+ "grad_norm": 2.012385606765747,
202
+ "learning_rate": 3.6585365853658535e-05,
203
+ "loss": 0.07,
204
+ "step": 78
205
+ },
206
+ {
207
+ "epoch": 1.0657894736842106,
208
+ "grad_norm": 1.5033577680587769,
209
+ "learning_rate": 3.585365853658537e-05,
210
+ "loss": 0.0595,
211
+ "step": 81
212
+ },
213
+ {
214
+ "epoch": 1.1052631578947367,
215
+ "grad_norm": 1.9349753856658936,
216
+ "learning_rate": 3.512195121951219e-05,
217
+ "loss": 0.0519,
218
+ "step": 84
219
+ },
220
+ {
221
+ "epoch": 1.1447368421052633,
222
+ "grad_norm": 1.1432642936706543,
223
+ "learning_rate": 3.4390243902439025e-05,
224
+ "loss": 0.0406,
225
+ "step": 87
226
+ },
227
+ {
228
+ "epoch": 1.1842105263157894,
229
+ "grad_norm": 1.307976245880127,
230
+ "learning_rate": 3.365853658536586e-05,
231
+ "loss": 0.0312,
232
+ "step": 90
233
+ },
234
+ {
235
+ "epoch": 1.2236842105263157,
236
+ "grad_norm": 2.569735050201416,
237
+ "learning_rate": 3.292682926829269e-05,
238
+ "loss": 0.0546,
239
+ "step": 93
240
+ },
241
+ {
242
+ "epoch": 1.263157894736842,
243
+ "grad_norm": 1.499475359916687,
244
+ "learning_rate": 3.2195121951219514e-05,
245
+ "loss": 0.0388,
246
+ "step": 96
247
+ },
248
+ {
249
+ "epoch": 1.3026315789473684,
250
+ "grad_norm": 2.22824764251709,
251
+ "learning_rate": 3.146341463414634e-05,
252
+ "loss": 0.0473,
253
+ "step": 99
254
+ },
255
+ {
256
+ "epoch": 1.3421052631578947,
257
+ "grad_norm": 2.3919763565063477,
258
+ "learning_rate": 3.073170731707317e-05,
259
+ "loss": 0.0633,
260
+ "step": 102
261
+ },
262
+ {
263
+ "epoch": 1.381578947368421,
264
+ "grad_norm": 2.487248420715332,
265
+ "learning_rate": 3e-05,
266
+ "loss": 0.0761,
267
+ "step": 105
268
+ },
269
+ {
270
+ "epoch": 1.4210526315789473,
271
+ "grad_norm": 1.0264114141464233,
272
+ "learning_rate": 2.926829268292683e-05,
273
+ "loss": 0.0243,
274
+ "step": 108
275
+ },
276
+ {
277
+ "epoch": 1.4605263157894737,
278
+ "grad_norm": 1.970423936843872,
279
+ "learning_rate": 2.8536585365853658e-05,
280
+ "loss": 0.0415,
281
+ "step": 111
282
+ },
283
+ {
284
+ "epoch": 1.5,
285
+ "grad_norm": 0.9041996598243713,
286
+ "learning_rate": 2.780487804878049e-05,
287
+ "loss": 0.032,
288
+ "step": 114
289
+ },
290
+ {
291
+ "epoch": 1.5394736842105263,
292
+ "grad_norm": 1.3935425281524658,
293
+ "learning_rate": 2.707317073170732e-05,
294
+ "loss": 0.0447,
295
+ "step": 117
296
+ },
297
+ {
298
+ "epoch": 1.5789473684210527,
299
+ "grad_norm": 1.299230694770813,
300
+ "learning_rate": 2.6341463414634148e-05,
301
+ "loss": 0.0362,
302
+ "step": 120
303
+ },
304
+ {
305
+ "epoch": 1.618421052631579,
306
+ "grad_norm": 1.1066516637802124,
307
+ "learning_rate": 2.5609756097560977e-05,
308
+ "loss": 0.0172,
309
+ "step": 123
310
+ },
311
+ {
312
+ "epoch": 1.6578947368421053,
313
+ "grad_norm": 1.3599709272384644,
314
+ "learning_rate": 2.4878048780487805e-05,
315
+ "loss": 0.0248,
316
+ "step": 126
317
+ },
318
+ {
319
+ "epoch": 1.6973684210526314,
320
+ "grad_norm": 1.9708794355392456,
321
+ "learning_rate": 2.4146341463414634e-05,
322
+ "loss": 0.0593,
323
+ "step": 129
324
+ },
325
+ {
326
+ "epoch": 1.736842105263158,
327
+ "grad_norm": 1.0266876220703125,
328
+ "learning_rate": 2.3414634146341466e-05,
329
+ "loss": 0.0247,
330
+ "step": 132
331
+ },
332
+ {
333
+ "epoch": 1.776315789473684,
334
+ "grad_norm": 2.81319260597229,
335
+ "learning_rate": 2.2682926829268295e-05,
336
+ "loss": 0.0489,
337
+ "step": 135
338
+ },
339
+ {
340
+ "epoch": 1.8157894736842106,
341
+ "grad_norm": 0.9068326950073242,
342
+ "learning_rate": 2.1951219512195124e-05,
343
+ "loss": 0.0494,
344
+ "step": 138
345
+ },
346
+ {
347
+ "epoch": 1.8552631578947367,
348
+ "grad_norm": 0.703267514705658,
349
+ "learning_rate": 2.1219512195121953e-05,
350
+ "loss": 0.0473,
351
+ "step": 141
352
+ },
353
+ {
354
+ "epoch": 1.8947368421052633,
355
+ "grad_norm": 1.852679967880249,
356
+ "learning_rate": 2.048780487804878e-05,
357
+ "loss": 0.0359,
358
+ "step": 144
359
+ },
360
+ {
361
+ "epoch": 1.9342105263157894,
362
+ "grad_norm": 1.1914632320404053,
363
+ "learning_rate": 1.975609756097561e-05,
364
+ "loss": 0.0228,
365
+ "step": 147
366
+ },
367
+ {
368
+ "epoch": 1.973684210526316,
369
+ "grad_norm": 1.3655986785888672,
370
+ "learning_rate": 1.902439024390244e-05,
371
+ "loss": 0.0281,
372
+ "step": 150
373
+ },
374
+ {
375
+ "epoch": 2.0,
376
+ "eval_explained_variance": 0.4443961977958679,
377
+ "eval_loss": 0.05540888383984566,
378
+ "eval_mae": 0.18671312928199768,
379
+ "eval_mse": 0.055408887565135956,
380
+ "eval_r2": 0.43730950355529785,
381
+ "eval_rmse": 0.23539092498466452,
382
+ "eval_runtime": 13.0934,
383
+ "eval_samples_per_second": 11.533,
384
+ "eval_steps_per_second": 0.764,
385
+ "step": 152
386
+ },
387
+ {
388
+ "epoch": 2.013157894736842,
389
+ "grad_norm": 1.1334549188613892,
390
+ "learning_rate": 1.8292682926829268e-05,
391
+ "loss": 0.0353,
392
+ "step": 153
393
+ },
394
+ {
395
+ "epoch": 2.0526315789473686,
396
+ "grad_norm": 1.1114447116851807,
397
+ "learning_rate": 1.7560975609756096e-05,
398
+ "loss": 0.0164,
399
+ "step": 156
400
+ },
401
+ {
402
+ "epoch": 2.0921052631578947,
403
+ "grad_norm": 1.5025893449783325,
404
+ "learning_rate": 1.682926829268293e-05,
405
+ "loss": 0.0221,
406
+ "step": 159
407
+ },
408
+ {
409
+ "epoch": 2.1315789473684212,
410
+ "grad_norm": 0.915424644947052,
411
+ "learning_rate": 1.6097560975609757e-05,
412
+ "loss": 0.0126,
413
+ "step": 162
414
+ },
415
+ {
416
+ "epoch": 2.1710526315789473,
417
+ "grad_norm": 1.6687657833099365,
418
+ "learning_rate": 1.5365853658536586e-05,
419
+ "loss": 0.0235,
420
+ "step": 165
421
+ },
422
+ {
423
+ "epoch": 2.2105263157894735,
424
+ "grad_norm": 1.230404257774353,
425
+ "learning_rate": 1.4634146341463415e-05,
426
+ "loss": 0.0232,
427
+ "step": 168
428
+ },
429
+ {
430
+ "epoch": 2.25,
431
+ "grad_norm": 0.943490207195282,
432
+ "learning_rate": 1.3902439024390245e-05,
433
+ "loss": 0.0152,
434
+ "step": 171
435
+ },
436
+ {
437
+ "epoch": 2.2894736842105265,
438
+ "grad_norm": 0.6173422932624817,
439
+ "learning_rate": 1.3170731707317074e-05,
440
+ "loss": 0.0126,
441
+ "step": 174
442
+ },
443
+ {
444
+ "epoch": 2.3289473684210527,
445
+ "grad_norm": 1.986954689025879,
446
+ "learning_rate": 1.2439024390243903e-05,
447
+ "loss": 0.0211,
448
+ "step": 177
449
+ },
450
+ {
451
+ "epoch": 2.3684210526315788,
452
+ "grad_norm": 0.650972843170166,
453
+ "learning_rate": 1.1707317073170733e-05,
454
+ "loss": 0.01,
455
+ "step": 180
456
+ },
457
+ {
458
+ "epoch": 2.4078947368421053,
459
+ "grad_norm": 2.853421688079834,
460
+ "learning_rate": 1.0975609756097562e-05,
461
+ "loss": 0.0286,
462
+ "step": 183
463
+ },
464
+ {
465
+ "epoch": 2.4473684210526314,
466
+ "grad_norm": 1.0729972124099731,
467
+ "learning_rate": 1.024390243902439e-05,
468
+ "loss": 0.02,
469
+ "step": 186
470
+ },
471
+ {
472
+ "epoch": 2.486842105263158,
473
+ "grad_norm": 0.8739410638809204,
474
+ "learning_rate": 9.51219512195122e-06,
475
+ "loss": 0.0124,
476
+ "step": 189
477
+ },
478
+ {
479
+ "epoch": 2.526315789473684,
480
+ "grad_norm": 0.8854183554649353,
481
+ "learning_rate": 8.780487804878048e-06,
482
+ "loss": 0.0117,
483
+ "step": 192
484
+ },
485
+ {
486
+ "epoch": 2.5657894736842106,
487
+ "grad_norm": 1.0242164134979248,
488
+ "learning_rate": 8.048780487804879e-06,
489
+ "loss": 0.0293,
490
+ "step": 195
491
+ },
492
+ {
493
+ "epoch": 2.6052631578947367,
494
+ "grad_norm": 1.419499158859253,
495
+ "learning_rate": 7.317073170731707e-06,
496
+ "loss": 0.0283,
497
+ "step": 198
498
+ },
499
+ {
500
+ "epoch": 2.6447368421052633,
501
+ "grad_norm": 0.8783360123634338,
502
+ "learning_rate": 6.585365853658537e-06,
503
+ "loss": 0.0318,
504
+ "step": 201
505
+ },
506
+ {
507
+ "epoch": 2.6842105263157894,
508
+ "grad_norm": 0.4939420521259308,
509
+ "learning_rate": 5.853658536585367e-06,
510
+ "loss": 0.0107,
511
+ "step": 204
512
+ },
513
+ {
514
+ "epoch": 2.723684210526316,
515
+ "grad_norm": 1.3545787334442139,
516
+ "learning_rate": 5.121951219512195e-06,
517
+ "loss": 0.0125,
518
+ "step": 207
519
+ },
520
+ {
521
+ "epoch": 2.763157894736842,
522
+ "grad_norm": 1.0449341535568237,
523
+ "learning_rate": 4.390243902439024e-06,
524
+ "loss": 0.0211,
525
+ "step": 210
526
+ },
527
+ {
528
+ "epoch": 2.8026315789473686,
529
+ "grad_norm": 1.5390199422836304,
530
+ "learning_rate": 3.6585365853658537e-06,
531
+ "loss": 0.0243,
532
+ "step": 213
533
+ },
534
+ {
535
+ "epoch": 2.8421052631578947,
536
+ "grad_norm": 1.0012524127960205,
537
+ "learning_rate": 2.9268292682926833e-06,
538
+ "loss": 0.026,
539
+ "step": 216
540
+ },
541
+ {
542
+ "epoch": 2.8815789473684212,
543
+ "grad_norm": 0.413464218378067,
544
+ "learning_rate": 2.195121951219512e-06,
545
+ "loss": 0.0155,
546
+ "step": 219
547
+ },
548
+ {
549
+ "epoch": 2.9210526315789473,
550
+ "grad_norm": 0.6869584918022156,
551
+ "learning_rate": 1.4634146341463416e-06,
552
+ "loss": 0.0174,
553
+ "step": 222
554
+ },
555
+ {
556
+ "epoch": 2.9605263157894735,
557
+ "grad_norm": 1.1535204648971558,
558
+ "learning_rate": 7.317073170731708e-07,
559
+ "loss": 0.0122,
560
+ "step": 225
561
+ },
562
+ {
563
+ "epoch": 3.0,
564
+ "grad_norm": 1.9466148614883423,
565
+ "learning_rate": 0.0,
566
+ "loss": 0.0198,
567
+ "step": 228
568
+ },
569
+ {
570
+ "epoch": 3.0,
571
+ "eval_explained_variance": 0.5387225151062012,
572
+ "eval_loss": 0.04696119576692581,
573
+ "eval_mae": 0.16446031630039215,
574
+ "eval_mse": 0.046961188316345215,
575
+ "eval_r2": 0.5230978727340698,
576
+ "eval_rmse": 0.216705302926221,
577
+ "eval_runtime": 12.6393,
578
+ "eval_samples_per_second": 11.947,
579
+ "eval_steps_per_second": 0.791,
580
+ "step": 228
581
+ }
582
+ ],
583
+ "logging_steps": 3,
584
+ "max_steps": 228,
585
+ "num_input_tokens_seen": 0,
586
+ "num_train_epochs": 3,
587
+ "save_steps": 500,
588
+ "stateful_callbacks": {
589
+ "EarlyStoppingCallback": {
590
+ "args": {
591
+ "early_stopping_patience": 5,
592
+ "early_stopping_threshold": 0.01
593
+ },
594
+ "attributes": {
595
+ "early_stopping_patience_counter": 1
596
+ }
597
+ },
598
+ "TrainerControl": {
599
+ "args": {
600
+ "should_epoch_stop": false,
601
+ "should_evaluate": false,
602
+ "should_log": false,
603
+ "should_save": true,
604
+ "should_training_stop": true
605
+ },
606
+ "attributes": {}
607
+ }
608
+ },
609
+ "total_flos": 60006661438464.0,
610
+ "train_batch_size": 8,
611
+ "trial_name": null,
612
+ "trial_params": null
613
+ }
checkpoint-228/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42cb593a413645126d2d1fc5c868a0cd79926c751e6e34f7e579d72a21ec645b
3
+ size 5368
config.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilbert/distilbert-base-cased",
3
+ "_num_labels": 1,
4
+ "activation": "gelu",
5
+ "architectures": [
6
+ "DistilBertForSequenceClassification"
7
+ ],
8
+ "attention_dropout": 0.1,
9
+ "dim": 768,
10
+ "dropout": 0.1,
11
+ "hidden_dim": 3072,
12
+ "id2label": {
13
+ "0": "target"
14
+ },
15
+ "initializer_range": 0.02,
16
+ "label2id": {
17
+ "target": 0
18
+ },
19
+ "max_position_embeddings": 512,
20
+ "model_type": "distilbert",
21
+ "n_heads": 12,
22
+ "n_layers": 6,
23
+ "output_past": true,
24
+ "pad_token_id": 0,
25
+ "problem_type": "regression",
26
+ "qa_dropout": 0.1,
27
+ "seq_classif_dropout": 0.2,
28
+ "sinusoidal_pos_embds": false,
29
+ "tie_weights_": true,
30
+ "torch_dtype": "float32",
31
+ "transformers_version": "4.48.0",
32
+ "vocab_size": 28996
33
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:705d4b85898e039d6d32d8851592d90fd631c16105efa2c7695c4151dbf63f1f
3
+ size 263141604
runs/Jul13_14-25-53_r-sebsa-asai-newstextbert-wpktfh4a-f38c3-0o175/events.out.tfevents.1752416754.r-sebsa-asai-newstextbert-wpktfh4a-f38c3-0o175.108.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:177198aa076736b880abd4c924207261b4ff4e8010daacb63a55864358d0db26
3
- size 20511
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9409fee11df43ec818eab7faede27077a1814ae256852a9017238ecee7ce0750
3
+ size 22863
runs/Jul13_14-25-53_r-sebsa-asai-newstextbert-wpktfh4a-f38c3-0o175/events.out.tfevents.1752417444.r-sebsa-asai-newstextbert-wpktfh4a-f38c3-0o175.108.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0673a8b599992c59f83cbc3e1bd6902cb67bfadd691bbec8f551fa4deba2cd03
3
+ size 609
special_tokens_map.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "cls_token": "[CLS]",
3
+ "mask_token": "[MASK]",
4
+ "pad_token": "[PAD]",
5
+ "sep_token": "[SEP]",
6
+ "unk_token": "[UNK]"
7
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "[PAD]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "100": {
12
+ "content": "[UNK]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "101": {
20
+ "content": "[CLS]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "102": {
28
+ "content": "[SEP]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "103": {
36
+ "content": "[MASK]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "clean_up_tokenization_spaces": false,
45
+ "cls_token": "[CLS]",
46
+ "do_lower_case": false,
47
+ "extra_special_tokens": {},
48
+ "mask_token": "[MASK]",
49
+ "model_max_length": 512,
50
+ "pad_token": "[PAD]",
51
+ "sep_token": "[SEP]",
52
+ "strip_accents": null,
53
+ "tokenize_chinese_chars": true,
54
+ "tokenizer_class": "DistilBertTokenizer",
55
+ "unk_token": "[UNK]"
56
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42cb593a413645126d2d1fc5c868a0cd79926c751e6e34f7e579d72a21ec645b
3
+ size 5368
training_params.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "data_path": "autotrain-bert-tone/autotrain-data",
3
+ "model": "distilbert/distilbert-base-cased",
4
+ "lr": 5e-05,
5
+ "epochs": 3,
6
+ "max_seq_length": 128,
7
+ "batch_size": 8,
8
+ "warmup_ratio": 0.1,
9
+ "gradient_accumulation": 1,
10
+ "optimizer": "adamw_torch",
11
+ "scheduler": "linear",
12
+ "weight_decay": 0.0,
13
+ "max_grad_norm": 1.0,
14
+ "seed": 42,
15
+ "train_split": "train",
16
+ "valid_split": "validation",
17
+ "text_column": "autotrain_text",
18
+ "target_column": "autotrain_label",
19
+ "logging_steps": -1,
20
+ "project_name": "autotrain-bert-tone",
21
+ "auto_find_batch_size": false,
22
+ "mixed_precision": "fp16",
23
+ "save_total_limit": 1,
24
+ "push_to_hub": true,
25
+ "eval_strategy": "epoch",
26
+ "username": "Sebsa",
27
+ "log": "tensorboard",
28
+ "early_stopping_patience": 5,
29
+ "early_stopping_threshold": 0.01
30
+ }
vocab.txt ADDED
The diff for this file is too large to render. See raw diff