47478c01bde169ebb7595bbc0189a751db3e0610,pyannote/audio/labeling/base.py,SequenceLabeling,postprocess_ndarray,#SequenceLabeling#Any#,107

Before Change

            Batch of sequence labelings.
        
        if isinstance(self.model, nn.Module):
            X = Variable(torch.from_numpy(np.rollaxis(np.array(X, dtype=np.float32), 0, 2)))
            if self.gpu:
                prediction = self.model(X.cuda()).data.cpu().numpy()
            else:
                prediction = self.model(X).data.numpy()

After Change

            Batch of sequence labelings.
        
        if isinstance(self.model, nn.Module):
            if not getattr(self.model, "batch_first", True):
                X = np.rollaxis(X, 0, 2)
            X = np.array(X, dtype=np.float32)
            X = Variable(torch.from_numpy(X))

            if self.gpu:
                prediction = self.model(X.cuda()).data.cpu().numpy()
            else:

Instances