Python全系列 教程
3567个小节阅读:5930k
目录
鸿蒙应用开发
C语言快速入门
JAVA全系列 教程
面向对象的程序设计语言
Python全系列 教程
Python3.x版本,未来主流的版本
人工智能 教程
顺势而为,AI创新未来
大厂算法 教程
算法,程序员自我提升必经之路
C++ 教程
一门通用计算机编程语言
微服务 教程
目前业界流行的框架组合
web前端全系列 教程
通向WEB技术世界的钥匙
大数据全系列 教程
站在云端操控万千数据
AIGC全能工具班
A A
White Night
xxxxxxxxxx
from torchvision import models
alexnet = models.alexnet(pretrained=True)
print(alexnet)
import torch.nn as nn
for param in alexnet.parameters():
param.requires_grad = False # 限制参数更新
alexnet.classifier = nn.Sequential(
nn.Dropout(p=0.5, inplace=False),
nn.Linear(in_features=9216, out_features=4096, bias=True),
nn.ReLU(inplace=True),
nn.Dropout(p=0.5, inplace=False),
nn.Linear(in_features=4096, out_features=4096, bias=True),
nn.ReLU(inplace=True),
# 此处将out_features参数改为了2
nn.Linear(in_features=4096, out_features=2, bias=True)
)
CUDA = torch.cuda.is_available()
if CUDA:
alexnet = alexnet.cuda()
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.SGD(alexnet.classifier.parameters(),lr=0.001)