From 80eed668da574eb89563e99adf8e0a9e2101a43a Mon Sep 17 00:00:00 2001 From: xiaxiaohua1 Date: Thu, 17 Nov 2022 14:26:36 +0800 Subject: [PATCH] fix TCAV for tiny-bert model report AssertionError: inputs must have type torch.Tensor but found: error happened --- captum/_utils/common.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/captum/_utils/common.py b/captum/_utils/common.py index bba0ea293b..e5373e9c83 100644 --- a/captum/_utils/common.py +++ b/captum/_utils/common.py @@ -174,6 +174,8 @@ def _format_tensor_into_tuples( ) -> Union[None, Tuple[Tensor, ...]]: if inputs is None: return None + if isinstance(inputs, list): + inputs = tuple(inputs) if not isinstance(inputs, tuple): assert isinstance( inputs, torch.Tensor