Skip to content

Commit 7bda0e8

Browse files
authored
Backward compat fix for Pytorch < 0.4 (#472)
1 parent 1a6038e commit 7bda0e8

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
@@ -58,7 +58,7 @@ def norm_range(t, range):
5858
if range is not None:
5959
norm_ip(t, range[0], range[1])
6060
else:
61-
norm_ip(t, t.min().item(), t.max().item())
61+
norm_ip(t, float(t.min()), float(t.max()))
6262

6363
if scale_each is True:
6464
for t in tensor: # loop over mini-batch dimension

0 commit comments

Comments
 (0)