Upload models/backbone.py with huggingface_hub

models/backbone.py

@@ -176,7 +176,7 @@ def build_backbone(args):
                     if keyword in name:
                         parameter.requires_grad_(False)
                         break
-        if "backbone_dir" in args:
+        if hasattr(args, "backbone_dir"):
             pretrained_dir = args.backbone_dir
             PTDICT = {
                 'swin_T_224_1k': 'swin_tiny_patch4_window7_224.pth',