ashutosh01012001 commited on
Commit
8cedfc2
1 Parent(s): e1e7609

Upload 9 files

Browse files
config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "vasista22/whisper-kannada-tiny",
3
+ "activation_dropout": 0.0,
4
+ "activation_function": "gelu",
5
+ "apply_spec_augment": false,
6
+ "architectures": [
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": 384,
17
+ "decoder_attention_heads": 6,
18
+ "decoder_ffn_dim": 1536,
19
+ "decoder_layerdrop": 0.0,
20
+ "decoder_layers": 4,
21
+ "decoder_start_token_id": 50258,
22
+ "dropout": 0.0,
23
+ "encoder_attention_heads": 6,
24
+ "encoder_ffn_dim": 1536,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 4,
27
+ "eos_token_id": 50257,
28
+ "forced_decoder_ids": null,
29
+ "init_std": 0.02,
30
+ "is_encoder_decoder": true,
31
+ "mask_feature_length": 10,
32
+ "mask_feature_min_masks": 0,
33
+ "mask_feature_prob": 0.0,
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,
41
+ "model_type": "whisper",
42
+ "num_hidden_layers": 4,
43
+ "num_mel_bins": 80,
44
+ "pad_token_id": 50257,
45
+ "scale_embedding": false,
46
+ "suppress_tokens": [],
47
+ "torch_dtype": "float32",
48
+ "transformers_version": "4.39.0.dev0",
49
+ "use_cache": false,
50
+ "use_weighted_layer_sum": false,
51
+ "vocab_size": 51865
52
+ }
generation_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "begin_suppress_tokens": [
3
+ 220,
4
+ 50257
5
+ ],
6
+ "bos_token_id": 50257,
7
+ "decoder_start_token_id": 50258,
8
+ "eos_token_id": 50257,
9
+ "max_length": 448,
10
+ "pad_token_id": 50257,
11
+ "suppress_tokens": [],
12
+ "transformers_version": "4.39.0.dev0",
13
+ "use_cache": false
14
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7e93abd11010d91a10c570c65411e4c287d8d6b3fa4f964e5c2cc30a3c3a010
3
+ size 151061672
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:58ca1f17a3908cc9ae7d5a08cdab8ec3771a190694314f4109881739a79dced5
3
+ size 297616186
preprocessor_config.json ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "chunk_length": 30,
3
+ "feature_extractor_type": "WhisperFeatureExtractor",
4
+ "feature_size": 80,
5
+ "hop_length": 160,
6
+ "n_fft": 400,
7
+ "n_samples": 480000,
8
+ "nb_max_frames": 3000,
9
+ "padding_side": "right",
10
+ "padding_value": 0.0,
11
+ "processor_class": "WhisperProcessor",
12
+ "return_attention_mask": false,
13
+ "sampling_rate": 16000
14
+ }
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3d3505da080f4dc5a8eea8abd004ed7ce11a56f181ee06d5af2cf5b030dae1d9
3
+ size 14244
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e35c47f6a21ee1bb8f2f8642578cb8c19ffdc3fefc7de53d643d321b1980b5e3
3
+ size 1064
trainer_state.json ADDED
@@ -0,0 +1,354 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": 24.489795918367346,
3
+ "best_model_checkpoint": "/content/drive/MyDrive/vavista_kannada_collection_sales_bot_v2/checkpoint-900",
4
+ "epoch": 14.285714285714286,
5
+ "eval_steps": 100,
6
+ "global_step": 900,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.4,
13
+ "grad_norm": 54.585182189941406,
14
+ "learning_rate": 3.8e-07,
15
+ "loss": 2.0199,
16
+ "step": 25
17
+ },
18
+ {
19
+ "epoch": 0.79,
20
+ "grad_norm": 38.83250427246094,
21
+ "learning_rate": 8.8e-07,
22
+ "loss": 1.7912,
23
+ "step": 50
24
+ },
25
+ {
26
+ "epoch": 1.19,
27
+ "grad_norm": 33.85667037963867,
28
+ "learning_rate": 1.3800000000000001e-06,
29
+ "loss": 1.2745,
30
+ "step": 75
31
+ },
32
+ {
33
+ "epoch": 1.59,
34
+ "grad_norm": 16.32790184020996,
35
+ "learning_rate": 1.8800000000000002e-06,
36
+ "loss": 0.7215,
37
+ "step": 100
38
+ },
39
+ {
40
+ "epoch": 1.59,
41
+ "eval_loss": 0.5508027672767639,
42
+ "eval_runtime": 174.0594,
43
+ "eval_samples_per_second": 1.419,
44
+ "eval_steps_per_second": 0.046,
45
+ "eval_wer": 116.3265306122449,
46
+ "step": 100
47
+ },
48
+ {
49
+ "epoch": 1.98,
50
+ "grad_norm": 9.59360408782959,
51
+ "learning_rate": 2.38e-06,
52
+ "loss": 0.4407,
53
+ "step": 125
54
+ },
55
+ {
56
+ "epoch": 2.38,
57
+ "grad_norm": 7.552663326263428,
58
+ "learning_rate": 2.88e-06,
59
+ "loss": 0.3329,
60
+ "step": 150
61
+ },
62
+ {
63
+ "epoch": 2.78,
64
+ "grad_norm": 7.011340141296387,
65
+ "learning_rate": 3.3800000000000007e-06,
66
+ "loss": 0.2694,
67
+ "step": 175
68
+ },
69
+ {
70
+ "epoch": 3.17,
71
+ "grad_norm": 5.541098117828369,
72
+ "learning_rate": 3.88e-06,
73
+ "loss": 0.2487,
74
+ "step": 200
75
+ },
76
+ {
77
+ "epoch": 3.17,
78
+ "eval_loss": 0.2373683899641037,
79
+ "eval_runtime": 132.9698,
80
+ "eval_samples_per_second": 1.858,
81
+ "eval_steps_per_second": 0.06,
82
+ "eval_wer": 71.68367346938776,
83
+ "step": 200
84
+ },
85
+ {
86
+ "epoch": 3.57,
87
+ "grad_norm": 4.505865573883057,
88
+ "learning_rate": 4.38e-06,
89
+ "loss": 0.1927,
90
+ "step": 225
91
+ },
92
+ {
93
+ "epoch": 3.97,
94
+ "grad_norm": 5.797987461090088,
95
+ "learning_rate": 4.880000000000001e-06,
96
+ "loss": 0.1946,
97
+ "step": 250
98
+ },
99
+ {
100
+ "epoch": 4.37,
101
+ "grad_norm": 4.06318998336792,
102
+ "learning_rate": 5.380000000000001e-06,
103
+ "loss": 0.161,
104
+ "step": 275
105
+ },
106
+ {
107
+ "epoch": 4.76,
108
+ "grad_norm": 3.223536252975464,
109
+ "learning_rate": 5.8800000000000005e-06,
110
+ "loss": 0.139,
111
+ "step": 300
112
+ },
113
+ {
114
+ "epoch": 4.76,
115
+ "eval_loss": 0.18038798868656158,
116
+ "eval_runtime": 87.9591,
117
+ "eval_samples_per_second": 2.808,
118
+ "eval_steps_per_second": 0.091,
119
+ "eval_wer": 40.05102040816326,
120
+ "step": 300
121
+ },
122
+ {
123
+ "epoch": 5.16,
124
+ "grad_norm": 5.675058841705322,
125
+ "learning_rate": 6.380000000000001e-06,
126
+ "loss": 0.1185,
127
+ "step": 325
128
+ },
129
+ {
130
+ "epoch": 5.56,
131
+ "grad_norm": 3.9274826049804688,
132
+ "learning_rate": 6.88e-06,
133
+ "loss": 0.1022,
134
+ "step": 350
135
+ },
136
+ {
137
+ "epoch": 5.95,
138
+ "grad_norm": 3.3899903297424316,
139
+ "learning_rate": 7.3800000000000005e-06,
140
+ "loss": 0.1012,
141
+ "step": 375
142
+ },
143
+ {
144
+ "epoch": 6.35,
145
+ "grad_norm": 2.7243378162384033,
146
+ "learning_rate": 7.88e-06,
147
+ "loss": 0.0712,
148
+ "step": 400
149
+ },
150
+ {
151
+ "epoch": 6.35,
152
+ "eval_loss": 0.16518759727478027,
153
+ "eval_runtime": 45.1355,
154
+ "eval_samples_per_second": 5.472,
155
+ "eval_steps_per_second": 0.177,
156
+ "eval_wer": 29.591836734693878,
157
+ "step": 400
158
+ },
159
+ {
160
+ "epoch": 6.75,
161
+ "grad_norm": 5.021995544433594,
162
+ "learning_rate": 8.380000000000001e-06,
163
+ "loss": 0.0718,
164
+ "step": 425
165
+ },
166
+ {
167
+ "epoch": 7.14,
168
+ "grad_norm": 2.2155253887176514,
169
+ "learning_rate": 8.880000000000001e-06,
170
+ "loss": 0.0687,
171
+ "step": 450
172
+ },
173
+ {
174
+ "epoch": 7.54,
175
+ "grad_norm": 3.375765562057495,
176
+ "learning_rate": 9.38e-06,
177
+ "loss": 0.0506,
178
+ "step": 475
179
+ },
180
+ {
181
+ "epoch": 7.94,
182
+ "grad_norm": 2.747950553894043,
183
+ "learning_rate": 9.88e-06,
184
+ "loss": 0.0448,
185
+ "step": 500
186
+ },
187
+ {
188
+ "epoch": 7.94,
189
+ "eval_loss": 0.15877747535705566,
190
+ "eval_runtime": 45.849,
191
+ "eval_samples_per_second": 5.387,
192
+ "eval_steps_per_second": 0.174,
193
+ "eval_wer": 26.53061224489796,
194
+ "step": 500
195
+ },
196
+ {
197
+ "epoch": 8.33,
198
+ "grad_norm": 2.5800018310546875,
199
+ "learning_rate": 9.620000000000001e-06,
200
+ "loss": 0.0336,
201
+ "step": 525
202
+ },
203
+ {
204
+ "epoch": 8.73,
205
+ "grad_norm": 2.498263359069824,
206
+ "learning_rate": 9.12e-06,
207
+ "loss": 0.0264,
208
+ "step": 550
209
+ },
210
+ {
211
+ "epoch": 9.13,
212
+ "grad_norm": 1.941878318786621,
213
+ "learning_rate": 8.62e-06,
214
+ "loss": 0.0295,
215
+ "step": 575
216
+ },
217
+ {
218
+ "epoch": 9.52,
219
+ "grad_norm": 2.1652817726135254,
220
+ "learning_rate": 8.120000000000002e-06,
221
+ "loss": 0.0186,
222
+ "step": 600
223
+ },
224
+ {
225
+ "epoch": 9.52,
226
+ "eval_loss": 0.17014923691749573,
227
+ "eval_runtime": 45.3455,
228
+ "eval_samples_per_second": 5.447,
229
+ "eval_steps_per_second": 0.176,
230
+ "eval_wer": 27.29591836734694,
231
+ "step": 600
232
+ },
233
+ {
234
+ "epoch": 9.92,
235
+ "grad_norm": 1.3376543521881104,
236
+ "learning_rate": 7.620000000000001e-06,
237
+ "loss": 0.02,
238
+ "step": 625
239
+ },
240
+ {
241
+ "epoch": 10.32,
242
+ "grad_norm": 1.175929069519043,
243
+ "learning_rate": 7.1200000000000004e-06,
244
+ "loss": 0.013,
245
+ "step": 650
246
+ },
247
+ {
248
+ "epoch": 10.71,
249
+ "grad_norm": 1.1778250932693481,
250
+ "learning_rate": 6.620000000000001e-06,
251
+ "loss": 0.0115,
252
+ "step": 675
253
+ },
254
+ {
255
+ "epoch": 11.11,
256
+ "grad_norm": 1.2574666738510132,
257
+ "learning_rate": 6.120000000000001e-06,
258
+ "loss": 0.0115,
259
+ "step": 700
260
+ },
261
+ {
262
+ "epoch": 11.11,
263
+ "eval_loss": 0.17647403478622437,
264
+ "eval_runtime": 45.0453,
265
+ "eval_samples_per_second": 5.483,
266
+ "eval_steps_per_second": 0.178,
267
+ "eval_wer": 25.637755102040817,
268
+ "step": 700
269
+ },
270
+ {
271
+ "epoch": 11.51,
272
+ "grad_norm": 0.981600284576416,
273
+ "learning_rate": 5.620000000000001e-06,
274
+ "loss": 0.0068,
275
+ "step": 725
276
+ },
277
+ {
278
+ "epoch": 11.9,
279
+ "grad_norm": 1.208804726600647,
280
+ "learning_rate": 5.12e-06,
281
+ "loss": 0.0097,
282
+ "step": 750
283
+ },
284
+ {
285
+ "epoch": 12.3,
286
+ "grad_norm": 0.9865374565124512,
287
+ "learning_rate": 4.620000000000001e-06,
288
+ "loss": 0.0058,
289
+ "step": 775
290
+ },
291
+ {
292
+ "epoch": 12.7,
293
+ "grad_norm": 0.5761059522628784,
294
+ "learning_rate": 4.12e-06,
295
+ "loss": 0.0057,
296
+ "step": 800
297
+ },
298
+ {
299
+ "epoch": 12.7,
300
+ "eval_loss": 0.1846890151500702,
301
+ "eval_runtime": 45.0489,
302
+ "eval_samples_per_second": 5.483,
303
+ "eval_steps_per_second": 0.178,
304
+ "eval_wer": 26.27551020408163,
305
+ "step": 800
306
+ },
307
+ {
308
+ "epoch": 13.1,
309
+ "grad_norm": 0.34698066115379333,
310
+ "learning_rate": 3.62e-06,
311
+ "loss": 0.0049,
312
+ "step": 825
313
+ },
314
+ {
315
+ "epoch": 13.49,
316
+ "grad_norm": 0.2580834925174713,
317
+ "learning_rate": 3.12e-06,
318
+ "loss": 0.0036,
319
+ "step": 850
320
+ },
321
+ {
322
+ "epoch": 13.89,
323
+ "grad_norm": 0.6976760625839233,
324
+ "learning_rate": 2.6200000000000003e-06,
325
+ "loss": 0.0043,
326
+ "step": 875
327
+ },
328
+ {
329
+ "epoch": 14.29,
330
+ "grad_norm": 0.257933646440506,
331
+ "learning_rate": 2.12e-06,
332
+ "loss": 0.0031,
333
+ "step": 900
334
+ },
335
+ {
336
+ "epoch": 14.29,
337
+ "eval_loss": 0.18790897727012634,
338
+ "eval_runtime": 45.02,
339
+ "eval_samples_per_second": 5.486,
340
+ "eval_steps_per_second": 0.178,
341
+ "eval_wer": 24.489795918367346,
342
+ "step": 900
343
+ }
344
+ ],
345
+ "logging_steps": 25,
346
+ "max_steps": 1000,
347
+ "num_input_tokens_seen": 0,
348
+ "num_train_epochs": 16,
349
+ "save_steps": 100,
350
+ "total_flos": 7.0350926118912e+17,
351
+ "train_batch_size": 32,
352
+ "trial_name": null,
353
+ "trial_params": null
354
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f77a39f7cd5157dd63feabbd16fecefb90d3b008a0f6ad2c4681ef69006a8664
3
+ size 5112