Torch / Lua，如何将训练好的神经网络模型保存到文件中？

Question

我正在开发一个Torch/Lua项目，其中我实现了一个人工神经网络模型。一切都运行良好，但现在我想按以下方式修改我的代码。由于我的输入数据集非常大，我想将其分成N=20个片段。

然后，我想仅在第一个数据集片段上训练我的神经网络，然后在其他N-1=19个片段上并行测试。

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

在Torch中有没有正确地将人工神经网络模型“写入”文件的方法？

- DavideChicco.it

Answer 1

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

.

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

- smhx

---

谢谢@smhx，比我想象的要容易！ - DavideChicco.it

有没有一种方法可以将它保存并加载到Lua中（不使用Torch）？ - Adam12344

在将网络加载回torch之前，您需要“require”序列化模型中使用的软件包，例如“nn”，否则加载将失败。因此，如果在加载模型之前无法加载“nn”软件包，则“Lua（不带Torch）”也肯定会失败。 - Andre Holzner