diff --git a/stargan/model.py b/stargan/model.py index a8e91e8..cff1a50 100644 --- a/stargan/model.py +++ b/stargan/model.py @@ -105,7 +105,7 @@ class AvgBlurGenerator(nn.Module): super(AvgBlurGenerator, self).__init__() layers = [] - layers.append(nn.Conv2d(3+c_dim, conv_dim, kernel_size=7, stride=1, padding=3, bias=False)) + layers.append(nn.Conv2d(3+c_dim, conv_dim, kernel_size=9, stride=1, padding=3, bias=False)) layers.append(nn.InstanceNorm2d(conv_dim, affine=True, track_running_stats=True)) layers.append(nn.ReLU(inplace=True))