upsample model in pytorch 0.4.0

xiao543348405 · Aug 6, 2018 · 1d440be · 1d440be
1 parent 47d7e97
commit 1d440be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/RFB_Net_E_vgg.py b/models/RFB_Net_E_vgg.py
@@ -243,7 +243,7 @@ def forward(self, x):
         for k in range(23, len(self.base)):
             x = self.base[k](x)
         s2 = self.up_reduce(x)
-        s2 = F.upsample(s2, scale_factor=2, mode='bilinear')
+        s2 = F.upsample(s2, scale_factor=2, mode='bilinear', align_corners=True)
         s = torch.cat((s1,s2),1)
 
         ss = self.Norm(s)