Update README.md

README.md

@@ -38,11 +38,11 @@ should probably proofread and complete it, then remove this comment. -->
 # Whisper Medium GA-EN Speech Translation
 
 This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on the IWSLT-2023, FLEURS, BiteSize, SpokenWords, Tatoeba, and Wikimedia dataset.
-It achieves the following results on the evaluation set:
+The best model checkpoint (this version) is at step 1400, epoch 1.84 (4 x 0.46), and it achieves the following results on the evaluation set:
 - Loss: 1.0240
-- Bleu: 32.14
-- Chrf: 50.84
-- Wer: 65.9613
+- Bleu: 33.55
+- Chrf: 50.95
+- Wer: 60.1981
 
 ## Model description
 
@@ -71,6 +71,10 @@ The following hyperparameters were used during training:
 - training_steps: 2000
 - mixed_precision_training: Native AMP
 
+### Hardware
+
+4 x A40 48GB VRAM, with batch size 4 per machine (total: 16)
+
 ### Training results
 
 | Training Loss | Epoch | Step | Bleu  | Chrf  | Validation Loss | Wer      |
@@ -90,11 +94,11 @@ The following hyperparameters were used during training:
 | 1.1818        | 0.43  | 1300 | 31.17 | 48.36 | 1.1304          | 61.6389  |
 | 1.2711        | 0.46  | 1400 | 33.55 | 50.95 | 1.0839          | 60.1981  |
 | 1.1305        | 0.49  | 1500 | 30.37 | 50.78 | 1.0718          | 68.6628  |
-| 1.0544        | 0.53  | 1600 | 1.1109| 26.98 | 48.1            | 73.7506  |
-| 1.125         | 0.56  | 1700 | 1.0709| 30.76 | 50.19           | 61.7740  |
-| 1.1348        | 0.59  | 1800 | 1.0530| 33.71 | 50.6            | 59.9280  |
-| 1.14          | 0.62  | 1900 | 1.0392| 31.45 | 50.16           | 66.9068  |
-| 1.1059        | 0.66  | 2000 | 1.0240| 32.14 | 50.84           | 65.9613  |
+| 1.0544        | 0.53  | 1600 | 26.98 | 48.1  | 1.1109          | 73.7506  |
+| 1.125         | 0.56  | 1700 | 30.76 | 50.19 | 1.0709          | 61.7740  |
+| 1.1348        | 0.59  | 1800 | 33.71 | 50.6  | 1.0530          | 59.9280  |
+| 1.14          | 0.62  | 1900 | 31.45 | 50.16 | 1.0392          | 66.9068  |
+| 1.1059        | 0.66  | 2000 | 32.14 | 50.84 | 1.0240          | 65.9613  |
 
 
 ### Framework versions