mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Increase pos-head-dim from 2 to 4.
This commit is contained in:
parent
fc74ff63fb
commit
526b5e59a6
@ -157,7 +157,7 @@ def add_model_arguments(parser: argparse.ArgumentParser):
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--pos-head-dim",
|
"--pos-head-dim",
|
||||||
type=str,
|
type=str,
|
||||||
default="2",
|
default="4",
|
||||||
help="Positional-encoding dimension per head in encoder stacks: a single int or comma-separated list."
|
help="Positional-encoding dimension per head in encoder stacks: a single int or comma-separated list."
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user