Unverified Commit 49aeab2d authored by Myle Ott's avatar Myle Ott Committed by GitHub
Browse files

Enforce upper-bound on maximum generation length (#121)

parent cbaf59d4

fairseq/sequence_generator.py

+1 −0
Changes for fairseq/sequence_generator.py: 1 added line, 0 removed lines.
Original line number Diff line number Diff line
@@ -37,6 +37,7 @@ class SequenceGenerator(object): 
        self.beam_size = beam_size

        self.minlen = minlen

        max_decoder_len = min([m.max_decoder_positions() for m in self.models])

        max_decoder_len -= 1  # we define maxlen not including the EOS marker

        self.maxlen = max_decoder_len if maxlen is None else min(maxlen, max_decoder_len)

        self.stop_early = stop_early

        self.normalize_scores = normalize_scores