mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
fix style
This commit is contained in:
parent
f49de8e8e0
commit
8b8cf6c68e
@ -36,22 +36,22 @@ def process_line(
|
|||||||
word_syllables = []
|
word_syllables = []
|
||||||
word_syllables_num = 1
|
word_syllables_num = 1
|
||||||
inited = False
|
inited = False
|
||||||
for char_syllables in pinyins :
|
for char_syllables in pinyins:
|
||||||
new_char_syllables_num = len(char_syllables)
|
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]]
|
word_syllables = [char_syllables[0]]
|
||||||
inited = True
|
inited = True
|
||||||
elif new_char_syllables_num == 1 :
|
elif new_char_syllables_num == 1:
|
||||||
for i in range(word_syllables_num) :
|
for i in range(word_syllables_num):
|
||||||
word_syllables[i] += " " + str(char_syllables)
|
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
|
word_syllables = word_syllables * new_char_syllables_num
|
||||||
for pre_index in range(word_syllables_num) :
|
for pre_index in range(word_syllables_num):
|
||||||
for expand_index in range(new_char_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[pre_index * new_char_syllables_num + expand_index] += " " + char_syllables[expand_index]
|
||||||
word_syllables_num *= new_char_syllables_num
|
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()))
|
print("{} {}".format(chars.strip(), str(word_syallable).strip()))
|
||||||
|
|
||||||
|
|
||||||
|
@ -336,19 +336,21 @@ class AishellAsrDataModule:
|
|||||||
)
|
)
|
||||||
return test_dl
|
return test_dl
|
||||||
|
|
||||||
|
|
||||||
@lru_cache()
|
@lru_cache()
|
||||||
def train_cuts(self) -> CutSet:
|
def train_cuts(self) -> CutSet:
|
||||||
logging.info("About to get train cuts")
|
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
|
return cuts_train
|
||||||
|
|
||||||
@lru_cache()
|
@lru_cache()
|
||||||
def valid_cuts(self) -> CutSet:
|
def valid_cuts(self) -> CutSet:
|
||||||
logging.info("About to get dev cuts")
|
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()
|
@lru_cache()
|
||||||
def test_cuts(self) -> List[CutSet]:
|
def test_cuts(self) -> List[CutSet]:
|
||||||
logging.info("About to get test cuts")
|
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")
|
||||||
|
@ -366,6 +366,8 @@ def main():
|
|||||||
model.eval()
|
model.eval()
|
||||||
|
|
||||||
aishell = AishellAsrDataModule(args)
|
aishell = AishellAsrDataModule(args)
|
||||||
|
test_cuts = aishell.test_cuts()
|
||||||
|
test_dl = aishell.test_dataloaders(test_cuts)
|
||||||
# CAUTION: `test_sets` is for displaying only.
|
# CAUTION: `test_sets` is for displaying only.
|
||||||
# If you want to skip test-clean, you have to skip
|
# If you want to skip test-clean, you have to skip
|
||||||
# it inside the for loop. That is, use
|
# it inside the for loop. That is, use
|
||||||
|
Loading…
x
Reference in New Issue
Block a user