python 文件路径(pytorch 数据集)

最近学深度学习,发现需要跑数据集,数据通常存在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:\深度学习

 

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值