ptrblck:

model.features.denseblock4.denselayer2.conv2


I’m looking for the a layer to create class activation maps, so probably one of the last ones like “module.densenet121.features.denseblock4.denselayer16.conv.2” or “module.densenet121.features.norm5”