release code

Dassl.ProGrad.pytorch/dassl/modeling/backbone/build.py

@@ -0,0 +1,11 @@
+from dassl.utils import Registry, check_availability
+
+BACKBONE_REGISTRY = Registry("BACKBONE")
+
+
+def build_backbone(name, verbose=True, **kwargs):
+    avai_backbones = BACKBONE_REGISTRY.registered_names()
+    check_availability(name, avai_backbones)
+    if verbose:
+        print("Backbone: {}".format(name))
+    return BACKBONE_REGISTRY.get(name)(**kwargs)