From 06a50a94a882a38021fdd2bd198fe837e8e3a1cd Mon Sep 17 00:00:00 2001 From: Chris Ha <15088501+VRandme@users.noreply.github.com> Date: Tue, 7 Apr 2020 21:15:57 +0900 Subject: [PATCH] Fix minor typos in create_attn.py and resnet.py 'eca'->'ceca' and doest not-> does not --- timm/models/layers/create_attn.py | 2 +- timm/models/resnet.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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