diff --git a/ppgan/models/generators/deoldify.py b/ppgan/models/generators/deoldify.py index ee4b950c73fc3..b806016a6f041 100644 --- 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)