From 526b5e59a6890139c7e80f0c7e670e74ed9c6225 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Tue, 15 Nov 2022 14:23:26 +0800 Subject: [PATCH] Increase pos-head-dim from 2 to 4. --- 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 4969ed96d..e5ab058fe 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -157,7 +157,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--pos-head-dim", type=str, - default="2", + default="4", help="Positional-encoding dimension per head in encoder stacks: a single int or comma-separated list." )