Torch: 在简单任务中出现了 luajit 内存不足的错误。

我尝试在 th repl 中加载 MNIST 数据集并执行以下平均值减法:

file = torch.load('data/mnist.t7/train_32x32.t7', 'ascii')
data = file.data:type(torch.getdefaulttensortype())
mean = data:mean()
data:add(-mean)

最后一行会导致以下错误:

.../torch/install/bin/luajit: not enough memory

我在一台配备 16GB RAM 的笔记本电脑上运行此命令。MNIST 已加载到 data 中,因此不确定为什么执行 data:add(-mean) 会导致此问题。有什么想法吗?

谢谢

点赞
用户225814
用户225814

问题在于它试图将整个矩阵(很大)打印到控制台。

可以通过以下方式解决: data = data:add(-mean) 或者 data:add(-mean); - 注意分号

Soumith Chintala 在 torch gitter 上提供了答案。

2015-07-09 14:27:53