mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Increase query-head-dim from 24 to 32.
This commit is contained in:
parent
364a4c3838
commit
423f9e3026
@ -141,7 +141,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--query-head-dim",
|
"--query-head-dim",
|
||||||
type=str,
|
type=str,
|
||||||
default="24",
|
default="32",
|
||||||
help="Query/key dimension per head in encoder stacks: a single int or comma-separated list."
|
help="Query/key dimension per head in encoder stacks: a single int or comma-separated list."
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user