我从事一个 Torch / Lua 项目,在其中实现了一个人工神经网络模型。
一切正常,但现在我想按以下方式修改我的代码。由于我的输入数据集非常大,我想将其划分为 N=20 个跨度。
然后我只想在第一个数据集跨度上训练我的神经网络,然后并行测试其他 N-1=19 跨度。
为了运行所有这些并行作业,我需要将神经网络模型详细信息保存到一个文件中,然后为每 19 个作业加载它。
torch 有什么方法可以正确地将人工神经网络模型“写入”文件吗?
-- save the model
torch.save(filename, model)
.
-- load the model
model = torch.load(filename)
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)