From fb7a967276ad14cb58cbef0d06d5a138b1f6f308 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 13 Jan 2023 17:38:11 +0800 Subject: [PATCH] Increase unmasked dims --- 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 3b668b229..8ebe627a5 100755 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/train.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/train.py @@ -195,7 +195,7 @@ def add_model_arguments(parser: argparse.ArgumentParser): parser.add_argument( "--encoder-unmasked-dim", type=str, - default="192,192,224,224,224,192", + default="192,192,256,256,256,192", help="Unmasked dimensions in the encoders, relates to augmentation during training. " "A single int or comma-separated list. Must be <= each corresponding encoder_dim." )