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

tests/utils.py

+1 −1
Changes for tests/utils.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -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)