67f153345af2423690753c09f3bd9537d1a5738b,src/models.py,MultiTaskModel,_seq_gen_forward,#MultiTaskModel#Any#Any#,307

Before Change

        s1, s1_mask = self.sent_encoder(batch["input1"])

        classifier = getattr(self, "%s_mdl" % task.name)
        logits = classifier(s1)
        out["logits"] = logits

        if "labels" in batch:
            labels = batch["labels"]

After Change

        b_size, seq_len = batch["inputs"]["words"].size()
        sent, sent_mask = self.sent_encoder(batch["inputs"])

        if isinstance(task, LanguageModelingTask):
            hid2voc = getattr(self, "%s_hid2voc" % task.name)
            logits = hid2voc(sent)
            logits = logits.view(b_size * seq_len, -1)
        else:
            pass
        out["logits"] = logits

        if "targs" in batch:
            targs = batch["targs"]["words"].view(-1)

Instances