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

Dav*_*.it 9 parallel-processing lua torch

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

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

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

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

小智 15

-- save the model
torch.save(filename, model)
Run Code Online (Sandbox Code Playgroud)

.

-- load the model
model = torch.load(filename)
Run Code Online (Sandbox Code Playgroud)