（十二）现有网络模型的使用及修改

【声明】来源b站视频小土堆PyTorch深度学习快速入门教程（绝对通俗易懂！）【小土堆】_哔哩哔哩_bilibili

import torch.nn as nn
import torchvision
vgg16_true = torchvision.models.vgg16(pretrained=True)

#在vgg16_true后面添加一层线性层linear,名字为add_linear,实现cifar10分类
# vgg16_true_add = vgg16_true.add_module('add_linear',nn.Linear(1000,10))
#打印修改后的vgg16_true
# print(vgg16_true)

#加到classifier
# vgg16_true_add_c = vgg16_true.classifier.add_module('add_linear_c',nn.Linear(1000,10))
# print(vgg16_true)

#加到classifier里面的(6)
vgg16_true_add_c = vgg16_true.classifier[6].add_module('add_linear_c',nn.Linear(1000,10))
print(vgg16_true)