Update lrp_pipeline_2.py

lrp_pipeline_2.py

@@ -318,7 +318,7 @@ def lrp_main(pixel_conversion):
         *classifier_list
     )  # Replace the model classifier
 
-    PATH = "herlev_best_adam_vgg16_modified12_final.pth"
+    PATH = "herlev_best_adam_vgg16_modified12.pth"
     checkpoint = torch.load(PATH, map_location=torch.device("cpu"))
     vgg16.load_state_dict(checkpoint)
     # vgg16.to(torch.device('cuda'))