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

Before Change

                raise ValueError("This model is not a deep clustering model!")
            embedding = output["embedding"]
            embedding_size = embedding.shape[-1]
            embedding = embedding.squeeze(-2)
            embedding = embedding.permute(2, 1, 0, 3)
            embedding = embedding.reshape(-1, embedding_size)
            embedding = embedding.data.cpu().numpy()
        return embedding

After Change

            embedding = output["embedding"]
            // 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