I’m having a PyTorch densenet model for classification purpose.In order to get layer wise output for an image,I can make a submodel upto the relevant layer and donthe inferencenupto that layer.
PyTorch allow me to create a submodel with the required layer,if it is a child.
In case I have a child of block which having some convolution layers.How can I add the particular layer within the child block?or anyways to get output of a layer inside a child-block.
Eg.There are layers inside a denseblock,if I want to do the inference upto a particular layer within the dense block.