Commit de5dcb93 authored by Adam Paszke's avatar Adam Paszke
Browse files

Remove long from make_grid

parent 97871cd9
......@@ -25,7 +25,7 @@ def make_grid(tensor, nrow=8, padding=2,
if isinstance(tensor, list):
tensorlist = tensor
numImages = len(tensorlist)
size = torch.Size(torch.Size([long(numImages)]) + tensorlist[0].size())
size = torch.Size(torch.Size([numImages]) + tensorlist[0].size())
tensor = tensorlist[0].new(size)
for i in irange(numImages):
tensor[i].copy_(tensorlist[i])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment