adjust output size

Andrewzh112 · Dec 14, 2020 · e4826c8 · e4826c8
1 parent 395ab49
commit e4826c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sagan/model.py b/sagan/model.py
@@ -36,7 +36,7 @@ def __init__(self, h_dim, z_dim, img_channels, img_size):
         self.gen = nn.Sequential(
             nn.BatchNorm2d(h_dim*8, momentum=0.9),
             nn.ReLU(),
-            ConvNormAct(h_dim*8, h_dim*8, 'sn', 'up', activation='relu', normalization='bn'),
+            ConvNormAct(h_dim*8, h_dim*8, 'sn', None, activation='relu', normalization='bn'),
             ConvNormAct(h_dim*8, h_dim*8, 'sn', 'up', activation='relu', normalization='bn'),
             ConvNormAct(h_dim*8, h_dim*4, 'sn', 'up', activation='relu', normalization='bn'),
             ConvNormAct(h_dim*4, h_dim*2, 'sn', 'up', activation='relu', normalization='bn'),