zhaokunjzk commited on
Commit
2ccaf63
1 Parent(s): bd527b4

Upload 10 files

Browse files
config.json ADDED
@@ -0,0 +1,61 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "./cache/base_model",
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
+ "bad_words_ids": [
12
+ [
13
+ 65000
14
+ ]
15
+ ],
16
+ "bos_token_id": 0,
17
+ "classif_dropout": 0.0,
18
+ "classifier_dropout": 0.0,
19
+ "d_model": 512,
20
+ "decoder_attention_heads": 8,
21
+ "decoder_ffn_dim": 2048,
22
+ "decoder_layerdrop": 0.0,
23
+ "decoder_layers": 6,
24
+ "decoder_start_token_id": 65000,
25
+ "decoder_vocab_size": 65001,
26
+ "dropout": 0.1,
27
+ "encoder_attention_heads": 8,
28
+ "encoder_ffn_dim": 2048,
29
+ "encoder_layerdrop": 0.0,
30
+ "encoder_layers": 6,
31
+ "eos_token_id": 0,
32
+ "extra_pos_embeddings": 65001,
33
+ "forced_eos_token_id": 0,
34
+ "id2label": {
35
+ "0": "LABEL_0",
36
+ "1": "LABEL_1",
37
+ "2": "LABEL_2"
38
+ },
39
+ "init_std": 0.02,
40
+ "is_encoder_decoder": true,
41
+ "label2id": {
42
+ "LABEL_0": 0,
43
+ "LABEL_1": 1,
44
+ "LABEL_2": 2
45
+ },
46
+ "max_length": 512,
47
+ "max_position_embeddings": 512,
48
+ "model_type": "marian",
49
+ "normalize_before": false,
50
+ "normalize_embedding": false,
51
+ "num_beams": 6,
52
+ "num_hidden_layers": 6,
53
+ "pad_token_id": 65000,
54
+ "scale_embedding": true,
55
+ "share_encoder_decoder_embeddings": true,
56
+ "static_position_embeddings": true,
57
+ "torch_dtype": "float32",
58
+ "transformers_version": "4.35.2",
59
+ "use_cache": true,
60
+ "vocab_size": 65001
61
+ }
generation_config.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bad_words_ids": [
3
+ [
4
+ 65000
5
+ ]
6
+ ],
7
+ "bos_token_id": 0,
8
+ "decoder_start_token_id": 65000,
9
+ "eos_token_id": 0,
10
+ "forced_eos_token_id": 0,
11
+ "max_length": 512,
12
+ "num_beams": 6,
13
+ "pad_token_id": 65000,
14
+ "renormalize_logits": true,
15
+ "transformers_version": "4.35.2"
16
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:884d3d5a1c0129a037080fe8ee57f1d853679676835ba0efb7311a9cf44d2473
3
+ size 309965092
source.spm ADDED
Binary file (805 kB). View file
 
special_tokens_map.json ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ {
2
+ "eos_token": "</s>",
3
+ "pad_token": "<pad>",
4
+ "unk_token": "<unk>"
5
+ }
target.spm ADDED
Binary file (807 kB). View file
 
tokenizer_config.json ADDED
@@ -0,0 +1,38 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": "<unk>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "65000": {
20
+ "content": "<pad>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ }
27
+ },
28
+ "clean_up_tokenization_spaces": true,
29
+ "eos_token": "</s>",
30
+ "model_max_length": 512,
31
+ "pad_token": "<pad>",
32
+ "separate_vocabs": false,
33
+ "source_lang": "zho",
34
+ "sp_model_kwargs": {},
35
+ "target_lang": "eng",
36
+ "tokenizer_class": "MarianTokenizer",
37
+ "unk_token": "<unk>"
38
+ }
trainer_state.json ADDED
@@ -0,0 +1,125 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 4.932735426008969,
5
+ "eval_steps": 500,
6
+ "global_step": 5500,
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.45,
13
+ "learning_rate": 1.820986547085202e-05,
14
+ "loss": 1.626,
15
+ "step": 500
16
+ },
17
+ {
18
+ "epoch": 0.9,
19
+ "learning_rate": 1.641614349775785e-05,
20
+ "loss": 1.4201,
21
+ "step": 1000
22
+ },
23
+ {
24
+ "epoch": 1.0,
25
+ "eval_bleu": 33.3044,
26
+ "eval_gen_len": 28.8563,
27
+ "eval_loss": 1.3592603206634521,
28
+ "eval_runtime": 172.5107,
29
+ "eval_samples_per_second": 9.762,
30
+ "eval_steps_per_second": 0.614,
31
+ "step": 1115
32
+ },
33
+ {
34
+ "epoch": 1.35,
35
+ "learning_rate": 1.462242152466368e-05,
36
+ "loss": 1.2694,
37
+ "step": 1500
38
+ },
39
+ {
40
+ "epoch": 1.79,
41
+ "learning_rate": 1.2828699551569507e-05,
42
+ "loss": 1.2085,
43
+ "step": 2000
44
+ },
45
+ {
46
+ "epoch": 2.0,
47
+ "eval_bleu": 34.5144,
48
+ "eval_gen_len": 29.0564,
49
+ "eval_loss": 1.3197557926177979,
50
+ "eval_runtime": 172.4795,
51
+ "eval_samples_per_second": 9.763,
52
+ "eval_steps_per_second": 0.615,
53
+ "step": 2230
54
+ },
55
+ {
56
+ "epoch": 2.24,
57
+ "learning_rate": 1.1034977578475336e-05,
58
+ "loss": 1.1369,
59
+ "step": 2500
60
+ },
61
+ {
62
+ "epoch": 2.69,
63
+ "learning_rate": 9.244843049327356e-06,
64
+ "loss": 1.0786,
65
+ "step": 3000
66
+ },
67
+ {
68
+ "epoch": 3.0,
69
+ "eval_bleu": 34.6844,
70
+ "eval_gen_len": 29.1496,
71
+ "eval_loss": 1.3080627918243408,
72
+ "eval_runtime": 175.6578,
73
+ "eval_samples_per_second": 9.587,
74
+ "eval_steps_per_second": 0.603,
75
+ "step": 3345
76
+ },
77
+ {
78
+ "epoch": 3.14,
79
+ "learning_rate": 7.451121076233184e-06,
80
+ "loss": 1.0529,
81
+ "step": 3500
82
+ },
83
+ {
84
+ "epoch": 3.59,
85
+ "learning_rate": 5.657399103139014e-06,
86
+ "loss": 0.9958,
87
+ "step": 4000
88
+ },
89
+ {
90
+ "epoch": 4.0,
91
+ "eval_bleu": 34.862,
92
+ "eval_gen_len": 29.0529,
93
+ "eval_loss": 1.309422254562378,
94
+ "eval_runtime": 174.4734,
95
+ "eval_samples_per_second": 9.652,
96
+ "eval_steps_per_second": 0.608,
97
+ "step": 4460
98
+ },
99
+ {
100
+ "epoch": 4.04,
101
+ "learning_rate": 3.863677130044843e-06,
102
+ "loss": 0.9933,
103
+ "step": 4500
104
+ },
105
+ {
106
+ "epoch": 4.48,
107
+ "learning_rate": 2.069955156950673e-06,
108
+ "loss": 0.9477,
109
+ "step": 5000
110
+ },
111
+ {
112
+ "epoch": 4.93,
113
+ "learning_rate": 2.762331838565023e-07,
114
+ "loss": 0.9504,
115
+ "step": 5500
116
+ }
117
+ ],
118
+ "logging_steps": 500,
119
+ "max_steps": 5575,
120
+ "num_train_epochs": 5,
121
+ "save_steps": 500,
122
+ "total_flos": 1101235524599808.0,
123
+ "trial_name": null,
124
+ "trial_params": null
125
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da90a111e5c48fccd8a546276563b6683ef1cde307a5b3c1ecb8438f97ef2c84
3
+ size 4271
vocab.json ADDED
The diff for this file is too large to render. See raw diff