Commit dfce069

committed

add experimental support for torch hub

1 parent fad2e34 commit dfce069Copy full SHA for dfce069

File tree

+16

-0

lines changed

+16

-0

lines changed

Lines changed: 16 additions & 0 deletions

Original file line number	Diff line number	Diff 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`	`+)`

Comments

(0)