From 912adfff7caccf72b5d7f59bf208e6b1751b9459 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 8 Dec 2022 21:11:58 +0800 Subject: [PATCH] Increase all ff dims by 256 --- 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 14934439a..a1d0061ef 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -123,7 +123,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--feedforward-dim", type=str, - default="1536,1536,2048,2048,2048,1536", + default="1792,1792,2304,2304,2304,1792", help="Feedforward dimension of the zipformer encoder layers, per stack, comma separated.", )