Skip to content

Commit ccf1a36

Browse files
fix comment
1 parent 00d4484 commit ccf1a36

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/transforms/v2/functional/_type_conversion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,6 @@ def cvcuda_to_tensor(cvcuda_img: "cvcuda.Tensor") -> torch.Tensor:
5959
cuda_tensor = torch.as_tensor(cvcuda_img.cuda(), device="cuda")
6060
if cvcuda_img.ndim != 4:
6161
raise ValueError(f"Image should be 4 dimensional. Got {cuda_tensor.ndim} dimensions.")
62-
# Convert to NCHW layout from CVCUDA default NHWC
62+
# Convert to NCHW shape from CVCUDA default NHWC
6363
img = cuda_tensor.permute(0, 3, 1, 2)
6464
return img

0 commit comments

Comments
 (0)