7a4f4148317f7274c0c88095c037e93f95b1d00d,python/baseline/pytorch/classify/model.py,LSTMModel,make_input,#LSTMModel#Any#,202

Before Change

            xch = xch[perm_idx]
        if y is not None:
            y = y[perm_idx]
        if self.gpu:
            x = x.cuda()
            if xch is not None:
                xch = xch.cuda()
            if y is not None:
                y = y.cuda()

        if y is not None:
            y = y.contiguous()

        return x, xch, lengths, y

After Change

        return hidden

    def make_input(self, batch_dict):
        inputs = super(LSTMModel).make_input(batch_dict)
        lengths = inputs["lengths"]
        lengths, perm_idx = lengths.sort(0, descending=True)
        for k, value in inputs.items():
            inputs[k] = value[perm_idx]

Instances