mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add an assertion

Browse Source
This commit is contained in:
Piotr Żelasko 2022-01-21 17:16:49 -05:00
parent 3d109b121d
commit f28951f2b6
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/conformer_ctc/train.py
View File

@ -588,6 +588,10 @@ def run(rank, world_size, args):
eos_token="<sos/eos>", eos_token="<sos/eos>",
) )
elif "lang_phone" in params.lang_dir: elif "lang_phone" in params.lang_dir:
assert params.att_rate == 0, (
"Attention decoder training does not support phone lang dirs at this time due to a missing "
"<sos/eos> symbol. Set --att-rate=0 for pure CTC training when using a phone-based lang dir."
)
graph_compiler = CtcTrainingGraphCompiler( graph_compiler = CtcTrainingGraphCompiler(
lexicon, lexicon,
device=device, device=device,