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

Merge pull request #325 from likyoo/master

Browse source 
delete convolution bias before BN

Former-commit-id: 0921374ee4ed02688ef5ed48c50b241ee8911069
This commit is contained in:
milesial 2021-12-13 12:02:03 +01:00 committed by GitHub
parent 473a03ce17 a953415a0d
commit c1f4757c32
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
unet/unet_parts.py
View file

@ -13,10 +13,10 @@ class DoubleConv(nn.Module):
if not mid_channels: if not mid_channels:
mid_channels = out_channels mid_channels = out_channels
self.double_conv = nn.Sequential( self.double_conv = nn.Sequential(
nn.Conv2d(in_channels, mid_channels, kernel_size=3, padding=1), nn.Conv2d(in_channels, mid_channels, kernel_size=3, padding=1, bias=False),
nn.BatchNorm2d(mid_channels), nn.BatchNorm2d(mid_channels),
nn.ReLU(inplace=True), nn.ReLU(inplace=True),
nn.Conv2d(mid_channels, out_channels, kernel_size=3, padding=1), nn.Conv2d(mid_channels, out_channels, kernel_size=3, padding=1, bias=False),
nn.BatchNorm2d(out_channels), nn.BatchNorm2d(out_channels),
nn.ReLU(inplace=True) nn.ReLU(inplace=True)
) )