from local

This commit is contained in:
dohe0342 2023-01-24 16:29:39 +09:00
parent da10455787
commit 4ad2861964
2 changed files with 1 additions and 3 deletions

View File

@ -71,9 +71,6 @@ def get_parser():
"--group-num", "--group-num",
type=int, type=int,
default=0, default=0,
help="""It specifies the checkpoint to use for decoding.
Note: Epoch counts from 1.
You can specify --avg to use more checkpoints for model averaging.""",
) )
parser.add_argument( parser.add_argument(
@ -851,6 +848,7 @@ def main():
subsampling_factor=params.subsampling_factor, subsampling_factor=params.subsampling_factor,
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,
group_num=params.group_num,
) )
if not params.use_averaged_model: if not params.use_averaged_model: