DewiBrynJones commited on
Commit
d4dda56
·
verified ·
1 Parent(s): aeecc20

Upload folder using huggingface_hub

Browse files
README_new.md ADDED
@@ -0,0 +1,69 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: apache-2.0
4
+ base_model: openai/whisper-large-v3
5
+ tags:
6
+ - generated_from_trainer
7
+ metrics:
8
+ - wer
9
+ model-index:
10
+ - name: whisper-large-v3-ft-btb-cv-cvad-ca-cy-2503
11
+ results: []
12
+ ---
13
+
14
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
15
+ should probably proofread and complete it, then remove this comment. -->
16
+
17
+ # whisper-large-v3-ft-btb-cv-cvad-ca-cy-2503
18
+
19
+ This model is a fine-tuned version of [openai/whisper-large-v3](https://huggingface.co/openai/whisper-large-v3) on the DewiBrynJones/banc-trawsgrifiadau-bangor train main, DewiBrynJones/commonvoice_18_0_cy train+dev+other_with_excluded main, cymen-arfor/lleisiau-arfor train+dev main, techiaith/commonvoice_vad_cy train main dataset.
20
+ It achieves the following results on the evaluation set:
21
+ - Loss: 0.3739
22
+ - Wer: 0.2915
23
+
24
+ ## Model description
25
+
26
+ More information needed
27
+
28
+ ## Intended uses & limitations
29
+
30
+ More information needed
31
+
32
+ ## Training and evaluation data
33
+
34
+ More information needed
35
+
36
+ ## Training procedure
37
+
38
+ ### Training hyperparameters
39
+
40
+ The following hyperparameters were used during training:
41
+ - learning_rate: 1e-05
42
+ - train_batch_size: 16
43
+ - eval_batch_size: 16
44
+ - seed: 42
45
+ - gradient_accumulation_steps: 2
46
+ - total_train_batch_size: 32
47
+ - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
48
+ - lr_scheduler_type: linear
49
+ - lr_scheduler_warmup_steps: 500
50
+ - training_steps: 5000
51
+ - mixed_precision_training: Native AMP
52
+
53
+ ### Training results
54
+
55
+ | Training Loss | Epoch | Step | Validation Loss | Wer |
56
+ |:-------------:|:------:|:----:|:---------------:|:------:|
57
+ | 0.5304 | 0.3240 | 1000 | 0.5236 | 0.3786 |
58
+ | 0.4409 | 0.6480 | 2000 | 0.4458 | 0.3563 |
59
+ | 0.3914 | 0.9720 | 3000 | 0.4035 | 0.3090 |
60
+ | 0.296 | 1.2958 | 4000 | 0.3868 | 0.2977 |
61
+ | 0.274 | 1.6198 | 5000 | 0.3739 | 0.2915 |
62
+
63
+
64
+ ### Framework versions
65
+
66
+ - Transformers 4.49.0
67
+ - Pytorch 2.6.0+cu124
68
+ - Datasets 3.3.2
69
+ - Tokenizers 0.21.0
all_results.json CHANGED
@@ -1,15 +1,15 @@
1
  {
2
- "epoch": 2.8555111364934325,
3
- "eval_loss": 0.38383349776268005,
4
- "eval_runtime": 1673.49,
5
- "eval_samples": 3901,
6
- "eval_samples_per_second": 2.331,
7
- "eval_steps_per_second": 0.146,
8
- "eval_wer": 0.27318168646769053,
9
- "total_flos": 5.435589590699213e+20,
10
- "train_loss": 0.3002769865989685,
11
- "train_runtime": 59305.2217,
12
- "train_samples": 56026,
13
- "train_samples_per_second": 2.698,
14
- "train_steps_per_second": 0.084
15
  }
 
1
  {
2
+ "epoch": 1.619795885306982,
3
+ "eval_loss": 0.3738669753074646,
4
+ "eval_runtime": 1862.9517,
5
+ "eval_samples": 3897,
6
+ "eval_samples_per_second": 2.092,
7
+ "eval_steps_per_second": 0.131,
8
+ "eval_wer": 0.29154381412902464,
9
+ "total_flos": 5.435419715783885e+20,
10
+ "train_loss": 0.4225531764030456,
11
+ "train_runtime": 64153.0147,
12
+ "train_samples": 98767,
13
+ "train_samples_per_second": 2.494,
14
+ "train_steps_per_second": 0.078
15
  }
config.json CHANGED
@@ -7,10 +7,7 @@
7
  "WhisperForConditionalGeneration"
8
  ],
9
  "attention_dropout": 0.0,
10
- "begin_suppress_tokens": [
11
- 220,
12
- 50257
13
- ],
14
  "bos_token_id": 50257,
15
  "classifier_proj_size": 256,
16
  "d_model": 1280,
@@ -34,7 +31,7 @@
34
  "mask_time_length": 10,
35
  "mask_time_min_masks": 2,
36
  "mask_time_prob": 0.05,
37
- "max_length": 448,
38
  "max_source_positions": 1500,
39
  "max_target_positions": 448,
40
  "median_filter_width": 7,
@@ -44,7 +41,7 @@
44
  "pad_token_id": 50256,
45
  "scale_embedding": false,
46
  "torch_dtype": "float32",
47
- "transformers_version": "4.44.0",
48
  "use_cache": true,
49
  "use_weighted_layer_sum": false,
50
  "vocab_size": 51866
 
7
  "WhisperForConditionalGeneration"
8
  ],
9
  "attention_dropout": 0.0,
10
+ "begin_suppress_tokens": null,
 
 
 
11
  "bos_token_id": 50257,
12
  "classifier_proj_size": 256,
13
  "d_model": 1280,
 
31
  "mask_time_length": 10,
32
  "mask_time_min_masks": 2,
33
  "mask_time_prob": 0.05,
34
+ "max_length": null,
35
  "max_source_positions": 1500,
36
  "max_target_positions": 448,
37
  "median_filter_width": 7,
 
41
  "pad_token_id": 50256,
42
  "scale_embedding": false,
43
  "torch_dtype": "float32",
44
+ "transformers_version": "4.49.0",
45
  "use_cache": true,
46
  "use_weighted_layer_sum": false,
47
  "vocab_size": 51866
eval_results.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
- "epoch": 2.8555111364934325,
3
- "eval_loss": 0.38383349776268005,
4
- "eval_runtime": 1673.49,
5
- "eval_samples": 3901,
6
- "eval_samples_per_second": 2.331,
7
- "eval_steps_per_second": 0.146,
8
- "eval_wer": 0.27318168646769053
9
  }
 
1
  {
2
+ "epoch": 1.619795885306982,
3
+ "eval_loss": 0.3738669753074646,
4
+ "eval_runtime": 1862.9517,
5
+ "eval_samples": 3897,
6
+ "eval_samples_per_second": 2.092,
7
+ "eval_steps_per_second": 0.131,
8
+ "eval_wer": 0.29154381412902464
9
  }
generation_config.json CHANGED
@@ -151,7 +151,7 @@
151
  "<|yue|>": 50358,
152
  "<|zh|>": 50260
153
  },
154
- "language": "welsh",
155
  "max_initial_timestamp_index": 50,
156
  "max_length": 448,
157
  "no_timestamps_token_id": 50364,
@@ -253,5 +253,5 @@
253
  "transcribe": 50360,
254
  "translate": 50359
255
  },
256
- "transformers_version": "4.44.0"
257
  }
 
151
  "<|yue|>": 50358,
152
  "<|zh|>": 50260
153
  },
154
+ "language": null,
155
  "max_initial_timestamp_index": 50,
156
  "max_length": 448,
157
  "no_timestamps_token_id": 50364,
 
253
  "transcribe": 50360,
254
  "translate": 50359
255
  },
256
+ "transformers_version": "4.49.0"
257
  }
model-00001-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:659ef032a319119312a7e5e3462e0b9a92789d2adcecabdea056d6acf30934a5
3
  size 4993448880
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89026adf241ab6e6f7d62cc3a2c3e4007c8b34ef54d1222ee9d8e204ccb4653d
3
  size 4993448880
model-00002-of-00002.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:0f64973e37d3581c270548790f99059bb3e41201557a5bb095fe22cefc89f00c
3
  size 1180663192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d5eb80d43bb52300a1166a1892b4d9fa06d3b0546ffb4338f7caa0422698d839
3
  size 1180663192
runs/Mar22_10-30-53_d4a283a05c40/events.out.tfevents.1742640704.d4a283a05c40.213.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da50f05716832fd4988146bdb5ba728d3f7382fe0cefec5700063559e37a0e67
3
+ size 50135
runs/Mar22_10-30-53_d4a283a05c40/events.out.tfevents.1742706967.d4a283a05c40.213.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:054c3fa54b2f8c098eb2a448fa6f33487c4a63baa3d8f235cbef0fa6d33f7fbc
3
+ size 406
tokenizer.json CHANGED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json CHANGED
@@ -12987,6 +12987,7 @@
12987
  "clean_up_tokenization_spaces": true,
12988
  "eos_token": "<|endoftext|>",
12989
  "errors": "replace",
 
12990
  "model_max_length": 1000000000000000019884624838656,
12991
  "pad_token": "<|endoftext|>",
12992
  "processor_class": "WhisperProcessor",
 
12987
  "clean_up_tokenization_spaces": true,
12988
  "eos_token": "<|endoftext|>",
12989
  "errors": "replace",
12990
+ "extra_special_tokens": {},
12991
  "model_max_length": 1000000000000000019884624838656,
12992
  "pad_token": "<|endoftext|>",
12993
  "processor_class": "WhisperProcessor",
train_results.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
- "epoch": 2.8555111364934325,
3
- "total_flos": 5.435589590699213e+20,
4
- "train_loss": 0.3002769865989685,
5
- "train_runtime": 59305.2217,
6
- "train_samples": 56026,
7
- "train_samples_per_second": 2.698,
8
- "train_steps_per_second": 0.084
9
  }
 
1
  {
2
+ "epoch": 1.619795885306982,
3
+ "total_flos": 5.435419715783885e+20,
4
+ "train_loss": 0.4225531764030456,
5
+ "train_runtime": 64153.0147,
6
+ "train_samples": 98767,
7
+ "train_samples_per_second": 2.494,
8
+ "train_steps_per_second": 0.078
9
  }
trainer_state.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
- "epoch": 2.8555111364934325,
5
  "eval_steps": 1000,
6
  "global_step": 5000,
7
  "is_hyper_param_search": false,
@@ -9,1464 +9,1464 @@
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
- "epoch": 0.014277555682467162,
13
- "grad_norm": 7.571424961090088,
14
  "learning_rate": 5.000000000000001e-07,
15
- "loss": 1.5088,
16
  "step": 25
17
  },
18
  {
19
- "epoch": 0.028555111364934323,
20
- "grad_norm": 5.992729187011719,
21
  "learning_rate": 1.0000000000000002e-06,
22
- "loss": 1.2038,
23
  "step": 50
24
  },
25
  {
26
- "epoch": 0.04283266704740148,
27
- "grad_norm": 5.949503421783447,
28
  "learning_rate": 1.5e-06,
29
- "loss": 0.8879,
30
  "step": 75
31
  },
32
  {
33
- "epoch": 0.05711022272986865,
34
- "grad_norm": 4.452832221984863,
35
  "learning_rate": 2.0000000000000003e-06,
36
- "loss": 0.7647,
37
  "step": 100
38
  },
39
  {
40
- "epoch": 0.0713877784123358,
41
- "grad_norm": 4.690545558929443,
42
  "learning_rate": 2.5e-06,
43
- "loss": 0.6792,
44
  "step": 125
45
  },
46
  {
47
- "epoch": 0.08566533409480297,
48
- "grad_norm": 4.969720840454102,
49
  "learning_rate": 3e-06,
50
- "loss": 0.6549,
51
  "step": 150
52
  },
53
  {
54
- "epoch": 0.09994288977727013,
55
- "grad_norm": 5.184281349182129,
56
  "learning_rate": 3.5e-06,
57
- "loss": 0.6376,
58
  "step": 175
59
  },
60
  {
61
- "epoch": 0.1142204454597373,
62
- "grad_norm": 5.00349235534668,
63
  "learning_rate": 4.000000000000001e-06,
64
- "loss": 0.5982,
65
  "step": 200
66
  },
67
  {
68
- "epoch": 0.12849800114220444,
69
- "grad_norm": 4.239490032196045,
70
  "learning_rate": 4.5e-06,
71
- "loss": 0.6084,
72
  "step": 225
73
  },
74
  {
75
- "epoch": 0.1427755568246716,
76
- "grad_norm": 4.2740068435668945,
77
  "learning_rate": 5e-06,
78
- "loss": 0.58,
79
  "step": 250
80
  },
81
  {
82
- "epoch": 0.15705311250713877,
83
- "grad_norm": 4.718848705291748,
84
  "learning_rate": 5.500000000000001e-06,
85
- "loss": 0.5759,
86
  "step": 275
87
  },
88
  {
89
- "epoch": 0.17133066818960593,
90
- "grad_norm": 4.2935638427734375,
91
  "learning_rate": 6e-06,
92
- "loss": 0.5625,
93
  "step": 300
94
  },
95
  {
96
- "epoch": 0.1856082238720731,
97
- "grad_norm": 4.917020797729492,
98
  "learning_rate": 6.5000000000000004e-06,
99
- "loss": 0.5621,
100
  "step": 325
101
  },
102
  {
103
- "epoch": 0.19988577955454026,
104
- "grad_norm": 3.9521942138671875,
105
  "learning_rate": 7e-06,
106
- "loss": 0.5644,
107
  "step": 350
108
  },
109
  {
110
- "epoch": 0.21416333523700742,
111
- "grad_norm": 4.506232738494873,
112
  "learning_rate": 7.500000000000001e-06,
113
- "loss": 0.5508,
114
  "step": 375
115
  },
116
  {
117
- "epoch": 0.2284408909194746,
118
- "grad_norm": 4.1483540534973145,
119
  "learning_rate": 8.000000000000001e-06,
120
- "loss": 0.5244,
121
  "step": 400
122
  },
123
  {
124
- "epoch": 0.24271844660194175,
125
- "grad_norm": 4.077396392822266,
126
  "learning_rate": 8.5e-06,
127
- "loss": 0.5051,
128
  "step": 425
129
  },
130
  {
131
- "epoch": 0.2569960022844089,
132
- "grad_norm": 4.375626087188721,
133
  "learning_rate": 9e-06,
134
- "loss": 0.5222,
135
  "step": 450
136
  },
137
  {
138
- "epoch": 0.2712735579668761,
139
- "grad_norm": 3.5698530673980713,
140
  "learning_rate": 9.5e-06,
141
- "loss": 0.5038,
142
  "step": 475
143
  },
144
  {
145
- "epoch": 0.2855511136493432,
146
- "grad_norm": 4.99509859085083,
147
  "learning_rate": 1e-05,
148
- "loss": 0.5196,
149
  "step": 500
150
  },
151
  {
152
- "epoch": 0.2998286693318104,
153
- "grad_norm": 3.666332721710205,
154
  "learning_rate": 9.944444444444445e-06,
155
- "loss": 0.5066,
156
  "step": 525
157
  },
158
  {
159
- "epoch": 0.31410622501427754,
160
- "grad_norm": 3.9203736782073975,
161
  "learning_rate": 9.88888888888889e-06,
162
- "loss": 0.4822,
163
  "step": 550
164
  },
165
  {
166
- "epoch": 0.32838378069674473,
167
- "grad_norm": 3.5677530765533447,
168
  "learning_rate": 9.833333333333333e-06,
169
- "loss": 0.519,
170
  "step": 575
171
  },
172
  {
173
- "epoch": 0.34266133637921187,
174
- "grad_norm": 3.3873414993286133,
175
  "learning_rate": 9.777777777777779e-06,
176
- "loss": 0.5205,
177
  "step": 600
178
  },
179
  {
180
- "epoch": 0.35693889206167906,
181
- "grad_norm": 3.9527816772460938,
182
  "learning_rate": 9.722222222222223e-06,
183
- "loss": 0.4769,
184
  "step": 625
185
  },
186
  {
187
- "epoch": 0.3712164477441462,
188
- "grad_norm": 3.3437490463256836,
189
  "learning_rate": 9.666666666666667e-06,
190
- "loss": 0.4629,
191
  "step": 650
192
  },
193
  {
194
- "epoch": 0.3854940034266134,
195
- "grad_norm": 3.7754790782928467,
196
  "learning_rate": 9.611111111111112e-06,
197
- "loss": 0.4812,
198
  "step": 675
199
  },
200
  {
201
- "epoch": 0.3997715591090805,
202
- "grad_norm": 3.744267225265503,
203
  "learning_rate": 9.555555555555556e-06,
204
- "loss": 0.467,
205
  "step": 700
206
  },
207
  {
208
- "epoch": 0.4140491147915477,
209
- "grad_norm": 3.5076072216033936,
210
  "learning_rate": 9.5e-06,
211
- "loss": 0.4454,
212
  "step": 725
213
  },
214
  {
215
- "epoch": 0.42832667047401485,
216
- "grad_norm": 3.556335687637329,
217
  "learning_rate": 9.444444444444445e-06,
218
- "loss": 0.4447,
219
  "step": 750
220
  },
221
  {
222
- "epoch": 0.442604226156482,
223
- "grad_norm": 4.256951332092285,
224
  "learning_rate": 9.38888888888889e-06,
225
- "loss": 0.4809,
226
  "step": 775
227
  },
228
  {
229
- "epoch": 0.4568817818389492,
230
- "grad_norm": 3.533447742462158,
231
  "learning_rate": 9.333333333333334e-06,
232
- "loss": 0.4425,
233
  "step": 800
234
  },
235
  {
236
- "epoch": 0.4711593375214163,
237
- "grad_norm": 4.324098587036133,
238
  "learning_rate": 9.277777777777778e-06,
239
- "loss": 0.424,
240
  "step": 825
241
  },
242
  {
243
- "epoch": 0.4854368932038835,
244
- "grad_norm": 2.913189649581909,
245
  "learning_rate": 9.222222222222224e-06,
246
- "loss": 0.4314,
247
  "step": 850
248
  },
249
  {
250
- "epoch": 0.49971444888635064,
251
- "grad_norm": 3.432490825653076,
252
  "learning_rate": 9.166666666666666e-06,
253
- "loss": 0.4355,
254
  "step": 875
255
  },
256
  {
257
- "epoch": 0.5139920045688178,
258
- "grad_norm": 3.645869255065918,
259
  "learning_rate": 9.111111111111112e-06,
260
- "loss": 0.4395,
261
  "step": 900
262
  },
263
  {
264
- "epoch": 0.528269560251285,
265
- "grad_norm": 3.2094240188598633,
266
  "learning_rate": 9.055555555555556e-06,
267
- "loss": 0.4144,
268
  "step": 925
269
  },
270
  {
271
- "epoch": 0.5425471159337522,
272
- "grad_norm": 3.4623546600341797,
273
  "learning_rate": 9e-06,
274
- "loss": 0.4277,
275
  "step": 950
276
  },
277
  {
278
- "epoch": 0.5568246716162193,
279
- "grad_norm": 3.640333414077759,
280
  "learning_rate": 8.944444444444446e-06,
281
- "loss": 0.4246,
282
  "step": 975
283
  },
284
  {
285
- "epoch": 0.5711022272986864,
286
- "grad_norm": 3.0283167362213135,
287
  "learning_rate": 8.888888888888888e-06,
288
- "loss": 0.4047,
289
  "step": 1000
290
  },
291
  {
292
- "epoch": 0.5711022272986864,
293
- "eval_loss": 0.4848648011684418,
294
- "eval_runtime": 1825.4203,
295
- "eval_samples_per_second": 2.137,
296
- "eval_steps_per_second": 0.134,
297
- "eval_wer": 0.35052641746353713,
298
  "step": 1000
299
  },
300
  {
301
- "epoch": 0.5853797829811537,
302
- "grad_norm": 3.7762739658355713,
303
  "learning_rate": 8.833333333333334e-06,
304
- "loss": 0.4218,
305
  "step": 1025
306
  },
307
  {
308
- "epoch": 0.5996573386636208,
309
- "grad_norm": 3.495347023010254,
310
  "learning_rate": 8.777777777777778e-06,
311
- "loss": 0.3968,
312
  "step": 1050
313
  },
314
  {
315
- "epoch": 0.613934894346088,
316
- "grad_norm": 3.5088939666748047,
317
  "learning_rate": 8.722222222222224e-06,
318
- "loss": 0.4108,
319
  "step": 1075
320
  },
321
  {
322
- "epoch": 0.6282124500285551,
323
- "grad_norm": 3.555328845977783,
324
  "learning_rate": 8.666666666666668e-06,
325
- "loss": 0.4063,
326
  "step": 1100
327
  },
328
  {
329
- "epoch": 0.6424900057110223,
330
- "grad_norm": 2.9576587677001953,
331
  "learning_rate": 8.611111111111112e-06,
332
- "loss": 0.4116,
333
  "step": 1125
334
  },
335
  {
336
- "epoch": 0.6567675613934895,
337
- "grad_norm": 3.280855178833008,
338
  "learning_rate": 8.555555555555556e-06,
339
- "loss": 0.4083,
340
  "step": 1150
341
  },
342
  {
343
- "epoch": 0.6710451170759566,
344
- "grad_norm": 3.903722047805786,
345
  "learning_rate": 8.5e-06,
346
- "loss": 0.411,
347
  "step": 1175
348
  },
349
  {
350
- "epoch": 0.6853226727584237,
351
- "grad_norm": 3.519038438796997,
352
  "learning_rate": 8.444444444444446e-06,
353
- "loss": 0.3964,
354
  "step": 1200
355
  },
356
  {
357
- "epoch": 0.6996002284408909,
358
- "grad_norm": 3.3553972244262695,
359
  "learning_rate": 8.38888888888889e-06,
360
- "loss": 0.4049,
361
  "step": 1225
362
  },
363
  {
364
- "epoch": 0.7138777841233581,
365
- "grad_norm": 3.3820197582244873,
366
  "learning_rate": 8.333333333333334e-06,
367
- "loss": 0.4159,
368
  "step": 1250
369
  },
370
  {
371
- "epoch": 0.7281553398058253,
372
- "grad_norm": 2.782127857208252,
373
  "learning_rate": 8.277777777777778e-06,
374
- "loss": 0.3859,
375
  "step": 1275
376
  },
377
  {
378
- "epoch": 0.7424328954882924,
379
- "grad_norm": 3.5839345455169678,
380
  "learning_rate": 8.222222222222222e-06,
381
- "loss": 0.392,
382
  "step": 1300
383
  },
384
  {
385
- "epoch": 0.7567104511707595,
386
- "grad_norm": 3.0308761596679688,
387
  "learning_rate": 8.166666666666668e-06,
388
- "loss": 0.3899,
389
  "step": 1325
390
  },
391
  {
392
- "epoch": 0.7709880068532268,
393
- "grad_norm": 3.136904001235962,
394
  "learning_rate": 8.111111111111112e-06,
395
- "loss": 0.3907,
396
  "step": 1350
397
  },
398
  {
399
- "epoch": 0.7852655625356939,
400
- "grad_norm": 3.3192756175994873,
401
  "learning_rate": 8.055555555555557e-06,
402
- "loss": 0.3941,
403
  "step": 1375
404
  },
405
  {
406
- "epoch": 0.799543118218161,
407
- "grad_norm": 4.766107082366943,
408
  "learning_rate": 8.000000000000001e-06,
409
- "loss": 0.3887,
410
  "step": 1400
411
  },
412
  {
413
- "epoch": 0.8138206739006282,
414
- "grad_norm": 4.241744041442871,
415
  "learning_rate": 7.944444444444445e-06,
416
- "loss": 0.4033,
417
  "step": 1425
418
  },
419
  {
420
- "epoch": 0.8280982295830954,
421
- "grad_norm": 3.1559460163116455,
422
  "learning_rate": 7.88888888888889e-06,
423
- "loss": 0.3567,
424
  "step": 1450
425
  },
426
  {
427
- "epoch": 0.8423757852655626,
428
- "grad_norm": 3.142645835876465,
429
  "learning_rate": 7.833333333333333e-06,
430
- "loss": 0.3731,
431
  "step": 1475
432
  },
433
  {
434
- "epoch": 0.8566533409480297,
435
- "grad_norm": 3.1183199882507324,
436
  "learning_rate": 7.77777777777778e-06,
437
- "loss": 0.3668,
438
  "step": 1500
439
  },
440
  {
441
- "epoch": 0.8709308966304968,
442
- "grad_norm": 2.7859325408935547,
443
  "learning_rate": 7.722222222222223e-06,
444
- "loss": 0.3965,
445
  "step": 1525
446
  },
447
  {
448
- "epoch": 0.885208452312964,
449
- "grad_norm": 3.191088914871216,
450
  "learning_rate": 7.666666666666667e-06,
451
- "loss": 0.3574,
452
  "step": 1550
453
  },
454
  {
455
- "epoch": 0.8994860079954312,
456
- "grad_norm": 3.0640053749084473,
457
  "learning_rate": 7.611111111111111e-06,
458
- "loss": 0.3811,
459
  "step": 1575
460
  },
461
  {
462
- "epoch": 0.9137635636778983,
463
- "grad_norm": 3.0769450664520264,
464
  "learning_rate": 7.555555555555556e-06,
465
- "loss": 0.3788,
466
  "step": 1600
467
  },
468
  {
469
- "epoch": 0.9280411193603655,
470
- "grad_norm": 3.1407933235168457,
471
  "learning_rate": 7.500000000000001e-06,
472
- "loss": 0.3698,
473
  "step": 1625
474
  },
475
  {
476
- "epoch": 0.9423186750428326,
477
- "grad_norm": 3.410187244415283,
478
  "learning_rate": 7.444444444444445e-06,
479
- "loss": 0.3907,
480
  "step": 1650
481
  },
482
  {
483
- "epoch": 0.9565962307252999,
484
- "grad_norm": 3.3382880687713623,
485
  "learning_rate": 7.38888888888889e-06,
486
- "loss": 0.3368,
487
  "step": 1675
488
  },
489
  {
490
- "epoch": 0.970873786407767,
491
- "grad_norm": 3.194368600845337,
492
  "learning_rate": 7.333333333333333e-06,
493
- "loss": 0.369,
494
  "step": 1700
495
  },
496
  {
497
- "epoch": 0.9851513420902341,
498
- "grad_norm": 3.089852809906006,
499
  "learning_rate": 7.277777777777778e-06,
500
- "loss": 0.3765,
501
  "step": 1725
502
  },
503
  {
504
- "epoch": 0.9994288977727013,
505
- "grad_norm": 3.0002810955047607,
506
  "learning_rate": 7.222222222222223e-06,
507
- "loss": 0.3705,
508
  "step": 1750
509
  },
510
  {
511
- "epoch": 1.0137064534551685,
512
- "grad_norm": 2.3977696895599365,
513
  "learning_rate": 7.166666666666667e-06,
514
- "loss": 0.2584,
515
  "step": 1775
516
  },
517
  {
518
- "epoch": 1.0279840091376355,
519
- "grad_norm": 2.3220465183258057,
520
  "learning_rate": 7.111111111111112e-06,
521
- "loss": 0.2538,
522
  "step": 1800
523
  },
524
  {
525
- "epoch": 1.0422615648201028,
526
- "grad_norm": 2.819687843322754,
527
  "learning_rate": 7.055555555555557e-06,
528
- "loss": 0.2571,
529
  "step": 1825
530
  },
531
  {
532
- "epoch": 1.05653912050257,
533
- "grad_norm": 2.514644145965576,
534
  "learning_rate": 7e-06,
535
- "loss": 0.2806,
536
  "step": 1850
537
  },
538
  {
539
- "epoch": 1.070816676185037,
540
- "grad_norm": 2.1887128353118896,
541
  "learning_rate": 6.944444444444445e-06,
542
- "loss": 0.2626,
543
  "step": 1875
544
  },
545
  {
546
- "epoch": 1.0850942318675043,
547
- "grad_norm": 2.592247486114502,
548
  "learning_rate": 6.88888888888889e-06,
549
- "loss": 0.2509,
550
  "step": 1900
551
  },
552
  {
553
- "epoch": 1.0993717875499716,
554
- "grad_norm": 2.371534824371338,
555
  "learning_rate": 6.833333333333334e-06,
556
- "loss": 0.2605,
557
  "step": 1925
558
  },
559
  {
560
- "epoch": 1.1136493432324386,
561
- "grad_norm": 3.1825778484344482,
562
  "learning_rate": 6.777777777777779e-06,
563
- "loss": 0.2495,
564
  "step": 1950
565
  },
566
  {
567
- "epoch": 1.1279268989149058,
568
- "grad_norm": 2.901749849319458,
569
  "learning_rate": 6.7222222222222235e-06,
570
- "loss": 0.261,
571
  "step": 1975
572
  },
573
  {
574
- "epoch": 1.1422044545973729,
575
- "grad_norm": 2.658766984939575,
576
  "learning_rate": 6.666666666666667e-06,
577
- "loss": 0.2476,
578
  "step": 2000
579
  },
580
  {
581
- "epoch": 1.1422044545973729,
582
- "eval_loss": 0.41870468854904175,
583
- "eval_runtime": 1722.2575,
584
- "eval_samples_per_second": 2.265,
585
- "eval_steps_per_second": 0.142,
586
- "eval_wer": 0.3136771950159374,
587
  "step": 2000
588
  },
589
  {
590
- "epoch": 1.15648201027984,
591
- "grad_norm": 2.711312770843506,
592
  "learning_rate": 6.6111111111111115e-06,
593
- "loss": 0.2414,
594
  "step": 2025
595
  },
596
  {
597
- "epoch": 1.1707595659623073,
598
- "grad_norm": 2.9044759273529053,
599
  "learning_rate": 6.555555555555556e-06,
600
- "loss": 0.2502,
601
  "step": 2050
602
  },
603
  {
604
- "epoch": 1.1850371216447744,
605
- "grad_norm": 2.549725294113159,
606
  "learning_rate": 6.5000000000000004e-06,
607
- "loss": 0.2511,
608
  "step": 2075
609
  },
610
  {
611
- "epoch": 1.1993146773272416,
612
- "grad_norm": 2.95792555809021,
613
  "learning_rate": 6.444444444444445e-06,
614
- "loss": 0.2427,
615
  "step": 2100
616
  },
617
  {
618
- "epoch": 1.2135922330097086,
619
- "grad_norm": 2.686870574951172,
620
  "learning_rate": 6.3888888888888885e-06,
621
- "loss": 0.2637,
622
  "step": 2125
623
  },
624
  {
625
- "epoch": 1.227869788692176,
626
- "grad_norm": 3.7834455966949463,
627
  "learning_rate": 6.333333333333333e-06,
628
- "loss": 0.2554,
629
  "step": 2150
630
  },
631
  {
632
- "epoch": 1.2421473443746431,
633
- "grad_norm": 3.0891430377960205,
634
  "learning_rate": 6.277777777777778e-06,
635
- "loss": 0.2467,
636
  "step": 2175
637
  },
638
  {
639
- "epoch": 1.2564249000571102,
640
- "grad_norm": 2.771472930908203,
641
  "learning_rate": 6.222222222222223e-06,
642
- "loss": 0.2467,
643
  "step": 2200
644
  },
645
  {
646
- "epoch": 1.2707024557395774,
647
- "grad_norm": 2.6807925701141357,
648
  "learning_rate": 6.166666666666667e-06,
649
- "loss": 0.2682,
650
  "step": 2225
651
  },
652
  {
653
- "epoch": 1.2849800114220447,
654
- "grad_norm": 2.2320196628570557,
655
  "learning_rate": 6.111111111111112e-06,
656
- "loss": 0.2408,
657
  "step": 2250
658
  },
659
  {
660
- "epoch": 1.2992575671045117,
661
- "grad_norm": 3.066009759902954,
662
  "learning_rate": 6.055555555555555e-06,
663
- "loss": 0.2363,
664
  "step": 2275
665
  },
666
  {
667
- "epoch": 1.313535122786979,
668
- "grad_norm": 2.6043167114257812,
669
  "learning_rate": 6e-06,
670
- "loss": 0.2483,
671
  "step": 2300
672
  },
673
  {
674
- "epoch": 1.327812678469446,
675
- "grad_norm": 2.6250624656677246,
676
  "learning_rate": 5.944444444444445e-06,
677
- "loss": 0.2563,
678
  "step": 2325
679
  },
680
  {
681
- "epoch": 1.3420902341519132,
682
- "grad_norm": 2.508998394012451,
683
  "learning_rate": 5.88888888888889e-06,
684
- "loss": 0.2581,
685
  "step": 2350
686
  },
687
  {
688
- "epoch": 1.3563677898343802,
689
- "grad_norm": 2.872715473175049,
690
  "learning_rate": 5.833333333333334e-06,
691
- "loss": 0.2371,
692
  "step": 2375
693
  },
694
  {
695
- "epoch": 1.3706453455168475,
696
- "grad_norm": 3.1910557746887207,
697
- "learning_rate": 5.777777777777778e-06,
698
- "loss": 0.2515,
699
  "step": 2400
700
  },
701
  {
702
- "epoch": 1.3849229011993147,
703
- "grad_norm": 2.7466485500335693,
704
- "learning_rate": 5.722222222222222e-06,
705
- "loss": 0.2578,
706
  "step": 2425
707
  },
708
  {
709
- "epoch": 1.3992004568817817,
710
- "grad_norm": 2.388066530227661,
711
- "learning_rate": 5.666666666666667e-06,
712
- "loss": 0.2541,
713
  "step": 2450
714
  },
715
  {
716
- "epoch": 1.413478012564249,
717
- "grad_norm": 2.688497304916382,
718
- "learning_rate": 5.611111111111112e-06,
719
- "loss": 0.2514,
720
  "step": 2475
721
  },
722
  {
723
- "epoch": 1.4277555682467162,
724
- "grad_norm": 2.710899591445923,
725
- "learning_rate": 5.555555555555557e-06,
726
- "loss": 0.2765,
727
  "step": 2500
728
  },
729
  {
730
- "epoch": 1.4420331239291833,
731
- "grad_norm": 2.296635389328003,
732
- "learning_rate": 5.500000000000001e-06,
733
- "loss": 0.2487,
734
  "step": 2525
735
  },
736
  {
737
- "epoch": 1.4563106796116505,
738
- "grad_norm": 2.7988133430480957,
739
- "learning_rate": 5.444444444444445e-06,
740
- "loss": 0.2499,
741
  "step": 2550
742
  },
743
  {
744
- "epoch": 1.4705882352941178,
745
- "grad_norm": 3.1988582611083984,
746
- "learning_rate": 5.388888888888889e-06,
747
- "loss": 0.2456,
748
  "step": 2575
749
  },
750
  {
751
- "epoch": 1.4848657909765848,
752
- "grad_norm": 2.657517910003662,
753
- "learning_rate": 5.333333333333334e-06,
754
- "loss": 0.2613,
755
  "step": 2600
756
  },
757
  {
758
- "epoch": 1.499143346659052,
759
- "grad_norm": 2.5517725944519043,
760
- "learning_rate": 5.2777777777777785e-06,
761
- "loss": 0.2528,
762
  "step": 2625
763
  },
764
  {
765
- "epoch": 1.5134209023415193,
766
- "grad_norm": 2.7166850566864014,
767
- "learning_rate": 5.2222222222222226e-06,
768
- "loss": 0.2476,
769
  "step": 2650
770
  },
771
  {
772
- "epoch": 1.5276984580239863,
773
- "grad_norm": 2.7338292598724365,
774
- "learning_rate": 5.1666666666666675e-06,
775
- "loss": 0.2489,
776
  "step": 2675
777
  },
778
  {
779
- "epoch": 1.5419760137064533,
780
- "grad_norm": 2.1498470306396484,
781
- "learning_rate": 5.1111111111111115e-06,
782
- "loss": 0.2388,
783
  "step": 2700
784
  },
785
  {
786
- "epoch": 1.5562535693889206,
787
- "grad_norm": 2.595247745513916,
788
- "learning_rate": 5.0555555555555555e-06,
789
- "loss": 0.2566,
790
  "step": 2725
791
  },
792
  {
793
- "epoch": 1.5705311250713878,
794
- "grad_norm": 2.652132987976074,
795
- "learning_rate": 5e-06,
796
- "loss": 0.239,
797
  "step": 2750
798
  },
799
  {
800
- "epoch": 1.5848086807538548,
801
- "grad_norm": 2.436605930328369,
802
- "learning_rate": 4.944444444444445e-06,
803
- "loss": 0.2419,
804
  "step": 2775
805
  },
806
  {
807
- "epoch": 1.599086236436322,
808
- "grad_norm": 2.618035316467285,
809
- "learning_rate": 4.888888888888889e-06,
810
- "loss": 0.2295,
811
  "step": 2800
812
  },
813
  {
814
- "epoch": 1.6133637921187893,
815
- "grad_norm": 2.2901298999786377,
816
- "learning_rate": 4.833333333333333e-06,
817
- "loss": 0.2446,
818
  "step": 2825
819
  },
820
  {
821
- "epoch": 1.6276413478012564,
822
- "grad_norm": 2.899315595626831,
823
- "learning_rate": 4.777777777777778e-06,
824
- "loss": 0.2628,
825
  "step": 2850
826
  },
827
  {
828
- "epoch": 1.6419189034837236,
829
- "grad_norm": 2.616224527359009,
830
- "learning_rate": 4.722222222222222e-06,
831
- "loss": 0.2273,
832
  "step": 2875
833
  },
834
  {
835
- "epoch": 1.6561964591661908,
836
- "grad_norm": 2.43113112449646,
837
- "learning_rate": 4.666666666666667e-06,
838
- "loss": 0.2362,
839
  "step": 2900
840
  },
841
  {
842
- "epoch": 1.6704740148486579,
843
- "grad_norm": 2.5203065872192383,
844
- "learning_rate": 4.611111111111112e-06,
845
- "loss": 0.2428,
846
  "step": 2925
847
  },
848
  {
849
- "epoch": 1.6847515705311251,
850
- "grad_norm": 2.3064985275268555,
851
- "learning_rate": 4.555555555555556e-06,
852
- "loss": 0.2441,
853
  "step": 2950
854
  },
855
  {
856
- "epoch": 1.6990291262135924,
857
- "grad_norm": 2.201695680618286,
858
- "learning_rate": 4.5e-06,
859
- "loss": 0.2324,
860
  "step": 2975
861
  },
862
  {
863
- "epoch": 1.7133066818960594,
864
- "grad_norm": 2.442471981048584,
865
- "learning_rate": 4.444444444444444e-06,
866
- "loss": 0.2527,
867
  "step": 3000
868
  },
869
  {
870
- "epoch": 1.7133066818960594,
871
- "eval_loss": 0.3882293701171875,
872
- "eval_runtime": 1749.1422,
873
- "eval_samples_per_second": 2.23,
874
- "eval_steps_per_second": 0.139,
875
- "eval_wer": 0.2901091471071187,
876
  "step": 3000
877
  },
878
  {
879
- "epoch": 1.7275842375785264,
880
- "grad_norm": 2.77786922454834,
881
- "learning_rate": 4.388888888888889e-06,
882
- "loss": 0.2492,
883
  "step": 3025
884
  },
885
  {
886
- "epoch": 1.7418617932609937,
887
- "grad_norm": 2.5009052753448486,
888
- "learning_rate": 4.333333333333334e-06,
889
- "loss": 0.2341,
890
  "step": 3050
891
  },
892
  {
893
- "epoch": 1.756139348943461,
894
- "grad_norm": 2.780186176300049,
895
- "learning_rate": 4.277777777777778e-06,
896
- "loss": 0.2407,
897
  "step": 3075
898
  },
899
  {
900
- "epoch": 1.770416904625928,
901
- "grad_norm": 1.9574618339538574,
902
- "learning_rate": 4.222222222222223e-06,
903
- "loss": 0.2437,
904
  "step": 3100
905
  },
906
  {
907
- "epoch": 1.7846944603083952,
908
- "grad_norm": 2.151125907897949,
909
- "learning_rate": 4.166666666666667e-06,
910
- "loss": 0.2341,
911
  "step": 3125
912
  },
913
  {
914
- "epoch": 1.7989720159908624,
915
- "grad_norm": 2.170015811920166,
916
- "learning_rate": 4.111111111111111e-06,
917
- "loss": 0.2373,
918
  "step": 3150
919
  },
920
  {
921
- "epoch": 1.8132495716733295,
922
- "grad_norm": 3.0467231273651123,
923
- "learning_rate": 4.055555555555556e-06,
924
- "loss": 0.2317,
925
  "step": 3175
926
  },
927
  {
928
- "epoch": 1.8275271273557967,
929
- "grad_norm": 3.0150015354156494,
930
- "learning_rate": 4.000000000000001e-06,
931
- "loss": 0.228,
932
  "step": 3200
933
  },
934
  {
935
- "epoch": 1.841804683038264,
936
- "grad_norm": 3.275949001312256,
937
- "learning_rate": 3.944444444444445e-06,
938
- "loss": 0.2438,
939
  "step": 3225
940
  },
941
  {
942
- "epoch": 1.856082238720731,
943
- "grad_norm": 3.0381839275360107,
944
- "learning_rate": 3.88888888888889e-06,
945
- "loss": 0.2478,
946
  "step": 3250
947
  },
948
  {
949
- "epoch": 1.8703597944031982,
950
- "grad_norm": 2.770716428756714,
951
- "learning_rate": 3.833333333333334e-06,
952
- "loss": 0.2312,
953
  "step": 3275
954
  },
955
  {
956
- "epoch": 1.8846373500856655,
957
- "grad_norm": 2.6976678371429443,
958
- "learning_rate": 3.777777777777778e-06,
959
- "loss": 0.2284,
960
  "step": 3300
961
  },
962
  {
963
- "epoch": 1.8989149057681325,
964
- "grad_norm": 2.8799102306365967,
965
- "learning_rate": 3.7222222222222225e-06,
966
- "loss": 0.2484,
967
  "step": 3325
968
  },
969
  {
970
- "epoch": 1.9131924614505995,
971
- "grad_norm": 2.574629545211792,
972
- "learning_rate": 3.6666666666666666e-06,
973
- "loss": 0.2295,
974
  "step": 3350
975
  },
976
  {
977
- "epoch": 1.927470017133067,
978
- "grad_norm": 2.4746835231781006,
979
- "learning_rate": 3.6111111111111115e-06,
980
- "loss": 0.2335,
981
  "step": 3375
982
  },
983
  {
984
- "epoch": 1.941747572815534,
985
- "grad_norm": 3.084383964538574,
986
- "learning_rate": 3.555555555555556e-06,
987
- "loss": 0.212,
988
  "step": 3400
989
  },
990
  {
991
- "epoch": 1.956025128498001,
992
- "grad_norm": 2.4441068172454834,
993
- "learning_rate": 3.5e-06,
994
- "loss": 0.221,
995
  "step": 3425
996
  },
997
  {
998
- "epoch": 1.9703026841804683,
999
- "grad_norm": 3.031568765640259,
1000
- "learning_rate": 3.444444444444445e-06,
1001
- "loss": 0.2341,
1002
  "step": 3450
1003
  },
1004
  {
1005
- "epoch": 1.9845802398629355,
1006
- "grad_norm": 2.3584327697753906,
1007
- "learning_rate": 3.3888888888888893e-06,
1008
- "loss": 0.2431,
1009
  "step": 3475
1010
  },
1011
  {
1012
- "epoch": 1.9988577955454025,
1013
- "grad_norm": 2.1590421199798584,
1014
- "learning_rate": 3.3333333333333333e-06,
1015
- "loss": 0.2357,
1016
  "step": 3500
1017
  },
1018
  {
1019
- "epoch": 2.0131353512278696,
1020
- "grad_norm": 2.2845587730407715,
1021
- "learning_rate": 3.277777777777778e-06,
1022
- "loss": 0.1576,
1023
  "step": 3525
1024
  },
1025
  {
1026
- "epoch": 2.027412906910337,
1027
- "grad_norm": 2.033133029937744,
1028
- "learning_rate": 3.2222222222222227e-06,
1029
- "loss": 0.1422,
1030
  "step": 3550
1031
  },
1032
  {
1033
- "epoch": 2.041690462592804,
1034
- "grad_norm": 2.2549259662628174,
1035
- "learning_rate": 3.1666666666666667e-06,
1036
- "loss": 0.1473,
1037
  "step": 3575
1038
  },
1039
  {
1040
- "epoch": 2.055968018275271,
1041
- "grad_norm": 1.5837754011154175,
1042
- "learning_rate": 3.1111111111111116e-06,
1043
- "loss": 0.143,
1044
  "step": 3600
1045
  },
1046
  {
1047
- "epoch": 2.0702455739577386,
1048
- "grad_norm": 1.9988360404968262,
1049
- "learning_rate": 3.055555555555556e-06,
1050
- "loss": 0.1416,
1051
  "step": 3625
1052
  },
1053
  {
1054
- "epoch": 2.0845231296402056,
1055
- "grad_norm": 2.148613929748535,
1056
- "learning_rate": 3e-06,
1057
- "loss": 0.1338,
1058
  "step": 3650
1059
  },
1060
  {
1061
- "epoch": 2.0988006853226726,
1062
- "grad_norm": 1.8176393508911133,
1063
- "learning_rate": 2.944444444444445e-06,
1064
- "loss": 0.1514,
1065
  "step": 3675
1066
  },
1067
  {
1068
- "epoch": 2.11307824100514,
1069
- "grad_norm": 2.60271954536438,
1070
- "learning_rate": 2.888888888888889e-06,
1071
- "loss": 0.1533,
1072
  "step": 3700
1073
  },
1074
  {
1075
- "epoch": 2.127355796687607,
1076
- "grad_norm": 2.120281457901001,
1077
- "learning_rate": 2.8333333333333335e-06,
1078
- "loss": 0.1404,
1079
  "step": 3725
1080
  },
1081
  {
1082
- "epoch": 2.141633352370074,
1083
- "grad_norm": 2.3522286415100098,
1084
- "learning_rate": 2.7777777777777783e-06,
1085
- "loss": 0.1511,
1086
  "step": 3750
1087
  },
1088
  {
1089
- "epoch": 2.1559109080525416,
1090
- "grad_norm": 1.8738924264907837,
1091
- "learning_rate": 2.7222222222222224e-06,
1092
- "loss": 0.1417,
1093
  "step": 3775
1094
  },
1095
  {
1096
- "epoch": 2.1701884637350086,
1097
- "grad_norm": 2.255291223526001,
1098
- "learning_rate": 2.666666666666667e-06,
1099
- "loss": 0.1437,
1100
  "step": 3800
1101
  },
1102
  {
1103
- "epoch": 2.1844660194174756,
1104
- "grad_norm": 1.7046154737472534,
1105
- "learning_rate": 2.6111111111111113e-06,
1106
- "loss": 0.1446,
1107
  "step": 3825
1108
  },
1109
  {
1110
- "epoch": 2.198743575099943,
1111
- "grad_norm": 2.0543861389160156,
1112
- "learning_rate": 2.5555555555555557e-06,
1113
- "loss": 0.1504,
1114
  "step": 3850
1115
  },
1116
  {
1117
- "epoch": 2.21302113078241,
1118
- "grad_norm": 2.139716863632202,
1119
- "learning_rate": 2.5e-06,
1120
- "loss": 0.1345,
1121
  "step": 3875
1122
  },
1123
  {
1124
- "epoch": 2.227298686464877,
1125
- "grad_norm": 1.7999951839447021,
1126
- "learning_rate": 2.4444444444444447e-06,
1127
- "loss": 0.1389,
1128
  "step": 3900
1129
  },
1130
  {
1131
- "epoch": 2.241576242147344,
1132
- "grad_norm": 1.7282090187072754,
1133
- "learning_rate": 2.388888888888889e-06,
1134
- "loss": 0.1324,
1135
  "step": 3925
1136
  },
1137
  {
1138
- "epoch": 2.2558537978298117,
1139
- "grad_norm": 2.6271605491638184,
1140
- "learning_rate": 2.3333333333333336e-06,
1141
- "loss": 0.1551,
1142
  "step": 3950
1143
  },
1144
  {
1145
- "epoch": 2.2701313535122787,
1146
- "grad_norm": 2.170382022857666,
1147
- "learning_rate": 2.277777777777778e-06,
1148
- "loss": 0.144,
1149
  "step": 3975
1150
  },
1151
  {
1152
- "epoch": 2.2844089091947457,
1153
- "grad_norm": 1.796635627746582,
1154
- "learning_rate": 2.222222222222222e-06,
1155
- "loss": 0.1568,
1156
  "step": 4000
1157
  },
1158
  {
1159
- "epoch": 2.2844089091947457,
1160
- "eval_loss": 0.3901652991771698,
1161
- "eval_runtime": 1765.3609,
1162
- "eval_samples_per_second": 2.21,
1163
- "eval_steps_per_second": 0.138,
1164
- "eval_wer": 0.28160919540229884,
1165
  "step": 4000
1166
  },
1167
  {
1168
- "epoch": 2.298686464877213,
1169
- "grad_norm": 2.0357980728149414,
1170
- "learning_rate": 2.166666666666667e-06,
1171
- "loss": 0.161,
1172
  "step": 4025
1173
  },
1174
  {
1175
- "epoch": 2.31296402055968,
1176
- "grad_norm": 2.027215003967285,
1177
- "learning_rate": 2.1111111111111114e-06,
1178
- "loss": 0.1353,
1179
  "step": 4050
1180
  },
1181
  {
1182
- "epoch": 2.3272415762421472,
1183
- "grad_norm": 2.8169405460357666,
1184
- "learning_rate": 2.0555555555555555e-06,
1185
- "loss": 0.1449,
1186
  "step": 4075
1187
  },
1188
  {
1189
- "epoch": 2.3415191319246147,
1190
- "grad_norm": 1.9528751373291016,
1191
- "learning_rate": 2.0000000000000003e-06,
1192
- "loss": 0.1376,
1193
  "step": 4100
1194
  },
1195
  {
1196
- "epoch": 2.3557966876070817,
1197
- "grad_norm": 2.5781335830688477,
1198
- "learning_rate": 1.944444444444445e-06,
1199
- "loss": 0.1383,
1200
  "step": 4125
1201
  },
1202
  {
1203
- "epoch": 2.3700742432895487,
1204
- "grad_norm": 2.083077907562256,
1205
- "learning_rate": 1.888888888888889e-06,
1206
- "loss": 0.1362,
1207
  "step": 4150
1208
  },
1209
  {
1210
- "epoch": 2.384351798972016,
1211
- "grad_norm": 2.431272029876709,
1212
- "learning_rate": 1.8333333333333333e-06,
1213
- "loss": 0.1329,
1214
  "step": 4175
1215
  },
1216
  {
1217
- "epoch": 2.3986293546544832,
1218
- "grad_norm": 2.157139539718628,
1219
- "learning_rate": 1.777777777777778e-06,
1220
- "loss": 0.1377,
1221
  "step": 4200
1222
  },
1223
  {
1224
- "epoch": 2.4129069103369503,
1225
- "grad_norm": 2.5328071117401123,
1226
- "learning_rate": 1.7222222222222224e-06,
1227
- "loss": 0.1361,
1228
  "step": 4225
1229
  },
1230
  {
1231
- "epoch": 2.4271844660194173,
1232
- "grad_norm": 2.433239459991455,
1233
- "learning_rate": 1.6666666666666667e-06,
1234
- "loss": 0.157,
1235
  "step": 4250
1236
  },
1237
  {
1238
- "epoch": 2.4414620217018848,
1239
- "grad_norm": 2.5167510509490967,
1240
- "learning_rate": 1.6111111111111113e-06,
1241
- "loss": 0.132,
1242
  "step": 4275
1243
  },
1244
  {
1245
- "epoch": 2.455739577384352,
1246
- "grad_norm": 1.9507442712783813,
1247
- "learning_rate": 1.5555555555555558e-06,
1248
- "loss": 0.1625,
1249
  "step": 4300
1250
  },
1251
  {
1252
- "epoch": 2.470017133066819,
1253
- "grad_norm": 2.2467007637023926,
1254
- "learning_rate": 1.5e-06,
1255
- "loss": 0.1333,
1256
  "step": 4325
1257
  },
1258
  {
1259
- "epoch": 2.4842946887492863,
1260
- "grad_norm": 2.4816768169403076,
1261
- "learning_rate": 1.4444444444444445e-06,
1262
- "loss": 0.1499,
1263
  "step": 4350
1264
  },
1265
  {
1266
- "epoch": 2.4985722444317533,
1267
- "grad_norm": 2.0616416931152344,
1268
- "learning_rate": 1.3888888888888892e-06,
1269
- "loss": 0.1508,
1270
  "step": 4375
1271
  },
1272
  {
1273
- "epoch": 2.5128498001142203,
1274
- "grad_norm": 2.089355230331421,
1275
- "learning_rate": 1.3333333333333334e-06,
1276
- "loss": 0.1344,
1277
  "step": 4400
1278
  },
1279
  {
1280
- "epoch": 2.5271273557966873,
1281
- "grad_norm": 2.2235498428344727,
1282
  "learning_rate": 1.28e-06,
1283
- "loss": 0.1717,
1284
  "step": 4425
1285
  },
1286
  {
1287
- "epoch": 2.541404911479155,
1288
- "grad_norm": 1.9268138408660889,
1289
  "learning_rate": 1.2244444444444445e-06,
1290
- "loss": 0.143,
1291
  "step": 4450
1292
  },
1293
  {
1294
- "epoch": 2.555682467161622,
1295
- "grad_norm": 1.8911551237106323,
1296
  "learning_rate": 1.168888888888889e-06,
1297
- "loss": 0.1439,
1298
  "step": 4475
1299
  },
1300
  {
1301
- "epoch": 2.5699600228440893,
1302
- "grad_norm": 2.5078868865966797,
1303
- "learning_rate": 1.1133333333333334e-06,
1304
- "loss": 0.1341,
1305
  "step": 4500
1306
  },
1307
  {
1308
- "epoch": 2.5842375785265563,
1309
- "grad_norm": 2.1232492923736572,
1310
- "learning_rate": 1.0577777777777779e-06,
1311
- "loss": 0.1415,
1312
  "step": 4525
1313
  },
1314
  {
1315
- "epoch": 2.5985151342090234,
1316
- "grad_norm": 1.9214311838150024,
1317
- "learning_rate": 1.0022222222222223e-06,
1318
- "loss": 0.1301,
1319
  "step": 4550
1320
  },
1321
  {
1322
- "epoch": 2.6127926898914904,
1323
- "grad_norm": 2.4226858615875244,
1324
- "learning_rate": 9.466666666666667e-07,
1325
- "loss": 0.1438,
1326
  "step": 4575
1327
  },
1328
  {
1329
- "epoch": 2.627070245573958,
1330
- "grad_norm": 2.324777126312256,
1331
- "learning_rate": 8.911111111111112e-07,
1332
- "loss": 0.1306,
1333
  "step": 4600
1334
  },
1335
  {
1336
- "epoch": 2.641347801256425,
1337
- "grad_norm": 2.427114486694336,
1338
- "learning_rate": 8.355555555555556e-07,
1339
- "loss": 0.1359,
1340
  "step": 4625
1341
  },
1342
  {
1343
- "epoch": 2.655625356938892,
1344
- "grad_norm": 1.989882469177246,
1345
- "learning_rate": 7.8e-07,
1346
- "loss": 0.1386,
1347
  "step": 4650
1348
  },
1349
  {
1350
- "epoch": 2.6699029126213594,
1351
- "grad_norm": 2.6079118251800537,
1352
- "learning_rate": 7.244444444444446e-07,
1353
- "loss": 0.135,
1354
  "step": 4675
1355
  },
1356
  {
1357
- "epoch": 2.6841804683038264,
1358
- "grad_norm": 2.3429243564605713,
1359
- "learning_rate": 6.68888888888889e-07,
1360
- "loss": 0.1356,
1361
  "step": 4700
1362
  },
1363
  {
1364
- "epoch": 2.6984580239862934,
1365
- "grad_norm": 2.3358540534973145,
1366
- "learning_rate": 6.133333333333333e-07,
1367
- "loss": 0.1304,
1368
  "step": 4725
1369
  },
1370
  {
1371
- "epoch": 2.7127355796687604,
1372
- "grad_norm": 1.917809247970581,
1373
- "learning_rate": 5.577777777777779e-07,
1374
- "loss": 0.1395,
1375
  "step": 4750
1376
  },
1377
  {
1378
- "epoch": 2.727013135351228,
1379
- "grad_norm": 2.0677952766418457,
1380
- "learning_rate": 5.022222222222222e-07,
1381
- "loss": 0.1309,
1382
  "step": 4775
1383
  },
1384
  {
1385
- "epoch": 2.741290691033695,
1386
- "grad_norm": 2.135127305984497,
1387
- "learning_rate": 4.466666666666667e-07,
1388
- "loss": 0.1424,
1389
  "step": 4800
1390
  },
1391
  {
1392
- "epoch": 2.7555682467161624,
1393
- "grad_norm": 2.3306682109832764,
1394
- "learning_rate": 3.9111111111111115e-07,
1395
- "loss": 0.1318,
1396
  "step": 4825
1397
  },
1398
  {
1399
- "epoch": 2.7698458023986294,
1400
- "grad_norm": 2.0700454711914062,
1401
- "learning_rate": 3.3555555555555556e-07,
1402
- "loss": 0.1566,
1403
  "step": 4850
1404
  },
1405
  {
1406
- "epoch": 2.7841233580810965,
1407
- "grad_norm": 1.8561683893203735,
1408
- "learning_rate": 2.8e-07,
1409
- "loss": 0.1453,
1410
  "step": 4875
1411
  },
1412
  {
1413
- "epoch": 2.7984009137635635,
1414
- "grad_norm": 2.2682347297668457,
1415
- "learning_rate": 2.2444444444444445e-07,
1416
- "loss": 0.1415,
1417
  "step": 4900
1418
  },
1419
  {
1420
- "epoch": 2.812678469446031,
1421
- "grad_norm": 2.2898778915405273,
1422
- "learning_rate": 1.6888888888888888e-07,
1423
- "loss": 0.1427,
1424
  "step": 4925
1425
  },
1426
  {
1427
- "epoch": 2.826956025128498,
1428
- "grad_norm": 2.328401803970337,
1429
- "learning_rate": 1.1333333333333336e-07,
1430
- "loss": 0.1357,
1431
  "step": 4950
1432
  },
1433
  {
1434
- "epoch": 2.841233580810965,
1435
- "grad_norm": 2.2169013023376465,
1436
- "learning_rate": 5.777777777777778e-08,
1437
- "loss": 0.1343,
1438
  "step": 4975
1439
  },
1440
  {
1441
- "epoch": 2.8555111364934325,
1442
- "grad_norm": 2.42340350151062,
1443
- "learning_rate": 2.2222222222222225e-09,
1444
- "loss": 0.1313,
1445
  "step": 5000
1446
  },
1447
  {
1448
- "epoch": 2.8555111364934325,
1449
- "eval_loss": 0.38383349776268005,
1450
- "eval_runtime": 1820.062,
1451
- "eval_samples_per_second": 2.143,
1452
- "eval_steps_per_second": 0.134,
1453
- "eval_wer": 0.27318168646769053,
1454
  "step": 5000
1455
  },
1456
  {
1457
- "epoch": 2.8555111364934325,
1458
  "step": 5000,
1459
- "total_flos": 5.435589590699213e+20,
1460
- "train_loss": 0.3002769865989685,
1461
- "train_runtime": 59305.2217,
1462
- "train_samples_per_second": 2.698,
1463
- "train_steps_per_second": 0.084
1464
  }
1465
  ],
1466
  "logging_steps": 25,
1467
  "max_steps": 5000,
1468
  "num_input_tokens_seen": 0,
1469
- "num_train_epochs": 3,
1470
  "save_steps": 1000,
1471
  "stateful_callbacks": {
1472
  "TrainerControl": {
@@ -1480,7 +1480,7 @@
1480
  "attributes": {}
1481
  }
1482
  },
1483
- "total_flos": 5.435589590699213e+20,
1484
  "train_batch_size": 16,
1485
  "trial_name": null,
1486
  "trial_params": null
 
1
  {
2
  "best_metric": null,
3
  "best_model_checkpoint": null,
4
+ "epoch": 1.619795885306982,
5
  "eval_steps": 1000,
6
  "global_step": 5000,
7
  "is_hyper_param_search": false,
 
9
  "is_world_process_zero": true,
10
  "log_history": [
11
  {
12
+ "epoch": 0.008099789405475458,
13
+ "grad_norm": 8.217827796936035,
14
  "learning_rate": 5.000000000000001e-07,
15
+ "loss": 1.6162,
16
  "step": 25
17
  },
18
  {
19
+ "epoch": 0.016199578810950917,
20
+ "grad_norm": 5.221391677856445,
21
  "learning_rate": 1.0000000000000002e-06,
22
+ "loss": 1.3426,
23
  "step": 50
24
  },
25
  {
26
+ "epoch": 0.02429936821642637,
27
+ "grad_norm": 6.4570441246032715,
28
  "learning_rate": 1.5e-06,
29
+ "loss": 1.0591,
30
  "step": 75
31
  },
32
  {
33
+ "epoch": 0.03239915762190183,
34
+ "grad_norm": 4.746372699737549,
35
  "learning_rate": 2.0000000000000003e-06,
36
+ "loss": 0.9419,
37
  "step": 100
38
  },
39
  {
40
+ "epoch": 0.04049894702737729,
41
+ "grad_norm": 5.625121593475342,
42
  "learning_rate": 2.5e-06,
43
+ "loss": 0.8389,
44
  "step": 125
45
  },
46
  {
47
+ "epoch": 0.04859873643285274,
48
+ "grad_norm": 6.477604866027832,
49
  "learning_rate": 3e-06,
50
+ "loss": 0.8019,
51
  "step": 150
52
  },
53
  {
54
+ "epoch": 0.056698525838328205,
55
+ "grad_norm": 4.935534954071045,
56
  "learning_rate": 3.5e-06,
57
+ "loss": 0.8007,
58
  "step": 175
59
  },
60
  {
61
+ "epoch": 0.06479831524380367,
62
+ "grad_norm": 5.263591289520264,
63
  "learning_rate": 4.000000000000001e-06,
64
+ "loss": 0.7419,
65
  "step": 200
66
  },
67
  {
68
+ "epoch": 0.07289810464927912,
69
+ "grad_norm": 5.378949165344238,
70
  "learning_rate": 4.5e-06,
71
+ "loss": 0.7391,
72
  "step": 225
73
  },
74
  {
75
+ "epoch": 0.08099789405475458,
76
+ "grad_norm": 5.626790523529053,
77
  "learning_rate": 5e-06,
78
+ "loss": 0.7243,
79
  "step": 250
80
  },
81
  {
82
+ "epoch": 0.08909768346023003,
83
+ "grad_norm": 5.457970142364502,
84
  "learning_rate": 5.500000000000001e-06,
85
+ "loss": 0.7273,
86
  "step": 275
87
  },
88
  {
89
+ "epoch": 0.09719747286570549,
90
+ "grad_norm": 4.948482036590576,
91
  "learning_rate": 6e-06,
92
+ "loss": 0.7105,
93
  "step": 300
94
  },
95
  {
96
+ "epoch": 0.10529726227118096,
97
+ "grad_norm": 4.352142333984375,
98
  "learning_rate": 6.5000000000000004e-06,
99
+ "loss": 0.7075,
100
  "step": 325
101
  },
102
  {
103
+ "epoch": 0.11339705167665641,
104
+ "grad_norm": 4.84460973739624,
105
  "learning_rate": 7e-06,
106
+ "loss": 0.6808,
107
  "step": 350
108
  },
109
  {
110
+ "epoch": 0.12149684108213187,
111
+ "grad_norm": 5.333044052124023,
112
  "learning_rate": 7.500000000000001e-06,
113
+ "loss": 0.6855,
114
  "step": 375
115
  },
116
  {
117
+ "epoch": 0.12959663048760733,
118
+ "grad_norm": 5.835122585296631,
119
  "learning_rate": 8.000000000000001e-06,
120
+ "loss": 0.642,
121
  "step": 400
122
  },
123
  {
124
+ "epoch": 0.1376964198930828,
125
+ "grad_norm": 6.060296058654785,
126
  "learning_rate": 8.5e-06,
127
+ "loss": 0.6443,
128
  "step": 425
129
  },
130
  {
131
+ "epoch": 0.14579620929855824,
132
+ "grad_norm": 4.375275135040283,
133
  "learning_rate": 9e-06,
134
+ "loss": 0.6477,
135
  "step": 450
136
  },
137
  {
138
+ "epoch": 0.1538959987040337,
139
+ "grad_norm": 4.223667144775391,
140
  "learning_rate": 9.5e-06,
141
+ "loss": 0.6369,
142
  "step": 475
143
  },
144
  {
145
+ "epoch": 0.16199578810950915,
146
+ "grad_norm": 10.034436225891113,
147
  "learning_rate": 1e-05,
148
+ "loss": 0.6853,
149
  "step": 500
150
  },
151
  {
152
+ "epoch": 0.1700955775149846,
153
+ "grad_norm": 4.590435028076172,
154
  "learning_rate": 9.944444444444445e-06,
155
+ "loss": 0.6149,
156
  "step": 525
157
  },
158
  {
159
+ "epoch": 0.17819536692046006,
160
+ "grad_norm": 4.104828357696533,
161
  "learning_rate": 9.88888888888889e-06,
162
+ "loss": 0.6267,
163
  "step": 550
164
  },
165
  {
166
+ "epoch": 0.18629515632593552,
167
+ "grad_norm": 3.9677858352661133,
168
  "learning_rate": 9.833333333333333e-06,
169
+ "loss": 0.5788,
170
  "step": 575
171
  },
172
  {
173
+ "epoch": 0.19439494573141097,
174
+ "grad_norm": 3.9867312908172607,
175
  "learning_rate": 9.777777777777779e-06,
176
+ "loss": 0.616,
177
  "step": 600
178
  },
179
  {
180
+ "epoch": 0.20249473513688643,
181
+ "grad_norm": 4.993474960327148,
182
  "learning_rate": 9.722222222222223e-06,
183
+ "loss": 0.6151,
184
  "step": 625
185
  },
186
  {
187
+ "epoch": 0.2105945245423619,
188
+ "grad_norm": 4.049670219421387,
189
  "learning_rate": 9.666666666666667e-06,
190
+ "loss": 0.5898,
191
  "step": 650
192
  },
193
  {
194
+ "epoch": 0.21869431394783737,
195
+ "grad_norm": 3.711517095565796,
196
  "learning_rate": 9.611111111111112e-06,
197
+ "loss": 0.5654,
198
  "step": 675
199
  },
200
  {
201
+ "epoch": 0.22679410335331282,
202
+ "grad_norm": 4.081414222717285,
203
  "learning_rate": 9.555555555555556e-06,
204
+ "loss": 0.5608,
205
  "step": 700
206
  },
207
  {
208
+ "epoch": 0.23489389275878828,
209
+ "grad_norm": 4.10880708694458,
210
  "learning_rate": 9.5e-06,
211
+ "loss": 0.5576,
212
  "step": 725
213
  },
214
  {
215
+ "epoch": 0.24299368216426373,
216
+ "grad_norm": 3.764904737472534,
217
  "learning_rate": 9.444444444444445e-06,
218
+ "loss": 0.5249,
219
  "step": 750
220
  },
221
  {
222
+ "epoch": 0.2510934715697392,
223
+ "grad_norm": 3.7500803470611572,
224
  "learning_rate": 9.38888888888889e-06,
225
+ "loss": 0.5724,
226
  "step": 775
227
  },
228
  {
229
+ "epoch": 0.25919326097521467,
230
+ "grad_norm": 4.319125175476074,
231
  "learning_rate": 9.333333333333334e-06,
232
+ "loss": 0.5368,
233
  "step": 800
234
  },
235
  {
236
+ "epoch": 0.2672930503806901,
237
+ "grad_norm": 3.590097665786743,
238
  "learning_rate": 9.277777777777778e-06,
239
+ "loss": 0.5494,
240
  "step": 825
241
  },
242
  {
243
+ "epoch": 0.2753928397861656,
244
+ "grad_norm": 5.081935405731201,
245
  "learning_rate": 9.222222222222224e-06,
246
+ "loss": 0.5443,
247
  "step": 850
248
  },
249
  {
250
+ "epoch": 0.28349262919164103,
251
+ "grad_norm": 3.8603386878967285,
252
  "learning_rate": 9.166666666666666e-06,
253
+ "loss": 0.5378,
254
  "step": 875
255
  },
256
  {
257
+ "epoch": 0.2915924185971165,
258
+ "grad_norm": 3.7635509967803955,
259
  "learning_rate": 9.111111111111112e-06,
260
+ "loss": 0.5121,
261
  "step": 900
262
  },
263
  {
264
+ "epoch": 0.29969220800259194,
265
+ "grad_norm": 4.4741692543029785,
266
  "learning_rate": 9.055555555555556e-06,
267
+ "loss": 0.5313,
268
  "step": 925
269
  },
270
  {
271
+ "epoch": 0.3077919974080674,
272
+ "grad_norm": 4.126248836517334,
273
  "learning_rate": 9e-06,
274
+ "loss": 0.5036,
275
  "step": 950
276
  },
277
  {
278
+ "epoch": 0.31589178681354285,
279
+ "grad_norm": 3.2901930809020996,
280
  "learning_rate": 8.944444444444446e-06,
281
+ "loss": 0.537,
282
  "step": 975
283
  },
284
  {
285
+ "epoch": 0.3239915762190183,
286
+ "grad_norm": 4.152955532073975,
287
  "learning_rate": 8.888888888888888e-06,
288
+ "loss": 0.5304,
289
  "step": 1000
290
  },
291
  {
292
+ "epoch": 0.3239915762190183,
293
+ "eval_loss": 0.5235934853553772,
294
+ "eval_runtime": 2073.7226,
295
+ "eval_samples_per_second": 1.879,
296
+ "eval_steps_per_second": 0.118,
297
+ "eval_wer": 0.3785823799976412,
298
  "step": 1000
299
  },
300
  {
301
+ "epoch": 0.33209136562449376,
302
+ "grad_norm": 3.5854649543762207,
303
  "learning_rate": 8.833333333333334e-06,
304
+ "loss": 0.5265,
305
  "step": 1025
306
  },
307
  {
308
+ "epoch": 0.3401911550299692,
309
+ "grad_norm": 3.5874733924865723,
310
  "learning_rate": 8.777777777777778e-06,
311
+ "loss": 0.4957,
312
  "step": 1050
313
  },
314
  {
315
+ "epoch": 0.34829094443544467,
316
+ "grad_norm": 4.2191338539123535,
317
  "learning_rate": 8.722222222222224e-06,
318
+ "loss": 0.4892,
319
  "step": 1075
320
  },
321
  {
322
+ "epoch": 0.3563907338409201,
323
+ "grad_norm": 4.080557823181152,
324
  "learning_rate": 8.666666666666668e-06,
325
+ "loss": 0.4992,
326
  "step": 1100
327
  },
328
  {
329
+ "epoch": 0.3644905232463956,
330
+ "grad_norm": 3.3749568462371826,
331
  "learning_rate": 8.611111111111112e-06,
332
+ "loss": 0.4999,
333
  "step": 1125
334
  },
335
  {
336
+ "epoch": 0.37259031265187104,
337
+ "grad_norm": 4.65897798538208,
338
  "learning_rate": 8.555555555555556e-06,
339
+ "loss": 0.4893,
340
  "step": 1150
341
  },
342
  {
343
+ "epoch": 0.3806901020573465,
344
+ "grad_norm": 4.345097541809082,
345
  "learning_rate": 8.5e-06,
346
+ "loss": 0.5078,
347
  "step": 1175
348
  },
349
  {
350
+ "epoch": 0.38878989146282195,
351
+ "grad_norm": 3.885808229446411,
352
  "learning_rate": 8.444444444444446e-06,
353
+ "loss": 0.4683,
354
  "step": 1200
355
  },
356
  {
357
+ "epoch": 0.3968896808682974,
358
+ "grad_norm": 3.44722318649292,
359
  "learning_rate": 8.38888888888889e-06,
360
+ "loss": 0.488,
361
  "step": 1225
362
  },
363
  {
364
+ "epoch": 0.40498947027377286,
365
+ "grad_norm": 3.6947743892669678,
366
  "learning_rate": 8.333333333333334e-06,
367
+ "loss": 0.4743,
368
  "step": 1250
369
  },
370
  {
371
+ "epoch": 0.41308925967924837,
372
+ "grad_norm": 4.338139533996582,
373
  "learning_rate": 8.277777777777778e-06,
374
+ "loss": 0.4913,
375
  "step": 1275
376
  },
377
  {
378
+ "epoch": 0.4211890490847238,
379
+ "grad_norm": 3.298166513442993,
380
  "learning_rate": 8.222222222222222e-06,
381
+ "loss": 0.4865,
382
  "step": 1300
383
  },
384
  {
385
+ "epoch": 0.4292888384901993,
386
+ "grad_norm": 3.604741334915161,
387
  "learning_rate": 8.166666666666668e-06,
388
+ "loss": 0.4826,
389
  "step": 1325
390
  },
391
  {
392
+ "epoch": 0.43738862789567473,
393
+ "grad_norm": 3.3499221801757812,
394
  "learning_rate": 8.111111111111112e-06,
395
+ "loss": 0.4722,
396
  "step": 1350
397
  },
398
  {
399
+ "epoch": 0.4454884173011502,
400
+ "grad_norm": 3.690180778503418,
401
  "learning_rate": 8.055555555555557e-06,
402
+ "loss": 0.4911,
403
  "step": 1375
404
  },
405
  {
406
+ "epoch": 0.45358820670662564,
407
+ "grad_norm": 2.948256492614746,
408
  "learning_rate": 8.000000000000001e-06,
409
+ "loss": 0.4757,
410
  "step": 1400
411
  },
412
  {
413
+ "epoch": 0.4616879961121011,
414
+ "grad_norm": 3.8048160076141357,
415
  "learning_rate": 7.944444444444445e-06,
416
+ "loss": 0.4637,
417
  "step": 1425
418
  },
419
  {
420
+ "epoch": 0.46978778551757655,
421
+ "grad_norm": 3.5490314960479736,
422
  "learning_rate": 7.88888888888889e-06,
423
+ "loss": 0.4889,
424
  "step": 1450
425
  },
426
  {
427
+ "epoch": 0.477887574923052,
428
+ "grad_norm": 5.1668009757995605,
429
  "learning_rate": 7.833333333333333e-06,
430
+ "loss": 0.4595,
431
  "step": 1475
432
  },
433
  {
434
+ "epoch": 0.48598736432852746,
435
+ "grad_norm": 3.1844429969787598,
436
  "learning_rate": 7.77777777777778e-06,
437
+ "loss": 0.4799,
438
  "step": 1500
439
  },
440
  {
441
+ "epoch": 0.4940871537340029,
442
+ "grad_norm": 3.055802583694458,
443
  "learning_rate": 7.722222222222223e-06,
444
+ "loss": 0.4701,
445
  "step": 1525
446
  },
447
  {
448
+ "epoch": 0.5021869431394784,
449
+ "grad_norm": 3.7342355251312256,
450
  "learning_rate": 7.666666666666667e-06,
451
+ "loss": 0.4866,
452
  "step": 1550
453
  },
454
  {
455
+ "epoch": 0.5102867325449538,
456
+ "grad_norm": 3.3651347160339355,
457
  "learning_rate": 7.611111111111111e-06,
458
+ "loss": 0.4462,
459
  "step": 1575
460
  },
461
  {
462
+ "epoch": 0.5183865219504293,
463
+ "grad_norm": 3.7382657527923584,
464
  "learning_rate": 7.555555555555556e-06,
465
+ "loss": 0.4491,
466
  "step": 1600
467
  },
468
  {
469
+ "epoch": 0.5264863113559047,
470
+ "grad_norm": 3.2548789978027344,
471
  "learning_rate": 7.500000000000001e-06,
472
+ "loss": 0.4604,
473
  "step": 1625
474
  },
475
  {
476
+ "epoch": 0.5345861007613802,
477
+ "grad_norm": 3.0430402755737305,
478
  "learning_rate": 7.444444444444445e-06,
479
+ "loss": 0.4354,
480
  "step": 1650
481
  },
482
  {
483
+ "epoch": 0.5426858901668556,
484
+ "grad_norm": 3.766554832458496,
485
  "learning_rate": 7.38888888888889e-06,
486
+ "loss": 0.465,
487
  "step": 1675
488
  },
489
  {
490
+ "epoch": 0.5507856795723312,
491
+ "grad_norm": 3.198700428009033,
492
  "learning_rate": 7.333333333333333e-06,
493
+ "loss": 0.4695,
494
  "step": 1700
495
  },
496
  {
497
+ "epoch": 0.5588854689778066,
498
+ "grad_norm": 2.7915539741516113,
499
  "learning_rate": 7.277777777777778e-06,
500
+ "loss": 0.4511,
501
  "step": 1725
502
  },
503
  {
504
+ "epoch": 0.5669852583832821,
505
+ "grad_norm": 3.3938117027282715,
506
  "learning_rate": 7.222222222222223e-06,
507
+ "loss": 0.4266,
508
  "step": 1750
509
  },
510
  {
511
+ "epoch": 0.5750850477887575,
512
+ "grad_norm": 3.4623613357543945,
513
  "learning_rate": 7.166666666666667e-06,
514
+ "loss": 0.4483,
515
  "step": 1775
516
  },
517
  {
518
+ "epoch": 0.583184837194233,
519
+ "grad_norm": 2.7853918075561523,
520
  "learning_rate": 7.111111111111112e-06,
521
+ "loss": 0.4197,
522
  "step": 1800
523
  },
524
  {
525
+ "epoch": 0.5912846265997084,
526
+ "grad_norm": 3.4827404022216797,
527
  "learning_rate": 7.055555555555557e-06,
528
+ "loss": 0.4319,
529
  "step": 1825
530
  },
531
  {
532
+ "epoch": 0.5993844160051839,
533
+ "grad_norm": 3.458853244781494,
534
  "learning_rate": 7e-06,
535
+ "loss": 0.4301,
536
  "step": 1850
537
  },
538
  {
539
+ "epoch": 0.6074842054106593,
540
+ "grad_norm": 3.726768970489502,
541
  "learning_rate": 6.944444444444445e-06,
542
+ "loss": 0.4339,
543
  "step": 1875
544
  },
545
  {
546
+ "epoch": 0.6155839948161348,
547
+ "grad_norm": 3.7597055435180664,
548
  "learning_rate": 6.88888888888889e-06,
549
+ "loss": 0.4265,
550
  "step": 1900
551
  },
552
  {
553
+ "epoch": 0.6236837842216102,
554
+ "grad_norm": 3.6740312576293945,
555
  "learning_rate": 6.833333333333334e-06,
556
+ "loss": 0.4302,
557
  "step": 1925
558
  },
559
  {
560
+ "epoch": 0.6317835736270857,
561
+ "grad_norm": 3.0535056591033936,
562
  "learning_rate": 6.777777777777779e-06,
563
+ "loss": 0.4336,
564
  "step": 1950
565
  },
566
  {
567
+ "epoch": 0.6398833630325611,
568
+ "grad_norm": 3.295497417449951,
569
  "learning_rate": 6.7222222222222235e-06,
570
+ "loss": 0.4526,
571
  "step": 1975
572
  },
573
  {
574
+ "epoch": 0.6479831524380366,
575
+ "grad_norm": 3.080634832382202,
576
  "learning_rate": 6.666666666666667e-06,
577
+ "loss": 0.4409,
578
  "step": 2000
579
  },
580
  {
581
+ "epoch": 0.6479831524380366,
582
+ "eval_loss": 0.4457571506500244,
583
+ "eval_runtime": 1994.8072,
584
+ "eval_samples_per_second": 1.954,
585
+ "eval_steps_per_second": 0.122,
586
+ "eval_wer": 0.35629201556787354,
587
  "step": 2000
588
  },
589
  {
590
+ "epoch": 0.6560829418435121,
591
+ "grad_norm": 3.1193337440490723,
592
  "learning_rate": 6.6111111111111115e-06,
593
+ "loss": 0.4111,
594
  "step": 2025
595
  },
596
  {
597
+ "epoch": 0.6641827312489875,
598
+ "grad_norm": 3.554410934448242,
599
  "learning_rate": 6.555555555555556e-06,
600
+ "loss": 0.4192,
601
  "step": 2050
602
  },
603
  {
604
+ "epoch": 0.672282520654463,
605
+ "grad_norm": 4.468347549438477,
606
  "learning_rate": 6.5000000000000004e-06,
607
+ "loss": 0.4329,
608
  "step": 2075
609
  },
610
  {
611
+ "epoch": 0.6803823100599384,
612
+ "grad_norm": 3.5781023502349854,
613
  "learning_rate": 6.444444444444445e-06,
614
+ "loss": 0.4191,
615
  "step": 2100
616
  },
617
  {
618
+ "epoch": 0.6884820994654139,
619
+ "grad_norm": 3.257659912109375,
620
  "learning_rate": 6.3888888888888885e-06,
621
+ "loss": 0.4046,
622
  "step": 2125
623
  },
624
  {
625
+ "epoch": 0.6965818888708893,
626
+ "grad_norm": 3.030454397201538,
627
  "learning_rate": 6.333333333333333e-06,
628
+ "loss": 0.4033,
629
  "step": 2150
630
  },
631
  {
632
+ "epoch": 0.7046816782763649,
633
+ "grad_norm": 3.3289663791656494,
634
  "learning_rate": 6.277777777777778e-06,
635
+ "loss": 0.4342,
636
  "step": 2175
637
  },
638
  {
639
+ "epoch": 0.7127814676818403,
640
+ "grad_norm": 3.4743051528930664,
641
  "learning_rate": 6.222222222222223e-06,
642
+ "loss": 0.4102,
643
  "step": 2200
644
  },
645
  {
646
+ "epoch": 0.7208812570873158,
647
+ "grad_norm": 3.300485610961914,
648
  "learning_rate": 6.166666666666667e-06,
649
+ "loss": 0.3951,
650
  "step": 2225
651
  },
652
  {
653
+ "epoch": 0.7289810464927912,
654
+ "grad_norm": 3.374356746673584,
655
  "learning_rate": 6.111111111111112e-06,
656
+ "loss": 0.3993,
657
  "step": 2250
658
  },
659
  {
660
+ "epoch": 0.7370808358982667,
661
+ "grad_norm": 3.108508586883545,
662
  "learning_rate": 6.055555555555555e-06,
663
+ "loss": 0.3927,
664
  "step": 2275
665
  },
666
  {
667
+ "epoch": 0.7451806253037421,
668
+ "grad_norm": 3.448174238204956,
669
  "learning_rate": 6e-06,
670
+ "loss": 0.4254,
671
  "step": 2300
672
  },
673
  {
674
+ "epoch": 0.7532804147092176,
675
+ "grad_norm": 3.17012095451355,
676
  "learning_rate": 5.944444444444445e-06,
677
+ "loss": 0.396,
678
  "step": 2325
679
  },
680
  {
681
+ "epoch": 0.761380204114693,
682
+ "grad_norm": 3.3391034603118896,
683
  "learning_rate": 5.88888888888889e-06,
684
+ "loss": 0.4035,
685
  "step": 2350
686
  },
687
  {
688
+ "epoch": 0.7694799935201685,
689
+ "grad_norm": 3.1390321254730225,
690
  "learning_rate": 5.833333333333334e-06,
691
+ "loss": 0.4387,
692
  "step": 2375
693
  },
694
  {
695
+ "epoch": 0.7775797829256439,
696
+ "grad_norm": 2.801060199737549,
697
+ "learning_rate": 5.78e-06,
698
+ "loss": 0.4101,
699
  "step": 2400
700
  },
701
  {
702
+ "epoch": 0.7856795723311194,
703
+ "grad_norm": 3.169654130935669,
704
+ "learning_rate": 5.724444444444445e-06,
705
+ "loss": 0.4019,
706
  "step": 2425
707
  },
708
  {
709
+ "epoch": 0.7937793617365948,
710
+ "grad_norm": 3.1517333984375,
711
+ "learning_rate": 5.6688888888888895e-06,
712
+ "loss": 0.3955,
713
  "step": 2450
714
  },
715
  {
716
+ "epoch": 0.8018791511420703,
717
+ "grad_norm": 2.812309980392456,
718
+ "learning_rate": 5.613333333333334e-06,
719
+ "loss": 0.3976,
720
  "step": 2475
721
  },
722
  {
723
+ "epoch": 0.8099789405475457,
724
+ "grad_norm": 3.095435619354248,
725
+ "learning_rate": 5.557777777777778e-06,
726
+ "loss": 0.4173,
727
  "step": 2500
728
  },
729
  {
730
+ "epoch": 0.8180787299530212,
731
+ "grad_norm": 3.530505418777466,
732
+ "learning_rate": 5.5022222222222224e-06,
733
+ "loss": 0.3968,
734
  "step": 2525
735
  },
736
  {
737
+ "epoch": 0.8261785193584967,
738
+ "grad_norm": 3.6718244552612305,
739
+ "learning_rate": 5.4466666666666665e-06,
740
+ "loss": 0.4365,
741
  "step": 2550
742
  },
743
  {
744
+ "epoch": 0.8342783087639721,
745
+ "grad_norm": 3.455793857574463,
746
+ "learning_rate": 5.391111111111111e-06,
747
+ "loss": 0.3995,
748
  "step": 2575
749
  },
750
  {
751
+ "epoch": 0.8423780981694476,
752
+ "grad_norm": 3.074331760406494,
753
+ "learning_rate": 5.335555555555556e-06,
754
+ "loss": 0.3995,
755
  "step": 2600
756
  },
757
  {
758
+ "epoch": 0.850477887574923,
759
+ "grad_norm": 3.732499361038208,
760
+ "learning_rate": 5.28e-06,
761
+ "loss": 0.3873,
762
  "step": 2625
763
  },
764
  {
765
+ "epoch": 0.8585776769803986,
766
+ "grad_norm": 2.719492197036743,
767
+ "learning_rate": 5.224444444444445e-06,
768
+ "loss": 0.3938,
769
  "step": 2650
770
  },
771
  {
772
+ "epoch": 0.866677466385874,
773
+ "grad_norm": 3.279792308807373,
774
+ "learning_rate": 5.168888888888889e-06,
775
+ "loss": 0.4087,
776
  "step": 2675
777
  },
778
  {
779
+ "epoch": 0.8747772557913495,
780
+ "grad_norm": 3.178786516189575,
781
+ "learning_rate": 5.113333333333333e-06,
782
+ "loss": 0.3771,
783
  "step": 2700
784
  },
785
  {
786
+ "epoch": 0.8828770451968249,
787
+ "grad_norm": 2.70697283744812,
788
+ "learning_rate": 5.057777777777778e-06,
789
+ "loss": 0.3991,
790
  "step": 2725
791
  },
792
  {
793
+ "epoch": 0.8909768346023004,
794
+ "grad_norm": 3.0300474166870117,
795
+ "learning_rate": 5.002222222222223e-06,
796
+ "loss": 0.4069,
797
  "step": 2750
798
  },
799
  {
800
+ "epoch": 0.8990766240077758,
801
+ "grad_norm": 4.001626491546631,
802
+ "learning_rate": 4.946666666666667e-06,
803
+ "loss": 0.4153,
804
  "step": 2775
805
  },
806
  {
807
+ "epoch": 0.9071764134132513,
808
+ "grad_norm": 3.1544246673583984,
809
+ "learning_rate": 4.891111111111111e-06,
810
+ "loss": 0.3733,
811
  "step": 2800
812
  },
813
  {
814
+ "epoch": 0.9152762028187267,
815
+ "grad_norm": 3.553840398788452,
816
+ "learning_rate": 4.835555555555556e-06,
817
+ "loss": 0.409,
818
  "step": 2825
819
  },
820
  {
821
+ "epoch": 0.9233759922242022,
822
+ "grad_norm": 3.6127443313598633,
823
+ "learning_rate": 4.78e-06,
824
+ "loss": 0.3728,
825
  "step": 2850
826
  },
827
  {
828
+ "epoch": 0.9314757816296776,
829
+ "grad_norm": 3.1611216068267822,
830
+ "learning_rate": 4.724444444444445e-06,
831
+ "loss": 0.3923,
832
  "step": 2875
833
  },
834
  {
835
+ "epoch": 0.9395755710351531,
836
+ "grad_norm": 3.2189152240753174,
837
+ "learning_rate": 4.66888888888889e-06,
838
+ "loss": 0.4179,
839
  "step": 2900
840
  },
841
  {
842
+ "epoch": 0.9476753604406285,
843
+ "grad_norm": 3.3200769424438477,
844
+ "learning_rate": 4.613333333333334e-06,
845
+ "loss": 0.4067,
846
  "step": 2925
847
  },
848
  {
849
+ "epoch": 0.955775149846104,
850
+ "grad_norm": 3.38618540763855,
851
+ "learning_rate": 4.557777777777778e-06,
852
+ "loss": 0.384,
853
  "step": 2950
854
  },
855
  {
856
+ "epoch": 0.9638749392515794,
857
+ "grad_norm": 3.5702147483825684,
858
+ "learning_rate": 4.502222222222223e-06,
859
+ "loss": 0.3949,
860
  "step": 2975
861
  },
862
  {
863
+ "epoch": 0.9719747286570549,
864
+ "grad_norm": 2.6733343601226807,
865
+ "learning_rate": 4.446666666666667e-06,
866
+ "loss": 0.3914,
867
  "step": 3000
868
  },
869
  {
870
+ "epoch": 0.9719747286570549,
871
+ "eval_loss": 0.40348634123802185,
872
+ "eval_runtime": 1837.3277,
873
+ "eval_samples_per_second": 2.121,
874
+ "eval_steps_per_second": 0.133,
875
+ "eval_wer": 0.3089751149899752,
876
  "step": 3000
877
  },
878
  {
879
+ "epoch": 0.9800745180625303,
880
+ "grad_norm": 3.1163229942321777,
881
+ "learning_rate": 4.391111111111112e-06,
882
+ "loss": 0.3614,
883
  "step": 3025
884
  },
885
  {
886
+ "epoch": 0.9881743074680058,
887
+ "grad_norm": 4.919127941131592,
888
+ "learning_rate": 4.3355555555555565e-06,
889
+ "loss": 0.3815,
890
  "step": 3050
891
  },
892
  {
893
+ "epoch": 0.9962740968734813,
894
+ "grad_norm": 3.523918867111206,
895
+ "learning_rate": 4.2800000000000005e-06,
896
+ "loss": 0.3679,
897
  "step": 3075
898
  },
899
  {
900
+ "epoch": 1.0042118904908472,
901
+ "grad_norm": 2.188014268875122,
902
+ "learning_rate": 4.2244444444444446e-06,
903
+ "loss": 0.3305,
904
  "step": 3100
905
  },
906
  {
907
+ "epoch": 1.0123116798963228,
908
+ "grad_norm": 3.0015525817871094,
909
+ "learning_rate": 4.168888888888889e-06,
910
+ "loss": 0.2856,
911
  "step": 3125
912
  },
913
  {
914
+ "epoch": 1.020411469301798,
915
+ "grad_norm": 3.565284490585327,
916
+ "learning_rate": 4.1133333333333335e-06,
917
+ "loss": 0.296,
918
  "step": 3150
919
  },
920
  {
921
+ "epoch": 1.0285112587072736,
922
+ "grad_norm": 2.8771865367889404,
923
+ "learning_rate": 4.057777777777778e-06,
924
+ "loss": 0.2912,
925
  "step": 3175
926
  },
927
  {
928
+ "epoch": 1.036611048112749,
929
+ "grad_norm": 2.7181403636932373,
930
+ "learning_rate": 4.002222222222222e-06,
931
+ "loss": 0.2871,
932
  "step": 3200
933
  },
934
  {
935
+ "epoch": 1.0447108375182246,
936
+ "grad_norm": 2.652308225631714,
937
+ "learning_rate": 3.946666666666667e-06,
938
+ "loss": 0.2669,
939
  "step": 3225
940
  },
941
  {
942
+ "epoch": 1.0528106269237,
943
+ "grad_norm": 2.390458345413208,
944
+ "learning_rate": 3.891111111111111e-06,
945
+ "loss": 0.2852,
946
  "step": 3250
947
  },
948
  {
949
+ "epoch": 1.0609104163291754,
950
+ "grad_norm": 3.036996603012085,
951
+ "learning_rate": 3.835555555555555e-06,
952
+ "loss": 0.2835,
953
  "step": 3275
954
  },
955
  {
956
+ "epoch": 1.0690102057346509,
957
+ "grad_norm": 3.0076892375946045,
958
+ "learning_rate": 3.7800000000000002e-06,
959
+ "loss": 0.2853,
960
  "step": 3300
961
  },
962
  {
963
+ "epoch": 1.0771099951401264,
964
+ "grad_norm": 2.6001665592193604,
965
+ "learning_rate": 3.724444444444445e-06,
966
+ "loss": 0.2552,
967
  "step": 3325
968
  },
969
  {
970
+ "epoch": 1.085209784545602,
971
+ "grad_norm": 2.9106733798980713,
972
+ "learning_rate": 3.668888888888889e-06,
973
+ "loss": 0.2921,
974
  "step": 3350
975
  },
976
  {
977
+ "epoch": 1.0933095739510772,
978
+ "grad_norm": 2.579561710357666,
979
+ "learning_rate": 3.6133333333333336e-06,
980
+ "loss": 0.2904,
981
  "step": 3375
982
  },
983
  {
984
+ "epoch": 1.1014093633565527,
985
+ "grad_norm": 2.4961764812469482,
986
+ "learning_rate": 3.5577777777777785e-06,
987
+ "loss": 0.2699,
988
  "step": 3400
989
  },
990
  {
991
+ "epoch": 1.1095091527620282,
992
+ "grad_norm": 2.9800219535827637,
993
+ "learning_rate": 3.5022222222222225e-06,
994
+ "loss": 0.2867,
995
  "step": 3425
996
  },
997
  {
998
+ "epoch": 1.1176089421675037,
999
+ "grad_norm": 2.7929482460021973,
1000
+ "learning_rate": 3.446666666666667e-06,
1001
+ "loss": 0.2707,
1002
  "step": 3450
1003
  },
1004
  {
1005
+ "epoch": 1.125708731572979,
1006
+ "grad_norm": 3.021549940109253,
1007
+ "learning_rate": 3.391111111111111e-06,
1008
+ "loss": 0.2793,
1009
  "step": 3475
1010
  },
1011
  {
1012
+ "epoch": 1.1338085209784545,
1013
+ "grad_norm": 2.6613991260528564,
1014
+ "learning_rate": 3.335555555555556e-06,
1015
+ "loss": 0.2705,
1016
  "step": 3500
1017
  },
1018
  {
1019
+ "epoch": 1.14190831038393,
1020
+ "grad_norm": 2.4502217769622803,
1021
+ "learning_rate": 3.2800000000000004e-06,
1022
+ "loss": 0.2776,
1023
  "step": 3525
1024
  },
1025
  {
1026
+ "epoch": 1.1500080997894055,
1027
+ "grad_norm": 2.698561668395996,
1028
+ "learning_rate": 3.2244444444444444e-06,
1029
+ "loss": 0.2685,
1030
  "step": 3550
1031
  },
1032
  {
1033
+ "epoch": 1.158107889194881,
1034
+ "grad_norm": 2.5799190998077393,
1035
+ "learning_rate": 3.1688888888888893e-06,
1036
+ "loss": 0.2788,
1037
  "step": 3575
1038
  },
1039
  {
1040
+ "epoch": 1.1662076786003563,
1041
+ "grad_norm": 2.5856029987335205,
1042
+ "learning_rate": 3.1133333333333337e-06,
1043
+ "loss": 0.2691,
1044
  "step": 3600
1045
  },
1046
  {
1047
+ "epoch": 1.1743074680058319,
1048
+ "grad_norm": 2.861074209213257,
1049
+ "learning_rate": 3.0577777777777778e-06,
1050
+ "loss": 0.2862,
1051
  "step": 3625
1052
  },
1053
  {
1054
+ "epoch": 1.1824072574113074,
1055
+ "grad_norm": 2.6794049739837646,
1056
+ "learning_rate": 3.0022222222222227e-06,
1057
+ "loss": 0.269,
1058
  "step": 3650
1059
  },
1060
  {
1061
+ "epoch": 1.1905070468167827,
1062
+ "grad_norm": 2.6813509464263916,
1063
+ "learning_rate": 2.946666666666667e-06,
1064
+ "loss": 0.2787,
1065
  "step": 3675
1066
  },
1067
  {
1068
+ "epoch": 1.1986068362222582,
1069
+ "grad_norm": 2.5139195919036865,
1070
+ "learning_rate": 2.891111111111111e-06,
1071
+ "loss": 0.2876,
1072
  "step": 3700
1073
  },
1074
  {
1075
+ "epoch": 1.2067066256277337,
1076
+ "grad_norm": 2.5221784114837646,
1077
+ "learning_rate": 2.835555555555556e-06,
1078
+ "loss": 0.2629,
1079
  "step": 3725
1080
  },
1081
  {
1082
+ "epoch": 1.2148064150332092,
1083
+ "grad_norm": 2.6317663192749023,
1084
+ "learning_rate": 2.7800000000000005e-06,
1085
+ "loss": 0.2792,
1086
  "step": 3750
1087
  },
1088
  {
1089
+ "epoch": 1.2229062044386847,
1090
+ "grad_norm": 3.165544271469116,
1091
+ "learning_rate": 2.7244444444444445e-06,
1092
+ "loss": 0.2798,
1093
  "step": 3775
1094
  },
1095
  {
1096
+ "epoch": 1.23100599384416,
1097
+ "grad_norm": 2.9240896701812744,
1098
+ "learning_rate": 2.6688888888888894e-06,
1099
+ "loss": 0.2745,
1100
  "step": 3800
1101
  },
1102
  {
1103
+ "epoch": 1.2391057832496355,
1104
+ "grad_norm": 2.7815427780151367,
1105
+ "learning_rate": 2.6133333333333334e-06,
1106
+ "loss": 0.2768,
1107
  "step": 3825
1108
  },
1109
  {
1110
+ "epoch": 1.247205572655111,
1111
+ "grad_norm": 3.016592025756836,
1112
+ "learning_rate": 2.557777777777778e-06,
1113
+ "loss": 0.2735,
1114
  "step": 3850
1115
  },
1116
  {
1117
+ "epoch": 1.2553053620605863,
1118
+ "grad_norm": 2.7359392642974854,
1119
+ "learning_rate": 2.5022222222222224e-06,
1120
+ "loss": 0.2757,
1121
  "step": 3875
1122
  },
1123
  {
1124
+ "epoch": 1.2634051514660618,
1125
+ "grad_norm": 2.575500726699829,
1126
+ "learning_rate": 2.446666666666667e-06,
1127
+ "loss": 0.2635,
1128
  "step": 3900
1129
  },
1130
  {
1131
+ "epoch": 1.2715049408715373,
1132
+ "grad_norm": 2.202298164367676,
1133
+ "learning_rate": 2.3911111111111113e-06,
1134
+ "loss": 0.2682,
1135
  "step": 3925
1136
  },
1137
  {
1138
+ "epoch": 1.2796047302770128,
1139
+ "grad_norm": 3.1898550987243652,
1140
+ "learning_rate": 2.3355555555555557e-06,
1141
+ "loss": 0.2708,
1142
  "step": 3950
1143
  },
1144
  {
1145
+ "epoch": 1.2877045196824883,
1146
+ "grad_norm": 2.5168306827545166,
1147
+ "learning_rate": 2.28e-06,
1148
+ "loss": 0.2639,
1149
  "step": 3975
1150
  },
1151
  {
1152
+ "epoch": 1.2958043090879636,
1153
+ "grad_norm": 2.682749032974243,
1154
+ "learning_rate": 2.2244444444444447e-06,
1155
+ "loss": 0.296,
1156
  "step": 4000
1157
  },
1158
  {
1159
+ "epoch": 1.2958043090879636,
1160
+ "eval_loss": 0.3868160843849182,
1161
+ "eval_runtime": 1836.049,
1162
+ "eval_samples_per_second": 2.122,
1163
+ "eval_steps_per_second": 0.133,
1164
+ "eval_wer": 0.2977002004953414,
1165
  "step": 4000
1166
  },
1167
  {
1168
+ "epoch": 1.3039040984934391,
1169
+ "grad_norm": 2.805448293685913,
1170
+ "learning_rate": 2.168888888888889e-06,
1171
+ "loss": 0.2728,
1172
  "step": 4025
1173
  },
1174
  {
1175
+ "epoch": 1.3120038878989146,
1176
+ "grad_norm": 2.2613837718963623,
1177
+ "learning_rate": 2.1133333333333336e-06,
1178
+ "loss": 0.2695,
1179
  "step": 4050
1180
  },
1181
  {
1182
+ "epoch": 1.3201036773043902,
1183
+ "grad_norm": 2.4388859272003174,
1184
+ "learning_rate": 2.057777777777778e-06,
1185
+ "loss": 0.286,
1186
  "step": 4075
1187
  },
1188
  {
1189
+ "epoch": 1.3282034667098657,
1190
+ "grad_norm": 2.807706832885742,
1191
+ "learning_rate": 2.0022222222222225e-06,
1192
+ "loss": 0.251,
1193
  "step": 4100
1194
  },
1195
  {
1196
+ "epoch": 1.336303256115341,
1197
+ "grad_norm": 2.927849292755127,
1198
+ "learning_rate": 1.9466666666666665e-06,
1199
+ "loss": 0.2747,
1200
  "step": 4125
1201
  },
1202
  {
1203
+ "epoch": 1.3444030455208165,
1204
+ "grad_norm": 2.2061877250671387,
1205
+ "learning_rate": 1.8911111111111114e-06,
1206
+ "loss": 0.267,
1207
  "step": 4150
1208
  },
1209
  {
1210
+ "epoch": 1.352502834926292,
1211
+ "grad_norm": 2.5290238857269287,
1212
+ "learning_rate": 1.8355555555555557e-06,
1213
+ "loss": 0.2727,
1214
  "step": 4175
1215
  },
1216
  {
1217
+ "epoch": 1.3606026243317673,
1218
+ "grad_norm": 2.74303936958313,
1219
+ "learning_rate": 1.7800000000000001e-06,
1220
+ "loss": 0.2726,
1221
  "step": 4200
1222
  },
1223
  {
1224
+ "epoch": 1.3687024137372428,
1225
+ "grad_norm": 2.194861888885498,
1226
+ "learning_rate": 1.7244444444444448e-06,
1227
+ "loss": 0.2645,
1228
  "step": 4225
1229
  },
1230
  {
1231
+ "epoch": 1.3768022031427183,
1232
+ "grad_norm": 2.4960832595825195,
1233
+ "learning_rate": 1.668888888888889e-06,
1234
+ "loss": 0.2788,
1235
  "step": 4250
1236
  },
1237
  {
1238
+ "epoch": 1.3849019925481938,
1239
+ "grad_norm": 2.3262386322021484,
1240
+ "learning_rate": 1.6133333333333335e-06,
1241
+ "loss": 0.2586,
1242
  "step": 4275
1243
  },
1244
  {
1245
+ "epoch": 1.3930017819536693,
1246
+ "grad_norm": 2.5423128604888916,
1247
+ "learning_rate": 1.5577777777777777e-06,
1248
+ "loss": 0.2719,
1249
  "step": 4300
1250
  },
1251
  {
1252
+ "epoch": 1.4011015713591446,
1253
+ "grad_norm": 2.9188232421875,
1254
+ "learning_rate": 1.5022222222222224e-06,
1255
+ "loss": 0.2671,
1256
  "step": 4325
1257
  },
1258
  {
1259
+ "epoch": 1.40920136076462,
1260
+ "grad_norm": 2.624691963195801,
1261
+ "learning_rate": 1.4466666666666669e-06,
1262
+ "loss": 0.2809,
1263
  "step": 4350
1264
  },
1265
  {
1266
+ "epoch": 1.4173011501700956,
1267
+ "grad_norm": 2.674189329147339,
1268
+ "learning_rate": 1.3911111111111111e-06,
1269
+ "loss": 0.2651,
1270
  "step": 4375
1271
  },
1272
  {
1273
+ "epoch": 1.425400939575571,
1274
+ "grad_norm": 2.832871913909912,
1275
+ "learning_rate": 1.3355555555555558e-06,
1276
+ "loss": 0.2676,
1277
  "step": 4400
1278
  },
1279
  {
1280
+ "epoch": 1.4335007289810464,
1281
+ "grad_norm": 2.8886163234710693,
1282
  "learning_rate": 1.28e-06,
1283
+ "loss": 0.2717,
1284
  "step": 4425
1285
  },
1286
  {
1287
+ "epoch": 1.441600518386522,
1288
+ "grad_norm": 2.9759998321533203,
1289
  "learning_rate": 1.2244444444444445e-06,
1290
+ "loss": 0.2709,
1291
  "step": 4450
1292
  },
1293
  {
1294
+ "epoch": 1.4497003077919974,
1295
+ "grad_norm": 2.8396642208099365,
1296
  "learning_rate": 1.168888888888889e-06,
1297
+ "loss": 0.2701,
1298
  "step": 4475
1299
  },
1300
  {
1301
+ "epoch": 1.457800097197473,
1302
+ "grad_norm": 3.0574100017547607,
1303
+ "learning_rate": 1.1155555555555558e-06,
1304
+ "loss": 0.2592,
1305
  "step": 4500
1306
  },
1307
  {
1308
+ "epoch": 1.4658998866029482,
1309
+ "grad_norm": 2.318142890930176,
1310
+ "learning_rate": 1.06e-06,
1311
+ "loss": 0.2731,
1312
  "step": 4525
1313
  },
1314
  {
1315
+ "epoch": 1.4739996760084237,
1316
+ "grad_norm": 2.372235059738159,
1317
+ "learning_rate": 1.0044444444444445e-06,
1318
+ "loss": 0.2694,
1319
  "step": 4550
1320
  },
1321
  {
1322
+ "epoch": 1.4820994654138993,
1323
+ "grad_norm": 2.6232292652130127,
1324
+ "learning_rate": 9.488888888888889e-07,
1325
+ "loss": 0.2529,
1326
  "step": 4575
1327
  },
1328
  {
1329
+ "epoch": 1.4901992548193748,
1330
+ "grad_norm": 2.717992067337036,
1331
+ "learning_rate": 8.933333333333334e-07,
1332
+ "loss": 0.2779,
1333
  "step": 4600
1334
  },
1335
  {
1336
+ "epoch": 1.4982990442248503,
1337
+ "grad_norm": 2.852125406265259,
1338
+ "learning_rate": 8.37777777777778e-07,
1339
+ "loss": 0.2596,
1340
  "step": 4625
1341
  },
1342
  {
1343
+ "epoch": 1.5063988336303256,
1344
+ "grad_norm": 2.2071590423583984,
1345
+ "learning_rate": 7.822222222222223e-07,
1346
+ "loss": 0.2765,
1347
  "step": 4650
1348
  },
1349
  {
1350
+ "epoch": 1.514498623035801,
1351
+ "grad_norm": 2.871159315109253,
1352
+ "learning_rate": 7.266666666666668e-07,
1353
+ "loss": 0.2594,
1354
  "step": 4675
1355
  },
1356
  {
1357
+ "epoch": 1.5225984124412766,
1358
+ "grad_norm": 2.287871837615967,
1359
+ "learning_rate": 6.711111111111111e-07,
1360
+ "loss": 0.2703,
1361
  "step": 4700
1362
  },
1363
  {
1364
+ "epoch": 1.5306982018467519,
1365
+ "grad_norm": 2.345743417739868,
1366
+ "learning_rate": 6.155555555555556e-07,
1367
+ "loss": 0.2641,
1368
  "step": 4725
1369
  },
1370
  {
1371
+ "epoch": 1.5387979912522276,
1372
+ "grad_norm": 2.734402656555176,
1373
+ "learning_rate": 5.6e-07,
1374
+ "loss": 0.2644,
1375
  "step": 4750
1376
  },
1377
  {
1378
+ "epoch": 1.546897780657703,
1379
+ "grad_norm": 2.6672370433807373,
1380
+ "learning_rate": 5.044444444444445e-07,
1381
+ "loss": 0.2568,
1382
  "step": 4775
1383
  },
1384
  {
1385
+ "epoch": 1.5549975700631784,
1386
+ "grad_norm": 2.9469447135925293,
1387
+ "learning_rate": 4.488888888888889e-07,
1388
+ "loss": 0.2646,
1389
  "step": 4800
1390
  },
1391
  {
1392
+ "epoch": 1.563097359468654,
1393
+ "grad_norm": 3.1269147396087646,
1394
+ "learning_rate": 3.9333333333333336e-07,
1395
+ "loss": 0.2643,
1396
  "step": 4825
1397
  },
1398
  {
1399
+ "epoch": 1.5711971488741292,
1400
+ "grad_norm": 2.543921709060669,
1401
+ "learning_rate": 3.3777777777777777e-07,
1402
+ "loss": 0.2738,
1403
  "step": 4850
1404
  },
1405
  {
1406
+ "epoch": 1.5792969382796047,
1407
+ "grad_norm": 2.7174246311187744,
1408
+ "learning_rate": 2.822222222222222e-07,
1409
+ "loss": 0.2439,
1410
  "step": 4875
1411
  },
1412
  {
1413
+ "epoch": 1.5873967276850802,
1414
+ "grad_norm": 2.381450891494751,
1415
+ "learning_rate": 2.266666666666667e-07,
1416
+ "loss": 0.2773,
1417
  "step": 4900
1418
  },
1419
  {
1420
+ "epoch": 1.5954965170905555,
1421
+ "grad_norm": 2.9978244304656982,
1422
+ "learning_rate": 1.7111111111111114e-07,
1423
+ "loss": 0.2653,
1424
  "step": 4925
1425
  },
1426
  {
1427
+ "epoch": 1.6035963064960312,
1428
+ "grad_norm": 2.594560384750366,
1429
+ "learning_rate": 1.1555555555555556e-07,
1430
+ "loss": 0.2472,
1431
  "step": 4950
1432
  },
1433
  {
1434
+ "epoch": 1.6116960959015065,
1435
+ "grad_norm": 2.8399264812469482,
1436
+ "learning_rate": 6.000000000000001e-08,
1437
+ "loss": 0.258,
1438
  "step": 4975
1439
  },
1440
  {
1441
+ "epoch": 1.619795885306982,
1442
+ "grad_norm": 2.9382212162017822,
1443
+ "learning_rate": 4.444444444444445e-09,
1444
+ "loss": 0.274,
1445
  "step": 5000
1446
  },
1447
  {
1448
+ "epoch": 1.619795885306982,
1449
+ "eval_loss": 0.3738669753074646,
1450
+ "eval_runtime": 1856.8663,
1451
+ "eval_samples_per_second": 2.099,
1452
+ "eval_steps_per_second": 0.131,
1453
+ "eval_wer": 0.29154381412902464,
1454
  "step": 5000
1455
  },
1456
  {
1457
+ "epoch": 1.619795885306982,
1458
  "step": 5000,
1459
+ "total_flos": 5.435419715783885e+20,
1460
+ "train_loss": 0.4225531764030456,
1461
+ "train_runtime": 64153.0147,
1462
+ "train_samples_per_second": 2.494,
1463
+ "train_steps_per_second": 0.078
1464
  }
1465
  ],
1466
  "logging_steps": 25,
1467
  "max_steps": 5000,
1468
  "num_input_tokens_seen": 0,
1469
+ "num_train_epochs": 2,
1470
  "save_steps": 1000,
1471
  "stateful_callbacks": {
1472
  "TrainerControl": {
 
1480
  "attributes": {}
1481
  }
1482
  },
1483
+ "total_flos": 5.435419715783885e+20,
1484
  "train_batch_size": 16,
1485
  "trial_name": null,
1486
  "trial_params": null
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:52d863451b68ca001911093c50580b07ade53625439e9961dceca82905be4d2f
3
- size 5432
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d3ec45a0f94508361d7a8afc788e43452d0d48a3e9839538d81547b065babf44
3
+ size 5560