Update face_fed.py

2020-10-08 16:49:58 -04:00
parent 3b0023ab50
commit be89fe24cd
1 changed files with 1 additions and 1 deletions
@@ -592,7 +592,7 @@ class scale_kernel_conf(nn.Module):
        x4=self.conv3(x3)
        x4 = self.trans_block4(x4)
        #print(x4.size())
-        residual = self.sig(self.conv_refin(self.sig(F.avg_pool2d(x4,16))))
+        residual = self.sig(self.conv_refin(self.sig(F.adaptive_avg_pool2d(x4,(1,1)))))
        #print(residual)
        residual = F.interpolate(residual, size=x.size()[2:])
        #print(residual.size())