from local

This commit is contained in:
dohe0342 2023-02-14 02:44:04 +09:00
parent aac6d1acd6
commit 2980df4f3e
3 changed files with 4 additions and 4 deletions

View File

@ -90,7 +90,7 @@ def add_model_arguments(parser: argparse.ArgumentParser) -> None:
parser.add_argument(
"--num-encoder-layers",
type=int,
default=24,
default=18,
help="Number of conformer encoder layers..",
)
@ -115,21 +115,21 @@ def add_model_arguments(parser: argparse.ArgumentParser) -> None:
parser.add_argument(
"--dim-feedforward",
type=int,
default=1536,
default=1024,
help="Feedforward module dimension of the conformer model.",
)
parser.add_argument(
"--nhead",
type=int,
default=8,
default=4,
help="Number of attention heads in the conformer multiheadattention modules.",
)
parser.add_argument(
"--dim-model",
type=int,
default=384,
default=256,
help="Attention dimension in the conformer model.",
)