mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Remove debug print from RexNet

Browse Source
This commit is contained in:
Ross Wightman 2020-07-29 11:15:19 -07:00
parent c53ec33ae0
commit ac18adb9c3
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
timm/models/rexnet.py
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)