如何处理torchvision数据集？

Question

我正在使用一个名为MNIST的火炬视觉数据集。我在python项目中运行代码来下载它。我是否需要删除我编写的代码，还是可以继续编写我的ai？下载代码会干扰它还是重新下载整个数据集？

我用来下载它的代码：

train = datasets.MNIST("", train=True, download=True,
                       transform = transforms.Compose([transforms.ToTensor()]))

test = datasets.MNIST("", train=False, download=True,
                       transform = transforms.Compose([transforms.ToTensor()]))

教程中的那个家伙使用的是jupyter笔记本，但我使用的是PyCharm。我该怎么办？

Answer 1

只需将其保留在那里，就需要定义数据集：download=True标志意味着只有在指定文件夹：https://pytorch.org/docs/stable/torchvision/datasets.html#mnist中不存在数据集时，才会下载数据集