From f8210e1d809d8c772b0e0e79daa860f660a60352 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 5 Nov 2022 13:50:14 +0800 Subject: [PATCH] Reduce feedforward dim of the 4th and 5th encoder stacks. # Conflicts: # egs/librispeech/ASR/pruned_transducer_stateless7/train.py --- 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 5019a7f9f..79860ffbc 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -120,7 +120,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--feedforward-dim", type=str, - default="1024,1024,1536,1536,1536,1024", + default="1024,1024,1536,1024,1024,1024", help="Feedforward dimension of the zipformer encoder layers, per stack, comma separated.", )