Spaces:

galatolo
/

oral-lesions-detection

Sleeping

Federico Galatolo commited on Jul 13, 2023

Commit

0a91a75

•

1 Parent(s): 97ba1b0

specify cpu in gradcam

Files changed (1) hide show

plots/plot_gradcam.py CHANGED Viewed

@@ -19,7 +19,8 @@ def plot_gradcam(**kwargs):
     cfg_list = [
     "MODEL.ROI_HEADS.SCORE_THRESH_TEST", str(kwargs.th),
     "MODEL.ROI_HEADS.NUM_CLASSES", "3",
-    "MODEL.WEIGHTS", kwargs.model
     ]
     metadata = Metadata()

     cfg_list = [
     "MODEL.ROI_HEADS.SCORE_THRESH_TEST", str(kwargs.th),
     "MODEL.ROI_HEADS.NUM_CLASSES", "3",
+    "MODEL.WEIGHTS", kwargs.model,
+    "MODEL.DEVICE", "cpu"
     ]
     metadata = Metadata()