mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/tedlium2/ASR/conformer_ctc3/.conformer.py.swp
View File

Binary file not shown.

BIN
egs/tedlium2/ASR/conformer_ctc3/.train.py.swp
View File

Binary file not shown.

8
egs/tedlium2/ASR/conformer_ctc3/train.py
View File

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