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

74efc214 · Myle Ott · GitHub · bd4db8fb · 74efc214
Unverified Commit 74efc214 authored Jun 25, 2018 by Myle Ott Committed by GitHub Jun 25, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/utils.py tests/utils.py +1 -1

No files found.
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -154,7 +154,7 @@ class TestIncrementalDecoder(FairseqIncrementalDecoder):
                    probs[:, i, self.dictionary.eos()] = 1.0

        # random attention
-        attn = torch.rand(bbsz, src_len, tgt_len)
+        attn = torch.rand(bbsz, tgt_len, src_len)

        return Variable(probs), Variable(attn)