Skip to content

Commit dfce069

Browse files
committed
add experimental support for torch hub
1 parent fad2e34 commit dfce069

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

hubconf.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)