时间戳

首页 > 解决方案 > 我如何处理 torchvision 数据集

python - 我如何处理 torchvision 数据集

问题描述

我正在使用一个名为 MNIST 的 torchvision 数据集。我在我的 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 notebook，但我正在使用 PyCharm。我需要做什么？

标签： pythonpytorchdatasettorchvision

解决方案

只需将其保留在那里，您需要它来定义数据集：该download=True标志表示数据集仅在指定文件夹中不存在时才被下载：https ://pytorch.org/docs/stable/torchvision /datasets.html#mnist

推荐阅读