replace ConvTranspose2d by up refle conv2d

according to junyanz#190 (comment)
adeshkin · Oct 20, 2023 · bcb1484 · bcb1484
1 parent d484d10
commit bcb1484
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/models/networks.py b/models/networks.py
@@ -357,10 +357,10 @@ def __init__(self, input_nc, output_nc, ngf=64, norm_layer=nn.BatchNorm2d, use_d
 
         for i in range(n_downsampling):  # add upsampling layers
             mult = 2 ** (n_downsampling - i)
-            model += [nn.ConvTranspose2d(ngf * mult, int(ngf * mult / 2),
-                                         kernel_size=3, stride=2,
-                                         padding=1, output_padding=1,
-                                         bias=use_bias),
+            model += [nn.Upsample(scale_factor = 2, mode='bilinear'),
+                      nn.ReflectionPad2d(1),
+                      nn.Conv2d(ngf * mult, int(ngf * mult / 2),
+                                         kernel_size=3, stride=1, padding=0),
                       norm_layer(int(ngf * mult / 2)),
                       nn.ReLU(True)]
         model += [nn.ReflectionPad2d(3)]