在封闭系统中从本地目录导入 MNIST 数据集

如何解决在封闭系统中从本地目录导入 MNIST 数据集

我正在尝试基于集群中的 MNIST 数据运行教程,而运行训练脚本的节点无法访问互联网,因此我手动将 MNIST 数据集放置在所需目录中,但我得到了 {{1} } 我正在尝试在集群上运行 this 教程。 我已经尝试了 this 答案,但答案并没有解决我的问题。 下面是我的代码修改 -

Dataset not found error.

如何解决?

解决方法

您必须指定根文件夹,而不是已处理文件的完整路径:

root (string):数据集的根目录,其中 MNIST/processed/training.ptMNIST/processed/test.pt 存在。

就你而言:

root is /scratch/netra

因此,

train_dataset = \
datasets.MNIST('/scratch/netra-%d' % hvd.rank(),train=True,download=True,transform=transforms.Compose([
                   transforms.ToTensor(),transforms.Normalize((0.1307,),(0.3081,))
               ]))
test_dataset = \
datasets.MNIST('/scratch/netra-%d' % hvd.rank(),train=False,))
               ]))

,

如果上述方法不起作用,请尝试将这些 .pt 文件放在当前工作目录中名为 .data 的文件夹中:

import os
CURR_DIR = os.getcwd()
print(CURR_DIR)

train = datasets.MNIST(root='./data',download=False,transform=transforms.Compose([transforms.ToTensor(),))]))
# works

train = datasets.MNIST(root=CURR_DIR + '\\data',))]))
# works

# same files also in this folder
train = datasets.MNIST(root=CURR_DIR + '\\processed',))
               ]))
# Dataset not found

有趣的是,在最后一个示例中,这正是生成数据的 Torch MNIST 数据集类放置 .pt 文件的位置。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其他元素将获得点击?
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。)
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbcDriver发生异常。为什么?
这是用Java进行XML解析的最佳库。
Java的PriorityQueue的内置迭代器不会以任何特定顺序遍历数据结构。为什么?
如何在Java中聆听按键时移动图像。
Java“Program to an interface”。这是什么意思?
Java在半透明框架/面板/组件上重新绘画。
Java“ Class.forName()”和“ Class.forName()。newInstance()”之间有什么区别?
在此环境中不提供编译器。也许是在JRE而不是JDK上运行?
Java用相同的方法在一个类中实现两个接口。哪种接口方法被覆盖?
Java 什么是Runtime.getRuntime()。totalMemory()和freeMemory()?
java.library.path中的java.lang.UnsatisfiedLinkError否*****。dll
JavaFX“位置是必需的。” 即使在同一包装中
Java 导入两个具有相同名称的类。怎么处理?
Java 是否应该在HttpServletResponse.getOutputStream()/。getWriter()上调用.close()?
Java RegEx元字符(。)和普通点?