search

albermax / innvestigate

A toolbox to iNNvestigate neural networks' predictions!
Other
1.25k stars 234 forks source link

Looking forward to pytorch suport #243

Closed cuge1995 closed 3 years ago

adrhill commented 3 years ago

Hi Jinlai Zhang,

thanks for your feedback! We're currently focusing on bringing iNNvestigate to TF2. PyTorch support is also on our wish list, but will have to wait until our TF2 port runs smoothly.

For the time being, I recommend you to check out Zennit, which implements LRP in PyTorch.