摘要:
一、Dataset基类介绍 在torch中提供了数据集的基类torch.utils.data.Dataset,继承这个基类,可以快速实现对数据的加载 torch.utils.data.Dataset的源码如下: class Dataset(Generic[T_co]): r"""An abstrac 阅读全文
摘要:
假设我们的基础模型是y = wx + b,其中w和b均为参数,我们使用y = 3x + 0.8来构造数据x、y,所以最后通过模型应该能够看得出w和b分别接近3和0.8。 实现过程: 1、准备数据 2、计算预测值 3、计算损失,把参数的梯度置为0,进行反向传播 4、更新参数 代码示例: import 阅读全文