yesj1234 commited on
Commit
42fc833
1 Parent(s): 7292ab0

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ checkpoint-7000/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,78 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - ko
4
+ - en
5
+ base_model: ./reduced_model
6
+ tags:
7
+ - generated_from_trainer
8
+ metrics:
9
+ - bleu
10
+ model-index:
11
+ - name: tst-translation-output
12
+ results: []
13
+ ---
14
+
15
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
16
+ should probably proofread and complete it, then remove this comment. -->
17
+
18
+ # tst-translation-output
19
+
20
+ This model is a fine-tuned version of [./reduced_model](https://huggingface.co/./reduced_model) on an unknown dataset.
21
+ It achieves the following results on the evaluation set:
22
+ - Loss: 3.7663
23
+ - Bleu: 19.3382
24
+ - Gen Len: 17.8929
25
+
26
+ ## Model description
27
+
28
+ More information needed
29
+
30
+ ## Intended uses & limitations
31
+
32
+ More information needed
33
+
34
+ ## Training and evaluation data
35
+
36
+ More information needed
37
+
38
+ ## Training procedure
39
+
40
+ ### Training hyperparameters
41
+
42
+ The following hyperparameters were used during training:
43
+ - learning_rate: 5e-05
44
+ - train_batch_size: 2
45
+ - eval_batch_size: 2
46
+ - seed: 42
47
+ - distributed_type: multi-GPU
48
+ - num_devices: 4
49
+ - gradient_accumulation_steps: 16
50
+ - total_train_batch_size: 128
51
+ - total_eval_batch_size: 8
52
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
53
+ - lr_scheduler_type: linear
54
+ - num_epochs: 40
55
+
56
+ ### Training results
57
+
58
+ | Training Loss | Epoch | Step | Validation Loss | Bleu | Gen Len |
59
+ |:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|
60
+ | 2.6161 | 11.09 | 2000 | 3.1762 | 13.5109 | 19.1966 |
61
+ | 2.6161 | 13.86 | 2500 | 3.0375 | 16.2868 | 18.7985 |
62
+ | 1.4467 | 16.62 | 3000 | 3.1328 | 17.6991 | 18.1949 |
63
+ | 1.4467 | 19.39 | 3500 | 3.2690 | 17.9052 | 18.3117 |
64
+ | 0.6809 | 22.15 | 4000 | 3.3850 | 18.4075 | 18.2149 |
65
+ | 0.6809 | 24.91 | 4500 | 3.4465 | 19.0339 | 18.009 |
66
+ | 0.3422 | 27.68 | 5000 | 3.5680 | 18.7281 | 17.5902 |
67
+ | 0.3422 | 30.44 | 5500 | 3.6350 | 19.1534 | 18.2177 |
68
+ | 0.1941 | 33.2 | 6000 | 3.7153 | 19.2575 | 17.8784 |
69
+ | 0.1941 | 35.97 | 6500 | 3.7382 | 19.2475 | 17.9831 |
70
+ | 0.1271 | 38.73 | 7000 | 3.7573 | 19.3045 | 17.9889 |
71
+
72
+
73
+ ### Framework versions
74
+
75
+ - Transformers 4.33.1
76
+ - Pytorch 2.0.1+cu117
77
+ - Datasets 2.14.5
78
+ - Tokenizers 0.13.3
all_results.json ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 39.83,
3
+ "eval_bleu": 19.3382,
4
+ "eval_gen_len": 17.8929,
5
+ "eval_loss": 3.7662878036499023,
6
+ "eval_runtime": 242.4649,
7
+ "eval_samples": 2894,
8
+ "eval_samples_per_second": 11.936,
9
+ "eval_steps_per_second": 1.493,
10
+ "predict_bleu": 27.4786,
11
+ "predict_gen_len": 18.3914,
12
+ "predict_loss": 3.0754947662353516,
13
+ "predict_runtime": 250.3792,
14
+ "predict_samples": 2895,
15
+ "predict_samples_per_second": 11.562,
16
+ "predict_steps_per_second": 1.446,
17
+ "train_loss": 0.5723760350545247,
18
+ "train_runtime": 23953.291,
19
+ "train_samples": 23160,
20
+ "train_samples_per_second": 38.675,
21
+ "train_steps_per_second": 0.301
22
+ }
checkpoint-7000/config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "./reduced_model",
3
+ "_num_labels": 3,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "gelu",
6
+ "add_bias_logits": false,
7
+ "add_final_layer_norm": true,
8
+ "architectures": [
9
+ "MBartForConditionalGeneration"
10
+ ],
11
+ "attention_dropout": 0.0,
12
+ "bos_token_id": 0,
13
+ "classif_dropout": 0.0,
14
+ "classifier_dropout": 0.0,
15
+ "d_model": 1024,
16
+ "decoder_attention_heads": 16,
17
+ "decoder_ffn_dim": 4096,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 12,
20
+ "decoder_start_token_id": 250004,
21
+ "dropout": 0.1,
22
+ "encoder_attention_heads": 16,
23
+ "encoder_ffn_dim": 4096,
24
+ "encoder_layerdrop": 0.0,
25
+ "encoder_layers": 12,
26
+ "eos_token_id": 2,
27
+ "forced_eos_token_id": 2,
28
+ "id2label": {
29
+ "0": "LABEL_0",
30
+ "1": "LABEL_1",
31
+ "2": "LABEL_2"
32
+ },
33
+ "init_std": 0.02,
34
+ "is_encoder_decoder": true,
35
+ "label2id": {
36
+ "LABEL_0": 0,
37
+ "LABEL_1": 1,
38
+ "LABEL_2": 2
39
+ },
40
+ "max_length": 1024,
41
+ "max_position_embeddings": 1024,
42
+ "model_type": "mbart",
43
+ "normalize_before": true,
44
+ "normalize_embedding": true,
45
+ "num_beams": 5,
46
+ "num_hidden_layers": 12,
47
+ "output_past": true,
48
+ "pad_token_id": 1,
49
+ "scale_embedding": true,
50
+ "static_position_embeddings": false,
51
+ "task_specific_params": {
52
+ "translation_en_to_ro": {
53
+ "decoder_start_token_id": 250020
54
+ }
55
+ },
56
+ "torch_dtype": "float32",
57
+ "transformers_version": "4.33.1",
58
+ "use_cache": true,
59
+ "vocab_size": 250027
60
+ }
checkpoint-7000/generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 0,
3
+ "decoder_start_token_id": 250004,
4
+ "eos_token_id": 2,
5
+ "forced_eos_token_id": 2,
6
+ "max_length": 1024,
7
+ "num_beams": 5,
8
+ "pad_token_id": 1,
9
+ "transformers_version": "4.33.1"
10
+ }
checkpoint-7000/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4b7b12f224fccf075eddd36af2f68f94cfd170eb68ff3f155606c6e7f2dc381f
3
+ size 4887259502
checkpoint-7000/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af1cb08d3a258969abcc45a8a2ba94eba91eb84f667a3e573461da49bb8d0f48
3
+ size 2444583325
checkpoint-7000/rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd69a76d5722493474b57a0fcdd821ad2ba74581ec9cdb56f3c68161c1a194e5
3
+ size 17655
checkpoint-7000/rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:374bc3352dde9998d5c6c01595b5ab6498e57429c3470ab01b66a755961028d3
3
+ size 17655
checkpoint-7000/rng_state_2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:38b0d6d8826ef37ff748881940d730f411b3264d6f957469c5dcb85db1b2bd0d
3
+ size 17655
checkpoint-7000/rng_state_3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5627e7e3b36a7c5cdb68605fbe6ef70f2f75894af07efbc44b376fb89eb31cd9
3
+ size 17655
checkpoint-7000/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c379d5e956ea2710523259c187b33fcf2c78ef55823f6c3ad6668cacb6aa4e0c
3
+ size 627
checkpoint-7000/sentencepiece.bpe.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cfc8146abe2a0488e9e2a0c56de7952f7c11ab059eca145a0a727afce0db2865
3
+ size 5069051
checkpoint-7000/special_tokens_map.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "ar_AR",
4
+ "cs_CZ",
5
+ "de_DE",
6
+ "en_XX",
7
+ "es_XX",
8
+ "et_EE",
9
+ "fi_FI",
10
+ "fr_XX",
11
+ "gu_IN",
12
+ "hi_IN",
13
+ "it_IT",
14
+ "ja_XX",
15
+ "kk_KZ",
16
+ "ko_KR",
17
+ "lt_LT",
18
+ "lv_LV",
19
+ "my_MM",
20
+ "ne_NP",
21
+ "nl_XX",
22
+ "ro_RO",
23
+ "ru_RU",
24
+ "si_LK",
25
+ "tr_TR",
26
+ "vi_VN",
27
+ "zh_CN"
28
+ ],
29
+ "bos_token": "<s>",
30
+ "cls_token": "<s>",
31
+ "eos_token": "</s>",
32
+ "mask_token": {
33
+ "content": "<mask>",
34
+ "lstrip": true,
35
+ "normalized": true,
36
+ "rstrip": false,
37
+ "single_word": false
38
+ },
39
+ "pad_token": "<pad>",
40
+ "sep_token": "</s>",
41
+ "unk_token": "<unk>"
42
+ }
checkpoint-7000/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c645d47f1b17b579114ff283e7a1023a596dbd37a6c8b189c3c9870afdbb00c0
3
+ size 17103594
checkpoint-7000/tokenizer_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": null,
3
+ "bos_token": "<s>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "cls_token": "<s>",
6
+ "eos_token": "</s>",
7
+ "mask_token": {
8
+ "__type": "AddedToken",
9
+ "content": "<mask>",
10
+ "lstrip": true,
11
+ "normalized": true,
12
+ "rstrip": false,
13
+ "single_word": false
14
+ },
15
+ "model_max_length": 1024,
16
+ "pad_token": "<pad>",
17
+ "sep_token": "</s>",
18
+ "sp_model_kwargs": {},
19
+ "src_lang": null,
20
+ "tgt_lang": null,
21
+ "tokenizer_class": "MBartTokenizer",
22
+ "unk_token": "<unk>"
23
+ }
checkpoint-7000/trainer_state.json ADDED
@@ -0,0 +1,171 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 38.72884283246977,
5
+ "eval_steps": 500,
6
+ "global_step": 7000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 5.53,
13
+ "learning_rate": 1.9800000000000004e-05,
14
+ "loss": 8.0967,
15
+ "step": 1000
16
+ },
17
+ {
18
+ "epoch": 11.09,
19
+ "learning_rate": 3.619444444444445e-05,
20
+ "loss": 2.6161,
21
+ "step": 2000
22
+ },
23
+ {
24
+ "epoch": 11.09,
25
+ "eval_bleu": 13.5109,
26
+ "eval_gen_len": 19.1966,
27
+ "eval_loss": 3.176215171813965,
28
+ "eval_runtime": 298.8186,
29
+ "eval_samples_per_second": 9.685,
30
+ "eval_steps_per_second": 1.211,
31
+ "step": 2000
32
+ },
33
+ {
34
+ "epoch": 13.86,
35
+ "eval_bleu": 16.2868,
36
+ "eval_gen_len": 18.7985,
37
+ "eval_loss": 3.037487268447876,
38
+ "eval_runtime": 263.6174,
39
+ "eval_samples_per_second": 10.978,
40
+ "eval_steps_per_second": 1.373,
41
+ "step": 2500
42
+ },
43
+ {
44
+ "epoch": 16.62,
45
+ "learning_rate": 2.9270833333333338e-05,
46
+ "loss": 1.4467,
47
+ "step": 3000
48
+ },
49
+ {
50
+ "epoch": 16.62,
51
+ "eval_bleu": 17.6991,
52
+ "eval_gen_len": 18.1949,
53
+ "eval_loss": 3.132826328277588,
54
+ "eval_runtime": 253.8432,
55
+ "eval_samples_per_second": 11.401,
56
+ "eval_steps_per_second": 1.426,
57
+ "step": 3000
58
+ },
59
+ {
60
+ "epoch": 19.39,
61
+ "eval_bleu": 17.9052,
62
+ "eval_gen_len": 18.3117,
63
+ "eval_loss": 3.2690258026123047,
64
+ "eval_runtime": 250.4816,
65
+ "eval_samples_per_second": 11.554,
66
+ "eval_steps_per_second": 1.445,
67
+ "step": 3500
68
+ },
69
+ {
70
+ "epoch": 22.15,
71
+ "learning_rate": 2.232638888888889e-05,
72
+ "loss": 0.6809,
73
+ "step": 4000
74
+ },
75
+ {
76
+ "epoch": 22.15,
77
+ "eval_bleu": 18.4075,
78
+ "eval_gen_len": 18.2149,
79
+ "eval_loss": 3.3850443363189697,
80
+ "eval_runtime": 248.7531,
81
+ "eval_samples_per_second": 11.634,
82
+ "eval_steps_per_second": 1.455,
83
+ "step": 4000
84
+ },
85
+ {
86
+ "epoch": 24.91,
87
+ "eval_bleu": 19.0339,
88
+ "eval_gen_len": 18.009,
89
+ "eval_loss": 3.446467876434326,
90
+ "eval_runtime": 246.4301,
91
+ "eval_samples_per_second": 11.744,
92
+ "eval_steps_per_second": 1.469,
93
+ "step": 4500
94
+ },
95
+ {
96
+ "epoch": 27.68,
97
+ "learning_rate": 1.5381944444444445e-05,
98
+ "loss": 0.3422,
99
+ "step": 5000
100
+ },
101
+ {
102
+ "epoch": 27.68,
103
+ "eval_bleu": 18.7281,
104
+ "eval_gen_len": 17.5902,
105
+ "eval_loss": 3.5680091381073,
106
+ "eval_runtime": 242.436,
107
+ "eval_samples_per_second": 11.937,
108
+ "eval_steps_per_second": 1.493,
109
+ "step": 5000
110
+ },
111
+ {
112
+ "epoch": 30.44,
113
+ "eval_bleu": 19.1534,
114
+ "eval_gen_len": 18.2177,
115
+ "eval_loss": 3.634960651397705,
116
+ "eval_runtime": 247.9202,
117
+ "eval_samples_per_second": 11.673,
118
+ "eval_steps_per_second": 1.46,
119
+ "step": 5500
120
+ },
121
+ {
122
+ "epoch": 33.2,
123
+ "learning_rate": 8.444444444444446e-06,
124
+ "loss": 0.1941,
125
+ "step": 6000
126
+ },
127
+ {
128
+ "epoch": 33.2,
129
+ "eval_bleu": 19.2575,
130
+ "eval_gen_len": 17.8784,
131
+ "eval_loss": 3.7152717113494873,
132
+ "eval_runtime": 241.6987,
133
+ "eval_samples_per_second": 11.974,
134
+ "eval_steps_per_second": 1.498,
135
+ "step": 6000
136
+ },
137
+ {
138
+ "epoch": 35.97,
139
+ "eval_bleu": 19.2475,
140
+ "eval_gen_len": 17.9831,
141
+ "eval_loss": 3.738178253173828,
142
+ "eval_runtime": 245.2172,
143
+ "eval_samples_per_second": 11.802,
144
+ "eval_steps_per_second": 1.476,
145
+ "step": 6500
146
+ },
147
+ {
148
+ "epoch": 38.73,
149
+ "learning_rate": 1.5e-06,
150
+ "loss": 0.1271,
151
+ "step": 7000
152
+ },
153
+ {
154
+ "epoch": 38.73,
155
+ "eval_bleu": 19.3045,
156
+ "eval_gen_len": 17.9889,
157
+ "eval_loss": 3.757262706756592,
158
+ "eval_runtime": 245.3062,
159
+ "eval_samples_per_second": 11.798,
160
+ "eval_steps_per_second": 1.476,
161
+ "step": 7000
162
+ }
163
+ ],
164
+ "logging_steps": 1000,
165
+ "max_steps": 7200,
166
+ "num_train_epochs": 40,
167
+ "save_steps": 500,
168
+ "total_flos": 1.941748272070656e+18,
169
+ "trial_name": null,
170
+ "trial_params": null
171
+ }
checkpoint-7000/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b8cd9179b4ffcf2f9bbae7daf2139334286c941918b41091b10dbba57133407
3
+ size 4219
config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "./reduced_model",
3
+ "_num_labels": 3,
4
+ "activation_dropout": 0.0,
5
+ "activation_function": "gelu",
6
+ "add_bias_logits": false,
7
+ "add_final_layer_norm": true,
8
+ "architectures": [
9
+ "MBartForConditionalGeneration"
10
+ ],
11
+ "attention_dropout": 0.0,
12
+ "bos_token_id": 0,
13
+ "classif_dropout": 0.0,
14
+ "classifier_dropout": 0.0,
15
+ "d_model": 1024,
16
+ "decoder_attention_heads": 16,
17
+ "decoder_ffn_dim": 4096,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 12,
20
+ "decoder_start_token_id": 250004,
21
+ "dropout": 0.1,
22
+ "encoder_attention_heads": 16,
23
+ "encoder_ffn_dim": 4096,
24
+ "encoder_layerdrop": 0.0,
25
+ "encoder_layers": 12,
26
+ "eos_token_id": 2,
27
+ "forced_eos_token_id": 2,
28
+ "id2label": {
29
+ "0": "LABEL_0",
30
+ "1": "LABEL_1",
31
+ "2": "LABEL_2"
32
+ },
33
+ "init_std": 0.02,
34
+ "is_encoder_decoder": true,
35
+ "label2id": {
36
+ "LABEL_0": 0,
37
+ "LABEL_1": 1,
38
+ "LABEL_2": 2
39
+ },
40
+ "max_length": 1024,
41
+ "max_position_embeddings": 1024,
42
+ "model_type": "mbart",
43
+ "normalize_before": true,
44
+ "normalize_embedding": true,
45
+ "num_beams": 5,
46
+ "num_hidden_layers": 12,
47
+ "output_past": true,
48
+ "pad_token_id": 1,
49
+ "scale_embedding": true,
50
+ "static_position_embeddings": false,
51
+ "task_specific_params": {
52
+ "translation_en_to_ro": {
53
+ "decoder_start_token_id": 250020
54
+ }
55
+ },
56
+ "torch_dtype": "float32",
57
+ "transformers_version": "4.33.1",
58
+ "use_cache": true,
59
+ "vocab_size": 250027
60
+ }
eval_results.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 39.83,
3
+ "eval_bleu": 19.3382,
4
+ "eval_gen_len": 17.8929,
5
+ "eval_loss": 3.7662878036499023,
6
+ "eval_runtime": 242.4649,
7
+ "eval_samples": 2894,
8
+ "eval_samples_per_second": 11.936,
9
+ "eval_steps_per_second": 1.493
10
+ }
generated_predictions.txt ADDED
The diff for this file is too large to render. See raw diff
 
generation_config.json ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 0,
3
+ "decoder_start_token_id": 250004,
4
+ "eos_token_id": 2,
5
+ "forced_eos_token_id": 2,
6
+ "max_length": 1024,
7
+ "num_beams": 5,
8
+ "pad_token_id": 1,
9
+ "transformers_version": "4.33.1"
10
+ }
predict_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "predict_bleu": 27.4786,
3
+ "predict_gen_len": 18.3914,
4
+ "predict_loss": 3.0754947662353516,
5
+ "predict_runtime": 250.3792,
6
+ "predict_samples": 2895,
7
+ "predict_samples_per_second": 11.562,
8
+ "predict_steps_per_second": 1.446
9
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fad538795d1f4dc04183c7059cddda9939185d06f284173aa715ccf58a040d74
3
+ size 2444583325
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,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "ar_AR",
4
+ "cs_CZ",
5
+ "de_DE",
6
+ "en_XX",
7
+ "es_XX",
8
+ "et_EE",
9
+ "fi_FI",
10
+ "fr_XX",
11
+ "gu_IN",
12
+ "hi_IN",
13
+ "it_IT",
14
+ "ja_XX",
15
+ "kk_KZ",
16
+ "ko_KR",
17
+ "lt_LT",
18
+ "lv_LV",
19
+ "my_MM",
20
+ "ne_NP",
21
+ "nl_XX",
22
+ "ro_RO",
23
+ "ru_RU",
24
+ "si_LK",
25
+ "tr_TR",
26
+ "vi_VN",
27
+ "zh_CN"
28
+ ],
29
+ "bos_token": "<s>",
30
+ "cls_token": "<s>",
31
+ "eos_token": "</s>",
32
+ "mask_token": {
33
+ "content": "<mask>",
34
+ "lstrip": true,
35
+ "normalized": true,
36
+ "rstrip": false,
37
+ "single_word": false
38
+ },
39
+ "pad_token": "<pad>",
40
+ "sep_token": "</s>",
41
+ "unk_token": "<unk>"
42
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c645d47f1b17b579114ff283e7a1023a596dbd37a6c8b189c3c9870afdbb00c0
3
+ size 17103594
tokenizer_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": null,
3
+ "bos_token": "<s>",
4
+ "clean_up_tokenization_spaces": true,
5
+ "cls_token": "<s>",
6
+ "eos_token": "</s>",
7
+ "mask_token": {
8
+ "__type": "AddedToken",
9
+ "content": "<mask>",
10
+ "lstrip": true,
11
+ "normalized": true,
12
+ "rstrip": false,
13
+ "single_word": false
14
+ },
15
+ "model_max_length": 1024,
16
+ "pad_token": "<pad>",
17
+ "sep_token": "</s>",
18
+ "sp_model_kwargs": {},
19
+ "src_lang": null,
20
+ "tgt_lang": null,
21
+ "tokenizer_class": "MBartTokenizer",
22
+ "unk_token": "<unk>"
23
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 39.83,
3
+ "train_loss": 0.5723760350545247,
4
+ "train_runtime": 23953.291,
5
+ "train_samples": 23160,
6
+ "train_samples_per_second": 38.675,
7
+ "train_steps_per_second": 0.301
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,180 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 39.83419689119171,
5
+ "eval_steps": 500,
6
+ "global_step": 7200,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 5.53,
13
+ "learning_rate": 1.9800000000000004e-05,
14
+ "loss": 8.0967,
15
+ "step": 1000
16
+ },
17
+ {
18
+ "epoch": 11.09,
19
+ "learning_rate": 3.619444444444445e-05,
20
+ "loss": 2.6161,
21
+ "step": 2000
22
+ },
23
+ {
24
+ "epoch": 11.09,
25
+ "eval_bleu": 13.5109,
26
+ "eval_gen_len": 19.1966,
27
+ "eval_loss": 3.176215171813965,
28
+ "eval_runtime": 298.8186,
29
+ "eval_samples_per_second": 9.685,
30
+ "eval_steps_per_second": 1.211,
31
+ "step": 2000
32
+ },
33
+ {
34
+ "epoch": 13.86,
35
+ "eval_bleu": 16.2868,
36
+ "eval_gen_len": 18.7985,
37
+ "eval_loss": 3.037487268447876,
38
+ "eval_runtime": 263.6174,
39
+ "eval_samples_per_second": 10.978,
40
+ "eval_steps_per_second": 1.373,
41
+ "step": 2500
42
+ },
43
+ {
44
+ "epoch": 16.62,
45
+ "learning_rate": 2.9270833333333338e-05,
46
+ "loss": 1.4467,
47
+ "step": 3000
48
+ },
49
+ {
50
+ "epoch": 16.62,
51
+ "eval_bleu": 17.6991,
52
+ "eval_gen_len": 18.1949,
53
+ "eval_loss": 3.132826328277588,
54
+ "eval_runtime": 253.8432,
55
+ "eval_samples_per_second": 11.401,
56
+ "eval_steps_per_second": 1.426,
57
+ "step": 3000
58
+ },
59
+ {
60
+ "epoch": 19.39,
61
+ "eval_bleu": 17.9052,
62
+ "eval_gen_len": 18.3117,
63
+ "eval_loss": 3.2690258026123047,
64
+ "eval_runtime": 250.4816,
65
+ "eval_samples_per_second": 11.554,
66
+ "eval_steps_per_second": 1.445,
67
+ "step": 3500
68
+ },
69
+ {
70
+ "epoch": 22.15,
71
+ "learning_rate": 2.232638888888889e-05,
72
+ "loss": 0.6809,
73
+ "step": 4000
74
+ },
75
+ {
76
+ "epoch": 22.15,
77
+ "eval_bleu": 18.4075,
78
+ "eval_gen_len": 18.2149,
79
+ "eval_loss": 3.3850443363189697,
80
+ "eval_runtime": 248.7531,
81
+ "eval_samples_per_second": 11.634,
82
+ "eval_steps_per_second": 1.455,
83
+ "step": 4000
84
+ },
85
+ {
86
+ "epoch": 24.91,
87
+ "eval_bleu": 19.0339,
88
+ "eval_gen_len": 18.009,
89
+ "eval_loss": 3.446467876434326,
90
+ "eval_runtime": 246.4301,
91
+ "eval_samples_per_second": 11.744,
92
+ "eval_steps_per_second": 1.469,
93
+ "step": 4500
94
+ },
95
+ {
96
+ "epoch": 27.68,
97
+ "learning_rate": 1.5381944444444445e-05,
98
+ "loss": 0.3422,
99
+ "step": 5000
100
+ },
101
+ {
102
+ "epoch": 27.68,
103
+ "eval_bleu": 18.7281,
104
+ "eval_gen_len": 17.5902,
105
+ "eval_loss": 3.5680091381073,
106
+ "eval_runtime": 242.436,
107
+ "eval_samples_per_second": 11.937,
108
+ "eval_steps_per_second": 1.493,
109
+ "step": 5000
110
+ },
111
+ {
112
+ "epoch": 30.44,
113
+ "eval_bleu": 19.1534,
114
+ "eval_gen_len": 18.2177,
115
+ "eval_loss": 3.634960651397705,
116
+ "eval_runtime": 247.9202,
117
+ "eval_samples_per_second": 11.673,
118
+ "eval_steps_per_second": 1.46,
119
+ "step": 5500
120
+ },
121
+ {
122
+ "epoch": 33.2,
123
+ "learning_rate": 8.444444444444446e-06,
124
+ "loss": 0.1941,
125
+ "step": 6000
126
+ },
127
+ {
128
+ "epoch": 33.2,
129
+ "eval_bleu": 19.2575,
130
+ "eval_gen_len": 17.8784,
131
+ "eval_loss": 3.7152717113494873,
132
+ "eval_runtime": 241.6987,
133
+ "eval_samples_per_second": 11.974,
134
+ "eval_steps_per_second": 1.498,
135
+ "step": 6000
136
+ },
137
+ {
138
+ "epoch": 35.97,
139
+ "eval_bleu": 19.2475,
140
+ "eval_gen_len": 17.9831,
141
+ "eval_loss": 3.738178253173828,
142
+ "eval_runtime": 245.2172,
143
+ "eval_samples_per_second": 11.802,
144
+ "eval_steps_per_second": 1.476,
145
+ "step": 6500
146
+ },
147
+ {
148
+ "epoch": 38.73,
149
+ "learning_rate": 1.5e-06,
150
+ "loss": 0.1271,
151
+ "step": 7000
152
+ },
153
+ {
154
+ "epoch": 38.73,
155
+ "eval_bleu": 19.3045,
156
+ "eval_gen_len": 17.9889,
157
+ "eval_loss": 3.757262706756592,
158
+ "eval_runtime": 245.3062,
159
+ "eval_samples_per_second": 11.798,
160
+ "eval_steps_per_second": 1.476,
161
+ "step": 7000
162
+ },
163
+ {
164
+ "epoch": 39.83,
165
+ "step": 7200,
166
+ "total_flos": 1.9972267941298176e+18,
167
+ "train_loss": 0.5723760350545247,
168
+ "train_runtime": 23953.291,
169
+ "train_samples_per_second": 38.675,
170
+ "train_steps_per_second": 0.301
171
+ }
172
+ ],
173
+ "logging_steps": 1000,
174
+ "max_steps": 7200,
175
+ "num_train_epochs": 40,
176
+ "save_steps": 500,
177
+ "total_flos": 1.9972267941298176e+18,
178
+ "trial_name": null,
179
+ "trial_params": null
180
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b8cd9179b4ffcf2f9bbae7daf2139334286c941918b41091b10dbba57133407
3
+ size 4219