火炬: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 上提供的答案。
我正在尝试在 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 上提供的答案。