shubov commited on
Commit
99b265d
·
verified ·
1 Parent(s): cbe1c01

Model save

Browse files
README.md ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - generated_from_trainer
4
+ model-index:
5
+ - name: omop_bert
6
+ results: []
7
+ ---
8
+
9
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
10
+ should probably proofread and complete it, then remove this comment. -->
11
+
12
+ # omop_bert
13
+
14
+ This model is a fine-tuned version of [](https://huggingface.co/) on the None dataset.
15
+
16
+ ## Model description
17
+
18
+ More information needed
19
+
20
+ ## Intended uses & limitations
21
+
22
+ More information needed
23
+
24
+ ## Training and evaluation data
25
+
26
+ More information needed
27
+
28
+ ## Training procedure
29
+
30
+ ### Training hyperparameters
31
+
32
+ The following hyperparameters were used during training:
33
+ - learning_rate: 5e-05
34
+ - train_batch_size: 16
35
+ - eval_batch_size: 128
36
+ - seed: 42
37
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
38
+ - lr_scheduler_type: linear
39
+ - lr_scheduler_warmup_ratio: 0.1
40
+ - training_steps: 2
41
+
42
+ ### Training results
43
+
44
+
45
+
46
+ ### Framework versions
47
+
48
+ - Transformers 4.36.2
49
+ - Pytorch 2.1.2
50
+ - Datasets 2.16.1
51
+ - Tokenizers 0.15.0
config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertForMaskedLM"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.1,
9
+ "hidden_size": 768,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 3072,
12
+ "layer_norm_eps": 1e-12,
13
+ "max_position_embeddings": 512,
14
+ "model_type": "bert",
15
+ "num_attention_heads": 12,
16
+ "num_hidden_layers": 12,
17
+ "pad_token_id": 0,
18
+ "position_embedding_type": "absolute",
19
+ "torch_dtype": "float32",
20
+ "transformers_version": "4.36.2",
21
+ "type_vocab_size": 2,
22
+ "use_cache": true,
23
+ "vocab_size": 26723
24
+ }
generation_config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "pad_token_id": 0,
4
+ "transformers_version": "4.36.2"
5
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f797389ca3a719da256713642bdf1dd8b67bdfc4cadaa1a00be7caf82ad6aa84
3
+ size 426395156
special_tokens_map.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ {
4
+ "content": "[DATE]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ {
11
+ "content": "[CODES]",
12
+ "lstrip": false,
13
+ "normalized": false,
14
+ "rstrip": false,
15
+ "single_word": false
16
+ }
17
+ ],
18
+ "cls_token": {
19
+ "content": "[CLS]",
20
+ "lstrip": false,
21
+ "normalized": false,
22
+ "rstrip": false,
23
+ "single_word": false
24
+ },
25
+ "mask_token": {
26
+ "content": "[MASK]",
27
+ "lstrip": false,
28
+ "normalized": false,
29
+ "rstrip": false,
30
+ "single_word": false
31
+ },
32
+ "pad_token": {
33
+ "content": "[PAD]",
34
+ "lstrip": false,
35
+ "normalized": false,
36
+ "rstrip": false,
37
+ "single_word": false
38
+ },
39
+ "sep_token": {
40
+ "content": "[SEP]",
41
+ "lstrip": false,
42
+ "normalized": false,
43
+ "rstrip": false,
44
+ "single_word": false
45
+ },
46
+ "unk_token": {
47
+ "content": "None",
48
+ "lstrip": false,
49
+ "normalized": false,
50
+ "rstrip": false,
51
+ "single_word": false
52
+ }
53
+ }
test/config.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "BertForMaskedLM"
4
+ ],
5
+ "attention_probs_dropout_prob": 0.1,
6
+ "classifier_dropout": null,
7
+ "hidden_act": "gelu",
8
+ "hidden_dropout_prob": 0.1,
9
+ "hidden_size": 768,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 3072,
12
+ "layer_norm_eps": 1e-12,
13
+ "max_position_embeddings": 512,
14
+ "model_type": "bert",
15
+ "num_attention_heads": 12,
16
+ "num_hidden_layers": 12,
17
+ "pad_token_id": 0,
18
+ "position_embedding_type": "absolute",
19
+ "torch_dtype": "float32",
20
+ "transformers_version": "4.36.2",
21
+ "type_vocab_size": 2,
22
+ "use_cache": true,
23
+ "vocab_size": 26723
24
+ }
test/generation_config.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "pad_token_id": 0,
4
+ "transformers_version": "4.36.2"
5
+ }
test/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b8923716339350df66fe7f9227fa35ee3b1bb19564f0eaf3b749d1a7f5387a7
3
+ size 426395156
test/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:37094b4aaa5a864158fdecc6c092c7cc40a62bee0c8a78954d95c087f7ffed3b
3
+ size 852911866
test/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d17d6c11e9441491cccbcf01cce37aa88baae51c29f56eed910755747be4745
3
+ size 14244
test/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:02f10014d5f6fa7cfb6822e8324cb833ec4e38e17bedf497cd8ae20f8573fd2e
3
+ size 1064
test/special_tokens_map.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ {
4
+ "content": "[DATE]",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ {
11
+ "content": "[CODES]",
12
+ "lstrip": false,
13
+ "normalized": false,
14
+ "rstrip": false,
15
+ "single_word": false
16
+ }
17
+ ],
18
+ "cls_token": {
19
+ "content": "[CLS]",
20
+ "lstrip": false,
21
+ "normalized": false,
22
+ "rstrip": false,
23
+ "single_word": false
24
+ },
25
+ "mask_token": {
26
+ "content": "[MASK]",
27
+ "lstrip": false,
28
+ "normalized": false,
29
+ "rstrip": false,
30
+ "single_word": false
31
+ },
32
+ "pad_token": {
33
+ "content": "[PAD]",
34
+ "lstrip": false,
35
+ "normalized": false,
36
+ "rstrip": false,
37
+ "single_word": false
38
+ },
39
+ "sep_token": {
40
+ "content": "[SEP]",
41
+ "lstrip": false,
42
+ "normalized": false,
43
+ "rstrip": false,
44
+ "single_word": false
45
+ },
46
+ "unk_token": {
47
+ "content": "None",
48
+ "lstrip": false,
49
+ "normalized": false,
50
+ "rstrip": false,
51
+ "single_word": false
52
+ }
53
+ }
test/tokenizer_config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "None",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "[CLS]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[SEP]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[PAD]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "[CODES]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "5": {
44
+ "content": "[DATE]",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "6": {
52
+ "content": "[MASK]",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ }
59
+ },
60
+ "additional_special_tokens": [
61
+ "[DATE]",
62
+ "[CODES]"
63
+ ],
64
+ "clean_up_tokenization_spaces": true,
65
+ "cls_token": "[CLS]",
66
+ "mask_token": "[MASK]",
67
+ "max_len": 512,
68
+ "model_max_length": 512,
69
+ "pad_token": "[PAD]",
70
+ "return_special_tokens_mask": true,
71
+ "sep_token": "[SEP]",
72
+ "tokenizer_class": "FixedVocabTokenizer",
73
+ "unk_token": "None",
74
+ "verbose": true
75
+ }
test/trainer_state.json ADDED
@@ -0,0 +1,427 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 0.9923011120615911,
5
+ "eval_steps": 500,
6
+ "global_step": 29000,
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.02,
13
+ "learning_rate": 4.914456800684346e-05,
14
+ "loss": 6.549,
15
+ "num_input_tokens_seen": 4096000,
16
+ "step": 500
17
+ },
18
+ {
19
+ "epoch": 0.03,
20
+ "learning_rate": 4.828913601368692e-05,
21
+ "loss": 5.7273,
22
+ "num_input_tokens_seen": 8192000,
23
+ "step": 1000
24
+ },
25
+ {
26
+ "epoch": 0.05,
27
+ "learning_rate": 4.7433704020530375e-05,
28
+ "loss": 5.4759,
29
+ "num_input_tokens_seen": 12288000,
30
+ "step": 1500
31
+ },
32
+ {
33
+ "epoch": 0.07,
34
+ "learning_rate": 4.6578272027373826e-05,
35
+ "loss": 5.2276,
36
+ "num_input_tokens_seen": 16384000,
37
+ "step": 2000
38
+ },
39
+ {
40
+ "epoch": 0.09,
41
+ "learning_rate": 4.572284003421728e-05,
42
+ "loss": 4.985,
43
+ "num_input_tokens_seen": 20480000,
44
+ "step": 2500
45
+ },
46
+ {
47
+ "epoch": 0.1,
48
+ "learning_rate": 4.486740804106074e-05,
49
+ "loss": 4.7303,
50
+ "num_input_tokens_seen": 24576000,
51
+ "step": 3000
52
+ },
53
+ {
54
+ "epoch": 0.12,
55
+ "learning_rate": 4.40119760479042e-05,
56
+ "loss": 4.4866,
57
+ "num_input_tokens_seen": 28672000,
58
+ "step": 3500
59
+ },
60
+ {
61
+ "epoch": 0.14,
62
+ "learning_rate": 4.315654405474765e-05,
63
+ "loss": 4.2891,
64
+ "num_input_tokens_seen": 32768000,
65
+ "step": 4000
66
+ },
67
+ {
68
+ "epoch": 0.15,
69
+ "learning_rate": 4.2301112061591106e-05,
70
+ "loss": 4.1568,
71
+ "num_input_tokens_seen": 36864000,
72
+ "step": 4500
73
+ },
74
+ {
75
+ "epoch": 0.17,
76
+ "learning_rate": 4.1445680068434564e-05,
77
+ "loss": 3.9851,
78
+ "num_input_tokens_seen": 40960000,
79
+ "step": 5000
80
+ },
81
+ {
82
+ "epoch": 0.19,
83
+ "learning_rate": 4.059024807527802e-05,
84
+ "loss": 3.8691,
85
+ "num_input_tokens_seen": 45056000,
86
+ "step": 5500
87
+ },
88
+ {
89
+ "epoch": 0.21,
90
+ "learning_rate": 3.973481608212147e-05,
91
+ "loss": 3.7303,
92
+ "num_input_tokens_seen": 49152000,
93
+ "step": 6000
94
+ },
95
+ {
96
+ "epoch": 0.22,
97
+ "learning_rate": 3.887938408896493e-05,
98
+ "loss": 3.6247,
99
+ "num_input_tokens_seen": 53248000,
100
+ "step": 6500
101
+ },
102
+ {
103
+ "epoch": 0.24,
104
+ "learning_rate": 3.802395209580839e-05,
105
+ "loss": 3.5127,
106
+ "num_input_tokens_seen": 57344000,
107
+ "step": 7000
108
+ },
109
+ {
110
+ "epoch": 0.26,
111
+ "learning_rate": 3.7168520102651845e-05,
112
+ "loss": 3.4229,
113
+ "num_input_tokens_seen": 61440000,
114
+ "step": 7500
115
+ },
116
+ {
117
+ "epoch": 0.27,
118
+ "learning_rate": 3.6313088109495295e-05,
119
+ "loss": 3.4015,
120
+ "num_input_tokens_seen": 65536000,
121
+ "step": 8000
122
+ },
123
+ {
124
+ "epoch": 0.29,
125
+ "learning_rate": 3.545765611633875e-05,
126
+ "loss": 3.3241,
127
+ "num_input_tokens_seen": 69632000,
128
+ "step": 8500
129
+ },
130
+ {
131
+ "epoch": 0.31,
132
+ "learning_rate": 3.460222412318221e-05,
133
+ "loss": 3.2992,
134
+ "num_input_tokens_seen": 73728000,
135
+ "step": 9000
136
+ },
137
+ {
138
+ "epoch": 0.33,
139
+ "learning_rate": 3.374679213002567e-05,
140
+ "loss": 3.2067,
141
+ "num_input_tokens_seen": 77824000,
142
+ "step": 9500
143
+ },
144
+ {
145
+ "epoch": 0.34,
146
+ "learning_rate": 3.289136013686912e-05,
147
+ "loss": 3.187,
148
+ "num_input_tokens_seen": 81920000,
149
+ "step": 10000
150
+ },
151
+ {
152
+ "epoch": 0.36,
153
+ "learning_rate": 3.2035928143712576e-05,
154
+ "loss": 3.142,
155
+ "num_input_tokens_seen": 86016000,
156
+ "step": 10500
157
+ },
158
+ {
159
+ "epoch": 0.38,
160
+ "learning_rate": 3.1180496150556034e-05,
161
+ "loss": 3.0925,
162
+ "num_input_tokens_seen": 90112000,
163
+ "step": 11000
164
+ },
165
+ {
166
+ "epoch": 0.39,
167
+ "learning_rate": 3.032506415739949e-05,
168
+ "loss": 3.048,
169
+ "num_input_tokens_seen": 94208000,
170
+ "step": 11500
171
+ },
172
+ {
173
+ "epoch": 0.41,
174
+ "learning_rate": 2.9469632164242945e-05,
175
+ "loss": 3.0123,
176
+ "num_input_tokens_seen": 98304000,
177
+ "step": 12000
178
+ },
179
+ {
180
+ "epoch": 0.43,
181
+ "learning_rate": 2.8614200171086403e-05,
182
+ "loss": 2.9888,
183
+ "num_input_tokens_seen": 102400000,
184
+ "step": 12500
185
+ },
186
+ {
187
+ "epoch": 0.44,
188
+ "learning_rate": 2.7758768177929857e-05,
189
+ "loss": 2.9684,
190
+ "num_input_tokens_seen": 106496000,
191
+ "step": 13000
192
+ },
193
+ {
194
+ "epoch": 0.46,
195
+ "learning_rate": 2.6903336184773314e-05,
196
+ "loss": 2.9042,
197
+ "num_input_tokens_seen": 110592000,
198
+ "step": 13500
199
+ },
200
+ {
201
+ "epoch": 0.48,
202
+ "learning_rate": 2.604790419161677e-05,
203
+ "loss": 2.8847,
204
+ "num_input_tokens_seen": 114688000,
205
+ "step": 14000
206
+ },
207
+ {
208
+ "epoch": 0.5,
209
+ "learning_rate": 2.5192472198460226e-05,
210
+ "loss": 2.8339,
211
+ "num_input_tokens_seen": 118784000,
212
+ "step": 14500
213
+ },
214
+ {
215
+ "epoch": 0.51,
216
+ "learning_rate": 2.433704020530368e-05,
217
+ "loss": 2.8091,
218
+ "num_input_tokens_seen": 122880000,
219
+ "step": 15000
220
+ },
221
+ {
222
+ "epoch": 0.53,
223
+ "learning_rate": 2.3481608212147134e-05,
224
+ "loss": 2.6936,
225
+ "num_input_tokens_seen": 126976000,
226
+ "step": 15500
227
+ },
228
+ {
229
+ "epoch": 0.55,
230
+ "learning_rate": 2.2626176218990592e-05,
231
+ "loss": 2.5118,
232
+ "num_input_tokens_seen": 131072000,
233
+ "step": 16000
234
+ },
235
+ {
236
+ "epoch": 0.56,
237
+ "learning_rate": 2.1770744225834046e-05,
238
+ "loss": 2.3521,
239
+ "num_input_tokens_seen": 135168000,
240
+ "step": 16500
241
+ },
242
+ {
243
+ "epoch": 0.58,
244
+ "learning_rate": 2.0915312232677503e-05,
245
+ "loss": 2.2735,
246
+ "num_input_tokens_seen": 139264000,
247
+ "step": 17000
248
+ },
249
+ {
250
+ "epoch": 0.6,
251
+ "learning_rate": 2.0059880239520957e-05,
252
+ "loss": 2.1603,
253
+ "num_input_tokens_seen": 143360000,
254
+ "step": 17500
255
+ },
256
+ {
257
+ "epoch": 0.62,
258
+ "learning_rate": 1.9204448246364415e-05,
259
+ "loss": 2.1129,
260
+ "num_input_tokens_seen": 147456000,
261
+ "step": 18000
262
+ },
263
+ {
264
+ "epoch": 0.63,
265
+ "learning_rate": 1.8349016253207872e-05,
266
+ "loss": 2.0428,
267
+ "num_input_tokens_seen": 151552000,
268
+ "step": 18500
269
+ },
270
+ {
271
+ "epoch": 0.65,
272
+ "learning_rate": 1.7493584260051327e-05,
273
+ "loss": 2.0321,
274
+ "num_input_tokens_seen": 155648000,
275
+ "step": 19000
276
+ },
277
+ {
278
+ "epoch": 0.67,
279
+ "learning_rate": 1.6638152266894784e-05,
280
+ "loss": 1.9463,
281
+ "num_input_tokens_seen": 159744000,
282
+ "step": 19500
283
+ },
284
+ {
285
+ "epoch": 0.68,
286
+ "learning_rate": 1.5782720273738238e-05,
287
+ "loss": 1.9171,
288
+ "num_input_tokens_seen": 163840000,
289
+ "step": 20000
290
+ },
291
+ {
292
+ "epoch": 0.7,
293
+ "learning_rate": 1.4927288280581694e-05,
294
+ "loss": 1.9084,
295
+ "num_input_tokens_seen": 167936000,
296
+ "step": 20500
297
+ },
298
+ {
299
+ "epoch": 0.72,
300
+ "learning_rate": 1.407185628742515e-05,
301
+ "loss": 1.8769,
302
+ "num_input_tokens_seen": 172032000,
303
+ "step": 21000
304
+ },
305
+ {
306
+ "epoch": 0.74,
307
+ "learning_rate": 1.3216424294268606e-05,
308
+ "loss": 1.8426,
309
+ "num_input_tokens_seen": 176128000,
310
+ "step": 21500
311
+ },
312
+ {
313
+ "epoch": 0.75,
314
+ "learning_rate": 1.2360992301112061e-05,
315
+ "loss": 1.7861,
316
+ "num_input_tokens_seen": 180224000,
317
+ "step": 22000
318
+ },
319
+ {
320
+ "epoch": 0.77,
321
+ "learning_rate": 1.1505560307955517e-05,
322
+ "loss": 1.7839,
323
+ "num_input_tokens_seen": 184320000,
324
+ "step": 22500
325
+ },
326
+ {
327
+ "epoch": 0.79,
328
+ "learning_rate": 1.0650128314798973e-05,
329
+ "loss": 1.7512,
330
+ "num_input_tokens_seen": 188416000,
331
+ "step": 23000
332
+ },
333
+ {
334
+ "epoch": 0.8,
335
+ "learning_rate": 9.79469632164243e-06,
336
+ "loss": 1.7208,
337
+ "num_input_tokens_seen": 192512000,
338
+ "step": 23500
339
+ },
340
+ {
341
+ "epoch": 0.82,
342
+ "learning_rate": 8.939264328485886e-06,
343
+ "loss": 1.7213,
344
+ "num_input_tokens_seen": 196608000,
345
+ "step": 24000
346
+ },
347
+ {
348
+ "epoch": 0.84,
349
+ "learning_rate": 8.083832335329342e-06,
350
+ "loss": 1.715,
351
+ "num_input_tokens_seen": 200704000,
352
+ "step": 24500
353
+ },
354
+ {
355
+ "epoch": 0.86,
356
+ "learning_rate": 7.228400342172797e-06,
357
+ "loss": 1.696,
358
+ "num_input_tokens_seen": 204800000,
359
+ "step": 25000
360
+ },
361
+ {
362
+ "epoch": 0.87,
363
+ "learning_rate": 6.372968349016254e-06,
364
+ "loss": 1.7104,
365
+ "num_input_tokens_seen": 208896000,
366
+ "step": 25500
367
+ },
368
+ {
369
+ "epoch": 0.89,
370
+ "learning_rate": 5.51753635585971e-06,
371
+ "loss": 1.6742,
372
+ "num_input_tokens_seen": 212992000,
373
+ "step": 26000
374
+ },
375
+ {
376
+ "epoch": 0.91,
377
+ "learning_rate": 4.662104362703165e-06,
378
+ "loss": 1.6933,
379
+ "num_input_tokens_seen": 217088000,
380
+ "step": 26500
381
+ },
382
+ {
383
+ "epoch": 0.92,
384
+ "learning_rate": 3.8066723695466212e-06,
385
+ "loss": 1.6325,
386
+ "num_input_tokens_seen": 221184000,
387
+ "step": 27000
388
+ },
389
+ {
390
+ "epoch": 0.94,
391
+ "learning_rate": 2.951240376390077e-06,
392
+ "loss": 1.6691,
393
+ "num_input_tokens_seen": 225280000,
394
+ "step": 27500
395
+ },
396
+ {
397
+ "epoch": 0.96,
398
+ "learning_rate": 2.095808383233533e-06,
399
+ "loss": 1.6348,
400
+ "num_input_tokens_seen": 229376000,
401
+ "step": 28000
402
+ },
403
+ {
404
+ "epoch": 0.98,
405
+ "learning_rate": 1.240376390076989e-06,
406
+ "loss": 1.6348,
407
+ "num_input_tokens_seen": 233472000,
408
+ "step": 28500
409
+ },
410
+ {
411
+ "epoch": 0.99,
412
+ "learning_rate": 3.8494439692044483e-07,
413
+ "loss": 1.6152,
414
+ "num_input_tokens_seen": 237568000,
415
+ "step": 29000
416
+ }
417
+ ],
418
+ "logging_steps": 500,
419
+ "max_steps": 29225,
420
+ "num_input_tokens_seen": 237568000,
421
+ "num_train_epochs": 1,
422
+ "save_steps": 500,
423
+ "total_flos": 1.22121618014208e+17,
424
+ "train_batch_size": 16,
425
+ "trial_name": null,
426
+ "trial_params": null
427
+ }
test/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0fc9830c44dc4142c9d6027f64dc2a55e22a8378874f3a7a45f3235a63e8ba45
3
+ size 4792
test/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "None",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "[CLS]",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "[SEP]",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "[PAD]",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "4": {
36
+ "content": "[CODES]",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "5": {
44
+ "content": "[DATE]",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "6": {
52
+ "content": "[MASK]",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ }
59
+ },
60
+ "additional_special_tokens": [
61
+ "[DATE]",
62
+ "[CODES]"
63
+ ],
64
+ "clean_up_tokenization_spaces": true,
65
+ "cls_token": "[CLS]",
66
+ "mask_token": "[MASK]",
67
+ "max_len": 512,
68
+ "model_max_length": 512,
69
+ "pad_token": "[PAD]",
70
+ "return_special_tokens_mask": true,
71
+ "sep_token": "[SEP]",
72
+ "tokenizer_class": "FixedVocabTokenizer",
73
+ "unk_token": "None",
74
+ "verbose": true
75
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1677399b3cb2408860cab4c11ff58ab765ad04a9c9f857bac1e7614361a71365
3
+ size 4728
vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
vocab_eval_acc.txt ADDED
The diff for this file is too large to render. See raw diff
 
vocab_sorted_visits.txt ADDED
The diff for this file is too large to render. See raw diff