Laurent/REVA-QCAV
1
0
Fork 0
mirror of https://github.com/Laurent2916/REVA-QCAV.git synced 2024-11-08 14:39:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed bad use of ConvTranspose2D

Browse source 
Former-commit-id: 245b072a82f2e87a9032a1b87136a9930c008afc
This commit is contained in:
milesial 2018-06-07 11:20:35 +03:00 committed by GitHub
parent 8008b77af6
commit 90e988c10f
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
unet/unet_parts.py
View file

@ -57,7 +57,7 @@ class up(nn.Module):
if bilinear: if bilinear:
self.up = nn.UpsamplingBilinear2d(scale_factor=2) self.up = nn.UpsamplingBilinear2d(scale_factor=2)
else: 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) self.conv = double_conv(in_ch, out_ch)