diff --git a/timm/models/layers/create_attn.py b/timm/models/layers/create_attn.py index 3bca254f..94c4e4e7 100644 --- a/timm/models/layers/create_attn.py +++ b/timm/models/layers/create_attn.py @@ -17,7 +17,7 @@ def create_attn(attn_type, channels, **kwargs): module_cls = SEModule elif attn_type == 'eca': module_cls = EcaModule - elif attn_type == 'eca': + elif attn_type == 'ceca': module_cls = CecaModule elif attn_type == 'cbam': module_cls = CbamModule diff --git a/timm/models/resnet.py b/timm/models/resnet.py index 0013cbe0..584fd0f6 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -121,7 +121,7 @@ class BasicBlock(nn.Module): super(BasicBlock, self).__init__() assert cardinality == 1, 'BasicBlock only supports cardinality of 1' - assert base_width == 64, 'BasicBlock doest not support changing base width' + assert base_width == 64, 'BasicBlock does not support changing base width' first_planes = planes // reduce_first outplanes = planes * self.expansion first_dilation = first_dilation or dilation