From c3287aafb3b226f291938df2ed6081a21931c329 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 29 Jun 2019 16:17:06 -0700 Subject: [PATCH] Slight improvement in EfficientNet-B2 native PyTorch weights --- README.md | 2 +- timm/models/gen_efficientnet.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index ac2fbdf6..194a6476 100644 --- a/README.md +++ b/README.md @@ -87,7 +87,7 @@ I've leveraged the training scripts in this repository to train a few of the mod #### @ 260x260 |Model | Prec@1 (Err) | Prec@5 (Err) | Param # | Image Scaling | |---|---|---|---|---| -| efficientnet_b2 | 79.668 (20.332) | 94.634 (5.366) | 9.11M | bicubic | +| efficientnet_b2 | 79.760 (20.240) | 94.714 (5.286) | 9.11M | bicubic | ### Ported Weights diff --git a/timm/models/gen_efficientnet.py b/timm/models/gen_efficientnet.py index 0642a1cb..9a0a6dd4 100644 --- a/timm/models/gen_efficientnet.py +++ b/timm/models/gen_efficientnet.py @@ -84,7 +84,7 @@ default_cfgs = { url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b1-533bc792.pth', input_size=(3, 240, 240), pool_size=(8, 8), interpolation='bicubic', crop_pct=0.882), 'efficientnet_b2': _cfg( - url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b2-d4105846.pth', + url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/efficientnet_b2-cf78dc4d.pth', input_size=(3, 260, 260), pool_size=(9, 9), interpolation='bicubic', crop_pct=0.890), 'efficientnet_b3': _cfg( url='', input_size=(3, 300, 300), pool_size=(10, 10), crop_pct=0.904),