Solved by vivekm in post #4 
              
              
                Hi @TinfoilHat0, you can take a look at this getting started tutorial on the Titanic dataset, it covers example usage of Captum for layer and neuron attribution.