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



我在做一个Torch/Lua项目,在这个项目中我实现了一个人工神经网络模型。一切工作,但现在我想修改我的代码在以下方式。因为我的输入数据集非常大,我想把它分成N=20个跨度。

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

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

有没有办法在火炬正确地"写"人工神经网络模型的文件?

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

.

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

最新更新