Update efficientnet.py

4 years ago · e16f526124
parent 7ebb99e5d6
commit e16f526124
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']):