f1b02cad435ce673a73fc997dc46d005c81e31c8,nussl/separation/deep_mask_estimation.py,DeepMaskEstimation,run,#DeepMaskEstimation#,63

Before Change

            if "estimates" not in output:
                raise ValueError("This model is not a mask estimation model!")

            _masks = output["estimates"].cpu()

        self.assignments = _masks
        self.num_sources = self.assignments.shape[0]
        self.masks = []

After Change

                raise ValueError("This model is not a mask estimation model!")

            _masks = (output["estimates"] / input_data["magnitude_spectrogram"].unsqueeze(-1)).squeeze(0)
            _masks = _masks.permute(3, 1, 0, 2)
            _masks = _masks.cpu().data.numpy()
        

        self.assignments = _masks

Instances