remove use_cuda param

jacobgil · Aug 29, 2024 · 15a9a21 · 15a9a21
1 parent cda6ea9
commit 15a9a21
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]