mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
585f346a95
commit
1b91ff52d4
Binary file not shown.
@ -294,14 +294,14 @@ def add_model_arguments(parser: argparse.ArgumentParser):
|
|||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--decoder-dim",
|
"--decoder-dim",
|
||||||
type=int,
|
type=int,
|
||||||
default=512,
|
default=768,
|
||||||
help="Embedding dimension in the decoder model.",
|
help="Embedding dimension in the decoder model.",
|
||||||
)
|
)
|
||||||
|
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--joiner-dim",
|
"--joiner-dim",
|
||||||
type=int,
|
type=int,
|
||||||
default=512,
|
default=768,
|
||||||
help="""Dimension used in the joiner model.
|
help="""Dimension used in the joiner model.
|
||||||
Outputs from the encoder and decoder model are projected
|
Outputs from the encoder and decoder model are projected
|
||||||
to this dimension before adding.
|
to this dimension before adding.
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user