如何修改VGG19預訓練模型

 

 

 

Given vgg16, how to remove pool5 layer and all the classify layers? And how to add new layers to that while keeping the pretrained weights?

 

pretrained_model = torchvision.models.vgg16(pretrained=True)

1. given vgg16, how to remove pool5 layer and all the classify layers?(only keep the conov1_1 to relu5_3 layers)
2. how to add new layers ,and with new layers with random initialize parameters and t