From a52ec3da28baa688d68b4abf42364a25f962bee1 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sun, 20 Nov 2022 14:24:41 +0800 Subject: [PATCH] Change feedforward dims: increase 1536->1792 for largest ff dim and move it one step later. --- 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 8d1c65362..d364560f3 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -122,7 +122,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--feedforward-dim", type=str, - default="1280,1280,1536,1280,1280,1280", + default="1280,1280,1280,1792,1280,1280", help="Feedforward dimension of the zipformer encoder layers, per stack, comma separated.", )