Merge pull request #117 from VRandme/typo_eca

minor PR to fix typos found in #116 and resnet.py
pull/123/head
Ross Wightman 5 years ago committed by GitHub
commit e01ccb88ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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