Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
vision
Commits
de5dcb93
Commit
de5dcb93
authored
Apr 02, 2017
by
Adam Paszke
Browse files
Remove long from make_grid
parent
97871cd9
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
torchvision/utils.py
torchvision/utils.py
+1
-1
No files found.
torchvision/utils.py
View file @
de5dcb93
...
...
@@ -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
])
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment