diff --git a/loss.py b/loss.py index 7aa4f9c..a683750 100644 --- a/loss.py +++ b/loss.py @@ -21,8 +21,8 @@ def scale_pyramid(self, img, num_scales): nh = h // ratio nw = w // ratio scaled_imgs.append(nn.functional.interpolate(img, - size=[nh, nw], mode='bilinear', - align_corners=True)) + size=[nh, nw], mode="area", + align_corners=None)) return scaled_imgs def gradient_x(self, img):