Skip to content

Commit e01ccb8

Browse files
authored
Merge pull request #117 from VRandme/typo_eca
minor PR to fix typos found in #116 and resnet.py
2 parents c99a5ab + 06a50a9 commit e01ccb8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

timm/models/layers/create_attn.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def create_attn(attn_type, channels, **kwargs):
1717
module_cls = SEModule
1818
elif attn_type == 'eca':
1919
module_cls = EcaModule
20-
elif attn_type == 'eca':
20+
elif attn_type == 'ceca':
2121
module_cls = CecaModule
2222
elif attn_type == 'cbam':
2323
module_cls = CbamModule

timm/models/resnet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ def __init__(self, inplanes, planes, stride=1, downsample=None, cardinality=1, b
121121
super(BasicBlock, self).__init__()
122122

123123
assert cardinality == 1, 'BasicBlock only supports cardinality of 1'
124-
assert base_width == 64, 'BasicBlock doest not support changing base width'
124+
assert base_width == 64, 'BasicBlock does not support changing base width'
125125
first_planes = planes // reduce_first
126126
outplanes = planes * self.expansion
127127
first_dilation = first_dilation or dilation

0 commit comments

Comments
 (0)