我在一个Torch / Lua项目上工作,其中我实现了一个人工神经网络模型.
一切正常,但现在我想以下列方式修改我的代码.由于我的输入数据集非常大,我希望将其分为N = 20个跨度.
一切正常,但现在我想以下列方式修改我的代码.由于我的输入数据集非常大,我希望将其分为N = 20个跨度.
然后我想在第一个数据集跨度上训练我的神经网络,然后并行测试另一个N-1 = 19个跨度.
要运行所有这些并行作业,我需要将我的神经网络模型细节保存到一个文件中,然后为每19个作业加载它.
有没有办法在火炬正确“写”一个人造神经网络模型文件？
解决方法
-- save the model torch.save(filename,model)
.
-- load the model model = torch.load(filename)