How to create a submodel from a pretrained model in pytorch without having to rewrite the whole architecture?

Question

So, I have been working on neural style transfer in Pytorch, but I\'m stuck at the point where we have to run the input image through limited number of layers and minimize the s