我如何处理 torchvision 数据集

Question

我正在使用名为 MNIST 的 torchvision 数据集。我运行在我的python项目中下载了代码。我需要删除我写的代码，还是我可以继续写我的人工智能？下载代码会干扰它还是重新下载整个数据集？

我用来下载的代码：

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 notebook，但我使用的是 PyCharm。我需要做什么？

Answer 1

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

我如何处理 torchvision 数据集

how do i deal with torchvision datasets

python

pytorch

dataset

torchvision