fix(loss): mag_weight term uses log mels instead of linear (+1 more)

[andrewwhitecdw]

This PR addresses 2 issues in loss.py.

Fix 1

fix: mag_weight term uses log mels instead of linear magnitudes

Fix: Replace:

            loss += self.log_weight * self.loss_fn(x_logmels, y_logmels)
            loss += self.mag_weight * self.loss_fn(x_logmels, y_logmels)

with:

            loss += self.log_weight * self.loss_fn(x_logmels, y_logmels)
            loss += self.mag_weight * self.loss_fn(x_mels, y_mels)

Fix 2

fix: self.weight is stored but never applied to the returned loss

Fix: Replace:

        return loss

with:

        return loss * self.weight

Files changed

• loss.py