Skip to content

Commit 9947bea

Browse files
authored
Fix use_cuda of swin transformer example code (#505)
1 parent 1ff3f58 commit 9947bea

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

usage_examples/swinT_example.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,13 +97,11 @@ def reshape_transform(tensor, height=7, width=7):
9797
if args.method == "ablationcam":
9898
cam = methods[args.method](model=model,
9999
target_layers=target_layers,
100-
use_cuda=args.use_cuda,
101100
reshape_transform=reshape_transform,
102101
ablation_layer=AblationLayerVit())
103102
else:
104103
cam = methods[args.method](model=model,
105104
target_layers=target_layers,
106-
use_cuda=args.use_cuda,
107105
reshape_transform=reshape_transform)
108106

109107
rgb_img = cv2.imread(args.image_path, 1)[:, :, ::-1]

0 commit comments

Comments
 (0)