Diezu commited on
Commit
8748681
·
verified ·
1 Parent(s): 609e9b8

Upload 11 files

Browse files
config.json ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "vinai/bartpho-syllable",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "gelu",
5
+ "architectures": [
6
+ "MBartForConditionalGeneration"
7
+ ],
8
+ "attention_dropout": 0.0,
9
+ "bos_token_id": 0,
10
+ "classifier_dropout": 0.0,
11
+ "d_model": 1024,
12
+ "decoder_attention_heads": 16,
13
+ "decoder_ffn_dim": 4096,
14
+ "decoder_layerdrop": 0.0,
15
+ "decoder_layers": 12,
16
+ "decoder_start_token_id": 2,
17
+ "dropout": 0.1,
18
+ "encoder_attention_heads": 16,
19
+ "encoder_ffn_dim": 4096,
20
+ "encoder_layerdrop": 0.0,
21
+ "encoder_layers": 12,
22
+ "eos_token_id": 2,
23
+ "forced_eos_token_id": 2,
24
+ "gradient_checkpointing": false,
25
+ "init_std": 0.02,
26
+ "is_encoder_decoder": true,
27
+ "max_position_embeddings": 1024,
28
+ "model_type": "mbart",
29
+ "num_hidden_layers": 12,
30
+ "pad_token_id": 1,
31
+ "scale_embedding": false,
32
+ "tokenizer_class": "BartphoTokenizer",
33
+ "torch_dtype": "float32",
34
+ "transformers_version": "4.44.2",
35
+ "use_cache": true,
36
+ "vocab_size": 40030
37
+ }
dict.txt ADDED
The diff for this file is too large to render. See raw diff
 
generation_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 0,
3
+ "decoder_start_token_id": 2,
4
+ "eos_token_id": 2,
5
+ "forced_eos_token_id": 2,
6
+ "pad_token_id": 1,
7
+ "transformers_version": "4.44.2"
8
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2764db94734dd5047142ce5c3be06ab4e7ea1ac88111ac79eb60096da46cb1dd
3
+ size 1583480280
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a8e4e19f123300dab2338c83cf138c1d32fc6a037b376cf975ee0f446f1c38e7
3
+ size 14244
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c217cde7544d321309999c61b6444065c558b71fbb0af7453fabff6d944be852
3
+ size 1064
sentencepiece.bpe.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cfc8146abe2a0488e9e2a0c56de7952f7c11ab059eca145a0a727afce0db2865
3
+ size 5069051
special_tokens_map.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": "<s>",
3
+ "cls_token": "<s>",
4
+ "eos_token": "</s>",
5
+ "mask_token": {
6
+ "content": "<mask>",
7
+ "lstrip": true,
8
+ "normalized": true,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "pad_token": "<pad>",
13
+ "sep_token": "</s>",
14
+ "unk_token": "<unk>"
15
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<s>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<pad>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "3": {
28
+ "content": "<unk>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "40029": {
36
+ "content": "<mask>",
37
+ "lstrip": true,
38
+ "normalized": true,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ }
43
+ },
44
+ "bos_token": "<s>",
45
+ "clean_up_tokenization_spaces": true,
46
+ "cls_token": "<s>",
47
+ "eos_token": "</s>",
48
+ "mask_token": "<mask>",
49
+ "model_max_length": 1000000000000000019884624838656,
50
+ "pad_token": "<pad>",
51
+ "sep_token": "</s>",
52
+ "sp_model_kwargs": {},
53
+ "tokenizer_class": "BartphoTokenizer",
54
+ "unk_token": "<unk>"
55
+ }
trainer_state.json ADDED
@@ -0,0 +1,227 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 0.11493609100580215,
3
+ "best_model_checkpoint": "/kaggle/working/checkpoint-11812",
4
+ "epoch": 8.999619047619047,
5
+ "eval_steps": 500,
6
+ "global_step": 11812,
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.7619047619047619,
13
+ "grad_norm": 1.9151228666305542,
14
+ "learning_rate": 9.240091463414634e-06,
15
+ "loss": 0.5245,
16
+ "step": 1000
17
+ },
18
+ {
19
+ "epoch": 0.9996190476190476,
20
+ "eval_avg_time_per_word": 0.2856894375340317,
21
+ "eval_bleu": 0.3241854207405991,
22
+ "eval_cer": 0.6666666865348816,
23
+ "eval_loss": 0.20001943409442902,
24
+ "eval_runtime": 241.8615,
25
+ "eval_sacrebleu": 32.41854207405989,
26
+ "eval_samples_per_second": 28.942,
27
+ "eval_steps_per_second": 0.604,
28
+ "eval_wer": 0.6888888888888889,
29
+ "step": 1312
30
+ },
31
+ {
32
+ "epoch": 1.5238095238095237,
33
+ "grad_norm": 2.0819993019104004,
34
+ "learning_rate": 8.478658536585368e-06,
35
+ "loss": 0.2796,
36
+ "step": 2000
37
+ },
38
+ {
39
+ "epoch": 2.0,
40
+ "eval_avg_time_per_word": 0.2867684533090989,
41
+ "eval_bleu": 0.33444843598328117,
42
+ "eval_cer": 0.6666666865348816,
43
+ "eval_loss": 0.15544836223125458,
44
+ "eval_runtime": 242.1808,
45
+ "eval_sacrebleu": 33.44484359832811,
46
+ "eval_samples_per_second": 28.904,
47
+ "eval_steps_per_second": 0.603,
48
+ "eval_wer": 0.6888888888888889,
49
+ "step": 2625
50
+ },
51
+ {
52
+ "epoch": 2.2857142857142856,
53
+ "grad_norm": 1.885176181793213,
54
+ "learning_rate": 7.716463414634146e-06,
55
+ "loss": 0.2149,
56
+ "step": 3000
57
+ },
58
+ {
59
+ "epoch": 2.9996190476190474,
60
+ "eval_avg_time_per_word": 0.2856631360126506,
61
+ "eval_bleu": 0.3407651434588302,
62
+ "eval_cer": 0.6666666865348816,
63
+ "eval_loss": 0.13238175213336945,
64
+ "eval_runtime": 243.4299,
65
+ "eval_sacrebleu": 34.076514345883034,
66
+ "eval_samples_per_second": 28.756,
67
+ "eval_steps_per_second": 0.6,
68
+ "eval_wer": 0.6888888888888889,
69
+ "step": 3937
70
+ },
71
+ {
72
+ "epoch": 3.0476190476190474,
73
+ "grad_norm": 1.629309892654419,
74
+ "learning_rate": 6.954268292682927e-06,
75
+ "loss": 0.1782,
76
+ "step": 4000
77
+ },
78
+ {
79
+ "epoch": 3.8095238095238093,
80
+ "grad_norm": 1.4296305179595947,
81
+ "learning_rate": 6.192073170731708e-06,
82
+ "loss": 0.1498,
83
+ "step": 5000
84
+ },
85
+ {
86
+ "epoch": 4.0,
87
+ "eval_avg_time_per_word": 0.28190079163009985,
88
+ "eval_bleu": 0.342484737426721,
89
+ "eval_cer": 0.6666666865348816,
90
+ "eval_loss": 0.12768897414207458,
91
+ "eval_runtime": 240.7041,
92
+ "eval_sacrebleu": 34.248473742672104,
93
+ "eval_samples_per_second": 29.081,
94
+ "eval_steps_per_second": 0.607,
95
+ "eval_wer": 0.6888888888888889,
96
+ "step": 5250
97
+ },
98
+ {
99
+ "epoch": 4.571428571428571,
100
+ "grad_norm": 1.3963040113449097,
101
+ "learning_rate": 5.429878048780488e-06,
102
+ "loss": 0.1318,
103
+ "step": 6000
104
+ },
105
+ {
106
+ "epoch": 4.999619047619047,
107
+ "eval_avg_time_per_word": 0.2839514807375623,
108
+ "eval_bleu": 0.3428596484328475,
109
+ "eval_cer": 0.6666666865348816,
110
+ "eval_loss": 0.12459039688110352,
111
+ "eval_runtime": 243.3533,
112
+ "eval_sacrebleu": 34.28596484328476,
113
+ "eval_samples_per_second": 28.765,
114
+ "eval_steps_per_second": 0.6,
115
+ "eval_wer": 0.6888888888888889,
116
+ "step": 6562
117
+ },
118
+ {
119
+ "epoch": 5.333333333333333,
120
+ "grad_norm": 1.245492935180664,
121
+ "learning_rate": 4.667682926829269e-06,
122
+ "loss": 0.1188,
123
+ "step": 7000
124
+ },
125
+ {
126
+ "epoch": 6.0,
127
+ "eval_avg_time_per_word": 0.2730801561768131,
128
+ "eval_bleu": 0.3446570827661078,
129
+ "eval_cer": 0.6666666865348816,
130
+ "eval_loss": 0.11544822156429291,
131
+ "eval_runtime": 241.1081,
132
+ "eval_sacrebleu": 34.46570827661079,
133
+ "eval_samples_per_second": 29.033,
134
+ "eval_steps_per_second": 0.606,
135
+ "eval_wer": 0.6888888888888889,
136
+ "step": 7875
137
+ },
138
+ {
139
+ "epoch": 6.095238095238095,
140
+ "grad_norm": 1.6760395765304565,
141
+ "learning_rate": 3.90625e-06,
142
+ "loss": 0.109,
143
+ "step": 8000
144
+ },
145
+ {
146
+ "epoch": 6.857142857142857,
147
+ "grad_norm": 1.5629112720489502,
148
+ "learning_rate": 3.1440548780487807e-06,
149
+ "loss": 0.0996,
150
+ "step": 9000
151
+ },
152
+ {
153
+ "epoch": 6.999619047619047,
154
+ "eval_avg_time_per_word": 0.2809329230335656,
155
+ "eval_bleu": 0.34531231334616685,
156
+ "eval_cer": 0.6666666865348816,
157
+ "eval_loss": 0.11558092385530472,
158
+ "eval_runtime": 240.4449,
159
+ "eval_sacrebleu": 34.5312313346167,
160
+ "eval_samples_per_second": 29.113,
161
+ "eval_steps_per_second": 0.607,
162
+ "eval_wer": 0.6888888888888889,
163
+ "step": 9187
164
+ },
165
+ {
166
+ "epoch": 7.619047619047619,
167
+ "grad_norm": 1.3507906198501587,
168
+ "learning_rate": 2.3818597560975614e-06,
169
+ "loss": 0.092,
170
+ "step": 10000
171
+ },
172
+ {
173
+ "epoch": 8.0,
174
+ "eval_avg_time_per_word": 0.27973620380425623,
175
+ "eval_bleu": 0.34567075563975486,
176
+ "eval_cer": 0.6666666865348816,
177
+ "eval_loss": 0.11671777069568634,
178
+ "eval_runtime": 240.0499,
179
+ "eval_sacrebleu": 34.56707556397547,
180
+ "eval_samples_per_second": 29.161,
181
+ "eval_steps_per_second": 0.608,
182
+ "eval_wer": 0.6888888888888889,
183
+ "step": 10500
184
+ },
185
+ {
186
+ "epoch": 8.380952380952381,
187
+ "grad_norm": 1.3803397417068481,
188
+ "learning_rate": 1.6196646341463414e-06,
189
+ "loss": 0.0873,
190
+ "step": 11000
191
+ },
192
+ {
193
+ "epoch": 8.999619047619047,
194
+ "eval_avg_time_per_word": 0.27570831504890414,
195
+ "eval_bleu": 0.3461272171150006,
196
+ "eval_cer": 0.6666666865348816,
197
+ "eval_loss": 0.11493609100580215,
198
+ "eval_runtime": 241.1666,
199
+ "eval_sacrebleu": 34.61272171150005,
200
+ "eval_samples_per_second": 29.026,
201
+ "eval_steps_per_second": 0.605,
202
+ "eval_wer": 0.6888888888888889,
203
+ "step": 11812
204
+ }
205
+ ],
206
+ "logging_steps": 1000,
207
+ "max_steps": 13120,
208
+ "num_input_tokens_seen": 0,
209
+ "num_train_epochs": 10,
210
+ "save_steps": 500,
211
+ "stateful_callbacks": {
212
+ "TrainerControl": {
213
+ "args": {
214
+ "should_epoch_stop": false,
215
+ "should_evaluate": false,
216
+ "should_log": false,
217
+ "should_save": true,
218
+ "should_training_stop": false
219
+ },
220
+ "attributes": {}
221
+ }
222
+ },
223
+ "total_flos": 1.0292481466328678e+17,
224
+ "train_batch_size": 12,
225
+ "trial_name": null,
226
+ "trial_params": null
227
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fb0aa8bf75dcc4ba5cd0daa415dc258ad2bfa2cd79a6a188f5a37cdfc835b1f0
3
+ size 5368