from local

This commit is contained in:
dohe0342 2023-02-25 16:06:17 +09:00
parent 9374e5d07e
commit 70d13296c1
2 changed files with 1 additions and 1 deletions

View File

@ -415,7 +415,7 @@ def decode_one_batch(
return {key: hyps}
if params.method == "greedy-search":
memory, memory_key_padding_mask
#memory, memory_key_padding_mask
att_loss = model.decoder_forward(
memory,
momory_key_padding_mask,