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

Update comments on the new SE-ResNeXt26 models

Browse Source
This commit is contained in:
Ross Wightman 2019-12-28 17:33:10 -08:00
parent 8894a6bc08
commit 19fc205a4d
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
timm/models/resnet.py
View File

@ -873,8 +873,8 @@ def swsl_resnext101_32x16d(pretrained=True, **kwargs):
@register_model
def seresnext26d_32x4d(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
"""Constructs a ResNet-26 v1d model.
This is technically a 28 layer ResNet, sticking with 'd' modifier from Gluon for now.
"""Constructs a SE-ResNeXt-26-D model.
This is technically a 28 layer ResNet, sticking with 'D' modifier from Gluon / bag-of-tricks.
"""
default_cfg = default_cfgs['seresnext26d_32x4d']
model = ResNet(
@ -889,7 +889,9 @@ def seresnext26d_32x4d(pretrained=False, num_classes=1000, in_chans=3, **kwargs)
@register_model
def seresnext26t_32x4d(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
"""Constructs a ResNet-26 v1d model.
"""Constructs a SE-ResNet-26-T model.
This is technically a 28 layer ResNet, like a 'D' bag-of-tricks model but with tiered 24, 48, 64 channels
in the deep stem. Stem channel counts suggested by Jeremy Howard.
"""
default_cfg = default_cfgs['seresnext26t_32x4d']
model = ResNet(