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 minor typos in create_attn.py and resnet.py

Browse Source 
'eca'->'ceca'
and
doest not-> does not
This commit is contained in:
Chris Ha 2020-04-07 21:15:57 +09:00
parent c99a5abed4
commit 06a50a94a8
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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

2
timm/models/resnet.py
View File

@ -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