最近学深度学习,发现需要跑数据集,数据通常存在data文件夹下。本文主要帮助那些看完代码后,不能正确安放数据集路径的同学。
案例1:
train_dataset = torchvision.datasets.MNIST(root='../../data',
train=True,
transform=transforms.ToTensor(),
download=True)
root='../../data'不知道应该把数据集放在那里????
案例2:
train_dataset = dsets.MNIST(root='./data',
train=True,
transform=transforms.ToTensor(),
download=True)
root='./data',不知道把数据集放在那里????
解决方案:直接查看文件路径:
path1=os.path.abspath('.')
print(path1)
path2=os.path.abspath('../..')
print(path2)
运行结果:
E:\深度学习\深度学习框架Pytorch快速开发与实践\chapter5
E:\深度学习