putazon commited on
Commit
a1a4713
·
verified ·
1 Parent(s): 0ef0125

putazon/SearchQueryNER-BERT

Browse files
README.md ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: bert-base-cased
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - precision
9
+ - recall
10
+ - f1
11
+ - accuracy
12
+ model-index:
13
+ - name: bert-finetuned-ner
14
+ results: []
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+ # bert-finetuned-ner
21
+
22
+ This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
23
+ It achieves the following results on the evaluation set:
24
+ - Loss: 0.0005
25
+ - Precision: 0.9999
26
+ - Recall: 0.9999
27
+ - F1: 0.9999
28
+ - Accuracy: 0.9999
29
+
30
+ ## Model description
31
+
32
+ More information needed
33
+
34
+ ## Intended uses & limitations
35
+
36
+ More information needed
37
+
38
+ ## Training and evaluation data
39
+
40
+ More information needed
41
+
42
+ ## Training procedure
43
+
44
+ ### Training hyperparameters
45
+
46
+ The following hyperparameters were used during training:
47
+ - learning_rate: 2e-05
48
+ - train_batch_size: 8
49
+ - eval_batch_size: 8
50
+ - seed: 42
51
+ - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
52
+ - lr_scheduler_type: linear
53
+ - num_epochs: 3
54
+
55
+ ### Training results
56
+
57
+ | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
58
+ |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
59
+ | 0.0011 | 1.0 | 12867 | 0.0009 | 0.9999 | 0.9999 | 0.9999 | 0.9999 |
60
+ | 0.002 | 2.0 | 25734 | 0.0004 | 0.9999 | 0.9999 | 0.9999 | 0.9999 |
61
+ | 0.0005 | 3.0 | 38601 | 0.0005 | 0.9999 | 0.9999 | 0.9999 | 0.9999 |
62
+
63
+
64
+ ### Framework versions
65
+
66
+ - Transformers 4.48.1
67
+ - Pytorch 2.5.1+cu124
68
+ - Datasets 3.2.0
69
+ - Tokenizers 0.21.0
config.json ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "bert-base-cased",
3
+ "architectures": [
4
+ "BertForTokenClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "classifier_dropout": null,
8
+ "gradient_checkpointing": false,
9
+ "hidden_act": "gelu",
10
+ "hidden_dropout_prob": 0.1,
11
+ "hidden_size": 768,
12
+ "id2label": {
13
+ "0": "O",
14
+ "1": "B-WORK_P",
15
+ "2": "I-WORK_P",
16
+ "3": "B-COUNTRY",
17
+ "4": "I-COUNTRY",
18
+ "5": "B-PHONE",
19
+ "6": "I-PHONE",
20
+ "7": "B-CITY",
21
+ "8": "I-CITY",
22
+ "9": "B-REL",
23
+ "10": "I-REL",
24
+ "11": "B-WORK_S",
25
+ "12": "I-WORK_S",
26
+ "13": "B-PER",
27
+ "14": "I-PER",
28
+ "15": "B-ORG",
29
+ "16": "I-ORG",
30
+ "17": "B-ATTR",
31
+ "18": "I-ATTR",
32
+ "19": "B-STREET",
33
+ "20": "I-STREET",
34
+ "21": "B-CITY_PART",
35
+ "22": "I-CITY_PART",
36
+ "23": "B-REGION",
37
+ "24": "I-REGION"
38
+ },
39
+ "initializer_range": 0.02,
40
+ "intermediate_size": 3072,
41
+ "label2id": {
42
+ "B-ATTR": 17,
43
+ "B-CITY": 7,
44
+ "B-CITY_PART": 21,
45
+ "B-COUNTRY": 3,
46
+ "B-ORG": 15,
47
+ "B-PER": 13,
48
+ "B-PHONE": 5,
49
+ "B-REGION": 23,
50
+ "B-REL": 9,
51
+ "B-STREET": 19,
52
+ "B-WORK_P": 1,
53
+ "B-WORK_S": 11,
54
+ "I-ATTR": 18,
55
+ "I-CITY": 8,
56
+ "I-CITY_PART": 22,
57
+ "I-COUNTRY": 4,
58
+ "I-ORG": 16,
59
+ "I-PER": 14,
60
+ "I-PHONE": 6,
61
+ "I-REGION": 24,
62
+ "I-REL": 10,
63
+ "I-STREET": 20,
64
+ "I-WORK_P": 2,
65
+ "I-WORK_S": 12,
66
+ "O": 0
67
+ },
68
+ "layer_norm_eps": 1e-12,
69
+ "max_position_embeddings": 512,
70
+ "model_type": "bert",
71
+ "num_attention_heads": 12,
72
+ "num_hidden_layers": 12,
73
+ "pad_token_id": 0,
74
+ "position_embedding_type": "absolute",
75
+ "torch_dtype": "float32",
76
+ "transformers_version": "4.48.1",
77
+ "type_vocab_size": 2,
78
+ "use_cache": true,
79
+ "vocab_size": 28996
80
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:32d8ef90a528548dae8e6cd1f72ac9bbd30e1c2c840f9338fe4fbc11da3f3873
3
+ size 430978956
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": "BertTokenizer",
55
+ "unk_token": "[UNK]"
56
+ }
trainer_state.json ADDED
@@ -0,0 +1,617 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 500,
6
+ "global_step": 38601,
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.03885909691458771,
13
+ "grad_norm": 0.45540618896484375,
14
+ "learning_rate": 1.9740939353902752e-05,
15
+ "loss": 0.4641,
16
+ "step": 500
17
+ },
18
+ {
19
+ "epoch": 0.07771819382917541,
20
+ "grad_norm": 0.05885039269924164,
21
+ "learning_rate": 1.94818787078055e-05,
22
+ "loss": 0.0073,
23
+ "step": 1000
24
+ },
25
+ {
26
+ "epoch": 0.11657729074376312,
27
+ "grad_norm": 0.0643157884478569,
28
+ "learning_rate": 1.922281806170825e-05,
29
+ "loss": 0.0066,
30
+ "step": 1500
31
+ },
32
+ {
33
+ "epoch": 0.15543638765835083,
34
+ "grad_norm": 0.025738418102264404,
35
+ "learning_rate": 1.8963757415611e-05,
36
+ "loss": 0.0055,
37
+ "step": 2000
38
+ },
39
+ {
40
+ "epoch": 0.19429548457293852,
41
+ "grad_norm": 0.006490091327577829,
42
+ "learning_rate": 1.8704696769513745e-05,
43
+ "loss": 0.0018,
44
+ "step": 2500
45
+ },
46
+ {
47
+ "epoch": 0.23315458148752624,
48
+ "grad_norm": 0.003259174292907119,
49
+ "learning_rate": 1.8445636123416492e-05,
50
+ "loss": 0.0017,
51
+ "step": 3000
52
+ },
53
+ {
54
+ "epoch": 0.27201367840211393,
55
+ "grad_norm": 0.002379123354330659,
56
+ "learning_rate": 1.8186575477319242e-05,
57
+ "loss": 0.0004,
58
+ "step": 3500
59
+ },
60
+ {
61
+ "epoch": 0.31087277531670165,
62
+ "grad_norm": 0.0013682579156011343,
63
+ "learning_rate": 1.7927514831221992e-05,
64
+ "loss": 0.0005,
65
+ "step": 4000
66
+ },
67
+ {
68
+ "epoch": 0.3497318722312893,
69
+ "grad_norm": 0.003730255179107189,
70
+ "learning_rate": 1.766845418512474e-05,
71
+ "loss": 0.0061,
72
+ "step": 4500
73
+ },
74
+ {
75
+ "epoch": 0.38859096914587704,
76
+ "grad_norm": 0.0010942475637421012,
77
+ "learning_rate": 1.740939353902749e-05,
78
+ "loss": 0.0015,
79
+ "step": 5000
80
+ },
81
+ {
82
+ "epoch": 0.42745006606046476,
83
+ "grad_norm": 0.0026132045313715935,
84
+ "learning_rate": 1.7150332892930236e-05,
85
+ "loss": 0.002,
86
+ "step": 5500
87
+ },
88
+ {
89
+ "epoch": 0.4663091629750525,
90
+ "grad_norm": 0.0005335228051990271,
91
+ "learning_rate": 1.6891272246832986e-05,
92
+ "loss": 0.0008,
93
+ "step": 6000
94
+ },
95
+ {
96
+ "epoch": 0.5051682598896402,
97
+ "grad_norm": 0.0023255019914358854,
98
+ "learning_rate": 1.6632211600735732e-05,
99
+ "loss": 0.0009,
100
+ "step": 6500
101
+ },
102
+ {
103
+ "epoch": 0.5440273568042279,
104
+ "grad_norm": 0.0013378489529713988,
105
+ "learning_rate": 1.6373150954638482e-05,
106
+ "loss": 0.0019,
107
+ "step": 7000
108
+ },
109
+ {
110
+ "epoch": 0.5828864537188155,
111
+ "grad_norm": 0.0021079631987959146,
112
+ "learning_rate": 1.611409030854123e-05,
113
+ "loss": 0.0099,
114
+ "step": 7500
115
+ },
116
+ {
117
+ "epoch": 0.6217455506334033,
118
+ "grad_norm": 0.00038598832907155156,
119
+ "learning_rate": 1.585502966244398e-05,
120
+ "loss": 0.0001,
121
+ "step": 8000
122
+ },
123
+ {
124
+ "epoch": 0.660604647547991,
125
+ "grad_norm": 0.0005731793353334069,
126
+ "learning_rate": 1.559596901634673e-05,
127
+ "loss": 0.0013,
128
+ "step": 8500
129
+ },
130
+ {
131
+ "epoch": 0.6994637444625786,
132
+ "grad_norm": 0.00023792145657353103,
133
+ "learning_rate": 1.5336908370249476e-05,
134
+ "loss": 0.0001,
135
+ "step": 9000
136
+ },
137
+ {
138
+ "epoch": 0.7383228413771664,
139
+ "grad_norm": 0.0007500049541704357,
140
+ "learning_rate": 1.5077847724152226e-05,
141
+ "loss": 0.0025,
142
+ "step": 9500
143
+ },
144
+ {
145
+ "epoch": 0.7771819382917541,
146
+ "grad_norm": 0.00044937318307347596,
147
+ "learning_rate": 1.4818787078054974e-05,
148
+ "loss": 0.0026,
149
+ "step": 10000
150
+ },
151
+ {
152
+ "epoch": 0.8160410352063419,
153
+ "grad_norm": 0.0006425898754969239,
154
+ "learning_rate": 1.4559726431957721e-05,
155
+ "loss": 0.0006,
156
+ "step": 10500
157
+ },
158
+ {
159
+ "epoch": 0.8549001321209295,
160
+ "grad_norm": 0.0006235554465092719,
161
+ "learning_rate": 1.430066578586047e-05,
162
+ "loss": 0.0012,
163
+ "step": 11000
164
+ },
165
+ {
166
+ "epoch": 0.8937592290355172,
167
+ "grad_norm": 0.0004008706600870937,
168
+ "learning_rate": 1.404160513976322e-05,
169
+ "loss": 0.0017,
170
+ "step": 11500
171
+ },
172
+ {
173
+ "epoch": 0.932618325950105,
174
+ "grad_norm": 0.0004875475133303553,
175
+ "learning_rate": 1.3782544493665968e-05,
176
+ "loss": 0.0001,
177
+ "step": 12000
178
+ },
179
+ {
180
+ "epoch": 0.9714774228646926,
181
+ "grad_norm": 0.0015087984502315521,
182
+ "learning_rate": 1.3523483847568716e-05,
183
+ "loss": 0.0011,
184
+ "step": 12500
185
+ },
186
+ {
187
+ "epoch": 1.0,
188
+ "eval_accuracy": 0.9999115207373271,
189
+ "eval_f1": 0.9998912136925701,
190
+ "eval_loss": 0.0008539558621123433,
191
+ "eval_precision": 0.9999032998912124,
192
+ "eval_recall": 0.9998791277861045,
193
+ "eval_runtime": 21.9409,
194
+ "eval_samples_per_second": 930.682,
195
+ "eval_steps_per_second": 116.358,
196
+ "step": 12867
197
+ },
198
+ {
199
+ "epoch": 1.0103365197792804,
200
+ "grad_norm": 0.0007504357490688562,
201
+ "learning_rate": 1.3264423201471466e-05,
202
+ "loss": 0.0012,
203
+ "step": 13000
204
+ },
205
+ {
206
+ "epoch": 1.049195616693868,
207
+ "grad_norm": 9.457436681259423e-05,
208
+ "learning_rate": 1.3005362555374215e-05,
209
+ "loss": 0.0001,
210
+ "step": 13500
211
+ },
212
+ {
213
+ "epoch": 1.0880547136084557,
214
+ "grad_norm": 0.0005767066031694412,
215
+ "learning_rate": 1.2746301909276963e-05,
216
+ "loss": 0.0006,
217
+ "step": 14000
218
+ },
219
+ {
220
+ "epoch": 1.1269138105230434,
221
+ "grad_norm": 0.0004582173132803291,
222
+ "learning_rate": 1.2487241263179713e-05,
223
+ "loss": 0.0045,
224
+ "step": 14500
225
+ },
226
+ {
227
+ "epoch": 1.165772907437631,
228
+ "grad_norm": 0.0006144808721728623,
229
+ "learning_rate": 1.222818061708246e-05,
230
+ "loss": 0.0053,
231
+ "step": 15000
232
+ },
233
+ {
234
+ "epoch": 1.2046320043522187,
235
+ "grad_norm": 0.0002801430528052151,
236
+ "learning_rate": 1.1969119970985208e-05,
237
+ "loss": 0.0005,
238
+ "step": 15500
239
+ },
240
+ {
241
+ "epoch": 1.2434911012668066,
242
+ "grad_norm": 0.001027058344334364,
243
+ "learning_rate": 1.1710059324887957e-05,
244
+ "loss": 0.002,
245
+ "step": 16000
246
+ },
247
+ {
248
+ "epoch": 1.2823501981813943,
249
+ "grad_norm": 0.00030287433764897287,
250
+ "learning_rate": 1.1450998678790705e-05,
251
+ "loss": 0.0001,
252
+ "step": 16500
253
+ },
254
+ {
255
+ "epoch": 1.321209295095982,
256
+ "grad_norm": 9.348896855954081e-05,
257
+ "learning_rate": 1.1191938032693455e-05,
258
+ "loss": 0.0,
259
+ "step": 17000
260
+ },
261
+ {
262
+ "epoch": 1.3600683920105696,
263
+ "grad_norm": 4.3804615415865555e-05,
264
+ "learning_rate": 1.0932877386596204e-05,
265
+ "loss": 0.0,
266
+ "step": 17500
267
+ },
268
+ {
269
+ "epoch": 1.3989274889251573,
270
+ "grad_norm": 5.0469727284507826e-05,
271
+ "learning_rate": 1.0673816740498952e-05,
272
+ "loss": 0.0012,
273
+ "step": 18000
274
+ },
275
+ {
276
+ "epoch": 1.4377865858397452,
277
+ "grad_norm": 8.691708353580907e-05,
278
+ "learning_rate": 1.0414756094401702e-05,
279
+ "loss": 0.0004,
280
+ "step": 18500
281
+ },
282
+ {
283
+ "epoch": 1.4766456827543328,
284
+ "grad_norm": 6.273793405853212e-05,
285
+ "learning_rate": 1.0155695448304449e-05,
286
+ "loss": 0.0005,
287
+ "step": 19000
288
+ },
289
+ {
290
+ "epoch": 1.5155047796689205,
291
+ "grad_norm": 9.822064748732373e-05,
292
+ "learning_rate": 9.896634802207197e-06,
293
+ "loss": 0.0,
294
+ "step": 19500
295
+ },
296
+ {
297
+ "epoch": 1.5543638765835082,
298
+ "grad_norm": 0.00012862567382398993,
299
+ "learning_rate": 9.637574156109945e-06,
300
+ "loss": 0.0005,
301
+ "step": 20000
302
+ },
303
+ {
304
+ "epoch": 1.5932229734980958,
305
+ "grad_norm": 0.013405167497694492,
306
+ "learning_rate": 9.378513510012694e-06,
307
+ "loss": 0.0005,
308
+ "step": 20500
309
+ },
310
+ {
311
+ "epoch": 1.6320820704126837,
312
+ "grad_norm": 6.82063982822001e-05,
313
+ "learning_rate": 9.119452863915444e-06,
314
+ "loss": 0.0,
315
+ "step": 21000
316
+ },
317
+ {
318
+ "epoch": 1.6709411673272712,
319
+ "grad_norm": 0.0001184305947390385,
320
+ "learning_rate": 8.860392217818192e-06,
321
+ "loss": 0.0006,
322
+ "step": 21500
323
+ },
324
+ {
325
+ "epoch": 1.709800264241859,
326
+ "grad_norm": 0.0004617497615981847,
327
+ "learning_rate": 8.60133157172094e-06,
328
+ "loss": 0.0,
329
+ "step": 22000
330
+ },
331
+ {
332
+ "epoch": 1.7486593611564467,
333
+ "grad_norm": 9.478950232733041e-05,
334
+ "learning_rate": 8.342270925623689e-06,
335
+ "loss": 0.0,
336
+ "step": 22500
337
+ },
338
+ {
339
+ "epoch": 1.7875184580710344,
340
+ "grad_norm": 0.0005689120152965188,
341
+ "learning_rate": 8.083210279526437e-06,
342
+ "loss": 0.0013,
343
+ "step": 23000
344
+ },
345
+ {
346
+ "epoch": 1.8263775549856223,
347
+ "grad_norm": 3.0777548090554774e-05,
348
+ "learning_rate": 7.824149633429187e-06,
349
+ "loss": 0.0002,
350
+ "step": 23500
351
+ },
352
+ {
353
+ "epoch": 1.8652366519002097,
354
+ "grad_norm": 0.00029757359880022705,
355
+ "learning_rate": 7.565088987331936e-06,
356
+ "loss": 0.001,
357
+ "step": 24000
358
+ },
359
+ {
360
+ "epoch": 1.9040957488147976,
361
+ "grad_norm": 0.0001594646309968084,
362
+ "learning_rate": 7.306028341234683e-06,
363
+ "loss": 0.0005,
364
+ "step": 24500
365
+ },
366
+ {
367
+ "epoch": 1.9429548457293853,
368
+ "grad_norm": 3.8429996493505314e-05,
369
+ "learning_rate": 7.046967695137432e-06,
370
+ "loss": 0.0009,
371
+ "step": 25000
372
+ },
373
+ {
374
+ "epoch": 1.981813942643973,
375
+ "grad_norm": 0.0016445047222077847,
376
+ "learning_rate": 6.787907049040181e-06,
377
+ "loss": 0.002,
378
+ "step": 25500
379
+ },
380
+ {
381
+ "epoch": 2.0,
382
+ "eval_accuracy": 0.9999188940092166,
383
+ "eval_f1": 0.9998670389575854,
384
+ "eval_loss": 0.0004210107435937971,
385
+ "eval_precision": 0.9998791248640154,
386
+ "eval_recall": 0.9998549533433254,
387
+ "eval_runtime": 26.3407,
388
+ "eval_samples_per_second": 775.225,
389
+ "eval_steps_per_second": 96.922,
390
+ "step": 25734
391
+ },
392
+ {
393
+ "epoch": 2.020673039558561,
394
+ "grad_norm": 0.010017353110015392,
395
+ "learning_rate": 6.52884640294293e-06,
396
+ "loss": 0.0015,
397
+ "step": 26000
398
+ },
399
+ {
400
+ "epoch": 2.0595321364731483,
401
+ "grad_norm": 0.0025657187215983868,
402
+ "learning_rate": 6.269785756845678e-06,
403
+ "loss": 0.0028,
404
+ "step": 26500
405
+ },
406
+ {
407
+ "epoch": 2.098391233387736,
408
+ "grad_norm": 0.00022906585945747793,
409
+ "learning_rate": 6.010725110748426e-06,
410
+ "loss": 0.0001,
411
+ "step": 27000
412
+ },
413
+ {
414
+ "epoch": 2.1372503303023236,
415
+ "grad_norm": 6.71066518407315e-05,
416
+ "learning_rate": 5.751664464651175e-06,
417
+ "loss": 0.0006,
418
+ "step": 27500
419
+ },
420
+ {
421
+ "epoch": 2.1761094272169115,
422
+ "grad_norm": 0.00021527826902456582,
423
+ "learning_rate": 5.492603818553925e-06,
424
+ "loss": 0.0,
425
+ "step": 28000
426
+ },
427
+ {
428
+ "epoch": 2.2149685241314994,
429
+ "grad_norm": 9.48450033320114e-05,
430
+ "learning_rate": 5.233543172456672e-06,
431
+ "loss": 0.0005,
432
+ "step": 28500
433
+ },
434
+ {
435
+ "epoch": 2.253827621046087,
436
+ "grad_norm": 0.00010121305240318179,
437
+ "learning_rate": 4.9744825263594205e-06,
438
+ "loss": 0.0,
439
+ "step": 29000
440
+ },
441
+ {
442
+ "epoch": 2.2926867179606747,
443
+ "grad_norm": 8.369102579308674e-05,
444
+ "learning_rate": 4.71542188026217e-06,
445
+ "loss": 0.0,
446
+ "step": 29500
447
+ },
448
+ {
449
+ "epoch": 2.331545814875262,
450
+ "grad_norm": 5.066412632004358e-05,
451
+ "learning_rate": 4.456361234164918e-06,
452
+ "loss": 0.0,
453
+ "step": 30000
454
+ },
455
+ {
456
+ "epoch": 2.37040491178985,
457
+ "grad_norm": 0.00012291729217395186,
458
+ "learning_rate": 4.197300588067667e-06,
459
+ "loss": 0.0008,
460
+ "step": 30500
461
+ },
462
+ {
463
+ "epoch": 2.4092640087044375,
464
+ "grad_norm": 0.00014957574603613466,
465
+ "learning_rate": 3.938239941970416e-06,
466
+ "loss": 0.0,
467
+ "step": 31000
468
+ },
469
+ {
470
+ "epoch": 2.4481231056190254,
471
+ "grad_norm": 4.759750663652085e-05,
472
+ "learning_rate": 3.679179295873164e-06,
473
+ "loss": 0.0,
474
+ "step": 31500
475
+ },
476
+ {
477
+ "epoch": 2.4869822025336132,
478
+ "grad_norm": 3.8031983422115445e-05,
479
+ "learning_rate": 3.4201186497759125e-06,
480
+ "loss": 0.0,
481
+ "step": 32000
482
+ },
483
+ {
484
+ "epoch": 2.5258412994482007,
485
+ "grad_norm": 8.281264308607206e-05,
486
+ "learning_rate": 3.1610580036786613e-06,
487
+ "loss": 0.0,
488
+ "step": 32500
489
+ },
490
+ {
491
+ "epoch": 2.5647003963627886,
492
+ "grad_norm": 8.475359209114686e-05,
493
+ "learning_rate": 2.9019973575814097e-06,
494
+ "loss": 0.0,
495
+ "step": 33000
496
+ },
497
+ {
498
+ "epoch": 2.603559493277376,
499
+ "grad_norm": 8.979766425909474e-05,
500
+ "learning_rate": 2.6429367114841585e-06,
501
+ "loss": 0.0,
502
+ "step": 33500
503
+ },
504
+ {
505
+ "epoch": 2.642418590191964,
506
+ "grad_norm": 0.0025826424825936556,
507
+ "learning_rate": 2.3838760653869073e-06,
508
+ "loss": 0.0007,
509
+ "step": 34000
510
+ },
511
+ {
512
+ "epoch": 2.681277687106552,
513
+ "grad_norm": 8.839767542667687e-05,
514
+ "learning_rate": 2.1248154192896557e-06,
515
+ "loss": 0.0007,
516
+ "step": 34500
517
+ },
518
+ {
519
+ "epoch": 2.7201367840211392,
520
+ "grad_norm": 3.709522934514098e-05,
521
+ "learning_rate": 1.8657547731924045e-06,
522
+ "loss": 0.0,
523
+ "step": 35000
524
+ },
525
+ {
526
+ "epoch": 2.758995880935727,
527
+ "grad_norm": 0.00011620597069850191,
528
+ "learning_rate": 1.6066941270951531e-06,
529
+ "loss": 0.0005,
530
+ "step": 35500
531
+ },
532
+ {
533
+ "epoch": 2.7978549778503146,
534
+ "grad_norm": 3.431947698118165e-05,
535
+ "learning_rate": 1.3476334809979017e-06,
536
+ "loss": 0.0,
537
+ "step": 36000
538
+ },
539
+ {
540
+ "epoch": 2.8367140747649024,
541
+ "grad_norm": 8.798589260550216e-05,
542
+ "learning_rate": 1.0885728349006503e-06,
543
+ "loss": 0.0,
544
+ "step": 36500
545
+ },
546
+ {
547
+ "epoch": 2.8755731716794903,
548
+ "grad_norm": 2.6514268029131927e-05,
549
+ "learning_rate": 8.29512188803399e-07,
550
+ "loss": 0.0,
551
+ "step": 37000
552
+ },
553
+ {
554
+ "epoch": 2.914432268594078,
555
+ "grad_norm": 2.451765067235101e-05,
556
+ "learning_rate": 5.704515427061476e-07,
557
+ "loss": 0.0,
558
+ "step": 37500
559
+ },
560
+ {
561
+ "epoch": 2.9532913655086657,
562
+ "grad_norm": 1.2271469131519552e-05,
563
+ "learning_rate": 3.113908966088962e-07,
564
+ "loss": 0.0,
565
+ "step": 38000
566
+ },
567
+ {
568
+ "epoch": 2.992150462423253,
569
+ "grad_norm": 3.4994409361388534e-05,
570
+ "learning_rate": 5.233025051164478e-08,
571
+ "loss": 0.0005,
572
+ "step": 38500
573
+ },
574
+ {
575
+ "epoch": 3.0,
576
+ "eval_accuracy": 0.9999188940092166,
577
+ "eval_f1": 0.9998549533433254,
578
+ "eval_loss": 0.0004571246390696615,
579
+ "eval_precision": 0.9998549533433254,
580
+ "eval_recall": 0.9998549533433254,
581
+ "eval_runtime": 40.6059,
582
+ "eval_samples_per_second": 502.882,
583
+ "eval_steps_per_second": 62.873,
584
+ "step": 38601
585
+ },
586
+ {
587
+ "epoch": 3.0,
588
+ "step": 38601,
589
+ "total_flos": 1954036863526950.0,
590
+ "train_loss": 0.007192428305773462,
591
+ "train_runtime": 2269.357,
592
+ "train_samples_per_second": 136.071,
593
+ "train_steps_per_second": 17.01
594
+ }
595
+ ],
596
+ "logging_steps": 500,
597
+ "max_steps": 38601,
598
+ "num_input_tokens_seen": 0,
599
+ "num_train_epochs": 3,
600
+ "save_steps": 500,
601
+ "stateful_callbacks": {
602
+ "TrainerControl": {
603
+ "args": {
604
+ "should_epoch_stop": false,
605
+ "should_evaluate": false,
606
+ "should_log": false,
607
+ "should_save": true,
608
+ "should_training_stop": true
609
+ },
610
+ "attributes": {}
611
+ }
612
+ },
613
+ "total_flos": 1954036863526950.0,
614
+ "train_batch_size": 8,
615
+ "trial_name": null,
616
+ "trial_params": null
617
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eb4e35aa0daabf4204885fcb8ad3a8cc216584fa19109f8940cad5ccfb227dc4
3
+ size 5304
vocab.txt ADDED
The diff for this file is too large to render. See raw diff