Fix a typo.

This commit is contained in:
Fangjun Kuang 2022-02-07 16:32:28 +08:00
parent 3b93761199
commit e614e0bd45

View File

@ -311,7 +311,7 @@ def main():
hyp = beam_search(
model=model, encoder_out=encoder_out_i, beam=params.beam_size
)
elif params.decoding_method == "modified_beam_search":
elif params.method == "modified_beam_search":
hyp = modified_beam_search(
model=model, encoder_out=encoder_out_i, beam=params.beam_size
)