from local
This commit is contained in:
parent
9374e5d07e
commit
70d13296c1
Binary file not shown.
@ -415,7 +415,7 @@ def decode_one_batch(
|
|||||||
return {key: hyps}
|
return {key: hyps}
|
||||||
|
|
||||||
if params.method == "greedy-search":
|
if params.method == "greedy-search":
|
||||||
memory, memory_key_padding_mask
|
#memory, memory_key_padding_mask
|
||||||
att_loss = model.decoder_forward(
|
att_loss = model.decoder_forward(
|
||||||
memory,
|
memory,
|
||||||
momory_key_padding_mask,
|
momory_key_padding_mask,
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user