From de1fe9dac42c62e9ebea52c7da508bb324ea18b9 Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Mon, 13 Jun 2022 13:42:12 +0800 Subject: [PATCH] update README.md --- egs/librispeech/ASR/README.md | 1 + .../ASR/conv_emformer_transducer_stateless/test_emformer.py | 4 +--- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/egs/librispeech/ASR/README.md b/egs/librispeech/ASR/README.md index e2aaa9d7e..318d908d1 100644 --- a/egs/librispeech/ASR/README.md +++ b/egs/librispeech/ASR/README.md @@ -23,6 +23,7 @@ The following table lists the differences among them. | `pruned_transducer_stateless5` | Conformer(modified) | Embedding + Conv1d | same as pruned_transducer_stateless4 + more layers + random combiner| | `pruned_transducer_stateless6` | Conformer(modified) | Embedding + Conv1d | same as pruned_transducer_stateless4 + distillation with hubert| | `pruned_stateless_emformer_rnnt2` | Emformer(from torchaudio) | Embedding + Conv1d | Using Emformer from torchaudio for streaming ASR| +| `conv_emformer_transducer_stateless` | Emformer | Embedding + Conv1d | Using Emformer augmented with convolution for streaming ASR + mechanisms in reworked model | The decoder in `transducer_stateless` is modified from the paper diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless/test_emformer.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless/test_emformer.py index f80f4e367..8cde6205b 100644 --- a/egs/librispeech/ASR/conv_emformer_transducer_stateless/test_emformer.py +++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless/test_emformer.py @@ -19,12 +19,10 @@ import torch -from emformer import Emformer, stack_states, unstack_states +from emformer import ConvolutionModule, Emformer, stack_states, unstack_states def test_convolution_module_forward(): - from emformer import ConvolutionModule - B, D = 2, 256 chunk_length = 4 right_context_length = 2