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

Я пытаюсь загрузить набор данных 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) вызовет эту проблему. Есть идеи?

Спасибо

1 2
lua torch

1 ответ:

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

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

Ответ, предоставленный Соумитом Чинталой на факеле Гиттера.

0