End of training

README.md

@@ -1,21 +1,36 @@
 ---
+language:
+- en
 license: apache-2.0
 base_model: openai/whisper-small
 tags:
 - generated_from_trainer
+datasets:
+- hf-internal-testing/librispeech_asr_dummy
 metrics:
 - wer
 model-index:
-- name: whisper-small-cpu-finetuned
-  results: []
+- name: Whisper Small CPU Finetuned
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: LibriSpeech ASR Dummy
+      type: hf-internal-testing/librispeech_asr_dummy
+      args: 'config: validation'
+    metrics:
+    - name: Wer
+      type: wer
+      value: 0.0
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# whisper-small-cpu-finetuned
+# Whisper Small CPU Finetuned
 
-This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on an unknown dataset.
+This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the LibriSpeech ASR Dummy dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.8936
 - Wer: 0.0

config.json

@@ -28,6 +28,7 @@
   "forced_decoder_ids": null,
   "init_std": 0.02,
   "is_encoder_decoder": true,
+  "low_cpu_mem_usage": true,
   "mask_feature_length": 10,
   "mask_feature_min_masks": 0,
   "mask_feature_prob": 0.0,