fix pool2d (PaddlePaddle#55)

AnnaTrainingG · Oct 27, 2020 · fb5f54b · fb5f54b
1 parent b4339a5
commit fb5f54b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ppgan/models/generators/deoldify.py b/ppgan/models/generators/deoldify.py
@@ -273,7 +273,7 @@ def __init__(self,
         self.shuf = PixelShuffle(scale)
 
         self.pad = ReplicationPad2d([1, 0, 1, 0])
-        self.blur = nn.Pool2D(2, pool_stride=1, pool_type='avg')
+        self.blur = nn.AvgPool2D(2, stride=1)
         self.relu = relu(True, leaky=leaky)
 
     def forward(self, x):
@@ -339,7 +339,7 @@ def __init__(self,
         self.shuf = PixelShuffle(scale)
 
         self.pad = ReplicationPad2d([1, 0, 1, 0])
-        self.blur = nn.Pool2D(2, pool_stride=1, pool_type='avg')
+        self.blur = nn.AvgPool2D(2, stride=1)
         self.relu = nn.LeakyReLU(
             leaky) if leaky is not None else nn.ReLU()  #relu(True, leaky=leaky)