bug fix

2026-09-10 11:50:20 +08:00 · 2020-05-12 16:23:32 +02:00
parent 68dbcee746
commit 1cd25ccf0d
1 changed files with 1 additions and 1 deletions
@@ -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