csteinmetz1 / auraloss

Collection of audio-focused loss functions in PyTorch
Apache License 2.0
717 stars 66 forks source link

fixing a bug with numerical stability in stft magnitude computation #12

Closed csteinmetz1 closed 3 years ago

csteinmetz1 commented 3 years ago

This fixes a bug when using torch.abs() on the complex STFT so that with small values there is no instability.