火炬:luajit 在简单任务上内存不足

Torch: luajit out of memory on simple task

我正在尝试在 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) 会导致此问题。有什么想法吗?

谢谢

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

这可以通过以下任一方式解决 data = data:add(-mean) 要么 data:add(-mean); - 注意分号

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