moshel commited on
Commit
13648f6
·
1 Parent(s): 5ba279b
app.py CHANGED
@@ -8,7 +8,7 @@ import cv2
8
 
9
  print(timm.__version__)
10
 
11
- checkpoint = torch.load('v5-epoch=19-val_loss=0.1464-val_accuracy=0.9514.ckpt', map_location=torch.device('cpu'))
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