深度学习-03 Pytorch

损失函数是用来衡量模型预测结果与真实值之间的差异,并用来优化模型的指标。在机器学习和神经网络中,常用的损失函数包括均方误差(Mean Squared Error,MSE)、交叉熵(Cross-Entropy)等。

反向传播(Backpropagation)是一种基于梯度下降算法的优化方法,用来计算模型中每个参数对于损失函数的梯度,从而更新参数以最小化损失函数。反向传播通过链式法则将损失函数的梯度传递回每个参数,从输出层向输入层反向计算梯度。具体而言,反向传播算法可以分为两个步骤:

前向传播(Forward Propagation):将输入数据通过模型的参数计算出预测值,并计算出损失函数的值。

反向传播(Backward Propagation):通过链式法则计算出损失函数对于每个参数的梯度,并更新参数。

反向传播的过程中,需要根据损失函数的类型来计算梯度。例如,对于均方误差损失函数,梯度的计算可以通过对每个参数的偏导数进行求解;对于交叉熵损失函数,梯度的计算可以通过softmax函数的导数进行求解。

反向传播算法的实现主要包括两个步骤:计算梯度和参数更新。在计算梯度时,通过对损失函数进行求导,得到每个参数的梯度;在参数更新时,根据梯度和学习率进行参数的更新。这个过程不断迭代,直到达到收敛条件或达到一定的迭代次数为止。

优化器是机器学习中一个重要的组件,用于调整模型的参数以使其最优化。在机器学习任务中,目标就是最小化或最大化一个特定的损失函数。优化器的作用就是通过调整模型的参数,使得损失函数的值最小化或最大化。

常见的优化器有梯度下降法(Gradient Descent)、随机梯度下降法(Stochastic Gradient Descent)、动量法(Momentum)、Nesterov Accelerated Gradient(NAG)、Adagrad、RMSprop、Adam等。

常见的Pytorch模型有:

1.线性回归模型(Linear Regression Model):用于拟合线性关系数据的模型。

2.逻辑回归模型(Logistic Regression Model):用于分类问题的模型。

3.多层感知机模型(Multi-Layer Perceptron Model):由多个全连接层组成的深度神经网络模型。

4.卷积神经网络模型(Convolutional Neural Network Model):用于处理图像和视觉数据的模型。

5.循环神经网络模型(Recurrent Neural Network Model):用于处理序列数据的模型。

6.长短期记忆网络模型(Long Short-Term Memory Model):一种循环神经网络的变种,用于处理长序列数据的模型。

7.生成对抗网络模型(Generative Adversarial Network Model):由生成器和判别器组成的模型,用于生成新的数据样本。

8.注意力机制模型(Attention Mechanism Model):用于处理序列数据的模型,通过对输入序列的不同部分赋予不同的注意权重来提升模型性能。

9.Transformer模型:基于注意力机制的模型,用于处理序列数据的模型,如自然语言处理任务中的机器翻译和文本生成等。

模型保存

python 复制代码
vgg16=torchvision.models.vgg16(pretrained=False)

# 保存方式一,保存模型结构+模型参数
torch.save(vgg16,"vgg16_method1.pth")

# 保存方式二,保存模型参数  (推荐)
torch.save(vgg16.state_dict(),"vgg16_method2.pth")




# 保存方式一:加载模型
model=torch.load("vgg16_method1.pth")

# 保存方式二:加载模型
vgg16=torchvision.models.vgg16(pretrained=False)
vgg16.load_state_dict()
model=torch.load("vgg16_method2.pth")
相关推荐
亚马逊云开发者5 分钟前
Amazon OpenSearch 助力高效 RAG 系统落地
人工智能
suke6 分钟前
听说前端又死了?
前端·人工智能·程序员
小小小怪兽22 分钟前
关于氛围编程(Vibe Coding)的一切
人工智能·全栈
付玉祥29 分钟前
从谷歌白皮书看 Prompt 工程
人工智能
松岛雾奈.23030 分钟前
机器学习--数据集的标准化和归一化算法;随机森林
人工智能·算法·机器学习
阿明Drift33 分钟前
用 RAG 搭建一个 AI 小说问答系统
前端·人工智能
朱龙凯37 分钟前
LangChain学习笔记
人工智能
飞哥数智坊1 小时前
Cursor 2.1 发布实测:计划能点了,审查能用了,CR 花多少?我也替你试了
人工智能·ai编程·cursor
凯子坚持 c1 小时前
Doubao-Seed-Code模型深度剖析:Agentic Coding在Obsidian插件开发中的应用实践
网络·人工智能
CoovallyAIHub1 小时前
存储风暴下的边缘智能韧性:瑞芯微RK3588如何将供应链挑战转化为市场机遇
深度学习·算法·计算机视觉