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>
- 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
|