Open YOLO-jbc opened 2 years ago
I cannot reproduce this. Are you operating on a custom dataset? What's your Captum version?
Thanks for your reply, I am using Cora dataset by dataset = Planetoid('./datasets', 'Cora', transform=T.NormalizeFeatures())
.
My Captum version is 0.2.0.
Concretely, my code is just the same as code
Can you try to upgrade captum
and test again?
pip install --upgrade captum
It works!!!! Really appreciate your help. I never thought about the Captum's version. Wish you best life ;)
🐛 Describe the bug
Hello!!! I am trying to run your example
captum_explainability.py
. When it goes toI got an error:
IndexError: index 1 is out of bounds for dimension 0 with size 1
The whole error report are as follows:
Thanks a lot! :)
Environment
conda
,pip
, source): piptorch-scatter
):