Update README.md

README.md

@@ -13,7 +13,7 @@ widget:
     src: >-
       https://huggingface.co/m3hrdadfi/wav2vec2-large-xlsr-persian-v3/resolve/main/sample5168.flac
 model-index:
-- name: wav2vec2-large-xlsr-persian-asr-shemo_me7494
+- name: wav2vec2-large-xlsr-persian-shemo
   results:
       - task:
           name: Speech Recognition
@@ -86,7 +86,21 @@ More information needed
 
 ## Training procedure
 
-### Training hyperparameters
+##### Model hyperparameters
+```python
+model = Wav2Vec2ForCTC.from_pretrained(
+    model_name_or_path if not last_checkpoint else last_checkpoint,
+    # hp-mehrdad: Hyperparams of 'm3hrdadfi/wav2vec2-large-xlsr-persian-v3'
+    attention_dropout = 0.05316,
+    hidden_dropout    = 0.01941,
+    feat_proj_dropout = 0.01249,
+    mask_time_prob    = 0.04529,
+    layerdrop         = 0.01377,
+    ctc_loss_reduction = 'mean',
+    ctc_zero_infinity = True,
+)
+```
+##### Training hyperparameters
 
 The following hyperparameters were used during training:
 - learning_rate: 1e-05
@@ -101,7 +115,7 @@ The following hyperparameters were used during training:
 - training_steps: 2000
 - mixed_precision_training: Native AMP
 
-### Training results
+##### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | Wer    |
 |:-------------:|:-----:|:----:|:---------------:|:------:|
@@ -127,7 +141,7 @@ The following hyperparameters were used during training:
 | 0.7618        | 12.5  | 2000 | 0.6728          | 0.3286 |
 
 
-### Framework versions
+##### Framework versions
 
 - Transformers 4.35.2
 - Pytorch 2.1.0+cu118