python 文件路径（pytorch 数据集）

2024-07-07 10:43| 来源: 网络整理| 查看: 265

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

【本文地址】

公司简介

联系我们