pytorch学习笔记-----现有模型的使用和修改

注意事项：

1.很多现有的，比较好的网络模性都在torchvision.models模块下

 下面观看完整代码

import torchvision
from torch import nn
#现有模型的使用和修改
vgg16_flase = torchvision.models.vgg16(pretrained=False)#pretrained=false仅仅就是将网络模型在这里进行替换，参数都是随机初始化的
#vgg16_true = torchvision.models.vgg16(pretrained=True,progress=True)#progress=true 会显示下载进度条，pretrained=true时会从网络中下载好预训练权重参数
print(vgg16_flase)

#在vgg16的基础上增加一个线性层，从而对10类别的物体进行分类
vgg16_flase.classifier.add_module("add_linear",nn.Linear(1000,10))#增加了一个线性层
print(vgg16_flase)
#或者直接对最后一个全连接层进行修改
#vgg16_flase.classifier[6] = nn.Linear(4096,10)