Mizuiro-sakura commited on
Commit
a82bd5d
1 Parent(s): 6a8f367

Upload 11 files

Browse files
added_tokens.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "<ent2>": 32771,
3
+ "<ent>": 32770
4
+ }
config.json ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "studio-ousia/luke-japanese-large-lite",
3
+ "architectures": [
4
+ "LukeForSequenceClassification"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.1,
7
+ "bert_model_name": "models/luke-japanese-large/hf_xlm_roberta",
8
+ "bos_token_id": 0,
9
+ "classifier_dropout": null,
10
+ "cls_entity_prediction": false,
11
+ "entity_emb_size": 256,
12
+ "entity_vocab_size": 4,
13
+ "eos_token_id": 2,
14
+ "hidden_act": "gelu",
15
+ "hidden_dropout_prob": 0.1,
16
+ "hidden_size": 1024,
17
+ "id2label": {
18
+ "0": "LABEL_0",
19
+ "1": "LABEL_1",
20
+ "2": "LABEL_2",
21
+ "3": "LABEL_3",
22
+ "4": "LABEL_4",
23
+ "5": "LABEL_5",
24
+ "6": "LABEL_6",
25
+ "7": "LABEL_7"
26
+ },
27
+ "initializer_range": 0.02,
28
+ "intermediate_size": 4096,
29
+ "label2id": {
30
+ "LABEL_0": 0,
31
+ "LABEL_1": 1,
32
+ "LABEL_2": 2,
33
+ "LABEL_3": 3,
34
+ "LABEL_4": 4,
35
+ "LABEL_5": 5,
36
+ "LABEL_6": 6,
37
+ "LABEL_7": 7
38
+ },
39
+ "layer_norm_eps": 1e-05,
40
+ "max_position_embeddings": 514,
41
+ "model_type": "luke",
42
+ "num_attention_heads": 16,
43
+ "num_hidden_layers": 24,
44
+ "pad_token_id": 1,
45
+ "position_embedding_type": "absolute",
46
+ "problem_type": "multi_label_classification",
47
+ "torch_dtype": "float32",
48
+ "transformers_version": "4.26.1",
49
+ "type_vocab_size": 1,
50
+ "use_cache": true,
51
+ "use_entity_aware_attention": true,
52
+ "vocab_size": 32772
53
+ }
entity_vocab.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "[MASK2]": 3,
3
+ "[MASK]": 0,
4
+ "[PAD]": 2,
5
+ "[UNK]": 1
6
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1dbb5124461d44a1bbfffa868cb498fb77daee7011262657d1d921c00750d82d
3
+ size 1655458499
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f34f56ba239ada44c80231ab5bf95ed34710adb531fd1cc7f0b377e4d2ce7c4a
3
+ size 14503
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f4cedc8a71c35f12124c99e1f7741fff4ad7d9474abced2c058bba036dce148
3
+ size 623
sentencepiece.bpe.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8b73a5e054936c920cf5b7d1ec21ce9c281977078269963beb821c6c86fbff7
3
+ size 841889
special_tokens_map.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ {
4
+ "content": "<ent>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ {
11
+ "content": "<ent2>",
12
+ "lstrip": false,
13
+ "normalized": false,
14
+ "rstrip": false,
15
+ "single_word": false
16
+ }
17
+ ],
18
+ "bos_token": "<s>",
19
+ "cls_token": "<s>",
20
+ "eos_token": "</s>",
21
+ "mask_token": {
22
+ "content": "<mask>",
23
+ "lstrip": true,
24
+ "normalized": true,
25
+ "rstrip": false,
26
+ "single_word": false
27
+ },
28
+ "pad_token": "<pad>",
29
+ "sep_token": "</s>",
30
+ "unk_token": "<unk>"
31
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ {
4
+ "__type": "AddedToken",
5
+ "content": "<ent>",
6
+ "lstrip": false,
7
+ "normalized": true,
8
+ "rstrip": false,
9
+ "single_word": false
10
+ },
11
+ {
12
+ "__type": "AddedToken",
13
+ "content": "<ent2>",
14
+ "lstrip": false,
15
+ "normalized": true,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ {
20
+ "__type": "AddedToken",
21
+ "content": "<ent>",
22
+ "lstrip": false,
23
+ "normalized": true,
24
+ "rstrip": false,
25
+ "single_word": false
26
+ },
27
+ {
28
+ "__type": "AddedToken",
29
+ "content": "<ent2>",
30
+ "lstrip": false,
31
+ "normalized": true,
32
+ "rstrip": false,
33
+ "single_word": false
34
+ }
35
+ ],
36
+ "bos_token": "<s>",
37
+ "cls_token": "<s>",
38
+ "entity_mask2_token": "[MASK2]",
39
+ "entity_mask_token": "[MASK]",
40
+ "entity_pad_token": "[PAD]",
41
+ "entity_token_1": {
42
+ "__type": "AddedToken",
43
+ "content": "<ent>",
44
+ "lstrip": false,
45
+ "normalized": true,
46
+ "rstrip": false,
47
+ "single_word": false
48
+ },
49
+ "entity_token_2": {
50
+ "__type": "AddedToken",
51
+ "content": "<ent2>",
52
+ "lstrip": false,
53
+ "normalized": true,
54
+ "rstrip": false,
55
+ "single_word": false
56
+ },
57
+ "entity_unk_token": "[UNK]",
58
+ "eos_token": "</s>",
59
+ "mask_token": {
60
+ "__type": "AddedToken",
61
+ "content": "<mask>",
62
+ "lstrip": true,
63
+ "normalized": true,
64
+ "rstrip": false,
65
+ "single_word": false
66
+ },
67
+ "max_entity_length": 32,
68
+ "max_mention_length": 30,
69
+ "model_max_length": 512,
70
+ "name_or_path": "studio-ousia/luke-japanese-large-lite",
71
+ "pad_token": "<pad>",
72
+ "sep_token": "</s>",
73
+ "sp_model_kwargs": {},
74
+ "special_tokens_map_file": "models/luke-japanese-large/hf_xlm_roberta/special_tokens_map.json",
75
+ "task": null,
76
+ "tokenizer_class": "MLukeTokenizer",
77
+ "tokenizer_file": "models/luke-japanese-large/hf_luke_japanese_large_lite_epoch20/tokenizer.json",
78
+ "unk_token": "<unk>"
79
+ }
trainer_state.json ADDED
@@ -0,0 +1,196 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "global_step": 15000,
6
+ "is_hyper_param_search": false,
7
+ "is_local_process_zero": true,
8
+ "is_world_process_zero": true,
9
+ "log_history": [
10
+ {
11
+ "epoch": 0.1,
12
+ "learning_rate": 9.666666666666667e-06,
13
+ "loss": 0.2806,
14
+ "step": 500
15
+ },
16
+ {
17
+ "epoch": 0.2,
18
+ "learning_rate": 9.333333333333334e-06,
19
+ "loss": 0.2282,
20
+ "step": 1000
21
+ },
22
+ {
23
+ "epoch": 0.3,
24
+ "learning_rate": 9e-06,
25
+ "loss": 0.2146,
26
+ "step": 1500
27
+ },
28
+ {
29
+ "epoch": 0.4,
30
+ "learning_rate": 8.666666666666668e-06,
31
+ "loss": 0.2067,
32
+ "step": 2000
33
+ },
34
+ {
35
+ "epoch": 0.5,
36
+ "learning_rate": 8.333333333333334e-06,
37
+ "loss": 0.2059,
38
+ "step": 2500
39
+ },
40
+ {
41
+ "epoch": 0.6,
42
+ "learning_rate": 8.000000000000001e-06,
43
+ "loss": 0.2037,
44
+ "step": 3000
45
+ },
46
+ {
47
+ "epoch": 0.7,
48
+ "learning_rate": 7.666666666666667e-06,
49
+ "loss": 0.1921,
50
+ "step": 3500
51
+ },
52
+ {
53
+ "epoch": 0.8,
54
+ "learning_rate": 7.333333333333333e-06,
55
+ "loss": 0.1955,
56
+ "step": 4000
57
+ },
58
+ {
59
+ "epoch": 0.9,
60
+ "learning_rate": 7e-06,
61
+ "loss": 0.1917,
62
+ "step": 4500
63
+ },
64
+ {
65
+ "epoch": 1.0,
66
+ "learning_rate": 6.666666666666667e-06,
67
+ "loss": 0.191,
68
+ "step": 5000
69
+ },
70
+ {
71
+ "epoch": 1.1,
72
+ "learning_rate": 6.333333333333333e-06,
73
+ "loss": 0.1565,
74
+ "step": 5500
75
+ },
76
+ {
77
+ "epoch": 1.2,
78
+ "learning_rate": 6e-06,
79
+ "loss": 0.1535,
80
+ "step": 6000
81
+ },
82
+ {
83
+ "epoch": 1.3,
84
+ "learning_rate": 5.666666666666667e-06,
85
+ "loss": 0.1555,
86
+ "step": 6500
87
+ },
88
+ {
89
+ "epoch": 1.4,
90
+ "learning_rate": 5.333333333333334e-06,
91
+ "loss": 0.1503,
92
+ "step": 7000
93
+ },
94
+ {
95
+ "epoch": 1.5,
96
+ "learning_rate": 5e-06,
97
+ "loss": 0.1541,
98
+ "step": 7500
99
+ },
100
+ {
101
+ "epoch": 1.6,
102
+ "learning_rate": 4.666666666666667e-06,
103
+ "loss": 0.1545,
104
+ "step": 8000
105
+ },
106
+ {
107
+ "epoch": 1.7,
108
+ "learning_rate": 4.333333333333334e-06,
109
+ "loss": 0.1524,
110
+ "step": 8500
111
+ },
112
+ {
113
+ "epoch": 1.8,
114
+ "learning_rate": 4.000000000000001e-06,
115
+ "loss": 0.1456,
116
+ "step": 9000
117
+ },
118
+ {
119
+ "epoch": 1.9,
120
+ "learning_rate": 3.6666666666666666e-06,
121
+ "loss": 0.1544,
122
+ "step": 9500
123
+ },
124
+ {
125
+ "epoch": 2.0,
126
+ "learning_rate": 3.3333333333333333e-06,
127
+ "loss": 0.1518,
128
+ "step": 10000
129
+ },
130
+ {
131
+ "epoch": 2.1,
132
+ "learning_rate": 3e-06,
133
+ "loss": 0.1049,
134
+ "step": 10500
135
+ },
136
+ {
137
+ "epoch": 2.2,
138
+ "learning_rate": 2.666666666666667e-06,
139
+ "loss": 0.1133,
140
+ "step": 11000
141
+ },
142
+ {
143
+ "epoch": 2.3,
144
+ "learning_rate": 2.3333333333333336e-06,
145
+ "loss": 0.1072,
146
+ "step": 11500
147
+ },
148
+ {
149
+ "epoch": 2.4,
150
+ "learning_rate": 2.0000000000000003e-06,
151
+ "loss": 0.1141,
152
+ "step": 12000
153
+ },
154
+ {
155
+ "epoch": 2.5,
156
+ "learning_rate": 1.6666666666666667e-06,
157
+ "loss": 0.1126,
158
+ "step": 12500
159
+ },
160
+ {
161
+ "epoch": 2.6,
162
+ "learning_rate": 1.3333333333333334e-06,
163
+ "loss": 0.1164,
164
+ "step": 13000
165
+ },
166
+ {
167
+ "epoch": 2.7,
168
+ "learning_rate": 1.0000000000000002e-06,
169
+ "loss": 0.1145,
170
+ "step": 13500
171
+ },
172
+ {
173
+ "epoch": 2.8,
174
+ "learning_rate": 6.666666666666667e-07,
175
+ "loss": 0.1132,
176
+ "step": 14000
177
+ },
178
+ {
179
+ "epoch": 2.9,
180
+ "learning_rate": 3.3333333333333335e-07,
181
+ "loss": 0.112,
182
+ "step": 14500
183
+ },
184
+ {
185
+ "epoch": 3.0,
186
+ "learning_rate": 0.0,
187
+ "loss": 0.1113,
188
+ "step": 15000
189
+ }
190
+ ],
191
+ "max_steps": 15000,
192
+ "num_train_epochs": 3,
193
+ "total_flos": 3.494749741056e+16,
194
+ "trial_name": null,
195
+ "trial_params": null
196
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:86f13921ee7a89de1f23965c17a045a5485ededc7e8e23c32a468b30a357ca30
3
+ size 3503