From 528453c8741d14a591018ff00e5fdb516557b79b Mon Sep 17 00:00:00 2001 From: Madhav Khosla Date: Tue, 9 Jul 2019 15:25:54 +0530 Subject: [PATCH] Better Model --- model/unet_parts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model/unet_parts.py b/model/unet_parts.py index 99cb41a..b24e375 100644 --- a/model/unet_parts.py +++ b/model/unet_parts.py @@ -53,7 +53,7 @@ def __init__(self, in_ch, out_ch, bilinear=True): # would be a nice idea if the upsampling could be learned too, # but my machine do not have enough memory to handle all those weights if bilinear: - self.up = nn.functional.interpolate(scale_factor=2, mode='bilinear', align_corners=True) + self.up = nn.Upsample(scale_factor=2, mode='bilinear', align_corners=True) else: self.up = nn.ConvTranspose2d(in_ch//2, in_ch//2, 2, stride=2)