Torch / Lua, как сохранить обученную модель нейронной сети в файл?

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

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

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

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

1 ответов


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

.

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