speechbrainteam sdelangen commited on
Commit
ad0a4a1
1 Parent(s): d3c6c0d

Add `model_input_keys`for g2p model (#4)

Browse files

- Add `model_input_keys`for g2p model (e5bb52b7ca562bdacc71cdef9f2fd525e22f68cc)


Co-authored-by: sdelangen <sdelangen@users.noreply.huggingface.co>

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