We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fad2e34 commit dfce069Copy full SHA for dfce069
1 file changed
hubconf.py
@@ -0,0 +1,16 @@
1
+# Optional list of dependencies required by the package
2
+dependencies = ["torch"]
3
+
4
+from torchvision.models import get_model_weights, get_weight
5
6
+from ImageNet.simplenet import (
7
+ simplenet,
8
+ simplenetv1_5m_m1,
9
+ simplenetv1_5m_m2,
10
+ simplenetv1_9m_m1,
11
+ simplenetv1_9m_m2,
12
+ simplenetv1_small_m1_05,
13
+ simplenetv1_small_m2_05,
14
+ simplenetv1_small_m1_075,
15
+ simplenetv1_small_m2_075,
16
+)
0 commit comments