Create __init__.py

3rdparty/SCHP/networks/__init__.py

@@ -0,0 +1,13 @@
+from __future__ import absolute_import
+
+from SCHP.networks.AugmentCE2P import resnet101
+
+__factory = {
+    "resnet101": resnet101,
+}
+
+
+def init_model(name, *args, **kwargs):
+    if name not in __factory.keys():
+        raise KeyError("Unknown model arch: {}".format(name))
+    return __factory[name](*args, **kwargs)