Fix minor typos in create_attn.py and resnet.py

'eca'->'ceca'
and
doest not-> does not
pull/117/head
Chris Ha 4 years ago
parent c99a5abed4
commit 06a50a94a8

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

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

Loading…
Cancel
Save