Issue while running on Test data for resnet50

[shejain-hub]

The model inject runs fine when run before train phase. But I would want to run it on test data as well. The model gives me an error Tensor 0 has no grad function. Is there some parameter that needs to be changed when same model is being used for test and train data.
Additionally what is the naming sequence. I am trying to show data patch and attention map along side, how do I map these two based on the name of the png file.