Can't find my own model's layer name

Dear @kazuto1011 Thank you so much for this amazing repo. I have one question regarding the customized model engaging. When I used a YOLOv3 model in this project, I used your script of finding layer name print(*list(model.named_modules()), sep='\n') and I get However, as you can see in the error below it doesn’t recognize it as a valid layer name. Do you have any idea of this?

[Sibozhu]

Thank you so much! I solved the issue eventually. Indeed it was because I lose the hook of backward when I load my own model. After all, everything works perfectly! 　本当にありがとうございました！

[kazuto1011]

Although I don’t know how you adapted to the detection model, please verify that you run GradCAM backward and enable autograd (don’t call torch.set_grad_enabled(False), with torch.no_grad(), etc.). To validate the backward, you can print self.gradient or insert any commands into the hook function.