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

时间:2015-07-08 22:23:16

标签: lua torch mnist

我正在尝试在LayoutInflater repl中加载MNIST数据集,并通过以下方式表示减法:

th

最后一行导致以下错误:

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

我在16GB内存的笔记本电脑上运行它。此外,MNIST已加载到.../torch/install/bin/luajit: not enough memory,因此不确定为什么执行data会导致此问题。有什么想法吗?

由于

1 个答案:

答案 0 :(得分:0)

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

这可以通过做到这一点来克服 data = data:add(-mean) 要么 data:add(-mean); - 注意分号

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