From 15a9a21347402713f8bdf3052e1082179f9f3073 Mon Sep 17 00:00:00 2001 From: jackyjinjing Date: Thu, 29 Aug 2024 02:46:39 +0000 Subject: [PATCH] remove use_cuda param --- usage_examples/vit_example.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/usage_examples/vit_example.py b/usage_examples/vit_example.py index 5cf51d69b..eb89f94fd 100644 --- 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]