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:51:33 +09:00
parent a649daa1cd
commit 2b1eb92a98
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

Binary file not shown.

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

@ -137,7 +137,6 @@ def add_model_arguments(parser: argparse.ArgumentParser) -> None:
"--kernel-size", "--kernel-size",
type=int, type=int,
default=15, default=15,
help="Attention dimension in the conformer model.",
) )
@ -881,6 +880,8 @@ def run(rank, world_size, args):
dim_feedforward=params.dim_feedforward, dim_feedforward=params.dim_feedforward,
num_encoder_layers=params.num_encoder_layers, num_encoder_layers=params.num_encoder_layers,
num_decoder_layers=params.num_decoder_layers, num_decoder_layers=params.num_decoder_layers,
cnn_module_kernel=params.kernel_size,
group_num=params.group_num,
) )
logging.info(model) logging.info(model)