Torch / Lua,如何将训练好的神经网络模型保存到文件中?
Torch / Lua, how to save a trained neural network model to file?
我在 Torch / Lua 项目上工作,我在其中实现了一个人工神经网络模型。
一切正常,但现在我想按以下方式修改我的代码。由于我的输入数据集非常大,我想将它分成 N=20 个跨度。
然后我只想在第一个数据集跨度上训练我的神经网络,然后并行测试其他 N-1=19 个跨度。
为了 运行 所有这些并行作业,我需要将我的神经网络模型详细信息保存到一个文件中,然后为每 19 个作业加载它。
torch有没有办法正确"write"人工神经网络模型归档?
-- save the model
torch.save(filename, model)
.
-- load the model
model = torch.load(filename)
我在 Torch / Lua 项目上工作,我在其中实现了一个人工神经网络模型。 一切正常,但现在我想按以下方式修改我的代码。由于我的输入数据集非常大,我想将它分成 N=20 个跨度。
然后我只想在第一个数据集跨度上训练我的神经网络,然后并行测试其他 N-1=19 个跨度。
为了 运行 所有这些并行作业,我需要将我的神经网络模型详细信息保存到一个文件中,然后为每 19 个作业加载它。
torch有没有办法正确"write"人工神经网络模型归档?
-- save the model
torch.save(filename, model)
.
-- load the model
model = torch.load(filename)