torchvision数据集使用

一、下载torchvision中的数据集文件

这段代码是使用PyTorch的torchvision库来加载CIFAR-10数据集。

import torchvision
train_set = torchvision.datasets.CIFAR10(root="./dataset",train=True,download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset",train=False,download=True)

root指在什么位置，train的True表示创建一个训练集，为False表示创建一个测试集，download为True的话则是直接下载
./dataset表示的是相对路径，把数据保存进名为Dataset中

回车后得到:

复制蓝色的链接，还可以在迅雷中添加链接进行下载，这样子下载的速度可能相对较快。

在下载好了文件包，可以在pycharm文件中见到Dataset文件包

二、断点知识点

• 只要我们在代码行的最左侧点击一下鼠标左键，就完成设置断点

• 设置好断点后我们可以进入调试模式
• 调试模式不同于运行模式，如果进行代码运行那么断点就依然忽略不计
  鼠标右键点击一下，可以看到一只瓢虫，点击就是进行调试