csteinmetz1 / auraloss

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

Cleanup #19

Closed csteinmetz1 closed 3 years ago

csteinmetz1 commented 3 years ago

Small refactor after the addition of the support for linear STFT. This fixes issue with the reduction in STFTLoss and also adding support **kwargs support for all the modules that use the STFTLoss module. This also fixes a bug where mel filterbanks needed to be manually moved to the correct device. See the updates in the README for usage.