From 90e988c10fb0aba0bce2bd8ee83349fb51b7002e Mon Sep 17 00:00:00 2001
From: milesial <milesial@users.noreply.github.com>
Date: Thu, 7 Jun 2018 11:20:35 +0300
Subject: [PATCH] Fixed bad use of ConvTranspose2D

Former-commit-id: 245b072a82f2e87a9032a1b87136a9930c008afc
---
 unet/unet_parts.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/unet/unet_parts.py b/unet/unet_parts.py
index 66149d2..c7128d0 100644
--- a/unet/unet_parts.py
+++ b/unet/unet_parts.py
@@ -57,7 +57,7 @@ class up(nn.Module):
         if bilinear:
             self.up = nn.UpsamplingBilinear2d(scale_factor=2)
         else:
-            self.up = nn.ConvTranspose2d(in_ch, out_ch, 2, stride=2)
+            self.up = nn.ConvTranspose2d(in_ch//2, in_ch//2, 2, stride=2)
 
         self.conv = double_conv(in_ch, out_ch)