Update README.md
Browse files
README.md
CHANGED
@@ -57,14 +57,29 @@ You can use given example code to load the model.
|
|
57 |
|
58 |
```
|
59 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
60 |
## Evaluation
|
61 |
|
62 |
1. Vietnamese
|
63 |
|
64 |
-
| Model Name | Codebook Size | Dataset test |
|
65 |
-
|
66 |
-
| **IchigoWhisper** | 2561 | viVoice |
|
67 |
-
| Whisper Medium | - | viVoice |
|
68 |
|
69 |
2. English
|
70 |
|
|
|
57 |
|
58 |
```
|
59 |
|
60 |
+
|
61 |
+
## Training Specs
|
62 |
+
|
63 |
+
| **Parameter** | **Value** |
|
64 |
+
|----------------------------|-------------------------|
|
65 |
+
| **Initialization Method** | |
|
66 |
+
| **Epochs** | |
|
67 |
+
| **Global Batch Size** | |
|
68 |
+
| **Learning Rate** | |
|
69 |
+
| **Learning Scheduler** | Cosine |
|
70 |
+
| **Optimizer** | AdamW |
|
71 |
+
| **Warmup Ratio** | |
|
72 |
+
| **Weight Decay** | |
|
73 |
+
| **Max Sequence Length** | |
|
74 |
+
|
75 |
## Evaluation
|
76 |
|
77 |
1. Vietnamese
|
78 |
|
79 |
+
| Model Name | Codebook Size | Dataset test | Test samples | WER |
|
80 |
+
|------------|---------------|--------------|--------------|-----|
|
81 |
+
| **IchigoWhisper** | 2561 | viVoice | 1000 | **11.36** |
|
82 |
+
| Whisper Medium | - | viVoice | 1000 | 18.64 |
|
83 |
|
84 |
2. English
|
85 |
|