update model card README.md

README.md

@@ -0,0 +1,82 @@
+---
+license: apache-2.0
+tags:
+- generated_from_trainer
+datasets:
+- common_voice_11_0
+metrics:
+- wer
+model-index:
+- name: openai/whisper-medium
+  results:
+  - task:
+      name: Automatic Speech Recognition
+      type: automatic-speech-recognition
+    dataset:
+      name: common_voice_11_0
+      type: common_voice_11_0
+      config: cs
+      split: test
+      args: cs
+    metrics:
+    - name: Wer
+      type: wer
+      value: 11.408629675328264
+---
+
+<!-- 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. -->
+
+# openai/whisper-medium
+
+This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the common_voice_11_0 dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.2417
+- Wer: 11.4086
+
+## 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: 32
+- eval_batch_size: 32
+- seed: 42
+- gradient_accumulation_steps: 2
+- total_train_batch_size: 64
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- lr_scheduler_warmup_steps: 500
+- training_steps: 5000
+- mixed_precision_training: Native AMP
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Wer     |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|
+| 0.0105        | 4.24  | 1000 | 0.1973          | 12.6130 |
+| 0.0016        | 8.47  | 2000 | 0.2198          | 11.8985 |
+| 0.0004        | 12.71 | 3000 | 0.2310          | 11.4547 |
+| 0.0003        | 16.95 | 4000 | 0.2380          | 11.4270 |
+| 0.0002        | 21.19 | 5000 | 0.2417          | 11.4086 |
+
+
+### Framework versions
+
+- Transformers 4.26.0.dev0
+- Pytorch 1.13.0+cu117
+- Datasets 2.7.1
+- Tokenizers 0.13.2

run.log

@@ -488,3 +488,12 @@ Dataset common_voice_11_0 downloaded and prepared to /root/.cache/huggingface/da
 {'loss': 0.0002, 'learning_rate': 1.1333333333333336e-07, 'epoch': 20.97}
 {'loss': 0.0002, 'learning_rate': 5.777777777777778e-08, 'epoch': 21.08}
 {'loss': 0.0002, 'learning_rate': 2.2222222222222225e-09, 'epoch': 21.19}
+{'eval_loss': 0.24174600839614868, 'eval_wer': 11.408629675328264, 'eval_runtime': 9721.5421, 'eval_samples_per_second': 0.793, 'eval_steps_per_second': 0.025, 'epoch': 21.19}
+{'train_runtime': 142520.4283, 'train_samples_per_second': 2.245, 'train_steps_per_second': 0.035, 'train_loss': 0.026235109101515264, 'epoch': 21.19}
+12/13/2022 09:40:58 - WARNING - huggingface_hub.repository - Several commits (2) will be pushed upstream.
+12/13/2022 09:40:58 - WARNING - huggingface_hub.repository - The progress bars may be unreliable.
+12/13/2022 09:52:06 - WARNING - huggingface_hub.repository - remote: Scanning LFS files for validity, may be slow...        
+remote: LFS file scan complete.        
+To https://huggingface.co/mikr/whisper-medium-czech-cv11
+   379048f..ae9f79e  main -> main
+