用脚本下载数据集时,怎么更改数据集的下载位置啊?老是下载到C盘里面。
2条回答 默认 最新
关注以MNIST为例:
train_dataset = torchvision.datasets.MNIST(root='./data', train=True, transform=transforms.ToTensor(), download=True)把root改一下试试,
root: 存放数据集的路径
train:(bool,可选) 如果为true,则从训练集创建数据集,否则从测试集创建。
transform:数据预处理(数据增强)
target_transform:标注的预处理。
download:是否下载(如果在root已存在,就算为true也不下载。解决 无用评论 打赏 举报 编辑记录