Update README.md

README.md

@@ -12,15 +12,18 @@ OpenAI의 whisper-base 모델을 아래 데이터셋으로 학습한 모델입
 - 저음질 전화망 음성인식 데이터 (https://www.aihub.or.kr/aihubdata/data/view.do?currMenu=115&topMenu=100&dataSetSn=571)
 - 방송 콘텐츠 대화체 음성인식 데이터 (https://www.aihub.or.kr/aihubdata/data/view.do?dataSetSn=463)
 
+Training setup
+
 ```
-train_steps: 20000
-warmup_steps: 2000
+train_steps: 50000
+warmup_steps: 500
 lr scheduler: linear warmup cosine decay
 max learning rate: 1e-4
-batch size: 256
+batch size: 1024
 max_grad_norm: 1.0
 adamw_beta1: 0.9
 adamw_beta2: 0.98
+adamw_eps: 1e-6
 ```
 
 ### Evaluation
@@ -30,14 +33,16 @@ https://github.com/rtzr/Awesome-Korean-Speech-Recognition
 위 레포지토리에서 주요 영역별 회의 음성을 제외한 테스트셋 결과입니다. 아래 테이블에서 whisper_base_komixv2가 본 모델 성능입니다.
 
 
-|         Model         | cv_15_ko | fleurs_ko | kcall_testset | kconf_test | kcounsel_test | klec_testset | kspon_clean | kspon_other |
-|-----------------------|----------|-----------|---------------|------------|---------------|--------------|-------------|-------------|
-| whisper_base         | 21.16    | 11.89     | 42.56         | 27.62      | 22.24         | 28.65        | 30.41       | 27.02       |
-| whisper_base_komix   | 15.42    | 7.16      | 20.86         | 14.24      | 12.64         | 13.44        | 12.26       | 12.12       |
-|   whisper_base_komixv2   |  13.04   |    7.04   |     10.54     |    13.1    |     10.65     |    12.99     |    12.44    |    12.56    |
-| whisper_large_v3     | 5.11     | 3.72      | 5.45          | 9.35       | 3.83          | 8.46         | 15.08       | 12.89       |
-| whisper_turbo        | 5.38     | 3.95      | 5.89          | 9.77       | 4.21          | 9.27         | 16.49       | 13.54       |
-| whisper_turbo_lora   | 6.25     | 4.0       | 6.51          | 9.94       | 5.05          | 8.84         | 9.35        | 9.29        |
+|          Model           | cv_15_ko | fleurs_ko | kcall_testset | kconf_test | kcounsel_test | klec_testset | kspon_clean | kspon_other | Average |
+|--------------------------|----------|-----------|---------------|------------|---------------|--------------|-------------|-------------|---------|
+| whisper_base            |  21.16   |   11.89   |     42.56     |   27.62    |     22.24     |    28.65     |    30.41    |    27.02    |  26.44  |
+| whisper_base_komix      |  15.42   |    7.16   |     20.86     |   14.24    |     12.64     |    13.44     |    12.26    |    12.12    |  13.52  |
+| whisper_base_komixv2    |  10.27   |    5.14   |      6.23     |   10.86    |      7.01     |    10.38     |     9.98    |     9.99    |   8.73  |
+| whisper_base_komixv2_phn|  12.81   |    8.27   |      9.5      |   13.26    |     11.33     |    14.24     |    13.11    |     13.3    |  11.98  |
+| whisper_large_v2        |   6.58   |    3.74   |     22.26     |   13.88    |      8.95     |    13.84     |    15.51    |     13.6    |  12.29  |
+| whisper_large_v3        |   5.11   |    3.72   |      5.45     |    9.35    |      3.83     |     8.46     |    15.08    |    12.89    |   7.99  |
+| whisper_large_v3_turbo  |   5.38   |    3.95   |      5.89     |    9.77    |      4.21     |     9.27     |    16.49    |    13.54    |   8.56  |
+
 
 ### Acknowledgement
 - 본 모델은 구글의 TRC 프로그램의 지원으로 학습했습니다.