End of training

README.md

@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [microsoft/speecht5_tts](https://huggingface.co/microsoft/speecht5_tts) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.4032
+- Loss: 0.5406
 
 ## Model description
 
@@ -36,10 +36,10 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 0.0001
-- train_batch_size: 4
+- train_batch_size: 16
 - eval_batch_size: 2
 - seed: 42
-- gradient_accumulation_steps: 8
+- gradient_accumulation_steps: 2
 - total_train_batch_size: 32
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
@@ -51,11 +51,11 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch  | Step | Validation Loss |
 |:-------------:|:------:|:----:|:---------------:|
-| 0.5844        | 0.1803 | 100  | 0.5010          |
-| 0.5228        | 0.3607 | 200  | 0.4566          |
-| 0.4881        | 0.5410 | 300  | 0.4294          |
-| 0.4633        | 0.7214 | 400  | 0.4137          |
-| 0.4515        | 0.9017 | 500  | 0.4032          |
+| 0.6213        | 0.0035 | 100  | 0.6012          |
+| 0.6087        | 0.0069 | 200  | 0.5743          |
+| 0.592         | 0.0104 | 300  | 0.5571          |
+| 0.5708        | 0.0139 | 400  | 0.5512          |
+| 0.5596        | 0.0173 | 500  | 0.5406          |
 
 
 ### Framework versions