update model card README.md

README.md

@@ -0,0 +1,86 @@
+---
+language:
+- ms_my
+license: apache-2.0
+tags:
+- whisper-event
+- generated_from_trainer
+datasets:
+- google/fleurs
+metrics:
+- wer
+model-index:
+- name: Whisper Medium MS - Augmented
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: google/fleurs
+      type: google/fleurs
+      config: ms_my
+      split: test
+      args: ms_my
+    metrics:
+    - name: Wer
+      type: wer
+      value: 9.578362255965294
+---
+
+<!-- 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 Medium MS - Augmented
+
+This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the google/fleurs dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.2066
+- Wer: 9.5784
+- Cer: 2.8109
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 1e-05
+- train_batch_size: 2
+- eval_batch_size: 2
+- seed: 42
+- gradient_accumulation_steps: 16
+- total_train_batch_size: 32
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_steps: 100
+- training_steps: 1000
+- mixed_precision_training: Native AMP
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Wer     | Cer    |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:------:|
+| 0.0876        | 2.15  | 200  | 0.1949          | 10.3105 | 3.0685 |
+| 0.0064        | 4.3   | 400  | 0.1974          | 9.7004  | 2.9596 |
+| 0.0014        | 6.45  | 600  | 0.2031          | 9.6190  | 2.8955 |
+| 0.001         | 8.6   | 800  | 0.2058          | 9.6055  | 2.8440 |
+| 0.0009        | 10.75 | 1000 | 0.2066          | 9.5784  | 2.8109 |
+
+
+### Framework versions
+
+- Transformers 4.26.0.dev0
+- Pytorch 1.13.0+cu117
+- Datasets 2.7.1.dev0
+- Tokenizers 0.13.2

fine-tune-whisper-non-streaming-ms.ipynb

@@ -1042,7 +1042,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 23,
    "id": "ee8b7b8e-1c9a-4d77-9137-1778a629e6de",
    "metadata": {
     "id": "ee8b7b8e-1c9a-4d77-9137-1778a629e6de",
@@ -1070,8 +1070,8 @@
        "\n",
        "    <div>\n",
        "      \n",
-       "      <progress value='1001' max='1000' style='width:300px; height:20px; vertical-align: middle;'></progress>\n",
-       "      [1000/1000 3:47:15, Epoch 10.75/11]\n",
+       "      <progress value='1000' max='1000' style='width:300px; height:20px; vertical-align: middle;'></progress>\n",
+       "      [1000/1000 4:05:08, Epoch 10/11]\n",
        "    </div>\n",
        "    <table border=\"1\" class=\"dataframe\">\n",
        "  <thead>\n",
@@ -1177,8 +1177,24 @@
       "Configuration saved in ./checkpoint-1000/config.json\n",
       "Model weights saved in ./checkpoint-1000/pytorch_model.bin\n",
       "Feature extractor saved in ./checkpoint-1000/preprocessor_config.json\n",
-      "Feature extractor saved in ./preprocessor_config.json\n"
+      "Feature extractor saved in ./preprocessor_config.json\n",
+      "\n",
+      "\n",
+      "Training completed. Do not forget to share your model on huggingface.co/models =)\n",
+      "\n",
+      "\n",
+      "Loading best model from ./checkpoint-1000 (score: 9.578362255965294).\n"
      ]
+    },
+    {
+     "data": {
+      "text/plain": [
+       "TrainOutput(global_step=1000, training_loss=0.12478019709698857, metrics={'train_runtime': 14718.8594, 'train_samples_per_second': 2.174, 'train_steps_per_second': 0.068, 'total_flos': 3.26797691387904e+19, 'train_loss': 0.12478019709698857, 'epoch': 10.75})"
+      ]
+     },
+     "execution_count": 23,
+     "metadata": {},
+     "output_type": "execute_result"
     }
    ],
    "source": [
@@ -1197,7 +1213,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 24,
    "id": "c704f91e-241b-48c9-b8e0-f0da396a9663",
    "metadata": {
     "id": "c704f91e-241b-48c9-b8e0-f0da396a9663"
@@ -1232,7 +1248,48 @@
    "metadata": {
     "id": "d7030622-caf7-4039-939b-6195cdaa2585"
    },
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "Saving model checkpoint to ./\n",
+      "Configuration saved in ./config.json\n",
+      "Model weights saved in ./pytorch_model.bin\n",
+      "Feature extractor saved in ./preprocessor_config.json\n",
+      "Several commits (2) will be pushed upstream.\n",
+      "The progress bars may be unreliable.\n"
+     ]
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "cfdf5c66dd1c4c61b8e49df34cf219bb",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Upload file pytorch_model.bin:   0%|          | 32.0k/2.85G [00:00<?, ?B/s]"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    },
+    {
+     "data": {
+      "application/vnd.jupyter.widget-view+json": {
+       "model_id": "d3c38f0c5bd843b7a1f6eb6f30797880",
+       "version_major": 2,
+       "version_minor": 0
+      },
+      "text/plain": [
+       "Upload file runs/Dec20_10-49-53_DANDAN/events.out.tfevents.1671504600.DANDAN.793.0: 100%|##########| 12.4k/12.…"
+      ]
+     },
+     "metadata": {},
+     "output_type": "display_data"
+    }
+   ],
    "source": [
     "trainer.push_to_hub(**kwargs)"
    ]