Skip to content

Commit b6be00b

Browse files
comments
1 parent 4a8d358 commit b6be00b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pytorch_grad_cam/shapley_cam.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def get_cam_weights(self,
6464
retain_graph=False,
6565
allow_unused=True
6666
)[0]
67-
# print(torch.max(hvp[0]).item()) # Use .item() to get the scalar value
67+
# print(torch.max(hvp[0]).item()) # check if hvp is not all zeros
6868
if hvp is None:
6969
hvp = torch.tensor(0).to(self.device)
7070
else:

0 commit comments

Comments
 (0)