A question about the LRP algorithm #12
Description
Thank you for sharing this excellent work. I had a problem when running the code on the CIFAR10 dataset.
The command is : python run.py -M cifar10-vgg16_model -DS cifar10 -L 1 -A idc -KS 10000 -KN 3
I trained a model namely 'cifar10-vgg16_model' on the CIFAR10 dataset. When I run the code to find the relevance score of each neuron, I got the following error message.
Could you please help me to solve this issue?
Thank you very much.