diff options
| author | Jun-Yan Zhu <junyanz@users.noreply.github.com> | 2017-09-21 23:57:51 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-09-21 23:57:51 -0700 |
| commit | 75c16069b6fa395c2dfa740602d088764ff09c14 (patch) | |
| tree | 9a22db113664ad338059840f415cc754abff54a5 /models/networks.py | |
| parent | 4062012515f4bcc2ee90ccef80d9492ebb64441f (diff) | |
| parent | 6360df153e46d33b27c46f16c78d732ce2f475cf (diff) | |
Merge pull request #110 from LambdaWill/patch-1
change the order of concanated channels
Diffstat (limited to 'models/networks.py')
| -rw-r--r-- | models/networks.py | 2 |
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. |