我正在尝试在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
会导致此问题。有什么想法吗?
由于
答案 0 :(得分:0)
问题是它试图将整个矩阵(很大)打印到控制台。
这可以通过做到这一点来克服
data = data:add(-mean)
要么
data:add(-mean);
- 注意分号
Soumith Chintala在火炬手上提供的答案。