mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-15 13:02:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

blank line

Browse Source
This commit is contained in:
qmpzzpmq 2022-08-15 14:59:57 +00:00
parent bf5c07381f
commit 208ac69519
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/aishell/ASR/conformer_ctc/generate_CTC_label.py
View File

@ -90,6 +90,7 @@ def get_params() -> AttributeDict:
) )
return params return params
def generate_ctc_label_batch( def generate_ctc_label_batch(
params: AttributeDict, params: AttributeDict,
model: nn.Module, model: nn.Module,
@ -105,6 +106,7 @@ def generate_ctc_label_batch(
nnet_output, memory, memory_key_padding_mask = model(feature, supervisions) nnet_output, memory, memory_key_padding_mask = model(feature, supervisions)
return nnet_output return nnet_output
def generate_ctc_label_dataset( def generate_ctc_label_dataset(
dl: torch.utils.data.DataLoader, dl: torch.utils.data.DataLoader,
params: AttributeDict, params: AttributeDict,
@ -125,6 +127,7 @@ def generate_ctc_label_dataset(
writer, writer,
) )
@torch.no_grad() @torch.no_grad()
def main(): def main():
parser = get_parser() parser = get_parser()