From 25c693605fb14bbe9383d7e6309f28f486f64781 Mon Sep 17 00:00:00 2001 From: Francisco Massa Date: Mon, 16 Apr 2018 13:44:54 +0200 Subject: [PATCH] Backward compat fix for Pytorch < 0.4 --- torchvision/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/utils.py b/torchvision/utils.py index 7c4f02257e5..e277d0c7253 100644 --- a/torchvision/utils.py +++ b/torchvision/utils.py @@ -58,7 +58,7 @@ def norm_range(t, range): 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