From 208ac69519b1fef7cdfa6a090940680c05e99374 Mon Sep 17 00:00:00 2001 From: qmpzzpmq <405691733@qq.com> Date: Mon, 15 Aug 2022 14:59:57 +0000 Subject: [PATCH] blank line --- egs/aishell/ASR/conformer_ctc/generate_CTC_label.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/egs/aishell/ASR/conformer_ctc/generate_CTC_label.py b/egs/aishell/ASR/conformer_ctc/generate_CTC_label.py index ef04d7f7b..83f74b580 100755 --- a/egs/aishell/ASR/conformer_ctc/generate_CTC_label.py +++ b/egs/aishell/ASR/conformer_ctc/generate_CTC_label.py @@ -90,6 +90,7 @@ def get_params() -> AttributeDict: ) return params + def generate_ctc_label_batch( params: AttributeDict, model: nn.Module, @@ -105,6 +106,7 @@ def generate_ctc_label_batch( nnet_output, memory, memory_key_padding_mask = model(feature, supervisions) return nnet_output + def generate_ctc_label_dataset( dl: torch.utils.data.DataLoader, params: AttributeDict, @@ -125,6 +127,7 @@ def generate_ctc_label_dataset( writer, ) + @torch.no_grad() def main(): parser = get_parser()