malarinv/tacotron2
1
0
Fork 0
mirror of https://github.com/malarinv/tacotron2 synced 2026-03-08 01:32:35 +00:00
Code Issues Releases Wiki Activity

stft.py: moving window_sum to cuda if magnitude is cuda

Browse Source
This commit is contained in:
rafaelvalle
2019-03-15 14:36:56 -07:00
parent f2c94d94fd
commit fc0d34cfce
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
stft.py
View File

@@ -124,6 +124,7 @@ class STFT(torch.nn.Module):
np.where(window_sum > tiny(window_sum))[0]) np.where(window_sum > tiny(window_sum))[0])
window_sum = torch.autograd.Variable( window_sum = torch.autograd.Variable(
torch.from_numpy(window_sum), requires_grad=False) torch.from_numpy(window_sum), requires_grad=False)
window_sum = window_sum.cuda() if magnitude.is_cuda else window_sum
inverse_transform[:, :, approx_nonzero_indices] /= window_sum[approx_nonzero_indices] inverse_transform[:, :, approx_nonzero_indices] /= window_sum[approx_nonzero_indices]
# scale by hop ratio # scale by hop ratio