v6
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ import cv2
|
|
| 8 |
|
| 9 |
print(timm.__version__)
|
| 10 |
|
| 11 |
-
checkpoint = torch.load('
|
| 12 |
state_dict = checkpoint["state_dict"]
|
| 13 |
model_weights = state_dict
|
| 14 |
for key in list(model_weights):
|
|
|
|
| 8 |
|
| 9 |
print(timm.__version__)
|
| 10 |
|
| 11 |
+
checkpoint = torch.load('v6-epoch=18-val_loss=0.0313-val_accuracy=0.9618.ckpt', map_location=torch.device('cpu'))
|
| 12 |
state_dict = checkpoint["state_dict"]
|
| 13 |
model_weights = state_dict
|
| 14 |
for key in list(model_weights):
|
v6-epoch=18-val_loss=0.0313-val_accuracy=0.9618.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7e48c5b4e7ab26724974fd06005fc24209cca21af7851f11e7a612d3234700fb
|
| 3 |
+
size 1051254575
|