Training in progress, step 1000

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "openai/whisper-small",
+  "_name_or_path": "openai/whisper-base",
   "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": 768,
-  "decoder_attention_heads": 12,
-  "decoder_ffn_dim": 3072,
+  "d_model": 512,
+  "decoder_attention_heads": 8,
+  "decoder_ffn_dim": 2048,
   "decoder_layerdrop": 0.0,
-  "decoder_layers": 12,
+  "decoder_layers": 6,
   "decoder_start_token_id": 50258,
   "dropout": 0.0,
-  "encoder_attention_heads": 12,
-  "encoder_ffn_dim": 3072,
+  "encoder_attention_heads": 8,
+  "encoder_ffn_dim": 2048,
   "encoder_layerdrop": 0.0,
-  "encoder_layers": 12,
+  "encoder_layers": 6,
   "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": 12,
+  "num_hidden_layers": 6,
   "num_mel_bins": 80,
   "pad_token_id": 50257,
   "scale_embedding": false,
@@ -140,6 +140,8 @@
     49870,
     50254,
     50258,
+    50358,
+    50359,
     50360,
     50361,
     50362

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:db3026e3b66ca4a9eee7434b8a130b443ee521640a8be373ae616838875ee6fa
-size 966995080
+oid sha256:ea0114db8bc0e54873499c406933b21a1dc6fc5876e2713d8701c9297f31c896
+size 290403936

runs/Jul29_17-31-46_e9b374fb131c/events.out.tfevents.1722274315.e9b374fb131c.2064.2

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9ba6e5a6758a04b9d09f6d8e6b148b786aa0e2ba658492fe300cf8b4d358cfa1
-size 14628
+oid sha256:4332e8698862e87980eb84516d7d83fa05541d3427af315a8b0b40696cb502ac
+size 16988

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:956e340eb138b3f2da69e2e0232ed2ea7a897075cc8e11e3639d5a09a541b39d
+oid sha256:236d12b487704ebf0de23dcf1e620d3905d1f9062845e589f253d4f5966e486a
 size 5240