Merge remote-tracking branch 'origin/develop'

hrnoh24 · Jul 30, 2024 · faa629f · faa629f
2 parents 6e890c9 + 581bf51
commit faa629f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/models/losses/mel_loss.py b/src/models/losses/mel_loss.py
@@ -45,12 +45,12 @@ def forward(self, audio):
  win_length=self.win_length,
  window=self.window,
  center=False,
- return_complex=False,
+ return_complex=True,
  )
- real_part, imag_part = fft.unbind(-1)
- magnitude = torch.sqrt(real_part ** 2 + imag_part ** 2)
+ magnitude = torch.abs(fft)
  mel_output = torch.matmul(self.mel_basis, magnitude)
  log_mel_spec = torch.log10(torch.clamp(mel_output, min=1e-5))
+
  return log_mel_spec
 
 class ReconstructionLoss(nn.Module):