Update README.md

README.md

@@ -11,14 +11,15 @@ model-index:
 
 # whisper-small-ko-normalized-1273h
 
-This model is a fine-tuned version of [openai/whisper-small](https://huggingface.co/openai/whisper-medium) on a custom dataset for improving Korean speech recognition.
+This model is a fine-tuned version of [openai/whisper-medium](https://huggingface.co/openai/whisper-medium) on a custom dataset for improving Korean speech recognition.
 It achieves the following results on the evaluation set:
 - Loss: 0.1254
 - Wer: 0.0551
 
 ## Model description
 
-The model was trained to transcript the Korean audio sources into text.
+The model was a fine-tuned version of `openai/whisper-medium` transcript the Korean audio sources into text.
+It was trained on GCP's `a2-highgpu-1g` (a100-40G) for 26 hours with about $90.
 
 ## Intended uses & limitations
 
@@ -40,7 +41,7 @@ Following indicates the hours information for each dastset.
 
 The following hyperparameters were used during training:
 - learning_rate: 1e-05
-- train_batch_size: 32
+- train_batch_size: 24
 - eval_batch_size: 32
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08