I’ve trained a DenseNet121 model, and am trying to get ahold of one of the final layers for a GradCAM implementation. The implementation uses model._modules[] as a dict from layer name to layer, but for some reason, my model has only one key-value pair in this dict: “module” to the entire DenseNet object. What should I do instead? My version is 1.2.0.
Thanks!