|
|
@ -1375,4 +1375,4 @@ def senet154(pretrained=False, **kwargs):
|
|
|
|
model_args = dict(
|
|
|
|
model_args = dict(
|
|
|
|
block=Bottleneck, layers=[3, 8, 36, 3], cardinality=64, base_width=4, stem_type='deep',
|
|
|
|
block=Bottleneck, layers=[3, 8, 36, 3], cardinality=64, base_width=4, stem_type='deep',
|
|
|
|
down_kernel_size=3, block_reduce_first=2, block_args=dict(attn_layer='se'), **kwargs)
|
|
|
|
down_kernel_size=3, block_reduce_first=2, block_args=dict(attn_layer='se'), **kwargs)
|
|
|
|
return _create_resnet('senet154', pretrained, **model_args)
|
|
|
|
return _create_resnet('senet154', pretrained, **model_args)
|
|
|
|