Update README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 39 |
|
| 40 |
# Whisper Small GA-EN Speech Translation
|
| 41 |
|
| 42 |
-
This model is a fine-tuned version of openai/whisper-small on the IWSLT-2023, FLEURS, BiteSize, and SpokenWords datasets.
|
| 43 |
The best model checkpoint (this version) based on ChrF is at step 2100, epoch 4.5259, and
|
| 44 |
it achieves the following results on the evaluation set:
|
| 45 |
- Loss: 1.7200
|
|
@@ -58,10 +58,15 @@ More information needed
|
|
| 58 |
|
| 59 |
## Training and evaluation data
|
| 60 |
|
| 61 |
-
|
|
|
|
| 62 |
|
| 63 |
## Training procedure
|
| 64 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
### Training hyperparameters
|
| 66 |
|
| 67 |
The following hyperparameters were used during training:
|
|
|
|
| 39 |
|
| 40 |
# Whisper Small GA-EN Speech Translation
|
| 41 |
|
| 42 |
+
This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-small) on the IWSLT-2023, FLEURS, BiteSize, and SpokenWords datasets.
|
| 43 |
The best model checkpoint (this version) based on ChrF is at step 2100, epoch 4.5259, and
|
| 44 |
it achieves the following results on the evaluation set:
|
| 45 |
- Loss: 1.7200
|
|
|
|
| 58 |
|
| 59 |
## Training and evaluation data
|
| 60 |
|
| 61 |
+
Training: IWSLT-2023 (train+dev), FLEURS, BiteSize, and SpokenWords
|
| 62 |
+
Evaluation: IWSLT-2023 (test)
|
| 63 |
|
| 64 |
## Training procedure
|
| 65 |
|
| 66 |
+
### Hardware:
|
| 67 |
+
|
| 68 |
+
1 NVIDIA A100-SXM4-80GB
|
| 69 |
+
|
| 70 |
### Training hyperparameters
|
| 71 |
|
| 72 |
The following hyperparameters were used during training:
|