change the order of concanated channels

To be consistent with Cyclegan(Torch), the latter half channels should receive the data from the skip connections.

1 files changed, 1 insertions, 1 deletions

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.