055caff00ce0b411c26a3d37ccc538a5f3f47ab9,nussl/separation/deep/deep_clustering.py,DeepClustering,extract_features,#DeepClustering#,83

Before Change

            embedding = embedding.squeeze(-2)
            embedding = embedding.permute(2, 1, 0, 3)
            embedding = embedding.reshape(-1, embedding_size)
            embedding = embedding.data.cpu().numpy()
        return embedding

    def make_audio_signals(self):
        signals = super().make_audio_signals()

After Change

            // swap back batch and sample dims
            if self.metadata["num_channels"] == 1:
                embedding = embedding.transpose(0, -2)
            embedding = embedding.squeeze(0).transpose(0, 1)
        return embedding.cpu().data.numpy()

Instances