Thank you for the link.
From what I can tell, in the example you are calling them by name I.e “fc2”, is there a way instead to call by indexing, since I know which layers are Conv2d, and I would like to access every conv2d output
Something like this:
model.features[i].register_forward_hook(get_activation(i))