[FBcode->GH] Fix to_tensor for accimage backend (#3439)

pytorch · Feb 23, 2021 · 67681a7 · 67681a7
1 parent a7b4bfd
commit 67681a7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchvision/transforms/functional.py b/torchvision/transforms/functional.py
@@ -119,9 +119,9 @@ def to_tensor(pic):
             return img
 
     if accimage is not None and isinstance(pic, accimage.Image):
-        nppic = np.zeros([pic.channels, pic.height, pic.width], dtype=default_float_dtype)
+        nppic = np.zeros([pic.channels, pic.height, pic.width], dtype=np.float32)
         pic.copyto(nppic)
-        return torch.from_numpy(nppic)
+        return torch.from_numpy(nppic).to(dtype=default_float_dtype)
 
     # handle PIL Image
     if pic.mode == 'I':