如何将训练完成的神经网络模型以文件形式保存至火炬Lua?
- 内容介绍
- 文章标签
- 相关推荐
本文共计268个文字,预计阅读时间需要2分钟。
我在一个Torch/Lua项目上工作,其中实现了一个人工神经网络模型。一直以来,我想以下列方式修改我的代码。由于我的输入数据集非常大,我希望将其分为N=20个跨度。然后,我想在第一个________。
一切正常,但现在我想以下列方式修改我的代码.由于我的输入数据集非常大,我希望将其分为N = 20个跨度.
然后我想在第一个数据集跨度上训练我的神经网络,然后并行测试另一个N-1 = 19个跨度.
要运行所有这些并行作业,我需要将我的神经网络模型细节保存到一个文件中,然后为每19个作业加载它.
有没有办法在火炬正确“写”一个人造神经网络模型文件?
-- save the model torch.save(filename, model)
.
-- load the model model = torch.load(filename)
本文共计268个文字,预计阅读时间需要2分钟。
我在一个Torch/Lua项目上工作,其中实现了一个人工神经网络模型。一直以来,我想以下列方式修改我的代码。由于我的输入数据集非常大,我希望将其分为N=20个跨度。然后,我想在第一个________。
一切正常,但现在我想以下列方式修改我的代码.由于我的输入数据集非常大,我希望将其分为N = 20个跨度.
然后我想在第一个数据集跨度上训练我的神经网络,然后并行测试另一个N-1 = 19个跨度.
要运行所有这些并行作业,我需要将我的神经网络模型细节保存到一个文件中,然后为每19个作业加载它.
有没有办法在火炬正确“写”一个人造神经网络模型文件?
-- save the model torch.save(filename, model)
.
-- load the model model = torch.load(filename)

