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 8, 2021

Commit

2c62b42

•

1 Parent(s): a6de363

remove transformer interface

Files changed (1) hide show

hyperparams.yaml +5 -2

hyperparams.yaml CHANGED Viewed

@@ -122,10 +122,13 @@ tokenizer: !new:sentencepiece.SentencePieceProcessor
 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>

 asr_model: !new:torch.nn.ModuleList
     - [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
+asr_encoder: !new:speechbrain.nnet.containers.Sequential
+    - !ref <CNN>
+    - !ref <Transformer.encode>
 modules:
    compute_features: !ref <compute_features>
+   asr_encoder: !ref <asr_encoder>
    asr_model: !ref <asr_model>
    normalize: !ref <normalize>
    lm_model: !ref <lm_model>