Upload model MarianMT - ver6
Browse files- MarianMT_ver6/config.json +56 -0
- MarianMT_ver6/generation_config.json +16 -0
- MarianMT_ver6/model.safetensors +3 -0
- MarianMT_ver6/optimizer.pt +3 -0
- MarianMT_ver6/rng_state.pth +3 -0
- MarianMT_ver6/scheduler.pt +3 -0
- MarianMT_ver6/trainer_state.json +243 -0
- MarianMT_ver6/training_args.bin +3 -0
MarianMT_ver6/config.json
ADDED
@@ -0,0 +1,56 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_name_or_path": "Helsinki-NLP/opus-mt-en-vi",
|
3 |
+
"activation_dropout": 0.0,
|
4 |
+
"activation_function": "swish",
|
5 |
+
"add_bias_logits": false,
|
6 |
+
"add_final_layer_norm": false,
|
7 |
+
"architectures": [
|
8 |
+
"MarianMTModel"
|
9 |
+
],
|
10 |
+
"attention_dropout": 0.0,
|
11 |
+
"bos_token_id": 0,
|
12 |
+
"classif_dropout": 0.0,
|
13 |
+
"classifier_dropout": 0.0,
|
14 |
+
"d_model": 512,
|
15 |
+
"decoder_attention_heads": 8,
|
16 |
+
"decoder_ffn_dim": 2048,
|
17 |
+
"decoder_layerdrop": 0.0,
|
18 |
+
"decoder_layers": 6,
|
19 |
+
"decoder_start_token_id": 53684,
|
20 |
+
"decoder_vocab_size": 53685,
|
21 |
+
"dropout": 0.1,
|
22 |
+
"encoder_attention_heads": 8,
|
23 |
+
"encoder_ffn_dim": 2048,
|
24 |
+
"encoder_layerdrop": 0.0,
|
25 |
+
"encoder_layers": 6,
|
26 |
+
"eos_token_id": 0,
|
27 |
+
"extra_pos_embeddings": 53685,
|
28 |
+
"forced_eos_token_id": 0,
|
29 |
+
"id2label": {
|
30 |
+
"0": "LABEL_0",
|
31 |
+
"1": "LABEL_1",
|
32 |
+
"2": "LABEL_2"
|
33 |
+
},
|
34 |
+
"init_std": 0.02,
|
35 |
+
"is_encoder_decoder": true,
|
36 |
+
"label2id": {
|
37 |
+
"LABEL_0": 0,
|
38 |
+
"LABEL_1": 1,
|
39 |
+
"LABEL_2": 2
|
40 |
+
},
|
41 |
+
"max_length": null,
|
42 |
+
"max_position_embeddings": 512,
|
43 |
+
"model_type": "marian",
|
44 |
+
"normalize_before": false,
|
45 |
+
"normalize_embedding": false,
|
46 |
+
"num_beams": null,
|
47 |
+
"num_hidden_layers": 6,
|
48 |
+
"pad_token_id": 53684,
|
49 |
+
"scale_embedding": true,
|
50 |
+
"share_encoder_decoder_embeddings": true,
|
51 |
+
"static_position_embeddings": true,
|
52 |
+
"torch_dtype": "float32",
|
53 |
+
"transformers_version": "4.46.3",
|
54 |
+
"use_cache": true,
|
55 |
+
"vocab_size": 53685
|
56 |
+
}
|
MarianMT_ver6/generation_config.json
ADDED
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"bad_words_ids": [
|
3 |
+
[
|
4 |
+
53684
|
5 |
+
]
|
6 |
+
],
|
7 |
+
"bos_token_id": 0,
|
8 |
+
"decoder_start_token_id": 53684,
|
9 |
+
"eos_token_id": 0,
|
10 |
+
"forced_eos_token_id": 0,
|
11 |
+
"max_length": 512,
|
12 |
+
"num_beams": 4,
|
13 |
+
"pad_token_id": 53684,
|
14 |
+
"renormalize_logits": true,
|
15 |
+
"transformers_version": "4.46.3"
|
16 |
+
}
|
MarianMT_ver6/model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:defe5ea169f670431edd57214700352a7bc352b58c528a441958fc2468aa004c
|
3 |
+
size 286744660
|
MarianMT_ver6/optimizer.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e3d91f4340a710a1437b8e4c4cf66695240a91092d1f749e06f6f01ffe2a92e
|
3 |
+
size 573213242
|
MarianMT_ver6/rng_state.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5d2dd2cf24d0834e80cbf1d671fee812082551158be23c39655c06e9fa2a5480
|
3 |
+
size 14244
|
MarianMT_ver6/scheduler.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:578becbd8b79fed48ed74d7a95af4bf311c3e23ac9a580fe52b5c20b0006b58b
|
3 |
+
size 1064
|
MarianMT_ver6/trainer_state.json
ADDED
@@ -0,0 +1,243 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"best_metric": 1.367624282836914,
|
3 |
+
"best_model_checkpoint": "./results/checkpoint-11635",
|
4 |
+
"epoch": 5.0,
|
5 |
+
"eval_steps": 500,
|
6 |
+
"global_step": 11635,
|
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.21486892995272883,
|
13 |
+
"grad_norm": 1.7943241596221924,
|
14 |
+
"learning_rate": 4.7851310700472715e-05,
|
15 |
+
"loss": 1.815,
|
16 |
+
"step": 500
|
17 |
+
},
|
18 |
+
{
|
19 |
+
"epoch": 0.42973785990545765,
|
20 |
+
"grad_norm": 1.5547664165496826,
|
21 |
+
"learning_rate": 4.570262140094543e-05,
|
22 |
+
"loss": 1.7137,
|
23 |
+
"step": 1000
|
24 |
+
},
|
25 |
+
{
|
26 |
+
"epoch": 0.6446067898581865,
|
27 |
+
"grad_norm": 1.829068899154663,
|
28 |
+
"learning_rate": 4.355393210141814e-05,
|
29 |
+
"loss": 1.6652,
|
30 |
+
"step": 1500
|
31 |
+
},
|
32 |
+
{
|
33 |
+
"epoch": 0.8594757198109153,
|
34 |
+
"grad_norm": 1.6881290674209595,
|
35 |
+
"learning_rate": 4.140524280189085e-05,
|
36 |
+
"loss": 1.6365,
|
37 |
+
"step": 2000
|
38 |
+
},
|
39 |
+
{
|
40 |
+
"epoch": 1.0,
|
41 |
+
"eval_loss": 1.4293888807296753,
|
42 |
+
"eval_runtime": 13.2899,
|
43 |
+
"eval_samples_per_second": 352.072,
|
44 |
+
"eval_steps_per_second": 2.784,
|
45 |
+
"step": 2327
|
46 |
+
},
|
47 |
+
{
|
48 |
+
"epoch": 1.0743446497636442,
|
49 |
+
"grad_norm": 1.6054766178131104,
|
50 |
+
"learning_rate": 3.9256553502363556e-05,
|
51 |
+
"loss": 1.5857,
|
52 |
+
"step": 2500
|
53 |
+
},
|
54 |
+
{
|
55 |
+
"epoch": 1.289213579716373,
|
56 |
+
"grad_norm": 1.550986647605896,
|
57 |
+
"learning_rate": 3.7107864202836275e-05,
|
58 |
+
"loss": 1.5097,
|
59 |
+
"step": 3000
|
60 |
+
},
|
61 |
+
{
|
62 |
+
"epoch": 1.504082509669102,
|
63 |
+
"grad_norm": 1.5548337697982788,
|
64 |
+
"learning_rate": 3.495917490330898e-05,
|
65 |
+
"loss": 1.495,
|
66 |
+
"step": 3500
|
67 |
+
},
|
68 |
+
{
|
69 |
+
"epoch": 1.7189514396218306,
|
70 |
+
"grad_norm": 1.5597646236419678,
|
71 |
+
"learning_rate": 3.281048560378169e-05,
|
72 |
+
"loss": 1.4887,
|
73 |
+
"step": 4000
|
74 |
+
},
|
75 |
+
{
|
76 |
+
"epoch": 1.9338203695745595,
|
77 |
+
"grad_norm": 1.4021880626678467,
|
78 |
+
"learning_rate": 3.066179630425441e-05,
|
79 |
+
"loss": 1.4862,
|
80 |
+
"step": 4500
|
81 |
+
},
|
82 |
+
{
|
83 |
+
"epoch": 2.0,
|
84 |
+
"eval_loss": 1.390964150428772,
|
85 |
+
"eval_runtime": 13.2972,
|
86 |
+
"eval_samples_per_second": 351.878,
|
87 |
+
"eval_steps_per_second": 2.783,
|
88 |
+
"step": 4654
|
89 |
+
},
|
90 |
+
{
|
91 |
+
"epoch": 2.1486892995272884,
|
92 |
+
"grad_norm": 1.399411916732788,
|
93 |
+
"learning_rate": 2.8513107004727117e-05,
|
94 |
+
"loss": 1.4227,
|
95 |
+
"step": 5000
|
96 |
+
},
|
97 |
+
{
|
98 |
+
"epoch": 2.363558229480017,
|
99 |
+
"grad_norm": 1.5368090867996216,
|
100 |
+
"learning_rate": 2.636441770519983e-05,
|
101 |
+
"loss": 1.3905,
|
102 |
+
"step": 5500
|
103 |
+
},
|
104 |
+
{
|
105 |
+
"epoch": 2.578427159432746,
|
106 |
+
"grad_norm": 1.406023383140564,
|
107 |
+
"learning_rate": 2.421572840567254e-05,
|
108 |
+
"loss": 1.3967,
|
109 |
+
"step": 6000
|
110 |
+
},
|
111 |
+
{
|
112 |
+
"epoch": 2.793296089385475,
|
113 |
+
"grad_norm": 1.437154769897461,
|
114 |
+
"learning_rate": 2.206703910614525e-05,
|
115 |
+
"loss": 1.4,
|
116 |
+
"step": 6500
|
117 |
+
},
|
118 |
+
{
|
119 |
+
"epoch": 3.0,
|
120 |
+
"eval_loss": 1.3757044076919556,
|
121 |
+
"eval_runtime": 13.2522,
|
122 |
+
"eval_samples_per_second": 353.074,
|
123 |
+
"eval_steps_per_second": 2.792,
|
124 |
+
"step": 6981
|
125 |
+
},
|
126 |
+
{
|
127 |
+
"epoch": 3.008165019338204,
|
128 |
+
"grad_norm": 1.37666916847229,
|
129 |
+
"learning_rate": 1.9918349806617966e-05,
|
130 |
+
"loss": 1.392,
|
131 |
+
"step": 7000
|
132 |
+
},
|
133 |
+
{
|
134 |
+
"epoch": 3.2230339492909326,
|
135 |
+
"grad_norm": 1.4489548206329346,
|
136 |
+
"learning_rate": 1.7769660507090678e-05,
|
137 |
+
"loss": 1.3282,
|
138 |
+
"step": 7500
|
139 |
+
},
|
140 |
+
{
|
141 |
+
"epoch": 3.4379028792436612,
|
142 |
+
"grad_norm": 1.3192425966262817,
|
143 |
+
"learning_rate": 1.5620971207563386e-05,
|
144 |
+
"loss": 1.333,
|
145 |
+
"step": 8000
|
146 |
+
},
|
147 |
+
{
|
148 |
+
"epoch": 3.6527718091963903,
|
149 |
+
"grad_norm": 1.4072356224060059,
|
150 |
+
"learning_rate": 1.34722819080361e-05,
|
151 |
+
"loss": 1.3261,
|
152 |
+
"step": 8500
|
153 |
+
},
|
154 |
+
{
|
155 |
+
"epoch": 3.867640739149119,
|
156 |
+
"grad_norm": 1.4121943712234497,
|
157 |
+
"learning_rate": 1.1323592608508809e-05,
|
158 |
+
"loss": 1.336,
|
159 |
+
"step": 9000
|
160 |
+
},
|
161 |
+
{
|
162 |
+
"epoch": 4.0,
|
163 |
+
"eval_loss": 1.368784785270691,
|
164 |
+
"eval_runtime": 13.295,
|
165 |
+
"eval_samples_per_second": 351.936,
|
166 |
+
"eval_steps_per_second": 2.783,
|
167 |
+
"step": 9308
|
168 |
+
},
|
169 |
+
{
|
170 |
+
"epoch": 4.082509669101848,
|
171 |
+
"grad_norm": 1.347611904144287,
|
172 |
+
"learning_rate": 9.174903308981523e-06,
|
173 |
+
"loss": 1.3144,
|
174 |
+
"step": 9500
|
175 |
+
},
|
176 |
+
{
|
177 |
+
"epoch": 4.297378599054577,
|
178 |
+
"grad_norm": 1.3636679649353027,
|
179 |
+
"learning_rate": 7.026214009454233e-06,
|
180 |
+
"loss": 1.2937,
|
181 |
+
"step": 10000
|
182 |
+
},
|
183 |
+
{
|
184 |
+
"epoch": 4.512247529007306,
|
185 |
+
"grad_norm": 1.3717917203903198,
|
186 |
+
"learning_rate": 4.877524709926945e-06,
|
187 |
+
"loss": 1.2912,
|
188 |
+
"step": 10500
|
189 |
+
},
|
190 |
+
{
|
191 |
+
"epoch": 4.727116458960034,
|
192 |
+
"grad_norm": 1.5116732120513916,
|
193 |
+
"learning_rate": 2.7288354103996563e-06,
|
194 |
+
"loss": 1.2907,
|
195 |
+
"step": 11000
|
196 |
+
},
|
197 |
+
{
|
198 |
+
"epoch": 4.941985388912763,
|
199 |
+
"grad_norm": 1.450828194618225,
|
200 |
+
"learning_rate": 5.801461108723678e-07,
|
201 |
+
"loss": 1.2871,
|
202 |
+
"step": 11500
|
203 |
+
},
|
204 |
+
{
|
205 |
+
"epoch": 5.0,
|
206 |
+
"eval_loss": 1.367624282836914,
|
207 |
+
"eval_runtime": 13.3043,
|
208 |
+
"eval_samples_per_second": 351.69,
|
209 |
+
"eval_steps_per_second": 2.781,
|
210 |
+
"step": 11635
|
211 |
+
}
|
212 |
+
],
|
213 |
+
"logging_steps": 500,
|
214 |
+
"max_steps": 11635,
|
215 |
+
"num_input_tokens_seen": 0,
|
216 |
+
"num_train_epochs": 5,
|
217 |
+
"save_steps": 500,
|
218 |
+
"stateful_callbacks": {
|
219 |
+
"EarlyStoppingCallback": {
|
220 |
+
"args": {
|
221 |
+
"early_stopping_patience": 2,
|
222 |
+
"early_stopping_threshold": 0.0
|
223 |
+
},
|
224 |
+
"attributes": {
|
225 |
+
"early_stopping_patience_counter": 0
|
226 |
+
}
|
227 |
+
},
|
228 |
+
"TrainerControl": {
|
229 |
+
"args": {
|
230 |
+
"should_epoch_stop": false,
|
231 |
+
"should_evaluate": false,
|
232 |
+
"should_log": false,
|
233 |
+
"should_save": true,
|
234 |
+
"should_training_stop": true
|
235 |
+
},
|
236 |
+
"attributes": {}
|
237 |
+
}
|
238 |
+
},
|
239 |
+
"total_flos": 2.523724794298368e+16,
|
240 |
+
"train_batch_size": 128,
|
241 |
+
"trial_name": null,
|
242 |
+
"trial_params": null
|
243 |
+
}
|
MarianMT_ver6/training_args.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:04105bfb3ae4249799e144d84e423e4d24aa40c9c8a36db245ca0fe5ed1cede4
|
3 |
+
size 5432
|