Increase dim of middle satck from 512 to 768

This commit is contained in:
Daniel Povey 2023-06-19 02:16:59 +08:00
parent 01ed3bbcc4
commit c7e8a7349d

View File

@ -143,7 +143,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
parser.add_argument( parser.add_argument(
"--encoder-dim", "--encoder-dim",
type=str, type=str,
default="384,512,512,512,512,512,384", default="384,512,512,768,512,512,384",
help="Embedding dimension in encoder stacks: a single int or comma-separated list." help="Embedding dimension in encoder stacks: a single int or comma-separated list."
) )