Add `model_input_keys`for g2p model
Browse filesSee https://github.com/speechbrain/speechbrain/pull/2600 for rationale and details
- 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
|