@torch.no_grad()
def mag(self, noisy, inference_args):
    noisy_complex = self.torch_stft(noisy)
    (noisy_mag, noisy_phase) = mag_phase(noisy_complex)
    enhanced_mag = self.model(noisy_mag.unsqueeze(1)).squeeze(1)
    enhanced = self.torch_istft((enhanced_mag, noisy_phase), length=noisy.size((- 1)), use_mag_phase=True)
    enhanced = enhanced.detach().squeeze(0).cpu().numpy()
    return enhanced
@torch.no_grad()
def mag(self, noisy, inference_args):
    noisy_complex = self.torch_stft(noisy)
    (noisy_mag, noisy_phase) = mag_phase(noisy_complex)
    enhanced_mag = self.model(noisy_mag.unsqueeze(1)).squeeze(1)
    enhanced = self.torch_istft((enhanced_mag, noisy_phase), length=noisy.size((- 1)), use_mag_phase=True)
    enhanced = enhanced.detach().squeeze(0).cpu().numpy()
    return enhanced