How to strip a pretrained network and add some layers to it using pytorch lightning?

Question

I am trying to use transfer learning for an image segmentation task, and my plan is to use the first few layers of a pretrained model (VGG16 for example) as an encoder and then