d477e70d888d8501cc0834c47f399a4f65a439d9,senteval/tools/classifier.py,PyTorchClassifier,prepare_split,#PyTorchClassifier#,39

Before Change

        else:
            trainX = torch.FloatTensor(trainX)
            trainy = torch.LongTensor(trainy)
            devX = torch.FloatTensor(devX)
            devy = torch.LongTensor(devy)

        return trainX, trainy, devX, devy

After Change

            trainX, trainy = X[trainidx], y[trainidx]
            devX, devy = X[devidx], y[devidx]

        device = torch.device("cpu") if self.cudaEfficient else torch.device("cuda")

        trainX = torch.from_numpy(trainX).to(device, dtype=torch.float32)
        trainy = torch.from_numpy(trainy).to(device, dtype=torch.int64)
        devX = torch.from_numpy(devX).to(device, dtype=torch.float32)

Instances