pytorch文档学习——QUICKSTART

最新推荐文章于 2023-05-27 14:46:38 发布

知之之

最新推荐文章于 2023-05-27 14:46:38 发布

阅读量181

点赞数

分类专栏： pytorch

本文链接：https://blog.csdn.net/weixin_43963453/article/details/118069981

版权

2 篇文章 0 订阅

订阅专栏

本节代码

讲述了如何处理数据，两个最关键的部分：

torch.utils.data.DataLoader and torch.utils.data.Dataset
#前者存储，后者迭代

PyTorch 提供领域特定的库，如 TorchText、 TorchVision 和 TorchAudio，所有这些都包含数据集

Dataset有俩参数： transform 和 target _ transform，分别用于修改样本和标签
Dataloader的描述：wraps an iterable over our dataset, and supports automatic batching, sampling, shuffling and multiprocess data loading.

在__init__()中定义层
在forward（）中指定数据如何flow
这里的loss 和opt是分开的。正如花书中说的，backprob只是计算梯度而已，并没有优化功能。因此loss指定了损失函数是什么，然后能计算梯度，剩下的怎么优化交给优化器做。

关注