From c4f669ef00a11cfbd9a31ee6618510a7246a297d Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 4 Apr 2023 14:41:23 +0800 Subject: [PATCH] Increase feedforward dims and num layers --- egs/librispeech/ASR/pruned_transducer_stateless7/train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py index b6ee6cdd4..33e376055 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -124,7 +124,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--num-encoder-layers", type=str, - default="1,2,2,3,2,2", + default="1,2,3,3,2,2", help="Number of zipformer encoder layers per stack, comma separated.", ) @@ -140,7 +140,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--feedforward-dim", type=str, - default="384,512,1024,1536,1024,512", + default="512,768,1024,1536,1024,768", help="Feedforward dimension of the zipformer encoder layers, per stack, comma separated.", )