1.CIFAR-10数据集介绍
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/b657c638e3460cc7d95ac1de7996db03.png)
2.torchvision内含数据集的下载
from torchvision import datasets
from torchvision import transforms
from torch.utils.tensorboard import SummaryWriter
compose = transforms.Compose([
transforms.ToTensor()
])
train_dataset = datasets.CIFAR10(root="./datasets", train=True, transform=compose, download=True)
test_dataset = datasets.CIFAR10(root="./datasets", train=False, transform=compose, download=True)
writer = SummaryWriter("data")
for i in range(100):
img_tensor, label = train_dataset[i]
target = train_dataset.classes[label]
writer.add_image(tag="CIFAR-10", img_tensor=img_tensor, global_step=i)
writer.close()
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/5fc0b8db21ece709479775ee9e7d1254.png)