We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b4339a5 commit fb5f54bCopy full SHA for fb5f54b
ppgan/models/generators/deoldify.py
@@ -273,7 +273,7 @@ def __init__(self,
273
self.shuf = PixelShuffle(scale)
274
275
self.pad = ReplicationPad2d([1, 0, 1, 0])
276
- self.blur = nn.Pool2D(2, pool_stride=1, pool_type='avg')
+ self.blur = nn.AvgPool2D(2, stride=1)
277
self.relu = relu(True, leaky=leaky)
278
279
def forward(self, x):
@@ -339,7 +339,7 @@ def __init__(self,
339
340
341
342
343
self.relu = nn.LeakyReLU(
344
leaky) if leaky is not None else nn.ReLU() #relu(True, leaky=leaky)
345
0 commit comments