torch 1.10 version not working

chr5tphr / zennit

Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.

Other

190 stars 32 forks source link

torch 1.10 version not working #60

Closed rachtibat closed 2 years ago

rachtibat commented 2 years ago

Hi Chris,

it seems that the pytorch version 1.10 produces wrong heatmaps and maybe also newer versions of torch. I updated my conda environment and suddenly the "epsilon_plus_flat" rule did not work anymore. Returning to pip install torch=1.9 torchvision=0.10 fixed the issue.

Best

chr5tphr commented 2 years ago

Hey Reduan,

thank you for raising this issue. I'm currently still quite sick, so I cannot fix this right away, but I will work on it as soon as I'm feeling better again. Until then, I guess torch 1.9.x should be used. Let's hope the fix for 1.10 will not break compatibility for 1.9.

rachtibat commented 2 years ago

Hi Chris,

I wish you a speedy recovery! Take it easy, it is not important.