From 7ebb99e5d651858b20809b345cd9e5ea9ba9f47b Mon Sep 17 00:00:00 2001 From: jim4399266 <32267011+jim4399266@users.noreply.github.com> Date: Fri, 12 Nov 2021 11:05:33 +0800 Subject: [PATCH] Update efficientnet.py --- timm/models/efficientnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 0c6dddf1..e3d39816 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -562,7 +562,7 @@ def _create_effnet(variant, pretrained=False, **kwargs): model_cls = EfficientNetFeatures -# default_cfg = default_cfgs[variant] + default_cfg = default_cfgs[variant] if kwargs.get('pth_path', None): # if use local weights and weights are existed if os.path.exists(kwargs['pth_path']):