【2022.11.03】pytorch的使用相关(一)

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

posted @ 2022-11-03 13:14  Mokou  阅读(43)  评论(0)  编辑  收藏  举报