如何将训练好的神经网络模型保存到文件中?在Torch/Lua中如何操作?

我正在开展一个 Torch / Lua 项目,其中我实现了一个人工神经网络模型。一切正常,但现在我想以以下方式修改代码。由于我的输入数据集非常大,我想将其分为 N=20 个区间。然后,我想仅在第一个数据集区间上训练神经网络,然后在其它 N-1=19 个区间上并行测试。

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

在 Torch 中有没有一种正确的方法“写”一个人工神经网络模型到文件中?

点赞
用户117844
用户117844
<!--保存模型-->
torch.save(filename, model)
<!--加载模型-->
model = torch.load(filename)
2015-10-22 20:04:16