当前位置 : 主页 > 网络编程 > lua >

火炬/ Lua,如何保存训练有素的神经网络模型文件?

来源:互联网 收集:自由互联 发布时间:2021-06-23
我在一个Torch / Lua项目上工作,其中我实现了一个人工神经网络模型. 一切正常,但现在我想以下列方式修改我的代码.由于我的输入数据集非常大,我希望将其分为N = 20个跨度. 然后我想在第
我在一个Torch / Lua项目上工作,其中我实现了一个人工神经网络模型.
一切正常,但现在我想以下列方式修改我的代码.由于我的输入数据集非常大,我希望将其分为N = 20个跨度.

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

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

有没有办法在火炬正确“写”一个人造神经网络模型文件?

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

.

-- load the model
model = torch.load(filename)
网友评论