mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-26 10:16:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix style

Browse Source
This commit is contained in:
PingFeng Luo 2021-12-30 18:43:47 +08:00
parent f49de8e8e0
commit 8b8cf6c68e
3 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
egs/aishell/ASR/local/make_syllable_lexicon.py
View File

@ -36,22 +36,22 @@ def process_line(
word_syllables = []
word_syllables_num = 1
inited = False
for char_syllables in pinyins :
for char_syllables in pinyins:
new_char_syllables_num = len(char_syllables)
if not inited and len(char_syllables) :
if not inited and len(char_syllables):
word_syllables = [char_syllables[0]]
inited = True
elif new_char_syllables_num == 1 :
for i in range(word_syllables_num) :
elif new_char_syllables_num == 1:
for i in range(word_syllables_num):
word_syllables[i] += " " + str(char_syllables)
elif new_char_syllables_num > 1 :
elif new_char_syllables_num > 1:
word_syllables = word_syllables * new_char_syllables_num
for pre_index in range(word_syllables_num) :
for expand_index in range(new_char_syllables_num) :
for pre_index in range(word_syllables_num):
for expand_index in range(new_char_syllables_num):
word_syllables[pre_index * new_char_syllables_num + expand_index] += " " + char_syllables[expand_index]
word_syllables_num *= new_char_syllables_num
for word_syallable in word_syllables :
for word_syallable in word_syllables:
print("{} {}".format(chars.strip(), str(word_syallable).strip()))

10
egs/aishell/ASR/tdnn_lstm_ctc/asr_datamodule.py
View File

@ -336,19 +336,21 @@ class AishellAsrDataModule:
)
return test_dl
@lru_cache()
def train_cuts(self) -> CutSet:
logging.info("About to get train cuts")
cuts_train = load_manifest(self.args.manifest_dir / "cuts_train.json.gz")
cuts_train = load_manifest(self.args.manifest_dir /
"cuts_train.json.gz")
return cuts_train
@lru_cache()
def valid_cuts(self) -> CutSet:
logging.info("About to get dev cuts")
return load_manifest(self.args.manifest_dir / "cuts_dev.json.gz")
return load_manifest(self.args.manifest_dir /
"cuts_dev.json.gz")
@lru_cache()
def test_cuts(self) -> List[CutSet]:
logging.info("About to get test cuts")
return load_manifest(self.args.manifest_dir / f"cuts_test.json.gz")
return load_manifest(self.args.manifest_dir /
"cuts_test.json.gz")

2
egs/aishell/ASR/tdnn_lstm_ctc/decode.py
View File

@ -366,6 +366,8 @@ def main():
model.eval()
aishell = AishellAsrDataModule(args)
test_cuts = aishell.test_cuts()
test_dl = aishell.test_dataloaders(test_cuts)
# CAUTION: `test_sets` is for displaying only.
# If you want to skip test-clean, you have to skip
# it inside the for loop. That is, use