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
|