Migrations
/
pytorch-image-models
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove debug print from RexNet

Browse Source
pull/175/head
Ross Wightman 4 years ago
parent c53ec33ae0
commit ac18adb9c3
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File

1
timm/models/rexnet.py
Unescape View File

@ -61,7 +61,6 @@ class SEWithNorm(nn.Module):
super(SEWithNorm, self).__init__() super(SEWithNorm, self).__init__()
self.avg_pool = nn.AdaptiveAvgPool2d(1) self.avg_pool = nn.AdaptiveAvgPool2d(1)
reduction_channels = reduction_channels or make_divisible(channels // reduction, divisor=divisor) reduction_channels = reduction_channels or make_divisible(channels // reduction, divisor=divisor)
print(reduction_channels)
self.fc1 = nn.Conv2d( self.fc1 = nn.Conv2d(
channels, reduction_channels, kernel_size=1, padding=0, bias=True) channels, reduction_channels, kernel_size=1, padding=0, bias=True)
self.bn = nn.BatchNorm2d(reduction_channels) self.bn = nn.BatchNorm2d(reduction_channels)

Write Preview
Loading…
Cancel
Save