Model save

README.md

@@ -22,7 +22,7 @@ model-index:
     metrics:
     - name: Wer
       type: wer
-      value: 2.3166023166023164
+      value: 80.0
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -32,8 +32,8 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the audiofolder dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0000
-- Wer: 2.3166
+- Loss: 1.9637
+- Wer: 80.0
 
 ## Model description
 
@@ -52,28 +52,22 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1e-05
+- learning_rate: 1e-06
 - train_batch_size: 16
 - eval_batch_size: 8
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
 - lr_scheduler_warmup_steps: 500
-- training_steps: 4000
+- training_steps: 1000
 - mixed_precision_training: Native AMP
 
 ### Training results
 
-| Training Loss | Epoch  | Step | Validation Loss | Wer     |
-|:-------------:|:------:|:----:|:---------------:|:-------:|
-| 0.0001        | 500.0  | 500  | 0.0001          | 37.0656 |
-| 0.0001        | 1000.0 | 1000 | 0.0001          | 37.0656 |
-| 0.0001        | 1500.0 | 1500 | 0.0001          | 2.3166  |
-| 0.0           | 2000.0 | 2000 | 0.0000          | 2.3166  |
-| 0.0           | 2500.0 | 2500 | 0.0000          | 2.3166  |
-| 0.0           | 3000.0 | 3000 | 0.0000          | 2.3166  |
-| 0.0           | 3500.0 | 3500 | 0.0000          | 2.3166  |
-| 0.0           | 4000.0 | 4000 | 0.0000          | 2.3166  |
+| Training Loss | Epoch | Step | Validation Loss | Wer   |
+|:-------------:|:-----:|:----:|:---------------:|:-----:|
+| 0.7875        | 35.71 | 500  | 2.2491          | 514.0 |
+| 0.0077        | 71.43 | 1000 | 1.9637          | 80.0  |
 
 
 ### Framework versions