This works well if there are limited number of layers or ones with fixed names. How do you go about when the model contains many layers so that you can’t explicitly write these statements?
Essentially, my aim is to obtain the activations from all the layers inside a model.
Thanks! 