v7

app.py

@@ -10,7 +10,8 @@ from pytorch_grad_cam.utils.image import show_cam_on_image
 
 print(timm.__version__)
 
-checkpoint = torch.load('v6-epoch=18-val_loss=0.0313-val_accuracy=0.9618.ckpt', map_location=torch.device('cpu'))
+# checkpoint = torch.load('v6-epoch=18-val_loss=0.0313-val_accuracy=0.9618.ckpt', map_location=torch.device('cpu'))
+checkpoint = torch.load('v7-epoch=39-val_loss=0.0222-val_accuracy=0.9806.ckpt', map_location=torch.device('cpu'))
 state_dict = checkpoint["state_dict"]                                                                                                                                                                                                         
 model_weights = state_dict                                                                                                                                                                                                                    
 for key in list(model_weights):                                                                                                                                                                                                               

v7-epoch=39-val_loss=0.0222-val_accuracy=0.9806.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:128abfd788e9c70a0ce26677c4ecfbf9b517139e41d66287c78f65a544efdae3
+size 1051254575