speechbrain
/

asr-transformer-transformerlm-librispeech

Automatic Speech Recognition speechbrain PyTorch

English CTC Attention Transformer hf-asr-leaderboard Eval Results

Model card Files Files and versions Community

Titouan commited on Mar 7, 2021

Commit

bc07a6d

•

1 Parent(s): 481fa97

hparams

Files changed (1) hide show

hyperparams.yaml +2 -5

hyperparams.yaml CHANGED Viewed

@@ -119,16 +119,13 @@ lm_model: !new:speechbrain.lobes.models.transformer.TransformerLM.TransformerLM
 tokenizer: !new:sentencepiece.SentencePieceProcessor
 # Models
-asr_encoder: !new:torch.nn.Sequential
-    - !ref <CNN>
-    - !ref <Transformer>
 asr_model: !new:torch.nn.ModuleList
     - [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
 modules:
    compute_features: !ref <compute_features>
-   asr_encoder: !ref <asr_encoder>
    asr_model: !ref <asr_model>
    normalize: !ref <normalize>
    lm_model: !ref <lm_model>

 tokenizer: !new:sentencepiece.SentencePieceProcessor
 # Models
 asr_model: !new:torch.nn.ModuleList
     - [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
 modules:
    compute_features: !ref <compute_features>
+   pre_transformer: !ref <CNN>
+   transformer: !ref <Transformer>
    asr_model: !ref <asr_model>
    normalize: !ref <normalize>
    lm_model: !ref <lm_model>