mirror of
https://github.com/facebookresearch/fairseq.git
synced 2024-09-11 17:25:31 +03:00
parent
bd4db8fbd0
commit
74efc21403
@ -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)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user