Commit 0714080b authored by Stephen Roller's avatar Stephen Roller Committed by Myle Ott
Browse files

Sequence generator bug fix.

parent e6d45d5c

fairseq/sequence_generator.py

+2 −2
Original line number Diff line number Diff line
@@ -84,9 +84,9 @@ class SequenceGenerator(object): 
            with torch.no_grad():

                hypos = self.generate(

                    beam_size=beam_size,

                    maxlen=int(maxlen_a*srclen + maxlen_b

                    **net_input),

                    maxlen=int(maxlen_a*srclen + maxlen_b),

                    prefix_tokens=s['target'][:, :prefix_size] if prefix_size > 0 else None,

                    **net_input,

                )

            if timer is not None:

                timer.stop(sum(len(h[0]['tokens']) for h in hypos))