from local

This commit is contained in:
dohe0342 2023-01-09 15:25:16 +09:00
parent 585f346a95
commit 1b91ff52d4
2 changed files with 2 additions and 2 deletions

View File

@ -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.