From 423f9e3026516d9fbd34271753e062a51935ac6c Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 9 Nov 2022 13:28:29 +0800 Subject: [PATCH] Increase query-head-dim from 24 to 32. --- 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 9cafe7047..23619b258 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -141,7 +141,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--query-head-dim", type=str, - default="24", + default="32", help="Query/key dimension per head in encoder stacks: a single int or comma-separated list." )