remove use_cuda param (#524)

jacobgil · Aug 29, 2024 · 18144f2 · 18144f2
1 parent cda6ea9
commit 18144f2
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/usage_examples/vit_example.py b/usage_examples/vit_example.py
@@ -98,13 +98,11 @@ def reshape_transform(tensor, height=14, width=14):
     if args.method == "ablationcam":
         cam = methods[args.method](model=model,
                                    target_layers=target_layers,
-                                   use_cuda=args.use_cuda,
                                    reshape_transform=reshape_transform,
                                    ablation_layer=AblationLayerVit())
     else:
         cam = methods[args.method](model=model,
                                    target_layers=target_layers,
-                                   use_cuda=args.use_cuda,
                                    reshape_transform=reshape_transform)
 
     rgb_img = cv2.imread(args.image_path, 1)[:, :, ::-1]