bda270d6255bca6c9518f4f6e8d2cdee8cd6a40c,parlai/agents/seq2seq/modules.py,Encoder,forward,#Encoder#Any#,294

Before Change

        // embed input tokens
        xes = self.dropout(self.lt(xs))
        try:
            x_lens = [x for x in torch.sum((xs > 0).int(), dim=1).data]
            xes = pack_padded_sequence(xes, x_lens, batch_first=True)
            packed = True
        except ValueError:
            // packing failed, don"t pack then

After Change

        // embed input tokens
        xes = self.dropout(self.lt(xs))
        attn_mask = xs.ne(0)
        try:
            x_lens = torch.sum(attn_mask.int(), dim=1)
            xes = pack_padded_sequence(xes, x_lens, batch_first=True)
            packed = True
        except ValueError:
            // packing failed, don"t pack then

Instances