From 2f0e945b46f90ea1e59fe0575265a70e735f6d29 Mon Sep 17 00:00:00 2001 From: Lmxyy Date: Sun, 26 Apr 2020 00:06:14 +0800 Subject: [PATCH] cast BoolTensor to ByteTensor; compatible with Pytorch 1.4 --- models/pix2pix_model.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py index 784cb15a..5da01aa1 100644 --- a/models/pix2pix_model.py +++ b/models/pix2pix_model.py @@ -236,10 +236,10 @@ def divide_pred(self, pred): def get_edges(self, t): edge = self.ByteTensor(t.size()).zero_() - edge[:, :, :, 1:] = edge[:, :, :, 1:] | (t[:, :, :, 1:] != t[:, :, :, :-1]) - edge[:, :, :, :-1] = edge[:, :, :, :-1] | (t[:, :, :, 1:] != t[:, :, :, :-1]) - edge[:, :, 1:, :] = edge[:, :, 1:, :] | (t[:, :, 1:, :] != t[:, :, :-1, :]) - edge[:, :, :-1, :] = edge[:, :, :-1, :] | (t[:, :, 1:, :] != t[:, :, :-1, :]) + edge[:, :, :, 1:] = edge[:, :, :, 1:] | ((t[:, :, :, 1:] != t[:, :, :, :-1]).byte()) + edge[:, :, :, :-1] = edge[:, :, :, :-1] | ((t[:, :, :, 1:] != t[:, :, :, :-1]).byte()) + edge[:, :, 1:, :] = edge[:, :, 1:, :] | ((t[:, :, 1:, :] != t[:, :, :-1, :]).byte()) + edge[:, :, :-1, :] = edge[:, :, :-1, :] | ((t[:, :, 1:, :] != t[:, :, :-1, :]).byte()) return edge.float() def reparameterize(self, mu, logvar):