Skip to content

Commit de5dcb9

Browse files
committed
Remove long from make_grid
1 parent 97871cd commit de5dcb9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def make_grid(tensor, nrow=8, padding=2,
2525
if isinstance(tensor, list):
2626
tensorlist = tensor
2727
numImages = len(tensorlist)
28-
size = torch.Size(torch.Size([long(numImages)]) + tensorlist[0].size())
28+
size = torch.Size(torch.Size([numImages]) + tensorlist[0].size())
2929
tensor = tensorlist[0].new(size)
3030
for i in irange(numImages):
3131
tensor[i].copy_(tensorlist[i])

0 commit comments

Comments
 (0)