Update melgan/utils/stft.py

melgan/utils/stft.py

@@ -91,8 +91,8 @@ class STFT(torch.nn.Module):
 
         # https://github.com/NVIDIA/tacotron2/issues/125
         forward_transform = F.conv1d(
-            input_data.cuda(),
-            Variable(self.forward_basis, requires_grad=False).cuda(),
+            input_data,  # cuda()
+            Variable(self.forward_basis, requires_grad=False),  # cuda()
             stride=self.hop_length,
             padding=0).cpu()
 
@@ -126,7 +126,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
+            # 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