Update utils.py

utils.py

@@ -2,10 +2,12 @@ import numpy as np
 from scipy.io.wavfile import read
 import torch
 
+use_cuda = torch.cuda.is_available()
+device = torch.device('cuda' if use_cuda else 'cpu')
 
 def get_mask_from_lengths(lengths):
     max_len = torch.max(lengths).item()
-    ids = torch.arange(0, max_len, out=torch.cuda.LongTensor(max_len))
+    ids = torch.arange(0, max_len, out=torch.cuda.LongTensor(max_len) if use_cuda else torch.LongTensor(max_len))
     mask = (ids < lengths.unsqueeze(1)).bool()
     return mask