sdelangen commited on
Commit
e5bb52b
1 Parent(s): d3c6c0d

Add `model_input_keys`for g2p model

Browse files

See https://github.com/speechbrain/speechbrain/pull/2600 for rationale and details

Files changed (1) hide show
  1. hyperparams.yaml +5 -0
hyperparams.yaml CHANGED
@@ -304,6 +304,11 @@ beam_searcher_valid: !new:speechbrain.decoders.S2SRNNBeamSearcher
304
 
305
  homograph_extractor: !new:speechbrain.lobes.models.g2p.homograph.SubsequenceExtractor
306
 
 
 
 
 
 
307
  model_output_keys:
308
  - p_seq
309
  - char_lens
 
304
 
305
  homograph_extractor: !new:speechbrain.lobes.models.g2p.homograph.SubsequenceExtractor
306
 
307
+ # keys to forward from the encoding pipeline to the model
308
+ model_input_keys:
309
+ - grapheme_encoded
310
+ - word_emb
311
+
312
  model_output_keys:
313
  - p_seq
314
  - char_lens