Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
57 commits
Select commit Hold shift + click to select a range
47e3097
shorter augmentations in yaml
Feb 8, 2024
5ab888a
layout to 80 char
Feb 8, 2024
a3bf472
listed label replication
Feb 8, 2024
c86d687
listed label replication
Feb 8, 2024
761bf93
listed label replication
Feb 8, 2024
09cfde3
Refact CTC
Feb 8, 2024
e60396f
Refact transducer
Feb 8, 2024
d6a5524
Refact seq2seq
Feb 8, 2024
9daba50
call replicate label instead of duplication
Feb 8, 2024
6bf2361
refactor aishell
Feb 8, 2024
7ec92c5
refactor aishell
Feb 8, 2024
ebae569
CommonLanuageÃ
Feb 8, 2024
088a0eb
fix error + CV CTC
Feb 8, 2024
bfb9bc2
Giga OOF
Feb 8, 2024
21353d5
Giga OOF
Feb 8, 2024
9971121
Giga OOF
Feb 8, 2024
f879302
Giga OOF
Feb 8, 2024
95c5ea4
Giga OOF
Feb 8, 2024
1b24844
Giga OOF
Feb 8, 2024
a5a97aa
Giga OOF
Feb 8, 2024
55904dd
Giga OOF
Feb 8, 2024
7f366bb
Giga OOF
Feb 8, 2024
963bda4
Finishing OOF
Feb 8, 2024
922024a
final touch LULZ
Feb 8, 2024
819f8c8
fix tests
Feb 8, 2024
8ade568
Tests???Ã
Feb 8, 2024
9e73c10
fix augment in some recipes
mravanelli Feb 10, 2024
b2b8f56
merge
Feb 20, 2024
f0e9f6d
Merge branch 'develop' of https://github.com/TParcollet/speechbrain-r…
Feb 20, 2024
ae6d106
voxpopuli recipe transducer recipe
Feb 20, 2024
65fd341
add recipe testing
Feb 20, 2024
c5e6ead
fix tests
Feb 20, 2024
4cfb5ba
Readme
Feb 20, 2024
92af50d
move readme
Feb 20, 2024
9f158c2
Ãaled chui con
Feb 20, 2024
03b2df2
test fixing
Feb 20, 2024
ea8615a
fix tests?
Feb 20, 2024
e99a7be
If not fixed, I dead
Feb 20, 2024
4d2b857
bon bah I dead
Feb 20, 2024
932a9e9
new auugment strategy for better WER
Feb 21, 2024
9b1f373
fix those mada dukin tests
Feb 21, 2024
18b2501
best recipe so far with test fix
Feb 22, 2024
afd37a1
Merge branch 'develop' of https://github.com/speechbrain/speechbrain …
Feb 22, 2024
331ff7d
Merge branch 'develop' of https://github.com/speechbrain/speechbrain …
Feb 26, 2024
81db8cc
Merge branch 'develop' of https://github.com/speechbrain/speechbrain …
Feb 28, 2024
9ba61e6
Merge branch 'develop' of https://github.com/speechbrain/speechbrain …
Mar 2, 2024
6d17536
Merge branch 'develop' of https://github.com/speechbrain/speechbrain …
Mar 18, 2024
f947231
fix all comments
Mar 18, 2024
017631a
small bug fixing
Mar 18, 2024
f169687
new results
Mar 19, 2024
56b5d3c
Merge branch 'develop' of https://github.com/speechbrain/speechbrain …
Mar 19, 2024
f44f8ca
add model link
Adel-Moumen Mar 20, 2024
bce63ad
inference csv link
Adel-Moumen Mar 20, 2024
d6f54f7
Merge remote-tracking branch 'origin/develop' into voxpopuli_transducer
Adel-Moumen Mar 20, 2024
904b1b4
Merge remote-tracking branch 'speechbrain/develop' into voxpopuli_tra…
Adel-Moumen Mar 20, 2024
6d5cdaf
fix cspell
Adel-Moumen Mar 20, 2024
74c74d1
ending line
Adel-Moumen Mar 20, 2024
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion .dict-speechbrain.txt
Original file line number Diff line number Diff line change
Expand Up @@ -1166,4 +1166,6 @@ noviembre
Politecnica
quelques
Université
Università
Università
AUJOURD'HUI
AUJOURD
8 changes: 5 additions & 3 deletions recipes/LibriSpeech/ASR/transducer/train.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,10 @@
language model.

To run this recipe, do the following:
> python train.py hparams/train.yaml
> python train.py hparams/conformer_transducer.yaml

With the default hyperparameters, the system employs a CRDNN encoder.
The decoder is based on a standard GRU. Beamsearch coupled with a RNN
With the default hyperparameters, the system employs a conformer encoder.
The decoder is based on a standard LSTM. Beamsearch coupled with a RNN
language model is used on the top of decoder probabilities.

The neural network is trained on both CTC and negative-log likelihood
Expand All @@ -24,6 +24,8 @@


Authors
* Sylvain de Langen 2024
* Titouan Parcollet 2024
* Abdel Heba 2020
* Mirco Ravanelli 2020
* Ju-Chieh Chou 2020
Expand Down
74 changes: 74 additions & 0 deletions recipes/VoxPopuli/ASR/transducer/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,74 @@
# VoxPopuli ASR with Transducers
This folder contains scripts necessary to run an ASR experiment with the VoxPopuli dataset;
Before running this recipe, make sure numba is installed (pip install numba) for faster training!
You can download VoxPopuli at: https://github.com/facebookresearch/voxpopuli

**We only report results for english but you simply need to download a different set to train with a different language!**

# Extra-Dependencies
This recipe supports two implementations of the transducer loss, see `use_torchaudio` arg in the yaml file:
1. Transducer loss from torchaudio (this requires torchaudio version >= 0.10.0).
2. Speechbrain implementation using Numba. To use it, please set `use_torchaudio=False` in the yaml file. This version is implemented within SpeechBrain and allows you to directly access the python code of the transducer loss (and directly modify it if needed).

The Numba implementation is currently enabled by default as the `use_torchaudio` option is incompatible with `bfloat16` training.

Note: Before running this recipe, make sure numba is installed. Otherwise, run:
```
pip install numba
```

# How to run it
```shell
python train.py hparams/conformer_transducer.yaml
```

## Precision Notes
If your GPU effectively supports fp16 (half-precision) computations, it is recommended to execute the training script with the `--precision=fp16` (or `--precision=bf16`) option.
Enabling half precision can significantly reduce the peak VRAM requirements. For example, in the case of the Conformer Transducer recipe trained with Librispeech, the peak VRAM decreases from 39GB to 12GB when using fp16.
According to our tests, the performance is not affected.

# VoxPopuli non-streaming results

Results are reported with beam search but without any language model. Models are
trained with dynamic chunk training but decoding is offline.


| Language | Hyperparams file | Train precision | Dev-clean Greedy | Test-clean Greedy | Model link | GPUs |
|:-------------:|:---------------------------:|:-:| :------:| :-----------:| :------------------:| :------------------:|
| English | conformer_transducer.yaml `streaming: True` | fp16 | 9.80 | 10.18 | [Model link](https://www.dropbox.com/scl/fo/y2if76ut4xur5rg9sszj3/h?rlkey=y8wmip8bd06cb82vm2cvmfaz3&dl=0) |6x A40|


# VoxPopuli streaming results

### WER vs chunk size & left context

The following matrix presents the Word Error Rate (WER%) achieved on the test set with various chunk sizes (in ms).

This is with greedy decoding only.


| | full | cs=32 (1280ms) | 16 (640ms) | 8 (320ms) |
|:-----:|:----:|:-----:|:-----:|:-----:|
| full | 10.18| - | - | - |
| lc=32 | - | 10.88 | 11.39 | 12.37 |

# **About SpeechBrain**
- Website: https://speechbrain.github.io/
- Code: https://github.com/speechbrain/speechbrain/
- HuggingFace: https://huggingface.co/speechbrain/


# **Citing SpeechBrain**
Please, cite SpeechBrain if you use it for your research or business.

```bibtex
@misc{speechbrain,
title={{SpeechBrain}: A General-Purpose Speech Toolkit},
author={Mirco Ravanelli and Titouan Parcollet and Peter Plantinga and Aku Rouhe and Samuele Cornell and Loren Lugosch and Cem Subakan and Nauman Dawalatabad and Abdelwahab Heba and Jianyuan Zhong and Ju-Chieh Chou and Sung-Lin Yeh and Szu-Wei Fu and Chien-Feng Liao and Elena Rastorgueva and François Grondin and William Aris and Hwidong Na and Yan Gao and Renato De Mori and Yoshua Bengio},
year={2021},
eprint={2106.04624},
archivePrefix={arXiv},
primaryClass={eess.AS},
note={arXiv:2106.04624}
}
```
Loading