Skip to content

Commit fb5f54b

Browse files
authored
fix pool2d (PaddlePaddle#55)
1 parent b4339a5 commit fb5f54b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ppgan/models/generators/deoldify.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -273,7 +273,7 @@ def __init__(self,
273273
self.shuf = PixelShuffle(scale)
274274

275275
self.pad = ReplicationPad2d([1, 0, 1, 0])
276-
self.blur = nn.Pool2D(2, pool_stride=1, pool_type='avg')
276+
self.blur = nn.AvgPool2D(2, stride=1)
277277
self.relu = relu(True, leaky=leaky)
278278

279279
def forward(self, x):
@@ -339,7 +339,7 @@ def __init__(self,
339339
self.shuf = PixelShuffle(scale)
340340

341341
self.pad = ReplicationPad2d([1, 0, 1, 0])
342-
self.blur = nn.Pool2D(2, pool_stride=1, pool_type='avg')
342+
self.blur = nn.AvgPool2D(2, stride=1)
343343
self.relu = nn.LeakyReLU(
344344
leaky) if leaky is not None else nn.ReLU() #relu(True, leaky=leaky)
345345

0 commit comments

Comments
 (0)