dk - edit

net/__init__.py

@@ -0,0 +1,9 @@
+from .models import Backbone_nFC, Backbone_nFC_Id
+
+
+def get_model(model_name, num_label, use_id=False, num_id=None):
+    if not use_id:
+        return Backbone_nFC(num_label, model_name)
+    else:
+        return Backbone_nFC_Id(num_label, num_id, model_name)
+