config.json CHANGED
@@ -8,11 +8,6 @@
8
  "MarianMTModel"
9
  ],
10
  "attention_dropout": 0.0,
11
- "bad_words_ids": [
12
- [
13
- 61109
14
- ]
15
- ],
16
  "bos_token_id": 0,
17
  "classif_dropout": 0.0,
18
  "classifier_dropout": 0.0,
@@ -43,19 +38,19 @@
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": 4,
52
  "num_hidden_layers": 6,
53
  "pad_token_id": 61109,
54
  "scale_embedding": true,
55
  "share_encoder_decoder_embeddings": true,
56
  "static_position_embeddings": true,
57
  "torch_dtype": "float32",
58
- "transformers_version": "4.44.2",
59
  "use_cache": true,
60
  "vocab_size": 61110
61
  }
 
8
  "MarianMTModel"
9
  ],
10
  "attention_dropout": 0.0,
 
 
 
 
 
11
  "bos_token_id": 0,
12
  "classif_dropout": 0.0,
13
  "classifier_dropout": 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": 61109,
49
  "scale_embedding": true,
50
  "share_encoder_decoder_embeddings": true,
51
  "static_position_embeddings": true,
52
  "torch_dtype": "float32",
53
+ "transformers_version": "4.47.0",
54
  "use_cache": true,
55
  "vocab_size": 61110
56
  }
generation_config.json CHANGED
@@ -12,5 +12,5 @@
12
  "num_beams": 4,
13
  "pad_token_id": 61109,
14
  "renormalize_logits": true,
15
- "transformers_version": "4.44.2"
16
  }
 
12
  "num_beams": 4,
13
  "pad_token_id": 61109,
14
  "renormalize_logits": true,
15
+ "transformers_version": "4.47.0"
16
  }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:936badc9c7f886bf746bedb457115bbf63890afefa6161ee81c6076c51c8ba84
3
+ size 301980760
optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:cdda2f75090a1fa6a2183cc5bbf5d356776970ca21e20aa147053a61b48f34d5
3
  size 603626106
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60e0dc487722a1366760c27d2ee970089f0fefc2794625d13aefa4169b2f0fc8
3
  size 603626106
rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f75bdecce60610b3bcd1d97c52de3f414a5305efa9e61c49a11d3e65936d2668
3
  size 14244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:269b6dd0942bbfcf02a52761caca71a971630a96b91a601c369a0236a15beb2f
3
  size 14244
scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6fc0f3a8a99bc754b73e55fd43baa7d572c046f5860346dbf5e6521b29bd9dff
3
  size 1064
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4daaed72356b285cd58234bc0f16bfd1bdd583b13d20d80a8722000fa49b28ef
3
  size 1064
special_tokens_map.json CHANGED
@@ -1,5 +1,23 @@
1
  {
2
- "eos_token": "</s>",
3
- "pad_token": "<pad>",
4
- "unk_token": "<unk>"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  }
 
1
  {
2
+ "eos_token": {
3
+ "content": "</s>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "pad_token": {
10
+ "content": "<pad>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "unk_token": {
17
+ "content": "<unk>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ }
23
  }
tokenizer_config.json CHANGED
@@ -27,6 +27,7 @@
27
  },
28
  "clean_up_tokenization_spaces": true,
29
  "eos_token": "</s>",
 
30
  "model_max_length": 512,
31
  "pad_token": "<pad>",
32
  "separate_vocabs": false,
 
27
  },
28
  "clean_up_tokenization_spaces": true,
29
  "eos_token": "</s>",
30
+ "extra_special_tokens": {},
31
  "model_max_length": 512,
32
  "pad_token": "<pad>",
33
  "separate_vocabs": false,
trainer_state.json CHANGED
@@ -3,182 +3,84 @@
3
  "best_model_checkpoint": null,
4
  "epoch": 8.0,
5
  "eval_steps": 500,
6
- "global_step": 7280,
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.5494505494505495,
13
- "grad_norm": 7.208542823791504,
14
- "learning_rate": 1.862637362637363e-05,
15
- "loss": 1.4229,
16
- "step": 500
17
- },
18
  {
19
  "epoch": 1.0,
20
- "eval_bleu": 45.1493,
21
- "eval_gen_len": 21.1512,
22
- "eval_loss": 0.9880260229110718,
23
- "eval_runtime": 159.8583,
24
- "eval_samples_per_second": 11.379,
25
- "eval_steps_per_second": 0.713,
26
- "step": 910
27
- },
28
- {
29
- "epoch": 1.098901098901099,
30
- "grad_norm": 5.381701469421387,
31
- "learning_rate": 1.7252747252747256e-05,
32
- "loss": 1.1197,
33
- "step": 1000
34
- },
35
- {
36
- "epoch": 1.6483516483516483,
37
- "grad_norm": 4.710964679718018,
38
- "learning_rate": 1.587912087912088e-05,
39
- "loss": 0.9205,
40
- "step": 1500
41
  },
42
  {
43
  "epoch": 2.0,
44
- "eval_bleu": 48.9448,
45
- "eval_gen_len": 21.4244,
46
- "eval_loss": 0.8794811367988586,
47
- "eval_runtime": 164.115,
48
- "eval_samples_per_second": 11.084,
49
- "eval_steps_per_second": 0.695,
50
- "step": 1820
51
- },
52
- {
53
- "epoch": 2.197802197802198,
54
- "grad_norm": 3.468052864074707,
55
- "learning_rate": 1.4505494505494506e-05,
56
- "loss": 0.8318,
57
- "step": 2000
58
- },
59
- {
60
- "epoch": 2.7472527472527473,
61
- "grad_norm": 3.9397318363189697,
62
- "learning_rate": 1.3131868131868134e-05,
63
- "loss": 0.7509,
64
- "step": 2500
65
  },
66
  {
67
  "epoch": 3.0,
68
- "eval_bleu": 50.7507,
69
- "eval_gen_len": 21.4739,
70
- "eval_loss": 0.83347487449646,
71
- "eval_runtime": 167.1805,
72
- "eval_samples_per_second": 10.88,
73
- "eval_steps_per_second": 0.682,
74
- "step": 2730
75
- },
76
- {
77
- "epoch": 3.2967032967032965,
78
- "grad_norm": 4.654043197631836,
79
- "learning_rate": 1.1758241758241759e-05,
80
- "loss": 0.7075,
81
- "step": 3000
82
- },
83
- {
84
- "epoch": 3.8461538461538463,
85
- "grad_norm": 3.244563341140747,
86
- "learning_rate": 1.0384615384615386e-05,
87
- "loss": 0.6522,
88
- "step": 3500
89
  },
90
  {
91
  "epoch": 4.0,
92
- "eval_bleu": 51.8451,
93
- "eval_gen_len": 21.4134,
94
- "eval_loss": 0.8046153783798218,
95
- "eval_runtime": 167.5891,
96
- "eval_samples_per_second": 10.854,
97
- "eval_steps_per_second": 0.68,
98
- "step": 3640
99
- },
100
- {
101
- "epoch": 4.395604395604396,
102
- "grad_norm": 4.202299118041992,
103
- "learning_rate": 9.010989010989011e-06,
104
- "loss": 0.6097,
105
- "step": 4000
106
- },
107
- {
108
- "epoch": 4.945054945054945,
109
- "grad_norm": 6.508543968200684,
110
- "learning_rate": 7.637362637362638e-06,
111
- "loss": 0.5877,
112
- "step": 4500
113
  },
114
  {
115
  "epoch": 5.0,
116
- "eval_bleu": 52.489,
117
- "eval_gen_len": 21.453,
118
- "eval_loss": 0.7915568947792053,
119
- "eval_runtime": 170.447,
120
- "eval_samples_per_second": 10.672,
121
- "eval_steps_per_second": 0.669,
122
- "step": 4550
123
- },
124
- {
125
- "epoch": 5.4945054945054945,
126
- "grad_norm": 4.693784236907959,
127
- "learning_rate": 6.2637362637362645e-06,
128
- "loss": 0.5454,
129
- "step": 5000
130
  },
131
  {
132
  "epoch": 6.0,
133
- "eval_bleu": 53.267,
134
- "eval_gen_len": 21.4063,
135
- "eval_loss": 0.7819094657897949,
136
- "eval_runtime": 166.0221,
137
- "eval_samples_per_second": 10.956,
138
- "eval_steps_per_second": 0.687,
139
- "step": 5460
140
- },
141
- {
142
- "epoch": 6.043956043956044,
143
- "grad_norm": 3.2858574390411377,
144
- "learning_rate": 4.890109890109891e-06,
145
- "loss": 0.5319,
146
- "step": 5500
147
- },
148
- {
149
- "epoch": 6.593406593406593,
150
- "grad_norm": 4.065381050109863,
151
- "learning_rate": 3.516483516483517e-06,
152
- "loss": 0.5039,
153
- "step": 6000
154
  },
155
  {
156
  "epoch": 7.0,
157
- "eval_bleu": 53.3718,
158
- "eval_gen_len": 21.5014,
159
- "eval_loss": 0.7780753970146179,
160
- "eval_runtime": 172.3653,
161
- "eval_samples_per_second": 10.553,
162
- "eval_steps_per_second": 0.661,
163
- "step": 6370
164
- },
165
- {
166
- "epoch": 7.142857142857143,
167
- "grad_norm": 3.438549280166626,
168
- "learning_rate": 2.1428571428571427e-06,
169
- "loss": 0.5018,
170
- "step": 6500
171
- },
172
- {
173
- "epoch": 7.6923076923076925,
174
- "grad_norm": 4.464223384857178,
175
- "learning_rate": 7.692307692307694e-07,
176
- "loss": 0.4847,
177
- "step": 7000
178
  }
179
  ],
180
  "logging_steps": 500,
181
- "max_steps": 7280,
182
  "num_input_tokens_seen": 0,
183
  "num_train_epochs": 8,
184
  "save_steps": 500,
@@ -194,7 +96,7 @@
194
  "attributes": {}
195
  }
196
  },
197
- "total_flos": 2444491603574784.0,
198
  "train_batch_size": 16,
199
  "trial_name": null,
200
  "trial_params": null
 
3
  "best_model_checkpoint": null,
4
  "epoch": 8.0,
5
  "eval_steps": 500,
6
+ "global_step": 224,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
10
  "log_history": [
 
 
 
 
 
 
 
11
  {
12
  "epoch": 1.0,
13
+ "eval_bleu": 28.3319,
14
+ "eval_gen_len": 32.6,
15
+ "eval_loss": 1.7957160472869873,
16
+ "eval_runtime": 11.0612,
17
+ "eval_samples_per_second": 4.972,
18
+ "eval_steps_per_second": 0.362,
19
+ "step": 28
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  },
21
  {
22
  "epoch": 2.0,
23
+ "eval_bleu": 36.2391,
24
+ "eval_gen_len": 35.6727,
25
+ "eval_loss": 1.7393983602523804,
26
+ "eval_runtime": 11.6367,
27
+ "eval_samples_per_second": 4.726,
28
+ "eval_steps_per_second": 0.344,
29
+ "step": 56
 
 
 
 
 
 
 
 
 
 
 
 
 
 
30
  },
31
  {
32
  "epoch": 3.0,
33
+ "eval_bleu": 38.5694,
34
+ "eval_gen_len": 36.0545,
35
+ "eval_loss": 1.6639301776885986,
36
+ "eval_runtime": 11.4577,
37
+ "eval_samples_per_second": 4.8,
38
+ "eval_steps_per_second": 0.349,
39
+ "step": 84
 
 
 
 
 
 
 
 
 
 
 
 
 
 
40
  },
41
  {
42
  "epoch": 4.0,
43
+ "eval_bleu": 39.6319,
44
+ "eval_gen_len": 35.3818,
45
+ "eval_loss": 1.6301159858703613,
46
+ "eval_runtime": 11.224,
47
+ "eval_samples_per_second": 4.9,
48
+ "eval_steps_per_second": 0.356,
49
+ "step": 112
 
 
 
 
 
 
 
 
 
 
 
 
 
 
50
  },
51
  {
52
  "epoch": 5.0,
53
+ "eval_bleu": 38.9538,
54
+ "eval_gen_len": 35.6182,
55
+ "eval_loss": 1.6278760433197021,
56
+ "eval_runtime": 11.1051,
57
+ "eval_samples_per_second": 4.953,
58
+ "eval_steps_per_second": 0.36,
59
+ "step": 140
 
 
 
 
 
 
 
60
  },
61
  {
62
  "epoch": 6.0,
63
+ "eval_bleu": 40.6136,
64
+ "eval_gen_len": 35.4909,
65
+ "eval_loss": 1.6254886388778687,
66
+ "eval_runtime": 11.4878,
67
+ "eval_samples_per_second": 4.788,
68
+ "eval_steps_per_second": 0.348,
69
+ "step": 168
 
 
 
 
 
 
 
 
 
 
 
 
 
 
70
  },
71
  {
72
  "epoch": 7.0,
73
+ "eval_bleu": 40.7205,
74
+ "eval_gen_len": 35.2545,
75
+ "eval_loss": 1.6089469194412231,
76
+ "eval_runtime": 10.925,
77
+ "eval_samples_per_second": 5.034,
78
+ "eval_steps_per_second": 0.366,
79
+ "step": 196
 
 
 
 
 
 
 
 
 
 
 
 
 
 
80
  }
81
  ],
82
  "logging_steps": 500,
83
+ "max_steps": 224,
84
  "num_input_tokens_seen": 0,
85
  "num_train_epochs": 8,
86
  "save_steps": 500,
 
96
  "attributes": {}
97
  }
98
  },
99
+ "total_flos": 114320117071872.0,
100
  "train_batch_size": 16,
101
  "trial_name": null,
102
  "trial_params": null
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:20449154fe68ba8250b81371c6832942b4bdd0b93d2c344989c41f78532d92ce
3
- size 5368
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1b823e3073621345cee91e7b5f83dc767f79444cac218b57c790356f7ac5e3e7
3
+ size 5496