Change feedforward dims: increase 1536->1792 for largest ff dim and move it one step later.

This commit is contained in:
Daniel Povey 2022-11-20 14:24:41 +08:00
parent 31b2a735b8
commit a52ec3da28

View File

@ -122,7 +122,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
parser.add_argument( parser.add_argument(
"--feedforward-dim", "--feedforward-dim",
type=str, type=str,
default="1280,1280,1536,1280,1280,1280", default="1280,1280,1280,1792,1280,1280",
help="Feedforward dimension of the zipformer encoder layers, per stack, comma separated.", help="Feedforward dimension of the zipformer encoder layers, per stack, comma separated.",
) )