Following this question, we should include the following in our GradCAM docstring:

for name, _ in model.named_modules(): print(name)