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

Fix two regressions

Browse Source
This commit is contained in:
Ross Wightman 2020-06-05 11:04:51 -07:00
parent a7e8cadd15
commit 4ddde1d3a4
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
timm/models/layers/se.py
View File

@ -5,7 +5,7 @@ from .create_act import get_act_fn
class SEModule(nn.Module): class SEModule(nn.Module):
def __init__(self, channels, reduction=16, act_layer=nn.ReLU, min_channels=8, reduction_channels=None, def __init__(self, channels, reduction=16, act_layer=nn.ReLU, min_channels=8, reduction_channels=None,
gate_fn='hard_sigmoid'): gate_fn='sigmoid'):
super(SEModule, self).__init__() super(SEModule, self).__init__()
self.avg_pool = nn.AdaptiveAvgPool2d(1) self.avg_pool = nn.AdaptiveAvgPool2d(1)
reduction_channels = reduction_channels or max(channels // reduction, min_channels) reduction_channels = reduction_channels or max(channels // reduction, min_channels)

4
timm/models/res2net.py
View File

@ -99,9 +99,9 @@ class Bottle2neck(nn.Module):
spx = torch.split(out, self.width, 1) spx = torch.split(out, self.width, 1)
spo = [] spo = []
sp = spx[0] sp = spx[0] # redundant, for torchscript
for i, (conv, bn) in enumerate(zip(self.convs, self.bns)): for i, (conv, bn) in enumerate(zip(self.convs, self.bns)):
if self.is_first: if i == 0 or self.is_first:
sp = spx[i] sp = spx[i]
else: else:
sp = sp + spx[i] sp = sp + spx[i]