ENSEEIHT/projet-long
1
0
Fork 0
Code Issues Pull requests 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:
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)