2eb4c5f77bd8da9b1e23851b0acb84543e442953,python/baseline/pytorch/classify/model.py,WordClassifierModelBase,make_input,#WordClassifierModelBase#Any#,61

Before Change

    def make_input(self, batch_dict):
        x = batch_dict["x"]
        xch = batch_dict.get("xch")
        y = batch_dict.get("y")
        lengths = batch_dict.get("lengths")
        if self.gpu:
            x = x.cuda()

After Change

        y = batch_dict.get("y")
        if y is not None:
            y = torch.from_numpy(y)
            if self.gpu is not None:
                y = y.cuda()
            example_dict["y"] = y

        return example_dict

    def forward(self, input):

Instances