Spaces:
Sleeping
Sleeping
Jason Adrian
commited on
Commit
•
3858779
1
Parent(s):
e8376fa
update on app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from resnet18 import ResNet18
|
|
7 |
|
8 |
model = ResNet18(1, 5)
|
9 |
|
10 |
-
checkpoint = torch.load('acc=0.94.ckpt')
|
11 |
|
12 |
# The state dict will contains net.layer_name
|
13 |
# Our model doesn't contains `net.` so we have to rename it
|
|
|
7 |
|
8 |
model = ResNet18(1, 5)
|
9 |
|
10 |
+
checkpoint = torch.load('acc=0.94.ckpt', map_location=torch.device('cpu'))
|
11 |
|
12 |
# The state dict will contains net.layer_name
|
13 |
# Our model doesn't contains `net.` so we have to rename it
|