lua - 火炬：简单任务的luajit内存不足

Question

我正在尝试在threpl 中加载 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)会导致这个问题。有任何想法吗？

谢谢

Answer 1

问题是它试图将整个矩阵（很大）打印到控制台。

这可以通过执行 data = data:add(-mean) 或 data:add(-mean);解决 - 注意分号

Soumith Chintala 在火炬手上提供的答案。