Update efficientnet.py

4 years ago · 7ebb99e5d6
parent caf28b6bc5
commit 7ebb99e5d6
1 changed files with 1 additions and 1 deletions
--- 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']):