我正在研究一个Torch / Lua项目,其中我实现了一个人工神经网络模型。 一切正常,但现在我想以下列方式修改我的代码。由于我的输入数据集非常大,我想将它除以N = 20个跨度。
然后我想仅在第一个数据集范围内训练我的神经网络,然后并行测试其他N-1 = 19个跨度。
要运行所有这些并行作业,我需要将神经网络模型详细信息保存到文件中,然后为每19个作业加载它。
火炬有没有正确的方法"写"一个人工神经网络模型来提交?
答案 0 :(得分:15)
-- save the model
torch.save(filename, model)
-- load the model
model = torch.load(filename)