This commit is contained in:
erogol
2020-05-12 16:16:55 +02:00
parent 68dbcee746
commit 1cd25ccf0d

View File

@@ -24,7 +24,7 @@ def text_to_seqvec(text, CONFIG, use_cuda):
def numpy_to_torch(np_array, dtype, cuda=False):
if np_array is None:
return None
tensor = torch.Tensor(np_array, dtype=dtype)
tensor = torch.as_tensor(np_array, dtype=dtype)
if cuda:
return tensor.cuda()
return tensor