Update README.md

README.md

@@ -7,25 +7,20 @@ tags:
 - generated_from_trainer
 datasets:
 - mozilla-foundation/common_voice_11_0
+- gigant/romanian_speech_synthesis_0_8_1
 model-index:
-- name: Whisper Small Romanian
+- name: Whisper Medium Romanian
   results: []
 ---
 
-<!-- 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 Romanian
 
-# Whisper Small Romanian
-
-This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the Common Voice 11.0 dataset.
+This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the Common Voice 11.0 dataset, and the Romanian speech synthesis corpus.
 It achieves the following results on the evaluation set:
-- eval_loss: 0.2068
-- eval_wer: 15.7031
-- eval_runtime: 1070.693
-- eval_samples_per_second: 3.604
-- eval_steps_per_second: 0.451
-- epoch: 5.03
-- step: 1000
+- eval_loss: 0.06453
+- eval_wer: 4.717
+- epoch: 7.03
+- step: 3500
 
 ## Model description
 
@@ -45,8 +40,8 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 1e-05
-- train_batch_size: 64
-- eval_batch_size: 8
+- train_batch_size: 32
+- eval_batch_size: 32
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear