2015-10-22 3 views
9

Я работаю над проектом Torch/Lua, в котором я реализовал модель искусственной нейронной сети. Все работает, но теперь я хочу изменить свой код следующим образом. Поскольку мой входной набор данных очень велик, я бы хотел разделить его на N = 20 интервалов.Torch/Lua, как сохранить подготовленную модель нейронной сети в файл?

Затем я хочу обучить свою нейронную сеть только 1-м диапазоном набора данных, а затем проверить на других N-1 = 19 параллельных параллельно.

Чтобы выполнить все эти параллельные задания, мне нужно сохранить данные моей модели нейронной сети в файл, а затем загрузить их для каждых 19 заданий.

Есть ли способ в факеле правильно написать «искусственную модель нейронной сети»?

ответ

15
-- save the model 
torch.save(filename, model) 

.

-- load the model 
model = torch.load(filename) 
+0

Спасибо @smhx, проще, чем я думал! –

+0

Есть ли способ сохранить его и загрузить в Lua (без факела)? – Adam12344

+0

@ Adam12344 перед загрузкой сети обратно в факел, вам нужно «потребовать» пакеты, используемые в сериализованной модели, такие как 'nn', иначе загрузка не удастся. Следовательно, это, безусловно, также потерпит неудачу с «Lua (без факела)», если вы не можете загрузить, например. пакет 'nn' перед загрузкой модели. –

Смежные вопросы