Change magnitude calculation in STFTLoss for better interpretability of the eps parameter

csteinmetz1 / auraloss

Collection of audio-focused loss functions in PyTorch

Apache License 2.0

745 stars 67 forks source link

Change magnitude calculation in STFTLoss for better interpretability of the eps parameter #72

Open simonschwaer opened 9 months ago

simonschwaer commented 9 months ago

Previously, small values were clamped inside a torch.sqrt, so that the smallest possible values were sqrt(eps) instead of eps. For example, the default eps=1e-8 led to a smallest value of 1e-4.