Skip to content

Commit 19fc205

Browse files
committed
Update comments on the new SE-ResNeXt26 models
1 parent 8894a6b commit 19fc205

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

timm/models/resnet.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -873,8 +873,8 @@ def swsl_resnext101_32x16d(pretrained=True, **kwargs):
873873

874874
@register_model
875875
def seresnext26d_32x4d(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
876-
"""Constructs a ResNet-26 v1d model.
877-
This is technically a 28 layer ResNet, sticking with 'd' modifier from Gluon for now.
876+
"""Constructs a SE-ResNeXt-26-D model.
877+
This is technically a 28 layer ResNet, sticking with 'D' modifier from Gluon / bag-of-tricks.
878878
"""
879879
default_cfg = default_cfgs['seresnext26d_32x4d']
880880
model = ResNet(
@@ -889,7 +889,9 @@ def seresnext26d_32x4d(pretrained=False, num_classes=1000, in_chans=3, **kwargs)
889889

890890
@register_model
891891
def seresnext26t_32x4d(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
892-
"""Constructs a ResNet-26 v1d model.
892+
"""Constructs a SE-ResNet-26-T model.
893+
This is technically a 28 layer ResNet, like a 'D' bag-of-tricks model but with tiered 24, 48, 64 channels
894+
in the deep stem. Stem channel counts suggested by Jeremy Howard.
893895
"""
894896
default_cfg = default_cfgs['seresnext26t_32x4d']
895897
model = ResNet(

0 commit comments

Comments
 (0)