网络模型使用及修改

#1.下载网络模型
#关于VGG-16模型，当您使用pretrained=True时，您会得到一个已经使用ImageNet数据集训练好的模型，
# 这意味着模型的权重已经根据ImageNet数据集进行了优化。
# 而当您使用pretrained=False时，您得到的模型权重将是随机初始化的，没有经过任何数据集的训练。
import torchvision
from torch import nn

vgg16_true=torchvision.models.vgg16(pretrained=True) # 下载卷积层对应的参数是多少、池化层对应的参数是多少，这些参数时ImageNet训练好了的
vgg16_false=torchvision.models.vgg16(pretrained=False)# 没有预训练的参数
print("下好了")
print(vgg16_true)

#2.查看函数用法
help(torchvision.models.vgg16)

#3.网络模型添加
vgg16_true.add_module("add_linear",nn.Linear(1000,10))
print(vgg16_true)

#4.网络模型修改
print(vgg16_false)
vgg16_false.classifier[6]=nn.Linear(4096,10)
print(vgg16_false)