From 512b2dd6459f6f3e50f2beab9d39756fbad7738d Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Tue, 11 Aug 2020 14:18:51 -0700 Subject: [PATCH] Add new EfficientNet-B3 and RegNetY-3.2GF weights, both just over 82 top-1 --- timm/models/efficientnet.py | 4 ++-- timm/models/regnet.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 2e64d7e1..6c30f7c8 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -93,10 +93,10 @@ default_cfgs = { url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b2_ra-bcdf34b7.pth', input_size=(3, 288, 288), pool_size=(9, 9), crop_pct=1.0), 'efficientnet_b3': _cfg( - url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b3_ra-a5e2fbc7.pth', + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b3_ra2-cf984f9c.pth', input_size=(3, 300, 300), pool_size=(10, 10), crop_pct=0.904), 'efficientnet_b3a': _cfg( - url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b3_ra-a5e2fbc7.pth', + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b3_ra2-cf984f9c.pth', input_size=(3, 320, 320), pool_size=(10, 10), crop_pct=1.0), 'efficientnet_b4': _cfg( url='', input_size=(3, 380, 380), pool_size=(12, 12), crop_pct=0.922), diff --git a/timm/models/regnet.py b/timm/models/regnet.py index aab94cf0..59554c09 100644 --- a/timm/models/regnet.py +++ b/timm/models/regnet.py @@ -84,7 +84,7 @@ default_cfgs = dict( regnety_006=_cfg(url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-regnet/regnety_006-c67e57ec.pth'), regnety_008=_cfg(url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-regnet/regnety_008-dc900dbe.pth'), regnety_016=_cfg(url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-regnet/regnety_016-54367f74.pth'), - regnety_032=_cfg(url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-regnet/regnety_032-62b47782.pth'), + regnety_032=_cfg(url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/regnety_032_ra-7f2439f9.pth'), regnety_040=_cfg(url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-regnet/regnety_040-f0d569f9.pth'), regnety_064=_cfg(url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-regnet/regnety_064-0a48325c.pth'), regnety_080=_cfg(url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-regnet/regnety_080-e7f3eb93.pth'),