Hi @jBloodless, thanks for sharing. In this case, I suggest changing the return_complex to False in torch.istft.
Also the n_fft and hop_length and window length need to be carefully selected, as torch.istft requires the settings to pass NOLA check to make the signal invertible. Check this doc for more information: Invertibility of overlap-add processing