Change feedforward dims: increase 1536->1792 for largest ff dim and move it one step later.
This commit is contained in:
parent
31b2a735b8
commit
a52ec3da28
@ -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.",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user