关键词1 关键词101 关键词201 关键词301 关键词401 关键词501 关键词601 关键词701 关键词801 关键词901 关键词1001 关键词1101 关键词1201 关键词1301 关键词1401 关键词1501 关键词1601 关键词1701 关键词1801 关键词1901 问答1 问答101 问答201 问答301 问答401 问答501 问答601 问答701 问答801 问答901 问答1001 问答1101 问答1201 问答1301 问答1401 问答1501 问答1601 问答1701 问答1801 问答1901 懂视1 懂视101 懂视201 懂视301 懂视401 懂视501 懂视601 懂视701 懂视801 懂视901 懂视1001 懂视1101 懂视1201 懂视1301 懂视1401 懂视1501 懂视1601 懂视1701 懂视1801 懂视1901 文库1 文库101 文库201 文库301 文库401 文库501 文库601 文库701 文库801 文库901 文库1001 文库1101 文库1201 文库1301 文库1401 文库1501 文库1601 文库1701 文库1801 文库1901 分类文章1 分类文章101 分类文章201 分类文章301 分类文章401 分类文章501 分类文章601 分类文章701 分类文章801 分类文章901 分类文章1001 分类文章1101 分类文章1201 分类文章1301 分类文章1401 分类文章1501 分类文章1601 分类文章1701 分类文章1801 分类文章1901

import torchvision
from torch import nn

# 加载vgg16网络模型，pretrained 是否使用优质网络的参数，并不是权重参数
vgg16_f = torchvision.models.vgg16(pretrained=False)
# 加载vgg16网络模型，pretrained 是否使用优质参数
vgg16_t = torchvision.models.vgg16(pretrained=True)

print(vgg16_t)

train_data = torchvision.datasets.CIFAR10("./data", train=True, transform=torchvision.transforms.ToTensor)

# 增加一层 网络 当前为全连接层
vgg16_f.classifier.add_module('add_linear', nn.Linear(1000, 10))
print(vgg16_f)

# 改层
vgg16_t.classifier[6] = nn.Linear(4096, 10)
print(vgg16_t)