use the explicit v1 form
Browse files
app.py
CHANGED
@@ -8,15 +8,15 @@ os.system("wget https://raw.githubusercontent.com/pytorch/hub/master/imagenet_cl
|
|
8 |
|
9 |
|
10 |
import torch
|
11 |
-
model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_5m_m2", pretrained=True)
|
12 |
# or any of these variants
|
13 |
-
# model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_5m_m2", pretrained=True)
|
14 |
-
# model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_9m_m1", pretrained=True)
|
15 |
-
# model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_9m_m2", pretrained=True)
|
16 |
-
# model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_small_m1_05", pretrained=True)
|
17 |
-
# model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_small_m2_05", pretrained=True)
|
18 |
-
# model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_small_m1_075", pretrained=True)
|
19 |
-
# model = torch.hub.load("coderx7/simplenet_pytorch","simplenetv1_small_m2_075", pretrained=True)
|
20 |
model.eval()
|
21 |
|
22 |
# Download an example image from the pytorch website
|
|
|
8 |
|
9 |
|
10 |
import torch
|
11 |
+
model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m2", pretrained=True)
|
12 |
# or any of these variants
|
13 |
+
# model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_5m_m2", pretrained=True)
|
14 |
+
# model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_9m_m1", pretrained=True)
|
15 |
+
# model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_9m_m2", pretrained=True)
|
16 |
+
# model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_small_m1_05", pretrained=True)
|
17 |
+
# model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_small_m2_05", pretrained=True)
|
18 |
+
# model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_small_m1_075", pretrained=True)
|
19 |
+
# model = torch.hub.load("coderx7/simplenet_pytorch:v1.0.0","simplenetv1_small_m2_075", pretrained=True)
|
20 |
model.eval()
|
21 |
|
22 |
# Download an example image from the pytorch website
|