fix decode.py

This commit is contained in:
yaozengwei 2022-05-15 16:25:34 +08:00
parent 27558858a3
commit 9828b8f628

View File

@ -75,7 +75,7 @@ from beam_search import (
greedy_search_batch,
modified_beam_search,
)
from train import get_params, get_transducer_model
from train import add_model_arguments, get_params, get_transducer_model
from icefall.checkpoint import (
average_checkpoints,
@ -212,6 +212,8 @@ def get_parser():
Used only when --decoding_method is greedy_search""",
)
add_model_arguments(parser)
return parser