Titouan commited on
Commit
e51c900
1 Parent(s): 2c62b42

remove transformer interface

Browse files
Files changed (1) hide show
  1. hyperparams.yaml +2 -5
hyperparams.yaml CHANGED
@@ -122,13 +122,10 @@ tokenizer: !new:sentencepiece.SentencePieceProcessor
122
  asr_model: !new:torch.nn.ModuleList
123
  - [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
124
 
125
- asr_encoder: !new:speechbrain.nnet.containers.Sequential
126
- - !ref <CNN>
127
- - !ref <Transformer.encode>
128
-
129
  modules:
130
  compute_features: !ref <compute_features>
131
- asr_encoder: !ref <asr_encoder>
 
132
  asr_model: !ref <asr_model>
133
  normalize: !ref <normalize>
134
  lm_model: !ref <lm_model>
122
  asr_model: !new:torch.nn.ModuleList
123
  - [!ref <CNN>, !ref <Transformer>, !ref <seq_lin>, !ref <ctc_lin>]
124
 
 
 
 
 
125
  modules:
126
  compute_features: !ref <compute_features>
127
+ pre_transformer: !ref <CNN>
128
+ transformer: !ref <Transformer>
129
  asr_model: !ref <asr_model>
130
  normalize: !ref <normalize>
131
  lm_model: !ref <lm_model>