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

发布时间：2020-12-15 00:27:03 所属栏目：大数据来源：网络整理

导读：我在一个Torch / Lua项目上工作,其中我实现了一个人工神经网络模型. 一切正常,但现在我想以下列方式修改我的代码.由于我的输入数据集非常大,我希望将其分为N = 20个跨度. 然后我想在第一个数据集跨度上训练我的神经网络,然后并行测试另一个N-1 = 19个跨度.

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

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

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

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

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

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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!