Skip to content

Commit 33436fa

Browse files
committed
Add weights for ResNeXt50d model
1 parent e78cd79 commit 33436fa

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

timm/models/resnet.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,9 @@ def _cfg(url='', **kwargs):
5353
'resnext50_32x4d': _cfg(
5454
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/resnext50_32x4d-068914d1.pth',
5555
interpolation='bicubic'),
56-
'resnext50d_32x4d': _cfg(url=''),
56+
'resnext50d_32x4d': _cfg(
57+
url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/resnext50d_32x4d-103e99f8.pth',
58+
interpolation='bicubic'),
5759
'resnext101_32x4d': _cfg(url=''),
5860
'resnext101_32x8d': _cfg(url='https://download.pytorch.org/models/resnext101_32x8d-8ba56ff5.pth'),
5961
'resnext101_64x4d': _cfg(url=''),
@@ -534,7 +536,7 @@ def resnext50_32x4d(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
534536

535537
@register_model
536538
def resnext50d_32x4d(pretrained=False, num_classes=1000, in_chans=3, **kwargs):
537-
"""Constructs a ResNeXt50-32x4d model.
539+
"""Constructs a ResNeXt50d-32x4d model. ResNext50 w/ deep stem & avg pool downsample
538540
"""
539541
default_cfg = default_cfgs['resnext50d_32x4d']
540542
model = ResNet(

0 commit comments

Comments
 (0)