Plasmoxy commited on
Commit
009464b
·
verified ·
1 Parent(s): 7fd0ceb

End of training

Browse files
README.md ADDED
@@ -0,0 +1,86 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: google/mt5-small
5
+ tags:
6
+ - generated_from_trainer
7
+ model-index:
8
+ - name: mt5-small-gigatrue-slovak-D-34567
9
+ results: []
10
+ ---
11
+
12
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
13
+ should probably proofread and complete it, then remove this comment. -->
14
+
15
+ # mt5-small-gigatrue-slovak-D-34567
16
+
17
+ This model is a fine-tuned version of [google/mt5-small](https://huggingface.co/google/mt5-small) on an unknown dataset.
18
+ It achieves the following results on the evaluation set:
19
+ - Loss: 3.1159
20
+
21
+ ## Model description
22
+
23
+ More information needed
24
+
25
+ ## Intended uses & limitations
26
+
27
+ More information needed
28
+
29
+ ## Training and evaluation data
30
+
31
+ More information needed
32
+
33
+ ## Training procedure
34
+
35
+ ### Training hyperparameters
36
+
37
+ The following hyperparameters were used during training:
38
+ - learning_rate: 0.0003
39
+ - train_batch_size: 128
40
+ - eval_batch_size: 128
41
+ - seed: 42
42
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
43
+ - lr_scheduler_type: linear
44
+ - num_epochs: 3
45
+
46
+ ### Training results
47
+
48
+ | Training Loss | Epoch | Step | Validation Loss |
49
+ |:-------------:|:------:|:-----:|:---------------:|
50
+ | 4.8045 | 0.1015 | 3000 | 3.3921 |
51
+ | 4.1932 | 0.2030 | 6000 | 3.2547 |
52
+ | 4.1098 | 0.3044 | 9000 | 3.2044 |
53
+ | 4.0752 | 0.4059 | 12000 | 3.1740 |
54
+ | 4.0515 | 0.5074 | 15000 | 3.1574 |
55
+ | 4.0446 | 0.6089 | 18000 | 3.1490 |
56
+ | 4.036 | 0.7104 | 21000 | 3.1390 |
57
+ | 4.0298 | 0.8119 | 24000 | 3.1365 |
58
+ | 4.0273 | 0.9133 | 27000 | 3.1308 |
59
+ | 4.023 | 1.0148 | 30000 | 3.1286 |
60
+ | 4.0229 | 1.1163 | 33000 | 3.1245 |
61
+ | 4.0158 | 1.2178 | 36000 | 3.1234 |
62
+ | 4.0163 | 1.3193 | 39000 | 3.1217 |
63
+ | 4.0156 | 1.4207 | 42000 | 3.1208 |
64
+ | 4.0147 | 1.5222 | 45000 | 3.1194 |
65
+ | 4.0114 | 1.6237 | 48000 | 3.1172 |
66
+ | 4.0113 | 1.7252 | 51000 | 3.1181 |
67
+ | 4.0097 | 1.8267 | 54000 | 3.1170 |
68
+ | 4.0105 | 1.9282 | 57000 | 3.1161 |
69
+ | 4.0115 | 2.0296 | 60000 | 3.1166 |
70
+ | 4.0117 | 2.1311 | 63000 | 3.1167 |
71
+ | 4.0082 | 2.2326 | 66000 | 3.1160 |
72
+ | 4.0089 | 2.3341 | 69000 | 3.1157 |
73
+ | 4.0087 | 2.4356 | 72000 | 3.1156 |
74
+ | 4.0104 | 2.5370 | 75000 | 3.1158 |
75
+ | 4.0088 | 2.6385 | 78000 | 3.1159 |
76
+ | 4.007 | 2.7400 | 81000 | 3.1159 |
77
+ | 4.0116 | 2.8415 | 84000 | 3.1160 |
78
+ | 4.0099 | 2.9430 | 87000 | 3.1159 |
79
+
80
+
81
+ ### Framework versions
82
+
83
+ - Transformers 4.45.2
84
+ - Pytorch 2.5.1
85
+ - Datasets 3.2.0
86
+ - Tokenizers 0.20.3
config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "google/mt5-small",
3
+ "architectures": [
4
+ "MT5ForConditionalGeneration"
5
+ ],
6
+ "classifier_dropout": 0.0,
7
+ "d_ff": 1024,
8
+ "d_kv": 64,
9
+ "d_model": 512,
10
+ "decoder_start_token_id": 0,
11
+ "dense_act_fn": "gelu_new",
12
+ "dropout_rate": 0.1,
13
+ "eos_token_id": 1,
14
+ "feed_forward_proj": "gated-gelu",
15
+ "initializer_factor": 1.0,
16
+ "is_encoder_decoder": true,
17
+ "is_gated_act": true,
18
+ "layer_norm_epsilon": 1e-06,
19
+ "model_type": "mt5",
20
+ "num_decoder_layers": 3,
21
+ "num_heads": 6,
22
+ "num_layers": 8,
23
+ "pad_token_id": 0,
24
+ "relative_attention_max_distance": 128,
25
+ "relative_attention_num_buckets": 32,
26
+ "tie_word_embeddings": false,
27
+ "tokenizer_class": "T5Tokenizer",
28
+ "torch_dtype": "bfloat16",
29
+ "transformers_version": "4.45.2",
30
+ "use_cache": true,
31
+ "vocab_size": 250112
32
+ }
coreconfig.json ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "out_name": "mt5-small-gigatrue-slovak-D-34567",
3
+ "remove_layers": true,
4
+ "model_archetype": "mt5",
5
+ "model_name": "google/mt5-small",
6
+ "tokenizer_name": "google/mt5-small",
7
+ "model_torch_dtype": "bfloat16",
8
+ "dataset_lang": "sk",
9
+ "dataset_name": "Plasmoxy/gigatrue-slovak",
10
+ "dataset_tokenized_cache_name": "gigatrue-slovak_tokenized_mt5",
11
+ "use_half_val_dataset": true,
12
+ "max_input_length": 110,
13
+ "max_target_length": 35,
14
+ "batch_size": 128,
15
+ "learning_rate": 0.0003,
16
+ "num_train_epochs": 3,
17
+ "pkg_versions": {
18
+ "optimum": "1.23.3",
19
+ "transformers": "4.45.2",
20
+ "openvino": "2024.6.0",
21
+ "nncf": "2.14.1",
22
+ "torch": "2.5.1",
23
+ "datasets": "3.2.0",
24
+ "peft": "0.13.3.dev0"
25
+ }
26
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "decoder_start_token_id": 0,
4
+ "eos_token_id": 1,
5
+ "pad_token_id": 0,
6
+ "transformers_version": "4.45.2"
7
+ }
logs/events.out.tfevents.1736379694.gna4000.758019.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e2c9a5e258f3ccd2f4dee56eaf177f3abe700cb54b9d86b796045c070ac9c79a
3
+ size 20067
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbf706331eb7de4ddaed80a0d2c411a9faf949a7e93e060494d9d851bf9dcedc
3
+ size 568895048
nvidia_smi_early.log ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Thu Jan 9 00:42:04 2025
2
+ +-----------------------------------------------------------------------------------------+
3
+ | NVIDIA-SMI 550.127.08 Driver Version: 550.127.08 CUDA Version: 12.4 |
4
+ |-----------------------------------------+------------------------+----------------------+
5
+ | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC |
6
+ | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. |
7
+ | | | MIG M. |
8
+ |=========================================+========================+======================|
9
+ | 0 NVIDIA RTX A4000 Off | 00000000:01:00.0 Off | Off |
10
+ | 41% 60C P2 128W / 140W | 14035MiB / 16376MiB | 83% Default |
11
+ | | | N/A |
12
+ +-----------------------------------------+------------------------+----------------------+
13
+
14
+ +-----------------------------------------------------------------------------------------+
15
+ | Processes: |
16
+ | GPU GI CI PID Type Process name GPU Memory |
17
+ | ID ID Usage |
18
+ |=========================================================================================|
19
+ | 0 N/A N/A 76833 G /usr/lib/xorg/Xorg 4MiB |
20
+ | 0 N/A N/A 758019 C ...ik/miniforge3/envs/alpha/bin/python 14020MiB |
21
+ +-----------------------------------------------------------------------------------------+
trainer_state.json ADDED
@@ -0,0 +1,484 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 3.0,
5
+ "eval_steps": 3000,
6
+ "global_step": 88686,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 3.382721060821325e-05,
13
+ "grad_norm": 146.0,
14
+ "learning_rate": 0.00029999661727893914,
15
+ "loss": 28.0,
16
+ "step": 1
17
+ },
18
+ {
19
+ "epoch": 0.10148163182463973,
20
+ "grad_norm": 1.0625,
21
+ "learning_rate": 0.000289851836817536,
22
+ "loss": 4.8045,
23
+ "step": 3000
24
+ },
25
+ {
26
+ "epoch": 0.10148163182463973,
27
+ "eval_loss": 3.3920648097991943,
28
+ "eval_runtime": 84.2066,
29
+ "eval_samples_per_second": 1121.111,
30
+ "eval_steps_per_second": 8.764,
31
+ "step": 3000
32
+ },
33
+ {
34
+ "epoch": 0.20296326364927947,
35
+ "grad_norm": 1.1875,
36
+ "learning_rate": 0.000279703673635072,
37
+ "loss": 4.1932,
38
+ "step": 6000
39
+ },
40
+ {
41
+ "epoch": 0.20296326364927947,
42
+ "eval_loss": 3.254700183868408,
43
+ "eval_runtime": 83.4018,
44
+ "eval_samples_per_second": 1131.93,
45
+ "eval_steps_per_second": 8.849,
46
+ "step": 6000
47
+ },
48
+ {
49
+ "epoch": 0.30444489547391923,
50
+ "grad_norm": 1.15625,
51
+ "learning_rate": 0.00026955551045260807,
52
+ "loss": 4.1098,
53
+ "step": 9000
54
+ },
55
+ {
56
+ "epoch": 0.30444489547391923,
57
+ "eval_loss": 3.204374074935913,
58
+ "eval_runtime": 83.3884,
59
+ "eval_samples_per_second": 1132.112,
60
+ "eval_steps_per_second": 8.85,
61
+ "step": 9000
62
+ },
63
+ {
64
+ "epoch": 0.40592652729855894,
65
+ "grad_norm": 1.15625,
66
+ "learning_rate": 0.0002594073472701441,
67
+ "loss": 4.0752,
68
+ "step": 12000
69
+ },
70
+ {
71
+ "epoch": 0.40592652729855894,
72
+ "eval_loss": 3.173970937728882,
73
+ "eval_runtime": 83.3988,
74
+ "eval_samples_per_second": 1131.971,
75
+ "eval_steps_per_second": 8.849,
76
+ "step": 12000
77
+ },
78
+ {
79
+ "epoch": 0.5074081591231987,
80
+ "grad_norm": 1.21875,
81
+ "learning_rate": 0.0002492591840876801,
82
+ "loss": 4.0515,
83
+ "step": 15000
84
+ },
85
+ {
86
+ "epoch": 0.5074081591231987,
87
+ "eval_loss": 3.157351016998291,
88
+ "eval_runtime": 83.395,
89
+ "eval_samples_per_second": 1132.022,
90
+ "eval_steps_per_second": 8.849,
91
+ "step": 15000
92
+ },
93
+ {
94
+ "epoch": 0.6088897909478385,
95
+ "grad_norm": 1.1328125,
96
+ "learning_rate": 0.0002391110209052161,
97
+ "loss": 4.0446,
98
+ "step": 18000
99
+ },
100
+ {
101
+ "epoch": 0.6088897909478385,
102
+ "eval_loss": 3.1490302085876465,
103
+ "eval_runtime": 83.4067,
104
+ "eval_samples_per_second": 1131.863,
105
+ "eval_steps_per_second": 8.848,
106
+ "step": 18000
107
+ },
108
+ {
109
+ "epoch": 0.7103714227724782,
110
+ "grad_norm": 1.1796875,
111
+ "learning_rate": 0.00022896285772275215,
112
+ "loss": 4.036,
113
+ "step": 21000
114
+ },
115
+ {
116
+ "epoch": 0.7103714227724782,
117
+ "eval_loss": 3.1390371322631836,
118
+ "eval_runtime": 83.3672,
119
+ "eval_samples_per_second": 1132.399,
120
+ "eval_steps_per_second": 8.852,
121
+ "step": 21000
122
+ },
123
+ {
124
+ "epoch": 0.8118530545971179,
125
+ "grad_norm": 1.1484375,
126
+ "learning_rate": 0.0002188146945402882,
127
+ "loss": 4.0298,
128
+ "step": 24000
129
+ },
130
+ {
131
+ "epoch": 0.8118530545971179,
132
+ "eval_loss": 3.1364541053771973,
133
+ "eval_runtime": 83.5197,
134
+ "eval_samples_per_second": 1130.332,
135
+ "eval_steps_per_second": 8.836,
136
+ "step": 24000
137
+ },
138
+ {
139
+ "epoch": 0.9133346864217576,
140
+ "grad_norm": 1.171875,
141
+ "learning_rate": 0.00020866653135782423,
142
+ "loss": 4.0273,
143
+ "step": 27000
144
+ },
145
+ {
146
+ "epoch": 0.9133346864217576,
147
+ "eval_loss": 3.1308434009552,
148
+ "eval_runtime": 83.5474,
149
+ "eval_samples_per_second": 1129.957,
150
+ "eval_steps_per_second": 8.833,
151
+ "step": 27000
152
+ },
153
+ {
154
+ "epoch": 1.0148163182463974,
155
+ "grad_norm": 1.171875,
156
+ "learning_rate": 0.00019851836817536025,
157
+ "loss": 4.023,
158
+ "step": 30000
159
+ },
160
+ {
161
+ "epoch": 1.0148163182463974,
162
+ "eval_loss": 3.1285781860351562,
163
+ "eval_runtime": 83.5362,
164
+ "eval_samples_per_second": 1130.109,
165
+ "eval_steps_per_second": 8.834,
166
+ "step": 30000
167
+ },
168
+ {
169
+ "epoch": 1.116297950071037,
170
+ "grad_norm": 1.1640625,
171
+ "learning_rate": 0.0001883702049928963,
172
+ "loss": 4.0229,
173
+ "step": 33000
174
+ },
175
+ {
176
+ "epoch": 1.116297950071037,
177
+ "eval_loss": 3.1244707107543945,
178
+ "eval_runtime": 83.5446,
179
+ "eval_samples_per_second": 1129.996,
180
+ "eval_steps_per_second": 8.834,
181
+ "step": 33000
182
+ },
183
+ {
184
+ "epoch": 1.217779581895677,
185
+ "grad_norm": 1.21875,
186
+ "learning_rate": 0.0001782220418104323,
187
+ "loss": 4.0158,
188
+ "step": 36000
189
+ },
190
+ {
191
+ "epoch": 1.217779581895677,
192
+ "eval_loss": 3.1234333515167236,
193
+ "eval_runtime": 83.5361,
194
+ "eval_samples_per_second": 1130.11,
195
+ "eval_steps_per_second": 8.835,
196
+ "step": 36000
197
+ },
198
+ {
199
+ "epoch": 1.3192612137203166,
200
+ "grad_norm": 1.1796875,
201
+ "learning_rate": 0.00016807387862796832,
202
+ "loss": 4.0163,
203
+ "step": 39000
204
+ },
205
+ {
206
+ "epoch": 1.3192612137203166,
207
+ "eval_loss": 3.121718406677246,
208
+ "eval_runtime": 83.5263,
209
+ "eval_samples_per_second": 1130.242,
210
+ "eval_steps_per_second": 8.836,
211
+ "step": 39000
212
+ },
213
+ {
214
+ "epoch": 1.4207428455449564,
215
+ "grad_norm": 1.2421875,
216
+ "learning_rate": 0.00015792571544550436,
217
+ "loss": 4.0156,
218
+ "step": 42000
219
+ },
220
+ {
221
+ "epoch": 1.4207428455449564,
222
+ "eval_loss": 3.1207656860351562,
223
+ "eval_runtime": 83.531,
224
+ "eval_samples_per_second": 1130.179,
225
+ "eval_steps_per_second": 8.835,
226
+ "step": 42000
227
+ },
228
+ {
229
+ "epoch": 1.522224477369596,
230
+ "grad_norm": 1.328125,
231
+ "learning_rate": 0.00014777755226304037,
232
+ "loss": 4.0147,
233
+ "step": 45000
234
+ },
235
+ {
236
+ "epoch": 1.522224477369596,
237
+ "eval_loss": 3.119389295578003,
238
+ "eval_runtime": 83.5355,
239
+ "eval_samples_per_second": 1130.119,
240
+ "eval_steps_per_second": 8.835,
241
+ "step": 45000
242
+ },
243
+ {
244
+ "epoch": 1.6237061091942357,
245
+ "grad_norm": 1.28125,
246
+ "learning_rate": 0.0001376293890805764,
247
+ "loss": 4.0114,
248
+ "step": 48000
249
+ },
250
+ {
251
+ "epoch": 1.6237061091942357,
252
+ "eval_loss": 3.11716628074646,
253
+ "eval_runtime": 83.398,
254
+ "eval_samples_per_second": 1131.981,
255
+ "eval_steps_per_second": 8.849,
256
+ "step": 48000
257
+ },
258
+ {
259
+ "epoch": 1.7251877410188756,
260
+ "grad_norm": 1.3515625,
261
+ "learning_rate": 0.00012748122589811243,
262
+ "loss": 4.0113,
263
+ "step": 51000
264
+ },
265
+ {
266
+ "epoch": 1.7251877410188756,
267
+ "eval_loss": 3.1180977821350098,
268
+ "eval_runtime": 83.3949,
269
+ "eval_samples_per_second": 1132.024,
270
+ "eval_steps_per_second": 8.849,
271
+ "step": 51000
272
+ },
273
+ {
274
+ "epoch": 1.8266693728435153,
275
+ "grad_norm": 1.40625,
276
+ "learning_rate": 0.00011733306271564845,
277
+ "loss": 4.0097,
278
+ "step": 54000
279
+ },
280
+ {
281
+ "epoch": 1.8266693728435153,
282
+ "eval_loss": 3.116975784301758,
283
+ "eval_runtime": 83.3757,
284
+ "eval_samples_per_second": 1132.284,
285
+ "eval_steps_per_second": 8.851,
286
+ "step": 54000
287
+ },
288
+ {
289
+ "epoch": 1.928151004668155,
290
+ "grad_norm": 1.140625,
291
+ "learning_rate": 0.00010718489953318448,
292
+ "loss": 4.0105,
293
+ "step": 57000
294
+ },
295
+ {
296
+ "epoch": 1.928151004668155,
297
+ "eval_loss": 3.116128921508789,
298
+ "eval_runtime": 83.3931,
299
+ "eval_samples_per_second": 1132.049,
300
+ "eval_steps_per_second": 8.85,
301
+ "step": 57000
302
+ },
303
+ {
304
+ "epoch": 2.029632636492795,
305
+ "grad_norm": 1.1953125,
306
+ "learning_rate": 9.703673635072052e-05,
307
+ "loss": 4.0115,
308
+ "step": 60000
309
+ },
310
+ {
311
+ "epoch": 2.029632636492795,
312
+ "eval_loss": 3.1166157722473145,
313
+ "eval_runtime": 83.3926,
314
+ "eval_samples_per_second": 1132.055,
315
+ "eval_steps_per_second": 8.85,
316
+ "step": 60000
317
+ },
318
+ {
319
+ "epoch": 2.1311142683174347,
320
+ "grad_norm": 1.4609375,
321
+ "learning_rate": 8.688857316825655e-05,
322
+ "loss": 4.0117,
323
+ "step": 63000
324
+ },
325
+ {
326
+ "epoch": 2.1311142683174347,
327
+ "eval_loss": 3.1167428493499756,
328
+ "eval_runtime": 83.4107,
329
+ "eval_samples_per_second": 1131.809,
330
+ "eval_steps_per_second": 8.848,
331
+ "step": 63000
332
+ },
333
+ {
334
+ "epoch": 2.232595900142074,
335
+ "grad_norm": 1.2109375,
336
+ "learning_rate": 7.674040998579256e-05,
337
+ "loss": 4.0082,
338
+ "step": 66000
339
+ },
340
+ {
341
+ "epoch": 2.232595900142074,
342
+ "eval_loss": 3.116044282913208,
343
+ "eval_runtime": 83.4033,
344
+ "eval_samples_per_second": 1131.909,
345
+ "eval_steps_per_second": 8.849,
346
+ "step": 66000
347
+ },
348
+ {
349
+ "epoch": 2.334077531966714,
350
+ "grad_norm": 1.265625,
351
+ "learning_rate": 6.659224680332859e-05,
352
+ "loss": 4.0089,
353
+ "step": 69000
354
+ },
355
+ {
356
+ "epoch": 2.334077531966714,
357
+ "eval_loss": 3.1157476902008057,
358
+ "eval_runtime": 83.3805,
359
+ "eval_samples_per_second": 1132.219,
360
+ "eval_steps_per_second": 8.851,
361
+ "step": 69000
362
+ },
363
+ {
364
+ "epoch": 2.435559163791354,
365
+ "grad_norm": 1.2421875,
366
+ "learning_rate": 5.644408362086462e-05,
367
+ "loss": 4.0087,
368
+ "step": 72000
369
+ },
370
+ {
371
+ "epoch": 2.435559163791354,
372
+ "eval_loss": 3.1155996322631836,
373
+ "eval_runtime": 83.4329,
374
+ "eval_samples_per_second": 1131.508,
375
+ "eval_steps_per_second": 8.845,
376
+ "step": 72000
377
+ },
378
+ {
379
+ "epoch": 2.5370407956159937,
380
+ "grad_norm": 1.34375,
381
+ "learning_rate": 4.629592043840065e-05,
382
+ "loss": 4.0104,
383
+ "step": 75000
384
+ },
385
+ {
386
+ "epoch": 2.5370407956159937,
387
+ "eval_loss": 3.115832567214966,
388
+ "eval_runtime": 83.4175,
389
+ "eval_samples_per_second": 1131.716,
390
+ "eval_steps_per_second": 8.847,
391
+ "step": 75000
392
+ },
393
+ {
394
+ "epoch": 2.638522427440633,
395
+ "grad_norm": 1.3203125,
396
+ "learning_rate": 3.614775725593667e-05,
397
+ "loss": 4.0088,
398
+ "step": 78000
399
+ },
400
+ {
401
+ "epoch": 2.638522427440633,
402
+ "eval_loss": 3.115938425064087,
403
+ "eval_runtime": 83.3984,
404
+ "eval_samples_per_second": 1131.977,
405
+ "eval_steps_per_second": 8.849,
406
+ "step": 78000
407
+ },
408
+ {
409
+ "epoch": 2.740004059265273,
410
+ "grad_norm": 1.234375,
411
+ "learning_rate": 2.59995940734727e-05,
412
+ "loss": 4.007,
413
+ "step": 81000
414
+ },
415
+ {
416
+ "epoch": 2.740004059265273,
417
+ "eval_loss": 3.115938425064087,
418
+ "eval_runtime": 83.4084,
419
+ "eval_samples_per_second": 1131.841,
420
+ "eval_steps_per_second": 8.848,
421
+ "step": 81000
422
+ },
423
+ {
424
+ "epoch": 2.841485691089913,
425
+ "grad_norm": 1.3984375,
426
+ "learning_rate": 1.5851430891008727e-05,
427
+ "loss": 4.0116,
428
+ "step": 84000
429
+ },
430
+ {
431
+ "epoch": 2.841485691089913,
432
+ "eval_loss": 3.116044282913208,
433
+ "eval_runtime": 83.4021,
434
+ "eval_samples_per_second": 1131.925,
435
+ "eval_steps_per_second": 8.849,
436
+ "step": 84000
437
+ },
438
+ {
439
+ "epoch": 2.9429673229145523,
440
+ "grad_norm": 1.2109375,
441
+ "learning_rate": 5.703267708544753e-06,
442
+ "loss": 4.0099,
443
+ "step": 87000
444
+ },
445
+ {
446
+ "epoch": 2.9429673229145523,
447
+ "eval_loss": 3.115874767303467,
448
+ "eval_runtime": 83.5591,
449
+ "eval_samples_per_second": 1129.799,
450
+ "eval_steps_per_second": 8.832,
451
+ "step": 87000
452
+ },
453
+ {
454
+ "epoch": 3.0,
455
+ "step": 88686,
456
+ "total_flos": 1.19291602746581e+18,
457
+ "train_loss": 4.055220102383691,
458
+ "train_runtime": 32766.006,
459
+ "train_samples_per_second": 346.44,
460
+ "train_steps_per_second": 2.707
461
+ }
462
+ ],
463
+ "logging_steps": 3000,
464
+ "max_steps": 88686,
465
+ "num_input_tokens_seen": 0,
466
+ "num_train_epochs": 3,
467
+ "save_steps": 3000,
468
+ "stateful_callbacks": {
469
+ "TrainerControl": {
470
+ "args": {
471
+ "should_epoch_stop": false,
472
+ "should_evaluate": false,
473
+ "should_log": false,
474
+ "should_save": true,
475
+ "should_training_stop": true
476
+ },
477
+ "attributes": {}
478
+ }
479
+ },
480
+ "total_flos": 1.19291602746581e+18,
481
+ "train_batch_size": 128,
482
+ "trial_name": null,
483
+ "trial_params": null
484
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a6eac3d7c2404d850551f86675cb0c3ca710dba990bbfb69b0f8d2533a6003f7
3
+ size 5368
training_args.json ADDED
@@ -0,0 +1,147 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "output_dir": "out_models/mt5-small-gigatrue-slovak-D-34567",
3
+ "overwrite_output_dir": false,
4
+ "do_train": false,
5
+ "do_eval": true,
6
+ "do_predict": false,
7
+ "eval_strategy": "steps",
8
+ "prediction_loss_only": false,
9
+ "per_device_train_batch_size": 128,
10
+ "per_device_eval_batch_size": 128,
11
+ "per_gpu_train_batch_size": null,
12
+ "per_gpu_eval_batch_size": null,
13
+ "gradient_accumulation_steps": 1,
14
+ "eval_accumulation_steps": null,
15
+ "eval_delay": 0,
16
+ "torch_empty_cache_steps": null,
17
+ "learning_rate": 0.0003,
18
+ "weight_decay": 0.0,
19
+ "adam_beta1": 0.9,
20
+ "adam_beta2": 0.999,
21
+ "adam_epsilon": 1e-08,
22
+ "max_grad_norm": 1.0,
23
+ "num_train_epochs": 3,
24
+ "max_steps": -1,
25
+ "lr_scheduler_type": "linear",
26
+ "lr_scheduler_kwargs": {},
27
+ "warmup_ratio": 0.0,
28
+ "warmup_steps": 0,
29
+ "log_level": "passive",
30
+ "log_level_replica": "warning",
31
+ "log_on_each_node": true,
32
+ "logging_dir": "out_models/mt5-small-gigatrue-slovak-D-34567/logs",
33
+ "logging_strategy": "steps",
34
+ "logging_first_step": true,
35
+ "logging_steps": 3000,
36
+ "logging_nan_inf_filter": true,
37
+ "save_strategy": "steps",
38
+ "save_steps": 3000,
39
+ "save_total_limit": null,
40
+ "save_safetensors": true,
41
+ "save_on_each_node": false,
42
+ "save_only_model": false,
43
+ "restore_callback_states_from_checkpoint": false,
44
+ "no_cuda": false,
45
+ "use_cpu": false,
46
+ "use_mps_device": false,
47
+ "seed": 42,
48
+ "data_seed": null,
49
+ "jit_mode_eval": false,
50
+ "use_ipex": false,
51
+ "bf16": false,
52
+ "fp16": false,
53
+ "fp16_opt_level": "O1",
54
+ "half_precision_backend": "auto",
55
+ "bf16_full_eval": false,
56
+ "fp16_full_eval": false,
57
+ "tf32": null,
58
+ "local_rank": 0,
59
+ "ddp_backend": null,
60
+ "tpu_num_cores": null,
61
+ "tpu_metrics_debug": false,
62
+ "debug": [],
63
+ "dataloader_drop_last": false,
64
+ "eval_steps": 3000,
65
+ "dataloader_num_workers": 0,
66
+ "dataloader_prefetch_factor": null,
67
+ "past_index": -1,
68
+ "run_name": "out_models/mt5-small-gigatrue-slovak-D-34567",
69
+ "disable_tqdm": false,
70
+ "remove_unused_columns": true,
71
+ "label_names": null,
72
+ "load_best_model_at_end": false,
73
+ "metric_for_best_model": null,
74
+ "greater_is_better": null,
75
+ "ignore_data_skip": false,
76
+ "fsdp": [],
77
+ "fsdp_min_num_params": 0,
78
+ "fsdp_config": {
79
+ "min_num_params": 0,
80
+ "xla": false,
81
+ "xla_fsdp_v2": false,
82
+ "xla_fsdp_grad_ckpt": false
83
+ },
84
+ "fsdp_transformer_layer_cls_to_wrap": null,
85
+ "accelerator_config": "AcceleratorConfig",
86
+ "deepspeed": null,
87
+ "label_smoothing_factor": 0.0,
88
+ "optim": "adamw_torch",
89
+ "optim_args": null,
90
+ "adafactor": false,
91
+ "group_by_length": false,
92
+ "length_column_name": "length",
93
+ "report_to": [
94
+ "tensorboard"
95
+ ],
96
+ "ddp_find_unused_parameters": null,
97
+ "ddp_bucket_cap_mb": null,
98
+ "ddp_broadcast_buffers": null,
99
+ "dataloader_pin_memory": true,
100
+ "dataloader_persistent_workers": false,
101
+ "skip_memory_metrics": true,
102
+ "use_legacy_prediction_loop": false,
103
+ "push_to_hub": false,
104
+ "resume_from_checkpoint": null,
105
+ "hub_model_id": null,
106
+ "hub_strategy": "every_save",
107
+ "hub_token": null,
108
+ "hub_private_repo": false,
109
+ "hub_always_push": false,
110
+ "gradient_checkpointing": false,
111
+ "gradient_checkpointing_kwargs": null,
112
+ "include_inputs_for_metrics": false,
113
+ "eval_do_concat_batches": true,
114
+ "fp16_backend": "auto",
115
+ "evaluation_strategy": null,
116
+ "push_to_hub_model_id": null,
117
+ "push_to_hub_organization": null,
118
+ "push_to_hub_token": null,
119
+ "mp_parameters": "",
120
+ "auto_find_batch_size": false,
121
+ "full_determinism": false,
122
+ "torchdynamo": null,
123
+ "ray_scope": "last",
124
+ "ddp_timeout": 1800,
125
+ "torch_compile": false,
126
+ "torch_compile_backend": null,
127
+ "torch_compile_mode": null,
128
+ "dispatch_batches": null,
129
+ "split_batches": null,
130
+ "include_tokens_per_second": false,
131
+ "include_num_input_tokens_seen": false,
132
+ "neftune_noise_alpha": null,
133
+ "optim_target_modules": null,
134
+ "batch_eval_metrics": false,
135
+ "eval_on_start": false,
136
+ "use_liger_kernel": false,
137
+ "eval_use_gather_object": false,
138
+ "sortish_sampler": false,
139
+ "predict_with_generate": false,
140
+ "generation_max_length": null,
141
+ "generation_num_beams": null,
142
+ "generation_config": null,
143
+ "distributed_state": "PartialState",
144
+ "_n_gpu": 1,
145
+ "__cached__setup_devices": "device",
146
+ "deepspeed_plugin": null
147
+ }