随笔分类 -  Pytorch

摘要:Pytorch的主要组成模块 一、基本配置 对于一个PyTorch项目,我们需要导入一些Python常用的包来帮助我们快速实现功能。常见的包有os、numpy等,此外还需要调用PyTorch自身一些模块便于灵活使用,比如torch、torch.nn、torch.utils.data.Dataset、 阅读全文
posted @ 2022-08-18 11:43 Xu_Lin 阅读(550) 评论(0) 推荐(0)
摘要:Utils 从torch.nn.utils模块 clip_grad_norm_ 裁剪参数迭代的梯度范数。 clip_grad_value_ 在指定值处剪辑可迭代参数的梯度。 parameters_to_vector 将参数转换为一个向量 vector_to_parameters 将一个向量转换为参数 阅读全文
posted @ 2022-03-23 20:58 Xu_Lin 阅读(508) 评论(0) 推荐(0)
摘要:卷积层 nn.Conv1d 对由多个输入平面组成的输入信号应用一维卷积。 nn.Conv2d 在由多个输入平面组成的输入信号上应用 2D 卷积。 nn.Conv3d 对由多个输入平面组成的输入信号应用 3D 卷积。 nn.ConvTranspose1d 在由多个输入平面组成的输入图像上应用一维转置卷 阅读全文
posted @ 2022-03-23 20:57 Xu_Lin 阅读(492) 评论(0) 推荐(0)
摘要:损失函数 nn.L1Loss 创建一个衡量输入中每个元素之间的平均绝对误差 (MAE) 的标准XX和目标是的是的. nn.MSELoss 创建一个标准,用于测量输入中每个元素之间的均方误差(平方 L2 范数)XX和目标是的是的. nn.CrossEntropyLoss 该标准计算输入和目标之间的交叉 阅读全文
posted @ 2022-03-23 20:56 Xu_Lin 阅读(250) 评论(0) 推荐(0)
摘要:使用Pytorch搭建模型的步骤及教程 我们知道,模型有一个特定的生命周期,了解这个为数据集建模和理解 PyTorch API 提供了指导方向。我们可以根据生命周期的每一个步骤进行设计和优化,同时更加方便调整各种细节。 模型的生命周期的五个步骤如下: 1.准备数据 2.定义模型 3.训练模型 4.评 阅读全文
posted @ 2021-10-14 11:22 Xu_Lin 阅读(1354) 评论(0) 推荐(1)