This commit is contained in:
Nataniel Ruiz
2020-01-17 15:47:27 -05:00
parent 5795810e6e
commit 698e494d2e
+1 -1
View File
@@ -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))