gradcam for a model with a module that outputs a tuple of tensors rather than a tensor

[pigen01]

In my model, I have a module like

class QKVLinear(nn.Module):
    def __init__(self, ...):
        ...
        self.qkv = nn.Linear(dim, qk_dim + qk_dim + dim, bias=bias)
        ...
    def forward(x):
        q, k, v = self.qkv(x).split([self.qk_dim, self.qk_dim, self.dim], dim=-1)
        return q, k, v

thus report AttributeError

  File "/home/xxx/anaconda3/envs/torchcam/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1547, in _call_impl
    hook_result = hook(self, args, result)
                  ^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/xxx/anaconda3/envs/torchcam/lib/python3.11/site-packages/torchcam/methods/_utils.py", line 34, in _record_output_shape
    output_shapes.append((name, output.shape))
                                ^^^^^^^^^^^^
AttributeError: 'tuple' object has no attribute 'shape'

any methods to avoid this?

[May2333]

Hi, i have the same issue. did you solve it?