1. import torchvision.datasets
    2. from torch import nn
    4. vgg16_false = torchvision.models.vgg16(pretrained=False)
    5. # vgg16_true = torchvision.models.vgg16(pretrained=True)  # 会下载已经训练好的模型参数
    7. print(vgg16_false)
    9. '''修改网络现有结构'''
    10. # 在最后添加一个线性层
    11. vgg16_false.classifier.add_module("add_linear", nn.Linear(1000, 10))
    12. print(vgg16_false)
    13. # 修改第7层
    14. vgg16_false.classifier[6] = nn.Linear(4096, 10)
    15. print(vgg16_false)