Torch / Lua,如何将经过训练的神经网络模型保存到文件中?

时间:2015-10-22 18:21:15

标签: lua parallel-processing torch

我正在研究一个Torch / Lua项目,其中我实现了一个人工神经网络模型。 一切正常,但现在我想以下列方式修改我的代码。由于我的输入数据集非常大,我想将它除以N = 20个跨度。

然后我想仅在第一个数据集范围内训练我的神经网络,然后并行测试其他N-1 = 19个跨度。

要运行所有这些并行作业,我需要将神经网络模型详细信息保存到文件中,然后为每19个作业加载它。

火炬有没有正确的方法"写"一个人工神经网络模型来提交?

1 个答案:

答案 0 :(得分:15)

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

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