2e90375960c8f10072bbf078e571735f846a2e57,onmt/translate/Translator.py,Translator,__init__,#Translator#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,26

Before Change

        // for debugging
        self.beam_accum = None
        if beam_trace:
            self.beam_accum = {
                "predicted_ids": [],
                "beam_parent_ids": [],
                "scores": [],
                "log_probs": []}

    def translate_batch(self, batch, data):
        
        Translate a batch of sentences.

After Change

                 beam_trace=False,
                 min_length=0,
                 stepwise_penalty=False,
                 block_ngram_repeat=0):
        self.model = model
        self.fields = fields
        self.n_best = n_best
        self.max_length = max_length
        self.global_scorer = global_scorer
        self.copy_attn = copy_attn

Instances