malarinv
/
tacotron2
mirror of https://github.com/malarinv/tacotron2
1
0
Fork 0
Code Issues Releases Wiki Activity

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

experiments
rafaelvalle 2019-03-15 14:36:56 -07:00
parent f2c94d94fd
commit fc0d34cfce
1 changed files with 1 additions and 0 deletions
Show Stats 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])
window_sum = torch.autograd.Variable(
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]
# scale by hop ratio