Hi everyone,
I’m running the Densenet121 Model. After loading weights. I have 2 unwanted blocks model.features.denseblock4 and model.features.transition3. I’m trying to remove these 2 blocks of layers. Could any please tell me the way to do it? I’ll very appreciate it.

prune830×505 26.1 KB