This commit is contained in:
erogol 2020-05-12 16:16:55 +02:00
parent 68dbcee746
commit 1cd25ccf0d
1 changed files with 1 additions and 1 deletions

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