From 6360df153e46d33b27c46f16c78d732ce2f475cf Mon Sep 17 00:00:00 2001
From: LambdaWill <574819595@qq.com>
Date: Fri, 22 Sep 2017 14:43:25 +0800
Subject: change the order of concanated channels

To be consistent with Cyclegan(Torch), the latter half channels should receive the data from the skip connections.
---
 models/networks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'models/networks.py')

diff --git a/models/networks.py b/models/networks.py
index f48e7fe..0410ca3 100644
--- a/models/networks.py
+++ b/models/networks.py
@@ -311,7 +311,7 @@ class UnetSkipConnectionBlock(nn.Module):
         if self.outermost:
             return self.model(x)
         else:
-            return torch.cat([self.model(x), x], 1)
+            return torch.cat([x, self.model(x)], 1)
 
 
 # Defines the PatchGAN discriminator with the specified arguments.
-- 
cgit v1.2.3-70-g09d2