pytorch学习笔记（四）---torchvision中数据集以及DataLoader

最新推荐文章于 2024-05-04 11:15:00 发布

橘春十三99

最新推荐文章于 2024-05-04 11:15:00 发布

阅读量1.6k

点赞数 1

分类专栏： pytorch 学习笔记文章标签： python pytorch 深度学习

本文链接：https://blog.csdn.net/qq_41858089/article/details/121502660

版权

本篇自学笔记来自于b站《PyTorch深度学习快速入门教程（绝对通俗易懂！）【小土堆】》，Up主讲的非常通俗易懂，文章下方有视频连接，如有需要可移步up主讲解视频，如有侵权，实非故意，深表歉意，请与我联系，删除相关内容！

首先是torchvision中数据集dataset的使用，如图所示为pytorch官网，展示了torchvision.datasets中的一些现有数据集。

以其中的CIFAR10数据集为例，展示一下他的使用方式。下图所示为官方文档所展示的参数以及具体的参数解释。

给出完整代码：

#展示使用torchvision中的datasets的使用，以其中的CIFAR为例
import torchvision
from torch.utils.tensorboard import SummaryWriter
from torchvision.transforms import tr

关注