d477e70d888d8501cc0834c47f399a4f65a439d9,senteval/tools/classifier.py,PyTorchClassifier,trainepoch,#PyTorchClassifier#Any#Any#Any#,90

Before Change

            all_costs = []
            for i in range(0, len(X), self.batch_size):
                // forward
                idx = torch.LongTensor(permutation[i:i + self.batch_size])
                if isinstance(X, torch.cuda.FloatTensor):
                    idx = idx.cuda()
                Xbatch = Variable(X.index_select(0, idx))
                ybatch = Variable(y.index_select(0, idx))

After Change

            all_costs = []
            for i in range(0, len(X), self.batch_size):
                // forward
                idx = torch.from_numpy(permutation[i:i + self.batch_size]).long().to(X.device)

                Xbatch = X[idx]
                ybatch = y[idx]

Instances