dm_aux
dm_aux copied to clipboard
Allow custom window functions for STFT and ISTFT
For istft and stft, the window_fn argument can now be a callable or a numpy/jax.numpy array. I added tests too.
Motivation: I wanted to be able to do a square-root Hann window like AudioTools can do.
@wanglouis49 Can you please take a look at this and my other PR? Or pass on to someone else at GDM?