pytorch快速入门（四）Dataloader（）

半甜田田

于 2021-12-28 21:02:20 发布

阅读量907

点赞数 1

分类专栏：深度学习可视化 pytorch 文章标签： pytorch python 深度学习

本文链接：https://blog.csdn.net/rensweet/article/details/122198488

版权

本文介绍了PyTorch中Dataloader的使用，包括通过官方指南理解Dataloader，解决运行代码时的维度不匹配问题，以及如何利用Tensorboard进行可视化。在解决报错中，指出了输入张量格式应与要求的CHW格式一致。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、查看官方指南

pytorch官方网站在这里插入图片描述

二、代码运行

1、pycharm中新建python文件，输入以下代码

import torchvision
from  torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter

#准备的测试数据集
test_set = torchvision.datasets.CIFAR10

最低0.47元/天解锁文章