From 33f8a1bf36148fe45bf0e6ed020e5bc7f3247124 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Thu, 3 Sep 2020 10:45:17 -0700 Subject: [PATCH] Updated README, add wide_resnet50_2 and seresnext50_32x4d weights --- README.md | 7 +++++++ timm/models/resnet.py | 5 ++++- 2 files changed, 11 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index e6d4269d..16f0b52e 100644 --- a/README.md +++ b/README.md @@ -2,6 +2,13 @@ ## What's New +### Sept 3, 2020 +* New weights + * Wide-ResNet50 - 81.5 top-1 (vs 78.5 torchvision) + * SEResNeXt50-32x4d - 81.3 top-1 (vs 79.1 cadene) +* Support for native Torch AMP and channels_last memory format added to train/validate scripts +* Models tested with channels_last on latest NGC 20.08 container. AdaptiveAvgPool in attn layers changed to mean((2,3)) to work around bug with NHWC kernel. + ### Aug 12, 2020 * New/updated weights from training experiments * EfficientNet-B3 - 82.1 top-1 (vs 81.6 for official with AA and 81.9 for AdvProp) diff --git a/timm/models/resnet.py b/timm/models/resnet.py index 1e7a6b08..fc86e960 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -55,7 +55,9 @@ default_cfgs = { 'resnet152': _cfg(url='https://download.pytorch.org/models/resnet152-b121ed2d.pth'), 'tv_resnet34': _cfg(url='https://download.pytorch.org/models/resnet34-333f7ec4.pth'), 'tv_resnet50': _cfg(url='https://download.pytorch.org/models/resnet50-19c8e357.pth'), - 'wide_resnet50_2': _cfg(url='https://download.pytorch.org/models/wide_resnet50_2-95faca4d.pth'), + 'wide_resnet50_2': _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/wide_resnet50_racm-8234f177.pth', + interpolation='bicubic'), 'wide_resnet101_2': _cfg(url='https://download.pytorch.org/models/wide_resnet101_2-32ee1156.pth'), # ResNeXt @@ -147,6 +149,7 @@ default_cfgs = { interpolation='bicubic', first_conv='conv1.0'), 'seresnext50_32x4d': _cfg( + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/seresnext50_32x4d_racm-a304a460.pth', interpolation='bicubic'), 'seresnext101_32x4d': _cfg( url='',