这是这周学习内容的总结,千里之行,始于足下
我们下载了一个别人训练好的模型,有时候需要对这个模型做一些修改,那我们应该怎么操作呢?废话不多说,直接上
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模型为代表,进行操作,具体的解释可以看代码