diff options
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. |