Yeah. sure. Let’s work on this together.
My goal is to have something in PyTorch that is similar to https://github.com/yosinski/deep-visualization-toolbox 892, yet more flexible and powerful. The visualization code for custom networks, in the deep visualization toolbox is painful.
One reference I can think of, on saliency map stuffs, is https://github.com/Lasagne/Recipes/blob/master/examples/Saliency%20Maps%20and%20Guided%20Backpropagation.ipynb 329