PyTorch现有网络模型的使用和修改

victor-gx

已于 2022-03-23 09:35:49 修改

阅读量1.8k

点赞数

分类专栏：笔记 PyTorch 文章标签： pytorch 网络深度学习

于 2022-01-30 00:59:10 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_51296032/article/details/122749670

版权

笔记同时被 2 个专栏收录

43 篇文章

订阅专栏

PyTorch

23 篇文章

订阅专栏

PyTorch现有网络模型的使用和修改

import torchvision

# train_data = torchvision.datasets.ImageNet("dataset", split = 'train', download = True,transform = torchvision.transforms.ToTensor())
from torch import nn

vgg16_false = torchvision.models.vgg16(pretrained = False)
vgg16_true = torchvision.models.vgg16(pretrained = True)
print(vgg16_true)

dataset = torchvision.datasets.CIFAR10(root = "./dataset", train = False,
                                       transform = torchvision.transforms.ToTensor(), download = True)

vgg16_true.classifier.add_module('add_linear', nn.Linear(1000, 10))
print(vgg16_true)

print(vgg16_false)
vgg16_false.classifier[6] = nn.Linear(4096, 10)
print(vgg16_false)