Skip to content

GitLab

Unverified Commit 5787787e authored by Soumith Chintala's avatar Soumith Chintala Committed by GitHub
Browse files

fixes for 0.4 (#40)

parent 7236e3f4
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
torchaudio/transforms.py
View file @ 5787787e
...@@ -213,7 +213,7 @@ class SPECTROGRAM(object): ...@@ -213,7 +213,7 @@ class SPECTROGRAM(object):
assert sig.dim() == 2 assert sig.dim() == 2
spec_f = torch.stft(sig, self.ws, self.hop, self.n_fft, spec_f = torch.stft(sig, self.ws, self.hop, self.n_fft,
True, self.window, self.pad) # (c, l, n_fft, 2) True, True, self.window, self.pad) # (c, l, n_fft, 2)
spec_f /= self.window.pow(2).sum().sqrt() spec_f /= self.window.pow(2).sum().sqrt()
spec_f = spec_f.pow(2).sum(-1) # get power of "complex" tensor (c, l, n_fft) spec_f = spec_f.pow(2).sum(-1) # get power of "complex" tensor (c, l, n_fft)
return spec_f if is_variable else spec_f.data return spec_f if is_variable else spec_f.data
...@@ -250,9 +250,9 @@ class F2M(object): ...@@ -250,9 +250,9 @@ class F2M(object):
fb = torch.zeros(n_fft, self.n_mels) fb = torch.zeros(n_fft, self.n_mels)
for m in range(1, self.n_mels + 1): for m in range(1, self.n_mels + 1):
f_m_minus = bins[m - 1] f_m_minus = bins[m - 1].item()
f_m = bins[m] f_m = bins[m].item()
f_m_plus = bins[m + 1] f_m_plus = bins[m + 1].item()
if f_m_minus != f_m: if f_m_minus != f_m:
fb[f_m_minus:f_m, m - 1] = (torch.arange(f_m_minus, f_m) - f_m_minus) / (f_m - f_m_minus) fb[f_m_minus:f_m, m - 1] = (torch.arange(f_m_minus, f_m) - f_m_minus) / (f_m - f_m_minus)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment