Minor fixes.

This commit is contained in:
Fangjun Kuang 2022-03-23 19:11:01 +08:00
parent 055406a67e
commit d21ac737ad

View File

@ -214,12 +214,6 @@ def decode_one_batch(
encoder_out=encoder_out_i,
beam=params.beam_size,
)
elif params.decoding_method == "modified_beam_search":
hyp = modified_beam_search(
model=model,
encoder_out=encoder_out_i,
beam=params.beam_size,
)
else:
raise ValueError(
f"Unsupported decoding method: {params.decoding_method}"