from local

This commit is contained in:
dohe0342 2023-02-25 16:05:33 +09:00
parent 3d1ff70da4
commit a66a1fcae0
2 changed files with 1 additions and 1 deletions

View File

@ -415,7 +415,7 @@ def decode_one_batch(
return {key: hyps} return {key: hyps}
if params.method == "greedy-search": if params.method == "greedy-search":
att_loss = mmodel.decoder_forward( att_loss = model.decoder_forward(
encoder_memory, encoder_memory,
memory_mask, memory_mask,
token_ids=unsorted_token_ids, token_ids=unsorted_token_ids,