Update README.md

README.md

@@ -17,9 +17,9 @@ This repository hosts an Automatic Speech Recognition (ASR) model for the Wolof
 
 ## Model Details
 
-- **Model Base**: Whisper-small
-- **Loss**: 0.123
-- **WER**: 0.17
+- **Model Base**: [wav2vec2-xls-r-300m](https://huggingface.co/facebook/wav2vec2-xls-r-300m)
+- **Loss**: 0.1604
+- **WER**: 0.24
 
 
 ## Dataset
@@ -38,13 +38,13 @@ The training process was adapted from the code in the [Finetune Wa2vec 2.0 For S
 The model was trained with the following configuration:
 
 - **Seed**: 19
-- **Training Batch Size**: 1
+- **Training Batch Size**: 4
 - **Gradient Accumulation Steps**: 8
 - **Number of GPUs**: 2
 
 ### Optimizer : AdamW
 
-- **Learning Rate**: 1e-7
+- **Learning Rate**: 1e-6
 
 ### Scheduler: OneCycleLR