pytorch-学习笔记-网络模型的保存和修改

这是这周学习内容的总结，千里之行，始于足下
我们下载了一个别人训练好的模型，有时候需要对这个模型做一些修改，那我们应该怎么操作呢？废话不多说，直接上

import torch
import torchvision
from torch import nn
from torch.nn import Linear

vgg_16=torchvision.models.vgg16(pretrained=False)#加载模型
print(vgg_16)#查看该模型
train_data=torchvision.datasets.CIFAR10('./dataset',train=True,transform=torchvision.transforms.ToTensor(),download=True)#加载数据
vgg_16.add_module("add_linear",nn.Linear(1000,10))#add_module函数，是向模型中添加其他的神经元
print(vgg_16)#查看是否添加成功
vgg_16.classifier[6]=nn.Linear(4096,10)#classifier函数是修改模型里面的参数，
print(vgg_16)

这里以vgg16模型为代表，进行操作，具体的解释可以看代码