Skip to content

GitLab

Unverified Commit 74efc214 authored by Myle Ott's avatar Myle Ott Committed by GitHub
Browse files

Fix attention order in unit tests (fixes #195) (#197)

parent bd4db8fb
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/utils.py
View file @ 74efc214
...@@ -154,7 +154,7 @@ class TestIncrementalDecoder(FairseqIncrementalDecoder): ...@@ -154,7 +154,7 @@ class TestIncrementalDecoder(FairseqIncrementalDecoder):
probs[:, i, self.dictionary.eos()] = 1.0 probs[:, i, self.dictionary.eos()] = 1.0
# random attention # random attention
attn = torch.rand(bbsz, src_len, tgt_len) attn = torch.rand(bbsz, tgt_len, src_len)
return Variable(probs), Variable(attn) return Variable(probs), Variable(attn)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment