Sag1012 commited on
Commit
ec9c774
1 Parent(s): 92f9338

Upload model MarianMT - ver6

Browse files
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