Update face_fed.py

2020-10-02 16:45:26 -04:00
parent 391254f6db
commit 3b0023ab50
1 changed files with 1 additions and 1 deletions
@@ -594,7 +594,7 @@ class scale_kernel_conf(nn.Module):
        #print(x4.size())
        residual = self.sig(self.conv_refin(self.sig(F.avg_pool2d(x4,16))))
        #print(residual)
-        residual = F.upsample_nearest(residual, scale_factor=128)
+        residual = F.interpolate(residual, size=x.size()[2:])
        #print(residual.size())
        return residual