From a2e3a9dc6493961d3d13f67a50452d53d8383441 Mon Sep 17 00:00:00 2001
From: Yin Li <eelregit@gmail.com>
Date: Sun, 9 Feb 2020 17:32:46 -0500
Subject: [PATCH] Change PatchGAN architecture

---
 map2map/models/patchgan.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/map2map/models/patchgan.py b/map2map/models/patchgan.py
index 803f894..5f8a466 100644
--- 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):