Amit234 commited on
Commit
e3b4db6
·
verified ·
1 Parent(s): 3464249

Upload model checkpoint

Browse files
config.json ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "distilbert-base-cased",
3
+ "activation": "gelu",
4
+ "architectures": [
5
+ "DistilBertForTokenClassification"
6
+ ],
7
+ "attention_dropout": 0.1,
8
+ "dim": 768,
9
+ "dropout": 0.1,
10
+ "hidden_dim": 3072,
11
+ "id2label": {
12
+ "0": "O",
13
+ "1": "B-Disease",
14
+ "2": "I-Disease"
15
+ },
16
+ "initializer_range": 0.02,
17
+ "label2id": {
18
+ "B-Disease": 1,
19
+ "I-Disease": 2,
20
+ "O": 0
21
+ },
22
+ "max_position_embeddings": 512,
23
+ "model_type": "distilbert",
24
+ "n_heads": 12,
25
+ "n_layers": 6,
26
+ "output_past": true,
27
+ "pad_token_id": 0,
28
+ "qa_dropout": 0.1,
29
+ "seq_classif_dropout": 0.2,
30
+ "sinusoidal_pos_embds": false,
31
+ "tie_weights_": true,
32
+ "torch_dtype": "float32",
33
+ "transformers_version": "4.45.2",
34
+ "vocab_size": 28996
35
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:27b44b058f6729fee2253fa980ba9d28ce2d63aab9696f594e74576635983f22
3
+ size 260785212
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0a77d39954e2568d367fa7f83347b900b154a8b76f82c53cc44484683cc852f
3
+ size 521631290
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:313f62a176aaf1d2f3bde04a4c4dcb7feea21da7035579370e2c4ed5a3e94f28
3
+ size 14244
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:122ce888b5437f2eb6f046ae78ee7415dd4a927f1414cd457078c697dc39617b
3
+ size 1064
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,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "mask_token": "[MASK]",
48
+ "model_max_length": 512,
49
+ "pad_token": "[PAD]",
50
+ "sep_token": "[SEP]",
51
+ "strip_accents": null,
52
+ "tokenize_chinese_chars": true,
53
+ "tokenizer_class": "DistilBertTokenizer",
54
+ "unk_token": "[UNK]"
55
+ }
trainer_state.json ADDED
@@ -0,0 +1,225 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 9.0,
5
+ "eval_steps": 500,
6
+ "global_step": 6120,
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.7352941176470589,
13
+ "grad_norm": 1.4399138689041138,
14
+ "learning_rate": 1.8366013071895427e-05,
15
+ "loss": 0.1305,
16
+ "step": 500
17
+ },
18
+ {
19
+ "epoch": 1.0,
20
+ "eval_accuracy": 0.9775613095979087,
21
+ "eval_f1": 0.740566037735849,
22
+ "eval_loss": 0.06862938404083252,
23
+ "eval_precision": 0.6908690869086909,
24
+ "eval_recall": 0.7979669631512071,
25
+ "eval_runtime": 2.5207,
26
+ "eval_samples_per_second": 366.567,
27
+ "eval_steps_per_second": 46.019,
28
+ "step": 680
29
+ },
30
+ {
31
+ "epoch": 1.4705882352941178,
32
+ "grad_norm": 3.059645652770996,
33
+ "learning_rate": 1.6732026143790852e-05,
34
+ "loss": 0.0475,
35
+ "step": 1000
36
+ },
37
+ {
38
+ "epoch": 2.0,
39
+ "eval_accuracy": 0.9815448773808042,
40
+ "eval_f1": 0.8041237113402061,
41
+ "eval_loss": 0.067368283867836,
42
+ "eval_precision": 0.7691415313225058,
43
+ "eval_recall": 0.8424396442185514,
44
+ "eval_runtime": 3.126,
45
+ "eval_samples_per_second": 295.582,
46
+ "eval_steps_per_second": 37.108,
47
+ "step": 1360
48
+ },
49
+ {
50
+ "epoch": 2.2058823529411766,
51
+ "grad_norm": 2.6966614723205566,
52
+ "learning_rate": 1.5098039215686276e-05,
53
+ "loss": 0.0354,
54
+ "step": 1500
55
+ },
56
+ {
57
+ "epoch": 2.9411764705882355,
58
+ "grad_norm": 3.640780210494995,
59
+ "learning_rate": 1.3464052287581701e-05,
60
+ "loss": 0.0222,
61
+ "step": 2000
62
+ },
63
+ {
64
+ "epoch": 3.0,
65
+ "eval_accuracy": 0.982322917963401,
66
+ "eval_f1": 0.8089053803339517,
67
+ "eval_loss": 0.07790510356426239,
68
+ "eval_precision": 0.7879518072289157,
69
+ "eval_recall": 0.8310038119440915,
70
+ "eval_runtime": 4.1074,
71
+ "eval_samples_per_second": 224.961,
72
+ "eval_steps_per_second": 28.242,
73
+ "step": 2040
74
+ },
75
+ {
76
+ "epoch": 3.6764705882352944,
77
+ "grad_norm": 0.008734635077416897,
78
+ "learning_rate": 1.1830065359477125e-05,
79
+ "loss": 0.012,
80
+ "step": 2500
81
+ },
82
+ {
83
+ "epoch": 4.0,
84
+ "eval_accuracy": 0.982789742312959,
85
+ "eval_f1": 0.8168498168498168,
86
+ "eval_loss": 0.0823572650551796,
87
+ "eval_precision": 0.7861339600470035,
88
+ "eval_recall": 0.8500635324015248,
89
+ "eval_runtime": 2.5823,
90
+ "eval_samples_per_second": 357.827,
91
+ "eval_steps_per_second": 44.922,
92
+ "step": 2720
93
+ },
94
+ {
95
+ "epoch": 4.411764705882353,
96
+ "grad_norm": 0.3800855576992035,
97
+ "learning_rate": 1.0196078431372549e-05,
98
+ "loss": 0.0081,
99
+ "step": 3000
100
+ },
101
+ {
102
+ "epoch": 5.0,
103
+ "eval_accuracy": 0.9815137557575003,
104
+ "eval_f1": 0.8112745098039216,
105
+ "eval_loss": 0.10165887326002121,
106
+ "eval_precision": 0.7834319526627219,
107
+ "eval_recall": 0.841168996188056,
108
+ "eval_runtime": 2.596,
109
+ "eval_samples_per_second": 355.931,
110
+ "eval_steps_per_second": 44.684,
111
+ "step": 3400
112
+ },
113
+ {
114
+ "epoch": 5.147058823529412,
115
+ "grad_norm": 1.0274019241333008,
116
+ "learning_rate": 8.562091503267974e-06,
117
+ "loss": 0.0054,
118
+ "step": 3500
119
+ },
120
+ {
121
+ "epoch": 5.882352941176471,
122
+ "grad_norm": 0.9368045926094055,
123
+ "learning_rate": 6.928104575163399e-06,
124
+ "loss": 0.0028,
125
+ "step": 4000
126
+ },
127
+ {
128
+ "epoch": 6.0,
129
+ "eval_accuracy": 0.981607120627412,
130
+ "eval_f1": 0.8175092478421702,
131
+ "eval_loss": 0.11204753071069717,
132
+ "eval_precision": 0.7940119760479042,
133
+ "eval_recall": 0.8424396442185514,
134
+ "eval_runtime": 2.6214,
135
+ "eval_samples_per_second": 352.478,
136
+ "eval_steps_per_second": 44.25,
137
+ "step": 4080
138
+ },
139
+ {
140
+ "epoch": 6.617647058823529,
141
+ "grad_norm": 0.0018368919845670462,
142
+ "learning_rate": 5.294117647058824e-06,
143
+ "loss": 0.0021,
144
+ "step": 4500
145
+ },
146
+ {
147
+ "epoch": 7.0,
148
+ "eval_accuracy": 0.9808290800448152,
149
+ "eval_f1": 0.8186215235792019,
150
+ "eval_loss": 0.12193118035793304,
151
+ "eval_precision": 0.7808535178777394,
152
+ "eval_recall": 0.8602287166454892,
153
+ "eval_runtime": 2.6902,
154
+ "eval_samples_per_second": 343.468,
155
+ "eval_steps_per_second": 43.119,
156
+ "step": 4760
157
+ },
158
+ {
159
+ "epoch": 7.352941176470588,
160
+ "grad_norm": 0.004133788403123617,
161
+ "learning_rate": 3.6601307189542484e-06,
162
+ "loss": 0.0011,
163
+ "step": 5000
164
+ },
165
+ {
166
+ "epoch": 8.0,
167
+ "eval_accuracy": 0.9814826341341965,
168
+ "eval_f1": 0.8152173913043478,
169
+ "eval_loss": 0.12062280625104904,
170
+ "eval_precision": 0.7767548906789413,
171
+ "eval_recall": 0.8576874205844981,
172
+ "eval_runtime": 2.6543,
173
+ "eval_samples_per_second": 348.116,
174
+ "eval_steps_per_second": 43.703,
175
+ "step": 5440
176
+ },
177
+ {
178
+ "epoch": 8.088235294117647,
179
+ "grad_norm": 0.00526324100792408,
180
+ "learning_rate": 2.0261437908496734e-06,
181
+ "loss": 0.0018,
182
+ "step": 5500
183
+ },
184
+ {
185
+ "epoch": 8.823529411764707,
186
+ "grad_norm": 0.01854279637336731,
187
+ "learning_rate": 3.921568627450981e-07,
188
+ "loss": 0.0005,
189
+ "step": 6000
190
+ },
191
+ {
192
+ "epoch": 9.0,
193
+ "eval_accuracy": 0.9821050666002739,
194
+ "eval_f1": 0.8220858895705522,
195
+ "eval_loss": 0.1177670955657959,
196
+ "eval_precision": 0.7947805456702254,
197
+ "eval_recall": 0.8513341804320204,
198
+ "eval_runtime": 2.7389,
199
+ "eval_samples_per_second": 337.365,
200
+ "eval_steps_per_second": 42.353,
201
+ "step": 6120
202
+ }
203
+ ],
204
+ "logging_steps": 500,
205
+ "max_steps": 6120,
206
+ "num_input_tokens_seen": 0,
207
+ "num_train_epochs": 9,
208
+ "save_steps": 500,
209
+ "stateful_callbacks": {
210
+ "TrainerControl": {
211
+ "args": {
212
+ "should_epoch_stop": false,
213
+ "should_evaluate": false,
214
+ "should_log": false,
215
+ "should_save": true,
216
+ "should_training_stop": true
217
+ },
218
+ "attributes": {}
219
+ }
220
+ },
221
+ "total_flos": 791914496183100.0,
222
+ "train_batch_size": 8,
223
+ "trial_name": null,
224
+ "trial_params": null
225
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:59a7d98ca53f20d7df8fbe1acaa972eb8f0a7f2ac71a8bf71806dd325ffd00fc
3
+ size 5240
vocab.txt ADDED
The diff for this file is too large to render. See raw diff