Training in progress, step 500

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "openai/whisper-tiny",
+  "_name_or_path": "openai/whisper-small",
   "activation_dropout": 0.0,
   "activation_function": "gelu",
   "apply_spec_augment": false,
@@ -13,17 +13,17 @@
   ],
   "bos_token_id": 50257,
   "classifier_proj_size": 256,
-  "d_model": 384,
-  "decoder_attention_heads": 6,
-  "decoder_ffn_dim": 1536,
+  "d_model": 768,
+  "decoder_attention_heads": 12,
+  "decoder_ffn_dim": 3072,
   "decoder_layerdrop": 0.0,
-  "decoder_layers": 4,
+  "decoder_layers": 12,
   "decoder_start_token_id": 50258,
   "dropout": 0.0,
-  "encoder_attention_heads": 6,
-  "encoder_ffn_dim": 1536,
+  "encoder_attention_heads": 12,
+  "encoder_ffn_dim": 3072,
   "encoder_layerdrop": 0.0,
-  "encoder_layers": 4,
+  "encoder_layers": 12,
   "eos_token_id": 50257,
   "forced_decoder_ids": [
     [
@@ -52,7 +52,7 @@
   "max_target_positions": 448,
   "median_filter_width": 7,
   "model_type": "whisper",
-  "num_hidden_layers": 4,
+  "num_hidden_layers": 12,
   "num_mel_bins": 80,
   "pad_token_id": 50257,
   "scale_embedding": false,
@@ -140,8 +140,6 @@
     49870,
     50254,
     50258,
-    50358,
-    50359,
     50360,
     50361,
     50362

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e402f541632d00e2956790750390c78a6094c78d742b62b91187f976d0862c2e
-size 151061672
+oid sha256:cb084c4a2d6e7371af97e661f1cad7df64b4fcced538c2b3425f63767d187259
+size 966995080

runs/Aug15_04-59-35_c17fec40b44e/events.out.tfevents.1723697977.c17fec40b44e.34.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2a1d08c94d9a661721d802e873bf012739da0642557896e7afe9cdc1dedf8e9a
-size 8036
+oid sha256:f494d22491313adcfbc72575cc3b9c5e319d9b868d24b55e8220f9dbe427a0be
+size 9816

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:a58d979b65d06f10ff7201a06c65afc6270fd97fc5e055e99a8af7a35d67c48f
+oid sha256:161cad3ce9212d9981716768b0f7cf7d7baeded183afbddebf680b12dd26e8ad
 size 5304