Change PatchGAN architecture

2020-02-09 17:32:46 -05:00 · 2020-02-09 17:32:46 -05:00 · a2e3a9dc64
commit a2e3a9dc64
parent 1dd2b6d748
1 changed files with 2 additions and 3 deletions
--- a/map2map/models/patchgan.py
+++ b/map2map/models/patchgan.py
@ -10,9 +10,8 @@ class PatchGAN(nn.Module):
        self.convs = nn.Sequential(
            ConvBlock(in_chan, 32, seq='CA'),
            ConvBlock(32, 64, seq='CBA'),
-            ConvBlock(64, seq='CBA'),
-            ConvBlock(64, 32, seq='CBA'),
-            nn.Conv3d(32, out_chan, 1),
+            ConvBlock(64, 128, seq='CBA'),
+            ConvBlock(128, out_chan, seq='C'),
        )

    def forward(self, x):