摘要: 一、简介 pytorch自带的数据集由两个上层api提供,分别是torchvision和torchtext,其中: 1、torchvision提供了对图片数据处理相关的api和数据 数据位置:torchvision.datasets,例如:torchvision.datasets.MNIST(手写数 阅读全文
posted @ 2024-02-04 11:49 ziff123 阅读(75) 评论(0) 推荐(0) 编辑
摘要: 一、Dataset基类介绍 在torch中提供了数据集的基类torch.utils.data.Dataset,继承这个基类,可以快速实现对数据的加载 torch.utils.data.Dataset的源码如下: class Dataset(Generic[T_co]): r"""An abstrac 阅读全文
posted @ 2024-02-04 10:55 ziff123 阅读(43) 评论(0) 推荐(0) 编辑
摘要: 假设我们的基础模型是y = wx + b,其中w和b均为参数,我们使用y = 3x + 0.8来构造数据x、y,所以最后通过模型应该能够看得出w和b分别接近3和0.8。 实现过程: 1、准备数据 2、计算预测值 3、计算损失,把参数的梯度置为0,进行反向传播 4、更新参数 代码示例: import 阅读全文
posted @ 2024-02-04 10:00 ziff123 阅读(17) 评论(0) 推荐(0) 编辑