c2684e5cf3719d15ca5a13c1919642f96b55f177,python/baseline/pytorch/classify/model.py,WordClassifierBase,make_input,#WordClassifierBase#Any#,49

Before Change

        x = batch_dict["x"]
        y = batch_dict["y"]
        if type(x) == list:
            x = [torch.autograd.Variable(item.cuda()) for item in x]
        else:
            x = torch.autograd.Variable(x.cuda())
        y = torch.autograd.Variable(y.cuda())
        return x, y

After Change

        xch = batch_dict.get("xch")
        y = batch_dict.get("y")
        lengths = batch_dict.get("lengths")
        if self.gpu:
            x = x.cuda()
            if xch is not None:
                xch = xch.cuda()
            if y is not None:
                y = y.cuda()

        return x, xch, lengths, y

    def forward(self, input):
        // BxTxC

Instances