Backward compat fix for Pytorch < 0.4 (#472)

7bda0e85 · Francisco Massa · GitHub · 1a6038ea · 7bda0e85
Unverified Commit 7bda0e85 authored Apr 16, 2018 by Francisco Massa Committed by GitHub Apr 16, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/utils.py torchvision/utils.py +1 -1

No files found.
--- a/torchvision/utils.py
+++ b/torchvision/utils.py
@@ -58,7 +58,7 @@ def make_grid(tensor, nrow=8, padding=2,
            if range is not None:
                norm_ip(t, range[0], range[1])
            else:
-                norm_ip(t, t.min().item(), t.max().item())
+                norm_ip(t, float(t.min()), float(t.max()))

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