maum-ai / phaseaug

ICASSP 2023 Accepted
https://maum-ai.github.io/phaseaug/
BSD 3-Clause "New" or "Revised" License
188 stars 14 forks source link

Changed in version 2.0: Real datatype inputs are no longer supported. #8

Closed Aloento closed 1 year ago

Aloento commented 1 year ago

https://pytorch.org/docs/stable/generated/torch.istft.html#:~:text=datatype%20inputs%20are-,no%20longer%20supported,-.%20Input%20must%20now

in PhaseAug -> forward9(_sync)

x_aug = torch.istft(

RuntimeError: istft requires a complex-valued input tensor matching the output from stft with return_complex=True.

Maybe:

X_aug = X * torch.complex(phi.cos(), -phi.sin())

junjun3518 commented 1 year ago

Yes, I plan to update it in the near future as mentioned in my todo list: PyTorch 2.0 is released, need to modify STFT and iSTFT for complex support.

However, it will take some time since I only have machines with older versions of drivers.

junjun3518 commented 1 year ago

Updated! Now phaseaug==1.0.0 is available for both pytorch>=2.0.0 and pytorch>=1.7.0