Факел: луаджит нехватка памяти при простом задании

Я пытаюсь загрузить набор данных MNIST в th repl и имею в виду вычитание следующим образом:

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

Я запускаю это на ноутбуке с 16 ГБ оперативной памяти. Кроме того, MNIST уже загружен в data, поэтому непонятно, почему выполнение data:add(-mean) может вызвать эту проблему. Любые идеи?

Спасибо


torch mnist lua
person Aly    schedule 08.07.2015    source источник

Ответы (1)


arrow_upward
0
arrow_downward

Проблема заключалась в том, что он пытался вывести всю матрицу (большую) на консоль.

Это можно преодолеть, выполнив data = data:add(-mean) или data:add(-mean); — обратите внимание на точку с запятой.

Ответ предоставлен Soumith Chintala о факеле.

person Aly    schedule 09.07.2015