diff --git a/models/pix2pix_model.py b/models/pix2pix_model.py
index 784cb15a..5ec24b1b 100644
--- a/models/pix2pix_model.py
+++ b/models/pix2pix_model.py
@@ -235,7 +235,7 @@ def divide_pred(self, pred):
         return fake, real
 
     def get_edges(self, t):
-        edge = self.ByteTensor(t.size()).zero_()
+        edge = self.ByteTensor(t.size()).zero_().bool()
         edge[:, :, :, 1:] = edge[:, :, :, 1:] | (t[:, :, :, 1:] != t[:, :, :, :-1])
         edge[:, :, :, :-1] = edge[:, :, :, :-1] | (t[:, :, :, 1:] != t[:, :, :, :-1])
         edge[:, :, 1:, :] = edge[:, :, 1:, :] | (t[:, :, 1:, :] != t[:, :, :-1, :])