现有模型的修改

方法一：
按照目标增加一层线性层对已有模型修改
方法二：
直接修改参数

import torch
import torchvision
from torch import nn

vgg16_false = torchvision.models.vgg16(pretrained=False)
vgg16_true = torchvision.models.vgg16(pretrained=True)

print(vgg16_true)

#train_data = torchvision.datasets.CIFAR10("../data",train=True,transform=torchvision.transforms.ToTensor(),download=True)
#增加一层线性层对已有模型修改
vgg16_true.classifier.add_module('add_linear',nn.Linear(1000,10))
print(vgg16_true)

print(vgg16_false)
#直接对已有模型修改
vgg16_false.classifier[6] = nn.Linear(4096,10)
print(vgg16_true)