Skip to content

Latest commit

 

History

History
91 lines (44 loc) · 2.99 KB

【2022.11.03】torch的一些函数.md

File metadata and controls

91 lines (44 loc) · 2.99 KB

Pytorch的使用相关,学习来源:https://www.bilibili.com/video/BV1Wv411h7kN/?p=6

加载数据有两种方法,一种是torch.utils.data.Dataset,一种是torch.utils.data.Dataloader

image-20221102163836735

Tensor是高维阵列

image-20221102164238486

张量的转置 transpose

image-20221103084334835

张量的降维 squeeze

image-20221103084618182

张量的增维 unsqueeze

image-20221103085611851

张量的合并 cat

image-20221103085651765

张量的计算

image-20221103090217568

image-20221103110531031

使用Pytorch定义网络

定义输入规范

image-20221103110657022

网络参数

image-20221103111053157

image-20221103111109607

建立自己的神经网络

image-20221103111535729

image-20221103112207837

损失函数

image-20221103112327094

优化,反向传播,梯度下降,学习率

image-20221103112432636

LR是learning rate

image-20221103112456990

全局变量

定义文件来源

使用dataset用一个dataloader包起来,定义model,定义损失函数和优化函数

image-20221103112703101

训练模型

image-20221103130315952

image-20221103130448102

测试模型

image-20221103130616136

梯度计算

image-20221103130722723

保存模型

image-20221103130815604