Automatic Speech Recognition
Transformers
PyTorch
TensorBoard
speech-encoder-decoder
Generated from Trainer
Instructions to use speech-seq2seq/wav2vec2-2-bert-large with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use speech-seq2seq/wav2vec2-2-bert-large with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("automatic-speech-recognition", model="speech-seq2seq/wav2vec2-2-bert-large")# Load model directly from transformers import AutoTokenizer, AutoModelForSpeechSeq2Seq tokenizer = AutoTokenizer.from_pretrained("speech-seq2seq/wav2vec2-2-bert-large") model = AutoModelForSpeechSeq2Seq.from_pretrained("speech-seq2seq/wav2vec2-2-bert-large") - Notebooks
- Google Colab
- Kaggle
Sanchit Gandhi commited on
Commit ·
5377a51
1
Parent(s): bca4b98
update model card README.md
Browse files
README.md
CHANGED
|
@@ -15,8 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 15 |
|
| 16 |
This model was trained from scratch on the librispeech_asr dataset.
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
-
- Loss:
|
| 19 |
-
- Wer:
|
| 20 |
|
| 21 |
## Model description
|
| 22 |
|
|
@@ -35,7 +35,7 @@ More information needed
|
|
| 35 |
### Training hyperparameters
|
| 36 |
|
| 37 |
The following hyperparameters were used during training:
|
| 38 |
-
- learning_rate: 0.
|
| 39 |
- train_batch_size: 8
|
| 40 |
- eval_batch_size: 8
|
| 41 |
- seed: 42
|
|
@@ -51,16 +51,16 @@ The following hyperparameters were used during training:
|
|
| 51 |
|
| 52 |
| Training Loss | Epoch | Step | Validation Loss | Wer |
|
| 53 |
|:-------------:|:-----:|:----:|:---------------:|:------:|
|
| 54 |
-
| 6.
|
| 55 |
-
| 6.
|
| 56 |
-
| 6.
|
| 57 |
-
|
|
| 58 |
-
|
|
| 59 |
-
|
|
| 60 |
-
|
|
| 61 |
-
|
|
| 62 |
-
|
|
| 63 |
-
|
|
| 64 |
|
| 65 |
|
| 66 |
### Framework versions
|
|
|
|
| 15 |
|
| 16 |
This model was trained from scratch on the librispeech_asr dataset.
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
+
- Loss: 6.9670
|
| 19 |
+
- Wer: 1.9878
|
| 20 |
|
| 21 |
## Model description
|
| 22 |
|
|
|
|
| 35 |
### Training hyperparameters
|
| 36 |
|
| 37 |
The following hyperparameters were used during training:
|
| 38 |
+
- learning_rate: 0.0001
|
| 39 |
- train_batch_size: 8
|
| 40 |
- eval_batch_size: 8
|
| 41 |
- seed: 42
|
|
|
|
| 51 |
|
| 52 |
| Training Loss | Epoch | Step | Validation Loss | Wer |
|
| 53 |
|:-------------:|:-----:|:----:|:---------------:|:------:|
|
| 54 |
+
| 6.7599 | 0.28 | 500 | 6.8755 | 1.2551 |
|
| 55 |
+
| 6.5943 | 0.56 | 1000 | 6.7702 | 1.5878 |
|
| 56 |
+
| 6.3146 | 0.84 | 1500 | 6.6981 | 1.6627 |
|
| 57 |
+
| 6.6112 | 1.12 | 2000 | 6.6760 | 1.9853 |
|
| 58 |
+
| 6.6894 | 1.4 | 2500 | 6.6323 | 1.9376 |
|
| 59 |
+
| 6.5525 | 1.68 | 3000 | 6.6185 | 1.9383 |
|
| 60 |
+
| 6.571 | 1.96 | 3500 | 6.6126 | 1.9580 |
|
| 61 |
+
| 6.3363 | 2.24 | 4000 | 6.7869 | 1.9818 |
|
| 62 |
+
| 6.5832 | 2.52 | 4500 | 6.9096 | 2.0025 |
|
| 63 |
+
| 6.3523 | 2.8 | 5000 | 6.9670 | 1.9878 |
|
| 64 |
|
| 65 |
|
| 66 |
### Framework versions
|