From d2b272ab509b974c8287e465744a107b7dbbf60d Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 20 Dec 2022 13:54:06 +0800 Subject: [PATCH] Add back 2 conformer layers in 1st stack. --- egs/librispeech/ASR/pruned_transducer_stateless7/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py index 7a31c4900..c015cddde 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -116,7 +116,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--num-encoder-layers", type=str, - default="2,4,4,4,4,4", + default="4,4,4,4,4,4", help="Number of zipformer encoder layers per stack, comma separated.", )