Update README.md
Browse files
README.md
CHANGED
@@ -86,7 +86,7 @@ More information needed
|
|
86 |
|
87 |
## Training procedure
|
88 |
|
89 |
-
|
90 |
```python
|
91 |
model = Wav2Vec2ForCTC.from_pretrained(
|
92 |
model_name_or_path if not last_checkpoint else last_checkpoint,
|
@@ -100,7 +100,7 @@ model = Wav2Vec2ForCTC.from_pretrained(
|
|
100 |
ctc_zero_infinity = True,
|
101 |
)
|
102 |
```
|
103 |
-
|
104 |
|
105 |
The following hyperparameters were used during training:
|
106 |
- learning_rate: 1e-05
|
@@ -115,7 +115,7 @@ The following hyperparameters were used during training:
|
|
115 |
- training_steps: 2000
|
116 |
- mixed_precision_training: Native AMP
|
117 |
|
118 |
-
|
119 |
|
120 |
| Training Loss | Epoch | Step | Validation Loss | Wer |
|
121 |
|:-------------:|:-----:|:----:|:---------------:|:------:|
|
@@ -141,7 +141,7 @@ The following hyperparameters were used during training:
|
|
141 |
| 0.7618 | 12.5 | 2000 | 0.6728 | 0.3286 |
|
142 |
|
143 |
|
144 |
-
|
145 |
|
146 |
- Transformers 4.35.2
|
147 |
- Pytorch 2.1.0+cu118
|
|
|
86 |
|
87 |
## Training procedure
|
88 |
|
89 |
+
#### Model hyperparameters
|
90 |
```python
|
91 |
model = Wav2Vec2ForCTC.from_pretrained(
|
92 |
model_name_or_path if not last_checkpoint else last_checkpoint,
|
|
|
100 |
ctc_zero_infinity = True,
|
101 |
)
|
102 |
```
|
103 |
+
#### Training hyperparameters
|
104 |
|
105 |
The following hyperparameters were used during training:
|
106 |
- learning_rate: 1e-05
|
|
|
115 |
- training_steps: 2000
|
116 |
- mixed_precision_training: Native AMP
|
117 |
|
118 |
+
#### Training results
|
119 |
|
120 |
| Training Loss | Epoch | Step | Validation Loss | Wer |
|
121 |
|:-------------:|:-----:|:----:|:---------------:|:------:|
|
|
|
141 |
| 0.7618 | 12.5 | 2000 | 0.6728 | 0.3286 |
|
142 |
|
143 |
|
144 |
+
#### Framework versions
|
145 |
|
146 |
- Transformers 4.35.2
|
147 |
- Pytorch 2.1.0+cu118
|